[
  {
    "path": ".dockerignore",
    "content": ".bin/\n.build/\ncadence\ncadence-server\ncadence-canary\ncadence-bench\ncadence-sql-tool\ncadence-cassandra-tool\nvendor/\n"
  },
  {
    "path": ".envrc",
    "content": "# Prepend project binary directories to the PATH\nPATH_add .bin\nPATH_add .build/bin\n"
  },
  {
    "path": ".fossa.yml",
    "content": "version: 2\ncli:\n  server: https://app.fossa.com\n  fetcher: custom\n  project: git@github.com:uber/cadence.git\nanalyze:\n  modules:\n  - name: github.com/uber/cadence/cmd/canary\n    type: go\n    target: github.com/uber/cadence/cmd/canary\n    path: cmd/canary\n  - name: github.com/uber/cadence/cmd/server\n    type: go\n    target: github.com/uber/cadence/cmd/server\n    path: cmd/server\n  - name: github.com/uber/cadence/cmd/tools/cassandra\n    type: go\n    target: github.com/uber/cadence/cmd/tools/cassandra\n    path: cmd/tools/cassandra\n  - name: github.com/uber/cadence/cmd/tools/cli\n    type: go\n    target: github.com/uber/cadence/cmd/tools/cli\n    path: cmd/tools/cli\n  - name: github.com/uber/cadence/cmd/tools/copyright\n    type: go\n    target: github.com/uber/cadence/cmd/tools/copyright\n    path: cmd/tools/copyright\n  - name: github.com/uber/cadence/cmd/tools/sql\n    type: go\n    target: github.com/uber/cadence/cmd/tools/sql\n    path: cmd/tools/sql\n"
  },
  {
    "path": ".gen/go/admin/admin.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage admin\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tconfig \"github.com/uber/cadence/.gen/go/config\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype AddSearchAttributeRequest struct {\n\tSearchAttribute map[string]shared.IndexedValueType `json:\"searchAttribute,omitempty\"`\n\tSecurityToken   *string                            `json:\"securityToken,omitempty\"`\n}\n\ntype _Map_String_IndexedValueType_MapItemList map[string]shared.IndexedValueType\n\nfunc (m _Map_String_IndexedValueType_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_IndexedValueType_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) ValueType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) Close() {}\n\n// ToWire translates a AddSearchAttributeRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AddSearchAttributeRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SearchAttribute != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_IndexedValueType_MapItemList(v.SearchAttribute)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tw, err = wire.NewValueString(*(v.SecurityToken)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IndexedValueType_Read(w wire.Value) (shared.IndexedValueType, error) {\n\tvar v shared.IndexedValueType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Map_String_IndexedValueType_Read(m wire.MapItemList) (map[string]shared.IndexedValueType, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]shared.IndexedValueType, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _IndexedValueType_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a AddSearchAttributeRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AddSearchAttributeRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AddSearchAttributeRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AddSearchAttributeRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.SearchAttribute, err = _Map_String_IndexedValueType_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SecurityToken = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_IndexedValueType_Encode(val map[string]shared.IndexedValueType, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TI32,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a AddSearchAttributeRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AddSearchAttributeRequest struct could not be encoded.\nfunc (v *AddSearchAttributeRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SearchAttribute != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_IndexedValueType_Encode(v.SearchAttribute, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SecurityToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SecurityToken)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IndexedValueType_Decode(sr stream.Reader) (shared.IndexedValueType, error) {\n\tvar v shared.IndexedValueType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Map_String_IndexedValueType_Decode(sr stream.Reader) (map[string]shared.IndexedValueType, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TI32 {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]shared.IndexedValueType, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _IndexedValueType_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a AddSearchAttributeRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AddSearchAttributeRequest struct could not be generated from the wire\n// representation.\nfunc (v *AddSearchAttributeRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.SearchAttribute, err = _Map_String_IndexedValueType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SecurityToken = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AddSearchAttributeRequest\n// struct.\nfunc (v *AddSearchAttributeRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.SearchAttribute != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttribute: %v\", v.SearchAttribute)\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"SecurityToken: %v\", *(v.SecurityToken))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AddSearchAttributeRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_IndexedValueType_Equals(lhs, rhs map[string]shared.IndexedValueType) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this AddSearchAttributeRequest match the\n// provided AddSearchAttributeRequest.\n//\n// This function performs a deep comparison.\nfunc (v *AddSearchAttributeRequest) Equals(rhs *AddSearchAttributeRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SearchAttribute == nil && rhs.SearchAttribute == nil) || (v.SearchAttribute != nil && rhs.SearchAttribute != nil && _Map_String_IndexedValueType_Equals(v.SearchAttribute, rhs.SearchAttribute))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SecurityToken, rhs.SecurityToken) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_IndexedValueType_Zapper map[string]shared.IndexedValueType\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_IndexedValueType_Zapper.\nfunc (m _Map_String_IndexedValueType_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AddSearchAttributeRequest.\nfunc (v *AddSearchAttributeRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SearchAttribute != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttribute\", (_Map_String_IndexedValueType_Zapper)(v.SearchAttribute)))\n\t}\n\tif v.SecurityToken != nil {\n\t\tenc.AddString(\"securityToken\", *v.SecurityToken)\n\t}\n\treturn err\n}\n\n// GetSearchAttribute returns the value of SearchAttribute if it is set or its\n// zero value if it is unset.\nfunc (v *AddSearchAttributeRequest) GetSearchAttribute() (o map[string]shared.IndexedValueType) {\n\tif v != nil && v.SearchAttribute != nil {\n\t\treturn v.SearchAttribute\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttribute returns true if SearchAttribute is not nil.\nfunc (v *AddSearchAttributeRequest) IsSetSearchAttribute() bool {\n\treturn v != nil && v.SearchAttribute != nil\n}\n\n// GetSecurityToken returns the value of SecurityToken if it is set or its\n// zero value if it is unset.\nfunc (v *AddSearchAttributeRequest) GetSecurityToken() (o string) {\n\tif v != nil && v.SecurityToken != nil {\n\t\treturn *v.SecurityToken\n\t}\n\n\treturn\n}\n\n// IsSetSecurityToken returns true if SecurityToken is not nil.\nfunc (v *AddSearchAttributeRequest) IsSetSecurityToken() bool {\n\treturn v != nil && v.SecurityToken != nil\n}\n\ntype AdminDeleteWorkflowRequest struct {\n\tDomain    *string                   `json:\"domain,omitempty\"`\n\tExecution *shared.WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a AdminDeleteWorkflowRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminDeleteWorkflowRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecution_Read(w wire.Value) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminDeleteWorkflowRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminDeleteWorkflowRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminDeleteWorkflowRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminDeleteWorkflowRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminDeleteWorkflowRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminDeleteWorkflowRequest struct could not be encoded.\nfunc (v *AdminDeleteWorkflowRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecution_Decode(sr stream.Reader) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminDeleteWorkflowRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminDeleteWorkflowRequest struct could not be generated from the wire\n// representation.\nfunc (v *AdminDeleteWorkflowRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminDeleteWorkflowRequest\n// struct.\nfunc (v *AdminDeleteWorkflowRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminDeleteWorkflowRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminDeleteWorkflowRequest match the\n// provided AdminDeleteWorkflowRequest.\n//\n// This function performs a deep comparison.\nfunc (v *AdminDeleteWorkflowRequest) Equals(rhs *AdminDeleteWorkflowRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminDeleteWorkflowRequest.\nfunc (v *AdminDeleteWorkflowRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *AdminDeleteWorkflowRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *AdminDeleteWorkflowRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *AdminDeleteWorkflowRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *AdminDeleteWorkflowRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype AdminDeleteWorkflowResponse struct {\n\tHistoryDeleted    *bool `json:\"historyDeleted,omitempty\"`\n\tExecutionsDeleted *bool `json:\"executionsDeleted,omitempty\"`\n\tVisibilityDeleted *bool `json:\"visibilityDeleted,omitempty\"`\n}\n\n// ToWire translates a AdminDeleteWorkflowResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminDeleteWorkflowResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HistoryDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.HistoryDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.ExecutionsDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.VisibilityDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminDeleteWorkflowResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminDeleteWorkflowResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminDeleteWorkflowResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminDeleteWorkflowResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.HistoryDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ExecutionsDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.VisibilityDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminDeleteWorkflowResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminDeleteWorkflowResponse struct could not be encoded.\nfunc (v *AdminDeleteWorkflowResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HistoryDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.HistoryDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionsDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ExecutionsDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.VisibilityDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminDeleteWorkflowResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminDeleteWorkflowResponse struct could not be generated from the wire\n// representation.\nfunc (v *AdminDeleteWorkflowResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.HistoryDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ExecutionsDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.VisibilityDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminDeleteWorkflowResponse\n// struct.\nfunc (v *AdminDeleteWorkflowResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.HistoryDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryDeleted: %v\", *(v.HistoryDeleted))\n\t\ti++\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionsDeleted: %v\", *(v.ExecutionsDeleted))\n\t\ti++\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityDeleted: %v\", *(v.VisibilityDeleted))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminDeleteWorkflowResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this AdminDeleteWorkflowResponse match the\n// provided AdminDeleteWorkflowResponse.\n//\n// This function performs a deep comparison.\nfunc (v *AdminDeleteWorkflowResponse) Equals(rhs *AdminDeleteWorkflowResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.HistoryDeleted, rhs.HistoryDeleted) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ExecutionsDeleted, rhs.ExecutionsDeleted) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.VisibilityDeleted, rhs.VisibilityDeleted) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminDeleteWorkflowResponse.\nfunc (v *AdminDeleteWorkflowResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HistoryDeleted != nil {\n\t\tenc.AddBool(\"historyDeleted\", *v.HistoryDeleted)\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tenc.AddBool(\"executionsDeleted\", *v.ExecutionsDeleted)\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tenc.AddBool(\"visibilityDeleted\", *v.VisibilityDeleted)\n\t}\n\treturn err\n}\n\n// GetHistoryDeleted returns the value of HistoryDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminDeleteWorkflowResponse) GetHistoryDeleted() (o bool) {\n\tif v != nil && v.HistoryDeleted != nil {\n\t\treturn *v.HistoryDeleted\n\t}\n\n\treturn\n}\n\n// IsSetHistoryDeleted returns true if HistoryDeleted is not nil.\nfunc (v *AdminDeleteWorkflowResponse) IsSetHistoryDeleted() bool {\n\treturn v != nil && v.HistoryDeleted != nil\n}\n\n// GetExecutionsDeleted returns the value of ExecutionsDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminDeleteWorkflowResponse) GetExecutionsDeleted() (o bool) {\n\tif v != nil && v.ExecutionsDeleted != nil {\n\t\treturn *v.ExecutionsDeleted\n\t}\n\n\treturn\n}\n\n// IsSetExecutionsDeleted returns true if ExecutionsDeleted is not nil.\nfunc (v *AdminDeleteWorkflowResponse) IsSetExecutionsDeleted() bool {\n\treturn v != nil && v.ExecutionsDeleted != nil\n}\n\n// GetVisibilityDeleted returns the value of VisibilityDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminDeleteWorkflowResponse) GetVisibilityDeleted() (o bool) {\n\tif v != nil && v.VisibilityDeleted != nil {\n\t\treturn *v.VisibilityDeleted\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityDeleted returns true if VisibilityDeleted is not nil.\nfunc (v *AdminDeleteWorkflowResponse) IsSetVisibilityDeleted() bool {\n\treturn v != nil && v.VisibilityDeleted != nil\n}\n\ntype AdminMaintainWorkflowRequest struct {\n\tDomain    *string                   `json:\"domain,omitempty\"`\n\tExecution *shared.WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a AdminMaintainWorkflowRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminMaintainWorkflowRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminMaintainWorkflowRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminMaintainWorkflowRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminMaintainWorkflowRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminMaintainWorkflowRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminMaintainWorkflowRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminMaintainWorkflowRequest struct could not be encoded.\nfunc (v *AdminMaintainWorkflowRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminMaintainWorkflowRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminMaintainWorkflowRequest struct could not be generated from the wire\n// representation.\nfunc (v *AdminMaintainWorkflowRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminMaintainWorkflowRequest\n// struct.\nfunc (v *AdminMaintainWorkflowRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminMaintainWorkflowRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminMaintainWorkflowRequest match the\n// provided AdminMaintainWorkflowRequest.\n//\n// This function performs a deep comparison.\nfunc (v *AdminMaintainWorkflowRequest) Equals(rhs *AdminMaintainWorkflowRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminMaintainWorkflowRequest.\nfunc (v *AdminMaintainWorkflowRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *AdminMaintainWorkflowRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *AdminMaintainWorkflowRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *AdminMaintainWorkflowRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *AdminMaintainWorkflowRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype AdminMaintainWorkflowResponse struct {\n\tHistoryDeleted    *bool `json:\"historyDeleted,omitempty\"`\n\tExecutionsDeleted *bool `json:\"executionsDeleted,omitempty\"`\n\tVisibilityDeleted *bool `json:\"visibilityDeleted,omitempty\"`\n}\n\n// ToWire translates a AdminMaintainWorkflowResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminMaintainWorkflowResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HistoryDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.HistoryDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.ExecutionsDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tw, err = wire.NewValueBool(*(v.VisibilityDeleted)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminMaintainWorkflowResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminMaintainWorkflowResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminMaintainWorkflowResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminMaintainWorkflowResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.HistoryDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ExecutionsDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.VisibilityDeleted = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminMaintainWorkflowResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminMaintainWorkflowResponse struct could not be encoded.\nfunc (v *AdminMaintainWorkflowResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HistoryDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.HistoryDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionsDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ExecutionsDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityDeleted != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.VisibilityDeleted)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminMaintainWorkflowResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminMaintainWorkflowResponse struct could not be generated from the wire\n// representation.\nfunc (v *AdminMaintainWorkflowResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.HistoryDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ExecutionsDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.VisibilityDeleted = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminMaintainWorkflowResponse\n// struct.\nfunc (v *AdminMaintainWorkflowResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.HistoryDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryDeleted: %v\", *(v.HistoryDeleted))\n\t\ti++\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionsDeleted: %v\", *(v.ExecutionsDeleted))\n\t\ti++\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityDeleted: %v\", *(v.VisibilityDeleted))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminMaintainWorkflowResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminMaintainWorkflowResponse match the\n// provided AdminMaintainWorkflowResponse.\n//\n// This function performs a deep comparison.\nfunc (v *AdminMaintainWorkflowResponse) Equals(rhs *AdminMaintainWorkflowResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.HistoryDeleted, rhs.HistoryDeleted) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ExecutionsDeleted, rhs.ExecutionsDeleted) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.VisibilityDeleted, rhs.VisibilityDeleted) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminMaintainWorkflowResponse.\nfunc (v *AdminMaintainWorkflowResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HistoryDeleted != nil {\n\t\tenc.AddBool(\"historyDeleted\", *v.HistoryDeleted)\n\t}\n\tif v.ExecutionsDeleted != nil {\n\t\tenc.AddBool(\"executionsDeleted\", *v.ExecutionsDeleted)\n\t}\n\tif v.VisibilityDeleted != nil {\n\t\tenc.AddBool(\"visibilityDeleted\", *v.VisibilityDeleted)\n\t}\n\treturn err\n}\n\n// GetHistoryDeleted returns the value of HistoryDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminMaintainWorkflowResponse) GetHistoryDeleted() (o bool) {\n\tif v != nil && v.HistoryDeleted != nil {\n\t\treturn *v.HistoryDeleted\n\t}\n\n\treturn\n}\n\n// IsSetHistoryDeleted returns true if HistoryDeleted is not nil.\nfunc (v *AdminMaintainWorkflowResponse) IsSetHistoryDeleted() bool {\n\treturn v != nil && v.HistoryDeleted != nil\n}\n\n// GetExecutionsDeleted returns the value of ExecutionsDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminMaintainWorkflowResponse) GetExecutionsDeleted() (o bool) {\n\tif v != nil && v.ExecutionsDeleted != nil {\n\t\treturn *v.ExecutionsDeleted\n\t}\n\n\treturn\n}\n\n// IsSetExecutionsDeleted returns true if ExecutionsDeleted is not nil.\nfunc (v *AdminMaintainWorkflowResponse) IsSetExecutionsDeleted() bool {\n\treturn v != nil && v.ExecutionsDeleted != nil\n}\n\n// GetVisibilityDeleted returns the value of VisibilityDeleted if it is set or its\n// zero value if it is unset.\nfunc (v *AdminMaintainWorkflowResponse) GetVisibilityDeleted() (o bool) {\n\tif v != nil && v.VisibilityDeleted != nil {\n\t\treturn *v.VisibilityDeleted\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityDeleted returns true if VisibilityDeleted is not nil.\nfunc (v *AdminMaintainWorkflowResponse) IsSetVisibilityDeleted() bool {\n\treturn v != nil && v.VisibilityDeleted != nil\n}\n\ntype DescribeClusterResponse struct {\n\tSupportedClientVersions *shared.SupportedClientVersions `json:\"supportedClientVersions,omitempty\"`\n\tMembershipInfo          *MembershipInfo                 `json:\"membershipInfo,omitempty\"`\n\tPersistenceInfo         map[string]*PersistenceInfo     `json:\"persistenceInfo,omitempty\"`\n}\n\ntype _Map_String_PersistenceInfo_MapItemList map[string]*PersistenceInfo\n\nfunc (m _Map_String_PersistenceInfo_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*PersistenceInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_PersistenceInfo_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_PersistenceInfo_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_PersistenceInfo_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_PersistenceInfo_MapItemList) Close() {}\n\n// ToWire translates a DescribeClusterResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeClusterResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SupportedClientVersions != nil {\n\t\tw, err = v.SupportedClientVersions.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.MembershipInfo != nil {\n\t\tw, err = v.MembershipInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.PersistenceInfo != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_PersistenceInfo_MapItemList(v.PersistenceInfo)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SupportedClientVersions_Read(w wire.Value) (*shared.SupportedClientVersions, error) {\n\tvar v shared.SupportedClientVersions\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _MembershipInfo_Read(w wire.Value) (*MembershipInfo, error) {\n\tvar v MembershipInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _PersistenceInfo_Read(w wire.Value) (*PersistenceInfo, error) {\n\tvar v PersistenceInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_PersistenceInfo_Read(m wire.MapItemList) (map[string]*PersistenceInfo, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*PersistenceInfo, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _PersistenceInfo_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DescribeClusterResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeClusterResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeClusterResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeClusterResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SupportedClientVersions, err = _SupportedClientVersions_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.MembershipInfo, err = _MembershipInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PersistenceInfo, err = _Map_String_PersistenceInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_PersistenceInfo_Encode(val map[string]*PersistenceInfo, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*PersistenceInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a DescribeClusterResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeClusterResponse struct could not be encoded.\nfunc (v *DescribeClusterResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SupportedClientVersions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SupportedClientVersions.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MembershipInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.MembershipInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PersistenceInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_PersistenceInfo_Encode(v.PersistenceInfo, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SupportedClientVersions_Decode(sr stream.Reader) (*shared.SupportedClientVersions, error) {\n\tvar v shared.SupportedClientVersions\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _MembershipInfo_Decode(sr stream.Reader) (*MembershipInfo, error) {\n\tvar v MembershipInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _PersistenceInfo_Decode(sr stream.Reader) (*PersistenceInfo, error) {\n\tvar v PersistenceInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_PersistenceInfo_Decode(sr stream.Reader) (map[string]*PersistenceInfo, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*PersistenceInfo, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _PersistenceInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DescribeClusterResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeClusterResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeClusterResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.SupportedClientVersions, err = _SupportedClientVersions_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.MembershipInfo, err = _MembershipInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TMap:\n\t\t\tv.PersistenceInfo, err = _Map_String_PersistenceInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeClusterResponse\n// struct.\nfunc (v *DescribeClusterResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.SupportedClientVersions != nil {\n\t\tfields[i] = fmt.Sprintf(\"SupportedClientVersions: %v\", v.SupportedClientVersions)\n\t\ti++\n\t}\n\tif v.MembershipInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"MembershipInfo: %v\", v.MembershipInfo)\n\t\ti++\n\t}\n\tif v.PersistenceInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"PersistenceInfo: %v\", v.PersistenceInfo)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeClusterResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_PersistenceInfo_Equals(lhs, rhs map[string]*PersistenceInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeClusterResponse match the\n// provided DescribeClusterResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeClusterResponse) Equals(rhs *DescribeClusterResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SupportedClientVersions == nil && rhs.SupportedClientVersions == nil) || (v.SupportedClientVersions != nil && rhs.SupportedClientVersions != nil && v.SupportedClientVersions.Equals(rhs.SupportedClientVersions))) {\n\t\treturn false\n\t}\n\tif !((v.MembershipInfo == nil && rhs.MembershipInfo == nil) || (v.MembershipInfo != nil && rhs.MembershipInfo != nil && v.MembershipInfo.Equals(rhs.MembershipInfo))) {\n\t\treturn false\n\t}\n\tif !((v.PersistenceInfo == nil && rhs.PersistenceInfo == nil) || (v.PersistenceInfo != nil && rhs.PersistenceInfo != nil && _Map_String_PersistenceInfo_Equals(v.PersistenceInfo, rhs.PersistenceInfo))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_PersistenceInfo_Zapper map[string]*PersistenceInfo\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_PersistenceInfo_Zapper.\nfunc (m _Map_String_PersistenceInfo_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeClusterResponse.\nfunc (v *DescribeClusterResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SupportedClientVersions != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"supportedClientVersions\", v.SupportedClientVersions))\n\t}\n\tif v.MembershipInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"membershipInfo\", v.MembershipInfo))\n\t}\n\tif v.PersistenceInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"persistenceInfo\", (_Map_String_PersistenceInfo_Zapper)(v.PersistenceInfo)))\n\t}\n\treturn err\n}\n\n// GetSupportedClientVersions returns the value of SupportedClientVersions if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeClusterResponse) GetSupportedClientVersions() (o *shared.SupportedClientVersions) {\n\tif v != nil && v.SupportedClientVersions != nil {\n\t\treturn v.SupportedClientVersions\n\t}\n\n\treturn\n}\n\n// IsSetSupportedClientVersions returns true if SupportedClientVersions is not nil.\nfunc (v *DescribeClusterResponse) IsSetSupportedClientVersions() bool {\n\treturn v != nil && v.SupportedClientVersions != nil\n}\n\n// GetMembershipInfo returns the value of MembershipInfo if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeClusterResponse) GetMembershipInfo() (o *MembershipInfo) {\n\tif v != nil && v.MembershipInfo != nil {\n\t\treturn v.MembershipInfo\n\t}\n\n\treturn\n}\n\n// IsSetMembershipInfo returns true if MembershipInfo is not nil.\nfunc (v *DescribeClusterResponse) IsSetMembershipInfo() bool {\n\treturn v != nil && v.MembershipInfo != nil\n}\n\n// GetPersistenceInfo returns the value of PersistenceInfo if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeClusterResponse) GetPersistenceInfo() (o map[string]*PersistenceInfo) {\n\tif v != nil && v.PersistenceInfo != nil {\n\t\treturn v.PersistenceInfo\n\t}\n\n\treturn\n}\n\n// IsSetPersistenceInfo returns true if PersistenceInfo is not nil.\nfunc (v *DescribeClusterResponse) IsSetPersistenceInfo() bool {\n\treturn v != nil && v.PersistenceInfo != nil\n}\n\ntype DescribeWorkflowExecutionRequest struct {\n\tDomain    *string                   `json:\"domain,omitempty\"`\n\tExecution *shared.WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a DescribeWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be encoded.\nfunc (v *DescribeWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeWorkflowExecutionRequest\n// struct.\nfunc (v *DescribeWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeWorkflowExecutionRequest match the\n// provided DescribeWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeWorkflowExecutionRequest) Equals(rhs *DescribeWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeWorkflowExecutionRequest.\nfunc (v *DescribeWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype DescribeWorkflowExecutionResponse struct {\n\tShardId                *string `json:\"shardId,omitempty\"`\n\tHistoryAddr            *string `json:\"historyAddr,omitempty\"`\n\tMutableStateInCache    *string `json:\"mutableStateInCache,omitempty\"`\n\tMutableStateInDatabase *string `json:\"mutableStateInDatabase,omitempty\"`\n}\n\n// ToWire translates a DescribeWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardId != nil {\n\t\tw, err = wire.NewValueString(*(v.ShardId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryAddr != nil {\n\t\tw, err = wire.NewValueString(*(v.HistoryAddr)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.MutableStateInCache != nil {\n\t\tw, err = wire.NewValueString(*(v.MutableStateInCache)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tw, err = wire.NewValueString(*(v.MutableStateInDatabase)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ShardId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.HistoryAddr = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MutableStateInCache = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MutableStateInDatabase = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionResponse struct could not be encoded.\nfunc (v *DescribeWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ShardId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryAddr != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.HistoryAddr)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MutableStateInCache != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MutableStateInCache)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MutableStateInDatabase != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MutableStateInDatabase)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ShardId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.HistoryAddr = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MutableStateInCache = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MutableStateInDatabase = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeWorkflowExecutionResponse\n// struct.\nfunc (v *DescribeWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.ShardId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardId: %v\", *(v.ShardId))\n\t\ti++\n\t}\n\tif v.HistoryAddr != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryAddr: %v\", *(v.HistoryAddr))\n\t\ti++\n\t}\n\tif v.MutableStateInCache != nil {\n\t\tfields[i] = fmt.Sprintf(\"MutableStateInCache: %v\", *(v.MutableStateInCache))\n\t\ti++\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tfields[i] = fmt.Sprintf(\"MutableStateInDatabase: %v\", *(v.MutableStateInDatabase))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeWorkflowExecutionResponse match the\n// provided DescribeWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeWorkflowExecutionResponse) Equals(rhs *DescribeWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ShardId, rhs.ShardId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.HistoryAddr, rhs.HistoryAddr) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MutableStateInCache, rhs.MutableStateInCache) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MutableStateInDatabase, rhs.MutableStateInDatabase) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeWorkflowExecutionResponse.\nfunc (v *DescribeWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardId != nil {\n\t\tenc.AddString(\"shardId\", *v.ShardId)\n\t}\n\tif v.HistoryAddr != nil {\n\t\tenc.AddString(\"historyAddr\", *v.HistoryAddr)\n\t}\n\tif v.MutableStateInCache != nil {\n\t\tenc.AddString(\"mutableStateInCache\", *v.MutableStateInCache)\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tenc.AddString(\"mutableStateInDatabase\", *v.MutableStateInDatabase)\n\t}\n\treturn err\n}\n\n// GetShardId returns the value of ShardId if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetShardId() (o string) {\n\tif v != nil && v.ShardId != nil {\n\t\treturn *v.ShardId\n\t}\n\n\treturn\n}\n\n// IsSetShardId returns true if ShardId is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetShardId() bool {\n\treturn v != nil && v.ShardId != nil\n}\n\n// GetHistoryAddr returns the value of HistoryAddr if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetHistoryAddr() (o string) {\n\tif v != nil && v.HistoryAddr != nil {\n\t\treturn *v.HistoryAddr\n\t}\n\n\treturn\n}\n\n// IsSetHistoryAddr returns true if HistoryAddr is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetHistoryAddr() bool {\n\treturn v != nil && v.HistoryAddr != nil\n}\n\n// GetMutableStateInCache returns the value of MutableStateInCache if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetMutableStateInCache() (o string) {\n\tif v != nil && v.MutableStateInCache != nil {\n\t\treturn *v.MutableStateInCache\n\t}\n\n\treturn\n}\n\n// IsSetMutableStateInCache returns true if MutableStateInCache is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetMutableStateInCache() bool {\n\treturn v != nil && v.MutableStateInCache != nil\n}\n\n// GetMutableStateInDatabase returns the value of MutableStateInDatabase if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetMutableStateInDatabase() (o string) {\n\tif v != nil && v.MutableStateInDatabase != nil {\n\t\treturn *v.MutableStateInDatabase\n\t}\n\n\treturn\n}\n\n// IsSetMutableStateInDatabase returns true if MutableStateInDatabase is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetMutableStateInDatabase() bool {\n\treturn v != nil && v.MutableStateInDatabase != nil\n}\n\ntype GetDomainAsyncWorkflowConfiguratonRequest struct {\n\tDomain *string `json:\"domain,omitempty\"`\n}\n\n// ToWire translates a GetDomainAsyncWorkflowConfiguratonRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetDomainAsyncWorkflowConfiguratonRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainAsyncWorkflowConfiguratonRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainAsyncWorkflowConfiguratonRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainAsyncWorkflowConfiguratonRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainAsyncWorkflowConfiguratonRequest struct could not be encoded.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetDomainAsyncWorkflowConfiguratonRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainAsyncWorkflowConfiguratonRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainAsyncWorkflowConfiguratonRequest\n// struct.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainAsyncWorkflowConfiguratonRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainAsyncWorkflowConfiguratonRequest match the\n// provided GetDomainAsyncWorkflowConfiguratonRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) Equals(rhs *GetDomainAsyncWorkflowConfiguratonRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainAsyncWorkflowConfiguratonRequest.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *GetDomainAsyncWorkflowConfiguratonRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\ntype GetDomainAsyncWorkflowConfiguratonResponse struct {\n\tConfiguration *shared.AsyncWorkflowConfiguration `json:\"configuration,omitempty\"`\n}\n\n// ToWire translates a GetDomainAsyncWorkflowConfiguratonResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AsyncWorkflowConfiguration_Read(w wire.Value) (*shared.AsyncWorkflowConfiguration, error) {\n\tvar v shared.AsyncWorkflowConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetDomainAsyncWorkflowConfiguratonResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainAsyncWorkflowConfiguratonResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainAsyncWorkflowConfiguratonResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _AsyncWorkflowConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainAsyncWorkflowConfiguratonResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainAsyncWorkflowConfiguratonResponse struct could not be encoded.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AsyncWorkflowConfiguration_Decode(sr stream.Reader) (*shared.AsyncWorkflowConfiguration, error) {\n\tvar v shared.AsyncWorkflowConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetDomainAsyncWorkflowConfiguratonResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainAsyncWorkflowConfiguratonResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _AsyncWorkflowConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainAsyncWorkflowConfiguratonResponse\n// struct.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainAsyncWorkflowConfiguratonResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainAsyncWorkflowConfiguratonResponse match the\n// provided GetDomainAsyncWorkflowConfiguratonResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) Equals(rhs *GetDomainAsyncWorkflowConfiguratonResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainAsyncWorkflowConfiguratonResponse.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\treturn err\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) GetConfiguration() (o *shared.AsyncWorkflowConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *GetDomainAsyncWorkflowConfiguratonResponse) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\ntype GetDomainIsolationGroupsRequest struct {\n\tDomain *string `json:\"domain,omitempty\"`\n}\n\n// ToWire translates a GetDomainIsolationGroupsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainIsolationGroupsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetDomainIsolationGroupsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainIsolationGroupsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainIsolationGroupsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainIsolationGroupsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainIsolationGroupsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainIsolationGroupsRequest struct could not be encoded.\nfunc (v *GetDomainIsolationGroupsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetDomainIsolationGroupsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainIsolationGroupsRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainIsolationGroupsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainIsolationGroupsRequest\n// struct.\nfunc (v *GetDomainIsolationGroupsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainIsolationGroupsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainIsolationGroupsRequest match the\n// provided GetDomainIsolationGroupsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainIsolationGroupsRequest) Equals(rhs *GetDomainIsolationGroupsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainIsolationGroupsRequest.\nfunc (v *GetDomainIsolationGroupsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainIsolationGroupsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *GetDomainIsolationGroupsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\ntype GetDomainIsolationGroupsResponse struct {\n\tIsolationGroups *shared.IsolationGroupConfiguration `json:\"isolationGroups,omitempty\"`\n}\n\n// ToWire translates a GetDomainIsolationGroupsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainIsolationGroupsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IsolationGroups != nil {\n\t\tw, err = v.IsolationGroups.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IsolationGroupConfiguration_Read(w wire.Value) (*shared.IsolationGroupConfiguration, error) {\n\tvar v shared.IsolationGroupConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetDomainIsolationGroupsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainIsolationGroupsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainIsolationGroupsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainIsolationGroupsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainIsolationGroupsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainIsolationGroupsResponse struct could not be encoded.\nfunc (v *GetDomainIsolationGroupsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.IsolationGroups.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IsolationGroupConfiguration_Decode(sr stream.Reader) (*shared.IsolationGroupConfiguration, error) {\n\tvar v shared.IsolationGroupConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetDomainIsolationGroupsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainIsolationGroupsResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainIsolationGroupsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainIsolationGroupsResponse\n// struct.\nfunc (v *GetDomainIsolationGroupsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainIsolationGroupsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainIsolationGroupsResponse match the\n// provided GetDomainIsolationGroupsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainIsolationGroupsResponse) Equals(rhs *GetDomainIsolationGroupsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && v.IsolationGroups.Equals(rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainIsolationGroupsResponse.\nfunc (v *GetDomainIsolationGroupsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationGroups\", v.IsolationGroups))\n\t}\n\treturn err\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainIsolationGroupsResponse) GetIsolationGroups() (o *shared.IsolationGroupConfiguration) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *GetDomainIsolationGroupsResponse) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\ntype GetDynamicConfigRequest struct {\n\tConfigName *string                       `json:\"configName,omitempty\"`\n\tFilters    []*config.DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\ntype _List_DynamicConfigFilter_ValueList []*config.DynamicConfigFilter\n\nfunc (v _List_DynamicConfigFilter_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigFilter', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigFilter_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigFilter_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigFilter_ValueList) Close() {}\n\n// ToWire translates a GetDynamicConfigRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDynamicConfigRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ConfigName != nil {\n\t\tw, err = wire.NewValueString(*(v.ConfigName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigFilter_ValueList(v.Filters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigFilter_Read(w wire.Value) (*config.DynamicConfigFilter, error) {\n\tvar v config.DynamicConfigFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigFilter_Read(l wire.ValueList) ([]*config.DynamicConfigFilter, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*config.DynamicConfigFilter, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigFilter_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetDynamicConfigRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDynamicConfigRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDynamicConfigRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDynamicConfigRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ConfigName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Filters, err = _List_DynamicConfigFilter_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigFilter_Encode(val []*config.DynamicConfigFilter, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigFilter', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetDynamicConfigRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDynamicConfigRequest struct could not be encoded.\nfunc (v *GetDynamicConfigRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ConfigName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ConfigName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Filters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigFilter_Encode(v.Filters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigFilter_Decode(sr stream.Reader) (*config.DynamicConfigFilter, error) {\n\tvar v config.DynamicConfigFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigFilter_Decode(sr stream.Reader) ([]*config.DynamicConfigFilter, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*config.DynamicConfigFilter, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigFilter_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetDynamicConfigRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDynamicConfigRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetDynamicConfigRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ConfigName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Filters, err = _List_DynamicConfigFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDynamicConfigRequest\n// struct.\nfunc (v *GetDynamicConfigRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ConfigName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigName: %v\", *(v.ConfigName))\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Filters: %v\", v.Filters)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDynamicConfigRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DynamicConfigFilter_Equals(lhs, rhs []*config.DynamicConfigFilter) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetDynamicConfigRequest match the\n// provided GetDynamicConfigRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetDynamicConfigRequest) Equals(rhs *GetDynamicConfigRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ConfigName, rhs.ConfigName) {\n\t\treturn false\n\t}\n\tif !((v.Filters == nil && rhs.Filters == nil) || (v.Filters != nil && rhs.Filters != nil && _List_DynamicConfigFilter_Equals(v.Filters, rhs.Filters))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigFilter_Zapper []*config.DynamicConfigFilter\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigFilter_Zapper.\nfunc (l _List_DynamicConfigFilter_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDynamicConfigRequest.\nfunc (v *GetDynamicConfigRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ConfigName != nil {\n\t\tenc.AddString(\"configName\", *v.ConfigName)\n\t}\n\tif v.Filters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"filters\", (_List_DynamicConfigFilter_Zapper)(v.Filters)))\n\t}\n\treturn err\n}\n\n// GetConfigName returns the value of ConfigName if it is set or its\n// zero value if it is unset.\nfunc (v *GetDynamicConfigRequest) GetConfigName() (o string) {\n\tif v != nil && v.ConfigName != nil {\n\t\treturn *v.ConfigName\n\t}\n\n\treturn\n}\n\n// IsSetConfigName returns true if ConfigName is not nil.\nfunc (v *GetDynamicConfigRequest) IsSetConfigName() bool {\n\treturn v != nil && v.ConfigName != nil\n}\n\n// GetFilters returns the value of Filters if it is set or its\n// zero value if it is unset.\nfunc (v *GetDynamicConfigRequest) GetFilters() (o []*config.DynamicConfigFilter) {\n\tif v != nil && v.Filters != nil {\n\t\treturn v.Filters\n\t}\n\n\treturn\n}\n\n// IsSetFilters returns true if Filters is not nil.\nfunc (v *GetDynamicConfigRequest) IsSetFilters() bool {\n\treturn v != nil && v.Filters != nil\n}\n\ntype GetDynamicConfigResponse struct {\n\tValue *shared.DataBlob `json:\"value,omitempty\"`\n}\n\n// ToWire translates a GetDynamicConfigResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDynamicConfigResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Value != nil {\n\t\tw, err = v.Value.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DataBlob_Read(w wire.Value) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetDynamicConfigResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDynamicConfigResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDynamicConfigResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDynamicConfigResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Value, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDynamicConfigResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDynamicConfigResponse struct could not be encoded.\nfunc (v *GetDynamicConfigResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Value != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Value.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DataBlob_Decode(sr stream.Reader) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetDynamicConfigResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDynamicConfigResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetDynamicConfigResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Value, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDynamicConfigResponse\n// struct.\nfunc (v *GetDynamicConfigResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Value != nil {\n\t\tfields[i] = fmt.Sprintf(\"Value: %v\", v.Value)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDynamicConfigResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDynamicConfigResponse match the\n// provided GetDynamicConfigResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetDynamicConfigResponse) Equals(rhs *GetDynamicConfigResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Value == nil && rhs.Value == nil) || (v.Value != nil && rhs.Value != nil && v.Value.Equals(rhs.Value))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDynamicConfigResponse.\nfunc (v *GetDynamicConfigResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Value != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\t}\n\treturn err\n}\n\n// GetValue returns the value of Value if it is set or its\n// zero value if it is unset.\nfunc (v *GetDynamicConfigResponse) GetValue() (o *shared.DataBlob) {\n\tif v != nil && v.Value != nil {\n\t\treturn v.Value\n\t}\n\n\treturn\n}\n\n// IsSetValue returns true if Value is not nil.\nfunc (v *GetDynamicConfigResponse) IsSetValue() bool {\n\treturn v != nil && v.Value != nil\n}\n\ntype GetGlobalIsolationGroupsRequest struct {\n}\n\n// ToWire translates a GetGlobalIsolationGroupsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetGlobalIsolationGroupsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetGlobalIsolationGroupsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetGlobalIsolationGroupsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetGlobalIsolationGroupsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetGlobalIsolationGroupsRequest) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetGlobalIsolationGroupsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetGlobalIsolationGroupsRequest struct could not be encoded.\nfunc (v *GetGlobalIsolationGroupsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetGlobalIsolationGroupsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetGlobalIsolationGroupsRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetGlobalIsolationGroupsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetGlobalIsolationGroupsRequest\n// struct.\nfunc (v *GetGlobalIsolationGroupsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"GetGlobalIsolationGroupsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetGlobalIsolationGroupsRequest match the\n// provided GetGlobalIsolationGroupsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetGlobalIsolationGroupsRequest) Equals(rhs *GetGlobalIsolationGroupsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetGlobalIsolationGroupsRequest.\nfunc (v *GetGlobalIsolationGroupsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype GetGlobalIsolationGroupsResponse struct {\n\tIsolationGroups *shared.IsolationGroupConfiguration `json:\"isolationGroups,omitempty\"`\n}\n\n// ToWire translates a GetGlobalIsolationGroupsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetGlobalIsolationGroupsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IsolationGroups != nil {\n\t\tw, err = v.IsolationGroups.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetGlobalIsolationGroupsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetGlobalIsolationGroupsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetGlobalIsolationGroupsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetGlobalIsolationGroupsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetGlobalIsolationGroupsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetGlobalIsolationGroupsResponse struct could not be encoded.\nfunc (v *GetGlobalIsolationGroupsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.IsolationGroups.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetGlobalIsolationGroupsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetGlobalIsolationGroupsResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetGlobalIsolationGroupsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetGlobalIsolationGroupsResponse\n// struct.\nfunc (v *GetGlobalIsolationGroupsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetGlobalIsolationGroupsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetGlobalIsolationGroupsResponse match the\n// provided GetGlobalIsolationGroupsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetGlobalIsolationGroupsResponse) Equals(rhs *GetGlobalIsolationGroupsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && v.IsolationGroups.Equals(rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetGlobalIsolationGroupsResponse.\nfunc (v *GetGlobalIsolationGroupsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationGroups\", v.IsolationGroups))\n\t}\n\treturn err\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *GetGlobalIsolationGroupsResponse) GetIsolationGroups() (o *shared.IsolationGroupConfiguration) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *GetGlobalIsolationGroupsResponse) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\n// StartEventId defines the beginning of the event to fetch. The first event is exclusive.\n// EndEventId and EndEventVersion defines the end of the event to fetch. The end event is exclusive.\ntype GetWorkflowExecutionRawHistoryV2Request struct {\n\tDomain            *string                   `json:\"domain,omitempty\"`\n\tExecution         *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tStartEventId      *int64                    `json:\"startEventId,omitempty\"`\n\tStartEventVersion *int64                    `json:\"startEventVersion,omitempty\"`\n\tEndEventId        *int64                    `json:\"endEventId,omitempty\"`\n\tEndEventVersion   *int64                    `json:\"endEventVersion,omitempty\"`\n\tMaximumPageSize   *int32                    `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken     []byte                    `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a GetWorkflowExecutionRawHistoryV2Request struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.EndEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.EndEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetWorkflowExecutionRawHistoryV2Request struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetWorkflowExecutionRawHistoryV2Request struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetWorkflowExecutionRawHistoryV2Request\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetWorkflowExecutionRawHistoryV2Request struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetWorkflowExecutionRawHistoryV2Request struct could not be encoded.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetWorkflowExecutionRawHistoryV2Request struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetWorkflowExecutionRawHistoryV2Request struct could not be generated from the wire\n// representation.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetWorkflowExecutionRawHistoryV2Request\n// struct.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.StartEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartEventId: %v\", *(v.StartEventId))\n\t\ti++\n\t}\n\tif v.StartEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartEventVersion: %v\", *(v.StartEventVersion))\n\t\ti++\n\t}\n\tif v.EndEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndEventId: %v\", *(v.EndEventId))\n\t\ti++\n\t}\n\tif v.EndEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndEventVersion: %v\", *(v.EndEventVersion))\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetWorkflowExecutionRawHistoryV2Request{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this GetWorkflowExecutionRawHistoryV2Request match the\n// provided GetWorkflowExecutionRawHistoryV2Request.\n//\n// This function performs a deep comparison.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) Equals(rhs *GetWorkflowExecutionRawHistoryV2Request) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartEventId, rhs.StartEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartEventVersion, rhs.StartEventVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndEventId, rhs.EndEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndEventVersion, rhs.EndEventVersion) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetWorkflowExecutionRawHistoryV2Request.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.StartEventId != nil {\n\t\tenc.AddInt64(\"startEventId\", *v.StartEventId)\n\t}\n\tif v.StartEventVersion != nil {\n\t\tenc.AddInt64(\"startEventVersion\", *v.StartEventVersion)\n\t}\n\tif v.EndEventId != nil {\n\t\tenc.AddInt64(\"endEventId\", *v.EndEventId)\n\t}\n\tif v.EndEventVersion != nil {\n\t\tenc.AddInt64(\"endEventVersion\", *v.EndEventVersion)\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetStartEventId returns the value of StartEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetStartEventId() (o int64) {\n\tif v != nil && v.StartEventId != nil {\n\t\treturn *v.StartEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartEventId returns true if StartEventId is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetStartEventId() bool {\n\treturn v != nil && v.StartEventId != nil\n}\n\n// GetStartEventVersion returns the value of StartEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetStartEventVersion() (o int64) {\n\tif v != nil && v.StartEventVersion != nil {\n\t\treturn *v.StartEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetStartEventVersion returns true if StartEventVersion is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetStartEventVersion() bool {\n\treturn v != nil && v.StartEventVersion != nil\n}\n\n// GetEndEventId returns the value of EndEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetEndEventId() (o int64) {\n\tif v != nil && v.EndEventId != nil {\n\t\treturn *v.EndEventId\n\t}\n\n\treturn\n}\n\n// IsSetEndEventId returns true if EndEventId is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetEndEventId() bool {\n\treturn v != nil && v.EndEventId != nil\n}\n\n// GetEndEventVersion returns the value of EndEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetEndEventVersion() (o int64) {\n\tif v != nil && v.EndEventVersion != nil {\n\t\treturn *v.EndEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetEndEventVersion returns true if EndEventVersion is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetEndEventVersion() bool {\n\treturn v != nil && v.EndEventVersion != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype GetWorkflowExecutionRawHistoryV2Response struct {\n\tNextPageToken  []byte                 `json:\"nextPageToken,omitempty\"`\n\tHistoryBatches []*shared.DataBlob     `json:\"historyBatches,omitempty\"`\n\tVersionHistory *shared.VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\ntype _List_DataBlob_ValueList []*shared.DataBlob\n\nfunc (v _List_DataBlob_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.DataBlob', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DataBlob_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DataBlob_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DataBlob_ValueList) Close() {}\n\n// ToWire translates a GetWorkflowExecutionRawHistoryV2Response struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryBatches != nil {\n\t\tw, err = wire.NewValueList(_List_DataBlob_ValueList(v.HistoryBatches)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tw, err = v.VersionHistory.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_DataBlob_Read(l wire.ValueList) ([]*shared.DataBlob, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*shared.DataBlob, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DataBlob_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _VersionHistory_Read(w wire.Value) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetWorkflowExecutionRawHistoryV2Response struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetWorkflowExecutionRawHistoryV2Response struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetWorkflowExecutionRawHistoryV2Response\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.HistoryBatches, err = _List_DataBlob_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistory, err = _VersionHistory_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DataBlob_Encode(val []*shared.DataBlob, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.DataBlob', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetWorkflowExecutionRawHistoryV2Response struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetWorkflowExecutionRawHistoryV2Response struct could not be encoded.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryBatches != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DataBlob_Encode(v.HistoryBatches, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistory != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistory.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_DataBlob_Decode(sr stream.Reader) ([]*shared.DataBlob, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*shared.DataBlob, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DataBlob_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _VersionHistory_Decode(sr stream.Reader) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetWorkflowExecutionRawHistoryV2Response struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetWorkflowExecutionRawHistoryV2Response struct could not be generated from the wire\n// representation.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.HistoryBatches, err = _List_DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistory, err = _VersionHistory_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetWorkflowExecutionRawHistoryV2Response\n// struct.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.HistoryBatches != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryBatches: %v\", v.HistoryBatches)\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistory: %v\", v.VersionHistory)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetWorkflowExecutionRawHistoryV2Response{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DataBlob_Equals(lhs, rhs []*shared.DataBlob) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetWorkflowExecutionRawHistoryV2Response match the\n// provided GetWorkflowExecutionRawHistoryV2Response.\n//\n// This function performs a deep comparison.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) Equals(rhs *GetWorkflowExecutionRawHistoryV2Response) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !((v.HistoryBatches == nil && rhs.HistoryBatches == nil) || (v.HistoryBatches != nil && rhs.HistoryBatches != nil && _List_DataBlob_Equals(v.HistoryBatches, rhs.HistoryBatches))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistory == nil && rhs.VersionHistory == nil) || (v.VersionHistory != nil && rhs.VersionHistory != nil && v.VersionHistory.Equals(rhs.VersionHistory))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DataBlob_Zapper []*shared.DataBlob\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DataBlob_Zapper.\nfunc (l _List_DataBlob_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetWorkflowExecutionRawHistoryV2Response.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.HistoryBatches != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"historyBatches\", (_List_DataBlob_Zapper)(v.HistoryBatches)))\n\t}\n\tif v.VersionHistory != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistory\", v.VersionHistory))\n\t}\n\treturn err\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetHistoryBatches returns the value of HistoryBatches if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) GetHistoryBatches() (o []*shared.DataBlob) {\n\tif v != nil && v.HistoryBatches != nil {\n\t\treturn v.HistoryBatches\n\t}\n\n\treturn\n}\n\n// IsSetHistoryBatches returns true if HistoryBatches is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) IsSetHistoryBatches() bool {\n\treturn v != nil && v.HistoryBatches != nil\n}\n\n// GetVersionHistory returns the value of VersionHistory if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) GetVersionHistory() (o *shared.VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistory returns true if VersionHistory is not nil.\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) IsSetVersionHistory() bool {\n\treturn v != nil && v.VersionHistory != nil\n}\n\ntype HostInfo struct {\n\tIdentity *string `json:\"Identity,omitempty\"`\n}\n\n// ToWire translates a HostInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HostInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HostInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HostInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HostInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HostInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HostInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HostInfo struct could not be encoded.\nfunc (v *HostInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HostInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HostInfo struct could not be generated from the wire\n// representation.\nfunc (v *HostInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HostInfo\n// struct.\nfunc (v *HostInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HostInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HostInfo match the\n// provided HostInfo.\n//\n// This function performs a deep comparison.\nfunc (v *HostInfo) Equals(rhs *HostInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HostInfo.\nfunc (v *HostInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"Identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *HostInfo) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *HostInfo) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype ListDynamicConfigRequest struct {\n\tConfigName *string `json:\"configName,omitempty\"`\n}\n\n// ToWire translates a ListDynamicConfigRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListDynamicConfigRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ConfigName != nil {\n\t\tw, err = wire.NewValueString(*(v.ConfigName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListDynamicConfigRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListDynamicConfigRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListDynamicConfigRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListDynamicConfigRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ConfigName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListDynamicConfigRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListDynamicConfigRequest struct could not be encoded.\nfunc (v *ListDynamicConfigRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ConfigName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ConfigName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListDynamicConfigRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListDynamicConfigRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListDynamicConfigRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ConfigName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListDynamicConfigRequest\n// struct.\nfunc (v *ListDynamicConfigRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ConfigName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigName: %v\", *(v.ConfigName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListDynamicConfigRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListDynamicConfigRequest match the\n// provided ListDynamicConfigRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListDynamicConfigRequest) Equals(rhs *ListDynamicConfigRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ConfigName, rhs.ConfigName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListDynamicConfigRequest.\nfunc (v *ListDynamicConfigRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ConfigName != nil {\n\t\tenc.AddString(\"configName\", *v.ConfigName)\n\t}\n\treturn err\n}\n\n// GetConfigName returns the value of ConfigName if it is set or its\n// zero value if it is unset.\nfunc (v *ListDynamicConfigRequest) GetConfigName() (o string) {\n\tif v != nil && v.ConfigName != nil {\n\t\treturn *v.ConfigName\n\t}\n\n\treturn\n}\n\n// IsSetConfigName returns true if ConfigName is not nil.\nfunc (v *ListDynamicConfigRequest) IsSetConfigName() bool {\n\treturn v != nil && v.ConfigName != nil\n}\n\ntype ListDynamicConfigResponse struct {\n\tEntries []*config.DynamicConfigEntry `json:\"entries,omitempty\"`\n}\n\ntype _List_DynamicConfigEntry_ValueList []*config.DynamicConfigEntry\n\nfunc (v _List_DynamicConfigEntry_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigEntry', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigEntry_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigEntry_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigEntry_ValueList) Close() {}\n\n// ToWire translates a ListDynamicConfigResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListDynamicConfigResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Entries != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigEntry_ValueList(v.Entries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigEntry_Read(w wire.Value) (*config.DynamicConfigEntry, error) {\n\tvar v config.DynamicConfigEntry\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigEntry_Read(l wire.ValueList) ([]*config.DynamicConfigEntry, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*config.DynamicConfigEntry, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigEntry_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ListDynamicConfigResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListDynamicConfigResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListDynamicConfigResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListDynamicConfigResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Entries, err = _List_DynamicConfigEntry_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigEntry_Encode(val []*config.DynamicConfigEntry, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigEntry', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ListDynamicConfigResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListDynamicConfigResponse struct could not be encoded.\nfunc (v *ListDynamicConfigResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Entries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigEntry_Encode(v.Entries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigEntry_Decode(sr stream.Reader) (*config.DynamicConfigEntry, error) {\n\tvar v config.DynamicConfigEntry\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigEntry_Decode(sr stream.Reader) ([]*config.DynamicConfigEntry, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*config.DynamicConfigEntry, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigEntry_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ListDynamicConfigResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListDynamicConfigResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListDynamicConfigResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Entries, err = _List_DynamicConfigEntry_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListDynamicConfigResponse\n// struct.\nfunc (v *ListDynamicConfigResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Entries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Entries: %v\", v.Entries)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListDynamicConfigResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DynamicConfigEntry_Equals(lhs, rhs []*config.DynamicConfigEntry) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ListDynamicConfigResponse match the\n// provided ListDynamicConfigResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListDynamicConfigResponse) Equals(rhs *ListDynamicConfigResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Entries == nil && rhs.Entries == nil) || (v.Entries != nil && rhs.Entries != nil && _List_DynamicConfigEntry_Equals(v.Entries, rhs.Entries))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigEntry_Zapper []*config.DynamicConfigEntry\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigEntry_Zapper.\nfunc (l _List_DynamicConfigEntry_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListDynamicConfigResponse.\nfunc (v *ListDynamicConfigResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Entries != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"entries\", (_List_DynamicConfigEntry_Zapper)(v.Entries)))\n\t}\n\treturn err\n}\n\n// GetEntries returns the value of Entries if it is set or its\n// zero value if it is unset.\nfunc (v *ListDynamicConfigResponse) GetEntries() (o []*config.DynamicConfigEntry) {\n\tif v != nil && v.Entries != nil {\n\t\treturn v.Entries\n\t}\n\n\treturn\n}\n\n// IsSetEntries returns true if Entries is not nil.\nfunc (v *ListDynamicConfigResponse) IsSetEntries() bool {\n\treturn v != nil && v.Entries != nil\n}\n\ntype MembershipInfo struct {\n\tCurrentHost      *HostInfo   `json:\"currentHost,omitempty\"`\n\tReachableMembers []string    `json:\"reachableMembers,omitempty\"`\n\tRings            []*RingInfo `json:\"rings,omitempty\"`\n}\n\ntype _List_String_ValueList []string\n\nfunc (v _List_String_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueString(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_String_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_String_ValueList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_List_String_ValueList) Close() {}\n\ntype _List_RingInfo_ValueList []*RingInfo\n\nfunc (v _List_RingInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*RingInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_RingInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_RingInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_RingInfo_ValueList) Close() {}\n\n// ToWire translates a MembershipInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MembershipInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CurrentHost != nil {\n\t\tw, err = v.CurrentHost.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ReachableMembers != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.ReachableMembers)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Rings != nil {\n\t\tw, err = wire.NewValueList(_List_RingInfo_ValueList(v.Rings)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HostInfo_Read(w wire.Value) (*HostInfo, error) {\n\tvar v HostInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_String_Read(l wire.ValueList) ([]string, error) {\n\tif l.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make([]string, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _RingInfo_Read(w wire.Value) (*RingInfo, error) {\n\tvar v RingInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_RingInfo_Read(l wire.ValueList) ([]*RingInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*RingInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _RingInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a MembershipInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MembershipInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MembershipInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MembershipInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CurrentHost, err = _HostInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ReachableMembers, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Rings, err = _List_RingInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_String_Encode(val []string, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TBinary,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _List_RingInfo_Encode(val []*RingInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*RingInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a MembershipInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MembershipInfo struct could not be encoded.\nfunc (v *MembershipInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CurrentHost != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CurrentHost.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReachableMembers != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.ReachableMembers, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Rings != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_RingInfo_Encode(v.Rings, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HostInfo_Decode(sr stream.Reader) (*HostInfo, error) {\n\tvar v HostInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_String_Decode(sr stream.Reader) ([]string, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TBinary {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]string, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _RingInfo_Decode(sr stream.Reader) (*RingInfo, error) {\n\tvar v RingInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_RingInfo_Decode(sr stream.Reader) ([]*RingInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*RingInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _RingInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a MembershipInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MembershipInfo struct could not be generated from the wire\n// representation.\nfunc (v *MembershipInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.CurrentHost, err = _HostInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.ReachableMembers, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.Rings, err = _List_RingInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MembershipInfo\n// struct.\nfunc (v *MembershipInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.CurrentHost != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentHost: %v\", v.CurrentHost)\n\t\ti++\n\t}\n\tif v.ReachableMembers != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReachableMembers: %v\", v.ReachableMembers)\n\t\ti++\n\t}\n\tif v.Rings != nil {\n\t\tfields[i] = fmt.Sprintf(\"Rings: %v\", v.Rings)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MembershipInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_String_Equals(lhs, rhs []string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _List_RingInfo_Equals(lhs, rhs []*RingInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this MembershipInfo match the\n// provided MembershipInfo.\n//\n// This function performs a deep comparison.\nfunc (v *MembershipInfo) Equals(rhs *MembershipInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CurrentHost == nil && rhs.CurrentHost == nil) || (v.CurrentHost != nil && rhs.CurrentHost != nil && v.CurrentHost.Equals(rhs.CurrentHost))) {\n\t\treturn false\n\t}\n\tif !((v.ReachableMembers == nil && rhs.ReachableMembers == nil) || (v.ReachableMembers != nil && rhs.ReachableMembers != nil && _List_String_Equals(v.ReachableMembers, rhs.ReachableMembers))) {\n\t\treturn false\n\t}\n\tif !((v.Rings == nil && rhs.Rings == nil) || (v.Rings != nil && rhs.Rings != nil && _List_RingInfo_Equals(v.Rings, rhs.Rings))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_String_Zapper []string\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_String_Zapper.\nfunc (l _List_String_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendString(v)\n\t}\n\treturn err\n}\n\ntype _List_RingInfo_Zapper []*RingInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_RingInfo_Zapper.\nfunc (l _List_RingInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MembershipInfo.\nfunc (v *MembershipInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CurrentHost != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"currentHost\", v.CurrentHost))\n\t}\n\tif v.ReachableMembers != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"reachableMembers\", (_List_String_Zapper)(v.ReachableMembers)))\n\t}\n\tif v.Rings != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"rings\", (_List_RingInfo_Zapper)(v.Rings)))\n\t}\n\treturn err\n}\n\n// GetCurrentHost returns the value of CurrentHost if it is set or its\n// zero value if it is unset.\nfunc (v *MembershipInfo) GetCurrentHost() (o *HostInfo) {\n\tif v != nil && v.CurrentHost != nil {\n\t\treturn v.CurrentHost\n\t}\n\n\treturn\n}\n\n// IsSetCurrentHost returns true if CurrentHost is not nil.\nfunc (v *MembershipInfo) IsSetCurrentHost() bool {\n\treturn v != nil && v.CurrentHost != nil\n}\n\n// GetReachableMembers returns the value of ReachableMembers if it is set or its\n// zero value if it is unset.\nfunc (v *MembershipInfo) GetReachableMembers() (o []string) {\n\tif v != nil && v.ReachableMembers != nil {\n\t\treturn v.ReachableMembers\n\t}\n\n\treturn\n}\n\n// IsSetReachableMembers returns true if ReachableMembers is not nil.\nfunc (v *MembershipInfo) IsSetReachableMembers() bool {\n\treturn v != nil && v.ReachableMembers != nil\n}\n\n// GetRings returns the value of Rings if it is set or its\n// zero value if it is unset.\nfunc (v *MembershipInfo) GetRings() (o []*RingInfo) {\n\tif v != nil && v.Rings != nil {\n\t\treturn v.Rings\n\t}\n\n\treturn\n}\n\n// IsSetRings returns true if Rings is not nil.\nfunc (v *MembershipInfo) IsSetRings() bool {\n\treturn v != nil && v.Rings != nil\n}\n\ntype PersistenceFeature struct {\n\tKey     *string `json:\"key,omitempty\"`\n\tEnabled *bool   `json:\"enabled,omitempty\"`\n}\n\n// ToWire translates a PersistenceFeature struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PersistenceFeature) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Key != nil {\n\t\tw, err = wire.NewValueString(*(v.Key)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Enabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.Enabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PersistenceFeature struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PersistenceFeature struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PersistenceFeature\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PersistenceFeature) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Key = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Enabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PersistenceFeature struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PersistenceFeature struct could not be encoded.\nfunc (v *PersistenceFeature) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Key != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Key)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Enabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Enabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PersistenceFeature struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PersistenceFeature struct could not be generated from the wire\n// representation.\nfunc (v *PersistenceFeature) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Key = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Enabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PersistenceFeature\n// struct.\nfunc (v *PersistenceFeature) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Key != nil {\n\t\tfields[i] = fmt.Sprintf(\"Key: %v\", *(v.Key))\n\t\ti++\n\t}\n\tif v.Enabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"Enabled: %v\", *(v.Enabled))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PersistenceFeature{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PersistenceFeature match the\n// provided PersistenceFeature.\n//\n// This function performs a deep comparison.\nfunc (v *PersistenceFeature) Equals(rhs *PersistenceFeature) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Key, rhs.Key) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Enabled, rhs.Enabled) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PersistenceFeature.\nfunc (v *PersistenceFeature) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Key != nil {\n\t\tenc.AddString(\"key\", *v.Key)\n\t}\n\tif v.Enabled != nil {\n\t\tenc.AddBool(\"enabled\", *v.Enabled)\n\t}\n\treturn err\n}\n\n// GetKey returns the value of Key if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceFeature) GetKey() (o string) {\n\tif v != nil && v.Key != nil {\n\t\treturn *v.Key\n\t}\n\n\treturn\n}\n\n// IsSetKey returns true if Key is not nil.\nfunc (v *PersistenceFeature) IsSetKey() bool {\n\treturn v != nil && v.Key != nil\n}\n\n// GetEnabled returns the value of Enabled if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceFeature) GetEnabled() (o bool) {\n\tif v != nil && v.Enabled != nil {\n\t\treturn *v.Enabled\n\t}\n\n\treturn\n}\n\n// IsSetEnabled returns true if Enabled is not nil.\nfunc (v *PersistenceFeature) IsSetEnabled() bool {\n\treturn v != nil && v.Enabled != nil\n}\n\ntype PersistenceInfo struct {\n\tBackend  *string               `json:\"backend,omitempty\"`\n\tSettings []*PersistenceSetting `json:\"settings,omitempty\"`\n\tFeatures []*PersistenceFeature `json:\"features,omitempty\"`\n}\n\ntype _List_PersistenceSetting_ValueList []*PersistenceSetting\n\nfunc (v _List_PersistenceSetting_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PersistenceSetting', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_PersistenceSetting_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_PersistenceSetting_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_PersistenceSetting_ValueList) Close() {}\n\ntype _List_PersistenceFeature_ValueList []*PersistenceFeature\n\nfunc (v _List_PersistenceFeature_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PersistenceFeature', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_PersistenceFeature_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_PersistenceFeature_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_PersistenceFeature_ValueList) Close() {}\n\n// ToWire translates a PersistenceInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PersistenceInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Backend != nil {\n\t\tw, err = wire.NewValueString(*(v.Backend)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Settings != nil {\n\t\tw, err = wire.NewValueList(_List_PersistenceSetting_ValueList(v.Settings)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Features != nil {\n\t\tw, err = wire.NewValueList(_List_PersistenceFeature_ValueList(v.Features)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PersistenceSetting_Read(w wire.Value) (*PersistenceSetting, error) {\n\tvar v PersistenceSetting\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_PersistenceSetting_Read(l wire.ValueList) ([]*PersistenceSetting, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*PersistenceSetting, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _PersistenceSetting_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _PersistenceFeature_Read(w wire.Value) (*PersistenceFeature, error) {\n\tvar v PersistenceFeature\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_PersistenceFeature_Read(l wire.ValueList) ([]*PersistenceFeature, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*PersistenceFeature, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _PersistenceFeature_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a PersistenceInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PersistenceInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PersistenceInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PersistenceInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Backend = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Settings, err = _List_PersistenceSetting_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Features, err = _List_PersistenceFeature_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_PersistenceSetting_Encode(val []*PersistenceSetting, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PersistenceSetting', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _List_PersistenceFeature_Encode(val []*PersistenceFeature, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PersistenceFeature', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a PersistenceInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PersistenceInfo struct could not be encoded.\nfunc (v *PersistenceInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Backend != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Backend)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Settings != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_PersistenceSetting_Encode(v.Settings, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Features != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_PersistenceFeature_Encode(v.Features, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PersistenceSetting_Decode(sr stream.Reader) (*PersistenceSetting, error) {\n\tvar v PersistenceSetting\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_PersistenceSetting_Decode(sr stream.Reader) ([]*PersistenceSetting, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*PersistenceSetting, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _PersistenceSetting_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _PersistenceFeature_Decode(sr stream.Reader) (*PersistenceFeature, error) {\n\tvar v PersistenceFeature\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_PersistenceFeature_Decode(sr stream.Reader) ([]*PersistenceFeature, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*PersistenceFeature, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _PersistenceFeature_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a PersistenceInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PersistenceInfo struct could not be generated from the wire\n// representation.\nfunc (v *PersistenceInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Backend = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Settings, err = _List_PersistenceSetting_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.Features, err = _List_PersistenceFeature_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PersistenceInfo\n// struct.\nfunc (v *PersistenceInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Backend != nil {\n\t\tfields[i] = fmt.Sprintf(\"Backend: %v\", *(v.Backend))\n\t\ti++\n\t}\n\tif v.Settings != nil {\n\t\tfields[i] = fmt.Sprintf(\"Settings: %v\", v.Settings)\n\t\ti++\n\t}\n\tif v.Features != nil {\n\t\tfields[i] = fmt.Sprintf(\"Features: %v\", v.Features)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PersistenceInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_PersistenceSetting_Equals(lhs, rhs []*PersistenceSetting) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _List_PersistenceFeature_Equals(lhs, rhs []*PersistenceFeature) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this PersistenceInfo match the\n// provided PersistenceInfo.\n//\n// This function performs a deep comparison.\nfunc (v *PersistenceInfo) Equals(rhs *PersistenceInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Backend, rhs.Backend) {\n\t\treturn false\n\t}\n\tif !((v.Settings == nil && rhs.Settings == nil) || (v.Settings != nil && rhs.Settings != nil && _List_PersistenceSetting_Equals(v.Settings, rhs.Settings))) {\n\t\treturn false\n\t}\n\tif !((v.Features == nil && rhs.Features == nil) || (v.Features != nil && rhs.Features != nil && _List_PersistenceFeature_Equals(v.Features, rhs.Features))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_PersistenceSetting_Zapper []*PersistenceSetting\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_PersistenceSetting_Zapper.\nfunc (l _List_PersistenceSetting_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\ntype _List_PersistenceFeature_Zapper []*PersistenceFeature\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_PersistenceFeature_Zapper.\nfunc (l _List_PersistenceFeature_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PersistenceInfo.\nfunc (v *PersistenceInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Backend != nil {\n\t\tenc.AddString(\"backend\", *v.Backend)\n\t}\n\tif v.Settings != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"settings\", (_List_PersistenceSetting_Zapper)(v.Settings)))\n\t}\n\tif v.Features != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"features\", (_List_PersistenceFeature_Zapper)(v.Features)))\n\t}\n\treturn err\n}\n\n// GetBackend returns the value of Backend if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceInfo) GetBackend() (o string) {\n\tif v != nil && v.Backend != nil {\n\t\treturn *v.Backend\n\t}\n\n\treturn\n}\n\n// IsSetBackend returns true if Backend is not nil.\nfunc (v *PersistenceInfo) IsSetBackend() bool {\n\treturn v != nil && v.Backend != nil\n}\n\n// GetSettings returns the value of Settings if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceInfo) GetSettings() (o []*PersistenceSetting) {\n\tif v != nil && v.Settings != nil {\n\t\treturn v.Settings\n\t}\n\n\treturn\n}\n\n// IsSetSettings returns true if Settings is not nil.\nfunc (v *PersistenceInfo) IsSetSettings() bool {\n\treturn v != nil && v.Settings != nil\n}\n\n// GetFeatures returns the value of Features if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceInfo) GetFeatures() (o []*PersistenceFeature) {\n\tif v != nil && v.Features != nil {\n\t\treturn v.Features\n\t}\n\n\treturn\n}\n\n// IsSetFeatures returns true if Features is not nil.\nfunc (v *PersistenceInfo) IsSetFeatures() bool {\n\treturn v != nil && v.Features != nil\n}\n\ntype PersistenceSetting struct {\n\tKey   *string `json:\"key,omitempty\"`\n\tValue *string `json:\"value,omitempty\"`\n}\n\n// ToWire translates a PersistenceSetting struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PersistenceSetting) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Key != nil {\n\t\tw, err = wire.NewValueString(*(v.Key)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tw, err = wire.NewValueString(*(v.Value)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PersistenceSetting struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PersistenceSetting struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PersistenceSetting\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PersistenceSetting) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Key = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Value = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PersistenceSetting struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PersistenceSetting struct could not be encoded.\nfunc (v *PersistenceSetting) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Key != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Key)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Value != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Value)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PersistenceSetting struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PersistenceSetting struct could not be generated from the wire\n// representation.\nfunc (v *PersistenceSetting) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Key = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Value = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PersistenceSetting\n// struct.\nfunc (v *PersistenceSetting) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Key != nil {\n\t\tfields[i] = fmt.Sprintf(\"Key: %v\", *(v.Key))\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tfields[i] = fmt.Sprintf(\"Value: %v\", *(v.Value))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PersistenceSetting{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PersistenceSetting match the\n// provided PersistenceSetting.\n//\n// This function performs a deep comparison.\nfunc (v *PersistenceSetting) Equals(rhs *PersistenceSetting) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Key, rhs.Key) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Value, rhs.Value) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PersistenceSetting.\nfunc (v *PersistenceSetting) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Key != nil {\n\t\tenc.AddString(\"key\", *v.Key)\n\t}\n\tif v.Value != nil {\n\t\tenc.AddString(\"value\", *v.Value)\n\t}\n\treturn err\n}\n\n// GetKey returns the value of Key if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceSetting) GetKey() (o string) {\n\tif v != nil && v.Key != nil {\n\t\treturn *v.Key\n\t}\n\n\treturn\n}\n\n// IsSetKey returns true if Key is not nil.\nfunc (v *PersistenceSetting) IsSetKey() bool {\n\treturn v != nil && v.Key != nil\n}\n\n// GetValue returns the value of Value if it is set or its\n// zero value if it is unset.\nfunc (v *PersistenceSetting) GetValue() (o string) {\n\tif v != nil && v.Value != nil {\n\t\treturn *v.Value\n\t}\n\n\treturn\n}\n\n// IsSetValue returns true if Value is not nil.\nfunc (v *PersistenceSetting) IsSetValue() bool {\n\treturn v != nil && v.Value != nil\n}\n\ntype ResendReplicationTasksRequest struct {\n\tDomainID      *string `json:\"domainID,omitempty\"`\n\tWorkflowID    *string `json:\"workflowID,omitempty\"`\n\tRunID         *string `json:\"runID,omitempty\"`\n\tRemoteCluster *string `json:\"remoteCluster,omitempty\"`\n\tStartEventID  *int64  `json:\"startEventID,omitempty\"`\n\tStartVersion  *int64  `json:\"startVersion,omitempty\"`\n\tEndEventID    *int64  `json:\"endEventID,omitempty\"`\n\tEndVersion    *int64  `json:\"endVersion,omitempty\"`\n}\n\n// ToWire translates a ResendReplicationTasksRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResendReplicationTasksRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RemoteCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.RemoteCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StartEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.EndEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.EndVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResendReplicationTasksRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResendReplicationTasksRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResendReplicationTasksRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResendReplicationTasksRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RemoteCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResendReplicationTasksRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResendReplicationTasksRequest struct could not be encoded.\nfunc (v *ResendReplicationTasksRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RemoteCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RemoteCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResendReplicationTasksRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResendReplicationTasksRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResendReplicationTasksRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RemoteCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResendReplicationTasksRequest\n// struct.\nfunc (v *ResendReplicationTasksRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.RemoteCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"RemoteCluster: %v\", *(v.RemoteCluster))\n\t\ti++\n\t}\n\tif v.StartEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartEventID: %v\", *(v.StartEventID))\n\t\ti++\n\t}\n\tif v.StartVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartVersion: %v\", *(v.StartVersion))\n\t\ti++\n\t}\n\tif v.EndEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndEventID: %v\", *(v.EndEventID))\n\t\ti++\n\t}\n\tif v.EndVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndVersion: %v\", *(v.EndVersion))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResendReplicationTasksRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResendReplicationTasksRequest match the\n// provided ResendReplicationTasksRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResendReplicationTasksRequest) Equals(rhs *ResendReplicationTasksRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RemoteCluster, rhs.RemoteCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartEventID, rhs.StartEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartVersion, rhs.StartVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndEventID, rhs.EndEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndVersion, rhs.EndVersion) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResendReplicationTasksRequest.\nfunc (v *ResendReplicationTasksRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.RemoteCluster != nil {\n\t\tenc.AddString(\"remoteCluster\", *v.RemoteCluster)\n\t}\n\tif v.StartEventID != nil {\n\t\tenc.AddInt64(\"startEventID\", *v.StartEventID)\n\t}\n\tif v.StartVersion != nil {\n\t\tenc.AddInt64(\"startVersion\", *v.StartVersion)\n\t}\n\tif v.EndEventID != nil {\n\t\tenc.AddInt64(\"endEventID\", *v.EndEventID)\n\t}\n\tif v.EndVersion != nil {\n\t\tenc.AddInt64(\"endVersion\", *v.EndVersion)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetRemoteCluster returns the value of RemoteCluster if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetRemoteCluster() (o string) {\n\tif v != nil && v.RemoteCluster != nil {\n\t\treturn *v.RemoteCluster\n\t}\n\n\treturn\n}\n\n// IsSetRemoteCluster returns true if RemoteCluster is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetRemoteCluster() bool {\n\treturn v != nil && v.RemoteCluster != nil\n}\n\n// GetStartEventID returns the value of StartEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetStartEventID() (o int64) {\n\tif v != nil && v.StartEventID != nil {\n\t\treturn *v.StartEventID\n\t}\n\n\treturn\n}\n\n// IsSetStartEventID returns true if StartEventID is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetStartEventID() bool {\n\treturn v != nil && v.StartEventID != nil\n}\n\n// GetStartVersion returns the value of StartVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetStartVersion() (o int64) {\n\tif v != nil && v.StartVersion != nil {\n\t\treturn *v.StartVersion\n\t}\n\n\treturn\n}\n\n// IsSetStartVersion returns true if StartVersion is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetStartVersion() bool {\n\treturn v != nil && v.StartVersion != nil\n}\n\n// GetEndEventID returns the value of EndEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetEndEventID() (o int64) {\n\tif v != nil && v.EndEventID != nil {\n\t\treturn *v.EndEventID\n\t}\n\n\treturn\n}\n\n// IsSetEndEventID returns true if EndEventID is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetEndEventID() bool {\n\treturn v != nil && v.EndEventID != nil\n}\n\n// GetEndVersion returns the value of EndVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ResendReplicationTasksRequest) GetEndVersion() (o int64) {\n\tif v != nil && v.EndVersion != nil {\n\t\treturn *v.EndVersion\n\t}\n\n\treturn\n}\n\n// IsSetEndVersion returns true if EndVersion is not nil.\nfunc (v *ResendReplicationTasksRequest) IsSetEndVersion() bool {\n\treturn v != nil && v.EndVersion != nil\n}\n\ntype RestoreDynamicConfigRequest struct {\n\tConfigName *string                       `json:\"configName,omitempty\"`\n\tFilters    []*config.DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\n// ToWire translates a RestoreDynamicConfigRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RestoreDynamicConfigRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ConfigName != nil {\n\t\tw, err = wire.NewValueString(*(v.ConfigName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigFilter_ValueList(v.Filters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RestoreDynamicConfigRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RestoreDynamicConfigRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RestoreDynamicConfigRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RestoreDynamicConfigRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ConfigName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Filters, err = _List_DynamicConfigFilter_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RestoreDynamicConfigRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RestoreDynamicConfigRequest struct could not be encoded.\nfunc (v *RestoreDynamicConfigRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ConfigName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ConfigName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Filters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigFilter_Encode(v.Filters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RestoreDynamicConfigRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RestoreDynamicConfigRequest struct could not be generated from the wire\n// representation.\nfunc (v *RestoreDynamicConfigRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ConfigName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Filters, err = _List_DynamicConfigFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RestoreDynamicConfigRequest\n// struct.\nfunc (v *RestoreDynamicConfigRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ConfigName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigName: %v\", *(v.ConfigName))\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Filters: %v\", v.Filters)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RestoreDynamicConfigRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RestoreDynamicConfigRequest match the\n// provided RestoreDynamicConfigRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RestoreDynamicConfigRequest) Equals(rhs *RestoreDynamicConfigRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ConfigName, rhs.ConfigName) {\n\t\treturn false\n\t}\n\tif !((v.Filters == nil && rhs.Filters == nil) || (v.Filters != nil && rhs.Filters != nil && _List_DynamicConfigFilter_Equals(v.Filters, rhs.Filters))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RestoreDynamicConfigRequest.\nfunc (v *RestoreDynamicConfigRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ConfigName != nil {\n\t\tenc.AddString(\"configName\", *v.ConfigName)\n\t}\n\tif v.Filters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"filters\", (_List_DynamicConfigFilter_Zapper)(v.Filters)))\n\t}\n\treturn err\n}\n\n// GetConfigName returns the value of ConfigName if it is set or its\n// zero value if it is unset.\nfunc (v *RestoreDynamicConfigRequest) GetConfigName() (o string) {\n\tif v != nil && v.ConfigName != nil {\n\t\treturn *v.ConfigName\n\t}\n\n\treturn\n}\n\n// IsSetConfigName returns true if ConfigName is not nil.\nfunc (v *RestoreDynamicConfigRequest) IsSetConfigName() bool {\n\treturn v != nil && v.ConfigName != nil\n}\n\n// GetFilters returns the value of Filters if it is set or its\n// zero value if it is unset.\nfunc (v *RestoreDynamicConfigRequest) GetFilters() (o []*config.DynamicConfigFilter) {\n\tif v != nil && v.Filters != nil {\n\t\treturn v.Filters\n\t}\n\n\treturn\n}\n\n// IsSetFilters returns true if Filters is not nil.\nfunc (v *RestoreDynamicConfigRequest) IsSetFilters() bool {\n\treturn v != nil && v.Filters != nil\n}\n\ntype RingInfo struct {\n\tRole        *string     `json:\"role,omitempty\"`\n\tMemberCount *int32      `json:\"memberCount,omitempty\"`\n\tMembers     []*HostInfo `json:\"members,omitempty\"`\n}\n\ntype _List_HostInfo_ValueList []*HostInfo\n\nfunc (v _List_HostInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HostInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_HostInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_HostInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_HostInfo_ValueList) Close() {}\n\n// ToWire translates a RingInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RingInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Role != nil {\n\t\tw, err = wire.NewValueString(*(v.Role)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.MemberCount != nil {\n\t\tw, err = wire.NewValueI32(*(v.MemberCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Members != nil {\n\t\tw, err = wire.NewValueList(_List_HostInfo_ValueList(v.Members)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_HostInfo_Read(l wire.ValueList) ([]*HostInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*HostInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _HostInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RingInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RingInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RingInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RingInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Role = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MemberCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Members, err = _List_HostInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_HostInfo_Encode(val []*HostInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HostInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a RingInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RingInfo struct could not be encoded.\nfunc (v *RingInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Role != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Role)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MemberCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MemberCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Members != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_HostInfo_Encode(v.Members, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_HostInfo_Decode(sr stream.Reader) ([]*HostInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*HostInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _HostInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RingInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RingInfo struct could not be generated from the wire\n// representation.\nfunc (v *RingInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Role = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MemberCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.Members, err = _List_HostInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RingInfo\n// struct.\nfunc (v *RingInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Role != nil {\n\t\tfields[i] = fmt.Sprintf(\"Role: %v\", *(v.Role))\n\t\ti++\n\t}\n\tif v.MemberCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"MemberCount: %v\", *(v.MemberCount))\n\t\ti++\n\t}\n\tif v.Members != nil {\n\t\tfields[i] = fmt.Sprintf(\"Members: %v\", v.Members)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RingInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_HostInfo_Equals(lhs, rhs []*HostInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this RingInfo match the\n// provided RingInfo.\n//\n// This function performs a deep comparison.\nfunc (v *RingInfo) Equals(rhs *RingInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Role, rhs.Role) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MemberCount, rhs.MemberCount) {\n\t\treturn false\n\t}\n\tif !((v.Members == nil && rhs.Members == nil) || (v.Members != nil && rhs.Members != nil && _List_HostInfo_Equals(v.Members, rhs.Members))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_HostInfo_Zapper []*HostInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_HostInfo_Zapper.\nfunc (l _List_HostInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RingInfo.\nfunc (v *RingInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Role != nil {\n\t\tenc.AddString(\"role\", *v.Role)\n\t}\n\tif v.MemberCount != nil {\n\t\tenc.AddInt32(\"memberCount\", *v.MemberCount)\n\t}\n\tif v.Members != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"members\", (_List_HostInfo_Zapper)(v.Members)))\n\t}\n\treturn err\n}\n\n// GetRole returns the value of Role if it is set or its\n// zero value if it is unset.\nfunc (v *RingInfo) GetRole() (o string) {\n\tif v != nil && v.Role != nil {\n\t\treturn *v.Role\n\t}\n\n\treturn\n}\n\n// IsSetRole returns true if Role is not nil.\nfunc (v *RingInfo) IsSetRole() bool {\n\treturn v != nil && v.Role != nil\n}\n\n// GetMemberCount returns the value of MemberCount if it is set or its\n// zero value if it is unset.\nfunc (v *RingInfo) GetMemberCount() (o int32) {\n\tif v != nil && v.MemberCount != nil {\n\t\treturn *v.MemberCount\n\t}\n\n\treturn\n}\n\n// IsSetMemberCount returns true if MemberCount is not nil.\nfunc (v *RingInfo) IsSetMemberCount() bool {\n\treturn v != nil && v.MemberCount != nil\n}\n\n// GetMembers returns the value of Members if it is set or its\n// zero value if it is unset.\nfunc (v *RingInfo) GetMembers() (o []*HostInfo) {\n\tif v != nil && v.Members != nil {\n\t\treturn v.Members\n\t}\n\n\treturn\n}\n\n// IsSetMembers returns true if Members is not nil.\nfunc (v *RingInfo) IsSetMembers() bool {\n\treturn v != nil && v.Members != nil\n}\n\ntype UpdateDomainAsyncWorkflowConfiguratonRequest struct {\n\tDomain        *string                            `json:\"domain,omitempty\"`\n\tConfiguration *shared.AsyncWorkflowConfiguration `json:\"configuration,omitempty\"`\n}\n\n// ToWire translates a UpdateDomainAsyncWorkflowConfiguratonRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainAsyncWorkflowConfiguratonRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainAsyncWorkflowConfiguratonRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainAsyncWorkflowConfiguratonRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _AsyncWorkflowConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainAsyncWorkflowConfiguratonRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainAsyncWorkflowConfiguratonRequest struct could not be encoded.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainAsyncWorkflowConfiguratonRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainAsyncWorkflowConfiguratonRequest struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _AsyncWorkflowConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainAsyncWorkflowConfiguratonRequest\n// struct.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDomainAsyncWorkflowConfiguratonRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainAsyncWorkflowConfiguratonRequest match the\n// provided UpdateDomainAsyncWorkflowConfiguratonRequest.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) Equals(rhs *UpdateDomainAsyncWorkflowConfiguratonRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainAsyncWorkflowConfiguratonRequest.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) GetConfiguration() (o *shared.AsyncWorkflowConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonRequest) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\ntype UpdateDomainAsyncWorkflowConfiguratonResponse struct {\n}\n\n// ToWire translates a UpdateDomainAsyncWorkflowConfiguratonResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainAsyncWorkflowConfiguratonResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainAsyncWorkflowConfiguratonResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainAsyncWorkflowConfiguratonResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainAsyncWorkflowConfiguratonResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainAsyncWorkflowConfiguratonResponse struct could not be encoded.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainAsyncWorkflowConfiguratonResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainAsyncWorkflowConfiguratonResponse struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainAsyncWorkflowConfiguratonResponse\n// struct.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"UpdateDomainAsyncWorkflowConfiguratonResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainAsyncWorkflowConfiguratonResponse match the\n// provided UpdateDomainAsyncWorkflowConfiguratonResponse.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) Equals(rhs *UpdateDomainAsyncWorkflowConfiguratonResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainAsyncWorkflowConfiguratonResponse.\nfunc (v *UpdateDomainAsyncWorkflowConfiguratonResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype UpdateDomainIsolationGroupsRequest struct {\n\tDomain          *string                             `json:\"domain,omitempty\"`\n\tIsolationGroups *shared.IsolationGroupConfiguration `json:\"isolationGroups,omitempty\"`\n}\n\n// ToWire translates a UpdateDomainIsolationGroupsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainIsolationGroupsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroups != nil {\n\t\tw, err = v.IsolationGroups.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainIsolationGroupsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainIsolationGroupsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainIsolationGroupsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainIsolationGroupsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainIsolationGroupsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainIsolationGroupsRequest struct could not be encoded.\nfunc (v *UpdateDomainIsolationGroupsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.IsolationGroups.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainIsolationGroupsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainIsolationGroupsRequest struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainIsolationGroupsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainIsolationGroupsRequest\n// struct.\nfunc (v *UpdateDomainIsolationGroupsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDomainIsolationGroupsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainIsolationGroupsRequest match the\n// provided UpdateDomainIsolationGroupsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainIsolationGroupsRequest) Equals(rhs *UpdateDomainIsolationGroupsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && v.IsolationGroups.Equals(rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainIsolationGroupsRequest.\nfunc (v *UpdateDomainIsolationGroupsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationGroups\", v.IsolationGroups))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainIsolationGroupsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *UpdateDomainIsolationGroupsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainIsolationGroupsRequest) GetIsolationGroups() (o *shared.IsolationGroupConfiguration) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *UpdateDomainIsolationGroupsRequest) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\ntype UpdateDomainIsolationGroupsResponse struct {\n}\n\n// ToWire translates a UpdateDomainIsolationGroupsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainIsolationGroupsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainIsolationGroupsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainIsolationGroupsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainIsolationGroupsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainIsolationGroupsResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainIsolationGroupsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainIsolationGroupsResponse struct could not be encoded.\nfunc (v *UpdateDomainIsolationGroupsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainIsolationGroupsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainIsolationGroupsResponse struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainIsolationGroupsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainIsolationGroupsResponse\n// struct.\nfunc (v *UpdateDomainIsolationGroupsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"UpdateDomainIsolationGroupsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainIsolationGroupsResponse match the\n// provided UpdateDomainIsolationGroupsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainIsolationGroupsResponse) Equals(rhs *UpdateDomainIsolationGroupsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainIsolationGroupsResponse.\nfunc (v *UpdateDomainIsolationGroupsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype UpdateDynamicConfigRequest struct {\n\tConfigName   *string                      `json:\"configName,omitempty\"`\n\tConfigValues []*config.DynamicConfigValue `json:\"configValues,omitempty\"`\n}\n\ntype _List_DynamicConfigValue_ValueList []*config.DynamicConfigValue\n\nfunc (v _List_DynamicConfigValue_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigValue', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigValue_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigValue_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigValue_ValueList) Close() {}\n\n// ToWire translates a UpdateDynamicConfigRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDynamicConfigRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ConfigName != nil {\n\t\tw, err = wire.NewValueString(*(v.ConfigName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ConfigValues != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigValue_ValueList(v.ConfigValues)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigValue_Read(w wire.Value) (*config.DynamicConfigValue, error) {\n\tvar v config.DynamicConfigValue\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigValue_Read(l wire.ValueList) ([]*config.DynamicConfigValue, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*config.DynamicConfigValue, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigValue_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a UpdateDynamicConfigRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDynamicConfigRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDynamicConfigRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDynamicConfigRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ConfigName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ConfigValues, err = _List_DynamicConfigValue_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigValue_Encode(val []*config.DynamicConfigValue, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*config.DynamicConfigValue', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a UpdateDynamicConfigRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDynamicConfigRequest struct could not be encoded.\nfunc (v *UpdateDynamicConfigRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ConfigName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ConfigName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ConfigValues != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigValue_Encode(v.ConfigValues, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigValue_Decode(sr stream.Reader) (*config.DynamicConfigValue, error) {\n\tvar v config.DynamicConfigValue\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigValue_Decode(sr stream.Reader) ([]*config.DynamicConfigValue, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*config.DynamicConfigValue, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigValue_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a UpdateDynamicConfigRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDynamicConfigRequest struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDynamicConfigRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ConfigName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.ConfigValues, err = _List_DynamicConfigValue_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDynamicConfigRequest\n// struct.\nfunc (v *UpdateDynamicConfigRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ConfigName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigName: %v\", *(v.ConfigName))\n\t\ti++\n\t}\n\tif v.ConfigValues != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigValues: %v\", v.ConfigValues)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDynamicConfigRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DynamicConfigValue_Equals(lhs, rhs []*config.DynamicConfigValue) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this UpdateDynamicConfigRequest match the\n// provided UpdateDynamicConfigRequest.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDynamicConfigRequest) Equals(rhs *UpdateDynamicConfigRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ConfigName, rhs.ConfigName) {\n\t\treturn false\n\t}\n\tif !((v.ConfigValues == nil && rhs.ConfigValues == nil) || (v.ConfigValues != nil && rhs.ConfigValues != nil && _List_DynamicConfigValue_Equals(v.ConfigValues, rhs.ConfigValues))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigValue_Zapper []*config.DynamicConfigValue\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigValue_Zapper.\nfunc (l _List_DynamicConfigValue_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDynamicConfigRequest.\nfunc (v *UpdateDynamicConfigRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ConfigName != nil {\n\t\tenc.AddString(\"configName\", *v.ConfigName)\n\t}\n\tif v.ConfigValues != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"configValues\", (_List_DynamicConfigValue_Zapper)(v.ConfigValues)))\n\t}\n\treturn err\n}\n\n// GetConfigName returns the value of ConfigName if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDynamicConfigRequest) GetConfigName() (o string) {\n\tif v != nil && v.ConfigName != nil {\n\t\treturn *v.ConfigName\n\t}\n\n\treturn\n}\n\n// IsSetConfigName returns true if ConfigName is not nil.\nfunc (v *UpdateDynamicConfigRequest) IsSetConfigName() bool {\n\treturn v != nil && v.ConfigName != nil\n}\n\n// GetConfigValues returns the value of ConfigValues if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDynamicConfigRequest) GetConfigValues() (o []*config.DynamicConfigValue) {\n\tif v != nil && v.ConfigValues != nil {\n\t\treturn v.ConfigValues\n\t}\n\n\treturn\n}\n\n// IsSetConfigValues returns true if ConfigValues is not nil.\nfunc (v *UpdateDynamicConfigRequest) IsSetConfigValues() bool {\n\treturn v != nil && v.ConfigValues != nil\n}\n\ntype UpdateGlobalIsolationGroupsRequest struct {\n\tIsolationGroups *shared.IsolationGroupConfiguration `json:\"isolationGroups,omitempty\"`\n}\n\n// ToWire translates a UpdateGlobalIsolationGroupsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateGlobalIsolationGroupsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IsolationGroups != nil {\n\t\tw, err = v.IsolationGroups.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateGlobalIsolationGroupsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateGlobalIsolationGroupsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateGlobalIsolationGroupsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateGlobalIsolationGroupsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateGlobalIsolationGroupsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateGlobalIsolationGroupsRequest struct could not be encoded.\nfunc (v *UpdateGlobalIsolationGroupsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.IsolationGroups.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateGlobalIsolationGroupsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateGlobalIsolationGroupsRequest struct could not be generated from the wire\n// representation.\nfunc (v *UpdateGlobalIsolationGroupsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.IsolationGroups, err = _IsolationGroupConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateGlobalIsolationGroupsRequest\n// struct.\nfunc (v *UpdateGlobalIsolationGroupsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateGlobalIsolationGroupsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateGlobalIsolationGroupsRequest match the\n// provided UpdateGlobalIsolationGroupsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateGlobalIsolationGroupsRequest) Equals(rhs *UpdateGlobalIsolationGroupsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && v.IsolationGroups.Equals(rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateGlobalIsolationGroupsRequest.\nfunc (v *UpdateGlobalIsolationGroupsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationGroups\", v.IsolationGroups))\n\t}\n\treturn err\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateGlobalIsolationGroupsRequest) GetIsolationGroups() (o *shared.IsolationGroupConfiguration) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *UpdateGlobalIsolationGroupsRequest) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\ntype UpdateGlobalIsolationGroupsResponse struct {\n}\n\n// ToWire translates a UpdateGlobalIsolationGroupsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateGlobalIsolationGroupsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateGlobalIsolationGroupsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateGlobalIsolationGroupsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateGlobalIsolationGroupsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateGlobalIsolationGroupsResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateGlobalIsolationGroupsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateGlobalIsolationGroupsResponse struct could not be encoded.\nfunc (v *UpdateGlobalIsolationGroupsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateGlobalIsolationGroupsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateGlobalIsolationGroupsResponse struct could not be generated from the wire\n// representation.\nfunc (v *UpdateGlobalIsolationGroupsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateGlobalIsolationGroupsResponse\n// struct.\nfunc (v *UpdateGlobalIsolationGroupsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"UpdateGlobalIsolationGroupsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateGlobalIsolationGroupsResponse match the\n// provided UpdateGlobalIsolationGroupsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateGlobalIsolationGroupsResponse) Equals(rhs *UpdateGlobalIsolationGroupsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateGlobalIsolationGroupsResponse.\nfunc (v *UpdateGlobalIsolationGroupsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"admin\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/admin\",\n\tFilePath: \"admin.thrift\",\n\tSHA1:     \"cd332257c2e6f1bcc07bdfaae63cb757ecf2cba3\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tconfig.ThriftModule,\n\t\treplicator.ThriftModule,\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.admin\\n\\ninclude \\\"shared.thrift\\\"\\ninclude \\\"replicator.thrift\\\"\\ninclude \\\"config.thrift\\\"\\n\\n/**\\n* AdminService provides advanced APIs for debugging and analysis with admin privilege\\n**/\\nservice AdminService {\\n  /**\\n  * DescribeWorkflowExecution returns information about the internal states of workflow execution.\\n  **/\\n  DescribeWorkflowExecutionResponse DescribeWorkflowExecution(1: DescribeWorkflowExecutionRequest request)\\n    throws (\\n      1: shared.BadRequestError         badRequestError,\\n      2: shared.InternalServiceError    internalServiceError,\\n      3: shared.EntityNotExistsError    entityNotExistError,\\n      4: shared.AccessDeniedError       accessDeniedError,\\n    )\\n\\n  /**\\n  * DescribeShardDistribution returns information about history shards within the cluster\\n  **/\\n  shared.DescribeShardDistributionResponse DescribeShardDistribution(1: shared.DescribeShardDistributionRequest request)\\n    throws (\\n      1: shared.InternalServiceError internalServiceError,\\n    )\\n\\n  /**\\n  * DescribeHistoryHost returns information about the internal states of a history host\\n  **/\\n  shared.DescribeHistoryHostResponse DescribeHistoryHost(1: shared.DescribeHistoryHostRequest request)\\n    throws (\\n      1: shared.BadRequestError       badRequestError,\\n      2: shared.InternalServiceError  internalServiceError,\\n      3: shared.AccessDeniedError     accessDeniedError,\\n    )\\n\\n  void CloseShard(1: shared.CloseShardRequest request)\\n    throws (\\n      1: shared.BadRequestError       badRequestError,\\n      2: shared.InternalServiceError  internalServiceError,\\n      3: shared.AccessDeniedError     accessDeniedError,\\n    )\\n\\n  void RemoveTask(1: shared.RemoveTaskRequest request)\\n    throws (\\n      1: shared.BadRequestError       badRequestError,\\n      2: shared.InternalServiceError  internalServiceError,\\n      3: shared.AccessDeniedError     accessDeniedError,\\n    )\\n\\n  void ResetQueue(1: shared.ResetQueueRequest request)\\n    throws (\\n      1: shared.BadRequestError       badRequestError,\\n      2: shared.InternalServiceError  internalServiceError,\\n      3: shared.AccessDeniedError     accessDeniedError,\\n    )\\n\\n  shared.DescribeQueueResponse DescribeQueue(1: shared.DescribeQueueRequest request)\\n    throws (\\n      1: shared.BadRequestError       badRequestError,\\n      2: shared.InternalServiceError  internalServiceError,\\n      3: shared.AccessDeniedError     accessDeniedError,\\n    )\\n\\n  /**\\n  * Returns the raw history of specified workflow execution.  It fails with 'EntityNotExistError' if speficied workflow\\n  * execution in unknown to the service.\\n  * StartEventId defines the beginning of the event to fetch. The first event is inclusive.\\n  * EndEventId and EndEventVersion defines the end of the event to fetch. The end event is exclusive.\\n  **/\\n  GetWorkflowExecutionRawHistoryV2Response GetWorkflowExecutionRawHistoryV2(1: GetWorkflowExecutionRawHistoryV2Request getRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  replicator.GetReplicationMessagesResponse GetReplicationMessages(1: replicator.GetReplicationMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n    )\\n\\n  replicator.GetDomainReplicationMessagesResponse GetDomainReplicationMessages(1: replicator.GetDomainReplicationMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n    )\\n\\n  replicator.GetDLQReplicationMessagesResponse GetDLQReplicationMessages(1: replicator.GetDLQReplicationMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * ReapplyEvents applies stale events to the current workflow and current run\\n  **/\\n  void ReapplyEvents(1: shared.ReapplyEventsRequest reapplyEventsRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.DomainNotActiveError domainNotActiveError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * AddSearchAttribute whitelist search attribute in request.\\n  **/\\n  void AddSearchAttribute(1: AddSearchAttributeRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * DescribeCluster returns information about cadence cluster\\n  **/\\n  DescribeClusterResponse DescribeCluster()\\n    throws (\\n      1: shared.InternalServiceError internalServiceError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * ReadDLQMessages returns messages from DLQ\\n  **/\\n  replicator.ReadDLQMessagesResponse ReadDLQMessages(1: replicator.ReadDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * PurgeDLQMessages purges messages from DLQ\\n  **/\\n  void PurgeDLQMessages(1: replicator.PurgeDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * MergeDLQMessages merges messages from DLQ\\n  **/\\n  replicator.MergeDLQMessagesResponse MergeDLQMessages(1: replicator.MergeDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * RefreshWorkflowTasks refreshes all tasks of a workflow\\n  **/\\n  void RefreshWorkflowTasks(1: shared.RefreshWorkflowTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.DomainNotActiveError domainNotActiveError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * ResendReplicationTasks requests replication tasks from remote cluster and apply tasks to current cluster\\n  **/\\n  void ResendReplicationTasks(1: ResendReplicationTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * GetCrossClusterTasks fetches cross cluster tasks\\n  **/\\n  shared.GetCrossClusterTasksResponse GetCrossClusterTasks(1: shared.GetCrossClusterTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * RespondCrossClusterTasksCompleted responds the result of processing cross cluster tasks\\n  **/\\n  shared.RespondCrossClusterTasksCompletedResponse RespondCrossClusterTasksCompleted(1: shared.RespondCrossClusterTasksCompletedRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * GetDynamicConfig returns values associated with a specified dynamic config parameter.\\n  **/\\n  GetDynamicConfigResponse GetDynamicConfig(1: GetDynamicConfigRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n    )\\n\\n  void UpdateDynamicConfig(1: UpdateDynamicConfigRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n    )\\n\\n  void RestoreDynamicConfig(1: RestoreDynamicConfigRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n    )\\n\\n  ListDynamicConfigResponse ListDynamicConfig(1: ListDynamicConfigRequest request)\\n    throws (\\n      1: shared.InternalServiceError internalServiceError,\\n    )\\n\\n  AdminDeleteWorkflowResponse DeleteWorkflow(1: AdminDeleteWorkflowRequest request)\\n    throws (\\n      1: shared.BadRequestError         badRequestError,\\n      2: shared.EntityNotExistsError    entityNotExistError,\\n      3: shared.InternalServiceError    internalServiceError,\\n    )\\n\\n  AdminMaintainWorkflowResponse MaintainCorruptWorkflow(1: AdminMaintainWorkflowRequest request)\\n    throws (\\n      1: shared.BadRequestError         badRequestError,\\n      2: shared.EntityNotExistsError    entityNotExistError,\\n      3: shared.InternalServiceError    internalServiceError,\\n    )\\n\\n  GetGlobalIsolationGroupsResponse GetGlobalIsolationGroups(1: GetGlobalIsolationGroupsRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n\\n  UpdateGlobalIsolationGroupsResponse UpdateGlobalIsolationGroups(1: UpdateGlobalIsolationGroupsRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n\\n  GetDomainIsolationGroupsResponse GetDomainIsolationGroups(1: GetDomainIsolationGroupsRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n\\n  UpdateDomainIsolationGroupsResponse UpdateDomainIsolationGroups(1: UpdateDomainIsolationGroupsRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n\\n\\n  GetDomainAsyncWorkflowConfiguratonResponse GetDomainAsyncWorkflowConfiguraton(1: GetDomainAsyncWorkflowConfiguratonRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n\\n  UpdateDomainAsyncWorkflowConfiguratonResponse UpdateDomainAsyncWorkflowConfiguraton(1: UpdateDomainAsyncWorkflowConfiguratonRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n    )\\n}\\n\\nstruct DescribeWorkflowExecutionRequest {\\n  10: optional string                       domain\\n  20: optional shared.WorkflowExecution     execution\\n}\\n\\nstruct DescribeWorkflowExecutionResponse {\\n  10: optional string shardId\\n  20: optional string historyAddr\\n  40: optional string mutableStateInCache\\n  50: optional string mutableStateInDatabase\\n}\\n\\n/**\\n  * StartEventId defines the beginning of the event to fetch. The first event is exclusive.\\n  * EndEventId and EndEventVersion defines the end of the event to fetch. The end event is exclusive.\\n  **/\\nstruct GetWorkflowExecutionRawHistoryV2Request {\\n  10: optional string domain\\n  20: optional shared.WorkflowExecution execution\\n  30: optional i64 (js.type = \\\"Long\\\") startEventId\\n  40: optional i64 (js.type = \\\"Long\\\") startEventVersion\\n  50: optional i64 (js.type = \\\"Long\\\") endEventId\\n  60: optional i64 (js.type = \\\"Long\\\") endEventVersion\\n  70: optional i32 maximumPageSize\\n  80: optional binary nextPageToken\\n}\\n\\nstruct GetWorkflowExecutionRawHistoryV2Response {\\n  10: optional binary nextPageToken\\n  20: optional list<shared.DataBlob> historyBatches\\n  30: optional shared.VersionHistory versionHistory\\n}\\n\\nstruct AddSearchAttributeRequest {\\n  10: optional map<string, shared.IndexedValueType> searchAttribute\\n  20: optional string securityToken\\n}\\n\\nstruct HostInfo {\\n  10: optional string Identity\\n}\\n\\nstruct RingInfo {\\n  10: optional string role\\n  20: optional i32 memberCount\\n  30: optional list<HostInfo> members\\n}\\n\\nstruct MembershipInfo {\\n  10: optional HostInfo currentHost\\n  20: optional list<string> reachableMembers\\n  30: optional list<RingInfo> rings\\n}\\n\\nstruct PersistenceSetting {\\n  10: optional string key\\n  20: optional string value\\n}\\n\\nstruct PersistenceFeature {\\n  10: optional string key\\n  20: optional bool enabled\\n}\\n\\nstruct PersistenceInfo {\\n  10: optional string backend\\n  20: optional list<PersistenceSetting> settings\\n  30: optional list<PersistenceFeature> features\\n}\\n\\nstruct DescribeClusterResponse {\\n  10: optional shared.SupportedClientVersions supportedClientVersions\\n  20: optional MembershipInfo membershipInfo\\n  30: optional map<string,PersistenceInfo> persistenceInfo\\n}\\n\\nstruct ResendReplicationTasksRequest {\\n  10: optional string domainID\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional string remoteCluster\\n  50: optional i64 (js.type = \\\"Long\\\") startEventID\\n  60: optional i64 (js.type = \\\"Long\\\") startVersion\\n  70: optional i64 (js.type = \\\"Long\\\") endEventID\\n  80: optional i64 (js.type = \\\"Long\\\") endVersion\\n}\\n\\nstruct GetDynamicConfigRequest {\\n  10: optional string configName\\n  20: optional list<config.DynamicConfigFilter> filters\\n}\\n\\nstruct GetDynamicConfigResponse {\\n  10: optional shared.DataBlob value\\n}\\n\\nstruct UpdateDynamicConfigRequest {\\n  10: optional string configName\\n  20: optional list<config.DynamicConfigValue> configValues\\n}\\n\\nstruct RestoreDynamicConfigRequest {\\n  10: optional string configName\\n  20: optional list<config.DynamicConfigFilter> filters\\n}\\n\\nstruct AdminDeleteWorkflowRequest {\\n  10: optional string                       domain\\n  20: optional shared.WorkflowExecution     execution\\n}\\n\\nstruct AdminDeleteWorkflowResponse {\\n  10: optional bool historyDeleted\\n  20: optional bool executionsDeleted\\n  30: optional bool visibilityDeleted\\n}\\n\\nstruct AdminMaintainWorkflowRequest {\\n  10: optional string                       domain\\n  20: optional shared.WorkflowExecution     execution\\n}\\n\\nstruct AdminMaintainWorkflowResponse {\\n  10: optional bool historyDeleted\\n  20: optional bool executionsDeleted\\n  30: optional bool visibilityDeleted\\n}\\n\\n//Eventually remove configName and integrate this functionality into Get.\\n//GetDynamicConfigResponse would need to change as well.\\nstruct ListDynamicConfigRequest {\\n  10: optional string configName\\n}\\n\\nstruct ListDynamicConfigResponse {\\n  10: optional list<config.DynamicConfigEntry> entries\\n}\\n\\n// global\\nstruct GetGlobalIsolationGroupsRequest{}\\n\\nstruct GetGlobalIsolationGroupsResponse{\\n    10: optional shared.IsolationGroupConfiguration isolationGroups\\n}\\n\\nstruct UpdateGlobalIsolationGroupsRequest{\\n    10: optional shared.IsolationGroupConfiguration isolationGroups\\n}\\n\\nstruct UpdateGlobalIsolationGroupsResponse{}\\n\\n\\n// For domains\\nstruct GetDomainIsolationGroupsRequest{\\n    10: optional string domain\\n}\\n\\nstruct GetDomainIsolationGroupsResponse{\\n    10: optional shared.IsolationGroupConfiguration isolationGroups\\n}\\n\\nstruct UpdateDomainIsolationGroupsRequest{\\n    10: optional string domain\\n    20: optional shared.IsolationGroupConfiguration isolationGroups\\n}\\n\\nstruct UpdateDomainIsolationGroupsResponse{}\\n\\n// Async workflow configuration request/response payloads\\nstruct GetDomainAsyncWorkflowConfiguratonRequest {\\n    10: optional string domain\\n}\\n\\nstruct GetDomainAsyncWorkflowConfiguratonResponse {\\n    10: optional shared.AsyncWorkflowConfiguration configuration\\n}\\n\\nstruct UpdateDomainAsyncWorkflowConfiguratonRequest {\\n    10: optional string domain\\n    20: optional shared.AsyncWorkflowConfiguration configuration\\n}\\n\\nstruct UpdateDomainAsyncWorkflowConfiguratonResponse {}\\n\"\n\n// AdminService_AddSearchAttribute_Args represents the arguments for the AdminService.AddSearchAttribute function.\n//\n// The arguments for AddSearchAttribute are sent and received over the wire as this struct.\ntype AdminService_AddSearchAttribute_Args struct {\n\tRequest *AddSearchAttributeRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_AddSearchAttribute_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_AddSearchAttribute_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AddSearchAttributeRequest_Read(w wire.Value) (*AddSearchAttributeRequest, error) {\n\tvar v AddSearchAttributeRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_AddSearchAttribute_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_AddSearchAttribute_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_AddSearchAttribute_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_AddSearchAttribute_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _AddSearchAttributeRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_AddSearchAttribute_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_AddSearchAttribute_Args struct could not be encoded.\nfunc (v *AdminService_AddSearchAttribute_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AddSearchAttributeRequest_Decode(sr stream.Reader) (*AddSearchAttributeRequest, error) {\n\tvar v AddSearchAttributeRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_AddSearchAttribute_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_AddSearchAttribute_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_AddSearchAttribute_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _AddSearchAttributeRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_AddSearchAttribute_Args\n// struct.\nfunc (v *AdminService_AddSearchAttribute_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_AddSearchAttribute_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_AddSearchAttribute_Args match the\n// provided AdminService_AddSearchAttribute_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_AddSearchAttribute_Args) Equals(rhs *AdminService_AddSearchAttribute_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_AddSearchAttribute_Args.\nfunc (v *AdminService_AddSearchAttribute_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_AddSearchAttribute_Args) GetRequest() (o *AddSearchAttributeRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_AddSearchAttribute_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"AddSearchAttribute\" for this struct.\nfunc (v *AdminService_AddSearchAttribute_Args) MethodName() string {\n\treturn \"AddSearchAttribute\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_AddSearchAttribute_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_AddSearchAttribute_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.AddSearchAttribute\n// function.\nvar AdminService_AddSearchAttribute_Helper = struct {\n\t// Args accepts the parameters of AddSearchAttribute in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *AddSearchAttributeRequest,\n\t) *AdminService_AddSearchAttribute_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by AddSearchAttribute.\n\t//\n\t// An error can be thrown by AddSearchAttribute only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for AddSearchAttribute\n\t// given the error returned by it. The provided error may\n\t// be nil if AddSearchAttribute did not fail.\n\t//\n\t// This allows mapping errors returned by AddSearchAttribute into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// AddSearchAttribute\n\t//\n\t//   err := AddSearchAttribute(args)\n\t//   result, err := AdminService_AddSearchAttribute_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from AddSearchAttribute: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_AddSearchAttribute_Result, error)\n\n\t// UnwrapResponse takes the result struct for AddSearchAttribute\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if AddSearchAttribute threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_AddSearchAttribute_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_AddSearchAttribute_Result) error\n}{}\n\nfunc init() {\n\tAdminService_AddSearchAttribute_Helper.Args = func(\n\t\trequest *AddSearchAttributeRequest,\n\t) *AdminService_AddSearchAttribute_Args {\n\t\treturn &AdminService_AddSearchAttribute_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_AddSearchAttribute_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_AddSearchAttribute_Helper.WrapResponse = func(err error) (*AdminService_AddSearchAttribute_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_AddSearchAttribute_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_AddSearchAttribute_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_AddSearchAttribute_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_AddSearchAttribute_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_AddSearchAttribute_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_AddSearchAttribute_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_AddSearchAttribute_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_AddSearchAttribute_Helper.UnwrapResponse = func(result *AdminService_AddSearchAttribute_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_AddSearchAttribute_Result represents the result of a AdminService.AddSearchAttribute function call.\n//\n// The result of a AddSearchAttribute execution is sent and received over the wire as this struct.\ntype AdminService_AddSearchAttribute_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_AddSearchAttribute_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_AddSearchAttribute_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_AddSearchAttribute_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _BadRequestError_Read(w wire.Value) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Read(w wire.Value) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Read(w wire.Value) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_AddSearchAttribute_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_AddSearchAttribute_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_AddSearchAttribute_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_AddSearchAttribute_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_AddSearchAttribute_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_AddSearchAttribute_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_AddSearchAttribute_Result struct could not be encoded.\nfunc (v *AdminService_AddSearchAttribute_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_AddSearchAttribute_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _BadRequestError_Decode(sr stream.Reader) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Decode(sr stream.Reader) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Decode(sr stream.Reader) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_AddSearchAttribute_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_AddSearchAttribute_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_AddSearchAttribute_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_AddSearchAttribute_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_AddSearchAttribute_Result\n// struct.\nfunc (v *AdminService_AddSearchAttribute_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_AddSearchAttribute_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_AddSearchAttribute_Result match the\n// provided AdminService_AddSearchAttribute_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_AddSearchAttribute_Result) Equals(rhs *AdminService_AddSearchAttribute_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_AddSearchAttribute_Result.\nfunc (v *AdminService_AddSearchAttribute_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_AddSearchAttribute_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_AddSearchAttribute_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_AddSearchAttribute_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_AddSearchAttribute_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_AddSearchAttribute_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_AddSearchAttribute_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"AddSearchAttribute\" for this struct.\nfunc (v *AdminService_AddSearchAttribute_Result) MethodName() string {\n\treturn \"AddSearchAttribute\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_AddSearchAttribute_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_CloseShard_Args represents the arguments for the AdminService.CloseShard function.\n//\n// The arguments for CloseShard are sent and received over the wire as this struct.\ntype AdminService_CloseShard_Args struct {\n\tRequest *shared.CloseShardRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_CloseShard_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_CloseShard_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CloseShardRequest_Read(w wire.Value) (*shared.CloseShardRequest, error) {\n\tvar v shared.CloseShardRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_CloseShard_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_CloseShard_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_CloseShard_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_CloseShard_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _CloseShardRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_CloseShard_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_CloseShard_Args struct could not be encoded.\nfunc (v *AdminService_CloseShard_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CloseShardRequest_Decode(sr stream.Reader) (*shared.CloseShardRequest, error) {\n\tvar v shared.CloseShardRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_CloseShard_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_CloseShard_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_CloseShard_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _CloseShardRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_CloseShard_Args\n// struct.\nfunc (v *AdminService_CloseShard_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_CloseShard_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_CloseShard_Args match the\n// provided AdminService_CloseShard_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_CloseShard_Args) Equals(rhs *AdminService_CloseShard_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_CloseShard_Args.\nfunc (v *AdminService_CloseShard_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_CloseShard_Args) GetRequest() (o *shared.CloseShardRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_CloseShard_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"CloseShard\" for this struct.\nfunc (v *AdminService_CloseShard_Args) MethodName() string {\n\treturn \"CloseShard\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_CloseShard_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_CloseShard_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.CloseShard\n// function.\nvar AdminService_CloseShard_Helper = struct {\n\t// Args accepts the parameters of CloseShard in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.CloseShardRequest,\n\t) *AdminService_CloseShard_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by CloseShard.\n\t//\n\t// An error can be thrown by CloseShard only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for CloseShard\n\t// given the error returned by it. The provided error may\n\t// be nil if CloseShard did not fail.\n\t//\n\t// This allows mapping errors returned by CloseShard into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// CloseShard\n\t//\n\t//   err := CloseShard(args)\n\t//   result, err := AdminService_CloseShard_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from CloseShard: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_CloseShard_Result, error)\n\n\t// UnwrapResponse takes the result struct for CloseShard\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if CloseShard threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_CloseShard_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_CloseShard_Result) error\n}{}\n\nfunc init() {\n\tAdminService_CloseShard_Helper.Args = func(\n\t\trequest *shared.CloseShardRequest,\n\t) *AdminService_CloseShard_Args {\n\t\treturn &AdminService_CloseShard_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_CloseShard_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_CloseShard_Helper.WrapResponse = func(err error) (*AdminService_CloseShard_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_CloseShard_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_CloseShard_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_CloseShard_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_CloseShard_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_CloseShard_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_CloseShard_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_CloseShard_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_CloseShard_Helper.UnwrapResponse = func(result *AdminService_CloseShard_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_CloseShard_Result represents the result of a AdminService.CloseShard function call.\n//\n// The result of a CloseShard execution is sent and received over the wire as this struct.\ntype AdminService_CloseShard_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_CloseShard_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_CloseShard_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_CloseShard_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AccessDeniedError_Read(w wire.Value) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_CloseShard_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_CloseShard_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_CloseShard_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_CloseShard_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_CloseShard_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_CloseShard_Result struct could not be encoded.\nfunc (v *AdminService_CloseShard_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AccessDeniedError_Decode(sr stream.Reader) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_CloseShard_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_CloseShard_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_CloseShard_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_CloseShard_Result\n// struct.\nfunc (v *AdminService_CloseShard_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_CloseShard_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_CloseShard_Result match the\n// provided AdminService_CloseShard_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_CloseShard_Result) Equals(rhs *AdminService_CloseShard_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_CloseShard_Result.\nfunc (v *AdminService_CloseShard_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_CloseShard_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_CloseShard_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_CloseShard_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_CloseShard_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_CloseShard_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_CloseShard_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"CloseShard\" for this struct.\nfunc (v *AdminService_CloseShard_Result) MethodName() string {\n\treturn \"CloseShard\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_CloseShard_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DeleteWorkflow_Args represents the arguments for the AdminService.DeleteWorkflow function.\n//\n// The arguments for DeleteWorkflow are sent and received over the wire as this struct.\ntype AdminService_DeleteWorkflow_Args struct {\n\tRequest *AdminDeleteWorkflowRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_DeleteWorkflow_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DeleteWorkflow_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AdminDeleteWorkflowRequest_Read(w wire.Value) (*AdminDeleteWorkflowRequest, error) {\n\tvar v AdminDeleteWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DeleteWorkflow_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DeleteWorkflow_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DeleteWorkflow_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DeleteWorkflow_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _AdminDeleteWorkflowRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DeleteWorkflow_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DeleteWorkflow_Args struct could not be encoded.\nfunc (v *AdminService_DeleteWorkflow_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AdminDeleteWorkflowRequest_Decode(sr stream.Reader) (*AdminDeleteWorkflowRequest, error) {\n\tvar v AdminDeleteWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DeleteWorkflow_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DeleteWorkflow_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DeleteWorkflow_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _AdminDeleteWorkflowRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DeleteWorkflow_Args\n// struct.\nfunc (v *AdminService_DeleteWorkflow_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DeleteWorkflow_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DeleteWorkflow_Args match the\n// provided AdminService_DeleteWorkflow_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DeleteWorkflow_Args) Equals(rhs *AdminService_DeleteWorkflow_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DeleteWorkflow_Args.\nfunc (v *AdminService_DeleteWorkflow_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DeleteWorkflow_Args) GetRequest() (o *AdminDeleteWorkflowRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_DeleteWorkflow_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DeleteWorkflow\" for this struct.\nfunc (v *AdminService_DeleteWorkflow_Args) MethodName() string {\n\treturn \"DeleteWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DeleteWorkflow_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DeleteWorkflow_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DeleteWorkflow\n// function.\nvar AdminService_DeleteWorkflow_Helper = struct {\n\t// Args accepts the parameters of DeleteWorkflow in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *AdminDeleteWorkflowRequest,\n\t) *AdminService_DeleteWorkflow_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DeleteWorkflow.\n\t//\n\t// An error can be thrown by DeleteWorkflow only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DeleteWorkflow\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DeleteWorkflow into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DeleteWorkflow\n\t//\n\t//   value, err := DeleteWorkflow(args)\n\t//   result, err := AdminService_DeleteWorkflow_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DeleteWorkflow: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*AdminDeleteWorkflowResponse, error) (*AdminService_DeleteWorkflow_Result, error)\n\n\t// UnwrapResponse takes the result struct for DeleteWorkflow\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DeleteWorkflow threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DeleteWorkflow_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DeleteWorkflow_Result) (*AdminDeleteWorkflowResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DeleteWorkflow_Helper.Args = func(\n\t\trequest *AdminDeleteWorkflowRequest,\n\t) *AdminService_DeleteWorkflow_Args {\n\t\treturn &AdminService_DeleteWorkflow_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_DeleteWorkflow_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DeleteWorkflow_Helper.WrapResponse = func(success *AdminDeleteWorkflowResponse, err error) (*AdminService_DeleteWorkflow_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DeleteWorkflow_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DeleteWorkflow_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_DeleteWorkflow_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DeleteWorkflow_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_DeleteWorkflow_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DeleteWorkflow_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DeleteWorkflow_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DeleteWorkflow_Helper.UnwrapResponse = func(result *AdminService_DeleteWorkflow_Result) (success *AdminDeleteWorkflowResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DeleteWorkflow_Result represents the result of a AdminService.DeleteWorkflow function call.\n//\n// The result of a DeleteWorkflow execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DeleteWorkflow_Result struct {\n\t// Value returned by DeleteWorkflow after a successful execution.\n\tSuccess              *AdminDeleteWorkflowResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DeleteWorkflow_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DeleteWorkflow_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DeleteWorkflow_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AdminDeleteWorkflowResponse_Read(w wire.Value) (*AdminDeleteWorkflowResponse, error) {\n\tvar v AdminDeleteWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Read(w wire.Value) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DeleteWorkflow_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DeleteWorkflow_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DeleteWorkflow_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DeleteWorkflow_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _AdminDeleteWorkflowResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DeleteWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DeleteWorkflow_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DeleteWorkflow_Result struct could not be encoded.\nfunc (v *AdminService_DeleteWorkflow_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DeleteWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AdminDeleteWorkflowResponse_Decode(sr stream.Reader) (*AdminDeleteWorkflowResponse, error) {\n\tvar v AdminDeleteWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Decode(sr stream.Reader) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DeleteWorkflow_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DeleteWorkflow_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DeleteWorkflow_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _AdminDeleteWorkflowResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DeleteWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DeleteWorkflow_Result\n// struct.\nfunc (v *AdminService_DeleteWorkflow_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DeleteWorkflow_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DeleteWorkflow_Result match the\n// provided AdminService_DeleteWorkflow_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DeleteWorkflow_Result) Equals(rhs *AdminService_DeleteWorkflow_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DeleteWorkflow_Result.\nfunc (v *AdminService_DeleteWorkflow_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DeleteWorkflow_Result) GetSuccess() (o *AdminDeleteWorkflowResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DeleteWorkflow_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DeleteWorkflow_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_DeleteWorkflow_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DeleteWorkflow_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_DeleteWorkflow_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DeleteWorkflow_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DeleteWorkflow_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DeleteWorkflow\" for this struct.\nfunc (v *AdminService_DeleteWorkflow_Result) MethodName() string {\n\treturn \"DeleteWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DeleteWorkflow_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DescribeCluster_Args represents the arguments for the AdminService.DescribeCluster function.\n//\n// The arguments for DescribeCluster are sent and received over the wire as this struct.\ntype AdminService_DescribeCluster_Args struct {\n}\n\n// ToWire translates a AdminService_DescribeCluster_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeCluster_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_DescribeCluster_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeCluster_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeCluster_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeCluster_Args) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeCluster_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeCluster_Args struct could not be encoded.\nfunc (v *AdminService_DescribeCluster_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_DescribeCluster_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeCluster_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeCluster_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeCluster_Args\n// struct.\nfunc (v *AdminService_DescribeCluster_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"AdminService_DescribeCluster_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeCluster_Args match the\n// provided AdminService_DescribeCluster_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeCluster_Args) Equals(rhs *AdminService_DescribeCluster_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeCluster_Args.\nfunc (v *AdminService_DescribeCluster_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeCluster\" for this struct.\nfunc (v *AdminService_DescribeCluster_Args) MethodName() string {\n\treturn \"DescribeCluster\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DescribeCluster_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DescribeCluster_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DescribeCluster\n// function.\nvar AdminService_DescribeCluster_Helper = struct {\n\t// Args accepts the parameters of DescribeCluster in-order and returns\n\t// the arguments struct for the function.\n\tArgs func() *AdminService_DescribeCluster_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeCluster.\n\t//\n\t// An error can be thrown by DescribeCluster only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeCluster\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeCluster into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeCluster\n\t//\n\t//   value, err := DescribeCluster(args)\n\t//   result, err := AdminService_DescribeCluster_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeCluster: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*DescribeClusterResponse, error) (*AdminService_DescribeCluster_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeCluster\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeCluster threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DescribeCluster_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DescribeCluster_Result) (*DescribeClusterResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DescribeCluster_Helper.Args = func() *AdminService_DescribeCluster_Args {\n\t\treturn &AdminService_DescribeCluster_Args{}\n\t}\n\n\tAdminService_DescribeCluster_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DescribeCluster_Helper.WrapResponse = func(success *DescribeClusterResponse, err error) (*AdminService_DescribeCluster_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DescribeCluster_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeCluster_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeCluster_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeCluster_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeCluster_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DescribeCluster_Helper.UnwrapResponse = func(result *AdminService_DescribeCluster_Result) (success *DescribeClusterResponse, err error) {\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DescribeCluster_Result represents the result of a AdminService.DescribeCluster function call.\n//\n// The result of a DescribeCluster execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DescribeCluster_Result struct {\n\t// Value returned by DescribeCluster after a successful execution.\n\tSuccess              *DescribeClusterResponse     `json:\"success,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeCluster_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeCluster_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DescribeCluster_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeClusterResponse_Read(w wire.Value) (*DescribeClusterResponse, error) {\n\tvar v DescribeClusterResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeCluster_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeCluster_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeCluster_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeCluster_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeClusterResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeCluster_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeCluster_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeCluster_Result struct could not be encoded.\nfunc (v *AdminService_DescribeCluster_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeCluster_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeClusterResponse_Decode(sr stream.Reader) (*DescribeClusterResponse, error) {\n\tvar v DescribeClusterResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeCluster_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeCluster_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeCluster_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeClusterResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeCluster_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeCluster_Result\n// struct.\nfunc (v *AdminService_DescribeCluster_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeCluster_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeCluster_Result match the\n// provided AdminService_DescribeCluster_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeCluster_Result) Equals(rhs *AdminService_DescribeCluster_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeCluster_Result.\nfunc (v *AdminService_DescribeCluster_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeCluster_Result) GetSuccess() (o *DescribeClusterResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DescribeCluster_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeCluster_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DescribeCluster_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeCluster_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_DescribeCluster_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeCluster\" for this struct.\nfunc (v *AdminService_DescribeCluster_Result) MethodName() string {\n\treturn \"DescribeCluster\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DescribeCluster_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DescribeHistoryHost_Args represents the arguments for the AdminService.DescribeHistoryHost function.\n//\n// The arguments for DescribeHistoryHost are sent and received over the wire as this struct.\ntype AdminService_DescribeHistoryHost_Args struct {\n\tRequest *shared.DescribeHistoryHostRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeHistoryHost_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeHistoryHost_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeHistoryHostRequest_Read(w wire.Value) (*shared.DescribeHistoryHostRequest, error) {\n\tvar v shared.DescribeHistoryHostRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeHistoryHost_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeHistoryHost_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeHistoryHost_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeHistoryHost_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeHistoryHostRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeHistoryHost_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeHistoryHost_Args struct could not be encoded.\nfunc (v *AdminService_DescribeHistoryHost_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeHistoryHostRequest_Decode(sr stream.Reader) (*shared.DescribeHistoryHostRequest, error) {\n\tvar v shared.DescribeHistoryHostRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeHistoryHost_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeHistoryHost_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeHistoryHost_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeHistoryHostRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeHistoryHost_Args\n// struct.\nfunc (v *AdminService_DescribeHistoryHost_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeHistoryHost_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeHistoryHost_Args match the\n// provided AdminService_DescribeHistoryHost_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeHistoryHost_Args) Equals(rhs *AdminService_DescribeHistoryHost_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeHistoryHost_Args.\nfunc (v *AdminService_DescribeHistoryHost_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeHistoryHost_Args) GetRequest() (o *shared.DescribeHistoryHostRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_DescribeHistoryHost_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeHistoryHost\" for this struct.\nfunc (v *AdminService_DescribeHistoryHost_Args) MethodName() string {\n\treturn \"DescribeHistoryHost\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DescribeHistoryHost_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DescribeHistoryHost_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DescribeHistoryHost\n// function.\nvar AdminService_DescribeHistoryHost_Helper = struct {\n\t// Args accepts the parameters of DescribeHistoryHost in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeHistoryHostRequest,\n\t) *AdminService_DescribeHistoryHost_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeHistoryHost.\n\t//\n\t// An error can be thrown by DescribeHistoryHost only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeHistoryHost\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeHistoryHost into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeHistoryHost\n\t//\n\t//   value, err := DescribeHistoryHost(args)\n\t//   result, err := AdminService_DescribeHistoryHost_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeHistoryHost: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeHistoryHostResponse, error) (*AdminService_DescribeHistoryHost_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeHistoryHost\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeHistoryHost threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DescribeHistoryHost_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DescribeHistoryHost_Result) (*shared.DescribeHistoryHostResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DescribeHistoryHost_Helper.Args = func(\n\t\trequest *shared.DescribeHistoryHostRequest,\n\t) *AdminService_DescribeHistoryHost_Args {\n\t\treturn &AdminService_DescribeHistoryHost_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_DescribeHistoryHost_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DescribeHistoryHost_Helper.WrapResponse = func(success *shared.DescribeHistoryHostResponse, err error) (*AdminService_DescribeHistoryHost_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DescribeHistoryHost_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeHistoryHost_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeHistoryHost_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeHistoryHost_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeHistoryHost_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeHistoryHost_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeHistoryHost_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DescribeHistoryHost_Helper.UnwrapResponse = func(result *AdminService_DescribeHistoryHost_Result) (success *shared.DescribeHistoryHostResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DescribeHistoryHost_Result represents the result of a AdminService.DescribeHistoryHost function call.\n//\n// The result of a DescribeHistoryHost execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DescribeHistoryHost_Result struct {\n\t// Value returned by DescribeHistoryHost after a successful execution.\n\tSuccess              *shared.DescribeHistoryHostResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError           `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeHistoryHost_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeHistoryHost_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeHistoryHostResponse_Read(w wire.Value) (*shared.DescribeHistoryHostResponse, error) {\n\tvar v shared.DescribeHistoryHostResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeHistoryHost_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeHistoryHost_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeHistoryHost_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeHistoryHost_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeHistoryHostResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeHistoryHost_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeHistoryHost_Result struct could not be encoded.\nfunc (v *AdminService_DescribeHistoryHost_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeHistoryHostResponse_Decode(sr stream.Reader) (*shared.DescribeHistoryHostResponse, error) {\n\tvar v shared.DescribeHistoryHostResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeHistoryHost_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeHistoryHost_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeHistoryHost_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeHistoryHostResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeHistoryHost_Result\n// struct.\nfunc (v *AdminService_DescribeHistoryHost_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeHistoryHost_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeHistoryHost_Result match the\n// provided AdminService_DescribeHistoryHost_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeHistoryHost_Result) Equals(rhs *AdminService_DescribeHistoryHost_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeHistoryHost_Result.\nfunc (v *AdminService_DescribeHistoryHost_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeHistoryHost_Result) GetSuccess() (o *shared.DescribeHistoryHostResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DescribeHistoryHost_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeHistoryHost_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_DescribeHistoryHost_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeHistoryHost_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DescribeHistoryHost_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeHistoryHost_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_DescribeHistoryHost_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeHistoryHost\" for this struct.\nfunc (v *AdminService_DescribeHistoryHost_Result) MethodName() string {\n\treturn \"DescribeHistoryHost\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DescribeHistoryHost_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DescribeQueue_Args represents the arguments for the AdminService.DescribeQueue function.\n//\n// The arguments for DescribeQueue are sent and received over the wire as this struct.\ntype AdminService_DescribeQueue_Args struct {\n\tRequest *shared.DescribeQueueRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeQueue_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeQueue_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeQueueRequest_Read(w wire.Value) (*shared.DescribeQueueRequest, error) {\n\tvar v shared.DescribeQueueRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeQueue_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeQueue_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeQueue_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeQueue_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeQueueRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeQueue_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeQueue_Args struct could not be encoded.\nfunc (v *AdminService_DescribeQueue_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeQueueRequest_Decode(sr stream.Reader) (*shared.DescribeQueueRequest, error) {\n\tvar v shared.DescribeQueueRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeQueue_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeQueue_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeQueue_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeQueueRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeQueue_Args\n// struct.\nfunc (v *AdminService_DescribeQueue_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeQueue_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeQueue_Args match the\n// provided AdminService_DescribeQueue_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeQueue_Args) Equals(rhs *AdminService_DescribeQueue_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeQueue_Args.\nfunc (v *AdminService_DescribeQueue_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeQueue_Args) GetRequest() (o *shared.DescribeQueueRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_DescribeQueue_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeQueue\" for this struct.\nfunc (v *AdminService_DescribeQueue_Args) MethodName() string {\n\treturn \"DescribeQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DescribeQueue_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DescribeQueue_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DescribeQueue\n// function.\nvar AdminService_DescribeQueue_Helper = struct {\n\t// Args accepts the parameters of DescribeQueue in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeQueueRequest,\n\t) *AdminService_DescribeQueue_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeQueue.\n\t//\n\t// An error can be thrown by DescribeQueue only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeQueue\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeQueue into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeQueue\n\t//\n\t//   value, err := DescribeQueue(args)\n\t//   result, err := AdminService_DescribeQueue_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeQueue: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeQueueResponse, error) (*AdminService_DescribeQueue_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeQueue\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeQueue threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DescribeQueue_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DescribeQueue_Result) (*shared.DescribeQueueResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DescribeQueue_Helper.Args = func(\n\t\trequest *shared.DescribeQueueRequest,\n\t) *AdminService_DescribeQueue_Args {\n\t\treturn &AdminService_DescribeQueue_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_DescribeQueue_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DescribeQueue_Helper.WrapResponse = func(success *shared.DescribeQueueResponse, err error) (*AdminService_DescribeQueue_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DescribeQueue_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeQueue_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeQueue_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeQueue_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeQueue_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeQueue_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeQueue_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DescribeQueue_Helper.UnwrapResponse = func(result *AdminService_DescribeQueue_Result) (success *shared.DescribeQueueResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DescribeQueue_Result represents the result of a AdminService.DescribeQueue function call.\n//\n// The result of a DescribeQueue execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DescribeQueue_Result struct {\n\t// Value returned by DescribeQueue after a successful execution.\n\tSuccess              *shared.DescribeQueueResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError       `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError  `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError     `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeQueue_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeQueue_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DescribeQueue_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeQueueResponse_Read(w wire.Value) (*shared.DescribeQueueResponse, error) {\n\tvar v shared.DescribeQueueResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeQueue_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeQueue_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeQueue_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeQueue_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeQueueResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeQueue_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeQueue_Result struct could not be encoded.\nfunc (v *AdminService_DescribeQueue_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeQueueResponse_Decode(sr stream.Reader) (*shared.DescribeQueueResponse, error) {\n\tvar v shared.DescribeQueueResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeQueue_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeQueue_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeQueue_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeQueueResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeQueue_Result\n// struct.\nfunc (v *AdminService_DescribeQueue_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeQueue_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeQueue_Result match the\n// provided AdminService_DescribeQueue_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeQueue_Result) Equals(rhs *AdminService_DescribeQueue_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeQueue_Result.\nfunc (v *AdminService_DescribeQueue_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeQueue_Result) GetSuccess() (o *shared.DescribeQueueResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DescribeQueue_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeQueue_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_DescribeQueue_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeQueue_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DescribeQueue_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeQueue_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_DescribeQueue_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeQueue\" for this struct.\nfunc (v *AdminService_DescribeQueue_Result) MethodName() string {\n\treturn \"DescribeQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DescribeQueue_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DescribeShardDistribution_Args represents the arguments for the AdminService.DescribeShardDistribution function.\n//\n// The arguments for DescribeShardDistribution are sent and received over the wire as this struct.\ntype AdminService_DescribeShardDistribution_Args struct {\n\tRequest *shared.DescribeShardDistributionRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeShardDistribution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeShardDistribution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeShardDistributionRequest_Read(w wire.Value) (*shared.DescribeShardDistributionRequest, error) {\n\tvar v shared.DescribeShardDistributionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeShardDistribution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeShardDistribution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeShardDistribution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeShardDistribution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeShardDistributionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeShardDistribution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeShardDistribution_Args struct could not be encoded.\nfunc (v *AdminService_DescribeShardDistribution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeShardDistributionRequest_Decode(sr stream.Reader) (*shared.DescribeShardDistributionRequest, error) {\n\tvar v shared.DescribeShardDistributionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeShardDistribution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeShardDistribution_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeShardDistribution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeShardDistributionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeShardDistribution_Args\n// struct.\nfunc (v *AdminService_DescribeShardDistribution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeShardDistribution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeShardDistribution_Args match the\n// provided AdminService_DescribeShardDistribution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeShardDistribution_Args) Equals(rhs *AdminService_DescribeShardDistribution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeShardDistribution_Args.\nfunc (v *AdminService_DescribeShardDistribution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeShardDistribution_Args) GetRequest() (o *shared.DescribeShardDistributionRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_DescribeShardDistribution_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeShardDistribution\" for this struct.\nfunc (v *AdminService_DescribeShardDistribution_Args) MethodName() string {\n\treturn \"DescribeShardDistribution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DescribeShardDistribution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DescribeShardDistribution_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DescribeShardDistribution\n// function.\nvar AdminService_DescribeShardDistribution_Helper = struct {\n\t// Args accepts the parameters of DescribeShardDistribution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeShardDistributionRequest,\n\t) *AdminService_DescribeShardDistribution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeShardDistribution.\n\t//\n\t// An error can be thrown by DescribeShardDistribution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeShardDistribution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeShardDistribution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeShardDistribution\n\t//\n\t//   value, err := DescribeShardDistribution(args)\n\t//   result, err := AdminService_DescribeShardDistribution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeShardDistribution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeShardDistributionResponse, error) (*AdminService_DescribeShardDistribution_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeShardDistribution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeShardDistribution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DescribeShardDistribution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DescribeShardDistribution_Result) (*shared.DescribeShardDistributionResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DescribeShardDistribution_Helper.Args = func(\n\t\trequest *shared.DescribeShardDistributionRequest,\n\t) *AdminService_DescribeShardDistribution_Args {\n\t\treturn &AdminService_DescribeShardDistribution_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_DescribeShardDistribution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DescribeShardDistribution_Helper.WrapResponse = func(success *shared.DescribeShardDistributionResponse, err error) (*AdminService_DescribeShardDistribution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DescribeShardDistribution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeShardDistribution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeShardDistribution_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DescribeShardDistribution_Helper.UnwrapResponse = func(result *AdminService_DescribeShardDistribution_Result) (success *shared.DescribeShardDistributionResponse, err error) {\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DescribeShardDistribution_Result represents the result of a AdminService.DescribeShardDistribution function call.\n//\n// The result of a DescribeShardDistribution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DescribeShardDistribution_Result struct {\n\t// Value returned by DescribeShardDistribution after a successful execution.\n\tSuccess              *shared.DescribeShardDistributionResponse `json:\"success,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError              `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeShardDistribution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeShardDistribution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DescribeShardDistribution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeShardDistributionResponse_Read(w wire.Value) (*shared.DescribeShardDistributionResponse, error) {\n\tvar v shared.DescribeShardDistributionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeShardDistribution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeShardDistribution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeShardDistribution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeShardDistribution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeShardDistributionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeShardDistribution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeShardDistribution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeShardDistribution_Result struct could not be encoded.\nfunc (v *AdminService_DescribeShardDistribution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeShardDistribution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeShardDistributionResponse_Decode(sr stream.Reader) (*shared.DescribeShardDistributionResponse, error) {\n\tvar v shared.DescribeShardDistributionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeShardDistribution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeShardDistribution_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeShardDistribution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeShardDistributionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeShardDistribution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeShardDistribution_Result\n// struct.\nfunc (v *AdminService_DescribeShardDistribution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeShardDistribution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeShardDistribution_Result match the\n// provided AdminService_DescribeShardDistribution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeShardDistribution_Result) Equals(rhs *AdminService_DescribeShardDistribution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeShardDistribution_Result.\nfunc (v *AdminService_DescribeShardDistribution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeShardDistribution_Result) GetSuccess() (o *shared.DescribeShardDistributionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DescribeShardDistribution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeShardDistribution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DescribeShardDistribution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeShardDistribution\" for this struct.\nfunc (v *AdminService_DescribeShardDistribution_Result) MethodName() string {\n\treturn \"DescribeShardDistribution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DescribeShardDistribution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_DescribeWorkflowExecution_Args represents the arguments for the AdminService.DescribeWorkflowExecution function.\n//\n// The arguments for DescribeWorkflowExecution are sent and received over the wire as this struct.\ntype AdminService_DescribeWorkflowExecution_Args struct {\n\tRequest *DescribeWorkflowExecutionRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionRequest_Read(w wire.Value) (*DescribeWorkflowExecutionRequest, error) {\n\tvar v DescribeWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeWorkflowExecution_Args struct could not be encoded.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionRequest_Decode(sr stream.Reader) (*DescribeWorkflowExecutionRequest, error) {\n\tvar v DescribeWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeWorkflowExecution_Args\n// struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeWorkflowExecution_Args match the\n// provided AdminService_DescribeWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) Equals(rhs *AdminService_DescribeWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeWorkflowExecution_Args.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) GetRequest() (o *DescribeWorkflowExecutionRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_DescribeWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.DescribeWorkflowExecution\n// function.\nvar AdminService_DescribeWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of DescribeWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *DescribeWorkflowExecutionRequest,\n\t) *AdminService_DescribeWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeWorkflowExecution.\n\t//\n\t// An error can be thrown by DescribeWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeWorkflowExecution\n\t//\n\t//   value, err := DescribeWorkflowExecution(args)\n\t//   result, err := AdminService_DescribeWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*DescribeWorkflowExecutionResponse, error) (*AdminService_DescribeWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_DescribeWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_DescribeWorkflowExecution_Result) (*DescribeWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_DescribeWorkflowExecution_Helper.Args = func(\n\t\trequest *DescribeWorkflowExecutionRequest,\n\t) *AdminService_DescribeWorkflowExecution_Args {\n\t\treturn &AdminService_DescribeWorkflowExecution_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_DescribeWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_DescribeWorkflowExecution_Helper.WrapResponse = func(success *DescribeWorkflowExecutionResponse, err error) (*AdminService_DescribeWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_DescribeWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_DescribeWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_DescribeWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_DescribeWorkflowExecution_Helper.UnwrapResponse = func(result *AdminService_DescribeWorkflowExecution_Result) (success *DescribeWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_DescribeWorkflowExecution_Result represents the result of a AdminService.DescribeWorkflowExecution function call.\n//\n// The result of a DescribeWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_DescribeWorkflowExecution_Result struct {\n\t// Value returned by DescribeWorkflowExecution after a successful execution.\n\tSuccess              *DescribeWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError            `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError       `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError       `json:\"entityNotExistError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError          `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_DescribeWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_DescribeWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionResponse_Read(w wire.Value) (*DescribeWorkflowExecutionResponse, error) {\n\tvar v DescribeWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_DescribeWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_DescribeWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_DescribeWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_DescribeWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_DescribeWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_DescribeWorkflowExecution_Result struct could not be encoded.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionResponse_Decode(sr stream.Reader) (*DescribeWorkflowExecutionResponse, error) {\n\tvar v DescribeWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_DescribeWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_DescribeWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_DescribeWorkflowExecution_Result\n// struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_DescribeWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_DescribeWorkflowExecution_Result match the\n// provided AdminService_DescribeWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) Equals(rhs *AdminService_DescribeWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_DescribeWorkflowExecution_Result.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) GetSuccess() (o *DescribeWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_DescribeWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetCrossClusterTasks_Args represents the arguments for the AdminService.GetCrossClusterTasks function.\n//\n// The arguments for GetCrossClusterTasks are sent and received over the wire as this struct.\ntype AdminService_GetCrossClusterTasks_Args struct {\n\tRequest *shared.GetCrossClusterTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetCrossClusterTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetCrossClusterTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetCrossClusterTasksRequest_Read(w wire.Value) (*shared.GetCrossClusterTasksRequest, error) {\n\tvar v shared.GetCrossClusterTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetCrossClusterTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetCrossClusterTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetCrossClusterTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetCrossClusterTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetCrossClusterTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetCrossClusterTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetCrossClusterTasks_Args struct could not be encoded.\nfunc (v *AdminService_GetCrossClusterTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetCrossClusterTasksRequest_Decode(sr stream.Reader) (*shared.GetCrossClusterTasksRequest, error) {\n\tvar v shared.GetCrossClusterTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetCrossClusterTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetCrossClusterTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetCrossClusterTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetCrossClusterTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetCrossClusterTasks_Args\n// struct.\nfunc (v *AdminService_GetCrossClusterTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetCrossClusterTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetCrossClusterTasks_Args match the\n// provided AdminService_GetCrossClusterTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetCrossClusterTasks_Args) Equals(rhs *AdminService_GetCrossClusterTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetCrossClusterTasks_Args.\nfunc (v *AdminService_GetCrossClusterTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetCrossClusterTasks_Args) GetRequest() (o *shared.GetCrossClusterTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetCrossClusterTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetCrossClusterTasks\" for this struct.\nfunc (v *AdminService_GetCrossClusterTasks_Args) MethodName() string {\n\treturn \"GetCrossClusterTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetCrossClusterTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetCrossClusterTasks_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetCrossClusterTasks\n// function.\nvar AdminService_GetCrossClusterTasks_Helper = struct {\n\t// Args accepts the parameters of GetCrossClusterTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.GetCrossClusterTasksRequest,\n\t) *AdminService_GetCrossClusterTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetCrossClusterTasks.\n\t//\n\t// An error can be thrown by GetCrossClusterTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetCrossClusterTasks\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetCrossClusterTasks into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetCrossClusterTasks\n\t//\n\t//   value, err := GetCrossClusterTasks(args)\n\t//   result, err := AdminService_GetCrossClusterTasks_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetCrossClusterTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetCrossClusterTasksResponse, error) (*AdminService_GetCrossClusterTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetCrossClusterTasks\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetCrossClusterTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetCrossClusterTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetCrossClusterTasks_Result) (*shared.GetCrossClusterTasksResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetCrossClusterTasks_Helper.Args = func(\n\t\trequest *shared.GetCrossClusterTasksRequest,\n\t) *AdminService_GetCrossClusterTasks_Args {\n\t\treturn &AdminService_GetCrossClusterTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetCrossClusterTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetCrossClusterTasks_Helper.WrapResponse = func(success *shared.GetCrossClusterTasksResponse, err error) (*AdminService_GetCrossClusterTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetCrossClusterTasks_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetCrossClusterTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetCrossClusterTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetCrossClusterTasks_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetCrossClusterTasks_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetCrossClusterTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetCrossClusterTasks_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetCrossClusterTasks_Helper.UnwrapResponse = func(result *AdminService_GetCrossClusterTasks_Result) (success *shared.GetCrossClusterTasksResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetCrossClusterTasks_Result represents the result of a AdminService.GetCrossClusterTasks function call.\n//\n// The result of a GetCrossClusterTasks execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetCrossClusterTasks_Result struct {\n\t// Value returned by GetCrossClusterTasks after a successful execution.\n\tSuccess              *shared.GetCrossClusterTasksResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetCrossClusterTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetCrossClusterTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetCrossClusterTasksResponse_Read(w wire.Value) (*shared.GetCrossClusterTasksResponse, error) {\n\tvar v shared.GetCrossClusterTasksResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetCrossClusterTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetCrossClusterTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetCrossClusterTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetCrossClusterTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetCrossClusterTasksResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetCrossClusterTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetCrossClusterTasks_Result struct could not be encoded.\nfunc (v *AdminService_GetCrossClusterTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetCrossClusterTasksResponse_Decode(sr stream.Reader) (*shared.GetCrossClusterTasksResponse, error) {\n\tvar v shared.GetCrossClusterTasksResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetCrossClusterTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetCrossClusterTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetCrossClusterTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetCrossClusterTasksResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetCrossClusterTasks_Result\n// struct.\nfunc (v *AdminService_GetCrossClusterTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetCrossClusterTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetCrossClusterTasks_Result match the\n// provided AdminService_GetCrossClusterTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetCrossClusterTasks_Result) Equals(rhs *AdminService_GetCrossClusterTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetCrossClusterTasks_Result.\nfunc (v *AdminService_GetCrossClusterTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetCrossClusterTasks_Result) GetSuccess() (o *shared.GetCrossClusterTasksResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetCrossClusterTasks_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetCrossClusterTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetCrossClusterTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetCrossClusterTasks_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_GetCrossClusterTasks_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetCrossClusterTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_GetCrossClusterTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetCrossClusterTasks\" for this struct.\nfunc (v *AdminService_GetCrossClusterTasks_Result) MethodName() string {\n\treturn \"GetCrossClusterTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetCrossClusterTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetDLQReplicationMessages_Args represents the arguments for the AdminService.GetDLQReplicationMessages function.\n//\n// The arguments for GetDLQReplicationMessages are sent and received over the wire as this struct.\ntype AdminService_GetDLQReplicationMessages_Args struct {\n\tRequest *replicator.GetDLQReplicationMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDLQReplicationMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDLQReplicationMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDLQReplicationMessagesRequest_Read(w wire.Value) (*replicator.GetDLQReplicationMessagesRequest, error) {\n\tvar v replicator.GetDLQReplicationMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDLQReplicationMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDLQReplicationMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDLQReplicationMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDLQReplicationMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDLQReplicationMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDLQReplicationMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDLQReplicationMessages_Args struct could not be encoded.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDLQReplicationMessagesRequest_Decode(sr stream.Reader) (*replicator.GetDLQReplicationMessagesRequest, error) {\n\tvar v replicator.GetDLQReplicationMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDLQReplicationMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDLQReplicationMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDLQReplicationMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDLQReplicationMessages_Args\n// struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDLQReplicationMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDLQReplicationMessages_Args match the\n// provided AdminService_GetDLQReplicationMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) Equals(rhs *AdminService_GetDLQReplicationMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDLQReplicationMessages_Args.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) GetRequest() (o *replicator.GetDLQReplicationMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDLQReplicationMessages\" for this struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) MethodName() string {\n\treturn \"GetDLQReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetDLQReplicationMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetDLQReplicationMessages\n// function.\nvar AdminService_GetDLQReplicationMessages_Helper = struct {\n\t// Args accepts the parameters of GetDLQReplicationMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.GetDLQReplicationMessagesRequest,\n\t) *AdminService_GetDLQReplicationMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDLQReplicationMessages.\n\t//\n\t// An error can be thrown by GetDLQReplicationMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDLQReplicationMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDLQReplicationMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDLQReplicationMessages\n\t//\n\t//   value, err := GetDLQReplicationMessages(args)\n\t//   result, err := AdminService_GetDLQReplicationMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDLQReplicationMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.GetDLQReplicationMessagesResponse, error) (*AdminService_GetDLQReplicationMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDLQReplicationMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDLQReplicationMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetDLQReplicationMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetDLQReplicationMessages_Result) (*replicator.GetDLQReplicationMessagesResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetDLQReplicationMessages_Helper.Args = func(\n\t\trequest *replicator.GetDLQReplicationMessagesRequest,\n\t) *AdminService_GetDLQReplicationMessages_Args {\n\t\treturn &AdminService_GetDLQReplicationMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetDLQReplicationMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetDLQReplicationMessages_Helper.WrapResponse = func(success *replicator.GetDLQReplicationMessagesResponse, err error) (*AdminService_GetDLQReplicationMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetDLQReplicationMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDLQReplicationMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDLQReplicationMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDLQReplicationMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDLQReplicationMessages_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetDLQReplicationMessages_Helper.UnwrapResponse = func(result *AdminService_GetDLQReplicationMessages_Result) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetDLQReplicationMessages_Result represents the result of a AdminService.GetDLQReplicationMessages function call.\n//\n// The result of a GetDLQReplicationMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetDLQReplicationMessages_Result struct {\n\t// Value returned by GetDLQReplicationMessages after a successful execution.\n\tSuccess          *replicator.GetDLQReplicationMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError  *shared.BadRequestError                       `json:\"badRequestError,omitempty\"`\n\tServiceBusyError *shared.ServiceBusyError                      `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDLQReplicationMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDLQReplicationMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDLQReplicationMessagesResponse_Read(w wire.Value) (*replicator.GetDLQReplicationMessagesResponse, error) {\n\tvar v replicator.GetDLQReplicationMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDLQReplicationMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDLQReplicationMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDLQReplicationMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDLQReplicationMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDLQReplicationMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDLQReplicationMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDLQReplicationMessages_Result struct could not be encoded.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDLQReplicationMessagesResponse_Decode(sr stream.Reader) (*replicator.GetDLQReplicationMessagesResponse, error) {\n\tvar v replicator.GetDLQReplicationMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDLQReplicationMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDLQReplicationMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDLQReplicationMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDLQReplicationMessages_Result\n// struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDLQReplicationMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDLQReplicationMessages_Result match the\n// provided AdminService_GetDLQReplicationMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) Equals(rhs *AdminService_GetDLQReplicationMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDLQReplicationMessages_Result.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) GetSuccess() (o *replicator.GetDLQReplicationMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDLQReplicationMessages\" for this struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) MethodName() string {\n\treturn \"GetDLQReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetDLQReplicationMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetDomainAsyncWorkflowConfiguraton_Args represents the arguments for the AdminService.GetDomainAsyncWorkflowConfiguraton function.\n//\n// The arguments for GetDomainAsyncWorkflowConfiguraton are sent and received over the wire as this struct.\ntype AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct {\n\tRequest *GetDomainAsyncWorkflowConfiguratonRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainAsyncWorkflowConfiguratonRequest_Read(w wire.Value) (*GetDomainAsyncWorkflowConfiguratonRequest, error) {\n\tvar v GetDomainAsyncWorkflowConfiguratonRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainAsyncWorkflowConfiguraton_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDomainAsyncWorkflowConfiguratonRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct could not be encoded.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainAsyncWorkflowConfiguratonRequest_Decode(sr stream.Reader) (*GetDomainAsyncWorkflowConfiguratonRequest, error) {\n\tvar v GetDomainAsyncWorkflowConfiguratonRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainAsyncWorkflowConfiguraton_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDomainAsyncWorkflowConfiguratonRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainAsyncWorkflowConfiguraton_Args\n// struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainAsyncWorkflowConfiguraton_Args match the\n// provided AdminService_GetDomainAsyncWorkflowConfiguraton_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) Equals(rhs *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainAsyncWorkflowConfiguraton_Args.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) GetRequest() (o *GetDomainAsyncWorkflowConfiguratonRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDomainAsyncWorkflowConfiguraton\" for this struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) MethodName() string {\n\treturn \"GetDomainAsyncWorkflowConfiguraton\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetDomainAsyncWorkflowConfiguraton_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetDomainAsyncWorkflowConfiguraton\n// function.\nvar AdminService_GetDomainAsyncWorkflowConfiguraton_Helper = struct {\n\t// Args accepts the parameters of GetDomainAsyncWorkflowConfiguraton in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *GetDomainAsyncWorkflowConfiguratonRequest,\n\t) *AdminService_GetDomainAsyncWorkflowConfiguraton_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDomainAsyncWorkflowConfiguraton.\n\t//\n\t// An error can be thrown by GetDomainAsyncWorkflowConfiguraton only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDomainAsyncWorkflowConfiguraton\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDomainAsyncWorkflowConfiguraton into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDomainAsyncWorkflowConfiguraton\n\t//\n\t//   value, err := GetDomainAsyncWorkflowConfiguraton(args)\n\t//   result, err := AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDomainAsyncWorkflowConfiguraton: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetDomainAsyncWorkflowConfiguratonResponse, error) (*AdminService_GetDomainAsyncWorkflowConfiguraton_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDomainAsyncWorkflowConfiguraton\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDomainAsyncWorkflowConfiguraton threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetDomainAsyncWorkflowConfiguraton_Result) (*GetDomainAsyncWorkflowConfiguratonResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetDomainAsyncWorkflowConfiguraton_Helper.Args = func(\n\t\trequest *GetDomainAsyncWorkflowConfiguratonRequest,\n\t) *AdminService_GetDomainAsyncWorkflowConfiguraton_Args {\n\t\treturn &AdminService_GetDomainAsyncWorkflowConfiguraton_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetDomainAsyncWorkflowConfiguraton_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetDomainAsyncWorkflowConfiguraton_Helper.WrapResponse = func(success *GetDomainAsyncWorkflowConfiguratonResponse, err error) (*AdminService_GetDomainAsyncWorkflowConfiguraton_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetDomainAsyncWorkflowConfiguraton_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainAsyncWorkflowConfiguraton_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainAsyncWorkflowConfiguraton_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse = func(result *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) (success *GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetDomainAsyncWorkflowConfiguraton_Result represents the result of a AdminService.GetDomainAsyncWorkflowConfiguraton function call.\n//\n// The result of a GetDomainAsyncWorkflowConfiguraton execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct {\n\t// Value returned by GetDomainAsyncWorkflowConfiguraton after a successful execution.\n\tSuccess         *GetDomainAsyncWorkflowConfiguratonResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError                     `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainAsyncWorkflowConfiguratonResponse_Read(w wire.Value) (*GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tvar v GetDomainAsyncWorkflowConfiguratonResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainAsyncWorkflowConfiguraton_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDomainAsyncWorkflowConfiguratonResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct could not be encoded.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainAsyncWorkflowConfiguratonResponse_Decode(sr stream.Reader) (*GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tvar v GetDomainAsyncWorkflowConfiguratonResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainAsyncWorkflowConfiguraton_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDomainAsyncWorkflowConfiguratonResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainAsyncWorkflowConfiguraton_Result\n// struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainAsyncWorkflowConfiguraton_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainAsyncWorkflowConfiguraton_Result match the\n// provided AdminService_GetDomainAsyncWorkflowConfiguraton_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) Equals(rhs *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainAsyncWorkflowConfiguraton_Result.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) GetSuccess() (o *GetDomainAsyncWorkflowConfiguratonResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDomainAsyncWorkflowConfiguraton\" for this struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) MethodName() string {\n\treturn \"GetDomainAsyncWorkflowConfiguraton\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetDomainAsyncWorkflowConfiguraton_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetDomainIsolationGroups_Args represents the arguments for the AdminService.GetDomainIsolationGroups function.\n//\n// The arguments for GetDomainIsolationGroups are sent and received over the wire as this struct.\ntype AdminService_GetDomainIsolationGroups_Args struct {\n\tRequest *GetDomainIsolationGroupsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainIsolationGroups_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainIsolationGroups_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainIsolationGroupsRequest_Read(w wire.Value) (*GetDomainIsolationGroupsRequest, error) {\n\tvar v GetDomainIsolationGroupsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainIsolationGroups_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainIsolationGroups_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainIsolationGroups_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainIsolationGroups_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDomainIsolationGroupsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainIsolationGroups_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainIsolationGroups_Args struct could not be encoded.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainIsolationGroupsRequest_Decode(sr stream.Reader) (*GetDomainIsolationGroupsRequest, error) {\n\tvar v GetDomainIsolationGroupsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainIsolationGroups_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainIsolationGroups_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDomainIsolationGroupsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainIsolationGroups_Args\n// struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainIsolationGroups_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainIsolationGroups_Args match the\n// provided AdminService_GetDomainIsolationGroups_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) Equals(rhs *AdminService_GetDomainIsolationGroups_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainIsolationGroups_Args.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) GetRequest() (o *GetDomainIsolationGroupsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDomainIsolationGroups\" for this struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) MethodName() string {\n\treturn \"GetDomainIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetDomainIsolationGroups_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetDomainIsolationGroups\n// function.\nvar AdminService_GetDomainIsolationGroups_Helper = struct {\n\t// Args accepts the parameters of GetDomainIsolationGroups in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *GetDomainIsolationGroupsRequest,\n\t) *AdminService_GetDomainIsolationGroups_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDomainIsolationGroups.\n\t//\n\t// An error can be thrown by GetDomainIsolationGroups only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDomainIsolationGroups\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDomainIsolationGroups into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDomainIsolationGroups\n\t//\n\t//   value, err := GetDomainIsolationGroups(args)\n\t//   result, err := AdminService_GetDomainIsolationGroups_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDomainIsolationGroups: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetDomainIsolationGroupsResponse, error) (*AdminService_GetDomainIsolationGroups_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDomainIsolationGroups\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDomainIsolationGroups threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetDomainIsolationGroups_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetDomainIsolationGroups_Result) (*GetDomainIsolationGroupsResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetDomainIsolationGroups_Helper.Args = func(\n\t\trequest *GetDomainIsolationGroupsRequest,\n\t) *AdminService_GetDomainIsolationGroups_Args {\n\t\treturn &AdminService_GetDomainIsolationGroups_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetDomainIsolationGroups_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetDomainIsolationGroups_Helper.WrapResponse = func(success *GetDomainIsolationGroupsResponse, err error) (*AdminService_GetDomainIsolationGroups_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetDomainIsolationGroups_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainIsolationGroups_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainIsolationGroups_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetDomainIsolationGroups_Helper.UnwrapResponse = func(result *AdminService_GetDomainIsolationGroups_Result) (success *GetDomainIsolationGroupsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetDomainIsolationGroups_Result represents the result of a AdminService.GetDomainIsolationGroups function call.\n//\n// The result of a GetDomainIsolationGroups execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetDomainIsolationGroups_Result struct {\n\t// Value returned by GetDomainIsolationGroups after a successful execution.\n\tSuccess         *GetDomainIsolationGroupsResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError           `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainIsolationGroups_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainIsolationGroups_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetDomainIsolationGroups_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainIsolationGroupsResponse_Read(w wire.Value) (*GetDomainIsolationGroupsResponse, error) {\n\tvar v GetDomainIsolationGroupsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainIsolationGroups_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainIsolationGroups_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainIsolationGroups_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainIsolationGroups_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDomainIsolationGroupsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainIsolationGroups_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainIsolationGroups_Result struct could not be encoded.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainIsolationGroupsResponse_Decode(sr stream.Reader) (*GetDomainIsolationGroupsResponse, error) {\n\tvar v GetDomainIsolationGroupsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainIsolationGroups_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainIsolationGroups_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDomainIsolationGroupsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainIsolationGroups_Result\n// struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainIsolationGroups_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainIsolationGroups_Result match the\n// provided AdminService_GetDomainIsolationGroups_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) Equals(rhs *AdminService_GetDomainIsolationGroups_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainIsolationGroups_Result.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) GetSuccess() (o *GetDomainIsolationGroupsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDomainIsolationGroups\" for this struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) MethodName() string {\n\treturn \"GetDomainIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetDomainIsolationGroups_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetDomainReplicationMessages_Args represents the arguments for the AdminService.GetDomainReplicationMessages function.\n//\n// The arguments for GetDomainReplicationMessages are sent and received over the wire as this struct.\ntype AdminService_GetDomainReplicationMessages_Args struct {\n\tRequest *replicator.GetDomainReplicationMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainReplicationMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainReplicationMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainReplicationMessagesRequest_Read(w wire.Value) (*replicator.GetDomainReplicationMessagesRequest, error) {\n\tvar v replicator.GetDomainReplicationMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainReplicationMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainReplicationMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainReplicationMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainReplicationMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDomainReplicationMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainReplicationMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainReplicationMessages_Args struct could not be encoded.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainReplicationMessagesRequest_Decode(sr stream.Reader) (*replicator.GetDomainReplicationMessagesRequest, error) {\n\tvar v replicator.GetDomainReplicationMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainReplicationMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainReplicationMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDomainReplicationMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainReplicationMessages_Args\n// struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainReplicationMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainReplicationMessages_Args match the\n// provided AdminService_GetDomainReplicationMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) Equals(rhs *AdminService_GetDomainReplicationMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainReplicationMessages_Args.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) GetRequest() (o *replicator.GetDomainReplicationMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDomainReplicationMessages\" for this struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) MethodName() string {\n\treturn \"GetDomainReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetDomainReplicationMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetDomainReplicationMessages\n// function.\nvar AdminService_GetDomainReplicationMessages_Helper = struct {\n\t// Args accepts the parameters of GetDomainReplicationMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.GetDomainReplicationMessagesRequest,\n\t) *AdminService_GetDomainReplicationMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDomainReplicationMessages.\n\t//\n\t// An error can be thrown by GetDomainReplicationMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDomainReplicationMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDomainReplicationMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDomainReplicationMessages\n\t//\n\t//   value, err := GetDomainReplicationMessages(args)\n\t//   result, err := AdminService_GetDomainReplicationMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDomainReplicationMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.GetDomainReplicationMessagesResponse, error) (*AdminService_GetDomainReplicationMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDomainReplicationMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDomainReplicationMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetDomainReplicationMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetDomainReplicationMessages_Result) (*replicator.GetDomainReplicationMessagesResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetDomainReplicationMessages_Helper.Args = func(\n\t\trequest *replicator.GetDomainReplicationMessagesRequest,\n\t) *AdminService_GetDomainReplicationMessages_Args {\n\t\treturn &AdminService_GetDomainReplicationMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetDomainReplicationMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetDomainReplicationMessages_Helper.WrapResponse = func(success *replicator.GetDomainReplicationMessagesResponse, err error) (*AdminService_GetDomainReplicationMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetDomainReplicationMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainReplicationMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainReplicationMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainReplicationMessages_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainReplicationMessages_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainReplicationMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainReplicationMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDomainReplicationMessages_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDomainReplicationMessages_Result{ClientVersionNotSupportedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetDomainReplicationMessages_Helper.UnwrapResponse = func(result *AdminService_GetDomainReplicationMessages_Result) (success *replicator.GetDomainReplicationMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetDomainReplicationMessages_Result represents the result of a AdminService.GetDomainReplicationMessages function call.\n//\n// The result of a GetDomainReplicationMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetDomainReplicationMessages_Result struct {\n\t// Value returned by GetDomainReplicationMessages after a successful execution.\n\tSuccess                        *replicator.GetDomainReplicationMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                          `json:\"badRequestError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                       `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                         `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError           `json:\"clientVersionNotSupportedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDomainReplicationMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDomainReplicationMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetDomainReplicationMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDomainReplicationMessagesResponse_Read(w wire.Value) (*replicator.GetDomainReplicationMessagesResponse, error) {\n\tvar v replicator.GetDomainReplicationMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Read(w wire.Value) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Read(w wire.Value) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDomainReplicationMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDomainReplicationMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDomainReplicationMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDomainReplicationMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDomainReplicationMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDomainReplicationMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDomainReplicationMessages_Result struct could not be encoded.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDomainReplicationMessagesResponse_Decode(sr stream.Reader) (*replicator.GetDomainReplicationMessagesResponse, error) {\n\tvar v replicator.GetDomainReplicationMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Decode(sr stream.Reader) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Decode(sr stream.Reader) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDomainReplicationMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDomainReplicationMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDomainReplicationMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDomainReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDomainReplicationMessages_Result\n// struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDomainReplicationMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDomainReplicationMessages_Result match the\n// provided AdminService_GetDomainReplicationMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) Equals(rhs *AdminService_GetDomainReplicationMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDomainReplicationMessages_Result.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) GetSuccess() (o *replicator.GetDomainReplicationMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDomainReplicationMessages\" for this struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) MethodName() string {\n\treturn \"GetDomainReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetDomainReplicationMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetDynamicConfig_Args represents the arguments for the AdminService.GetDynamicConfig function.\n//\n// The arguments for GetDynamicConfig are sent and received over the wire as this struct.\ntype AdminService_GetDynamicConfig_Args struct {\n\tRequest *GetDynamicConfigRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDynamicConfig_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDynamicConfig_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDynamicConfigRequest_Read(w wire.Value) (*GetDynamicConfigRequest, error) {\n\tvar v GetDynamicConfigRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDynamicConfig_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDynamicConfig_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDynamicConfig_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDynamicConfig_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDynamicConfigRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDynamicConfig_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDynamicConfig_Args struct could not be encoded.\nfunc (v *AdminService_GetDynamicConfig_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDynamicConfigRequest_Decode(sr stream.Reader) (*GetDynamicConfigRequest, error) {\n\tvar v GetDynamicConfigRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDynamicConfig_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDynamicConfig_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDynamicConfig_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDynamicConfigRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDynamicConfig_Args\n// struct.\nfunc (v *AdminService_GetDynamicConfig_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDynamicConfig_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDynamicConfig_Args match the\n// provided AdminService_GetDynamicConfig_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDynamicConfig_Args) Equals(rhs *AdminService_GetDynamicConfig_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDynamicConfig_Args.\nfunc (v *AdminService_GetDynamicConfig_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDynamicConfig_Args) GetRequest() (o *GetDynamicConfigRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetDynamicConfig_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDynamicConfig\" for this struct.\nfunc (v *AdminService_GetDynamicConfig_Args) MethodName() string {\n\treturn \"GetDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetDynamicConfig_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetDynamicConfig_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetDynamicConfig\n// function.\nvar AdminService_GetDynamicConfig_Helper = struct {\n\t// Args accepts the parameters of GetDynamicConfig in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *GetDynamicConfigRequest,\n\t) *AdminService_GetDynamicConfig_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDynamicConfig.\n\t//\n\t// An error can be thrown by GetDynamicConfig only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDynamicConfig\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDynamicConfig into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDynamicConfig\n\t//\n\t//   value, err := GetDynamicConfig(args)\n\t//   result, err := AdminService_GetDynamicConfig_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDynamicConfig: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetDynamicConfigResponse, error) (*AdminService_GetDynamicConfig_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDynamicConfig\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDynamicConfig threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetDynamicConfig_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetDynamicConfig_Result) (*GetDynamicConfigResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetDynamicConfig_Helper.Args = func(\n\t\trequest *GetDynamicConfigRequest,\n\t) *AdminService_GetDynamicConfig_Args {\n\t\treturn &AdminService_GetDynamicConfig_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetDynamicConfig_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetDynamicConfig_Helper.WrapResponse = func(success *GetDynamicConfigResponse, err error) (*AdminService_GetDynamicConfig_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetDynamicConfig_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDynamicConfig_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDynamicConfig_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetDynamicConfig_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetDynamicConfig_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetDynamicConfig_Helper.UnwrapResponse = func(result *AdminService_GetDynamicConfig_Result) (success *GetDynamicConfigResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetDynamicConfig_Result represents the result of a AdminService.GetDynamicConfig function call.\n//\n// The result of a GetDynamicConfig execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetDynamicConfig_Result struct {\n\t// Value returned by GetDynamicConfig after a successful execution.\n\tSuccess              *GetDynamicConfigResponse    `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetDynamicConfig_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetDynamicConfig_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetDynamicConfig_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDynamicConfigResponse_Read(w wire.Value) (*GetDynamicConfigResponse, error) {\n\tvar v GetDynamicConfigResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetDynamicConfig_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetDynamicConfig_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetDynamicConfig_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetDynamicConfig_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDynamicConfigResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetDynamicConfig_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetDynamicConfig_Result struct could not be encoded.\nfunc (v *AdminService_GetDynamicConfig_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDynamicConfigResponse_Decode(sr stream.Reader) (*GetDynamicConfigResponse, error) {\n\tvar v GetDynamicConfigResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetDynamicConfig_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetDynamicConfig_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetDynamicConfig_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDynamicConfigResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetDynamicConfig_Result\n// struct.\nfunc (v *AdminService_GetDynamicConfig_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetDynamicConfig_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetDynamicConfig_Result match the\n// provided AdminService_GetDynamicConfig_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetDynamicConfig_Result) Equals(rhs *AdminService_GetDynamicConfig_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetDynamicConfig_Result.\nfunc (v *AdminService_GetDynamicConfig_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDynamicConfig_Result) GetSuccess() (o *GetDynamicConfigResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetDynamicConfig_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDynamicConfig_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetDynamicConfig_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetDynamicConfig_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_GetDynamicConfig_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDynamicConfig\" for this struct.\nfunc (v *AdminService_GetDynamicConfig_Result) MethodName() string {\n\treturn \"GetDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetDynamicConfig_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetGlobalIsolationGroups_Args represents the arguments for the AdminService.GetGlobalIsolationGroups function.\n//\n// The arguments for GetGlobalIsolationGroups are sent and received over the wire as this struct.\ntype AdminService_GetGlobalIsolationGroups_Args struct {\n\tRequest *GetGlobalIsolationGroupsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetGlobalIsolationGroups_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetGlobalIsolationGroupsRequest_Read(w wire.Value) (*GetGlobalIsolationGroupsRequest, error) {\n\tvar v GetGlobalIsolationGroupsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetGlobalIsolationGroups_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetGlobalIsolationGroups_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetGlobalIsolationGroups_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetGlobalIsolationGroupsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetGlobalIsolationGroups_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetGlobalIsolationGroups_Args struct could not be encoded.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetGlobalIsolationGroupsRequest_Decode(sr stream.Reader) (*GetGlobalIsolationGroupsRequest, error) {\n\tvar v GetGlobalIsolationGroupsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetGlobalIsolationGroups_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetGlobalIsolationGroups_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetGlobalIsolationGroupsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetGlobalIsolationGroups_Args\n// struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetGlobalIsolationGroups_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetGlobalIsolationGroups_Args match the\n// provided AdminService_GetGlobalIsolationGroups_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) Equals(rhs *AdminService_GetGlobalIsolationGroups_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetGlobalIsolationGroups_Args.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) GetRequest() (o *GetGlobalIsolationGroupsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetGlobalIsolationGroups\" for this struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) MethodName() string {\n\treturn \"GetGlobalIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetGlobalIsolationGroups_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetGlobalIsolationGroups\n// function.\nvar AdminService_GetGlobalIsolationGroups_Helper = struct {\n\t// Args accepts the parameters of GetGlobalIsolationGroups in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *GetGlobalIsolationGroupsRequest,\n\t) *AdminService_GetGlobalIsolationGroups_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetGlobalIsolationGroups.\n\t//\n\t// An error can be thrown by GetGlobalIsolationGroups only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetGlobalIsolationGroups\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetGlobalIsolationGroups into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetGlobalIsolationGroups\n\t//\n\t//   value, err := GetGlobalIsolationGroups(args)\n\t//   result, err := AdminService_GetGlobalIsolationGroups_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetGlobalIsolationGroups: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetGlobalIsolationGroupsResponse, error) (*AdminService_GetGlobalIsolationGroups_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetGlobalIsolationGroups\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetGlobalIsolationGroups threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetGlobalIsolationGroups_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetGlobalIsolationGroups_Result) (*GetGlobalIsolationGroupsResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetGlobalIsolationGroups_Helper.Args = func(\n\t\trequest *GetGlobalIsolationGroupsRequest,\n\t) *AdminService_GetGlobalIsolationGroups_Args {\n\t\treturn &AdminService_GetGlobalIsolationGroups_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetGlobalIsolationGroups_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetGlobalIsolationGroups_Helper.WrapResponse = func(success *GetGlobalIsolationGroupsResponse, err error) (*AdminService_GetGlobalIsolationGroups_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetGlobalIsolationGroups_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetGlobalIsolationGroups_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetGlobalIsolationGroups_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetGlobalIsolationGroups_Helper.UnwrapResponse = func(result *AdminService_GetGlobalIsolationGroups_Result) (success *GetGlobalIsolationGroupsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetGlobalIsolationGroups_Result represents the result of a AdminService.GetGlobalIsolationGroups function call.\n//\n// The result of a GetGlobalIsolationGroups execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetGlobalIsolationGroups_Result struct {\n\t// Value returned by GetGlobalIsolationGroups after a successful execution.\n\tSuccess         *GetGlobalIsolationGroupsResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError           `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetGlobalIsolationGroups_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetGlobalIsolationGroups_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetGlobalIsolationGroupsResponse_Read(w wire.Value) (*GetGlobalIsolationGroupsResponse, error) {\n\tvar v GetGlobalIsolationGroupsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetGlobalIsolationGroups_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetGlobalIsolationGroups_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetGlobalIsolationGroups_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetGlobalIsolationGroupsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetGlobalIsolationGroups_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetGlobalIsolationGroups_Result struct could not be encoded.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetGlobalIsolationGroupsResponse_Decode(sr stream.Reader) (*GetGlobalIsolationGroupsResponse, error) {\n\tvar v GetGlobalIsolationGroupsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetGlobalIsolationGroups_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetGlobalIsolationGroups_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetGlobalIsolationGroupsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetGlobalIsolationGroups_Result\n// struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetGlobalIsolationGroups_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetGlobalIsolationGroups_Result match the\n// provided AdminService_GetGlobalIsolationGroups_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) Equals(rhs *AdminService_GetGlobalIsolationGroups_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetGlobalIsolationGroups_Result.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) GetSuccess() (o *GetGlobalIsolationGroupsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetGlobalIsolationGroups\" for this struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) MethodName() string {\n\treturn \"GetGlobalIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetGlobalIsolationGroups_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetReplicationMessages_Args represents the arguments for the AdminService.GetReplicationMessages function.\n//\n// The arguments for GetReplicationMessages are sent and received over the wire as this struct.\ntype AdminService_GetReplicationMessages_Args struct {\n\tRequest *replicator.GetReplicationMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetReplicationMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetReplicationMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetReplicationMessagesRequest_Read(w wire.Value) (*replicator.GetReplicationMessagesRequest, error) {\n\tvar v replicator.GetReplicationMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetReplicationMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetReplicationMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetReplicationMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetReplicationMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetReplicationMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetReplicationMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetReplicationMessages_Args struct could not be encoded.\nfunc (v *AdminService_GetReplicationMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetReplicationMessagesRequest_Decode(sr stream.Reader) (*replicator.GetReplicationMessagesRequest, error) {\n\tvar v replicator.GetReplicationMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetReplicationMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetReplicationMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetReplicationMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetReplicationMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetReplicationMessages_Args\n// struct.\nfunc (v *AdminService_GetReplicationMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetReplicationMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetReplicationMessages_Args match the\n// provided AdminService_GetReplicationMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetReplicationMessages_Args) Equals(rhs *AdminService_GetReplicationMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetReplicationMessages_Args.\nfunc (v *AdminService_GetReplicationMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Args) GetRequest() (o *replicator.GetReplicationMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_GetReplicationMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetReplicationMessages\" for this struct.\nfunc (v *AdminService_GetReplicationMessages_Args) MethodName() string {\n\treturn \"GetReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetReplicationMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetReplicationMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetReplicationMessages\n// function.\nvar AdminService_GetReplicationMessages_Helper = struct {\n\t// Args accepts the parameters of GetReplicationMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.GetReplicationMessagesRequest,\n\t) *AdminService_GetReplicationMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetReplicationMessages.\n\t//\n\t// An error can be thrown by GetReplicationMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetReplicationMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetReplicationMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetReplicationMessages\n\t//\n\t//   value, err := GetReplicationMessages(args)\n\t//   result, err := AdminService_GetReplicationMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetReplicationMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.GetReplicationMessagesResponse, error) (*AdminService_GetReplicationMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetReplicationMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetReplicationMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetReplicationMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetReplicationMessages_Result) (*replicator.GetReplicationMessagesResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_GetReplicationMessages_Helper.Args = func(\n\t\trequest *replicator.GetReplicationMessagesRequest,\n\t) *AdminService_GetReplicationMessages_Args {\n\t\treturn &AdminService_GetReplicationMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_GetReplicationMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetReplicationMessages_Helper.WrapResponse = func(success *replicator.GetReplicationMessagesResponse, err error) (*AdminService_GetReplicationMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetReplicationMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetReplicationMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetReplicationMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetReplicationMessages_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetReplicationMessages_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetReplicationMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetReplicationMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetReplicationMessages_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetReplicationMessages_Result{ClientVersionNotSupportedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetReplicationMessages_Helper.UnwrapResponse = func(result *AdminService_GetReplicationMessages_Result) (success *replicator.GetReplicationMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetReplicationMessages_Result represents the result of a AdminService.GetReplicationMessages function call.\n//\n// The result of a GetReplicationMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetReplicationMessages_Result struct {\n\t// Value returned by GetReplicationMessages after a successful execution.\n\tSuccess                        *replicator.GetReplicationMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                    `json:\"badRequestError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                 `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                   `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError     `json:\"clientVersionNotSupportedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetReplicationMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetReplicationMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetReplicationMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetReplicationMessagesResponse_Read(w wire.Value) (*replicator.GetReplicationMessagesResponse, error) {\n\tvar v replicator.GetReplicationMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetReplicationMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetReplicationMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetReplicationMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetReplicationMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetReplicationMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetReplicationMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetReplicationMessages_Result struct could not be encoded.\nfunc (v *AdminService_GetReplicationMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetReplicationMessagesResponse_Decode(sr stream.Reader) (*replicator.GetReplicationMessagesResponse, error) {\n\tvar v replicator.GetReplicationMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetReplicationMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetReplicationMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetReplicationMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetReplicationMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetReplicationMessages_Result\n// struct.\nfunc (v *AdminService_GetReplicationMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetReplicationMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetReplicationMessages_Result match the\n// provided AdminService_GetReplicationMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetReplicationMessages_Result) Equals(rhs *AdminService_GetReplicationMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetReplicationMessages_Result.\nfunc (v *AdminService_GetReplicationMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Result) GetSuccess() (o *replicator.GetReplicationMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetReplicationMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetReplicationMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *AdminService_GetReplicationMessages_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_GetReplicationMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetReplicationMessages_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *AdminService_GetReplicationMessages_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetReplicationMessages\" for this struct.\nfunc (v *AdminService_GetReplicationMessages_Result) MethodName() string {\n\treturn \"GetReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetReplicationMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_GetWorkflowExecutionRawHistoryV2_Args represents the arguments for the AdminService.GetWorkflowExecutionRawHistoryV2 function.\n//\n// The arguments for GetWorkflowExecutionRawHistoryV2 are sent and received over the wire as this struct.\ntype AdminService_GetWorkflowExecutionRawHistoryV2_Args struct {\n\tGetRequest *GetWorkflowExecutionRawHistoryV2Request `json:\"getRequest,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.GetRequest != nil {\n\t\tw, err = v.GetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetWorkflowExecutionRawHistoryV2Request_Read(w wire.Value) (*GetWorkflowExecutionRawHistoryV2Request, error) {\n\tvar v GetWorkflowExecutionRawHistoryV2Request\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetWorkflowExecutionRawHistoryV2_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.GetRequest, err = _GetWorkflowExecutionRawHistoryV2Request_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct could not be encoded.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.GetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.GetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetWorkflowExecutionRawHistoryV2Request_Decode(sr stream.Reader) (*GetWorkflowExecutionRawHistoryV2Request, error) {\n\tvar v GetWorkflowExecutionRawHistoryV2Request\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetWorkflowExecutionRawHistoryV2_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.GetRequest, err = _GetWorkflowExecutionRawHistoryV2Request_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetWorkflowExecutionRawHistoryV2_Args\n// struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.GetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"GetRequest: %v\", v.GetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetWorkflowExecutionRawHistoryV2_Args match the\n// provided AdminService_GetWorkflowExecutionRawHistoryV2_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) Equals(rhs *AdminService_GetWorkflowExecutionRawHistoryV2_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.GetRequest == nil && rhs.GetRequest == nil) || (v.GetRequest != nil && rhs.GetRequest != nil && v.GetRequest.Equals(rhs.GetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetWorkflowExecutionRawHistoryV2_Args.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.GetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"getRequest\", v.GetRequest))\n\t}\n\treturn err\n}\n\n// GetGetRequest returns the value of GetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) GetGetRequest() (o *GetWorkflowExecutionRawHistoryV2Request) {\n\tif v != nil && v.GetRequest != nil {\n\t\treturn v.GetRequest\n\t}\n\n\treturn\n}\n\n// IsSetGetRequest returns true if GetRequest is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) IsSetGetRequest() bool {\n\treturn v != nil && v.GetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetWorkflowExecutionRawHistoryV2\" for this struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) MethodName() string {\n\treturn \"GetWorkflowExecutionRawHistoryV2\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_GetWorkflowExecutionRawHistoryV2_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.GetWorkflowExecutionRawHistoryV2\n// function.\nvar AdminService_GetWorkflowExecutionRawHistoryV2_Helper = struct {\n\t// Args accepts the parameters of GetWorkflowExecutionRawHistoryV2 in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tgetRequest *GetWorkflowExecutionRawHistoryV2Request,\n\t) *AdminService_GetWorkflowExecutionRawHistoryV2_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetWorkflowExecutionRawHistoryV2.\n\t//\n\t// An error can be thrown by GetWorkflowExecutionRawHistoryV2 only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetWorkflowExecutionRawHistoryV2\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetWorkflowExecutionRawHistoryV2 into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetWorkflowExecutionRawHistoryV2\n\t//\n\t//   value, err := GetWorkflowExecutionRawHistoryV2(args)\n\t//   result, err := AdminService_GetWorkflowExecutionRawHistoryV2_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetWorkflowExecutionRawHistoryV2: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetWorkflowExecutionRawHistoryV2Response, error) (*AdminService_GetWorkflowExecutionRawHistoryV2_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetWorkflowExecutionRawHistoryV2\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetWorkflowExecutionRawHistoryV2 threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_GetWorkflowExecutionRawHistoryV2_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_GetWorkflowExecutionRawHistoryV2_Result) (*GetWorkflowExecutionRawHistoryV2Response, error)\n}{}\n\nfunc init() {\n\tAdminService_GetWorkflowExecutionRawHistoryV2_Helper.Args = func(\n\t\tgetRequest *GetWorkflowExecutionRawHistoryV2Request,\n\t) *AdminService_GetWorkflowExecutionRawHistoryV2_Args {\n\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Args{\n\t\t\tGetRequest: getRequest,\n\t\t}\n\t}\n\n\tAdminService_GetWorkflowExecutionRawHistoryV2_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_GetWorkflowExecutionRawHistoryV2_Helper.WrapResponse = func(success *GetWorkflowExecutionRawHistoryV2Response, err error) (*AdminService_GetWorkflowExecutionRawHistoryV2_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetWorkflowExecutionRawHistoryV2_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetWorkflowExecutionRawHistoryV2_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetWorkflowExecutionRawHistoryV2_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_GetWorkflowExecutionRawHistoryV2_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_GetWorkflowExecutionRawHistoryV2_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_GetWorkflowExecutionRawHistoryV2_Helper.UnwrapResponse = func(result *AdminService_GetWorkflowExecutionRawHistoryV2_Result) (success *GetWorkflowExecutionRawHistoryV2Response, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_GetWorkflowExecutionRawHistoryV2_Result represents the result of a AdminService.GetWorkflowExecutionRawHistoryV2 function call.\n//\n// The result of a GetWorkflowExecutionRawHistoryV2 execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_GetWorkflowExecutionRawHistoryV2_Result struct {\n\t// Value returned by GetWorkflowExecutionRawHistoryV2 after a successful execution.\n\tSuccess              *GetWorkflowExecutionRawHistoryV2Response `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError                   `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError              `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError              `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError                  `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetWorkflowExecutionRawHistoryV2Response_Read(w wire.Value) (*GetWorkflowExecutionRawHistoryV2Response, error) {\n\tvar v GetWorkflowExecutionRawHistoryV2Response\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_GetWorkflowExecutionRawHistoryV2_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetWorkflowExecutionRawHistoryV2Response_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct could not be encoded.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetWorkflowExecutionRawHistoryV2Response_Decode(sr stream.Reader) (*GetWorkflowExecutionRawHistoryV2Response, error) {\n\tvar v GetWorkflowExecutionRawHistoryV2Response\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_GetWorkflowExecutionRawHistoryV2_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetWorkflowExecutionRawHistoryV2Response_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_GetWorkflowExecutionRawHistoryV2_Result\n// struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_GetWorkflowExecutionRawHistoryV2_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_GetWorkflowExecutionRawHistoryV2_Result match the\n// provided AdminService_GetWorkflowExecutionRawHistoryV2_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) Equals(rhs *AdminService_GetWorkflowExecutionRawHistoryV2_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_GetWorkflowExecutionRawHistoryV2_Result.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) GetSuccess() (o *GetWorkflowExecutionRawHistoryV2Response) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetWorkflowExecutionRawHistoryV2\" for this struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) MethodName() string {\n\treturn \"GetWorkflowExecutionRawHistoryV2\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_GetWorkflowExecutionRawHistoryV2_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_ListDynamicConfig_Args represents the arguments for the AdminService.ListDynamicConfig function.\n//\n// The arguments for ListDynamicConfig are sent and received over the wire as this struct.\ntype AdminService_ListDynamicConfig_Args struct {\n\tRequest *ListDynamicConfigRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_ListDynamicConfig_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ListDynamicConfig_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListDynamicConfigRequest_Read(w wire.Value) (*ListDynamicConfigRequest, error) {\n\tvar v ListDynamicConfigRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ListDynamicConfig_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ListDynamicConfig_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ListDynamicConfig_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ListDynamicConfig_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ListDynamicConfigRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ListDynamicConfig_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ListDynamicConfig_Args struct could not be encoded.\nfunc (v *AdminService_ListDynamicConfig_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListDynamicConfigRequest_Decode(sr stream.Reader) (*ListDynamicConfigRequest, error) {\n\tvar v ListDynamicConfigRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ListDynamicConfig_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ListDynamicConfig_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ListDynamicConfig_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ListDynamicConfigRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ListDynamicConfig_Args\n// struct.\nfunc (v *AdminService_ListDynamicConfig_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ListDynamicConfig_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ListDynamicConfig_Args match the\n// provided AdminService_ListDynamicConfig_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ListDynamicConfig_Args) Equals(rhs *AdminService_ListDynamicConfig_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ListDynamicConfig_Args.\nfunc (v *AdminService_ListDynamicConfig_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ListDynamicConfig_Args) GetRequest() (o *ListDynamicConfigRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_ListDynamicConfig_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListDynamicConfig\" for this struct.\nfunc (v *AdminService_ListDynamicConfig_Args) MethodName() string {\n\treturn \"ListDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_ListDynamicConfig_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_ListDynamicConfig_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.ListDynamicConfig\n// function.\nvar AdminService_ListDynamicConfig_Helper = struct {\n\t// Args accepts the parameters of ListDynamicConfig in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *ListDynamicConfigRequest,\n\t) *AdminService_ListDynamicConfig_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListDynamicConfig.\n\t//\n\t// An error can be thrown by ListDynamicConfig only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListDynamicConfig\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListDynamicConfig into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListDynamicConfig\n\t//\n\t//   value, err := ListDynamicConfig(args)\n\t//   result, err := AdminService_ListDynamicConfig_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListDynamicConfig: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*ListDynamicConfigResponse, error) (*AdminService_ListDynamicConfig_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListDynamicConfig\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListDynamicConfig threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_ListDynamicConfig_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_ListDynamicConfig_Result) (*ListDynamicConfigResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_ListDynamicConfig_Helper.Args = func(\n\t\trequest *ListDynamicConfigRequest,\n\t) *AdminService_ListDynamicConfig_Args {\n\t\treturn &AdminService_ListDynamicConfig_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_ListDynamicConfig_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_ListDynamicConfig_Helper.WrapResponse = func(success *ListDynamicConfigResponse, err error) (*AdminService_ListDynamicConfig_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_ListDynamicConfig_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ListDynamicConfig_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_ListDynamicConfig_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_ListDynamicConfig_Helper.UnwrapResponse = func(result *AdminService_ListDynamicConfig_Result) (success *ListDynamicConfigResponse, err error) {\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_ListDynamicConfig_Result represents the result of a AdminService.ListDynamicConfig function call.\n//\n// The result of a ListDynamicConfig execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_ListDynamicConfig_Result struct {\n\t// Value returned by ListDynamicConfig after a successful execution.\n\tSuccess              *ListDynamicConfigResponse   `json:\"success,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_ListDynamicConfig_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ListDynamicConfig_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_ListDynamicConfig_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListDynamicConfigResponse_Read(w wire.Value) (*ListDynamicConfigResponse, error) {\n\tvar v ListDynamicConfigResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ListDynamicConfig_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ListDynamicConfig_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ListDynamicConfig_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ListDynamicConfig_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListDynamicConfigResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ListDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ListDynamicConfig_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ListDynamicConfig_Result struct could not be encoded.\nfunc (v *AdminService_ListDynamicConfig_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ListDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListDynamicConfigResponse_Decode(sr stream.Reader) (*ListDynamicConfigResponse, error) {\n\tvar v ListDynamicConfigResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ListDynamicConfig_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ListDynamicConfig_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ListDynamicConfig_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListDynamicConfigResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ListDynamicConfig_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ListDynamicConfig_Result\n// struct.\nfunc (v *AdminService_ListDynamicConfig_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ListDynamicConfig_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ListDynamicConfig_Result match the\n// provided AdminService_ListDynamicConfig_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ListDynamicConfig_Result) Equals(rhs *AdminService_ListDynamicConfig_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ListDynamicConfig_Result.\nfunc (v *AdminService_ListDynamicConfig_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ListDynamicConfig_Result) GetSuccess() (o *ListDynamicConfigResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_ListDynamicConfig_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ListDynamicConfig_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_ListDynamicConfig_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListDynamicConfig\" for this struct.\nfunc (v *AdminService_ListDynamicConfig_Result) MethodName() string {\n\treturn \"ListDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_ListDynamicConfig_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_MaintainCorruptWorkflow_Args represents the arguments for the AdminService.MaintainCorruptWorkflow function.\n//\n// The arguments for MaintainCorruptWorkflow are sent and received over the wire as this struct.\ntype AdminService_MaintainCorruptWorkflow_Args struct {\n\tRequest *AdminMaintainWorkflowRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_MaintainCorruptWorkflow_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AdminMaintainWorkflowRequest_Read(w wire.Value) (*AdminMaintainWorkflowRequest, error) {\n\tvar v AdminMaintainWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_MaintainCorruptWorkflow_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_MaintainCorruptWorkflow_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_MaintainCorruptWorkflow_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _AdminMaintainWorkflowRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_MaintainCorruptWorkflow_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_MaintainCorruptWorkflow_Args struct could not be encoded.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AdminMaintainWorkflowRequest_Decode(sr stream.Reader) (*AdminMaintainWorkflowRequest, error) {\n\tvar v AdminMaintainWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_MaintainCorruptWorkflow_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_MaintainCorruptWorkflow_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _AdminMaintainWorkflowRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_MaintainCorruptWorkflow_Args\n// struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_MaintainCorruptWorkflow_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_MaintainCorruptWorkflow_Args match the\n// provided AdminService_MaintainCorruptWorkflow_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) Equals(rhs *AdminService_MaintainCorruptWorkflow_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_MaintainCorruptWorkflow_Args.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) GetRequest() (o *AdminMaintainWorkflowRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"MaintainCorruptWorkflow\" for this struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) MethodName() string {\n\treturn \"MaintainCorruptWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_MaintainCorruptWorkflow_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.MaintainCorruptWorkflow\n// function.\nvar AdminService_MaintainCorruptWorkflow_Helper = struct {\n\t// Args accepts the parameters of MaintainCorruptWorkflow in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *AdminMaintainWorkflowRequest,\n\t) *AdminService_MaintainCorruptWorkflow_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by MaintainCorruptWorkflow.\n\t//\n\t// An error can be thrown by MaintainCorruptWorkflow only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for MaintainCorruptWorkflow\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// MaintainCorruptWorkflow into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by MaintainCorruptWorkflow\n\t//\n\t//   value, err := MaintainCorruptWorkflow(args)\n\t//   result, err := AdminService_MaintainCorruptWorkflow_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from MaintainCorruptWorkflow: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*AdminMaintainWorkflowResponse, error) (*AdminService_MaintainCorruptWorkflow_Result, error)\n\n\t// UnwrapResponse takes the result struct for MaintainCorruptWorkflow\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if MaintainCorruptWorkflow threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_MaintainCorruptWorkflow_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_MaintainCorruptWorkflow_Result) (*AdminMaintainWorkflowResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_MaintainCorruptWorkflow_Helper.Args = func(\n\t\trequest *AdminMaintainWorkflowRequest,\n\t) *AdminService_MaintainCorruptWorkflow_Args {\n\t\treturn &AdminService_MaintainCorruptWorkflow_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_MaintainCorruptWorkflow_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_MaintainCorruptWorkflow_Helper.WrapResponse = func(success *AdminMaintainWorkflowResponse, err error) (*AdminService_MaintainCorruptWorkflow_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_MaintainCorruptWorkflow_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MaintainCorruptWorkflow_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_MaintainCorruptWorkflow_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MaintainCorruptWorkflow_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_MaintainCorruptWorkflow_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MaintainCorruptWorkflow_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_MaintainCorruptWorkflow_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_MaintainCorruptWorkflow_Helper.UnwrapResponse = func(result *AdminService_MaintainCorruptWorkflow_Result) (success *AdminMaintainWorkflowResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_MaintainCorruptWorkflow_Result represents the result of a AdminService.MaintainCorruptWorkflow function call.\n//\n// The result of a MaintainCorruptWorkflow execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_MaintainCorruptWorkflow_Result struct {\n\t// Value returned by MaintainCorruptWorkflow after a successful execution.\n\tSuccess              *AdminMaintainWorkflowResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError   `json:\"entityNotExistError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError   `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_MaintainCorruptWorkflow_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_MaintainCorruptWorkflow_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AdminMaintainWorkflowResponse_Read(w wire.Value) (*AdminMaintainWorkflowResponse, error) {\n\tvar v AdminMaintainWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_MaintainCorruptWorkflow_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_MaintainCorruptWorkflow_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_MaintainCorruptWorkflow_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _AdminMaintainWorkflowResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MaintainCorruptWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_MaintainCorruptWorkflow_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_MaintainCorruptWorkflow_Result struct could not be encoded.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MaintainCorruptWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AdminMaintainWorkflowResponse_Decode(sr stream.Reader) (*AdminMaintainWorkflowResponse, error) {\n\tvar v AdminMaintainWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_MaintainCorruptWorkflow_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_MaintainCorruptWorkflow_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _AdminMaintainWorkflowResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MaintainCorruptWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_MaintainCorruptWorkflow_Result\n// struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_MaintainCorruptWorkflow_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_MaintainCorruptWorkflow_Result match the\n// provided AdminService_MaintainCorruptWorkflow_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) Equals(rhs *AdminService_MaintainCorruptWorkflow_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_MaintainCorruptWorkflow_Result.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) GetSuccess() (o *AdminMaintainWorkflowResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"MaintainCorruptWorkflow\" for this struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) MethodName() string {\n\treturn \"MaintainCorruptWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_MaintainCorruptWorkflow_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_MergeDLQMessages_Args represents the arguments for the AdminService.MergeDLQMessages function.\n//\n// The arguments for MergeDLQMessages are sent and received over the wire as this struct.\ntype AdminService_MergeDLQMessages_Args struct {\n\tRequest *replicator.MergeDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_MergeDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_MergeDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _MergeDLQMessagesRequest_Read(w wire.Value) (*replicator.MergeDLQMessagesRequest, error) {\n\tvar v replicator.MergeDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_MergeDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_MergeDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_MergeDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_MergeDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _MergeDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_MergeDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_MergeDLQMessages_Args struct could not be encoded.\nfunc (v *AdminService_MergeDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _MergeDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.MergeDLQMessagesRequest, error) {\n\tvar v replicator.MergeDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_MergeDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_MergeDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_MergeDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _MergeDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_MergeDLQMessages_Args\n// struct.\nfunc (v *AdminService_MergeDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_MergeDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_MergeDLQMessages_Args match the\n// provided AdminService_MergeDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_MergeDLQMessages_Args) Equals(rhs *AdminService_MergeDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_MergeDLQMessages_Args.\nfunc (v *AdminService_MergeDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Args) GetRequest() (o *replicator.MergeDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_MergeDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"MergeDLQMessages\" for this struct.\nfunc (v *AdminService_MergeDLQMessages_Args) MethodName() string {\n\treturn \"MergeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_MergeDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_MergeDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.MergeDLQMessages\n// function.\nvar AdminService_MergeDLQMessages_Helper = struct {\n\t// Args accepts the parameters of MergeDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.MergeDLQMessagesRequest,\n\t) *AdminService_MergeDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by MergeDLQMessages.\n\t//\n\t// An error can be thrown by MergeDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for MergeDLQMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// MergeDLQMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by MergeDLQMessages\n\t//\n\t//   value, err := MergeDLQMessages(args)\n\t//   result, err := AdminService_MergeDLQMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from MergeDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.MergeDLQMessagesResponse, error) (*AdminService_MergeDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for MergeDLQMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if MergeDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_MergeDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_MergeDLQMessages_Result) (*replicator.MergeDLQMessagesResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_MergeDLQMessages_Helper.Args = func(\n\t\trequest *replicator.MergeDLQMessagesRequest,\n\t) *AdminService_MergeDLQMessages_Args {\n\t\treturn &AdminService_MergeDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_MergeDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_MergeDLQMessages_Helper.WrapResponse = func(success *replicator.MergeDLQMessagesResponse, err error) (*AdminService_MergeDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_MergeDLQMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MergeDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_MergeDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MergeDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_MergeDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MergeDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_MergeDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_MergeDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_MergeDLQMessages_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_MergeDLQMessages_Helper.UnwrapResponse = func(result *AdminService_MergeDLQMessages_Result) (success *replicator.MergeDLQMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_MergeDLQMessages_Result represents the result of a AdminService.MergeDLQMessages function call.\n//\n// The result of a MergeDLQMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_MergeDLQMessages_Result struct {\n\t// Value returned by MergeDLQMessages after a successful execution.\n\tSuccess              *replicator.MergeDLQMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError         `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_MergeDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_MergeDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_MergeDLQMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _MergeDLQMessagesResponse_Read(w wire.Value) (*replicator.MergeDLQMessagesResponse, error) {\n\tvar v replicator.MergeDLQMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_MergeDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_MergeDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_MergeDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_MergeDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _MergeDLQMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_MergeDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_MergeDLQMessages_Result struct could not be encoded.\nfunc (v *AdminService_MergeDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _MergeDLQMessagesResponse_Decode(sr stream.Reader) (*replicator.MergeDLQMessagesResponse, error) {\n\tvar v replicator.MergeDLQMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_MergeDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_MergeDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_MergeDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _MergeDLQMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_MergeDLQMessages_Result\n// struct.\nfunc (v *AdminService_MergeDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_MergeDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_MergeDLQMessages_Result match the\n// provided AdminService_MergeDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_MergeDLQMessages_Result) Equals(rhs *AdminService_MergeDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_MergeDLQMessages_Result.\nfunc (v *AdminService_MergeDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Result) GetSuccess() (o *replicator.MergeDLQMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_MergeDLQMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_MergeDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_MergeDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_MergeDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_MergeDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_MergeDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"MergeDLQMessages\" for this struct.\nfunc (v *AdminService_MergeDLQMessages_Result) MethodName() string {\n\treturn \"MergeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_MergeDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_PurgeDLQMessages_Args represents the arguments for the AdminService.PurgeDLQMessages function.\n//\n// The arguments for PurgeDLQMessages are sent and received over the wire as this struct.\ntype AdminService_PurgeDLQMessages_Args struct {\n\tRequest *replicator.PurgeDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_PurgeDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_PurgeDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PurgeDLQMessagesRequest_Read(w wire.Value) (*replicator.PurgeDLQMessagesRequest, error) {\n\tvar v replicator.PurgeDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_PurgeDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_PurgeDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_PurgeDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_PurgeDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _PurgeDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_PurgeDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_PurgeDLQMessages_Args struct could not be encoded.\nfunc (v *AdminService_PurgeDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PurgeDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.PurgeDLQMessagesRequest, error) {\n\tvar v replicator.PurgeDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_PurgeDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_PurgeDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_PurgeDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _PurgeDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_PurgeDLQMessages_Args\n// struct.\nfunc (v *AdminService_PurgeDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_PurgeDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_PurgeDLQMessages_Args match the\n// provided AdminService_PurgeDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_PurgeDLQMessages_Args) Equals(rhs *AdminService_PurgeDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_PurgeDLQMessages_Args.\nfunc (v *AdminService_PurgeDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_PurgeDLQMessages_Args) GetRequest() (o *replicator.PurgeDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_PurgeDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PurgeDLQMessages\" for this struct.\nfunc (v *AdminService_PurgeDLQMessages_Args) MethodName() string {\n\treturn \"PurgeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_PurgeDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_PurgeDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.PurgeDLQMessages\n// function.\nvar AdminService_PurgeDLQMessages_Helper = struct {\n\t// Args accepts the parameters of PurgeDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.PurgeDLQMessagesRequest,\n\t) *AdminService_PurgeDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PurgeDLQMessages.\n\t//\n\t// An error can be thrown by PurgeDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PurgeDLQMessages\n\t// given the error returned by it. The provided error may\n\t// be nil if PurgeDLQMessages did not fail.\n\t//\n\t// This allows mapping errors returned by PurgeDLQMessages into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// PurgeDLQMessages\n\t//\n\t//   err := PurgeDLQMessages(args)\n\t//   result, err := AdminService_PurgeDLQMessages_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PurgeDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_PurgeDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for PurgeDLQMessages\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if PurgeDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_PurgeDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_PurgeDLQMessages_Result) error\n}{}\n\nfunc init() {\n\tAdminService_PurgeDLQMessages_Helper.Args = func(\n\t\trequest *replicator.PurgeDLQMessagesRequest,\n\t) *AdminService_PurgeDLQMessages_Args {\n\t\treturn &AdminService_PurgeDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_PurgeDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_PurgeDLQMessages_Helper.WrapResponse = func(err error) (*AdminService_PurgeDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_PurgeDLQMessages_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_PurgeDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_PurgeDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_PurgeDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_PurgeDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_PurgeDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_PurgeDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_PurgeDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_PurgeDLQMessages_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_PurgeDLQMessages_Helper.UnwrapResponse = func(result *AdminService_PurgeDLQMessages_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_PurgeDLQMessages_Result represents the result of a AdminService.PurgeDLQMessages function call.\n//\n// The result of a PurgeDLQMessages execution is sent and received over the wire as this struct.\ntype AdminService_PurgeDLQMessages_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_PurgeDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_PurgeDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_PurgeDLQMessages_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_PurgeDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_PurgeDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_PurgeDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_PurgeDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_PurgeDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_PurgeDLQMessages_Result struct could not be encoded.\nfunc (v *AdminService_PurgeDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_PurgeDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_PurgeDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_PurgeDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_PurgeDLQMessages_Result\n// struct.\nfunc (v *AdminService_PurgeDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_PurgeDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_PurgeDLQMessages_Result match the\n// provided AdminService_PurgeDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_PurgeDLQMessages_Result) Equals(rhs *AdminService_PurgeDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_PurgeDLQMessages_Result.\nfunc (v *AdminService_PurgeDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_PurgeDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_PurgeDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_PurgeDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_PurgeDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_PurgeDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_PurgeDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_PurgeDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_PurgeDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PurgeDLQMessages\" for this struct.\nfunc (v *AdminService_PurgeDLQMessages_Result) MethodName() string {\n\treturn \"PurgeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_PurgeDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_ReadDLQMessages_Args represents the arguments for the AdminService.ReadDLQMessages function.\n//\n// The arguments for ReadDLQMessages are sent and received over the wire as this struct.\ntype AdminService_ReadDLQMessages_Args struct {\n\tRequest *replicator.ReadDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_ReadDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ReadDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReadDLQMessagesRequest_Read(w wire.Value) (*replicator.ReadDLQMessagesRequest, error) {\n\tvar v replicator.ReadDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ReadDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ReadDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ReadDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ReadDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ReadDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ReadDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ReadDLQMessages_Args struct could not be encoded.\nfunc (v *AdminService_ReadDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReadDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.ReadDLQMessagesRequest, error) {\n\tvar v replicator.ReadDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ReadDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ReadDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ReadDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ReadDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ReadDLQMessages_Args\n// struct.\nfunc (v *AdminService_ReadDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ReadDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ReadDLQMessages_Args match the\n// provided AdminService_ReadDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ReadDLQMessages_Args) Equals(rhs *AdminService_ReadDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ReadDLQMessages_Args.\nfunc (v *AdminService_ReadDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Args) GetRequest() (o *replicator.ReadDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_ReadDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ReadDLQMessages\" for this struct.\nfunc (v *AdminService_ReadDLQMessages_Args) MethodName() string {\n\treturn \"ReadDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_ReadDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_ReadDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.ReadDLQMessages\n// function.\nvar AdminService_ReadDLQMessages_Helper = struct {\n\t// Args accepts the parameters of ReadDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.ReadDLQMessagesRequest,\n\t) *AdminService_ReadDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ReadDLQMessages.\n\t//\n\t// An error can be thrown by ReadDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ReadDLQMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ReadDLQMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ReadDLQMessages\n\t//\n\t//   value, err := ReadDLQMessages(args)\n\t//   result, err := AdminService_ReadDLQMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ReadDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.ReadDLQMessagesResponse, error) (*AdminService_ReadDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for ReadDLQMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ReadDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_ReadDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_ReadDLQMessages_Result) (*replicator.ReadDLQMessagesResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_ReadDLQMessages_Helper.Args = func(\n\t\trequest *replicator.ReadDLQMessagesRequest,\n\t) *AdminService_ReadDLQMessages_Args {\n\t\treturn &AdminService_ReadDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_ReadDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_ReadDLQMessages_Helper.WrapResponse = func(success *replicator.ReadDLQMessagesResponse, err error) (*AdminService_ReadDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_ReadDLQMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReadDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReadDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReadDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReadDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReadDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReadDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReadDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReadDLQMessages_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_ReadDLQMessages_Helper.UnwrapResponse = func(result *AdminService_ReadDLQMessages_Result) (success *replicator.ReadDLQMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_ReadDLQMessages_Result represents the result of a AdminService.ReadDLQMessages function call.\n//\n// The result of a ReadDLQMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_ReadDLQMessages_Result struct {\n\t// Value returned by ReadDLQMessages after a successful execution.\n\tSuccess              *replicator.ReadDLQMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError        `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_ReadDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ReadDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_ReadDLQMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReadDLQMessagesResponse_Read(w wire.Value) (*replicator.ReadDLQMessagesResponse, error) {\n\tvar v replicator.ReadDLQMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ReadDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ReadDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ReadDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ReadDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ReadDLQMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ReadDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ReadDLQMessages_Result struct could not be encoded.\nfunc (v *AdminService_ReadDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReadDLQMessagesResponse_Decode(sr stream.Reader) (*replicator.ReadDLQMessagesResponse, error) {\n\tvar v replicator.ReadDLQMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ReadDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ReadDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ReadDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ReadDLQMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ReadDLQMessages_Result\n// struct.\nfunc (v *AdminService_ReadDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ReadDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ReadDLQMessages_Result match the\n// provided AdminService_ReadDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ReadDLQMessages_Result) Equals(rhs *AdminService_ReadDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ReadDLQMessages_Result.\nfunc (v *AdminService_ReadDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Result) GetSuccess() (o *replicator.ReadDLQMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_ReadDLQMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_ReadDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_ReadDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_ReadDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReadDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_ReadDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ReadDLQMessages\" for this struct.\nfunc (v *AdminService_ReadDLQMessages_Result) MethodName() string {\n\treturn \"ReadDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_ReadDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_ReapplyEvents_Args represents the arguments for the AdminService.ReapplyEvents function.\n//\n// The arguments for ReapplyEvents are sent and received over the wire as this struct.\ntype AdminService_ReapplyEvents_Args struct {\n\tReapplyEventsRequest *shared.ReapplyEventsRequest `json:\"reapplyEventsRequest,omitempty\"`\n}\n\n// ToWire translates a AdminService_ReapplyEvents_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ReapplyEvents_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ReapplyEventsRequest != nil {\n\t\tw, err = v.ReapplyEventsRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReapplyEventsRequest_Read(w wire.Value) (*shared.ReapplyEventsRequest, error) {\n\tvar v shared.ReapplyEventsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ReapplyEvents_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ReapplyEvents_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ReapplyEvents_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ReapplyEvents_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReapplyEventsRequest, err = _ReapplyEventsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ReapplyEvents_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ReapplyEvents_Args struct could not be encoded.\nfunc (v *AdminService_ReapplyEvents_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ReapplyEventsRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReapplyEventsRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReapplyEventsRequest_Decode(sr stream.Reader) (*shared.ReapplyEventsRequest, error) {\n\tvar v shared.ReapplyEventsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ReapplyEvents_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ReapplyEvents_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ReapplyEvents_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ReapplyEventsRequest, err = _ReapplyEventsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ReapplyEvents_Args\n// struct.\nfunc (v *AdminService_ReapplyEvents_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ReapplyEventsRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReapplyEventsRequest: %v\", v.ReapplyEventsRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ReapplyEvents_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ReapplyEvents_Args match the\n// provided AdminService_ReapplyEvents_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ReapplyEvents_Args) Equals(rhs *AdminService_ReapplyEvents_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ReapplyEventsRequest == nil && rhs.ReapplyEventsRequest == nil) || (v.ReapplyEventsRequest != nil && rhs.ReapplyEventsRequest != nil && v.ReapplyEventsRequest.Equals(rhs.ReapplyEventsRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ReapplyEvents_Args.\nfunc (v *AdminService_ReapplyEvents_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ReapplyEventsRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"reapplyEventsRequest\", v.ReapplyEventsRequest))\n\t}\n\treturn err\n}\n\n// GetReapplyEventsRequest returns the value of ReapplyEventsRequest if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Args) GetReapplyEventsRequest() (o *shared.ReapplyEventsRequest) {\n\tif v != nil && v.ReapplyEventsRequest != nil {\n\t\treturn v.ReapplyEventsRequest\n\t}\n\n\treturn\n}\n\n// IsSetReapplyEventsRequest returns true if ReapplyEventsRequest is not nil.\nfunc (v *AdminService_ReapplyEvents_Args) IsSetReapplyEventsRequest() bool {\n\treturn v != nil && v.ReapplyEventsRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ReapplyEvents\" for this struct.\nfunc (v *AdminService_ReapplyEvents_Args) MethodName() string {\n\treturn \"ReapplyEvents\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_ReapplyEvents_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_ReapplyEvents_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.ReapplyEvents\n// function.\nvar AdminService_ReapplyEvents_Helper = struct {\n\t// Args accepts the parameters of ReapplyEvents in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\treapplyEventsRequest *shared.ReapplyEventsRequest,\n\t) *AdminService_ReapplyEvents_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ReapplyEvents.\n\t//\n\t// An error can be thrown by ReapplyEvents only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ReapplyEvents\n\t// given the error returned by it. The provided error may\n\t// be nil if ReapplyEvents did not fail.\n\t//\n\t// This allows mapping errors returned by ReapplyEvents into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ReapplyEvents\n\t//\n\t//   err := ReapplyEvents(args)\n\t//   result, err := AdminService_ReapplyEvents_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ReapplyEvents: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_ReapplyEvents_Result, error)\n\n\t// UnwrapResponse takes the result struct for ReapplyEvents\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ReapplyEvents threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_ReapplyEvents_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_ReapplyEvents_Result) error\n}{}\n\nfunc init() {\n\tAdminService_ReapplyEvents_Helper.Args = func(\n\t\treapplyEventsRequest *shared.ReapplyEventsRequest,\n\t) *AdminService_ReapplyEvents_Args {\n\t\treturn &AdminService_ReapplyEvents_Args{\n\t\t\tReapplyEventsRequest: reapplyEventsRequest,\n\t\t}\n\t}\n\n\tAdminService_ReapplyEvents_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_ReapplyEvents_Helper.WrapResponse = func(err error) (*AdminService_ReapplyEvents_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_ReapplyEvents_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReapplyEvents_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReapplyEvents_Result{BadRequestError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReapplyEvents_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReapplyEvents_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReapplyEvents_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReapplyEvents_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReapplyEvents_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReapplyEvents_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ReapplyEvents_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_ReapplyEvents_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_ReapplyEvents_Helper.UnwrapResponse = func(result *AdminService_ReapplyEvents_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_ReapplyEvents_Result represents the result of a AdminService.ReapplyEvents function call.\n//\n// The result of a ReapplyEvents execution is sent and received over the wire as this struct.\ntype AdminService_ReapplyEvents_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tDomainNotActiveError *shared.DomainNotActiveError `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError   *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_ReapplyEvents_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ReapplyEvents_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_ReapplyEvents_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainNotActiveError_Read(w wire.Value) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ReapplyEvents_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ReapplyEvents_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ReapplyEvents_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ReapplyEvents_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ReapplyEvents_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ReapplyEvents_Result struct could not be encoded.\nfunc (v *AdminService_ReapplyEvents_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainNotActiveError_Decode(sr stream.Reader) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ReapplyEvents_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ReapplyEvents_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ReapplyEvents_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ReapplyEvents_Result\n// struct.\nfunc (v *AdminService_ReapplyEvents_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ReapplyEvents_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ReapplyEvents_Result match the\n// provided AdminService_ReapplyEvents_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ReapplyEvents_Result) Equals(rhs *AdminService_ReapplyEvents_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ReapplyEvents_Result.\nfunc (v *AdminService_ReapplyEvents_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_ReapplyEvents_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *AdminService_ReapplyEvents_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *AdminService_ReapplyEvents_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_ReapplyEvents_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ReapplyEvents_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_ReapplyEvents_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ReapplyEvents\" for this struct.\nfunc (v *AdminService_ReapplyEvents_Result) MethodName() string {\n\treturn \"ReapplyEvents\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_ReapplyEvents_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_RefreshWorkflowTasks_Args represents the arguments for the AdminService.RefreshWorkflowTasks function.\n//\n// The arguments for RefreshWorkflowTasks are sent and received over the wire as this struct.\ntype AdminService_RefreshWorkflowTasks_Args struct {\n\tRequest *shared.RefreshWorkflowTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_RefreshWorkflowTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RefreshWorkflowTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RefreshWorkflowTasksRequest_Read(w wire.Value) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_RefreshWorkflowTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RefreshWorkflowTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RefreshWorkflowTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RefreshWorkflowTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RefreshWorkflowTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RefreshWorkflowTasks_Args struct could not be encoded.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RefreshWorkflowTasksRequest_Decode(sr stream.Reader) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_RefreshWorkflowTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RefreshWorkflowTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RefreshWorkflowTasks_Args\n// struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RefreshWorkflowTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RefreshWorkflowTasks_Args match the\n// provided AdminService_RefreshWorkflowTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) Equals(rhs *AdminService_RefreshWorkflowTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RefreshWorkflowTasks_Args.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) GetRequest() (o *shared.RefreshWorkflowTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_RefreshWorkflowTasks_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.RefreshWorkflowTasks\n// function.\nvar AdminService_RefreshWorkflowTasks_Helper = struct {\n\t// Args accepts the parameters of RefreshWorkflowTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RefreshWorkflowTasksRequest,\n\t) *AdminService_RefreshWorkflowTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RefreshWorkflowTasks.\n\t//\n\t// An error can be thrown by RefreshWorkflowTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RefreshWorkflowTasks\n\t// given the error returned by it. The provided error may\n\t// be nil if RefreshWorkflowTasks did not fail.\n\t//\n\t// This allows mapping errors returned by RefreshWorkflowTasks into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RefreshWorkflowTasks\n\t//\n\t//   err := RefreshWorkflowTasks(args)\n\t//   result, err := AdminService_RefreshWorkflowTasks_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RefreshWorkflowTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_RefreshWorkflowTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for RefreshWorkflowTasks\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RefreshWorkflowTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_RefreshWorkflowTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_RefreshWorkflowTasks_Result) error\n}{}\n\nfunc init() {\n\tAdminService_RefreshWorkflowTasks_Helper.Args = func(\n\t\trequest *shared.RefreshWorkflowTasksRequest,\n\t) *AdminService_RefreshWorkflowTasks_Args {\n\t\treturn &AdminService_RefreshWorkflowTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_RefreshWorkflowTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_RefreshWorkflowTasks_Helper.WrapResponse = func(err error) (*AdminService_RefreshWorkflowTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_RefreshWorkflowTasks_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RefreshWorkflowTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_RefreshWorkflowTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RefreshWorkflowTasks_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &AdminService_RefreshWorkflowTasks_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RefreshWorkflowTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_RefreshWorkflowTasks_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RefreshWorkflowTasks_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_RefreshWorkflowTasks_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_RefreshWorkflowTasks_Helper.UnwrapResponse = func(result *AdminService_RefreshWorkflowTasks_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_RefreshWorkflowTasks_Result represents the result of a AdminService.RefreshWorkflowTasks function call.\n//\n// The result of a RefreshWorkflowTasks execution is sent and received over the wire as this struct.\ntype AdminService_RefreshWorkflowTasks_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tDomainNotActiveError *shared.DomainNotActiveError `json:\"domainNotActiveError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_RefreshWorkflowTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RefreshWorkflowTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_RefreshWorkflowTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RefreshWorkflowTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RefreshWorkflowTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RefreshWorkflowTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RefreshWorkflowTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RefreshWorkflowTasks_Result struct could not be encoded.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_RefreshWorkflowTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RefreshWorkflowTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RefreshWorkflowTasks_Result\n// struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RefreshWorkflowTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RefreshWorkflowTasks_Result match the\n// provided AdminService_RefreshWorkflowTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) Equals(rhs *AdminService_RefreshWorkflowTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RefreshWorkflowTasks_Result.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_RefreshWorkflowTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_RemoveTask_Args represents the arguments for the AdminService.RemoveTask function.\n//\n// The arguments for RemoveTask are sent and received over the wire as this struct.\ntype AdminService_RemoveTask_Args struct {\n\tRequest *shared.RemoveTaskRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_RemoveTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RemoveTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RemoveTaskRequest_Read(w wire.Value) (*shared.RemoveTaskRequest, error) {\n\tvar v shared.RemoveTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_RemoveTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RemoveTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RemoveTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RemoveTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RemoveTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RemoveTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RemoveTask_Args struct could not be encoded.\nfunc (v *AdminService_RemoveTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RemoveTaskRequest_Decode(sr stream.Reader) (*shared.RemoveTaskRequest, error) {\n\tvar v shared.RemoveTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_RemoveTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RemoveTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RemoveTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RemoveTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RemoveTask_Args\n// struct.\nfunc (v *AdminService_RemoveTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RemoveTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RemoveTask_Args match the\n// provided AdminService_RemoveTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RemoveTask_Args) Equals(rhs *AdminService_RemoveTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RemoveTask_Args.\nfunc (v *AdminService_RemoveTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RemoveTask_Args) GetRequest() (o *shared.RemoveTaskRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_RemoveTask_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RemoveTask\" for this struct.\nfunc (v *AdminService_RemoveTask_Args) MethodName() string {\n\treturn \"RemoveTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_RemoveTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_RemoveTask_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.RemoveTask\n// function.\nvar AdminService_RemoveTask_Helper = struct {\n\t// Args accepts the parameters of RemoveTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RemoveTaskRequest,\n\t) *AdminService_RemoveTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RemoveTask.\n\t//\n\t// An error can be thrown by RemoveTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RemoveTask\n\t// given the error returned by it. The provided error may\n\t// be nil if RemoveTask did not fail.\n\t//\n\t// This allows mapping errors returned by RemoveTask into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RemoveTask\n\t//\n\t//   err := RemoveTask(args)\n\t//   result, err := AdminService_RemoveTask_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RemoveTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_RemoveTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for RemoveTask\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RemoveTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_RemoveTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_RemoveTask_Result) error\n}{}\n\nfunc init() {\n\tAdminService_RemoveTask_Helper.Args = func(\n\t\trequest *shared.RemoveTaskRequest,\n\t) *AdminService_RemoveTask_Args {\n\t\treturn &AdminService_RemoveTask_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_RemoveTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_RemoveTask_Helper.WrapResponse = func(err error) (*AdminService_RemoveTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_RemoveTask_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RemoveTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_RemoveTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RemoveTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_RemoveTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RemoveTask_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_RemoveTask_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_RemoveTask_Helper.UnwrapResponse = func(result *AdminService_RemoveTask_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_RemoveTask_Result represents the result of a AdminService.RemoveTask function call.\n//\n// The result of a RemoveTask execution is sent and received over the wire as this struct.\ntype AdminService_RemoveTask_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_RemoveTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RemoveTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_RemoveTask_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_RemoveTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RemoveTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RemoveTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RemoveTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RemoveTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RemoveTask_Result struct could not be encoded.\nfunc (v *AdminService_RemoveTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_RemoveTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RemoveTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RemoveTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RemoveTask_Result\n// struct.\nfunc (v *AdminService_RemoveTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RemoveTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RemoveTask_Result match the\n// provided AdminService_RemoveTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RemoveTask_Result) Equals(rhs *AdminService_RemoveTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RemoveTask_Result.\nfunc (v *AdminService_RemoveTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RemoveTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_RemoveTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RemoveTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_RemoveTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RemoveTask_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_RemoveTask_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RemoveTask\" for this struct.\nfunc (v *AdminService_RemoveTask_Result) MethodName() string {\n\treturn \"RemoveTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_RemoveTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_ResendReplicationTasks_Args represents the arguments for the AdminService.ResendReplicationTasks function.\n//\n// The arguments for ResendReplicationTasks are sent and received over the wire as this struct.\ntype AdminService_ResendReplicationTasks_Args struct {\n\tRequest *ResendReplicationTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_ResendReplicationTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ResendReplicationTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResendReplicationTasksRequest_Read(w wire.Value) (*ResendReplicationTasksRequest, error) {\n\tvar v ResendReplicationTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ResendReplicationTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ResendReplicationTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ResendReplicationTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ResendReplicationTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ResendReplicationTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ResendReplicationTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ResendReplicationTasks_Args struct could not be encoded.\nfunc (v *AdminService_ResendReplicationTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResendReplicationTasksRequest_Decode(sr stream.Reader) (*ResendReplicationTasksRequest, error) {\n\tvar v ResendReplicationTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ResendReplicationTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ResendReplicationTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ResendReplicationTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ResendReplicationTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ResendReplicationTasks_Args\n// struct.\nfunc (v *AdminService_ResendReplicationTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ResendReplicationTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ResendReplicationTasks_Args match the\n// provided AdminService_ResendReplicationTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ResendReplicationTasks_Args) Equals(rhs *AdminService_ResendReplicationTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ResendReplicationTasks_Args.\nfunc (v *AdminService_ResendReplicationTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResendReplicationTasks_Args) GetRequest() (o *ResendReplicationTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_ResendReplicationTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResendReplicationTasks\" for this struct.\nfunc (v *AdminService_ResendReplicationTasks_Args) MethodName() string {\n\treturn \"ResendReplicationTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_ResendReplicationTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_ResendReplicationTasks_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.ResendReplicationTasks\n// function.\nvar AdminService_ResendReplicationTasks_Helper = struct {\n\t// Args accepts the parameters of ResendReplicationTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *ResendReplicationTasksRequest,\n\t) *AdminService_ResendReplicationTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResendReplicationTasks.\n\t//\n\t// An error can be thrown by ResendReplicationTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResendReplicationTasks\n\t// given the error returned by it. The provided error may\n\t// be nil if ResendReplicationTasks did not fail.\n\t//\n\t// This allows mapping errors returned by ResendReplicationTasks into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ResendReplicationTasks\n\t//\n\t//   err := ResendReplicationTasks(args)\n\t//   result, err := AdminService_ResendReplicationTasks_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResendReplicationTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_ResendReplicationTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResendReplicationTasks\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ResendReplicationTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_ResendReplicationTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_ResendReplicationTasks_Result) error\n}{}\n\nfunc init() {\n\tAdminService_ResendReplicationTasks_Helper.Args = func(\n\t\trequest *ResendReplicationTasksRequest,\n\t) *AdminService_ResendReplicationTasks_Args {\n\t\treturn &AdminService_ResendReplicationTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_ResendReplicationTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_ResendReplicationTasks_Helper.WrapResponse = func(err error) (*AdminService_ResendReplicationTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_ResendReplicationTasks_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResendReplicationTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResendReplicationTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResendReplicationTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResendReplicationTasks_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResendReplicationTasks_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResendReplicationTasks_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_ResendReplicationTasks_Helper.UnwrapResponse = func(result *AdminService_ResendReplicationTasks_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_ResendReplicationTasks_Result represents the result of a AdminService.ResendReplicationTasks function call.\n//\n// The result of a ResendReplicationTasks execution is sent and received over the wire as this struct.\ntype AdminService_ResendReplicationTasks_Result struct {\n\tBadRequestError     *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tServiceBusyError    *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a AdminService_ResendReplicationTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ResendReplicationTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_ResendReplicationTasks_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_ResendReplicationTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ResendReplicationTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ResendReplicationTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ResendReplicationTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResendReplicationTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ResendReplicationTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ResendReplicationTasks_Result struct could not be encoded.\nfunc (v *AdminService_ResendReplicationTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResendReplicationTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_ResendReplicationTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ResendReplicationTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ResendReplicationTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResendReplicationTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ResendReplicationTasks_Result\n// struct.\nfunc (v *AdminService_ResendReplicationTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ResendReplicationTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ResendReplicationTasks_Result match the\n// provided AdminService_ResendReplicationTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ResendReplicationTasks_Result) Equals(rhs *AdminService_ResendReplicationTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ResendReplicationTasks_Result.\nfunc (v *AdminService_ResendReplicationTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResendReplicationTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_ResendReplicationTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResendReplicationTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_ResendReplicationTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResendReplicationTasks_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *AdminService_ResendReplicationTasks_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResendReplicationTasks\" for this struct.\nfunc (v *AdminService_ResendReplicationTasks_Result) MethodName() string {\n\treturn \"ResendReplicationTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_ResendReplicationTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_ResetQueue_Args represents the arguments for the AdminService.ResetQueue function.\n//\n// The arguments for ResetQueue are sent and received over the wire as this struct.\ntype AdminService_ResetQueue_Args struct {\n\tRequest *shared.ResetQueueRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_ResetQueue_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ResetQueue_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetQueueRequest_Read(w wire.Value) (*shared.ResetQueueRequest, error) {\n\tvar v shared.ResetQueueRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_ResetQueue_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ResetQueue_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ResetQueue_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ResetQueue_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ResetQueueRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ResetQueue_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ResetQueue_Args struct could not be encoded.\nfunc (v *AdminService_ResetQueue_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetQueueRequest_Decode(sr stream.Reader) (*shared.ResetQueueRequest, error) {\n\tvar v shared.ResetQueueRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_ResetQueue_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ResetQueue_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ResetQueue_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ResetQueueRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ResetQueue_Args\n// struct.\nfunc (v *AdminService_ResetQueue_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ResetQueue_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ResetQueue_Args match the\n// provided AdminService_ResetQueue_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ResetQueue_Args) Equals(rhs *AdminService_ResetQueue_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ResetQueue_Args.\nfunc (v *AdminService_ResetQueue_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResetQueue_Args) GetRequest() (o *shared.ResetQueueRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_ResetQueue_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetQueue\" for this struct.\nfunc (v *AdminService_ResetQueue_Args) MethodName() string {\n\treturn \"ResetQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_ResetQueue_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_ResetQueue_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.ResetQueue\n// function.\nvar AdminService_ResetQueue_Helper = struct {\n\t// Args accepts the parameters of ResetQueue in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.ResetQueueRequest,\n\t) *AdminService_ResetQueue_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetQueue.\n\t//\n\t// An error can be thrown by ResetQueue only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetQueue\n\t// given the error returned by it. The provided error may\n\t// be nil if ResetQueue did not fail.\n\t//\n\t// This allows mapping errors returned by ResetQueue into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ResetQueue\n\t//\n\t//   err := ResetQueue(args)\n\t//   result, err := AdminService_ResetQueue_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetQueue: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_ResetQueue_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetQueue\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ResetQueue threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_ResetQueue_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_ResetQueue_Result) error\n}{}\n\nfunc init() {\n\tAdminService_ResetQueue_Helper.Args = func(\n\t\trequest *shared.ResetQueueRequest,\n\t) *AdminService_ResetQueue_Args {\n\t\treturn &AdminService_ResetQueue_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_ResetQueue_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_ResetQueue_Helper.WrapResponse = func(err error) (*AdminService_ResetQueue_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_ResetQueue_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResetQueue_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResetQueue_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResetQueue_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResetQueue_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_ResetQueue_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &AdminService_ResetQueue_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_ResetQueue_Helper.UnwrapResponse = func(result *AdminService_ResetQueue_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_ResetQueue_Result represents the result of a AdminService.ResetQueue function call.\n//\n// The result of a ResetQueue execution is sent and received over the wire as this struct.\ntype AdminService_ResetQueue_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a AdminService_ResetQueue_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_ResetQueue_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_ResetQueue_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_ResetQueue_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_ResetQueue_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_ResetQueue_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_ResetQueue_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_ResetQueue_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_ResetQueue_Result struct could not be encoded.\nfunc (v *AdminService_ResetQueue_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_ResetQueue_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_ResetQueue_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_ResetQueue_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_ResetQueue_Result\n// struct.\nfunc (v *AdminService_ResetQueue_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_ResetQueue_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_ResetQueue_Result match the\n// provided AdminService_ResetQueue_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_ResetQueue_Result) Equals(rhs *AdminService_ResetQueue_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_ResetQueue_Result.\nfunc (v *AdminService_ResetQueue_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResetQueue_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_ResetQueue_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResetQueue_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_ResetQueue_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_ResetQueue_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *AdminService_ResetQueue_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetQueue\" for this struct.\nfunc (v *AdminService_ResetQueue_Result) MethodName() string {\n\treturn \"ResetQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_ResetQueue_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_RespondCrossClusterTasksCompleted_Args represents the arguments for the AdminService.RespondCrossClusterTasksCompleted function.\n//\n// The arguments for RespondCrossClusterTasksCompleted are sent and received over the wire as this struct.\ntype AdminService_RespondCrossClusterTasksCompleted_Args struct {\n\tRequest *shared.RespondCrossClusterTasksCompletedRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_RespondCrossClusterTasksCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondCrossClusterTasksCompletedRequest_Read(w wire.Value) (*shared.RespondCrossClusterTasksCompletedRequest, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_RespondCrossClusterTasksCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RespondCrossClusterTasksCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RespondCrossClusterTasksCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RespondCrossClusterTasksCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RespondCrossClusterTasksCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RespondCrossClusterTasksCompleted_Args struct could not be encoded.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondCrossClusterTasksCompletedRequest_Decode(sr stream.Reader) (*shared.RespondCrossClusterTasksCompletedRequest, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_RespondCrossClusterTasksCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RespondCrossClusterTasksCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RespondCrossClusterTasksCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RespondCrossClusterTasksCompleted_Args\n// struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RespondCrossClusterTasksCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RespondCrossClusterTasksCompleted_Args match the\n// provided AdminService_RespondCrossClusterTasksCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) Equals(rhs *AdminService_RespondCrossClusterTasksCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RespondCrossClusterTasksCompleted_Args.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) GetRequest() (o *shared.RespondCrossClusterTasksCompletedRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondCrossClusterTasksCompleted\" for this struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) MethodName() string {\n\treturn \"RespondCrossClusterTasksCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_RespondCrossClusterTasksCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.RespondCrossClusterTasksCompleted\n// function.\nvar AdminService_RespondCrossClusterTasksCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondCrossClusterTasksCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) *AdminService_RespondCrossClusterTasksCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondCrossClusterTasksCompleted.\n\t//\n\t// An error can be thrown by RespondCrossClusterTasksCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondCrossClusterTasksCompleted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RespondCrossClusterTasksCompleted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RespondCrossClusterTasksCompleted\n\t//\n\t//   value, err := RespondCrossClusterTasksCompleted(args)\n\t//   result, err := AdminService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondCrossClusterTasksCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RespondCrossClusterTasksCompletedResponse, error) (*AdminService_RespondCrossClusterTasksCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondCrossClusterTasksCompleted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RespondCrossClusterTasksCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_RespondCrossClusterTasksCompleted_Result) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_RespondCrossClusterTasksCompleted_Helper.Args = func(\n\t\trequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) *AdminService_RespondCrossClusterTasksCompleted_Args {\n\t\treturn &AdminService_RespondCrossClusterTasksCompleted_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_RespondCrossClusterTasksCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_RespondCrossClusterTasksCompleted_Helper.WrapResponse = func(success *shared.RespondCrossClusterTasksCompletedResponse, err error) (*AdminService_RespondCrossClusterTasksCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_RespondCrossClusterTasksCompleted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RespondCrossClusterTasksCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_RespondCrossClusterTasksCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RespondCrossClusterTasksCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_RespondCrossClusterTasksCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RespondCrossClusterTasksCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &AdminService_RespondCrossClusterTasksCompleted_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse = func(result *AdminService_RespondCrossClusterTasksCompleted_Result) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_RespondCrossClusterTasksCompleted_Result represents the result of a AdminService.RespondCrossClusterTasksCompleted function call.\n//\n// The result of a RespondCrossClusterTasksCompleted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_RespondCrossClusterTasksCompleted_Result struct {\n\t// Value returned by RespondCrossClusterTasksCompleted after a successful execution.\n\tSuccess              *shared.RespondCrossClusterTasksCompletedResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError                           `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError                      `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError                          `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a AdminService_RespondCrossClusterTasksCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondCrossClusterTasksCompletedResponse_Read(w wire.Value) (*shared.RespondCrossClusterTasksCompletedResponse, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_RespondCrossClusterTasksCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RespondCrossClusterTasksCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RespondCrossClusterTasksCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RespondCrossClusterTasksCompletedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RespondCrossClusterTasksCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RespondCrossClusterTasksCompleted_Result struct could not be encoded.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondCrossClusterTasksCompletedResponse_Decode(sr stream.Reader) (*shared.RespondCrossClusterTasksCompletedResponse, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_RespondCrossClusterTasksCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RespondCrossClusterTasksCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RespondCrossClusterTasksCompletedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RespondCrossClusterTasksCompleted_Result\n// struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RespondCrossClusterTasksCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RespondCrossClusterTasksCompleted_Result match the\n// provided AdminService_RespondCrossClusterTasksCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) Equals(rhs *AdminService_RespondCrossClusterTasksCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RespondCrossClusterTasksCompleted_Result.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) GetSuccess() (o *shared.RespondCrossClusterTasksCompletedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondCrossClusterTasksCompleted\" for this struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) MethodName() string {\n\treturn \"RespondCrossClusterTasksCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_RespondCrossClusterTasksCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_RestoreDynamicConfig_Args represents the arguments for the AdminService.RestoreDynamicConfig function.\n//\n// The arguments for RestoreDynamicConfig are sent and received over the wire as this struct.\ntype AdminService_RestoreDynamicConfig_Args struct {\n\tRequest *RestoreDynamicConfigRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_RestoreDynamicConfig_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RestoreDynamicConfig_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RestoreDynamicConfigRequest_Read(w wire.Value) (*RestoreDynamicConfigRequest, error) {\n\tvar v RestoreDynamicConfigRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_RestoreDynamicConfig_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RestoreDynamicConfig_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RestoreDynamicConfig_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RestoreDynamicConfig_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RestoreDynamicConfigRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RestoreDynamicConfig_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RestoreDynamicConfig_Args struct could not be encoded.\nfunc (v *AdminService_RestoreDynamicConfig_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RestoreDynamicConfigRequest_Decode(sr stream.Reader) (*RestoreDynamicConfigRequest, error) {\n\tvar v RestoreDynamicConfigRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_RestoreDynamicConfig_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RestoreDynamicConfig_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RestoreDynamicConfig_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RestoreDynamicConfigRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RestoreDynamicConfig_Args\n// struct.\nfunc (v *AdminService_RestoreDynamicConfig_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RestoreDynamicConfig_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RestoreDynamicConfig_Args match the\n// provided AdminService_RestoreDynamicConfig_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RestoreDynamicConfig_Args) Equals(rhs *AdminService_RestoreDynamicConfig_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RestoreDynamicConfig_Args.\nfunc (v *AdminService_RestoreDynamicConfig_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RestoreDynamicConfig_Args) GetRequest() (o *RestoreDynamicConfigRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_RestoreDynamicConfig_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RestoreDynamicConfig\" for this struct.\nfunc (v *AdminService_RestoreDynamicConfig_Args) MethodName() string {\n\treturn \"RestoreDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_RestoreDynamicConfig_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_RestoreDynamicConfig_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.RestoreDynamicConfig\n// function.\nvar AdminService_RestoreDynamicConfig_Helper = struct {\n\t// Args accepts the parameters of RestoreDynamicConfig in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *RestoreDynamicConfigRequest,\n\t) *AdminService_RestoreDynamicConfig_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RestoreDynamicConfig.\n\t//\n\t// An error can be thrown by RestoreDynamicConfig only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RestoreDynamicConfig\n\t// given the error returned by it. The provided error may\n\t// be nil if RestoreDynamicConfig did not fail.\n\t//\n\t// This allows mapping errors returned by RestoreDynamicConfig into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RestoreDynamicConfig\n\t//\n\t//   err := RestoreDynamicConfig(args)\n\t//   result, err := AdminService_RestoreDynamicConfig_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RestoreDynamicConfig: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_RestoreDynamicConfig_Result, error)\n\n\t// UnwrapResponse takes the result struct for RestoreDynamicConfig\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RestoreDynamicConfig threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_RestoreDynamicConfig_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_RestoreDynamicConfig_Result) error\n}{}\n\nfunc init() {\n\tAdminService_RestoreDynamicConfig_Helper.Args = func(\n\t\trequest *RestoreDynamicConfigRequest,\n\t) *AdminService_RestoreDynamicConfig_Args {\n\t\treturn &AdminService_RestoreDynamicConfig_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_RestoreDynamicConfig_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_RestoreDynamicConfig_Helper.WrapResponse = func(err error) (*AdminService_RestoreDynamicConfig_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_RestoreDynamicConfig_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RestoreDynamicConfig_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_RestoreDynamicConfig_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_RestoreDynamicConfig_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_RestoreDynamicConfig_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_RestoreDynamicConfig_Helper.UnwrapResponse = func(result *AdminService_RestoreDynamicConfig_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_RestoreDynamicConfig_Result represents the result of a AdminService.RestoreDynamicConfig function call.\n//\n// The result of a RestoreDynamicConfig execution is sent and received over the wire as this struct.\ntype AdminService_RestoreDynamicConfig_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_RestoreDynamicConfig_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_RestoreDynamicConfig_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_RestoreDynamicConfig_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_RestoreDynamicConfig_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_RestoreDynamicConfig_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_RestoreDynamicConfig_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_RestoreDynamicConfig_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RestoreDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_RestoreDynamicConfig_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_RestoreDynamicConfig_Result struct could not be encoded.\nfunc (v *AdminService_RestoreDynamicConfig_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RestoreDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_RestoreDynamicConfig_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_RestoreDynamicConfig_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_RestoreDynamicConfig_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_RestoreDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_RestoreDynamicConfig_Result\n// struct.\nfunc (v *AdminService_RestoreDynamicConfig_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_RestoreDynamicConfig_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_RestoreDynamicConfig_Result match the\n// provided AdminService_RestoreDynamicConfig_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_RestoreDynamicConfig_Result) Equals(rhs *AdminService_RestoreDynamicConfig_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_RestoreDynamicConfig_Result.\nfunc (v *AdminService_RestoreDynamicConfig_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RestoreDynamicConfig_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_RestoreDynamicConfig_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_RestoreDynamicConfig_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_RestoreDynamicConfig_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RestoreDynamicConfig\" for this struct.\nfunc (v *AdminService_RestoreDynamicConfig_Result) MethodName() string {\n\treturn \"RestoreDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_RestoreDynamicConfig_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args represents the arguments for the AdminService.UpdateDomainAsyncWorkflowConfiguraton function.\n//\n// The arguments for UpdateDomainAsyncWorkflowConfiguraton are sent and received over the wire as this struct.\ntype AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct {\n\tRequest *UpdateDomainAsyncWorkflowConfiguratonRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainAsyncWorkflowConfiguratonRequest_Read(w wire.Value) (*UpdateDomainAsyncWorkflowConfiguratonRequest, error) {\n\tvar v UpdateDomainAsyncWorkflowConfiguratonRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _UpdateDomainAsyncWorkflowConfiguratonRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct could not be encoded.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainAsyncWorkflowConfiguratonRequest_Decode(sr stream.Reader) (*UpdateDomainAsyncWorkflowConfiguratonRequest, error) {\n\tvar v UpdateDomainAsyncWorkflowConfiguratonRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _UpdateDomainAsyncWorkflowConfiguratonRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args\n// struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args match the\n// provided AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) Equals(rhs *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) GetRequest() (o *UpdateDomainAsyncWorkflowConfiguratonRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"UpdateDomainAsyncWorkflowConfiguraton\" for this struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) MethodName() string {\n\treturn \"UpdateDomainAsyncWorkflowConfiguraton\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.UpdateDomainAsyncWorkflowConfiguraton\n// function.\nvar AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper = struct {\n\t// Args accepts the parameters of UpdateDomainAsyncWorkflowConfiguraton in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *UpdateDomainAsyncWorkflowConfiguratonRequest,\n\t) *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by UpdateDomainAsyncWorkflowConfiguraton.\n\t//\n\t// An error can be thrown by UpdateDomainAsyncWorkflowConfiguraton only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for UpdateDomainAsyncWorkflowConfiguraton\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// UpdateDomainAsyncWorkflowConfiguraton into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by UpdateDomainAsyncWorkflowConfiguraton\n\t//\n\t//   value, err := UpdateDomainAsyncWorkflowConfiguraton(args)\n\t//   result, err := AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from UpdateDomainAsyncWorkflowConfiguraton: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*UpdateDomainAsyncWorkflowConfiguratonResponse, error) (*AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result, error)\n\n\t// UnwrapResponse takes the result struct for UpdateDomainAsyncWorkflowConfiguraton\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if UpdateDomainAsyncWorkflowConfiguraton threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) (*UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.Args = func(\n\t\trequest *UpdateDomainAsyncWorkflowConfiguratonRequest,\n\t) *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args {\n\t\treturn &AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.WrapResponse = func(success *UpdateDomainAsyncWorkflowConfiguratonResponse, err error) (*AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse = func(result *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) (success *UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result represents the result of a AdminService.UpdateDomainAsyncWorkflowConfiguraton function call.\n//\n// The result of a UpdateDomainAsyncWorkflowConfiguraton execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct {\n\t// Value returned by UpdateDomainAsyncWorkflowConfiguraton after a successful execution.\n\tSuccess         *UpdateDomainAsyncWorkflowConfiguratonResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainAsyncWorkflowConfiguratonResponse_Read(w wire.Value) (*UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tvar v UpdateDomainAsyncWorkflowConfiguratonResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _UpdateDomainAsyncWorkflowConfiguratonResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct could not be encoded.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainAsyncWorkflowConfiguratonResponse_Decode(sr stream.Reader) (*UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tvar v UpdateDomainAsyncWorkflowConfiguratonResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _UpdateDomainAsyncWorkflowConfiguratonResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result\n// struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result match the\n// provided AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) Equals(rhs *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) GetSuccess() (o *UpdateDomainAsyncWorkflowConfiguratonResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"UpdateDomainAsyncWorkflowConfiguraton\" for this struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) MethodName() string {\n\treturn \"UpdateDomainAsyncWorkflowConfiguraton\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_UpdateDomainIsolationGroups_Args represents the arguments for the AdminService.UpdateDomainIsolationGroups function.\n//\n// The arguments for UpdateDomainIsolationGroups are sent and received over the wire as this struct.\ntype AdminService_UpdateDomainIsolationGroups_Args struct {\n\tRequest *UpdateDomainIsolationGroupsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDomainIsolationGroups_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainIsolationGroupsRequest_Read(w wire.Value) (*UpdateDomainIsolationGroupsRequest, error) {\n\tvar v UpdateDomainIsolationGroupsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateDomainIsolationGroups_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDomainIsolationGroups_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDomainIsolationGroups_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _UpdateDomainIsolationGroupsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDomainIsolationGroups_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDomainIsolationGroups_Args struct could not be encoded.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainIsolationGroupsRequest_Decode(sr stream.Reader) (*UpdateDomainIsolationGroupsRequest, error) {\n\tvar v UpdateDomainIsolationGroupsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateDomainIsolationGroups_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDomainIsolationGroups_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _UpdateDomainIsolationGroupsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDomainIsolationGroups_Args\n// struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDomainIsolationGroups_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDomainIsolationGroups_Args match the\n// provided AdminService_UpdateDomainIsolationGroups_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) Equals(rhs *AdminService_UpdateDomainIsolationGroups_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDomainIsolationGroups_Args.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) GetRequest() (o *UpdateDomainIsolationGroupsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"UpdateDomainIsolationGroups\" for this struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) MethodName() string {\n\treturn \"UpdateDomainIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_UpdateDomainIsolationGroups_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.UpdateDomainIsolationGroups\n// function.\nvar AdminService_UpdateDomainIsolationGroups_Helper = struct {\n\t// Args accepts the parameters of UpdateDomainIsolationGroups in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *UpdateDomainIsolationGroupsRequest,\n\t) *AdminService_UpdateDomainIsolationGroups_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by UpdateDomainIsolationGroups.\n\t//\n\t// An error can be thrown by UpdateDomainIsolationGroups only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for UpdateDomainIsolationGroups\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// UpdateDomainIsolationGroups into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by UpdateDomainIsolationGroups\n\t//\n\t//   value, err := UpdateDomainIsolationGroups(args)\n\t//   result, err := AdminService_UpdateDomainIsolationGroups_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from UpdateDomainIsolationGroups: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*UpdateDomainIsolationGroupsResponse, error) (*AdminService_UpdateDomainIsolationGroups_Result, error)\n\n\t// UnwrapResponse takes the result struct for UpdateDomainIsolationGroups\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if UpdateDomainIsolationGroups threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_UpdateDomainIsolationGroups_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_UpdateDomainIsolationGroups_Result) (*UpdateDomainIsolationGroupsResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_UpdateDomainIsolationGroups_Helper.Args = func(\n\t\trequest *UpdateDomainIsolationGroupsRequest,\n\t) *AdminService_UpdateDomainIsolationGroups_Args {\n\t\treturn &AdminService_UpdateDomainIsolationGroups_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_UpdateDomainIsolationGroups_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_UpdateDomainIsolationGroups_Helper.WrapResponse = func(success *UpdateDomainIsolationGroupsResponse, err error) (*AdminService_UpdateDomainIsolationGroups_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_UpdateDomainIsolationGroups_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_UpdateDomainIsolationGroups_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_UpdateDomainIsolationGroups_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_UpdateDomainIsolationGroups_Helper.UnwrapResponse = func(result *AdminService_UpdateDomainIsolationGroups_Result) (success *UpdateDomainIsolationGroupsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_UpdateDomainIsolationGroups_Result represents the result of a AdminService.UpdateDomainIsolationGroups function call.\n//\n// The result of a UpdateDomainIsolationGroups execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_UpdateDomainIsolationGroups_Result struct {\n\t// Value returned by UpdateDomainIsolationGroups after a successful execution.\n\tSuccess         *UpdateDomainIsolationGroupsResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDomainIsolationGroups_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_UpdateDomainIsolationGroups_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainIsolationGroupsResponse_Read(w wire.Value) (*UpdateDomainIsolationGroupsResponse, error) {\n\tvar v UpdateDomainIsolationGroupsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateDomainIsolationGroups_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDomainIsolationGroups_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDomainIsolationGroups_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _UpdateDomainIsolationGroupsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDomainIsolationGroups_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDomainIsolationGroups_Result struct could not be encoded.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainIsolationGroupsResponse_Decode(sr stream.Reader) (*UpdateDomainIsolationGroupsResponse, error) {\n\tvar v UpdateDomainIsolationGroupsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateDomainIsolationGroups_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDomainIsolationGroups_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _UpdateDomainIsolationGroupsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDomainIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDomainIsolationGroups_Result\n// struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDomainIsolationGroups_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDomainIsolationGroups_Result match the\n// provided AdminService_UpdateDomainIsolationGroups_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) Equals(rhs *AdminService_UpdateDomainIsolationGroups_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDomainIsolationGroups_Result.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) GetSuccess() (o *UpdateDomainIsolationGroupsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"UpdateDomainIsolationGroups\" for this struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) MethodName() string {\n\treturn \"UpdateDomainIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_UpdateDomainIsolationGroups_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_UpdateDynamicConfig_Args represents the arguments for the AdminService.UpdateDynamicConfig function.\n//\n// The arguments for UpdateDynamicConfig are sent and received over the wire as this struct.\ntype AdminService_UpdateDynamicConfig_Args struct {\n\tRequest *UpdateDynamicConfigRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDynamicConfig_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDynamicConfig_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDynamicConfigRequest_Read(w wire.Value) (*UpdateDynamicConfigRequest, error) {\n\tvar v UpdateDynamicConfigRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateDynamicConfig_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDynamicConfig_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDynamicConfig_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDynamicConfig_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _UpdateDynamicConfigRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDynamicConfig_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDynamicConfig_Args struct could not be encoded.\nfunc (v *AdminService_UpdateDynamicConfig_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDynamicConfigRequest_Decode(sr stream.Reader) (*UpdateDynamicConfigRequest, error) {\n\tvar v UpdateDynamicConfigRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateDynamicConfig_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDynamicConfig_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDynamicConfig_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _UpdateDynamicConfigRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDynamicConfig_Args\n// struct.\nfunc (v *AdminService_UpdateDynamicConfig_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDynamicConfig_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDynamicConfig_Args match the\n// provided AdminService_UpdateDynamicConfig_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDynamicConfig_Args) Equals(rhs *AdminService_UpdateDynamicConfig_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDynamicConfig_Args.\nfunc (v *AdminService_UpdateDynamicConfig_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDynamicConfig_Args) GetRequest() (o *UpdateDynamicConfigRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_UpdateDynamicConfig_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"UpdateDynamicConfig\" for this struct.\nfunc (v *AdminService_UpdateDynamicConfig_Args) MethodName() string {\n\treturn \"UpdateDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_UpdateDynamicConfig_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_UpdateDynamicConfig_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.UpdateDynamicConfig\n// function.\nvar AdminService_UpdateDynamicConfig_Helper = struct {\n\t// Args accepts the parameters of UpdateDynamicConfig in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *UpdateDynamicConfigRequest,\n\t) *AdminService_UpdateDynamicConfig_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by UpdateDynamicConfig.\n\t//\n\t// An error can be thrown by UpdateDynamicConfig only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for UpdateDynamicConfig\n\t// given the error returned by it. The provided error may\n\t// be nil if UpdateDynamicConfig did not fail.\n\t//\n\t// This allows mapping errors returned by UpdateDynamicConfig into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// UpdateDynamicConfig\n\t//\n\t//   err := UpdateDynamicConfig(args)\n\t//   result, err := AdminService_UpdateDynamicConfig_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from UpdateDynamicConfig: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*AdminService_UpdateDynamicConfig_Result, error)\n\n\t// UnwrapResponse takes the result struct for UpdateDynamicConfig\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if UpdateDynamicConfig threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := AdminService_UpdateDynamicConfig_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_UpdateDynamicConfig_Result) error\n}{}\n\nfunc init() {\n\tAdminService_UpdateDynamicConfig_Helper.Args = func(\n\t\trequest *UpdateDynamicConfigRequest,\n\t) *AdminService_UpdateDynamicConfig_Args {\n\t\treturn &AdminService_UpdateDynamicConfig_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_UpdateDynamicConfig_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_UpdateDynamicConfig_Helper.WrapResponse = func(err error) (*AdminService_UpdateDynamicConfig_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_UpdateDynamicConfig_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_UpdateDynamicConfig_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_UpdateDynamicConfig_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_UpdateDynamicConfig_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &AdminService_UpdateDynamicConfig_Result{InternalServiceError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_UpdateDynamicConfig_Helper.UnwrapResponse = func(result *AdminService_UpdateDynamicConfig_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// AdminService_UpdateDynamicConfig_Result represents the result of a AdminService.UpdateDynamicConfig function call.\n//\n// The result of a UpdateDynamicConfig execution is sent and received over the wire as this struct.\ntype AdminService_UpdateDynamicConfig_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateDynamicConfig_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateDynamicConfig_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_UpdateDynamicConfig_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AdminService_UpdateDynamicConfig_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateDynamicConfig_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateDynamicConfig_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateDynamicConfig_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateDynamicConfig_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateDynamicConfig_Result struct could not be encoded.\nfunc (v *AdminService_UpdateDynamicConfig_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AdminService_UpdateDynamicConfig_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateDynamicConfig_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateDynamicConfig_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateDynamicConfig_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateDynamicConfig_Result\n// struct.\nfunc (v *AdminService_UpdateDynamicConfig_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateDynamicConfig_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateDynamicConfig_Result match the\n// provided AdminService_UpdateDynamicConfig_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateDynamicConfig_Result) Equals(rhs *AdminService_UpdateDynamicConfig_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateDynamicConfig_Result.\nfunc (v *AdminService_UpdateDynamicConfig_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDynamicConfig_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_UpdateDynamicConfig_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateDynamicConfig_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *AdminService_UpdateDynamicConfig_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"UpdateDynamicConfig\" for this struct.\nfunc (v *AdminService_UpdateDynamicConfig_Result) MethodName() string {\n\treturn \"UpdateDynamicConfig\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_UpdateDynamicConfig_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// AdminService_UpdateGlobalIsolationGroups_Args represents the arguments for the AdminService.UpdateGlobalIsolationGroups function.\n//\n// The arguments for UpdateGlobalIsolationGroups are sent and received over the wire as this struct.\ntype AdminService_UpdateGlobalIsolationGroups_Args struct {\n\tRequest *UpdateGlobalIsolationGroupsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateGlobalIsolationGroups_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateGlobalIsolationGroupsRequest_Read(w wire.Value) (*UpdateGlobalIsolationGroupsRequest, error) {\n\tvar v UpdateGlobalIsolationGroupsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateGlobalIsolationGroups_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateGlobalIsolationGroups_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateGlobalIsolationGroups_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _UpdateGlobalIsolationGroupsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateGlobalIsolationGroups_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateGlobalIsolationGroups_Args struct could not be encoded.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateGlobalIsolationGroupsRequest_Decode(sr stream.Reader) (*UpdateGlobalIsolationGroupsRequest, error) {\n\tvar v UpdateGlobalIsolationGroupsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateGlobalIsolationGroups_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateGlobalIsolationGroups_Args struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _UpdateGlobalIsolationGroupsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateGlobalIsolationGroups_Args\n// struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateGlobalIsolationGroups_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateGlobalIsolationGroups_Args match the\n// provided AdminService_UpdateGlobalIsolationGroups_Args.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) Equals(rhs *AdminService_UpdateGlobalIsolationGroups_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateGlobalIsolationGroups_Args.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) GetRequest() (o *UpdateGlobalIsolationGroupsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"UpdateGlobalIsolationGroups\" for this struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) MethodName() string {\n\treturn \"UpdateGlobalIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// AdminService_UpdateGlobalIsolationGroups_Helper provides functions that aid in handling the\n// parameters and return values of the AdminService.UpdateGlobalIsolationGroups\n// function.\nvar AdminService_UpdateGlobalIsolationGroups_Helper = struct {\n\t// Args accepts the parameters of UpdateGlobalIsolationGroups in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *UpdateGlobalIsolationGroupsRequest,\n\t) *AdminService_UpdateGlobalIsolationGroups_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by UpdateGlobalIsolationGroups.\n\t//\n\t// An error can be thrown by UpdateGlobalIsolationGroups only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for UpdateGlobalIsolationGroups\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// UpdateGlobalIsolationGroups into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by UpdateGlobalIsolationGroups\n\t//\n\t//   value, err := UpdateGlobalIsolationGroups(args)\n\t//   result, err := AdminService_UpdateGlobalIsolationGroups_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from UpdateGlobalIsolationGroups: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*UpdateGlobalIsolationGroupsResponse, error) (*AdminService_UpdateGlobalIsolationGroups_Result, error)\n\n\t// UnwrapResponse takes the result struct for UpdateGlobalIsolationGroups\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if UpdateGlobalIsolationGroups threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := AdminService_UpdateGlobalIsolationGroups_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*AdminService_UpdateGlobalIsolationGroups_Result) (*UpdateGlobalIsolationGroupsResponse, error)\n}{}\n\nfunc init() {\n\tAdminService_UpdateGlobalIsolationGroups_Helper.Args = func(\n\t\trequest *UpdateGlobalIsolationGroupsRequest,\n\t) *AdminService_UpdateGlobalIsolationGroups_Args {\n\t\treturn &AdminService_UpdateGlobalIsolationGroups_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tAdminService_UpdateGlobalIsolationGroups_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tAdminService_UpdateGlobalIsolationGroups_Helper.WrapResponse = func(success *UpdateGlobalIsolationGroupsResponse, err error) (*AdminService_UpdateGlobalIsolationGroups_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &AdminService_UpdateGlobalIsolationGroups_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for AdminService_UpdateGlobalIsolationGroups_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &AdminService_UpdateGlobalIsolationGroups_Result{BadRequestError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tAdminService_UpdateGlobalIsolationGroups_Helper.UnwrapResponse = func(result *AdminService_UpdateGlobalIsolationGroups_Result) (success *UpdateGlobalIsolationGroupsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// AdminService_UpdateGlobalIsolationGroups_Result represents the result of a AdminService.UpdateGlobalIsolationGroups function call.\n//\n// The result of a UpdateGlobalIsolationGroups execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype AdminService_UpdateGlobalIsolationGroups_Result struct {\n\t// Value returned by UpdateGlobalIsolationGroups after a successful execution.\n\tSuccess         *UpdateGlobalIsolationGroupsResponse `json:\"success,omitempty\"`\n\tBadRequestError *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n}\n\n// ToWire translates a AdminService_UpdateGlobalIsolationGroups_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"AdminService_UpdateGlobalIsolationGroups_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateGlobalIsolationGroupsResponse_Read(w wire.Value) (*UpdateGlobalIsolationGroupsResponse, error) {\n\tvar v UpdateGlobalIsolationGroupsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AdminService_UpdateGlobalIsolationGroups_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AdminService_UpdateGlobalIsolationGroups_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AdminService_UpdateGlobalIsolationGroups_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _UpdateGlobalIsolationGroupsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AdminService_UpdateGlobalIsolationGroups_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AdminService_UpdateGlobalIsolationGroups_Result struct could not be encoded.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateGlobalIsolationGroupsResponse_Decode(sr stream.Reader) (*UpdateGlobalIsolationGroupsResponse, error) {\n\tvar v UpdateGlobalIsolationGroupsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AdminService_UpdateGlobalIsolationGroups_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AdminService_UpdateGlobalIsolationGroups_Result struct could not be generated from the wire\n// representation.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _UpdateGlobalIsolationGroupsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"AdminService_UpdateGlobalIsolationGroups_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AdminService_UpdateGlobalIsolationGroups_Result\n// struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AdminService_UpdateGlobalIsolationGroups_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AdminService_UpdateGlobalIsolationGroups_Result match the\n// provided AdminService_UpdateGlobalIsolationGroups_Result.\n//\n// This function performs a deep comparison.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) Equals(rhs *AdminService_UpdateGlobalIsolationGroups_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AdminService_UpdateGlobalIsolationGroups_Result.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) GetSuccess() (o *UpdateGlobalIsolationGroupsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"UpdateGlobalIsolationGroups\" for this struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) MethodName() string {\n\treturn \"UpdateGlobalIsolationGroups\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *AdminService_UpdateGlobalIsolationGroups_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n"
  },
  {
    "path": ".gen/go/admin/adminserviceclient/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage adminserviceclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\twire \"go.uber.org/thriftrw/wire\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tadmin \"github.com/uber/cadence/.gen/go/admin\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is a client for the AdminService service.\ntype Interface interface {\n\tAddSearchAttribute(\n\t\tctx context.Context,\n\t\tRequest *admin.AddSearchAttributeRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tCloseShard(\n\t\tctx context.Context,\n\t\tRequest *shared.CloseShardRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tDeleteWorkflow(\n\t\tctx context.Context,\n\t\tRequest *admin.AdminDeleteWorkflowRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.AdminDeleteWorkflowResponse, error)\n\n\tDescribeCluster(\n\t\tctx context.Context,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.DescribeClusterResponse, error)\n\n\tDescribeHistoryHost(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeHistoryHostRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeHistoryHostResponse, error)\n\n\tDescribeQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeQueueRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeQueueResponse, error)\n\n\tDescribeShardDistribution(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeShardDistributionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeShardDistributionResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tRequest *admin.DescribeWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.DescribeWorkflowExecutionResponse, error)\n\n\tGetCrossClusterTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.GetCrossClusterTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetCrossClusterTasksResponse, error)\n\n\tGetDLQReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDLQReplicationMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.GetDLQReplicationMessagesResponse, error)\n\n\tGetDomainAsyncWorkflowConfiguraton(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDomainAsyncWorkflowConfiguratonRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.GetDomainAsyncWorkflowConfiguratonResponse, error)\n\n\tGetDomainIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDomainIsolationGroupsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.GetDomainIsolationGroupsResponse, error)\n\n\tGetDomainReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDomainReplicationMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.GetDomainReplicationMessagesResponse, error)\n\n\tGetDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDynamicConfigRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.GetDynamicConfigResponse, error)\n\n\tGetGlobalIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.GetGlobalIsolationGroupsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.GetGlobalIsolationGroupsResponse, error)\n\n\tGetReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetReplicationMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.GetReplicationMessagesResponse, error)\n\n\tGetWorkflowExecutionRawHistoryV2(\n\t\tctx context.Context,\n\t\tGetRequest *admin.GetWorkflowExecutionRawHistoryV2Request,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.GetWorkflowExecutionRawHistoryV2Response, error)\n\n\tListDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.ListDynamicConfigRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.ListDynamicConfigResponse, error)\n\n\tMaintainCorruptWorkflow(\n\t\tctx context.Context,\n\t\tRequest *admin.AdminMaintainWorkflowRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.AdminMaintainWorkflowResponse, error)\n\n\tMergeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.MergeDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.MergeDLQMessagesResponse, error)\n\n\tPurgeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.PurgeDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tReadDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.ReadDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.ReadDLQMessagesResponse, error)\n\n\tReapplyEvents(\n\t\tctx context.Context,\n\t\tReapplyEventsRequest *shared.ReapplyEventsRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.RefreshWorkflowTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRemoveTask(\n\t\tctx context.Context,\n\t\tRequest *shared.RemoveTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tResendReplicationTasks(\n\t\tctx context.Context,\n\t\tRequest *admin.ResendReplicationTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tResetQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.ResetQueueRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondCrossClusterTasksCompleted(\n\t\tctx context.Context,\n\t\tRequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n\n\tRestoreDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.RestoreDynamicConfigRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tUpdateDomainAsyncWorkflowConfiguraton(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDomainAsyncWorkflowConfiguratonRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n\n\tUpdateDomainIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDomainIsolationGroupsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.UpdateDomainIsolationGroupsResponse, error)\n\n\tUpdateDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDynamicConfigRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tUpdateGlobalIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateGlobalIsolationGroupsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*admin.UpdateGlobalIsolationGroupsResponse, error)\n}\n\n// New builds a new client for the AdminService service.\n//\n//\tclient := adminserviceclient.New(dispatcher.ClientConfig(\"adminservice\"))\nfunc New(c transport.ClientConfig, opts ...thrift.ClientOption) Interface {\n\treturn client{\n\t\tc: thrift.New(thrift.Config{\n\t\t\tService:      \"AdminService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t\tnwc: thrift.NewNoWire(thrift.Config{\n\t\t\tService:      \"AdminService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t}\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(c transport.ClientConfig, f reflect.StructField) Interface {\n\t\t\treturn New(c, thrift.ClientBuilderOptions(c, f)...)\n\t\t},\n\t)\n}\n\ntype client struct {\n\tc   thrift.Client\n\tnwc thrift.NoWireClient\n}\n\nfunc (c client) AddSearchAttribute(\n\tctx context.Context,\n\t_Request *admin.AddSearchAttributeRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_AddSearchAttribute_Result\n\targs := admin.AdminService_AddSearchAttribute_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_AddSearchAttribute_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) CloseShard(\n\tctx context.Context,\n\t_Request *shared.CloseShardRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_CloseShard_Result\n\targs := admin.AdminService_CloseShard_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_CloseShard_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DeleteWorkflow(\n\tctx context.Context,\n\t_Request *admin.AdminDeleteWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.AdminDeleteWorkflowResponse, err error) {\n\n\tvar result admin.AdminService_DeleteWorkflow_Result\n\targs := admin.AdminService_DeleteWorkflow_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DeleteWorkflow_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeCluster(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *admin.DescribeClusterResponse, err error) {\n\n\tvar result admin.AdminService_DescribeCluster_Result\n\targs := admin.AdminService_DescribeCluster_Helper.Args()\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DescribeCluster_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeHistoryHost(\n\tctx context.Context,\n\t_Request *shared.DescribeHistoryHostRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeHistoryHostResponse, err error) {\n\n\tvar result admin.AdminService_DescribeHistoryHost_Result\n\targs := admin.AdminService_DescribeHistoryHost_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DescribeHistoryHost_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeQueue(\n\tctx context.Context,\n\t_Request *shared.DescribeQueueRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeQueueResponse, err error) {\n\n\tvar result admin.AdminService_DescribeQueue_Result\n\targs := admin.AdminService_DescribeQueue_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DescribeQueue_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeShardDistribution(\n\tctx context.Context,\n\t_Request *shared.DescribeShardDistributionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeShardDistributionResponse, err error) {\n\n\tvar result admin.AdminService_DescribeShardDistribution_Result\n\targs := admin.AdminService_DescribeShardDistribution_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DescribeShardDistribution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_Request *admin.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.DescribeWorkflowExecutionResponse, err error) {\n\n\tvar result admin.AdminService_DescribeWorkflowExecution_Result\n\targs := admin.AdminService_DescribeWorkflowExecution_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_DescribeWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetCrossClusterTasks(\n\tctx context.Context,\n\t_Request *shared.GetCrossClusterTasksRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetCrossClusterTasksResponse, err error) {\n\n\tvar result admin.AdminService_GetCrossClusterTasks_Result\n\targs := admin.AdminService_GetCrossClusterTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetCrossClusterTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDLQReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDLQReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\n\tvar result admin.AdminService_GetDLQReplicationMessages_Result\n\targs := admin.AdminService_GetDLQReplicationMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetDLQReplicationMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDomainAsyncWorkflowConfiguraton(\n\tctx context.Context,\n\t_Request *admin.GetDomainAsyncWorkflowConfiguratonRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\n\tvar result admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Result\n\targs := admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDomainIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.GetDomainIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDomainIsolationGroupsResponse, err error) {\n\n\tvar result admin.AdminService_GetDomainIsolationGroups_Result\n\targs := admin.AdminService_GetDomainIsolationGroups_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetDomainIsolationGroups_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDomainReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDomainReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDomainReplicationMessagesResponse, err error) {\n\n\tvar result admin.AdminService_GetDomainReplicationMessages_Result\n\targs := admin.AdminService_GetDomainReplicationMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetDomainReplicationMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.GetDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDynamicConfigResponse, err error) {\n\n\tvar result admin.AdminService_GetDynamicConfig_Result\n\targs := admin.AdminService_GetDynamicConfig_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetDynamicConfig_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetGlobalIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.GetGlobalIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetGlobalIsolationGroupsResponse, err error) {\n\n\tvar result admin.AdminService_GetGlobalIsolationGroups_Result\n\targs := admin.AdminService_GetGlobalIsolationGroups_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetGlobalIsolationGroups_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetReplicationMessagesResponse, err error) {\n\n\tvar result admin.AdminService_GetReplicationMessages_Result\n\targs := admin.AdminService_GetReplicationMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetReplicationMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetWorkflowExecutionRawHistoryV2(\n\tctx context.Context,\n\t_GetRequest *admin.GetWorkflowExecutionRawHistoryV2Request,\n\topts ...yarpc.CallOption,\n) (success *admin.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\n\tvar result admin.AdminService_GetWorkflowExecutionRawHistoryV2_Result\n\targs := admin.AdminService_GetWorkflowExecutionRawHistoryV2_Helper.Args(_GetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_GetWorkflowExecutionRawHistoryV2_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.ListDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.ListDynamicConfigResponse, err error) {\n\n\tvar result admin.AdminService_ListDynamicConfig_Result\n\targs := admin.AdminService_ListDynamicConfig_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_ListDynamicConfig_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) MaintainCorruptWorkflow(\n\tctx context.Context,\n\t_Request *admin.AdminMaintainWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.AdminMaintainWorkflowResponse, err error) {\n\n\tvar result admin.AdminService_MaintainCorruptWorkflow_Result\n\targs := admin.AdminService_MaintainCorruptWorkflow_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_MaintainCorruptWorkflow_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) MergeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.MergeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.MergeDLQMessagesResponse, err error) {\n\n\tvar result admin.AdminService_MergeDLQMessages_Result\n\targs := admin.AdminService_MergeDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_MergeDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PurgeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.PurgeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_PurgeDLQMessages_Result\n\targs := admin.AdminService_PurgeDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_PurgeDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ReadDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.ReadDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.ReadDLQMessagesResponse, err error) {\n\n\tvar result admin.AdminService_ReadDLQMessages_Result\n\targs := admin.AdminService_ReadDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_ReadDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ReapplyEvents(\n\tctx context.Context,\n\t_ReapplyEventsRequest *shared.ReapplyEventsRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_ReapplyEvents_Result\n\targs := admin.AdminService_ReapplyEvents_Helper.Args(_ReapplyEventsRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_ReapplyEvents_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *shared.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_RefreshWorkflowTasks_Result\n\targs := admin.AdminService_RefreshWorkflowTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_RefreshWorkflowTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RemoveTask(\n\tctx context.Context,\n\t_Request *shared.RemoveTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_RemoveTask_Result\n\targs := admin.AdminService_RemoveTask_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_RemoveTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResendReplicationTasks(\n\tctx context.Context,\n\t_Request *admin.ResendReplicationTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_ResendReplicationTasks_Result\n\targs := admin.AdminService_ResendReplicationTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_ResendReplicationTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetQueue(\n\tctx context.Context,\n\t_Request *shared.ResetQueueRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_ResetQueue_Result\n\targs := admin.AdminService_ResetQueue_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_ResetQueue_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\t_Request *shared.RespondCrossClusterTasksCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\n\tvar result admin.AdminService_RespondCrossClusterTasksCompleted_Result\n\targs := admin.AdminService_RespondCrossClusterTasksCompleted_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RestoreDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.RestoreDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_RestoreDynamicConfig_Result\n\targs := admin.AdminService_RestoreDynamicConfig_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_RestoreDynamicConfig_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) UpdateDomainAsyncWorkflowConfiguraton(\n\tctx context.Context,\n\t_Request *admin.UpdateDomainAsyncWorkflowConfiguratonRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\n\tvar result admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Result\n\targs := admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) UpdateDomainIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.UpdateDomainIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateDomainIsolationGroupsResponse, err error) {\n\n\tvar result admin.AdminService_UpdateDomainIsolationGroups_Result\n\targs := admin.AdminService_UpdateDomainIsolationGroups_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_UpdateDomainIsolationGroups_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) UpdateDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.UpdateDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result admin.AdminService_UpdateDynamicConfig_Result\n\targs := admin.AdminService_UpdateDynamicConfig_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = admin.AdminService_UpdateDynamicConfig_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) UpdateGlobalIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.UpdateGlobalIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateGlobalIsolationGroupsResponse, err error) {\n\n\tvar result admin.AdminService_UpdateGlobalIsolationGroups_Result\n\targs := admin.AdminService_UpdateGlobalIsolationGroups_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = admin.AdminService_UpdateGlobalIsolationGroups_Helper.UnwrapResponse(&result)\n\treturn\n}\n"
  },
  {
    "path": ".gen/go/admin/adminservicefx/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage adminservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\trestriction \"go.uber.org/yarpc/api/x/restriction\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tadminserviceclient \"github.com/uber/cadence/.gen/go/admin/adminserviceclient\"\n)\n\n// Params defines the dependencies for the AdminService client.\ntype Params struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// Result defines the output of the AdminService client module. It provides a\n// AdminService client to an Fx application.\ntype Result struct {\n\tfx.Out\n\n\tClient adminserviceclient.Interface\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// Client provides a AdminService client to an Fx application using the given name\n// for routing.\n//\n//\tfx.Provide(\n//\t\tadminservicefx.Client(\"...\"),\n//\t\tnewHandler,\n//\t)\nfunc Client(name string, opts ...thrift.ClientOption) interface{} {\n\treturn func(p Params) Result {\n\t\tcc := p.Provider.ClientConfig(name)\n\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok && p.Restriction != nil {\n\t\t\tif err := p.Restriction.Check(thrift.Encoding, namer.TransportName()); err != nil {\n\t\t\t\tpanic(err.Error())\n\t\t\t}\n\t\t}\n\t\tclient := adminserviceclient.New(cc, opts...)\n\t\treturn Result{Client: client}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/admin/adminservicefx/doc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\n// Package adminservicefx provides better integration for Fx for services\n// implementing or calling AdminService.\n//\n// # Clients\n//\n// If you are making requests to AdminService, use the Client function to inject a\n// AdminService client into your container.\n//\n//\tfx.Provide(adminservicefx.Client(\"...\"))\n//\n// # Servers\n//\n// If you are implementing AdminService, provide a adminserviceserver.Interface into\n// the container and use the Server function.\n//\n// Given,\n//\n//\tfunc NewAdminServiceHandler() adminserviceserver.Interface\n//\n// You can do the following to have the procedures of AdminService made available\n// to an Fx application.\n//\n//\tfx.Provide(\n//\t\tNewAdminServiceHandler,\n//\t\tadminservicefx.Server(),\n//\t)\npackage adminservicefx\n"
  },
  {
    "path": ".gen/go/admin/adminservicefx/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage adminservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tadminserviceserver \"github.com/uber/cadence/.gen/go/admin/adminserviceserver\"\n)\n\n// ServerParams defines the dependencies for the AdminService server.\ntype ServerParams struct {\n\tfx.In\n\n\tHandler adminserviceserver.Interface\n}\n\n// ServerResult defines the output of AdminService server module. It provides the\n// procedures of a AdminService handler to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group. Dig 1.2 or newer\n// must be used for this feature to work.\ntype ServerResult struct {\n\tfx.Out\n\n\tProcedures []transport.Procedure `group:\"yarpcfx\"`\n}\n\n// Server provides procedures for AdminService to an Fx application. It expects a\n// adminservicefx.Interface to be present in the container.\n//\n//\tfx.Provide(\n//\t\tfunc(h *MyAdminServiceHandler) adminserviceserver.Interface {\n//\t\t\treturn h\n//\t\t},\n//\t\tadminservicefx.Server(),\n//\t)\nfunc Server(opts ...thrift.RegisterOption) interface{} {\n\treturn func(p ServerParams) ServerResult {\n\t\tprocedures := adminserviceserver.New(p.Handler, opts...)\n\t\treturn ServerResult{Procedures: procedures}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/admin/adminserviceserver/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage adminserviceserver\n\nimport (\n\tcontext \"context\"\n\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\tyarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n\tadmin \"github.com/uber/cadence/.gen/go/admin\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is the server-side interface for the AdminService service.\ntype Interface interface {\n\tAddSearchAttribute(\n\t\tctx context.Context,\n\t\tRequest *admin.AddSearchAttributeRequest,\n\t) error\n\n\tCloseShard(\n\t\tctx context.Context,\n\t\tRequest *shared.CloseShardRequest,\n\t) error\n\n\tDeleteWorkflow(\n\t\tctx context.Context,\n\t\tRequest *admin.AdminDeleteWorkflowRequest,\n\t) (*admin.AdminDeleteWorkflowResponse, error)\n\n\tDescribeCluster(\n\t\tctx context.Context,\n\t) (*admin.DescribeClusterResponse, error)\n\n\tDescribeHistoryHost(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeHistoryHostRequest,\n\t) (*shared.DescribeHistoryHostResponse, error)\n\n\tDescribeQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeQueueRequest,\n\t) (*shared.DescribeQueueResponse, error)\n\n\tDescribeShardDistribution(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeShardDistributionRequest,\n\t) (*shared.DescribeShardDistributionResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tRequest *admin.DescribeWorkflowExecutionRequest,\n\t) (*admin.DescribeWorkflowExecutionResponse, error)\n\n\tGetCrossClusterTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.GetCrossClusterTasksRequest,\n\t) (*shared.GetCrossClusterTasksResponse, error)\n\n\tGetDLQReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDLQReplicationMessagesRequest,\n\t) (*replicator.GetDLQReplicationMessagesResponse, error)\n\n\tGetDomainAsyncWorkflowConfiguraton(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDomainAsyncWorkflowConfiguratonRequest,\n\t) (*admin.GetDomainAsyncWorkflowConfiguratonResponse, error)\n\n\tGetDomainIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDomainIsolationGroupsRequest,\n\t) (*admin.GetDomainIsolationGroupsResponse, error)\n\n\tGetDomainReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDomainReplicationMessagesRequest,\n\t) (*replicator.GetDomainReplicationMessagesResponse, error)\n\n\tGetDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.GetDynamicConfigRequest,\n\t) (*admin.GetDynamicConfigResponse, error)\n\n\tGetGlobalIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.GetGlobalIsolationGroupsRequest,\n\t) (*admin.GetGlobalIsolationGroupsResponse, error)\n\n\tGetReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetReplicationMessagesRequest,\n\t) (*replicator.GetReplicationMessagesResponse, error)\n\n\tGetWorkflowExecutionRawHistoryV2(\n\t\tctx context.Context,\n\t\tGetRequest *admin.GetWorkflowExecutionRawHistoryV2Request,\n\t) (*admin.GetWorkflowExecutionRawHistoryV2Response, error)\n\n\tListDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.ListDynamicConfigRequest,\n\t) (*admin.ListDynamicConfigResponse, error)\n\n\tMaintainCorruptWorkflow(\n\t\tctx context.Context,\n\t\tRequest *admin.AdminMaintainWorkflowRequest,\n\t) (*admin.AdminMaintainWorkflowResponse, error)\n\n\tMergeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.MergeDLQMessagesRequest,\n\t) (*replicator.MergeDLQMessagesResponse, error)\n\n\tPurgeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.PurgeDLQMessagesRequest,\n\t) error\n\n\tReadDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.ReadDLQMessagesRequest,\n\t) (*replicator.ReadDLQMessagesResponse, error)\n\n\tReapplyEvents(\n\t\tctx context.Context,\n\t\tReapplyEventsRequest *shared.ReapplyEventsRequest,\n\t) error\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.RefreshWorkflowTasksRequest,\n\t) error\n\n\tRemoveTask(\n\t\tctx context.Context,\n\t\tRequest *shared.RemoveTaskRequest,\n\t) error\n\n\tResendReplicationTasks(\n\t\tctx context.Context,\n\t\tRequest *admin.ResendReplicationTasksRequest,\n\t) error\n\n\tResetQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.ResetQueueRequest,\n\t) error\n\n\tRespondCrossClusterTasksCompleted(\n\t\tctx context.Context,\n\t\tRequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n\n\tRestoreDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.RestoreDynamicConfigRequest,\n\t) error\n\n\tUpdateDomainAsyncWorkflowConfiguraton(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDomainAsyncWorkflowConfiguratonRequest,\n\t) (*admin.UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n\n\tUpdateDomainIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDomainIsolationGroupsRequest,\n\t) (*admin.UpdateDomainIsolationGroupsResponse, error)\n\n\tUpdateDynamicConfig(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateDynamicConfigRequest,\n\t) error\n\n\tUpdateGlobalIsolationGroups(\n\t\tctx context.Context,\n\t\tRequest *admin.UpdateGlobalIsolationGroupsRequest,\n\t) (*admin.UpdateGlobalIsolationGroupsResponse, error)\n}\n\n// New prepares an implementation of the AdminService service for\n// registration.\n//\n//\thandler := AdminServiceHandler{}\n//\tdispatcher.Register(adminserviceserver.New(handler))\nfunc New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {\n\th := handler{impl}\n\tservice := thrift.Service{\n\t\tName: \"AdminService\",\n\t\tMethods: []thrift.Method{\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"AddSearchAttribute\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.AddSearchAttribute),\n\t\t\t\t\tNoWire: addsearchattribute_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"AddSearchAttribute(Request *admin.AddSearchAttributeRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"CloseShard\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.CloseShard),\n\t\t\t\t\tNoWire: closeshard_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"CloseShard(Request *shared.CloseShardRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DeleteWorkflow\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DeleteWorkflow),\n\t\t\t\t\tNoWire: deleteworkflow_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DeleteWorkflow(Request *admin.AdminDeleteWorkflowRequest) (*admin.AdminDeleteWorkflowResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeCluster\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeCluster),\n\t\t\t\t\tNoWire: describecluster_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeCluster() (*admin.DescribeClusterResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeHistoryHost\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeHistoryHost),\n\t\t\t\t\tNoWire: describehistoryhost_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeHistoryHost(Request *shared.DescribeHistoryHostRequest) (*shared.DescribeHistoryHostResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeQueue\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeQueue),\n\t\t\t\t\tNoWire: describequeue_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeQueue(Request *shared.DescribeQueueRequest) (*shared.DescribeQueueResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeShardDistribution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeShardDistribution),\n\t\t\t\t\tNoWire: describesharddistribution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeShardDistribution(Request *shared.DescribeShardDistributionRequest) (*shared.DescribeShardDistributionResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeWorkflowExecution),\n\t\t\t\t\tNoWire: describeworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeWorkflowExecution(Request *admin.DescribeWorkflowExecutionRequest) (*admin.DescribeWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetCrossClusterTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetCrossClusterTasks),\n\t\t\t\t\tNoWire: getcrossclustertasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetCrossClusterTasks(Request *shared.GetCrossClusterTasksRequest) (*shared.GetCrossClusterTasksResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDLQReplicationMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDLQReplicationMessages),\n\t\t\t\t\tNoWire: getdlqreplicationmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDLQReplicationMessages(Request *replicator.GetDLQReplicationMessagesRequest) (*replicator.GetDLQReplicationMessagesResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDomainAsyncWorkflowConfiguraton\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDomainAsyncWorkflowConfiguraton),\n\t\t\t\t\tNoWire: getdomainasyncworkflowconfiguraton_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDomainAsyncWorkflowConfiguraton(Request *admin.GetDomainAsyncWorkflowConfiguratonRequest) (*admin.GetDomainAsyncWorkflowConfiguratonResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDomainIsolationGroups\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDomainIsolationGroups),\n\t\t\t\t\tNoWire: getdomainisolationgroups_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDomainIsolationGroups(Request *admin.GetDomainIsolationGroupsRequest) (*admin.GetDomainIsolationGroupsResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDomainReplicationMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDomainReplicationMessages),\n\t\t\t\t\tNoWire: getdomainreplicationmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDomainReplicationMessages(Request *replicator.GetDomainReplicationMessagesRequest) (*replicator.GetDomainReplicationMessagesResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDynamicConfig\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDynamicConfig),\n\t\t\t\t\tNoWire: getdynamicconfig_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDynamicConfig(Request *admin.GetDynamicConfigRequest) (*admin.GetDynamicConfigResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetGlobalIsolationGroups\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetGlobalIsolationGroups),\n\t\t\t\t\tNoWire: getglobalisolationgroups_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetGlobalIsolationGroups(Request *admin.GetGlobalIsolationGroupsRequest) (*admin.GetGlobalIsolationGroupsResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetReplicationMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetReplicationMessages),\n\t\t\t\t\tNoWire: getreplicationmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetReplicationMessages(Request *replicator.GetReplicationMessagesRequest) (*replicator.GetReplicationMessagesResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetWorkflowExecutionRawHistoryV2\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetWorkflowExecutionRawHistoryV2),\n\t\t\t\t\tNoWire: getworkflowexecutionrawhistoryv2_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetWorkflowExecutionRawHistoryV2(GetRequest *admin.GetWorkflowExecutionRawHistoryV2Request) (*admin.GetWorkflowExecutionRawHistoryV2Response)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListDynamicConfig\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListDynamicConfig),\n\t\t\t\t\tNoWire: listdynamicconfig_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListDynamicConfig(Request *admin.ListDynamicConfigRequest) (*admin.ListDynamicConfigResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"MaintainCorruptWorkflow\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.MaintainCorruptWorkflow),\n\t\t\t\t\tNoWire: maintaincorruptworkflow_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"MaintainCorruptWorkflow(Request *admin.AdminMaintainWorkflowRequest) (*admin.AdminMaintainWorkflowResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"MergeDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.MergeDLQMessages),\n\t\t\t\t\tNoWire: mergedlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"MergeDLQMessages(Request *replicator.MergeDLQMessagesRequest) (*replicator.MergeDLQMessagesResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PurgeDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PurgeDLQMessages),\n\t\t\t\t\tNoWire: purgedlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PurgeDLQMessages(Request *replicator.PurgeDLQMessagesRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ReadDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ReadDLQMessages),\n\t\t\t\t\tNoWire: readdlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ReadDLQMessages(Request *replicator.ReadDLQMessagesRequest) (*replicator.ReadDLQMessagesResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ReapplyEvents\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ReapplyEvents),\n\t\t\t\t\tNoWire: reapplyevents_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ReapplyEvents(ReapplyEventsRequest *shared.ReapplyEventsRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RefreshWorkflowTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RefreshWorkflowTasks),\n\t\t\t\t\tNoWire: refreshworkflowtasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RefreshWorkflowTasks(Request *shared.RefreshWorkflowTasksRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RemoveTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RemoveTask),\n\t\t\t\t\tNoWire: removetask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RemoveTask(Request *shared.RemoveTaskRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResendReplicationTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResendReplicationTasks),\n\t\t\t\t\tNoWire: resendreplicationtasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResendReplicationTasks(Request *admin.ResendReplicationTasksRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetQueue\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetQueue),\n\t\t\t\t\tNoWire: resetqueue_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetQueue(Request *shared.ResetQueueRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondCrossClusterTasksCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondCrossClusterTasksCompleted),\n\t\t\t\t\tNoWire: respondcrossclustertaskscompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondCrossClusterTasksCompleted(Request *shared.RespondCrossClusterTasksCompletedRequest) (*shared.RespondCrossClusterTasksCompletedResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RestoreDynamicConfig\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RestoreDynamicConfig),\n\t\t\t\t\tNoWire: restoredynamicconfig_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RestoreDynamicConfig(Request *admin.RestoreDynamicConfigRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"UpdateDomainAsyncWorkflowConfiguraton\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.UpdateDomainAsyncWorkflowConfiguraton),\n\t\t\t\t\tNoWire: updatedomainasyncworkflowconfiguraton_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"UpdateDomainAsyncWorkflowConfiguraton(Request *admin.UpdateDomainAsyncWorkflowConfiguratonRequest) (*admin.UpdateDomainAsyncWorkflowConfiguratonResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"UpdateDomainIsolationGroups\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.UpdateDomainIsolationGroups),\n\t\t\t\t\tNoWire: updatedomainisolationgroups_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"UpdateDomainIsolationGroups(Request *admin.UpdateDomainIsolationGroupsRequest) (*admin.UpdateDomainIsolationGroupsResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"UpdateDynamicConfig\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.UpdateDynamicConfig),\n\t\t\t\t\tNoWire: updatedynamicconfig_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"UpdateDynamicConfig(Request *admin.UpdateDynamicConfigRequest)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"UpdateGlobalIsolationGroups\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.UpdateGlobalIsolationGroups),\n\t\t\t\t\tNoWire: updateglobalisolationgroups_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"UpdateGlobalIsolationGroups(Request *admin.UpdateGlobalIsolationGroupsRequest) (*admin.UpdateGlobalIsolationGroupsResponse)\",\n\t\t\t\tThriftModule: admin.ThriftModule,\n\t\t\t},\n\t\t},\n\t}\n\n\tprocedures := make([]transport.Procedure, 0, 33)\n\tprocedures = append(procedures, thrift.BuildProcedures(service, opts...)...)\n\treturn procedures\n}\n\ntype handler struct{ impl Interface }\n\ntype yarpcErrorNamer interface{ YARPCErrorName() string }\n\ntype yarpcErrorCoder interface{ YARPCErrorCode() *yarpcerrors.Code }\n\nfunc (h handler) AddSearchAttribute(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_AddSearchAttribute_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'AddSearchAttribute': %w\", err)\n\t}\n\n\tappErr := h.impl.AddSearchAttribute(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_AddSearchAttribute_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) CloseShard(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_CloseShard_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'CloseShard': %w\", err)\n\t}\n\n\tappErr := h.impl.CloseShard(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_CloseShard_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DeleteWorkflow(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DeleteWorkflow_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DeleteWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DeleteWorkflow(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DeleteWorkflow_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeCluster(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DescribeCluster_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DescribeCluster': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeCluster(ctx)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeCluster_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeHistoryHost(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DescribeHistoryHost_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DescribeHistoryHost': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeHistoryHost(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeHistoryHost_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeQueue(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DescribeQueue_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DescribeQueue': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeQueue_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeShardDistribution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DescribeShardDistribution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DescribeShardDistribution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeShardDistribution(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeShardDistribution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_DescribeWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetCrossClusterTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetCrossClusterTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetCrossClusterTasks': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetCrossClusterTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetCrossClusterTasks_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDLQReplicationMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetDLQReplicationMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetDLQReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDLQReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDLQReplicationMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetDomainAsyncWorkflowConfiguraton': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainAsyncWorkflowConfiguraton(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDomainIsolationGroups(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetDomainIsolationGroups_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetDomainIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainIsolationGroups_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDomainReplicationMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetDomainReplicationMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetDomainReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainReplicationMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDynamicConfig(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetDynamicConfig_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetDynamicConfig': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDynamicConfig_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetGlobalIsolationGroups(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetGlobalIsolationGroups_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetGlobalIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetGlobalIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetGlobalIsolationGroups_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetReplicationMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetReplicationMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetReplicationMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetWorkflowExecutionRawHistoryV2(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_GetWorkflowExecutionRawHistoryV2_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'GetWorkflowExecutionRawHistoryV2': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetWorkflowExecutionRawHistoryV2(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetWorkflowExecutionRawHistoryV2_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListDynamicConfig(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_ListDynamicConfig_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'ListDynamicConfig': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ListDynamicConfig_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) MaintainCorruptWorkflow(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_MaintainCorruptWorkflow_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'MaintainCorruptWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MaintainCorruptWorkflow(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_MaintainCorruptWorkflow_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) MergeDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_MergeDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'MergeDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MergeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_MergeDLQMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PurgeDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_PurgeDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'PurgeDLQMessages': %w\", err)\n\t}\n\n\tappErr := h.impl.PurgeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_PurgeDLQMessages_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ReadDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_ReadDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'ReadDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ReadDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ReadDLQMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ReapplyEvents(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_ReapplyEvents_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'ReapplyEvents': %w\", err)\n\t}\n\n\tappErr := h.impl.ReapplyEvents(ctx, args.ReapplyEventsRequest)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ReapplyEvents_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RefreshWorkflowTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_RefreshWorkflowTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RemoveTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_RemoveTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'RemoveTask': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveTask(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RemoveTask_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResendReplicationTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_ResendReplicationTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'ResendReplicationTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.ResendReplicationTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ResendReplicationTasks_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetQueue(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_ResetQueue_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'ResetQueue': %w\", err)\n\t}\n\n\tappErr := h.impl.ResetQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ResetQueue_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondCrossClusterTasksCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_RespondCrossClusterTasksCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'RespondCrossClusterTasksCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondCrossClusterTasksCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RestoreDynamicConfig(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_RestoreDynamicConfig_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'RestoreDynamicConfig': %w\", err)\n\t}\n\n\tappErr := h.impl.RestoreDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RestoreDynamicConfig_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'UpdateDomainAsyncWorkflowConfiguraton': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomainAsyncWorkflowConfiguraton(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) UpdateDomainIsolationGroups(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_UpdateDomainIsolationGroups_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'UpdateDomainIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomainIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDomainIsolationGroups_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) UpdateDynamicConfig(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_UpdateDynamicConfig_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'UpdateDynamicConfig': %w\", err)\n\t}\n\n\tappErr := h.impl.UpdateDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDynamicConfig_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) UpdateGlobalIsolationGroups(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args admin.AdminService_UpdateGlobalIsolationGroups_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'AdminService' procedure 'UpdateGlobalIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateGlobalIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateGlobalIsolationGroups_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\ntype addsearchattribute_NoWireHandler struct{ impl Interface }\n\nfunc (h addsearchattribute_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_AddSearchAttribute_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'AddSearchAttribute': %w\", err)\n\t}\n\n\tappErr := h.impl.AddSearchAttribute(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_AddSearchAttribute_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype closeshard_NoWireHandler struct{ impl Interface }\n\nfunc (h closeshard_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_CloseShard_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'CloseShard': %w\", err)\n\t}\n\n\tappErr := h.impl.CloseShard(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_CloseShard_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype deleteworkflow_NoWireHandler struct{ impl Interface }\n\nfunc (h deleteworkflow_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DeleteWorkflow_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DeleteWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DeleteWorkflow(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DeleteWorkflow_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describecluster_NoWireHandler struct{ impl Interface }\n\nfunc (h describecluster_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DescribeCluster_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DescribeCluster': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeCluster(ctx)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeCluster_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describehistoryhost_NoWireHandler struct{ impl Interface }\n\nfunc (h describehistoryhost_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DescribeHistoryHost_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DescribeHistoryHost': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeHistoryHost(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeHistoryHost_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describequeue_NoWireHandler struct{ impl Interface }\n\nfunc (h describequeue_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DescribeQueue_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DescribeQueue': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeQueue_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describesharddistribution_NoWireHandler struct{ impl Interface }\n\nfunc (h describesharddistribution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DescribeShardDistribution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DescribeShardDistribution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeShardDistribution(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeShardDistribution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describeworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h describeworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_DescribeWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getcrossclustertasks_NoWireHandler struct{ impl Interface }\n\nfunc (h getcrossclustertasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetCrossClusterTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetCrossClusterTasks': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetCrossClusterTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetCrossClusterTasks_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdlqreplicationmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h getdlqreplicationmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetDLQReplicationMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetDLQReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDLQReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDLQReplicationMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdomainasyncworkflowconfiguraton_NoWireHandler struct{ impl Interface }\n\nfunc (h getdomainasyncworkflowconfiguraton_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetDomainAsyncWorkflowConfiguraton': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainAsyncWorkflowConfiguraton(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdomainisolationgroups_NoWireHandler struct{ impl Interface }\n\nfunc (h getdomainisolationgroups_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetDomainIsolationGroups_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetDomainIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainIsolationGroups_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdomainreplicationmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h getdomainreplicationmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetDomainReplicationMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetDomainReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDomainReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDomainReplicationMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdynamicconfig_NoWireHandler struct{ impl Interface }\n\nfunc (h getdynamicconfig_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetDynamicConfig_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetDynamicConfig': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetDynamicConfig_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getglobalisolationgroups_NoWireHandler struct{ impl Interface }\n\nfunc (h getglobalisolationgroups_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetGlobalIsolationGroups_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetGlobalIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetGlobalIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetGlobalIsolationGroups_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getreplicationmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h getreplicationmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetReplicationMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetReplicationMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getworkflowexecutionrawhistoryv2_NoWireHandler struct{ impl Interface }\n\nfunc (h getworkflowexecutionrawhistoryv2_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_GetWorkflowExecutionRawHistoryV2_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'GetWorkflowExecutionRawHistoryV2': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetWorkflowExecutionRawHistoryV2(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_GetWorkflowExecutionRawHistoryV2_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listdynamicconfig_NoWireHandler struct{ impl Interface }\n\nfunc (h listdynamicconfig_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_ListDynamicConfig_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'ListDynamicConfig': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ListDynamicConfig_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype maintaincorruptworkflow_NoWireHandler struct{ impl Interface }\n\nfunc (h maintaincorruptworkflow_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_MaintainCorruptWorkflow_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'MaintainCorruptWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MaintainCorruptWorkflow(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_MaintainCorruptWorkflow_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype mergedlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h mergedlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_MergeDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'MergeDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MergeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_MergeDLQMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype purgedlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h purgedlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_PurgeDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'PurgeDLQMessages': %w\", err)\n\t}\n\n\tappErr := h.impl.PurgeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_PurgeDLQMessages_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype readdlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h readdlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_ReadDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'ReadDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ReadDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ReadDLQMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype reapplyevents_NoWireHandler struct{ impl Interface }\n\nfunc (h reapplyevents_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_ReapplyEvents_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'ReapplyEvents': %w\", err)\n\t}\n\n\tappErr := h.impl.ReapplyEvents(ctx, args.ReapplyEventsRequest)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ReapplyEvents_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype refreshworkflowtasks_NoWireHandler struct{ impl Interface }\n\nfunc (h refreshworkflowtasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_RefreshWorkflowTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype removetask_NoWireHandler struct{ impl Interface }\n\nfunc (h removetask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_RemoveTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'RemoveTask': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveTask(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RemoveTask_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resendreplicationtasks_NoWireHandler struct{ impl Interface }\n\nfunc (h resendreplicationtasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_ResendReplicationTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'ResendReplicationTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.ResendReplicationTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ResendReplicationTasks_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetqueue_NoWireHandler struct{ impl Interface }\n\nfunc (h resetqueue_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_ResetQueue_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'ResetQueue': %w\", err)\n\t}\n\n\tappErr := h.impl.ResetQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_ResetQueue_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondcrossclustertaskscompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondcrossclustertaskscompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_RespondCrossClusterTasksCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'RespondCrossClusterTasksCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondCrossClusterTasksCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype restoredynamicconfig_NoWireHandler struct{ impl Interface }\n\nfunc (h restoredynamicconfig_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_RestoreDynamicConfig_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'RestoreDynamicConfig': %w\", err)\n\t}\n\n\tappErr := h.impl.RestoreDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_RestoreDynamicConfig_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype updatedomainasyncworkflowconfiguraton_NoWireHandler struct{ impl Interface }\n\nfunc (h updatedomainasyncworkflowconfiguraton_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'UpdateDomainAsyncWorkflowConfiguraton': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomainAsyncWorkflowConfiguraton(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDomainAsyncWorkflowConfiguraton_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype updatedomainisolationgroups_NoWireHandler struct{ impl Interface }\n\nfunc (h updatedomainisolationgroups_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_UpdateDomainIsolationGroups_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'UpdateDomainIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomainIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDomainIsolationGroups_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype updatedynamicconfig_NoWireHandler struct{ impl Interface }\n\nfunc (h updatedynamicconfig_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_UpdateDynamicConfig_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'UpdateDynamicConfig': %w\", err)\n\t}\n\n\tappErr := h.impl.UpdateDynamicConfig(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateDynamicConfig_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype updateglobalisolationgroups_NoWireHandler struct{ impl Interface }\n\nfunc (h updateglobalisolationgroups_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs admin.AdminService_UpdateGlobalIsolationGroups_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'AdminService' procedure 'UpdateGlobalIsolationGroups': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateGlobalIsolationGroups(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := admin.AdminService_UpdateGlobalIsolationGroups_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n"
  },
  {
    "path": ".gen/go/admin/adminservicetest/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage adminservicetest\n\nimport (\n\tcontext \"context\"\n\n\tgomock \"github.com/golang/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tadmin \"github.com/uber/cadence/.gen/go/admin\"\n\tadminserviceclient \"github.com/uber/cadence/.gen/go/admin/adminserviceclient\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// MockClient implements a gomock-compatible mock client for service\n// AdminService.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *_MockClientRecorder\n}\n\nvar _ adminserviceclient.Interface = (*MockClient)(nil)\n\ntype _MockClientRecorder struct {\n\tmock *MockClient\n}\n\n// Build a new mock client for service AdminService.\n//\n//\tmockCtrl := gomock.NewController(t)\n//\tclient := adminservicetest.NewMockClient(mockCtrl)\n//\n// Use EXPECT() to set expectations on the mock.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &_MockClientRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows you to define an expectation on the\n// AdminService mock client.\nfunc (m *MockClient) EXPECT() *_MockClientRecorder {\n\treturn m.recorder\n}\n\n// AddSearchAttribute responds to a AddSearchAttribute call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().AddSearchAttribute(gomock.Any(), ...).Return(...)\n//\t... := client.AddSearchAttribute(...)\nfunc (m *MockClient) AddSearchAttribute(\n\tctx context.Context,\n\t_Request *admin.AddSearchAttributeRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"AddSearchAttribute\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) AddSearchAttribute(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"AddSearchAttribute\", args...)\n}\n\n// CloseShard responds to a CloseShard call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().CloseShard(gomock.Any(), ...).Return(...)\n//\t... := client.CloseShard(...)\nfunc (m *MockClient) CloseShard(\n\tctx context.Context,\n\t_Request *shared.CloseShardRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"CloseShard\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) CloseShard(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"CloseShard\", args...)\n}\n\n// DeleteWorkflow responds to a DeleteWorkflow call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DeleteWorkflow(gomock.Any(), ...).Return(...)\n//\t... := client.DeleteWorkflow(...)\nfunc (m *MockClient) DeleteWorkflow(\n\tctx context.Context,\n\t_Request *admin.AdminDeleteWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.AdminDeleteWorkflowResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DeleteWorkflow\", args...)\n\tsuccess, _ = ret[i].(*admin.AdminDeleteWorkflowResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DeleteWorkflow(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DeleteWorkflow\", args...)\n}\n\n// DescribeCluster responds to a DescribeCluster call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeCluster(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeCluster(...)\nfunc (m *MockClient) DescribeCluster(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *admin.DescribeClusterResponse, err error) {\n\n\targs := []interface{}{ctx}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeCluster\", args...)\n\tsuccess, _ = ret[i].(*admin.DescribeClusterResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeCluster(\n\tctx interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeCluster\", args...)\n}\n\n// DescribeHistoryHost responds to a DescribeHistoryHost call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeHistoryHost(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeHistoryHost(...)\nfunc (m *MockClient) DescribeHistoryHost(\n\tctx context.Context,\n\t_Request *shared.DescribeHistoryHostRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeHistoryHostResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeHistoryHostResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeHistoryHost(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeHistoryHost\", args...)\n}\n\n// DescribeQueue responds to a DescribeQueue call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeQueue(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeQueue(...)\nfunc (m *MockClient) DescribeQueue(\n\tctx context.Context,\n\t_Request *shared.DescribeQueueRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeQueueResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeQueue\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeQueueResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeQueue(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeQueue\", args...)\n}\n\n// DescribeShardDistribution responds to a DescribeShardDistribution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeShardDistribution(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeShardDistribution(...)\nfunc (m *MockClient) DescribeShardDistribution(\n\tctx context.Context,\n\t_Request *shared.DescribeShardDistributionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeShardDistributionResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeShardDistribution\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeShardDistributionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeShardDistribution(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeShardDistribution\", args...)\n}\n\n// DescribeWorkflowExecution responds to a DescribeWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeWorkflowExecution(...)\nfunc (m *MockClient) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_Request *admin.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.DescribeWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*admin.DescribeWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeWorkflowExecution(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeWorkflowExecution\", args...)\n}\n\n// GetCrossClusterTasks responds to a GetCrossClusterTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetCrossClusterTasks(gomock.Any(), ...).Return(...)\n//\t... := client.GetCrossClusterTasks(...)\nfunc (m *MockClient) GetCrossClusterTasks(\n\tctx context.Context,\n\t_Request *shared.GetCrossClusterTasksRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetCrossClusterTasksResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetCrossClusterTasks\", args...)\n\tsuccess, _ = ret[i].(*shared.GetCrossClusterTasksResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetCrossClusterTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetCrossClusterTasks\", args...)\n}\n\n// GetDLQReplicationMessages responds to a GetDLQReplicationMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDLQReplicationMessages(gomock.Any(), ...).Return(...)\n//\t... := client.GetDLQReplicationMessages(...)\nfunc (m *MockClient) GetDLQReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDLQReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.GetDLQReplicationMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDLQReplicationMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDLQReplicationMessages\", args...)\n}\n\n// GetDomainAsyncWorkflowConfiguraton responds to a GetDomainAsyncWorkflowConfiguraton call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDomainAsyncWorkflowConfiguraton(gomock.Any(), ...).Return(...)\n//\t... := client.GetDomainAsyncWorkflowConfiguraton(...)\nfunc (m *MockClient) GetDomainAsyncWorkflowConfiguraton(\n\tctx context.Context,\n\t_Request *admin.GetDomainAsyncWorkflowConfiguratonRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDomainAsyncWorkflowConfiguraton\", args...)\n\tsuccess, _ = ret[i].(*admin.GetDomainAsyncWorkflowConfiguratonResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDomainAsyncWorkflowConfiguraton(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDomainAsyncWorkflowConfiguraton\", args...)\n}\n\n// GetDomainIsolationGroups responds to a GetDomainIsolationGroups call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDomainIsolationGroups(gomock.Any(), ...).Return(...)\n//\t... := client.GetDomainIsolationGroups(...)\nfunc (m *MockClient) GetDomainIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.GetDomainIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDomainIsolationGroupsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDomainIsolationGroups\", args...)\n\tsuccess, _ = ret[i].(*admin.GetDomainIsolationGroupsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDomainIsolationGroups(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDomainIsolationGroups\", args...)\n}\n\n// GetDomainReplicationMessages responds to a GetDomainReplicationMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDomainReplicationMessages(gomock.Any(), ...).Return(...)\n//\t... := client.GetDomainReplicationMessages(...)\nfunc (m *MockClient) GetDomainReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDomainReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDomainReplicationMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDomainReplicationMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.GetDomainReplicationMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDomainReplicationMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDomainReplicationMessages\", args...)\n}\n\n// GetDynamicConfig responds to a GetDynamicConfig call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDynamicConfig(gomock.Any(), ...).Return(...)\n//\t... := client.GetDynamicConfig(...)\nfunc (m *MockClient) GetDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.GetDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetDynamicConfigResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDynamicConfig\", args...)\n\tsuccess, _ = ret[i].(*admin.GetDynamicConfigResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDynamicConfig(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDynamicConfig\", args...)\n}\n\n// GetGlobalIsolationGroups responds to a GetGlobalIsolationGroups call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetGlobalIsolationGroups(gomock.Any(), ...).Return(...)\n//\t... := client.GetGlobalIsolationGroups(...)\nfunc (m *MockClient) GetGlobalIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.GetGlobalIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.GetGlobalIsolationGroupsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetGlobalIsolationGroups\", args...)\n\tsuccess, _ = ret[i].(*admin.GetGlobalIsolationGroupsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetGlobalIsolationGroups(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetGlobalIsolationGroups\", args...)\n}\n\n// GetReplicationMessages responds to a GetReplicationMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetReplicationMessages(gomock.Any(), ...).Return(...)\n//\t... := client.GetReplicationMessages(...)\nfunc (m *MockClient) GetReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetReplicationMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.GetReplicationMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetReplicationMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetReplicationMessages\", args...)\n}\n\n// GetWorkflowExecutionRawHistoryV2 responds to a GetWorkflowExecutionRawHistoryV2 call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetWorkflowExecutionRawHistoryV2(gomock.Any(), ...).Return(...)\n//\t... := client.GetWorkflowExecutionRawHistoryV2(...)\nfunc (m *MockClient) GetWorkflowExecutionRawHistoryV2(\n\tctx context.Context,\n\t_GetRequest *admin.GetWorkflowExecutionRawHistoryV2Request,\n\topts ...yarpc.CallOption,\n) (success *admin.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\n\targs := []interface{}{ctx, _GetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionRawHistoryV2\", args...)\n\tsuccess, _ = ret[i].(*admin.GetWorkflowExecutionRawHistoryV2Response)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetWorkflowExecutionRawHistoryV2(\n\tctx interface{},\n\t_GetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _GetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetWorkflowExecutionRawHistoryV2\", args...)\n}\n\n// ListDynamicConfig responds to a ListDynamicConfig call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListDynamicConfig(gomock.Any(), ...).Return(...)\n//\t... := client.ListDynamicConfig(...)\nfunc (m *MockClient) ListDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.ListDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.ListDynamicConfigResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListDynamicConfig\", args...)\n\tsuccess, _ = ret[i].(*admin.ListDynamicConfigResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListDynamicConfig(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListDynamicConfig\", args...)\n}\n\n// MaintainCorruptWorkflow responds to a MaintainCorruptWorkflow call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().MaintainCorruptWorkflow(gomock.Any(), ...).Return(...)\n//\t... := client.MaintainCorruptWorkflow(...)\nfunc (m *MockClient) MaintainCorruptWorkflow(\n\tctx context.Context,\n\t_Request *admin.AdminMaintainWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.AdminMaintainWorkflowResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"MaintainCorruptWorkflow\", args...)\n\tsuccess, _ = ret[i].(*admin.AdminMaintainWorkflowResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) MaintainCorruptWorkflow(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"MaintainCorruptWorkflow\", args...)\n}\n\n// MergeDLQMessages responds to a MergeDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().MergeDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.MergeDLQMessages(...)\nfunc (m *MockClient) MergeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.MergeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.MergeDLQMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.MergeDLQMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) MergeDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"MergeDLQMessages\", args...)\n}\n\n// PurgeDLQMessages responds to a PurgeDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PurgeDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.PurgeDLQMessages(...)\nfunc (m *MockClient) PurgeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.PurgeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PurgeDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PurgeDLQMessages\", args...)\n}\n\n// ReadDLQMessages responds to a ReadDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ReadDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.ReadDLQMessages(...)\nfunc (m *MockClient) ReadDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.ReadDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.ReadDLQMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.ReadDLQMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ReadDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ReadDLQMessages\", args...)\n}\n\n// ReapplyEvents responds to a ReapplyEvents call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ReapplyEvents(gomock.Any(), ...).Return(...)\n//\t... := client.ReapplyEvents(...)\nfunc (m *MockClient) ReapplyEvents(\n\tctx context.Context,\n\t_ReapplyEventsRequest *shared.ReapplyEventsRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _ReapplyEventsRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ReapplyEvents(\n\tctx interface{},\n\t_ReapplyEventsRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ReapplyEventsRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ReapplyEvents\", args...)\n}\n\n// RefreshWorkflowTasks responds to a RefreshWorkflowTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RefreshWorkflowTasks(gomock.Any(), ...).Return(...)\n//\t... := client.RefreshWorkflowTasks(...)\nfunc (m *MockClient) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *shared.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RefreshWorkflowTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RefreshWorkflowTasks\", args...)\n}\n\n// RemoveTask responds to a RemoveTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RemoveTask(gomock.Any(), ...).Return(...)\n//\t... := client.RemoveTask(...)\nfunc (m *MockClient) RemoveTask(\n\tctx context.Context,\n\t_Request *shared.RemoveTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RemoveTask\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RemoveTask(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RemoveTask\", args...)\n}\n\n// ResendReplicationTasks responds to a ResendReplicationTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResendReplicationTasks(gomock.Any(), ...).Return(...)\n//\t... := client.ResendReplicationTasks(...)\nfunc (m *MockClient) ResendReplicationTasks(\n\tctx context.Context,\n\t_Request *admin.ResendReplicationTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResendReplicationTasks\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResendReplicationTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResendReplicationTasks\", args...)\n}\n\n// ResetQueue responds to a ResetQueue call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetQueue(gomock.Any(), ...).Return(...)\n//\t... := client.ResetQueue(...)\nfunc (m *MockClient) ResetQueue(\n\tctx context.Context,\n\t_Request *shared.ResetQueueRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetQueue\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetQueue(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetQueue\", args...)\n}\n\n// RespondCrossClusterTasksCompleted responds to a RespondCrossClusterTasksCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondCrossClusterTasksCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondCrossClusterTasksCompleted(...)\nfunc (m *MockClient) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\t_Request *shared.RespondCrossClusterTasksCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondCrossClusterTasksCompleted\", args...)\n\tsuccess, _ = ret[i].(*shared.RespondCrossClusterTasksCompletedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondCrossClusterTasksCompleted(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondCrossClusterTasksCompleted\", args...)\n}\n\n// RestoreDynamicConfig responds to a RestoreDynamicConfig call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RestoreDynamicConfig(gomock.Any(), ...).Return(...)\n//\t... := client.RestoreDynamicConfig(...)\nfunc (m *MockClient) RestoreDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.RestoreDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RestoreDynamicConfig\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RestoreDynamicConfig(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RestoreDynamicConfig\", args...)\n}\n\n// UpdateDomainAsyncWorkflowConfiguraton responds to a UpdateDomainAsyncWorkflowConfiguraton call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().UpdateDomainAsyncWorkflowConfiguraton(gomock.Any(), ...).Return(...)\n//\t... := client.UpdateDomainAsyncWorkflowConfiguraton(...)\nfunc (m *MockClient) UpdateDomainAsyncWorkflowConfiguraton(\n\tctx context.Context,\n\t_Request *admin.UpdateDomainAsyncWorkflowConfiguratonRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"UpdateDomainAsyncWorkflowConfiguraton\", args...)\n\tsuccess, _ = ret[i].(*admin.UpdateDomainAsyncWorkflowConfiguratonResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) UpdateDomainAsyncWorkflowConfiguraton(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"UpdateDomainAsyncWorkflowConfiguraton\", args...)\n}\n\n// UpdateDomainIsolationGroups responds to a UpdateDomainIsolationGroups call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().UpdateDomainIsolationGroups(gomock.Any(), ...).Return(...)\n//\t... := client.UpdateDomainIsolationGroups(...)\nfunc (m *MockClient) UpdateDomainIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.UpdateDomainIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateDomainIsolationGroupsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"UpdateDomainIsolationGroups\", args...)\n\tsuccess, _ = ret[i].(*admin.UpdateDomainIsolationGroupsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) UpdateDomainIsolationGroups(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"UpdateDomainIsolationGroups\", args...)\n}\n\n// UpdateDynamicConfig responds to a UpdateDynamicConfig call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().UpdateDynamicConfig(gomock.Any(), ...).Return(...)\n//\t... := client.UpdateDynamicConfig(...)\nfunc (m *MockClient) UpdateDynamicConfig(\n\tctx context.Context,\n\t_Request *admin.UpdateDynamicConfigRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"UpdateDynamicConfig\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) UpdateDynamicConfig(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"UpdateDynamicConfig\", args...)\n}\n\n// UpdateGlobalIsolationGroups responds to a UpdateGlobalIsolationGroups call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().UpdateGlobalIsolationGroups(gomock.Any(), ...).Return(...)\n//\t... := client.UpdateGlobalIsolationGroups(...)\nfunc (m *MockClient) UpdateGlobalIsolationGroups(\n\tctx context.Context,\n\t_Request *admin.UpdateGlobalIsolationGroupsRequest,\n\topts ...yarpc.CallOption,\n) (success *admin.UpdateGlobalIsolationGroupsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"UpdateGlobalIsolationGroups\", args...)\n\tsuccess, _ = ret[i].(*admin.UpdateGlobalIsolationGroupsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) UpdateGlobalIsolationGroups(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"UpdateGlobalIsolationGroups\", args...)\n}\n"
  },
  {
    "path": ".gen/go/admin/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage admin\n"
  },
  {
    "path": ".gen/go/cadence/cadence.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage cadence\n\nimport (\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"cadence\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/cadence\",\n\tFilePath: \"cadence.thrift\",\n\tSHA1:     \"8c644a4a8acae7e865a84d625bc845ffae7ff693\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\ninclude \\\"shared.thrift\\\"\\n\\nnamespace java com.uber.cadence\\n\\n/**\\n* WorkflowService API is exposed to provide support for long running applications.  Application is expected to call\\n* StartWorkflowExecution to create an instance for each instance of long running workflow.  Such applications are expected\\n* to have a worker which regularly polls for DecisionTask and ActivityTask from the WorkflowService.  For each\\n* DecisionTask, application is expected to process the history of events for that session and respond back with next\\n* decisions.  For each ActivityTask, application is expected to execute the actual logic for that task and respond back\\n* with completion or failure.  Worker is expected to regularly heartbeat while activity task is running.\\n**/\\nservice WorkflowService {\\n  /**\\n  * RegisterDomain creates a new domain which can be used as a container for all resources.  Domain is a top level\\n  * entity within Cadence, used as a container for all resources like workflow executions, tasklists, etc.  Domain\\n  * acts as a sandbox and provides isolation for all resources within the domain.  All resources belongs to exactly one\\n  * domain.\\n  **/\\n  void RegisterDomain(1: shared.RegisterDomainRequest registerRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.DomainAlreadyExistsError domainExistsError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * DescribeDomain returns the information and configuration for a registered domain.\\n  **/\\n  shared.DescribeDomainResponse DescribeDomain(1: shared.DescribeDomainRequest describeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n    * ListDomains returns the information and configuration for all domains.\\n    **/\\n    shared.ListDomainsResponse ListDomains(1: shared.ListDomainsRequest listRequest)\\n      throws (\\n        1: shared.BadRequestError badRequestError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n        5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n        6: shared.AccessDeniedError accessDeniedError,\\n      )\\n\\n  /**\\n  * UpdateDomain is used to update the information and configuration for a registered domain.\\n  **/\\n  shared.UpdateDomainResponse UpdateDomain(1: shared.UpdateDomainRequest updateRequest)\\n      throws (\\n        1: shared.BadRequestError badRequestError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n        5: shared.DomainNotActiveError domainNotActiveError,\\n        6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n        7: shared.AccessDeniedError accessDeniedError,\\n      )\\n\\n  /**\\n  * FailoverDomain is used to failover a registered domain to different cluster.\\n  **/\\n  shared.FailoverDomainResponse FailoverDomain(1: shared.FailoverDomainRequest failoverRequest)\\n      throws (\\n        1: shared.BadRequestError badRequestError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n        5: shared.DomainNotActiveError domainNotActiveError,\\n        6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n        7: shared.AccessDeniedError accessDeniedError,\\n      )\\n\\n  /**\\n  * DeprecateDomain us used to update status of a registered domain to DEPRECATED.  Once the domain is deprecated\\n  * it cannot be used to start new workflow executions.  Existing workflow executions will continue to run on\\n  * deprecated domains.\\n  **/\\n  void DeprecateDomain(1: shared.DeprecateDomainRequest deprecateRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      7: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * DeleteDomain permanently removes a domain record. This operation:\\n  * - Requires domain to be in DEPRECATED status\\n  * - Cannot be performed on domains with running workflows\\n  * - Is irreversible and removes all domain data\\n  * - Requires proper permissions and security token\\n  **/\\n  void DeleteDomain(1: shared.DeleteDomainRequest deleteRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      4: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ListFailoverHistory returns the history of failover events for a domain.\\n  **/\\n  shared.ListFailoverHistoryResponse ListFailoverHistory(1: shared.ListFailoverHistoryRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      4: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RestartWorkflowExecution restarts a previous workflow\\n  * If the workflow is currently running it will terminate and restart\\n  **/\\n  shared.RestartWorkflowExecutionResponse RestartWorkflowExecution(1: shared.RestartWorkflowExecutionRequest restartRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.DomainNotActiveError domainNotActiveError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.EntityNotExistsError entityNotExistError,\\n      6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      7: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * DiagnoseWorkflowExecution diagnoses a previous workflow execution\\n  **/\\n  shared.DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(1: shared.DiagnoseWorkflowExecutionRequest diagnoseRequest)\\n    throws (\\n      1: shared.DomainNotActiveError domainNotActiveError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      5: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * StartWorkflowExecution starts a new long running workflow instance.  It will create the instance with\\n  * 'WorkflowExecutionStarted' event in history and also schedule the first DecisionTask for the worker to make the\\n  * first decision for this instance.  It will return 'WorkflowExecutionAlreadyStartedError', if an instance already\\n  * exists with same workflowId.\\n  **/\\n  shared.StartWorkflowExecutionResponse StartWorkflowExecution(1: shared.StartWorkflowExecutionRequest startRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.WorkflowExecutionAlreadyStartedError sessionAlreadyExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.EntityNotExistsError entityNotExistError,\\n      8: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n  /**\\n  * StartWorkflowExecutionAsync starts a new long running workflow instance asynchronously. It will push a StartWorkflowExecutionRequest to a queue\\n  * and immediately return a response. The request will be processed by a separate consumer eventually.\\n  **/\\n  shared.StartWorkflowExecutionAsyncResponse StartWorkflowExecutionAsync(1: shared.StartWorkflowExecutionAsyncRequest startRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.WorkflowExecutionAlreadyStartedError sessionAlreadyExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.EntityNotExistsError entityNotExistError,\\n      8: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n  /**\\n  * Returns the history of specified workflow execution.  It fails with 'EntityNotExistError' if speficied workflow\\n  * execution in unknown to the service.\\n  **/\\n  shared.GetWorkflowExecutionHistoryResponse GetWorkflowExecutionHistory(1: shared.GetWorkflowExecutionHistoryRequest getRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * PollForDecisionTask is called by application worker to process DecisionTask from a specific taskList.  A\\n  * DecisionTask is dispatched to callers for active workflow executions, with pending decisions.\\n  * Application is then expected to call 'RespondDecisionTaskCompleted' API when it is done processing the DecisionTask.\\n  * It will also create a 'DecisionTaskStarted' event in the history for that session before handing off DecisionTask to\\n  * application worker.\\n  **/\\n  shared.PollForDecisionTaskResponse PollForDecisionTask(1: shared.PollForDecisionTaskRequest pollRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.EntityNotExistsError entityNotExistError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondDecisionTaskCompleted is called by application worker to complete a DecisionTask handed as a result of\\n  * 'PollForDecisionTask' API call.  Completing a DecisionTask will result in new events for the workflow execution and\\n  * potentially new ActivityTask being created for corresponding decisions.  It will also create a DecisionTaskCompleted\\n  * event in the history for that session.  Use the 'taskToken' provided as response of PollForDecisionTask API call\\n  * for completing the DecisionTask.\\n  * The response could contain a new decision task if there is one or if the request asking for one.\\n  **/\\n  shared.RespondDecisionTaskCompletedResponse RespondDecisionTaskCompleted(1: shared.RespondDecisionTaskCompletedRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondDecisionTaskFailed is called by application worker to indicate failure.  This results in\\n  * DecisionTaskFailedEvent written to the history and a new DecisionTask created.  This API can be used by client to\\n  * either clear sticky tasklist or report any panics during DecisionTask processing.  Cadence will only append first\\n  * DecisionTaskFailed event to the history of workflow execution for consecutive failures.\\n  **/\\n  void RespondDecisionTaskFailed(1: shared.RespondDecisionTaskFailedRequest failedRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * PollForActivityTask is called by application worker to process ActivityTask from a specific taskList.  ActivityTask\\n  * is dispatched to callers whenever a ScheduleTask decision is made for a workflow execution.\\n  * Application is expected to call 'RespondActivityTaskCompleted' or 'RespondActivityTaskFailed' once it is done\\n  * processing the task.\\n  * Application also needs to call 'RecordActivityTaskHeartbeat' API within 'heartbeatTimeoutSeconds' interval to\\n  * prevent the task from getting timed out.  An event 'ActivityTaskStarted' event is also written to workflow execution\\n  * history before the ActivityTask is dispatched to application worker.\\n  **/\\n  shared.PollForActivityTaskResponse PollForActivityTask(1: shared.PollForActivityTaskRequest pollRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.EntityNotExistsError entityNotExistError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RecordActivityTaskHeartbeat is called by application worker while it is processing an ActivityTask.  If worker fails\\n  * to heartbeat within 'heartbeatTimeoutSeconds' interval for the ActivityTask, then it will be marked as timedout and\\n  * 'ActivityTaskTimedOut' event will be written to the workflow history.  Calling 'RecordActivityTaskHeartbeat' will\\n  * fail with 'EntityNotExistsError' in such situations.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for heartbeating.\\n  **/\\n  shared.RecordActivityTaskHeartbeatResponse RecordActivityTaskHeartbeat(1: shared.RecordActivityTaskHeartbeatRequest heartbeatRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RecordActivityTaskHeartbeatByID is called by application worker while it is processing an ActivityTask.  If worker fails\\n  * to heartbeat within 'heartbeatTimeoutSeconds' interval for the ActivityTask, then it will be marked as timedout and\\n  * 'ActivityTaskTimedOut' event will be written to the workflow history.  Calling 'RecordActivityTaskHeartbeatByID' will\\n  * fail with 'EntityNotExistsError' in such situations.  Instead of using 'taskToken' like in RecordActivityTaskHeartbeat,\\n  * use Domain, WorkflowID and ActivityID\\n  **/\\n  shared.RecordActivityTaskHeartbeatResponse RecordActivityTaskHeartbeatByID(1: shared.RecordActivityTaskHeartbeatByIDRequest heartbeatRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCompleted is called by application worker when it is done processing an ActivityTask.  It will\\n  * result in a new 'ActivityTaskCompleted' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void  RespondActivityTaskCompleted(1: shared.RespondActivityTaskCompletedRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCompletedByID is called by application worker when it is done processing an ActivityTask.\\n  * It will result in a new 'ActivityTaskCompleted' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow so new decisions could be made.  Similar to RespondActivityTaskCompleted but use Domain,\\n  * WorkflowID and ActivityID instead of 'taskToken' for completion. It fails with 'EntityNotExistsError'\\n  * if the these IDs are not valid anymore due to activity timeout.\\n  **/\\n  void  RespondActivityTaskCompletedByID(1: shared.RespondActivityTaskCompletedByIDRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskFailed is called by application worker when it is done processing an ActivityTask.  It will\\n  * result in a new 'ActivityTaskFailed' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void  RespondActivityTaskFailed(1: shared.RespondActivityTaskFailedRequest failRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskFailedByID is called by application worker when it is done processing an ActivityTask.\\n  * It will result in a new 'ActivityTaskFailed' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Similar to RespondActivityTaskFailed but use\\n  * Domain, WorkflowID and ActivityID instead of 'taskToken' for completion. It fails with 'EntityNotExistsError'\\n  * if the these IDs are not valid anymore due to activity timeout.\\n  **/\\n  void  RespondActivityTaskFailedByID(1: shared.RespondActivityTaskFailedByIDRequest failRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCanceled is called by application worker when it is successfully canceled an ActivityTask.  It will\\n  * result in a new 'ActivityTaskCanceled' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void RespondActivityTaskCanceled(1: shared.RespondActivityTaskCanceledRequest canceledRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCanceledByID is called by application worker when it is successfully canceled an ActivityTask.\\n  * It will result in a new 'ActivityTaskCanceled' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Similar to RespondActivityTaskCanceled but use\\n  * Domain, WorkflowID and ActivityID instead of 'taskToken' for completion. It fails with 'EntityNotExistsError'\\n  * if the these IDs are not valid anymore due to activity timeout.\\n  **/\\n  void RespondActivityTaskCanceledByID(1: shared.RespondActivityTaskCanceledByIDRequest canceledRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RequestCancelWorkflowExecution is called by application worker when it wants to request cancellation of a workflow instance.\\n  * It will result in a new 'WorkflowExecutionCancelRequested' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made. It fails with 'EntityNotExistsError' if the workflow is not valid\\n  * anymore due to completion or doesn't exist.\\n  **/\\n  void RequestCancelWorkflowExecution(1: shared.RequestCancelWorkflowExecutionRequest cancelRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.CancellationAlreadyRequestedError cancellationAlreadyRequestedError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.LimitExceededError limitExceededError,\\n      8: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      9: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      10: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * SignalWorkflowExecution is used to send a signal event to running workflow execution.  This results in\\n  * WorkflowExecutionSignaled event recorded in the history and a decision task being created for the execution.\\n  **/\\n  void SignalWorkflowExecution(1: shared.SignalWorkflowExecutionRequest signalRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * SignalWithStartWorkflowExecution is used to ensure sending signal to a workflow.\\n  * If the workflow is running, this results in WorkflowExecutionSignaled event being recorded in the history\\n  * and a decision task being created for the execution.\\n  * If the workflow is not running or not found, this results in WorkflowExecutionStarted and WorkflowExecutionSignaled\\n  * events being recorded in history, and a decision task being created for the execution\\n  **/\\n  shared.StartWorkflowExecutionResponse SignalWithStartWorkflowExecution(1: shared.SignalWithStartWorkflowExecutionRequest signalWithStartRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.WorkflowExecutionAlreadyStartedError workflowAlreadyStartedError,\\n      8: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * SignalWithStartWorkflowExecutionAsync is used to ensure sending signal to a workflow asynchronously.  It will push a SignalWithStartWorkflowExecutionRequest to a queue\\n  * and immediately return a response. The request will be processed by a separate consumer eventually.\\n  **/\\n  shared.SignalWithStartWorkflowExecutionAsyncResponse SignalWithStartWorkflowExecutionAsync(1: shared.SignalWithStartWorkflowExecutionAsyncRequest signalWithStartRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.WorkflowExecutionAlreadyStartedError sessionAlreadyExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.EntityNotExistsError entityNotExistError,\\n      8: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n  /**\\n    * ResetWorkflowExecution reset an existing workflow execution to DecisionTaskCompleted event(exclusive).\\n    * And it will immediately terminating the current execution instance.\\n    **/\\n  shared.ResetWorkflowExecutionResponse ResetWorkflowExecution(1: shared.ResetWorkflowExecutionRequest resetRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * TerminateWorkflowExecution terminates an existing workflow execution by recording WorkflowExecutionTerminated event\\n  * in the history and immediately terminating the execution instance.\\n  **/\\n  void TerminateWorkflowExecution(1: shared.TerminateWorkflowExecutionRequest terminateRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ListOpenWorkflowExecutions is a visibility API to list the open executions in a specific domain.\\n  **/\\n  shared.ListOpenWorkflowExecutionsResponse ListOpenWorkflowExecutions(1: shared.ListOpenWorkflowExecutionsRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      7: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ListClosedWorkflowExecutions is a visibility API to list the closed executions in a specific domain.\\n  **/\\n  shared.ListClosedWorkflowExecutionsResponse ListClosedWorkflowExecutions(1: shared.ListClosedWorkflowExecutionsRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ListWorkflowExecutions is a visibility API to list workflow executions in a specific domain.\\n  **/\\n  shared.ListWorkflowExecutionsResponse ListWorkflowExecutions(1: shared.ListWorkflowExecutionsRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ListArchivedWorkflowExecutions is a visibility API to list archived workflow executions in a specific domain.\\n  **/\\n  shared.ListArchivedWorkflowExecutionsResponse ListArchivedWorkflowExecutions(1: shared.ListArchivedWorkflowExecutionsRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ScanWorkflowExecutions is a visibility API to list large amount of workflow executions in a specific domain without order.\\n  **/\\n  shared.ListWorkflowExecutionsResponse ScanWorkflowExecutions(1: shared.ListWorkflowExecutionsRequest listRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * CountWorkflowExecutions is a visibility API to count of workflow executions in a specific domain.\\n  **/\\n  shared.CountWorkflowExecutionsResponse CountWorkflowExecutions(1: shared.CountWorkflowExecutionsRequest countRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * GetSearchAttributes is a visibility API to get all legal keys that could be used in list APIs\\n  **/\\n  shared.GetSearchAttributesResponse GetSearchAttributes()\\n    throws (\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      4: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RespondQueryTaskCompleted is called by application worker to complete a QueryTask (which is a DecisionTask for query)\\n  * as a result of 'PollForDecisionTask' API call. Completing a QueryTask will unblock the client call to 'QueryWorkflow'\\n  * API and return the query result to client as a response to 'QueryWorkflow' API call.\\n  **/\\n  void RespondQueryTaskCompleted(1: shared.RespondQueryTaskCompletedRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * Reset the sticky tasklist related information in mutable state of a given workflow.\\n  * Things cleared are:\\n  * 1. StickyTaskList\\n  * 2. StickyScheduleToStartTimeout\\n  * 3. ClientLibraryVersion\\n  * 4. ClientFeatureVersion\\n  * 5. ClientImpl\\n  **/\\n  shared.ResetStickyTaskListResponse ResetStickyTaskList(1: shared.ResetStickyTaskListRequest resetRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n      9: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * QueryWorkflow returns query result for a specified workflow execution\\n  **/\\n  shared.QueryWorkflowResponse QueryWorkflow(1: shared.QueryWorkflowRequest queryRequest)\\n\\tthrows (\\n\\t  1: shared.BadRequestError badRequestError,\\n\\t  3: shared.EntityNotExistsError entityNotExistError,\\n\\t  4: shared.QueryFailedError queryFailedError,\\n\\t  5: shared.LimitExceededError limitExceededError,\\n\\t  6: shared.ServiceBusyError serviceBusyError,\\n\\t  7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n    8: shared.AccessDeniedError accessDeniedError,\\n\\t)\\n\\n  /**\\n  * DescribeWorkflowExecution returns information about the specified workflow execution.\\n  **/\\n  shared.DescribeWorkflowExecutionResponse DescribeWorkflowExecution(1: shared.DescribeWorkflowExecutionRequest describeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      7: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * DescribeTaskList returns information about the target tasklist, right now this API returns the\\n  * pollers which polled this tasklist in last few minutes.\\n  **/\\n  shared.DescribeTaskListResponse DescribeTaskList(1: shared.DescribeTaskListRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      7: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * GetClusterInfo returns information about cadence cluster\\n  **/\\n  shared.ClusterInfo GetClusterInfo()\\n    throws (\\n      1: shared.InternalServiceError internalServiceError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * GetTaskListsByDomain returns the list of all the task lists for a domainName.\\n  **/\\n  shared.GetTaskListsByDomainResponse GetTaskListsByDomain(1: shared.GetTaskListsByDomainRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.EntityNotExistsError entityNotExistError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n   /**\\n   * ReapplyEvents applies stale events to the current workflow and current run\\n   **/\\n  shared.ListTaskListPartitionsResponse ListTaskListPartitions(1: shared.ListTaskListPartitionsRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RefreshWorkflowTasks refreshes all tasks of a workflow\\n  **/\\n  void RefreshWorkflowTasks(1: shared.RefreshWorkflowTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.DomainNotActiveError domainNotActiveError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: shared.AccessDeniedError accessDeniedError,\\n    )\\n}\\n\"\n\n// WorkflowService_CountWorkflowExecutions_Args represents the arguments for the WorkflowService.CountWorkflowExecutions function.\n//\n// The arguments for CountWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_CountWorkflowExecutions_Args struct {\n\tCountRequest *shared.CountWorkflowExecutionsRequest `json:\"countRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_CountWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CountRequest != nil {\n\t\tw, err = v.CountRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CountWorkflowExecutionsRequest_Read(w wire.Value) (*shared.CountWorkflowExecutionsRequest, error) {\n\tvar v shared.CountWorkflowExecutionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_CountWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_CountWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_CountWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CountRequest, err = _CountWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_CountWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_CountWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CountRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CountRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CountWorkflowExecutionsRequest_Decode(sr stream.Reader) (*shared.CountWorkflowExecutionsRequest, error) {\n\tvar v shared.CountWorkflowExecutionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_CountWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_CountWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CountRequest, err = _CountWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_CountWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CountRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CountRequest: %v\", v.CountRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_CountWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_CountWorkflowExecutions_Args match the\n// provided WorkflowService_CountWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) Equals(rhs *WorkflowService_CountWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CountRequest == nil && rhs.CountRequest == nil) || (v.CountRequest != nil && rhs.CountRequest != nil && v.CountRequest.Equals(rhs.CountRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_CountWorkflowExecutions_Args.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CountRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"countRequest\", v.CountRequest))\n\t}\n\treturn err\n}\n\n// GetCountRequest returns the value of CountRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) GetCountRequest() (o *shared.CountWorkflowExecutionsRequest) {\n\tif v != nil && v.CountRequest != nil {\n\t\treturn v.CountRequest\n\t}\n\n\treturn\n}\n\n// IsSetCountRequest returns true if CountRequest is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) IsSetCountRequest() bool {\n\treturn v != nil && v.CountRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"CountWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) MethodName() string {\n\treturn \"CountWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_CountWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.CountWorkflowExecutions\n// function.\nvar WorkflowService_CountWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of CountWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcountRequest *shared.CountWorkflowExecutionsRequest,\n\t) *WorkflowService_CountWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by CountWorkflowExecutions.\n\t//\n\t// An error can be thrown by CountWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for CountWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// CountWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by CountWorkflowExecutions\n\t//\n\t//   value, err := CountWorkflowExecutions(args)\n\t//   result, err := WorkflowService_CountWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from CountWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.CountWorkflowExecutionsResponse, error) (*WorkflowService_CountWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for CountWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if CountWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_CountWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_CountWorkflowExecutions_Result) (*shared.CountWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_CountWorkflowExecutions_Helper.Args = func(\n\t\tcountRequest *shared.CountWorkflowExecutionsRequest,\n\t) *WorkflowService_CountWorkflowExecutions_Args {\n\t\treturn &WorkflowService_CountWorkflowExecutions_Args{\n\t\t\tCountRequest: countRequest,\n\t\t}\n\t}\n\n\tWorkflowService_CountWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_CountWorkflowExecutions_Helper.WrapResponse = func(success *shared.CountWorkflowExecutionsResponse, err error) (*WorkflowService_CountWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_CountWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_CountWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_CountWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_CountWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_CountWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_CountWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_CountWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_CountWorkflowExecutions_Result) (success *shared.CountWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_CountWorkflowExecutions_Result represents the result of a WorkflowService.CountWorkflowExecutions function call.\n//\n// The result of a CountWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_CountWorkflowExecutions_Result struct {\n\t// Value returned by CountWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.CountWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                 `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError            `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError  `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError               `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_CountWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_CountWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CountWorkflowExecutionsResponse_Read(w wire.Value) (*shared.CountWorkflowExecutionsResponse, error) {\n\tvar v shared.CountWorkflowExecutionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _BadRequestError_Read(w wire.Value) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Read(w wire.Value) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Read(w wire.Value) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Read(w wire.Value) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _AccessDeniedError_Read(w wire.Value) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_CountWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_CountWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_CountWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _CountWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_CountWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_CountWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_CountWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_CountWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CountWorkflowExecutionsResponse_Decode(sr stream.Reader) (*shared.CountWorkflowExecutionsResponse, error) {\n\tvar v shared.CountWorkflowExecutionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _BadRequestError_Decode(sr stream.Reader) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Decode(sr stream.Reader) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Decode(sr stream.Reader) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Decode(sr stream.Reader) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _AccessDeniedError_Decode(sr stream.Reader) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_CountWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_CountWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _CountWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_CountWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_CountWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_CountWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_CountWorkflowExecutions_Result match the\n// provided WorkflowService_CountWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) Equals(rhs *WorkflowService_CountWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_CountWorkflowExecutions_Result.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetSuccess() (o *shared.CountWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"CountWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) MethodName() string {\n\treturn \"CountWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_CountWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DeleteDomain_Args represents the arguments for the WorkflowService.DeleteDomain function.\n//\n// The arguments for DeleteDomain are sent and received over the wire as this struct.\ntype WorkflowService_DeleteDomain_Args struct {\n\tDeleteRequest *shared.DeleteDomainRequest `json:\"deleteRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DeleteDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DeleteDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DeleteRequest != nil {\n\t\tw, err = v.DeleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DeleteDomainRequest_Read(w wire.Value) (*shared.DeleteDomainRequest, error) {\n\tvar v shared.DeleteDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DeleteDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DeleteDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DeleteDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DeleteDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DeleteRequest, err = _DeleteDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DeleteDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DeleteDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_DeleteDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DeleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DeleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DeleteDomainRequest_Decode(sr stream.Reader) (*shared.DeleteDomainRequest, error) {\n\tvar v shared.DeleteDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DeleteDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DeleteDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DeleteDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DeleteRequest, err = _DeleteDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DeleteDomain_Args\n// struct.\nfunc (v *WorkflowService_DeleteDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DeleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DeleteRequest: %v\", v.DeleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DeleteDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DeleteDomain_Args match the\n// provided WorkflowService_DeleteDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DeleteDomain_Args) Equals(rhs *WorkflowService_DeleteDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DeleteRequest == nil && rhs.DeleteRequest == nil) || (v.DeleteRequest != nil && rhs.DeleteRequest != nil && v.DeleteRequest.Equals(rhs.DeleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DeleteDomain_Args.\nfunc (v *WorkflowService_DeleteDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DeleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"deleteRequest\", v.DeleteRequest))\n\t}\n\treturn err\n}\n\n// GetDeleteRequest returns the value of DeleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeleteDomain_Args) GetDeleteRequest() (o *shared.DeleteDomainRequest) {\n\tif v != nil && v.DeleteRequest != nil {\n\t\treturn v.DeleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetDeleteRequest returns true if DeleteRequest is not nil.\nfunc (v *WorkflowService_DeleteDomain_Args) IsSetDeleteRequest() bool {\n\treturn v != nil && v.DeleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DeleteDomain\" for this struct.\nfunc (v *WorkflowService_DeleteDomain_Args) MethodName() string {\n\treturn \"DeleteDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DeleteDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DeleteDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DeleteDomain\n// function.\nvar WorkflowService_DeleteDomain_Helper = struct {\n\t// Args accepts the parameters of DeleteDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdeleteRequest *shared.DeleteDomainRequest,\n\t) *WorkflowService_DeleteDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DeleteDomain.\n\t//\n\t// An error can be thrown by DeleteDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DeleteDomain\n\t// given the error returned by it. The provided error may\n\t// be nil if DeleteDomain did not fail.\n\t//\n\t// This allows mapping errors returned by DeleteDomain into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// DeleteDomain\n\t//\n\t//   err := DeleteDomain(args)\n\t//   result, err := WorkflowService_DeleteDomain_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DeleteDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_DeleteDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for DeleteDomain\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if DeleteDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_DeleteDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DeleteDomain_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_DeleteDomain_Helper.Args = func(\n\t\tdeleteRequest *shared.DeleteDomainRequest,\n\t) *WorkflowService_DeleteDomain_Args {\n\t\treturn &WorkflowService_DeleteDomain_Args{\n\t\t\tDeleteRequest: deleteRequest,\n\t\t}\n\t}\n\n\tWorkflowService_DeleteDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DeleteDomain_Helper.WrapResponse = func(err error) (*WorkflowService_DeleteDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DeleteDomain_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeleteDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeleteDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeleteDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeleteDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeleteDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeleteDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeleteDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeleteDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DeleteDomain_Helper.UnwrapResponse = func(result *WorkflowService_DeleteDomain_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DeleteDomain_Result represents the result of a WorkflowService.DeleteDomain function call.\n//\n// The result of a DeleteDomain execution is sent and received over the wire as this struct.\ntype WorkflowService_DeleteDomain_Result struct {\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DeleteDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DeleteDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DeleteDomain_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_DeleteDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DeleteDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DeleteDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DeleteDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeleteDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DeleteDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DeleteDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_DeleteDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeleteDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_DeleteDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DeleteDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DeleteDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeleteDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DeleteDomain_Result\n// struct.\nfunc (v *WorkflowService_DeleteDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DeleteDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DeleteDomain_Result match the\n// provided WorkflowService_DeleteDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DeleteDomain_Result) Equals(rhs *WorkflowService_DeleteDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DeleteDomain_Result.\nfunc (v *WorkflowService_DeleteDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeleteDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_DeleteDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeleteDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DeleteDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeleteDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DeleteDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeleteDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DeleteDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DeleteDomain\" for this struct.\nfunc (v *WorkflowService_DeleteDomain_Result) MethodName() string {\n\treturn \"DeleteDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DeleteDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DeprecateDomain_Args represents the arguments for the WorkflowService.DeprecateDomain function.\n//\n// The arguments for DeprecateDomain are sent and received over the wire as this struct.\ntype WorkflowService_DeprecateDomain_Args struct {\n\tDeprecateRequest *shared.DeprecateDomainRequest `json:\"deprecateRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DeprecateDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DeprecateDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DeprecateRequest != nil {\n\t\tw, err = v.DeprecateRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DeprecateDomainRequest_Read(w wire.Value) (*shared.DeprecateDomainRequest, error) {\n\tvar v shared.DeprecateDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DeprecateDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DeprecateDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DeprecateDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DeprecateDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DeprecateRequest, err = _DeprecateDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DeprecateDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DeprecateDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_DeprecateDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DeprecateRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DeprecateRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DeprecateDomainRequest_Decode(sr stream.Reader) (*shared.DeprecateDomainRequest, error) {\n\tvar v shared.DeprecateDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DeprecateDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DeprecateDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DeprecateDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DeprecateRequest, err = _DeprecateDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DeprecateDomain_Args\n// struct.\nfunc (v *WorkflowService_DeprecateDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DeprecateRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DeprecateRequest: %v\", v.DeprecateRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DeprecateDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DeprecateDomain_Args match the\n// provided WorkflowService_DeprecateDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DeprecateDomain_Args) Equals(rhs *WorkflowService_DeprecateDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DeprecateRequest == nil && rhs.DeprecateRequest == nil) || (v.DeprecateRequest != nil && rhs.DeprecateRequest != nil && v.DeprecateRequest.Equals(rhs.DeprecateRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DeprecateDomain_Args.\nfunc (v *WorkflowService_DeprecateDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DeprecateRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"deprecateRequest\", v.DeprecateRequest))\n\t}\n\treturn err\n}\n\n// GetDeprecateRequest returns the value of DeprecateRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Args) GetDeprecateRequest() (o *shared.DeprecateDomainRequest) {\n\tif v != nil && v.DeprecateRequest != nil {\n\t\treturn v.DeprecateRequest\n\t}\n\n\treturn\n}\n\n// IsSetDeprecateRequest returns true if DeprecateRequest is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Args) IsSetDeprecateRequest() bool {\n\treturn v != nil && v.DeprecateRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DeprecateDomain\" for this struct.\nfunc (v *WorkflowService_DeprecateDomain_Args) MethodName() string {\n\treturn \"DeprecateDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DeprecateDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DeprecateDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DeprecateDomain\n// function.\nvar WorkflowService_DeprecateDomain_Helper = struct {\n\t// Args accepts the parameters of DeprecateDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdeprecateRequest *shared.DeprecateDomainRequest,\n\t) *WorkflowService_DeprecateDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DeprecateDomain.\n\t//\n\t// An error can be thrown by DeprecateDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DeprecateDomain\n\t// given the error returned by it. The provided error may\n\t// be nil if DeprecateDomain did not fail.\n\t//\n\t// This allows mapping errors returned by DeprecateDomain into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// DeprecateDomain\n\t//\n\t//   err := DeprecateDomain(args)\n\t//   result, err := WorkflowService_DeprecateDomain_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DeprecateDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_DeprecateDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for DeprecateDomain\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if DeprecateDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_DeprecateDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DeprecateDomain_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_DeprecateDomain_Helper.Args = func(\n\t\tdeprecateRequest *shared.DeprecateDomainRequest,\n\t) *WorkflowService_DeprecateDomain_Args {\n\t\treturn &WorkflowService_DeprecateDomain_Args{\n\t\t\tDeprecateRequest: deprecateRequest,\n\t\t}\n\t}\n\n\tWorkflowService_DeprecateDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DeprecateDomain_Helper.WrapResponse = func(err error) (*WorkflowService_DeprecateDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DeprecateDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DeprecateDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DeprecateDomain_Helper.UnwrapResponse = func(result *WorkflowService_DeprecateDomain_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DeprecateDomain_Result represents the result of a WorkflowService.DeprecateDomain function call.\n//\n// The result of a DeprecateDomain execution is sent and received over the wire as this struct.\ntype WorkflowService_DeprecateDomain_Result struct {\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DeprecateDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DeprecateDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DeprecateDomain_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainNotActiveError_Read(w wire.Value) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DeprecateDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DeprecateDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DeprecateDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DeprecateDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeprecateDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DeprecateDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DeprecateDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_DeprecateDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeprecateDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainNotActiveError_Decode(sr stream.Reader) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DeprecateDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DeprecateDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DeprecateDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DeprecateDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DeprecateDomain_Result\n// struct.\nfunc (v *WorkflowService_DeprecateDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DeprecateDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DeprecateDomain_Result match the\n// provided WorkflowService_DeprecateDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DeprecateDomain_Result) Equals(rhs *WorkflowService_DeprecateDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DeprecateDomain_Result.\nfunc (v *WorkflowService_DeprecateDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DeprecateDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DeprecateDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DeprecateDomain\" for this struct.\nfunc (v *WorkflowService_DeprecateDomain_Result) MethodName() string {\n\treturn \"DeprecateDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DeprecateDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DescribeDomain_Args represents the arguments for the WorkflowService.DescribeDomain function.\n//\n// The arguments for DescribeDomain are sent and received over the wire as this struct.\ntype WorkflowService_DescribeDomain_Args struct {\n\tDescribeRequest *shared.DescribeDomainRequest `json:\"describeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DescribeRequest != nil {\n\t\tw, err = v.DescribeRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeDomainRequest_Read(w wire.Value) (*shared.DescribeDomainRequest, error) {\n\tvar v shared.DescribeDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DescribeRequest, err = _DescribeDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_DescribeDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DescribeRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DescribeRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeDomainRequest_Decode(sr stream.Reader) (*shared.DescribeDomainRequest, error) {\n\tvar v shared.DescribeDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DescribeRequest, err = _DescribeDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeDomain_Args\n// struct.\nfunc (v *WorkflowService_DescribeDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DescribeRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DescribeRequest: %v\", v.DescribeRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeDomain_Args match the\n// provided WorkflowService_DescribeDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeDomain_Args) Equals(rhs *WorkflowService_DescribeDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DescribeRequest == nil && rhs.DescribeRequest == nil) || (v.DescribeRequest != nil && rhs.DescribeRequest != nil && v.DescribeRequest.Equals(rhs.DescribeRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeDomain_Args.\nfunc (v *WorkflowService_DescribeDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DescribeRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"describeRequest\", v.DescribeRequest))\n\t}\n\treturn err\n}\n\n// GetDescribeRequest returns the value of DescribeRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Args) GetDescribeRequest() (o *shared.DescribeDomainRequest) {\n\tif v != nil && v.DescribeRequest != nil {\n\t\treturn v.DescribeRequest\n\t}\n\n\treturn\n}\n\n// IsSetDescribeRequest returns true if DescribeRequest is not nil.\nfunc (v *WorkflowService_DescribeDomain_Args) IsSetDescribeRequest() bool {\n\treturn v != nil && v.DescribeRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeDomain\" for this struct.\nfunc (v *WorkflowService_DescribeDomain_Args) MethodName() string {\n\treturn \"DescribeDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DescribeDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DescribeDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DescribeDomain\n// function.\nvar WorkflowService_DescribeDomain_Helper = struct {\n\t// Args accepts the parameters of DescribeDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdescribeRequest *shared.DescribeDomainRequest,\n\t) *WorkflowService_DescribeDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeDomain.\n\t//\n\t// An error can be thrown by DescribeDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeDomain\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeDomain into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeDomain\n\t//\n\t//   value, err := DescribeDomain(args)\n\t//   result, err := WorkflowService_DescribeDomain_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeDomainResponse, error) (*WorkflowService_DescribeDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeDomain\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_DescribeDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DescribeDomain_Result) (*shared.DescribeDomainResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_DescribeDomain_Helper.Args = func(\n\t\tdescribeRequest *shared.DescribeDomainRequest,\n\t) *WorkflowService_DescribeDomain_Args {\n\t\treturn &WorkflowService_DescribeDomain_Args{\n\t\t\tDescribeRequest: describeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_DescribeDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DescribeDomain_Helper.WrapResponse = func(success *shared.DescribeDomainResponse, err error) (*WorkflowService_DescribeDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DescribeDomain_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DescribeDomain_Helper.UnwrapResponse = func(result *WorkflowService_DescribeDomain_Result) (success *shared.DescribeDomainResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DescribeDomain_Result represents the result of a WorkflowService.DescribeDomain function call.\n//\n// The result of a DescribeDomain execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_DescribeDomain_Result struct {\n\t// Value returned by DescribeDomain after a successful execution.\n\tSuccess                        *shared.DescribeDomainResponse         `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DescribeDomain_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeDomainResponse_Read(w wire.Value) (*shared.DescribeDomainResponse, error) {\n\tvar v shared.DescribeDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeDomainResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_DescribeDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeDomainResponse_Decode(sr stream.Reader) (*shared.DescribeDomainResponse, error) {\n\tvar v shared.DescribeDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeDomain_Result\n// struct.\nfunc (v *WorkflowService_DescribeDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeDomain_Result match the\n// provided WorkflowService_DescribeDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeDomain_Result) Equals(rhs *WorkflowService_DescribeDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeDomain_Result.\nfunc (v *WorkflowService_DescribeDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetSuccess() (o *shared.DescribeDomainResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DescribeDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeDomain\" for this struct.\nfunc (v *WorkflowService_DescribeDomain_Result) MethodName() string {\n\treturn \"DescribeDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DescribeDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DescribeTaskList_Args represents the arguments for the WorkflowService.DescribeTaskList function.\n//\n// The arguments for DescribeTaskList are sent and received over the wire as this struct.\ntype WorkflowService_DescribeTaskList_Args struct {\n\tRequest *shared.DescribeTaskListRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeTaskList_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeTaskList_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListRequest_Read(w wire.Value) (*shared.DescribeTaskListRequest, error) {\n\tvar v shared.DescribeTaskListRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeTaskList_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeTaskList_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeTaskList_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeTaskList_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeTaskListRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeTaskList_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeTaskList_Args struct could not be encoded.\nfunc (v *WorkflowService_DescribeTaskList_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListRequest_Decode(sr stream.Reader) (*shared.DescribeTaskListRequest, error) {\n\tvar v shared.DescribeTaskListRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeTaskList_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeTaskList_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeTaskList_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeTaskListRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeTaskList_Args\n// struct.\nfunc (v *WorkflowService_DescribeTaskList_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeTaskList_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeTaskList_Args match the\n// provided WorkflowService_DescribeTaskList_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeTaskList_Args) Equals(rhs *WorkflowService_DescribeTaskList_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeTaskList_Args.\nfunc (v *WorkflowService_DescribeTaskList_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Args) GetRequest() (o *shared.DescribeTaskListRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeTaskList\" for this struct.\nfunc (v *WorkflowService_DescribeTaskList_Args) MethodName() string {\n\treturn \"DescribeTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DescribeTaskList_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DescribeTaskList_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DescribeTaskList\n// function.\nvar WorkflowService_DescribeTaskList_Helper = struct {\n\t// Args accepts the parameters of DescribeTaskList in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeTaskListRequest,\n\t) *WorkflowService_DescribeTaskList_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeTaskList.\n\t//\n\t// An error can be thrown by DescribeTaskList only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeTaskList\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeTaskList into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeTaskList\n\t//\n\t//   value, err := DescribeTaskList(args)\n\t//   result, err := WorkflowService_DescribeTaskList_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeTaskList: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeTaskListResponse, error) (*WorkflowService_DescribeTaskList_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeTaskList\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeTaskList threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_DescribeTaskList_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DescribeTaskList_Result) (*shared.DescribeTaskListResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_DescribeTaskList_Helper.Args = func(\n\t\trequest *shared.DescribeTaskListRequest,\n\t) *WorkflowService_DescribeTaskList_Args {\n\t\treturn &WorkflowService_DescribeTaskList_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tWorkflowService_DescribeTaskList_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DescribeTaskList_Helper.WrapResponse = func(success *shared.DescribeTaskListResponse, err error) (*WorkflowService_DescribeTaskList_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeTaskList_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeTaskList_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DescribeTaskList_Helper.UnwrapResponse = func(result *WorkflowService_DescribeTaskList_Result) (success *shared.DescribeTaskListResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DescribeTaskList_Result represents the result of a WorkflowService.DescribeTaskList function call.\n//\n// The result of a DescribeTaskList execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_DescribeTaskList_Result struct {\n\t// Value returned by DescribeTaskList after a successful execution.\n\tSuccess                        *shared.DescribeTaskListResponse       `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeTaskList_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeTaskList_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DescribeTaskList_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListResponse_Read(w wire.Value) (*shared.DescribeTaskListResponse, error) {\n\tvar v shared.DescribeTaskListResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Read(w wire.Value) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeTaskList_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeTaskList_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeTaskList_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeTaskList_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeTaskListResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeTaskList_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeTaskList_Result struct could not be encoded.\nfunc (v *WorkflowService_DescribeTaskList_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListResponse_Decode(sr stream.Reader) (*shared.DescribeTaskListResponse, error) {\n\tvar v shared.DescribeTaskListResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Decode(sr stream.Reader) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeTaskList_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeTaskList_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeTaskList_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeTaskList_Result\n// struct.\nfunc (v *WorkflowService_DescribeTaskList_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeTaskList_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeTaskList_Result match the\n// provided WorkflowService_DescribeTaskList_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeTaskList_Result) Equals(rhs *WorkflowService_DescribeTaskList_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeTaskList_Result.\nfunc (v *WorkflowService_DescribeTaskList_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetSuccess() (o *shared.DescribeTaskListResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeTaskList_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DescribeTaskList_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeTaskList\" for this struct.\nfunc (v *WorkflowService_DescribeTaskList_Result) MethodName() string {\n\treturn \"DescribeTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DescribeTaskList_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DescribeWorkflowExecution_Args represents the arguments for the WorkflowService.DescribeWorkflowExecution function.\n//\n// The arguments for DescribeWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_DescribeWorkflowExecution_Args struct {\n\tDescribeRequest *shared.DescribeWorkflowExecutionRequest `json:\"describeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DescribeRequest != nil {\n\t\tw, err = v.DescribeRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionRequest_Read(w wire.Value) (*shared.DescribeWorkflowExecutionRequest, error) {\n\tvar v shared.DescribeWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DescribeRequest, err = _DescribeWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DescribeRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DescribeRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.DescribeWorkflowExecutionRequest, error) {\n\tvar v shared.DescribeWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DescribeRequest, err = _DescribeWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DescribeRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DescribeRequest: %v\", v.DescribeRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeWorkflowExecution_Args match the\n// provided WorkflowService_DescribeWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) Equals(rhs *WorkflowService_DescribeWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DescribeRequest == nil && rhs.DescribeRequest == nil) || (v.DescribeRequest != nil && rhs.DescribeRequest != nil && v.DescribeRequest.Equals(rhs.DescribeRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeWorkflowExecution_Args.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DescribeRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"describeRequest\", v.DescribeRequest))\n\t}\n\treturn err\n}\n\n// GetDescribeRequest returns the value of DescribeRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) GetDescribeRequest() (o *shared.DescribeWorkflowExecutionRequest) {\n\tif v != nil && v.DescribeRequest != nil {\n\t\treturn v.DescribeRequest\n\t}\n\n\treturn\n}\n\n// IsSetDescribeRequest returns true if DescribeRequest is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) IsSetDescribeRequest() bool {\n\treturn v != nil && v.DescribeRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DescribeWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DescribeWorkflowExecution\n// function.\nvar WorkflowService_DescribeWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of DescribeWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\t) *WorkflowService_DescribeWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeWorkflowExecution.\n\t//\n\t// An error can be thrown by DescribeWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeWorkflowExecution\n\t//\n\t//   value, err := DescribeWorkflowExecution(args)\n\t//   result, err := WorkflowService_DescribeWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeWorkflowExecutionResponse, error) (*WorkflowService_DescribeWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_DescribeWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DescribeWorkflowExecution_Result) (*shared.DescribeWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_DescribeWorkflowExecution_Helper.Args = func(\n\t\tdescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\t) *WorkflowService_DescribeWorkflowExecution_Args {\n\t\treturn &WorkflowService_DescribeWorkflowExecution_Args{\n\t\t\tDescribeRequest: describeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_DescribeWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DescribeWorkflowExecution_Helper.WrapResponse = func(success *shared.DescribeWorkflowExecutionResponse, err error) (*WorkflowService_DescribeWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DescribeWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DescribeWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DescribeWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_DescribeWorkflowExecution_Result) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DescribeWorkflowExecution_Result represents the result of a WorkflowService.DescribeWorkflowExecution function call.\n//\n// The result of a DescribeWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_DescribeWorkflowExecution_Result struct {\n\t// Value returned by DescribeWorkflowExecution after a successful execution.\n\tSuccess                        *shared.DescribeWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                   `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError              `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                  `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError    `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                 `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DescribeWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionResponse_Read(w wire.Value) (*shared.DescribeWorkflowExecutionResponse, error) {\n\tvar v shared.DescribeWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DescribeWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DescribeWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DescribeWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DescribeWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DescribeWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.DescribeWorkflowExecutionResponse, error) {\n\tvar v shared.DescribeWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DescribeWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DescribeWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DescribeWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DescribeWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DescribeWorkflowExecution_Result match the\n// provided WorkflowService_DescribeWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) Equals(rhs *WorkflowService_DescribeWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DescribeWorkflowExecution_Result.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetSuccess() (o *shared.DescribeWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DescribeWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_DiagnoseWorkflowExecution_Args represents the arguments for the WorkflowService.DiagnoseWorkflowExecution function.\n//\n// The arguments for DiagnoseWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_DiagnoseWorkflowExecution_Args struct {\n\tDiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest `json:\"diagnoseRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DiagnoseWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DiagnoseRequest != nil {\n\t\tw, err = v.DiagnoseRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DiagnoseWorkflowExecutionRequest_Read(w wire.Value) (*shared.DiagnoseWorkflowExecutionRequest, error) {\n\tvar v shared.DiagnoseWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DiagnoseWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DiagnoseWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DiagnoseWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DiagnoseRequest, err = _DiagnoseWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DiagnoseWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DiagnoseWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DiagnoseRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DiagnoseRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DiagnoseWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.DiagnoseWorkflowExecutionRequest, error) {\n\tvar v shared.DiagnoseWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DiagnoseWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DiagnoseWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DiagnoseRequest, err = _DiagnoseWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DiagnoseWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DiagnoseRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DiagnoseRequest: %v\", v.DiagnoseRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DiagnoseWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DiagnoseWorkflowExecution_Args match the\n// provided WorkflowService_DiagnoseWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) Equals(rhs *WorkflowService_DiagnoseWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DiagnoseRequest == nil && rhs.DiagnoseRequest == nil) || (v.DiagnoseRequest != nil && rhs.DiagnoseRequest != nil && v.DiagnoseRequest.Equals(rhs.DiagnoseRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DiagnoseWorkflowExecution_Args.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DiagnoseRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"diagnoseRequest\", v.DiagnoseRequest))\n\t}\n\treturn err\n}\n\n// GetDiagnoseRequest returns the value of DiagnoseRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) GetDiagnoseRequest() (o *shared.DiagnoseWorkflowExecutionRequest) {\n\tif v != nil && v.DiagnoseRequest != nil {\n\t\treturn v.DiagnoseRequest\n\t}\n\n\treturn\n}\n\n// IsSetDiagnoseRequest returns true if DiagnoseRequest is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) IsSetDiagnoseRequest() bool {\n\treturn v != nil && v.DiagnoseRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DiagnoseWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) MethodName() string {\n\treturn \"DiagnoseWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_DiagnoseWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.DiagnoseWorkflowExecution\n// function.\nvar WorkflowService_DiagnoseWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of DiagnoseWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\t) *WorkflowService_DiagnoseWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DiagnoseWorkflowExecution.\n\t//\n\t// An error can be thrown by DiagnoseWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DiagnoseWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DiagnoseWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DiagnoseWorkflowExecution\n\t//\n\t//   value, err := DiagnoseWorkflowExecution(args)\n\t//   result, err := WorkflowService_DiagnoseWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DiagnoseWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DiagnoseWorkflowExecutionResponse, error) (*WorkflowService_DiagnoseWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for DiagnoseWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DiagnoseWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_DiagnoseWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_DiagnoseWorkflowExecution_Result) (*shared.DiagnoseWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_DiagnoseWorkflowExecution_Helper.Args = func(\n\t\tdiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\t) *WorkflowService_DiagnoseWorkflowExecution_Args {\n\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Args{\n\t\t\tDiagnoseRequest: diagnoseRequest,\n\t\t}\n\t}\n\n\tWorkflowService_DiagnoseWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_DiagnoseWorkflowExecution_Helper.WrapResponse = func(success *shared.DiagnoseWorkflowExecutionResponse, err error) (*WorkflowService_DiagnoseWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DiagnoseWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DiagnoseWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DiagnoseWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DiagnoseWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_DiagnoseWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_DiagnoseWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_DiagnoseWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_DiagnoseWorkflowExecution_Result) (success *shared.DiagnoseWorkflowExecutionResponse, err error) {\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_DiagnoseWorkflowExecution_Result represents the result of a WorkflowService.DiagnoseWorkflowExecution function call.\n//\n// The result of a DiagnoseWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_DiagnoseWorkflowExecution_Result struct {\n\t// Value returned by DiagnoseWorkflowExecution after a successful execution.\n\tSuccess                        *shared.DiagnoseWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError              `json:\"domainNotActiveError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                  `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError              `json:\"entityNotExistError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError    `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                 `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_DiagnoseWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_DiagnoseWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DiagnoseWorkflowExecutionResponse_Read(w wire.Value) (*shared.DiagnoseWorkflowExecutionResponse, error) {\n\tvar v shared.DiagnoseWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_DiagnoseWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_DiagnoseWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_DiagnoseWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DiagnoseWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DiagnoseWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_DiagnoseWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_DiagnoseWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DiagnoseWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DiagnoseWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.DiagnoseWorkflowExecutionResponse, error) {\n\tvar v shared.DiagnoseWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_DiagnoseWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_DiagnoseWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DiagnoseWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_DiagnoseWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_DiagnoseWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_DiagnoseWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_DiagnoseWorkflowExecution_Result match the\n// provided WorkflowService_DiagnoseWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) Equals(rhs *WorkflowService_DiagnoseWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_DiagnoseWorkflowExecution_Result.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetSuccess() (o *shared.DiagnoseWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DiagnoseWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) MethodName() string {\n\treturn \"DiagnoseWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_DiagnoseWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_FailoverDomain_Args represents the arguments for the WorkflowService.FailoverDomain function.\n//\n// The arguments for FailoverDomain are sent and received over the wire as this struct.\ntype WorkflowService_FailoverDomain_Args struct {\n\tFailoverRequest *shared.FailoverDomainRequest `json:\"failoverRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_FailoverDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_FailoverDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailoverRequest != nil {\n\t\tw, err = v.FailoverRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverDomainRequest_Read(w wire.Value) (*shared.FailoverDomainRequest, error) {\n\tvar v shared.FailoverDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_FailoverDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_FailoverDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_FailoverDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_FailoverDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailoverRequest, err = _FailoverDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_FailoverDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_FailoverDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_FailoverDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailoverRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailoverRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverDomainRequest_Decode(sr stream.Reader) (*shared.FailoverDomainRequest, error) {\n\tvar v shared.FailoverDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_FailoverDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_FailoverDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_FailoverDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailoverRequest, err = _FailoverDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_FailoverDomain_Args\n// struct.\nfunc (v *WorkflowService_FailoverDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailoverRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverRequest: %v\", v.FailoverRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_FailoverDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_FailoverDomain_Args match the\n// provided WorkflowService_FailoverDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_FailoverDomain_Args) Equals(rhs *WorkflowService_FailoverDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailoverRequest == nil && rhs.FailoverRequest == nil) || (v.FailoverRequest != nil && rhs.FailoverRequest != nil && v.FailoverRequest.Equals(rhs.FailoverRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_FailoverDomain_Args.\nfunc (v *WorkflowService_FailoverDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailoverRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failoverRequest\", v.FailoverRequest))\n\t}\n\treturn err\n}\n\n// GetFailoverRequest returns the value of FailoverRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Args) GetFailoverRequest() (o *shared.FailoverDomainRequest) {\n\tif v != nil && v.FailoverRequest != nil {\n\t\treturn v.FailoverRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailoverRequest returns true if FailoverRequest is not nil.\nfunc (v *WorkflowService_FailoverDomain_Args) IsSetFailoverRequest() bool {\n\treturn v != nil && v.FailoverRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"FailoverDomain\" for this struct.\nfunc (v *WorkflowService_FailoverDomain_Args) MethodName() string {\n\treturn \"FailoverDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_FailoverDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_FailoverDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.FailoverDomain\n// function.\nvar WorkflowService_FailoverDomain_Helper = struct {\n\t// Args accepts the parameters of FailoverDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailoverRequest *shared.FailoverDomainRequest,\n\t) *WorkflowService_FailoverDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by FailoverDomain.\n\t//\n\t// An error can be thrown by FailoverDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for FailoverDomain\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// FailoverDomain into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by FailoverDomain\n\t//\n\t//   value, err := FailoverDomain(args)\n\t//   result, err := WorkflowService_FailoverDomain_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from FailoverDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.FailoverDomainResponse, error) (*WorkflowService_FailoverDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for FailoverDomain\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if FailoverDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_FailoverDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_FailoverDomain_Result) (*shared.FailoverDomainResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_FailoverDomain_Helper.Args = func(\n\t\tfailoverRequest *shared.FailoverDomainRequest,\n\t) *WorkflowService_FailoverDomain_Args {\n\t\treturn &WorkflowService_FailoverDomain_Args{\n\t\t\tFailoverRequest: failoverRequest,\n\t\t}\n\t}\n\n\tWorkflowService_FailoverDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_FailoverDomain_Helper.WrapResponse = func(success *shared.FailoverDomainResponse, err error) (*WorkflowService_FailoverDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_FailoverDomain_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_FailoverDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_FailoverDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_FailoverDomain_Helper.UnwrapResponse = func(result *WorkflowService_FailoverDomain_Result) (success *shared.FailoverDomainResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_FailoverDomain_Result represents the result of a WorkflowService.FailoverDomain function call.\n//\n// The result of a FailoverDomain execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_FailoverDomain_Result struct {\n\t// Value returned by FailoverDomain after a successful execution.\n\tSuccess                        *shared.FailoverDomainResponse         `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_FailoverDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_FailoverDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_FailoverDomain_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverDomainResponse_Read(w wire.Value) (*shared.FailoverDomainResponse, error) {\n\tvar v shared.FailoverDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_FailoverDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_FailoverDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_FailoverDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_FailoverDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _FailoverDomainResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_FailoverDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_FailoverDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_FailoverDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_FailoverDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_FailoverDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverDomainResponse_Decode(sr stream.Reader) (*shared.FailoverDomainResponse, error) {\n\tvar v shared.FailoverDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_FailoverDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_FailoverDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_FailoverDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _FailoverDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_FailoverDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_FailoverDomain_Result\n// struct.\nfunc (v *WorkflowService_FailoverDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_FailoverDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_FailoverDomain_Result match the\n// provided WorkflowService_FailoverDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_FailoverDomain_Result) Equals(rhs *WorkflowService_FailoverDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_FailoverDomain_Result.\nfunc (v *WorkflowService_FailoverDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetSuccess() (o *shared.FailoverDomainResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_FailoverDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_FailoverDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"FailoverDomain\" for this struct.\nfunc (v *WorkflowService_FailoverDomain_Result) MethodName() string {\n\treturn \"FailoverDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_FailoverDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_GetClusterInfo_Args represents the arguments for the WorkflowService.GetClusterInfo function.\n//\n// The arguments for GetClusterInfo are sent and received over the wire as this struct.\ntype WorkflowService_GetClusterInfo_Args struct {\n}\n\n// ToWire translates a WorkflowService_GetClusterInfo_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetClusterInfo_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_GetClusterInfo_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetClusterInfo_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetClusterInfo_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetClusterInfo_Args) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetClusterInfo_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetClusterInfo_Args struct could not be encoded.\nfunc (v *WorkflowService_GetClusterInfo_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_GetClusterInfo_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetClusterInfo_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetClusterInfo_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetClusterInfo_Args\n// struct.\nfunc (v *WorkflowService_GetClusterInfo_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"WorkflowService_GetClusterInfo_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetClusterInfo_Args match the\n// provided WorkflowService_GetClusterInfo_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetClusterInfo_Args) Equals(rhs *WorkflowService_GetClusterInfo_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetClusterInfo_Args.\nfunc (v *WorkflowService_GetClusterInfo_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetClusterInfo\" for this struct.\nfunc (v *WorkflowService_GetClusterInfo_Args) MethodName() string {\n\treturn \"GetClusterInfo\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_GetClusterInfo_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_GetClusterInfo_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.GetClusterInfo\n// function.\nvar WorkflowService_GetClusterInfo_Helper = struct {\n\t// Args accepts the parameters of GetClusterInfo in-order and returns\n\t// the arguments struct for the function.\n\tArgs func() *WorkflowService_GetClusterInfo_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetClusterInfo.\n\t//\n\t// An error can be thrown by GetClusterInfo only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetClusterInfo\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetClusterInfo into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetClusterInfo\n\t//\n\t//   value, err := GetClusterInfo(args)\n\t//   result, err := WorkflowService_GetClusterInfo_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetClusterInfo: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ClusterInfo, error) (*WorkflowService_GetClusterInfo_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetClusterInfo\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetClusterInfo threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_GetClusterInfo_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_GetClusterInfo_Result) (*shared.ClusterInfo, error)\n}{}\n\nfunc init() {\n\tWorkflowService_GetClusterInfo_Helper.Args = func() *WorkflowService_GetClusterInfo_Args {\n\t\treturn &WorkflowService_GetClusterInfo_Args{}\n\t}\n\n\tWorkflowService_GetClusterInfo_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_GetClusterInfo_Helper.WrapResponse = func(success *shared.ClusterInfo, err error) (*WorkflowService_GetClusterInfo_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_GetClusterInfo_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetClusterInfo_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetClusterInfo_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetClusterInfo_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetClusterInfo_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetClusterInfo_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetClusterInfo_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_GetClusterInfo_Helper.UnwrapResponse = func(result *WorkflowService_GetClusterInfo_Result) (success *shared.ClusterInfo, err error) {\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_GetClusterInfo_Result represents the result of a WorkflowService.GetClusterInfo function call.\n//\n// The result of a GetClusterInfo execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_GetClusterInfo_Result struct {\n\t// Value returned by GetClusterInfo after a successful execution.\n\tSuccess              *shared.ClusterInfo          `json:\"success,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetClusterInfo_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetClusterInfo_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_GetClusterInfo_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ClusterInfo_Read(w wire.Value) (*shared.ClusterInfo, error) {\n\tvar v shared.ClusterInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Read(w wire.Value) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetClusterInfo_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetClusterInfo_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetClusterInfo_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetClusterInfo_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ClusterInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetClusterInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetClusterInfo_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetClusterInfo_Result struct could not be encoded.\nfunc (v *WorkflowService_GetClusterInfo_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetClusterInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ClusterInfo_Decode(sr stream.Reader) (*shared.ClusterInfo, error) {\n\tvar v shared.ClusterInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Decode(sr stream.Reader) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetClusterInfo_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetClusterInfo_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetClusterInfo_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ClusterInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetClusterInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetClusterInfo_Result\n// struct.\nfunc (v *WorkflowService_GetClusterInfo_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetClusterInfo_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetClusterInfo_Result match the\n// provided WorkflowService_GetClusterInfo_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetClusterInfo_Result) Equals(rhs *WorkflowService_GetClusterInfo_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetClusterInfo_Result.\nfunc (v *WorkflowService_GetClusterInfo_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetClusterInfo_Result) GetSuccess() (o *shared.ClusterInfo) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_GetClusterInfo_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetClusterInfo_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *WorkflowService_GetClusterInfo_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetClusterInfo_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_GetClusterInfo_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetClusterInfo_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_GetClusterInfo_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetClusterInfo\" for this struct.\nfunc (v *WorkflowService_GetClusterInfo_Result) MethodName() string {\n\treturn \"GetClusterInfo\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_GetClusterInfo_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_GetSearchAttributes_Args represents the arguments for the WorkflowService.GetSearchAttributes function.\n//\n// The arguments for GetSearchAttributes are sent and received over the wire as this struct.\ntype WorkflowService_GetSearchAttributes_Args struct {\n}\n\n// ToWire translates a WorkflowService_GetSearchAttributes_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetSearchAttributes_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_GetSearchAttributes_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetSearchAttributes_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetSearchAttributes_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetSearchAttributes_Args) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetSearchAttributes_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetSearchAttributes_Args struct could not be encoded.\nfunc (v *WorkflowService_GetSearchAttributes_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_GetSearchAttributes_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetSearchAttributes_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetSearchAttributes_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetSearchAttributes_Args\n// struct.\nfunc (v *WorkflowService_GetSearchAttributes_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"WorkflowService_GetSearchAttributes_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetSearchAttributes_Args match the\n// provided WorkflowService_GetSearchAttributes_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetSearchAttributes_Args) Equals(rhs *WorkflowService_GetSearchAttributes_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetSearchAttributes_Args.\nfunc (v *WorkflowService_GetSearchAttributes_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetSearchAttributes\" for this struct.\nfunc (v *WorkflowService_GetSearchAttributes_Args) MethodName() string {\n\treturn \"GetSearchAttributes\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_GetSearchAttributes_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_GetSearchAttributes_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.GetSearchAttributes\n// function.\nvar WorkflowService_GetSearchAttributes_Helper = struct {\n\t// Args accepts the parameters of GetSearchAttributes in-order and returns\n\t// the arguments struct for the function.\n\tArgs func() *WorkflowService_GetSearchAttributes_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetSearchAttributes.\n\t//\n\t// An error can be thrown by GetSearchAttributes only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetSearchAttributes\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetSearchAttributes into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetSearchAttributes\n\t//\n\t//   value, err := GetSearchAttributes(args)\n\t//   result, err := WorkflowService_GetSearchAttributes_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetSearchAttributes: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetSearchAttributesResponse, error) (*WorkflowService_GetSearchAttributes_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetSearchAttributes\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetSearchAttributes threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_GetSearchAttributes_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_GetSearchAttributes_Result) (*shared.GetSearchAttributesResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_GetSearchAttributes_Helper.Args = func() *WorkflowService_GetSearchAttributes_Args {\n\t\treturn &WorkflowService_GetSearchAttributes_Args{}\n\t}\n\n\tWorkflowService_GetSearchAttributes_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_GetSearchAttributes_Helper.WrapResponse = func(success *shared.GetSearchAttributesResponse, err error) (*WorkflowService_GetSearchAttributes_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_GetSearchAttributes_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetSearchAttributes_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetSearchAttributes_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetSearchAttributes_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetSearchAttributes_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetSearchAttributes_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetSearchAttributes_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_GetSearchAttributes_Helper.UnwrapResponse = func(result *WorkflowService_GetSearchAttributes_Result) (success *shared.GetSearchAttributesResponse, err error) {\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_GetSearchAttributes_Result represents the result of a WorkflowService.GetSearchAttributes function call.\n//\n// The result of a GetSearchAttributes execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_GetSearchAttributes_Result struct {\n\t// Value returned by GetSearchAttributes after a successful execution.\n\tSuccess                        *shared.GetSearchAttributesResponse    `json:\"success,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetSearchAttributes_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetSearchAttributes_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_GetSearchAttributes_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetSearchAttributesResponse_Read(w wire.Value) (*shared.GetSearchAttributesResponse, error) {\n\tvar v shared.GetSearchAttributesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetSearchAttributes_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetSearchAttributes_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetSearchAttributes_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetSearchAttributes_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetSearchAttributesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetSearchAttributes_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetSearchAttributes_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetSearchAttributes_Result struct could not be encoded.\nfunc (v *WorkflowService_GetSearchAttributes_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetSearchAttributes_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetSearchAttributesResponse_Decode(sr stream.Reader) (*shared.GetSearchAttributesResponse, error) {\n\tvar v shared.GetSearchAttributesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetSearchAttributes_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetSearchAttributes_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetSearchAttributes_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetSearchAttributesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetSearchAttributes_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetSearchAttributes_Result\n// struct.\nfunc (v *WorkflowService_GetSearchAttributes_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetSearchAttributes_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetSearchAttributes_Result match the\n// provided WorkflowService_GetSearchAttributes_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetSearchAttributes_Result) Equals(rhs *WorkflowService_GetSearchAttributes_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetSearchAttributes_Result.\nfunc (v *WorkflowService_GetSearchAttributes_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetSearchAttributes_Result) GetSuccess() (o *shared.GetSearchAttributesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_GetSearchAttributes_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetSearchAttributes_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_GetSearchAttributes_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetSearchAttributes_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_GetSearchAttributes_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetSearchAttributes_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_GetSearchAttributes_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetSearchAttributes\" for this struct.\nfunc (v *WorkflowService_GetSearchAttributes_Result) MethodName() string {\n\treturn \"GetSearchAttributes\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_GetSearchAttributes_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_GetTaskListsByDomain_Args represents the arguments for the WorkflowService.GetTaskListsByDomain function.\n//\n// The arguments for GetTaskListsByDomain are sent and received over the wire as this struct.\ntype WorkflowService_GetTaskListsByDomain_Args struct {\n\tRequest *shared.GetTaskListsByDomainRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetTaskListsByDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetTaskListsByDomainRequest_Read(w wire.Value) (*shared.GetTaskListsByDomainRequest, error) {\n\tvar v shared.GetTaskListsByDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetTaskListsByDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetTaskListsByDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetTaskListsByDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetTaskListsByDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetTaskListsByDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetTaskListsByDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetTaskListsByDomainRequest_Decode(sr stream.Reader) (*shared.GetTaskListsByDomainRequest, error) {\n\tvar v shared.GetTaskListsByDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetTaskListsByDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetTaskListsByDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetTaskListsByDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetTaskListsByDomain_Args\n// struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetTaskListsByDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetTaskListsByDomain_Args match the\n// provided WorkflowService_GetTaskListsByDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) Equals(rhs *WorkflowService_GetTaskListsByDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetTaskListsByDomain_Args.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) GetRequest() (o *shared.GetTaskListsByDomainRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetTaskListsByDomain\" for this struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) MethodName() string {\n\treturn \"GetTaskListsByDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_GetTaskListsByDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.GetTaskListsByDomain\n// function.\nvar WorkflowService_GetTaskListsByDomain_Helper = struct {\n\t// Args accepts the parameters of GetTaskListsByDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.GetTaskListsByDomainRequest,\n\t) *WorkflowService_GetTaskListsByDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetTaskListsByDomain.\n\t//\n\t// An error can be thrown by GetTaskListsByDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetTaskListsByDomain\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetTaskListsByDomain into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetTaskListsByDomain\n\t//\n\t//   value, err := GetTaskListsByDomain(args)\n\t//   result, err := WorkflowService_GetTaskListsByDomain_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetTaskListsByDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetTaskListsByDomainResponse, error) (*WorkflowService_GetTaskListsByDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetTaskListsByDomain\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetTaskListsByDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_GetTaskListsByDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_GetTaskListsByDomain_Result) (*shared.GetTaskListsByDomainResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_GetTaskListsByDomain_Helper.Args = func(\n\t\trequest *shared.GetTaskListsByDomainRequest,\n\t) *WorkflowService_GetTaskListsByDomain_Args {\n\t\treturn &WorkflowService_GetTaskListsByDomain_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tWorkflowService_GetTaskListsByDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_GetTaskListsByDomain_Helper.WrapResponse = func(success *shared.GetTaskListsByDomainResponse, err error) (*WorkflowService_GetTaskListsByDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetTaskListsByDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetTaskListsByDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_GetTaskListsByDomain_Helper.UnwrapResponse = func(result *WorkflowService_GetTaskListsByDomain_Result) (success *shared.GetTaskListsByDomainResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_GetTaskListsByDomain_Result represents the result of a WorkflowService.GetTaskListsByDomain function call.\n//\n// The result of a GetTaskListsByDomain execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_GetTaskListsByDomain_Result struct {\n\t// Value returned by GetTaskListsByDomain after a successful execution.\n\tSuccess                        *shared.GetTaskListsByDomainResponse   `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetTaskListsByDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetTaskListsByDomainResponse_Read(w wire.Value) (*shared.GetTaskListsByDomainResponse, error) {\n\tvar v shared.GetTaskListsByDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetTaskListsByDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetTaskListsByDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetTaskListsByDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetTaskListsByDomainResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetTaskListsByDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetTaskListsByDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetTaskListsByDomainResponse_Decode(sr stream.Reader) (*shared.GetTaskListsByDomainResponse, error) {\n\tvar v shared.GetTaskListsByDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetTaskListsByDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetTaskListsByDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetTaskListsByDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetTaskListsByDomain_Result\n// struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetTaskListsByDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetTaskListsByDomain_Result match the\n// provided WorkflowService_GetTaskListsByDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) Equals(rhs *WorkflowService_GetTaskListsByDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetTaskListsByDomain_Result.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetSuccess() (o *shared.GetTaskListsByDomainResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetTaskListsByDomain\" for this struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) MethodName() string {\n\treturn \"GetTaskListsByDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_GetTaskListsByDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_GetWorkflowExecutionHistory_Args represents the arguments for the WorkflowService.GetWorkflowExecutionHistory function.\n//\n// The arguments for GetWorkflowExecutionHistory are sent and received over the wire as this struct.\ntype WorkflowService_GetWorkflowExecutionHistory_Args struct {\n\tGetRequest *shared.GetWorkflowExecutionHistoryRequest `json:\"getRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetWorkflowExecutionHistory_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.GetRequest != nil {\n\t\tw, err = v.GetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetWorkflowExecutionHistoryRequest_Read(w wire.Value) (*shared.GetWorkflowExecutionHistoryRequest, error) {\n\tvar v shared.GetWorkflowExecutionHistoryRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetWorkflowExecutionHistory_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetWorkflowExecutionHistory_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetWorkflowExecutionHistory_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.GetRequest, err = _GetWorkflowExecutionHistoryRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetWorkflowExecutionHistory_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetWorkflowExecutionHistory_Args struct could not be encoded.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.GetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.GetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetWorkflowExecutionHistoryRequest_Decode(sr stream.Reader) (*shared.GetWorkflowExecutionHistoryRequest, error) {\n\tvar v shared.GetWorkflowExecutionHistoryRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetWorkflowExecutionHistory_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetWorkflowExecutionHistory_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.GetRequest, err = _GetWorkflowExecutionHistoryRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetWorkflowExecutionHistory_Args\n// struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.GetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"GetRequest: %v\", v.GetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetWorkflowExecutionHistory_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetWorkflowExecutionHistory_Args match the\n// provided WorkflowService_GetWorkflowExecutionHistory_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) Equals(rhs *WorkflowService_GetWorkflowExecutionHistory_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.GetRequest == nil && rhs.GetRequest == nil) || (v.GetRequest != nil && rhs.GetRequest != nil && v.GetRequest.Equals(rhs.GetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetWorkflowExecutionHistory_Args.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.GetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"getRequest\", v.GetRequest))\n\t}\n\treturn err\n}\n\n// GetGetRequest returns the value of GetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) GetGetRequest() (o *shared.GetWorkflowExecutionHistoryRequest) {\n\tif v != nil && v.GetRequest != nil {\n\t\treturn v.GetRequest\n\t}\n\n\treturn\n}\n\n// IsSetGetRequest returns true if GetRequest is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) IsSetGetRequest() bool {\n\treturn v != nil && v.GetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetWorkflowExecutionHistory\" for this struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) MethodName() string {\n\treturn \"GetWorkflowExecutionHistory\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_GetWorkflowExecutionHistory_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.GetWorkflowExecutionHistory\n// function.\nvar WorkflowService_GetWorkflowExecutionHistory_Helper = struct {\n\t// Args accepts the parameters of GetWorkflowExecutionHistory in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tgetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\t) *WorkflowService_GetWorkflowExecutionHistory_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetWorkflowExecutionHistory.\n\t//\n\t// An error can be thrown by GetWorkflowExecutionHistory only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetWorkflowExecutionHistory\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetWorkflowExecutionHistory into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetWorkflowExecutionHistory\n\t//\n\t//   value, err := GetWorkflowExecutionHistory(args)\n\t//   result, err := WorkflowService_GetWorkflowExecutionHistory_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetWorkflowExecutionHistory: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetWorkflowExecutionHistoryResponse, error) (*WorkflowService_GetWorkflowExecutionHistory_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetWorkflowExecutionHistory\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetWorkflowExecutionHistory threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_GetWorkflowExecutionHistory_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_GetWorkflowExecutionHistory_Result) (*shared.GetWorkflowExecutionHistoryResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_GetWorkflowExecutionHistory_Helper.Args = func(\n\t\tgetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\t) *WorkflowService_GetWorkflowExecutionHistory_Args {\n\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Args{\n\t\t\tGetRequest: getRequest,\n\t\t}\n\t}\n\n\tWorkflowService_GetWorkflowExecutionHistory_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_GetWorkflowExecutionHistory_Helper.WrapResponse = func(success *shared.GetWorkflowExecutionHistoryResponse, err error) (*WorkflowService_GetWorkflowExecutionHistory_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetWorkflowExecutionHistory_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetWorkflowExecutionHistory_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetWorkflowExecutionHistory_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetWorkflowExecutionHistory_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_GetWorkflowExecutionHistory_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_GetWorkflowExecutionHistory_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_GetWorkflowExecutionHistory_Helper.UnwrapResponse = func(result *WorkflowService_GetWorkflowExecutionHistory_Result) (success *shared.GetWorkflowExecutionHistoryResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_GetWorkflowExecutionHistory_Result represents the result of a WorkflowService.GetWorkflowExecutionHistory function call.\n//\n// The result of a GetWorkflowExecutionHistory execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_GetWorkflowExecutionHistory_Result struct {\n\t// Value returned by GetWorkflowExecutionHistory after a successful execution.\n\tSuccess                        *shared.GetWorkflowExecutionHistoryResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                     `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                    `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError      `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                   `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_GetWorkflowExecutionHistory_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_GetWorkflowExecutionHistory_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetWorkflowExecutionHistoryResponse_Read(w wire.Value) (*shared.GetWorkflowExecutionHistoryResponse, error) {\n\tvar v shared.GetWorkflowExecutionHistoryResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_GetWorkflowExecutionHistory_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_GetWorkflowExecutionHistory_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_GetWorkflowExecutionHistory_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetWorkflowExecutionHistoryResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetWorkflowExecutionHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_GetWorkflowExecutionHistory_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_GetWorkflowExecutionHistory_Result struct could not be encoded.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetWorkflowExecutionHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetWorkflowExecutionHistoryResponse_Decode(sr stream.Reader) (*shared.GetWorkflowExecutionHistoryResponse, error) {\n\tvar v shared.GetWorkflowExecutionHistoryResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_GetWorkflowExecutionHistory_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_GetWorkflowExecutionHistory_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetWorkflowExecutionHistoryResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_GetWorkflowExecutionHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_GetWorkflowExecutionHistory_Result\n// struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_GetWorkflowExecutionHistory_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_GetWorkflowExecutionHistory_Result match the\n// provided WorkflowService_GetWorkflowExecutionHistory_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) Equals(rhs *WorkflowService_GetWorkflowExecutionHistory_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_GetWorkflowExecutionHistory_Result.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetSuccess() (o *shared.GetWorkflowExecutionHistoryResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetWorkflowExecutionHistory\" for this struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) MethodName() string {\n\treturn \"GetWorkflowExecutionHistory\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_GetWorkflowExecutionHistory_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListArchivedWorkflowExecutions_Args represents the arguments for the WorkflowService.ListArchivedWorkflowExecutions function.\n//\n// The arguments for ListArchivedWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_ListArchivedWorkflowExecutions_Args struct {\n\tListRequest *shared.ListArchivedWorkflowExecutionsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListArchivedWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListArchivedWorkflowExecutionsRequest_Read(w wire.Value) (*shared.ListArchivedWorkflowExecutionsRequest, error) {\n\tvar v shared.ListArchivedWorkflowExecutionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListArchivedWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListArchivedWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListArchivedWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListArchivedWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListArchivedWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListArchivedWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListArchivedWorkflowExecutionsRequest_Decode(sr stream.Reader) (*shared.ListArchivedWorkflowExecutionsRequest, error) {\n\tvar v shared.ListArchivedWorkflowExecutionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListArchivedWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListArchivedWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListArchivedWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListArchivedWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListArchivedWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListArchivedWorkflowExecutions_Args match the\n// provided WorkflowService_ListArchivedWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) Equals(rhs *WorkflowService_ListArchivedWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListArchivedWorkflowExecutions_Args.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) GetListRequest() (o *shared.ListArchivedWorkflowExecutionsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListArchivedWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) MethodName() string {\n\treturn \"ListArchivedWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListArchivedWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListArchivedWorkflowExecutions\n// function.\nvar WorkflowService_ListArchivedWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of ListArchivedWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\t) *WorkflowService_ListArchivedWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListArchivedWorkflowExecutions.\n\t//\n\t// An error can be thrown by ListArchivedWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListArchivedWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListArchivedWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListArchivedWorkflowExecutions\n\t//\n\t//   value, err := ListArchivedWorkflowExecutions(args)\n\t//   result, err := WorkflowService_ListArchivedWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListArchivedWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListArchivedWorkflowExecutionsResponse, error) (*WorkflowService_ListArchivedWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListArchivedWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListArchivedWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListArchivedWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListArchivedWorkflowExecutions_Result) (*shared.ListArchivedWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListArchivedWorkflowExecutions_Helper.Args = func(\n\t\tlistRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\t) *WorkflowService_ListArchivedWorkflowExecutions_Args {\n\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListArchivedWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListArchivedWorkflowExecutions_Helper.WrapResponse = func(success *shared.ListArchivedWorkflowExecutionsResponse, err error) (*WorkflowService_ListArchivedWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListArchivedWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListArchivedWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListArchivedWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListArchivedWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListArchivedWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListArchivedWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListArchivedWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_ListArchivedWorkflowExecutions_Result) (success *shared.ListArchivedWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListArchivedWorkflowExecutions_Result represents the result of a WorkflowService.ListArchivedWorkflowExecutions function call.\n//\n// The result of a ListArchivedWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListArchivedWorkflowExecutions_Result struct {\n\t// Value returned by ListArchivedWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.ListArchivedWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListArchivedWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListArchivedWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListArchivedWorkflowExecutionsResponse_Read(w wire.Value) (*shared.ListArchivedWorkflowExecutionsResponse, error) {\n\tvar v shared.ListArchivedWorkflowExecutionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListArchivedWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListArchivedWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListArchivedWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListArchivedWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListArchivedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListArchivedWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListArchivedWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListArchivedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListArchivedWorkflowExecutionsResponse_Decode(sr stream.Reader) (*shared.ListArchivedWorkflowExecutionsResponse, error) {\n\tvar v shared.ListArchivedWorkflowExecutionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListArchivedWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListArchivedWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListArchivedWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListArchivedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListArchivedWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListArchivedWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListArchivedWorkflowExecutions_Result match the\n// provided WorkflowService_ListArchivedWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) Equals(rhs *WorkflowService_ListArchivedWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListArchivedWorkflowExecutions_Result.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetSuccess() (o *shared.ListArchivedWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListArchivedWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) MethodName() string {\n\treturn \"ListArchivedWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListArchivedWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListClosedWorkflowExecutions_Args represents the arguments for the WorkflowService.ListClosedWorkflowExecutions function.\n//\n// The arguments for ListClosedWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_ListClosedWorkflowExecutions_Args struct {\n\tListRequest *shared.ListClosedWorkflowExecutionsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListClosedWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListClosedWorkflowExecutionsRequest_Read(w wire.Value) (*shared.ListClosedWorkflowExecutionsRequest, error) {\n\tvar v shared.ListClosedWorkflowExecutionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListClosedWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListClosedWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListClosedWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListClosedWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListClosedWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListClosedWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListClosedWorkflowExecutionsRequest_Decode(sr stream.Reader) (*shared.ListClosedWorkflowExecutionsRequest, error) {\n\tvar v shared.ListClosedWorkflowExecutionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListClosedWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListClosedWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListClosedWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListClosedWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListClosedWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListClosedWorkflowExecutions_Args match the\n// provided WorkflowService_ListClosedWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) Equals(rhs *WorkflowService_ListClosedWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListClosedWorkflowExecutions_Args.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) GetListRequest() (o *shared.ListClosedWorkflowExecutionsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListClosedWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) MethodName() string {\n\treturn \"ListClosedWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListClosedWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListClosedWorkflowExecutions\n// function.\nvar WorkflowService_ListClosedWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of ListClosedWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListClosedWorkflowExecutionsRequest,\n\t) *WorkflowService_ListClosedWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListClosedWorkflowExecutions.\n\t//\n\t// An error can be thrown by ListClosedWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListClosedWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListClosedWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListClosedWorkflowExecutions\n\t//\n\t//   value, err := ListClosedWorkflowExecutions(args)\n\t//   result, err := WorkflowService_ListClosedWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListClosedWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListClosedWorkflowExecutionsResponse, error) (*WorkflowService_ListClosedWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListClosedWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListClosedWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListClosedWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListClosedWorkflowExecutions_Result) (*shared.ListClosedWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListClosedWorkflowExecutions_Helper.Args = func(\n\t\tlistRequest *shared.ListClosedWorkflowExecutionsRequest,\n\t) *WorkflowService_ListClosedWorkflowExecutions_Args {\n\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListClosedWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListClosedWorkflowExecutions_Helper.WrapResponse = func(success *shared.ListClosedWorkflowExecutionsResponse, err error) (*WorkflowService_ListClosedWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListClosedWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListClosedWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListClosedWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListClosedWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListClosedWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListClosedWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListClosedWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_ListClosedWorkflowExecutions_Result) (success *shared.ListClosedWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListClosedWorkflowExecutions_Result represents the result of a WorkflowService.ListClosedWorkflowExecutions function call.\n//\n// The result of a ListClosedWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListClosedWorkflowExecutions_Result struct {\n\t// Value returned by ListClosedWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.ListClosedWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                 `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError       `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListClosedWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListClosedWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListClosedWorkflowExecutionsResponse_Read(w wire.Value) (*shared.ListClosedWorkflowExecutionsResponse, error) {\n\tvar v shared.ListClosedWorkflowExecutionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListClosedWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListClosedWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListClosedWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListClosedWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListClosedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListClosedWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListClosedWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListClosedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListClosedWorkflowExecutionsResponse_Decode(sr stream.Reader) (*shared.ListClosedWorkflowExecutionsResponse, error) {\n\tvar v shared.ListClosedWorkflowExecutionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListClosedWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListClosedWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListClosedWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListClosedWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListClosedWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListClosedWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListClosedWorkflowExecutions_Result match the\n// provided WorkflowService_ListClosedWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) Equals(rhs *WorkflowService_ListClosedWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListClosedWorkflowExecutions_Result.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetSuccess() (o *shared.ListClosedWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListClosedWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) MethodName() string {\n\treturn \"ListClosedWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListClosedWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListDomains_Args represents the arguments for the WorkflowService.ListDomains function.\n//\n// The arguments for ListDomains are sent and received over the wire as this struct.\ntype WorkflowService_ListDomains_Args struct {\n\tListRequest *shared.ListDomainsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListDomains_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListDomains_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListDomainsRequest_Read(w wire.Value) (*shared.ListDomainsRequest, error) {\n\tvar v shared.ListDomainsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListDomains_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListDomains_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListDomains_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListDomains_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListDomainsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListDomains_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListDomains_Args struct could not be encoded.\nfunc (v *WorkflowService_ListDomains_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListDomainsRequest_Decode(sr stream.Reader) (*shared.ListDomainsRequest, error) {\n\tvar v shared.ListDomainsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListDomains_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListDomains_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListDomains_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListDomainsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListDomains_Args\n// struct.\nfunc (v *WorkflowService_ListDomains_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListDomains_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListDomains_Args match the\n// provided WorkflowService_ListDomains_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListDomains_Args) Equals(rhs *WorkflowService_ListDomains_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListDomains_Args.\nfunc (v *WorkflowService_ListDomains_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Args) GetListRequest() (o *shared.ListDomainsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListDomains_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListDomains\" for this struct.\nfunc (v *WorkflowService_ListDomains_Args) MethodName() string {\n\treturn \"ListDomains\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListDomains_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListDomains_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListDomains\n// function.\nvar WorkflowService_ListDomains_Helper = struct {\n\t// Args accepts the parameters of ListDomains in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListDomainsRequest,\n\t) *WorkflowService_ListDomains_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListDomains.\n\t//\n\t// An error can be thrown by ListDomains only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListDomains\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListDomains into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListDomains\n\t//\n\t//   value, err := ListDomains(args)\n\t//   result, err := WorkflowService_ListDomains_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListDomains: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListDomainsResponse, error) (*WorkflowService_ListDomains_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListDomains\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListDomains threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListDomains_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListDomains_Result) (*shared.ListDomainsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListDomains_Helper.Args = func(\n\t\tlistRequest *shared.ListDomainsRequest,\n\t) *WorkflowService_ListDomains_Args {\n\t\treturn &WorkflowService_ListDomains_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListDomains_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListDomains_Helper.WrapResponse = func(success *shared.ListDomainsResponse, err error) (*WorkflowService_ListDomains_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListDomains_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListDomains_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListDomains_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListDomains_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListDomains_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListDomains_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListDomains_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListDomains_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListDomains_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListDomains_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListDomains_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListDomains_Helper.UnwrapResponse = func(result *WorkflowService_ListDomains_Result) (success *shared.ListDomainsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListDomains_Result represents the result of a WorkflowService.ListDomains function call.\n//\n// The result of a ListDomains execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListDomains_Result struct {\n\t// Value returned by ListDomains after a successful execution.\n\tSuccess                        *shared.ListDomainsResponse            `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListDomains_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListDomains_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListDomains_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListDomainsResponse_Read(w wire.Value) (*shared.ListDomainsResponse, error) {\n\tvar v shared.ListDomainsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListDomains_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListDomains_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListDomains_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListDomains_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListDomainsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListDomains_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListDomains_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListDomains_Result struct could not be encoded.\nfunc (v *WorkflowService_ListDomains_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListDomains_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListDomainsResponse_Decode(sr stream.Reader) (*shared.ListDomainsResponse, error) {\n\tvar v shared.ListDomainsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListDomains_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListDomains_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListDomains_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListDomainsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListDomains_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListDomains_Result\n// struct.\nfunc (v *WorkflowService_ListDomains_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListDomains_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListDomains_Result match the\n// provided WorkflowService_ListDomains_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListDomains_Result) Equals(rhs *WorkflowService_ListDomains_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListDomains_Result.\nfunc (v *WorkflowService_ListDomains_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetSuccess() (o *shared.ListDomainsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListDomains_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListDomains_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListDomains\" for this struct.\nfunc (v *WorkflowService_ListDomains_Result) MethodName() string {\n\treturn \"ListDomains\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListDomains_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListFailoverHistory_Args represents the arguments for the WorkflowService.ListFailoverHistory function.\n//\n// The arguments for ListFailoverHistory are sent and received over the wire as this struct.\ntype WorkflowService_ListFailoverHistory_Args struct {\n\tListRequest *shared.ListFailoverHistoryRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListFailoverHistory_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListFailoverHistory_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListFailoverHistoryRequest_Read(w wire.Value) (*shared.ListFailoverHistoryRequest, error) {\n\tvar v shared.ListFailoverHistoryRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListFailoverHistory_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListFailoverHistory_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListFailoverHistory_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListFailoverHistory_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListFailoverHistoryRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListFailoverHistory_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListFailoverHistory_Args struct could not be encoded.\nfunc (v *WorkflowService_ListFailoverHistory_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListFailoverHistoryRequest_Decode(sr stream.Reader) (*shared.ListFailoverHistoryRequest, error) {\n\tvar v shared.ListFailoverHistoryRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListFailoverHistory_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListFailoverHistory_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListFailoverHistory_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListFailoverHistoryRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListFailoverHistory_Args\n// struct.\nfunc (v *WorkflowService_ListFailoverHistory_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListFailoverHistory_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListFailoverHistory_Args match the\n// provided WorkflowService_ListFailoverHistory_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListFailoverHistory_Args) Equals(rhs *WorkflowService_ListFailoverHistory_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListFailoverHistory_Args.\nfunc (v *WorkflowService_ListFailoverHistory_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Args) GetListRequest() (o *shared.ListFailoverHistoryRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListFailoverHistory\" for this struct.\nfunc (v *WorkflowService_ListFailoverHistory_Args) MethodName() string {\n\treturn \"ListFailoverHistory\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListFailoverHistory_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListFailoverHistory_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListFailoverHistory\n// function.\nvar WorkflowService_ListFailoverHistory_Helper = struct {\n\t// Args accepts the parameters of ListFailoverHistory in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListFailoverHistoryRequest,\n\t) *WorkflowService_ListFailoverHistory_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListFailoverHistory.\n\t//\n\t// An error can be thrown by ListFailoverHistory only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListFailoverHistory\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListFailoverHistory into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListFailoverHistory\n\t//\n\t//   value, err := ListFailoverHistory(args)\n\t//   result, err := WorkflowService_ListFailoverHistory_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListFailoverHistory: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListFailoverHistoryResponse, error) (*WorkflowService_ListFailoverHistory_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListFailoverHistory\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListFailoverHistory threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListFailoverHistory_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListFailoverHistory_Result) (*shared.ListFailoverHistoryResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListFailoverHistory_Helper.Args = func(\n\t\tlistRequest *shared.ListFailoverHistoryRequest,\n\t) *WorkflowService_ListFailoverHistory_Args {\n\t\treturn &WorkflowService_ListFailoverHistory_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListFailoverHistory_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListFailoverHistory_Helper.WrapResponse = func(success *shared.ListFailoverHistoryResponse, err error) (*WorkflowService_ListFailoverHistory_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListFailoverHistory_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListFailoverHistory_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListFailoverHistory_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListFailoverHistory_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListFailoverHistory_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListFailoverHistory_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListFailoverHistory_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListFailoverHistory_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListFailoverHistory_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListFailoverHistory_Helper.UnwrapResponse = func(result *WorkflowService_ListFailoverHistory_Result) (success *shared.ListFailoverHistoryResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListFailoverHistory_Result represents the result of a WorkflowService.ListFailoverHistory function call.\n//\n// The result of a ListFailoverHistory execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListFailoverHistory_Result struct {\n\t// Value returned by ListFailoverHistory after a successful execution.\n\tSuccess                        *shared.ListFailoverHistoryResponse    `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListFailoverHistory_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListFailoverHistory_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListFailoverHistory_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListFailoverHistoryResponse_Read(w wire.Value) (*shared.ListFailoverHistoryResponse, error) {\n\tvar v shared.ListFailoverHistoryResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListFailoverHistory_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListFailoverHistory_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListFailoverHistory_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListFailoverHistory_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListFailoverHistoryResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListFailoverHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListFailoverHistory_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListFailoverHistory_Result struct could not be encoded.\nfunc (v *WorkflowService_ListFailoverHistory_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListFailoverHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListFailoverHistoryResponse_Decode(sr stream.Reader) (*shared.ListFailoverHistoryResponse, error) {\n\tvar v shared.ListFailoverHistoryResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListFailoverHistory_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListFailoverHistory_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListFailoverHistory_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListFailoverHistoryResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListFailoverHistory_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListFailoverHistory_Result\n// struct.\nfunc (v *WorkflowService_ListFailoverHistory_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListFailoverHistory_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListFailoverHistory_Result match the\n// provided WorkflowService_ListFailoverHistory_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListFailoverHistory_Result) Equals(rhs *WorkflowService_ListFailoverHistory_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListFailoverHistory_Result.\nfunc (v *WorkflowService_ListFailoverHistory_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Result) GetSuccess() (o *shared.ListFailoverHistoryResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListFailoverHistory_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListFailoverHistory_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListFailoverHistory\" for this struct.\nfunc (v *WorkflowService_ListFailoverHistory_Result) MethodName() string {\n\treturn \"ListFailoverHistory\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListFailoverHistory_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListOpenWorkflowExecutions_Args represents the arguments for the WorkflowService.ListOpenWorkflowExecutions function.\n//\n// The arguments for ListOpenWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_ListOpenWorkflowExecutions_Args struct {\n\tListRequest *shared.ListOpenWorkflowExecutionsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListOpenWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListOpenWorkflowExecutionsRequest_Read(w wire.Value) (*shared.ListOpenWorkflowExecutionsRequest, error) {\n\tvar v shared.ListOpenWorkflowExecutionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListOpenWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListOpenWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListOpenWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListOpenWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListOpenWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListOpenWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListOpenWorkflowExecutionsRequest_Decode(sr stream.Reader) (*shared.ListOpenWorkflowExecutionsRequest, error) {\n\tvar v shared.ListOpenWorkflowExecutionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListOpenWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListOpenWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListOpenWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListOpenWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListOpenWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListOpenWorkflowExecutions_Args match the\n// provided WorkflowService_ListOpenWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) Equals(rhs *WorkflowService_ListOpenWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListOpenWorkflowExecutions_Args.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) GetListRequest() (o *shared.ListOpenWorkflowExecutionsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListOpenWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) MethodName() string {\n\treturn \"ListOpenWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListOpenWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListOpenWorkflowExecutions\n// function.\nvar WorkflowService_ListOpenWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of ListOpenWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListOpenWorkflowExecutionsRequest,\n\t) *WorkflowService_ListOpenWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListOpenWorkflowExecutions.\n\t//\n\t// An error can be thrown by ListOpenWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListOpenWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListOpenWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListOpenWorkflowExecutions\n\t//\n\t//   value, err := ListOpenWorkflowExecutions(args)\n\t//   result, err := WorkflowService_ListOpenWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListOpenWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListOpenWorkflowExecutionsResponse, error) (*WorkflowService_ListOpenWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListOpenWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListOpenWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListOpenWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListOpenWorkflowExecutions_Result) (*shared.ListOpenWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListOpenWorkflowExecutions_Helper.Args = func(\n\t\tlistRequest *shared.ListOpenWorkflowExecutionsRequest,\n\t) *WorkflowService_ListOpenWorkflowExecutions_Args {\n\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListOpenWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListOpenWorkflowExecutions_Helper.WrapResponse = func(success *shared.ListOpenWorkflowExecutionsResponse, err error) (*WorkflowService_ListOpenWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListOpenWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListOpenWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListOpenWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_ListOpenWorkflowExecutions_Result) (success *shared.ListOpenWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListOpenWorkflowExecutions_Result represents the result of a WorkflowService.ListOpenWorkflowExecutions function call.\n//\n// The result of a ListOpenWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListOpenWorkflowExecutions_Result struct {\n\t// Value returned by ListOpenWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.ListOpenWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                    `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError               `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                   `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                 `json:\"limitExceededError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError     `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                  `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListOpenWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListOpenWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListOpenWorkflowExecutionsResponse_Read(w wire.Value) (*shared.ListOpenWorkflowExecutionsResponse, error) {\n\tvar v shared.ListOpenWorkflowExecutionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListOpenWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListOpenWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListOpenWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListOpenWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListOpenWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListOpenWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListOpenWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListOpenWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListOpenWorkflowExecutionsResponse_Decode(sr stream.Reader) (*shared.ListOpenWorkflowExecutionsResponse, error) {\n\tvar v shared.ListOpenWorkflowExecutionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListOpenWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListOpenWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListOpenWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListOpenWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListOpenWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListOpenWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListOpenWorkflowExecutions_Result match the\n// provided WorkflowService_ListOpenWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) Equals(rhs *WorkflowService_ListOpenWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListOpenWorkflowExecutions_Result.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetSuccess() (o *shared.ListOpenWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListOpenWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) MethodName() string {\n\treturn \"ListOpenWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListOpenWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListTaskListPartitions_Args represents the arguments for the WorkflowService.ListTaskListPartitions function.\n//\n// The arguments for ListTaskListPartitions are sent and received over the wire as this struct.\ntype WorkflowService_ListTaskListPartitions_Args struct {\n\tRequest *shared.ListTaskListPartitionsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListTaskListPartitions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListTaskListPartitions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListTaskListPartitionsRequest_Read(w wire.Value) (*shared.ListTaskListPartitionsRequest, error) {\n\tvar v shared.ListTaskListPartitionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListTaskListPartitions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListTaskListPartitions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListTaskListPartitions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListTaskListPartitions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ListTaskListPartitionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListTaskListPartitions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListTaskListPartitions_Args struct could not be encoded.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListTaskListPartitionsRequest_Decode(sr stream.Reader) (*shared.ListTaskListPartitionsRequest, error) {\n\tvar v shared.ListTaskListPartitionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListTaskListPartitions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListTaskListPartitions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ListTaskListPartitionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListTaskListPartitions_Args\n// struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListTaskListPartitions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListTaskListPartitions_Args match the\n// provided WorkflowService_ListTaskListPartitions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) Equals(rhs *WorkflowService_ListTaskListPartitions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListTaskListPartitions_Args.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) GetRequest() (o *shared.ListTaskListPartitionsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListTaskListPartitions\" for this struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) MethodName() string {\n\treturn \"ListTaskListPartitions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListTaskListPartitions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListTaskListPartitions\n// function.\nvar WorkflowService_ListTaskListPartitions_Helper = struct {\n\t// Args accepts the parameters of ListTaskListPartitions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.ListTaskListPartitionsRequest,\n\t) *WorkflowService_ListTaskListPartitions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListTaskListPartitions.\n\t//\n\t// An error can be thrown by ListTaskListPartitions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListTaskListPartitions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListTaskListPartitions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListTaskListPartitions\n\t//\n\t//   value, err := ListTaskListPartitions(args)\n\t//   result, err := WorkflowService_ListTaskListPartitions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListTaskListPartitions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListTaskListPartitionsResponse, error) (*WorkflowService_ListTaskListPartitions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListTaskListPartitions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListTaskListPartitions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListTaskListPartitions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListTaskListPartitions_Result) (*shared.ListTaskListPartitionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListTaskListPartitions_Helper.Args = func(\n\t\trequest *shared.ListTaskListPartitionsRequest,\n\t) *WorkflowService_ListTaskListPartitions_Args {\n\t\treturn &WorkflowService_ListTaskListPartitions_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tWorkflowService_ListTaskListPartitions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListTaskListPartitions_Helper.WrapResponse = func(success *shared.ListTaskListPartitionsResponse, err error) (*WorkflowService_ListTaskListPartitions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListTaskListPartitions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListTaskListPartitions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListTaskListPartitions_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListTaskListPartitions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListTaskListPartitions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListTaskListPartitions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListTaskListPartitions_Helper.UnwrapResponse = func(result *WorkflowService_ListTaskListPartitions_Result) (success *shared.ListTaskListPartitionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListTaskListPartitions_Result represents the result of a WorkflowService.ListTaskListPartitions function call.\n//\n// The result of a ListTaskListPartitions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListTaskListPartitions_Result struct {\n\t// Value returned by ListTaskListPartitions after a successful execution.\n\tSuccess             *shared.ListTaskListPartitionsResponse `json:\"success,omitempty\"`\n\tBadRequestError     *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError  *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError    *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tAccessDeniedError   *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListTaskListPartitions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListTaskListPartitions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListTaskListPartitionsResponse_Read(w wire.Value) (*shared.ListTaskListPartitionsResponse, error) {\n\tvar v shared.ListTaskListPartitionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListTaskListPartitions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListTaskListPartitions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListTaskListPartitions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListTaskListPartitions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListTaskListPartitionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListTaskListPartitions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListTaskListPartitions_Result struct could not be encoded.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListTaskListPartitionsResponse_Decode(sr stream.Reader) (*shared.ListTaskListPartitionsResponse, error) {\n\tvar v shared.ListTaskListPartitionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListTaskListPartitions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListTaskListPartitions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListTaskListPartitionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListTaskListPartitions_Result\n// struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListTaskListPartitions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListTaskListPartitions_Result match the\n// provided WorkflowService_ListTaskListPartitions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) Equals(rhs *WorkflowService_ListTaskListPartitions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListTaskListPartitions_Result.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetSuccess() (o *shared.ListTaskListPartitionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListTaskListPartitions\" for this struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) MethodName() string {\n\treturn \"ListTaskListPartitions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListTaskListPartitions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ListWorkflowExecutions_Args represents the arguments for the WorkflowService.ListWorkflowExecutions function.\n//\n// The arguments for ListWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_ListWorkflowExecutions_Args struct {\n\tListRequest *shared.ListWorkflowExecutionsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListWorkflowExecutionsRequest_Read(w wire.Value) (*shared.ListWorkflowExecutionsRequest, error) {\n\tvar v shared.ListWorkflowExecutionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListWorkflowExecutionsRequest_Decode(sr stream.Reader) (*shared.ListWorkflowExecutionsRequest, error) {\n\tvar v shared.ListWorkflowExecutionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListWorkflowExecutions_Args match the\n// provided WorkflowService_ListWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) Equals(rhs *WorkflowService_ListWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListWorkflowExecutions_Args.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) GetListRequest() (o *shared.ListWorkflowExecutionsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) MethodName() string {\n\treturn \"ListWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ListWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ListWorkflowExecutions\n// function.\nvar WorkflowService_ListWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of ListWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListWorkflowExecutionsRequest,\n\t) *WorkflowService_ListWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListWorkflowExecutions.\n\t//\n\t// An error can be thrown by ListWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListWorkflowExecutions\n\t//\n\t//   value, err := ListWorkflowExecutions(args)\n\t//   result, err := WorkflowService_ListWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListWorkflowExecutionsResponse, error) (*WorkflowService_ListWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ListWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ListWorkflowExecutions_Result) (*shared.ListWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ListWorkflowExecutions_Helper.Args = func(\n\t\tlistRequest *shared.ListWorkflowExecutionsRequest,\n\t) *WorkflowService_ListWorkflowExecutions_Args {\n\t\treturn &WorkflowService_ListWorkflowExecutions_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ListWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ListWorkflowExecutions_Helper.WrapResponse = func(success *shared.ListWorkflowExecutionsResponse, err error) (*WorkflowService_ListWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ListWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ListWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ListWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_ListWorkflowExecutions_Result) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ListWorkflowExecutions_Result represents the result of a WorkflowService.ListWorkflowExecutions function call.\n//\n// The result of a ListWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ListWorkflowExecutions_Result struct {\n\t// Value returned by ListWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.ListWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ListWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ListWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListWorkflowExecutionsResponse_Read(w wire.Value) (*shared.ListWorkflowExecutionsResponse, error) {\n\tvar v shared.ListWorkflowExecutionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ListWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ListWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ListWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ListWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ListWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListWorkflowExecutionsResponse_Decode(sr stream.Reader) (*shared.ListWorkflowExecutionsResponse, error) {\n\tvar v shared.ListWorkflowExecutionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ListWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ListWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ListWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ListWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ListWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ListWorkflowExecutions_Result match the\n// provided WorkflowService_ListWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) Equals(rhs *WorkflowService_ListWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ListWorkflowExecutions_Result.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetSuccess() (o *shared.ListWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) MethodName() string {\n\treturn \"ListWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ListWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_PollForActivityTask_Args represents the arguments for the WorkflowService.PollForActivityTask function.\n//\n// The arguments for PollForActivityTask are sent and received over the wire as this struct.\ntype WorkflowService_PollForActivityTask_Args struct {\n\tPollRequest *shared.PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_PollForActivityTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_PollForActivityTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskRequest_Read(w wire.Value) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_PollForActivityTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_PollForActivityTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_PollForActivityTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_PollForActivityTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_PollForActivityTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_PollForActivityTask_Args struct could not be encoded.\nfunc (v *WorkflowService_PollForActivityTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskRequest_Decode(sr stream.Reader) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_PollForActivityTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_PollForActivityTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_PollForActivityTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_PollForActivityTask_Args\n// struct.\nfunc (v *WorkflowService_PollForActivityTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_PollForActivityTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_PollForActivityTask_Args match the\n// provided WorkflowService_PollForActivityTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_PollForActivityTask_Args) Equals(rhs *WorkflowService_PollForActivityTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_PollForActivityTask_Args.\nfunc (v *WorkflowService_PollForActivityTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Args) GetPollRequest() (o *shared.PollForActivityTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Args) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PollForActivityTask\" for this struct.\nfunc (v *WorkflowService_PollForActivityTask_Args) MethodName() string {\n\treturn \"PollForActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_PollForActivityTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_PollForActivityTask_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.PollForActivityTask\n// function.\nvar WorkflowService_PollForActivityTask_Helper = struct {\n\t// Args accepts the parameters of PollForActivityTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tpollRequest *shared.PollForActivityTaskRequest,\n\t) *WorkflowService_PollForActivityTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PollForActivityTask.\n\t//\n\t// An error can be thrown by PollForActivityTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PollForActivityTask\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// PollForActivityTask into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by PollForActivityTask\n\t//\n\t//   value, err := PollForActivityTask(args)\n\t//   result, err := WorkflowService_PollForActivityTask_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PollForActivityTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.PollForActivityTaskResponse, error) (*WorkflowService_PollForActivityTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for PollForActivityTask\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if PollForActivityTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_PollForActivityTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_PollForActivityTask_Result) (*shared.PollForActivityTaskResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_PollForActivityTask_Helper.Args = func(\n\t\tpollRequest *shared.PollForActivityTaskRequest,\n\t) *WorkflowService_PollForActivityTask_Args {\n\t\treturn &WorkflowService_PollForActivityTask_Args{\n\t\t\tPollRequest: pollRequest,\n\t\t}\n\t}\n\n\tWorkflowService_PollForActivityTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_PollForActivityTask_Helper.WrapResponse = func(success *shared.PollForActivityTaskResponse, err error) (*WorkflowService_PollForActivityTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForActivityTask_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForActivityTask_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_PollForActivityTask_Helper.UnwrapResponse = func(result *WorkflowService_PollForActivityTask_Result) (success *shared.PollForActivityTaskResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_PollForActivityTask_Result represents the result of a WorkflowService.PollForActivityTask function call.\n//\n// The result of a PollForActivityTask execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_PollForActivityTask_Result struct {\n\t// Value returned by PollForActivityTask after a successful execution.\n\tSuccess                        *shared.PollForActivityTaskResponse    `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_PollForActivityTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_PollForActivityTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_PollForActivityTask_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskResponse_Read(w wire.Value) (*shared.PollForActivityTaskResponse, error) {\n\tvar v shared.PollForActivityTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_PollForActivityTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_PollForActivityTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_PollForActivityTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_PollForActivityTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _PollForActivityTaskResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_PollForActivityTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_PollForActivityTask_Result struct could not be encoded.\nfunc (v *WorkflowService_PollForActivityTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskResponse_Decode(sr stream.Reader) (*shared.PollForActivityTaskResponse, error) {\n\tvar v shared.PollForActivityTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_PollForActivityTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_PollForActivityTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_PollForActivityTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _PollForActivityTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_PollForActivityTask_Result\n// struct.\nfunc (v *WorkflowService_PollForActivityTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_PollForActivityTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_PollForActivityTask_Result match the\n// provided WorkflowService_PollForActivityTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_PollForActivityTask_Result) Equals(rhs *WorkflowService_PollForActivityTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_PollForActivityTask_Result.\nfunc (v *WorkflowService_PollForActivityTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetSuccess() (o *shared.PollForActivityTaskResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForActivityTask_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_PollForActivityTask_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PollForActivityTask\" for this struct.\nfunc (v *WorkflowService_PollForActivityTask_Result) MethodName() string {\n\treturn \"PollForActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_PollForActivityTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_PollForDecisionTask_Args represents the arguments for the WorkflowService.PollForDecisionTask function.\n//\n// The arguments for PollForDecisionTask are sent and received over the wire as this struct.\ntype WorkflowService_PollForDecisionTask_Args struct {\n\tPollRequest *shared.PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_PollForDecisionTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_PollForDecisionTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskRequest_Read(w wire.Value) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_PollForDecisionTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_PollForDecisionTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_PollForDecisionTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_PollForDecisionTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_PollForDecisionTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_PollForDecisionTask_Args struct could not be encoded.\nfunc (v *WorkflowService_PollForDecisionTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskRequest_Decode(sr stream.Reader) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_PollForDecisionTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_PollForDecisionTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_PollForDecisionTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_PollForDecisionTask_Args\n// struct.\nfunc (v *WorkflowService_PollForDecisionTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_PollForDecisionTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_PollForDecisionTask_Args match the\n// provided WorkflowService_PollForDecisionTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_PollForDecisionTask_Args) Equals(rhs *WorkflowService_PollForDecisionTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_PollForDecisionTask_Args.\nfunc (v *WorkflowService_PollForDecisionTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Args) GetPollRequest() (o *shared.PollForDecisionTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Args) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PollForDecisionTask\" for this struct.\nfunc (v *WorkflowService_PollForDecisionTask_Args) MethodName() string {\n\treturn \"PollForDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_PollForDecisionTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_PollForDecisionTask_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.PollForDecisionTask\n// function.\nvar WorkflowService_PollForDecisionTask_Helper = struct {\n\t// Args accepts the parameters of PollForDecisionTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tpollRequest *shared.PollForDecisionTaskRequest,\n\t) *WorkflowService_PollForDecisionTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PollForDecisionTask.\n\t//\n\t// An error can be thrown by PollForDecisionTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PollForDecisionTask\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// PollForDecisionTask into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by PollForDecisionTask\n\t//\n\t//   value, err := PollForDecisionTask(args)\n\t//   result, err := WorkflowService_PollForDecisionTask_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PollForDecisionTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.PollForDecisionTaskResponse, error) (*WorkflowService_PollForDecisionTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for PollForDecisionTask\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if PollForDecisionTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_PollForDecisionTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_PollForDecisionTask_Result) (*shared.PollForDecisionTaskResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_PollForDecisionTask_Helper.Args = func(\n\t\tpollRequest *shared.PollForDecisionTaskRequest,\n\t) *WorkflowService_PollForDecisionTask_Args {\n\t\treturn &WorkflowService_PollForDecisionTask_Args{\n\t\t\tPollRequest: pollRequest,\n\t\t}\n\t}\n\n\tWorkflowService_PollForDecisionTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_PollForDecisionTask_Helper.WrapResponse = func(success *shared.PollForDecisionTaskResponse, err error) (*WorkflowService_PollForDecisionTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_PollForDecisionTask_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_PollForDecisionTask_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_PollForDecisionTask_Helper.UnwrapResponse = func(result *WorkflowService_PollForDecisionTask_Result) (success *shared.PollForDecisionTaskResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_PollForDecisionTask_Result represents the result of a WorkflowService.PollForDecisionTask function call.\n//\n// The result of a PollForDecisionTask execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_PollForDecisionTask_Result struct {\n\t// Value returned by PollForDecisionTask after a successful execution.\n\tSuccess                        *shared.PollForDecisionTaskResponse    `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_PollForDecisionTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_PollForDecisionTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_PollForDecisionTask_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskResponse_Read(w wire.Value) (*shared.PollForDecisionTaskResponse, error) {\n\tvar v shared.PollForDecisionTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_PollForDecisionTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_PollForDecisionTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_PollForDecisionTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_PollForDecisionTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _PollForDecisionTaskResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_PollForDecisionTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_PollForDecisionTask_Result struct could not be encoded.\nfunc (v *WorkflowService_PollForDecisionTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskResponse_Decode(sr stream.Reader) (*shared.PollForDecisionTaskResponse, error) {\n\tvar v shared.PollForDecisionTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_PollForDecisionTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_PollForDecisionTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_PollForDecisionTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _PollForDecisionTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_PollForDecisionTask_Result\n// struct.\nfunc (v *WorkflowService_PollForDecisionTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_PollForDecisionTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_PollForDecisionTask_Result match the\n// provided WorkflowService_PollForDecisionTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_PollForDecisionTask_Result) Equals(rhs *WorkflowService_PollForDecisionTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_PollForDecisionTask_Result.\nfunc (v *WorkflowService_PollForDecisionTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetSuccess() (o *shared.PollForDecisionTaskResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_PollForDecisionTask_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_PollForDecisionTask_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PollForDecisionTask\" for this struct.\nfunc (v *WorkflowService_PollForDecisionTask_Result) MethodName() string {\n\treturn \"PollForDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_PollForDecisionTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_QueryWorkflow_Args represents the arguments for the WorkflowService.QueryWorkflow function.\n//\n// The arguments for QueryWorkflow are sent and received over the wire as this struct.\ntype WorkflowService_QueryWorkflow_Args struct {\n\tQueryRequest *shared.QueryWorkflowRequest `json:\"queryRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_QueryWorkflow_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_QueryWorkflow_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.QueryRequest != nil {\n\t\tw, err = v.QueryRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowRequest_Read(w wire.Value) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_QueryWorkflow_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_QueryWorkflow_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_QueryWorkflow_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_QueryWorkflow_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_QueryWorkflow_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_QueryWorkflow_Args struct could not be encoded.\nfunc (v *WorkflowService_QueryWorkflow_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.QueryRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowRequest_Decode(sr stream.Reader) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_QueryWorkflow_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_QueryWorkflow_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_QueryWorkflow_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_QueryWorkflow_Args\n// struct.\nfunc (v *WorkflowService_QueryWorkflow_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.QueryRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRequest: %v\", v.QueryRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_QueryWorkflow_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_QueryWorkflow_Args match the\n// provided WorkflowService_QueryWorkflow_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_QueryWorkflow_Args) Equals(rhs *WorkflowService_QueryWorkflow_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.QueryRequest == nil && rhs.QueryRequest == nil) || (v.QueryRequest != nil && rhs.QueryRequest != nil && v.QueryRequest.Equals(rhs.QueryRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_QueryWorkflow_Args.\nfunc (v *WorkflowService_QueryWorkflow_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.QueryRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRequest\", v.QueryRequest))\n\t}\n\treturn err\n}\n\n// GetQueryRequest returns the value of QueryRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Args) GetQueryRequest() (o *shared.QueryWorkflowRequest) {\n\tif v != nil && v.QueryRequest != nil {\n\t\treturn v.QueryRequest\n\t}\n\n\treturn\n}\n\n// IsSetQueryRequest returns true if QueryRequest is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Args) IsSetQueryRequest() bool {\n\treturn v != nil && v.QueryRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *WorkflowService_QueryWorkflow_Args) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_QueryWorkflow_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_QueryWorkflow_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.QueryWorkflow\n// function.\nvar WorkflowService_QueryWorkflow_Helper = struct {\n\t// Args accepts the parameters of QueryWorkflow in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tqueryRequest *shared.QueryWorkflowRequest,\n\t) *WorkflowService_QueryWorkflow_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by QueryWorkflow.\n\t//\n\t// An error can be thrown by QueryWorkflow only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for QueryWorkflow\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// QueryWorkflow into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by QueryWorkflow\n\t//\n\t//   value, err := QueryWorkflow(args)\n\t//   result, err := WorkflowService_QueryWorkflow_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from QueryWorkflow: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.QueryWorkflowResponse, error) (*WorkflowService_QueryWorkflow_Result, error)\n\n\t// UnwrapResponse takes the result struct for QueryWorkflow\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if QueryWorkflow threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_QueryWorkflow_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_QueryWorkflow_Result) (*shared.QueryWorkflowResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_QueryWorkflow_Helper.Args = func(\n\t\tqueryRequest *shared.QueryWorkflowRequest,\n\t) *WorkflowService_QueryWorkflow_Args {\n\t\treturn &WorkflowService_QueryWorkflow_Args{\n\t\t\tQueryRequest: queryRequest,\n\t\t}\n\t}\n\n\tWorkflowService_QueryWorkflow_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.QueryFailedError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_QueryWorkflow_Helper.WrapResponse = func(success *shared.QueryWorkflowResponse, err error) (*WorkflowService_QueryWorkflow_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.QueryFailedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.QueryFailedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{QueryFailedError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_QueryWorkflow_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_QueryWorkflow_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_QueryWorkflow_Helper.UnwrapResponse = func(result *WorkflowService_QueryWorkflow_Result) (success *shared.QueryWorkflowResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.QueryFailedError != nil {\n\t\t\terr = result.QueryFailedError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_QueryWorkflow_Result represents the result of a WorkflowService.QueryWorkflow function call.\n//\n// The result of a QueryWorkflow execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_QueryWorkflow_Result struct {\n\t// Value returned by QueryWorkflow after a successful execution.\n\tSuccess                        *shared.QueryWorkflowResponse          `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tQueryFailedError               *shared.QueryFailedError               `json:\"queryFailedError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_QueryWorkflow_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_QueryWorkflow_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tw, err = v.QueryFailedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_QueryWorkflow_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowResponse_Read(w wire.Value) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Read(w wire.Value) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_QueryWorkflow_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_QueryWorkflow_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_QueryWorkflow_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_QueryWorkflow_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _QueryWorkflowResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryFailedError, err = _QueryFailedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_QueryWorkflow_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_QueryWorkflow_Result struct could not be encoded.\nfunc (v *WorkflowService_QueryWorkflow_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryFailedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryFailedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowResponse_Decode(sr stream.Reader) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Decode(sr stream.Reader) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_QueryWorkflow_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_QueryWorkflow_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_QueryWorkflow_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _QueryWorkflowResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.QueryFailedError, err = _QueryFailedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_QueryWorkflow_Result\n// struct.\nfunc (v *WorkflowService_QueryWorkflow_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryFailedError: %v\", v.QueryFailedError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_QueryWorkflow_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_QueryWorkflow_Result match the\n// provided WorkflowService_QueryWorkflow_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_QueryWorkflow_Result) Equals(rhs *WorkflowService_QueryWorkflow_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.QueryFailedError == nil && rhs.QueryFailedError == nil) || (v.QueryFailedError != nil && rhs.QueryFailedError != nil && v.QueryFailedError.Equals(rhs.QueryFailedError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_QueryWorkflow_Result.\nfunc (v *WorkflowService_QueryWorkflow_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.QueryFailedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryFailedError\", v.QueryFailedError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetSuccess() (o *shared.QueryWorkflowResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetQueryFailedError returns the value of QueryFailedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetQueryFailedError() (o *shared.QueryFailedError) {\n\tif v != nil && v.QueryFailedError != nil {\n\t\treturn v.QueryFailedError\n\t}\n\n\treturn\n}\n\n// IsSetQueryFailedError returns true if QueryFailedError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetQueryFailedError() bool {\n\treturn v != nil && v.QueryFailedError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_QueryWorkflow_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_QueryWorkflow_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *WorkflowService_QueryWorkflow_Result) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_QueryWorkflow_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RecordActivityTaskHeartbeat_Args represents the arguments for the WorkflowService.RecordActivityTaskHeartbeat function.\n//\n// The arguments for RecordActivityTaskHeartbeat are sent and received over the wire as this struct.\ntype WorkflowService_RecordActivityTaskHeartbeat_Args struct {\n\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest `json:\"heartbeatRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RecordActivityTaskHeartbeat_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HeartbeatRequest != nil {\n\t\tw, err = v.HeartbeatRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_Read(w wire.Value) (*shared.RecordActivityTaskHeartbeatRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RecordActivityTaskHeartbeat_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RecordActivityTaskHeartbeat_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RecordActivityTaskHeartbeat_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RecordActivityTaskHeartbeat_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeat_Args struct could not be encoded.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HeartbeatRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HeartbeatRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_Decode(sr stream.Reader) (*shared.RecordActivityTaskHeartbeatRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RecordActivityTaskHeartbeat_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeat_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RecordActivityTaskHeartbeat_Args\n// struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.HeartbeatRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatRequest: %v\", v.HeartbeatRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RecordActivityTaskHeartbeat_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RecordActivityTaskHeartbeat_Args match the\n// provided WorkflowService_RecordActivityTaskHeartbeat_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) Equals(rhs *WorkflowService_RecordActivityTaskHeartbeat_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatRequest == nil && rhs.HeartbeatRequest == nil) || (v.HeartbeatRequest != nil && rhs.HeartbeatRequest != nil && v.HeartbeatRequest.Equals(rhs.HeartbeatRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RecordActivityTaskHeartbeat_Args.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"heartbeatRequest\", v.HeartbeatRequest))\n\t}\n\treturn err\n}\n\n// GetHeartbeatRequest returns the value of HeartbeatRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) GetHeartbeatRequest() (o *shared.RecordActivityTaskHeartbeatRequest) {\n\tif v != nil && v.HeartbeatRequest != nil {\n\t\treturn v.HeartbeatRequest\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatRequest returns true if HeartbeatRequest is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) IsSetHeartbeatRequest() bool {\n\treturn v != nil && v.HeartbeatRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordActivityTaskHeartbeat\" for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeat\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RecordActivityTaskHeartbeat_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RecordActivityTaskHeartbeat\n// function.\nvar WorkflowService_RecordActivityTaskHeartbeat_Helper = struct {\n\t// Args accepts the parameters of RecordActivityTaskHeartbeat in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\theartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\t) *WorkflowService_RecordActivityTaskHeartbeat_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordActivityTaskHeartbeat.\n\t//\n\t// An error can be thrown by RecordActivityTaskHeartbeat only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordActivityTaskHeartbeat\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RecordActivityTaskHeartbeat into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RecordActivityTaskHeartbeat\n\t//\n\t//   value, err := RecordActivityTaskHeartbeat(args)\n\t//   result, err := WorkflowService_RecordActivityTaskHeartbeat_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordActivityTaskHeartbeat: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RecordActivityTaskHeartbeatResponse, error) (*WorkflowService_RecordActivityTaskHeartbeat_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordActivityTaskHeartbeat\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RecordActivityTaskHeartbeat threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RecordActivityTaskHeartbeat_Result) (*shared.RecordActivityTaskHeartbeatResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_RecordActivityTaskHeartbeat_Helper.Args = func(\n\t\theartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\t) *WorkflowService_RecordActivityTaskHeartbeat_Args {\n\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Args{\n\t\t\tHeartbeatRequest: heartbeatRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RecordActivityTaskHeartbeat_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RecordActivityTaskHeartbeat_Helper.WrapResponse = func(success *shared.RecordActivityTaskHeartbeatResponse, err error) (*WorkflowService_RecordActivityTaskHeartbeat_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeat_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeat_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse = func(result *WorkflowService_RecordActivityTaskHeartbeat_Result) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RecordActivityTaskHeartbeat_Result represents the result of a WorkflowService.RecordActivityTaskHeartbeat function call.\n//\n// The result of a RecordActivityTaskHeartbeat execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_RecordActivityTaskHeartbeat_Result struct {\n\t// Value returned by RecordActivityTaskHeartbeat after a successful execution.\n\tSuccess                                *shared.RecordActivityTaskHeartbeatResponse    `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RecordActivityTaskHeartbeat_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatResponse_Read(w wire.Value) (*shared.RecordActivityTaskHeartbeatResponse, error) {\n\tvar v shared.RecordActivityTaskHeartbeatResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyCompletedError_Read(w wire.Value) (*shared.WorkflowExecutionAlreadyCompletedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyCompletedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RecordActivityTaskHeartbeat_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RecordActivityTaskHeartbeat_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RecordActivityTaskHeartbeat_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RecordActivityTaskHeartbeat_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeat_Result struct could not be encoded.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatResponse_Decode(sr stream.Reader) (*shared.RecordActivityTaskHeartbeatResponse, error) {\n\tvar v shared.RecordActivityTaskHeartbeatResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyCompletedError_Decode(sr stream.Reader) (*shared.WorkflowExecutionAlreadyCompletedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyCompletedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RecordActivityTaskHeartbeat_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeat_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RecordActivityTaskHeartbeat_Result\n// struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RecordActivityTaskHeartbeat_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RecordActivityTaskHeartbeat_Result match the\n// provided WorkflowService_RecordActivityTaskHeartbeat_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) Equals(rhs *WorkflowService_RecordActivityTaskHeartbeat_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RecordActivityTaskHeartbeat_Result.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetSuccess() (o *shared.RecordActivityTaskHeartbeatResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordActivityTaskHeartbeat\" for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeat\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeat_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RecordActivityTaskHeartbeatByID_Args represents the arguments for the WorkflowService.RecordActivityTaskHeartbeatByID function.\n//\n// The arguments for RecordActivityTaskHeartbeatByID are sent and received over the wire as this struct.\ntype WorkflowService_RecordActivityTaskHeartbeatByID_Args struct {\n\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest `json:\"heartbeatRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HeartbeatRequest != nil {\n\t\tw, err = v.HeartbeatRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatByIDRequest_Read(w wire.Value) (*shared.RecordActivityTaskHeartbeatByIDRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatByIDRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RecordActivityTaskHeartbeatByID_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatByIDRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct could not be encoded.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HeartbeatRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HeartbeatRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatByIDRequest_Decode(sr stream.Reader) (*shared.RecordActivityTaskHeartbeatByIDRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatByIDRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeatByID_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatByIDRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RecordActivityTaskHeartbeatByID_Args\n// struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.HeartbeatRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatRequest: %v\", v.HeartbeatRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RecordActivityTaskHeartbeatByID_Args match the\n// provided WorkflowService_RecordActivityTaskHeartbeatByID_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) Equals(rhs *WorkflowService_RecordActivityTaskHeartbeatByID_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatRequest == nil && rhs.HeartbeatRequest == nil) || (v.HeartbeatRequest != nil && rhs.HeartbeatRequest != nil && v.HeartbeatRequest.Equals(rhs.HeartbeatRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RecordActivityTaskHeartbeatByID_Args.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"heartbeatRequest\", v.HeartbeatRequest))\n\t}\n\treturn err\n}\n\n// GetHeartbeatRequest returns the value of HeartbeatRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) GetHeartbeatRequest() (o *shared.RecordActivityTaskHeartbeatByIDRequest) {\n\tif v != nil && v.HeartbeatRequest != nil {\n\t\treturn v.HeartbeatRequest\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatRequest returns true if HeartbeatRequest is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) IsSetHeartbeatRequest() bool {\n\treturn v != nil && v.HeartbeatRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordActivityTaskHeartbeatByID\" for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeatByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RecordActivityTaskHeartbeatByID_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RecordActivityTaskHeartbeatByID\n// function.\nvar WorkflowService_RecordActivityTaskHeartbeatByID_Helper = struct {\n\t// Args accepts the parameters of RecordActivityTaskHeartbeatByID in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\theartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\t) *WorkflowService_RecordActivityTaskHeartbeatByID_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordActivityTaskHeartbeatByID.\n\t//\n\t// An error can be thrown by RecordActivityTaskHeartbeatByID only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordActivityTaskHeartbeatByID\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RecordActivityTaskHeartbeatByID into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RecordActivityTaskHeartbeatByID\n\t//\n\t//   value, err := RecordActivityTaskHeartbeatByID(args)\n\t//   result, err := WorkflowService_RecordActivityTaskHeartbeatByID_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordActivityTaskHeartbeatByID: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RecordActivityTaskHeartbeatResponse, error) (*WorkflowService_RecordActivityTaskHeartbeatByID_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordActivityTaskHeartbeatByID\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RecordActivityTaskHeartbeatByID threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_RecordActivityTaskHeartbeatByID_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RecordActivityTaskHeartbeatByID_Result) (*shared.RecordActivityTaskHeartbeatResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_RecordActivityTaskHeartbeatByID_Helper.Args = func(\n\t\theartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\t) *WorkflowService_RecordActivityTaskHeartbeatByID_Args {\n\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Args{\n\t\t\tHeartbeatRequest: heartbeatRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RecordActivityTaskHeartbeatByID_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RecordActivityTaskHeartbeatByID_Helper.WrapResponse = func(success *shared.RecordActivityTaskHeartbeatResponse, err error) (*WorkflowService_RecordActivityTaskHeartbeatByID_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RecordActivityTaskHeartbeatByID_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RecordActivityTaskHeartbeatByID_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RecordActivityTaskHeartbeatByID_Helper.UnwrapResponse = func(result *WorkflowService_RecordActivityTaskHeartbeatByID_Result) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RecordActivityTaskHeartbeatByID_Result represents the result of a WorkflowService.RecordActivityTaskHeartbeatByID function call.\n//\n// The result of a RecordActivityTaskHeartbeatByID execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_RecordActivityTaskHeartbeatByID_Result struct {\n\t// Value returned by RecordActivityTaskHeartbeatByID after a successful execution.\n\tSuccess                                *shared.RecordActivityTaskHeartbeatResponse    `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RecordActivityTaskHeartbeatByID_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct could not be encoded.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RecordActivityTaskHeartbeatByID_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RecordActivityTaskHeartbeatByID_Result\n// struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RecordActivityTaskHeartbeatByID_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RecordActivityTaskHeartbeatByID_Result match the\n// provided WorkflowService_RecordActivityTaskHeartbeatByID_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) Equals(rhs *WorkflowService_RecordActivityTaskHeartbeatByID_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RecordActivityTaskHeartbeatByID_Result.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetSuccess() (o *shared.RecordActivityTaskHeartbeatResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordActivityTaskHeartbeatByID\" for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeatByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RecordActivityTaskHeartbeatByID_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RefreshWorkflowTasks_Args represents the arguments for the WorkflowService.RefreshWorkflowTasks function.\n//\n// The arguments for RefreshWorkflowTasks are sent and received over the wire as this struct.\ntype WorkflowService_RefreshWorkflowTasks_Args struct {\n\tRequest *shared.RefreshWorkflowTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RefreshWorkflowTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RefreshWorkflowTasksRequest_Read(w wire.Value) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RefreshWorkflowTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RefreshWorkflowTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RefreshWorkflowTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RefreshWorkflowTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RefreshWorkflowTasks_Args struct could not be encoded.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RefreshWorkflowTasksRequest_Decode(sr stream.Reader) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RefreshWorkflowTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RefreshWorkflowTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RefreshWorkflowTasks_Args\n// struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RefreshWorkflowTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RefreshWorkflowTasks_Args match the\n// provided WorkflowService_RefreshWorkflowTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) Equals(rhs *WorkflowService_RefreshWorkflowTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RefreshWorkflowTasks_Args.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) GetRequest() (o *shared.RefreshWorkflowTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RefreshWorkflowTasks_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RefreshWorkflowTasks\n// function.\nvar WorkflowService_RefreshWorkflowTasks_Helper = struct {\n\t// Args accepts the parameters of RefreshWorkflowTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RefreshWorkflowTasksRequest,\n\t) *WorkflowService_RefreshWorkflowTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RefreshWorkflowTasks.\n\t//\n\t// An error can be thrown by RefreshWorkflowTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RefreshWorkflowTasks\n\t// given the error returned by it. The provided error may\n\t// be nil if RefreshWorkflowTasks did not fail.\n\t//\n\t// This allows mapping errors returned by RefreshWorkflowTasks into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RefreshWorkflowTasks\n\t//\n\t//   err := RefreshWorkflowTasks(args)\n\t//   result, err := WorkflowService_RefreshWorkflowTasks_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RefreshWorkflowTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RefreshWorkflowTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for RefreshWorkflowTasks\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RefreshWorkflowTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RefreshWorkflowTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RefreshWorkflowTasks_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RefreshWorkflowTasks_Helper.Args = func(\n\t\trequest *shared.RefreshWorkflowTasksRequest,\n\t) *WorkflowService_RefreshWorkflowTasks_Args {\n\t\treturn &WorkflowService_RefreshWorkflowTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tWorkflowService_RefreshWorkflowTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RefreshWorkflowTasks_Helper.WrapResponse = func(err error) (*WorkflowService_RefreshWorkflowTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RefreshWorkflowTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RefreshWorkflowTasks_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RefreshWorkflowTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RefreshWorkflowTasks_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RefreshWorkflowTasks_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RefreshWorkflowTasks_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RefreshWorkflowTasks_Helper.UnwrapResponse = func(result *WorkflowService_RefreshWorkflowTasks_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RefreshWorkflowTasks_Result represents the result of a WorkflowService.RefreshWorkflowTasks function call.\n//\n// The result of a RefreshWorkflowTasks execution is sent and received over the wire as this struct.\ntype WorkflowService_RefreshWorkflowTasks_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tDomainNotActiveError *shared.DomainNotActiveError `json:\"domainNotActiveError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RefreshWorkflowTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RefreshWorkflowTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RefreshWorkflowTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RefreshWorkflowTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RefreshWorkflowTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RefreshWorkflowTasks_Result struct could not be encoded.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RefreshWorkflowTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RefreshWorkflowTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RefreshWorkflowTasks_Result\n// struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RefreshWorkflowTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RefreshWorkflowTasks_Result match the\n// provided WorkflowService_RefreshWorkflowTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) Equals(rhs *WorkflowService_RefreshWorkflowTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RefreshWorkflowTasks_Result.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RefreshWorkflowTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RegisterDomain_Args represents the arguments for the WorkflowService.RegisterDomain function.\n//\n// The arguments for RegisterDomain are sent and received over the wire as this struct.\ntype WorkflowService_RegisterDomain_Args struct {\n\tRegisterRequest *shared.RegisterDomainRequest `json:\"registerRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RegisterDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RegisterDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RegisterRequest != nil {\n\t\tw, err = v.RegisterRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RegisterDomainRequest_Read(w wire.Value) (*shared.RegisterDomainRequest, error) {\n\tvar v shared.RegisterDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RegisterDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RegisterDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RegisterDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RegisterDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RegisterRequest, err = _RegisterDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RegisterDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RegisterDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_RegisterDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RegisterRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RegisterRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RegisterDomainRequest_Decode(sr stream.Reader) (*shared.RegisterDomainRequest, error) {\n\tvar v shared.RegisterDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RegisterDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RegisterDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RegisterDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.RegisterRequest, err = _RegisterDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RegisterDomain_Args\n// struct.\nfunc (v *WorkflowService_RegisterDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RegisterRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"RegisterRequest: %v\", v.RegisterRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RegisterDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RegisterDomain_Args match the\n// provided WorkflowService_RegisterDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RegisterDomain_Args) Equals(rhs *WorkflowService_RegisterDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.RegisterRequest == nil && rhs.RegisterRequest == nil) || (v.RegisterRequest != nil && rhs.RegisterRequest != nil && v.RegisterRequest.Equals(rhs.RegisterRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RegisterDomain_Args.\nfunc (v *WorkflowService_RegisterDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RegisterRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"registerRequest\", v.RegisterRequest))\n\t}\n\treturn err\n}\n\n// GetRegisterRequest returns the value of RegisterRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Args) GetRegisterRequest() (o *shared.RegisterDomainRequest) {\n\tif v != nil && v.RegisterRequest != nil {\n\t\treturn v.RegisterRequest\n\t}\n\n\treturn\n}\n\n// IsSetRegisterRequest returns true if RegisterRequest is not nil.\nfunc (v *WorkflowService_RegisterDomain_Args) IsSetRegisterRequest() bool {\n\treturn v != nil && v.RegisterRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RegisterDomain\" for this struct.\nfunc (v *WorkflowService_RegisterDomain_Args) MethodName() string {\n\treturn \"RegisterDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RegisterDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RegisterDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RegisterDomain\n// function.\nvar WorkflowService_RegisterDomain_Helper = struct {\n\t// Args accepts the parameters of RegisterDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tregisterRequest *shared.RegisterDomainRequest,\n\t) *WorkflowService_RegisterDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RegisterDomain.\n\t//\n\t// An error can be thrown by RegisterDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RegisterDomain\n\t// given the error returned by it. The provided error may\n\t// be nil if RegisterDomain did not fail.\n\t//\n\t// This allows mapping errors returned by RegisterDomain into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RegisterDomain\n\t//\n\t//   err := RegisterDomain(args)\n\t//   result, err := WorkflowService_RegisterDomain_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RegisterDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RegisterDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for RegisterDomain\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RegisterDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RegisterDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RegisterDomain_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RegisterDomain_Helper.Args = func(\n\t\tregisterRequest *shared.RegisterDomainRequest,\n\t) *WorkflowService_RegisterDomain_Args {\n\t\treturn &WorkflowService_RegisterDomain_Args{\n\t\t\tRegisterRequest: registerRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RegisterDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.DomainAlreadyExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RegisterDomain_Helper.WrapResponse = func(err error) (*WorkflowService_RegisterDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RegisterDomain_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RegisterDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RegisterDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.DomainAlreadyExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RegisterDomain_Result.DomainExistsError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RegisterDomain_Result{DomainExistsError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RegisterDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RegisterDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RegisterDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RegisterDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RegisterDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RegisterDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RegisterDomain_Helper.UnwrapResponse = func(result *WorkflowService_RegisterDomain_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainExistsError != nil {\n\t\t\terr = result.DomainExistsError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RegisterDomain_Result represents the result of a WorkflowService.RegisterDomain function call.\n//\n// The result of a RegisterDomain execution is sent and received over the wire as this struct.\ntype WorkflowService_RegisterDomain_Result struct {\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tDomainExistsError              *shared.DomainAlreadyExistsError       `json:\"domainExistsError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RegisterDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RegisterDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.DomainExistsError != nil {\n\t\tw, err = v.DomainExistsError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RegisterDomain_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainAlreadyExistsError_Read(w wire.Value) (*shared.DomainAlreadyExistsError, error) {\n\tvar v shared.DomainAlreadyExistsError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RegisterDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RegisterDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RegisterDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RegisterDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainExistsError, err = _DomainAlreadyExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainExistsError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RegisterDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RegisterDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RegisterDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_RegisterDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainExistsError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainExistsError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainExistsError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RegisterDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainAlreadyExistsError_Decode(sr stream.Reader) (*shared.DomainAlreadyExistsError, error) {\n\tvar v shared.DomainAlreadyExistsError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RegisterDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RegisterDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RegisterDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.DomainExistsError, err = _DomainAlreadyExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.DomainExistsError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RegisterDomain_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RegisterDomain_Result\n// struct.\nfunc (v *WorkflowService_RegisterDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.DomainExistsError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainExistsError: %v\", v.DomainExistsError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RegisterDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RegisterDomain_Result match the\n// provided WorkflowService_RegisterDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RegisterDomain_Result) Equals(rhs *WorkflowService_RegisterDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainExistsError == nil && rhs.DomainExistsError == nil) || (v.DomainExistsError != nil && rhs.DomainExistsError != nil && v.DomainExistsError.Equals(rhs.DomainExistsError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RegisterDomain_Result.\nfunc (v *WorkflowService_RegisterDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.DomainExistsError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainExistsError\", v.DomainExistsError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RegisterDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetDomainExistsError returns the value of DomainExistsError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Result) GetDomainExistsError() (o *shared.DomainAlreadyExistsError) {\n\tif v != nil && v.DomainExistsError != nil {\n\t\treturn v.DomainExistsError\n\t}\n\n\treturn\n}\n\n// IsSetDomainExistsError returns true if DomainExistsError is not nil.\nfunc (v *WorkflowService_RegisterDomain_Result) IsSetDomainExistsError() bool {\n\treturn v != nil && v.DomainExistsError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RegisterDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RegisterDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RegisterDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RegisterDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RegisterDomain\" for this struct.\nfunc (v *WorkflowService_RegisterDomain_Result) MethodName() string {\n\treturn \"RegisterDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RegisterDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RequestCancelWorkflowExecution_Args represents the arguments for the WorkflowService.RequestCancelWorkflowExecution function.\n//\n// The arguments for RequestCancelWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_RequestCancelWorkflowExecution_Args struct {\n\tCancelRequest *shared.RequestCancelWorkflowExecutionRequest `json:\"cancelRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RequestCancelWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CancelRequest != nil {\n\t\tw, err = v.CancelRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_Read(w wire.Value) (*shared.RequestCancelWorkflowExecutionRequest, error) {\n\tvar v shared.RequestCancelWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RequestCancelWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RequestCancelWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RequestCancelWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RequestCancelWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RequestCancelWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CancelRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.RequestCancelWorkflowExecutionRequest, error) {\n\tvar v shared.RequestCancelWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RequestCancelWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RequestCancelWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RequestCancelWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CancelRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequest: %v\", v.CancelRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RequestCancelWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RequestCancelWorkflowExecution_Args match the\n// provided WorkflowService_RequestCancelWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) Equals(rhs *WorkflowService_RequestCancelWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CancelRequest == nil && rhs.CancelRequest == nil) || (v.CancelRequest != nil && rhs.CancelRequest != nil && v.CancelRequest.Equals(rhs.CancelRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RequestCancelWorkflowExecution_Args.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CancelRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelRequest\", v.CancelRequest))\n\t}\n\treturn err\n}\n\n// GetCancelRequest returns the value of CancelRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) GetCancelRequest() (o *shared.RequestCancelWorkflowExecutionRequest) {\n\tif v != nil && v.CancelRequest != nil {\n\t\treturn v.CancelRequest\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequest returns true if CancelRequest is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) IsSetCancelRequest() bool {\n\treturn v != nil && v.CancelRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RequestCancelWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) MethodName() string {\n\treturn \"RequestCancelWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RequestCancelWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RequestCancelWorkflowExecution\n// function.\nvar WorkflowService_RequestCancelWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of RequestCancelWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\t) *WorkflowService_RequestCancelWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RequestCancelWorkflowExecution.\n\t//\n\t// An error can be thrown by RequestCancelWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RequestCancelWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if RequestCancelWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by RequestCancelWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RequestCancelWorkflowExecution\n\t//\n\t//   err := RequestCancelWorkflowExecution(args)\n\t//   result, err := WorkflowService_RequestCancelWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RequestCancelWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RequestCancelWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for RequestCancelWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RequestCancelWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RequestCancelWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RequestCancelWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RequestCancelWorkflowExecution_Helper.Args = func(\n\t\tcancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\t) *WorkflowService_RequestCancelWorkflowExecution_Args {\n\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Args{\n\t\t\tCancelRequest: cancelRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RequestCancelWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.CancellationAlreadyRequestedError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RequestCancelWorkflowExecution_Helper.WrapResponse = func(err error) (*WorkflowService_RequestCancelWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.CancellationAlreadyRequestedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.CancellationAlreadyRequestedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{CancellationAlreadyRequestedError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RequestCancelWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RequestCancelWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RequestCancelWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_RequestCancelWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.CancellationAlreadyRequestedError != nil {\n\t\t\terr = result.CancellationAlreadyRequestedError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RequestCancelWorkflowExecution_Result represents the result of a WorkflowService.RequestCancelWorkflowExecution function call.\n//\n// The result of a RequestCancelWorkflowExecution execution is sent and received over the wire as this struct.\ntype WorkflowService_RequestCancelWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tCancellationAlreadyRequestedError      *shared.CancellationAlreadyRequestedError      `json:\"cancellationAlreadyRequestedError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RequestCancelWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tw, err = v.CancellationAlreadyRequestedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CancellationAlreadyRequestedError_Read(w wire.Value) (*shared.CancellationAlreadyRequestedError, error) {\n\tvar v shared.CancellationAlreadyRequestedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RequestCancelWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RequestCancelWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RequestCancelWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancellationAlreadyRequestedError, err = _CancellationAlreadyRequestedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RequestCancelWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RequestCancelWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancellationAlreadyRequestedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CancellationAlreadyRequestedError_Decode(sr stream.Reader) (*shared.CancellationAlreadyRequestedError, error) {\n\tvar v shared.CancellationAlreadyRequestedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RequestCancelWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RequestCancelWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.CancellationAlreadyRequestedError, err = _CancellationAlreadyRequestedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RequestCancelWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancellationAlreadyRequestedError: %v\", v.CancellationAlreadyRequestedError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RequestCancelWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RequestCancelWorkflowExecution_Result match the\n// provided WorkflowService_RequestCancelWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) Equals(rhs *WorkflowService_RequestCancelWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.CancellationAlreadyRequestedError == nil && rhs.CancellationAlreadyRequestedError == nil) || (v.CancellationAlreadyRequestedError != nil && rhs.CancellationAlreadyRequestedError != nil && v.CancellationAlreadyRequestedError.Equals(rhs.CancellationAlreadyRequestedError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RequestCancelWorkflowExecution_Result.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancellationAlreadyRequestedError\", v.CancellationAlreadyRequestedError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetCancellationAlreadyRequestedError returns the value of CancellationAlreadyRequestedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetCancellationAlreadyRequestedError() (o *shared.CancellationAlreadyRequestedError) {\n\tif v != nil && v.CancellationAlreadyRequestedError != nil {\n\t\treturn v.CancellationAlreadyRequestedError\n\t}\n\n\treturn\n}\n\n// IsSetCancellationAlreadyRequestedError returns true if CancellationAlreadyRequestedError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetCancellationAlreadyRequestedError() bool {\n\treturn v != nil && v.CancellationAlreadyRequestedError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RequestCancelWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) MethodName() string {\n\treturn \"RequestCancelWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RequestCancelWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ResetStickyTaskList_Args represents the arguments for the WorkflowService.ResetStickyTaskList function.\n//\n// The arguments for ResetStickyTaskList are sent and received over the wire as this struct.\ntype WorkflowService_ResetStickyTaskList_Args struct {\n\tResetRequest *shared.ResetStickyTaskListRequest `json:\"resetRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ResetStickyTaskList_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ResetStickyTaskList_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ResetRequest != nil {\n\t\tw, err = v.ResetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetStickyTaskListRequest_Read(w wire.Value) (*shared.ResetStickyTaskListRequest, error) {\n\tvar v shared.ResetStickyTaskListRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ResetStickyTaskList_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ResetStickyTaskList_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ResetStickyTaskList_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ResetStickyTaskList_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ResetRequest, err = _ResetStickyTaskListRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ResetStickyTaskList_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ResetStickyTaskList_Args struct could not be encoded.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ResetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetStickyTaskListRequest_Decode(sr stream.Reader) (*shared.ResetStickyTaskListRequest, error) {\n\tvar v shared.ResetStickyTaskListRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ResetStickyTaskList_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ResetStickyTaskList_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ResetRequest, err = _ResetStickyTaskListRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ResetStickyTaskList_Args\n// struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ResetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResetRequest: %v\", v.ResetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ResetStickyTaskList_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ResetStickyTaskList_Args match the\n// provided WorkflowService_ResetStickyTaskList_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) Equals(rhs *WorkflowService_ResetStickyTaskList_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ResetRequest == nil && rhs.ResetRequest == nil) || (v.ResetRequest != nil && rhs.ResetRequest != nil && v.ResetRequest.Equals(rhs.ResetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ResetStickyTaskList_Args.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ResetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resetRequest\", v.ResetRequest))\n\t}\n\treturn err\n}\n\n// GetResetRequest returns the value of ResetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) GetResetRequest() (o *shared.ResetStickyTaskListRequest) {\n\tif v != nil && v.ResetRequest != nil {\n\t\treturn v.ResetRequest\n\t}\n\n\treturn\n}\n\n// IsSetResetRequest returns true if ResetRequest is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) IsSetResetRequest() bool {\n\treturn v != nil && v.ResetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetStickyTaskList\" for this struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) MethodName() string {\n\treturn \"ResetStickyTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ResetStickyTaskList_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ResetStickyTaskList\n// function.\nvar WorkflowService_ResetStickyTaskList_Helper = struct {\n\t// Args accepts the parameters of ResetStickyTaskList in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tresetRequest *shared.ResetStickyTaskListRequest,\n\t) *WorkflowService_ResetStickyTaskList_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetStickyTaskList.\n\t//\n\t// An error can be thrown by ResetStickyTaskList only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetStickyTaskList\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ResetStickyTaskList into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ResetStickyTaskList\n\t//\n\t//   value, err := ResetStickyTaskList(args)\n\t//   result, err := WorkflowService_ResetStickyTaskList_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetStickyTaskList: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ResetStickyTaskListResponse, error) (*WorkflowService_ResetStickyTaskList_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetStickyTaskList\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ResetStickyTaskList threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ResetStickyTaskList_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ResetStickyTaskList_Result) (*shared.ResetStickyTaskListResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ResetStickyTaskList_Helper.Args = func(\n\t\tresetRequest *shared.ResetStickyTaskListRequest,\n\t) *WorkflowService_ResetStickyTaskList_Args {\n\t\treturn &WorkflowService_ResetStickyTaskList_Args{\n\t\t\tResetRequest: resetRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ResetStickyTaskList_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ResetStickyTaskList_Helper.WrapResponse = func(success *shared.ResetStickyTaskListResponse, err error) (*WorkflowService_ResetStickyTaskList_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetStickyTaskList_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetStickyTaskList_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ResetStickyTaskList_Helper.UnwrapResponse = func(result *WorkflowService_ResetStickyTaskList_Result) (success *shared.ResetStickyTaskListResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ResetStickyTaskList_Result represents the result of a WorkflowService.ResetStickyTaskList function call.\n//\n// The result of a ResetStickyTaskList execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ResetStickyTaskList_Result struct {\n\t// Value returned by ResetStickyTaskList after a successful execution.\n\tSuccess                                *shared.ResetStickyTaskListResponse            `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ResetStickyTaskList_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ResetStickyTaskList_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetStickyTaskListResponse_Read(w wire.Value) (*shared.ResetStickyTaskListResponse, error) {\n\tvar v shared.ResetStickyTaskListResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ResetStickyTaskList_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ResetStickyTaskList_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ResetStickyTaskList_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ResetStickyTaskList_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ResetStickyTaskListResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ResetStickyTaskList_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ResetStickyTaskList_Result struct could not be encoded.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetStickyTaskListResponse_Decode(sr stream.Reader) (*shared.ResetStickyTaskListResponse, error) {\n\tvar v shared.ResetStickyTaskListResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ResetStickyTaskList_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ResetStickyTaskList_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ResetStickyTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ResetStickyTaskList_Result\n// struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ResetStickyTaskList_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ResetStickyTaskList_Result match the\n// provided WorkflowService_ResetStickyTaskList_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) Equals(rhs *WorkflowService_ResetStickyTaskList_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ResetStickyTaskList_Result.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetSuccess() (o *shared.ResetStickyTaskListResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetStickyTaskList\" for this struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) MethodName() string {\n\treturn \"ResetStickyTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ResetStickyTaskList_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ResetWorkflowExecution_Args represents the arguments for the WorkflowService.ResetWorkflowExecution function.\n//\n// The arguments for ResetWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_ResetWorkflowExecution_Args struct {\n\tResetRequest *shared.ResetWorkflowExecutionRequest `json:\"resetRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ResetWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ResetRequest != nil {\n\t\tw, err = v.ResetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetWorkflowExecutionRequest_Read(w wire.Value) (*shared.ResetWorkflowExecutionRequest, error) {\n\tvar v shared.ResetWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ResetWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ResetWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ResetWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ResetWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ResetWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ResetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.ResetWorkflowExecutionRequest, error) {\n\tvar v shared.ResetWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ResetWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ResetWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ResetWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ResetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResetRequest: %v\", v.ResetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ResetWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ResetWorkflowExecution_Args match the\n// provided WorkflowService_ResetWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) Equals(rhs *WorkflowService_ResetWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ResetRequest == nil && rhs.ResetRequest == nil) || (v.ResetRequest != nil && rhs.ResetRequest != nil && v.ResetRequest.Equals(rhs.ResetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ResetWorkflowExecution_Args.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ResetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resetRequest\", v.ResetRequest))\n\t}\n\treturn err\n}\n\n// GetResetRequest returns the value of ResetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) GetResetRequest() (o *shared.ResetWorkflowExecutionRequest) {\n\tif v != nil && v.ResetRequest != nil {\n\t\treturn v.ResetRequest\n\t}\n\n\treturn\n}\n\n// IsSetResetRequest returns true if ResetRequest is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) IsSetResetRequest() bool {\n\treturn v != nil && v.ResetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) MethodName() string {\n\treturn \"ResetWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ResetWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ResetWorkflowExecution\n// function.\nvar WorkflowService_ResetWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of ResetWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tresetRequest *shared.ResetWorkflowExecutionRequest,\n\t) *WorkflowService_ResetWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetWorkflowExecution.\n\t//\n\t// An error can be thrown by ResetWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ResetWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ResetWorkflowExecution\n\t//\n\t//   value, err := ResetWorkflowExecution(args)\n\t//   result, err := WorkflowService_ResetWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ResetWorkflowExecutionResponse, error) (*WorkflowService_ResetWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ResetWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ResetWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ResetWorkflowExecution_Result) (*shared.ResetWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ResetWorkflowExecution_Helper.Args = func(\n\t\tresetRequest *shared.ResetWorkflowExecutionRequest,\n\t) *WorkflowService_ResetWorkflowExecution_Args {\n\t\treturn &WorkflowService_ResetWorkflowExecution_Args{\n\t\t\tResetRequest: resetRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ResetWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ResetWorkflowExecution_Helper.WrapResponse = func(success *shared.ResetWorkflowExecutionResponse, err error) (*WorkflowService_ResetWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ResetWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ResetWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ResetWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_ResetWorkflowExecution_Result) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ResetWorkflowExecution_Result represents the result of a WorkflowService.ResetWorkflowExecution function call.\n//\n// The result of a ResetWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ResetWorkflowExecution_Result struct {\n\t// Value returned by ResetWorkflowExecution after a successful execution.\n\tSuccess                        *shared.ResetWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ResetWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetWorkflowExecutionResponse_Read(w wire.Value) (*shared.ResetWorkflowExecutionResponse, error) {\n\tvar v shared.ResetWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_ResetWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ResetWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ResetWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ResetWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ResetWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ResetWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.ResetWorkflowExecutionResponse, error) {\n\tvar v shared.ResetWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_ResetWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ResetWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ResetWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ResetWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ResetWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ResetWorkflowExecution_Result match the\n// provided WorkflowService_ResetWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) Equals(rhs *WorkflowService_ResetWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ResetWorkflowExecution_Result.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetSuccess() (o *shared.ResetWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) MethodName() string {\n\treturn \"ResetWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ResetWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskCanceled_Args represents the arguments for the WorkflowService.RespondActivityTaskCanceled function.\n//\n// The arguments for RespondActivityTaskCanceled are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCanceled_Args struct {\n\tCanceledRequest *shared.RespondActivityTaskCanceledRequest `json:\"canceledRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCanceled_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CanceledRequest != nil {\n\t\tw, err = v.CanceledRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCanceledRequest_Read(w wire.Value) (*shared.RespondActivityTaskCanceledRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCanceled_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCanceled_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCanceled_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCanceled_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceled_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CanceledRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CanceledRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCanceledRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCanceledRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCanceled_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceled_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCanceled_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CanceledRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CanceledRequest: %v\", v.CanceledRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCanceled_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCanceled_Args match the\n// provided WorkflowService_RespondActivityTaskCanceled_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) Equals(rhs *WorkflowService_RespondActivityTaskCanceled_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CanceledRequest == nil && rhs.CanceledRequest == nil) || (v.CanceledRequest != nil && rhs.CanceledRequest != nil && v.CanceledRequest.Equals(rhs.CanceledRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCanceled_Args.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CanceledRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"canceledRequest\", v.CanceledRequest))\n\t}\n\treturn err\n}\n\n// GetCanceledRequest returns the value of CanceledRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) GetCanceledRequest() (o *shared.RespondActivityTaskCanceledRequest) {\n\tif v != nil && v.CanceledRequest != nil {\n\t\treturn v.CanceledRequest\n\t}\n\n\treturn\n}\n\n// IsSetCanceledRequest returns true if CanceledRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) IsSetCanceledRequest() bool {\n\treturn v != nil && v.CanceledRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCanceled\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) MethodName() string {\n\treturn \"RespondActivityTaskCanceled\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskCanceled_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskCanceled\n// function.\nvar WorkflowService_RespondActivityTaskCanceled_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCanceled in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\t) *WorkflowService_RespondActivityTaskCanceled_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCanceled.\n\t//\n\t// An error can be thrown by RespondActivityTaskCanceled only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCanceled\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCanceled did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCanceled into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCanceled\n\t//\n\t//   err := RespondActivityTaskCanceled(args)\n\t//   result, err := WorkflowService_RespondActivityTaskCanceled_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCanceled: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskCanceled_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCanceled\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCanceled threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskCanceled_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskCanceled_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskCanceled_Helper.Args = func(\n\t\tcanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\t) *WorkflowService_RespondActivityTaskCanceled_Args {\n\t\treturn &WorkflowService_RespondActivityTaskCanceled_Args{\n\t\t\tCanceledRequest: canceledRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCanceled_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCanceled_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskCanceled_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceled_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceled_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskCanceled_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskCanceled_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskCanceled_Result represents the result of a WorkflowService.RespondActivityTaskCanceled function call.\n//\n// The result of a RespondActivityTaskCanceled execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCanceled_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCanceled_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCanceled_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCanceled_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCanceled_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCanceled_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceled_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCanceled_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceled_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCanceled_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCanceled_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCanceled_Result match the\n// provided WorkflowService_RespondActivityTaskCanceled_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) Equals(rhs *WorkflowService_RespondActivityTaskCanceled_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCanceled_Result.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCanceled\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) MethodName() string {\n\treturn \"RespondActivityTaskCanceled\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceled_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskCanceledByID_Args represents the arguments for the WorkflowService.RespondActivityTaskCanceledByID function.\n//\n// The arguments for RespondActivityTaskCanceledByID are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCanceledByID_Args struct {\n\tCanceledRequest *shared.RespondActivityTaskCanceledByIDRequest `json:\"canceledRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCanceledByID_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CanceledRequest != nil {\n\t\tw, err = v.CanceledRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCanceledByIDRequest_Read(w wire.Value) (*shared.RespondActivityTaskCanceledByIDRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledByIDRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCanceledByID_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCanceledByID_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCanceledByID_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledByIDRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCanceledByID_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceledByID_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CanceledRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CanceledRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCanceledByIDRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCanceledByIDRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledByIDRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCanceledByID_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceledByID_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledByIDRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCanceledByID_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CanceledRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CanceledRequest: %v\", v.CanceledRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCanceledByID_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCanceledByID_Args match the\n// provided WorkflowService_RespondActivityTaskCanceledByID_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) Equals(rhs *WorkflowService_RespondActivityTaskCanceledByID_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CanceledRequest == nil && rhs.CanceledRequest == nil) || (v.CanceledRequest != nil && rhs.CanceledRequest != nil && v.CanceledRequest.Equals(rhs.CanceledRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCanceledByID_Args.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CanceledRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"canceledRequest\", v.CanceledRequest))\n\t}\n\treturn err\n}\n\n// GetCanceledRequest returns the value of CanceledRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) GetCanceledRequest() (o *shared.RespondActivityTaskCanceledByIDRequest) {\n\tif v != nil && v.CanceledRequest != nil {\n\t\treturn v.CanceledRequest\n\t}\n\n\treturn\n}\n\n// IsSetCanceledRequest returns true if CanceledRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) IsSetCanceledRequest() bool {\n\treturn v != nil && v.CanceledRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCanceledByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) MethodName() string {\n\treturn \"RespondActivityTaskCanceledByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskCanceledByID_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskCanceledByID\n// function.\nvar WorkflowService_RespondActivityTaskCanceledByID_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCanceledByID in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\t) *WorkflowService_RespondActivityTaskCanceledByID_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCanceledByID.\n\t//\n\t// An error can be thrown by RespondActivityTaskCanceledByID only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCanceledByID\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCanceledByID did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCanceledByID into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCanceledByID\n\t//\n\t//   err := RespondActivityTaskCanceledByID(args)\n\t//   result, err := WorkflowService_RespondActivityTaskCanceledByID_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCanceledByID: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskCanceledByID_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCanceledByID\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCanceledByID threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskCanceledByID_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskCanceledByID_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskCanceledByID_Helper.Args = func(\n\t\tcanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\t) *WorkflowService_RespondActivityTaskCanceledByID_Args {\n\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Args{\n\t\t\tCanceledRequest: canceledRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCanceledByID_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCanceledByID_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskCanceledByID_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCanceledByID_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCanceledByID_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskCanceledByID_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskCanceledByID_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskCanceledByID_Result represents the result of a WorkflowService.RespondActivityTaskCanceledByID function call.\n//\n// The result of a RespondActivityTaskCanceledByID execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCanceledByID_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCanceledByID_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceledByID_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCanceledByID_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCanceledByID_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCanceledByID_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceledByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCanceledByID_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceledByID_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceledByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCanceledByID_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCanceledByID_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCanceledByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCanceledByID_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCanceledByID_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCanceledByID_Result match the\n// provided WorkflowService_RespondActivityTaskCanceledByID_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) Equals(rhs *WorkflowService_RespondActivityTaskCanceledByID_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCanceledByID_Result.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCanceledByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) MethodName() string {\n\treturn \"RespondActivityTaskCanceledByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCanceledByID_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskCompleted_Args represents the arguments for the WorkflowService.RespondActivityTaskCompleted function.\n//\n// The arguments for RespondActivityTaskCompleted are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCompleted_Args struct {\n\tCompleteRequest *shared.RespondActivityTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCompletedRequest_Read(w wire.Value) (*shared.RespondActivityTaskCompletedRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompleted_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCompletedRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCompleted_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCompleted_Args match the\n// provided WorkflowService_RespondActivityTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) Equals(rhs *WorkflowService_RespondActivityTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCompleted_Args.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) GetCompleteRequest() (o *shared.RespondActivityTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) MethodName() string {\n\treturn \"RespondActivityTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskCompleted\n// function.\nvar WorkflowService_RespondActivityTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\t) *WorkflowService_RespondActivityTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCompleted.\n\t//\n\t// An error can be thrown by RespondActivityTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCompleted\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCompleted did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCompleted into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCompleted\n\t//\n\t//   err := RespondActivityTaskCompleted(args)\n\t//   result, err := WorkflowService_RespondActivityTaskCompleted_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCompleted\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskCompleted_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskCompleted_Helper.Args = func(\n\t\tcompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\t) *WorkflowService_RespondActivityTaskCompleted_Args {\n\t\treturn &WorkflowService_RespondActivityTaskCompleted_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCompleted_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompleted_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompleted_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskCompleted_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskCompleted_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskCompleted_Result represents the result of a WorkflowService.RespondActivityTaskCompleted function call.\n//\n// The result of a RespondActivityTaskCompleted execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCompleted_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompleted_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCompleted_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCompleted_Result match the\n// provided WorkflowService_RespondActivityTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) Equals(rhs *WorkflowService_RespondActivityTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCompleted_Result.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) MethodName() string {\n\treturn \"RespondActivityTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskCompletedByID_Args represents the arguments for the WorkflowService.RespondActivityTaskCompletedByID function.\n//\n// The arguments for RespondActivityTaskCompletedByID are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCompletedByID_Args struct {\n\tCompleteRequest *shared.RespondActivityTaskCompletedByIDRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCompletedByID_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCompletedByIDRequest_Read(w wire.Value) (*shared.RespondActivityTaskCompletedByIDRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedByIDRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCompletedByID_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCompletedByID_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCompletedByID_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedByIDRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCompletedByID_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompletedByID_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCompletedByIDRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCompletedByIDRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedByIDRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCompletedByID_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompletedByID_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedByIDRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCompletedByID_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCompletedByID_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCompletedByID_Args match the\n// provided WorkflowService_RespondActivityTaskCompletedByID_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) Equals(rhs *WorkflowService_RespondActivityTaskCompletedByID_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCompletedByID_Args.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) GetCompleteRequest() (o *shared.RespondActivityTaskCompletedByIDRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCompletedByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) MethodName() string {\n\treturn \"RespondActivityTaskCompletedByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskCompletedByID_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskCompletedByID\n// function.\nvar WorkflowService_RespondActivityTaskCompletedByID_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCompletedByID in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\t) *WorkflowService_RespondActivityTaskCompletedByID_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCompletedByID.\n\t//\n\t// An error can be thrown by RespondActivityTaskCompletedByID only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCompletedByID\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCompletedByID did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCompletedByID into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCompletedByID\n\t//\n\t//   err := RespondActivityTaskCompletedByID(args)\n\t//   result, err := WorkflowService_RespondActivityTaskCompletedByID_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCompletedByID: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskCompletedByID_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCompletedByID\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCompletedByID threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskCompletedByID_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskCompletedByID_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskCompletedByID_Helper.Args = func(\n\t\tcompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\t) *WorkflowService_RespondActivityTaskCompletedByID_Args {\n\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCompletedByID_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskCompletedByID_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskCompletedByID_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskCompletedByID_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskCompletedByID_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskCompletedByID_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskCompletedByID_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskCompletedByID_Result represents the result of a WorkflowService.RespondActivityTaskCompletedByID function call.\n//\n// The result of a RespondActivityTaskCompletedByID execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskCompletedByID_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskCompletedByID_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskCompletedByID_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskCompletedByID_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskCompletedByID_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskCompletedByID_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompletedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskCompletedByID_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompletedByID_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompletedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskCompletedByID_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskCompletedByID_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskCompletedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskCompletedByID_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskCompletedByID_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskCompletedByID_Result match the\n// provided WorkflowService_RespondActivityTaskCompletedByID_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) Equals(rhs *WorkflowService_RespondActivityTaskCompletedByID_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskCompletedByID_Result.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCompletedByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) MethodName() string {\n\treturn \"RespondActivityTaskCompletedByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskCompletedByID_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskFailed_Args represents the arguments for the WorkflowService.RespondActivityTaskFailed function.\n//\n// The arguments for RespondActivityTaskFailed are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskFailed_Args struct {\n\tFailRequest *shared.RespondActivityTaskFailedRequest `json:\"failRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskFailed_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailRequest != nil {\n\t\tw, err = v.FailRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskFailedRequest_Read(w wire.Value) (*shared.RespondActivityTaskFailedRequest, error) {\n\tvar v shared.RespondActivityTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskFailed_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskFailed_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskFailed_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailRequest, err = _RespondActivityTaskFailedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskFailed_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailed_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskFailedRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskFailedRequest, error) {\n\tvar v shared.RespondActivityTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskFailed_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailed_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailRequest, err = _RespondActivityTaskFailedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskFailed_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailRequest: %v\", v.FailRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskFailed_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskFailed_Args match the\n// provided WorkflowService_RespondActivityTaskFailed_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) Equals(rhs *WorkflowService_RespondActivityTaskFailed_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailRequest == nil && rhs.FailRequest == nil) || (v.FailRequest != nil && rhs.FailRequest != nil && v.FailRequest.Equals(rhs.FailRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskFailed_Args.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failRequest\", v.FailRequest))\n\t}\n\treturn err\n}\n\n// GetFailRequest returns the value of FailRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) GetFailRequest() (o *shared.RespondActivityTaskFailedRequest) {\n\tif v != nil && v.FailRequest != nil {\n\t\treturn v.FailRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailRequest returns true if FailRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) IsSetFailRequest() bool {\n\treturn v != nil && v.FailRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskFailed\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) MethodName() string {\n\treturn \"RespondActivityTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskFailed_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskFailed\n// function.\nvar WorkflowService_RespondActivityTaskFailed_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskFailed in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailRequest *shared.RespondActivityTaskFailedRequest,\n\t) *WorkflowService_RespondActivityTaskFailed_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskFailed.\n\t//\n\t// An error can be thrown by RespondActivityTaskFailed only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskFailed\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskFailed did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskFailed into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskFailed\n\t//\n\t//   err := RespondActivityTaskFailed(args)\n\t//   result, err := WorkflowService_RespondActivityTaskFailed_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskFailed: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskFailed_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskFailed\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskFailed threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskFailed_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskFailed_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskFailed_Helper.Args = func(\n\t\tfailRequest *shared.RespondActivityTaskFailedRequest,\n\t) *WorkflowService_RespondActivityTaskFailed_Args {\n\t\treturn &WorkflowService_RespondActivityTaskFailed_Args{\n\t\t\tFailRequest: failRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskFailed_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskFailed_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskFailed_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailed_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailed_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskFailed_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskFailed_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskFailed_Result represents the result of a WorkflowService.RespondActivityTaskFailed function call.\n//\n// The result of a RespondActivityTaskFailed execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskFailed_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskFailed_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskFailed_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskFailed_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskFailed_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskFailed_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailed_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskFailed_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailed_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskFailed_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskFailed_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskFailed_Result match the\n// provided WorkflowService_RespondActivityTaskFailed_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) Equals(rhs *WorkflowService_RespondActivityTaskFailed_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskFailed_Result.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskFailed\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) MethodName() string {\n\treturn \"RespondActivityTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailed_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondActivityTaskFailedByID_Args represents the arguments for the WorkflowService.RespondActivityTaskFailedByID function.\n//\n// The arguments for RespondActivityTaskFailedByID are sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskFailedByID_Args struct {\n\tFailRequest *shared.RespondActivityTaskFailedByIDRequest `json:\"failRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskFailedByID_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailRequest != nil {\n\t\tw, err = v.FailRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskFailedByIDRequest_Read(w wire.Value) (*shared.RespondActivityTaskFailedByIDRequest, error) {\n\tvar v shared.RespondActivityTaskFailedByIDRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskFailedByID_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskFailedByID_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskFailedByID_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailRequest, err = _RespondActivityTaskFailedByIDRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskFailedByID_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailedByID_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskFailedByIDRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskFailedByIDRequest, error) {\n\tvar v shared.RespondActivityTaskFailedByIDRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskFailedByID_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailedByID_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailRequest, err = _RespondActivityTaskFailedByIDRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskFailedByID_Args\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailRequest: %v\", v.FailRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskFailedByID_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskFailedByID_Args match the\n// provided WorkflowService_RespondActivityTaskFailedByID_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) Equals(rhs *WorkflowService_RespondActivityTaskFailedByID_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailRequest == nil && rhs.FailRequest == nil) || (v.FailRequest != nil && rhs.FailRequest != nil && v.FailRequest.Equals(rhs.FailRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskFailedByID_Args.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failRequest\", v.FailRequest))\n\t}\n\treturn err\n}\n\n// GetFailRequest returns the value of FailRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) GetFailRequest() (o *shared.RespondActivityTaskFailedByIDRequest) {\n\tif v != nil && v.FailRequest != nil {\n\t\treturn v.FailRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailRequest returns true if FailRequest is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) IsSetFailRequest() bool {\n\treturn v != nil && v.FailRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskFailedByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) MethodName() string {\n\treturn \"RespondActivityTaskFailedByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondActivityTaskFailedByID_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondActivityTaskFailedByID\n// function.\nvar WorkflowService_RespondActivityTaskFailedByID_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskFailedByID in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\t) *WorkflowService_RespondActivityTaskFailedByID_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskFailedByID.\n\t//\n\t// An error can be thrown by RespondActivityTaskFailedByID only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskFailedByID\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskFailedByID did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskFailedByID into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskFailedByID\n\t//\n\t//   err := RespondActivityTaskFailedByID(args)\n\t//   result, err := WorkflowService_RespondActivityTaskFailedByID_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskFailedByID: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondActivityTaskFailedByID_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskFailedByID\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskFailedByID threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondActivityTaskFailedByID_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondActivityTaskFailedByID_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondActivityTaskFailedByID_Helper.Args = func(\n\t\tfailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\t) *WorkflowService_RespondActivityTaskFailedByID_Args {\n\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Args{\n\t\t\tFailRequest: failRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskFailedByID_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondActivityTaskFailedByID_Helper.WrapResponse = func(err error) (*WorkflowService_RespondActivityTaskFailedByID_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondActivityTaskFailedByID_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondActivityTaskFailedByID_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondActivityTaskFailedByID_Helper.UnwrapResponse = func(result *WorkflowService_RespondActivityTaskFailedByID_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondActivityTaskFailedByID_Result represents the result of a WorkflowService.RespondActivityTaskFailedByID function call.\n//\n// The result of a RespondActivityTaskFailedByID execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondActivityTaskFailedByID_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondActivityTaskFailedByID_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondActivityTaskFailedByID_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondActivityTaskFailedByID_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondActivityTaskFailedByID_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondActivityTaskFailedByID_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondActivityTaskFailedByID_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailedByID_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondActivityTaskFailedByID_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondActivityTaskFailedByID_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondActivityTaskFailedByID_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondActivityTaskFailedByID_Result\n// struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondActivityTaskFailedByID_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondActivityTaskFailedByID_Result match the\n// provided WorkflowService_RespondActivityTaskFailedByID_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) Equals(rhs *WorkflowService_RespondActivityTaskFailedByID_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondActivityTaskFailedByID_Result.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskFailedByID\" for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) MethodName() string {\n\treturn \"RespondActivityTaskFailedByID\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondActivityTaskFailedByID_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondDecisionTaskCompleted_Args represents the arguments for the WorkflowService.RespondDecisionTaskCompleted function.\n//\n// The arguments for RespondDecisionTaskCompleted are sent and received over the wire as this struct.\ntype WorkflowService_RespondDecisionTaskCompleted_Args struct {\n\tCompleteRequest *shared.RespondDecisionTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondDecisionTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskCompletedRequest_Read(w wire.Value) (*shared.RespondDecisionTaskCompletedRequest, error) {\n\tvar v shared.RespondDecisionTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondDecisionTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondDecisionTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondDecisionTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondDecisionTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskCompleted_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondDecisionTaskCompletedRequest, error) {\n\tvar v shared.RespondDecisionTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondDecisionTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondDecisionTaskCompleted_Args\n// struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondDecisionTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondDecisionTaskCompleted_Args match the\n// provided WorkflowService_RespondDecisionTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) Equals(rhs *WorkflowService_RespondDecisionTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondDecisionTaskCompleted_Args.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) GetCompleteRequest() (o *shared.RespondDecisionTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondDecisionTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) MethodName() string {\n\treturn \"RespondDecisionTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondDecisionTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondDecisionTaskCompleted\n// function.\nvar WorkflowService_RespondDecisionTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondDecisionTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\t) *WorkflowService_RespondDecisionTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondDecisionTaskCompleted.\n\t//\n\t// An error can be thrown by RespondDecisionTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondDecisionTaskCompleted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RespondDecisionTaskCompleted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RespondDecisionTaskCompleted\n\t//\n\t//   value, err := RespondDecisionTaskCompleted(args)\n\t//   result, err := WorkflowService_RespondDecisionTaskCompleted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondDecisionTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RespondDecisionTaskCompletedResponse, error) (*WorkflowService_RespondDecisionTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondDecisionTaskCompleted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RespondDecisionTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_RespondDecisionTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondDecisionTaskCompleted_Result) (*shared.RespondDecisionTaskCompletedResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_RespondDecisionTaskCompleted_Helper.Args = func(\n\t\tcompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\t) *WorkflowService_RespondDecisionTaskCompleted_Args {\n\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondDecisionTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondDecisionTaskCompleted_Helper.WrapResponse = func(success *shared.RespondDecisionTaskCompletedResponse, err error) (*WorkflowService_RespondDecisionTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskCompleted_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskCompleted_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondDecisionTaskCompleted_Helper.UnwrapResponse = func(result *WorkflowService_RespondDecisionTaskCompleted_Result) (success *shared.RespondDecisionTaskCompletedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondDecisionTaskCompleted_Result represents the result of a WorkflowService.RespondDecisionTaskCompleted function call.\n//\n// The result of a RespondDecisionTaskCompleted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_RespondDecisionTaskCompleted_Result struct {\n\t// Value returned by RespondDecisionTaskCompleted after a successful execution.\n\tSuccess                                *shared.RespondDecisionTaskCompletedResponse   `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondDecisionTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskCompletedResponse_Read(w wire.Value) (*shared.RespondDecisionTaskCompletedResponse, error) {\n\tvar v shared.RespondDecisionTaskCompletedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondDecisionTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondDecisionTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondDecisionTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RespondDecisionTaskCompletedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondDecisionTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskCompleted_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskCompletedResponse_Decode(sr stream.Reader) (*shared.RespondDecisionTaskCompletedResponse, error) {\n\tvar v shared.RespondDecisionTaskCompletedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondDecisionTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RespondDecisionTaskCompletedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondDecisionTaskCompleted_Result\n// struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondDecisionTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondDecisionTaskCompleted_Result match the\n// provided WorkflowService_RespondDecisionTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) Equals(rhs *WorkflowService_RespondDecisionTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondDecisionTaskCompleted_Result.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetSuccess() (o *shared.RespondDecisionTaskCompletedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondDecisionTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) MethodName() string {\n\treturn \"RespondDecisionTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondDecisionTaskFailed_Args represents the arguments for the WorkflowService.RespondDecisionTaskFailed function.\n//\n// The arguments for RespondDecisionTaskFailed are sent and received over the wire as this struct.\ntype WorkflowService_RespondDecisionTaskFailed_Args struct {\n\tFailedRequest *shared.RespondDecisionTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondDecisionTaskFailed_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailedRequest != nil {\n\t\tw, err = v.FailedRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskFailedRequest_Read(w wire.Value) (*shared.RespondDecisionTaskFailedRequest, error) {\n\tvar v shared.RespondDecisionTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondDecisionTaskFailed_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondDecisionTaskFailed_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondDecisionTaskFailed_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondDecisionTaskFailed_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskFailed_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailedRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskFailedRequest_Decode(sr stream.Reader) (*shared.RespondDecisionTaskFailedRequest, error) {\n\tvar v shared.RespondDecisionTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondDecisionTaskFailed_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskFailed_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondDecisionTaskFailed_Args\n// struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailedRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedRequest: %v\", v.FailedRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondDecisionTaskFailed_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondDecisionTaskFailed_Args match the\n// provided WorkflowService_RespondDecisionTaskFailed_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) Equals(rhs *WorkflowService_RespondDecisionTaskFailed_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailedRequest == nil && rhs.FailedRequest == nil) || (v.FailedRequest != nil && rhs.FailedRequest != nil && v.FailedRequest.Equals(rhs.FailedRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondDecisionTaskFailed_Args.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailedRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedRequest\", v.FailedRequest))\n\t}\n\treturn err\n}\n\n// GetFailedRequest returns the value of FailedRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) GetFailedRequest() (o *shared.RespondDecisionTaskFailedRequest) {\n\tif v != nil && v.FailedRequest != nil {\n\t\treturn v.FailedRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailedRequest returns true if FailedRequest is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) IsSetFailedRequest() bool {\n\treturn v != nil && v.FailedRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondDecisionTaskFailed\" for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) MethodName() string {\n\treturn \"RespondDecisionTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondDecisionTaskFailed_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondDecisionTaskFailed\n// function.\nvar WorkflowService_RespondDecisionTaskFailed_Helper = struct {\n\t// Args accepts the parameters of RespondDecisionTaskFailed in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailedRequest *shared.RespondDecisionTaskFailedRequest,\n\t) *WorkflowService_RespondDecisionTaskFailed_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondDecisionTaskFailed.\n\t//\n\t// An error can be thrown by RespondDecisionTaskFailed only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondDecisionTaskFailed\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondDecisionTaskFailed did not fail.\n\t//\n\t// This allows mapping errors returned by RespondDecisionTaskFailed into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondDecisionTaskFailed\n\t//\n\t//   err := RespondDecisionTaskFailed(args)\n\t//   result, err := WorkflowService_RespondDecisionTaskFailed_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondDecisionTaskFailed: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondDecisionTaskFailed_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondDecisionTaskFailed\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondDecisionTaskFailed threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondDecisionTaskFailed_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondDecisionTaskFailed_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondDecisionTaskFailed_Helper.Args = func(\n\t\tfailedRequest *shared.RespondDecisionTaskFailedRequest,\n\t) *WorkflowService_RespondDecisionTaskFailed_Args {\n\t\treturn &WorkflowService_RespondDecisionTaskFailed_Args{\n\t\t\tFailedRequest: failedRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondDecisionTaskFailed_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondDecisionTaskFailed_Helper.WrapResponse = func(err error) (*WorkflowService_RespondDecisionTaskFailed_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondDecisionTaskFailed_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondDecisionTaskFailed_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondDecisionTaskFailed_Helper.UnwrapResponse = func(result *WorkflowService_RespondDecisionTaskFailed_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondDecisionTaskFailed_Result represents the result of a WorkflowService.RespondDecisionTaskFailed function call.\n//\n// The result of a RespondDecisionTaskFailed execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondDecisionTaskFailed_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondDecisionTaskFailed_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondDecisionTaskFailed_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondDecisionTaskFailed_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondDecisionTaskFailed_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondDecisionTaskFailed_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskFailed_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondDecisionTaskFailed_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondDecisionTaskFailed_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondDecisionTaskFailed_Result\n// struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondDecisionTaskFailed_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondDecisionTaskFailed_Result match the\n// provided WorkflowService_RespondDecisionTaskFailed_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) Equals(rhs *WorkflowService_RespondDecisionTaskFailed_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondDecisionTaskFailed_Result.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondDecisionTaskFailed\" for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) MethodName() string {\n\treturn \"RespondDecisionTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondDecisionTaskFailed_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RespondQueryTaskCompleted_Args represents the arguments for the WorkflowService.RespondQueryTaskCompleted function.\n//\n// The arguments for RespondQueryTaskCompleted are sent and received over the wire as this struct.\ntype WorkflowService_RespondQueryTaskCompleted_Args struct {\n\tCompleteRequest *shared.RespondQueryTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondQueryTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondQueryTaskCompletedRequest_Read(w wire.Value) (*shared.RespondQueryTaskCompletedRequest, error) {\n\tvar v shared.RespondQueryTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RespondQueryTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondQueryTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondQueryTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondQueryTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondQueryTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondQueryTaskCompleted_Args struct could not be encoded.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondQueryTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondQueryTaskCompletedRequest, error) {\n\tvar v shared.RespondQueryTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RespondQueryTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondQueryTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondQueryTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondQueryTaskCompleted_Args\n// struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondQueryTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondQueryTaskCompleted_Args match the\n// provided WorkflowService_RespondQueryTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) Equals(rhs *WorkflowService_RespondQueryTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondQueryTaskCompleted_Args.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) GetCompleteRequest() (o *shared.RespondQueryTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondQueryTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) MethodName() string {\n\treturn \"RespondQueryTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RespondQueryTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RespondQueryTaskCompleted\n// function.\nvar WorkflowService_RespondQueryTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondQueryTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\t) *WorkflowService_RespondQueryTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondQueryTaskCompleted.\n\t//\n\t// An error can be thrown by RespondQueryTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondQueryTaskCompleted\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondQueryTaskCompleted did not fail.\n\t//\n\t// This allows mapping errors returned by RespondQueryTaskCompleted into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondQueryTaskCompleted\n\t//\n\t//   err := RespondQueryTaskCompleted(args)\n\t//   result, err := WorkflowService_RespondQueryTaskCompleted_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondQueryTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_RespondQueryTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondQueryTaskCompleted\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondQueryTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_RespondQueryTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RespondQueryTaskCompleted_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_RespondQueryTaskCompleted_Helper.Args = func(\n\t\tcompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\t) *WorkflowService_RespondQueryTaskCompleted_Args {\n\t\treturn &WorkflowService_RespondQueryTaskCompleted_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RespondQueryTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RespondQueryTaskCompleted_Helper.WrapResponse = func(err error) (*WorkflowService_RespondQueryTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RespondQueryTaskCompleted_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RespondQueryTaskCompleted_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RespondQueryTaskCompleted_Helper.UnwrapResponse = func(result *WorkflowService_RespondQueryTaskCompleted_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RespondQueryTaskCompleted_Result represents the result of a WorkflowService.RespondQueryTaskCompleted function call.\n//\n// The result of a RespondQueryTaskCompleted execution is sent and received over the wire as this struct.\ntype WorkflowService_RespondQueryTaskCompleted_Result struct {\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RespondQueryTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_RespondQueryTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RespondQueryTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RespondQueryTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RespondQueryTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RespondQueryTaskCompleted_Result struct could not be encoded.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_RespondQueryTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RespondQueryTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RespondQueryTaskCompleted_Result\n// struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RespondQueryTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RespondQueryTaskCompleted_Result match the\n// provided WorkflowService_RespondQueryTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) Equals(rhs *WorkflowService_RespondQueryTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RespondQueryTaskCompleted_Result.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondQueryTaskCompleted\" for this struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) MethodName() string {\n\treturn \"RespondQueryTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RespondQueryTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_RestartWorkflowExecution_Args represents the arguments for the WorkflowService.RestartWorkflowExecution function.\n//\n// The arguments for RestartWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_RestartWorkflowExecution_Args struct {\n\tRestartRequest *shared.RestartWorkflowExecutionRequest `json:\"restartRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RestartWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RestartRequest != nil {\n\t\tw, err = v.RestartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RestartWorkflowExecutionRequest_Read(w wire.Value) (*shared.RestartWorkflowExecutionRequest, error) {\n\tvar v shared.RestartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RestartWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RestartWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RestartWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RestartRequest, err = _RestartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RestartWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RestartWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RestartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RestartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RestartWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.RestartWorkflowExecutionRequest, error) {\n\tvar v shared.RestartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RestartWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RestartWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.RestartRequest, err = _RestartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RestartWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RestartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"RestartRequest: %v\", v.RestartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RestartWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RestartWorkflowExecution_Args match the\n// provided WorkflowService_RestartWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) Equals(rhs *WorkflowService_RestartWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.RestartRequest == nil && rhs.RestartRequest == nil) || (v.RestartRequest != nil && rhs.RestartRequest != nil && v.RestartRequest.Equals(rhs.RestartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RestartWorkflowExecution_Args.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RestartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"restartRequest\", v.RestartRequest))\n\t}\n\treturn err\n}\n\n// GetRestartRequest returns the value of RestartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) GetRestartRequest() (o *shared.RestartWorkflowExecutionRequest) {\n\tif v != nil && v.RestartRequest != nil {\n\t\treturn v.RestartRequest\n\t}\n\n\treturn\n}\n\n// IsSetRestartRequest returns true if RestartRequest is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) IsSetRestartRequest() bool {\n\treturn v != nil && v.RestartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RestartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) MethodName() string {\n\treturn \"RestartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_RestartWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.RestartWorkflowExecution\n// function.\nvar WorkflowService_RestartWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of RestartWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trestartRequest *shared.RestartWorkflowExecutionRequest,\n\t) *WorkflowService_RestartWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RestartWorkflowExecution.\n\t//\n\t// An error can be thrown by RestartWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RestartWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RestartWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RestartWorkflowExecution\n\t//\n\t//   value, err := RestartWorkflowExecution(args)\n\t//   result, err := WorkflowService_RestartWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RestartWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RestartWorkflowExecutionResponse, error) (*WorkflowService_RestartWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for RestartWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RestartWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_RestartWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_RestartWorkflowExecution_Result) (*shared.RestartWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_RestartWorkflowExecution_Helper.Args = func(\n\t\trestartRequest *shared.RestartWorkflowExecutionRequest,\n\t) *WorkflowService_RestartWorkflowExecution_Args {\n\t\treturn &WorkflowService_RestartWorkflowExecution_Args{\n\t\t\tRestartRequest: restartRequest,\n\t\t}\n\t}\n\n\tWorkflowService_RestartWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_RestartWorkflowExecution_Helper.WrapResponse = func(success *shared.RestartWorkflowExecutionResponse, err error) (*WorkflowService_RestartWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_RestartWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_RestartWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_RestartWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_RestartWorkflowExecution_Result) (success *shared.RestartWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_RestartWorkflowExecution_Result represents the result of a WorkflowService.RestartWorkflowExecution function call.\n//\n// The result of a RestartWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_RestartWorkflowExecution_Result struct {\n\t// Value returned by RestartWorkflowExecution after a successful execution.\n\tSuccess                        *shared.RestartWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                  `json:\"badRequestError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                 `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError             `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError               `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError             `json:\"entityNotExistError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError   `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_RestartWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_RestartWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RestartWorkflowExecutionResponse_Read(w wire.Value) (*shared.RestartWorkflowExecutionResponse, error) {\n\tvar v shared.RestartWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_RestartWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_RestartWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_RestartWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RestartWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RestartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_RestartWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_RestartWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RestartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RestartWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.RestartWorkflowExecutionResponse, error) {\n\tvar v shared.RestartWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_RestartWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_RestartWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RestartWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_RestartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_RestartWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_RestartWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_RestartWorkflowExecution_Result match the\n// provided WorkflowService_RestartWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) Equals(rhs *WorkflowService_RestartWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_RestartWorkflowExecution_Result.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetSuccess() (o *shared.RestartWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RestartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) MethodName() string {\n\treturn \"RestartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_RestartWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_ScanWorkflowExecutions_Args represents the arguments for the WorkflowService.ScanWorkflowExecutions function.\n//\n// The arguments for ScanWorkflowExecutions are sent and received over the wire as this struct.\ntype WorkflowService_ScanWorkflowExecutions_Args struct {\n\tListRequest *shared.ListWorkflowExecutionsRequest `json:\"listRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ScanWorkflowExecutions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ListRequest != nil {\n\t\tw, err = v.ListRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_ScanWorkflowExecutions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ScanWorkflowExecutions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ScanWorkflowExecutions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ListRequest, err = _ListWorkflowExecutionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ScanWorkflowExecutions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ScanWorkflowExecutions_Args struct could not be encoded.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ListRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ListRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_ScanWorkflowExecutions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ScanWorkflowExecutions_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ListRequest, err = _ListWorkflowExecutionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ScanWorkflowExecutions_Args\n// struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ListRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ListRequest: %v\", v.ListRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ScanWorkflowExecutions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ScanWorkflowExecutions_Args match the\n// provided WorkflowService_ScanWorkflowExecutions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) Equals(rhs *WorkflowService_ScanWorkflowExecutions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ListRequest == nil && rhs.ListRequest == nil) || (v.ListRequest != nil && rhs.ListRequest != nil && v.ListRequest.Equals(rhs.ListRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ScanWorkflowExecutions_Args.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ListRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"listRequest\", v.ListRequest))\n\t}\n\treturn err\n}\n\n// GetListRequest returns the value of ListRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) GetListRequest() (o *shared.ListWorkflowExecutionsRequest) {\n\tif v != nil && v.ListRequest != nil {\n\t\treturn v.ListRequest\n\t}\n\n\treturn\n}\n\n// IsSetListRequest returns true if ListRequest is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) IsSetListRequest() bool {\n\treturn v != nil && v.ListRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ScanWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) MethodName() string {\n\treturn \"ScanWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_ScanWorkflowExecutions_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.ScanWorkflowExecutions\n// function.\nvar WorkflowService_ScanWorkflowExecutions_Helper = struct {\n\t// Args accepts the parameters of ScanWorkflowExecutions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tlistRequest *shared.ListWorkflowExecutionsRequest,\n\t) *WorkflowService_ScanWorkflowExecutions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ScanWorkflowExecutions.\n\t//\n\t// An error can be thrown by ScanWorkflowExecutions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ScanWorkflowExecutions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ScanWorkflowExecutions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ScanWorkflowExecutions\n\t//\n\t//   value, err := ScanWorkflowExecutions(args)\n\t//   result, err := WorkflowService_ScanWorkflowExecutions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ScanWorkflowExecutions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListWorkflowExecutionsResponse, error) (*WorkflowService_ScanWorkflowExecutions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ScanWorkflowExecutions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ScanWorkflowExecutions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_ScanWorkflowExecutions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_ScanWorkflowExecutions_Result) (*shared.ListWorkflowExecutionsResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_ScanWorkflowExecutions_Helper.Args = func(\n\t\tlistRequest *shared.ListWorkflowExecutionsRequest,\n\t) *WorkflowService_ScanWorkflowExecutions_Args {\n\t\treturn &WorkflowService_ScanWorkflowExecutions_Args{\n\t\t\tListRequest: listRequest,\n\t\t}\n\t}\n\n\tWorkflowService_ScanWorkflowExecutions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_ScanWorkflowExecutions_Helper.WrapResponse = func(success *shared.ListWorkflowExecutionsResponse, err error) (*WorkflowService_ScanWorkflowExecutions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ScanWorkflowExecutions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ScanWorkflowExecutions_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ScanWorkflowExecutions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ScanWorkflowExecutions_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_ScanWorkflowExecutions_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_ScanWorkflowExecutions_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_ScanWorkflowExecutions_Helper.UnwrapResponse = func(result *WorkflowService_ScanWorkflowExecutions_Result) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_ScanWorkflowExecutions_Result represents the result of a WorkflowService.ScanWorkflowExecutions function call.\n//\n// The result of a ScanWorkflowExecutions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_ScanWorkflowExecutions_Result struct {\n\t// Value returned by ScanWorkflowExecutions after a successful execution.\n\tSuccess                        *shared.ListWorkflowExecutionsResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_ScanWorkflowExecutions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_ScanWorkflowExecutions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_ScanWorkflowExecutions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_ScanWorkflowExecutions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_ScanWorkflowExecutions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListWorkflowExecutionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ScanWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_ScanWorkflowExecutions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_ScanWorkflowExecutions_Result struct could not be encoded.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ScanWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_ScanWorkflowExecutions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_ScanWorkflowExecutions_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListWorkflowExecutionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_ScanWorkflowExecutions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_ScanWorkflowExecutions_Result\n// struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_ScanWorkflowExecutions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_ScanWorkflowExecutions_Result match the\n// provided WorkflowService_ScanWorkflowExecutions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) Equals(rhs *WorkflowService_ScanWorkflowExecutions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_ScanWorkflowExecutions_Result.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetSuccess() (o *shared.ListWorkflowExecutionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ScanWorkflowExecutions\" for this struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) MethodName() string {\n\treturn \"ScanWorkflowExecutions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_ScanWorkflowExecutions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_SignalWithStartWorkflowExecution_Args represents the arguments for the WorkflowService.SignalWithStartWorkflowExecution function.\n//\n// The arguments for SignalWithStartWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_SignalWithStartWorkflowExecution_Args struct {\n\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest `json:\"signalWithStartRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWithStartWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalWithStartRequest != nil {\n\t\tw, err = v.SignalWithStartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Read(w wire.Value) (*shared.SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_SignalWithStartWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWithStartWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWithStartWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWithStartWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalWithStartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalWithStartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_SignalWithStartWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWithStartWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SignalWithStartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalWithStartRequest: %v\", v.SignalWithStartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWithStartWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWithStartWorkflowExecution_Args match the\n// provided WorkflowService_SignalWithStartWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) Equals(rhs *WorkflowService_SignalWithStartWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SignalWithStartRequest == nil && rhs.SignalWithStartRequest == nil) || (v.SignalWithStartRequest != nil && rhs.SignalWithStartRequest != nil && v.SignalWithStartRequest.Equals(rhs.SignalWithStartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWithStartWorkflowExecution_Args.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalWithStartRequest\", v.SignalWithStartRequest))\n\t}\n\treturn err\n}\n\n// GetSignalWithStartRequest returns the value of SignalWithStartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) GetSignalWithStartRequest() (o *shared.SignalWithStartWorkflowExecutionRequest) {\n\tif v != nil && v.SignalWithStartRequest != nil {\n\t\treturn v.SignalWithStartRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalWithStartRequest returns true if SignalWithStartRequest is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) IsSetSignalWithStartRequest() bool {\n\treturn v != nil && v.SignalWithStartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SignalWithStartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_SignalWithStartWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.SignalWithStartWorkflowExecution\n// function.\nvar WorkflowService_SignalWithStartWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of SignalWithStartWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\t) *WorkflowService_SignalWithStartWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SignalWithStartWorkflowExecution.\n\t//\n\t// An error can be thrown by SignalWithStartWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SignalWithStartWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// SignalWithStartWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by SignalWithStartWorkflowExecution\n\t//\n\t//   value, err := SignalWithStartWorkflowExecution(args)\n\t//   result, err := WorkflowService_SignalWithStartWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SignalWithStartWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.StartWorkflowExecutionResponse, error) (*WorkflowService_SignalWithStartWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for SignalWithStartWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if SignalWithStartWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_SignalWithStartWorkflowExecution_Result) (*shared.StartWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_SignalWithStartWorkflowExecution_Helper.Args = func(\n\t\tsignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\t) *WorkflowService_SignalWithStartWorkflowExecution_Args {\n\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Args{\n\t\t\tSignalWithStartRequest: signalWithStartRequest,\n\t\t}\n\t}\n\n\tWorkflowService_SignalWithStartWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_SignalWithStartWorkflowExecution_Helper.WrapResponse = func(success *shared.StartWorkflowExecutionResponse, err error) (*WorkflowService_SignalWithStartWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.WorkflowAlreadyStartedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{WorkflowAlreadyStartedError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_SignalWithStartWorkflowExecution_Result) (success *shared.StartWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowAlreadyStartedError != nil {\n\t\t\terr = result.WorkflowAlreadyStartedError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_SignalWithStartWorkflowExecution_Result represents the result of a WorkflowService.SignalWithStartWorkflowExecution function call.\n//\n// The result of a SignalWithStartWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_SignalWithStartWorkflowExecution_Result struct {\n\t// Value returned by SignalWithStartWorkflowExecution after a successful execution.\n\tSuccess                        *shared.StartWorkflowExecutionResponse       `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                 `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError                 `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                   `json:\"limitExceededError,omitempty\"`\n\tWorkflowAlreadyStartedError    *shared.WorkflowExecutionAlreadyStartedError `json:\"workflowAlreadyStartedError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError       `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWithStartWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tw, err = v.WorkflowAlreadyStartedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionResponse_Read(w wire.Value) (*shared.StartWorkflowExecutionResponse, error) {\n\tvar v shared.StartWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyStartedError_Read(w wire.Value) (*shared.WorkflowExecutionAlreadyStartedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyStartedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_SignalWithStartWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWithStartWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWithStartWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowAlreadyStartedError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWithStartWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowAlreadyStartedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionResponse, error) {\n\tvar v shared.StartWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyStartedError_Decode(sr stream.Reader) (*shared.WorkflowExecutionAlreadyStartedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyStartedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_SignalWithStartWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowAlreadyStartedError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWithStartWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowAlreadyStartedError: %v\", v.WorkflowAlreadyStartedError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWithStartWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWithStartWorkflowExecution_Result match the\n// provided WorkflowService_SignalWithStartWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) Equals(rhs *WorkflowService_SignalWithStartWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowAlreadyStartedError == nil && rhs.WorkflowAlreadyStartedError == nil) || (v.WorkflowAlreadyStartedError != nil && rhs.WorkflowAlreadyStartedError != nil && v.WorkflowAlreadyStartedError.Equals(rhs.WorkflowAlreadyStartedError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWithStartWorkflowExecution_Result.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowAlreadyStartedError\", v.WorkflowAlreadyStartedError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetSuccess() (o *shared.StartWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetWorkflowAlreadyStartedError returns the value of WorkflowAlreadyStartedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetWorkflowAlreadyStartedError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.WorkflowAlreadyStartedError != nil {\n\t\treturn v.WorkflowAlreadyStartedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowAlreadyStartedError returns true if WorkflowAlreadyStartedError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetWorkflowAlreadyStartedError() bool {\n\treturn v != nil && v.WorkflowAlreadyStartedError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SignalWithStartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_SignalWithStartWorkflowExecutionAsync_Args represents the arguments for the WorkflowService.SignalWithStartWorkflowExecutionAsync function.\n//\n// The arguments for SignalWithStartWorkflowExecutionAsync are sent and received over the wire as this struct.\ntype WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct {\n\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest `json:\"signalWithStartRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalWithStartRequest != nil {\n\t\tw, err = v.SignalWithStartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionAsyncRequest_Read(w wire.Value) (*shared.SignalWithStartWorkflowExecutionAsyncRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionAsyncRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWithStartWorkflowExecutionAsync_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionAsyncRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct could not be encoded.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalWithStartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalWithStartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionAsyncRequest_Decode(sr stream.Reader) (*shared.SignalWithStartWorkflowExecutionAsyncRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionAsyncRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionAsyncRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWithStartWorkflowExecutionAsync_Args\n// struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SignalWithStartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalWithStartRequest: %v\", v.SignalWithStartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWithStartWorkflowExecutionAsync_Args match the\n// provided WorkflowService_SignalWithStartWorkflowExecutionAsync_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) Equals(rhs *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SignalWithStartRequest == nil && rhs.SignalWithStartRequest == nil) || (v.SignalWithStartRequest != nil && rhs.SignalWithStartRequest != nil && v.SignalWithStartRequest.Equals(rhs.SignalWithStartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWithStartWorkflowExecutionAsync_Args.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalWithStartRequest\", v.SignalWithStartRequest))\n\t}\n\treturn err\n}\n\n// GetSignalWithStartRequest returns the value of SignalWithStartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) GetSignalWithStartRequest() (o *shared.SignalWithStartWorkflowExecutionAsyncRequest) {\n\tif v != nil && v.SignalWithStartRequest != nil {\n\t\treturn v.SignalWithStartRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalWithStartRequest returns true if SignalWithStartRequest is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) IsSetSignalWithStartRequest() bool {\n\treturn v != nil && v.SignalWithStartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SignalWithStartWorkflowExecutionAsync\" for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecutionAsync\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.SignalWithStartWorkflowExecutionAsync\n// function.\nvar WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper = struct {\n\t// Args accepts the parameters of SignalWithStartWorkflowExecutionAsync in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\t) *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SignalWithStartWorkflowExecutionAsync.\n\t//\n\t// An error can be thrown by SignalWithStartWorkflowExecutionAsync only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SignalWithStartWorkflowExecutionAsync\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// SignalWithStartWorkflowExecutionAsync into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by SignalWithStartWorkflowExecutionAsync\n\t//\n\t//   value, err := SignalWithStartWorkflowExecutionAsync(args)\n\t//   result, err := WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SignalWithStartWorkflowExecutionAsync: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.SignalWithStartWorkflowExecutionAsyncResponse, error) (*WorkflowService_SignalWithStartWorkflowExecutionAsync_Result, error)\n\n\t// UnwrapResponse takes the result struct for SignalWithStartWorkflowExecutionAsync\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if SignalWithStartWorkflowExecutionAsync threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) (*shared.SignalWithStartWorkflowExecutionAsyncResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.Args = func(\n\t\tsignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\t) *WorkflowService_SignalWithStartWorkflowExecutionAsync_Args {\n\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Args{\n\t\t\tSignalWithStartRequest: signalWithStartRequest,\n\t\t}\n\t}\n\n\tWorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.WrapResponse = func(success *shared.SignalWithStartWorkflowExecutionAsyncResponse, err error) (*WorkflowService_SignalWithStartWorkflowExecutionAsync_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{BadRequestError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.SessionAlreadyExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{SessionAlreadyExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.UnwrapResponse = func(result *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) (success *shared.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.SessionAlreadyExistError != nil {\n\t\t\terr = result.SessionAlreadyExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_SignalWithStartWorkflowExecutionAsync_Result represents the result of a WorkflowService.SignalWithStartWorkflowExecutionAsync function call.\n//\n// The result of a SignalWithStartWorkflowExecutionAsync execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct {\n\t// Value returned by SignalWithStartWorkflowExecutionAsync after a successful execution.\n\tSuccess                        *shared.SignalWithStartWorkflowExecutionAsyncResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                               `json:\"badRequestError,omitempty\"`\n\tSessionAlreadyExistError       *shared.WorkflowExecutionAlreadyStartedError          `json:\"sessionAlreadyExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                              `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError                          `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                            `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                          `json:\"entityNotExistError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError                `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                             `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tw, err = v.SessionAlreadyExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionAsyncResponse_Read(w wire.Value) (*shared.SignalWithStartWorkflowExecutionAsyncResponse, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionAsyncResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWithStartWorkflowExecutionAsync_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _SignalWithStartWorkflowExecutionAsyncResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct could not be encoded.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SessionAlreadyExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SessionAlreadyExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionAsyncResponse_Decode(sr stream.Reader) (*shared.SignalWithStartWorkflowExecutionAsyncResponse, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionAsyncResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _SignalWithStartWorkflowExecutionAsyncResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWithStartWorkflowExecutionAsync_Result\n// struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"SessionAlreadyExistError: %v\", v.SessionAlreadyExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWithStartWorkflowExecutionAsync_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWithStartWorkflowExecutionAsync_Result match the\n// provided WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) Equals(rhs *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.SessionAlreadyExistError == nil && rhs.SessionAlreadyExistError == nil) || (v.SessionAlreadyExistError != nil && rhs.SessionAlreadyExistError != nil && v.SessionAlreadyExistError.Equals(rhs.SessionAlreadyExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWithStartWorkflowExecutionAsync_Result.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"sessionAlreadyExistError\", v.SessionAlreadyExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetSuccess() (o *shared.SignalWithStartWorkflowExecutionAsyncResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetSessionAlreadyExistError returns the value of SessionAlreadyExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetSessionAlreadyExistError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.SessionAlreadyExistError != nil {\n\t\treturn v.SessionAlreadyExistError\n\t}\n\n\treturn\n}\n\n// IsSetSessionAlreadyExistError returns true if SessionAlreadyExistError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetSessionAlreadyExistError() bool {\n\treturn v != nil && v.SessionAlreadyExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SignalWithStartWorkflowExecutionAsync\" for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecutionAsync\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_SignalWithStartWorkflowExecutionAsync_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_SignalWorkflowExecution_Args represents the arguments for the WorkflowService.SignalWorkflowExecution function.\n//\n// The arguments for SignalWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_SignalWorkflowExecution_Args struct {\n\tSignalRequest *shared.SignalWorkflowExecutionRequest `json:\"signalRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalRequest != nil {\n\t\tw, err = v.SignalRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWorkflowExecutionRequest_Read(w wire.Value) (*shared.SignalWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_SignalWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.SignalWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_SignalWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SignalRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalRequest: %v\", v.SignalRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWorkflowExecution_Args match the\n// provided WorkflowService_SignalWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) Equals(rhs *WorkflowService_SignalWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SignalRequest == nil && rhs.SignalRequest == nil) || (v.SignalRequest != nil && rhs.SignalRequest != nil && v.SignalRequest.Equals(rhs.SignalRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWorkflowExecution_Args.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalRequest\", v.SignalRequest))\n\t}\n\treturn err\n}\n\n// GetSignalRequest returns the value of SignalRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) GetSignalRequest() (o *shared.SignalWorkflowExecutionRequest) {\n\tif v != nil && v.SignalRequest != nil {\n\t\treturn v.SignalRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalRequest returns true if SignalRequest is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) IsSetSignalRequest() bool {\n\treturn v != nil && v.SignalRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SignalWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) MethodName() string {\n\treturn \"SignalWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_SignalWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.SignalWorkflowExecution\n// function.\nvar WorkflowService_SignalWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of SignalWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsignalRequest *shared.SignalWorkflowExecutionRequest,\n\t) *WorkflowService_SignalWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SignalWorkflowExecution.\n\t//\n\t// An error can be thrown by SignalWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SignalWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if SignalWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by SignalWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// SignalWorkflowExecution\n\t//\n\t//   err := SignalWorkflowExecution(args)\n\t//   result, err := WorkflowService_SignalWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SignalWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_SignalWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for SignalWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if SignalWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_SignalWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_SignalWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_SignalWorkflowExecution_Helper.Args = func(\n\t\tsignalRequest *shared.SignalWorkflowExecutionRequest,\n\t) *WorkflowService_SignalWorkflowExecution_Args {\n\t\treturn &WorkflowService_SignalWorkflowExecution_Args{\n\t\t\tSignalRequest: signalRequest,\n\t\t}\n\t}\n\n\tWorkflowService_SignalWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_SignalWorkflowExecution_Helper.WrapResponse = func(err error) (*WorkflowService_SignalWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_SignalWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_SignalWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_SignalWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_SignalWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_SignalWorkflowExecution_Result represents the result of a WorkflowService.SignalWorkflowExecution function call.\n//\n// The result of a SignalWorkflowExecution execution is sent and received over the wire as this struct.\ntype WorkflowService_SignalWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_SignalWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_SignalWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_SignalWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_SignalWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_SignalWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_SignalWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_SignalWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_SignalWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_SignalWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_SignalWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_SignalWorkflowExecution_Result match the\n// provided WorkflowService_SignalWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) Equals(rhs *WorkflowService_SignalWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_SignalWorkflowExecution_Result.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SignalWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) MethodName() string {\n\treturn \"SignalWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_SignalWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_StartWorkflowExecution_Args represents the arguments for the WorkflowService.StartWorkflowExecution function.\n//\n// The arguments for StartWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_StartWorkflowExecution_Args struct {\n\tStartRequest *shared.StartWorkflowExecutionRequest `json:\"startRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_StartWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_StartWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StartRequest != nil {\n\t\tw, err = v.StartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionRequest_Read(w wire.Value) (*shared.StartWorkflowExecutionRequest, error) {\n\tvar v shared.StartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_StartWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_StartWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_StartWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_StartWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_StartWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionRequest, error) {\n\tvar v shared.StartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_StartWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_StartWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.StartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartRequest: %v\", v.StartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_StartWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_StartWorkflowExecution_Args match the\n// provided WorkflowService_StartWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) Equals(rhs *WorkflowService_StartWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.StartRequest == nil && rhs.StartRequest == nil) || (v.StartRequest != nil && rhs.StartRequest != nil && v.StartRequest.Equals(rhs.StartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_StartWorkflowExecution_Args.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startRequest\", v.StartRequest))\n\t}\n\treturn err\n}\n\n// GetStartRequest returns the value of StartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) GetStartRequest() (o *shared.StartWorkflowExecutionRequest) {\n\tif v != nil && v.StartRequest != nil {\n\t\treturn v.StartRequest\n\t}\n\n\treturn\n}\n\n// IsSetStartRequest returns true if StartRequest is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) IsSetStartRequest() bool {\n\treturn v != nil && v.StartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"StartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) MethodName() string {\n\treturn \"StartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_StartWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.StartWorkflowExecution\n// function.\nvar WorkflowService_StartWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of StartWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tstartRequest *shared.StartWorkflowExecutionRequest,\n\t) *WorkflowService_StartWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by StartWorkflowExecution.\n\t//\n\t// An error can be thrown by StartWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for StartWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// StartWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by StartWorkflowExecution\n\t//\n\t//   value, err := StartWorkflowExecution(args)\n\t//   result, err := WorkflowService_StartWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from StartWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.StartWorkflowExecutionResponse, error) (*WorkflowService_StartWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for StartWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if StartWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_StartWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_StartWorkflowExecution_Result) (*shared.StartWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_StartWorkflowExecution_Helper.Args = func(\n\t\tstartRequest *shared.StartWorkflowExecutionRequest,\n\t) *WorkflowService_StartWorkflowExecution_Args {\n\t\treturn &WorkflowService_StartWorkflowExecution_Args{\n\t\t\tStartRequest: startRequest,\n\t\t}\n\t}\n\n\tWorkflowService_StartWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_StartWorkflowExecution_Helper.WrapResponse = func(success *shared.StartWorkflowExecutionResponse, err error) (*WorkflowService_StartWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.SessionAlreadyExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{SessionAlreadyExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_StartWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_StartWorkflowExecution_Result) (success *shared.StartWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.SessionAlreadyExistError != nil {\n\t\t\terr = result.SessionAlreadyExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_StartWorkflowExecution_Result represents the result of a WorkflowService.StartWorkflowExecution function call.\n//\n// The result of a StartWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_StartWorkflowExecution_Result struct {\n\t// Value returned by StartWorkflowExecution after a successful execution.\n\tSuccess                        *shared.StartWorkflowExecutionResponse       `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tSessionAlreadyExistError       *shared.WorkflowExecutionAlreadyStartedError `json:\"sessionAlreadyExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError                 `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                   `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                 `json:\"entityNotExistError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError       `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_StartWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_StartWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tw, err = v.SessionAlreadyExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_StartWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_StartWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_StartWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_StartWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_StartWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SessionAlreadyExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SessionAlreadyExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_StartWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_StartWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"SessionAlreadyExistError: %v\", v.SessionAlreadyExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_StartWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_StartWorkflowExecution_Result match the\n// provided WorkflowService_StartWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) Equals(rhs *WorkflowService_StartWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.SessionAlreadyExistError == nil && rhs.SessionAlreadyExistError == nil) || (v.SessionAlreadyExistError != nil && rhs.SessionAlreadyExistError != nil && v.SessionAlreadyExistError.Equals(rhs.SessionAlreadyExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_StartWorkflowExecution_Result.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"sessionAlreadyExistError\", v.SessionAlreadyExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetSuccess() (o *shared.StartWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetSessionAlreadyExistError returns the value of SessionAlreadyExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetSessionAlreadyExistError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.SessionAlreadyExistError != nil {\n\t\treturn v.SessionAlreadyExistError\n\t}\n\n\treturn\n}\n\n// IsSetSessionAlreadyExistError returns true if SessionAlreadyExistError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetSessionAlreadyExistError() bool {\n\treturn v != nil && v.SessionAlreadyExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"StartWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) MethodName() string {\n\treturn \"StartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_StartWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_StartWorkflowExecutionAsync_Args represents the arguments for the WorkflowService.StartWorkflowExecutionAsync function.\n//\n// The arguments for StartWorkflowExecutionAsync are sent and received over the wire as this struct.\ntype WorkflowService_StartWorkflowExecutionAsync_Args struct {\n\tStartRequest *shared.StartWorkflowExecutionAsyncRequest `json:\"startRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_StartWorkflowExecutionAsync_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StartRequest != nil {\n\t\tw, err = v.StartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionAsyncRequest_Read(w wire.Value) (*shared.StartWorkflowExecutionAsyncRequest, error) {\n\tvar v shared.StartWorkflowExecutionAsyncRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_StartWorkflowExecutionAsync_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_StartWorkflowExecutionAsync_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_StartWorkflowExecutionAsync_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartRequest, err = _StartWorkflowExecutionAsyncRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_StartWorkflowExecutionAsync_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecutionAsync_Args struct could not be encoded.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionAsyncRequest_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionAsyncRequest, error) {\n\tvar v shared.StartWorkflowExecutionAsyncRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_StartWorkflowExecutionAsync_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecutionAsync_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.StartRequest, err = _StartWorkflowExecutionAsyncRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_StartWorkflowExecutionAsync_Args\n// struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.StartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartRequest: %v\", v.StartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_StartWorkflowExecutionAsync_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_StartWorkflowExecutionAsync_Args match the\n// provided WorkflowService_StartWorkflowExecutionAsync_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) Equals(rhs *WorkflowService_StartWorkflowExecutionAsync_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.StartRequest == nil && rhs.StartRequest == nil) || (v.StartRequest != nil && rhs.StartRequest != nil && v.StartRequest.Equals(rhs.StartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_StartWorkflowExecutionAsync_Args.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startRequest\", v.StartRequest))\n\t}\n\treturn err\n}\n\n// GetStartRequest returns the value of StartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) GetStartRequest() (o *shared.StartWorkflowExecutionAsyncRequest) {\n\tif v != nil && v.StartRequest != nil {\n\t\treturn v.StartRequest\n\t}\n\n\treturn\n}\n\n// IsSetStartRequest returns true if StartRequest is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) IsSetStartRequest() bool {\n\treturn v != nil && v.StartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"StartWorkflowExecutionAsync\" for this struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) MethodName() string {\n\treturn \"StartWorkflowExecutionAsync\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_StartWorkflowExecutionAsync_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.StartWorkflowExecutionAsync\n// function.\nvar WorkflowService_StartWorkflowExecutionAsync_Helper = struct {\n\t// Args accepts the parameters of StartWorkflowExecutionAsync in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tstartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\t) *WorkflowService_StartWorkflowExecutionAsync_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by StartWorkflowExecutionAsync.\n\t//\n\t// An error can be thrown by StartWorkflowExecutionAsync only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for StartWorkflowExecutionAsync\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// StartWorkflowExecutionAsync into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by StartWorkflowExecutionAsync\n\t//\n\t//   value, err := StartWorkflowExecutionAsync(args)\n\t//   result, err := WorkflowService_StartWorkflowExecutionAsync_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from StartWorkflowExecutionAsync: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.StartWorkflowExecutionAsyncResponse, error) (*WorkflowService_StartWorkflowExecutionAsync_Result, error)\n\n\t// UnwrapResponse takes the result struct for StartWorkflowExecutionAsync\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if StartWorkflowExecutionAsync threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_StartWorkflowExecutionAsync_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_StartWorkflowExecutionAsync_Result) (*shared.StartWorkflowExecutionAsyncResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_StartWorkflowExecutionAsync_Helper.Args = func(\n\t\tstartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\t) *WorkflowService_StartWorkflowExecutionAsync_Args {\n\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Args{\n\t\t\tStartRequest: startRequest,\n\t\t}\n\t}\n\n\tWorkflowService_StartWorkflowExecutionAsync_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_StartWorkflowExecutionAsync_Helper.WrapResponse = func(success *shared.StartWorkflowExecutionAsyncResponse, err error) (*WorkflowService_StartWorkflowExecutionAsync_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{BadRequestError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.SessionAlreadyExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{SessionAlreadyExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{LimitExceededError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_StartWorkflowExecutionAsync_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_StartWorkflowExecutionAsync_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_StartWorkflowExecutionAsync_Helper.UnwrapResponse = func(result *WorkflowService_StartWorkflowExecutionAsync_Result) (success *shared.StartWorkflowExecutionAsyncResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.SessionAlreadyExistError != nil {\n\t\t\terr = result.SessionAlreadyExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_StartWorkflowExecutionAsync_Result represents the result of a WorkflowService.StartWorkflowExecutionAsync function call.\n//\n// The result of a StartWorkflowExecutionAsync execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_StartWorkflowExecutionAsync_Result struct {\n\t// Value returned by StartWorkflowExecutionAsync after a successful execution.\n\tSuccess                        *shared.StartWorkflowExecutionAsyncResponse  `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tSessionAlreadyExistError       *shared.WorkflowExecutionAlreadyStartedError `json:\"sessionAlreadyExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError                 `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                   `json:\"limitExceededError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError                 `json:\"entityNotExistError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError       `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError                    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_StartWorkflowExecutionAsync_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tw, err = v.SessionAlreadyExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_StartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionAsyncResponse_Read(w wire.Value) (*shared.StartWorkflowExecutionAsyncResponse, error) {\n\tvar v shared.StartWorkflowExecutionAsyncResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_StartWorkflowExecutionAsync_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_StartWorkflowExecutionAsync_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_StartWorkflowExecutionAsync_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _StartWorkflowExecutionAsyncResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_StartWorkflowExecutionAsync_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecutionAsync_Result struct could not be encoded.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SessionAlreadyExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SessionAlreadyExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionAsyncResponse_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionAsyncResponse, error) {\n\tvar v shared.StartWorkflowExecutionAsyncResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_StartWorkflowExecutionAsync_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_StartWorkflowExecutionAsync_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _StartWorkflowExecutionAsyncResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_StartWorkflowExecutionAsync_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_StartWorkflowExecutionAsync_Result\n// struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"SessionAlreadyExistError: %v\", v.SessionAlreadyExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_StartWorkflowExecutionAsync_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_StartWorkflowExecutionAsync_Result match the\n// provided WorkflowService_StartWorkflowExecutionAsync_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) Equals(rhs *WorkflowService_StartWorkflowExecutionAsync_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.SessionAlreadyExistError == nil && rhs.SessionAlreadyExistError == nil) || (v.SessionAlreadyExistError != nil && rhs.SessionAlreadyExistError != nil && v.SessionAlreadyExistError.Equals(rhs.SessionAlreadyExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_StartWorkflowExecutionAsync_Result.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"sessionAlreadyExistError\", v.SessionAlreadyExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetSuccess() (o *shared.StartWorkflowExecutionAsyncResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetSessionAlreadyExistError returns the value of SessionAlreadyExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetSessionAlreadyExistError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.SessionAlreadyExistError != nil {\n\t\treturn v.SessionAlreadyExistError\n\t}\n\n\treturn\n}\n\n// IsSetSessionAlreadyExistError returns true if SessionAlreadyExistError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetSessionAlreadyExistError() bool {\n\treturn v != nil && v.SessionAlreadyExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"StartWorkflowExecutionAsync\" for this struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) MethodName() string {\n\treturn \"StartWorkflowExecutionAsync\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_StartWorkflowExecutionAsync_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_TerminateWorkflowExecution_Args represents the arguments for the WorkflowService.TerminateWorkflowExecution function.\n//\n// The arguments for TerminateWorkflowExecution are sent and received over the wire as this struct.\ntype WorkflowService_TerminateWorkflowExecution_Args struct {\n\tTerminateRequest *shared.TerminateWorkflowExecutionRequest `json:\"terminateRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_TerminateWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TerminateRequest != nil {\n\t\tw, err = v.TerminateRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TerminateWorkflowExecutionRequest_Read(w wire.Value) (*shared.TerminateWorkflowExecutionRequest, error) {\n\tvar v shared.TerminateWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_TerminateWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_TerminateWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_TerminateWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_TerminateWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_TerminateWorkflowExecution_Args struct could not be encoded.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TerminateRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TerminateRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TerminateWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.TerminateWorkflowExecutionRequest, error) {\n\tvar v shared.TerminateWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_TerminateWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_TerminateWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_TerminateWorkflowExecution_Args\n// struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.TerminateRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"TerminateRequest: %v\", v.TerminateRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_TerminateWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_TerminateWorkflowExecution_Args match the\n// provided WorkflowService_TerminateWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) Equals(rhs *WorkflowService_TerminateWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TerminateRequest == nil && rhs.TerminateRequest == nil) || (v.TerminateRequest != nil && rhs.TerminateRequest != nil && v.TerminateRequest.Equals(rhs.TerminateRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_TerminateWorkflowExecution_Args.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TerminateRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"terminateRequest\", v.TerminateRequest))\n\t}\n\treturn err\n}\n\n// GetTerminateRequest returns the value of TerminateRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) GetTerminateRequest() (o *shared.TerminateWorkflowExecutionRequest) {\n\tif v != nil && v.TerminateRequest != nil {\n\t\treturn v.TerminateRequest\n\t}\n\n\treturn\n}\n\n// IsSetTerminateRequest returns true if TerminateRequest is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) IsSetTerminateRequest() bool {\n\treturn v != nil && v.TerminateRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"TerminateWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) MethodName() string {\n\treturn \"TerminateWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_TerminateWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.TerminateWorkflowExecution\n// function.\nvar WorkflowService_TerminateWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of TerminateWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tterminateRequest *shared.TerminateWorkflowExecutionRequest,\n\t) *WorkflowService_TerminateWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by TerminateWorkflowExecution.\n\t//\n\t// An error can be thrown by TerminateWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for TerminateWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if TerminateWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by TerminateWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// TerminateWorkflowExecution\n\t//\n\t//   err := TerminateWorkflowExecution(args)\n\t//   result, err := WorkflowService_TerminateWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from TerminateWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*WorkflowService_TerminateWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for TerminateWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if TerminateWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := WorkflowService_TerminateWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_TerminateWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tWorkflowService_TerminateWorkflowExecution_Helper.Args = func(\n\t\tterminateRequest *shared.TerminateWorkflowExecutionRequest,\n\t) *WorkflowService_TerminateWorkflowExecution_Args {\n\t\treturn &WorkflowService_TerminateWorkflowExecution_Args{\n\t\t\tTerminateRequest: terminateRequest,\n\t\t}\n\t}\n\n\tWorkflowService_TerminateWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_TerminateWorkflowExecution_Helper.WrapResponse = func(err error) (*WorkflowService_TerminateWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_TerminateWorkflowExecution_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_TerminateWorkflowExecution_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_TerminateWorkflowExecution_Helper.UnwrapResponse = func(result *WorkflowService_TerminateWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_TerminateWorkflowExecution_Result represents the result of a WorkflowService.TerminateWorkflowExecution function call.\n//\n// The result of a TerminateWorkflowExecution execution is sent and received over the wire as this struct.\ntype WorkflowService_TerminateWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tClientVersionNotSupportedError         *shared.ClientVersionNotSupportedError         `json:\"clientVersionNotSupportedError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n\tAccessDeniedError                      *shared.AccessDeniedError                      `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_TerminateWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowService_TerminateWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_TerminateWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_TerminateWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_TerminateWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_TerminateWorkflowExecution_Result struct could not be encoded.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowService_TerminateWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_TerminateWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_TerminateWorkflowExecution_Result\n// struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_TerminateWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_TerminateWorkflowExecution_Result match the\n// provided WorkflowService_TerminateWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) Equals(rhs *WorkflowService_TerminateWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_TerminateWorkflowExecution_Result.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"TerminateWorkflowExecution\" for this struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) MethodName() string {\n\treturn \"TerminateWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_TerminateWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// WorkflowService_UpdateDomain_Args represents the arguments for the WorkflowService.UpdateDomain function.\n//\n// The arguments for UpdateDomain are sent and received over the wire as this struct.\ntype WorkflowService_UpdateDomain_Args struct {\n\tUpdateRequest *shared.UpdateDomainRequest `json:\"updateRequest,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_UpdateDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_UpdateDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.UpdateRequest != nil {\n\t\tw, err = v.UpdateRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainRequest_Read(w wire.Value) (*shared.UpdateDomainRequest, error) {\n\tvar v shared.UpdateDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_UpdateDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_UpdateDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_UpdateDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_UpdateDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.UpdateRequest, err = _UpdateDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_UpdateDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_UpdateDomain_Args struct could not be encoded.\nfunc (v *WorkflowService_UpdateDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.UpdateRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.UpdateRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainRequest_Decode(sr stream.Reader) (*shared.UpdateDomainRequest, error) {\n\tvar v shared.UpdateDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_UpdateDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_UpdateDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_UpdateDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.UpdateRequest, err = _UpdateDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_UpdateDomain_Args\n// struct.\nfunc (v *WorkflowService_UpdateDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.UpdateRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpdateRequest: %v\", v.UpdateRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_UpdateDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_UpdateDomain_Args match the\n// provided WorkflowService_UpdateDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_UpdateDomain_Args) Equals(rhs *WorkflowService_UpdateDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.UpdateRequest == nil && rhs.UpdateRequest == nil) || (v.UpdateRequest != nil && rhs.UpdateRequest != nil && v.UpdateRequest.Equals(rhs.UpdateRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_UpdateDomain_Args.\nfunc (v *WorkflowService_UpdateDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.UpdateRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"updateRequest\", v.UpdateRequest))\n\t}\n\treturn err\n}\n\n// GetUpdateRequest returns the value of UpdateRequest if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Args) GetUpdateRequest() (o *shared.UpdateDomainRequest) {\n\tif v != nil && v.UpdateRequest != nil {\n\t\treturn v.UpdateRequest\n\t}\n\n\treturn\n}\n\n// IsSetUpdateRequest returns true if UpdateRequest is not nil.\nfunc (v *WorkflowService_UpdateDomain_Args) IsSetUpdateRequest() bool {\n\treturn v != nil && v.UpdateRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"UpdateDomain\" for this struct.\nfunc (v *WorkflowService_UpdateDomain_Args) MethodName() string {\n\treturn \"UpdateDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *WorkflowService_UpdateDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// WorkflowService_UpdateDomain_Helper provides functions that aid in handling the\n// parameters and return values of the WorkflowService.UpdateDomain\n// function.\nvar WorkflowService_UpdateDomain_Helper = struct {\n\t// Args accepts the parameters of UpdateDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tupdateRequest *shared.UpdateDomainRequest,\n\t) *WorkflowService_UpdateDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by UpdateDomain.\n\t//\n\t// An error can be thrown by UpdateDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for UpdateDomain\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// UpdateDomain into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by UpdateDomain\n\t//\n\t//   value, err := UpdateDomain(args)\n\t//   result, err := WorkflowService_UpdateDomain_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from UpdateDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.UpdateDomainResponse, error) (*WorkflowService_UpdateDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for UpdateDomain\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if UpdateDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := WorkflowService_UpdateDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*WorkflowService_UpdateDomain_Result) (*shared.UpdateDomainResponse, error)\n}{}\n\nfunc init() {\n\tWorkflowService_UpdateDomain_Helper.Args = func(\n\t\tupdateRequest *shared.UpdateDomainRequest,\n\t) *WorkflowService_UpdateDomain_Args {\n\t\treturn &WorkflowService_UpdateDomain_Args{\n\t\t\tUpdateRequest: updateRequest,\n\t\t}\n\t}\n\n\tWorkflowService_UpdateDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tWorkflowService_UpdateDomain_Helper.WrapResponse = func(success *shared.UpdateDomainResponse, err error) (*WorkflowService_UpdateDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &WorkflowService_UpdateDomain_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{ClientVersionNotSupportedError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for WorkflowService_UpdateDomain_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &WorkflowService_UpdateDomain_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tWorkflowService_UpdateDomain_Helper.UnwrapResponse = func(result *WorkflowService_UpdateDomain_Result) (success *shared.UpdateDomainResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// WorkflowService_UpdateDomain_Result represents the result of a WorkflowService.UpdateDomain function call.\n//\n// The result of a UpdateDomain execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype WorkflowService_UpdateDomain_Result struct {\n\t// Value returned by UpdateDomain after a successful execution.\n\tSuccess                        *shared.UpdateDomainResponse           `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tDomainNotActiveError           *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n\tAccessDeniedError              *shared.AccessDeniedError              `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a WorkflowService_UpdateDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowService_UpdateDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"WorkflowService_UpdateDomain_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainResponse_Read(w wire.Value) (*shared.UpdateDomainResponse, error) {\n\tvar v shared.UpdateDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowService_UpdateDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowService_UpdateDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowService_UpdateDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowService_UpdateDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _UpdateDomainResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_UpdateDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowService_UpdateDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowService_UpdateDomain_Result struct could not be encoded.\nfunc (v *WorkflowService_UpdateDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_UpdateDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainResponse_Decode(sr stream.Reader) (*shared.UpdateDomainResponse, error) {\n\tvar v shared.UpdateDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowService_UpdateDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowService_UpdateDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowService_UpdateDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _UpdateDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"WorkflowService_UpdateDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowService_UpdateDomain_Result\n// struct.\nfunc (v *WorkflowService_UpdateDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowService_UpdateDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowService_UpdateDomain_Result match the\n// provided WorkflowService_UpdateDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowService_UpdateDomain_Result) Equals(rhs *WorkflowService_UpdateDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowService_UpdateDomain_Result.\nfunc (v *WorkflowService_UpdateDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetSuccess() (o *shared.UpdateDomainResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowService_UpdateDomain_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *WorkflowService_UpdateDomain_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"UpdateDomain\" for this struct.\nfunc (v *WorkflowService_UpdateDomain_Result) MethodName() string {\n\treturn \"UpdateDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *WorkflowService_UpdateDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n"
  },
  {
    "path": ".gen/go/cadence/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage cadence\n"
  },
  {
    "path": ".gen/go/cadence/workflowserviceclient/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage workflowserviceclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\twire \"go.uber.org/thriftrw/wire\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tcadence \"github.com/uber/cadence/.gen/go/cadence\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is a client for the WorkflowService service.\ntype Interface interface {\n\tCountWorkflowExecutions(\n\t\tctx context.Context,\n\t\tCountRequest *shared.CountWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.CountWorkflowExecutionsResponse, error)\n\n\tDeleteDomain(\n\t\tctx context.Context,\n\t\tDeleteRequest *shared.DeleteDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tDeprecateDomain(\n\t\tctx context.Context,\n\t\tDeprecateRequest *shared.DeprecateDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tDescribeDomain(\n\t\tctx context.Context,\n\t\tDescribeRequest *shared.DescribeDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeDomainResponse, error)\n\n\tDescribeTaskList(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeTaskListRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeTaskListResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tDescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeWorkflowExecutionResponse, error)\n\n\tDiagnoseWorkflowExecution(\n\t\tctx context.Context,\n\t\tDiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DiagnoseWorkflowExecutionResponse, error)\n\n\tFailoverDomain(\n\t\tctx context.Context,\n\t\tFailoverRequest *shared.FailoverDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.FailoverDomainResponse, error)\n\n\tGetClusterInfo(\n\t\tctx context.Context,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ClusterInfo, error)\n\n\tGetSearchAttributes(\n\t\tctx context.Context,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetSearchAttributesResponse, error)\n\n\tGetTaskListsByDomain(\n\t\tctx context.Context,\n\t\tRequest *shared.GetTaskListsByDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetTaskListsByDomainResponse, error)\n\n\tGetWorkflowExecutionHistory(\n\t\tctx context.Context,\n\t\tGetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetWorkflowExecutionHistoryResponse, error)\n\n\tListArchivedWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListArchivedWorkflowExecutionsResponse, error)\n\n\tListClosedWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListClosedWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListClosedWorkflowExecutionsResponse, error)\n\n\tListDomains(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListDomainsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListDomainsResponse, error)\n\n\tListFailoverHistory(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListFailoverHistoryRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListFailoverHistoryResponse, error)\n\n\tListOpenWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListOpenWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListOpenWorkflowExecutionsResponse, error)\n\n\tListTaskListPartitions(\n\t\tctx context.Context,\n\t\tRequest *shared.ListTaskListPartitionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListTaskListPartitionsResponse, error)\n\n\tListWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListWorkflowExecutionsResponse, error)\n\n\tPollForActivityTask(\n\t\tctx context.Context,\n\t\tPollRequest *shared.PollForActivityTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.PollForActivityTaskResponse, error)\n\n\tPollForDecisionTask(\n\t\tctx context.Context,\n\t\tPollRequest *shared.PollForDecisionTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.PollForDecisionTaskResponse, error)\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *shared.QueryWorkflowRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.QueryWorkflowResponse, error)\n\n\tRecordActivityTaskHeartbeat(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRecordActivityTaskHeartbeatByID(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.RefreshWorkflowTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRegisterDomain(\n\t\tctx context.Context,\n\t\tRegisterRequest *shared.RegisterDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRequestCancelWorkflowExecution(\n\t\tctx context.Context,\n\t\tCancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tResetStickyTaskList(\n\t\tctx context.Context,\n\t\tResetRequest *shared.ResetStickyTaskListRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ResetStickyTaskListResponse, error)\n\n\tResetWorkflowExecution(\n\t\tctx context.Context,\n\t\tResetRequest *shared.ResetWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ResetWorkflowExecutionResponse, error)\n\n\tRespondActivityTaskCanceled(\n\t\tctx context.Context,\n\t\tCanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskCanceledByID(\n\t\tctx context.Context,\n\t\tCanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskCompletedByID(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskFailed(\n\t\tctx context.Context,\n\t\tFailRequest *shared.RespondActivityTaskFailedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskFailedByID(\n\t\tctx context.Context,\n\t\tFailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondDecisionTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RespondDecisionTaskCompletedResponse, error)\n\n\tRespondDecisionTaskFailed(\n\t\tctx context.Context,\n\t\tFailedRequest *shared.RespondDecisionTaskFailedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondQueryTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRestartWorkflowExecution(\n\t\tctx context.Context,\n\t\tRestartRequest *shared.RestartWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RestartWorkflowExecutionResponse, error)\n\n\tScanWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListWorkflowExecutionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListWorkflowExecutionsResponse, error)\n\n\tSignalWithStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSignalWithStartWorkflowExecutionAsync(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.SignalWithStartWorkflowExecutionAsyncResponse, error)\n\n\tSignalWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalRequest *shared.SignalWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tStartRequest *shared.StartWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tStartWorkflowExecutionAsync(\n\t\tctx context.Context,\n\t\tStartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.StartWorkflowExecutionAsyncResponse, error)\n\n\tTerminateWorkflowExecution(\n\t\tctx context.Context,\n\t\tTerminateRequest *shared.TerminateWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tUpdateDomain(\n\t\tctx context.Context,\n\t\tUpdateRequest *shared.UpdateDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.UpdateDomainResponse, error)\n}\n\n// New builds a new client for the WorkflowService service.\n//\n//\tclient := workflowserviceclient.New(dispatcher.ClientConfig(\"workflowservice\"))\nfunc New(c transport.ClientConfig, opts ...thrift.ClientOption) Interface {\n\treturn client{\n\t\tc: thrift.New(thrift.Config{\n\t\t\tService:      \"WorkflowService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t\tnwc: thrift.NewNoWire(thrift.Config{\n\t\t\tService:      \"WorkflowService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t}\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(c transport.ClientConfig, f reflect.StructField) Interface {\n\t\t\treturn New(c, thrift.ClientBuilderOptions(c, f)...)\n\t\t},\n\t)\n}\n\ntype client struct {\n\tc   thrift.Client\n\tnwc thrift.NoWireClient\n}\n\nfunc (c client) CountWorkflowExecutions(\n\tctx context.Context,\n\t_CountRequest *shared.CountWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.CountWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_CountWorkflowExecutions_Result\n\targs := cadence.WorkflowService_CountWorkflowExecutions_Helper.Args(_CountRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_CountWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DeleteDomain(\n\tctx context.Context,\n\t_DeleteRequest *shared.DeleteDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_DeleteDomain_Result\n\targs := cadence.WorkflowService_DeleteDomain_Helper.Args(_DeleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_DeleteDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DeprecateDomain(\n\tctx context.Context,\n\t_DeprecateRequest *shared.DeprecateDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_DeprecateDomain_Result\n\targs := cadence.WorkflowService_DeprecateDomain_Helper.Args(_DeprecateRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_DeprecateDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeDomain(\n\tctx context.Context,\n\t_DescribeRequest *shared.DescribeDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeDomainResponse, err error) {\n\n\tvar result cadence.WorkflowService_DescribeDomain_Result\n\targs := cadence.WorkflowService_DescribeDomain_Helper.Args(_DescribeRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_DescribeDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeTaskList(\n\tctx context.Context,\n\t_Request *shared.DescribeTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeTaskListResponse, err error) {\n\n\tvar result cadence.WorkflowService_DescribeTaskList_Result\n\targs := cadence.WorkflowService_DescribeTaskList_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_DescribeTaskList_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_DescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_DescribeWorkflowExecution_Result\n\targs := cadence.WorkflowService_DescribeWorkflowExecution_Helper.Args(_DescribeRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_DescribeWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DiagnoseWorkflowExecution(\n\tctx context.Context,\n\t_DiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DiagnoseWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_DiagnoseWorkflowExecution_Result\n\targs := cadence.WorkflowService_DiagnoseWorkflowExecution_Helper.Args(_DiagnoseRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_DiagnoseWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) FailoverDomain(\n\tctx context.Context,\n\t_FailoverRequest *shared.FailoverDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.FailoverDomainResponse, err error) {\n\n\tvar result cadence.WorkflowService_FailoverDomain_Result\n\targs := cadence.WorkflowService_FailoverDomain_Helper.Args(_FailoverRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_FailoverDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetClusterInfo(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *shared.ClusterInfo, err error) {\n\n\tvar result cadence.WorkflowService_GetClusterInfo_Result\n\targs := cadence.WorkflowService_GetClusterInfo_Helper.Args()\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_GetClusterInfo_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetSearchAttributes(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *shared.GetSearchAttributesResponse, err error) {\n\n\tvar result cadence.WorkflowService_GetSearchAttributes_Result\n\targs := cadence.WorkflowService_GetSearchAttributes_Helper.Args()\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_GetSearchAttributes_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetTaskListsByDomain(\n\tctx context.Context,\n\t_Request *shared.GetTaskListsByDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetTaskListsByDomainResponse, err error) {\n\n\tvar result cadence.WorkflowService_GetTaskListsByDomain_Result\n\targs := cadence.WorkflowService_GetTaskListsByDomain_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_GetTaskListsByDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetWorkflowExecutionHistory(\n\tctx context.Context,\n\t_GetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetWorkflowExecutionHistoryResponse, err error) {\n\n\tvar result cadence.WorkflowService_GetWorkflowExecutionHistory_Result\n\targs := cadence.WorkflowService_GetWorkflowExecutionHistory_Helper.Args(_GetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_GetWorkflowExecutionHistory_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListArchivedWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListArchivedWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListArchivedWorkflowExecutions_Result\n\targs := cadence.WorkflowService_ListArchivedWorkflowExecutions_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListArchivedWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListClosedWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListClosedWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListClosedWorkflowExecutions_Result\n\targs := cadence.WorkflowService_ListClosedWorkflowExecutions_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListClosedWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListDomains(\n\tctx context.Context,\n\t_ListRequest *shared.ListDomainsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListDomainsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListDomains_Result\n\targs := cadence.WorkflowService_ListDomains_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListDomains_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListFailoverHistory(\n\tctx context.Context,\n\t_ListRequest *shared.ListFailoverHistoryRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListFailoverHistoryResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListFailoverHistory_Result\n\targs := cadence.WorkflowService_ListFailoverHistory_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListFailoverHistory_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListOpenWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListOpenWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListOpenWorkflowExecutions_Result\n\targs := cadence.WorkflowService_ListOpenWorkflowExecutions_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListOpenWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListTaskListPartitions(\n\tctx context.Context,\n\t_Request *shared.ListTaskListPartitionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListTaskListPartitionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListTaskListPartitions_Result\n\targs := cadence.WorkflowService_ListTaskListPartitions_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListTaskListPartitions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ListWorkflowExecutions_Result\n\targs := cadence.WorkflowService_ListWorkflowExecutions_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ListWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PollForActivityTask(\n\tctx context.Context,\n\t_PollRequest *shared.PollForActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForActivityTaskResponse, err error) {\n\n\tvar result cadence.WorkflowService_PollForActivityTask_Result\n\targs := cadence.WorkflowService_PollForActivityTask_Helper.Args(_PollRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_PollForActivityTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PollForDecisionTask(\n\tctx context.Context,\n\t_PollRequest *shared.PollForDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForDecisionTaskResponse, err error) {\n\n\tvar result cadence.WorkflowService_PollForDecisionTask_Result\n\targs := cadence.WorkflowService_PollForDecisionTask_Helper.Args(_PollRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_PollForDecisionTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *shared.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.QueryWorkflowResponse, err error) {\n\n\tvar result cadence.WorkflowService_QueryWorkflow_Result\n\targs := cadence.WorkflowService_QueryWorkflow_Helper.Args(_QueryRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_QueryWorkflow_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\t_HeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\tvar result cadence.WorkflowService_RecordActivityTaskHeartbeat_Result\n\targs := cadence.WorkflowService_RecordActivityTaskHeartbeat_Helper.Args(_HeartbeatRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordActivityTaskHeartbeatByID(\n\tctx context.Context,\n\t_HeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\tvar result cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Result\n\targs := cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Helper.Args(_HeartbeatRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *shared.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RefreshWorkflowTasks_Result\n\targs := cadence.WorkflowService_RefreshWorkflowTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RefreshWorkflowTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RegisterDomain(\n\tctx context.Context,\n\t_RegisterRequest *shared.RegisterDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RegisterDomain_Result\n\targs := cadence.WorkflowService_RegisterDomain_Helper.Args(_RegisterRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RegisterDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\t_CancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RequestCancelWorkflowExecution_Result\n\targs := cadence.WorkflowService_RequestCancelWorkflowExecution_Helper.Args(_CancelRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RequestCancelWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetStickyTaskList(\n\tctx context.Context,\n\t_ResetRequest *shared.ResetStickyTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetStickyTaskListResponse, err error) {\n\n\tvar result cadence.WorkflowService_ResetStickyTaskList_Result\n\targs := cadence.WorkflowService_ResetStickyTaskList_Helper.Args(_ResetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ResetStickyTaskList_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetWorkflowExecution(\n\tctx context.Context,\n\t_ResetRequest *shared.ResetWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_ResetWorkflowExecution_Result\n\targs := cadence.WorkflowService_ResetWorkflowExecution_Helper.Args(_ResetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ResetWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCanceled(\n\tctx context.Context,\n\t_CanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskCanceled_Result\n\targs := cadence.WorkflowService_RespondActivityTaskCanceled_Helper.Args(_CanceledRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskCanceled_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCanceledByID(\n\tctx context.Context,\n\t_CanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskCanceledByID_Result\n\targs := cadence.WorkflowService_RespondActivityTaskCanceledByID_Helper.Args(_CanceledRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskCanceledByID_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskCompleted_Result\n\targs := cadence.WorkflowService_RespondActivityTaskCompleted_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCompletedByID(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskCompletedByID_Result\n\targs := cadence.WorkflowService_RespondActivityTaskCompletedByID_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskCompletedByID_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskFailed(\n\tctx context.Context,\n\t_FailRequest *shared.RespondActivityTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskFailed_Result\n\targs := cadence.WorkflowService_RespondActivityTaskFailed_Helper.Args(_FailRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskFailed_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskFailedByID(\n\tctx context.Context,\n\t_FailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondActivityTaskFailedByID_Result\n\targs := cadence.WorkflowService_RespondActivityTaskFailedByID_Helper.Args(_FailRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondActivityTaskFailedByID_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondDecisionTaskCompletedResponse, err error) {\n\n\tvar result cadence.WorkflowService_RespondDecisionTaskCompleted_Result\n\targs := cadence.WorkflowService_RespondDecisionTaskCompleted_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_RespondDecisionTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondDecisionTaskFailed(\n\tctx context.Context,\n\t_FailedRequest *shared.RespondDecisionTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondDecisionTaskFailed_Result\n\targs := cadence.WorkflowService_RespondDecisionTaskFailed_Helper.Args(_FailedRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondDecisionTaskFailed_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondQueryTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_RespondQueryTaskCompleted_Result\n\targs := cadence.WorkflowService_RespondQueryTaskCompleted_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_RespondQueryTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RestartWorkflowExecution(\n\tctx context.Context,\n\t_RestartRequest *shared.RestartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RestartWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_RestartWorkflowExecution_Result\n\targs := cadence.WorkflowService_RestartWorkflowExecution_Helper.Args(_RestartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_RestartWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ScanWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\n\tvar result cadence.WorkflowService_ScanWorkflowExecutions_Result\n\targs := cadence.WorkflowService_ScanWorkflowExecutions_Helper.Args(_ListRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_ScanWorkflowExecutions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\t_SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_SignalWithStartWorkflowExecution_Result\n\targs := cadence.WorkflowService_SignalWithStartWorkflowExecution_Helper.Args(_SignalWithStartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SignalWithStartWorkflowExecutionAsync(\n\tctx context.Context,\n\t_SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\n\tvar result cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Result\n\targs := cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.Args(_SignalWithStartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SignalWorkflowExecution(\n\tctx context.Context,\n\t_SignalRequest *shared.SignalWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_SignalWorkflowExecution_Result\n\targs := cadence.WorkflowService_SignalWorkflowExecution_Helper.Args(_SignalRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_SignalWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) StartWorkflowExecution(\n\tctx context.Context,\n\t_StartRequest *shared.StartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\tvar result cadence.WorkflowService_StartWorkflowExecution_Result\n\targs := cadence.WorkflowService_StartWorkflowExecution_Helper.Args(_StartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_StartWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) StartWorkflowExecutionAsync(\n\tctx context.Context,\n\t_StartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionAsyncResponse, err error) {\n\n\tvar result cadence.WorkflowService_StartWorkflowExecutionAsync_Result\n\targs := cadence.WorkflowService_StartWorkflowExecutionAsync_Helper.Args(_StartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_StartWorkflowExecutionAsync_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) TerminateWorkflowExecution(\n\tctx context.Context,\n\t_TerminateRequest *shared.TerminateWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result cadence.WorkflowService_TerminateWorkflowExecution_Result\n\targs := cadence.WorkflowService_TerminateWorkflowExecution_Helper.Args(_TerminateRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = cadence.WorkflowService_TerminateWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) UpdateDomain(\n\tctx context.Context,\n\t_UpdateRequest *shared.UpdateDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.UpdateDomainResponse, err error) {\n\n\tvar result cadence.WorkflowService_UpdateDomain_Result\n\targs := cadence.WorkflowService_UpdateDomain_Helper.Args(_UpdateRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = cadence.WorkflowService_UpdateDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n"
  },
  {
    "path": ".gen/go/cadence/workflowservicefx/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage workflowservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\trestriction \"go.uber.org/yarpc/api/x/restriction\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tworkflowserviceclient \"github.com/uber/cadence/.gen/go/cadence/workflowserviceclient\"\n)\n\n// Params defines the dependencies for the WorkflowService client.\ntype Params struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// Result defines the output of the WorkflowService client module. It provides a\n// WorkflowService client to an Fx application.\ntype Result struct {\n\tfx.Out\n\n\tClient workflowserviceclient.Interface\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// Client provides a WorkflowService client to an Fx application using the given name\n// for routing.\n//\n//\tfx.Provide(\n//\t\tworkflowservicefx.Client(\"...\"),\n//\t\tnewHandler,\n//\t)\nfunc Client(name string, opts ...thrift.ClientOption) interface{} {\n\treturn func(p Params) Result {\n\t\tcc := p.Provider.ClientConfig(name)\n\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok && p.Restriction != nil {\n\t\t\tif err := p.Restriction.Check(thrift.Encoding, namer.TransportName()); err != nil {\n\t\t\t\tpanic(err.Error())\n\t\t\t}\n\t\t}\n\t\tclient := workflowserviceclient.New(cc, opts...)\n\t\treturn Result{Client: client}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/cadence/workflowservicefx/doc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\n// Package workflowservicefx provides better integration for Fx for services\n// implementing or calling WorkflowService.\n//\n// # Clients\n//\n// If you are making requests to WorkflowService, use the Client function to inject a\n// WorkflowService client into your container.\n//\n//\tfx.Provide(workflowservicefx.Client(\"...\"))\n//\n// # Servers\n//\n// If you are implementing WorkflowService, provide a workflowserviceserver.Interface into\n// the container and use the Server function.\n//\n// Given,\n//\n//\tfunc NewWorkflowServiceHandler() workflowserviceserver.Interface\n//\n// You can do the following to have the procedures of WorkflowService made available\n// to an Fx application.\n//\n//\tfx.Provide(\n//\t\tNewWorkflowServiceHandler,\n//\t\tworkflowservicefx.Server(),\n//\t)\npackage workflowservicefx\n"
  },
  {
    "path": ".gen/go/cadence/workflowservicefx/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage workflowservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tworkflowserviceserver \"github.com/uber/cadence/.gen/go/cadence/workflowserviceserver\"\n)\n\n// ServerParams defines the dependencies for the WorkflowService server.\ntype ServerParams struct {\n\tfx.In\n\n\tHandler workflowserviceserver.Interface\n}\n\n// ServerResult defines the output of WorkflowService server module. It provides the\n// procedures of a WorkflowService handler to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group. Dig 1.2 or newer\n// must be used for this feature to work.\ntype ServerResult struct {\n\tfx.Out\n\n\tProcedures []transport.Procedure `group:\"yarpcfx\"`\n}\n\n// Server provides procedures for WorkflowService to an Fx application. It expects a\n// workflowservicefx.Interface to be present in the container.\n//\n//\tfx.Provide(\n//\t\tfunc(h *MyWorkflowServiceHandler) workflowserviceserver.Interface {\n//\t\t\treturn h\n//\t\t},\n//\t\tworkflowservicefx.Server(),\n//\t)\nfunc Server(opts ...thrift.RegisterOption) interface{} {\n\treturn func(p ServerParams) ServerResult {\n\t\tprocedures := workflowserviceserver.New(p.Handler, opts...)\n\t\treturn ServerResult{Procedures: procedures}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/cadence/workflowserviceserver/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage workflowserviceserver\n\nimport (\n\tcontext \"context\"\n\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\tyarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n\tcadence \"github.com/uber/cadence/.gen/go/cadence\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is the server-side interface for the WorkflowService service.\ntype Interface interface {\n\tCountWorkflowExecutions(\n\t\tctx context.Context,\n\t\tCountRequest *shared.CountWorkflowExecutionsRequest,\n\t) (*shared.CountWorkflowExecutionsResponse, error)\n\n\tDeleteDomain(\n\t\tctx context.Context,\n\t\tDeleteRequest *shared.DeleteDomainRequest,\n\t) error\n\n\tDeprecateDomain(\n\t\tctx context.Context,\n\t\tDeprecateRequest *shared.DeprecateDomainRequest,\n\t) error\n\n\tDescribeDomain(\n\t\tctx context.Context,\n\t\tDescribeRequest *shared.DescribeDomainRequest,\n\t) (*shared.DescribeDomainResponse, error)\n\n\tDescribeTaskList(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeTaskListRequest,\n\t) (*shared.DescribeTaskListResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tDescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\t) (*shared.DescribeWorkflowExecutionResponse, error)\n\n\tDiagnoseWorkflowExecution(\n\t\tctx context.Context,\n\t\tDiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\t) (*shared.DiagnoseWorkflowExecutionResponse, error)\n\n\tFailoverDomain(\n\t\tctx context.Context,\n\t\tFailoverRequest *shared.FailoverDomainRequest,\n\t) (*shared.FailoverDomainResponse, error)\n\n\tGetClusterInfo(\n\t\tctx context.Context,\n\t) (*shared.ClusterInfo, error)\n\n\tGetSearchAttributes(\n\t\tctx context.Context,\n\t) (*shared.GetSearchAttributesResponse, error)\n\n\tGetTaskListsByDomain(\n\t\tctx context.Context,\n\t\tRequest *shared.GetTaskListsByDomainRequest,\n\t) (*shared.GetTaskListsByDomainResponse, error)\n\n\tGetWorkflowExecutionHistory(\n\t\tctx context.Context,\n\t\tGetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\t) (*shared.GetWorkflowExecutionHistoryResponse, error)\n\n\tListArchivedWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\t) (*shared.ListArchivedWorkflowExecutionsResponse, error)\n\n\tListClosedWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListClosedWorkflowExecutionsRequest,\n\t) (*shared.ListClosedWorkflowExecutionsResponse, error)\n\n\tListDomains(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListDomainsRequest,\n\t) (*shared.ListDomainsResponse, error)\n\n\tListFailoverHistory(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListFailoverHistoryRequest,\n\t) (*shared.ListFailoverHistoryResponse, error)\n\n\tListOpenWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListOpenWorkflowExecutionsRequest,\n\t) (*shared.ListOpenWorkflowExecutionsResponse, error)\n\n\tListTaskListPartitions(\n\t\tctx context.Context,\n\t\tRequest *shared.ListTaskListPartitionsRequest,\n\t) (*shared.ListTaskListPartitionsResponse, error)\n\n\tListWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListWorkflowExecutionsRequest,\n\t) (*shared.ListWorkflowExecutionsResponse, error)\n\n\tPollForActivityTask(\n\t\tctx context.Context,\n\t\tPollRequest *shared.PollForActivityTaskRequest,\n\t) (*shared.PollForActivityTaskResponse, error)\n\n\tPollForDecisionTask(\n\t\tctx context.Context,\n\t\tPollRequest *shared.PollForDecisionTaskRequest,\n\t) (*shared.PollForDecisionTaskResponse, error)\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *shared.QueryWorkflowRequest,\n\t) (*shared.QueryWorkflowResponse, error)\n\n\tRecordActivityTaskHeartbeat(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRecordActivityTaskHeartbeatByID(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.RefreshWorkflowTasksRequest,\n\t) error\n\n\tRegisterDomain(\n\t\tctx context.Context,\n\t\tRegisterRequest *shared.RegisterDomainRequest,\n\t) error\n\n\tRequestCancelWorkflowExecution(\n\t\tctx context.Context,\n\t\tCancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\t) error\n\n\tResetStickyTaskList(\n\t\tctx context.Context,\n\t\tResetRequest *shared.ResetStickyTaskListRequest,\n\t) (*shared.ResetStickyTaskListResponse, error)\n\n\tResetWorkflowExecution(\n\t\tctx context.Context,\n\t\tResetRequest *shared.ResetWorkflowExecutionRequest,\n\t) (*shared.ResetWorkflowExecutionResponse, error)\n\n\tRespondActivityTaskCanceled(\n\t\tctx context.Context,\n\t\tCanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\t) error\n\n\tRespondActivityTaskCanceledByID(\n\t\tctx context.Context,\n\t\tCanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\t) error\n\n\tRespondActivityTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\t) error\n\n\tRespondActivityTaskCompletedByID(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\t) error\n\n\tRespondActivityTaskFailed(\n\t\tctx context.Context,\n\t\tFailRequest *shared.RespondActivityTaskFailedRequest,\n\t) error\n\n\tRespondActivityTaskFailedByID(\n\t\tctx context.Context,\n\t\tFailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\t) error\n\n\tRespondDecisionTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\t) (*shared.RespondDecisionTaskCompletedResponse, error)\n\n\tRespondDecisionTaskFailed(\n\t\tctx context.Context,\n\t\tFailedRequest *shared.RespondDecisionTaskFailedRequest,\n\t) error\n\n\tRespondQueryTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\t) error\n\n\tRestartWorkflowExecution(\n\t\tctx context.Context,\n\t\tRestartRequest *shared.RestartWorkflowExecutionRequest,\n\t) (*shared.RestartWorkflowExecutionResponse, error)\n\n\tScanWorkflowExecutions(\n\t\tctx context.Context,\n\t\tListRequest *shared.ListWorkflowExecutionsRequest,\n\t) (*shared.ListWorkflowExecutionsResponse, error)\n\n\tSignalWithStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSignalWithStartWorkflowExecutionAsync(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\t) (*shared.SignalWithStartWorkflowExecutionAsyncResponse, error)\n\n\tSignalWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalRequest *shared.SignalWorkflowExecutionRequest,\n\t) error\n\n\tStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tStartRequest *shared.StartWorkflowExecutionRequest,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tStartWorkflowExecutionAsync(\n\t\tctx context.Context,\n\t\tStartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\t) (*shared.StartWorkflowExecutionAsyncResponse, error)\n\n\tTerminateWorkflowExecution(\n\t\tctx context.Context,\n\t\tTerminateRequest *shared.TerminateWorkflowExecutionRequest,\n\t) error\n\n\tUpdateDomain(\n\t\tctx context.Context,\n\t\tUpdateRequest *shared.UpdateDomainRequest,\n\t) (*shared.UpdateDomainResponse, error)\n}\n\n// New prepares an implementation of the WorkflowService service for\n// registration.\n//\n//\thandler := WorkflowServiceHandler{}\n//\tdispatcher.Register(workflowserviceserver.New(handler))\nfunc New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {\n\th := handler{impl}\n\tservice := thrift.Service{\n\t\tName: \"WorkflowService\",\n\t\tMethods: []thrift.Method{\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"CountWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.CountWorkflowExecutions),\n\t\t\t\t\tNoWire: countworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"CountWorkflowExecutions(CountRequest *shared.CountWorkflowExecutionsRequest) (*shared.CountWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DeleteDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DeleteDomain),\n\t\t\t\t\tNoWire: deletedomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DeleteDomain(DeleteRequest *shared.DeleteDomainRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DeprecateDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DeprecateDomain),\n\t\t\t\t\tNoWire: deprecatedomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DeprecateDomain(DeprecateRequest *shared.DeprecateDomainRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeDomain),\n\t\t\t\t\tNoWire: describedomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeDomain(DescribeRequest *shared.DescribeDomainRequest) (*shared.DescribeDomainResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeTaskList\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeTaskList),\n\t\t\t\t\tNoWire: describetasklist_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeTaskList(Request *shared.DescribeTaskListRequest) (*shared.DescribeTaskListResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeWorkflowExecution),\n\t\t\t\t\tNoWire: describeworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeWorkflowExecution(DescribeRequest *shared.DescribeWorkflowExecutionRequest) (*shared.DescribeWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DiagnoseWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DiagnoseWorkflowExecution),\n\t\t\t\t\tNoWire: diagnoseworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DiagnoseWorkflowExecution(DiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest) (*shared.DiagnoseWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"FailoverDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.FailoverDomain),\n\t\t\t\t\tNoWire: failoverdomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"FailoverDomain(FailoverRequest *shared.FailoverDomainRequest) (*shared.FailoverDomainResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetClusterInfo\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetClusterInfo),\n\t\t\t\t\tNoWire: getclusterinfo_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetClusterInfo() (*shared.ClusterInfo)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetSearchAttributes\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetSearchAttributes),\n\t\t\t\t\tNoWire: getsearchattributes_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetSearchAttributes() (*shared.GetSearchAttributesResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetTaskListsByDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetTaskListsByDomain),\n\t\t\t\t\tNoWire: gettasklistsbydomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetTaskListsByDomain(Request *shared.GetTaskListsByDomainRequest) (*shared.GetTaskListsByDomainResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetWorkflowExecutionHistory\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetWorkflowExecutionHistory),\n\t\t\t\t\tNoWire: getworkflowexecutionhistory_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetWorkflowExecutionHistory(GetRequest *shared.GetWorkflowExecutionHistoryRequest) (*shared.GetWorkflowExecutionHistoryResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListArchivedWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListArchivedWorkflowExecutions),\n\t\t\t\t\tNoWire: listarchivedworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListArchivedWorkflowExecutions(ListRequest *shared.ListArchivedWorkflowExecutionsRequest) (*shared.ListArchivedWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListClosedWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListClosedWorkflowExecutions),\n\t\t\t\t\tNoWire: listclosedworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListClosedWorkflowExecutions(ListRequest *shared.ListClosedWorkflowExecutionsRequest) (*shared.ListClosedWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListDomains\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListDomains),\n\t\t\t\t\tNoWire: listdomains_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListDomains(ListRequest *shared.ListDomainsRequest) (*shared.ListDomainsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListFailoverHistory\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListFailoverHistory),\n\t\t\t\t\tNoWire: listfailoverhistory_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListFailoverHistory(ListRequest *shared.ListFailoverHistoryRequest) (*shared.ListFailoverHistoryResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListOpenWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListOpenWorkflowExecutions),\n\t\t\t\t\tNoWire: listopenworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListOpenWorkflowExecutions(ListRequest *shared.ListOpenWorkflowExecutionsRequest) (*shared.ListOpenWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListTaskListPartitions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListTaskListPartitions),\n\t\t\t\t\tNoWire: listtasklistpartitions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListTaskListPartitions(Request *shared.ListTaskListPartitionsRequest) (*shared.ListTaskListPartitionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListWorkflowExecutions),\n\t\t\t\t\tNoWire: listworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListWorkflowExecutions(ListRequest *shared.ListWorkflowExecutionsRequest) (*shared.ListWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PollForActivityTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PollForActivityTask),\n\t\t\t\t\tNoWire: pollforactivitytask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PollForActivityTask(PollRequest *shared.PollForActivityTaskRequest) (*shared.PollForActivityTaskResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PollForDecisionTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PollForDecisionTask),\n\t\t\t\t\tNoWire: pollfordecisiontask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PollForDecisionTask(PollRequest *shared.PollForDecisionTaskRequest) (*shared.PollForDecisionTaskResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"QueryWorkflow\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.QueryWorkflow),\n\t\t\t\t\tNoWire: queryworkflow_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"QueryWorkflow(QueryRequest *shared.QueryWorkflowRequest) (*shared.QueryWorkflowResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordActivityTaskHeartbeat\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordActivityTaskHeartbeat),\n\t\t\t\t\tNoWire: recordactivitytaskheartbeat_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordActivityTaskHeartbeat(HeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest) (*shared.RecordActivityTaskHeartbeatResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordActivityTaskHeartbeatByID\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordActivityTaskHeartbeatByID),\n\t\t\t\t\tNoWire: recordactivitytaskheartbeatbyid_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordActivityTaskHeartbeatByID(HeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest) (*shared.RecordActivityTaskHeartbeatResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RefreshWorkflowTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RefreshWorkflowTasks),\n\t\t\t\t\tNoWire: refreshworkflowtasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RefreshWorkflowTasks(Request *shared.RefreshWorkflowTasksRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RegisterDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RegisterDomain),\n\t\t\t\t\tNoWire: registerdomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RegisterDomain(RegisterRequest *shared.RegisterDomainRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RequestCancelWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RequestCancelWorkflowExecution),\n\t\t\t\t\tNoWire: requestcancelworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RequestCancelWorkflowExecution(CancelRequest *shared.RequestCancelWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetStickyTaskList\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetStickyTaskList),\n\t\t\t\t\tNoWire: resetstickytasklist_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetStickyTaskList(ResetRequest *shared.ResetStickyTaskListRequest) (*shared.ResetStickyTaskListResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetWorkflowExecution),\n\t\t\t\t\tNoWire: resetworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetWorkflowExecution(ResetRequest *shared.ResetWorkflowExecutionRequest) (*shared.ResetWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCanceled\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCanceled),\n\t\t\t\t\tNoWire: respondactivitytaskcanceled_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCanceled(CanceledRequest *shared.RespondActivityTaskCanceledRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCanceledByID\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCanceledByID),\n\t\t\t\t\tNoWire: respondactivitytaskcanceledbyid_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCanceledByID(CanceledRequest *shared.RespondActivityTaskCanceledByIDRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCompleted),\n\t\t\t\t\tNoWire: respondactivitytaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCompleted(CompleteRequest *shared.RespondActivityTaskCompletedRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCompletedByID\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCompletedByID),\n\t\t\t\t\tNoWire: respondactivitytaskcompletedbyid_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCompletedByID(CompleteRequest *shared.RespondActivityTaskCompletedByIDRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskFailed\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskFailed),\n\t\t\t\t\tNoWire: respondactivitytaskfailed_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskFailed(FailRequest *shared.RespondActivityTaskFailedRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskFailedByID\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskFailedByID),\n\t\t\t\t\tNoWire: respondactivitytaskfailedbyid_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskFailedByID(FailRequest *shared.RespondActivityTaskFailedByIDRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondDecisionTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondDecisionTaskCompleted),\n\t\t\t\t\tNoWire: responddecisiontaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondDecisionTaskCompleted(CompleteRequest *shared.RespondDecisionTaskCompletedRequest) (*shared.RespondDecisionTaskCompletedResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondDecisionTaskFailed\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondDecisionTaskFailed),\n\t\t\t\t\tNoWire: responddecisiontaskfailed_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondDecisionTaskFailed(FailedRequest *shared.RespondDecisionTaskFailedRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondQueryTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondQueryTaskCompleted),\n\t\t\t\t\tNoWire: respondquerytaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondQueryTaskCompleted(CompleteRequest *shared.RespondQueryTaskCompletedRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RestartWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RestartWorkflowExecution),\n\t\t\t\t\tNoWire: restartworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RestartWorkflowExecution(RestartRequest *shared.RestartWorkflowExecutionRequest) (*shared.RestartWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ScanWorkflowExecutions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ScanWorkflowExecutions),\n\t\t\t\t\tNoWire: scanworkflowexecutions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ScanWorkflowExecutions(ListRequest *shared.ListWorkflowExecutionsRequest) (*shared.ListWorkflowExecutionsResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SignalWithStartWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SignalWithStartWorkflowExecution),\n\t\t\t\t\tNoWire: signalwithstartworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SignalWithStartWorkflowExecution(SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest) (*shared.StartWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SignalWithStartWorkflowExecutionAsync\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SignalWithStartWorkflowExecutionAsync),\n\t\t\t\t\tNoWire: signalwithstartworkflowexecutionasync_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SignalWithStartWorkflowExecutionAsync(SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest) (*shared.SignalWithStartWorkflowExecutionAsyncResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SignalWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SignalWorkflowExecution),\n\t\t\t\t\tNoWire: signalworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SignalWorkflowExecution(SignalRequest *shared.SignalWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"StartWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.StartWorkflowExecution),\n\t\t\t\t\tNoWire: startworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"StartWorkflowExecution(StartRequest *shared.StartWorkflowExecutionRequest) (*shared.StartWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"StartWorkflowExecutionAsync\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.StartWorkflowExecutionAsync),\n\t\t\t\t\tNoWire: startworkflowexecutionasync_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"StartWorkflowExecutionAsync(StartRequest *shared.StartWorkflowExecutionAsyncRequest) (*shared.StartWorkflowExecutionAsyncResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"TerminateWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.TerminateWorkflowExecution),\n\t\t\t\t\tNoWire: terminateworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"TerminateWorkflowExecution(TerminateRequest *shared.TerminateWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"UpdateDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.UpdateDomain),\n\t\t\t\t\tNoWire: updatedomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"UpdateDomain(UpdateRequest *shared.UpdateDomainRequest) (*shared.UpdateDomainResponse)\",\n\t\t\t\tThriftModule: cadence.ThriftModule,\n\t\t\t},\n\t\t},\n\t}\n\n\tprocedures := make([]transport.Procedure, 0, 47)\n\tprocedures = append(procedures, thrift.BuildProcedures(service, opts...)...)\n\treturn procedures\n}\n\ntype handler struct{ impl Interface }\n\ntype yarpcErrorNamer interface{ YARPCErrorName() string }\n\ntype yarpcErrorCoder interface{ YARPCErrorCode() *yarpcerrors.Code }\n\nfunc (h handler) CountWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_CountWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'CountWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.CountWorkflowExecutions(ctx, args.CountRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_CountWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DeleteDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DeleteDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DeleteDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.DeleteDomain(ctx, args.DeleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DeleteDomain_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DeprecateDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DeprecateDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DeprecateDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.DeprecateDomain(ctx, args.DeprecateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DeprecateDomain_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DescribeDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DescribeDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeDomain(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeDomain_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeTaskList(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DescribeTaskList_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DescribeTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeTaskList(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeTaskList_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DescribeWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DiagnoseWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_DiagnoseWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'DiagnoseWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DiagnoseWorkflowExecution(ctx, args.DiagnoseRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DiagnoseWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) FailoverDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_FailoverDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'FailoverDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.FailoverDomain(ctx, args.FailoverRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_FailoverDomain_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetClusterInfo(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_GetClusterInfo_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'GetClusterInfo': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetClusterInfo(ctx)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetClusterInfo_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetSearchAttributes(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_GetSearchAttributes_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'GetSearchAttributes': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetSearchAttributes(ctx)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetSearchAttributes_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetTaskListsByDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_GetTaskListsByDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'GetTaskListsByDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetTaskListsByDomain(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetTaskListsByDomain_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetWorkflowExecutionHistory(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_GetWorkflowExecutionHistory_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'GetWorkflowExecutionHistory': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetWorkflowExecutionHistory(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetWorkflowExecutionHistory_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListArchivedWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListArchivedWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListArchivedWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListArchivedWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListArchivedWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListClosedWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListClosedWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListClosedWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListClosedWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListClosedWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListDomains(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListDomains_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListDomains': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListDomains(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListDomains_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListFailoverHistory(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListFailoverHistory_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListFailoverHistory': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListFailoverHistory(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListFailoverHistory_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListOpenWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListOpenWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListOpenWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListOpenWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListOpenWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListTaskListPartitions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListTaskListPartitions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListTaskListPartitions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListTaskListPartitions(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListTaskListPartitions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ListWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ListWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PollForActivityTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_PollForActivityTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'PollForActivityTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForActivityTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_PollForActivityTask_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PollForDecisionTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_PollForDecisionTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'PollForDecisionTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForDecisionTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_PollForDecisionTask_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) QueryWorkflow(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_QueryWorkflow_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordActivityTaskHeartbeat(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RecordActivityTaskHeartbeat_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RecordActivityTaskHeartbeat': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeat(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RecordActivityTaskHeartbeat_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordActivityTaskHeartbeatByID(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RecordActivityTaskHeartbeatByID': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeatByID(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RefreshWorkflowTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RefreshWorkflowTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RegisterDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RegisterDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RegisterDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.RegisterDomain(ctx, args.RegisterRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RegisterDomain_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RequestCancelWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RequestCancelWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RequestCancelWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.RequestCancelWorkflowExecution(ctx, args.CancelRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RequestCancelWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetStickyTaskList(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ResetStickyTaskList_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ResetStickyTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetStickyTaskList(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ResetStickyTaskList_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ResetWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ResetWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetWorkflowExecution(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ResetWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCanceled(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskCanceled_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCanceled': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceled(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCanceled_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCanceledByID(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskCanceledByID_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCanceledByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceledByID(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCanceledByID_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCompleted_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCompletedByID(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskCompletedByID_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCompletedByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompletedByID(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCompletedByID_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskFailed(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskFailed_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailed(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskFailed_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskFailedByID(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondActivityTaskFailedByID_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskFailedByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailedByID(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskFailedByID_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondDecisionTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondDecisionTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondDecisionTaskCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondDecisionTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondDecisionTaskCompleted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondDecisionTaskFailed(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondDecisionTaskFailed_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondDecisionTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondDecisionTaskFailed(ctx, args.FailedRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondDecisionTaskFailed_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondQueryTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RespondQueryTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RespondQueryTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondQueryTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondQueryTaskCompleted_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RestartWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_RestartWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'RestartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RestartWorkflowExecution(ctx, args.RestartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RestartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ScanWorkflowExecutions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_ScanWorkflowExecutions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'ScanWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ScanWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ScanWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SignalWithStartWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_SignalWithStartWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'SignalWithStartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecution(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWithStartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'SignalWithStartWorkflowExecutionAsync': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecutionAsync(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SignalWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_SignalWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'SignalWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.SignalWorkflowExecution(ctx, args.SignalRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) StartWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_StartWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'StartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecution(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_StartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) StartWorkflowExecutionAsync(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_StartWorkflowExecutionAsync_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'StartWorkflowExecutionAsync': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecutionAsync(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_StartWorkflowExecutionAsync_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) TerminateWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_TerminateWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'TerminateWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.TerminateWorkflowExecution(ctx, args.TerminateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_TerminateWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) UpdateDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args cadence.WorkflowService_UpdateDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'WorkflowService' procedure 'UpdateDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomain(ctx, args.UpdateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_UpdateDomain_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\ntype countworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h countworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_CountWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'CountWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.CountWorkflowExecutions(ctx, args.CountRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_CountWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype deletedomain_NoWireHandler struct{ impl Interface }\n\nfunc (h deletedomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DeleteDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DeleteDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.DeleteDomain(ctx, args.DeleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DeleteDomain_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype deprecatedomain_NoWireHandler struct{ impl Interface }\n\nfunc (h deprecatedomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DeprecateDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DeprecateDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.DeprecateDomain(ctx, args.DeprecateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DeprecateDomain_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describedomain_NoWireHandler struct{ impl Interface }\n\nfunc (h describedomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DescribeDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DescribeDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeDomain(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeDomain_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describetasklist_NoWireHandler struct{ impl Interface }\n\nfunc (h describetasklist_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DescribeTaskList_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DescribeTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeTaskList(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeTaskList_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describeworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h describeworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DescribeWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype diagnoseworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h diagnoseworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_DiagnoseWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'DiagnoseWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DiagnoseWorkflowExecution(ctx, args.DiagnoseRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_DiagnoseWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype failoverdomain_NoWireHandler struct{ impl Interface }\n\nfunc (h failoverdomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_FailoverDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'FailoverDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.FailoverDomain(ctx, args.FailoverRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_FailoverDomain_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getclusterinfo_NoWireHandler struct{ impl Interface }\n\nfunc (h getclusterinfo_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_GetClusterInfo_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'GetClusterInfo': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetClusterInfo(ctx)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetClusterInfo_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getsearchattributes_NoWireHandler struct{ impl Interface }\n\nfunc (h getsearchattributes_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_GetSearchAttributes_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'GetSearchAttributes': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetSearchAttributes(ctx)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetSearchAttributes_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype gettasklistsbydomain_NoWireHandler struct{ impl Interface }\n\nfunc (h gettasklistsbydomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_GetTaskListsByDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'GetTaskListsByDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetTaskListsByDomain(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetTaskListsByDomain_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getworkflowexecutionhistory_NoWireHandler struct{ impl Interface }\n\nfunc (h getworkflowexecutionhistory_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_GetWorkflowExecutionHistory_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'GetWorkflowExecutionHistory': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetWorkflowExecutionHistory(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_GetWorkflowExecutionHistory_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listarchivedworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h listarchivedworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListArchivedWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListArchivedWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListArchivedWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListArchivedWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listclosedworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h listclosedworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListClosedWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListClosedWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListClosedWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListClosedWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listdomains_NoWireHandler struct{ impl Interface }\n\nfunc (h listdomains_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListDomains_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListDomains': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListDomains(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListDomains_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listfailoverhistory_NoWireHandler struct{ impl Interface }\n\nfunc (h listfailoverhistory_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListFailoverHistory_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListFailoverHistory': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListFailoverHistory(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListFailoverHistory_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listopenworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h listopenworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListOpenWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListOpenWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListOpenWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListOpenWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listtasklistpartitions_NoWireHandler struct{ impl Interface }\n\nfunc (h listtasklistpartitions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListTaskListPartitions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListTaskListPartitions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListTaskListPartitions(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListTaskListPartitions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h listworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ListWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ListWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ListWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype pollforactivitytask_NoWireHandler struct{ impl Interface }\n\nfunc (h pollforactivitytask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_PollForActivityTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'PollForActivityTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForActivityTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_PollForActivityTask_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype pollfordecisiontask_NoWireHandler struct{ impl Interface }\n\nfunc (h pollfordecisiontask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_PollForDecisionTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'PollForDecisionTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForDecisionTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_PollForDecisionTask_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype queryworkflow_NoWireHandler struct{ impl Interface }\n\nfunc (h queryworkflow_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_QueryWorkflow_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recordactivitytaskheartbeat_NoWireHandler struct{ impl Interface }\n\nfunc (h recordactivitytaskheartbeat_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RecordActivityTaskHeartbeat_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RecordActivityTaskHeartbeat': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeat(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RecordActivityTaskHeartbeat_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recordactivitytaskheartbeatbyid_NoWireHandler struct{ impl Interface }\n\nfunc (h recordactivitytaskheartbeatbyid_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RecordActivityTaskHeartbeatByID': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeatByID(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RecordActivityTaskHeartbeatByID_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype refreshworkflowtasks_NoWireHandler struct{ impl Interface }\n\nfunc (h refreshworkflowtasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RefreshWorkflowTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype registerdomain_NoWireHandler struct{ impl Interface }\n\nfunc (h registerdomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RegisterDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RegisterDomain': %w\", err)\n\t}\n\n\tappErr := h.impl.RegisterDomain(ctx, args.RegisterRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RegisterDomain_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype requestcancelworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h requestcancelworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RequestCancelWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RequestCancelWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.RequestCancelWorkflowExecution(ctx, args.CancelRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RequestCancelWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetstickytasklist_NoWireHandler struct{ impl Interface }\n\nfunc (h resetstickytasklist_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ResetStickyTaskList_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ResetStickyTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetStickyTaskList(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ResetStickyTaskList_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h resetworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ResetWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ResetWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetWorkflowExecution(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ResetWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcanceled_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcanceled_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskCanceled_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCanceled': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceled(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCanceled_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcanceledbyid_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcanceledbyid_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskCanceledByID_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCanceledByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceledByID(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCanceledByID_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCompleted_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcompletedbyid_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcompletedbyid_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskCompletedByID_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskCompletedByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompletedByID(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskCompletedByID_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskfailed_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskfailed_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskFailed_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailed(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskFailed_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskfailedbyid_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskfailedbyid_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondActivityTaskFailedByID_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondActivityTaskFailedByID': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailedByID(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondActivityTaskFailedByID_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype responddecisiontaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h responddecisiontaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondDecisionTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondDecisionTaskCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondDecisionTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondDecisionTaskCompleted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype responddecisiontaskfailed_NoWireHandler struct{ impl Interface }\n\nfunc (h responddecisiontaskfailed_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondDecisionTaskFailed_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondDecisionTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondDecisionTaskFailed(ctx, args.FailedRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondDecisionTaskFailed_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondquerytaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondquerytaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RespondQueryTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RespondQueryTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondQueryTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RespondQueryTaskCompleted_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype restartworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h restartworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_RestartWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'RestartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RestartWorkflowExecution(ctx, args.RestartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_RestartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype scanworkflowexecutions_NoWireHandler struct{ impl Interface }\n\nfunc (h scanworkflowexecutions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_ScanWorkflowExecutions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'ScanWorkflowExecutions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ScanWorkflowExecutions(ctx, args.ListRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_ScanWorkflowExecutions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype signalwithstartworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h signalwithstartworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_SignalWithStartWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'SignalWithStartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecution(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWithStartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype signalwithstartworkflowexecutionasync_NoWireHandler struct{ impl Interface }\n\nfunc (h signalwithstartworkflowexecutionasync_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'SignalWithStartWorkflowExecutionAsync': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecutionAsync(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWithStartWorkflowExecutionAsync_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype signalworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h signalworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_SignalWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'SignalWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.SignalWorkflowExecution(ctx, args.SignalRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_SignalWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype startworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h startworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_StartWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'StartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecution(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_StartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype startworkflowexecutionasync_NoWireHandler struct{ impl Interface }\n\nfunc (h startworkflowexecutionasync_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_StartWorkflowExecutionAsync_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'StartWorkflowExecutionAsync': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecutionAsync(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_StartWorkflowExecutionAsync_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype terminateworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h terminateworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_TerminateWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'TerminateWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.TerminateWorkflowExecution(ctx, args.TerminateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_TerminateWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype updatedomain_NoWireHandler struct{ impl Interface }\n\nfunc (h updatedomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs cadence.WorkflowService_UpdateDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'WorkflowService' procedure 'UpdateDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.UpdateDomain(ctx, args.UpdateRequest)\n\n\thadError := appErr != nil\n\tresult, err := cadence.WorkflowService_UpdateDomain_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n"
  },
  {
    "path": ".gen/go/cadence/workflowservicetest/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage workflowservicetest\n\nimport (\n\tcontext \"context\"\n\n\tgomock \"github.com/golang/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tworkflowserviceclient \"github.com/uber/cadence/.gen/go/cadence/workflowserviceclient\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// MockClient implements a gomock-compatible mock client for service\n// WorkflowService.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *_MockClientRecorder\n}\n\nvar _ workflowserviceclient.Interface = (*MockClient)(nil)\n\ntype _MockClientRecorder struct {\n\tmock *MockClient\n}\n\n// Build a new mock client for service WorkflowService.\n//\n//\tmockCtrl := gomock.NewController(t)\n//\tclient := workflowservicetest.NewMockClient(mockCtrl)\n//\n// Use EXPECT() to set expectations on the mock.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &_MockClientRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows you to define an expectation on the\n// WorkflowService mock client.\nfunc (m *MockClient) EXPECT() *_MockClientRecorder {\n\treturn m.recorder\n}\n\n// CountWorkflowExecutions responds to a CountWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().CountWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.CountWorkflowExecutions(...)\nfunc (m *MockClient) CountWorkflowExecutions(\n\tctx context.Context,\n\t_CountRequest *shared.CountWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.CountWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _CountRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"CountWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.CountWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) CountWorkflowExecutions(\n\tctx interface{},\n\t_CountRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CountRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"CountWorkflowExecutions\", args...)\n}\n\n// DeleteDomain responds to a DeleteDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DeleteDomain(gomock.Any(), ...).Return(...)\n//\t... := client.DeleteDomain(...)\nfunc (m *MockClient) DeleteDomain(\n\tctx context.Context,\n\t_DeleteRequest *shared.DeleteDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _DeleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DeleteDomain\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DeleteDomain(\n\tctx interface{},\n\t_DeleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DeleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DeleteDomain\", args...)\n}\n\n// DeprecateDomain responds to a DeprecateDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DeprecateDomain(gomock.Any(), ...).Return(...)\n//\t... := client.DeprecateDomain(...)\nfunc (m *MockClient) DeprecateDomain(\n\tctx context.Context,\n\t_DeprecateRequest *shared.DeprecateDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _DeprecateRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DeprecateDomain\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DeprecateDomain(\n\tctx interface{},\n\t_DeprecateRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DeprecateRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DeprecateDomain\", args...)\n}\n\n// DescribeDomain responds to a DescribeDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeDomain(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeDomain(...)\nfunc (m *MockClient) DescribeDomain(\n\tctx context.Context,\n\t_DescribeRequest *shared.DescribeDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeDomainResponse, err error) {\n\n\targs := []interface{}{ctx, _DescribeRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeDomain\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeDomainResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeDomain(\n\tctx interface{},\n\t_DescribeRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DescribeRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeDomain\", args...)\n}\n\n// DescribeTaskList responds to a DescribeTaskList call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeTaskList(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeTaskList(...)\nfunc (m *MockClient) DescribeTaskList(\n\tctx context.Context,\n\t_Request *shared.DescribeTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeTaskListResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeTaskListResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeTaskList(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeTaskList\", args...)\n}\n\n// DescribeWorkflowExecution responds to a DescribeWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeWorkflowExecution(...)\nfunc (m *MockClient) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_DescribeRequest *shared.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _DescribeRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeWorkflowExecution(\n\tctx interface{},\n\t_DescribeRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DescribeRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeWorkflowExecution\", args...)\n}\n\n// DiagnoseWorkflowExecution responds to a DiagnoseWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DiagnoseWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.DiagnoseWorkflowExecution(...)\nfunc (m *MockClient) DiagnoseWorkflowExecution(\n\tctx context.Context,\n\t_DiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DiagnoseWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _DiagnoseRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DiagnoseWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.DiagnoseWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DiagnoseWorkflowExecution(\n\tctx interface{},\n\t_DiagnoseRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DiagnoseRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DiagnoseWorkflowExecution\", args...)\n}\n\n// FailoverDomain responds to a FailoverDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().FailoverDomain(gomock.Any(), ...).Return(...)\n//\t... := client.FailoverDomain(...)\nfunc (m *MockClient) FailoverDomain(\n\tctx context.Context,\n\t_FailoverRequest *shared.FailoverDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.FailoverDomainResponse, err error) {\n\n\targs := []interface{}{ctx, _FailoverRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"FailoverDomain\", args...)\n\tsuccess, _ = ret[i].(*shared.FailoverDomainResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) FailoverDomain(\n\tctx interface{},\n\t_FailoverRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailoverRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"FailoverDomain\", args...)\n}\n\n// GetClusterInfo responds to a GetClusterInfo call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetClusterInfo(gomock.Any(), ...).Return(...)\n//\t... := client.GetClusterInfo(...)\nfunc (m *MockClient) GetClusterInfo(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *shared.ClusterInfo, err error) {\n\n\targs := []interface{}{ctx}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetClusterInfo\", args...)\n\tsuccess, _ = ret[i].(*shared.ClusterInfo)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetClusterInfo(\n\tctx interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetClusterInfo\", args...)\n}\n\n// GetSearchAttributes responds to a GetSearchAttributes call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetSearchAttributes(gomock.Any(), ...).Return(...)\n//\t... := client.GetSearchAttributes(...)\nfunc (m *MockClient) GetSearchAttributes(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *shared.GetSearchAttributesResponse, err error) {\n\n\targs := []interface{}{ctx}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetSearchAttributes\", args...)\n\tsuccess, _ = ret[i].(*shared.GetSearchAttributesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetSearchAttributes(\n\tctx interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetSearchAttributes\", args...)\n}\n\n// GetTaskListsByDomain responds to a GetTaskListsByDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetTaskListsByDomain(gomock.Any(), ...).Return(...)\n//\t... := client.GetTaskListsByDomain(...)\nfunc (m *MockClient) GetTaskListsByDomain(\n\tctx context.Context,\n\t_Request *shared.GetTaskListsByDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetTaskListsByDomainResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", args...)\n\tsuccess, _ = ret[i].(*shared.GetTaskListsByDomainResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetTaskListsByDomain(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetTaskListsByDomain\", args...)\n}\n\n// GetWorkflowExecutionHistory responds to a GetWorkflowExecutionHistory call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), ...).Return(...)\n//\t... := client.GetWorkflowExecutionHistory(...)\nfunc (m *MockClient) GetWorkflowExecutionHistory(\n\tctx context.Context,\n\t_GetRequest *shared.GetWorkflowExecutionHistoryRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetWorkflowExecutionHistoryResponse, err error) {\n\n\targs := []interface{}{ctx, _GetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionHistory\", args...)\n\tsuccess, _ = ret[i].(*shared.GetWorkflowExecutionHistoryResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetWorkflowExecutionHistory(\n\tctx interface{},\n\t_GetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _GetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetWorkflowExecutionHistory\", args...)\n}\n\n// ListArchivedWorkflowExecutions responds to a ListArchivedWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListArchivedWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.ListArchivedWorkflowExecutions(...)\nfunc (m *MockClient) ListArchivedWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListArchivedWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListArchivedWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListArchivedWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListArchivedWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListArchivedWorkflowExecutions(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListArchivedWorkflowExecutions\", args...)\n}\n\n// ListClosedWorkflowExecutions responds to a ListClosedWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.ListClosedWorkflowExecutions(...)\nfunc (m *MockClient) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListClosedWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListClosedWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListClosedWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListClosedWorkflowExecutions(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListClosedWorkflowExecutions\", args...)\n}\n\n// ListDomains responds to a ListDomains call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListDomains(gomock.Any(), ...).Return(...)\n//\t... := client.ListDomains(...)\nfunc (m *MockClient) ListDomains(\n\tctx context.Context,\n\t_ListRequest *shared.ListDomainsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListDomainsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListDomains\", args...)\n\tsuccess, _ = ret[i].(*shared.ListDomainsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListDomains(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListDomains\", args...)\n}\n\n// ListFailoverHistory responds to a ListFailoverHistory call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListFailoverHistory(gomock.Any(), ...).Return(...)\n//\t... := client.ListFailoverHistory(...)\nfunc (m *MockClient) ListFailoverHistory(\n\tctx context.Context,\n\t_ListRequest *shared.ListFailoverHistoryRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListFailoverHistoryResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListFailoverHistory\", args...)\n\tsuccess, _ = ret[i].(*shared.ListFailoverHistoryResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListFailoverHistory(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListFailoverHistory\", args...)\n}\n\n// ListOpenWorkflowExecutions responds to a ListOpenWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.ListOpenWorkflowExecutions(...)\nfunc (m *MockClient) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListOpenWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListOpenWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListOpenWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListOpenWorkflowExecutions(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListOpenWorkflowExecutions\", args...)\n}\n\n// ListTaskListPartitions responds to a ListTaskListPartitions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListTaskListPartitions(gomock.Any(), ...).Return(...)\n//\t... := client.ListTaskListPartitions(...)\nfunc (m *MockClient) ListTaskListPartitions(\n\tctx context.Context,\n\t_Request *shared.ListTaskListPartitionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListTaskListPartitionsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListTaskListPartitionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListTaskListPartitions(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListTaskListPartitions\", args...)\n}\n\n// ListWorkflowExecutions responds to a ListWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.ListWorkflowExecutions(...)\nfunc (m *MockClient) ListWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListWorkflowExecutions(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListWorkflowExecutions\", args...)\n}\n\n// PollForActivityTask responds to a PollForActivityTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PollForActivityTask(gomock.Any(), ...).Return(...)\n//\t... := client.PollForActivityTask(...)\nfunc (m *MockClient) PollForActivityTask(\n\tctx context.Context,\n\t_PollRequest *shared.PollForActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForActivityTaskResponse, err error) {\n\n\targs := []interface{}{ctx, _PollRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", args...)\n\tsuccess, _ = ret[i].(*shared.PollForActivityTaskResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PollForActivityTask(\n\tctx interface{},\n\t_PollRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _PollRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PollForActivityTask\", args...)\n}\n\n// PollForDecisionTask responds to a PollForDecisionTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PollForDecisionTask(gomock.Any(), ...).Return(...)\n//\t... := client.PollForDecisionTask(...)\nfunc (m *MockClient) PollForDecisionTask(\n\tctx context.Context,\n\t_PollRequest *shared.PollForDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForDecisionTaskResponse, err error) {\n\n\targs := []interface{}{ctx, _PollRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", args...)\n\tsuccess, _ = ret[i].(*shared.PollForDecisionTaskResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PollForDecisionTask(\n\tctx interface{},\n\t_PollRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _PollRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PollForDecisionTask\", args...)\n}\n\n// QueryWorkflow responds to a QueryWorkflow call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().QueryWorkflow(gomock.Any(), ...).Return(...)\n//\t... := client.QueryWorkflow(...)\nfunc (m *MockClient) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *shared.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.QueryWorkflowResponse, err error) {\n\n\targs := []interface{}{ctx, _QueryRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", args...)\n\tsuccess, _ = ret[i].(*shared.QueryWorkflowResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) QueryWorkflow(\n\tctx interface{},\n\t_QueryRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _QueryRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"QueryWorkflow\", args...)\n}\n\n// RecordActivityTaskHeartbeat responds to a RecordActivityTaskHeartbeat call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), ...).Return(...)\n//\t... := client.RecordActivityTaskHeartbeat(...)\nfunc (m *MockClient) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\t_HeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\targs := []interface{}{ctx, _HeartbeatRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", args...)\n\tsuccess, _ = ret[i].(*shared.RecordActivityTaskHeartbeatResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordActivityTaskHeartbeat(\n\tctx interface{},\n\t_HeartbeatRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _HeartbeatRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordActivityTaskHeartbeat\", args...)\n}\n\n// RecordActivityTaskHeartbeatByID responds to a RecordActivityTaskHeartbeatByID call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordActivityTaskHeartbeatByID(gomock.Any(), ...).Return(...)\n//\t... := client.RecordActivityTaskHeartbeatByID(...)\nfunc (m *MockClient) RecordActivityTaskHeartbeatByID(\n\tctx context.Context,\n\t_HeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\targs := []interface{}{ctx, _HeartbeatRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeatByID\", args...)\n\tsuccess, _ = ret[i].(*shared.RecordActivityTaskHeartbeatResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordActivityTaskHeartbeatByID(\n\tctx interface{},\n\t_HeartbeatRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _HeartbeatRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordActivityTaskHeartbeatByID\", args...)\n}\n\n// RefreshWorkflowTasks responds to a RefreshWorkflowTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RefreshWorkflowTasks(gomock.Any(), ...).Return(...)\n//\t... := client.RefreshWorkflowTasks(...)\nfunc (m *MockClient) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *shared.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RefreshWorkflowTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RefreshWorkflowTasks\", args...)\n}\n\n// RegisterDomain responds to a RegisterDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RegisterDomain(gomock.Any(), ...).Return(...)\n//\t... := client.RegisterDomain(...)\nfunc (m *MockClient) RegisterDomain(\n\tctx context.Context,\n\t_RegisterRequest *shared.RegisterDomainRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _RegisterRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RegisterDomain\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RegisterDomain(\n\tctx interface{},\n\t_RegisterRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _RegisterRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RegisterDomain\", args...)\n}\n\n// RequestCancelWorkflowExecution responds to a RequestCancelWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.RequestCancelWorkflowExecution(...)\nfunc (m *MockClient) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\t_CancelRequest *shared.RequestCancelWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CancelRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RequestCancelWorkflowExecution(\n\tctx interface{},\n\t_CancelRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CancelRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RequestCancelWorkflowExecution\", args...)\n}\n\n// ResetStickyTaskList responds to a ResetStickyTaskList call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetStickyTaskList(gomock.Any(), ...).Return(...)\n//\t... := client.ResetStickyTaskList(...)\nfunc (m *MockClient) ResetStickyTaskList(\n\tctx context.Context,\n\t_ResetRequest *shared.ResetStickyTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetStickyTaskListResponse, err error) {\n\n\targs := []interface{}{ctx, _ResetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", args...)\n\tsuccess, _ = ret[i].(*shared.ResetStickyTaskListResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetStickyTaskList(\n\tctx interface{},\n\t_ResetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ResetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetStickyTaskList\", args...)\n}\n\n// ResetWorkflowExecution responds to a ResetWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.ResetWorkflowExecution(...)\nfunc (m *MockClient) ResetWorkflowExecution(\n\tctx context.Context,\n\t_ResetRequest *shared.ResetWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _ResetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.ResetWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetWorkflowExecution(\n\tctx interface{},\n\t_ResetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ResetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetWorkflowExecution\", args...)\n}\n\n// RespondActivityTaskCanceled responds to a RespondActivityTaskCanceled call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCanceled(...)\nfunc (m *MockClient) RespondActivityTaskCanceled(\n\tctx context.Context,\n\t_CanceledRequest *shared.RespondActivityTaskCanceledRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CanceledRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCanceled(\n\tctx interface{},\n\t_CanceledRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CanceledRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCanceled\", args...)\n}\n\n// RespondActivityTaskCanceledByID responds to a RespondActivityTaskCanceledByID call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCanceledByID(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCanceledByID(...)\nfunc (m *MockClient) RespondActivityTaskCanceledByID(\n\tctx context.Context,\n\t_CanceledRequest *shared.RespondActivityTaskCanceledByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CanceledRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceledByID\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCanceledByID(\n\tctx interface{},\n\t_CanceledRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CanceledRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCanceledByID\", args...)\n}\n\n// RespondActivityTaskCompleted responds to a RespondActivityTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCompleted(...)\nfunc (m *MockClient) RespondActivityTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondActivityTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCompleted(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCompleted\", args...)\n}\n\n// RespondActivityTaskCompletedByID responds to a RespondActivityTaskCompletedByID call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCompletedByID(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCompletedByID(...)\nfunc (m *MockClient) RespondActivityTaskCompletedByID(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondActivityTaskCompletedByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompletedByID\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCompletedByID(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCompletedByID\", args...)\n}\n\n// RespondActivityTaskFailed responds to a RespondActivityTaskFailed call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskFailed(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskFailed(...)\nfunc (m *MockClient) RespondActivityTaskFailed(\n\tctx context.Context,\n\t_FailRequest *shared.RespondActivityTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _FailRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskFailed(\n\tctx interface{},\n\t_FailRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskFailed\", args...)\n}\n\n// RespondActivityTaskFailedByID responds to a RespondActivityTaskFailedByID call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskFailedByID(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskFailedByID(...)\nfunc (m *MockClient) RespondActivityTaskFailedByID(\n\tctx context.Context,\n\t_FailRequest *shared.RespondActivityTaskFailedByIDRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _FailRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailedByID\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskFailedByID(\n\tctx interface{},\n\t_FailRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskFailedByID\", args...)\n}\n\n// RespondDecisionTaskCompleted responds to a RespondDecisionTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondDecisionTaskCompleted(...)\nfunc (m *MockClient) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondDecisionTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondDecisionTaskCompletedResponse, err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", args...)\n\tsuccess, _ = ret[i].(*shared.RespondDecisionTaskCompletedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondDecisionTaskCompleted(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondDecisionTaskCompleted\", args...)\n}\n\n// RespondDecisionTaskFailed responds to a RespondDecisionTaskFailed call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondDecisionTaskFailed(gomock.Any(), ...).Return(...)\n//\t... := client.RespondDecisionTaskFailed(...)\nfunc (m *MockClient) RespondDecisionTaskFailed(\n\tctx context.Context,\n\t_FailedRequest *shared.RespondDecisionTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _FailedRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondDecisionTaskFailed(\n\tctx interface{},\n\t_FailedRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailedRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondDecisionTaskFailed\", args...)\n}\n\n// RespondQueryTaskCompleted responds to a RespondQueryTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondQueryTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondQueryTaskCompleted(...)\nfunc (m *MockClient) RespondQueryTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *shared.RespondQueryTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondQueryTaskCompleted(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondQueryTaskCompleted\", args...)\n}\n\n// RestartWorkflowExecution responds to a RestartWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RestartWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.RestartWorkflowExecution(...)\nfunc (m *MockClient) RestartWorkflowExecution(\n\tctx context.Context,\n\t_RestartRequest *shared.RestartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RestartWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _RestartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RestartWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.RestartWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RestartWorkflowExecution(\n\tctx interface{},\n\t_RestartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _RestartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RestartWorkflowExecution\", args...)\n}\n\n// ScanWorkflowExecutions responds to a ScanWorkflowExecutions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ScanWorkflowExecutions(gomock.Any(), ...).Return(...)\n//\t... := client.ScanWorkflowExecutions(...)\nfunc (m *MockClient) ScanWorkflowExecutions(\n\tctx context.Context,\n\t_ListRequest *shared.ListWorkflowExecutionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListWorkflowExecutionsResponse, err error) {\n\n\targs := []interface{}{ctx, _ListRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ScanWorkflowExecutions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListWorkflowExecutionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ScanWorkflowExecutions(\n\tctx interface{},\n\t_ListRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ListRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ScanWorkflowExecutions\", args...)\n}\n\n// SignalWithStartWorkflowExecution responds to a SignalWithStartWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.SignalWithStartWorkflowExecution(...)\nfunc (m *MockClient) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\t_SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _SignalWithStartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.StartWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SignalWithStartWorkflowExecution(\n\tctx interface{},\n\t_SignalWithStartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SignalWithStartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SignalWithStartWorkflowExecution\", args...)\n}\n\n// SignalWithStartWorkflowExecutionAsync responds to a SignalWithStartWorkflowExecutionAsync call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SignalWithStartWorkflowExecutionAsync(gomock.Any(), ...).Return(...)\n//\t... := client.SignalWithStartWorkflowExecutionAsync(...)\nfunc (m *MockClient) SignalWithStartWorkflowExecutionAsync(\n\tctx context.Context,\n\t_SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\n\targs := []interface{}{ctx, _SignalWithStartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecutionAsync\", args...)\n\tsuccess, _ = ret[i].(*shared.SignalWithStartWorkflowExecutionAsyncResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SignalWithStartWorkflowExecutionAsync(\n\tctx interface{},\n\t_SignalWithStartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SignalWithStartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SignalWithStartWorkflowExecutionAsync\", args...)\n}\n\n// SignalWorkflowExecution responds to a SignalWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SignalWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.SignalWorkflowExecution(...)\nfunc (m *MockClient) SignalWorkflowExecution(\n\tctx context.Context,\n\t_SignalRequest *shared.SignalWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _SignalRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SignalWorkflowExecution(\n\tctx interface{},\n\t_SignalRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SignalRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SignalWorkflowExecution\", args...)\n}\n\n// StartWorkflowExecution responds to a StartWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().StartWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.StartWorkflowExecution(...)\nfunc (m *MockClient) StartWorkflowExecution(\n\tctx context.Context,\n\t_StartRequest *shared.StartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _StartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.StartWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) StartWorkflowExecution(\n\tctx interface{},\n\t_StartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _StartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"StartWorkflowExecution\", args...)\n}\n\n// StartWorkflowExecutionAsync responds to a StartWorkflowExecutionAsync call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().StartWorkflowExecutionAsync(gomock.Any(), ...).Return(...)\n//\t... := client.StartWorkflowExecutionAsync(...)\nfunc (m *MockClient) StartWorkflowExecutionAsync(\n\tctx context.Context,\n\t_StartRequest *shared.StartWorkflowExecutionAsyncRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionAsyncResponse, err error) {\n\n\targs := []interface{}{ctx, _StartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"StartWorkflowExecutionAsync\", args...)\n\tsuccess, _ = ret[i].(*shared.StartWorkflowExecutionAsyncResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) StartWorkflowExecutionAsync(\n\tctx interface{},\n\t_StartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _StartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"StartWorkflowExecutionAsync\", args...)\n}\n\n// TerminateWorkflowExecution responds to a TerminateWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().TerminateWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.TerminateWorkflowExecution(...)\nfunc (m *MockClient) TerminateWorkflowExecution(\n\tctx context.Context,\n\t_TerminateRequest *shared.TerminateWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _TerminateRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) TerminateWorkflowExecution(\n\tctx interface{},\n\t_TerminateRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _TerminateRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"TerminateWorkflowExecution\", args...)\n}\n\n// UpdateDomain responds to a UpdateDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().UpdateDomain(gomock.Any(), ...).Return(...)\n//\t... := client.UpdateDomain(...)\nfunc (m *MockClient) UpdateDomain(\n\tctx context.Context,\n\t_UpdateRequest *shared.UpdateDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.UpdateDomainResponse, err error) {\n\n\targs := []interface{}{ctx, _UpdateRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"UpdateDomain\", args...)\n\tsuccess, _ = ret[i].(*shared.UpdateDomainResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) UpdateDomain(\n\tctx interface{},\n\t_UpdateRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _UpdateRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"UpdateDomain\", args...)\n}\n"
  },
  {
    "path": ".gen/go/checksum/checksum.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage checksum\n\nimport (\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype MutableStateChecksumPayload struct {\n\tCancelRequested              *bool                    `json:\"cancelRequested,omitempty\"`\n\tState                        *int16                   `json:\"state,omitempty\"`\n\tCloseStatus                  *int16                   `json:\"closeStatus,omitempty\"`\n\tLastWriteVersion             *int64                   `json:\"lastWriteVersion,omitempty\"`\n\tLastWriteEventID             *int64                   `json:\"lastWriteEventID,omitempty\"`\n\tLastFirstEventID             *int64                   `json:\"lastFirstEventID,omitempty\"`\n\tNextEventID                  *int64                   `json:\"nextEventID,omitempty\"`\n\tLastProcessedEventID         *int64                   `json:\"lastProcessedEventID,omitempty\"`\n\tSignalCount                  *int64                   `json:\"signalCount,omitempty\"`\n\tDecisionAttempt              *int32                   `json:\"decisionAttempt,omitempty\"`\n\tDecisionVersion              *int64                   `json:\"decisionVersion,omitempty\"`\n\tDecisionScheduledID          *int64                   `json:\"decisionScheduledID,omitempty\"`\n\tDecisionStartedID            *int64                   `json:\"decisionStartedID,omitempty\"`\n\tPendingTimerStartedIDs       []int64                  `json:\"pendingTimerStartedIDs,omitempty\"`\n\tPendingActivityScheduledIDs  []int64                  `json:\"pendingActivityScheduledIDs,omitempty\"`\n\tPendingSignalInitiatedIDs    []int64                  `json:\"pendingSignalInitiatedIDs,omitempty\"`\n\tPendingReqCancelInitiatedIDs []int64                  `json:\"pendingReqCancelInitiatedIDs,omitempty\"`\n\tPendingChildInitiatedIDs     []int64                  `json:\"pendingChildInitiatedIDs,omitempty\"`\n\tStickyTaskListName           *string                  `json:\"stickyTaskListName,omitempty\"`\n\tVersionHistories             *shared.VersionHistories `json:\"VersionHistories,omitempty\"`\n}\n\ntype _List_I64_ValueList []int64\n\nfunc (v _List_I64_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueI64(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_I64_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_I64_ValueList) ValueType() wire.Type {\n\treturn wire.TI64\n}\n\nfunc (_List_I64_ValueList) Close() {}\n\n// ToWire translates a MutableStateChecksumPayload struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MutableStateChecksumPayload) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [20]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CancelRequested != nil {\n\t\tw, err = wire.NewValueBool(*(v.CancelRequested)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tw, err = wire.NewValueI16(*(v.State)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 15, Value: w}\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tw, err = wire.NewValueI16(*(v.CloseStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.LastWriteVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastWriteVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 21, Value: w}\n\t\ti++\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastWriteEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastFirstEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 23, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.LastProcessedEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastProcessedEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 25, Value: w}\n\t\ti++\n\t}\n\tif v.SignalCount != nil {\n\t\tw, err = wire.NewValueI64(*(v.SignalCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.DecisionAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionScheduledID != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionScheduledID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 37, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionStartedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\tif v.PendingTimerStartedIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I64_ValueList(v.PendingTimerStartedIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 45, Value: w}\n\t\ti++\n\t}\n\tif v.PendingActivityScheduledIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I64_ValueList(v.PendingActivityScheduledIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 46, Value: w}\n\t\ti++\n\t}\n\tif v.PendingSignalInitiatedIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I64_ValueList(v.PendingSignalInitiatedIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 47, Value: w}\n\t\ti++\n\t}\n\tif v.PendingReqCancelInitiatedIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I64_ValueList(v.PendingReqCancelInitiatedIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 48, Value: w}\n\t\ti++\n\t}\n\tif v.PendingChildInitiatedIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I64_ValueList(v.PendingChildInitiatedIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 49, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskListName != nil {\n\t\tw, err = wire.NewValueString(*(v.StickyTaskListName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tw, err = v.VersionHistories.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_I64_Read(l wire.ValueList) ([]int64, error) {\n\tif l.ValueType() != wire.TI64 {\n\t\treturn nil, nil\n\t}\n\n\to := make([]int64, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetI64(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _VersionHistories_Read(w wire.Value) (*shared.VersionHistories, error) {\n\tvar v shared.VersionHistories\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MutableStateChecksumPayload struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MutableStateChecksumPayload struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MutableStateChecksumPayload\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MutableStateChecksumPayload) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.CancelRequested = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 15:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.State = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.CloseStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 21:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastWriteVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastWriteEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 23:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastFirstEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 25:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastProcessedEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.SignalCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DecisionAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 37:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionScheduledID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionStartedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 45:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingTimerStartedIDs, err = _List_I64_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 46:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingActivityScheduledIDs, err = _List_I64_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 47:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingSignalInitiatedIDs, err = _List_I64_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 48:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingReqCancelInitiatedIDs, err = _List_I64_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 49:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingChildInitiatedIDs, err = _List_I64_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StickyTaskListName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistories, err = _VersionHistories_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_I64_Encode(val []int64, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TI64,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteInt64(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a MutableStateChecksumPayload struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MutableStateChecksumPayload struct could not be encoded.\nfunc (v *MutableStateChecksumPayload) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CancelRequested != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.CancelRequested)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.State != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 15, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.State)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CloseStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.CloseStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWriteVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 21, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastWriteVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWriteEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastWriteEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFirstEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 23, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastFirstEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastProcessedEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 25, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastProcessedEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.SignalCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DecisionAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionScheduledID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 37, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionScheduledID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionStartedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionStartedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingTimerStartedIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 45, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I64_Encode(v.PendingTimerStartedIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingActivityScheduledIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 46, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I64_Encode(v.PendingActivityScheduledIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingSignalInitiatedIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 47, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I64_Encode(v.PendingSignalInitiatedIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingReqCancelInitiatedIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 48, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I64_Encode(v.PendingReqCancelInitiatedIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingChildInitiatedIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 49, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I64_Encode(v.PendingChildInitiatedIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskListName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StickyTaskListName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistories != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistories.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_I64_Decode(sr stream.Reader) ([]int64, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TI64 {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]int64, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadInt64()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _VersionHistories_Decode(sr stream.Reader) (*shared.VersionHistories, error) {\n\tvar v shared.VersionHistories\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MutableStateChecksumPayload struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MutableStateChecksumPayload struct could not be generated from the wire\n// representation.\nfunc (v *MutableStateChecksumPayload) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.CancelRequested = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 15 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.State = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.CloseStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 21 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastWriteVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastWriteEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 23 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastFirstEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 25 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastProcessedEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.SignalCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DecisionAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 37 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionScheduledID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionStartedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 45 && fh.Type == wire.TList:\n\t\t\tv.PendingTimerStartedIDs, err = _List_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 46 && fh.Type == wire.TList:\n\t\t\tv.PendingActivityScheduledIDs, err = _List_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 47 && fh.Type == wire.TList:\n\t\t\tv.PendingSignalInitiatedIDs, err = _List_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 48 && fh.Type == wire.TList:\n\t\t\tv.PendingReqCancelInitiatedIDs, err = _List_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 49 && fh.Type == wire.TList:\n\t\t\tv.PendingChildInitiatedIDs, err = _List_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StickyTaskListName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistories, err = _VersionHistories_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MutableStateChecksumPayload\n// struct.\nfunc (v *MutableStateChecksumPayload) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [20]string\n\ti := 0\n\tif v.CancelRequested != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequested: %v\", *(v.CancelRequested))\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tfields[i] = fmt.Sprintf(\"State: %v\", *(v.State))\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"CloseStatus: %v\", *(v.CloseStatus))\n\t\ti++\n\t}\n\tif v.LastWriteVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWriteVersion: %v\", *(v.LastWriteVersion))\n\t\ti++\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWriteEventID: %v\", *(v.LastWriteEventID))\n\t\ti++\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFirstEventID: %v\", *(v.LastFirstEventID))\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventID: %v\", *(v.NextEventID))\n\t\ti++\n\t}\n\tif v.LastProcessedEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastProcessedEventID: %v\", *(v.LastProcessedEventID))\n\t\ti++\n\t}\n\tif v.SignalCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalCount: %v\", *(v.SignalCount))\n\t\ti++\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionAttempt: %v\", *(v.DecisionAttempt))\n\t\ti++\n\t}\n\tif v.DecisionVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionVersion: %v\", *(v.DecisionVersion))\n\t\ti++\n\t}\n\tif v.DecisionScheduledID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionScheduledID: %v\", *(v.DecisionScheduledID))\n\t\ti++\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionStartedID: %v\", *(v.DecisionStartedID))\n\t\ti++\n\t}\n\tif v.PendingTimerStartedIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingTimerStartedIDs: %v\", v.PendingTimerStartedIDs)\n\t\ti++\n\t}\n\tif v.PendingActivityScheduledIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingActivityScheduledIDs: %v\", v.PendingActivityScheduledIDs)\n\t\ti++\n\t}\n\tif v.PendingSignalInitiatedIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingSignalInitiatedIDs: %v\", v.PendingSignalInitiatedIDs)\n\t\ti++\n\t}\n\tif v.PendingReqCancelInitiatedIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingReqCancelInitiatedIDs: %v\", v.PendingReqCancelInitiatedIDs)\n\t\ti++\n\t}\n\tif v.PendingChildInitiatedIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingChildInitiatedIDs: %v\", v.PendingChildInitiatedIDs)\n\t\ti++\n\t}\n\tif v.StickyTaskListName != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskListName: %v\", *(v.StickyTaskListName))\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistories: %v\", v.VersionHistories)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MutableStateChecksumPayload{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I16_EqualsPtr(lhs, rhs *int16) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_I64_Equals(lhs, rhs []int64) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this MutableStateChecksumPayload match the\n// provided MutableStateChecksumPayload.\n//\n// This function performs a deep comparison.\nfunc (v *MutableStateChecksumPayload) Equals(rhs *MutableStateChecksumPayload) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.CancelRequested, rhs.CancelRequested) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.State, rhs.State) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.CloseStatus, rhs.CloseStatus) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastWriteVersion, rhs.LastWriteVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastWriteEventID, rhs.LastWriteEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastFirstEventID, rhs.LastFirstEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventID, rhs.NextEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastProcessedEventID, rhs.LastProcessedEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.SignalCount, rhs.SignalCount) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DecisionAttempt, rhs.DecisionAttempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionVersion, rhs.DecisionVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionScheduledID, rhs.DecisionScheduledID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionStartedID, rhs.DecisionStartedID) {\n\t\treturn false\n\t}\n\tif !((v.PendingTimerStartedIDs == nil && rhs.PendingTimerStartedIDs == nil) || (v.PendingTimerStartedIDs != nil && rhs.PendingTimerStartedIDs != nil && _List_I64_Equals(v.PendingTimerStartedIDs, rhs.PendingTimerStartedIDs))) {\n\t\treturn false\n\t}\n\tif !((v.PendingActivityScheduledIDs == nil && rhs.PendingActivityScheduledIDs == nil) || (v.PendingActivityScheduledIDs != nil && rhs.PendingActivityScheduledIDs != nil && _List_I64_Equals(v.PendingActivityScheduledIDs, rhs.PendingActivityScheduledIDs))) {\n\t\treturn false\n\t}\n\tif !((v.PendingSignalInitiatedIDs == nil && rhs.PendingSignalInitiatedIDs == nil) || (v.PendingSignalInitiatedIDs != nil && rhs.PendingSignalInitiatedIDs != nil && _List_I64_Equals(v.PendingSignalInitiatedIDs, rhs.PendingSignalInitiatedIDs))) {\n\t\treturn false\n\t}\n\tif !((v.PendingReqCancelInitiatedIDs == nil && rhs.PendingReqCancelInitiatedIDs == nil) || (v.PendingReqCancelInitiatedIDs != nil && rhs.PendingReqCancelInitiatedIDs != nil && _List_I64_Equals(v.PendingReqCancelInitiatedIDs, rhs.PendingReqCancelInitiatedIDs))) {\n\t\treturn false\n\t}\n\tif !((v.PendingChildInitiatedIDs == nil && rhs.PendingChildInitiatedIDs == nil) || (v.PendingChildInitiatedIDs != nil && rhs.PendingChildInitiatedIDs != nil && _List_I64_Equals(v.PendingChildInitiatedIDs, rhs.PendingChildInitiatedIDs))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StickyTaskListName, rhs.StickyTaskListName) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistories == nil && rhs.VersionHistories == nil) || (v.VersionHistories != nil && rhs.VersionHistories != nil && v.VersionHistories.Equals(rhs.VersionHistories))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_I64_Zapper []int64\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_I64_Zapper.\nfunc (l _List_I64_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendInt64(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MutableStateChecksumPayload.\nfunc (v *MutableStateChecksumPayload) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CancelRequested != nil {\n\t\tenc.AddBool(\"cancelRequested\", *v.CancelRequested)\n\t}\n\tif v.State != nil {\n\t\tenc.AddInt16(\"state\", *v.State)\n\t}\n\tif v.CloseStatus != nil {\n\t\tenc.AddInt16(\"closeStatus\", *v.CloseStatus)\n\t}\n\tif v.LastWriteVersion != nil {\n\t\tenc.AddInt64(\"lastWriteVersion\", *v.LastWriteVersion)\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tenc.AddInt64(\"lastWriteEventID\", *v.LastWriteEventID)\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tenc.AddInt64(\"lastFirstEventID\", *v.LastFirstEventID)\n\t}\n\tif v.NextEventID != nil {\n\t\tenc.AddInt64(\"nextEventID\", *v.NextEventID)\n\t}\n\tif v.LastProcessedEventID != nil {\n\t\tenc.AddInt64(\"lastProcessedEventID\", *v.LastProcessedEventID)\n\t}\n\tif v.SignalCount != nil {\n\t\tenc.AddInt64(\"signalCount\", *v.SignalCount)\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tenc.AddInt32(\"decisionAttempt\", *v.DecisionAttempt)\n\t}\n\tif v.DecisionVersion != nil {\n\t\tenc.AddInt64(\"decisionVersion\", *v.DecisionVersion)\n\t}\n\tif v.DecisionScheduledID != nil {\n\t\tenc.AddInt64(\"decisionScheduledID\", *v.DecisionScheduledID)\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tenc.AddInt64(\"decisionStartedID\", *v.DecisionStartedID)\n\t}\n\tif v.PendingTimerStartedIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingTimerStartedIDs\", (_List_I64_Zapper)(v.PendingTimerStartedIDs)))\n\t}\n\tif v.PendingActivityScheduledIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingActivityScheduledIDs\", (_List_I64_Zapper)(v.PendingActivityScheduledIDs)))\n\t}\n\tif v.PendingSignalInitiatedIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingSignalInitiatedIDs\", (_List_I64_Zapper)(v.PendingSignalInitiatedIDs)))\n\t}\n\tif v.PendingReqCancelInitiatedIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingReqCancelInitiatedIDs\", (_List_I64_Zapper)(v.PendingReqCancelInitiatedIDs)))\n\t}\n\tif v.PendingChildInitiatedIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingChildInitiatedIDs\", (_List_I64_Zapper)(v.PendingChildInitiatedIDs)))\n\t}\n\tif v.StickyTaskListName != nil {\n\t\tenc.AddString(\"stickyTaskListName\", *v.StickyTaskListName)\n\t}\n\tif v.VersionHistories != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"VersionHistories\", v.VersionHistories))\n\t}\n\treturn err\n}\n\n// GetCancelRequested returns the value of CancelRequested if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetCancelRequested() (o bool) {\n\tif v != nil && v.CancelRequested != nil {\n\t\treturn *v.CancelRequested\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequested returns true if CancelRequested is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetCancelRequested() bool {\n\treturn v != nil && v.CancelRequested != nil\n}\n\n// GetState returns the value of State if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetState() (o int16) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\n\treturn\n}\n\n// IsSetState returns true if State is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetState() bool {\n\treturn v != nil && v.State != nil\n}\n\n// GetCloseStatus returns the value of CloseStatus if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetCloseStatus() (o int16) {\n\tif v != nil && v.CloseStatus != nil {\n\t\treturn *v.CloseStatus\n\t}\n\n\treturn\n}\n\n// IsSetCloseStatus returns true if CloseStatus is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetCloseStatus() bool {\n\treturn v != nil && v.CloseStatus != nil\n}\n\n// GetLastWriteVersion returns the value of LastWriteVersion if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetLastWriteVersion() (o int64) {\n\tif v != nil && v.LastWriteVersion != nil {\n\t\treturn *v.LastWriteVersion\n\t}\n\n\treturn\n}\n\n// IsSetLastWriteVersion returns true if LastWriteVersion is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetLastWriteVersion() bool {\n\treturn v != nil && v.LastWriteVersion != nil\n}\n\n// GetLastWriteEventID returns the value of LastWriteEventID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetLastWriteEventID() (o int64) {\n\tif v != nil && v.LastWriteEventID != nil {\n\t\treturn *v.LastWriteEventID\n\t}\n\n\treturn\n}\n\n// IsSetLastWriteEventID returns true if LastWriteEventID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetLastWriteEventID() bool {\n\treturn v != nil && v.LastWriteEventID != nil\n}\n\n// GetLastFirstEventID returns the value of LastFirstEventID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetLastFirstEventID() (o int64) {\n\tif v != nil && v.LastFirstEventID != nil {\n\t\treturn *v.LastFirstEventID\n\t}\n\n\treturn\n}\n\n// IsSetLastFirstEventID returns true if LastFirstEventID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetLastFirstEventID() bool {\n\treturn v != nil && v.LastFirstEventID != nil\n}\n\n// GetNextEventID returns the value of NextEventID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetNextEventID() (o int64) {\n\tif v != nil && v.NextEventID != nil {\n\t\treturn *v.NextEventID\n\t}\n\n\treturn\n}\n\n// IsSetNextEventID returns true if NextEventID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetNextEventID() bool {\n\treturn v != nil && v.NextEventID != nil\n}\n\n// GetLastProcessedEventID returns the value of LastProcessedEventID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetLastProcessedEventID() (o int64) {\n\tif v != nil && v.LastProcessedEventID != nil {\n\t\treturn *v.LastProcessedEventID\n\t}\n\n\treturn\n}\n\n// IsSetLastProcessedEventID returns true if LastProcessedEventID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetLastProcessedEventID() bool {\n\treturn v != nil && v.LastProcessedEventID != nil\n}\n\n// GetSignalCount returns the value of SignalCount if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetSignalCount() (o int64) {\n\tif v != nil && v.SignalCount != nil {\n\t\treturn *v.SignalCount\n\t}\n\n\treturn\n}\n\n// IsSetSignalCount returns true if SignalCount is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetSignalCount() bool {\n\treturn v != nil && v.SignalCount != nil\n}\n\n// GetDecisionAttempt returns the value of DecisionAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetDecisionAttempt() (o int32) {\n\tif v != nil && v.DecisionAttempt != nil {\n\t\treturn *v.DecisionAttempt\n\t}\n\n\treturn\n}\n\n// IsSetDecisionAttempt returns true if DecisionAttempt is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetDecisionAttempt() bool {\n\treturn v != nil && v.DecisionAttempt != nil\n}\n\n// GetDecisionVersion returns the value of DecisionVersion if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetDecisionVersion() (o int64) {\n\tif v != nil && v.DecisionVersion != nil {\n\t\treturn *v.DecisionVersion\n\t}\n\n\treturn\n}\n\n// IsSetDecisionVersion returns true if DecisionVersion is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetDecisionVersion() bool {\n\treturn v != nil && v.DecisionVersion != nil\n}\n\n// GetDecisionScheduledID returns the value of DecisionScheduledID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetDecisionScheduledID() (o int64) {\n\tif v != nil && v.DecisionScheduledID != nil {\n\t\treturn *v.DecisionScheduledID\n\t}\n\n\treturn\n}\n\n// IsSetDecisionScheduledID returns true if DecisionScheduledID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetDecisionScheduledID() bool {\n\treturn v != nil && v.DecisionScheduledID != nil\n}\n\n// GetDecisionStartedID returns the value of DecisionStartedID if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetDecisionStartedID() (o int64) {\n\tif v != nil && v.DecisionStartedID != nil {\n\t\treturn *v.DecisionStartedID\n\t}\n\n\treturn\n}\n\n// IsSetDecisionStartedID returns true if DecisionStartedID is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetDecisionStartedID() bool {\n\treturn v != nil && v.DecisionStartedID != nil\n}\n\n// GetPendingTimerStartedIDs returns the value of PendingTimerStartedIDs if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetPendingTimerStartedIDs() (o []int64) {\n\tif v != nil && v.PendingTimerStartedIDs != nil {\n\t\treturn v.PendingTimerStartedIDs\n\t}\n\n\treturn\n}\n\n// IsSetPendingTimerStartedIDs returns true if PendingTimerStartedIDs is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetPendingTimerStartedIDs() bool {\n\treturn v != nil && v.PendingTimerStartedIDs != nil\n}\n\n// GetPendingActivityScheduledIDs returns the value of PendingActivityScheduledIDs if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetPendingActivityScheduledIDs() (o []int64) {\n\tif v != nil && v.PendingActivityScheduledIDs != nil {\n\t\treturn v.PendingActivityScheduledIDs\n\t}\n\n\treturn\n}\n\n// IsSetPendingActivityScheduledIDs returns true if PendingActivityScheduledIDs is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetPendingActivityScheduledIDs() bool {\n\treturn v != nil && v.PendingActivityScheduledIDs != nil\n}\n\n// GetPendingSignalInitiatedIDs returns the value of PendingSignalInitiatedIDs if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetPendingSignalInitiatedIDs() (o []int64) {\n\tif v != nil && v.PendingSignalInitiatedIDs != nil {\n\t\treturn v.PendingSignalInitiatedIDs\n\t}\n\n\treturn\n}\n\n// IsSetPendingSignalInitiatedIDs returns true if PendingSignalInitiatedIDs is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetPendingSignalInitiatedIDs() bool {\n\treturn v != nil && v.PendingSignalInitiatedIDs != nil\n}\n\n// GetPendingReqCancelInitiatedIDs returns the value of PendingReqCancelInitiatedIDs if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetPendingReqCancelInitiatedIDs() (o []int64) {\n\tif v != nil && v.PendingReqCancelInitiatedIDs != nil {\n\t\treturn v.PendingReqCancelInitiatedIDs\n\t}\n\n\treturn\n}\n\n// IsSetPendingReqCancelInitiatedIDs returns true if PendingReqCancelInitiatedIDs is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetPendingReqCancelInitiatedIDs() bool {\n\treturn v != nil && v.PendingReqCancelInitiatedIDs != nil\n}\n\n// GetPendingChildInitiatedIDs returns the value of PendingChildInitiatedIDs if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetPendingChildInitiatedIDs() (o []int64) {\n\tif v != nil && v.PendingChildInitiatedIDs != nil {\n\t\treturn v.PendingChildInitiatedIDs\n\t}\n\n\treturn\n}\n\n// IsSetPendingChildInitiatedIDs returns true if PendingChildInitiatedIDs is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetPendingChildInitiatedIDs() bool {\n\treturn v != nil && v.PendingChildInitiatedIDs != nil\n}\n\n// GetStickyTaskListName returns the value of StickyTaskListName if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetStickyTaskListName() (o string) {\n\tif v != nil && v.StickyTaskListName != nil {\n\t\treturn *v.StickyTaskListName\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskListName returns true if StickyTaskListName is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetStickyTaskListName() bool {\n\treturn v != nil && v.StickyTaskListName != nil\n}\n\n// GetVersionHistories returns the value of VersionHistories if it is set or its\n// zero value if it is unset.\nfunc (v *MutableStateChecksumPayload) GetVersionHistories() (o *shared.VersionHistories) {\n\tif v != nil && v.VersionHistories != nil {\n\t\treturn v.VersionHistories\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistories returns true if VersionHistories is not nil.\nfunc (v *MutableStateChecksumPayload) IsSetVersionHistories() bool {\n\treturn v != nil && v.VersionHistories != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"checksum\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/checksum\",\n\tFilePath: \"checksum.thrift\",\n\tSHA1:     \"c3ee77b53c2e06c35a3296cfeeeadf140711ed95\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2019 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\ninclude \\\"shared.thrift\\\"\\n\\nnamespace java com.uber.cadence\\n\\nstruct MutableStateChecksumPayload {\\n    10: optional bool cancelRequested\\n    15: optional i16 state\\n    16: optional i16 closeStatus\\n\\n    21: optional i64 (js.type = \\\"Long\\\") lastWriteVersion\\n    22: optional i64 (js.type = \\\"Long\\\") lastWriteEventID\\n    23: optional i64 (js.type = \\\"Long\\\") lastFirstEventID\\n    24: optional i64 (js.type = \\\"Long\\\") nextEventID\\n    25: optional i64 (js.type = \\\"Long\\\") lastProcessedEventID\\n    26: optional i64 (js.type = \\\"Long\\\") signalCount\\n\\n    35: optional i32 decisionAttempt\\n    36: optional i64 (js.type = \\\"Long\\\") decisionVersion\\n    37: optional i64 (js.type = \\\"Long\\\") decisionScheduledID\\n    38: optional i64 (js.type = \\\"Long\\\") decisionStartedID\\n\\n    45: optional list<i64> pendingTimerStartedIDs\\n    46: optional list<i64> pendingActivityScheduledIDs\\n    47: optional list<i64> pendingSignalInitiatedIDs\\n    48: optional list<i64> pendingReqCancelInitiatedIDs\\n    49: optional list<i64> pendingChildInitiatedIDs\\n\\n    55: optional string stickyTaskListName\\n    56: optional shared.VersionHistories VersionHistories\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/checksum/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage checksum\n"
  },
  {
    "path": ".gen/go/config/config.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage config\n\nimport (\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype DynamicConfigBlob struct {\n\tSchemaVersion *int64                `json:\"schemaVersion,omitempty\"`\n\tEntries       []*DynamicConfigEntry `json:\"entries,omitempty\"`\n}\n\ntype _List_DynamicConfigEntry_ValueList []*DynamicConfigEntry\n\nfunc (v _List_DynamicConfigEntry_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigEntry', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigEntry_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigEntry_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigEntry_ValueList) Close() {}\n\n// ToWire translates a DynamicConfigBlob struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DynamicConfigBlob) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SchemaVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.SchemaVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Entries != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigEntry_ValueList(v.Entries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigEntry_Read(w wire.Value) (*DynamicConfigEntry, error) {\n\tvar v DynamicConfigEntry\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigEntry_Read(l wire.ValueList) ([]*DynamicConfigEntry, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*DynamicConfigEntry, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigEntry_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DynamicConfigBlob struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DynamicConfigBlob struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DynamicConfigBlob\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DynamicConfigBlob) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.SchemaVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Entries, err = _List_DynamicConfigEntry_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigEntry_Encode(val []*DynamicConfigEntry, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigEntry', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DynamicConfigBlob struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DynamicConfigBlob struct could not be encoded.\nfunc (v *DynamicConfigBlob) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SchemaVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.SchemaVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Entries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigEntry_Encode(v.Entries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigEntry_Decode(sr stream.Reader) (*DynamicConfigEntry, error) {\n\tvar v DynamicConfigEntry\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigEntry_Decode(sr stream.Reader) ([]*DynamicConfigEntry, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*DynamicConfigEntry, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigEntry_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DynamicConfigBlob struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DynamicConfigBlob struct could not be generated from the wire\n// representation.\nfunc (v *DynamicConfigBlob) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.SchemaVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Entries, err = _List_DynamicConfigEntry_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DynamicConfigBlob\n// struct.\nfunc (v *DynamicConfigBlob) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.SchemaVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"SchemaVersion: %v\", *(v.SchemaVersion))\n\t\ti++\n\t}\n\tif v.Entries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Entries: %v\", v.Entries)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DynamicConfigBlob{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_DynamicConfigEntry_Equals(lhs, rhs []*DynamicConfigEntry) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DynamicConfigBlob match the\n// provided DynamicConfigBlob.\n//\n// This function performs a deep comparison.\nfunc (v *DynamicConfigBlob) Equals(rhs *DynamicConfigBlob) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.SchemaVersion, rhs.SchemaVersion) {\n\t\treturn false\n\t}\n\tif !((v.Entries == nil && rhs.Entries == nil) || (v.Entries != nil && rhs.Entries != nil && _List_DynamicConfigEntry_Equals(v.Entries, rhs.Entries))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigEntry_Zapper []*DynamicConfigEntry\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigEntry_Zapper.\nfunc (l _List_DynamicConfigEntry_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DynamicConfigBlob.\nfunc (v *DynamicConfigBlob) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SchemaVersion != nil {\n\t\tenc.AddInt64(\"schemaVersion\", *v.SchemaVersion)\n\t}\n\tif v.Entries != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"entries\", (_List_DynamicConfigEntry_Zapper)(v.Entries)))\n\t}\n\treturn err\n}\n\n// GetSchemaVersion returns the value of SchemaVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigBlob) GetSchemaVersion() (o int64) {\n\tif v != nil && v.SchemaVersion != nil {\n\t\treturn *v.SchemaVersion\n\t}\n\n\treturn\n}\n\n// IsSetSchemaVersion returns true if SchemaVersion is not nil.\nfunc (v *DynamicConfigBlob) IsSetSchemaVersion() bool {\n\treturn v != nil && v.SchemaVersion != nil\n}\n\n// GetEntries returns the value of Entries if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigBlob) GetEntries() (o []*DynamicConfigEntry) {\n\tif v != nil && v.Entries != nil {\n\t\treturn v.Entries\n\t}\n\n\treturn\n}\n\n// IsSetEntries returns true if Entries is not nil.\nfunc (v *DynamicConfigBlob) IsSetEntries() bool {\n\treturn v != nil && v.Entries != nil\n}\n\ntype DynamicConfigEntry struct {\n\tName   *string               `json:\"name,omitempty\"`\n\tValues []*DynamicConfigValue `json:\"values,omitempty\"`\n}\n\ntype _List_DynamicConfigValue_ValueList []*DynamicConfigValue\n\nfunc (v _List_DynamicConfigValue_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigValue', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigValue_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigValue_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigValue_ValueList) Close() {}\n\n// ToWire translates a DynamicConfigEntry struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DynamicConfigEntry) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Values != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigValue_ValueList(v.Values)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigValue_Read(w wire.Value) (*DynamicConfigValue, error) {\n\tvar v DynamicConfigValue\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigValue_Read(l wire.ValueList) ([]*DynamicConfigValue, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*DynamicConfigValue, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigValue_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DynamicConfigEntry struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DynamicConfigEntry struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DynamicConfigEntry\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DynamicConfigEntry) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Values, err = _List_DynamicConfigValue_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigValue_Encode(val []*DynamicConfigValue, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigValue', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DynamicConfigEntry struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DynamicConfigEntry struct could not be encoded.\nfunc (v *DynamicConfigEntry) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Values != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigValue_Encode(v.Values, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigValue_Decode(sr stream.Reader) (*DynamicConfigValue, error) {\n\tvar v DynamicConfigValue\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigValue_Decode(sr stream.Reader) ([]*DynamicConfigValue, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*DynamicConfigValue, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigValue_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DynamicConfigEntry struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DynamicConfigEntry struct could not be generated from the wire\n// representation.\nfunc (v *DynamicConfigEntry) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Values, err = _List_DynamicConfigValue_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DynamicConfigEntry\n// struct.\nfunc (v *DynamicConfigEntry) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Values != nil {\n\t\tfields[i] = fmt.Sprintf(\"Values: %v\", v.Values)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DynamicConfigEntry{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_DynamicConfigValue_Equals(lhs, rhs []*DynamicConfigValue) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DynamicConfigEntry match the\n// provided DynamicConfigEntry.\n//\n// This function performs a deep comparison.\nfunc (v *DynamicConfigEntry) Equals(rhs *DynamicConfigEntry) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !((v.Values == nil && rhs.Values == nil) || (v.Values != nil && rhs.Values != nil && _List_DynamicConfigValue_Equals(v.Values, rhs.Values))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigValue_Zapper []*DynamicConfigValue\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigValue_Zapper.\nfunc (l _List_DynamicConfigValue_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DynamicConfigEntry.\nfunc (v *DynamicConfigEntry) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Values != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"values\", (_List_DynamicConfigValue_Zapper)(v.Values)))\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigEntry) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DynamicConfigEntry) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetValues returns the value of Values if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigEntry) GetValues() (o []*DynamicConfigValue) {\n\tif v != nil && v.Values != nil {\n\t\treturn v.Values\n\t}\n\n\treturn\n}\n\n// IsSetValues returns true if Values is not nil.\nfunc (v *DynamicConfigEntry) IsSetValues() bool {\n\treturn v != nil && v.Values != nil\n}\n\ntype DynamicConfigFilter struct {\n\tName  *string          `json:\"name,omitempty\"`\n\tValue *shared.DataBlob `json:\"value,omitempty\"`\n}\n\n// ToWire translates a DynamicConfigFilter struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DynamicConfigFilter) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tw, err = v.Value.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DataBlob_Read(w wire.Value) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DynamicConfigFilter struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DynamicConfigFilter struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DynamicConfigFilter\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DynamicConfigFilter) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Value, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DynamicConfigFilter struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DynamicConfigFilter struct could not be encoded.\nfunc (v *DynamicConfigFilter) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Value != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Value.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DataBlob_Decode(sr stream.Reader) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DynamicConfigFilter struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DynamicConfigFilter struct could not be generated from the wire\n// representation.\nfunc (v *DynamicConfigFilter) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Value, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DynamicConfigFilter\n// struct.\nfunc (v *DynamicConfigFilter) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tfields[i] = fmt.Sprintf(\"Value: %v\", v.Value)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DynamicConfigFilter{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DynamicConfigFilter match the\n// provided DynamicConfigFilter.\n//\n// This function performs a deep comparison.\nfunc (v *DynamicConfigFilter) Equals(rhs *DynamicConfigFilter) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !((v.Value == nil && rhs.Value == nil) || (v.Value != nil && rhs.Value != nil && v.Value.Equals(rhs.Value))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DynamicConfigFilter.\nfunc (v *DynamicConfigFilter) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Value != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigFilter) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DynamicConfigFilter) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetValue returns the value of Value if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigFilter) GetValue() (o *shared.DataBlob) {\n\tif v != nil && v.Value != nil {\n\t\treturn v.Value\n\t}\n\n\treturn\n}\n\n// IsSetValue returns true if Value is not nil.\nfunc (v *DynamicConfigFilter) IsSetValue() bool {\n\treturn v != nil && v.Value != nil\n}\n\ntype DynamicConfigValue struct {\n\tValue   *shared.DataBlob       `json:\"value,omitempty\"`\n\tFilters []*DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\ntype _List_DynamicConfigFilter_ValueList []*DynamicConfigFilter\n\nfunc (v _List_DynamicConfigFilter_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigFilter', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DynamicConfigFilter_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DynamicConfigFilter_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DynamicConfigFilter_ValueList) Close() {}\n\n// ToWire translates a DynamicConfigValue struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DynamicConfigValue) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Value != nil {\n\t\tw, err = v.Value.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tw, err = wire.NewValueList(_List_DynamicConfigFilter_ValueList(v.Filters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DynamicConfigFilter_Read(w wire.Value) (*DynamicConfigFilter, error) {\n\tvar v DynamicConfigFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigFilter_Read(l wire.ValueList) ([]*DynamicConfigFilter, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*DynamicConfigFilter, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DynamicConfigFilter_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DynamicConfigValue struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DynamicConfigValue struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DynamicConfigValue\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DynamicConfigValue) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Value, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Filters, err = _List_DynamicConfigFilter_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DynamicConfigFilter_Encode(val []*DynamicConfigFilter, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DynamicConfigFilter', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DynamicConfigValue struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DynamicConfigValue struct could not be encoded.\nfunc (v *DynamicConfigValue) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Value != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Value.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Filters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DynamicConfigFilter_Encode(v.Filters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DynamicConfigFilter_Decode(sr stream.Reader) (*DynamicConfigFilter, error) {\n\tvar v DynamicConfigFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DynamicConfigFilter_Decode(sr stream.Reader) ([]*DynamicConfigFilter, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*DynamicConfigFilter, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DynamicConfigFilter_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DynamicConfigValue struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DynamicConfigValue struct could not be generated from the wire\n// representation.\nfunc (v *DynamicConfigValue) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Value, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Filters, err = _List_DynamicConfigFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DynamicConfigValue\n// struct.\nfunc (v *DynamicConfigValue) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Value != nil {\n\t\tfields[i] = fmt.Sprintf(\"Value: %v\", v.Value)\n\t\ti++\n\t}\n\tif v.Filters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Filters: %v\", v.Filters)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DynamicConfigValue{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DynamicConfigFilter_Equals(lhs, rhs []*DynamicConfigFilter) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DynamicConfigValue match the\n// provided DynamicConfigValue.\n//\n// This function performs a deep comparison.\nfunc (v *DynamicConfigValue) Equals(rhs *DynamicConfigValue) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Value == nil && rhs.Value == nil) || (v.Value != nil && rhs.Value != nil && v.Value.Equals(rhs.Value))) {\n\t\treturn false\n\t}\n\tif !((v.Filters == nil && rhs.Filters == nil) || (v.Filters != nil && rhs.Filters != nil && _List_DynamicConfigFilter_Equals(v.Filters, rhs.Filters))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DynamicConfigFilter_Zapper []*DynamicConfigFilter\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DynamicConfigFilter_Zapper.\nfunc (l _List_DynamicConfigFilter_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DynamicConfigValue.\nfunc (v *DynamicConfigValue) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Value != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\t}\n\tif v.Filters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"filters\", (_List_DynamicConfigFilter_Zapper)(v.Filters)))\n\t}\n\treturn err\n}\n\n// GetValue returns the value of Value if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigValue) GetValue() (o *shared.DataBlob) {\n\tif v != nil && v.Value != nil {\n\t\treturn v.Value\n\t}\n\n\treturn\n}\n\n// IsSetValue returns true if Value is not nil.\nfunc (v *DynamicConfigValue) IsSetValue() bool {\n\treturn v != nil && v.Value != nil\n}\n\n// GetFilters returns the value of Filters if it is set or its\n// zero value if it is unset.\nfunc (v *DynamicConfigValue) GetFilters() (o []*DynamicConfigFilter) {\n\tif v != nil && v.Filters != nil {\n\t\treturn v.Filters\n\t}\n\n\treturn\n}\n\n// IsSetFilters returns true if Filters is not nil.\nfunc (v *DynamicConfigValue) IsSetFilters() bool {\n\treturn v != nil && v.Filters != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"config\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/config\",\n\tFilePath: \"config.thrift\",\n\tSHA1:     \"cbc9d97e2a2f4820452fc2d273d5102e4721cf34\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.config\\n\\ninclude \\\"shared.thrift\\\"\\n\\nstruct DynamicConfigBlob {\\n\\t10: optional i64 schemaVersion\\n\\t20: optional list<DynamicConfigEntry> entries\\n}\\n\\nstruct DynamicConfigEntry {\\n  10: optional string name\\n  20: optional list<DynamicConfigValue> values\\n}\\n\\nstruct DynamicConfigValue {\\n  10: optional shared.DataBlob value\\n  20: optional list<DynamicConfigFilter> filters\\n}\\n\\nstruct DynamicConfigFilter {\\n  10: optional string name\\n  20: optional shared.DataBlob value\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/config/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage config\n"
  },
  {
    "path": ".gen/go/health/health.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage health\n\nimport (\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n)\n\ntype HealthStatus struct {\n\tOk  bool    `json:\"ok,required\"`\n\tMsg *string `json:\"msg,omitempty\"`\n}\n\n// ToWire translates a HealthStatus struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HealthStatus) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueBool(v.Ok), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\tif v.Msg != nil {\n\t\tw, err = wire.NewValueString(*(v.Msg)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HealthStatus struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HealthStatus struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HealthStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HealthStatus) FromWire(w wire.Value) error {\n\tvar err error\n\n\tokIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tv.Ok, err = field.Value.GetBool(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tokIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Msg = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tif !okIsSet {\n\t\treturn errors.New(\"field Ok of HealthStatus is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HealthStatus struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HealthStatus struct could not be encoded.\nfunc (v *HealthStatus) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBool}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteBool(v.Ok); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Msg != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Msg)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HealthStatus struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HealthStatus struct could not be generated from the wire\n// representation.\nfunc (v *HealthStatus) Decode(sr stream.Reader) error {\n\n\tokIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBool:\n\t\t\tv.Ok, err = sr.ReadBool()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tokIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Msg = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !okIsSet {\n\t\treturn errors.New(\"field Ok of HealthStatus is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HealthStatus\n// struct.\nfunc (v *HealthStatus) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Ok: %v\", v.Ok)\n\ti++\n\tif v.Msg != nil {\n\t\tfields[i] = fmt.Sprintf(\"Msg: %v\", *(v.Msg))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HealthStatus{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this HealthStatus match the\n// provided HealthStatus.\n//\n// This function performs a deep comparison.\nfunc (v *HealthStatus) Equals(rhs *HealthStatus) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Ok == rhs.Ok) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Msg, rhs.Msg) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HealthStatus.\nfunc (v *HealthStatus) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddBool(\"ok\", v.Ok)\n\tif v.Msg != nil {\n\t\tenc.AddString(\"msg\", *v.Msg)\n\t}\n\treturn err\n}\n\n// GetOk returns the value of Ok if it is set or its\n// zero value if it is unset.\nfunc (v *HealthStatus) GetOk() (o bool) {\n\tif v != nil {\n\t\to = v.Ok\n\t}\n\treturn\n}\n\n// GetMsg returns the value of Msg if it is set or its\n// zero value if it is unset.\nfunc (v *HealthStatus) GetMsg() (o string) {\n\tif v != nil && v.Msg != nil {\n\t\treturn *v.Msg\n\t}\n\n\treturn\n}\n\n// IsSetMsg returns true if Msg is not nil.\nfunc (v *HealthStatus) IsSetMsg() bool {\n\treturn v != nil && v.Msg != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"health\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/health\",\n\tFilePath: \"health.thrift\",\n\tSHA1:     \"8d52f05c157e47bef27c86d2133e1cdb475f8024\",\n\tRaw:      rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence\\n\\n/* ==================== Health Check ==================== */\\n\\nstruct HealthStatus {\\n    1: required bool ok\\n    2: optional string msg\\n}\\n\\nservice Meta {\\n    HealthStatus health()\\n}\\n\\n\"\n\n// Meta_Health_Args represents the arguments for the Meta.health function.\n//\n// The arguments for health are sent and received over the wire as this struct.\ntype Meta_Health_Args struct {\n}\n\n// ToWire translates a Meta_Health_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Meta_Health_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a Meta_Health_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Meta_Health_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Meta_Health_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Meta_Health_Args) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Meta_Health_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Meta_Health_Args struct could not be encoded.\nfunc (v *Meta_Health_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a Meta_Health_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Meta_Health_Args struct could not be generated from the wire\n// representation.\nfunc (v *Meta_Health_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Meta_Health_Args\n// struct.\nfunc (v *Meta_Health_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"Meta_Health_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this Meta_Health_Args match the\n// provided Meta_Health_Args.\n//\n// This function performs a deep comparison.\nfunc (v *Meta_Health_Args) Equals(rhs *Meta_Health_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Meta_Health_Args.\nfunc (v *Meta_Health_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"health\" for this struct.\nfunc (v *Meta_Health_Args) MethodName() string {\n\treturn \"health\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *Meta_Health_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// Meta_Health_Helper provides functions that aid in handling the\n// parameters and return values of the Meta.health\n// function.\nvar Meta_Health_Helper = struct {\n\t// Args accepts the parameters of health in-order and returns\n\t// the arguments struct for the function.\n\tArgs func() *Meta_Health_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by health.\n\t//\n\t// An error can be thrown by health only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for health\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// health into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by health\n\t//\n\t//   value, err := health(args)\n\t//   result, err := Meta_Health_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from health: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*HealthStatus, error) (*Meta_Health_Result, error)\n\n\t// UnwrapResponse takes the result struct for health\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if health threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := Meta_Health_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*Meta_Health_Result) (*HealthStatus, error)\n}{}\n\nfunc init() {\n\tMeta_Health_Helper.Args = func() *Meta_Health_Args {\n\t\treturn &Meta_Health_Args{}\n\t}\n\n\tMeta_Health_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMeta_Health_Helper.WrapResponse = func(success *HealthStatus, err error) (*Meta_Health_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &Meta_Health_Result{Success: success}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMeta_Health_Helper.UnwrapResponse = func(result *Meta_Health_Result) (success *HealthStatus, err error) {\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// Meta_Health_Result represents the result of a Meta.health function call.\n//\n// The result of a health execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype Meta_Health_Result struct {\n\t// Value returned by health after a successful execution.\n\tSuccess *HealthStatus `json:\"success,omitempty\"`\n}\n\n// ToWire translates a Meta_Health_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Meta_Health_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"Meta_Health_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HealthStatus_Read(w wire.Value) (*HealthStatus, error) {\n\tvar v HealthStatus\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a Meta_Health_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Meta_Health_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Meta_Health_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Meta_Health_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _HealthStatus_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"Meta_Health_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Meta_Health_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Meta_Health_Result struct could not be encoded.\nfunc (v *Meta_Health_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"Meta_Health_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HealthStatus_Decode(sr stream.Reader) (*HealthStatus, error) {\n\tvar v HealthStatus\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a Meta_Health_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Meta_Health_Result struct could not be generated from the wire\n// representation.\nfunc (v *Meta_Health_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _HealthStatus_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"Meta_Health_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Meta_Health_Result\n// struct.\nfunc (v *Meta_Health_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Meta_Health_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this Meta_Health_Result match the\n// provided Meta_Health_Result.\n//\n// This function performs a deep comparison.\nfunc (v *Meta_Health_Result) Equals(rhs *Meta_Health_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Meta_Health_Result.\nfunc (v *Meta_Health_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *Meta_Health_Result) GetSuccess() (o *HealthStatus) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *Meta_Health_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"health\" for this struct.\nfunc (v *Meta_Health_Result) MethodName() string {\n\treturn \"health\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *Meta_Health_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n"
  },
  {
    "path": ".gen/go/health/metaclient/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage metaclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\twire \"go.uber.org/thriftrw/wire\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\thealth \"github.com/uber/cadence/.gen/go/health\"\n)\n\n// Interface is a client for the Meta service.\ntype Interface interface {\n\tHealth(\n\t\tctx context.Context,\n\t\topts ...yarpc.CallOption,\n\t) (*health.HealthStatus, error)\n}\n\n// New builds a new client for the Meta service.\n//\n//\tclient := metaclient.New(dispatcher.ClientConfig(\"meta\"))\nfunc New(c transport.ClientConfig, opts ...thrift.ClientOption) Interface {\n\treturn client{\n\t\tc: thrift.New(thrift.Config{\n\t\t\tService:      \"Meta\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t\tnwc: thrift.NewNoWire(thrift.Config{\n\t\t\tService:      \"Meta\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t}\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(c transport.ClientConfig, f reflect.StructField) Interface {\n\t\t\treturn New(c, thrift.ClientBuilderOptions(c, f)...)\n\t\t},\n\t)\n}\n\ntype client struct {\n\tc   thrift.Client\n\tnwc thrift.NoWireClient\n}\n\nfunc (c client) Health(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *health.HealthStatus, err error) {\n\n\tvar result health.Meta_Health_Result\n\targs := health.Meta_Health_Helper.Args()\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = health.Meta_Health_Helper.UnwrapResponse(&result)\n\treturn\n}\n"
  },
  {
    "path": ".gen/go/health/metafx/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage metafx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\trestriction \"go.uber.org/yarpc/api/x/restriction\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tmetaclient \"github.com/uber/cadence/.gen/go/health/metaclient\"\n)\n\n// Params defines the dependencies for the Meta client.\ntype Params struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// Result defines the output of the Meta client module. It provides a\n// Meta client to an Fx application.\ntype Result struct {\n\tfx.Out\n\n\tClient metaclient.Interface\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// Client provides a Meta client to an Fx application using the given name\n// for routing.\n//\n//\tfx.Provide(\n//\t\tmetafx.Client(\"...\"),\n//\t\tnewHandler,\n//\t)\nfunc Client(name string, opts ...thrift.ClientOption) interface{} {\n\treturn func(p Params) Result {\n\t\tcc := p.Provider.ClientConfig(name)\n\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok && p.Restriction != nil {\n\t\t\tif err := p.Restriction.Check(thrift.Encoding, namer.TransportName()); err != nil {\n\t\t\t\tpanic(err.Error())\n\t\t\t}\n\t\t}\n\t\tclient := metaclient.New(cc, opts...)\n\t\treturn Result{Client: client}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/health/metafx/doc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\n// Package metafx provides better integration for Fx for services\n// implementing or calling Meta.\n//\n// # Clients\n//\n// If you are making requests to Meta, use the Client function to inject a\n// Meta client into your container.\n//\n//\tfx.Provide(metafx.Client(\"...\"))\n//\n// # Servers\n//\n// If you are implementing Meta, provide a metaserver.Interface into\n// the container and use the Server function.\n//\n// Given,\n//\n//\tfunc NewMetaHandler() metaserver.Interface\n//\n// You can do the following to have the procedures of Meta made available\n// to an Fx application.\n//\n//\tfx.Provide(\n//\t\tNewMetaHandler,\n//\t\tmetafx.Server(),\n//\t)\npackage metafx\n"
  },
  {
    "path": ".gen/go/health/metafx/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage metafx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tmetaserver \"github.com/uber/cadence/.gen/go/health/metaserver\"\n)\n\n// ServerParams defines the dependencies for the Meta server.\ntype ServerParams struct {\n\tfx.In\n\n\tHandler metaserver.Interface\n}\n\n// ServerResult defines the output of Meta server module. It provides the\n// procedures of a Meta handler to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group. Dig 1.2 or newer\n// must be used for this feature to work.\ntype ServerResult struct {\n\tfx.Out\n\n\tProcedures []transport.Procedure `group:\"yarpcfx\"`\n}\n\n// Server provides procedures for Meta to an Fx application. It expects a\n// metafx.Interface to be present in the container.\n//\n//\tfx.Provide(\n//\t\tfunc(h *MyMetaHandler) metaserver.Interface {\n//\t\t\treturn h\n//\t\t},\n//\t\tmetafx.Server(),\n//\t)\nfunc Server(opts ...thrift.RegisterOption) interface{} {\n\treturn func(p ServerParams) ServerResult {\n\t\tprocedures := metaserver.New(p.Handler, opts...)\n\t\treturn ServerResult{Procedures: procedures}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/health/metaserver/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage metaserver\n\nimport (\n\tcontext \"context\"\n\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\tyarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n\thealth \"github.com/uber/cadence/.gen/go/health\"\n)\n\n// Interface is the server-side interface for the Meta service.\ntype Interface interface {\n\tHealth(\n\t\tctx context.Context,\n\t) (*health.HealthStatus, error)\n}\n\n// New prepares an implementation of the Meta service for\n// registration.\n//\n//\thandler := MetaHandler{}\n//\tdispatcher.Register(metaserver.New(handler))\nfunc New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {\n\th := handler{impl}\n\tservice := thrift.Service{\n\t\tName: \"Meta\",\n\t\tMethods: []thrift.Method{\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"health\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.Health),\n\t\t\t\t\tNoWire: health_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"Health() (*health.HealthStatus)\",\n\t\t\t\tThriftModule: health.ThriftModule,\n\t\t\t},\n\t\t},\n\t}\n\n\tprocedures := make([]transport.Procedure, 0, 1)\n\tprocedures = append(procedures, thrift.BuildProcedures(service, opts...)...)\n\treturn procedures\n}\n\ntype handler struct{ impl Interface }\n\ntype yarpcErrorNamer interface{ YARPCErrorName() string }\n\ntype yarpcErrorCoder interface{ YARPCErrorCode() *yarpcerrors.Code }\n\nfunc (h handler) Health(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args health.Meta_Health_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'Meta' procedure 'Health': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.Health(ctx)\n\n\thadError := appErr != nil\n\tresult, err := health.Meta_Health_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\ntype health_NoWireHandler struct{ impl Interface }\n\nfunc (h health_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs health.Meta_Health_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'Meta' procedure 'Health': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.Health(ctx)\n\n\thadError := appErr != nil\n\tresult, err := health.Meta_Health_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n"
  },
  {
    "path": ".gen/go/health/metatest/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage metatest\n\nimport (\n\tcontext \"context\"\n\n\tgomock \"github.com/golang/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\thealth \"github.com/uber/cadence/.gen/go/health\"\n\tmetaclient \"github.com/uber/cadence/.gen/go/health/metaclient\"\n)\n\n// MockClient implements a gomock-compatible mock client for service\n// Meta.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *_MockClientRecorder\n}\n\nvar _ metaclient.Interface = (*MockClient)(nil)\n\ntype _MockClientRecorder struct {\n\tmock *MockClient\n}\n\n// Build a new mock client for service Meta.\n//\n//\tmockCtrl := gomock.NewController(t)\n//\tclient := metatest.NewMockClient(mockCtrl)\n//\n// Use EXPECT() to set expectations on the mock.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &_MockClientRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows you to define an expectation on the\n// Meta mock client.\nfunc (m *MockClient) EXPECT() *_MockClientRecorder {\n\treturn m.recorder\n}\n\n// Health responds to a Health call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().Health(gomock.Any(), ...).Return(...)\n//\t... := client.Health(...)\nfunc (m *MockClient) Health(\n\tctx context.Context,\n\topts ...yarpc.CallOption,\n) (success *health.HealthStatus, err error) {\n\n\targs := []interface{}{ctx}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"Health\", args...)\n\tsuccess, _ = ret[i].(*health.HealthStatus)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) Health(\n\tctx interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"Health\", args...)\n}\n"
  },
  {
    "path": ".gen/go/health/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage health\n"
  },
  {
    "path": ".gen/go/history/history.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage history\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Any{ValueType} identifier for WeightedRatelimitQuotas data\nconst WeightedRatelimitQuotasAnyType string = \"cadence:loadbalanced:update_response\"\n\n// Any{ValueType} identifier for WeightedRatelimitUsage data\nconst WeightedRatelimitUsageAnyType string = \"cadence:loadbalanced:update_request\"\n\nconst WeightedRatelimitUsageQuotasAnyType string = \"cadence:loadbalanced:update_response_used\"\n\ntype DescribeMutableStateRequest struct {\n\tDomainUUID *string                   `json:\"domainUUID,omitempty\"`\n\tExecution  *shared.WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a DescribeMutableStateRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeMutableStateRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecution_Read(w wire.Value) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeMutableStateRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeMutableStateRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeMutableStateRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeMutableStateRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeMutableStateRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeMutableStateRequest struct could not be encoded.\nfunc (v *DescribeMutableStateRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecution_Decode(sr stream.Reader) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeMutableStateRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeMutableStateRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeMutableStateRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeMutableStateRequest\n// struct.\nfunc (v *DescribeMutableStateRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeMutableStateRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DescribeMutableStateRequest match the\n// provided DescribeMutableStateRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeMutableStateRequest) Equals(rhs *DescribeMutableStateRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeMutableStateRequest.\nfunc (v *DescribeMutableStateRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *DescribeMutableStateRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeMutableStateRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *DescribeMutableStateRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype DescribeMutableStateResponse struct {\n\tMutableStateInCache    *string `json:\"mutableStateInCache,omitempty\"`\n\tMutableStateInDatabase *string `json:\"mutableStateInDatabase,omitempty\"`\n}\n\n// ToWire translates a DescribeMutableStateResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeMutableStateResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MutableStateInCache != nil {\n\t\tw, err = wire.NewValueString(*(v.MutableStateInCache)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tw, err = wire.NewValueString(*(v.MutableStateInDatabase)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeMutableStateResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeMutableStateResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeMutableStateResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeMutableStateResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MutableStateInCache = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MutableStateInDatabase = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeMutableStateResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeMutableStateResponse struct could not be encoded.\nfunc (v *DescribeMutableStateResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MutableStateInCache != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MutableStateInCache)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MutableStateInDatabase != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MutableStateInDatabase)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeMutableStateResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeMutableStateResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeMutableStateResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MutableStateInCache = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MutableStateInDatabase = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeMutableStateResponse\n// struct.\nfunc (v *DescribeMutableStateResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.MutableStateInCache != nil {\n\t\tfields[i] = fmt.Sprintf(\"MutableStateInCache: %v\", *(v.MutableStateInCache))\n\t\ti++\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tfields[i] = fmt.Sprintf(\"MutableStateInDatabase: %v\", *(v.MutableStateInDatabase))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeMutableStateResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeMutableStateResponse match the\n// provided DescribeMutableStateResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeMutableStateResponse) Equals(rhs *DescribeMutableStateResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MutableStateInCache, rhs.MutableStateInCache) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MutableStateInDatabase, rhs.MutableStateInDatabase) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeMutableStateResponse.\nfunc (v *DescribeMutableStateResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MutableStateInCache != nil {\n\t\tenc.AddString(\"mutableStateInCache\", *v.MutableStateInCache)\n\t}\n\tif v.MutableStateInDatabase != nil {\n\t\tenc.AddString(\"mutableStateInDatabase\", *v.MutableStateInDatabase)\n\t}\n\treturn err\n}\n\n// GetMutableStateInCache returns the value of MutableStateInCache if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeMutableStateResponse) GetMutableStateInCache() (o string) {\n\tif v != nil && v.MutableStateInCache != nil {\n\t\treturn *v.MutableStateInCache\n\t}\n\n\treturn\n}\n\n// IsSetMutableStateInCache returns true if MutableStateInCache is not nil.\nfunc (v *DescribeMutableStateResponse) IsSetMutableStateInCache() bool {\n\treturn v != nil && v.MutableStateInCache != nil\n}\n\n// GetMutableStateInDatabase returns the value of MutableStateInDatabase if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeMutableStateResponse) GetMutableStateInDatabase() (o string) {\n\tif v != nil && v.MutableStateInDatabase != nil {\n\t\treturn *v.MutableStateInDatabase\n\t}\n\n\treturn\n}\n\n// IsSetMutableStateInDatabase returns true if MutableStateInDatabase is not nil.\nfunc (v *DescribeMutableStateResponse) IsSetMutableStateInDatabase() bool {\n\treturn v != nil && v.MutableStateInDatabase != nil\n}\n\ntype DescribeWorkflowExecutionRequest struct {\n\tDomainUUID *string                                  `json:\"domainUUID,omitempty\"`\n\tRequest    *shared.DescribeWorkflowExecutionRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a DescribeWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionRequest_Read(w wire.Value) (*shared.DescribeWorkflowExecutionRequest, error) {\n\tvar v shared.DescribeWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be encoded.\nfunc (v *DescribeWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.DescribeWorkflowExecutionRequest, error) {\n\tvar v shared.DescribeWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeWorkflowExecutionRequest\n// struct.\nfunc (v *DescribeWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeWorkflowExecutionRequest match the\n// provided DescribeWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeWorkflowExecutionRequest) Equals(rhs *DescribeWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeWorkflowExecutionRequest.\nfunc (v *DescribeWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetRequest() (o *shared.DescribeWorkflowExecutionRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype DomainFilter struct {\n\tDomainIDs    []string `json:\"domainIDs,omitempty\"`\n\tReverseMatch *bool    `json:\"reverseMatch,omitempty\"`\n}\n\ntype _List_String_ValueList []string\n\nfunc (v _List_String_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueString(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_String_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_String_ValueList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_List_String_ValueList) Close() {}\n\n// ToWire translates a DomainFilter struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainFilter) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainIDs != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.DomainIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ReverseMatch != nil {\n\t\tw, err = wire.NewValueBool(*(v.ReverseMatch)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_String_Read(l wire.ValueList) ([]string, error) {\n\tif l.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make([]string, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DomainFilter struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainFilter struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainFilter\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainFilter) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.DomainIDs, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ReverseMatch = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_String_Encode(val []string, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TBinary,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DomainFilter struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainFilter struct could not be encoded.\nfunc (v *DomainFilter) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.DomainIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReverseMatch != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ReverseMatch)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_String_Decode(sr stream.Reader) ([]string, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TBinary {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]string, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DomainFilter struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainFilter struct could not be generated from the wire\n// representation.\nfunc (v *DomainFilter) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.DomainIDs, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ReverseMatch = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainFilter\n// struct.\nfunc (v *DomainFilter) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainIDs: %v\", v.DomainIDs)\n\t\ti++\n\t}\n\tif v.ReverseMatch != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReverseMatch: %v\", *(v.ReverseMatch))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainFilter{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_String_Equals(lhs, rhs []string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DomainFilter match the\n// provided DomainFilter.\n//\n// This function performs a deep comparison.\nfunc (v *DomainFilter) Equals(rhs *DomainFilter) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainIDs == nil && rhs.DomainIDs == nil) || (v.DomainIDs != nil && rhs.DomainIDs != nil && _List_String_Equals(v.DomainIDs, rhs.DomainIDs))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ReverseMatch, rhs.ReverseMatch) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_String_Zapper []string\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_String_Zapper.\nfunc (l _List_String_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendString(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainFilter.\nfunc (v *DomainFilter) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"domainIDs\", (_List_String_Zapper)(v.DomainIDs)))\n\t}\n\tif v.ReverseMatch != nil {\n\t\tenc.AddBool(\"reverseMatch\", *v.ReverseMatch)\n\t}\n\treturn err\n}\n\n// GetDomainIDs returns the value of DomainIDs if it is set or its\n// zero value if it is unset.\nfunc (v *DomainFilter) GetDomainIDs() (o []string) {\n\tif v != nil && v.DomainIDs != nil {\n\t\treturn v.DomainIDs\n\t}\n\n\treturn\n}\n\n// IsSetDomainIDs returns true if DomainIDs is not nil.\nfunc (v *DomainFilter) IsSetDomainIDs() bool {\n\treturn v != nil && v.DomainIDs != nil\n}\n\n// GetReverseMatch returns the value of ReverseMatch if it is set or its\n// zero value if it is unset.\nfunc (v *DomainFilter) GetReverseMatch() (o bool) {\n\tif v != nil && v.ReverseMatch != nil {\n\t\treturn *v.ReverseMatch\n\t}\n\n\treturn\n}\n\n// IsSetReverseMatch returns true if ReverseMatch is not nil.\nfunc (v *DomainFilter) IsSetReverseMatch() bool {\n\treturn v != nil && v.ReverseMatch != nil\n}\n\ntype EventAlreadyStartedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a EventAlreadyStartedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *EventAlreadyStartedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a EventAlreadyStartedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a EventAlreadyStartedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v EventAlreadyStartedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *EventAlreadyStartedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of EventAlreadyStartedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a EventAlreadyStartedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a EventAlreadyStartedError struct could not be encoded.\nfunc (v *EventAlreadyStartedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a EventAlreadyStartedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a EventAlreadyStartedError struct could not be generated from the wire\n// representation.\nfunc (v *EventAlreadyStartedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of EventAlreadyStartedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a EventAlreadyStartedError\n// struct.\nfunc (v *EventAlreadyStartedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"EventAlreadyStartedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*EventAlreadyStartedError) ErrorName() string {\n\treturn \"EventAlreadyStartedError\"\n}\n\n// Equals returns true if all the fields of this EventAlreadyStartedError match the\n// provided EventAlreadyStartedError.\n//\n// This function performs a deep comparison.\nfunc (v *EventAlreadyStartedError) Equals(rhs *EventAlreadyStartedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of EventAlreadyStartedError.\nfunc (v *EventAlreadyStartedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *EventAlreadyStartedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *EventAlreadyStartedError) Error() string {\n\treturn v.String()\n}\n\ntype FailoverMarkerToken struct {\n\tShardIDs       []int32                              `json:\"shardIDs,omitempty\"`\n\tFailoverMarker *replicator.FailoverMarkerAttributes `json:\"failoverMarker,omitempty\"`\n}\n\ntype _List_I32_ValueList []int32\n\nfunc (v _List_I32_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueI32(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_I32_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_I32_ValueList) ValueType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_List_I32_ValueList) Close() {}\n\n// ToWire translates a FailoverMarkerToken struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverMarkerToken) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I32_ValueList(v.ShardIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverMarker != nil {\n\t\tw, err = v.FailoverMarker.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_I32_Read(l wire.ValueList) ([]int32, error) {\n\tif l.ValueType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\to := make([]int32, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _FailoverMarkerAttributes_Read(w wire.Value) (*replicator.FailoverMarkerAttributes, error) {\n\tvar v replicator.FailoverMarkerAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a FailoverMarkerToken struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverMarkerToken struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverMarkerToken\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverMarkerToken) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ShardIDs, err = _List_I32_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailoverMarker, err = _FailoverMarkerAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_I32_Encode(val []int32, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TI32,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteInt32(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a FailoverMarkerToken struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverMarkerToken struct could not be encoded.\nfunc (v *FailoverMarkerToken) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I32_Encode(v.ShardIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverMarker != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailoverMarker.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_I32_Decode(sr stream.Reader) ([]int32, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TI32 {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]int32, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _FailoverMarkerAttributes_Decode(sr stream.Reader) (*replicator.FailoverMarkerAttributes, error) {\n\tvar v replicator.FailoverMarkerAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a FailoverMarkerToken struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverMarkerToken struct could not be generated from the wire\n// representation.\nfunc (v *FailoverMarkerToken) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ShardIDs, err = _List_I32_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.FailoverMarker, err = _FailoverMarkerAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverMarkerToken\n// struct.\nfunc (v *FailoverMarkerToken) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ShardIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardIDs: %v\", v.ShardIDs)\n\t\ti++\n\t}\n\tif v.FailoverMarker != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverMarker: %v\", v.FailoverMarker)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverMarkerToken{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_I32_Equals(lhs, rhs []int32) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this FailoverMarkerToken match the\n// provided FailoverMarkerToken.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverMarkerToken) Equals(rhs *FailoverMarkerToken) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ShardIDs == nil && rhs.ShardIDs == nil) || (v.ShardIDs != nil && rhs.ShardIDs != nil && _List_I32_Equals(v.ShardIDs, rhs.ShardIDs))) {\n\t\treturn false\n\t}\n\tif !((v.FailoverMarker == nil && rhs.FailoverMarker == nil) || (v.FailoverMarker != nil && rhs.FailoverMarker != nil && v.FailoverMarker.Equals(rhs.FailoverMarker))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_I32_Zapper []int32\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_I32_Zapper.\nfunc (l _List_I32_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendInt32(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverMarkerToken.\nfunc (v *FailoverMarkerToken) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"shardIDs\", (_List_I32_Zapper)(v.ShardIDs)))\n\t}\n\tif v.FailoverMarker != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failoverMarker\", v.FailoverMarker))\n\t}\n\treturn err\n}\n\n// GetShardIDs returns the value of ShardIDs if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkerToken) GetShardIDs() (o []int32) {\n\tif v != nil && v.ShardIDs != nil {\n\t\treturn v.ShardIDs\n\t}\n\n\treturn\n}\n\n// IsSetShardIDs returns true if ShardIDs is not nil.\nfunc (v *FailoverMarkerToken) IsSetShardIDs() bool {\n\treturn v != nil && v.ShardIDs != nil\n}\n\n// GetFailoverMarker returns the value of FailoverMarker if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkerToken) GetFailoverMarker() (o *replicator.FailoverMarkerAttributes) {\n\tif v != nil && v.FailoverMarker != nil {\n\t\treturn v.FailoverMarker\n\t}\n\n\treturn\n}\n\n// IsSetFailoverMarker returns true if FailoverMarker is not nil.\nfunc (v *FailoverMarkerToken) IsSetFailoverMarker() bool {\n\treturn v != nil && v.FailoverMarker != nil\n}\n\ntype GetFailoverInfoRequest struct {\n\tDomainID *string `json:\"domainID,omitempty\"`\n}\n\n// ToWire translates a GetFailoverInfoRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetFailoverInfoRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetFailoverInfoRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetFailoverInfoRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetFailoverInfoRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetFailoverInfoRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetFailoverInfoRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetFailoverInfoRequest struct could not be encoded.\nfunc (v *GetFailoverInfoRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetFailoverInfoRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetFailoverInfoRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetFailoverInfoRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetFailoverInfoRequest\n// struct.\nfunc (v *GetFailoverInfoRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetFailoverInfoRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetFailoverInfoRequest match the\n// provided GetFailoverInfoRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetFailoverInfoRequest) Equals(rhs *GetFailoverInfoRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetFailoverInfoRequest.\nfunc (v *GetFailoverInfoRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *GetFailoverInfoRequest) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *GetFailoverInfoRequest) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\ntype GetFailoverInfoResponse struct {\n\tCompletedShardCount *int32  `json:\"completedShardCount,omitempty\"`\n\tPendingShards       []int32 `json:\"pendingShards,omitempty\"`\n}\n\n// ToWire translates a GetFailoverInfoResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetFailoverInfoResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompletedShardCount != nil {\n\t\tw, err = wire.NewValueI32(*(v.CompletedShardCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PendingShards != nil {\n\t\tw, err = wire.NewValueList(_List_I32_ValueList(v.PendingShards)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetFailoverInfoResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetFailoverInfoResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetFailoverInfoResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetFailoverInfoResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.CompletedShardCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingShards, err = _List_I32_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetFailoverInfoResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetFailoverInfoResponse struct could not be encoded.\nfunc (v *GetFailoverInfoResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompletedShardCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.CompletedShardCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingShards != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I32_Encode(v.PendingShards, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetFailoverInfoResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetFailoverInfoResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetFailoverInfoResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.CompletedShardCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.PendingShards, err = _List_I32_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetFailoverInfoResponse\n// struct.\nfunc (v *GetFailoverInfoResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.CompletedShardCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletedShardCount: %v\", *(v.CompletedShardCount))\n\t\ti++\n\t}\n\tif v.PendingShards != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingShards: %v\", v.PendingShards)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetFailoverInfoResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this GetFailoverInfoResponse match the\n// provided GetFailoverInfoResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetFailoverInfoResponse) Equals(rhs *GetFailoverInfoResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.CompletedShardCount, rhs.CompletedShardCount) {\n\t\treturn false\n\t}\n\tif !((v.PendingShards == nil && rhs.PendingShards == nil) || (v.PendingShards != nil && rhs.PendingShards != nil && _List_I32_Equals(v.PendingShards, rhs.PendingShards))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetFailoverInfoResponse.\nfunc (v *GetFailoverInfoResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompletedShardCount != nil {\n\t\tenc.AddInt32(\"completedShardCount\", *v.CompletedShardCount)\n\t}\n\tif v.PendingShards != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingShards\", (_List_I32_Zapper)(v.PendingShards)))\n\t}\n\treturn err\n}\n\n// GetCompletedShardCount returns the value of CompletedShardCount if it is set or its\n// zero value if it is unset.\nfunc (v *GetFailoverInfoResponse) GetCompletedShardCount() (o int32) {\n\tif v != nil && v.CompletedShardCount != nil {\n\t\treturn *v.CompletedShardCount\n\t}\n\n\treturn\n}\n\n// IsSetCompletedShardCount returns true if CompletedShardCount is not nil.\nfunc (v *GetFailoverInfoResponse) IsSetCompletedShardCount() bool {\n\treturn v != nil && v.CompletedShardCount != nil\n}\n\n// GetPendingShards returns the value of PendingShards if it is set or its\n// zero value if it is unset.\nfunc (v *GetFailoverInfoResponse) GetPendingShards() (o []int32) {\n\tif v != nil && v.PendingShards != nil {\n\t\treturn v.PendingShards\n\t}\n\n\treturn\n}\n\n// IsSetPendingShards returns true if PendingShards is not nil.\nfunc (v *GetFailoverInfoResponse) IsSetPendingShards() bool {\n\treturn v != nil && v.PendingShards != nil\n}\n\ntype GetMutableStateRequest struct {\n\tDomainUUID          *string                    `json:\"domainUUID,omitempty\"`\n\tExecution           *shared.WorkflowExecution  `json:\"execution,omitempty\"`\n\tExpectedNextEventId *int64                     `json:\"expectedNextEventId,omitempty\"`\n\tCurrentBranchToken  []byte                     `json:\"currentBranchToken,omitempty\"`\n\tVersionHistoryItem  *shared.VersionHistoryItem `json:\"versionHistoryItem,omitempty\"`\n}\n\n// ToWire translates a GetMutableStateRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetMutableStateRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpectedNextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.CurrentBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistoryItem != nil {\n\t\tw, err = v.VersionHistoryItem.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistoryItem_Read(w wire.Value) (*shared.VersionHistoryItem, error) {\n\tvar v shared.VersionHistoryItem\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetMutableStateRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetMutableStateRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetMutableStateRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetMutableStateRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpectedNextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistoryItem, err = _VersionHistoryItem_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetMutableStateRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetMutableStateRequest struct could not be encoded.\nfunc (v *GetMutableStateRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpectedNextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpectedNextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CurrentBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistoryItem != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistoryItem.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistoryItem_Decode(sr stream.Reader) (*shared.VersionHistoryItem, error) {\n\tvar v shared.VersionHistoryItem\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetMutableStateRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetMutableStateRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetMutableStateRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpectedNextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistoryItem, err = _VersionHistoryItem_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetMutableStateRequest\n// struct.\nfunc (v *GetMutableStateRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpectedNextEventId: %v\", *(v.ExpectedNextEventId))\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchToken: %v\", v.CurrentBranchToken)\n\t\ti++\n\t}\n\tif v.VersionHistoryItem != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistoryItem: %v\", v.VersionHistoryItem)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetMutableStateRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this GetMutableStateRequest match the\n// provided GetMutableStateRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetMutableStateRequest) Equals(rhs *GetMutableStateRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpectedNextEventId, rhs.ExpectedNextEventId) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchToken == nil && rhs.CurrentBranchToken == nil) || (v.CurrentBranchToken != nil && rhs.CurrentBranchToken != nil && bytes.Equal(v.CurrentBranchToken, rhs.CurrentBranchToken))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistoryItem == nil && rhs.VersionHistoryItem == nil) || (v.VersionHistoryItem != nil && rhs.VersionHistoryItem != nil && v.VersionHistoryItem.Equals(rhs.VersionHistoryItem))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetMutableStateRequest.\nfunc (v *GetMutableStateRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tenc.AddInt64(\"expectedNextEventId\", *v.ExpectedNextEventId)\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tenc.AddString(\"currentBranchToken\", base64.StdEncoding.EncodeToString(v.CurrentBranchToken))\n\t}\n\tif v.VersionHistoryItem != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistoryItem\", v.VersionHistoryItem))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *GetMutableStateRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *GetMutableStateRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetExpectedNextEventId returns the value of ExpectedNextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateRequest) GetExpectedNextEventId() (o int64) {\n\tif v != nil && v.ExpectedNextEventId != nil {\n\t\treturn *v.ExpectedNextEventId\n\t}\n\n\treturn\n}\n\n// IsSetExpectedNextEventId returns true if ExpectedNextEventId is not nil.\nfunc (v *GetMutableStateRequest) IsSetExpectedNextEventId() bool {\n\treturn v != nil && v.ExpectedNextEventId != nil\n}\n\n// GetCurrentBranchToken returns the value of CurrentBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateRequest) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchToken returns true if CurrentBranchToken is not nil.\nfunc (v *GetMutableStateRequest) IsSetCurrentBranchToken() bool {\n\treturn v != nil && v.CurrentBranchToken != nil\n}\n\n// GetVersionHistoryItem returns the value of VersionHistoryItem if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateRequest) GetVersionHistoryItem() (o *shared.VersionHistoryItem) {\n\tif v != nil && v.VersionHistoryItem != nil {\n\t\treturn v.VersionHistoryItem\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistoryItem returns true if VersionHistoryItem is not nil.\nfunc (v *GetMutableStateRequest) IsSetVersionHistoryItem() bool {\n\treturn v != nil && v.VersionHistoryItem != nil\n}\n\ntype GetMutableStateResponse struct {\n\tExecution                            *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tWorkflowType                         *shared.WorkflowType      `json:\"workflowType,omitempty\"`\n\tNextEventId                          *int64                    `json:\"NextEventId,omitempty\"`\n\tPreviousStartedEventId               *int64                    `json:\"PreviousStartedEventId,omitempty\"`\n\tLastFirstEventId                     *int64                    `json:\"LastFirstEventId,omitempty\"`\n\tTaskList                             *shared.TaskList          `json:\"taskList,omitempty\"`\n\tStickyTaskList                       *shared.TaskList          `json:\"stickyTaskList,omitempty\"`\n\tClientLibraryVersion                 *string                   `json:\"clientLibraryVersion,omitempty\"`\n\tClientFeatureVersion                 *string                   `json:\"clientFeatureVersion,omitempty\"`\n\tClientImpl                           *string                   `json:\"clientImpl,omitempty\"`\n\tIsWorkflowRunning                    *bool                     `json:\"isWorkflowRunning,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *int32                    `json:\"stickyTaskListScheduleToStartTimeout,omitempty\"`\n\tEventStoreVersion                    *int32                    `json:\"eventStoreVersion,omitempty\"`\n\tCurrentBranchToken                   []byte                    `json:\"currentBranchToken,omitempty\"`\n\tWorkflowState                        *int32                    `json:\"workflowState,omitempty\"`\n\tWorkflowCloseState                   *int32                    `json:\"workflowCloseState,omitempty\"`\n\tVersionHistories                     *shared.VersionHistories  `json:\"versionHistories,omitempty\"`\n\tIsStickyTaskListEnabled              *bool                     `json:\"isStickyTaskListEnabled,omitempty\"`\n\tHistorySize                          *int64                    `json:\"historySize,omitempty\"`\n}\n\n// ToWire translates a GetMutableStateResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetMutableStateResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [19]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousStartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastFirstEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tw, err = v.StickyTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientLibraryVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientFeatureVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientImpl)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.IsWorkflowRunning != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsWorkflowRunning)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tw, err = wire.NewValueI32(*(v.StickyTaskListScheduleToStartTimeout)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.EventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.CurrentBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowState != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowCloseState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tw, err = v.VersionHistories.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.IsStickyTaskListEnabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsStickyTaskListEnabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tw, err = wire.NewValueI64(*(v.HistorySize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowType_Read(w wire.Value) (*shared.WorkflowType, error) {\n\tvar v shared.WorkflowType\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TaskList_Read(w wire.Value) (*shared.TaskList, error) {\n\tvar v shared.TaskList\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _VersionHistories_Read(w wire.Value) (*shared.VersionHistories, error) {\n\tvar v shared.VersionHistories\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetMutableStateResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetMutableStateResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetMutableStateResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetMutableStateResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousStartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastFirstEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StickyTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientLibraryVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientFeatureVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientImpl = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsWorkflowRunning = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StickyTaskListScheduleToStartTimeout = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.EventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowCloseState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistories, err = _VersionHistories_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsStickyTaskListEnabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.HistorySize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetMutableStateResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetMutableStateResponse struct could not be encoded.\nfunc (v *GetMutableStateResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousStartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousStartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFirstEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastFirstEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StickyTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientLibraryVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientLibraryVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientFeatureVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientFeatureVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientImpl != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientImpl)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsWorkflowRunning != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsWorkflowRunning)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StickyTaskListScheduleToStartTimeout)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.EventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CurrentBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowCloseState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowCloseState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistories != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistories.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsStickyTaskListEnabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsStickyTaskListEnabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistorySize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.HistorySize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowType_Decode(sr stream.Reader) (*shared.WorkflowType, error) {\n\tvar v shared.WorkflowType\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TaskList_Decode(sr stream.Reader) (*shared.TaskList, error) {\n\tvar v shared.TaskList\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _VersionHistories_Decode(sr stream.Reader) (*shared.VersionHistories, error) {\n\tvar v shared.VersionHistories\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetMutableStateResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetMutableStateResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetMutableStateResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousStartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastFirstEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.StickyTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientLibraryVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientFeatureVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientImpl = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsWorkflowRunning = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StickyTaskListScheduleToStartTimeout = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.EventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowCloseState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistories, err = _VersionHistories_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsStickyTaskListEnabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.HistorySize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetMutableStateResponse\n// struct.\nfunc (v *GetMutableStateResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [19]string\n\ti := 0\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventId: %v\", *(v.NextEventId))\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousStartedEventId: %v\", *(v.PreviousStartedEventId))\n\t\ti++\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFirstEventId: %v\", *(v.LastFirstEventId))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskList: %v\", v.StickyTaskList)\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientLibraryVersion: %v\", *(v.ClientLibraryVersion))\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientFeatureVersion: %v\", *(v.ClientFeatureVersion))\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientImpl: %v\", *(v.ClientImpl))\n\t\ti++\n\t}\n\tif v.IsWorkflowRunning != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsWorkflowRunning: %v\", *(v.IsWorkflowRunning))\n\t\ti++\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskListScheduleToStartTimeout: %v\", *(v.StickyTaskListScheduleToStartTimeout))\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventStoreVersion: %v\", *(v.EventStoreVersion))\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchToken: %v\", v.CurrentBranchToken)\n\t\ti++\n\t}\n\tif v.WorkflowState != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowState: %v\", *(v.WorkflowState))\n\t\ti++\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowCloseState: %v\", *(v.WorkflowCloseState))\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistories: %v\", v.VersionHistories)\n\t\ti++\n\t}\n\tif v.IsStickyTaskListEnabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsStickyTaskListEnabled: %v\", *(v.IsStickyTaskListEnabled))\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistorySize: %v\", *(v.HistorySize))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetMutableStateResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetMutableStateResponse match the\n// provided GetMutableStateResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetMutableStateResponse) Equals(rhs *GetMutableStateResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventId, rhs.NextEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousStartedEventId, rhs.PreviousStartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastFirstEventId, rhs.LastFirstEventId) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.StickyTaskList == nil && rhs.StickyTaskList == nil) || (v.StickyTaskList != nil && rhs.StickyTaskList != nil && v.StickyTaskList.Equals(rhs.StickyTaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientLibraryVersion, rhs.ClientLibraryVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientFeatureVersion, rhs.ClientFeatureVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientImpl, rhs.ClientImpl) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsWorkflowRunning, rhs.IsWorkflowRunning) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StickyTaskListScheduleToStartTimeout, rhs.StickyTaskListScheduleToStartTimeout) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.EventStoreVersion, rhs.EventStoreVersion) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchToken == nil && rhs.CurrentBranchToken == nil) || (v.CurrentBranchToken != nil && rhs.CurrentBranchToken != nil && bytes.Equal(v.CurrentBranchToken, rhs.CurrentBranchToken))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowState, rhs.WorkflowState) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowCloseState, rhs.WorkflowCloseState) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistories == nil && rhs.VersionHistories == nil) || (v.VersionHistories != nil && rhs.VersionHistories != nil && v.VersionHistories.Equals(rhs.VersionHistories))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsStickyTaskListEnabled, rhs.IsStickyTaskListEnabled) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.HistorySize, rhs.HistorySize) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetMutableStateResponse.\nfunc (v *GetMutableStateResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.NextEventId != nil {\n\t\tenc.AddInt64(\"NextEventId\", *v.NextEventId)\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tenc.AddInt64(\"PreviousStartedEventId\", *v.PreviousStartedEventId)\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tenc.AddInt64(\"LastFirstEventId\", *v.LastFirstEventId)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.StickyTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"stickyTaskList\", v.StickyTaskList))\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tenc.AddString(\"clientLibraryVersion\", *v.ClientLibraryVersion)\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tenc.AddString(\"clientFeatureVersion\", *v.ClientFeatureVersion)\n\t}\n\tif v.ClientImpl != nil {\n\t\tenc.AddString(\"clientImpl\", *v.ClientImpl)\n\t}\n\tif v.IsWorkflowRunning != nil {\n\t\tenc.AddBool(\"isWorkflowRunning\", *v.IsWorkflowRunning)\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tenc.AddInt32(\"stickyTaskListScheduleToStartTimeout\", *v.StickyTaskListScheduleToStartTimeout)\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tenc.AddInt32(\"eventStoreVersion\", *v.EventStoreVersion)\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tenc.AddString(\"currentBranchToken\", base64.StdEncoding.EncodeToString(v.CurrentBranchToken))\n\t}\n\tif v.WorkflowState != nil {\n\t\tenc.AddInt32(\"workflowState\", *v.WorkflowState)\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tenc.AddInt32(\"workflowCloseState\", *v.WorkflowCloseState)\n\t}\n\tif v.VersionHistories != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistories\", v.VersionHistories))\n\t}\n\tif v.IsStickyTaskListEnabled != nil {\n\t\tenc.AddBool(\"isStickyTaskListEnabled\", *v.IsStickyTaskListEnabled)\n\t}\n\tif v.HistorySize != nil {\n\t\tenc.AddInt64(\"historySize\", *v.HistorySize)\n\t}\n\treturn err\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *GetMutableStateResponse) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *GetMutableStateResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetNextEventId returns the value of NextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetNextEventId() (o int64) {\n\tif v != nil && v.NextEventId != nil {\n\t\treturn *v.NextEventId\n\t}\n\n\treturn\n}\n\n// IsSetNextEventId returns true if NextEventId is not nil.\nfunc (v *GetMutableStateResponse) IsSetNextEventId() bool {\n\treturn v != nil && v.NextEventId != nil\n}\n\n// GetPreviousStartedEventId returns the value of PreviousStartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetPreviousStartedEventId() (o int64) {\n\tif v != nil && v.PreviousStartedEventId != nil {\n\t\treturn *v.PreviousStartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetPreviousStartedEventId returns true if PreviousStartedEventId is not nil.\nfunc (v *GetMutableStateResponse) IsSetPreviousStartedEventId() bool {\n\treturn v != nil && v.PreviousStartedEventId != nil\n}\n\n// GetLastFirstEventId returns the value of LastFirstEventId if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetLastFirstEventId() (o int64) {\n\tif v != nil && v.LastFirstEventId != nil {\n\t\treturn *v.LastFirstEventId\n\t}\n\n\treturn\n}\n\n// IsSetLastFirstEventId returns true if LastFirstEventId is not nil.\nfunc (v *GetMutableStateResponse) IsSetLastFirstEventId() bool {\n\treturn v != nil && v.LastFirstEventId != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *GetMutableStateResponse) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetStickyTaskList returns the value of StickyTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetStickyTaskList() (o *shared.TaskList) {\n\tif v != nil && v.StickyTaskList != nil {\n\t\treturn v.StickyTaskList\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskList returns true if StickyTaskList is not nil.\nfunc (v *GetMutableStateResponse) IsSetStickyTaskList() bool {\n\treturn v != nil && v.StickyTaskList != nil\n}\n\n// GetClientLibraryVersion returns the value of ClientLibraryVersion if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetClientLibraryVersion() (o string) {\n\tif v != nil && v.ClientLibraryVersion != nil {\n\t\treturn *v.ClientLibraryVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientLibraryVersion returns true if ClientLibraryVersion is not nil.\nfunc (v *GetMutableStateResponse) IsSetClientLibraryVersion() bool {\n\treturn v != nil && v.ClientLibraryVersion != nil\n}\n\n// GetClientFeatureVersion returns the value of ClientFeatureVersion if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetClientFeatureVersion() (o string) {\n\tif v != nil && v.ClientFeatureVersion != nil {\n\t\treturn *v.ClientFeatureVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientFeatureVersion returns true if ClientFeatureVersion is not nil.\nfunc (v *GetMutableStateResponse) IsSetClientFeatureVersion() bool {\n\treturn v != nil && v.ClientFeatureVersion != nil\n}\n\n// GetClientImpl returns the value of ClientImpl if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetClientImpl() (o string) {\n\tif v != nil && v.ClientImpl != nil {\n\t\treturn *v.ClientImpl\n\t}\n\n\treturn\n}\n\n// IsSetClientImpl returns true if ClientImpl is not nil.\nfunc (v *GetMutableStateResponse) IsSetClientImpl() bool {\n\treturn v != nil && v.ClientImpl != nil\n}\n\n// GetIsWorkflowRunning returns the value of IsWorkflowRunning if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetIsWorkflowRunning() (o bool) {\n\tif v != nil && v.IsWorkflowRunning != nil {\n\t\treturn *v.IsWorkflowRunning\n\t}\n\n\treturn\n}\n\n// IsSetIsWorkflowRunning returns true if IsWorkflowRunning is not nil.\nfunc (v *GetMutableStateResponse) IsSetIsWorkflowRunning() bool {\n\treturn v != nil && v.IsWorkflowRunning != nil\n}\n\n// GetStickyTaskListScheduleToStartTimeout returns the value of StickyTaskListScheduleToStartTimeout if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetStickyTaskListScheduleToStartTimeout() (o int32) {\n\tif v != nil && v.StickyTaskListScheduleToStartTimeout != nil {\n\t\treturn *v.StickyTaskListScheduleToStartTimeout\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskListScheduleToStartTimeout returns true if StickyTaskListScheduleToStartTimeout is not nil.\nfunc (v *GetMutableStateResponse) IsSetStickyTaskListScheduleToStartTimeout() bool {\n\treturn v != nil && v.StickyTaskListScheduleToStartTimeout != nil\n}\n\n// GetEventStoreVersion returns the value of EventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetEventStoreVersion() (o int32) {\n\tif v != nil && v.EventStoreVersion != nil {\n\t\treturn *v.EventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetEventStoreVersion returns true if EventStoreVersion is not nil.\nfunc (v *GetMutableStateResponse) IsSetEventStoreVersion() bool {\n\treturn v != nil && v.EventStoreVersion != nil\n}\n\n// GetCurrentBranchToken returns the value of CurrentBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchToken returns true if CurrentBranchToken is not nil.\nfunc (v *GetMutableStateResponse) IsSetCurrentBranchToken() bool {\n\treturn v != nil && v.CurrentBranchToken != nil\n}\n\n// GetWorkflowState returns the value of WorkflowState if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetWorkflowState() (o int32) {\n\tif v != nil && v.WorkflowState != nil {\n\t\treturn *v.WorkflowState\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowState returns true if WorkflowState is not nil.\nfunc (v *GetMutableStateResponse) IsSetWorkflowState() bool {\n\treturn v != nil && v.WorkflowState != nil\n}\n\n// GetWorkflowCloseState returns the value of WorkflowCloseState if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetWorkflowCloseState() (o int32) {\n\tif v != nil && v.WorkflowCloseState != nil {\n\t\treturn *v.WorkflowCloseState\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowCloseState returns true if WorkflowCloseState is not nil.\nfunc (v *GetMutableStateResponse) IsSetWorkflowCloseState() bool {\n\treturn v != nil && v.WorkflowCloseState != nil\n}\n\n// GetVersionHistories returns the value of VersionHistories if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetVersionHistories() (o *shared.VersionHistories) {\n\tif v != nil && v.VersionHistories != nil {\n\t\treturn v.VersionHistories\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistories returns true if VersionHistories is not nil.\nfunc (v *GetMutableStateResponse) IsSetVersionHistories() bool {\n\treturn v != nil && v.VersionHistories != nil\n}\n\n// GetIsStickyTaskListEnabled returns the value of IsStickyTaskListEnabled if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetIsStickyTaskListEnabled() (o bool) {\n\tif v != nil && v.IsStickyTaskListEnabled != nil {\n\t\treturn *v.IsStickyTaskListEnabled\n\t}\n\n\treturn\n}\n\n// IsSetIsStickyTaskListEnabled returns true if IsStickyTaskListEnabled is not nil.\nfunc (v *GetMutableStateResponse) IsSetIsStickyTaskListEnabled() bool {\n\treturn v != nil && v.IsStickyTaskListEnabled != nil\n}\n\n// GetHistorySize returns the value of HistorySize if it is set or its\n// zero value if it is unset.\nfunc (v *GetMutableStateResponse) GetHistorySize() (o int64) {\n\tif v != nil && v.HistorySize != nil {\n\t\treturn *v.HistorySize\n\t}\n\n\treturn\n}\n\n// IsSetHistorySize returns true if HistorySize is not nil.\nfunc (v *GetMutableStateResponse) IsSetHistorySize() bool {\n\treturn v != nil && v.HistorySize != nil\n}\n\ntype NotifyFailoverMarkersRequest struct {\n\tFailoverMarkerTokens []*FailoverMarkerToken `json:\"failoverMarkerTokens,omitempty\"`\n}\n\ntype _List_FailoverMarkerToken_ValueList []*FailoverMarkerToken\n\nfunc (v _List_FailoverMarkerToken_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverMarkerToken', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_FailoverMarkerToken_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_FailoverMarkerToken_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_FailoverMarkerToken_ValueList) Close() {}\n\n// ToWire translates a NotifyFailoverMarkersRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *NotifyFailoverMarkersRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailoverMarkerTokens != nil {\n\t\tw, err = wire.NewValueList(_List_FailoverMarkerToken_ValueList(v.FailoverMarkerTokens)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverMarkerToken_Read(w wire.Value) (*FailoverMarkerToken, error) {\n\tvar v FailoverMarkerToken\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_FailoverMarkerToken_Read(l wire.ValueList) ([]*FailoverMarkerToken, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*FailoverMarkerToken, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _FailoverMarkerToken_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a NotifyFailoverMarkersRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a NotifyFailoverMarkersRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v NotifyFailoverMarkersRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *NotifyFailoverMarkersRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.FailoverMarkerTokens, err = _List_FailoverMarkerToken_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_FailoverMarkerToken_Encode(val []*FailoverMarkerToken, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverMarkerToken', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a NotifyFailoverMarkersRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a NotifyFailoverMarkersRequest struct could not be encoded.\nfunc (v *NotifyFailoverMarkersRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailoverMarkerTokens != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_FailoverMarkerToken_Encode(v.FailoverMarkerTokens, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverMarkerToken_Decode(sr stream.Reader) (*FailoverMarkerToken, error) {\n\tvar v FailoverMarkerToken\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_FailoverMarkerToken_Decode(sr stream.Reader) ([]*FailoverMarkerToken, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*FailoverMarkerToken, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _FailoverMarkerToken_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a NotifyFailoverMarkersRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a NotifyFailoverMarkersRequest struct could not be generated from the wire\n// representation.\nfunc (v *NotifyFailoverMarkersRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.FailoverMarkerTokens, err = _List_FailoverMarkerToken_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a NotifyFailoverMarkersRequest\n// struct.\nfunc (v *NotifyFailoverMarkersRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailoverMarkerTokens != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverMarkerTokens: %v\", v.FailoverMarkerTokens)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"NotifyFailoverMarkersRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_FailoverMarkerToken_Equals(lhs, rhs []*FailoverMarkerToken) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this NotifyFailoverMarkersRequest match the\n// provided NotifyFailoverMarkersRequest.\n//\n// This function performs a deep comparison.\nfunc (v *NotifyFailoverMarkersRequest) Equals(rhs *NotifyFailoverMarkersRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailoverMarkerTokens == nil && rhs.FailoverMarkerTokens == nil) || (v.FailoverMarkerTokens != nil && rhs.FailoverMarkerTokens != nil && _List_FailoverMarkerToken_Equals(v.FailoverMarkerTokens, rhs.FailoverMarkerTokens))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_FailoverMarkerToken_Zapper []*FailoverMarkerToken\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_FailoverMarkerToken_Zapper.\nfunc (l _List_FailoverMarkerToken_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of NotifyFailoverMarkersRequest.\nfunc (v *NotifyFailoverMarkersRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailoverMarkerTokens != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"failoverMarkerTokens\", (_List_FailoverMarkerToken_Zapper)(v.FailoverMarkerTokens)))\n\t}\n\treturn err\n}\n\n// GetFailoverMarkerTokens returns the value of FailoverMarkerTokens if it is set or its\n// zero value if it is unset.\nfunc (v *NotifyFailoverMarkersRequest) GetFailoverMarkerTokens() (o []*FailoverMarkerToken) {\n\tif v != nil && v.FailoverMarkerTokens != nil {\n\t\treturn v.FailoverMarkerTokens\n\t}\n\n\treturn\n}\n\n// IsSetFailoverMarkerTokens returns true if FailoverMarkerTokens is not nil.\nfunc (v *NotifyFailoverMarkersRequest) IsSetFailoverMarkerTokens() bool {\n\treturn v != nil && v.FailoverMarkerTokens != nil\n}\n\ntype ParentExecutionInfo struct {\n\tDomainUUID  *string                   `json:\"domainUUID,omitempty\"`\n\tDomain      *string                   `json:\"domain,omitempty\"`\n\tExecution   *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tInitiatedId *int64                    `json:\"initiatedId,omitempty\"`\n}\n\n// ToWire translates a ParentExecutionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ParentExecutionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 15, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ParentExecutionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ParentExecutionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ParentExecutionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ParentExecutionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 15:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ParentExecutionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ParentExecutionInfo struct could not be encoded.\nfunc (v *ParentExecutionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 15, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ParentExecutionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ParentExecutionInfo struct could not be generated from the wire\n// representation.\nfunc (v *ParentExecutionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 15 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ParentExecutionInfo\n// struct.\nfunc (v *ParentExecutionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.InitiatedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedId: %v\", *(v.InitiatedId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ParentExecutionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ParentExecutionInfo match the\n// provided ParentExecutionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ParentExecutionInfo) Equals(rhs *ParentExecutionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedId, rhs.InitiatedId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ParentExecutionInfo.\nfunc (v *ParentExecutionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.InitiatedId != nil {\n\t\tenc.AddInt64(\"initiatedId\", *v.InitiatedId)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ParentExecutionInfo) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ParentExecutionInfo) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ParentExecutionInfo) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ParentExecutionInfo) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *ParentExecutionInfo) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *ParentExecutionInfo) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetInitiatedId returns the value of InitiatedId if it is set or its\n// zero value if it is unset.\nfunc (v *ParentExecutionInfo) GetInitiatedId() (o int64) {\n\tif v != nil && v.InitiatedId != nil {\n\t\treturn *v.InitiatedId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedId returns true if InitiatedId is not nil.\nfunc (v *ParentExecutionInfo) IsSetInitiatedId() bool {\n\treturn v != nil && v.InitiatedId != nil\n}\n\ntype PollMutableStateRequest struct {\n\tDomainUUID          *string                   `json:\"domainUUID,omitempty\"`\n\tExecution           *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tExpectedNextEventId *int64                    `json:\"expectedNextEventId,omitempty\"`\n\tCurrentBranchToken  []byte                    `json:\"currentBranchToken,omitempty\"`\n}\n\n// ToWire translates a PollMutableStateRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollMutableStateRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpectedNextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.CurrentBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PollMutableStateRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollMutableStateRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollMutableStateRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollMutableStateRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpectedNextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollMutableStateRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollMutableStateRequest struct could not be encoded.\nfunc (v *PollMutableStateRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpectedNextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpectedNextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CurrentBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PollMutableStateRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollMutableStateRequest struct could not be generated from the wire\n// representation.\nfunc (v *PollMutableStateRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpectedNextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollMutableStateRequest\n// struct.\nfunc (v *PollMutableStateRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpectedNextEventId: %v\", *(v.ExpectedNextEventId))\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchToken: %v\", v.CurrentBranchToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollMutableStateRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollMutableStateRequest match the\n// provided PollMutableStateRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PollMutableStateRequest) Equals(rhs *PollMutableStateRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpectedNextEventId, rhs.ExpectedNextEventId) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchToken == nil && rhs.CurrentBranchToken == nil) || (v.CurrentBranchToken != nil && rhs.CurrentBranchToken != nil && bytes.Equal(v.CurrentBranchToken, rhs.CurrentBranchToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollMutableStateRequest.\nfunc (v *PollMutableStateRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.ExpectedNextEventId != nil {\n\t\tenc.AddInt64(\"expectedNextEventId\", *v.ExpectedNextEventId)\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tenc.AddString(\"currentBranchToken\", base64.StdEncoding.EncodeToString(v.CurrentBranchToken))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *PollMutableStateRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *PollMutableStateRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetExpectedNextEventId returns the value of ExpectedNextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateRequest) GetExpectedNextEventId() (o int64) {\n\tif v != nil && v.ExpectedNextEventId != nil {\n\t\treturn *v.ExpectedNextEventId\n\t}\n\n\treturn\n}\n\n// IsSetExpectedNextEventId returns true if ExpectedNextEventId is not nil.\nfunc (v *PollMutableStateRequest) IsSetExpectedNextEventId() bool {\n\treturn v != nil && v.ExpectedNextEventId != nil\n}\n\n// GetCurrentBranchToken returns the value of CurrentBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateRequest) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchToken returns true if CurrentBranchToken is not nil.\nfunc (v *PollMutableStateRequest) IsSetCurrentBranchToken() bool {\n\treturn v != nil && v.CurrentBranchToken != nil\n}\n\ntype PollMutableStateResponse struct {\n\tExecution                            *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tWorkflowType                         *shared.WorkflowType      `json:\"workflowType,omitempty\"`\n\tNextEventId                          *int64                    `json:\"NextEventId,omitempty\"`\n\tPreviousStartedEventId               *int64                    `json:\"PreviousStartedEventId,omitempty\"`\n\tLastFirstEventId                     *int64                    `json:\"LastFirstEventId,omitempty\"`\n\tTaskList                             *shared.TaskList          `json:\"taskList,omitempty\"`\n\tStickyTaskList                       *shared.TaskList          `json:\"stickyTaskList,omitempty\"`\n\tClientLibraryVersion                 *string                   `json:\"clientLibraryVersion,omitempty\"`\n\tClientFeatureVersion                 *string                   `json:\"clientFeatureVersion,omitempty\"`\n\tClientImpl                           *string                   `json:\"clientImpl,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *int32                    `json:\"stickyTaskListScheduleToStartTimeout,omitempty\"`\n\tCurrentBranchToken                   []byte                    `json:\"currentBranchToken,omitempty\"`\n\tVersionHistories                     *shared.VersionHistories  `json:\"versionHistories,omitempty\"`\n\tWorkflowState                        *int32                    `json:\"workflowState,omitempty\"`\n\tWorkflowCloseState                   *int32                    `json:\"workflowCloseState,omitempty\"`\n}\n\n// ToWire translates a PollMutableStateResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollMutableStateResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousStartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastFirstEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tw, err = v.StickyTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientLibraryVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientFeatureVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientImpl)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tw, err = wire.NewValueI32(*(v.StickyTaskListScheduleToStartTimeout)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.CurrentBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tw, err = v.VersionHistories.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowState != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowCloseState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PollMutableStateResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollMutableStateResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollMutableStateResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollMutableStateResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousStartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastFirstEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StickyTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientLibraryVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientFeatureVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientImpl = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StickyTaskListScheduleToStartTimeout = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistories, err = _VersionHistories_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowCloseState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollMutableStateResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollMutableStateResponse struct could not be encoded.\nfunc (v *PollMutableStateResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousStartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousStartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFirstEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastFirstEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StickyTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientLibraryVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientLibraryVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientFeatureVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientFeatureVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientImpl != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientImpl)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StickyTaskListScheduleToStartTimeout)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CurrentBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistories != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistories.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowCloseState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowCloseState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PollMutableStateResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollMutableStateResponse struct could not be generated from the wire\n// representation.\nfunc (v *PollMutableStateResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousStartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastFirstEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.StickyTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientLibraryVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientFeatureVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientImpl = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StickyTaskListScheduleToStartTimeout = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistories, err = _VersionHistories_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowCloseState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollMutableStateResponse\n// struct.\nfunc (v *PollMutableStateResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventId: %v\", *(v.NextEventId))\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousStartedEventId: %v\", *(v.PreviousStartedEventId))\n\t\ti++\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFirstEventId: %v\", *(v.LastFirstEventId))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskList: %v\", v.StickyTaskList)\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientLibraryVersion: %v\", *(v.ClientLibraryVersion))\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientFeatureVersion: %v\", *(v.ClientFeatureVersion))\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientImpl: %v\", *(v.ClientImpl))\n\t\ti++\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskListScheduleToStartTimeout: %v\", *(v.StickyTaskListScheduleToStartTimeout))\n\t\ti++\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchToken: %v\", v.CurrentBranchToken)\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistories: %v\", v.VersionHistories)\n\t\ti++\n\t}\n\tif v.WorkflowState != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowState: %v\", *(v.WorkflowState))\n\t\ti++\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowCloseState: %v\", *(v.WorkflowCloseState))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollMutableStateResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollMutableStateResponse match the\n// provided PollMutableStateResponse.\n//\n// This function performs a deep comparison.\nfunc (v *PollMutableStateResponse) Equals(rhs *PollMutableStateResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventId, rhs.NextEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousStartedEventId, rhs.PreviousStartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastFirstEventId, rhs.LastFirstEventId) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.StickyTaskList == nil && rhs.StickyTaskList == nil) || (v.StickyTaskList != nil && rhs.StickyTaskList != nil && v.StickyTaskList.Equals(rhs.StickyTaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientLibraryVersion, rhs.ClientLibraryVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientFeatureVersion, rhs.ClientFeatureVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientImpl, rhs.ClientImpl) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StickyTaskListScheduleToStartTimeout, rhs.StickyTaskListScheduleToStartTimeout) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchToken == nil && rhs.CurrentBranchToken == nil) || (v.CurrentBranchToken != nil && rhs.CurrentBranchToken != nil && bytes.Equal(v.CurrentBranchToken, rhs.CurrentBranchToken))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistories == nil && rhs.VersionHistories == nil) || (v.VersionHistories != nil && rhs.VersionHistories != nil && v.VersionHistories.Equals(rhs.VersionHistories))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowState, rhs.WorkflowState) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowCloseState, rhs.WorkflowCloseState) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollMutableStateResponse.\nfunc (v *PollMutableStateResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.NextEventId != nil {\n\t\tenc.AddInt64(\"NextEventId\", *v.NextEventId)\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tenc.AddInt64(\"PreviousStartedEventId\", *v.PreviousStartedEventId)\n\t}\n\tif v.LastFirstEventId != nil {\n\t\tenc.AddInt64(\"LastFirstEventId\", *v.LastFirstEventId)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.StickyTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"stickyTaskList\", v.StickyTaskList))\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tenc.AddString(\"clientLibraryVersion\", *v.ClientLibraryVersion)\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tenc.AddString(\"clientFeatureVersion\", *v.ClientFeatureVersion)\n\t}\n\tif v.ClientImpl != nil {\n\t\tenc.AddString(\"clientImpl\", *v.ClientImpl)\n\t}\n\tif v.StickyTaskListScheduleToStartTimeout != nil {\n\t\tenc.AddInt32(\"stickyTaskListScheduleToStartTimeout\", *v.StickyTaskListScheduleToStartTimeout)\n\t}\n\tif v.CurrentBranchToken != nil {\n\t\tenc.AddString(\"currentBranchToken\", base64.StdEncoding.EncodeToString(v.CurrentBranchToken))\n\t}\n\tif v.VersionHistories != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistories\", v.VersionHistories))\n\t}\n\tif v.WorkflowState != nil {\n\t\tenc.AddInt32(\"workflowState\", *v.WorkflowState)\n\t}\n\tif v.WorkflowCloseState != nil {\n\t\tenc.AddInt32(\"workflowCloseState\", *v.WorkflowCloseState)\n\t}\n\treturn err\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *PollMutableStateResponse) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *PollMutableStateResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetNextEventId returns the value of NextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetNextEventId() (o int64) {\n\tif v != nil && v.NextEventId != nil {\n\t\treturn *v.NextEventId\n\t}\n\n\treturn\n}\n\n// IsSetNextEventId returns true if NextEventId is not nil.\nfunc (v *PollMutableStateResponse) IsSetNextEventId() bool {\n\treturn v != nil && v.NextEventId != nil\n}\n\n// GetPreviousStartedEventId returns the value of PreviousStartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetPreviousStartedEventId() (o int64) {\n\tif v != nil && v.PreviousStartedEventId != nil {\n\t\treturn *v.PreviousStartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetPreviousStartedEventId returns true if PreviousStartedEventId is not nil.\nfunc (v *PollMutableStateResponse) IsSetPreviousStartedEventId() bool {\n\treturn v != nil && v.PreviousStartedEventId != nil\n}\n\n// GetLastFirstEventId returns the value of LastFirstEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetLastFirstEventId() (o int64) {\n\tif v != nil && v.LastFirstEventId != nil {\n\t\treturn *v.LastFirstEventId\n\t}\n\n\treturn\n}\n\n// IsSetLastFirstEventId returns true if LastFirstEventId is not nil.\nfunc (v *PollMutableStateResponse) IsSetLastFirstEventId() bool {\n\treturn v != nil && v.LastFirstEventId != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *PollMutableStateResponse) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetStickyTaskList returns the value of StickyTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetStickyTaskList() (o *shared.TaskList) {\n\tif v != nil && v.StickyTaskList != nil {\n\t\treturn v.StickyTaskList\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskList returns true if StickyTaskList is not nil.\nfunc (v *PollMutableStateResponse) IsSetStickyTaskList() bool {\n\treturn v != nil && v.StickyTaskList != nil\n}\n\n// GetClientLibraryVersion returns the value of ClientLibraryVersion if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetClientLibraryVersion() (o string) {\n\tif v != nil && v.ClientLibraryVersion != nil {\n\t\treturn *v.ClientLibraryVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientLibraryVersion returns true if ClientLibraryVersion is not nil.\nfunc (v *PollMutableStateResponse) IsSetClientLibraryVersion() bool {\n\treturn v != nil && v.ClientLibraryVersion != nil\n}\n\n// GetClientFeatureVersion returns the value of ClientFeatureVersion if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetClientFeatureVersion() (o string) {\n\tif v != nil && v.ClientFeatureVersion != nil {\n\t\treturn *v.ClientFeatureVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientFeatureVersion returns true if ClientFeatureVersion is not nil.\nfunc (v *PollMutableStateResponse) IsSetClientFeatureVersion() bool {\n\treturn v != nil && v.ClientFeatureVersion != nil\n}\n\n// GetClientImpl returns the value of ClientImpl if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetClientImpl() (o string) {\n\tif v != nil && v.ClientImpl != nil {\n\t\treturn *v.ClientImpl\n\t}\n\n\treturn\n}\n\n// IsSetClientImpl returns true if ClientImpl is not nil.\nfunc (v *PollMutableStateResponse) IsSetClientImpl() bool {\n\treturn v != nil && v.ClientImpl != nil\n}\n\n// GetStickyTaskListScheduleToStartTimeout returns the value of StickyTaskListScheduleToStartTimeout if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetStickyTaskListScheduleToStartTimeout() (o int32) {\n\tif v != nil && v.StickyTaskListScheduleToStartTimeout != nil {\n\t\treturn *v.StickyTaskListScheduleToStartTimeout\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskListScheduleToStartTimeout returns true if StickyTaskListScheduleToStartTimeout is not nil.\nfunc (v *PollMutableStateResponse) IsSetStickyTaskListScheduleToStartTimeout() bool {\n\treturn v != nil && v.StickyTaskListScheduleToStartTimeout != nil\n}\n\n// GetCurrentBranchToken returns the value of CurrentBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchToken returns true if CurrentBranchToken is not nil.\nfunc (v *PollMutableStateResponse) IsSetCurrentBranchToken() bool {\n\treturn v != nil && v.CurrentBranchToken != nil\n}\n\n// GetVersionHistories returns the value of VersionHistories if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetVersionHistories() (o *shared.VersionHistories) {\n\tif v != nil && v.VersionHistories != nil {\n\t\treturn v.VersionHistories\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistories returns true if VersionHistories is not nil.\nfunc (v *PollMutableStateResponse) IsSetVersionHistories() bool {\n\treturn v != nil && v.VersionHistories != nil\n}\n\n// GetWorkflowState returns the value of WorkflowState if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetWorkflowState() (o int32) {\n\tif v != nil && v.WorkflowState != nil {\n\t\treturn *v.WorkflowState\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowState returns true if WorkflowState is not nil.\nfunc (v *PollMutableStateResponse) IsSetWorkflowState() bool {\n\treturn v != nil && v.WorkflowState != nil\n}\n\n// GetWorkflowCloseState returns the value of WorkflowCloseState if it is set or its\n// zero value if it is unset.\nfunc (v *PollMutableStateResponse) GetWorkflowCloseState() (o int32) {\n\tif v != nil && v.WorkflowCloseState != nil {\n\t\treturn *v.WorkflowCloseState\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowCloseState returns true if WorkflowCloseState is not nil.\nfunc (v *PollMutableStateResponse) IsSetWorkflowCloseState() bool {\n\treturn v != nil && v.WorkflowCloseState != nil\n}\n\ntype ProcessingQueueState struct {\n\tLevel        *int32        `json:\"level,omitempty\"`\n\tAckLevel     *int64        `json:\"ackLevel,omitempty\"`\n\tMaxLevel     *int64        `json:\"maxLevel,omitempty\"`\n\tDomainFilter *DomainFilter `json:\"domainFilter,omitempty\"`\n}\n\n// ToWire translates a ProcessingQueueState struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ProcessingQueueState) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Level != nil {\n\t\tw, err = wire.NewValueI32(*(v.Level)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.AckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.MaxLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.MaxLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.DomainFilter != nil {\n\t\tw, err = v.DomainFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainFilter_Read(w wire.Value) (*DomainFilter, error) {\n\tvar v DomainFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ProcessingQueueState struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ProcessingQueueState struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ProcessingQueueState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ProcessingQueueState) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Level = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.AckLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.MaxLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainFilter, err = _DomainFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ProcessingQueueState struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ProcessingQueueState struct could not be encoded.\nfunc (v *ProcessingQueueState) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Level != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Level)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.AckLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaxLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.MaxLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainFilter_Decode(sr stream.Reader) (*DomainFilter, error) {\n\tvar v DomainFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ProcessingQueueState struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ProcessingQueueState struct could not be generated from the wire\n// representation.\nfunc (v *ProcessingQueueState) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Level = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.AckLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.MaxLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.DomainFilter, err = _DomainFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ProcessingQueueState\n// struct.\nfunc (v *ProcessingQueueState) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Level != nil {\n\t\tfields[i] = fmt.Sprintf(\"Level: %v\", *(v.Level))\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"AckLevel: %v\", *(v.AckLevel))\n\t\ti++\n\t}\n\tif v.MaxLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaxLevel: %v\", *(v.MaxLevel))\n\t\ti++\n\t}\n\tif v.DomainFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainFilter: %v\", v.DomainFilter)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ProcessingQueueState{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ProcessingQueueState match the\n// provided ProcessingQueueState.\n//\n// This function performs a deep comparison.\nfunc (v *ProcessingQueueState) Equals(rhs *ProcessingQueueState) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Level, rhs.Level) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.AckLevel, rhs.AckLevel) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.MaxLevel, rhs.MaxLevel) {\n\t\treturn false\n\t}\n\tif !((v.DomainFilter == nil && rhs.DomainFilter == nil) || (v.DomainFilter != nil && rhs.DomainFilter != nil && v.DomainFilter.Equals(rhs.DomainFilter))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ProcessingQueueState.\nfunc (v *ProcessingQueueState) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Level != nil {\n\t\tenc.AddInt32(\"level\", *v.Level)\n\t}\n\tif v.AckLevel != nil {\n\t\tenc.AddInt64(\"ackLevel\", *v.AckLevel)\n\t}\n\tif v.MaxLevel != nil {\n\t\tenc.AddInt64(\"maxLevel\", *v.MaxLevel)\n\t}\n\tif v.DomainFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainFilter\", v.DomainFilter))\n\t}\n\treturn err\n}\n\n// GetLevel returns the value of Level if it is set or its\n// zero value if it is unset.\nfunc (v *ProcessingQueueState) GetLevel() (o int32) {\n\tif v != nil && v.Level != nil {\n\t\treturn *v.Level\n\t}\n\n\treturn\n}\n\n// IsSetLevel returns true if Level is not nil.\nfunc (v *ProcessingQueueState) IsSetLevel() bool {\n\treturn v != nil && v.Level != nil\n}\n\n// GetAckLevel returns the value of AckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ProcessingQueueState) GetAckLevel() (o int64) {\n\tif v != nil && v.AckLevel != nil {\n\t\treturn *v.AckLevel\n\t}\n\n\treturn\n}\n\n// IsSetAckLevel returns true if AckLevel is not nil.\nfunc (v *ProcessingQueueState) IsSetAckLevel() bool {\n\treturn v != nil && v.AckLevel != nil\n}\n\n// GetMaxLevel returns the value of MaxLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ProcessingQueueState) GetMaxLevel() (o int64) {\n\tif v != nil && v.MaxLevel != nil {\n\t\treturn *v.MaxLevel\n\t}\n\n\treturn\n}\n\n// IsSetMaxLevel returns true if MaxLevel is not nil.\nfunc (v *ProcessingQueueState) IsSetMaxLevel() bool {\n\treturn v != nil && v.MaxLevel != nil\n}\n\n// GetDomainFilter returns the value of DomainFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ProcessingQueueState) GetDomainFilter() (o *DomainFilter) {\n\tif v != nil && v.DomainFilter != nil {\n\t\treturn v.DomainFilter\n\t}\n\n\treturn\n}\n\n// IsSetDomainFilter returns true if DomainFilter is not nil.\nfunc (v *ProcessingQueueState) IsSetDomainFilter() bool {\n\treturn v != nil && v.DomainFilter != nil\n}\n\ntype ProcessingQueueStates struct {\n\tStatesByCluster map[string][]*ProcessingQueueState `json:\"statesByCluster,omitempty\"`\n}\n\ntype _List_ProcessingQueueState_ValueList []*ProcessingQueueState\n\nfunc (v _List_ProcessingQueueState_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ProcessingQueueState', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ProcessingQueueState_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ProcessingQueueState_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ProcessingQueueState_ValueList) Close() {}\n\ntype _Map_String_List_ProcessingQueueState_MapItemList map[string][]*ProcessingQueueState\n\nfunc (m _Map_String_List_ProcessingQueueState_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]*ProcessingQueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueList(_List_ProcessingQueueState_ValueList(v)), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_List_ProcessingQueueState_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_List_ProcessingQueueState_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_List_ProcessingQueueState_MapItemList) ValueType() wire.Type {\n\treturn wire.TList\n}\n\nfunc (_Map_String_List_ProcessingQueueState_MapItemList) Close() {}\n\n// ToWire translates a ProcessingQueueStates struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ProcessingQueueStates) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StatesByCluster != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_List_ProcessingQueueState_MapItemList(v.StatesByCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ProcessingQueueState_Read(w wire.Value) (*ProcessingQueueState, error) {\n\tvar v ProcessingQueueState\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ProcessingQueueState_Read(l wire.ValueList) ([]*ProcessingQueueState, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ProcessingQueueState, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ProcessingQueueState_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _Map_String_List_ProcessingQueueState_Read(m wire.MapItemList) (map[string][]*ProcessingQueueState, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TList {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string][]*ProcessingQueueState, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _List_ProcessingQueueState_Read(x.Value.GetList())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ProcessingQueueStates struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ProcessingQueueStates struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ProcessingQueueStates\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ProcessingQueueStates) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.StatesByCluster, err = _Map_String_List_ProcessingQueueState_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ProcessingQueueState_Encode(val []*ProcessingQueueState, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ProcessingQueueState', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _Map_String_List_ProcessingQueueState_Encode(val map[string][]*ProcessingQueueState, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TList,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]*ProcessingQueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ProcessingQueueState_Encode(v, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a ProcessingQueueStates struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ProcessingQueueStates struct could not be encoded.\nfunc (v *ProcessingQueueStates) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StatesByCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_List_ProcessingQueueState_Encode(v.StatesByCluster, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ProcessingQueueState_Decode(sr stream.Reader) (*ProcessingQueueState, error) {\n\tvar v ProcessingQueueState\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ProcessingQueueState_Decode(sr stream.Reader) ([]*ProcessingQueueState, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ProcessingQueueState, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ProcessingQueueState_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _Map_String_List_ProcessingQueueState_Decode(sr stream.Reader) (map[string][]*ProcessingQueueState, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TList {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string][]*ProcessingQueueState, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _List_ProcessingQueueState_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ProcessingQueueStates struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ProcessingQueueStates struct could not be generated from the wire\n// representation.\nfunc (v *ProcessingQueueStates) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.StatesByCluster, err = _Map_String_List_ProcessingQueueState_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ProcessingQueueStates\n// struct.\nfunc (v *ProcessingQueueStates) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.StatesByCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"StatesByCluster: %v\", v.StatesByCluster)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ProcessingQueueStates{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ProcessingQueueState_Equals(lhs, rhs []*ProcessingQueueState) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _Map_String_List_ProcessingQueueState_Equals(lhs, rhs map[string][]*ProcessingQueueState) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !_List_ProcessingQueueState_Equals(lv, rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this ProcessingQueueStates match the\n// provided ProcessingQueueStates.\n//\n// This function performs a deep comparison.\nfunc (v *ProcessingQueueStates) Equals(rhs *ProcessingQueueStates) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.StatesByCluster == nil && rhs.StatesByCluster == nil) || (v.StatesByCluster != nil && rhs.StatesByCluster != nil && _Map_String_List_ProcessingQueueState_Equals(v.StatesByCluster, rhs.StatesByCluster))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ProcessingQueueState_Zapper []*ProcessingQueueState\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ProcessingQueueState_Zapper.\nfunc (l _List_ProcessingQueueState_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\ntype _Map_String_List_ProcessingQueueState_Zapper map[string][]*ProcessingQueueState\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_List_ProcessingQueueState_Zapper.\nfunc (m _Map_String_List_ProcessingQueueState_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddArray((string)(k), (_List_ProcessingQueueState_Zapper)(v)))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ProcessingQueueStates.\nfunc (v *ProcessingQueueStates) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StatesByCluster != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"statesByCluster\", (_Map_String_List_ProcessingQueueState_Zapper)(v.StatesByCluster)))\n\t}\n\treturn err\n}\n\n// GetStatesByCluster returns the value of StatesByCluster if it is set or its\n// zero value if it is unset.\nfunc (v *ProcessingQueueStates) GetStatesByCluster() (o map[string][]*ProcessingQueueState) {\n\tif v != nil && v.StatesByCluster != nil {\n\t\treturn v.StatesByCluster\n\t}\n\n\treturn\n}\n\n// IsSetStatesByCluster returns true if StatesByCluster is not nil.\nfunc (v *ProcessingQueueStates) IsSetStatesByCluster() bool {\n\treturn v != nil && v.StatesByCluster != nil\n}\n\ntype QueryWorkflowRequest struct {\n\tDomainUUID *string                      `json:\"domainUUID,omitempty\"`\n\tRequest    *shared.QueryWorkflowRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a QueryWorkflowRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryWorkflowRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowRequest_Read(w wire.Value) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a QueryWorkflowRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryWorkflowRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryWorkflowRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryWorkflowRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _QueryWorkflowRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryWorkflowRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be encoded.\nfunc (v *QueryWorkflowRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowRequest_Decode(sr stream.Reader) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a QueryWorkflowRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be generated from the wire\n// representation.\nfunc (v *QueryWorkflowRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _QueryWorkflowRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryWorkflowRequest\n// struct.\nfunc (v *QueryWorkflowRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryWorkflowRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this QueryWorkflowRequest match the\n// provided QueryWorkflowRequest.\n//\n// This function performs a deep comparison.\nfunc (v *QueryWorkflowRequest) Equals(rhs *QueryWorkflowRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryWorkflowRequest.\nfunc (v *QueryWorkflowRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *QueryWorkflowRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetRequest() (o *shared.QueryWorkflowRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *QueryWorkflowRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype QueryWorkflowResponse struct {\n\tResponse *shared.QueryWorkflowResponse `json:\"response,omitempty\"`\n}\n\n// ToWire translates a QueryWorkflowResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryWorkflowResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Response != nil {\n\t\tw, err = v.Response.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowResponse_Read(w wire.Value) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a QueryWorkflowResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryWorkflowResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryWorkflowResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryWorkflowResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Response, err = _QueryWorkflowResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryWorkflowResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryWorkflowResponse struct could not be encoded.\nfunc (v *QueryWorkflowResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Response != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Response.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowResponse_Decode(sr stream.Reader) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a QueryWorkflowResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryWorkflowResponse struct could not be generated from the wire\n// representation.\nfunc (v *QueryWorkflowResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Response, err = _QueryWorkflowResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryWorkflowResponse\n// struct.\nfunc (v *QueryWorkflowResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Response != nil {\n\t\tfields[i] = fmt.Sprintf(\"Response: %v\", v.Response)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryWorkflowResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this QueryWorkflowResponse match the\n// provided QueryWorkflowResponse.\n//\n// This function performs a deep comparison.\nfunc (v *QueryWorkflowResponse) Equals(rhs *QueryWorkflowResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Response == nil && rhs.Response == nil) || (v.Response != nil && rhs.Response != nil && v.Response.Equals(rhs.Response))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryWorkflowResponse.\nfunc (v *QueryWorkflowResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Response != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"response\", v.Response))\n\t}\n\treturn err\n}\n\n// GetResponse returns the value of Response if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowResponse) GetResponse() (o *shared.QueryWorkflowResponse) {\n\tif v != nil && v.Response != nil {\n\t\treturn v.Response\n\t}\n\n\treturn\n}\n\n// IsSetResponse returns true if Response is not nil.\nfunc (v *QueryWorkflowResponse) IsSetResponse() bool {\n\treturn v != nil && v.Response != nil\n}\n\ntype RatelimitUpdateRequest struct {\n\t// impl-specific data.\n\t//\n\t// likely some simple top-level keys and then either:\n\t//   - map<ratelimit-key-string, something>\n\t//   - list<something>\n\t//\n\t// this is a single blob rather than a collection to save on\n\t// repeated serialization of the type name, and to allow impls\n\t// to choose whatever structures are most-convenient for them.\n\tData *shared.Any `json:\"data,omitempty\"`\n}\n\n// ToWire translates a RatelimitUpdateRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RatelimitUpdateRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Data != nil {\n\t\tw, err = v.Data.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Any_Read(w wire.Value) (*shared.Any, error) {\n\tvar v shared.Any\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RatelimitUpdateRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RatelimitUpdateRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RatelimitUpdateRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RatelimitUpdateRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Data, err = _Any_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RatelimitUpdateRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RatelimitUpdateRequest struct could not be encoded.\nfunc (v *RatelimitUpdateRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Data.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Any_Decode(sr stream.Reader) (*shared.Any, error) {\n\tvar v shared.Any\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RatelimitUpdateRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RatelimitUpdateRequest struct could not be generated from the wire\n// representation.\nfunc (v *RatelimitUpdateRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Data, err = _Any_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RatelimitUpdateRequest\n// struct.\nfunc (v *RatelimitUpdateRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RatelimitUpdateRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RatelimitUpdateRequest match the\n// provided RatelimitUpdateRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RatelimitUpdateRequest) Equals(rhs *RatelimitUpdateRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && v.Data.Equals(rhs.Data))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RatelimitUpdateRequest.\nfunc (v *RatelimitUpdateRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", v.Data))\n\t}\n\treturn err\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *RatelimitUpdateRequest) GetData() (o *shared.Any) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *RatelimitUpdateRequest) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\ntype RatelimitUpdateResponse struct {\n\t// impl-specific data.\n\t//\n\t// likely some simple top-level keys and then either:\n\t//   - map<ratelimit-key-string, something>\n\t//   - list<something>\n\t//\n\t// this is a single blob rather than a collection to save on\n\t// repeated serialization of the type name, and to allow impls\n\t// to choose whatever structures are most-convenient for them.\n\tData *shared.Any `json:\"data,omitempty\"`\n}\n\n// ToWire translates a RatelimitUpdateResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RatelimitUpdateResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Data != nil {\n\t\tw, err = v.Data.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RatelimitUpdateResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RatelimitUpdateResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RatelimitUpdateResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RatelimitUpdateResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Data, err = _Any_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RatelimitUpdateResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RatelimitUpdateResponse struct could not be encoded.\nfunc (v *RatelimitUpdateResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Data.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RatelimitUpdateResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RatelimitUpdateResponse struct could not be generated from the wire\n// representation.\nfunc (v *RatelimitUpdateResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Data, err = _Any_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RatelimitUpdateResponse\n// struct.\nfunc (v *RatelimitUpdateResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RatelimitUpdateResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RatelimitUpdateResponse match the\n// provided RatelimitUpdateResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RatelimitUpdateResponse) Equals(rhs *RatelimitUpdateResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && v.Data.Equals(rhs.Data))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RatelimitUpdateResponse.\nfunc (v *RatelimitUpdateResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", v.Data))\n\t}\n\treturn err\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *RatelimitUpdateResponse) GetData() (o *shared.Any) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *RatelimitUpdateResponse) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\ntype ReapplyEventsRequest struct {\n\tDomainUUID *string                      `json:\"domainUUID,omitempty\"`\n\tRequest    *shared.ReapplyEventsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a ReapplyEventsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReapplyEventsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReapplyEventsRequest_Read(w wire.Value) (*shared.ReapplyEventsRequest, error) {\n\tvar v shared.ReapplyEventsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ReapplyEventsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReapplyEventsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReapplyEventsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReapplyEventsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ReapplyEventsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReapplyEventsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReapplyEventsRequest struct could not be encoded.\nfunc (v *ReapplyEventsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReapplyEventsRequest_Decode(sr stream.Reader) (*shared.ReapplyEventsRequest, error) {\n\tvar v shared.ReapplyEventsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ReapplyEventsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReapplyEventsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ReapplyEventsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ReapplyEventsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReapplyEventsRequest\n// struct.\nfunc (v *ReapplyEventsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReapplyEventsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReapplyEventsRequest match the\n// provided ReapplyEventsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ReapplyEventsRequest) Equals(rhs *ReapplyEventsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReapplyEventsRequest.\nfunc (v *ReapplyEventsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ReapplyEventsRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ReapplyEventsRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *ReapplyEventsRequest) GetRequest() (o *shared.ReapplyEventsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *ReapplyEventsRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype RecordActivityTaskHeartbeatRequest struct {\n\tDomainUUID       *string                                    `json:\"domainUUID,omitempty\"`\n\tHeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest `json:\"heartbeatRequest,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskHeartbeatRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskHeartbeatRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\tw, err = v.HeartbeatRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_Read(w wire.Value) (*shared.RecordActivityTaskHeartbeatRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RecordActivityTaskHeartbeatRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskHeartbeatRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskHeartbeatRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskHeartbeatRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskHeartbeatRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatRequest struct could not be encoded.\nfunc (v *RecordActivityTaskHeartbeatRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HeartbeatRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_Decode(sr stream.Reader) (*shared.RecordActivityTaskHeartbeatRequest, error) {\n\tvar v shared.RecordActivityTaskHeartbeatRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RecordActivityTaskHeartbeatRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskHeartbeatRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskHeartbeatRequest\n// struct.\nfunc (v *RecordActivityTaskHeartbeatRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatRequest: %v\", v.HeartbeatRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskHeartbeatRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskHeartbeatRequest match the\n// provided RecordActivityTaskHeartbeatRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskHeartbeatRequest) Equals(rhs *RecordActivityTaskHeartbeatRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatRequest == nil && rhs.HeartbeatRequest == nil) || (v.HeartbeatRequest != nil && rhs.HeartbeatRequest != nil && v.HeartbeatRequest.Equals(rhs.HeartbeatRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskHeartbeatRequest.\nfunc (v *RecordActivityTaskHeartbeatRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"heartbeatRequest\", v.HeartbeatRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RecordActivityTaskHeartbeatRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetHeartbeatRequest returns the value of HeartbeatRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatRequest) GetHeartbeatRequest() (o *shared.RecordActivityTaskHeartbeatRequest) {\n\tif v != nil && v.HeartbeatRequest != nil {\n\t\treturn v.HeartbeatRequest\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatRequest returns true if HeartbeatRequest is not nil.\nfunc (v *RecordActivityTaskHeartbeatRequest) IsSetHeartbeatRequest() bool {\n\treturn v != nil && v.HeartbeatRequest != nil\n}\n\ntype RecordActivityTaskStartedRequest struct {\n\tDomainUUID        *string                            `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *shared.WorkflowExecution          `json:\"workflowExecution,omitempty\"`\n\tScheduleId        *int64                             `json:\"scheduleId,omitempty\"`\n\tTaskId            *int64                             `json:\"taskId,omitempty\"`\n\tRequestId         *string                            `json:\"requestId,omitempty\"`\n\tPollRequest       *shared.PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskStartedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskStartedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 45, Value: w}\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskRequest_Read(w wire.Value) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RecordActivityTaskStartedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskStartedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskStartedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskStartedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 45:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskStartedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskStartedRequest struct could not be encoded.\nfunc (v *RecordActivityTaskStartedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 45, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskRequest_Decode(sr stream.Reader) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RecordActivityTaskStartedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskStartedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskStartedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 45 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskStartedRequest\n// struct.\nfunc (v *RecordActivityTaskStartedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleId: %v\", *(v.ScheduleId))\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskId: %v\", *(v.TaskId))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskStartedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskStartedRequest match the\n// provided RecordActivityTaskStartedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskStartedRequest) Equals(rhs *RecordActivityTaskStartedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleId, rhs.ScheduleId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskId, rhs.TaskId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskStartedRequest.\nfunc (v *RecordActivityTaskStartedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.ScheduleId != nil {\n\t\tenc.AddInt64(\"scheduleId\", *v.ScheduleId)\n\t}\n\tif v.TaskId != nil {\n\t\tenc.AddInt64(\"taskId\", *v.TaskId)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetScheduleId returns the value of ScheduleId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetScheduleId() (o int64) {\n\tif v != nil && v.ScheduleId != nil {\n\t\treturn *v.ScheduleId\n\t}\n\n\treturn\n}\n\n// IsSetScheduleId returns true if ScheduleId is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetScheduleId() bool {\n\treturn v != nil && v.ScheduleId != nil\n}\n\n// GetTaskId returns the value of TaskId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetTaskId() (o int64) {\n\tif v != nil && v.TaskId != nil {\n\t\treturn *v.TaskId\n\t}\n\n\treturn\n}\n\n// IsSetTaskId returns true if TaskId is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetTaskId() bool {\n\treturn v != nil && v.TaskId != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedRequest) GetPollRequest() (o *shared.PollForActivityTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *RecordActivityTaskStartedRequest) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\ntype RecordActivityTaskStartedResponse struct {\n\tScheduledEvent                  *shared.HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedTimestamp                *int64               `json:\"startedTimestamp,omitempty\"`\n\tAttempt                         *int64               `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64               `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte               `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *shared.WorkflowType `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  *string              `json:\"workflowDomain,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskStartedResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskStartedResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEvent != nil {\n\t\tw, err = v.ScheduledEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestampOfThisAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.HeartbeatDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryEvent_Read(w wire.Value) (*shared.HistoryEvent, error) {\n\tvar v shared.HistoryEvent\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RecordActivityTaskStartedResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskStartedResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskStartedResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskStartedResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ScheduledEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.HeartbeatDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskStartedResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskStartedResponse struct could not be encoded.\nfunc (v *RecordActivityTaskStartedResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ScheduledEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestampOfThisAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.HeartbeatDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryEvent_Decode(sr stream.Reader) (*shared.HistoryEvent, error) {\n\tvar v shared.HistoryEvent\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RecordActivityTaskStartedResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskStartedResponse struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskStartedResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ScheduledEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.HeartbeatDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskStartedResponse\n// struct.\nfunc (v *RecordActivityTaskStartedResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.ScheduledEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEvent: %v\", v.ScheduledEvent)\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestampOfThisAttempt: %v\", *(v.ScheduledTimestampOfThisAttempt))\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatDetails: %v\", v.HeartbeatDetails)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowDomain: %v\", *(v.WorkflowDomain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskStartedResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskStartedResponse match the\n// provided RecordActivityTaskStartedResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskStartedResponse) Equals(rhs *RecordActivityTaskStartedResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ScheduledEvent == nil && rhs.ScheduledEvent == nil) || (v.ScheduledEvent != nil && rhs.ScheduledEvent != nil && v.ScheduledEvent.Equals(rhs.ScheduledEvent))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestampOfThisAttempt, rhs.ScheduledTimestampOfThisAttempt) {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatDetails == nil && rhs.HeartbeatDetails == nil) || (v.HeartbeatDetails != nil && rhs.HeartbeatDetails != nil && bytes.Equal(v.HeartbeatDetails, rhs.HeartbeatDetails))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowDomain, rhs.WorkflowDomain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskStartedResponse.\nfunc (v *RecordActivityTaskStartedResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"scheduledEvent\", v.ScheduledEvent))\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tenc.AddInt64(\"scheduledTimestampOfThisAttempt\", *v.ScheduledTimestampOfThisAttempt)\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tenc.AddString(\"heartbeatDetails\", base64.StdEncoding.EncodeToString(v.HeartbeatDetails))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tenc.AddString(\"workflowDomain\", *v.WorkflowDomain)\n\t}\n\treturn err\n}\n\n// GetScheduledEvent returns the value of ScheduledEvent if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetScheduledEvent() (o *shared.HistoryEvent) {\n\tif v != nil && v.ScheduledEvent != nil {\n\t\treturn v.ScheduledEvent\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEvent returns true if ScheduledEvent is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetScheduledEvent() bool {\n\treturn v != nil && v.ScheduledEvent != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetScheduledTimestampOfThisAttempt returns the value of ScheduledTimestampOfThisAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetScheduledTimestampOfThisAttempt() (o int64) {\n\tif v != nil && v.ScheduledTimestampOfThisAttempt != nil {\n\t\treturn *v.ScheduledTimestampOfThisAttempt\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestampOfThisAttempt returns true if ScheduledTimestampOfThisAttempt is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetScheduledTimestampOfThisAttempt() bool {\n\treturn v != nil && v.ScheduledTimestampOfThisAttempt != nil\n}\n\n// GetHeartbeatDetails returns the value of HeartbeatDetails if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetHeartbeatDetails() (o []byte) {\n\tif v != nil && v.HeartbeatDetails != nil {\n\t\treturn v.HeartbeatDetails\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatDetails returns true if HeartbeatDetails is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetHeartbeatDetails() bool {\n\treturn v != nil && v.HeartbeatDetails != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetWorkflowDomain returns the value of WorkflowDomain if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskStartedResponse) GetWorkflowDomain() (o string) {\n\tif v != nil && v.WorkflowDomain != nil {\n\t\treturn *v.WorkflowDomain\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowDomain returns true if WorkflowDomain is not nil.\nfunc (v *RecordActivityTaskStartedResponse) IsSetWorkflowDomain() bool {\n\treturn v != nil && v.WorkflowDomain != nil\n}\n\n// RecordChildExecutionCompletedRequest is used for reporting the completion of child execution to parent workflow\n// execution which started it.  When a child execution is completed it creates this request and calls the\n// RecordChildExecutionCompleted API with the workflowExecution of parent.  It also sets the completedExecution of the\n// child as it could potentially be different than the ChildExecutionStartedEvent of parent in the situation when\n// child creates multiple runs through ContinueAsNew before finally completing.\ntype RecordChildExecutionCompletedRequest struct {\n\tDomainUUID         *string                   `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution  *shared.WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tInitiatedId        *int64                    `json:\"initiatedId,omitempty\"`\n\tCompletedExecution *shared.WorkflowExecution `json:\"completedExecution,omitempty\"`\n\tCompletionEvent    *shared.HistoryEvent      `json:\"completionEvent,omitempty\"`\n\tStartedId          *int64                    `json:\"startedId,omitempty\"`\n}\n\n// ToWire translates a RecordChildExecutionCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordChildExecutionCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CompletedExecution != nil {\n\t\tw, err = v.CompletedExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tw, err = v.CompletionEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RecordChildExecutionCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordChildExecutionCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordChildExecutionCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordChildExecutionCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompletedExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompletionEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordChildExecutionCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordChildExecutionCompletedRequest struct could not be encoded.\nfunc (v *RecordChildExecutionCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletedExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletedExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletionEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletionEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RecordChildExecutionCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordChildExecutionCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordChildExecutionCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.CompletedExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.CompletionEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordChildExecutionCompletedRequest\n// struct.\nfunc (v *RecordChildExecutionCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.InitiatedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedId: %v\", *(v.InitiatedId))\n\t\ti++\n\t}\n\tif v.CompletedExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletedExecution: %v\", v.CompletedExecution)\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionEvent: %v\", v.CompletionEvent)\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedId: %v\", *(v.StartedId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordChildExecutionCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordChildExecutionCompletedRequest match the\n// provided RecordChildExecutionCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordChildExecutionCompletedRequest) Equals(rhs *RecordChildExecutionCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedId, rhs.InitiatedId) {\n\t\treturn false\n\t}\n\tif !((v.CompletedExecution == nil && rhs.CompletedExecution == nil) || (v.CompletedExecution != nil && rhs.CompletedExecution != nil && v.CompletedExecution.Equals(rhs.CompletedExecution))) {\n\t\treturn false\n\t}\n\tif !((v.CompletionEvent == nil && rhs.CompletionEvent == nil) || (v.CompletionEvent != nil && rhs.CompletionEvent != nil && v.CompletionEvent.Equals(rhs.CompletionEvent))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedId, rhs.StartedId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordChildExecutionCompletedRequest.\nfunc (v *RecordChildExecutionCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.InitiatedId != nil {\n\t\tenc.AddInt64(\"initiatedId\", *v.InitiatedId)\n\t}\n\tif v.CompletedExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completedExecution\", v.CompletedExecution))\n\t}\n\tif v.CompletionEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completionEvent\", v.CompletionEvent))\n\t}\n\tif v.StartedId != nil {\n\t\tenc.AddInt64(\"startedId\", *v.StartedId)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetInitiatedId returns the value of InitiatedId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetInitiatedId() (o int64) {\n\tif v != nil && v.InitiatedId != nil {\n\t\treturn *v.InitiatedId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedId returns true if InitiatedId is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetInitiatedId() bool {\n\treturn v != nil && v.InitiatedId != nil\n}\n\n// GetCompletedExecution returns the value of CompletedExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetCompletedExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.CompletedExecution != nil {\n\t\treturn v.CompletedExecution\n\t}\n\n\treturn\n}\n\n// IsSetCompletedExecution returns true if CompletedExecution is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetCompletedExecution() bool {\n\treturn v != nil && v.CompletedExecution != nil\n}\n\n// GetCompletionEvent returns the value of CompletionEvent if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetCompletionEvent() (o *shared.HistoryEvent) {\n\tif v != nil && v.CompletionEvent != nil {\n\t\treturn v.CompletionEvent\n\t}\n\n\treturn\n}\n\n// IsSetCompletionEvent returns true if CompletionEvent is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetCompletionEvent() bool {\n\treturn v != nil && v.CompletionEvent != nil\n}\n\n// GetStartedId returns the value of StartedId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordChildExecutionCompletedRequest) GetStartedId() (o int64) {\n\tif v != nil && v.StartedId != nil {\n\t\treturn *v.StartedId\n\t}\n\n\treturn\n}\n\n// IsSetStartedId returns true if StartedId is not nil.\nfunc (v *RecordChildExecutionCompletedRequest) IsSetStartedId() bool {\n\treturn v != nil && v.StartedId != nil\n}\n\ntype RecordDecisionTaskStartedRequest struct {\n\tDomainUUID        *string                            `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *shared.WorkflowExecution          `json:\"workflowExecution,omitempty\"`\n\tScheduleId        *int64                             `json:\"scheduleId,omitempty\"`\n\tTaskId            *int64                             `json:\"taskId,omitempty\"`\n\tRequestId         *string                            `json:\"requestId,omitempty\"`\n\tPollRequest       *shared.PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a RecordDecisionTaskStartedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordDecisionTaskStartedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 45, Value: w}\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskRequest_Read(w wire.Value) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RecordDecisionTaskStartedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordDecisionTaskStartedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordDecisionTaskStartedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordDecisionTaskStartedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 45:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordDecisionTaskStartedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordDecisionTaskStartedRequest struct could not be encoded.\nfunc (v *RecordDecisionTaskStartedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 45, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskRequest_Decode(sr stream.Reader) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RecordDecisionTaskStartedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordDecisionTaskStartedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordDecisionTaskStartedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 45 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordDecisionTaskStartedRequest\n// struct.\nfunc (v *RecordDecisionTaskStartedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleId: %v\", *(v.ScheduleId))\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskId: %v\", *(v.TaskId))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordDecisionTaskStartedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordDecisionTaskStartedRequest match the\n// provided RecordDecisionTaskStartedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordDecisionTaskStartedRequest) Equals(rhs *RecordDecisionTaskStartedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleId, rhs.ScheduleId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskId, rhs.TaskId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordDecisionTaskStartedRequest.\nfunc (v *RecordDecisionTaskStartedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.ScheduleId != nil {\n\t\tenc.AddInt64(\"scheduleId\", *v.ScheduleId)\n\t}\n\tif v.TaskId != nil {\n\t\tenc.AddInt64(\"taskId\", *v.TaskId)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetScheduleId returns the value of ScheduleId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetScheduleId() (o int64) {\n\tif v != nil && v.ScheduleId != nil {\n\t\treturn *v.ScheduleId\n\t}\n\n\treturn\n}\n\n// IsSetScheduleId returns true if ScheduleId is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetScheduleId() bool {\n\treturn v != nil && v.ScheduleId != nil\n}\n\n// GetTaskId returns the value of TaskId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetTaskId() (o int64) {\n\tif v != nil && v.TaskId != nil {\n\t\treturn *v.TaskId\n\t}\n\n\treturn\n}\n\n// IsSetTaskId returns true if TaskId is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetTaskId() bool {\n\treturn v != nil && v.TaskId != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedRequest) GetPollRequest() (o *shared.PollForDecisionTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *RecordDecisionTaskStartedRequest) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\ntype RecordDecisionTaskStartedResponse struct {\n\tWorkflowType              *shared.WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventId    *int64                           `json:\"previousStartedEventId,omitempty\"`\n\tScheduledEventId          *int64                           `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId            *int64                           `json:\"startedEventId,omitempty\"`\n\tNextEventId               *int64                           `json:\"nextEventId,omitempty\"`\n\tAttempt                   *int64                           `json:\"attempt,omitempty\"`\n\tStickyExecutionEnabled    *bool                            `json:\"stickyExecutionEnabled,omitempty\"`\n\tDecisionInfo              *shared.TransientDecisionInfo    `json:\"decisionInfo,omitempty\"`\n\tWorkflowExecutionTaskList *shared.TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tEventStoreVersion         *int32                           `json:\"eventStoreVersion,omitempty\"`\n\tBranchToken               []byte                           `json:\"branchToken,omitempty\"`\n\tScheduledTimestamp        *int64                           `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                           `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*shared.WorkflowQuery `json:\"queries,omitempty\"`\n\tHistorySize               *int64                           `json:\"historySize,omitempty\"`\n}\n\ntype _Map_String_WorkflowQuery_MapItemList map[string]*shared.WorkflowQuery\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) Close() {}\n\n// ToWire translates a RecordDecisionTaskStartedResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordDecisionTaskStartedResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousStartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.StickyExecutionEnabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionInfo != nil {\n\t\tw, err = v.DecisionInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tw, err = v.WorkflowExecutionTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.EventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.BranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_WorkflowQuery_MapItemList(v.Queries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tw, err = wire.NewValueI64(*(v.HistorySize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TransientDecisionInfo_Read(w wire.Value) (*shared.TransientDecisionInfo, error) {\n\tvar v shared.TransientDecisionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowQuery_Read(w wire.Value) (*shared.WorkflowQuery, error) {\n\tvar v shared.WorkflowQuery\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Read(m wire.MapItemList) (map[string]*shared.WorkflowQuery, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*shared.WorkflowQuery, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RecordDecisionTaskStartedResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordDecisionTaskStartedResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordDecisionTaskStartedResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordDecisionTaskStartedResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousStartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.StickyExecutionEnabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionInfo, err = _TransientDecisionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.EventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.HistorySize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_WorkflowQuery_Encode(val map[string]*shared.WorkflowQuery, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a RecordDecisionTaskStartedResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordDecisionTaskStartedResponse struct could not be encoded.\nfunc (v *RecordDecisionTaskStartedResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousStartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousStartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyExecutionEnabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.StickyExecutionEnabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.EventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Queries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_WorkflowQuery_Encode(v.Queries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistorySize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.HistorySize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TransientDecisionInfo_Decode(sr stream.Reader) (*shared.TransientDecisionInfo, error) {\n\tvar v shared.TransientDecisionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowQuery_Decode(sr stream.Reader) (*shared.WorkflowQuery, error) {\n\tvar v shared.WorkflowQuery\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Decode(sr stream.Reader) (map[string]*shared.WorkflowQuery, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*shared.WorkflowQuery, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RecordDecisionTaskStartedResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordDecisionTaskStartedResponse struct could not be generated from the wire\n// representation.\nfunc (v *RecordDecisionTaskStartedResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousStartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.StickyExecutionEnabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionInfo, err = _TransientDecisionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.EventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tv.BranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TMap:\n\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.HistorySize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordDecisionTaskStartedResponse\n// struct.\nfunc (v *RecordDecisionTaskStartedResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousStartedEventId: %v\", *(v.PreviousStartedEventId))\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventId: %v\", *(v.NextEventId))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyExecutionEnabled: %v\", *(v.StickyExecutionEnabled))\n\t\ti++\n\t}\n\tif v.DecisionInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionInfo: %v\", v.DecisionInfo)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionTaskList: %v\", v.WorkflowExecutionTaskList)\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventStoreVersion: %v\", *(v.EventStoreVersion))\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchToken: %v\", v.BranchToken)\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Queries: %v\", v.Queries)\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistorySize: %v\", *(v.HistorySize))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordDecisionTaskStartedResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_WorkflowQuery_Equals(lhs, rhs map[string]*shared.WorkflowQuery) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this RecordDecisionTaskStartedResponse match the\n// provided RecordDecisionTaskStartedResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RecordDecisionTaskStartedResponse) Equals(rhs *RecordDecisionTaskStartedResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousStartedEventId, rhs.PreviousStartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventId, rhs.NextEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.StickyExecutionEnabled, rhs.StickyExecutionEnabled) {\n\t\treturn false\n\t}\n\tif !((v.DecisionInfo == nil && rhs.DecisionInfo == nil) || (v.DecisionInfo != nil && rhs.DecisionInfo != nil && v.DecisionInfo.Equals(rhs.DecisionInfo))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionTaskList == nil && rhs.WorkflowExecutionTaskList == nil) || (v.WorkflowExecutionTaskList != nil && rhs.WorkflowExecutionTaskList != nil && v.WorkflowExecutionTaskList.Equals(rhs.WorkflowExecutionTaskList))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.EventStoreVersion, rhs.EventStoreVersion) {\n\t\treturn false\n\t}\n\tif !((v.BranchToken == nil && rhs.BranchToken == nil) || (v.BranchToken != nil && rhs.BranchToken != nil && bytes.Equal(v.BranchToken, rhs.BranchToken))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !((v.Queries == nil && rhs.Queries == nil) || (v.Queries != nil && rhs.Queries != nil && _Map_String_WorkflowQuery_Equals(v.Queries, rhs.Queries))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.HistorySize, rhs.HistorySize) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_WorkflowQuery_Zapper map[string]*shared.WorkflowQuery\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WorkflowQuery_Zapper.\nfunc (m _Map_String_WorkflowQuery_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordDecisionTaskStartedResponse.\nfunc (v *RecordDecisionTaskStartedResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tenc.AddInt64(\"previousStartedEventId\", *v.PreviousStartedEventId)\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.NextEventId != nil {\n\t\tenc.AddInt64(\"nextEventId\", *v.NextEventId)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tenc.AddBool(\"stickyExecutionEnabled\", *v.StickyExecutionEnabled)\n\t}\n\tif v.DecisionInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionInfo\", v.DecisionInfo))\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"WorkflowExecutionTaskList\", v.WorkflowExecutionTaskList))\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tenc.AddInt32(\"eventStoreVersion\", *v.EventStoreVersion)\n\t}\n\tif v.BranchToken != nil {\n\t\tenc.AddString(\"branchToken\", base64.StdEncoding.EncodeToString(v.BranchToken))\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Queries != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queries\", (_Map_String_WorkflowQuery_Zapper)(v.Queries)))\n\t}\n\tif v.HistorySize != nil {\n\t\tenc.AddInt64(\"historySize\", *v.HistorySize)\n\t}\n\treturn err\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetPreviousStartedEventId returns the value of PreviousStartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetPreviousStartedEventId() (o int64) {\n\tif v != nil && v.PreviousStartedEventId != nil {\n\t\treturn *v.PreviousStartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetPreviousStartedEventId returns true if PreviousStartedEventId is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetPreviousStartedEventId() bool {\n\treturn v != nil && v.PreviousStartedEventId != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetNextEventId returns the value of NextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetNextEventId() (o int64) {\n\tif v != nil && v.NextEventId != nil {\n\t\treturn *v.NextEventId\n\t}\n\n\treturn\n}\n\n// IsSetNextEventId returns true if NextEventId is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetNextEventId() bool {\n\treturn v != nil && v.NextEventId != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetStickyExecutionEnabled returns the value of StickyExecutionEnabled if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetStickyExecutionEnabled() (o bool) {\n\tif v != nil && v.StickyExecutionEnabled != nil {\n\t\treturn *v.StickyExecutionEnabled\n\t}\n\n\treturn\n}\n\n// IsSetStickyExecutionEnabled returns true if StickyExecutionEnabled is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetStickyExecutionEnabled() bool {\n\treturn v != nil && v.StickyExecutionEnabled != nil\n}\n\n// GetDecisionInfo returns the value of DecisionInfo if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetDecisionInfo() (o *shared.TransientDecisionInfo) {\n\tif v != nil && v.DecisionInfo != nil {\n\t\treturn v.DecisionInfo\n\t}\n\n\treturn\n}\n\n// IsSetDecisionInfo returns true if DecisionInfo is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetDecisionInfo() bool {\n\treturn v != nil && v.DecisionInfo != nil\n}\n\n// GetWorkflowExecutionTaskList returns the value of WorkflowExecutionTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetWorkflowExecutionTaskList() (o *shared.TaskList) {\n\tif v != nil && v.WorkflowExecutionTaskList != nil {\n\t\treturn v.WorkflowExecutionTaskList\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionTaskList returns true if WorkflowExecutionTaskList is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetWorkflowExecutionTaskList() bool {\n\treturn v != nil && v.WorkflowExecutionTaskList != nil\n}\n\n// GetEventStoreVersion returns the value of EventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetEventStoreVersion() (o int32) {\n\tif v != nil && v.EventStoreVersion != nil {\n\t\treturn *v.EventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetEventStoreVersion returns true if EventStoreVersion is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetEventStoreVersion() bool {\n\treturn v != nil && v.EventStoreVersion != nil\n}\n\n// GetBranchToken returns the value of BranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetBranchToken() (o []byte) {\n\tif v != nil && v.BranchToken != nil {\n\t\treturn v.BranchToken\n\t}\n\n\treturn\n}\n\n// IsSetBranchToken returns true if BranchToken is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetBranchToken() bool {\n\treturn v != nil && v.BranchToken != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetQueries returns the value of Queries if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetQueries() (o map[string]*shared.WorkflowQuery) {\n\tif v != nil && v.Queries != nil {\n\t\treturn v.Queries\n\t}\n\n\treturn\n}\n\n// IsSetQueries returns true if Queries is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetQueries() bool {\n\treturn v != nil && v.Queries != nil\n}\n\n// GetHistorySize returns the value of HistorySize if it is set or its\n// zero value if it is unset.\nfunc (v *RecordDecisionTaskStartedResponse) GetHistorySize() (o int64) {\n\tif v != nil && v.HistorySize != nil {\n\t\treturn *v.HistorySize\n\t}\n\n\treturn\n}\n\n// IsSetHistorySize returns true if HistorySize is not nil.\nfunc (v *RecordDecisionTaskStartedResponse) IsSetHistorySize() bool {\n\treturn v != nil && v.HistorySize != nil\n}\n\ntype RefreshWorkflowTasksRequest struct {\n\tDomainUIID *string                             `json:\"domainUIID,omitempty\"`\n\tRequest    *shared.RefreshWorkflowTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a RefreshWorkflowTasksRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RefreshWorkflowTasksRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUIID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUIID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RefreshWorkflowTasksRequest_Read(w wire.Value) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RefreshWorkflowTasksRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RefreshWorkflowTasksRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RefreshWorkflowTasksRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RefreshWorkflowTasksRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUIID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RefreshWorkflowTasksRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RefreshWorkflowTasksRequest struct could not be encoded.\nfunc (v *RefreshWorkflowTasksRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUIID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUIID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RefreshWorkflowTasksRequest_Decode(sr stream.Reader) (*shared.RefreshWorkflowTasksRequest, error) {\n\tvar v shared.RefreshWorkflowTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RefreshWorkflowTasksRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RefreshWorkflowTasksRequest struct could not be generated from the wire\n// representation.\nfunc (v *RefreshWorkflowTasksRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUIID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RefreshWorkflowTasksRequest\n// struct.\nfunc (v *RefreshWorkflowTasksRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUIID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUIID: %v\", *(v.DomainUIID))\n\t\ti++\n\t}\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RefreshWorkflowTasksRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RefreshWorkflowTasksRequest match the\n// provided RefreshWorkflowTasksRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RefreshWorkflowTasksRequest) Equals(rhs *RefreshWorkflowTasksRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUIID, rhs.DomainUIID) {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RefreshWorkflowTasksRequest.\nfunc (v *RefreshWorkflowTasksRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUIID != nil {\n\t\tenc.AddString(\"domainUIID\", *v.DomainUIID)\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetDomainUIID returns the value of DomainUIID if it is set or its\n// zero value if it is unset.\nfunc (v *RefreshWorkflowTasksRequest) GetDomainUIID() (o string) {\n\tif v != nil && v.DomainUIID != nil {\n\t\treturn *v.DomainUIID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUIID returns true if DomainUIID is not nil.\nfunc (v *RefreshWorkflowTasksRequest) IsSetDomainUIID() bool {\n\treturn v != nil && v.DomainUIID != nil\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *RefreshWorkflowTasksRequest) GetRequest() (o *shared.RefreshWorkflowTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *RefreshWorkflowTasksRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype RemoveSignalMutableStateRequest struct {\n\tDomainUUID        *string                   `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *shared.WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tRequestId         *string                   `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a RemoveSignalMutableStateRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RemoveSignalMutableStateRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RemoveSignalMutableStateRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RemoveSignalMutableStateRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RemoveSignalMutableStateRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RemoveSignalMutableStateRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RemoveSignalMutableStateRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RemoveSignalMutableStateRequest struct could not be encoded.\nfunc (v *RemoveSignalMutableStateRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RemoveSignalMutableStateRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RemoveSignalMutableStateRequest struct could not be generated from the wire\n// representation.\nfunc (v *RemoveSignalMutableStateRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RemoveSignalMutableStateRequest\n// struct.\nfunc (v *RemoveSignalMutableStateRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RemoveSignalMutableStateRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RemoveSignalMutableStateRequest match the\n// provided RemoveSignalMutableStateRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RemoveSignalMutableStateRequest) Equals(rhs *RemoveSignalMutableStateRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RemoveSignalMutableStateRequest.\nfunc (v *RemoveSignalMutableStateRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveSignalMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RemoveSignalMutableStateRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveSignalMutableStateRequest) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RemoveSignalMutableStateRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveSignalMutableStateRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *RemoveSignalMutableStateRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype ReplicateEventsV2Request struct {\n\tDomainUUID          *string                      `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution   *shared.WorkflowExecution    `json:\"workflowExecution,omitempty\"`\n\tVersionHistoryItems []*shared.VersionHistoryItem `json:\"versionHistoryItems,omitempty\"`\n\tEvents              *shared.DataBlob             `json:\"events,omitempty\"`\n\tNewRunEvents        *shared.DataBlob             `json:\"newRunEvents,omitempty\"`\n}\n\ntype _List_VersionHistoryItem_ValueList []*shared.VersionHistoryItem\n\nfunc (v _List_VersionHistoryItem_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_VersionHistoryItem_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_VersionHistoryItem_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_VersionHistoryItem_ValueList) Close() {}\n\n// ToWire translates a ReplicateEventsV2Request struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicateEventsV2Request) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\tw, err = wire.NewValueList(_List_VersionHistoryItem_ValueList(v.VersionHistoryItems)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tw, err = v.Events.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunEvents != nil {\n\t\tw, err = v.NewRunEvents.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_VersionHistoryItem_Read(l wire.ValueList) ([]*shared.VersionHistoryItem, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*shared.VersionHistoryItem, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _VersionHistoryItem_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _DataBlob_Read(w wire.Value) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ReplicateEventsV2Request struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicateEventsV2Request struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicateEventsV2Request\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicateEventsV2Request) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.VersionHistoryItems, err = _List_VersionHistoryItem_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Events, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.NewRunEvents, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_VersionHistoryItem_Encode(val []*shared.VersionHistoryItem, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ReplicateEventsV2Request struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicateEventsV2Request struct could not be encoded.\nfunc (v *ReplicateEventsV2Request) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistoryItems != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_VersionHistoryItem_Encode(v.VersionHistoryItems, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Events != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Events.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunEvents != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.NewRunEvents.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_VersionHistoryItem_Decode(sr stream.Reader) ([]*shared.VersionHistoryItem, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*shared.VersionHistoryItem, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _VersionHistoryItem_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _DataBlob_Decode(sr stream.Reader) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ReplicateEventsV2Request struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicateEventsV2Request struct could not be generated from the wire\n// representation.\nfunc (v *ReplicateEventsV2Request) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.VersionHistoryItems, err = _List_VersionHistoryItem_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.Events, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.NewRunEvents, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicateEventsV2Request\n// struct.\nfunc (v *ReplicateEventsV2Request) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistoryItems: %v\", v.VersionHistoryItems)\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tfields[i] = fmt.Sprintf(\"Events: %v\", v.Events)\n\t\ti++\n\t}\n\tif v.NewRunEvents != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunEvents: %v\", v.NewRunEvents)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicateEventsV2Request{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_VersionHistoryItem_Equals(lhs, rhs []*shared.VersionHistoryItem) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ReplicateEventsV2Request match the\n// provided ReplicateEventsV2Request.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicateEventsV2Request) Equals(rhs *ReplicateEventsV2Request) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistoryItems == nil && rhs.VersionHistoryItems == nil) || (v.VersionHistoryItems != nil && rhs.VersionHistoryItems != nil && _List_VersionHistoryItem_Equals(v.VersionHistoryItems, rhs.VersionHistoryItems))) {\n\t\treturn false\n\t}\n\tif !((v.Events == nil && rhs.Events == nil) || (v.Events != nil && rhs.Events != nil && v.Events.Equals(rhs.Events))) {\n\t\treturn false\n\t}\n\tif !((v.NewRunEvents == nil && rhs.NewRunEvents == nil) || (v.NewRunEvents != nil && rhs.NewRunEvents != nil && v.NewRunEvents.Equals(rhs.NewRunEvents))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_VersionHistoryItem_Zapper []*shared.VersionHistoryItem\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_VersionHistoryItem_Zapper.\nfunc (l _List_VersionHistoryItem_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicateEventsV2Request.\nfunc (v *ReplicateEventsV2Request) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"versionHistoryItems\", (_List_VersionHistoryItem_Zapper)(v.VersionHistoryItems)))\n\t}\n\tif v.Events != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"events\", v.Events))\n\t}\n\tif v.NewRunEvents != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"newRunEvents\", v.NewRunEvents))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicateEventsV2Request) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ReplicateEventsV2Request) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicateEventsV2Request) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ReplicateEventsV2Request) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetVersionHistoryItems returns the value of VersionHistoryItems if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicateEventsV2Request) GetVersionHistoryItems() (o []*shared.VersionHistoryItem) {\n\tif v != nil && v.VersionHistoryItems != nil {\n\t\treturn v.VersionHistoryItems\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistoryItems returns true if VersionHistoryItems is not nil.\nfunc (v *ReplicateEventsV2Request) IsSetVersionHistoryItems() bool {\n\treturn v != nil && v.VersionHistoryItems != nil\n}\n\n// GetEvents returns the value of Events if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicateEventsV2Request) GetEvents() (o *shared.DataBlob) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\n\treturn\n}\n\n// IsSetEvents returns true if Events is not nil.\nfunc (v *ReplicateEventsV2Request) IsSetEvents() bool {\n\treturn v != nil && v.Events != nil\n}\n\n// GetNewRunEvents returns the value of NewRunEvents if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicateEventsV2Request) GetNewRunEvents() (o *shared.DataBlob) {\n\tif v != nil && v.NewRunEvents != nil {\n\t\treturn v.NewRunEvents\n\t}\n\n\treturn\n}\n\n// IsSetNewRunEvents returns true if NewRunEvents is not nil.\nfunc (v *ReplicateEventsV2Request) IsSetNewRunEvents() bool {\n\treturn v != nil && v.NewRunEvents != nil\n}\n\ntype RequestCancelWorkflowExecutionRequest struct {\n\tDomainUUID                *string                                       `json:\"domainUUID,omitempty\"`\n\tCancelRequest             *shared.RequestCancelWorkflowExecutionRequest `json:\"cancelRequest,omitempty\"`\n\tExternalInitiatedEventId  *int64                                        `json:\"externalInitiatedEventId,omitempty\"`\n\tExternalWorkflowExecution *shared.WorkflowExecution                     `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         *bool                                         `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a RequestCancelWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequest != nil {\n\t\tw, err = v.CancelRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExternalInitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tw, err = v.ExternalWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_Read(w wire.Value) (*shared.RequestCancelWorkflowExecutionRequest, error) {\n\tvar v shared.RequestCancelWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RequestCancelWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExternalInitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelWorkflowExecutionRequest struct could not be encoded.\nfunc (v *RequestCancelWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalInitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExternalInitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.RequestCancelWorkflowExecutionRequest, error) {\n\tvar v shared.RequestCancelWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RequestCancelWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExternalInitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelWorkflowExecutionRequest\n// struct.\nfunc (v *RequestCancelWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.CancelRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequest: %v\", v.CancelRequest)\n\t\ti++\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalInitiatedEventId: %v\", *(v.ExternalInitiatedEventId))\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecution: %v\", v.ExternalWorkflowExecution)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelWorkflowExecutionRequest match the\n// provided RequestCancelWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelWorkflowExecutionRequest) Equals(rhs *RequestCancelWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.CancelRequest == nil && rhs.CancelRequest == nil) || (v.CancelRequest != nil && rhs.CancelRequest != nil && v.CancelRequest.Equals(rhs.CancelRequest))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExternalInitiatedEventId, rhs.ExternalInitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecution == nil && rhs.ExternalWorkflowExecution == nil) || (v.ExternalWorkflowExecution != nil && rhs.ExternalWorkflowExecution != nil && v.ExternalWorkflowExecution.Equals(rhs.ExternalWorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelWorkflowExecutionRequest.\nfunc (v *RequestCancelWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.CancelRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelRequest\", v.CancelRequest))\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tenc.AddInt64(\"externalInitiatedEventId\", *v.ExternalInitiatedEventId)\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecution\", v.ExternalWorkflowExecution))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetCancelRequest returns the value of CancelRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetCancelRequest() (o *shared.RequestCancelWorkflowExecutionRequest) {\n\tif v != nil && v.CancelRequest != nil {\n\t\treturn v.CancelRequest\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequest returns true if CancelRequest is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetCancelRequest() bool {\n\treturn v != nil && v.CancelRequest != nil\n}\n\n// GetExternalInitiatedEventId returns the value of ExternalInitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetExternalInitiatedEventId() (o int64) {\n\tif v != nil && v.ExternalInitiatedEventId != nil {\n\t\treturn *v.ExternalInitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetExternalInitiatedEventId returns true if ExternalInitiatedEventId is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetExternalInitiatedEventId() bool {\n\treturn v != nil && v.ExternalInitiatedEventId != nil\n}\n\n// GetExternalWorkflowExecution returns the value of ExternalWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetExternalWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecution returns true if ExternalWorkflowExecution is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetExternalWorkflowExecution() bool {\n\treturn v != nil && v.ExternalWorkflowExecution != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype ResetStickyTaskListRequest struct {\n\tDomainUUID *string                   `json:\"domainUUID,omitempty\"`\n\tExecution  *shared.WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a ResetStickyTaskListRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetStickyTaskListRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetStickyTaskListRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetStickyTaskListRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetStickyTaskListRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetStickyTaskListRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetStickyTaskListRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetStickyTaskListRequest struct could not be encoded.\nfunc (v *ResetStickyTaskListRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetStickyTaskListRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetStickyTaskListRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResetStickyTaskListRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetStickyTaskListRequest\n// struct.\nfunc (v *ResetStickyTaskListRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetStickyTaskListRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetStickyTaskListRequest match the\n// provided ResetStickyTaskListRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResetStickyTaskListRequest) Equals(rhs *ResetStickyTaskListRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetStickyTaskListRequest.\nfunc (v *ResetStickyTaskListRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ResetStickyTaskListRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ResetStickyTaskListRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *ResetStickyTaskListRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *ResetStickyTaskListRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype ResetStickyTaskListResponse struct {\n}\n\n// ToWire translates a ResetStickyTaskListResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetStickyTaskListResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetStickyTaskListResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetStickyTaskListResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetStickyTaskListResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetStickyTaskListResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetStickyTaskListResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetStickyTaskListResponse struct could not be encoded.\nfunc (v *ResetStickyTaskListResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetStickyTaskListResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetStickyTaskListResponse struct could not be generated from the wire\n// representation.\nfunc (v *ResetStickyTaskListResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetStickyTaskListResponse\n// struct.\nfunc (v *ResetStickyTaskListResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"ResetStickyTaskListResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetStickyTaskListResponse match the\n// provided ResetStickyTaskListResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ResetStickyTaskListResponse) Equals(rhs *ResetStickyTaskListResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetStickyTaskListResponse.\nfunc (v *ResetStickyTaskListResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype ResetWorkflowExecutionRequest struct {\n\tDomainUUID   *string                               `json:\"domainUUID,omitempty\"`\n\tResetRequest *shared.ResetWorkflowExecutionRequest `json:\"resetRequest,omitempty\"`\n}\n\n// ToWire translates a ResetWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ResetRequest != nil {\n\t\tw, err = v.ResetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetWorkflowExecutionRequest_Read(w wire.Value) (*shared.ResetWorkflowExecutionRequest, error) {\n\tvar v shared.ResetWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ResetWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetWorkflowExecutionRequest struct could not be encoded.\nfunc (v *ResetWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ResetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.ResetWorkflowExecutionRequest, error) {\n\tvar v shared.ResetWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ResetWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResetWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetWorkflowExecutionRequest\n// struct.\nfunc (v *ResetWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.ResetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResetRequest: %v\", v.ResetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetWorkflowExecutionRequest match the\n// provided ResetWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResetWorkflowExecutionRequest) Equals(rhs *ResetWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.ResetRequest == nil && rhs.ResetRequest == nil) || (v.ResetRequest != nil && rhs.ResetRequest != nil && v.ResetRequest.Equals(rhs.ResetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetWorkflowExecutionRequest.\nfunc (v *ResetWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.ResetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resetRequest\", v.ResetRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetResetRequest returns the value of ResetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetResetRequest() (o *shared.ResetWorkflowExecutionRequest) {\n\tif v != nil && v.ResetRequest != nil {\n\t\treturn v.ResetRequest\n\t}\n\n\treturn\n}\n\n// IsSetResetRequest returns true if ResetRequest is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetResetRequest() bool {\n\treturn v != nil && v.ResetRequest != nil\n}\n\ntype RespondActivityTaskCanceledRequest struct {\n\tDomainUUID    *string                                    `json:\"domainUUID,omitempty\"`\n\tCancelRequest *shared.RespondActivityTaskCanceledRequest `json:\"cancelRequest,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCanceledRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCanceledRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequest != nil {\n\t\tw, err = v.CancelRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCanceledRequest_Read(w wire.Value) (*shared.RespondActivityTaskCanceledRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondActivityTaskCanceledRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCanceledRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCanceledRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCanceledRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelRequest, err = _RespondActivityTaskCanceledRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCanceledRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCanceledRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCanceledRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCanceledRequest, error) {\n\tvar v shared.RespondActivityTaskCanceledRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondActivityTaskCanceledRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCanceledRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.CancelRequest, err = _RespondActivityTaskCanceledRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCanceledRequest\n// struct.\nfunc (v *RespondActivityTaskCanceledRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.CancelRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequest: %v\", v.CancelRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCanceledRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCanceledRequest match the\n// provided RespondActivityTaskCanceledRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCanceledRequest) Equals(rhs *RespondActivityTaskCanceledRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.CancelRequest == nil && rhs.CancelRequest == nil) || (v.CancelRequest != nil && rhs.CancelRequest != nil && v.CancelRequest.Equals(rhs.CancelRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCanceledRequest.\nfunc (v *RespondActivityTaskCanceledRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.CancelRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelRequest\", v.CancelRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondActivityTaskCanceledRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetCancelRequest returns the value of CancelRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledRequest) GetCancelRequest() (o *shared.RespondActivityTaskCanceledRequest) {\n\tif v != nil && v.CancelRequest != nil {\n\t\treturn v.CancelRequest\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequest returns true if CancelRequest is not nil.\nfunc (v *RespondActivityTaskCanceledRequest) IsSetCancelRequest() bool {\n\treturn v != nil && v.CancelRequest != nil\n}\n\ntype RespondActivityTaskCompletedRequest struct {\n\tDomainUUID      *string                                     `json:\"domainUUID,omitempty\"`\n\tCompleteRequest *shared.RespondActivityTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCompletedRequest_Read(w wire.Value) (*shared.RespondActivityTaskCompletedRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondActivityTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskCompletedRequest, error) {\n\tvar v shared.RespondActivityTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondActivityTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCompletedRequest\n// struct.\nfunc (v *RespondActivityTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCompletedRequest match the\n// provided RespondActivityTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCompletedRequest) Equals(rhs *RespondActivityTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCompletedRequest.\nfunc (v *RespondActivityTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondActivityTaskCompletedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedRequest) GetCompleteRequest() (o *shared.RespondActivityTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *RespondActivityTaskCompletedRequest) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\ntype RespondActivityTaskFailedRequest struct {\n\tDomainUUID    *string                                  `json:\"domainUUID,omitempty\"`\n\tFailedRequest *shared.RespondActivityTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskFailedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskFailedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailedRequest != nil {\n\t\tw, err = v.FailedRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskFailedRequest_Read(w wire.Value) (*shared.RespondActivityTaskFailedRequest, error) {\n\tvar v shared.RespondActivityTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondActivityTaskFailedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskFailedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskFailedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskFailedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailedRequest, err = _RespondActivityTaskFailedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskFailedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskFailedRequest struct could not be encoded.\nfunc (v *RespondActivityTaskFailedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskFailedRequest_Decode(sr stream.Reader) (*shared.RespondActivityTaskFailedRequest, error) {\n\tvar v shared.RespondActivityTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondActivityTaskFailedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskFailedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskFailedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.FailedRequest, err = _RespondActivityTaskFailedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskFailedRequest\n// struct.\nfunc (v *RespondActivityTaskFailedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.FailedRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedRequest: %v\", v.FailedRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskFailedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskFailedRequest match the\n// provided RespondActivityTaskFailedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskFailedRequest) Equals(rhs *RespondActivityTaskFailedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.FailedRequest == nil && rhs.FailedRequest == nil) || (v.FailedRequest != nil && rhs.FailedRequest != nil && v.FailedRequest.Equals(rhs.FailedRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskFailedRequest.\nfunc (v *RespondActivityTaskFailedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.FailedRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedRequest\", v.FailedRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetFailedRequest returns the value of FailedRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetFailedRequest() (o *shared.RespondActivityTaskFailedRequest) {\n\tif v != nil && v.FailedRequest != nil {\n\t\treturn v.FailedRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailedRequest returns true if FailedRequest is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetFailedRequest() bool {\n\treturn v != nil && v.FailedRequest != nil\n}\n\ntype RespondDecisionTaskCompletedRequest struct {\n\tDomainUUID      *string                                     `json:\"domainUUID,omitempty\"`\n\tCompleteRequest *shared.RespondDecisionTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a RespondDecisionTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskCompletedRequest_Read(w wire.Value) (*shared.RespondDecisionTaskCompletedRequest, error) {\n\tvar v shared.RespondDecisionTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondDecisionTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondDecisionTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondDecisionTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondDecisionTaskCompletedRequest, error) {\n\tvar v shared.RespondDecisionTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondDecisionTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskCompletedRequest\n// struct.\nfunc (v *RespondDecisionTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskCompletedRequest match the\n// provided RespondDecisionTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskCompletedRequest) Equals(rhs *RespondDecisionTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskCompletedRequest.\nfunc (v *RespondDecisionTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetCompleteRequest() (o *shared.RespondDecisionTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\ntype RespondDecisionTaskCompletedResponse struct {\n\tStartedResponse             *RecordDecisionTaskStartedResponse           `json:\"startedResponse,omitempty\"`\n\tActivitiesToDispatchLocally map[string]*shared.ActivityLocalDispatchInfo `json:\"activitiesToDispatchLocally,omitempty\"`\n}\n\ntype _Map_String_ActivityLocalDispatchInfo_MapItemList map[string]*shared.ActivityLocalDispatchInfo\n\nfunc (m _Map_String_ActivityLocalDispatchInfo_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.ActivityLocalDispatchInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_ActivityLocalDispatchInfo_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) Close() {}\n\n// ToWire translates a RespondDecisionTaskCompletedResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskCompletedResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StartedResponse != nil {\n\t\tw, err = v.StartedResponse.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_ActivityLocalDispatchInfo_MapItemList(v.ActivitiesToDispatchLocally)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordDecisionTaskStartedResponse_Read(w wire.Value) (*RecordDecisionTaskStartedResponse, error) {\n\tvar v RecordDecisionTaskStartedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityLocalDispatchInfo_Read(w wire.Value) (*shared.ActivityLocalDispatchInfo, error) {\n\tvar v shared.ActivityLocalDispatchInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Read(m wire.MapItemList) (map[string]*shared.ActivityLocalDispatchInfo, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*shared.ActivityLocalDispatchInfo, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _ActivityLocalDispatchInfo_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RespondDecisionTaskCompletedResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskCompletedResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskCompletedResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskCompletedResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartedResponse, err = _RecordDecisionTaskStartedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActivitiesToDispatchLocally, err = _Map_String_ActivityLocalDispatchInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Encode(val map[string]*shared.ActivityLocalDispatchInfo, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.ActivityLocalDispatchInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a RespondDecisionTaskCompletedResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedResponse struct could not be encoded.\nfunc (v *RespondDecisionTaskCompletedResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StartedResponse != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartedResponse.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_ActivityLocalDispatchInfo_Encode(v.ActivitiesToDispatchLocally, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordDecisionTaskStartedResponse_Decode(sr stream.Reader) (*RecordDecisionTaskStartedResponse, error) {\n\tvar v RecordDecisionTaskStartedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityLocalDispatchInfo_Decode(sr stream.Reader) (*shared.ActivityLocalDispatchInfo, error) {\n\tvar v shared.ActivityLocalDispatchInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Decode(sr stream.Reader) (map[string]*shared.ActivityLocalDispatchInfo, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*shared.ActivityLocalDispatchInfo, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _ActivityLocalDispatchInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RespondDecisionTaskCompletedResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedResponse struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskCompletedResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.StartedResponse, err = _RecordDecisionTaskStartedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TMap:\n\t\t\tv.ActivitiesToDispatchLocally, err = _Map_String_ActivityLocalDispatchInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskCompletedResponse\n// struct.\nfunc (v *RespondDecisionTaskCompletedResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.StartedResponse != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedResponse: %v\", v.StartedResponse)\n\t\ti++\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivitiesToDispatchLocally: %v\", v.ActivitiesToDispatchLocally)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskCompletedResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Equals(lhs, rhs map[string]*shared.ActivityLocalDispatchInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskCompletedResponse match the\n// provided RespondDecisionTaskCompletedResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskCompletedResponse) Equals(rhs *RespondDecisionTaskCompletedResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.StartedResponse == nil && rhs.StartedResponse == nil) || (v.StartedResponse != nil && rhs.StartedResponse != nil && v.StartedResponse.Equals(rhs.StartedResponse))) {\n\t\treturn false\n\t}\n\tif !((v.ActivitiesToDispatchLocally == nil && rhs.ActivitiesToDispatchLocally == nil) || (v.ActivitiesToDispatchLocally != nil && rhs.ActivitiesToDispatchLocally != nil && _Map_String_ActivityLocalDispatchInfo_Equals(v.ActivitiesToDispatchLocally, rhs.ActivitiesToDispatchLocally))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_ActivityLocalDispatchInfo_Zapper map[string]*shared.ActivityLocalDispatchInfo\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_ActivityLocalDispatchInfo_Zapper.\nfunc (m _Map_String_ActivityLocalDispatchInfo_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskCompletedResponse.\nfunc (v *RespondDecisionTaskCompletedResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StartedResponse != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startedResponse\", v.StartedResponse))\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activitiesToDispatchLocally\", (_Map_String_ActivityLocalDispatchInfo_Zapper)(v.ActivitiesToDispatchLocally)))\n\t}\n\treturn err\n}\n\n// GetStartedResponse returns the value of StartedResponse if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedResponse) GetStartedResponse() (o *RecordDecisionTaskStartedResponse) {\n\tif v != nil && v.StartedResponse != nil {\n\t\treturn v.StartedResponse\n\t}\n\n\treturn\n}\n\n// IsSetStartedResponse returns true if StartedResponse is not nil.\nfunc (v *RespondDecisionTaskCompletedResponse) IsSetStartedResponse() bool {\n\treturn v != nil && v.StartedResponse != nil\n}\n\n// GetActivitiesToDispatchLocally returns the value of ActivitiesToDispatchLocally if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedResponse) GetActivitiesToDispatchLocally() (o map[string]*shared.ActivityLocalDispatchInfo) {\n\tif v != nil && v.ActivitiesToDispatchLocally != nil {\n\t\treturn v.ActivitiesToDispatchLocally\n\t}\n\n\treturn\n}\n\n// IsSetActivitiesToDispatchLocally returns true if ActivitiesToDispatchLocally is not nil.\nfunc (v *RespondDecisionTaskCompletedResponse) IsSetActivitiesToDispatchLocally() bool {\n\treturn v != nil && v.ActivitiesToDispatchLocally != nil\n}\n\ntype RespondDecisionTaskFailedRequest struct {\n\tDomainUUID    *string                                  `json:\"domainUUID,omitempty\"`\n\tFailedRequest *shared.RespondDecisionTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// ToWire translates a RespondDecisionTaskFailedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskFailedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailedRequest != nil {\n\t\tw, err = v.FailedRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskFailedRequest_Read(w wire.Value) (*shared.RespondDecisionTaskFailedRequest, error) {\n\tvar v shared.RespondDecisionTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondDecisionTaskFailedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskFailedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskFailedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskFailedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondDecisionTaskFailedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskFailedRequest struct could not be encoded.\nfunc (v *RespondDecisionTaskFailedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskFailedRequest_Decode(sr stream.Reader) (*shared.RespondDecisionTaskFailedRequest, error) {\n\tvar v shared.RespondDecisionTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondDecisionTaskFailedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskFailedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskFailedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskFailedRequest\n// struct.\nfunc (v *RespondDecisionTaskFailedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.FailedRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedRequest: %v\", v.FailedRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskFailedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskFailedRequest match the\n// provided RespondDecisionTaskFailedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskFailedRequest) Equals(rhs *RespondDecisionTaskFailedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.FailedRequest == nil && rhs.FailedRequest == nil) || (v.FailedRequest != nil && rhs.FailedRequest != nil && v.FailedRequest.Equals(rhs.FailedRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskFailedRequest.\nfunc (v *RespondDecisionTaskFailedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.FailedRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedRequest\", v.FailedRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetFailedRequest returns the value of FailedRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetFailedRequest() (o *shared.RespondDecisionTaskFailedRequest) {\n\tif v != nil && v.FailedRequest != nil {\n\t\treturn v.FailedRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailedRequest returns true if FailedRequest is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetFailedRequest() bool {\n\treturn v != nil && v.FailedRequest != nil\n}\n\ntype ScheduleDecisionTaskRequest struct {\n\tDomainUUID        *string                   `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *shared.WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIsFirstDecision   *bool                     `json:\"isFirstDecision,omitempty\"`\n}\n\n// ToWire translates a ScheduleDecisionTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ScheduleDecisionTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.IsFirstDecision != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsFirstDecision)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ScheduleDecisionTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ScheduleDecisionTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ScheduleDecisionTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ScheduleDecisionTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsFirstDecision = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ScheduleDecisionTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ScheduleDecisionTaskRequest struct could not be encoded.\nfunc (v *ScheduleDecisionTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsFirstDecision != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsFirstDecision)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ScheduleDecisionTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ScheduleDecisionTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *ScheduleDecisionTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsFirstDecision = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ScheduleDecisionTaskRequest\n// struct.\nfunc (v *ScheduleDecisionTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.IsFirstDecision != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsFirstDecision: %v\", *(v.IsFirstDecision))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ScheduleDecisionTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ScheduleDecisionTaskRequest match the\n// provided ScheduleDecisionTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ScheduleDecisionTaskRequest) Equals(rhs *ScheduleDecisionTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsFirstDecision, rhs.IsFirstDecision) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ScheduleDecisionTaskRequest.\nfunc (v *ScheduleDecisionTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.IsFirstDecision != nil {\n\t\tenc.AddBool(\"isFirstDecision\", *v.IsFirstDecision)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *ScheduleDecisionTaskRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleDecisionTaskRequest) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ScheduleDecisionTaskRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetIsFirstDecision returns the value of IsFirstDecision if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleDecisionTaskRequest) GetIsFirstDecision() (o bool) {\n\tif v != nil && v.IsFirstDecision != nil {\n\t\treturn *v.IsFirstDecision\n\t}\n\n\treturn\n}\n\n// IsSetIsFirstDecision returns true if IsFirstDecision is not nil.\nfunc (v *ScheduleDecisionTaskRequest) IsSetIsFirstDecision() bool {\n\treturn v != nil && v.IsFirstDecision != nil\n}\n\ntype ShardOwnershipLostError struct {\n\tMessage *string `json:\"message,omitempty\"`\n\tOwner   *string `json:\"owner,omitempty\"`\n}\n\n// ToWire translates a ShardOwnershipLostError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ShardOwnershipLostError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Message != nil {\n\t\tw, err = wire.NewValueString(*(v.Message)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tw, err = wire.NewValueString(*(v.Owner)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ShardOwnershipLostError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ShardOwnershipLostError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ShardOwnershipLostError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ShardOwnershipLostError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Message = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Owner = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ShardOwnershipLostError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ShardOwnershipLostError struct could not be encoded.\nfunc (v *ShardOwnershipLostError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Message != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Message)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Owner != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Owner)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ShardOwnershipLostError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ShardOwnershipLostError struct could not be generated from the wire\n// representation.\nfunc (v *ShardOwnershipLostError) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Message = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Owner = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ShardOwnershipLostError\n// struct.\nfunc (v *ShardOwnershipLostError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Message != nil {\n\t\tfields[i] = fmt.Sprintf(\"Message: %v\", *(v.Message))\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tfields[i] = fmt.Sprintf(\"Owner: %v\", *(v.Owner))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ShardOwnershipLostError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*ShardOwnershipLostError) ErrorName() string {\n\treturn \"ShardOwnershipLostError\"\n}\n\n// Equals returns true if all the fields of this ShardOwnershipLostError match the\n// provided ShardOwnershipLostError.\n//\n// This function performs a deep comparison.\nfunc (v *ShardOwnershipLostError) Equals(rhs *ShardOwnershipLostError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Message, rhs.Message) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Owner, rhs.Owner) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ShardOwnershipLostError.\nfunc (v *ShardOwnershipLostError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Message != nil {\n\t\tenc.AddString(\"message\", *v.Message)\n\t}\n\tif v.Owner != nil {\n\t\tenc.AddString(\"owner\", *v.Owner)\n\t}\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *ShardOwnershipLostError) GetMessage() (o string) {\n\tif v != nil && v.Message != nil {\n\t\treturn *v.Message\n\t}\n\n\treturn\n}\n\n// IsSetMessage returns true if Message is not nil.\nfunc (v *ShardOwnershipLostError) IsSetMessage() bool {\n\treturn v != nil && v.Message != nil\n}\n\n// GetOwner returns the value of Owner if it is set or its\n// zero value if it is unset.\nfunc (v *ShardOwnershipLostError) GetOwner() (o string) {\n\tif v != nil && v.Owner != nil {\n\t\treturn *v.Owner\n\t}\n\n\treturn\n}\n\n// IsSetOwner returns true if Owner is not nil.\nfunc (v *ShardOwnershipLostError) IsSetOwner() bool {\n\treturn v != nil && v.Owner != nil\n}\n\nfunc (v *ShardOwnershipLostError) Error() string {\n\treturn v.String()\n}\n\ntype SignalWithStartWorkflowExecutionRequest struct {\n\tDomainUUID             *string                                         `json:\"domainUUID,omitempty\"`\n\tSignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest `json:\"signalWithStartRequest,omitempty\"`\n\tPartitionConfig        map[string]string                               `json:\"partitionConfig,omitempty\"`\n}\n\ntype _Map_String_String_MapItemList map[string]string\n\nfunc (m _Map_String_String_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueString(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_String_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_String_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) Close() {}\n\n// ToWire translates a SignalWithStartWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWithStartWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\tw, err = v.SignalWithStartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Read(w wire.Value) (*shared.SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Read(m wire.MapItemList) (map[string]string, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]string, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a SignalWithStartWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWithStartWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWithStartWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWithStartWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_String_Encode(val map[string]string, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a SignalWithStartWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionRequest struct could not be encoded.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalWithStartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalWithStartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWithStartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Decode(sr stream.Reader) (map[string]string, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]string, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a SignalWithStartWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWithStartWorkflowExecutionRequest\n// struct.\nfunc (v *SignalWithStartWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalWithStartRequest: %v\", v.SignalWithStartRequest)\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalWithStartWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_String_Equals(lhs, rhs map[string]string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this SignalWithStartWorkflowExecutionRequest match the\n// provided SignalWithStartWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Equals(rhs *SignalWithStartWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.SignalWithStartRequest == nil && rhs.SignalWithStartRequest == nil) || (v.SignalWithStartRequest != nil && rhs.SignalWithStartRequest != nil && v.SignalWithStartRequest.Equals(rhs.SignalWithStartRequest))) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_String_Zapper map[string]string\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_String_Zapper.\nfunc (m _Map_String_String_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWithStartWorkflowExecutionRequest.\nfunc (v *SignalWithStartWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalWithStartRequest\", v.SignalWithStartRequest))\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetSignalWithStartRequest returns the value of SignalWithStartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSignalWithStartRequest() (o *shared.SignalWithStartWorkflowExecutionRequest) {\n\tif v != nil && v.SignalWithStartRequest != nil {\n\t\treturn v.SignalWithStartRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalWithStartRequest returns true if SignalWithStartRequest is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetSignalWithStartRequest() bool {\n\treturn v != nil && v.SignalWithStartRequest != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype SignalWorkflowExecutionRequest struct {\n\tDomainUUID                *string                                `json:\"domainUUID,omitempty\"`\n\tSignalRequest             *shared.SignalWorkflowExecutionRequest `json:\"signalRequest,omitempty\"`\n\tExternalWorkflowExecution *shared.WorkflowExecution              `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         *bool                                  `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a SignalWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SignalRequest != nil {\n\t\tw, err = v.SignalRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tw, err = v.ExternalWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWorkflowExecutionRequest_Read(w wire.Value) (*shared.SignalWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a SignalWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWorkflowExecutionRequest struct could not be encoded.\nfunc (v *SignalWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.SignalWorkflowExecutionRequest, error) {\n\tvar v shared.SignalWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a SignalWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *SignalWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWorkflowExecutionRequest\n// struct.\nfunc (v *SignalWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.SignalRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalRequest: %v\", v.SignalRequest)\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecution: %v\", v.ExternalWorkflowExecution)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalWorkflowExecutionRequest match the\n// provided SignalWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWorkflowExecutionRequest) Equals(rhs *SignalWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.SignalRequest == nil && rhs.SignalRequest == nil) || (v.SignalRequest != nil && rhs.SignalRequest != nil && v.SignalRequest.Equals(rhs.SignalRequest))) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecution == nil && rhs.ExternalWorkflowExecution == nil) || (v.ExternalWorkflowExecution != nil && rhs.ExternalWorkflowExecution != nil && v.ExternalWorkflowExecution.Equals(rhs.ExternalWorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWorkflowExecutionRequest.\nfunc (v *SignalWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.SignalRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalRequest\", v.SignalRequest))\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecution\", v.ExternalWorkflowExecution))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetSignalRequest returns the value of SignalRequest if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetSignalRequest() (o *shared.SignalWorkflowExecutionRequest) {\n\tif v != nil && v.SignalRequest != nil {\n\t\treturn v.SignalRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalRequest returns true if SignalRequest is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetSignalRequest() bool {\n\treturn v != nil && v.SignalRequest != nil\n}\n\n// GetExternalWorkflowExecution returns the value of ExternalWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetExternalWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecution returns true if ExternalWorkflowExecution is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetExternalWorkflowExecution() bool {\n\treturn v != nil && v.ExternalWorkflowExecution != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype StartWorkflowExecutionRequest struct {\n\tDomainUUID                      *string                               `json:\"domainUUID,omitempty\"`\n\tStartRequest                    *shared.StartWorkflowExecutionRequest `json:\"startRequest,omitempty\"`\n\tParentExecutionInfo             *ParentExecutionInfo                  `json:\"parentExecutionInfo,omitempty\"`\n\tAttempt                         *int32                                `json:\"attempt,omitempty\"`\n\tExpirationTimestamp             *int64                                `json:\"expirationTimestamp,omitempty\"`\n\tContinueAsNewInitiator          *shared.ContinueAsNewInitiator        `json:\"continueAsNewInitiator,omitempty\"`\n\tContinuedFailureReason          *string                               `json:\"continuedFailureReason,omitempty\"`\n\tContinuedFailureDetails         []byte                                `json:\"continuedFailureDetails,omitempty\"`\n\tLastCompletionResult            []byte                                `json:\"lastCompletionResult,omitempty\"`\n\tFirstDecisionTaskBackoffSeconds *int32                                `json:\"firstDecisionTaskBackoffSeconds,omitempty\"`\n\tPartitionConfig                 map[string]string                     `json:\"partitionConfig,omitempty\"`\n}\n\n// ToWire translates a StartWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [11]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartRequest != nil {\n\t\tw, err = v.StartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ParentExecutionInfo != nil {\n\t\tw, err = v.ParentExecutionInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpirationTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ContinueAsNewInitiator != nil {\n\t\tw, err = v.ContinueAsNewInitiator.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.ContinuedFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.ContinuedFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 57, Value: w}\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tw, err = wire.NewValueBinary(v.LastCompletionResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 58, Value: w}\n\t\ti++\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.FirstDecisionTaskBackoffSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 62, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionRequest_Read(w wire.Value) (*shared.StartWorkflowExecutionRequest, error) {\n\tvar v shared.StartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ParentExecutionInfo_Read(w wire.Value) (*ParentExecutionInfo, error) {\n\tvar v ParentExecutionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ContinueAsNewInitiator_Read(w wire.Value) (shared.ContinueAsNewInitiator, error) {\n\tvar v shared.ContinueAsNewInitiator\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a StartWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ParentExecutionInfo, err = _ParentExecutionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpirationTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x shared.ContinueAsNewInitiator\n\t\t\t\tx, err = _ContinueAsNewInitiator_Read(field.Value)\n\t\t\t\tv.ContinueAsNewInitiator = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ContinuedFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 57:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ContinuedFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 58:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastCompletionResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.FirstDecisionTaskBackoffSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 62:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartWorkflowExecutionRequest struct could not be encoded.\nfunc (v *StartWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentExecutionInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentExecutionInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpirationTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpirationTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinueAsNewInitiator != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ContinueAsNewInitiator.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinuedFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ContinuedFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinuedFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 57, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ContinuedFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastCompletionResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 58, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastCompletionResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.FirstDecisionTaskBackoffSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 62, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionRequest, error) {\n\tvar v shared.StartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ParentExecutionInfo_Decode(sr stream.Reader) (*ParentExecutionInfo, error) {\n\tvar v ParentExecutionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ContinueAsNewInitiator_Decode(sr stream.Reader) (shared.ContinueAsNewInitiator, error) {\n\tvar v shared.ContinueAsNewInitiator\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a StartWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *StartWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ParentExecutionInfo, err = _ParentExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpirationTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TI32:\n\t\t\tvar x shared.ContinueAsNewInitiator\n\t\t\tx, err = _ContinueAsNewInitiator_Decode(sr)\n\t\t\tv.ContinueAsNewInitiator = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ContinuedFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 57 && fh.Type == wire.TBinary:\n\t\t\tv.ContinuedFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 58 && fh.Type == wire.TBinary:\n\t\t\tv.LastCompletionResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.FirstDecisionTaskBackoffSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 62 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartWorkflowExecutionRequest\n// struct.\nfunc (v *StartWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [11]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.StartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartRequest: %v\", v.StartRequest)\n\t\ti++\n\t}\n\tif v.ParentExecutionInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentExecutionInfo: %v\", v.ParentExecutionInfo)\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpirationTimestamp: %v\", *(v.ExpirationTimestamp))\n\t\ti++\n\t}\n\tif v.ContinueAsNewInitiator != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinueAsNewInitiator: %v\", *(v.ContinueAsNewInitiator))\n\t\ti++\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinuedFailureReason: %v\", *(v.ContinuedFailureReason))\n\t\ti++\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinuedFailureDetails: %v\", v.ContinuedFailureDetails)\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastCompletionResult: %v\", v.LastCompletionResult)\n\t\ti++\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstDecisionTaskBackoffSeconds: %v\", *(v.FirstDecisionTaskBackoffSeconds))\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ContinueAsNewInitiator_EqualsPtr(lhs, rhs *shared.ContinueAsNewInitiator) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this StartWorkflowExecutionRequest match the\n// provided StartWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *StartWorkflowExecutionRequest) Equals(rhs *StartWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.StartRequest == nil && rhs.StartRequest == nil) || (v.StartRequest != nil && rhs.StartRequest != nil && v.StartRequest.Equals(rhs.StartRequest))) {\n\t\treturn false\n\t}\n\tif !((v.ParentExecutionInfo == nil && rhs.ParentExecutionInfo == nil) || (v.ParentExecutionInfo != nil && rhs.ParentExecutionInfo != nil && v.ParentExecutionInfo.Equals(rhs.ParentExecutionInfo))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpirationTimestamp, rhs.ExpirationTimestamp) {\n\t\treturn false\n\t}\n\tif !_ContinueAsNewInitiator_EqualsPtr(v.ContinueAsNewInitiator, rhs.ContinueAsNewInitiator) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ContinuedFailureReason, rhs.ContinuedFailureReason) {\n\t\treturn false\n\t}\n\tif !((v.ContinuedFailureDetails == nil && rhs.ContinuedFailureDetails == nil) || (v.ContinuedFailureDetails != nil && rhs.ContinuedFailureDetails != nil && bytes.Equal(v.ContinuedFailureDetails, rhs.ContinuedFailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.LastCompletionResult == nil && rhs.LastCompletionResult == nil) || (v.LastCompletionResult != nil && rhs.LastCompletionResult != nil && bytes.Equal(v.LastCompletionResult, rhs.LastCompletionResult))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.FirstDecisionTaskBackoffSeconds, rhs.FirstDecisionTaskBackoffSeconds) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartWorkflowExecutionRequest.\nfunc (v *StartWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.StartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startRequest\", v.StartRequest))\n\t}\n\tif v.ParentExecutionInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentExecutionInfo\", v.ParentExecutionInfo))\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tenc.AddInt64(\"expirationTimestamp\", *v.ExpirationTimestamp)\n\t}\n\tif v.ContinueAsNewInitiator != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"continueAsNewInitiator\", *v.ContinueAsNewInitiator))\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tenc.AddString(\"continuedFailureReason\", *v.ContinuedFailureReason)\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tenc.AddString(\"continuedFailureDetails\", base64.StdEncoding.EncodeToString(v.ContinuedFailureDetails))\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tenc.AddString(\"lastCompletionResult\", base64.StdEncoding.EncodeToString(v.LastCompletionResult))\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tenc.AddInt32(\"firstDecisionTaskBackoffSeconds\", *v.FirstDecisionTaskBackoffSeconds)\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetStartRequest returns the value of StartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetStartRequest() (o *shared.StartWorkflowExecutionRequest) {\n\tif v != nil && v.StartRequest != nil {\n\t\treturn v.StartRequest\n\t}\n\n\treturn\n}\n\n// IsSetStartRequest returns true if StartRequest is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetStartRequest() bool {\n\treturn v != nil && v.StartRequest != nil\n}\n\n// GetParentExecutionInfo returns the value of ParentExecutionInfo if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetParentExecutionInfo() (o *ParentExecutionInfo) {\n\tif v != nil && v.ParentExecutionInfo != nil {\n\t\treturn v.ParentExecutionInfo\n\t}\n\n\treturn\n}\n\n// IsSetParentExecutionInfo returns true if ParentExecutionInfo is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetParentExecutionInfo() bool {\n\treturn v != nil && v.ParentExecutionInfo != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetExpirationTimestamp returns the value of ExpirationTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetExpirationTimestamp() (o int64) {\n\tif v != nil && v.ExpirationTimestamp != nil {\n\t\treturn *v.ExpirationTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetExpirationTimestamp returns true if ExpirationTimestamp is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetExpirationTimestamp() bool {\n\treturn v != nil && v.ExpirationTimestamp != nil\n}\n\n// GetContinueAsNewInitiator returns the value of ContinueAsNewInitiator if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetContinueAsNewInitiator() (o shared.ContinueAsNewInitiator) {\n\tif v != nil && v.ContinueAsNewInitiator != nil {\n\t\treturn *v.ContinueAsNewInitiator\n\t}\n\n\treturn\n}\n\n// IsSetContinueAsNewInitiator returns true if ContinueAsNewInitiator is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetContinueAsNewInitiator() bool {\n\treturn v != nil && v.ContinueAsNewInitiator != nil\n}\n\n// GetContinuedFailureReason returns the value of ContinuedFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetContinuedFailureReason() (o string) {\n\tif v != nil && v.ContinuedFailureReason != nil {\n\t\treturn *v.ContinuedFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetContinuedFailureReason returns true if ContinuedFailureReason is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetContinuedFailureReason() bool {\n\treturn v != nil && v.ContinuedFailureReason != nil\n}\n\n// GetContinuedFailureDetails returns the value of ContinuedFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetContinuedFailureDetails() (o []byte) {\n\tif v != nil && v.ContinuedFailureDetails != nil {\n\t\treturn v.ContinuedFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetContinuedFailureDetails returns true if ContinuedFailureDetails is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetContinuedFailureDetails() bool {\n\treturn v != nil && v.ContinuedFailureDetails != nil\n}\n\n// GetLastCompletionResult returns the value of LastCompletionResult if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetLastCompletionResult() (o []byte) {\n\tif v != nil && v.LastCompletionResult != nil {\n\t\treturn v.LastCompletionResult\n\t}\n\n\treturn\n}\n\n// IsSetLastCompletionResult returns true if LastCompletionResult is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetLastCompletionResult() bool {\n\treturn v != nil && v.LastCompletionResult != nil\n}\n\n// GetFirstDecisionTaskBackoffSeconds returns the value of FirstDecisionTaskBackoffSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetFirstDecisionTaskBackoffSeconds() (o int32) {\n\tif v != nil && v.FirstDecisionTaskBackoffSeconds != nil {\n\t\treturn *v.FirstDecisionTaskBackoffSeconds\n\t}\n\n\treturn\n}\n\n// IsSetFirstDecisionTaskBackoffSeconds returns true if FirstDecisionTaskBackoffSeconds is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetFirstDecisionTaskBackoffSeconds() bool {\n\treturn v != nil && v.FirstDecisionTaskBackoffSeconds != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype SyncActivityRequest struct {\n\tDomainId           *string                `json:\"domainId,omitempty\"`\n\tWorkflowId         *string                `json:\"workflowId,omitempty\"`\n\tRunId              *string                `json:\"runId,omitempty\"`\n\tVersion            *int64                 `json:\"version,omitempty\"`\n\tScheduledId        *int64                 `json:\"scheduledId,omitempty\"`\n\tScheduledTime      *int64                 `json:\"scheduledTime,omitempty\"`\n\tStartedId          *int64                 `json:\"startedId,omitempty\"`\n\tStartedTime        *int64                 `json:\"startedTime,omitempty\"`\n\tLastHeartbeatTime  *int64                 `json:\"lastHeartbeatTime,omitempty\"`\n\tDetails            []byte                 `json:\"details,omitempty\"`\n\tAttempt            *int32                 `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string                `json:\"lastFailureReason,omitempty\"`\n\tLastWorkerIdentity *string                `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails []byte                 `json:\"lastFailureDetails,omitempty\"`\n\tVersionHistory     *shared.VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\n// ToWire translates a SyncActivityRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SyncActivityRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainId != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastHeartbeatTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.LastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.LastWorkerIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.LastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tw, err = v.VersionHistory.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistory_Read(w wire.Value) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a SyncActivityRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SyncActivityRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SyncActivityRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SyncActivityRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastHeartbeatTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastWorkerIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistory, err = _VersionHistory_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SyncActivityRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SyncActivityRequest struct could not be encoded.\nfunc (v *SyncActivityRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastHeartbeatTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastHeartbeatTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWorkerIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastWorkerIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistory != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistory.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistory_Decode(sr stream.Reader) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a SyncActivityRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SyncActivityRequest struct could not be generated from the wire\n// representation.\nfunc (v *SyncActivityRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastHeartbeatTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastWorkerIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TBinary:\n\t\t\tv.LastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistory, err = _VersionHistory_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SyncActivityRequest\n// struct.\nfunc (v *SyncActivityRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.DomainId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainId: %v\", *(v.DomainId))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.ScheduledId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledId: %v\", *(v.ScheduledId))\n\t\ti++\n\t}\n\tif v.ScheduledTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTime: %v\", *(v.ScheduledTime))\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedId: %v\", *(v.StartedId))\n\t\ti++\n\t}\n\tif v.StartedTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTime: %v\", *(v.StartedTime))\n\t\ti++\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastHeartbeatTime: %v\", *(v.LastHeartbeatTime))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureReason: %v\", *(v.LastFailureReason))\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWorkerIdentity: %v\", *(v.LastWorkerIdentity))\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureDetails: %v\", v.LastFailureDetails)\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistory: %v\", v.VersionHistory)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SyncActivityRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SyncActivityRequest match the\n// provided SyncActivityRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SyncActivityRequest) Equals(rhs *SyncActivityRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainId, rhs.DomainId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledId, rhs.ScheduledId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTime, rhs.ScheduledTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedId, rhs.StartedId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTime, rhs.StartedTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastHeartbeatTime, rhs.LastHeartbeatTime) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastFailureReason, rhs.LastFailureReason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastWorkerIdentity, rhs.LastWorkerIdentity) {\n\t\treturn false\n\t}\n\tif !((v.LastFailureDetails == nil && rhs.LastFailureDetails == nil) || (v.LastFailureDetails != nil && rhs.LastFailureDetails != nil && bytes.Equal(v.LastFailureDetails, rhs.LastFailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistory == nil && rhs.VersionHistory == nil) || (v.VersionHistory != nil && rhs.VersionHistory != nil && v.VersionHistory.Equals(rhs.VersionHistory))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SyncActivityRequest.\nfunc (v *SyncActivityRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainId != nil {\n\t\tenc.AddString(\"domainId\", *v.DomainId)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.ScheduledId != nil {\n\t\tenc.AddInt64(\"scheduledId\", *v.ScheduledId)\n\t}\n\tif v.ScheduledTime != nil {\n\t\tenc.AddInt64(\"scheduledTime\", *v.ScheduledTime)\n\t}\n\tif v.StartedId != nil {\n\t\tenc.AddInt64(\"startedId\", *v.StartedId)\n\t}\n\tif v.StartedTime != nil {\n\t\tenc.AddInt64(\"startedTime\", *v.StartedTime)\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tenc.AddInt64(\"lastHeartbeatTime\", *v.LastHeartbeatTime)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.LastFailureReason != nil {\n\t\tenc.AddString(\"lastFailureReason\", *v.LastFailureReason)\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tenc.AddString(\"lastWorkerIdentity\", *v.LastWorkerIdentity)\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tenc.AddString(\"lastFailureDetails\", base64.StdEncoding.EncodeToString(v.LastFailureDetails))\n\t}\n\tif v.VersionHistory != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistory\", v.VersionHistory))\n\t}\n\treturn err\n}\n\n// GetDomainId returns the value of DomainId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetDomainId() (o string) {\n\tif v != nil && v.DomainId != nil {\n\t\treturn *v.DomainId\n\t}\n\n\treturn\n}\n\n// IsSetDomainId returns true if DomainId is not nil.\nfunc (v *SyncActivityRequest) IsSetDomainId() bool {\n\treturn v != nil && v.DomainId != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *SyncActivityRequest) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *SyncActivityRequest) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *SyncActivityRequest) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetScheduledId returns the value of ScheduledId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetScheduledId() (o int64) {\n\tif v != nil && v.ScheduledId != nil {\n\t\treturn *v.ScheduledId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledId returns true if ScheduledId is not nil.\nfunc (v *SyncActivityRequest) IsSetScheduledId() bool {\n\treturn v != nil && v.ScheduledId != nil\n}\n\n// GetScheduledTime returns the value of ScheduledTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetScheduledTime() (o int64) {\n\tif v != nil && v.ScheduledTime != nil {\n\t\treturn *v.ScheduledTime\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTime returns true if ScheduledTime is not nil.\nfunc (v *SyncActivityRequest) IsSetScheduledTime() bool {\n\treturn v != nil && v.ScheduledTime != nil\n}\n\n// GetStartedId returns the value of StartedId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetStartedId() (o int64) {\n\tif v != nil && v.StartedId != nil {\n\t\treturn *v.StartedId\n\t}\n\n\treturn\n}\n\n// IsSetStartedId returns true if StartedId is not nil.\nfunc (v *SyncActivityRequest) IsSetStartedId() bool {\n\treturn v != nil && v.StartedId != nil\n}\n\n// GetStartedTime returns the value of StartedTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetStartedTime() (o int64) {\n\tif v != nil && v.StartedTime != nil {\n\t\treturn *v.StartedTime\n\t}\n\n\treturn\n}\n\n// IsSetStartedTime returns true if StartedTime is not nil.\nfunc (v *SyncActivityRequest) IsSetStartedTime() bool {\n\treturn v != nil && v.StartedTime != nil\n}\n\n// GetLastHeartbeatTime returns the value of LastHeartbeatTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetLastHeartbeatTime() (o int64) {\n\tif v != nil && v.LastHeartbeatTime != nil {\n\t\treturn *v.LastHeartbeatTime\n\t}\n\n\treturn\n}\n\n// IsSetLastHeartbeatTime returns true if LastHeartbeatTime is not nil.\nfunc (v *SyncActivityRequest) IsSetLastHeartbeatTime() bool {\n\treturn v != nil && v.LastHeartbeatTime != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *SyncActivityRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *SyncActivityRequest) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetLastFailureReason returns the value of LastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureReason returns true if LastFailureReason is not nil.\nfunc (v *SyncActivityRequest) IsSetLastFailureReason() bool {\n\treturn v != nil && v.LastFailureReason != nil\n}\n\n// GetLastWorkerIdentity returns the value of LastWorkerIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetLastWorkerIdentity() (o string) {\n\tif v != nil && v.LastWorkerIdentity != nil {\n\t\treturn *v.LastWorkerIdentity\n\t}\n\n\treturn\n}\n\n// IsSetLastWorkerIdentity returns true if LastWorkerIdentity is not nil.\nfunc (v *SyncActivityRequest) IsSetLastWorkerIdentity() bool {\n\treturn v != nil && v.LastWorkerIdentity != nil\n}\n\n// GetLastFailureDetails returns the value of LastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureDetails returns true if LastFailureDetails is not nil.\nfunc (v *SyncActivityRequest) IsSetLastFailureDetails() bool {\n\treturn v != nil && v.LastFailureDetails != nil\n}\n\n// GetVersionHistory returns the value of VersionHistory if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityRequest) GetVersionHistory() (o *shared.VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistory returns true if VersionHistory is not nil.\nfunc (v *SyncActivityRequest) IsSetVersionHistory() bool {\n\treturn v != nil && v.VersionHistory != nil\n}\n\ntype SyncShardStatusRequest struct {\n\tSourceCluster *string `json:\"sourceCluster,omitempty\"`\n\tShardId       *int64  `json:\"shardId,omitempty\"`\n\tTimestamp     *int64  `json:\"timestamp,omitempty\"`\n}\n\n// ToWire translates a SyncShardStatusRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SyncShardStatusRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SourceCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ShardId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.Timestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SyncShardStatusRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SyncShardStatusRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SyncShardStatusRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SyncShardStatusRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ShardId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Timestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SyncShardStatusRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SyncShardStatusRequest struct could not be encoded.\nfunc (v *SyncShardStatusRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SourceCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ShardId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Timestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Timestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SyncShardStatusRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SyncShardStatusRequest struct could not be generated from the wire\n// representation.\nfunc (v *SyncShardStatusRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ShardId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Timestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SyncShardStatusRequest\n// struct.\nfunc (v *SyncShardStatusRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.SourceCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceCluster: %v\", *(v.SourceCluster))\n\t\ti++\n\t}\n\tif v.ShardId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardId: %v\", *(v.ShardId))\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"Timestamp: %v\", *(v.Timestamp))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SyncShardStatusRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SyncShardStatusRequest match the\n// provided SyncShardStatusRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SyncShardStatusRequest) Equals(rhs *SyncShardStatusRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceCluster, rhs.SourceCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ShardId, rhs.ShardId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Timestamp, rhs.Timestamp) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SyncShardStatusRequest.\nfunc (v *SyncShardStatusRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SourceCluster != nil {\n\t\tenc.AddString(\"sourceCluster\", *v.SourceCluster)\n\t}\n\tif v.ShardId != nil {\n\t\tenc.AddInt64(\"shardId\", *v.ShardId)\n\t}\n\tif v.Timestamp != nil {\n\t\tenc.AddInt64(\"timestamp\", *v.Timestamp)\n\t}\n\treturn err\n}\n\n// GetSourceCluster returns the value of SourceCluster if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusRequest) GetSourceCluster() (o string) {\n\tif v != nil && v.SourceCluster != nil {\n\t\treturn *v.SourceCluster\n\t}\n\n\treturn\n}\n\n// IsSetSourceCluster returns true if SourceCluster is not nil.\nfunc (v *SyncShardStatusRequest) IsSetSourceCluster() bool {\n\treturn v != nil && v.SourceCluster != nil\n}\n\n// GetShardId returns the value of ShardId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusRequest) GetShardId() (o int64) {\n\tif v != nil && v.ShardId != nil {\n\t\treturn *v.ShardId\n\t}\n\n\treturn\n}\n\n// IsSetShardId returns true if ShardId is not nil.\nfunc (v *SyncShardStatusRequest) IsSetShardId() bool {\n\treturn v != nil && v.ShardId != nil\n}\n\n// GetTimestamp returns the value of Timestamp if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusRequest) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\n\treturn\n}\n\n// IsSetTimestamp returns true if Timestamp is not nil.\nfunc (v *SyncShardStatusRequest) IsSetTimestamp() bool {\n\treturn v != nil && v.Timestamp != nil\n}\n\ntype TerminateWorkflowExecutionRequest struct {\n\tDomainUUID                *string                                   `json:\"domainUUID,omitempty\"`\n\tTerminateRequest          *shared.TerminateWorkflowExecutionRequest `json:\"terminateRequest,omitempty\"`\n\tExternalWorkflowExecution *shared.WorkflowExecution                 `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         *bool                                     `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a TerminateWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TerminateWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TerminateRequest != nil {\n\t\tw, err = v.TerminateRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tw, err = v.ExternalWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TerminateWorkflowExecutionRequest_Read(w wire.Value) (*shared.TerminateWorkflowExecutionRequest, error) {\n\tvar v shared.TerminateWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a TerminateWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TerminateWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TerminateWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TerminateWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TerminateWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TerminateWorkflowExecutionRequest struct could not be encoded.\nfunc (v *TerminateWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TerminateRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TerminateRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TerminateWorkflowExecutionRequest_Decode(sr stream.Reader) (*shared.TerminateWorkflowExecutionRequest, error) {\n\tvar v shared.TerminateWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a TerminateWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TerminateWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *TerminateWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TerminateWorkflowExecutionRequest\n// struct.\nfunc (v *TerminateWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.TerminateRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"TerminateRequest: %v\", v.TerminateRequest)\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecution: %v\", v.ExternalWorkflowExecution)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TerminateWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TerminateWorkflowExecutionRequest match the\n// provided TerminateWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *TerminateWorkflowExecutionRequest) Equals(rhs *TerminateWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.TerminateRequest == nil && rhs.TerminateRequest == nil) || (v.TerminateRequest != nil && rhs.TerminateRequest != nil && v.TerminateRequest.Equals(rhs.TerminateRequest))) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecution == nil && rhs.ExternalWorkflowExecution == nil) || (v.ExternalWorkflowExecution != nil && rhs.ExternalWorkflowExecution != nil && v.ExternalWorkflowExecution.Equals(rhs.ExternalWorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TerminateWorkflowExecutionRequest.\nfunc (v *TerminateWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.TerminateRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"terminateRequest\", v.TerminateRequest))\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecution\", v.ExternalWorkflowExecution))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetTerminateRequest returns the value of TerminateRequest if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetTerminateRequest() (o *shared.TerminateWorkflowExecutionRequest) {\n\tif v != nil && v.TerminateRequest != nil {\n\t\treturn v.TerminateRequest\n\t}\n\n\treturn\n}\n\n// IsSetTerminateRequest returns true if TerminateRequest is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetTerminateRequest() bool {\n\treturn v != nil && v.TerminateRequest != nil\n}\n\n// GetExternalWorkflowExecution returns the value of ExternalWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetExternalWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecution returns true if ExternalWorkflowExecution is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetExternalWorkflowExecution() bool {\n\treturn v != nil && v.ExternalWorkflowExecution != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\n// fields are required to encourage compact serialization, zeros are expected\ntype WeightedRatelimitCalls struct {\n\t// number of allowed requests since last call.\n\t// assumed to be <1m or so, saturates at MAX_INT32.\n\tAllowed int32 `json:\"allowed,required\"`\n\t// number of rejected requests since last call.\n\t// assumed to be <1m or so, saturates at MAX_INT32.\n\tRejected int32 `json:\"rejected,required\"`\n}\n\n// ToWire translates a WeightedRatelimitCalls struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WeightedRatelimitCalls) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueI32(v.Allowed), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\tw, err = wire.NewValueI32(v.Rejected), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 20, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WeightedRatelimitCalls struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WeightedRatelimitCalls struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WeightedRatelimitCalls\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WeightedRatelimitCalls) FromWire(w wire.Value) error {\n\tvar err error\n\n\tallowedIsSet := false\n\trejectedIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tv.Allowed, err = field.Value.GetI32(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tallowedIsSet = true\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tv.Rejected, err = field.Value.GetI32(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\trejectedIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !allowedIsSet {\n\t\treturn errors.New(\"field Allowed of WeightedRatelimitCalls is required\")\n\t}\n\n\tif !rejectedIsSet {\n\t\treturn errors.New(\"field Rejected of WeightedRatelimitCalls is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WeightedRatelimitCalls struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WeightedRatelimitCalls struct could not be encoded.\nfunc (v *WeightedRatelimitCalls) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteInt32(v.Allowed); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteInt32(v.Rejected); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WeightedRatelimitCalls struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WeightedRatelimitCalls struct could not be generated from the wire\n// representation.\nfunc (v *WeightedRatelimitCalls) Decode(sr stream.Reader) error {\n\n\tallowedIsSet := false\n\trejectedIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tv.Allowed, err = sr.ReadInt32()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tallowedIsSet = true\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tv.Rejected, err = sr.ReadInt32()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trejectedIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !allowedIsSet {\n\t\treturn errors.New(\"field Allowed of WeightedRatelimitCalls is required\")\n\t}\n\n\tif !rejectedIsSet {\n\t\treturn errors.New(\"field Rejected of WeightedRatelimitCalls is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WeightedRatelimitCalls\n// struct.\nfunc (v *WeightedRatelimitCalls) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Allowed: %v\", v.Allowed)\n\ti++\n\tfields[i] = fmt.Sprintf(\"Rejected: %v\", v.Rejected)\n\ti++\n\n\treturn fmt.Sprintf(\"WeightedRatelimitCalls{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WeightedRatelimitCalls match the\n// provided WeightedRatelimitCalls.\n//\n// This function performs a deep comparison.\nfunc (v *WeightedRatelimitCalls) Equals(rhs *WeightedRatelimitCalls) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Allowed == rhs.Allowed) {\n\t\treturn false\n\t}\n\tif !(v.Rejected == rhs.Rejected) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WeightedRatelimitCalls.\nfunc (v *WeightedRatelimitCalls) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddInt32(\"allowed\", v.Allowed)\n\tenc.AddInt32(\"rejected\", v.Rejected)\n\treturn err\n}\n\n// GetAllowed returns the value of Allowed if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitCalls) GetAllowed() (o int32) {\n\tif v != nil {\n\t\to = v.Allowed\n\t}\n\treturn\n}\n\n// GetRejected returns the value of Rejected if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitCalls) GetRejected() (o int32) {\n\tif v != nil {\n\t\to = v.Rejected\n\t}\n\treturn\n}\n\n// first impl of ratelimiting data, result from aggregator to limiter.\n//\n// used in an Any with ValueType: WeightedRatelimitQuotasAnyType\ntype WeightedRatelimitQuotas struct {\n\t// RPS-weights to allow per key\n\tQuotas map[string]float64 `json:\"quotas,required\"`\n}\n\ntype _Map_String_Double_MapItemList map[string]float64\n\nfunc (m _Map_String_Double_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueDouble(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_Double_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_Double_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Double_MapItemList) ValueType() wire.Type {\n\treturn wire.TDouble\n}\n\nfunc (_Map_String_Double_MapItemList) Close() {}\n\n// ToWire translates a WeightedRatelimitQuotas struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WeightedRatelimitQuotas) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Quotas == nil {\n\t\treturn w, errors.New(\"field Quotas of WeightedRatelimitQuotas is required\")\n\t}\n\tw, err = wire.NewValueMap(_Map_String_Double_MapItemList(v.Quotas)), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_String_Double_Read(m wire.MapItemList) (map[string]float64, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TDouble {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]float64, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetDouble(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a WeightedRatelimitQuotas struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WeightedRatelimitQuotas struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WeightedRatelimitQuotas\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WeightedRatelimitQuotas) FromWire(w wire.Value) error {\n\tvar err error\n\n\tquotasIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Quotas, err = _Map_String_Double_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tquotasIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !quotasIsSet {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitQuotas is required\")\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_Double_Encode(val map[string]float64, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TDouble,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a WeightedRatelimitQuotas struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WeightedRatelimitQuotas struct could not be encoded.\nfunc (v *WeightedRatelimitQuotas) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Quotas == nil {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitQuotas is required\")\n\t}\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\treturn err\n\t}\n\tif err := _Map_String_Double_Encode(v.Quotas, sw); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_String_Double_Decode(sr stream.Reader) (map[string]float64, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TDouble {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]float64, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadDouble()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a WeightedRatelimitQuotas struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WeightedRatelimitQuotas struct could not be generated from the wire\n// representation.\nfunc (v *WeightedRatelimitQuotas) Decode(sr stream.Reader) error {\n\n\tquotasIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Quotas, err = _Map_String_Double_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tquotasIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !quotasIsSet {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitQuotas is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WeightedRatelimitQuotas\n// struct.\nfunc (v *WeightedRatelimitQuotas) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Quotas: %v\", v.Quotas)\n\ti++\n\n\treturn fmt.Sprintf(\"WeightedRatelimitQuotas{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_Double_Equals(lhs, rhs map[string]float64) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this WeightedRatelimitQuotas match the\n// provided WeightedRatelimitQuotas.\n//\n// This function performs a deep comparison.\nfunc (v *WeightedRatelimitQuotas) Equals(rhs *WeightedRatelimitQuotas) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Map_String_Double_Equals(v.Quotas, rhs.Quotas) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_Double_Zapper map[string]float64\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_Double_Zapper.\nfunc (m _Map_String_Double_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddFloat64((string)(k), v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WeightedRatelimitQuotas.\nfunc (v *WeightedRatelimitQuotas) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\terr = multierr.Append(err, enc.AddObject(\"quotas\", (_Map_String_Double_Zapper)(v.Quotas)))\n\treturn err\n}\n\n// GetQuotas returns the value of Quotas if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitQuotas) GetQuotas() (o map[string]float64) {\n\tif v != nil {\n\t\to = v.Quotas\n\t}\n\treturn\n}\n\n// IsSetQuotas returns true if Quotas is not nil.\nfunc (v *WeightedRatelimitQuotas) IsSetQuotas() bool {\n\treturn v != nil && v.Quotas != nil\n}\n\n// first impl of ratelimiting data, collected by limiters and sent to aggregators.\n//\n// used in an Any with ValueType: WeightedRatelimitUsageAnyType\ntype WeightedRatelimitUsage struct {\n\t// unique, stable identifier of the calling host, to identify future data from the same host\n\tCaller string `json:\"caller,required\"`\n\t// milliseconds since last update call.  expected to be on the order of a few seconds or less.\n\tElapsedMS int32 `json:\"elapsedMS,required\"`\n\t// per key, number of allowed vs rejected calls since last update.\n\tCalls map[string]*WeightedRatelimitCalls `json:\"calls,required\"`\n}\n\ntype _Map_String_WeightedRatelimitCalls_MapItemList map[string]*WeightedRatelimitCalls\n\nfunc (m _Map_String_WeightedRatelimitCalls_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WeightedRatelimitCalls', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WeightedRatelimitCalls_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WeightedRatelimitCalls_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WeightedRatelimitCalls_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WeightedRatelimitCalls_MapItemList) Close() {}\n\n// ToWire translates a WeightedRatelimitUsage struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WeightedRatelimitUsage) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Caller), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\tw, err = wire.NewValueI32(v.ElapsedMS), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 20, Value: w}\n\ti++\n\tif v.Calls == nil {\n\t\treturn w, errors.New(\"field Calls of WeightedRatelimitUsage is required\")\n\t}\n\tw, err = wire.NewValueMap(_Map_String_WeightedRatelimitCalls_MapItemList(v.Calls)), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 30, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WeightedRatelimitCalls_Read(w wire.Value) (*WeightedRatelimitCalls, error) {\n\tvar v WeightedRatelimitCalls\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WeightedRatelimitCalls_Read(m wire.MapItemList) (map[string]*WeightedRatelimitCalls, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*WeightedRatelimitCalls, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WeightedRatelimitCalls_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a WeightedRatelimitUsage struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WeightedRatelimitUsage struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WeightedRatelimitUsage\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WeightedRatelimitUsage) FromWire(w wire.Value) error {\n\tvar err error\n\n\tcallerIsSet := false\n\telapsedMSIsSet := false\n\tcallsIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Caller, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcallerIsSet = true\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tv.ElapsedMS, err = field.Value.GetI32(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\telapsedMSIsSet = true\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Calls, err = _Map_String_WeightedRatelimitCalls_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcallsIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !callerIsSet {\n\t\treturn errors.New(\"field Caller of WeightedRatelimitUsage is required\")\n\t}\n\n\tif !elapsedMSIsSet {\n\t\treturn errors.New(\"field ElapsedMS of WeightedRatelimitUsage is required\")\n\t}\n\n\tif !callsIsSet {\n\t\treturn errors.New(\"field Calls of WeightedRatelimitUsage is required\")\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_WeightedRatelimitCalls_Encode(val map[string]*WeightedRatelimitCalls, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WeightedRatelimitCalls', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a WeightedRatelimitUsage struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WeightedRatelimitUsage struct could not be encoded.\nfunc (v *WeightedRatelimitUsage) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Caller); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteInt32(v.ElapsedMS); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Calls == nil {\n\t\treturn errors.New(\"field Calls of WeightedRatelimitUsage is required\")\n\t}\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TMap}); err != nil {\n\t\treturn err\n\t}\n\tif err := _Map_String_WeightedRatelimitCalls_Encode(v.Calls, sw); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WeightedRatelimitCalls_Decode(sr stream.Reader) (*WeightedRatelimitCalls, error) {\n\tvar v WeightedRatelimitCalls\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WeightedRatelimitCalls_Decode(sr stream.Reader) (map[string]*WeightedRatelimitCalls, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*WeightedRatelimitCalls, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WeightedRatelimitCalls_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a WeightedRatelimitUsage struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WeightedRatelimitUsage struct could not be generated from the wire\n// representation.\nfunc (v *WeightedRatelimitUsage) Decode(sr stream.Reader) error {\n\n\tcallerIsSet := false\n\telapsedMSIsSet := false\n\tcallsIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Caller, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tcallerIsSet = true\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tv.ElapsedMS, err = sr.ReadInt32()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\telapsedMSIsSet = true\n\t\tcase fh.ID == 30 && fh.Type == wire.TMap:\n\t\t\tv.Calls, err = _Map_String_WeightedRatelimitCalls_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tcallsIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !callerIsSet {\n\t\treturn errors.New(\"field Caller of WeightedRatelimitUsage is required\")\n\t}\n\n\tif !elapsedMSIsSet {\n\t\treturn errors.New(\"field ElapsedMS of WeightedRatelimitUsage is required\")\n\t}\n\n\tif !callsIsSet {\n\t\treturn errors.New(\"field Calls of WeightedRatelimitUsage is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WeightedRatelimitUsage\n// struct.\nfunc (v *WeightedRatelimitUsage) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Caller: %v\", v.Caller)\n\ti++\n\tfields[i] = fmt.Sprintf(\"ElapsedMS: %v\", v.ElapsedMS)\n\ti++\n\tfields[i] = fmt.Sprintf(\"Calls: %v\", v.Calls)\n\ti++\n\n\treturn fmt.Sprintf(\"WeightedRatelimitUsage{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_WeightedRatelimitCalls_Equals(lhs, rhs map[string]*WeightedRatelimitCalls) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this WeightedRatelimitUsage match the\n// provided WeightedRatelimitUsage.\n//\n// This function performs a deep comparison.\nfunc (v *WeightedRatelimitUsage) Equals(rhs *WeightedRatelimitUsage) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Caller == rhs.Caller) {\n\t\treturn false\n\t}\n\tif !(v.ElapsedMS == rhs.ElapsedMS) {\n\t\treturn false\n\t}\n\tif !_Map_String_WeightedRatelimitCalls_Equals(v.Calls, rhs.Calls) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_WeightedRatelimitCalls_Zapper map[string]*WeightedRatelimitCalls\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WeightedRatelimitCalls_Zapper.\nfunc (m _Map_String_WeightedRatelimitCalls_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WeightedRatelimitUsage.\nfunc (v *WeightedRatelimitUsage) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"caller\", v.Caller)\n\tenc.AddInt32(\"elapsedMS\", v.ElapsedMS)\n\terr = multierr.Append(err, enc.AddObject(\"calls\", (_Map_String_WeightedRatelimitCalls_Zapper)(v.Calls)))\n\treturn err\n}\n\n// GetCaller returns the value of Caller if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsage) GetCaller() (o string) {\n\tif v != nil {\n\t\to = v.Caller\n\t}\n\treturn\n}\n\n// GetElapsedMS returns the value of ElapsedMS if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsage) GetElapsedMS() (o int32) {\n\tif v != nil {\n\t\to = v.ElapsedMS\n\t}\n\treturn\n}\n\n// GetCalls returns the value of Calls if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsage) GetCalls() (o map[string]*WeightedRatelimitCalls) {\n\tif v != nil {\n\t\to = v.Calls\n\t}\n\treturn\n}\n\n// IsSetCalls returns true if Calls is not nil.\nfunc (v *WeightedRatelimitUsage) IsSetCalls() bool {\n\treturn v != nil && v.Calls != nil\n}\n\ntype WeightedRatelimitUsageQuotaEntry struct {\n\t// Amount of the quota that the receiving host can use, between 0 and 1\n\tWeight float64 `json:\"weight,required\"`\n\t// RPS estimated across the whole cluster\n\tUsed float64 `json:\"used,required\"`\n}\n\n// ToWire translates a WeightedRatelimitUsageQuotaEntry struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WeightedRatelimitUsageQuotaEntry) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueDouble(v.Weight), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\tw, err = wire.NewValueDouble(v.Used), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 20, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WeightedRatelimitUsageQuotaEntry struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WeightedRatelimitUsageQuotaEntry struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WeightedRatelimitUsageQuotaEntry\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WeightedRatelimitUsageQuotaEntry) FromWire(w wire.Value) error {\n\tvar err error\n\n\tweightIsSet := false\n\tusedIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tv.Weight, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tweightIsSet = true\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tv.Used, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tusedIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !weightIsSet {\n\t\treturn errors.New(\"field Weight of WeightedRatelimitUsageQuotaEntry is required\")\n\t}\n\n\tif !usedIsSet {\n\t\treturn errors.New(\"field Used of WeightedRatelimitUsageQuotaEntry is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WeightedRatelimitUsageQuotaEntry struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WeightedRatelimitUsageQuotaEntry struct could not be encoded.\nfunc (v *WeightedRatelimitUsageQuotaEntry) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TDouble}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteDouble(v.Weight); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TDouble}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteDouble(v.Used); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WeightedRatelimitUsageQuotaEntry struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WeightedRatelimitUsageQuotaEntry struct could not be generated from the wire\n// representation.\nfunc (v *WeightedRatelimitUsageQuotaEntry) Decode(sr stream.Reader) error {\n\n\tweightIsSet := false\n\tusedIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TDouble:\n\t\t\tv.Weight, err = sr.ReadDouble()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tweightIsSet = true\n\t\tcase fh.ID == 20 && fh.Type == wire.TDouble:\n\t\t\tv.Used, err = sr.ReadDouble()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tusedIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !weightIsSet {\n\t\treturn errors.New(\"field Weight of WeightedRatelimitUsageQuotaEntry is required\")\n\t}\n\n\tif !usedIsSet {\n\t\treturn errors.New(\"field Used of WeightedRatelimitUsageQuotaEntry is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WeightedRatelimitUsageQuotaEntry\n// struct.\nfunc (v *WeightedRatelimitUsageQuotaEntry) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Weight: %v\", v.Weight)\n\ti++\n\tfields[i] = fmt.Sprintf(\"Used: %v\", v.Used)\n\ti++\n\n\treturn fmt.Sprintf(\"WeightedRatelimitUsageQuotaEntry{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WeightedRatelimitUsageQuotaEntry match the\n// provided WeightedRatelimitUsageQuotaEntry.\n//\n// This function performs a deep comparison.\nfunc (v *WeightedRatelimitUsageQuotaEntry) Equals(rhs *WeightedRatelimitUsageQuotaEntry) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Weight == rhs.Weight) {\n\t\treturn false\n\t}\n\tif !(v.Used == rhs.Used) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WeightedRatelimitUsageQuotaEntry.\nfunc (v *WeightedRatelimitUsageQuotaEntry) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddFloat64(\"weight\", v.Weight)\n\tenc.AddFloat64(\"used\", v.Used)\n\treturn err\n}\n\n// GetWeight returns the value of Weight if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsageQuotaEntry) GetWeight() (o float64) {\n\tif v != nil {\n\t\to = v.Weight\n\t}\n\treturn\n}\n\n// GetUsed returns the value of Used if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsageQuotaEntry) GetUsed() (o float64) {\n\tif v != nil {\n\t\to = v.Used\n\t}\n\treturn\n}\n\n// second impl, includes unused-RPS data so limiters can decide if they\n// want to allow exceeding limits when there is free space.\n//\n// used in an Any with ValueType: WeightedRatelimitUsageQuotasAnyType\ntype WeightedRatelimitUsageQuotas struct {\n\t// RPS weights and total usage per key\n\tQuotas map[string]*WeightedRatelimitUsageQuotaEntry `json:\"quotas,required\"`\n}\n\ntype _Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList map[string]*WeightedRatelimitUsageQuotaEntry\n\nfunc (m _Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WeightedRatelimitUsageQuotaEntry', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList) Close() {}\n\n// ToWire translates a WeightedRatelimitUsageQuotas struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WeightedRatelimitUsageQuotas) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Quotas == nil {\n\t\treturn w, errors.New(\"field Quotas of WeightedRatelimitUsageQuotas is required\")\n\t}\n\tw, err = wire.NewValueMap(_Map_String_WeightedRatelimitUsageQuotaEntry_MapItemList(v.Quotas)), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WeightedRatelimitUsageQuotaEntry_Read(w wire.Value) (*WeightedRatelimitUsageQuotaEntry, error) {\n\tvar v WeightedRatelimitUsageQuotaEntry\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WeightedRatelimitUsageQuotaEntry_Read(m wire.MapItemList) (map[string]*WeightedRatelimitUsageQuotaEntry, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*WeightedRatelimitUsageQuotaEntry, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WeightedRatelimitUsageQuotaEntry_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a WeightedRatelimitUsageQuotas struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WeightedRatelimitUsageQuotas struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WeightedRatelimitUsageQuotas\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WeightedRatelimitUsageQuotas) FromWire(w wire.Value) error {\n\tvar err error\n\n\tquotasIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Quotas, err = _Map_String_WeightedRatelimitUsageQuotaEntry_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tquotasIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !quotasIsSet {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitUsageQuotas is required\")\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_WeightedRatelimitUsageQuotaEntry_Encode(val map[string]*WeightedRatelimitUsageQuotaEntry, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WeightedRatelimitUsageQuotaEntry', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a WeightedRatelimitUsageQuotas struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WeightedRatelimitUsageQuotas struct could not be encoded.\nfunc (v *WeightedRatelimitUsageQuotas) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Quotas == nil {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitUsageQuotas is required\")\n\t}\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\treturn err\n\t}\n\tif err := _Map_String_WeightedRatelimitUsageQuotaEntry_Encode(v.Quotas, sw); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WeightedRatelimitUsageQuotaEntry_Decode(sr stream.Reader) (*WeightedRatelimitUsageQuotaEntry, error) {\n\tvar v WeightedRatelimitUsageQuotaEntry\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WeightedRatelimitUsageQuotaEntry_Decode(sr stream.Reader) (map[string]*WeightedRatelimitUsageQuotaEntry, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*WeightedRatelimitUsageQuotaEntry, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WeightedRatelimitUsageQuotaEntry_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a WeightedRatelimitUsageQuotas struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WeightedRatelimitUsageQuotas struct could not be generated from the wire\n// representation.\nfunc (v *WeightedRatelimitUsageQuotas) Decode(sr stream.Reader) error {\n\n\tquotasIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Quotas, err = _Map_String_WeightedRatelimitUsageQuotaEntry_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tquotasIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !quotasIsSet {\n\t\treturn errors.New(\"field Quotas of WeightedRatelimitUsageQuotas is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WeightedRatelimitUsageQuotas\n// struct.\nfunc (v *WeightedRatelimitUsageQuotas) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Quotas: %v\", v.Quotas)\n\ti++\n\n\treturn fmt.Sprintf(\"WeightedRatelimitUsageQuotas{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_WeightedRatelimitUsageQuotaEntry_Equals(lhs, rhs map[string]*WeightedRatelimitUsageQuotaEntry) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this WeightedRatelimitUsageQuotas match the\n// provided WeightedRatelimitUsageQuotas.\n//\n// This function performs a deep comparison.\nfunc (v *WeightedRatelimitUsageQuotas) Equals(rhs *WeightedRatelimitUsageQuotas) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Map_String_WeightedRatelimitUsageQuotaEntry_Equals(v.Quotas, rhs.Quotas) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_WeightedRatelimitUsageQuotaEntry_Zapper map[string]*WeightedRatelimitUsageQuotaEntry\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WeightedRatelimitUsageQuotaEntry_Zapper.\nfunc (m _Map_String_WeightedRatelimitUsageQuotaEntry_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WeightedRatelimitUsageQuotas.\nfunc (v *WeightedRatelimitUsageQuotas) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\terr = multierr.Append(err, enc.AddObject(\"quotas\", (_Map_String_WeightedRatelimitUsageQuotaEntry_Zapper)(v.Quotas)))\n\treturn err\n}\n\n// GetQuotas returns the value of Quotas if it is set or its\n// zero value if it is unset.\nfunc (v *WeightedRatelimitUsageQuotas) GetQuotas() (o map[string]*WeightedRatelimitUsageQuotaEntry) {\n\tif v != nil {\n\t\to = v.Quotas\n\t}\n\treturn\n}\n\n// IsSetQuotas returns true if Quotas is not nil.\nfunc (v *WeightedRatelimitUsageQuotas) IsSetQuotas() bool {\n\treturn v != nil && v.Quotas != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"history\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/history\",\n\tFilePath: \"history.thrift\",\n\tSHA1:     \"4f9bc03480287dbdd2c4c579e95be64a2750f07a\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\treplicator.ThriftModule,\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\ninclude \\\"shared.thrift\\\"\\ninclude \\\"replicator.thrift\\\"\\n\\nnamespace java com.uber.cadence.history\\n\\nexception EventAlreadyStartedError {\\n  1: required string message\\n}\\n\\nexception ShardOwnershipLostError {\\n  10: optional string message\\n  20: optional string owner\\n}\\n\\nstruct ParentExecutionInfo {\\n  10: optional string domainUUID\\n  15: optional string domain\\n  20: optional shared.WorkflowExecution execution\\n  30: optional i64 (js.type = \\\"Long\\\") initiatedId\\n}\\n\\nstruct StartWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.StartWorkflowExecutionRequest startRequest\\n  30: optional ParentExecutionInfo parentExecutionInfo\\n  40: optional i32 attempt\\n  50: optional i64 (js.type = \\\"Long\\\") expirationTimestamp\\n  55: optional shared.ContinueAsNewInitiator continueAsNewInitiator\\n  56: optional string continuedFailureReason\\n  57: optional binary continuedFailureDetails\\n  58: optional binary lastCompletionResult\\n  60: optional i32 firstDecisionTaskBackoffSeconds\\n  62: optional map<string, string> partitionConfig\\n}\\n\\nstruct DescribeMutableStateRequest{\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n}\\n\\nstruct DescribeMutableStateResponse{\\n  30: optional string mutableStateInCache\\n  40: optional string mutableStateInDatabase\\n}\\n\\nstruct GetMutableStateRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n  30: optional i64 (js.type = \\\"Long\\\") expectedNextEventId\\n  40: optional binary currentBranchToken\\n  50: optional shared.VersionHistoryItem versionHistoryItem\\n}\\n\\nstruct GetMutableStateResponse {\\n  10: optional shared.WorkflowExecution execution\\n  20: optional shared.WorkflowType workflowType\\n  30: optional i64 (js.type = \\\"Long\\\") NextEventId\\n  35: optional i64 (js.type = \\\"Long\\\") PreviousStartedEventId\\n  40: optional i64 (js.type = \\\"Long\\\") LastFirstEventId\\n  50: optional shared.TaskList taskList\\n  60: optional shared.TaskList stickyTaskList\\n  70: optional string clientLibraryVersion\\n  80: optional string clientFeatureVersion\\n  90: optional string clientImpl\\n  //TODO: isWorkflowRunning is deprecating. workflowState is going replace this field\\n  100: optional bool isWorkflowRunning\\n  110: optional i32 stickyTaskListScheduleToStartTimeout\\n  120: optional i32 eventStoreVersion\\n  130: optional binary currentBranchToken\\n  // TODO: when migrating to gRPC, make this a enum\\n  // TODO: when migrating to gRPC, unify internal & external representation\\n  // NOTE: workflowState & workflowCloseState are the same as persistence representation\\n  150: optional i32 workflowState\\n  160: optional i32 workflowCloseState\\n  170: optional shared.VersionHistories versionHistories\\n  180: optional bool isStickyTaskListEnabled\\n  190: optional i64 (js.type = \\\"Long\\\") historySize\\n}\\n\\nstruct PollMutableStateRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n  30: optional i64 (js.type = \\\"Long\\\") expectedNextEventId\\n  40: optional binary currentBranchToken\\n}\\n\\nstruct PollMutableStateResponse {\\n  10: optional shared.WorkflowExecution execution\\n  20: optional shared.WorkflowType workflowType\\n  30: optional i64 (js.type = \\\"Long\\\") NextEventId\\n  35: optional i64 (js.type = \\\"Long\\\") PreviousStartedEventId\\n  40: optional i64 (js.type = \\\"Long\\\") LastFirstEventId\\n  50: optional shared.TaskList taskList\\n  60: optional shared.TaskList stickyTaskList\\n  70: optional string clientLibraryVersion\\n  80: optional string clientFeatureVersion\\n  90: optional string clientImpl\\n  100: optional i32 stickyTaskListScheduleToStartTimeout\\n  110: optional binary currentBranchToken\\n  130: optional shared.VersionHistories versionHistories\\n  // TODO: when migrating to gRPC, make this a enum\\n  // TODO: when migrating to gRPC, unify internal & external representation\\n  // NOTE: workflowState & workflowCloseState are the same as persistence representation\\n  140: optional i32 workflowState\\n  150: optional i32 workflowCloseState\\n}\\n\\nstruct ResetStickyTaskListRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n}\\n\\nstruct ResetStickyTaskListResponse {\\n  // The reason to keep this response is to allow returning\\n  // information in the future.\\n}\\n\\nstruct RespondDecisionTaskCompletedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RespondDecisionTaskCompletedRequest completeRequest\\n}\\n\\nstruct RespondDecisionTaskCompletedResponse {\\n  10: optional RecordDecisionTaskStartedResponse startedResponse\\n  20: optional map<string,shared.ActivityLocalDispatchInfo> activitiesToDispatchLocally\\n}\\n\\nstruct RespondDecisionTaskFailedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RespondDecisionTaskFailedRequest failedRequest\\n}\\n\\nstruct RecordActivityTaskHeartbeatRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RecordActivityTaskHeartbeatRequest heartbeatRequest\\n}\\n\\nstruct RespondActivityTaskCompletedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RespondActivityTaskCompletedRequest completeRequest\\n}\\n\\nstruct RespondActivityTaskFailedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RespondActivityTaskFailedRequest failedRequest\\n}\\n\\nstruct RespondActivityTaskCanceledRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RespondActivityTaskCanceledRequest cancelRequest\\n}\\n\\nstruct RefreshWorkflowTasksRequest {\\n  10: optional string domainUIID\\n  20: optional shared.RefreshWorkflowTasksRequest request\\n}\\n\\nstruct RecordActivityTaskStartedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional i64 (js.type = \\\"Long\\\") scheduleId\\n  40: optional i64 (js.type = \\\"Long\\\") taskId\\n  45: optional string requestId // Unique id of each poll request. Used to ensure at most once delivery of tasks.\\n  50: optional shared.PollForActivityTaskRequest pollRequest\\n}\\n\\nstruct RecordActivityTaskStartedResponse {\\n  20: optional shared.HistoryEvent scheduledEvent\\n  30: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  40: optional i64 (js.type = \\\"Long\\\") attempt\\n  50: optional i64 (js.type = \\\"Long\\\") scheduledTimestampOfThisAttempt\\n  60: optional binary heartbeatDetails\\n  70: optional shared.WorkflowType workflowType\\n  80: optional string workflowDomain\\n}\\n\\nstruct RecordDecisionTaskStartedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional i64 (js.type = \\\"Long\\\") scheduleId\\n  40: optional i64 (js.type = \\\"Long\\\") taskId\\n  45: optional string requestId // Unique id of each poll request. Used to ensure at most once delivery of tasks.\\n  50: optional shared.PollForDecisionTaskRequest pollRequest\\n}\\n\\nstruct RecordDecisionTaskStartedResponse {\\n  10: optional shared.WorkflowType workflowType\\n  20: optional i64 (js.type = \\\"Long\\\") previousStartedEventId\\n  30: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  40: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  50: optional i64 (js.type = \\\"Long\\\") nextEventId\\n  60: optional i64 (js.type = \\\"Long\\\") attempt\\n  70: optional bool stickyExecutionEnabled\\n  80: optional shared.TransientDecisionInfo decisionInfo\\n  90: optional shared.TaskList WorkflowExecutionTaskList\\n  100: optional i32 eventStoreVersion\\n  110: optional binary branchToken\\n  120: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  130: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  140: optional map<string, shared.WorkflowQuery> queries\\n  150: optional i64 (js.type = \\\"Long\\\") historySize\\n}\\n\\nstruct SignalWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.SignalWorkflowExecutionRequest signalRequest\\n  // workflow execution that requests this signal, for making sure\\n  // the workflow being signaled is actually a child of the workflow\\n  // making the request\\n  30: optional shared.WorkflowExecution externalWorkflowExecution\\n  40: optional bool childWorkflowOnly\\n}\\n\\nstruct SignalWithStartWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.SignalWithStartWorkflowExecutionRequest signalWithStartRequest\\n  30: optional map<string, string> partitionConfig\\n}\\n\\nstruct RemoveSignalMutableStateRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional string requestId\\n}\\n\\nstruct TerminateWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.TerminateWorkflowExecutionRequest terminateRequest\\n  // workflow execution that requests this termination, for making sure\\n  // the workflow being terminated is actually a child of the workflow\\n  // making the request\\n  30: optional shared.WorkflowExecution externalWorkflowExecution\\n  40: optional bool childWorkflowOnly\\n}\\n\\nstruct ResetWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.ResetWorkflowExecutionRequest resetRequest\\n}\\n\\nstruct RequestCancelWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.RequestCancelWorkflowExecutionRequest cancelRequest\\n  // workflow execution that requests this cancellation, for making sure\\n  // the workflow being cancelled is actually a child of the workflow\\n  // making the request\\n  30: optional i64 (js.type = \\\"Long\\\") externalInitiatedEventId\\n  40: optional shared.WorkflowExecution externalWorkflowExecution\\n  50: optional bool childWorkflowOnly\\n}\\n\\nstruct ScheduleDecisionTaskRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional bool isFirstDecision\\n}\\n\\nstruct DescribeWorkflowExecutionRequest {\\n  10: optional string domainUUID\\n  20: optional shared.DescribeWorkflowExecutionRequest request\\n}\\n\\n/**\\n* RecordChildExecutionCompletedRequest is used for reporting the completion of child execution to parent workflow\\n* execution which started it.  When a child execution is completed it creates this request and calls the\\n* RecordChildExecutionCompleted API with the workflowExecution of parent.  It also sets the completedExecution of the\\n* child as it could potentially be different than the ChildExecutionStartedEvent of parent in the situation when\\n* child creates multiple runs through ContinueAsNew before finally completing.\\n**/\\nstruct RecordChildExecutionCompletedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional i64 (js.type = \\\"Long\\\") initiatedId\\n  40: optional shared.WorkflowExecution completedExecution\\n  50: optional shared.HistoryEvent completionEvent\\n  60: optional i64 (js.type = \\\"Long\\\") startedId\\n}\\n\\nstruct ReplicateEventsV2Request {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional list<shared.VersionHistoryItem> versionHistoryItems\\n  40: optional shared.DataBlob events\\n  // new run events does not need version history since there is no prior events\\n  60: optional shared.DataBlob newRunEvents\\n}\\n\\nstruct SyncShardStatusRequest {\\n  10: optional string sourceCluster\\n  20: optional i64 (js.type = \\\"Long\\\") shardId\\n  30: optional i64 (js.type = \\\"Long\\\") timestamp\\n}\\n\\nstruct SyncActivityRequest {\\n  10: optional string domainId\\n  20: optional string workflowId\\n  30: optional string runId\\n  40: optional i64 (js.type = \\\"Long\\\") version\\n  50: optional i64 (js.type = \\\"Long\\\") scheduledId\\n  60: optional i64 (js.type = \\\"Long\\\") scheduledTime\\n  70: optional i64 (js.type = \\\"Long\\\") startedId\\n  80: optional i64 (js.type = \\\"Long\\\") startedTime\\n  90: optional i64 (js.type = \\\"Long\\\") lastHeartbeatTime\\n  100: optional binary details\\n  110: optional i32 attempt\\n  120: optional string lastFailureReason\\n  130: optional string lastWorkerIdentity\\n  140: optional binary lastFailureDetails\\n  150: optional shared.VersionHistory versionHistory\\n}\\n\\nstruct QueryWorkflowRequest {\\n  10: optional string domainUUID\\n  20: optional shared.QueryWorkflowRequest request\\n}\\n\\nstruct QueryWorkflowResponse {\\n  10: optional shared.QueryWorkflowResponse response\\n}\\n\\nstruct ReapplyEventsRequest {\\n  10: optional string domainUUID\\n  20: optional shared.ReapplyEventsRequest request\\n}\\n\\nstruct FailoverMarkerToken {\\n  10: optional list<i32> shardIDs\\n  20: optional replicator.FailoverMarkerAttributes failoverMarker\\n}\\n\\nstruct NotifyFailoverMarkersRequest {\\n  10: optional list<FailoverMarkerToken> failoverMarkerTokens\\n}\\n\\nstruct ProcessingQueueStates {\\n  10: optional map<string, list<ProcessingQueueState>> statesByCluster\\n}\\n\\nstruct ProcessingQueueState {\\n  10: optional i32 level\\n  20: optional i64 ackLevel\\n  30: optional i64 maxLevel\\n  40: optional DomainFilter domainFilter\\n}\\n\\nstruct DomainFilter {\\n  10: optional list<string> domainIDs\\n  20: optional bool reverseMatch\\n}\\n\\nstruct GetFailoverInfoRequest {\\n  10: optional string domainID\\n}\\n\\nstruct GetFailoverInfoResponse {\\n  10: optional i32 completedShardCount\\n  20: optional list<i32> pendingShards\\n}\\n\\nstruct RatelimitUpdateRequest {\\n  /**\\n  * impl-specific data.\\n  *\\n  * likely some simple top-level keys and then either:\\n  *   - map<ratelimit-key-string, something>\\n  *   - list<something>\\n  *\\n  * this is a single blob rather than a collection to save on\\n  * repeated serialization of the type name, and to allow impls\\n  * to choose whatever structures are most-convenient for them.\\n  */\\n  10: optional shared.Any data\\n}\\n\\nstruct RatelimitUpdateResponse {\\n  /**\\n  * impl-specific data.\\n  *\\n  * likely some simple top-level keys and then either:\\n  *   - map<ratelimit-key-string, something>\\n  *   - list<something>\\n  *\\n  * this is a single blob rather than a collection to save on\\n  * repeated serialization of the type name, and to allow impls\\n  * to choose whatever structures are most-convenient for them.\\n  */\\n  10: optional shared.Any data\\n}\\n\\n/**\\n* first impl of ratelimiting data, collected by limiters and sent to aggregators.\\n*\\n* used in an Any with ValueType: WeightedRatelimitUsageAnyType\\n*/\\nstruct WeightedRatelimitUsage {\\n  /** unique, stable identifier of the calling host, to identify future data from the same host */\\n  10: required string caller\\n  /** milliseconds since last update call.  expected to be on the order of a few seconds or less. */\\n  20: required i32 elapsedMS\\n  /** per key, number of allowed vs rejected calls since last update. */\\n  30: required map<string, WeightedRatelimitCalls> calls\\n}\\n\\n/** Any{ValueType} identifier for WeightedRatelimitUsage data */\\nconst string WeightedRatelimitUsageAnyType = \\\"cadence:loadbalanced:update_request\\\"\\n\\n/** fields are required to encourage compact serialization, zeros are expected */\\nstruct WeightedRatelimitCalls {\\n  /**\\n  * number of allowed requests since last call.\\n  * assumed to be <1m or so, saturates at MAX_INT32.\\n  */\\n  10: required i32 allowed\\n  /**\\n  * number of rejected requests since last call.\\n  * assumed to be <1m or so, saturates at MAX_INT32.\\n  */\\n  20: required i32 rejected\\n}\\n\\n/**\\n* first impl of ratelimiting data, result from aggregator to limiter.\\n*\\n* used in an Any with ValueType: WeightedRatelimitQuotasAnyType\\n*/\\nstruct WeightedRatelimitQuotas {\\n  /** RPS-weights to allow per key */\\n  10: required map<string,double> quotas\\n}\\n\\n/** Any{ValueType} identifier for WeightedRatelimitQuotas data */\\nconst string WeightedRatelimitQuotasAnyType = \\\"cadence:loadbalanced:update_response\\\"\\n\\n/**\\n* second impl, includes unused-RPS data so limiters can decide if they\\n* want to allow exceeding limits when there is free space.\\n*\\n* used in an Any with ValueType: WeightedRatelimitUsageQuotasAnyType\\n*/\\nstruct WeightedRatelimitUsageQuotas {\\n  /** RPS weights and total usage per key */\\n  10: required map<string,WeightedRatelimitUsageQuotaEntry> quotas\\n}\\n\\nstruct WeightedRatelimitUsageQuotaEntry {\\n  /** Amount of the quota that the receiving host can use, between 0 and 1 */\\n  10: required double weight\\n  /** RPS estimated across the whole cluster */\\n  20: required double used\\n}\\n\\nconst string WeightedRatelimitUsageQuotasAnyType = \\\"cadence:loadbalanced:update_response_used\\\"\\n\\n/**\\n* HistoryService provides API to start a new long running workflow instance, as well as query and update the history\\n* of workflow instances already created.\\n**/\\nservice HistoryService {\\n  /**\\n  * StartWorkflowExecution starts a new long running workflow instance.  It will create the instance with\\n  * 'WorkflowExecutionStarted' event in history and also schedule the first DecisionTask for the worker to make the\\n  * first decision for this instance.  It will return 'WorkflowExecutionAlreadyStartedError', if an instance already\\n  * exists with same workflowId.\\n  **/\\n  shared.StartWorkflowExecutionResponse StartWorkflowExecution(1: StartWorkflowExecutionRequest startRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.WorkflowExecutionAlreadyStartedError sessionAlreadyExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * Returns the information from mutable state of workflow execution.\\n  * It fails with 'EntityNotExistError' if specified workflow execution in unknown to the service.\\n  * It returns CurrentBranchChangedError if the workflow version branch has changed.\\n  **/\\n  GetMutableStateResponse GetMutableState(1: GetMutableStateRequest getRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.CurrentBranchChangedError currentBranchChangedError,\\n    )\\n\\n  /**\\n   * Returns the information from mutable state of workflow execution.\\n   * It fails with 'EntityNotExistError' if specified workflow execution in unknown to the service.\\n   * It returns CurrentBranchChangedError if the workflow version branch has changed.\\n   **/\\n   PollMutableStateResponse PollMutableState(1: PollMutableStateRequest pollRequest)\\n     throws (\\n       1: shared.BadRequestError badRequestError,\\n       2: shared.InternalServiceError internalServiceError,\\n       3: shared.EntityNotExistsError entityNotExistError,\\n       4: ShardOwnershipLostError shardOwnershipLostError,\\n       5: shared.LimitExceededError limitExceededError,\\n       6: shared.ServiceBusyError serviceBusyError,\\n       7: shared.CurrentBranchChangedError currentBranchChangedError,\\n     )\\n\\n  /**\\n  * Reset the sticky tasklist related information in mutable state of a given workflow.\\n  * Things cleared are:\\n  * 1. StickyTaskList\\n  * 2. StickyScheduleToStartTimeout\\n  * 3. ClientLibraryVersion\\n  * 4. ClientFeatureVersion\\n  * 5. ClientImpl\\n  **/\\n  ResetStickyTaskListResponse ResetStickyTaskList(1: ResetStickyTaskListRequest resetRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RecordDecisionTaskStarted is called by the Matchingservice before it hands a decision task to the application worker in response to\\n  * a PollForDecisionTask call. It records in the history the event that the decision task has started. It will return 'EventAlreadyStartedError',\\n  * if the workflow's execution history already includes a record of the event starting.\\n  **/\\n  RecordDecisionTaskStartedResponse RecordDecisionTaskStarted(1: RecordDecisionTaskStartedRequest addRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: EventAlreadyStartedError eventAlreadyStartedError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.LimitExceededError limitExceededError,\\n      8: shared.ServiceBusyError serviceBusyError,\\n      9: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RecordActivityTaskStarted is called by the Matchingservice before it hands a decision task to the application worker in response to\\n  * a PollForActivityTask call. It records in the history the event that the decision task has started. It will return 'EventAlreadyStartedError',\\n  * if the workflow's execution history already includes a record of the event starting.\\n  **/\\n  RecordActivityTaskStartedResponse RecordActivityTaskStarted(1: RecordActivityTaskStartedRequest addRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: EventAlreadyStartedError eventAlreadyStartedError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.LimitExceededError limitExceededError,\\n      8: shared.ServiceBusyError serviceBusyError,\\n      9: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RespondDecisionTaskCompleted is called by application worker to complete a DecisionTask handed as a result of\\n  * 'PollForDecisionTask' API call.  Completing a DecisionTask will result in new events for the workflow execution and\\n  * potentially new ActivityTask being created for corresponding decisions.  It will also create a DecisionTaskCompleted\\n  * event in the history for that session.  Use the 'taskToken' provided as response of PollForDecisionTask API call\\n  * for completing the DecisionTask.\\n  **/\\n  RespondDecisionTaskCompletedResponse RespondDecisionTaskCompleted(1: RespondDecisionTaskCompletedRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RespondDecisionTaskFailed is called by application worker to indicate failure.  This results in\\n  * DecisionTaskFailedEvent written to the history and a new DecisionTask created.  This API can be used by client to\\n  * either clear sticky tasklist or report ny panics during DecisionTask processing.\\n  **/\\n  void RespondDecisionTaskFailed(1: RespondDecisionTaskFailedRequest failedRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RecordActivityTaskHeartbeat is called by application worker while it is processing an ActivityTask.  If worker fails\\n  * to heartbeat within 'heartbeatTimeoutSeconds' interval for the ActivityTask, then it will be marked as timedout and\\n  * 'ActivityTaskTimedOut' event will be written to the workflow history.  Calling 'RecordActivityTaskHeartbeat' will\\n  * fail with 'EntityNotExistsError' in such situations.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for heartbeating.\\n  **/\\n  shared.RecordActivityTaskHeartbeatResponse RecordActivityTaskHeartbeat(1: RecordActivityTaskHeartbeatRequest heartbeatRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCompleted is called by application worker when it is done processing an ActivityTask.  It will\\n  * result in a new 'ActivityTaskCompleted' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void  RespondActivityTaskCompleted(1: RespondActivityTaskCompletedRequest completeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskFailed is called by application worker when it is done processing an ActivityTask.  It will\\n  * result in a new 'ActivityTaskFailed' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void RespondActivityTaskFailed(1: RespondActivityTaskFailedRequest failRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RespondActivityTaskCanceled is called by application worker when it is successfully canceled an ActivityTask.  It will\\n  * result in a new 'ActivityTaskCanceled' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\\n  * PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\\n  * anymore due to activity timeout.\\n  **/\\n  void RespondActivityTaskCanceled(1: RespondActivityTaskCanceledRequest canceledRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * SignalWorkflowExecution is used to send a signal event to running workflow execution.  This results in\\n  * WorkflowExecutionSignaled event recorded in the history and a decision task being created for the execution.\\n  **/\\n  void SignalWorkflowExecution(1: SignalWorkflowExecutionRequest signalRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.LimitExceededError limitExceededError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * SignalWithStartWorkflowExecution is used to ensure sending a signal event to a workflow execution.\\n  * If workflow is running, this results in WorkflowExecutionSignaled event recorded in the history\\n  * and a decision task being created for the execution.\\n  * If workflow is not running or not found, it will first try start workflow with given WorkflowIDResuePolicy,\\n  * and record WorkflowExecutionStarted and WorkflowExecutionSignaled event in case of success.\\n  * It will return `WorkflowExecutionAlreadyStartedError` if start workflow failed with given policy.\\n  **/\\n  shared.StartWorkflowExecutionResponse SignalWithStartWorkflowExecution(1: SignalWithStartWorkflowExecutionRequest signalWithStartRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: ShardOwnershipLostError shardOwnershipLostError,\\n      4: shared.DomainNotActiveError domainNotActiveError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.WorkflowExecutionAlreadyStartedError workflowAlreadyStartedError,\\n    )\\n\\n  /**\\n  * RemoveSignalMutableState is used to remove a signal request ID that was previously recorded.  This is currently\\n  * used to clean execution info when signal decision finished.\\n  **/\\n  void RemoveSignalMutableState(1: RemoveSignalMutableStateRequest removeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * TerminateWorkflowExecution terminates an existing workflow execution by recording WorkflowExecutionTerminated event\\n  * in the history and immediately terminating the execution instance.\\n  **/\\n  void TerminateWorkflowExecution(1: TerminateWorkflowExecutionRequest terminateRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * ResetWorkflowExecution reset an existing workflow execution by a firstEventID of a existing event batch\\n  * in the history and immediately terminating the current execution instance.\\n  * After reset, the history will grow from nextFirstEventID.\\n  **/\\n  shared.ResetWorkflowExecutionResponse ResetWorkflowExecution(1: ResetWorkflowExecutionRequest resetRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * RequestCancelWorkflowExecution is called by application worker when it wants to request cancellation of a workflow instance.\\n  * It will result in a new 'WorkflowExecutionCancelRequested' event being written to the workflow history and a new DecisionTask\\n  * created for the workflow instance so new decisions could be made. It fails with\\n  * 'WorkflowExecutionAlreadyCompletedError' if the workflow is not valid\\n  * anymore due to completion or with 'EntityNotExistsError' if worfklow doesn't exist.\\n  **/\\n  void RequestCancelWorkflowExecution(1: RequestCancelWorkflowExecutionRequest cancelRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.CancellationAlreadyRequestedError cancellationAlreadyRequestedError,\\n      6: shared.DomainNotActiveError domainNotActiveError,\\n      7: shared.LimitExceededError limitExceededError,\\n      8: shared.ServiceBusyError serviceBusyError,\\n      10: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * ScheduleDecisionTask is used for creating a decision task for already started workflow execution.  This is mainly\\n  * used by transfer queue processor during the processing of StartChildWorkflowExecution task, where it first starts\\n  * child execution without creating the decision task and then calls this API after updating the mutable state of\\n  * parent execution.\\n  **/\\n  void ScheduleDecisionTask(1: ScheduleDecisionTaskRequest scheduleRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * RecordChildExecutionCompleted is used for reporting the completion of child workflow execution to parent.\\n  * This is mainly called by transfer queue processor during the processing of DeleteExecution task.\\n  **/\\n  void RecordChildExecutionCompleted(1: RecordChildExecutionCompletedRequest completionRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.LimitExceededError limitExceededError,\\n      7: shared.ServiceBusyError serviceBusyError,\\n      8: shared.WorkflowExecutionAlreadyCompletedError workflowExecutionAlreadyCompletedError,\\n    )\\n\\n  /**\\n  * DescribeWorkflowExecution returns information about the specified workflow execution.\\n  **/\\n  shared.DescribeWorkflowExecutionResponse DescribeWorkflowExecution(1: DescribeWorkflowExecutionRequest describeRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  void ReplicateEventsV2(1: ReplicateEventsV2Request replicateV2Request)\\n    throws (\\n        1: shared.BadRequestError badRequestError,\\n        2: shared.InternalServiceError internalServiceError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: ShardOwnershipLostError shardOwnershipLostError,\\n        5: shared.LimitExceededError limitExceededError,\\n        6: shared.RetryTaskV2Error retryTaskError,\\n        7: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * SyncShardStatus sync the status between shards\\n  **/\\n  void SyncShardStatus(1: SyncShardStatusRequest syncShardStatusRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * SyncActivity sync the activity status\\n  **/\\n  void SyncActivity(1: SyncActivityRequest syncActivityRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      7: shared.RetryTaskV2Error retryTaskV2Error,\\n    )\\n\\n  /**\\n  * DescribeMutableState returns information about the internal states of workflow mutable state.\\n  **/\\n  DescribeMutableStateResponse DescribeMutableState(1: DescribeMutableStateRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.AccessDeniedError accessDeniedError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n      6: shared.LimitExceededError limitExceededError,\\n    )\\n\\n  /**\\n  * DescribeHistoryHost returns information about the internal states of a history host\\n  **/\\n  shared.DescribeHistoryHostResponse DescribeHistoryHost(1: shared.DescribeHistoryHostRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * CloseShard close the shard\\n  **/\\n  void CloseShard(1: shared.CloseShardRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * RemoveTask remove task based on type, taskid, shardid\\n  **/\\n  void RemoveTask(1: shared.RemoveTaskRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * ResetQueue reset processing queue state based on cluster name and type\\n  **/\\n  void ResetQueue(1: shared.ResetQueueRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * DescribeQueue return queue states based on cluster name and type\\n  **/\\n  shared.DescribeQueueResponse DescribeQueue(1: shared.DescribeQueueRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.AccessDeniedError accessDeniedError,\\n    )\\n\\n  /**\\n  * GetReplicationMessages return replication messages based on the read level\\n  **/\\n  replicator.GetReplicationMessagesResponse GetReplicationMessages(1: replicator.GetReplicationMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n      5: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n    )\\n\\n  /**\\n  * GetDLQReplicationMessages return replication messages based on dlq info\\n  **/\\n  replicator.GetDLQReplicationMessagesResponse GetDLQReplicationMessages(1: replicator.GetDLQReplicationMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * QueryWorkflow returns query result for a specified workflow execution\\n  **/\\n  QueryWorkflowResponse QueryWorkflow(1: QueryWorkflowRequest queryRequest)\\n\\tthrows (\\n\\t  1: shared.BadRequestError badRequestError,\\n\\t  2: shared.InternalServiceError internalServiceError,\\n\\t  3: shared.EntityNotExistsError entityNotExistError,\\n\\t  4: shared.QueryFailedError queryFailedError,\\n\\t  5: shared.LimitExceededError limitExceededError,\\n\\t  6: shared.ServiceBusyError serviceBusyError,\\n\\t  7: shared.ClientVersionNotSupportedError clientVersionNotSupportedError,\\n\\t)\\n\\n  /**\\n  * ReapplyEvents applies stale events to the current workflow and current run\\n  **/\\n  void ReapplyEvents(1: ReapplyEventsRequest reapplyEventsRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.DomainNotActiveError domainNotActiveError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: ShardOwnershipLostError shardOwnershipLostError,\\n      7: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * RefreshWorkflowTasks refreshes all tasks of a workflow\\n  **/\\n  void RefreshWorkflowTasks(1: RefreshWorkflowTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.DomainNotActiveError domainNotActiveError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n      6: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * ReadDLQMessages returns messages from DLQ\\n  **/\\n  replicator.ReadDLQMessagesResponse ReadDLQMessages(1: replicator.ReadDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n    )\\n\\n  /**\\n  * PurgeDLQMessages purges messages from DLQ\\n  **/\\n  void PurgeDLQMessages(1: replicator.PurgeDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n    )\\n\\n  /**\\n  * MergeDLQMessages merges messages from DLQ\\n  **/\\n  replicator.MergeDLQMessagesResponse MergeDLQMessages(1: replicator.MergeDLQMessagesRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n      5: ShardOwnershipLostError shardOwnershipLostError,\\n    )\\n\\n  /**\\n  * NotifyFailoverMarkers sends failover marker to the failover coordinator\\n  **/\\n  void NotifyFailoverMarkers(1: NotifyFailoverMarkersRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * GetCrossClusterTasks fetches cross cluster tasks\\n  **/\\n  shared.GetCrossClusterTasksResponse GetCrossClusterTasks(1: shared.GetCrossClusterTasksRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * RespondCrossClusterTasksCompleted responds the result of processing cross cluster tasks\\n  **/\\n  shared.RespondCrossClusterTasksCompletedResponse RespondCrossClusterTasksCompleted(1: shared.RespondCrossClusterTasksCompletedRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n    )\\n\\n  /**\\n  * GetFailoverInfo responds the failover info about an on-going graceful failover\\n  **/\\n  GetFailoverInfoResponse GetFailoverInfo(1: GetFailoverInfoRequest request)\\n    throws (\\n      1: shared.InternalServiceError internalServiceError,\\n      2: shared.ServiceBusyError serviceBusyError,\\n      3: ShardOwnershipLostError shardOwnershipLostError,\\n      4: shared.EntityNotExistsError entityNotExistError,\\n    )\\n\\n  /**\\n  * RatelimitUpdate pushes global-ratelimiting data to aggregating hosts,\\n  * and returns data describing how to update the caller's ratelimits.\\n  *\\n  * For more details, see github.com/uber/cadence/common/quotas/global documentation.\\n  *\\n  * Request and response structures are intentionally loosely defined, to allow plugging\\n  * in externally-defined algorithms without changing protocol-level details.\\n  **/\\n  RatelimitUpdateResponse RatelimitUpdate(1: RatelimitUpdateRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: ShardOwnershipLostError shardOwnershipLostError,\\n    )\\n}\\n\"\n\n// HistoryService_CloseShard_Args represents the arguments for the HistoryService.CloseShard function.\n//\n// The arguments for CloseShard are sent and received over the wire as this struct.\ntype HistoryService_CloseShard_Args struct {\n\tRequest *shared.CloseShardRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_CloseShard_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_CloseShard_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CloseShardRequest_Read(w wire.Value) (*shared.CloseShardRequest, error) {\n\tvar v shared.CloseShardRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_CloseShard_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_CloseShard_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_CloseShard_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_CloseShard_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _CloseShardRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_CloseShard_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_CloseShard_Args struct could not be encoded.\nfunc (v *HistoryService_CloseShard_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CloseShardRequest_Decode(sr stream.Reader) (*shared.CloseShardRequest, error) {\n\tvar v shared.CloseShardRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_CloseShard_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_CloseShard_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_CloseShard_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _CloseShardRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_CloseShard_Args\n// struct.\nfunc (v *HistoryService_CloseShard_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_CloseShard_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_CloseShard_Args match the\n// provided HistoryService_CloseShard_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_CloseShard_Args) Equals(rhs *HistoryService_CloseShard_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_CloseShard_Args.\nfunc (v *HistoryService_CloseShard_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_CloseShard_Args) GetRequest() (o *shared.CloseShardRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_CloseShard_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"CloseShard\" for this struct.\nfunc (v *HistoryService_CloseShard_Args) MethodName() string {\n\treturn \"CloseShard\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_CloseShard_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_CloseShard_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.CloseShard\n// function.\nvar HistoryService_CloseShard_Helper = struct {\n\t// Args accepts the parameters of CloseShard in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.CloseShardRequest,\n\t) *HistoryService_CloseShard_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by CloseShard.\n\t//\n\t// An error can be thrown by CloseShard only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for CloseShard\n\t// given the error returned by it. The provided error may\n\t// be nil if CloseShard did not fail.\n\t//\n\t// This allows mapping errors returned by CloseShard into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// CloseShard\n\t//\n\t//   err := CloseShard(args)\n\t//   result, err := HistoryService_CloseShard_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from CloseShard: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_CloseShard_Result, error)\n\n\t// UnwrapResponse takes the result struct for CloseShard\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if CloseShard threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_CloseShard_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_CloseShard_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_CloseShard_Helper.Args = func(\n\t\trequest *shared.CloseShardRequest,\n\t) *HistoryService_CloseShard_Args {\n\t\treturn &HistoryService_CloseShard_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_CloseShard_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_CloseShard_Helper.WrapResponse = func(err error) (*HistoryService_CloseShard_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_CloseShard_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_CloseShard_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_CloseShard_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_CloseShard_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_CloseShard_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_CloseShard_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_CloseShard_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_CloseShard_Helper.UnwrapResponse = func(result *HistoryService_CloseShard_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_CloseShard_Result represents the result of a HistoryService.CloseShard function call.\n//\n// The result of a CloseShard execution is sent and received over the wire as this struct.\ntype HistoryService_CloseShard_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_CloseShard_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_CloseShard_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_CloseShard_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _BadRequestError_Read(w wire.Value) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Read(w wire.Value) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _AccessDeniedError_Read(w wire.Value) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_CloseShard_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_CloseShard_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_CloseShard_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_CloseShard_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_CloseShard_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_CloseShard_Result struct could not be encoded.\nfunc (v *HistoryService_CloseShard_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _BadRequestError_Decode(sr stream.Reader) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Decode(sr stream.Reader) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _AccessDeniedError_Decode(sr stream.Reader) (*shared.AccessDeniedError, error) {\n\tvar v shared.AccessDeniedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_CloseShard_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_CloseShard_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_CloseShard_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_CloseShard_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_CloseShard_Result\n// struct.\nfunc (v *HistoryService_CloseShard_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_CloseShard_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_CloseShard_Result match the\n// provided HistoryService_CloseShard_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_CloseShard_Result) Equals(rhs *HistoryService_CloseShard_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_CloseShard_Result.\nfunc (v *HistoryService_CloseShard_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_CloseShard_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_CloseShard_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_CloseShard_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_CloseShard_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_CloseShard_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_CloseShard_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"CloseShard\" for this struct.\nfunc (v *HistoryService_CloseShard_Result) MethodName() string {\n\treturn \"CloseShard\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_CloseShard_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_DescribeHistoryHost_Args represents the arguments for the HistoryService.DescribeHistoryHost function.\n//\n// The arguments for DescribeHistoryHost are sent and received over the wire as this struct.\ntype HistoryService_DescribeHistoryHost_Args struct {\n\tRequest *shared.DescribeHistoryHostRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeHistoryHost_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeHistoryHost_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeHistoryHostRequest_Read(w wire.Value) (*shared.DescribeHistoryHostRequest, error) {\n\tvar v shared.DescribeHistoryHostRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeHistoryHost_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeHistoryHost_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeHistoryHost_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeHistoryHost_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeHistoryHostRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeHistoryHost_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeHistoryHost_Args struct could not be encoded.\nfunc (v *HistoryService_DescribeHistoryHost_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeHistoryHostRequest_Decode(sr stream.Reader) (*shared.DescribeHistoryHostRequest, error) {\n\tvar v shared.DescribeHistoryHostRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeHistoryHost_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeHistoryHost_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeHistoryHost_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeHistoryHostRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeHistoryHost_Args\n// struct.\nfunc (v *HistoryService_DescribeHistoryHost_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeHistoryHost_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeHistoryHost_Args match the\n// provided HistoryService_DescribeHistoryHost_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeHistoryHost_Args) Equals(rhs *HistoryService_DescribeHistoryHost_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeHistoryHost_Args.\nfunc (v *HistoryService_DescribeHistoryHost_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeHistoryHost_Args) GetRequest() (o *shared.DescribeHistoryHostRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_DescribeHistoryHost_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeHistoryHost\" for this struct.\nfunc (v *HistoryService_DescribeHistoryHost_Args) MethodName() string {\n\treturn \"DescribeHistoryHost\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_DescribeHistoryHost_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_DescribeHistoryHost_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.DescribeHistoryHost\n// function.\nvar HistoryService_DescribeHistoryHost_Helper = struct {\n\t// Args accepts the parameters of DescribeHistoryHost in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeHistoryHostRequest,\n\t) *HistoryService_DescribeHistoryHost_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeHistoryHost.\n\t//\n\t// An error can be thrown by DescribeHistoryHost only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeHistoryHost\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeHistoryHost into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeHistoryHost\n\t//\n\t//   value, err := DescribeHistoryHost(args)\n\t//   result, err := HistoryService_DescribeHistoryHost_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeHistoryHost: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeHistoryHostResponse, error) (*HistoryService_DescribeHistoryHost_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeHistoryHost\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeHistoryHost threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_DescribeHistoryHost_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_DescribeHistoryHost_Result) (*shared.DescribeHistoryHostResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_DescribeHistoryHost_Helper.Args = func(\n\t\trequest *shared.DescribeHistoryHostRequest,\n\t) *HistoryService_DescribeHistoryHost_Args {\n\t\treturn &HistoryService_DescribeHistoryHost_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_DescribeHistoryHost_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_DescribeHistoryHost_Helper.WrapResponse = func(success *shared.DescribeHistoryHostResponse, err error) (*HistoryService_DescribeHistoryHost_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_DescribeHistoryHost_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeHistoryHost_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeHistoryHost_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeHistoryHost_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeHistoryHost_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeHistoryHost_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeHistoryHost_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_DescribeHistoryHost_Helper.UnwrapResponse = func(result *HistoryService_DescribeHistoryHost_Result) (success *shared.DescribeHistoryHostResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_DescribeHistoryHost_Result represents the result of a HistoryService.DescribeHistoryHost function call.\n//\n// The result of a DescribeHistoryHost execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_DescribeHistoryHost_Result struct {\n\t// Value returned by DescribeHistoryHost after a successful execution.\n\tSuccess              *shared.DescribeHistoryHostResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError           `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeHistoryHost_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeHistoryHost_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeHistoryHostResponse_Read(w wire.Value) (*shared.DescribeHistoryHostResponse, error) {\n\tvar v shared.DescribeHistoryHostResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeHistoryHost_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeHistoryHost_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeHistoryHost_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeHistoryHost_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeHistoryHostResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeHistoryHost_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeHistoryHost_Result struct could not be encoded.\nfunc (v *HistoryService_DescribeHistoryHost_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeHistoryHostResponse_Decode(sr stream.Reader) (*shared.DescribeHistoryHostResponse, error) {\n\tvar v shared.DescribeHistoryHostResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeHistoryHost_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeHistoryHost_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeHistoryHost_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeHistoryHostResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeHistoryHost_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeHistoryHost_Result\n// struct.\nfunc (v *HistoryService_DescribeHistoryHost_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeHistoryHost_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeHistoryHost_Result match the\n// provided HistoryService_DescribeHistoryHost_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeHistoryHost_Result) Equals(rhs *HistoryService_DescribeHistoryHost_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeHistoryHost_Result.\nfunc (v *HistoryService_DescribeHistoryHost_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeHistoryHost_Result) GetSuccess() (o *shared.DescribeHistoryHostResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_DescribeHistoryHost_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeHistoryHost_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_DescribeHistoryHost_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeHistoryHost_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_DescribeHistoryHost_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeHistoryHost_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_DescribeHistoryHost_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeHistoryHost\" for this struct.\nfunc (v *HistoryService_DescribeHistoryHost_Result) MethodName() string {\n\treturn \"DescribeHistoryHost\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_DescribeHistoryHost_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_DescribeMutableState_Args represents the arguments for the HistoryService.DescribeMutableState function.\n//\n// The arguments for DescribeMutableState are sent and received over the wire as this struct.\ntype HistoryService_DescribeMutableState_Args struct {\n\tRequest *DescribeMutableStateRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeMutableState_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeMutableState_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeMutableStateRequest_Read(w wire.Value) (*DescribeMutableStateRequest, error) {\n\tvar v DescribeMutableStateRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeMutableState_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeMutableState_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeMutableState_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeMutableState_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeMutableStateRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeMutableState_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeMutableState_Args struct could not be encoded.\nfunc (v *HistoryService_DescribeMutableState_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeMutableStateRequest_Decode(sr stream.Reader) (*DescribeMutableStateRequest, error) {\n\tvar v DescribeMutableStateRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeMutableState_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeMutableState_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeMutableState_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeMutableStateRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeMutableState_Args\n// struct.\nfunc (v *HistoryService_DescribeMutableState_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeMutableState_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeMutableState_Args match the\n// provided HistoryService_DescribeMutableState_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeMutableState_Args) Equals(rhs *HistoryService_DescribeMutableState_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeMutableState_Args.\nfunc (v *HistoryService_DescribeMutableState_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Args) GetRequest() (o *DescribeMutableStateRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_DescribeMutableState_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeMutableState\" for this struct.\nfunc (v *HistoryService_DescribeMutableState_Args) MethodName() string {\n\treturn \"DescribeMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_DescribeMutableState_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_DescribeMutableState_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.DescribeMutableState\n// function.\nvar HistoryService_DescribeMutableState_Helper = struct {\n\t// Args accepts the parameters of DescribeMutableState in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *DescribeMutableStateRequest,\n\t) *HistoryService_DescribeMutableState_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeMutableState.\n\t//\n\t// An error can be thrown by DescribeMutableState only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeMutableState\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeMutableState into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeMutableState\n\t//\n\t//   value, err := DescribeMutableState(args)\n\t//   result, err := HistoryService_DescribeMutableState_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeMutableState: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*DescribeMutableStateResponse, error) (*HistoryService_DescribeMutableState_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeMutableState\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeMutableState threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_DescribeMutableState_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_DescribeMutableState_Result) (*DescribeMutableStateResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_DescribeMutableState_Helper.Args = func(\n\t\trequest *DescribeMutableStateRequest,\n\t) *HistoryService_DescribeMutableState_Args {\n\t\treturn &HistoryService_DescribeMutableState_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_DescribeMutableState_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_DescribeMutableState_Helper.WrapResponse = func(success *DescribeMutableStateResponse, err error) (*HistoryService_DescribeMutableState_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_DescribeMutableState_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{AccessDeniedError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeMutableState_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeMutableState_Result{LimitExceededError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_DescribeMutableState_Helper.UnwrapResponse = func(result *HistoryService_DescribeMutableState_Result) (success *DescribeMutableStateResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_DescribeMutableState_Result represents the result of a HistoryService.DescribeMutableState function call.\n//\n// The result of a DescribeMutableState execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_DescribeMutableState_Result struct {\n\t// Value returned by DescribeMutableState after a successful execution.\n\tSuccess                 *DescribeMutableStateResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError       `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError  `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError  `json:\"entityNotExistError,omitempty\"`\n\tAccessDeniedError       *shared.AccessDeniedError     `json:\"accessDeniedError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError      `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError    `json:\"limitExceededError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeMutableState_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeMutableState_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_DescribeMutableState_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeMutableStateResponse_Read(w wire.Value) (*DescribeMutableStateResponse, error) {\n\tvar v DescribeMutableStateResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Read(w wire.Value) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ShardOwnershipLostError_Read(w wire.Value) (*ShardOwnershipLostError, error) {\n\tvar v ShardOwnershipLostError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Read(w wire.Value) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeMutableState_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeMutableState_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeMutableState_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeMutableState_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeMutableStateResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeMutableState_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeMutableState_Result struct could not be encoded.\nfunc (v *HistoryService_DescribeMutableState_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeMutableStateResponse_Decode(sr stream.Reader) (*DescribeMutableStateResponse, error) {\n\tvar v DescribeMutableStateResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Decode(sr stream.Reader) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ShardOwnershipLostError_Decode(sr stream.Reader) (*ShardOwnershipLostError, error) {\n\tvar v ShardOwnershipLostError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Decode(sr stream.Reader) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeMutableState_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeMutableState_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeMutableState_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeMutableStateResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeMutableState_Result\n// struct.\nfunc (v *HistoryService_DescribeMutableState_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeMutableState_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeMutableState_Result match the\n// provided HistoryService_DescribeMutableState_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeMutableState_Result) Equals(rhs *HistoryService_DescribeMutableState_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeMutableState_Result.\nfunc (v *HistoryService_DescribeMutableState_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetSuccess() (o *DescribeMutableStateResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeMutableState_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_DescribeMutableState_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeMutableState\" for this struct.\nfunc (v *HistoryService_DescribeMutableState_Result) MethodName() string {\n\treturn \"DescribeMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_DescribeMutableState_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_DescribeQueue_Args represents the arguments for the HistoryService.DescribeQueue function.\n//\n// The arguments for DescribeQueue are sent and received over the wire as this struct.\ntype HistoryService_DescribeQueue_Args struct {\n\tRequest *shared.DescribeQueueRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeQueue_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeQueue_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeQueueRequest_Read(w wire.Value) (*shared.DescribeQueueRequest, error) {\n\tvar v shared.DescribeQueueRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeQueue_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeQueue_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeQueue_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeQueue_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeQueueRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeQueue_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeQueue_Args struct could not be encoded.\nfunc (v *HistoryService_DescribeQueue_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeQueueRequest_Decode(sr stream.Reader) (*shared.DescribeQueueRequest, error) {\n\tvar v shared.DescribeQueueRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeQueue_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeQueue_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeQueue_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeQueueRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeQueue_Args\n// struct.\nfunc (v *HistoryService_DescribeQueue_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeQueue_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeQueue_Args match the\n// provided HistoryService_DescribeQueue_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeQueue_Args) Equals(rhs *HistoryService_DescribeQueue_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeQueue_Args.\nfunc (v *HistoryService_DescribeQueue_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeQueue_Args) GetRequest() (o *shared.DescribeQueueRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_DescribeQueue_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeQueue\" for this struct.\nfunc (v *HistoryService_DescribeQueue_Args) MethodName() string {\n\treturn \"DescribeQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_DescribeQueue_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_DescribeQueue_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.DescribeQueue\n// function.\nvar HistoryService_DescribeQueue_Helper = struct {\n\t// Args accepts the parameters of DescribeQueue in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.DescribeQueueRequest,\n\t) *HistoryService_DescribeQueue_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeQueue.\n\t//\n\t// An error can be thrown by DescribeQueue only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeQueue\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeQueue into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeQueue\n\t//\n\t//   value, err := DescribeQueue(args)\n\t//   result, err := HistoryService_DescribeQueue_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeQueue: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeQueueResponse, error) (*HistoryService_DescribeQueue_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeQueue\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeQueue threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_DescribeQueue_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_DescribeQueue_Result) (*shared.DescribeQueueResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_DescribeQueue_Helper.Args = func(\n\t\trequest *shared.DescribeQueueRequest,\n\t) *HistoryService_DescribeQueue_Args {\n\t\treturn &HistoryService_DescribeQueue_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_DescribeQueue_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_DescribeQueue_Helper.WrapResponse = func(success *shared.DescribeQueueResponse, err error) (*HistoryService_DescribeQueue_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_DescribeQueue_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeQueue_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeQueue_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeQueue_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeQueue_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeQueue_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeQueue_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_DescribeQueue_Helper.UnwrapResponse = func(result *HistoryService_DescribeQueue_Result) (success *shared.DescribeQueueResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_DescribeQueue_Result represents the result of a HistoryService.DescribeQueue function call.\n//\n// The result of a DescribeQueue execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_DescribeQueue_Result struct {\n\t// Value returned by DescribeQueue after a successful execution.\n\tSuccess              *shared.DescribeQueueResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError       `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError  `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError     `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeQueue_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeQueue_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_DescribeQueue_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeQueueResponse_Read(w wire.Value) (*shared.DescribeQueueResponse, error) {\n\tvar v shared.DescribeQueueResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeQueue_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeQueue_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeQueue_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeQueue_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeQueueResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeQueue_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeQueue_Result struct could not be encoded.\nfunc (v *HistoryService_DescribeQueue_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeQueueResponse_Decode(sr stream.Reader) (*shared.DescribeQueueResponse, error) {\n\tvar v shared.DescribeQueueResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeQueue_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeQueue_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeQueue_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeQueueResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeQueue_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeQueue_Result\n// struct.\nfunc (v *HistoryService_DescribeQueue_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeQueue_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeQueue_Result match the\n// provided HistoryService_DescribeQueue_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeQueue_Result) Equals(rhs *HistoryService_DescribeQueue_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeQueue_Result.\nfunc (v *HistoryService_DescribeQueue_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeQueue_Result) GetSuccess() (o *shared.DescribeQueueResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_DescribeQueue_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeQueue_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_DescribeQueue_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeQueue_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_DescribeQueue_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeQueue_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_DescribeQueue_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeQueue\" for this struct.\nfunc (v *HistoryService_DescribeQueue_Result) MethodName() string {\n\treturn \"DescribeQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_DescribeQueue_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_DescribeWorkflowExecution_Args represents the arguments for the HistoryService.DescribeWorkflowExecution function.\n//\n// The arguments for DescribeWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_DescribeWorkflowExecution_Args struct {\n\tDescribeRequest *DescribeWorkflowExecutionRequest `json:\"describeRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DescribeRequest != nil {\n\t\tw, err = v.DescribeRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionRequest_1_Read(w wire.Value) (*DescribeWorkflowExecutionRequest, error) {\n\tvar v DescribeWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DescribeRequest, err = _DescribeWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DescribeRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DescribeRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*DescribeWorkflowExecutionRequest, error) {\n\tvar v DescribeWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.DescribeRequest, err = _DescribeWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DescribeRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DescribeRequest: %v\", v.DescribeRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeWorkflowExecution_Args match the\n// provided HistoryService_DescribeWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) Equals(rhs *HistoryService_DescribeWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DescribeRequest == nil && rhs.DescribeRequest == nil) || (v.DescribeRequest != nil && rhs.DescribeRequest != nil && v.DescribeRequest.Equals(rhs.DescribeRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeWorkflowExecution_Args.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DescribeRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"describeRequest\", v.DescribeRequest))\n\t}\n\treturn err\n}\n\n// GetDescribeRequest returns the value of DescribeRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) GetDescribeRequest() (o *DescribeWorkflowExecutionRequest) {\n\tif v != nil && v.DescribeRequest != nil {\n\t\treturn v.DescribeRequest\n\t}\n\n\treturn\n}\n\n// IsSetDescribeRequest returns true if DescribeRequest is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) IsSetDescribeRequest() bool {\n\treturn v != nil && v.DescribeRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_DescribeWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.DescribeWorkflowExecution\n// function.\nvar HistoryService_DescribeWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of DescribeWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tdescribeRequest *DescribeWorkflowExecutionRequest,\n\t) *HistoryService_DescribeWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeWorkflowExecution.\n\t//\n\t// An error can be thrown by DescribeWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeWorkflowExecution\n\t//\n\t//   value, err := DescribeWorkflowExecution(args)\n\t//   result, err := HistoryService_DescribeWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeWorkflowExecutionResponse, error) (*HistoryService_DescribeWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_DescribeWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_DescribeWorkflowExecution_Result) (*shared.DescribeWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_DescribeWorkflowExecution_Helper.Args = func(\n\t\tdescribeRequest *DescribeWorkflowExecutionRequest,\n\t) *HistoryService_DescribeWorkflowExecution_Args {\n\t\treturn &HistoryService_DescribeWorkflowExecution_Args{\n\t\t\tDescribeRequest: describeRequest,\n\t\t}\n\t}\n\n\tHistoryService_DescribeWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_DescribeWorkflowExecution_Helper.WrapResponse = func(success *shared.DescribeWorkflowExecutionResponse, err error) (*HistoryService_DescribeWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_DescribeWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_DescribeWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_DescribeWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_DescribeWorkflowExecution_Result) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_DescribeWorkflowExecution_Result represents the result of a HistoryService.DescribeWorkflowExecution function call.\n//\n// The result of a DescribeWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_DescribeWorkflowExecution_Result struct {\n\t// Value returned by DescribeWorkflowExecution after a successful execution.\n\tSuccess                 *shared.DescribeWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError                   `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError              `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError              `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError                  `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError                `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError                  `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_DescribeWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeWorkflowExecutionResponse_Read(w wire.Value) (*shared.DescribeWorkflowExecutionResponse, error) {\n\tvar v shared.DescribeWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Read(w wire.Value) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_DescribeWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_DescribeWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_DescribeWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_DescribeWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_DescribeWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.DescribeWorkflowExecutionResponse, error) {\n\tvar v shared.DescribeWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Decode(sr stream.Reader) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_DescribeWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_DescribeWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_DescribeWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_DescribeWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_DescribeWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_DescribeWorkflowExecution_Result match the\n// provided HistoryService_DescribeWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) Equals(rhs *HistoryService_DescribeWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_DescribeWorkflowExecution_Result.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetSuccess() (o *shared.DescribeWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeWorkflowExecution\" for this struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) MethodName() string {\n\treturn \"DescribeWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_DescribeWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_GetCrossClusterTasks_Args represents the arguments for the HistoryService.GetCrossClusterTasks function.\n//\n// The arguments for GetCrossClusterTasks are sent and received over the wire as this struct.\ntype HistoryService_GetCrossClusterTasks_Args struct {\n\tRequest *shared.GetCrossClusterTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetCrossClusterTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetCrossClusterTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetCrossClusterTasksRequest_Read(w wire.Value) (*shared.GetCrossClusterTasksRequest, error) {\n\tvar v shared.GetCrossClusterTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetCrossClusterTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetCrossClusterTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetCrossClusterTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetCrossClusterTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetCrossClusterTasksRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetCrossClusterTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetCrossClusterTasks_Args struct could not be encoded.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetCrossClusterTasksRequest_Decode(sr stream.Reader) (*shared.GetCrossClusterTasksRequest, error) {\n\tvar v shared.GetCrossClusterTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetCrossClusterTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetCrossClusterTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetCrossClusterTasksRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetCrossClusterTasks_Args\n// struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetCrossClusterTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetCrossClusterTasks_Args match the\n// provided HistoryService_GetCrossClusterTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) Equals(rhs *HistoryService_GetCrossClusterTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetCrossClusterTasks_Args.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) GetRequest() (o *shared.GetCrossClusterTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetCrossClusterTasks\" for this struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) MethodName() string {\n\treturn \"GetCrossClusterTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_GetCrossClusterTasks_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.GetCrossClusterTasks\n// function.\nvar HistoryService_GetCrossClusterTasks_Helper = struct {\n\t// Args accepts the parameters of GetCrossClusterTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.GetCrossClusterTasksRequest,\n\t) *HistoryService_GetCrossClusterTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetCrossClusterTasks.\n\t//\n\t// An error can be thrown by GetCrossClusterTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetCrossClusterTasks\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetCrossClusterTasks into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetCrossClusterTasks\n\t//\n\t//   value, err := GetCrossClusterTasks(args)\n\t//   result, err := HistoryService_GetCrossClusterTasks_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetCrossClusterTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetCrossClusterTasksResponse, error) (*HistoryService_GetCrossClusterTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetCrossClusterTasks\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetCrossClusterTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_GetCrossClusterTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_GetCrossClusterTasks_Result) (*shared.GetCrossClusterTasksResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_GetCrossClusterTasks_Helper.Args = func(\n\t\trequest *shared.GetCrossClusterTasksRequest,\n\t) *HistoryService_GetCrossClusterTasks_Args {\n\t\treturn &HistoryService_GetCrossClusterTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_GetCrossClusterTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_GetCrossClusterTasks_Helper.WrapResponse = func(success *shared.GetCrossClusterTasksResponse, err error) (*HistoryService_GetCrossClusterTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_GetCrossClusterTasks_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetCrossClusterTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetCrossClusterTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetCrossClusterTasks_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetCrossClusterTasks_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetCrossClusterTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetCrossClusterTasks_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_GetCrossClusterTasks_Helper.UnwrapResponse = func(result *HistoryService_GetCrossClusterTasks_Result) (success *shared.GetCrossClusterTasksResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_GetCrossClusterTasks_Result represents the result of a HistoryService.GetCrossClusterTasks function call.\n//\n// The result of a GetCrossClusterTasks execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_GetCrossClusterTasks_Result struct {\n\t// Value returned by GetCrossClusterTasks after a successful execution.\n\tSuccess              *shared.GetCrossClusterTasksResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetCrossClusterTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetCrossClusterTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetCrossClusterTasksResponse_Read(w wire.Value) (*shared.GetCrossClusterTasksResponse, error) {\n\tvar v shared.GetCrossClusterTasksResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetCrossClusterTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetCrossClusterTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetCrossClusterTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetCrossClusterTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetCrossClusterTasksResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetCrossClusterTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetCrossClusterTasks_Result struct could not be encoded.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetCrossClusterTasksResponse_Decode(sr stream.Reader) (*shared.GetCrossClusterTasksResponse, error) {\n\tvar v shared.GetCrossClusterTasksResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetCrossClusterTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetCrossClusterTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetCrossClusterTasksResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetCrossClusterTasks_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetCrossClusterTasks_Result\n// struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetCrossClusterTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetCrossClusterTasks_Result match the\n// provided HistoryService_GetCrossClusterTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) Equals(rhs *HistoryService_GetCrossClusterTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetCrossClusterTasks_Result.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) GetSuccess() (o *shared.GetCrossClusterTasksResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetCrossClusterTasks\" for this struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) MethodName() string {\n\treturn \"GetCrossClusterTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_GetCrossClusterTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_GetDLQReplicationMessages_Args represents the arguments for the HistoryService.GetDLQReplicationMessages function.\n//\n// The arguments for GetDLQReplicationMessages are sent and received over the wire as this struct.\ntype HistoryService_GetDLQReplicationMessages_Args struct {\n\tRequest *replicator.GetDLQReplicationMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetDLQReplicationMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDLQReplicationMessagesRequest_Read(w wire.Value) (*replicator.GetDLQReplicationMessagesRequest, error) {\n\tvar v replicator.GetDLQReplicationMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetDLQReplicationMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetDLQReplicationMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetDLQReplicationMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetDLQReplicationMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetDLQReplicationMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetDLQReplicationMessages_Args struct could not be encoded.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDLQReplicationMessagesRequest_Decode(sr stream.Reader) (*replicator.GetDLQReplicationMessagesRequest, error) {\n\tvar v replicator.GetDLQReplicationMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetDLQReplicationMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetDLQReplicationMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetDLQReplicationMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetDLQReplicationMessages_Args\n// struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetDLQReplicationMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetDLQReplicationMessages_Args match the\n// provided HistoryService_GetDLQReplicationMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) Equals(rhs *HistoryService_GetDLQReplicationMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetDLQReplicationMessages_Args.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) GetRequest() (o *replicator.GetDLQReplicationMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetDLQReplicationMessages\" for this struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) MethodName() string {\n\treturn \"GetDLQReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_GetDLQReplicationMessages_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.GetDLQReplicationMessages\n// function.\nvar HistoryService_GetDLQReplicationMessages_Helper = struct {\n\t// Args accepts the parameters of GetDLQReplicationMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.GetDLQReplicationMessagesRequest,\n\t) *HistoryService_GetDLQReplicationMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetDLQReplicationMessages.\n\t//\n\t// An error can be thrown by GetDLQReplicationMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetDLQReplicationMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetDLQReplicationMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetDLQReplicationMessages\n\t//\n\t//   value, err := GetDLQReplicationMessages(args)\n\t//   result, err := HistoryService_GetDLQReplicationMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetDLQReplicationMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.GetDLQReplicationMessagesResponse, error) (*HistoryService_GetDLQReplicationMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetDLQReplicationMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetDLQReplicationMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_GetDLQReplicationMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_GetDLQReplicationMessages_Result) (*replicator.GetDLQReplicationMessagesResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_GetDLQReplicationMessages_Helper.Args = func(\n\t\trequest *replicator.GetDLQReplicationMessagesRequest,\n\t) *HistoryService_GetDLQReplicationMessages_Args {\n\t\treturn &HistoryService_GetDLQReplicationMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_GetDLQReplicationMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_GetDLQReplicationMessages_Helper.WrapResponse = func(success *replicator.GetDLQReplicationMessagesResponse, err error) (*HistoryService_GetDLQReplicationMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_GetDLQReplicationMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetDLQReplicationMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetDLQReplicationMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetDLQReplicationMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetDLQReplicationMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetDLQReplicationMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetDLQReplicationMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetDLQReplicationMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetDLQReplicationMessages_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_GetDLQReplicationMessages_Helper.UnwrapResponse = func(result *HistoryService_GetDLQReplicationMessages_Result) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_GetDLQReplicationMessages_Result represents the result of a HistoryService.GetDLQReplicationMessages function call.\n//\n// The result of a GetDLQReplicationMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_GetDLQReplicationMessages_Result struct {\n\t// Value returned by GetDLQReplicationMessages after a successful execution.\n\tSuccess              *replicator.GetDLQReplicationMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError                       `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError                  `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError                      `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError                  `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetDLQReplicationMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetDLQReplicationMessagesResponse_Read(w wire.Value) (*replicator.GetDLQReplicationMessagesResponse, error) {\n\tvar v replicator.GetDLQReplicationMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetDLQReplicationMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetDLQReplicationMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetDLQReplicationMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetDLQReplicationMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetDLQReplicationMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetDLQReplicationMessages_Result struct could not be encoded.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetDLQReplicationMessagesResponse_Decode(sr stream.Reader) (*replicator.GetDLQReplicationMessagesResponse, error) {\n\tvar v replicator.GetDLQReplicationMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetDLQReplicationMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetDLQReplicationMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetDLQReplicationMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetDLQReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetDLQReplicationMessages_Result\n// struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetDLQReplicationMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetDLQReplicationMessages_Result match the\n// provided HistoryService_GetDLQReplicationMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) Equals(rhs *HistoryService_GetDLQReplicationMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetDLQReplicationMessages_Result.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) GetSuccess() (o *replicator.GetDLQReplicationMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetDLQReplicationMessages\" for this struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) MethodName() string {\n\treturn \"GetDLQReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_GetDLQReplicationMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_GetFailoverInfo_Args represents the arguments for the HistoryService.GetFailoverInfo function.\n//\n// The arguments for GetFailoverInfo are sent and received over the wire as this struct.\ntype HistoryService_GetFailoverInfo_Args struct {\n\tRequest *GetFailoverInfoRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetFailoverInfo_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetFailoverInfo_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetFailoverInfoRequest_Read(w wire.Value) (*GetFailoverInfoRequest, error) {\n\tvar v GetFailoverInfoRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetFailoverInfo_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetFailoverInfo_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetFailoverInfo_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetFailoverInfo_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetFailoverInfoRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetFailoverInfo_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetFailoverInfo_Args struct could not be encoded.\nfunc (v *HistoryService_GetFailoverInfo_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetFailoverInfoRequest_Decode(sr stream.Reader) (*GetFailoverInfoRequest, error) {\n\tvar v GetFailoverInfoRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetFailoverInfo_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetFailoverInfo_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetFailoverInfo_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetFailoverInfoRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetFailoverInfo_Args\n// struct.\nfunc (v *HistoryService_GetFailoverInfo_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetFailoverInfo_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetFailoverInfo_Args match the\n// provided HistoryService_GetFailoverInfo_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetFailoverInfo_Args) Equals(rhs *HistoryService_GetFailoverInfo_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetFailoverInfo_Args.\nfunc (v *HistoryService_GetFailoverInfo_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Args) GetRequest() (o *GetFailoverInfoRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetFailoverInfo\" for this struct.\nfunc (v *HistoryService_GetFailoverInfo_Args) MethodName() string {\n\treturn \"GetFailoverInfo\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_GetFailoverInfo_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_GetFailoverInfo_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.GetFailoverInfo\n// function.\nvar HistoryService_GetFailoverInfo_Helper = struct {\n\t// Args accepts the parameters of GetFailoverInfo in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *GetFailoverInfoRequest,\n\t) *HistoryService_GetFailoverInfo_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetFailoverInfo.\n\t//\n\t// An error can be thrown by GetFailoverInfo only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetFailoverInfo\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetFailoverInfo into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetFailoverInfo\n\t//\n\t//   value, err := GetFailoverInfo(args)\n\t//   result, err := HistoryService_GetFailoverInfo_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetFailoverInfo: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetFailoverInfoResponse, error) (*HistoryService_GetFailoverInfo_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetFailoverInfo\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetFailoverInfo threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_GetFailoverInfo_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_GetFailoverInfo_Result) (*GetFailoverInfoResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_GetFailoverInfo_Helper.Args = func(\n\t\trequest *GetFailoverInfoRequest,\n\t) *HistoryService_GetFailoverInfo_Args {\n\t\treturn &HistoryService_GetFailoverInfo_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_GetFailoverInfo_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_GetFailoverInfo_Helper.WrapResponse = func(success *GetFailoverInfoResponse, err error) (*HistoryService_GetFailoverInfo_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_GetFailoverInfo_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetFailoverInfo_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetFailoverInfo_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetFailoverInfo_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetFailoverInfo_Result{ServiceBusyError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetFailoverInfo_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetFailoverInfo_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetFailoverInfo_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetFailoverInfo_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_GetFailoverInfo_Helper.UnwrapResponse = func(result *HistoryService_GetFailoverInfo_Result) (success *GetFailoverInfoResponse, err error) {\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_GetFailoverInfo_Result represents the result of a HistoryService.GetFailoverInfo function call.\n//\n// The result of a GetFailoverInfo execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_GetFailoverInfo_Result struct {\n\t// Value returned by GetFailoverInfo after a successful execution.\n\tSuccess                 *GetFailoverInfoResponse     `json:\"success,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetFailoverInfo_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetFailoverInfo_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_GetFailoverInfo_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetFailoverInfoResponse_Read(w wire.Value) (*GetFailoverInfoResponse, error) {\n\tvar v GetFailoverInfoResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetFailoverInfo_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetFailoverInfo_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetFailoverInfo_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetFailoverInfo_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetFailoverInfoResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetFailoverInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetFailoverInfo_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetFailoverInfo_Result struct could not be encoded.\nfunc (v *HistoryService_GetFailoverInfo_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetFailoverInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetFailoverInfoResponse_Decode(sr stream.Reader) (*GetFailoverInfoResponse, error) {\n\tvar v GetFailoverInfoResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetFailoverInfo_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetFailoverInfo_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetFailoverInfo_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetFailoverInfoResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetFailoverInfo_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetFailoverInfo_Result\n// struct.\nfunc (v *HistoryService_GetFailoverInfo_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetFailoverInfo_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetFailoverInfo_Result match the\n// provided HistoryService_GetFailoverInfo_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetFailoverInfo_Result) Equals(rhs *HistoryService_GetFailoverInfo_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetFailoverInfo_Result.\nfunc (v *HistoryService_GetFailoverInfo_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Result) GetSuccess() (o *GetFailoverInfoResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetFailoverInfo_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_GetFailoverInfo_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetFailoverInfo\" for this struct.\nfunc (v *HistoryService_GetFailoverInfo_Result) MethodName() string {\n\treturn \"GetFailoverInfo\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_GetFailoverInfo_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_GetMutableState_Args represents the arguments for the HistoryService.GetMutableState function.\n//\n// The arguments for GetMutableState are sent and received over the wire as this struct.\ntype HistoryService_GetMutableState_Args struct {\n\tGetRequest *GetMutableStateRequest `json:\"getRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetMutableState_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetMutableState_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.GetRequest != nil {\n\t\tw, err = v.GetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetMutableStateRequest_Read(w wire.Value) (*GetMutableStateRequest, error) {\n\tvar v GetMutableStateRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetMutableState_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetMutableState_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetMutableState_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetMutableState_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.GetRequest, err = _GetMutableStateRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetMutableState_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetMutableState_Args struct could not be encoded.\nfunc (v *HistoryService_GetMutableState_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.GetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.GetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetMutableStateRequest_Decode(sr stream.Reader) (*GetMutableStateRequest, error) {\n\tvar v GetMutableStateRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetMutableState_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetMutableState_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetMutableState_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.GetRequest, err = _GetMutableStateRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetMutableState_Args\n// struct.\nfunc (v *HistoryService_GetMutableState_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.GetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"GetRequest: %v\", v.GetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetMutableState_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetMutableState_Args match the\n// provided HistoryService_GetMutableState_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetMutableState_Args) Equals(rhs *HistoryService_GetMutableState_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.GetRequest == nil && rhs.GetRequest == nil) || (v.GetRequest != nil && rhs.GetRequest != nil && v.GetRequest.Equals(rhs.GetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetMutableState_Args.\nfunc (v *HistoryService_GetMutableState_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.GetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"getRequest\", v.GetRequest))\n\t}\n\treturn err\n}\n\n// GetGetRequest returns the value of GetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Args) GetGetRequest() (o *GetMutableStateRequest) {\n\tif v != nil && v.GetRequest != nil {\n\t\treturn v.GetRequest\n\t}\n\n\treturn\n}\n\n// IsSetGetRequest returns true if GetRequest is not nil.\nfunc (v *HistoryService_GetMutableState_Args) IsSetGetRequest() bool {\n\treturn v != nil && v.GetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetMutableState\" for this struct.\nfunc (v *HistoryService_GetMutableState_Args) MethodName() string {\n\treturn \"GetMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_GetMutableState_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_GetMutableState_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.GetMutableState\n// function.\nvar HistoryService_GetMutableState_Helper = struct {\n\t// Args accepts the parameters of GetMutableState in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tgetRequest *GetMutableStateRequest,\n\t) *HistoryService_GetMutableState_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetMutableState.\n\t//\n\t// An error can be thrown by GetMutableState only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetMutableState\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetMutableState into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetMutableState\n\t//\n\t//   value, err := GetMutableState(args)\n\t//   result, err := HistoryService_GetMutableState_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetMutableState: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*GetMutableStateResponse, error) (*HistoryService_GetMutableState_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetMutableState\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetMutableState threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_GetMutableState_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_GetMutableState_Result) (*GetMutableStateResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_GetMutableState_Helper.Args = func(\n\t\tgetRequest *GetMutableStateRequest,\n\t) *HistoryService_GetMutableState_Args {\n\t\treturn &HistoryService_GetMutableState_Args{\n\t\t\tGetRequest: getRequest,\n\t\t}\n\t}\n\n\tHistoryService_GetMutableState_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.CurrentBranchChangedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_GetMutableState_Helper.WrapResponse = func(success *GetMutableStateResponse, err error) (*HistoryService_GetMutableState_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_GetMutableState_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.CurrentBranchChangedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetMutableState_Result.CurrentBranchChangedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetMutableState_Result{CurrentBranchChangedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_GetMutableState_Helper.UnwrapResponse = func(result *HistoryService_GetMutableState_Result) (success *GetMutableStateResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.CurrentBranchChangedError != nil {\n\t\t\terr = result.CurrentBranchChangedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_GetMutableState_Result represents the result of a HistoryService.GetMutableState function call.\n//\n// The result of a GetMutableState execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_GetMutableState_Result struct {\n\t// Value returned by GetMutableState after a successful execution.\n\tSuccess                   *GetMutableStateResponse          `json:\"success,omitempty\"`\n\tBadRequestError           *shared.BadRequestError           `json:\"badRequestError,omitempty\"`\n\tInternalServiceError      *shared.InternalServiceError      `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError       *shared.EntityNotExistsError      `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError   *ShardOwnershipLostError          `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError        *shared.LimitExceededError        `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError          *shared.ServiceBusyError          `json:\"serviceBusyError,omitempty\"`\n\tCurrentBranchChangedError *shared.CurrentBranchChangedError `json:\"currentBranchChangedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetMutableState_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetMutableState_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tw, err = v.CurrentBranchChangedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_GetMutableState_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetMutableStateResponse_Read(w wire.Value) (*GetMutableStateResponse, error) {\n\tvar v GetMutableStateResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CurrentBranchChangedError_Read(w wire.Value) (*shared.CurrentBranchChangedError, error) {\n\tvar v shared.CurrentBranchChangedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetMutableState_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetMutableState_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetMutableState_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetMutableState_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetMutableStateResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CurrentBranchChangedError, err = _CurrentBranchChangedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetMutableState_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetMutableState_Result struct could not be encoded.\nfunc (v *HistoryService_GetMutableState_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchChangedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CurrentBranchChangedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetMutableStateResponse_Decode(sr stream.Reader) (*GetMutableStateResponse, error) {\n\tvar v GetMutableStateResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CurrentBranchChangedError_Decode(sr stream.Reader) (*shared.CurrentBranchChangedError, error) {\n\tvar v shared.CurrentBranchChangedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetMutableState_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetMutableState_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetMutableState_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetMutableStateResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.CurrentBranchChangedError, err = _CurrentBranchChangedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetMutableState_Result\n// struct.\nfunc (v *HistoryService_GetMutableState_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchChangedError: %v\", v.CurrentBranchChangedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetMutableState_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetMutableState_Result match the\n// provided HistoryService_GetMutableState_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetMutableState_Result) Equals(rhs *HistoryService_GetMutableState_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchChangedError == nil && rhs.CurrentBranchChangedError == nil) || (v.CurrentBranchChangedError != nil && rhs.CurrentBranchChangedError != nil && v.CurrentBranchChangedError.Equals(rhs.CurrentBranchChangedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetMutableState_Result.\nfunc (v *HistoryService_GetMutableState_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"currentBranchChangedError\", v.CurrentBranchChangedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetSuccess() (o *GetMutableStateResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetCurrentBranchChangedError returns the value of CurrentBranchChangedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetMutableState_Result) GetCurrentBranchChangedError() (o *shared.CurrentBranchChangedError) {\n\tif v != nil && v.CurrentBranchChangedError != nil {\n\t\treturn v.CurrentBranchChangedError\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchChangedError returns true if CurrentBranchChangedError is not nil.\nfunc (v *HistoryService_GetMutableState_Result) IsSetCurrentBranchChangedError() bool {\n\treturn v != nil && v.CurrentBranchChangedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetMutableState\" for this struct.\nfunc (v *HistoryService_GetMutableState_Result) MethodName() string {\n\treturn \"GetMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_GetMutableState_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_GetReplicationMessages_Args represents the arguments for the HistoryService.GetReplicationMessages function.\n//\n// The arguments for GetReplicationMessages are sent and received over the wire as this struct.\ntype HistoryService_GetReplicationMessages_Args struct {\n\tRequest *replicator.GetReplicationMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetReplicationMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetReplicationMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetReplicationMessagesRequest_Read(w wire.Value) (*replicator.GetReplicationMessagesRequest, error) {\n\tvar v replicator.GetReplicationMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetReplicationMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetReplicationMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetReplicationMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetReplicationMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetReplicationMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetReplicationMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetReplicationMessages_Args struct could not be encoded.\nfunc (v *HistoryService_GetReplicationMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetReplicationMessagesRequest_Decode(sr stream.Reader) (*replicator.GetReplicationMessagesRequest, error) {\n\tvar v replicator.GetReplicationMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetReplicationMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetReplicationMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetReplicationMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetReplicationMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetReplicationMessages_Args\n// struct.\nfunc (v *HistoryService_GetReplicationMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetReplicationMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetReplicationMessages_Args match the\n// provided HistoryService_GetReplicationMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetReplicationMessages_Args) Equals(rhs *HistoryService_GetReplicationMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetReplicationMessages_Args.\nfunc (v *HistoryService_GetReplicationMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Args) GetRequest() (o *replicator.GetReplicationMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetReplicationMessages\" for this struct.\nfunc (v *HistoryService_GetReplicationMessages_Args) MethodName() string {\n\treturn \"GetReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_GetReplicationMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_GetReplicationMessages_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.GetReplicationMessages\n// function.\nvar HistoryService_GetReplicationMessages_Helper = struct {\n\t// Args accepts the parameters of GetReplicationMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.GetReplicationMessagesRequest,\n\t) *HistoryService_GetReplicationMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetReplicationMessages.\n\t//\n\t// An error can be thrown by GetReplicationMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetReplicationMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetReplicationMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetReplicationMessages\n\t//\n\t//   value, err := GetReplicationMessages(args)\n\t//   result, err := HistoryService_GetReplicationMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetReplicationMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.GetReplicationMessagesResponse, error) (*HistoryService_GetReplicationMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetReplicationMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetReplicationMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_GetReplicationMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_GetReplicationMessages_Result) (*replicator.GetReplicationMessagesResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_GetReplicationMessages_Helper.Args = func(\n\t\trequest *replicator.GetReplicationMessagesRequest,\n\t) *HistoryService_GetReplicationMessages_Args {\n\t\treturn &HistoryService_GetReplicationMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_GetReplicationMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_GetReplicationMessages_Helper.WrapResponse = func(success *replicator.GetReplicationMessagesResponse, err error) (*HistoryService_GetReplicationMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetReplicationMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetReplicationMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetReplicationMessages_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetReplicationMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_GetReplicationMessages_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_GetReplicationMessages_Result{ClientVersionNotSupportedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_GetReplicationMessages_Helper.UnwrapResponse = func(result *HistoryService_GetReplicationMessages_Result) (success *replicator.GetReplicationMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_GetReplicationMessages_Result represents the result of a HistoryService.GetReplicationMessages function call.\n//\n// The result of a GetReplicationMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_GetReplicationMessages_Result struct {\n\t// Value returned by GetReplicationMessages after a successful execution.\n\tSuccess                        *replicator.GetReplicationMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                    `json:\"badRequestError,omitempty\"`\n\tInternalServiceError           *shared.InternalServiceError               `json:\"internalServiceError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError                 `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError                   `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError     `json:\"clientVersionNotSupportedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_GetReplicationMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_GetReplicationMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_GetReplicationMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetReplicationMessagesResponse_Read(w wire.Value) (*replicator.GetReplicationMessagesResponse, error) {\n\tvar v replicator.GetReplicationMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Read(w wire.Value) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_GetReplicationMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_GetReplicationMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_GetReplicationMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_GetReplicationMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetReplicationMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_GetReplicationMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_GetReplicationMessages_Result struct could not be encoded.\nfunc (v *HistoryService_GetReplicationMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetReplicationMessagesResponse_Decode(sr stream.Reader) (*replicator.GetReplicationMessagesResponse, error) {\n\tvar v replicator.GetReplicationMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ClientVersionNotSupportedError_Decode(sr stream.Reader) (*shared.ClientVersionNotSupportedError, error) {\n\tvar v shared.ClientVersionNotSupportedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_GetReplicationMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_GetReplicationMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_GetReplicationMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetReplicationMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_GetReplicationMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_GetReplicationMessages_Result\n// struct.\nfunc (v *HistoryService_GetReplicationMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_GetReplicationMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_GetReplicationMessages_Result match the\n// provided HistoryService_GetReplicationMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_GetReplicationMessages_Result) Equals(rhs *HistoryService_GetReplicationMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_GetReplicationMessages_Result.\nfunc (v *HistoryService_GetReplicationMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetSuccess() (o *replicator.GetReplicationMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_GetReplicationMessages_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *HistoryService_GetReplicationMessages_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetReplicationMessages\" for this struct.\nfunc (v *HistoryService_GetReplicationMessages_Result) MethodName() string {\n\treturn \"GetReplicationMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_GetReplicationMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_MergeDLQMessages_Args represents the arguments for the HistoryService.MergeDLQMessages function.\n//\n// The arguments for MergeDLQMessages are sent and received over the wire as this struct.\ntype HistoryService_MergeDLQMessages_Args struct {\n\tRequest *replicator.MergeDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_MergeDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_MergeDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _MergeDLQMessagesRequest_Read(w wire.Value) (*replicator.MergeDLQMessagesRequest, error) {\n\tvar v replicator.MergeDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_MergeDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_MergeDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_MergeDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_MergeDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _MergeDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_MergeDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_MergeDLQMessages_Args struct could not be encoded.\nfunc (v *HistoryService_MergeDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _MergeDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.MergeDLQMessagesRequest, error) {\n\tvar v replicator.MergeDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_MergeDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_MergeDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_MergeDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _MergeDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_MergeDLQMessages_Args\n// struct.\nfunc (v *HistoryService_MergeDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_MergeDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_MergeDLQMessages_Args match the\n// provided HistoryService_MergeDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_MergeDLQMessages_Args) Equals(rhs *HistoryService_MergeDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_MergeDLQMessages_Args.\nfunc (v *HistoryService_MergeDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Args) GetRequest() (o *replicator.MergeDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"MergeDLQMessages\" for this struct.\nfunc (v *HistoryService_MergeDLQMessages_Args) MethodName() string {\n\treturn \"MergeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_MergeDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_MergeDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.MergeDLQMessages\n// function.\nvar HistoryService_MergeDLQMessages_Helper = struct {\n\t// Args accepts the parameters of MergeDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.MergeDLQMessagesRequest,\n\t) *HistoryService_MergeDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by MergeDLQMessages.\n\t//\n\t// An error can be thrown by MergeDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for MergeDLQMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// MergeDLQMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by MergeDLQMessages\n\t//\n\t//   value, err := MergeDLQMessages(args)\n\t//   result, err := HistoryService_MergeDLQMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from MergeDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.MergeDLQMessagesResponse, error) (*HistoryService_MergeDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for MergeDLQMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if MergeDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_MergeDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_MergeDLQMessages_Result) (*replicator.MergeDLQMessagesResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_MergeDLQMessages_Helper.Args = func(\n\t\trequest *replicator.MergeDLQMessagesRequest,\n\t) *HistoryService_MergeDLQMessages_Args {\n\t\treturn &HistoryService_MergeDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_MergeDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_MergeDLQMessages_Helper.WrapResponse = func(success *replicator.MergeDLQMessagesResponse, err error) (*HistoryService_MergeDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_MergeDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_MergeDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_MergeDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_MergeDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_MergeDLQMessages_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_MergeDLQMessages_Result{ShardOwnershipLostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_MergeDLQMessages_Helper.UnwrapResponse = func(result *HistoryService_MergeDLQMessages_Result) (success *replicator.MergeDLQMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_MergeDLQMessages_Result represents the result of a HistoryService.MergeDLQMessages function call.\n//\n// The result of a MergeDLQMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_MergeDLQMessages_Result struct {\n\t// Value returned by MergeDLQMessages after a successful execution.\n\tSuccess                 *replicator.MergeDLQMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError         `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError             `json:\"shardOwnershipLostError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_MergeDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_MergeDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_MergeDLQMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _MergeDLQMessagesResponse_Read(w wire.Value) (*replicator.MergeDLQMessagesResponse, error) {\n\tvar v replicator.MergeDLQMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_MergeDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_MergeDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_MergeDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_MergeDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _MergeDLQMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_MergeDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_MergeDLQMessages_Result struct could not be encoded.\nfunc (v *HistoryService_MergeDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _MergeDLQMessagesResponse_Decode(sr stream.Reader) (*replicator.MergeDLQMessagesResponse, error) {\n\tvar v replicator.MergeDLQMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_MergeDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_MergeDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_MergeDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _MergeDLQMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_MergeDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_MergeDLQMessages_Result\n// struct.\nfunc (v *HistoryService_MergeDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_MergeDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_MergeDLQMessages_Result match the\n// provided HistoryService_MergeDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_MergeDLQMessages_Result) Equals(rhs *HistoryService_MergeDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_MergeDLQMessages_Result.\nfunc (v *HistoryService_MergeDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetSuccess() (o *replicator.MergeDLQMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_MergeDLQMessages_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_MergeDLQMessages_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"MergeDLQMessages\" for this struct.\nfunc (v *HistoryService_MergeDLQMessages_Result) MethodName() string {\n\treturn \"MergeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_MergeDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_NotifyFailoverMarkers_Args represents the arguments for the HistoryService.NotifyFailoverMarkers function.\n//\n// The arguments for NotifyFailoverMarkers are sent and received over the wire as this struct.\ntype HistoryService_NotifyFailoverMarkers_Args struct {\n\tRequest *NotifyFailoverMarkersRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_NotifyFailoverMarkers_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _NotifyFailoverMarkersRequest_Read(w wire.Value) (*NotifyFailoverMarkersRequest, error) {\n\tvar v NotifyFailoverMarkersRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_NotifyFailoverMarkers_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_NotifyFailoverMarkers_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_NotifyFailoverMarkers_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _NotifyFailoverMarkersRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_NotifyFailoverMarkers_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_NotifyFailoverMarkers_Args struct could not be encoded.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _NotifyFailoverMarkersRequest_Decode(sr stream.Reader) (*NotifyFailoverMarkersRequest, error) {\n\tvar v NotifyFailoverMarkersRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_NotifyFailoverMarkers_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_NotifyFailoverMarkers_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _NotifyFailoverMarkersRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_NotifyFailoverMarkers_Args\n// struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_NotifyFailoverMarkers_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_NotifyFailoverMarkers_Args match the\n// provided HistoryService_NotifyFailoverMarkers_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) Equals(rhs *HistoryService_NotifyFailoverMarkers_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_NotifyFailoverMarkers_Args.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) GetRequest() (o *NotifyFailoverMarkersRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"NotifyFailoverMarkers\" for this struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) MethodName() string {\n\treturn \"NotifyFailoverMarkers\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_NotifyFailoverMarkers_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.NotifyFailoverMarkers\n// function.\nvar HistoryService_NotifyFailoverMarkers_Helper = struct {\n\t// Args accepts the parameters of NotifyFailoverMarkers in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *NotifyFailoverMarkersRequest,\n\t) *HistoryService_NotifyFailoverMarkers_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by NotifyFailoverMarkers.\n\t//\n\t// An error can be thrown by NotifyFailoverMarkers only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for NotifyFailoverMarkers\n\t// given the error returned by it. The provided error may\n\t// be nil if NotifyFailoverMarkers did not fail.\n\t//\n\t// This allows mapping errors returned by NotifyFailoverMarkers into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// NotifyFailoverMarkers\n\t//\n\t//   err := NotifyFailoverMarkers(args)\n\t//   result, err := HistoryService_NotifyFailoverMarkers_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from NotifyFailoverMarkers: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_NotifyFailoverMarkers_Result, error)\n\n\t// UnwrapResponse takes the result struct for NotifyFailoverMarkers\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if NotifyFailoverMarkers threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_NotifyFailoverMarkers_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_NotifyFailoverMarkers_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_NotifyFailoverMarkers_Helper.Args = func(\n\t\trequest *NotifyFailoverMarkersRequest,\n\t) *HistoryService_NotifyFailoverMarkers_Args {\n\t\treturn &HistoryService_NotifyFailoverMarkers_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_NotifyFailoverMarkers_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_NotifyFailoverMarkers_Helper.WrapResponse = func(err error) (*HistoryService_NotifyFailoverMarkers_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_NotifyFailoverMarkers_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_NotifyFailoverMarkers_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_NotifyFailoverMarkers_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_NotifyFailoverMarkers_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_NotifyFailoverMarkers_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_NotifyFailoverMarkers_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_NotifyFailoverMarkers_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_NotifyFailoverMarkers_Helper.UnwrapResponse = func(result *HistoryService_NotifyFailoverMarkers_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_NotifyFailoverMarkers_Result represents the result of a HistoryService.NotifyFailoverMarkers function call.\n//\n// The result of a NotifyFailoverMarkers execution is sent and received over the wire as this struct.\ntype HistoryService_NotifyFailoverMarkers_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_NotifyFailoverMarkers_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_NotifyFailoverMarkers_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_NotifyFailoverMarkers_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_NotifyFailoverMarkers_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_NotifyFailoverMarkers_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_NotifyFailoverMarkers_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_NotifyFailoverMarkers_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_NotifyFailoverMarkers_Result struct could not be encoded.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_NotifyFailoverMarkers_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_NotifyFailoverMarkers_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_NotifyFailoverMarkers_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_NotifyFailoverMarkers_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_NotifyFailoverMarkers_Result\n// struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_NotifyFailoverMarkers_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_NotifyFailoverMarkers_Result match the\n// provided HistoryService_NotifyFailoverMarkers_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) Equals(rhs *HistoryService_NotifyFailoverMarkers_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_NotifyFailoverMarkers_Result.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"NotifyFailoverMarkers\" for this struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) MethodName() string {\n\treturn \"NotifyFailoverMarkers\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_NotifyFailoverMarkers_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_PollMutableState_Args represents the arguments for the HistoryService.PollMutableState function.\n//\n// The arguments for PollMutableState are sent and received over the wire as this struct.\ntype HistoryService_PollMutableState_Args struct {\n\tPollRequest *PollMutableStateRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_PollMutableState_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_PollMutableState_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollMutableStateRequest_Read(w wire.Value) (*PollMutableStateRequest, error) {\n\tvar v PollMutableStateRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_PollMutableState_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_PollMutableState_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_PollMutableState_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_PollMutableState_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollMutableStateRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_PollMutableState_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_PollMutableState_Args struct could not be encoded.\nfunc (v *HistoryService_PollMutableState_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollMutableStateRequest_Decode(sr stream.Reader) (*PollMutableStateRequest, error) {\n\tvar v PollMutableStateRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_PollMutableState_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_PollMutableState_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_PollMutableState_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollMutableStateRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_PollMutableState_Args\n// struct.\nfunc (v *HistoryService_PollMutableState_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_PollMutableState_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_PollMutableState_Args match the\n// provided HistoryService_PollMutableState_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_PollMutableState_Args) Equals(rhs *HistoryService_PollMutableState_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_PollMutableState_Args.\nfunc (v *HistoryService_PollMutableState_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Args) GetPollRequest() (o *PollMutableStateRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *HistoryService_PollMutableState_Args) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PollMutableState\" for this struct.\nfunc (v *HistoryService_PollMutableState_Args) MethodName() string {\n\treturn \"PollMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_PollMutableState_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_PollMutableState_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.PollMutableState\n// function.\nvar HistoryService_PollMutableState_Helper = struct {\n\t// Args accepts the parameters of PollMutableState in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tpollRequest *PollMutableStateRequest,\n\t) *HistoryService_PollMutableState_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PollMutableState.\n\t//\n\t// An error can be thrown by PollMutableState only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PollMutableState\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// PollMutableState into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by PollMutableState\n\t//\n\t//   value, err := PollMutableState(args)\n\t//   result, err := HistoryService_PollMutableState_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PollMutableState: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*PollMutableStateResponse, error) (*HistoryService_PollMutableState_Result, error)\n\n\t// UnwrapResponse takes the result struct for PollMutableState\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if PollMutableState threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_PollMutableState_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_PollMutableState_Result) (*PollMutableStateResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_PollMutableState_Helper.Args = func(\n\t\tpollRequest *PollMutableStateRequest,\n\t) *HistoryService_PollMutableState_Args {\n\t\treturn &HistoryService_PollMutableState_Args{\n\t\t\tPollRequest: pollRequest,\n\t\t}\n\t}\n\n\tHistoryService_PollMutableState_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.CurrentBranchChangedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_PollMutableState_Helper.WrapResponse = func(success *PollMutableStateResponse, err error) (*HistoryService_PollMutableState_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_PollMutableState_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.CurrentBranchChangedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PollMutableState_Result.CurrentBranchChangedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PollMutableState_Result{CurrentBranchChangedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_PollMutableState_Helper.UnwrapResponse = func(result *HistoryService_PollMutableState_Result) (success *PollMutableStateResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.CurrentBranchChangedError != nil {\n\t\t\terr = result.CurrentBranchChangedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_PollMutableState_Result represents the result of a HistoryService.PollMutableState function call.\n//\n// The result of a PollMutableState execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_PollMutableState_Result struct {\n\t// Value returned by PollMutableState after a successful execution.\n\tSuccess                   *PollMutableStateResponse         `json:\"success,omitempty\"`\n\tBadRequestError           *shared.BadRequestError           `json:\"badRequestError,omitempty\"`\n\tInternalServiceError      *shared.InternalServiceError      `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError       *shared.EntityNotExistsError      `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError   *ShardOwnershipLostError          `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError        *shared.LimitExceededError        `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError          *shared.ServiceBusyError          `json:\"serviceBusyError,omitempty\"`\n\tCurrentBranchChangedError *shared.CurrentBranchChangedError `json:\"currentBranchChangedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_PollMutableState_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_PollMutableState_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tw, err = v.CurrentBranchChangedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_PollMutableState_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollMutableStateResponse_Read(w wire.Value) (*PollMutableStateResponse, error) {\n\tvar v PollMutableStateResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_PollMutableState_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_PollMutableState_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_PollMutableState_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_PollMutableState_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _PollMutableStateResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CurrentBranchChangedError, err = _CurrentBranchChangedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PollMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_PollMutableState_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_PollMutableState_Result struct could not be encoded.\nfunc (v *HistoryService_PollMutableState_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CurrentBranchChangedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CurrentBranchChangedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PollMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollMutableStateResponse_Decode(sr stream.Reader) (*PollMutableStateResponse, error) {\n\tvar v PollMutableStateResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_PollMutableState_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_PollMutableState_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_PollMutableState_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _PollMutableStateResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.CurrentBranchChangedError, err = _CurrentBranchChangedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PollMutableState_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_PollMutableState_Result\n// struct.\nfunc (v *HistoryService_PollMutableState_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentBranchChangedError: %v\", v.CurrentBranchChangedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_PollMutableState_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_PollMutableState_Result match the\n// provided HistoryService_PollMutableState_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_PollMutableState_Result) Equals(rhs *HistoryService_PollMutableState_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.CurrentBranchChangedError == nil && rhs.CurrentBranchChangedError == nil) || (v.CurrentBranchChangedError != nil && rhs.CurrentBranchChangedError != nil && v.CurrentBranchChangedError.Equals(rhs.CurrentBranchChangedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_PollMutableState_Result.\nfunc (v *HistoryService_PollMutableState_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.CurrentBranchChangedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"currentBranchChangedError\", v.CurrentBranchChangedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetSuccess() (o *PollMutableStateResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetCurrentBranchChangedError returns the value of CurrentBranchChangedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PollMutableState_Result) GetCurrentBranchChangedError() (o *shared.CurrentBranchChangedError) {\n\tif v != nil && v.CurrentBranchChangedError != nil {\n\t\treturn v.CurrentBranchChangedError\n\t}\n\n\treturn\n}\n\n// IsSetCurrentBranchChangedError returns true if CurrentBranchChangedError is not nil.\nfunc (v *HistoryService_PollMutableState_Result) IsSetCurrentBranchChangedError() bool {\n\treturn v != nil && v.CurrentBranchChangedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PollMutableState\" for this struct.\nfunc (v *HistoryService_PollMutableState_Result) MethodName() string {\n\treturn \"PollMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_PollMutableState_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_PurgeDLQMessages_Args represents the arguments for the HistoryService.PurgeDLQMessages function.\n//\n// The arguments for PurgeDLQMessages are sent and received over the wire as this struct.\ntype HistoryService_PurgeDLQMessages_Args struct {\n\tRequest *replicator.PurgeDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_PurgeDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_PurgeDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PurgeDLQMessagesRequest_Read(w wire.Value) (*replicator.PurgeDLQMessagesRequest, error) {\n\tvar v replicator.PurgeDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_PurgeDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_PurgeDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_PurgeDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_PurgeDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _PurgeDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_PurgeDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_PurgeDLQMessages_Args struct could not be encoded.\nfunc (v *HistoryService_PurgeDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PurgeDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.PurgeDLQMessagesRequest, error) {\n\tvar v replicator.PurgeDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_PurgeDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_PurgeDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_PurgeDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _PurgeDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_PurgeDLQMessages_Args\n// struct.\nfunc (v *HistoryService_PurgeDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_PurgeDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_PurgeDLQMessages_Args match the\n// provided HistoryService_PurgeDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_PurgeDLQMessages_Args) Equals(rhs *HistoryService_PurgeDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_PurgeDLQMessages_Args.\nfunc (v *HistoryService_PurgeDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Args) GetRequest() (o *replicator.PurgeDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PurgeDLQMessages\" for this struct.\nfunc (v *HistoryService_PurgeDLQMessages_Args) MethodName() string {\n\treturn \"PurgeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_PurgeDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_PurgeDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.PurgeDLQMessages\n// function.\nvar HistoryService_PurgeDLQMessages_Helper = struct {\n\t// Args accepts the parameters of PurgeDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.PurgeDLQMessagesRequest,\n\t) *HistoryService_PurgeDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PurgeDLQMessages.\n\t//\n\t// An error can be thrown by PurgeDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PurgeDLQMessages\n\t// given the error returned by it. The provided error may\n\t// be nil if PurgeDLQMessages did not fail.\n\t//\n\t// This allows mapping errors returned by PurgeDLQMessages into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// PurgeDLQMessages\n\t//\n\t//   err := PurgeDLQMessages(args)\n\t//   result, err := HistoryService_PurgeDLQMessages_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PurgeDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_PurgeDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for PurgeDLQMessages\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if PurgeDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_PurgeDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_PurgeDLQMessages_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_PurgeDLQMessages_Helper.Args = func(\n\t\trequest *replicator.PurgeDLQMessagesRequest,\n\t) *HistoryService_PurgeDLQMessages_Args {\n\t\treturn &HistoryService_PurgeDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_PurgeDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_PurgeDLQMessages_Helper.WrapResponse = func(err error) (*HistoryService_PurgeDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PurgeDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PurgeDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PurgeDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PurgeDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_PurgeDLQMessages_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_PurgeDLQMessages_Result{ShardOwnershipLostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_PurgeDLQMessages_Helper.UnwrapResponse = func(result *HistoryService_PurgeDLQMessages_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_PurgeDLQMessages_Result represents the result of a HistoryService.PurgeDLQMessages function call.\n//\n// The result of a PurgeDLQMessages execution is sent and received over the wire as this struct.\ntype HistoryService_PurgeDLQMessages_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_PurgeDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_PurgeDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_PurgeDLQMessages_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_PurgeDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_PurgeDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_PurgeDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_PurgeDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_PurgeDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_PurgeDLQMessages_Result struct could not be encoded.\nfunc (v *HistoryService_PurgeDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_PurgeDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_PurgeDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_PurgeDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_PurgeDLQMessages_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_PurgeDLQMessages_Result\n// struct.\nfunc (v *HistoryService_PurgeDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_PurgeDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_PurgeDLQMessages_Result match the\n// provided HistoryService_PurgeDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_PurgeDLQMessages_Result) Equals(rhs *HistoryService_PurgeDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_PurgeDLQMessages_Result.\nfunc (v *HistoryService_PurgeDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_PurgeDLQMessages_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_PurgeDLQMessages_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PurgeDLQMessages\" for this struct.\nfunc (v *HistoryService_PurgeDLQMessages_Result) MethodName() string {\n\treturn \"PurgeDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_PurgeDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_QueryWorkflow_Args represents the arguments for the HistoryService.QueryWorkflow function.\n//\n// The arguments for QueryWorkflow are sent and received over the wire as this struct.\ntype HistoryService_QueryWorkflow_Args struct {\n\tQueryRequest *QueryWorkflowRequest `json:\"queryRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_QueryWorkflow_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_QueryWorkflow_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.QueryRequest != nil {\n\t\tw, err = v.QueryRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowRequest_1_Read(w wire.Value) (*QueryWorkflowRequest, error) {\n\tvar v QueryWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_QueryWorkflow_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_QueryWorkflow_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_QueryWorkflow_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_QueryWorkflow_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_QueryWorkflow_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_QueryWorkflow_Args struct could not be encoded.\nfunc (v *HistoryService_QueryWorkflow_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.QueryRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowRequest_1_Decode(sr stream.Reader) (*QueryWorkflowRequest, error) {\n\tvar v QueryWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_QueryWorkflow_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_QueryWorkflow_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_QueryWorkflow_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_QueryWorkflow_Args\n// struct.\nfunc (v *HistoryService_QueryWorkflow_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.QueryRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRequest: %v\", v.QueryRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_QueryWorkflow_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_QueryWorkflow_Args match the\n// provided HistoryService_QueryWorkflow_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_QueryWorkflow_Args) Equals(rhs *HistoryService_QueryWorkflow_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.QueryRequest == nil && rhs.QueryRequest == nil) || (v.QueryRequest != nil && rhs.QueryRequest != nil && v.QueryRequest.Equals(rhs.QueryRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_QueryWorkflow_Args.\nfunc (v *HistoryService_QueryWorkflow_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.QueryRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRequest\", v.QueryRequest))\n\t}\n\treturn err\n}\n\n// GetQueryRequest returns the value of QueryRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Args) GetQueryRequest() (o *QueryWorkflowRequest) {\n\tif v != nil && v.QueryRequest != nil {\n\t\treturn v.QueryRequest\n\t}\n\n\treturn\n}\n\n// IsSetQueryRequest returns true if QueryRequest is not nil.\nfunc (v *HistoryService_QueryWorkflow_Args) IsSetQueryRequest() bool {\n\treturn v != nil && v.QueryRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *HistoryService_QueryWorkflow_Args) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_QueryWorkflow_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_QueryWorkflow_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.QueryWorkflow\n// function.\nvar HistoryService_QueryWorkflow_Helper = struct {\n\t// Args accepts the parameters of QueryWorkflow in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tqueryRequest *QueryWorkflowRequest,\n\t) *HistoryService_QueryWorkflow_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by QueryWorkflow.\n\t//\n\t// An error can be thrown by QueryWorkflow only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for QueryWorkflow\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// QueryWorkflow into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by QueryWorkflow\n\t//\n\t//   value, err := QueryWorkflow(args)\n\t//   result, err := HistoryService_QueryWorkflow_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from QueryWorkflow: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*QueryWorkflowResponse, error) (*HistoryService_QueryWorkflow_Result, error)\n\n\t// UnwrapResponse takes the result struct for QueryWorkflow\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if QueryWorkflow threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_QueryWorkflow_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_QueryWorkflow_Result) (*QueryWorkflowResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_QueryWorkflow_Helper.Args = func(\n\t\tqueryRequest *QueryWorkflowRequest,\n\t) *HistoryService_QueryWorkflow_Args {\n\t\treturn &HistoryService_QueryWorkflow_Args{\n\t\t\tQueryRequest: queryRequest,\n\t\t}\n\t}\n\n\tHistoryService_QueryWorkflow_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.QueryFailedError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_QueryWorkflow_Helper.WrapResponse = func(success *QueryWorkflowResponse, err error) (*HistoryService_QueryWorkflow_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_QueryWorkflow_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.QueryFailedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.QueryFailedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{QueryFailedError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.ClientVersionNotSupportedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_QueryWorkflow_Result.ClientVersionNotSupportedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_QueryWorkflow_Result{ClientVersionNotSupportedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_QueryWorkflow_Helper.UnwrapResponse = func(result *HistoryService_QueryWorkflow_Result) (success *QueryWorkflowResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.QueryFailedError != nil {\n\t\t\terr = result.QueryFailedError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ClientVersionNotSupportedError != nil {\n\t\t\terr = result.ClientVersionNotSupportedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_QueryWorkflow_Result represents the result of a HistoryService.QueryWorkflow function call.\n//\n// The result of a QueryWorkflow execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_QueryWorkflow_Result struct {\n\t// Value returned by QueryWorkflow after a successful execution.\n\tSuccess                        *QueryWorkflowResponse                 `json:\"success,omitempty\"`\n\tBadRequestError                *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tInternalServiceError           *shared.InternalServiceError           `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError            *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tQueryFailedError               *shared.QueryFailedError               `json:\"queryFailedError,omitempty\"`\n\tLimitExceededError             *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError               *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n\tClientVersionNotSupportedError *shared.ClientVersionNotSupportedError `json:\"clientVersionNotSupportedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_QueryWorkflow_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_QueryWorkflow_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tw, err = v.QueryFailedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tw, err = v.ClientVersionNotSupportedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_QueryWorkflow_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowResponse_1_Read(w wire.Value) (*QueryWorkflowResponse, error) {\n\tvar v QueryWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Read(w wire.Value) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_QueryWorkflow_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_QueryWorkflow_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_QueryWorkflow_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_QueryWorkflow_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _QueryWorkflowResponse_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryFailedError, err = _QueryFailedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_QueryWorkflow_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_QueryWorkflow_Result struct could not be encoded.\nfunc (v *HistoryService_QueryWorkflow_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryFailedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryFailedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClientVersionNotSupportedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowResponse_1_Decode(sr stream.Reader) (*QueryWorkflowResponse, error) {\n\tvar v QueryWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Decode(sr stream.Reader) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_QueryWorkflow_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_QueryWorkflow_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_QueryWorkflow_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _QueryWorkflowResponse_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.QueryFailedError, err = _QueryFailedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ClientVersionNotSupportedError, err = _ClientVersionNotSupportedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_QueryWorkflow_Result\n// struct.\nfunc (v *HistoryService_QueryWorkflow_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryFailedError: %v\", v.QueryFailedError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientVersionNotSupportedError: %v\", v.ClientVersionNotSupportedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_QueryWorkflow_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_QueryWorkflow_Result match the\n// provided HistoryService_QueryWorkflow_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_QueryWorkflow_Result) Equals(rhs *HistoryService_QueryWorkflow_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.QueryFailedError == nil && rhs.QueryFailedError == nil) || (v.QueryFailedError != nil && rhs.QueryFailedError != nil && v.QueryFailedError.Equals(rhs.QueryFailedError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ClientVersionNotSupportedError == nil && rhs.ClientVersionNotSupportedError == nil) || (v.ClientVersionNotSupportedError != nil && rhs.ClientVersionNotSupportedError != nil && v.ClientVersionNotSupportedError.Equals(rhs.ClientVersionNotSupportedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_QueryWorkflow_Result.\nfunc (v *HistoryService_QueryWorkflow_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.QueryFailedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryFailedError\", v.QueryFailedError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ClientVersionNotSupportedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clientVersionNotSupportedError\", v.ClientVersionNotSupportedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetSuccess() (o *QueryWorkflowResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetQueryFailedError returns the value of QueryFailedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetQueryFailedError() (o *shared.QueryFailedError) {\n\tif v != nil && v.QueryFailedError != nil {\n\t\treturn v.QueryFailedError\n\t}\n\n\treturn\n}\n\n// IsSetQueryFailedError returns true if QueryFailedError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetQueryFailedError() bool {\n\treturn v != nil && v.QueryFailedError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetClientVersionNotSupportedError returns the value of ClientVersionNotSupportedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_QueryWorkflow_Result) GetClientVersionNotSupportedError() (o *shared.ClientVersionNotSupportedError) {\n\tif v != nil && v.ClientVersionNotSupportedError != nil {\n\t\treturn v.ClientVersionNotSupportedError\n\t}\n\n\treturn\n}\n\n// IsSetClientVersionNotSupportedError returns true if ClientVersionNotSupportedError is not nil.\nfunc (v *HistoryService_QueryWorkflow_Result) IsSetClientVersionNotSupportedError() bool {\n\treturn v != nil && v.ClientVersionNotSupportedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *HistoryService_QueryWorkflow_Result) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_QueryWorkflow_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RatelimitUpdate_Args represents the arguments for the HistoryService.RatelimitUpdate function.\n//\n// The arguments for RatelimitUpdate are sent and received over the wire as this struct.\ntype HistoryService_RatelimitUpdate_Args struct {\n\tRequest *RatelimitUpdateRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RatelimitUpdate_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RatelimitUpdate_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RatelimitUpdateRequest_Read(w wire.Value) (*RatelimitUpdateRequest, error) {\n\tvar v RatelimitUpdateRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RatelimitUpdate_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RatelimitUpdate_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RatelimitUpdate_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RatelimitUpdate_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RatelimitUpdateRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RatelimitUpdate_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RatelimitUpdate_Args struct could not be encoded.\nfunc (v *HistoryService_RatelimitUpdate_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RatelimitUpdateRequest_Decode(sr stream.Reader) (*RatelimitUpdateRequest, error) {\n\tvar v RatelimitUpdateRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RatelimitUpdate_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RatelimitUpdate_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RatelimitUpdate_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RatelimitUpdateRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RatelimitUpdate_Args\n// struct.\nfunc (v *HistoryService_RatelimitUpdate_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RatelimitUpdate_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RatelimitUpdate_Args match the\n// provided HistoryService_RatelimitUpdate_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RatelimitUpdate_Args) Equals(rhs *HistoryService_RatelimitUpdate_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RatelimitUpdate_Args.\nfunc (v *HistoryService_RatelimitUpdate_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Args) GetRequest() (o *RatelimitUpdateRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RatelimitUpdate\" for this struct.\nfunc (v *HistoryService_RatelimitUpdate_Args) MethodName() string {\n\treturn \"RatelimitUpdate\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RatelimitUpdate_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RatelimitUpdate_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RatelimitUpdate\n// function.\nvar HistoryService_RatelimitUpdate_Helper = struct {\n\t// Args accepts the parameters of RatelimitUpdate in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *RatelimitUpdateRequest,\n\t) *HistoryService_RatelimitUpdate_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RatelimitUpdate.\n\t//\n\t// An error can be thrown by RatelimitUpdate only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RatelimitUpdate\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RatelimitUpdate into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RatelimitUpdate\n\t//\n\t//   value, err := RatelimitUpdate(args)\n\t//   result, err := HistoryService_RatelimitUpdate_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RatelimitUpdate: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*RatelimitUpdateResponse, error) (*HistoryService_RatelimitUpdate_Result, error)\n\n\t// UnwrapResponse takes the result struct for RatelimitUpdate\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RatelimitUpdate threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RatelimitUpdate_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RatelimitUpdate_Result) (*RatelimitUpdateResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RatelimitUpdate_Helper.Args = func(\n\t\trequest *RatelimitUpdateRequest,\n\t) *HistoryService_RatelimitUpdate_Args {\n\t\treturn &HistoryService_RatelimitUpdate_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_RatelimitUpdate_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RatelimitUpdate_Helper.WrapResponse = func(success *RatelimitUpdateResponse, err error) (*HistoryService_RatelimitUpdate_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RatelimitUpdate_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RatelimitUpdate_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RatelimitUpdate_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RatelimitUpdate_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RatelimitUpdate_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RatelimitUpdate_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RatelimitUpdate_Result{ServiceBusyError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RatelimitUpdate_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RatelimitUpdate_Result{ShardOwnershipLostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RatelimitUpdate_Helper.UnwrapResponse = func(result *HistoryService_RatelimitUpdate_Result) (success *RatelimitUpdateResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RatelimitUpdate_Result represents the result of a HistoryService.RatelimitUpdate function call.\n//\n// The result of a RatelimitUpdate execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RatelimitUpdate_Result struct {\n\t// Value returned by RatelimitUpdate after a successful execution.\n\tSuccess                 *RatelimitUpdateResponse     `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RatelimitUpdate_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RatelimitUpdate_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RatelimitUpdate_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RatelimitUpdateResponse_Read(w wire.Value) (*RatelimitUpdateResponse, error) {\n\tvar v RatelimitUpdateResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RatelimitUpdate_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RatelimitUpdate_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RatelimitUpdate_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RatelimitUpdate_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RatelimitUpdateResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RatelimitUpdate_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RatelimitUpdate_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RatelimitUpdate_Result struct could not be encoded.\nfunc (v *HistoryService_RatelimitUpdate_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RatelimitUpdate_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RatelimitUpdateResponse_Decode(sr stream.Reader) (*RatelimitUpdateResponse, error) {\n\tvar v RatelimitUpdateResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RatelimitUpdate_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RatelimitUpdate_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RatelimitUpdate_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RatelimitUpdateResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RatelimitUpdate_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RatelimitUpdate_Result\n// struct.\nfunc (v *HistoryService_RatelimitUpdate_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RatelimitUpdate_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RatelimitUpdate_Result match the\n// provided HistoryService_RatelimitUpdate_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RatelimitUpdate_Result) Equals(rhs *HistoryService_RatelimitUpdate_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RatelimitUpdate_Result.\nfunc (v *HistoryService_RatelimitUpdate_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Result) GetSuccess() (o *RatelimitUpdateResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RatelimitUpdate_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RatelimitUpdate_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RatelimitUpdate\" for this struct.\nfunc (v *HistoryService_RatelimitUpdate_Result) MethodName() string {\n\treturn \"RatelimitUpdate\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RatelimitUpdate_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ReadDLQMessages_Args represents the arguments for the HistoryService.ReadDLQMessages function.\n//\n// The arguments for ReadDLQMessages are sent and received over the wire as this struct.\ntype HistoryService_ReadDLQMessages_Args struct {\n\tRequest *replicator.ReadDLQMessagesRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReadDLQMessages_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReadDLQMessages_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReadDLQMessagesRequest_Read(w wire.Value) (*replicator.ReadDLQMessagesRequest, error) {\n\tvar v replicator.ReadDLQMessagesRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReadDLQMessages_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReadDLQMessages_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReadDLQMessages_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReadDLQMessages_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ReadDLQMessagesRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReadDLQMessages_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReadDLQMessages_Args struct could not be encoded.\nfunc (v *HistoryService_ReadDLQMessages_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReadDLQMessagesRequest_Decode(sr stream.Reader) (*replicator.ReadDLQMessagesRequest, error) {\n\tvar v replicator.ReadDLQMessagesRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReadDLQMessages_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReadDLQMessages_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReadDLQMessages_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ReadDLQMessagesRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReadDLQMessages_Args\n// struct.\nfunc (v *HistoryService_ReadDLQMessages_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReadDLQMessages_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReadDLQMessages_Args match the\n// provided HistoryService_ReadDLQMessages_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReadDLQMessages_Args) Equals(rhs *HistoryService_ReadDLQMessages_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReadDLQMessages_Args.\nfunc (v *HistoryService_ReadDLQMessages_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Args) GetRequest() (o *replicator.ReadDLQMessagesRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ReadDLQMessages\" for this struct.\nfunc (v *HistoryService_ReadDLQMessages_Args) MethodName() string {\n\treturn \"ReadDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ReadDLQMessages_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ReadDLQMessages_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ReadDLQMessages\n// function.\nvar HistoryService_ReadDLQMessages_Helper = struct {\n\t// Args accepts the parameters of ReadDLQMessages in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *replicator.ReadDLQMessagesRequest,\n\t) *HistoryService_ReadDLQMessages_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ReadDLQMessages.\n\t//\n\t// An error can be thrown by ReadDLQMessages only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ReadDLQMessages\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ReadDLQMessages into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ReadDLQMessages\n\t//\n\t//   value, err := ReadDLQMessages(args)\n\t//   result, err := HistoryService_ReadDLQMessages_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ReadDLQMessages: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*replicator.ReadDLQMessagesResponse, error) (*HistoryService_ReadDLQMessages_Result, error)\n\n\t// UnwrapResponse takes the result struct for ReadDLQMessages\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ReadDLQMessages threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_ReadDLQMessages_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ReadDLQMessages_Result) (*replicator.ReadDLQMessagesResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_ReadDLQMessages_Helper.Args = func(\n\t\trequest *replicator.ReadDLQMessagesRequest,\n\t) *HistoryService_ReadDLQMessages_Args {\n\t\treturn &HistoryService_ReadDLQMessages_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_ReadDLQMessages_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ReadDLQMessages_Helper.WrapResponse = func(success *replicator.ReadDLQMessagesResponse, err error) (*HistoryService_ReadDLQMessages_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReadDLQMessages_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReadDLQMessages_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReadDLQMessages_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReadDLQMessages_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReadDLQMessages_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReadDLQMessages_Result{ShardOwnershipLostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ReadDLQMessages_Helper.UnwrapResponse = func(result *HistoryService_ReadDLQMessages_Result) (success *replicator.ReadDLQMessagesResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ReadDLQMessages_Result represents the result of a HistoryService.ReadDLQMessages function call.\n//\n// The result of a ReadDLQMessages execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_ReadDLQMessages_Result struct {\n\t// Value returned by ReadDLQMessages after a successful execution.\n\tSuccess                 *replicator.ReadDLQMessagesResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError        `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError            `json:\"shardOwnershipLostError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReadDLQMessages_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReadDLQMessages_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ReadDLQMessages_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReadDLQMessagesResponse_Read(w wire.Value) (*replicator.ReadDLQMessagesResponse, error) {\n\tvar v replicator.ReadDLQMessagesResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReadDLQMessages_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReadDLQMessages_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReadDLQMessages_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReadDLQMessages_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ReadDLQMessagesResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReadDLQMessages_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReadDLQMessages_Result struct could not be encoded.\nfunc (v *HistoryService_ReadDLQMessages_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReadDLQMessagesResponse_Decode(sr stream.Reader) (*replicator.ReadDLQMessagesResponse, error) {\n\tvar v replicator.ReadDLQMessagesResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReadDLQMessages_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReadDLQMessages_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReadDLQMessages_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ReadDLQMessagesResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReadDLQMessages_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReadDLQMessages_Result\n// struct.\nfunc (v *HistoryService_ReadDLQMessages_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReadDLQMessages_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReadDLQMessages_Result match the\n// provided HistoryService_ReadDLQMessages_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReadDLQMessages_Result) Equals(rhs *HistoryService_ReadDLQMessages_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReadDLQMessages_Result.\nfunc (v *HistoryService_ReadDLQMessages_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetSuccess() (o *replicator.ReadDLQMessagesResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReadDLQMessages_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ReadDLQMessages_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ReadDLQMessages\" for this struct.\nfunc (v *HistoryService_ReadDLQMessages_Result) MethodName() string {\n\treturn \"ReadDLQMessages\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ReadDLQMessages_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ReapplyEvents_Args represents the arguments for the HistoryService.ReapplyEvents function.\n//\n// The arguments for ReapplyEvents are sent and received over the wire as this struct.\ntype HistoryService_ReapplyEvents_Args struct {\n\tReapplyEventsRequest *ReapplyEventsRequest `json:\"reapplyEventsRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReapplyEvents_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReapplyEvents_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ReapplyEventsRequest != nil {\n\t\tw, err = v.ReapplyEventsRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReapplyEventsRequest_1_Read(w wire.Value) (*ReapplyEventsRequest, error) {\n\tvar v ReapplyEventsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReapplyEvents_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReapplyEvents_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReapplyEvents_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReapplyEvents_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReapplyEventsRequest, err = _ReapplyEventsRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReapplyEvents_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReapplyEvents_Args struct could not be encoded.\nfunc (v *HistoryService_ReapplyEvents_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ReapplyEventsRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReapplyEventsRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReapplyEventsRequest_1_Decode(sr stream.Reader) (*ReapplyEventsRequest, error) {\n\tvar v ReapplyEventsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReapplyEvents_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReapplyEvents_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReapplyEvents_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ReapplyEventsRequest, err = _ReapplyEventsRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReapplyEvents_Args\n// struct.\nfunc (v *HistoryService_ReapplyEvents_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ReapplyEventsRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReapplyEventsRequest: %v\", v.ReapplyEventsRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReapplyEvents_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReapplyEvents_Args match the\n// provided HistoryService_ReapplyEvents_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReapplyEvents_Args) Equals(rhs *HistoryService_ReapplyEvents_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ReapplyEventsRequest == nil && rhs.ReapplyEventsRequest == nil) || (v.ReapplyEventsRequest != nil && rhs.ReapplyEventsRequest != nil && v.ReapplyEventsRequest.Equals(rhs.ReapplyEventsRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReapplyEvents_Args.\nfunc (v *HistoryService_ReapplyEvents_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ReapplyEventsRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"reapplyEventsRequest\", v.ReapplyEventsRequest))\n\t}\n\treturn err\n}\n\n// GetReapplyEventsRequest returns the value of ReapplyEventsRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Args) GetReapplyEventsRequest() (o *ReapplyEventsRequest) {\n\tif v != nil && v.ReapplyEventsRequest != nil {\n\t\treturn v.ReapplyEventsRequest\n\t}\n\n\treturn\n}\n\n// IsSetReapplyEventsRequest returns true if ReapplyEventsRequest is not nil.\nfunc (v *HistoryService_ReapplyEvents_Args) IsSetReapplyEventsRequest() bool {\n\treturn v != nil && v.ReapplyEventsRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ReapplyEvents\" for this struct.\nfunc (v *HistoryService_ReapplyEvents_Args) MethodName() string {\n\treturn \"ReapplyEvents\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ReapplyEvents_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ReapplyEvents_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ReapplyEvents\n// function.\nvar HistoryService_ReapplyEvents_Helper = struct {\n\t// Args accepts the parameters of ReapplyEvents in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\treapplyEventsRequest *ReapplyEventsRequest,\n\t) *HistoryService_ReapplyEvents_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ReapplyEvents.\n\t//\n\t// An error can be thrown by ReapplyEvents only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ReapplyEvents\n\t// given the error returned by it. The provided error may\n\t// be nil if ReapplyEvents did not fail.\n\t//\n\t// This allows mapping errors returned by ReapplyEvents into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ReapplyEvents\n\t//\n\t//   err := ReapplyEvents(args)\n\t//   result, err := HistoryService_ReapplyEvents_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ReapplyEvents: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_ReapplyEvents_Result, error)\n\n\t// UnwrapResponse takes the result struct for ReapplyEvents\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ReapplyEvents threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_ReapplyEvents_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ReapplyEvents_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_ReapplyEvents_Helper.Args = func(\n\t\treapplyEventsRequest *ReapplyEventsRequest,\n\t) *HistoryService_ReapplyEvents_Args {\n\t\treturn &HistoryService_ReapplyEvents_Args{\n\t\t\tReapplyEventsRequest: reapplyEventsRequest,\n\t\t}\n\t}\n\n\tHistoryService_ReapplyEvents_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ReapplyEvents_Helper.WrapResponse = func(err error) (*HistoryService_ReapplyEvents_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ReapplyEvents_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{InternalServiceError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{ServiceBusyError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReapplyEvents_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReapplyEvents_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ReapplyEvents_Helper.UnwrapResponse = func(result *HistoryService_ReapplyEvents_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ReapplyEvents_Result represents the result of a HistoryService.ReapplyEvents function call.\n//\n// The result of a ReapplyEvents execution is sent and received over the wire as this struct.\ntype HistoryService_ReapplyEvents_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tDomainNotActiveError    *shared.DomainNotActiveError `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReapplyEvents_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReapplyEvents_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ReapplyEvents_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainNotActiveError_Read(w wire.Value) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReapplyEvents_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReapplyEvents_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReapplyEvents_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReapplyEvents_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReapplyEvents_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReapplyEvents_Result struct could not be encoded.\nfunc (v *HistoryService_ReapplyEvents_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainNotActiveError_Decode(sr stream.Reader) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReapplyEvents_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReapplyEvents_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReapplyEvents_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReapplyEvents_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReapplyEvents_Result\n// struct.\nfunc (v *HistoryService_ReapplyEvents_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReapplyEvents_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReapplyEvents_Result match the\n// provided HistoryService_ReapplyEvents_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReapplyEvents_Result) Equals(rhs *HistoryService_ReapplyEvents_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReapplyEvents_Result.\nfunc (v *HistoryService_ReapplyEvents_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReapplyEvents_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ReapplyEvents_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ReapplyEvents\" for this struct.\nfunc (v *HistoryService_ReapplyEvents_Result) MethodName() string {\n\treturn \"ReapplyEvents\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ReapplyEvents_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RecordActivityTaskHeartbeat_Args represents the arguments for the HistoryService.RecordActivityTaskHeartbeat function.\n//\n// The arguments for RecordActivityTaskHeartbeat are sent and received over the wire as this struct.\ntype HistoryService_RecordActivityTaskHeartbeat_Args struct {\n\tHeartbeatRequest *RecordActivityTaskHeartbeatRequest `json:\"heartbeatRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordActivityTaskHeartbeat_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HeartbeatRequest != nil {\n\t\tw, err = v.HeartbeatRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_1_Read(w wire.Value) (*RecordActivityTaskHeartbeatRequest, error) {\n\tvar v RecordActivityTaskHeartbeatRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordActivityTaskHeartbeat_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordActivityTaskHeartbeat_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordActivityTaskHeartbeat_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordActivityTaskHeartbeat_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskHeartbeat_Args struct could not be encoded.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HeartbeatRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HeartbeatRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatRequest_1_Decode(sr stream.Reader) (*RecordActivityTaskHeartbeatRequest, error) {\n\tvar v RecordActivityTaskHeartbeatRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordActivityTaskHeartbeat_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskHeartbeat_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.HeartbeatRequest, err = _RecordActivityTaskHeartbeatRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordActivityTaskHeartbeat_Args\n// struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.HeartbeatRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatRequest: %v\", v.HeartbeatRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordActivityTaskHeartbeat_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordActivityTaskHeartbeat_Args match the\n// provided HistoryService_RecordActivityTaskHeartbeat_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) Equals(rhs *HistoryService_RecordActivityTaskHeartbeat_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatRequest == nil && rhs.HeartbeatRequest == nil) || (v.HeartbeatRequest != nil && rhs.HeartbeatRequest != nil && v.HeartbeatRequest.Equals(rhs.HeartbeatRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordActivityTaskHeartbeat_Args.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HeartbeatRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"heartbeatRequest\", v.HeartbeatRequest))\n\t}\n\treturn err\n}\n\n// GetHeartbeatRequest returns the value of HeartbeatRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) GetHeartbeatRequest() (o *RecordActivityTaskHeartbeatRequest) {\n\tif v != nil && v.HeartbeatRequest != nil {\n\t\treturn v.HeartbeatRequest\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatRequest returns true if HeartbeatRequest is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) IsSetHeartbeatRequest() bool {\n\treturn v != nil && v.HeartbeatRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordActivityTaskHeartbeat\" for this struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeat\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RecordActivityTaskHeartbeat_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RecordActivityTaskHeartbeat\n// function.\nvar HistoryService_RecordActivityTaskHeartbeat_Helper = struct {\n\t// Args accepts the parameters of RecordActivityTaskHeartbeat in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\theartbeatRequest *RecordActivityTaskHeartbeatRequest,\n\t) *HistoryService_RecordActivityTaskHeartbeat_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordActivityTaskHeartbeat.\n\t//\n\t// An error can be thrown by RecordActivityTaskHeartbeat only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordActivityTaskHeartbeat\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RecordActivityTaskHeartbeat into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RecordActivityTaskHeartbeat\n\t//\n\t//   value, err := RecordActivityTaskHeartbeat(args)\n\t//   result, err := HistoryService_RecordActivityTaskHeartbeat_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordActivityTaskHeartbeat: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RecordActivityTaskHeartbeatResponse, error) (*HistoryService_RecordActivityTaskHeartbeat_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordActivityTaskHeartbeat\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RecordActivityTaskHeartbeat threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RecordActivityTaskHeartbeat_Result) (*shared.RecordActivityTaskHeartbeatResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RecordActivityTaskHeartbeat_Helper.Args = func(\n\t\theartbeatRequest *RecordActivityTaskHeartbeatRequest,\n\t) *HistoryService_RecordActivityTaskHeartbeat_Args {\n\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Args{\n\t\t\tHeartbeatRequest: heartbeatRequest,\n\t\t}\n\t}\n\n\tHistoryService_RecordActivityTaskHeartbeat_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RecordActivityTaskHeartbeat_Helper.WrapResponse = func(success *shared.RecordActivityTaskHeartbeatResponse, err error) (*HistoryService_RecordActivityTaskHeartbeat_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskHeartbeat_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskHeartbeat_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse = func(result *HistoryService_RecordActivityTaskHeartbeat_Result) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RecordActivityTaskHeartbeat_Result represents the result of a HistoryService.RecordActivityTaskHeartbeat function call.\n//\n// The result of a RecordActivityTaskHeartbeat execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RecordActivityTaskHeartbeat_Result struct {\n\t// Value returned by RecordActivityTaskHeartbeat after a successful execution.\n\tSuccess                                *shared.RecordActivityTaskHeartbeatResponse    `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordActivityTaskHeartbeat_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskHeartbeatResponse_Read(w wire.Value) (*shared.RecordActivityTaskHeartbeatResponse, error) {\n\tvar v shared.RecordActivityTaskHeartbeatResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyCompletedError_Read(w wire.Value) (*shared.WorkflowExecutionAlreadyCompletedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyCompletedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordActivityTaskHeartbeat_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordActivityTaskHeartbeat_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordActivityTaskHeartbeat_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordActivityTaskHeartbeat_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskHeartbeat_Result struct could not be encoded.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskHeartbeatResponse_Decode(sr stream.Reader) (*shared.RecordActivityTaskHeartbeatResponse, error) {\n\tvar v shared.RecordActivityTaskHeartbeatResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyCompletedError_Decode(sr stream.Reader) (*shared.WorkflowExecutionAlreadyCompletedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyCompletedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordActivityTaskHeartbeat_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskHeartbeat_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RecordActivityTaskHeartbeatResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskHeartbeat_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordActivityTaskHeartbeat_Result\n// struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordActivityTaskHeartbeat_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordActivityTaskHeartbeat_Result match the\n// provided HistoryService_RecordActivityTaskHeartbeat_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) Equals(rhs *HistoryService_RecordActivityTaskHeartbeat_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordActivityTaskHeartbeat_Result.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetSuccess() (o *shared.RecordActivityTaskHeartbeatResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordActivityTaskHeartbeat\" for this struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) MethodName() string {\n\treturn \"RecordActivityTaskHeartbeat\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RecordActivityTaskHeartbeat_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RecordActivityTaskStarted_Args represents the arguments for the HistoryService.RecordActivityTaskStarted function.\n//\n// The arguments for RecordActivityTaskStarted are sent and received over the wire as this struct.\ntype HistoryService_RecordActivityTaskStarted_Args struct {\n\tAddRequest *RecordActivityTaskStartedRequest `json:\"addRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordActivityTaskStarted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.AddRequest != nil {\n\t\tw, err = v.AddRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskStartedRequest_Read(w wire.Value) (*RecordActivityTaskStartedRequest, error) {\n\tvar v RecordActivityTaskStartedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordActivityTaskStarted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordActivityTaskStarted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordActivityTaskStarted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AddRequest, err = _RecordActivityTaskStartedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordActivityTaskStarted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskStarted_Args struct could not be encoded.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.AddRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AddRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskStartedRequest_Decode(sr stream.Reader) (*RecordActivityTaskStartedRequest, error) {\n\tvar v RecordActivityTaskStartedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordActivityTaskStarted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskStarted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.AddRequest, err = _RecordActivityTaskStartedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordActivityTaskStarted_Args\n// struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.AddRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"AddRequest: %v\", v.AddRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordActivityTaskStarted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordActivityTaskStarted_Args match the\n// provided HistoryService_RecordActivityTaskStarted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) Equals(rhs *HistoryService_RecordActivityTaskStarted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.AddRequest == nil && rhs.AddRequest == nil) || (v.AddRequest != nil && rhs.AddRequest != nil && v.AddRequest.Equals(rhs.AddRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordActivityTaskStarted_Args.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.AddRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"addRequest\", v.AddRequest))\n\t}\n\treturn err\n}\n\n// GetAddRequest returns the value of AddRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) GetAddRequest() (o *RecordActivityTaskStartedRequest) {\n\tif v != nil && v.AddRequest != nil {\n\t\treturn v.AddRequest\n\t}\n\n\treturn\n}\n\n// IsSetAddRequest returns true if AddRequest is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) IsSetAddRequest() bool {\n\treturn v != nil && v.AddRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordActivityTaskStarted\" for this struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) MethodName() string {\n\treturn \"RecordActivityTaskStarted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RecordActivityTaskStarted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RecordActivityTaskStarted\n// function.\nvar HistoryService_RecordActivityTaskStarted_Helper = struct {\n\t// Args accepts the parameters of RecordActivityTaskStarted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\taddRequest *RecordActivityTaskStartedRequest,\n\t) *HistoryService_RecordActivityTaskStarted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordActivityTaskStarted.\n\t//\n\t// An error can be thrown by RecordActivityTaskStarted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordActivityTaskStarted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RecordActivityTaskStarted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RecordActivityTaskStarted\n\t//\n\t//   value, err := RecordActivityTaskStarted(args)\n\t//   result, err := HistoryService_RecordActivityTaskStarted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordActivityTaskStarted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*RecordActivityTaskStartedResponse, error) (*HistoryService_RecordActivityTaskStarted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordActivityTaskStarted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RecordActivityTaskStarted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RecordActivityTaskStarted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RecordActivityTaskStarted_Result) (*RecordActivityTaskStartedResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RecordActivityTaskStarted_Helper.Args = func(\n\t\taddRequest *RecordActivityTaskStartedRequest,\n\t) *HistoryService_RecordActivityTaskStarted_Args {\n\t\treturn &HistoryService_RecordActivityTaskStarted_Args{\n\t\t\tAddRequest: addRequest,\n\t\t}\n\t}\n\n\tHistoryService_RecordActivityTaskStarted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *EventAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RecordActivityTaskStarted_Helper.WrapResponse = func(success *RecordActivityTaskStartedResponse, err error) (*HistoryService_RecordActivityTaskStarted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{InternalServiceError: e}, nil\n\t\tcase *EventAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.EventAlreadyStartedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{EventAlreadyStartedError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordActivityTaskStarted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordActivityTaskStarted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RecordActivityTaskStarted_Helper.UnwrapResponse = func(result *HistoryService_RecordActivityTaskStarted_Result) (success *RecordActivityTaskStartedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EventAlreadyStartedError != nil {\n\t\t\terr = result.EventAlreadyStartedError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RecordActivityTaskStarted_Result represents the result of a HistoryService.RecordActivityTaskStarted function call.\n//\n// The result of a RecordActivityTaskStarted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RecordActivityTaskStarted_Result struct {\n\t// Value returned by RecordActivityTaskStarted after a successful execution.\n\tSuccess                                *RecordActivityTaskStartedResponse             `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEventAlreadyStartedError               *EventAlreadyStartedError                      `json:\"eventAlreadyStartedError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordActivityTaskStarted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [10]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tw, err = v.EventAlreadyStartedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RecordActivityTaskStarted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordActivityTaskStartedResponse_Read(w wire.Value) (*RecordActivityTaskStartedResponse, error) {\n\tvar v RecordActivityTaskStartedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EventAlreadyStartedError_Read(w wire.Value) (*EventAlreadyStartedError, error) {\n\tvar v EventAlreadyStartedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordActivityTaskStarted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordActivityTaskStarted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordActivityTaskStarted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RecordActivityTaskStartedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EventAlreadyStartedError, err = _EventAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordActivityTaskStarted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskStarted_Result struct could not be encoded.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventAlreadyStartedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EventAlreadyStartedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordActivityTaskStartedResponse_Decode(sr stream.Reader) (*RecordActivityTaskStartedResponse, error) {\n\tvar v RecordActivityTaskStartedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EventAlreadyStartedError_Decode(sr stream.Reader) (*EventAlreadyStartedError, error) {\n\tvar v EventAlreadyStartedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordActivityTaskStarted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordActivityTaskStarted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RecordActivityTaskStartedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EventAlreadyStartedError, err = _EventAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordActivityTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordActivityTaskStarted_Result\n// struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [10]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventAlreadyStartedError: %v\", v.EventAlreadyStartedError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordActivityTaskStarted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordActivityTaskStarted_Result match the\n// provided HistoryService_RecordActivityTaskStarted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) Equals(rhs *HistoryService_RecordActivityTaskStarted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EventAlreadyStartedError == nil && rhs.EventAlreadyStartedError == nil) || (v.EventAlreadyStartedError != nil && rhs.EventAlreadyStartedError != nil && v.EventAlreadyStartedError.Equals(rhs.EventAlreadyStartedError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordActivityTaskStarted_Result.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"eventAlreadyStartedError\", v.EventAlreadyStartedError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetSuccess() (o *RecordActivityTaskStartedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEventAlreadyStartedError returns the value of EventAlreadyStartedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetEventAlreadyStartedError() (o *EventAlreadyStartedError) {\n\tif v != nil && v.EventAlreadyStartedError != nil {\n\t\treturn v.EventAlreadyStartedError\n\t}\n\n\treturn\n}\n\n// IsSetEventAlreadyStartedError returns true if EventAlreadyStartedError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetEventAlreadyStartedError() bool {\n\treturn v != nil && v.EventAlreadyStartedError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordActivityTaskStarted\" for this struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) MethodName() string {\n\treturn \"RecordActivityTaskStarted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RecordActivityTaskStarted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RecordChildExecutionCompleted_Args represents the arguments for the HistoryService.RecordChildExecutionCompleted function.\n//\n// The arguments for RecordChildExecutionCompleted are sent and received over the wire as this struct.\ntype HistoryService_RecordChildExecutionCompleted_Args struct {\n\tCompletionRequest *RecordChildExecutionCompletedRequest `json:\"completionRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordChildExecutionCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompletionRequest != nil {\n\t\tw, err = v.CompletionRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordChildExecutionCompletedRequest_Read(w wire.Value) (*RecordChildExecutionCompletedRequest, error) {\n\tvar v RecordChildExecutionCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordChildExecutionCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordChildExecutionCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordChildExecutionCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompletionRequest, err = _RecordChildExecutionCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordChildExecutionCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordChildExecutionCompleted_Args struct could not be encoded.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompletionRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletionRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordChildExecutionCompletedRequest_Decode(sr stream.Reader) (*RecordChildExecutionCompletedRequest, error) {\n\tvar v RecordChildExecutionCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordChildExecutionCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordChildExecutionCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompletionRequest, err = _RecordChildExecutionCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordChildExecutionCompleted_Args\n// struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompletionRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionRequest: %v\", v.CompletionRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordChildExecutionCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordChildExecutionCompleted_Args match the\n// provided HistoryService_RecordChildExecutionCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) Equals(rhs *HistoryService_RecordChildExecutionCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompletionRequest == nil && rhs.CompletionRequest == nil) || (v.CompletionRequest != nil && rhs.CompletionRequest != nil && v.CompletionRequest.Equals(rhs.CompletionRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordChildExecutionCompleted_Args.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompletionRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completionRequest\", v.CompletionRequest))\n\t}\n\treturn err\n}\n\n// GetCompletionRequest returns the value of CompletionRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) GetCompletionRequest() (o *RecordChildExecutionCompletedRequest) {\n\tif v != nil && v.CompletionRequest != nil {\n\t\treturn v.CompletionRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompletionRequest returns true if CompletionRequest is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) IsSetCompletionRequest() bool {\n\treturn v != nil && v.CompletionRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordChildExecutionCompleted\" for this struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) MethodName() string {\n\treturn \"RecordChildExecutionCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RecordChildExecutionCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RecordChildExecutionCompleted\n// function.\nvar HistoryService_RecordChildExecutionCompleted_Helper = struct {\n\t// Args accepts the parameters of RecordChildExecutionCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompletionRequest *RecordChildExecutionCompletedRequest,\n\t) *HistoryService_RecordChildExecutionCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordChildExecutionCompleted.\n\t//\n\t// An error can be thrown by RecordChildExecutionCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordChildExecutionCompleted\n\t// given the error returned by it. The provided error may\n\t// be nil if RecordChildExecutionCompleted did not fail.\n\t//\n\t// This allows mapping errors returned by RecordChildExecutionCompleted into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RecordChildExecutionCompleted\n\t//\n\t//   err := RecordChildExecutionCompleted(args)\n\t//   result, err := HistoryService_RecordChildExecutionCompleted_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordChildExecutionCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RecordChildExecutionCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordChildExecutionCompleted\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RecordChildExecutionCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RecordChildExecutionCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RecordChildExecutionCompleted_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RecordChildExecutionCompleted_Helper.Args = func(\n\t\tcompletionRequest *RecordChildExecutionCompletedRequest,\n\t) *HistoryService_RecordChildExecutionCompleted_Args {\n\t\treturn &HistoryService_RecordChildExecutionCompleted_Args{\n\t\t\tCompletionRequest: completionRequest,\n\t\t}\n\t}\n\n\tHistoryService_RecordChildExecutionCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RecordChildExecutionCompleted_Helper.WrapResponse = func(err error) (*HistoryService_RecordChildExecutionCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordChildExecutionCompleted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordChildExecutionCompleted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RecordChildExecutionCompleted_Helper.UnwrapResponse = func(result *HistoryService_RecordChildExecutionCompleted_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RecordChildExecutionCompleted_Result represents the result of a HistoryService.RecordChildExecutionCompleted function call.\n//\n// The result of a RecordChildExecutionCompleted execution is sent and received over the wire as this struct.\ntype HistoryService_RecordChildExecutionCompleted_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordChildExecutionCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RecordChildExecutionCompleted_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RecordChildExecutionCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordChildExecutionCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordChildExecutionCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordChildExecutionCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordChildExecutionCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordChildExecutionCompleted_Result struct could not be encoded.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordChildExecutionCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RecordChildExecutionCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordChildExecutionCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordChildExecutionCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordChildExecutionCompleted_Result\n// struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordChildExecutionCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordChildExecutionCompleted_Result match the\n// provided HistoryService_RecordChildExecutionCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) Equals(rhs *HistoryService_RecordChildExecutionCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordChildExecutionCompleted_Result.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordChildExecutionCompleted\" for this struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) MethodName() string {\n\treturn \"RecordChildExecutionCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RecordChildExecutionCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RecordDecisionTaskStarted_Args represents the arguments for the HistoryService.RecordDecisionTaskStarted function.\n//\n// The arguments for RecordDecisionTaskStarted are sent and received over the wire as this struct.\ntype HistoryService_RecordDecisionTaskStarted_Args struct {\n\tAddRequest *RecordDecisionTaskStartedRequest `json:\"addRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordDecisionTaskStarted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.AddRequest != nil {\n\t\tw, err = v.AddRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RecordDecisionTaskStartedRequest_Read(w wire.Value) (*RecordDecisionTaskStartedRequest, error) {\n\tvar v RecordDecisionTaskStartedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RecordDecisionTaskStarted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordDecisionTaskStarted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordDecisionTaskStarted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AddRequest, err = _RecordDecisionTaskStartedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordDecisionTaskStarted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordDecisionTaskStarted_Args struct could not be encoded.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.AddRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AddRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RecordDecisionTaskStartedRequest_Decode(sr stream.Reader) (*RecordDecisionTaskStartedRequest, error) {\n\tvar v RecordDecisionTaskStartedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RecordDecisionTaskStarted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordDecisionTaskStarted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.AddRequest, err = _RecordDecisionTaskStartedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordDecisionTaskStarted_Args\n// struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.AddRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"AddRequest: %v\", v.AddRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordDecisionTaskStarted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordDecisionTaskStarted_Args match the\n// provided HistoryService_RecordDecisionTaskStarted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) Equals(rhs *HistoryService_RecordDecisionTaskStarted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.AddRequest == nil && rhs.AddRequest == nil) || (v.AddRequest != nil && rhs.AddRequest != nil && v.AddRequest.Equals(rhs.AddRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordDecisionTaskStarted_Args.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.AddRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"addRequest\", v.AddRequest))\n\t}\n\treturn err\n}\n\n// GetAddRequest returns the value of AddRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) GetAddRequest() (o *RecordDecisionTaskStartedRequest) {\n\tif v != nil && v.AddRequest != nil {\n\t\treturn v.AddRequest\n\t}\n\n\treturn\n}\n\n// IsSetAddRequest returns true if AddRequest is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) IsSetAddRequest() bool {\n\treturn v != nil && v.AddRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RecordDecisionTaskStarted\" for this struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) MethodName() string {\n\treturn \"RecordDecisionTaskStarted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RecordDecisionTaskStarted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RecordDecisionTaskStarted\n// function.\nvar HistoryService_RecordDecisionTaskStarted_Helper = struct {\n\t// Args accepts the parameters of RecordDecisionTaskStarted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\taddRequest *RecordDecisionTaskStartedRequest,\n\t) *HistoryService_RecordDecisionTaskStarted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RecordDecisionTaskStarted.\n\t//\n\t// An error can be thrown by RecordDecisionTaskStarted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RecordDecisionTaskStarted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RecordDecisionTaskStarted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RecordDecisionTaskStarted\n\t//\n\t//   value, err := RecordDecisionTaskStarted(args)\n\t//   result, err := HistoryService_RecordDecisionTaskStarted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RecordDecisionTaskStarted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*RecordDecisionTaskStartedResponse, error) (*HistoryService_RecordDecisionTaskStarted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RecordDecisionTaskStarted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RecordDecisionTaskStarted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RecordDecisionTaskStarted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RecordDecisionTaskStarted_Result) (*RecordDecisionTaskStartedResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RecordDecisionTaskStarted_Helper.Args = func(\n\t\taddRequest *RecordDecisionTaskStartedRequest,\n\t) *HistoryService_RecordDecisionTaskStarted_Args {\n\t\treturn &HistoryService_RecordDecisionTaskStarted_Args{\n\t\t\tAddRequest: addRequest,\n\t\t}\n\t}\n\n\tHistoryService_RecordDecisionTaskStarted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *EventAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RecordDecisionTaskStarted_Helper.WrapResponse = func(success *RecordDecisionTaskStartedResponse, err error) (*HistoryService_RecordDecisionTaskStarted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{InternalServiceError: e}, nil\n\t\tcase *EventAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.EventAlreadyStartedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{EventAlreadyStartedError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RecordDecisionTaskStarted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RecordDecisionTaskStarted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RecordDecisionTaskStarted_Helper.UnwrapResponse = func(result *HistoryService_RecordDecisionTaskStarted_Result) (success *RecordDecisionTaskStartedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EventAlreadyStartedError != nil {\n\t\t\terr = result.EventAlreadyStartedError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RecordDecisionTaskStarted_Result represents the result of a HistoryService.RecordDecisionTaskStarted function call.\n//\n// The result of a RecordDecisionTaskStarted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RecordDecisionTaskStarted_Result struct {\n\t// Value returned by RecordDecisionTaskStarted after a successful execution.\n\tSuccess                                *RecordDecisionTaskStartedResponse             `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEventAlreadyStartedError               *EventAlreadyStartedError                      `json:\"eventAlreadyStartedError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RecordDecisionTaskStarted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [10]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tw, err = v.EventAlreadyStartedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 9, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RecordDecisionTaskStarted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RecordDecisionTaskStarted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RecordDecisionTaskStarted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RecordDecisionTaskStarted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RecordDecisionTaskStartedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EventAlreadyStartedError, err = _EventAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 9:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordDecisionTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RecordDecisionTaskStarted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RecordDecisionTaskStarted_Result struct could not be encoded.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventAlreadyStartedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EventAlreadyStartedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 9, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordDecisionTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RecordDecisionTaskStarted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RecordDecisionTaskStarted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RecordDecisionTaskStartedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EventAlreadyStartedError, err = _EventAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 9 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RecordDecisionTaskStarted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RecordDecisionTaskStarted_Result\n// struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [10]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventAlreadyStartedError: %v\", v.EventAlreadyStartedError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RecordDecisionTaskStarted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RecordDecisionTaskStarted_Result match the\n// provided HistoryService_RecordDecisionTaskStarted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) Equals(rhs *HistoryService_RecordDecisionTaskStarted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EventAlreadyStartedError == nil && rhs.EventAlreadyStartedError == nil) || (v.EventAlreadyStartedError != nil && rhs.EventAlreadyStartedError != nil && v.EventAlreadyStartedError.Equals(rhs.EventAlreadyStartedError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RecordDecisionTaskStarted_Result.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EventAlreadyStartedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"eventAlreadyStartedError\", v.EventAlreadyStartedError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetSuccess() (o *RecordDecisionTaskStartedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEventAlreadyStartedError returns the value of EventAlreadyStartedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetEventAlreadyStartedError() (o *EventAlreadyStartedError) {\n\tif v != nil && v.EventAlreadyStartedError != nil {\n\t\treturn v.EventAlreadyStartedError\n\t}\n\n\treturn\n}\n\n// IsSetEventAlreadyStartedError returns true if EventAlreadyStartedError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetEventAlreadyStartedError() bool {\n\treturn v != nil && v.EventAlreadyStartedError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RecordDecisionTaskStarted\" for this struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) MethodName() string {\n\treturn \"RecordDecisionTaskStarted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RecordDecisionTaskStarted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RefreshWorkflowTasks_Args represents the arguments for the HistoryService.RefreshWorkflowTasks function.\n//\n// The arguments for RefreshWorkflowTasks are sent and received over the wire as this struct.\ntype HistoryService_RefreshWorkflowTasks_Args struct {\n\tRequest *RefreshWorkflowTasksRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RefreshWorkflowTasks_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RefreshWorkflowTasksRequest_1_Read(w wire.Value) (*RefreshWorkflowTasksRequest, error) {\n\tvar v RefreshWorkflowTasksRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RefreshWorkflowTasks_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RefreshWorkflowTasks_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RefreshWorkflowTasks_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RefreshWorkflowTasks_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RefreshWorkflowTasks_Args struct could not be encoded.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RefreshWorkflowTasksRequest_1_Decode(sr stream.Reader) (*RefreshWorkflowTasksRequest, error) {\n\tvar v RefreshWorkflowTasksRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RefreshWorkflowTasks_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RefreshWorkflowTasks_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RefreshWorkflowTasksRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RefreshWorkflowTasks_Args\n// struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RefreshWorkflowTasks_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RefreshWorkflowTasks_Args match the\n// provided HistoryService_RefreshWorkflowTasks_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) Equals(rhs *HistoryService_RefreshWorkflowTasks_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RefreshWorkflowTasks_Args.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) GetRequest() (o *RefreshWorkflowTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RefreshWorkflowTasks_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RefreshWorkflowTasks\n// function.\nvar HistoryService_RefreshWorkflowTasks_Helper = struct {\n\t// Args accepts the parameters of RefreshWorkflowTasks in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *RefreshWorkflowTasksRequest,\n\t) *HistoryService_RefreshWorkflowTasks_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RefreshWorkflowTasks.\n\t//\n\t// An error can be thrown by RefreshWorkflowTasks only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RefreshWorkflowTasks\n\t// given the error returned by it. The provided error may\n\t// be nil if RefreshWorkflowTasks did not fail.\n\t//\n\t// This allows mapping errors returned by RefreshWorkflowTasks into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RefreshWorkflowTasks\n\t//\n\t//   err := RefreshWorkflowTasks(args)\n\t//   result, err := HistoryService_RefreshWorkflowTasks_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RefreshWorkflowTasks: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RefreshWorkflowTasks_Result, error)\n\n\t// UnwrapResponse takes the result struct for RefreshWorkflowTasks\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RefreshWorkflowTasks threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RefreshWorkflowTasks_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RefreshWorkflowTasks_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RefreshWorkflowTasks_Helper.Args = func(\n\t\trequest *RefreshWorkflowTasksRequest,\n\t) *HistoryService_RefreshWorkflowTasks_Args {\n\t\treturn &HistoryService_RefreshWorkflowTasks_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_RefreshWorkflowTasks_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RefreshWorkflowTasks_Helper.WrapResponse = func(err error) (*HistoryService_RefreshWorkflowTasks_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{InternalServiceError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{DomainNotActiveError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RefreshWorkflowTasks_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RefreshWorkflowTasks_Result{EntityNotExistError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RefreshWorkflowTasks_Helper.UnwrapResponse = func(result *HistoryService_RefreshWorkflowTasks_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RefreshWorkflowTasks_Result represents the result of a HistoryService.RefreshWorkflowTasks function call.\n//\n// The result of a RefreshWorkflowTasks execution is sent and received over the wire as this struct.\ntype HistoryService_RefreshWorkflowTasks_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tDomainNotActiveError    *shared.DomainNotActiveError `json:\"domainNotActiveError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RefreshWorkflowTasks_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RefreshWorkflowTasks_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RefreshWorkflowTasks_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RefreshWorkflowTasks_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RefreshWorkflowTasks_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RefreshWorkflowTasks_Result struct could not be encoded.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RefreshWorkflowTasks_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RefreshWorkflowTasks_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RefreshWorkflowTasks_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RefreshWorkflowTasks_Result\n// struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RefreshWorkflowTasks_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RefreshWorkflowTasks_Result match the\n// provided HistoryService_RefreshWorkflowTasks_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) Equals(rhs *HistoryService_RefreshWorkflowTasks_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RefreshWorkflowTasks_Result.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RefreshWorkflowTasks\" for this struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) MethodName() string {\n\treturn \"RefreshWorkflowTasks\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RefreshWorkflowTasks_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RemoveSignalMutableState_Args represents the arguments for the HistoryService.RemoveSignalMutableState function.\n//\n// The arguments for RemoveSignalMutableState are sent and received over the wire as this struct.\ntype HistoryService_RemoveSignalMutableState_Args struct {\n\tRemoveRequest *RemoveSignalMutableStateRequest `json:\"removeRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RemoveSignalMutableState_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RemoveSignalMutableState_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RemoveRequest != nil {\n\t\tw, err = v.RemoveRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RemoveSignalMutableStateRequest_Read(w wire.Value) (*RemoveSignalMutableStateRequest, error) {\n\tvar v RemoveSignalMutableStateRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RemoveSignalMutableState_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RemoveSignalMutableState_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RemoveSignalMutableState_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RemoveSignalMutableState_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RemoveRequest, err = _RemoveSignalMutableStateRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RemoveSignalMutableState_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RemoveSignalMutableState_Args struct could not be encoded.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RemoveRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RemoveRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RemoveSignalMutableStateRequest_Decode(sr stream.Reader) (*RemoveSignalMutableStateRequest, error) {\n\tvar v RemoveSignalMutableStateRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RemoveSignalMutableState_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RemoveSignalMutableState_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.RemoveRequest, err = _RemoveSignalMutableStateRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RemoveSignalMutableState_Args\n// struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RemoveRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"RemoveRequest: %v\", v.RemoveRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RemoveSignalMutableState_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RemoveSignalMutableState_Args match the\n// provided HistoryService_RemoveSignalMutableState_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) Equals(rhs *HistoryService_RemoveSignalMutableState_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.RemoveRequest == nil && rhs.RemoveRequest == nil) || (v.RemoveRequest != nil && rhs.RemoveRequest != nil && v.RemoveRequest.Equals(rhs.RemoveRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RemoveSignalMutableState_Args.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RemoveRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"removeRequest\", v.RemoveRequest))\n\t}\n\treturn err\n}\n\n// GetRemoveRequest returns the value of RemoveRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) GetRemoveRequest() (o *RemoveSignalMutableStateRequest) {\n\tif v != nil && v.RemoveRequest != nil {\n\t\treturn v.RemoveRequest\n\t}\n\n\treturn\n}\n\n// IsSetRemoveRequest returns true if RemoveRequest is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) IsSetRemoveRequest() bool {\n\treturn v != nil && v.RemoveRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RemoveSignalMutableState\" for this struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) MethodName() string {\n\treturn \"RemoveSignalMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RemoveSignalMutableState_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RemoveSignalMutableState\n// function.\nvar HistoryService_RemoveSignalMutableState_Helper = struct {\n\t// Args accepts the parameters of RemoveSignalMutableState in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tremoveRequest *RemoveSignalMutableStateRequest,\n\t) *HistoryService_RemoveSignalMutableState_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RemoveSignalMutableState.\n\t//\n\t// An error can be thrown by RemoveSignalMutableState only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RemoveSignalMutableState\n\t// given the error returned by it. The provided error may\n\t// be nil if RemoveSignalMutableState did not fail.\n\t//\n\t// This allows mapping errors returned by RemoveSignalMutableState into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RemoveSignalMutableState\n\t//\n\t//   err := RemoveSignalMutableState(args)\n\t//   result, err := HistoryService_RemoveSignalMutableState_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RemoveSignalMutableState: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RemoveSignalMutableState_Result, error)\n\n\t// UnwrapResponse takes the result struct for RemoveSignalMutableState\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RemoveSignalMutableState threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RemoveSignalMutableState_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RemoveSignalMutableState_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RemoveSignalMutableState_Helper.Args = func(\n\t\tremoveRequest *RemoveSignalMutableStateRequest,\n\t) *HistoryService_RemoveSignalMutableState_Args {\n\t\treturn &HistoryService_RemoveSignalMutableState_Args{\n\t\t\tRemoveRequest: removeRequest,\n\t\t}\n\t}\n\n\tHistoryService_RemoveSignalMutableState_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RemoveSignalMutableState_Helper.WrapResponse = func(err error) (*HistoryService_RemoveSignalMutableState_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveSignalMutableState_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveSignalMutableState_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RemoveSignalMutableState_Helper.UnwrapResponse = func(result *HistoryService_RemoveSignalMutableState_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RemoveSignalMutableState_Result represents the result of a HistoryService.RemoveSignalMutableState function call.\n//\n// The result of a RemoveSignalMutableState execution is sent and received over the wire as this struct.\ntype HistoryService_RemoveSignalMutableState_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RemoveSignalMutableState_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RemoveSignalMutableState_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RemoveSignalMutableState_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RemoveSignalMutableState_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RemoveSignalMutableState_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RemoveSignalMutableState_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RemoveSignalMutableState_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveSignalMutableState_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RemoveSignalMutableState_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RemoveSignalMutableState_Result struct could not be encoded.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveSignalMutableState_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RemoveSignalMutableState_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RemoveSignalMutableState_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveSignalMutableState_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RemoveSignalMutableState_Result\n// struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RemoveSignalMutableState_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RemoveSignalMutableState_Result match the\n// provided HistoryService_RemoveSignalMutableState_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) Equals(rhs *HistoryService_RemoveSignalMutableState_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RemoveSignalMutableState_Result.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RemoveSignalMutableState\" for this struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) MethodName() string {\n\treturn \"RemoveSignalMutableState\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RemoveSignalMutableState_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RemoveTask_Args represents the arguments for the HistoryService.RemoveTask function.\n//\n// The arguments for RemoveTask are sent and received over the wire as this struct.\ntype HistoryService_RemoveTask_Args struct {\n\tRequest *shared.RemoveTaskRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RemoveTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RemoveTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RemoveTaskRequest_Read(w wire.Value) (*shared.RemoveTaskRequest, error) {\n\tvar v shared.RemoveTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RemoveTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RemoveTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RemoveTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RemoveTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RemoveTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RemoveTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RemoveTask_Args struct could not be encoded.\nfunc (v *HistoryService_RemoveTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RemoveTaskRequest_Decode(sr stream.Reader) (*shared.RemoveTaskRequest, error) {\n\tvar v shared.RemoveTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RemoveTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RemoveTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RemoveTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RemoveTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RemoveTask_Args\n// struct.\nfunc (v *HistoryService_RemoveTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RemoveTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RemoveTask_Args match the\n// provided HistoryService_RemoveTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RemoveTask_Args) Equals(rhs *HistoryService_RemoveTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RemoveTask_Args.\nfunc (v *HistoryService_RemoveTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveTask_Args) GetRequest() (o *shared.RemoveTaskRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_RemoveTask_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RemoveTask\" for this struct.\nfunc (v *HistoryService_RemoveTask_Args) MethodName() string {\n\treturn \"RemoveTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RemoveTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RemoveTask_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RemoveTask\n// function.\nvar HistoryService_RemoveTask_Helper = struct {\n\t// Args accepts the parameters of RemoveTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RemoveTaskRequest,\n\t) *HistoryService_RemoveTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RemoveTask.\n\t//\n\t// An error can be thrown by RemoveTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RemoveTask\n\t// given the error returned by it. The provided error may\n\t// be nil if RemoveTask did not fail.\n\t//\n\t// This allows mapping errors returned by RemoveTask into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RemoveTask\n\t//\n\t//   err := RemoveTask(args)\n\t//   result, err := HistoryService_RemoveTask_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RemoveTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RemoveTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for RemoveTask\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RemoveTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RemoveTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RemoveTask_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RemoveTask_Helper.Args = func(\n\t\trequest *shared.RemoveTaskRequest,\n\t) *HistoryService_RemoveTask_Args {\n\t\treturn &HistoryService_RemoveTask_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_RemoveTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RemoveTask_Helper.WrapResponse = func(err error) (*HistoryService_RemoveTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RemoveTask_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RemoveTask_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RemoveTask_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RemoveTask_Helper.UnwrapResponse = func(result *HistoryService_RemoveTask_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RemoveTask_Result represents the result of a HistoryService.RemoveTask function call.\n//\n// The result of a RemoveTask execution is sent and received over the wire as this struct.\ntype HistoryService_RemoveTask_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RemoveTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RemoveTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RemoveTask_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RemoveTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RemoveTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RemoveTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RemoveTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RemoveTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RemoveTask_Result struct could not be encoded.\nfunc (v *HistoryService_RemoveTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RemoveTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RemoveTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RemoveTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RemoveTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RemoveTask_Result\n// struct.\nfunc (v *HistoryService_RemoveTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RemoveTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RemoveTask_Result match the\n// provided HistoryService_RemoveTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RemoveTask_Result) Equals(rhs *HistoryService_RemoveTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RemoveTask_Result.\nfunc (v *HistoryService_RemoveTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RemoveTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RemoveTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RemoveTask_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_RemoveTask_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RemoveTask\" for this struct.\nfunc (v *HistoryService_RemoveTask_Result) MethodName() string {\n\treturn \"RemoveTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RemoveTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ReplicateEventsV2_Args represents the arguments for the HistoryService.ReplicateEventsV2 function.\n//\n// The arguments for ReplicateEventsV2 are sent and received over the wire as this struct.\ntype HistoryService_ReplicateEventsV2_Args struct {\n\tReplicateV2Request *ReplicateEventsV2Request `json:\"replicateV2Request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReplicateEventsV2_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReplicateEventsV2_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ReplicateV2Request != nil {\n\t\tw, err = v.ReplicateV2Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicateEventsV2Request_Read(w wire.Value) (*ReplicateEventsV2Request, error) {\n\tvar v ReplicateEventsV2Request\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReplicateEventsV2_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReplicateEventsV2_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReplicateEventsV2_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReplicateEventsV2_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicateV2Request, err = _ReplicateEventsV2Request_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReplicateEventsV2_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReplicateEventsV2_Args struct could not be encoded.\nfunc (v *HistoryService_ReplicateEventsV2_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ReplicateV2Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicateV2Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicateEventsV2Request_Decode(sr stream.Reader) (*ReplicateEventsV2Request, error) {\n\tvar v ReplicateEventsV2Request\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReplicateEventsV2_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReplicateEventsV2_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReplicateEventsV2_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicateV2Request, err = _ReplicateEventsV2Request_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReplicateEventsV2_Args\n// struct.\nfunc (v *HistoryService_ReplicateEventsV2_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ReplicateV2Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicateV2Request: %v\", v.ReplicateV2Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReplicateEventsV2_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReplicateEventsV2_Args match the\n// provided HistoryService_ReplicateEventsV2_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReplicateEventsV2_Args) Equals(rhs *HistoryService_ReplicateEventsV2_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ReplicateV2Request == nil && rhs.ReplicateV2Request == nil) || (v.ReplicateV2Request != nil && rhs.ReplicateV2Request != nil && v.ReplicateV2Request.Equals(rhs.ReplicateV2Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReplicateEventsV2_Args.\nfunc (v *HistoryService_ReplicateEventsV2_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ReplicateV2Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicateV2Request\", v.ReplicateV2Request))\n\t}\n\treturn err\n}\n\n// GetReplicateV2Request returns the value of ReplicateV2Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Args) GetReplicateV2Request() (o *ReplicateEventsV2Request) {\n\tif v != nil && v.ReplicateV2Request != nil {\n\t\treturn v.ReplicateV2Request\n\t}\n\n\treturn\n}\n\n// IsSetReplicateV2Request returns true if ReplicateV2Request is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Args) IsSetReplicateV2Request() bool {\n\treturn v != nil && v.ReplicateV2Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ReplicateEventsV2\" for this struct.\nfunc (v *HistoryService_ReplicateEventsV2_Args) MethodName() string {\n\treturn \"ReplicateEventsV2\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ReplicateEventsV2_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ReplicateEventsV2_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ReplicateEventsV2\n// function.\nvar HistoryService_ReplicateEventsV2_Helper = struct {\n\t// Args accepts the parameters of ReplicateEventsV2 in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\treplicateV2Request *ReplicateEventsV2Request,\n\t) *HistoryService_ReplicateEventsV2_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ReplicateEventsV2.\n\t//\n\t// An error can be thrown by ReplicateEventsV2 only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ReplicateEventsV2\n\t// given the error returned by it. The provided error may\n\t// be nil if ReplicateEventsV2 did not fail.\n\t//\n\t// This allows mapping errors returned by ReplicateEventsV2 into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ReplicateEventsV2\n\t//\n\t//   err := ReplicateEventsV2(args)\n\t//   result, err := HistoryService_ReplicateEventsV2_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ReplicateEventsV2: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_ReplicateEventsV2_Result, error)\n\n\t// UnwrapResponse takes the result struct for ReplicateEventsV2\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ReplicateEventsV2 threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_ReplicateEventsV2_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ReplicateEventsV2_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_ReplicateEventsV2_Helper.Args = func(\n\t\treplicateV2Request *ReplicateEventsV2Request,\n\t) *HistoryService_ReplicateEventsV2_Args {\n\t\treturn &HistoryService_ReplicateEventsV2_Args{\n\t\t\tReplicateV2Request: replicateV2Request,\n\t\t}\n\t}\n\n\tHistoryService_ReplicateEventsV2_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.RetryTaskV2Error:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ReplicateEventsV2_Helper.WrapResponse = func(err error) (*HistoryService_ReplicateEventsV2_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{LimitExceededError: e}, nil\n\t\tcase *shared.RetryTaskV2Error:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.RetryTaskError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{RetryTaskError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ReplicateEventsV2_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ReplicateEventsV2_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ReplicateEventsV2_Helper.UnwrapResponse = func(result *HistoryService_ReplicateEventsV2_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.RetryTaskError != nil {\n\t\t\terr = result.RetryTaskError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ReplicateEventsV2_Result represents the result of a HistoryService.ReplicateEventsV2 function call.\n//\n// The result of a ReplicateEventsV2 execution is sent and received over the wire as this struct.\ntype HistoryService_ReplicateEventsV2_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tRetryTaskError          *shared.RetryTaskV2Error     `json:\"retryTaskError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ReplicateEventsV2_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ReplicateEventsV2_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.RetryTaskError != nil {\n\t\tw, err = v.RetryTaskError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ReplicateEventsV2_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RetryTaskV2Error_Read(w wire.Value) (*shared.RetryTaskV2Error, error) {\n\tvar v shared.RetryTaskV2Error\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ReplicateEventsV2_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ReplicateEventsV2_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ReplicateEventsV2_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ReplicateEventsV2_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryTaskError, err = _RetryTaskV2Error_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReplicateEventsV2_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ReplicateEventsV2_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ReplicateEventsV2_Result struct could not be encoded.\nfunc (v *HistoryService_ReplicateEventsV2_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryTaskError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryTaskError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReplicateEventsV2_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RetryTaskV2Error_Decode(sr stream.Reader) (*shared.RetryTaskV2Error, error) {\n\tvar v shared.RetryTaskV2Error\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ReplicateEventsV2_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ReplicateEventsV2_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ReplicateEventsV2_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.RetryTaskError, err = _RetryTaskV2Error_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ReplicateEventsV2_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ReplicateEventsV2_Result\n// struct.\nfunc (v *HistoryService_ReplicateEventsV2_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.RetryTaskError != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryTaskError: %v\", v.RetryTaskError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ReplicateEventsV2_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ReplicateEventsV2_Result match the\n// provided HistoryService_ReplicateEventsV2_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ReplicateEventsV2_Result) Equals(rhs *HistoryService_ReplicateEventsV2_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.RetryTaskError == nil && rhs.RetryTaskError == nil) || (v.RetryTaskError != nil && rhs.RetryTaskError != nil && v.RetryTaskError.Equals(rhs.RetryTaskError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ReplicateEventsV2_Result.\nfunc (v *HistoryService_ReplicateEventsV2_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.RetryTaskError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryTaskError\", v.RetryTaskError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetRetryTaskError returns the value of RetryTaskError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetRetryTaskError() (o *shared.RetryTaskV2Error) {\n\tif v != nil && v.RetryTaskError != nil {\n\t\treturn v.RetryTaskError\n\t}\n\n\treturn\n}\n\n// IsSetRetryTaskError returns true if RetryTaskError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetRetryTaskError() bool {\n\treturn v != nil && v.RetryTaskError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ReplicateEventsV2_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ReplicateEventsV2_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ReplicateEventsV2\" for this struct.\nfunc (v *HistoryService_ReplicateEventsV2_Result) MethodName() string {\n\treturn \"ReplicateEventsV2\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ReplicateEventsV2_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RequestCancelWorkflowExecution_Args represents the arguments for the HistoryService.RequestCancelWorkflowExecution function.\n//\n// The arguments for RequestCancelWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_RequestCancelWorkflowExecution_Args struct {\n\tCancelRequest *RequestCancelWorkflowExecutionRequest `json:\"cancelRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RequestCancelWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CancelRequest != nil {\n\t\tw, err = v.CancelRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_1_Read(w wire.Value) (*RequestCancelWorkflowExecutionRequest, error) {\n\tvar v RequestCancelWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RequestCancelWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RequestCancelWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RequestCancelWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RequestCancelWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RequestCancelWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CancelRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RequestCancelWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*RequestCancelWorkflowExecutionRequest, error) {\n\tvar v RequestCancelWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RequestCancelWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RequestCancelWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CancelRequest, err = _RequestCancelWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RequestCancelWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CancelRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequest: %v\", v.CancelRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RequestCancelWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RequestCancelWorkflowExecution_Args match the\n// provided HistoryService_RequestCancelWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) Equals(rhs *HistoryService_RequestCancelWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CancelRequest == nil && rhs.CancelRequest == nil) || (v.CancelRequest != nil && rhs.CancelRequest != nil && v.CancelRequest.Equals(rhs.CancelRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RequestCancelWorkflowExecution_Args.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CancelRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelRequest\", v.CancelRequest))\n\t}\n\treturn err\n}\n\n// GetCancelRequest returns the value of CancelRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) GetCancelRequest() (o *RequestCancelWorkflowExecutionRequest) {\n\tif v != nil && v.CancelRequest != nil {\n\t\treturn v.CancelRequest\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequest returns true if CancelRequest is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) IsSetCancelRequest() bool {\n\treturn v != nil && v.CancelRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RequestCancelWorkflowExecution\" for this struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) MethodName() string {\n\treturn \"RequestCancelWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RequestCancelWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RequestCancelWorkflowExecution\n// function.\nvar HistoryService_RequestCancelWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of RequestCancelWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcancelRequest *RequestCancelWorkflowExecutionRequest,\n\t) *HistoryService_RequestCancelWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RequestCancelWorkflowExecution.\n\t//\n\t// An error can be thrown by RequestCancelWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RequestCancelWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if RequestCancelWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by RequestCancelWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RequestCancelWorkflowExecution\n\t//\n\t//   err := RequestCancelWorkflowExecution(args)\n\t//   result, err := HistoryService_RequestCancelWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RequestCancelWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RequestCancelWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for RequestCancelWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RequestCancelWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RequestCancelWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RequestCancelWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RequestCancelWorkflowExecution_Helper.Args = func(\n\t\tcancelRequest *RequestCancelWorkflowExecutionRequest,\n\t) *HistoryService_RequestCancelWorkflowExecution_Args {\n\t\treturn &HistoryService_RequestCancelWorkflowExecution_Args{\n\t\t\tCancelRequest: cancelRequest,\n\t\t}\n\t}\n\n\tHistoryService_RequestCancelWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.CancellationAlreadyRequestedError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RequestCancelWorkflowExecution_Helper.WrapResponse = func(err error) (*HistoryService_RequestCancelWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.CancellationAlreadyRequestedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.CancellationAlreadyRequestedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{CancellationAlreadyRequestedError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RequestCancelWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RequestCancelWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RequestCancelWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_RequestCancelWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.CancellationAlreadyRequestedError != nil {\n\t\t\terr = result.CancellationAlreadyRequestedError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RequestCancelWorkflowExecution_Result represents the result of a HistoryService.RequestCancelWorkflowExecution function call.\n//\n// The result of a RequestCancelWorkflowExecution execution is sent and received over the wire as this struct.\ntype HistoryService_RequestCancelWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tCancellationAlreadyRequestedError      *shared.CancellationAlreadyRequestedError      `json:\"cancellationAlreadyRequestedError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RequestCancelWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tw, err = v.CancellationAlreadyRequestedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CancellationAlreadyRequestedError_Read(w wire.Value) (*shared.CancellationAlreadyRequestedError, error) {\n\tvar v shared.CancellationAlreadyRequestedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RequestCancelWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RequestCancelWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RequestCancelWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancellationAlreadyRequestedError, err = _CancellationAlreadyRequestedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RequestCancelWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RequestCancelWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancellationAlreadyRequestedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CancellationAlreadyRequestedError_Decode(sr stream.Reader) (*shared.CancellationAlreadyRequestedError, error) {\n\tvar v shared.CancellationAlreadyRequestedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RequestCancelWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RequestCancelWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.CancellationAlreadyRequestedError, err = _CancellationAlreadyRequestedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RequestCancelWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RequestCancelWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancellationAlreadyRequestedError: %v\", v.CancellationAlreadyRequestedError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RequestCancelWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RequestCancelWorkflowExecution_Result match the\n// provided HistoryService_RequestCancelWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) Equals(rhs *HistoryService_RequestCancelWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.CancellationAlreadyRequestedError == nil && rhs.CancellationAlreadyRequestedError == nil) || (v.CancellationAlreadyRequestedError != nil && rhs.CancellationAlreadyRequestedError != nil && v.CancellationAlreadyRequestedError.Equals(rhs.CancellationAlreadyRequestedError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RequestCancelWorkflowExecution_Result.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.CancellationAlreadyRequestedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancellationAlreadyRequestedError\", v.CancellationAlreadyRequestedError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetCancellationAlreadyRequestedError returns the value of CancellationAlreadyRequestedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetCancellationAlreadyRequestedError() (o *shared.CancellationAlreadyRequestedError) {\n\tif v != nil && v.CancellationAlreadyRequestedError != nil {\n\t\treturn v.CancellationAlreadyRequestedError\n\t}\n\n\treturn\n}\n\n// IsSetCancellationAlreadyRequestedError returns true if CancellationAlreadyRequestedError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetCancellationAlreadyRequestedError() bool {\n\treturn v != nil && v.CancellationAlreadyRequestedError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RequestCancelWorkflowExecution\" for this struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) MethodName() string {\n\treturn \"RequestCancelWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RequestCancelWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ResetQueue_Args represents the arguments for the HistoryService.ResetQueue function.\n//\n// The arguments for ResetQueue are sent and received over the wire as this struct.\ntype HistoryService_ResetQueue_Args struct {\n\tRequest *shared.ResetQueueRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetQueue_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetQueue_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetQueueRequest_Read(w wire.Value) (*shared.ResetQueueRequest, error) {\n\tvar v shared.ResetQueueRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ResetQueue_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetQueue_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetQueue_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetQueue_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ResetQueueRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetQueue_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetQueue_Args struct could not be encoded.\nfunc (v *HistoryService_ResetQueue_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetQueueRequest_Decode(sr stream.Reader) (*shared.ResetQueueRequest, error) {\n\tvar v shared.ResetQueueRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ResetQueue_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetQueue_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetQueue_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ResetQueueRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetQueue_Args\n// struct.\nfunc (v *HistoryService_ResetQueue_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetQueue_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetQueue_Args match the\n// provided HistoryService_ResetQueue_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetQueue_Args) Equals(rhs *HistoryService_ResetQueue_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetQueue_Args.\nfunc (v *HistoryService_ResetQueue_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetQueue_Args) GetRequest() (o *shared.ResetQueueRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_ResetQueue_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetQueue\" for this struct.\nfunc (v *HistoryService_ResetQueue_Args) MethodName() string {\n\treturn \"ResetQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ResetQueue_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ResetQueue_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ResetQueue\n// function.\nvar HistoryService_ResetQueue_Helper = struct {\n\t// Args accepts the parameters of ResetQueue in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.ResetQueueRequest,\n\t) *HistoryService_ResetQueue_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetQueue.\n\t//\n\t// An error can be thrown by ResetQueue only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetQueue\n\t// given the error returned by it. The provided error may\n\t// be nil if ResetQueue did not fail.\n\t//\n\t// This allows mapping errors returned by ResetQueue into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ResetQueue\n\t//\n\t//   err := ResetQueue(args)\n\t//   result, err := HistoryService_ResetQueue_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetQueue: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_ResetQueue_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetQueue\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ResetQueue threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_ResetQueue_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ResetQueue_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_ResetQueue_Helper.Args = func(\n\t\trequest *shared.ResetQueueRequest,\n\t) *HistoryService_ResetQueue_Args {\n\t\treturn &HistoryService_ResetQueue_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_ResetQueue_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.AccessDeniedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ResetQueue_Helper.WrapResponse = func(err error) (*HistoryService_ResetQueue_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ResetQueue_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetQueue_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetQueue_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetQueue_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetQueue_Result{InternalServiceError: e}, nil\n\t\tcase *shared.AccessDeniedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetQueue_Result.AccessDeniedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetQueue_Result{AccessDeniedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ResetQueue_Helper.UnwrapResponse = func(result *HistoryService_ResetQueue_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.AccessDeniedError != nil {\n\t\t\terr = result.AccessDeniedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ResetQueue_Result represents the result of a HistoryService.ResetQueue function call.\n//\n// The result of a ResetQueue execution is sent and received over the wire as this struct.\ntype HistoryService_ResetQueue_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tAccessDeniedError    *shared.AccessDeniedError    `json:\"accessDeniedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetQueue_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetQueue_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tw, err = v.AccessDeniedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ResetQueue_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_ResetQueue_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetQueue_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetQueue_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetQueue_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetQueue_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetQueue_Result struct could not be encoded.\nfunc (v *HistoryService_ResetQueue_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AccessDeniedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AccessDeniedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_ResetQueue_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetQueue_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetQueue_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.AccessDeniedError, err = _AccessDeniedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetQueue_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetQueue_Result\n// struct.\nfunc (v *HistoryService_ResetQueue_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.AccessDeniedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"AccessDeniedError: %v\", v.AccessDeniedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetQueue_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetQueue_Result match the\n// provided HistoryService_ResetQueue_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetQueue_Result) Equals(rhs *HistoryService_ResetQueue_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.AccessDeniedError == nil && rhs.AccessDeniedError == nil) || (v.AccessDeniedError != nil && rhs.AccessDeniedError != nil && v.AccessDeniedError.Equals(rhs.AccessDeniedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetQueue_Result.\nfunc (v *HistoryService_ResetQueue_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.AccessDeniedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"accessDeniedError\", v.AccessDeniedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetQueue_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ResetQueue_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetQueue_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ResetQueue_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetAccessDeniedError returns the value of AccessDeniedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetQueue_Result) GetAccessDeniedError() (o *shared.AccessDeniedError) {\n\tif v != nil && v.AccessDeniedError != nil {\n\t\treturn v.AccessDeniedError\n\t}\n\n\treturn\n}\n\n// IsSetAccessDeniedError returns true if AccessDeniedError is not nil.\nfunc (v *HistoryService_ResetQueue_Result) IsSetAccessDeniedError() bool {\n\treturn v != nil && v.AccessDeniedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetQueue\" for this struct.\nfunc (v *HistoryService_ResetQueue_Result) MethodName() string {\n\treturn \"ResetQueue\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ResetQueue_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ResetStickyTaskList_Args represents the arguments for the HistoryService.ResetStickyTaskList function.\n//\n// The arguments for ResetStickyTaskList are sent and received over the wire as this struct.\ntype HistoryService_ResetStickyTaskList_Args struct {\n\tResetRequest *ResetStickyTaskListRequest `json:\"resetRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetStickyTaskList_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetStickyTaskList_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ResetRequest != nil {\n\t\tw, err = v.ResetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetStickyTaskListRequest_Read(w wire.Value) (*ResetStickyTaskListRequest, error) {\n\tvar v ResetStickyTaskListRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ResetStickyTaskList_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetStickyTaskList_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetStickyTaskList_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetStickyTaskList_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ResetRequest, err = _ResetStickyTaskListRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetStickyTaskList_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetStickyTaskList_Args struct could not be encoded.\nfunc (v *HistoryService_ResetStickyTaskList_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ResetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetStickyTaskListRequest_Decode(sr stream.Reader) (*ResetStickyTaskListRequest, error) {\n\tvar v ResetStickyTaskListRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ResetStickyTaskList_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetStickyTaskList_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetStickyTaskList_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ResetRequest, err = _ResetStickyTaskListRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetStickyTaskList_Args\n// struct.\nfunc (v *HistoryService_ResetStickyTaskList_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ResetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResetRequest: %v\", v.ResetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetStickyTaskList_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetStickyTaskList_Args match the\n// provided HistoryService_ResetStickyTaskList_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetStickyTaskList_Args) Equals(rhs *HistoryService_ResetStickyTaskList_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ResetRequest == nil && rhs.ResetRequest == nil) || (v.ResetRequest != nil && rhs.ResetRequest != nil && v.ResetRequest.Equals(rhs.ResetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetStickyTaskList_Args.\nfunc (v *HistoryService_ResetStickyTaskList_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ResetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resetRequest\", v.ResetRequest))\n\t}\n\treturn err\n}\n\n// GetResetRequest returns the value of ResetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Args) GetResetRequest() (o *ResetStickyTaskListRequest) {\n\tif v != nil && v.ResetRequest != nil {\n\t\treturn v.ResetRequest\n\t}\n\n\treturn\n}\n\n// IsSetResetRequest returns true if ResetRequest is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Args) IsSetResetRequest() bool {\n\treturn v != nil && v.ResetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetStickyTaskList\" for this struct.\nfunc (v *HistoryService_ResetStickyTaskList_Args) MethodName() string {\n\treturn \"ResetStickyTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ResetStickyTaskList_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ResetStickyTaskList_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ResetStickyTaskList\n// function.\nvar HistoryService_ResetStickyTaskList_Helper = struct {\n\t// Args accepts the parameters of ResetStickyTaskList in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tresetRequest *ResetStickyTaskListRequest,\n\t) *HistoryService_ResetStickyTaskList_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetStickyTaskList.\n\t//\n\t// An error can be thrown by ResetStickyTaskList only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetStickyTaskList\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ResetStickyTaskList into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ResetStickyTaskList\n\t//\n\t//   value, err := ResetStickyTaskList(args)\n\t//   result, err := HistoryService_ResetStickyTaskList_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetStickyTaskList: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*ResetStickyTaskListResponse, error) (*HistoryService_ResetStickyTaskList_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetStickyTaskList\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ResetStickyTaskList threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_ResetStickyTaskList_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ResetStickyTaskList_Result) (*ResetStickyTaskListResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_ResetStickyTaskList_Helper.Args = func(\n\t\tresetRequest *ResetStickyTaskListRequest,\n\t) *HistoryService_ResetStickyTaskList_Args {\n\t\treturn &HistoryService_ResetStickyTaskList_Args{\n\t\t\tResetRequest: resetRequest,\n\t\t}\n\t}\n\n\tHistoryService_ResetStickyTaskList_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ResetStickyTaskList_Helper.WrapResponse = func(success *ResetStickyTaskListResponse, err error) (*HistoryService_ResetStickyTaskList_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetStickyTaskList_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetStickyTaskList_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ResetStickyTaskList_Helper.UnwrapResponse = func(result *HistoryService_ResetStickyTaskList_Result) (success *ResetStickyTaskListResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ResetStickyTaskList_Result represents the result of a HistoryService.ResetStickyTaskList function call.\n//\n// The result of a ResetStickyTaskList execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_ResetStickyTaskList_Result struct {\n\t// Value returned by ResetStickyTaskList after a successful execution.\n\tSuccess                                *ResetStickyTaskListResponse                   `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetStickyTaskList_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetStickyTaskList_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetStickyTaskListResponse_Read(w wire.Value) (*ResetStickyTaskListResponse, error) {\n\tvar v ResetStickyTaskListResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ResetStickyTaskList_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetStickyTaskList_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetStickyTaskList_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetStickyTaskList_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ResetStickyTaskListResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetStickyTaskList_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetStickyTaskList_Result struct could not be encoded.\nfunc (v *HistoryService_ResetStickyTaskList_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetStickyTaskListResponse_Decode(sr stream.Reader) (*ResetStickyTaskListResponse, error) {\n\tvar v ResetStickyTaskListResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ResetStickyTaskList_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetStickyTaskList_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetStickyTaskList_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ResetStickyTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetStickyTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetStickyTaskList_Result\n// struct.\nfunc (v *HistoryService_ResetStickyTaskList_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetStickyTaskList_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetStickyTaskList_Result match the\n// provided HistoryService_ResetStickyTaskList_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetStickyTaskList_Result) Equals(rhs *HistoryService_ResetStickyTaskList_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetStickyTaskList_Result.\nfunc (v *HistoryService_ResetStickyTaskList_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetSuccess() (o *ResetStickyTaskListResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetStickyTaskList_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_ResetStickyTaskList_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetStickyTaskList\" for this struct.\nfunc (v *HistoryService_ResetStickyTaskList_Result) MethodName() string {\n\treturn \"ResetStickyTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ResetStickyTaskList_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ResetWorkflowExecution_Args represents the arguments for the HistoryService.ResetWorkflowExecution function.\n//\n// The arguments for ResetWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_ResetWorkflowExecution_Args struct {\n\tResetRequest *ResetWorkflowExecutionRequest `json:\"resetRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ResetRequest != nil {\n\t\tw, err = v.ResetRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetWorkflowExecutionRequest_1_Read(w wire.Value) (*ResetWorkflowExecutionRequest, error) {\n\tvar v ResetWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ResetWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ResetRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResetRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*ResetWorkflowExecutionRequest, error) {\n\tvar v ResetWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ResetWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ResetRequest, err = _ResetWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ResetRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResetRequest: %v\", v.ResetRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetWorkflowExecution_Args match the\n// provided HistoryService_ResetWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) Equals(rhs *HistoryService_ResetWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ResetRequest == nil && rhs.ResetRequest == nil) || (v.ResetRequest != nil && rhs.ResetRequest != nil && v.ResetRequest.Equals(rhs.ResetRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetWorkflowExecution_Args.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ResetRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resetRequest\", v.ResetRequest))\n\t}\n\treturn err\n}\n\n// GetResetRequest returns the value of ResetRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) GetResetRequest() (o *ResetWorkflowExecutionRequest) {\n\tif v != nil && v.ResetRequest != nil {\n\t\treturn v.ResetRequest\n\t}\n\n\treturn\n}\n\n// IsSetResetRequest returns true if ResetRequest is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) IsSetResetRequest() bool {\n\treturn v != nil && v.ResetRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ResetWorkflowExecution\" for this struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) MethodName() string {\n\treturn \"ResetWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ResetWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ResetWorkflowExecution\n// function.\nvar HistoryService_ResetWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of ResetWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tresetRequest *ResetWorkflowExecutionRequest,\n\t) *HistoryService_ResetWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ResetWorkflowExecution.\n\t//\n\t// An error can be thrown by ResetWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ResetWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ResetWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ResetWorkflowExecution\n\t//\n\t//   value, err := ResetWorkflowExecution(args)\n\t//   result, err := HistoryService_ResetWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ResetWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ResetWorkflowExecutionResponse, error) (*HistoryService_ResetWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for ResetWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ResetWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_ResetWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ResetWorkflowExecution_Result) (*shared.ResetWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_ResetWorkflowExecution_Helper.Args = func(\n\t\tresetRequest *ResetWorkflowExecutionRequest,\n\t) *HistoryService_ResetWorkflowExecution_Args {\n\t\treturn &HistoryService_ResetWorkflowExecution_Args{\n\t\t\tResetRequest: resetRequest,\n\t\t}\n\t}\n\n\tHistoryService_ResetWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ResetWorkflowExecution_Helper.WrapResponse = func(success *shared.ResetWorkflowExecutionResponse, err error) (*HistoryService_ResetWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ResetWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ResetWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ResetWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_ResetWorkflowExecution_Result) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ResetWorkflowExecution_Result represents the result of a HistoryService.ResetWorkflowExecution function call.\n//\n// The result of a ResetWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_ResetWorkflowExecution_Result struct {\n\t// Value returned by ResetWorkflowExecution after a successful execution.\n\tSuccess                 *shared.ResetWorkflowExecutionResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError           `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError           `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError               `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError    *shared.DomainNotActiveError           `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError             `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ResetWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ResetWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetWorkflowExecutionResponse_Read(w wire.Value) (*shared.ResetWorkflowExecutionResponse, error) {\n\tvar v shared.ResetWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ResetWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ResetWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ResetWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ResetWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ResetWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ResetWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ResetWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.ResetWorkflowExecutionResponse, error) {\n\tvar v shared.ResetWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ResetWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ResetWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ResetWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ResetWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ResetWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ResetWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ResetWorkflowExecution_Result match the\n// provided HistoryService_ResetWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) Equals(rhs *HistoryService_ResetWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ResetWorkflowExecution_Result.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetSuccess() (o *shared.ResetWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ResetWorkflowExecution\" for this struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) MethodName() string {\n\treturn \"ResetWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ResetWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondActivityTaskCanceled_Args represents the arguments for the HistoryService.RespondActivityTaskCanceled function.\n//\n// The arguments for RespondActivityTaskCanceled are sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskCanceled_Args struct {\n\tCanceledRequest *RespondActivityTaskCanceledRequest `json:\"canceledRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskCanceled_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CanceledRequest != nil {\n\t\tw, err = v.CanceledRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCanceledRequest_1_Read(w wire.Value) (*RespondActivityTaskCanceledRequest, error) {\n\tvar v RespondActivityTaskCanceledRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskCanceled_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskCanceled_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskCanceled_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskCanceled_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCanceled_Args struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CanceledRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CanceledRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCanceledRequest_1_Decode(sr stream.Reader) (*RespondActivityTaskCanceledRequest, error) {\n\tvar v RespondActivityTaskCanceledRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskCanceled_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCanceled_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CanceledRequest, err = _RespondActivityTaskCanceledRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskCanceled_Args\n// struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CanceledRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CanceledRequest: %v\", v.CanceledRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskCanceled_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskCanceled_Args match the\n// provided HistoryService_RespondActivityTaskCanceled_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) Equals(rhs *HistoryService_RespondActivityTaskCanceled_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CanceledRequest == nil && rhs.CanceledRequest == nil) || (v.CanceledRequest != nil && rhs.CanceledRequest != nil && v.CanceledRequest.Equals(rhs.CanceledRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskCanceled_Args.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CanceledRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"canceledRequest\", v.CanceledRequest))\n\t}\n\treturn err\n}\n\n// GetCanceledRequest returns the value of CanceledRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) GetCanceledRequest() (o *RespondActivityTaskCanceledRequest) {\n\tif v != nil && v.CanceledRequest != nil {\n\t\treturn v.CanceledRequest\n\t}\n\n\treturn\n}\n\n// IsSetCanceledRequest returns true if CanceledRequest is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) IsSetCanceledRequest() bool {\n\treturn v != nil && v.CanceledRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCanceled\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) MethodName() string {\n\treturn \"RespondActivityTaskCanceled\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondActivityTaskCanceled_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondActivityTaskCanceled\n// function.\nvar HistoryService_RespondActivityTaskCanceled_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCanceled in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcanceledRequest *RespondActivityTaskCanceledRequest,\n\t) *HistoryService_RespondActivityTaskCanceled_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCanceled.\n\t//\n\t// An error can be thrown by RespondActivityTaskCanceled only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCanceled\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCanceled did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCanceled into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCanceled\n\t//\n\t//   err := RespondActivityTaskCanceled(args)\n\t//   result, err := HistoryService_RespondActivityTaskCanceled_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCanceled: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RespondActivityTaskCanceled_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCanceled\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCanceled threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RespondActivityTaskCanceled_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondActivityTaskCanceled_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RespondActivityTaskCanceled_Helper.Args = func(\n\t\tcanceledRequest *RespondActivityTaskCanceledRequest,\n\t) *HistoryService_RespondActivityTaskCanceled_Args {\n\t\treturn &HistoryService_RespondActivityTaskCanceled_Args{\n\t\t\tCanceledRequest: canceledRequest,\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskCanceled_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskCanceled_Helper.WrapResponse = func(err error) (*HistoryService_RespondActivityTaskCanceled_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCanceled_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCanceled_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondActivityTaskCanceled_Helper.UnwrapResponse = func(result *HistoryService_RespondActivityTaskCanceled_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondActivityTaskCanceled_Result represents the result of a HistoryService.RespondActivityTaskCanceled function call.\n//\n// The result of a RespondActivityTaskCanceled execution is sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskCanceled_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskCanceled_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskCanceled_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskCanceled_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskCanceled_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskCanceled_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCanceled_Result struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskCanceled_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCanceled_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCanceled_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskCanceled_Result\n// struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskCanceled_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskCanceled_Result match the\n// provided HistoryService_RespondActivityTaskCanceled_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) Equals(rhs *HistoryService_RespondActivityTaskCanceled_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskCanceled_Result.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCanceled\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) MethodName() string {\n\treturn \"RespondActivityTaskCanceled\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondActivityTaskCanceled_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondActivityTaskCompleted_Args represents the arguments for the HistoryService.RespondActivityTaskCompleted function.\n//\n// The arguments for RespondActivityTaskCompleted are sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskCompleted_Args struct {\n\tCompleteRequest *RespondActivityTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskCompletedRequest_1_Read(w wire.Value) (*RespondActivityTaskCompletedRequest, error) {\n\tvar v RespondActivityTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCompleted_Args struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskCompletedRequest_1_Decode(sr stream.Reader) (*RespondActivityTaskCompletedRequest, error) {\n\tvar v RespondActivityTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondActivityTaskCompletedRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskCompleted_Args\n// struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskCompleted_Args match the\n// provided HistoryService_RespondActivityTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) Equals(rhs *HistoryService_RespondActivityTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskCompleted_Args.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) GetCompleteRequest() (o *RespondActivityTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskCompleted\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) MethodName() string {\n\treturn \"RespondActivityTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondActivityTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondActivityTaskCompleted\n// function.\nvar HistoryService_RespondActivityTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *RespondActivityTaskCompletedRequest,\n\t) *HistoryService_RespondActivityTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskCompleted.\n\t//\n\t// An error can be thrown by RespondActivityTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskCompleted\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskCompleted did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskCompleted into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskCompleted\n\t//\n\t//   err := RespondActivityTaskCompleted(args)\n\t//   result, err := HistoryService_RespondActivityTaskCompleted_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RespondActivityTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskCompleted\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RespondActivityTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondActivityTaskCompleted_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RespondActivityTaskCompleted_Helper.Args = func(\n\t\tcompleteRequest *RespondActivityTaskCompletedRequest,\n\t) *HistoryService_RespondActivityTaskCompleted_Args {\n\t\treturn &HistoryService_RespondActivityTaskCompleted_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskCompleted_Helper.WrapResponse = func(err error) (*HistoryService_RespondActivityTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskCompleted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskCompleted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondActivityTaskCompleted_Helper.UnwrapResponse = func(result *HistoryService_RespondActivityTaskCompleted_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondActivityTaskCompleted_Result represents the result of a HistoryService.RespondActivityTaskCompleted function call.\n//\n// The result of a RespondActivityTaskCompleted execution is sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskCompleted_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCompleted_Result struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskCompleted_Result\n// struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskCompleted_Result match the\n// provided HistoryService_RespondActivityTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) Equals(rhs *HistoryService_RespondActivityTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskCompleted_Result.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskCompleted\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) MethodName() string {\n\treturn \"RespondActivityTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondActivityTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondActivityTaskFailed_Args represents the arguments for the HistoryService.RespondActivityTaskFailed function.\n//\n// The arguments for RespondActivityTaskFailed are sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskFailed_Args struct {\n\tFailRequest *RespondActivityTaskFailedRequest `json:\"failRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskFailed_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailRequest != nil {\n\t\tw, err = v.FailRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondActivityTaskFailedRequest_1_Read(w wire.Value) (*RespondActivityTaskFailedRequest, error) {\n\tvar v RespondActivityTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskFailed_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskFailed_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskFailed_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailRequest, err = _RespondActivityTaskFailedRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskFailed_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskFailed_Args struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondActivityTaskFailedRequest_1_Decode(sr stream.Reader) (*RespondActivityTaskFailedRequest, error) {\n\tvar v RespondActivityTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskFailed_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskFailed_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailRequest, err = _RespondActivityTaskFailedRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskFailed_Args\n// struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailRequest: %v\", v.FailRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskFailed_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskFailed_Args match the\n// provided HistoryService_RespondActivityTaskFailed_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) Equals(rhs *HistoryService_RespondActivityTaskFailed_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailRequest == nil && rhs.FailRequest == nil) || (v.FailRequest != nil && rhs.FailRequest != nil && v.FailRequest.Equals(rhs.FailRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskFailed_Args.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failRequest\", v.FailRequest))\n\t}\n\treturn err\n}\n\n// GetFailRequest returns the value of FailRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) GetFailRequest() (o *RespondActivityTaskFailedRequest) {\n\tif v != nil && v.FailRequest != nil {\n\t\treturn v.FailRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailRequest returns true if FailRequest is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) IsSetFailRequest() bool {\n\treturn v != nil && v.FailRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondActivityTaskFailed\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) MethodName() string {\n\treturn \"RespondActivityTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondActivityTaskFailed_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondActivityTaskFailed\n// function.\nvar HistoryService_RespondActivityTaskFailed_Helper = struct {\n\t// Args accepts the parameters of RespondActivityTaskFailed in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailRequest *RespondActivityTaskFailedRequest,\n\t) *HistoryService_RespondActivityTaskFailed_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondActivityTaskFailed.\n\t//\n\t// An error can be thrown by RespondActivityTaskFailed only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondActivityTaskFailed\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondActivityTaskFailed did not fail.\n\t//\n\t// This allows mapping errors returned by RespondActivityTaskFailed into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondActivityTaskFailed\n\t//\n\t//   err := RespondActivityTaskFailed(args)\n\t//   result, err := HistoryService_RespondActivityTaskFailed_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondActivityTaskFailed: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RespondActivityTaskFailed_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondActivityTaskFailed\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondActivityTaskFailed threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RespondActivityTaskFailed_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondActivityTaskFailed_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RespondActivityTaskFailed_Helper.Args = func(\n\t\tfailRequest *RespondActivityTaskFailedRequest,\n\t) *HistoryService_RespondActivityTaskFailed_Args {\n\t\treturn &HistoryService_RespondActivityTaskFailed_Args{\n\t\t\tFailRequest: failRequest,\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskFailed_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondActivityTaskFailed_Helper.WrapResponse = func(err error) (*HistoryService_RespondActivityTaskFailed_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondActivityTaskFailed_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondActivityTaskFailed_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondActivityTaskFailed_Helper.UnwrapResponse = func(result *HistoryService_RespondActivityTaskFailed_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondActivityTaskFailed_Result represents the result of a HistoryService.RespondActivityTaskFailed function call.\n//\n// The result of a RespondActivityTaskFailed execution is sent and received over the wire as this struct.\ntype HistoryService_RespondActivityTaskFailed_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondActivityTaskFailed_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RespondActivityTaskFailed_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondActivityTaskFailed_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondActivityTaskFailed_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondActivityTaskFailed_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskFailed_Result struct could not be encoded.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RespondActivityTaskFailed_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondActivityTaskFailed_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondActivityTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondActivityTaskFailed_Result\n// struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondActivityTaskFailed_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondActivityTaskFailed_Result match the\n// provided HistoryService_RespondActivityTaskFailed_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) Equals(rhs *HistoryService_RespondActivityTaskFailed_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondActivityTaskFailed_Result.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondActivityTaskFailed\" for this struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) MethodName() string {\n\treturn \"RespondActivityTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondActivityTaskFailed_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondCrossClusterTasksCompleted_Args represents the arguments for the HistoryService.RespondCrossClusterTasksCompleted function.\n//\n// The arguments for RespondCrossClusterTasksCompleted are sent and received over the wire as this struct.\ntype HistoryService_RespondCrossClusterTasksCompleted_Args struct {\n\tRequest *shared.RespondCrossClusterTasksCompletedRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondCrossClusterTasksCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondCrossClusterTasksCompletedRequest_Read(w wire.Value) (*shared.RespondCrossClusterTasksCompletedRequest, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondCrossClusterTasksCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondCrossClusterTasksCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondCrossClusterTasksCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RespondCrossClusterTasksCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondCrossClusterTasksCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondCrossClusterTasksCompleted_Args struct could not be encoded.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondCrossClusterTasksCompletedRequest_Decode(sr stream.Reader) (*shared.RespondCrossClusterTasksCompletedRequest, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondCrossClusterTasksCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondCrossClusterTasksCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RespondCrossClusterTasksCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondCrossClusterTasksCompleted_Args\n// struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondCrossClusterTasksCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondCrossClusterTasksCompleted_Args match the\n// provided HistoryService_RespondCrossClusterTasksCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) Equals(rhs *HistoryService_RespondCrossClusterTasksCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondCrossClusterTasksCompleted_Args.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) GetRequest() (o *shared.RespondCrossClusterTasksCompletedRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondCrossClusterTasksCompleted\" for this struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) MethodName() string {\n\treturn \"RespondCrossClusterTasksCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondCrossClusterTasksCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondCrossClusterTasksCompleted\n// function.\nvar HistoryService_RespondCrossClusterTasksCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondCrossClusterTasksCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) *HistoryService_RespondCrossClusterTasksCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondCrossClusterTasksCompleted.\n\t//\n\t// An error can be thrown by RespondCrossClusterTasksCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondCrossClusterTasksCompleted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RespondCrossClusterTasksCompleted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RespondCrossClusterTasksCompleted\n\t//\n\t//   value, err := RespondCrossClusterTasksCompleted(args)\n\t//   result, err := HistoryService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondCrossClusterTasksCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.RespondCrossClusterTasksCompletedResponse, error) (*HistoryService_RespondCrossClusterTasksCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondCrossClusterTasksCompleted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RespondCrossClusterTasksCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondCrossClusterTasksCompleted_Result) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RespondCrossClusterTasksCompleted_Helper.Args = func(\n\t\trequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) *HistoryService_RespondCrossClusterTasksCompleted_Args {\n\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tHistoryService_RespondCrossClusterTasksCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondCrossClusterTasksCompleted_Helper.WrapResponse = func(success *shared.RespondCrossClusterTasksCompletedResponse, err error) (*HistoryService_RespondCrossClusterTasksCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondCrossClusterTasksCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondCrossClusterTasksCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondCrossClusterTasksCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondCrossClusterTasksCompleted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondCrossClusterTasksCompleted_Result{ShardOwnershipLostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse = func(result *HistoryService_RespondCrossClusterTasksCompleted_Result) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondCrossClusterTasksCompleted_Result represents the result of a HistoryService.RespondCrossClusterTasksCompleted function call.\n//\n// The result of a RespondCrossClusterTasksCompleted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RespondCrossClusterTasksCompleted_Result struct {\n\t// Value returned by RespondCrossClusterTasksCompleted after a successful execution.\n\tSuccess                 *shared.RespondCrossClusterTasksCompletedResponse `json:\"success,omitempty\"`\n\tBadRequestError         *shared.BadRequestError                           `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError                      `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError                          `json:\"serviceBusyError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError                          `json:\"shardOwnershipLostError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondCrossClusterTasksCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondCrossClusterTasksCompletedResponse_Read(w wire.Value) (*shared.RespondCrossClusterTasksCompletedResponse, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondCrossClusterTasksCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondCrossClusterTasksCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondCrossClusterTasksCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RespondCrossClusterTasksCompletedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondCrossClusterTasksCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondCrossClusterTasksCompleted_Result struct could not be encoded.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondCrossClusterTasksCompletedResponse_Decode(sr stream.Reader) (*shared.RespondCrossClusterTasksCompletedResponse, error) {\n\tvar v shared.RespondCrossClusterTasksCompletedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondCrossClusterTasksCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondCrossClusterTasksCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RespondCrossClusterTasksCompletedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondCrossClusterTasksCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondCrossClusterTasksCompleted_Result\n// struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondCrossClusterTasksCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondCrossClusterTasksCompleted_Result match the\n// provided HistoryService_RespondCrossClusterTasksCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) Equals(rhs *HistoryService_RespondCrossClusterTasksCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondCrossClusterTasksCompleted_Result.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) GetSuccess() (o *shared.RespondCrossClusterTasksCompletedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondCrossClusterTasksCompleted\" for this struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) MethodName() string {\n\treturn \"RespondCrossClusterTasksCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondCrossClusterTasksCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondDecisionTaskCompleted_Args represents the arguments for the HistoryService.RespondDecisionTaskCompleted function.\n//\n// The arguments for RespondDecisionTaskCompleted are sent and received over the wire as this struct.\ntype HistoryService_RespondDecisionTaskCompleted_Args struct {\n\tCompleteRequest *RespondDecisionTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondDecisionTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CompleteRequest != nil {\n\t\tw, err = v.CompleteRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskCompletedRequest_1_Read(w wire.Value) (*RespondDecisionTaskCompletedRequest, error) {\n\tvar v RespondDecisionTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondDecisionTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondDecisionTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondDecisionTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondDecisionTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskCompleted_Args struct could not be encoded.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CompleteRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskCompletedRequest_1_Decode(sr stream.Reader) (*RespondDecisionTaskCompletedRequest, error) {\n\tvar v RespondDecisionTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondDecisionTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteRequest, err = _RespondDecisionTaskCompletedRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondDecisionTaskCompleted_Args\n// struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CompleteRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteRequest: %v\", v.CompleteRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondDecisionTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondDecisionTaskCompleted_Args match the\n// provided HistoryService_RespondDecisionTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) Equals(rhs *HistoryService_RespondDecisionTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.CompleteRequest == nil && rhs.CompleteRequest == nil) || (v.CompleteRequest != nil && rhs.CompleteRequest != nil && v.CompleteRequest.Equals(rhs.CompleteRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondDecisionTaskCompleted_Args.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CompleteRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeRequest\", v.CompleteRequest))\n\t}\n\treturn err\n}\n\n// GetCompleteRequest returns the value of CompleteRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) GetCompleteRequest() (o *RespondDecisionTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompleteRequest returns true if CompleteRequest is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) IsSetCompleteRequest() bool {\n\treturn v != nil && v.CompleteRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondDecisionTaskCompleted\" for this struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) MethodName() string {\n\treturn \"RespondDecisionTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondDecisionTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondDecisionTaskCompleted\n// function.\nvar HistoryService_RespondDecisionTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondDecisionTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tcompleteRequest *RespondDecisionTaskCompletedRequest,\n\t) *HistoryService_RespondDecisionTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondDecisionTaskCompleted.\n\t//\n\t// An error can be thrown by RespondDecisionTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondDecisionTaskCompleted\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// RespondDecisionTaskCompleted into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by RespondDecisionTaskCompleted\n\t//\n\t//   value, err := RespondDecisionTaskCompleted(args)\n\t//   result, err := HistoryService_RespondDecisionTaskCompleted_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondDecisionTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*RespondDecisionTaskCompletedResponse, error) (*HistoryService_RespondDecisionTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondDecisionTaskCompleted\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if RespondDecisionTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_RespondDecisionTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondDecisionTaskCompleted_Result) (*RespondDecisionTaskCompletedResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_RespondDecisionTaskCompleted_Helper.Args = func(\n\t\tcompleteRequest *RespondDecisionTaskCompletedRequest,\n\t) *HistoryService_RespondDecisionTaskCompleted_Args {\n\t\treturn &HistoryService_RespondDecisionTaskCompleted_Args{\n\t\t\tCompleteRequest: completeRequest,\n\t\t}\n\t}\n\n\tHistoryService_RespondDecisionTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondDecisionTaskCompleted_Helper.WrapResponse = func(success *RespondDecisionTaskCompletedResponse, err error) (*HistoryService_RespondDecisionTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskCompleted_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskCompleted_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondDecisionTaskCompleted_Helper.UnwrapResponse = func(result *HistoryService_RespondDecisionTaskCompleted_Result) (success *RespondDecisionTaskCompletedResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondDecisionTaskCompleted_Result represents the result of a HistoryService.RespondDecisionTaskCompleted function call.\n//\n// The result of a RespondDecisionTaskCompleted execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_RespondDecisionTaskCompleted_Result struct {\n\t// Value returned by RespondDecisionTaskCompleted after a successful execution.\n\tSuccess                                *RespondDecisionTaskCompletedResponse          `json:\"success,omitempty\"`\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondDecisionTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskCompletedResponse_Read(w wire.Value) (*RespondDecisionTaskCompletedResponse, error) {\n\tvar v RespondDecisionTaskCompletedResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondDecisionTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondDecisionTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondDecisionTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _RespondDecisionTaskCompletedResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondDecisionTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskCompleted_Result struct could not be encoded.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskCompletedResponse_Decode(sr stream.Reader) (*RespondDecisionTaskCompletedResponse, error) {\n\tvar v RespondDecisionTaskCompletedResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondDecisionTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _RespondDecisionTaskCompletedResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskCompleted_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondDecisionTaskCompleted_Result\n// struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondDecisionTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondDecisionTaskCompleted_Result match the\n// provided HistoryService_RespondDecisionTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) Equals(rhs *HistoryService_RespondDecisionTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondDecisionTaskCompleted_Result.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetSuccess() (o *RespondDecisionTaskCompletedResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondDecisionTaskCompleted\" for this struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) MethodName() string {\n\treturn \"RespondDecisionTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondDecisionTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_RespondDecisionTaskFailed_Args represents the arguments for the HistoryService.RespondDecisionTaskFailed function.\n//\n// The arguments for RespondDecisionTaskFailed are sent and received over the wire as this struct.\ntype HistoryService_RespondDecisionTaskFailed_Args struct {\n\tFailedRequest *RespondDecisionTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondDecisionTaskFailed_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailedRequest != nil {\n\t\tw, err = v.FailedRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondDecisionTaskFailedRequest_1_Read(w wire.Value) (*RespondDecisionTaskFailedRequest, error) {\n\tvar v RespondDecisionTaskFailedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_RespondDecisionTaskFailed_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondDecisionTaskFailed_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondDecisionTaskFailed_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondDecisionTaskFailed_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskFailed_Args struct could not be encoded.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailedRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondDecisionTaskFailedRequest_1_Decode(sr stream.Reader) (*RespondDecisionTaskFailedRequest, error) {\n\tvar v RespondDecisionTaskFailedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_RespondDecisionTaskFailed_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskFailed_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.FailedRequest, err = _RespondDecisionTaskFailedRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondDecisionTaskFailed_Args\n// struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailedRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedRequest: %v\", v.FailedRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondDecisionTaskFailed_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondDecisionTaskFailed_Args match the\n// provided HistoryService_RespondDecisionTaskFailed_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) Equals(rhs *HistoryService_RespondDecisionTaskFailed_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailedRequest == nil && rhs.FailedRequest == nil) || (v.FailedRequest != nil && rhs.FailedRequest != nil && v.FailedRequest.Equals(rhs.FailedRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondDecisionTaskFailed_Args.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailedRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedRequest\", v.FailedRequest))\n\t}\n\treturn err\n}\n\n// GetFailedRequest returns the value of FailedRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) GetFailedRequest() (o *RespondDecisionTaskFailedRequest) {\n\tif v != nil && v.FailedRequest != nil {\n\t\treturn v.FailedRequest\n\t}\n\n\treturn\n}\n\n// IsSetFailedRequest returns true if FailedRequest is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) IsSetFailedRequest() bool {\n\treturn v != nil && v.FailedRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondDecisionTaskFailed\" for this struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) MethodName() string {\n\treturn \"RespondDecisionTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_RespondDecisionTaskFailed_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.RespondDecisionTaskFailed\n// function.\nvar HistoryService_RespondDecisionTaskFailed_Helper = struct {\n\t// Args accepts the parameters of RespondDecisionTaskFailed in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tfailedRequest *RespondDecisionTaskFailedRequest,\n\t) *HistoryService_RespondDecisionTaskFailed_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondDecisionTaskFailed.\n\t//\n\t// An error can be thrown by RespondDecisionTaskFailed only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondDecisionTaskFailed\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondDecisionTaskFailed did not fail.\n\t//\n\t// This allows mapping errors returned by RespondDecisionTaskFailed into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondDecisionTaskFailed\n\t//\n\t//   err := RespondDecisionTaskFailed(args)\n\t//   result, err := HistoryService_RespondDecisionTaskFailed_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondDecisionTaskFailed: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_RespondDecisionTaskFailed_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondDecisionTaskFailed\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondDecisionTaskFailed threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_RespondDecisionTaskFailed_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_RespondDecisionTaskFailed_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_RespondDecisionTaskFailed_Helper.Args = func(\n\t\tfailedRequest *RespondDecisionTaskFailedRequest,\n\t) *HistoryService_RespondDecisionTaskFailed_Args {\n\t\treturn &HistoryService_RespondDecisionTaskFailed_Args{\n\t\t\tFailedRequest: failedRequest,\n\t\t}\n\t}\n\n\tHistoryService_RespondDecisionTaskFailed_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_RespondDecisionTaskFailed_Helper.WrapResponse = func(err error) (*HistoryService_RespondDecisionTaskFailed_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_RespondDecisionTaskFailed_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_RespondDecisionTaskFailed_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_RespondDecisionTaskFailed_Helper.UnwrapResponse = func(result *HistoryService_RespondDecisionTaskFailed_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_RespondDecisionTaskFailed_Result represents the result of a HistoryService.RespondDecisionTaskFailed function call.\n//\n// The result of a RespondDecisionTaskFailed execution is sent and received over the wire as this struct.\ntype HistoryService_RespondDecisionTaskFailed_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_RespondDecisionTaskFailed_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_RespondDecisionTaskFailed_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_RespondDecisionTaskFailed_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_RespondDecisionTaskFailed_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_RespondDecisionTaskFailed_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskFailed_Result struct could not be encoded.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_RespondDecisionTaskFailed_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_RespondDecisionTaskFailed_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_RespondDecisionTaskFailed_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_RespondDecisionTaskFailed_Result\n// struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_RespondDecisionTaskFailed_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_RespondDecisionTaskFailed_Result match the\n// provided HistoryService_RespondDecisionTaskFailed_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) Equals(rhs *HistoryService_RespondDecisionTaskFailed_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_RespondDecisionTaskFailed_Result.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondDecisionTaskFailed\" for this struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) MethodName() string {\n\treturn \"RespondDecisionTaskFailed\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_RespondDecisionTaskFailed_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_ScheduleDecisionTask_Args represents the arguments for the HistoryService.ScheduleDecisionTask function.\n//\n// The arguments for ScheduleDecisionTask are sent and received over the wire as this struct.\ntype HistoryService_ScheduleDecisionTask_Args struct {\n\tScheduleRequest *ScheduleDecisionTaskRequest `json:\"scheduleRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ScheduleDecisionTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ScheduleDecisionTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduleRequest != nil {\n\t\tw, err = v.ScheduleRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ScheduleDecisionTaskRequest_Read(w wire.Value) (*ScheduleDecisionTaskRequest, error) {\n\tvar v ScheduleDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_ScheduleDecisionTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ScheduleDecisionTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ScheduleDecisionTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ScheduleDecisionTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ScheduleRequest, err = _ScheduleDecisionTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ScheduleDecisionTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ScheduleDecisionTask_Args struct could not be encoded.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduleRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ScheduleRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ScheduleDecisionTaskRequest_Decode(sr stream.Reader) (*ScheduleDecisionTaskRequest, error) {\n\tvar v ScheduleDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_ScheduleDecisionTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ScheduleDecisionTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ScheduleRequest, err = _ScheduleDecisionTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ScheduleDecisionTask_Args\n// struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ScheduleRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleRequest: %v\", v.ScheduleRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ScheduleDecisionTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ScheduleDecisionTask_Args match the\n// provided HistoryService_ScheduleDecisionTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) Equals(rhs *HistoryService_ScheduleDecisionTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ScheduleRequest == nil && rhs.ScheduleRequest == nil) || (v.ScheduleRequest != nil && rhs.ScheduleRequest != nil && v.ScheduleRequest.Equals(rhs.ScheduleRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ScheduleDecisionTask_Args.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduleRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"scheduleRequest\", v.ScheduleRequest))\n\t}\n\treturn err\n}\n\n// GetScheduleRequest returns the value of ScheduleRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) GetScheduleRequest() (o *ScheduleDecisionTaskRequest) {\n\tif v != nil && v.ScheduleRequest != nil {\n\t\treturn v.ScheduleRequest\n\t}\n\n\treturn\n}\n\n// IsSetScheduleRequest returns true if ScheduleRequest is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) IsSetScheduleRequest() bool {\n\treturn v != nil && v.ScheduleRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ScheduleDecisionTask\" for this struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) MethodName() string {\n\treturn \"ScheduleDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_ScheduleDecisionTask_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.ScheduleDecisionTask\n// function.\nvar HistoryService_ScheduleDecisionTask_Helper = struct {\n\t// Args accepts the parameters of ScheduleDecisionTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tscheduleRequest *ScheduleDecisionTaskRequest,\n\t) *HistoryService_ScheduleDecisionTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ScheduleDecisionTask.\n\t//\n\t// An error can be thrown by ScheduleDecisionTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ScheduleDecisionTask\n\t// given the error returned by it. The provided error may\n\t// be nil if ScheduleDecisionTask did not fail.\n\t//\n\t// This allows mapping errors returned by ScheduleDecisionTask into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// ScheduleDecisionTask\n\t//\n\t//   err := ScheduleDecisionTask(args)\n\t//   result, err := HistoryService_ScheduleDecisionTask_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ScheduleDecisionTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_ScheduleDecisionTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for ScheduleDecisionTask\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if ScheduleDecisionTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_ScheduleDecisionTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_ScheduleDecisionTask_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_ScheduleDecisionTask_Helper.Args = func(\n\t\tscheduleRequest *ScheduleDecisionTaskRequest,\n\t) *HistoryService_ScheduleDecisionTask_Args {\n\t\treturn &HistoryService_ScheduleDecisionTask_Args{\n\t\t\tScheduleRequest: scheduleRequest,\n\t\t}\n\t}\n\n\tHistoryService_ScheduleDecisionTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_ScheduleDecisionTask_Helper.WrapResponse = func(err error) (*HistoryService_ScheduleDecisionTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_ScheduleDecisionTask_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_ScheduleDecisionTask_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_ScheduleDecisionTask_Helper.UnwrapResponse = func(result *HistoryService_ScheduleDecisionTask_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_ScheduleDecisionTask_Result represents the result of a HistoryService.ScheduleDecisionTask function call.\n//\n// The result of a ScheduleDecisionTask execution is sent and received over the wire as this struct.\ntype HistoryService_ScheduleDecisionTask_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_ScheduleDecisionTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_ScheduleDecisionTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_ScheduleDecisionTask_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_ScheduleDecisionTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_ScheduleDecisionTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_ScheduleDecisionTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_ScheduleDecisionTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ScheduleDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_ScheduleDecisionTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_ScheduleDecisionTask_Result struct could not be encoded.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ScheduleDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_ScheduleDecisionTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_ScheduleDecisionTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_ScheduleDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_ScheduleDecisionTask_Result\n// struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_ScheduleDecisionTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_ScheduleDecisionTask_Result match the\n// provided HistoryService_ScheduleDecisionTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) Equals(rhs *HistoryService_ScheduleDecisionTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_ScheduleDecisionTask_Result.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ScheduleDecisionTask\" for this struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) MethodName() string {\n\treturn \"ScheduleDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_ScheduleDecisionTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_SignalWithStartWorkflowExecution_Args represents the arguments for the HistoryService.SignalWithStartWorkflowExecution function.\n//\n// The arguments for SignalWithStartWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_SignalWithStartWorkflowExecution_Args struct {\n\tSignalWithStartRequest *SignalWithStartWorkflowExecutionRequest `json:\"signalWithStartRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SignalWithStartWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalWithStartRequest != nil {\n\t\tw, err = v.SignalWithStartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_1_Read(w wire.Value) (*SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v SignalWithStartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_SignalWithStartWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SignalWithStartWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SignalWithStartWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SignalWithStartWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SignalWithStartWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalWithStartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalWithStartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v SignalWithStartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_SignalWithStartWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SignalWithStartWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SignalWithStartRequest, err = _SignalWithStartWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SignalWithStartWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SignalWithStartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalWithStartRequest: %v\", v.SignalWithStartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SignalWithStartWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SignalWithStartWorkflowExecution_Args match the\n// provided HistoryService_SignalWithStartWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) Equals(rhs *HistoryService_SignalWithStartWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SignalWithStartRequest == nil && rhs.SignalWithStartRequest == nil) || (v.SignalWithStartRequest != nil && rhs.SignalWithStartRequest != nil && v.SignalWithStartRequest.Equals(rhs.SignalWithStartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SignalWithStartWorkflowExecution_Args.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalWithStartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalWithStartRequest\", v.SignalWithStartRequest))\n\t}\n\treturn err\n}\n\n// GetSignalWithStartRequest returns the value of SignalWithStartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) GetSignalWithStartRequest() (o *SignalWithStartWorkflowExecutionRequest) {\n\tif v != nil && v.SignalWithStartRequest != nil {\n\t\treturn v.SignalWithStartRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalWithStartRequest returns true if SignalWithStartRequest is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) IsSetSignalWithStartRequest() bool {\n\treturn v != nil && v.SignalWithStartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SignalWithStartWorkflowExecution\" for this struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_SignalWithStartWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.SignalWithStartWorkflowExecution\n// function.\nvar HistoryService_SignalWithStartWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of SignalWithStartWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsignalWithStartRequest *SignalWithStartWorkflowExecutionRequest,\n\t) *HistoryService_SignalWithStartWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SignalWithStartWorkflowExecution.\n\t//\n\t// An error can be thrown by SignalWithStartWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SignalWithStartWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// SignalWithStartWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by SignalWithStartWorkflowExecution\n\t//\n\t//   value, err := SignalWithStartWorkflowExecution(args)\n\t//   result, err := HistoryService_SignalWithStartWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SignalWithStartWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.StartWorkflowExecutionResponse, error) (*HistoryService_SignalWithStartWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for SignalWithStartWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if SignalWithStartWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_SignalWithStartWorkflowExecution_Result) (*shared.StartWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_SignalWithStartWorkflowExecution_Helper.Args = func(\n\t\tsignalWithStartRequest *SignalWithStartWorkflowExecutionRequest,\n\t) *HistoryService_SignalWithStartWorkflowExecution_Args {\n\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Args{\n\t\t\tSignalWithStartRequest: signalWithStartRequest,\n\t\t}\n\t}\n\n\tHistoryService_SignalWithStartWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_SignalWithStartWorkflowExecution_Helper.WrapResponse = func(success *shared.StartWorkflowExecutionResponse, err error) (*HistoryService_SignalWithStartWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWithStartWorkflowExecution_Result.WorkflowAlreadyStartedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWithStartWorkflowExecution_Result{WorkflowAlreadyStartedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_SignalWithStartWorkflowExecution_Result) (success *shared.StartWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowAlreadyStartedError != nil {\n\t\t\terr = result.WorkflowAlreadyStartedError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_SignalWithStartWorkflowExecution_Result represents the result of a HistoryService.SignalWithStartWorkflowExecution function call.\n//\n// The result of a SignalWithStartWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_SignalWithStartWorkflowExecution_Result struct {\n\t// Value returned by SignalWithStartWorkflowExecution after a successful execution.\n\tSuccess                     *shared.StartWorkflowExecutionResponse       `json:\"success,omitempty\"`\n\tBadRequestError             *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError        *shared.InternalServiceError                 `json:\"internalServiceError,omitempty\"`\n\tShardOwnershipLostError     *ShardOwnershipLostError                     `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError        *shared.DomainNotActiveError                 `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError          *shared.LimitExceededError                   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError            *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n\tWorkflowAlreadyStartedError *shared.WorkflowExecutionAlreadyStartedError `json:\"workflowAlreadyStartedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SignalWithStartWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tw, err = v.WorkflowAlreadyStartedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionResponse_Read(w wire.Value) (*shared.StartWorkflowExecutionResponse, error) {\n\tvar v shared.StartWorkflowExecutionResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyStartedError_Read(w wire.Value) (*shared.WorkflowExecutionAlreadyStartedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyStartedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_SignalWithStartWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SignalWithStartWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SignalWithStartWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowAlreadyStartedError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SignalWithStartWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SignalWithStartWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowAlreadyStartedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionResponse_Decode(sr stream.Reader) (*shared.StartWorkflowExecutionResponse, error) {\n\tvar v shared.StartWorkflowExecutionResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionAlreadyStartedError_Decode(sr stream.Reader) (*shared.WorkflowExecutionAlreadyStartedError, error) {\n\tvar v shared.WorkflowExecutionAlreadyStartedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_SignalWithStartWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SignalWithStartWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowAlreadyStartedError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWithStartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SignalWithStartWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowAlreadyStartedError: %v\", v.WorkflowAlreadyStartedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SignalWithStartWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SignalWithStartWorkflowExecution_Result match the\n// provided HistoryService_SignalWithStartWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) Equals(rhs *HistoryService_SignalWithStartWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowAlreadyStartedError == nil && rhs.WorkflowAlreadyStartedError == nil) || (v.WorkflowAlreadyStartedError != nil && rhs.WorkflowAlreadyStartedError != nil && v.WorkflowAlreadyStartedError.Equals(rhs.WorkflowAlreadyStartedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SignalWithStartWorkflowExecution_Result.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowAlreadyStartedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowAlreadyStartedError\", v.WorkflowAlreadyStartedError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetSuccess() (o *shared.StartWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowAlreadyStartedError returns the value of WorkflowAlreadyStartedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) GetWorkflowAlreadyStartedError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.WorkflowAlreadyStartedError != nil {\n\t\treturn v.WorkflowAlreadyStartedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowAlreadyStartedError returns true if WorkflowAlreadyStartedError is not nil.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) IsSetWorkflowAlreadyStartedError() bool {\n\treturn v != nil && v.WorkflowAlreadyStartedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SignalWithStartWorkflowExecution\" for this struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) MethodName() string {\n\treturn \"SignalWithStartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_SignalWithStartWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_SignalWorkflowExecution_Args represents the arguments for the HistoryService.SignalWorkflowExecution function.\n//\n// The arguments for SignalWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_SignalWorkflowExecution_Args struct {\n\tSignalRequest *SignalWorkflowExecutionRequest `json:\"signalRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SignalWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SignalWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalRequest != nil {\n\t\tw, err = v.SignalRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWorkflowExecutionRequest_1_Read(w wire.Value) (*SignalWorkflowExecutionRequest, error) {\n\tvar v SignalWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_SignalWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SignalWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SignalWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SignalWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SignalWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SignalWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*SignalWorkflowExecutionRequest, error) {\n\tvar v SignalWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_SignalWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SignalWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SignalRequest, err = _SignalWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SignalWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SignalRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalRequest: %v\", v.SignalRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SignalWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SignalWorkflowExecution_Args match the\n// provided HistoryService_SignalWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) Equals(rhs *HistoryService_SignalWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SignalRequest == nil && rhs.SignalRequest == nil) || (v.SignalRequest != nil && rhs.SignalRequest != nil && v.SignalRequest.Equals(rhs.SignalRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SignalWorkflowExecution_Args.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalRequest\", v.SignalRequest))\n\t}\n\treturn err\n}\n\n// GetSignalRequest returns the value of SignalRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) GetSignalRequest() (o *SignalWorkflowExecutionRequest) {\n\tif v != nil && v.SignalRequest != nil {\n\t\treturn v.SignalRequest\n\t}\n\n\treturn\n}\n\n// IsSetSignalRequest returns true if SignalRequest is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) IsSetSignalRequest() bool {\n\treturn v != nil && v.SignalRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SignalWorkflowExecution\" for this struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) MethodName() string {\n\treturn \"SignalWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_SignalWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.SignalWorkflowExecution\n// function.\nvar HistoryService_SignalWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of SignalWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsignalRequest *SignalWorkflowExecutionRequest,\n\t) *HistoryService_SignalWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SignalWorkflowExecution.\n\t//\n\t// An error can be thrown by SignalWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SignalWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if SignalWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by SignalWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// SignalWorkflowExecution\n\t//\n\t//   err := SignalWorkflowExecution(args)\n\t//   result, err := HistoryService_SignalWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SignalWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_SignalWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for SignalWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if SignalWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_SignalWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_SignalWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_SignalWorkflowExecution_Helper.Args = func(\n\t\tsignalRequest *SignalWorkflowExecutionRequest,\n\t) *HistoryService_SignalWorkflowExecution_Args {\n\t\treturn &HistoryService_SignalWorkflowExecution_Args{\n\t\t\tSignalRequest: signalRequest,\n\t\t}\n\t}\n\n\tHistoryService_SignalWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_SignalWorkflowExecution_Helper.WrapResponse = func(err error) (*HistoryService_SignalWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SignalWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SignalWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_SignalWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_SignalWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_SignalWorkflowExecution_Result represents the result of a HistoryService.SignalWorkflowExecution function call.\n//\n// The result of a SignalWorkflowExecution execution is sent and received over the wire as this struct.\ntype HistoryService_SignalWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SignalWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SignalWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_SignalWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SignalWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SignalWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SignalWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SignalWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SignalWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_SignalWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SignalWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SignalWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SignalWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SignalWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SignalWorkflowExecution_Result match the\n// provided HistoryService_SignalWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) Equals(rhs *HistoryService_SignalWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SignalWorkflowExecution_Result.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SignalWorkflowExecution\" for this struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) MethodName() string {\n\treturn \"SignalWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_SignalWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_StartWorkflowExecution_Args represents the arguments for the HistoryService.StartWorkflowExecution function.\n//\n// The arguments for StartWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_StartWorkflowExecution_Args struct {\n\tStartRequest *StartWorkflowExecutionRequest `json:\"startRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_StartWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_StartWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StartRequest != nil {\n\t\tw, err = v.StartRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionRequest_1_Read(w wire.Value) (*StartWorkflowExecutionRequest, error) {\n\tvar v StartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_StartWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_StartWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_StartWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_StartWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_StartWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_StartWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_StartWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StartRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*StartWorkflowExecutionRequest, error) {\n\tvar v StartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_StartWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_StartWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_StartWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.StartRequest, err = _StartWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_StartWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_StartWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.StartRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartRequest: %v\", v.StartRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_StartWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_StartWorkflowExecution_Args match the\n// provided HistoryService_StartWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_StartWorkflowExecution_Args) Equals(rhs *HistoryService_StartWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.StartRequest == nil && rhs.StartRequest == nil) || (v.StartRequest != nil && rhs.StartRequest != nil && v.StartRequest.Equals(rhs.StartRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_StartWorkflowExecution_Args.\nfunc (v *HistoryService_StartWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StartRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startRequest\", v.StartRequest))\n\t}\n\treturn err\n}\n\n// GetStartRequest returns the value of StartRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Args) GetStartRequest() (o *StartWorkflowExecutionRequest) {\n\tif v != nil && v.StartRequest != nil {\n\t\treturn v.StartRequest\n\t}\n\n\treturn\n}\n\n// IsSetStartRequest returns true if StartRequest is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Args) IsSetStartRequest() bool {\n\treturn v != nil && v.StartRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"StartWorkflowExecution\" for this struct.\nfunc (v *HistoryService_StartWorkflowExecution_Args) MethodName() string {\n\treturn \"StartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_StartWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_StartWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.StartWorkflowExecution\n// function.\nvar HistoryService_StartWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of StartWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tstartRequest *StartWorkflowExecutionRequest,\n\t) *HistoryService_StartWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by StartWorkflowExecution.\n\t//\n\t// An error can be thrown by StartWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for StartWorkflowExecution\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// StartWorkflowExecution into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by StartWorkflowExecution\n\t//\n\t//   value, err := StartWorkflowExecution(args)\n\t//   result, err := HistoryService_StartWorkflowExecution_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from StartWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.StartWorkflowExecutionResponse, error) (*HistoryService_StartWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for StartWorkflowExecution\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if StartWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := HistoryService_StartWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_StartWorkflowExecution_Result) (*shared.StartWorkflowExecutionResponse, error)\n}{}\n\nfunc init() {\n\tHistoryService_StartWorkflowExecution_Helper.Args = func(\n\t\tstartRequest *StartWorkflowExecutionRequest,\n\t) *HistoryService_StartWorkflowExecution_Args {\n\t\treturn &HistoryService_StartWorkflowExecution_Args{\n\t\t\tStartRequest: startRequest,\n\t\t}\n\t}\n\n\tHistoryService_StartWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_StartWorkflowExecution_Helper.WrapResponse = func(success *shared.StartWorkflowExecutionResponse, err error) (*HistoryService_StartWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.SessionAlreadyExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{SessionAlreadyExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_StartWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_StartWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_StartWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_StartWorkflowExecution_Result) (success *shared.StartWorkflowExecutionResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.SessionAlreadyExistError != nil {\n\t\t\terr = result.SessionAlreadyExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// HistoryService_StartWorkflowExecution_Result represents the result of a HistoryService.StartWorkflowExecution function call.\n//\n// The result of a StartWorkflowExecution execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype HistoryService_StartWorkflowExecution_Result struct {\n\t// Value returned by StartWorkflowExecution after a successful execution.\n\tSuccess                  *shared.StartWorkflowExecutionResponse       `json:\"success,omitempty\"`\n\tBadRequestError          *shared.BadRequestError                      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError     *shared.InternalServiceError                 `json:\"internalServiceError,omitempty\"`\n\tSessionAlreadyExistError *shared.WorkflowExecutionAlreadyStartedError `json:\"sessionAlreadyExistError,omitempty\"`\n\tShardOwnershipLostError  *ShardOwnershipLostError                     `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError     *shared.DomainNotActiveError                 `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError       *shared.LimitExceededError                   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError         *shared.ServiceBusyError                     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_StartWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_StartWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tw, err = v.SessionAlreadyExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_StartWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_StartWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_StartWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_StartWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_StartWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_StartWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_StartWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SessionAlreadyExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SessionAlreadyExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_StartWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_StartWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_StartWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _StartWorkflowExecutionResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.SessionAlreadyExistError, err = _WorkflowExecutionAlreadyStartedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"HistoryService_StartWorkflowExecution_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_StartWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_StartWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"SessionAlreadyExistError: %v\", v.SessionAlreadyExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_StartWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_StartWorkflowExecution_Result match the\n// provided HistoryService_StartWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_StartWorkflowExecution_Result) Equals(rhs *HistoryService_StartWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.SessionAlreadyExistError == nil && rhs.SessionAlreadyExistError == nil) || (v.SessionAlreadyExistError != nil && rhs.SessionAlreadyExistError != nil && v.SessionAlreadyExistError.Equals(rhs.SessionAlreadyExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_StartWorkflowExecution_Result.\nfunc (v *HistoryService_StartWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.SessionAlreadyExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"sessionAlreadyExistError\", v.SessionAlreadyExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetSuccess() (o *shared.StartWorkflowExecutionResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetSessionAlreadyExistError returns the value of SessionAlreadyExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetSessionAlreadyExistError() (o *shared.WorkflowExecutionAlreadyStartedError) {\n\tif v != nil && v.SessionAlreadyExistError != nil {\n\t\treturn v.SessionAlreadyExistError\n\t}\n\n\treturn\n}\n\n// IsSetSessionAlreadyExistError returns true if SessionAlreadyExistError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetSessionAlreadyExistError() bool {\n\treturn v != nil && v.SessionAlreadyExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_StartWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_StartWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"StartWorkflowExecution\" for this struct.\nfunc (v *HistoryService_StartWorkflowExecution_Result) MethodName() string {\n\treturn \"StartWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_StartWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_SyncActivity_Args represents the arguments for the HistoryService.SyncActivity function.\n//\n// The arguments for SyncActivity are sent and received over the wire as this struct.\ntype HistoryService_SyncActivity_Args struct {\n\tSyncActivityRequest *SyncActivityRequest `json:\"syncActivityRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SyncActivity_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SyncActivity_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SyncActivityRequest != nil {\n\t\tw, err = v.SyncActivityRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SyncActivityRequest_Read(w wire.Value) (*SyncActivityRequest, error) {\n\tvar v SyncActivityRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_SyncActivity_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SyncActivity_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SyncActivity_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SyncActivity_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SyncActivityRequest, err = _SyncActivityRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SyncActivity_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SyncActivity_Args struct could not be encoded.\nfunc (v *HistoryService_SyncActivity_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SyncActivityRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SyncActivityRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SyncActivityRequest_Decode(sr stream.Reader) (*SyncActivityRequest, error) {\n\tvar v SyncActivityRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_SyncActivity_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SyncActivity_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SyncActivity_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SyncActivityRequest, err = _SyncActivityRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SyncActivity_Args\n// struct.\nfunc (v *HistoryService_SyncActivity_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SyncActivityRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SyncActivityRequest: %v\", v.SyncActivityRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SyncActivity_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SyncActivity_Args match the\n// provided HistoryService_SyncActivity_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SyncActivity_Args) Equals(rhs *HistoryService_SyncActivity_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SyncActivityRequest == nil && rhs.SyncActivityRequest == nil) || (v.SyncActivityRequest != nil && rhs.SyncActivityRequest != nil && v.SyncActivityRequest.Equals(rhs.SyncActivityRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SyncActivity_Args.\nfunc (v *HistoryService_SyncActivity_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SyncActivityRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"syncActivityRequest\", v.SyncActivityRequest))\n\t}\n\treturn err\n}\n\n// GetSyncActivityRequest returns the value of SyncActivityRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Args) GetSyncActivityRequest() (o *SyncActivityRequest) {\n\tif v != nil && v.SyncActivityRequest != nil {\n\t\treturn v.SyncActivityRequest\n\t}\n\n\treturn\n}\n\n// IsSetSyncActivityRequest returns true if SyncActivityRequest is not nil.\nfunc (v *HistoryService_SyncActivity_Args) IsSetSyncActivityRequest() bool {\n\treturn v != nil && v.SyncActivityRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SyncActivity\" for this struct.\nfunc (v *HistoryService_SyncActivity_Args) MethodName() string {\n\treturn \"SyncActivity\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_SyncActivity_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_SyncActivity_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.SyncActivity\n// function.\nvar HistoryService_SyncActivity_Helper = struct {\n\t// Args accepts the parameters of SyncActivity in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsyncActivityRequest *SyncActivityRequest,\n\t) *HistoryService_SyncActivity_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SyncActivity.\n\t//\n\t// An error can be thrown by SyncActivity only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SyncActivity\n\t// given the error returned by it. The provided error may\n\t// be nil if SyncActivity did not fail.\n\t//\n\t// This allows mapping errors returned by SyncActivity into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// SyncActivity\n\t//\n\t//   err := SyncActivity(args)\n\t//   result, err := HistoryService_SyncActivity_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SyncActivity: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_SyncActivity_Result, error)\n\n\t// UnwrapResponse takes the result struct for SyncActivity\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if SyncActivity threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_SyncActivity_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_SyncActivity_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_SyncActivity_Helper.Args = func(\n\t\tsyncActivityRequest *SyncActivityRequest,\n\t) *HistoryService_SyncActivity_Args {\n\t\treturn &HistoryService_SyncActivity_Args{\n\t\t\tSyncActivityRequest: syncActivityRequest,\n\t\t}\n\t}\n\n\tHistoryService_SyncActivity_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.RetryTaskV2Error:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_SyncActivity_Helper.WrapResponse = func(err error) (*HistoryService_SyncActivity_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_SyncActivity_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.RetryTaskV2Error:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncActivity_Result.RetryTaskV2Error\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncActivity_Result{RetryTaskV2Error: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_SyncActivity_Helper.UnwrapResponse = func(result *HistoryService_SyncActivity_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.RetryTaskV2Error != nil {\n\t\t\terr = result.RetryTaskV2Error\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_SyncActivity_Result represents the result of a HistoryService.SyncActivity function call.\n//\n// The result of a SyncActivity execution is sent and received over the wire as this struct.\ntype HistoryService_SyncActivity_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError     *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n\tRetryTaskV2Error        *shared.RetryTaskV2Error     `json:\"retryTaskV2Error,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SyncActivity_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SyncActivity_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\tw, err = v.RetryTaskV2Error.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_SyncActivity_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_SyncActivity_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SyncActivity_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SyncActivity_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SyncActivity_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryTaskV2Error, err = _RetryTaskV2Error_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncActivity_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SyncActivity_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SyncActivity_Result struct could not be encoded.\nfunc (v *HistoryService_SyncActivity_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryTaskV2Error != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryTaskV2Error.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncActivity_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_SyncActivity_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SyncActivity_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SyncActivity_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.RetryTaskV2Error, err = _RetryTaskV2Error_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncActivity_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SyncActivity_Result\n// struct.\nfunc (v *HistoryService_SyncActivity_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryTaskV2Error: %v\", v.RetryTaskV2Error)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SyncActivity_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SyncActivity_Result match the\n// provided HistoryService_SyncActivity_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SyncActivity_Result) Equals(rhs *HistoryService_SyncActivity_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.RetryTaskV2Error == nil && rhs.RetryTaskV2Error == nil) || (v.RetryTaskV2Error != nil && rhs.RetryTaskV2Error != nil && v.RetryTaskV2Error.Equals(rhs.RetryTaskV2Error))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SyncActivity_Result.\nfunc (v *HistoryService_SyncActivity_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.RetryTaskV2Error != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryTaskV2Error\", v.RetryTaskV2Error))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetRetryTaskV2Error returns the value of RetryTaskV2Error if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncActivity_Result) GetRetryTaskV2Error() (o *shared.RetryTaskV2Error) {\n\tif v != nil && v.RetryTaskV2Error != nil {\n\t\treturn v.RetryTaskV2Error\n\t}\n\n\treturn\n}\n\n// IsSetRetryTaskV2Error returns true if RetryTaskV2Error is not nil.\nfunc (v *HistoryService_SyncActivity_Result) IsSetRetryTaskV2Error() bool {\n\treturn v != nil && v.RetryTaskV2Error != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SyncActivity\" for this struct.\nfunc (v *HistoryService_SyncActivity_Result) MethodName() string {\n\treturn \"SyncActivity\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_SyncActivity_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_SyncShardStatus_Args represents the arguments for the HistoryService.SyncShardStatus function.\n//\n// The arguments for SyncShardStatus are sent and received over the wire as this struct.\ntype HistoryService_SyncShardStatus_Args struct {\n\tSyncShardStatusRequest *SyncShardStatusRequest `json:\"syncShardStatusRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SyncShardStatus_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SyncShardStatus_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SyncShardStatusRequest != nil {\n\t\tw, err = v.SyncShardStatusRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SyncShardStatusRequest_Read(w wire.Value) (*SyncShardStatusRequest, error) {\n\tvar v SyncShardStatusRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_SyncShardStatus_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SyncShardStatus_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SyncShardStatus_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SyncShardStatus_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SyncShardStatusRequest, err = _SyncShardStatusRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SyncShardStatus_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SyncShardStatus_Args struct could not be encoded.\nfunc (v *HistoryService_SyncShardStatus_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SyncShardStatusRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SyncShardStatusRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SyncShardStatusRequest_Decode(sr stream.Reader) (*SyncShardStatusRequest, error) {\n\tvar v SyncShardStatusRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_SyncShardStatus_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SyncShardStatus_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SyncShardStatus_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.SyncShardStatusRequest, err = _SyncShardStatusRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SyncShardStatus_Args\n// struct.\nfunc (v *HistoryService_SyncShardStatus_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SyncShardStatusRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"SyncShardStatusRequest: %v\", v.SyncShardStatusRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SyncShardStatus_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SyncShardStatus_Args match the\n// provided HistoryService_SyncShardStatus_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SyncShardStatus_Args) Equals(rhs *HistoryService_SyncShardStatus_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SyncShardStatusRequest == nil && rhs.SyncShardStatusRequest == nil) || (v.SyncShardStatusRequest != nil && rhs.SyncShardStatusRequest != nil && v.SyncShardStatusRequest.Equals(rhs.SyncShardStatusRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SyncShardStatus_Args.\nfunc (v *HistoryService_SyncShardStatus_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SyncShardStatusRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"syncShardStatusRequest\", v.SyncShardStatusRequest))\n\t}\n\treturn err\n}\n\n// GetSyncShardStatusRequest returns the value of SyncShardStatusRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Args) GetSyncShardStatusRequest() (o *SyncShardStatusRequest) {\n\tif v != nil && v.SyncShardStatusRequest != nil {\n\t\treturn v.SyncShardStatusRequest\n\t}\n\n\treturn\n}\n\n// IsSetSyncShardStatusRequest returns true if SyncShardStatusRequest is not nil.\nfunc (v *HistoryService_SyncShardStatus_Args) IsSetSyncShardStatusRequest() bool {\n\treturn v != nil && v.SyncShardStatusRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"SyncShardStatus\" for this struct.\nfunc (v *HistoryService_SyncShardStatus_Args) MethodName() string {\n\treturn \"SyncShardStatus\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_SyncShardStatus_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_SyncShardStatus_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.SyncShardStatus\n// function.\nvar HistoryService_SyncShardStatus_Helper = struct {\n\t// Args accepts the parameters of SyncShardStatus in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tsyncShardStatusRequest *SyncShardStatusRequest,\n\t) *HistoryService_SyncShardStatus_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by SyncShardStatus.\n\t//\n\t// An error can be thrown by SyncShardStatus only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for SyncShardStatus\n\t// given the error returned by it. The provided error may\n\t// be nil if SyncShardStatus did not fail.\n\t//\n\t// This allows mapping errors returned by SyncShardStatus into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// SyncShardStatus\n\t//\n\t//   err := SyncShardStatus(args)\n\t//   result, err := HistoryService_SyncShardStatus_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from SyncShardStatus: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_SyncShardStatus_Result, error)\n\n\t// UnwrapResponse takes the result struct for SyncShardStatus\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if SyncShardStatus threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_SyncShardStatus_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_SyncShardStatus_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_SyncShardStatus_Helper.Args = func(\n\t\tsyncShardStatusRequest *SyncShardStatusRequest,\n\t) *HistoryService_SyncShardStatus_Args {\n\t\treturn &HistoryService_SyncShardStatus_Args{\n\t\t\tSyncShardStatusRequest: syncShardStatusRequest,\n\t\t}\n\t}\n\n\tHistoryService_SyncShardStatus_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_SyncShardStatus_Helper.WrapResponse = func(err error) (*HistoryService_SyncShardStatus_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_SyncShardStatus_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncShardStatus_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncShardStatus_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncShardStatus_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncShardStatus_Result{InternalServiceError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncShardStatus_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncShardStatus_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncShardStatus_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncShardStatus_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_SyncShardStatus_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_SyncShardStatus_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_SyncShardStatus_Helper.UnwrapResponse = func(result *HistoryService_SyncShardStatus_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_SyncShardStatus_Result represents the result of a HistoryService.SyncShardStatus function call.\n//\n// The result of a SyncShardStatus execution is sent and received over the wire as this struct.\ntype HistoryService_SyncShardStatus_Result struct {\n\tBadRequestError         *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError    *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tShardOwnershipLostError *ShardOwnershipLostError     `json:\"shardOwnershipLostError,omitempty\"`\n\tLimitExceededError      *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError        *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_SyncShardStatus_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_SyncShardStatus_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_SyncShardStatus_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_SyncShardStatus_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_SyncShardStatus_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_SyncShardStatus_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_SyncShardStatus_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncShardStatus_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_SyncShardStatus_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_SyncShardStatus_Result struct could not be encoded.\nfunc (v *HistoryService_SyncShardStatus_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncShardStatus_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_SyncShardStatus_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_SyncShardStatus_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_SyncShardStatus_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_SyncShardStatus_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_SyncShardStatus_Result\n// struct.\nfunc (v *HistoryService_SyncShardStatus_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_SyncShardStatus_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_SyncShardStatus_Result match the\n// provided HistoryService_SyncShardStatus_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_SyncShardStatus_Result) Equals(rhs *HistoryService_SyncShardStatus_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_SyncShardStatus_Result.\nfunc (v *HistoryService_SyncShardStatus_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_SyncShardStatus_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_SyncShardStatus_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_SyncShardStatus_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_SyncShardStatus_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_SyncShardStatus_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_SyncShardStatus_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"SyncShardStatus\" for this struct.\nfunc (v *HistoryService_SyncShardStatus_Result) MethodName() string {\n\treturn \"SyncShardStatus\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_SyncShardStatus_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// HistoryService_TerminateWorkflowExecution_Args represents the arguments for the HistoryService.TerminateWorkflowExecution function.\n//\n// The arguments for TerminateWorkflowExecution are sent and received over the wire as this struct.\ntype HistoryService_TerminateWorkflowExecution_Args struct {\n\tTerminateRequest *TerminateWorkflowExecutionRequest `json:\"terminateRequest,omitempty\"`\n}\n\n// ToWire translates a HistoryService_TerminateWorkflowExecution_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TerminateRequest != nil {\n\t\tw, err = v.TerminateRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TerminateWorkflowExecutionRequest_1_Read(w wire.Value) (*TerminateWorkflowExecutionRequest, error) {\n\tvar v TerminateWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryService_TerminateWorkflowExecution_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_TerminateWorkflowExecution_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_TerminateWorkflowExecution_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_TerminateWorkflowExecution_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_TerminateWorkflowExecution_Args struct could not be encoded.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TerminateRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TerminateRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TerminateWorkflowExecutionRequest_1_Decode(sr stream.Reader) (*TerminateWorkflowExecutionRequest, error) {\n\tvar v TerminateWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryService_TerminateWorkflowExecution_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_TerminateWorkflowExecution_Args struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.TerminateRequest, err = _TerminateWorkflowExecutionRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_TerminateWorkflowExecution_Args\n// struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.TerminateRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"TerminateRequest: %v\", v.TerminateRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_TerminateWorkflowExecution_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_TerminateWorkflowExecution_Args match the\n// provided HistoryService_TerminateWorkflowExecution_Args.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) Equals(rhs *HistoryService_TerminateWorkflowExecution_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TerminateRequest == nil && rhs.TerminateRequest == nil) || (v.TerminateRequest != nil && rhs.TerminateRequest != nil && v.TerminateRequest.Equals(rhs.TerminateRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_TerminateWorkflowExecution_Args.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TerminateRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"terminateRequest\", v.TerminateRequest))\n\t}\n\treturn err\n}\n\n// GetTerminateRequest returns the value of TerminateRequest if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) GetTerminateRequest() (o *TerminateWorkflowExecutionRequest) {\n\tif v != nil && v.TerminateRequest != nil {\n\t\treturn v.TerminateRequest\n\t}\n\n\treturn\n}\n\n// IsSetTerminateRequest returns true if TerminateRequest is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) IsSetTerminateRequest() bool {\n\treturn v != nil && v.TerminateRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"TerminateWorkflowExecution\" for this struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) MethodName() string {\n\treturn \"TerminateWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// HistoryService_TerminateWorkflowExecution_Helper provides functions that aid in handling the\n// parameters and return values of the HistoryService.TerminateWorkflowExecution\n// function.\nvar HistoryService_TerminateWorkflowExecution_Helper = struct {\n\t// Args accepts the parameters of TerminateWorkflowExecution in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tterminateRequest *TerminateWorkflowExecutionRequest,\n\t) *HistoryService_TerminateWorkflowExecution_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by TerminateWorkflowExecution.\n\t//\n\t// An error can be thrown by TerminateWorkflowExecution only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for TerminateWorkflowExecution\n\t// given the error returned by it. The provided error may\n\t// be nil if TerminateWorkflowExecution did not fail.\n\t//\n\t// This allows mapping errors returned by TerminateWorkflowExecution into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// TerminateWorkflowExecution\n\t//\n\t//   err := TerminateWorkflowExecution(args)\n\t//   result, err := HistoryService_TerminateWorkflowExecution_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from TerminateWorkflowExecution: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*HistoryService_TerminateWorkflowExecution_Result, error)\n\n\t// UnwrapResponse takes the result struct for TerminateWorkflowExecution\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if TerminateWorkflowExecution threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := HistoryService_TerminateWorkflowExecution_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*HistoryService_TerminateWorkflowExecution_Result) error\n}{}\n\nfunc init() {\n\tHistoryService_TerminateWorkflowExecution_Helper.Args = func(\n\t\tterminateRequest *TerminateWorkflowExecutionRequest,\n\t) *HistoryService_TerminateWorkflowExecution_Args {\n\t\treturn &HistoryService_TerminateWorkflowExecution_Args{\n\t\t\tTerminateRequest: terminateRequest,\n\t\t}\n\t}\n\n\tHistoryService_TerminateWorkflowExecution_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *ShardOwnershipLostError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tHistoryService_TerminateWorkflowExecution_Helper.WrapResponse = func(err error) (*HistoryService_TerminateWorkflowExecution_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{EntityNotExistError: e}, nil\n\t\tcase *ShardOwnershipLostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.ShardOwnershipLostError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{ShardOwnershipLostError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.WorkflowExecutionAlreadyCompletedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for HistoryService_TerminateWorkflowExecution_Result.WorkflowExecutionAlreadyCompletedError\")\n\t\t\t}\n\t\t\treturn &HistoryService_TerminateWorkflowExecution_Result{WorkflowExecutionAlreadyCompletedError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tHistoryService_TerminateWorkflowExecution_Helper.UnwrapResponse = func(result *HistoryService_TerminateWorkflowExecution_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ShardOwnershipLostError != nil {\n\t\t\terr = result.ShardOwnershipLostError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.WorkflowExecutionAlreadyCompletedError != nil {\n\t\t\terr = result.WorkflowExecutionAlreadyCompletedError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// HistoryService_TerminateWorkflowExecution_Result represents the result of a HistoryService.TerminateWorkflowExecution function call.\n//\n// The result of a TerminateWorkflowExecution execution is sent and received over the wire as this struct.\ntype HistoryService_TerminateWorkflowExecution_Result struct {\n\tBadRequestError                        *shared.BadRequestError                        `json:\"badRequestError,omitempty\"`\n\tInternalServiceError                   *shared.InternalServiceError                   `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError                    *shared.EntityNotExistsError                   `json:\"entityNotExistError,omitempty\"`\n\tShardOwnershipLostError                *ShardOwnershipLostError                       `json:\"shardOwnershipLostError,omitempty\"`\n\tDomainNotActiveError                   *shared.DomainNotActiveError                   `json:\"domainNotActiveError,omitempty\"`\n\tLimitExceededError                     *shared.LimitExceededError                     `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError                       *shared.ServiceBusyError                       `json:\"serviceBusyError,omitempty\"`\n\tWorkflowExecutionAlreadyCompletedError *shared.WorkflowExecutionAlreadyCompletedError `json:\"workflowExecutionAlreadyCompletedError,omitempty\"`\n}\n\n// ToWire translates a HistoryService_TerminateWorkflowExecution_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tw, err = v.ShardOwnershipLostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tw, err = v.WorkflowExecutionAlreadyCompletedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"HistoryService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryService_TerminateWorkflowExecution_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryService_TerminateWorkflowExecution_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryService_TerminateWorkflowExecution_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryService_TerminateWorkflowExecution_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryService_TerminateWorkflowExecution_Result struct could not be encoded.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardOwnershipLostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShardOwnershipLostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionAlreadyCompletedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryService_TerminateWorkflowExecution_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryService_TerminateWorkflowExecution_Result struct could not be generated from the wire\n// representation.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ShardOwnershipLostError, err = _ShardOwnershipLostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionAlreadyCompletedError, err = _WorkflowExecutionAlreadyCompletedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"HistoryService_TerminateWorkflowExecution_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryService_TerminateWorkflowExecution_Result\n// struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardOwnershipLostError: %v\", v.ShardOwnershipLostError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError: %v\", v.WorkflowExecutionAlreadyCompletedError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryService_TerminateWorkflowExecution_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryService_TerminateWorkflowExecution_Result match the\n// provided HistoryService_TerminateWorkflowExecution_Result.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) Equals(rhs *HistoryService_TerminateWorkflowExecution_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ShardOwnershipLostError == nil && rhs.ShardOwnershipLostError == nil) || (v.ShardOwnershipLostError != nil && rhs.ShardOwnershipLostError != nil && v.ShardOwnershipLostError.Equals(rhs.ShardOwnershipLostError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionAlreadyCompletedError == nil && rhs.WorkflowExecutionAlreadyCompletedError == nil) || (v.WorkflowExecutionAlreadyCompletedError != nil && rhs.WorkflowExecutionAlreadyCompletedError != nil && v.WorkflowExecutionAlreadyCompletedError.Equals(rhs.WorkflowExecutionAlreadyCompletedError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryService_TerminateWorkflowExecution_Result.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ShardOwnershipLostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shardOwnershipLostError\", v.ShardOwnershipLostError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionAlreadyCompletedError\", v.WorkflowExecutionAlreadyCompletedError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetShardOwnershipLostError returns the value of ShardOwnershipLostError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetShardOwnershipLostError() (o *ShardOwnershipLostError) {\n\tif v != nil && v.ShardOwnershipLostError != nil {\n\t\treturn v.ShardOwnershipLostError\n\t}\n\n\treturn\n}\n\n// IsSetShardOwnershipLostError returns true if ShardOwnershipLostError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetShardOwnershipLostError() bool {\n\treturn v != nil && v.ShardOwnershipLostError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetWorkflowExecutionAlreadyCompletedError returns the value of WorkflowExecutionAlreadyCompletedError if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) GetWorkflowExecutionAlreadyCompletedError() (o *shared.WorkflowExecutionAlreadyCompletedError) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedError != nil {\n\t\treturn v.WorkflowExecutionAlreadyCompletedError\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedError returns true if WorkflowExecutionAlreadyCompletedError is not nil.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) IsSetWorkflowExecutionAlreadyCompletedError() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"TerminateWorkflowExecution\" for this struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) MethodName() string {\n\treturn \"TerminateWorkflowExecution\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *HistoryService_TerminateWorkflowExecution_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n"
  },
  {
    "path": ".gen/go/history/historyserviceclient/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage historyserviceclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\twire \"go.uber.org/thriftrw/wire\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\thistory \"github.com/uber/cadence/.gen/go/history\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is a client for the HistoryService service.\ntype Interface interface {\n\tCloseShard(\n\t\tctx context.Context,\n\t\tRequest *shared.CloseShardRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tDescribeHistoryHost(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeHistoryHostRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeHistoryHostResponse, error)\n\n\tDescribeMutableState(\n\t\tctx context.Context,\n\t\tRequest *history.DescribeMutableStateRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.DescribeMutableStateResponse, error)\n\n\tDescribeQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeQueueRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeQueueResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tDescribeRequest *history.DescribeWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeWorkflowExecutionResponse, error)\n\n\tGetCrossClusterTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.GetCrossClusterTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetCrossClusterTasksResponse, error)\n\n\tGetDLQReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDLQReplicationMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.GetDLQReplicationMessagesResponse, error)\n\n\tGetFailoverInfo(\n\t\tctx context.Context,\n\t\tRequest *history.GetFailoverInfoRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.GetFailoverInfoResponse, error)\n\n\tGetMutableState(\n\t\tctx context.Context,\n\t\tGetRequest *history.GetMutableStateRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.GetMutableStateResponse, error)\n\n\tGetReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetReplicationMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.GetReplicationMessagesResponse, error)\n\n\tMergeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.MergeDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.MergeDLQMessagesResponse, error)\n\n\tNotifyFailoverMarkers(\n\t\tctx context.Context,\n\t\tRequest *history.NotifyFailoverMarkersRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tPollMutableState(\n\t\tctx context.Context,\n\t\tPollRequest *history.PollMutableStateRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.PollMutableStateResponse, error)\n\n\tPurgeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.PurgeDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *history.QueryWorkflowRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.QueryWorkflowResponse, error)\n\n\tRatelimitUpdate(\n\t\tctx context.Context,\n\t\tRequest *history.RatelimitUpdateRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.RatelimitUpdateResponse, error)\n\n\tReadDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.ReadDLQMessagesRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*replicator.ReadDLQMessagesResponse, error)\n\n\tReapplyEvents(\n\t\tctx context.Context,\n\t\tReapplyEventsRequest *history.ReapplyEventsRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRecordActivityTaskHeartbeat(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *history.RecordActivityTaskHeartbeatRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRecordActivityTaskStarted(\n\t\tctx context.Context,\n\t\tAddRequest *history.RecordActivityTaskStartedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.RecordActivityTaskStartedResponse, error)\n\n\tRecordChildExecutionCompleted(\n\t\tctx context.Context,\n\t\tCompletionRequest *history.RecordChildExecutionCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRecordDecisionTaskStarted(\n\t\tctx context.Context,\n\t\tAddRequest *history.RecordDecisionTaskStartedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.RecordDecisionTaskStartedResponse, error)\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *history.RefreshWorkflowTasksRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRemoveSignalMutableState(\n\t\tctx context.Context,\n\t\tRemoveRequest *history.RemoveSignalMutableStateRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRemoveTask(\n\t\tctx context.Context,\n\t\tRequest *shared.RemoveTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tReplicateEventsV2(\n\t\tctx context.Context,\n\t\tReplicateV2Request *history.ReplicateEventsV2Request,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRequestCancelWorkflowExecution(\n\t\tctx context.Context,\n\t\tCancelRequest *history.RequestCancelWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tResetQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.ResetQueueRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tResetStickyTaskList(\n\t\tctx context.Context,\n\t\tResetRequest *history.ResetStickyTaskListRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.ResetStickyTaskListResponse, error)\n\n\tResetWorkflowExecution(\n\t\tctx context.Context,\n\t\tResetRequest *history.ResetWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ResetWorkflowExecutionResponse, error)\n\n\tRespondActivityTaskCanceled(\n\t\tctx context.Context,\n\t\tCanceledRequest *history.RespondActivityTaskCanceledRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *history.RespondActivityTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondActivityTaskFailed(\n\t\tctx context.Context,\n\t\tFailRequest *history.RespondActivityTaskFailedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tRespondCrossClusterTasksCompleted(\n\t\tctx context.Context,\n\t\tRequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n\n\tRespondDecisionTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *history.RespondDecisionTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*history.RespondDecisionTaskCompletedResponse, error)\n\n\tRespondDecisionTaskFailed(\n\t\tctx context.Context,\n\t\tFailedRequest *history.RespondDecisionTaskFailedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tScheduleDecisionTask(\n\t\tctx context.Context,\n\t\tScheduleRequest *history.ScheduleDecisionTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tSignalWithStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSignalWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalRequest *history.SignalWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tStartRequest *history.StartWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSyncActivity(\n\t\tctx context.Context,\n\t\tSyncActivityRequest *history.SyncActivityRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tSyncShardStatus(\n\t\tctx context.Context,\n\t\tSyncShardStatusRequest *history.SyncShardStatusRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tTerminateWorkflowExecution(\n\t\tctx context.Context,\n\t\tTerminateRequest *history.TerminateWorkflowExecutionRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n}\n\n// New builds a new client for the HistoryService service.\n//\n//\tclient := historyserviceclient.New(dispatcher.ClientConfig(\"historyservice\"))\nfunc New(c transport.ClientConfig, opts ...thrift.ClientOption) Interface {\n\treturn client{\n\t\tc: thrift.New(thrift.Config{\n\t\t\tService:      \"HistoryService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t\tnwc: thrift.NewNoWire(thrift.Config{\n\t\t\tService:      \"HistoryService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t}\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(c transport.ClientConfig, f reflect.StructField) Interface {\n\t\t\treturn New(c, thrift.ClientBuilderOptions(c, f)...)\n\t\t},\n\t)\n}\n\ntype client struct {\n\tc   thrift.Client\n\tnwc thrift.NoWireClient\n}\n\nfunc (c client) CloseShard(\n\tctx context.Context,\n\t_Request *shared.CloseShardRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_CloseShard_Result\n\targs := history.HistoryService_CloseShard_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_CloseShard_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeHistoryHost(\n\tctx context.Context,\n\t_Request *shared.DescribeHistoryHostRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeHistoryHostResponse, err error) {\n\n\tvar result history.HistoryService_DescribeHistoryHost_Result\n\targs := history.HistoryService_DescribeHistoryHost_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_DescribeHistoryHost_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeMutableState(\n\tctx context.Context,\n\t_Request *history.DescribeMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.DescribeMutableStateResponse, err error) {\n\n\tvar result history.HistoryService_DescribeMutableState_Result\n\targs := history.HistoryService_DescribeMutableState_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_DescribeMutableState_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeQueue(\n\tctx context.Context,\n\t_Request *shared.DescribeQueueRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeQueueResponse, err error) {\n\n\tvar result history.HistoryService_DescribeQueue_Result\n\targs := history.HistoryService_DescribeQueue_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_DescribeQueue_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_DescribeRequest *history.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\n\tvar result history.HistoryService_DescribeWorkflowExecution_Result\n\targs := history.HistoryService_DescribeWorkflowExecution_Helper.Args(_DescribeRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_DescribeWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetCrossClusterTasks(\n\tctx context.Context,\n\t_Request *shared.GetCrossClusterTasksRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetCrossClusterTasksResponse, err error) {\n\n\tvar result history.HistoryService_GetCrossClusterTasks_Result\n\targs := history.HistoryService_GetCrossClusterTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_GetCrossClusterTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetDLQReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDLQReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\n\tvar result history.HistoryService_GetDLQReplicationMessages_Result\n\targs := history.HistoryService_GetDLQReplicationMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_GetDLQReplicationMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetFailoverInfo(\n\tctx context.Context,\n\t_Request *history.GetFailoverInfoRequest,\n\topts ...yarpc.CallOption,\n) (success *history.GetFailoverInfoResponse, err error) {\n\n\tvar result history.HistoryService_GetFailoverInfo_Result\n\targs := history.HistoryService_GetFailoverInfo_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_GetFailoverInfo_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetMutableState(\n\tctx context.Context,\n\t_GetRequest *history.GetMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.GetMutableStateResponse, err error) {\n\n\tvar result history.HistoryService_GetMutableState_Result\n\targs := history.HistoryService_GetMutableState_Helper.Args(_GetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_GetMutableState_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetReplicationMessagesResponse, err error) {\n\n\tvar result history.HistoryService_GetReplicationMessages_Result\n\targs := history.HistoryService_GetReplicationMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_GetReplicationMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) MergeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.MergeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.MergeDLQMessagesResponse, err error) {\n\n\tvar result history.HistoryService_MergeDLQMessages_Result\n\targs := history.HistoryService_MergeDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_MergeDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) NotifyFailoverMarkers(\n\tctx context.Context,\n\t_Request *history.NotifyFailoverMarkersRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_NotifyFailoverMarkers_Result\n\targs := history.HistoryService_NotifyFailoverMarkers_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_NotifyFailoverMarkers_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PollMutableState(\n\tctx context.Context,\n\t_PollRequest *history.PollMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.PollMutableStateResponse, err error) {\n\n\tvar result history.HistoryService_PollMutableState_Result\n\targs := history.HistoryService_PollMutableState_Helper.Args(_PollRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_PollMutableState_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PurgeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.PurgeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_PurgeDLQMessages_Result\n\targs := history.HistoryService_PurgeDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_PurgeDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *history.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *history.QueryWorkflowResponse, err error) {\n\n\tvar result history.HistoryService_QueryWorkflow_Result\n\targs := history.HistoryService_QueryWorkflow_Helper.Args(_QueryRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_QueryWorkflow_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RatelimitUpdate(\n\tctx context.Context,\n\t_Request *history.RatelimitUpdateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RatelimitUpdateResponse, err error) {\n\n\tvar result history.HistoryService_RatelimitUpdate_Result\n\targs := history.HistoryService_RatelimitUpdate_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RatelimitUpdate_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ReadDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.ReadDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.ReadDLQMessagesResponse, err error) {\n\n\tvar result history.HistoryService_ReadDLQMessages_Result\n\targs := history.HistoryService_ReadDLQMessages_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_ReadDLQMessages_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ReapplyEvents(\n\tctx context.Context,\n\t_ReapplyEventsRequest *history.ReapplyEventsRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_ReapplyEvents_Result\n\targs := history.HistoryService_ReapplyEvents_Helper.Args(_ReapplyEventsRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_ReapplyEvents_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\t_HeartbeatRequest *history.RecordActivityTaskHeartbeatRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\tvar result history.HistoryService_RecordActivityTaskHeartbeat_Result\n\targs := history.HistoryService_RecordActivityTaskHeartbeat_Helper.Args(_HeartbeatRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RecordActivityTaskHeartbeat_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordActivityTaskStarted(\n\tctx context.Context,\n\t_AddRequest *history.RecordActivityTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RecordActivityTaskStartedResponse, err error) {\n\n\tvar result history.HistoryService_RecordActivityTaskStarted_Result\n\targs := history.HistoryService_RecordActivityTaskStarted_Helper.Args(_AddRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RecordActivityTaskStarted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordChildExecutionCompleted(\n\tctx context.Context,\n\t_CompletionRequest *history.RecordChildExecutionCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RecordChildExecutionCompleted_Result\n\targs := history.HistoryService_RecordChildExecutionCompleted_Helper.Args(_CompletionRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RecordChildExecutionCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RecordDecisionTaskStarted(\n\tctx context.Context,\n\t_AddRequest *history.RecordDecisionTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RecordDecisionTaskStartedResponse, err error) {\n\n\tvar result history.HistoryService_RecordDecisionTaskStarted_Result\n\targs := history.HistoryService_RecordDecisionTaskStarted_Helper.Args(_AddRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RecordDecisionTaskStarted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *history.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RefreshWorkflowTasks_Result\n\targs := history.HistoryService_RefreshWorkflowTasks_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RefreshWorkflowTasks_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RemoveSignalMutableState(\n\tctx context.Context,\n\t_RemoveRequest *history.RemoveSignalMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RemoveSignalMutableState_Result\n\targs := history.HistoryService_RemoveSignalMutableState_Helper.Args(_RemoveRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RemoveSignalMutableState_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RemoveTask(\n\tctx context.Context,\n\t_Request *shared.RemoveTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RemoveTask_Result\n\targs := history.HistoryService_RemoveTask_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RemoveTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ReplicateEventsV2(\n\tctx context.Context,\n\t_ReplicateV2Request *history.ReplicateEventsV2Request,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_ReplicateEventsV2_Result\n\targs := history.HistoryService_ReplicateEventsV2_Helper.Args(_ReplicateV2Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_ReplicateEventsV2_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\t_CancelRequest *history.RequestCancelWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RequestCancelWorkflowExecution_Result\n\targs := history.HistoryService_RequestCancelWorkflowExecution_Helper.Args(_CancelRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RequestCancelWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetQueue(\n\tctx context.Context,\n\t_Request *shared.ResetQueueRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_ResetQueue_Result\n\targs := history.HistoryService_ResetQueue_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_ResetQueue_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetStickyTaskList(\n\tctx context.Context,\n\t_ResetRequest *history.ResetStickyTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *history.ResetStickyTaskListResponse, err error) {\n\n\tvar result history.HistoryService_ResetStickyTaskList_Result\n\targs := history.HistoryService_ResetStickyTaskList_Helper.Args(_ResetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_ResetStickyTaskList_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ResetWorkflowExecution(\n\tctx context.Context,\n\t_ResetRequest *history.ResetWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\n\tvar result history.HistoryService_ResetWorkflowExecution_Result\n\targs := history.HistoryService_ResetWorkflowExecution_Helper.Args(_ResetRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_ResetWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCanceled(\n\tctx context.Context,\n\t_CanceledRequest *history.RespondActivityTaskCanceledRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RespondActivityTaskCanceled_Result\n\targs := history.HistoryService_RespondActivityTaskCanceled_Helper.Args(_CanceledRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RespondActivityTaskCanceled_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *history.RespondActivityTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RespondActivityTaskCompleted_Result\n\targs := history.HistoryService_RespondActivityTaskCompleted_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RespondActivityTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondActivityTaskFailed(\n\tctx context.Context,\n\t_FailRequest *history.RespondActivityTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RespondActivityTaskFailed_Result\n\targs := history.HistoryService_RespondActivityTaskFailed_Helper.Args(_FailRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RespondActivityTaskFailed_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\t_Request *shared.RespondCrossClusterTasksCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\n\tvar result history.HistoryService_RespondCrossClusterTasksCompleted_Result\n\targs := history.HistoryService_RespondCrossClusterTasksCompleted_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RespondCrossClusterTasksCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *history.RespondDecisionTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RespondDecisionTaskCompletedResponse, err error) {\n\n\tvar result history.HistoryService_RespondDecisionTaskCompleted_Result\n\targs := history.HistoryService_RespondDecisionTaskCompleted_Helper.Args(_CompleteRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_RespondDecisionTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondDecisionTaskFailed(\n\tctx context.Context,\n\t_FailedRequest *history.RespondDecisionTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_RespondDecisionTaskFailed_Result\n\targs := history.HistoryService_RespondDecisionTaskFailed_Helper.Args(_FailedRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_RespondDecisionTaskFailed_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ScheduleDecisionTask(\n\tctx context.Context,\n\t_ScheduleRequest *history.ScheduleDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_ScheduleDecisionTask_Result\n\targs := history.HistoryService_ScheduleDecisionTask_Helper.Args(_ScheduleRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_ScheduleDecisionTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\t_SignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\tvar result history.HistoryService_SignalWithStartWorkflowExecution_Result\n\targs := history.HistoryService_SignalWithStartWorkflowExecution_Helper.Args(_SignalWithStartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_SignalWithStartWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SignalWorkflowExecution(\n\tctx context.Context,\n\t_SignalRequest *history.SignalWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_SignalWorkflowExecution_Result\n\targs := history.HistoryService_SignalWorkflowExecution_Helper.Args(_SignalRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_SignalWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) StartWorkflowExecution(\n\tctx context.Context,\n\t_StartRequest *history.StartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\tvar result history.HistoryService_StartWorkflowExecution_Result\n\targs := history.HistoryService_StartWorkflowExecution_Helper.Args(_StartRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = history.HistoryService_StartWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SyncActivity(\n\tctx context.Context,\n\t_SyncActivityRequest *history.SyncActivityRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_SyncActivity_Result\n\targs := history.HistoryService_SyncActivity_Helper.Args(_SyncActivityRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_SyncActivity_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) SyncShardStatus(\n\tctx context.Context,\n\t_SyncShardStatusRequest *history.SyncShardStatusRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_SyncShardStatus_Result\n\targs := history.HistoryService_SyncShardStatus_Helper.Args(_SyncShardStatusRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_SyncShardStatus_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) TerminateWorkflowExecution(\n\tctx context.Context,\n\t_TerminateRequest *history.TerminateWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result history.HistoryService_TerminateWorkflowExecution_Result\n\targs := history.HistoryService_TerminateWorkflowExecution_Helper.Args(_TerminateRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = history.HistoryService_TerminateWorkflowExecution_Helper.UnwrapResponse(&result)\n\treturn\n}\n"
  },
  {
    "path": ".gen/go/history/historyservicefx/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage historyservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\trestriction \"go.uber.org/yarpc/api/x/restriction\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\thistoryserviceclient \"github.com/uber/cadence/.gen/go/history/historyserviceclient\"\n)\n\n// Params defines the dependencies for the HistoryService client.\ntype Params struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// Result defines the output of the HistoryService client module. It provides a\n// HistoryService client to an Fx application.\ntype Result struct {\n\tfx.Out\n\n\tClient historyserviceclient.Interface\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// Client provides a HistoryService client to an Fx application using the given name\n// for routing.\n//\n//\tfx.Provide(\n//\t\thistoryservicefx.Client(\"...\"),\n//\t\tnewHandler,\n//\t)\nfunc Client(name string, opts ...thrift.ClientOption) interface{} {\n\treturn func(p Params) Result {\n\t\tcc := p.Provider.ClientConfig(name)\n\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok && p.Restriction != nil {\n\t\t\tif err := p.Restriction.Check(thrift.Encoding, namer.TransportName()); err != nil {\n\t\t\t\tpanic(err.Error())\n\t\t\t}\n\t\t}\n\t\tclient := historyserviceclient.New(cc, opts...)\n\t\treturn Result{Client: client}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/history/historyservicefx/doc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\n// Package historyservicefx provides better integration for Fx for services\n// implementing or calling HistoryService.\n//\n// # Clients\n//\n// If you are making requests to HistoryService, use the Client function to inject a\n// HistoryService client into your container.\n//\n//\tfx.Provide(historyservicefx.Client(\"...\"))\n//\n// # Servers\n//\n// If you are implementing HistoryService, provide a historyserviceserver.Interface into\n// the container and use the Server function.\n//\n// Given,\n//\n//\tfunc NewHistoryServiceHandler() historyserviceserver.Interface\n//\n// You can do the following to have the procedures of HistoryService made available\n// to an Fx application.\n//\n//\tfx.Provide(\n//\t\tNewHistoryServiceHandler,\n//\t\thistoryservicefx.Server(),\n//\t)\npackage historyservicefx\n"
  },
  {
    "path": ".gen/go/history/historyservicefx/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage historyservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\thistoryserviceserver \"github.com/uber/cadence/.gen/go/history/historyserviceserver\"\n)\n\n// ServerParams defines the dependencies for the HistoryService server.\ntype ServerParams struct {\n\tfx.In\n\n\tHandler historyserviceserver.Interface\n}\n\n// ServerResult defines the output of HistoryService server module. It provides the\n// procedures of a HistoryService handler to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group. Dig 1.2 or newer\n// must be used for this feature to work.\ntype ServerResult struct {\n\tfx.Out\n\n\tProcedures []transport.Procedure `group:\"yarpcfx\"`\n}\n\n// Server provides procedures for HistoryService to an Fx application. It expects a\n// historyservicefx.Interface to be present in the container.\n//\n//\tfx.Provide(\n//\t\tfunc(h *MyHistoryServiceHandler) historyserviceserver.Interface {\n//\t\t\treturn h\n//\t\t},\n//\t\thistoryservicefx.Server(),\n//\t)\nfunc Server(opts ...thrift.RegisterOption) interface{} {\n\treturn func(p ServerParams) ServerResult {\n\t\tprocedures := historyserviceserver.New(p.Handler, opts...)\n\t\treturn ServerResult{Procedures: procedures}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/history/historyserviceserver/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage historyserviceserver\n\nimport (\n\tcontext \"context\"\n\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\tyarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n\thistory \"github.com/uber/cadence/.gen/go/history\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is the server-side interface for the HistoryService service.\ntype Interface interface {\n\tCloseShard(\n\t\tctx context.Context,\n\t\tRequest *shared.CloseShardRequest,\n\t) error\n\n\tDescribeHistoryHost(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeHistoryHostRequest,\n\t) (*shared.DescribeHistoryHostResponse, error)\n\n\tDescribeMutableState(\n\t\tctx context.Context,\n\t\tRequest *history.DescribeMutableStateRequest,\n\t) (*history.DescribeMutableStateResponse, error)\n\n\tDescribeQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.DescribeQueueRequest,\n\t) (*shared.DescribeQueueResponse, error)\n\n\tDescribeWorkflowExecution(\n\t\tctx context.Context,\n\t\tDescribeRequest *history.DescribeWorkflowExecutionRequest,\n\t) (*shared.DescribeWorkflowExecutionResponse, error)\n\n\tGetCrossClusterTasks(\n\t\tctx context.Context,\n\t\tRequest *shared.GetCrossClusterTasksRequest,\n\t) (*shared.GetCrossClusterTasksResponse, error)\n\n\tGetDLQReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetDLQReplicationMessagesRequest,\n\t) (*replicator.GetDLQReplicationMessagesResponse, error)\n\n\tGetFailoverInfo(\n\t\tctx context.Context,\n\t\tRequest *history.GetFailoverInfoRequest,\n\t) (*history.GetFailoverInfoResponse, error)\n\n\tGetMutableState(\n\t\tctx context.Context,\n\t\tGetRequest *history.GetMutableStateRequest,\n\t) (*history.GetMutableStateResponse, error)\n\n\tGetReplicationMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.GetReplicationMessagesRequest,\n\t) (*replicator.GetReplicationMessagesResponse, error)\n\n\tMergeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.MergeDLQMessagesRequest,\n\t) (*replicator.MergeDLQMessagesResponse, error)\n\n\tNotifyFailoverMarkers(\n\t\tctx context.Context,\n\t\tRequest *history.NotifyFailoverMarkersRequest,\n\t) error\n\n\tPollMutableState(\n\t\tctx context.Context,\n\t\tPollRequest *history.PollMutableStateRequest,\n\t) (*history.PollMutableStateResponse, error)\n\n\tPurgeDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.PurgeDLQMessagesRequest,\n\t) error\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *history.QueryWorkflowRequest,\n\t) (*history.QueryWorkflowResponse, error)\n\n\tRatelimitUpdate(\n\t\tctx context.Context,\n\t\tRequest *history.RatelimitUpdateRequest,\n\t) (*history.RatelimitUpdateResponse, error)\n\n\tReadDLQMessages(\n\t\tctx context.Context,\n\t\tRequest *replicator.ReadDLQMessagesRequest,\n\t) (*replicator.ReadDLQMessagesResponse, error)\n\n\tReapplyEvents(\n\t\tctx context.Context,\n\t\tReapplyEventsRequest *history.ReapplyEventsRequest,\n\t) error\n\n\tRecordActivityTaskHeartbeat(\n\t\tctx context.Context,\n\t\tHeartbeatRequest *history.RecordActivityTaskHeartbeatRequest,\n\t) (*shared.RecordActivityTaskHeartbeatResponse, error)\n\n\tRecordActivityTaskStarted(\n\t\tctx context.Context,\n\t\tAddRequest *history.RecordActivityTaskStartedRequest,\n\t) (*history.RecordActivityTaskStartedResponse, error)\n\n\tRecordChildExecutionCompleted(\n\t\tctx context.Context,\n\t\tCompletionRequest *history.RecordChildExecutionCompletedRequest,\n\t) error\n\n\tRecordDecisionTaskStarted(\n\t\tctx context.Context,\n\t\tAddRequest *history.RecordDecisionTaskStartedRequest,\n\t) (*history.RecordDecisionTaskStartedResponse, error)\n\n\tRefreshWorkflowTasks(\n\t\tctx context.Context,\n\t\tRequest *history.RefreshWorkflowTasksRequest,\n\t) error\n\n\tRemoveSignalMutableState(\n\t\tctx context.Context,\n\t\tRemoveRequest *history.RemoveSignalMutableStateRequest,\n\t) error\n\n\tRemoveTask(\n\t\tctx context.Context,\n\t\tRequest *shared.RemoveTaskRequest,\n\t) error\n\n\tReplicateEventsV2(\n\t\tctx context.Context,\n\t\tReplicateV2Request *history.ReplicateEventsV2Request,\n\t) error\n\n\tRequestCancelWorkflowExecution(\n\t\tctx context.Context,\n\t\tCancelRequest *history.RequestCancelWorkflowExecutionRequest,\n\t) error\n\n\tResetQueue(\n\t\tctx context.Context,\n\t\tRequest *shared.ResetQueueRequest,\n\t) error\n\n\tResetStickyTaskList(\n\t\tctx context.Context,\n\t\tResetRequest *history.ResetStickyTaskListRequest,\n\t) (*history.ResetStickyTaskListResponse, error)\n\n\tResetWorkflowExecution(\n\t\tctx context.Context,\n\t\tResetRequest *history.ResetWorkflowExecutionRequest,\n\t) (*shared.ResetWorkflowExecutionResponse, error)\n\n\tRespondActivityTaskCanceled(\n\t\tctx context.Context,\n\t\tCanceledRequest *history.RespondActivityTaskCanceledRequest,\n\t) error\n\n\tRespondActivityTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *history.RespondActivityTaskCompletedRequest,\n\t) error\n\n\tRespondActivityTaskFailed(\n\t\tctx context.Context,\n\t\tFailRequest *history.RespondActivityTaskFailedRequest,\n\t) error\n\n\tRespondCrossClusterTasksCompleted(\n\t\tctx context.Context,\n\t\tRequest *shared.RespondCrossClusterTasksCompletedRequest,\n\t) (*shared.RespondCrossClusterTasksCompletedResponse, error)\n\n\tRespondDecisionTaskCompleted(\n\t\tctx context.Context,\n\t\tCompleteRequest *history.RespondDecisionTaskCompletedRequest,\n\t) (*history.RespondDecisionTaskCompletedResponse, error)\n\n\tRespondDecisionTaskFailed(\n\t\tctx context.Context,\n\t\tFailedRequest *history.RespondDecisionTaskFailedRequest,\n\t) error\n\n\tScheduleDecisionTask(\n\t\tctx context.Context,\n\t\tScheduleRequest *history.ScheduleDecisionTaskRequest,\n\t) error\n\n\tSignalWithStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSignalWorkflowExecution(\n\t\tctx context.Context,\n\t\tSignalRequest *history.SignalWorkflowExecutionRequest,\n\t) error\n\n\tStartWorkflowExecution(\n\t\tctx context.Context,\n\t\tStartRequest *history.StartWorkflowExecutionRequest,\n\t) (*shared.StartWorkflowExecutionResponse, error)\n\n\tSyncActivity(\n\t\tctx context.Context,\n\t\tSyncActivityRequest *history.SyncActivityRequest,\n\t) error\n\n\tSyncShardStatus(\n\t\tctx context.Context,\n\t\tSyncShardStatusRequest *history.SyncShardStatusRequest,\n\t) error\n\n\tTerminateWorkflowExecution(\n\t\tctx context.Context,\n\t\tTerminateRequest *history.TerminateWorkflowExecutionRequest,\n\t) error\n}\n\n// New prepares an implementation of the HistoryService service for\n// registration.\n//\n//\thandler := HistoryServiceHandler{}\n//\tdispatcher.Register(historyserviceserver.New(handler))\nfunc New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {\n\th := handler{impl}\n\tservice := thrift.Service{\n\t\tName: \"HistoryService\",\n\t\tMethods: []thrift.Method{\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"CloseShard\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.CloseShard),\n\t\t\t\t\tNoWire: closeshard_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"CloseShard(Request *shared.CloseShardRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeHistoryHost\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeHistoryHost),\n\t\t\t\t\tNoWire: describehistoryhost_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeHistoryHost(Request *shared.DescribeHistoryHostRequest) (*shared.DescribeHistoryHostResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeMutableState\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeMutableState),\n\t\t\t\t\tNoWire: describemutablestate_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeMutableState(Request *history.DescribeMutableStateRequest) (*history.DescribeMutableStateResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeQueue\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeQueue),\n\t\t\t\t\tNoWire: describequeue_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeQueue(Request *shared.DescribeQueueRequest) (*shared.DescribeQueueResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeWorkflowExecution),\n\t\t\t\t\tNoWire: describeworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeWorkflowExecution(DescribeRequest *history.DescribeWorkflowExecutionRequest) (*shared.DescribeWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetCrossClusterTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetCrossClusterTasks),\n\t\t\t\t\tNoWire: getcrossclustertasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetCrossClusterTasks(Request *shared.GetCrossClusterTasksRequest) (*shared.GetCrossClusterTasksResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetDLQReplicationMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetDLQReplicationMessages),\n\t\t\t\t\tNoWire: getdlqreplicationmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetDLQReplicationMessages(Request *replicator.GetDLQReplicationMessagesRequest) (*replicator.GetDLQReplicationMessagesResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetFailoverInfo\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetFailoverInfo),\n\t\t\t\t\tNoWire: getfailoverinfo_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetFailoverInfo(Request *history.GetFailoverInfoRequest) (*history.GetFailoverInfoResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetMutableState\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetMutableState),\n\t\t\t\t\tNoWire: getmutablestate_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetMutableState(GetRequest *history.GetMutableStateRequest) (*history.GetMutableStateResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetReplicationMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetReplicationMessages),\n\t\t\t\t\tNoWire: getreplicationmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetReplicationMessages(Request *replicator.GetReplicationMessagesRequest) (*replicator.GetReplicationMessagesResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"MergeDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.MergeDLQMessages),\n\t\t\t\t\tNoWire: mergedlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"MergeDLQMessages(Request *replicator.MergeDLQMessagesRequest) (*replicator.MergeDLQMessagesResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"NotifyFailoverMarkers\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.NotifyFailoverMarkers),\n\t\t\t\t\tNoWire: notifyfailovermarkers_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"NotifyFailoverMarkers(Request *history.NotifyFailoverMarkersRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PollMutableState\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PollMutableState),\n\t\t\t\t\tNoWire: pollmutablestate_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PollMutableState(PollRequest *history.PollMutableStateRequest) (*history.PollMutableStateResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PurgeDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PurgeDLQMessages),\n\t\t\t\t\tNoWire: purgedlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PurgeDLQMessages(Request *replicator.PurgeDLQMessagesRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"QueryWorkflow\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.QueryWorkflow),\n\t\t\t\t\tNoWire: queryworkflow_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"QueryWorkflow(QueryRequest *history.QueryWorkflowRequest) (*history.QueryWorkflowResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RatelimitUpdate\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RatelimitUpdate),\n\t\t\t\t\tNoWire: ratelimitupdate_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RatelimitUpdate(Request *history.RatelimitUpdateRequest) (*history.RatelimitUpdateResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ReadDLQMessages\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ReadDLQMessages),\n\t\t\t\t\tNoWire: readdlqmessages_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ReadDLQMessages(Request *replicator.ReadDLQMessagesRequest) (*replicator.ReadDLQMessagesResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ReapplyEvents\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ReapplyEvents),\n\t\t\t\t\tNoWire: reapplyevents_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ReapplyEvents(ReapplyEventsRequest *history.ReapplyEventsRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordActivityTaskHeartbeat\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordActivityTaskHeartbeat),\n\t\t\t\t\tNoWire: recordactivitytaskheartbeat_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordActivityTaskHeartbeat(HeartbeatRequest *history.RecordActivityTaskHeartbeatRequest) (*shared.RecordActivityTaskHeartbeatResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordActivityTaskStarted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordActivityTaskStarted),\n\t\t\t\t\tNoWire: recordactivitytaskstarted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordActivityTaskStarted(AddRequest *history.RecordActivityTaskStartedRequest) (*history.RecordActivityTaskStartedResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordChildExecutionCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordChildExecutionCompleted),\n\t\t\t\t\tNoWire: recordchildexecutioncompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordChildExecutionCompleted(CompletionRequest *history.RecordChildExecutionCompletedRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RecordDecisionTaskStarted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RecordDecisionTaskStarted),\n\t\t\t\t\tNoWire: recorddecisiontaskstarted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RecordDecisionTaskStarted(AddRequest *history.RecordDecisionTaskStartedRequest) (*history.RecordDecisionTaskStartedResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RefreshWorkflowTasks\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RefreshWorkflowTasks),\n\t\t\t\t\tNoWire: refreshworkflowtasks_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RefreshWorkflowTasks(Request *history.RefreshWorkflowTasksRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RemoveSignalMutableState\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RemoveSignalMutableState),\n\t\t\t\t\tNoWire: removesignalmutablestate_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RemoveSignalMutableState(RemoveRequest *history.RemoveSignalMutableStateRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RemoveTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RemoveTask),\n\t\t\t\t\tNoWire: removetask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RemoveTask(Request *shared.RemoveTaskRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ReplicateEventsV2\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ReplicateEventsV2),\n\t\t\t\t\tNoWire: replicateeventsv2_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ReplicateEventsV2(ReplicateV2Request *history.ReplicateEventsV2Request)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RequestCancelWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RequestCancelWorkflowExecution),\n\t\t\t\t\tNoWire: requestcancelworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RequestCancelWorkflowExecution(CancelRequest *history.RequestCancelWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetQueue\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetQueue),\n\t\t\t\t\tNoWire: resetqueue_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetQueue(Request *shared.ResetQueueRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetStickyTaskList\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetStickyTaskList),\n\t\t\t\t\tNoWire: resetstickytasklist_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetStickyTaskList(ResetRequest *history.ResetStickyTaskListRequest) (*history.ResetStickyTaskListResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ResetWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ResetWorkflowExecution),\n\t\t\t\t\tNoWire: resetworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ResetWorkflowExecution(ResetRequest *history.ResetWorkflowExecutionRequest) (*shared.ResetWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCanceled\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCanceled),\n\t\t\t\t\tNoWire: respondactivitytaskcanceled_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCanceled(CanceledRequest *history.RespondActivityTaskCanceledRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskCompleted),\n\t\t\t\t\tNoWire: respondactivitytaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskCompleted(CompleteRequest *history.RespondActivityTaskCompletedRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondActivityTaskFailed\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondActivityTaskFailed),\n\t\t\t\t\tNoWire: respondactivitytaskfailed_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondActivityTaskFailed(FailRequest *history.RespondActivityTaskFailedRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondCrossClusterTasksCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondCrossClusterTasksCompleted),\n\t\t\t\t\tNoWire: respondcrossclustertaskscompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondCrossClusterTasksCompleted(Request *shared.RespondCrossClusterTasksCompletedRequest) (*shared.RespondCrossClusterTasksCompletedResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondDecisionTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondDecisionTaskCompleted),\n\t\t\t\t\tNoWire: responddecisiontaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondDecisionTaskCompleted(CompleteRequest *history.RespondDecisionTaskCompletedRequest) (*history.RespondDecisionTaskCompletedResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondDecisionTaskFailed\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondDecisionTaskFailed),\n\t\t\t\t\tNoWire: responddecisiontaskfailed_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondDecisionTaskFailed(FailedRequest *history.RespondDecisionTaskFailedRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ScheduleDecisionTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ScheduleDecisionTask),\n\t\t\t\t\tNoWire: scheduledecisiontask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ScheduleDecisionTask(ScheduleRequest *history.ScheduleDecisionTaskRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SignalWithStartWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SignalWithStartWorkflowExecution),\n\t\t\t\t\tNoWire: signalwithstartworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SignalWithStartWorkflowExecution(SignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest) (*shared.StartWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SignalWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SignalWorkflowExecution),\n\t\t\t\t\tNoWire: signalworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SignalWorkflowExecution(SignalRequest *history.SignalWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"StartWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.StartWorkflowExecution),\n\t\t\t\t\tNoWire: startworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"StartWorkflowExecution(StartRequest *history.StartWorkflowExecutionRequest) (*shared.StartWorkflowExecutionResponse)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SyncActivity\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SyncActivity),\n\t\t\t\t\tNoWire: syncactivity_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SyncActivity(SyncActivityRequest *history.SyncActivityRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"SyncShardStatus\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.SyncShardStatus),\n\t\t\t\t\tNoWire: syncshardstatus_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"SyncShardStatus(SyncShardStatusRequest *history.SyncShardStatusRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"TerminateWorkflowExecution\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.TerminateWorkflowExecution),\n\t\t\t\t\tNoWire: terminateworkflowexecution_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"TerminateWorkflowExecution(TerminateRequest *history.TerminateWorkflowExecutionRequest)\",\n\t\t\t\tThriftModule: history.ThriftModule,\n\t\t\t},\n\t\t},\n\t}\n\n\tprocedures := make([]transport.Procedure, 0, 43)\n\tprocedures = append(procedures, thrift.BuildProcedures(service, opts...)...)\n\treturn procedures\n}\n\ntype handler struct{ impl Interface }\n\ntype yarpcErrorNamer interface{ YARPCErrorName() string }\n\ntype yarpcErrorCoder interface{ YARPCErrorCode() *yarpcerrors.Code }\n\nfunc (h handler) CloseShard(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_CloseShard_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'CloseShard': %w\", err)\n\t}\n\n\tappErr := h.impl.CloseShard(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_CloseShard_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeHistoryHost(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_DescribeHistoryHost_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'DescribeHistoryHost': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeHistoryHost(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeHistoryHost_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeMutableState(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_DescribeMutableState_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'DescribeMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeMutableState(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeMutableState_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeQueue(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_DescribeQueue_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'DescribeQueue': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeQueue_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_DescribeWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetCrossClusterTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_GetCrossClusterTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'GetCrossClusterTasks': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetCrossClusterTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetCrossClusterTasks_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetDLQReplicationMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_GetDLQReplicationMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'GetDLQReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDLQReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetDLQReplicationMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetFailoverInfo(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_GetFailoverInfo_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'GetFailoverInfo': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetFailoverInfo(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetFailoverInfo_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetMutableState(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_GetMutableState_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'GetMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetMutableState(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetMutableState_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetReplicationMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_GetReplicationMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'GetReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetReplicationMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) MergeDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_MergeDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'MergeDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MergeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_MergeDLQMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) NotifyFailoverMarkers(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_NotifyFailoverMarkers_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'NotifyFailoverMarkers': %w\", err)\n\t}\n\n\tappErr := h.impl.NotifyFailoverMarkers(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_NotifyFailoverMarkers_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PollMutableState(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_PollMutableState_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'PollMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollMutableState(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_PollMutableState_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PurgeDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_PurgeDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'PurgeDLQMessages': %w\", err)\n\t}\n\n\tappErr := h.impl.PurgeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_PurgeDLQMessages_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) QueryWorkflow(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_QueryWorkflow_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RatelimitUpdate(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RatelimitUpdate_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RatelimitUpdate': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RatelimitUpdate(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RatelimitUpdate_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ReadDLQMessages(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ReadDLQMessages_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ReadDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ReadDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReadDLQMessages_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ReapplyEvents(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ReapplyEvents_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ReapplyEvents': %w\", err)\n\t}\n\n\tappErr := h.impl.ReapplyEvents(ctx, args.ReapplyEventsRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReapplyEvents_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordActivityTaskHeartbeat(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RecordActivityTaskHeartbeat_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RecordActivityTaskHeartbeat': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeat(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordActivityTaskHeartbeat_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordActivityTaskStarted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RecordActivityTaskStarted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RecordActivityTaskStarted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskStarted(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordActivityTaskStarted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordChildExecutionCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RecordChildExecutionCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RecordChildExecutionCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RecordChildExecutionCompleted(ctx, args.CompletionRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordChildExecutionCompleted_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RecordDecisionTaskStarted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RecordDecisionTaskStarted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RecordDecisionTaskStarted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordDecisionTaskStarted(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordDecisionTaskStarted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RefreshWorkflowTasks(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RefreshWorkflowTasks_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RemoveSignalMutableState(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RemoveSignalMutableState_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RemoveSignalMutableState': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveSignalMutableState(ctx, args.RemoveRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RemoveSignalMutableState_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RemoveTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RemoveTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RemoveTask': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveTask(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RemoveTask_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ReplicateEventsV2(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ReplicateEventsV2_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ReplicateEventsV2': %w\", err)\n\t}\n\n\tappErr := h.impl.ReplicateEventsV2(ctx, args.ReplicateV2Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReplicateEventsV2_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RequestCancelWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RequestCancelWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RequestCancelWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.RequestCancelWorkflowExecution(ctx, args.CancelRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RequestCancelWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetQueue(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ResetQueue_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ResetQueue': %w\", err)\n\t}\n\n\tappErr := h.impl.ResetQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetQueue_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetStickyTaskList(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ResetStickyTaskList_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ResetStickyTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetStickyTaskList(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetStickyTaskList_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ResetWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ResetWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ResetWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetWorkflowExecution(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCanceled(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondActivityTaskCanceled_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondActivityTaskCanceled': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceled(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskCanceled_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondActivityTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondActivityTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskCompleted_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondActivityTaskFailed(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondActivityTaskFailed_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondActivityTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailed(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskFailed_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondCrossClusterTasksCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondCrossClusterTasksCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondCrossClusterTasksCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondCrossClusterTasksCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondDecisionTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondDecisionTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondDecisionTaskCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondDecisionTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondDecisionTaskCompleted_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondDecisionTaskFailed(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_RespondDecisionTaskFailed_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'RespondDecisionTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondDecisionTaskFailed(ctx, args.FailedRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondDecisionTaskFailed_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ScheduleDecisionTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_ScheduleDecisionTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'ScheduleDecisionTask': %w\", err)\n\t}\n\n\tappErr := h.impl.ScheduleDecisionTask(ctx, args.ScheduleRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ScheduleDecisionTask_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SignalWithStartWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_SignalWithStartWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'SignalWithStartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecution(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SignalWithStartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SignalWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_SignalWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'SignalWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.SignalWorkflowExecution(ctx, args.SignalRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SignalWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) StartWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_StartWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'StartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecution(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_StartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SyncActivity(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_SyncActivity_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'SyncActivity': %w\", err)\n\t}\n\n\tappErr := h.impl.SyncActivity(ctx, args.SyncActivityRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SyncActivity_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) SyncShardStatus(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_SyncShardStatus_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'SyncShardStatus': %w\", err)\n\t}\n\n\tappErr := h.impl.SyncShardStatus(ctx, args.SyncShardStatusRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SyncShardStatus_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) TerminateWorkflowExecution(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args history.HistoryService_TerminateWorkflowExecution_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'HistoryService' procedure 'TerminateWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.TerminateWorkflowExecution(ctx, args.TerminateRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_TerminateWorkflowExecution_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\ntype closeshard_NoWireHandler struct{ impl Interface }\n\nfunc (h closeshard_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_CloseShard_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'CloseShard': %w\", err)\n\t}\n\n\tappErr := h.impl.CloseShard(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_CloseShard_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describehistoryhost_NoWireHandler struct{ impl Interface }\n\nfunc (h describehistoryhost_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_DescribeHistoryHost_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'DescribeHistoryHost': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeHistoryHost(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeHistoryHost_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describemutablestate_NoWireHandler struct{ impl Interface }\n\nfunc (h describemutablestate_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_DescribeMutableState_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'DescribeMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeMutableState(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeMutableState_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describequeue_NoWireHandler struct{ impl Interface }\n\nfunc (h describequeue_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_DescribeQueue_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'DescribeQueue': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeQueue_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describeworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h describeworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_DescribeWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'DescribeWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeWorkflowExecution(ctx, args.DescribeRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_DescribeWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getcrossclustertasks_NoWireHandler struct{ impl Interface }\n\nfunc (h getcrossclustertasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_GetCrossClusterTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'GetCrossClusterTasks': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetCrossClusterTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetCrossClusterTasks_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getdlqreplicationmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h getdlqreplicationmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_GetDLQReplicationMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'GetDLQReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetDLQReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetDLQReplicationMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getfailoverinfo_NoWireHandler struct{ impl Interface }\n\nfunc (h getfailoverinfo_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_GetFailoverInfo_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'GetFailoverInfo': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetFailoverInfo(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetFailoverInfo_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getmutablestate_NoWireHandler struct{ impl Interface }\n\nfunc (h getmutablestate_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_GetMutableState_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'GetMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetMutableState(ctx, args.GetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetMutableState_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype getreplicationmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h getreplicationmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_GetReplicationMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'GetReplicationMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetReplicationMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_GetReplicationMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype mergedlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h mergedlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_MergeDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'MergeDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.MergeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_MergeDLQMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype notifyfailovermarkers_NoWireHandler struct{ impl Interface }\n\nfunc (h notifyfailovermarkers_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_NotifyFailoverMarkers_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'NotifyFailoverMarkers': %w\", err)\n\t}\n\n\tappErr := h.impl.NotifyFailoverMarkers(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_NotifyFailoverMarkers_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype pollmutablestate_NoWireHandler struct{ impl Interface }\n\nfunc (h pollmutablestate_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_PollMutableState_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'PollMutableState': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollMutableState(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_PollMutableState_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype purgedlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h purgedlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_PurgeDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'PurgeDLQMessages': %w\", err)\n\t}\n\n\tappErr := h.impl.PurgeDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_PurgeDLQMessages_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype queryworkflow_NoWireHandler struct{ impl Interface }\n\nfunc (h queryworkflow_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_QueryWorkflow_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype ratelimitupdate_NoWireHandler struct{ impl Interface }\n\nfunc (h ratelimitupdate_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RatelimitUpdate_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RatelimitUpdate': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RatelimitUpdate(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RatelimitUpdate_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype readdlqmessages_NoWireHandler struct{ impl Interface }\n\nfunc (h readdlqmessages_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ReadDLQMessages_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ReadDLQMessages': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ReadDLQMessages(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReadDLQMessages_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype reapplyevents_NoWireHandler struct{ impl Interface }\n\nfunc (h reapplyevents_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ReapplyEvents_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ReapplyEvents': %w\", err)\n\t}\n\n\tappErr := h.impl.ReapplyEvents(ctx, args.ReapplyEventsRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReapplyEvents_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recordactivitytaskheartbeat_NoWireHandler struct{ impl Interface }\n\nfunc (h recordactivitytaskheartbeat_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RecordActivityTaskHeartbeat_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RecordActivityTaskHeartbeat': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskHeartbeat(ctx, args.HeartbeatRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordActivityTaskHeartbeat_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recordactivitytaskstarted_NoWireHandler struct{ impl Interface }\n\nfunc (h recordactivitytaskstarted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RecordActivityTaskStarted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RecordActivityTaskStarted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordActivityTaskStarted(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordActivityTaskStarted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recordchildexecutioncompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h recordchildexecutioncompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RecordChildExecutionCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RecordChildExecutionCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RecordChildExecutionCompleted(ctx, args.CompletionRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordChildExecutionCompleted_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype recorddecisiontaskstarted_NoWireHandler struct{ impl Interface }\n\nfunc (h recorddecisiontaskstarted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RecordDecisionTaskStarted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RecordDecisionTaskStarted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RecordDecisionTaskStarted(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RecordDecisionTaskStarted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype refreshworkflowtasks_NoWireHandler struct{ impl Interface }\n\nfunc (h refreshworkflowtasks_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RefreshWorkflowTasks_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RefreshWorkflowTasks': %w\", err)\n\t}\n\n\tappErr := h.impl.RefreshWorkflowTasks(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RefreshWorkflowTasks_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype removesignalmutablestate_NoWireHandler struct{ impl Interface }\n\nfunc (h removesignalmutablestate_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RemoveSignalMutableState_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RemoveSignalMutableState': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveSignalMutableState(ctx, args.RemoveRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RemoveSignalMutableState_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype removetask_NoWireHandler struct{ impl Interface }\n\nfunc (h removetask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RemoveTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RemoveTask': %w\", err)\n\t}\n\n\tappErr := h.impl.RemoveTask(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RemoveTask_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype replicateeventsv2_NoWireHandler struct{ impl Interface }\n\nfunc (h replicateeventsv2_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ReplicateEventsV2_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ReplicateEventsV2': %w\", err)\n\t}\n\n\tappErr := h.impl.ReplicateEventsV2(ctx, args.ReplicateV2Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ReplicateEventsV2_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype requestcancelworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h requestcancelworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RequestCancelWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RequestCancelWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.RequestCancelWorkflowExecution(ctx, args.CancelRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RequestCancelWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetqueue_NoWireHandler struct{ impl Interface }\n\nfunc (h resetqueue_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ResetQueue_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ResetQueue': %w\", err)\n\t}\n\n\tappErr := h.impl.ResetQueue(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetQueue_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetstickytasklist_NoWireHandler struct{ impl Interface }\n\nfunc (h resetstickytasklist_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ResetStickyTaskList_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ResetStickyTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetStickyTaskList(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetStickyTaskList_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype resetworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h resetworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ResetWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ResetWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ResetWorkflowExecution(ctx, args.ResetRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ResetWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcanceled_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcanceled_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondActivityTaskCanceled_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondActivityTaskCanceled': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCanceled(ctx, args.CanceledRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskCanceled_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondActivityTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondActivityTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskCompleted_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondactivitytaskfailed_NoWireHandler struct{ impl Interface }\n\nfunc (h respondactivitytaskfailed_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondActivityTaskFailed_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondActivityTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondActivityTaskFailed(ctx, args.FailRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondActivityTaskFailed_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondcrossclustertaskscompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondcrossclustertaskscompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondCrossClusterTasksCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondCrossClusterTasksCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondCrossClusterTasksCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondCrossClusterTasksCompleted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype responddecisiontaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h responddecisiontaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondDecisionTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondDecisionTaskCompleted': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.RespondDecisionTaskCompleted(ctx, args.CompleteRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondDecisionTaskCompleted_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype responddecisiontaskfailed_NoWireHandler struct{ impl Interface }\n\nfunc (h responddecisiontaskfailed_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_RespondDecisionTaskFailed_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'RespondDecisionTaskFailed': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondDecisionTaskFailed(ctx, args.FailedRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_RespondDecisionTaskFailed_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype scheduledecisiontask_NoWireHandler struct{ impl Interface }\n\nfunc (h scheduledecisiontask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_ScheduleDecisionTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'ScheduleDecisionTask': %w\", err)\n\t}\n\n\tappErr := h.impl.ScheduleDecisionTask(ctx, args.ScheduleRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_ScheduleDecisionTask_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype signalwithstartworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h signalwithstartworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_SignalWithStartWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'SignalWithStartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.SignalWithStartWorkflowExecution(ctx, args.SignalWithStartRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SignalWithStartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype signalworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h signalworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_SignalWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'SignalWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.SignalWorkflowExecution(ctx, args.SignalRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SignalWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype startworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h startworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_StartWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'StartWorkflowExecution': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.StartWorkflowExecution(ctx, args.StartRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_StartWorkflowExecution_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype syncactivity_NoWireHandler struct{ impl Interface }\n\nfunc (h syncactivity_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_SyncActivity_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'SyncActivity': %w\", err)\n\t}\n\n\tappErr := h.impl.SyncActivity(ctx, args.SyncActivityRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SyncActivity_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype syncshardstatus_NoWireHandler struct{ impl Interface }\n\nfunc (h syncshardstatus_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_SyncShardStatus_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'SyncShardStatus': %w\", err)\n\t}\n\n\tappErr := h.impl.SyncShardStatus(ctx, args.SyncShardStatusRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_SyncShardStatus_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype terminateworkflowexecution_NoWireHandler struct{ impl Interface }\n\nfunc (h terminateworkflowexecution_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs history.HistoryService_TerminateWorkflowExecution_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'HistoryService' procedure 'TerminateWorkflowExecution': %w\", err)\n\t}\n\n\tappErr := h.impl.TerminateWorkflowExecution(ctx, args.TerminateRequest)\n\n\thadError := appErr != nil\n\tresult, err := history.HistoryService_TerminateWorkflowExecution_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n"
  },
  {
    "path": ".gen/go/history/historyservicetest/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage historyservicetest\n\nimport (\n\tcontext \"context\"\n\n\tgomock \"github.com/golang/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\thistory \"github.com/uber/cadence/.gen/go/history\"\n\thistoryserviceclient \"github.com/uber/cadence/.gen/go/history/historyserviceclient\"\n\treplicator \"github.com/uber/cadence/.gen/go/replicator\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// MockClient implements a gomock-compatible mock client for service\n// HistoryService.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *_MockClientRecorder\n}\n\nvar _ historyserviceclient.Interface = (*MockClient)(nil)\n\ntype _MockClientRecorder struct {\n\tmock *MockClient\n}\n\n// Build a new mock client for service HistoryService.\n//\n//\tmockCtrl := gomock.NewController(t)\n//\tclient := historyservicetest.NewMockClient(mockCtrl)\n//\n// Use EXPECT() to set expectations on the mock.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &_MockClientRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows you to define an expectation on the\n// HistoryService mock client.\nfunc (m *MockClient) EXPECT() *_MockClientRecorder {\n\treturn m.recorder\n}\n\n// CloseShard responds to a CloseShard call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().CloseShard(gomock.Any(), ...).Return(...)\n//\t... := client.CloseShard(...)\nfunc (m *MockClient) CloseShard(\n\tctx context.Context,\n\t_Request *shared.CloseShardRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"CloseShard\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) CloseShard(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"CloseShard\", args...)\n}\n\n// DescribeHistoryHost responds to a DescribeHistoryHost call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeHistoryHost(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeHistoryHost(...)\nfunc (m *MockClient) DescribeHistoryHost(\n\tctx context.Context,\n\t_Request *shared.DescribeHistoryHostRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeHistoryHostResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeHistoryHostResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeHistoryHost(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeHistoryHost\", args...)\n}\n\n// DescribeMutableState responds to a DescribeMutableState call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeMutableState(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeMutableState(...)\nfunc (m *MockClient) DescribeMutableState(\n\tctx context.Context,\n\t_Request *history.DescribeMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.DescribeMutableStateResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeMutableState\", args...)\n\tsuccess, _ = ret[i].(*history.DescribeMutableStateResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeMutableState(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeMutableState\", args...)\n}\n\n// DescribeQueue responds to a DescribeQueue call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeQueue(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeQueue(...)\nfunc (m *MockClient) DescribeQueue(\n\tctx context.Context,\n\t_Request *shared.DescribeQueueRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeQueueResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeQueue\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeQueueResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeQueue(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeQueue\", args...)\n}\n\n// DescribeWorkflowExecution responds to a DescribeWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeWorkflowExecution(...)\nfunc (m *MockClient) DescribeWorkflowExecution(\n\tctx context.Context,\n\t_DescribeRequest *history.DescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _DescribeRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeWorkflowExecution(\n\tctx interface{},\n\t_DescribeRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _DescribeRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeWorkflowExecution\", args...)\n}\n\n// GetCrossClusterTasks responds to a GetCrossClusterTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetCrossClusterTasks(gomock.Any(), ...).Return(...)\n//\t... := client.GetCrossClusterTasks(...)\nfunc (m *MockClient) GetCrossClusterTasks(\n\tctx context.Context,\n\t_Request *shared.GetCrossClusterTasksRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetCrossClusterTasksResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetCrossClusterTasks\", args...)\n\tsuccess, _ = ret[i].(*shared.GetCrossClusterTasksResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetCrossClusterTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetCrossClusterTasks\", args...)\n}\n\n// GetDLQReplicationMessages responds to a GetDLQReplicationMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetDLQReplicationMessages(gomock.Any(), ...).Return(...)\n//\t... := client.GetDLQReplicationMessages(...)\nfunc (m *MockClient) GetDLQReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetDLQReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetDLQReplicationMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.GetDLQReplicationMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetDLQReplicationMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetDLQReplicationMessages\", args...)\n}\n\n// GetFailoverInfo responds to a GetFailoverInfo call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetFailoverInfo(gomock.Any(), ...).Return(...)\n//\t... := client.GetFailoverInfo(...)\nfunc (m *MockClient) GetFailoverInfo(\n\tctx context.Context,\n\t_Request *history.GetFailoverInfoRequest,\n\topts ...yarpc.CallOption,\n) (success *history.GetFailoverInfoResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetFailoverInfo\", args...)\n\tsuccess, _ = ret[i].(*history.GetFailoverInfoResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetFailoverInfo(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetFailoverInfo\", args...)\n}\n\n// GetMutableState responds to a GetMutableState call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetMutableState(gomock.Any(), ...).Return(...)\n//\t... := client.GetMutableState(...)\nfunc (m *MockClient) GetMutableState(\n\tctx context.Context,\n\t_GetRequest *history.GetMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.GetMutableStateResponse, err error) {\n\n\targs := []interface{}{ctx, _GetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetMutableState\", args...)\n\tsuccess, _ = ret[i].(*history.GetMutableStateResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetMutableState(\n\tctx interface{},\n\t_GetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _GetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetMutableState\", args...)\n}\n\n// GetReplicationMessages responds to a GetReplicationMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetReplicationMessages(gomock.Any(), ...).Return(...)\n//\t... := client.GetReplicationMessages(...)\nfunc (m *MockClient) GetReplicationMessages(\n\tctx context.Context,\n\t_Request *replicator.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.GetReplicationMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.GetReplicationMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetReplicationMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetReplicationMessages\", args...)\n}\n\n// MergeDLQMessages responds to a MergeDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().MergeDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.MergeDLQMessages(...)\nfunc (m *MockClient) MergeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.MergeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.MergeDLQMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.MergeDLQMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) MergeDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"MergeDLQMessages\", args...)\n}\n\n// NotifyFailoverMarkers responds to a NotifyFailoverMarkers call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().NotifyFailoverMarkers(gomock.Any(), ...).Return(...)\n//\t... := client.NotifyFailoverMarkers(...)\nfunc (m *MockClient) NotifyFailoverMarkers(\n\tctx context.Context,\n\t_Request *history.NotifyFailoverMarkersRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"NotifyFailoverMarkers\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) NotifyFailoverMarkers(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"NotifyFailoverMarkers\", args...)\n}\n\n// PollMutableState responds to a PollMutableState call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PollMutableState(gomock.Any(), ...).Return(...)\n//\t... := client.PollMutableState(...)\nfunc (m *MockClient) PollMutableState(\n\tctx context.Context,\n\t_PollRequest *history.PollMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.PollMutableStateResponse, err error) {\n\n\targs := []interface{}{ctx, _PollRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PollMutableState\", args...)\n\tsuccess, _ = ret[i].(*history.PollMutableStateResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PollMutableState(\n\tctx interface{},\n\t_PollRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _PollRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PollMutableState\", args...)\n}\n\n// PurgeDLQMessages responds to a PurgeDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PurgeDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.PurgeDLQMessages(...)\nfunc (m *MockClient) PurgeDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.PurgeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PurgeDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PurgeDLQMessages\", args...)\n}\n\n// QueryWorkflow responds to a QueryWorkflow call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().QueryWorkflow(gomock.Any(), ...).Return(...)\n//\t... := client.QueryWorkflow(...)\nfunc (m *MockClient) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *history.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *history.QueryWorkflowResponse, err error) {\n\n\targs := []interface{}{ctx, _QueryRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", args...)\n\tsuccess, _ = ret[i].(*history.QueryWorkflowResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) QueryWorkflow(\n\tctx interface{},\n\t_QueryRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _QueryRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"QueryWorkflow\", args...)\n}\n\n// RatelimitUpdate responds to a RatelimitUpdate call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RatelimitUpdate(gomock.Any(), ...).Return(...)\n//\t... := client.RatelimitUpdate(...)\nfunc (m *MockClient) RatelimitUpdate(\n\tctx context.Context,\n\t_Request *history.RatelimitUpdateRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RatelimitUpdateResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RatelimitUpdate\", args...)\n\tsuccess, _ = ret[i].(*history.RatelimitUpdateResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RatelimitUpdate(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RatelimitUpdate\", args...)\n}\n\n// ReadDLQMessages responds to a ReadDLQMessages call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ReadDLQMessages(gomock.Any(), ...).Return(...)\n//\t... := client.ReadDLQMessages(...)\nfunc (m *MockClient) ReadDLQMessages(\n\tctx context.Context,\n\t_Request *replicator.ReadDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (success *replicator.ReadDLQMessagesResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", args...)\n\tsuccess, _ = ret[i].(*replicator.ReadDLQMessagesResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ReadDLQMessages(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ReadDLQMessages\", args...)\n}\n\n// ReapplyEvents responds to a ReapplyEvents call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ReapplyEvents(gomock.Any(), ...).Return(...)\n//\t... := client.ReapplyEvents(...)\nfunc (m *MockClient) ReapplyEvents(\n\tctx context.Context,\n\t_ReapplyEventsRequest *history.ReapplyEventsRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _ReapplyEventsRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ReapplyEvents(\n\tctx interface{},\n\t_ReapplyEventsRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ReapplyEventsRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ReapplyEvents\", args...)\n}\n\n// RecordActivityTaskHeartbeat responds to a RecordActivityTaskHeartbeat call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), ...).Return(...)\n//\t... := client.RecordActivityTaskHeartbeat(...)\nfunc (m *MockClient) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\t_HeartbeatRequest *history.RecordActivityTaskHeartbeatRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\n\targs := []interface{}{ctx, _HeartbeatRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", args...)\n\tsuccess, _ = ret[i].(*shared.RecordActivityTaskHeartbeatResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordActivityTaskHeartbeat(\n\tctx interface{},\n\t_HeartbeatRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _HeartbeatRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordActivityTaskHeartbeat\", args...)\n}\n\n// RecordActivityTaskStarted responds to a RecordActivityTaskStarted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordActivityTaskStarted(gomock.Any(), ...).Return(...)\n//\t... := client.RecordActivityTaskStarted(...)\nfunc (m *MockClient) RecordActivityTaskStarted(\n\tctx context.Context,\n\t_AddRequest *history.RecordActivityTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RecordActivityTaskStartedResponse, err error) {\n\n\targs := []interface{}{ctx, _AddRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordActivityTaskStarted\", args...)\n\tsuccess, _ = ret[i].(*history.RecordActivityTaskStartedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordActivityTaskStarted(\n\tctx interface{},\n\t_AddRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _AddRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordActivityTaskStarted\", args...)\n}\n\n// RecordChildExecutionCompleted responds to a RecordChildExecutionCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordChildExecutionCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RecordChildExecutionCompleted(...)\nfunc (m *MockClient) RecordChildExecutionCompleted(\n\tctx context.Context,\n\t_CompletionRequest *history.RecordChildExecutionCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CompletionRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordChildExecutionCompleted\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordChildExecutionCompleted(\n\tctx interface{},\n\t_CompletionRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompletionRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordChildExecutionCompleted\", args...)\n}\n\n// RecordDecisionTaskStarted responds to a RecordDecisionTaskStarted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RecordDecisionTaskStarted(gomock.Any(), ...).Return(...)\n//\t... := client.RecordDecisionTaskStarted(...)\nfunc (m *MockClient) RecordDecisionTaskStarted(\n\tctx context.Context,\n\t_AddRequest *history.RecordDecisionTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RecordDecisionTaskStartedResponse, err error) {\n\n\targs := []interface{}{ctx, _AddRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RecordDecisionTaskStarted\", args...)\n\tsuccess, _ = ret[i].(*history.RecordDecisionTaskStartedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RecordDecisionTaskStarted(\n\tctx interface{},\n\t_AddRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _AddRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RecordDecisionTaskStarted\", args...)\n}\n\n// RefreshWorkflowTasks responds to a RefreshWorkflowTasks call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RefreshWorkflowTasks(gomock.Any(), ...).Return(...)\n//\t... := client.RefreshWorkflowTasks(...)\nfunc (m *MockClient) RefreshWorkflowTasks(\n\tctx context.Context,\n\t_Request *history.RefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RefreshWorkflowTasks(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RefreshWorkflowTasks\", args...)\n}\n\n// RemoveSignalMutableState responds to a RemoveSignalMutableState call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RemoveSignalMutableState(gomock.Any(), ...).Return(...)\n//\t... := client.RemoveSignalMutableState(...)\nfunc (m *MockClient) RemoveSignalMutableState(\n\tctx context.Context,\n\t_RemoveRequest *history.RemoveSignalMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _RemoveRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RemoveSignalMutableState\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RemoveSignalMutableState(\n\tctx interface{},\n\t_RemoveRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _RemoveRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RemoveSignalMutableState\", args...)\n}\n\n// RemoveTask responds to a RemoveTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RemoveTask(gomock.Any(), ...).Return(...)\n//\t... := client.RemoveTask(...)\nfunc (m *MockClient) RemoveTask(\n\tctx context.Context,\n\t_Request *shared.RemoveTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RemoveTask\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RemoveTask(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RemoveTask\", args...)\n}\n\n// ReplicateEventsV2 responds to a ReplicateEventsV2 call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ReplicateEventsV2(gomock.Any(), ...).Return(...)\n//\t... := client.ReplicateEventsV2(...)\nfunc (m *MockClient) ReplicateEventsV2(\n\tctx context.Context,\n\t_ReplicateV2Request *history.ReplicateEventsV2Request,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _ReplicateV2Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ReplicateEventsV2\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ReplicateEventsV2(\n\tctx interface{},\n\t_ReplicateV2Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ReplicateV2Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ReplicateEventsV2\", args...)\n}\n\n// RequestCancelWorkflowExecution responds to a RequestCancelWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.RequestCancelWorkflowExecution(...)\nfunc (m *MockClient) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\t_CancelRequest *history.RequestCancelWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CancelRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RequestCancelWorkflowExecution(\n\tctx interface{},\n\t_CancelRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CancelRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RequestCancelWorkflowExecution\", args...)\n}\n\n// ResetQueue responds to a ResetQueue call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetQueue(gomock.Any(), ...).Return(...)\n//\t... := client.ResetQueue(...)\nfunc (m *MockClient) ResetQueue(\n\tctx context.Context,\n\t_Request *shared.ResetQueueRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetQueue\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetQueue(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetQueue\", args...)\n}\n\n// ResetStickyTaskList responds to a ResetStickyTaskList call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetStickyTaskList(gomock.Any(), ...).Return(...)\n//\t... := client.ResetStickyTaskList(...)\nfunc (m *MockClient) ResetStickyTaskList(\n\tctx context.Context,\n\t_ResetRequest *history.ResetStickyTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *history.ResetStickyTaskListResponse, err error) {\n\n\targs := []interface{}{ctx, _ResetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", args...)\n\tsuccess, _ = ret[i].(*history.ResetStickyTaskListResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetStickyTaskList(\n\tctx interface{},\n\t_ResetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ResetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetStickyTaskList\", args...)\n}\n\n// ResetWorkflowExecution responds to a ResetWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ResetWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.ResetWorkflowExecution(...)\nfunc (m *MockClient) ResetWorkflowExecution(\n\tctx context.Context,\n\t_ResetRequest *history.ResetWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ResetWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _ResetRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.ResetWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ResetWorkflowExecution(\n\tctx interface{},\n\t_ResetRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ResetRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ResetWorkflowExecution\", args...)\n}\n\n// RespondActivityTaskCanceled responds to a RespondActivityTaskCanceled call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCanceled(...)\nfunc (m *MockClient) RespondActivityTaskCanceled(\n\tctx context.Context,\n\t_CanceledRequest *history.RespondActivityTaskCanceledRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CanceledRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCanceled(\n\tctx interface{},\n\t_CanceledRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CanceledRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCanceled\", args...)\n}\n\n// RespondActivityTaskCompleted responds to a RespondActivityTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskCompleted(...)\nfunc (m *MockClient) RespondActivityTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *history.RespondActivityTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskCompleted(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskCompleted\", args...)\n}\n\n// RespondActivityTaskFailed responds to a RespondActivityTaskFailed call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondActivityTaskFailed(gomock.Any(), ...).Return(...)\n//\t... := client.RespondActivityTaskFailed(...)\nfunc (m *MockClient) RespondActivityTaskFailed(\n\tctx context.Context,\n\t_FailRequest *history.RespondActivityTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _FailRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondActivityTaskFailed(\n\tctx interface{},\n\t_FailRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondActivityTaskFailed\", args...)\n}\n\n// RespondCrossClusterTasksCompleted responds to a RespondCrossClusterTasksCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondCrossClusterTasksCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondCrossClusterTasksCompleted(...)\nfunc (m *MockClient) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\t_Request *shared.RespondCrossClusterTasksCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondCrossClusterTasksCompleted\", args...)\n\tsuccess, _ = ret[i].(*shared.RespondCrossClusterTasksCompletedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondCrossClusterTasksCompleted(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondCrossClusterTasksCompleted\", args...)\n}\n\n// RespondDecisionTaskCompleted responds to a RespondDecisionTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondDecisionTaskCompleted(...)\nfunc (m *MockClient) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\t_CompleteRequest *history.RespondDecisionTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (success *history.RespondDecisionTaskCompletedResponse, err error) {\n\n\targs := []interface{}{ctx, _CompleteRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", args...)\n\tsuccess, _ = ret[i].(*history.RespondDecisionTaskCompletedResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondDecisionTaskCompleted(\n\tctx interface{},\n\t_CompleteRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _CompleteRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondDecisionTaskCompleted\", args...)\n}\n\n// RespondDecisionTaskFailed responds to a RespondDecisionTaskFailed call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondDecisionTaskFailed(gomock.Any(), ...).Return(...)\n//\t... := client.RespondDecisionTaskFailed(...)\nfunc (m *MockClient) RespondDecisionTaskFailed(\n\tctx context.Context,\n\t_FailedRequest *history.RespondDecisionTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _FailedRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondDecisionTaskFailed(\n\tctx interface{},\n\t_FailedRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _FailedRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondDecisionTaskFailed\", args...)\n}\n\n// ScheduleDecisionTask responds to a ScheduleDecisionTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ScheduleDecisionTask(gomock.Any(), ...).Return(...)\n//\t... := client.ScheduleDecisionTask(...)\nfunc (m *MockClient) ScheduleDecisionTask(\n\tctx context.Context,\n\t_ScheduleRequest *history.ScheduleDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _ScheduleRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ScheduleDecisionTask\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ScheduleDecisionTask(\n\tctx interface{},\n\t_ScheduleRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _ScheduleRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ScheduleDecisionTask\", args...)\n}\n\n// SignalWithStartWorkflowExecution responds to a SignalWithStartWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.SignalWithStartWorkflowExecution(...)\nfunc (m *MockClient) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\t_SignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _SignalWithStartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.StartWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SignalWithStartWorkflowExecution(\n\tctx interface{},\n\t_SignalWithStartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SignalWithStartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SignalWithStartWorkflowExecution\", args...)\n}\n\n// SignalWorkflowExecution responds to a SignalWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SignalWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.SignalWorkflowExecution(...)\nfunc (m *MockClient) SignalWorkflowExecution(\n\tctx context.Context,\n\t_SignalRequest *history.SignalWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _SignalRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SignalWorkflowExecution(\n\tctx interface{},\n\t_SignalRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SignalRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SignalWorkflowExecution\", args...)\n}\n\n// StartWorkflowExecution responds to a StartWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().StartWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.StartWorkflowExecution(...)\nfunc (m *MockClient) StartWorkflowExecution(\n\tctx context.Context,\n\t_StartRequest *history.StartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.StartWorkflowExecutionResponse, err error) {\n\n\targs := []interface{}{ctx, _StartRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", args...)\n\tsuccess, _ = ret[i].(*shared.StartWorkflowExecutionResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) StartWorkflowExecution(\n\tctx interface{},\n\t_StartRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _StartRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"StartWorkflowExecution\", args...)\n}\n\n// SyncActivity responds to a SyncActivity call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SyncActivity(gomock.Any(), ...).Return(...)\n//\t... := client.SyncActivity(...)\nfunc (m *MockClient) SyncActivity(\n\tctx context.Context,\n\t_SyncActivityRequest *history.SyncActivityRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _SyncActivityRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SyncActivity\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SyncActivity(\n\tctx interface{},\n\t_SyncActivityRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SyncActivityRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SyncActivity\", args...)\n}\n\n// SyncShardStatus responds to a SyncShardStatus call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().SyncShardStatus(gomock.Any(), ...).Return(...)\n//\t... := client.SyncShardStatus(...)\nfunc (m *MockClient) SyncShardStatus(\n\tctx context.Context,\n\t_SyncShardStatusRequest *history.SyncShardStatusRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _SyncShardStatusRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"SyncShardStatus\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) SyncShardStatus(\n\tctx interface{},\n\t_SyncShardStatusRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _SyncShardStatusRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"SyncShardStatus\", args...)\n}\n\n// TerminateWorkflowExecution responds to a TerminateWorkflowExecution call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().TerminateWorkflowExecution(gomock.Any(), ...).Return(...)\n//\t... := client.TerminateWorkflowExecution(...)\nfunc (m *MockClient) TerminateWorkflowExecution(\n\tctx context.Context,\n\t_TerminateRequest *history.TerminateWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _TerminateRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) TerminateWorkflowExecution(\n\tctx interface{},\n\t_TerminateRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _TerminateRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"TerminateWorkflowExecution\", args...)\n}\n"
  },
  {
    "path": ".gen/go/history/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage history\n\nimport yarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n// YARPCErrorCode returns nil for EventAlreadyStartedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *EventAlreadyStartedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for EventAlreadyStartedError.\nfunc (e *EventAlreadyStartedError) YARPCErrorName() string { return \"EventAlreadyStartedError\" }\n\n// YARPCErrorCode returns nil for ShardOwnershipLostError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *ShardOwnershipLostError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for ShardOwnershipLostError.\nfunc (e *ShardOwnershipLostError) YARPCErrorName() string { return \"ShardOwnershipLostError\" }\n"
  },
  {
    "path": ".gen/go/indexer/indexer.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage indexer\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype Field struct {\n\tType       *FieldType `json:\"type,omitempty\"`\n\tStringData *string    `json:\"stringData,omitempty\"`\n\tIntData    *int64     `json:\"intData,omitempty\"`\n\tBoolData   *bool      `json:\"boolData,omitempty\"`\n\tBinaryData []byte     `json:\"binaryData,omitempty\"`\n}\n\n// ToWire translates a Field struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Field) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StringData != nil {\n\t\tw, err = wire.NewValueString(*(v.StringData)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.IntData != nil {\n\t\tw, err = wire.NewValueI64(*(v.IntData)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.BoolData != nil {\n\t\tw, err = wire.NewValueBool(*(v.BoolData)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryData != nil {\n\t\tw, err = wire.NewValueBinary(v.BinaryData), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FieldType_Read(w wire.Value) (FieldType, error) {\n\tvar v FieldType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a Field struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Field struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Field\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Field) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x FieldType\n\t\t\t\tx, err = _FieldType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StringData = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.IntData = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.BoolData = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BinaryData, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Field struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Field struct could not be encoded.\nfunc (v *Field) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StringData != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StringData)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IntData != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.IntData)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BoolData != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.BoolData)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryData != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BinaryData); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FieldType_Decode(sr stream.Reader) (FieldType, error) {\n\tvar v FieldType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a Field struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Field struct could not be generated from the wire\n// representation.\nfunc (v *Field) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x FieldType\n\t\t\tx, err = _FieldType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StringData = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.IntData = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.BoolData = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.BinaryData, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Field\n// struct.\nfunc (v *Field) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.StringData != nil {\n\t\tfields[i] = fmt.Sprintf(\"StringData: %v\", *(v.StringData))\n\t\ti++\n\t}\n\tif v.IntData != nil {\n\t\tfields[i] = fmt.Sprintf(\"IntData: %v\", *(v.IntData))\n\t\ti++\n\t}\n\tif v.BoolData != nil {\n\t\tfields[i] = fmt.Sprintf(\"BoolData: %v\", *(v.BoolData))\n\t\ti++\n\t}\n\tif v.BinaryData != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryData: %v\", v.BinaryData)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Field{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _FieldType_EqualsPtr(lhs, rhs *FieldType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this Field match the\n// provided Field.\n//\n// This function performs a deep comparison.\nfunc (v *Field) Equals(rhs *Field) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_FieldType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StringData, rhs.StringData) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.IntData, rhs.IntData) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.BoolData, rhs.BoolData) {\n\t\treturn false\n\t}\n\tif !((v.BinaryData == nil && rhs.BinaryData == nil) || (v.BinaryData != nil && rhs.BinaryData != nil && bytes.Equal(v.BinaryData, rhs.BinaryData))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Field.\nfunc (v *Field) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.StringData != nil {\n\t\tenc.AddString(\"stringData\", *v.StringData)\n\t}\n\tif v.IntData != nil {\n\t\tenc.AddInt64(\"intData\", *v.IntData)\n\t}\n\tif v.BoolData != nil {\n\t\tenc.AddBool(\"boolData\", *v.BoolData)\n\t}\n\tif v.BinaryData != nil {\n\t\tenc.AddString(\"binaryData\", base64.StdEncoding.EncodeToString(v.BinaryData))\n\t}\n\treturn err\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *Field) GetType() (o FieldType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *Field) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetStringData returns the value of StringData if it is set or its\n// zero value if it is unset.\nfunc (v *Field) GetStringData() (o string) {\n\tif v != nil && v.StringData != nil {\n\t\treturn *v.StringData\n\t}\n\n\treturn\n}\n\n// IsSetStringData returns true if StringData is not nil.\nfunc (v *Field) IsSetStringData() bool {\n\treturn v != nil && v.StringData != nil\n}\n\n// GetIntData returns the value of IntData if it is set or its\n// zero value if it is unset.\nfunc (v *Field) GetIntData() (o int64) {\n\tif v != nil && v.IntData != nil {\n\t\treturn *v.IntData\n\t}\n\n\treturn\n}\n\n// IsSetIntData returns true if IntData is not nil.\nfunc (v *Field) IsSetIntData() bool {\n\treturn v != nil && v.IntData != nil\n}\n\n// GetBoolData returns the value of BoolData if it is set or its\n// zero value if it is unset.\nfunc (v *Field) GetBoolData() (o bool) {\n\tif v != nil && v.BoolData != nil {\n\t\treturn *v.BoolData\n\t}\n\n\treturn\n}\n\n// IsSetBoolData returns true if BoolData is not nil.\nfunc (v *Field) IsSetBoolData() bool {\n\treturn v != nil && v.BoolData != nil\n}\n\n// GetBinaryData returns the value of BinaryData if it is set or its\n// zero value if it is unset.\nfunc (v *Field) GetBinaryData() (o []byte) {\n\tif v != nil && v.BinaryData != nil {\n\t\treturn v.BinaryData\n\t}\n\n\treturn\n}\n\n// IsSetBinaryData returns true if BinaryData is not nil.\nfunc (v *Field) IsSetBinaryData() bool {\n\treturn v != nil && v.BinaryData != nil\n}\n\ntype FieldType int32\n\nconst (\n\tFieldTypeString FieldType = 0\n\tFieldTypeInt    FieldType = 1\n\tFieldTypeBool   FieldType = 2\n\tFieldTypeBinary FieldType = 3\n)\n\n// FieldType_Values returns all recognized values of FieldType.\nfunc FieldType_Values() []FieldType {\n\treturn []FieldType{\n\t\tFieldTypeString,\n\t\tFieldTypeInt,\n\t\tFieldTypeBool,\n\t\tFieldTypeBinary,\n\t}\n}\n\n// UnmarshalText tries to decode FieldType from a byte slice\n// containing its name.\n//\n//\tvar v FieldType\n//\terr := v.UnmarshalText([]byte(\"String\"))\nfunc (v *FieldType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"String\":\n\t\t*v = FieldTypeString\n\t\treturn nil\n\tcase \"Int\":\n\t\t*v = FieldTypeInt\n\t\treturn nil\n\tcase \"Bool\":\n\t\t*v = FieldTypeBool\n\t\treturn nil\n\tcase \"Binary\":\n\t\t*v = FieldTypeBinary\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"FieldType\", err)\n\t\t}\n\t\t*v = FieldType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes FieldType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v FieldType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"String\"), nil\n\tcase 1:\n\t\treturn []byte(\"Int\"), nil\n\tcase 2:\n\t\treturn []byte(\"Bool\"), nil\n\tcase 3:\n\t\treturn []byte(\"Binary\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FieldType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v FieldType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"String\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Int\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"Bool\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"Binary\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v FieldType) Ptr() *FieldType {\n\treturn &v\n}\n\n// Encode encodes FieldType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v FieldType\n//\treturn v.Encode(sWriter)\nfunc (v FieldType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates FieldType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v FieldType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes FieldType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return FieldType(0), err\n//\t}\n//\n//\tvar v FieldType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return FieldType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *FieldType) FromWire(w wire.Value) error {\n\t*v = (FieldType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded FieldType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v FieldType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return FieldType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *FieldType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (FieldType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of FieldType.\nfunc (v FieldType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"String\"\n\tcase 1:\n\t\treturn \"Int\"\n\tcase 2:\n\t\treturn \"Bool\"\n\tcase 3:\n\t\treturn \"Binary\"\n\t}\n\treturn fmt.Sprintf(\"FieldType(%d)\", w)\n}\n\n// Equals returns true if this FieldType value matches the provided\n// value.\nfunc (v FieldType) Equals(rhs FieldType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes FieldType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v FieldType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"String\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Int\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"Bool\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"Binary\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode FieldType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *FieldType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"FieldType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"FieldType\")\n\t\t}\n\t\t*v = (FieldType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"FieldType\")\n\t}\n}\n\ntype Message struct {\n\tMessageType         *MessageType         `json:\"messageType,omitempty\"`\n\tDomainID            *string              `json:\"domainID,omitempty\"`\n\tWorkflowID          *string              `json:\"workflowID,omitempty\"`\n\tRunID               *string              `json:\"runID,omitempty\"`\n\tVersion             *int64               `json:\"version,omitempty\"`\n\tFields              map[string]*Field    `json:\"fields,omitempty\"`\n\tVisibilityOperation *VisibilityOperation `json:\"visibilityOperation,omitempty\"`\n}\n\ntype _Map_String_Field_MapItemList map[string]*Field\n\nfunc (m _Map_String_Field_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*Field', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_Field_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_Field_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Field_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_Field_MapItemList) Close() {}\n\n// ToWire translates a Message struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Message) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MessageType != nil {\n\t\tw, err = v.MessageType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Fields != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Field_MapItemList(v.Fields)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityOperation != nil {\n\t\tw, err = v.VisibilityOperation.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _MessageType_Read(w wire.Value) (MessageType, error) {\n\tvar v MessageType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Field_Read(w wire.Value) (*Field, error) {\n\tvar v Field\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_Field_Read(m wire.MapItemList) (map[string]*Field, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*Field, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _Field_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _VisibilityOperation_Read(w wire.Value) (VisibilityOperation, error) {\n\tvar v VisibilityOperation\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a Message struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Message struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Message\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Message) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x MessageType\n\t\t\t\tx, err = _MessageType_Read(field.Value)\n\t\t\t\tv.MessageType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Fields, err = _Map_String_Field_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x VisibilityOperation\n\t\t\t\tx, err = _VisibilityOperation_Read(field.Value)\n\t\t\t\tv.VisibilityOperation = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_Field_Encode(val map[string]*Field, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*Field', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a Message struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Message struct could not be encoded.\nfunc (v *Message) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MessageType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.MessageType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Fields != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Field_Encode(v.Fields, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityOperation != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VisibilityOperation.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _MessageType_Decode(sr stream.Reader) (MessageType, error) {\n\tvar v MessageType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Field_Decode(sr stream.Reader) (*Field, error) {\n\tvar v Field\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_Field_Decode(sr stream.Reader) (map[string]*Field, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*Field, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _Field_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _VisibilityOperation_Decode(sr stream.Reader) (VisibilityOperation, error) {\n\tvar v VisibilityOperation\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a Message struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Message struct could not be generated from the wire\n// representation.\nfunc (v *Message) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x MessageType\n\t\t\tx, err = _MessageType_Decode(sr)\n\t\t\tv.MessageType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TMap:\n\t\t\tv.Fields, err = _Map_String_Field_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x VisibilityOperation\n\t\t\tx, err = _VisibilityOperation_Decode(sr)\n\t\t\tv.VisibilityOperation = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Message\n// struct.\nfunc (v *Message) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.MessageType != nil {\n\t\tfields[i] = fmt.Sprintf(\"MessageType: %v\", *(v.MessageType))\n\t\ti++\n\t}\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.Fields != nil {\n\t\tfields[i] = fmt.Sprintf(\"Fields: %v\", v.Fields)\n\t\ti++\n\t}\n\tif v.VisibilityOperation != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityOperation: %v\", *(v.VisibilityOperation))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Message{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _MessageType_EqualsPtr(lhs, rhs *MessageType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Map_String_Field_Equals(lhs, rhs map[string]*Field) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _VisibilityOperation_EqualsPtr(lhs, rhs *VisibilityOperation) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this Message match the\n// provided Message.\n//\n// This function performs a deep comparison.\nfunc (v *Message) Equals(rhs *Message) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_MessageType_EqualsPtr(v.MessageType, rhs.MessageType) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !((v.Fields == nil && rhs.Fields == nil) || (v.Fields != nil && rhs.Fields != nil && _Map_String_Field_Equals(v.Fields, rhs.Fields))) {\n\t\treturn false\n\t}\n\tif !_VisibilityOperation_EqualsPtr(v.VisibilityOperation, rhs.VisibilityOperation) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_Field_Zapper map[string]*Field\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_Field_Zapper.\nfunc (m _Map_String_Field_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Message.\nfunc (v *Message) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MessageType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"messageType\", *v.MessageType))\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.Fields != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"fields\", (_Map_String_Field_Zapper)(v.Fields)))\n\t}\n\tif v.VisibilityOperation != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"visibilityOperation\", *v.VisibilityOperation))\n\t}\n\treturn err\n}\n\n// GetMessageType returns the value of MessageType if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetMessageType() (o MessageType) {\n\tif v != nil && v.MessageType != nil {\n\t\treturn *v.MessageType\n\t}\n\n\treturn\n}\n\n// IsSetMessageType returns true if MessageType is not nil.\nfunc (v *Message) IsSetMessageType() bool {\n\treturn v != nil && v.MessageType != nil\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *Message) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *Message) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *Message) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *Message) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetFields returns the value of Fields if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetFields() (o map[string]*Field) {\n\tif v != nil && v.Fields != nil {\n\t\treturn v.Fields\n\t}\n\n\treturn\n}\n\n// IsSetFields returns true if Fields is not nil.\nfunc (v *Message) IsSetFields() bool {\n\treturn v != nil && v.Fields != nil\n}\n\n// GetVisibilityOperation returns the value of VisibilityOperation if it is set or its\n// zero value if it is unset.\nfunc (v *Message) GetVisibilityOperation() (o VisibilityOperation) {\n\tif v != nil && v.VisibilityOperation != nil {\n\t\treturn *v.VisibilityOperation\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityOperation returns true if VisibilityOperation is not nil.\nfunc (v *Message) IsSetVisibilityOperation() bool {\n\treturn v != nil && v.VisibilityOperation != nil\n}\n\ntype MessageType int32\n\nconst (\n\tMessageTypeIndex  MessageType = 0\n\tMessageTypeDelete MessageType = 1\n\tMessageTypeCreate MessageType = 2\n)\n\n// MessageType_Values returns all recognized values of MessageType.\nfunc MessageType_Values() []MessageType {\n\treturn []MessageType{\n\t\tMessageTypeIndex,\n\t\tMessageTypeDelete,\n\t\tMessageTypeCreate,\n\t}\n}\n\n// UnmarshalText tries to decode MessageType from a byte slice\n// containing its name.\n//\n//\tvar v MessageType\n//\terr := v.UnmarshalText([]byte(\"Index\"))\nfunc (v *MessageType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Index\":\n\t\t*v = MessageTypeIndex\n\t\treturn nil\n\tcase \"Delete\":\n\t\t*v = MessageTypeDelete\n\t\treturn nil\n\tcase \"Create\":\n\t\t*v = MessageTypeCreate\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"MessageType\", err)\n\t\t}\n\t\t*v = MessageType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes MessageType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v MessageType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Index\"), nil\n\tcase 1:\n\t\treturn []byte(\"Delete\"), nil\n\tcase 2:\n\t\treturn []byte(\"Create\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MessageType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v MessageType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Index\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Delete\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"Create\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v MessageType) Ptr() *MessageType {\n\treturn &v\n}\n\n// Encode encodes MessageType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v MessageType\n//\treturn v.Encode(sWriter)\nfunc (v MessageType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates MessageType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v MessageType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes MessageType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return MessageType(0), err\n//\t}\n//\n//\tvar v MessageType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return MessageType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *MessageType) FromWire(w wire.Value) error {\n\t*v = (MessageType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded MessageType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v MessageType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return MessageType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *MessageType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (MessageType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of MessageType.\nfunc (v MessageType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Index\"\n\tcase 1:\n\t\treturn \"Delete\"\n\tcase 2:\n\t\treturn \"Create\"\n\t}\n\treturn fmt.Sprintf(\"MessageType(%d)\", w)\n}\n\n// Equals returns true if this MessageType value matches the provided\n// value.\nfunc (v MessageType) Equals(rhs MessageType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes MessageType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v MessageType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Index\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Delete\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"Create\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode MessageType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *MessageType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"MessageType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"MessageType\")\n\t\t}\n\t\t*v = (MessageType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"MessageType\")\n\t}\n}\n\ntype PinotMessage struct {\n\tWorkflowID *string `json:\"workflowID,omitempty\"`\n\tPayload    []byte  `json:\"payload,omitempty\"`\n}\n\n// ToWire translates a PinotMessage struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PinotMessage) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Payload != nil {\n\t\tw, err = wire.NewValueBinary(v.Payload), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PinotMessage struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PinotMessage struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PinotMessage\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PinotMessage) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Payload, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PinotMessage struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PinotMessage struct could not be encoded.\nfunc (v *PinotMessage) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Payload != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Payload); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PinotMessage struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PinotMessage struct could not be generated from the wire\n// representation.\nfunc (v *PinotMessage) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Payload, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PinotMessage\n// struct.\nfunc (v *PinotMessage) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.Payload != nil {\n\t\tfields[i] = fmt.Sprintf(\"Payload: %v\", v.Payload)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PinotMessage{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PinotMessage match the\n// provided PinotMessage.\n//\n// This function performs a deep comparison.\nfunc (v *PinotMessage) Equals(rhs *PinotMessage) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.Payload == nil && rhs.Payload == nil) || (v.Payload != nil && rhs.Payload != nil && bytes.Equal(v.Payload, rhs.Payload))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PinotMessage.\nfunc (v *PinotMessage) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.Payload != nil {\n\t\tenc.AddString(\"payload\", base64.StdEncoding.EncodeToString(v.Payload))\n\t}\n\treturn err\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *PinotMessage) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *PinotMessage) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetPayload returns the value of Payload if it is set or its\n// zero value if it is unset.\nfunc (v *PinotMessage) GetPayload() (o []byte) {\n\tif v != nil && v.Payload != nil {\n\t\treturn v.Payload\n\t}\n\n\treturn\n}\n\n// IsSetPayload returns true if Payload is not nil.\nfunc (v *PinotMessage) IsSetPayload() bool {\n\treturn v != nil && v.Payload != nil\n}\n\ntype VisibilityOperation int32\n\nconst (\n\tVisibilityOperationRecordStarted          VisibilityOperation = 0\n\tVisibilityOperationRecordClosed           VisibilityOperation = 1\n\tVisibilityOperationUpsertSearchAttributes VisibilityOperation = 2\n)\n\n// VisibilityOperation_Values returns all recognized values of VisibilityOperation.\nfunc VisibilityOperation_Values() []VisibilityOperation {\n\treturn []VisibilityOperation{\n\t\tVisibilityOperationRecordStarted,\n\t\tVisibilityOperationRecordClosed,\n\t\tVisibilityOperationUpsertSearchAttributes,\n\t}\n}\n\n// UnmarshalText tries to decode VisibilityOperation from a byte slice\n// containing its name.\n//\n//\tvar v VisibilityOperation\n//\terr := v.UnmarshalText([]byte(\"RecordStarted\"))\nfunc (v *VisibilityOperation) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"RecordStarted\":\n\t\t*v = VisibilityOperationRecordStarted\n\t\treturn nil\n\tcase \"RecordClosed\":\n\t\t*v = VisibilityOperationRecordClosed\n\t\treturn nil\n\tcase \"UpsertSearchAttributes\":\n\t\t*v = VisibilityOperationUpsertSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"VisibilityOperation\", err)\n\t\t}\n\t\t*v = VisibilityOperation(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes VisibilityOperation to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v VisibilityOperation) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"RecordStarted\"), nil\n\tcase 1:\n\t\treturn []byte(\"RecordClosed\"), nil\n\tcase 2:\n\t\treturn []byte(\"UpsertSearchAttributes\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VisibilityOperation.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v VisibilityOperation) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"RecordStarted\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"RecordClosed\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"UpsertSearchAttributes\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v VisibilityOperation) Ptr() *VisibilityOperation {\n\treturn &v\n}\n\n// Encode encodes VisibilityOperation directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v VisibilityOperation\n//\treturn v.Encode(sWriter)\nfunc (v VisibilityOperation) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates VisibilityOperation into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v VisibilityOperation) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes VisibilityOperation from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return VisibilityOperation(0), err\n//\t}\n//\n//\tvar v VisibilityOperation\n//\tif err := v.FromWire(x); err != nil {\n//\t  return VisibilityOperation(0), err\n//\t}\n//\treturn v, nil\nfunc (v *VisibilityOperation) FromWire(w wire.Value) error {\n\t*v = (VisibilityOperation)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded VisibilityOperation directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v VisibilityOperation\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return VisibilityOperation(0), err\n//\t}\n//\treturn v, nil\nfunc (v *VisibilityOperation) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (VisibilityOperation)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of VisibilityOperation.\nfunc (v VisibilityOperation) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"RecordStarted\"\n\tcase 1:\n\t\treturn \"RecordClosed\"\n\tcase 2:\n\t\treturn \"UpsertSearchAttributes\"\n\t}\n\treturn fmt.Sprintf(\"VisibilityOperation(%d)\", w)\n}\n\n// Equals returns true if this VisibilityOperation value matches the provided\n// value.\nfunc (v VisibilityOperation) Equals(rhs VisibilityOperation) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes VisibilityOperation into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v VisibilityOperation) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"RecordStarted\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"RecordClosed\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"UpsertSearchAttributes\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode VisibilityOperation from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *VisibilityOperation) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"VisibilityOperation\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"VisibilityOperation\")\n\t\t}\n\t\t*v = (VisibilityOperation)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"VisibilityOperation\")\n\t}\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"indexer\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/indexer\",\n\tFilePath: \"indexer.thrift\",\n\tSHA1:     \"d5b60fa082530a64a77c19c573c0f3ccfce5c408\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.indexer\\n\\ninclude \\\"shared.thrift\\\"\\n\\nenum MessageType {\\n  Index\\n  Delete\\n  Create\\n}\\n\\nenum VisibilityOperation {\\n  RecordStarted\\n  RecordClosed\\n  UpsertSearchAttributes\\n}\\n\\nenum FieldType {\\n  String\\n  Int\\n  Bool\\n  Binary\\n}\\n\\nstruct Field {\\n  10: optional FieldType type\\n  20: optional string stringData\\n  30: optional i64 (js.type = \\\"Long\\\") intData\\n  40: optional bool boolData\\n  50: optional binary binaryData\\n}\\n\\nstruct Message {\\n  10: optional MessageType messageType\\n  20: optional string domainID\\n  30: optional string workflowID\\n  40: optional string runID\\n  50: optional i64 (js.type = \\\"Long\\\") version\\n  60: optional map<string,Field> fields\\n  70: optional VisibilityOperation visibilityOperation\\n}\\n\\nstruct PinotMessage {\\n  10: optional string workflowID\\n  20: optional binary payload\\n}\"\n"
  },
  {
    "path": ".gen/go/indexer/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage indexer\n"
  },
  {
    "path": ".gen/go/matching/matching.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage matching\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype ActivityTaskDispatchInfo struct {\n\tScheduledEvent                  *shared.HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedTimestamp                *int64               `json:\"startedTimestamp,omitempty\"`\n\tAttempt                         *int64               `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64               `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tScheduledTimestamp              *int64               `json:\"scheduledTimestamp,omitempty\"`\n\tHeartbeatDetails                []byte               `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *shared.WorkflowType `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  *string              `json:\"workflowDomain,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskDispatchInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskDispatchInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEvent != nil {\n\t\tw, err = v.ScheduledEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestampOfThisAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.HeartbeatDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryEvent_Read(w wire.Value) (*shared.HistoryEvent, error) {\n\tvar v shared.HistoryEvent\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowType_Read(w wire.Value) (*shared.WorkflowType, error) {\n\tvar v shared.WorkflowType\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ActivityTaskDispatchInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskDispatchInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskDispatchInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskDispatchInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ScheduledEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.HeartbeatDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskDispatchInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskDispatchInfo struct could not be encoded.\nfunc (v *ActivityTaskDispatchInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ScheduledEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestampOfThisAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.HeartbeatDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryEvent_Decode(sr stream.Reader) (*shared.HistoryEvent, error) {\n\tvar v shared.HistoryEvent\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowType_Decode(sr stream.Reader) (*shared.WorkflowType, error) {\n\tvar v shared.WorkflowType\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ActivityTaskDispatchInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskDispatchInfo struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskDispatchInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.ScheduledEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.HeartbeatDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskDispatchInfo\n// struct.\nfunc (v *ActivityTaskDispatchInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.ScheduledEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEvent: %v\", v.ScheduledEvent)\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestampOfThisAttempt: %v\", *(v.ScheduledTimestampOfThisAttempt))\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatDetails: %v\", v.HeartbeatDetails)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowDomain: %v\", *(v.WorkflowDomain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskDispatchInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ActivityTaskDispatchInfo match the\n// provided ActivityTaskDispatchInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskDispatchInfo) Equals(rhs *ActivityTaskDispatchInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ScheduledEvent == nil && rhs.ScheduledEvent == nil) || (v.ScheduledEvent != nil && rhs.ScheduledEvent != nil && v.ScheduledEvent.Equals(rhs.ScheduledEvent))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestampOfThisAttempt, rhs.ScheduledTimestampOfThisAttempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatDetails == nil && rhs.HeartbeatDetails == nil) || (v.HeartbeatDetails != nil && rhs.HeartbeatDetails != nil && bytes.Equal(v.HeartbeatDetails, rhs.HeartbeatDetails))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowDomain, rhs.WorkflowDomain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskDispatchInfo.\nfunc (v *ActivityTaskDispatchInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"scheduledEvent\", v.ScheduledEvent))\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tenc.AddInt64(\"scheduledTimestampOfThisAttempt\", *v.ScheduledTimestampOfThisAttempt)\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tenc.AddString(\"heartbeatDetails\", base64.StdEncoding.EncodeToString(v.HeartbeatDetails))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tenc.AddString(\"workflowDomain\", *v.WorkflowDomain)\n\t}\n\treturn err\n}\n\n// GetScheduledEvent returns the value of ScheduledEvent if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetScheduledEvent() (o *shared.HistoryEvent) {\n\tif v != nil && v.ScheduledEvent != nil {\n\t\treturn v.ScheduledEvent\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEvent returns true if ScheduledEvent is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetScheduledEvent() bool {\n\treturn v != nil && v.ScheduledEvent != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetScheduledTimestampOfThisAttempt returns the value of ScheduledTimestampOfThisAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetScheduledTimestampOfThisAttempt() (o int64) {\n\tif v != nil && v.ScheduledTimestampOfThisAttempt != nil {\n\t\treturn *v.ScheduledTimestampOfThisAttempt\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestampOfThisAttempt returns true if ScheduledTimestampOfThisAttempt is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetScheduledTimestampOfThisAttempt() bool {\n\treturn v != nil && v.ScheduledTimestampOfThisAttempt != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetHeartbeatDetails returns the value of HeartbeatDetails if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetHeartbeatDetails() (o []byte) {\n\tif v != nil && v.HeartbeatDetails != nil {\n\t\treturn v.HeartbeatDetails\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatDetails returns true if HeartbeatDetails is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetHeartbeatDetails() bool {\n\treturn v != nil && v.HeartbeatDetails != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetWorkflowDomain returns the value of WorkflowDomain if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskDispatchInfo) GetWorkflowDomain() (o string) {\n\tif v != nil && v.WorkflowDomain != nil {\n\t\treturn *v.WorkflowDomain\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowDomain returns true if WorkflowDomain is not nil.\nfunc (v *ActivityTaskDispatchInfo) IsSetWorkflowDomain() bool {\n\treturn v != nil && v.WorkflowDomain != nil\n}\n\ntype AddActivityTaskRequest struct {\n\tDomainUUID                    *string                   `json:\"domainUUID,omitempty\"`\n\tExecution                     *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tSourceDomainUUID              *string                   `json:\"sourceDomainUUID,omitempty\"`\n\tTaskList                      *shared.TaskList          `json:\"taskList,omitempty\"`\n\tScheduleId                    *int64                    `json:\"scheduleId,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32                    `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tSource                        *TaskSource               `json:\"source,omitempty\"`\n\tForwardedFrom                 *string                   `json:\"forwardedFrom,omitempty\"`\n\tActivityTaskDispatchInfo      *ActivityTaskDispatchInfo `json:\"activityTaskDispatchInfo,omitempty\"`\n\tPartitionConfig               map[string]string         `json:\"partitionConfig,omitempty\"`\n}\n\ntype _Map_String_String_MapItemList map[string]string\n\nfunc (m _Map_String_String_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueString(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_String_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_String_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) Close() {}\n\n// ToWire translates a AddActivityTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AddActivityTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [10]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SourceDomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceDomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Source != nil {\n\t\tw, err = v.Source.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 69, Value: w}\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tw, err = wire.NewValueString(*(v.ForwardedFrom)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskDispatchInfo != nil {\n\t\tw, err = v.ActivityTaskDispatchInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecution_Read(w wire.Value) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TaskList_Read(w wire.Value) (*shared.TaskList, error) {\n\tvar v shared.TaskList\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TaskSource_Read(w wire.Value) (TaskSource, error) {\n\tvar v TaskSource\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _ActivityTaskDispatchInfo_Read(w wire.Value) (*ActivityTaskDispatchInfo, error) {\n\tvar v ActivityTaskDispatchInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Read(m wire.MapItemList) (map[string]string, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]string, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a AddActivityTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AddActivityTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AddActivityTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AddActivityTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceDomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 69:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TaskSource\n\t\t\t\tx, err = _TaskSource_Read(field.Value)\n\t\t\t\tv.Source = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ForwardedFrom = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskDispatchInfo, err = _ActivityTaskDispatchInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_String_Encode(val map[string]string, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a AddActivityTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AddActivityTaskRequest struct could not be encoded.\nfunc (v *AddActivityTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SourceDomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceDomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Source != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 69, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Source.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForwardedFrom != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ForwardedFrom)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskDispatchInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskDispatchInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecution_Decode(sr stream.Reader) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TaskList_Decode(sr stream.Reader) (*shared.TaskList, error) {\n\tvar v shared.TaskList\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TaskSource_Decode(sr stream.Reader) (TaskSource, error) {\n\tvar v TaskSource\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _ActivityTaskDispatchInfo_Decode(sr stream.Reader) (*ActivityTaskDispatchInfo, error) {\n\tvar v ActivityTaskDispatchInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Decode(sr stream.Reader) (map[string]string, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]string, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a AddActivityTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AddActivityTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *AddActivityTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceDomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 69 && fh.Type == wire.TI32:\n\t\t\tvar x TaskSource\n\t\t\tx, err = _TaskSource_Decode(sr)\n\t\t\tv.Source = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ForwardedFrom = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskDispatchInfo, err = _ActivityTaskDispatchInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AddActivityTaskRequest\n// struct.\nfunc (v *AddActivityTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [10]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.SourceDomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceDomainUUID: %v\", *(v.SourceDomainUUID))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleId: %v\", *(v.ScheduleId))\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Source != nil {\n\t\tfields[i] = fmt.Sprintf(\"Source: %v\", *(v.Source))\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForwardedFrom: %v\", *(v.ForwardedFrom))\n\t\ti++\n\t}\n\tif v.ActivityTaskDispatchInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskDispatchInfo: %v\", v.ActivityTaskDispatchInfo)\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AddActivityTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _TaskSource_EqualsPtr(lhs, rhs *TaskSource) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Map_String_String_Equals(lhs, rhs map[string]string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this AddActivityTaskRequest match the\n// provided AddActivityTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *AddActivityTaskRequest) Equals(rhs *AddActivityTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceDomainUUID, rhs.SourceDomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleId, rhs.ScheduleId) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_TaskSource_EqualsPtr(v.Source, rhs.Source) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ForwardedFrom, rhs.ForwardedFrom) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskDispatchInfo == nil && rhs.ActivityTaskDispatchInfo == nil) || (v.ActivityTaskDispatchInfo != nil && rhs.ActivityTaskDispatchInfo != nil && v.ActivityTaskDispatchInfo.Equals(rhs.ActivityTaskDispatchInfo))) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_String_Zapper map[string]string\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_String_Zapper.\nfunc (m _Map_String_String_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AddActivityTaskRequest.\nfunc (v *AddActivityTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.SourceDomainUUID != nil {\n\t\tenc.AddString(\"sourceDomainUUID\", *v.SourceDomainUUID)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.ScheduleId != nil {\n\t\tenc.AddInt64(\"scheduleId\", *v.ScheduleId)\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\tif v.Source != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"source\", *v.Source))\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tenc.AddString(\"forwardedFrom\", *v.ForwardedFrom)\n\t}\n\tif v.ActivityTaskDispatchInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskDispatchInfo\", v.ActivityTaskDispatchInfo))\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *AddActivityTaskRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *AddActivityTaskRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetSourceDomainUUID returns the value of SourceDomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetSourceDomainUUID() (o string) {\n\tif v != nil && v.SourceDomainUUID != nil {\n\t\treturn *v.SourceDomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetSourceDomainUUID returns true if SourceDomainUUID is not nil.\nfunc (v *AddActivityTaskRequest) IsSetSourceDomainUUID() bool {\n\treturn v != nil && v.SourceDomainUUID != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *AddActivityTaskRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetScheduleId returns the value of ScheduleId if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetScheduleId() (o int64) {\n\tif v != nil && v.ScheduleId != nil {\n\t\treturn *v.ScheduleId\n\t}\n\n\treturn\n}\n\n// IsSetScheduleId returns true if ScheduleId is not nil.\nfunc (v *AddActivityTaskRequest) IsSetScheduleId() bool {\n\treturn v != nil && v.ScheduleId != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *AddActivityTaskRequest) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\n// GetSource returns the value of Source if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetSource() (o TaskSource) {\n\tif v != nil && v.Source != nil {\n\t\treturn *v.Source\n\t}\n\n\treturn\n}\n\n// IsSetSource returns true if Source is not nil.\nfunc (v *AddActivityTaskRequest) IsSetSource() bool {\n\treturn v != nil && v.Source != nil\n}\n\n// GetForwardedFrom returns the value of ForwardedFrom if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil && v.ForwardedFrom != nil {\n\t\treturn *v.ForwardedFrom\n\t}\n\n\treturn\n}\n\n// IsSetForwardedFrom returns true if ForwardedFrom is not nil.\nfunc (v *AddActivityTaskRequest) IsSetForwardedFrom() bool {\n\treturn v != nil && v.ForwardedFrom != nil\n}\n\n// GetActivityTaskDispatchInfo returns the value of ActivityTaskDispatchInfo if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetActivityTaskDispatchInfo() (o *ActivityTaskDispatchInfo) {\n\tif v != nil && v.ActivityTaskDispatchInfo != nil {\n\t\treturn v.ActivityTaskDispatchInfo\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskDispatchInfo returns true if ActivityTaskDispatchInfo is not nil.\nfunc (v *AddActivityTaskRequest) IsSetActivityTaskDispatchInfo() bool {\n\treturn v != nil && v.ActivityTaskDispatchInfo != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *AddActivityTaskRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *AddActivityTaskRequest) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype AddDecisionTaskRequest struct {\n\tDomainUUID                    *string                   `json:\"domainUUID,omitempty\"`\n\tExecution                     *shared.WorkflowExecution `json:\"execution,omitempty\"`\n\tTaskList                      *shared.TaskList          `json:\"taskList,omitempty\"`\n\tScheduleId                    *int64                    `json:\"scheduleId,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32                    `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tSource                        *TaskSource               `json:\"source,omitempty\"`\n\tForwardedFrom                 *string                   `json:\"forwardedFrom,omitempty\"`\n\tPartitionConfig               map[string]string         `json:\"partitionConfig,omitempty\"`\n}\n\n// ToWire translates a AddDecisionTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AddDecisionTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Source != nil {\n\t\tw, err = v.Source.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 59, Value: w}\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tw, err = wire.NewValueString(*(v.ForwardedFrom)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AddDecisionTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AddDecisionTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AddDecisionTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AddDecisionTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 59:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TaskSource\n\t\t\t\tx, err = _TaskSource_Read(field.Value)\n\t\t\t\tv.Source = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ForwardedFrom = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AddDecisionTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AddDecisionTaskRequest struct could not be encoded.\nfunc (v *AddDecisionTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Source != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 59, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Source.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForwardedFrom != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ForwardedFrom)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AddDecisionTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AddDecisionTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *AddDecisionTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 59 && fh.Type == wire.TI32:\n\t\t\tvar x TaskSource\n\t\t\tx, err = _TaskSource_Decode(sr)\n\t\t\tv.Source = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ForwardedFrom = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AddDecisionTaskRequest\n// struct.\nfunc (v *AddDecisionTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.ScheduleId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleId: %v\", *(v.ScheduleId))\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Source != nil {\n\t\tfields[i] = fmt.Sprintf(\"Source: %v\", *(v.Source))\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForwardedFrom: %v\", *(v.ForwardedFrom))\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AddDecisionTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AddDecisionTaskRequest match the\n// provided AddDecisionTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *AddDecisionTaskRequest) Equals(rhs *AddDecisionTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleId, rhs.ScheduleId) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_TaskSource_EqualsPtr(v.Source, rhs.Source) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ForwardedFrom, rhs.ForwardedFrom) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AddDecisionTaskRequest.\nfunc (v *AddDecisionTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.ScheduleId != nil {\n\t\tenc.AddInt64(\"scheduleId\", *v.ScheduleId)\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\tif v.Source != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"source\", *v.Source))\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tenc.AddString(\"forwardedFrom\", *v.ForwardedFrom)\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetScheduleId returns the value of ScheduleId if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetScheduleId() (o int64) {\n\tif v != nil && v.ScheduleId != nil {\n\t\treturn *v.ScheduleId\n\t}\n\n\treturn\n}\n\n// IsSetScheduleId returns true if ScheduleId is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetScheduleId() bool {\n\treturn v != nil && v.ScheduleId != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\n// GetSource returns the value of Source if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetSource() (o TaskSource) {\n\tif v != nil && v.Source != nil {\n\t\treturn *v.Source\n\t}\n\n\treturn\n}\n\n// IsSetSource returns true if Source is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetSource() bool {\n\treturn v != nil && v.Source != nil\n}\n\n// GetForwardedFrom returns the value of ForwardedFrom if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil && v.ForwardedFrom != nil {\n\t\treturn *v.ForwardedFrom\n\t}\n\n\treturn\n}\n\n// IsSetForwardedFrom returns true if ForwardedFrom is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetForwardedFrom() bool {\n\treturn v != nil && v.ForwardedFrom != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *AddDecisionTaskRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *AddDecisionTaskRequest) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype CancelOutstandingPollRequest struct {\n\tDomainUUID   *string          `json:\"domainUUID,omitempty\"`\n\tTaskListType *int32           `json:\"taskListType,omitempty\"`\n\tTaskList     *shared.TaskList `json:\"taskList,omitempty\"`\n\tPollerID     *string          `json:\"pollerID,omitempty\"`\n}\n\n// ToWire translates a CancelOutstandingPollRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CancelOutstandingPollRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListType != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskListType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tw, err = wire.NewValueString(*(v.PollerID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CancelOutstandingPollRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CancelOutstandingPollRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CancelOutstandingPollRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CancelOutstandingPollRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskListType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PollerID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CancelOutstandingPollRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CancelOutstandingPollRequest struct could not be encoded.\nfunc (v *CancelOutstandingPollRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskListType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollerID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PollerID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CancelOutstandingPollRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CancelOutstandingPollRequest struct could not be generated from the wire\n// representation.\nfunc (v *CancelOutstandingPollRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskListType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PollerID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CancelOutstandingPollRequest\n// struct.\nfunc (v *CancelOutstandingPollRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.TaskListType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListType: %v\", *(v.TaskListType))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollerID: %v\", *(v.PollerID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CancelOutstandingPollRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CancelOutstandingPollRequest match the\n// provided CancelOutstandingPollRequest.\n//\n// This function performs a deep comparison.\nfunc (v *CancelOutstandingPollRequest) Equals(rhs *CancelOutstandingPollRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskListType, rhs.TaskListType) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PollerID, rhs.PollerID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancelOutstandingPollRequest.\nfunc (v *CancelOutstandingPollRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.TaskListType != nil {\n\t\tenc.AddInt32(\"taskListType\", *v.TaskListType)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.PollerID != nil {\n\t\tenc.AddString(\"pollerID\", *v.PollerID)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *CancelOutstandingPollRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *CancelOutstandingPollRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetTaskListType returns the value of TaskListType if it is set or its\n// zero value if it is unset.\nfunc (v *CancelOutstandingPollRequest) GetTaskListType() (o int32) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\n\treturn\n}\n\n// IsSetTaskListType returns true if TaskListType is not nil.\nfunc (v *CancelOutstandingPollRequest) IsSetTaskListType() bool {\n\treturn v != nil && v.TaskListType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *CancelOutstandingPollRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *CancelOutstandingPollRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetPollerID returns the value of PollerID if it is set or its\n// zero value if it is unset.\nfunc (v *CancelOutstandingPollRequest) GetPollerID() (o string) {\n\tif v != nil && v.PollerID != nil {\n\t\treturn *v.PollerID\n\t}\n\n\treturn\n}\n\n// IsSetPollerID returns true if PollerID is not nil.\nfunc (v *CancelOutstandingPollRequest) IsSetPollerID() bool {\n\treturn v != nil && v.PollerID != nil\n}\n\ntype DescribeTaskListRequest struct {\n\tDomainUUID  *string                         `json:\"domainUUID,omitempty\"`\n\tDescRequest *shared.DescribeTaskListRequest `json:\"descRequest,omitempty\"`\n}\n\n// ToWire translates a DescribeTaskListRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeTaskListRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DescRequest != nil {\n\t\tw, err = v.DescRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListRequest_Read(w wire.Value) (*shared.DescribeTaskListRequest, error) {\n\tvar v shared.DescribeTaskListRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeTaskListRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeTaskListRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeTaskListRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeTaskListRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DescRequest, err = _DescribeTaskListRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeTaskListRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeTaskListRequest struct could not be encoded.\nfunc (v *DescribeTaskListRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DescRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DescRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListRequest_Decode(sr stream.Reader) (*shared.DescribeTaskListRequest, error) {\n\tvar v shared.DescribeTaskListRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeTaskListRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeTaskListRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeTaskListRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.DescRequest, err = _DescribeTaskListRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeTaskListRequest\n// struct.\nfunc (v *DescribeTaskListRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.DescRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"DescRequest: %v\", v.DescRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeTaskListRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeTaskListRequest match the\n// provided DescribeTaskListRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeTaskListRequest) Equals(rhs *DescribeTaskListRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.DescRequest == nil && rhs.DescRequest == nil) || (v.DescRequest != nil && rhs.DescRequest != nil && v.DescRequest.Equals(rhs.DescRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeTaskListRequest.\nfunc (v *DescribeTaskListRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.DescRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"descRequest\", v.DescRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *DescribeTaskListRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetDescRequest returns the value of DescRequest if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetDescRequest() (o *shared.DescribeTaskListRequest) {\n\tif v != nil && v.DescRequest != nil {\n\t\treturn v.DescRequest\n\t}\n\n\treturn\n}\n\n// IsSetDescRequest returns true if DescRequest is not nil.\nfunc (v *DescribeTaskListRequest) IsSetDescRequest() bool {\n\treturn v != nil && v.DescRequest != nil\n}\n\ntype ListTaskListPartitionsRequest struct {\n\tDomain   *string          `json:\"domain,omitempty\"`\n\tTaskList *shared.TaskList `json:\"taskList,omitempty\"`\n}\n\n// ToWire translates a ListTaskListPartitionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListTaskListPartitionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListTaskListPartitionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListTaskListPartitionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListTaskListPartitionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListTaskListPartitionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListTaskListPartitionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListTaskListPartitionsRequest struct could not be encoded.\nfunc (v *ListTaskListPartitionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListTaskListPartitionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListTaskListPartitionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListTaskListPartitionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListTaskListPartitionsRequest\n// struct.\nfunc (v *ListTaskListPartitionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListTaskListPartitionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListTaskListPartitionsRequest match the\n// provided ListTaskListPartitionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListTaskListPartitionsRequest) Equals(rhs *ListTaskListPartitionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListTaskListPartitionsRequest.\nfunc (v *ListTaskListPartitionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListTaskListPartitionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ListTaskListPartitionsRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\ntype PollForActivityTaskRequest struct {\n\tDomainUUID     *string                            `json:\"domainUUID,omitempty\"`\n\tPollerID       *string                            `json:\"pollerID,omitempty\"`\n\tPollRequest    *shared.PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n\tForwardedFrom  *string                            `json:\"forwardedFrom,omitempty\"`\n\tIsolationGroup *string                            `json:\"isolationGroup,omitempty\"`\n}\n\n// ToWire translates a PollForActivityTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForActivityTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tw, err = wire.NewValueString(*(v.PollerID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 15, Value: w}\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tw, err = wire.NewValueString(*(v.ForwardedFrom)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroup != nil {\n\t\tw, err = wire.NewValueString(*(v.IsolationGroup)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskRequest_Read(w wire.Value) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a PollForActivityTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForActivityTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForActivityTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForActivityTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 15:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PollerID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ForwardedFrom = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.IsolationGroup = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollForActivityTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForActivityTaskRequest struct could not be encoded.\nfunc (v *PollForActivityTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollerID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 15, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PollerID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForwardedFrom != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ForwardedFrom)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroup != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.IsolationGroup)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskRequest_Decode(sr stream.Reader) (*shared.PollForActivityTaskRequest, error) {\n\tvar v shared.PollForActivityTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a PollForActivityTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForActivityTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *PollForActivityTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 15 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PollerID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ForwardedFrom = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.IsolationGroup = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForActivityTaskRequest\n// struct.\nfunc (v *PollForActivityTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollerID: %v\", *(v.PollerID))\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForwardedFrom: %v\", *(v.ForwardedFrom))\n\t\ti++\n\t}\n\tif v.IsolationGroup != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroup: %v\", *(v.IsolationGroup))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForActivityTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollForActivityTaskRequest match the\n// provided PollForActivityTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PollForActivityTaskRequest) Equals(rhs *PollForActivityTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PollerID, rhs.PollerID) {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ForwardedFrom, rhs.ForwardedFrom) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.IsolationGroup, rhs.IsolationGroup) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForActivityTaskRequest.\nfunc (v *PollForActivityTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.PollerID != nil {\n\t\tenc.AddString(\"pollerID\", *v.PollerID)\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tenc.AddString(\"forwardedFrom\", *v.ForwardedFrom)\n\t}\n\tif v.IsolationGroup != nil {\n\t\tenc.AddString(\"isolationGroup\", *v.IsolationGroup)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetPollerID returns the value of PollerID if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetPollerID() (o string) {\n\tif v != nil && v.PollerID != nil {\n\t\treturn *v.PollerID\n\t}\n\n\treturn\n}\n\n// IsSetPollerID returns true if PollerID is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetPollerID() bool {\n\treturn v != nil && v.PollerID != nil\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetPollRequest() (o *shared.PollForActivityTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// GetForwardedFrom returns the value of ForwardedFrom if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil && v.ForwardedFrom != nil {\n\t\treturn *v.ForwardedFrom\n\t}\n\n\treturn\n}\n\n// IsSetForwardedFrom returns true if ForwardedFrom is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetForwardedFrom() bool {\n\treturn v != nil && v.ForwardedFrom != nil\n}\n\n// GetIsolationGroup returns the value of IsolationGroup if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetIsolationGroup() (o string) {\n\tif v != nil && v.IsolationGroup != nil {\n\t\treturn *v.IsolationGroup\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroup returns true if IsolationGroup is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetIsolationGroup() bool {\n\treturn v != nil && v.IsolationGroup != nil\n}\n\ntype PollForDecisionTaskRequest struct {\n\tDomainUUID     *string                            `json:\"domainUUID,omitempty\"`\n\tPollerID       *string                            `json:\"pollerID,omitempty\"`\n\tPollRequest    *shared.PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n\tForwardedFrom  *string                            `json:\"forwardedFrom,omitempty\"`\n\tIsolationGroup *string                            `json:\"isolationGroup,omitempty\"`\n}\n\n// ToWire translates a PollForDecisionTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForDecisionTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tw, err = wire.NewValueString(*(v.PollerID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 15, Value: w}\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tw, err = wire.NewValueString(*(v.ForwardedFrom)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroup != nil {\n\t\tw, err = wire.NewValueString(*(v.IsolationGroup)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskRequest_Read(w wire.Value) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a PollForDecisionTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForDecisionTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForDecisionTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForDecisionTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 15:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PollerID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ForwardedFrom = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.IsolationGroup = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollForDecisionTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForDecisionTaskRequest struct could not be encoded.\nfunc (v *PollForDecisionTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollerID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 15, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PollerID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForwardedFrom != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ForwardedFrom)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroup != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.IsolationGroup)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskRequest_Decode(sr stream.Reader) (*shared.PollForDecisionTaskRequest, error) {\n\tvar v shared.PollForDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a PollForDecisionTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForDecisionTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *PollForDecisionTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 15 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PollerID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ForwardedFrom = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.IsolationGroup = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForDecisionTaskRequest\n// struct.\nfunc (v *PollForDecisionTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.PollerID != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollerID: %v\", *(v.PollerID))\n\t\ti++\n\t}\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForwardedFrom: %v\", *(v.ForwardedFrom))\n\t\ti++\n\t}\n\tif v.IsolationGroup != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroup: %v\", *(v.IsolationGroup))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForDecisionTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollForDecisionTaskRequest match the\n// provided PollForDecisionTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PollForDecisionTaskRequest) Equals(rhs *PollForDecisionTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PollerID, rhs.PollerID) {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ForwardedFrom, rhs.ForwardedFrom) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.IsolationGroup, rhs.IsolationGroup) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForDecisionTaskRequest.\nfunc (v *PollForDecisionTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.PollerID != nil {\n\t\tenc.AddString(\"pollerID\", *v.PollerID)\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tenc.AddString(\"forwardedFrom\", *v.ForwardedFrom)\n\t}\n\tif v.IsolationGroup != nil {\n\t\tenc.AddString(\"isolationGroup\", *v.IsolationGroup)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetPollerID returns the value of PollerID if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetPollerID() (o string) {\n\tif v != nil && v.PollerID != nil {\n\t\treturn *v.PollerID\n\t}\n\n\treturn\n}\n\n// IsSetPollerID returns true if PollerID is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetPollerID() bool {\n\treturn v != nil && v.PollerID != nil\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetPollRequest() (o *shared.PollForDecisionTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// GetForwardedFrom returns the value of ForwardedFrom if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil && v.ForwardedFrom != nil {\n\t\treturn *v.ForwardedFrom\n\t}\n\n\treturn\n}\n\n// IsSetForwardedFrom returns true if ForwardedFrom is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetForwardedFrom() bool {\n\treturn v != nil && v.ForwardedFrom != nil\n}\n\n// GetIsolationGroup returns the value of IsolationGroup if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetIsolationGroup() (o string) {\n\tif v != nil && v.IsolationGroup != nil {\n\t\treturn *v.IsolationGroup\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroup returns true if IsolationGroup is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetIsolationGroup() bool {\n\treturn v != nil && v.IsolationGroup != nil\n}\n\ntype PollForDecisionTaskResponse struct {\n\tTaskToken                 []byte                           `json:\"taskToken,omitempty\"`\n\tWorkflowExecution         *shared.WorkflowExecution        `json:\"workflowExecution,omitempty\"`\n\tWorkflowType              *shared.WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventId    *int64                           `json:\"previousStartedEventId,omitempty\"`\n\tStartedEventId            *int64                           `json:\"startedEventId,omitempty\"`\n\tAttempt                   *int64                           `json:\"attempt,omitempty\"`\n\tNextEventId               *int64                           `json:\"nextEventId,omitempty\"`\n\tBacklogCountHint          *int64                           `json:\"backlogCountHint,omitempty\"`\n\tStickyExecutionEnabled    *bool                            `json:\"stickyExecutionEnabled,omitempty\"`\n\tQuery                     *shared.WorkflowQuery            `json:\"query,omitempty\"`\n\tDecisionInfo              *shared.TransientDecisionInfo    `json:\"decisionInfo,omitempty\"`\n\tWorkflowExecutionTaskList *shared.TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tEventStoreVersion         *int32                           `json:\"eventStoreVersion,omitempty\"`\n\tBranchToken               []byte                           `json:\"branchToken,omitempty\"`\n\tScheduledTimestamp        *int64                           `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                           `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*shared.WorkflowQuery `json:\"queries,omitempty\"`\n\tTotalHistoryBytes         *int64                           `json:\"totalHistoryBytes,omitempty\"`\n\tAutoConfigHint            *shared.AutoConfigHint           `json:\"autoConfigHint,omitempty\"`\n}\n\ntype _Map_String_WorkflowQuery_MapItemList map[string]*shared.WorkflowQuery\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) Close() {}\n\n// ToWire translates a PollForDecisionTaskResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForDecisionTaskResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [19]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousStartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 51, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tw, err = wire.NewValueI64(*(v.BacklogCountHint)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 65, Value: w}\n\t\ti++\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.StickyExecutionEnabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = v.Query.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionInfo != nil {\n\t\tw, err = v.DecisionInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tw, err = v.WorkflowExecutionTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.EventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.BranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_WorkflowQuery_MapItemList(v.Queries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tw, err = wire.NewValueI64(*(v.TotalHistoryBytes)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tw, err = v.AutoConfigHint.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowQuery_Read(w wire.Value) (*shared.WorkflowQuery, error) {\n\tvar v shared.WorkflowQuery\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TransientDecisionInfo_Read(w wire.Value) (*shared.TransientDecisionInfo, error) {\n\tvar v shared.TransientDecisionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Read(m wire.MapItemList) (map[string]*shared.WorkflowQuery, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*shared.WorkflowQuery, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _AutoConfigHint_Read(w wire.Value) (*shared.AutoConfigHint, error) {\n\tvar v shared.AutoConfigHint\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a PollForDecisionTaskResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForDecisionTaskResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForDecisionTaskResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForDecisionTaskResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousStartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 51:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 65:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.BacklogCountHint = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.StickyExecutionEnabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Query, err = _WorkflowQuery_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionInfo, err = _TransientDecisionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.EventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TotalHistoryBytes = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_WorkflowQuery_Encode(val map[string]*shared.WorkflowQuery, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*shared.WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a PollForDecisionTaskResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForDecisionTaskResponse struct could not be encoded.\nfunc (v *PollForDecisionTaskResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousStartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousStartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 51, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BacklogCountHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 65, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.BacklogCountHint)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyExecutionEnabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.StickyExecutionEnabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Query.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.EventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Queries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_WorkflowQuery_Encode(v.Queries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TotalHistoryBytes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TotalHistoryBytes)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoConfigHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AutoConfigHint.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowQuery_Decode(sr stream.Reader) (*shared.WorkflowQuery, error) {\n\tvar v shared.WorkflowQuery\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TransientDecisionInfo_Decode(sr stream.Reader) (*shared.TransientDecisionInfo, error) {\n\tvar v shared.TransientDecisionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Decode(sr stream.Reader) (map[string]*shared.WorkflowQuery, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*shared.WorkflowQuery, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _AutoConfigHint_Decode(sr stream.Reader) (*shared.AutoConfigHint, error) {\n\tvar v shared.AutoConfigHint\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a PollForDecisionTaskResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForDecisionTaskResponse struct could not be generated from the wire\n// representation.\nfunc (v *PollForDecisionTaskResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousStartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 51 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 65 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.BacklogCountHint = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.StickyExecutionEnabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.Query, err = _WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionInfo, err = _TransientDecisionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.EventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tv.BranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TMap:\n\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TotalHistoryBytes = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForDecisionTaskResponse\n// struct.\nfunc (v *PollForDecisionTaskResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [19]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousStartedEventId: %v\", *(v.PreviousStartedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventId: %v\", *(v.NextEventId))\n\t\ti++\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"BacklogCountHint: %v\", *(v.BacklogCountHint))\n\t\ti++\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyExecutionEnabled: %v\", *(v.StickyExecutionEnabled))\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", v.Query)\n\t\ti++\n\t}\n\tif v.DecisionInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionInfo: %v\", v.DecisionInfo)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionTaskList: %v\", v.WorkflowExecutionTaskList)\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventStoreVersion: %v\", *(v.EventStoreVersion))\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchToken: %v\", v.BranchToken)\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Queries: %v\", v.Queries)\n\t\ti++\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tfields[i] = fmt.Sprintf(\"TotalHistoryBytes: %v\", *(v.TotalHistoryBytes))\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoConfigHint: %v\", v.AutoConfigHint)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForDecisionTaskResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Map_String_WorkflowQuery_Equals(lhs, rhs map[string]*shared.WorkflowQuery) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this PollForDecisionTaskResponse match the\n// provided PollForDecisionTaskResponse.\n//\n// This function performs a deep comparison.\nfunc (v *PollForDecisionTaskResponse) Equals(rhs *PollForDecisionTaskResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousStartedEventId, rhs.PreviousStartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventId, rhs.NextEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.BacklogCountHint, rhs.BacklogCountHint) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.StickyExecutionEnabled, rhs.StickyExecutionEnabled) {\n\t\treturn false\n\t}\n\tif !((v.Query == nil && rhs.Query == nil) || (v.Query != nil && rhs.Query != nil && v.Query.Equals(rhs.Query))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionInfo == nil && rhs.DecisionInfo == nil) || (v.DecisionInfo != nil && rhs.DecisionInfo != nil && v.DecisionInfo.Equals(rhs.DecisionInfo))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionTaskList == nil && rhs.WorkflowExecutionTaskList == nil) || (v.WorkflowExecutionTaskList != nil && rhs.WorkflowExecutionTaskList != nil && v.WorkflowExecutionTaskList.Equals(rhs.WorkflowExecutionTaskList))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.EventStoreVersion, rhs.EventStoreVersion) {\n\t\treturn false\n\t}\n\tif !((v.BranchToken == nil && rhs.BranchToken == nil) || (v.BranchToken != nil && rhs.BranchToken != nil && bytes.Equal(v.BranchToken, rhs.BranchToken))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !((v.Queries == nil && rhs.Queries == nil) || (v.Queries != nil && rhs.Queries != nil && _Map_String_WorkflowQuery_Equals(v.Queries, rhs.Queries))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TotalHistoryBytes, rhs.TotalHistoryBytes) {\n\t\treturn false\n\t}\n\tif !((v.AutoConfigHint == nil && rhs.AutoConfigHint == nil) || (v.AutoConfigHint != nil && rhs.AutoConfigHint != nil && v.AutoConfigHint.Equals(rhs.AutoConfigHint))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_WorkflowQuery_Zapper map[string]*shared.WorkflowQuery\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WorkflowQuery_Zapper.\nfunc (m _Map_String_WorkflowQuery_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForDecisionTaskResponse.\nfunc (v *PollForDecisionTaskResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tenc.AddInt64(\"previousStartedEventId\", *v.PreviousStartedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.NextEventId != nil {\n\t\tenc.AddInt64(\"nextEventId\", *v.NextEventId)\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tenc.AddInt64(\"backlogCountHint\", *v.BacklogCountHint)\n\t}\n\tif v.StickyExecutionEnabled != nil {\n\t\tenc.AddBool(\"stickyExecutionEnabled\", *v.StickyExecutionEnabled)\n\t}\n\tif v.Query != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"query\", v.Query))\n\t}\n\tif v.DecisionInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionInfo\", v.DecisionInfo))\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"WorkflowExecutionTaskList\", v.WorkflowExecutionTaskList))\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tenc.AddInt32(\"eventStoreVersion\", *v.EventStoreVersion)\n\t}\n\tif v.BranchToken != nil {\n\t\tenc.AddString(\"branchToken\", base64.StdEncoding.EncodeToString(v.BranchToken))\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Queries != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queries\", (_Map_String_WorkflowQuery_Zapper)(v.Queries)))\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tenc.AddInt64(\"totalHistoryBytes\", *v.TotalHistoryBytes)\n\t}\n\tif v.AutoConfigHint != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"autoConfigHint\", v.AutoConfigHint))\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowExecution() (o *shared.WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowType() (o *shared.WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetPreviousStartedEventId returns the value of PreviousStartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetPreviousStartedEventId() (o int64) {\n\tif v != nil && v.PreviousStartedEventId != nil {\n\t\treturn *v.PreviousStartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetPreviousStartedEventId returns true if PreviousStartedEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetPreviousStartedEventId() bool {\n\treturn v != nil && v.PreviousStartedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetNextEventId returns the value of NextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetNextEventId() (o int64) {\n\tif v != nil && v.NextEventId != nil {\n\t\treturn *v.NextEventId\n\t}\n\n\treturn\n}\n\n// IsSetNextEventId returns true if NextEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetNextEventId() bool {\n\treturn v != nil && v.NextEventId != nil\n}\n\n// GetBacklogCountHint returns the value of BacklogCountHint if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetBacklogCountHint() (o int64) {\n\tif v != nil && v.BacklogCountHint != nil {\n\t\treturn *v.BacklogCountHint\n\t}\n\n\treturn\n}\n\n// IsSetBacklogCountHint returns true if BacklogCountHint is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetBacklogCountHint() bool {\n\treturn v != nil && v.BacklogCountHint != nil\n}\n\n// GetStickyExecutionEnabled returns the value of StickyExecutionEnabled if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetStickyExecutionEnabled() (o bool) {\n\tif v != nil && v.StickyExecutionEnabled != nil {\n\t\treturn *v.StickyExecutionEnabled\n\t}\n\n\treturn\n}\n\n// IsSetStickyExecutionEnabled returns true if StickyExecutionEnabled is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetStickyExecutionEnabled() bool {\n\treturn v != nil && v.StickyExecutionEnabled != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetQuery() (o *shared.WorkflowQuery) {\n\tif v != nil && v.Query != nil {\n\t\treturn v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\n// GetDecisionInfo returns the value of DecisionInfo if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetDecisionInfo() (o *shared.TransientDecisionInfo) {\n\tif v != nil && v.DecisionInfo != nil {\n\t\treturn v.DecisionInfo\n\t}\n\n\treturn\n}\n\n// IsSetDecisionInfo returns true if DecisionInfo is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetDecisionInfo() bool {\n\treturn v != nil && v.DecisionInfo != nil\n}\n\n// GetWorkflowExecutionTaskList returns the value of WorkflowExecutionTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowExecutionTaskList() (o *shared.TaskList) {\n\tif v != nil && v.WorkflowExecutionTaskList != nil {\n\t\treturn v.WorkflowExecutionTaskList\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionTaskList returns true if WorkflowExecutionTaskList is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowExecutionTaskList() bool {\n\treturn v != nil && v.WorkflowExecutionTaskList != nil\n}\n\n// GetEventStoreVersion returns the value of EventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetEventStoreVersion() (o int32) {\n\tif v != nil && v.EventStoreVersion != nil {\n\t\treturn *v.EventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetEventStoreVersion returns true if EventStoreVersion is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetEventStoreVersion() bool {\n\treturn v != nil && v.EventStoreVersion != nil\n}\n\n// GetBranchToken returns the value of BranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetBranchToken() (o []byte) {\n\tif v != nil && v.BranchToken != nil {\n\t\treturn v.BranchToken\n\t}\n\n\treturn\n}\n\n// IsSetBranchToken returns true if BranchToken is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetBranchToken() bool {\n\treturn v != nil && v.BranchToken != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetQueries returns the value of Queries if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetQueries() (o map[string]*shared.WorkflowQuery) {\n\tif v != nil && v.Queries != nil {\n\t\treturn v.Queries\n\t}\n\n\treturn\n}\n\n// IsSetQueries returns true if Queries is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetQueries() bool {\n\treturn v != nil && v.Queries != nil\n}\n\n// GetTotalHistoryBytes returns the value of TotalHistoryBytes if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetTotalHistoryBytes() (o int64) {\n\tif v != nil && v.TotalHistoryBytes != nil {\n\t\treturn *v.TotalHistoryBytes\n\t}\n\n\treturn\n}\n\n// IsSetTotalHistoryBytes returns true if TotalHistoryBytes is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetTotalHistoryBytes() bool {\n\treturn v != nil && v.TotalHistoryBytes != nil\n}\n\n// GetAutoConfigHint returns the value of AutoConfigHint if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetAutoConfigHint() (o *shared.AutoConfigHint) {\n\tif v != nil && v.AutoConfigHint != nil {\n\t\treturn v.AutoConfigHint\n\t}\n\n\treturn\n}\n\n// IsSetAutoConfigHint returns true if AutoConfigHint is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetAutoConfigHint() bool {\n\treturn v != nil && v.AutoConfigHint != nil\n}\n\ntype QueryWorkflowRequest struct {\n\tDomainUUID    *string                      `json:\"domainUUID,omitempty\"`\n\tTaskList      *shared.TaskList             `json:\"taskList,omitempty\"`\n\tQueryRequest  *shared.QueryWorkflowRequest `json:\"queryRequest,omitempty\"`\n\tForwardedFrom *string                      `json:\"forwardedFrom,omitempty\"`\n}\n\n// ToWire translates a QueryWorkflowRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryWorkflowRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.QueryRequest != nil {\n\t\tw, err = v.QueryRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tw, err = wire.NewValueString(*(v.ForwardedFrom)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowRequest_Read(w wire.Value) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a QueryWorkflowRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryWorkflowRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryWorkflowRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryWorkflowRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ForwardedFrom = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryWorkflowRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be encoded.\nfunc (v *QueryWorkflowRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForwardedFrom != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ForwardedFrom)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowRequest_Decode(sr stream.Reader) (*shared.QueryWorkflowRequest, error) {\n\tvar v shared.QueryWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a QueryWorkflowRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be generated from the wire\n// representation.\nfunc (v *QueryWorkflowRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ForwardedFrom = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryWorkflowRequest\n// struct.\nfunc (v *QueryWorkflowRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.QueryRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRequest: %v\", v.QueryRequest)\n\t\ti++\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForwardedFrom: %v\", *(v.ForwardedFrom))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryWorkflowRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this QueryWorkflowRequest match the\n// provided QueryWorkflowRequest.\n//\n// This function performs a deep comparison.\nfunc (v *QueryWorkflowRequest) Equals(rhs *QueryWorkflowRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.QueryRequest == nil && rhs.QueryRequest == nil) || (v.QueryRequest != nil && rhs.QueryRequest != nil && v.QueryRequest.Equals(rhs.QueryRequest))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ForwardedFrom, rhs.ForwardedFrom) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryWorkflowRequest.\nfunc (v *QueryWorkflowRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.QueryRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRequest\", v.QueryRequest))\n\t}\n\tif v.ForwardedFrom != nil {\n\t\tenc.AddString(\"forwardedFrom\", *v.ForwardedFrom)\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *QueryWorkflowRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *QueryWorkflowRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetQueryRequest returns the value of QueryRequest if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetQueryRequest() (o *shared.QueryWorkflowRequest) {\n\tif v != nil && v.QueryRequest != nil {\n\t\treturn v.QueryRequest\n\t}\n\n\treturn\n}\n\n// IsSetQueryRequest returns true if QueryRequest is not nil.\nfunc (v *QueryWorkflowRequest) IsSetQueryRequest() bool {\n\treturn v != nil && v.QueryRequest != nil\n}\n\n// GetForwardedFrom returns the value of ForwardedFrom if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetForwardedFrom() (o string) {\n\tif v != nil && v.ForwardedFrom != nil {\n\t\treturn *v.ForwardedFrom\n\t}\n\n\treturn\n}\n\n// IsSetForwardedFrom returns true if ForwardedFrom is not nil.\nfunc (v *QueryWorkflowRequest) IsSetForwardedFrom() bool {\n\treturn v != nil && v.ForwardedFrom != nil\n}\n\ntype RespondQueryTaskCompletedRequest struct {\n\tDomainUUID       *string                                  `json:\"domainUUID,omitempty\"`\n\tTaskList         *shared.TaskList                         `json:\"taskList,omitempty\"`\n\tTaskID           *string                                  `json:\"taskID,omitempty\"`\n\tCompletedRequest *shared.RespondQueryTaskCompletedRequest `json:\"completedRequest,omitempty\"`\n}\n\n// ToWire translates a RespondQueryTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondQueryTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainUUID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainUUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CompletedRequest != nil {\n\t\tw, err = v.CompletedRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondQueryTaskCompletedRequest_Read(w wire.Value) (*shared.RespondQueryTaskCompletedRequest, error) {\n\tvar v shared.RespondQueryTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondQueryTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondQueryTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondQueryTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondQueryTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainUUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompletedRequest, err = _RespondQueryTaskCompletedRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondQueryTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondQueryTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondQueryTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainUUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainUUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletedRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletedRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondQueryTaskCompletedRequest_Decode(sr stream.Reader) (*shared.RespondQueryTaskCompletedRequest, error) {\n\tvar v shared.RespondQueryTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondQueryTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondQueryTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondQueryTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainUUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.CompletedRequest, err = _RespondQueryTaskCompletedRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondQueryTaskCompletedRequest\n// struct.\nfunc (v *RespondQueryTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainUUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainUUID: %v\", *(v.DomainUUID))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.CompletedRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletedRequest: %v\", v.CompletedRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondQueryTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondQueryTaskCompletedRequest match the\n// provided RespondQueryTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondQueryTaskCompletedRequest) Equals(rhs *RespondQueryTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainUUID, rhs.DomainUUID) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !((v.CompletedRequest == nil && rhs.CompletedRequest == nil) || (v.CompletedRequest != nil && rhs.CompletedRequest != nil && v.CompletedRequest.Equals(rhs.CompletedRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondQueryTaskCompletedRequest.\nfunc (v *RespondQueryTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainUUID != nil {\n\t\tenc.AddString(\"domainUUID\", *v.DomainUUID)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddString(\"taskID\", *v.TaskID)\n\t}\n\tif v.CompletedRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completedRequest\", v.CompletedRequest))\n\t}\n\treturn err\n}\n\n// GetDomainUUID returns the value of DomainUUID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil && v.DomainUUID != nil {\n\t\treturn *v.DomainUUID\n\t}\n\n\treturn\n}\n\n// IsSetDomainUUID returns true if DomainUUID is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetDomainUUID() bool {\n\treturn v != nil && v.DomainUUID != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetTaskList() (o *shared.TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetTaskID() (o string) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetCompletedRequest returns the value of CompletedRequest if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetCompletedRequest() (o *shared.RespondQueryTaskCompletedRequest) {\n\tif v != nil && v.CompletedRequest != nil {\n\t\treturn v.CompletedRequest\n\t}\n\n\treturn\n}\n\n// IsSetCompletedRequest returns true if CompletedRequest is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetCompletedRequest() bool {\n\treturn v != nil && v.CompletedRequest != nil\n}\n\ntype TaskSource int32\n\nconst (\n\tTaskSourceHistory   TaskSource = 0\n\tTaskSourceDbBacklog TaskSource = 1\n)\n\n// TaskSource_Values returns all recognized values of TaskSource.\nfunc TaskSource_Values() []TaskSource {\n\treturn []TaskSource{\n\t\tTaskSourceHistory,\n\t\tTaskSourceDbBacklog,\n\t}\n}\n\n// UnmarshalText tries to decode TaskSource from a byte slice\n// containing its name.\n//\n//\tvar v TaskSource\n//\terr := v.UnmarshalText([]byte(\"HISTORY\"))\nfunc (v *TaskSource) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"HISTORY\":\n\t\t*v = TaskSourceHistory\n\t\treturn nil\n\tcase \"DB_BACKLOG\":\n\t\t*v = TaskSourceDbBacklog\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskSource\", err)\n\t\t}\n\t\t*v = TaskSource(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskSource to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v TaskSource) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"HISTORY\"), nil\n\tcase 1:\n\t\treturn []byte(\"DB_BACKLOG\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskSource.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v TaskSource) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"HISTORY\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"DB_BACKLOG\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v TaskSource) Ptr() *TaskSource {\n\treturn &v\n}\n\n// Encode encodes TaskSource directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v TaskSource\n//\treturn v.Encode(sWriter)\nfunc (v TaskSource) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates TaskSource into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v TaskSource) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes TaskSource from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return TaskSource(0), err\n//\t}\n//\n//\tvar v TaskSource\n//\tif err := v.FromWire(x); err != nil {\n//\t  return TaskSource(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskSource) FromWire(w wire.Value) error {\n\t*v = (TaskSource)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded TaskSource directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v TaskSource\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return TaskSource(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskSource) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (TaskSource)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of TaskSource.\nfunc (v TaskSource) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"HISTORY\"\n\tcase 1:\n\t\treturn \"DB_BACKLOG\"\n\t}\n\treturn fmt.Sprintf(\"TaskSource(%d)\", w)\n}\n\n// Equals returns true if this TaskSource value matches the provided\n// value.\nfunc (v TaskSource) Equals(rhs TaskSource) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes TaskSource into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v TaskSource) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"HISTORY\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"DB_BACKLOG\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode TaskSource from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *TaskSource) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"TaskSource\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"TaskSource\")\n\t\t}\n\t\t*v = (TaskSource)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"TaskSource\")\n\t}\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"matching\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/matching\",\n\tFilePath: \"matching.thrift\",\n\tSHA1:     \"e7db050ae64980b33e3f8ed3f6a3bba3d63174b3\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\ninclude \\\"shared.thrift\\\"\\n\\nnamespace java com.uber.cadence.matching\\n\\n// TaskSource is the source from which a task was produced\\nenum TaskSource {\\n    HISTORY,    // Task produced by history service\\n    DB_BACKLOG // Task produced from matching db backlog\\n}\\n\\nstruct PollForDecisionTaskRequest {\\n  10: optional string domainUUID\\n  15: optional string pollerID\\n  20: optional shared.PollForDecisionTaskRequest pollRequest\\n  30: optional string forwardedFrom\\n  40: optional string isolationGroup\\n}\\n\\nstruct PollForDecisionTaskResponse {\\n  10: optional binary taskToken\\n  20: optional shared.WorkflowExecution workflowExecution\\n  30: optional shared.WorkflowType workflowType\\n  40: optional i64 (js.type = \\\"Long\\\") previousStartedEventId\\n  50: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  51: optional i64 (js.type = \\\"Long\\\") attempt\\n  60: optional i64 (js.type = \\\"Long\\\") nextEventId\\n  65: optional i64 (js.type = \\\"Long\\\") backlogCountHint\\n  70: optional bool stickyExecutionEnabled\\n  80: optional shared.WorkflowQuery query\\n  90: optional shared.TransientDecisionInfo decisionInfo\\n  100: optional shared.TaskList WorkflowExecutionTaskList\\n  110: optional i32 eventStoreVersion\\n  120: optional binary branchToken\\n  130: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  140: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  150: optional map<string, shared.WorkflowQuery> queries\\n  160: optional i64 (js.type = \\\"Long\\\") totalHistoryBytes\\n  170: optional shared.AutoConfigHint autoConfigHint\\n}\\n\\nstruct PollForActivityTaskRequest {\\n  10: optional string domainUUID\\n  15: optional string pollerID\\n  20: optional shared.PollForActivityTaskRequest pollRequest\\n  30: optional string forwardedFrom\\n  40: optional string isolationGroup\\n}\\n\\nstruct AddDecisionTaskRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n  30: optional shared.TaskList taskList\\n  40: optional i64 (js.type = \\\"Long\\\") scheduleId\\n  50: optional i32 scheduleToStartTimeoutSeconds\\n  59: optional TaskSource source\\n  60: optional string forwardedFrom\\n  70: optional map<string, string> partitionConfig\\n}\\n\\nstruct AddActivityTaskRequest {\\n  10: optional string domainUUID\\n  20: optional shared.WorkflowExecution execution\\n  30: optional string sourceDomainUUID\\n  40: optional shared.TaskList taskList\\n  50: optional i64 (js.type = \\\"Long\\\") scheduleId\\n  60: optional i32 scheduleToStartTimeoutSeconds\\n  69: optional TaskSource source\\n  70: optional string forwardedFrom\\n  80: optional ActivityTaskDispatchInfo activityTaskDispatchInfo\\n  90: optional map<string, string> partitionConfig\\n}\\n\\nstruct ActivityTaskDispatchInfo {\\n   10: optional shared.HistoryEvent scheduledEvent\\n   20: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n   30: optional i64 (js.type = \\\"Long\\\") attempt\\n   40: optional i64 (js.type = \\\"Long\\\") scheduledTimestampOfThisAttempt\\n   50: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n   60: optional binary heartbeatDetails\\n   70: optional shared.WorkflowType workflowType\\n   80: optional string workflowDomain\\n}\\n\\nstruct QueryWorkflowRequest {\\n  10: optional string domainUUID\\n  20: optional shared.TaskList taskList\\n  30: optional shared.QueryWorkflowRequest queryRequest\\n  40: optional string forwardedFrom\\n}\\n\\nstruct RespondQueryTaskCompletedRequest {\\n  10: optional string domainUUID\\n  20: optional shared.TaskList taskList\\n  30: optional string taskID\\n  40: optional shared.RespondQueryTaskCompletedRequest completedRequest\\n}\\n\\nstruct CancelOutstandingPollRequest {\\n  10: optional string domainUUID\\n  20: optional i32 taskListType\\n  30: optional shared.TaskList taskList\\n  40: optional string pollerID\\n}\\n\\nstruct DescribeTaskListRequest {\\n  10: optional string domainUUID\\n  20: optional shared.DescribeTaskListRequest descRequest\\n}\\n\\nstruct ListTaskListPartitionsRequest {\\n  10: optional string domain\\n  20: optional shared.TaskList taskList\\n}\\n\\n/**\\n* MatchingService API is exposed to provide support for polling from long running applications.\\n* Such applications are expected to have a worker which regularly polls for DecisionTask and ActivityTask.  For each\\n* DecisionTask, application is expected to process the history of events for that session and respond back with next\\n* decisions.  For each ActivityTask, application is expected to execute the actual logic for that task and respond back\\n* with completion or failure.\\n**/\\nservice MatchingService {\\n  /**\\n  * PollForDecisionTask is called by frontend to process DecisionTask from a specific taskList.  A\\n  * DecisionTask is dispatched to callers for active workflow executions, with pending decisions.\\n  **/\\n  PollForDecisionTaskResponse PollForDecisionTask(1: PollForDecisionTaskRequest pollRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * PollForActivityTask is called by frontend to process ActivityTask from a specific taskList.  ActivityTask\\n  * is dispatched to callers whenever a ScheduleTask decision is made for a workflow execution.\\n  **/\\n  shared.PollForActivityTaskResponse PollForActivityTask(1: PollForActivityTaskRequest pollRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.LimitExceededError limitExceededError,\\n      4: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n  * AddDecisionTask is called by the history service when a decision task is scheduled, so that it can be dispatched\\n  * by the MatchingEngine.\\n  **/\\n  void AddDecisionTask(1: AddDecisionTaskRequest addRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.RemoteSyncMatchedError remoteSyncMatchedError,\\n      7: shared.StickyWorkerUnavailableError stickyWorkerUnavailableError,\\n      8: shared.TaskListNotOwnedByHostError taskListNotOwnedByHostError,\\n    )\\n\\n  /**\\n  * AddActivityTask is called by the history service when a decision task is scheduled, so that it can be dispatched\\n  * by the MatchingEngine.\\n  **/\\n  void AddActivityTask(1: AddActivityTaskRequest addRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.DomainNotActiveError domainNotActiveError,\\n      6: shared.RemoteSyncMatchedError remoteSyncMatchedError,\\n      7: shared.TaskListNotOwnedByHostError taskListNotOwnedByHostError,\\n    )\\n\\n  /**\\n  * QueryWorkflow is called by frontend to query a workflow.\\n  **/\\n  shared.QueryWorkflowResponse QueryWorkflow(1: QueryWorkflowRequest queryRequest)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.QueryFailedError queryFailedError,\\n      5: shared.LimitExceededError limitExceededError,\\n      6: shared.ServiceBusyError serviceBusyError,\\n      7: shared.StickyWorkerUnavailableError stickyWorkerUnavailableError,\\n      8: shared.TaskListNotOwnedByHostError taskListNotOwnedByHostError,\\n    )\\n\\n  /**\\n  * RespondQueryTaskCompleted is called by frontend to respond query completed.\\n  **/\\n  void RespondQueryTaskCompleted(1: RespondQueryTaskCompletedRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.EntityNotExistsError entityNotExistError,\\n      4: shared.LimitExceededError limitExceededError,\\n      5: shared.ServiceBusyError serviceBusyError,\\n    )\\n\\n  /**\\n    * CancelOutstandingPoll is called by frontend to unblock long polls on matching for zombie pollers.\\n    * Our rpc stack does not support context propagation, so when a client connection goes away frontend sees\\n    * cancellation of context for that handler, but any corresponding calls (long-poll) to matching service does not\\n    * see the cancellation propagated so it can unblock corresponding long-polls on its end.  This results is tasks\\n    * being dispatched to zombie pollers in this situation.  This API is added so everytime frontend makes a long-poll\\n    * api call to matching it passes in a pollerID and then calls this API when it detects client connection is closed\\n    * to unblock long polls for this poller and prevent tasks being sent to these zombie pollers.\\n    **/\\n  void CancelOutstandingPoll(1: CancelOutstandingPollRequest request)\\n    throws (\\n      1: shared.BadRequestError badRequestError,\\n      2: shared.InternalServiceError internalServiceError,\\n      3: shared.ServiceBusyError serviceBusyError,\\n      4: shared.TaskListNotOwnedByHostError taskListNotOwnedByHostError,\\n    )\\n\\n  /**\\n  * DescribeTaskList returns information about the target tasklist, right now this API returns the\\n  * pollers which polled this tasklist in last few minutes.\\n  **/\\n  shared.DescribeTaskListResponse DescribeTaskList(1: DescribeTaskListRequest request)\\n    throws (\\n        1: shared.BadRequestError badRequestError,\\n        2: shared.InternalServiceError internalServiceError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n        5: shared.TaskListNotOwnedByHostError taskListNotOwnedByHostError,\\n      )\\n\\n  /**\\n  * GetTaskListsByDomain returns the list of all the task lists for a domainName.\\n  **/\\n  shared.GetTaskListsByDomainResponse GetTaskListsByDomain(1: shared.GetTaskListsByDomainRequest request)\\n    throws (\\n        1: shared.BadRequestError badRequestError,\\n        2: shared.InternalServiceError internalServiceError,\\n        3: shared.EntityNotExistsError entityNotExistError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n      )\\n\\n  /**\\n  * ListTaskListPartitions returns a map of partitionKey and hostAddress for a taskList\\n  **/\\n  shared.ListTaskListPartitionsResponse ListTaskListPartitions(1: ListTaskListPartitionsRequest request)\\n    throws (\\n        1: shared.BadRequestError badRequestError,\\n        2: shared.InternalServiceError internalServiceError,\\n        4: shared.ServiceBusyError serviceBusyError,\\n    )\\n}\\n\"\n\n// MatchingService_AddActivityTask_Args represents the arguments for the MatchingService.AddActivityTask function.\n//\n// The arguments for AddActivityTask are sent and received over the wire as this struct.\ntype MatchingService_AddActivityTask_Args struct {\n\tAddRequest *AddActivityTaskRequest `json:\"addRequest,omitempty\"`\n}\n\n// ToWire translates a MatchingService_AddActivityTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_AddActivityTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.AddRequest != nil {\n\t\tw, err = v.AddRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AddActivityTaskRequest_Read(w wire.Value) (*AddActivityTaskRequest, error) {\n\tvar v AddActivityTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_AddActivityTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_AddActivityTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_AddActivityTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_AddActivityTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AddRequest, err = _AddActivityTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_AddActivityTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_AddActivityTask_Args struct could not be encoded.\nfunc (v *MatchingService_AddActivityTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.AddRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AddRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AddActivityTaskRequest_Decode(sr stream.Reader) (*AddActivityTaskRequest, error) {\n\tvar v AddActivityTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_AddActivityTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_AddActivityTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_AddActivityTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.AddRequest, err = _AddActivityTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_AddActivityTask_Args\n// struct.\nfunc (v *MatchingService_AddActivityTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.AddRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"AddRequest: %v\", v.AddRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_AddActivityTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_AddActivityTask_Args match the\n// provided MatchingService_AddActivityTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_AddActivityTask_Args) Equals(rhs *MatchingService_AddActivityTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.AddRequest == nil && rhs.AddRequest == nil) || (v.AddRequest != nil && rhs.AddRequest != nil && v.AddRequest.Equals(rhs.AddRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_AddActivityTask_Args.\nfunc (v *MatchingService_AddActivityTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.AddRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"addRequest\", v.AddRequest))\n\t}\n\treturn err\n}\n\n// GetAddRequest returns the value of AddRequest if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Args) GetAddRequest() (o *AddActivityTaskRequest) {\n\tif v != nil && v.AddRequest != nil {\n\t\treturn v.AddRequest\n\t}\n\n\treturn\n}\n\n// IsSetAddRequest returns true if AddRequest is not nil.\nfunc (v *MatchingService_AddActivityTask_Args) IsSetAddRequest() bool {\n\treturn v != nil && v.AddRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"AddActivityTask\" for this struct.\nfunc (v *MatchingService_AddActivityTask_Args) MethodName() string {\n\treturn \"AddActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_AddActivityTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_AddActivityTask_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.AddActivityTask\n// function.\nvar MatchingService_AddActivityTask_Helper = struct {\n\t// Args accepts the parameters of AddActivityTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\taddRequest *AddActivityTaskRequest,\n\t) *MatchingService_AddActivityTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by AddActivityTask.\n\t//\n\t// An error can be thrown by AddActivityTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for AddActivityTask\n\t// given the error returned by it. The provided error may\n\t// be nil if AddActivityTask did not fail.\n\t//\n\t// This allows mapping errors returned by AddActivityTask into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// AddActivityTask\n\t//\n\t//   err := AddActivityTask(args)\n\t//   result, err := MatchingService_AddActivityTask_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from AddActivityTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*MatchingService_AddActivityTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for AddActivityTask\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if AddActivityTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := MatchingService_AddActivityTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_AddActivityTask_Result) error\n}{}\n\nfunc init() {\n\tMatchingService_AddActivityTask_Helper.Args = func(\n\t\taddRequest *AddActivityTaskRequest,\n\t) *MatchingService_AddActivityTask_Args {\n\t\treturn &MatchingService_AddActivityTask_Args{\n\t\t\tAddRequest: addRequest,\n\t\t}\n\t}\n\n\tMatchingService_AddActivityTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.RemoteSyncMatchedError:\n\t\t\treturn true\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_AddActivityTask_Helper.WrapResponse = func(err error) (*MatchingService_AddActivityTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_AddActivityTask_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.RemoteSyncMatchedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.RemoteSyncMatchedError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{RemoteSyncMatchedError: e}, nil\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddActivityTask_Result.TaskListNotOwnedByHostError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddActivityTask_Result{TaskListNotOwnedByHostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_AddActivityTask_Helper.UnwrapResponse = func(result *MatchingService_AddActivityTask_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.RemoteSyncMatchedError != nil {\n\t\t\terr = result.RemoteSyncMatchedError\n\t\t\treturn\n\t\t}\n\t\tif result.TaskListNotOwnedByHostError != nil {\n\t\t\terr = result.TaskListNotOwnedByHostError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// MatchingService_AddActivityTask_Result represents the result of a MatchingService.AddActivityTask function call.\n//\n// The result of a AddActivityTask execution is sent and received over the wire as this struct.\ntype MatchingService_AddActivityTask_Result struct {\n\tBadRequestError             *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError        *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError            *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError          *shared.LimitExceededError          `json:\"limitExceededError,omitempty\"`\n\tDomainNotActiveError        *shared.DomainNotActiveError        `json:\"domainNotActiveError,omitempty\"`\n\tRemoteSyncMatchedError      *shared.RemoteSyncMatchedError      `json:\"remoteSyncMatchedError,omitempty\"`\n\tTaskListNotOwnedByHostError *shared.TaskListNotOwnedByHostError `json:\"taskListNotOwnedByHostError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_AddActivityTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_AddActivityTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tw, err = v.RemoteSyncMatchedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tw, err = v.TaskListNotOwnedByHostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_AddActivityTask_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _BadRequestError_Read(w wire.Value) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Read(w wire.Value) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Read(w wire.Value) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Read(w wire.Value) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainNotActiveError_Read(w wire.Value) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RemoteSyncMatchedError_Read(w wire.Value) (*shared.RemoteSyncMatchedError, error) {\n\tvar v shared.RemoteSyncMatchedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TaskListNotOwnedByHostError_Read(w wire.Value) (*shared.TaskListNotOwnedByHostError, error) {\n\tvar v shared.TaskListNotOwnedByHostError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_AddActivityTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_AddActivityTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_AddActivityTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_AddActivityTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RemoteSyncMatchedError, err = _RemoteSyncMatchedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddActivityTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_AddActivityTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_AddActivityTask_Result struct could not be encoded.\nfunc (v *MatchingService_AddActivityTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RemoteSyncMatchedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RemoteSyncMatchedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListNotOwnedByHostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddActivityTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _BadRequestError_Decode(sr stream.Reader) (*shared.BadRequestError, error) {\n\tvar v shared.BadRequestError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _InternalServiceError_Decode(sr stream.Reader) (*shared.InternalServiceError, error) {\n\tvar v shared.InternalServiceError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ServiceBusyError_Decode(sr stream.Reader) (*shared.ServiceBusyError, error) {\n\tvar v shared.ServiceBusyError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _LimitExceededError_Decode(sr stream.Reader) (*shared.LimitExceededError, error) {\n\tvar v shared.LimitExceededError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainNotActiveError_Decode(sr stream.Reader) (*shared.DomainNotActiveError, error) {\n\tvar v shared.DomainNotActiveError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RemoteSyncMatchedError_Decode(sr stream.Reader) (*shared.RemoteSyncMatchedError, error) {\n\tvar v shared.RemoteSyncMatchedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TaskListNotOwnedByHostError_Decode(sr stream.Reader) (*shared.TaskListNotOwnedByHostError, error) {\n\tvar v shared.TaskListNotOwnedByHostError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_AddActivityTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_AddActivityTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_AddActivityTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.RemoteSyncMatchedError, err = _RemoteSyncMatchedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddActivityTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_AddActivityTask_Result\n// struct.\nfunc (v *MatchingService_AddActivityTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"RemoteSyncMatchedError: %v\", v.RemoteSyncMatchedError)\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListNotOwnedByHostError: %v\", v.TaskListNotOwnedByHostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_AddActivityTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_AddActivityTask_Result match the\n// provided MatchingService_AddActivityTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_AddActivityTask_Result) Equals(rhs *MatchingService_AddActivityTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.RemoteSyncMatchedError == nil && rhs.RemoteSyncMatchedError == nil) || (v.RemoteSyncMatchedError != nil && rhs.RemoteSyncMatchedError != nil && v.RemoteSyncMatchedError.Equals(rhs.RemoteSyncMatchedError))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListNotOwnedByHostError == nil && rhs.TaskListNotOwnedByHostError == nil) || (v.TaskListNotOwnedByHostError != nil && rhs.TaskListNotOwnedByHostError != nil && v.TaskListNotOwnedByHostError.Equals(rhs.TaskListNotOwnedByHostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_AddActivityTask_Result.\nfunc (v *MatchingService_AddActivityTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"remoteSyncMatchedError\", v.RemoteSyncMatchedError))\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListNotOwnedByHostError\", v.TaskListNotOwnedByHostError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetRemoteSyncMatchedError returns the value of RemoteSyncMatchedError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetRemoteSyncMatchedError() (o *shared.RemoteSyncMatchedError) {\n\tif v != nil && v.RemoteSyncMatchedError != nil {\n\t\treturn v.RemoteSyncMatchedError\n\t}\n\n\treturn\n}\n\n// IsSetRemoteSyncMatchedError returns true if RemoteSyncMatchedError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetRemoteSyncMatchedError() bool {\n\treturn v != nil && v.RemoteSyncMatchedError != nil\n}\n\n// GetTaskListNotOwnedByHostError returns the value of TaskListNotOwnedByHostError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddActivityTask_Result) GetTaskListNotOwnedByHostError() (o *shared.TaskListNotOwnedByHostError) {\n\tif v != nil && v.TaskListNotOwnedByHostError != nil {\n\t\treturn v.TaskListNotOwnedByHostError\n\t}\n\n\treturn\n}\n\n// IsSetTaskListNotOwnedByHostError returns true if TaskListNotOwnedByHostError is not nil.\nfunc (v *MatchingService_AddActivityTask_Result) IsSetTaskListNotOwnedByHostError() bool {\n\treturn v != nil && v.TaskListNotOwnedByHostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"AddActivityTask\" for this struct.\nfunc (v *MatchingService_AddActivityTask_Result) MethodName() string {\n\treturn \"AddActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_AddActivityTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_AddDecisionTask_Args represents the arguments for the MatchingService.AddDecisionTask function.\n//\n// The arguments for AddDecisionTask are sent and received over the wire as this struct.\ntype MatchingService_AddDecisionTask_Args struct {\n\tAddRequest *AddDecisionTaskRequest `json:\"addRequest,omitempty\"`\n}\n\n// ToWire translates a MatchingService_AddDecisionTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_AddDecisionTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.AddRequest != nil {\n\t\tw, err = v.AddRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AddDecisionTaskRequest_Read(w wire.Value) (*AddDecisionTaskRequest, error) {\n\tvar v AddDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_AddDecisionTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_AddDecisionTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_AddDecisionTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_AddDecisionTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AddRequest, err = _AddDecisionTaskRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_AddDecisionTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_AddDecisionTask_Args struct could not be encoded.\nfunc (v *MatchingService_AddDecisionTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.AddRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AddRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AddDecisionTaskRequest_Decode(sr stream.Reader) (*AddDecisionTaskRequest, error) {\n\tvar v AddDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_AddDecisionTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_AddDecisionTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_AddDecisionTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.AddRequest, err = _AddDecisionTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_AddDecisionTask_Args\n// struct.\nfunc (v *MatchingService_AddDecisionTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.AddRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"AddRequest: %v\", v.AddRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_AddDecisionTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_AddDecisionTask_Args match the\n// provided MatchingService_AddDecisionTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_AddDecisionTask_Args) Equals(rhs *MatchingService_AddDecisionTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.AddRequest == nil && rhs.AddRequest == nil) || (v.AddRequest != nil && rhs.AddRequest != nil && v.AddRequest.Equals(rhs.AddRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_AddDecisionTask_Args.\nfunc (v *MatchingService_AddDecisionTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.AddRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"addRequest\", v.AddRequest))\n\t}\n\treturn err\n}\n\n// GetAddRequest returns the value of AddRequest if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Args) GetAddRequest() (o *AddDecisionTaskRequest) {\n\tif v != nil && v.AddRequest != nil {\n\t\treturn v.AddRequest\n\t}\n\n\treturn\n}\n\n// IsSetAddRequest returns true if AddRequest is not nil.\nfunc (v *MatchingService_AddDecisionTask_Args) IsSetAddRequest() bool {\n\treturn v != nil && v.AddRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"AddDecisionTask\" for this struct.\nfunc (v *MatchingService_AddDecisionTask_Args) MethodName() string {\n\treturn \"AddDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_AddDecisionTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_AddDecisionTask_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.AddDecisionTask\n// function.\nvar MatchingService_AddDecisionTask_Helper = struct {\n\t// Args accepts the parameters of AddDecisionTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\taddRequest *AddDecisionTaskRequest,\n\t) *MatchingService_AddDecisionTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by AddDecisionTask.\n\t//\n\t// An error can be thrown by AddDecisionTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for AddDecisionTask\n\t// given the error returned by it. The provided error may\n\t// be nil if AddDecisionTask did not fail.\n\t//\n\t// This allows mapping errors returned by AddDecisionTask into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// AddDecisionTask\n\t//\n\t//   err := AddDecisionTask(args)\n\t//   result, err := MatchingService_AddDecisionTask_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from AddDecisionTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*MatchingService_AddDecisionTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for AddDecisionTask\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if AddDecisionTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := MatchingService_AddDecisionTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_AddDecisionTask_Result) error\n}{}\n\nfunc init() {\n\tMatchingService_AddDecisionTask_Helper.Args = func(\n\t\taddRequest *AddDecisionTaskRequest,\n\t) *MatchingService_AddDecisionTask_Args {\n\t\treturn &MatchingService_AddDecisionTask_Args{\n\t\t\tAddRequest: addRequest,\n\t\t}\n\t}\n\n\tMatchingService_AddDecisionTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.DomainNotActiveError:\n\t\t\treturn true\n\t\tcase *shared.RemoteSyncMatchedError:\n\t\t\treturn true\n\t\tcase *shared.StickyWorkerUnavailableError:\n\t\t\treturn true\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_AddDecisionTask_Helper.WrapResponse = func(err error) (*MatchingService_AddDecisionTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_AddDecisionTask_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.DomainNotActiveError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{DomainNotActiveError: e}, nil\n\t\tcase *shared.RemoteSyncMatchedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.RemoteSyncMatchedError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{RemoteSyncMatchedError: e}, nil\n\t\tcase *shared.StickyWorkerUnavailableError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.StickyWorkerUnavailableError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{StickyWorkerUnavailableError: e}, nil\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_AddDecisionTask_Result.TaskListNotOwnedByHostError\")\n\t\t\t}\n\t\t\treturn &MatchingService_AddDecisionTask_Result{TaskListNotOwnedByHostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_AddDecisionTask_Helper.UnwrapResponse = func(result *MatchingService_AddDecisionTask_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.DomainNotActiveError != nil {\n\t\t\terr = result.DomainNotActiveError\n\t\t\treturn\n\t\t}\n\t\tif result.RemoteSyncMatchedError != nil {\n\t\t\terr = result.RemoteSyncMatchedError\n\t\t\treturn\n\t\t}\n\t\tif result.StickyWorkerUnavailableError != nil {\n\t\t\terr = result.StickyWorkerUnavailableError\n\t\t\treturn\n\t\t}\n\t\tif result.TaskListNotOwnedByHostError != nil {\n\t\t\terr = result.TaskListNotOwnedByHostError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// MatchingService_AddDecisionTask_Result represents the result of a MatchingService.AddDecisionTask function call.\n//\n// The result of a AddDecisionTask execution is sent and received over the wire as this struct.\ntype MatchingService_AddDecisionTask_Result struct {\n\tBadRequestError              *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError         *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError             *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n\tLimitExceededError           *shared.LimitExceededError           `json:\"limitExceededError,omitempty\"`\n\tDomainNotActiveError         *shared.DomainNotActiveError         `json:\"domainNotActiveError,omitempty\"`\n\tRemoteSyncMatchedError       *shared.RemoteSyncMatchedError       `json:\"remoteSyncMatchedError,omitempty\"`\n\tStickyWorkerUnavailableError *shared.StickyWorkerUnavailableError `json:\"stickyWorkerUnavailableError,omitempty\"`\n\tTaskListNotOwnedByHostError  *shared.TaskListNotOwnedByHostError  `json:\"taskListNotOwnedByHostError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_AddDecisionTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_AddDecisionTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tw, err = v.DomainNotActiveError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tw, err = v.RemoteSyncMatchedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tw, err = v.StickyWorkerUnavailableError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tw, err = v.TaskListNotOwnedByHostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_AddDecisionTask_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StickyWorkerUnavailableError_Read(w wire.Value) (*shared.StickyWorkerUnavailableError, error) {\n\tvar v shared.StickyWorkerUnavailableError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_AddDecisionTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_AddDecisionTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_AddDecisionTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_AddDecisionTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RemoteSyncMatchedError, err = _RemoteSyncMatchedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StickyWorkerUnavailableError, err = _StickyWorkerUnavailableError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_AddDecisionTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_AddDecisionTask_Result struct could not be encoded.\nfunc (v *MatchingService_AddDecisionTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotActiveError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainNotActiveError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RemoteSyncMatchedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RemoteSyncMatchedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StickyWorkerUnavailableError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListNotOwnedByHostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StickyWorkerUnavailableError_Decode(sr stream.Reader) (*shared.StickyWorkerUnavailableError, error) {\n\tvar v shared.StickyWorkerUnavailableError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_AddDecisionTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_AddDecisionTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_AddDecisionTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.DomainNotActiveError, err = _DomainNotActiveError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.RemoteSyncMatchedError, err = _RemoteSyncMatchedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.StickyWorkerUnavailableError, err = _StickyWorkerUnavailableError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tcount++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_AddDecisionTask_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_AddDecisionTask_Result\n// struct.\nfunc (v *MatchingService_AddDecisionTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotActiveError: %v\", v.DomainNotActiveError)\n\t\ti++\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"RemoteSyncMatchedError: %v\", v.RemoteSyncMatchedError)\n\t\ti++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyWorkerUnavailableError: %v\", v.StickyWorkerUnavailableError)\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListNotOwnedByHostError: %v\", v.TaskListNotOwnedByHostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_AddDecisionTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_AddDecisionTask_Result match the\n// provided MatchingService_AddDecisionTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_AddDecisionTask_Result) Equals(rhs *MatchingService_AddDecisionTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.DomainNotActiveError == nil && rhs.DomainNotActiveError == nil) || (v.DomainNotActiveError != nil && rhs.DomainNotActiveError != nil && v.DomainNotActiveError.Equals(rhs.DomainNotActiveError))) {\n\t\treturn false\n\t}\n\tif !((v.RemoteSyncMatchedError == nil && rhs.RemoteSyncMatchedError == nil) || (v.RemoteSyncMatchedError != nil && rhs.RemoteSyncMatchedError != nil && v.RemoteSyncMatchedError.Equals(rhs.RemoteSyncMatchedError))) {\n\t\treturn false\n\t}\n\tif !((v.StickyWorkerUnavailableError == nil && rhs.StickyWorkerUnavailableError == nil) || (v.StickyWorkerUnavailableError != nil && rhs.StickyWorkerUnavailableError != nil && v.StickyWorkerUnavailableError.Equals(rhs.StickyWorkerUnavailableError))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListNotOwnedByHostError == nil && rhs.TaskListNotOwnedByHostError == nil) || (v.TaskListNotOwnedByHostError != nil && rhs.TaskListNotOwnedByHostError != nil && v.TaskListNotOwnedByHostError.Equals(rhs.TaskListNotOwnedByHostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_AddDecisionTask_Result.\nfunc (v *MatchingService_AddDecisionTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.DomainNotActiveError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainNotActiveError\", v.DomainNotActiveError))\n\t}\n\tif v.RemoteSyncMatchedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"remoteSyncMatchedError\", v.RemoteSyncMatchedError))\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"stickyWorkerUnavailableError\", v.StickyWorkerUnavailableError))\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListNotOwnedByHostError\", v.TaskListNotOwnedByHostError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetDomainNotActiveError returns the value of DomainNotActiveError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetDomainNotActiveError() (o *shared.DomainNotActiveError) {\n\tif v != nil && v.DomainNotActiveError != nil {\n\t\treturn v.DomainNotActiveError\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotActiveError returns true if DomainNotActiveError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetDomainNotActiveError() bool {\n\treturn v != nil && v.DomainNotActiveError != nil\n}\n\n// GetRemoteSyncMatchedError returns the value of RemoteSyncMatchedError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetRemoteSyncMatchedError() (o *shared.RemoteSyncMatchedError) {\n\tif v != nil && v.RemoteSyncMatchedError != nil {\n\t\treturn v.RemoteSyncMatchedError\n\t}\n\n\treturn\n}\n\n// IsSetRemoteSyncMatchedError returns true if RemoteSyncMatchedError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetRemoteSyncMatchedError() bool {\n\treturn v != nil && v.RemoteSyncMatchedError != nil\n}\n\n// GetStickyWorkerUnavailableError returns the value of StickyWorkerUnavailableError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetStickyWorkerUnavailableError() (o *shared.StickyWorkerUnavailableError) {\n\tif v != nil && v.StickyWorkerUnavailableError != nil {\n\t\treturn v.StickyWorkerUnavailableError\n\t}\n\n\treturn\n}\n\n// IsSetStickyWorkerUnavailableError returns true if StickyWorkerUnavailableError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetStickyWorkerUnavailableError() bool {\n\treturn v != nil && v.StickyWorkerUnavailableError != nil\n}\n\n// GetTaskListNotOwnedByHostError returns the value of TaskListNotOwnedByHostError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_AddDecisionTask_Result) GetTaskListNotOwnedByHostError() (o *shared.TaskListNotOwnedByHostError) {\n\tif v != nil && v.TaskListNotOwnedByHostError != nil {\n\t\treturn v.TaskListNotOwnedByHostError\n\t}\n\n\treturn\n}\n\n// IsSetTaskListNotOwnedByHostError returns true if TaskListNotOwnedByHostError is not nil.\nfunc (v *MatchingService_AddDecisionTask_Result) IsSetTaskListNotOwnedByHostError() bool {\n\treturn v != nil && v.TaskListNotOwnedByHostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"AddDecisionTask\" for this struct.\nfunc (v *MatchingService_AddDecisionTask_Result) MethodName() string {\n\treturn \"AddDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_AddDecisionTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_CancelOutstandingPoll_Args represents the arguments for the MatchingService.CancelOutstandingPoll function.\n//\n// The arguments for CancelOutstandingPoll are sent and received over the wire as this struct.\ntype MatchingService_CancelOutstandingPoll_Args struct {\n\tRequest *CancelOutstandingPollRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a MatchingService_CancelOutstandingPoll_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_CancelOutstandingPoll_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CancelOutstandingPollRequest_Read(w wire.Value) (*CancelOutstandingPollRequest, error) {\n\tvar v CancelOutstandingPollRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_CancelOutstandingPoll_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_CancelOutstandingPoll_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_CancelOutstandingPoll_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_CancelOutstandingPoll_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _CancelOutstandingPollRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_CancelOutstandingPoll_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_CancelOutstandingPoll_Args struct could not be encoded.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CancelOutstandingPollRequest_Decode(sr stream.Reader) (*CancelOutstandingPollRequest, error) {\n\tvar v CancelOutstandingPollRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_CancelOutstandingPoll_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_CancelOutstandingPoll_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _CancelOutstandingPollRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_CancelOutstandingPoll_Args\n// struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_CancelOutstandingPoll_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_CancelOutstandingPoll_Args match the\n// provided MatchingService_CancelOutstandingPoll_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) Equals(rhs *MatchingService_CancelOutstandingPoll_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_CancelOutstandingPoll_Args.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) GetRequest() (o *CancelOutstandingPollRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"CancelOutstandingPoll\" for this struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) MethodName() string {\n\treturn \"CancelOutstandingPoll\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_CancelOutstandingPoll_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.CancelOutstandingPoll\n// function.\nvar MatchingService_CancelOutstandingPoll_Helper = struct {\n\t// Args accepts the parameters of CancelOutstandingPoll in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *CancelOutstandingPollRequest,\n\t) *MatchingService_CancelOutstandingPoll_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by CancelOutstandingPoll.\n\t//\n\t// An error can be thrown by CancelOutstandingPoll only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for CancelOutstandingPoll\n\t// given the error returned by it. The provided error may\n\t// be nil if CancelOutstandingPoll did not fail.\n\t//\n\t// This allows mapping errors returned by CancelOutstandingPoll into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// CancelOutstandingPoll\n\t//\n\t//   err := CancelOutstandingPoll(args)\n\t//   result, err := MatchingService_CancelOutstandingPoll_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from CancelOutstandingPoll: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*MatchingService_CancelOutstandingPoll_Result, error)\n\n\t// UnwrapResponse takes the result struct for CancelOutstandingPoll\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if CancelOutstandingPoll threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := MatchingService_CancelOutstandingPoll_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_CancelOutstandingPoll_Result) error\n}{}\n\nfunc init() {\n\tMatchingService_CancelOutstandingPoll_Helper.Args = func(\n\t\trequest *CancelOutstandingPollRequest,\n\t) *MatchingService_CancelOutstandingPoll_Args {\n\t\treturn &MatchingService_CancelOutstandingPoll_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tMatchingService_CancelOutstandingPoll_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_CancelOutstandingPoll_Helper.WrapResponse = func(err error) (*MatchingService_CancelOutstandingPoll_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_CancelOutstandingPoll_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_CancelOutstandingPoll_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_CancelOutstandingPoll_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_CancelOutstandingPoll_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_CancelOutstandingPoll_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_CancelOutstandingPoll_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_CancelOutstandingPoll_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_CancelOutstandingPoll_Result.TaskListNotOwnedByHostError\")\n\t\t\t}\n\t\t\treturn &MatchingService_CancelOutstandingPoll_Result{TaskListNotOwnedByHostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_CancelOutstandingPoll_Helper.UnwrapResponse = func(result *MatchingService_CancelOutstandingPoll_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.TaskListNotOwnedByHostError != nil {\n\t\t\terr = result.TaskListNotOwnedByHostError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// MatchingService_CancelOutstandingPoll_Result represents the result of a MatchingService.CancelOutstandingPoll function call.\n//\n// The result of a CancelOutstandingPoll execution is sent and received over the wire as this struct.\ntype MatchingService_CancelOutstandingPoll_Result struct {\n\tBadRequestError             *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError        *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError            *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n\tTaskListNotOwnedByHostError *shared.TaskListNotOwnedByHostError `json:\"taskListNotOwnedByHostError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_CancelOutstandingPoll_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_CancelOutstandingPoll_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tw, err = v.TaskListNotOwnedByHostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_CancelOutstandingPoll_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a MatchingService_CancelOutstandingPoll_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_CancelOutstandingPoll_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_CancelOutstandingPoll_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_CancelOutstandingPoll_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_CancelOutstandingPoll_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_CancelOutstandingPoll_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_CancelOutstandingPoll_Result struct could not be encoded.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListNotOwnedByHostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_CancelOutstandingPoll_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a MatchingService_CancelOutstandingPoll_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_CancelOutstandingPoll_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_CancelOutstandingPoll_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_CancelOutstandingPoll_Result\n// struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListNotOwnedByHostError: %v\", v.TaskListNotOwnedByHostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_CancelOutstandingPoll_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_CancelOutstandingPoll_Result match the\n// provided MatchingService_CancelOutstandingPoll_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) Equals(rhs *MatchingService_CancelOutstandingPoll_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListNotOwnedByHostError == nil && rhs.TaskListNotOwnedByHostError == nil) || (v.TaskListNotOwnedByHostError != nil && rhs.TaskListNotOwnedByHostError != nil && v.TaskListNotOwnedByHostError.Equals(rhs.TaskListNotOwnedByHostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_CancelOutstandingPoll_Result.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListNotOwnedByHostError\", v.TaskListNotOwnedByHostError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetTaskListNotOwnedByHostError returns the value of TaskListNotOwnedByHostError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) GetTaskListNotOwnedByHostError() (o *shared.TaskListNotOwnedByHostError) {\n\tif v != nil && v.TaskListNotOwnedByHostError != nil {\n\t\treturn v.TaskListNotOwnedByHostError\n\t}\n\n\treturn\n}\n\n// IsSetTaskListNotOwnedByHostError returns true if TaskListNotOwnedByHostError is not nil.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) IsSetTaskListNotOwnedByHostError() bool {\n\treturn v != nil && v.TaskListNotOwnedByHostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"CancelOutstandingPoll\" for this struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) MethodName() string {\n\treturn \"CancelOutstandingPoll\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_CancelOutstandingPoll_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_DescribeTaskList_Args represents the arguments for the MatchingService.DescribeTaskList function.\n//\n// The arguments for DescribeTaskList are sent and received over the wire as this struct.\ntype MatchingService_DescribeTaskList_Args struct {\n\tRequest *DescribeTaskListRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a MatchingService_DescribeTaskList_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_DescribeTaskList_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListRequest_1_Read(w wire.Value) (*DescribeTaskListRequest, error) {\n\tvar v DescribeTaskListRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_DescribeTaskList_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_DescribeTaskList_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_DescribeTaskList_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_DescribeTaskList_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _DescribeTaskListRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_DescribeTaskList_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_DescribeTaskList_Args struct could not be encoded.\nfunc (v *MatchingService_DescribeTaskList_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListRequest_1_Decode(sr stream.Reader) (*DescribeTaskListRequest, error) {\n\tvar v DescribeTaskListRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_DescribeTaskList_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_DescribeTaskList_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_DescribeTaskList_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _DescribeTaskListRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_DescribeTaskList_Args\n// struct.\nfunc (v *MatchingService_DescribeTaskList_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_DescribeTaskList_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_DescribeTaskList_Args match the\n// provided MatchingService_DescribeTaskList_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_DescribeTaskList_Args) Equals(rhs *MatchingService_DescribeTaskList_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_DescribeTaskList_Args.\nfunc (v *MatchingService_DescribeTaskList_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Args) GetRequest() (o *DescribeTaskListRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *MatchingService_DescribeTaskList_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"DescribeTaskList\" for this struct.\nfunc (v *MatchingService_DescribeTaskList_Args) MethodName() string {\n\treturn \"DescribeTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_DescribeTaskList_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_DescribeTaskList_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.DescribeTaskList\n// function.\nvar MatchingService_DescribeTaskList_Helper = struct {\n\t// Args accepts the parameters of DescribeTaskList in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *DescribeTaskListRequest,\n\t) *MatchingService_DescribeTaskList_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by DescribeTaskList.\n\t//\n\t// An error can be thrown by DescribeTaskList only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for DescribeTaskList\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// DescribeTaskList into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by DescribeTaskList\n\t//\n\t//   value, err := DescribeTaskList(args)\n\t//   result, err := MatchingService_DescribeTaskList_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from DescribeTaskList: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.DescribeTaskListResponse, error) (*MatchingService_DescribeTaskList_Result, error)\n\n\t// UnwrapResponse takes the result struct for DescribeTaskList\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if DescribeTaskList threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_DescribeTaskList_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_DescribeTaskList_Result) (*shared.DescribeTaskListResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_DescribeTaskList_Helper.Args = func(\n\t\trequest *DescribeTaskListRequest,\n\t) *MatchingService_DescribeTaskList_Args {\n\t\treturn &MatchingService_DescribeTaskList_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tMatchingService_DescribeTaskList_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_DescribeTaskList_Helper.WrapResponse = func(success *shared.DescribeTaskListResponse, err error) (*MatchingService_DescribeTaskList_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_DescribeTaskList_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_DescribeTaskList_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_DescribeTaskList_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_DescribeTaskList_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_DescribeTaskList_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_DescribeTaskList_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &MatchingService_DescribeTaskList_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_DescribeTaskList_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_DescribeTaskList_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_DescribeTaskList_Result.TaskListNotOwnedByHostError\")\n\t\t\t}\n\t\t\treturn &MatchingService_DescribeTaskList_Result{TaskListNotOwnedByHostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_DescribeTaskList_Helper.UnwrapResponse = func(result *MatchingService_DescribeTaskList_Result) (success *shared.DescribeTaskListResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.TaskListNotOwnedByHostError != nil {\n\t\t\terr = result.TaskListNotOwnedByHostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_DescribeTaskList_Result represents the result of a MatchingService.DescribeTaskList function call.\n//\n// The result of a DescribeTaskList execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_DescribeTaskList_Result struct {\n\t// Value returned by DescribeTaskList after a successful execution.\n\tSuccess                     *shared.DescribeTaskListResponse    `json:\"success,omitempty\"`\n\tBadRequestError             *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError        *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError         *shared.EntityNotExistsError        `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError            *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n\tTaskListNotOwnedByHostError *shared.TaskListNotOwnedByHostError `json:\"taskListNotOwnedByHostError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_DescribeTaskList_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_DescribeTaskList_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tw, err = v.TaskListNotOwnedByHostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_DescribeTaskList_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListResponse_Read(w wire.Value) (*shared.DescribeTaskListResponse, error) {\n\tvar v shared.DescribeTaskListResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Read(w wire.Value) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_DescribeTaskList_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_DescribeTaskList_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_DescribeTaskList_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_DescribeTaskList_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _DescribeTaskListResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_DescribeTaskList_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_DescribeTaskList_Result struct could not be encoded.\nfunc (v *MatchingService_DescribeTaskList_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListNotOwnedByHostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListResponse_Decode(sr stream.Reader) (*shared.DescribeTaskListResponse, error) {\n\tvar v shared.DescribeTaskListResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EntityNotExistsError_Decode(sr stream.Reader) (*shared.EntityNotExistsError, error) {\n\tvar v shared.EntityNotExistsError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_DescribeTaskList_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_DescribeTaskList_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_DescribeTaskList_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _DescribeTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_DescribeTaskList_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_DescribeTaskList_Result\n// struct.\nfunc (v *MatchingService_DescribeTaskList_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListNotOwnedByHostError: %v\", v.TaskListNotOwnedByHostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_DescribeTaskList_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_DescribeTaskList_Result match the\n// provided MatchingService_DescribeTaskList_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_DescribeTaskList_Result) Equals(rhs *MatchingService_DescribeTaskList_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListNotOwnedByHostError == nil && rhs.TaskListNotOwnedByHostError == nil) || (v.TaskListNotOwnedByHostError != nil && rhs.TaskListNotOwnedByHostError != nil && v.TaskListNotOwnedByHostError.Equals(rhs.TaskListNotOwnedByHostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_DescribeTaskList_Result.\nfunc (v *MatchingService_DescribeTaskList_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListNotOwnedByHostError\", v.TaskListNotOwnedByHostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetSuccess() (o *shared.DescribeTaskListResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetTaskListNotOwnedByHostError returns the value of TaskListNotOwnedByHostError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_DescribeTaskList_Result) GetTaskListNotOwnedByHostError() (o *shared.TaskListNotOwnedByHostError) {\n\tif v != nil && v.TaskListNotOwnedByHostError != nil {\n\t\treturn v.TaskListNotOwnedByHostError\n\t}\n\n\treturn\n}\n\n// IsSetTaskListNotOwnedByHostError returns true if TaskListNotOwnedByHostError is not nil.\nfunc (v *MatchingService_DescribeTaskList_Result) IsSetTaskListNotOwnedByHostError() bool {\n\treturn v != nil && v.TaskListNotOwnedByHostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"DescribeTaskList\" for this struct.\nfunc (v *MatchingService_DescribeTaskList_Result) MethodName() string {\n\treturn \"DescribeTaskList\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_DescribeTaskList_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_GetTaskListsByDomain_Args represents the arguments for the MatchingService.GetTaskListsByDomain function.\n//\n// The arguments for GetTaskListsByDomain are sent and received over the wire as this struct.\ntype MatchingService_GetTaskListsByDomain_Args struct {\n\tRequest *shared.GetTaskListsByDomainRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a MatchingService_GetTaskListsByDomain_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_GetTaskListsByDomain_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetTaskListsByDomainRequest_Read(w wire.Value) (*shared.GetTaskListsByDomainRequest, error) {\n\tvar v shared.GetTaskListsByDomainRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_GetTaskListsByDomain_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_GetTaskListsByDomain_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_GetTaskListsByDomain_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_GetTaskListsByDomain_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _GetTaskListsByDomainRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_GetTaskListsByDomain_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_GetTaskListsByDomain_Args struct could not be encoded.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetTaskListsByDomainRequest_Decode(sr stream.Reader) (*shared.GetTaskListsByDomainRequest, error) {\n\tvar v shared.GetTaskListsByDomainRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_GetTaskListsByDomain_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_GetTaskListsByDomain_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _GetTaskListsByDomainRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_GetTaskListsByDomain_Args\n// struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_GetTaskListsByDomain_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_GetTaskListsByDomain_Args match the\n// provided MatchingService_GetTaskListsByDomain_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) Equals(rhs *MatchingService_GetTaskListsByDomain_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_GetTaskListsByDomain_Args.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) GetRequest() (o *shared.GetTaskListsByDomainRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"GetTaskListsByDomain\" for this struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) MethodName() string {\n\treturn \"GetTaskListsByDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_GetTaskListsByDomain_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.GetTaskListsByDomain\n// function.\nvar MatchingService_GetTaskListsByDomain_Helper = struct {\n\t// Args accepts the parameters of GetTaskListsByDomain in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *shared.GetTaskListsByDomainRequest,\n\t) *MatchingService_GetTaskListsByDomain_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by GetTaskListsByDomain.\n\t//\n\t// An error can be thrown by GetTaskListsByDomain only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for GetTaskListsByDomain\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// GetTaskListsByDomain into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by GetTaskListsByDomain\n\t//\n\t//   value, err := GetTaskListsByDomain(args)\n\t//   result, err := MatchingService_GetTaskListsByDomain_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from GetTaskListsByDomain: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.GetTaskListsByDomainResponse, error) (*MatchingService_GetTaskListsByDomain_Result, error)\n\n\t// UnwrapResponse takes the result struct for GetTaskListsByDomain\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if GetTaskListsByDomain threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_GetTaskListsByDomain_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_GetTaskListsByDomain_Result) (*shared.GetTaskListsByDomainResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_GetTaskListsByDomain_Helper.Args = func(\n\t\trequest *shared.GetTaskListsByDomainRequest,\n\t) *MatchingService_GetTaskListsByDomain_Args {\n\t\treturn &MatchingService_GetTaskListsByDomain_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tMatchingService_GetTaskListsByDomain_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_GetTaskListsByDomain_Helper.WrapResponse = func(success *shared.GetTaskListsByDomainResponse, err error) (*MatchingService_GetTaskListsByDomain_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_GetTaskListsByDomain_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_GetTaskListsByDomain_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_GetTaskListsByDomain_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_GetTaskListsByDomain_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_GetTaskListsByDomain_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_GetTaskListsByDomain_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &MatchingService_GetTaskListsByDomain_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_GetTaskListsByDomain_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_GetTaskListsByDomain_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_GetTaskListsByDomain_Helper.UnwrapResponse = func(result *MatchingService_GetTaskListsByDomain_Result) (success *shared.GetTaskListsByDomainResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_GetTaskListsByDomain_Result represents the result of a MatchingService.GetTaskListsByDomain function call.\n//\n// The result of a GetTaskListsByDomain execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_GetTaskListsByDomain_Result struct {\n\t// Value returned by GetTaskListsByDomain after a successful execution.\n\tSuccess              *shared.GetTaskListsByDomainResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError         `json:\"entityNotExistError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_GetTaskListsByDomain_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_GetTaskListsByDomain_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _GetTaskListsByDomainResponse_Read(w wire.Value) (*shared.GetTaskListsByDomainResponse, error) {\n\tvar v shared.GetTaskListsByDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_GetTaskListsByDomain_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_GetTaskListsByDomain_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_GetTaskListsByDomain_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_GetTaskListsByDomain_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _GetTaskListsByDomainResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_GetTaskListsByDomain_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_GetTaskListsByDomain_Result struct could not be encoded.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _GetTaskListsByDomainResponse_Decode(sr stream.Reader) (*shared.GetTaskListsByDomainResponse, error) {\n\tvar v shared.GetTaskListsByDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_GetTaskListsByDomain_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_GetTaskListsByDomain_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _GetTaskListsByDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_GetTaskListsByDomain_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_GetTaskListsByDomain_Result\n// struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_GetTaskListsByDomain_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_GetTaskListsByDomain_Result match the\n// provided MatchingService_GetTaskListsByDomain_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) Equals(rhs *MatchingService_GetTaskListsByDomain_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_GetTaskListsByDomain_Result.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) GetSuccess() (o *shared.GetTaskListsByDomainResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"GetTaskListsByDomain\" for this struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) MethodName() string {\n\treturn \"GetTaskListsByDomain\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_GetTaskListsByDomain_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_ListTaskListPartitions_Args represents the arguments for the MatchingService.ListTaskListPartitions function.\n//\n// The arguments for ListTaskListPartitions are sent and received over the wire as this struct.\ntype MatchingService_ListTaskListPartitions_Args struct {\n\tRequest *ListTaskListPartitionsRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a MatchingService_ListTaskListPartitions_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_ListTaskListPartitions_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListTaskListPartitionsRequest_Read(w wire.Value) (*ListTaskListPartitionsRequest, error) {\n\tvar v ListTaskListPartitionsRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_ListTaskListPartitions_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_ListTaskListPartitions_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_ListTaskListPartitions_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_ListTaskListPartitions_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _ListTaskListPartitionsRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_ListTaskListPartitions_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_ListTaskListPartitions_Args struct could not be encoded.\nfunc (v *MatchingService_ListTaskListPartitions_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListTaskListPartitionsRequest_Decode(sr stream.Reader) (*ListTaskListPartitionsRequest, error) {\n\tvar v ListTaskListPartitionsRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_ListTaskListPartitions_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_ListTaskListPartitions_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_ListTaskListPartitions_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _ListTaskListPartitionsRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_ListTaskListPartitions_Args\n// struct.\nfunc (v *MatchingService_ListTaskListPartitions_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_ListTaskListPartitions_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_ListTaskListPartitions_Args match the\n// provided MatchingService_ListTaskListPartitions_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_ListTaskListPartitions_Args) Equals(rhs *MatchingService_ListTaskListPartitions_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_ListTaskListPartitions_Args.\nfunc (v *MatchingService_ListTaskListPartitions_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_ListTaskListPartitions_Args) GetRequest() (o *ListTaskListPartitionsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *MatchingService_ListTaskListPartitions_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"ListTaskListPartitions\" for this struct.\nfunc (v *MatchingService_ListTaskListPartitions_Args) MethodName() string {\n\treturn \"ListTaskListPartitions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_ListTaskListPartitions_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_ListTaskListPartitions_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.ListTaskListPartitions\n// function.\nvar MatchingService_ListTaskListPartitions_Helper = struct {\n\t// Args accepts the parameters of ListTaskListPartitions in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *ListTaskListPartitionsRequest,\n\t) *MatchingService_ListTaskListPartitions_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by ListTaskListPartitions.\n\t//\n\t// An error can be thrown by ListTaskListPartitions only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for ListTaskListPartitions\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// ListTaskListPartitions into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by ListTaskListPartitions\n\t//\n\t//   value, err := ListTaskListPartitions(args)\n\t//   result, err := MatchingService_ListTaskListPartitions_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from ListTaskListPartitions: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.ListTaskListPartitionsResponse, error) (*MatchingService_ListTaskListPartitions_Result, error)\n\n\t// UnwrapResponse takes the result struct for ListTaskListPartitions\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if ListTaskListPartitions threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_ListTaskListPartitions_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_ListTaskListPartitions_Result) (*shared.ListTaskListPartitionsResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_ListTaskListPartitions_Helper.Args = func(\n\t\trequest *ListTaskListPartitionsRequest,\n\t) *MatchingService_ListTaskListPartitions_Args {\n\t\treturn &MatchingService_ListTaskListPartitions_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tMatchingService_ListTaskListPartitions_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_ListTaskListPartitions_Helper.WrapResponse = func(success *shared.ListTaskListPartitionsResponse, err error) (*MatchingService_ListTaskListPartitions_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_ListTaskListPartitions_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_ListTaskListPartitions_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_ListTaskListPartitions_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_ListTaskListPartitions_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_ListTaskListPartitions_Result{InternalServiceError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_ListTaskListPartitions_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_ListTaskListPartitions_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_ListTaskListPartitions_Helper.UnwrapResponse = func(result *MatchingService_ListTaskListPartitions_Result) (success *shared.ListTaskListPartitionsResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_ListTaskListPartitions_Result represents the result of a MatchingService.ListTaskListPartitions function call.\n//\n// The result of a ListTaskListPartitions execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_ListTaskListPartitions_Result struct {\n\t// Value returned by ListTaskListPartitions after a successful execution.\n\tSuccess              *shared.ListTaskListPartitionsResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError                `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError           `json:\"internalServiceError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError               `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_ListTaskListPartitions_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_ListTaskListPartitions_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListTaskListPartitionsResponse_Read(w wire.Value) (*shared.ListTaskListPartitionsResponse, error) {\n\tvar v shared.ListTaskListPartitionsResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_ListTaskListPartitions_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_ListTaskListPartitions_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_ListTaskListPartitions_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_ListTaskListPartitions_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _ListTaskListPartitionsResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_ListTaskListPartitions_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_ListTaskListPartitions_Result struct could not be encoded.\nfunc (v *MatchingService_ListTaskListPartitions_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListTaskListPartitionsResponse_Decode(sr stream.Reader) (*shared.ListTaskListPartitionsResponse, error) {\n\tvar v shared.ListTaskListPartitionsResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_ListTaskListPartitions_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_ListTaskListPartitions_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_ListTaskListPartitions_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _ListTaskListPartitionsResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_ListTaskListPartitions_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_ListTaskListPartitions_Result\n// struct.\nfunc (v *MatchingService_ListTaskListPartitions_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_ListTaskListPartitions_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_ListTaskListPartitions_Result match the\n// provided MatchingService_ListTaskListPartitions_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_ListTaskListPartitions_Result) Equals(rhs *MatchingService_ListTaskListPartitions_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_ListTaskListPartitions_Result.\nfunc (v *MatchingService_ListTaskListPartitions_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_ListTaskListPartitions_Result) GetSuccess() (o *shared.ListTaskListPartitionsResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_ListTaskListPartitions_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_ListTaskListPartitions_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_ListTaskListPartitions_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_ListTaskListPartitions_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_ListTaskListPartitions_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_ListTaskListPartitions_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_ListTaskListPartitions_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"ListTaskListPartitions\" for this struct.\nfunc (v *MatchingService_ListTaskListPartitions_Result) MethodName() string {\n\treturn \"ListTaskListPartitions\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_ListTaskListPartitions_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_PollForActivityTask_Args represents the arguments for the MatchingService.PollForActivityTask function.\n//\n// The arguments for PollForActivityTask are sent and received over the wire as this struct.\ntype MatchingService_PollForActivityTask_Args struct {\n\tPollRequest *PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a MatchingService_PollForActivityTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_PollForActivityTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskRequest_1_Read(w wire.Value) (*PollForActivityTaskRequest, error) {\n\tvar v PollForActivityTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_PollForActivityTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_PollForActivityTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_PollForActivityTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_PollForActivityTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_PollForActivityTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_PollForActivityTask_Args struct could not be encoded.\nfunc (v *MatchingService_PollForActivityTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskRequest_1_Decode(sr stream.Reader) (*PollForActivityTaskRequest, error) {\n\tvar v PollForActivityTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_PollForActivityTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_PollForActivityTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_PollForActivityTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForActivityTaskRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_PollForActivityTask_Args\n// struct.\nfunc (v *MatchingService_PollForActivityTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_PollForActivityTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_PollForActivityTask_Args match the\n// provided MatchingService_PollForActivityTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_PollForActivityTask_Args) Equals(rhs *MatchingService_PollForActivityTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_PollForActivityTask_Args.\nfunc (v *MatchingService_PollForActivityTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Args) GetPollRequest() (o *PollForActivityTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *MatchingService_PollForActivityTask_Args) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PollForActivityTask\" for this struct.\nfunc (v *MatchingService_PollForActivityTask_Args) MethodName() string {\n\treturn \"PollForActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_PollForActivityTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_PollForActivityTask_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.PollForActivityTask\n// function.\nvar MatchingService_PollForActivityTask_Helper = struct {\n\t// Args accepts the parameters of PollForActivityTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tpollRequest *PollForActivityTaskRequest,\n\t) *MatchingService_PollForActivityTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PollForActivityTask.\n\t//\n\t// An error can be thrown by PollForActivityTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PollForActivityTask\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// PollForActivityTask into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by PollForActivityTask\n\t//\n\t//   value, err := PollForActivityTask(args)\n\t//   result, err := MatchingService_PollForActivityTask_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PollForActivityTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.PollForActivityTaskResponse, error) (*MatchingService_PollForActivityTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for PollForActivityTask\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if PollForActivityTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_PollForActivityTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_PollForActivityTask_Result) (*shared.PollForActivityTaskResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_PollForActivityTask_Helper.Args = func(\n\t\tpollRequest *PollForActivityTaskRequest,\n\t) *MatchingService_PollForActivityTask_Args {\n\t\treturn &MatchingService_PollForActivityTask_Args{\n\t\t\tPollRequest: pollRequest,\n\t\t}\n\t}\n\n\tMatchingService_PollForActivityTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_PollForActivityTask_Helper.WrapResponse = func(success *shared.PollForActivityTaskResponse, err error) (*MatchingService_PollForActivityTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_PollForActivityTask_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForActivityTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForActivityTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForActivityTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForActivityTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForActivityTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForActivityTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForActivityTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForActivityTask_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_PollForActivityTask_Helper.UnwrapResponse = func(result *MatchingService_PollForActivityTask_Result) (success *shared.PollForActivityTaskResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_PollForActivityTask_Result represents the result of a MatchingService.PollForActivityTask function call.\n//\n// The result of a PollForActivityTask execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_PollForActivityTask_Result struct {\n\t// Value returned by PollForActivityTask after a successful execution.\n\tSuccess              *shared.PollForActivityTaskResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError             `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError        `json:\"internalServiceError,omitempty\"`\n\tLimitExceededError   *shared.LimitExceededError          `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError            `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_PollForActivityTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_PollForActivityTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_PollForActivityTask_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForActivityTaskResponse_Read(w wire.Value) (*shared.PollForActivityTaskResponse, error) {\n\tvar v shared.PollForActivityTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_PollForActivityTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_PollForActivityTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_PollForActivityTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_PollForActivityTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _PollForActivityTaskResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_PollForActivityTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_PollForActivityTask_Result struct could not be encoded.\nfunc (v *MatchingService_PollForActivityTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForActivityTaskResponse_Decode(sr stream.Reader) (*shared.PollForActivityTaskResponse, error) {\n\tvar v shared.PollForActivityTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_PollForActivityTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_PollForActivityTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_PollForActivityTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _PollForActivityTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForActivityTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_PollForActivityTask_Result\n// struct.\nfunc (v *MatchingService_PollForActivityTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_PollForActivityTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_PollForActivityTask_Result match the\n// provided MatchingService_PollForActivityTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_PollForActivityTask_Result) Equals(rhs *MatchingService_PollForActivityTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_PollForActivityTask_Result.\nfunc (v *MatchingService_PollForActivityTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Result) GetSuccess() (o *shared.PollForActivityTaskResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_PollForActivityTask_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_PollForActivityTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_PollForActivityTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_PollForActivityTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForActivityTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_PollForActivityTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PollForActivityTask\" for this struct.\nfunc (v *MatchingService_PollForActivityTask_Result) MethodName() string {\n\treturn \"PollForActivityTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_PollForActivityTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_PollForDecisionTask_Args represents the arguments for the MatchingService.PollForDecisionTask function.\n//\n// The arguments for PollForDecisionTask are sent and received over the wire as this struct.\ntype MatchingService_PollForDecisionTask_Args struct {\n\tPollRequest *PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// ToWire translates a MatchingService_PollForDecisionTask_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_PollForDecisionTask_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PollRequest != nil {\n\t\tw, err = v.PollRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskRequest_1_Read(w wire.Value) (*PollForDecisionTaskRequest, error) {\n\tvar v PollForDecisionTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_PollForDecisionTask_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_PollForDecisionTask_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_PollForDecisionTask_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_PollForDecisionTask_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_PollForDecisionTask_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_PollForDecisionTask_Args struct could not be encoded.\nfunc (v *MatchingService_PollForDecisionTask_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PollRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PollRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskRequest_1_Decode(sr stream.Reader) (*PollForDecisionTaskRequest, error) {\n\tvar v PollForDecisionTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_PollForDecisionTask_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_PollForDecisionTask_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_PollForDecisionTask_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.PollRequest, err = _PollForDecisionTaskRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_PollForDecisionTask_Args\n// struct.\nfunc (v *MatchingService_PollForDecisionTask_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.PollRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollRequest: %v\", v.PollRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_PollForDecisionTask_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_PollForDecisionTask_Args match the\n// provided MatchingService_PollForDecisionTask_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_PollForDecisionTask_Args) Equals(rhs *MatchingService_PollForDecisionTask_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.PollRequest == nil && rhs.PollRequest == nil) || (v.PollRequest != nil && rhs.PollRequest != nil && v.PollRequest.Equals(rhs.PollRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_PollForDecisionTask_Args.\nfunc (v *MatchingService_PollForDecisionTask_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PollRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pollRequest\", v.PollRequest))\n\t}\n\treturn err\n}\n\n// GetPollRequest returns the value of PollRequest if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Args) GetPollRequest() (o *PollForDecisionTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\n\treturn\n}\n\n// IsSetPollRequest returns true if PollRequest is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Args) IsSetPollRequest() bool {\n\treturn v != nil && v.PollRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"PollForDecisionTask\" for this struct.\nfunc (v *MatchingService_PollForDecisionTask_Args) MethodName() string {\n\treturn \"PollForDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_PollForDecisionTask_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_PollForDecisionTask_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.PollForDecisionTask\n// function.\nvar MatchingService_PollForDecisionTask_Helper = struct {\n\t// Args accepts the parameters of PollForDecisionTask in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tpollRequest *PollForDecisionTaskRequest,\n\t) *MatchingService_PollForDecisionTask_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by PollForDecisionTask.\n\t//\n\t// An error can be thrown by PollForDecisionTask only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for PollForDecisionTask\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// PollForDecisionTask into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by PollForDecisionTask\n\t//\n\t//   value, err := PollForDecisionTask(args)\n\t//   result, err := MatchingService_PollForDecisionTask_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from PollForDecisionTask: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*PollForDecisionTaskResponse, error) (*MatchingService_PollForDecisionTask_Result, error)\n\n\t// UnwrapResponse takes the result struct for PollForDecisionTask\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if PollForDecisionTask threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_PollForDecisionTask_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_PollForDecisionTask_Result) (*PollForDecisionTaskResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_PollForDecisionTask_Helper.Args = func(\n\t\tpollRequest *PollForDecisionTaskRequest,\n\t) *MatchingService_PollForDecisionTask_Args {\n\t\treturn &MatchingService_PollForDecisionTask_Args{\n\t\t\tPollRequest: pollRequest,\n\t\t}\n\t}\n\n\tMatchingService_PollForDecisionTask_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_PollForDecisionTask_Helper.WrapResponse = func(success *PollForDecisionTaskResponse, err error) (*MatchingService_PollForDecisionTask_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_PollForDecisionTask_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForDecisionTask_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForDecisionTask_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForDecisionTask_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForDecisionTask_Result{InternalServiceError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForDecisionTask_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForDecisionTask_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_PollForDecisionTask_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_PollForDecisionTask_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_PollForDecisionTask_Helper.UnwrapResponse = func(result *MatchingService_PollForDecisionTask_Result) (success *PollForDecisionTaskResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_PollForDecisionTask_Result represents the result of a MatchingService.PollForDecisionTask function call.\n//\n// The result of a PollForDecisionTask execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_PollForDecisionTask_Result struct {\n\t// Value returned by PollForDecisionTask after a successful execution.\n\tSuccess              *PollForDecisionTaskResponse `json:\"success,omitempty\"`\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tLimitExceededError   *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_PollForDecisionTask_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_PollForDecisionTask_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_PollForDecisionTask_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskResponse_Read(w wire.Value) (*PollForDecisionTaskResponse, error) {\n\tvar v PollForDecisionTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_PollForDecisionTask_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_PollForDecisionTask_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_PollForDecisionTask_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_PollForDecisionTask_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _PollForDecisionTaskResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_PollForDecisionTask_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_PollForDecisionTask_Result struct could not be encoded.\nfunc (v *MatchingService_PollForDecisionTask_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskResponse_Decode(sr stream.Reader) (*PollForDecisionTaskResponse, error) {\n\tvar v PollForDecisionTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_PollForDecisionTask_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_PollForDecisionTask_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_PollForDecisionTask_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _PollForDecisionTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_PollForDecisionTask_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_PollForDecisionTask_Result\n// struct.\nfunc (v *MatchingService_PollForDecisionTask_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_PollForDecisionTask_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_PollForDecisionTask_Result match the\n// provided MatchingService_PollForDecisionTask_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_PollForDecisionTask_Result) Equals(rhs *MatchingService_PollForDecisionTask_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_PollForDecisionTask_Result.\nfunc (v *MatchingService_PollForDecisionTask_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Result) GetSuccess() (o *PollForDecisionTaskResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_PollForDecisionTask_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_PollForDecisionTask_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"PollForDecisionTask\" for this struct.\nfunc (v *MatchingService_PollForDecisionTask_Result) MethodName() string {\n\treturn \"PollForDecisionTask\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_PollForDecisionTask_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_QueryWorkflow_Args represents the arguments for the MatchingService.QueryWorkflow function.\n//\n// The arguments for QueryWorkflow are sent and received over the wire as this struct.\ntype MatchingService_QueryWorkflow_Args struct {\n\tQueryRequest *QueryWorkflowRequest `json:\"queryRequest,omitempty\"`\n}\n\n// ToWire translates a MatchingService_QueryWorkflow_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_QueryWorkflow_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.QueryRequest != nil {\n\t\tw, err = v.QueryRequest.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowRequest_1_Read(w wire.Value) (*QueryWorkflowRequest, error) {\n\tvar v QueryWorkflowRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_QueryWorkflow_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_QueryWorkflow_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_QueryWorkflow_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_QueryWorkflow_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_QueryWorkflow_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_QueryWorkflow_Args struct could not be encoded.\nfunc (v *MatchingService_QueryWorkflow_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.QueryRequest != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRequest.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowRequest_1_Decode(sr stream.Reader) (*QueryWorkflowRequest, error) {\n\tvar v QueryWorkflowRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_QueryWorkflow_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_QueryWorkflow_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_QueryWorkflow_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.QueryRequest, err = _QueryWorkflowRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_QueryWorkflow_Args\n// struct.\nfunc (v *MatchingService_QueryWorkflow_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.QueryRequest != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRequest: %v\", v.QueryRequest)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_QueryWorkflow_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_QueryWorkflow_Args match the\n// provided MatchingService_QueryWorkflow_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_QueryWorkflow_Args) Equals(rhs *MatchingService_QueryWorkflow_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.QueryRequest == nil && rhs.QueryRequest == nil) || (v.QueryRequest != nil && rhs.QueryRequest != nil && v.QueryRequest.Equals(rhs.QueryRequest))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_QueryWorkflow_Args.\nfunc (v *MatchingService_QueryWorkflow_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.QueryRequest != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRequest\", v.QueryRequest))\n\t}\n\treturn err\n}\n\n// GetQueryRequest returns the value of QueryRequest if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Args) GetQueryRequest() (o *QueryWorkflowRequest) {\n\tif v != nil && v.QueryRequest != nil {\n\t\treturn v.QueryRequest\n\t}\n\n\treturn\n}\n\n// IsSetQueryRequest returns true if QueryRequest is not nil.\nfunc (v *MatchingService_QueryWorkflow_Args) IsSetQueryRequest() bool {\n\treturn v != nil && v.QueryRequest != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *MatchingService_QueryWorkflow_Args) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_QueryWorkflow_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_QueryWorkflow_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.QueryWorkflow\n// function.\nvar MatchingService_QueryWorkflow_Helper = struct {\n\t// Args accepts the parameters of QueryWorkflow in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\tqueryRequest *QueryWorkflowRequest,\n\t) *MatchingService_QueryWorkflow_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by QueryWorkflow.\n\t//\n\t// An error can be thrown by QueryWorkflow only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for QueryWorkflow\n\t// given its return value and error.\n\t//\n\t// This allows mapping values and errors returned by\n\t// QueryWorkflow into a serializable result struct.\n\t// WrapResponse returns a non-nil error if the provided\n\t// error cannot be thrown by QueryWorkflow\n\t//\n\t//   value, err := QueryWorkflow(args)\n\t//   result, err := MatchingService_QueryWorkflow_Helper.WrapResponse(value, err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from QueryWorkflow: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(*shared.QueryWorkflowResponse, error) (*MatchingService_QueryWorkflow_Result, error)\n\n\t// UnwrapResponse takes the result struct for QueryWorkflow\n\t// and returns the value or error returned by it.\n\t//\n\t// The error is non-nil only if QueryWorkflow threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   value, err := MatchingService_QueryWorkflow_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_QueryWorkflow_Result) (*shared.QueryWorkflowResponse, error)\n}{}\n\nfunc init() {\n\tMatchingService_QueryWorkflow_Helper.Args = func(\n\t\tqueryRequest *QueryWorkflowRequest,\n\t) *MatchingService_QueryWorkflow_Args {\n\t\treturn &MatchingService_QueryWorkflow_Args{\n\t\t\tQueryRequest: queryRequest,\n\t\t}\n\t}\n\n\tMatchingService_QueryWorkflow_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.QueryFailedError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tcase *shared.StickyWorkerUnavailableError:\n\t\t\treturn true\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_QueryWorkflow_Helper.WrapResponse = func(success *shared.QueryWorkflowResponse, err error) (*MatchingService_QueryWorkflow_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_QueryWorkflow_Result{Success: success}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.QueryFailedError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.QueryFailedError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{QueryFailedError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{ServiceBusyError: e}, nil\n\t\tcase *shared.StickyWorkerUnavailableError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.StickyWorkerUnavailableError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{StickyWorkerUnavailableError: e}, nil\n\t\tcase *shared.TaskListNotOwnedByHostError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_QueryWorkflow_Result.TaskListNotOwnedByHostError\")\n\t\t\t}\n\t\t\treturn &MatchingService_QueryWorkflow_Result{TaskListNotOwnedByHostError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_QueryWorkflow_Helper.UnwrapResponse = func(result *MatchingService_QueryWorkflow_Result) (success *shared.QueryWorkflowResponse, err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.QueryFailedError != nil {\n\t\t\terr = result.QueryFailedError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\tif result.StickyWorkerUnavailableError != nil {\n\t\t\terr = result.StickyWorkerUnavailableError\n\t\t\treturn\n\t\t}\n\t\tif result.TaskListNotOwnedByHostError != nil {\n\t\t\terr = result.TaskListNotOwnedByHostError\n\t\t\treturn\n\t\t}\n\n\t\tif result.Success != nil {\n\t\t\tsuccess = result.Success\n\t\t\treturn\n\t\t}\n\n\t\terr = errors.New(\"expected a non-void result\")\n\t\treturn\n\t}\n\n}\n\n// MatchingService_QueryWorkflow_Result represents the result of a MatchingService.QueryWorkflow function call.\n//\n// The result of a QueryWorkflow execution is sent and received over the wire as this struct.\n//\n// Success is set only if the function did not throw an exception.\ntype MatchingService_QueryWorkflow_Result struct {\n\t// Value returned by QueryWorkflow after a successful execution.\n\tSuccess                      *shared.QueryWorkflowResponse        `json:\"success,omitempty\"`\n\tBadRequestError              *shared.BadRequestError              `json:\"badRequestError,omitempty\"`\n\tInternalServiceError         *shared.InternalServiceError         `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError          *shared.EntityNotExistsError         `json:\"entityNotExistError,omitempty\"`\n\tQueryFailedError             *shared.QueryFailedError             `json:\"queryFailedError,omitempty\"`\n\tLimitExceededError           *shared.LimitExceededError           `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError             *shared.ServiceBusyError             `json:\"serviceBusyError,omitempty\"`\n\tStickyWorkerUnavailableError *shared.StickyWorkerUnavailableError `json:\"stickyWorkerUnavailableError,omitempty\"`\n\tTaskListNotOwnedByHostError  *shared.TaskListNotOwnedByHostError  `json:\"taskListNotOwnedByHostError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_QueryWorkflow_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_QueryWorkflow_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Success != nil {\n\t\tw, err = v.Success.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 0, Value: w}\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tw, err = v.QueryFailedError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tw, err = v.StickyWorkerUnavailableError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tw, err = v.TaskListNotOwnedByHostError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\tif i != 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_QueryWorkflow_Result should have exactly one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryWorkflowResponse_Read(w wire.Value) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Read(w wire.Value) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_QueryWorkflow_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_QueryWorkflow_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_QueryWorkflow_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_QueryWorkflow_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 0:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Success, err = _QueryWorkflowResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryFailedError, err = _QueryFailedError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StickyWorkerUnavailableError, err = _StickyWorkerUnavailableError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_QueryWorkflow_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_QueryWorkflow_Result struct could not be encoded.\nfunc (v *MatchingService_QueryWorkflow_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Success != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 0, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Success.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryFailedError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryFailedError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StickyWorkerUnavailableError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListNotOwnedByHostError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryWorkflowResponse_Decode(sr stream.Reader) (*shared.QueryWorkflowResponse, error) {\n\tvar v shared.QueryWorkflowResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _QueryFailedError_Decode(sr stream.Reader) (*shared.QueryFailedError, error) {\n\tvar v shared.QueryFailedError\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_QueryWorkflow_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_QueryWorkflow_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_QueryWorkflow_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 0 && fh.Type == wire.TStruct:\n\t\t\tv.Success, err = _QueryWorkflowResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.QueryFailedError, err = _QueryFailedError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TStruct:\n\t\t\tv.StickyWorkerUnavailableError, err = _StickyWorkerUnavailableError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListNotOwnedByHostError, err = _TaskListNotOwnedByHostError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.Success != nil {\n\t\tcount++\n\t}\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tcount++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tcount++\n\t}\n\tif count != 1 {\n\t\treturn fmt.Errorf(\"MatchingService_QueryWorkflow_Result should have exactly one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_QueryWorkflow_Result\n// struct.\nfunc (v *MatchingService_QueryWorkflow_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Success != nil {\n\t\tfields[i] = fmt.Sprintf(\"Success: %v\", v.Success)\n\t\ti++\n\t}\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.QueryFailedError != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryFailedError: %v\", v.QueryFailedError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyWorkerUnavailableError: %v\", v.StickyWorkerUnavailableError)\n\t\ti++\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListNotOwnedByHostError: %v\", v.TaskListNotOwnedByHostError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_QueryWorkflow_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_QueryWorkflow_Result match the\n// provided MatchingService_QueryWorkflow_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_QueryWorkflow_Result) Equals(rhs *MatchingService_QueryWorkflow_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Success == nil && rhs.Success == nil) || (v.Success != nil && rhs.Success != nil && v.Success.Equals(rhs.Success))) {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.QueryFailedError == nil && rhs.QueryFailedError == nil) || (v.QueryFailedError != nil && rhs.QueryFailedError != nil && v.QueryFailedError.Equals(rhs.QueryFailedError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\tif !((v.StickyWorkerUnavailableError == nil && rhs.StickyWorkerUnavailableError == nil) || (v.StickyWorkerUnavailableError != nil && rhs.StickyWorkerUnavailableError != nil && v.StickyWorkerUnavailableError.Equals(rhs.StickyWorkerUnavailableError))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListNotOwnedByHostError == nil && rhs.TaskListNotOwnedByHostError == nil) || (v.TaskListNotOwnedByHostError != nil && rhs.TaskListNotOwnedByHostError != nil && v.TaskListNotOwnedByHostError.Equals(rhs.TaskListNotOwnedByHostError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_QueryWorkflow_Result.\nfunc (v *MatchingService_QueryWorkflow_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Success != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"success\", v.Success))\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.QueryFailedError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryFailedError\", v.QueryFailedError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\tif v.StickyWorkerUnavailableError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"stickyWorkerUnavailableError\", v.StickyWorkerUnavailableError))\n\t}\n\tif v.TaskListNotOwnedByHostError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListNotOwnedByHostError\", v.TaskListNotOwnedByHostError))\n\t}\n\treturn err\n}\n\n// GetSuccess returns the value of Success if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetSuccess() (o *shared.QueryWorkflowResponse) {\n\tif v != nil && v.Success != nil {\n\t\treturn v.Success\n\t}\n\n\treturn\n}\n\n// IsSetSuccess returns true if Success is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetSuccess() bool {\n\treturn v != nil && v.Success != nil\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetQueryFailedError returns the value of QueryFailedError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetQueryFailedError() (o *shared.QueryFailedError) {\n\tif v != nil && v.QueryFailedError != nil {\n\t\treturn v.QueryFailedError\n\t}\n\n\treturn\n}\n\n// IsSetQueryFailedError returns true if QueryFailedError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetQueryFailedError() bool {\n\treturn v != nil && v.QueryFailedError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// GetStickyWorkerUnavailableError returns the value of StickyWorkerUnavailableError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetStickyWorkerUnavailableError() (o *shared.StickyWorkerUnavailableError) {\n\tif v != nil && v.StickyWorkerUnavailableError != nil {\n\t\treturn v.StickyWorkerUnavailableError\n\t}\n\n\treturn\n}\n\n// IsSetStickyWorkerUnavailableError returns true if StickyWorkerUnavailableError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetStickyWorkerUnavailableError() bool {\n\treturn v != nil && v.StickyWorkerUnavailableError != nil\n}\n\n// GetTaskListNotOwnedByHostError returns the value of TaskListNotOwnedByHostError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_QueryWorkflow_Result) GetTaskListNotOwnedByHostError() (o *shared.TaskListNotOwnedByHostError) {\n\tif v != nil && v.TaskListNotOwnedByHostError != nil {\n\t\treturn v.TaskListNotOwnedByHostError\n\t}\n\n\treturn\n}\n\n// IsSetTaskListNotOwnedByHostError returns true if TaskListNotOwnedByHostError is not nil.\nfunc (v *MatchingService_QueryWorkflow_Result) IsSetTaskListNotOwnedByHostError() bool {\n\treturn v != nil && v.TaskListNotOwnedByHostError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"QueryWorkflow\" for this struct.\nfunc (v *MatchingService_QueryWorkflow_Result) MethodName() string {\n\treturn \"QueryWorkflow\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_QueryWorkflow_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n\n// MatchingService_RespondQueryTaskCompleted_Args represents the arguments for the MatchingService.RespondQueryTaskCompleted function.\n//\n// The arguments for RespondQueryTaskCompleted are sent and received over the wire as this struct.\ntype MatchingService_RespondQueryTaskCompleted_Args struct {\n\tRequest *RespondQueryTaskCompletedRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a MatchingService_RespondQueryTaskCompleted_Args struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _RespondQueryTaskCompletedRequest_1_Read(w wire.Value) (*RespondQueryTaskCompletedRequest, error) {\n\tvar v RespondQueryTaskCompletedRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a MatchingService_RespondQueryTaskCompleted_Args struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_RespondQueryTaskCompleted_Args struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_RespondQueryTaskCompleted_Args\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _RespondQueryTaskCompletedRequest_1_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_RespondQueryTaskCompleted_Args struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_RespondQueryTaskCompleted_Args struct could not be encoded.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _RespondQueryTaskCompletedRequest_1_Decode(sr stream.Reader) (*RespondQueryTaskCompletedRequest, error) {\n\tvar v RespondQueryTaskCompletedRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a MatchingService_RespondQueryTaskCompleted_Args struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_RespondQueryTaskCompleted_Args struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _RespondQueryTaskCompletedRequest_1_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_RespondQueryTaskCompleted_Args\n// struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_RespondQueryTaskCompleted_Args{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_RespondQueryTaskCompleted_Args match the\n// provided MatchingService_RespondQueryTaskCompleted_Args.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) Equals(rhs *MatchingService_RespondQueryTaskCompleted_Args) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_RespondQueryTaskCompleted_Args.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) GetRequest() (o *RespondQueryTaskCompletedRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the arguments.\n//\n// This will always be \"RespondQueryTaskCompleted\" for this struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) MethodName() string {\n\treturn \"RespondQueryTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Call for this struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Args) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Call\n}\n\n// MatchingService_RespondQueryTaskCompleted_Helper provides functions that aid in handling the\n// parameters and return values of the MatchingService.RespondQueryTaskCompleted\n// function.\nvar MatchingService_RespondQueryTaskCompleted_Helper = struct {\n\t// Args accepts the parameters of RespondQueryTaskCompleted in-order and returns\n\t// the arguments struct for the function.\n\tArgs func(\n\t\trequest *RespondQueryTaskCompletedRequest,\n\t) *MatchingService_RespondQueryTaskCompleted_Args\n\n\t// IsException returns true if the given error can be thrown\n\t// by RespondQueryTaskCompleted.\n\t//\n\t// An error can be thrown by RespondQueryTaskCompleted only if the\n\t// corresponding exception type was mentioned in the 'throws'\n\t// section for it in the Thrift file.\n\tIsException func(error) bool\n\n\t// WrapResponse returns the result struct for RespondQueryTaskCompleted\n\t// given the error returned by it. The provided error may\n\t// be nil if RespondQueryTaskCompleted did not fail.\n\t//\n\t// This allows mapping errors returned by RespondQueryTaskCompleted into a\n\t// serializable result struct. WrapResponse returns a\n\t// non-nil error if the provided error cannot be thrown by\n\t// RespondQueryTaskCompleted\n\t//\n\t//   err := RespondQueryTaskCompleted(args)\n\t//   result, err := MatchingService_RespondQueryTaskCompleted_Helper.WrapResponse(err)\n\t//   if err != nil {\n\t//     return fmt.Errorf(\"unexpected error from RespondQueryTaskCompleted: %v\", err)\n\t//   }\n\t//   serialize(result)\n\tWrapResponse func(error) (*MatchingService_RespondQueryTaskCompleted_Result, error)\n\n\t// UnwrapResponse takes the result struct for RespondQueryTaskCompleted\n\t// and returns the erorr returned by it (if any).\n\t//\n\t// The error is non-nil only if RespondQueryTaskCompleted threw an\n\t// exception.\n\t//\n\t//   result := deserialize(bytes)\n\t//   err := MatchingService_RespondQueryTaskCompleted_Helper.UnwrapResponse(result)\n\tUnwrapResponse func(*MatchingService_RespondQueryTaskCompleted_Result) error\n}{}\n\nfunc init() {\n\tMatchingService_RespondQueryTaskCompleted_Helper.Args = func(\n\t\trequest *RespondQueryTaskCompletedRequest,\n\t) *MatchingService_RespondQueryTaskCompleted_Args {\n\t\treturn &MatchingService_RespondQueryTaskCompleted_Args{\n\t\t\tRequest: request,\n\t\t}\n\t}\n\n\tMatchingService_RespondQueryTaskCompleted_Helper.IsException = func(err error) bool {\n\t\tswitch err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\treturn true\n\t\tcase *shared.InternalServiceError:\n\t\t\treturn true\n\t\tcase *shared.EntityNotExistsError:\n\t\t\treturn true\n\t\tcase *shared.LimitExceededError:\n\t\t\treturn true\n\t\tcase *shared.ServiceBusyError:\n\t\t\treturn true\n\t\tdefault:\n\t\t\treturn false\n\t\t}\n\t}\n\n\tMatchingService_RespondQueryTaskCompleted_Helper.WrapResponse = func(err error) (*MatchingService_RespondQueryTaskCompleted_Result, error) {\n\t\tif err == nil {\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{}, nil\n\t\t}\n\n\t\tswitch e := err.(type) {\n\t\tcase *shared.BadRequestError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_RespondQueryTaskCompleted_Result.BadRequestError\")\n\t\t\t}\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{BadRequestError: e}, nil\n\t\tcase *shared.InternalServiceError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_RespondQueryTaskCompleted_Result.InternalServiceError\")\n\t\t\t}\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{InternalServiceError: e}, nil\n\t\tcase *shared.EntityNotExistsError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_RespondQueryTaskCompleted_Result.EntityNotExistError\")\n\t\t\t}\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{EntityNotExistError: e}, nil\n\t\tcase *shared.LimitExceededError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_RespondQueryTaskCompleted_Result.LimitExceededError\")\n\t\t\t}\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{LimitExceededError: e}, nil\n\t\tcase *shared.ServiceBusyError:\n\t\t\tif e == nil {\n\t\t\t\treturn nil, errors.New(\"WrapResponse received non-nil error type with nil value for MatchingService_RespondQueryTaskCompleted_Result.ServiceBusyError\")\n\t\t\t}\n\t\t\treturn &MatchingService_RespondQueryTaskCompleted_Result{ServiceBusyError: e}, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\tMatchingService_RespondQueryTaskCompleted_Helper.UnwrapResponse = func(result *MatchingService_RespondQueryTaskCompleted_Result) (err error) {\n\t\tif result.BadRequestError != nil {\n\t\t\terr = result.BadRequestError\n\t\t\treturn\n\t\t}\n\t\tif result.InternalServiceError != nil {\n\t\t\terr = result.InternalServiceError\n\t\t\treturn\n\t\t}\n\t\tif result.EntityNotExistError != nil {\n\t\t\terr = result.EntityNotExistError\n\t\t\treturn\n\t\t}\n\t\tif result.LimitExceededError != nil {\n\t\t\terr = result.LimitExceededError\n\t\t\treturn\n\t\t}\n\t\tif result.ServiceBusyError != nil {\n\t\t\terr = result.ServiceBusyError\n\t\t\treturn\n\t\t}\n\t\treturn\n\t}\n\n}\n\n// MatchingService_RespondQueryTaskCompleted_Result represents the result of a MatchingService.RespondQueryTaskCompleted function call.\n//\n// The result of a RespondQueryTaskCompleted execution is sent and received over the wire as this struct.\ntype MatchingService_RespondQueryTaskCompleted_Result struct {\n\tBadRequestError      *shared.BadRequestError      `json:\"badRequestError,omitempty\"`\n\tInternalServiceError *shared.InternalServiceError `json:\"internalServiceError,omitempty\"`\n\tEntityNotExistError  *shared.EntityNotExistsError `json:\"entityNotExistError,omitempty\"`\n\tLimitExceededError   *shared.LimitExceededError   `json:\"limitExceededError,omitempty\"`\n\tServiceBusyError     *shared.ServiceBusyError     `json:\"serviceBusyError,omitempty\"`\n}\n\n// ToWire translates a MatchingService_RespondQueryTaskCompleted_Result struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BadRequestError != nil {\n\t\tw, err = v.BadRequestError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tw, err = v.InternalServiceError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tw, err = v.EntityNotExistError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tw, err = v.LimitExceededError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tw, err = v.ServiceBusyError.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\n\tif i > 1 {\n\t\treturn wire.Value{}, fmt.Errorf(\"MatchingService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", i)\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a MatchingService_RespondQueryTaskCompleted_Result struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MatchingService_RespondQueryTaskCompleted_Result struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MatchingService_RespondQueryTaskCompleted_Result\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadRequestError, err = _BadRequestError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InternalServiceError, err = _InternalServiceError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LimitExceededError, err = _LimitExceededError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MatchingService_RespondQueryTaskCompleted_Result struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MatchingService_RespondQueryTaskCompleted_Result struct could not be encoded.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BadRequestError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadRequestError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InternalServiceError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InternalServiceError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EntityNotExistError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EntityNotExistError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LimitExceededError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LimitExceededError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ServiceBusyError != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ServiceBusyError.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a MatchingService_RespondQueryTaskCompleted_Result struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MatchingService_RespondQueryTaskCompleted_Result struct could not be generated from the wire\n// representation.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.BadRequestError, err = _BadRequestError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TStruct:\n\t\t\tv.InternalServiceError, err = _InternalServiceError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TStruct:\n\t\t\tv.EntityNotExistError, err = _EntityNotExistsError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TStruct:\n\t\t\tv.LimitExceededError, err = _LimitExceededError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TStruct:\n\t\t\tv.ServiceBusyError, err = _ServiceBusyError_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tcount := 0\n\tif v.BadRequestError != nil {\n\t\tcount++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tcount++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tcount++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tcount++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tcount++\n\t}\n\tif count > 1 {\n\t\treturn fmt.Errorf(\"MatchingService_RespondQueryTaskCompleted_Result should have at most one field: got %v fields\", count)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MatchingService_RespondQueryTaskCompleted_Result\n// struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.BadRequestError != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadRequestError: %v\", v.BadRequestError)\n\t\ti++\n\t}\n\tif v.InternalServiceError != nil {\n\t\tfields[i] = fmt.Sprintf(\"InternalServiceError: %v\", v.InternalServiceError)\n\t\ti++\n\t}\n\tif v.EntityNotExistError != nil {\n\t\tfields[i] = fmt.Sprintf(\"EntityNotExistError: %v\", v.EntityNotExistError)\n\t\ti++\n\t}\n\tif v.LimitExceededError != nil {\n\t\tfields[i] = fmt.Sprintf(\"LimitExceededError: %v\", v.LimitExceededError)\n\t\ti++\n\t}\n\tif v.ServiceBusyError != nil {\n\t\tfields[i] = fmt.Sprintf(\"ServiceBusyError: %v\", v.ServiceBusyError)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MatchingService_RespondQueryTaskCompleted_Result{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MatchingService_RespondQueryTaskCompleted_Result match the\n// provided MatchingService_RespondQueryTaskCompleted_Result.\n//\n// This function performs a deep comparison.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) Equals(rhs *MatchingService_RespondQueryTaskCompleted_Result) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BadRequestError == nil && rhs.BadRequestError == nil) || (v.BadRequestError != nil && rhs.BadRequestError != nil && v.BadRequestError.Equals(rhs.BadRequestError))) {\n\t\treturn false\n\t}\n\tif !((v.InternalServiceError == nil && rhs.InternalServiceError == nil) || (v.InternalServiceError != nil && rhs.InternalServiceError != nil && v.InternalServiceError.Equals(rhs.InternalServiceError))) {\n\t\treturn false\n\t}\n\tif !((v.EntityNotExistError == nil && rhs.EntityNotExistError == nil) || (v.EntityNotExistError != nil && rhs.EntityNotExistError != nil && v.EntityNotExistError.Equals(rhs.EntityNotExistError))) {\n\t\treturn false\n\t}\n\tif !((v.LimitExceededError == nil && rhs.LimitExceededError == nil) || (v.LimitExceededError != nil && rhs.LimitExceededError != nil && v.LimitExceededError.Equals(rhs.LimitExceededError))) {\n\t\treturn false\n\t}\n\tif !((v.ServiceBusyError == nil && rhs.ServiceBusyError == nil) || (v.ServiceBusyError != nil && rhs.ServiceBusyError != nil && v.ServiceBusyError.Equals(rhs.ServiceBusyError))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MatchingService_RespondQueryTaskCompleted_Result.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BadRequestError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badRequestError\", v.BadRequestError))\n\t}\n\tif v.InternalServiceError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"internalServiceError\", v.InternalServiceError))\n\t}\n\tif v.EntityNotExistError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"entityNotExistError\", v.EntityNotExistError))\n\t}\n\tif v.LimitExceededError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"limitExceededError\", v.LimitExceededError))\n\t}\n\tif v.ServiceBusyError != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"serviceBusyError\", v.ServiceBusyError))\n\t}\n\treturn err\n}\n\n// GetBadRequestError returns the value of BadRequestError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) GetBadRequestError() (o *shared.BadRequestError) {\n\tif v != nil && v.BadRequestError != nil {\n\t\treturn v.BadRequestError\n\t}\n\n\treturn\n}\n\n// IsSetBadRequestError returns true if BadRequestError is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) IsSetBadRequestError() bool {\n\treturn v != nil && v.BadRequestError != nil\n}\n\n// GetInternalServiceError returns the value of InternalServiceError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) GetInternalServiceError() (o *shared.InternalServiceError) {\n\tif v != nil && v.InternalServiceError != nil {\n\t\treturn v.InternalServiceError\n\t}\n\n\treturn\n}\n\n// IsSetInternalServiceError returns true if InternalServiceError is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) IsSetInternalServiceError() bool {\n\treturn v != nil && v.InternalServiceError != nil\n}\n\n// GetEntityNotExistError returns the value of EntityNotExistError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) GetEntityNotExistError() (o *shared.EntityNotExistsError) {\n\tif v != nil && v.EntityNotExistError != nil {\n\t\treturn v.EntityNotExistError\n\t}\n\n\treturn\n}\n\n// IsSetEntityNotExistError returns true if EntityNotExistError is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) IsSetEntityNotExistError() bool {\n\treturn v != nil && v.EntityNotExistError != nil\n}\n\n// GetLimitExceededError returns the value of LimitExceededError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) GetLimitExceededError() (o *shared.LimitExceededError) {\n\tif v != nil && v.LimitExceededError != nil {\n\t\treturn v.LimitExceededError\n\t}\n\n\treturn\n}\n\n// IsSetLimitExceededError returns true if LimitExceededError is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) IsSetLimitExceededError() bool {\n\treturn v != nil && v.LimitExceededError != nil\n}\n\n// GetServiceBusyError returns the value of ServiceBusyError if it is set or its\n// zero value if it is unset.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) GetServiceBusyError() (o *shared.ServiceBusyError) {\n\tif v != nil && v.ServiceBusyError != nil {\n\t\treturn v.ServiceBusyError\n\t}\n\n\treturn\n}\n\n// IsSetServiceBusyError returns true if ServiceBusyError is not nil.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) IsSetServiceBusyError() bool {\n\treturn v != nil && v.ServiceBusyError != nil\n}\n\n// MethodName returns the name of the Thrift function as specified in\n// the IDL, for which this struct represent the result.\n//\n// This will always be \"RespondQueryTaskCompleted\" for this struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) MethodName() string {\n\treturn \"RespondQueryTaskCompleted\"\n}\n\n// EnvelopeType returns the kind of value inside this struct.\n//\n// This will always be Reply for this struct.\nfunc (v *MatchingService_RespondQueryTaskCompleted_Result) EnvelopeType() wire.EnvelopeType {\n\treturn wire.Reply\n}\n"
  },
  {
    "path": ".gen/go/matching/matchingserviceclient/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matchingserviceclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\twire \"go.uber.org/thriftrw/wire\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tmatching \"github.com/uber/cadence/.gen/go/matching\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is a client for the MatchingService service.\ntype Interface interface {\n\tAddActivityTask(\n\t\tctx context.Context,\n\t\tAddRequest *matching.AddActivityTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tAddDecisionTask(\n\t\tctx context.Context,\n\t\tAddRequest *matching.AddDecisionTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tCancelOutstandingPoll(\n\t\tctx context.Context,\n\t\tRequest *matching.CancelOutstandingPollRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n\n\tDescribeTaskList(\n\t\tctx context.Context,\n\t\tRequest *matching.DescribeTaskListRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.DescribeTaskListResponse, error)\n\n\tGetTaskListsByDomain(\n\t\tctx context.Context,\n\t\tRequest *shared.GetTaskListsByDomainRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.GetTaskListsByDomainResponse, error)\n\n\tListTaskListPartitions(\n\t\tctx context.Context,\n\t\tRequest *matching.ListTaskListPartitionsRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.ListTaskListPartitionsResponse, error)\n\n\tPollForActivityTask(\n\t\tctx context.Context,\n\t\tPollRequest *matching.PollForActivityTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.PollForActivityTaskResponse, error)\n\n\tPollForDecisionTask(\n\t\tctx context.Context,\n\t\tPollRequest *matching.PollForDecisionTaskRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*matching.PollForDecisionTaskResponse, error)\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *matching.QueryWorkflowRequest,\n\t\topts ...yarpc.CallOption,\n\t) (*shared.QueryWorkflowResponse, error)\n\n\tRespondQueryTaskCompleted(\n\t\tctx context.Context,\n\t\tRequest *matching.RespondQueryTaskCompletedRequest,\n\t\topts ...yarpc.CallOption,\n\t) error\n}\n\n// New builds a new client for the MatchingService service.\n//\n//\tclient := matchingserviceclient.New(dispatcher.ClientConfig(\"matchingservice\"))\nfunc New(c transport.ClientConfig, opts ...thrift.ClientOption) Interface {\n\treturn client{\n\t\tc: thrift.New(thrift.Config{\n\t\t\tService:      \"MatchingService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t\tnwc: thrift.NewNoWire(thrift.Config{\n\t\t\tService:      \"MatchingService\",\n\t\t\tClientConfig: c,\n\t\t}, opts...),\n\t}\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(c transport.ClientConfig, f reflect.StructField) Interface {\n\t\t\treturn New(c, thrift.ClientBuilderOptions(c, f)...)\n\t\t},\n\t)\n}\n\ntype client struct {\n\tc   thrift.Client\n\tnwc thrift.NoWireClient\n}\n\nfunc (c client) AddActivityTask(\n\tctx context.Context,\n\t_AddRequest *matching.AddActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result matching.MatchingService_AddActivityTask_Result\n\targs := matching.MatchingService_AddActivityTask_Helper.Args(_AddRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = matching.MatchingService_AddActivityTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) AddDecisionTask(\n\tctx context.Context,\n\t_AddRequest *matching.AddDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result matching.MatchingService_AddDecisionTask_Result\n\targs := matching.MatchingService_AddDecisionTask_Helper.Args(_AddRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = matching.MatchingService_AddDecisionTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) CancelOutstandingPoll(\n\tctx context.Context,\n\t_Request *matching.CancelOutstandingPollRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result matching.MatchingService_CancelOutstandingPoll_Result\n\targs := matching.MatchingService_CancelOutstandingPoll_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = matching.MatchingService_CancelOutstandingPoll_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) DescribeTaskList(\n\tctx context.Context,\n\t_Request *matching.DescribeTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeTaskListResponse, err error) {\n\n\tvar result matching.MatchingService_DescribeTaskList_Result\n\targs := matching.MatchingService_DescribeTaskList_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_DescribeTaskList_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) GetTaskListsByDomain(\n\tctx context.Context,\n\t_Request *shared.GetTaskListsByDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetTaskListsByDomainResponse, err error) {\n\n\tvar result matching.MatchingService_GetTaskListsByDomain_Result\n\targs := matching.MatchingService_GetTaskListsByDomain_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_GetTaskListsByDomain_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) ListTaskListPartitions(\n\tctx context.Context,\n\t_Request *matching.ListTaskListPartitionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListTaskListPartitionsResponse, err error) {\n\n\tvar result matching.MatchingService_ListTaskListPartitions_Result\n\targs := matching.MatchingService_ListTaskListPartitions_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_ListTaskListPartitions_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PollForActivityTask(\n\tctx context.Context,\n\t_PollRequest *matching.PollForActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForActivityTaskResponse, err error) {\n\n\tvar result matching.MatchingService_PollForActivityTask_Result\n\targs := matching.MatchingService_PollForActivityTask_Helper.Args(_PollRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_PollForActivityTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) PollForDecisionTask(\n\tctx context.Context,\n\t_PollRequest *matching.PollForDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *matching.PollForDecisionTaskResponse, err error) {\n\n\tvar result matching.MatchingService_PollForDecisionTask_Result\n\targs := matching.MatchingService_PollForDecisionTask_Helper.Args(_PollRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_PollForDecisionTask_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *matching.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.QueryWorkflowResponse, err error) {\n\n\tvar result matching.MatchingService_QueryWorkflow_Result\n\targs := matching.MatchingService_QueryWorkflow_Helper.Args(_QueryRequest)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tsuccess, err = matching.MatchingService_QueryWorkflow_Helper.UnwrapResponse(&result)\n\treturn\n}\n\nfunc (c client) RespondQueryTaskCompleted(\n\tctx context.Context,\n\t_Request *matching.RespondQueryTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\tvar result matching.MatchingService_RespondQueryTaskCompleted_Result\n\targs := matching.MatchingService_RespondQueryTaskCompleted_Helper.Args(_Request)\n\n\tif c.nwc != nil && c.nwc.Enabled() {\n\t\tif err = c.nwc.Call(ctx, args, &result, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\t} else {\n\t\tvar body wire.Value\n\t\tif body, err = c.c.Call(ctx, args, opts...); err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tif err = result.FromWire(body); err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\terr = matching.MatchingService_RespondQueryTaskCompleted_Helper.UnwrapResponse(&result)\n\treturn\n}\n"
  },
  {
    "path": ".gen/go/matching/matchingservicefx/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matchingservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\trestriction \"go.uber.org/yarpc/api/x/restriction\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tmatchingserviceclient \"github.com/uber/cadence/.gen/go/matching/matchingserviceclient\"\n)\n\n// Params defines the dependencies for the MatchingService client.\ntype Params struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// Result defines the output of the MatchingService client module. It provides a\n// MatchingService client to an Fx application.\ntype Result struct {\n\tfx.Out\n\n\tClient matchingserviceclient.Interface\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// Client provides a MatchingService client to an Fx application using the given name\n// for routing.\n//\n//\tfx.Provide(\n//\t\tmatchingservicefx.Client(\"...\"),\n//\t\tnewHandler,\n//\t)\nfunc Client(name string, opts ...thrift.ClientOption) interface{} {\n\treturn func(p Params) Result {\n\t\tcc := p.Provider.ClientConfig(name)\n\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok && p.Restriction != nil {\n\t\t\tif err := p.Restriction.Check(thrift.Encoding, namer.TransportName()); err != nil {\n\t\t\t\tpanic(err.Error())\n\t\t\t}\n\t\t}\n\t\tclient := matchingserviceclient.New(cc, opts...)\n\t\treturn Result{Client: client}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/matching/matchingservicefx/doc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\n// Package matchingservicefx provides better integration for Fx for services\n// implementing or calling MatchingService.\n//\n// # Clients\n//\n// If you are making requests to MatchingService, use the Client function to inject a\n// MatchingService client into your container.\n//\n//\tfx.Provide(matchingservicefx.Client(\"...\"))\n//\n// # Servers\n//\n// If you are implementing MatchingService, provide a matchingserviceserver.Interface into\n// the container and use the Server function.\n//\n// Given,\n//\n//\tfunc NewMatchingServiceHandler() matchingserviceserver.Interface\n//\n// You can do the following to have the procedures of MatchingService made available\n// to an Fx application.\n//\n//\tfx.Provide(\n//\t\tNewMatchingServiceHandler,\n//\t\tmatchingservicefx.Server(),\n//\t)\npackage matchingservicefx\n"
  },
  {
    "path": ".gen/go/matching/matchingservicefx/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matchingservicefx\n\nimport (\n\tfx \"go.uber.org/fx\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\n\tmatchingserviceserver \"github.com/uber/cadence/.gen/go/matching/matchingserviceserver\"\n)\n\n// ServerParams defines the dependencies for the MatchingService server.\ntype ServerParams struct {\n\tfx.In\n\n\tHandler matchingserviceserver.Interface\n}\n\n// ServerResult defines the output of MatchingService server module. It provides the\n// procedures of a MatchingService handler to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group. Dig 1.2 or newer\n// must be used for this feature to work.\ntype ServerResult struct {\n\tfx.Out\n\n\tProcedures []transport.Procedure `group:\"yarpcfx\"`\n}\n\n// Server provides procedures for MatchingService to an Fx application. It expects a\n// matchingservicefx.Interface to be present in the container.\n//\n//\tfx.Provide(\n//\t\tfunc(h *MyMatchingServiceHandler) matchingserviceserver.Interface {\n//\t\t\treturn h\n//\t\t},\n//\t\tmatchingservicefx.Server(),\n//\t)\nfunc Server(opts ...thrift.RegisterOption) interface{} {\n\treturn func(p ServerParams) ServerResult {\n\t\tprocedures := matchingserviceserver.New(p.Handler, opts...)\n\t\treturn ServerResult{Procedures: procedures}\n\t}\n}\n"
  },
  {
    "path": ".gen/go/matching/matchingserviceserver/server.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matchingserviceserver\n\nimport (\n\tcontext \"context\"\n\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\tthrift \"go.uber.org/yarpc/encoding/thrift\"\n\tyarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n\tmatching \"github.com/uber/cadence/.gen/go/matching\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Interface is the server-side interface for the MatchingService service.\ntype Interface interface {\n\tAddActivityTask(\n\t\tctx context.Context,\n\t\tAddRequest *matching.AddActivityTaskRequest,\n\t) error\n\n\tAddDecisionTask(\n\t\tctx context.Context,\n\t\tAddRequest *matching.AddDecisionTaskRequest,\n\t) error\n\n\tCancelOutstandingPoll(\n\t\tctx context.Context,\n\t\tRequest *matching.CancelOutstandingPollRequest,\n\t) error\n\n\tDescribeTaskList(\n\t\tctx context.Context,\n\t\tRequest *matching.DescribeTaskListRequest,\n\t) (*shared.DescribeTaskListResponse, error)\n\n\tGetTaskListsByDomain(\n\t\tctx context.Context,\n\t\tRequest *shared.GetTaskListsByDomainRequest,\n\t) (*shared.GetTaskListsByDomainResponse, error)\n\n\tListTaskListPartitions(\n\t\tctx context.Context,\n\t\tRequest *matching.ListTaskListPartitionsRequest,\n\t) (*shared.ListTaskListPartitionsResponse, error)\n\n\tPollForActivityTask(\n\t\tctx context.Context,\n\t\tPollRequest *matching.PollForActivityTaskRequest,\n\t) (*shared.PollForActivityTaskResponse, error)\n\n\tPollForDecisionTask(\n\t\tctx context.Context,\n\t\tPollRequest *matching.PollForDecisionTaskRequest,\n\t) (*matching.PollForDecisionTaskResponse, error)\n\n\tQueryWorkflow(\n\t\tctx context.Context,\n\t\tQueryRequest *matching.QueryWorkflowRequest,\n\t) (*shared.QueryWorkflowResponse, error)\n\n\tRespondQueryTaskCompleted(\n\t\tctx context.Context,\n\t\tRequest *matching.RespondQueryTaskCompletedRequest,\n\t) error\n}\n\n// New prepares an implementation of the MatchingService service for\n// registration.\n//\n//\thandler := MatchingServiceHandler{}\n//\tdispatcher.Register(matchingserviceserver.New(handler))\nfunc New(impl Interface, opts ...thrift.RegisterOption) []transport.Procedure {\n\th := handler{impl}\n\tservice := thrift.Service{\n\t\tName: \"MatchingService\",\n\t\tMethods: []thrift.Method{\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"AddActivityTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.AddActivityTask),\n\t\t\t\t\tNoWire: addactivitytask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"AddActivityTask(AddRequest *matching.AddActivityTaskRequest)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"AddDecisionTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.AddDecisionTask),\n\t\t\t\t\tNoWire: adddecisiontask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"AddDecisionTask(AddRequest *matching.AddDecisionTaskRequest)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"CancelOutstandingPoll\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.CancelOutstandingPoll),\n\t\t\t\t\tNoWire: canceloutstandingpoll_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"CancelOutstandingPoll(Request *matching.CancelOutstandingPollRequest)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"DescribeTaskList\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.DescribeTaskList),\n\t\t\t\t\tNoWire: describetasklist_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"DescribeTaskList(Request *matching.DescribeTaskListRequest) (*shared.DescribeTaskListResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"GetTaskListsByDomain\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.GetTaskListsByDomain),\n\t\t\t\t\tNoWire: gettasklistsbydomain_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"GetTaskListsByDomain(Request *shared.GetTaskListsByDomainRequest) (*shared.GetTaskListsByDomainResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"ListTaskListPartitions\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.ListTaskListPartitions),\n\t\t\t\t\tNoWire: listtasklistpartitions_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"ListTaskListPartitions(Request *matching.ListTaskListPartitionsRequest) (*shared.ListTaskListPartitionsResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PollForActivityTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PollForActivityTask),\n\t\t\t\t\tNoWire: pollforactivitytask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PollForActivityTask(PollRequest *matching.PollForActivityTaskRequest) (*shared.PollForActivityTaskResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"PollForDecisionTask\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.PollForDecisionTask),\n\t\t\t\t\tNoWire: pollfordecisiontask_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"PollForDecisionTask(PollRequest *matching.PollForDecisionTaskRequest) (*matching.PollForDecisionTaskResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"QueryWorkflow\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.QueryWorkflow),\n\t\t\t\t\tNoWire: queryworkflow_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"QueryWorkflow(QueryRequest *matching.QueryWorkflowRequest) (*shared.QueryWorkflowResponse)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\n\t\t\tthrift.Method{\n\t\t\t\tName: \"RespondQueryTaskCompleted\",\n\t\t\t\tHandlerSpec: thrift.HandlerSpec{\n\n\t\t\t\t\tType:   transport.Unary,\n\t\t\t\t\tUnary:  thrift.UnaryHandler(h.RespondQueryTaskCompleted),\n\t\t\t\t\tNoWire: respondquerytaskcompleted_NoWireHandler{impl},\n\t\t\t\t},\n\t\t\t\tSignature:    \"RespondQueryTaskCompleted(Request *matching.RespondQueryTaskCompletedRequest)\",\n\t\t\t\tThriftModule: matching.ThriftModule,\n\t\t\t},\n\t\t},\n\t}\n\n\tprocedures := make([]transport.Procedure, 0, 10)\n\tprocedures = append(procedures, thrift.BuildProcedures(service, opts...)...)\n\treturn procedures\n}\n\ntype handler struct{ impl Interface }\n\ntype yarpcErrorNamer interface{ YARPCErrorName() string }\n\ntype yarpcErrorCoder interface{ YARPCErrorCode() *yarpcerrors.Code }\n\nfunc (h handler) AddActivityTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_AddActivityTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'AddActivityTask': %w\", err)\n\t}\n\n\tappErr := h.impl.AddActivityTask(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_AddActivityTask_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) AddDecisionTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_AddDecisionTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'AddDecisionTask': %w\", err)\n\t}\n\n\tappErr := h.impl.AddDecisionTask(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_AddDecisionTask_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) CancelOutstandingPoll(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_CancelOutstandingPoll_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'CancelOutstandingPoll': %w\", err)\n\t}\n\n\tappErr := h.impl.CancelOutstandingPoll(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_CancelOutstandingPoll_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) DescribeTaskList(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_DescribeTaskList_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'DescribeTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeTaskList(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_DescribeTaskList_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) GetTaskListsByDomain(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_GetTaskListsByDomain_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'GetTaskListsByDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetTaskListsByDomain(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_GetTaskListsByDomain_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) ListTaskListPartitions(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_ListTaskListPartitions_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'ListTaskListPartitions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListTaskListPartitions(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_ListTaskListPartitions_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PollForActivityTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_PollForActivityTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'PollForActivityTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForActivityTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_PollForActivityTask_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) PollForDecisionTask(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_PollForDecisionTask_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'PollForDecisionTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForDecisionTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_PollForDecisionTask_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) QueryWorkflow(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_QueryWorkflow_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\nfunc (h handler) RespondQueryTaskCompleted(ctx context.Context, body wire.Value) (thrift.Response, error) {\n\tvar args matching.MatchingService_RespondQueryTaskCompleted_Args\n\tif err := args.FromWire(body); err != nil {\n\t\treturn thrift.Response{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode Thrift request for service 'MatchingService' procedure 'RespondQueryTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondQueryTaskCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_RespondQueryTaskCompleted_Helper.WrapResponse(appErr)\n\n\tvar response thrift.Response\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\n\treturn response, err\n}\n\ntype addactivitytask_NoWireHandler struct{ impl Interface }\n\nfunc (h addactivitytask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_AddActivityTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'AddActivityTask': %w\", err)\n\t}\n\n\tappErr := h.impl.AddActivityTask(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_AddActivityTask_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype adddecisiontask_NoWireHandler struct{ impl Interface }\n\nfunc (h adddecisiontask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_AddDecisionTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'AddDecisionTask': %w\", err)\n\t}\n\n\tappErr := h.impl.AddDecisionTask(ctx, args.AddRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_AddDecisionTask_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype canceloutstandingpoll_NoWireHandler struct{ impl Interface }\n\nfunc (h canceloutstandingpoll_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_CancelOutstandingPoll_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'CancelOutstandingPoll': %w\", err)\n\t}\n\n\tappErr := h.impl.CancelOutstandingPoll(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_CancelOutstandingPoll_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype describetasklist_NoWireHandler struct{ impl Interface }\n\nfunc (h describetasklist_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_DescribeTaskList_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'DescribeTaskList': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.DescribeTaskList(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_DescribeTaskList_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype gettasklistsbydomain_NoWireHandler struct{ impl Interface }\n\nfunc (h gettasklistsbydomain_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_GetTaskListsByDomain_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'GetTaskListsByDomain': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.GetTaskListsByDomain(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_GetTaskListsByDomain_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype listtasklistpartitions_NoWireHandler struct{ impl Interface }\n\nfunc (h listtasklistpartitions_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_ListTaskListPartitions_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'ListTaskListPartitions': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.ListTaskListPartitions(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_ListTaskListPartitions_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype pollforactivitytask_NoWireHandler struct{ impl Interface }\n\nfunc (h pollforactivitytask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_PollForActivityTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'PollForActivityTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForActivityTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_PollForActivityTask_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype pollfordecisiontask_NoWireHandler struct{ impl Interface }\n\nfunc (h pollfordecisiontask_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_PollForDecisionTask_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'PollForDecisionTask': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.PollForDecisionTask(ctx, args.PollRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_PollForDecisionTask_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype queryworkflow_NoWireHandler struct{ impl Interface }\n\nfunc (h queryworkflow_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_QueryWorkflow_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'QueryWorkflow': %w\", err)\n\t}\n\n\tsuccess, appErr := h.impl.QueryWorkflow(ctx, args.QueryRequest)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_QueryWorkflow_Helper.WrapResponse(success, appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n\ntype respondquerytaskcompleted_NoWireHandler struct{ impl Interface }\n\nfunc (h respondquerytaskcompleted_NoWireHandler) HandleNoWire(ctx context.Context, nwc *thrift.NoWireCall) (thrift.NoWireResponse, error) {\n\tvar (\n\t\targs matching.MatchingService_RespondQueryTaskCompleted_Args\n\t\trw   stream.ResponseWriter\n\t\terr  error\n\t)\n\n\trw, err = nwc.RequestReader.ReadRequest(ctx, nwc.EnvelopeType, nwc.Reader, &args)\n\tif err != nil {\n\t\treturn thrift.NoWireResponse{}, yarpcerrors.InvalidArgumentErrorf(\n\t\t\t\"could not decode (via no wire) Thrift request for service 'MatchingService' procedure 'RespondQueryTaskCompleted': %w\", err)\n\t}\n\n\tappErr := h.impl.RespondQueryTaskCompleted(ctx, args.Request)\n\n\thadError := appErr != nil\n\tresult, err := matching.MatchingService_RespondQueryTaskCompleted_Helper.WrapResponse(appErr)\n\tresponse := thrift.NoWireResponse{ResponseWriter: rw}\n\tif err == nil {\n\t\tresponse.IsApplicationError = hadError\n\t\tresponse.Body = result\n\t\tif namer, ok := appErr.(yarpcErrorNamer); ok {\n\t\t\tresponse.ApplicationErrorName = namer.YARPCErrorName()\n\t\t}\n\t\tif extractor, ok := appErr.(yarpcErrorCoder); ok {\n\t\t\tresponse.ApplicationErrorCode = extractor.YARPCErrorCode()\n\t\t}\n\t\tif appErr != nil {\n\t\t\tresponse.ApplicationErrorDetails = appErr.Error()\n\t\t}\n\t}\n\treturn response, err\n\n}\n"
  },
  {
    "path": ".gen/go/matching/matchingservicetest/client.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matchingservicetest\n\nimport (\n\tcontext \"context\"\n\n\tgomock \"github.com/golang/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tmatching \"github.com/uber/cadence/.gen/go/matching\"\n\tmatchingserviceclient \"github.com/uber/cadence/.gen/go/matching/matchingserviceclient\"\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// MockClient implements a gomock-compatible mock client for service\n// MatchingService.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *_MockClientRecorder\n}\n\nvar _ matchingserviceclient.Interface = (*MockClient)(nil)\n\ntype _MockClientRecorder struct {\n\tmock *MockClient\n}\n\n// Build a new mock client for service MatchingService.\n//\n//\tmockCtrl := gomock.NewController(t)\n//\tclient := matchingservicetest.NewMockClient(mockCtrl)\n//\n// Use EXPECT() to set expectations on the mock.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &_MockClientRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows you to define an expectation on the\n// MatchingService mock client.\nfunc (m *MockClient) EXPECT() *_MockClientRecorder {\n\treturn m.recorder\n}\n\n// AddActivityTask responds to a AddActivityTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().AddActivityTask(gomock.Any(), ...).Return(...)\n//\t... := client.AddActivityTask(...)\nfunc (m *MockClient) AddActivityTask(\n\tctx context.Context,\n\t_AddRequest *matching.AddActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _AddRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"AddActivityTask\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) AddActivityTask(\n\tctx interface{},\n\t_AddRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _AddRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"AddActivityTask\", args...)\n}\n\n// AddDecisionTask responds to a AddDecisionTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().AddDecisionTask(gomock.Any(), ...).Return(...)\n//\t... := client.AddDecisionTask(...)\nfunc (m *MockClient) AddDecisionTask(\n\tctx context.Context,\n\t_AddRequest *matching.AddDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _AddRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"AddDecisionTask\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) AddDecisionTask(\n\tctx interface{},\n\t_AddRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _AddRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"AddDecisionTask\", args...)\n}\n\n// CancelOutstandingPoll responds to a CancelOutstandingPoll call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().CancelOutstandingPoll(gomock.Any(), ...).Return(...)\n//\t... := client.CancelOutstandingPoll(...)\nfunc (m *MockClient) CancelOutstandingPoll(\n\tctx context.Context,\n\t_Request *matching.CancelOutstandingPollRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"CancelOutstandingPoll\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) CancelOutstandingPoll(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"CancelOutstandingPoll\", args...)\n}\n\n// DescribeTaskList responds to a DescribeTaskList call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().DescribeTaskList(gomock.Any(), ...).Return(...)\n//\t... := client.DescribeTaskList(...)\nfunc (m *MockClient) DescribeTaskList(\n\tctx context.Context,\n\t_Request *matching.DescribeTaskListRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.DescribeTaskListResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", args...)\n\tsuccess, _ = ret[i].(*shared.DescribeTaskListResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) DescribeTaskList(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"DescribeTaskList\", args...)\n}\n\n// GetTaskListsByDomain responds to a GetTaskListsByDomain call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().GetTaskListsByDomain(gomock.Any(), ...).Return(...)\n//\t... := client.GetTaskListsByDomain(...)\nfunc (m *MockClient) GetTaskListsByDomain(\n\tctx context.Context,\n\t_Request *shared.GetTaskListsByDomainRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.GetTaskListsByDomainResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", args...)\n\tsuccess, _ = ret[i].(*shared.GetTaskListsByDomainResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) GetTaskListsByDomain(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"GetTaskListsByDomain\", args...)\n}\n\n// ListTaskListPartitions responds to a ListTaskListPartitions call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().ListTaskListPartitions(gomock.Any(), ...).Return(...)\n//\t... := client.ListTaskListPartitions(...)\nfunc (m *MockClient) ListTaskListPartitions(\n\tctx context.Context,\n\t_Request *matching.ListTaskListPartitionsRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.ListTaskListPartitionsResponse, err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", args...)\n\tsuccess, _ = ret[i].(*shared.ListTaskListPartitionsResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) ListTaskListPartitions(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"ListTaskListPartitions\", args...)\n}\n\n// PollForActivityTask responds to a PollForActivityTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PollForActivityTask(gomock.Any(), ...).Return(...)\n//\t... := client.PollForActivityTask(...)\nfunc (m *MockClient) PollForActivityTask(\n\tctx context.Context,\n\t_PollRequest *matching.PollForActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.PollForActivityTaskResponse, err error) {\n\n\targs := []interface{}{ctx, _PollRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", args...)\n\tsuccess, _ = ret[i].(*shared.PollForActivityTaskResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PollForActivityTask(\n\tctx interface{},\n\t_PollRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _PollRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PollForActivityTask\", args...)\n}\n\n// PollForDecisionTask responds to a PollForDecisionTask call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().PollForDecisionTask(gomock.Any(), ...).Return(...)\n//\t... := client.PollForDecisionTask(...)\nfunc (m *MockClient) PollForDecisionTask(\n\tctx context.Context,\n\t_PollRequest *matching.PollForDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (success *matching.PollForDecisionTaskResponse, err error) {\n\n\targs := []interface{}{ctx, _PollRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", args...)\n\tsuccess, _ = ret[i].(*matching.PollForDecisionTaskResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) PollForDecisionTask(\n\tctx interface{},\n\t_PollRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _PollRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"PollForDecisionTask\", args...)\n}\n\n// QueryWorkflow responds to a QueryWorkflow call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().QueryWorkflow(gomock.Any(), ...).Return(...)\n//\t... := client.QueryWorkflow(...)\nfunc (m *MockClient) QueryWorkflow(\n\tctx context.Context,\n\t_QueryRequest *matching.QueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (success *shared.QueryWorkflowResponse, err error) {\n\n\targs := []interface{}{ctx, _QueryRequest}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", args...)\n\tsuccess, _ = ret[i].(*shared.QueryWorkflowResponse)\n\ti++\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) QueryWorkflow(\n\tctx interface{},\n\t_QueryRequest interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _QueryRequest}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"QueryWorkflow\", args...)\n}\n\n// RespondQueryTaskCompleted responds to a RespondQueryTaskCompleted call based on the mock expectations. This\n// call will fail if the mock does not expect this call. Use EXPECT to expect\n// a call to this function.\n//\n//\tclient.EXPECT().RespondQueryTaskCompleted(gomock.Any(), ...).Return(...)\n//\t... := client.RespondQueryTaskCompleted(...)\nfunc (m *MockClient) RespondQueryTaskCompleted(\n\tctx context.Context,\n\t_Request *matching.RespondQueryTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (err error) {\n\n\targs := []interface{}{ctx, _Request}\n\tfor _, o := range opts {\n\t\targs = append(args, o)\n\t}\n\ti := 0\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", args...)\n\terr, _ = ret[i].(error)\n\treturn\n}\n\nfunc (mr *_MockClientRecorder) RespondQueryTaskCompleted(\n\tctx interface{},\n\t_Request interface{},\n\topts ...interface{},\n) *gomock.Call {\n\targs := append([]interface{}{ctx, _Request}, opts...)\n\treturn mr.mock.ctrl.RecordCall(mr.mock, \"RespondQueryTaskCompleted\", args...)\n}\n"
  },
  {
    "path": ".gen/go/matching/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage matching\n"
  },
  {
    "path": ".gen/go/replicator/replicator.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage replicator\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype DLQType int32\n\nconst (\n\tDLQTypeReplication DLQType = 0\n\tDLQTypeDomain      DLQType = 1\n)\n\n// DLQType_Values returns all recognized values of DLQType.\nfunc DLQType_Values() []DLQType {\n\treturn []DLQType{\n\t\tDLQTypeReplication,\n\t\tDLQTypeDomain,\n\t}\n}\n\n// UnmarshalText tries to decode DLQType from a byte slice\n// containing its name.\n//\n//\tvar v DLQType\n//\terr := v.UnmarshalText([]byte(\"Replication\"))\nfunc (v *DLQType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Replication\":\n\t\t*v = DLQTypeReplication\n\t\treturn nil\n\tcase \"Domain\":\n\t\t*v = DLQTypeDomain\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DLQType\", err)\n\t\t}\n\t\t*v = DLQType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DLQType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DLQType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Replication\"), nil\n\tcase 1:\n\t\treturn []byte(\"Domain\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DLQType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DLQType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Replication\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Domain\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DLQType) Ptr() *DLQType {\n\treturn &v\n}\n\n// Encode encodes DLQType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DLQType\n//\treturn v.Encode(sWriter)\nfunc (v DLQType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DLQType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DLQType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DLQType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DLQType(0), err\n//\t}\n//\n//\tvar v DLQType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DLQType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DLQType) FromWire(w wire.Value) error {\n\t*v = (DLQType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DLQType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DLQType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DLQType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DLQType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DLQType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DLQType.\nfunc (v DLQType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Replication\"\n\tcase 1:\n\t\treturn \"Domain\"\n\t}\n\treturn fmt.Sprintf(\"DLQType(%d)\", w)\n}\n\n// Equals returns true if this DLQType value matches the provided\n// value.\nfunc (v DLQType) Equals(rhs DLQType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DLQType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DLQType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Replication\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Domain\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DLQType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DLQType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DLQType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DLQType\")\n\t\t}\n\t\t*v = (DLQType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DLQType\")\n\t}\n}\n\ntype DomainOperation int32\n\nconst (\n\tDomainOperationCreate DomainOperation = 0\n\tDomainOperationUpdate DomainOperation = 1\n\tDomainOperationDelete DomainOperation = 2\n)\n\n// DomainOperation_Values returns all recognized values of DomainOperation.\nfunc DomainOperation_Values() []DomainOperation {\n\treturn []DomainOperation{\n\t\tDomainOperationCreate,\n\t\tDomainOperationUpdate,\n\t\tDomainOperationDelete,\n\t}\n}\n\n// UnmarshalText tries to decode DomainOperation from a byte slice\n// containing its name.\n//\n//\tvar v DomainOperation\n//\terr := v.UnmarshalText([]byte(\"Create\"))\nfunc (v *DomainOperation) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Create\":\n\t\t*v = DomainOperationCreate\n\t\treturn nil\n\tcase \"Update\":\n\t\t*v = DomainOperationUpdate\n\t\treturn nil\n\tcase \"Delete\":\n\t\t*v = DomainOperationDelete\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DomainOperation\", err)\n\t\t}\n\t\t*v = DomainOperation(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DomainOperation to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DomainOperation) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Create\"), nil\n\tcase 1:\n\t\treturn []byte(\"Update\"), nil\n\tcase 2:\n\t\treturn []byte(\"Delete\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainOperation.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DomainOperation) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Create\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Update\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"Delete\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DomainOperation) Ptr() *DomainOperation {\n\treturn &v\n}\n\n// Encode encodes DomainOperation directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DomainOperation\n//\treturn v.Encode(sWriter)\nfunc (v DomainOperation) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DomainOperation into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DomainOperation) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DomainOperation from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DomainOperation(0), err\n//\t}\n//\n//\tvar v DomainOperation\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DomainOperation(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DomainOperation) FromWire(w wire.Value) error {\n\t*v = (DomainOperation)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DomainOperation directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DomainOperation\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DomainOperation(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DomainOperation) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DomainOperation)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DomainOperation.\nfunc (v DomainOperation) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Create\"\n\tcase 1:\n\t\treturn \"Update\"\n\tcase 2:\n\t\treturn \"Delete\"\n\t}\n\treturn fmt.Sprintf(\"DomainOperation(%d)\", w)\n}\n\n// Equals returns true if this DomainOperation value matches the provided\n// value.\nfunc (v DomainOperation) Equals(rhs DomainOperation) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DomainOperation into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DomainOperation) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Create\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Update\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"Delete\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DomainOperation from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DomainOperation) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DomainOperation\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DomainOperation\")\n\t\t}\n\t\t*v = (DomainOperation)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DomainOperation\")\n\t}\n}\n\ntype DomainTaskAttributes struct {\n\tDomainOperation         *DomainOperation                       `json:\"domainOperation,omitempty\"`\n\tID                      *string                                `json:\"id,omitempty\"`\n\tInfo                    *shared.DomainInfo                     `json:\"info,omitempty\"`\n\tConfig                  *shared.DomainConfiguration            `json:\"config,omitempty\"`\n\tReplicationConfig       *shared.DomainReplicationConfiguration `json:\"replicationConfig,omitempty\"`\n\tConfigVersion           *int64                                 `json:\"configVersion,omitempty\"`\n\tFailoverVersion         *int64                                 `json:\"failoverVersion,omitempty\"`\n\tPreviousFailoverVersion *int64                                 `json:\"previousFailoverVersion,omitempty\"`\n}\n\n// ToWire translates a DomainTaskAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainTaskAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainOperation != nil {\n\t\tw, err = v.DomainOperation.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ID != nil {\n\t\tw, err = wire.NewValueString(*(v.ID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Info != nil {\n\t\tw, err = v.Info.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Config != nil {\n\t\tw, err = v.Config.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationConfig != nil {\n\t\tw, err = v.ReplicationConfig.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ConfigVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.ConfigVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousFailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainOperation_Read(w wire.Value) (DomainOperation, error) {\n\tvar v DomainOperation\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _DomainInfo_Read(w wire.Value) (*shared.DomainInfo, error) {\n\tvar v shared.DomainInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainConfiguration_Read(w wire.Value) (*shared.DomainConfiguration, error) {\n\tvar v shared.DomainConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainReplicationConfiguration_Read(w wire.Value) (*shared.DomainReplicationConfiguration, error) {\n\tvar v shared.DomainReplicationConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DomainTaskAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainTaskAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainTaskAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainTaskAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DomainOperation\n\t\t\t\tx, err = _DomainOperation_Read(field.Value)\n\t\t\t\tv.DomainOperation = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Info, err = _DomainInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Config, err = _DomainConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicationConfig, err = _DomainReplicationConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ConfigVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousFailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainTaskAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainTaskAttributes struct could not be encoded.\nfunc (v *DomainTaskAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainOperation != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainOperation.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Info != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Info.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Config != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Config.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicationConfig.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ConfigVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ConfigVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousFailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousFailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainOperation_Decode(sr stream.Reader) (DomainOperation, error) {\n\tvar v DomainOperation\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _DomainInfo_Decode(sr stream.Reader) (*shared.DomainInfo, error) {\n\tvar v shared.DomainInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainConfiguration_Decode(sr stream.Reader) (*shared.DomainConfiguration, error) {\n\tvar v shared.DomainConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainReplicationConfiguration_Decode(sr stream.Reader) (*shared.DomainReplicationConfiguration, error) {\n\tvar v shared.DomainReplicationConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DomainTaskAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainTaskAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DomainTaskAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 5 && fh.Type == wire.TI32:\n\t\t\tvar x DomainOperation\n\t\t\tx, err = _DomainOperation_Decode(sr)\n\t\t\tv.DomainOperation = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Info, err = _DomainInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.Config, err = _DomainConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicationConfig, err = _DomainReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ConfigVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousFailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainTaskAttributes\n// struct.\nfunc (v *DomainTaskAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.DomainOperation != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainOperation: %v\", *(v.DomainOperation))\n\t\ti++\n\t}\n\tif v.ID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ID: %v\", *(v.ID))\n\t\ti++\n\t}\n\tif v.Info != nil {\n\t\tfields[i] = fmt.Sprintf(\"Info: %v\", v.Info)\n\t\ti++\n\t}\n\tif v.Config != nil {\n\t\tfields[i] = fmt.Sprintf(\"Config: %v\", v.Config)\n\t\ti++\n\t}\n\tif v.ReplicationConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationConfig: %v\", v.ReplicationConfig)\n\t\ti++\n\t}\n\tif v.ConfigVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigVersion: %v\", *(v.ConfigVersion))\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousFailoverVersion: %v\", *(v.PreviousFailoverVersion))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainTaskAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DomainOperation_EqualsPtr(lhs, rhs *DomainOperation) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DomainTaskAttributes match the\n// provided DomainTaskAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DomainTaskAttributes) Equals(rhs *DomainTaskAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DomainOperation_EqualsPtr(v.DomainOperation, rhs.DomainOperation) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ID, rhs.ID) {\n\t\treturn false\n\t}\n\tif !((v.Info == nil && rhs.Info == nil) || (v.Info != nil && rhs.Info != nil && v.Info.Equals(rhs.Info))) {\n\t\treturn false\n\t}\n\tif !((v.Config == nil && rhs.Config == nil) || (v.Config != nil && rhs.Config != nil && v.Config.Equals(rhs.Config))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationConfig == nil && rhs.ReplicationConfig == nil) || (v.ReplicationConfig != nil && rhs.ReplicationConfig != nil && v.ReplicationConfig.Equals(rhs.ReplicationConfig))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ConfigVersion, rhs.ConfigVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousFailoverVersion, rhs.PreviousFailoverVersion) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainTaskAttributes.\nfunc (v *DomainTaskAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainOperation != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainOperation\", *v.DomainOperation))\n\t}\n\tif v.ID != nil {\n\t\tenc.AddString(\"id\", *v.ID)\n\t}\n\tif v.Info != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"info\", v.Info))\n\t}\n\tif v.Config != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"config\", v.Config))\n\t}\n\tif v.ReplicationConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationConfig\", v.ReplicationConfig))\n\t}\n\tif v.ConfigVersion != nil {\n\t\tenc.AddInt64(\"configVersion\", *v.ConfigVersion)\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tenc.AddInt64(\"previousFailoverVersion\", *v.PreviousFailoverVersion)\n\t}\n\treturn err\n}\n\n// GetDomainOperation returns the value of DomainOperation if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetDomainOperation() (o DomainOperation) {\n\tif v != nil && v.DomainOperation != nil {\n\t\treturn *v.DomainOperation\n\t}\n\n\treturn\n}\n\n// IsSetDomainOperation returns true if DomainOperation is not nil.\nfunc (v *DomainTaskAttributes) IsSetDomainOperation() bool {\n\treturn v != nil && v.DomainOperation != nil\n}\n\n// GetID returns the value of ID if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetID() (o string) {\n\tif v != nil && v.ID != nil {\n\t\treturn *v.ID\n\t}\n\n\treturn\n}\n\n// IsSetID returns true if ID is not nil.\nfunc (v *DomainTaskAttributes) IsSetID() bool {\n\treturn v != nil && v.ID != nil\n}\n\n// GetInfo returns the value of Info if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetInfo() (o *shared.DomainInfo) {\n\tif v != nil && v.Info != nil {\n\t\treturn v.Info\n\t}\n\n\treturn\n}\n\n// IsSetInfo returns true if Info is not nil.\nfunc (v *DomainTaskAttributes) IsSetInfo() bool {\n\treturn v != nil && v.Info != nil\n}\n\n// GetConfig returns the value of Config if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetConfig() (o *shared.DomainConfiguration) {\n\tif v != nil && v.Config != nil {\n\t\treturn v.Config\n\t}\n\n\treturn\n}\n\n// IsSetConfig returns true if Config is not nil.\nfunc (v *DomainTaskAttributes) IsSetConfig() bool {\n\treturn v != nil && v.Config != nil\n}\n\n// GetReplicationConfig returns the value of ReplicationConfig if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetReplicationConfig() (o *shared.DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfig != nil {\n\t\treturn v.ReplicationConfig\n\t}\n\n\treturn\n}\n\n// IsSetReplicationConfig returns true if ReplicationConfig is not nil.\nfunc (v *DomainTaskAttributes) IsSetReplicationConfig() bool {\n\treturn v != nil && v.ReplicationConfig != nil\n}\n\n// GetConfigVersion returns the value of ConfigVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetConfigVersion() (o int64) {\n\tif v != nil && v.ConfigVersion != nil {\n\t\treturn *v.ConfigVersion\n\t}\n\n\treturn\n}\n\n// IsSetConfigVersion returns true if ConfigVersion is not nil.\nfunc (v *DomainTaskAttributes) IsSetConfigVersion() bool {\n\treturn v != nil && v.ConfigVersion != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *DomainTaskAttributes) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetPreviousFailoverVersion returns the value of PreviousFailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainTaskAttributes) GetPreviousFailoverVersion() (o int64) {\n\tif v != nil && v.PreviousFailoverVersion != nil {\n\t\treturn *v.PreviousFailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetPreviousFailoverVersion returns true if PreviousFailoverVersion is not nil.\nfunc (v *DomainTaskAttributes) IsSetPreviousFailoverVersion() bool {\n\treturn v != nil && v.PreviousFailoverVersion != nil\n}\n\ntype FailoverMarkerAttributes struct {\n\tDomainID        *string `json:\"domainID,omitempty\"`\n\tFailoverVersion *int64  `json:\"failoverVersion,omitempty\"`\n\tCreationTime    *int64  `json:\"creationTime,omitempty\"`\n}\n\n// ToWire translates a FailoverMarkerAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverMarkerAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreationTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FailoverMarkerAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverMarkerAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverMarkerAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverMarkerAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreationTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FailoverMarkerAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverMarkerAttributes struct could not be encoded.\nfunc (v *FailoverMarkerAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreationTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreationTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FailoverMarkerAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverMarkerAttributes struct could not be generated from the wire\n// representation.\nfunc (v *FailoverMarkerAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreationTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverMarkerAttributes\n// struct.\nfunc (v *FailoverMarkerAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreationTime: %v\", *(v.CreationTime))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverMarkerAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FailoverMarkerAttributes match the\n// provided FailoverMarkerAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverMarkerAttributes) Equals(rhs *FailoverMarkerAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreationTime, rhs.CreationTime) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverMarkerAttributes.\nfunc (v *FailoverMarkerAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.CreationTime != nil {\n\t\tenc.AddInt64(\"creationTime\", *v.CreationTime)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkerAttributes) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *FailoverMarkerAttributes) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkerAttributes) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *FailoverMarkerAttributes) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetCreationTime returns the value of CreationTime if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkerAttributes) GetCreationTime() (o int64) {\n\tif v != nil && v.CreationTime != nil {\n\t\treturn *v.CreationTime\n\t}\n\n\treturn\n}\n\n// IsSetCreationTime returns true if CreationTime is not nil.\nfunc (v *FailoverMarkerAttributes) IsSetCreationTime() bool {\n\treturn v != nil && v.CreationTime != nil\n}\n\ntype FailoverMarkers struct {\n\tFailoverMarkers []*FailoverMarkerAttributes `json:\"failoverMarkers,omitempty\"`\n}\n\ntype _List_FailoverMarkerAttributes_ValueList []*FailoverMarkerAttributes\n\nfunc (v _List_FailoverMarkerAttributes_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverMarkerAttributes', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_FailoverMarkerAttributes_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_FailoverMarkerAttributes_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_FailoverMarkerAttributes_ValueList) Close() {}\n\n// ToWire translates a FailoverMarkers struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverMarkers) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailoverMarkers != nil {\n\t\tw, err = wire.NewValueList(_List_FailoverMarkerAttributes_ValueList(v.FailoverMarkers)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverMarkerAttributes_Read(w wire.Value) (*FailoverMarkerAttributes, error) {\n\tvar v FailoverMarkerAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_FailoverMarkerAttributes_Read(l wire.ValueList) ([]*FailoverMarkerAttributes, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*FailoverMarkerAttributes, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _FailoverMarkerAttributes_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a FailoverMarkers struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverMarkers struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverMarkers\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverMarkers) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.FailoverMarkers, err = _List_FailoverMarkerAttributes_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_FailoverMarkerAttributes_Encode(val []*FailoverMarkerAttributes, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverMarkerAttributes', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a FailoverMarkers struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverMarkers struct could not be encoded.\nfunc (v *FailoverMarkers) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailoverMarkers != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_FailoverMarkerAttributes_Encode(v.FailoverMarkers, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverMarkerAttributes_Decode(sr stream.Reader) (*FailoverMarkerAttributes, error) {\n\tvar v FailoverMarkerAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_FailoverMarkerAttributes_Decode(sr stream.Reader) ([]*FailoverMarkerAttributes, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*FailoverMarkerAttributes, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _FailoverMarkerAttributes_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a FailoverMarkers struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverMarkers struct could not be generated from the wire\n// representation.\nfunc (v *FailoverMarkers) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.FailoverMarkers, err = _List_FailoverMarkerAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverMarkers\n// struct.\nfunc (v *FailoverMarkers) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.FailoverMarkers != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverMarkers: %v\", v.FailoverMarkers)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverMarkers{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_FailoverMarkerAttributes_Equals(lhs, rhs []*FailoverMarkerAttributes) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this FailoverMarkers match the\n// provided FailoverMarkers.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverMarkers) Equals(rhs *FailoverMarkers) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailoverMarkers == nil && rhs.FailoverMarkers == nil) || (v.FailoverMarkers != nil && rhs.FailoverMarkers != nil && _List_FailoverMarkerAttributes_Equals(v.FailoverMarkers, rhs.FailoverMarkers))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_FailoverMarkerAttributes_Zapper []*FailoverMarkerAttributes\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_FailoverMarkerAttributes_Zapper.\nfunc (l _List_FailoverMarkerAttributes_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverMarkers.\nfunc (v *FailoverMarkers) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailoverMarkers != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"failoverMarkers\", (_List_FailoverMarkerAttributes_Zapper)(v.FailoverMarkers)))\n\t}\n\treturn err\n}\n\n// GetFailoverMarkers returns the value of FailoverMarkers if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverMarkers) GetFailoverMarkers() (o []*FailoverMarkerAttributes) {\n\tif v != nil && v.FailoverMarkers != nil {\n\t\treturn v.FailoverMarkers\n\t}\n\n\treturn\n}\n\n// IsSetFailoverMarkers returns true if FailoverMarkers is not nil.\nfunc (v *FailoverMarkers) IsSetFailoverMarkers() bool {\n\treturn v != nil && v.FailoverMarkers != nil\n}\n\ntype GetDLQReplicationMessagesRequest struct {\n\tTaskInfos []*ReplicationTaskInfo `json:\"taskInfos,omitempty\"`\n}\n\ntype _List_ReplicationTaskInfo_ValueList []*ReplicationTaskInfo\n\nfunc (v _List_ReplicationTaskInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationTaskInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ReplicationTaskInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ReplicationTaskInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ReplicationTaskInfo_ValueList) Close() {}\n\n// ToWire translates a GetDLQReplicationMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDLQReplicationMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskInfos != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationTaskInfo_ValueList(v.TaskInfos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicationTaskInfo_Read(w wire.Value) (*ReplicationTaskInfo, error) {\n\tvar v ReplicationTaskInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ReplicationTaskInfo_Read(l wire.ValueList) ([]*ReplicationTaskInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ReplicationTaskInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ReplicationTaskInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetDLQReplicationMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDLQReplicationMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDLQReplicationMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDLQReplicationMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.TaskInfos, err = _List_ReplicationTaskInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ReplicationTaskInfo_Encode(val []*ReplicationTaskInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationTaskInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetDLQReplicationMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDLQReplicationMessagesRequest struct could not be encoded.\nfunc (v *GetDLQReplicationMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskInfos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationTaskInfo_Encode(v.TaskInfos, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicationTaskInfo_Decode(sr stream.Reader) (*ReplicationTaskInfo, error) {\n\tvar v ReplicationTaskInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ReplicationTaskInfo_Decode(sr stream.Reader) ([]*ReplicationTaskInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ReplicationTaskInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ReplicationTaskInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetDLQReplicationMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDLQReplicationMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetDLQReplicationMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.TaskInfos, err = _List_ReplicationTaskInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDLQReplicationMessagesRequest\n// struct.\nfunc (v *GetDLQReplicationMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.TaskInfos != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskInfos: %v\", v.TaskInfos)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDLQReplicationMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ReplicationTaskInfo_Equals(lhs, rhs []*ReplicationTaskInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetDLQReplicationMessagesRequest match the\n// provided GetDLQReplicationMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetDLQReplicationMessagesRequest) Equals(rhs *GetDLQReplicationMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskInfos == nil && rhs.TaskInfos == nil) || (v.TaskInfos != nil && rhs.TaskInfos != nil && _List_ReplicationTaskInfo_Equals(v.TaskInfos, rhs.TaskInfos))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ReplicationTaskInfo_Zapper []*ReplicationTaskInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ReplicationTaskInfo_Zapper.\nfunc (l _List_ReplicationTaskInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDLQReplicationMessagesRequest.\nfunc (v *GetDLQReplicationMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskInfos != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"taskInfos\", (_List_ReplicationTaskInfo_Zapper)(v.TaskInfos)))\n\t}\n\treturn err\n}\n\n// GetTaskInfos returns the value of TaskInfos if it is set or its\n// zero value if it is unset.\nfunc (v *GetDLQReplicationMessagesRequest) GetTaskInfos() (o []*ReplicationTaskInfo) {\n\tif v != nil && v.TaskInfos != nil {\n\t\treturn v.TaskInfos\n\t}\n\n\treturn\n}\n\n// IsSetTaskInfos returns true if TaskInfos is not nil.\nfunc (v *GetDLQReplicationMessagesRequest) IsSetTaskInfos() bool {\n\treturn v != nil && v.TaskInfos != nil\n}\n\ntype GetDLQReplicationMessagesResponse struct {\n\tReplicationTasks []*ReplicationTask `json:\"replicationTasks,omitempty\"`\n}\n\ntype _List_ReplicationTask_ValueList []*ReplicationTask\n\nfunc (v _List_ReplicationTask_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationTask', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ReplicationTask_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ReplicationTask_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ReplicationTask_ValueList) Close() {}\n\n// ToWire translates a GetDLQReplicationMessagesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDLQReplicationMessagesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ReplicationTasks != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationTask_ValueList(v.ReplicationTasks)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicationTask_Read(w wire.Value) (*ReplicationTask, error) {\n\tvar v ReplicationTask\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ReplicationTask_Read(l wire.ValueList) ([]*ReplicationTask, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ReplicationTask, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ReplicationTask_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetDLQReplicationMessagesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDLQReplicationMessagesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDLQReplicationMessagesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDLQReplicationMessagesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ReplicationTask_Encode(val []*ReplicationTask, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationTask', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetDLQReplicationMessagesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDLQReplicationMessagesResponse struct could not be encoded.\nfunc (v *GetDLQReplicationMessagesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ReplicationTasks != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationTask_Encode(v.ReplicationTasks, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicationTask_Decode(sr stream.Reader) (*ReplicationTask, error) {\n\tvar v ReplicationTask\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ReplicationTask_Decode(sr stream.Reader) ([]*ReplicationTask, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ReplicationTask, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ReplicationTask_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetDLQReplicationMessagesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDLQReplicationMessagesResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetDLQReplicationMessagesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDLQReplicationMessagesResponse\n// struct.\nfunc (v *GetDLQReplicationMessagesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ReplicationTasks != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationTasks: %v\", v.ReplicationTasks)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDLQReplicationMessagesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ReplicationTask_Equals(lhs, rhs []*ReplicationTask) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetDLQReplicationMessagesResponse match the\n// provided GetDLQReplicationMessagesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetDLQReplicationMessagesResponse) Equals(rhs *GetDLQReplicationMessagesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ReplicationTasks == nil && rhs.ReplicationTasks == nil) || (v.ReplicationTasks != nil && rhs.ReplicationTasks != nil && _List_ReplicationTask_Equals(v.ReplicationTasks, rhs.ReplicationTasks))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ReplicationTask_Zapper []*ReplicationTask\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ReplicationTask_Zapper.\nfunc (l _List_ReplicationTask_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDLQReplicationMessagesResponse.\nfunc (v *GetDLQReplicationMessagesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ReplicationTasks != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"replicationTasks\", (_List_ReplicationTask_Zapper)(v.ReplicationTasks)))\n\t}\n\treturn err\n}\n\n// GetReplicationTasks returns the value of ReplicationTasks if it is set or its\n// zero value if it is unset.\nfunc (v *GetDLQReplicationMessagesResponse) GetReplicationTasks() (o []*ReplicationTask) {\n\tif v != nil && v.ReplicationTasks != nil {\n\t\treturn v.ReplicationTasks\n\t}\n\n\treturn\n}\n\n// IsSetReplicationTasks returns true if ReplicationTasks is not nil.\nfunc (v *GetDLQReplicationMessagesResponse) IsSetReplicationTasks() bool {\n\treturn v != nil && v.ReplicationTasks != nil\n}\n\ntype GetDomainReplicationMessagesRequest struct {\n\tLastRetrievedMessageId *int64  `json:\"lastRetrievedMessageId,omitempty\"`\n\tLastProcessedMessageId *int64  `json:\"lastProcessedMessageId,omitempty\"`\n\tClusterName            *string `json:\"clusterName,omitempty\"`\n}\n\n// ToWire translates a GetDomainReplicationMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainReplicationMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.LastRetrievedMessageId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastRetrievedMessageId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastProcessedMessageId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetDomainReplicationMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainReplicationMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainReplicationMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainReplicationMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastRetrievedMessageId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastProcessedMessageId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainReplicationMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainReplicationMessagesRequest struct could not be encoded.\nfunc (v *GetDomainReplicationMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.LastRetrievedMessageId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastRetrievedMessageId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastProcessedMessageId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastProcessedMessageId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetDomainReplicationMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainReplicationMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainReplicationMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastRetrievedMessageId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastProcessedMessageId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainReplicationMessagesRequest\n// struct.\nfunc (v *GetDomainReplicationMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.LastRetrievedMessageId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastRetrievedMessageId: %v\", *(v.LastRetrievedMessageId))\n\t\ti++\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastProcessedMessageId: %v\", *(v.LastProcessedMessageId))\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainReplicationMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainReplicationMessagesRequest match the\n// provided GetDomainReplicationMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainReplicationMessagesRequest) Equals(rhs *GetDomainReplicationMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastRetrievedMessageId, rhs.LastRetrievedMessageId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastProcessedMessageId, rhs.LastProcessedMessageId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainReplicationMessagesRequest.\nfunc (v *GetDomainReplicationMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tenc.AddInt64(\"lastRetrievedMessageId\", *v.LastRetrievedMessageId)\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tenc.AddInt64(\"lastProcessedMessageId\", *v.LastProcessedMessageId)\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\treturn err\n}\n\n// GetLastRetrievedMessageId returns the value of LastRetrievedMessageId if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainReplicationMessagesRequest) GetLastRetrievedMessageId() (o int64) {\n\tif v != nil && v.LastRetrievedMessageId != nil {\n\t\treturn *v.LastRetrievedMessageId\n\t}\n\n\treturn\n}\n\n// IsSetLastRetrievedMessageId returns true if LastRetrievedMessageId is not nil.\nfunc (v *GetDomainReplicationMessagesRequest) IsSetLastRetrievedMessageId() bool {\n\treturn v != nil && v.LastRetrievedMessageId != nil\n}\n\n// GetLastProcessedMessageId returns the value of LastProcessedMessageId if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainReplicationMessagesRequest) GetLastProcessedMessageId() (o int64) {\n\tif v != nil && v.LastProcessedMessageId != nil {\n\t\treturn *v.LastProcessedMessageId\n\t}\n\n\treturn\n}\n\n// IsSetLastProcessedMessageId returns true if LastProcessedMessageId is not nil.\nfunc (v *GetDomainReplicationMessagesRequest) IsSetLastProcessedMessageId() bool {\n\treturn v != nil && v.LastProcessedMessageId != nil\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainReplicationMessagesRequest) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *GetDomainReplicationMessagesRequest) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\ntype GetDomainReplicationMessagesResponse struct {\n\tMessages *ReplicationMessages `json:\"messages,omitempty\"`\n}\n\n// ToWire translates a GetDomainReplicationMessagesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetDomainReplicationMessagesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Messages != nil {\n\t\tw, err = v.Messages.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicationMessages_Read(w wire.Value) (*ReplicationMessages, error) {\n\tvar v ReplicationMessages\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a GetDomainReplicationMessagesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetDomainReplicationMessagesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetDomainReplicationMessagesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetDomainReplicationMessagesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Messages, err = _ReplicationMessages_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetDomainReplicationMessagesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetDomainReplicationMessagesResponse struct could not be encoded.\nfunc (v *GetDomainReplicationMessagesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Messages != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Messages.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicationMessages_Decode(sr stream.Reader) (*ReplicationMessages, error) {\n\tvar v ReplicationMessages\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a GetDomainReplicationMessagesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetDomainReplicationMessagesResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetDomainReplicationMessagesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Messages, err = _ReplicationMessages_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetDomainReplicationMessagesResponse\n// struct.\nfunc (v *GetDomainReplicationMessagesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Messages != nil {\n\t\tfields[i] = fmt.Sprintf(\"Messages: %v\", v.Messages)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetDomainReplicationMessagesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetDomainReplicationMessagesResponse match the\n// provided GetDomainReplicationMessagesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetDomainReplicationMessagesResponse) Equals(rhs *GetDomainReplicationMessagesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Messages == nil && rhs.Messages == nil) || (v.Messages != nil && rhs.Messages != nil && v.Messages.Equals(rhs.Messages))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetDomainReplicationMessagesResponse.\nfunc (v *GetDomainReplicationMessagesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Messages != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"messages\", v.Messages))\n\t}\n\treturn err\n}\n\n// GetMessages returns the value of Messages if it is set or its\n// zero value if it is unset.\nfunc (v *GetDomainReplicationMessagesResponse) GetMessages() (o *ReplicationMessages) {\n\tif v != nil && v.Messages != nil {\n\t\treturn v.Messages\n\t}\n\n\treturn\n}\n\n// IsSetMessages returns true if Messages is not nil.\nfunc (v *GetDomainReplicationMessagesResponse) IsSetMessages() bool {\n\treturn v != nil && v.Messages != nil\n}\n\ntype GetReplicationMessagesRequest struct {\n\tTokens      []*ReplicationToken `json:\"tokens,omitempty\"`\n\tClusterName *string             `json:\"clusterName,omitempty\"`\n}\n\ntype _List_ReplicationToken_ValueList []*ReplicationToken\n\nfunc (v _List_ReplicationToken_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationToken', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ReplicationToken_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ReplicationToken_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ReplicationToken_ValueList) Close() {}\n\n// ToWire translates a GetReplicationMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetReplicationMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Tokens != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationToken_ValueList(v.Tokens)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicationToken_Read(w wire.Value) (*ReplicationToken, error) {\n\tvar v ReplicationToken\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ReplicationToken_Read(l wire.ValueList) ([]*ReplicationToken, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ReplicationToken, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ReplicationToken_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetReplicationMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetReplicationMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetReplicationMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetReplicationMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Tokens, err = _List_ReplicationToken_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ReplicationToken_Encode(val []*ReplicationToken, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ReplicationToken', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetReplicationMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetReplicationMessagesRequest struct could not be encoded.\nfunc (v *GetReplicationMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Tokens != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationToken_Encode(v.Tokens, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicationToken_Decode(sr stream.Reader) (*ReplicationToken, error) {\n\tvar v ReplicationToken\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ReplicationToken_Decode(sr stream.Reader) ([]*ReplicationToken, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ReplicationToken, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ReplicationToken_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetReplicationMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetReplicationMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetReplicationMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Tokens, err = _List_ReplicationToken_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetReplicationMessagesRequest\n// struct.\nfunc (v *GetReplicationMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Tokens != nil {\n\t\tfields[i] = fmt.Sprintf(\"Tokens: %v\", v.Tokens)\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetReplicationMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ReplicationToken_Equals(lhs, rhs []*ReplicationToken) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetReplicationMessagesRequest match the\n// provided GetReplicationMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetReplicationMessagesRequest) Equals(rhs *GetReplicationMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Tokens == nil && rhs.Tokens == nil) || (v.Tokens != nil && rhs.Tokens != nil && _List_ReplicationToken_Equals(v.Tokens, rhs.Tokens))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ReplicationToken_Zapper []*ReplicationToken\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ReplicationToken_Zapper.\nfunc (l _List_ReplicationToken_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetReplicationMessagesRequest.\nfunc (v *GetReplicationMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Tokens != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"tokens\", (_List_ReplicationToken_Zapper)(v.Tokens)))\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\treturn err\n}\n\n// GetTokens returns the value of Tokens if it is set or its\n// zero value if it is unset.\nfunc (v *GetReplicationMessagesRequest) GetTokens() (o []*ReplicationToken) {\n\tif v != nil && v.Tokens != nil {\n\t\treturn v.Tokens\n\t}\n\n\treturn\n}\n\n// IsSetTokens returns true if Tokens is not nil.\nfunc (v *GetReplicationMessagesRequest) IsSetTokens() bool {\n\treturn v != nil && v.Tokens != nil\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *GetReplicationMessagesRequest) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *GetReplicationMessagesRequest) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\ntype GetReplicationMessagesResponse struct {\n\tMessagesByShard map[int32]*ReplicationMessages `json:\"messagesByShard,omitempty\"`\n}\n\ntype _Map_I32_ReplicationMessages_MapItemList map[int32]*ReplicationMessages\n\nfunc (m _Map_I32_ReplicationMessages_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*ReplicationMessages', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_ReplicationMessages_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_ReplicationMessages_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_ReplicationMessages_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_I32_ReplicationMessages_MapItemList) Close() {}\n\n// ToWire translates a GetReplicationMessagesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetReplicationMessagesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MessagesByShard != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_ReplicationMessages_MapItemList(v.MessagesByShard)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_I32_ReplicationMessages_Read(m wire.MapItemList) (map[int32]*ReplicationMessages, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32]*ReplicationMessages, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _ReplicationMessages_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetReplicationMessagesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetReplicationMessagesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetReplicationMessagesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetReplicationMessagesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.MessagesByShard, err = _Map_I32_ReplicationMessages_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_I32_ReplicationMessages_Encode(val map[int32]*ReplicationMessages, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*ReplicationMessages', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a GetReplicationMessagesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetReplicationMessagesResponse struct could not be encoded.\nfunc (v *GetReplicationMessagesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MessagesByShard != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_ReplicationMessages_Encode(v.MessagesByShard, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_I32_ReplicationMessages_Decode(sr stream.Reader) (map[int32]*ReplicationMessages, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32]*ReplicationMessages, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _ReplicationMessages_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetReplicationMessagesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetReplicationMessagesResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetReplicationMessagesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.MessagesByShard, err = _Map_I32_ReplicationMessages_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetReplicationMessagesResponse\n// struct.\nfunc (v *GetReplicationMessagesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.MessagesByShard != nil {\n\t\tfields[i] = fmt.Sprintf(\"MessagesByShard: %v\", v.MessagesByShard)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetReplicationMessagesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_I32_ReplicationMessages_Equals(lhs, rhs map[int32]*ReplicationMessages) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetReplicationMessagesResponse match the\n// provided GetReplicationMessagesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetReplicationMessagesResponse) Equals(rhs *GetReplicationMessagesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.MessagesByShard == nil && rhs.MessagesByShard == nil) || (v.MessagesByShard != nil && rhs.MessagesByShard != nil && _Map_I32_ReplicationMessages_Equals(v.MessagesByShard, rhs.MessagesByShard))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_I32_ReplicationMessages_Item_Zapper struct {\n\tKey   int32\n\tValue *ReplicationMessages\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_ReplicationMessages_Item_Zapper.\nfunc (v _Map_I32_ReplicationMessages_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\treturn err\n}\n\ntype _Map_I32_ReplicationMessages_Zapper map[int32]*ReplicationMessages\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_ReplicationMessages_Zapper.\nfunc (m _Map_I32_ReplicationMessages_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_ReplicationMessages_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetReplicationMessagesResponse.\nfunc (v *GetReplicationMessagesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MessagesByShard != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"messagesByShard\", (_Map_I32_ReplicationMessages_Zapper)(v.MessagesByShard)))\n\t}\n\treturn err\n}\n\n// GetMessagesByShard returns the value of MessagesByShard if it is set or its\n// zero value if it is unset.\nfunc (v *GetReplicationMessagesResponse) GetMessagesByShard() (o map[int32]*ReplicationMessages) {\n\tif v != nil && v.MessagesByShard != nil {\n\t\treturn v.MessagesByShard\n\t}\n\n\treturn\n}\n\n// IsSetMessagesByShard returns true if MessagesByShard is not nil.\nfunc (v *GetReplicationMessagesResponse) IsSetMessagesByShard() bool {\n\treturn v != nil && v.MessagesByShard != nil\n}\n\ntype HistoryTaskV2Attributes struct {\n\tTaskId              *int64                       `json:\"taskId,omitempty\"`\n\tDomainId            *string                      `json:\"domainId,omitempty\"`\n\tWorkflowId          *string                      `json:\"workflowId,omitempty\"`\n\tRunId               *string                      `json:\"runId,omitempty\"`\n\tVersionHistoryItems []*shared.VersionHistoryItem `json:\"versionHistoryItems,omitempty\"`\n\tEvents              *shared.DataBlob             `json:\"events,omitempty\"`\n\tNewRunEvents        *shared.DataBlob             `json:\"newRunEvents,omitempty\"`\n}\n\ntype _List_VersionHistoryItem_ValueList []*shared.VersionHistoryItem\n\nfunc (v _List_VersionHistoryItem_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_VersionHistoryItem_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_VersionHistoryItem_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_VersionHistoryItem_ValueList) Close() {}\n\n// ToWire translates a HistoryTaskV2Attributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryTaskV2Attributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskId != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.DomainId != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\tw, err = wire.NewValueList(_List_VersionHistoryItem_ValueList(v.VersionHistoryItems)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tw, err = v.Events.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunEvents != nil {\n\t\tw, err = v.NewRunEvents.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistoryItem_Read(w wire.Value) (*shared.VersionHistoryItem, error) {\n\tvar v shared.VersionHistoryItem\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_VersionHistoryItem_Read(l wire.ValueList) ([]*shared.VersionHistoryItem, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*shared.VersionHistoryItem, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _VersionHistoryItem_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _DataBlob_Read(w wire.Value) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryTaskV2Attributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryTaskV2Attributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryTaskV2Attributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryTaskV2Attributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.VersionHistoryItems, err = _List_VersionHistoryItem_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Events, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.NewRunEvents, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_VersionHistoryItem_Encode(val []*shared.VersionHistoryItem, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a HistoryTaskV2Attributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryTaskV2Attributes struct could not be encoded.\nfunc (v *HistoryTaskV2Attributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistoryItems != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_VersionHistoryItem_Encode(v.VersionHistoryItems, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Events != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Events.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunEvents != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.NewRunEvents.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistoryItem_Decode(sr stream.Reader) (*shared.VersionHistoryItem, error) {\n\tvar v shared.VersionHistoryItem\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_VersionHistoryItem_Decode(sr stream.Reader) ([]*shared.VersionHistoryItem, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*shared.VersionHistoryItem, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _VersionHistoryItem_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _DataBlob_Decode(sr stream.Reader) (*shared.DataBlob, error) {\n\tvar v shared.DataBlob\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryTaskV2Attributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryTaskV2Attributes struct could not be generated from the wire\n// representation.\nfunc (v *HistoryTaskV2Attributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 5 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TList:\n\t\t\tv.VersionHistoryItems, err = _List_VersionHistoryItem_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.Events, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.NewRunEvents, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryTaskV2Attributes\n// struct.\nfunc (v *HistoryTaskV2Attributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.TaskId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskId: %v\", *(v.TaskId))\n\t\ti++\n\t}\n\tif v.DomainId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainId: %v\", *(v.DomainId))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistoryItems: %v\", v.VersionHistoryItems)\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tfields[i] = fmt.Sprintf(\"Events: %v\", v.Events)\n\t\ti++\n\t}\n\tif v.NewRunEvents != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunEvents: %v\", v.NewRunEvents)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryTaskV2Attributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_VersionHistoryItem_Equals(lhs, rhs []*shared.VersionHistoryItem) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this HistoryTaskV2Attributes match the\n// provided HistoryTaskV2Attributes.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryTaskV2Attributes) Equals(rhs *HistoryTaskV2Attributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskId, rhs.TaskId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainId, rhs.DomainId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistoryItems == nil && rhs.VersionHistoryItems == nil) || (v.VersionHistoryItems != nil && rhs.VersionHistoryItems != nil && _List_VersionHistoryItem_Equals(v.VersionHistoryItems, rhs.VersionHistoryItems))) {\n\t\treturn false\n\t}\n\tif !((v.Events == nil && rhs.Events == nil) || (v.Events != nil && rhs.Events != nil && v.Events.Equals(rhs.Events))) {\n\t\treturn false\n\t}\n\tif !((v.NewRunEvents == nil && rhs.NewRunEvents == nil) || (v.NewRunEvents != nil && rhs.NewRunEvents != nil && v.NewRunEvents.Equals(rhs.NewRunEvents))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_VersionHistoryItem_Zapper []*shared.VersionHistoryItem\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_VersionHistoryItem_Zapper.\nfunc (l _List_VersionHistoryItem_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryTaskV2Attributes.\nfunc (v *HistoryTaskV2Attributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskId != nil {\n\t\tenc.AddInt64(\"taskId\", *v.TaskId)\n\t}\n\tif v.DomainId != nil {\n\t\tenc.AddString(\"domainId\", *v.DomainId)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.VersionHistoryItems != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"versionHistoryItems\", (_List_VersionHistoryItem_Zapper)(v.VersionHistoryItems)))\n\t}\n\tif v.Events != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"events\", v.Events))\n\t}\n\tif v.NewRunEvents != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"newRunEvents\", v.NewRunEvents))\n\t}\n\treturn err\n}\n\n// GetTaskId returns the value of TaskId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetTaskId() (o int64) {\n\tif v != nil && v.TaskId != nil {\n\t\treturn *v.TaskId\n\t}\n\n\treturn\n}\n\n// IsSetTaskId returns true if TaskId is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetTaskId() bool {\n\treturn v != nil && v.TaskId != nil\n}\n\n// GetDomainId returns the value of DomainId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetDomainId() (o string) {\n\tif v != nil && v.DomainId != nil {\n\t\treturn *v.DomainId\n\t}\n\n\treturn\n}\n\n// IsSetDomainId returns true if DomainId is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetDomainId() bool {\n\treturn v != nil && v.DomainId != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetVersionHistoryItems returns the value of VersionHistoryItems if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetVersionHistoryItems() (o []*shared.VersionHistoryItem) {\n\tif v != nil && v.VersionHistoryItems != nil {\n\t\treturn v.VersionHistoryItems\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistoryItems returns true if VersionHistoryItems is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetVersionHistoryItems() bool {\n\treturn v != nil && v.VersionHistoryItems != nil\n}\n\n// GetEvents returns the value of Events if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetEvents() (o *shared.DataBlob) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\n\treturn\n}\n\n// IsSetEvents returns true if Events is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetEvents() bool {\n\treturn v != nil && v.Events != nil\n}\n\n// GetNewRunEvents returns the value of NewRunEvents if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTaskV2Attributes) GetNewRunEvents() (o *shared.DataBlob) {\n\tif v != nil && v.NewRunEvents != nil {\n\t\treturn v.NewRunEvents\n\t}\n\n\treturn\n}\n\n// IsSetNewRunEvents returns true if NewRunEvents is not nil.\nfunc (v *HistoryTaskV2Attributes) IsSetNewRunEvents() bool {\n\treturn v != nil && v.NewRunEvents != nil\n}\n\ntype MergeDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               *int32   `json:\"shardID,omitempty\"`\n\tSourceCluster         *string  `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n\tMaximumPageSize       *int32   `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken         []byte   `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a MergeDLQMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MergeDLQMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InclusiveEndMessageID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DLQType_Read(w wire.Value) (DLQType, error) {\n\tvar v DLQType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a MergeDLQMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MergeDLQMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MergeDLQMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MergeDLQMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DLQType\n\t\t\t\tx, err = _DLQType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InclusiveEndMessageID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MergeDLQMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MergeDLQMessagesRequest struct could not be encoded.\nfunc (v *MergeDLQMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SourceCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InclusiveEndMessageID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InclusiveEndMessageID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DLQType_Decode(sr stream.Reader) (DLQType, error) {\n\tvar v DLQType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a MergeDLQMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MergeDLQMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *MergeDLQMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x DLQType\n\t\t\tx, err = _DLQType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InclusiveEndMessageID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MergeDLQMessagesRequest\n// struct.\nfunc (v *MergeDLQMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceCluster: %v\", *(v.SourceCluster))\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InclusiveEndMessageID: %v\", *(v.InclusiveEndMessageID))\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MergeDLQMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DLQType_EqualsPtr(lhs, rhs *DLQType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this MergeDLQMessagesRequest match the\n// provided MergeDLQMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *MergeDLQMessagesRequest) Equals(rhs *MergeDLQMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DLQType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceCluster, rhs.SourceCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InclusiveEndMessageID, rhs.InclusiveEndMessageID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MergeDLQMessagesRequest.\nfunc (v *MergeDLQMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.SourceCluster != nil {\n\t\tenc.AddString(\"sourceCluster\", *v.SourceCluster)\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tenc.AddInt64(\"inclusiveEndMessageID\", *v.InclusiveEndMessageID)\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetSourceCluster returns the value of SourceCluster if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil && v.SourceCluster != nil {\n\t\treturn *v.SourceCluster\n\t}\n\n\treturn\n}\n\n// IsSetSourceCluster returns true if SourceCluster is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetSourceCluster() bool {\n\treturn v != nil && v.SourceCluster != nil\n}\n\n// GetInclusiveEndMessageID returns the value of InclusiveEndMessageID if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\n\treturn\n}\n\n// IsSetInclusiveEndMessageID returns true if InclusiveEndMessageID is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetInclusiveEndMessageID() bool {\n\treturn v != nil && v.InclusiveEndMessageID != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *MergeDLQMessagesRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype MergeDLQMessagesResponse struct {\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a MergeDLQMessagesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MergeDLQMessagesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a MergeDLQMessagesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MergeDLQMessagesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MergeDLQMessagesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MergeDLQMessagesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MergeDLQMessagesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MergeDLQMessagesResponse struct could not be encoded.\nfunc (v *MergeDLQMessagesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a MergeDLQMessagesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MergeDLQMessagesResponse struct could not be generated from the wire\n// representation.\nfunc (v *MergeDLQMessagesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MergeDLQMessagesResponse\n// struct.\nfunc (v *MergeDLQMessagesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MergeDLQMessagesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MergeDLQMessagesResponse match the\n// provided MergeDLQMessagesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *MergeDLQMessagesResponse) Equals(rhs *MergeDLQMessagesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MergeDLQMessagesResponse.\nfunc (v *MergeDLQMessagesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *MergeDLQMessagesResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *MergeDLQMessagesResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype PurgeDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               *int32   `json:\"shardID,omitempty\"`\n\tSourceCluster         *string  `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n}\n\n// ToWire translates a PurgeDLQMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PurgeDLQMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InclusiveEndMessageID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PurgeDLQMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PurgeDLQMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PurgeDLQMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PurgeDLQMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DLQType\n\t\t\t\tx, err = _DLQType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InclusiveEndMessageID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PurgeDLQMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PurgeDLQMessagesRequest struct could not be encoded.\nfunc (v *PurgeDLQMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SourceCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InclusiveEndMessageID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InclusiveEndMessageID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PurgeDLQMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PurgeDLQMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *PurgeDLQMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x DLQType\n\t\t\tx, err = _DLQType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InclusiveEndMessageID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PurgeDLQMessagesRequest\n// struct.\nfunc (v *PurgeDLQMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceCluster: %v\", *(v.SourceCluster))\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InclusiveEndMessageID: %v\", *(v.InclusiveEndMessageID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PurgeDLQMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PurgeDLQMessagesRequest match the\n// provided PurgeDLQMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PurgeDLQMessagesRequest) Equals(rhs *PurgeDLQMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DLQType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceCluster, rhs.SourceCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InclusiveEndMessageID, rhs.InclusiveEndMessageID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PurgeDLQMessagesRequest.\nfunc (v *PurgeDLQMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.SourceCluster != nil {\n\t\tenc.AddString(\"sourceCluster\", *v.SourceCluster)\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tenc.AddInt64(\"inclusiveEndMessageID\", *v.InclusiveEndMessageID)\n\t}\n\treturn err\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *PurgeDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *PurgeDLQMessagesRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *PurgeDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *PurgeDLQMessagesRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetSourceCluster returns the value of SourceCluster if it is set or its\n// zero value if it is unset.\nfunc (v *PurgeDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil && v.SourceCluster != nil {\n\t\treturn *v.SourceCluster\n\t}\n\n\treturn\n}\n\n// IsSetSourceCluster returns true if SourceCluster is not nil.\nfunc (v *PurgeDLQMessagesRequest) IsSetSourceCluster() bool {\n\treturn v != nil && v.SourceCluster != nil\n}\n\n// GetInclusiveEndMessageID returns the value of InclusiveEndMessageID if it is set or its\n// zero value if it is unset.\nfunc (v *PurgeDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\n\treturn\n}\n\n// IsSetInclusiveEndMessageID returns true if InclusiveEndMessageID is not nil.\nfunc (v *PurgeDLQMessagesRequest) IsSetInclusiveEndMessageID() bool {\n\treturn v != nil && v.InclusiveEndMessageID != nil\n}\n\ntype ReadDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               *int32   `json:\"shardID,omitempty\"`\n\tSourceCluster         *string  `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n\tMaximumPageSize       *int32   `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken         []byte   `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ReadDLQMessagesRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReadDLQMessagesRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InclusiveEndMessageID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReadDLQMessagesRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReadDLQMessagesRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReadDLQMessagesRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReadDLQMessagesRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DLQType\n\t\t\t\tx, err = _DLQType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InclusiveEndMessageID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReadDLQMessagesRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReadDLQMessagesRequest struct could not be encoded.\nfunc (v *ReadDLQMessagesRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SourceCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InclusiveEndMessageID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InclusiveEndMessageID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReadDLQMessagesRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReadDLQMessagesRequest struct could not be generated from the wire\n// representation.\nfunc (v *ReadDLQMessagesRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x DLQType\n\t\t\tx, err = _DLQType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InclusiveEndMessageID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReadDLQMessagesRequest\n// struct.\nfunc (v *ReadDLQMessagesRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.SourceCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceCluster: %v\", *(v.SourceCluster))\n\t\ti++\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InclusiveEndMessageID: %v\", *(v.InclusiveEndMessageID))\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReadDLQMessagesRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReadDLQMessagesRequest match the\n// provided ReadDLQMessagesRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ReadDLQMessagesRequest) Equals(rhs *ReadDLQMessagesRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DLQType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceCluster, rhs.SourceCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InclusiveEndMessageID, rhs.InclusiveEndMessageID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReadDLQMessagesRequest.\nfunc (v *ReadDLQMessagesRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.SourceCluster != nil {\n\t\tenc.AddString(\"sourceCluster\", *v.SourceCluster)\n\t}\n\tif v.InclusiveEndMessageID != nil {\n\t\tenc.AddInt64(\"inclusiveEndMessageID\", *v.InclusiveEndMessageID)\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetSourceCluster returns the value of SourceCluster if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil && v.SourceCluster != nil {\n\t\treturn *v.SourceCluster\n\t}\n\n\treturn\n}\n\n// IsSetSourceCluster returns true if SourceCluster is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetSourceCluster() bool {\n\treturn v != nil && v.SourceCluster != nil\n}\n\n// GetInclusiveEndMessageID returns the value of InclusiveEndMessageID if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\n\treturn\n}\n\n// IsSetInclusiveEndMessageID returns true if InclusiveEndMessageID is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetInclusiveEndMessageID() bool {\n\treturn v != nil && v.InclusiveEndMessageID != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ReadDLQMessagesRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ReadDLQMessagesResponse struct {\n\tType                 *DLQType               `json:\"type,omitempty\"`\n\tReplicationTasks     []*ReplicationTask     `json:\"replicationTasks,omitempty\"`\n\tNextPageToken        []byte                 `json:\"nextPageToken,omitempty\"`\n\tReplicationTasksInfo []*ReplicationTaskInfo `json:\"replicationTasksInfo,omitempty\"`\n}\n\n// ToWire translates a ReadDLQMessagesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReadDLQMessagesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationTasks != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationTask_ValueList(v.ReplicationTasks)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationTasksInfo != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationTaskInfo_ValueList(v.ReplicationTasksInfo)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReadDLQMessagesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReadDLQMessagesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReadDLQMessagesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReadDLQMessagesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DLQType\n\t\t\t\tx, err = _DLQType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ReplicationTasksInfo, err = _List_ReplicationTaskInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReadDLQMessagesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReadDLQMessagesResponse struct could not be encoded.\nfunc (v *ReadDLQMessagesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationTasks != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationTask_Encode(v.ReplicationTasks, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationTasksInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationTaskInfo_Encode(v.ReplicationTasksInfo, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReadDLQMessagesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReadDLQMessagesResponse struct could not be generated from the wire\n// representation.\nfunc (v *ReadDLQMessagesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x DLQType\n\t\t\tx, err = _DLQType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TList:\n\t\t\tv.ReplicationTasksInfo, err = _List_ReplicationTaskInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReadDLQMessagesResponse\n// struct.\nfunc (v *ReadDLQMessagesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.ReplicationTasks != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationTasks: %v\", v.ReplicationTasks)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.ReplicationTasksInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationTasksInfo: %v\", v.ReplicationTasksInfo)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReadDLQMessagesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReadDLQMessagesResponse match the\n// provided ReadDLQMessagesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ReadDLQMessagesResponse) Equals(rhs *ReadDLQMessagesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DLQType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationTasks == nil && rhs.ReplicationTasks == nil) || (v.ReplicationTasks != nil && rhs.ReplicationTasks != nil && _List_ReplicationTask_Equals(v.ReplicationTasks, rhs.ReplicationTasks))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationTasksInfo == nil && rhs.ReplicationTasksInfo == nil) || (v.ReplicationTasksInfo != nil && rhs.ReplicationTasksInfo != nil && _List_ReplicationTaskInfo_Equals(v.ReplicationTasksInfo, rhs.ReplicationTasksInfo))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReadDLQMessagesResponse.\nfunc (v *ReadDLQMessagesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.ReplicationTasks != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"replicationTasks\", (_List_ReplicationTask_Zapper)(v.ReplicationTasks)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.ReplicationTasksInfo != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"replicationTasksInfo\", (_List_ReplicationTaskInfo_Zapper)(v.ReplicationTasksInfo)))\n\t}\n\treturn err\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesResponse) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *ReadDLQMessagesResponse) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetReplicationTasks returns the value of ReplicationTasks if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesResponse) GetReplicationTasks() (o []*ReplicationTask) {\n\tif v != nil && v.ReplicationTasks != nil {\n\t\treturn v.ReplicationTasks\n\t}\n\n\treturn\n}\n\n// IsSetReplicationTasks returns true if ReplicationTasks is not nil.\nfunc (v *ReadDLQMessagesResponse) IsSetReplicationTasks() bool {\n\treturn v != nil && v.ReplicationTasks != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ReadDLQMessagesResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetReplicationTasksInfo returns the value of ReplicationTasksInfo if it is set or its\n// zero value if it is unset.\nfunc (v *ReadDLQMessagesResponse) GetReplicationTasksInfo() (o []*ReplicationTaskInfo) {\n\tif v != nil && v.ReplicationTasksInfo != nil {\n\t\treturn v.ReplicationTasksInfo\n\t}\n\n\treturn\n}\n\n// IsSetReplicationTasksInfo returns true if ReplicationTasksInfo is not nil.\nfunc (v *ReadDLQMessagesResponse) IsSetReplicationTasksInfo() bool {\n\treturn v != nil && v.ReplicationTasksInfo != nil\n}\n\ntype ReplicationMessages struct {\n\tReplicationTasks       []*ReplicationTask `json:\"replicationTasks,omitempty\"`\n\tLastRetrievedMessageId *int64             `json:\"lastRetrievedMessageId,omitempty\"`\n\tHasMore                *bool              `json:\"hasMore,omitempty\"`\n\tSyncShardStatus        *SyncShardStatus   `json:\"syncShardStatus,omitempty\"`\n}\n\n// ToWire translates a ReplicationMessages struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicationMessages) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ReplicationTasks != nil {\n\t\tw, err = wire.NewValueList(_List_ReplicationTask_ValueList(v.ReplicationTasks)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastRetrievedMessageId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.HasMore != nil {\n\t\tw, err = wire.NewValueBool(*(v.HasMore)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.SyncShardStatus != nil {\n\t\tw, err = v.SyncShardStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SyncShardStatus_Read(w wire.Value) (*SyncShardStatus, error) {\n\tvar v SyncShardStatus\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ReplicationMessages struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicationMessages struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicationMessages\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicationMessages) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastRetrievedMessageId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.HasMore = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SyncShardStatus, err = _SyncShardStatus_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplicationMessages struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicationMessages struct could not be encoded.\nfunc (v *ReplicationMessages) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ReplicationTasks != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ReplicationTask_Encode(v.ReplicationTasks, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastRetrievedMessageId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastRetrievedMessageId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HasMore != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.HasMore)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SyncShardStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SyncShardStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SyncShardStatus_Decode(sr stream.Reader) (*SyncShardStatus, error) {\n\tvar v SyncShardStatus\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ReplicationMessages struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicationMessages struct could not be generated from the wire\n// representation.\nfunc (v *ReplicationMessages) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ReplicationTasks, err = _List_ReplicationTask_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastRetrievedMessageId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.HasMore = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.SyncShardStatus, err = _SyncShardStatus_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicationMessages\n// struct.\nfunc (v *ReplicationMessages) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.ReplicationTasks != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationTasks: %v\", v.ReplicationTasks)\n\t\ti++\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastRetrievedMessageId: %v\", *(v.LastRetrievedMessageId))\n\t\ti++\n\t}\n\tif v.HasMore != nil {\n\t\tfields[i] = fmt.Sprintf(\"HasMore: %v\", *(v.HasMore))\n\t\ti++\n\t}\n\tif v.SyncShardStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"SyncShardStatus: %v\", v.SyncShardStatus)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicationMessages{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ReplicationMessages match the\n// provided ReplicationMessages.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicationMessages) Equals(rhs *ReplicationMessages) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ReplicationTasks == nil && rhs.ReplicationTasks == nil) || (v.ReplicationTasks != nil && rhs.ReplicationTasks != nil && _List_ReplicationTask_Equals(v.ReplicationTasks, rhs.ReplicationTasks))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastRetrievedMessageId, rhs.LastRetrievedMessageId) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.HasMore, rhs.HasMore) {\n\t\treturn false\n\t}\n\tif !((v.SyncShardStatus == nil && rhs.SyncShardStatus == nil) || (v.SyncShardStatus != nil && rhs.SyncShardStatus != nil && v.SyncShardStatus.Equals(rhs.SyncShardStatus))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationMessages.\nfunc (v *ReplicationMessages) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ReplicationTasks != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"replicationTasks\", (_List_ReplicationTask_Zapper)(v.ReplicationTasks)))\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tenc.AddInt64(\"lastRetrievedMessageId\", *v.LastRetrievedMessageId)\n\t}\n\tif v.HasMore != nil {\n\t\tenc.AddBool(\"hasMore\", *v.HasMore)\n\t}\n\tif v.SyncShardStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"syncShardStatus\", v.SyncShardStatus))\n\t}\n\treturn err\n}\n\n// GetReplicationTasks returns the value of ReplicationTasks if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationMessages) GetReplicationTasks() (o []*ReplicationTask) {\n\tif v != nil && v.ReplicationTasks != nil {\n\t\treturn v.ReplicationTasks\n\t}\n\n\treturn\n}\n\n// IsSetReplicationTasks returns true if ReplicationTasks is not nil.\nfunc (v *ReplicationMessages) IsSetReplicationTasks() bool {\n\treturn v != nil && v.ReplicationTasks != nil\n}\n\n// GetLastRetrievedMessageId returns the value of LastRetrievedMessageId if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationMessages) GetLastRetrievedMessageId() (o int64) {\n\tif v != nil && v.LastRetrievedMessageId != nil {\n\t\treturn *v.LastRetrievedMessageId\n\t}\n\n\treturn\n}\n\n// IsSetLastRetrievedMessageId returns true if LastRetrievedMessageId is not nil.\nfunc (v *ReplicationMessages) IsSetLastRetrievedMessageId() bool {\n\treturn v != nil && v.LastRetrievedMessageId != nil\n}\n\n// GetHasMore returns the value of HasMore if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationMessages) GetHasMore() (o bool) {\n\tif v != nil && v.HasMore != nil {\n\t\treturn *v.HasMore\n\t}\n\n\treturn\n}\n\n// IsSetHasMore returns true if HasMore is not nil.\nfunc (v *ReplicationMessages) IsSetHasMore() bool {\n\treturn v != nil && v.HasMore != nil\n}\n\n// GetSyncShardStatus returns the value of SyncShardStatus if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationMessages) GetSyncShardStatus() (o *SyncShardStatus) {\n\tif v != nil && v.SyncShardStatus != nil {\n\t\treturn v.SyncShardStatus\n\t}\n\n\treturn\n}\n\n// IsSetSyncShardStatus returns true if SyncShardStatus is not nil.\nfunc (v *ReplicationMessages) IsSetSyncShardStatus() bool {\n\treturn v != nil && v.SyncShardStatus != nil\n}\n\ntype ReplicationTask struct {\n\tTaskType                      *ReplicationTaskType           `json:\"taskType,omitempty\"`\n\tSourceTaskId                  *int64                         `json:\"sourceTaskId,omitempty\"`\n\tDomainTaskAttributes          *DomainTaskAttributes          `json:\"domainTaskAttributes,omitempty\"`\n\tSyncShardStatusTaskAttributes *SyncShardStatusTaskAttributes `json:\"syncShardStatusTaskAttributes,omitempty\"`\n\tSyncActivityTaskAttributes    *SyncActivityTaskAttributes    `json:\"syncActivityTaskAttributes,omitempty\"`\n\tHistoryTaskV2Attributes       *HistoryTaskV2Attributes       `json:\"historyTaskV2Attributes,omitempty\"`\n\tFailoverMarkerAttributes      *FailoverMarkerAttributes      `json:\"failoverMarkerAttributes,omitempty\"`\n\tCreationTime                  *int64                         `json:\"creationTime,omitempty\"`\n}\n\n// ToWire translates a ReplicationTask struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicationTask) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskType != nil {\n\t\tw, err = v.TaskType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SourceTaskId != nil {\n\t\tw, err = wire.NewValueI64(*(v.SourceTaskId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\tif v.DomainTaskAttributes != nil {\n\t\tw, err = v.DomainTaskAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SyncShardStatusTaskAttributes != nil {\n\t\tw, err = v.SyncShardStatusTaskAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.SyncActivityTaskAttributes != nil {\n\t\tw, err = v.SyncActivityTaskAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryTaskV2Attributes != nil {\n\t\tw, err = v.HistoryTaskV2Attributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverMarkerAttributes != nil {\n\t\tw, err = v.FailoverMarkerAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreationTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ReplicationTaskType_Read(w wire.Value) (ReplicationTaskType, error) {\n\tvar v ReplicationTaskType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _DomainTaskAttributes_Read(w wire.Value) (*DomainTaskAttributes, error) {\n\tvar v DomainTaskAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SyncShardStatusTaskAttributes_Read(w wire.Value) (*SyncShardStatusTaskAttributes, error) {\n\tvar v SyncShardStatusTaskAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SyncActivityTaskAttributes_Read(w wire.Value) (*SyncActivityTaskAttributes, error) {\n\tvar v SyncActivityTaskAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _HistoryTaskV2Attributes_Read(w wire.Value) (*HistoryTaskV2Attributes, error) {\n\tvar v HistoryTaskV2Attributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ReplicationTask struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicationTask struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicationTask\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicationTask) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ReplicationTaskType\n\t\t\t\tx, err = _ReplicationTaskType_Read(field.Value)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.SourceTaskId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainTaskAttributes, err = _DomainTaskAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SyncShardStatusTaskAttributes, err = _SyncShardStatusTaskAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SyncActivityTaskAttributes, err = _SyncActivityTaskAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.HistoryTaskV2Attributes, err = _HistoryTaskV2Attributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailoverMarkerAttributes, err = _FailoverMarkerAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreationTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplicationTask struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicationTask struct could not be encoded.\nfunc (v *ReplicationTask) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SourceTaskId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.SourceTaskId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainTaskAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainTaskAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SyncShardStatusTaskAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SyncShardStatusTaskAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SyncActivityTaskAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SyncActivityTaskAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryTaskV2Attributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HistoryTaskV2Attributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverMarkerAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailoverMarkerAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreationTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreationTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ReplicationTaskType_Decode(sr stream.Reader) (ReplicationTaskType, error) {\n\tvar v ReplicationTaskType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _DomainTaskAttributes_Decode(sr stream.Reader) (*DomainTaskAttributes, error) {\n\tvar v DomainTaskAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SyncShardStatusTaskAttributes_Decode(sr stream.Reader) (*SyncShardStatusTaskAttributes, error) {\n\tvar v SyncShardStatusTaskAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SyncActivityTaskAttributes_Decode(sr stream.Reader) (*SyncActivityTaskAttributes, error) {\n\tvar v SyncActivityTaskAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _HistoryTaskV2Attributes_Decode(sr stream.Reader) (*HistoryTaskV2Attributes, error) {\n\tvar v HistoryTaskV2Attributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ReplicationTask struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicationTask struct could not be generated from the wire\n// representation.\nfunc (v *ReplicationTask) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x ReplicationTaskType\n\t\t\tx, err = _ReplicationTaskType_Decode(sr)\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.SourceTaskId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.DomainTaskAttributes, err = _DomainTaskAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.SyncShardStatusTaskAttributes, err = _SyncShardStatusTaskAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.SyncActivityTaskAttributes, err = _SyncActivityTaskAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.HistoryTaskV2Attributes, err = _HistoryTaskV2Attributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.FailoverMarkerAttributes, err = _FailoverMarkerAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreationTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicationTask\n// struct.\nfunc (v *ReplicationTask) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.SourceTaskId != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceTaskId: %v\", *(v.SourceTaskId))\n\t\ti++\n\t}\n\tif v.DomainTaskAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainTaskAttributes: %v\", v.DomainTaskAttributes)\n\t\ti++\n\t}\n\tif v.SyncShardStatusTaskAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SyncShardStatusTaskAttributes: %v\", v.SyncShardStatusTaskAttributes)\n\t\ti++\n\t}\n\tif v.SyncActivityTaskAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SyncActivityTaskAttributes: %v\", v.SyncActivityTaskAttributes)\n\t\ti++\n\t}\n\tif v.HistoryTaskV2Attributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryTaskV2Attributes: %v\", v.HistoryTaskV2Attributes)\n\t\ti++\n\t}\n\tif v.FailoverMarkerAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverMarkerAttributes: %v\", v.FailoverMarkerAttributes)\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreationTime: %v\", *(v.CreationTime))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicationTask{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ReplicationTaskType_EqualsPtr(lhs, rhs *ReplicationTaskType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ReplicationTask match the\n// provided ReplicationTask.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicationTask) Equals(rhs *ReplicationTask) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_ReplicationTaskType_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.SourceTaskId, rhs.SourceTaskId) {\n\t\treturn false\n\t}\n\tif !((v.DomainTaskAttributes == nil && rhs.DomainTaskAttributes == nil) || (v.DomainTaskAttributes != nil && rhs.DomainTaskAttributes != nil && v.DomainTaskAttributes.Equals(rhs.DomainTaskAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SyncShardStatusTaskAttributes == nil && rhs.SyncShardStatusTaskAttributes == nil) || (v.SyncShardStatusTaskAttributes != nil && rhs.SyncShardStatusTaskAttributes != nil && v.SyncShardStatusTaskAttributes.Equals(rhs.SyncShardStatusTaskAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SyncActivityTaskAttributes == nil && rhs.SyncActivityTaskAttributes == nil) || (v.SyncActivityTaskAttributes != nil && rhs.SyncActivityTaskAttributes != nil && v.SyncActivityTaskAttributes.Equals(rhs.SyncActivityTaskAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.HistoryTaskV2Attributes == nil && rhs.HistoryTaskV2Attributes == nil) || (v.HistoryTaskV2Attributes != nil && rhs.HistoryTaskV2Attributes != nil && v.HistoryTaskV2Attributes.Equals(rhs.HistoryTaskV2Attributes))) {\n\t\treturn false\n\t}\n\tif !((v.FailoverMarkerAttributes == nil && rhs.FailoverMarkerAttributes == nil) || (v.FailoverMarkerAttributes != nil && rhs.FailoverMarkerAttributes != nil && v.FailoverMarkerAttributes.Equals(rhs.FailoverMarkerAttributes))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreationTime, rhs.CreationTime) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationTask.\nfunc (v *ReplicationTask) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskType\", *v.TaskType))\n\t}\n\tif v.SourceTaskId != nil {\n\t\tenc.AddInt64(\"sourceTaskId\", *v.SourceTaskId)\n\t}\n\tif v.DomainTaskAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainTaskAttributes\", v.DomainTaskAttributes))\n\t}\n\tif v.SyncShardStatusTaskAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"syncShardStatusTaskAttributes\", v.SyncShardStatusTaskAttributes))\n\t}\n\tif v.SyncActivityTaskAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"syncActivityTaskAttributes\", v.SyncActivityTaskAttributes))\n\t}\n\tif v.HistoryTaskV2Attributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"historyTaskV2Attributes\", v.HistoryTaskV2Attributes))\n\t}\n\tif v.FailoverMarkerAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failoverMarkerAttributes\", v.FailoverMarkerAttributes))\n\t}\n\tif v.CreationTime != nil {\n\t\tenc.AddInt64(\"creationTime\", *v.CreationTime)\n\t}\n\treturn err\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetTaskType() (o ReplicationTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *ReplicationTask) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetSourceTaskId returns the value of SourceTaskId if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetSourceTaskId() (o int64) {\n\tif v != nil && v.SourceTaskId != nil {\n\t\treturn *v.SourceTaskId\n\t}\n\n\treturn\n}\n\n// IsSetSourceTaskId returns true if SourceTaskId is not nil.\nfunc (v *ReplicationTask) IsSetSourceTaskId() bool {\n\treturn v != nil && v.SourceTaskId != nil\n}\n\n// GetDomainTaskAttributes returns the value of DomainTaskAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetDomainTaskAttributes() (o *DomainTaskAttributes) {\n\tif v != nil && v.DomainTaskAttributes != nil {\n\t\treturn v.DomainTaskAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDomainTaskAttributes returns true if DomainTaskAttributes is not nil.\nfunc (v *ReplicationTask) IsSetDomainTaskAttributes() bool {\n\treturn v != nil && v.DomainTaskAttributes != nil\n}\n\n// GetSyncShardStatusTaskAttributes returns the value of SyncShardStatusTaskAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetSyncShardStatusTaskAttributes() (o *SyncShardStatusTaskAttributes) {\n\tif v != nil && v.SyncShardStatusTaskAttributes != nil {\n\t\treturn v.SyncShardStatusTaskAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSyncShardStatusTaskAttributes returns true if SyncShardStatusTaskAttributes is not nil.\nfunc (v *ReplicationTask) IsSetSyncShardStatusTaskAttributes() bool {\n\treturn v != nil && v.SyncShardStatusTaskAttributes != nil\n}\n\n// GetSyncActivityTaskAttributes returns the value of SyncActivityTaskAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetSyncActivityTaskAttributes() (o *SyncActivityTaskAttributes) {\n\tif v != nil && v.SyncActivityTaskAttributes != nil {\n\t\treturn v.SyncActivityTaskAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSyncActivityTaskAttributes returns true if SyncActivityTaskAttributes is not nil.\nfunc (v *ReplicationTask) IsSetSyncActivityTaskAttributes() bool {\n\treturn v != nil && v.SyncActivityTaskAttributes != nil\n}\n\n// GetHistoryTaskV2Attributes returns the value of HistoryTaskV2Attributes if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetHistoryTaskV2Attributes() (o *HistoryTaskV2Attributes) {\n\tif v != nil && v.HistoryTaskV2Attributes != nil {\n\t\treturn v.HistoryTaskV2Attributes\n\t}\n\n\treturn\n}\n\n// IsSetHistoryTaskV2Attributes returns true if HistoryTaskV2Attributes is not nil.\nfunc (v *ReplicationTask) IsSetHistoryTaskV2Attributes() bool {\n\treturn v != nil && v.HistoryTaskV2Attributes != nil\n}\n\n// GetFailoverMarkerAttributes returns the value of FailoverMarkerAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetFailoverMarkerAttributes() (o *FailoverMarkerAttributes) {\n\tif v != nil && v.FailoverMarkerAttributes != nil {\n\t\treturn v.FailoverMarkerAttributes\n\t}\n\n\treturn\n}\n\n// IsSetFailoverMarkerAttributes returns true if FailoverMarkerAttributes is not nil.\nfunc (v *ReplicationTask) IsSetFailoverMarkerAttributes() bool {\n\treturn v != nil && v.FailoverMarkerAttributes != nil\n}\n\n// GetCreationTime returns the value of CreationTime if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTask) GetCreationTime() (o int64) {\n\tif v != nil && v.CreationTime != nil {\n\t\treturn *v.CreationTime\n\t}\n\n\treturn\n}\n\n// IsSetCreationTime returns true if CreationTime is not nil.\nfunc (v *ReplicationTask) IsSetCreationTime() bool {\n\treturn v != nil && v.CreationTime != nil\n}\n\ntype ReplicationTaskInfo struct {\n\tDomainID     *string `json:\"domainID,omitempty\"`\n\tWorkflowID   *string `json:\"workflowID,omitempty\"`\n\tRunID        *string `json:\"runID,omitempty\"`\n\tTaskType     *int16  `json:\"taskType,omitempty\"`\n\tTaskID       *int64  `json:\"taskID,omitempty\"`\n\tVersion      *int64  `json:\"version,omitempty\"`\n\tFirstEventID *int64  `json:\"firstEventID,omitempty\"`\n\tNextEventID  *int64  `json:\"nextEventID,omitempty\"`\n\tScheduledID  *int64  `json:\"scheduledID,omitempty\"`\n}\n\n// ToWire translates a ReplicationTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicationTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.FirstEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReplicationTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicationTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicationTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicationTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplicationTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicationTaskInfo struct could not be encoded.\nfunc (v *ReplicationTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReplicationTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicationTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *ReplicationTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicationTaskInfo\n// struct.\nfunc (v *ReplicationTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.FirstEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstEventID: %v\", *(v.FirstEventID))\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventID: %v\", *(v.NextEventID))\n\t\ti++\n\t}\n\tif v.ScheduledID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledID: %v\", *(v.ScheduledID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicationTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I16_EqualsPtr(lhs, rhs *int16) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ReplicationTaskInfo match the\n// provided ReplicationTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicationTaskInfo) Equals(rhs *ReplicationTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstEventID, rhs.FirstEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventID, rhs.NextEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledID, rhs.ScheduledID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationTaskInfo.\nfunc (v *ReplicationTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.TaskType != nil {\n\t\tenc.AddInt16(\"taskType\", *v.TaskType)\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.FirstEventID != nil {\n\t\tenc.AddInt64(\"firstEventID\", *v.FirstEventID)\n\t}\n\tif v.NextEventID != nil {\n\t\tenc.AddInt64(\"nextEventID\", *v.NextEventID)\n\t}\n\tif v.ScheduledID != nil {\n\t\tenc.AddInt64(\"scheduledID\", *v.ScheduledID)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetTaskType() (o int16) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *ReplicationTaskInfo) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *ReplicationTaskInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetFirstEventID returns the value of FirstEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetFirstEventID() (o int64) {\n\tif v != nil && v.FirstEventID != nil {\n\t\treturn *v.FirstEventID\n\t}\n\n\treturn\n}\n\n// IsSetFirstEventID returns true if FirstEventID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetFirstEventID() bool {\n\treturn v != nil && v.FirstEventID != nil\n}\n\n// GetNextEventID returns the value of NextEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetNextEventID() (o int64) {\n\tif v != nil && v.NextEventID != nil {\n\t\treturn *v.NextEventID\n\t}\n\n\treturn\n}\n\n// IsSetNextEventID returns true if NextEventID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetNextEventID() bool {\n\treturn v != nil && v.NextEventID != nil\n}\n\n// GetScheduledID returns the value of ScheduledID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetScheduledID() (o int64) {\n\tif v != nil && v.ScheduledID != nil {\n\t\treturn *v.ScheduledID\n\t}\n\n\treturn\n}\n\n// IsSetScheduledID returns true if ScheduledID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetScheduledID() bool {\n\treturn v != nil && v.ScheduledID != nil\n}\n\ntype ReplicationTaskType int32\n\nconst (\n\tReplicationTaskTypeDomain          ReplicationTaskType = 0\n\tReplicationTaskTypeHistory         ReplicationTaskType = 1\n\tReplicationTaskTypeSyncShardStatus ReplicationTaskType = 2\n\tReplicationTaskTypeSyncActivity    ReplicationTaskType = 3\n\tReplicationTaskTypeHistoryMetadata ReplicationTaskType = 4\n\tReplicationTaskTypeHistoryV2       ReplicationTaskType = 5\n\tReplicationTaskTypeFailoverMarker  ReplicationTaskType = 6\n)\n\n// ReplicationTaskType_Values returns all recognized values of ReplicationTaskType.\nfunc ReplicationTaskType_Values() []ReplicationTaskType {\n\treturn []ReplicationTaskType{\n\t\tReplicationTaskTypeDomain,\n\t\tReplicationTaskTypeHistory,\n\t\tReplicationTaskTypeSyncShardStatus,\n\t\tReplicationTaskTypeSyncActivity,\n\t\tReplicationTaskTypeHistoryMetadata,\n\t\tReplicationTaskTypeHistoryV2,\n\t\tReplicationTaskTypeFailoverMarker,\n\t}\n}\n\n// UnmarshalText tries to decode ReplicationTaskType from a byte slice\n// containing its name.\n//\n//\tvar v ReplicationTaskType\n//\terr := v.UnmarshalText([]byte(\"Domain\"))\nfunc (v *ReplicationTaskType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Domain\":\n\t\t*v = ReplicationTaskTypeDomain\n\t\treturn nil\n\tcase \"History\":\n\t\t*v = ReplicationTaskTypeHistory\n\t\treturn nil\n\tcase \"SyncShardStatus\":\n\t\t*v = ReplicationTaskTypeSyncShardStatus\n\t\treturn nil\n\tcase \"SyncActivity\":\n\t\t*v = ReplicationTaskTypeSyncActivity\n\t\treturn nil\n\tcase \"HistoryMetadata\":\n\t\t*v = ReplicationTaskTypeHistoryMetadata\n\t\treturn nil\n\tcase \"HistoryV2\":\n\t\t*v = ReplicationTaskTypeHistoryV2\n\t\treturn nil\n\tcase \"FailoverMarker\":\n\t\t*v = ReplicationTaskTypeFailoverMarker\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ReplicationTaskType\", err)\n\t\t}\n\t\t*v = ReplicationTaskType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ReplicationTaskType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ReplicationTaskType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Domain\"), nil\n\tcase 1:\n\t\treturn []byte(\"History\"), nil\n\tcase 2:\n\t\treturn []byte(\"SyncShardStatus\"), nil\n\tcase 3:\n\t\treturn []byte(\"SyncActivity\"), nil\n\tcase 4:\n\t\treturn []byte(\"HistoryMetadata\"), nil\n\tcase 5:\n\t\treturn []byte(\"HistoryV2\"), nil\n\tcase 6:\n\t\treturn []byte(\"FailoverMarker\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationTaskType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ReplicationTaskType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Domain\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"History\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"SyncShardStatus\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"SyncActivity\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"HistoryMetadata\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"HistoryV2\")\n\tcase 6:\n\t\tenc.AddString(\"name\", \"FailoverMarker\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ReplicationTaskType) Ptr() *ReplicationTaskType {\n\treturn &v\n}\n\n// Encode encodes ReplicationTaskType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ReplicationTaskType\n//\treturn v.Encode(sWriter)\nfunc (v ReplicationTaskType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ReplicationTaskType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ReplicationTaskType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ReplicationTaskType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ReplicationTaskType(0), err\n//\t}\n//\n//\tvar v ReplicationTaskType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ReplicationTaskType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ReplicationTaskType) FromWire(w wire.Value) error {\n\t*v = (ReplicationTaskType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ReplicationTaskType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ReplicationTaskType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ReplicationTaskType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ReplicationTaskType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ReplicationTaskType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ReplicationTaskType.\nfunc (v ReplicationTaskType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Domain\"\n\tcase 1:\n\t\treturn \"History\"\n\tcase 2:\n\t\treturn \"SyncShardStatus\"\n\tcase 3:\n\t\treturn \"SyncActivity\"\n\tcase 4:\n\t\treturn \"HistoryMetadata\"\n\tcase 5:\n\t\treturn \"HistoryV2\"\n\tcase 6:\n\t\treturn \"FailoverMarker\"\n\t}\n\treturn fmt.Sprintf(\"ReplicationTaskType(%d)\", w)\n}\n\n// Equals returns true if this ReplicationTaskType value matches the provided\n// value.\nfunc (v ReplicationTaskType) Equals(rhs ReplicationTaskType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ReplicationTaskType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ReplicationTaskType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Domain\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"History\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"SyncShardStatus\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"SyncActivity\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"HistoryMetadata\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"HistoryV2\\\"\"), nil\n\tcase 6:\n\t\treturn ([]byte)(\"\\\"FailoverMarker\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ReplicationTaskType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ReplicationTaskType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ReplicationTaskType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ReplicationTaskType\")\n\t\t}\n\t\t*v = (ReplicationTaskType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ReplicationTaskType\")\n\t}\n}\n\ntype ReplicationToken struct {\n\tShardID                *int32 `json:\"shardID,omitempty\"`\n\tLastRetrievedMessageId *int64 `json:\"lastRetrievedMessageId,omitempty\"`\n\tLastProcessedMessageId *int64 `json:\"lastProcessedMessageId,omitempty\"`\n}\n\n// ToWire translates a ReplicationToken struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicationToken) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastRetrievedMessageId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastProcessedMessageId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReplicationToken struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicationToken struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicationToken\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicationToken) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastRetrievedMessageId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastProcessedMessageId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplicationToken struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicationToken struct could not be encoded.\nfunc (v *ReplicationToken) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastRetrievedMessageId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastRetrievedMessageId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastProcessedMessageId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastProcessedMessageId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReplicationToken struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicationToken struct could not be generated from the wire\n// representation.\nfunc (v *ReplicationToken) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastRetrievedMessageId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastProcessedMessageId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicationToken\n// struct.\nfunc (v *ReplicationToken) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastRetrievedMessageId: %v\", *(v.LastRetrievedMessageId))\n\t\ti++\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastProcessedMessageId: %v\", *(v.LastProcessedMessageId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicationToken{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReplicationToken match the\n// provided ReplicationToken.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicationToken) Equals(rhs *ReplicationToken) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastRetrievedMessageId, rhs.LastRetrievedMessageId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastProcessedMessageId, rhs.LastProcessedMessageId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationToken.\nfunc (v *ReplicationToken) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.LastRetrievedMessageId != nil {\n\t\tenc.AddInt64(\"lastRetrievedMessageId\", *v.LastRetrievedMessageId)\n\t}\n\tif v.LastProcessedMessageId != nil {\n\t\tenc.AddInt64(\"lastProcessedMessageId\", *v.LastProcessedMessageId)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationToken) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *ReplicationToken) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetLastRetrievedMessageId returns the value of LastRetrievedMessageId if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationToken) GetLastRetrievedMessageId() (o int64) {\n\tif v != nil && v.LastRetrievedMessageId != nil {\n\t\treturn *v.LastRetrievedMessageId\n\t}\n\n\treturn\n}\n\n// IsSetLastRetrievedMessageId returns true if LastRetrievedMessageId is not nil.\nfunc (v *ReplicationToken) IsSetLastRetrievedMessageId() bool {\n\treturn v != nil && v.LastRetrievedMessageId != nil\n}\n\n// GetLastProcessedMessageId returns the value of LastProcessedMessageId if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationToken) GetLastProcessedMessageId() (o int64) {\n\tif v != nil && v.LastProcessedMessageId != nil {\n\t\treturn *v.LastProcessedMessageId\n\t}\n\n\treturn\n}\n\n// IsSetLastProcessedMessageId returns true if LastProcessedMessageId is not nil.\nfunc (v *ReplicationToken) IsSetLastProcessedMessageId() bool {\n\treturn v != nil && v.LastProcessedMessageId != nil\n}\n\ntype SyncActivityTaskAttributes struct {\n\tDomainId           *string                `json:\"domainId,omitempty\"`\n\tWorkflowId         *string                `json:\"workflowId,omitempty\"`\n\tRunId              *string                `json:\"runId,omitempty\"`\n\tVersion            *int64                 `json:\"version,omitempty\"`\n\tScheduledId        *int64                 `json:\"scheduledId,omitempty\"`\n\tScheduledTime      *int64                 `json:\"scheduledTime,omitempty\"`\n\tStartedId          *int64                 `json:\"startedId,omitempty\"`\n\tStartedTime        *int64                 `json:\"startedTime,omitempty\"`\n\tLastHeartbeatTime  *int64                 `json:\"lastHeartbeatTime,omitempty\"`\n\tDetails            []byte                 `json:\"details,omitempty\"`\n\tAttempt            *int32                 `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string                `json:\"lastFailureReason,omitempty\"`\n\tLastWorkerIdentity *string                `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails []byte                 `json:\"lastFailureDetails,omitempty\"`\n\tVersionHistory     *shared.VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\n// ToWire translates a SyncActivityTaskAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SyncActivityTaskAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainId != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastHeartbeatTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.LastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.LastWorkerIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.LastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tw, err = v.VersionHistory.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistory_Read(w wire.Value) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a SyncActivityTaskAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SyncActivityTaskAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SyncActivityTaskAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SyncActivityTaskAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastHeartbeatTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastWorkerIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.VersionHistory, err = _VersionHistory_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SyncActivityTaskAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SyncActivityTaskAttributes struct could not be encoded.\nfunc (v *SyncActivityTaskAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastHeartbeatTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastHeartbeatTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWorkerIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastWorkerIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistory != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VersionHistory.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistory_Decode(sr stream.Reader) (*shared.VersionHistory, error) {\n\tvar v shared.VersionHistory\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a SyncActivityTaskAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SyncActivityTaskAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SyncActivityTaskAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastHeartbeatTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastWorkerIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TBinary:\n\t\t\tv.LastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.VersionHistory, err = _VersionHistory_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SyncActivityTaskAttributes\n// struct.\nfunc (v *SyncActivityTaskAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.DomainId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainId: %v\", *(v.DomainId))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.ScheduledId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledId: %v\", *(v.ScheduledId))\n\t\ti++\n\t}\n\tif v.ScheduledTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTime: %v\", *(v.ScheduledTime))\n\t\ti++\n\t}\n\tif v.StartedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedId: %v\", *(v.StartedId))\n\t\ti++\n\t}\n\tif v.StartedTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTime: %v\", *(v.StartedTime))\n\t\ti++\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastHeartbeatTime: %v\", *(v.LastHeartbeatTime))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureReason: %v\", *(v.LastFailureReason))\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWorkerIdentity: %v\", *(v.LastWorkerIdentity))\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureDetails: %v\", v.LastFailureDetails)\n\t\ti++\n\t}\n\tif v.VersionHistory != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistory: %v\", v.VersionHistory)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SyncActivityTaskAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SyncActivityTaskAttributes match the\n// provided SyncActivityTaskAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SyncActivityTaskAttributes) Equals(rhs *SyncActivityTaskAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainId, rhs.DomainId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledId, rhs.ScheduledId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTime, rhs.ScheduledTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedId, rhs.StartedId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTime, rhs.StartedTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastHeartbeatTime, rhs.LastHeartbeatTime) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastFailureReason, rhs.LastFailureReason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastWorkerIdentity, rhs.LastWorkerIdentity) {\n\t\treturn false\n\t}\n\tif !((v.LastFailureDetails == nil && rhs.LastFailureDetails == nil) || (v.LastFailureDetails != nil && rhs.LastFailureDetails != nil && bytes.Equal(v.LastFailureDetails, rhs.LastFailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistory == nil && rhs.VersionHistory == nil) || (v.VersionHistory != nil && rhs.VersionHistory != nil && v.VersionHistory.Equals(rhs.VersionHistory))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SyncActivityTaskAttributes.\nfunc (v *SyncActivityTaskAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainId != nil {\n\t\tenc.AddString(\"domainId\", *v.DomainId)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.ScheduledId != nil {\n\t\tenc.AddInt64(\"scheduledId\", *v.ScheduledId)\n\t}\n\tif v.ScheduledTime != nil {\n\t\tenc.AddInt64(\"scheduledTime\", *v.ScheduledTime)\n\t}\n\tif v.StartedId != nil {\n\t\tenc.AddInt64(\"startedId\", *v.StartedId)\n\t}\n\tif v.StartedTime != nil {\n\t\tenc.AddInt64(\"startedTime\", *v.StartedTime)\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tenc.AddInt64(\"lastHeartbeatTime\", *v.LastHeartbeatTime)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.LastFailureReason != nil {\n\t\tenc.AddString(\"lastFailureReason\", *v.LastFailureReason)\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tenc.AddString(\"lastWorkerIdentity\", *v.LastWorkerIdentity)\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tenc.AddString(\"lastFailureDetails\", base64.StdEncoding.EncodeToString(v.LastFailureDetails))\n\t}\n\tif v.VersionHistory != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"versionHistory\", v.VersionHistory))\n\t}\n\treturn err\n}\n\n// GetDomainId returns the value of DomainId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetDomainId() (o string) {\n\tif v != nil && v.DomainId != nil {\n\t\treturn *v.DomainId\n\t}\n\n\treturn\n}\n\n// IsSetDomainId returns true if DomainId is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetDomainId() bool {\n\treturn v != nil && v.DomainId != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetScheduledId returns the value of ScheduledId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetScheduledId() (o int64) {\n\tif v != nil && v.ScheduledId != nil {\n\t\treturn *v.ScheduledId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledId returns true if ScheduledId is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetScheduledId() bool {\n\treturn v != nil && v.ScheduledId != nil\n}\n\n// GetScheduledTime returns the value of ScheduledTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetScheduledTime() (o int64) {\n\tif v != nil && v.ScheduledTime != nil {\n\t\treturn *v.ScheduledTime\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTime returns true if ScheduledTime is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetScheduledTime() bool {\n\treturn v != nil && v.ScheduledTime != nil\n}\n\n// GetStartedId returns the value of StartedId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetStartedId() (o int64) {\n\tif v != nil && v.StartedId != nil {\n\t\treturn *v.StartedId\n\t}\n\n\treturn\n}\n\n// IsSetStartedId returns true if StartedId is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetStartedId() bool {\n\treturn v != nil && v.StartedId != nil\n}\n\n// GetStartedTime returns the value of StartedTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetStartedTime() (o int64) {\n\tif v != nil && v.StartedTime != nil {\n\t\treturn *v.StartedTime\n\t}\n\n\treturn\n}\n\n// IsSetStartedTime returns true if StartedTime is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetStartedTime() bool {\n\treturn v != nil && v.StartedTime != nil\n}\n\n// GetLastHeartbeatTime returns the value of LastHeartbeatTime if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetLastHeartbeatTime() (o int64) {\n\tif v != nil && v.LastHeartbeatTime != nil {\n\t\treturn *v.LastHeartbeatTime\n\t}\n\n\treturn\n}\n\n// IsSetLastHeartbeatTime returns true if LastHeartbeatTime is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetLastHeartbeatTime() bool {\n\treturn v != nil && v.LastHeartbeatTime != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetLastFailureReason returns the value of LastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureReason returns true if LastFailureReason is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetLastFailureReason() bool {\n\treturn v != nil && v.LastFailureReason != nil\n}\n\n// GetLastWorkerIdentity returns the value of LastWorkerIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetLastWorkerIdentity() (o string) {\n\tif v != nil && v.LastWorkerIdentity != nil {\n\t\treturn *v.LastWorkerIdentity\n\t}\n\n\treturn\n}\n\n// IsSetLastWorkerIdentity returns true if LastWorkerIdentity is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetLastWorkerIdentity() bool {\n\treturn v != nil && v.LastWorkerIdentity != nil\n}\n\n// GetLastFailureDetails returns the value of LastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureDetails returns true if LastFailureDetails is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetLastFailureDetails() bool {\n\treturn v != nil && v.LastFailureDetails != nil\n}\n\n// GetVersionHistory returns the value of VersionHistory if it is set or its\n// zero value if it is unset.\nfunc (v *SyncActivityTaskAttributes) GetVersionHistory() (o *shared.VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistory returns true if VersionHistory is not nil.\nfunc (v *SyncActivityTaskAttributes) IsSetVersionHistory() bool {\n\treturn v != nil && v.VersionHistory != nil\n}\n\ntype SyncShardStatus struct {\n\tTimestamp *int64 `json:\"timestamp,omitempty\"`\n}\n\n// ToWire translates a SyncShardStatus struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SyncShardStatus) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Timestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.Timestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SyncShardStatus struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SyncShardStatus struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SyncShardStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SyncShardStatus) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Timestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SyncShardStatus struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SyncShardStatus struct could not be encoded.\nfunc (v *SyncShardStatus) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Timestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Timestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SyncShardStatus struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SyncShardStatus struct could not be generated from the wire\n// representation.\nfunc (v *SyncShardStatus) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Timestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SyncShardStatus\n// struct.\nfunc (v *SyncShardStatus) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Timestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"Timestamp: %v\", *(v.Timestamp))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SyncShardStatus{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SyncShardStatus match the\n// provided SyncShardStatus.\n//\n// This function performs a deep comparison.\nfunc (v *SyncShardStatus) Equals(rhs *SyncShardStatus) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Timestamp, rhs.Timestamp) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SyncShardStatus.\nfunc (v *SyncShardStatus) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Timestamp != nil {\n\t\tenc.AddInt64(\"timestamp\", *v.Timestamp)\n\t}\n\treturn err\n}\n\n// GetTimestamp returns the value of Timestamp if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatus) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\n\treturn\n}\n\n// IsSetTimestamp returns true if Timestamp is not nil.\nfunc (v *SyncShardStatus) IsSetTimestamp() bool {\n\treturn v != nil && v.Timestamp != nil\n}\n\ntype SyncShardStatusTaskAttributes struct {\n\tSourceCluster *string `json:\"sourceCluster,omitempty\"`\n\tShardId       *int64  `json:\"shardId,omitempty\"`\n\tTimestamp     *int64  `json:\"timestamp,omitempty\"`\n}\n\n// ToWire translates a SyncShardStatusTaskAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SyncShardStatusTaskAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SourceCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.SourceCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ShardId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.Timestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SyncShardStatusTaskAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SyncShardStatusTaskAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SyncShardStatusTaskAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SyncShardStatusTaskAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SourceCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ShardId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Timestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SyncShardStatusTaskAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SyncShardStatusTaskAttributes struct could not be encoded.\nfunc (v *SyncShardStatusTaskAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SourceCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SourceCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ShardId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Timestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Timestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SyncShardStatusTaskAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SyncShardStatusTaskAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SyncShardStatusTaskAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SourceCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ShardId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Timestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SyncShardStatusTaskAttributes\n// struct.\nfunc (v *SyncShardStatusTaskAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.SourceCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"SourceCluster: %v\", *(v.SourceCluster))\n\t\ti++\n\t}\n\tif v.ShardId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardId: %v\", *(v.ShardId))\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"Timestamp: %v\", *(v.Timestamp))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SyncShardStatusTaskAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SyncShardStatusTaskAttributes match the\n// provided SyncShardStatusTaskAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SyncShardStatusTaskAttributes) Equals(rhs *SyncShardStatusTaskAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SourceCluster, rhs.SourceCluster) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ShardId, rhs.ShardId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Timestamp, rhs.Timestamp) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SyncShardStatusTaskAttributes.\nfunc (v *SyncShardStatusTaskAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SourceCluster != nil {\n\t\tenc.AddString(\"sourceCluster\", *v.SourceCluster)\n\t}\n\tif v.ShardId != nil {\n\t\tenc.AddInt64(\"shardId\", *v.ShardId)\n\t}\n\tif v.Timestamp != nil {\n\t\tenc.AddInt64(\"timestamp\", *v.Timestamp)\n\t}\n\treturn err\n}\n\n// GetSourceCluster returns the value of SourceCluster if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusTaskAttributes) GetSourceCluster() (o string) {\n\tif v != nil && v.SourceCluster != nil {\n\t\treturn *v.SourceCluster\n\t}\n\n\treturn\n}\n\n// IsSetSourceCluster returns true if SourceCluster is not nil.\nfunc (v *SyncShardStatusTaskAttributes) IsSetSourceCluster() bool {\n\treturn v != nil && v.SourceCluster != nil\n}\n\n// GetShardId returns the value of ShardId if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusTaskAttributes) GetShardId() (o int64) {\n\tif v != nil && v.ShardId != nil {\n\t\treturn *v.ShardId\n\t}\n\n\treturn\n}\n\n// IsSetShardId returns true if ShardId is not nil.\nfunc (v *SyncShardStatusTaskAttributes) IsSetShardId() bool {\n\treturn v != nil && v.ShardId != nil\n}\n\n// GetTimestamp returns the value of Timestamp if it is set or its\n// zero value if it is unset.\nfunc (v *SyncShardStatusTaskAttributes) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\n\treturn\n}\n\n// IsSetTimestamp returns true if Timestamp is not nil.\nfunc (v *SyncShardStatusTaskAttributes) IsSetTimestamp() bool {\n\treturn v != nil && v.Timestamp != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"replicator\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/replicator\",\n\tFilePath: \"replicator.thrift\",\n\tSHA1:     \"5511dc1dfc4b6025f16d77cac580134c62303bd6\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.replicator\\n\\ninclude \\\"shared.thrift\\\"\\n\\nenum ReplicationTaskType {\\n  Domain\\n  History\\n  SyncShardStatus\\n  SyncActivity\\n  HistoryMetadata\\n  HistoryV2\\n  FailoverMarker\\n}\\n\\nenum DomainOperation {\\n  Create\\n  Update\\n  Delete\\n}\\n\\nstruct DomainTaskAttributes {\\n  05: optional DomainOperation domainOperation\\n  10: optional string id\\n  20: optional shared.DomainInfo info\\n  30: optional shared.DomainConfiguration config\\n  40: optional shared.DomainReplicationConfiguration replicationConfig\\n  50: optional i64 (js.type = \\\"Long\\\") configVersion\\n  60: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n  70: optional i64 (js.type = \\\"Long\\\") previousFailoverVersion\\n}\\n\\nstruct SyncShardStatusTaskAttributes {\\n  10: optional string sourceCluster\\n  20: optional i64 (js.type = \\\"Long\\\") shardId\\n  30: optional i64 (js.type = \\\"Long\\\") timestamp\\n}\\n\\nstruct SyncActivityTaskAttributes {\\n  10: optional string domainId\\n  20: optional string workflowId\\n  30: optional string runId\\n  40: optional i64 (js.type = \\\"Long\\\") version\\n  50: optional i64 (js.type = \\\"Long\\\") scheduledId\\n  60: optional i64 (js.type = \\\"Long\\\") scheduledTime\\n  70: optional i64 (js.type = \\\"Long\\\") startedId\\n  80: optional i64 (js.type = \\\"Long\\\") startedTime\\n  90: optional i64 (js.type = \\\"Long\\\") lastHeartbeatTime\\n  100: optional binary details\\n  110: optional i32 attempt\\n  120: optional string lastFailureReason\\n  130: optional string lastWorkerIdentity\\n  140: optional binary lastFailureDetails\\n  150: optional shared.VersionHistory versionHistory\\n}\\n\\nstruct HistoryTaskV2Attributes {\\n  05: optional i64 (js.type = \\\"Long\\\") taskId\\n  10: optional string domainId\\n  20: optional string workflowId\\n  30: optional string runId\\n  40: optional list<shared.VersionHistoryItem> versionHistoryItems\\n  50: optional shared.DataBlob events\\n  // new run events does not need version history since there is no prior events\\n  70: optional shared.DataBlob newRunEvents\\n}\\n\\nstruct FailoverMarkerAttributes{\\n\\t10: optional string domainID\\n\\t20: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n\\t30: optional i64 (js.type = \\\"Long\\\") creationTime\\n}\\n\\nstruct FailoverMarkers{\\n\\t10: optional list<FailoverMarkerAttributes> failoverMarkers\\n}\\n\\nstruct ReplicationTask {\\n  10: optional ReplicationTaskType taskType\\n  11: optional i64 (js.type = \\\"Long\\\") sourceTaskId\\n  20: optional DomainTaskAttributes domainTaskAttributes\\n  40: optional SyncShardStatusTaskAttributes syncShardStatusTaskAttributes\\n  50: optional SyncActivityTaskAttributes syncActivityTaskAttributes\\n  70: optional HistoryTaskV2Attributes historyTaskV2Attributes\\n  80: optional FailoverMarkerAttributes failoverMarkerAttributes\\n  90: optional i64 (js.type = \\\"Long\\\") creationTime\\n}\\n\\nstruct ReplicationToken {\\n  10: optional i32 shardID\\n  // lastRetrivedMessageId is where the next fetch should begin with\\n  20: optional i64 (js.type = \\\"Long\\\") lastRetrievedMessageId\\n  // lastProcessedMessageId is the last messageId that is processed on the passive side.\\n  // This can be different than lastRetrievedMessageId if passive side supports prefetching messages.\\n  30: optional i64 (js.type = \\\"Long\\\") lastProcessedMessageId\\n}\\n\\nstruct SyncShardStatus {\\n  10: optional i64 (js.type = \\\"Long\\\") timestamp\\n}\\n\\nstruct ReplicationMessages {\\n  10: optional list<ReplicationTask> replicationTasks\\n  // This can be different than the last taskId in the above list, because sender can decide to skip tasks (e.g. for completed workflows).\\n  20: optional i64 (js.type = \\\"Long\\\") lastRetrievedMessageId\\n  30: optional bool hasMore // Hint for flow control\\n  40: optional SyncShardStatus syncShardStatus\\n}\\n\\nstruct ReplicationTaskInfo {\\n  10: optional string domainID\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional i16 taskType\\n  50: optional i64 (js.type = \\\"Long\\\") taskID\\n  60: optional i64 (js.type = \\\"Long\\\") version\\n  70: optional i64 (js.type = \\\"Long\\\") firstEventID\\n  80: optional i64 (js.type = \\\"Long\\\") nextEventID\\n  90: optional i64 (js.type = \\\"Long\\\") scheduledID\\n}\\n\\nstruct GetReplicationMessagesRequest {\\n  10: optional list<ReplicationToken> tokens\\n  20: optional string clusterName\\n}\\n\\nstruct GetReplicationMessagesResponse {\\n  10: optional map<i32, ReplicationMessages> messagesByShard\\n}\\n\\nstruct GetDomainReplicationMessagesRequest {\\n  // lastRetrievedMessageId is where the next fetch should begin with\\n  10: optional i64 (js.type = \\\"Long\\\") lastRetrievedMessageId\\n  // lastProcessedMessageId is the last messageId that is processed on the passive side.\\n  // This can be different than lastRetrievedMessageId if passive side supports prefetching messages.\\n  20: optional i64 (js.type = \\\"Long\\\") lastProcessedMessageId\\n  // clusterName is the name of the pulling cluster\\n  30: optional string clusterName\\n}\\n\\nstruct GetDomainReplicationMessagesResponse {\\n  10: optional ReplicationMessages messages\\n}\\n\\nstruct GetDLQReplicationMessagesRequest {\\n  10: optional list<ReplicationTaskInfo> taskInfos\\n}\\n\\nstruct GetDLQReplicationMessagesResponse {\\n  10: optional list<ReplicationTask> replicationTasks\\n}\\n\\nenum DLQType {\\n  Replication,\\n  Domain,\\n}\\n\\nstruct ReadDLQMessagesRequest{\\n  10: optional DLQType type\\n  20: optional i32 shardID\\n  30: optional string sourceCluster\\n  40: optional i64 (js.type = \\\"Long\\\") inclusiveEndMessageID\\n  50: optional i32 maximumPageSize\\n  60: optional binary nextPageToken\\n}\\n\\nstruct ReadDLQMessagesResponse{\\n  10: optional DLQType type\\n  20: optional list<ReplicationTask> replicationTasks\\n  30: optional binary nextPageToken\\n  40: optional list<ReplicationTaskInfo> replicationTasksInfo\\n}\\n\\nstruct PurgeDLQMessagesRequest{\\n  10: optional DLQType type\\n  20: optional i32 shardID\\n  30: optional string sourceCluster\\n  40: optional i64 (js.type = \\\"Long\\\") inclusiveEndMessageID\\n}\\n\\nstruct MergeDLQMessagesRequest{\\n  10: optional DLQType type\\n  20: optional i32 shardID\\n  30: optional string sourceCluster\\n  40: optional i64 (js.type = \\\"Long\\\") inclusiveEndMessageID\\n  50: optional i32 maximumPageSize\\n  60: optional binary nextPageToken\\n}\\n\\nstruct MergeDLQMessagesResponse{\\n  10: optional binary nextPageToken\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/replicator/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage replicator\n"
  },
  {
    "path": ".gen/go/shadower/shadower.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage shadower\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\nconst ErrNonRetryableType string = \"com.uber.cadence.internal.shadowing.NonRetryableException\"\n\nconst ErrReasonDomainNotExists string = \"domain not exists\"\n\nconst ErrReasonInvalidQuery string = \"invalid visibility query\"\n\nconst ErrReasonWorkflowTypeNotRegistered string = \"workflow type not registered\"\n\nconst LocalDomainName string = \"cadence-shadower\"\n\nconst ReplayWorkflowActivityName string = \"replayWorkflowActivity\"\n\nconst ScanWorkflowActivityName string = \"scanWorkflowActivity\"\n\nconst TaskList string = \"cadence-shadower-tl\"\n\nconst WorkflowIDSuffix string = \"-shadow-workflow\"\n\nconst WorkflowName string = \"cadence-shadow-workflow\"\n\ntype ExitCondition struct {\n\tExpirationIntervalInSeconds *int32 `json:\"expirationIntervalInSeconds,omitempty\"`\n\tShadowCount                 *int32 `json:\"shadowCount,omitempty\"`\n}\n\n// ToWire translates a ExitCondition struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ExitCondition) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExpirationIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShadowCount != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShadowCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ExitCondition struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ExitCondition struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ExitCondition\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ExitCondition) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExpirationIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShadowCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ExitCondition struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ExitCondition struct could not be encoded.\nfunc (v *ExitCondition) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExpirationIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShadowCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShadowCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ExitCondition struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ExitCondition struct could not be generated from the wire\n// representation.\nfunc (v *ExitCondition) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExpirationIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShadowCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ExitCondition\n// struct.\nfunc (v *ExitCondition) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpirationIntervalInSeconds: %v\", *(v.ExpirationIntervalInSeconds))\n\t\ti++\n\t}\n\tif v.ShadowCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShadowCount: %v\", *(v.ShadowCount))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ExitCondition{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ExitCondition match the\n// provided ExitCondition.\n//\n// This function performs a deep comparison.\nfunc (v *ExitCondition) Equals(rhs *ExitCondition) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExpirationIntervalInSeconds, rhs.ExpirationIntervalInSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShadowCount, rhs.ShadowCount) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ExitCondition.\nfunc (v *ExitCondition) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"expirationIntervalInSeconds\", *v.ExpirationIntervalInSeconds)\n\t}\n\tif v.ShadowCount != nil {\n\t\tenc.AddInt32(\"shadowCount\", *v.ShadowCount)\n\t}\n\treturn err\n}\n\n// GetExpirationIntervalInSeconds returns the value of ExpirationIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ExitCondition) GetExpirationIntervalInSeconds() (o int32) {\n\tif v != nil && v.ExpirationIntervalInSeconds != nil {\n\t\treturn *v.ExpirationIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExpirationIntervalInSeconds returns true if ExpirationIntervalInSeconds is not nil.\nfunc (v *ExitCondition) IsSetExpirationIntervalInSeconds() bool {\n\treturn v != nil && v.ExpirationIntervalInSeconds != nil\n}\n\n// GetShadowCount returns the value of ShadowCount if it is set or its\n// zero value if it is unset.\nfunc (v *ExitCondition) GetShadowCount() (o int32) {\n\tif v != nil && v.ShadowCount != nil {\n\t\treturn *v.ShadowCount\n\t}\n\n\treturn\n}\n\n// IsSetShadowCount returns true if ShadowCount is not nil.\nfunc (v *ExitCondition) IsSetShadowCount() bool {\n\treturn v != nil && v.ShadowCount != nil\n}\n\ntype Mode int32\n\nconst (\n\tModeNormal     Mode = 0\n\tModeContinuous Mode = 1\n)\n\n// Mode_Values returns all recognized values of Mode.\nfunc Mode_Values() []Mode {\n\treturn []Mode{\n\t\tModeNormal,\n\t\tModeContinuous,\n\t}\n}\n\n// UnmarshalText tries to decode Mode from a byte slice\n// containing its name.\n//\n//\tvar v Mode\n//\terr := v.UnmarshalText([]byte(\"Normal\"))\nfunc (v *Mode) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Normal\":\n\t\t*v = ModeNormal\n\t\treturn nil\n\tcase \"Continuous\":\n\t\t*v = ModeContinuous\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"Mode\", err)\n\t\t}\n\t\t*v = Mode(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes Mode to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v Mode) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Normal\"), nil\n\tcase 1:\n\t\treturn []byte(\"Continuous\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Mode.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v Mode) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Normal\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Continuous\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v Mode) Ptr() *Mode {\n\treturn &v\n}\n\n// Encode encodes Mode directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v Mode\n//\treturn v.Encode(sWriter)\nfunc (v Mode) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates Mode into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v Mode) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes Mode from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return Mode(0), err\n//\t}\n//\n//\tvar v Mode\n//\tif err := v.FromWire(x); err != nil {\n//\t  return Mode(0), err\n//\t}\n//\treturn v, nil\nfunc (v *Mode) FromWire(w wire.Value) error {\n\t*v = (Mode)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded Mode directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v Mode\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return Mode(0), err\n//\t}\n//\treturn v, nil\nfunc (v *Mode) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (Mode)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of Mode.\nfunc (v Mode) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Normal\"\n\tcase 1:\n\t\treturn \"Continuous\"\n\t}\n\treturn fmt.Sprintf(\"Mode(%d)\", w)\n}\n\n// Equals returns true if this Mode value matches the provided\n// value.\nfunc (v Mode) Equals(rhs Mode) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes Mode into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v Mode) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Normal\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Continuous\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode Mode from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *Mode) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"Mode\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"Mode\")\n\t\t}\n\t\t*v = (Mode)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"Mode\")\n\t}\n}\n\ntype ReplayWorkflowActivityParams struct {\n\tDomain     *string                     `json:\"domain,omitempty\"`\n\tExecutions []*shared.WorkflowExecution `json:\"executions,omitempty\"`\n}\n\ntype _List_WorkflowExecution_ValueList []*shared.WorkflowExecution\n\nfunc (v _List_WorkflowExecution_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.WorkflowExecution', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_WorkflowExecution_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_WorkflowExecution_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_WorkflowExecution_ValueList) Close() {}\n\n// ToWire translates a ReplayWorkflowActivityParams struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplayWorkflowActivityParams) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecution_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecution_Read(w wire.Value) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_WorkflowExecution_Read(l wire.ValueList) ([]*shared.WorkflowExecution, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*shared.WorkflowExecution, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _WorkflowExecution_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ReplayWorkflowActivityParams struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplayWorkflowActivityParams struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplayWorkflowActivityParams\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplayWorkflowActivityParams) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecution_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_WorkflowExecution_Encode(val []*shared.WorkflowExecution, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.WorkflowExecution', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ReplayWorkflowActivityParams struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplayWorkflowActivityParams struct could not be encoded.\nfunc (v *ReplayWorkflowActivityParams) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecution_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecution_Decode(sr stream.Reader) (*shared.WorkflowExecution, error) {\n\tvar v shared.WorkflowExecution\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_WorkflowExecution_Decode(sr stream.Reader) ([]*shared.WorkflowExecution, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*shared.WorkflowExecution, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _WorkflowExecution_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ReplayWorkflowActivityParams struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplayWorkflowActivityParams struct could not be generated from the wire\n// representation.\nfunc (v *ReplayWorkflowActivityParams) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplayWorkflowActivityParams\n// struct.\nfunc (v *ReplayWorkflowActivityParams) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplayWorkflowActivityParams{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_WorkflowExecution_Equals(lhs, rhs []*shared.WorkflowExecution) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ReplayWorkflowActivityParams match the\n// provided ReplayWorkflowActivityParams.\n//\n// This function performs a deep comparison.\nfunc (v *ReplayWorkflowActivityParams) Equals(rhs *ReplayWorkflowActivityParams) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecution_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_WorkflowExecution_Zapper []*shared.WorkflowExecution\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_WorkflowExecution_Zapper.\nfunc (l _List_WorkflowExecution_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplayWorkflowActivityParams.\nfunc (v *ReplayWorkflowActivityParams) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecution_Zapper)(v.Executions)))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ReplayWorkflowActivityParams) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ReplayWorkflowActivityParams) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ReplayWorkflowActivityParams) GetExecutions() (o []*shared.WorkflowExecution) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ReplayWorkflowActivityParams) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\ntype ReplayWorkflowActivityResult struct {\n\tSucceeded *int32 `json:\"succeeded,omitempty\"`\n\tSkipped   *int32 `json:\"skipped,omitempty\"`\n\tFailed    *int32 `json:\"failed,omitempty\"`\n}\n\n// ToWire translates a ReplayWorkflowActivityResult struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplayWorkflowActivityResult) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Succeeded != nil {\n\t\tw, err = wire.NewValueI32(*(v.Succeeded)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Skipped != nil {\n\t\tw, err = wire.NewValueI32(*(v.Skipped)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Failed != nil {\n\t\tw, err = wire.NewValueI32(*(v.Failed)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReplayWorkflowActivityResult struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplayWorkflowActivityResult struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplayWorkflowActivityResult\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplayWorkflowActivityResult) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Succeeded = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Skipped = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Failed = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplayWorkflowActivityResult struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplayWorkflowActivityResult struct could not be encoded.\nfunc (v *ReplayWorkflowActivityResult) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Succeeded != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Succeeded)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Skipped != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Skipped)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Failed != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Failed)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReplayWorkflowActivityResult struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplayWorkflowActivityResult struct could not be generated from the wire\n// representation.\nfunc (v *ReplayWorkflowActivityResult) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Succeeded = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Skipped = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Failed = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplayWorkflowActivityResult\n// struct.\nfunc (v *ReplayWorkflowActivityResult) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Succeeded != nil {\n\t\tfields[i] = fmt.Sprintf(\"Succeeded: %v\", *(v.Succeeded))\n\t\ti++\n\t}\n\tif v.Skipped != nil {\n\t\tfields[i] = fmt.Sprintf(\"Skipped: %v\", *(v.Skipped))\n\t\ti++\n\t}\n\tif v.Failed != nil {\n\t\tfields[i] = fmt.Sprintf(\"Failed: %v\", *(v.Failed))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplayWorkflowActivityResult{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReplayWorkflowActivityResult match the\n// provided ReplayWorkflowActivityResult.\n//\n// This function performs a deep comparison.\nfunc (v *ReplayWorkflowActivityResult) Equals(rhs *ReplayWorkflowActivityResult) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Succeeded, rhs.Succeeded) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Skipped, rhs.Skipped) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Failed, rhs.Failed) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplayWorkflowActivityResult.\nfunc (v *ReplayWorkflowActivityResult) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Succeeded != nil {\n\t\tenc.AddInt32(\"succeeded\", *v.Succeeded)\n\t}\n\tif v.Skipped != nil {\n\t\tenc.AddInt32(\"skipped\", *v.Skipped)\n\t}\n\tif v.Failed != nil {\n\t\tenc.AddInt32(\"failed\", *v.Failed)\n\t}\n\treturn err\n}\n\n// GetSucceeded returns the value of Succeeded if it is set or its\n// zero value if it is unset.\nfunc (v *ReplayWorkflowActivityResult) GetSucceeded() (o int32) {\n\tif v != nil && v.Succeeded != nil {\n\t\treturn *v.Succeeded\n\t}\n\n\treturn\n}\n\n// IsSetSucceeded returns true if Succeeded is not nil.\nfunc (v *ReplayWorkflowActivityResult) IsSetSucceeded() bool {\n\treturn v != nil && v.Succeeded != nil\n}\n\n// GetSkipped returns the value of Skipped if it is set or its\n// zero value if it is unset.\nfunc (v *ReplayWorkflowActivityResult) GetSkipped() (o int32) {\n\tif v != nil && v.Skipped != nil {\n\t\treturn *v.Skipped\n\t}\n\n\treturn\n}\n\n// IsSetSkipped returns true if Skipped is not nil.\nfunc (v *ReplayWorkflowActivityResult) IsSetSkipped() bool {\n\treturn v != nil && v.Skipped != nil\n}\n\n// GetFailed returns the value of Failed if it is set or its\n// zero value if it is unset.\nfunc (v *ReplayWorkflowActivityResult) GetFailed() (o int32) {\n\tif v != nil && v.Failed != nil {\n\t\treturn *v.Failed\n\t}\n\n\treturn\n}\n\n// IsSetFailed returns true if Failed is not nil.\nfunc (v *ReplayWorkflowActivityResult) IsSetFailed() bool {\n\treturn v != nil && v.Failed != nil\n}\n\ntype ScanWorkflowActivityParams struct {\n\tDomain        *string  `json:\"domain,omitempty\"`\n\tWorkflowQuery *string  `json:\"workflowQuery,omitempty\"`\n\tNextPageToken []byte   `json:\"nextPageToken,omitempty\"`\n\tPageSize      *int32   `json:\"pageSize,omitempty\"`\n\tSamplingRate  *float64 `json:\"samplingRate,omitempty\"`\n}\n\n// ToWire translates a ScanWorkflowActivityParams struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ScanWorkflowActivityParams) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowQuery)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.SamplingRate != nil {\n\t\tw, err = wire.NewValueDouble(*(v.SamplingRate)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ScanWorkflowActivityParams struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ScanWorkflowActivityParams struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ScanWorkflowActivityParams\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ScanWorkflowActivityParams) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowQuery = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.SamplingRate = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ScanWorkflowActivityParams struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ScanWorkflowActivityParams struct could not be encoded.\nfunc (v *ScanWorkflowActivityParams) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowQuery != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowQuery)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SamplingRate != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.SamplingRate)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ScanWorkflowActivityParams struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ScanWorkflowActivityParams struct could not be generated from the wire\n// representation.\nfunc (v *ScanWorkflowActivityParams) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowQuery = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.SamplingRate = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ScanWorkflowActivityParams\n// struct.\nfunc (v *ScanWorkflowActivityParams) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowQuery: %v\", *(v.WorkflowQuery))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.SamplingRate != nil {\n\t\tfields[i] = fmt.Sprintf(\"SamplingRate: %v\", *(v.SamplingRate))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ScanWorkflowActivityParams{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Double_EqualsPtr(lhs, rhs *float64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ScanWorkflowActivityParams match the\n// provided ScanWorkflowActivityParams.\n//\n// This function performs a deep comparison.\nfunc (v *ScanWorkflowActivityParams) Equals(rhs *ScanWorkflowActivityParams) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowQuery, rhs.WorkflowQuery) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.SamplingRate, rhs.SamplingRate) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ScanWorkflowActivityParams.\nfunc (v *ScanWorkflowActivityParams) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tenc.AddString(\"workflowQuery\", *v.WorkflowQuery)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.SamplingRate != nil {\n\t\tenc.AddFloat64(\"samplingRate\", *v.SamplingRate)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityParams) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ScanWorkflowActivityParams) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowQuery returns the value of WorkflowQuery if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityParams) GetWorkflowQuery() (o string) {\n\tif v != nil && v.WorkflowQuery != nil {\n\t\treturn *v.WorkflowQuery\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowQuery returns true if WorkflowQuery is not nil.\nfunc (v *ScanWorkflowActivityParams) IsSetWorkflowQuery() bool {\n\treturn v != nil && v.WorkflowQuery != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityParams) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ScanWorkflowActivityParams) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityParams) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *ScanWorkflowActivityParams) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetSamplingRate returns the value of SamplingRate if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityParams) GetSamplingRate() (o float64) {\n\tif v != nil && v.SamplingRate != nil {\n\t\treturn *v.SamplingRate\n\t}\n\n\treturn\n}\n\n// IsSetSamplingRate returns true if SamplingRate is not nil.\nfunc (v *ScanWorkflowActivityParams) IsSetSamplingRate() bool {\n\treturn v != nil && v.SamplingRate != nil\n}\n\ntype ScanWorkflowActivityResult struct {\n\tExecutions    []*shared.WorkflowExecution `json:\"executions,omitempty\"`\n\tNextPageToken []byte                      `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ScanWorkflowActivityResult struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ScanWorkflowActivityResult) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecution_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ScanWorkflowActivityResult struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ScanWorkflowActivityResult struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ScanWorkflowActivityResult\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ScanWorkflowActivityResult) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecution_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ScanWorkflowActivityResult struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ScanWorkflowActivityResult struct could not be encoded.\nfunc (v *ScanWorkflowActivityResult) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecution_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ScanWorkflowActivityResult struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ScanWorkflowActivityResult struct could not be generated from the wire\n// representation.\nfunc (v *ScanWorkflowActivityResult) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ScanWorkflowActivityResult\n// struct.\nfunc (v *ScanWorkflowActivityResult) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ScanWorkflowActivityResult{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ScanWorkflowActivityResult match the\n// provided ScanWorkflowActivityResult.\n//\n// This function performs a deep comparison.\nfunc (v *ScanWorkflowActivityResult) Equals(rhs *ScanWorkflowActivityResult) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecution_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ScanWorkflowActivityResult.\nfunc (v *ScanWorkflowActivityResult) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecution_Zapper)(v.Executions)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityResult) GetExecutions() (o []*shared.WorkflowExecution) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ScanWorkflowActivityResult) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ScanWorkflowActivityResult) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ScanWorkflowActivityResult) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype WorkflowParams struct {\n\tDomain        *string         `json:\"domain,omitempty\"`\n\tTaskList      *string         `json:\"taskList,omitempty\"`\n\tWorkflowQuery *string         `json:\"workflowQuery,omitempty\"`\n\tNextPageToken []byte          `json:\"nextPageToken,omitempty\"`\n\tSamplingRate  *float64        `json:\"samplingRate,omitempty\"`\n\tShadowMode    *Mode           `json:\"shadowMode,omitempty\"`\n\tExitCondition *ExitCondition  `json:\"exitCondition,omitempty\"`\n\tConcurrency   *int32          `json:\"concurrency,omitempty\"`\n\tLastRunResult *WorkflowResult `json:\"lastRunResult,omitempty\"`\n}\n\n// ToWire translates a WorkflowParams struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowParams) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowQuery)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.SamplingRate != nil {\n\t\tw, err = wire.NewValueDouble(*(v.SamplingRate)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ShadowMode != nil {\n\t\tw, err = v.ShadowMode.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ExitCondition != nil {\n\t\tw, err = v.ExitCondition.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Concurrency != nil {\n\t\tw, err = wire.NewValueI32(*(v.Concurrency)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.LastRunResult != nil {\n\t\tw, err = v.LastRunResult.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Mode_Read(w wire.Value) (Mode, error) {\n\tvar v Mode\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _ExitCondition_Read(w wire.Value) (*ExitCondition, error) {\n\tvar v ExitCondition\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowResult_Read(w wire.Value) (*WorkflowResult, error) {\n\tvar v WorkflowResult\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a WorkflowParams struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowParams struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowParams\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowParams) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowQuery = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.SamplingRate = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x Mode\n\t\t\t\tx, err = _Mode_Read(field.Value)\n\t\t\t\tv.ShadowMode = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExitCondition, err = _ExitCondition_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Concurrency = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.LastRunResult, err = _WorkflowResult_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowParams struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowParams struct could not be encoded.\nfunc (v *WorkflowParams) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowQuery != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowQuery)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SamplingRate != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.SamplingRate)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShadowMode != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ShadowMode.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExitCondition != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExitCondition.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Concurrency != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Concurrency)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastRunResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.LastRunResult.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Mode_Decode(sr stream.Reader) (Mode, error) {\n\tvar v Mode\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _ExitCondition_Decode(sr stream.Reader) (*ExitCondition, error) {\n\tvar v ExitCondition\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowResult_Decode(sr stream.Reader) (*WorkflowResult, error) {\n\tvar v WorkflowResult\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a WorkflowParams struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowParams struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowParams) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowQuery = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.SamplingRate = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x Mode\n\t\t\tx, err = _Mode_Decode(sr)\n\t\t\tv.ShadowMode = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.ExitCondition, err = _ExitCondition_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Concurrency = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.LastRunResult, err = _WorkflowResult_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowParams\n// struct.\nfunc (v *WorkflowParams) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowQuery: %v\", *(v.WorkflowQuery))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.SamplingRate != nil {\n\t\tfields[i] = fmt.Sprintf(\"SamplingRate: %v\", *(v.SamplingRate))\n\t\ti++\n\t}\n\tif v.ShadowMode != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShadowMode: %v\", *(v.ShadowMode))\n\t\ti++\n\t}\n\tif v.ExitCondition != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExitCondition: %v\", v.ExitCondition)\n\t\ti++\n\t}\n\tif v.Concurrency != nil {\n\t\tfields[i] = fmt.Sprintf(\"Concurrency: %v\", *(v.Concurrency))\n\t\ti++\n\t}\n\tif v.LastRunResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastRunResult: %v\", v.LastRunResult)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowParams{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Mode_EqualsPtr(lhs, rhs *Mode) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this WorkflowParams match the\n// provided WorkflowParams.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowParams) Equals(rhs *WorkflowParams) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowQuery, rhs.WorkflowQuery) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.SamplingRate, rhs.SamplingRate) {\n\t\treturn false\n\t}\n\tif !_Mode_EqualsPtr(v.ShadowMode, rhs.ShadowMode) {\n\t\treturn false\n\t}\n\tif !((v.ExitCondition == nil && rhs.ExitCondition == nil) || (v.ExitCondition != nil && rhs.ExitCondition != nil && v.ExitCondition.Equals(rhs.ExitCondition))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Concurrency, rhs.Concurrency) {\n\t\treturn false\n\t}\n\tif !((v.LastRunResult == nil && rhs.LastRunResult == nil) || (v.LastRunResult != nil && rhs.LastRunResult != nil && v.LastRunResult.Equals(rhs.LastRunResult))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowParams.\nfunc (v *WorkflowParams) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\tif v.WorkflowQuery != nil {\n\t\tenc.AddString(\"workflowQuery\", *v.WorkflowQuery)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.SamplingRate != nil {\n\t\tenc.AddFloat64(\"samplingRate\", *v.SamplingRate)\n\t}\n\tif v.ShadowMode != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"shadowMode\", *v.ShadowMode))\n\t}\n\tif v.ExitCondition != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"exitCondition\", v.ExitCondition))\n\t}\n\tif v.Concurrency != nil {\n\t\tenc.AddInt32(\"concurrency\", *v.Concurrency)\n\t}\n\tif v.LastRunResult != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"lastRunResult\", v.LastRunResult))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *WorkflowParams) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowParams) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetWorkflowQuery returns the value of WorkflowQuery if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetWorkflowQuery() (o string) {\n\tif v != nil && v.WorkflowQuery != nil {\n\t\treturn *v.WorkflowQuery\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowQuery returns true if WorkflowQuery is not nil.\nfunc (v *WorkflowParams) IsSetWorkflowQuery() bool {\n\treturn v != nil && v.WorkflowQuery != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *WorkflowParams) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetSamplingRate returns the value of SamplingRate if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetSamplingRate() (o float64) {\n\tif v != nil && v.SamplingRate != nil {\n\t\treturn *v.SamplingRate\n\t}\n\n\treturn\n}\n\n// IsSetSamplingRate returns true if SamplingRate is not nil.\nfunc (v *WorkflowParams) IsSetSamplingRate() bool {\n\treturn v != nil && v.SamplingRate != nil\n}\n\n// GetShadowMode returns the value of ShadowMode if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetShadowMode() (o Mode) {\n\tif v != nil && v.ShadowMode != nil {\n\t\treturn *v.ShadowMode\n\t}\n\n\treturn\n}\n\n// IsSetShadowMode returns true if ShadowMode is not nil.\nfunc (v *WorkflowParams) IsSetShadowMode() bool {\n\treturn v != nil && v.ShadowMode != nil\n}\n\n// GetExitCondition returns the value of ExitCondition if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetExitCondition() (o *ExitCondition) {\n\tif v != nil && v.ExitCondition != nil {\n\t\treturn v.ExitCondition\n\t}\n\n\treturn\n}\n\n// IsSetExitCondition returns true if ExitCondition is not nil.\nfunc (v *WorkflowParams) IsSetExitCondition() bool {\n\treturn v != nil && v.ExitCondition != nil\n}\n\n// GetConcurrency returns the value of Concurrency if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetConcurrency() (o int32) {\n\tif v != nil && v.Concurrency != nil {\n\t\treturn *v.Concurrency\n\t}\n\n\treturn\n}\n\n// IsSetConcurrency returns true if Concurrency is not nil.\nfunc (v *WorkflowParams) IsSetConcurrency() bool {\n\treturn v != nil && v.Concurrency != nil\n}\n\n// GetLastRunResult returns the value of LastRunResult if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowParams) GetLastRunResult() (o *WorkflowResult) {\n\tif v != nil && v.LastRunResult != nil {\n\t\treturn v.LastRunResult\n\t}\n\n\treturn\n}\n\n// IsSetLastRunResult returns true if LastRunResult is not nil.\nfunc (v *WorkflowParams) IsSetLastRunResult() bool {\n\treturn v != nil && v.LastRunResult != nil\n}\n\ntype WorkflowResult struct {\n\tSucceeded *int32 `json:\"succeeded,omitempty\"`\n\tSkipped   *int32 `json:\"skipped,omitempty\"`\n\tFailed    *int32 `json:\"failed,omitempty\"`\n}\n\n// ToWire translates a WorkflowResult struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowResult) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Succeeded != nil {\n\t\tw, err = wire.NewValueI32(*(v.Succeeded)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Skipped != nil {\n\t\tw, err = wire.NewValueI32(*(v.Skipped)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Failed != nil {\n\t\tw, err = wire.NewValueI32(*(v.Failed)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowResult struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowResult struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowResult\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowResult) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Succeeded = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Skipped = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Failed = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowResult struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowResult struct could not be encoded.\nfunc (v *WorkflowResult) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Succeeded != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Succeeded)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Skipped != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Skipped)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Failed != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Failed)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowResult struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowResult struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowResult) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Succeeded = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Skipped = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Failed = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowResult\n// struct.\nfunc (v *WorkflowResult) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Succeeded != nil {\n\t\tfields[i] = fmt.Sprintf(\"Succeeded: %v\", *(v.Succeeded))\n\t\ti++\n\t}\n\tif v.Skipped != nil {\n\t\tfields[i] = fmt.Sprintf(\"Skipped: %v\", *(v.Skipped))\n\t\ti++\n\t}\n\tif v.Failed != nil {\n\t\tfields[i] = fmt.Sprintf(\"Failed: %v\", *(v.Failed))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowResult{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowResult match the\n// provided WorkflowResult.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowResult) Equals(rhs *WorkflowResult) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Succeeded, rhs.Succeeded) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Skipped, rhs.Skipped) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Failed, rhs.Failed) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowResult.\nfunc (v *WorkflowResult) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Succeeded != nil {\n\t\tenc.AddInt32(\"succeeded\", *v.Succeeded)\n\t}\n\tif v.Skipped != nil {\n\t\tenc.AddInt32(\"skipped\", *v.Skipped)\n\t}\n\tif v.Failed != nil {\n\t\tenc.AddInt32(\"failed\", *v.Failed)\n\t}\n\treturn err\n}\n\n// GetSucceeded returns the value of Succeeded if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowResult) GetSucceeded() (o int32) {\n\tif v != nil && v.Succeeded != nil {\n\t\treturn *v.Succeeded\n\t}\n\n\treturn\n}\n\n// IsSetSucceeded returns true if Succeeded is not nil.\nfunc (v *WorkflowResult) IsSetSucceeded() bool {\n\treturn v != nil && v.Succeeded != nil\n}\n\n// GetSkipped returns the value of Skipped if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowResult) GetSkipped() (o int32) {\n\tif v != nil && v.Skipped != nil {\n\t\treturn *v.Skipped\n\t}\n\n\treturn\n}\n\n// IsSetSkipped returns true if Skipped is not nil.\nfunc (v *WorkflowResult) IsSetSkipped() bool {\n\treturn v != nil && v.Skipped != nil\n}\n\n// GetFailed returns the value of Failed if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowResult) GetFailed() (o int32) {\n\tif v != nil && v.Failed != nil {\n\t\treturn *v.Failed\n\t}\n\n\treturn\n}\n\n// IsSetFailed returns true if Failed is not nil.\nfunc (v *WorkflowResult) IsSetFailed() bool {\n\treturn v != nil && v.Failed != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"shadower\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/shadower\",\n\tFilePath: \"shadower.thrift\",\n\tSHA1:     \"f29a425548641e51e02ec1978279c1e37b9df792\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017-2021 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.shadower\\n\\ninclude \\\"shared.thrift\\\"\\n\\nconst string LocalDomainName = \\\"cadence-shadower\\\"\\nconst string TaskList = \\\"cadence-shadower-tl\\\"\\n\\nconst string WorkflowName = \\\"cadence-shadow-workflow\\\"\\n\\nconst string ScanWorkflowActivityName = \\\"scanWorkflowActivity\\\"\\nconst string ReplayWorkflowActivityName = \\\"replayWorkflowActivity\\\"\\n\\nconst string WorkflowIDSuffix = \\\"-shadow-workflow\\\"\\n\\nconst string ErrReasonDomainNotExists = \\\"domain not exists\\\"\\nconst string ErrReasonInvalidQuery = \\\"invalid visibility query\\\"\\nconst string ErrReasonWorkflowTypeNotRegistered = \\\"workflow type not registered\\\"\\nconst string ErrNonRetryableType = \\\"com.uber.cadence.internal.shadowing.NonRetryableException\\\"\\n\\nenum Mode {\\n  Normal,\\n  Continuous,\\n}\\n\\nstruct ExitCondition {\\n  10: optional i32 expirationIntervalInSeconds\\n  20: optional i32 shadowCount\\n}\\n\\nstruct WorkflowParams {\\n  10: optional string domain \\n  20: optional string taskList\\n  30: optional string workflowQuery\\n  40: optional binary nextPageToken\\n  50: optional double samplingRate\\n  60: optional Mode shadowMode\\n  70: optional ExitCondition exitCondition\\n  80: optional i32 concurrency\\n  90: optional WorkflowResult lastRunResult\\n}\\n\\nstruct WorkflowResult {\\n  10: optional i32 succeeded\\n  20: optional i32 skipped\\n  30: optional i32 failed\\n}\\n\\nstruct ScanWorkflowActivityParams {\\n  10: optional string domain\\n  20: optional string workflowQuery\\n  30: optional binary nextPageToken\\n  40: optional i32 pageSize\\n  50: optional double samplingRate\\n}\\n\\nstruct ScanWorkflowActivityResult {\\n  10: optional list<shared.WorkflowExecution> executions\\n  20: optional binary nextPageToken\\n}\\n\\nstruct ReplayWorkflowActivityParams {\\n  10: optional string domain\\n  20: optional list<shared.WorkflowExecution> executions\\n}\\n\\nstruct ReplayWorkflowActivityResult {\\n  10: optional i32 succeeded\\n  20: optional i32 skipped\\n  30: optional i32 failed\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/shadower/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage shadower\n"
  },
  {
    "path": ".gen/go/shared/shared.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage shared\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\terrors \"errors\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tptr \"go.uber.org/thriftrw/ptr\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n)\n\ntype AccessDeniedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a AccessDeniedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AccessDeniedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AccessDeniedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AccessDeniedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AccessDeniedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AccessDeniedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of AccessDeniedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AccessDeniedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AccessDeniedError struct could not be encoded.\nfunc (v *AccessDeniedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AccessDeniedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AccessDeniedError struct could not be generated from the wire\n// representation.\nfunc (v *AccessDeniedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of AccessDeniedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AccessDeniedError\n// struct.\nfunc (v *AccessDeniedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"AccessDeniedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*AccessDeniedError) ErrorName() string {\n\treturn \"AccessDeniedError\"\n}\n\n// Equals returns true if all the fields of this AccessDeniedError match the\n// provided AccessDeniedError.\n//\n// This function performs a deep comparison.\nfunc (v *AccessDeniedError) Equals(rhs *AccessDeniedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AccessDeniedError.\nfunc (v *AccessDeniedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *AccessDeniedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *AccessDeniedError) Error() string {\n\treturn v.String()\n}\n\ntype ActiveClusterInfo struct {\n\tActiveClusterName *string `json:\"activeClusterName,omitempty\"`\n\tFailoverVersion   *int64  `json:\"failoverVersion,omitempty\"`\n}\n\n// ToWire translates a ActiveClusterInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActiveClusterInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActiveClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActiveClusterInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActiveClusterInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActiveClusterInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActiveClusterInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActiveClusterInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActiveClusterInfo struct could not be encoded.\nfunc (v *ActiveClusterInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActiveClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActiveClusterInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActiveClusterInfo struct could not be generated from the wire\n// representation.\nfunc (v *ActiveClusterInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActiveClusterInfo\n// struct.\nfunc (v *ActiveClusterInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ActiveClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterName: %v\", *(v.ActiveClusterName))\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActiveClusterInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ActiveClusterInfo match the\n// provided ActiveClusterInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ActiveClusterInfo) Equals(rhs *ActiveClusterInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClusterName, rhs.ActiveClusterName) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActiveClusterInfo.\nfunc (v *ActiveClusterInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tenc.AddString(\"activeClusterName\", *v.ActiveClusterName)\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\treturn err\n}\n\n// GetActiveClusterName returns the value of ActiveClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterInfo) GetActiveClusterName() (o string) {\n\tif v != nil && v.ActiveClusterName != nil {\n\t\treturn *v.ActiveClusterName\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterName returns true if ActiveClusterName is not nil.\nfunc (v *ActiveClusterInfo) IsSetActiveClusterName() bool {\n\treturn v != nil && v.ActiveClusterName != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterInfo) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *ActiveClusterInfo) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\ntype ActiveClusterSelectionPolicy struct {\n\tClusterAttribute   *ClusterAttribute               `json:\"clusterAttribute,omitempty\"`\n\tStrategy           *ActiveClusterSelectionStrategy `json:\"strategy,omitempty\"`\n\tStickyRegion       *string                         `json:\"stickyRegion,omitempty\"`\n\tExternalEntityType *string                         `json:\"externalEntityType,omitempty\"`\n\tExternalEntityKey  *string                         `json:\"externalEntityKey,omitempty\"`\n}\n\n// ToWire translates a ActiveClusterSelectionPolicy struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActiveClusterSelectionPolicy) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ClusterAttribute != nil {\n\t\tw, err = v.ClusterAttribute.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.Strategy != nil {\n\t\tw, err = v.Strategy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StickyRegion != nil {\n\t\tw, err = wire.NewValueString(*(v.StickyRegion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalEntityType != nil {\n\t\tw, err = wire.NewValueString(*(v.ExternalEntityType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalEntityKey != nil {\n\t\tw, err = wire.NewValueString(*(v.ExternalEntityKey)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ClusterAttribute_Read(w wire.Value) (*ClusterAttribute, error) {\n\tvar v ClusterAttribute\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActiveClusterSelectionStrategy_Read(w wire.Value) (ActiveClusterSelectionStrategy, error) {\n\tvar v ActiveClusterSelectionStrategy\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a ActiveClusterSelectionPolicy struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActiveClusterSelectionPolicy struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActiveClusterSelectionPolicy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActiveClusterSelectionPolicy) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClusterAttribute, err = _ClusterAttribute_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ActiveClusterSelectionStrategy\n\t\t\t\tx, err = _ActiveClusterSelectionStrategy_Read(field.Value)\n\t\t\t\tv.Strategy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StickyRegion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ExternalEntityType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ExternalEntityKey = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActiveClusterSelectionPolicy struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActiveClusterSelectionPolicy struct could not be encoded.\nfunc (v *ActiveClusterSelectionPolicy) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ClusterAttribute != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClusterAttribute.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Strategy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Strategy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyRegion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StickyRegion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalEntityType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ExternalEntityType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalEntityKey != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ExternalEntityKey)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ClusterAttribute_Decode(sr stream.Reader) (*ClusterAttribute, error) {\n\tvar v ClusterAttribute\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActiveClusterSelectionStrategy_Decode(sr stream.Reader) (ActiveClusterSelectionStrategy, error) {\n\tvar v ActiveClusterSelectionStrategy\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a ActiveClusterSelectionPolicy struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActiveClusterSelectionPolicy struct could not be generated from the wire\n// representation.\nfunc (v *ActiveClusterSelectionPolicy) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TStruct:\n\t\t\tv.ClusterAttribute, err = _ClusterAttribute_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x ActiveClusterSelectionStrategy\n\t\t\tx, err = _ActiveClusterSelectionStrategy_Decode(sr)\n\t\t\tv.Strategy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StickyRegion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ExternalEntityType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ExternalEntityKey = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActiveClusterSelectionPolicy\n// struct.\nfunc (v *ActiveClusterSelectionPolicy) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.ClusterAttribute != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterAttribute: %v\", v.ClusterAttribute)\n\t\ti++\n\t}\n\tif v.Strategy != nil {\n\t\tfields[i] = fmt.Sprintf(\"Strategy: %v\", *(v.Strategy))\n\t\ti++\n\t}\n\tif v.StickyRegion != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyRegion: %v\", *(v.StickyRegion))\n\t\ti++\n\t}\n\tif v.ExternalEntityType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalEntityType: %v\", *(v.ExternalEntityType))\n\t\ti++\n\t}\n\tif v.ExternalEntityKey != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalEntityKey: %v\", *(v.ExternalEntityKey))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActiveClusterSelectionPolicy{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ActiveClusterSelectionStrategy_EqualsPtr(lhs, rhs *ActiveClusterSelectionStrategy) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ActiveClusterSelectionPolicy match the\n// provided ActiveClusterSelectionPolicy.\n//\n// This function performs a deep comparison.\nfunc (v *ActiveClusterSelectionPolicy) Equals(rhs *ActiveClusterSelectionPolicy) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ClusterAttribute == nil && rhs.ClusterAttribute == nil) || (v.ClusterAttribute != nil && rhs.ClusterAttribute != nil && v.ClusterAttribute.Equals(rhs.ClusterAttribute))) {\n\t\treturn false\n\t}\n\tif !_ActiveClusterSelectionStrategy_EqualsPtr(v.Strategy, rhs.Strategy) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StickyRegion, rhs.StickyRegion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ExternalEntityType, rhs.ExternalEntityType) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ExternalEntityKey, rhs.ExternalEntityKey) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActiveClusterSelectionPolicy.\nfunc (v *ActiveClusterSelectionPolicy) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ClusterAttribute != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterAttribute\", v.ClusterAttribute))\n\t}\n\tif v.Strategy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"strategy\", *v.Strategy))\n\t}\n\tif v.StickyRegion != nil {\n\t\tenc.AddString(\"stickyRegion\", *v.StickyRegion)\n\t}\n\tif v.ExternalEntityType != nil {\n\t\tenc.AddString(\"externalEntityType\", *v.ExternalEntityType)\n\t}\n\tif v.ExternalEntityKey != nil {\n\t\tenc.AddString(\"externalEntityKey\", *v.ExternalEntityKey)\n\t}\n\treturn err\n}\n\n// GetClusterAttribute returns the value of ClusterAttribute if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterSelectionPolicy) GetClusterAttribute() (o *ClusterAttribute) {\n\tif v != nil && v.ClusterAttribute != nil {\n\t\treturn v.ClusterAttribute\n\t}\n\n\treturn\n}\n\n// IsSetClusterAttribute returns true if ClusterAttribute is not nil.\nfunc (v *ActiveClusterSelectionPolicy) IsSetClusterAttribute() bool {\n\treturn v != nil && v.ClusterAttribute != nil\n}\n\n// GetStrategy returns the value of Strategy if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterSelectionPolicy) GetStrategy() (o ActiveClusterSelectionStrategy) {\n\tif v != nil && v.Strategy != nil {\n\t\treturn *v.Strategy\n\t}\n\n\treturn\n}\n\n// IsSetStrategy returns true if Strategy is not nil.\nfunc (v *ActiveClusterSelectionPolicy) IsSetStrategy() bool {\n\treturn v != nil && v.Strategy != nil\n}\n\n// GetStickyRegion returns the value of StickyRegion if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterSelectionPolicy) GetStickyRegion() (o string) {\n\tif v != nil && v.StickyRegion != nil {\n\t\treturn *v.StickyRegion\n\t}\n\n\treturn\n}\n\n// IsSetStickyRegion returns true if StickyRegion is not nil.\nfunc (v *ActiveClusterSelectionPolicy) IsSetStickyRegion() bool {\n\treturn v != nil && v.StickyRegion != nil\n}\n\n// GetExternalEntityType returns the value of ExternalEntityType if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterSelectionPolicy) GetExternalEntityType() (o string) {\n\tif v != nil && v.ExternalEntityType != nil {\n\t\treturn *v.ExternalEntityType\n\t}\n\n\treturn\n}\n\n// IsSetExternalEntityType returns true if ExternalEntityType is not nil.\nfunc (v *ActiveClusterSelectionPolicy) IsSetExternalEntityType() bool {\n\treturn v != nil && v.ExternalEntityType != nil\n}\n\n// GetExternalEntityKey returns the value of ExternalEntityKey if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusterSelectionPolicy) GetExternalEntityKey() (o string) {\n\tif v != nil && v.ExternalEntityKey != nil {\n\t\treturn *v.ExternalEntityKey\n\t}\n\n\treturn\n}\n\n// IsSetExternalEntityKey returns true if ExternalEntityKey is not nil.\nfunc (v *ActiveClusterSelectionPolicy) IsSetExternalEntityKey() bool {\n\treturn v != nil && v.ExternalEntityKey != nil\n}\n\ntype ActiveClusterSelectionStrategy int32\n\nconst (\n\tActiveClusterSelectionStrategyRegionSticky   ActiveClusterSelectionStrategy = 0\n\tActiveClusterSelectionStrategyExternalEntity ActiveClusterSelectionStrategy = 1\n)\n\n// ActiveClusterSelectionStrategy_Values returns all recognized values of ActiveClusterSelectionStrategy.\nfunc ActiveClusterSelectionStrategy_Values() []ActiveClusterSelectionStrategy {\n\treturn []ActiveClusterSelectionStrategy{\n\t\tActiveClusterSelectionStrategyRegionSticky,\n\t\tActiveClusterSelectionStrategyExternalEntity,\n\t}\n}\n\n// UnmarshalText tries to decode ActiveClusterSelectionStrategy from a byte slice\n// containing its name.\n//\n//\tvar v ActiveClusterSelectionStrategy\n//\terr := v.UnmarshalText([]byte(\"REGION_STICKY\"))\nfunc (v *ActiveClusterSelectionStrategy) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"REGION_STICKY\":\n\t\t*v = ActiveClusterSelectionStrategyRegionSticky\n\t\treturn nil\n\tcase \"EXTERNAL_ENTITY\":\n\t\t*v = ActiveClusterSelectionStrategyExternalEntity\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ActiveClusterSelectionStrategy\", err)\n\t\t}\n\t\t*v = ActiveClusterSelectionStrategy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ActiveClusterSelectionStrategy to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ActiveClusterSelectionStrategy) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"REGION_STICKY\"), nil\n\tcase 1:\n\t\treturn []byte(\"EXTERNAL_ENTITY\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActiveClusterSelectionStrategy.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ActiveClusterSelectionStrategy) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"REGION_STICKY\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"EXTERNAL_ENTITY\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ActiveClusterSelectionStrategy) Ptr() *ActiveClusterSelectionStrategy {\n\treturn &v\n}\n\n// Encode encodes ActiveClusterSelectionStrategy directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ActiveClusterSelectionStrategy\n//\treturn v.Encode(sWriter)\nfunc (v ActiveClusterSelectionStrategy) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ActiveClusterSelectionStrategy into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ActiveClusterSelectionStrategy) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ActiveClusterSelectionStrategy from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ActiveClusterSelectionStrategy(0), err\n//\t}\n//\n//\tvar v ActiveClusterSelectionStrategy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ActiveClusterSelectionStrategy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ActiveClusterSelectionStrategy) FromWire(w wire.Value) error {\n\t*v = (ActiveClusterSelectionStrategy)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ActiveClusterSelectionStrategy directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ActiveClusterSelectionStrategy\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ActiveClusterSelectionStrategy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ActiveClusterSelectionStrategy) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ActiveClusterSelectionStrategy)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ActiveClusterSelectionStrategy.\nfunc (v ActiveClusterSelectionStrategy) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"REGION_STICKY\"\n\tcase 1:\n\t\treturn \"EXTERNAL_ENTITY\"\n\t}\n\treturn fmt.Sprintf(\"ActiveClusterSelectionStrategy(%d)\", w)\n}\n\n// Equals returns true if this ActiveClusterSelectionStrategy value matches the provided\n// value.\nfunc (v ActiveClusterSelectionStrategy) Equals(rhs ActiveClusterSelectionStrategy) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ActiveClusterSelectionStrategy into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ActiveClusterSelectionStrategy) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"REGION_STICKY\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"EXTERNAL_ENTITY\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ActiveClusterSelectionStrategy from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ActiveClusterSelectionStrategy) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ActiveClusterSelectionStrategy\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ActiveClusterSelectionStrategy\")\n\t\t}\n\t\t*v = (ActiveClusterSelectionStrategy)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ActiveClusterSelectionStrategy\")\n\t}\n}\n\ntype ActiveClusters struct {\n\tActiveClustersByRegion           map[string]*ActiveClusterInfo     `json:\"activeClustersByRegion,omitempty\"`\n\tActiveClustersByClusterAttribute map[string]*ClusterAttributeScope `json:\"activeClustersByClusterAttribute,omitempty\"`\n}\n\ntype _Map_String_ActiveClusterInfo_MapItemList map[string]*ActiveClusterInfo\n\nfunc (m _Map_String_ActiveClusterInfo_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ActiveClusterInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_ActiveClusterInfo_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_ActiveClusterInfo_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_ActiveClusterInfo_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_ActiveClusterInfo_MapItemList) Close() {}\n\ntype _Map_String_ClusterAttributeScope_MapItemList map[string]*ClusterAttributeScope\n\nfunc (m _Map_String_ClusterAttributeScope_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ClusterAttributeScope', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_ClusterAttributeScope_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_ClusterAttributeScope_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_ClusterAttributeScope_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_ClusterAttributeScope_MapItemList) Close() {}\n\n// ToWire translates a ActiveClusters struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActiveClusters) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActiveClustersByRegion != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_ActiveClusterInfo_MapItemList(v.ActiveClustersByRegion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClustersByClusterAttribute != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_ClusterAttributeScope_MapItemList(v.ActiveClustersByClusterAttribute)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ActiveClusterInfo_Read(w wire.Value) (*ActiveClusterInfo, error) {\n\tvar v ActiveClusterInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_ActiveClusterInfo_Read(m wire.MapItemList) (map[string]*ActiveClusterInfo, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*ActiveClusterInfo, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _ActiveClusterInfo_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _ClusterAttributeScope_Read(w wire.Value) (*ClusterAttributeScope, error) {\n\tvar v ClusterAttributeScope\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_ClusterAttributeScope_Read(m wire.MapItemList) (map[string]*ClusterAttributeScope, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*ClusterAttributeScope, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _ClusterAttributeScope_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ActiveClusters struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActiveClusters struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActiveClusters\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActiveClusters) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActiveClustersByRegion, err = _Map_String_ActiveClusterInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActiveClustersByClusterAttribute, err = _Map_String_ClusterAttributeScope_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_ActiveClusterInfo_Encode(val map[string]*ActiveClusterInfo, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ActiveClusterInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\nfunc _Map_String_ClusterAttributeScope_Encode(val map[string]*ClusterAttributeScope, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ClusterAttributeScope', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a ActiveClusters struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActiveClusters struct could not be encoded.\nfunc (v *ActiveClusters) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActiveClustersByRegion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_ActiveClusterInfo_Encode(v.ActiveClustersByRegion, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClustersByClusterAttribute != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_ClusterAttributeScope_Encode(v.ActiveClustersByClusterAttribute, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ActiveClusterInfo_Decode(sr stream.Reader) (*ActiveClusterInfo, error) {\n\tvar v ActiveClusterInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_ActiveClusterInfo_Decode(sr stream.Reader) (map[string]*ActiveClusterInfo, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*ActiveClusterInfo, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _ActiveClusterInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _ClusterAttributeScope_Decode(sr stream.Reader) (*ClusterAttributeScope, error) {\n\tvar v ClusterAttributeScope\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_ClusterAttributeScope_Decode(sr stream.Reader) (map[string]*ClusterAttributeScope, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*ClusterAttributeScope, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _ClusterAttributeScope_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ActiveClusters struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActiveClusters struct could not be generated from the wire\n// representation.\nfunc (v *ActiveClusters) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.ActiveClustersByRegion, err = _Map_String_ActiveClusterInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TMap:\n\t\t\tv.ActiveClustersByClusterAttribute, err = _Map_String_ClusterAttributeScope_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActiveClusters\n// struct.\nfunc (v *ActiveClusters) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ActiveClustersByRegion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClustersByRegion: %v\", v.ActiveClustersByRegion)\n\t\ti++\n\t}\n\tif v.ActiveClustersByClusterAttribute != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClustersByClusterAttribute: %v\", v.ActiveClustersByClusterAttribute)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActiveClusters{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_ActiveClusterInfo_Equals(lhs, rhs map[string]*ActiveClusterInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _Map_String_ClusterAttributeScope_Equals(lhs, rhs map[string]*ClusterAttributeScope) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this ActiveClusters match the\n// provided ActiveClusters.\n//\n// This function performs a deep comparison.\nfunc (v *ActiveClusters) Equals(rhs *ActiveClusters) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ActiveClustersByRegion == nil && rhs.ActiveClustersByRegion == nil) || (v.ActiveClustersByRegion != nil && rhs.ActiveClustersByRegion != nil && _Map_String_ActiveClusterInfo_Equals(v.ActiveClustersByRegion, rhs.ActiveClustersByRegion))) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClustersByClusterAttribute == nil && rhs.ActiveClustersByClusterAttribute == nil) || (v.ActiveClustersByClusterAttribute != nil && rhs.ActiveClustersByClusterAttribute != nil && _Map_String_ClusterAttributeScope_Equals(v.ActiveClustersByClusterAttribute, rhs.ActiveClustersByClusterAttribute))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_ActiveClusterInfo_Zapper map[string]*ActiveClusterInfo\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_ActiveClusterInfo_Zapper.\nfunc (m _Map_String_ActiveClusterInfo_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\ntype _Map_String_ClusterAttributeScope_Zapper map[string]*ClusterAttributeScope\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_ClusterAttributeScope_Zapper.\nfunc (m _Map_String_ClusterAttributeScope_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActiveClusters.\nfunc (v *ActiveClusters) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActiveClustersByRegion != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClustersByRegion\", (_Map_String_ActiveClusterInfo_Zapper)(v.ActiveClustersByRegion)))\n\t}\n\tif v.ActiveClustersByClusterAttribute != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClustersByClusterAttribute\", (_Map_String_ClusterAttributeScope_Zapper)(v.ActiveClustersByClusterAttribute)))\n\t}\n\treturn err\n}\n\n// GetActiveClustersByRegion returns the value of ActiveClustersByRegion if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusters) GetActiveClustersByRegion() (o map[string]*ActiveClusterInfo) {\n\tif v != nil && v.ActiveClustersByRegion != nil {\n\t\treturn v.ActiveClustersByRegion\n\t}\n\n\treturn\n}\n\n// IsSetActiveClustersByRegion returns true if ActiveClustersByRegion is not nil.\nfunc (v *ActiveClusters) IsSetActiveClustersByRegion() bool {\n\treturn v != nil && v.ActiveClustersByRegion != nil\n}\n\n// GetActiveClustersByClusterAttribute returns the value of ActiveClustersByClusterAttribute if it is set or its\n// zero value if it is unset.\nfunc (v *ActiveClusters) GetActiveClustersByClusterAttribute() (o map[string]*ClusterAttributeScope) {\n\tif v != nil && v.ActiveClustersByClusterAttribute != nil {\n\t\treturn v.ActiveClustersByClusterAttribute\n\t}\n\n\treturn\n}\n\n// IsSetActiveClustersByClusterAttribute returns true if ActiveClustersByClusterAttribute is not nil.\nfunc (v *ActiveClusters) IsSetActiveClustersByClusterAttribute() bool {\n\treturn v != nil && v.ActiveClustersByClusterAttribute != nil\n}\n\ntype ActivityLocalDispatchInfo struct {\n\tActivityId                      *string `json:\"activityId,omitempty\"`\n\tScheduledTimestamp              *int64  `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp                *int64  `json:\"startedTimestamp,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64  `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tTaskToken                       []byte  `json:\"taskToken,omitempty\"`\n}\n\n// ToWire translates a ActivityLocalDispatchInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityLocalDispatchInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestampOfThisAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityLocalDispatchInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityLocalDispatchInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityLocalDispatchInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityLocalDispatchInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityLocalDispatchInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityLocalDispatchInfo struct could not be encoded.\nfunc (v *ActivityLocalDispatchInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestampOfThisAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityLocalDispatchInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityLocalDispatchInfo struct could not be generated from the wire\n// representation.\nfunc (v *ActivityLocalDispatchInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityLocalDispatchInfo\n// struct.\nfunc (v *ActivityLocalDispatchInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestampOfThisAttempt: %v\", *(v.ScheduledTimestampOfThisAttempt))\n\t\ti++\n\t}\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityLocalDispatchInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityLocalDispatchInfo match the\n// provided ActivityLocalDispatchInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityLocalDispatchInfo) Equals(rhs *ActivityLocalDispatchInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestampOfThisAttempt, rhs.ScheduledTimestampOfThisAttempt) {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityLocalDispatchInfo.\nfunc (v *ActivityLocalDispatchInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tenc.AddInt64(\"scheduledTimestampOfThisAttempt\", *v.ScheduledTimestampOfThisAttempt)\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityLocalDispatchInfo) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *ActivityLocalDispatchInfo) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityLocalDispatchInfo) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *ActivityLocalDispatchInfo) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityLocalDispatchInfo) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *ActivityLocalDispatchInfo) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetScheduledTimestampOfThisAttempt returns the value of ScheduledTimestampOfThisAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityLocalDispatchInfo) GetScheduledTimestampOfThisAttempt() (o int64) {\n\tif v != nil && v.ScheduledTimestampOfThisAttempt != nil {\n\t\treturn *v.ScheduledTimestampOfThisAttempt\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestampOfThisAttempt returns true if ScheduledTimestampOfThisAttempt is not nil.\nfunc (v *ActivityLocalDispatchInfo) IsSetScheduledTimestampOfThisAttempt() bool {\n\treturn v != nil && v.ScheduledTimestampOfThisAttempt != nil\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityLocalDispatchInfo) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *ActivityLocalDispatchInfo) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\ntype ActivityTaskCancelRequestedEventAttributes struct {\n\tActivityId                   *string `json:\"activityId,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskCancelRequestedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskCancelRequestedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityTaskCancelRequestedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskCancelRequestedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskCancelRequestedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskCancelRequestedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskCancelRequestedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskCancelRequestedEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityTaskCancelRequestedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskCancelRequestedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskCancelRequestedEventAttributes\n// struct.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskCancelRequestedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityTaskCancelRequestedEventAttributes match the\n// provided ActivityTaskCancelRequestedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) Equals(rhs *ActivityTaskCancelRequestedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskCancelRequestedEventAttributes.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *ActivityTaskCancelRequestedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype ActivityTaskCanceledEventAttributes struct {\n\tDetails                      []byte  `json:\"details,omitempty\"`\n\tLatestCancelRequestedEventId *int64  `json:\"latestCancelRequestedEventId,omitempty\"`\n\tScheduledEventId             *int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId               *int64  `json:\"startedEventId,omitempty\"`\n\tIdentity                     *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskCanceledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskCanceledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.LatestCancelRequestedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.LatestCancelRequestedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityTaskCanceledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskCanceledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskCanceledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskCanceledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LatestCancelRequestedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskCanceledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskCanceledEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskCanceledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LatestCancelRequestedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LatestCancelRequestedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityTaskCanceledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskCanceledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskCanceledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LatestCancelRequestedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskCanceledEventAttributes\n// struct.\nfunc (v *ActivityTaskCanceledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.LatestCancelRequestedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"LatestCancelRequestedEventId: %v\", *(v.LatestCancelRequestedEventId))\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskCanceledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityTaskCanceledEventAttributes match the\n// provided ActivityTaskCanceledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskCanceledEventAttributes) Equals(rhs *ActivityTaskCanceledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LatestCancelRequestedEventId, rhs.LatestCancelRequestedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskCanceledEventAttributes.\nfunc (v *ActivityTaskCanceledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.LatestCancelRequestedEventId != nil {\n\t\tenc.AddInt64(\"latestCancelRequestedEventId\", *v.LatestCancelRequestedEventId)\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCanceledEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *ActivityTaskCanceledEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetLatestCancelRequestedEventId returns the value of LatestCancelRequestedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCanceledEventAttributes) GetLatestCancelRequestedEventId() (o int64) {\n\tif v != nil && v.LatestCancelRequestedEventId != nil {\n\t\treturn *v.LatestCancelRequestedEventId\n\t}\n\n\treturn\n}\n\n// IsSetLatestCancelRequestedEventId returns true if LatestCancelRequestedEventId is not nil.\nfunc (v *ActivityTaskCanceledEventAttributes) IsSetLatestCancelRequestedEventId() bool {\n\treturn v != nil && v.LatestCancelRequestedEventId != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCanceledEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *ActivityTaskCanceledEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCanceledEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ActivityTaskCanceledEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCanceledEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *ActivityTaskCanceledEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype ActivityTaskCompletedEventAttributes struct {\n\tResult           []byte  `json:\"result,omitempty\"`\n\tScheduledEventId *int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId   *int64  `json:\"startedEventId,omitempty\"`\n\tIdentity         *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskCompletedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskCompletedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityTaskCompletedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskCompletedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskCompletedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskCompletedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskCompletedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskCompletedEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskCompletedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityTaskCompletedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskCompletedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskCompletedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskCompletedEventAttributes\n// struct.\nfunc (v *ActivityTaskCompletedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskCompletedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityTaskCompletedEventAttributes match the\n// provided ActivityTaskCompletedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskCompletedEventAttributes) Equals(rhs *ActivityTaskCompletedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskCompletedEventAttributes.\nfunc (v *ActivityTaskCompletedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCompletedEventAttributes) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *ActivityTaskCompletedEventAttributes) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCompletedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *ActivityTaskCompletedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCompletedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ActivityTaskCompletedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskCompletedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *ActivityTaskCompletedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype ActivityTaskFailedEventAttributes struct {\n\tReason           *string `json:\"reason,omitempty\"`\n\tDetails          []byte  `json:\"details,omitempty\"`\n\tScheduledEventId *int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId   *int64  `json:\"startedEventId,omitempty\"`\n\tIdentity         *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityTaskFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskFailedEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityTaskFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskFailedEventAttributes\n// struct.\nfunc (v *ActivityTaskFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityTaskFailedEventAttributes match the\n// provided ActivityTaskFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskFailedEventAttributes) Equals(rhs *ActivityTaskFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskFailedEventAttributes.\nfunc (v *ActivityTaskFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskFailedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *ActivityTaskFailedEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskFailedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *ActivityTaskFailedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskFailedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *ActivityTaskFailedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskFailedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ActivityTaskFailedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskFailedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *ActivityTaskFailedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype ActivityTaskScheduledEventAttributes struct {\n\tActivityId                    *string       `json:\"activityId,omitempty\"`\n\tActivityType                  *ActivityType `json:\"activityType,omitempty\"`\n\tDomain                        *string       `json:\"domain,omitempty\"`\n\tTaskList                      *TaskList     `json:\"taskList,omitempty\"`\n\tInput                         []byte        `json:\"input,omitempty\"`\n\tScheduleToCloseTimeoutSeconds *int32        `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32        `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tStartToCloseTimeoutSeconds    *int32        `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds       *int32        `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventId  *int64        `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tRetryPolicy                   *RetryPolicy  `json:\"retryPolicy,omitempty\"`\n\tHeader                        *Header       `json:\"header,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskScheduledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskScheduledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [12]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tw, err = v.ActivityType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 25, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 45, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.StartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.HeartbeatTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ActivityType_Read(w wire.Value) (*ActivityType, error) {\n\tvar v ActivityType\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TaskList_Read(w wire.Value) (*TaskList, error) {\n\tvar v TaskList\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RetryPolicy_Read(w wire.Value) (*RetryPolicy, error) {\n\tvar v RetryPolicy\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Header_Read(w wire.Value) (*Header, error) {\n\tvar v Header\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ActivityTaskScheduledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskScheduledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskScheduledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskScheduledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityType, err = _ActivityType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 25:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 45:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskScheduledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskScheduledEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskScheduledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 25, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 45, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.HeartbeatTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ActivityType_Decode(sr stream.Reader) (*ActivityType, error) {\n\tvar v ActivityType\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TaskList_Decode(sr stream.Reader) (*TaskList, error) {\n\tvar v TaskList\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RetryPolicy_Decode(sr stream.Reader) (*RetryPolicy, error) {\n\tvar v RetryPolicy\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Header_Decode(sr stream.Reader) (*Header, error) {\n\tvar v Header\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ActivityTaskScheduledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskScheduledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskScheduledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityType, err = _ActivityType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 25 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 45 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskScheduledEventAttributes\n// struct.\nfunc (v *ActivityTaskScheduledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [12]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityType: %v\", v.ActivityType)\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToCloseTimeoutSeconds: %v\", *(v.ScheduleToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToCloseTimeoutSeconds: %v\", *(v.StartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatTimeoutSeconds: %v\", *(v.HeartbeatTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskScheduledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ActivityTaskScheduledEventAttributes match the\n// provided ActivityTaskScheduledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskScheduledEventAttributes) Equals(rhs *ActivityTaskScheduledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !((v.ActivityType == nil && rhs.ActivityType == nil) || (v.ActivityType != nil && rhs.ActivityType != nil && v.ActivityType.Equals(rhs.ActivityType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToCloseTimeoutSeconds, rhs.ScheduleToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StartToCloseTimeoutSeconds, rhs.StartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.HeartbeatTimeoutSeconds, rhs.HeartbeatTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskScheduledEventAttributes.\nfunc (v *ActivityTaskScheduledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.ActivityType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityType\", v.ActivityType))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToCloseTimeoutSeconds\", *v.ScheduleToCloseTimeoutSeconds)\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"startToCloseTimeoutSeconds\", *v.StartToCloseTimeoutSeconds)\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"heartbeatTimeoutSeconds\", *v.HeartbeatTimeoutSeconds)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetActivityType returns the value of ActivityType if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\n\treturn\n}\n\n// IsSetActivityType returns true if ActivityType is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetActivityType() bool {\n\treturn v != nil && v.ActivityType != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetScheduleToCloseTimeoutSeconds returns the value of ScheduleToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToCloseTimeoutSeconds returns true if ScheduleToCloseTimeoutSeconds is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetScheduleToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToCloseTimeoutSeconds != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\n// GetStartToCloseTimeoutSeconds returns the value of StartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToCloseTimeoutSeconds returns true if StartToCloseTimeoutSeconds is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.StartToCloseTimeoutSeconds != nil\n}\n\n// GetHeartbeatTimeoutSeconds returns the value of HeartbeatTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatTimeoutSeconds returns true if HeartbeatTimeoutSeconds is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetHeartbeatTimeoutSeconds() bool {\n\treturn v != nil && v.HeartbeatTimeoutSeconds != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskScheduledEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *ActivityTaskScheduledEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\ntype ActivityTaskStartedEventAttributes struct {\n\tScheduledEventId   *int64  `json:\"scheduledEventId,omitempty\"`\n\tIdentity           *string `json:\"identity,omitempty\"`\n\tRequestId          *string `json:\"requestId,omitempty\"`\n\tAttempt            *int32  `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string `json:\"lastFailureReason,omitempty\"`\n\tLastFailureDetails []byte  `json:\"lastFailureDetails,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskStartedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskStartedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.LastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.LastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityTaskStartedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskStartedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskStartedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskStartedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskStartedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskStartedEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskStartedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityTaskStartedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskStartedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskStartedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.LastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskStartedEventAttributes\n// struct.\nfunc (v *ActivityTaskStartedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureReason: %v\", *(v.LastFailureReason))\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureDetails: %v\", v.LastFailureDetails)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskStartedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityTaskStartedEventAttributes match the\n// provided ActivityTaskStartedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskStartedEventAttributes) Equals(rhs *ActivityTaskStartedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastFailureReason, rhs.LastFailureReason) {\n\t\treturn false\n\t}\n\tif !((v.LastFailureDetails == nil && rhs.LastFailureDetails == nil) || (v.LastFailureDetails != nil && rhs.LastFailureDetails != nil && bytes.Equal(v.LastFailureDetails, rhs.LastFailureDetails))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskStartedEventAttributes.\nfunc (v *ActivityTaskStartedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.LastFailureReason != nil {\n\t\tenc.AddString(\"lastFailureReason\", *v.LastFailureReason)\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tenc.AddString(\"lastFailureDetails\", base64.StdEncoding.EncodeToString(v.LastFailureDetails))\n\t}\n\treturn err\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetLastFailureReason returns the value of LastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureReason returns true if LastFailureReason is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetLastFailureReason() bool {\n\treturn v != nil && v.LastFailureReason != nil\n}\n\n// GetLastFailureDetails returns the value of LastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskStartedEventAttributes) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureDetails returns true if LastFailureDetails is not nil.\nfunc (v *ActivityTaskStartedEventAttributes) IsSetLastFailureDetails() bool {\n\treturn v != nil && v.LastFailureDetails != nil\n}\n\ntype ActivityTaskTimedOutEventAttributes struct {\n\tDetails            []byte       `json:\"details,omitempty\"`\n\tScheduledEventId   *int64       `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId     *int64       `json:\"startedEventId,omitempty\"`\n\tTimeoutType        *TimeoutType `json:\"timeoutType,omitempty\"`\n\tLastFailureReason  *string      `json:\"lastFailureReason,omitempty\"`\n\tLastFailureDetails []byte       `json:\"lastFailureDetails,omitempty\"`\n}\n\n// ToWire translates a ActivityTaskTimedOutEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityTaskTimedOutEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tw, err = v.TimeoutType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.LastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.LastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TimeoutType_Read(w wire.Value) (TimeoutType, error) {\n\tvar v TimeoutType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a ActivityTaskTimedOutEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityTaskTimedOutEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityTaskTimedOutEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityTaskTimedOutEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TimeoutType\n\t\t\t\tx, err = _TimeoutType_Read(field.Value)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityTaskTimedOutEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityTaskTimedOutEventAttributes struct could not be encoded.\nfunc (v *ActivityTaskTimedOutEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimeoutType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TimeoutType_Decode(sr stream.Reader) (TimeoutType, error) {\n\tvar v TimeoutType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a ActivityTaskTimedOutEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityTaskTimedOutEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ActivityTaskTimedOutEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 5 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x TimeoutType\n\t\t\tx, err = _TimeoutType_Decode(sr)\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.LastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityTaskTimedOutEventAttributes\n// struct.\nfunc (v *ActivityTaskTimedOutEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureReason: %v\", *(v.LastFailureReason))\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureDetails: %v\", v.LastFailureDetails)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityTaskTimedOutEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _TimeoutType_EqualsPtr(lhs, rhs *TimeoutType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ActivityTaskTimedOutEventAttributes match the\n// provided ActivityTaskTimedOutEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityTaskTimedOutEventAttributes) Equals(rhs *ActivityTaskTimedOutEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_TimeoutType_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastFailureReason, rhs.LastFailureReason) {\n\t\treturn false\n\t}\n\tif !((v.LastFailureDetails == nil && rhs.LastFailureDetails == nil) || (v.LastFailureDetails != nil && rhs.LastFailureDetails != nil && bytes.Equal(v.LastFailureDetails, rhs.LastFailureDetails))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityTaskTimedOutEventAttributes.\nfunc (v *ActivityTaskTimedOutEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.TimeoutType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timeoutType\", *v.TimeoutType))\n\t}\n\tif v.LastFailureReason != nil {\n\t\tenc.AddString(\"lastFailureReason\", *v.LastFailureReason)\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tenc.AddString(\"lastFailureDetails\", base64.StdEncoding.EncodeToString(v.LastFailureDetails))\n\t}\n\treturn err\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\n// GetLastFailureReason returns the value of LastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureReason returns true if LastFailureReason is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetLastFailureReason() bool {\n\treturn v != nil && v.LastFailureReason != nil\n}\n\n// GetLastFailureDetails returns the value of LastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityTaskTimedOutEventAttributes) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureDetails returns true if LastFailureDetails is not nil.\nfunc (v *ActivityTaskTimedOutEventAttributes) IsSetLastFailureDetails() bool {\n\treturn v != nil && v.LastFailureDetails != nil\n}\n\ntype ActivityType struct {\n\tName *string `json:\"name,omitempty\"`\n}\n\n// ToWire translates a ActivityType struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityType) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ActivityType struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityType struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityType) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ActivityType struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityType struct could not be encoded.\nfunc (v *ActivityType) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ActivityType struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityType struct could not be generated from the wire\n// representation.\nfunc (v *ActivityType) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityType\n// struct.\nfunc (v *ActivityType) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityType{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ActivityType match the\n// provided ActivityType.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityType) Equals(rhs *ActivityType) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityType.\nfunc (v *ActivityType) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityType) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *ActivityType) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// Any is a logical duplicate of google.protobuf.Any.\n//\n// The intent of the type is the same, but it is not intended to be directly\n// compatible with google.protobuf.Any or any Thrift equivalent - this blob is\n// RPC-type agnostic by design (as the underlying data may be transported over\n// proto or thrift), and the data-bytes may be in any encoding.\n//\n// This is intentionally different from DataBlob, which supports only a handful\n// of known encodings so it can be interpreted everywhere.  Any supports literally\n// any contents, and needs to be considered opaque until it is given to something\n// that is expecting it.\n//\n// See ValueType to interpret the contents.\ntype Any struct {\n\tValueType *string `json:\"ValueType,omitempty\"`\n\tValue     []byte  `json:\"Value,omitempty\"`\n}\n\n// ToWire translates a Any struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Any) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ValueType != nil {\n\t\tw, err = wire.NewValueString(*(v.ValueType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tw, err = wire.NewValueBinary(v.Value), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a Any struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Any struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Any\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Any) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ValueType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Value, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Any struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Any struct could not be encoded.\nfunc (v *Any) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ValueType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ValueType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Value != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Value); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a Any struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Any struct could not be generated from the wire\n// representation.\nfunc (v *Any) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ValueType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Value, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Any\n// struct.\nfunc (v *Any) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ValueType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ValueType: %v\", *(v.ValueType))\n\t\ti++\n\t}\n\tif v.Value != nil {\n\t\tfields[i] = fmt.Sprintf(\"Value: %v\", v.Value)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Any{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this Any match the\n// provided Any.\n//\n// This function performs a deep comparison.\nfunc (v *Any) Equals(rhs *Any) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ValueType, rhs.ValueType) {\n\t\treturn false\n\t}\n\tif !((v.Value == nil && rhs.Value == nil) || (v.Value != nil && rhs.Value != nil && bytes.Equal(v.Value, rhs.Value))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Any.\nfunc (v *Any) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ValueType != nil {\n\t\tenc.AddString(\"ValueType\", *v.ValueType)\n\t}\n\tif v.Value != nil {\n\t\tenc.AddString(\"Value\", base64.StdEncoding.EncodeToString(v.Value))\n\t}\n\treturn err\n}\n\n// GetValueType returns the value of ValueType if it is set or its\n// zero value if it is unset.\nfunc (v *Any) GetValueType() (o string) {\n\tif v != nil && v.ValueType != nil {\n\t\treturn *v.ValueType\n\t}\n\n\treturn\n}\n\n// IsSetValueType returns true if ValueType is not nil.\nfunc (v *Any) IsSetValueType() bool {\n\treturn v != nil && v.ValueType != nil\n}\n\n// GetValue returns the value of Value if it is set or its\n// zero value if it is unset.\nfunc (v *Any) GetValue() (o []byte) {\n\tif v != nil && v.Value != nil {\n\t\treturn v.Value\n\t}\n\n\treturn\n}\n\n// IsSetValue returns true if Value is not nil.\nfunc (v *Any) IsSetValue() bool {\n\treturn v != nil && v.Value != nil\n}\n\ntype ApplyParentClosePolicyAttributes struct {\n\tChildDomainID     *string            `json:\"childDomainID,omitempty\"`\n\tChildWorkflowID   *string            `json:\"childWorkflowID,omitempty\"`\n\tChildRunID        *string            `json:\"childRunID,omitempty\"`\n\tParentClosePolicy *ParentClosePolicy `json:\"parentClosePolicy,omitempty\"`\n}\n\n// ToWire translates a ApplyParentClosePolicyAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ApplyParentClosePolicyAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ChildDomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.ChildDomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.ChildWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ChildRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.ChildRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tw, err = v.ParentClosePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ParentClosePolicy_Read(w wire.Value) (ParentClosePolicy, error) {\n\tvar v ParentClosePolicy\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a ApplyParentClosePolicyAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ApplyParentClosePolicyAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ApplyParentClosePolicyAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ApplyParentClosePolicyAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ChildDomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ChildWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ChildRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ParentClosePolicy\n\t\t\t\tx, err = _ParentClosePolicy_Read(field.Value)\n\t\t\t\tv.ParentClosePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ApplyParentClosePolicyAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ApplyParentClosePolicyAttributes struct could not be encoded.\nfunc (v *ApplyParentClosePolicyAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ChildDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ChildDomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ChildWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ChildRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentClosePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentClosePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ParentClosePolicy_Decode(sr stream.Reader) (ParentClosePolicy, error) {\n\tvar v ParentClosePolicy\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a ApplyParentClosePolicyAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ApplyParentClosePolicyAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ApplyParentClosePolicyAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ChildDomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ChildWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ChildRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x ParentClosePolicy\n\t\t\tx, err = _ParentClosePolicy_Decode(sr)\n\t\t\tv.ParentClosePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ApplyParentClosePolicyAttributes\n// struct.\nfunc (v *ApplyParentClosePolicyAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.ChildDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildDomainID: %v\", *(v.ChildDomainID))\n\t\ti++\n\t}\n\tif v.ChildWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowID: %v\", *(v.ChildWorkflowID))\n\t\ti++\n\t}\n\tif v.ChildRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildRunID: %v\", *(v.ChildRunID))\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentClosePolicy: %v\", *(v.ParentClosePolicy))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ApplyParentClosePolicyAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ParentClosePolicy_EqualsPtr(lhs, rhs *ParentClosePolicy) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ApplyParentClosePolicyAttributes match the\n// provided ApplyParentClosePolicyAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ApplyParentClosePolicyAttributes) Equals(rhs *ApplyParentClosePolicyAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ChildDomainID, rhs.ChildDomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ChildWorkflowID, rhs.ChildWorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ChildRunID, rhs.ChildRunID) {\n\t\treturn false\n\t}\n\tif !_ParentClosePolicy_EqualsPtr(v.ParentClosePolicy, rhs.ParentClosePolicy) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ApplyParentClosePolicyAttributes.\nfunc (v *ApplyParentClosePolicyAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ChildDomainID != nil {\n\t\tenc.AddString(\"childDomainID\", *v.ChildDomainID)\n\t}\n\tif v.ChildWorkflowID != nil {\n\t\tenc.AddString(\"childWorkflowID\", *v.ChildWorkflowID)\n\t}\n\tif v.ChildRunID != nil {\n\t\tenc.AddString(\"childRunID\", *v.ChildRunID)\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentClosePolicy\", *v.ParentClosePolicy))\n\t}\n\treturn err\n}\n\n// GetChildDomainID returns the value of ChildDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyAttributes) GetChildDomainID() (o string) {\n\tif v != nil && v.ChildDomainID != nil {\n\t\treturn *v.ChildDomainID\n\t}\n\n\treturn\n}\n\n// IsSetChildDomainID returns true if ChildDomainID is not nil.\nfunc (v *ApplyParentClosePolicyAttributes) IsSetChildDomainID() bool {\n\treturn v != nil && v.ChildDomainID != nil\n}\n\n// GetChildWorkflowID returns the value of ChildWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyAttributes) GetChildWorkflowID() (o string) {\n\tif v != nil && v.ChildWorkflowID != nil {\n\t\treturn *v.ChildWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowID returns true if ChildWorkflowID is not nil.\nfunc (v *ApplyParentClosePolicyAttributes) IsSetChildWorkflowID() bool {\n\treturn v != nil && v.ChildWorkflowID != nil\n}\n\n// GetChildRunID returns the value of ChildRunID if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyAttributes) GetChildRunID() (o string) {\n\tif v != nil && v.ChildRunID != nil {\n\t\treturn *v.ChildRunID\n\t}\n\n\treturn\n}\n\n// IsSetChildRunID returns true if ChildRunID is not nil.\nfunc (v *ApplyParentClosePolicyAttributes) IsSetChildRunID() bool {\n\treturn v != nil && v.ChildRunID != nil\n}\n\n// GetParentClosePolicy returns the value of ParentClosePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyAttributes) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\n\treturn\n}\n\n// IsSetParentClosePolicy returns true if ParentClosePolicy is not nil.\nfunc (v *ApplyParentClosePolicyAttributes) IsSetParentClosePolicy() bool {\n\treturn v != nil && v.ParentClosePolicy != nil\n}\n\ntype ApplyParentClosePolicyRequest struct {\n\tChild  *ApplyParentClosePolicyAttributes `json:\"child,omitempty\"`\n\tStatus *ApplyParentClosePolicyStatus     `json:\"status,omitempty\"`\n}\n\n// ToWire translates a ApplyParentClosePolicyRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ApplyParentClosePolicyRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Child != nil {\n\t\tw, err = v.Child.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tw, err = v.Status.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ApplyParentClosePolicyAttributes_Read(w wire.Value) (*ApplyParentClosePolicyAttributes, error) {\n\tvar v ApplyParentClosePolicyAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ApplyParentClosePolicyStatus_Read(w wire.Value) (*ApplyParentClosePolicyStatus, error) {\n\tvar v ApplyParentClosePolicyStatus\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ApplyParentClosePolicyRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ApplyParentClosePolicyRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ApplyParentClosePolicyRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ApplyParentClosePolicyRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Child, err = _ApplyParentClosePolicyAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Status, err = _ApplyParentClosePolicyStatus_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ApplyParentClosePolicyRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ApplyParentClosePolicyRequest struct could not be encoded.\nfunc (v *ApplyParentClosePolicyRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Child != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Child.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Status != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Status.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ApplyParentClosePolicyAttributes_Decode(sr stream.Reader) (*ApplyParentClosePolicyAttributes, error) {\n\tvar v ApplyParentClosePolicyAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ApplyParentClosePolicyStatus_Decode(sr stream.Reader) (*ApplyParentClosePolicyStatus, error) {\n\tvar v ApplyParentClosePolicyStatus\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ApplyParentClosePolicyRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ApplyParentClosePolicyRequest struct could not be generated from the wire\n// representation.\nfunc (v *ApplyParentClosePolicyRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Child, err = _ApplyParentClosePolicyAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Status, err = _ApplyParentClosePolicyStatus_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ApplyParentClosePolicyRequest\n// struct.\nfunc (v *ApplyParentClosePolicyRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Child != nil {\n\t\tfields[i] = fmt.Sprintf(\"Child: %v\", v.Child)\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tfields[i] = fmt.Sprintf(\"Status: %v\", v.Status)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ApplyParentClosePolicyRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ApplyParentClosePolicyRequest match the\n// provided ApplyParentClosePolicyRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ApplyParentClosePolicyRequest) Equals(rhs *ApplyParentClosePolicyRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Child == nil && rhs.Child == nil) || (v.Child != nil && rhs.Child != nil && v.Child.Equals(rhs.Child))) {\n\t\treturn false\n\t}\n\tif !((v.Status == nil && rhs.Status == nil) || (v.Status != nil && rhs.Status != nil && v.Status.Equals(rhs.Status))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ApplyParentClosePolicyRequest.\nfunc (v *ApplyParentClosePolicyRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Child != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"child\", v.Child))\n\t}\n\tif v.Status != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"status\", v.Status))\n\t}\n\treturn err\n}\n\n// GetChild returns the value of Child if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyRequest) GetChild() (o *ApplyParentClosePolicyAttributes) {\n\tif v != nil && v.Child != nil {\n\t\treturn v.Child\n\t}\n\n\treturn\n}\n\n// IsSetChild returns true if Child is not nil.\nfunc (v *ApplyParentClosePolicyRequest) IsSetChild() bool {\n\treturn v != nil && v.Child != nil\n}\n\n// GetStatus returns the value of Status if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyRequest) GetStatus() (o *ApplyParentClosePolicyStatus) {\n\tif v != nil && v.Status != nil {\n\t\treturn v.Status\n\t}\n\n\treturn\n}\n\n// IsSetStatus returns true if Status is not nil.\nfunc (v *ApplyParentClosePolicyRequest) IsSetStatus() bool {\n\treturn v != nil && v.Status != nil\n}\n\ntype ApplyParentClosePolicyResult struct {\n\tChild       *ApplyParentClosePolicyAttributes `json:\"child,omitempty\"`\n\tFailedCause *CrossClusterTaskFailedCause      `json:\"failedCause,omitempty\"`\n}\n\n// ToWire translates a ApplyParentClosePolicyResult struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ApplyParentClosePolicyResult) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Child != nil {\n\t\tw, err = v.Child.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tw, err = v.FailedCause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterTaskFailedCause_Read(w wire.Value) (CrossClusterTaskFailedCause, error) {\n\tvar v CrossClusterTaskFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a ApplyParentClosePolicyResult struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ApplyParentClosePolicyResult struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ApplyParentClosePolicyResult\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ApplyParentClosePolicyResult) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Child, err = _ApplyParentClosePolicyAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\t\tx, err = _CrossClusterTaskFailedCause_Read(field.Value)\n\t\t\t\tv.FailedCause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ApplyParentClosePolicyResult struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ApplyParentClosePolicyResult struct could not be encoded.\nfunc (v *ApplyParentClosePolicyResult) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Child != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Child.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedCause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedCause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterTaskFailedCause_Decode(sr stream.Reader) (CrossClusterTaskFailedCause, error) {\n\tvar v CrossClusterTaskFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a ApplyParentClosePolicyResult struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ApplyParentClosePolicyResult struct could not be generated from the wire\n// representation.\nfunc (v *ApplyParentClosePolicyResult) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Child, err = _ApplyParentClosePolicyAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\tx, err = _CrossClusterTaskFailedCause_Decode(sr)\n\t\t\tv.FailedCause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ApplyParentClosePolicyResult\n// struct.\nfunc (v *ApplyParentClosePolicyResult) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Child != nil {\n\t\tfields[i] = fmt.Sprintf(\"Child: %v\", v.Child)\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedCause: %v\", *(v.FailedCause))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ApplyParentClosePolicyResult{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _CrossClusterTaskFailedCause_EqualsPtr(lhs, rhs *CrossClusterTaskFailedCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ApplyParentClosePolicyResult match the\n// provided ApplyParentClosePolicyResult.\n//\n// This function performs a deep comparison.\nfunc (v *ApplyParentClosePolicyResult) Equals(rhs *ApplyParentClosePolicyResult) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Child == nil && rhs.Child == nil) || (v.Child != nil && rhs.Child != nil && v.Child.Equals(rhs.Child))) {\n\t\treturn false\n\t}\n\tif !_CrossClusterTaskFailedCause_EqualsPtr(v.FailedCause, rhs.FailedCause) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ApplyParentClosePolicyResult.\nfunc (v *ApplyParentClosePolicyResult) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Child != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"child\", v.Child))\n\t}\n\tif v.FailedCause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedCause\", *v.FailedCause))\n\t}\n\treturn err\n}\n\n// GetChild returns the value of Child if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyResult) GetChild() (o *ApplyParentClosePolicyAttributes) {\n\tif v != nil && v.Child != nil {\n\t\treturn v.Child\n\t}\n\n\treturn\n}\n\n// IsSetChild returns true if Child is not nil.\nfunc (v *ApplyParentClosePolicyResult) IsSetChild() bool {\n\treturn v != nil && v.Child != nil\n}\n\n// GetFailedCause returns the value of FailedCause if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyResult) GetFailedCause() (o CrossClusterTaskFailedCause) {\n\tif v != nil && v.FailedCause != nil {\n\t\treturn *v.FailedCause\n\t}\n\n\treturn\n}\n\n// IsSetFailedCause returns true if FailedCause is not nil.\nfunc (v *ApplyParentClosePolicyResult) IsSetFailedCause() bool {\n\treturn v != nil && v.FailedCause != nil\n}\n\ntype ApplyParentClosePolicyStatus struct {\n\tCompleted   *bool                        `json:\"completed,omitempty\"`\n\tFailedCause *CrossClusterTaskFailedCause `json:\"failedCause,omitempty\"`\n}\n\n// ToWire translates a ApplyParentClosePolicyStatus struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ApplyParentClosePolicyStatus) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Completed != nil {\n\t\tw, err = wire.NewValueBool(*(v.Completed)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tw, err = v.FailedCause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ApplyParentClosePolicyStatus struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ApplyParentClosePolicyStatus struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ApplyParentClosePolicyStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ApplyParentClosePolicyStatus) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Completed = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\t\tx, err = _CrossClusterTaskFailedCause_Read(field.Value)\n\t\t\t\tv.FailedCause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ApplyParentClosePolicyStatus struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ApplyParentClosePolicyStatus struct could not be encoded.\nfunc (v *ApplyParentClosePolicyStatus) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Completed != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Completed)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedCause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedCause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ApplyParentClosePolicyStatus struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ApplyParentClosePolicyStatus struct could not be generated from the wire\n// representation.\nfunc (v *ApplyParentClosePolicyStatus) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Completed = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\tx, err = _CrossClusterTaskFailedCause_Decode(sr)\n\t\t\tv.FailedCause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ApplyParentClosePolicyStatus\n// struct.\nfunc (v *ApplyParentClosePolicyStatus) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Completed != nil {\n\t\tfields[i] = fmt.Sprintf(\"Completed: %v\", *(v.Completed))\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedCause: %v\", *(v.FailedCause))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ApplyParentClosePolicyStatus{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ApplyParentClosePolicyStatus match the\n// provided ApplyParentClosePolicyStatus.\n//\n// This function performs a deep comparison.\nfunc (v *ApplyParentClosePolicyStatus) Equals(rhs *ApplyParentClosePolicyStatus) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Completed, rhs.Completed) {\n\t\treturn false\n\t}\n\tif !_CrossClusterTaskFailedCause_EqualsPtr(v.FailedCause, rhs.FailedCause) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ApplyParentClosePolicyStatus.\nfunc (v *ApplyParentClosePolicyStatus) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Completed != nil {\n\t\tenc.AddBool(\"completed\", *v.Completed)\n\t}\n\tif v.FailedCause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedCause\", *v.FailedCause))\n\t}\n\treturn err\n}\n\n// GetCompleted returns the value of Completed if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyStatus) GetCompleted() (o bool) {\n\tif v != nil && v.Completed != nil {\n\t\treturn *v.Completed\n\t}\n\n\treturn\n}\n\n// IsSetCompleted returns true if Completed is not nil.\nfunc (v *ApplyParentClosePolicyStatus) IsSetCompleted() bool {\n\treturn v != nil && v.Completed != nil\n}\n\n// GetFailedCause returns the value of FailedCause if it is set or its\n// zero value if it is unset.\nfunc (v *ApplyParentClosePolicyStatus) GetFailedCause() (o CrossClusterTaskFailedCause) {\n\tif v != nil && v.FailedCause != nil {\n\t\treturn *v.FailedCause\n\t}\n\n\treturn\n}\n\n// IsSetFailedCause returns true if FailedCause is not nil.\nfunc (v *ApplyParentClosePolicyStatus) IsSetFailedCause() bool {\n\treturn v != nil && v.FailedCause != nil\n}\n\ntype ArchivalStatus int32\n\nconst (\n\tArchivalStatusDisabled ArchivalStatus = 0\n\tArchivalStatusEnabled  ArchivalStatus = 1\n)\n\n// ArchivalStatus_Values returns all recognized values of ArchivalStatus.\nfunc ArchivalStatus_Values() []ArchivalStatus {\n\treturn []ArchivalStatus{\n\t\tArchivalStatusDisabled,\n\t\tArchivalStatusEnabled,\n\t}\n}\n\n// UnmarshalText tries to decode ArchivalStatus from a byte slice\n// containing its name.\n//\n//\tvar v ArchivalStatus\n//\terr := v.UnmarshalText([]byte(\"DISABLED\"))\nfunc (v *ArchivalStatus) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"DISABLED\":\n\t\t*v = ArchivalStatusDisabled\n\t\treturn nil\n\tcase \"ENABLED\":\n\t\t*v = ArchivalStatusEnabled\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ArchivalStatus\", err)\n\t\t}\n\t\t*v = ArchivalStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ArchivalStatus to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ArchivalStatus) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"DISABLED\"), nil\n\tcase 1:\n\t\treturn []byte(\"ENABLED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ArchivalStatus.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ArchivalStatus) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"DISABLED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"ENABLED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ArchivalStatus) Ptr() *ArchivalStatus {\n\treturn &v\n}\n\n// Encode encodes ArchivalStatus directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ArchivalStatus\n//\treturn v.Encode(sWriter)\nfunc (v ArchivalStatus) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ArchivalStatus into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ArchivalStatus) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ArchivalStatus from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ArchivalStatus(0), err\n//\t}\n//\n//\tvar v ArchivalStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ArchivalStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ArchivalStatus) FromWire(w wire.Value) error {\n\t*v = (ArchivalStatus)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ArchivalStatus directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ArchivalStatus\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ArchivalStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ArchivalStatus) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ArchivalStatus)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ArchivalStatus.\nfunc (v ArchivalStatus) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"DISABLED\"\n\tcase 1:\n\t\treturn \"ENABLED\"\n\t}\n\treturn fmt.Sprintf(\"ArchivalStatus(%d)\", w)\n}\n\n// Equals returns true if this ArchivalStatus value matches the provided\n// value.\nfunc (v ArchivalStatus) Equals(rhs ArchivalStatus) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ArchivalStatus into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ArchivalStatus) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"DISABLED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"ENABLED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ArchivalStatus from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ArchivalStatus) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ArchivalStatus\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ArchivalStatus\")\n\t\t}\n\t\t*v = (ArchivalStatus)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ArchivalStatus\")\n\t}\n}\n\ntype AsyncWorkflowConfiguration struct {\n\tEnabled             *bool     `json:\"enabled,omitempty\"`\n\tPredefinedQueueName *string   `json:\"predefinedQueueName,omitempty\"`\n\tQueueType           *string   `json:\"queueType,omitempty\"`\n\tQueueConfig         *DataBlob `json:\"queueConfig,omitempty\"`\n}\n\n// ToWire translates a AsyncWorkflowConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AsyncWorkflowConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Enabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.Enabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PredefinedQueueName != nil {\n\t\tw, err = wire.NewValueString(*(v.PredefinedQueueName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.QueueType != nil {\n\t\tw, err = wire.NewValueString(*(v.QueueType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.QueueConfig != nil {\n\t\tw, err = v.QueueConfig.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DataBlob_Read(w wire.Value) (*DataBlob, error) {\n\tvar v DataBlob\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AsyncWorkflowConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AsyncWorkflowConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AsyncWorkflowConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AsyncWorkflowConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Enabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PredefinedQueueName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.QueueType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueueConfig, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AsyncWorkflowConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AsyncWorkflowConfiguration struct could not be encoded.\nfunc (v *AsyncWorkflowConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Enabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Enabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PredefinedQueueName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PredefinedQueueName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueueType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.QueueType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueueConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueueConfig.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DataBlob_Decode(sr stream.Reader) (*DataBlob, error) {\n\tvar v DataBlob\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AsyncWorkflowConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AsyncWorkflowConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *AsyncWorkflowConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Enabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PredefinedQueueName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.QueueType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.QueueConfig, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AsyncWorkflowConfiguration\n// struct.\nfunc (v *AsyncWorkflowConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Enabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"Enabled: %v\", *(v.Enabled))\n\t\ti++\n\t}\n\tif v.PredefinedQueueName != nil {\n\t\tfields[i] = fmt.Sprintf(\"PredefinedQueueName: %v\", *(v.PredefinedQueueName))\n\t\ti++\n\t}\n\tif v.QueueType != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueueType: %v\", *(v.QueueType))\n\t\ti++\n\t}\n\tif v.QueueConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueueConfig: %v\", v.QueueConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AsyncWorkflowConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AsyncWorkflowConfiguration match the\n// provided AsyncWorkflowConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *AsyncWorkflowConfiguration) Equals(rhs *AsyncWorkflowConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Enabled, rhs.Enabled) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PredefinedQueueName, rhs.PredefinedQueueName) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.QueueType, rhs.QueueType) {\n\t\treturn false\n\t}\n\tif !((v.QueueConfig == nil && rhs.QueueConfig == nil) || (v.QueueConfig != nil && rhs.QueueConfig != nil && v.QueueConfig.Equals(rhs.QueueConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AsyncWorkflowConfiguration.\nfunc (v *AsyncWorkflowConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Enabled != nil {\n\t\tenc.AddBool(\"enabled\", *v.Enabled)\n\t}\n\tif v.PredefinedQueueName != nil {\n\t\tenc.AddString(\"predefinedQueueName\", *v.PredefinedQueueName)\n\t}\n\tif v.QueueType != nil {\n\t\tenc.AddString(\"queueType\", *v.QueueType)\n\t}\n\tif v.QueueConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queueConfig\", v.QueueConfig))\n\t}\n\treturn err\n}\n\n// GetEnabled returns the value of Enabled if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncWorkflowConfiguration) GetEnabled() (o bool) {\n\tif v != nil && v.Enabled != nil {\n\t\treturn *v.Enabled\n\t}\n\n\treturn\n}\n\n// IsSetEnabled returns true if Enabled is not nil.\nfunc (v *AsyncWorkflowConfiguration) IsSetEnabled() bool {\n\treturn v != nil && v.Enabled != nil\n}\n\n// GetPredefinedQueueName returns the value of PredefinedQueueName if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncWorkflowConfiguration) GetPredefinedQueueName() (o string) {\n\tif v != nil && v.PredefinedQueueName != nil {\n\t\treturn *v.PredefinedQueueName\n\t}\n\n\treturn\n}\n\n// IsSetPredefinedQueueName returns true if PredefinedQueueName is not nil.\nfunc (v *AsyncWorkflowConfiguration) IsSetPredefinedQueueName() bool {\n\treturn v != nil && v.PredefinedQueueName != nil\n}\n\n// GetQueueType returns the value of QueueType if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncWorkflowConfiguration) GetQueueType() (o string) {\n\tif v != nil && v.QueueType != nil {\n\t\treturn *v.QueueType\n\t}\n\n\treturn\n}\n\n// IsSetQueueType returns true if QueueType is not nil.\nfunc (v *AsyncWorkflowConfiguration) IsSetQueueType() bool {\n\treturn v != nil && v.QueueType != nil\n}\n\n// GetQueueConfig returns the value of QueueConfig if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncWorkflowConfiguration) GetQueueConfig() (o *DataBlob) {\n\tif v != nil && v.QueueConfig != nil {\n\t\treturn v.QueueConfig\n\t}\n\n\treturn\n}\n\n// IsSetQueueConfig returns true if QueueConfig is not nil.\nfunc (v *AsyncWorkflowConfiguration) IsSetQueueConfig() bool {\n\treturn v != nil && v.QueueConfig != nil\n}\n\ntype AutoConfigHint struct {\n\tEnableAutoConfig   *bool  `json:\"enableAutoConfig,omitempty\"`\n\tPollerWaitTimeInMs *int64 `json:\"pollerWaitTimeInMs,omitempty\"`\n}\n\n// ToWire translates a AutoConfigHint struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AutoConfigHint) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.EnableAutoConfig != nil {\n\t\tw, err = wire.NewValueBool(*(v.EnableAutoConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PollerWaitTimeInMs != nil {\n\t\tw, err = wire.NewValueI64(*(v.PollerWaitTimeInMs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a AutoConfigHint struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AutoConfigHint struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AutoConfigHint\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AutoConfigHint) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.EnableAutoConfig = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PollerWaitTimeInMs = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AutoConfigHint struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AutoConfigHint struct could not be encoded.\nfunc (v *AutoConfigHint) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EnableAutoConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.EnableAutoConfig)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollerWaitTimeInMs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PollerWaitTimeInMs)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a AutoConfigHint struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AutoConfigHint struct could not be generated from the wire\n// representation.\nfunc (v *AutoConfigHint) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.EnableAutoConfig = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PollerWaitTimeInMs = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AutoConfigHint\n// struct.\nfunc (v *AutoConfigHint) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.EnableAutoConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"EnableAutoConfig: %v\", *(v.EnableAutoConfig))\n\t\ti++\n\t}\n\tif v.PollerWaitTimeInMs != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollerWaitTimeInMs: %v\", *(v.PollerWaitTimeInMs))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AutoConfigHint{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this AutoConfigHint match the\n// provided AutoConfigHint.\n//\n// This function performs a deep comparison.\nfunc (v *AutoConfigHint) Equals(rhs *AutoConfigHint) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.EnableAutoConfig, rhs.EnableAutoConfig) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PollerWaitTimeInMs, rhs.PollerWaitTimeInMs) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AutoConfigHint.\nfunc (v *AutoConfigHint) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.EnableAutoConfig != nil {\n\t\tenc.AddBool(\"enableAutoConfig\", *v.EnableAutoConfig)\n\t}\n\tif v.PollerWaitTimeInMs != nil {\n\t\tenc.AddInt64(\"pollerWaitTimeInMs\", *v.PollerWaitTimeInMs)\n\t}\n\treturn err\n}\n\n// GetEnableAutoConfig returns the value of EnableAutoConfig if it is set or its\n// zero value if it is unset.\nfunc (v *AutoConfigHint) GetEnableAutoConfig() (o bool) {\n\tif v != nil && v.EnableAutoConfig != nil {\n\t\treturn *v.EnableAutoConfig\n\t}\n\n\treturn\n}\n\n// IsSetEnableAutoConfig returns true if EnableAutoConfig is not nil.\nfunc (v *AutoConfigHint) IsSetEnableAutoConfig() bool {\n\treturn v != nil && v.EnableAutoConfig != nil\n}\n\n// GetPollerWaitTimeInMs returns the value of PollerWaitTimeInMs if it is set or its\n// zero value if it is unset.\nfunc (v *AutoConfigHint) GetPollerWaitTimeInMs() (o int64) {\n\tif v != nil && v.PollerWaitTimeInMs != nil {\n\t\treturn *v.PollerWaitTimeInMs\n\t}\n\n\treturn\n}\n\n// IsSetPollerWaitTimeInMs returns true if PollerWaitTimeInMs is not nil.\nfunc (v *AutoConfigHint) IsSetPollerWaitTimeInMs() bool {\n\treturn v != nil && v.PollerWaitTimeInMs != nil\n}\n\ntype BadBinaries struct {\n\tBinaries map[string]*BadBinaryInfo `json:\"binaries,omitempty\"`\n}\n\ntype _Map_String_BadBinaryInfo_MapItemList map[string]*BadBinaryInfo\n\nfunc (m _Map_String_BadBinaryInfo_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*BadBinaryInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_BadBinaryInfo_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_BadBinaryInfo_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_BadBinaryInfo_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_BadBinaryInfo_MapItemList) Close() {}\n\n// ToWire translates a BadBinaries struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *BadBinaries) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Binaries != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_BadBinaryInfo_MapItemList(v.Binaries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _BadBinaryInfo_Read(w wire.Value) (*BadBinaryInfo, error) {\n\tvar v BadBinaryInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_BadBinaryInfo_Read(m wire.MapItemList) (map[string]*BadBinaryInfo, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*BadBinaryInfo, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _BadBinaryInfo_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a BadBinaries struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a BadBinaries struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v BadBinaries\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *BadBinaries) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Binaries, err = _Map_String_BadBinaryInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_BadBinaryInfo_Encode(val map[string]*BadBinaryInfo, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*BadBinaryInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a BadBinaries struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a BadBinaries struct could not be encoded.\nfunc (v *BadBinaries) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Binaries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_BadBinaryInfo_Encode(v.Binaries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _BadBinaryInfo_Decode(sr stream.Reader) (*BadBinaryInfo, error) {\n\tvar v BadBinaryInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_BadBinaryInfo_Decode(sr stream.Reader) (map[string]*BadBinaryInfo, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*BadBinaryInfo, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _BadBinaryInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a BadBinaries struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a BadBinaries struct could not be generated from the wire\n// representation.\nfunc (v *BadBinaries) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Binaries, err = _Map_String_BadBinaryInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a BadBinaries\n// struct.\nfunc (v *BadBinaries) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Binaries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Binaries: %v\", v.Binaries)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"BadBinaries{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_BadBinaryInfo_Equals(lhs, rhs map[string]*BadBinaryInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this BadBinaries match the\n// provided BadBinaries.\n//\n// This function performs a deep comparison.\nfunc (v *BadBinaries) Equals(rhs *BadBinaries) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Binaries == nil && rhs.Binaries == nil) || (v.Binaries != nil && rhs.Binaries != nil && _Map_String_BadBinaryInfo_Equals(v.Binaries, rhs.Binaries))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_BadBinaryInfo_Zapper map[string]*BadBinaryInfo\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_BadBinaryInfo_Zapper.\nfunc (m _Map_String_BadBinaryInfo_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of BadBinaries.\nfunc (v *BadBinaries) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Binaries != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"binaries\", (_Map_String_BadBinaryInfo_Zapper)(v.Binaries)))\n\t}\n\treturn err\n}\n\n// GetBinaries returns the value of Binaries if it is set or its\n// zero value if it is unset.\nfunc (v *BadBinaries) GetBinaries() (o map[string]*BadBinaryInfo) {\n\tif v != nil && v.Binaries != nil {\n\t\treturn v.Binaries\n\t}\n\n\treturn\n}\n\n// IsSetBinaries returns true if Binaries is not nil.\nfunc (v *BadBinaries) IsSetBinaries() bool {\n\treturn v != nil && v.Binaries != nil\n}\n\ntype BadBinaryInfo struct {\n\tReason          *string `json:\"reason,omitempty\"`\n\tOperator        *string `json:\"operator,omitempty\"`\n\tCreatedTimeNano *int64  `json:\"createdTimeNano,omitempty\"`\n}\n\n// ToWire translates a BadBinaryInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *BadBinaryInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Operator != nil {\n\t\tw, err = wire.NewValueString(*(v.Operator)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreatedTimeNano)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a BadBinaryInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a BadBinaryInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v BadBinaryInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *BadBinaryInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Operator = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreatedTimeNano = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a BadBinaryInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a BadBinaryInfo struct could not be encoded.\nfunc (v *BadBinaryInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Operator != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Operator)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreatedTimeNano != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreatedTimeNano)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a BadBinaryInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a BadBinaryInfo struct could not be generated from the wire\n// representation.\nfunc (v *BadBinaryInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Operator = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreatedTimeNano = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a BadBinaryInfo\n// struct.\nfunc (v *BadBinaryInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Operator != nil {\n\t\tfields[i] = fmt.Sprintf(\"Operator: %v\", *(v.Operator))\n\t\ti++\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreatedTimeNano: %v\", *(v.CreatedTimeNano))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"BadBinaryInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this BadBinaryInfo match the\n// provided BadBinaryInfo.\n//\n// This function performs a deep comparison.\nfunc (v *BadBinaryInfo) Equals(rhs *BadBinaryInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Operator, rhs.Operator) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreatedTimeNano, rhs.CreatedTimeNano) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of BadBinaryInfo.\nfunc (v *BadBinaryInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Operator != nil {\n\t\tenc.AddString(\"operator\", *v.Operator)\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tenc.AddInt64(\"createdTimeNano\", *v.CreatedTimeNano)\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *BadBinaryInfo) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *BadBinaryInfo) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetOperator returns the value of Operator if it is set or its\n// zero value if it is unset.\nfunc (v *BadBinaryInfo) GetOperator() (o string) {\n\tif v != nil && v.Operator != nil {\n\t\treturn *v.Operator\n\t}\n\n\treturn\n}\n\n// IsSetOperator returns true if Operator is not nil.\nfunc (v *BadBinaryInfo) IsSetOperator() bool {\n\treturn v != nil && v.Operator != nil\n}\n\n// GetCreatedTimeNano returns the value of CreatedTimeNano if it is set or its\n// zero value if it is unset.\nfunc (v *BadBinaryInfo) GetCreatedTimeNano() (o int64) {\n\tif v != nil && v.CreatedTimeNano != nil {\n\t\treturn *v.CreatedTimeNano\n\t}\n\n\treturn\n}\n\n// IsSetCreatedTimeNano returns true if CreatedTimeNano is not nil.\nfunc (v *BadBinaryInfo) IsSetCreatedTimeNano() bool {\n\treturn v != nil && v.CreatedTimeNano != nil\n}\n\ntype BadRequestError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a BadRequestError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *BadRequestError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a BadRequestError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a BadRequestError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v BadRequestError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *BadRequestError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of BadRequestError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a BadRequestError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a BadRequestError struct could not be encoded.\nfunc (v *BadRequestError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a BadRequestError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a BadRequestError struct could not be generated from the wire\n// representation.\nfunc (v *BadRequestError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of BadRequestError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a BadRequestError\n// struct.\nfunc (v *BadRequestError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"BadRequestError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*BadRequestError) ErrorName() string {\n\treturn \"BadRequestError\"\n}\n\n// Equals returns true if all the fields of this BadRequestError match the\n// provided BadRequestError.\n//\n// This function performs a deep comparison.\nfunc (v *BadRequestError) Equals(rhs *BadRequestError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of BadRequestError.\nfunc (v *BadRequestError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *BadRequestError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *BadRequestError) Error() string {\n\treturn v.String()\n}\n\ntype CancelExternalWorkflowExecutionFailedCause int32\n\nconst (\n\tCancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution CancelExternalWorkflowExecutionFailedCause = 0\n\tCancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted         CancelExternalWorkflowExecutionFailedCause = 1\n)\n\n// CancelExternalWorkflowExecutionFailedCause_Values returns all recognized values of CancelExternalWorkflowExecutionFailedCause.\nfunc CancelExternalWorkflowExecutionFailedCause_Values() []CancelExternalWorkflowExecutionFailedCause {\n\treturn []CancelExternalWorkflowExecutionFailedCause{\n\t\tCancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution,\n\t\tCancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted,\n\t}\n}\n\n// UnmarshalText tries to decode CancelExternalWorkflowExecutionFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v CancelExternalWorkflowExecutionFailedCause\n//\terr := v.UnmarshalText([]byte(\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"))\nfunc (v *CancelExternalWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\":\n\t\t*v = CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*v = CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CancelExternalWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*v = CancelExternalWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CancelExternalWorkflowExecutionFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v CancelExternalWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"), nil\n\tcase 1:\n\t\treturn []byte(\"WORKFLOW_ALREADY_COMPLETED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancelExternalWorkflowExecutionFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v CancelExternalWorkflowExecutionFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"WORKFLOW_ALREADY_COMPLETED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v CancelExternalWorkflowExecutionFailedCause) Ptr() *CancelExternalWorkflowExecutionFailedCause {\n\treturn &v\n}\n\n// Encode encodes CancelExternalWorkflowExecutionFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v CancelExternalWorkflowExecutionFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v CancelExternalWorkflowExecutionFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates CancelExternalWorkflowExecutionFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v CancelExternalWorkflowExecutionFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes CancelExternalWorkflowExecutionFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return CancelExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\n//\tvar v CancelExternalWorkflowExecutionFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return CancelExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CancelExternalWorkflowExecutionFailedCause) FromWire(w wire.Value) error {\n\t*v = (CancelExternalWorkflowExecutionFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded CancelExternalWorkflowExecutionFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v CancelExternalWorkflowExecutionFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return CancelExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CancelExternalWorkflowExecutionFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (CancelExternalWorkflowExecutionFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of CancelExternalWorkflowExecutionFailedCause.\nfunc (v CancelExternalWorkflowExecutionFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"\n\tcase 1:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\t}\n\treturn fmt.Sprintf(\"CancelExternalWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// Equals returns true if this CancelExternalWorkflowExecutionFailedCause value matches the provided\n// value.\nfunc (v CancelExternalWorkflowExecutionFailedCause) Equals(rhs CancelExternalWorkflowExecutionFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes CancelExternalWorkflowExecutionFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v CancelExternalWorkflowExecutionFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_ALREADY_COMPLETED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode CancelExternalWorkflowExecutionFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *CancelExternalWorkflowExecutionFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"CancelExternalWorkflowExecutionFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"CancelExternalWorkflowExecutionFailedCause\")\n\t\t}\n\t\t*v = (CancelExternalWorkflowExecutionFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"CancelExternalWorkflowExecutionFailedCause\")\n\t}\n}\n\ntype CancelTimerDecisionAttributes struct {\n\tTimerId *string `json:\"timerId,omitempty\"`\n}\n\n// ToWire translates a CancelTimerDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CancelTimerDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CancelTimerDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CancelTimerDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CancelTimerDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CancelTimerDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CancelTimerDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CancelTimerDecisionAttributes struct could not be encoded.\nfunc (v *CancelTimerDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CancelTimerDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CancelTimerDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CancelTimerDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CancelTimerDecisionAttributes\n// struct.\nfunc (v *CancelTimerDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CancelTimerDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CancelTimerDecisionAttributes match the\n// provided CancelTimerDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CancelTimerDecisionAttributes) Equals(rhs *CancelTimerDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancelTimerDecisionAttributes.\nfunc (v *CancelTimerDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *CancelTimerDecisionAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *CancelTimerDecisionAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\ntype CancelTimerFailedEventAttributes struct {\n\tTimerId                      *string `json:\"timerId,omitempty\"`\n\tCause                        *string `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tIdentity                     *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a CancelTimerFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CancelTimerFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = wire.NewValueString(*(v.Cause)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CancelTimerFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CancelTimerFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CancelTimerFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CancelTimerFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CancelTimerFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CancelTimerFailedEventAttributes struct could not be encoded.\nfunc (v *CancelTimerFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Cause)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CancelTimerFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CancelTimerFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CancelTimerFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CancelTimerFailedEventAttributes\n// struct.\nfunc (v *CancelTimerFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CancelTimerFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CancelTimerFailedEventAttributes match the\n// provided CancelTimerFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CancelTimerFailedEventAttributes) Equals(rhs *CancelTimerFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancelTimerFailedEventAttributes.\nfunc (v *CancelTimerFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\tif v.Cause != nil {\n\t\tenc.AddString(\"cause\", *v.Cause)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *CancelTimerFailedEventAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *CancelTimerFailedEventAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *CancelTimerFailedEventAttributes) GetCause() (o string) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *CancelTimerFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *CancelTimerFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *CancelTimerFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *CancelTimerFailedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *CancelTimerFailedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype CancelWorkflowExecutionDecisionAttributes struct {\n\tDetails []byte `json:\"details,omitempty\"`\n}\n\n// ToWire translates a CancelWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CancelWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CancelWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CancelWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CancelWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CancelWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CancelWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CancelWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CancelWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CancelWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CancelWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CancelWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CancelWorkflowExecutionDecisionAttributes match the\n// provided CancelWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) Equals(rhs *CancelWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancelWorkflowExecutionDecisionAttributes.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\treturn err\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *CancelWorkflowExecutionDecisionAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\ntype CancellationAlreadyRequestedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a CancellationAlreadyRequestedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CancellationAlreadyRequestedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CancellationAlreadyRequestedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CancellationAlreadyRequestedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CancellationAlreadyRequestedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CancellationAlreadyRequestedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of CancellationAlreadyRequestedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CancellationAlreadyRequestedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CancellationAlreadyRequestedError struct could not be encoded.\nfunc (v *CancellationAlreadyRequestedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CancellationAlreadyRequestedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CancellationAlreadyRequestedError struct could not be generated from the wire\n// representation.\nfunc (v *CancellationAlreadyRequestedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of CancellationAlreadyRequestedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CancellationAlreadyRequestedError\n// struct.\nfunc (v *CancellationAlreadyRequestedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"CancellationAlreadyRequestedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*CancellationAlreadyRequestedError) ErrorName() string {\n\treturn \"CancellationAlreadyRequestedError\"\n}\n\n// Equals returns true if all the fields of this CancellationAlreadyRequestedError match the\n// provided CancellationAlreadyRequestedError.\n//\n// This function performs a deep comparison.\nfunc (v *CancellationAlreadyRequestedError) Equals(rhs *CancellationAlreadyRequestedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CancellationAlreadyRequestedError.\nfunc (v *CancellationAlreadyRequestedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *CancellationAlreadyRequestedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *CancellationAlreadyRequestedError) Error() string {\n\treturn v.String()\n}\n\ntype ChildWorkflowExecutionCanceledEventAttributes struct {\n\tDetails           []byte             `json:\"details,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tStartedEventId    *int64             `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionCanceledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecution_Read(w wire.Value) (*WorkflowExecution, error) {\n\tvar v WorkflowExecution\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowType_Read(w wire.Value) (*WorkflowType, error) {\n\tvar v WorkflowType\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ChildWorkflowExecutionCanceledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionCanceledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionCanceledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionCanceledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionCanceledEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecution_Decode(sr stream.Reader) (*WorkflowExecution, error) {\n\tvar v WorkflowExecution\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowType_Decode(sr stream.Reader) (*WorkflowType, error) {\n\tvar v WorkflowType\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ChildWorkflowExecutionCanceledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionCanceledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionCanceledEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionCanceledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionCanceledEventAttributes match the\n// provided ChildWorkflowExecutionCanceledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) Equals(rhs *ChildWorkflowExecutionCanceledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionCanceledEventAttributes.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype ChildWorkflowExecutionCompletedEventAttributes struct {\n\tResult            []byte             `json:\"result,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tStartedEventId    *int64             `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionCompletedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildWorkflowExecutionCompletedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionCompletedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionCompletedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionCompletedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionCompletedEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildWorkflowExecutionCompletedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionCompletedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionCompletedEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionCompletedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionCompletedEventAttributes match the\n// provided ChildWorkflowExecutionCompletedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) Equals(rhs *ChildWorkflowExecutionCompletedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionCompletedEventAttributes.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype ChildWorkflowExecutionFailedCause int32\n\nconst (\n\tChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning ChildWorkflowExecutionFailedCause = 0\n)\n\n// ChildWorkflowExecutionFailedCause_Values returns all recognized values of ChildWorkflowExecutionFailedCause.\nfunc ChildWorkflowExecutionFailedCause_Values() []ChildWorkflowExecutionFailedCause {\n\treturn []ChildWorkflowExecutionFailedCause{\n\t\tChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning,\n\t}\n}\n\n// UnmarshalText tries to decode ChildWorkflowExecutionFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v ChildWorkflowExecutionFailedCause\n//\terr := v.UnmarshalText([]byte(\"WORKFLOW_ALREADY_RUNNING\"))\nfunc (v *ChildWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"WORKFLOW_ALREADY_RUNNING\":\n\t\t*v = ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ChildWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*v = ChildWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ChildWorkflowExecutionFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ChildWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"WORKFLOW_ALREADY_RUNNING\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ChildWorkflowExecutionFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"WORKFLOW_ALREADY_RUNNING\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ChildWorkflowExecutionFailedCause) Ptr() *ChildWorkflowExecutionFailedCause {\n\treturn &v\n}\n\n// Encode encodes ChildWorkflowExecutionFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ChildWorkflowExecutionFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v ChildWorkflowExecutionFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ChildWorkflowExecutionFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ChildWorkflowExecutionFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ChildWorkflowExecutionFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ChildWorkflowExecutionFailedCause(0), err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ChildWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ChildWorkflowExecutionFailedCause) FromWire(w wire.Value) error {\n\t*v = (ChildWorkflowExecutionFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ChildWorkflowExecutionFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ChildWorkflowExecutionFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ChildWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ChildWorkflowExecutionFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ChildWorkflowExecutionFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ChildWorkflowExecutionFailedCause.\nfunc (v ChildWorkflowExecutionFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"WORKFLOW_ALREADY_RUNNING\"\n\t}\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// Equals returns true if this ChildWorkflowExecutionFailedCause value matches the provided\n// value.\nfunc (v ChildWorkflowExecutionFailedCause) Equals(rhs ChildWorkflowExecutionFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ChildWorkflowExecutionFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ChildWorkflowExecutionFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_ALREADY_RUNNING\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ChildWorkflowExecutionFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ChildWorkflowExecutionFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ChildWorkflowExecutionFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ChildWorkflowExecutionFailedCause\")\n\t\t}\n\t\t*v = (ChildWorkflowExecutionFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ChildWorkflowExecutionFailedCause\")\n\t}\n}\n\ntype ChildWorkflowExecutionFailedEventAttributes struct {\n\tReason            *string            `json:\"reason,omitempty\"`\n\tDetails           []byte             `json:\"details,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tStartedEventId    *int64             `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildWorkflowExecutionFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionFailedEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildWorkflowExecutionFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionFailedEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionFailedEventAttributes match the\n// provided ChildWorkflowExecutionFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) Equals(rhs *ChildWorkflowExecutionFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionFailedEventAttributes.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype ChildWorkflowExecutionStartedEventAttributes struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tHeader            *Header            `json:\"header,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionStartedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildWorkflowExecutionStartedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionStartedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionStartedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionStartedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionStartedEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildWorkflowExecutionStartedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionStartedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionStartedEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionStartedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionStartedEventAttributes match the\n// provided ChildWorkflowExecutionStartedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) Equals(rhs *ChildWorkflowExecutionStartedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionStartedEventAttributes.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\ntype ChildWorkflowExecutionTerminatedEventAttributes struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tStartedEventId    *int64             `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionTerminatedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildWorkflowExecutionTerminatedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionTerminatedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionTerminatedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionTerminatedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionTerminatedEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildWorkflowExecutionTerminatedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionTerminatedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionTerminatedEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionTerminatedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionTerminatedEventAttributes match the\n// provided ChildWorkflowExecutionTerminatedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) Equals(rhs *ChildWorkflowExecutionTerminatedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionTerminatedEventAttributes.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype ChildWorkflowExecutionTimedOutEventAttributes struct {\n\tTimeoutType       *TimeoutType       `json:\"timeoutType,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tStartedEventId    *int64             `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a ChildWorkflowExecutionTimedOutEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimeoutType != nil {\n\t\tw, err = v.TimeoutType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildWorkflowExecutionTimedOutEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildWorkflowExecutionTimedOutEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildWorkflowExecutionTimedOutEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TimeoutType\n\t\t\t\tx, err = _TimeoutType_Read(field.Value)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildWorkflowExecutionTimedOutEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildWorkflowExecutionTimedOutEventAttributes struct could not be encoded.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimeoutType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildWorkflowExecutionTimedOutEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildWorkflowExecutionTimedOutEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x TimeoutType\n\t\t\tx, err = _TimeoutType_Decode(sr)\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildWorkflowExecutionTimedOutEventAttributes\n// struct.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionTimedOutEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildWorkflowExecutionTimedOutEventAttributes match the\n// provided ChildWorkflowExecutionTimedOutEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) Equals(rhs *ChildWorkflowExecutionTimedOutEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_TimeoutType_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildWorkflowExecutionTimedOutEventAttributes.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimeoutType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timeoutType\", *v.TimeoutType))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype ClientVersionNotSupportedError struct {\n\tFeatureVersion    string `json:\"featureVersion,required\"`\n\tClientImpl        string `json:\"clientImpl,required\"`\n\tSupportedVersions string `json:\"supportedVersions,required\"`\n}\n\n// ToWire translates a ClientVersionNotSupportedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClientVersionNotSupportedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.FeatureVersion), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.ClientImpl), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 2, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.SupportedVersions), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 3, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ClientVersionNotSupportedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClientVersionNotSupportedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClientVersionNotSupportedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClientVersionNotSupportedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfeatureVersionIsSet := false\n\tclientImplIsSet := false\n\tsupportedVersionsIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.FeatureVersion, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tfeatureVersionIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ClientImpl, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tclientImplIsSet = true\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.SupportedVersions, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tsupportedVersionsIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !featureVersionIsSet {\n\t\treturn errors.New(\"field FeatureVersion of ClientVersionNotSupportedError is required\")\n\t}\n\n\tif !clientImplIsSet {\n\t\treturn errors.New(\"field ClientImpl of ClientVersionNotSupportedError is required\")\n\t}\n\n\tif !supportedVersionsIsSet {\n\t\treturn errors.New(\"field SupportedVersions of ClientVersionNotSupportedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClientVersionNotSupportedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClientVersionNotSupportedError struct could not be encoded.\nfunc (v *ClientVersionNotSupportedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.FeatureVersion); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.ClientImpl); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.SupportedVersions); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ClientVersionNotSupportedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClientVersionNotSupportedError struct could not be generated from the wire\n// representation.\nfunc (v *ClientVersionNotSupportedError) Decode(sr stream.Reader) error {\n\n\tfeatureVersionIsSet := false\n\tclientImplIsSet := false\n\tsupportedVersionsIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.FeatureVersion, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tfeatureVersionIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tv.ClientImpl, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tclientImplIsSet = true\n\t\tcase fh.ID == 3 && fh.Type == wire.TBinary:\n\t\t\tv.SupportedVersions, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tsupportedVersionsIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !featureVersionIsSet {\n\t\treturn errors.New(\"field FeatureVersion of ClientVersionNotSupportedError is required\")\n\t}\n\n\tif !clientImplIsSet {\n\t\treturn errors.New(\"field ClientImpl of ClientVersionNotSupportedError is required\")\n\t}\n\n\tif !supportedVersionsIsSet {\n\t\treturn errors.New(\"field SupportedVersions of ClientVersionNotSupportedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClientVersionNotSupportedError\n// struct.\nfunc (v *ClientVersionNotSupportedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"FeatureVersion: %v\", v.FeatureVersion)\n\ti++\n\tfields[i] = fmt.Sprintf(\"ClientImpl: %v\", v.ClientImpl)\n\ti++\n\tfields[i] = fmt.Sprintf(\"SupportedVersions: %v\", v.SupportedVersions)\n\ti++\n\n\treturn fmt.Sprintf(\"ClientVersionNotSupportedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*ClientVersionNotSupportedError) ErrorName() string {\n\treturn \"ClientVersionNotSupportedError\"\n}\n\n// Equals returns true if all the fields of this ClientVersionNotSupportedError match the\n// provided ClientVersionNotSupportedError.\n//\n// This function performs a deep comparison.\nfunc (v *ClientVersionNotSupportedError) Equals(rhs *ClientVersionNotSupportedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.FeatureVersion == rhs.FeatureVersion) {\n\t\treturn false\n\t}\n\tif !(v.ClientImpl == rhs.ClientImpl) {\n\t\treturn false\n\t}\n\tif !(v.SupportedVersions == rhs.SupportedVersions) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClientVersionNotSupportedError.\nfunc (v *ClientVersionNotSupportedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"featureVersion\", v.FeatureVersion)\n\tenc.AddString(\"clientImpl\", v.ClientImpl)\n\tenc.AddString(\"supportedVersions\", v.SupportedVersions)\n\treturn err\n}\n\n// GetFeatureVersion returns the value of FeatureVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ClientVersionNotSupportedError) GetFeatureVersion() (o string) {\n\tif v != nil {\n\t\to = v.FeatureVersion\n\t}\n\treturn\n}\n\n// GetClientImpl returns the value of ClientImpl if it is set or its\n// zero value if it is unset.\nfunc (v *ClientVersionNotSupportedError) GetClientImpl() (o string) {\n\tif v != nil {\n\t\to = v.ClientImpl\n\t}\n\treturn\n}\n\n// GetSupportedVersions returns the value of SupportedVersions if it is set or its\n// zero value if it is unset.\nfunc (v *ClientVersionNotSupportedError) GetSupportedVersions() (o string) {\n\tif v != nil {\n\t\to = v.SupportedVersions\n\t}\n\treturn\n}\n\nfunc (v *ClientVersionNotSupportedError) Error() string {\n\treturn v.String()\n}\n\ntype CloseShardRequest struct {\n\tShardID *int32 `json:\"shardID,omitempty\"`\n}\n\n// ToWire translates a CloseShardRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CloseShardRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CloseShardRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CloseShardRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CloseShardRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CloseShardRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CloseShardRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CloseShardRequest struct could not be encoded.\nfunc (v *CloseShardRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CloseShardRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CloseShardRequest struct could not be generated from the wire\n// representation.\nfunc (v *CloseShardRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CloseShardRequest\n// struct.\nfunc (v *CloseShardRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CloseShardRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CloseShardRequest match the\n// provided CloseShardRequest.\n//\n// This function performs a deep comparison.\nfunc (v *CloseShardRequest) Equals(rhs *CloseShardRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CloseShardRequest.\nfunc (v *CloseShardRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *CloseShardRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *CloseShardRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\ntype ClusterAttribute struct {\n\tScope *string `json:\"scope,omitempty\"`\n\tName  *string `json:\"name,omitempty\"`\n}\n\n// ToWire translates a ClusterAttribute struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClusterAttribute) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Scope != nil {\n\t\tw, err = wire.NewValueString(*(v.Scope)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ClusterAttribute struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClusterAttribute struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClusterAttribute\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClusterAttribute) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Scope = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClusterAttribute struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClusterAttribute struct could not be encoded.\nfunc (v *ClusterAttribute) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Scope != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Scope)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ClusterAttribute struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClusterAttribute struct could not be generated from the wire\n// representation.\nfunc (v *ClusterAttribute) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Scope = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClusterAttribute\n// struct.\nfunc (v *ClusterAttribute) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Scope != nil {\n\t\tfields[i] = fmt.Sprintf(\"Scope: %v\", *(v.Scope))\n\t\ti++\n\t}\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ClusterAttribute{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ClusterAttribute match the\n// provided ClusterAttribute.\n//\n// This function performs a deep comparison.\nfunc (v *ClusterAttribute) Equals(rhs *ClusterAttribute) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Scope, rhs.Scope) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClusterAttribute.\nfunc (v *ClusterAttribute) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Scope != nil {\n\t\tenc.AddString(\"scope\", *v.Scope)\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\treturn err\n}\n\n// GetScope returns the value of Scope if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterAttribute) GetScope() (o string) {\n\tif v != nil && v.Scope != nil {\n\t\treturn *v.Scope\n\t}\n\n\treturn\n}\n\n// IsSetScope returns true if Scope is not nil.\nfunc (v *ClusterAttribute) IsSetScope() bool {\n\treturn v != nil && v.Scope != nil\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterAttribute) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *ClusterAttribute) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\ntype ClusterAttributeScope struct {\n\tClusterAttributes map[string]*ActiveClusterInfo `json:\"clusterAttributes,omitempty\"`\n}\n\n// ToWire translates a ClusterAttributeScope struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClusterAttributeScope) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ClusterAttributes != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_ActiveClusterInfo_MapItemList(v.ClusterAttributes)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ClusterAttributeScope struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClusterAttributeScope struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClusterAttributeScope\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClusterAttributeScope) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ClusterAttributes, err = _Map_String_ActiveClusterInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClusterAttributeScope struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClusterAttributeScope struct could not be encoded.\nfunc (v *ClusterAttributeScope) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ClusterAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_ActiveClusterInfo_Encode(v.ClusterAttributes, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ClusterAttributeScope struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClusterAttributeScope struct could not be generated from the wire\n// representation.\nfunc (v *ClusterAttributeScope) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.ClusterAttributes, err = _Map_String_ActiveClusterInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClusterAttributeScope\n// struct.\nfunc (v *ClusterAttributeScope) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ClusterAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterAttributes: %v\", v.ClusterAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ClusterAttributeScope{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ClusterAttributeScope match the\n// provided ClusterAttributeScope.\n//\n// This function performs a deep comparison.\nfunc (v *ClusterAttributeScope) Equals(rhs *ClusterAttributeScope) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ClusterAttributes == nil && rhs.ClusterAttributes == nil) || (v.ClusterAttributes != nil && rhs.ClusterAttributes != nil && _Map_String_ActiveClusterInfo_Equals(v.ClusterAttributes, rhs.ClusterAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClusterAttributeScope.\nfunc (v *ClusterAttributeScope) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ClusterAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterAttributes\", (_Map_String_ActiveClusterInfo_Zapper)(v.ClusterAttributes)))\n\t}\n\treturn err\n}\n\n// GetClusterAttributes returns the value of ClusterAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterAttributeScope) GetClusterAttributes() (o map[string]*ActiveClusterInfo) {\n\tif v != nil && v.ClusterAttributes != nil {\n\t\treturn v.ClusterAttributes\n\t}\n\n\treturn\n}\n\n// IsSetClusterAttributes returns true if ClusterAttributes is not nil.\nfunc (v *ClusterAttributeScope) IsSetClusterAttributes() bool {\n\treturn v != nil && v.ClusterAttributes != nil\n}\n\ntype ClusterFailover struct {\n\tFromCluster      *ActiveClusterInfo `json:\"fromCluster,omitempty\"`\n\tToCluster        *ActiveClusterInfo `json:\"toCluster,omitempty\"`\n\tClusterAttribute *ClusterAttribute  `json:\"clusterAttribute,omitempty\"`\n}\n\n// ToWire translates a ClusterFailover struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClusterFailover) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FromCluster != nil {\n\t\tw, err = v.FromCluster.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ToCluster != nil {\n\t\tw, err = v.ToCluster.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterAttribute != nil {\n\t\tw, err = v.ClusterAttribute.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ClusterFailover struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClusterFailover struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClusterFailover\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClusterFailover) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FromCluster, err = _ActiveClusterInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ToCluster, err = _ActiveClusterInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ClusterAttribute, err = _ClusterAttribute_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClusterFailover struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClusterFailover struct could not be encoded.\nfunc (v *ClusterFailover) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FromCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FromCluster.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ToCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ToCluster.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterAttribute != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ClusterAttribute.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ClusterFailover struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClusterFailover struct could not be generated from the wire\n// representation.\nfunc (v *ClusterFailover) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.FromCluster, err = _ActiveClusterInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ToCluster, err = _ActiveClusterInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ClusterAttribute, err = _ClusterAttribute_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClusterFailover\n// struct.\nfunc (v *ClusterFailover) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.FromCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"FromCluster: %v\", v.FromCluster)\n\t\ti++\n\t}\n\tif v.ToCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"ToCluster: %v\", v.ToCluster)\n\t\ti++\n\t}\n\tif v.ClusterAttribute != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterAttribute: %v\", v.ClusterAttribute)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ClusterFailover{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ClusterFailover match the\n// provided ClusterFailover.\n//\n// This function performs a deep comparison.\nfunc (v *ClusterFailover) Equals(rhs *ClusterFailover) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FromCluster == nil && rhs.FromCluster == nil) || (v.FromCluster != nil && rhs.FromCluster != nil && v.FromCluster.Equals(rhs.FromCluster))) {\n\t\treturn false\n\t}\n\tif !((v.ToCluster == nil && rhs.ToCluster == nil) || (v.ToCluster != nil && rhs.ToCluster != nil && v.ToCluster.Equals(rhs.ToCluster))) {\n\t\treturn false\n\t}\n\tif !((v.ClusterAttribute == nil && rhs.ClusterAttribute == nil) || (v.ClusterAttribute != nil && rhs.ClusterAttribute != nil && v.ClusterAttribute.Equals(rhs.ClusterAttribute))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClusterFailover.\nfunc (v *ClusterFailover) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FromCluster != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"fromCluster\", v.FromCluster))\n\t}\n\tif v.ToCluster != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"toCluster\", v.ToCluster))\n\t}\n\tif v.ClusterAttribute != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterAttribute\", v.ClusterAttribute))\n\t}\n\treturn err\n}\n\n// GetFromCluster returns the value of FromCluster if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterFailover) GetFromCluster() (o *ActiveClusterInfo) {\n\tif v != nil && v.FromCluster != nil {\n\t\treturn v.FromCluster\n\t}\n\n\treturn\n}\n\n// IsSetFromCluster returns true if FromCluster is not nil.\nfunc (v *ClusterFailover) IsSetFromCluster() bool {\n\treturn v != nil && v.FromCluster != nil\n}\n\n// GetToCluster returns the value of ToCluster if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterFailover) GetToCluster() (o *ActiveClusterInfo) {\n\tif v != nil && v.ToCluster != nil {\n\t\treturn v.ToCluster\n\t}\n\n\treturn\n}\n\n// IsSetToCluster returns true if ToCluster is not nil.\nfunc (v *ClusterFailover) IsSetToCluster() bool {\n\treturn v != nil && v.ToCluster != nil\n}\n\n// GetClusterAttribute returns the value of ClusterAttribute if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterFailover) GetClusterAttribute() (o *ClusterAttribute) {\n\tif v != nil && v.ClusterAttribute != nil {\n\t\treturn v.ClusterAttribute\n\t}\n\n\treturn\n}\n\n// IsSetClusterAttribute returns true if ClusterAttribute is not nil.\nfunc (v *ClusterFailover) IsSetClusterAttribute() bool {\n\treturn v != nil && v.ClusterAttribute != nil\n}\n\ntype ClusterInfo struct {\n\tSupportedClientVersions *SupportedClientVersions `json:\"supportedClientVersions,omitempty\"`\n}\n\n// ToWire translates a ClusterInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClusterInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SupportedClientVersions != nil {\n\t\tw, err = v.SupportedClientVersions.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SupportedClientVersions_Read(w wire.Value) (*SupportedClientVersions, error) {\n\tvar v SupportedClientVersions\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ClusterInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClusterInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClusterInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClusterInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SupportedClientVersions, err = _SupportedClientVersions_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClusterInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClusterInfo struct could not be encoded.\nfunc (v *ClusterInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SupportedClientVersions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SupportedClientVersions.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SupportedClientVersions_Decode(sr stream.Reader) (*SupportedClientVersions, error) {\n\tvar v SupportedClientVersions\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ClusterInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClusterInfo struct could not be generated from the wire\n// representation.\nfunc (v *ClusterInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.SupportedClientVersions, err = _SupportedClientVersions_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClusterInfo\n// struct.\nfunc (v *ClusterInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SupportedClientVersions != nil {\n\t\tfields[i] = fmt.Sprintf(\"SupportedClientVersions: %v\", v.SupportedClientVersions)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ClusterInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ClusterInfo match the\n// provided ClusterInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ClusterInfo) Equals(rhs *ClusterInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SupportedClientVersions == nil && rhs.SupportedClientVersions == nil) || (v.SupportedClientVersions != nil && rhs.SupportedClientVersions != nil && v.SupportedClientVersions.Equals(rhs.SupportedClientVersions))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClusterInfo.\nfunc (v *ClusterInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SupportedClientVersions != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"supportedClientVersions\", v.SupportedClientVersions))\n\t}\n\treturn err\n}\n\n// GetSupportedClientVersions returns the value of SupportedClientVersions if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterInfo) GetSupportedClientVersions() (o *SupportedClientVersions) {\n\tif v != nil && v.SupportedClientVersions != nil {\n\t\treturn v.SupportedClientVersions\n\t}\n\n\treturn\n}\n\n// IsSetSupportedClientVersions returns true if SupportedClientVersions is not nil.\nfunc (v *ClusterInfo) IsSetSupportedClientVersions() bool {\n\treturn v != nil && v.SupportedClientVersions != nil\n}\n\ntype ClusterReplicationConfiguration struct {\n\tClusterName *string `json:\"clusterName,omitempty\"`\n}\n\n// ToWire translates a ClusterReplicationConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ClusterReplicationConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ClusterReplicationConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ClusterReplicationConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ClusterReplicationConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ClusterReplicationConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ClusterReplicationConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ClusterReplicationConfiguration struct could not be encoded.\nfunc (v *ClusterReplicationConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ClusterReplicationConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ClusterReplicationConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *ClusterReplicationConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ClusterReplicationConfiguration\n// struct.\nfunc (v *ClusterReplicationConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ClusterReplicationConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ClusterReplicationConfiguration match the\n// provided ClusterReplicationConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *ClusterReplicationConfiguration) Equals(rhs *ClusterReplicationConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ClusterReplicationConfiguration.\nfunc (v *ClusterReplicationConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\treturn err\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *ClusterReplicationConfiguration) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *ClusterReplicationConfiguration) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\ntype CompleteWorkflowExecutionDecisionAttributes struct {\n\tResult []byte `json:\"result,omitempty\"`\n}\n\n// ToWire translates a CompleteWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CompleteWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CompleteWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CompleteWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CompleteWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CompleteWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CompleteWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CompleteWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CompleteWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CompleteWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CompleteWorkflowExecutionDecisionAttributes match the\n// provided CompleteWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) Equals(rhs *CompleteWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CompleteWorkflowExecutionDecisionAttributes.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\treturn err\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *CompleteWorkflowExecutionDecisionAttributes) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\ntype ContinueAsNewInitiator int32\n\nconst (\n\tContinueAsNewInitiatorDecider      ContinueAsNewInitiator = 0\n\tContinueAsNewInitiatorRetryPolicy  ContinueAsNewInitiator = 1\n\tContinueAsNewInitiatorCronSchedule ContinueAsNewInitiator = 2\n)\n\n// ContinueAsNewInitiator_Values returns all recognized values of ContinueAsNewInitiator.\nfunc ContinueAsNewInitiator_Values() []ContinueAsNewInitiator {\n\treturn []ContinueAsNewInitiator{\n\t\tContinueAsNewInitiatorDecider,\n\t\tContinueAsNewInitiatorRetryPolicy,\n\t\tContinueAsNewInitiatorCronSchedule,\n\t}\n}\n\n// UnmarshalText tries to decode ContinueAsNewInitiator from a byte slice\n// containing its name.\n//\n//\tvar v ContinueAsNewInitiator\n//\terr := v.UnmarshalText([]byte(\"Decider\"))\nfunc (v *ContinueAsNewInitiator) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Decider\":\n\t\t*v = ContinueAsNewInitiatorDecider\n\t\treturn nil\n\tcase \"RetryPolicy\":\n\t\t*v = ContinueAsNewInitiatorRetryPolicy\n\t\treturn nil\n\tcase \"CronSchedule\":\n\t\t*v = ContinueAsNewInitiatorCronSchedule\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ContinueAsNewInitiator\", err)\n\t\t}\n\t\t*v = ContinueAsNewInitiator(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ContinueAsNewInitiator to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ContinueAsNewInitiator) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Decider\"), nil\n\tcase 1:\n\t\treturn []byte(\"RetryPolicy\"), nil\n\tcase 2:\n\t\treturn []byte(\"CronSchedule\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ContinueAsNewInitiator.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ContinueAsNewInitiator) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Decider\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"RetryPolicy\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"CronSchedule\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ContinueAsNewInitiator) Ptr() *ContinueAsNewInitiator {\n\treturn &v\n}\n\n// Encode encodes ContinueAsNewInitiator directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ContinueAsNewInitiator\n//\treturn v.Encode(sWriter)\nfunc (v ContinueAsNewInitiator) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ContinueAsNewInitiator into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ContinueAsNewInitiator) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ContinueAsNewInitiator from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ContinueAsNewInitiator(0), err\n//\t}\n//\n//\tvar v ContinueAsNewInitiator\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ContinueAsNewInitiator(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ContinueAsNewInitiator) FromWire(w wire.Value) error {\n\t*v = (ContinueAsNewInitiator)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ContinueAsNewInitiator directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ContinueAsNewInitiator\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ContinueAsNewInitiator(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ContinueAsNewInitiator) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ContinueAsNewInitiator)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ContinueAsNewInitiator.\nfunc (v ContinueAsNewInitiator) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Decider\"\n\tcase 1:\n\t\treturn \"RetryPolicy\"\n\tcase 2:\n\t\treturn \"CronSchedule\"\n\t}\n\treturn fmt.Sprintf(\"ContinueAsNewInitiator(%d)\", w)\n}\n\n// Equals returns true if this ContinueAsNewInitiator value matches the provided\n// value.\nfunc (v ContinueAsNewInitiator) Equals(rhs ContinueAsNewInitiator) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ContinueAsNewInitiator into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ContinueAsNewInitiator) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Decider\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"RetryPolicy\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"CronSchedule\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ContinueAsNewInitiator from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ContinueAsNewInitiator) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ContinueAsNewInitiator\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ContinueAsNewInitiator\")\n\t\t}\n\t\t*v = (ContinueAsNewInitiator)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ContinueAsNewInitiator\")\n\t}\n}\n\ntype ContinueAsNewWorkflowExecutionDecisionAttributes struct {\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tBackoffStartIntervalInSeconds       *int32                        `json:\"backoffStartIntervalInSeconds,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tFailureReason                       *string                       `json:\"failureReason,omitempty\"`\n\tFailureDetails                      []byte                        `json:\"failureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a ContinueAsNewWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [18]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.BackoffStartIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tw, err = v.Initiator.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.FailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.FailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.FailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.FailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tw, err = wire.NewValueBinary(v.LastCompletionResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.JitterStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ContinueAsNewInitiator_Read(w wire.Value) (ContinueAsNewInitiator, error) {\n\tvar v ContinueAsNewInitiator\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Memo_Read(w wire.Value) (*Memo, error) {\n\tvar v Memo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SearchAttributes_Read(w wire.Value) (*SearchAttributes, error) {\n\tvar v SearchAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CronOverlapPolicy_Read(w wire.Value) (CronOverlapPolicy, error) {\n\tvar v CronOverlapPolicy\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _ActiveClusterSelectionPolicy_Read(w wire.Value) (*ActiveClusterSelectionPolicy, error) {\n\tvar v ActiveClusterSelectionPolicy\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ContinueAsNewWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ContinueAsNewWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ContinueAsNewWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.BackoffStartIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ContinueAsNewInitiator\n\t\t\t\tx, err = _ContinueAsNewInitiator_Read(field.Value)\n\t\t\t\tv.Initiator = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.FailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastCompletionResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.JitterStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ContinueAsNewWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ContinueAsNewWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.BackoffStartIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Initiator != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Initiator.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.FailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastCompletionResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastCompletionResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.JitterStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.JitterStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ContinueAsNewInitiator_Decode(sr stream.Reader) (ContinueAsNewInitiator, error) {\n\tvar v ContinueAsNewInitiator\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Memo_Decode(sr stream.Reader) (*Memo, error) {\n\tvar v Memo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SearchAttributes_Decode(sr stream.Reader) (*SearchAttributes, error) {\n\tvar v SearchAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CronOverlapPolicy_Decode(sr stream.Reader) (CronOverlapPolicy, error) {\n\tvar v CronOverlapPolicy\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _ActiveClusterSelectionPolicy_Decode(sr stream.Reader) (*ActiveClusterSelectionPolicy, error) {\n\tvar v ActiveClusterSelectionPolicy\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ContinueAsNewWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ContinueAsNewWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.BackoffStartIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x ContinueAsNewInitiator\n\t\t\tx, err = _ContinueAsNewInitiator_Decode(sr)\n\t\t\tv.Initiator = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tv.FailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tv.LastCompletionResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.JitterStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ContinueAsNewWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [18]string\n\ti := 0\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"BackoffStartIntervalInSeconds: %v\", *(v.BackoffStartIntervalInSeconds))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tfields[i] = fmt.Sprintf(\"Initiator: %v\", *(v.Initiator))\n\t\ti++\n\t}\n\tif v.FailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailureReason: %v\", *(v.FailureReason))\n\t\ti++\n\t}\n\tif v.FailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailureDetails: %v\", v.FailureDetails)\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastCompletionResult: %v\", v.LastCompletionResult)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"JitterStartSeconds: %v\", *(v.JitterStartSeconds))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ContinueAsNewWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ContinueAsNewInitiator_EqualsPtr(lhs, rhs *ContinueAsNewInitiator) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _CronOverlapPolicy_EqualsPtr(lhs, rhs *CronOverlapPolicy) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ContinueAsNewWorkflowExecutionDecisionAttributes match the\n// provided ContinueAsNewWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) Equals(rhs *ContinueAsNewWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.BackoffStartIntervalInSeconds, rhs.BackoffStartIntervalInSeconds) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_ContinueAsNewInitiator_EqualsPtr(v.Initiator, rhs.Initiator) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FailureReason, rhs.FailureReason) {\n\t\treturn false\n\t}\n\tif !((v.FailureDetails == nil && rhs.FailureDetails == nil) || (v.FailureDetails != nil && rhs.FailureDetails != nil && bytes.Equal(v.FailureDetails, rhs.FailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.LastCompletionResult == nil && rhs.LastCompletionResult == nil) || (v.LastCompletionResult != nil && rhs.LastCompletionResult != nil && bytes.Equal(v.LastCompletionResult, rhs.LastCompletionResult))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.JitterStartSeconds, rhs.JitterStartSeconds) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ContinueAsNewWorkflowExecutionDecisionAttributes.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"backoffStartIntervalInSeconds\", *v.BackoffStartIntervalInSeconds)\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.Initiator != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"initiator\", *v.Initiator))\n\t}\n\tif v.FailureReason != nil {\n\t\tenc.AddString(\"failureReason\", *v.FailureReason)\n\t}\n\tif v.FailureDetails != nil {\n\t\tenc.AddString(\"failureDetails\", base64.StdEncoding.EncodeToString(v.FailureDetails))\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tenc.AddString(\"lastCompletionResult\", base64.StdEncoding.EncodeToString(v.LastCompletionResult))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tenc.AddInt32(\"jitterStartSeconds\", *v.JitterStartSeconds)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetBackoffStartIntervalInSeconds returns the value of BackoffStartIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetBackoffStartIntervalInSeconds() (o int32) {\n\tif v != nil && v.BackoffStartIntervalInSeconds != nil {\n\t\treturn *v.BackoffStartIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetBackoffStartIntervalInSeconds returns true if BackoffStartIntervalInSeconds is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetBackoffStartIntervalInSeconds() bool {\n\treturn v != nil && v.BackoffStartIntervalInSeconds != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetInitiator returns the value of Initiator if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\n\treturn\n}\n\n// IsSetInitiator returns true if Initiator is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetInitiator() bool {\n\treturn v != nil && v.Initiator != nil\n}\n\n// GetFailureReason returns the value of FailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetFailureReason() (o string) {\n\tif v != nil && v.FailureReason != nil {\n\t\treturn *v.FailureReason\n\t}\n\n\treturn\n}\n\n// IsSetFailureReason returns true if FailureReason is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetFailureReason() bool {\n\treturn v != nil && v.FailureReason != nil\n}\n\n// GetFailureDetails returns the value of FailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetFailureDetails() (o []byte) {\n\tif v != nil && v.FailureDetails != nil {\n\t\treturn v.FailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetFailureDetails returns true if FailureDetails is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetFailureDetails() bool {\n\treturn v != nil && v.FailureDetails != nil\n}\n\n// GetLastCompletionResult returns the value of LastCompletionResult if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetLastCompletionResult() (o []byte) {\n\tif v != nil && v.LastCompletionResult != nil {\n\t\treturn v.LastCompletionResult\n\t}\n\n\treturn\n}\n\n// IsSetLastCompletionResult returns true if LastCompletionResult is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetLastCompletionResult() bool {\n\treturn v != nil && v.LastCompletionResult != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetJitterStartSeconds returns the value of JitterStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetJitterStartSeconds returns true if JitterStartSeconds is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetJitterStartSeconds() bool {\n\treturn v != nil && v.JitterStartSeconds != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype CountWorkflowExecutionsRequest struct {\n\tDomain *string `json:\"domain,omitempty\"`\n\tQuery  *string `json:\"query,omitempty\"`\n}\n\n// ToWire translates a CountWorkflowExecutionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CountWorkflowExecutionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = wire.NewValueString(*(v.Query)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CountWorkflowExecutionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CountWorkflowExecutionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CountWorkflowExecutionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CountWorkflowExecutionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Query = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CountWorkflowExecutionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CountWorkflowExecutionsRequest struct could not be encoded.\nfunc (v *CountWorkflowExecutionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Query)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CountWorkflowExecutionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CountWorkflowExecutionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *CountWorkflowExecutionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Query = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CountWorkflowExecutionsRequest\n// struct.\nfunc (v *CountWorkflowExecutionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", *(v.Query))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CountWorkflowExecutionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CountWorkflowExecutionsRequest match the\n// provided CountWorkflowExecutionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *CountWorkflowExecutionsRequest) Equals(rhs *CountWorkflowExecutionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Query, rhs.Query) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CountWorkflowExecutionsRequest.\nfunc (v *CountWorkflowExecutionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Query != nil {\n\t\tenc.AddString(\"query\", *v.Query)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *CountWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *CountWorkflowExecutionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *CountWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil && v.Query != nil {\n\t\treturn *v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *CountWorkflowExecutionsRequest) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\ntype CountWorkflowExecutionsResponse struct {\n\tCount *int64 `json:\"count,omitempty\"`\n}\n\n// ToWire translates a CountWorkflowExecutionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CountWorkflowExecutionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Count != nil {\n\t\tw, err = wire.NewValueI64(*(v.Count)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CountWorkflowExecutionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CountWorkflowExecutionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CountWorkflowExecutionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CountWorkflowExecutionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Count = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CountWorkflowExecutionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CountWorkflowExecutionsResponse struct could not be encoded.\nfunc (v *CountWorkflowExecutionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Count != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Count)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CountWorkflowExecutionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CountWorkflowExecutionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *CountWorkflowExecutionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Count = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CountWorkflowExecutionsResponse\n// struct.\nfunc (v *CountWorkflowExecutionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Count != nil {\n\t\tfields[i] = fmt.Sprintf(\"Count: %v\", *(v.Count))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CountWorkflowExecutionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CountWorkflowExecutionsResponse match the\n// provided CountWorkflowExecutionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *CountWorkflowExecutionsResponse) Equals(rhs *CountWorkflowExecutionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Count, rhs.Count) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CountWorkflowExecutionsResponse.\nfunc (v *CountWorkflowExecutionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Count != nil {\n\t\tenc.AddInt64(\"count\", *v.Count)\n\t}\n\treturn err\n}\n\n// GetCount returns the value of Count if it is set or its\n// zero value if it is unset.\nfunc (v *CountWorkflowExecutionsResponse) GetCount() (o int64) {\n\tif v != nil && v.Count != nil {\n\t\treturn *v.Count\n\t}\n\n\treturn\n}\n\n// IsSetCount returns true if Count is not nil.\nfunc (v *CountWorkflowExecutionsResponse) IsSetCount() bool {\n\treturn v != nil && v.Count != nil\n}\n\ntype CronOverlapPolicy int32\n\nconst (\n\tCronOverlapPolicySkipped   CronOverlapPolicy = 0\n\tCronOverlapPolicyBufferone CronOverlapPolicy = 1\n)\n\n// CronOverlapPolicy_Values returns all recognized values of CronOverlapPolicy.\nfunc CronOverlapPolicy_Values() []CronOverlapPolicy {\n\treturn []CronOverlapPolicy{\n\t\tCronOverlapPolicySkipped,\n\t\tCronOverlapPolicyBufferone,\n\t}\n}\n\n// UnmarshalText tries to decode CronOverlapPolicy from a byte slice\n// containing its name.\n//\n//\tvar v CronOverlapPolicy\n//\terr := v.UnmarshalText([]byte(\"SKIPPED\"))\nfunc (v *CronOverlapPolicy) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"SKIPPED\":\n\t\t*v = CronOverlapPolicySkipped\n\t\treturn nil\n\tcase \"BUFFERONE\":\n\t\t*v = CronOverlapPolicyBufferone\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CronOverlapPolicy\", err)\n\t\t}\n\t\t*v = CronOverlapPolicy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CronOverlapPolicy to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v CronOverlapPolicy) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"SKIPPED\"), nil\n\tcase 1:\n\t\treturn []byte(\"BUFFERONE\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CronOverlapPolicy.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v CronOverlapPolicy) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"SKIPPED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"BUFFERONE\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v CronOverlapPolicy) Ptr() *CronOverlapPolicy {\n\treturn &v\n}\n\n// Encode encodes CronOverlapPolicy directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v CronOverlapPolicy\n//\treturn v.Encode(sWriter)\nfunc (v CronOverlapPolicy) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates CronOverlapPolicy into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v CronOverlapPolicy) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes CronOverlapPolicy from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return CronOverlapPolicy(0), err\n//\t}\n//\n//\tvar v CronOverlapPolicy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return CronOverlapPolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CronOverlapPolicy) FromWire(w wire.Value) error {\n\t*v = (CronOverlapPolicy)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded CronOverlapPolicy directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v CronOverlapPolicy\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return CronOverlapPolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CronOverlapPolicy) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (CronOverlapPolicy)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of CronOverlapPolicy.\nfunc (v CronOverlapPolicy) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SKIPPED\"\n\tcase 1:\n\t\treturn \"BUFFERONE\"\n\t}\n\treturn fmt.Sprintf(\"CronOverlapPolicy(%d)\", w)\n}\n\n// Equals returns true if this CronOverlapPolicy value matches the provided\n// value.\nfunc (v CronOverlapPolicy) Equals(rhs CronOverlapPolicy) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes CronOverlapPolicy into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v CronOverlapPolicy) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"SKIPPED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"BUFFERONE\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode CronOverlapPolicy from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *CronOverlapPolicy) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"CronOverlapPolicy\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"CronOverlapPolicy\")\n\t\t}\n\t\t*v = (CronOverlapPolicy)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"CronOverlapPolicy\")\n\t}\n}\n\ntype CrossClusterApplyParentClosePolicyRequestAttributes struct {\n\tChildren []*ApplyParentClosePolicyRequest `json:\"children,omitempty\"`\n}\n\ntype _List_ApplyParentClosePolicyRequest_ValueList []*ApplyParentClosePolicyRequest\n\nfunc (v _List_ApplyParentClosePolicyRequest_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ApplyParentClosePolicyRequest', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ApplyParentClosePolicyRequest_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ApplyParentClosePolicyRequest_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ApplyParentClosePolicyRequest_ValueList) Close() {}\n\n// ToWire translates a CrossClusterApplyParentClosePolicyRequestAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Children != nil {\n\t\tw, err = wire.NewValueList(_List_ApplyParentClosePolicyRequest_ValueList(v.Children)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ApplyParentClosePolicyRequest_Read(w wire.Value) (*ApplyParentClosePolicyRequest, error) {\n\tvar v ApplyParentClosePolicyRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ApplyParentClosePolicyRequest_Read(l wire.ValueList) ([]*ApplyParentClosePolicyRequest, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ApplyParentClosePolicyRequest, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ApplyParentClosePolicyRequest_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a CrossClusterApplyParentClosePolicyRequestAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterApplyParentClosePolicyRequestAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterApplyParentClosePolicyRequestAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Children, err = _List_ApplyParentClosePolicyRequest_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ApplyParentClosePolicyRequest_Encode(val []*ApplyParentClosePolicyRequest, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ApplyParentClosePolicyRequest', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a CrossClusterApplyParentClosePolicyRequestAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterApplyParentClosePolicyRequestAttributes struct could not be encoded.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Children != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ApplyParentClosePolicyRequest_Encode(v.Children, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ApplyParentClosePolicyRequest_Decode(sr stream.Reader) (*ApplyParentClosePolicyRequest, error) {\n\tvar v ApplyParentClosePolicyRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ApplyParentClosePolicyRequest_Decode(sr stream.Reader) ([]*ApplyParentClosePolicyRequest, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ApplyParentClosePolicyRequest, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ApplyParentClosePolicyRequest_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a CrossClusterApplyParentClosePolicyRequestAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterApplyParentClosePolicyRequestAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Children, err = _List_ApplyParentClosePolicyRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterApplyParentClosePolicyRequestAttributes\n// struct.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Children != nil {\n\t\tfields[i] = fmt.Sprintf(\"Children: %v\", v.Children)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterApplyParentClosePolicyRequestAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ApplyParentClosePolicyRequest_Equals(lhs, rhs []*ApplyParentClosePolicyRequest) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this CrossClusterApplyParentClosePolicyRequestAttributes match the\n// provided CrossClusterApplyParentClosePolicyRequestAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) Equals(rhs *CrossClusterApplyParentClosePolicyRequestAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Children == nil && rhs.Children == nil) || (v.Children != nil && rhs.Children != nil && _List_ApplyParentClosePolicyRequest_Equals(v.Children, rhs.Children))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ApplyParentClosePolicyRequest_Zapper []*ApplyParentClosePolicyRequest\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ApplyParentClosePolicyRequest_Zapper.\nfunc (l _List_ApplyParentClosePolicyRequest_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterApplyParentClosePolicyRequestAttributes.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Children != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"children\", (_List_ApplyParentClosePolicyRequest_Zapper)(v.Children)))\n\t}\n\treturn err\n}\n\n// GetChildren returns the value of Children if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) GetChildren() (o []*ApplyParentClosePolicyRequest) {\n\tif v != nil && v.Children != nil {\n\t\treturn v.Children\n\t}\n\n\treturn\n}\n\n// IsSetChildren returns true if Children is not nil.\nfunc (v *CrossClusterApplyParentClosePolicyRequestAttributes) IsSetChildren() bool {\n\treturn v != nil && v.Children != nil\n}\n\ntype CrossClusterApplyParentClosePolicyResponseAttributes struct {\n\tChildrenStatus []*ApplyParentClosePolicyResult `json:\"childrenStatus,omitempty\"`\n}\n\ntype _List_ApplyParentClosePolicyResult_ValueList []*ApplyParentClosePolicyResult\n\nfunc (v _List_ApplyParentClosePolicyResult_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ApplyParentClosePolicyResult', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ApplyParentClosePolicyResult_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ApplyParentClosePolicyResult_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ApplyParentClosePolicyResult_ValueList) Close() {}\n\n// ToWire translates a CrossClusterApplyParentClosePolicyResponseAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ChildrenStatus != nil {\n\t\tw, err = wire.NewValueList(_List_ApplyParentClosePolicyResult_ValueList(v.ChildrenStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ApplyParentClosePolicyResult_Read(w wire.Value) (*ApplyParentClosePolicyResult, error) {\n\tvar v ApplyParentClosePolicyResult\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ApplyParentClosePolicyResult_Read(l wire.ValueList) ([]*ApplyParentClosePolicyResult, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ApplyParentClosePolicyResult, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ApplyParentClosePolicyResult_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a CrossClusterApplyParentClosePolicyResponseAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterApplyParentClosePolicyResponseAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterApplyParentClosePolicyResponseAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ChildrenStatus, err = _List_ApplyParentClosePolicyResult_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ApplyParentClosePolicyResult_Encode(val []*ApplyParentClosePolicyResult, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ApplyParentClosePolicyResult', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a CrossClusterApplyParentClosePolicyResponseAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterApplyParentClosePolicyResponseAttributes struct could not be encoded.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ChildrenStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ApplyParentClosePolicyResult_Encode(v.ChildrenStatus, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ApplyParentClosePolicyResult_Decode(sr stream.Reader) (*ApplyParentClosePolicyResult, error) {\n\tvar v ApplyParentClosePolicyResult\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ApplyParentClosePolicyResult_Decode(sr stream.Reader) ([]*ApplyParentClosePolicyResult, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ApplyParentClosePolicyResult, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ApplyParentClosePolicyResult_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a CrossClusterApplyParentClosePolicyResponseAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterApplyParentClosePolicyResponseAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ChildrenStatus, err = _List_ApplyParentClosePolicyResult_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterApplyParentClosePolicyResponseAttributes\n// struct.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ChildrenStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildrenStatus: %v\", v.ChildrenStatus)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterApplyParentClosePolicyResponseAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ApplyParentClosePolicyResult_Equals(lhs, rhs []*ApplyParentClosePolicyResult) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this CrossClusterApplyParentClosePolicyResponseAttributes match the\n// provided CrossClusterApplyParentClosePolicyResponseAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) Equals(rhs *CrossClusterApplyParentClosePolicyResponseAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ChildrenStatus == nil && rhs.ChildrenStatus == nil) || (v.ChildrenStatus != nil && rhs.ChildrenStatus != nil && _List_ApplyParentClosePolicyResult_Equals(v.ChildrenStatus, rhs.ChildrenStatus))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ApplyParentClosePolicyResult_Zapper []*ApplyParentClosePolicyResult\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ApplyParentClosePolicyResult_Zapper.\nfunc (l _List_ApplyParentClosePolicyResult_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterApplyParentClosePolicyResponseAttributes.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ChildrenStatus != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"childrenStatus\", (_List_ApplyParentClosePolicyResult_Zapper)(v.ChildrenStatus)))\n\t}\n\treturn err\n}\n\n// GetChildrenStatus returns the value of ChildrenStatus if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) GetChildrenStatus() (o []*ApplyParentClosePolicyResult) {\n\tif v != nil && v.ChildrenStatus != nil {\n\t\treturn v.ChildrenStatus\n\t}\n\n\treturn\n}\n\n// IsSetChildrenStatus returns true if ChildrenStatus is not nil.\nfunc (v *CrossClusterApplyParentClosePolicyResponseAttributes) IsSetChildrenStatus() bool {\n\treturn v != nil && v.ChildrenStatus != nil\n}\n\ntype CrossClusterCancelExecutionRequestAttributes struct {\n\tTargetDomainID    *string `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID  *string `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID       *string `json:\"targetRunID,omitempty\"`\n\tRequestID         *string `json:\"requestID,omitempty\"`\n\tInitiatedEventID  *int64  `json:\"initiatedEventID,omitempty\"`\n\tChildWorkflowOnly *bool   `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a CrossClusterCancelExecutionRequestAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterCancelExecutionRequestAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TargetDomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetDomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterCancelExecutionRequestAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterCancelExecutionRequestAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterCancelExecutionRequestAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterCancelExecutionRequestAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetDomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterCancelExecutionRequestAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterCancelExecutionRequestAttributes struct could not be encoded.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TargetDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetDomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterCancelExecutionRequestAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterCancelExecutionRequestAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetDomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterCancelExecutionRequestAttributes\n// struct.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.TargetDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainID: %v\", *(v.TargetDomainID))\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetWorkflowID: %v\", *(v.TargetWorkflowID))\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetRunID: %v\", *(v.TargetRunID))\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestID: %v\", *(v.RequestID))\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventID: %v\", *(v.InitiatedEventID))\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterCancelExecutionRequestAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterCancelExecutionRequestAttributes match the\n// provided CrossClusterCancelExecutionRequestAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) Equals(rhs *CrossClusterCancelExecutionRequestAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetDomainID, rhs.TargetDomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetWorkflowID, rhs.TargetWorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetRunID, rhs.TargetRunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestID, rhs.RequestID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventID, rhs.InitiatedEventID) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterCancelExecutionRequestAttributes.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TargetDomainID != nil {\n\t\tenc.AddString(\"targetDomainID\", *v.TargetDomainID)\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tenc.AddString(\"targetWorkflowID\", *v.TargetWorkflowID)\n\t}\n\tif v.TargetRunID != nil {\n\t\tenc.AddString(\"targetRunID\", *v.TargetRunID)\n\t}\n\tif v.RequestID != nil {\n\t\tenc.AddString(\"requestID\", *v.RequestID)\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tenc.AddInt64(\"initiatedEventID\", *v.InitiatedEventID)\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetTargetDomainID returns the value of TargetDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetTargetDomainID() (o string) {\n\tif v != nil && v.TargetDomainID != nil {\n\t\treturn *v.TargetDomainID\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainID returns true if TargetDomainID is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetTargetDomainID() bool {\n\treturn v != nil && v.TargetDomainID != nil\n}\n\n// GetTargetWorkflowID returns the value of TargetWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetTargetWorkflowID() (o string) {\n\tif v != nil && v.TargetWorkflowID != nil {\n\t\treturn *v.TargetWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetTargetWorkflowID returns true if TargetWorkflowID is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetTargetWorkflowID() bool {\n\treturn v != nil && v.TargetWorkflowID != nil\n}\n\n// GetTargetRunID returns the value of TargetRunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetTargetRunID() (o string) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn *v.TargetRunID\n\t}\n\n\treturn\n}\n\n// IsSetTargetRunID returns true if TargetRunID is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetTargetRunID() bool {\n\treturn v != nil && v.TargetRunID != nil\n}\n\n// GetRequestID returns the value of RequestID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetRequestID() (o string) {\n\tif v != nil && v.RequestID != nil {\n\t\treturn *v.RequestID\n\t}\n\n\treturn\n}\n\n// IsSetRequestID returns true if RequestID is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetRequestID() bool {\n\treturn v != nil && v.RequestID != nil\n}\n\n// GetInitiatedEventID returns the value of InitiatedEventID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil && v.InitiatedEventID != nil {\n\t\treturn *v.InitiatedEventID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventID returns true if InitiatedEventID is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetInitiatedEventID() bool {\n\treturn v != nil && v.InitiatedEventID != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *CrossClusterCancelExecutionRequestAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype CrossClusterCancelExecutionResponseAttributes struct {\n}\n\n// ToWire translates a CrossClusterCancelExecutionResponseAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterCancelExecutionResponseAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterCancelExecutionResponseAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterCancelExecutionResponseAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterCancelExecutionResponseAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterCancelExecutionResponseAttributes) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterCancelExecutionResponseAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterCancelExecutionResponseAttributes struct could not be encoded.\nfunc (v *CrossClusterCancelExecutionResponseAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterCancelExecutionResponseAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterCancelExecutionResponseAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterCancelExecutionResponseAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterCancelExecutionResponseAttributes\n// struct.\nfunc (v *CrossClusterCancelExecutionResponseAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"CrossClusterCancelExecutionResponseAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterCancelExecutionResponseAttributes match the\n// provided CrossClusterCancelExecutionResponseAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterCancelExecutionResponseAttributes) Equals(rhs *CrossClusterCancelExecutionResponseAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterCancelExecutionResponseAttributes.\nfunc (v *CrossClusterCancelExecutionResponseAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct {\n\tTargetDomainID   *string       `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID *string       `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID      *string       `json:\"targetRunID,omitempty\"`\n\tInitiatedEventID *int64        `json:\"initiatedEventID,omitempty\"`\n\tCompletionEvent  *HistoryEvent `json:\"completionEvent,omitempty\"`\n}\n\n// ToWire translates a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TargetDomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetDomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tw, err = v.CompletionEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryEvent_Read(w wire.Value) (*HistoryEvent, error) {\n\tvar v HistoryEvent\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetDomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompletionEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct could not be encoded.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TargetDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetDomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletionEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletionEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryEvent_Decode(sr stream.Reader) (*HistoryEvent, error) {\n\tvar v HistoryEvent\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetDomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.CompletionEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes\n// struct.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.TargetDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainID: %v\", *(v.TargetDomainID))\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetWorkflowID: %v\", *(v.TargetWorkflowID))\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetRunID: %v\", *(v.TargetRunID))\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventID: %v\", *(v.InitiatedEventID))\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionEvent: %v\", v.CompletionEvent)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes match the\n// provided CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) Equals(rhs *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetDomainID, rhs.TargetDomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetWorkflowID, rhs.TargetWorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetRunID, rhs.TargetRunID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventID, rhs.InitiatedEventID) {\n\t\treturn false\n\t}\n\tif !((v.CompletionEvent == nil && rhs.CompletionEvent == nil) || (v.CompletionEvent != nil && rhs.CompletionEvent != nil && v.CompletionEvent.Equals(rhs.CompletionEvent))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TargetDomainID != nil {\n\t\tenc.AddString(\"targetDomainID\", *v.TargetDomainID)\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tenc.AddString(\"targetWorkflowID\", *v.TargetWorkflowID)\n\t}\n\tif v.TargetRunID != nil {\n\t\tenc.AddString(\"targetRunID\", *v.TargetRunID)\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tenc.AddInt64(\"initiatedEventID\", *v.InitiatedEventID)\n\t}\n\tif v.CompletionEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completionEvent\", v.CompletionEvent))\n\t}\n\treturn err\n}\n\n// GetTargetDomainID returns the value of TargetDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) GetTargetDomainID() (o string) {\n\tif v != nil && v.TargetDomainID != nil {\n\t\treturn *v.TargetDomainID\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainID returns true if TargetDomainID is not nil.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) IsSetTargetDomainID() bool {\n\treturn v != nil && v.TargetDomainID != nil\n}\n\n// GetTargetWorkflowID returns the value of TargetWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) GetTargetWorkflowID() (o string) {\n\tif v != nil && v.TargetWorkflowID != nil {\n\t\treturn *v.TargetWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetTargetWorkflowID returns true if TargetWorkflowID is not nil.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) IsSetTargetWorkflowID() bool {\n\treturn v != nil && v.TargetWorkflowID != nil\n}\n\n// GetTargetRunID returns the value of TargetRunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) GetTargetRunID() (o string) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn *v.TargetRunID\n\t}\n\n\treturn\n}\n\n// IsSetTargetRunID returns true if TargetRunID is not nil.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) IsSetTargetRunID() bool {\n\treturn v != nil && v.TargetRunID != nil\n}\n\n// GetInitiatedEventID returns the value of InitiatedEventID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil && v.InitiatedEventID != nil {\n\t\treturn *v.InitiatedEventID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventID returns true if InitiatedEventID is not nil.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) IsSetInitiatedEventID() bool {\n\treturn v != nil && v.InitiatedEventID != nil\n}\n\n// GetCompletionEvent returns the value of CompletionEvent if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) GetCompletionEvent() (o *HistoryEvent) {\n\tif v != nil && v.CompletionEvent != nil {\n\t\treturn v.CompletionEvent\n\t}\n\n\treturn\n}\n\n// IsSetCompletionEvent returns true if CompletionEvent is not nil.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) IsSetCompletionEvent() bool {\n\treturn v != nil && v.CompletionEvent != nil\n}\n\ntype CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct {\n}\n\n// ToWire translates a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct could not be encoded.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes\n// struct.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes match the\n// provided CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) Equals(rhs *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes.\nfunc (v *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype CrossClusterSignalExecutionRequestAttributes struct {\n\tTargetDomainID    *string `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID  *string `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID       *string `json:\"targetRunID,omitempty\"`\n\tRequestID         *string `json:\"requestID,omitempty\"`\n\tInitiatedEventID  *int64  `json:\"initiatedEventID,omitempty\"`\n\tChildWorkflowOnly *bool   `json:\"childWorkflowOnly,omitempty\"`\n\tSignalName        *string `json:\"signalName,omitempty\"`\n\tSignalInput       []byte  `json:\"signalInput,omitempty\"`\n\tControl           []byte  `json:\"control,omitempty\"`\n}\n\n// ToWire translates a CrossClusterSignalExecutionRequestAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterSignalExecutionRequestAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TargetDomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetDomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.SignalInput != nil {\n\t\tw, err = wire.NewValueBinary(v.SignalInput), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterSignalExecutionRequestAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterSignalExecutionRequestAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterSignalExecutionRequestAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterSignalExecutionRequestAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetDomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.SignalInput, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterSignalExecutionRequestAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterSignalExecutionRequestAttributes struct could not be encoded.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TargetDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetDomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalInput != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.SignalInput); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterSignalExecutionRequestAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterSignalExecutionRequestAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetDomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tv.SignalInput, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterSignalExecutionRequestAttributes\n// struct.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.TargetDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainID: %v\", *(v.TargetDomainID))\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetWorkflowID: %v\", *(v.TargetWorkflowID))\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetRunID: %v\", *(v.TargetRunID))\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestID: %v\", *(v.RequestID))\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventID: %v\", *(v.InitiatedEventID))\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.SignalInput != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalInput: %v\", v.SignalInput)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterSignalExecutionRequestAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterSignalExecutionRequestAttributes match the\n// provided CrossClusterSignalExecutionRequestAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) Equals(rhs *CrossClusterSignalExecutionRequestAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetDomainID, rhs.TargetDomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetWorkflowID, rhs.TargetWorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetRunID, rhs.TargetRunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestID, rhs.RequestID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventID, rhs.InitiatedEventID) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.SignalInput == nil && rhs.SignalInput == nil) || (v.SignalInput != nil && rhs.SignalInput != nil && bytes.Equal(v.SignalInput, rhs.SignalInput))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterSignalExecutionRequestAttributes.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TargetDomainID != nil {\n\t\tenc.AddString(\"targetDomainID\", *v.TargetDomainID)\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tenc.AddString(\"targetWorkflowID\", *v.TargetWorkflowID)\n\t}\n\tif v.TargetRunID != nil {\n\t\tenc.AddString(\"targetRunID\", *v.TargetRunID)\n\t}\n\tif v.RequestID != nil {\n\t\tenc.AddString(\"requestID\", *v.RequestID)\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tenc.AddInt64(\"initiatedEventID\", *v.InitiatedEventID)\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.SignalInput != nil {\n\t\tenc.AddString(\"signalInput\", base64.StdEncoding.EncodeToString(v.SignalInput))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetTargetDomainID returns the value of TargetDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetTargetDomainID() (o string) {\n\tif v != nil && v.TargetDomainID != nil {\n\t\treturn *v.TargetDomainID\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainID returns true if TargetDomainID is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetTargetDomainID() bool {\n\treturn v != nil && v.TargetDomainID != nil\n}\n\n// GetTargetWorkflowID returns the value of TargetWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetTargetWorkflowID() (o string) {\n\tif v != nil && v.TargetWorkflowID != nil {\n\t\treturn *v.TargetWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetTargetWorkflowID returns true if TargetWorkflowID is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetTargetWorkflowID() bool {\n\treturn v != nil && v.TargetWorkflowID != nil\n}\n\n// GetTargetRunID returns the value of TargetRunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetTargetRunID() (o string) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn *v.TargetRunID\n\t}\n\n\treturn\n}\n\n// IsSetTargetRunID returns true if TargetRunID is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetTargetRunID() bool {\n\treturn v != nil && v.TargetRunID != nil\n}\n\n// GetRequestID returns the value of RequestID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetRequestID() (o string) {\n\tif v != nil && v.RequestID != nil {\n\t\treturn *v.RequestID\n\t}\n\n\treturn\n}\n\n// IsSetRequestID returns true if RequestID is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetRequestID() bool {\n\treturn v != nil && v.RequestID != nil\n}\n\n// GetInitiatedEventID returns the value of InitiatedEventID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil && v.InitiatedEventID != nil {\n\t\treturn *v.InitiatedEventID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventID returns true if InitiatedEventID is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetInitiatedEventID() bool {\n\treturn v != nil && v.InitiatedEventID != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetSignalInput returns the value of SignalInput if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetSignalInput() (o []byte) {\n\tif v != nil && v.SignalInput != nil {\n\t\treturn v.SignalInput\n\t}\n\n\treturn\n}\n\n// IsSetSignalInput returns true if SignalInput is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetSignalInput() bool {\n\treturn v != nil && v.SignalInput != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *CrossClusterSignalExecutionRequestAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype CrossClusterSignalExecutionResponseAttributes struct {\n}\n\n// ToWire translates a CrossClusterSignalExecutionResponseAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterSignalExecutionResponseAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterSignalExecutionResponseAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterSignalExecutionResponseAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterSignalExecutionResponseAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterSignalExecutionResponseAttributes) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterSignalExecutionResponseAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterSignalExecutionResponseAttributes struct could not be encoded.\nfunc (v *CrossClusterSignalExecutionResponseAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterSignalExecutionResponseAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterSignalExecutionResponseAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterSignalExecutionResponseAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterSignalExecutionResponseAttributes\n// struct.\nfunc (v *CrossClusterSignalExecutionResponseAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"CrossClusterSignalExecutionResponseAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterSignalExecutionResponseAttributes match the\n// provided CrossClusterSignalExecutionResponseAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterSignalExecutionResponseAttributes) Equals(rhs *CrossClusterSignalExecutionResponseAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterSignalExecutionResponseAttributes.\nfunc (v *CrossClusterSignalExecutionResponseAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype CrossClusterStartChildExecutionRequestAttributes struct {\n\tTargetDomainID           *string                                              `json:\"targetDomainID,omitempty\"`\n\tRequestID                *string                                              `json:\"requestID,omitempty\"`\n\tInitiatedEventID         *int64                                               `json:\"initiatedEventID,omitempty\"`\n\tInitiatedEventAttributes *StartChildWorkflowExecutionInitiatedEventAttributes `json:\"initiatedEventAttributes,omitempty\"`\n\tTargetRunID              *string                                              `json:\"targetRunID,omitempty\"`\n\tPartitionConfig          map[string]string                                    `json:\"partitionConfig,omitempty\"`\n}\n\ntype _Map_String_String_MapItemList map[string]string\n\nfunc (m _Map_String_String_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueString(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_String_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_String_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) Close() {}\n\n// ToWire translates a CrossClusterStartChildExecutionRequestAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TargetDomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetDomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventAttributes != nil {\n\t\tw, err = v.InitiatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartChildWorkflowExecutionInitiatedEventAttributes_Read(w wire.Value) (*StartChildWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v StartChildWorkflowExecutionInitiatedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Read(m wire.MapItemList) (map[string]string, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]string, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a CrossClusterStartChildExecutionRequestAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterStartChildExecutionRequestAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterStartChildExecutionRequestAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetDomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InitiatedEventAttributes, err = _StartChildWorkflowExecutionInitiatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_String_Encode(val map[string]string, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a CrossClusterStartChildExecutionRequestAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterStartChildExecutionRequestAttributes struct could not be encoded.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TargetDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetDomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InitiatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartChildWorkflowExecutionInitiatedEventAttributes_Decode(sr stream.Reader) (*StartChildWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v StartChildWorkflowExecutionInitiatedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_String_Decode(sr stream.Reader) (map[string]string, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]string, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a CrossClusterStartChildExecutionRequestAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterStartChildExecutionRequestAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetDomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.InitiatedEventAttributes, err = _StartChildWorkflowExecutionInitiatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterStartChildExecutionRequestAttributes\n// struct.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.TargetDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainID: %v\", *(v.TargetDomainID))\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestID: %v\", *(v.RequestID))\n\t\ti++\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventID: %v\", *(v.InitiatedEventID))\n\t\ti++\n\t}\n\tif v.InitiatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventAttributes: %v\", v.InitiatedEventAttributes)\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetRunID: %v\", *(v.TargetRunID))\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterStartChildExecutionRequestAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_String_Equals(lhs, rhs map[string]string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this CrossClusterStartChildExecutionRequestAttributes match the\n// provided CrossClusterStartChildExecutionRequestAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) Equals(rhs *CrossClusterStartChildExecutionRequestAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetDomainID, rhs.TargetDomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestID, rhs.RequestID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventID, rhs.InitiatedEventID) {\n\t\treturn false\n\t}\n\tif !((v.InitiatedEventAttributes == nil && rhs.InitiatedEventAttributes == nil) || (v.InitiatedEventAttributes != nil && rhs.InitiatedEventAttributes != nil && v.InitiatedEventAttributes.Equals(rhs.InitiatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetRunID, rhs.TargetRunID) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_String_Zapper map[string]string\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_String_Zapper.\nfunc (m _Map_String_String_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterStartChildExecutionRequestAttributes.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TargetDomainID != nil {\n\t\tenc.AddString(\"targetDomainID\", *v.TargetDomainID)\n\t}\n\tif v.RequestID != nil {\n\t\tenc.AddString(\"requestID\", *v.RequestID)\n\t}\n\tif v.InitiatedEventID != nil {\n\t\tenc.AddInt64(\"initiatedEventID\", *v.InitiatedEventID)\n\t}\n\tif v.InitiatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"initiatedEventAttributes\", v.InitiatedEventAttributes))\n\t}\n\tif v.TargetRunID != nil {\n\t\tenc.AddString(\"targetRunID\", *v.TargetRunID)\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetTargetDomainID returns the value of TargetDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetTargetDomainID() (o string) {\n\tif v != nil && v.TargetDomainID != nil {\n\t\treturn *v.TargetDomainID\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainID returns true if TargetDomainID is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetTargetDomainID() bool {\n\treturn v != nil && v.TargetDomainID != nil\n}\n\n// GetRequestID returns the value of RequestID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetRequestID() (o string) {\n\tif v != nil && v.RequestID != nil {\n\t\treturn *v.RequestID\n\t}\n\n\treturn\n}\n\n// IsSetRequestID returns true if RequestID is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetRequestID() bool {\n\treturn v != nil && v.RequestID != nil\n}\n\n// GetInitiatedEventID returns the value of InitiatedEventID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil && v.InitiatedEventID != nil {\n\t\treturn *v.InitiatedEventID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventID returns true if InitiatedEventID is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetInitiatedEventID() bool {\n\treturn v != nil && v.InitiatedEventID != nil\n}\n\n// GetInitiatedEventAttributes returns the value of InitiatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetInitiatedEventAttributes() (o *StartChildWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.InitiatedEventAttributes != nil {\n\t\treturn v.InitiatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventAttributes returns true if InitiatedEventAttributes is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetInitiatedEventAttributes() bool {\n\treturn v != nil && v.InitiatedEventAttributes != nil\n}\n\n// GetTargetRunID returns the value of TargetRunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetTargetRunID() (o string) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn *v.TargetRunID\n\t}\n\n\treturn\n}\n\n// IsSetTargetRunID returns true if TargetRunID is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetTargetRunID() bool {\n\treturn v != nil && v.TargetRunID != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype CrossClusterStartChildExecutionResponseAttributes struct {\n\tRunID *string `json:\"runID,omitempty\"`\n}\n\n// ToWire translates a CrossClusterStartChildExecutionResponseAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CrossClusterStartChildExecutionResponseAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterStartChildExecutionResponseAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterStartChildExecutionResponseAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterStartChildExecutionResponseAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterStartChildExecutionResponseAttributes struct could not be encoded.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CrossClusterStartChildExecutionResponseAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterStartChildExecutionResponseAttributes struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterStartChildExecutionResponseAttributes\n// struct.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterStartChildExecutionResponseAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterStartChildExecutionResponseAttributes match the\n// provided CrossClusterStartChildExecutionResponseAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) Equals(rhs *CrossClusterStartChildExecutionResponseAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterStartChildExecutionResponseAttributes.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\treturn err\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\ntype CrossClusterTaskFailedCause int32\n\nconst (\n\tCrossClusterTaskFailedCauseDomainNotActive          CrossClusterTaskFailedCause = 0\n\tCrossClusterTaskFailedCauseDomainNotExists          CrossClusterTaskFailedCause = 1\n\tCrossClusterTaskFailedCauseWorkflowAlreadyRunning   CrossClusterTaskFailedCause = 2\n\tCrossClusterTaskFailedCauseWorkflowNotExists        CrossClusterTaskFailedCause = 3\n\tCrossClusterTaskFailedCauseWorkflowAlreadyCompleted CrossClusterTaskFailedCause = 4\n\tCrossClusterTaskFailedCauseUncategorized            CrossClusterTaskFailedCause = 5\n)\n\n// CrossClusterTaskFailedCause_Values returns all recognized values of CrossClusterTaskFailedCause.\nfunc CrossClusterTaskFailedCause_Values() []CrossClusterTaskFailedCause {\n\treturn []CrossClusterTaskFailedCause{\n\t\tCrossClusterTaskFailedCauseDomainNotActive,\n\t\tCrossClusterTaskFailedCauseDomainNotExists,\n\t\tCrossClusterTaskFailedCauseWorkflowAlreadyRunning,\n\t\tCrossClusterTaskFailedCauseWorkflowNotExists,\n\t\tCrossClusterTaskFailedCauseWorkflowAlreadyCompleted,\n\t\tCrossClusterTaskFailedCauseUncategorized,\n\t}\n}\n\n// UnmarshalText tries to decode CrossClusterTaskFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v CrossClusterTaskFailedCause\n//\terr := v.UnmarshalText([]byte(\"DOMAIN_NOT_ACTIVE\"))\nfunc (v *CrossClusterTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"DOMAIN_NOT_ACTIVE\":\n\t\t*v = CrossClusterTaskFailedCauseDomainNotActive\n\t\treturn nil\n\tcase \"DOMAIN_NOT_EXISTS\":\n\t\t*v = CrossClusterTaskFailedCauseDomainNotExists\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_RUNNING\":\n\t\t*v = CrossClusterTaskFailedCauseWorkflowAlreadyRunning\n\t\treturn nil\n\tcase \"WORKFLOW_NOT_EXISTS\":\n\t\t*v = CrossClusterTaskFailedCauseWorkflowNotExists\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*v = CrossClusterTaskFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tcase \"UNCATEGORIZED\":\n\t\t*v = CrossClusterTaskFailedCauseUncategorized\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CrossClusterTaskFailedCause\", err)\n\t\t}\n\t\t*v = CrossClusterTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CrossClusterTaskFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v CrossClusterTaskFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"DOMAIN_NOT_ACTIVE\"), nil\n\tcase 1:\n\t\treturn []byte(\"DOMAIN_NOT_EXISTS\"), nil\n\tcase 2:\n\t\treturn []byte(\"WORKFLOW_ALREADY_RUNNING\"), nil\n\tcase 3:\n\t\treturn []byte(\"WORKFLOW_NOT_EXISTS\"), nil\n\tcase 4:\n\t\treturn []byte(\"WORKFLOW_ALREADY_COMPLETED\"), nil\n\tcase 5:\n\t\treturn []byte(\"UNCATEGORIZED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterTaskFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v CrossClusterTaskFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"DOMAIN_NOT_ACTIVE\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"DOMAIN_NOT_EXISTS\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"WORKFLOW_ALREADY_RUNNING\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"WORKFLOW_NOT_EXISTS\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"WORKFLOW_ALREADY_COMPLETED\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"UNCATEGORIZED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v CrossClusterTaskFailedCause) Ptr() *CrossClusterTaskFailedCause {\n\treturn &v\n}\n\n// Encode encodes CrossClusterTaskFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v CrossClusterTaskFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v CrossClusterTaskFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates CrossClusterTaskFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v CrossClusterTaskFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes CrossClusterTaskFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return CrossClusterTaskFailedCause(0), err\n//\t}\n//\n//\tvar v CrossClusterTaskFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return CrossClusterTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CrossClusterTaskFailedCause) FromWire(w wire.Value) error {\n\t*v = (CrossClusterTaskFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded CrossClusterTaskFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v CrossClusterTaskFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return CrossClusterTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CrossClusterTaskFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (CrossClusterTaskFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of CrossClusterTaskFailedCause.\nfunc (v CrossClusterTaskFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"DOMAIN_NOT_ACTIVE\"\n\tcase 1:\n\t\treturn \"DOMAIN_NOT_EXISTS\"\n\tcase 2:\n\t\treturn \"WORKFLOW_ALREADY_RUNNING\"\n\tcase 3:\n\t\treturn \"WORKFLOW_NOT_EXISTS\"\n\tcase 4:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\tcase 5:\n\t\treturn \"UNCATEGORIZED\"\n\t}\n\treturn fmt.Sprintf(\"CrossClusterTaskFailedCause(%d)\", w)\n}\n\n// Equals returns true if this CrossClusterTaskFailedCause value matches the provided\n// value.\nfunc (v CrossClusterTaskFailedCause) Equals(rhs CrossClusterTaskFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes CrossClusterTaskFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v CrossClusterTaskFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"DOMAIN_NOT_ACTIVE\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"DOMAIN_NOT_EXISTS\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_ALREADY_RUNNING\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_NOT_EXISTS\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_ALREADY_COMPLETED\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"UNCATEGORIZED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode CrossClusterTaskFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *CrossClusterTaskFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"CrossClusterTaskFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"CrossClusterTaskFailedCause\")\n\t\t}\n\t\t*v = (CrossClusterTaskFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"CrossClusterTaskFailedCause\")\n\t}\n}\n\ntype CrossClusterTaskInfo struct {\n\tDomainID            *string               `json:\"domainID,omitempty\"`\n\tWorkflowID          *string               `json:\"workflowID,omitempty\"`\n\tRunID               *string               `json:\"runID,omitempty\"`\n\tTaskType            *CrossClusterTaskType `json:\"taskType,omitempty\"`\n\tTaskState           *int16                `json:\"taskState,omitempty\"`\n\tTaskID              *int64                `json:\"taskID,omitempty\"`\n\tVisibilityTimestamp *int64                `json:\"visibilityTimestamp,omitempty\"`\n}\n\n// ToWire translates a CrossClusterTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = v.TaskType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskState != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.VisibilityTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterTaskType_Read(w wire.Value) (CrossClusterTaskType, error) {\n\tvar v CrossClusterTaskType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a CrossClusterTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CrossClusterTaskType\n\t\t\t\tx, err = _CrossClusterTaskType_Read(field.Value)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.VisibilityTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterTaskInfo struct could not be encoded.\nfunc (v *CrossClusterTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.VisibilityTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterTaskType_Decode(sr stream.Reader) (CrossClusterTaskType, error) {\n\tvar v CrossClusterTaskType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a CrossClusterTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x CrossClusterTaskType\n\t\t\tx, err = _CrossClusterTaskType_Decode(sr)\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.VisibilityTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterTaskInfo\n// struct.\nfunc (v *CrossClusterTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.TaskState != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskState: %v\", *(v.TaskState))\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityTimestamp: %v\", *(v.VisibilityTimestamp))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _CrossClusterTaskType_EqualsPtr(lhs, rhs *CrossClusterTaskType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I16_EqualsPtr(lhs, rhs *int16) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this CrossClusterTaskInfo match the\n// provided CrossClusterTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterTaskInfo) Equals(rhs *CrossClusterTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_CrossClusterTaskType_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskState, rhs.TaskState) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.VisibilityTimestamp, rhs.VisibilityTimestamp) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterTaskInfo.\nfunc (v *CrossClusterTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.TaskType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskType\", *v.TaskType))\n\t}\n\tif v.TaskState != nil {\n\t\tenc.AddInt16(\"taskState\", *v.TaskState)\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tenc.AddInt64(\"visibilityTimestamp\", *v.VisibilityTimestamp)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetTaskType() (o CrossClusterTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetTaskState returns the value of TaskState if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetTaskState() (o int16) {\n\tif v != nil && v.TaskState != nil {\n\t\treturn *v.TaskState\n\t}\n\n\treturn\n}\n\n// IsSetTaskState returns true if TaskState is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetTaskState() bool {\n\treturn v != nil && v.TaskState != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetVisibilityTimestamp returns the value of VisibilityTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskInfo) GetVisibilityTimestamp() (o int64) {\n\tif v != nil && v.VisibilityTimestamp != nil {\n\t\treturn *v.VisibilityTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityTimestamp returns true if VisibilityTimestamp is not nil.\nfunc (v *CrossClusterTaskInfo) IsSetVisibilityTimestamp() bool {\n\treturn v != nil && v.VisibilityTimestamp != nil\n}\n\ntype CrossClusterTaskRequest struct {\n\tTaskInfo                                       *CrossClusterTaskInfo                                              `json:\"taskInfo,omitempty\"`\n\tStartChildExecutionAttributes                  *CrossClusterStartChildExecutionRequestAttributes                  `json:\"startChildExecutionAttributes,omitempty\"`\n\tCancelExecutionAttributes                      *CrossClusterCancelExecutionRequestAttributes                      `json:\"cancelExecutionAttributes,omitempty\"`\n\tSignalExecutionAttributes                      *CrossClusterSignalExecutionRequestAttributes                      `json:\"signalExecutionAttributes,omitempty\"`\n\tRecordChildWorkflowExecutionCompleteAttributes *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes `json:\"recordChildWorkflowExecutionCompleteAttributes,omitempty\"`\n\tApplyParentClosePolicyAttributes               *CrossClusterApplyParentClosePolicyRequestAttributes               `json:\"applyParentClosePolicyAttributes,omitempty\"`\n}\n\n// ToWire translates a CrossClusterTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskInfo != nil {\n\t\tw, err = v.TaskInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\tw, err = v.StartChildExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\tw, err = v.CancelExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\tw, err = v.SignalExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tw, err = v.RecordChildWorkflowExecutionCompleteAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tw, err = v.ApplyParentClosePolicyAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterTaskInfo_Read(w wire.Value) (*CrossClusterTaskInfo, error) {\n\tvar v CrossClusterTaskInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterStartChildExecutionRequestAttributes_Read(w wire.Value) (*CrossClusterStartChildExecutionRequestAttributes, error) {\n\tvar v CrossClusterStartChildExecutionRequestAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterCancelExecutionRequestAttributes_Read(w wire.Value) (*CrossClusterCancelExecutionRequestAttributes, error) {\n\tvar v CrossClusterCancelExecutionRequestAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterSignalExecutionRequestAttributes_Read(w wire.Value) (*CrossClusterSignalExecutionRequestAttributes, error) {\n\tvar v CrossClusterSignalExecutionRequestAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes_Read(w wire.Value) (*CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes, error) {\n\tvar v CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterApplyParentClosePolicyRequestAttributes_Read(w wire.Value) (*CrossClusterApplyParentClosePolicyRequestAttributes, error) {\n\tvar v CrossClusterApplyParentClosePolicyRequestAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a CrossClusterTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskInfo, err = _CrossClusterTaskInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartChildExecutionAttributes, err = _CrossClusterStartChildExecutionRequestAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelExecutionAttributes, err = _CrossClusterCancelExecutionRequestAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalExecutionAttributes, err = _CrossClusterSignalExecutionRequestAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RecordChildWorkflowExecutionCompleteAttributes, err = _CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ApplyParentClosePolicyAttributes, err = _CrossClusterApplyParentClosePolicyRequestAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterTaskRequest struct could not be encoded.\nfunc (v *CrossClusterTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartChildExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartChildExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RecordChildWorkflowExecutionCompleteAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ApplyParentClosePolicyAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterTaskInfo_Decode(sr stream.Reader) (*CrossClusterTaskInfo, error) {\n\tvar v CrossClusterTaskInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterStartChildExecutionRequestAttributes_Decode(sr stream.Reader) (*CrossClusterStartChildExecutionRequestAttributes, error) {\n\tvar v CrossClusterStartChildExecutionRequestAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterCancelExecutionRequestAttributes_Decode(sr stream.Reader) (*CrossClusterCancelExecutionRequestAttributes, error) {\n\tvar v CrossClusterCancelExecutionRequestAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterSignalExecutionRequestAttributes_Decode(sr stream.Reader) (*CrossClusterSignalExecutionRequestAttributes, error) {\n\tvar v CrossClusterSignalExecutionRequestAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes_Decode(sr stream.Reader) (*CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes, error) {\n\tvar v CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterApplyParentClosePolicyRequestAttributes_Decode(sr stream.Reader) (*CrossClusterApplyParentClosePolicyRequestAttributes, error) {\n\tvar v CrossClusterApplyParentClosePolicyRequestAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a CrossClusterTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.TaskInfo, err = _CrossClusterTaskInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.StartChildExecutionAttributes, err = _CrossClusterStartChildExecutionRequestAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.CancelExecutionAttributes, err = _CrossClusterCancelExecutionRequestAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.SignalExecutionAttributes, err = _CrossClusterSignalExecutionRequestAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.RecordChildWorkflowExecutionCompleteAttributes, err = _CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.ApplyParentClosePolicyAttributes, err = _CrossClusterApplyParentClosePolicyRequestAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterTaskRequest\n// struct.\nfunc (v *CrossClusterTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.TaskInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskInfo: %v\", v.TaskInfo)\n\t\ti++\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartChildExecutionAttributes: %v\", v.StartChildExecutionAttributes)\n\t\ti++\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelExecutionAttributes: %v\", v.CancelExecutionAttributes)\n\t\ti++\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalExecutionAttributes: %v\", v.SignalExecutionAttributes)\n\t\ti++\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RecordChildWorkflowExecutionCompleteAttributes: %v\", v.RecordChildWorkflowExecutionCompleteAttributes)\n\t\ti++\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ApplyParentClosePolicyAttributes: %v\", v.ApplyParentClosePolicyAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterTaskRequest match the\n// provided CrossClusterTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterTaskRequest) Equals(rhs *CrossClusterTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskInfo == nil && rhs.TaskInfo == nil) || (v.TaskInfo != nil && rhs.TaskInfo != nil && v.TaskInfo.Equals(rhs.TaskInfo))) {\n\t\treturn false\n\t}\n\tif !((v.StartChildExecutionAttributes == nil && rhs.StartChildExecutionAttributes == nil) || (v.StartChildExecutionAttributes != nil && rhs.StartChildExecutionAttributes != nil && v.StartChildExecutionAttributes.Equals(rhs.StartChildExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CancelExecutionAttributes == nil && rhs.CancelExecutionAttributes == nil) || (v.CancelExecutionAttributes != nil && rhs.CancelExecutionAttributes != nil && v.CancelExecutionAttributes.Equals(rhs.CancelExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SignalExecutionAttributes == nil && rhs.SignalExecutionAttributes == nil) || (v.SignalExecutionAttributes != nil && rhs.SignalExecutionAttributes != nil && v.SignalExecutionAttributes.Equals(rhs.SignalExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RecordChildWorkflowExecutionCompleteAttributes == nil && rhs.RecordChildWorkflowExecutionCompleteAttributes == nil) || (v.RecordChildWorkflowExecutionCompleteAttributes != nil && rhs.RecordChildWorkflowExecutionCompleteAttributes != nil && v.RecordChildWorkflowExecutionCompleteAttributes.Equals(rhs.RecordChildWorkflowExecutionCompleteAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ApplyParentClosePolicyAttributes == nil && rhs.ApplyParentClosePolicyAttributes == nil) || (v.ApplyParentClosePolicyAttributes != nil && rhs.ApplyParentClosePolicyAttributes != nil && v.ApplyParentClosePolicyAttributes.Equals(rhs.ApplyParentClosePolicyAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterTaskRequest.\nfunc (v *CrossClusterTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskInfo\", v.TaskInfo))\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startChildExecutionAttributes\", v.StartChildExecutionAttributes))\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelExecutionAttributes\", v.CancelExecutionAttributes))\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalExecutionAttributes\", v.SignalExecutionAttributes))\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"recordChildWorkflowExecutionCompleteAttributes\", v.RecordChildWorkflowExecutionCompleteAttributes))\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"applyParentClosePolicyAttributes\", v.ApplyParentClosePolicyAttributes))\n\t}\n\treturn err\n}\n\n// GetTaskInfo returns the value of TaskInfo if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetTaskInfo() (o *CrossClusterTaskInfo) {\n\tif v != nil && v.TaskInfo != nil {\n\t\treturn v.TaskInfo\n\t}\n\n\treturn\n}\n\n// IsSetTaskInfo returns true if TaskInfo is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetTaskInfo() bool {\n\treturn v != nil && v.TaskInfo != nil\n}\n\n// GetStartChildExecutionAttributes returns the value of StartChildExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetStartChildExecutionAttributes() (o *CrossClusterStartChildExecutionRequestAttributes) {\n\tif v != nil && v.StartChildExecutionAttributes != nil {\n\t\treturn v.StartChildExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartChildExecutionAttributes returns true if StartChildExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetStartChildExecutionAttributes() bool {\n\treturn v != nil && v.StartChildExecutionAttributes != nil\n}\n\n// GetCancelExecutionAttributes returns the value of CancelExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetCancelExecutionAttributes() (o *CrossClusterCancelExecutionRequestAttributes) {\n\tif v != nil && v.CancelExecutionAttributes != nil {\n\t\treturn v.CancelExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCancelExecutionAttributes returns true if CancelExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetCancelExecutionAttributes() bool {\n\treturn v != nil && v.CancelExecutionAttributes != nil\n}\n\n// GetSignalExecutionAttributes returns the value of SignalExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetSignalExecutionAttributes() (o *CrossClusterSignalExecutionRequestAttributes) {\n\tif v != nil && v.SignalExecutionAttributes != nil {\n\t\treturn v.SignalExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSignalExecutionAttributes returns true if SignalExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetSignalExecutionAttributes() bool {\n\treturn v != nil && v.SignalExecutionAttributes != nil\n}\n\n// GetRecordChildWorkflowExecutionCompleteAttributes returns the value of RecordChildWorkflowExecutionCompleteAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetRecordChildWorkflowExecutionCompleteAttributes() (o *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) {\n\tif v != nil && v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\treturn v.RecordChildWorkflowExecutionCompleteAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRecordChildWorkflowExecutionCompleteAttributes returns true if RecordChildWorkflowExecutionCompleteAttributes is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetRecordChildWorkflowExecutionCompleteAttributes() bool {\n\treturn v != nil && v.RecordChildWorkflowExecutionCompleteAttributes != nil\n}\n\n// GetApplyParentClosePolicyAttributes returns the value of ApplyParentClosePolicyAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskRequest) GetApplyParentClosePolicyAttributes() (o *CrossClusterApplyParentClosePolicyRequestAttributes) {\n\tif v != nil && v.ApplyParentClosePolicyAttributes != nil {\n\t\treturn v.ApplyParentClosePolicyAttributes\n\t}\n\n\treturn\n}\n\n// IsSetApplyParentClosePolicyAttributes returns true if ApplyParentClosePolicyAttributes is not nil.\nfunc (v *CrossClusterTaskRequest) IsSetApplyParentClosePolicyAttributes() bool {\n\treturn v != nil && v.ApplyParentClosePolicyAttributes != nil\n}\n\ntype CrossClusterTaskResponse struct {\n\tTaskID                                         *int64                                                              `json:\"taskID,omitempty\"`\n\tTaskType                                       *CrossClusterTaskType                                               `json:\"taskType,omitempty\"`\n\tTaskState                                      *int16                                                              `json:\"taskState,omitempty\"`\n\tFailedCause                                    *CrossClusterTaskFailedCause                                        `json:\"failedCause,omitempty\"`\n\tStartChildExecutionAttributes                  *CrossClusterStartChildExecutionResponseAttributes                  `json:\"startChildExecutionAttributes,omitempty\"`\n\tCancelExecutionAttributes                      *CrossClusterCancelExecutionResponseAttributes                      `json:\"cancelExecutionAttributes,omitempty\"`\n\tSignalExecutionAttributes                      *CrossClusterSignalExecutionResponseAttributes                      `json:\"signalExecutionAttributes,omitempty\"`\n\tRecordChildWorkflowExecutionCompleteAttributes *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes `json:\"recordChildWorkflowExecutionCompleteAttributes,omitempty\"`\n\tApplyParentClosePolicyAttributes               *CrossClusterApplyParentClosePolicyResponseAttributes               `json:\"applyParentClosePolicyAttributes,omitempty\"`\n}\n\n// ToWire translates a CrossClusterTaskResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CrossClusterTaskResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = v.TaskType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskState != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskState)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tw, err = v.FailedCause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\tw, err = v.StartChildExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\tw, err = v.CancelExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\tw, err = v.SignalExecutionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tw, err = v.RecordChildWorkflowExecutionCompleteAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tw, err = v.ApplyParentClosePolicyAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterStartChildExecutionResponseAttributes_Read(w wire.Value) (*CrossClusterStartChildExecutionResponseAttributes, error) {\n\tvar v CrossClusterStartChildExecutionResponseAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterCancelExecutionResponseAttributes_Read(w wire.Value) (*CrossClusterCancelExecutionResponseAttributes, error) {\n\tvar v CrossClusterCancelExecutionResponseAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterSignalExecutionResponseAttributes_Read(w wire.Value) (*CrossClusterSignalExecutionResponseAttributes, error) {\n\tvar v CrossClusterSignalExecutionResponseAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes_Read(w wire.Value) (*CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes, error) {\n\tvar v CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CrossClusterApplyParentClosePolicyResponseAttributes_Read(w wire.Value) (*CrossClusterApplyParentClosePolicyResponseAttributes, error) {\n\tvar v CrossClusterApplyParentClosePolicyResponseAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a CrossClusterTaskResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CrossClusterTaskResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CrossClusterTaskResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CrossClusterTaskResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CrossClusterTaskType\n\t\t\t\tx, err = _CrossClusterTaskType_Read(field.Value)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskState = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\t\tx, err = _CrossClusterTaskFailedCause_Read(field.Value)\n\t\t\t\tv.FailedCause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartChildExecutionAttributes, err = _CrossClusterStartChildExecutionResponseAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelExecutionAttributes, err = _CrossClusterCancelExecutionResponseAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalExecutionAttributes, err = _CrossClusterSignalExecutionResponseAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RecordChildWorkflowExecutionCompleteAttributes, err = _CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ApplyParentClosePolicyAttributes, err = _CrossClusterApplyParentClosePolicyResponseAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CrossClusterTaskResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CrossClusterTaskResponse struct could not be encoded.\nfunc (v *CrossClusterTaskResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskState != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskState)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedCause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailedCause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartChildExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartChildExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalExecutionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalExecutionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RecordChildWorkflowExecutionCompleteAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ApplyParentClosePolicyAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterStartChildExecutionResponseAttributes_Decode(sr stream.Reader) (*CrossClusterStartChildExecutionResponseAttributes, error) {\n\tvar v CrossClusterStartChildExecutionResponseAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterCancelExecutionResponseAttributes_Decode(sr stream.Reader) (*CrossClusterCancelExecutionResponseAttributes, error) {\n\tvar v CrossClusterCancelExecutionResponseAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterSignalExecutionResponseAttributes_Decode(sr stream.Reader) (*CrossClusterSignalExecutionResponseAttributes, error) {\n\tvar v CrossClusterSignalExecutionResponseAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes_Decode(sr stream.Reader) (*CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes, error) {\n\tvar v CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CrossClusterApplyParentClosePolicyResponseAttributes_Decode(sr stream.Reader) (*CrossClusterApplyParentClosePolicyResponseAttributes, error) {\n\tvar v CrossClusterApplyParentClosePolicyResponseAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a CrossClusterTaskResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CrossClusterTaskResponse struct could not be generated from the wire\n// representation.\nfunc (v *CrossClusterTaskResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x CrossClusterTaskType\n\t\t\tx, err = _CrossClusterTaskType_Decode(sr)\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskState = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x CrossClusterTaskFailedCause\n\t\t\tx, err = _CrossClusterTaskFailedCause_Decode(sr)\n\t\t\tv.FailedCause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.StartChildExecutionAttributes, err = _CrossClusterStartChildExecutionResponseAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.CancelExecutionAttributes, err = _CrossClusterCancelExecutionResponseAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.SignalExecutionAttributes, err = _CrossClusterSignalExecutionResponseAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.RecordChildWorkflowExecutionCompleteAttributes, err = _CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.ApplyParentClosePolicyAttributes, err = _CrossClusterApplyParentClosePolicyResponseAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CrossClusterTaskResponse\n// struct.\nfunc (v *CrossClusterTaskResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.TaskState != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskState: %v\", *(v.TaskState))\n\t\ti++\n\t}\n\tif v.FailedCause != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedCause: %v\", *(v.FailedCause))\n\t\ti++\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartChildExecutionAttributes: %v\", v.StartChildExecutionAttributes)\n\t\ti++\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelExecutionAttributes: %v\", v.CancelExecutionAttributes)\n\t\ti++\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalExecutionAttributes: %v\", v.SignalExecutionAttributes)\n\t\ti++\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RecordChildWorkflowExecutionCompleteAttributes: %v\", v.RecordChildWorkflowExecutionCompleteAttributes)\n\t\ti++\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ApplyParentClosePolicyAttributes: %v\", v.ApplyParentClosePolicyAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"CrossClusterTaskResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this CrossClusterTaskResponse match the\n// provided CrossClusterTaskResponse.\n//\n// This function performs a deep comparison.\nfunc (v *CrossClusterTaskResponse) Equals(rhs *CrossClusterTaskResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !_CrossClusterTaskType_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskState, rhs.TaskState) {\n\t\treturn false\n\t}\n\tif !_CrossClusterTaskFailedCause_EqualsPtr(v.FailedCause, rhs.FailedCause) {\n\t\treturn false\n\t}\n\tif !((v.StartChildExecutionAttributes == nil && rhs.StartChildExecutionAttributes == nil) || (v.StartChildExecutionAttributes != nil && rhs.StartChildExecutionAttributes != nil && v.StartChildExecutionAttributes.Equals(rhs.StartChildExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CancelExecutionAttributes == nil && rhs.CancelExecutionAttributes == nil) || (v.CancelExecutionAttributes != nil && rhs.CancelExecutionAttributes != nil && v.CancelExecutionAttributes.Equals(rhs.CancelExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SignalExecutionAttributes == nil && rhs.SignalExecutionAttributes == nil) || (v.SignalExecutionAttributes != nil && rhs.SignalExecutionAttributes != nil && v.SignalExecutionAttributes.Equals(rhs.SignalExecutionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RecordChildWorkflowExecutionCompleteAttributes == nil && rhs.RecordChildWorkflowExecutionCompleteAttributes == nil) || (v.RecordChildWorkflowExecutionCompleteAttributes != nil && rhs.RecordChildWorkflowExecutionCompleteAttributes != nil && v.RecordChildWorkflowExecutionCompleteAttributes.Equals(rhs.RecordChildWorkflowExecutionCompleteAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ApplyParentClosePolicyAttributes == nil && rhs.ApplyParentClosePolicyAttributes == nil) || (v.ApplyParentClosePolicyAttributes != nil && rhs.ApplyParentClosePolicyAttributes != nil && v.ApplyParentClosePolicyAttributes.Equals(rhs.ApplyParentClosePolicyAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterTaskResponse.\nfunc (v *CrossClusterTaskResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\tif v.TaskType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskType\", *v.TaskType))\n\t}\n\tif v.TaskState != nil {\n\t\tenc.AddInt16(\"taskState\", *v.TaskState)\n\t}\n\tif v.FailedCause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failedCause\", *v.FailedCause))\n\t}\n\tif v.StartChildExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startChildExecutionAttributes\", v.StartChildExecutionAttributes))\n\t}\n\tif v.CancelExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelExecutionAttributes\", v.CancelExecutionAttributes))\n\t}\n\tif v.SignalExecutionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalExecutionAttributes\", v.SignalExecutionAttributes))\n\t}\n\tif v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"recordChildWorkflowExecutionCompleteAttributes\", v.RecordChildWorkflowExecutionCompleteAttributes))\n\t}\n\tif v.ApplyParentClosePolicyAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"applyParentClosePolicyAttributes\", v.ApplyParentClosePolicyAttributes))\n\t}\n\treturn err\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetTaskType() (o CrossClusterTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetTaskState returns the value of TaskState if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetTaskState() (o int16) {\n\tif v != nil && v.TaskState != nil {\n\t\treturn *v.TaskState\n\t}\n\n\treturn\n}\n\n// IsSetTaskState returns true if TaskState is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetTaskState() bool {\n\treturn v != nil && v.TaskState != nil\n}\n\n// GetFailedCause returns the value of FailedCause if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetFailedCause() (o CrossClusterTaskFailedCause) {\n\tif v != nil && v.FailedCause != nil {\n\t\treturn *v.FailedCause\n\t}\n\n\treturn\n}\n\n// IsSetFailedCause returns true if FailedCause is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetFailedCause() bool {\n\treturn v != nil && v.FailedCause != nil\n}\n\n// GetStartChildExecutionAttributes returns the value of StartChildExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetStartChildExecutionAttributes() (o *CrossClusterStartChildExecutionResponseAttributes) {\n\tif v != nil && v.StartChildExecutionAttributes != nil {\n\t\treturn v.StartChildExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartChildExecutionAttributes returns true if StartChildExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetStartChildExecutionAttributes() bool {\n\treturn v != nil && v.StartChildExecutionAttributes != nil\n}\n\n// GetCancelExecutionAttributes returns the value of CancelExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetCancelExecutionAttributes() (o *CrossClusterCancelExecutionResponseAttributes) {\n\tif v != nil && v.CancelExecutionAttributes != nil {\n\t\treturn v.CancelExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCancelExecutionAttributes returns true if CancelExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetCancelExecutionAttributes() bool {\n\treturn v != nil && v.CancelExecutionAttributes != nil\n}\n\n// GetSignalExecutionAttributes returns the value of SignalExecutionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetSignalExecutionAttributes() (o *CrossClusterSignalExecutionResponseAttributes) {\n\tif v != nil && v.SignalExecutionAttributes != nil {\n\t\treturn v.SignalExecutionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSignalExecutionAttributes returns true if SignalExecutionAttributes is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetSignalExecutionAttributes() bool {\n\treturn v != nil && v.SignalExecutionAttributes != nil\n}\n\n// GetRecordChildWorkflowExecutionCompleteAttributes returns the value of RecordChildWorkflowExecutionCompleteAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetRecordChildWorkflowExecutionCompleteAttributes() (o *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) {\n\tif v != nil && v.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\treturn v.RecordChildWorkflowExecutionCompleteAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRecordChildWorkflowExecutionCompleteAttributes returns true if RecordChildWorkflowExecutionCompleteAttributes is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetRecordChildWorkflowExecutionCompleteAttributes() bool {\n\treturn v != nil && v.RecordChildWorkflowExecutionCompleteAttributes != nil\n}\n\n// GetApplyParentClosePolicyAttributes returns the value of ApplyParentClosePolicyAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *CrossClusterTaskResponse) GetApplyParentClosePolicyAttributes() (o *CrossClusterApplyParentClosePolicyResponseAttributes) {\n\tif v != nil && v.ApplyParentClosePolicyAttributes != nil {\n\t\treturn v.ApplyParentClosePolicyAttributes\n\t}\n\n\treturn\n}\n\n// IsSetApplyParentClosePolicyAttributes returns true if ApplyParentClosePolicyAttributes is not nil.\nfunc (v *CrossClusterTaskResponse) IsSetApplyParentClosePolicyAttributes() bool {\n\treturn v != nil && v.ApplyParentClosePolicyAttributes != nil\n}\n\ntype CrossClusterTaskType int32\n\nconst (\n\tCrossClusterTaskTypeStartChildExecution                  CrossClusterTaskType = 0\n\tCrossClusterTaskTypeCancelExecution                      CrossClusterTaskType = 1\n\tCrossClusterTaskTypeSignalExecution                      CrossClusterTaskType = 2\n\tCrossClusterTaskTypeRecordChildWorkflowExecutionComplete CrossClusterTaskType = 3\n\tCrossClusterTaskTypeApplyParentClosePolicy               CrossClusterTaskType = 4\n)\n\n// CrossClusterTaskType_Values returns all recognized values of CrossClusterTaskType.\nfunc CrossClusterTaskType_Values() []CrossClusterTaskType {\n\treturn []CrossClusterTaskType{\n\t\tCrossClusterTaskTypeStartChildExecution,\n\t\tCrossClusterTaskTypeCancelExecution,\n\t\tCrossClusterTaskTypeSignalExecution,\n\t\tCrossClusterTaskTypeRecordChildWorkflowExecutionComplete,\n\t\tCrossClusterTaskTypeApplyParentClosePolicy,\n\t}\n}\n\n// UnmarshalText tries to decode CrossClusterTaskType from a byte slice\n// containing its name.\n//\n//\tvar v CrossClusterTaskType\n//\terr := v.UnmarshalText([]byte(\"StartChildExecution\"))\nfunc (v *CrossClusterTaskType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"StartChildExecution\":\n\t\t*v = CrossClusterTaskTypeStartChildExecution\n\t\treturn nil\n\tcase \"CancelExecution\":\n\t\t*v = CrossClusterTaskTypeCancelExecution\n\t\treturn nil\n\tcase \"SignalExecution\":\n\t\t*v = CrossClusterTaskTypeSignalExecution\n\t\treturn nil\n\tcase \"RecordChildWorkflowExecutionComplete\":\n\t\t*v = CrossClusterTaskTypeRecordChildWorkflowExecutionComplete\n\t\treturn nil\n\tcase \"ApplyParentClosePolicy\":\n\t\t*v = CrossClusterTaskTypeApplyParentClosePolicy\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CrossClusterTaskType\", err)\n\t\t}\n\t\t*v = CrossClusterTaskType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CrossClusterTaskType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v CrossClusterTaskType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"StartChildExecution\"), nil\n\tcase 1:\n\t\treturn []byte(\"CancelExecution\"), nil\n\tcase 2:\n\t\treturn []byte(\"SignalExecution\"), nil\n\tcase 3:\n\t\treturn []byte(\"RecordChildWorkflowExecutionComplete\"), nil\n\tcase 4:\n\t\treturn []byte(\"ApplyParentClosePolicy\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CrossClusterTaskType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v CrossClusterTaskType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"StartChildExecution\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"CancelExecution\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"SignalExecution\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"RecordChildWorkflowExecutionComplete\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"ApplyParentClosePolicy\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v CrossClusterTaskType) Ptr() *CrossClusterTaskType {\n\treturn &v\n}\n\n// Encode encodes CrossClusterTaskType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v CrossClusterTaskType\n//\treturn v.Encode(sWriter)\nfunc (v CrossClusterTaskType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates CrossClusterTaskType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v CrossClusterTaskType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes CrossClusterTaskType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return CrossClusterTaskType(0), err\n//\t}\n//\n//\tvar v CrossClusterTaskType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return CrossClusterTaskType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CrossClusterTaskType) FromWire(w wire.Value) error {\n\t*v = (CrossClusterTaskType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded CrossClusterTaskType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v CrossClusterTaskType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return CrossClusterTaskType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *CrossClusterTaskType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (CrossClusterTaskType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of CrossClusterTaskType.\nfunc (v CrossClusterTaskType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"StartChildExecution\"\n\tcase 1:\n\t\treturn \"CancelExecution\"\n\tcase 2:\n\t\treturn \"SignalExecution\"\n\tcase 3:\n\t\treturn \"RecordChildWorkflowExecutionComplete\"\n\tcase 4:\n\t\treturn \"ApplyParentClosePolicy\"\n\t}\n\treturn fmt.Sprintf(\"CrossClusterTaskType(%d)\", w)\n}\n\n// Equals returns true if this CrossClusterTaskType value matches the provided\n// value.\nfunc (v CrossClusterTaskType) Equals(rhs CrossClusterTaskType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes CrossClusterTaskType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v CrossClusterTaskType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"StartChildExecution\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"CancelExecution\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"SignalExecution\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"RecordChildWorkflowExecutionComplete\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"ApplyParentClosePolicy\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode CrossClusterTaskType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *CrossClusterTaskType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"CrossClusterTaskType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"CrossClusterTaskType\")\n\t\t}\n\t\t*v = (CrossClusterTaskType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"CrossClusterTaskType\")\n\t}\n}\n\ntype CurrentBranchChangedError struct {\n\tMessage            string `json:\"message,required\"`\n\tCurrentBranchToken []byte `json:\"currentBranchToken,required\"`\n}\n\n// ToWire translates a CurrentBranchChangedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *CurrentBranchChangedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\tif v.CurrentBranchToken == nil {\n\t\treturn w, errors.New(\"field CurrentBranchToken of CurrentBranchChangedError is required\")\n\t}\n\tw, err = wire.NewValueBinary(v.CurrentBranchToken), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 20, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a CurrentBranchChangedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a CurrentBranchChangedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v CurrentBranchChangedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *CurrentBranchChangedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\tcurrentBranchTokenIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcurrentBranchTokenIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of CurrentBranchChangedError is required\")\n\t}\n\n\tif !currentBranchTokenIsSet {\n\t\treturn errors.New(\"field CurrentBranchToken of CurrentBranchChangedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a CurrentBranchChangedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a CurrentBranchChangedError struct could not be encoded.\nfunc (v *CurrentBranchChangedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CurrentBranchToken == nil {\n\t\treturn errors.New(\"field CurrentBranchToken of CurrentBranchChangedError is required\")\n\t}\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteBinary(v.CurrentBranchToken); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a CurrentBranchChangedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a CurrentBranchChangedError struct could not be generated from the wire\n// representation.\nfunc (v *CurrentBranchChangedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\tcurrentBranchTokenIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tcurrentBranchTokenIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of CurrentBranchChangedError is required\")\n\t}\n\n\tif !currentBranchTokenIsSet {\n\t\treturn errors.New(\"field CurrentBranchToken of CurrentBranchChangedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a CurrentBranchChangedError\n// struct.\nfunc (v *CurrentBranchChangedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\tfields[i] = fmt.Sprintf(\"CurrentBranchToken: %v\", v.CurrentBranchToken)\n\ti++\n\n\treturn fmt.Sprintf(\"CurrentBranchChangedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*CurrentBranchChangedError) ErrorName() string {\n\treturn \"CurrentBranchChangedError\"\n}\n\n// Equals returns true if all the fields of this CurrentBranchChangedError match the\n// provided CurrentBranchChangedError.\n//\n// This function performs a deep comparison.\nfunc (v *CurrentBranchChangedError) Equals(rhs *CurrentBranchChangedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\tif !bytes.Equal(v.CurrentBranchToken, rhs.CurrentBranchToken) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of CurrentBranchChangedError.\nfunc (v *CurrentBranchChangedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\tenc.AddString(\"currentBranchToken\", base64.StdEncoding.EncodeToString(v.CurrentBranchToken))\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *CurrentBranchChangedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\n// GetCurrentBranchToken returns the value of CurrentBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *CurrentBranchChangedError) GetCurrentBranchToken() (o []byte) {\n\tif v != nil {\n\t\to = v.CurrentBranchToken\n\t}\n\treturn\n}\n\n// IsSetCurrentBranchToken returns true if CurrentBranchToken is not nil.\nfunc (v *CurrentBranchChangedError) IsSetCurrentBranchToken() bool {\n\treturn v != nil && v.CurrentBranchToken != nil\n}\n\nfunc (v *CurrentBranchChangedError) Error() string {\n\treturn v.String()\n}\n\ntype DataBlob struct {\n\tEncodingType *EncodingType `json:\"EncodingType,omitempty\"`\n\tData         []byte        `json:\"Data,omitempty\"`\n}\n\n// ToWire translates a DataBlob struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DataBlob) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.EncodingType != nil {\n\t\tw, err = v.EncodingType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tw, err = wire.NewValueBinary(v.Data), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _EncodingType_Read(w wire.Value) (EncodingType, error) {\n\tvar v EncodingType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DataBlob struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DataBlob struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DataBlob\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DataBlob) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x EncodingType\n\t\t\t\tx, err = _EncodingType_Read(field.Value)\n\t\t\t\tv.EncodingType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Data, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DataBlob struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DataBlob struct could not be encoded.\nfunc (v *DataBlob) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EncodingType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EncodingType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Data); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _EncodingType_Decode(sr stream.Reader) (EncodingType, error) {\n\tvar v EncodingType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DataBlob struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DataBlob struct could not be generated from the wire\n// representation.\nfunc (v *DataBlob) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x EncodingType\n\t\t\tx, err = _EncodingType_Decode(sr)\n\t\t\tv.EncodingType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Data, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DataBlob\n// struct.\nfunc (v *DataBlob) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.EncodingType != nil {\n\t\tfields[i] = fmt.Sprintf(\"EncodingType: %v\", *(v.EncodingType))\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DataBlob{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _EncodingType_EqualsPtr(lhs, rhs *EncodingType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DataBlob match the\n// provided DataBlob.\n//\n// This function performs a deep comparison.\nfunc (v *DataBlob) Equals(rhs *DataBlob) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_EncodingType_EqualsPtr(v.EncodingType, rhs.EncodingType) {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && bytes.Equal(v.Data, rhs.Data))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DataBlob.\nfunc (v *DataBlob) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.EncodingType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"EncodingType\", *v.EncodingType))\n\t}\n\tif v.Data != nil {\n\t\tenc.AddString(\"Data\", base64.StdEncoding.EncodeToString(v.Data))\n\t}\n\treturn err\n}\n\n// GetEncodingType returns the value of EncodingType if it is set or its\n// zero value if it is unset.\nfunc (v *DataBlob) GetEncodingType() (o EncodingType) {\n\tif v != nil && v.EncodingType != nil {\n\t\treturn *v.EncodingType\n\t}\n\n\treturn\n}\n\n// IsSetEncodingType returns true if EncodingType is not nil.\nfunc (v *DataBlob) IsSetEncodingType() bool {\n\treturn v != nil && v.EncodingType != nil\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *DataBlob) GetData() (o []byte) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *DataBlob) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\ntype Decision struct {\n\tDecisionType                                             *DecisionType                                             `json:\"decisionType,omitempty\"`\n\tScheduleActivityTaskDecisionAttributes                   *ScheduleActivityTaskDecisionAttributes                   `json:\"scheduleActivityTaskDecisionAttributes,omitempty\"`\n\tStartTimerDecisionAttributes                             *StartTimerDecisionAttributes                             `json:\"startTimerDecisionAttributes,omitempty\"`\n\tCompleteWorkflowExecutionDecisionAttributes              *CompleteWorkflowExecutionDecisionAttributes              `json:\"completeWorkflowExecutionDecisionAttributes,omitempty\"`\n\tFailWorkflowExecutionDecisionAttributes                  *FailWorkflowExecutionDecisionAttributes                  `json:\"failWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRequestCancelActivityTaskDecisionAttributes              *RequestCancelActivityTaskDecisionAttributes              `json:\"requestCancelActivityTaskDecisionAttributes,omitempty\"`\n\tCancelTimerDecisionAttributes                            *CancelTimerDecisionAttributes                            `json:\"cancelTimerDecisionAttributes,omitempty\"`\n\tCancelWorkflowExecutionDecisionAttributes                *CancelWorkflowExecutionDecisionAttributes                `json:\"cancelWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionDecisionAttributes *RequestCancelExternalWorkflowExecutionDecisionAttributes `json:\"requestCancelExternalWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRecordMarkerDecisionAttributes                           *RecordMarkerDecisionAttributes                           `json:\"recordMarkerDecisionAttributes,omitempty\"`\n\tContinueAsNewWorkflowExecutionDecisionAttributes         *ContinueAsNewWorkflowExecutionDecisionAttributes         `json:\"continueAsNewWorkflowExecutionDecisionAttributes,omitempty\"`\n\tStartChildWorkflowExecutionDecisionAttributes            *StartChildWorkflowExecutionDecisionAttributes            `json:\"startChildWorkflowExecutionDecisionAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionDecisionAttributes        *SignalExternalWorkflowExecutionDecisionAttributes        `json:\"signalExternalWorkflowExecutionDecisionAttributes,omitempty\"`\n\tUpsertWorkflowSearchAttributesDecisionAttributes         *UpsertWorkflowSearchAttributesDecisionAttributes         `json:\"upsertWorkflowSearchAttributesDecisionAttributes,omitempty\"`\n}\n\n// ToWire translates a Decision struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Decision) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [14]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionType != nil {\n\t\tw, err = v.DecisionType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleActivityTaskDecisionAttributes != nil {\n\t\tw, err = v.ScheduleActivityTaskDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartTimerDecisionAttributes != nil {\n\t\tw, err = v.StartTimerDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 25, Value: w}\n\t\ti++\n\t}\n\tif v.CompleteWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.CompleteWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.FailWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.RequestCancelActivityTaskDecisionAttributes != nil {\n\t\tw, err = v.RequestCancelActivityTaskDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.CancelTimerDecisionAttributes != nil {\n\t\tw, err = v.CancelTimerDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.CancelWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.CancelWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.RequestCancelExternalWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.RecordMarkerDecisionAttributes != nil {\n\t\tw, err = v.RecordMarkerDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.ContinueAsNewWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.StartChildWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tw, err = v.SignalExternalWorkflowExecutionDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.UpsertWorkflowSearchAttributesDecisionAttributes != nil {\n\t\tw, err = v.UpsertWorkflowSearchAttributesDecisionAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DecisionType_Read(w wire.Value) (DecisionType, error) {\n\tvar v DecisionType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _ScheduleActivityTaskDecisionAttributes_Read(w wire.Value) (*ScheduleActivityTaskDecisionAttributes, error) {\n\tvar v ScheduleActivityTaskDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _StartTimerDecisionAttributes_Read(w wire.Value) (*StartTimerDecisionAttributes, error) {\n\tvar v StartTimerDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CompleteWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*CompleteWorkflowExecutionDecisionAttributes, error) {\n\tvar v CompleteWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _FailWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*FailWorkflowExecutionDecisionAttributes, error) {\n\tvar v FailWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RequestCancelActivityTaskDecisionAttributes_Read(w wire.Value) (*RequestCancelActivityTaskDecisionAttributes, error) {\n\tvar v RequestCancelActivityTaskDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CancelTimerDecisionAttributes_Read(w wire.Value) (*CancelTimerDecisionAttributes, error) {\n\tvar v CancelTimerDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CancelWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*CancelWorkflowExecutionDecisionAttributes, error) {\n\tvar v CancelWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*RequestCancelExternalWorkflowExecutionDecisionAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RecordMarkerDecisionAttributes_Read(w wire.Value) (*RecordMarkerDecisionAttributes, error) {\n\tvar v RecordMarkerDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ContinueAsNewWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*ContinueAsNewWorkflowExecutionDecisionAttributes, error) {\n\tvar v ContinueAsNewWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _StartChildWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*StartChildWorkflowExecutionDecisionAttributes, error) {\n\tvar v StartChildWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionDecisionAttributes_Read(w wire.Value) (*SignalExternalWorkflowExecutionDecisionAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _UpsertWorkflowSearchAttributesDecisionAttributes_Read(w wire.Value) (*UpsertWorkflowSearchAttributesDecisionAttributes, error) {\n\tvar v UpsertWorkflowSearchAttributesDecisionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a Decision struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Decision struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Decision\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Decision) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DecisionType\n\t\t\t\tx, err = _DecisionType_Read(field.Value)\n\t\t\t\tv.DecisionType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ScheduleActivityTaskDecisionAttributes, err = _ScheduleActivityTaskDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 25:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartTimerDecisionAttributes, err = _StartTimerDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CompleteWorkflowExecutionDecisionAttributes, err = _CompleteWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailWorkflowExecutionDecisionAttributes, err = _FailWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RequestCancelActivityTaskDecisionAttributes, err = _RequestCancelActivityTaskDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelTimerDecisionAttributes, err = _CancelTimerDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelWorkflowExecutionDecisionAttributes, err = _CancelWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RequestCancelExternalWorkflowExecutionDecisionAttributes, err = _RequestCancelExternalWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RecordMarkerDecisionAttributes, err = _RecordMarkerDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ContinueAsNewWorkflowExecutionDecisionAttributes, err = _ContinueAsNewWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartChildWorkflowExecutionDecisionAttributes, err = _StartChildWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalExternalWorkflowExecutionDecisionAttributes, err = _SignalExternalWorkflowExecutionDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.UpsertWorkflowSearchAttributesDecisionAttributes, err = _UpsertWorkflowSearchAttributesDecisionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Decision struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Decision struct could not be encoded.\nfunc (v *Decision) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleActivityTaskDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ScheduleActivityTaskDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartTimerDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 25, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartTimerDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompleteWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompleteWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestCancelActivityTaskDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RequestCancelActivityTaskDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelTimerDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelTimerDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RequestCancelExternalWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RecordMarkerDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RecordMarkerDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ContinueAsNewWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartChildWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartChildWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalExternalWorkflowExecutionDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UpsertWorkflowSearchAttributesDecisionAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.UpsertWorkflowSearchAttributesDecisionAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DecisionType_Decode(sr stream.Reader) (DecisionType, error) {\n\tvar v DecisionType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _ScheduleActivityTaskDecisionAttributes_Decode(sr stream.Reader) (*ScheduleActivityTaskDecisionAttributes, error) {\n\tvar v ScheduleActivityTaskDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _StartTimerDecisionAttributes_Decode(sr stream.Reader) (*StartTimerDecisionAttributes, error) {\n\tvar v StartTimerDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CompleteWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*CompleteWorkflowExecutionDecisionAttributes, error) {\n\tvar v CompleteWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _FailWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*FailWorkflowExecutionDecisionAttributes, error) {\n\tvar v FailWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RequestCancelActivityTaskDecisionAttributes_Decode(sr stream.Reader) (*RequestCancelActivityTaskDecisionAttributes, error) {\n\tvar v RequestCancelActivityTaskDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CancelTimerDecisionAttributes_Decode(sr stream.Reader) (*CancelTimerDecisionAttributes, error) {\n\tvar v CancelTimerDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CancelWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*CancelWorkflowExecutionDecisionAttributes, error) {\n\tvar v CancelWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*RequestCancelExternalWorkflowExecutionDecisionAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RecordMarkerDecisionAttributes_Decode(sr stream.Reader) (*RecordMarkerDecisionAttributes, error) {\n\tvar v RecordMarkerDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ContinueAsNewWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*ContinueAsNewWorkflowExecutionDecisionAttributes, error) {\n\tvar v ContinueAsNewWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _StartChildWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*StartChildWorkflowExecutionDecisionAttributes, error) {\n\tvar v StartChildWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionDecisionAttributes_Decode(sr stream.Reader) (*SignalExternalWorkflowExecutionDecisionAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _UpsertWorkflowSearchAttributesDecisionAttributes_Decode(sr stream.Reader) (*UpsertWorkflowSearchAttributesDecisionAttributes, error) {\n\tvar v UpsertWorkflowSearchAttributesDecisionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a Decision struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Decision struct could not be generated from the wire\n// representation.\nfunc (v *Decision) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x DecisionType\n\t\t\tx, err = _DecisionType_Decode(sr)\n\t\t\tv.DecisionType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ScheduleActivityTaskDecisionAttributes, err = _ScheduleActivityTaskDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 25 && fh.Type == wire.TStruct:\n\t\t\tv.StartTimerDecisionAttributes, err = _StartTimerDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.CompleteWorkflowExecutionDecisionAttributes, err = _CompleteWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TStruct:\n\t\t\tv.FailWorkflowExecutionDecisionAttributes, err = _FailWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.RequestCancelActivityTaskDecisionAttributes, err = _RequestCancelActivityTaskDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.CancelTimerDecisionAttributes, err = _CancelTimerDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.CancelWorkflowExecutionDecisionAttributes, err = _CancelWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.RequestCancelExternalWorkflowExecutionDecisionAttributes, err = _RequestCancelExternalWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.RecordMarkerDecisionAttributes, err = _RecordMarkerDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.ContinueAsNewWorkflowExecutionDecisionAttributes, err = _ContinueAsNewWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TStruct:\n\t\t\tv.StartChildWorkflowExecutionDecisionAttributes, err = _StartChildWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TStruct:\n\t\t\tv.SignalExternalWorkflowExecutionDecisionAttributes, err = _SignalExternalWorkflowExecutionDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.UpsertWorkflowSearchAttributesDecisionAttributes, err = _UpsertWorkflowSearchAttributesDecisionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Decision\n// struct.\nfunc (v *Decision) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [14]string\n\ti := 0\n\tif v.DecisionType != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionType: %v\", *(v.DecisionType))\n\t\ti++\n\t}\n\tif v.ScheduleActivityTaskDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleActivityTaskDecisionAttributes: %v\", v.ScheduleActivityTaskDecisionAttributes)\n\t\ti++\n\t}\n\tif v.StartTimerDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartTimerDecisionAttributes: %v\", v.StartTimerDecisionAttributes)\n\t\ti++\n\t}\n\tif v.CompleteWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompleteWorkflowExecutionDecisionAttributes: %v\", v.CompleteWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.FailWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailWorkflowExecutionDecisionAttributes: %v\", v.FailWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.RequestCancelActivityTaskDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestCancelActivityTaskDecisionAttributes: %v\", v.RequestCancelActivityTaskDecisionAttributes)\n\t\ti++\n\t}\n\tif v.CancelTimerDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelTimerDecisionAttributes: %v\", v.CancelTimerDecisionAttributes)\n\t\ti++\n\t}\n\tif v.CancelWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelWorkflowExecutionDecisionAttributes: %v\", v.CancelWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionDecisionAttributes: %v\", v.RequestCancelExternalWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.RecordMarkerDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RecordMarkerDecisionAttributes: %v\", v.RecordMarkerDecisionAttributes)\n\t\ti++\n\t}\n\tif v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinueAsNewWorkflowExecutionDecisionAttributes: %v\", v.ContinueAsNewWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartChildWorkflowExecutionDecisionAttributes: %v\", v.StartChildWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalExternalWorkflowExecutionDecisionAttributes: %v\", v.SignalExternalWorkflowExecutionDecisionAttributes)\n\t\ti++\n\t}\n\tif v.UpsertWorkflowSearchAttributesDecisionAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpsertWorkflowSearchAttributesDecisionAttributes: %v\", v.UpsertWorkflowSearchAttributesDecisionAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Decision{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DecisionType_EqualsPtr(lhs, rhs *DecisionType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this Decision match the\n// provided Decision.\n//\n// This function performs a deep comparison.\nfunc (v *Decision) Equals(rhs *Decision) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_DecisionType_EqualsPtr(v.DecisionType, rhs.DecisionType) {\n\t\treturn false\n\t}\n\tif !((v.ScheduleActivityTaskDecisionAttributes == nil && rhs.ScheduleActivityTaskDecisionAttributes == nil) || (v.ScheduleActivityTaskDecisionAttributes != nil && rhs.ScheduleActivityTaskDecisionAttributes != nil && v.ScheduleActivityTaskDecisionAttributes.Equals(rhs.ScheduleActivityTaskDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.StartTimerDecisionAttributes == nil && rhs.StartTimerDecisionAttributes == nil) || (v.StartTimerDecisionAttributes != nil && rhs.StartTimerDecisionAttributes != nil && v.StartTimerDecisionAttributes.Equals(rhs.StartTimerDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CompleteWorkflowExecutionDecisionAttributes == nil && rhs.CompleteWorkflowExecutionDecisionAttributes == nil) || (v.CompleteWorkflowExecutionDecisionAttributes != nil && rhs.CompleteWorkflowExecutionDecisionAttributes != nil && v.CompleteWorkflowExecutionDecisionAttributes.Equals(rhs.CompleteWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.FailWorkflowExecutionDecisionAttributes == nil && rhs.FailWorkflowExecutionDecisionAttributes == nil) || (v.FailWorkflowExecutionDecisionAttributes != nil && rhs.FailWorkflowExecutionDecisionAttributes != nil && v.FailWorkflowExecutionDecisionAttributes.Equals(rhs.FailWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RequestCancelActivityTaskDecisionAttributes == nil && rhs.RequestCancelActivityTaskDecisionAttributes == nil) || (v.RequestCancelActivityTaskDecisionAttributes != nil && rhs.RequestCancelActivityTaskDecisionAttributes != nil && v.RequestCancelActivityTaskDecisionAttributes.Equals(rhs.RequestCancelActivityTaskDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CancelTimerDecisionAttributes == nil && rhs.CancelTimerDecisionAttributes == nil) || (v.CancelTimerDecisionAttributes != nil && rhs.CancelTimerDecisionAttributes != nil && v.CancelTimerDecisionAttributes.Equals(rhs.CancelTimerDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CancelWorkflowExecutionDecisionAttributes == nil && rhs.CancelWorkflowExecutionDecisionAttributes == nil) || (v.CancelWorkflowExecutionDecisionAttributes != nil && rhs.CancelWorkflowExecutionDecisionAttributes != nil && v.CancelWorkflowExecutionDecisionAttributes.Equals(rhs.CancelWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RequestCancelExternalWorkflowExecutionDecisionAttributes == nil && rhs.RequestCancelExternalWorkflowExecutionDecisionAttributes == nil) || (v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil && rhs.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil && v.RequestCancelExternalWorkflowExecutionDecisionAttributes.Equals(rhs.RequestCancelExternalWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RecordMarkerDecisionAttributes == nil && rhs.RecordMarkerDecisionAttributes == nil) || (v.RecordMarkerDecisionAttributes != nil && rhs.RecordMarkerDecisionAttributes != nil && v.RecordMarkerDecisionAttributes.Equals(rhs.RecordMarkerDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ContinueAsNewWorkflowExecutionDecisionAttributes == nil && rhs.ContinueAsNewWorkflowExecutionDecisionAttributes == nil) || (v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil && rhs.ContinueAsNewWorkflowExecutionDecisionAttributes != nil && v.ContinueAsNewWorkflowExecutionDecisionAttributes.Equals(rhs.ContinueAsNewWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.StartChildWorkflowExecutionDecisionAttributes == nil && rhs.StartChildWorkflowExecutionDecisionAttributes == nil) || (v.StartChildWorkflowExecutionDecisionAttributes != nil && rhs.StartChildWorkflowExecutionDecisionAttributes != nil && v.StartChildWorkflowExecutionDecisionAttributes.Equals(rhs.StartChildWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SignalExternalWorkflowExecutionDecisionAttributes == nil && rhs.SignalExternalWorkflowExecutionDecisionAttributes == nil) || (v.SignalExternalWorkflowExecutionDecisionAttributes != nil && rhs.SignalExternalWorkflowExecutionDecisionAttributes != nil && v.SignalExternalWorkflowExecutionDecisionAttributes.Equals(rhs.SignalExternalWorkflowExecutionDecisionAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.UpsertWorkflowSearchAttributesDecisionAttributes == nil && rhs.UpsertWorkflowSearchAttributesDecisionAttributes == nil) || (v.UpsertWorkflowSearchAttributesDecisionAttributes != nil && rhs.UpsertWorkflowSearchAttributesDecisionAttributes != nil && v.UpsertWorkflowSearchAttributesDecisionAttributes.Equals(rhs.UpsertWorkflowSearchAttributesDecisionAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Decision.\nfunc (v *Decision) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionType\", *v.DecisionType))\n\t}\n\tif v.ScheduleActivityTaskDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"scheduleActivityTaskDecisionAttributes\", v.ScheduleActivityTaskDecisionAttributes))\n\t}\n\tif v.StartTimerDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startTimerDecisionAttributes\", v.StartTimerDecisionAttributes))\n\t}\n\tif v.CompleteWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completeWorkflowExecutionDecisionAttributes\", v.CompleteWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.FailWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failWorkflowExecutionDecisionAttributes\", v.FailWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.RequestCancelActivityTaskDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"requestCancelActivityTaskDecisionAttributes\", v.RequestCancelActivityTaskDecisionAttributes))\n\t}\n\tif v.CancelTimerDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelTimerDecisionAttributes\", v.CancelTimerDecisionAttributes))\n\t}\n\tif v.CancelWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelWorkflowExecutionDecisionAttributes\", v.CancelWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"requestCancelExternalWorkflowExecutionDecisionAttributes\", v.RequestCancelExternalWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.RecordMarkerDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"recordMarkerDecisionAttributes\", v.RecordMarkerDecisionAttributes))\n\t}\n\tif v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"continueAsNewWorkflowExecutionDecisionAttributes\", v.ContinueAsNewWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.StartChildWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startChildWorkflowExecutionDecisionAttributes\", v.StartChildWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.SignalExternalWorkflowExecutionDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalExternalWorkflowExecutionDecisionAttributes\", v.SignalExternalWorkflowExecutionDecisionAttributes))\n\t}\n\tif v.UpsertWorkflowSearchAttributesDecisionAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"upsertWorkflowSearchAttributesDecisionAttributes\", v.UpsertWorkflowSearchAttributesDecisionAttributes))\n\t}\n\treturn err\n}\n\n// GetDecisionType returns the value of DecisionType if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetDecisionType() (o DecisionType) {\n\tif v != nil && v.DecisionType != nil {\n\t\treturn *v.DecisionType\n\t}\n\n\treturn\n}\n\n// IsSetDecisionType returns true if DecisionType is not nil.\nfunc (v *Decision) IsSetDecisionType() bool {\n\treturn v != nil && v.DecisionType != nil\n}\n\n// GetScheduleActivityTaskDecisionAttributes returns the value of ScheduleActivityTaskDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetScheduleActivityTaskDecisionAttributes() (o *ScheduleActivityTaskDecisionAttributes) {\n\tif v != nil && v.ScheduleActivityTaskDecisionAttributes != nil {\n\t\treturn v.ScheduleActivityTaskDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetScheduleActivityTaskDecisionAttributes returns true if ScheduleActivityTaskDecisionAttributes is not nil.\nfunc (v *Decision) IsSetScheduleActivityTaskDecisionAttributes() bool {\n\treturn v != nil && v.ScheduleActivityTaskDecisionAttributes != nil\n}\n\n// GetStartTimerDecisionAttributes returns the value of StartTimerDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetStartTimerDecisionAttributes() (o *StartTimerDecisionAttributes) {\n\tif v != nil && v.StartTimerDecisionAttributes != nil {\n\t\treturn v.StartTimerDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartTimerDecisionAttributes returns true if StartTimerDecisionAttributes is not nil.\nfunc (v *Decision) IsSetStartTimerDecisionAttributes() bool {\n\treturn v != nil && v.StartTimerDecisionAttributes != nil\n}\n\n// GetCompleteWorkflowExecutionDecisionAttributes returns the value of CompleteWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetCompleteWorkflowExecutionDecisionAttributes() (o *CompleteWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.CompleteWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.CompleteWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCompleteWorkflowExecutionDecisionAttributes returns true if CompleteWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetCompleteWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.CompleteWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetFailWorkflowExecutionDecisionAttributes returns the value of FailWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetFailWorkflowExecutionDecisionAttributes() (o *FailWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.FailWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.FailWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetFailWorkflowExecutionDecisionAttributes returns true if FailWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetFailWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.FailWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetRequestCancelActivityTaskDecisionAttributes returns the value of RequestCancelActivityTaskDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetRequestCancelActivityTaskDecisionAttributes() (o *RequestCancelActivityTaskDecisionAttributes) {\n\tif v != nil && v.RequestCancelActivityTaskDecisionAttributes != nil {\n\t\treturn v.RequestCancelActivityTaskDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRequestCancelActivityTaskDecisionAttributes returns true if RequestCancelActivityTaskDecisionAttributes is not nil.\nfunc (v *Decision) IsSetRequestCancelActivityTaskDecisionAttributes() bool {\n\treturn v != nil && v.RequestCancelActivityTaskDecisionAttributes != nil\n}\n\n// GetCancelTimerDecisionAttributes returns the value of CancelTimerDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetCancelTimerDecisionAttributes() (o *CancelTimerDecisionAttributes) {\n\tif v != nil && v.CancelTimerDecisionAttributes != nil {\n\t\treturn v.CancelTimerDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCancelTimerDecisionAttributes returns true if CancelTimerDecisionAttributes is not nil.\nfunc (v *Decision) IsSetCancelTimerDecisionAttributes() bool {\n\treturn v != nil && v.CancelTimerDecisionAttributes != nil\n}\n\n// GetCancelWorkflowExecutionDecisionAttributes returns the value of CancelWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetCancelWorkflowExecutionDecisionAttributes() (o *CancelWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.CancelWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.CancelWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCancelWorkflowExecutionDecisionAttributes returns true if CancelWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetCancelWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.CancelWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetRequestCancelExternalWorkflowExecutionDecisionAttributes returns the value of RequestCancelExternalWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetRequestCancelExternalWorkflowExecutionDecisionAttributes() (o *RequestCancelExternalWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.RequestCancelExternalWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRequestCancelExternalWorkflowExecutionDecisionAttributes returns true if RequestCancelExternalWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetRequestCancelExternalWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.RequestCancelExternalWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetRecordMarkerDecisionAttributes returns the value of RecordMarkerDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetRecordMarkerDecisionAttributes() (o *RecordMarkerDecisionAttributes) {\n\tif v != nil && v.RecordMarkerDecisionAttributes != nil {\n\t\treturn v.RecordMarkerDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRecordMarkerDecisionAttributes returns true if RecordMarkerDecisionAttributes is not nil.\nfunc (v *Decision) IsSetRecordMarkerDecisionAttributes() bool {\n\treturn v != nil && v.RecordMarkerDecisionAttributes != nil\n}\n\n// GetContinueAsNewWorkflowExecutionDecisionAttributes returns the value of ContinueAsNewWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetContinueAsNewWorkflowExecutionDecisionAttributes() (o *ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.ContinueAsNewWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetContinueAsNewWorkflowExecutionDecisionAttributes returns true if ContinueAsNewWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetContinueAsNewWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.ContinueAsNewWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetStartChildWorkflowExecutionDecisionAttributes returns the value of StartChildWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetStartChildWorkflowExecutionDecisionAttributes() (o *StartChildWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.StartChildWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.StartChildWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartChildWorkflowExecutionDecisionAttributes returns true if StartChildWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetStartChildWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.StartChildWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetSignalExternalWorkflowExecutionDecisionAttributes returns the value of SignalExternalWorkflowExecutionDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetSignalExternalWorkflowExecutionDecisionAttributes() (o *SignalExternalWorkflowExecutionDecisionAttributes) {\n\tif v != nil && v.SignalExternalWorkflowExecutionDecisionAttributes != nil {\n\t\treturn v.SignalExternalWorkflowExecutionDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSignalExternalWorkflowExecutionDecisionAttributes returns true if SignalExternalWorkflowExecutionDecisionAttributes is not nil.\nfunc (v *Decision) IsSetSignalExternalWorkflowExecutionDecisionAttributes() bool {\n\treturn v != nil && v.SignalExternalWorkflowExecutionDecisionAttributes != nil\n}\n\n// GetUpsertWorkflowSearchAttributesDecisionAttributes returns the value of UpsertWorkflowSearchAttributesDecisionAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Decision) GetUpsertWorkflowSearchAttributesDecisionAttributes() (o *UpsertWorkflowSearchAttributesDecisionAttributes) {\n\tif v != nil && v.UpsertWorkflowSearchAttributesDecisionAttributes != nil {\n\t\treturn v.UpsertWorkflowSearchAttributesDecisionAttributes\n\t}\n\n\treturn\n}\n\n// IsSetUpsertWorkflowSearchAttributesDecisionAttributes returns true if UpsertWorkflowSearchAttributesDecisionAttributes is not nil.\nfunc (v *Decision) IsSetUpsertWorkflowSearchAttributesDecisionAttributes() bool {\n\treturn v != nil && v.UpsertWorkflowSearchAttributesDecisionAttributes != nil\n}\n\ntype DecisionTaskCompletedEventAttributes struct {\n\tExecutionContext []byte  `json:\"executionContext,omitempty\"`\n\tScheduledEventId *int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId   *int64  `json:\"startedEventId,omitempty\"`\n\tIdentity         *string `json:\"identity,omitempty\"`\n\tBinaryChecksum   *string `json:\"binaryChecksum,omitempty\"`\n}\n\n// ToWire translates a DecisionTaskCompletedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DecisionTaskCompletedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ExecutionContext != nil {\n\t\tw, err = wire.NewValueBinary(v.ExecutionContext), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DecisionTaskCompletedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DecisionTaskCompletedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DecisionTaskCompletedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DecisionTaskCompletedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ExecutionContext, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DecisionTaskCompletedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DecisionTaskCompletedEventAttributes struct could not be encoded.\nfunc (v *DecisionTaskCompletedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ExecutionContext != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ExecutionContext); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DecisionTaskCompletedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DecisionTaskCompletedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DecisionTaskCompletedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.ExecutionContext, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DecisionTaskCompletedEventAttributes\n// struct.\nfunc (v *DecisionTaskCompletedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.ExecutionContext != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionContext: %v\", v.ExecutionContext)\n\t\ti++\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DecisionTaskCompletedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DecisionTaskCompletedEventAttributes match the\n// provided DecisionTaskCompletedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DecisionTaskCompletedEventAttributes) Equals(rhs *DecisionTaskCompletedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ExecutionContext == nil && rhs.ExecutionContext == nil) || (v.ExecutionContext != nil && rhs.ExecutionContext != nil && bytes.Equal(v.ExecutionContext, rhs.ExecutionContext))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskCompletedEventAttributes.\nfunc (v *DecisionTaskCompletedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ExecutionContext != nil {\n\t\tenc.AddString(\"executionContext\", base64.StdEncoding.EncodeToString(v.ExecutionContext))\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\treturn err\n}\n\n// GetExecutionContext returns the value of ExecutionContext if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskCompletedEventAttributes) GetExecutionContext() (o []byte) {\n\tif v != nil && v.ExecutionContext != nil {\n\t\treturn v.ExecutionContext\n\t}\n\n\treturn\n}\n\n// IsSetExecutionContext returns true if ExecutionContext is not nil.\nfunc (v *DecisionTaskCompletedEventAttributes) IsSetExecutionContext() bool {\n\treturn v != nil && v.ExecutionContext != nil\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskCompletedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *DecisionTaskCompletedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskCompletedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *DecisionTaskCompletedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskCompletedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *DecisionTaskCompletedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskCompletedEventAttributes) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *DecisionTaskCompletedEventAttributes) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\ntype DecisionTaskFailedCause int32\n\nconst (\n\tDecisionTaskFailedCauseUnhandledDecision                                   DecisionTaskFailedCause = 0\n\tDecisionTaskFailedCauseBadScheduleActivityAttributes                       DecisionTaskFailedCause = 1\n\tDecisionTaskFailedCauseBadRequestCancelActivityAttributes                  DecisionTaskFailedCause = 2\n\tDecisionTaskFailedCauseBadStartTimerAttributes                             DecisionTaskFailedCause = 3\n\tDecisionTaskFailedCauseBadCancelTimerAttributes                            DecisionTaskFailedCause = 4\n\tDecisionTaskFailedCauseBadRecordMarkerAttributes                           DecisionTaskFailedCause = 5\n\tDecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes              DecisionTaskFailedCause = 6\n\tDecisionTaskFailedCauseBadFailWorkflowExecutionAttributes                  DecisionTaskFailedCause = 7\n\tDecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes                DecisionTaskFailedCause = 8\n\tDecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes DecisionTaskFailedCause = 9\n\tDecisionTaskFailedCauseBadContinueAsNewAttributes                          DecisionTaskFailedCause = 10\n\tDecisionTaskFailedCauseStartTimerDuplicateID                               DecisionTaskFailedCause = 11\n\tDecisionTaskFailedCauseResetStickyTasklist                                 DecisionTaskFailedCause = 12\n\tDecisionTaskFailedCauseWorkflowWorkerUnhandledFailure                      DecisionTaskFailedCause = 13\n\tDecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes                DecisionTaskFailedCause = 14\n\tDecisionTaskFailedCauseBadStartChildExecutionAttributes                    DecisionTaskFailedCause = 15\n\tDecisionTaskFailedCauseForceCloseDecision                                  DecisionTaskFailedCause = 16\n\tDecisionTaskFailedCauseFailoverCloseDecision                               DecisionTaskFailedCause = 17\n\tDecisionTaskFailedCauseBadSignalInputSize                                  DecisionTaskFailedCause = 18\n\tDecisionTaskFailedCauseResetWorkflow                                       DecisionTaskFailedCause = 19\n\tDecisionTaskFailedCauseBadBinary                                           DecisionTaskFailedCause = 20\n\tDecisionTaskFailedCauseScheduleActivityDuplicateID                         DecisionTaskFailedCause = 21\n\tDecisionTaskFailedCauseBadSearchAttributes                                 DecisionTaskFailedCause = 22\n)\n\n// DecisionTaskFailedCause_Values returns all recognized values of DecisionTaskFailedCause.\nfunc DecisionTaskFailedCause_Values() []DecisionTaskFailedCause {\n\treturn []DecisionTaskFailedCause{\n\t\tDecisionTaskFailedCauseUnhandledDecision,\n\t\tDecisionTaskFailedCauseBadScheduleActivityAttributes,\n\t\tDecisionTaskFailedCauseBadRequestCancelActivityAttributes,\n\t\tDecisionTaskFailedCauseBadStartTimerAttributes,\n\t\tDecisionTaskFailedCauseBadCancelTimerAttributes,\n\t\tDecisionTaskFailedCauseBadRecordMarkerAttributes,\n\t\tDecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes,\n\t\tDecisionTaskFailedCauseBadFailWorkflowExecutionAttributes,\n\t\tDecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes,\n\t\tDecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes,\n\t\tDecisionTaskFailedCauseBadContinueAsNewAttributes,\n\t\tDecisionTaskFailedCauseStartTimerDuplicateID,\n\t\tDecisionTaskFailedCauseResetStickyTasklist,\n\t\tDecisionTaskFailedCauseWorkflowWorkerUnhandledFailure,\n\t\tDecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes,\n\t\tDecisionTaskFailedCauseBadStartChildExecutionAttributes,\n\t\tDecisionTaskFailedCauseForceCloseDecision,\n\t\tDecisionTaskFailedCauseFailoverCloseDecision,\n\t\tDecisionTaskFailedCauseBadSignalInputSize,\n\t\tDecisionTaskFailedCauseResetWorkflow,\n\t\tDecisionTaskFailedCauseBadBinary,\n\t\tDecisionTaskFailedCauseScheduleActivityDuplicateID,\n\t\tDecisionTaskFailedCauseBadSearchAttributes,\n\t}\n}\n\n// UnmarshalText tries to decode DecisionTaskFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v DecisionTaskFailedCause\n//\terr := v.UnmarshalText([]byte(\"UNHANDLED_DECISION\"))\nfunc (v *DecisionTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"UNHANDLED_DECISION\":\n\t\t*v = DecisionTaskFailedCauseUnhandledDecision\n\t\treturn nil\n\tcase \"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadScheduleActivityAttributes\n\t\treturn nil\n\tcase \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadRequestCancelActivityAttributes\n\t\treturn nil\n\tcase \"BAD_START_TIMER_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadStartTimerAttributes\n\t\treturn nil\n\tcase \"BAD_CANCEL_TIMER_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadCancelTimerAttributes\n\t\treturn nil\n\tcase \"BAD_RECORD_MARKER_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadRecordMarkerAttributes\n\t\treturn nil\n\tcase \"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_CONTINUE_AS_NEW_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadContinueAsNewAttributes\n\t\treturn nil\n\tcase \"START_TIMER_DUPLICATE_ID\":\n\t\t*v = DecisionTaskFailedCauseStartTimerDuplicateID\n\t\treturn nil\n\tcase \"RESET_STICKY_TASKLIST\":\n\t\t*v = DecisionTaskFailedCauseResetStickyTasklist\n\t\treturn nil\n\tcase \"WORKFLOW_WORKER_UNHANDLED_FAILURE\":\n\t\t*v = DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t\treturn nil\n\tcase \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadStartChildExecutionAttributes\n\t\treturn nil\n\tcase \"FORCE_CLOSE_DECISION\":\n\t\t*v = DecisionTaskFailedCauseForceCloseDecision\n\t\treturn nil\n\tcase \"FAILOVER_CLOSE_DECISION\":\n\t\t*v = DecisionTaskFailedCauseFailoverCloseDecision\n\t\treturn nil\n\tcase \"BAD_SIGNAL_INPUT_SIZE\":\n\t\t*v = DecisionTaskFailedCauseBadSignalInputSize\n\t\treturn nil\n\tcase \"RESET_WORKFLOW\":\n\t\t*v = DecisionTaskFailedCauseResetWorkflow\n\t\treturn nil\n\tcase \"BAD_BINARY\":\n\t\t*v = DecisionTaskFailedCauseBadBinary\n\t\treturn nil\n\tcase \"SCHEDULE_ACTIVITY_DUPLICATE_ID\":\n\t\t*v = DecisionTaskFailedCauseScheduleActivityDuplicateID\n\t\treturn nil\n\tcase \"BAD_SEARCH_ATTRIBUTES\":\n\t\t*v = DecisionTaskFailedCauseBadSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionTaskFailedCause\", err)\n\t\t}\n\t\t*v = DecisionTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionTaskFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DecisionTaskFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"UNHANDLED_DECISION\"), nil\n\tcase 1:\n\t\treturn []byte(\"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\"), nil\n\tcase 2:\n\t\treturn []byte(\"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\"), nil\n\tcase 3:\n\t\treturn []byte(\"BAD_START_TIMER_ATTRIBUTES\"), nil\n\tcase 4:\n\t\treturn []byte(\"BAD_CANCEL_TIMER_ATTRIBUTES\"), nil\n\tcase 5:\n\t\treturn []byte(\"BAD_RECORD_MARKER_ATTRIBUTES\"), nil\n\tcase 6:\n\t\treturn []byte(\"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\"), nil\n\tcase 7:\n\t\treturn []byte(\"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\"), nil\n\tcase 8:\n\t\treturn []byte(\"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\"), nil\n\tcase 9:\n\t\treturn []byte(\"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"), nil\n\tcase 10:\n\t\treturn []byte(\"BAD_CONTINUE_AS_NEW_ATTRIBUTES\"), nil\n\tcase 11:\n\t\treturn []byte(\"START_TIMER_DUPLICATE_ID\"), nil\n\tcase 12:\n\t\treturn []byte(\"RESET_STICKY_TASKLIST\"), nil\n\tcase 13:\n\t\treturn []byte(\"WORKFLOW_WORKER_UNHANDLED_FAILURE\"), nil\n\tcase 14:\n\t\treturn []byte(\"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"), nil\n\tcase 15:\n\t\treturn []byte(\"BAD_START_CHILD_EXECUTION_ATTRIBUTES\"), nil\n\tcase 16:\n\t\treturn []byte(\"FORCE_CLOSE_DECISION\"), nil\n\tcase 17:\n\t\treturn []byte(\"FAILOVER_CLOSE_DECISION\"), nil\n\tcase 18:\n\t\treturn []byte(\"BAD_SIGNAL_INPUT_SIZE\"), nil\n\tcase 19:\n\t\treturn []byte(\"RESET_WORKFLOW\"), nil\n\tcase 20:\n\t\treturn []byte(\"BAD_BINARY\"), nil\n\tcase 21:\n\t\treturn []byte(\"SCHEDULE_ACTIVITY_DUPLICATE_ID\"), nil\n\tcase 22:\n\t\treturn []byte(\"BAD_SEARCH_ATTRIBUTES\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DecisionTaskFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"UNHANDLED_DECISION\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"BAD_START_TIMER_ATTRIBUTES\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"BAD_CANCEL_TIMER_ATTRIBUTES\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"BAD_RECORD_MARKER_ATTRIBUTES\")\n\tcase 6:\n\t\tenc.AddString(\"name\", \"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\")\n\tcase 7:\n\t\tenc.AddString(\"name\", \"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\")\n\tcase 8:\n\t\tenc.AddString(\"name\", \"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\")\n\tcase 9:\n\t\tenc.AddString(\"name\", \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\")\n\tcase 10:\n\t\tenc.AddString(\"name\", \"BAD_CONTINUE_AS_NEW_ATTRIBUTES\")\n\tcase 11:\n\t\tenc.AddString(\"name\", \"START_TIMER_DUPLICATE_ID\")\n\tcase 12:\n\t\tenc.AddString(\"name\", \"RESET_STICKY_TASKLIST\")\n\tcase 13:\n\t\tenc.AddString(\"name\", \"WORKFLOW_WORKER_UNHANDLED_FAILURE\")\n\tcase 14:\n\t\tenc.AddString(\"name\", \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\")\n\tcase 15:\n\t\tenc.AddString(\"name\", \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\")\n\tcase 16:\n\t\tenc.AddString(\"name\", \"FORCE_CLOSE_DECISION\")\n\tcase 17:\n\t\tenc.AddString(\"name\", \"FAILOVER_CLOSE_DECISION\")\n\tcase 18:\n\t\tenc.AddString(\"name\", \"BAD_SIGNAL_INPUT_SIZE\")\n\tcase 19:\n\t\tenc.AddString(\"name\", \"RESET_WORKFLOW\")\n\tcase 20:\n\t\tenc.AddString(\"name\", \"BAD_BINARY\")\n\tcase 21:\n\t\tenc.AddString(\"name\", \"SCHEDULE_ACTIVITY_DUPLICATE_ID\")\n\tcase 22:\n\t\tenc.AddString(\"name\", \"BAD_SEARCH_ATTRIBUTES\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DecisionTaskFailedCause) Ptr() *DecisionTaskFailedCause {\n\treturn &v\n}\n\n// Encode encodes DecisionTaskFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DecisionTaskFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v DecisionTaskFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DecisionTaskFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DecisionTaskFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DecisionTaskFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DecisionTaskFailedCause(0), err\n//\t}\n//\n//\tvar v DecisionTaskFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DecisionTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionTaskFailedCause) FromWire(w wire.Value) error {\n\t*v = (DecisionTaskFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DecisionTaskFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DecisionTaskFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DecisionTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionTaskFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DecisionTaskFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DecisionTaskFailedCause.\nfunc (v DecisionTaskFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNHANDLED_DECISION\"\n\tcase 1:\n\t\treturn \"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\"\n\tcase 2:\n\t\treturn \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\"\n\tcase 3:\n\t\treturn \"BAD_START_TIMER_ATTRIBUTES\"\n\tcase 4:\n\t\treturn \"BAD_CANCEL_TIMER_ATTRIBUTES\"\n\tcase 5:\n\t\treturn \"BAD_RECORD_MARKER_ATTRIBUTES\"\n\tcase 6:\n\t\treturn \"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 7:\n\t\treturn \"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 8:\n\t\treturn \"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 9:\n\t\treturn \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 10:\n\t\treturn \"BAD_CONTINUE_AS_NEW_ATTRIBUTES\"\n\tcase 11:\n\t\treturn \"START_TIMER_DUPLICATE_ID\"\n\tcase 12:\n\t\treturn \"RESET_STICKY_TASKLIST\"\n\tcase 13:\n\t\treturn \"WORKFLOW_WORKER_UNHANDLED_FAILURE\"\n\tcase 14:\n\t\treturn \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 15:\n\t\treturn \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\"\n\tcase 16:\n\t\treturn \"FORCE_CLOSE_DECISION\"\n\tcase 17:\n\t\treturn \"FAILOVER_CLOSE_DECISION\"\n\tcase 18:\n\t\treturn \"BAD_SIGNAL_INPUT_SIZE\"\n\tcase 19:\n\t\treturn \"RESET_WORKFLOW\"\n\tcase 20:\n\t\treturn \"BAD_BINARY\"\n\tcase 21:\n\t\treturn \"SCHEDULE_ACTIVITY_DUPLICATE_ID\"\n\tcase 22:\n\t\treturn \"BAD_SEARCH_ATTRIBUTES\"\n\t}\n\treturn fmt.Sprintf(\"DecisionTaskFailedCause(%d)\", w)\n}\n\n// Equals returns true if this DecisionTaskFailedCause value matches the provided\n// value.\nfunc (v DecisionTaskFailedCause) Equals(rhs DecisionTaskFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DecisionTaskFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DecisionTaskFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"UNHANDLED_DECISION\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"BAD_START_TIMER_ATTRIBUTES\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"BAD_CANCEL_TIMER_ATTRIBUTES\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"BAD_RECORD_MARKER_ATTRIBUTES\\\"\"), nil\n\tcase 6:\n\t\treturn ([]byte)(\"\\\"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 7:\n\t\treturn ([]byte)(\"\\\"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 8:\n\t\treturn ([]byte)(\"\\\"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 9:\n\t\treturn ([]byte)(\"\\\"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 10:\n\t\treturn ([]byte)(\"\\\"BAD_CONTINUE_AS_NEW_ATTRIBUTES\\\"\"), nil\n\tcase 11:\n\t\treturn ([]byte)(\"\\\"START_TIMER_DUPLICATE_ID\\\"\"), nil\n\tcase 12:\n\t\treturn ([]byte)(\"\\\"RESET_STICKY_TASKLIST\\\"\"), nil\n\tcase 13:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_WORKER_UNHANDLED_FAILURE\\\"\"), nil\n\tcase 14:\n\t\treturn ([]byte)(\"\\\"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 15:\n\t\treturn ([]byte)(\"\\\"BAD_START_CHILD_EXECUTION_ATTRIBUTES\\\"\"), nil\n\tcase 16:\n\t\treturn ([]byte)(\"\\\"FORCE_CLOSE_DECISION\\\"\"), nil\n\tcase 17:\n\t\treturn ([]byte)(\"\\\"FAILOVER_CLOSE_DECISION\\\"\"), nil\n\tcase 18:\n\t\treturn ([]byte)(\"\\\"BAD_SIGNAL_INPUT_SIZE\\\"\"), nil\n\tcase 19:\n\t\treturn ([]byte)(\"\\\"RESET_WORKFLOW\\\"\"), nil\n\tcase 20:\n\t\treturn ([]byte)(\"\\\"BAD_BINARY\\\"\"), nil\n\tcase 21:\n\t\treturn ([]byte)(\"\\\"SCHEDULE_ACTIVITY_DUPLICATE_ID\\\"\"), nil\n\tcase 22:\n\t\treturn ([]byte)(\"\\\"BAD_SEARCH_ATTRIBUTES\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DecisionTaskFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DecisionTaskFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DecisionTaskFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DecisionTaskFailedCause\")\n\t\t}\n\t\t*v = (DecisionTaskFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DecisionTaskFailedCause\")\n\t}\n}\n\ntype DecisionTaskFailedEventAttributes struct {\n\tScheduledEventId *int64                   `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId   *int64                   `json:\"startedEventId,omitempty\"`\n\tCause            *DecisionTaskFailedCause `json:\"cause,omitempty\"`\n\tDetails          []byte                   `json:\"details,omitempty\"`\n\tIdentity         *string                  `json:\"identity,omitempty\"`\n\tReason           *string                  `json:\"reason,omitempty\"`\n\tBaseRunId        *string                  `json:\"baseRunId,omitempty\"`\n\tNewRunId         *string                  `json:\"newRunId,omitempty\"`\n\tForkEventVersion *int64                   `json:\"forkEventVersion,omitempty\"`\n\tBinaryChecksum   *string                  `json:\"binaryChecksum,omitempty\"`\n\tRequestId        *string                  `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a DecisionTaskFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DecisionTaskFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [11]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.BaseRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.BaseRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.NewRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.ForkEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DecisionTaskFailedCause_Read(w wire.Value) (DecisionTaskFailedCause, error) {\n\tvar v DecisionTaskFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DecisionTaskFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DecisionTaskFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DecisionTaskFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DecisionTaskFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DecisionTaskFailedCause\n\t\t\t\tx, err = _DecisionTaskFailedCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BaseRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.NewRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ForkEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DecisionTaskFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DecisionTaskFailedEventAttributes struct could not be encoded.\nfunc (v *DecisionTaskFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BaseRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BaseRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.NewRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForkEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ForkEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DecisionTaskFailedCause_Decode(sr stream.Reader) (DecisionTaskFailedCause, error) {\n\tvar v DecisionTaskFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DecisionTaskFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DecisionTaskFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DecisionTaskFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x DecisionTaskFailedCause\n\t\t\tx, err = _DecisionTaskFailedCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BaseRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.NewRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ForkEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DecisionTaskFailedEventAttributes\n// struct.\nfunc (v *DecisionTaskFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [11]string\n\ti := 0\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.BaseRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"BaseRunId: %v\", *(v.BaseRunId))\n\t\ti++\n\t}\n\tif v.NewRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunId: %v\", *(v.NewRunId))\n\t\ti++\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForkEventVersion: %v\", *(v.ForkEventVersion))\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DecisionTaskFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DecisionTaskFailedCause_EqualsPtr(lhs, rhs *DecisionTaskFailedCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DecisionTaskFailedEventAttributes match the\n// provided DecisionTaskFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DecisionTaskFailedEventAttributes) Equals(rhs *DecisionTaskFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_DecisionTaskFailedCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BaseRunId, rhs.BaseRunId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.NewRunId, rhs.NewRunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ForkEventVersion, rhs.ForkEventVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskFailedEventAttributes.\nfunc (v *DecisionTaskFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.BaseRunId != nil {\n\t\tenc.AddString(\"baseRunId\", *v.BaseRunId)\n\t}\n\tif v.NewRunId != nil {\n\t\tenc.AddString(\"newRunId\", *v.NewRunId)\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tenc.AddInt64(\"forkEventVersion\", *v.ForkEventVersion)\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetCause() (o DecisionTaskFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetBaseRunId returns the value of BaseRunId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetBaseRunId() (o string) {\n\tif v != nil && v.BaseRunId != nil {\n\t\treturn *v.BaseRunId\n\t}\n\n\treturn\n}\n\n// IsSetBaseRunId returns true if BaseRunId is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetBaseRunId() bool {\n\treturn v != nil && v.BaseRunId != nil\n}\n\n// GetNewRunId returns the value of NewRunId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetNewRunId() (o string) {\n\tif v != nil && v.NewRunId != nil {\n\t\treturn *v.NewRunId\n\t}\n\n\treturn\n}\n\n// IsSetNewRunId returns true if NewRunId is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetNewRunId() bool {\n\treturn v != nil && v.NewRunId != nil\n}\n\n// GetForkEventVersion returns the value of ForkEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetForkEventVersion() (o int64) {\n\tif v != nil && v.ForkEventVersion != nil {\n\t\treturn *v.ForkEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetForkEventVersion returns true if ForkEventVersion is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetForkEventVersion() bool {\n\treturn v != nil && v.ForkEventVersion != nil\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskFailedEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *DecisionTaskFailedEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype DecisionTaskScheduledEventAttributes struct {\n\tTaskList                   *TaskList `json:\"taskList,omitempty\"`\n\tStartToCloseTimeoutSeconds *int32    `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tAttempt                    *int64    `json:\"attempt,omitempty\"`\n}\n\n// ToWire translates a DecisionTaskScheduledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DecisionTaskScheduledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.StartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DecisionTaskScheduledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DecisionTaskScheduledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DecisionTaskScheduledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DecisionTaskScheduledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DecisionTaskScheduledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DecisionTaskScheduledEventAttributes struct could not be encoded.\nfunc (v *DecisionTaskScheduledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DecisionTaskScheduledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DecisionTaskScheduledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DecisionTaskScheduledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DecisionTaskScheduledEventAttributes\n// struct.\nfunc (v *DecisionTaskScheduledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToCloseTimeoutSeconds: %v\", *(v.StartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DecisionTaskScheduledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DecisionTaskScheduledEventAttributes match the\n// provided DecisionTaskScheduledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DecisionTaskScheduledEventAttributes) Equals(rhs *DecisionTaskScheduledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StartToCloseTimeoutSeconds, rhs.StartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskScheduledEventAttributes.\nfunc (v *DecisionTaskScheduledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"startToCloseTimeoutSeconds\", *v.StartToCloseTimeoutSeconds)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\treturn err\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskScheduledEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *DecisionTaskScheduledEventAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetStartToCloseTimeoutSeconds returns the value of StartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskScheduledEventAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToCloseTimeoutSeconds returns true if StartToCloseTimeoutSeconds is not nil.\nfunc (v *DecisionTaskScheduledEventAttributes) IsSetStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.StartToCloseTimeoutSeconds != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskScheduledEventAttributes) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *DecisionTaskScheduledEventAttributes) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\ntype DecisionTaskStartedEventAttributes struct {\n\tScheduledEventId *int64  `json:\"scheduledEventId,omitempty\"`\n\tIdentity         *string `json:\"identity,omitempty\"`\n\tRequestId        *string `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a DecisionTaskStartedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DecisionTaskStartedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DecisionTaskStartedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DecisionTaskStartedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DecisionTaskStartedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DecisionTaskStartedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DecisionTaskStartedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DecisionTaskStartedEventAttributes struct could not be encoded.\nfunc (v *DecisionTaskStartedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DecisionTaskStartedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DecisionTaskStartedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DecisionTaskStartedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DecisionTaskStartedEventAttributes\n// struct.\nfunc (v *DecisionTaskStartedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DecisionTaskStartedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DecisionTaskStartedEventAttributes match the\n// provided DecisionTaskStartedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DecisionTaskStartedEventAttributes) Equals(rhs *DecisionTaskStartedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskStartedEventAttributes.\nfunc (v *DecisionTaskStartedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskStartedEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *DecisionTaskStartedEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskStartedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *DecisionTaskStartedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskStartedEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *DecisionTaskStartedEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype DecisionTaskTimedOutCause int32\n\nconst (\n\tDecisionTaskTimedOutCauseTimeout DecisionTaskTimedOutCause = 0\n\tDecisionTaskTimedOutCauseReset   DecisionTaskTimedOutCause = 1\n)\n\n// DecisionTaskTimedOutCause_Values returns all recognized values of DecisionTaskTimedOutCause.\nfunc DecisionTaskTimedOutCause_Values() []DecisionTaskTimedOutCause {\n\treturn []DecisionTaskTimedOutCause{\n\t\tDecisionTaskTimedOutCauseTimeout,\n\t\tDecisionTaskTimedOutCauseReset,\n\t}\n}\n\n// UnmarshalText tries to decode DecisionTaskTimedOutCause from a byte slice\n// containing its name.\n//\n//\tvar v DecisionTaskTimedOutCause\n//\terr := v.UnmarshalText([]byte(\"TIMEOUT\"))\nfunc (v *DecisionTaskTimedOutCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"TIMEOUT\":\n\t\t*v = DecisionTaskTimedOutCauseTimeout\n\t\treturn nil\n\tcase \"RESET\":\n\t\t*v = DecisionTaskTimedOutCauseReset\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionTaskTimedOutCause\", err)\n\t\t}\n\t\t*v = DecisionTaskTimedOutCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionTaskTimedOutCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DecisionTaskTimedOutCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"TIMEOUT\"), nil\n\tcase 1:\n\t\treturn []byte(\"RESET\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskTimedOutCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DecisionTaskTimedOutCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"TIMEOUT\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"RESET\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DecisionTaskTimedOutCause) Ptr() *DecisionTaskTimedOutCause {\n\treturn &v\n}\n\n// Encode encodes DecisionTaskTimedOutCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DecisionTaskTimedOutCause\n//\treturn v.Encode(sWriter)\nfunc (v DecisionTaskTimedOutCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DecisionTaskTimedOutCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DecisionTaskTimedOutCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DecisionTaskTimedOutCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DecisionTaskTimedOutCause(0), err\n//\t}\n//\n//\tvar v DecisionTaskTimedOutCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DecisionTaskTimedOutCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionTaskTimedOutCause) FromWire(w wire.Value) error {\n\t*v = (DecisionTaskTimedOutCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DecisionTaskTimedOutCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DecisionTaskTimedOutCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DecisionTaskTimedOutCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionTaskTimedOutCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DecisionTaskTimedOutCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DecisionTaskTimedOutCause.\nfunc (v DecisionTaskTimedOutCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"TIMEOUT\"\n\tcase 1:\n\t\treturn \"RESET\"\n\t}\n\treturn fmt.Sprintf(\"DecisionTaskTimedOutCause(%d)\", w)\n}\n\n// Equals returns true if this DecisionTaskTimedOutCause value matches the provided\n// value.\nfunc (v DecisionTaskTimedOutCause) Equals(rhs DecisionTaskTimedOutCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DecisionTaskTimedOutCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DecisionTaskTimedOutCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"TIMEOUT\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"RESET\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DecisionTaskTimedOutCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DecisionTaskTimedOutCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DecisionTaskTimedOutCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DecisionTaskTimedOutCause\")\n\t\t}\n\t\t*v = (DecisionTaskTimedOutCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DecisionTaskTimedOutCause\")\n\t}\n}\n\ntype DecisionTaskTimedOutEventAttributes struct {\n\tScheduledEventId *int64                     `json:\"scheduledEventId,omitempty\"`\n\tStartedEventId   *int64                     `json:\"startedEventId,omitempty\"`\n\tTimeoutType      *TimeoutType               `json:\"timeoutType,omitempty\"`\n\tBaseRunId        *string                    `json:\"baseRunId,omitempty\"`\n\tNewRunId         *string                    `json:\"newRunId,omitempty\"`\n\tForkEventVersion *int64                     `json:\"forkEventVersion,omitempty\"`\n\tReason           *string                    `json:\"reason,omitempty\"`\n\tCause            *DecisionTaskTimedOutCause `json:\"cause,omitempty\"`\n\tRequestId        *string                    `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a DecisionTaskTimedOutEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DecisionTaskTimedOutEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tw, err = v.TimeoutType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.BaseRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.BaseRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.NewRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.ForkEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DecisionTaskTimedOutCause_Read(w wire.Value) (DecisionTaskTimedOutCause, error) {\n\tvar v DecisionTaskTimedOutCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DecisionTaskTimedOutEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DecisionTaskTimedOutEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DecisionTaskTimedOutEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DecisionTaskTimedOutEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TimeoutType\n\t\t\t\tx, err = _TimeoutType_Read(field.Value)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BaseRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.NewRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ForkEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DecisionTaskTimedOutCause\n\t\t\t\tx, err = _DecisionTaskTimedOutCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DecisionTaskTimedOutEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DecisionTaskTimedOutEventAttributes struct could not be encoded.\nfunc (v *DecisionTaskTimedOutEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimeoutType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BaseRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BaseRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.NewRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForkEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ForkEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DecisionTaskTimedOutCause_Decode(sr stream.Reader) (DecisionTaskTimedOutCause, error) {\n\tvar v DecisionTaskTimedOutCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DecisionTaskTimedOutEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DecisionTaskTimedOutEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DecisionTaskTimedOutEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x TimeoutType\n\t\t\tx, err = _TimeoutType_Decode(sr)\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BaseRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.NewRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ForkEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x DecisionTaskTimedOutCause\n\t\t\tx, err = _DecisionTaskTimedOutCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DecisionTaskTimedOutEventAttributes\n// struct.\nfunc (v *DecisionTaskTimedOutEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.ScheduledEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventId: %v\", *(v.ScheduledEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\tif v.BaseRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"BaseRunId: %v\", *(v.BaseRunId))\n\t\ti++\n\t}\n\tif v.NewRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunId: %v\", *(v.NewRunId))\n\t\ti++\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForkEventVersion: %v\", *(v.ForkEventVersion))\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DecisionTaskTimedOutEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DecisionTaskTimedOutCause_EqualsPtr(lhs, rhs *DecisionTaskTimedOutCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DecisionTaskTimedOutEventAttributes match the\n// provided DecisionTaskTimedOutEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DecisionTaskTimedOutEventAttributes) Equals(rhs *DecisionTaskTimedOutEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventId, rhs.ScheduledEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_TimeoutType_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BaseRunId, rhs.BaseRunId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.NewRunId, rhs.NewRunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ForkEventVersion, rhs.ForkEventVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !_DecisionTaskTimedOutCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionTaskTimedOutEventAttributes.\nfunc (v *DecisionTaskTimedOutEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEventId != nil {\n\t\tenc.AddInt64(\"scheduledEventId\", *v.ScheduledEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.TimeoutType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timeoutType\", *v.TimeoutType))\n\t}\n\tif v.BaseRunId != nil {\n\t\tenc.AddString(\"baseRunId\", *v.BaseRunId)\n\t}\n\tif v.NewRunId != nil {\n\t\tenc.AddString(\"newRunId\", *v.NewRunId)\n\t}\n\tif v.ForkEventVersion != nil {\n\t\tenc.AddInt64(\"forkEventVersion\", *v.ForkEventVersion)\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetScheduledEventId returns the value of ScheduledEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetScheduledEventId() (o int64) {\n\tif v != nil && v.ScheduledEventId != nil {\n\t\treturn *v.ScheduledEventId\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventId returns true if ScheduledEventId is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetScheduledEventId() bool {\n\treturn v != nil && v.ScheduledEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\n// GetBaseRunId returns the value of BaseRunId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetBaseRunId() (o string) {\n\tif v != nil && v.BaseRunId != nil {\n\t\treturn *v.BaseRunId\n\t}\n\n\treturn\n}\n\n// IsSetBaseRunId returns true if BaseRunId is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetBaseRunId() bool {\n\treturn v != nil && v.BaseRunId != nil\n}\n\n// GetNewRunId returns the value of NewRunId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetNewRunId() (o string) {\n\tif v != nil && v.NewRunId != nil {\n\t\treturn *v.NewRunId\n\t}\n\n\treturn\n}\n\n// IsSetNewRunId returns true if NewRunId is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetNewRunId() bool {\n\treturn v != nil && v.NewRunId != nil\n}\n\n// GetForkEventVersion returns the value of ForkEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetForkEventVersion() (o int64) {\n\tif v != nil && v.ForkEventVersion != nil {\n\t\treturn *v.ForkEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetForkEventVersion returns true if ForkEventVersion is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetForkEventVersion() bool {\n\treturn v != nil && v.ForkEventVersion != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetCause() (o DecisionTaskTimedOutCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *DecisionTaskTimedOutEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *DecisionTaskTimedOutEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype DecisionType int32\n\nconst (\n\tDecisionTypeScheduleActivityTask                   DecisionType = 0\n\tDecisionTypeRequestCancelActivityTask              DecisionType = 1\n\tDecisionTypeStartTimer                             DecisionType = 2\n\tDecisionTypeCompleteWorkflowExecution              DecisionType = 3\n\tDecisionTypeFailWorkflowExecution                  DecisionType = 4\n\tDecisionTypeCancelTimer                            DecisionType = 5\n\tDecisionTypeCancelWorkflowExecution                DecisionType = 6\n\tDecisionTypeRequestCancelExternalWorkflowExecution DecisionType = 7\n\tDecisionTypeRecordMarker                           DecisionType = 8\n\tDecisionTypeContinueAsNewWorkflowExecution         DecisionType = 9\n\tDecisionTypeStartChildWorkflowExecution            DecisionType = 10\n\tDecisionTypeSignalExternalWorkflowExecution        DecisionType = 11\n\tDecisionTypeUpsertWorkflowSearchAttributes         DecisionType = 12\n)\n\n// DecisionType_Values returns all recognized values of DecisionType.\nfunc DecisionType_Values() []DecisionType {\n\treturn []DecisionType{\n\t\tDecisionTypeScheduleActivityTask,\n\t\tDecisionTypeRequestCancelActivityTask,\n\t\tDecisionTypeStartTimer,\n\t\tDecisionTypeCompleteWorkflowExecution,\n\t\tDecisionTypeFailWorkflowExecution,\n\t\tDecisionTypeCancelTimer,\n\t\tDecisionTypeCancelWorkflowExecution,\n\t\tDecisionTypeRequestCancelExternalWorkflowExecution,\n\t\tDecisionTypeRecordMarker,\n\t\tDecisionTypeContinueAsNewWorkflowExecution,\n\t\tDecisionTypeStartChildWorkflowExecution,\n\t\tDecisionTypeSignalExternalWorkflowExecution,\n\t\tDecisionTypeUpsertWorkflowSearchAttributes,\n\t}\n}\n\n// UnmarshalText tries to decode DecisionType from a byte slice\n// containing its name.\n//\n//\tvar v DecisionType\n//\terr := v.UnmarshalText([]byte(\"ScheduleActivityTask\"))\nfunc (v *DecisionType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"ScheduleActivityTask\":\n\t\t*v = DecisionTypeScheduleActivityTask\n\t\treturn nil\n\tcase \"RequestCancelActivityTask\":\n\t\t*v = DecisionTypeRequestCancelActivityTask\n\t\treturn nil\n\tcase \"StartTimer\":\n\t\t*v = DecisionTypeStartTimer\n\t\treturn nil\n\tcase \"CompleteWorkflowExecution\":\n\t\t*v = DecisionTypeCompleteWorkflowExecution\n\t\treturn nil\n\tcase \"FailWorkflowExecution\":\n\t\t*v = DecisionTypeFailWorkflowExecution\n\t\treturn nil\n\tcase \"CancelTimer\":\n\t\t*v = DecisionTypeCancelTimer\n\t\treturn nil\n\tcase \"CancelWorkflowExecution\":\n\t\t*v = DecisionTypeCancelWorkflowExecution\n\t\treturn nil\n\tcase \"RequestCancelExternalWorkflowExecution\":\n\t\t*v = DecisionTypeRequestCancelExternalWorkflowExecution\n\t\treturn nil\n\tcase \"RecordMarker\":\n\t\t*v = DecisionTypeRecordMarker\n\t\treturn nil\n\tcase \"ContinueAsNewWorkflowExecution\":\n\t\t*v = DecisionTypeContinueAsNewWorkflowExecution\n\t\treturn nil\n\tcase \"StartChildWorkflowExecution\":\n\t\t*v = DecisionTypeStartChildWorkflowExecution\n\t\treturn nil\n\tcase \"SignalExternalWorkflowExecution\":\n\t\t*v = DecisionTypeSignalExternalWorkflowExecution\n\t\treturn nil\n\tcase \"UpsertWorkflowSearchAttributes\":\n\t\t*v = DecisionTypeUpsertWorkflowSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionType\", err)\n\t\t}\n\t\t*v = DecisionType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DecisionType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"ScheduleActivityTask\"), nil\n\tcase 1:\n\t\treturn []byte(\"RequestCancelActivityTask\"), nil\n\tcase 2:\n\t\treturn []byte(\"StartTimer\"), nil\n\tcase 3:\n\t\treturn []byte(\"CompleteWorkflowExecution\"), nil\n\tcase 4:\n\t\treturn []byte(\"FailWorkflowExecution\"), nil\n\tcase 5:\n\t\treturn []byte(\"CancelTimer\"), nil\n\tcase 6:\n\t\treturn []byte(\"CancelWorkflowExecution\"), nil\n\tcase 7:\n\t\treturn []byte(\"RequestCancelExternalWorkflowExecution\"), nil\n\tcase 8:\n\t\treturn []byte(\"RecordMarker\"), nil\n\tcase 9:\n\t\treturn []byte(\"ContinueAsNewWorkflowExecution\"), nil\n\tcase 10:\n\t\treturn []byte(\"StartChildWorkflowExecution\"), nil\n\tcase 11:\n\t\treturn []byte(\"SignalExternalWorkflowExecution\"), nil\n\tcase 12:\n\t\treturn []byte(\"UpsertWorkflowSearchAttributes\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DecisionType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DecisionType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"ScheduleActivityTask\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"RequestCancelActivityTask\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"StartTimer\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"CompleteWorkflowExecution\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"FailWorkflowExecution\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"CancelTimer\")\n\tcase 6:\n\t\tenc.AddString(\"name\", \"CancelWorkflowExecution\")\n\tcase 7:\n\t\tenc.AddString(\"name\", \"RequestCancelExternalWorkflowExecution\")\n\tcase 8:\n\t\tenc.AddString(\"name\", \"RecordMarker\")\n\tcase 9:\n\t\tenc.AddString(\"name\", \"ContinueAsNewWorkflowExecution\")\n\tcase 10:\n\t\tenc.AddString(\"name\", \"StartChildWorkflowExecution\")\n\tcase 11:\n\t\tenc.AddString(\"name\", \"SignalExternalWorkflowExecution\")\n\tcase 12:\n\t\tenc.AddString(\"name\", \"UpsertWorkflowSearchAttributes\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DecisionType) Ptr() *DecisionType {\n\treturn &v\n}\n\n// Encode encodes DecisionType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DecisionType\n//\treturn v.Encode(sWriter)\nfunc (v DecisionType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DecisionType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DecisionType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DecisionType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DecisionType(0), err\n//\t}\n//\n//\tvar v DecisionType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DecisionType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionType) FromWire(w wire.Value) error {\n\t*v = (DecisionType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DecisionType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DecisionType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DecisionType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DecisionType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DecisionType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DecisionType.\nfunc (v DecisionType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ScheduleActivityTask\"\n\tcase 1:\n\t\treturn \"RequestCancelActivityTask\"\n\tcase 2:\n\t\treturn \"StartTimer\"\n\tcase 3:\n\t\treturn \"CompleteWorkflowExecution\"\n\tcase 4:\n\t\treturn \"FailWorkflowExecution\"\n\tcase 5:\n\t\treturn \"CancelTimer\"\n\tcase 6:\n\t\treturn \"CancelWorkflowExecution\"\n\tcase 7:\n\t\treturn \"RequestCancelExternalWorkflowExecution\"\n\tcase 8:\n\t\treturn \"RecordMarker\"\n\tcase 9:\n\t\treturn \"ContinueAsNewWorkflowExecution\"\n\tcase 10:\n\t\treturn \"StartChildWorkflowExecution\"\n\tcase 11:\n\t\treturn \"SignalExternalWorkflowExecution\"\n\tcase 12:\n\t\treturn \"UpsertWorkflowSearchAttributes\"\n\t}\n\treturn fmt.Sprintf(\"DecisionType(%d)\", w)\n}\n\n// Equals returns true if this DecisionType value matches the provided\n// value.\nfunc (v DecisionType) Equals(rhs DecisionType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DecisionType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DecisionType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"ScheduleActivityTask\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"RequestCancelActivityTask\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"StartTimer\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"CompleteWorkflowExecution\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"FailWorkflowExecution\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"CancelTimer\\\"\"), nil\n\tcase 6:\n\t\treturn ([]byte)(\"\\\"CancelWorkflowExecution\\\"\"), nil\n\tcase 7:\n\t\treturn ([]byte)(\"\\\"RequestCancelExternalWorkflowExecution\\\"\"), nil\n\tcase 8:\n\t\treturn ([]byte)(\"\\\"RecordMarker\\\"\"), nil\n\tcase 9:\n\t\treturn ([]byte)(\"\\\"ContinueAsNewWorkflowExecution\\\"\"), nil\n\tcase 10:\n\t\treturn ([]byte)(\"\\\"StartChildWorkflowExecution\\\"\"), nil\n\tcase 11:\n\t\treturn ([]byte)(\"\\\"SignalExternalWorkflowExecution\\\"\"), nil\n\tcase 12:\n\t\treturn ([]byte)(\"\\\"UpsertWorkflowSearchAttributes\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DecisionType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DecisionType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DecisionType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DecisionType\")\n\t\t}\n\t\t*v = (DecisionType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DecisionType\")\n\t}\n}\n\ntype DeleteDomainRequest struct {\n\tName          *string `json:\"name,omitempty\"`\n\tSecurityToken *string `json:\"securityToken,omitempty\"`\n}\n\n// ToWire translates a DeleteDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DeleteDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tw, err = wire.NewValueString(*(v.SecurityToken)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DeleteDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DeleteDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DeleteDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DeleteDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SecurityToken = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DeleteDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DeleteDomainRequest struct could not be encoded.\nfunc (v *DeleteDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SecurityToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SecurityToken)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DeleteDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DeleteDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *DeleteDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SecurityToken = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DeleteDomainRequest\n// struct.\nfunc (v *DeleteDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"SecurityToken: %v\", *(v.SecurityToken))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DeleteDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DeleteDomainRequest match the\n// provided DeleteDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DeleteDomainRequest) Equals(rhs *DeleteDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SecurityToken, rhs.SecurityToken) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DeleteDomainRequest.\nfunc (v *DeleteDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.SecurityToken != nil {\n\t\tenc.AddString(\"securityToken\", *v.SecurityToken)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DeleteDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DeleteDomainRequest) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetSecurityToken returns the value of SecurityToken if it is set or its\n// zero value if it is unset.\nfunc (v *DeleteDomainRequest) GetSecurityToken() (o string) {\n\tif v != nil && v.SecurityToken != nil {\n\t\treturn *v.SecurityToken\n\t}\n\n\treturn\n}\n\n// IsSetSecurityToken returns true if SecurityToken is not nil.\nfunc (v *DeleteDomainRequest) IsSetSecurityToken() bool {\n\treturn v != nil && v.SecurityToken != nil\n}\n\ntype DeprecateDomainRequest struct {\n\tName          *string `json:\"name,omitempty\"`\n\tSecurityToken *string `json:\"securityToken,omitempty\"`\n}\n\n// ToWire translates a DeprecateDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DeprecateDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tw, err = wire.NewValueString(*(v.SecurityToken)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DeprecateDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DeprecateDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DeprecateDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DeprecateDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SecurityToken = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DeprecateDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DeprecateDomainRequest struct could not be encoded.\nfunc (v *DeprecateDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SecurityToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SecurityToken)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DeprecateDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DeprecateDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *DeprecateDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SecurityToken = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DeprecateDomainRequest\n// struct.\nfunc (v *DeprecateDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"SecurityToken: %v\", *(v.SecurityToken))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DeprecateDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DeprecateDomainRequest match the\n// provided DeprecateDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DeprecateDomainRequest) Equals(rhs *DeprecateDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SecurityToken, rhs.SecurityToken) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DeprecateDomainRequest.\nfunc (v *DeprecateDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.SecurityToken != nil {\n\t\tenc.AddString(\"securityToken\", *v.SecurityToken)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DeprecateDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DeprecateDomainRequest) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetSecurityToken returns the value of SecurityToken if it is set or its\n// zero value if it is unset.\nfunc (v *DeprecateDomainRequest) GetSecurityToken() (o string) {\n\tif v != nil && v.SecurityToken != nil {\n\t\treturn *v.SecurityToken\n\t}\n\n\treturn\n}\n\n// IsSetSecurityToken returns true if SecurityToken is not nil.\nfunc (v *DeprecateDomainRequest) IsSetSecurityToken() bool {\n\treturn v != nil && v.SecurityToken != nil\n}\n\ntype DescribeDomainRequest struct {\n\tName *string `json:\"name,omitempty\"`\n\tUUID *string `json:\"uuid,omitempty\"`\n}\n\n// ToWire translates a DescribeDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.UUID != nil {\n\t\tw, err = wire.NewValueString(*(v.UUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.UUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeDomainRequest struct could not be encoded.\nfunc (v *DescribeDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.UUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.UUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeDomainRequest\n// struct.\nfunc (v *DescribeDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.UUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"UUID: %v\", *(v.UUID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeDomainRequest match the\n// provided DescribeDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeDomainRequest) Equals(rhs *DescribeDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.UUID, rhs.UUID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeDomainRequest.\nfunc (v *DescribeDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.UUID != nil {\n\t\tenc.AddString(\"uuid\", *v.UUID)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DescribeDomainRequest) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetUUID returns the value of UUID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainRequest) GetUUID() (o string) {\n\tif v != nil && v.UUID != nil {\n\t\treturn *v.UUID\n\t}\n\n\treturn\n}\n\n// IsSetUUID returns true if UUID is not nil.\nfunc (v *DescribeDomainRequest) IsSetUUID() bool {\n\treturn v != nil && v.UUID != nil\n}\n\ntype DescribeDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          *int64                          `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           *bool                           `json:\"isGlobalDomain,omitempty\"`\n\tFailoverInfo             *FailoverInfo                   `json:\"failoverInfo,omitempty\"`\n}\n\n// ToWire translates a DescribeDomainResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeDomainResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainInfo != nil {\n\t\tw, err = v.DomainInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tw, err = v.ReplicationConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsGlobalDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverInfo != nil {\n\t\tw, err = v.FailoverInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainInfo_Read(w wire.Value) (*DomainInfo, error) {\n\tvar v DomainInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainConfiguration_Read(w wire.Value) (*DomainConfiguration, error) {\n\tvar v DomainConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainReplicationConfiguration_Read(w wire.Value) (*DomainReplicationConfiguration, error) {\n\tvar v DomainReplicationConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _FailoverInfo_Read(w wire.Value) (*FailoverInfo, error) {\n\tvar v FailoverInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeDomainResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeDomainResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeDomainResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeDomainResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainInfo, err = _DomainInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _DomainConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsGlobalDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.FailoverInfo, err = _FailoverInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeDomainResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeDomainResponse struct could not be encoded.\nfunc (v *DescribeDomainResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicationConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsGlobalDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsGlobalDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailoverInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainInfo_Decode(sr stream.Reader) (*DomainInfo, error) {\n\tvar v DomainInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainConfiguration_Decode(sr stream.Reader) (*DomainConfiguration, error) {\n\tvar v DomainConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainReplicationConfiguration_Decode(sr stream.Reader) (*DomainReplicationConfiguration, error) {\n\tvar v DomainReplicationConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _FailoverInfo_Decode(sr stream.Reader) (*FailoverInfo, error) {\n\tvar v FailoverInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeDomainResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeDomainResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeDomainResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.DomainInfo, err = _DomainInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _DomainConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsGlobalDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.FailoverInfo, err = _FailoverInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeDomainResponse\n// struct.\nfunc (v *DescribeDomainResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.DomainInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainInfo: %v\", v.DomainInfo)\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationConfiguration: %v\", v.ReplicationConfiguration)\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsGlobalDomain: %v\", *(v.IsGlobalDomain))\n\t\ti++\n\t}\n\tif v.FailoverInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverInfo: %v\", v.FailoverInfo)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeDomainResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeDomainResponse match the\n// provided DescribeDomainResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeDomainResponse) Equals(rhs *DescribeDomainResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainInfo == nil && rhs.DomainInfo == nil) || (v.DomainInfo != nil && rhs.DomainInfo != nil && v.DomainInfo.Equals(rhs.DomainInfo))) {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationConfiguration == nil && rhs.ReplicationConfiguration == nil) || (v.ReplicationConfiguration != nil && rhs.ReplicationConfiguration != nil && v.ReplicationConfiguration.Equals(rhs.ReplicationConfiguration))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsGlobalDomain, rhs.IsGlobalDomain) {\n\t\treturn false\n\t}\n\tif !((v.FailoverInfo == nil && rhs.FailoverInfo == nil) || (v.FailoverInfo != nil && rhs.FailoverInfo != nil && v.FailoverInfo.Equals(rhs.FailoverInfo))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeDomainResponse.\nfunc (v *DescribeDomainResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainInfo\", v.DomainInfo))\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationConfiguration\", v.ReplicationConfiguration))\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tenc.AddBool(\"isGlobalDomain\", *v.IsGlobalDomain)\n\t}\n\tif v.FailoverInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failoverInfo\", v.FailoverInfo))\n\t}\n\treturn err\n}\n\n// GetDomainInfo returns the value of DomainInfo if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\n\treturn\n}\n\n// IsSetDomainInfo returns true if DomainInfo is not nil.\nfunc (v *DescribeDomainResponse) IsSetDomainInfo() bool {\n\treturn v != nil && v.DomainInfo != nil\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetConfiguration() (o *DomainConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *DescribeDomainResponse) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\n// GetReplicationConfiguration returns the value of ReplicationConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetReplicationConfiguration() (o *DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfiguration != nil {\n\t\treturn v.ReplicationConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetReplicationConfiguration returns true if ReplicationConfiguration is not nil.\nfunc (v *DescribeDomainResponse) IsSetReplicationConfiguration() bool {\n\treturn v != nil && v.ReplicationConfiguration != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *DescribeDomainResponse) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetIsGlobalDomain returns the value of IsGlobalDomain if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil && v.IsGlobalDomain != nil {\n\t\treturn *v.IsGlobalDomain\n\t}\n\n\treturn\n}\n\n// IsSetIsGlobalDomain returns true if IsGlobalDomain is not nil.\nfunc (v *DescribeDomainResponse) IsSetIsGlobalDomain() bool {\n\treturn v != nil && v.IsGlobalDomain != nil\n}\n\n// GetFailoverInfo returns the value of FailoverInfo if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeDomainResponse) GetFailoverInfo() (o *FailoverInfo) {\n\tif v != nil && v.FailoverInfo != nil {\n\t\treturn v.FailoverInfo\n\t}\n\n\treturn\n}\n\n// IsSetFailoverInfo returns true if FailoverInfo is not nil.\nfunc (v *DescribeDomainResponse) IsSetFailoverInfo() bool {\n\treturn v != nil && v.FailoverInfo != nil\n}\n\ntype DescribeHistoryHostRequest struct {\n\tHostAddress      *string            `json:\"hostAddress,omitempty\"`\n\tShardIdForHost   *int32             `json:\"shardIdForHost,omitempty\"`\n\tExecutionForHost *WorkflowExecution `json:\"executionForHost,omitempty\"`\n}\n\n// ToWire translates a DescribeHistoryHostRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeHistoryHostRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.HostAddress != nil {\n\t\tw, err = wire.NewValueString(*(v.HostAddress)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardIdForHost != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardIdForHost)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionForHost != nil {\n\t\tw, err = v.ExecutionForHost.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeHistoryHostRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeHistoryHostRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeHistoryHostRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeHistoryHostRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.HostAddress = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardIdForHost = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExecutionForHost, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeHistoryHostRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeHistoryHostRequest struct could not be encoded.\nfunc (v *DescribeHistoryHostRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.HostAddress != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.HostAddress)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardIdForHost != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardIdForHost)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionForHost != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExecutionForHost.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeHistoryHostRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeHistoryHostRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeHistoryHostRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.HostAddress = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardIdForHost = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ExecutionForHost, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeHistoryHostRequest\n// struct.\nfunc (v *DescribeHistoryHostRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.HostAddress != nil {\n\t\tfields[i] = fmt.Sprintf(\"HostAddress: %v\", *(v.HostAddress))\n\t\ti++\n\t}\n\tif v.ShardIdForHost != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardIdForHost: %v\", *(v.ShardIdForHost))\n\t\ti++\n\t}\n\tif v.ExecutionForHost != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionForHost: %v\", v.ExecutionForHost)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeHistoryHostRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeHistoryHostRequest match the\n// provided DescribeHistoryHostRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeHistoryHostRequest) Equals(rhs *DescribeHistoryHostRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.HostAddress, rhs.HostAddress) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardIdForHost, rhs.ShardIdForHost) {\n\t\treturn false\n\t}\n\tif !((v.ExecutionForHost == nil && rhs.ExecutionForHost == nil) || (v.ExecutionForHost != nil && rhs.ExecutionForHost != nil && v.ExecutionForHost.Equals(rhs.ExecutionForHost))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeHistoryHostRequest.\nfunc (v *DescribeHistoryHostRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.HostAddress != nil {\n\t\tenc.AddString(\"hostAddress\", *v.HostAddress)\n\t}\n\tif v.ShardIdForHost != nil {\n\t\tenc.AddInt32(\"shardIdForHost\", *v.ShardIdForHost)\n\t}\n\tif v.ExecutionForHost != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"executionForHost\", v.ExecutionForHost))\n\t}\n\treturn err\n}\n\n// GetHostAddress returns the value of HostAddress if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostRequest) GetHostAddress() (o string) {\n\tif v != nil && v.HostAddress != nil {\n\t\treturn *v.HostAddress\n\t}\n\n\treturn\n}\n\n// IsSetHostAddress returns true if HostAddress is not nil.\nfunc (v *DescribeHistoryHostRequest) IsSetHostAddress() bool {\n\treturn v != nil && v.HostAddress != nil\n}\n\n// GetShardIdForHost returns the value of ShardIdForHost if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostRequest) GetShardIdForHost() (o int32) {\n\tif v != nil && v.ShardIdForHost != nil {\n\t\treturn *v.ShardIdForHost\n\t}\n\n\treturn\n}\n\n// IsSetShardIdForHost returns true if ShardIdForHost is not nil.\nfunc (v *DescribeHistoryHostRequest) IsSetShardIdForHost() bool {\n\treturn v != nil && v.ShardIdForHost != nil\n}\n\n// GetExecutionForHost returns the value of ExecutionForHost if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostRequest) GetExecutionForHost() (o *WorkflowExecution) {\n\tif v != nil && v.ExecutionForHost != nil {\n\t\treturn v.ExecutionForHost\n\t}\n\n\treturn\n}\n\n// IsSetExecutionForHost returns true if ExecutionForHost is not nil.\nfunc (v *DescribeHistoryHostRequest) IsSetExecutionForHost() bool {\n\treturn v != nil && v.ExecutionForHost != nil\n}\n\ntype DescribeHistoryHostResponse struct {\n\tNumberOfShards        *int32           `json:\"numberOfShards,omitempty\"`\n\tShardIDs              []int32          `json:\"shardIDs,omitempty\"`\n\tDomainCache           *DomainCacheInfo `json:\"domainCache,omitempty\"`\n\tShardControllerStatus *string          `json:\"shardControllerStatus,omitempty\"`\n\tAddress               *string          `json:\"address,omitempty\"`\n}\n\ntype _List_I32_ValueList []int32\n\nfunc (v _List_I32_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueI32(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_I32_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_I32_ValueList) ValueType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_List_I32_ValueList) Close() {}\n\n// ToWire translates a DescribeHistoryHostResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeHistoryHostResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NumberOfShards != nil {\n\t\tw, err = wire.NewValueI32(*(v.NumberOfShards)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ShardIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I32_ValueList(v.ShardIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DomainCache != nil {\n\t\tw, err = v.DomainCache.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ShardControllerStatus != nil {\n\t\tw, err = wire.NewValueString(*(v.ShardControllerStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Address != nil {\n\t\tw, err = wire.NewValueString(*(v.Address)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_I32_Read(l wire.ValueList) ([]int32, error) {\n\tif l.ValueType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\to := make([]int32, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _DomainCacheInfo_Read(w wire.Value) (*DomainCacheInfo, error) {\n\tvar v DomainCacheInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeHistoryHostResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeHistoryHostResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeHistoryHostResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeHistoryHostResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.NumberOfShards = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ShardIDs, err = _List_I32_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainCache, err = _DomainCacheInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ShardControllerStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Address = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_I32_Encode(val []int32, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TI32,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteInt32(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DescribeHistoryHostResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeHistoryHostResponse struct could not be encoded.\nfunc (v *DescribeHistoryHostResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NumberOfShards != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.NumberOfShards)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I32_Encode(v.ShardIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainCache != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainCache.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ShardControllerStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ShardControllerStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Address != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Address)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_I32_Decode(sr stream.Reader) ([]int32, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TI32 {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]int32, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _DomainCacheInfo_Decode(sr stream.Reader) (*DomainCacheInfo, error) {\n\tvar v DomainCacheInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeHistoryHostResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeHistoryHostResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeHistoryHostResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.NumberOfShards = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.ShardIDs, err = _List_I32_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.DomainCache, err = _DomainCacheInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ShardControllerStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Address = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeHistoryHostResponse\n// struct.\nfunc (v *DescribeHistoryHostResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.NumberOfShards != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumberOfShards: %v\", *(v.NumberOfShards))\n\t\ti++\n\t}\n\tif v.ShardIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardIDs: %v\", v.ShardIDs)\n\t\ti++\n\t}\n\tif v.DomainCache != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainCache: %v\", v.DomainCache)\n\t\ti++\n\t}\n\tif v.ShardControllerStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardControllerStatus: %v\", *(v.ShardControllerStatus))\n\t\ti++\n\t}\n\tif v.Address != nil {\n\t\tfields[i] = fmt.Sprintf(\"Address: %v\", *(v.Address))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeHistoryHostResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_I32_Equals(lhs, rhs []int32) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeHistoryHostResponse match the\n// provided DescribeHistoryHostResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeHistoryHostResponse) Equals(rhs *DescribeHistoryHostResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.NumberOfShards, rhs.NumberOfShards) {\n\t\treturn false\n\t}\n\tif !((v.ShardIDs == nil && rhs.ShardIDs == nil) || (v.ShardIDs != nil && rhs.ShardIDs != nil && _List_I32_Equals(v.ShardIDs, rhs.ShardIDs))) {\n\t\treturn false\n\t}\n\tif !((v.DomainCache == nil && rhs.DomainCache == nil) || (v.DomainCache != nil && rhs.DomainCache != nil && v.DomainCache.Equals(rhs.DomainCache))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ShardControllerStatus, rhs.ShardControllerStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Address, rhs.Address) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_I32_Zapper []int32\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_I32_Zapper.\nfunc (l _List_I32_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendInt32(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeHistoryHostResponse.\nfunc (v *DescribeHistoryHostResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NumberOfShards != nil {\n\t\tenc.AddInt32(\"numberOfShards\", *v.NumberOfShards)\n\t}\n\tif v.ShardIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"shardIDs\", (_List_I32_Zapper)(v.ShardIDs)))\n\t}\n\tif v.DomainCache != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainCache\", v.DomainCache))\n\t}\n\tif v.ShardControllerStatus != nil {\n\t\tenc.AddString(\"shardControllerStatus\", *v.ShardControllerStatus)\n\t}\n\tif v.Address != nil {\n\t\tenc.AddString(\"address\", *v.Address)\n\t}\n\treturn err\n}\n\n// GetNumberOfShards returns the value of NumberOfShards if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostResponse) GetNumberOfShards() (o int32) {\n\tif v != nil && v.NumberOfShards != nil {\n\t\treturn *v.NumberOfShards\n\t}\n\n\treturn\n}\n\n// IsSetNumberOfShards returns true if NumberOfShards is not nil.\nfunc (v *DescribeHistoryHostResponse) IsSetNumberOfShards() bool {\n\treturn v != nil && v.NumberOfShards != nil\n}\n\n// GetShardIDs returns the value of ShardIDs if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostResponse) GetShardIDs() (o []int32) {\n\tif v != nil && v.ShardIDs != nil {\n\t\treturn v.ShardIDs\n\t}\n\n\treturn\n}\n\n// IsSetShardIDs returns true if ShardIDs is not nil.\nfunc (v *DescribeHistoryHostResponse) IsSetShardIDs() bool {\n\treturn v != nil && v.ShardIDs != nil\n}\n\n// GetDomainCache returns the value of DomainCache if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostResponse) GetDomainCache() (o *DomainCacheInfo) {\n\tif v != nil && v.DomainCache != nil {\n\t\treturn v.DomainCache\n\t}\n\n\treturn\n}\n\n// IsSetDomainCache returns true if DomainCache is not nil.\nfunc (v *DescribeHistoryHostResponse) IsSetDomainCache() bool {\n\treturn v != nil && v.DomainCache != nil\n}\n\n// GetShardControllerStatus returns the value of ShardControllerStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostResponse) GetShardControllerStatus() (o string) {\n\tif v != nil && v.ShardControllerStatus != nil {\n\t\treturn *v.ShardControllerStatus\n\t}\n\n\treturn\n}\n\n// IsSetShardControllerStatus returns true if ShardControllerStatus is not nil.\nfunc (v *DescribeHistoryHostResponse) IsSetShardControllerStatus() bool {\n\treturn v != nil && v.ShardControllerStatus != nil\n}\n\n// GetAddress returns the value of Address if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeHistoryHostResponse) GetAddress() (o string) {\n\tif v != nil && v.Address != nil {\n\t\treturn *v.Address\n\t}\n\n\treturn\n}\n\n// IsSetAddress returns true if Address is not nil.\nfunc (v *DescribeHistoryHostResponse) IsSetAddress() bool {\n\treturn v != nil && v.Address != nil\n}\n\ntype DescribeQueueRequest struct {\n\tShardID     *int32  `json:\"shardID,omitempty\"`\n\tClusterName *string `json:\"clusterName,omitempty\"`\n\tType        *int32  `json:\"type,omitempty\"`\n}\n\n// ToWire translates a DescribeQueueRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeQueueRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tw, err = wire.NewValueI32(*(v.Type)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeQueueRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeQueueRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeQueueRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeQueueRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeQueueRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeQueueRequest struct could not be encoded.\nfunc (v *DescribeQueueRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Type)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeQueueRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeQueueRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeQueueRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeQueueRequest\n// struct.\nfunc (v *DescribeQueueRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeQueueRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeQueueRequest match the\n// provided DescribeQueueRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeQueueRequest) Equals(rhs *DescribeQueueRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeQueueRequest.\nfunc (v *DescribeQueueRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\tif v.Type != nil {\n\t\tenc.AddInt32(\"type\", *v.Type)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeQueueRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *DescribeQueueRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeQueueRequest) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *DescribeQueueRequest) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeQueueRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *DescribeQueueRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\ntype DescribeQueueResponse struct {\n\tProcessingQueueStates []string `json:\"processingQueueStates,omitempty\"`\n}\n\ntype _List_String_ValueList []string\n\nfunc (v _List_String_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueString(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_String_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_String_ValueList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_List_String_ValueList) Close() {}\n\n// ToWire translates a DescribeQueueResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeQueueResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ProcessingQueueStates != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.ProcessingQueueStates)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_String_Read(l wire.ValueList) ([]string, error) {\n\tif l.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make([]string, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DescribeQueueResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeQueueResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeQueueResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeQueueResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ProcessingQueueStates, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_String_Encode(val []string, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TBinary,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DescribeQueueResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeQueueResponse struct could not be encoded.\nfunc (v *DescribeQueueResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ProcessingQueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.ProcessingQueueStates, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_String_Decode(sr stream.Reader) ([]string, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TBinary {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]string, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DescribeQueueResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeQueueResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeQueueResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ProcessingQueueStates, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeQueueResponse\n// struct.\nfunc (v *DescribeQueueResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ProcessingQueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"ProcessingQueueStates: %v\", v.ProcessingQueueStates)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeQueueResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_String_Equals(lhs, rhs []string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeQueueResponse match the\n// provided DescribeQueueResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeQueueResponse) Equals(rhs *DescribeQueueResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ProcessingQueueStates == nil && rhs.ProcessingQueueStates == nil) || (v.ProcessingQueueStates != nil && rhs.ProcessingQueueStates != nil && _List_String_Equals(v.ProcessingQueueStates, rhs.ProcessingQueueStates))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_String_Zapper []string\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_String_Zapper.\nfunc (l _List_String_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendString(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeQueueResponse.\nfunc (v *DescribeQueueResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ProcessingQueueStates != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"processingQueueStates\", (_List_String_Zapper)(v.ProcessingQueueStates)))\n\t}\n\treturn err\n}\n\n// GetProcessingQueueStates returns the value of ProcessingQueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeQueueResponse) GetProcessingQueueStates() (o []string) {\n\tif v != nil && v.ProcessingQueueStates != nil {\n\t\treturn v.ProcessingQueueStates\n\t}\n\n\treturn\n}\n\n// IsSetProcessingQueueStates returns true if ProcessingQueueStates is not nil.\nfunc (v *DescribeQueueResponse) IsSetProcessingQueueStates() bool {\n\treturn v != nil && v.ProcessingQueueStates != nil\n}\n\ntype DescribeShardDistributionRequest struct {\n\tPageSize *int32 `json:\"pageSize,omitempty\"`\n\tPageID   *int32 `json:\"pageID,omitempty\"`\n}\n\n// ToWire translates a DescribeShardDistributionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeShardDistributionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PageID != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DescribeShardDistributionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeShardDistributionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeShardDistributionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeShardDistributionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeShardDistributionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeShardDistributionRequest struct could not be encoded.\nfunc (v *DescribeShardDistributionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PageID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DescribeShardDistributionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeShardDistributionRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeShardDistributionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeShardDistributionRequest\n// struct.\nfunc (v *DescribeShardDistributionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.PageID != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageID: %v\", *(v.PageID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeShardDistributionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DescribeShardDistributionRequest match the\n// provided DescribeShardDistributionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeShardDistributionRequest) Equals(rhs *DescribeShardDistributionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageID, rhs.PageID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeShardDistributionRequest.\nfunc (v *DescribeShardDistributionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.PageID != nil {\n\t\tenc.AddInt32(\"pageID\", *v.PageID)\n\t}\n\treturn err\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeShardDistributionRequest) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *DescribeShardDistributionRequest) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetPageID returns the value of PageID if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeShardDistributionRequest) GetPageID() (o int32) {\n\tif v != nil && v.PageID != nil {\n\t\treturn *v.PageID\n\t}\n\n\treturn\n}\n\n// IsSetPageID returns true if PageID is not nil.\nfunc (v *DescribeShardDistributionRequest) IsSetPageID() bool {\n\treturn v != nil && v.PageID != nil\n}\n\ntype DescribeShardDistributionResponse struct {\n\tNumberOfShards *int32           `json:\"numberOfShards,omitempty\"`\n\tShards         map[int32]string `json:\"shards,omitempty\"`\n}\n\ntype _Map_I32_String_MapItemList map[int32]string\n\nfunc (m _Map_I32_String_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueString(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_String_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_String_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_String_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_I32_String_MapItemList) Close() {}\n\n// ToWire translates a DescribeShardDistributionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeShardDistributionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NumberOfShards != nil {\n\t\tw, err = wire.NewValueI32(*(v.NumberOfShards)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Shards != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_String_MapItemList(v.Shards)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_I32_String_Read(m wire.MapItemList) (map[int32]string, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32]string, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DescribeShardDistributionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeShardDistributionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeShardDistributionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeShardDistributionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.NumberOfShards = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Shards, err = _Map_I32_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_I32_String_Encode(val map[int32]string, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a DescribeShardDistributionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeShardDistributionResponse struct could not be encoded.\nfunc (v *DescribeShardDistributionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NumberOfShards != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.NumberOfShards)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Shards != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_String_Encode(v.Shards, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_I32_String_Decode(sr stream.Reader) (map[int32]string, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32]string, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DescribeShardDistributionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeShardDistributionResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeShardDistributionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.NumberOfShards = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TMap:\n\t\t\tv.Shards, err = _Map_I32_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeShardDistributionResponse\n// struct.\nfunc (v *DescribeShardDistributionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.NumberOfShards != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumberOfShards: %v\", *(v.NumberOfShards))\n\t\ti++\n\t}\n\tif v.Shards != nil {\n\t\tfields[i] = fmt.Sprintf(\"Shards: %v\", v.Shards)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeShardDistributionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_I32_String_Equals(lhs, rhs map[int32]string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeShardDistributionResponse match the\n// provided DescribeShardDistributionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeShardDistributionResponse) Equals(rhs *DescribeShardDistributionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.NumberOfShards, rhs.NumberOfShards) {\n\t\treturn false\n\t}\n\tif !((v.Shards == nil && rhs.Shards == nil) || (v.Shards != nil && rhs.Shards != nil && _Map_I32_String_Equals(v.Shards, rhs.Shards))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_I32_String_Item_Zapper struct {\n\tKey   int32\n\tValue string\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_String_Item_Zapper.\nfunc (v _Map_I32_String_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\tenc.AddString(\"value\", v.Value)\n\treturn err\n}\n\ntype _Map_I32_String_Zapper map[int32]string\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_String_Zapper.\nfunc (m _Map_I32_String_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_String_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeShardDistributionResponse.\nfunc (v *DescribeShardDistributionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NumberOfShards != nil {\n\t\tenc.AddInt32(\"numberOfShards\", *v.NumberOfShards)\n\t}\n\tif v.Shards != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"shards\", (_Map_I32_String_Zapper)(v.Shards)))\n\t}\n\treturn err\n}\n\n// GetNumberOfShards returns the value of NumberOfShards if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeShardDistributionResponse) GetNumberOfShards() (o int32) {\n\tif v != nil && v.NumberOfShards != nil {\n\t\treturn *v.NumberOfShards\n\t}\n\n\treturn\n}\n\n// IsSetNumberOfShards returns true if NumberOfShards is not nil.\nfunc (v *DescribeShardDistributionResponse) IsSetNumberOfShards() bool {\n\treturn v != nil && v.NumberOfShards != nil\n}\n\n// GetShards returns the value of Shards if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeShardDistributionResponse) GetShards() (o map[int32]string) {\n\tif v != nil && v.Shards != nil {\n\t\treturn v.Shards\n\t}\n\n\treturn\n}\n\n// IsSetShards returns true if Shards is not nil.\nfunc (v *DescribeShardDistributionResponse) IsSetShards() bool {\n\treturn v != nil && v.Shards != nil\n}\n\ntype DescribeTaskListRequest struct {\n\tDomain                *string       `json:\"domain,omitempty\"`\n\tTaskList              *TaskList     `json:\"taskList,omitempty\"`\n\tTaskListType          *TaskListType `json:\"taskListType,omitempty\"`\n\tIncludeTaskListStatus *bool         `json:\"includeTaskListStatus,omitempty\"`\n}\n\n// ToWire translates a DescribeTaskListRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeTaskListRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListType != nil {\n\t\tw, err = v.TaskListType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.IncludeTaskListStatus != nil {\n\t\tw, err = wire.NewValueBool(*(v.IncludeTaskListStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListType_Read(w wire.Value) (TaskListType, error) {\n\tvar v TaskListType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DescribeTaskListRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeTaskListRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeTaskListRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeTaskListRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TaskListType\n\t\t\t\tx, err = _TaskListType_Read(field.Value)\n\t\t\t\tv.TaskListType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IncludeTaskListStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeTaskListRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeTaskListRequest struct could not be encoded.\nfunc (v *DescribeTaskListRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IncludeTaskListStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IncludeTaskListStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListType_Decode(sr stream.Reader) (TaskListType, error) {\n\tvar v TaskListType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DescribeTaskListRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeTaskListRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeTaskListRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x TaskListType\n\t\t\tx, err = _TaskListType_Decode(sr)\n\t\t\tv.TaskListType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IncludeTaskListStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeTaskListRequest\n// struct.\nfunc (v *DescribeTaskListRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.TaskListType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListType: %v\", *(v.TaskListType))\n\t\ti++\n\t}\n\tif v.IncludeTaskListStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"IncludeTaskListStatus: %v\", *(v.IncludeTaskListStatus))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeTaskListRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _TaskListType_EqualsPtr(lhs, rhs *TaskListType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DescribeTaskListRequest match the\n// provided DescribeTaskListRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeTaskListRequest) Equals(rhs *DescribeTaskListRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_TaskListType_EqualsPtr(v.TaskListType, rhs.TaskListType) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IncludeTaskListStatus, rhs.IncludeTaskListStatus) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeTaskListRequest.\nfunc (v *DescribeTaskListRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.TaskListType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListType\", *v.TaskListType))\n\t}\n\tif v.IncludeTaskListStatus != nil {\n\t\tenc.AddBool(\"includeTaskListStatus\", *v.IncludeTaskListStatus)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *DescribeTaskListRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *DescribeTaskListRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTaskListType returns the value of TaskListType if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetTaskListType() (o TaskListType) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\n\treturn\n}\n\n// IsSetTaskListType returns true if TaskListType is not nil.\nfunc (v *DescribeTaskListRequest) IsSetTaskListType() bool {\n\treturn v != nil && v.TaskListType != nil\n}\n\n// GetIncludeTaskListStatus returns the value of IncludeTaskListStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListRequest) GetIncludeTaskListStatus() (o bool) {\n\tif v != nil && v.IncludeTaskListStatus != nil {\n\t\treturn *v.IncludeTaskListStatus\n\t}\n\n\treturn\n}\n\n// IsSetIncludeTaskListStatus returns true if IncludeTaskListStatus is not nil.\nfunc (v *DescribeTaskListRequest) IsSetIncludeTaskListStatus() bool {\n\treturn v != nil && v.IncludeTaskListStatus != nil\n}\n\ntype DescribeTaskListResponse struct {\n\tPollers        []*PollerInfo   `json:\"pollers,omitempty\"`\n\tTaskListStatus *TaskListStatus `json:\"taskListStatus,omitempty\"`\n\tTaskList       *TaskList       `json:\"taskList,omitempty\"`\n}\n\ntype _List_PollerInfo_ValueList []*PollerInfo\n\nfunc (v _List_PollerInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PollerInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_PollerInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_PollerInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_PollerInfo_ValueList) Close() {}\n\n// ToWire translates a DescribeTaskListResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeTaskListResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Pollers != nil {\n\t\tw, err = wire.NewValueList(_List_PollerInfo_ValueList(v.Pollers)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListStatus != nil {\n\t\tw, err = v.TaskListStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollerInfo_Read(w wire.Value) (*PollerInfo, error) {\n\tvar v PollerInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_PollerInfo_Read(l wire.ValueList) ([]*PollerInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*PollerInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _PollerInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _TaskListStatus_Read(w wire.Value) (*TaskListStatus, error) {\n\tvar v TaskListStatus\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeTaskListResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeTaskListResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeTaskListResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeTaskListResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Pollers, err = _List_PollerInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListStatus, err = _TaskListStatus_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_PollerInfo_Encode(val []*PollerInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PollerInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DescribeTaskListResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeTaskListResponse struct could not be encoded.\nfunc (v *DescribeTaskListResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Pollers != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_PollerInfo_Encode(v.Pollers, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollerInfo_Decode(sr stream.Reader) (*PollerInfo, error) {\n\tvar v PollerInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_PollerInfo_Decode(sr stream.Reader) ([]*PollerInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*PollerInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _PollerInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _TaskListStatus_Decode(sr stream.Reader) (*TaskListStatus, error) {\n\tvar v TaskListStatus\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeTaskListResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeTaskListResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeTaskListResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Pollers, err = _List_PollerInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListStatus, err = _TaskListStatus_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeTaskListResponse\n// struct.\nfunc (v *DescribeTaskListResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Pollers != nil {\n\t\tfields[i] = fmt.Sprintf(\"Pollers: %v\", v.Pollers)\n\t\ti++\n\t}\n\tif v.TaskListStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListStatus: %v\", v.TaskListStatus)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeTaskListResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_PollerInfo_Equals(lhs, rhs []*PollerInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeTaskListResponse match the\n// provided DescribeTaskListResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeTaskListResponse) Equals(rhs *DescribeTaskListResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Pollers == nil && rhs.Pollers == nil) || (v.Pollers != nil && rhs.Pollers != nil && _List_PollerInfo_Equals(v.Pollers, rhs.Pollers))) {\n\t\treturn false\n\t}\n\tif !((v.TaskListStatus == nil && rhs.TaskListStatus == nil) || (v.TaskListStatus != nil && rhs.TaskListStatus != nil && v.TaskListStatus.Equals(rhs.TaskListStatus))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_PollerInfo_Zapper []*PollerInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_PollerInfo_Zapper.\nfunc (l _List_PollerInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeTaskListResponse.\nfunc (v *DescribeTaskListResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Pollers != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pollers\", (_List_PollerInfo_Zapper)(v.Pollers)))\n\t}\n\tif v.TaskListStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListStatus\", v.TaskListStatus))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\treturn err\n}\n\n// GetPollers returns the value of Pollers if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListResponse) GetPollers() (o []*PollerInfo) {\n\tif v != nil && v.Pollers != nil {\n\t\treturn v.Pollers\n\t}\n\n\treturn\n}\n\n// IsSetPollers returns true if Pollers is not nil.\nfunc (v *DescribeTaskListResponse) IsSetPollers() bool {\n\treturn v != nil && v.Pollers != nil\n}\n\n// GetTaskListStatus returns the value of TaskListStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListResponse) GetTaskListStatus() (o *TaskListStatus) {\n\tif v != nil && v.TaskListStatus != nil {\n\t\treturn v.TaskListStatus\n\t}\n\n\treturn\n}\n\n// IsSetTaskListStatus returns true if TaskListStatus is not nil.\nfunc (v *DescribeTaskListResponse) IsSetTaskListStatus() bool {\n\treturn v != nil && v.TaskListStatus != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeTaskListResponse) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *DescribeTaskListResponse) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\ntype DescribeWorkflowExecutionRequest struct {\n\tDomain                *string                `json:\"domain,omitempty\"`\n\tExecution             *WorkflowExecution     `json:\"execution,omitempty\"`\n\tQueryConsistencyLevel *QueryConsistencyLevel `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// ToWire translates a DescribeWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tw, err = v.QueryConsistencyLevel.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryConsistencyLevel_Read(w wire.Value) (QueryConsistencyLevel, error) {\n\tvar v QueryConsistencyLevel\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DescribeWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryConsistencyLevel\n\t\t\t\tx, err = _QueryConsistencyLevel_Read(field.Value)\n\t\t\t\tv.QueryConsistencyLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DescribeWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be encoded.\nfunc (v *DescribeWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryConsistencyLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryConsistencyLevel.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryConsistencyLevel_Decode(sr stream.Reader) (QueryConsistencyLevel, error) {\n\tvar v QueryConsistencyLevel\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DescribeWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *DescribeWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x QueryConsistencyLevel\n\t\t\tx, err = _QueryConsistencyLevel_Decode(sr)\n\t\t\tv.QueryConsistencyLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeWorkflowExecutionRequest\n// struct.\nfunc (v *DescribeWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryConsistencyLevel: %v\", *(v.QueryConsistencyLevel))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _QueryConsistencyLevel_EqualsPtr(lhs, rhs *QueryConsistencyLevel) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DescribeWorkflowExecutionRequest match the\n// provided DescribeWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeWorkflowExecutionRequest) Equals(rhs *DescribeWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_QueryConsistencyLevel_EqualsPtr(v.QueryConsistencyLevel, rhs.QueryConsistencyLevel) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeWorkflowExecutionRequest.\nfunc (v *DescribeWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryConsistencyLevel\", *v.QueryConsistencyLevel))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetQueryConsistencyLevel returns the value of QueryConsistencyLevel if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\n\treturn\n}\n\n// IsSetQueryConsistencyLevel returns true if QueryConsistencyLevel is not nil.\nfunc (v *DescribeWorkflowExecutionRequest) IsSetQueryConsistencyLevel() bool {\n\treturn v != nil && v.QueryConsistencyLevel != nil\n}\n\ntype DescribeWorkflowExecutionResponse struct {\n\tExecutionConfiguration *WorkflowExecutionConfiguration `json:\"executionConfiguration,omitempty\"`\n\tWorkflowExecutionInfo  *WorkflowExecutionInfo          `json:\"workflowExecutionInfo,omitempty\"`\n\tPendingActivities      []*PendingActivityInfo          `json:\"pendingActivities,omitempty\"`\n\tPendingChildren        []*PendingChildExecutionInfo    `json:\"pendingChildren,omitempty\"`\n\tPendingDecision        *PendingDecisionInfo            `json:\"pendingDecision,omitempty\"`\n}\n\ntype _List_PendingActivityInfo_ValueList []*PendingActivityInfo\n\nfunc (v _List_PendingActivityInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PendingActivityInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_PendingActivityInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_PendingActivityInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_PendingActivityInfo_ValueList) Close() {}\n\ntype _List_PendingChildExecutionInfo_ValueList []*PendingChildExecutionInfo\n\nfunc (v _List_PendingChildExecutionInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PendingChildExecutionInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_PendingChildExecutionInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_PendingChildExecutionInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_PendingChildExecutionInfo_ValueList) Close() {}\n\n// ToWire translates a DescribeWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DescribeWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ExecutionConfiguration != nil {\n\t\tw, err = v.ExecutionConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionInfo != nil {\n\t\tw, err = v.WorkflowExecutionInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.PendingActivities != nil {\n\t\tw, err = wire.NewValueList(_List_PendingActivityInfo_ValueList(v.PendingActivities)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PendingChildren != nil {\n\t\tw, err = wire.NewValueList(_List_PendingChildExecutionInfo_ValueList(v.PendingChildren)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.PendingDecision != nil {\n\t\tw, err = v.PendingDecision.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowExecutionConfiguration_Read(w wire.Value) (*WorkflowExecutionConfiguration, error) {\n\tvar v WorkflowExecutionConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionInfo_Read(w wire.Value) (*WorkflowExecutionInfo, error) {\n\tvar v WorkflowExecutionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _PendingActivityInfo_Read(w wire.Value) (*PendingActivityInfo, error) {\n\tvar v PendingActivityInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_PendingActivityInfo_Read(l wire.ValueList) ([]*PendingActivityInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*PendingActivityInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _PendingActivityInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _PendingChildExecutionInfo_Read(w wire.Value) (*PendingChildExecutionInfo, error) {\n\tvar v PendingChildExecutionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_PendingChildExecutionInfo_Read(l wire.ValueList) ([]*PendingChildExecutionInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*PendingChildExecutionInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _PendingChildExecutionInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _PendingDecisionInfo_Read(w wire.Value) (*PendingDecisionInfo, error) {\n\tvar v PendingDecisionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DescribeWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DescribeWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DescribeWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DescribeWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExecutionConfiguration, err = _WorkflowExecutionConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionInfo, err = _WorkflowExecutionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingActivities, err = _List_PendingActivityInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingChildren, err = _List_PendingChildExecutionInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PendingDecision, err = _PendingDecisionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_PendingActivityInfo_Encode(val []*PendingActivityInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PendingActivityInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _List_PendingChildExecutionInfo_Encode(val []*PendingChildExecutionInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*PendingChildExecutionInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DescribeWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionResponse struct could not be encoded.\nfunc (v *DescribeWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ExecutionConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExecutionConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingActivities != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_PendingActivityInfo_Encode(v.PendingActivities, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingChildren != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_PendingChildExecutionInfo_Encode(v.PendingChildren, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingDecision != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PendingDecision.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowExecutionConfiguration_Decode(sr stream.Reader) (*WorkflowExecutionConfiguration, error) {\n\tvar v WorkflowExecutionConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionInfo_Decode(sr stream.Reader) (*WorkflowExecutionInfo, error) {\n\tvar v WorkflowExecutionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _PendingActivityInfo_Decode(sr stream.Reader) (*PendingActivityInfo, error) {\n\tvar v PendingActivityInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_PendingActivityInfo_Decode(sr stream.Reader) ([]*PendingActivityInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*PendingActivityInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _PendingActivityInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _PendingChildExecutionInfo_Decode(sr stream.Reader) (*PendingChildExecutionInfo, error) {\n\tvar v PendingChildExecutionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_PendingChildExecutionInfo_Decode(sr stream.Reader) ([]*PendingChildExecutionInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*PendingChildExecutionInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _PendingChildExecutionInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _PendingDecisionInfo_Decode(sr stream.Reader) (*PendingDecisionInfo, error) {\n\tvar v PendingDecisionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DescribeWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DescribeWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *DescribeWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.ExecutionConfiguration, err = _WorkflowExecutionConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionInfo, err = _WorkflowExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.PendingActivities, err = _List_PendingActivityInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TList:\n\t\t\tv.PendingChildren, err = _List_PendingChildExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.PendingDecision, err = _PendingDecisionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DescribeWorkflowExecutionResponse\n// struct.\nfunc (v *DescribeWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.ExecutionConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionConfiguration: %v\", v.ExecutionConfiguration)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionInfo: %v\", v.WorkflowExecutionInfo)\n\t\ti++\n\t}\n\tif v.PendingActivities != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingActivities: %v\", v.PendingActivities)\n\t\ti++\n\t}\n\tif v.PendingChildren != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingChildren: %v\", v.PendingChildren)\n\t\ti++\n\t}\n\tif v.PendingDecision != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingDecision: %v\", v.PendingDecision)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DescribeWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_PendingActivityInfo_Equals(lhs, rhs []*PendingActivityInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _List_PendingChildExecutionInfo_Equals(lhs, rhs []*PendingChildExecutionInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DescribeWorkflowExecutionResponse match the\n// provided DescribeWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DescribeWorkflowExecutionResponse) Equals(rhs *DescribeWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ExecutionConfiguration == nil && rhs.ExecutionConfiguration == nil) || (v.ExecutionConfiguration != nil && rhs.ExecutionConfiguration != nil && v.ExecutionConfiguration.Equals(rhs.ExecutionConfiguration))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionInfo == nil && rhs.WorkflowExecutionInfo == nil) || (v.WorkflowExecutionInfo != nil && rhs.WorkflowExecutionInfo != nil && v.WorkflowExecutionInfo.Equals(rhs.WorkflowExecutionInfo))) {\n\t\treturn false\n\t}\n\tif !((v.PendingActivities == nil && rhs.PendingActivities == nil) || (v.PendingActivities != nil && rhs.PendingActivities != nil && _List_PendingActivityInfo_Equals(v.PendingActivities, rhs.PendingActivities))) {\n\t\treturn false\n\t}\n\tif !((v.PendingChildren == nil && rhs.PendingChildren == nil) || (v.PendingChildren != nil && rhs.PendingChildren != nil && _List_PendingChildExecutionInfo_Equals(v.PendingChildren, rhs.PendingChildren))) {\n\t\treturn false\n\t}\n\tif !((v.PendingDecision == nil && rhs.PendingDecision == nil) || (v.PendingDecision != nil && rhs.PendingDecision != nil && v.PendingDecision.Equals(rhs.PendingDecision))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_PendingActivityInfo_Zapper []*PendingActivityInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_PendingActivityInfo_Zapper.\nfunc (l _List_PendingActivityInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\ntype _List_PendingChildExecutionInfo_Zapper []*PendingChildExecutionInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_PendingChildExecutionInfo_Zapper.\nfunc (l _List_PendingChildExecutionInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DescribeWorkflowExecutionResponse.\nfunc (v *DescribeWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ExecutionConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"executionConfiguration\", v.ExecutionConfiguration))\n\t}\n\tif v.WorkflowExecutionInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionInfo\", v.WorkflowExecutionInfo))\n\t}\n\tif v.PendingActivities != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingActivities\", (_List_PendingActivityInfo_Zapper)(v.PendingActivities)))\n\t}\n\tif v.PendingChildren != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingChildren\", (_List_PendingChildExecutionInfo_Zapper)(v.PendingChildren)))\n\t}\n\tif v.PendingDecision != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pendingDecision\", v.PendingDecision))\n\t}\n\treturn err\n}\n\n// GetExecutionConfiguration returns the value of ExecutionConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetExecutionConfiguration() (o *WorkflowExecutionConfiguration) {\n\tif v != nil && v.ExecutionConfiguration != nil {\n\t\treturn v.ExecutionConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetExecutionConfiguration returns true if ExecutionConfiguration is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetExecutionConfiguration() bool {\n\treturn v != nil && v.ExecutionConfiguration != nil\n}\n\n// GetWorkflowExecutionInfo returns the value of WorkflowExecutionInfo if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetWorkflowExecutionInfo() (o *WorkflowExecutionInfo) {\n\tif v != nil && v.WorkflowExecutionInfo != nil {\n\t\treturn v.WorkflowExecutionInfo\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionInfo returns true if WorkflowExecutionInfo is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetWorkflowExecutionInfo() bool {\n\treturn v != nil && v.WorkflowExecutionInfo != nil\n}\n\n// GetPendingActivities returns the value of PendingActivities if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetPendingActivities() (o []*PendingActivityInfo) {\n\tif v != nil && v.PendingActivities != nil {\n\t\treturn v.PendingActivities\n\t}\n\n\treturn\n}\n\n// IsSetPendingActivities returns true if PendingActivities is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetPendingActivities() bool {\n\treturn v != nil && v.PendingActivities != nil\n}\n\n// GetPendingChildren returns the value of PendingChildren if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetPendingChildren() (o []*PendingChildExecutionInfo) {\n\tif v != nil && v.PendingChildren != nil {\n\t\treturn v.PendingChildren\n\t}\n\n\treturn\n}\n\n// IsSetPendingChildren returns true if PendingChildren is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetPendingChildren() bool {\n\treturn v != nil && v.PendingChildren != nil\n}\n\n// GetPendingDecision returns the value of PendingDecision if it is set or its\n// zero value if it is unset.\nfunc (v *DescribeWorkflowExecutionResponse) GetPendingDecision() (o *PendingDecisionInfo) {\n\tif v != nil && v.PendingDecision != nil {\n\t\treturn v.PendingDecision\n\t}\n\n\treturn\n}\n\n// IsSetPendingDecision returns true if PendingDecision is not nil.\nfunc (v *DescribeWorkflowExecutionResponse) IsSetPendingDecision() bool {\n\treturn v != nil && v.PendingDecision != nil\n}\n\ntype DiagnoseWorkflowExecutionRequest struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIdentity          *string            `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a DiagnoseWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DiagnoseWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DiagnoseWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DiagnoseWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DiagnoseWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DiagnoseWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DiagnoseWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DiagnoseWorkflowExecutionRequest struct could not be encoded.\nfunc (v *DiagnoseWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DiagnoseWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DiagnoseWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *DiagnoseWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DiagnoseWorkflowExecutionRequest\n// struct.\nfunc (v *DiagnoseWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DiagnoseWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DiagnoseWorkflowExecutionRequest match the\n// provided DiagnoseWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *DiagnoseWorkflowExecutionRequest) Equals(rhs *DiagnoseWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DiagnoseWorkflowExecutionRequest.\nfunc (v *DiagnoseWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *DiagnoseWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *DiagnoseWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *DiagnoseWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *DiagnoseWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *DiagnoseWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *DiagnoseWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype DiagnoseWorkflowExecutionResponse struct {\n\tDomain                      *string            `json:\"domain,omitempty\"`\n\tDiagnosticWorkflowExecution *WorkflowExecution `json:\"diagnosticWorkflowExecution,omitempty\"`\n}\n\n// ToWire translates a DiagnoseWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DiagnoseWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DiagnosticWorkflowExecution != nil {\n\t\tw, err = v.DiagnosticWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DiagnoseWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DiagnoseWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DiagnoseWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DiagnoseWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DiagnosticWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DiagnoseWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DiagnoseWorkflowExecutionResponse struct could not be encoded.\nfunc (v *DiagnoseWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DiagnosticWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DiagnosticWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DiagnoseWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DiagnoseWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *DiagnoseWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.DiagnosticWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DiagnoseWorkflowExecutionResponse\n// struct.\nfunc (v *DiagnoseWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.DiagnosticWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"DiagnosticWorkflowExecution: %v\", v.DiagnosticWorkflowExecution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DiagnoseWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DiagnoseWorkflowExecutionResponse match the\n// provided DiagnoseWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *DiagnoseWorkflowExecutionResponse) Equals(rhs *DiagnoseWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.DiagnosticWorkflowExecution == nil && rhs.DiagnosticWorkflowExecution == nil) || (v.DiagnosticWorkflowExecution != nil && rhs.DiagnosticWorkflowExecution != nil && v.DiagnosticWorkflowExecution.Equals(rhs.DiagnosticWorkflowExecution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DiagnoseWorkflowExecutionResponse.\nfunc (v *DiagnoseWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.DiagnosticWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"diagnosticWorkflowExecution\", v.DiagnosticWorkflowExecution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *DiagnoseWorkflowExecutionResponse) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *DiagnoseWorkflowExecutionResponse) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetDiagnosticWorkflowExecution returns the value of DiagnosticWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *DiagnoseWorkflowExecutionResponse) GetDiagnosticWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.DiagnosticWorkflowExecution != nil {\n\t\treturn v.DiagnosticWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetDiagnosticWorkflowExecution returns true if DiagnosticWorkflowExecution is not nil.\nfunc (v *DiagnoseWorkflowExecutionResponse) IsSetDiagnosticWorkflowExecution() bool {\n\treturn v != nil && v.DiagnosticWorkflowExecution != nil\n}\n\ntype DomainAlreadyExistsError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a DomainAlreadyExistsError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainAlreadyExistsError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DomainAlreadyExistsError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainAlreadyExistsError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainAlreadyExistsError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainAlreadyExistsError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of DomainAlreadyExistsError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainAlreadyExistsError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainAlreadyExistsError struct could not be encoded.\nfunc (v *DomainAlreadyExistsError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DomainAlreadyExistsError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainAlreadyExistsError struct could not be generated from the wire\n// representation.\nfunc (v *DomainAlreadyExistsError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of DomainAlreadyExistsError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainAlreadyExistsError\n// struct.\nfunc (v *DomainAlreadyExistsError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"DomainAlreadyExistsError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*DomainAlreadyExistsError) ErrorName() string {\n\treturn \"DomainAlreadyExistsError\"\n}\n\n// Equals returns true if all the fields of this DomainAlreadyExistsError match the\n// provided DomainAlreadyExistsError.\n//\n// This function performs a deep comparison.\nfunc (v *DomainAlreadyExistsError) Equals(rhs *DomainAlreadyExistsError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainAlreadyExistsError.\nfunc (v *DomainAlreadyExistsError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *DomainAlreadyExistsError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *DomainAlreadyExistsError) Error() string {\n\treturn v.String()\n}\n\ntype DomainCacheInfo struct {\n\tNumOfItemsInCacheByID   *int64 `json:\"numOfItemsInCacheByID,omitempty\"`\n\tNumOfItemsInCacheByName *int64 `json:\"numOfItemsInCacheByName,omitempty\"`\n}\n\n// ToWire translates a DomainCacheInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainCacheInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NumOfItemsInCacheByID != nil {\n\t\tw, err = wire.NewValueI64(*(v.NumOfItemsInCacheByID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NumOfItemsInCacheByName != nil {\n\t\tw, err = wire.NewValueI64(*(v.NumOfItemsInCacheByName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DomainCacheInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainCacheInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainCacheInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainCacheInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NumOfItemsInCacheByID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NumOfItemsInCacheByName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainCacheInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainCacheInfo struct could not be encoded.\nfunc (v *DomainCacheInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NumOfItemsInCacheByID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NumOfItemsInCacheByID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NumOfItemsInCacheByName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NumOfItemsInCacheByName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DomainCacheInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainCacheInfo struct could not be generated from the wire\n// representation.\nfunc (v *DomainCacheInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NumOfItemsInCacheByID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NumOfItemsInCacheByName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainCacheInfo\n// struct.\nfunc (v *DomainCacheInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.NumOfItemsInCacheByID != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumOfItemsInCacheByID: %v\", *(v.NumOfItemsInCacheByID))\n\t\ti++\n\t}\n\tif v.NumOfItemsInCacheByName != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumOfItemsInCacheByName: %v\", *(v.NumOfItemsInCacheByName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainCacheInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DomainCacheInfo match the\n// provided DomainCacheInfo.\n//\n// This function performs a deep comparison.\nfunc (v *DomainCacheInfo) Equals(rhs *DomainCacheInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NumOfItemsInCacheByID, rhs.NumOfItemsInCacheByID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NumOfItemsInCacheByName, rhs.NumOfItemsInCacheByName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainCacheInfo.\nfunc (v *DomainCacheInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NumOfItemsInCacheByID != nil {\n\t\tenc.AddInt64(\"numOfItemsInCacheByID\", *v.NumOfItemsInCacheByID)\n\t}\n\tif v.NumOfItemsInCacheByName != nil {\n\t\tenc.AddInt64(\"numOfItemsInCacheByName\", *v.NumOfItemsInCacheByName)\n\t}\n\treturn err\n}\n\n// GetNumOfItemsInCacheByID returns the value of NumOfItemsInCacheByID if it is set or its\n// zero value if it is unset.\nfunc (v *DomainCacheInfo) GetNumOfItemsInCacheByID() (o int64) {\n\tif v != nil && v.NumOfItemsInCacheByID != nil {\n\t\treturn *v.NumOfItemsInCacheByID\n\t}\n\n\treturn\n}\n\n// IsSetNumOfItemsInCacheByID returns true if NumOfItemsInCacheByID is not nil.\nfunc (v *DomainCacheInfo) IsSetNumOfItemsInCacheByID() bool {\n\treturn v != nil && v.NumOfItemsInCacheByID != nil\n}\n\n// GetNumOfItemsInCacheByName returns the value of NumOfItemsInCacheByName if it is set or its\n// zero value if it is unset.\nfunc (v *DomainCacheInfo) GetNumOfItemsInCacheByName() (o int64) {\n\tif v != nil && v.NumOfItemsInCacheByName != nil {\n\t\treturn *v.NumOfItemsInCacheByName\n\t}\n\n\treturn\n}\n\n// IsSetNumOfItemsInCacheByName returns true if NumOfItemsInCacheByName is not nil.\nfunc (v *DomainCacheInfo) IsSetNumOfItemsInCacheByName() bool {\n\treturn v != nil && v.NumOfItemsInCacheByName != nil\n}\n\ntype DomainConfiguration struct {\n\tWorkflowExecutionRetentionPeriodInDays *int32                       `json:\"workflowExecutionRetentionPeriodInDays,omitempty\"`\n\tEmitMetric                             *bool                        `json:\"emitMetric,omitempty\"`\n\tIsolationgroups                        *IsolationGroupConfiguration `json:\"isolationgroups,omitempty\"`\n\tBadBinaries                            *BadBinaries                 `json:\"badBinaries,omitempty\"`\n\tHistoryArchivalStatus                  *ArchivalStatus              `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                     *string                      `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus               *ArchivalStatus              `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                  *string                      `json:\"visibilityArchivalURI,omitempty\"`\n\tAsyncWorkflowConfiguration             *AsyncWorkflowConfiguration  `json:\"AsyncWorkflowConfiguration,omitempty\"`\n}\n\n// ToWire translates a DomainConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowExecutionRetentionPeriodInDays)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.EmitMetric != nil {\n\t\tw, err = wire.NewValueBool(*(v.EmitMetric)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Isolationgroups != nil {\n\t\tw, err = v.Isolationgroups.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.BadBinaries != nil {\n\t\tw, err = v.BadBinaries.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tw, err = v.HistoryArchivalStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.HistoryArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tw, err = v.VisibilityArchivalStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.VisibilityArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tw, err = v.AsyncWorkflowConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IsolationGroupConfiguration_Read(w wire.Value) (*IsolationGroupConfiguration, error) {\n\tvar v IsolationGroupConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _BadBinaries_Read(w wire.Value) (*BadBinaries, error) {\n\tvar v BadBinaries\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ArchivalStatus_Read(w wire.Value) (ArchivalStatus, error) {\n\tvar v ArchivalStatus\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _AsyncWorkflowConfiguration_Read(w wire.Value) (*AsyncWorkflowConfiguration, error) {\n\tvar v AsyncWorkflowConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DomainConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowExecutionRetentionPeriodInDays = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.EmitMetric = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Isolationgroups, err = _IsolationGroupConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.BadBinaries, err = _BadBinaries_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ArchivalStatus\n\t\t\t\tx, err = _ArchivalStatus_Read(field.Value)\n\t\t\t\tv.HistoryArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.HistoryArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ArchivalStatus\n\t\t\t\tx, err = _ArchivalStatus_Read(field.Value)\n\t\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.VisibilityArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AsyncWorkflowConfiguration, err = _AsyncWorkflowConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainConfiguration struct could not be encoded.\nfunc (v *DomainConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowExecutionRetentionPeriodInDays)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EmitMetric != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.EmitMetric)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Isolationgroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Isolationgroups.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadBinaries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.BadBinaries.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HistoryArchivalStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.HistoryArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VisibilityArchivalStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.VisibilityArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AsyncWorkflowConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IsolationGroupConfiguration_Decode(sr stream.Reader) (*IsolationGroupConfiguration, error) {\n\tvar v IsolationGroupConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _BadBinaries_Decode(sr stream.Reader) (*BadBinaries, error) {\n\tvar v BadBinaries\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ArchivalStatus_Decode(sr stream.Reader) (ArchivalStatus, error) {\n\tvar v ArchivalStatus\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _AsyncWorkflowConfiguration_Decode(sr stream.Reader) (*AsyncWorkflowConfiguration, error) {\n\tvar v AsyncWorkflowConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DomainConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *DomainConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowExecutionRetentionPeriodInDays = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.EmitMetric = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.Isolationgroups, err = _IsolationGroupConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.BadBinaries, err = _BadBinaries_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x ArchivalStatus\n\t\t\tx, err = _ArchivalStatus_Decode(sr)\n\t\t\tv.HistoryArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.HistoryArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x ArchivalStatus\n\t\t\tx, err = _ArchivalStatus_Decode(sr)\n\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.VisibilityArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.AsyncWorkflowConfiguration, err = _AsyncWorkflowConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainConfiguration\n// struct.\nfunc (v *DomainConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionRetentionPeriodInDays: %v\", *(v.WorkflowExecutionRetentionPeriodInDays))\n\t\ti++\n\t}\n\tif v.EmitMetric != nil {\n\t\tfields[i] = fmt.Sprintf(\"EmitMetric: %v\", *(v.EmitMetric))\n\t\ti++\n\t}\n\tif v.Isolationgroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"Isolationgroups: %v\", v.Isolationgroups)\n\t\ti++\n\t}\n\tif v.BadBinaries != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadBinaries: %v\", v.BadBinaries)\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalStatus: %v\", *(v.HistoryArchivalStatus))\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalURI: %v\", *(v.HistoryArchivalURI))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalStatus: %v\", *(v.VisibilityArchivalStatus))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalURI: %v\", *(v.VisibilityArchivalURI))\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"AsyncWorkflowConfiguration: %v\", v.AsyncWorkflowConfiguration)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ArchivalStatus_EqualsPtr(lhs, rhs *ArchivalStatus) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DomainConfiguration match the\n// provided DomainConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *DomainConfiguration) Equals(rhs *DomainConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowExecutionRetentionPeriodInDays, rhs.WorkflowExecutionRetentionPeriodInDays) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.EmitMetric, rhs.EmitMetric) {\n\t\treturn false\n\t}\n\tif !((v.Isolationgroups == nil && rhs.Isolationgroups == nil) || (v.Isolationgroups != nil && rhs.Isolationgroups != nil && v.Isolationgroups.Equals(rhs.Isolationgroups))) {\n\t\treturn false\n\t}\n\tif !((v.BadBinaries == nil && rhs.BadBinaries == nil) || (v.BadBinaries != nil && rhs.BadBinaries != nil && v.BadBinaries.Equals(rhs.BadBinaries))) {\n\t\treturn false\n\t}\n\tif !_ArchivalStatus_EqualsPtr(v.HistoryArchivalStatus, rhs.HistoryArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.HistoryArchivalURI, rhs.HistoryArchivalURI) {\n\t\treturn false\n\t}\n\tif !_ArchivalStatus_EqualsPtr(v.VisibilityArchivalStatus, rhs.VisibilityArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.VisibilityArchivalURI, rhs.VisibilityArchivalURI) {\n\t\treturn false\n\t}\n\tif !((v.AsyncWorkflowConfiguration == nil && rhs.AsyncWorkflowConfiguration == nil) || (v.AsyncWorkflowConfiguration != nil && rhs.AsyncWorkflowConfiguration != nil && v.AsyncWorkflowConfiguration.Equals(rhs.AsyncWorkflowConfiguration))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainConfiguration.\nfunc (v *DomainConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tenc.AddInt32(\"workflowExecutionRetentionPeriodInDays\", *v.WorkflowExecutionRetentionPeriodInDays)\n\t}\n\tif v.EmitMetric != nil {\n\t\tenc.AddBool(\"emitMetric\", *v.EmitMetric)\n\t}\n\tif v.Isolationgroups != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationgroups\", v.Isolationgroups))\n\t}\n\tif v.BadBinaries != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"badBinaries\", v.BadBinaries))\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"historyArchivalStatus\", *v.HistoryArchivalStatus))\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tenc.AddString(\"historyArchivalURI\", *v.HistoryArchivalURI)\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"visibilityArchivalStatus\", *v.VisibilityArchivalStatus))\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tenc.AddString(\"visibilityArchivalURI\", *v.VisibilityArchivalURI)\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"AsyncWorkflowConfiguration\", v.AsyncWorkflowConfiguration))\n\t}\n\treturn err\n}\n\n// GetWorkflowExecutionRetentionPeriodInDays returns the value of WorkflowExecutionRetentionPeriodInDays if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetWorkflowExecutionRetentionPeriodInDays() (o int32) {\n\tif v != nil && v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\treturn *v.WorkflowExecutionRetentionPeriodInDays\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionRetentionPeriodInDays returns true if WorkflowExecutionRetentionPeriodInDays is not nil.\nfunc (v *DomainConfiguration) IsSetWorkflowExecutionRetentionPeriodInDays() bool {\n\treturn v != nil && v.WorkflowExecutionRetentionPeriodInDays != nil\n}\n\n// GetEmitMetric returns the value of EmitMetric if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetEmitMetric() (o bool) {\n\tif v != nil && v.EmitMetric != nil {\n\t\treturn *v.EmitMetric\n\t}\n\n\treturn\n}\n\n// IsSetEmitMetric returns true if EmitMetric is not nil.\nfunc (v *DomainConfiguration) IsSetEmitMetric() bool {\n\treturn v != nil && v.EmitMetric != nil\n}\n\n// GetIsolationgroups returns the value of Isolationgroups if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetIsolationgroups() (o *IsolationGroupConfiguration) {\n\tif v != nil && v.Isolationgroups != nil {\n\t\treturn v.Isolationgroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationgroups returns true if Isolationgroups is not nil.\nfunc (v *DomainConfiguration) IsSetIsolationgroups() bool {\n\treturn v != nil && v.Isolationgroups != nil\n}\n\n// GetBadBinaries returns the value of BadBinaries if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetBadBinaries() (o *BadBinaries) {\n\tif v != nil && v.BadBinaries != nil {\n\t\treturn v.BadBinaries\n\t}\n\n\treturn\n}\n\n// IsSetBadBinaries returns true if BadBinaries is not nil.\nfunc (v *DomainConfiguration) IsSetBadBinaries() bool {\n\treturn v != nil && v.BadBinaries != nil\n}\n\n// GetHistoryArchivalStatus returns the value of HistoryArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetHistoryArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.HistoryArchivalStatus != nil {\n\t\treturn *v.HistoryArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalStatus returns true if HistoryArchivalStatus is not nil.\nfunc (v *DomainConfiguration) IsSetHistoryArchivalStatus() bool {\n\treturn v != nil && v.HistoryArchivalStatus != nil\n}\n\n// GetHistoryArchivalURI returns the value of HistoryArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetHistoryArchivalURI() (o string) {\n\tif v != nil && v.HistoryArchivalURI != nil {\n\t\treturn *v.HistoryArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalURI returns true if HistoryArchivalURI is not nil.\nfunc (v *DomainConfiguration) IsSetHistoryArchivalURI() bool {\n\treturn v != nil && v.HistoryArchivalURI != nil\n}\n\n// GetVisibilityArchivalStatus returns the value of VisibilityArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetVisibilityArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.VisibilityArchivalStatus != nil {\n\t\treturn *v.VisibilityArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalStatus returns true if VisibilityArchivalStatus is not nil.\nfunc (v *DomainConfiguration) IsSetVisibilityArchivalStatus() bool {\n\treturn v != nil && v.VisibilityArchivalStatus != nil\n}\n\n// GetVisibilityArchivalURI returns the value of VisibilityArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetVisibilityArchivalURI() (o string) {\n\tif v != nil && v.VisibilityArchivalURI != nil {\n\t\treturn *v.VisibilityArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalURI returns true if VisibilityArchivalURI is not nil.\nfunc (v *DomainConfiguration) IsSetVisibilityArchivalURI() bool {\n\treturn v != nil && v.VisibilityArchivalURI != nil\n}\n\n// GetAsyncWorkflowConfiguration returns the value of AsyncWorkflowConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DomainConfiguration) GetAsyncWorkflowConfiguration() (o *AsyncWorkflowConfiguration) {\n\tif v != nil && v.AsyncWorkflowConfiguration != nil {\n\t\treturn v.AsyncWorkflowConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetAsyncWorkflowConfiguration returns true if AsyncWorkflowConfiguration is not nil.\nfunc (v *DomainConfiguration) IsSetAsyncWorkflowConfiguration() bool {\n\treturn v != nil && v.AsyncWorkflowConfiguration != nil\n}\n\ntype DomainIDPredicateAttributes struct {\n\tDomainIDs   []string `json:\"domainIDs,omitempty\"`\n\tIsExclusive *bool    `json:\"isExclusive,omitempty\"`\n}\n\n// ToWire translates a DomainIDPredicateAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainIDPredicateAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainIDs != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.DomainIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.IsExclusive != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsExclusive)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DomainIDPredicateAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainIDPredicateAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainIDPredicateAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainIDPredicateAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.DomainIDs, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsExclusive = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainIDPredicateAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainIDPredicateAttributes struct could not be encoded.\nfunc (v *DomainIDPredicateAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.DomainIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsExclusive != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsExclusive)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DomainIDPredicateAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainIDPredicateAttributes struct could not be generated from the wire\n// representation.\nfunc (v *DomainIDPredicateAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.DomainIDs, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsExclusive = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainIDPredicateAttributes\n// struct.\nfunc (v *DomainIDPredicateAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DomainIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainIDs: %v\", v.DomainIDs)\n\t\ti++\n\t}\n\tif v.IsExclusive != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsExclusive: %v\", *(v.IsExclusive))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainIDPredicateAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this DomainIDPredicateAttributes match the\n// provided DomainIDPredicateAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *DomainIDPredicateAttributes) Equals(rhs *DomainIDPredicateAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainIDs == nil && rhs.DomainIDs == nil) || (v.DomainIDs != nil && rhs.DomainIDs != nil && _List_String_Equals(v.DomainIDs, rhs.DomainIDs))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsExclusive, rhs.IsExclusive) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainIDPredicateAttributes.\nfunc (v *DomainIDPredicateAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"domainIDs\", (_List_String_Zapper)(v.DomainIDs)))\n\t}\n\tif v.IsExclusive != nil {\n\t\tenc.AddBool(\"isExclusive\", *v.IsExclusive)\n\t}\n\treturn err\n}\n\n// GetDomainIDs returns the value of DomainIDs if it is set or its\n// zero value if it is unset.\nfunc (v *DomainIDPredicateAttributes) GetDomainIDs() (o []string) {\n\tif v != nil && v.DomainIDs != nil {\n\t\treturn v.DomainIDs\n\t}\n\n\treturn\n}\n\n// IsSetDomainIDs returns true if DomainIDs is not nil.\nfunc (v *DomainIDPredicateAttributes) IsSetDomainIDs() bool {\n\treturn v != nil && v.DomainIDs != nil\n}\n\n// GetIsExclusive returns the value of IsExclusive if it is set or its\n// zero value if it is unset.\nfunc (v *DomainIDPredicateAttributes) GetIsExclusive() (o bool) {\n\tif v != nil && v.IsExclusive != nil {\n\t\treturn *v.IsExclusive\n\t}\n\n\treturn\n}\n\n// IsSetIsExclusive returns true if IsExclusive is not nil.\nfunc (v *DomainIDPredicateAttributes) IsSetIsExclusive() bool {\n\treturn v != nil && v.IsExclusive != nil\n}\n\ntype DomainInfo struct {\n\tName        *string           `json:\"name,omitempty\"`\n\tStatus      *DomainStatus     `json:\"status,omitempty\"`\n\tDescription *string           `json:\"description,omitempty\"`\n\tOwnerEmail  *string           `json:\"ownerEmail,omitempty\"`\n\tData        map[string]string `json:\"data,omitempty\"`\n\tUUID        *string           `json:\"uuid,omitempty\"`\n}\n\n// ToWire translates a DomainInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tw, err = v.Status.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tw, err = wire.NewValueString(*(v.Description)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tw, err = wire.NewValueString(*(v.OwnerEmail)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.Data)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.UUID != nil {\n\t\tw, err = wire.NewValueString(*(v.UUID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DomainStatus_Read(w wire.Value) (DomainStatus, error) {\n\tvar v DomainStatus\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a DomainInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DomainStatus\n\t\t\t\tx, err = _DomainStatus_Read(field.Value)\n\t\t\t\tv.Status = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Description = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OwnerEmail = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Data, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.UUID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainInfo struct could not be encoded.\nfunc (v *DomainInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Status != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Status.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Description != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Description)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OwnerEmail != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OwnerEmail)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.Data, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UUID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.UUID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DomainStatus_Decode(sr stream.Reader) (DomainStatus, error) {\n\tvar v DomainStatus\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a DomainInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainInfo struct could not be generated from the wire\n// representation.\nfunc (v *DomainInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x DomainStatus\n\t\t\tx, err = _DomainStatus_Decode(sr)\n\t\t\tv.Status = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Description = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OwnerEmail = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TMap:\n\t\t\tv.Data, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.UUID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainInfo\n// struct.\nfunc (v *DomainInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tfields[i] = fmt.Sprintf(\"Status: %v\", *(v.Status))\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tfields[i] = fmt.Sprintf(\"Description: %v\", *(v.Description))\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tfields[i] = fmt.Sprintf(\"OwnerEmail: %v\", *(v.OwnerEmail))\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\tif v.UUID != nil {\n\t\tfields[i] = fmt.Sprintf(\"UUID: %v\", *(v.UUID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _DomainStatus_EqualsPtr(lhs, rhs *DomainStatus) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this DomainInfo match the\n// provided DomainInfo.\n//\n// This function performs a deep comparison.\nfunc (v *DomainInfo) Equals(rhs *DomainInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_DomainStatus_EqualsPtr(v.Status, rhs.Status) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Description, rhs.Description) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OwnerEmail, rhs.OwnerEmail) {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && _Map_String_String_Equals(v.Data, rhs.Data))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.UUID, rhs.UUID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainInfo.\nfunc (v *DomainInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Status != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"status\", *v.Status))\n\t}\n\tif v.Description != nil {\n\t\tenc.AddString(\"description\", *v.Description)\n\t}\n\tif v.OwnerEmail != nil {\n\t\tenc.AddString(\"ownerEmail\", *v.OwnerEmail)\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", (_Map_String_String_Zapper)(v.Data)))\n\t}\n\tif v.UUID != nil {\n\t\tenc.AddString(\"uuid\", *v.UUID)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DomainInfo) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetStatus returns the value of Status if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetStatus() (o DomainStatus) {\n\tif v != nil && v.Status != nil {\n\t\treturn *v.Status\n\t}\n\n\treturn\n}\n\n// IsSetStatus returns true if Status is not nil.\nfunc (v *DomainInfo) IsSetStatus() bool {\n\treturn v != nil && v.Status != nil\n}\n\n// GetDescription returns the value of Description if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetDescription() (o string) {\n\tif v != nil && v.Description != nil {\n\t\treturn *v.Description\n\t}\n\n\treturn\n}\n\n// IsSetDescription returns true if Description is not nil.\nfunc (v *DomainInfo) IsSetDescription() bool {\n\treturn v != nil && v.Description != nil\n}\n\n// GetOwnerEmail returns the value of OwnerEmail if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetOwnerEmail() (o string) {\n\tif v != nil && v.OwnerEmail != nil {\n\t\treturn *v.OwnerEmail\n\t}\n\n\treturn\n}\n\n// IsSetOwnerEmail returns true if OwnerEmail is not nil.\nfunc (v *DomainInfo) IsSetOwnerEmail() bool {\n\treturn v != nil && v.OwnerEmail != nil\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *DomainInfo) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\n// GetUUID returns the value of UUID if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetUUID() (o string) {\n\tif v != nil && v.UUID != nil {\n\t\treturn *v.UUID\n\t}\n\n\treturn\n}\n\n// IsSetUUID returns true if UUID is not nil.\nfunc (v *DomainInfo) IsSetUUID() bool {\n\treturn v != nil && v.UUID != nil\n}\n\ntype DomainNotActiveError struct {\n\tMessage        string   `json:\"message,required\"`\n\tDomainName     string   `json:\"domainName,required\"`\n\tCurrentCluster string   `json:\"currentCluster,required\"`\n\tActiveCluster  string   `json:\"activeCluster,required\"`\n\tActiveClusters []string `json:\"activeClusters,required\"`\n}\n\n// ToWire translates a DomainNotActiveError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainNotActiveError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.DomainName), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 2, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.CurrentCluster), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 3, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.ActiveCluster), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 4, Value: w}\n\ti++\n\n\tw, err = wire.NewValueList(_List_String_ValueList(v.ActiveClusters)), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 5, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a DomainNotActiveError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainNotActiveError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainNotActiveError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainNotActiveError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\tdomainNameIsSet := false\n\tcurrentClusterIsSet := false\n\tactiveClusterIsSet := false\n\tactiveClustersIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.DomainName, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tdomainNameIsSet = true\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CurrentCluster, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcurrentClusterIsSet = true\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ActiveCluster, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tactiveClusterIsSet = true\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ActiveClusters, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tactiveClustersIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of DomainNotActiveError is required\")\n\t}\n\n\tif !domainNameIsSet {\n\t\treturn errors.New(\"field DomainName of DomainNotActiveError is required\")\n\t}\n\n\tif !currentClusterIsSet {\n\t\treturn errors.New(\"field CurrentCluster of DomainNotActiveError is required\")\n\t}\n\n\tif !activeClusterIsSet {\n\t\treturn errors.New(\"field ActiveCluster of DomainNotActiveError is required\")\n\t}\n\n\tif !activeClustersIsSet {\n\t\treturn errors.New(\"field ActiveClusters of DomainNotActiveError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a DomainNotActiveError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainNotActiveError struct could not be encoded.\nfunc (v *DomainNotActiveError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.DomainName); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.CurrentCluster); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.ActiveCluster); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TList}); err != nil {\n\t\treturn err\n\t}\n\tif err := _List_String_Encode(v.ActiveClusters, sw); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a DomainNotActiveError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainNotActiveError struct could not be generated from the wire\n// representation.\nfunc (v *DomainNotActiveError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\tdomainNameIsSet := false\n\tcurrentClusterIsSet := false\n\tactiveClusterIsSet := false\n\tactiveClustersIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tv.DomainName, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tdomainNameIsSet = true\n\t\tcase fh.ID == 3 && fh.Type == wire.TBinary:\n\t\t\tv.CurrentCluster, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tcurrentClusterIsSet = true\n\t\tcase fh.ID == 4 && fh.Type == wire.TBinary:\n\t\t\tv.ActiveCluster, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tactiveClusterIsSet = true\n\t\tcase fh.ID == 5 && fh.Type == wire.TList:\n\t\t\tv.ActiveClusters, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tactiveClustersIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of DomainNotActiveError is required\")\n\t}\n\n\tif !domainNameIsSet {\n\t\treturn errors.New(\"field DomainName of DomainNotActiveError is required\")\n\t}\n\n\tif !currentClusterIsSet {\n\t\treturn errors.New(\"field CurrentCluster of DomainNotActiveError is required\")\n\t}\n\n\tif !activeClusterIsSet {\n\t\treturn errors.New(\"field ActiveCluster of DomainNotActiveError is required\")\n\t}\n\n\tif !activeClustersIsSet {\n\t\treturn errors.New(\"field ActiveClusters of DomainNotActiveError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainNotActiveError\n// struct.\nfunc (v *DomainNotActiveError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\tfields[i] = fmt.Sprintf(\"DomainName: %v\", v.DomainName)\n\ti++\n\tfields[i] = fmt.Sprintf(\"CurrentCluster: %v\", v.CurrentCluster)\n\ti++\n\tfields[i] = fmt.Sprintf(\"ActiveCluster: %v\", v.ActiveCluster)\n\ti++\n\tfields[i] = fmt.Sprintf(\"ActiveClusters: %v\", v.ActiveClusters)\n\ti++\n\n\treturn fmt.Sprintf(\"DomainNotActiveError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*DomainNotActiveError) ErrorName() string {\n\treturn \"DomainNotActiveError\"\n}\n\n// Equals returns true if all the fields of this DomainNotActiveError match the\n// provided DomainNotActiveError.\n//\n// This function performs a deep comparison.\nfunc (v *DomainNotActiveError) Equals(rhs *DomainNotActiveError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\tif !(v.DomainName == rhs.DomainName) {\n\t\treturn false\n\t}\n\tif !(v.CurrentCluster == rhs.CurrentCluster) {\n\t\treturn false\n\t}\n\tif !(v.ActiveCluster == rhs.ActiveCluster) {\n\t\treturn false\n\t}\n\tif !_List_String_Equals(v.ActiveClusters, rhs.ActiveClusters) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainNotActiveError.\nfunc (v *DomainNotActiveError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\tenc.AddString(\"domainName\", v.DomainName)\n\tenc.AddString(\"currentCluster\", v.CurrentCluster)\n\tenc.AddString(\"activeCluster\", v.ActiveCluster)\n\terr = multierr.Append(err, enc.AddArray(\"activeClusters\", (_List_String_Zapper)(v.ActiveClusters)))\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *DomainNotActiveError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\n// GetDomainName returns the value of DomainName if it is set or its\n// zero value if it is unset.\nfunc (v *DomainNotActiveError) GetDomainName() (o string) {\n\tif v != nil {\n\t\to = v.DomainName\n\t}\n\treturn\n}\n\n// GetCurrentCluster returns the value of CurrentCluster if it is set or its\n// zero value if it is unset.\nfunc (v *DomainNotActiveError) GetCurrentCluster() (o string) {\n\tif v != nil {\n\t\to = v.CurrentCluster\n\t}\n\treturn\n}\n\n// GetActiveCluster returns the value of ActiveCluster if it is set or its\n// zero value if it is unset.\nfunc (v *DomainNotActiveError) GetActiveCluster() (o string) {\n\tif v != nil {\n\t\to = v.ActiveCluster\n\t}\n\treturn\n}\n\n// GetActiveClusters returns the value of ActiveClusters if it is set or its\n// zero value if it is unset.\nfunc (v *DomainNotActiveError) GetActiveClusters() (o []string) {\n\tif v != nil {\n\t\to = v.ActiveClusters\n\t}\n\treturn\n}\n\n// IsSetActiveClusters returns true if ActiveClusters is not nil.\nfunc (v *DomainNotActiveError) IsSetActiveClusters() bool {\n\treturn v != nil && v.ActiveClusters != nil\n}\n\nfunc (v *DomainNotActiveError) Error() string {\n\treturn v.String()\n}\n\ntype DomainReplicationConfiguration struct {\n\tActiveClusterName *string                            `json:\"activeClusterName,omitempty\"`\n\tClusters          []*ClusterReplicationConfiguration `json:\"clusters,omitempty\"`\n\tActiveClusters    *ActiveClusters                    `json:\"activeClusters,omitempty\"`\n}\n\ntype _List_ClusterReplicationConfiguration_ValueList []*ClusterReplicationConfiguration\n\nfunc (v _List_ClusterReplicationConfiguration_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ClusterReplicationConfiguration', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ClusterReplicationConfiguration_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ClusterReplicationConfiguration_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ClusterReplicationConfiguration_ValueList) Close() {}\n\n// ToWire translates a DomainReplicationConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainReplicationConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActiveClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tw, err = wire.NewValueList(_List_ClusterReplicationConfiguration_ValueList(v.Clusters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tw, err = v.ActiveClusters.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ClusterReplicationConfiguration_Read(w wire.Value) (*ClusterReplicationConfiguration, error) {\n\tvar v ClusterReplicationConfiguration\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ClusterReplicationConfiguration_Read(l wire.ValueList) ([]*ClusterReplicationConfiguration, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ClusterReplicationConfiguration, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ClusterReplicationConfiguration_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _ActiveClusters_Read(w wire.Value) (*ActiveClusters, error) {\n\tvar v ActiveClusters\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a DomainReplicationConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainReplicationConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainReplicationConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainReplicationConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Clusters, err = _List_ClusterReplicationConfiguration_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusters, err = _ActiveClusters_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ClusterReplicationConfiguration_Encode(val []*ClusterReplicationConfiguration, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ClusterReplicationConfiguration', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a DomainReplicationConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainReplicationConfiguration struct could not be encoded.\nfunc (v *DomainReplicationConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActiveClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Clusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ClusterReplicationConfiguration_Encode(v.Clusters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusters.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ClusterReplicationConfiguration_Decode(sr stream.Reader) (*ClusterReplicationConfiguration, error) {\n\tvar v ClusterReplicationConfiguration\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ClusterReplicationConfiguration_Decode(sr stream.Reader) ([]*ClusterReplicationConfiguration, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ClusterReplicationConfiguration, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ClusterReplicationConfiguration_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _ActiveClusters_Decode(sr stream.Reader) (*ActiveClusters, error) {\n\tvar v ActiveClusters\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a DomainReplicationConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainReplicationConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *DomainReplicationConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Clusters, err = _List_ClusterReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusters, err = _ActiveClusters_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainReplicationConfiguration\n// struct.\nfunc (v *DomainReplicationConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ActiveClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterName: %v\", *(v.ActiveClusterName))\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Clusters: %v\", v.Clusters)\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusters: %v\", v.ActiveClusters)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainReplicationConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ClusterReplicationConfiguration_Equals(lhs, rhs []*ClusterReplicationConfiguration) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this DomainReplicationConfiguration match the\n// provided DomainReplicationConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *DomainReplicationConfiguration) Equals(rhs *DomainReplicationConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClusterName, rhs.ActiveClusterName) {\n\t\treturn false\n\t}\n\tif !((v.Clusters == nil && rhs.Clusters == nil) || (v.Clusters != nil && rhs.Clusters != nil && _List_ClusterReplicationConfiguration_Equals(v.Clusters, rhs.Clusters))) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusters == nil && rhs.ActiveClusters == nil) || (v.ActiveClusters != nil && rhs.ActiveClusters != nil && v.ActiveClusters.Equals(rhs.ActiveClusters))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ClusterReplicationConfiguration_Zapper []*ClusterReplicationConfiguration\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ClusterReplicationConfiguration_Zapper.\nfunc (l _List_ClusterReplicationConfiguration_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainReplicationConfiguration.\nfunc (v *DomainReplicationConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tenc.AddString(\"activeClusterName\", *v.ActiveClusterName)\n\t}\n\tif v.Clusters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"clusters\", (_List_ClusterReplicationConfiguration_Zapper)(v.Clusters)))\n\t}\n\tif v.ActiveClusters != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusters\", v.ActiveClusters))\n\t}\n\treturn err\n}\n\n// GetActiveClusterName returns the value of ActiveClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *DomainReplicationConfiguration) GetActiveClusterName() (o string) {\n\tif v != nil && v.ActiveClusterName != nil {\n\t\treturn *v.ActiveClusterName\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterName returns true if ActiveClusterName is not nil.\nfunc (v *DomainReplicationConfiguration) IsSetActiveClusterName() bool {\n\treturn v != nil && v.ActiveClusterName != nil\n}\n\n// GetClusters returns the value of Clusters if it is set or its\n// zero value if it is unset.\nfunc (v *DomainReplicationConfiguration) GetClusters() (o []*ClusterReplicationConfiguration) {\n\tif v != nil && v.Clusters != nil {\n\t\treturn v.Clusters\n\t}\n\n\treturn\n}\n\n// IsSetClusters returns true if Clusters is not nil.\nfunc (v *DomainReplicationConfiguration) IsSetClusters() bool {\n\treturn v != nil && v.Clusters != nil\n}\n\n// GetActiveClusters returns the value of ActiveClusters if it is set or its\n// zero value if it is unset.\nfunc (v *DomainReplicationConfiguration) GetActiveClusters() (o *ActiveClusters) {\n\tif v != nil && v.ActiveClusters != nil {\n\t\treturn v.ActiveClusters\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusters returns true if ActiveClusters is not nil.\nfunc (v *DomainReplicationConfiguration) IsSetActiveClusters() bool {\n\treturn v != nil && v.ActiveClusters != nil\n}\n\ntype DomainStatus int32\n\nconst (\n\tDomainStatusRegistered DomainStatus = 0\n\tDomainStatusDeprecated DomainStatus = 1\n\tDomainStatusDeleted    DomainStatus = 2\n)\n\n// DomainStatus_Values returns all recognized values of DomainStatus.\nfunc DomainStatus_Values() []DomainStatus {\n\treturn []DomainStatus{\n\t\tDomainStatusRegistered,\n\t\tDomainStatusDeprecated,\n\t\tDomainStatusDeleted,\n\t}\n}\n\n// UnmarshalText tries to decode DomainStatus from a byte slice\n// containing its name.\n//\n//\tvar v DomainStatus\n//\terr := v.UnmarshalText([]byte(\"REGISTERED\"))\nfunc (v *DomainStatus) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"REGISTERED\":\n\t\t*v = DomainStatusRegistered\n\t\treturn nil\n\tcase \"DEPRECATED\":\n\t\t*v = DomainStatusDeprecated\n\t\treturn nil\n\tcase \"DELETED\":\n\t\t*v = DomainStatusDeleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DomainStatus\", err)\n\t\t}\n\t\t*v = DomainStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DomainStatus to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v DomainStatus) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"REGISTERED\"), nil\n\tcase 1:\n\t\treturn []byte(\"DEPRECATED\"), nil\n\tcase 2:\n\t\treturn []byte(\"DELETED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainStatus.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v DomainStatus) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"REGISTERED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"DEPRECATED\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"DELETED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v DomainStatus) Ptr() *DomainStatus {\n\treturn &v\n}\n\n// Encode encodes DomainStatus directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v DomainStatus\n//\treturn v.Encode(sWriter)\nfunc (v DomainStatus) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates DomainStatus into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v DomainStatus) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes DomainStatus from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return DomainStatus(0), err\n//\t}\n//\n//\tvar v DomainStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return DomainStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DomainStatus) FromWire(w wire.Value) error {\n\t*v = (DomainStatus)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded DomainStatus directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v DomainStatus\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return DomainStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *DomainStatus) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (DomainStatus)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of DomainStatus.\nfunc (v DomainStatus) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"REGISTERED\"\n\tcase 1:\n\t\treturn \"DEPRECATED\"\n\tcase 2:\n\t\treturn \"DELETED\"\n\t}\n\treturn fmt.Sprintf(\"DomainStatus(%d)\", w)\n}\n\n// Equals returns true if this DomainStatus value matches the provided\n// value.\nfunc (v DomainStatus) Equals(rhs DomainStatus) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes DomainStatus into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v DomainStatus) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"REGISTERED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"DEPRECATED\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"DELETED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode DomainStatus from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *DomainStatus) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"DomainStatus\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"DomainStatus\")\n\t\t}\n\t\t*v = (DomainStatus)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"DomainStatus\")\n\t}\n}\n\ntype EmptyPredicateAttributes struct {\n}\n\n// ToWire translates a EmptyPredicateAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *EmptyPredicateAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a EmptyPredicateAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a EmptyPredicateAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v EmptyPredicateAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *EmptyPredicateAttributes) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a EmptyPredicateAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a EmptyPredicateAttributes struct could not be encoded.\nfunc (v *EmptyPredicateAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a EmptyPredicateAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a EmptyPredicateAttributes struct could not be generated from the wire\n// representation.\nfunc (v *EmptyPredicateAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a EmptyPredicateAttributes\n// struct.\nfunc (v *EmptyPredicateAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"EmptyPredicateAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this EmptyPredicateAttributes match the\n// provided EmptyPredicateAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *EmptyPredicateAttributes) Equals(rhs *EmptyPredicateAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of EmptyPredicateAttributes.\nfunc (v *EmptyPredicateAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype EncodingType int32\n\nconst (\n\tEncodingTypeThriftRW EncodingType = 0\n\tEncodingTypeJSON     EncodingType = 1\n)\n\n// EncodingType_Values returns all recognized values of EncodingType.\nfunc EncodingType_Values() []EncodingType {\n\treturn []EncodingType{\n\t\tEncodingTypeThriftRW,\n\t\tEncodingTypeJSON,\n\t}\n}\n\n// UnmarshalText tries to decode EncodingType from a byte slice\n// containing its name.\n//\n//\tvar v EncodingType\n//\terr := v.UnmarshalText([]byte(\"ThriftRW\"))\nfunc (v *EncodingType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"ThriftRW\":\n\t\t*v = EncodingTypeThriftRW\n\t\treturn nil\n\tcase \"JSON\":\n\t\t*v = EncodingTypeJSON\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"EncodingType\", err)\n\t\t}\n\t\t*v = EncodingType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes EncodingType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v EncodingType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"ThriftRW\"), nil\n\tcase 1:\n\t\treturn []byte(\"JSON\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of EncodingType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v EncodingType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"ThriftRW\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"JSON\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v EncodingType) Ptr() *EncodingType {\n\treturn &v\n}\n\n// Encode encodes EncodingType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v EncodingType\n//\treturn v.Encode(sWriter)\nfunc (v EncodingType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates EncodingType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v EncodingType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes EncodingType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return EncodingType(0), err\n//\t}\n//\n//\tvar v EncodingType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return EncodingType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *EncodingType) FromWire(w wire.Value) error {\n\t*v = (EncodingType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded EncodingType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v EncodingType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return EncodingType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *EncodingType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (EncodingType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of EncodingType.\nfunc (v EncodingType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ThriftRW\"\n\tcase 1:\n\t\treturn \"JSON\"\n\t}\n\treturn fmt.Sprintf(\"EncodingType(%d)\", w)\n}\n\n// Equals returns true if this EncodingType value matches the provided\n// value.\nfunc (v EncodingType) Equals(rhs EncodingType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes EncodingType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v EncodingType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"ThriftRW\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"JSON\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode EncodingType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *EncodingType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"EncodingType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"EncodingType\")\n\t\t}\n\t\t*v = (EncodingType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"EncodingType\")\n\t}\n}\n\ntype EntityNotExistsError struct {\n\tMessage        string   `json:\"message,required\"`\n\tCurrentCluster *string  `json:\"currentCluster,omitempty\"`\n\tActiveCluster  *string  `json:\"activeCluster,omitempty\"`\n\tActiveClusters []string `json:\"activeClusters,required\"`\n}\n\n// ToWire translates a EntityNotExistsError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *EntityNotExistsError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\tif v.CurrentCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.CurrentCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\n\tw, err = wire.NewValueList(_List_String_ValueList(v.ActiveClusters)), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 4, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a EntityNotExistsError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a EntityNotExistsError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v EntityNotExistsError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *EntityNotExistsError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tactiveClustersIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CurrentCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ActiveClusters, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tactiveClustersIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of EntityNotExistsError is required\")\n\t}\n\n\tif !activeClustersIsSet {\n\t\treturn errors.New(\"field ActiveClusters of EntityNotExistsError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a EntityNotExistsError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a EntityNotExistsError struct could not be encoded.\nfunc (v *EntityNotExistsError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CurrentCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CurrentCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TList}); err != nil {\n\t\treturn err\n\t}\n\tif err := _List_String_Encode(v.ActiveClusters, sw); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a EntityNotExistsError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a EntityNotExistsError struct could not be generated from the wire\n// representation.\nfunc (v *EntityNotExistsError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tactiveClustersIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CurrentCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TList:\n\t\t\tv.ActiveClusters, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tactiveClustersIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of EntityNotExistsError is required\")\n\t}\n\n\tif !activeClustersIsSet {\n\t\treturn errors.New(\"field ActiveClusters of EntityNotExistsError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a EntityNotExistsError\n// struct.\nfunc (v *EntityNotExistsError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\tif v.CurrentCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentCluster: %v\", *(v.CurrentCluster))\n\t\ti++\n\t}\n\tif v.ActiveCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveCluster: %v\", *(v.ActiveCluster))\n\t\ti++\n\t}\n\tfields[i] = fmt.Sprintf(\"ActiveClusters: %v\", v.ActiveClusters)\n\ti++\n\n\treturn fmt.Sprintf(\"EntityNotExistsError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*EntityNotExistsError) ErrorName() string {\n\treturn \"EntityNotExistsError\"\n}\n\n// Equals returns true if all the fields of this EntityNotExistsError match the\n// provided EntityNotExistsError.\n//\n// This function performs a deep comparison.\nfunc (v *EntityNotExistsError) Equals(rhs *EntityNotExistsError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CurrentCluster, rhs.CurrentCluster) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveCluster, rhs.ActiveCluster) {\n\t\treturn false\n\t}\n\tif !_List_String_Equals(v.ActiveClusters, rhs.ActiveClusters) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of EntityNotExistsError.\nfunc (v *EntityNotExistsError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\tif v.CurrentCluster != nil {\n\t\tenc.AddString(\"currentCluster\", *v.CurrentCluster)\n\t}\n\tif v.ActiveCluster != nil {\n\t\tenc.AddString(\"activeCluster\", *v.ActiveCluster)\n\t}\n\terr = multierr.Append(err, enc.AddArray(\"activeClusters\", (_List_String_Zapper)(v.ActiveClusters)))\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *EntityNotExistsError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\n// GetCurrentCluster returns the value of CurrentCluster if it is set or its\n// zero value if it is unset.\nfunc (v *EntityNotExistsError) GetCurrentCluster() (o string) {\n\tif v != nil && v.CurrentCluster != nil {\n\t\treturn *v.CurrentCluster\n\t}\n\n\treturn\n}\n\n// IsSetCurrentCluster returns true if CurrentCluster is not nil.\nfunc (v *EntityNotExistsError) IsSetCurrentCluster() bool {\n\treturn v != nil && v.CurrentCluster != nil\n}\n\n// GetActiveCluster returns the value of ActiveCluster if it is set or its\n// zero value if it is unset.\nfunc (v *EntityNotExistsError) GetActiveCluster() (o string) {\n\tif v != nil && v.ActiveCluster != nil {\n\t\treturn *v.ActiveCluster\n\t}\n\n\treturn\n}\n\n// IsSetActiveCluster returns true if ActiveCluster is not nil.\nfunc (v *EntityNotExistsError) IsSetActiveCluster() bool {\n\treturn v != nil && v.ActiveCluster != nil\n}\n\n// GetActiveClusters returns the value of ActiveClusters if it is set or its\n// zero value if it is unset.\nfunc (v *EntityNotExistsError) GetActiveClusters() (o []string) {\n\tif v != nil {\n\t\to = v.ActiveClusters\n\t}\n\treturn\n}\n\n// IsSetActiveClusters returns true if ActiveClusters is not nil.\nfunc (v *EntityNotExistsError) IsSetActiveClusters() bool {\n\treturn v != nil && v.ActiveClusters != nil\n}\n\nfunc (v *EntityNotExistsError) Error() string {\n\treturn v.String()\n}\n\ntype EventType int32\n\nconst (\n\tEventTypeWorkflowExecutionStarted                        EventType = 0\n\tEventTypeWorkflowExecutionCompleted                      EventType = 1\n\tEventTypeWorkflowExecutionFailed                         EventType = 2\n\tEventTypeWorkflowExecutionTimedOut                       EventType = 3\n\tEventTypeDecisionTaskScheduled                           EventType = 4\n\tEventTypeDecisionTaskStarted                             EventType = 5\n\tEventTypeDecisionTaskCompleted                           EventType = 6\n\tEventTypeDecisionTaskTimedOut                            EventType = 7\n\tEventTypeDecisionTaskFailed                              EventType = 8\n\tEventTypeActivityTaskScheduled                           EventType = 9\n\tEventTypeActivityTaskStarted                             EventType = 10\n\tEventTypeActivityTaskCompleted                           EventType = 11\n\tEventTypeActivityTaskFailed                              EventType = 12\n\tEventTypeActivityTaskTimedOut                            EventType = 13\n\tEventTypeActivityTaskCancelRequested                     EventType = 14\n\tEventTypeRequestCancelActivityTaskFailed                 EventType = 15\n\tEventTypeActivityTaskCanceled                            EventType = 16\n\tEventTypeTimerStarted                                    EventType = 17\n\tEventTypeTimerFired                                      EventType = 18\n\tEventTypeCancelTimerFailed                               EventType = 19\n\tEventTypeTimerCanceled                                   EventType = 20\n\tEventTypeWorkflowExecutionCancelRequested                EventType = 21\n\tEventTypeWorkflowExecutionCanceled                       EventType = 22\n\tEventTypeRequestCancelExternalWorkflowExecutionInitiated EventType = 23\n\tEventTypeRequestCancelExternalWorkflowExecutionFailed    EventType = 24\n\tEventTypeExternalWorkflowExecutionCancelRequested        EventType = 25\n\tEventTypeMarkerRecorded                                  EventType = 26\n\tEventTypeWorkflowExecutionSignaled                       EventType = 27\n\tEventTypeWorkflowExecutionTerminated                     EventType = 28\n\tEventTypeWorkflowExecutionContinuedAsNew                 EventType = 29\n\tEventTypeStartChildWorkflowExecutionInitiated            EventType = 30\n\tEventTypeStartChildWorkflowExecutionFailed               EventType = 31\n\tEventTypeChildWorkflowExecutionStarted                   EventType = 32\n\tEventTypeChildWorkflowExecutionCompleted                 EventType = 33\n\tEventTypeChildWorkflowExecutionFailed                    EventType = 34\n\tEventTypeChildWorkflowExecutionCanceled                  EventType = 35\n\tEventTypeChildWorkflowExecutionTimedOut                  EventType = 36\n\tEventTypeChildWorkflowExecutionTerminated                EventType = 37\n\tEventTypeSignalExternalWorkflowExecutionInitiated        EventType = 38\n\tEventTypeSignalExternalWorkflowExecutionFailed           EventType = 39\n\tEventTypeExternalWorkflowExecutionSignaled               EventType = 40\n\tEventTypeUpsertWorkflowSearchAttributes                  EventType = 41\n)\n\n// EventType_Values returns all recognized values of EventType.\nfunc EventType_Values() []EventType {\n\treturn []EventType{\n\t\tEventTypeWorkflowExecutionStarted,\n\t\tEventTypeWorkflowExecutionCompleted,\n\t\tEventTypeWorkflowExecutionFailed,\n\t\tEventTypeWorkflowExecutionTimedOut,\n\t\tEventTypeDecisionTaskScheduled,\n\t\tEventTypeDecisionTaskStarted,\n\t\tEventTypeDecisionTaskCompleted,\n\t\tEventTypeDecisionTaskTimedOut,\n\t\tEventTypeDecisionTaskFailed,\n\t\tEventTypeActivityTaskScheduled,\n\t\tEventTypeActivityTaskStarted,\n\t\tEventTypeActivityTaskCompleted,\n\t\tEventTypeActivityTaskFailed,\n\t\tEventTypeActivityTaskTimedOut,\n\t\tEventTypeActivityTaskCancelRequested,\n\t\tEventTypeRequestCancelActivityTaskFailed,\n\t\tEventTypeActivityTaskCanceled,\n\t\tEventTypeTimerStarted,\n\t\tEventTypeTimerFired,\n\t\tEventTypeCancelTimerFailed,\n\t\tEventTypeTimerCanceled,\n\t\tEventTypeWorkflowExecutionCancelRequested,\n\t\tEventTypeWorkflowExecutionCanceled,\n\t\tEventTypeRequestCancelExternalWorkflowExecutionInitiated,\n\t\tEventTypeRequestCancelExternalWorkflowExecutionFailed,\n\t\tEventTypeExternalWorkflowExecutionCancelRequested,\n\t\tEventTypeMarkerRecorded,\n\t\tEventTypeWorkflowExecutionSignaled,\n\t\tEventTypeWorkflowExecutionTerminated,\n\t\tEventTypeWorkflowExecutionContinuedAsNew,\n\t\tEventTypeStartChildWorkflowExecutionInitiated,\n\t\tEventTypeStartChildWorkflowExecutionFailed,\n\t\tEventTypeChildWorkflowExecutionStarted,\n\t\tEventTypeChildWorkflowExecutionCompleted,\n\t\tEventTypeChildWorkflowExecutionFailed,\n\t\tEventTypeChildWorkflowExecutionCanceled,\n\t\tEventTypeChildWorkflowExecutionTimedOut,\n\t\tEventTypeChildWorkflowExecutionTerminated,\n\t\tEventTypeSignalExternalWorkflowExecutionInitiated,\n\t\tEventTypeSignalExternalWorkflowExecutionFailed,\n\t\tEventTypeExternalWorkflowExecutionSignaled,\n\t\tEventTypeUpsertWorkflowSearchAttributes,\n\t}\n}\n\n// UnmarshalText tries to decode EventType from a byte slice\n// containing its name.\n//\n//\tvar v EventType\n//\terr := v.UnmarshalText([]byte(\"WorkflowExecutionStarted\"))\nfunc (v *EventType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"WorkflowExecutionStarted\":\n\t\t*v = EventTypeWorkflowExecutionStarted\n\t\treturn nil\n\tcase \"WorkflowExecutionCompleted\":\n\t\t*v = EventTypeWorkflowExecutionCompleted\n\t\treturn nil\n\tcase \"WorkflowExecutionFailed\":\n\t\t*v = EventTypeWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"WorkflowExecutionTimedOut\":\n\t\t*v = EventTypeWorkflowExecutionTimedOut\n\t\treturn nil\n\tcase \"DecisionTaskScheduled\":\n\t\t*v = EventTypeDecisionTaskScheduled\n\t\treturn nil\n\tcase \"DecisionTaskStarted\":\n\t\t*v = EventTypeDecisionTaskStarted\n\t\treturn nil\n\tcase \"DecisionTaskCompleted\":\n\t\t*v = EventTypeDecisionTaskCompleted\n\t\treturn nil\n\tcase \"DecisionTaskTimedOut\":\n\t\t*v = EventTypeDecisionTaskTimedOut\n\t\treturn nil\n\tcase \"DecisionTaskFailed\":\n\t\t*v = EventTypeDecisionTaskFailed\n\t\treturn nil\n\tcase \"ActivityTaskScheduled\":\n\t\t*v = EventTypeActivityTaskScheduled\n\t\treturn nil\n\tcase \"ActivityTaskStarted\":\n\t\t*v = EventTypeActivityTaskStarted\n\t\treturn nil\n\tcase \"ActivityTaskCompleted\":\n\t\t*v = EventTypeActivityTaskCompleted\n\t\treturn nil\n\tcase \"ActivityTaskFailed\":\n\t\t*v = EventTypeActivityTaskFailed\n\t\treturn nil\n\tcase \"ActivityTaskTimedOut\":\n\t\t*v = EventTypeActivityTaskTimedOut\n\t\treturn nil\n\tcase \"ActivityTaskCancelRequested\":\n\t\t*v = EventTypeActivityTaskCancelRequested\n\t\treturn nil\n\tcase \"RequestCancelActivityTaskFailed\":\n\t\t*v = EventTypeRequestCancelActivityTaskFailed\n\t\treturn nil\n\tcase \"ActivityTaskCanceled\":\n\t\t*v = EventTypeActivityTaskCanceled\n\t\treturn nil\n\tcase \"TimerStarted\":\n\t\t*v = EventTypeTimerStarted\n\t\treturn nil\n\tcase \"TimerFired\":\n\t\t*v = EventTypeTimerFired\n\t\treturn nil\n\tcase \"CancelTimerFailed\":\n\t\t*v = EventTypeCancelTimerFailed\n\t\treturn nil\n\tcase \"TimerCanceled\":\n\t\t*v = EventTypeTimerCanceled\n\t\treturn nil\n\tcase \"WorkflowExecutionCancelRequested\":\n\t\t*v = EventTypeWorkflowExecutionCancelRequested\n\t\treturn nil\n\tcase \"WorkflowExecutionCanceled\":\n\t\t*v = EventTypeWorkflowExecutionCanceled\n\t\treturn nil\n\tcase \"RequestCancelExternalWorkflowExecutionInitiated\":\n\t\t*v = EventTypeRequestCancelExternalWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"RequestCancelExternalWorkflowExecutionFailed\":\n\t\t*v = EventTypeRequestCancelExternalWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"ExternalWorkflowExecutionCancelRequested\":\n\t\t*v = EventTypeExternalWorkflowExecutionCancelRequested\n\t\treturn nil\n\tcase \"MarkerRecorded\":\n\t\t*v = EventTypeMarkerRecorded\n\t\treturn nil\n\tcase \"WorkflowExecutionSignaled\":\n\t\t*v = EventTypeWorkflowExecutionSignaled\n\t\treturn nil\n\tcase \"WorkflowExecutionTerminated\":\n\t\t*v = EventTypeWorkflowExecutionTerminated\n\t\treturn nil\n\tcase \"WorkflowExecutionContinuedAsNew\":\n\t\t*v = EventTypeWorkflowExecutionContinuedAsNew\n\t\treturn nil\n\tcase \"StartChildWorkflowExecutionInitiated\":\n\t\t*v = EventTypeStartChildWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"StartChildWorkflowExecutionFailed\":\n\t\t*v = EventTypeStartChildWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionStarted\":\n\t\t*v = EventTypeChildWorkflowExecutionStarted\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionCompleted\":\n\t\t*v = EventTypeChildWorkflowExecutionCompleted\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionFailed\":\n\t\t*v = EventTypeChildWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionCanceled\":\n\t\t*v = EventTypeChildWorkflowExecutionCanceled\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionTimedOut\":\n\t\t*v = EventTypeChildWorkflowExecutionTimedOut\n\t\treturn nil\n\tcase \"ChildWorkflowExecutionTerminated\":\n\t\t*v = EventTypeChildWorkflowExecutionTerminated\n\t\treturn nil\n\tcase \"SignalExternalWorkflowExecutionInitiated\":\n\t\t*v = EventTypeSignalExternalWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"SignalExternalWorkflowExecutionFailed\":\n\t\t*v = EventTypeSignalExternalWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"ExternalWorkflowExecutionSignaled\":\n\t\t*v = EventTypeExternalWorkflowExecutionSignaled\n\t\treturn nil\n\tcase \"UpsertWorkflowSearchAttributes\":\n\t\t*v = EventTypeUpsertWorkflowSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"EventType\", err)\n\t\t}\n\t\t*v = EventType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes EventType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v EventType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"WorkflowExecutionStarted\"), nil\n\tcase 1:\n\t\treturn []byte(\"WorkflowExecutionCompleted\"), nil\n\tcase 2:\n\t\treturn []byte(\"WorkflowExecutionFailed\"), nil\n\tcase 3:\n\t\treturn []byte(\"WorkflowExecutionTimedOut\"), nil\n\tcase 4:\n\t\treturn []byte(\"DecisionTaskScheduled\"), nil\n\tcase 5:\n\t\treturn []byte(\"DecisionTaskStarted\"), nil\n\tcase 6:\n\t\treturn []byte(\"DecisionTaskCompleted\"), nil\n\tcase 7:\n\t\treturn []byte(\"DecisionTaskTimedOut\"), nil\n\tcase 8:\n\t\treturn []byte(\"DecisionTaskFailed\"), nil\n\tcase 9:\n\t\treturn []byte(\"ActivityTaskScheduled\"), nil\n\tcase 10:\n\t\treturn []byte(\"ActivityTaskStarted\"), nil\n\tcase 11:\n\t\treturn []byte(\"ActivityTaskCompleted\"), nil\n\tcase 12:\n\t\treturn []byte(\"ActivityTaskFailed\"), nil\n\tcase 13:\n\t\treturn []byte(\"ActivityTaskTimedOut\"), nil\n\tcase 14:\n\t\treturn []byte(\"ActivityTaskCancelRequested\"), nil\n\tcase 15:\n\t\treturn []byte(\"RequestCancelActivityTaskFailed\"), nil\n\tcase 16:\n\t\treturn []byte(\"ActivityTaskCanceled\"), nil\n\tcase 17:\n\t\treturn []byte(\"TimerStarted\"), nil\n\tcase 18:\n\t\treturn []byte(\"TimerFired\"), nil\n\tcase 19:\n\t\treturn []byte(\"CancelTimerFailed\"), nil\n\tcase 20:\n\t\treturn []byte(\"TimerCanceled\"), nil\n\tcase 21:\n\t\treturn []byte(\"WorkflowExecutionCancelRequested\"), nil\n\tcase 22:\n\t\treturn []byte(\"WorkflowExecutionCanceled\"), nil\n\tcase 23:\n\t\treturn []byte(\"RequestCancelExternalWorkflowExecutionInitiated\"), nil\n\tcase 24:\n\t\treturn []byte(\"RequestCancelExternalWorkflowExecutionFailed\"), nil\n\tcase 25:\n\t\treturn []byte(\"ExternalWorkflowExecutionCancelRequested\"), nil\n\tcase 26:\n\t\treturn []byte(\"MarkerRecorded\"), nil\n\tcase 27:\n\t\treturn []byte(\"WorkflowExecutionSignaled\"), nil\n\tcase 28:\n\t\treturn []byte(\"WorkflowExecutionTerminated\"), nil\n\tcase 29:\n\t\treturn []byte(\"WorkflowExecutionContinuedAsNew\"), nil\n\tcase 30:\n\t\treturn []byte(\"StartChildWorkflowExecutionInitiated\"), nil\n\tcase 31:\n\t\treturn []byte(\"StartChildWorkflowExecutionFailed\"), nil\n\tcase 32:\n\t\treturn []byte(\"ChildWorkflowExecutionStarted\"), nil\n\tcase 33:\n\t\treturn []byte(\"ChildWorkflowExecutionCompleted\"), nil\n\tcase 34:\n\t\treturn []byte(\"ChildWorkflowExecutionFailed\"), nil\n\tcase 35:\n\t\treturn []byte(\"ChildWorkflowExecutionCanceled\"), nil\n\tcase 36:\n\t\treturn []byte(\"ChildWorkflowExecutionTimedOut\"), nil\n\tcase 37:\n\t\treturn []byte(\"ChildWorkflowExecutionTerminated\"), nil\n\tcase 38:\n\t\treturn []byte(\"SignalExternalWorkflowExecutionInitiated\"), nil\n\tcase 39:\n\t\treturn []byte(\"SignalExternalWorkflowExecutionFailed\"), nil\n\tcase 40:\n\t\treturn []byte(\"ExternalWorkflowExecutionSignaled\"), nil\n\tcase 41:\n\t\treturn []byte(\"UpsertWorkflowSearchAttributes\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of EventType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v EventType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionStarted\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionCompleted\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionFailed\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionTimedOut\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"DecisionTaskScheduled\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"DecisionTaskStarted\")\n\tcase 6:\n\t\tenc.AddString(\"name\", \"DecisionTaskCompleted\")\n\tcase 7:\n\t\tenc.AddString(\"name\", \"DecisionTaskTimedOut\")\n\tcase 8:\n\t\tenc.AddString(\"name\", \"DecisionTaskFailed\")\n\tcase 9:\n\t\tenc.AddString(\"name\", \"ActivityTaskScheduled\")\n\tcase 10:\n\t\tenc.AddString(\"name\", \"ActivityTaskStarted\")\n\tcase 11:\n\t\tenc.AddString(\"name\", \"ActivityTaskCompleted\")\n\tcase 12:\n\t\tenc.AddString(\"name\", \"ActivityTaskFailed\")\n\tcase 13:\n\t\tenc.AddString(\"name\", \"ActivityTaskTimedOut\")\n\tcase 14:\n\t\tenc.AddString(\"name\", \"ActivityTaskCancelRequested\")\n\tcase 15:\n\t\tenc.AddString(\"name\", \"RequestCancelActivityTaskFailed\")\n\tcase 16:\n\t\tenc.AddString(\"name\", \"ActivityTaskCanceled\")\n\tcase 17:\n\t\tenc.AddString(\"name\", \"TimerStarted\")\n\tcase 18:\n\t\tenc.AddString(\"name\", \"TimerFired\")\n\tcase 19:\n\t\tenc.AddString(\"name\", \"CancelTimerFailed\")\n\tcase 20:\n\t\tenc.AddString(\"name\", \"TimerCanceled\")\n\tcase 21:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionCancelRequested\")\n\tcase 22:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionCanceled\")\n\tcase 23:\n\t\tenc.AddString(\"name\", \"RequestCancelExternalWorkflowExecutionInitiated\")\n\tcase 24:\n\t\tenc.AddString(\"name\", \"RequestCancelExternalWorkflowExecutionFailed\")\n\tcase 25:\n\t\tenc.AddString(\"name\", \"ExternalWorkflowExecutionCancelRequested\")\n\tcase 26:\n\t\tenc.AddString(\"name\", \"MarkerRecorded\")\n\tcase 27:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionSignaled\")\n\tcase 28:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionTerminated\")\n\tcase 29:\n\t\tenc.AddString(\"name\", \"WorkflowExecutionContinuedAsNew\")\n\tcase 30:\n\t\tenc.AddString(\"name\", \"StartChildWorkflowExecutionInitiated\")\n\tcase 31:\n\t\tenc.AddString(\"name\", \"StartChildWorkflowExecutionFailed\")\n\tcase 32:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionStarted\")\n\tcase 33:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionCompleted\")\n\tcase 34:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionFailed\")\n\tcase 35:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionCanceled\")\n\tcase 36:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionTimedOut\")\n\tcase 37:\n\t\tenc.AddString(\"name\", \"ChildWorkflowExecutionTerminated\")\n\tcase 38:\n\t\tenc.AddString(\"name\", \"SignalExternalWorkflowExecutionInitiated\")\n\tcase 39:\n\t\tenc.AddString(\"name\", \"SignalExternalWorkflowExecutionFailed\")\n\tcase 40:\n\t\tenc.AddString(\"name\", \"ExternalWorkflowExecutionSignaled\")\n\tcase 41:\n\t\tenc.AddString(\"name\", \"UpsertWorkflowSearchAttributes\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v EventType) Ptr() *EventType {\n\treturn &v\n}\n\n// Encode encodes EventType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v EventType\n//\treturn v.Encode(sWriter)\nfunc (v EventType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates EventType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v EventType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes EventType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return EventType(0), err\n//\t}\n//\n//\tvar v EventType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return EventType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *EventType) FromWire(w wire.Value) error {\n\t*v = (EventType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded EventType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v EventType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return EventType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *EventType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (EventType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of EventType.\nfunc (v EventType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"WorkflowExecutionStarted\"\n\tcase 1:\n\t\treturn \"WorkflowExecutionCompleted\"\n\tcase 2:\n\t\treturn \"WorkflowExecutionFailed\"\n\tcase 3:\n\t\treturn \"WorkflowExecutionTimedOut\"\n\tcase 4:\n\t\treturn \"DecisionTaskScheduled\"\n\tcase 5:\n\t\treturn \"DecisionTaskStarted\"\n\tcase 6:\n\t\treturn \"DecisionTaskCompleted\"\n\tcase 7:\n\t\treturn \"DecisionTaskTimedOut\"\n\tcase 8:\n\t\treturn \"DecisionTaskFailed\"\n\tcase 9:\n\t\treturn \"ActivityTaskScheduled\"\n\tcase 10:\n\t\treturn \"ActivityTaskStarted\"\n\tcase 11:\n\t\treturn \"ActivityTaskCompleted\"\n\tcase 12:\n\t\treturn \"ActivityTaskFailed\"\n\tcase 13:\n\t\treturn \"ActivityTaskTimedOut\"\n\tcase 14:\n\t\treturn \"ActivityTaskCancelRequested\"\n\tcase 15:\n\t\treturn \"RequestCancelActivityTaskFailed\"\n\tcase 16:\n\t\treturn \"ActivityTaskCanceled\"\n\tcase 17:\n\t\treturn \"TimerStarted\"\n\tcase 18:\n\t\treturn \"TimerFired\"\n\tcase 19:\n\t\treturn \"CancelTimerFailed\"\n\tcase 20:\n\t\treturn \"TimerCanceled\"\n\tcase 21:\n\t\treturn \"WorkflowExecutionCancelRequested\"\n\tcase 22:\n\t\treturn \"WorkflowExecutionCanceled\"\n\tcase 23:\n\t\treturn \"RequestCancelExternalWorkflowExecutionInitiated\"\n\tcase 24:\n\t\treturn \"RequestCancelExternalWorkflowExecutionFailed\"\n\tcase 25:\n\t\treturn \"ExternalWorkflowExecutionCancelRequested\"\n\tcase 26:\n\t\treturn \"MarkerRecorded\"\n\tcase 27:\n\t\treturn \"WorkflowExecutionSignaled\"\n\tcase 28:\n\t\treturn \"WorkflowExecutionTerminated\"\n\tcase 29:\n\t\treturn \"WorkflowExecutionContinuedAsNew\"\n\tcase 30:\n\t\treturn \"StartChildWorkflowExecutionInitiated\"\n\tcase 31:\n\t\treturn \"StartChildWorkflowExecutionFailed\"\n\tcase 32:\n\t\treturn \"ChildWorkflowExecutionStarted\"\n\tcase 33:\n\t\treturn \"ChildWorkflowExecutionCompleted\"\n\tcase 34:\n\t\treturn \"ChildWorkflowExecutionFailed\"\n\tcase 35:\n\t\treturn \"ChildWorkflowExecutionCanceled\"\n\tcase 36:\n\t\treturn \"ChildWorkflowExecutionTimedOut\"\n\tcase 37:\n\t\treturn \"ChildWorkflowExecutionTerminated\"\n\tcase 38:\n\t\treturn \"SignalExternalWorkflowExecutionInitiated\"\n\tcase 39:\n\t\treturn \"SignalExternalWorkflowExecutionFailed\"\n\tcase 40:\n\t\treturn \"ExternalWorkflowExecutionSignaled\"\n\tcase 41:\n\t\treturn \"UpsertWorkflowSearchAttributes\"\n\t}\n\treturn fmt.Sprintf(\"EventType(%d)\", w)\n}\n\n// Equals returns true if this EventType value matches the provided\n// value.\nfunc (v EventType) Equals(rhs EventType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes EventType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v EventType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionStarted\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionCompleted\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionFailed\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionTimedOut\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"DecisionTaskScheduled\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"DecisionTaskStarted\\\"\"), nil\n\tcase 6:\n\t\treturn ([]byte)(\"\\\"DecisionTaskCompleted\\\"\"), nil\n\tcase 7:\n\t\treturn ([]byte)(\"\\\"DecisionTaskTimedOut\\\"\"), nil\n\tcase 8:\n\t\treturn ([]byte)(\"\\\"DecisionTaskFailed\\\"\"), nil\n\tcase 9:\n\t\treturn ([]byte)(\"\\\"ActivityTaskScheduled\\\"\"), nil\n\tcase 10:\n\t\treturn ([]byte)(\"\\\"ActivityTaskStarted\\\"\"), nil\n\tcase 11:\n\t\treturn ([]byte)(\"\\\"ActivityTaskCompleted\\\"\"), nil\n\tcase 12:\n\t\treturn ([]byte)(\"\\\"ActivityTaskFailed\\\"\"), nil\n\tcase 13:\n\t\treturn ([]byte)(\"\\\"ActivityTaskTimedOut\\\"\"), nil\n\tcase 14:\n\t\treturn ([]byte)(\"\\\"ActivityTaskCancelRequested\\\"\"), nil\n\tcase 15:\n\t\treturn ([]byte)(\"\\\"RequestCancelActivityTaskFailed\\\"\"), nil\n\tcase 16:\n\t\treturn ([]byte)(\"\\\"ActivityTaskCanceled\\\"\"), nil\n\tcase 17:\n\t\treturn ([]byte)(\"\\\"TimerStarted\\\"\"), nil\n\tcase 18:\n\t\treturn ([]byte)(\"\\\"TimerFired\\\"\"), nil\n\tcase 19:\n\t\treturn ([]byte)(\"\\\"CancelTimerFailed\\\"\"), nil\n\tcase 20:\n\t\treturn ([]byte)(\"\\\"TimerCanceled\\\"\"), nil\n\tcase 21:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionCancelRequested\\\"\"), nil\n\tcase 22:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionCanceled\\\"\"), nil\n\tcase 23:\n\t\treturn ([]byte)(\"\\\"RequestCancelExternalWorkflowExecutionInitiated\\\"\"), nil\n\tcase 24:\n\t\treturn ([]byte)(\"\\\"RequestCancelExternalWorkflowExecutionFailed\\\"\"), nil\n\tcase 25:\n\t\treturn ([]byte)(\"\\\"ExternalWorkflowExecutionCancelRequested\\\"\"), nil\n\tcase 26:\n\t\treturn ([]byte)(\"\\\"MarkerRecorded\\\"\"), nil\n\tcase 27:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionSignaled\\\"\"), nil\n\tcase 28:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionTerminated\\\"\"), nil\n\tcase 29:\n\t\treturn ([]byte)(\"\\\"WorkflowExecutionContinuedAsNew\\\"\"), nil\n\tcase 30:\n\t\treturn ([]byte)(\"\\\"StartChildWorkflowExecutionInitiated\\\"\"), nil\n\tcase 31:\n\t\treturn ([]byte)(\"\\\"StartChildWorkflowExecutionFailed\\\"\"), nil\n\tcase 32:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionStarted\\\"\"), nil\n\tcase 33:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionCompleted\\\"\"), nil\n\tcase 34:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionFailed\\\"\"), nil\n\tcase 35:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionCanceled\\\"\"), nil\n\tcase 36:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionTimedOut\\\"\"), nil\n\tcase 37:\n\t\treturn ([]byte)(\"\\\"ChildWorkflowExecutionTerminated\\\"\"), nil\n\tcase 38:\n\t\treturn ([]byte)(\"\\\"SignalExternalWorkflowExecutionInitiated\\\"\"), nil\n\tcase 39:\n\t\treturn ([]byte)(\"\\\"SignalExternalWorkflowExecutionFailed\\\"\"), nil\n\tcase 40:\n\t\treturn ([]byte)(\"\\\"ExternalWorkflowExecutionSignaled\\\"\"), nil\n\tcase 41:\n\t\treturn ([]byte)(\"\\\"UpsertWorkflowSearchAttributes\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode EventType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *EventType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"EventType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"EventType\")\n\t\t}\n\t\t*v = (EventType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"EventType\")\n\t}\n}\n\ntype ExternalWorkflowExecutionCancelRequestedEventAttributes struct {\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n}\n\n// ToWire translates a ExternalWorkflowExecutionCancelRequestedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ExternalWorkflowExecutionCancelRequestedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ExternalWorkflowExecutionCancelRequestedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ExternalWorkflowExecutionCancelRequestedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ExternalWorkflowExecutionCancelRequestedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ExternalWorkflowExecutionCancelRequestedEventAttributes struct could not be encoded.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ExternalWorkflowExecutionCancelRequestedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ExternalWorkflowExecutionCancelRequestedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ExternalWorkflowExecutionCancelRequestedEventAttributes\n// struct.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ExternalWorkflowExecutionCancelRequestedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ExternalWorkflowExecutionCancelRequestedEventAttributes match the\n// provided ExternalWorkflowExecutionCancelRequestedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) Equals(rhs *ExternalWorkflowExecutionCancelRequestedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ExternalWorkflowExecutionCancelRequestedEventAttributes.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\treturn err\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\ntype ExternalWorkflowExecutionSignaledEventAttributes struct {\n\tInitiatedEventId  *int64             `json:\"initiatedEventId,omitempty\"`\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n}\n\n// ToWire translates a ExternalWorkflowExecutionSignaledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ExternalWorkflowExecutionSignaledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ExternalWorkflowExecutionSignaledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ExternalWorkflowExecutionSignaledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ExternalWorkflowExecutionSignaledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ExternalWorkflowExecutionSignaledEventAttributes struct could not be encoded.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ExternalWorkflowExecutionSignaledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ExternalWorkflowExecutionSignaledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ExternalWorkflowExecutionSignaledEventAttributes\n// struct.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ExternalWorkflowExecutionSignaledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ExternalWorkflowExecutionSignaledEventAttributes match the\n// provided ExternalWorkflowExecutionSignaledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) Equals(rhs *ExternalWorkflowExecutionSignaledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ExternalWorkflowExecutionSignaledEventAttributes.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype FailWorkflowExecutionDecisionAttributes struct {\n\tReason  *string `json:\"reason,omitempty\"`\n\tDetails []byte  `json:\"details,omitempty\"`\n}\n\n// ToWire translates a FailWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FailWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FailWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *FailWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FailWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *FailWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *FailWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FailWorkflowExecutionDecisionAttributes match the\n// provided FailWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *FailWorkflowExecutionDecisionAttributes) Equals(rhs *FailWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailWorkflowExecutionDecisionAttributes.\nfunc (v *FailWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *FailWorkflowExecutionDecisionAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *FailWorkflowExecutionDecisionAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *FailWorkflowExecutionDecisionAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *FailWorkflowExecutionDecisionAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\ntype FailoverDomainRequest struct {\n\tDomainName              *string         `json:\"domainName,omitempty\"`\n\tDomainActiveClusterName *string         `json:\"domainActiveClusterName,omitempty\"`\n\tActiveClusters          *ActiveClusters `json:\"activeClusters,omitempty\"`\n\tReason                  *string         `json:\"reason,omitempty\"`\n}\n\n// ToWire translates a FailoverDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainName != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DomainActiveClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainActiveClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tw, err = v.ActiveClusters.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FailoverDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainActiveClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusters, err = _ActiveClusters_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FailoverDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverDomainRequest struct could not be encoded.\nfunc (v *FailoverDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainActiveClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainActiveClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusters.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FailoverDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *FailoverDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainActiveClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusters, err = _ActiveClusters_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverDomainRequest\n// struct.\nfunc (v *FailoverDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.DomainName != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainName: %v\", *(v.DomainName))\n\t\ti++\n\t}\n\tif v.DomainActiveClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainActiveClusterName: %v\", *(v.DomainActiveClusterName))\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusters: %v\", v.ActiveClusters)\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FailoverDomainRequest match the\n// provided FailoverDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverDomainRequest) Equals(rhs *FailoverDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainName, rhs.DomainName) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainActiveClusterName, rhs.DomainActiveClusterName) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusters == nil && rhs.ActiveClusters == nil) || (v.ActiveClusters != nil && rhs.ActiveClusters != nil && v.ActiveClusters.Equals(rhs.ActiveClusters))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverDomainRequest.\nfunc (v *FailoverDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainName != nil {\n\t\tenc.AddString(\"domainName\", *v.DomainName)\n\t}\n\tif v.DomainActiveClusterName != nil {\n\t\tenc.AddString(\"domainActiveClusterName\", *v.DomainActiveClusterName)\n\t}\n\tif v.ActiveClusters != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusters\", v.ActiveClusters))\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\treturn err\n}\n\n// GetDomainName returns the value of DomainName if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainRequest) GetDomainName() (o string) {\n\tif v != nil && v.DomainName != nil {\n\t\treturn *v.DomainName\n\t}\n\n\treturn\n}\n\n// IsSetDomainName returns true if DomainName is not nil.\nfunc (v *FailoverDomainRequest) IsSetDomainName() bool {\n\treturn v != nil && v.DomainName != nil\n}\n\n// GetDomainActiveClusterName returns the value of DomainActiveClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainRequest) GetDomainActiveClusterName() (o string) {\n\tif v != nil && v.DomainActiveClusterName != nil {\n\t\treturn *v.DomainActiveClusterName\n\t}\n\n\treturn\n}\n\n// IsSetDomainActiveClusterName returns true if DomainActiveClusterName is not nil.\nfunc (v *FailoverDomainRequest) IsSetDomainActiveClusterName() bool {\n\treturn v != nil && v.DomainActiveClusterName != nil\n}\n\n// GetActiveClusters returns the value of ActiveClusters if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainRequest) GetActiveClusters() (o *ActiveClusters) {\n\tif v != nil && v.ActiveClusters != nil {\n\t\treturn v.ActiveClusters\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusters returns true if ActiveClusters is not nil.\nfunc (v *FailoverDomainRequest) IsSetActiveClusters() bool {\n\treturn v != nil && v.ActiveClusters != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *FailoverDomainRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\ntype FailoverDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          *int64                          `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           *bool                           `json:\"isGlobalDomain,omitempty\"`\n}\n\n// ToWire translates a FailoverDomainResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverDomainResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainInfo != nil {\n\t\tw, err = v.DomainInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tw, err = v.ReplicationConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsGlobalDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FailoverDomainResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverDomainResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverDomainResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverDomainResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainInfo, err = _DomainInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _DomainConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsGlobalDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FailoverDomainResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverDomainResponse struct could not be encoded.\nfunc (v *FailoverDomainResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicationConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsGlobalDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsGlobalDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FailoverDomainResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverDomainResponse struct could not be generated from the wire\n// representation.\nfunc (v *FailoverDomainResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.DomainInfo, err = _DomainInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _DomainConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsGlobalDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverDomainResponse\n// struct.\nfunc (v *FailoverDomainResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainInfo: %v\", v.DomainInfo)\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationConfiguration: %v\", v.ReplicationConfiguration)\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsGlobalDomain: %v\", *(v.IsGlobalDomain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverDomainResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FailoverDomainResponse match the\n// provided FailoverDomainResponse.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverDomainResponse) Equals(rhs *FailoverDomainResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainInfo == nil && rhs.DomainInfo == nil) || (v.DomainInfo != nil && rhs.DomainInfo != nil && v.DomainInfo.Equals(rhs.DomainInfo))) {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationConfiguration == nil && rhs.ReplicationConfiguration == nil) || (v.ReplicationConfiguration != nil && rhs.ReplicationConfiguration != nil && v.ReplicationConfiguration.Equals(rhs.ReplicationConfiguration))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsGlobalDomain, rhs.IsGlobalDomain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverDomainResponse.\nfunc (v *FailoverDomainResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainInfo\", v.DomainInfo))\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationConfiguration\", v.ReplicationConfiguration))\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tenc.AddBool(\"isGlobalDomain\", *v.IsGlobalDomain)\n\t}\n\treturn err\n}\n\n// GetDomainInfo returns the value of DomainInfo if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\n\treturn\n}\n\n// IsSetDomainInfo returns true if DomainInfo is not nil.\nfunc (v *FailoverDomainResponse) IsSetDomainInfo() bool {\n\treturn v != nil && v.DomainInfo != nil\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainResponse) GetConfiguration() (o *DomainConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *FailoverDomainResponse) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\n// GetReplicationConfiguration returns the value of ReplicationConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainResponse) GetReplicationConfiguration() (o *DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfiguration != nil {\n\t\treturn v.ReplicationConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetReplicationConfiguration returns true if ReplicationConfiguration is not nil.\nfunc (v *FailoverDomainResponse) IsSetReplicationConfiguration() bool {\n\treturn v != nil && v.ReplicationConfiguration != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *FailoverDomainResponse) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetIsGlobalDomain returns the value of IsGlobalDomain if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil && v.IsGlobalDomain != nil {\n\t\treturn *v.IsGlobalDomain\n\t}\n\n\treturn\n}\n\n// IsSetIsGlobalDomain returns true if IsGlobalDomain is not nil.\nfunc (v *FailoverDomainResponse) IsSetIsGlobalDomain() bool {\n\treturn v != nil && v.IsGlobalDomain != nil\n}\n\ntype FailoverEvent struct {\n\tID               *string            `json:\"id,omitempty\"`\n\tCreatedTime      *int64             `json:\"createdTime,omitempty\"`\n\tFailoverType     *FailoverType      `json:\"failoverType,omitempty\"`\n\tClusterFailovers []*ClusterFailover `json:\"clusterFailovers,omitempty\"`\n}\n\ntype _List_ClusterFailover_ValueList []*ClusterFailover\n\nfunc (v _List_ClusterFailover_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ClusterFailover', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ClusterFailover_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ClusterFailover_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ClusterFailover_ValueList) Close() {}\n\n// ToWire translates a FailoverEvent struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverEvent) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ID != nil {\n\t\tw, err = wire.NewValueString(*(v.ID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CreatedTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreatedTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverType != nil {\n\t\tw, err = v.FailoverType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterFailovers != nil {\n\t\tw, err = wire.NewValueList(_List_ClusterFailover_ValueList(v.ClusterFailovers)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverType_Read(w wire.Value) (FailoverType, error) {\n\tvar v FailoverType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _ClusterFailover_Read(w wire.Value) (*ClusterFailover, error) {\n\tvar v ClusterFailover\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ClusterFailover_Read(l wire.ValueList) ([]*ClusterFailover, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ClusterFailover, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ClusterFailover_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a FailoverEvent struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverEvent struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverEvent\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverEvent) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreatedTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x FailoverType\n\t\t\t\tx, err = _FailoverType_Read(field.Value)\n\t\t\t\tv.FailoverType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ClusterFailovers, err = _List_ClusterFailover_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ClusterFailover_Encode(val []*ClusterFailover, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ClusterFailover', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a FailoverEvent struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverEvent struct could not be encoded.\nfunc (v *FailoverEvent) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreatedTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreatedTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.FailoverType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterFailovers != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ClusterFailover_Encode(v.ClusterFailovers, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverType_Decode(sr stream.Reader) (FailoverType, error) {\n\tvar v FailoverType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _ClusterFailover_Decode(sr stream.Reader) (*ClusterFailover, error) {\n\tvar v ClusterFailover\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ClusterFailover_Decode(sr stream.Reader) ([]*ClusterFailover, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ClusterFailover, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ClusterFailover_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a FailoverEvent struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverEvent struct could not be generated from the wire\n// representation.\nfunc (v *FailoverEvent) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreatedTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x FailoverType\n\t\t\tx, err = _FailoverType_Decode(sr)\n\t\t\tv.FailoverType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TList:\n\t\t\tv.ClusterFailovers, err = _List_ClusterFailover_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverEvent\n// struct.\nfunc (v *FailoverEvent) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.ID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ID: %v\", *(v.ID))\n\t\ti++\n\t}\n\tif v.CreatedTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreatedTime: %v\", *(v.CreatedTime))\n\t\ti++\n\t}\n\tif v.FailoverType != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverType: %v\", *(v.FailoverType))\n\t\ti++\n\t}\n\tif v.ClusterFailovers != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterFailovers: %v\", v.ClusterFailovers)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverEvent{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _FailoverType_EqualsPtr(lhs, rhs *FailoverType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_ClusterFailover_Equals(lhs, rhs []*ClusterFailover) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this FailoverEvent match the\n// provided FailoverEvent.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverEvent) Equals(rhs *FailoverEvent) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ID, rhs.ID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreatedTime, rhs.CreatedTime) {\n\t\treturn false\n\t}\n\tif !_FailoverType_EqualsPtr(v.FailoverType, rhs.FailoverType) {\n\t\treturn false\n\t}\n\tif !((v.ClusterFailovers == nil && rhs.ClusterFailovers == nil) || (v.ClusterFailovers != nil && rhs.ClusterFailovers != nil && _List_ClusterFailover_Equals(v.ClusterFailovers, rhs.ClusterFailovers))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ClusterFailover_Zapper []*ClusterFailover\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ClusterFailover_Zapper.\nfunc (l _List_ClusterFailover_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverEvent.\nfunc (v *FailoverEvent) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ID != nil {\n\t\tenc.AddString(\"id\", *v.ID)\n\t}\n\tif v.CreatedTime != nil {\n\t\tenc.AddInt64(\"createdTime\", *v.CreatedTime)\n\t}\n\tif v.FailoverType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"failoverType\", *v.FailoverType))\n\t}\n\tif v.ClusterFailovers != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"clusterFailovers\", (_List_ClusterFailover_Zapper)(v.ClusterFailovers)))\n\t}\n\treturn err\n}\n\n// GetID returns the value of ID if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverEvent) GetID() (o string) {\n\tif v != nil && v.ID != nil {\n\t\treturn *v.ID\n\t}\n\n\treturn\n}\n\n// IsSetID returns true if ID is not nil.\nfunc (v *FailoverEvent) IsSetID() bool {\n\treturn v != nil && v.ID != nil\n}\n\n// GetCreatedTime returns the value of CreatedTime if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverEvent) GetCreatedTime() (o int64) {\n\tif v != nil && v.CreatedTime != nil {\n\t\treturn *v.CreatedTime\n\t}\n\n\treturn\n}\n\n// IsSetCreatedTime returns true if CreatedTime is not nil.\nfunc (v *FailoverEvent) IsSetCreatedTime() bool {\n\treturn v != nil && v.CreatedTime != nil\n}\n\n// GetFailoverType returns the value of FailoverType if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverEvent) GetFailoverType() (o FailoverType) {\n\tif v != nil && v.FailoverType != nil {\n\t\treturn *v.FailoverType\n\t}\n\n\treturn\n}\n\n// IsSetFailoverType returns true if FailoverType is not nil.\nfunc (v *FailoverEvent) IsSetFailoverType() bool {\n\treturn v != nil && v.FailoverType != nil\n}\n\n// GetClusterFailovers returns the value of ClusterFailovers if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverEvent) GetClusterFailovers() (o []*ClusterFailover) {\n\tif v != nil && v.ClusterFailovers != nil {\n\t\treturn v.ClusterFailovers\n\t}\n\n\treturn\n}\n\n// IsSetClusterFailovers returns true if ClusterFailovers is not nil.\nfunc (v *FailoverEvent) IsSetClusterFailovers() bool {\n\treturn v != nil && v.ClusterFailovers != nil\n}\n\ntype FailoverInfo struct {\n\tFailoverVersion         *int64  `json:\"failoverVersion,omitempty\"`\n\tFailoverStartTimestamp  *int64  `json:\"failoverStartTimestamp,omitempty\"`\n\tFailoverExpireTimestamp *int64  `json:\"failoverExpireTimestamp,omitempty\"`\n\tCompletedShardCount     *int32  `json:\"completedShardCount,omitempty\"`\n\tPendingShards           []int32 `json:\"pendingShards,omitempty\"`\n}\n\n// ToWire translates a FailoverInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FailoverInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverStartTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverStartTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverExpireTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverExpireTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CompletedShardCount != nil {\n\t\tw, err = wire.NewValueI32(*(v.CompletedShardCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.PendingShards != nil {\n\t\tw, err = wire.NewValueList(_List_I32_ValueList(v.PendingShards)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FailoverInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FailoverInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FailoverInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FailoverInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverStartTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverExpireTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.CompletedShardCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.PendingShards, err = _List_I32_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FailoverInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FailoverInfo struct could not be encoded.\nfunc (v *FailoverInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverStartTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverStartTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverExpireTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverExpireTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletedShardCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.CompletedShardCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingShards != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I32_Encode(v.PendingShards, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FailoverInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FailoverInfo struct could not be generated from the wire\n// representation.\nfunc (v *FailoverInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverStartTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverExpireTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.CompletedShardCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TList:\n\t\t\tv.PendingShards, err = _List_I32_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FailoverInfo\n// struct.\nfunc (v *FailoverInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.FailoverStartTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverStartTimestamp: %v\", *(v.FailoverStartTimestamp))\n\t\ti++\n\t}\n\tif v.FailoverExpireTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverExpireTimestamp: %v\", *(v.FailoverExpireTimestamp))\n\t\ti++\n\t}\n\tif v.CompletedShardCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletedShardCount: %v\", *(v.CompletedShardCount))\n\t\ti++\n\t}\n\tif v.PendingShards != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingShards: %v\", v.PendingShards)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FailoverInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FailoverInfo match the\n// provided FailoverInfo.\n//\n// This function performs a deep comparison.\nfunc (v *FailoverInfo) Equals(rhs *FailoverInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverStartTimestamp, rhs.FailoverStartTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverExpireTimestamp, rhs.FailoverExpireTimestamp) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.CompletedShardCount, rhs.CompletedShardCount) {\n\t\treturn false\n\t}\n\tif !((v.PendingShards == nil && rhs.PendingShards == nil) || (v.PendingShards != nil && rhs.PendingShards != nil && _List_I32_Equals(v.PendingShards, rhs.PendingShards))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverInfo.\nfunc (v *FailoverInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.FailoverStartTimestamp != nil {\n\t\tenc.AddInt64(\"failoverStartTimestamp\", *v.FailoverStartTimestamp)\n\t}\n\tif v.FailoverExpireTimestamp != nil {\n\t\tenc.AddInt64(\"failoverExpireTimestamp\", *v.FailoverExpireTimestamp)\n\t}\n\tif v.CompletedShardCount != nil {\n\t\tenc.AddInt32(\"completedShardCount\", *v.CompletedShardCount)\n\t}\n\tif v.PendingShards != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"pendingShards\", (_List_I32_Zapper)(v.PendingShards)))\n\t}\n\treturn err\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverInfo) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *FailoverInfo) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetFailoverStartTimestamp returns the value of FailoverStartTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverInfo) GetFailoverStartTimestamp() (o int64) {\n\tif v != nil && v.FailoverStartTimestamp != nil {\n\t\treturn *v.FailoverStartTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetFailoverStartTimestamp returns true if FailoverStartTimestamp is not nil.\nfunc (v *FailoverInfo) IsSetFailoverStartTimestamp() bool {\n\treturn v != nil && v.FailoverStartTimestamp != nil\n}\n\n// GetFailoverExpireTimestamp returns the value of FailoverExpireTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverInfo) GetFailoverExpireTimestamp() (o int64) {\n\tif v != nil && v.FailoverExpireTimestamp != nil {\n\t\treturn *v.FailoverExpireTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetFailoverExpireTimestamp returns true if FailoverExpireTimestamp is not nil.\nfunc (v *FailoverInfo) IsSetFailoverExpireTimestamp() bool {\n\treturn v != nil && v.FailoverExpireTimestamp != nil\n}\n\n// GetCompletedShardCount returns the value of CompletedShardCount if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverInfo) GetCompletedShardCount() (o int32) {\n\tif v != nil && v.CompletedShardCount != nil {\n\t\treturn *v.CompletedShardCount\n\t}\n\n\treturn\n}\n\n// IsSetCompletedShardCount returns true if CompletedShardCount is not nil.\nfunc (v *FailoverInfo) IsSetCompletedShardCount() bool {\n\treturn v != nil && v.CompletedShardCount != nil\n}\n\n// GetPendingShards returns the value of PendingShards if it is set or its\n// zero value if it is unset.\nfunc (v *FailoverInfo) GetPendingShards() (o []int32) {\n\tif v != nil && v.PendingShards != nil {\n\t\treturn v.PendingShards\n\t}\n\n\treturn\n}\n\n// IsSetPendingShards returns true if PendingShards is not nil.\nfunc (v *FailoverInfo) IsSetPendingShards() bool {\n\treturn v != nil && v.PendingShards != nil\n}\n\ntype FailoverType int32\n\nconst (\n\tFailoverTypeInvalid  FailoverType = 0\n\tFailoverTypeForce    FailoverType = 1\n\tFailoverTypeGraceful FailoverType = 2\n)\n\n// FailoverType_Values returns all recognized values of FailoverType.\nfunc FailoverType_Values() []FailoverType {\n\treturn []FailoverType{\n\t\tFailoverTypeInvalid,\n\t\tFailoverTypeForce,\n\t\tFailoverTypeGraceful,\n\t}\n}\n\n// UnmarshalText tries to decode FailoverType from a byte slice\n// containing its name.\n//\n//\tvar v FailoverType\n//\terr := v.UnmarshalText([]byte(\"INVALID\"))\nfunc (v *FailoverType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"INVALID\":\n\t\t*v = FailoverTypeInvalid\n\t\treturn nil\n\tcase \"FORCE\":\n\t\t*v = FailoverTypeForce\n\t\treturn nil\n\tcase \"GRACEFUL\":\n\t\t*v = FailoverTypeGraceful\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"FailoverType\", err)\n\t\t}\n\t\t*v = FailoverType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes FailoverType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v FailoverType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"INVALID\"), nil\n\tcase 1:\n\t\treturn []byte(\"FORCE\"), nil\n\tcase 2:\n\t\treturn []byte(\"GRACEFUL\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FailoverType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v FailoverType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"INVALID\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"FORCE\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"GRACEFUL\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v FailoverType) Ptr() *FailoverType {\n\treturn &v\n}\n\n// Encode encodes FailoverType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v FailoverType\n//\treturn v.Encode(sWriter)\nfunc (v FailoverType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates FailoverType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v FailoverType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes FailoverType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return FailoverType(0), err\n//\t}\n//\n//\tvar v FailoverType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return FailoverType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *FailoverType) FromWire(w wire.Value) error {\n\t*v = (FailoverType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded FailoverType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v FailoverType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return FailoverType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *FailoverType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (FailoverType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of FailoverType.\nfunc (v FailoverType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"INVALID\"\n\tcase 1:\n\t\treturn \"FORCE\"\n\tcase 2:\n\t\treturn \"GRACEFUL\"\n\t}\n\treturn fmt.Sprintf(\"FailoverType(%d)\", w)\n}\n\n// Equals returns true if this FailoverType value matches the provided\n// value.\nfunc (v FailoverType) Equals(rhs FailoverType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes FailoverType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v FailoverType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"INVALID\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"FORCE\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"GRACEFUL\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode FailoverType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *FailoverType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"FailoverType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"FailoverType\")\n\t\t}\n\t\t*v = (FailoverType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"FailoverType\")\n\t}\n}\n\ntype FeatureFlags struct {\n\tWorkflowExecutionAlreadyCompletedErrorEnabled *bool `json:\"WorkflowExecutionAlreadyCompletedErrorEnabled,omitempty\"`\n\tAutoForwardingEnabled                         *bool `json:\"AutoForwardingEnabled,omitempty\"`\n}\n\n// ToWire translates a FeatureFlags struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FeatureFlags) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.WorkflowExecutionAlreadyCompletedErrorEnabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.AutoForwardingEnabled != nil {\n\t\tw, err = wire.NewValueBool(*(v.AutoForwardingEnabled)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FeatureFlags struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FeatureFlags struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FeatureFlags\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FeatureFlags) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.WorkflowExecutionAlreadyCompletedErrorEnabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.AutoForwardingEnabled = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FeatureFlags struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FeatureFlags struct could not be encoded.\nfunc (v *FeatureFlags) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.WorkflowExecutionAlreadyCompletedErrorEnabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoForwardingEnabled != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.AutoForwardingEnabled)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FeatureFlags struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FeatureFlags struct could not be generated from the wire\n// representation.\nfunc (v *FeatureFlags) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.WorkflowExecutionAlreadyCompletedErrorEnabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.AutoForwardingEnabled = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FeatureFlags\n// struct.\nfunc (v *FeatureFlags) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedErrorEnabled: %v\", *(v.WorkflowExecutionAlreadyCompletedErrorEnabled))\n\t\ti++\n\t}\n\tif v.AutoForwardingEnabled != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoForwardingEnabled: %v\", *(v.AutoForwardingEnabled))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"FeatureFlags{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this FeatureFlags match the\n// provided FeatureFlags.\n//\n// This function performs a deep comparison.\nfunc (v *FeatureFlags) Equals(rhs *FeatureFlags) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.WorkflowExecutionAlreadyCompletedErrorEnabled, rhs.WorkflowExecutionAlreadyCompletedErrorEnabled) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.AutoForwardingEnabled, rhs.AutoForwardingEnabled) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FeatureFlags.\nfunc (v *FeatureFlags) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil {\n\t\tenc.AddBool(\"WorkflowExecutionAlreadyCompletedErrorEnabled\", *v.WorkflowExecutionAlreadyCompletedErrorEnabled)\n\t}\n\tif v.AutoForwardingEnabled != nil {\n\t\tenc.AddBool(\"AutoForwardingEnabled\", *v.AutoForwardingEnabled)\n\t}\n\treturn err\n}\n\n// GetWorkflowExecutionAlreadyCompletedErrorEnabled returns the value of WorkflowExecutionAlreadyCompletedErrorEnabled if it is set or its\n// zero value if it is unset.\nfunc (v *FeatureFlags) GetWorkflowExecutionAlreadyCompletedErrorEnabled() (o bool) {\n\tif v != nil && v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil {\n\t\treturn *v.WorkflowExecutionAlreadyCompletedErrorEnabled\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionAlreadyCompletedErrorEnabled returns true if WorkflowExecutionAlreadyCompletedErrorEnabled is not nil.\nfunc (v *FeatureFlags) IsSetWorkflowExecutionAlreadyCompletedErrorEnabled() bool {\n\treturn v != nil && v.WorkflowExecutionAlreadyCompletedErrorEnabled != nil\n}\n\n// GetAutoForwardingEnabled returns the value of AutoForwardingEnabled if it is set or its\n// zero value if it is unset.\nfunc (v *FeatureFlags) GetAutoForwardingEnabled() (o bool) {\n\tif v != nil && v.AutoForwardingEnabled != nil {\n\t\treturn *v.AutoForwardingEnabled\n\t}\n\n\treturn\n}\n\n// IsSetAutoForwardingEnabled returns true if AutoForwardingEnabled is not nil.\nfunc (v *FeatureFlags) IsSetAutoForwardingEnabled() bool {\n\treturn v != nil && v.AutoForwardingEnabled != nil\n}\n\ntype FeatureNotEnabledError struct {\n\tFeatureFlag string `json:\"featureFlag,required\"`\n}\n\n// ToWire translates a FeatureNotEnabledError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *FeatureNotEnabledError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.FeatureFlag), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a FeatureNotEnabledError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a FeatureNotEnabledError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v FeatureNotEnabledError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *FeatureNotEnabledError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfeatureFlagIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.FeatureFlag, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tfeatureFlagIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !featureFlagIsSet {\n\t\treturn errors.New(\"field FeatureFlag of FeatureNotEnabledError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a FeatureNotEnabledError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a FeatureNotEnabledError struct could not be encoded.\nfunc (v *FeatureNotEnabledError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.FeatureFlag); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a FeatureNotEnabledError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a FeatureNotEnabledError struct could not be generated from the wire\n// representation.\nfunc (v *FeatureNotEnabledError) Decode(sr stream.Reader) error {\n\n\tfeatureFlagIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.FeatureFlag, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tfeatureFlagIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !featureFlagIsSet {\n\t\treturn errors.New(\"field FeatureFlag of FeatureNotEnabledError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a FeatureNotEnabledError\n// struct.\nfunc (v *FeatureNotEnabledError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"FeatureFlag: %v\", v.FeatureFlag)\n\ti++\n\n\treturn fmt.Sprintf(\"FeatureNotEnabledError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*FeatureNotEnabledError) ErrorName() string {\n\treturn \"FeatureNotEnabledError\"\n}\n\n// Equals returns true if all the fields of this FeatureNotEnabledError match the\n// provided FeatureNotEnabledError.\n//\n// This function performs a deep comparison.\nfunc (v *FeatureNotEnabledError) Equals(rhs *FeatureNotEnabledError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.FeatureFlag == rhs.FeatureFlag) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of FeatureNotEnabledError.\nfunc (v *FeatureNotEnabledError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"featureFlag\", v.FeatureFlag)\n\treturn err\n}\n\n// GetFeatureFlag returns the value of FeatureFlag if it is set or its\n// zero value if it is unset.\nfunc (v *FeatureNotEnabledError) GetFeatureFlag() (o string) {\n\tif v != nil {\n\t\to = v.FeatureFlag\n\t}\n\treturn\n}\n\nfunc (v *FeatureNotEnabledError) Error() string {\n\treturn v.String()\n}\n\ntype GetCrossClusterTasksRequest struct {\n\tShardIDs      []int32 `json:\"shardIDs,omitempty\"`\n\tTargetCluster *string `json:\"targetCluster,omitempty\"`\n}\n\n// ToWire translates a GetCrossClusterTasksRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetCrossClusterTasksRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardIDs != nil {\n\t\tw, err = wire.NewValueList(_List_I32_ValueList(v.ShardIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TargetCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetCrossClusterTasksRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetCrossClusterTasksRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetCrossClusterTasksRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetCrossClusterTasksRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ShardIDs, err = _List_I32_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetCrossClusterTasksRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetCrossClusterTasksRequest struct could not be encoded.\nfunc (v *GetCrossClusterTasksRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_I32_Encode(v.ShardIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetCrossClusterTasksRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetCrossClusterTasksRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetCrossClusterTasksRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ShardIDs, err = _List_I32_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetCrossClusterTasksRequest\n// struct.\nfunc (v *GetCrossClusterTasksRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ShardIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardIDs: %v\", v.ShardIDs)\n\t\ti++\n\t}\n\tif v.TargetCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetCluster: %v\", *(v.TargetCluster))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetCrossClusterTasksRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetCrossClusterTasksRequest match the\n// provided GetCrossClusterTasksRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetCrossClusterTasksRequest) Equals(rhs *GetCrossClusterTasksRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ShardIDs == nil && rhs.ShardIDs == nil) || (v.ShardIDs != nil && rhs.ShardIDs != nil && _List_I32_Equals(v.ShardIDs, rhs.ShardIDs))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetCluster, rhs.TargetCluster) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetCrossClusterTasksRequest.\nfunc (v *GetCrossClusterTasksRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"shardIDs\", (_List_I32_Zapper)(v.ShardIDs)))\n\t}\n\tif v.TargetCluster != nil {\n\t\tenc.AddString(\"targetCluster\", *v.TargetCluster)\n\t}\n\treturn err\n}\n\n// GetShardIDs returns the value of ShardIDs if it is set or its\n// zero value if it is unset.\nfunc (v *GetCrossClusterTasksRequest) GetShardIDs() (o []int32) {\n\tif v != nil && v.ShardIDs != nil {\n\t\treturn v.ShardIDs\n\t}\n\n\treturn\n}\n\n// IsSetShardIDs returns true if ShardIDs is not nil.\nfunc (v *GetCrossClusterTasksRequest) IsSetShardIDs() bool {\n\treturn v != nil && v.ShardIDs != nil\n}\n\n// GetTargetCluster returns the value of TargetCluster if it is set or its\n// zero value if it is unset.\nfunc (v *GetCrossClusterTasksRequest) GetTargetCluster() (o string) {\n\tif v != nil && v.TargetCluster != nil {\n\t\treturn *v.TargetCluster\n\t}\n\n\treturn\n}\n\n// IsSetTargetCluster returns true if TargetCluster is not nil.\nfunc (v *GetCrossClusterTasksRequest) IsSetTargetCluster() bool {\n\treturn v != nil && v.TargetCluster != nil\n}\n\ntype GetCrossClusterTasksResponse struct {\n\tTasksByShard       map[int32][]*CrossClusterTaskRequest `json:\"tasksByShard,omitempty\"`\n\tFailedCauseByShard map[int32]GetTaskFailedCause         `json:\"failedCauseByShard,omitempty\"`\n}\n\ntype _List_CrossClusterTaskRequest_ValueList []*CrossClusterTaskRequest\n\nfunc (v _List_CrossClusterTaskRequest_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*CrossClusterTaskRequest', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_CrossClusterTaskRequest_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_CrossClusterTaskRequest_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_CrossClusterTaskRequest_ValueList) Close() {}\n\ntype _Map_I32_List_CrossClusterTaskRequest_MapItemList map[int32][]*CrossClusterTaskRequest\n\nfunc (m _Map_I32_List_CrossClusterTaskRequest_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32][]*CrossClusterTaskRequest', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueList(_List_CrossClusterTaskRequest_ValueList(v)), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_List_CrossClusterTaskRequest_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_List_CrossClusterTaskRequest_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_List_CrossClusterTaskRequest_MapItemList) ValueType() wire.Type {\n\treturn wire.TList\n}\n\nfunc (_Map_I32_List_CrossClusterTaskRequest_MapItemList) Close() {}\n\ntype _Map_I32_GetTaskFailedCause_MapItemList map[int32]GetTaskFailedCause\n\nfunc (m _Map_I32_GetTaskFailedCause_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_GetTaskFailedCause_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_GetTaskFailedCause_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_GetTaskFailedCause_MapItemList) ValueType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_GetTaskFailedCause_MapItemList) Close() {}\n\n// ToWire translates a GetCrossClusterTasksResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetCrossClusterTasksResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TasksByShard != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_List_CrossClusterTaskRequest_MapItemList(v.TasksByShard)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FailedCauseByShard != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_GetTaskFailedCause_MapItemList(v.FailedCauseByShard)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterTaskRequest_Read(w wire.Value) (*CrossClusterTaskRequest, error) {\n\tvar v CrossClusterTaskRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_CrossClusterTaskRequest_Read(l wire.ValueList) ([]*CrossClusterTaskRequest, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*CrossClusterTaskRequest, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _CrossClusterTaskRequest_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _Map_I32_List_CrossClusterTaskRequest_Read(m wire.MapItemList) (map[int32][]*CrossClusterTaskRequest, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TList {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32][]*CrossClusterTaskRequest, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _List_CrossClusterTaskRequest_Read(x.Value.GetList())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _GetTaskFailedCause_Read(w wire.Value) (GetTaskFailedCause, error) {\n\tvar v GetTaskFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Map_I32_GetTaskFailedCause_Read(m wire.MapItemList) (map[int32]GetTaskFailedCause, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32]GetTaskFailedCause, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _GetTaskFailedCause_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetCrossClusterTasksResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetCrossClusterTasksResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetCrossClusterTasksResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetCrossClusterTasksResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.TasksByShard, err = _Map_I32_List_CrossClusterTaskRequest_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.FailedCauseByShard, err = _Map_I32_GetTaskFailedCause_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_CrossClusterTaskRequest_Encode(val []*CrossClusterTaskRequest, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*CrossClusterTaskRequest', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _Map_I32_List_CrossClusterTaskRequest_Encode(val map[int32][]*CrossClusterTaskRequest, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TList,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32][]*CrossClusterTaskRequest', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_CrossClusterTaskRequest_Encode(v, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\nfunc _Map_I32_GetTaskFailedCause_Encode(val map[int32]GetTaskFailedCause, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TI32,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a GetCrossClusterTasksResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetCrossClusterTasksResponse struct could not be encoded.\nfunc (v *GetCrossClusterTasksResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TasksByShard != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_List_CrossClusterTaskRequest_Encode(v.TasksByShard, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailedCauseByShard != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_GetTaskFailedCause_Encode(v.FailedCauseByShard, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterTaskRequest_Decode(sr stream.Reader) (*CrossClusterTaskRequest, error) {\n\tvar v CrossClusterTaskRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_CrossClusterTaskRequest_Decode(sr stream.Reader) ([]*CrossClusterTaskRequest, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*CrossClusterTaskRequest, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _CrossClusterTaskRequest_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _Map_I32_List_CrossClusterTaskRequest_Decode(sr stream.Reader) (map[int32][]*CrossClusterTaskRequest, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TList {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32][]*CrossClusterTaskRequest, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _List_CrossClusterTaskRequest_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _GetTaskFailedCause_Decode(sr stream.Reader) (GetTaskFailedCause, error) {\n\tvar v GetTaskFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Map_I32_GetTaskFailedCause_Decode(sr stream.Reader) (map[int32]GetTaskFailedCause, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TI32 {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32]GetTaskFailedCause, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _GetTaskFailedCause_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetCrossClusterTasksResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetCrossClusterTasksResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetCrossClusterTasksResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.TasksByShard, err = _Map_I32_List_CrossClusterTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TMap:\n\t\t\tv.FailedCauseByShard, err = _Map_I32_GetTaskFailedCause_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetCrossClusterTasksResponse\n// struct.\nfunc (v *GetCrossClusterTasksResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.TasksByShard != nil {\n\t\tfields[i] = fmt.Sprintf(\"TasksByShard: %v\", v.TasksByShard)\n\t\ti++\n\t}\n\tif v.FailedCauseByShard != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailedCauseByShard: %v\", v.FailedCauseByShard)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetCrossClusterTasksResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_CrossClusterTaskRequest_Equals(lhs, rhs []*CrossClusterTaskRequest) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _Map_I32_List_CrossClusterTaskRequest_Equals(lhs, rhs map[int32][]*CrossClusterTaskRequest) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !_List_CrossClusterTaskRequest_Equals(lv, rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _Map_I32_GetTaskFailedCause_Equals(lhs, rhs map[int32]GetTaskFailedCause) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetCrossClusterTasksResponse match the\n// provided GetCrossClusterTasksResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetCrossClusterTasksResponse) Equals(rhs *GetCrossClusterTasksResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TasksByShard == nil && rhs.TasksByShard == nil) || (v.TasksByShard != nil && rhs.TasksByShard != nil && _Map_I32_List_CrossClusterTaskRequest_Equals(v.TasksByShard, rhs.TasksByShard))) {\n\t\treturn false\n\t}\n\tif !((v.FailedCauseByShard == nil && rhs.FailedCauseByShard == nil) || (v.FailedCauseByShard != nil && rhs.FailedCauseByShard != nil && _Map_I32_GetTaskFailedCause_Equals(v.FailedCauseByShard, rhs.FailedCauseByShard))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_CrossClusterTaskRequest_Zapper []*CrossClusterTaskRequest\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_CrossClusterTaskRequest_Zapper.\nfunc (l _List_CrossClusterTaskRequest_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\ntype _Map_I32_List_CrossClusterTaskRequest_Item_Zapper struct {\n\tKey   int32\n\tValue []*CrossClusterTaskRequest\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_List_CrossClusterTaskRequest_Item_Zapper.\nfunc (v _Map_I32_List_CrossClusterTaskRequest_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddArray(\"value\", (_List_CrossClusterTaskRequest_Zapper)(v.Value)))\n\treturn err\n}\n\ntype _Map_I32_List_CrossClusterTaskRequest_Zapper map[int32][]*CrossClusterTaskRequest\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_List_CrossClusterTaskRequest_Zapper.\nfunc (m _Map_I32_List_CrossClusterTaskRequest_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_List_CrossClusterTaskRequest_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\ntype _Map_I32_GetTaskFailedCause_Item_Zapper struct {\n\tKey   int32\n\tValue GetTaskFailedCause\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_GetTaskFailedCause_Item_Zapper.\nfunc (v _Map_I32_GetTaskFailedCause_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\treturn err\n}\n\ntype _Map_I32_GetTaskFailedCause_Zapper map[int32]GetTaskFailedCause\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_GetTaskFailedCause_Zapper.\nfunc (m _Map_I32_GetTaskFailedCause_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_GetTaskFailedCause_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetCrossClusterTasksResponse.\nfunc (v *GetCrossClusterTasksResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TasksByShard != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"tasksByShard\", (_Map_I32_List_CrossClusterTaskRequest_Zapper)(v.TasksByShard)))\n\t}\n\tif v.FailedCauseByShard != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"failedCauseByShard\", (_Map_I32_GetTaskFailedCause_Zapper)(v.FailedCauseByShard)))\n\t}\n\treturn err\n}\n\n// GetTasksByShard returns the value of TasksByShard if it is set or its\n// zero value if it is unset.\nfunc (v *GetCrossClusterTasksResponse) GetTasksByShard() (o map[int32][]*CrossClusterTaskRequest) {\n\tif v != nil && v.TasksByShard != nil {\n\t\treturn v.TasksByShard\n\t}\n\n\treturn\n}\n\n// IsSetTasksByShard returns true if TasksByShard is not nil.\nfunc (v *GetCrossClusterTasksResponse) IsSetTasksByShard() bool {\n\treturn v != nil && v.TasksByShard != nil\n}\n\n// GetFailedCauseByShard returns the value of FailedCauseByShard if it is set or its\n// zero value if it is unset.\nfunc (v *GetCrossClusterTasksResponse) GetFailedCauseByShard() (o map[int32]GetTaskFailedCause) {\n\tif v != nil && v.FailedCauseByShard != nil {\n\t\treturn v.FailedCauseByShard\n\t}\n\n\treturn\n}\n\n// IsSetFailedCauseByShard returns true if FailedCauseByShard is not nil.\nfunc (v *GetCrossClusterTasksResponse) IsSetFailedCauseByShard() bool {\n\treturn v != nil && v.FailedCauseByShard != nil\n}\n\ntype GetSearchAttributesResponse struct {\n\tKeys map[string]IndexedValueType `json:\"keys,omitempty\"`\n}\n\ntype _Map_String_IndexedValueType_MapItemList map[string]IndexedValueType\n\nfunc (m _Map_String_IndexedValueType_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_IndexedValueType_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) ValueType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_String_IndexedValueType_MapItemList) Close() {}\n\n// ToWire translates a GetSearchAttributesResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetSearchAttributesResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Keys != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_IndexedValueType_MapItemList(v.Keys)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IndexedValueType_Read(w wire.Value) (IndexedValueType, error) {\n\tvar v IndexedValueType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Map_String_IndexedValueType_Read(m wire.MapItemList) (map[string]IndexedValueType, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]IndexedValueType, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _IndexedValueType_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetSearchAttributesResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetSearchAttributesResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetSearchAttributesResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetSearchAttributesResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Keys, err = _Map_String_IndexedValueType_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_IndexedValueType_Encode(val map[string]IndexedValueType, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TI32,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a GetSearchAttributesResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetSearchAttributesResponse struct could not be encoded.\nfunc (v *GetSearchAttributesResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Keys != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_IndexedValueType_Encode(v.Keys, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IndexedValueType_Decode(sr stream.Reader) (IndexedValueType, error) {\n\tvar v IndexedValueType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Map_String_IndexedValueType_Decode(sr stream.Reader) (map[string]IndexedValueType, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TI32 {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]IndexedValueType, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _IndexedValueType_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetSearchAttributesResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetSearchAttributesResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetSearchAttributesResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Keys, err = _Map_String_IndexedValueType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetSearchAttributesResponse\n// struct.\nfunc (v *GetSearchAttributesResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Keys != nil {\n\t\tfields[i] = fmt.Sprintf(\"Keys: %v\", v.Keys)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetSearchAttributesResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_IndexedValueType_Equals(lhs, rhs map[string]IndexedValueType) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetSearchAttributesResponse match the\n// provided GetSearchAttributesResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetSearchAttributesResponse) Equals(rhs *GetSearchAttributesResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Keys == nil && rhs.Keys == nil) || (v.Keys != nil && rhs.Keys != nil && _Map_String_IndexedValueType_Equals(v.Keys, rhs.Keys))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_IndexedValueType_Zapper map[string]IndexedValueType\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_IndexedValueType_Zapper.\nfunc (m _Map_String_IndexedValueType_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetSearchAttributesResponse.\nfunc (v *GetSearchAttributesResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Keys != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"keys\", (_Map_String_IndexedValueType_Zapper)(v.Keys)))\n\t}\n\treturn err\n}\n\n// GetKeys returns the value of Keys if it is set or its\n// zero value if it is unset.\nfunc (v *GetSearchAttributesResponse) GetKeys() (o map[string]IndexedValueType) {\n\tif v != nil && v.Keys != nil {\n\t\treturn v.Keys\n\t}\n\n\treturn\n}\n\n// IsSetKeys returns true if Keys is not nil.\nfunc (v *GetSearchAttributesResponse) IsSetKeys() bool {\n\treturn v != nil && v.Keys != nil\n}\n\ntype GetTaskFailedCause int32\n\nconst (\n\tGetTaskFailedCauseServiceBusy        GetTaskFailedCause = 0\n\tGetTaskFailedCauseTimeout            GetTaskFailedCause = 1\n\tGetTaskFailedCauseShardOwnershipLost GetTaskFailedCause = 2\n\tGetTaskFailedCauseUncategorized      GetTaskFailedCause = 3\n)\n\n// GetTaskFailedCause_Values returns all recognized values of GetTaskFailedCause.\nfunc GetTaskFailedCause_Values() []GetTaskFailedCause {\n\treturn []GetTaskFailedCause{\n\t\tGetTaskFailedCauseServiceBusy,\n\t\tGetTaskFailedCauseTimeout,\n\t\tGetTaskFailedCauseShardOwnershipLost,\n\t\tGetTaskFailedCauseUncategorized,\n\t}\n}\n\n// UnmarshalText tries to decode GetTaskFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v GetTaskFailedCause\n//\terr := v.UnmarshalText([]byte(\"SERVICE_BUSY\"))\nfunc (v *GetTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"SERVICE_BUSY\":\n\t\t*v = GetTaskFailedCauseServiceBusy\n\t\treturn nil\n\tcase \"TIMEOUT\":\n\t\t*v = GetTaskFailedCauseTimeout\n\t\treturn nil\n\tcase \"SHARD_OWNERSHIP_LOST\":\n\t\t*v = GetTaskFailedCauseShardOwnershipLost\n\t\treturn nil\n\tcase \"UNCATEGORIZED\":\n\t\t*v = GetTaskFailedCauseUncategorized\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"GetTaskFailedCause\", err)\n\t\t}\n\t\t*v = GetTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes GetTaskFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v GetTaskFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"SERVICE_BUSY\"), nil\n\tcase 1:\n\t\treturn []byte(\"TIMEOUT\"), nil\n\tcase 2:\n\t\treturn []byte(\"SHARD_OWNERSHIP_LOST\"), nil\n\tcase 3:\n\t\treturn []byte(\"UNCATEGORIZED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetTaskFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v GetTaskFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"SERVICE_BUSY\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"TIMEOUT\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"SHARD_OWNERSHIP_LOST\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"UNCATEGORIZED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v GetTaskFailedCause) Ptr() *GetTaskFailedCause {\n\treturn &v\n}\n\n// Encode encodes GetTaskFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v GetTaskFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v GetTaskFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates GetTaskFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v GetTaskFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes GetTaskFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return GetTaskFailedCause(0), err\n//\t}\n//\n//\tvar v GetTaskFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return GetTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *GetTaskFailedCause) FromWire(w wire.Value) error {\n\t*v = (GetTaskFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded GetTaskFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v GetTaskFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return GetTaskFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *GetTaskFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (GetTaskFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of GetTaskFailedCause.\nfunc (v GetTaskFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SERVICE_BUSY\"\n\tcase 1:\n\t\treturn \"TIMEOUT\"\n\tcase 2:\n\t\treturn \"SHARD_OWNERSHIP_LOST\"\n\tcase 3:\n\t\treturn \"UNCATEGORIZED\"\n\t}\n\treturn fmt.Sprintf(\"GetTaskFailedCause(%d)\", w)\n}\n\n// Equals returns true if this GetTaskFailedCause value matches the provided\n// value.\nfunc (v GetTaskFailedCause) Equals(rhs GetTaskFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes GetTaskFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v GetTaskFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"SERVICE_BUSY\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"TIMEOUT\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"SHARD_OWNERSHIP_LOST\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"UNCATEGORIZED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode GetTaskFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *GetTaskFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"GetTaskFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"GetTaskFailedCause\")\n\t\t}\n\t\t*v = (GetTaskFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"GetTaskFailedCause\")\n\t}\n}\n\ntype GetTaskListsByDomainRequest struct {\n\tDomainName *string `json:\"domainName,omitempty\"`\n}\n\n// ToWire translates a GetTaskListsByDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetTaskListsByDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainName != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a GetTaskListsByDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetTaskListsByDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetTaskListsByDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetTaskListsByDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetTaskListsByDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetTaskListsByDomainRequest struct could not be encoded.\nfunc (v *GetTaskListsByDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a GetTaskListsByDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetTaskListsByDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetTaskListsByDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetTaskListsByDomainRequest\n// struct.\nfunc (v *GetTaskListsByDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DomainName != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainName: %v\", *(v.DomainName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetTaskListsByDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this GetTaskListsByDomainRequest match the\n// provided GetTaskListsByDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetTaskListsByDomainRequest) Equals(rhs *GetTaskListsByDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainName, rhs.DomainName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetTaskListsByDomainRequest.\nfunc (v *GetTaskListsByDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainName != nil {\n\t\tenc.AddString(\"domainName\", *v.DomainName)\n\t}\n\treturn err\n}\n\n// GetDomainName returns the value of DomainName if it is set or its\n// zero value if it is unset.\nfunc (v *GetTaskListsByDomainRequest) GetDomainName() (o string) {\n\tif v != nil && v.DomainName != nil {\n\t\treturn *v.DomainName\n\t}\n\n\treturn\n}\n\n// IsSetDomainName returns true if DomainName is not nil.\nfunc (v *GetTaskListsByDomainRequest) IsSetDomainName() bool {\n\treturn v != nil && v.DomainName != nil\n}\n\ntype GetTaskListsByDomainResponse struct {\n\tDecisionTaskListMap map[string]*DescribeTaskListResponse `json:\"decisionTaskListMap,omitempty\"`\n\tActivityTaskListMap map[string]*DescribeTaskListResponse `json:\"activityTaskListMap,omitempty\"`\n}\n\ntype _Map_String_DescribeTaskListResponse_MapItemList map[string]*DescribeTaskListResponse\n\nfunc (m _Map_String_DescribeTaskListResponse_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*DescribeTaskListResponse', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_DescribeTaskListResponse_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_DescribeTaskListResponse_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_DescribeTaskListResponse_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_DescribeTaskListResponse_MapItemList) Close() {}\n\n// ToWire translates a GetTaskListsByDomainResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetTaskListsByDomainResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTaskListMap != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_DescribeTaskListResponse_MapItemList(v.DecisionTaskListMap)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskListMap != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_DescribeTaskListResponse_MapItemList(v.ActivityTaskListMap)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeTaskListResponse_Read(w wire.Value) (*DescribeTaskListResponse, error) {\n\tvar v DescribeTaskListResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_DescribeTaskListResponse_Read(m wire.MapItemList) (map[string]*DescribeTaskListResponse, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*DescribeTaskListResponse, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _DescribeTaskListResponse_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetTaskListsByDomainResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetTaskListsByDomainResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetTaskListsByDomainResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetTaskListsByDomainResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.DecisionTaskListMap, err = _Map_String_DescribeTaskListResponse_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActivityTaskListMap, err = _Map_String_DescribeTaskListResponse_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_DescribeTaskListResponse_Encode(val map[string]*DescribeTaskListResponse, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*DescribeTaskListResponse', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a GetTaskListsByDomainResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetTaskListsByDomainResponse struct could not be encoded.\nfunc (v *GetTaskListsByDomainResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTaskListMap != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_DescribeTaskListResponse_Encode(v.DecisionTaskListMap, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskListMap != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_DescribeTaskListResponse_Encode(v.ActivityTaskListMap, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeTaskListResponse_Decode(sr stream.Reader) (*DescribeTaskListResponse, error) {\n\tvar v DescribeTaskListResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_DescribeTaskListResponse_Decode(sr stream.Reader) (map[string]*DescribeTaskListResponse, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*DescribeTaskListResponse, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _DescribeTaskListResponse_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetTaskListsByDomainResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetTaskListsByDomainResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetTaskListsByDomainResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.DecisionTaskListMap, err = _Map_String_DescribeTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TMap:\n\t\t\tv.ActivityTaskListMap, err = _Map_String_DescribeTaskListResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetTaskListsByDomainResponse\n// struct.\nfunc (v *GetTaskListsByDomainResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DecisionTaskListMap != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskListMap: %v\", v.DecisionTaskListMap)\n\t\ti++\n\t}\n\tif v.ActivityTaskListMap != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskListMap: %v\", v.ActivityTaskListMap)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetTaskListsByDomainResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_DescribeTaskListResponse_Equals(lhs, rhs map[string]*DescribeTaskListResponse) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetTaskListsByDomainResponse match the\n// provided GetTaskListsByDomainResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetTaskListsByDomainResponse) Equals(rhs *GetTaskListsByDomainResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskListMap == nil && rhs.DecisionTaskListMap == nil) || (v.DecisionTaskListMap != nil && rhs.DecisionTaskListMap != nil && _Map_String_DescribeTaskListResponse_Equals(v.DecisionTaskListMap, rhs.DecisionTaskListMap))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskListMap == nil && rhs.ActivityTaskListMap == nil) || (v.ActivityTaskListMap != nil && rhs.ActivityTaskListMap != nil && _Map_String_DescribeTaskListResponse_Equals(v.ActivityTaskListMap, rhs.ActivityTaskListMap))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_DescribeTaskListResponse_Zapper map[string]*DescribeTaskListResponse\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_DescribeTaskListResponse_Zapper.\nfunc (m _Map_String_DescribeTaskListResponse_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetTaskListsByDomainResponse.\nfunc (v *GetTaskListsByDomainResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTaskListMap != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskListMap\", (_Map_String_DescribeTaskListResponse_Zapper)(v.DecisionTaskListMap)))\n\t}\n\tif v.ActivityTaskListMap != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskListMap\", (_Map_String_DescribeTaskListResponse_Zapper)(v.ActivityTaskListMap)))\n\t}\n\treturn err\n}\n\n// GetDecisionTaskListMap returns the value of DecisionTaskListMap if it is set or its\n// zero value if it is unset.\nfunc (v *GetTaskListsByDomainResponse) GetDecisionTaskListMap() (o map[string]*DescribeTaskListResponse) {\n\tif v != nil && v.DecisionTaskListMap != nil {\n\t\treturn v.DecisionTaskListMap\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskListMap returns true if DecisionTaskListMap is not nil.\nfunc (v *GetTaskListsByDomainResponse) IsSetDecisionTaskListMap() bool {\n\treturn v != nil && v.DecisionTaskListMap != nil\n}\n\n// GetActivityTaskListMap returns the value of ActivityTaskListMap if it is set or its\n// zero value if it is unset.\nfunc (v *GetTaskListsByDomainResponse) GetActivityTaskListMap() (o map[string]*DescribeTaskListResponse) {\n\tif v != nil && v.ActivityTaskListMap != nil {\n\t\treturn v.ActivityTaskListMap\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskListMap returns true if ActivityTaskListMap is not nil.\nfunc (v *GetTaskListsByDomainResponse) IsSetActivityTaskListMap() bool {\n\treturn v != nil && v.ActivityTaskListMap != nil\n}\n\ntype GetWorkflowExecutionHistoryRequest struct {\n\tDomain                 *string                 `json:\"domain,omitempty\"`\n\tExecution              *WorkflowExecution      `json:\"execution,omitempty\"`\n\tMaximumPageSize        *int32                  `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken          []byte                  `json:\"nextPageToken,omitempty\"`\n\tWaitForNewEvent        *bool                   `json:\"waitForNewEvent,omitempty\"`\n\tHistoryEventFilterType *HistoryEventFilterType `json:\"HistoryEventFilterType,omitempty\"`\n\tSkipArchival           *bool                   `json:\"skipArchival,omitempty\"`\n\tQueryConsistencyLevel  *QueryConsistencyLevel  `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// ToWire translates a GetWorkflowExecutionHistoryRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetWorkflowExecutionHistoryRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.WaitForNewEvent != nil {\n\t\tw, err = wire.NewValueBool(*(v.WaitForNewEvent)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryEventFilterType != nil {\n\t\tw, err = v.HistoryEventFilterType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.SkipArchival != nil {\n\t\tw, err = wire.NewValueBool(*(v.SkipArchival)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tw, err = v.QueryConsistencyLevel.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryEventFilterType_Read(w wire.Value) (HistoryEventFilterType, error) {\n\tvar v HistoryEventFilterType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a GetWorkflowExecutionHistoryRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetWorkflowExecutionHistoryRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetWorkflowExecutionHistoryRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetWorkflowExecutionHistoryRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.WaitForNewEvent = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x HistoryEventFilterType\n\t\t\t\tx, err = _HistoryEventFilterType_Read(field.Value)\n\t\t\t\tv.HistoryEventFilterType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.SkipArchival = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryConsistencyLevel\n\t\t\t\tx, err = _QueryConsistencyLevel_Read(field.Value)\n\t\t\t\tv.QueryConsistencyLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a GetWorkflowExecutionHistoryRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetWorkflowExecutionHistoryRequest struct could not be encoded.\nfunc (v *GetWorkflowExecutionHistoryRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WaitForNewEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.WaitForNewEvent)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryEventFilterType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HistoryEventFilterType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SkipArchival != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.SkipArchival)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryConsistencyLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryConsistencyLevel.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryEventFilterType_Decode(sr stream.Reader) (HistoryEventFilterType, error) {\n\tvar v HistoryEventFilterType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a GetWorkflowExecutionHistoryRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetWorkflowExecutionHistoryRequest struct could not be generated from the wire\n// representation.\nfunc (v *GetWorkflowExecutionHistoryRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.WaitForNewEvent = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x HistoryEventFilterType\n\t\t\tx, err = _HistoryEventFilterType_Decode(sr)\n\t\t\tv.HistoryEventFilterType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.SkipArchival = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x QueryConsistencyLevel\n\t\t\tx, err = _QueryConsistencyLevel_Decode(sr)\n\t\t\tv.QueryConsistencyLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetWorkflowExecutionHistoryRequest\n// struct.\nfunc (v *GetWorkflowExecutionHistoryRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.WaitForNewEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"WaitForNewEvent: %v\", *(v.WaitForNewEvent))\n\t\ti++\n\t}\n\tif v.HistoryEventFilterType != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryEventFilterType: %v\", *(v.HistoryEventFilterType))\n\t\ti++\n\t}\n\tif v.SkipArchival != nil {\n\t\tfields[i] = fmt.Sprintf(\"SkipArchival: %v\", *(v.SkipArchival))\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryConsistencyLevel: %v\", *(v.QueryConsistencyLevel))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetWorkflowExecutionHistoryRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _HistoryEventFilterType_EqualsPtr(lhs, rhs *HistoryEventFilterType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this GetWorkflowExecutionHistoryRequest match the\n// provided GetWorkflowExecutionHistoryRequest.\n//\n// This function performs a deep comparison.\nfunc (v *GetWorkflowExecutionHistoryRequest) Equals(rhs *GetWorkflowExecutionHistoryRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.WaitForNewEvent, rhs.WaitForNewEvent) {\n\t\treturn false\n\t}\n\tif !_HistoryEventFilterType_EqualsPtr(v.HistoryEventFilterType, rhs.HistoryEventFilterType) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.SkipArchival, rhs.SkipArchival) {\n\t\treturn false\n\t}\n\tif !_QueryConsistencyLevel_EqualsPtr(v.QueryConsistencyLevel, rhs.QueryConsistencyLevel) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetWorkflowExecutionHistoryRequest.\nfunc (v *GetWorkflowExecutionHistoryRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.WaitForNewEvent != nil {\n\t\tenc.AddBool(\"waitForNewEvent\", *v.WaitForNewEvent)\n\t}\n\tif v.HistoryEventFilterType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"HistoryEventFilterType\", *v.HistoryEventFilterType))\n\t}\n\tif v.SkipArchival != nil {\n\t\tenc.AddBool(\"skipArchival\", *v.SkipArchival)\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryConsistencyLevel\", *v.QueryConsistencyLevel))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetWaitForNewEvent returns the value of WaitForNewEvent if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetWaitForNewEvent() (o bool) {\n\tif v != nil && v.WaitForNewEvent != nil {\n\t\treturn *v.WaitForNewEvent\n\t}\n\n\treturn\n}\n\n// IsSetWaitForNewEvent returns true if WaitForNewEvent is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetWaitForNewEvent() bool {\n\treturn v != nil && v.WaitForNewEvent != nil\n}\n\n// GetHistoryEventFilterType returns the value of HistoryEventFilterType if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetHistoryEventFilterType() (o HistoryEventFilterType) {\n\tif v != nil && v.HistoryEventFilterType != nil {\n\t\treturn *v.HistoryEventFilterType\n\t}\n\n\treturn\n}\n\n// IsSetHistoryEventFilterType returns true if HistoryEventFilterType is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetHistoryEventFilterType() bool {\n\treturn v != nil && v.HistoryEventFilterType != nil\n}\n\n// GetSkipArchival returns the value of SkipArchival if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetSkipArchival() (o bool) {\n\tif v != nil && v.SkipArchival != nil {\n\t\treturn *v.SkipArchival\n\t}\n\n\treturn\n}\n\n// IsSetSkipArchival returns true if SkipArchival is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetSkipArchival() bool {\n\treturn v != nil && v.SkipArchival != nil\n}\n\n// GetQueryConsistencyLevel returns the value of QueryConsistencyLevel if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\n\treturn\n}\n\n// IsSetQueryConsistencyLevel returns true if QueryConsistencyLevel is not nil.\nfunc (v *GetWorkflowExecutionHistoryRequest) IsSetQueryConsistencyLevel() bool {\n\treturn v != nil && v.QueryConsistencyLevel != nil\n}\n\ntype GetWorkflowExecutionHistoryResponse struct {\n\tHistory       *History    `json:\"history,omitempty\"`\n\tRawHistory    []*DataBlob `json:\"rawHistory,omitempty\"`\n\tNextPageToken []byte      `json:\"nextPageToken,omitempty\"`\n\tArchived      *bool       `json:\"archived,omitempty\"`\n}\n\ntype _List_DataBlob_ValueList []*DataBlob\n\nfunc (v _List_DataBlob_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DataBlob', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DataBlob_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DataBlob_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DataBlob_ValueList) Close() {}\n\n// ToWire translates a GetWorkflowExecutionHistoryResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *GetWorkflowExecutionHistoryResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.History != nil {\n\t\tw, err = v.History.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RawHistory != nil {\n\t\tw, err = wire.NewValueList(_List_DataBlob_ValueList(v.RawHistory)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Archived != nil {\n\t\tw, err = wire.NewValueBool(*(v.Archived)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _History_Read(w wire.Value) (*History, error) {\n\tvar v History\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DataBlob_Read(l wire.ValueList) ([]*DataBlob, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*DataBlob, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DataBlob_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a GetWorkflowExecutionHistoryResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a GetWorkflowExecutionHistoryResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v GetWorkflowExecutionHistoryResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *GetWorkflowExecutionHistoryResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.History, err = _History_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.RawHistory, err = _List_DataBlob_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Archived = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DataBlob_Encode(val []*DataBlob, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DataBlob', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a GetWorkflowExecutionHistoryResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a GetWorkflowExecutionHistoryResponse struct could not be encoded.\nfunc (v *GetWorkflowExecutionHistoryResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.History != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.History.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RawHistory != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DataBlob_Encode(v.RawHistory, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Archived != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Archived)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _History_Decode(sr stream.Reader) (*History, error) {\n\tvar v History\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DataBlob_Decode(sr stream.Reader) ([]*DataBlob, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*DataBlob, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DataBlob_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a GetWorkflowExecutionHistoryResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a GetWorkflowExecutionHistoryResponse struct could not be generated from the wire\n// representation.\nfunc (v *GetWorkflowExecutionHistoryResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.History, err = _History_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TList:\n\t\t\tv.RawHistory, err = _List_DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Archived = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a GetWorkflowExecutionHistoryResponse\n// struct.\nfunc (v *GetWorkflowExecutionHistoryResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.History != nil {\n\t\tfields[i] = fmt.Sprintf(\"History: %v\", v.History)\n\t\ti++\n\t}\n\tif v.RawHistory != nil {\n\t\tfields[i] = fmt.Sprintf(\"RawHistory: %v\", v.RawHistory)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.Archived != nil {\n\t\tfields[i] = fmt.Sprintf(\"Archived: %v\", *(v.Archived))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"GetWorkflowExecutionHistoryResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DataBlob_Equals(lhs, rhs []*DataBlob) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this GetWorkflowExecutionHistoryResponse match the\n// provided GetWorkflowExecutionHistoryResponse.\n//\n// This function performs a deep comparison.\nfunc (v *GetWorkflowExecutionHistoryResponse) Equals(rhs *GetWorkflowExecutionHistoryResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.History == nil && rhs.History == nil) || (v.History != nil && rhs.History != nil && v.History.Equals(rhs.History))) {\n\t\treturn false\n\t}\n\tif !((v.RawHistory == nil && rhs.RawHistory == nil) || (v.RawHistory != nil && rhs.RawHistory != nil && _List_DataBlob_Equals(v.RawHistory, rhs.RawHistory))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Archived, rhs.Archived) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DataBlob_Zapper []*DataBlob\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DataBlob_Zapper.\nfunc (l _List_DataBlob_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of GetWorkflowExecutionHistoryResponse.\nfunc (v *GetWorkflowExecutionHistoryResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.History != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"history\", v.History))\n\t}\n\tif v.RawHistory != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"rawHistory\", (_List_DataBlob_Zapper)(v.RawHistory)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.Archived != nil {\n\t\tenc.AddBool(\"archived\", *v.Archived)\n\t}\n\treturn err\n}\n\n// GetHistory returns the value of History if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryResponse) GetHistory() (o *History) {\n\tif v != nil && v.History != nil {\n\t\treturn v.History\n\t}\n\n\treturn\n}\n\n// IsSetHistory returns true if History is not nil.\nfunc (v *GetWorkflowExecutionHistoryResponse) IsSetHistory() bool {\n\treturn v != nil && v.History != nil\n}\n\n// GetRawHistory returns the value of RawHistory if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryResponse) GetRawHistory() (o []*DataBlob) {\n\tif v != nil && v.RawHistory != nil {\n\t\treturn v.RawHistory\n\t}\n\n\treturn\n}\n\n// IsSetRawHistory returns true if RawHistory is not nil.\nfunc (v *GetWorkflowExecutionHistoryResponse) IsSetRawHistory() bool {\n\treturn v != nil && v.RawHistory != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *GetWorkflowExecutionHistoryResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetArchived returns the value of Archived if it is set or its\n// zero value if it is unset.\nfunc (v *GetWorkflowExecutionHistoryResponse) GetArchived() (o bool) {\n\tif v != nil && v.Archived != nil {\n\t\treturn *v.Archived\n\t}\n\n\treturn\n}\n\n// IsSetArchived returns true if Archived is not nil.\nfunc (v *GetWorkflowExecutionHistoryResponse) IsSetArchived() bool {\n\treturn v != nil && v.Archived != nil\n}\n\ntype Header struct {\n\tFields map[string][]byte `json:\"fields,omitempty\"`\n}\n\ntype _Map_String_Binary_MapItemList map[string][]byte\n\nfunc (m _Map_String_Binary_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]byte', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueBinary(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_Binary_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_Binary_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Binary_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Binary_MapItemList) Close() {}\n\n// ToWire translates a Header struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Header) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Fields != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Binary_MapItemList(v.Fields)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_String_Binary_Read(m wire.MapItemList) (map[string][]byte, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string][]byte, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetBinary(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a Header struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Header struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Header\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Header) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Fields, err = _Map_String_Binary_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_Binary_Encode(val map[string][]byte, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]byte', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a Header struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Header struct could not be encoded.\nfunc (v *Header) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Fields != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Binary_Encode(v.Fields, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_String_Binary_Decode(sr stream.Reader) (map[string][]byte, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string][]byte, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadBinary()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a Header struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Header struct could not be generated from the wire\n// representation.\nfunc (v *Header) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Fields, err = _Map_String_Binary_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Header\n// struct.\nfunc (v *Header) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Fields != nil {\n\t\tfields[i] = fmt.Sprintf(\"Fields: %v\", v.Fields)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Header{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_Binary_Equals(lhs, rhs map[string][]byte) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !bytes.Equal(lv, rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this Header match the\n// provided Header.\n//\n// This function performs a deep comparison.\nfunc (v *Header) Equals(rhs *Header) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Fields == nil && rhs.Fields == nil) || (v.Fields != nil && rhs.Fields != nil && _Map_String_Binary_Equals(v.Fields, rhs.Fields))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_Binary_Zapper map[string][]byte\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_Binary_Zapper.\nfunc (m _Map_String_Binary_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), base64.StdEncoding.EncodeToString(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Header.\nfunc (v *Header) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Fields != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"fields\", (_Map_String_Binary_Zapper)(v.Fields)))\n\t}\n\treturn err\n}\n\n// GetFields returns the value of Fields if it is set or its\n// zero value if it is unset.\nfunc (v *Header) GetFields() (o map[string][]byte) {\n\tif v != nil && v.Fields != nil {\n\t\treturn v.Fields\n\t}\n\n\treturn\n}\n\n// IsSetFields returns true if Fields is not nil.\nfunc (v *Header) IsSetFields() bool {\n\treturn v != nil && v.Fields != nil\n}\n\ntype History struct {\n\tEvents []*HistoryEvent `json:\"events,omitempty\"`\n}\n\ntype _List_HistoryEvent_ValueList []*HistoryEvent\n\nfunc (v _List_HistoryEvent_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HistoryEvent', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_HistoryEvent_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_HistoryEvent_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_HistoryEvent_ValueList) Close() {}\n\n// ToWire translates a History struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *History) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Events != nil {\n\t\tw, err = wire.NewValueList(_List_HistoryEvent_ValueList(v.Events)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_HistoryEvent_Read(l wire.ValueList) ([]*HistoryEvent, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*HistoryEvent, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _HistoryEvent_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a History struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a History struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v History\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *History) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Events, err = _List_HistoryEvent_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_HistoryEvent_Encode(val []*HistoryEvent, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HistoryEvent', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a History struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a History struct could not be encoded.\nfunc (v *History) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Events != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_HistoryEvent_Encode(v.Events, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_HistoryEvent_Decode(sr stream.Reader) ([]*HistoryEvent, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*HistoryEvent, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _HistoryEvent_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a History struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a History struct could not be generated from the wire\n// representation.\nfunc (v *History) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Events, err = _List_HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a History\n// struct.\nfunc (v *History) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Events != nil {\n\t\tfields[i] = fmt.Sprintf(\"Events: %v\", v.Events)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"History{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_HistoryEvent_Equals(lhs, rhs []*HistoryEvent) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this History match the\n// provided History.\n//\n// This function performs a deep comparison.\nfunc (v *History) Equals(rhs *History) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Events == nil && rhs.Events == nil) || (v.Events != nil && rhs.Events != nil && _List_HistoryEvent_Equals(v.Events, rhs.Events))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_HistoryEvent_Zapper []*HistoryEvent\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_HistoryEvent_Zapper.\nfunc (l _List_HistoryEvent_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of History.\nfunc (v *History) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Events != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"events\", (_List_HistoryEvent_Zapper)(v.Events)))\n\t}\n\treturn err\n}\n\n// GetEvents returns the value of Events if it is set or its\n// zero value if it is unset.\nfunc (v *History) GetEvents() (o []*HistoryEvent) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\n\treturn\n}\n\n// IsSetEvents returns true if Events is not nil.\nfunc (v *History) IsSetEvents() bool {\n\treturn v != nil && v.Events != nil\n}\n\ntype HistoryBranch struct {\n\tTreeID    *string               `json:\"treeID,omitempty\"`\n\tBranchID  *string               `json:\"branchID,omitempty\"`\n\tAncestors []*HistoryBranchRange `json:\"ancestors,omitempty\"`\n}\n\ntype _List_HistoryBranchRange_ValueList []*HistoryBranchRange\n\nfunc (v _List_HistoryBranchRange_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HistoryBranchRange', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_HistoryBranchRange_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_HistoryBranchRange_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_HistoryBranchRange_ValueList) Close() {}\n\n// ToWire translates a HistoryBranch struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryBranch) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TreeID != nil {\n\t\tw, err = wire.NewValueString(*(v.TreeID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.BranchID != nil {\n\t\tw, err = wire.NewValueString(*(v.BranchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Ancestors != nil {\n\t\tw, err = wire.NewValueList(_List_HistoryBranchRange_ValueList(v.Ancestors)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryBranchRange_Read(w wire.Value) (*HistoryBranchRange, error) {\n\tvar v HistoryBranchRange\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_HistoryBranchRange_Read(l wire.ValueList) ([]*HistoryBranchRange, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*HistoryBranchRange, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _HistoryBranchRange_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a HistoryBranch struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryBranch struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryBranch\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryBranch) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TreeID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BranchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Ancestors, err = _List_HistoryBranchRange_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_HistoryBranchRange_Encode(val []*HistoryBranchRange, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*HistoryBranchRange', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a HistoryBranch struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryBranch struct could not be encoded.\nfunc (v *HistoryBranch) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TreeID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TreeID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BranchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BranchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Ancestors != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_HistoryBranchRange_Encode(v.Ancestors, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryBranchRange_Decode(sr stream.Reader) (*HistoryBranchRange, error) {\n\tvar v HistoryBranchRange\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_HistoryBranchRange_Decode(sr stream.Reader) ([]*HistoryBranchRange, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*HistoryBranchRange, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _HistoryBranchRange_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a HistoryBranch struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryBranch struct could not be generated from the wire\n// representation.\nfunc (v *HistoryBranch) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TreeID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BranchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.Ancestors, err = _List_HistoryBranchRange_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryBranch\n// struct.\nfunc (v *HistoryBranch) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TreeID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TreeID: %v\", *(v.TreeID))\n\t\ti++\n\t}\n\tif v.BranchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchID: %v\", *(v.BranchID))\n\t\ti++\n\t}\n\tif v.Ancestors != nil {\n\t\tfields[i] = fmt.Sprintf(\"Ancestors: %v\", v.Ancestors)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryBranch{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_HistoryBranchRange_Equals(lhs, rhs []*HistoryBranchRange) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this HistoryBranch match the\n// provided HistoryBranch.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryBranch) Equals(rhs *HistoryBranch) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TreeID, rhs.TreeID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BranchID, rhs.BranchID) {\n\t\treturn false\n\t}\n\tif !((v.Ancestors == nil && rhs.Ancestors == nil) || (v.Ancestors != nil && rhs.Ancestors != nil && _List_HistoryBranchRange_Equals(v.Ancestors, rhs.Ancestors))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_HistoryBranchRange_Zapper []*HistoryBranchRange\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_HistoryBranchRange_Zapper.\nfunc (l _List_HistoryBranchRange_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryBranch.\nfunc (v *HistoryBranch) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TreeID != nil {\n\t\tenc.AddString(\"treeID\", *v.TreeID)\n\t}\n\tif v.BranchID != nil {\n\t\tenc.AddString(\"branchID\", *v.BranchID)\n\t}\n\tif v.Ancestors != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"ancestors\", (_List_HistoryBranchRange_Zapper)(v.Ancestors)))\n\t}\n\treturn err\n}\n\n// GetTreeID returns the value of TreeID if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranch) GetTreeID() (o string) {\n\tif v != nil && v.TreeID != nil {\n\t\treturn *v.TreeID\n\t}\n\n\treturn\n}\n\n// IsSetTreeID returns true if TreeID is not nil.\nfunc (v *HistoryBranch) IsSetTreeID() bool {\n\treturn v != nil && v.TreeID != nil\n}\n\n// GetBranchID returns the value of BranchID if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranch) GetBranchID() (o string) {\n\tif v != nil && v.BranchID != nil {\n\t\treturn *v.BranchID\n\t}\n\n\treturn\n}\n\n// IsSetBranchID returns true if BranchID is not nil.\nfunc (v *HistoryBranch) IsSetBranchID() bool {\n\treturn v != nil && v.BranchID != nil\n}\n\n// GetAncestors returns the value of Ancestors if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranch) GetAncestors() (o []*HistoryBranchRange) {\n\tif v != nil && v.Ancestors != nil {\n\t\treturn v.Ancestors\n\t}\n\n\treturn\n}\n\n// IsSetAncestors returns true if Ancestors is not nil.\nfunc (v *HistoryBranch) IsSetAncestors() bool {\n\treturn v != nil && v.Ancestors != nil\n}\n\ntype HistoryBranchRange struct {\n\tBranchID    *string `json:\"branchID,omitempty\"`\n\tBeginNodeID *int64  `json:\"beginNodeID,omitempty\"`\n\tEndNodeID   *int64  `json:\"endNodeID,omitempty\"`\n}\n\n// ToWire translates a HistoryBranchRange struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryBranchRange) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BranchID != nil {\n\t\tw, err = wire.NewValueString(*(v.BranchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.BeginNodeID != nil {\n\t\tw, err = wire.NewValueI64(*(v.BeginNodeID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.EndNodeID != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndNodeID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a HistoryBranchRange struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryBranchRange struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryBranchRange\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryBranchRange) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BranchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.BeginNodeID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndNodeID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryBranchRange struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryBranchRange struct could not be encoded.\nfunc (v *HistoryBranchRange) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BranchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BranchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BeginNodeID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.BeginNodeID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndNodeID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndNodeID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a HistoryBranchRange struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryBranchRange struct could not be generated from the wire\n// representation.\nfunc (v *HistoryBranchRange) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BranchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.BeginNodeID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndNodeID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryBranchRange\n// struct.\nfunc (v *HistoryBranchRange) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.BranchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchID: %v\", *(v.BranchID))\n\t\ti++\n\t}\n\tif v.BeginNodeID != nil {\n\t\tfields[i] = fmt.Sprintf(\"BeginNodeID: %v\", *(v.BeginNodeID))\n\t\ti++\n\t}\n\tif v.EndNodeID != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndNodeID: %v\", *(v.EndNodeID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryBranchRange{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this HistoryBranchRange match the\n// provided HistoryBranchRange.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryBranchRange) Equals(rhs *HistoryBranchRange) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BranchID, rhs.BranchID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.BeginNodeID, rhs.BeginNodeID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndNodeID, rhs.EndNodeID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryBranchRange.\nfunc (v *HistoryBranchRange) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BranchID != nil {\n\t\tenc.AddString(\"branchID\", *v.BranchID)\n\t}\n\tif v.BeginNodeID != nil {\n\t\tenc.AddInt64(\"beginNodeID\", *v.BeginNodeID)\n\t}\n\tif v.EndNodeID != nil {\n\t\tenc.AddInt64(\"endNodeID\", *v.EndNodeID)\n\t}\n\treturn err\n}\n\n// GetBranchID returns the value of BranchID if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranchRange) GetBranchID() (o string) {\n\tif v != nil && v.BranchID != nil {\n\t\treturn *v.BranchID\n\t}\n\n\treturn\n}\n\n// IsSetBranchID returns true if BranchID is not nil.\nfunc (v *HistoryBranchRange) IsSetBranchID() bool {\n\treturn v != nil && v.BranchID != nil\n}\n\n// GetBeginNodeID returns the value of BeginNodeID if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranchRange) GetBeginNodeID() (o int64) {\n\tif v != nil && v.BeginNodeID != nil {\n\t\treturn *v.BeginNodeID\n\t}\n\n\treturn\n}\n\n// IsSetBeginNodeID returns true if BeginNodeID is not nil.\nfunc (v *HistoryBranchRange) IsSetBeginNodeID() bool {\n\treturn v != nil && v.BeginNodeID != nil\n}\n\n// GetEndNodeID returns the value of EndNodeID if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryBranchRange) GetEndNodeID() (o int64) {\n\tif v != nil && v.EndNodeID != nil {\n\t\treturn *v.EndNodeID\n\t}\n\n\treturn\n}\n\n// IsSetEndNodeID returns true if EndNodeID is not nil.\nfunc (v *HistoryBranchRange) IsSetEndNodeID() bool {\n\treturn v != nil && v.EndNodeID != nil\n}\n\ntype HistoryEvent struct {\n\tEventId                                                        *int64                                                          `json:\"eventId,omitempty\"`\n\tTimestamp                                                      *int64                                                          `json:\"timestamp,omitempty\"`\n\tEventType                                                      *EventType                                                      `json:\"eventType,omitempty\"`\n\tVersion                                                        *int64                                                          `json:\"version,omitempty\"`\n\tTaskId                                                         *int64                                                          `json:\"taskId,omitempty\"`\n\tWorkflowExecutionStartedEventAttributes                        *WorkflowExecutionStartedEventAttributes                        `json:\"workflowExecutionStartedEventAttributes,omitempty\"`\n\tWorkflowExecutionCompletedEventAttributes                      *WorkflowExecutionCompletedEventAttributes                      `json:\"workflowExecutionCompletedEventAttributes,omitempty\"`\n\tWorkflowExecutionFailedEventAttributes                         *WorkflowExecutionFailedEventAttributes                         `json:\"workflowExecutionFailedEventAttributes,omitempty\"`\n\tWorkflowExecutionTimedOutEventAttributes                       *WorkflowExecutionTimedOutEventAttributes                       `json:\"workflowExecutionTimedOutEventAttributes,omitempty\"`\n\tDecisionTaskScheduledEventAttributes                           *DecisionTaskScheduledEventAttributes                           `json:\"decisionTaskScheduledEventAttributes,omitempty\"`\n\tDecisionTaskStartedEventAttributes                             *DecisionTaskStartedEventAttributes                             `json:\"decisionTaskStartedEventAttributes,omitempty\"`\n\tDecisionTaskCompletedEventAttributes                           *DecisionTaskCompletedEventAttributes                           `json:\"decisionTaskCompletedEventAttributes,omitempty\"`\n\tDecisionTaskTimedOutEventAttributes                            *DecisionTaskTimedOutEventAttributes                            `json:\"decisionTaskTimedOutEventAttributes,omitempty\"`\n\tDecisionTaskFailedEventAttributes                              *DecisionTaskFailedEventAttributes                              `json:\"decisionTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskScheduledEventAttributes                           *ActivityTaskScheduledEventAttributes                           `json:\"activityTaskScheduledEventAttributes,omitempty\"`\n\tActivityTaskStartedEventAttributes                             *ActivityTaskStartedEventAttributes                             `json:\"activityTaskStartedEventAttributes,omitempty\"`\n\tActivityTaskCompletedEventAttributes                           *ActivityTaskCompletedEventAttributes                           `json:\"activityTaskCompletedEventAttributes,omitempty\"`\n\tActivityTaskFailedEventAttributes                              *ActivityTaskFailedEventAttributes                              `json:\"activityTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskTimedOutEventAttributes                            *ActivityTaskTimedOutEventAttributes                            `json:\"activityTaskTimedOutEventAttributes,omitempty\"`\n\tTimerStartedEventAttributes                                    *TimerStartedEventAttributes                                    `json:\"timerStartedEventAttributes,omitempty\"`\n\tTimerFiredEventAttributes                                      *TimerFiredEventAttributes                                      `json:\"timerFiredEventAttributes,omitempty\"`\n\tActivityTaskCancelRequestedEventAttributes                     *ActivityTaskCancelRequestedEventAttributes                     `json:\"activityTaskCancelRequestedEventAttributes,omitempty\"`\n\tRequestCancelActivityTaskFailedEventAttributes                 *RequestCancelActivityTaskFailedEventAttributes                 `json:\"requestCancelActivityTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskCanceledEventAttributes                            *ActivityTaskCanceledEventAttributes                            `json:\"activityTaskCanceledEventAttributes,omitempty\"`\n\tTimerCanceledEventAttributes                                   *TimerCanceledEventAttributes                                   `json:\"timerCanceledEventAttributes,omitempty\"`\n\tCancelTimerFailedEventAttributes                               *CancelTimerFailedEventAttributes                               `json:\"cancelTimerFailedEventAttributes,omitempty\"`\n\tMarkerRecordedEventAttributes                                  *MarkerRecordedEventAttributes                                  `json:\"markerRecordedEventAttributes,omitempty\"`\n\tWorkflowExecutionSignaledEventAttributes                       *WorkflowExecutionSignaledEventAttributes                       `json:\"workflowExecutionSignaledEventAttributes,omitempty\"`\n\tWorkflowExecutionTerminatedEventAttributes                     *WorkflowExecutionTerminatedEventAttributes                     `json:\"workflowExecutionTerminatedEventAttributes,omitempty\"`\n\tWorkflowExecutionCancelRequestedEventAttributes                *WorkflowExecutionCancelRequestedEventAttributes                `json:\"workflowExecutionCancelRequestedEventAttributes,omitempty\"`\n\tWorkflowExecutionCanceledEventAttributes                       *WorkflowExecutionCanceledEventAttributes                       `json:\"workflowExecutionCanceledEventAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes `json:\"requestCancelExternalWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionFailedEventAttributes    *RequestCancelExternalWorkflowExecutionFailedEventAttributes    `json:\"requestCancelExternalWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tExternalWorkflowExecutionCancelRequestedEventAttributes        *ExternalWorkflowExecutionCancelRequestedEventAttributes        `json:\"externalWorkflowExecutionCancelRequestedEventAttributes,omitempty\"`\n\tWorkflowExecutionContinuedAsNewEventAttributes                 *WorkflowExecutionContinuedAsNewEventAttributes                 `json:\"workflowExecutionContinuedAsNewEventAttributes,omitempty\"`\n\tStartChildWorkflowExecutionInitiatedEventAttributes            *StartChildWorkflowExecutionInitiatedEventAttributes            `json:\"startChildWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tStartChildWorkflowExecutionFailedEventAttributes               *StartChildWorkflowExecutionFailedEventAttributes               `json:\"startChildWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionStartedEventAttributes                   *ChildWorkflowExecutionStartedEventAttributes                   `json:\"childWorkflowExecutionStartedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionCompletedEventAttributes                 *ChildWorkflowExecutionCompletedEventAttributes                 `json:\"childWorkflowExecutionCompletedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionFailedEventAttributes                    *ChildWorkflowExecutionFailedEventAttributes                    `json:\"childWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionCanceledEventAttributes                  *ChildWorkflowExecutionCanceledEventAttributes                  `json:\"childWorkflowExecutionCanceledEventAttributes,omitempty\"`\n\tChildWorkflowExecutionTimedOutEventAttributes                  *ChildWorkflowExecutionTimedOutEventAttributes                  `json:\"childWorkflowExecutionTimedOutEventAttributes,omitempty\"`\n\tChildWorkflowExecutionTerminatedEventAttributes                *ChildWorkflowExecutionTerminatedEventAttributes                `json:\"childWorkflowExecutionTerminatedEventAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionInitiatedEventAttributes        *SignalExternalWorkflowExecutionInitiatedEventAttributes        `json:\"signalExternalWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionFailedEventAttributes           *SignalExternalWorkflowExecutionFailedEventAttributes           `json:\"signalExternalWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tExternalWorkflowExecutionSignaledEventAttributes               *ExternalWorkflowExecutionSignaledEventAttributes               `json:\"externalWorkflowExecutionSignaledEventAttributes,omitempty\"`\n\tUpsertWorkflowSearchAttributesEventAttributes                  *UpsertWorkflowSearchAttributesEventAttributes                  `json:\"upsertWorkflowSearchAttributesEventAttributes,omitempty\"`\n}\n\n// ToWire translates a HistoryEvent struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryEvent) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [47]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.EventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.EventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.Timestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.EventType != nil {\n\t\tw, err = v.EventType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionStartedEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionStartedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionCompletedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionFailedEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionTimedOutEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskScheduledEventAttributes != nil {\n\t\tw, err = v.DecisionTaskScheduledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskStartedEventAttributes != nil {\n\t\tw, err = v.DecisionTaskStartedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventAttributes != nil {\n\t\tw, err = v.DecisionTaskCompletedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskTimedOutEventAttributes != nil {\n\t\tw, err = v.DecisionTaskTimedOutEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskFailedEventAttributes != nil {\n\t\tw, err = v.DecisionTaskFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskScheduledEventAttributes != nil {\n\t\tw, err = v.ActivityTaskScheduledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskStartedEventAttributes != nil {\n\t\tw, err = v.ActivityTaskStartedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskCompletedEventAttributes != nil {\n\t\tw, err = v.ActivityTaskCompletedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskFailedEventAttributes != nil {\n\t\tw, err = v.ActivityTaskFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskTimedOutEventAttributes != nil {\n\t\tw, err = v.ActivityTaskTimedOutEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.TimerStartedEventAttributes != nil {\n\t\tw, err = v.TimerStartedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.TimerFiredEventAttributes != nil {\n\t\tw, err = v.TimerFiredEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\tw, err = v.ActivityTaskCancelRequestedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 200, Value: w}\n\t\ti++\n\t}\n\tif v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\tw, err = v.RequestCancelActivityTaskFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 210, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityTaskCanceledEventAttributes != nil {\n\t\tw, err = v.ActivityTaskCanceledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 220, Value: w}\n\t\ti++\n\t}\n\tif v.TimerCanceledEventAttributes != nil {\n\t\tw, err = v.TimerCanceledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 230, Value: w}\n\t\ti++\n\t}\n\tif v.CancelTimerFailedEventAttributes != nil {\n\t\tw, err = v.CancelTimerFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 240, Value: w}\n\t\ti++\n\t}\n\tif v.MarkerRecordedEventAttributes != nil {\n\t\tw, err = v.MarkerRecordedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 250, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionSignaledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 260, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionTerminatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 270, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionCancelRequestedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 280, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionCanceledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 290, Value: w}\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tw, err = v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 300, Value: w}\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tw, err = v.RequestCancelExternalWorkflowExecutionFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 310, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tw, err = v.ExternalWorkflowExecutionCancelRequestedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 320, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\tw, err = v.WorkflowExecutionContinuedAsNewEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 330, Value: w}\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tw, err = v.StartChildWorkflowExecutionInitiatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 340, Value: w}\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tw, err = v.StartChildWorkflowExecutionFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 350, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionStartedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 360, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionCompletedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 370, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 380, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionCanceledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 390, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionTimedOutEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 400, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\tw, err = v.ChildWorkflowExecutionTerminatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 410, Value: w}\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tw, err = v.SignalExternalWorkflowExecutionInitiatedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 420, Value: w}\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tw, err = v.SignalExternalWorkflowExecutionFailedEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 430, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\tw, err = v.ExternalWorkflowExecutionSignaledEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 440, Value: w}\n\t\ti++\n\t}\n\tif v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\tw, err = v.UpsertWorkflowSearchAttributesEventAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 450, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _EventType_Read(w wire.Value) (EventType, error) {\n\tvar v EventType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _WorkflowExecutionStartedEventAttributes_Read(w wire.Value) (*WorkflowExecutionStartedEventAttributes, error) {\n\tvar v WorkflowExecutionStartedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCompletedEventAttributes_Read(w wire.Value) (*WorkflowExecutionCompletedEventAttributes, error) {\n\tvar v WorkflowExecutionCompletedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionFailedEventAttributes_Read(w wire.Value) (*WorkflowExecutionFailedEventAttributes, error) {\n\tvar v WorkflowExecutionFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionTimedOutEventAttributes_Read(w wire.Value) (*WorkflowExecutionTimedOutEventAttributes, error) {\n\tvar v WorkflowExecutionTimedOutEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DecisionTaskScheduledEventAttributes_Read(w wire.Value) (*DecisionTaskScheduledEventAttributes, error) {\n\tvar v DecisionTaskScheduledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DecisionTaskStartedEventAttributes_Read(w wire.Value) (*DecisionTaskStartedEventAttributes, error) {\n\tvar v DecisionTaskStartedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DecisionTaskCompletedEventAttributes_Read(w wire.Value) (*DecisionTaskCompletedEventAttributes, error) {\n\tvar v DecisionTaskCompletedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DecisionTaskTimedOutEventAttributes_Read(w wire.Value) (*DecisionTaskTimedOutEventAttributes, error) {\n\tvar v DecisionTaskTimedOutEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DecisionTaskFailedEventAttributes_Read(w wire.Value) (*DecisionTaskFailedEventAttributes, error) {\n\tvar v DecisionTaskFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskScheduledEventAttributes_Read(w wire.Value) (*ActivityTaskScheduledEventAttributes, error) {\n\tvar v ActivityTaskScheduledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskStartedEventAttributes_Read(w wire.Value) (*ActivityTaskStartedEventAttributes, error) {\n\tvar v ActivityTaskStartedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCompletedEventAttributes_Read(w wire.Value) (*ActivityTaskCompletedEventAttributes, error) {\n\tvar v ActivityTaskCompletedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskFailedEventAttributes_Read(w wire.Value) (*ActivityTaskFailedEventAttributes, error) {\n\tvar v ActivityTaskFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskTimedOutEventAttributes_Read(w wire.Value) (*ActivityTaskTimedOutEventAttributes, error) {\n\tvar v ActivityTaskTimedOutEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TimerStartedEventAttributes_Read(w wire.Value) (*TimerStartedEventAttributes, error) {\n\tvar v TimerStartedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TimerFiredEventAttributes_Read(w wire.Value) (*TimerFiredEventAttributes, error) {\n\tvar v TimerFiredEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCancelRequestedEventAttributes_Read(w wire.Value) (*ActivityTaskCancelRequestedEventAttributes, error) {\n\tvar v ActivityTaskCancelRequestedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RequestCancelActivityTaskFailedEventAttributes_Read(w wire.Value) (*RequestCancelActivityTaskFailedEventAttributes, error) {\n\tvar v RequestCancelActivityTaskFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCanceledEventAttributes_Read(w wire.Value) (*ActivityTaskCanceledEventAttributes, error) {\n\tvar v ActivityTaskCanceledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _TimerCanceledEventAttributes_Read(w wire.Value) (*TimerCanceledEventAttributes, error) {\n\tvar v TimerCanceledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _CancelTimerFailedEventAttributes_Read(w wire.Value) (*CancelTimerFailedEventAttributes, error) {\n\tvar v CancelTimerFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _MarkerRecordedEventAttributes_Read(w wire.Value) (*MarkerRecordedEventAttributes, error) {\n\tvar v MarkerRecordedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionSignaledEventAttributes_Read(w wire.Value) (*WorkflowExecutionSignaledEventAttributes, error) {\n\tvar v WorkflowExecutionSignaledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionTerminatedEventAttributes_Read(w wire.Value) (*WorkflowExecutionTerminatedEventAttributes, error) {\n\tvar v WorkflowExecutionTerminatedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCancelRequestedEventAttributes_Read(w wire.Value) (*WorkflowExecutionCancelRequestedEventAttributes, error) {\n\tvar v WorkflowExecutionCancelRequestedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCanceledEventAttributes_Read(w wire.Value) (*WorkflowExecutionCanceledEventAttributes, error) {\n\tvar v WorkflowExecutionCanceledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionInitiatedEventAttributes_Read(w wire.Value) (*RequestCancelExternalWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionFailedEventAttributes_Read(w wire.Value) (*RequestCancelExternalWorkflowExecutionFailedEventAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ExternalWorkflowExecutionCancelRequestedEventAttributes_Read(w wire.Value) (*ExternalWorkflowExecutionCancelRequestedEventAttributes, error) {\n\tvar v ExternalWorkflowExecutionCancelRequestedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionContinuedAsNewEventAttributes_Read(w wire.Value) (*WorkflowExecutionContinuedAsNewEventAttributes, error) {\n\tvar v WorkflowExecutionContinuedAsNewEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _StartChildWorkflowExecutionFailedEventAttributes_Read(w wire.Value) (*StartChildWorkflowExecutionFailedEventAttributes, error) {\n\tvar v StartChildWorkflowExecutionFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionStartedEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionStartedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionStartedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionCompletedEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionCompletedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionCompletedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionFailedEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionFailedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionCanceledEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionCanceledEventAttributes, error) {\n\tvar v ChildWorkflowExecutionCanceledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionTimedOutEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionTimedOutEventAttributes, error) {\n\tvar v ChildWorkflowExecutionTimedOutEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionTerminatedEventAttributes_Read(w wire.Value) (*ChildWorkflowExecutionTerminatedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionTerminatedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionInitiatedEventAttributes_Read(w wire.Value) (*SignalExternalWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionInitiatedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionFailedEventAttributes_Read(w wire.Value) (*SignalExternalWorkflowExecutionFailedEventAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionFailedEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ExternalWorkflowExecutionSignaledEventAttributes_Read(w wire.Value) (*ExternalWorkflowExecutionSignaledEventAttributes, error) {\n\tvar v ExternalWorkflowExecutionSignaledEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _UpsertWorkflowSearchAttributesEventAttributes_Read(w wire.Value) (*UpsertWorkflowSearchAttributesEventAttributes, error) {\n\tvar v UpsertWorkflowSearchAttributesEventAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a HistoryEvent struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryEvent struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryEvent\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryEvent) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Timestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x EventType\n\t\t\t\tx, err = _EventType_Read(field.Value)\n\t\t\t\tv.EventType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionStartedEventAttributes, err = _WorkflowExecutionStartedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionCompletedEventAttributes, err = _WorkflowExecutionCompletedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionFailedEventAttributes, err = _WorkflowExecutionFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionTimedOutEventAttributes, err = _WorkflowExecutionTimedOutEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTaskScheduledEventAttributes, err = _DecisionTaskScheduledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTaskStartedEventAttributes, err = _DecisionTaskStartedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTaskCompletedEventAttributes, err = _DecisionTaskCompletedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTaskTimedOutEventAttributes, err = _DecisionTaskTimedOutEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTaskFailedEventAttributes, err = _DecisionTaskFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskScheduledEventAttributes, err = _ActivityTaskScheduledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskStartedEventAttributes, err = _ActivityTaskStartedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskCompletedEventAttributes, err = _ActivityTaskCompletedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskFailedEventAttributes, err = _ActivityTaskFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskTimedOutEventAttributes, err = _ActivityTaskTimedOutEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TimerStartedEventAttributes, err = _TimerStartedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TimerFiredEventAttributes, err = _TimerFiredEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 200:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskCancelRequestedEventAttributes, err = _ActivityTaskCancelRequestedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 210:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RequestCancelActivityTaskFailedEventAttributes, err = _RequestCancelActivityTaskFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 220:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityTaskCanceledEventAttributes, err = _ActivityTaskCanceledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 230:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TimerCanceledEventAttributes, err = _TimerCanceledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 240:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.CancelTimerFailedEventAttributes, err = _CancelTimerFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 250:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.MarkerRecordedEventAttributes, err = _MarkerRecordedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 260:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionSignaledEventAttributes, err = _WorkflowExecutionSignaledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 270:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionTerminatedEventAttributes, err = _WorkflowExecutionTerminatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 280:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionCancelRequestedEventAttributes, err = _WorkflowExecutionCancelRequestedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 290:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionCanceledEventAttributes, err = _WorkflowExecutionCanceledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 300:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes, err = _RequestCancelExternalWorkflowExecutionInitiatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 310:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RequestCancelExternalWorkflowExecutionFailedEventAttributes, err = _RequestCancelExternalWorkflowExecutionFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 320:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecutionCancelRequestedEventAttributes, err = _ExternalWorkflowExecutionCancelRequestedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 330:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionContinuedAsNewEventAttributes, err = _WorkflowExecutionContinuedAsNewEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 340:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartChildWorkflowExecutionInitiatedEventAttributes, err = _StartChildWorkflowExecutionInitiatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 350:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartChildWorkflowExecutionFailedEventAttributes, err = _StartChildWorkflowExecutionFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 360:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionStartedEventAttributes, err = _ChildWorkflowExecutionStartedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 370:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionCompletedEventAttributes, err = _ChildWorkflowExecutionCompletedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 380:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionFailedEventAttributes, err = _ChildWorkflowExecutionFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 390:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionCanceledEventAttributes, err = _ChildWorkflowExecutionCanceledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 400:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionTimedOutEventAttributes, err = _ChildWorkflowExecutionTimedOutEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 410:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ChildWorkflowExecutionTerminatedEventAttributes, err = _ChildWorkflowExecutionTerminatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 420:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalExternalWorkflowExecutionInitiatedEventAttributes, err = _SignalExternalWorkflowExecutionInitiatedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 430:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SignalExternalWorkflowExecutionFailedEventAttributes, err = _SignalExternalWorkflowExecutionFailedEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 440:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecutionSignaledEventAttributes, err = _ExternalWorkflowExecutionSignaledEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 450:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.UpsertWorkflowSearchAttributesEventAttributes, err = _UpsertWorkflowSearchAttributesEventAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a HistoryEvent struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryEvent struct could not be encoded.\nfunc (v *HistoryEvent) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Timestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Timestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EventType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionStartedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionStartedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionCompletedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionTimedOutEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskScheduledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTaskScheduledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskStartedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTaskStartedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTaskCompletedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskTimedOutEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTaskTimedOutEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTaskFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskScheduledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskScheduledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskStartedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskStartedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskCompletedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskCompletedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskTimedOutEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskTimedOutEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerStartedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimerStartedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerFiredEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimerFiredEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 200, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskCancelRequestedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 210, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RequestCancelActivityTaskFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityTaskCanceledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 220, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityTaskCanceledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerCanceledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 230, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimerCanceledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelTimerFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 240, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CancelTimerFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MarkerRecordedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 250, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.MarkerRecordedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 260, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionSignaledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 270, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionTerminatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 280, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionCancelRequestedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 290, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionCanceledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 300, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 310, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RequestCancelExternalWorkflowExecutionFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 320, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecutionCancelRequestedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 330, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionContinuedAsNewEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 340, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartChildWorkflowExecutionInitiatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 350, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartChildWorkflowExecutionFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 360, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionStartedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 370, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionCompletedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 380, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 390, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionCanceledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 400, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionTimedOutEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 410, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ChildWorkflowExecutionTerminatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 420, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalExternalWorkflowExecutionInitiatedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 430, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SignalExternalWorkflowExecutionFailedEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 440, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecutionSignaledEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 450, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.UpsertWorkflowSearchAttributesEventAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _EventType_Decode(sr stream.Reader) (EventType, error) {\n\tvar v EventType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _WorkflowExecutionStartedEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionStartedEventAttributes, error) {\n\tvar v WorkflowExecutionStartedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCompletedEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionCompletedEventAttributes, error) {\n\tvar v WorkflowExecutionCompletedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionFailedEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionFailedEventAttributes, error) {\n\tvar v WorkflowExecutionFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionTimedOutEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionTimedOutEventAttributes, error) {\n\tvar v WorkflowExecutionTimedOutEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DecisionTaskScheduledEventAttributes_Decode(sr stream.Reader) (*DecisionTaskScheduledEventAttributes, error) {\n\tvar v DecisionTaskScheduledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DecisionTaskStartedEventAttributes_Decode(sr stream.Reader) (*DecisionTaskStartedEventAttributes, error) {\n\tvar v DecisionTaskStartedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DecisionTaskCompletedEventAttributes_Decode(sr stream.Reader) (*DecisionTaskCompletedEventAttributes, error) {\n\tvar v DecisionTaskCompletedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DecisionTaskTimedOutEventAttributes_Decode(sr stream.Reader) (*DecisionTaskTimedOutEventAttributes, error) {\n\tvar v DecisionTaskTimedOutEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DecisionTaskFailedEventAttributes_Decode(sr stream.Reader) (*DecisionTaskFailedEventAttributes, error) {\n\tvar v DecisionTaskFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskScheduledEventAttributes_Decode(sr stream.Reader) (*ActivityTaskScheduledEventAttributes, error) {\n\tvar v ActivityTaskScheduledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskStartedEventAttributes_Decode(sr stream.Reader) (*ActivityTaskStartedEventAttributes, error) {\n\tvar v ActivityTaskStartedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCompletedEventAttributes_Decode(sr stream.Reader) (*ActivityTaskCompletedEventAttributes, error) {\n\tvar v ActivityTaskCompletedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskFailedEventAttributes_Decode(sr stream.Reader) (*ActivityTaskFailedEventAttributes, error) {\n\tvar v ActivityTaskFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskTimedOutEventAttributes_Decode(sr stream.Reader) (*ActivityTaskTimedOutEventAttributes, error) {\n\tvar v ActivityTaskTimedOutEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TimerStartedEventAttributes_Decode(sr stream.Reader) (*TimerStartedEventAttributes, error) {\n\tvar v TimerStartedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TimerFiredEventAttributes_Decode(sr stream.Reader) (*TimerFiredEventAttributes, error) {\n\tvar v TimerFiredEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCancelRequestedEventAttributes_Decode(sr stream.Reader) (*ActivityTaskCancelRequestedEventAttributes, error) {\n\tvar v ActivityTaskCancelRequestedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RequestCancelActivityTaskFailedEventAttributes_Decode(sr stream.Reader) (*RequestCancelActivityTaskFailedEventAttributes, error) {\n\tvar v RequestCancelActivityTaskFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityTaskCanceledEventAttributes_Decode(sr stream.Reader) (*ActivityTaskCanceledEventAttributes, error) {\n\tvar v ActivityTaskCanceledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _TimerCanceledEventAttributes_Decode(sr stream.Reader) (*TimerCanceledEventAttributes, error) {\n\tvar v TimerCanceledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _CancelTimerFailedEventAttributes_Decode(sr stream.Reader) (*CancelTimerFailedEventAttributes, error) {\n\tvar v CancelTimerFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _MarkerRecordedEventAttributes_Decode(sr stream.Reader) (*MarkerRecordedEventAttributes, error) {\n\tvar v MarkerRecordedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionSignaledEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionSignaledEventAttributes, error) {\n\tvar v WorkflowExecutionSignaledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionTerminatedEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionTerminatedEventAttributes, error) {\n\tvar v WorkflowExecutionTerminatedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCancelRequestedEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionCancelRequestedEventAttributes, error) {\n\tvar v WorkflowExecutionCancelRequestedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCanceledEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionCanceledEventAttributes, error) {\n\tvar v WorkflowExecutionCanceledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionInitiatedEventAttributes_Decode(sr stream.Reader) (*RequestCancelExternalWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _RequestCancelExternalWorkflowExecutionFailedEventAttributes_Decode(sr stream.Reader) (*RequestCancelExternalWorkflowExecutionFailedEventAttributes, error) {\n\tvar v RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ExternalWorkflowExecutionCancelRequestedEventAttributes_Decode(sr stream.Reader) (*ExternalWorkflowExecutionCancelRequestedEventAttributes, error) {\n\tvar v ExternalWorkflowExecutionCancelRequestedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionContinuedAsNewEventAttributes_Decode(sr stream.Reader) (*WorkflowExecutionContinuedAsNewEventAttributes, error) {\n\tvar v WorkflowExecutionContinuedAsNewEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _StartChildWorkflowExecutionFailedEventAttributes_Decode(sr stream.Reader) (*StartChildWorkflowExecutionFailedEventAttributes, error) {\n\tvar v StartChildWorkflowExecutionFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionStartedEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionStartedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionStartedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionCompletedEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionCompletedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionCompletedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionFailedEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionFailedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionCanceledEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionCanceledEventAttributes, error) {\n\tvar v ChildWorkflowExecutionCanceledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionTimedOutEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionTimedOutEventAttributes, error) {\n\tvar v ChildWorkflowExecutionTimedOutEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ChildWorkflowExecutionTerminatedEventAttributes_Decode(sr stream.Reader) (*ChildWorkflowExecutionTerminatedEventAttributes, error) {\n\tvar v ChildWorkflowExecutionTerminatedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionInitiatedEventAttributes_Decode(sr stream.Reader) (*SignalExternalWorkflowExecutionInitiatedEventAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionInitiatedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _SignalExternalWorkflowExecutionFailedEventAttributes_Decode(sr stream.Reader) (*SignalExternalWorkflowExecutionFailedEventAttributes, error) {\n\tvar v SignalExternalWorkflowExecutionFailedEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ExternalWorkflowExecutionSignaledEventAttributes_Decode(sr stream.Reader) (*ExternalWorkflowExecutionSignaledEventAttributes, error) {\n\tvar v ExternalWorkflowExecutionSignaledEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _UpsertWorkflowSearchAttributesEventAttributes_Decode(sr stream.Reader) (*UpsertWorkflowSearchAttributesEventAttributes, error) {\n\tvar v UpsertWorkflowSearchAttributesEventAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a HistoryEvent struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryEvent struct could not be generated from the wire\n// representation.\nfunc (v *HistoryEvent) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Timestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x EventType\n\t\t\tx, err = _EventType_Decode(sr)\n\t\t\tv.EventType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionStartedEventAttributes, err = _WorkflowExecutionStartedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionCompletedEventAttributes, err = _WorkflowExecutionCompletedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionFailedEventAttributes, err = _WorkflowExecutionFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionTimedOutEventAttributes, err = _WorkflowExecutionTimedOutEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTaskScheduledEventAttributes, err = _DecisionTaskScheduledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTaskStartedEventAttributes, err = _DecisionTaskStartedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTaskCompletedEventAttributes, err = _DecisionTaskCompletedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTaskTimedOutEventAttributes, err = _DecisionTaskTimedOutEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTaskFailedEventAttributes, err = _DecisionTaskFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskScheduledEventAttributes, err = _ActivityTaskScheduledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskStartedEventAttributes, err = _ActivityTaskStartedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskCompletedEventAttributes, err = _ActivityTaskCompletedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskFailedEventAttributes, err = _ActivityTaskFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskTimedOutEventAttributes, err = _ActivityTaskTimedOutEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TStruct:\n\t\t\tv.TimerStartedEventAttributes, err = _TimerStartedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TStruct:\n\t\t\tv.TimerFiredEventAttributes, err = _TimerFiredEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 200 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskCancelRequestedEventAttributes, err = _ActivityTaskCancelRequestedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 210 && fh.Type == wire.TStruct:\n\t\t\tv.RequestCancelActivityTaskFailedEventAttributes, err = _RequestCancelActivityTaskFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 220 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityTaskCanceledEventAttributes, err = _ActivityTaskCanceledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 230 && fh.Type == wire.TStruct:\n\t\t\tv.TimerCanceledEventAttributes, err = _TimerCanceledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 240 && fh.Type == wire.TStruct:\n\t\t\tv.CancelTimerFailedEventAttributes, err = _CancelTimerFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 250 && fh.Type == wire.TStruct:\n\t\t\tv.MarkerRecordedEventAttributes, err = _MarkerRecordedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 260 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionSignaledEventAttributes, err = _WorkflowExecutionSignaledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 270 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionTerminatedEventAttributes, err = _WorkflowExecutionTerminatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 280 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionCancelRequestedEventAttributes, err = _WorkflowExecutionCancelRequestedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 290 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionCanceledEventAttributes, err = _WorkflowExecutionCanceledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 300 && fh.Type == wire.TStruct:\n\t\t\tv.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes, err = _RequestCancelExternalWorkflowExecutionInitiatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 310 && fh.Type == wire.TStruct:\n\t\t\tv.RequestCancelExternalWorkflowExecutionFailedEventAttributes, err = _RequestCancelExternalWorkflowExecutionFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 320 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecutionCancelRequestedEventAttributes, err = _ExternalWorkflowExecutionCancelRequestedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 330 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionContinuedAsNewEventAttributes, err = _WorkflowExecutionContinuedAsNewEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 340 && fh.Type == wire.TStruct:\n\t\t\tv.StartChildWorkflowExecutionInitiatedEventAttributes, err = _StartChildWorkflowExecutionInitiatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 350 && fh.Type == wire.TStruct:\n\t\t\tv.StartChildWorkflowExecutionFailedEventAttributes, err = _StartChildWorkflowExecutionFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 360 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionStartedEventAttributes, err = _ChildWorkflowExecutionStartedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 370 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionCompletedEventAttributes, err = _ChildWorkflowExecutionCompletedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 380 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionFailedEventAttributes, err = _ChildWorkflowExecutionFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 390 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionCanceledEventAttributes, err = _ChildWorkflowExecutionCanceledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 400 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionTimedOutEventAttributes, err = _ChildWorkflowExecutionTimedOutEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 410 && fh.Type == wire.TStruct:\n\t\t\tv.ChildWorkflowExecutionTerminatedEventAttributes, err = _ChildWorkflowExecutionTerminatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 420 && fh.Type == wire.TStruct:\n\t\t\tv.SignalExternalWorkflowExecutionInitiatedEventAttributes, err = _SignalExternalWorkflowExecutionInitiatedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 430 && fh.Type == wire.TStruct:\n\t\t\tv.SignalExternalWorkflowExecutionFailedEventAttributes, err = _SignalExternalWorkflowExecutionFailedEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 440 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecutionSignaledEventAttributes, err = _ExternalWorkflowExecutionSignaledEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 450 && fh.Type == wire.TStruct:\n\t\t\tv.UpsertWorkflowSearchAttributesEventAttributes, err = _UpsertWorkflowSearchAttributesEventAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryEvent\n// struct.\nfunc (v *HistoryEvent) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [47]string\n\ti := 0\n\tif v.EventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventId: %v\", *(v.EventId))\n\t\ti++\n\t}\n\tif v.Timestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"Timestamp: %v\", *(v.Timestamp))\n\t\ti++\n\t}\n\tif v.EventType != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventType: %v\", *(v.EventType))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.TaskId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskId: %v\", *(v.TaskId))\n\t\ti++\n\t}\n\tif v.WorkflowExecutionStartedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionStartedEventAttributes: %v\", v.WorkflowExecutionStartedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionCompletedEventAttributes: %v\", v.WorkflowExecutionCompletedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionFailedEventAttributes: %v\", v.WorkflowExecutionFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionTimedOutEventAttributes: %v\", v.WorkflowExecutionTimedOutEventAttributes)\n\t\ti++\n\t}\n\tif v.DecisionTaskScheduledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskScheduledEventAttributes: %v\", v.DecisionTaskScheduledEventAttributes)\n\t\ti++\n\t}\n\tif v.DecisionTaskStartedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskStartedEventAttributes: %v\", v.DecisionTaskStartedEventAttributes)\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventAttributes: %v\", v.DecisionTaskCompletedEventAttributes)\n\t\ti++\n\t}\n\tif v.DecisionTaskTimedOutEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskTimedOutEventAttributes: %v\", v.DecisionTaskTimedOutEventAttributes)\n\t\ti++\n\t}\n\tif v.DecisionTaskFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskFailedEventAttributes: %v\", v.DecisionTaskFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskScheduledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskScheduledEventAttributes: %v\", v.ActivityTaskScheduledEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskStartedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskStartedEventAttributes: %v\", v.ActivityTaskStartedEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskCompletedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskCompletedEventAttributes: %v\", v.ActivityTaskCompletedEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskFailedEventAttributes: %v\", v.ActivityTaskFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskTimedOutEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskTimedOutEventAttributes: %v\", v.ActivityTaskTimedOutEventAttributes)\n\t\ti++\n\t}\n\tif v.TimerStartedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerStartedEventAttributes: %v\", v.TimerStartedEventAttributes)\n\t\ti++\n\t}\n\tif v.TimerFiredEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerFiredEventAttributes: %v\", v.TimerFiredEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskCancelRequestedEventAttributes: %v\", v.ActivityTaskCancelRequestedEventAttributes)\n\t\ti++\n\t}\n\tif v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestCancelActivityTaskFailedEventAttributes: %v\", v.RequestCancelActivityTaskFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ActivityTaskCanceledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskCanceledEventAttributes: %v\", v.ActivityTaskCanceledEventAttributes)\n\t\ti++\n\t}\n\tif v.TimerCanceledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerCanceledEventAttributes: %v\", v.TimerCanceledEventAttributes)\n\t\ti++\n\t}\n\tif v.CancelTimerFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelTimerFailedEventAttributes: %v\", v.CancelTimerFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.MarkerRecordedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"MarkerRecordedEventAttributes: %v\", v.MarkerRecordedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionSignaledEventAttributes: %v\", v.WorkflowExecutionSignaledEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionTerminatedEventAttributes: %v\", v.WorkflowExecutionTerminatedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionCancelRequestedEventAttributes: %v\", v.WorkflowExecutionCancelRequestedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionCanceledEventAttributes: %v\", v.WorkflowExecutionCanceledEventAttributes)\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionInitiatedEventAttributes: %v\", v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes)\n\t\ti++\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionFailedEventAttributes: %v\", v.RequestCancelExternalWorkflowExecutionFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecutionCancelRequestedEventAttributes: %v\", v.ExternalWorkflowExecutionCancelRequestedEventAttributes)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionContinuedAsNewEventAttributes: %v\", v.WorkflowExecutionContinuedAsNewEventAttributes)\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartChildWorkflowExecutionInitiatedEventAttributes: %v\", v.StartChildWorkflowExecutionInitiatedEventAttributes)\n\t\ti++\n\t}\n\tif v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartChildWorkflowExecutionFailedEventAttributes: %v\", v.StartChildWorkflowExecutionFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionStartedEventAttributes: %v\", v.ChildWorkflowExecutionStartedEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionCompletedEventAttributes: %v\", v.ChildWorkflowExecutionCompletedEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionFailedEventAttributes: %v\", v.ChildWorkflowExecutionFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionCanceledEventAttributes: %v\", v.ChildWorkflowExecutionCanceledEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionTimedOutEventAttributes: %v\", v.ChildWorkflowExecutionTimedOutEventAttributes)\n\t\ti++\n\t}\n\tif v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowExecutionTerminatedEventAttributes: %v\", v.ChildWorkflowExecutionTerminatedEventAttributes)\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalExternalWorkflowExecutionInitiatedEventAttributes: %v\", v.SignalExternalWorkflowExecutionInitiatedEventAttributes)\n\t\ti++\n\t}\n\tif v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalExternalWorkflowExecutionFailedEventAttributes: %v\", v.SignalExternalWorkflowExecutionFailedEventAttributes)\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecutionSignaledEventAttributes: %v\", v.ExternalWorkflowExecutionSignaledEventAttributes)\n\t\ti++\n\t}\n\tif v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpsertWorkflowSearchAttributesEventAttributes: %v\", v.UpsertWorkflowSearchAttributesEventAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryEvent{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _EventType_EqualsPtr(lhs, rhs *EventType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this HistoryEvent match the\n// provided HistoryEvent.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryEvent) Equals(rhs *HistoryEvent) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EventId, rhs.EventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Timestamp, rhs.Timestamp) {\n\t\treturn false\n\t}\n\tif !_EventType_EqualsPtr(v.EventType, rhs.EventType) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskId, rhs.TaskId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionStartedEventAttributes == nil && rhs.WorkflowExecutionStartedEventAttributes == nil) || (v.WorkflowExecutionStartedEventAttributes != nil && rhs.WorkflowExecutionStartedEventAttributes != nil && v.WorkflowExecutionStartedEventAttributes.Equals(rhs.WorkflowExecutionStartedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionCompletedEventAttributes == nil && rhs.WorkflowExecutionCompletedEventAttributes == nil) || (v.WorkflowExecutionCompletedEventAttributes != nil && rhs.WorkflowExecutionCompletedEventAttributes != nil && v.WorkflowExecutionCompletedEventAttributes.Equals(rhs.WorkflowExecutionCompletedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionFailedEventAttributes == nil && rhs.WorkflowExecutionFailedEventAttributes == nil) || (v.WorkflowExecutionFailedEventAttributes != nil && rhs.WorkflowExecutionFailedEventAttributes != nil && v.WorkflowExecutionFailedEventAttributes.Equals(rhs.WorkflowExecutionFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionTimedOutEventAttributes == nil && rhs.WorkflowExecutionTimedOutEventAttributes == nil) || (v.WorkflowExecutionTimedOutEventAttributes != nil && rhs.WorkflowExecutionTimedOutEventAttributes != nil && v.WorkflowExecutionTimedOutEventAttributes.Equals(rhs.WorkflowExecutionTimedOutEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskScheduledEventAttributes == nil && rhs.DecisionTaskScheduledEventAttributes == nil) || (v.DecisionTaskScheduledEventAttributes != nil && rhs.DecisionTaskScheduledEventAttributes != nil && v.DecisionTaskScheduledEventAttributes.Equals(rhs.DecisionTaskScheduledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskStartedEventAttributes == nil && rhs.DecisionTaskStartedEventAttributes == nil) || (v.DecisionTaskStartedEventAttributes != nil && rhs.DecisionTaskStartedEventAttributes != nil && v.DecisionTaskStartedEventAttributes.Equals(rhs.DecisionTaskStartedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskCompletedEventAttributes == nil && rhs.DecisionTaskCompletedEventAttributes == nil) || (v.DecisionTaskCompletedEventAttributes != nil && rhs.DecisionTaskCompletedEventAttributes != nil && v.DecisionTaskCompletedEventAttributes.Equals(rhs.DecisionTaskCompletedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskTimedOutEventAttributes == nil && rhs.DecisionTaskTimedOutEventAttributes == nil) || (v.DecisionTaskTimedOutEventAttributes != nil && rhs.DecisionTaskTimedOutEventAttributes != nil && v.DecisionTaskTimedOutEventAttributes.Equals(rhs.DecisionTaskTimedOutEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskFailedEventAttributes == nil && rhs.DecisionTaskFailedEventAttributes == nil) || (v.DecisionTaskFailedEventAttributes != nil && rhs.DecisionTaskFailedEventAttributes != nil && v.DecisionTaskFailedEventAttributes.Equals(rhs.DecisionTaskFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskScheduledEventAttributes == nil && rhs.ActivityTaskScheduledEventAttributes == nil) || (v.ActivityTaskScheduledEventAttributes != nil && rhs.ActivityTaskScheduledEventAttributes != nil && v.ActivityTaskScheduledEventAttributes.Equals(rhs.ActivityTaskScheduledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskStartedEventAttributes == nil && rhs.ActivityTaskStartedEventAttributes == nil) || (v.ActivityTaskStartedEventAttributes != nil && rhs.ActivityTaskStartedEventAttributes != nil && v.ActivityTaskStartedEventAttributes.Equals(rhs.ActivityTaskStartedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskCompletedEventAttributes == nil && rhs.ActivityTaskCompletedEventAttributes == nil) || (v.ActivityTaskCompletedEventAttributes != nil && rhs.ActivityTaskCompletedEventAttributes != nil && v.ActivityTaskCompletedEventAttributes.Equals(rhs.ActivityTaskCompletedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskFailedEventAttributes == nil && rhs.ActivityTaskFailedEventAttributes == nil) || (v.ActivityTaskFailedEventAttributes != nil && rhs.ActivityTaskFailedEventAttributes != nil && v.ActivityTaskFailedEventAttributes.Equals(rhs.ActivityTaskFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskTimedOutEventAttributes == nil && rhs.ActivityTaskTimedOutEventAttributes == nil) || (v.ActivityTaskTimedOutEventAttributes != nil && rhs.ActivityTaskTimedOutEventAttributes != nil && v.ActivityTaskTimedOutEventAttributes.Equals(rhs.ActivityTaskTimedOutEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.TimerStartedEventAttributes == nil && rhs.TimerStartedEventAttributes == nil) || (v.TimerStartedEventAttributes != nil && rhs.TimerStartedEventAttributes != nil && v.TimerStartedEventAttributes.Equals(rhs.TimerStartedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.TimerFiredEventAttributes == nil && rhs.TimerFiredEventAttributes == nil) || (v.TimerFiredEventAttributes != nil && rhs.TimerFiredEventAttributes != nil && v.TimerFiredEventAttributes.Equals(rhs.TimerFiredEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskCancelRequestedEventAttributes == nil && rhs.ActivityTaskCancelRequestedEventAttributes == nil) || (v.ActivityTaskCancelRequestedEventAttributes != nil && rhs.ActivityTaskCancelRequestedEventAttributes != nil && v.ActivityTaskCancelRequestedEventAttributes.Equals(rhs.ActivityTaskCancelRequestedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RequestCancelActivityTaskFailedEventAttributes == nil && rhs.RequestCancelActivityTaskFailedEventAttributes == nil) || (v.RequestCancelActivityTaskFailedEventAttributes != nil && rhs.RequestCancelActivityTaskFailedEventAttributes != nil && v.RequestCancelActivityTaskFailedEventAttributes.Equals(rhs.RequestCancelActivityTaskFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskCanceledEventAttributes == nil && rhs.ActivityTaskCanceledEventAttributes == nil) || (v.ActivityTaskCanceledEventAttributes != nil && rhs.ActivityTaskCanceledEventAttributes != nil && v.ActivityTaskCanceledEventAttributes.Equals(rhs.ActivityTaskCanceledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.TimerCanceledEventAttributes == nil && rhs.TimerCanceledEventAttributes == nil) || (v.TimerCanceledEventAttributes != nil && rhs.TimerCanceledEventAttributes != nil && v.TimerCanceledEventAttributes.Equals(rhs.TimerCanceledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.CancelTimerFailedEventAttributes == nil && rhs.CancelTimerFailedEventAttributes == nil) || (v.CancelTimerFailedEventAttributes != nil && rhs.CancelTimerFailedEventAttributes != nil && v.CancelTimerFailedEventAttributes.Equals(rhs.CancelTimerFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.MarkerRecordedEventAttributes == nil && rhs.MarkerRecordedEventAttributes == nil) || (v.MarkerRecordedEventAttributes != nil && rhs.MarkerRecordedEventAttributes != nil && v.MarkerRecordedEventAttributes.Equals(rhs.MarkerRecordedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionSignaledEventAttributes == nil && rhs.WorkflowExecutionSignaledEventAttributes == nil) || (v.WorkflowExecutionSignaledEventAttributes != nil && rhs.WorkflowExecutionSignaledEventAttributes != nil && v.WorkflowExecutionSignaledEventAttributes.Equals(rhs.WorkflowExecutionSignaledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionTerminatedEventAttributes == nil && rhs.WorkflowExecutionTerminatedEventAttributes == nil) || (v.WorkflowExecutionTerminatedEventAttributes != nil && rhs.WorkflowExecutionTerminatedEventAttributes != nil && v.WorkflowExecutionTerminatedEventAttributes.Equals(rhs.WorkflowExecutionTerminatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionCancelRequestedEventAttributes == nil && rhs.WorkflowExecutionCancelRequestedEventAttributes == nil) || (v.WorkflowExecutionCancelRequestedEventAttributes != nil && rhs.WorkflowExecutionCancelRequestedEventAttributes != nil && v.WorkflowExecutionCancelRequestedEventAttributes.Equals(rhs.WorkflowExecutionCancelRequestedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionCanceledEventAttributes == nil && rhs.WorkflowExecutionCanceledEventAttributes == nil) || (v.WorkflowExecutionCanceledEventAttributes != nil && rhs.WorkflowExecutionCanceledEventAttributes != nil && v.WorkflowExecutionCanceledEventAttributes.Equals(rhs.WorkflowExecutionCanceledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes == nil && rhs.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes == nil) || (v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil && rhs.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil && v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.Equals(rhs.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.RequestCancelExternalWorkflowExecutionFailedEventAttributes == nil && rhs.RequestCancelExternalWorkflowExecutionFailedEventAttributes == nil) || (v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil && rhs.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil && v.RequestCancelExternalWorkflowExecutionFailedEventAttributes.Equals(rhs.RequestCancelExternalWorkflowExecutionFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecutionCancelRequestedEventAttributes == nil && rhs.ExternalWorkflowExecutionCancelRequestedEventAttributes == nil) || (v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil && rhs.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil && v.ExternalWorkflowExecutionCancelRequestedEventAttributes.Equals(rhs.ExternalWorkflowExecutionCancelRequestedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionContinuedAsNewEventAttributes == nil && rhs.WorkflowExecutionContinuedAsNewEventAttributes == nil) || (v.WorkflowExecutionContinuedAsNewEventAttributes != nil && rhs.WorkflowExecutionContinuedAsNewEventAttributes != nil && v.WorkflowExecutionContinuedAsNewEventAttributes.Equals(rhs.WorkflowExecutionContinuedAsNewEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.StartChildWorkflowExecutionInitiatedEventAttributes == nil && rhs.StartChildWorkflowExecutionInitiatedEventAttributes == nil) || (v.StartChildWorkflowExecutionInitiatedEventAttributes != nil && rhs.StartChildWorkflowExecutionInitiatedEventAttributes != nil && v.StartChildWorkflowExecutionInitiatedEventAttributes.Equals(rhs.StartChildWorkflowExecutionInitiatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.StartChildWorkflowExecutionFailedEventAttributes == nil && rhs.StartChildWorkflowExecutionFailedEventAttributes == nil) || (v.StartChildWorkflowExecutionFailedEventAttributes != nil && rhs.StartChildWorkflowExecutionFailedEventAttributes != nil && v.StartChildWorkflowExecutionFailedEventAttributes.Equals(rhs.StartChildWorkflowExecutionFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionStartedEventAttributes == nil && rhs.ChildWorkflowExecutionStartedEventAttributes == nil) || (v.ChildWorkflowExecutionStartedEventAttributes != nil && rhs.ChildWorkflowExecutionStartedEventAttributes != nil && v.ChildWorkflowExecutionStartedEventAttributes.Equals(rhs.ChildWorkflowExecutionStartedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionCompletedEventAttributes == nil && rhs.ChildWorkflowExecutionCompletedEventAttributes == nil) || (v.ChildWorkflowExecutionCompletedEventAttributes != nil && rhs.ChildWorkflowExecutionCompletedEventAttributes != nil && v.ChildWorkflowExecutionCompletedEventAttributes.Equals(rhs.ChildWorkflowExecutionCompletedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionFailedEventAttributes == nil && rhs.ChildWorkflowExecutionFailedEventAttributes == nil) || (v.ChildWorkflowExecutionFailedEventAttributes != nil && rhs.ChildWorkflowExecutionFailedEventAttributes != nil && v.ChildWorkflowExecutionFailedEventAttributes.Equals(rhs.ChildWorkflowExecutionFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionCanceledEventAttributes == nil && rhs.ChildWorkflowExecutionCanceledEventAttributes == nil) || (v.ChildWorkflowExecutionCanceledEventAttributes != nil && rhs.ChildWorkflowExecutionCanceledEventAttributes != nil && v.ChildWorkflowExecutionCanceledEventAttributes.Equals(rhs.ChildWorkflowExecutionCanceledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionTimedOutEventAttributes == nil && rhs.ChildWorkflowExecutionTimedOutEventAttributes == nil) || (v.ChildWorkflowExecutionTimedOutEventAttributes != nil && rhs.ChildWorkflowExecutionTimedOutEventAttributes != nil && v.ChildWorkflowExecutionTimedOutEventAttributes.Equals(rhs.ChildWorkflowExecutionTimedOutEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ChildWorkflowExecutionTerminatedEventAttributes == nil && rhs.ChildWorkflowExecutionTerminatedEventAttributes == nil) || (v.ChildWorkflowExecutionTerminatedEventAttributes != nil && rhs.ChildWorkflowExecutionTerminatedEventAttributes != nil && v.ChildWorkflowExecutionTerminatedEventAttributes.Equals(rhs.ChildWorkflowExecutionTerminatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SignalExternalWorkflowExecutionInitiatedEventAttributes == nil && rhs.SignalExternalWorkflowExecutionInitiatedEventAttributes == nil) || (v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil && rhs.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil && v.SignalExternalWorkflowExecutionInitiatedEventAttributes.Equals(rhs.SignalExternalWorkflowExecutionInitiatedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.SignalExternalWorkflowExecutionFailedEventAttributes == nil && rhs.SignalExternalWorkflowExecutionFailedEventAttributes == nil) || (v.SignalExternalWorkflowExecutionFailedEventAttributes != nil && rhs.SignalExternalWorkflowExecutionFailedEventAttributes != nil && v.SignalExternalWorkflowExecutionFailedEventAttributes.Equals(rhs.SignalExternalWorkflowExecutionFailedEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecutionSignaledEventAttributes == nil && rhs.ExternalWorkflowExecutionSignaledEventAttributes == nil) || (v.ExternalWorkflowExecutionSignaledEventAttributes != nil && rhs.ExternalWorkflowExecutionSignaledEventAttributes != nil && v.ExternalWorkflowExecutionSignaledEventAttributes.Equals(rhs.ExternalWorkflowExecutionSignaledEventAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.UpsertWorkflowSearchAttributesEventAttributes == nil && rhs.UpsertWorkflowSearchAttributesEventAttributes == nil) || (v.UpsertWorkflowSearchAttributesEventAttributes != nil && rhs.UpsertWorkflowSearchAttributesEventAttributes != nil && v.UpsertWorkflowSearchAttributesEventAttributes.Equals(rhs.UpsertWorkflowSearchAttributesEventAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryEvent.\nfunc (v *HistoryEvent) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.EventId != nil {\n\t\tenc.AddInt64(\"eventId\", *v.EventId)\n\t}\n\tif v.Timestamp != nil {\n\t\tenc.AddInt64(\"timestamp\", *v.Timestamp)\n\t}\n\tif v.EventType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"eventType\", *v.EventType))\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.TaskId != nil {\n\t\tenc.AddInt64(\"taskId\", *v.TaskId)\n\t}\n\tif v.WorkflowExecutionStartedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionStartedEventAttributes\", v.WorkflowExecutionStartedEventAttributes))\n\t}\n\tif v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionCompletedEventAttributes\", v.WorkflowExecutionCompletedEventAttributes))\n\t}\n\tif v.WorkflowExecutionFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionFailedEventAttributes\", v.WorkflowExecutionFailedEventAttributes))\n\t}\n\tif v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionTimedOutEventAttributes\", v.WorkflowExecutionTimedOutEventAttributes))\n\t}\n\tif v.DecisionTaskScheduledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskScheduledEventAttributes\", v.DecisionTaskScheduledEventAttributes))\n\t}\n\tif v.DecisionTaskStartedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskStartedEventAttributes\", v.DecisionTaskStartedEventAttributes))\n\t}\n\tif v.DecisionTaskCompletedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskCompletedEventAttributes\", v.DecisionTaskCompletedEventAttributes))\n\t}\n\tif v.DecisionTaskTimedOutEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskTimedOutEventAttributes\", v.DecisionTaskTimedOutEventAttributes))\n\t}\n\tif v.DecisionTaskFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTaskFailedEventAttributes\", v.DecisionTaskFailedEventAttributes))\n\t}\n\tif v.ActivityTaskScheduledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskScheduledEventAttributes\", v.ActivityTaskScheduledEventAttributes))\n\t}\n\tif v.ActivityTaskStartedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskStartedEventAttributes\", v.ActivityTaskStartedEventAttributes))\n\t}\n\tif v.ActivityTaskCompletedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskCompletedEventAttributes\", v.ActivityTaskCompletedEventAttributes))\n\t}\n\tif v.ActivityTaskFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskFailedEventAttributes\", v.ActivityTaskFailedEventAttributes))\n\t}\n\tif v.ActivityTaskTimedOutEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskTimedOutEventAttributes\", v.ActivityTaskTimedOutEventAttributes))\n\t}\n\tif v.TimerStartedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timerStartedEventAttributes\", v.TimerStartedEventAttributes))\n\t}\n\tif v.TimerFiredEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timerFiredEventAttributes\", v.TimerFiredEventAttributes))\n\t}\n\tif v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskCancelRequestedEventAttributes\", v.ActivityTaskCancelRequestedEventAttributes))\n\t}\n\tif v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"requestCancelActivityTaskFailedEventAttributes\", v.RequestCancelActivityTaskFailedEventAttributes))\n\t}\n\tif v.ActivityTaskCanceledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityTaskCanceledEventAttributes\", v.ActivityTaskCanceledEventAttributes))\n\t}\n\tif v.TimerCanceledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timerCanceledEventAttributes\", v.TimerCanceledEventAttributes))\n\t}\n\tif v.CancelTimerFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cancelTimerFailedEventAttributes\", v.CancelTimerFailedEventAttributes))\n\t}\n\tif v.MarkerRecordedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"markerRecordedEventAttributes\", v.MarkerRecordedEventAttributes))\n\t}\n\tif v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionSignaledEventAttributes\", v.WorkflowExecutionSignaledEventAttributes))\n\t}\n\tif v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionTerminatedEventAttributes\", v.WorkflowExecutionTerminatedEventAttributes))\n\t}\n\tif v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionCancelRequestedEventAttributes\", v.WorkflowExecutionCancelRequestedEventAttributes))\n\t}\n\tif v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionCanceledEventAttributes\", v.WorkflowExecutionCanceledEventAttributes))\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"requestCancelExternalWorkflowExecutionInitiatedEventAttributes\", v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes))\n\t}\n\tif v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"requestCancelExternalWorkflowExecutionFailedEventAttributes\", v.RequestCancelExternalWorkflowExecutionFailedEventAttributes))\n\t}\n\tif v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecutionCancelRequestedEventAttributes\", v.ExternalWorkflowExecutionCancelRequestedEventAttributes))\n\t}\n\tif v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecutionContinuedAsNewEventAttributes\", v.WorkflowExecutionContinuedAsNewEventAttributes))\n\t}\n\tif v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startChildWorkflowExecutionInitiatedEventAttributes\", v.StartChildWorkflowExecutionInitiatedEventAttributes))\n\t}\n\tif v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startChildWorkflowExecutionFailedEventAttributes\", v.StartChildWorkflowExecutionFailedEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionStartedEventAttributes\", v.ChildWorkflowExecutionStartedEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionCompletedEventAttributes\", v.ChildWorkflowExecutionCompletedEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionFailedEventAttributes\", v.ChildWorkflowExecutionFailedEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionCanceledEventAttributes\", v.ChildWorkflowExecutionCanceledEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionTimedOutEventAttributes\", v.ChildWorkflowExecutionTimedOutEventAttributes))\n\t}\n\tif v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"childWorkflowExecutionTerminatedEventAttributes\", v.ChildWorkflowExecutionTerminatedEventAttributes))\n\t}\n\tif v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalExternalWorkflowExecutionInitiatedEventAttributes\", v.SignalExternalWorkflowExecutionInitiatedEventAttributes))\n\t}\n\tif v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"signalExternalWorkflowExecutionFailedEventAttributes\", v.SignalExternalWorkflowExecutionFailedEventAttributes))\n\t}\n\tif v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecutionSignaledEventAttributes\", v.ExternalWorkflowExecutionSignaledEventAttributes))\n\t}\n\tif v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"upsertWorkflowSearchAttributesEventAttributes\", v.UpsertWorkflowSearchAttributesEventAttributes))\n\t}\n\treturn err\n}\n\n// GetEventId returns the value of EventId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetEventId() (o int64) {\n\tif v != nil && v.EventId != nil {\n\t\treturn *v.EventId\n\t}\n\n\treturn\n}\n\n// IsSetEventId returns true if EventId is not nil.\nfunc (v *HistoryEvent) IsSetEventId() bool {\n\treturn v != nil && v.EventId != nil\n}\n\n// GetTimestamp returns the value of Timestamp if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\n\treturn\n}\n\n// IsSetTimestamp returns true if Timestamp is not nil.\nfunc (v *HistoryEvent) IsSetTimestamp() bool {\n\treturn v != nil && v.Timestamp != nil\n}\n\n// GetEventType returns the value of EventType if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetEventType() (o EventType) {\n\tif v != nil && v.EventType != nil {\n\t\treturn *v.EventType\n\t}\n\n\treturn\n}\n\n// IsSetEventType returns true if EventType is not nil.\nfunc (v *HistoryEvent) IsSetEventType() bool {\n\treturn v != nil && v.EventType != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *HistoryEvent) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetTaskId returns the value of TaskId if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetTaskId() (o int64) {\n\tif v != nil && v.TaskId != nil {\n\t\treturn *v.TaskId\n\t}\n\n\treturn\n}\n\n// IsSetTaskId returns true if TaskId is not nil.\nfunc (v *HistoryEvent) IsSetTaskId() bool {\n\treturn v != nil && v.TaskId != nil\n}\n\n// GetWorkflowExecutionStartedEventAttributes returns the value of WorkflowExecutionStartedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionStartedEventAttributes() (o *WorkflowExecutionStartedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionStartedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionStartedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionStartedEventAttributes returns true if WorkflowExecutionStartedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionStartedEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionStartedEventAttributes != nil\n}\n\n// GetWorkflowExecutionCompletedEventAttributes returns the value of WorkflowExecutionCompletedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionCompletedEventAttributes() (o *WorkflowExecutionCompletedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCompletedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionCompletedEventAttributes returns true if WorkflowExecutionCompletedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionCompletedEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionCompletedEventAttributes != nil\n}\n\n// GetWorkflowExecutionFailedEventAttributes returns the value of WorkflowExecutionFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionFailedEventAttributes() (o *WorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionFailedEventAttributes returns true if WorkflowExecutionFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionFailedEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionFailedEventAttributes != nil\n}\n\n// GetWorkflowExecutionTimedOutEventAttributes returns the value of WorkflowExecutionTimedOutEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionTimedOutEventAttributes() (o *WorkflowExecutionTimedOutEventAttributes) {\n\tif v != nil && v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\treturn v.WorkflowExecutionTimedOutEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionTimedOutEventAttributes returns true if WorkflowExecutionTimedOutEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionTimedOutEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionTimedOutEventAttributes != nil\n}\n\n// GetDecisionTaskScheduledEventAttributes returns the value of DecisionTaskScheduledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetDecisionTaskScheduledEventAttributes() (o *DecisionTaskScheduledEventAttributes) {\n\tif v != nil && v.DecisionTaskScheduledEventAttributes != nil {\n\t\treturn v.DecisionTaskScheduledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskScheduledEventAttributes returns true if DecisionTaskScheduledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetDecisionTaskScheduledEventAttributes() bool {\n\treturn v != nil && v.DecisionTaskScheduledEventAttributes != nil\n}\n\n// GetDecisionTaskStartedEventAttributes returns the value of DecisionTaskStartedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetDecisionTaskStartedEventAttributes() (o *DecisionTaskStartedEventAttributes) {\n\tif v != nil && v.DecisionTaskStartedEventAttributes != nil {\n\t\treturn v.DecisionTaskStartedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskStartedEventAttributes returns true if DecisionTaskStartedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetDecisionTaskStartedEventAttributes() bool {\n\treturn v != nil && v.DecisionTaskStartedEventAttributes != nil\n}\n\n// GetDecisionTaskCompletedEventAttributes returns the value of DecisionTaskCompletedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetDecisionTaskCompletedEventAttributes() (o *DecisionTaskCompletedEventAttributes) {\n\tif v != nil && v.DecisionTaskCompletedEventAttributes != nil {\n\t\treturn v.DecisionTaskCompletedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventAttributes returns true if DecisionTaskCompletedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetDecisionTaskCompletedEventAttributes() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventAttributes != nil\n}\n\n// GetDecisionTaskTimedOutEventAttributes returns the value of DecisionTaskTimedOutEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetDecisionTaskTimedOutEventAttributes() (o *DecisionTaskTimedOutEventAttributes) {\n\tif v != nil && v.DecisionTaskTimedOutEventAttributes != nil {\n\t\treturn v.DecisionTaskTimedOutEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskTimedOutEventAttributes returns true if DecisionTaskTimedOutEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetDecisionTaskTimedOutEventAttributes() bool {\n\treturn v != nil && v.DecisionTaskTimedOutEventAttributes != nil\n}\n\n// GetDecisionTaskFailedEventAttributes returns the value of DecisionTaskFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetDecisionTaskFailedEventAttributes() (o *DecisionTaskFailedEventAttributes) {\n\tif v != nil && v.DecisionTaskFailedEventAttributes != nil {\n\t\treturn v.DecisionTaskFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskFailedEventAttributes returns true if DecisionTaskFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetDecisionTaskFailedEventAttributes() bool {\n\treturn v != nil && v.DecisionTaskFailedEventAttributes != nil\n}\n\n// GetActivityTaskScheduledEventAttributes returns the value of ActivityTaskScheduledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskScheduledEventAttributes() (o *ActivityTaskScheduledEventAttributes) {\n\tif v != nil && v.ActivityTaskScheduledEventAttributes != nil {\n\t\treturn v.ActivityTaskScheduledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskScheduledEventAttributes returns true if ActivityTaskScheduledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskScheduledEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskScheduledEventAttributes != nil\n}\n\n// GetActivityTaskStartedEventAttributes returns the value of ActivityTaskStartedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskStartedEventAttributes() (o *ActivityTaskStartedEventAttributes) {\n\tif v != nil && v.ActivityTaskStartedEventAttributes != nil {\n\t\treturn v.ActivityTaskStartedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskStartedEventAttributes returns true if ActivityTaskStartedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskStartedEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskStartedEventAttributes != nil\n}\n\n// GetActivityTaskCompletedEventAttributes returns the value of ActivityTaskCompletedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskCompletedEventAttributes() (o *ActivityTaskCompletedEventAttributes) {\n\tif v != nil && v.ActivityTaskCompletedEventAttributes != nil {\n\t\treturn v.ActivityTaskCompletedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskCompletedEventAttributes returns true if ActivityTaskCompletedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskCompletedEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskCompletedEventAttributes != nil\n}\n\n// GetActivityTaskFailedEventAttributes returns the value of ActivityTaskFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskFailedEventAttributes() (o *ActivityTaskFailedEventAttributes) {\n\tif v != nil && v.ActivityTaskFailedEventAttributes != nil {\n\t\treturn v.ActivityTaskFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskFailedEventAttributes returns true if ActivityTaskFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskFailedEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskFailedEventAttributes != nil\n}\n\n// GetActivityTaskTimedOutEventAttributes returns the value of ActivityTaskTimedOutEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskTimedOutEventAttributes() (o *ActivityTaskTimedOutEventAttributes) {\n\tif v != nil && v.ActivityTaskTimedOutEventAttributes != nil {\n\t\treturn v.ActivityTaskTimedOutEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskTimedOutEventAttributes returns true if ActivityTaskTimedOutEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskTimedOutEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskTimedOutEventAttributes != nil\n}\n\n// GetTimerStartedEventAttributes returns the value of TimerStartedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetTimerStartedEventAttributes() (o *TimerStartedEventAttributes) {\n\tif v != nil && v.TimerStartedEventAttributes != nil {\n\t\treturn v.TimerStartedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetTimerStartedEventAttributes returns true if TimerStartedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetTimerStartedEventAttributes() bool {\n\treturn v != nil && v.TimerStartedEventAttributes != nil\n}\n\n// GetTimerFiredEventAttributes returns the value of TimerFiredEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetTimerFiredEventAttributes() (o *TimerFiredEventAttributes) {\n\tif v != nil && v.TimerFiredEventAttributes != nil {\n\t\treturn v.TimerFiredEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetTimerFiredEventAttributes returns true if TimerFiredEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetTimerFiredEventAttributes() bool {\n\treturn v != nil && v.TimerFiredEventAttributes != nil\n}\n\n// GetActivityTaskCancelRequestedEventAttributes returns the value of ActivityTaskCancelRequestedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskCancelRequestedEventAttributes() (o *ActivityTaskCancelRequestedEventAttributes) {\n\tif v != nil && v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\treturn v.ActivityTaskCancelRequestedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskCancelRequestedEventAttributes returns true if ActivityTaskCancelRequestedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskCancelRequestedEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskCancelRequestedEventAttributes != nil\n}\n\n// GetRequestCancelActivityTaskFailedEventAttributes returns the value of RequestCancelActivityTaskFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetRequestCancelActivityTaskFailedEventAttributes() (o *RequestCancelActivityTaskFailedEventAttributes) {\n\tif v != nil && v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\treturn v.RequestCancelActivityTaskFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRequestCancelActivityTaskFailedEventAttributes returns true if RequestCancelActivityTaskFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetRequestCancelActivityTaskFailedEventAttributes() bool {\n\treturn v != nil && v.RequestCancelActivityTaskFailedEventAttributes != nil\n}\n\n// GetActivityTaskCanceledEventAttributes returns the value of ActivityTaskCanceledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetActivityTaskCanceledEventAttributes() (o *ActivityTaskCanceledEventAttributes) {\n\tif v != nil && v.ActivityTaskCanceledEventAttributes != nil {\n\t\treturn v.ActivityTaskCanceledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskCanceledEventAttributes returns true if ActivityTaskCanceledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetActivityTaskCanceledEventAttributes() bool {\n\treturn v != nil && v.ActivityTaskCanceledEventAttributes != nil\n}\n\n// GetTimerCanceledEventAttributes returns the value of TimerCanceledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetTimerCanceledEventAttributes() (o *TimerCanceledEventAttributes) {\n\tif v != nil && v.TimerCanceledEventAttributes != nil {\n\t\treturn v.TimerCanceledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetTimerCanceledEventAttributes returns true if TimerCanceledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetTimerCanceledEventAttributes() bool {\n\treturn v != nil && v.TimerCanceledEventAttributes != nil\n}\n\n// GetCancelTimerFailedEventAttributes returns the value of CancelTimerFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetCancelTimerFailedEventAttributes() (o *CancelTimerFailedEventAttributes) {\n\tif v != nil && v.CancelTimerFailedEventAttributes != nil {\n\t\treturn v.CancelTimerFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetCancelTimerFailedEventAttributes returns true if CancelTimerFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetCancelTimerFailedEventAttributes() bool {\n\treturn v != nil && v.CancelTimerFailedEventAttributes != nil\n}\n\n// GetMarkerRecordedEventAttributes returns the value of MarkerRecordedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetMarkerRecordedEventAttributes() (o *MarkerRecordedEventAttributes) {\n\tif v != nil && v.MarkerRecordedEventAttributes != nil {\n\t\treturn v.MarkerRecordedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetMarkerRecordedEventAttributes returns true if MarkerRecordedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetMarkerRecordedEventAttributes() bool {\n\treturn v != nil && v.MarkerRecordedEventAttributes != nil\n}\n\n// GetWorkflowExecutionSignaledEventAttributes returns the value of WorkflowExecutionSignaledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionSignaledEventAttributes() (o *WorkflowExecutionSignaledEventAttributes) {\n\tif v != nil && v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\treturn v.WorkflowExecutionSignaledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionSignaledEventAttributes returns true if WorkflowExecutionSignaledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionSignaledEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionSignaledEventAttributes != nil\n}\n\n// GetWorkflowExecutionTerminatedEventAttributes returns the value of WorkflowExecutionTerminatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionTerminatedEventAttributes() (o *WorkflowExecutionTerminatedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionTerminatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionTerminatedEventAttributes returns true if WorkflowExecutionTerminatedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionTerminatedEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionTerminatedEventAttributes != nil\n}\n\n// GetWorkflowExecutionCancelRequestedEventAttributes returns the value of WorkflowExecutionCancelRequestedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionCancelRequestedEventAttributes() (o *WorkflowExecutionCancelRequestedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCancelRequestedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionCancelRequestedEventAttributes returns true if WorkflowExecutionCancelRequestedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionCancelRequestedEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionCancelRequestedEventAttributes != nil\n}\n\n// GetWorkflowExecutionCanceledEventAttributes returns the value of WorkflowExecutionCanceledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionCanceledEventAttributes() (o *WorkflowExecutionCanceledEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCanceledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionCanceledEventAttributes returns true if WorkflowExecutionCanceledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionCanceledEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionCanceledEventAttributes != nil\n}\n\n// GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes returns the value of RequestCancelExternalWorkflowExecutionInitiatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes() (o *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes returns true if RequestCancelExternalWorkflowExecutionInitiatedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes() bool {\n\treturn v != nil && v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil\n}\n\n// GetRequestCancelExternalWorkflowExecutionFailedEventAttributes returns the value of RequestCancelExternalWorkflowExecutionFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetRequestCancelExternalWorkflowExecutionFailedEventAttributes() (o *RequestCancelExternalWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetRequestCancelExternalWorkflowExecutionFailedEventAttributes returns true if RequestCancelExternalWorkflowExecutionFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetRequestCancelExternalWorkflowExecutionFailedEventAttributes() bool {\n\treturn v != nil && v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil\n}\n\n// GetExternalWorkflowExecutionCancelRequestedEventAttributes returns the value of ExternalWorkflowExecutionCancelRequestedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetExternalWorkflowExecutionCancelRequestedEventAttributes() (o *ExternalWorkflowExecutionCancelRequestedEventAttributes) {\n\tif v != nil && v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\treturn v.ExternalWorkflowExecutionCancelRequestedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecutionCancelRequestedEventAttributes returns true if ExternalWorkflowExecutionCancelRequestedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetExternalWorkflowExecutionCancelRequestedEventAttributes() bool {\n\treturn v != nil && v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil\n}\n\n// GetWorkflowExecutionContinuedAsNewEventAttributes returns the value of WorkflowExecutionContinuedAsNewEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetWorkflowExecutionContinuedAsNewEventAttributes() (o *WorkflowExecutionContinuedAsNewEventAttributes) {\n\tif v != nil && v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\treturn v.WorkflowExecutionContinuedAsNewEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionContinuedAsNewEventAttributes returns true if WorkflowExecutionContinuedAsNewEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetWorkflowExecutionContinuedAsNewEventAttributes() bool {\n\treturn v != nil && v.WorkflowExecutionContinuedAsNewEventAttributes != nil\n}\n\n// GetStartChildWorkflowExecutionInitiatedEventAttributes returns the value of StartChildWorkflowExecutionInitiatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetStartChildWorkflowExecutionInitiatedEventAttributes() (o *StartChildWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.StartChildWorkflowExecutionInitiatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartChildWorkflowExecutionInitiatedEventAttributes returns true if StartChildWorkflowExecutionInitiatedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetStartChildWorkflowExecutionInitiatedEventAttributes() bool {\n\treturn v != nil && v.StartChildWorkflowExecutionInitiatedEventAttributes != nil\n}\n\n// GetStartChildWorkflowExecutionFailedEventAttributes returns the value of StartChildWorkflowExecutionFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetStartChildWorkflowExecutionFailedEventAttributes() (o *StartChildWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.StartChildWorkflowExecutionFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStartChildWorkflowExecutionFailedEventAttributes returns true if StartChildWorkflowExecutionFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetStartChildWorkflowExecutionFailedEventAttributes() bool {\n\treturn v != nil && v.StartChildWorkflowExecutionFailedEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionStartedEventAttributes returns the value of ChildWorkflowExecutionStartedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionStartedEventAttributes() (o *ChildWorkflowExecutionStartedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionStartedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionStartedEventAttributes returns true if ChildWorkflowExecutionStartedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionStartedEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionStartedEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionCompletedEventAttributes returns the value of ChildWorkflowExecutionCompletedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionCompletedEventAttributes() (o *ChildWorkflowExecutionCompletedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionCompletedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionCompletedEventAttributes returns true if ChildWorkflowExecutionCompletedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionCompletedEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionCompletedEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionFailedEventAttributes returns the value of ChildWorkflowExecutionFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionFailedEventAttributes() (o *ChildWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionFailedEventAttributes returns true if ChildWorkflowExecutionFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionFailedEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionFailedEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionCanceledEventAttributes returns the value of ChildWorkflowExecutionCanceledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionCanceledEventAttributes() (o *ChildWorkflowExecutionCanceledEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionCanceledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionCanceledEventAttributes returns true if ChildWorkflowExecutionCanceledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionCanceledEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionCanceledEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionTimedOutEventAttributes returns the value of ChildWorkflowExecutionTimedOutEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionTimedOutEventAttributes() (o *ChildWorkflowExecutionTimedOutEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionTimedOutEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionTimedOutEventAttributes returns true if ChildWorkflowExecutionTimedOutEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionTimedOutEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionTimedOutEventAttributes != nil\n}\n\n// GetChildWorkflowExecutionTerminatedEventAttributes returns the value of ChildWorkflowExecutionTerminatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetChildWorkflowExecutionTerminatedEventAttributes() (o *ChildWorkflowExecutionTerminatedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionTerminatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowExecutionTerminatedEventAttributes returns true if ChildWorkflowExecutionTerminatedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetChildWorkflowExecutionTerminatedEventAttributes() bool {\n\treturn v != nil && v.ChildWorkflowExecutionTerminatedEventAttributes != nil\n}\n\n// GetSignalExternalWorkflowExecutionInitiatedEventAttributes returns the value of SignalExternalWorkflowExecutionInitiatedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetSignalExternalWorkflowExecutionInitiatedEventAttributes() (o *SignalExternalWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.SignalExternalWorkflowExecutionInitiatedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSignalExternalWorkflowExecutionInitiatedEventAttributes returns true if SignalExternalWorkflowExecutionInitiatedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetSignalExternalWorkflowExecutionInitiatedEventAttributes() bool {\n\treturn v != nil && v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil\n}\n\n// GetSignalExternalWorkflowExecutionFailedEventAttributes returns the value of SignalExternalWorkflowExecutionFailedEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetSignalExternalWorkflowExecutionFailedEventAttributes() (o *SignalExternalWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.SignalExternalWorkflowExecutionFailedEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSignalExternalWorkflowExecutionFailedEventAttributes returns true if SignalExternalWorkflowExecutionFailedEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetSignalExternalWorkflowExecutionFailedEventAttributes() bool {\n\treturn v != nil && v.SignalExternalWorkflowExecutionFailedEventAttributes != nil\n}\n\n// GetExternalWorkflowExecutionSignaledEventAttributes returns the value of ExternalWorkflowExecutionSignaledEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetExternalWorkflowExecutionSignaledEventAttributes() (o *ExternalWorkflowExecutionSignaledEventAttributes) {\n\tif v != nil && v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\treturn v.ExternalWorkflowExecutionSignaledEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecutionSignaledEventAttributes returns true if ExternalWorkflowExecutionSignaledEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetExternalWorkflowExecutionSignaledEventAttributes() bool {\n\treturn v != nil && v.ExternalWorkflowExecutionSignaledEventAttributes != nil\n}\n\n// GetUpsertWorkflowSearchAttributesEventAttributes returns the value of UpsertWorkflowSearchAttributesEventAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryEvent) GetUpsertWorkflowSearchAttributesEventAttributes() (o *UpsertWorkflowSearchAttributesEventAttributes) {\n\tif v != nil && v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\treturn v.UpsertWorkflowSearchAttributesEventAttributes\n\t}\n\n\treturn\n}\n\n// IsSetUpsertWorkflowSearchAttributesEventAttributes returns true if UpsertWorkflowSearchAttributesEventAttributes is not nil.\nfunc (v *HistoryEvent) IsSetUpsertWorkflowSearchAttributesEventAttributes() bool {\n\treturn v != nil && v.UpsertWorkflowSearchAttributesEventAttributes != nil\n}\n\ntype HistoryEventFilterType int32\n\nconst (\n\tHistoryEventFilterTypeAllEvent   HistoryEventFilterType = 0\n\tHistoryEventFilterTypeCloseEvent HistoryEventFilterType = 1\n)\n\n// HistoryEventFilterType_Values returns all recognized values of HistoryEventFilterType.\nfunc HistoryEventFilterType_Values() []HistoryEventFilterType {\n\treturn []HistoryEventFilterType{\n\t\tHistoryEventFilterTypeAllEvent,\n\t\tHistoryEventFilterTypeCloseEvent,\n\t}\n}\n\n// UnmarshalText tries to decode HistoryEventFilterType from a byte slice\n// containing its name.\n//\n//\tvar v HistoryEventFilterType\n//\terr := v.UnmarshalText([]byte(\"ALL_EVENT\"))\nfunc (v *HistoryEventFilterType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"ALL_EVENT\":\n\t\t*v = HistoryEventFilterTypeAllEvent\n\t\treturn nil\n\tcase \"CLOSE_EVENT\":\n\t\t*v = HistoryEventFilterTypeCloseEvent\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"HistoryEventFilterType\", err)\n\t\t}\n\t\t*v = HistoryEventFilterType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes HistoryEventFilterType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v HistoryEventFilterType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"ALL_EVENT\"), nil\n\tcase 1:\n\t\treturn []byte(\"CLOSE_EVENT\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryEventFilterType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v HistoryEventFilterType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"ALL_EVENT\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"CLOSE_EVENT\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v HistoryEventFilterType) Ptr() *HistoryEventFilterType {\n\treturn &v\n}\n\n// Encode encodes HistoryEventFilterType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v HistoryEventFilterType\n//\treturn v.Encode(sWriter)\nfunc (v HistoryEventFilterType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates HistoryEventFilterType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v HistoryEventFilterType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes HistoryEventFilterType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return HistoryEventFilterType(0), err\n//\t}\n//\n//\tvar v HistoryEventFilterType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return HistoryEventFilterType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *HistoryEventFilterType) FromWire(w wire.Value) error {\n\t*v = (HistoryEventFilterType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded HistoryEventFilterType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v HistoryEventFilterType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return HistoryEventFilterType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *HistoryEventFilterType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (HistoryEventFilterType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of HistoryEventFilterType.\nfunc (v HistoryEventFilterType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ALL_EVENT\"\n\tcase 1:\n\t\treturn \"CLOSE_EVENT\"\n\t}\n\treturn fmt.Sprintf(\"HistoryEventFilterType(%d)\", w)\n}\n\n// Equals returns true if this HistoryEventFilterType value matches the provided\n// value.\nfunc (v HistoryEventFilterType) Equals(rhs HistoryEventFilterType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes HistoryEventFilterType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v HistoryEventFilterType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"ALL_EVENT\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"CLOSE_EVENT\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode HistoryEventFilterType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *HistoryEventFilterType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"HistoryEventFilterType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"HistoryEventFilterType\")\n\t\t}\n\t\t*v = (HistoryEventFilterType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"HistoryEventFilterType\")\n\t}\n}\n\ntype IndexedValueType int32\n\nconst (\n\tIndexedValueTypeString   IndexedValueType = 0\n\tIndexedValueTypeKeyword  IndexedValueType = 1\n\tIndexedValueTypeInt      IndexedValueType = 2\n\tIndexedValueTypeDouble   IndexedValueType = 3\n\tIndexedValueTypeBool     IndexedValueType = 4\n\tIndexedValueTypeDatetime IndexedValueType = 5\n)\n\n// IndexedValueType_Values returns all recognized values of IndexedValueType.\nfunc IndexedValueType_Values() []IndexedValueType {\n\treturn []IndexedValueType{\n\t\tIndexedValueTypeString,\n\t\tIndexedValueTypeKeyword,\n\t\tIndexedValueTypeInt,\n\t\tIndexedValueTypeDouble,\n\t\tIndexedValueTypeBool,\n\t\tIndexedValueTypeDatetime,\n\t}\n}\n\n// UnmarshalText tries to decode IndexedValueType from a byte slice\n// containing its name.\n//\n//\tvar v IndexedValueType\n//\terr := v.UnmarshalText([]byte(\"STRING\"))\nfunc (v *IndexedValueType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"STRING\":\n\t\t*v = IndexedValueTypeString\n\t\treturn nil\n\tcase \"KEYWORD\":\n\t\t*v = IndexedValueTypeKeyword\n\t\treturn nil\n\tcase \"INT\":\n\t\t*v = IndexedValueTypeInt\n\t\treturn nil\n\tcase \"DOUBLE\":\n\t\t*v = IndexedValueTypeDouble\n\t\treturn nil\n\tcase \"BOOL\":\n\t\t*v = IndexedValueTypeBool\n\t\treturn nil\n\tcase \"DATETIME\":\n\t\t*v = IndexedValueTypeDatetime\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"IndexedValueType\", err)\n\t\t}\n\t\t*v = IndexedValueType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes IndexedValueType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v IndexedValueType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"STRING\"), nil\n\tcase 1:\n\t\treturn []byte(\"KEYWORD\"), nil\n\tcase 2:\n\t\treturn []byte(\"INT\"), nil\n\tcase 3:\n\t\treturn []byte(\"DOUBLE\"), nil\n\tcase 4:\n\t\treturn []byte(\"BOOL\"), nil\n\tcase 5:\n\t\treturn []byte(\"DATETIME\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of IndexedValueType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v IndexedValueType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"STRING\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"KEYWORD\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"INT\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"DOUBLE\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"BOOL\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"DATETIME\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v IndexedValueType) Ptr() *IndexedValueType {\n\treturn &v\n}\n\n// Encode encodes IndexedValueType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v IndexedValueType\n//\treturn v.Encode(sWriter)\nfunc (v IndexedValueType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates IndexedValueType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v IndexedValueType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes IndexedValueType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return IndexedValueType(0), err\n//\t}\n//\n//\tvar v IndexedValueType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return IndexedValueType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *IndexedValueType) FromWire(w wire.Value) error {\n\t*v = (IndexedValueType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded IndexedValueType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v IndexedValueType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return IndexedValueType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *IndexedValueType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (IndexedValueType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of IndexedValueType.\nfunc (v IndexedValueType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"STRING\"\n\tcase 1:\n\t\treturn \"KEYWORD\"\n\tcase 2:\n\t\treturn \"INT\"\n\tcase 3:\n\t\treturn \"DOUBLE\"\n\tcase 4:\n\t\treturn \"BOOL\"\n\tcase 5:\n\t\treturn \"DATETIME\"\n\t}\n\treturn fmt.Sprintf(\"IndexedValueType(%d)\", w)\n}\n\n// Equals returns true if this IndexedValueType value matches the provided\n// value.\nfunc (v IndexedValueType) Equals(rhs IndexedValueType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes IndexedValueType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v IndexedValueType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"STRING\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"KEYWORD\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"INT\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"DOUBLE\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"BOOL\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"DATETIME\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode IndexedValueType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *IndexedValueType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"IndexedValueType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"IndexedValueType\")\n\t\t}\n\t\t*v = (IndexedValueType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"IndexedValueType\")\n\t}\n}\n\ntype InternalDataInconsistencyError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a InternalDataInconsistencyError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *InternalDataInconsistencyError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a InternalDataInconsistencyError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a InternalDataInconsistencyError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v InternalDataInconsistencyError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *InternalDataInconsistencyError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of InternalDataInconsistencyError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a InternalDataInconsistencyError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a InternalDataInconsistencyError struct could not be encoded.\nfunc (v *InternalDataInconsistencyError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a InternalDataInconsistencyError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a InternalDataInconsistencyError struct could not be generated from the wire\n// representation.\nfunc (v *InternalDataInconsistencyError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of InternalDataInconsistencyError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a InternalDataInconsistencyError\n// struct.\nfunc (v *InternalDataInconsistencyError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"InternalDataInconsistencyError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*InternalDataInconsistencyError) ErrorName() string {\n\treturn \"InternalDataInconsistencyError\"\n}\n\n// Equals returns true if all the fields of this InternalDataInconsistencyError match the\n// provided InternalDataInconsistencyError.\n//\n// This function performs a deep comparison.\nfunc (v *InternalDataInconsistencyError) Equals(rhs *InternalDataInconsistencyError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of InternalDataInconsistencyError.\nfunc (v *InternalDataInconsistencyError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *InternalDataInconsistencyError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *InternalDataInconsistencyError) Error() string {\n\treturn v.String()\n}\n\ntype InternalServiceError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a InternalServiceError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *InternalServiceError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a InternalServiceError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a InternalServiceError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v InternalServiceError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *InternalServiceError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of InternalServiceError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a InternalServiceError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a InternalServiceError struct could not be encoded.\nfunc (v *InternalServiceError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a InternalServiceError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a InternalServiceError struct could not be generated from the wire\n// representation.\nfunc (v *InternalServiceError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of InternalServiceError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a InternalServiceError\n// struct.\nfunc (v *InternalServiceError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"InternalServiceError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*InternalServiceError) ErrorName() string {\n\treturn \"InternalServiceError\"\n}\n\n// Equals returns true if all the fields of this InternalServiceError match the\n// provided InternalServiceError.\n//\n// This function performs a deep comparison.\nfunc (v *InternalServiceError) Equals(rhs *InternalServiceError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of InternalServiceError.\nfunc (v *InternalServiceError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *InternalServiceError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *InternalServiceError) Error() string {\n\treturn v.String()\n}\n\ntype IsolationGroupConfiguration struct {\n\tIsolationGroups []*IsolationGroupPartition `json:\"isolationGroups,omitempty\"`\n}\n\ntype _List_IsolationGroupPartition_ValueList []*IsolationGroupPartition\n\nfunc (v _List_IsolationGroupPartition_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*IsolationGroupPartition', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_IsolationGroupPartition_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_IsolationGroupPartition_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_IsolationGroupPartition_ValueList) Close() {}\n\n// ToWire translates a IsolationGroupConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *IsolationGroupConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IsolationGroups != nil {\n\t\tw, err = wire.NewValueList(_List_IsolationGroupPartition_ValueList(v.IsolationGroups)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IsolationGroupPartition_Read(w wire.Value) (*IsolationGroupPartition, error) {\n\tvar v IsolationGroupPartition\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_IsolationGroupPartition_Read(l wire.ValueList) ([]*IsolationGroupPartition, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*IsolationGroupPartition, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _IsolationGroupPartition_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a IsolationGroupConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a IsolationGroupConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v IsolationGroupConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *IsolationGroupConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.IsolationGroups, err = _List_IsolationGroupPartition_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_IsolationGroupPartition_Encode(val []*IsolationGroupPartition, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*IsolationGroupPartition', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a IsolationGroupConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a IsolationGroupConfiguration struct could not be encoded.\nfunc (v *IsolationGroupConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_IsolationGroupPartition_Encode(v.IsolationGroups, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IsolationGroupPartition_Decode(sr stream.Reader) (*IsolationGroupPartition, error) {\n\tvar v IsolationGroupPartition\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_IsolationGroupPartition_Decode(sr stream.Reader) ([]*IsolationGroupPartition, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*IsolationGroupPartition, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _IsolationGroupPartition_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a IsolationGroupConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a IsolationGroupConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *IsolationGroupConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.IsolationGroups, err = _List_IsolationGroupPartition_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a IsolationGroupConfiguration\n// struct.\nfunc (v *IsolationGroupConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"IsolationGroupConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_IsolationGroupPartition_Equals(lhs, rhs []*IsolationGroupPartition) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this IsolationGroupConfiguration match the\n// provided IsolationGroupConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *IsolationGroupConfiguration) Equals(rhs *IsolationGroupConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && _List_IsolationGroupPartition_Equals(v.IsolationGroups, rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_IsolationGroupPartition_Zapper []*IsolationGroupPartition\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_IsolationGroupPartition_Zapper.\nfunc (l _List_IsolationGroupPartition_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of IsolationGroupConfiguration.\nfunc (v *IsolationGroupConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"isolationGroups\", (_List_IsolationGroupPartition_Zapper)(v.IsolationGroups)))\n\t}\n\treturn err\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *IsolationGroupConfiguration) GetIsolationGroups() (o []*IsolationGroupPartition) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *IsolationGroupConfiguration) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\ntype IsolationGroupMetrics struct {\n\tNewTasksPerSecond *float64 `json:\"newTasksPerSecond,omitempty\"`\n\tPollerCount       *int64   `json:\"pollerCount,omitempty\"`\n}\n\n// ToWire translates a IsolationGroupMetrics struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *IsolationGroupMetrics) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NewTasksPerSecond != nil {\n\t\tw, err = wire.NewValueDouble(*(v.NewTasksPerSecond)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PollerCount != nil {\n\t\tw, err = wire.NewValueI64(*(v.PollerCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a IsolationGroupMetrics struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a IsolationGroupMetrics struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v IsolationGroupMetrics\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *IsolationGroupMetrics) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.NewTasksPerSecond = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PollerCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a IsolationGroupMetrics struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a IsolationGroupMetrics struct could not be encoded.\nfunc (v *IsolationGroupMetrics) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NewTasksPerSecond != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.NewTasksPerSecond)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PollerCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PollerCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a IsolationGroupMetrics struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a IsolationGroupMetrics struct could not be generated from the wire\n// representation.\nfunc (v *IsolationGroupMetrics) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.NewTasksPerSecond = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PollerCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a IsolationGroupMetrics\n// struct.\nfunc (v *IsolationGroupMetrics) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.NewTasksPerSecond != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewTasksPerSecond: %v\", *(v.NewTasksPerSecond))\n\t\ti++\n\t}\n\tif v.PollerCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"PollerCount: %v\", *(v.PollerCount))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"IsolationGroupMetrics{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Double_EqualsPtr(lhs, rhs *float64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this IsolationGroupMetrics match the\n// provided IsolationGroupMetrics.\n//\n// This function performs a deep comparison.\nfunc (v *IsolationGroupMetrics) Equals(rhs *IsolationGroupMetrics) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.NewTasksPerSecond, rhs.NewTasksPerSecond) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PollerCount, rhs.PollerCount) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of IsolationGroupMetrics.\nfunc (v *IsolationGroupMetrics) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NewTasksPerSecond != nil {\n\t\tenc.AddFloat64(\"newTasksPerSecond\", *v.NewTasksPerSecond)\n\t}\n\tif v.PollerCount != nil {\n\t\tenc.AddInt64(\"pollerCount\", *v.PollerCount)\n\t}\n\treturn err\n}\n\n// GetNewTasksPerSecond returns the value of NewTasksPerSecond if it is set or its\n// zero value if it is unset.\nfunc (v *IsolationGroupMetrics) GetNewTasksPerSecond() (o float64) {\n\tif v != nil && v.NewTasksPerSecond != nil {\n\t\treturn *v.NewTasksPerSecond\n\t}\n\n\treturn\n}\n\n// IsSetNewTasksPerSecond returns true if NewTasksPerSecond is not nil.\nfunc (v *IsolationGroupMetrics) IsSetNewTasksPerSecond() bool {\n\treturn v != nil && v.NewTasksPerSecond != nil\n}\n\n// GetPollerCount returns the value of PollerCount if it is set or its\n// zero value if it is unset.\nfunc (v *IsolationGroupMetrics) GetPollerCount() (o int64) {\n\tif v != nil && v.PollerCount != nil {\n\t\treturn *v.PollerCount\n\t}\n\n\treturn\n}\n\n// IsSetPollerCount returns true if PollerCount is not nil.\nfunc (v *IsolationGroupMetrics) IsSetPollerCount() bool {\n\treturn v != nil && v.PollerCount != nil\n}\n\ntype IsolationGroupPartition struct {\n\tName  *string              `json:\"name,omitempty\"`\n\tState *IsolationGroupState `json:\"state,omitempty\"`\n}\n\n// ToWire translates a IsolationGroupPartition struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *IsolationGroupPartition) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tw, err = v.State.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _IsolationGroupState_Read(w wire.Value) (IsolationGroupState, error) {\n\tvar v IsolationGroupState\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a IsolationGroupPartition struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a IsolationGroupPartition struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v IsolationGroupPartition\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *IsolationGroupPartition) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x IsolationGroupState\n\t\t\t\tx, err = _IsolationGroupState_Read(field.Value)\n\t\t\t\tv.State = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a IsolationGroupPartition struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a IsolationGroupPartition struct could not be encoded.\nfunc (v *IsolationGroupPartition) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.State != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.State.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _IsolationGroupState_Decode(sr stream.Reader) (IsolationGroupState, error) {\n\tvar v IsolationGroupState\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a IsolationGroupPartition struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a IsolationGroupPartition struct could not be generated from the wire\n// representation.\nfunc (v *IsolationGroupPartition) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x IsolationGroupState\n\t\t\tx, err = _IsolationGroupState_Decode(sr)\n\t\t\tv.State = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a IsolationGroupPartition\n// struct.\nfunc (v *IsolationGroupPartition) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tfields[i] = fmt.Sprintf(\"State: %v\", *(v.State))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"IsolationGroupPartition{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _IsolationGroupState_EqualsPtr(lhs, rhs *IsolationGroupState) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this IsolationGroupPartition match the\n// provided IsolationGroupPartition.\n//\n// This function performs a deep comparison.\nfunc (v *IsolationGroupPartition) Equals(rhs *IsolationGroupPartition) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_IsolationGroupState_EqualsPtr(v.State, rhs.State) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of IsolationGroupPartition.\nfunc (v *IsolationGroupPartition) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.State != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"state\", *v.State))\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *IsolationGroupPartition) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *IsolationGroupPartition) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetState returns the value of State if it is set or its\n// zero value if it is unset.\nfunc (v *IsolationGroupPartition) GetState() (o IsolationGroupState) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\n\treturn\n}\n\n// IsSetState returns true if State is not nil.\nfunc (v *IsolationGroupPartition) IsSetState() bool {\n\treturn v != nil && v.State != nil\n}\n\ntype IsolationGroupState int32\n\nconst (\n\tIsolationGroupStateInvalid IsolationGroupState = 0\n\tIsolationGroupStateHealthy IsolationGroupState = 1\n\tIsolationGroupStateDrained IsolationGroupState = 2\n)\n\n// IsolationGroupState_Values returns all recognized values of IsolationGroupState.\nfunc IsolationGroupState_Values() []IsolationGroupState {\n\treturn []IsolationGroupState{\n\t\tIsolationGroupStateInvalid,\n\t\tIsolationGroupStateHealthy,\n\t\tIsolationGroupStateDrained,\n\t}\n}\n\n// UnmarshalText tries to decode IsolationGroupState from a byte slice\n// containing its name.\n//\n//\tvar v IsolationGroupState\n//\terr := v.UnmarshalText([]byte(\"INVALID\"))\nfunc (v *IsolationGroupState) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"INVALID\":\n\t\t*v = IsolationGroupStateInvalid\n\t\treturn nil\n\tcase \"HEALTHY\":\n\t\t*v = IsolationGroupStateHealthy\n\t\treturn nil\n\tcase \"DRAINED\":\n\t\t*v = IsolationGroupStateDrained\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"IsolationGroupState\", err)\n\t\t}\n\t\t*v = IsolationGroupState(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes IsolationGroupState to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v IsolationGroupState) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"INVALID\"), nil\n\tcase 1:\n\t\treturn []byte(\"HEALTHY\"), nil\n\tcase 2:\n\t\treturn []byte(\"DRAINED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of IsolationGroupState.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v IsolationGroupState) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"INVALID\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"HEALTHY\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"DRAINED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v IsolationGroupState) Ptr() *IsolationGroupState {\n\treturn &v\n}\n\n// Encode encodes IsolationGroupState directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v IsolationGroupState\n//\treturn v.Encode(sWriter)\nfunc (v IsolationGroupState) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates IsolationGroupState into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v IsolationGroupState) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes IsolationGroupState from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return IsolationGroupState(0), err\n//\t}\n//\n//\tvar v IsolationGroupState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return IsolationGroupState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *IsolationGroupState) FromWire(w wire.Value) error {\n\t*v = (IsolationGroupState)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded IsolationGroupState directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v IsolationGroupState\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return IsolationGroupState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *IsolationGroupState) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (IsolationGroupState)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of IsolationGroupState.\nfunc (v IsolationGroupState) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"INVALID\"\n\tcase 1:\n\t\treturn \"HEALTHY\"\n\tcase 2:\n\t\treturn \"DRAINED\"\n\t}\n\treturn fmt.Sprintf(\"IsolationGroupState(%d)\", w)\n}\n\n// Equals returns true if this IsolationGroupState value matches the provided\n// value.\nfunc (v IsolationGroupState) Equals(rhs IsolationGroupState) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes IsolationGroupState into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v IsolationGroupState) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"INVALID\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"HEALTHY\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"DRAINED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode IsolationGroupState from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *IsolationGroupState) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"IsolationGroupState\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"IsolationGroupState\")\n\t\t}\n\t\t*v = (IsolationGroupState)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"IsolationGroupState\")\n\t}\n}\n\ntype LimitExceededError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a LimitExceededError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *LimitExceededError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a LimitExceededError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a LimitExceededError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v LimitExceededError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *LimitExceededError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of LimitExceededError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a LimitExceededError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a LimitExceededError struct could not be encoded.\nfunc (v *LimitExceededError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a LimitExceededError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a LimitExceededError struct could not be generated from the wire\n// representation.\nfunc (v *LimitExceededError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of LimitExceededError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a LimitExceededError\n// struct.\nfunc (v *LimitExceededError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"LimitExceededError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*LimitExceededError) ErrorName() string {\n\treturn \"LimitExceededError\"\n}\n\n// Equals returns true if all the fields of this LimitExceededError match the\n// provided LimitExceededError.\n//\n// This function performs a deep comparison.\nfunc (v *LimitExceededError) Equals(rhs *LimitExceededError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of LimitExceededError.\nfunc (v *LimitExceededError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *LimitExceededError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *LimitExceededError) Error() string {\n\treturn v.String()\n}\n\ntype ListArchivedWorkflowExecutionsRequest struct {\n\tDomain        *string `json:\"domain,omitempty\"`\n\tPageSize      *int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte  `json:\"nextPageToken,omitempty\"`\n\tQuery         *string `json:\"query,omitempty\"`\n}\n\n// ToWire translates a ListArchivedWorkflowExecutionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListArchivedWorkflowExecutionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = wire.NewValueString(*(v.Query)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListArchivedWorkflowExecutionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListArchivedWorkflowExecutionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListArchivedWorkflowExecutionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListArchivedWorkflowExecutionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Query = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListArchivedWorkflowExecutionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListArchivedWorkflowExecutionsRequest struct could not be encoded.\nfunc (v *ListArchivedWorkflowExecutionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Query)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListArchivedWorkflowExecutionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListArchivedWorkflowExecutionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListArchivedWorkflowExecutionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Query = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListArchivedWorkflowExecutionsRequest\n// struct.\nfunc (v *ListArchivedWorkflowExecutionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", *(v.Query))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListArchivedWorkflowExecutionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListArchivedWorkflowExecutionsRequest match the\n// provided ListArchivedWorkflowExecutionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListArchivedWorkflowExecutionsRequest) Equals(rhs *ListArchivedWorkflowExecutionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Query, rhs.Query) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListArchivedWorkflowExecutionsRequest.\nfunc (v *ListArchivedWorkflowExecutionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.Query != nil {\n\t\tenc.AddString(\"query\", *v.Query)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListArchivedWorkflowExecutionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *ListArchivedWorkflowExecutionsRequest) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListArchivedWorkflowExecutionsRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil && v.Query != nil {\n\t\treturn *v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *ListArchivedWorkflowExecutionsRequest) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\ntype ListArchivedWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\ntype _List_WorkflowExecutionInfo_ValueList []*WorkflowExecutionInfo\n\nfunc (v _List_WorkflowExecutionInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*WorkflowExecutionInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_WorkflowExecutionInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_WorkflowExecutionInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_WorkflowExecutionInfo_ValueList) Close() {}\n\n// ToWire translates a ListArchivedWorkflowExecutionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListArchivedWorkflowExecutionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecutionInfo_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _List_WorkflowExecutionInfo_Read(l wire.ValueList) ([]*WorkflowExecutionInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*WorkflowExecutionInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _WorkflowExecutionInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ListArchivedWorkflowExecutionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListArchivedWorkflowExecutionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListArchivedWorkflowExecutionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListArchivedWorkflowExecutionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_WorkflowExecutionInfo_Encode(val []*WorkflowExecutionInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*WorkflowExecutionInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ListArchivedWorkflowExecutionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListArchivedWorkflowExecutionsResponse struct could not be encoded.\nfunc (v *ListArchivedWorkflowExecutionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecutionInfo_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _List_WorkflowExecutionInfo_Decode(sr stream.Reader) ([]*WorkflowExecutionInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*WorkflowExecutionInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _WorkflowExecutionInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ListArchivedWorkflowExecutionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListArchivedWorkflowExecutionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListArchivedWorkflowExecutionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListArchivedWorkflowExecutionsResponse\n// struct.\nfunc (v *ListArchivedWorkflowExecutionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListArchivedWorkflowExecutionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_WorkflowExecutionInfo_Equals(lhs, rhs []*WorkflowExecutionInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ListArchivedWorkflowExecutionsResponse match the\n// provided ListArchivedWorkflowExecutionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListArchivedWorkflowExecutionsResponse) Equals(rhs *ListArchivedWorkflowExecutionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecutionInfo_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_WorkflowExecutionInfo_Zapper []*WorkflowExecutionInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_WorkflowExecutionInfo_Zapper.\nfunc (l _List_WorkflowExecutionInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListArchivedWorkflowExecutionsResponse.\nfunc (v *ListArchivedWorkflowExecutionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecutionInfo_Zapper)(v.Executions)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ListArchivedWorkflowExecutionsResponse) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListArchivedWorkflowExecutionsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListArchivedWorkflowExecutionsResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListClosedWorkflowExecutionsRequest struct {\n\tDomain          *string                       `json:\"domain,omitempty\"`\n\tMaximumPageSize *int32                        `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken   []byte                        `json:\"nextPageToken,omitempty\"`\n\tStartTimeFilter *StartTimeFilter              `json:\"StartTimeFilter,omitempty\"`\n\tExecutionFilter *WorkflowExecutionFilter      `json:\"executionFilter,omitempty\"`\n\tTypeFilter      *WorkflowTypeFilter           `json:\"typeFilter,omitempty\"`\n\tStatusFilter    *WorkflowExecutionCloseStatus `json:\"statusFilter,omitempty\"`\n}\n\n// ToWire translates a ListClosedWorkflowExecutionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListClosedWorkflowExecutionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartTimeFilter != nil {\n\t\tw, err = v.StartTimeFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionFilter != nil {\n\t\tw, err = v.ExecutionFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.TypeFilter != nil {\n\t\tw, err = v.TypeFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.StatusFilter != nil {\n\t\tw, err = v.StatusFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartTimeFilter_Read(w wire.Value) (*StartTimeFilter, error) {\n\tvar v StartTimeFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionFilter_Read(w wire.Value) (*WorkflowExecutionFilter, error) {\n\tvar v WorkflowExecutionFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowTypeFilter_Read(w wire.Value) (*WorkflowTypeFilter, error) {\n\tvar v WorkflowTypeFilter\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCloseStatus_Read(w wire.Value) (WorkflowExecutionCloseStatus, error) {\n\tvar v WorkflowExecutionCloseStatus\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a ListClosedWorkflowExecutionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListClosedWorkflowExecutionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListClosedWorkflowExecutionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListClosedWorkflowExecutionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartTimeFilter, err = _StartTimeFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExecutionFilter, err = _WorkflowExecutionFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TypeFilter, err = _WorkflowTypeFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\t\tx, err = _WorkflowExecutionCloseStatus_Read(field.Value)\n\t\t\t\tv.StatusFilter = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListClosedWorkflowExecutionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListClosedWorkflowExecutionsRequest struct could not be encoded.\nfunc (v *ListClosedWorkflowExecutionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartTimeFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartTimeFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExecutionFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TypeFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TypeFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StatusFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StatusFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartTimeFilter_Decode(sr stream.Reader) (*StartTimeFilter, error) {\n\tvar v StartTimeFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionFilter_Decode(sr stream.Reader) (*WorkflowExecutionFilter, error) {\n\tvar v WorkflowExecutionFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowTypeFilter_Decode(sr stream.Reader) (*WorkflowTypeFilter, error) {\n\tvar v WorkflowTypeFilter\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionCloseStatus_Decode(sr stream.Reader) (WorkflowExecutionCloseStatus, error) {\n\tvar v WorkflowExecutionCloseStatus\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a ListClosedWorkflowExecutionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListClosedWorkflowExecutionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListClosedWorkflowExecutionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.StartTimeFilter, err = _StartTimeFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.ExecutionFilter, err = _WorkflowExecutionFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.TypeFilter, err = _WorkflowTypeFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\tx, err = _WorkflowExecutionCloseStatus_Decode(sr)\n\t\t\tv.StatusFilter = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListClosedWorkflowExecutionsRequest\n// struct.\nfunc (v *ListClosedWorkflowExecutionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.StartTimeFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartTimeFilter: %v\", v.StartTimeFilter)\n\t\ti++\n\t}\n\tif v.ExecutionFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionFilter: %v\", v.ExecutionFilter)\n\t\ti++\n\t}\n\tif v.TypeFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"TypeFilter: %v\", v.TypeFilter)\n\t\ti++\n\t}\n\tif v.StatusFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"StatusFilter: %v\", *(v.StatusFilter))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListClosedWorkflowExecutionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _WorkflowExecutionCloseStatus_EqualsPtr(lhs, rhs *WorkflowExecutionCloseStatus) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this ListClosedWorkflowExecutionsRequest match the\n// provided ListClosedWorkflowExecutionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListClosedWorkflowExecutionsRequest) Equals(rhs *ListClosedWorkflowExecutionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !((v.StartTimeFilter == nil && rhs.StartTimeFilter == nil) || (v.StartTimeFilter != nil && rhs.StartTimeFilter != nil && v.StartTimeFilter.Equals(rhs.StartTimeFilter))) {\n\t\treturn false\n\t}\n\tif !((v.ExecutionFilter == nil && rhs.ExecutionFilter == nil) || (v.ExecutionFilter != nil && rhs.ExecutionFilter != nil && v.ExecutionFilter.Equals(rhs.ExecutionFilter))) {\n\t\treturn false\n\t}\n\tif !((v.TypeFilter == nil && rhs.TypeFilter == nil) || (v.TypeFilter != nil && rhs.TypeFilter != nil && v.TypeFilter.Equals(rhs.TypeFilter))) {\n\t\treturn false\n\t}\n\tif !_WorkflowExecutionCloseStatus_EqualsPtr(v.StatusFilter, rhs.StatusFilter) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListClosedWorkflowExecutionsRequest.\nfunc (v *ListClosedWorkflowExecutionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.StartTimeFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"StartTimeFilter\", v.StartTimeFilter))\n\t}\n\tif v.ExecutionFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"executionFilter\", v.ExecutionFilter))\n\t}\n\tif v.TypeFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"typeFilter\", v.TypeFilter))\n\t}\n\tif v.StatusFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"statusFilter\", *v.StatusFilter))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetStartTimeFilter returns the value of StartTimeFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetStartTimeFilter() (o *StartTimeFilter) {\n\tif v != nil && v.StartTimeFilter != nil {\n\t\treturn v.StartTimeFilter\n\t}\n\n\treturn\n}\n\n// IsSetStartTimeFilter returns true if StartTimeFilter is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetStartTimeFilter() bool {\n\treturn v != nil && v.StartTimeFilter != nil\n}\n\n// GetExecutionFilter returns the value of ExecutionFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetExecutionFilter() (o *WorkflowExecutionFilter) {\n\tif v != nil && v.ExecutionFilter != nil {\n\t\treturn v.ExecutionFilter\n\t}\n\n\treturn\n}\n\n// IsSetExecutionFilter returns true if ExecutionFilter is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetExecutionFilter() bool {\n\treturn v != nil && v.ExecutionFilter != nil\n}\n\n// GetTypeFilter returns the value of TypeFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetTypeFilter() (o *WorkflowTypeFilter) {\n\tif v != nil && v.TypeFilter != nil {\n\t\treturn v.TypeFilter\n\t}\n\n\treturn\n}\n\n// IsSetTypeFilter returns true if TypeFilter is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetTypeFilter() bool {\n\treturn v != nil && v.TypeFilter != nil\n}\n\n// GetStatusFilter returns the value of StatusFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsRequest) GetStatusFilter() (o WorkflowExecutionCloseStatus) {\n\tif v != nil && v.StatusFilter != nil {\n\t\treturn *v.StatusFilter\n\t}\n\n\treturn\n}\n\n// IsSetStatusFilter returns true if StatusFilter is not nil.\nfunc (v *ListClosedWorkflowExecutionsRequest) IsSetStatusFilter() bool {\n\treturn v != nil && v.StatusFilter != nil\n}\n\ntype ListClosedWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ListClosedWorkflowExecutionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListClosedWorkflowExecutionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecutionInfo_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListClosedWorkflowExecutionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListClosedWorkflowExecutionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListClosedWorkflowExecutionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListClosedWorkflowExecutionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListClosedWorkflowExecutionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListClosedWorkflowExecutionsResponse struct could not be encoded.\nfunc (v *ListClosedWorkflowExecutionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecutionInfo_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListClosedWorkflowExecutionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListClosedWorkflowExecutionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListClosedWorkflowExecutionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListClosedWorkflowExecutionsResponse\n// struct.\nfunc (v *ListClosedWorkflowExecutionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListClosedWorkflowExecutionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListClosedWorkflowExecutionsResponse match the\n// provided ListClosedWorkflowExecutionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListClosedWorkflowExecutionsResponse) Equals(rhs *ListClosedWorkflowExecutionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecutionInfo_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListClosedWorkflowExecutionsResponse.\nfunc (v *ListClosedWorkflowExecutionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecutionInfo_Zapper)(v.Executions)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ListClosedWorkflowExecutionsResponse) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListClosedWorkflowExecutionsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListClosedWorkflowExecutionsResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListDomainsRequest struct {\n\tPageSize      *int32 `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ListDomainsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListDomainsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListDomainsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListDomainsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListDomainsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListDomainsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListDomainsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListDomainsRequest struct could not be encoded.\nfunc (v *ListDomainsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListDomainsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListDomainsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListDomainsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListDomainsRequest\n// struct.\nfunc (v *ListDomainsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListDomainsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListDomainsRequest match the\n// provided ListDomainsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListDomainsRequest) Equals(rhs *ListDomainsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListDomainsRequest.\nfunc (v *ListDomainsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ListDomainsRequest) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *ListDomainsRequest) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListDomainsRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListDomainsRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListDomainsResponse struct {\n\tDomains       []*DescribeDomainResponse `json:\"domains,omitempty\"`\n\tNextPageToken []byte                    `json:\"nextPageToken,omitempty\"`\n}\n\ntype _List_DescribeDomainResponse_ValueList []*DescribeDomainResponse\n\nfunc (v _List_DescribeDomainResponse_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DescribeDomainResponse', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_DescribeDomainResponse_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_DescribeDomainResponse_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_DescribeDomainResponse_ValueList) Close() {}\n\n// ToWire translates a ListDomainsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListDomainsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domains != nil {\n\t\tw, err = wire.NewValueList(_List_DescribeDomainResponse_ValueList(v.Domains)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _DescribeDomainResponse_Read(w wire.Value) (*DescribeDomainResponse, error) {\n\tvar v DescribeDomainResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_DescribeDomainResponse_Read(l wire.ValueList) ([]*DescribeDomainResponse, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*DescribeDomainResponse, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _DescribeDomainResponse_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ListDomainsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListDomainsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListDomainsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListDomainsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Domains, err = _List_DescribeDomainResponse_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_DescribeDomainResponse_Encode(val []*DescribeDomainResponse, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*DescribeDomainResponse', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ListDomainsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListDomainsResponse struct could not be encoded.\nfunc (v *ListDomainsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domains != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_DescribeDomainResponse_Encode(v.Domains, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _DescribeDomainResponse_Decode(sr stream.Reader) (*DescribeDomainResponse, error) {\n\tvar v DescribeDomainResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_DescribeDomainResponse_Decode(sr stream.Reader) ([]*DescribeDomainResponse, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*DescribeDomainResponse, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _DescribeDomainResponse_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ListDomainsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListDomainsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListDomainsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Domains, err = _List_DescribeDomainResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListDomainsResponse\n// struct.\nfunc (v *ListDomainsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domains != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domains: %v\", v.Domains)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListDomainsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_DescribeDomainResponse_Equals(lhs, rhs []*DescribeDomainResponse) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ListDomainsResponse match the\n// provided ListDomainsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListDomainsResponse) Equals(rhs *ListDomainsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Domains == nil && rhs.Domains == nil) || (v.Domains != nil && rhs.Domains != nil && _List_DescribeDomainResponse_Equals(v.Domains, rhs.Domains))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_DescribeDomainResponse_Zapper []*DescribeDomainResponse\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_DescribeDomainResponse_Zapper.\nfunc (l _List_DescribeDomainResponse_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListDomainsResponse.\nfunc (v *ListDomainsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domains != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"domains\", (_List_DescribeDomainResponse_Zapper)(v.Domains)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetDomains returns the value of Domains if it is set or its\n// zero value if it is unset.\nfunc (v *ListDomainsResponse) GetDomains() (o []*DescribeDomainResponse) {\n\tif v != nil && v.Domains != nil {\n\t\treturn v.Domains\n\t}\n\n\treturn\n}\n\n// IsSetDomains returns true if Domains is not nil.\nfunc (v *ListDomainsResponse) IsSetDomains() bool {\n\treturn v != nil && v.Domains != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListDomainsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListDomainsResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListFailoverHistoryRequest struct {\n\tFilters    *ListFailoverHistoryRequestFilters `json:\"filters,omitempty\"`\n\tPagination *PaginationOptions                 `json:\"pagination,omitempty\"`\n}\n\n// ToWire translates a ListFailoverHistoryRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListFailoverHistoryRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Filters != nil {\n\t\tw, err = v.Filters.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Pagination != nil {\n\t\tw, err = v.Pagination.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ListFailoverHistoryRequestFilters_Read(w wire.Value) (*ListFailoverHistoryRequestFilters, error) {\n\tvar v ListFailoverHistoryRequestFilters\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _PaginationOptions_Read(w wire.Value) (*PaginationOptions, error) {\n\tvar v PaginationOptions\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a ListFailoverHistoryRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListFailoverHistoryRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListFailoverHistoryRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListFailoverHistoryRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Filters, err = _ListFailoverHistoryRequestFilters_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Pagination, err = _PaginationOptions_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListFailoverHistoryRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListFailoverHistoryRequest struct could not be encoded.\nfunc (v *ListFailoverHistoryRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Filters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Filters.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Pagination != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Pagination.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ListFailoverHistoryRequestFilters_Decode(sr stream.Reader) (*ListFailoverHistoryRequestFilters, error) {\n\tvar v ListFailoverHistoryRequestFilters\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _PaginationOptions_Decode(sr stream.Reader) (*PaginationOptions, error) {\n\tvar v PaginationOptions\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a ListFailoverHistoryRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListFailoverHistoryRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListFailoverHistoryRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Filters, err = _ListFailoverHistoryRequestFilters_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Pagination, err = _PaginationOptions_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListFailoverHistoryRequest\n// struct.\nfunc (v *ListFailoverHistoryRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Filters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Filters: %v\", v.Filters)\n\t\ti++\n\t}\n\tif v.Pagination != nil {\n\t\tfields[i] = fmt.Sprintf(\"Pagination: %v\", v.Pagination)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListFailoverHistoryRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListFailoverHistoryRequest match the\n// provided ListFailoverHistoryRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListFailoverHistoryRequest) Equals(rhs *ListFailoverHistoryRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Filters == nil && rhs.Filters == nil) || (v.Filters != nil && rhs.Filters != nil && v.Filters.Equals(rhs.Filters))) {\n\t\treturn false\n\t}\n\tif !((v.Pagination == nil && rhs.Pagination == nil) || (v.Pagination != nil && rhs.Pagination != nil && v.Pagination.Equals(rhs.Pagination))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListFailoverHistoryRequest.\nfunc (v *ListFailoverHistoryRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Filters != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"filters\", v.Filters))\n\t}\n\tif v.Pagination != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"pagination\", v.Pagination))\n\t}\n\treturn err\n}\n\n// GetFilters returns the value of Filters if it is set or its\n// zero value if it is unset.\nfunc (v *ListFailoverHistoryRequest) GetFilters() (o *ListFailoverHistoryRequestFilters) {\n\tif v != nil && v.Filters != nil {\n\t\treturn v.Filters\n\t}\n\n\treturn\n}\n\n// IsSetFilters returns true if Filters is not nil.\nfunc (v *ListFailoverHistoryRequest) IsSetFilters() bool {\n\treturn v != nil && v.Filters != nil\n}\n\n// GetPagination returns the value of Pagination if it is set or its\n// zero value if it is unset.\nfunc (v *ListFailoverHistoryRequest) GetPagination() (o *PaginationOptions) {\n\tif v != nil && v.Pagination != nil {\n\t\treturn v.Pagination\n\t}\n\n\treturn\n}\n\n// IsSetPagination returns true if Pagination is not nil.\nfunc (v *ListFailoverHistoryRequest) IsSetPagination() bool {\n\treturn v != nil && v.Pagination != nil\n}\n\ntype ListFailoverHistoryRequestFilters struct {\n\tDomainID *string `json:\"domainID,omitempty\"`\n}\n\n// ToWire translates a ListFailoverHistoryRequestFilters struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListFailoverHistoryRequestFilters) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListFailoverHistoryRequestFilters struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListFailoverHistoryRequestFilters struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListFailoverHistoryRequestFilters\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListFailoverHistoryRequestFilters) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListFailoverHistoryRequestFilters struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListFailoverHistoryRequestFilters struct could not be encoded.\nfunc (v *ListFailoverHistoryRequestFilters) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListFailoverHistoryRequestFilters struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListFailoverHistoryRequestFilters struct could not be generated from the wire\n// representation.\nfunc (v *ListFailoverHistoryRequestFilters) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListFailoverHistoryRequestFilters\n// struct.\nfunc (v *ListFailoverHistoryRequestFilters) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListFailoverHistoryRequestFilters{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListFailoverHistoryRequestFilters match the\n// provided ListFailoverHistoryRequestFilters.\n//\n// This function performs a deep comparison.\nfunc (v *ListFailoverHistoryRequestFilters) Equals(rhs *ListFailoverHistoryRequestFilters) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListFailoverHistoryRequestFilters.\nfunc (v *ListFailoverHistoryRequestFilters) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ListFailoverHistoryRequestFilters) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *ListFailoverHistoryRequestFilters) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\ntype ListFailoverHistoryResponse struct {\n\tFailoverEvents []*FailoverEvent `json:\"failoverEvents,omitempty\"`\n\tNextPageToken  []byte           `json:\"nextPageToken,omitempty\"`\n}\n\ntype _List_FailoverEvent_ValueList []*FailoverEvent\n\nfunc (v _List_FailoverEvent_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverEvent', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_FailoverEvent_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_FailoverEvent_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_FailoverEvent_ValueList) Close() {}\n\n// ToWire translates a ListFailoverHistoryResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListFailoverHistoryResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.FailoverEvents != nil {\n\t\tw, err = wire.NewValueList(_List_FailoverEvent_ValueList(v.FailoverEvents)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _FailoverEvent_Read(w wire.Value) (*FailoverEvent, error) {\n\tvar v FailoverEvent\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_FailoverEvent_Read(l wire.ValueList) ([]*FailoverEvent, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*FailoverEvent, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _FailoverEvent_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ListFailoverHistoryResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListFailoverHistoryResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListFailoverHistoryResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListFailoverHistoryResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.FailoverEvents, err = _List_FailoverEvent_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_FailoverEvent_Encode(val []*FailoverEvent, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*FailoverEvent', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ListFailoverHistoryResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListFailoverHistoryResponse struct could not be encoded.\nfunc (v *ListFailoverHistoryResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.FailoverEvents != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_FailoverEvent_Encode(v.FailoverEvents, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _FailoverEvent_Decode(sr stream.Reader) (*FailoverEvent, error) {\n\tvar v FailoverEvent\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_FailoverEvent_Decode(sr stream.Reader) ([]*FailoverEvent, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*FailoverEvent, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _FailoverEvent_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ListFailoverHistoryResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListFailoverHistoryResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListFailoverHistoryResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.FailoverEvents, err = _List_FailoverEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListFailoverHistoryResponse\n// struct.\nfunc (v *ListFailoverHistoryResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.FailoverEvents != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverEvents: %v\", v.FailoverEvents)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListFailoverHistoryResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_FailoverEvent_Equals(lhs, rhs []*FailoverEvent) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ListFailoverHistoryResponse match the\n// provided ListFailoverHistoryResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListFailoverHistoryResponse) Equals(rhs *ListFailoverHistoryResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.FailoverEvents == nil && rhs.FailoverEvents == nil) || (v.FailoverEvents != nil && rhs.FailoverEvents != nil && _List_FailoverEvent_Equals(v.FailoverEvents, rhs.FailoverEvents))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_FailoverEvent_Zapper []*FailoverEvent\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_FailoverEvent_Zapper.\nfunc (l _List_FailoverEvent_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListFailoverHistoryResponse.\nfunc (v *ListFailoverHistoryResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.FailoverEvents != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"failoverEvents\", (_List_FailoverEvent_Zapper)(v.FailoverEvents)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetFailoverEvents returns the value of FailoverEvents if it is set or its\n// zero value if it is unset.\nfunc (v *ListFailoverHistoryResponse) GetFailoverEvents() (o []*FailoverEvent) {\n\tif v != nil && v.FailoverEvents != nil {\n\t\treturn v.FailoverEvents\n\t}\n\n\treturn\n}\n\n// IsSetFailoverEvents returns true if FailoverEvents is not nil.\nfunc (v *ListFailoverHistoryResponse) IsSetFailoverEvents() bool {\n\treturn v != nil && v.FailoverEvents != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListFailoverHistoryResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListFailoverHistoryResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListOpenWorkflowExecutionsRequest struct {\n\tDomain          *string                  `json:\"domain,omitempty\"`\n\tMaximumPageSize *int32                   `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken   []byte                   `json:\"nextPageToken,omitempty\"`\n\tStartTimeFilter *StartTimeFilter         `json:\"StartTimeFilter,omitempty\"`\n\tExecutionFilter *WorkflowExecutionFilter `json:\"executionFilter,omitempty\"`\n\tTypeFilter      *WorkflowTypeFilter      `json:\"typeFilter,omitempty\"`\n}\n\n// ToWire translates a ListOpenWorkflowExecutionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListOpenWorkflowExecutionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumPageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.StartTimeFilter != nil {\n\t\tw, err = v.StartTimeFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionFilter != nil {\n\t\tw, err = v.ExecutionFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.TypeFilter != nil {\n\t\tw, err = v.TypeFilter.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListOpenWorkflowExecutionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListOpenWorkflowExecutionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListOpenWorkflowExecutionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListOpenWorkflowExecutionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumPageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartTimeFilter, err = _StartTimeFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExecutionFilter, err = _WorkflowExecutionFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TypeFilter, err = _WorkflowTypeFilter_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListOpenWorkflowExecutionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListOpenWorkflowExecutionsRequest struct could not be encoded.\nfunc (v *ListOpenWorkflowExecutionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumPageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumPageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartTimeFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartTimeFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExecutionFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TypeFilter != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TypeFilter.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListOpenWorkflowExecutionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListOpenWorkflowExecutionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListOpenWorkflowExecutionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumPageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.StartTimeFilter, err = _StartTimeFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.ExecutionFilter, err = _WorkflowExecutionFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.TypeFilter, err = _WorkflowTypeFilter_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListOpenWorkflowExecutionsRequest\n// struct.\nfunc (v *ListOpenWorkflowExecutionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumPageSize: %v\", *(v.MaximumPageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.StartTimeFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartTimeFilter: %v\", v.StartTimeFilter)\n\t\ti++\n\t}\n\tif v.ExecutionFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionFilter: %v\", v.ExecutionFilter)\n\t\ti++\n\t}\n\tif v.TypeFilter != nil {\n\t\tfields[i] = fmt.Sprintf(\"TypeFilter: %v\", v.TypeFilter)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListOpenWorkflowExecutionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListOpenWorkflowExecutionsRequest match the\n// provided ListOpenWorkflowExecutionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListOpenWorkflowExecutionsRequest) Equals(rhs *ListOpenWorkflowExecutionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumPageSize, rhs.MaximumPageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !((v.StartTimeFilter == nil && rhs.StartTimeFilter == nil) || (v.StartTimeFilter != nil && rhs.StartTimeFilter != nil && v.StartTimeFilter.Equals(rhs.StartTimeFilter))) {\n\t\treturn false\n\t}\n\tif !((v.ExecutionFilter == nil && rhs.ExecutionFilter == nil) || (v.ExecutionFilter != nil && rhs.ExecutionFilter != nil && v.ExecutionFilter.Equals(rhs.ExecutionFilter))) {\n\t\treturn false\n\t}\n\tif !((v.TypeFilter == nil && rhs.TypeFilter == nil) || (v.TypeFilter != nil && rhs.TypeFilter != nil && v.TypeFilter.Equals(rhs.TypeFilter))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListOpenWorkflowExecutionsRequest.\nfunc (v *ListOpenWorkflowExecutionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.MaximumPageSize != nil {\n\t\tenc.AddInt32(\"maximumPageSize\", *v.MaximumPageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.StartTimeFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"StartTimeFilter\", v.StartTimeFilter))\n\t}\n\tif v.ExecutionFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"executionFilter\", v.ExecutionFilter))\n\t}\n\tif v.TypeFilter != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"typeFilter\", v.TypeFilter))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetMaximumPageSize returns the value of MaximumPageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil && v.MaximumPageSize != nil {\n\t\treturn *v.MaximumPageSize\n\t}\n\n\treturn\n}\n\n// IsSetMaximumPageSize returns true if MaximumPageSize is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetMaximumPageSize() bool {\n\treturn v != nil && v.MaximumPageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetStartTimeFilter returns the value of StartTimeFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetStartTimeFilter() (o *StartTimeFilter) {\n\tif v != nil && v.StartTimeFilter != nil {\n\t\treturn v.StartTimeFilter\n\t}\n\n\treturn\n}\n\n// IsSetStartTimeFilter returns true if StartTimeFilter is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetStartTimeFilter() bool {\n\treturn v != nil && v.StartTimeFilter != nil\n}\n\n// GetExecutionFilter returns the value of ExecutionFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetExecutionFilter() (o *WorkflowExecutionFilter) {\n\tif v != nil && v.ExecutionFilter != nil {\n\t\treturn v.ExecutionFilter\n\t}\n\n\treturn\n}\n\n// IsSetExecutionFilter returns true if ExecutionFilter is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetExecutionFilter() bool {\n\treturn v != nil && v.ExecutionFilter != nil\n}\n\n// GetTypeFilter returns the value of TypeFilter if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsRequest) GetTypeFilter() (o *WorkflowTypeFilter) {\n\tif v != nil && v.TypeFilter != nil {\n\t\treturn v.TypeFilter\n\t}\n\n\treturn\n}\n\n// IsSetTypeFilter returns true if TypeFilter is not nil.\nfunc (v *ListOpenWorkflowExecutionsRequest) IsSetTypeFilter() bool {\n\treturn v != nil && v.TypeFilter != nil\n}\n\ntype ListOpenWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ListOpenWorkflowExecutionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListOpenWorkflowExecutionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecutionInfo_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListOpenWorkflowExecutionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListOpenWorkflowExecutionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListOpenWorkflowExecutionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListOpenWorkflowExecutionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListOpenWorkflowExecutionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListOpenWorkflowExecutionsResponse struct could not be encoded.\nfunc (v *ListOpenWorkflowExecutionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecutionInfo_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListOpenWorkflowExecutionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListOpenWorkflowExecutionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListOpenWorkflowExecutionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListOpenWorkflowExecutionsResponse\n// struct.\nfunc (v *ListOpenWorkflowExecutionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListOpenWorkflowExecutionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListOpenWorkflowExecutionsResponse match the\n// provided ListOpenWorkflowExecutionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListOpenWorkflowExecutionsResponse) Equals(rhs *ListOpenWorkflowExecutionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecutionInfo_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListOpenWorkflowExecutionsResponse.\nfunc (v *ListOpenWorkflowExecutionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecutionInfo_Zapper)(v.Executions)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ListOpenWorkflowExecutionsResponse) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListOpenWorkflowExecutionsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListOpenWorkflowExecutionsResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ListTaskListPartitionsRequest struct {\n\tDomain   *string   `json:\"domain,omitempty\"`\n\tTaskList *TaskList `json:\"taskList,omitempty\"`\n}\n\n// ToWire translates a ListTaskListPartitionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListTaskListPartitionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListTaskListPartitionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListTaskListPartitionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListTaskListPartitionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListTaskListPartitionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListTaskListPartitionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListTaskListPartitionsRequest struct could not be encoded.\nfunc (v *ListTaskListPartitionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListTaskListPartitionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListTaskListPartitionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListTaskListPartitionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListTaskListPartitionsRequest\n// struct.\nfunc (v *ListTaskListPartitionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListTaskListPartitionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListTaskListPartitionsRequest match the\n// provided ListTaskListPartitionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListTaskListPartitionsRequest) Equals(rhs *ListTaskListPartitionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListTaskListPartitionsRequest.\nfunc (v *ListTaskListPartitionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListTaskListPartitionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ListTaskListPartitionsRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\ntype ListTaskListPartitionsResponse struct {\n\tActivityTaskListPartitions []*TaskListPartitionMetadata `json:\"activityTaskListPartitions,omitempty\"`\n\tDecisionTaskListPartitions []*TaskListPartitionMetadata `json:\"decisionTaskListPartitions,omitempty\"`\n}\n\ntype _List_TaskListPartitionMetadata_ValueList []*TaskListPartitionMetadata\n\nfunc (v _List_TaskListPartitionMetadata_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*TaskListPartitionMetadata', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_TaskListPartitionMetadata_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_TaskListPartitionMetadata_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_TaskListPartitionMetadata_ValueList) Close() {}\n\n// ToWire translates a ListTaskListPartitionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListTaskListPartitionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityTaskListPartitions != nil {\n\t\tw, err = wire.NewValueList(_List_TaskListPartitionMetadata_ValueList(v.ActivityTaskListPartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskListPartitions != nil {\n\t\tw, err = wire.NewValueList(_List_TaskListPartitionMetadata_ValueList(v.DecisionTaskListPartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListPartitionMetadata_Read(w wire.Value) (*TaskListPartitionMetadata, error) {\n\tvar v TaskListPartitionMetadata\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_TaskListPartitionMetadata_Read(l wire.ValueList) ([]*TaskListPartitionMetadata, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*TaskListPartitionMetadata, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _TaskListPartitionMetadata_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ListTaskListPartitionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListTaskListPartitionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListTaskListPartitionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListTaskListPartitionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.ActivityTaskListPartitions, err = _List_TaskListPartitionMetadata_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.DecisionTaskListPartitions, err = _List_TaskListPartitionMetadata_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_TaskListPartitionMetadata_Encode(val []*TaskListPartitionMetadata, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*TaskListPartitionMetadata', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ListTaskListPartitionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListTaskListPartitionsResponse struct could not be encoded.\nfunc (v *ListTaskListPartitionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityTaskListPartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_TaskListPartitionMetadata_Encode(v.ActivityTaskListPartitions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskListPartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_TaskListPartitionMetadata_Encode(v.DecisionTaskListPartitions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListPartitionMetadata_Decode(sr stream.Reader) (*TaskListPartitionMetadata, error) {\n\tvar v TaskListPartitionMetadata\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_TaskListPartitionMetadata_Decode(sr stream.Reader) ([]*TaskListPartitionMetadata, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*TaskListPartitionMetadata, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _TaskListPartitionMetadata_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ListTaskListPartitionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListTaskListPartitionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListTaskListPartitionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.ActivityTaskListPartitions, err = _List_TaskListPartitionMetadata_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.DecisionTaskListPartitions, err = _List_TaskListPartitionMetadata_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListTaskListPartitionsResponse\n// struct.\nfunc (v *ListTaskListPartitionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ActivityTaskListPartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityTaskListPartitions: %v\", v.ActivityTaskListPartitions)\n\t\ti++\n\t}\n\tif v.DecisionTaskListPartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskListPartitions: %v\", v.DecisionTaskListPartitions)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListTaskListPartitionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_TaskListPartitionMetadata_Equals(lhs, rhs []*TaskListPartitionMetadata) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ListTaskListPartitionsResponse match the\n// provided ListTaskListPartitionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListTaskListPartitionsResponse) Equals(rhs *ListTaskListPartitionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ActivityTaskListPartitions == nil && rhs.ActivityTaskListPartitions == nil) || (v.ActivityTaskListPartitions != nil && rhs.ActivityTaskListPartitions != nil && _List_TaskListPartitionMetadata_Equals(v.ActivityTaskListPartitions, rhs.ActivityTaskListPartitions))) {\n\t\treturn false\n\t}\n\tif !((v.DecisionTaskListPartitions == nil && rhs.DecisionTaskListPartitions == nil) || (v.DecisionTaskListPartitions != nil && rhs.DecisionTaskListPartitions != nil && _List_TaskListPartitionMetadata_Equals(v.DecisionTaskListPartitions, rhs.DecisionTaskListPartitions))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_TaskListPartitionMetadata_Zapper []*TaskListPartitionMetadata\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_TaskListPartitionMetadata_Zapper.\nfunc (l _List_TaskListPartitionMetadata_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListTaskListPartitionsResponse.\nfunc (v *ListTaskListPartitionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityTaskListPartitions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"activityTaskListPartitions\", (_List_TaskListPartitionMetadata_Zapper)(v.ActivityTaskListPartitions)))\n\t}\n\tif v.DecisionTaskListPartitions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"decisionTaskListPartitions\", (_List_TaskListPartitionMetadata_Zapper)(v.DecisionTaskListPartitions)))\n\t}\n\treturn err\n}\n\n// GetActivityTaskListPartitions returns the value of ActivityTaskListPartitions if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsResponse) GetActivityTaskListPartitions() (o []*TaskListPartitionMetadata) {\n\tif v != nil && v.ActivityTaskListPartitions != nil {\n\t\treturn v.ActivityTaskListPartitions\n\t}\n\n\treturn\n}\n\n// IsSetActivityTaskListPartitions returns true if ActivityTaskListPartitions is not nil.\nfunc (v *ListTaskListPartitionsResponse) IsSetActivityTaskListPartitions() bool {\n\treturn v != nil && v.ActivityTaskListPartitions != nil\n}\n\n// GetDecisionTaskListPartitions returns the value of DecisionTaskListPartitions if it is set or its\n// zero value if it is unset.\nfunc (v *ListTaskListPartitionsResponse) GetDecisionTaskListPartitions() (o []*TaskListPartitionMetadata) {\n\tif v != nil && v.DecisionTaskListPartitions != nil {\n\t\treturn v.DecisionTaskListPartitions\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskListPartitions returns true if DecisionTaskListPartitions is not nil.\nfunc (v *ListTaskListPartitionsResponse) IsSetDecisionTaskListPartitions() bool {\n\treturn v != nil && v.DecisionTaskListPartitions != nil\n}\n\ntype ListWorkflowExecutionsRequest struct {\n\tDomain        *string `json:\"domain,omitempty\"`\n\tPageSize      *int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte  `json:\"nextPageToken,omitempty\"`\n\tQuery         *string `json:\"query,omitempty\"`\n}\n\n// ToWire translates a ListWorkflowExecutionsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListWorkflowExecutionsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = wire.NewValueString(*(v.Query)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListWorkflowExecutionsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListWorkflowExecutionsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListWorkflowExecutionsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListWorkflowExecutionsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Query = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListWorkflowExecutionsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListWorkflowExecutionsRequest struct could not be encoded.\nfunc (v *ListWorkflowExecutionsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Query)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListWorkflowExecutionsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListWorkflowExecutionsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ListWorkflowExecutionsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Query = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListWorkflowExecutionsRequest\n// struct.\nfunc (v *ListWorkflowExecutionsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", *(v.Query))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListWorkflowExecutionsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListWorkflowExecutionsRequest match the\n// provided ListWorkflowExecutionsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ListWorkflowExecutionsRequest) Equals(rhs *ListWorkflowExecutionsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Query, rhs.Query) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListWorkflowExecutionsRequest.\nfunc (v *ListWorkflowExecutionsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.Query != nil {\n\t\tenc.AddString(\"query\", *v.Query)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ListWorkflowExecutionsRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsRequest) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *ListWorkflowExecutionsRequest) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListWorkflowExecutionsRequest) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil && v.Query != nil {\n\t\treturn *v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *ListWorkflowExecutionsRequest) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\ntype ListWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a ListWorkflowExecutionsResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ListWorkflowExecutionsResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Executions != nil {\n\t\tw, err = wire.NewValueList(_List_WorkflowExecutionInfo_ValueList(v.Executions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ListWorkflowExecutionsResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ListWorkflowExecutionsResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ListWorkflowExecutionsResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ListWorkflowExecutionsResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ListWorkflowExecutionsResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ListWorkflowExecutionsResponse struct could not be encoded.\nfunc (v *ListWorkflowExecutionsResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Executions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_WorkflowExecutionInfo_Encode(v.Executions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ListWorkflowExecutionsResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ListWorkflowExecutionsResponse struct could not be generated from the wire\n// representation.\nfunc (v *ListWorkflowExecutionsResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Executions, err = _List_WorkflowExecutionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ListWorkflowExecutionsResponse\n// struct.\nfunc (v *ListWorkflowExecutionsResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Executions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Executions: %v\", v.Executions)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ListWorkflowExecutionsResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ListWorkflowExecutionsResponse match the\n// provided ListWorkflowExecutionsResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ListWorkflowExecutionsResponse) Equals(rhs *ListWorkflowExecutionsResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Executions == nil && rhs.Executions == nil) || (v.Executions != nil && rhs.Executions != nil && _List_WorkflowExecutionInfo_Equals(v.Executions, rhs.Executions))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ListWorkflowExecutionsResponse.\nfunc (v *ListWorkflowExecutionsResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Executions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"executions\", (_List_WorkflowExecutionInfo_Zapper)(v.Executions)))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetExecutions returns the value of Executions if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\n\treturn\n}\n\n// IsSetExecutions returns true if Executions is not nil.\nfunc (v *ListWorkflowExecutionsResponse) IsSetExecutions() bool {\n\treturn v != nil && v.Executions != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *ListWorkflowExecutionsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *ListWorkflowExecutionsResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype MarkerRecordedEventAttributes struct {\n\tMarkerName                   *string `json:\"markerName,omitempty\"`\n\tDetails                      []byte  `json:\"details,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tHeader                       *Header `json:\"header,omitempty\"`\n}\n\n// ToWire translates a MarkerRecordedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *MarkerRecordedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MarkerName != nil {\n\t\tw, err = wire.NewValueString(*(v.MarkerName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a MarkerRecordedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a MarkerRecordedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v MarkerRecordedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *MarkerRecordedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MarkerName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a MarkerRecordedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a MarkerRecordedEventAttributes struct could not be encoded.\nfunc (v *MarkerRecordedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MarkerName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MarkerName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a MarkerRecordedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a MarkerRecordedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *MarkerRecordedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MarkerName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a MarkerRecordedEventAttributes\n// struct.\nfunc (v *MarkerRecordedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.MarkerName != nil {\n\t\tfields[i] = fmt.Sprintf(\"MarkerName: %v\", *(v.MarkerName))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"MarkerRecordedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this MarkerRecordedEventAttributes match the\n// provided MarkerRecordedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *MarkerRecordedEventAttributes) Equals(rhs *MarkerRecordedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MarkerName, rhs.MarkerName) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of MarkerRecordedEventAttributes.\nfunc (v *MarkerRecordedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MarkerName != nil {\n\t\tenc.AddString(\"markerName\", *v.MarkerName)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\treturn err\n}\n\n// GetMarkerName returns the value of MarkerName if it is set or its\n// zero value if it is unset.\nfunc (v *MarkerRecordedEventAttributes) GetMarkerName() (o string) {\n\tif v != nil && v.MarkerName != nil {\n\t\treturn *v.MarkerName\n\t}\n\n\treturn\n}\n\n// IsSetMarkerName returns true if MarkerName is not nil.\nfunc (v *MarkerRecordedEventAttributes) IsSetMarkerName() bool {\n\treturn v != nil && v.MarkerName != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *MarkerRecordedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *MarkerRecordedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *MarkerRecordedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *MarkerRecordedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *MarkerRecordedEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *MarkerRecordedEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\ntype Memo struct {\n\tFields map[string][]byte `json:\"fields,omitempty\"`\n}\n\n// ToWire translates a Memo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Memo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Fields != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Binary_MapItemList(v.Fields)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a Memo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Memo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Memo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Memo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Fields, err = _Map_String_Binary_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Memo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Memo struct could not be encoded.\nfunc (v *Memo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Fields != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Binary_Encode(v.Fields, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a Memo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Memo struct could not be generated from the wire\n// representation.\nfunc (v *Memo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.Fields, err = _Map_String_Binary_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Memo\n// struct.\nfunc (v *Memo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Fields != nil {\n\t\tfields[i] = fmt.Sprintf(\"Fields: %v\", v.Fields)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Memo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this Memo match the\n// provided Memo.\n//\n// This function performs a deep comparison.\nfunc (v *Memo) Equals(rhs *Memo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Fields == nil && rhs.Fields == nil) || (v.Fields != nil && rhs.Fields != nil && _Map_String_Binary_Equals(v.Fields, rhs.Fields))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Memo.\nfunc (v *Memo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Fields != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"fields\", (_Map_String_Binary_Zapper)(v.Fields)))\n\t}\n\treturn err\n}\n\n// GetFields returns the value of Fields if it is set or its\n// zero value if it is unset.\nfunc (v *Memo) GetFields() (o map[string][]byte) {\n\tif v != nil && v.Fields != nil {\n\t\treturn v.Fields\n\t}\n\n\treturn\n}\n\n// IsSetFields returns true if Fields is not nil.\nfunc (v *Memo) IsSetFields() bool {\n\treturn v != nil && v.Fields != nil\n}\n\ntype PaginationOptions struct {\n\tPageSize      *int32 `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// ToWire translates a PaginationOptions struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PaginationOptions) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PageSize != nil {\n\t\tw, err = wire.NewValueI32(*(v.PageSize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PaginationOptions struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PaginationOptions struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PaginationOptions\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PaginationOptions) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.PageSize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PaginationOptions struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PaginationOptions struct could not be encoded.\nfunc (v *PaginationOptions) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PageSize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.PageSize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PaginationOptions struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PaginationOptions struct could not be generated from the wire\n// representation.\nfunc (v *PaginationOptions) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.PageSize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PaginationOptions\n// struct.\nfunc (v *PaginationOptions) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.PageSize != nil {\n\t\tfields[i] = fmt.Sprintf(\"PageSize: %v\", *(v.PageSize))\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PaginationOptions{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PaginationOptions match the\n// provided PaginationOptions.\n//\n// This function performs a deep comparison.\nfunc (v *PaginationOptions) Equals(rhs *PaginationOptions) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.PageSize, rhs.PageSize) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PaginationOptions.\nfunc (v *PaginationOptions) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PageSize != nil {\n\t\tenc.AddInt32(\"pageSize\", *v.PageSize)\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\treturn err\n}\n\n// GetPageSize returns the value of PageSize if it is set or its\n// zero value if it is unset.\nfunc (v *PaginationOptions) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\n\treturn\n}\n\n// IsSetPageSize returns true if PageSize is not nil.\nfunc (v *PaginationOptions) IsSetPageSize() bool {\n\treturn v != nil && v.PageSize != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *PaginationOptions) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *PaginationOptions) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\ntype ParentClosePolicy int32\n\nconst (\n\tParentClosePolicyAbandon       ParentClosePolicy = 0\n\tParentClosePolicyRequestCancel ParentClosePolicy = 1\n\tParentClosePolicyTerminate     ParentClosePolicy = 2\n)\n\n// ParentClosePolicy_Values returns all recognized values of ParentClosePolicy.\nfunc ParentClosePolicy_Values() []ParentClosePolicy {\n\treturn []ParentClosePolicy{\n\t\tParentClosePolicyAbandon,\n\t\tParentClosePolicyRequestCancel,\n\t\tParentClosePolicyTerminate,\n\t}\n}\n\n// UnmarshalText tries to decode ParentClosePolicy from a byte slice\n// containing its name.\n//\n//\tvar v ParentClosePolicy\n//\terr := v.UnmarshalText([]byte(\"ABANDON\"))\nfunc (v *ParentClosePolicy) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"ABANDON\":\n\t\t*v = ParentClosePolicyAbandon\n\t\treturn nil\n\tcase \"REQUEST_CANCEL\":\n\t\t*v = ParentClosePolicyRequestCancel\n\t\treturn nil\n\tcase \"TERMINATE\":\n\t\t*v = ParentClosePolicyTerminate\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ParentClosePolicy\", err)\n\t\t}\n\t\t*v = ParentClosePolicy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ParentClosePolicy to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v ParentClosePolicy) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"ABANDON\"), nil\n\tcase 1:\n\t\treturn []byte(\"REQUEST_CANCEL\"), nil\n\tcase 2:\n\t\treturn []byte(\"TERMINATE\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ParentClosePolicy.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v ParentClosePolicy) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"ABANDON\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"REQUEST_CANCEL\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"TERMINATE\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v ParentClosePolicy) Ptr() *ParentClosePolicy {\n\treturn &v\n}\n\n// Encode encodes ParentClosePolicy directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v ParentClosePolicy\n//\treturn v.Encode(sWriter)\nfunc (v ParentClosePolicy) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates ParentClosePolicy into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v ParentClosePolicy) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes ParentClosePolicy from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return ParentClosePolicy(0), err\n//\t}\n//\n//\tvar v ParentClosePolicy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return ParentClosePolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ParentClosePolicy) FromWire(w wire.Value) error {\n\t*v = (ParentClosePolicy)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded ParentClosePolicy directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v ParentClosePolicy\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return ParentClosePolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *ParentClosePolicy) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (ParentClosePolicy)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of ParentClosePolicy.\nfunc (v ParentClosePolicy) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ABANDON\"\n\tcase 1:\n\t\treturn \"REQUEST_CANCEL\"\n\tcase 2:\n\t\treturn \"TERMINATE\"\n\t}\n\treturn fmt.Sprintf(\"ParentClosePolicy(%d)\", w)\n}\n\n// Equals returns true if this ParentClosePolicy value matches the provided\n// value.\nfunc (v ParentClosePolicy) Equals(rhs ParentClosePolicy) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes ParentClosePolicy into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v ParentClosePolicy) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"ABANDON\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"REQUEST_CANCEL\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"TERMINATE\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode ParentClosePolicy from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *ParentClosePolicy) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"ParentClosePolicy\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"ParentClosePolicy\")\n\t\t}\n\t\t*v = (ParentClosePolicy)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"ParentClosePolicy\")\n\t}\n}\n\ntype PendingActivityInfo struct {\n\tActivityID             *string               `json:\"activityID,omitempty\"`\n\tActivityType           *ActivityType         `json:\"activityType,omitempty\"`\n\tState                  *PendingActivityState `json:\"state,omitempty\"`\n\tHeartbeatDetails       []byte                `json:\"heartbeatDetails,omitempty\"`\n\tLastHeartbeatTimestamp *int64                `json:\"lastHeartbeatTimestamp,omitempty\"`\n\tLastStartedTimestamp   *int64                `json:\"lastStartedTimestamp,omitempty\"`\n\tAttempt                *int32                `json:\"attempt,omitempty\"`\n\tMaximumAttempts        *int32                `json:\"maximumAttempts,omitempty\"`\n\tScheduledTimestamp     *int64                `json:\"scheduledTimestamp,omitempty\"`\n\tExpirationTimestamp    *int64                `json:\"expirationTimestamp,omitempty\"`\n\tLastFailureReason      *string               `json:\"lastFailureReason,omitempty\"`\n\tLastWorkerIdentity     *string               `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails     []byte                `json:\"lastFailureDetails,omitempty\"`\n\tStartedWorkerIdentity  *string               `json:\"startedWorkerIdentity,omitempty\"`\n\tScheduleID             *int64                `json:\"scheduleID,omitempty\"`\n}\n\n// ToWire translates a PendingActivityInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PendingActivityInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tw, err = v.ActivityType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tw, err = v.State.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.HeartbeatDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.LastHeartbeatTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastHeartbeatTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.LastStartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastStartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumAttempts)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpirationTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.LastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.LastWorkerIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.LastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.StartedWorkerIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.StartedWorkerIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PendingActivityState_Read(w wire.Value) (PendingActivityState, error) {\n\tvar v PendingActivityState\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a PendingActivityInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PendingActivityInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PendingActivityInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PendingActivityInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityType, err = _ActivityType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x PendingActivityState\n\t\t\t\tx, err = _PendingActivityState_Read(field.Value)\n\t\t\t\tv.State = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.HeartbeatDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastHeartbeatTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastStartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumAttempts = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpirationTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.LastWorkerIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartedWorkerIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PendingActivityInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PendingActivityInfo struct could not be encoded.\nfunc (v *PendingActivityInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.State != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.State.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.HeartbeatDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastHeartbeatTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastHeartbeatTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastStartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastStartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumAttempts != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumAttempts)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpirationTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpirationTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWorkerIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.LastWorkerIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedWorkerIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartedWorkerIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PendingActivityState_Decode(sr stream.Reader) (PendingActivityState, error) {\n\tvar v PendingActivityState\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a PendingActivityInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PendingActivityInfo struct could not be generated from the wire\n// representation.\nfunc (v *PendingActivityInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityType, err = _ActivityType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x PendingActivityState\n\t\t\tx, err = _PendingActivityState_Decode(sr)\n\t\t\tv.State = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.HeartbeatDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastHeartbeatTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastStartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumAttempts = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpirationTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.LastWorkerIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tv.LastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartedWorkerIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PendingActivityInfo\n// struct.\nfunc (v *PendingActivityInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityType: %v\", v.ActivityType)\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tfields[i] = fmt.Sprintf(\"State: %v\", *(v.State))\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatDetails: %v\", v.HeartbeatDetails)\n\t\ti++\n\t}\n\tif v.LastHeartbeatTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastHeartbeatTimestamp: %v\", *(v.LastHeartbeatTimestamp))\n\t\ti++\n\t}\n\tif v.LastStartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastStartedTimestamp: %v\", *(v.LastStartedTimestamp))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumAttempts: %v\", *(v.MaximumAttempts))\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpirationTimestamp: %v\", *(v.ExpirationTimestamp))\n\t\ti++\n\t}\n\tif v.LastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureReason: %v\", *(v.LastFailureReason))\n\t\ti++\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWorkerIdentity: %v\", *(v.LastWorkerIdentity))\n\t\ti++\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFailureDetails: %v\", v.LastFailureDetails)\n\t\ti++\n\t}\n\tif v.StartedWorkerIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedWorkerIdentity: %v\", *(v.StartedWorkerIdentity))\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleID: %v\", *(v.ScheduleID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PendingActivityInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _PendingActivityState_EqualsPtr(lhs, rhs *PendingActivityState) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this PendingActivityInfo match the\n// provided PendingActivityInfo.\n//\n// This function performs a deep comparison.\nfunc (v *PendingActivityInfo) Equals(rhs *PendingActivityInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !((v.ActivityType == nil && rhs.ActivityType == nil) || (v.ActivityType != nil && rhs.ActivityType != nil && v.ActivityType.Equals(rhs.ActivityType))) {\n\t\treturn false\n\t}\n\tif !_PendingActivityState_EqualsPtr(v.State, rhs.State) {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatDetails == nil && rhs.HeartbeatDetails == nil) || (v.HeartbeatDetails != nil && rhs.HeartbeatDetails != nil && bytes.Equal(v.HeartbeatDetails, rhs.HeartbeatDetails))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastHeartbeatTimestamp, rhs.LastHeartbeatTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastStartedTimestamp, rhs.LastStartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumAttempts, rhs.MaximumAttempts) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpirationTimestamp, rhs.ExpirationTimestamp) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastFailureReason, rhs.LastFailureReason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.LastWorkerIdentity, rhs.LastWorkerIdentity) {\n\t\treturn false\n\t}\n\tif !((v.LastFailureDetails == nil && rhs.LastFailureDetails == nil) || (v.LastFailureDetails != nil && rhs.LastFailureDetails != nil && bytes.Equal(v.LastFailureDetails, rhs.LastFailureDetails))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartedWorkerIdentity, rhs.StartedWorkerIdentity) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleID, rhs.ScheduleID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PendingActivityInfo.\nfunc (v *PendingActivityInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.ActivityType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityType\", v.ActivityType))\n\t}\n\tif v.State != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"state\", *v.State))\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tenc.AddString(\"heartbeatDetails\", base64.StdEncoding.EncodeToString(v.HeartbeatDetails))\n\t}\n\tif v.LastHeartbeatTimestamp != nil {\n\t\tenc.AddInt64(\"lastHeartbeatTimestamp\", *v.LastHeartbeatTimestamp)\n\t}\n\tif v.LastStartedTimestamp != nil {\n\t\tenc.AddInt64(\"lastStartedTimestamp\", *v.LastStartedTimestamp)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tenc.AddInt32(\"maximumAttempts\", *v.MaximumAttempts)\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tenc.AddInt64(\"expirationTimestamp\", *v.ExpirationTimestamp)\n\t}\n\tif v.LastFailureReason != nil {\n\t\tenc.AddString(\"lastFailureReason\", *v.LastFailureReason)\n\t}\n\tif v.LastWorkerIdentity != nil {\n\t\tenc.AddString(\"lastWorkerIdentity\", *v.LastWorkerIdentity)\n\t}\n\tif v.LastFailureDetails != nil {\n\t\tenc.AddString(\"lastFailureDetails\", base64.StdEncoding.EncodeToString(v.LastFailureDetails))\n\t}\n\tif v.StartedWorkerIdentity != nil {\n\t\tenc.AddString(\"startedWorkerIdentity\", *v.StartedWorkerIdentity)\n\t}\n\tif v.ScheduleID != nil {\n\t\tenc.AddInt64(\"scheduleID\", *v.ScheduleID)\n\t}\n\treturn err\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *PendingActivityInfo) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetActivityType returns the value of ActivityType if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\n\treturn\n}\n\n// IsSetActivityType returns true if ActivityType is not nil.\nfunc (v *PendingActivityInfo) IsSetActivityType() bool {\n\treturn v != nil && v.ActivityType != nil\n}\n\n// GetState returns the value of State if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetState() (o PendingActivityState) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\n\treturn\n}\n\n// IsSetState returns true if State is not nil.\nfunc (v *PendingActivityInfo) IsSetState() bool {\n\treturn v != nil && v.State != nil\n}\n\n// GetHeartbeatDetails returns the value of HeartbeatDetails if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetHeartbeatDetails() (o []byte) {\n\tif v != nil && v.HeartbeatDetails != nil {\n\t\treturn v.HeartbeatDetails\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatDetails returns true if HeartbeatDetails is not nil.\nfunc (v *PendingActivityInfo) IsSetHeartbeatDetails() bool {\n\treturn v != nil && v.HeartbeatDetails != nil\n}\n\n// GetLastHeartbeatTimestamp returns the value of LastHeartbeatTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetLastHeartbeatTimestamp() (o int64) {\n\tif v != nil && v.LastHeartbeatTimestamp != nil {\n\t\treturn *v.LastHeartbeatTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetLastHeartbeatTimestamp returns true if LastHeartbeatTimestamp is not nil.\nfunc (v *PendingActivityInfo) IsSetLastHeartbeatTimestamp() bool {\n\treturn v != nil && v.LastHeartbeatTimestamp != nil\n}\n\n// GetLastStartedTimestamp returns the value of LastStartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetLastStartedTimestamp() (o int64) {\n\tif v != nil && v.LastStartedTimestamp != nil {\n\t\treturn *v.LastStartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetLastStartedTimestamp returns true if LastStartedTimestamp is not nil.\nfunc (v *PendingActivityInfo) IsSetLastStartedTimestamp() bool {\n\treturn v != nil && v.LastStartedTimestamp != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *PendingActivityInfo) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetMaximumAttempts returns the value of MaximumAttempts if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetMaximumAttempts() (o int32) {\n\tif v != nil && v.MaximumAttempts != nil {\n\t\treturn *v.MaximumAttempts\n\t}\n\n\treturn\n}\n\n// IsSetMaximumAttempts returns true if MaximumAttempts is not nil.\nfunc (v *PendingActivityInfo) IsSetMaximumAttempts() bool {\n\treturn v != nil && v.MaximumAttempts != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *PendingActivityInfo) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetExpirationTimestamp returns the value of ExpirationTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetExpirationTimestamp() (o int64) {\n\tif v != nil && v.ExpirationTimestamp != nil {\n\t\treturn *v.ExpirationTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetExpirationTimestamp returns true if ExpirationTimestamp is not nil.\nfunc (v *PendingActivityInfo) IsSetExpirationTimestamp() bool {\n\treturn v != nil && v.ExpirationTimestamp != nil\n}\n\n// GetLastFailureReason returns the value of LastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureReason returns true if LastFailureReason is not nil.\nfunc (v *PendingActivityInfo) IsSetLastFailureReason() bool {\n\treturn v != nil && v.LastFailureReason != nil\n}\n\n// GetLastWorkerIdentity returns the value of LastWorkerIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetLastWorkerIdentity() (o string) {\n\tif v != nil && v.LastWorkerIdentity != nil {\n\t\treturn *v.LastWorkerIdentity\n\t}\n\n\treturn\n}\n\n// IsSetLastWorkerIdentity returns true if LastWorkerIdentity is not nil.\nfunc (v *PendingActivityInfo) IsSetLastWorkerIdentity() bool {\n\treturn v != nil && v.LastWorkerIdentity != nil\n}\n\n// GetLastFailureDetails returns the value of LastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetLastFailureDetails returns true if LastFailureDetails is not nil.\nfunc (v *PendingActivityInfo) IsSetLastFailureDetails() bool {\n\treturn v != nil && v.LastFailureDetails != nil\n}\n\n// GetStartedWorkerIdentity returns the value of StartedWorkerIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetStartedWorkerIdentity() (o string) {\n\tif v != nil && v.StartedWorkerIdentity != nil {\n\t\treturn *v.StartedWorkerIdentity\n\t}\n\n\treturn\n}\n\n// IsSetStartedWorkerIdentity returns true if StartedWorkerIdentity is not nil.\nfunc (v *PendingActivityInfo) IsSetStartedWorkerIdentity() bool {\n\treturn v != nil && v.StartedWorkerIdentity != nil\n}\n\n// GetScheduleID returns the value of ScheduleID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingActivityInfo) GetScheduleID() (o int64) {\n\tif v != nil && v.ScheduleID != nil {\n\t\treturn *v.ScheduleID\n\t}\n\n\treturn\n}\n\n// IsSetScheduleID returns true if ScheduleID is not nil.\nfunc (v *PendingActivityInfo) IsSetScheduleID() bool {\n\treturn v != nil && v.ScheduleID != nil\n}\n\ntype PendingActivityState int32\n\nconst (\n\tPendingActivityStateScheduled       PendingActivityState = 0\n\tPendingActivityStateStarted         PendingActivityState = 1\n\tPendingActivityStateCancelRequested PendingActivityState = 2\n)\n\n// PendingActivityState_Values returns all recognized values of PendingActivityState.\nfunc PendingActivityState_Values() []PendingActivityState {\n\treturn []PendingActivityState{\n\t\tPendingActivityStateScheduled,\n\t\tPendingActivityStateStarted,\n\t\tPendingActivityStateCancelRequested,\n\t}\n}\n\n// UnmarshalText tries to decode PendingActivityState from a byte slice\n// containing its name.\n//\n//\tvar v PendingActivityState\n//\terr := v.UnmarshalText([]byte(\"SCHEDULED\"))\nfunc (v *PendingActivityState) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"SCHEDULED\":\n\t\t*v = PendingActivityStateScheduled\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*v = PendingActivityStateStarted\n\t\treturn nil\n\tcase \"CANCEL_REQUESTED\":\n\t\t*v = PendingActivityStateCancelRequested\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"PendingActivityState\", err)\n\t\t}\n\t\t*v = PendingActivityState(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes PendingActivityState to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v PendingActivityState) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"SCHEDULED\"), nil\n\tcase 1:\n\t\treturn []byte(\"STARTED\"), nil\n\tcase 2:\n\t\treturn []byte(\"CANCEL_REQUESTED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PendingActivityState.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v PendingActivityState) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"SCHEDULED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"STARTED\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"CANCEL_REQUESTED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v PendingActivityState) Ptr() *PendingActivityState {\n\treturn &v\n}\n\n// Encode encodes PendingActivityState directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v PendingActivityState\n//\treturn v.Encode(sWriter)\nfunc (v PendingActivityState) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates PendingActivityState into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v PendingActivityState) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes PendingActivityState from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return PendingActivityState(0), err\n//\t}\n//\n//\tvar v PendingActivityState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return PendingActivityState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PendingActivityState) FromWire(w wire.Value) error {\n\t*v = (PendingActivityState)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded PendingActivityState directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v PendingActivityState\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return PendingActivityState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PendingActivityState) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (PendingActivityState)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of PendingActivityState.\nfunc (v PendingActivityState) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SCHEDULED\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\tcase 2:\n\t\treturn \"CANCEL_REQUESTED\"\n\t}\n\treturn fmt.Sprintf(\"PendingActivityState(%d)\", w)\n}\n\n// Equals returns true if this PendingActivityState value matches the provided\n// value.\nfunc (v PendingActivityState) Equals(rhs PendingActivityState) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes PendingActivityState into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v PendingActivityState) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"SCHEDULED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"STARTED\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"CANCEL_REQUESTED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode PendingActivityState from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *PendingActivityState) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"PendingActivityState\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"PendingActivityState\")\n\t\t}\n\t\t*v = (PendingActivityState)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"PendingActivityState\")\n\t}\n}\n\ntype PendingChildExecutionInfo struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowID        *string            `json:\"workflowID,omitempty\"`\n\tRunID             *string            `json:\"runID,omitempty\"`\n\tWorkflowTypName   *string            `json:\"workflowTypName,omitempty\"`\n\tInitiatedID       *int64             `json:\"initiatedID,omitempty\"`\n\tParentClosePolicy *ParentClosePolicy `json:\"parentClosePolicy,omitempty\"`\n}\n\n// ToWire translates a PendingChildExecutionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PendingChildExecutionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 1, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowTypName != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowTypName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tw, err = v.ParentClosePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PendingChildExecutionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PendingChildExecutionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PendingChildExecutionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PendingChildExecutionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowTypName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ParentClosePolicy\n\t\t\t\tx, err = _ParentClosePolicy_Read(field.Value)\n\t\t\t\tv.ParentClosePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PendingChildExecutionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PendingChildExecutionInfo struct could not be encoded.\nfunc (v *PendingChildExecutionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowTypName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowTypName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentClosePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentClosePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PendingChildExecutionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PendingChildExecutionInfo struct could not be generated from the wire\n// representation.\nfunc (v *PendingChildExecutionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowTypName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x ParentClosePolicy\n\t\t\tx, err = _ParentClosePolicy_Decode(sr)\n\t\t\tv.ParentClosePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PendingChildExecutionInfo\n// struct.\nfunc (v *PendingChildExecutionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.WorkflowTypName != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowTypName: %v\", *(v.WorkflowTypName))\n\t\ti++\n\t}\n\tif v.InitiatedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedID: %v\", *(v.InitiatedID))\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentClosePolicy: %v\", *(v.ParentClosePolicy))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PendingChildExecutionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PendingChildExecutionInfo match the\n// provided PendingChildExecutionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *PendingChildExecutionInfo) Equals(rhs *PendingChildExecutionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowTypName, rhs.WorkflowTypName) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedID, rhs.InitiatedID) {\n\t\treturn false\n\t}\n\tif !_ParentClosePolicy_EqualsPtr(v.ParentClosePolicy, rhs.ParentClosePolicy) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PendingChildExecutionInfo.\nfunc (v *PendingChildExecutionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.WorkflowTypName != nil {\n\t\tenc.AddString(\"workflowTypName\", *v.WorkflowTypName)\n\t}\n\tif v.InitiatedID != nil {\n\t\tenc.AddInt64(\"initiatedID\", *v.InitiatedID)\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentClosePolicy\", *v.ParentClosePolicy))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetWorkflowTypName returns the value of WorkflowTypName if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetWorkflowTypName() (o string) {\n\tif v != nil && v.WorkflowTypName != nil {\n\t\treturn *v.WorkflowTypName\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowTypName returns true if WorkflowTypName is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetWorkflowTypName() bool {\n\treturn v != nil && v.WorkflowTypName != nil\n}\n\n// GetInitiatedID returns the value of InitiatedID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetInitiatedID() (o int64) {\n\tif v != nil && v.InitiatedID != nil {\n\t\treturn *v.InitiatedID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedID returns true if InitiatedID is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetInitiatedID() bool {\n\treturn v != nil && v.InitiatedID != nil\n}\n\n// GetParentClosePolicy returns the value of ParentClosePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *PendingChildExecutionInfo) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\n\treturn\n}\n\n// IsSetParentClosePolicy returns true if ParentClosePolicy is not nil.\nfunc (v *PendingChildExecutionInfo) IsSetParentClosePolicy() bool {\n\treturn v != nil && v.ParentClosePolicy != nil\n}\n\ntype PendingDecisionInfo struct {\n\tState                      *PendingDecisionState `json:\"state,omitempty\"`\n\tScheduledTimestamp         *int64                `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp           *int64                `json:\"startedTimestamp,omitempty\"`\n\tAttempt                    *int64                `json:\"attempt,omitempty\"`\n\tOriginalScheduledTimestamp *int64                `json:\"originalScheduledTimestamp,omitempty\"`\n\tScheduleID                 *int64                `json:\"scheduleID,omitempty\"`\n}\n\n// ToWire translates a PendingDecisionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PendingDecisionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.State != nil {\n\t\tw, err = v.State.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.OriginalScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.OriginalScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PendingDecisionState_Read(w wire.Value) (PendingDecisionState, error) {\n\tvar v PendingDecisionState\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a PendingDecisionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PendingDecisionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PendingDecisionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PendingDecisionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x PendingDecisionState\n\t\t\t\tx, err = _PendingDecisionState_Read(field.Value)\n\t\t\t\tv.State = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.OriginalScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PendingDecisionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PendingDecisionInfo struct could not be encoded.\nfunc (v *PendingDecisionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.State != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.State.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OriginalScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.OriginalScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PendingDecisionState_Decode(sr stream.Reader) (PendingDecisionState, error) {\n\tvar v PendingDecisionState\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a PendingDecisionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PendingDecisionInfo struct could not be generated from the wire\n// representation.\nfunc (v *PendingDecisionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x PendingDecisionState\n\t\t\tx, err = _PendingDecisionState_Decode(sr)\n\t\t\tv.State = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.OriginalScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PendingDecisionInfo\n// struct.\nfunc (v *PendingDecisionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.State != nil {\n\t\tfields[i] = fmt.Sprintf(\"State: %v\", *(v.State))\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.OriginalScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"OriginalScheduledTimestamp: %v\", *(v.OriginalScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleID: %v\", *(v.ScheduleID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PendingDecisionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _PendingDecisionState_EqualsPtr(lhs, rhs *PendingDecisionState) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this PendingDecisionInfo match the\n// provided PendingDecisionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *PendingDecisionInfo) Equals(rhs *PendingDecisionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_PendingDecisionState_EqualsPtr(v.State, rhs.State) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.OriginalScheduledTimestamp, rhs.OriginalScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleID, rhs.ScheduleID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PendingDecisionInfo.\nfunc (v *PendingDecisionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.State != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"state\", *v.State))\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.OriginalScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"originalScheduledTimestamp\", *v.OriginalScheduledTimestamp)\n\t}\n\tif v.ScheduleID != nil {\n\t\tenc.AddInt64(\"scheduleID\", *v.ScheduleID)\n\t}\n\treturn err\n}\n\n// GetState returns the value of State if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetState() (o PendingDecisionState) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\n\treturn\n}\n\n// IsSetState returns true if State is not nil.\nfunc (v *PendingDecisionInfo) IsSetState() bool {\n\treturn v != nil && v.State != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *PendingDecisionInfo) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *PendingDecisionInfo) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *PendingDecisionInfo) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetOriginalScheduledTimestamp returns the value of OriginalScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetOriginalScheduledTimestamp() (o int64) {\n\tif v != nil && v.OriginalScheduledTimestamp != nil {\n\t\treturn *v.OriginalScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetOriginalScheduledTimestamp returns true if OriginalScheduledTimestamp is not nil.\nfunc (v *PendingDecisionInfo) IsSetOriginalScheduledTimestamp() bool {\n\treturn v != nil && v.OriginalScheduledTimestamp != nil\n}\n\n// GetScheduleID returns the value of ScheduleID if it is set or its\n// zero value if it is unset.\nfunc (v *PendingDecisionInfo) GetScheduleID() (o int64) {\n\tif v != nil && v.ScheduleID != nil {\n\t\treturn *v.ScheduleID\n\t}\n\n\treturn\n}\n\n// IsSetScheduleID returns true if ScheduleID is not nil.\nfunc (v *PendingDecisionInfo) IsSetScheduleID() bool {\n\treturn v != nil && v.ScheduleID != nil\n}\n\ntype PendingDecisionState int32\n\nconst (\n\tPendingDecisionStateScheduled PendingDecisionState = 0\n\tPendingDecisionStateStarted   PendingDecisionState = 1\n)\n\n// PendingDecisionState_Values returns all recognized values of PendingDecisionState.\nfunc PendingDecisionState_Values() []PendingDecisionState {\n\treturn []PendingDecisionState{\n\t\tPendingDecisionStateScheduled,\n\t\tPendingDecisionStateStarted,\n\t}\n}\n\n// UnmarshalText tries to decode PendingDecisionState from a byte slice\n// containing its name.\n//\n//\tvar v PendingDecisionState\n//\terr := v.UnmarshalText([]byte(\"SCHEDULED\"))\nfunc (v *PendingDecisionState) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"SCHEDULED\":\n\t\t*v = PendingDecisionStateScheduled\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*v = PendingDecisionStateStarted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"PendingDecisionState\", err)\n\t\t}\n\t\t*v = PendingDecisionState(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes PendingDecisionState to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v PendingDecisionState) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"SCHEDULED\"), nil\n\tcase 1:\n\t\treturn []byte(\"STARTED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PendingDecisionState.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v PendingDecisionState) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"SCHEDULED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"STARTED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v PendingDecisionState) Ptr() *PendingDecisionState {\n\treturn &v\n}\n\n// Encode encodes PendingDecisionState directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v PendingDecisionState\n//\treturn v.Encode(sWriter)\nfunc (v PendingDecisionState) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates PendingDecisionState into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v PendingDecisionState) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes PendingDecisionState from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return PendingDecisionState(0), err\n//\t}\n//\n//\tvar v PendingDecisionState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return PendingDecisionState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PendingDecisionState) FromWire(w wire.Value) error {\n\t*v = (PendingDecisionState)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded PendingDecisionState directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v PendingDecisionState\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return PendingDecisionState(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PendingDecisionState) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (PendingDecisionState)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of PendingDecisionState.\nfunc (v PendingDecisionState) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SCHEDULED\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\t}\n\treturn fmt.Sprintf(\"PendingDecisionState(%d)\", w)\n}\n\n// Equals returns true if this PendingDecisionState value matches the provided\n// value.\nfunc (v PendingDecisionState) Equals(rhs PendingDecisionState) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes PendingDecisionState into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v PendingDecisionState) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"SCHEDULED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"STARTED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode PendingDecisionState from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *PendingDecisionState) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"PendingDecisionState\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"PendingDecisionState\")\n\t\t}\n\t\t*v = (PendingDecisionState)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"PendingDecisionState\")\n\t}\n}\n\ntype PollForActivityTaskRequest struct {\n\tDomain           *string           `json:\"domain,omitempty\"`\n\tTaskList         *TaskList         `json:\"taskList,omitempty\"`\n\tIdentity         *string           `json:\"identity,omitempty\"`\n\tTaskListMetadata *TaskListMetadata `json:\"taskListMetadata,omitempty\"`\n}\n\n// ToWire translates a PollForActivityTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForActivityTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListMetadata != nil {\n\t\tw, err = v.TaskListMetadata.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListMetadata_Read(w wire.Value) (*TaskListMetadata, error) {\n\tvar v TaskListMetadata\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a PollForActivityTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForActivityTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForActivityTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForActivityTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListMetadata, err = _TaskListMetadata_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollForActivityTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForActivityTaskRequest struct could not be encoded.\nfunc (v *PollForActivityTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListMetadata != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListMetadata.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListMetadata_Decode(sr stream.Reader) (*TaskListMetadata, error) {\n\tvar v TaskListMetadata\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a PollForActivityTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForActivityTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *PollForActivityTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListMetadata, err = _TaskListMetadata_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForActivityTaskRequest\n// struct.\nfunc (v *PollForActivityTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.TaskListMetadata != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListMetadata: %v\", v.TaskListMetadata)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForActivityTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollForActivityTaskRequest match the\n// provided PollForActivityTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PollForActivityTaskRequest) Equals(rhs *PollForActivityTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !((v.TaskListMetadata == nil && rhs.TaskListMetadata == nil) || (v.TaskListMetadata != nil && rhs.TaskListMetadata != nil && v.TaskListMetadata.Equals(rhs.TaskListMetadata))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForActivityTaskRequest.\nfunc (v *PollForActivityTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.TaskListMetadata != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListMetadata\", v.TaskListMetadata))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetTaskListMetadata returns the value of TaskListMetadata if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskRequest) GetTaskListMetadata() (o *TaskListMetadata) {\n\tif v != nil && v.TaskListMetadata != nil {\n\t\treturn v.TaskListMetadata\n\t}\n\n\treturn\n}\n\n// IsSetTaskListMetadata returns true if TaskListMetadata is not nil.\nfunc (v *PollForActivityTaskRequest) IsSetTaskListMetadata() bool {\n\treturn v != nil && v.TaskListMetadata != nil\n}\n\ntype PollForActivityTaskResponse struct {\n\tTaskToken                       []byte             `json:\"taskToken,omitempty\"`\n\tWorkflowExecution               *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tActivityId                      *string            `json:\"activityId,omitempty\"`\n\tActivityType                    *ActivityType      `json:\"activityType,omitempty\"`\n\tInput                           []byte             `json:\"input,omitempty\"`\n\tScheduledTimestamp              *int64             `json:\"scheduledTimestamp,omitempty\"`\n\tScheduleToCloseTimeoutSeconds   *int32             `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tStartedTimestamp                *int64             `json:\"startedTimestamp,omitempty\"`\n\tStartToCloseTimeoutSeconds      *int32             `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds         *int32             `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tAttempt                         *int32             `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64             `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte             `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *WorkflowType      `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  *string            `json:\"workflowDomain,omitempty\"`\n\tHeader                          *Header            `json:\"header,omitempty\"`\n\tAutoConfigHint                  *AutoConfigHint    `json:\"autoConfigHint,omitempty\"`\n}\n\n// ToWire translates a PollForActivityTaskResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForActivityTaskResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [17]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tw, err = v.ActivityType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.StartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.HeartbeatTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestampOfThisAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.HeartbeatDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tw, err = v.AutoConfigHint.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AutoConfigHint_Read(w wire.Value) (*AutoConfigHint, error) {\n\tvar v AutoConfigHint\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a PollForActivityTaskResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForActivityTaskResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForActivityTaskResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForActivityTaskResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityType, err = _ActivityType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.HeartbeatDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollForActivityTaskResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForActivityTaskResponse struct could not be encoded.\nfunc (v *PollForActivityTaskResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.HeartbeatTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestampOfThisAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.HeartbeatDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoConfigHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AutoConfigHint.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AutoConfigHint_Decode(sr stream.Reader) (*AutoConfigHint, error) {\n\tvar v AutoConfigHint\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a PollForActivityTaskResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForActivityTaskResponse struct could not be generated from the wire\n// representation.\nfunc (v *PollForActivityTaskResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityType, err = _ActivityType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestampOfThisAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TBinary:\n\t\t\tv.HeartbeatDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TStruct:\n\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForActivityTaskResponse\n// struct.\nfunc (v *PollForActivityTaskResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [17]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityType: %v\", v.ActivityType)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToCloseTimeoutSeconds: %v\", *(v.ScheduleToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToCloseTimeoutSeconds: %v\", *(v.StartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatTimeoutSeconds: %v\", *(v.HeartbeatTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestampOfThisAttempt: %v\", *(v.ScheduledTimestampOfThisAttempt))\n\t\ti++\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatDetails: %v\", v.HeartbeatDetails)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowDomain: %v\", *(v.WorkflowDomain))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoConfigHint: %v\", v.AutoConfigHint)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForActivityTaskResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollForActivityTaskResponse match the\n// provided PollForActivityTaskResponse.\n//\n// This function performs a deep comparison.\nfunc (v *PollForActivityTaskResponse) Equals(rhs *PollForActivityTaskResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !((v.ActivityType == nil && rhs.ActivityType == nil) || (v.ActivityType != nil && rhs.ActivityType != nil && v.ActivityType.Equals(rhs.ActivityType))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToCloseTimeoutSeconds, rhs.ScheduleToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StartToCloseTimeoutSeconds, rhs.StartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.HeartbeatTimeoutSeconds, rhs.HeartbeatTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestampOfThisAttempt, rhs.ScheduledTimestampOfThisAttempt) {\n\t\treturn false\n\t}\n\tif !((v.HeartbeatDetails == nil && rhs.HeartbeatDetails == nil) || (v.HeartbeatDetails != nil && rhs.HeartbeatDetails != nil && bytes.Equal(v.HeartbeatDetails, rhs.HeartbeatDetails))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowDomain, rhs.WorkflowDomain) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.AutoConfigHint == nil && rhs.AutoConfigHint == nil) || (v.AutoConfigHint != nil && rhs.AutoConfigHint != nil && v.AutoConfigHint.Equals(rhs.AutoConfigHint))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForActivityTaskResponse.\nfunc (v *PollForActivityTaskResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.ActivityType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityType\", v.ActivityType))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToCloseTimeoutSeconds\", *v.ScheduleToCloseTimeoutSeconds)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"startToCloseTimeoutSeconds\", *v.StartToCloseTimeoutSeconds)\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"heartbeatTimeoutSeconds\", *v.HeartbeatTimeoutSeconds)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.ScheduledTimestampOfThisAttempt != nil {\n\t\tenc.AddInt64(\"scheduledTimestampOfThisAttempt\", *v.ScheduledTimestampOfThisAttempt)\n\t}\n\tif v.HeartbeatDetails != nil {\n\t\tenc.AddString(\"heartbeatDetails\", base64.StdEncoding.EncodeToString(v.HeartbeatDetails))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.WorkflowDomain != nil {\n\t\tenc.AddString(\"workflowDomain\", *v.WorkflowDomain)\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.AutoConfigHint != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"autoConfigHint\", v.AutoConfigHint))\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetActivityType returns the value of ActivityType if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\n\treturn\n}\n\n// IsSetActivityType returns true if ActivityType is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetActivityType() bool {\n\treturn v != nil && v.ActivityType != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetScheduleToCloseTimeoutSeconds returns the value of ScheduleToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToCloseTimeoutSeconds returns true if ScheduleToCloseTimeoutSeconds is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetScheduleToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToCloseTimeoutSeconds != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetStartToCloseTimeoutSeconds returns the value of StartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToCloseTimeoutSeconds returns true if StartToCloseTimeoutSeconds is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.StartToCloseTimeoutSeconds != nil\n}\n\n// GetHeartbeatTimeoutSeconds returns the value of HeartbeatTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatTimeoutSeconds returns true if HeartbeatTimeoutSeconds is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetHeartbeatTimeoutSeconds() bool {\n\treturn v != nil && v.HeartbeatTimeoutSeconds != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetScheduledTimestampOfThisAttempt returns the value of ScheduledTimestampOfThisAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetScheduledTimestampOfThisAttempt() (o int64) {\n\tif v != nil && v.ScheduledTimestampOfThisAttempt != nil {\n\t\treturn *v.ScheduledTimestampOfThisAttempt\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestampOfThisAttempt returns true if ScheduledTimestampOfThisAttempt is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetScheduledTimestampOfThisAttempt() bool {\n\treturn v != nil && v.ScheduledTimestampOfThisAttempt != nil\n}\n\n// GetHeartbeatDetails returns the value of HeartbeatDetails if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetHeartbeatDetails() (o []byte) {\n\tif v != nil && v.HeartbeatDetails != nil {\n\t\treturn v.HeartbeatDetails\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatDetails returns true if HeartbeatDetails is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetHeartbeatDetails() bool {\n\treturn v != nil && v.HeartbeatDetails != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetWorkflowDomain returns the value of WorkflowDomain if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetWorkflowDomain() (o string) {\n\tif v != nil && v.WorkflowDomain != nil {\n\t\treturn *v.WorkflowDomain\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowDomain returns true if WorkflowDomain is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetWorkflowDomain() bool {\n\treturn v != nil && v.WorkflowDomain != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetAutoConfigHint returns the value of AutoConfigHint if it is set or its\n// zero value if it is unset.\nfunc (v *PollForActivityTaskResponse) GetAutoConfigHint() (o *AutoConfigHint) {\n\tif v != nil && v.AutoConfigHint != nil {\n\t\treturn v.AutoConfigHint\n\t}\n\n\treturn\n}\n\n// IsSetAutoConfigHint returns true if AutoConfigHint is not nil.\nfunc (v *PollForActivityTaskResponse) IsSetAutoConfigHint() bool {\n\treturn v != nil && v.AutoConfigHint != nil\n}\n\ntype PollForDecisionTaskRequest struct {\n\tDomain         *string   `json:\"domain,omitempty\"`\n\tTaskList       *TaskList `json:\"taskList,omitempty\"`\n\tIdentity       *string   `json:\"identity,omitempty\"`\n\tBinaryChecksum *string   `json:\"binaryChecksum,omitempty\"`\n}\n\n// ToWire translates a PollForDecisionTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForDecisionTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PollForDecisionTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForDecisionTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForDecisionTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForDecisionTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollForDecisionTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForDecisionTaskRequest struct could not be encoded.\nfunc (v *PollForDecisionTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PollForDecisionTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForDecisionTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *PollForDecisionTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForDecisionTaskRequest\n// struct.\nfunc (v *PollForDecisionTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForDecisionTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollForDecisionTaskRequest match the\n// provided PollForDecisionTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *PollForDecisionTaskRequest) Equals(rhs *PollForDecisionTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForDecisionTaskRequest.\nfunc (v *PollForDecisionTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskRequest) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *PollForDecisionTaskRequest) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\ntype PollForDecisionTaskResponse struct {\n\tTaskToken                 []byte                    `json:\"taskToken,omitempty\"`\n\tWorkflowExecution         *WorkflowExecution        `json:\"workflowExecution,omitempty\"`\n\tWorkflowType              *WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventId    *int64                    `json:\"previousStartedEventId,omitempty\"`\n\tStartedEventId            *int64                    `json:\"startedEventId,omitempty\"`\n\tAttempt                   *int64                    `json:\"attempt,omitempty\"`\n\tBacklogCountHint          *int64                    `json:\"backlogCountHint,omitempty\"`\n\tHistory                   *History                  `json:\"history,omitempty\"`\n\tNextPageToken             []byte                    `json:\"nextPageToken,omitempty\"`\n\tQuery                     *WorkflowQuery            `json:\"query,omitempty\"`\n\tWorkflowExecutionTaskList *TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tScheduledTimestamp        *int64                    `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                    `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*WorkflowQuery `json:\"queries,omitempty\"`\n\tNextEventId               *int64                    `json:\"nextEventId,omitempty\"`\n\tTotalHistoryBytes         *int64                    `json:\"totalHistoryBytes,omitempty\"`\n\tAutoConfigHint            *AutoConfigHint           `json:\"autoConfigHint,omitempty\"`\n}\n\ntype _Map_String_WorkflowQuery_MapItemList map[string]*WorkflowQuery\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WorkflowQuery_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WorkflowQuery_MapItemList) Close() {}\n\n// ToWire translates a PollForDecisionTaskResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollForDecisionTaskResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [17]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousStartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 51, Value: w}\n\t\ti++\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tw, err = wire.NewValueI64(*(v.BacklogCountHint)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 54, Value: w}\n\t\ti++\n\t}\n\tif v.History != nil {\n\t\tw, err = v.History.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NextPageToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = v.Query.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tw, err = v.WorkflowExecutionTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_WorkflowQuery_MapItemList(v.Queries)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tw, err = wire.NewValueI64(*(v.TotalHistoryBytes)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tw, err = v.AutoConfigHint.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowQuery_Read(w wire.Value) (*WorkflowQuery, error) {\n\tvar v WorkflowQuery\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Read(m wire.MapItemList) (map[string]*WorkflowQuery, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*WorkflowQuery, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a PollForDecisionTaskResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollForDecisionTaskResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollForDecisionTaskResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollForDecisionTaskResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousStartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 51:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 54:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.BacklogCountHint = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.History, err = _History_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NextPageToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Query, err = _WorkflowQuery_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TotalHistoryBytes = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_WorkflowQuery_Encode(val map[string]*WorkflowQuery, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WorkflowQuery', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a PollForDecisionTaskResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollForDecisionTaskResponse struct could not be encoded.\nfunc (v *PollForDecisionTaskResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousStartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousStartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 51, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BacklogCountHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 54, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.BacklogCountHint)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.History != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.History.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextPageToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NextPageToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Query.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecutionTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Queries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_WorkflowQuery_Encode(v.Queries, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TotalHistoryBytes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TotalHistoryBytes)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoConfigHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AutoConfigHint.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowQuery_Decode(sr stream.Reader) (*WorkflowQuery, error) {\n\tvar v WorkflowQuery\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQuery_Decode(sr stream.Reader) (map[string]*WorkflowQuery, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*WorkflowQuery, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WorkflowQuery_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a PollForDecisionTaskResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollForDecisionTaskResponse struct could not be generated from the wire\n// representation.\nfunc (v *PollForDecisionTaskResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousStartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 51 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 54 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.BacklogCountHint = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TStruct:\n\t\t\tv.History, err = _History_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tv.NextPageToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.Query, err = _WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecutionTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TMap:\n\t\t\tv.Queries, err = _Map_String_WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TotalHistoryBytes = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.AutoConfigHint, err = _AutoConfigHint_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollForDecisionTaskResponse\n// struct.\nfunc (v *PollForDecisionTaskResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [17]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousStartedEventId: %v\", *(v.PreviousStartedEventId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"BacklogCountHint: %v\", *(v.BacklogCountHint))\n\t\ti++\n\t}\n\tif v.History != nil {\n\t\tfields[i] = fmt.Sprintf(\"History: %v\", v.History)\n\t\ti++\n\t}\n\tif v.NextPageToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextPageToken: %v\", v.NextPageToken)\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", v.Query)\n\t\ti++\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionTaskList: %v\", v.WorkflowExecutionTaskList)\n\t\ti++\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimestamp: %v\", *(v.ScheduledTimestamp))\n\t\ti++\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimestamp: %v\", *(v.StartedTimestamp))\n\t\ti++\n\t}\n\tif v.Queries != nil {\n\t\tfields[i] = fmt.Sprintf(\"Queries: %v\", v.Queries)\n\t\ti++\n\t}\n\tif v.NextEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventId: %v\", *(v.NextEventId))\n\t\ti++\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tfields[i] = fmt.Sprintf(\"TotalHistoryBytes: %v\", *(v.TotalHistoryBytes))\n\t\ti++\n\t}\n\tif v.AutoConfigHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoConfigHint: %v\", v.AutoConfigHint)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollForDecisionTaskResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_WorkflowQuery_Equals(lhs, rhs map[string]*WorkflowQuery) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this PollForDecisionTaskResponse match the\n// provided PollForDecisionTaskResponse.\n//\n// This function performs a deep comparison.\nfunc (v *PollForDecisionTaskResponse) Equals(rhs *PollForDecisionTaskResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousStartedEventId, rhs.PreviousStartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.BacklogCountHint, rhs.BacklogCountHint) {\n\t\treturn false\n\t}\n\tif !((v.History == nil && rhs.History == nil) || (v.History != nil && rhs.History != nil && v.History.Equals(rhs.History))) {\n\t\treturn false\n\t}\n\tif !((v.NextPageToken == nil && rhs.NextPageToken == nil) || (v.NextPageToken != nil && rhs.NextPageToken != nil && bytes.Equal(v.NextPageToken, rhs.NextPageToken))) {\n\t\treturn false\n\t}\n\tif !((v.Query == nil && rhs.Query == nil) || (v.Query != nil && rhs.Query != nil && v.Query.Equals(rhs.Query))) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecutionTaskList == nil && rhs.WorkflowExecutionTaskList == nil) || (v.WorkflowExecutionTaskList != nil && rhs.WorkflowExecutionTaskList != nil && v.WorkflowExecutionTaskList.Equals(rhs.WorkflowExecutionTaskList))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimestamp, rhs.ScheduledTimestamp) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimestamp, rhs.StartedTimestamp) {\n\t\treturn false\n\t}\n\tif !((v.Queries == nil && rhs.Queries == nil) || (v.Queries != nil && rhs.Queries != nil && _Map_String_WorkflowQuery_Equals(v.Queries, rhs.Queries))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventId, rhs.NextEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TotalHistoryBytes, rhs.TotalHistoryBytes) {\n\t\treturn false\n\t}\n\tif !((v.AutoConfigHint == nil && rhs.AutoConfigHint == nil) || (v.AutoConfigHint != nil && rhs.AutoConfigHint != nil && v.AutoConfigHint.Equals(rhs.AutoConfigHint))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_WorkflowQuery_Zapper map[string]*WorkflowQuery\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WorkflowQuery_Zapper.\nfunc (m _Map_String_WorkflowQuery_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollForDecisionTaskResponse.\nfunc (v *PollForDecisionTaskResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.PreviousStartedEventId != nil {\n\t\tenc.AddInt64(\"previousStartedEventId\", *v.PreviousStartedEventId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt64(\"attempt\", *v.Attempt)\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tenc.AddInt64(\"backlogCountHint\", *v.BacklogCountHint)\n\t}\n\tif v.History != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"history\", v.History))\n\t}\n\tif v.NextPageToken != nil {\n\t\tenc.AddString(\"nextPageToken\", base64.StdEncoding.EncodeToString(v.NextPageToken))\n\t}\n\tif v.Query != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"query\", v.Query))\n\t}\n\tif v.WorkflowExecutionTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"WorkflowExecutionTaskList\", v.WorkflowExecutionTaskList))\n\t}\n\tif v.ScheduledTimestamp != nil {\n\t\tenc.AddInt64(\"scheduledTimestamp\", *v.ScheduledTimestamp)\n\t}\n\tif v.StartedTimestamp != nil {\n\t\tenc.AddInt64(\"startedTimestamp\", *v.StartedTimestamp)\n\t}\n\tif v.Queries != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queries\", (_Map_String_WorkflowQuery_Zapper)(v.Queries)))\n\t}\n\tif v.NextEventId != nil {\n\t\tenc.AddInt64(\"nextEventId\", *v.NextEventId)\n\t}\n\tif v.TotalHistoryBytes != nil {\n\t\tenc.AddInt64(\"totalHistoryBytes\", *v.TotalHistoryBytes)\n\t}\n\tif v.AutoConfigHint != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"autoConfigHint\", v.AutoConfigHint))\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetPreviousStartedEventId returns the value of PreviousStartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetPreviousStartedEventId() (o int64) {\n\tif v != nil && v.PreviousStartedEventId != nil {\n\t\treturn *v.PreviousStartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetPreviousStartedEventId returns true if PreviousStartedEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetPreviousStartedEventId() bool {\n\treturn v != nil && v.PreviousStartedEventId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetAttempt() (o int64) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetBacklogCountHint returns the value of BacklogCountHint if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetBacklogCountHint() (o int64) {\n\tif v != nil && v.BacklogCountHint != nil {\n\t\treturn *v.BacklogCountHint\n\t}\n\n\treturn\n}\n\n// IsSetBacklogCountHint returns true if BacklogCountHint is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetBacklogCountHint() bool {\n\treturn v != nil && v.BacklogCountHint != nil\n}\n\n// GetHistory returns the value of History if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetHistory() (o *History) {\n\tif v != nil && v.History != nil {\n\t\treturn v.History\n\t}\n\n\treturn\n}\n\n// IsSetHistory returns true if History is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetHistory() bool {\n\treturn v != nil && v.History != nil\n}\n\n// GetNextPageToken returns the value of NextPageToken if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\n\treturn\n}\n\n// IsSetNextPageToken returns true if NextPageToken is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetNextPageToken() bool {\n\treturn v != nil && v.NextPageToken != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetQuery() (o *WorkflowQuery) {\n\tif v != nil && v.Query != nil {\n\t\treturn v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\n// GetWorkflowExecutionTaskList returns the value of WorkflowExecutionTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetWorkflowExecutionTaskList() (o *TaskList) {\n\tif v != nil && v.WorkflowExecutionTaskList != nil {\n\t\treturn v.WorkflowExecutionTaskList\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionTaskList returns true if WorkflowExecutionTaskList is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetWorkflowExecutionTaskList() bool {\n\treturn v != nil && v.WorkflowExecutionTaskList != nil\n}\n\n// GetScheduledTimestamp returns the value of ScheduledTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimestamp returns true if ScheduledTimestamp is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetScheduledTimestamp() bool {\n\treturn v != nil && v.ScheduledTimestamp != nil\n}\n\n// GetStartedTimestamp returns the value of StartedTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetStartedTimestamp() (o int64) {\n\tif v != nil && v.StartedTimestamp != nil {\n\t\treturn *v.StartedTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimestamp returns true if StartedTimestamp is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetStartedTimestamp() bool {\n\treturn v != nil && v.StartedTimestamp != nil\n}\n\n// GetQueries returns the value of Queries if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetQueries() (o map[string]*WorkflowQuery) {\n\tif v != nil && v.Queries != nil {\n\t\treturn v.Queries\n\t}\n\n\treturn\n}\n\n// IsSetQueries returns true if Queries is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetQueries() bool {\n\treturn v != nil && v.Queries != nil\n}\n\n// GetNextEventId returns the value of NextEventId if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetNextEventId() (o int64) {\n\tif v != nil && v.NextEventId != nil {\n\t\treturn *v.NextEventId\n\t}\n\n\treturn\n}\n\n// IsSetNextEventId returns true if NextEventId is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetNextEventId() bool {\n\treturn v != nil && v.NextEventId != nil\n}\n\n// GetTotalHistoryBytes returns the value of TotalHistoryBytes if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetTotalHistoryBytes() (o int64) {\n\tif v != nil && v.TotalHistoryBytes != nil {\n\t\treturn *v.TotalHistoryBytes\n\t}\n\n\treturn\n}\n\n// IsSetTotalHistoryBytes returns true if TotalHistoryBytes is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetTotalHistoryBytes() bool {\n\treturn v != nil && v.TotalHistoryBytes != nil\n}\n\n// GetAutoConfigHint returns the value of AutoConfigHint if it is set or its\n// zero value if it is unset.\nfunc (v *PollForDecisionTaskResponse) GetAutoConfigHint() (o *AutoConfigHint) {\n\tif v != nil && v.AutoConfigHint != nil {\n\t\treturn v.AutoConfigHint\n\t}\n\n\treturn\n}\n\n// IsSetAutoConfigHint returns true if AutoConfigHint is not nil.\nfunc (v *PollForDecisionTaskResponse) IsSetAutoConfigHint() bool {\n\treturn v != nil && v.AutoConfigHint != nil\n}\n\ntype PollerInfo struct {\n\tLastAccessTime *int64   `json:\"lastAccessTime,omitempty\"`\n\tIdentity       *string  `json:\"identity,omitempty\"`\n\tRatePerSecond  *float64 `json:\"ratePerSecond,omitempty\"`\n}\n\n// ToWire translates a PollerInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *PollerInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.LastAccessTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastAccessTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RatePerSecond != nil {\n\t\tw, err = wire.NewValueDouble(*(v.RatePerSecond)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a PollerInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a PollerInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v PollerInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *PollerInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastAccessTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.RatePerSecond = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a PollerInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a PollerInfo struct could not be encoded.\nfunc (v *PollerInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.LastAccessTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastAccessTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RatePerSecond != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.RatePerSecond)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a PollerInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a PollerInfo struct could not be generated from the wire\n// representation.\nfunc (v *PollerInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastAccessTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.RatePerSecond = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a PollerInfo\n// struct.\nfunc (v *PollerInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.LastAccessTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastAccessTime: %v\", *(v.LastAccessTime))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RatePerSecond != nil {\n\t\tfields[i] = fmt.Sprintf(\"RatePerSecond: %v\", *(v.RatePerSecond))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"PollerInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this PollerInfo match the\n// provided PollerInfo.\n//\n// This function performs a deep comparison.\nfunc (v *PollerInfo) Equals(rhs *PollerInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastAccessTime, rhs.LastAccessTime) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.RatePerSecond, rhs.RatePerSecond) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PollerInfo.\nfunc (v *PollerInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.LastAccessTime != nil {\n\t\tenc.AddInt64(\"lastAccessTime\", *v.LastAccessTime)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RatePerSecond != nil {\n\t\tenc.AddFloat64(\"ratePerSecond\", *v.RatePerSecond)\n\t}\n\treturn err\n}\n\n// GetLastAccessTime returns the value of LastAccessTime if it is set or its\n// zero value if it is unset.\nfunc (v *PollerInfo) GetLastAccessTime() (o int64) {\n\tif v != nil && v.LastAccessTime != nil {\n\t\treturn *v.LastAccessTime\n\t}\n\n\treturn\n}\n\n// IsSetLastAccessTime returns true if LastAccessTime is not nil.\nfunc (v *PollerInfo) IsSetLastAccessTime() bool {\n\treturn v != nil && v.LastAccessTime != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *PollerInfo) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *PollerInfo) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRatePerSecond returns the value of RatePerSecond if it is set or its\n// zero value if it is unset.\nfunc (v *PollerInfo) GetRatePerSecond() (o float64) {\n\tif v != nil && v.RatePerSecond != nil {\n\t\treturn *v.RatePerSecond\n\t}\n\n\treturn\n}\n\n// IsSetRatePerSecond returns true if RatePerSecond is not nil.\nfunc (v *PollerInfo) IsSetRatePerSecond() bool {\n\treturn v != nil && v.RatePerSecond != nil\n}\n\ntype Predicate struct {\n\tPredicateType                *PredicateType                `json:\"predicateType,omitempty\"`\n\tUniversalPredicateAttributes *UniversalPredicateAttributes `json:\"universalPredicateAttributes,omitempty\"`\n\tEmptyPredicateAttributes     *EmptyPredicateAttributes     `json:\"emptyPredicateAttributes,omitempty\"`\n\tDomainIDPredicateAttributes  *DomainIDPredicateAttributes  `json:\"domainIDPredicateAttributes,omitempty\"`\n}\n\n// ToWire translates a Predicate struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *Predicate) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PredicateType != nil {\n\t\tw, err = v.PredicateType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.UniversalPredicateAttributes != nil {\n\t\tw, err = v.UniversalPredicateAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.EmptyPredicateAttributes != nil {\n\t\tw, err = v.EmptyPredicateAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.DomainIDPredicateAttributes != nil {\n\t\tw, err = v.DomainIDPredicateAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PredicateType_Read(w wire.Value) (PredicateType, error) {\n\tvar v PredicateType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _UniversalPredicateAttributes_Read(w wire.Value) (*UniversalPredicateAttributes, error) {\n\tvar v UniversalPredicateAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _EmptyPredicateAttributes_Read(w wire.Value) (*EmptyPredicateAttributes, error) {\n\tvar v EmptyPredicateAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _DomainIDPredicateAttributes_Read(w wire.Value) (*DomainIDPredicateAttributes, error) {\n\tvar v DomainIDPredicateAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a Predicate struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a Predicate struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v Predicate\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *Predicate) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x PredicateType\n\t\t\t\tx, err = _PredicateType_Read(field.Value)\n\t\t\t\tv.PredicateType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.UniversalPredicateAttributes, err = _UniversalPredicateAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.EmptyPredicateAttributes, err = _EmptyPredicateAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainIDPredicateAttributes, err = _DomainIDPredicateAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a Predicate struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a Predicate struct could not be encoded.\nfunc (v *Predicate) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PredicateType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PredicateType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UniversalPredicateAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.UniversalPredicateAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EmptyPredicateAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.EmptyPredicateAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainIDPredicateAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainIDPredicateAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PredicateType_Decode(sr stream.Reader) (PredicateType, error) {\n\tvar v PredicateType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _UniversalPredicateAttributes_Decode(sr stream.Reader) (*UniversalPredicateAttributes, error) {\n\tvar v UniversalPredicateAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _EmptyPredicateAttributes_Decode(sr stream.Reader) (*EmptyPredicateAttributes, error) {\n\tvar v EmptyPredicateAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _DomainIDPredicateAttributes_Decode(sr stream.Reader) (*DomainIDPredicateAttributes, error) {\n\tvar v DomainIDPredicateAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a Predicate struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a Predicate struct could not be generated from the wire\n// representation.\nfunc (v *Predicate) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x PredicateType\n\t\t\tx, err = _PredicateType_Decode(sr)\n\t\t\tv.PredicateType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.UniversalPredicateAttributes, err = _UniversalPredicateAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.EmptyPredicateAttributes, err = _EmptyPredicateAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.DomainIDPredicateAttributes, err = _DomainIDPredicateAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a Predicate\n// struct.\nfunc (v *Predicate) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.PredicateType != nil {\n\t\tfields[i] = fmt.Sprintf(\"PredicateType: %v\", *(v.PredicateType))\n\t\ti++\n\t}\n\tif v.UniversalPredicateAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"UniversalPredicateAttributes: %v\", v.UniversalPredicateAttributes)\n\t\ti++\n\t}\n\tif v.EmptyPredicateAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"EmptyPredicateAttributes: %v\", v.EmptyPredicateAttributes)\n\t\ti++\n\t}\n\tif v.DomainIDPredicateAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainIDPredicateAttributes: %v\", v.DomainIDPredicateAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"Predicate{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _PredicateType_EqualsPtr(lhs, rhs *PredicateType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this Predicate match the\n// provided Predicate.\n//\n// This function performs a deep comparison.\nfunc (v *Predicate) Equals(rhs *Predicate) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_PredicateType_EqualsPtr(v.PredicateType, rhs.PredicateType) {\n\t\treturn false\n\t}\n\tif !((v.UniversalPredicateAttributes == nil && rhs.UniversalPredicateAttributes == nil) || (v.UniversalPredicateAttributes != nil && rhs.UniversalPredicateAttributes != nil && v.UniversalPredicateAttributes.Equals(rhs.UniversalPredicateAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.EmptyPredicateAttributes == nil && rhs.EmptyPredicateAttributes == nil) || (v.EmptyPredicateAttributes != nil && rhs.EmptyPredicateAttributes != nil && v.EmptyPredicateAttributes.Equals(rhs.EmptyPredicateAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.DomainIDPredicateAttributes == nil && rhs.DomainIDPredicateAttributes == nil) || (v.DomainIDPredicateAttributes != nil && rhs.DomainIDPredicateAttributes != nil && v.DomainIDPredicateAttributes.Equals(rhs.DomainIDPredicateAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of Predicate.\nfunc (v *Predicate) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PredicateType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"predicateType\", *v.PredicateType))\n\t}\n\tif v.UniversalPredicateAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"universalPredicateAttributes\", v.UniversalPredicateAttributes))\n\t}\n\tif v.EmptyPredicateAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"emptyPredicateAttributes\", v.EmptyPredicateAttributes))\n\t}\n\tif v.DomainIDPredicateAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainIDPredicateAttributes\", v.DomainIDPredicateAttributes))\n\t}\n\treturn err\n}\n\n// GetPredicateType returns the value of PredicateType if it is set or its\n// zero value if it is unset.\nfunc (v *Predicate) GetPredicateType() (o PredicateType) {\n\tif v != nil && v.PredicateType != nil {\n\t\treturn *v.PredicateType\n\t}\n\n\treturn\n}\n\n// IsSetPredicateType returns true if PredicateType is not nil.\nfunc (v *Predicate) IsSetPredicateType() bool {\n\treturn v != nil && v.PredicateType != nil\n}\n\n// GetUniversalPredicateAttributes returns the value of UniversalPredicateAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Predicate) GetUniversalPredicateAttributes() (o *UniversalPredicateAttributes) {\n\tif v != nil && v.UniversalPredicateAttributes != nil {\n\t\treturn v.UniversalPredicateAttributes\n\t}\n\n\treturn\n}\n\n// IsSetUniversalPredicateAttributes returns true if UniversalPredicateAttributes is not nil.\nfunc (v *Predicate) IsSetUniversalPredicateAttributes() bool {\n\treturn v != nil && v.UniversalPredicateAttributes != nil\n}\n\n// GetEmptyPredicateAttributes returns the value of EmptyPredicateAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Predicate) GetEmptyPredicateAttributes() (o *EmptyPredicateAttributes) {\n\tif v != nil && v.EmptyPredicateAttributes != nil {\n\t\treturn v.EmptyPredicateAttributes\n\t}\n\n\treturn\n}\n\n// IsSetEmptyPredicateAttributes returns true if EmptyPredicateAttributes is not nil.\nfunc (v *Predicate) IsSetEmptyPredicateAttributes() bool {\n\treturn v != nil && v.EmptyPredicateAttributes != nil\n}\n\n// GetDomainIDPredicateAttributes returns the value of DomainIDPredicateAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *Predicate) GetDomainIDPredicateAttributes() (o *DomainIDPredicateAttributes) {\n\tif v != nil && v.DomainIDPredicateAttributes != nil {\n\t\treturn v.DomainIDPredicateAttributes\n\t}\n\n\treturn\n}\n\n// IsSetDomainIDPredicateAttributes returns true if DomainIDPredicateAttributes is not nil.\nfunc (v *Predicate) IsSetDomainIDPredicateAttributes() bool {\n\treturn v != nil && v.DomainIDPredicateAttributes != nil\n}\n\ntype PredicateType int32\n\nconst (\n\tPredicateTypeUniversal PredicateType = 0\n\tPredicateTypeEmpty     PredicateType = 1\n\tPredicateTypeDomainID  PredicateType = 2\n)\n\n// PredicateType_Values returns all recognized values of PredicateType.\nfunc PredicateType_Values() []PredicateType {\n\treturn []PredicateType{\n\t\tPredicateTypeUniversal,\n\t\tPredicateTypeEmpty,\n\t\tPredicateTypeDomainID,\n\t}\n}\n\n// UnmarshalText tries to decode PredicateType from a byte slice\n// containing its name.\n//\n//\tvar v PredicateType\n//\terr := v.UnmarshalText([]byte(\"Universal\"))\nfunc (v *PredicateType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Universal\":\n\t\t*v = PredicateTypeUniversal\n\t\treturn nil\n\tcase \"Empty\":\n\t\t*v = PredicateTypeEmpty\n\t\treturn nil\n\tcase \"DomainID\":\n\t\t*v = PredicateTypeDomainID\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"PredicateType\", err)\n\t\t}\n\t\t*v = PredicateType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes PredicateType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v PredicateType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Universal\"), nil\n\tcase 1:\n\t\treturn []byte(\"Empty\"), nil\n\tcase 2:\n\t\treturn []byte(\"DomainID\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of PredicateType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v PredicateType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Universal\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Empty\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"DomainID\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v PredicateType) Ptr() *PredicateType {\n\treturn &v\n}\n\n// Encode encodes PredicateType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v PredicateType\n//\treturn v.Encode(sWriter)\nfunc (v PredicateType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates PredicateType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v PredicateType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes PredicateType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return PredicateType(0), err\n//\t}\n//\n//\tvar v PredicateType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return PredicateType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PredicateType) FromWire(w wire.Value) error {\n\t*v = (PredicateType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded PredicateType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v PredicateType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return PredicateType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *PredicateType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (PredicateType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of PredicateType.\nfunc (v PredicateType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Universal\"\n\tcase 1:\n\t\treturn \"Empty\"\n\tcase 2:\n\t\treturn \"DomainID\"\n\t}\n\treturn fmt.Sprintf(\"PredicateType(%d)\", w)\n}\n\n// Equals returns true if this PredicateType value matches the provided\n// value.\nfunc (v PredicateType) Equals(rhs PredicateType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes PredicateType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v PredicateType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Universal\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Empty\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"DomainID\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode PredicateType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *PredicateType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"PredicateType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"PredicateType\")\n\t\t}\n\t\t*v = (PredicateType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"PredicateType\")\n\t}\n}\n\ntype QueryConsistencyLevel int32\n\nconst (\n\tQueryConsistencyLevelEventual QueryConsistencyLevel = 0\n\tQueryConsistencyLevelStrong   QueryConsistencyLevel = 1\n)\n\n// QueryConsistencyLevel_Values returns all recognized values of QueryConsistencyLevel.\nfunc QueryConsistencyLevel_Values() []QueryConsistencyLevel {\n\treturn []QueryConsistencyLevel{\n\t\tQueryConsistencyLevelEventual,\n\t\tQueryConsistencyLevelStrong,\n\t}\n}\n\n// UnmarshalText tries to decode QueryConsistencyLevel from a byte slice\n// containing its name.\n//\n//\tvar v QueryConsistencyLevel\n//\terr := v.UnmarshalText([]byte(\"EVENTUAL\"))\nfunc (v *QueryConsistencyLevel) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"EVENTUAL\":\n\t\t*v = QueryConsistencyLevelEventual\n\t\treturn nil\n\tcase \"STRONG\":\n\t\t*v = QueryConsistencyLevelStrong\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryConsistencyLevel\", err)\n\t\t}\n\t\t*v = QueryConsistencyLevel(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryConsistencyLevel to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v QueryConsistencyLevel) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"EVENTUAL\"), nil\n\tcase 1:\n\t\treturn []byte(\"STRONG\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryConsistencyLevel.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v QueryConsistencyLevel) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"EVENTUAL\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"STRONG\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v QueryConsistencyLevel) Ptr() *QueryConsistencyLevel {\n\treturn &v\n}\n\n// Encode encodes QueryConsistencyLevel directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v QueryConsistencyLevel\n//\treturn v.Encode(sWriter)\nfunc (v QueryConsistencyLevel) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates QueryConsistencyLevel into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v QueryConsistencyLevel) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes QueryConsistencyLevel from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return QueryConsistencyLevel(0), err\n//\t}\n//\n//\tvar v QueryConsistencyLevel\n//\tif err := v.FromWire(x); err != nil {\n//\t  return QueryConsistencyLevel(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryConsistencyLevel) FromWire(w wire.Value) error {\n\t*v = (QueryConsistencyLevel)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded QueryConsistencyLevel directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v QueryConsistencyLevel\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return QueryConsistencyLevel(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryConsistencyLevel) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (QueryConsistencyLevel)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of QueryConsistencyLevel.\nfunc (v QueryConsistencyLevel) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"EVENTUAL\"\n\tcase 1:\n\t\treturn \"STRONG\"\n\t}\n\treturn fmt.Sprintf(\"QueryConsistencyLevel(%d)\", w)\n}\n\n// Equals returns true if this QueryConsistencyLevel value matches the provided\n// value.\nfunc (v QueryConsistencyLevel) Equals(rhs QueryConsistencyLevel) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes QueryConsistencyLevel into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v QueryConsistencyLevel) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"EVENTUAL\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"STRONG\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode QueryConsistencyLevel from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *QueryConsistencyLevel) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"QueryConsistencyLevel\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"QueryConsistencyLevel\")\n\t\t}\n\t\t*v = (QueryConsistencyLevel)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"QueryConsistencyLevel\")\n\t}\n}\n\ntype QueryFailedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a QueryFailedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryFailedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a QueryFailedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryFailedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryFailedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryFailedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of QueryFailedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryFailedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryFailedError struct could not be encoded.\nfunc (v *QueryFailedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a QueryFailedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryFailedError struct could not be generated from the wire\n// representation.\nfunc (v *QueryFailedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of QueryFailedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryFailedError\n// struct.\nfunc (v *QueryFailedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"QueryFailedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*QueryFailedError) ErrorName() string {\n\treturn \"QueryFailedError\"\n}\n\n// Equals returns true if all the fields of this QueryFailedError match the\n// provided QueryFailedError.\n//\n// This function performs a deep comparison.\nfunc (v *QueryFailedError) Equals(rhs *QueryFailedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryFailedError.\nfunc (v *QueryFailedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *QueryFailedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *QueryFailedError) Error() string {\n\treturn v.String()\n}\n\ntype QueryRejectCondition int32\n\nconst (\n\tQueryRejectConditionNotOpen             QueryRejectCondition = 0\n\tQueryRejectConditionNotCompletedCleanly QueryRejectCondition = 1\n)\n\n// QueryRejectCondition_Values returns all recognized values of QueryRejectCondition.\nfunc QueryRejectCondition_Values() []QueryRejectCondition {\n\treturn []QueryRejectCondition{\n\t\tQueryRejectConditionNotOpen,\n\t\tQueryRejectConditionNotCompletedCleanly,\n\t}\n}\n\n// UnmarshalText tries to decode QueryRejectCondition from a byte slice\n// containing its name.\n//\n//\tvar v QueryRejectCondition\n//\terr := v.UnmarshalText([]byte(\"NOT_OPEN\"))\nfunc (v *QueryRejectCondition) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"NOT_OPEN\":\n\t\t*v = QueryRejectConditionNotOpen\n\t\treturn nil\n\tcase \"NOT_COMPLETED_CLEANLY\":\n\t\t*v = QueryRejectConditionNotCompletedCleanly\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryRejectCondition\", err)\n\t\t}\n\t\t*v = QueryRejectCondition(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryRejectCondition to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v QueryRejectCondition) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"NOT_OPEN\"), nil\n\tcase 1:\n\t\treturn []byte(\"NOT_COMPLETED_CLEANLY\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryRejectCondition.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v QueryRejectCondition) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"NOT_OPEN\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"NOT_COMPLETED_CLEANLY\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v QueryRejectCondition) Ptr() *QueryRejectCondition {\n\treturn &v\n}\n\n// Encode encodes QueryRejectCondition directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v QueryRejectCondition\n//\treturn v.Encode(sWriter)\nfunc (v QueryRejectCondition) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates QueryRejectCondition into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v QueryRejectCondition) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes QueryRejectCondition from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return QueryRejectCondition(0), err\n//\t}\n//\n//\tvar v QueryRejectCondition\n//\tif err := v.FromWire(x); err != nil {\n//\t  return QueryRejectCondition(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryRejectCondition) FromWire(w wire.Value) error {\n\t*v = (QueryRejectCondition)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded QueryRejectCondition directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v QueryRejectCondition\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return QueryRejectCondition(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryRejectCondition) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (QueryRejectCondition)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of QueryRejectCondition.\nfunc (v QueryRejectCondition) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"NOT_OPEN\"\n\tcase 1:\n\t\treturn \"NOT_COMPLETED_CLEANLY\"\n\t}\n\treturn fmt.Sprintf(\"QueryRejectCondition(%d)\", w)\n}\n\n// Equals returns true if this QueryRejectCondition value matches the provided\n// value.\nfunc (v QueryRejectCondition) Equals(rhs QueryRejectCondition) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes QueryRejectCondition into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v QueryRejectCondition) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"NOT_OPEN\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"NOT_COMPLETED_CLEANLY\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode QueryRejectCondition from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *QueryRejectCondition) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"QueryRejectCondition\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"QueryRejectCondition\")\n\t\t}\n\t\t*v = (QueryRejectCondition)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"QueryRejectCondition\")\n\t}\n}\n\ntype QueryRejected struct {\n\tCloseStatus *WorkflowExecutionCloseStatus `json:\"closeStatus,omitempty\"`\n}\n\n// ToWire translates a QueryRejected struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryRejected) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CloseStatus != nil {\n\t\tw, err = v.CloseStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a QueryRejected struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryRejected struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryRejected\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryRejected) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\t\tx, err = _WorkflowExecutionCloseStatus_Read(field.Value)\n\t\t\t\tv.CloseStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryRejected struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryRejected struct could not be encoded.\nfunc (v *QueryRejected) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CloseStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CloseStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a QueryRejected struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryRejected struct could not be generated from the wire\n// representation.\nfunc (v *QueryRejected) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\tx, err = _WorkflowExecutionCloseStatus_Decode(sr)\n\t\t\tv.CloseStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryRejected\n// struct.\nfunc (v *QueryRejected) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CloseStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"CloseStatus: %v\", *(v.CloseStatus))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryRejected{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this QueryRejected match the\n// provided QueryRejected.\n//\n// This function performs a deep comparison.\nfunc (v *QueryRejected) Equals(rhs *QueryRejected) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_WorkflowExecutionCloseStatus_EqualsPtr(v.CloseStatus, rhs.CloseStatus) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryRejected.\nfunc (v *QueryRejected) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CloseStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"closeStatus\", *v.CloseStatus))\n\t}\n\treturn err\n}\n\n// GetCloseStatus returns the value of CloseStatus if it is set or its\n// zero value if it is unset.\nfunc (v *QueryRejected) GetCloseStatus() (o WorkflowExecutionCloseStatus) {\n\tif v != nil && v.CloseStatus != nil {\n\t\treturn *v.CloseStatus\n\t}\n\n\treturn\n}\n\n// IsSetCloseStatus returns true if CloseStatus is not nil.\nfunc (v *QueryRejected) IsSetCloseStatus() bool {\n\treturn v != nil && v.CloseStatus != nil\n}\n\ntype QueryResultType int32\n\nconst (\n\tQueryResultTypeAnswered QueryResultType = 0\n\tQueryResultTypeFailed   QueryResultType = 1\n)\n\n// QueryResultType_Values returns all recognized values of QueryResultType.\nfunc QueryResultType_Values() []QueryResultType {\n\treturn []QueryResultType{\n\t\tQueryResultTypeAnswered,\n\t\tQueryResultTypeFailed,\n\t}\n}\n\n// UnmarshalText tries to decode QueryResultType from a byte slice\n// containing its name.\n//\n//\tvar v QueryResultType\n//\terr := v.UnmarshalText([]byte(\"ANSWERED\"))\nfunc (v *QueryResultType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"ANSWERED\":\n\t\t*v = QueryResultTypeAnswered\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*v = QueryResultTypeFailed\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryResultType\", err)\n\t\t}\n\t\t*v = QueryResultType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryResultType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v QueryResultType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"ANSWERED\"), nil\n\tcase 1:\n\t\treturn []byte(\"FAILED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryResultType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v QueryResultType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"ANSWERED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"FAILED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v QueryResultType) Ptr() *QueryResultType {\n\treturn &v\n}\n\n// Encode encodes QueryResultType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v QueryResultType\n//\treturn v.Encode(sWriter)\nfunc (v QueryResultType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates QueryResultType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v QueryResultType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes QueryResultType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return QueryResultType(0), err\n//\t}\n//\n//\tvar v QueryResultType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return QueryResultType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryResultType) FromWire(w wire.Value) error {\n\t*v = (QueryResultType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded QueryResultType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v QueryResultType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return QueryResultType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryResultType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (QueryResultType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of QueryResultType.\nfunc (v QueryResultType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ANSWERED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\t}\n\treturn fmt.Sprintf(\"QueryResultType(%d)\", w)\n}\n\n// Equals returns true if this QueryResultType value matches the provided\n// value.\nfunc (v QueryResultType) Equals(rhs QueryResultType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes QueryResultType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v QueryResultType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"ANSWERED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"FAILED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode QueryResultType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *QueryResultType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"QueryResultType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"QueryResultType\")\n\t\t}\n\t\t*v = (QueryResultType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"QueryResultType\")\n\t}\n}\n\ntype QueryTaskCompletedType int32\n\nconst (\n\tQueryTaskCompletedTypeCompleted QueryTaskCompletedType = 0\n\tQueryTaskCompletedTypeFailed    QueryTaskCompletedType = 1\n)\n\n// QueryTaskCompletedType_Values returns all recognized values of QueryTaskCompletedType.\nfunc QueryTaskCompletedType_Values() []QueryTaskCompletedType {\n\treturn []QueryTaskCompletedType{\n\t\tQueryTaskCompletedTypeCompleted,\n\t\tQueryTaskCompletedTypeFailed,\n\t}\n}\n\n// UnmarshalText tries to decode QueryTaskCompletedType from a byte slice\n// containing its name.\n//\n//\tvar v QueryTaskCompletedType\n//\terr := v.UnmarshalText([]byte(\"COMPLETED\"))\nfunc (v *QueryTaskCompletedType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"COMPLETED\":\n\t\t*v = QueryTaskCompletedTypeCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*v = QueryTaskCompletedTypeFailed\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryTaskCompletedType\", err)\n\t\t}\n\t\t*v = QueryTaskCompletedType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryTaskCompletedType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v QueryTaskCompletedType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"COMPLETED\"), nil\n\tcase 1:\n\t\treturn []byte(\"FAILED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryTaskCompletedType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v QueryTaskCompletedType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"COMPLETED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"FAILED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v QueryTaskCompletedType) Ptr() *QueryTaskCompletedType {\n\treturn &v\n}\n\n// Encode encodes QueryTaskCompletedType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v QueryTaskCompletedType\n//\treturn v.Encode(sWriter)\nfunc (v QueryTaskCompletedType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates QueryTaskCompletedType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v QueryTaskCompletedType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes QueryTaskCompletedType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return QueryTaskCompletedType(0), err\n//\t}\n//\n//\tvar v QueryTaskCompletedType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return QueryTaskCompletedType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryTaskCompletedType) FromWire(w wire.Value) error {\n\t*v = (QueryTaskCompletedType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded QueryTaskCompletedType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v QueryTaskCompletedType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return QueryTaskCompletedType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *QueryTaskCompletedType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (QueryTaskCompletedType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of QueryTaskCompletedType.\nfunc (v QueryTaskCompletedType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"COMPLETED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\t}\n\treturn fmt.Sprintf(\"QueryTaskCompletedType(%d)\", w)\n}\n\n// Equals returns true if this QueryTaskCompletedType value matches the provided\n// value.\nfunc (v QueryTaskCompletedType) Equals(rhs QueryTaskCompletedType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes QueryTaskCompletedType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v QueryTaskCompletedType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"COMPLETED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"FAILED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode QueryTaskCompletedType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *QueryTaskCompletedType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"QueryTaskCompletedType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"QueryTaskCompletedType\")\n\t\t}\n\t\t*v = (QueryTaskCompletedType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"QueryTaskCompletedType\")\n\t}\n}\n\ntype QueryWorkflowRequest struct {\n\tDomain                *string                `json:\"domain,omitempty\"`\n\tExecution             *WorkflowExecution     `json:\"execution,omitempty\"`\n\tQuery                 *WorkflowQuery         `json:\"query,omitempty\"`\n\tQueryRejectCondition  *QueryRejectCondition  `json:\"queryRejectCondition,omitempty\"`\n\tQueryConsistencyLevel *QueryConsistencyLevel `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// ToWire translates a QueryWorkflowRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryWorkflowRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tw, err = v.Query.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.QueryRejectCondition != nil {\n\t\tw, err = v.QueryRejectCondition.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tw, err = v.QueryConsistencyLevel.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryRejectCondition_Read(w wire.Value) (QueryRejectCondition, error) {\n\tvar v QueryRejectCondition\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a QueryWorkflowRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryWorkflowRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryWorkflowRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryWorkflowRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Query, err = _WorkflowQuery_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryRejectCondition\n\t\t\t\tx, err = _QueryRejectCondition_Read(field.Value)\n\t\t\t\tv.QueryRejectCondition = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryConsistencyLevel\n\t\t\t\tx, err = _QueryConsistencyLevel_Read(field.Value)\n\t\t\t\tv.QueryConsistencyLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryWorkflowRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be encoded.\nfunc (v *QueryWorkflowRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Query != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Query.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryRejectCondition != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRejectCondition.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryConsistencyLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryConsistencyLevel.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryRejectCondition_Decode(sr stream.Reader) (QueryRejectCondition, error) {\n\tvar v QueryRejectCondition\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a QueryWorkflowRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryWorkflowRequest struct could not be generated from the wire\n// representation.\nfunc (v *QueryWorkflowRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.Query, err = _WorkflowQuery_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x QueryRejectCondition\n\t\t\tx, err = _QueryRejectCondition_Decode(sr)\n\t\t\tv.QueryRejectCondition = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x QueryConsistencyLevel\n\t\t\tx, err = _QueryConsistencyLevel_Decode(sr)\n\t\t\tv.QueryConsistencyLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryWorkflowRequest\n// struct.\nfunc (v *QueryWorkflowRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.Query != nil {\n\t\tfields[i] = fmt.Sprintf(\"Query: %v\", v.Query)\n\t\ti++\n\t}\n\tif v.QueryRejectCondition != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRejectCondition: %v\", *(v.QueryRejectCondition))\n\t\ti++\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryConsistencyLevel: %v\", *(v.QueryConsistencyLevel))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryWorkflowRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _QueryRejectCondition_EqualsPtr(lhs, rhs *QueryRejectCondition) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this QueryWorkflowRequest match the\n// provided QueryWorkflowRequest.\n//\n// This function performs a deep comparison.\nfunc (v *QueryWorkflowRequest) Equals(rhs *QueryWorkflowRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !((v.Query == nil && rhs.Query == nil) || (v.Query != nil && rhs.Query != nil && v.Query.Equals(rhs.Query))) {\n\t\treturn false\n\t}\n\tif !_QueryRejectCondition_EqualsPtr(v.QueryRejectCondition, rhs.QueryRejectCondition) {\n\t\treturn false\n\t}\n\tif !_QueryConsistencyLevel_EqualsPtr(v.QueryConsistencyLevel, rhs.QueryConsistencyLevel) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryWorkflowRequest.\nfunc (v *QueryWorkflowRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.Query != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"query\", v.Query))\n\t}\n\tif v.QueryRejectCondition != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRejectCondition\", *v.QueryRejectCondition))\n\t}\n\tif v.QueryConsistencyLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryConsistencyLevel\", *v.QueryConsistencyLevel))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *QueryWorkflowRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *QueryWorkflowRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetQuery returns the value of Query if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetQuery() (o *WorkflowQuery) {\n\tif v != nil && v.Query != nil {\n\t\treturn v.Query\n\t}\n\n\treturn\n}\n\n// IsSetQuery returns true if Query is not nil.\nfunc (v *QueryWorkflowRequest) IsSetQuery() bool {\n\treturn v != nil && v.Query != nil\n}\n\n// GetQueryRejectCondition returns the value of QueryRejectCondition if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetQueryRejectCondition() (o QueryRejectCondition) {\n\tif v != nil && v.QueryRejectCondition != nil {\n\t\treturn *v.QueryRejectCondition\n\t}\n\n\treturn\n}\n\n// IsSetQueryRejectCondition returns true if QueryRejectCondition is not nil.\nfunc (v *QueryWorkflowRequest) IsSetQueryRejectCondition() bool {\n\treturn v != nil && v.QueryRejectCondition != nil\n}\n\n// GetQueryConsistencyLevel returns the value of QueryConsistencyLevel if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\n\treturn\n}\n\n// IsSetQueryConsistencyLevel returns true if QueryConsistencyLevel is not nil.\nfunc (v *QueryWorkflowRequest) IsSetQueryConsistencyLevel() bool {\n\treturn v != nil && v.QueryConsistencyLevel != nil\n}\n\ntype QueryWorkflowResponse struct {\n\tQueryResult   []byte         `json:\"queryResult,omitempty\"`\n\tQueryRejected *QueryRejected `json:\"queryRejected,omitempty\"`\n}\n\n// ToWire translates a QueryWorkflowResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueryWorkflowResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.QueryResult != nil {\n\t\tw, err = wire.NewValueBinary(v.QueryResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.QueryRejected != nil {\n\t\tw, err = v.QueryRejected.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryRejected_Read(w wire.Value) (*QueryRejected, error) {\n\tvar v QueryRejected\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a QueryWorkflowResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueryWorkflowResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueryWorkflowResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueryWorkflowResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.QueryResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.QueryRejected, err = _QueryRejected_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a QueryWorkflowResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueryWorkflowResponse struct could not be encoded.\nfunc (v *QueryWorkflowResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.QueryResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.QueryResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryRejected != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.QueryRejected.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryRejected_Decode(sr stream.Reader) (*QueryRejected, error) {\n\tvar v QueryRejected\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a QueryWorkflowResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueryWorkflowResponse struct could not be generated from the wire\n// representation.\nfunc (v *QueryWorkflowResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.QueryResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.QueryRejected, err = _QueryRejected_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueryWorkflowResponse\n// struct.\nfunc (v *QueryWorkflowResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.QueryResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryResult: %v\", v.QueryResult)\n\t\ti++\n\t}\n\tif v.QueryRejected != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryRejected: %v\", v.QueryRejected)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueryWorkflowResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this QueryWorkflowResponse match the\n// provided QueryWorkflowResponse.\n//\n// This function performs a deep comparison.\nfunc (v *QueryWorkflowResponse) Equals(rhs *QueryWorkflowResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.QueryResult == nil && rhs.QueryResult == nil) || (v.QueryResult != nil && rhs.QueryResult != nil && bytes.Equal(v.QueryResult, rhs.QueryResult))) {\n\t\treturn false\n\t}\n\tif !((v.QueryRejected == nil && rhs.QueryRejected == nil) || (v.QueryRejected != nil && rhs.QueryRejected != nil && v.QueryRejected.Equals(rhs.QueryRejected))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueryWorkflowResponse.\nfunc (v *QueryWorkflowResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.QueryResult != nil {\n\t\tenc.AddString(\"queryResult\", base64.StdEncoding.EncodeToString(v.QueryResult))\n\t}\n\tif v.QueryRejected != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryRejected\", v.QueryRejected))\n\t}\n\treturn err\n}\n\n// GetQueryResult returns the value of QueryResult if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowResponse) GetQueryResult() (o []byte) {\n\tif v != nil && v.QueryResult != nil {\n\t\treturn v.QueryResult\n\t}\n\n\treturn\n}\n\n// IsSetQueryResult returns true if QueryResult is not nil.\nfunc (v *QueryWorkflowResponse) IsSetQueryResult() bool {\n\treturn v != nil && v.QueryResult != nil\n}\n\n// GetQueryRejected returns the value of QueryRejected if it is set or its\n// zero value if it is unset.\nfunc (v *QueryWorkflowResponse) GetQueryRejected() (o *QueryRejected) {\n\tif v != nil && v.QueryRejected != nil {\n\t\treturn v.QueryRejected\n\t}\n\n\treturn\n}\n\n// IsSetQueryRejected returns true if QueryRejected is not nil.\nfunc (v *QueryWorkflowResponse) IsSetQueryRejected() bool {\n\treturn v != nil && v.QueryRejected != nil\n}\n\ntype QueueState struct {\n\tVirtualQueueStates    map[int64]*VirtualQueueState `json:\"virtualQueueStates,omitempty\"`\n\tExclusiveMaxReadLevel *TaskKey                     `json:\"exclusiveMaxReadLevel,omitempty\"`\n}\n\ntype _Map_I64_VirtualQueueState_MapItemList map[int64]*VirtualQueueState\n\nfunc (m _Map_I64_VirtualQueueState_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int64]*VirtualQueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueI64(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I64_VirtualQueueState_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I64_VirtualQueueState_MapItemList) KeyType() wire.Type {\n\treturn wire.TI64\n}\n\nfunc (_Map_I64_VirtualQueueState_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_I64_VirtualQueueState_MapItemList) Close() {}\n\n// ToWire translates a QueueState struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *QueueState) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.VirtualQueueStates != nil {\n\t\tw, err = wire.NewValueMap(_Map_I64_VirtualQueueState_MapItemList(v.VirtualQueueStates)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExclusiveMaxReadLevel != nil {\n\t\tw, err = v.ExclusiveMaxReadLevel.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VirtualQueueState_Read(w wire.Value) (*VirtualQueueState, error) {\n\tvar v VirtualQueueState\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_I64_VirtualQueueState_Read(m wire.MapItemList) (map[int64]*VirtualQueueState, error) {\n\tif m.KeyType() != wire.TI64 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int64]*VirtualQueueState, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI64(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _VirtualQueueState_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _TaskKey_Read(w wire.Value) (*TaskKey, error) {\n\tvar v TaskKey\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a QueueState struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a QueueState struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v QueueState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *QueueState) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.VirtualQueueStates, err = _Map_I64_VirtualQueueState_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExclusiveMaxReadLevel, err = _TaskKey_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_I64_VirtualQueueState_Encode(val map[int64]*VirtualQueueState, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI64,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int64]*VirtualQueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteInt64(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a QueueState struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a QueueState struct could not be encoded.\nfunc (v *QueueState) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.VirtualQueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I64_VirtualQueueState_Encode(v.VirtualQueueStates, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExclusiveMaxReadLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExclusiveMaxReadLevel.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VirtualQueueState_Decode(sr stream.Reader) (*VirtualQueueState, error) {\n\tvar v VirtualQueueState\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_I64_VirtualQueueState_Decode(sr stream.Reader) (map[int64]*VirtualQueueState, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI64 || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int64]*VirtualQueueState, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt64()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _VirtualQueueState_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _TaskKey_Decode(sr stream.Reader) (*TaskKey, error) {\n\tvar v TaskKey\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a QueueState struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a QueueState struct could not be generated from the wire\n// representation.\nfunc (v *QueueState) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.VirtualQueueStates, err = _Map_I64_VirtualQueueState_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ExclusiveMaxReadLevel, err = _TaskKey_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a QueueState\n// struct.\nfunc (v *QueueState) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.VirtualQueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"VirtualQueueStates: %v\", v.VirtualQueueStates)\n\t\ti++\n\t}\n\tif v.ExclusiveMaxReadLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExclusiveMaxReadLevel: %v\", v.ExclusiveMaxReadLevel)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"QueueState{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_I64_VirtualQueueState_Equals(lhs, rhs map[int64]*VirtualQueueState) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this QueueState match the\n// provided QueueState.\n//\n// This function performs a deep comparison.\nfunc (v *QueueState) Equals(rhs *QueueState) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.VirtualQueueStates == nil && rhs.VirtualQueueStates == nil) || (v.VirtualQueueStates != nil && rhs.VirtualQueueStates != nil && _Map_I64_VirtualQueueState_Equals(v.VirtualQueueStates, rhs.VirtualQueueStates))) {\n\t\treturn false\n\t}\n\tif !((v.ExclusiveMaxReadLevel == nil && rhs.ExclusiveMaxReadLevel == nil) || (v.ExclusiveMaxReadLevel != nil && rhs.ExclusiveMaxReadLevel != nil && v.ExclusiveMaxReadLevel.Equals(rhs.ExclusiveMaxReadLevel))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_I64_VirtualQueueState_Item_Zapper struct {\n\tKey   int64\n\tValue *VirtualQueueState\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I64_VirtualQueueState_Item_Zapper.\nfunc (v _Map_I64_VirtualQueueState_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt64(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\treturn err\n}\n\ntype _Map_I64_VirtualQueueState_Zapper map[int64]*VirtualQueueState\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I64_VirtualQueueState_Zapper.\nfunc (m _Map_I64_VirtualQueueState_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I64_VirtualQueueState_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of QueueState.\nfunc (v *QueueState) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.VirtualQueueStates != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"virtualQueueStates\", (_Map_I64_VirtualQueueState_Zapper)(v.VirtualQueueStates)))\n\t}\n\tif v.ExclusiveMaxReadLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"exclusiveMaxReadLevel\", v.ExclusiveMaxReadLevel))\n\t}\n\treturn err\n}\n\n// GetVirtualQueueStates returns the value of VirtualQueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *QueueState) GetVirtualQueueStates() (o map[int64]*VirtualQueueState) {\n\tif v != nil && v.VirtualQueueStates != nil {\n\t\treturn v.VirtualQueueStates\n\t}\n\n\treturn\n}\n\n// IsSetVirtualQueueStates returns true if VirtualQueueStates is not nil.\nfunc (v *QueueState) IsSetVirtualQueueStates() bool {\n\treturn v != nil && v.VirtualQueueStates != nil\n}\n\n// GetExclusiveMaxReadLevel returns the value of ExclusiveMaxReadLevel if it is set or its\n// zero value if it is unset.\nfunc (v *QueueState) GetExclusiveMaxReadLevel() (o *TaskKey) {\n\tif v != nil && v.ExclusiveMaxReadLevel != nil {\n\t\treturn v.ExclusiveMaxReadLevel\n\t}\n\n\treturn\n}\n\n// IsSetExclusiveMaxReadLevel returns true if ExclusiveMaxReadLevel is not nil.\nfunc (v *QueueState) IsSetExclusiveMaxReadLevel() bool {\n\treturn v != nil && v.ExclusiveMaxReadLevel != nil\n}\n\ntype ReapplyEventsRequest struct {\n\tDomainName        *string            `json:\"domainName,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tEvents            *DataBlob          `json:\"events,omitempty\"`\n}\n\n// ToWire translates a ReapplyEventsRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReapplyEventsRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainName != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tw, err = v.Events.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReapplyEventsRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReapplyEventsRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReapplyEventsRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReapplyEventsRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Events, err = _DataBlob_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReapplyEventsRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReapplyEventsRequest struct could not be encoded.\nfunc (v *ReapplyEventsRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Events != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Events.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReapplyEventsRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReapplyEventsRequest struct could not be generated from the wire\n// representation.\nfunc (v *ReapplyEventsRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.Events, err = _DataBlob_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReapplyEventsRequest\n// struct.\nfunc (v *ReapplyEventsRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.DomainName != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainName: %v\", *(v.DomainName))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Events != nil {\n\t\tfields[i] = fmt.Sprintf(\"Events: %v\", v.Events)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReapplyEventsRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReapplyEventsRequest match the\n// provided ReapplyEventsRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ReapplyEventsRequest) Equals(rhs *ReapplyEventsRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainName, rhs.DomainName) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.Events == nil && rhs.Events == nil) || (v.Events != nil && rhs.Events != nil && v.Events.Equals(rhs.Events))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReapplyEventsRequest.\nfunc (v *ReapplyEventsRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainName != nil {\n\t\tenc.AddString(\"domainName\", *v.DomainName)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Events != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"events\", v.Events))\n\t}\n\treturn err\n}\n\n// GetDomainName returns the value of DomainName if it is set or its\n// zero value if it is unset.\nfunc (v *ReapplyEventsRequest) GetDomainName() (o string) {\n\tif v != nil && v.DomainName != nil {\n\t\treturn *v.DomainName\n\t}\n\n\treturn\n}\n\n// IsSetDomainName returns true if DomainName is not nil.\nfunc (v *ReapplyEventsRequest) IsSetDomainName() bool {\n\treturn v != nil && v.DomainName != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ReapplyEventsRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ReapplyEventsRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetEvents returns the value of Events if it is set or its\n// zero value if it is unset.\nfunc (v *ReapplyEventsRequest) GetEvents() (o *DataBlob) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\n\treturn\n}\n\n// IsSetEvents returns true if Events is not nil.\nfunc (v *ReapplyEventsRequest) IsSetEvents() bool {\n\treturn v != nil && v.Events != nil\n}\n\ntype RecordActivityTaskHeartbeatByIDRequest struct {\n\tDomain     *string `json:\"domain,omitempty\"`\n\tWorkflowID *string `json:\"workflowID,omitempty\"`\n\tRunID      *string `json:\"runID,omitempty\"`\n\tActivityID *string `json:\"activityID,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tIdentity   *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskHeartbeatByIDRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RecordActivityTaskHeartbeatByIDRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskHeartbeatByIDRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskHeartbeatByIDRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskHeartbeatByIDRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatByIDRequest struct could not be encoded.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RecordActivityTaskHeartbeatByIDRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatByIDRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskHeartbeatByIDRequest\n// struct.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskHeartbeatByIDRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskHeartbeatByIDRequest match the\n// provided RecordActivityTaskHeartbeatByIDRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) Equals(rhs *RecordActivityTaskHeartbeatByIDRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskHeartbeatByIDRequest.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RecordActivityTaskHeartbeatRequest struct {\n\tTaskToken []byte  `json:\"taskToken,omitempty\"`\n\tDetails   []byte  `json:\"details,omitempty\"`\n\tIdentity  *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskHeartbeatRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskHeartbeatRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RecordActivityTaskHeartbeatRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskHeartbeatRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskHeartbeatRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskHeartbeatRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskHeartbeatRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatRequest struct could not be encoded.\nfunc (v *RecordActivityTaskHeartbeatRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RecordActivityTaskHeartbeatRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatRequest struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskHeartbeatRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskHeartbeatRequest\n// struct.\nfunc (v *RecordActivityTaskHeartbeatRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskHeartbeatRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskHeartbeatRequest match the\n// provided RecordActivityTaskHeartbeatRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskHeartbeatRequest) Equals(rhs *RecordActivityTaskHeartbeatRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskHeartbeatRequest.\nfunc (v *RecordActivityTaskHeartbeatRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RecordActivityTaskHeartbeatRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RecordActivityTaskHeartbeatRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RecordActivityTaskHeartbeatRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RecordActivityTaskHeartbeatResponse struct {\n\tCancelRequested *bool `json:\"cancelRequested,omitempty\"`\n}\n\n// ToWire translates a RecordActivityTaskHeartbeatResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordActivityTaskHeartbeatResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CancelRequested != nil {\n\t\tw, err = wire.NewValueBool(*(v.CancelRequested)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RecordActivityTaskHeartbeatResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordActivityTaskHeartbeatResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordActivityTaskHeartbeatResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordActivityTaskHeartbeatResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.CancelRequested = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordActivityTaskHeartbeatResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatResponse struct could not be encoded.\nfunc (v *RecordActivityTaskHeartbeatResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CancelRequested != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.CancelRequested)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RecordActivityTaskHeartbeatResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordActivityTaskHeartbeatResponse struct could not be generated from the wire\n// representation.\nfunc (v *RecordActivityTaskHeartbeatResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.CancelRequested = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordActivityTaskHeartbeatResponse\n// struct.\nfunc (v *RecordActivityTaskHeartbeatResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.CancelRequested != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequested: %v\", *(v.CancelRequested))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordActivityTaskHeartbeatResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordActivityTaskHeartbeatResponse match the\n// provided RecordActivityTaskHeartbeatResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RecordActivityTaskHeartbeatResponse) Equals(rhs *RecordActivityTaskHeartbeatResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.CancelRequested, rhs.CancelRequested) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordActivityTaskHeartbeatResponse.\nfunc (v *RecordActivityTaskHeartbeatResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CancelRequested != nil {\n\t\tenc.AddBool(\"cancelRequested\", *v.CancelRequested)\n\t}\n\treturn err\n}\n\n// GetCancelRequested returns the value of CancelRequested if it is set or its\n// zero value if it is unset.\nfunc (v *RecordActivityTaskHeartbeatResponse) GetCancelRequested() (o bool) {\n\tif v != nil && v.CancelRequested != nil {\n\t\treturn *v.CancelRequested\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequested returns true if CancelRequested is not nil.\nfunc (v *RecordActivityTaskHeartbeatResponse) IsSetCancelRequested() bool {\n\treturn v != nil && v.CancelRequested != nil\n}\n\ntype RecordMarkerDecisionAttributes struct {\n\tMarkerName *string `json:\"markerName,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tHeader     *Header `json:\"header,omitempty\"`\n}\n\n// ToWire translates a RecordMarkerDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RecordMarkerDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MarkerName != nil {\n\t\tw, err = wire.NewValueString(*(v.MarkerName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RecordMarkerDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RecordMarkerDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RecordMarkerDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RecordMarkerDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.MarkerName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RecordMarkerDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RecordMarkerDecisionAttributes struct could not be encoded.\nfunc (v *RecordMarkerDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MarkerName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.MarkerName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RecordMarkerDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RecordMarkerDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RecordMarkerDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.MarkerName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RecordMarkerDecisionAttributes\n// struct.\nfunc (v *RecordMarkerDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.MarkerName != nil {\n\t\tfields[i] = fmt.Sprintf(\"MarkerName: %v\", *(v.MarkerName))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RecordMarkerDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RecordMarkerDecisionAttributes match the\n// provided RecordMarkerDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RecordMarkerDecisionAttributes) Equals(rhs *RecordMarkerDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.MarkerName, rhs.MarkerName) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RecordMarkerDecisionAttributes.\nfunc (v *RecordMarkerDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MarkerName != nil {\n\t\tenc.AddString(\"markerName\", *v.MarkerName)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\treturn err\n}\n\n// GetMarkerName returns the value of MarkerName if it is set or its\n// zero value if it is unset.\nfunc (v *RecordMarkerDecisionAttributes) GetMarkerName() (o string) {\n\tif v != nil && v.MarkerName != nil {\n\t\treturn *v.MarkerName\n\t}\n\n\treturn\n}\n\n// IsSetMarkerName returns true if MarkerName is not nil.\nfunc (v *RecordMarkerDecisionAttributes) IsSetMarkerName() bool {\n\treturn v != nil && v.MarkerName != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RecordMarkerDecisionAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RecordMarkerDecisionAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *RecordMarkerDecisionAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *RecordMarkerDecisionAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\ntype RefreshWorkflowTasksRequest struct {\n\tDomain    *string            `json:\"domain,omitempty\"`\n\tExecution *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a RefreshWorkflowTasksRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RefreshWorkflowTasksRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RefreshWorkflowTasksRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RefreshWorkflowTasksRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RefreshWorkflowTasksRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RefreshWorkflowTasksRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RefreshWorkflowTasksRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RefreshWorkflowTasksRequest struct could not be encoded.\nfunc (v *RefreshWorkflowTasksRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RefreshWorkflowTasksRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RefreshWorkflowTasksRequest struct could not be generated from the wire\n// representation.\nfunc (v *RefreshWorkflowTasksRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RefreshWorkflowTasksRequest\n// struct.\nfunc (v *RefreshWorkflowTasksRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RefreshWorkflowTasksRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RefreshWorkflowTasksRequest match the\n// provided RefreshWorkflowTasksRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RefreshWorkflowTasksRequest) Equals(rhs *RefreshWorkflowTasksRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RefreshWorkflowTasksRequest.\nfunc (v *RefreshWorkflowTasksRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RefreshWorkflowTasksRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RefreshWorkflowTasksRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *RefreshWorkflowTasksRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *RefreshWorkflowTasksRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype RegisterDomainRequest struct {\n\tName                                   *string                            `json:\"name,omitempty\"`\n\tDescription                            *string                            `json:\"description,omitempty\"`\n\tOwnerEmail                             *string                            `json:\"ownerEmail,omitempty\"`\n\tWorkflowExecutionRetentionPeriodInDays *int32                             `json:\"workflowExecutionRetentionPeriodInDays,omitempty\"`\n\tEmitMetric                             *bool                              `json:\"emitMetric,omitempty\"`\n\tClusters                               []*ClusterReplicationConfiguration `json:\"clusters,omitempty\"`\n\tActiveClusterName                      *string                            `json:\"activeClusterName,omitempty\"`\n\tActiveClustersByRegion                 map[string]string                  `json:\"activeClustersByRegion,omitempty\"`\n\tActiveClusters                         *ActiveClusters                    `json:\"activeClusters,omitempty\"`\n\tData                                   map[string]string                  `json:\"data,omitempty\"`\n\tSecurityToken                          *string                            `json:\"securityToken,omitempty\"`\n\tIsGlobalDomain                         *bool                              `json:\"isGlobalDomain,omitempty\"`\n\tHistoryArchivalStatus                  *ArchivalStatus                    `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                     *string                            `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus               *ArchivalStatus                    `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                  *string                            `json:\"visibilityArchivalURI,omitempty\"`\n}\n\n// Default_RegisterDomainRequest constructs a new RegisterDomainRequest struct,\n// pre-populating any fields with defined default values.\nfunc Default_RegisterDomainRequest() *RegisterDomainRequest {\n\tvar v RegisterDomainRequest\n\tv.EmitMetric = ptr.Bool(true)\n\treturn &v\n}\n\n// ToWire translates a RegisterDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RegisterDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [16]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tw, err = wire.NewValueString(*(v.Description)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tw, err = wire.NewValueString(*(v.OwnerEmail)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowExecutionRetentionPeriodInDays)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tvEmitMetric := v.EmitMetric\n\tif vEmitMetric == nil {\n\t\tvEmitMetric = ptr.Bool(true)\n\t}\n\t{\n\t\tw, err = wire.NewValueBool(*(vEmitMetric)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tw, err = wire.NewValueList(_List_ClusterReplicationConfiguration_ValueList(v.Clusters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClustersByRegion != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.ActiveClustersByRegion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 75, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tw, err = v.ActiveClusters.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 76, Value: w}\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.Data)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tw, err = wire.NewValueString(*(v.SecurityToken)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsGlobalDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tw, err = v.HistoryArchivalStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.HistoryArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tw, err = v.VisibilityArchivalStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.VisibilityArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RegisterDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RegisterDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RegisterDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RegisterDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Description = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OwnerEmail = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowExecutionRetentionPeriodInDays = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.EmitMetric = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Clusters, err = _List_ClusterReplicationConfiguration_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 75:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActiveClustersByRegion, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 76:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusters, err = _ActiveClusters_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Data, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SecurityToken = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsGlobalDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ArchivalStatus\n\t\t\t\tx, err = _ArchivalStatus_Read(field.Value)\n\t\t\t\tv.HistoryArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.HistoryArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ArchivalStatus\n\t\t\t\tx, err = _ArchivalStatus_Read(field.Value)\n\t\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.VisibilityArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tif v.EmitMetric == nil {\n\t\tv.EmitMetric = ptr.Bool(true)\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RegisterDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RegisterDomainRequest struct could not be encoded.\nfunc (v *RegisterDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Description != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Description)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OwnerEmail != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OwnerEmail)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowExecutionRetentionPeriodInDays)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tvEmitMetric := v.EmitMetric\n\tif vEmitMetric == nil {\n\t\tvEmitMetric = ptr.Bool(true)\n\t}\n\t{\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(vEmitMetric)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Clusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ClusterReplicationConfiguration_Encode(v.Clusters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClustersByRegion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 75, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.ActiveClustersByRegion, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 76, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusters.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.Data, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SecurityToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SecurityToken)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsGlobalDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsGlobalDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.HistoryArchivalStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.HistoryArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.VisibilityArchivalStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.VisibilityArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RegisterDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RegisterDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *RegisterDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Description = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OwnerEmail = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowExecutionRetentionPeriodInDays = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.EmitMetric = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TList:\n\t\t\tv.Clusters, err = _List_ClusterReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 75 && fh.Type == wire.TMap:\n\t\t\tv.ActiveClustersByRegion, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 76 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusters, err = _ActiveClusters_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TMap:\n\t\t\tv.Data, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SecurityToken = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsGlobalDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TI32:\n\t\t\tvar x ArchivalStatus\n\t\t\tx, err = _ArchivalStatus_Decode(sr)\n\t\t\tv.HistoryArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.HistoryArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TI32:\n\t\t\tvar x ArchivalStatus\n\t\t\tx, err = _ArchivalStatus_Decode(sr)\n\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.VisibilityArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EmitMetric == nil {\n\t\tv.EmitMetric = ptr.Bool(true)\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RegisterDomainRequest\n// struct.\nfunc (v *RegisterDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [16]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tfields[i] = fmt.Sprintf(\"Description: %v\", *(v.Description))\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tfields[i] = fmt.Sprintf(\"OwnerEmail: %v\", *(v.OwnerEmail))\n\t\ti++\n\t}\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecutionRetentionPeriodInDays: %v\", *(v.WorkflowExecutionRetentionPeriodInDays))\n\t\ti++\n\t}\n\tif v.EmitMetric != nil {\n\t\tfields[i] = fmt.Sprintf(\"EmitMetric: %v\", *(v.EmitMetric))\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Clusters: %v\", v.Clusters)\n\t\ti++\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterName: %v\", *(v.ActiveClusterName))\n\t\ti++\n\t}\n\tif v.ActiveClustersByRegion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClustersByRegion: %v\", v.ActiveClustersByRegion)\n\t\ti++\n\t}\n\tif v.ActiveClusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusters: %v\", v.ActiveClusters)\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"SecurityToken: %v\", *(v.SecurityToken))\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsGlobalDomain: %v\", *(v.IsGlobalDomain))\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalStatus: %v\", *(v.HistoryArchivalStatus))\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalURI: %v\", *(v.HistoryArchivalURI))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalStatus: %v\", *(v.VisibilityArchivalStatus))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalURI: %v\", *(v.VisibilityArchivalURI))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RegisterDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RegisterDomainRequest match the\n// provided RegisterDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RegisterDomainRequest) Equals(rhs *RegisterDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Description, rhs.Description) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OwnerEmail, rhs.OwnerEmail) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowExecutionRetentionPeriodInDays, rhs.WorkflowExecutionRetentionPeriodInDays) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.EmitMetric, rhs.EmitMetric) {\n\t\treturn false\n\t}\n\tif !((v.Clusters == nil && rhs.Clusters == nil) || (v.Clusters != nil && rhs.Clusters != nil && _List_ClusterReplicationConfiguration_Equals(v.Clusters, rhs.Clusters))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClusterName, rhs.ActiveClusterName) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClustersByRegion == nil && rhs.ActiveClustersByRegion == nil) || (v.ActiveClustersByRegion != nil && rhs.ActiveClustersByRegion != nil && _Map_String_String_Equals(v.ActiveClustersByRegion, rhs.ActiveClustersByRegion))) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusters == nil && rhs.ActiveClusters == nil) || (v.ActiveClusters != nil && rhs.ActiveClusters != nil && v.ActiveClusters.Equals(rhs.ActiveClusters))) {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && _Map_String_String_Equals(v.Data, rhs.Data))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SecurityToken, rhs.SecurityToken) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsGlobalDomain, rhs.IsGlobalDomain) {\n\t\treturn false\n\t}\n\tif !_ArchivalStatus_EqualsPtr(v.HistoryArchivalStatus, rhs.HistoryArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.HistoryArchivalURI, rhs.HistoryArchivalURI) {\n\t\treturn false\n\t}\n\tif !_ArchivalStatus_EqualsPtr(v.VisibilityArchivalStatus, rhs.VisibilityArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.VisibilityArchivalURI, rhs.VisibilityArchivalURI) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RegisterDomainRequest.\nfunc (v *RegisterDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Description != nil {\n\t\tenc.AddString(\"description\", *v.Description)\n\t}\n\tif v.OwnerEmail != nil {\n\t\tenc.AddString(\"ownerEmail\", *v.OwnerEmail)\n\t}\n\tif v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tenc.AddInt32(\"workflowExecutionRetentionPeriodInDays\", *v.WorkflowExecutionRetentionPeriodInDays)\n\t}\n\tif v.EmitMetric != nil {\n\t\tenc.AddBool(\"emitMetric\", *v.EmitMetric)\n\t}\n\tif v.Clusters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"clusters\", (_List_ClusterReplicationConfiguration_Zapper)(v.Clusters)))\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tenc.AddString(\"activeClusterName\", *v.ActiveClusterName)\n\t}\n\tif v.ActiveClustersByRegion != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClustersByRegion\", (_Map_String_String_Zapper)(v.ActiveClustersByRegion)))\n\t}\n\tif v.ActiveClusters != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusters\", v.ActiveClusters))\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", (_Map_String_String_Zapper)(v.Data)))\n\t}\n\tif v.SecurityToken != nil {\n\t\tenc.AddString(\"securityToken\", *v.SecurityToken)\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tenc.AddBool(\"isGlobalDomain\", *v.IsGlobalDomain)\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"historyArchivalStatus\", *v.HistoryArchivalStatus))\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tenc.AddString(\"historyArchivalURI\", *v.HistoryArchivalURI)\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"visibilityArchivalStatus\", *v.VisibilityArchivalStatus))\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tenc.AddString(\"visibilityArchivalURI\", *v.VisibilityArchivalURI)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *RegisterDomainRequest) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetDescription returns the value of Description if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetDescription() (o string) {\n\tif v != nil && v.Description != nil {\n\t\treturn *v.Description\n\t}\n\n\treturn\n}\n\n// IsSetDescription returns true if Description is not nil.\nfunc (v *RegisterDomainRequest) IsSetDescription() bool {\n\treturn v != nil && v.Description != nil\n}\n\n// GetOwnerEmail returns the value of OwnerEmail if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetOwnerEmail() (o string) {\n\tif v != nil && v.OwnerEmail != nil {\n\t\treturn *v.OwnerEmail\n\t}\n\n\treturn\n}\n\n// IsSetOwnerEmail returns true if OwnerEmail is not nil.\nfunc (v *RegisterDomainRequest) IsSetOwnerEmail() bool {\n\treturn v != nil && v.OwnerEmail != nil\n}\n\n// GetWorkflowExecutionRetentionPeriodInDays returns the value of WorkflowExecutionRetentionPeriodInDays if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetWorkflowExecutionRetentionPeriodInDays() (o int32) {\n\tif v != nil && v.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\treturn *v.WorkflowExecutionRetentionPeriodInDays\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecutionRetentionPeriodInDays returns true if WorkflowExecutionRetentionPeriodInDays is not nil.\nfunc (v *RegisterDomainRequest) IsSetWorkflowExecutionRetentionPeriodInDays() bool {\n\treturn v != nil && v.WorkflowExecutionRetentionPeriodInDays != nil\n}\n\n// GetEmitMetric returns the value of EmitMetric if it is set or its\n// default value if it is unset.\nfunc (v *RegisterDomainRequest) GetEmitMetric() (o bool) {\n\tif v != nil && v.EmitMetric != nil {\n\t\treturn *v.EmitMetric\n\t}\n\to = true\n\treturn\n}\n\n// IsSetEmitMetric returns true if EmitMetric is not nil.\nfunc (v *RegisterDomainRequest) IsSetEmitMetric() bool {\n\treturn v != nil && v.EmitMetric != nil\n}\n\n// GetClusters returns the value of Clusters if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetClusters() (o []*ClusterReplicationConfiguration) {\n\tif v != nil && v.Clusters != nil {\n\t\treturn v.Clusters\n\t}\n\n\treturn\n}\n\n// IsSetClusters returns true if Clusters is not nil.\nfunc (v *RegisterDomainRequest) IsSetClusters() bool {\n\treturn v != nil && v.Clusters != nil\n}\n\n// GetActiveClusterName returns the value of ActiveClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetActiveClusterName() (o string) {\n\tif v != nil && v.ActiveClusterName != nil {\n\t\treturn *v.ActiveClusterName\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterName returns true if ActiveClusterName is not nil.\nfunc (v *RegisterDomainRequest) IsSetActiveClusterName() bool {\n\treturn v != nil && v.ActiveClusterName != nil\n}\n\n// GetActiveClustersByRegion returns the value of ActiveClustersByRegion if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetActiveClustersByRegion() (o map[string]string) {\n\tif v != nil && v.ActiveClustersByRegion != nil {\n\t\treturn v.ActiveClustersByRegion\n\t}\n\n\treturn\n}\n\n// IsSetActiveClustersByRegion returns true if ActiveClustersByRegion is not nil.\nfunc (v *RegisterDomainRequest) IsSetActiveClustersByRegion() bool {\n\treturn v != nil && v.ActiveClustersByRegion != nil\n}\n\n// GetActiveClusters returns the value of ActiveClusters if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetActiveClusters() (o *ActiveClusters) {\n\tif v != nil && v.ActiveClusters != nil {\n\t\treturn v.ActiveClusters\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusters returns true if ActiveClusters is not nil.\nfunc (v *RegisterDomainRequest) IsSetActiveClusters() bool {\n\treturn v != nil && v.ActiveClusters != nil\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *RegisterDomainRequest) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\n// GetSecurityToken returns the value of SecurityToken if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetSecurityToken() (o string) {\n\tif v != nil && v.SecurityToken != nil {\n\t\treturn *v.SecurityToken\n\t}\n\n\treturn\n}\n\n// IsSetSecurityToken returns true if SecurityToken is not nil.\nfunc (v *RegisterDomainRequest) IsSetSecurityToken() bool {\n\treturn v != nil && v.SecurityToken != nil\n}\n\n// GetIsGlobalDomain returns the value of IsGlobalDomain if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetIsGlobalDomain() (o bool) {\n\tif v != nil && v.IsGlobalDomain != nil {\n\t\treturn *v.IsGlobalDomain\n\t}\n\n\treturn\n}\n\n// IsSetIsGlobalDomain returns true if IsGlobalDomain is not nil.\nfunc (v *RegisterDomainRequest) IsSetIsGlobalDomain() bool {\n\treturn v != nil && v.IsGlobalDomain != nil\n}\n\n// GetHistoryArchivalStatus returns the value of HistoryArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetHistoryArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.HistoryArchivalStatus != nil {\n\t\treturn *v.HistoryArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalStatus returns true if HistoryArchivalStatus is not nil.\nfunc (v *RegisterDomainRequest) IsSetHistoryArchivalStatus() bool {\n\treturn v != nil && v.HistoryArchivalStatus != nil\n}\n\n// GetHistoryArchivalURI returns the value of HistoryArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetHistoryArchivalURI() (o string) {\n\tif v != nil && v.HistoryArchivalURI != nil {\n\t\treturn *v.HistoryArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalURI returns true if HistoryArchivalURI is not nil.\nfunc (v *RegisterDomainRequest) IsSetHistoryArchivalURI() bool {\n\treturn v != nil && v.HistoryArchivalURI != nil\n}\n\n// GetVisibilityArchivalStatus returns the value of VisibilityArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetVisibilityArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.VisibilityArchivalStatus != nil {\n\t\treturn *v.VisibilityArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalStatus returns true if VisibilityArchivalStatus is not nil.\nfunc (v *RegisterDomainRequest) IsSetVisibilityArchivalStatus() bool {\n\treturn v != nil && v.VisibilityArchivalStatus != nil\n}\n\n// GetVisibilityArchivalURI returns the value of VisibilityArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *RegisterDomainRequest) GetVisibilityArchivalURI() (o string) {\n\tif v != nil && v.VisibilityArchivalURI != nil {\n\t\treturn *v.VisibilityArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalURI returns true if VisibilityArchivalURI is not nil.\nfunc (v *RegisterDomainRequest) IsSetVisibilityArchivalURI() bool {\n\treturn v != nil && v.VisibilityArchivalURI != nil\n}\n\ntype RemoteSyncMatchedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a RemoteSyncMatchedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RemoteSyncMatchedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 10, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RemoteSyncMatchedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RemoteSyncMatchedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RemoteSyncMatchedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RemoteSyncMatchedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of RemoteSyncMatchedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RemoteSyncMatchedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RemoteSyncMatchedError struct could not be encoded.\nfunc (v *RemoteSyncMatchedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RemoteSyncMatchedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RemoteSyncMatchedError struct could not be generated from the wire\n// representation.\nfunc (v *RemoteSyncMatchedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of RemoteSyncMatchedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RemoteSyncMatchedError\n// struct.\nfunc (v *RemoteSyncMatchedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"RemoteSyncMatchedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*RemoteSyncMatchedError) ErrorName() string {\n\treturn \"RemoteSyncMatchedError\"\n}\n\n// Equals returns true if all the fields of this RemoteSyncMatchedError match the\n// provided RemoteSyncMatchedError.\n//\n// This function performs a deep comparison.\nfunc (v *RemoteSyncMatchedError) Equals(rhs *RemoteSyncMatchedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RemoteSyncMatchedError.\nfunc (v *RemoteSyncMatchedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *RemoteSyncMatchedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *RemoteSyncMatchedError) Error() string {\n\treturn v.String()\n}\n\ntype RemoveTaskRequest struct {\n\tShardID             *int32  `json:\"shardID,omitempty\"`\n\tType                *int32  `json:\"type,omitempty\"`\n\tTaskID              *int64  `json:\"taskID,omitempty\"`\n\tVisibilityTimestamp *int64  `json:\"visibilityTimestamp,omitempty\"`\n\tClusterName         *string `json:\"clusterName,omitempty\"`\n}\n\n// ToWire translates a RemoveTaskRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RemoveTaskRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tw, err = wire.NewValueI32(*(v.Type)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.VisibilityTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RemoveTaskRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RemoveTaskRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RemoveTaskRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RemoveTaskRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.VisibilityTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RemoveTaskRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RemoveTaskRequest struct could not be encoded.\nfunc (v *RemoveTaskRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Type)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.VisibilityTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RemoveTaskRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RemoveTaskRequest struct could not be generated from the wire\n// representation.\nfunc (v *RemoveTaskRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.VisibilityTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RemoveTaskRequest\n// struct.\nfunc (v *RemoveTaskRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityTimestamp: %v\", *(v.VisibilityTimestamp))\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RemoveTaskRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RemoveTaskRequest match the\n// provided RemoveTaskRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RemoveTaskRequest) Equals(rhs *RemoveTaskRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.VisibilityTimestamp, rhs.VisibilityTimestamp) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RemoveTaskRequest.\nfunc (v *RemoveTaskRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.Type != nil {\n\t\tenc.AddInt32(\"type\", *v.Type)\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tenc.AddInt64(\"visibilityTimestamp\", *v.VisibilityTimestamp)\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveTaskRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *RemoveTaskRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveTaskRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *RemoveTaskRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveTaskRequest) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *RemoveTaskRequest) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetVisibilityTimestamp returns the value of VisibilityTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveTaskRequest) GetVisibilityTimestamp() (o int64) {\n\tif v != nil && v.VisibilityTimestamp != nil {\n\t\treturn *v.VisibilityTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityTimestamp returns true if VisibilityTimestamp is not nil.\nfunc (v *RemoveTaskRequest) IsSetVisibilityTimestamp() bool {\n\treturn v != nil && v.VisibilityTimestamp != nil\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *RemoveTaskRequest) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *RemoveTaskRequest) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\ntype RequestCancelActivityTaskDecisionAttributes struct {\n\tActivityId *string `json:\"activityId,omitempty\"`\n}\n\n// ToWire translates a RequestCancelActivityTaskDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelActivityTaskDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelActivityTaskDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelActivityTaskDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelActivityTaskDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelActivityTaskDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelActivityTaskDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelActivityTaskDecisionAttributes struct could not be encoded.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelActivityTaskDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelActivityTaskDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelActivityTaskDecisionAttributes\n// struct.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelActivityTaskDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelActivityTaskDecisionAttributes match the\n// provided RequestCancelActivityTaskDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) Equals(rhs *RequestCancelActivityTaskDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelActivityTaskDecisionAttributes.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *RequestCancelActivityTaskDecisionAttributes) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\ntype RequestCancelActivityTaskFailedEventAttributes struct {\n\tActivityId                   *string `json:\"activityId,omitempty\"`\n\tCause                        *string `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a RequestCancelActivityTaskFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = wire.NewValueString(*(v.Cause)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelActivityTaskFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelActivityTaskFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelActivityTaskFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelActivityTaskFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelActivityTaskFailedEventAttributes struct could not be encoded.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Cause)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelActivityTaskFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelActivityTaskFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelActivityTaskFailedEventAttributes\n// struct.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelActivityTaskFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelActivityTaskFailedEventAttributes match the\n// provided RequestCancelActivityTaskFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) Equals(rhs *RequestCancelActivityTaskFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelActivityTaskFailedEventAttributes.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.Cause != nil {\n\t\tenc.AddString(\"cause\", *v.Cause)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) GetCause() (o string) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype RequestCancelExternalWorkflowExecutionDecisionAttributes struct {\n\tDomain            *string `json:\"domain,omitempty\"`\n\tWorkflowId        *string `json:\"workflowId,omitempty\"`\n\tRunId             *string `json:\"runId,omitempty\"`\n\tControl           []byte  `json:\"control,omitempty\"`\n\tChildWorkflowOnly *bool   `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a RequestCancelExternalWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelExternalWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelExternalWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelExternalWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelExternalWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelExternalWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelExternalWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelExternalWorkflowExecutionDecisionAttributes match the\n// provided RequestCancelExternalWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) Equals(rhs *RequestCancelExternalWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelExternalWorkflowExecutionDecisionAttributes.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype RequestCancelExternalWorkflowExecutionFailedEventAttributes struct {\n\tCause                        *CancelExternalWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventId *int64                                      `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       *string                                     `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution                          `json:\"workflowExecution,omitempty\"`\n\tInitiatedEventId             *int64                                      `json:\"initiatedEventId,omitempty\"`\n\tControl                      []byte                                      `json:\"control,omitempty\"`\n}\n\n// ToWire translates a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CancelExternalWorkflowExecutionFailedCause_Read(w wire.Value) (CancelExternalWorkflowExecutionFailedCause, error) {\n\tvar v CancelExternalWorkflowExecutionFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelExternalWorkflowExecutionFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CancelExternalWorkflowExecutionFailedCause\n\t\t\t\tx, err = _CancelExternalWorkflowExecutionFailedCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct could not be encoded.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CancelExternalWorkflowExecutionFailedCause_Decode(sr stream.Reader) (CancelExternalWorkflowExecutionFailedCause, error) {\n\tvar v CancelExternalWorkflowExecutionFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x CancelExternalWorkflowExecutionFailedCause\n\t\t\tx, err = _CancelExternalWorkflowExecutionFailedCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelExternalWorkflowExecutionFailedEventAttributes\n// struct.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _CancelExternalWorkflowExecutionFailedCause_EqualsPtr(lhs, rhs *CancelExternalWorkflowExecutionFailedCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this RequestCancelExternalWorkflowExecutionFailedEventAttributes match the\n// provided RequestCancelExternalWorkflowExecutionFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) Equals(rhs *RequestCancelExternalWorkflowExecutionFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_CancelExternalWorkflowExecutionFailedCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelExternalWorkflowExecutionFailedEventAttributes.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetCause() (o CancelExternalWorkflowExecutionFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct {\n\tDecisionTaskCompletedEventId *int64             `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tControl                      []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly            *bool              `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct could not be encoded.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n// struct.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelExternalWorkflowExecutionInitiatedEventAttributes match the\n// provided RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) Equals(rhs *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype RequestCancelWorkflowExecutionRequest struct {\n\tDomain              *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution   *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIdentity            *string            `json:\"identity,omitempty\"`\n\tRequestId           *string            `json:\"requestId,omitempty\"`\n\tCause               *string            `json:\"cause,omitempty\"`\n\tFirstExecutionRunID *string            `json:\"firstExecutionRunID,omitempty\"`\n}\n\n// ToWire translates a RequestCancelWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = wire.NewValueString(*(v.Cause)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.FirstExecutionRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FirstExecutionRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelWorkflowExecutionRequest struct could not be encoded.\nfunc (v *RequestCancelWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Cause)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstExecutionRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FirstExecutionRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FirstExecutionRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelWorkflowExecutionRequest\n// struct.\nfunc (v *RequestCancelWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstExecutionRunID: %v\", *(v.FirstExecutionRunID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelWorkflowExecutionRequest match the\n// provided RequestCancelWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelWorkflowExecutionRequest) Equals(rhs *RequestCancelWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FirstExecutionRunID, rhs.FirstExecutionRunID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelWorkflowExecutionRequest.\nfunc (v *RequestCancelWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.Cause != nil {\n\t\tenc.AddString(\"cause\", *v.Cause)\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tenc.AddString(\"firstExecutionRunID\", *v.FirstExecutionRunID)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetCause() (o string) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetFirstExecutionRunID returns the value of FirstExecutionRunID if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelWorkflowExecutionRequest) GetFirstExecutionRunID() (o string) {\n\tif v != nil && v.FirstExecutionRunID != nil {\n\t\treturn *v.FirstExecutionRunID\n\t}\n\n\treturn\n}\n\n// IsSetFirstExecutionRunID returns true if FirstExecutionRunID is not nil.\nfunc (v *RequestCancelWorkflowExecutionRequest) IsSetFirstExecutionRunID() bool {\n\treturn v != nil && v.FirstExecutionRunID != nil\n}\n\ntype ResetPointInfo struct {\n\tBinaryChecksum           *string `json:\"binaryChecksum,omitempty\"`\n\tRunId                    *string `json:\"runId,omitempty\"`\n\tFirstDecisionCompletedId *int64  `json:\"firstDecisionCompletedId,omitempty\"`\n\tCreatedTimeNano          *int64  `json:\"createdTimeNano,omitempty\"`\n\tExpiringTimeNano         *int64  `json:\"expiringTimeNano,omitempty\"`\n\tResettable               *bool   `json:\"resettable,omitempty\"`\n}\n\n// ToWire translates a ResetPointInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetPointInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.FirstDecisionCompletedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstDecisionCompletedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreatedTimeNano)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ExpiringTimeNano != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpiringTimeNano)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Resettable != nil {\n\t\tw, err = wire.NewValueBool(*(v.Resettable)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetPointInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetPointInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetPointInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetPointInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstDecisionCompletedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreatedTimeNano = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpiringTimeNano = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Resettable = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetPointInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetPointInfo struct could not be encoded.\nfunc (v *ResetPointInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstDecisionCompletedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstDecisionCompletedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreatedTimeNano != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreatedTimeNano)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpiringTimeNano != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpiringTimeNano)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Resettable != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Resettable)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetPointInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetPointInfo struct could not be generated from the wire\n// representation.\nfunc (v *ResetPointInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstDecisionCompletedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreatedTimeNano = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpiringTimeNano = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Resettable = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetPointInfo\n// struct.\nfunc (v *ResetPointInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.FirstDecisionCompletedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstDecisionCompletedId: %v\", *(v.FirstDecisionCompletedId))\n\t\ti++\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreatedTimeNano: %v\", *(v.CreatedTimeNano))\n\t\ti++\n\t}\n\tif v.ExpiringTimeNano != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpiringTimeNano: %v\", *(v.ExpiringTimeNano))\n\t\ti++\n\t}\n\tif v.Resettable != nil {\n\t\tfields[i] = fmt.Sprintf(\"Resettable: %v\", *(v.Resettable))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetPointInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetPointInfo match the\n// provided ResetPointInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ResetPointInfo) Equals(rhs *ResetPointInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstDecisionCompletedId, rhs.FirstDecisionCompletedId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreatedTimeNano, rhs.CreatedTimeNano) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpiringTimeNano, rhs.ExpiringTimeNano) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Resettable, rhs.Resettable) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetPointInfo.\nfunc (v *ResetPointInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.FirstDecisionCompletedId != nil {\n\t\tenc.AddInt64(\"firstDecisionCompletedId\", *v.FirstDecisionCompletedId)\n\t}\n\tif v.CreatedTimeNano != nil {\n\t\tenc.AddInt64(\"createdTimeNano\", *v.CreatedTimeNano)\n\t}\n\tif v.ExpiringTimeNano != nil {\n\t\tenc.AddInt64(\"expiringTimeNano\", *v.ExpiringTimeNano)\n\t}\n\tif v.Resettable != nil {\n\t\tenc.AddBool(\"resettable\", *v.Resettable)\n\t}\n\treturn err\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *ResetPointInfo) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *ResetPointInfo) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetFirstDecisionCompletedId returns the value of FirstDecisionCompletedId if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetFirstDecisionCompletedId() (o int64) {\n\tif v != nil && v.FirstDecisionCompletedId != nil {\n\t\treturn *v.FirstDecisionCompletedId\n\t}\n\n\treturn\n}\n\n// IsSetFirstDecisionCompletedId returns true if FirstDecisionCompletedId is not nil.\nfunc (v *ResetPointInfo) IsSetFirstDecisionCompletedId() bool {\n\treturn v != nil && v.FirstDecisionCompletedId != nil\n}\n\n// GetCreatedTimeNano returns the value of CreatedTimeNano if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetCreatedTimeNano() (o int64) {\n\tif v != nil && v.CreatedTimeNano != nil {\n\t\treturn *v.CreatedTimeNano\n\t}\n\n\treturn\n}\n\n// IsSetCreatedTimeNano returns true if CreatedTimeNano is not nil.\nfunc (v *ResetPointInfo) IsSetCreatedTimeNano() bool {\n\treturn v != nil && v.CreatedTimeNano != nil\n}\n\n// GetExpiringTimeNano returns the value of ExpiringTimeNano if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetExpiringTimeNano() (o int64) {\n\tif v != nil && v.ExpiringTimeNano != nil {\n\t\treturn *v.ExpiringTimeNano\n\t}\n\n\treturn\n}\n\n// IsSetExpiringTimeNano returns true if ExpiringTimeNano is not nil.\nfunc (v *ResetPointInfo) IsSetExpiringTimeNano() bool {\n\treturn v != nil && v.ExpiringTimeNano != nil\n}\n\n// GetResettable returns the value of Resettable if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPointInfo) GetResettable() (o bool) {\n\tif v != nil && v.Resettable != nil {\n\t\treturn *v.Resettable\n\t}\n\n\treturn\n}\n\n// IsSetResettable returns true if Resettable is not nil.\nfunc (v *ResetPointInfo) IsSetResettable() bool {\n\treturn v != nil && v.Resettable != nil\n}\n\ntype ResetPoints struct {\n\tPoints []*ResetPointInfo `json:\"points,omitempty\"`\n}\n\ntype _List_ResetPointInfo_ValueList []*ResetPointInfo\n\nfunc (v _List_ResetPointInfo_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ResetPointInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_ResetPointInfo_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_ResetPointInfo_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_ResetPointInfo_ValueList) Close() {}\n\n// ToWire translates a ResetPoints struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetPoints) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Points != nil {\n\t\tw, err = wire.NewValueList(_List_ResetPointInfo_ValueList(v.Points)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetPointInfo_Read(w wire.Value) (*ResetPointInfo, error) {\n\tvar v ResetPointInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_ResetPointInfo_Read(l wire.ValueList) ([]*ResetPointInfo, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*ResetPointInfo, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _ResetPointInfo_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ResetPoints struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetPoints struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetPoints\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetPoints) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Points, err = _List_ResetPointInfo_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_ResetPointInfo_Encode(val []*ResetPointInfo, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*ResetPointInfo', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ResetPoints struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetPoints struct could not be encoded.\nfunc (v *ResetPoints) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Points != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_ResetPointInfo_Encode(v.Points, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetPointInfo_Decode(sr stream.Reader) (*ResetPointInfo, error) {\n\tvar v ResetPointInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_ResetPointInfo_Decode(sr stream.Reader) ([]*ResetPointInfo, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*ResetPointInfo, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _ResetPointInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ResetPoints struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetPoints struct could not be generated from the wire\n// representation.\nfunc (v *ResetPoints) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Points, err = _List_ResetPointInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetPoints\n// struct.\nfunc (v *ResetPoints) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Points != nil {\n\t\tfields[i] = fmt.Sprintf(\"Points: %v\", v.Points)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetPoints{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_ResetPointInfo_Equals(lhs, rhs []*ResetPointInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ResetPoints match the\n// provided ResetPoints.\n//\n// This function performs a deep comparison.\nfunc (v *ResetPoints) Equals(rhs *ResetPoints) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Points == nil && rhs.Points == nil) || (v.Points != nil && rhs.Points != nil && _List_ResetPointInfo_Equals(v.Points, rhs.Points))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_ResetPointInfo_Zapper []*ResetPointInfo\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_ResetPointInfo_Zapper.\nfunc (l _List_ResetPointInfo_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetPoints.\nfunc (v *ResetPoints) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Points != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"points\", (_List_ResetPointInfo_Zapper)(v.Points)))\n\t}\n\treturn err\n}\n\n// GetPoints returns the value of Points if it is set or its\n// zero value if it is unset.\nfunc (v *ResetPoints) GetPoints() (o []*ResetPointInfo) {\n\tif v != nil && v.Points != nil {\n\t\treturn v.Points\n\t}\n\n\treturn\n}\n\n// IsSetPoints returns true if Points is not nil.\nfunc (v *ResetPoints) IsSetPoints() bool {\n\treturn v != nil && v.Points != nil\n}\n\ntype ResetQueueRequest struct {\n\tShardID     *int32  `json:\"shardID,omitempty\"`\n\tClusterName *string `json:\"clusterName,omitempty\"`\n\tType        *int32  `json:\"type,omitempty\"`\n}\n\n// ToWire translates a ResetQueueRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetQueueRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tw, err = wire.NewValueI32(*(v.Type)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetQueueRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetQueueRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetQueueRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetQueueRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetQueueRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetQueueRequest struct could not be encoded.\nfunc (v *ResetQueueRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Type)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetQueueRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetQueueRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResetQueueRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetQueueRequest\n// struct.\nfunc (v *ResetQueueRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.ClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterName: %v\", *(v.ClusterName))\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetQueueRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetQueueRequest match the\n// provided ResetQueueRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResetQueueRequest) Equals(rhs *ResetQueueRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClusterName, rhs.ClusterName) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetQueueRequest.\nfunc (v *ResetQueueRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.ClusterName != nil {\n\t\tenc.AddString(\"clusterName\", *v.ClusterName)\n\t}\n\tif v.Type != nil {\n\t\tenc.AddInt32(\"type\", *v.Type)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *ResetQueueRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *ResetQueueRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetClusterName returns the value of ClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *ResetQueueRequest) GetClusterName() (o string) {\n\tif v != nil && v.ClusterName != nil {\n\t\treturn *v.ClusterName\n\t}\n\n\treturn\n}\n\n// IsSetClusterName returns true if ClusterName is not nil.\nfunc (v *ResetQueueRequest) IsSetClusterName() bool {\n\treturn v != nil && v.ClusterName != nil\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *ResetQueueRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *ResetQueueRequest) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\ntype ResetStickyTaskListRequest struct {\n\tDomain    *string            `json:\"domain,omitempty\"`\n\tExecution *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// ToWire translates a ResetStickyTaskListRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetStickyTaskListRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetStickyTaskListRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetStickyTaskListRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetStickyTaskListRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetStickyTaskListRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetStickyTaskListRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetStickyTaskListRequest struct could not be encoded.\nfunc (v *ResetStickyTaskListRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetStickyTaskListRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetStickyTaskListRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResetStickyTaskListRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetStickyTaskListRequest\n// struct.\nfunc (v *ResetStickyTaskListRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetStickyTaskListRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetStickyTaskListRequest match the\n// provided ResetStickyTaskListRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResetStickyTaskListRequest) Equals(rhs *ResetStickyTaskListRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetStickyTaskListRequest.\nfunc (v *ResetStickyTaskListRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ResetStickyTaskListRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ResetStickyTaskListRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *ResetStickyTaskListRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *ResetStickyTaskListRequest) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\ntype ResetStickyTaskListResponse struct {\n}\n\n// ToWire translates a ResetStickyTaskListResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetStickyTaskListResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetStickyTaskListResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetStickyTaskListResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetStickyTaskListResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetStickyTaskListResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetStickyTaskListResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetStickyTaskListResponse struct could not be encoded.\nfunc (v *ResetStickyTaskListResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetStickyTaskListResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetStickyTaskListResponse struct could not be generated from the wire\n// representation.\nfunc (v *ResetStickyTaskListResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetStickyTaskListResponse\n// struct.\nfunc (v *ResetStickyTaskListResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"ResetStickyTaskListResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetStickyTaskListResponse match the\n// provided ResetStickyTaskListResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ResetStickyTaskListResponse) Equals(rhs *ResetStickyTaskListResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetStickyTaskListResponse.\nfunc (v *ResetStickyTaskListResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype ResetWorkflowExecutionRequest struct {\n\tDomain                *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution     *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tReason                *string            `json:\"reason,omitempty\"`\n\tDecisionFinishEventId *int64             `json:\"decisionFinishEventId,omitempty\"`\n\tRequestId             *string            `json:\"requestId,omitempty\"`\n\tSkipSignalReapply     *bool              `json:\"skipSignalReapply,omitempty\"`\n}\n\n// ToWire translates a ResetWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionFinishEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionFinishEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.SkipSignalReapply != nil {\n\t\tw, err = wire.NewValueBool(*(v.SkipSignalReapply)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionFinishEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.SkipSignalReapply = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetWorkflowExecutionRequest struct could not be encoded.\nfunc (v *ResetWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionFinishEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionFinishEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SkipSignalReapply != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.SkipSignalReapply)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *ResetWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionFinishEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.SkipSignalReapply = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetWorkflowExecutionRequest\n// struct.\nfunc (v *ResetWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.DecisionFinishEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionFinishEventId: %v\", *(v.DecisionFinishEventId))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.SkipSignalReapply != nil {\n\t\tfields[i] = fmt.Sprintf(\"SkipSignalReapply: %v\", *(v.SkipSignalReapply))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetWorkflowExecutionRequest match the\n// provided ResetWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *ResetWorkflowExecutionRequest) Equals(rhs *ResetWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionFinishEventId, rhs.DecisionFinishEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.SkipSignalReapply, rhs.SkipSignalReapply) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetWorkflowExecutionRequest.\nfunc (v *ResetWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.DecisionFinishEventId != nil {\n\t\tenc.AddInt64(\"decisionFinishEventId\", *v.DecisionFinishEventId)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.SkipSignalReapply != nil {\n\t\tenc.AddBool(\"skipSignalReapply\", *v.SkipSignalReapply)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDecisionFinishEventId returns the value of DecisionFinishEventId if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetDecisionFinishEventId() (o int64) {\n\tif v != nil && v.DecisionFinishEventId != nil {\n\t\treturn *v.DecisionFinishEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionFinishEventId returns true if DecisionFinishEventId is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetDecisionFinishEventId() bool {\n\treturn v != nil && v.DecisionFinishEventId != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetSkipSignalReapply returns the value of SkipSignalReapply if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionRequest) GetSkipSignalReapply() (o bool) {\n\tif v != nil && v.SkipSignalReapply != nil {\n\t\treturn *v.SkipSignalReapply\n\t}\n\n\treturn\n}\n\n// IsSetSkipSignalReapply returns true if SkipSignalReapply is not nil.\nfunc (v *ResetWorkflowExecutionRequest) IsSetSkipSignalReapply() bool {\n\treturn v != nil && v.SkipSignalReapply != nil\n}\n\ntype ResetWorkflowExecutionResponse struct {\n\tRunId *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a ResetWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ResetWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ResetWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ResetWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ResetWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ResetWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ResetWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ResetWorkflowExecutionResponse struct could not be encoded.\nfunc (v *ResetWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ResetWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ResetWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *ResetWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ResetWorkflowExecutionResponse\n// struct.\nfunc (v *ResetWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ResetWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ResetWorkflowExecutionResponse match the\n// provided ResetWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *ResetWorkflowExecutionResponse) Equals(rhs *ResetWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ResetWorkflowExecutionResponse.\nfunc (v *ResetWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *ResetWorkflowExecutionResponse) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *ResetWorkflowExecutionResponse) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\ntype RespondActivityTaskCanceledByIDRequest struct {\n\tDomain     *string `json:\"domain,omitempty\"`\n\tWorkflowID *string `json:\"workflowID,omitempty\"`\n\tRunID      *string `json:\"runID,omitempty\"`\n\tActivityID *string `json:\"activityID,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tIdentity   *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCanceledByIDRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCanceledByIDRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskCanceledByIDRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCanceledByIDRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCanceledByIDRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCanceledByIDRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCanceledByIDRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledByIDRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCanceledByIDRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskCanceledByIDRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledByIDRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCanceledByIDRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCanceledByIDRequest\n// struct.\nfunc (v *RespondActivityTaskCanceledByIDRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCanceledByIDRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCanceledByIDRequest match the\n// provided RespondActivityTaskCanceledByIDRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCanceledByIDRequest) Equals(rhs *RespondActivityTaskCanceledByIDRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCanceledByIDRequest.\nfunc (v *RespondActivityTaskCanceledByIDRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskCanceledByIDRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondActivityTaskCanceledRequest struct {\n\tTaskToken []byte  `json:\"taskToken,omitempty\"`\n\tDetails   []byte  `json:\"details,omitempty\"`\n\tIdentity  *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCanceledRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCanceledRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskCanceledRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCanceledRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCanceledRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCanceledRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCanceledRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCanceledRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskCanceledRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCanceledRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCanceledRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCanceledRequest\n// struct.\nfunc (v *RespondActivityTaskCanceledRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCanceledRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCanceledRequest match the\n// provided RespondActivityTaskCanceledRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCanceledRequest) Equals(rhs *RespondActivityTaskCanceledRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCanceledRequest.\nfunc (v *RespondActivityTaskCanceledRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondActivityTaskCanceledRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RespondActivityTaskCanceledRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCanceledRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskCanceledRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondActivityTaskCompletedByIDRequest struct {\n\tDomain     *string `json:\"domain,omitempty\"`\n\tWorkflowID *string `json:\"workflowID,omitempty\"`\n\tRunID      *string `json:\"runID,omitempty\"`\n\tActivityID *string `json:\"activityID,omitempty\"`\n\tResult     []byte  `json:\"result,omitempty\"`\n\tIdentity   *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCompletedByIDRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCompletedByIDRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskCompletedByIDRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCompletedByIDRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCompletedByIDRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCompletedByIDRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCompletedByIDRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedByIDRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCompletedByIDRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskCompletedByIDRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedByIDRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCompletedByIDRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCompletedByIDRequest\n// struct.\nfunc (v *RespondActivityTaskCompletedByIDRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCompletedByIDRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCompletedByIDRequest match the\n// provided RespondActivityTaskCompletedByIDRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCompletedByIDRequest) Equals(rhs *RespondActivityTaskCompletedByIDRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCompletedByIDRequest.\nfunc (v *RespondActivityTaskCompletedByIDRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskCompletedByIDRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondActivityTaskCompletedRequest struct {\n\tTaskToken []byte  `json:\"taskToken,omitempty\"`\n\tResult    []byte  `json:\"result,omitempty\"`\n\tIdentity  *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondActivityTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskCompletedRequest\n// struct.\nfunc (v *RespondActivityTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskCompletedRequest match the\n// provided RespondActivityTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskCompletedRequest) Equals(rhs *RespondActivityTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskCompletedRequest.\nfunc (v *RespondActivityTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondActivityTaskCompletedRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedRequest) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *RespondActivityTaskCompletedRequest) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskCompletedRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskCompletedRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondActivityTaskFailedByIDRequest struct {\n\tDomain     *string `json:\"domain,omitempty\"`\n\tWorkflowID *string `json:\"workflowID,omitempty\"`\n\tRunID      *string `json:\"runID,omitempty\"`\n\tActivityID *string `json:\"activityID,omitempty\"`\n\tReason     *string `json:\"reason,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tIdentity   *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskFailedByIDRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskFailedByIDRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueString(*(v.RunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskFailedByIDRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskFailedByIDRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskFailedByIDRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskFailedByIDRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskFailedByIDRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskFailedByIDRequest struct could not be encoded.\nfunc (v *RespondActivityTaskFailedByIDRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskFailedByIDRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskFailedByIDRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskFailedByIDRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskFailedByIDRequest\n// struct.\nfunc (v *RespondActivityTaskFailedByIDRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", *(v.RunID))\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskFailedByIDRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskFailedByIDRequest match the\n// provided RespondActivityTaskFailedByIDRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskFailedByIDRequest) Equals(rhs *RespondActivityTaskFailedByIDRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunID, rhs.RunID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskFailedByIDRequest.\nfunc (v *RespondActivityTaskFailedByIDRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", *v.RunID)\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetRunID() (o string) {\n\tif v != nil && v.RunID != nil {\n\t\treturn *v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedByIDRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskFailedByIDRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondActivityTaskFailedRequest struct {\n\tTaskToken []byte  `json:\"taskToken,omitempty\"`\n\tReason    *string `json:\"reason,omitempty\"`\n\tDetails   []byte  `json:\"details,omitempty\"`\n\tIdentity  *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RespondActivityTaskFailedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondActivityTaskFailedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondActivityTaskFailedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondActivityTaskFailedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondActivityTaskFailedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondActivityTaskFailedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondActivityTaskFailedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondActivityTaskFailedRequest struct could not be encoded.\nfunc (v *RespondActivityTaskFailedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondActivityTaskFailedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondActivityTaskFailedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondActivityTaskFailedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondActivityTaskFailedRequest\n// struct.\nfunc (v *RespondActivityTaskFailedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondActivityTaskFailedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondActivityTaskFailedRequest match the\n// provided RespondActivityTaskFailedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondActivityTaskFailedRequest) Equals(rhs *RespondActivityTaskFailedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondActivityTaskFailedRequest.\nfunc (v *RespondActivityTaskFailedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondActivityTaskFailedRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondActivityTaskFailedRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RespondCrossClusterTasksCompletedRequest struct {\n\tShardID       *int32                      `json:\"shardID,omitempty\"`\n\tTargetCluster *string                     `json:\"targetCluster,omitempty\"`\n\tTaskResponses []*CrossClusterTaskResponse `json:\"taskResponses,omitempty\"`\n\tFetchNewTasks *bool                       `json:\"fetchNewTasks,omitempty\"`\n}\n\ntype _List_CrossClusterTaskResponse_ValueList []*CrossClusterTaskResponse\n\nfunc (v _List_CrossClusterTaskResponse_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*CrossClusterTaskResponse', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_CrossClusterTaskResponse_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_CrossClusterTaskResponse_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_CrossClusterTaskResponse_ValueList) Close() {}\n\n// ToWire translates a RespondCrossClusterTasksCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondCrossClusterTasksCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ShardID != nil {\n\t\tw, err = wire.NewValueI32(*(v.ShardID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TargetCluster != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetCluster)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskResponses != nil {\n\t\tw, err = wire.NewValueList(_List_CrossClusterTaskResponse_ValueList(v.TaskResponses)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FetchNewTasks != nil {\n\t\tw, err = wire.NewValueBool(*(v.FetchNewTasks)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _CrossClusterTaskResponse_Read(w wire.Value) (*CrossClusterTaskResponse, error) {\n\tvar v CrossClusterTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_CrossClusterTaskResponse_Read(l wire.ValueList) ([]*CrossClusterTaskResponse, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*CrossClusterTaskResponse, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _CrossClusterTaskResponse_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RespondCrossClusterTasksCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondCrossClusterTasksCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondCrossClusterTasksCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondCrossClusterTasksCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ShardID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetCluster = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.TaskResponses, err = _List_CrossClusterTaskResponse_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.FetchNewTasks = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_CrossClusterTaskResponse_Encode(val []*CrossClusterTaskResponse, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*CrossClusterTaskResponse', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a RespondCrossClusterTasksCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondCrossClusterTasksCompletedRequest struct could not be encoded.\nfunc (v *RespondCrossClusterTasksCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ShardID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ShardID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetCluster != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetCluster)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskResponses != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_CrossClusterTaskResponse_Encode(v.TaskResponses, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FetchNewTasks != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.FetchNewTasks)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _CrossClusterTaskResponse_Decode(sr stream.Reader) (*CrossClusterTaskResponse, error) {\n\tvar v CrossClusterTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_CrossClusterTaskResponse_Decode(sr stream.Reader) ([]*CrossClusterTaskResponse, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*CrossClusterTaskResponse, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _CrossClusterTaskResponse_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RespondCrossClusterTasksCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondCrossClusterTasksCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondCrossClusterTasksCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ShardID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetCluster = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TList:\n\t\t\tv.TaskResponses, err = _List_CrossClusterTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.FetchNewTasks = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondCrossClusterTasksCompletedRequest\n// struct.\nfunc (v *RespondCrossClusterTasksCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.ShardID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ShardID: %v\", *(v.ShardID))\n\t\ti++\n\t}\n\tif v.TargetCluster != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetCluster: %v\", *(v.TargetCluster))\n\t\ti++\n\t}\n\tif v.TaskResponses != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskResponses: %v\", v.TaskResponses)\n\t\ti++\n\t}\n\tif v.FetchNewTasks != nil {\n\t\tfields[i] = fmt.Sprintf(\"FetchNewTasks: %v\", *(v.FetchNewTasks))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondCrossClusterTasksCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_CrossClusterTaskResponse_Equals(lhs, rhs []*CrossClusterTaskResponse) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this RespondCrossClusterTasksCompletedRequest match the\n// provided RespondCrossClusterTasksCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondCrossClusterTasksCompletedRequest) Equals(rhs *RespondCrossClusterTasksCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ShardID, rhs.ShardID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetCluster, rhs.TargetCluster) {\n\t\treturn false\n\t}\n\tif !((v.TaskResponses == nil && rhs.TaskResponses == nil) || (v.TaskResponses != nil && rhs.TaskResponses != nil && _List_CrossClusterTaskResponse_Equals(v.TaskResponses, rhs.TaskResponses))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.FetchNewTasks, rhs.FetchNewTasks) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_CrossClusterTaskResponse_Zapper []*CrossClusterTaskResponse\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_CrossClusterTaskResponse_Zapper.\nfunc (l _List_CrossClusterTaskResponse_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondCrossClusterTasksCompletedRequest.\nfunc (v *RespondCrossClusterTasksCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ShardID != nil {\n\t\tenc.AddInt32(\"shardID\", *v.ShardID)\n\t}\n\tif v.TargetCluster != nil {\n\t\tenc.AddString(\"targetCluster\", *v.TargetCluster)\n\t}\n\tif v.TaskResponses != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"taskResponses\", (_List_CrossClusterTaskResponse_Zapper)(v.TaskResponses)))\n\t}\n\tif v.FetchNewTasks != nil {\n\t\tenc.AddBool(\"fetchNewTasks\", *v.FetchNewTasks)\n\t}\n\treturn err\n}\n\n// GetShardID returns the value of ShardID if it is set or its\n// zero value if it is unset.\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetShardID() (o int32) {\n\tif v != nil && v.ShardID != nil {\n\t\treturn *v.ShardID\n\t}\n\n\treturn\n}\n\n// IsSetShardID returns true if ShardID is not nil.\nfunc (v *RespondCrossClusterTasksCompletedRequest) IsSetShardID() bool {\n\treturn v != nil && v.ShardID != nil\n}\n\n// GetTargetCluster returns the value of TargetCluster if it is set or its\n// zero value if it is unset.\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetTargetCluster() (o string) {\n\tif v != nil && v.TargetCluster != nil {\n\t\treturn *v.TargetCluster\n\t}\n\n\treturn\n}\n\n// IsSetTargetCluster returns true if TargetCluster is not nil.\nfunc (v *RespondCrossClusterTasksCompletedRequest) IsSetTargetCluster() bool {\n\treturn v != nil && v.TargetCluster != nil\n}\n\n// GetTaskResponses returns the value of TaskResponses if it is set or its\n// zero value if it is unset.\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetTaskResponses() (o []*CrossClusterTaskResponse) {\n\tif v != nil && v.TaskResponses != nil {\n\t\treturn v.TaskResponses\n\t}\n\n\treturn\n}\n\n// IsSetTaskResponses returns true if TaskResponses is not nil.\nfunc (v *RespondCrossClusterTasksCompletedRequest) IsSetTaskResponses() bool {\n\treturn v != nil && v.TaskResponses != nil\n}\n\n// GetFetchNewTasks returns the value of FetchNewTasks if it is set or its\n// zero value if it is unset.\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetFetchNewTasks() (o bool) {\n\tif v != nil && v.FetchNewTasks != nil {\n\t\treturn *v.FetchNewTasks\n\t}\n\n\treturn\n}\n\n// IsSetFetchNewTasks returns true if FetchNewTasks is not nil.\nfunc (v *RespondCrossClusterTasksCompletedRequest) IsSetFetchNewTasks() bool {\n\treturn v != nil && v.FetchNewTasks != nil\n}\n\ntype RespondCrossClusterTasksCompletedResponse struct {\n\tTasks []*CrossClusterTaskRequest `json:\"tasks,omitempty\"`\n}\n\n// ToWire translates a RespondCrossClusterTasksCompletedResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondCrossClusterTasksCompletedResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Tasks != nil {\n\t\tw, err = wire.NewValueList(_List_CrossClusterTaskRequest_ValueList(v.Tasks)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondCrossClusterTasksCompletedResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondCrossClusterTasksCompletedResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondCrossClusterTasksCompletedResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondCrossClusterTasksCompletedResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Tasks, err = _List_CrossClusterTaskRequest_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondCrossClusterTasksCompletedResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondCrossClusterTasksCompletedResponse struct could not be encoded.\nfunc (v *RespondCrossClusterTasksCompletedResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Tasks != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_CrossClusterTaskRequest_Encode(v.Tasks, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondCrossClusterTasksCompletedResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondCrossClusterTasksCompletedResponse struct could not be generated from the wire\n// representation.\nfunc (v *RespondCrossClusterTasksCompletedResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.Tasks, err = _List_CrossClusterTaskRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondCrossClusterTasksCompletedResponse\n// struct.\nfunc (v *RespondCrossClusterTasksCompletedResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Tasks != nil {\n\t\tfields[i] = fmt.Sprintf(\"Tasks: %v\", v.Tasks)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondCrossClusterTasksCompletedResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondCrossClusterTasksCompletedResponse match the\n// provided RespondCrossClusterTasksCompletedResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RespondCrossClusterTasksCompletedResponse) Equals(rhs *RespondCrossClusterTasksCompletedResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Tasks == nil && rhs.Tasks == nil) || (v.Tasks != nil && rhs.Tasks != nil && _List_CrossClusterTaskRequest_Equals(v.Tasks, rhs.Tasks))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondCrossClusterTasksCompletedResponse.\nfunc (v *RespondCrossClusterTasksCompletedResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Tasks != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"tasks\", (_List_CrossClusterTaskRequest_Zapper)(v.Tasks)))\n\t}\n\treturn err\n}\n\n// GetTasks returns the value of Tasks if it is set or its\n// zero value if it is unset.\nfunc (v *RespondCrossClusterTasksCompletedResponse) GetTasks() (o []*CrossClusterTaskRequest) {\n\tif v != nil && v.Tasks != nil {\n\t\treturn v.Tasks\n\t}\n\n\treturn\n}\n\n// IsSetTasks returns true if Tasks is not nil.\nfunc (v *RespondCrossClusterTasksCompletedResponse) IsSetTasks() bool {\n\treturn v != nil && v.Tasks != nil\n}\n\ntype RespondDecisionTaskCompletedRequest struct {\n\tTaskToken                  []byte                          `json:\"taskToken,omitempty\"`\n\tDecisions                  []*Decision                     `json:\"decisions,omitempty\"`\n\tExecutionContext           []byte                          `json:\"executionContext,omitempty\"`\n\tIdentity                   *string                         `json:\"identity,omitempty\"`\n\tStickyAttributes           *StickyExecutionAttributes      `json:\"stickyAttributes,omitempty\"`\n\tReturnNewDecisionTask      *bool                           `json:\"returnNewDecisionTask,omitempty\"`\n\tForceCreateNewDecisionTask *bool                           `json:\"forceCreateNewDecisionTask,omitempty\"`\n\tBinaryChecksum             *string                         `json:\"binaryChecksum,omitempty\"`\n\tQueryResults               map[string]*WorkflowQueryResult `json:\"queryResults,omitempty\"`\n}\n\ntype _List_Decision_ValueList []*Decision\n\nfunc (v _List_Decision_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*Decision', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_Decision_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_Decision_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_Decision_ValueList) Close() {}\n\ntype _Map_String_WorkflowQueryResult_MapItemList map[string]*WorkflowQueryResult\n\nfunc (m _Map_String_WorkflowQueryResult_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WorkflowQueryResult', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_WorkflowQueryResult_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_WorkflowQueryResult_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_WorkflowQueryResult_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_WorkflowQueryResult_MapItemList) Close() {}\n\n// ToWire translates a RespondDecisionTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Decisions != nil {\n\t\tw, err = wire.NewValueList(_List_Decision_ValueList(v.Decisions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionContext != nil {\n\t\tw, err = wire.NewValueBinary(v.ExecutionContext), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.StickyAttributes != nil {\n\t\tw, err = v.StickyAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ReturnNewDecisionTask != nil {\n\t\tw, err = wire.NewValueBool(*(v.ReturnNewDecisionTask)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ForceCreateNewDecisionTask != nil {\n\t\tw, err = wire.NewValueBool(*(v.ForceCreateNewDecisionTask)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.QueryResults != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_WorkflowQueryResult_MapItemList(v.QueryResults)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Decision_Read(w wire.Value) (*Decision, error) {\n\tvar v Decision\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_Decision_Read(l wire.ValueList) ([]*Decision, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*Decision, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _Decision_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\nfunc _StickyExecutionAttributes_Read(w wire.Value) (*StickyExecutionAttributes, error) {\n\tvar v StickyExecutionAttributes\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowQueryResult_Read(w wire.Value) (*WorkflowQueryResult, error) {\n\tvar v WorkflowQueryResult\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQueryResult_Read(m wire.MapItemList) (map[string]*WorkflowQueryResult, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*WorkflowQueryResult, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _WorkflowQueryResult_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RespondDecisionTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Decisions, err = _List_Decision_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ExecutionContext, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StickyAttributes, err = _StickyExecutionAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ReturnNewDecisionTask = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ForceCreateNewDecisionTask = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.QueryResults, err = _Map_String_WorkflowQueryResult_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_Decision_Encode(val []*Decision, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*Decision', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\nfunc _Map_String_WorkflowQueryResult_Encode(val map[string]*WorkflowQueryResult, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*WorkflowQueryResult', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a RespondDecisionTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondDecisionTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Decisions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_Decision_Encode(v.Decisions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionContext != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ExecutionContext); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StickyAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReturnNewDecisionTask != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ReturnNewDecisionTask)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ForceCreateNewDecisionTask != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ForceCreateNewDecisionTask)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryResults != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_WorkflowQueryResult_Encode(v.QueryResults, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Decision_Decode(sr stream.Reader) (*Decision, error) {\n\tvar v Decision\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_Decision_Decode(sr stream.Reader) ([]*Decision, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*Decision, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _Decision_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _StickyExecutionAttributes_Decode(sr stream.Reader) (*StickyExecutionAttributes, error) {\n\tvar v StickyExecutionAttributes\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowQueryResult_Decode(sr stream.Reader) (*WorkflowQueryResult, error) {\n\tvar v WorkflowQueryResult\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_WorkflowQueryResult_Decode(sr stream.Reader) (map[string]*WorkflowQueryResult, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*WorkflowQueryResult, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _WorkflowQueryResult_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RespondDecisionTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Decisions, err = _List_Decision_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.ExecutionContext, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.StickyAttributes, err = _StickyExecutionAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ReturnNewDecisionTask = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ForceCreateNewDecisionTask = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TMap:\n\t\t\tv.QueryResults, err = _Map_String_WorkflowQueryResult_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskCompletedRequest\n// struct.\nfunc (v *RespondDecisionTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Decisions != nil {\n\t\tfields[i] = fmt.Sprintf(\"Decisions: %v\", v.Decisions)\n\t\ti++\n\t}\n\tif v.ExecutionContext != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionContext: %v\", v.ExecutionContext)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.StickyAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyAttributes: %v\", v.StickyAttributes)\n\t\ti++\n\t}\n\tif v.ReturnNewDecisionTask != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReturnNewDecisionTask: %v\", *(v.ReturnNewDecisionTask))\n\t\ti++\n\t}\n\tif v.ForceCreateNewDecisionTask != nil {\n\t\tfields[i] = fmt.Sprintf(\"ForceCreateNewDecisionTask: %v\", *(v.ForceCreateNewDecisionTask))\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\tif v.QueryResults != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryResults: %v\", v.QueryResults)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_Decision_Equals(lhs, rhs []*Decision) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc _Map_String_WorkflowQueryResult_Equals(lhs, rhs map[string]*WorkflowQueryResult) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskCompletedRequest match the\n// provided RespondDecisionTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskCompletedRequest) Equals(rhs *RespondDecisionTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !((v.Decisions == nil && rhs.Decisions == nil) || (v.Decisions != nil && rhs.Decisions != nil && _List_Decision_Equals(v.Decisions, rhs.Decisions))) {\n\t\treturn false\n\t}\n\tif !((v.ExecutionContext == nil && rhs.ExecutionContext == nil) || (v.ExecutionContext != nil && rhs.ExecutionContext != nil && bytes.Equal(v.ExecutionContext, rhs.ExecutionContext))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !((v.StickyAttributes == nil && rhs.StickyAttributes == nil) || (v.StickyAttributes != nil && rhs.StickyAttributes != nil && v.StickyAttributes.Equals(rhs.StickyAttributes))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ReturnNewDecisionTask, rhs.ReturnNewDecisionTask) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ForceCreateNewDecisionTask, rhs.ForceCreateNewDecisionTask) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\tif !((v.QueryResults == nil && rhs.QueryResults == nil) || (v.QueryResults != nil && rhs.QueryResults != nil && _Map_String_WorkflowQueryResult_Equals(v.QueryResults, rhs.QueryResults))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_Decision_Zapper []*Decision\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_Decision_Zapper.\nfunc (l _List_Decision_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\ntype _Map_String_WorkflowQueryResult_Zapper map[string]*WorkflowQueryResult\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_WorkflowQueryResult_Zapper.\nfunc (m _Map_String_WorkflowQueryResult_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskCompletedRequest.\nfunc (v *RespondDecisionTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Decisions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"decisions\", (_List_Decision_Zapper)(v.Decisions)))\n\t}\n\tif v.ExecutionContext != nil {\n\t\tenc.AddString(\"executionContext\", base64.StdEncoding.EncodeToString(v.ExecutionContext))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.StickyAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"stickyAttributes\", v.StickyAttributes))\n\t}\n\tif v.ReturnNewDecisionTask != nil {\n\t\tenc.AddBool(\"returnNewDecisionTask\", *v.ReturnNewDecisionTask)\n\t}\n\tif v.ForceCreateNewDecisionTask != nil {\n\t\tenc.AddBool(\"forceCreateNewDecisionTask\", *v.ForceCreateNewDecisionTask)\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\tif v.QueryResults != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"queryResults\", (_Map_String_WorkflowQueryResult_Zapper)(v.QueryResults)))\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetDecisions returns the value of Decisions if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetDecisions() (o []*Decision) {\n\tif v != nil && v.Decisions != nil {\n\t\treturn v.Decisions\n\t}\n\n\treturn\n}\n\n// IsSetDecisions returns true if Decisions is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetDecisions() bool {\n\treturn v != nil && v.Decisions != nil\n}\n\n// GetExecutionContext returns the value of ExecutionContext if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetExecutionContext() (o []byte) {\n\tif v != nil && v.ExecutionContext != nil {\n\t\treturn v.ExecutionContext\n\t}\n\n\treturn\n}\n\n// IsSetExecutionContext returns true if ExecutionContext is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetExecutionContext() bool {\n\treturn v != nil && v.ExecutionContext != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetStickyAttributes returns the value of StickyAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetStickyAttributes() (o *StickyExecutionAttributes) {\n\tif v != nil && v.StickyAttributes != nil {\n\t\treturn v.StickyAttributes\n\t}\n\n\treturn\n}\n\n// IsSetStickyAttributes returns true if StickyAttributes is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetStickyAttributes() bool {\n\treturn v != nil && v.StickyAttributes != nil\n}\n\n// GetReturnNewDecisionTask returns the value of ReturnNewDecisionTask if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetReturnNewDecisionTask() (o bool) {\n\tif v != nil && v.ReturnNewDecisionTask != nil {\n\t\treturn *v.ReturnNewDecisionTask\n\t}\n\n\treturn\n}\n\n// IsSetReturnNewDecisionTask returns true if ReturnNewDecisionTask is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetReturnNewDecisionTask() bool {\n\treturn v != nil && v.ReturnNewDecisionTask != nil\n}\n\n// GetForceCreateNewDecisionTask returns the value of ForceCreateNewDecisionTask if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetForceCreateNewDecisionTask() (o bool) {\n\tif v != nil && v.ForceCreateNewDecisionTask != nil {\n\t\treturn *v.ForceCreateNewDecisionTask\n\t}\n\n\treturn\n}\n\n// IsSetForceCreateNewDecisionTask returns true if ForceCreateNewDecisionTask is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetForceCreateNewDecisionTask() bool {\n\treturn v != nil && v.ForceCreateNewDecisionTask != nil\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\n// GetQueryResults returns the value of QueryResults if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedRequest) GetQueryResults() (o map[string]*WorkflowQueryResult) {\n\tif v != nil && v.QueryResults != nil {\n\t\treturn v.QueryResults\n\t}\n\n\treturn\n}\n\n// IsSetQueryResults returns true if QueryResults is not nil.\nfunc (v *RespondDecisionTaskCompletedRequest) IsSetQueryResults() bool {\n\treturn v != nil && v.QueryResults != nil\n}\n\ntype RespondDecisionTaskCompletedResponse struct {\n\tDecisionTask                *PollForDecisionTaskResponse          `json:\"decisionTask,omitempty\"`\n\tActivitiesToDispatchLocally map[string]*ActivityLocalDispatchInfo `json:\"activitiesToDispatchLocally,omitempty\"`\n}\n\ntype _Map_String_ActivityLocalDispatchInfo_MapItemList map[string]*ActivityLocalDispatchInfo\n\nfunc (m _Map_String_ActivityLocalDispatchInfo_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ActivityLocalDispatchInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_ActivityLocalDispatchInfo_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_ActivityLocalDispatchInfo_MapItemList) Close() {}\n\n// ToWire translates a RespondDecisionTaskCompletedResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskCompletedResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTask != nil {\n\t\tw, err = v.DecisionTask.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_ActivityLocalDispatchInfo_MapItemList(v.ActivitiesToDispatchLocally)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _PollForDecisionTaskResponse_Read(w wire.Value) (*PollForDecisionTaskResponse, error) {\n\tvar v PollForDecisionTaskResponse\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _ActivityLocalDispatchInfo_Read(w wire.Value) (*ActivityLocalDispatchInfo, error) {\n\tvar v ActivityLocalDispatchInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Read(m wire.MapItemList) (map[string]*ActivityLocalDispatchInfo, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*ActivityLocalDispatchInfo, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _ActivityLocalDispatchInfo_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a RespondDecisionTaskCompletedResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskCompletedResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskCompletedResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskCompletedResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DecisionTask, err = _PollForDecisionTaskResponse_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ActivitiesToDispatchLocally, err = _Map_String_ActivityLocalDispatchInfo_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Encode(val map[string]*ActivityLocalDispatchInfo, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*ActivityLocalDispatchInfo', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a RespondDecisionTaskCompletedResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedResponse struct could not be encoded.\nfunc (v *RespondDecisionTaskCompletedResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTask != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DecisionTask.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_ActivityLocalDispatchInfo_Encode(v.ActivitiesToDispatchLocally, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _PollForDecisionTaskResponse_Decode(sr stream.Reader) (*PollForDecisionTaskResponse, error) {\n\tvar v PollForDecisionTaskResponse\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _ActivityLocalDispatchInfo_Decode(sr stream.Reader) (*ActivityLocalDispatchInfo, error) {\n\tvar v ActivityLocalDispatchInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Decode(sr stream.Reader) (map[string]*ActivityLocalDispatchInfo, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*ActivityLocalDispatchInfo, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _ActivityLocalDispatchInfo_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a RespondDecisionTaskCompletedResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskCompletedResponse struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskCompletedResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.DecisionTask, err = _PollForDecisionTaskResponse_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TMap:\n\t\t\tv.ActivitiesToDispatchLocally, err = _Map_String_ActivityLocalDispatchInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskCompletedResponse\n// struct.\nfunc (v *RespondDecisionTaskCompletedResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DecisionTask != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTask: %v\", v.DecisionTask)\n\t\ti++\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivitiesToDispatchLocally: %v\", v.ActivitiesToDispatchLocally)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskCompletedResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_ActivityLocalDispatchInfo_Equals(lhs, rhs map[string]*ActivityLocalDispatchInfo) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskCompletedResponse match the\n// provided RespondDecisionTaskCompletedResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskCompletedResponse) Equals(rhs *RespondDecisionTaskCompletedResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DecisionTask == nil && rhs.DecisionTask == nil) || (v.DecisionTask != nil && rhs.DecisionTask != nil && v.DecisionTask.Equals(rhs.DecisionTask))) {\n\t\treturn false\n\t}\n\tif !((v.ActivitiesToDispatchLocally == nil && rhs.ActivitiesToDispatchLocally == nil) || (v.ActivitiesToDispatchLocally != nil && rhs.ActivitiesToDispatchLocally != nil && _Map_String_ActivityLocalDispatchInfo_Equals(v.ActivitiesToDispatchLocally, rhs.ActivitiesToDispatchLocally))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_ActivityLocalDispatchInfo_Zapper map[string]*ActivityLocalDispatchInfo\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_ActivityLocalDispatchInfo_Zapper.\nfunc (m _Map_String_ActivityLocalDispatchInfo_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskCompletedResponse.\nfunc (v *RespondDecisionTaskCompletedResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTask != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"decisionTask\", v.DecisionTask))\n\t}\n\tif v.ActivitiesToDispatchLocally != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activitiesToDispatchLocally\", (_Map_String_ActivityLocalDispatchInfo_Zapper)(v.ActivitiesToDispatchLocally)))\n\t}\n\treturn err\n}\n\n// GetDecisionTask returns the value of DecisionTask if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedResponse) GetDecisionTask() (o *PollForDecisionTaskResponse) {\n\tif v != nil && v.DecisionTask != nil {\n\t\treturn v.DecisionTask\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTask returns true if DecisionTask is not nil.\nfunc (v *RespondDecisionTaskCompletedResponse) IsSetDecisionTask() bool {\n\treturn v != nil && v.DecisionTask != nil\n}\n\n// GetActivitiesToDispatchLocally returns the value of ActivitiesToDispatchLocally if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskCompletedResponse) GetActivitiesToDispatchLocally() (o map[string]*ActivityLocalDispatchInfo) {\n\tif v != nil && v.ActivitiesToDispatchLocally != nil {\n\t\treturn v.ActivitiesToDispatchLocally\n\t}\n\n\treturn\n}\n\n// IsSetActivitiesToDispatchLocally returns true if ActivitiesToDispatchLocally is not nil.\nfunc (v *RespondDecisionTaskCompletedResponse) IsSetActivitiesToDispatchLocally() bool {\n\treturn v != nil && v.ActivitiesToDispatchLocally != nil\n}\n\ntype RespondDecisionTaskFailedRequest struct {\n\tTaskToken      []byte                   `json:\"taskToken,omitempty\"`\n\tCause          *DecisionTaskFailedCause `json:\"cause,omitempty\"`\n\tDetails        []byte                   `json:\"details,omitempty\"`\n\tIdentity       *string                  `json:\"identity,omitempty\"`\n\tBinaryChecksum *string                  `json:\"binaryChecksum,omitempty\"`\n}\n\n// ToWire translates a RespondDecisionTaskFailedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondDecisionTaskFailedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tw, err = wire.NewValueString(*(v.BinaryChecksum)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RespondDecisionTaskFailedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondDecisionTaskFailedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondDecisionTaskFailedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondDecisionTaskFailedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x DecisionTaskFailedCause\n\t\t\t\tx, err = _DecisionTaskFailedCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BinaryChecksum = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondDecisionTaskFailedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondDecisionTaskFailedRequest struct could not be encoded.\nfunc (v *RespondDecisionTaskFailedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BinaryChecksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BinaryChecksum)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RespondDecisionTaskFailedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondDecisionTaskFailedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondDecisionTaskFailedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x DecisionTaskFailedCause\n\t\t\tx, err = _DecisionTaskFailedCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BinaryChecksum = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondDecisionTaskFailedRequest\n// struct.\nfunc (v *RespondDecisionTaskFailedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"BinaryChecksum: %v\", *(v.BinaryChecksum))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondDecisionTaskFailedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RespondDecisionTaskFailedRequest match the\n// provided RespondDecisionTaskFailedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondDecisionTaskFailedRequest) Equals(rhs *RespondDecisionTaskFailedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !_DecisionTaskFailedCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BinaryChecksum, rhs.BinaryChecksum) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondDecisionTaskFailedRequest.\nfunc (v *RespondDecisionTaskFailedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.BinaryChecksum != nil {\n\t\tenc.AddString(\"binaryChecksum\", *v.BinaryChecksum)\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetCause() (o DecisionTaskFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetBinaryChecksum returns the value of BinaryChecksum if it is set or its\n// zero value if it is unset.\nfunc (v *RespondDecisionTaskFailedRequest) GetBinaryChecksum() (o string) {\n\tif v != nil && v.BinaryChecksum != nil {\n\t\treturn *v.BinaryChecksum\n\t}\n\n\treturn\n}\n\n// IsSetBinaryChecksum returns true if BinaryChecksum is not nil.\nfunc (v *RespondDecisionTaskFailedRequest) IsSetBinaryChecksum() bool {\n\treturn v != nil && v.BinaryChecksum != nil\n}\n\ntype RespondQueryTaskCompletedRequest struct {\n\tTaskToken         []byte                  `json:\"taskToken,omitempty\"`\n\tCompletedType     *QueryTaskCompletedType `json:\"completedType,omitempty\"`\n\tQueryResult       []byte                  `json:\"queryResult,omitempty\"`\n\tErrorMessage      *string                 `json:\"errorMessage,omitempty\"`\n\tWorkerVersionInfo *WorkerVersionInfo      `json:\"workerVersionInfo,omitempty\"`\n}\n\n// ToWire translates a RespondQueryTaskCompletedRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RespondQueryTaskCompletedRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskToken != nil {\n\t\tw, err = wire.NewValueBinary(v.TaskToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.CompletedType != nil {\n\t\tw, err = v.CompletedType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.QueryResult != nil {\n\t\tw, err = wire.NewValueBinary(v.QueryResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ErrorMessage != nil {\n\t\tw, err = wire.NewValueString(*(v.ErrorMessage)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.WorkerVersionInfo != nil {\n\t\tw, err = v.WorkerVersionInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryTaskCompletedType_Read(w wire.Value) (QueryTaskCompletedType, error) {\n\tvar v QueryTaskCompletedType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _WorkerVersionInfo_Read(w wire.Value) (*WorkerVersionInfo, error) {\n\tvar v WorkerVersionInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a RespondQueryTaskCompletedRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RespondQueryTaskCompletedRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RespondQueryTaskCompletedRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RespondQueryTaskCompletedRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TaskToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryTaskCompletedType\n\t\t\t\tx, err = _QueryTaskCompletedType_Read(field.Value)\n\t\t\t\tv.CompletedType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.QueryResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ErrorMessage = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkerVersionInfo, err = _WorkerVersionInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RespondQueryTaskCompletedRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RespondQueryTaskCompletedRequest struct could not be encoded.\nfunc (v *RespondQueryTaskCompletedRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TaskToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletedType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CompletedType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.QueryResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ErrorMessage != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ErrorMessage)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkerVersionInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkerVersionInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryTaskCompletedType_Decode(sr stream.Reader) (QueryTaskCompletedType, error) {\n\tvar v QueryTaskCompletedType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _WorkerVersionInfo_Decode(sr stream.Reader) (*WorkerVersionInfo, error) {\n\tvar v WorkerVersionInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a RespondQueryTaskCompletedRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RespondQueryTaskCompletedRequest struct could not be generated from the wire\n// representation.\nfunc (v *RespondQueryTaskCompletedRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.TaskToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x QueryTaskCompletedType\n\t\t\tx, err = _QueryTaskCompletedType_Decode(sr)\n\t\t\tv.CompletedType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.QueryResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ErrorMessage = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TStruct:\n\t\t\tv.WorkerVersionInfo, err = _WorkerVersionInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RespondQueryTaskCompletedRequest\n// struct.\nfunc (v *RespondQueryTaskCompletedRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.TaskToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskToken: %v\", v.TaskToken)\n\t\ti++\n\t}\n\tif v.CompletedType != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletedType: %v\", *(v.CompletedType))\n\t\ti++\n\t}\n\tif v.QueryResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryResult: %v\", v.QueryResult)\n\t\ti++\n\t}\n\tif v.ErrorMessage != nil {\n\t\tfields[i] = fmt.Sprintf(\"ErrorMessage: %v\", *(v.ErrorMessage))\n\t\ti++\n\t}\n\tif v.WorkerVersionInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkerVersionInfo: %v\", v.WorkerVersionInfo)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RespondQueryTaskCompletedRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _QueryTaskCompletedType_EqualsPtr(lhs, rhs *QueryTaskCompletedType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this RespondQueryTaskCompletedRequest match the\n// provided RespondQueryTaskCompletedRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RespondQueryTaskCompletedRequest) Equals(rhs *RespondQueryTaskCompletedRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskToken == nil && rhs.TaskToken == nil) || (v.TaskToken != nil && rhs.TaskToken != nil && bytes.Equal(v.TaskToken, rhs.TaskToken))) {\n\t\treturn false\n\t}\n\tif !_QueryTaskCompletedType_EqualsPtr(v.CompletedType, rhs.CompletedType) {\n\t\treturn false\n\t}\n\tif !((v.QueryResult == nil && rhs.QueryResult == nil) || (v.QueryResult != nil && rhs.QueryResult != nil && bytes.Equal(v.QueryResult, rhs.QueryResult))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ErrorMessage, rhs.ErrorMessage) {\n\t\treturn false\n\t}\n\tif !((v.WorkerVersionInfo == nil && rhs.WorkerVersionInfo == nil) || (v.WorkerVersionInfo != nil && rhs.WorkerVersionInfo != nil && v.WorkerVersionInfo.Equals(rhs.WorkerVersionInfo))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RespondQueryTaskCompletedRequest.\nfunc (v *RespondQueryTaskCompletedRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskToken != nil {\n\t\tenc.AddString(\"taskToken\", base64.StdEncoding.EncodeToString(v.TaskToken))\n\t}\n\tif v.CompletedType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"completedType\", *v.CompletedType))\n\t}\n\tif v.QueryResult != nil {\n\t\tenc.AddString(\"queryResult\", base64.StdEncoding.EncodeToString(v.QueryResult))\n\t}\n\tif v.ErrorMessage != nil {\n\t\tenc.AddString(\"errorMessage\", *v.ErrorMessage)\n\t}\n\tif v.WorkerVersionInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workerVersionInfo\", v.WorkerVersionInfo))\n\t}\n\treturn err\n}\n\n// GetTaskToken returns the value of TaskToken if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\n\treturn\n}\n\n// IsSetTaskToken returns true if TaskToken is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetTaskToken() bool {\n\treturn v != nil && v.TaskToken != nil\n}\n\n// GetCompletedType returns the value of CompletedType if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetCompletedType() (o QueryTaskCompletedType) {\n\tif v != nil && v.CompletedType != nil {\n\t\treturn *v.CompletedType\n\t}\n\n\treturn\n}\n\n// IsSetCompletedType returns true if CompletedType is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetCompletedType() bool {\n\treturn v != nil && v.CompletedType != nil\n}\n\n// GetQueryResult returns the value of QueryResult if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetQueryResult() (o []byte) {\n\tif v != nil && v.QueryResult != nil {\n\t\treturn v.QueryResult\n\t}\n\n\treturn\n}\n\n// IsSetQueryResult returns true if QueryResult is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetQueryResult() bool {\n\treturn v != nil && v.QueryResult != nil\n}\n\n// GetErrorMessage returns the value of ErrorMessage if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetErrorMessage() (o string) {\n\tif v != nil && v.ErrorMessage != nil {\n\t\treturn *v.ErrorMessage\n\t}\n\n\treturn\n}\n\n// IsSetErrorMessage returns true if ErrorMessage is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetErrorMessage() bool {\n\treturn v != nil && v.ErrorMessage != nil\n}\n\n// GetWorkerVersionInfo returns the value of WorkerVersionInfo if it is set or its\n// zero value if it is unset.\nfunc (v *RespondQueryTaskCompletedRequest) GetWorkerVersionInfo() (o *WorkerVersionInfo) {\n\tif v != nil && v.WorkerVersionInfo != nil {\n\t\treturn v.WorkerVersionInfo\n\t}\n\n\treturn\n}\n\n// IsSetWorkerVersionInfo returns true if WorkerVersionInfo is not nil.\nfunc (v *RespondQueryTaskCompletedRequest) IsSetWorkerVersionInfo() bool {\n\treturn v != nil && v.WorkerVersionInfo != nil\n}\n\ntype RestartWorkflowExecutionRequest struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tReason            *string            `json:\"reason,omitempty\"`\n\tIdentity          *string            `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a RestartWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RestartWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RestartWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RestartWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RestartWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RestartWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RestartWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RestartWorkflowExecutionRequest struct could not be encoded.\nfunc (v *RestartWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RestartWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RestartWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *RestartWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RestartWorkflowExecutionRequest\n// struct.\nfunc (v *RestartWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RestartWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RestartWorkflowExecutionRequest match the\n// provided RestartWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *RestartWorkflowExecutionRequest) Equals(rhs *RestartWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RestartWorkflowExecutionRequest.\nfunc (v *RestartWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *RestartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *RestartWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *RestartWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *RestartWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *RestartWorkflowExecutionRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *RestartWorkflowExecutionRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *RestartWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *RestartWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype RestartWorkflowExecutionResponse struct {\n\tRunId *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a RestartWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RestartWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RestartWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RestartWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RestartWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RestartWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RestartWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RestartWorkflowExecutionResponse struct could not be encoded.\nfunc (v *RestartWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RestartWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RestartWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *RestartWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RestartWorkflowExecutionResponse\n// struct.\nfunc (v *RestartWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RestartWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RestartWorkflowExecutionResponse match the\n// provided RestartWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *RestartWorkflowExecutionResponse) Equals(rhs *RestartWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RestartWorkflowExecutionResponse.\nfunc (v *RestartWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *RestartWorkflowExecutionResponse) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *RestartWorkflowExecutionResponse) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\ntype RetryPolicy struct {\n\tInitialIntervalInSeconds    *int32   `json:\"initialIntervalInSeconds,omitempty\"`\n\tBackoffCoefficient          *float64 `json:\"backoffCoefficient,omitempty\"`\n\tMaximumIntervalInSeconds    *int32   `json:\"maximumIntervalInSeconds,omitempty\"`\n\tMaximumAttempts             *int32   `json:\"maximumAttempts,omitempty\"`\n\tNonRetriableErrorReasons    []string `json:\"nonRetriableErrorReasons,omitempty\"`\n\tExpirationIntervalInSeconds *int32   `json:\"expirationIntervalInSeconds,omitempty\"`\n}\n\n// ToWire translates a RetryPolicy struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RetryPolicy) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.InitialIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.InitialIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.BackoffCoefficient != nil {\n\t\tw, err = wire.NewValueDouble(*(v.BackoffCoefficient)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tw, err = wire.NewValueI32(*(v.MaximumAttempts)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.NonRetriableErrorReasons != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.NonRetriableErrorReasons)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExpirationIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RetryPolicy struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RetryPolicy struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RetryPolicy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RetryPolicy) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.InitialIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.BackoffCoefficient = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.MaximumAttempts = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.NonRetriableErrorReasons, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExpirationIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RetryPolicy struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RetryPolicy struct could not be encoded.\nfunc (v *RetryPolicy) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.InitialIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.InitialIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BackoffCoefficient != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.BackoffCoefficient)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.MaximumAttempts != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.MaximumAttempts)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NonRetriableErrorReasons != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.NonRetriableErrorReasons, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExpirationIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RetryPolicy struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RetryPolicy struct could not be generated from the wire\n// representation.\nfunc (v *RetryPolicy) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.InitialIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.BackoffCoefficient = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.MaximumAttempts = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TList:\n\t\t\tv.NonRetriableErrorReasons, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExpirationIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RetryPolicy\n// struct.\nfunc (v *RetryPolicy) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.InitialIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitialIntervalInSeconds: %v\", *(v.InitialIntervalInSeconds))\n\t\ti++\n\t}\n\tif v.BackoffCoefficient != nil {\n\t\tfields[i] = fmt.Sprintf(\"BackoffCoefficient: %v\", *(v.BackoffCoefficient))\n\t\ti++\n\t}\n\tif v.MaximumIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumIntervalInSeconds: %v\", *(v.MaximumIntervalInSeconds))\n\t\ti++\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaximumAttempts: %v\", *(v.MaximumAttempts))\n\t\ti++\n\t}\n\tif v.NonRetriableErrorReasons != nil {\n\t\tfields[i] = fmt.Sprintf(\"NonRetriableErrorReasons: %v\", v.NonRetriableErrorReasons)\n\t\ti++\n\t}\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpirationIntervalInSeconds: %v\", *(v.ExpirationIntervalInSeconds))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RetryPolicy{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RetryPolicy match the\n// provided RetryPolicy.\n//\n// This function performs a deep comparison.\nfunc (v *RetryPolicy) Equals(rhs *RetryPolicy) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.InitialIntervalInSeconds, rhs.InitialIntervalInSeconds) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.BackoffCoefficient, rhs.BackoffCoefficient) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumIntervalInSeconds, rhs.MaximumIntervalInSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.MaximumAttempts, rhs.MaximumAttempts) {\n\t\treturn false\n\t}\n\tif !((v.NonRetriableErrorReasons == nil && rhs.NonRetriableErrorReasons == nil) || (v.NonRetriableErrorReasons != nil && rhs.NonRetriableErrorReasons != nil && _List_String_Equals(v.NonRetriableErrorReasons, rhs.NonRetriableErrorReasons))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExpirationIntervalInSeconds, rhs.ExpirationIntervalInSeconds) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RetryPolicy.\nfunc (v *RetryPolicy) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.InitialIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"initialIntervalInSeconds\", *v.InitialIntervalInSeconds)\n\t}\n\tif v.BackoffCoefficient != nil {\n\t\tenc.AddFloat64(\"backoffCoefficient\", *v.BackoffCoefficient)\n\t}\n\tif v.MaximumIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"maximumIntervalInSeconds\", *v.MaximumIntervalInSeconds)\n\t}\n\tif v.MaximumAttempts != nil {\n\t\tenc.AddInt32(\"maximumAttempts\", *v.MaximumAttempts)\n\t}\n\tif v.NonRetriableErrorReasons != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"nonRetriableErrorReasons\", (_List_String_Zapper)(v.NonRetriableErrorReasons)))\n\t}\n\tif v.ExpirationIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"expirationIntervalInSeconds\", *v.ExpirationIntervalInSeconds)\n\t}\n\treturn err\n}\n\n// GetInitialIntervalInSeconds returns the value of InitialIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetInitialIntervalInSeconds() (o int32) {\n\tif v != nil && v.InitialIntervalInSeconds != nil {\n\t\treturn *v.InitialIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetInitialIntervalInSeconds returns true if InitialIntervalInSeconds is not nil.\nfunc (v *RetryPolicy) IsSetInitialIntervalInSeconds() bool {\n\treturn v != nil && v.InitialIntervalInSeconds != nil\n}\n\n// GetBackoffCoefficient returns the value of BackoffCoefficient if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetBackoffCoefficient() (o float64) {\n\tif v != nil && v.BackoffCoefficient != nil {\n\t\treturn *v.BackoffCoefficient\n\t}\n\n\treturn\n}\n\n// IsSetBackoffCoefficient returns true if BackoffCoefficient is not nil.\nfunc (v *RetryPolicy) IsSetBackoffCoefficient() bool {\n\treturn v != nil && v.BackoffCoefficient != nil\n}\n\n// GetMaximumIntervalInSeconds returns the value of MaximumIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetMaximumIntervalInSeconds() (o int32) {\n\tif v != nil && v.MaximumIntervalInSeconds != nil {\n\t\treturn *v.MaximumIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetMaximumIntervalInSeconds returns true if MaximumIntervalInSeconds is not nil.\nfunc (v *RetryPolicy) IsSetMaximumIntervalInSeconds() bool {\n\treturn v != nil && v.MaximumIntervalInSeconds != nil\n}\n\n// GetMaximumAttempts returns the value of MaximumAttempts if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetMaximumAttempts() (o int32) {\n\tif v != nil && v.MaximumAttempts != nil {\n\t\treturn *v.MaximumAttempts\n\t}\n\n\treturn\n}\n\n// IsSetMaximumAttempts returns true if MaximumAttempts is not nil.\nfunc (v *RetryPolicy) IsSetMaximumAttempts() bool {\n\treturn v != nil && v.MaximumAttempts != nil\n}\n\n// GetNonRetriableErrorReasons returns the value of NonRetriableErrorReasons if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetNonRetriableErrorReasons() (o []string) {\n\tif v != nil && v.NonRetriableErrorReasons != nil {\n\t\treturn v.NonRetriableErrorReasons\n\t}\n\n\treturn\n}\n\n// IsSetNonRetriableErrorReasons returns true if NonRetriableErrorReasons is not nil.\nfunc (v *RetryPolicy) IsSetNonRetriableErrorReasons() bool {\n\treturn v != nil && v.NonRetriableErrorReasons != nil\n}\n\n// GetExpirationIntervalInSeconds returns the value of ExpirationIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *RetryPolicy) GetExpirationIntervalInSeconds() (o int32) {\n\tif v != nil && v.ExpirationIntervalInSeconds != nil {\n\t\treturn *v.ExpirationIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExpirationIntervalInSeconds returns true if ExpirationIntervalInSeconds is not nil.\nfunc (v *RetryPolicy) IsSetExpirationIntervalInSeconds() bool {\n\treturn v != nil && v.ExpirationIntervalInSeconds != nil\n}\n\ntype RetryTaskV2Error struct {\n\tMessage           string  `json:\"message,required\"`\n\tDomainId          *string `json:\"domainId,omitempty\"`\n\tWorkflowId        *string `json:\"workflowId,omitempty\"`\n\tRunId             *string `json:\"runId,omitempty\"`\n\tStartEventId      *int64  `json:\"startEventId,omitempty\"`\n\tStartEventVersion *int64  `json:\"startEventVersion,omitempty\"`\n\tEndEventId        *int64  `json:\"endEventId,omitempty\"`\n\tEndEventVersion   *int64  `json:\"endEventVersion,omitempty\"`\n}\n\n// ToWire translates a RetryTaskV2Error struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RetryTaskV2Error) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\tif v.DomainId != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 3, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 4, Value: w}\n\t\ti++\n\t}\n\tif v.StartEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 5, Value: w}\n\t\ti++\n\t}\n\tif v.StartEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 6, Value: w}\n\t\ti++\n\t}\n\tif v.EndEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 7, Value: w}\n\t\ti++\n\t}\n\tif v.EndEventVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndEventVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 8, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RetryTaskV2Error struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RetryTaskV2Error struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RetryTaskV2Error\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RetryTaskV2Error) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 4:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 5:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 6:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 7:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 8:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndEventVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of RetryTaskV2Error is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RetryTaskV2Error struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RetryTaskV2Error struct could not be encoded.\nfunc (v *RetryTaskV2Error) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 4, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 5, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 6, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 7, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndEventVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 8, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndEventVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RetryTaskV2Error struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RetryTaskV2Error struct could not be generated from the wire\n// representation.\nfunc (v *RetryTaskV2Error) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 3 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 4 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 5 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 6 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 7 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 8 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndEventVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of RetryTaskV2Error is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RetryTaskV2Error\n// struct.\nfunc (v *RetryTaskV2Error) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\tif v.DomainId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainId: %v\", *(v.DomainId))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\tif v.StartEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartEventId: %v\", *(v.StartEventId))\n\t\ti++\n\t}\n\tif v.StartEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartEventVersion: %v\", *(v.StartEventVersion))\n\t\ti++\n\t}\n\tif v.EndEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndEventId: %v\", *(v.EndEventId))\n\t\ti++\n\t}\n\tif v.EndEventVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndEventVersion: %v\", *(v.EndEventVersion))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RetryTaskV2Error{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*RetryTaskV2Error) ErrorName() string {\n\treturn \"RetryTaskV2Error\"\n}\n\n// Equals returns true if all the fields of this RetryTaskV2Error match the\n// provided RetryTaskV2Error.\n//\n// This function performs a deep comparison.\nfunc (v *RetryTaskV2Error) Equals(rhs *RetryTaskV2Error) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainId, rhs.DomainId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartEventId, rhs.StartEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartEventVersion, rhs.StartEventVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndEventId, rhs.EndEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndEventVersion, rhs.EndEventVersion) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RetryTaskV2Error.\nfunc (v *RetryTaskV2Error) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\tif v.DomainId != nil {\n\t\tenc.AddString(\"domainId\", *v.DomainId)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\tif v.StartEventId != nil {\n\t\tenc.AddInt64(\"startEventId\", *v.StartEventId)\n\t}\n\tif v.StartEventVersion != nil {\n\t\tenc.AddInt64(\"startEventVersion\", *v.StartEventVersion)\n\t}\n\tif v.EndEventId != nil {\n\t\tenc.AddInt64(\"endEventId\", *v.EndEventId)\n\t}\n\tif v.EndEventVersion != nil {\n\t\tenc.AddInt64(\"endEventVersion\", *v.EndEventVersion)\n\t}\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\n// GetDomainId returns the value of DomainId if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetDomainId() (o string) {\n\tif v != nil && v.DomainId != nil {\n\t\treturn *v.DomainId\n\t}\n\n\treturn\n}\n\n// IsSetDomainId returns true if DomainId is not nil.\nfunc (v *RetryTaskV2Error) IsSetDomainId() bool {\n\treturn v != nil && v.DomainId != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *RetryTaskV2Error) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *RetryTaskV2Error) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\n// GetStartEventId returns the value of StartEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetStartEventId() (o int64) {\n\tif v != nil && v.StartEventId != nil {\n\t\treturn *v.StartEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartEventId returns true if StartEventId is not nil.\nfunc (v *RetryTaskV2Error) IsSetStartEventId() bool {\n\treturn v != nil && v.StartEventId != nil\n}\n\n// GetStartEventVersion returns the value of StartEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetStartEventVersion() (o int64) {\n\tif v != nil && v.StartEventVersion != nil {\n\t\treturn *v.StartEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetStartEventVersion returns true if StartEventVersion is not nil.\nfunc (v *RetryTaskV2Error) IsSetStartEventVersion() bool {\n\treturn v != nil && v.StartEventVersion != nil\n}\n\n// GetEndEventId returns the value of EndEventId if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetEndEventId() (o int64) {\n\tif v != nil && v.EndEventId != nil {\n\t\treturn *v.EndEventId\n\t}\n\n\treturn\n}\n\n// IsSetEndEventId returns true if EndEventId is not nil.\nfunc (v *RetryTaskV2Error) IsSetEndEventId() bool {\n\treturn v != nil && v.EndEventId != nil\n}\n\n// GetEndEventVersion returns the value of EndEventVersion if it is set or its\n// zero value if it is unset.\nfunc (v *RetryTaskV2Error) GetEndEventVersion() (o int64) {\n\tif v != nil && v.EndEventVersion != nil {\n\t\treturn *v.EndEventVersion\n\t}\n\n\treturn\n}\n\n// IsSetEndEventVersion returns true if EndEventVersion is not nil.\nfunc (v *RetryTaskV2Error) IsSetEndEventVersion() bool {\n\treturn v != nil && v.EndEventVersion != nil\n}\n\nfunc (v *RetryTaskV2Error) Error() string {\n\treturn v.String()\n}\n\ntype ScheduleActivityTaskDecisionAttributes struct {\n\tActivityId                    *string       `json:\"activityId,omitempty\"`\n\tActivityType                  *ActivityType `json:\"activityType,omitempty\"`\n\tDomain                        *string       `json:\"domain,omitempty\"`\n\tTaskList                      *TaskList     `json:\"taskList,omitempty\"`\n\tInput                         []byte        `json:\"input,omitempty\"`\n\tScheduleToCloseTimeoutSeconds *int32        `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32        `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tStartToCloseTimeoutSeconds    *int32        `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds       *int32        `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tRetryPolicy                   *RetryPolicy  `json:\"retryPolicy,omitempty\"`\n\tHeader                        *Header       `json:\"header,omitempty\"`\n\tRequestLocalDispatch          *bool         `json:\"requestLocalDispatch,omitempty\"`\n}\n\n// ToWire translates a ScheduleActivityTaskDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ScheduleActivityTaskDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [12]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ActivityId != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tw, err = v.ActivityType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 25, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 45, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.StartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.HeartbeatTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.RequestLocalDispatch != nil {\n\t\tw, err = wire.NewValueBool(*(v.RequestLocalDispatch)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ScheduleActivityTaskDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ScheduleActivityTaskDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ScheduleActivityTaskDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ScheduleActivityTaskDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActivityType, err = _ActivityType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 25:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 45:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.RequestLocalDispatch = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ScheduleActivityTaskDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ScheduleActivityTaskDecisionAttributes struct could not be encoded.\nfunc (v *ScheduleActivityTaskDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ActivityId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActivityType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 25, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 45, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.HeartbeatTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestLocalDispatch != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.RequestLocalDispatch)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ScheduleActivityTaskDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ScheduleActivityTaskDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *ScheduleActivityTaskDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ActivityType, err = _ActivityType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 25 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 45 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.RequestLocalDispatch = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ScheduleActivityTaskDecisionAttributes\n// struct.\nfunc (v *ScheduleActivityTaskDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [12]string\n\ti := 0\n\tif v.ActivityId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityId: %v\", *(v.ActivityId))\n\t\ti++\n\t}\n\tif v.ActivityType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityType: %v\", v.ActivityType)\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToCloseTimeoutSeconds: %v\", *(v.ScheduleToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToCloseTimeoutSeconds: %v\", *(v.StartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatTimeoutSeconds: %v\", *(v.HeartbeatTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.RequestLocalDispatch != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestLocalDispatch: %v\", *(v.RequestLocalDispatch))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ScheduleActivityTaskDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ScheduleActivityTaskDecisionAttributes match the\n// provided ScheduleActivityTaskDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *ScheduleActivityTaskDecisionAttributes) Equals(rhs *ScheduleActivityTaskDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityId, rhs.ActivityId) {\n\t\treturn false\n\t}\n\tif !((v.ActivityType == nil && rhs.ActivityType == nil) || (v.ActivityType != nil && rhs.ActivityType != nil && v.ActivityType.Equals(rhs.ActivityType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToCloseTimeoutSeconds, rhs.ScheduleToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StartToCloseTimeoutSeconds, rhs.StartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.HeartbeatTimeoutSeconds, rhs.HeartbeatTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.RequestLocalDispatch, rhs.RequestLocalDispatch) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ScheduleActivityTaskDecisionAttributes.\nfunc (v *ScheduleActivityTaskDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ActivityId != nil {\n\t\tenc.AddString(\"activityId\", *v.ActivityId)\n\t}\n\tif v.ActivityType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activityType\", v.ActivityType))\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToCloseTimeoutSeconds\", *v.ScheduleToCloseTimeoutSeconds)\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"startToCloseTimeoutSeconds\", *v.StartToCloseTimeoutSeconds)\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"heartbeatTimeoutSeconds\", *v.HeartbeatTimeoutSeconds)\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.RequestLocalDispatch != nil {\n\t\tenc.AddBool(\"requestLocalDispatch\", *v.RequestLocalDispatch)\n\t}\n\treturn err\n}\n\n// GetActivityId returns the value of ActivityId if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetActivityId() (o string) {\n\tif v != nil && v.ActivityId != nil {\n\t\treturn *v.ActivityId\n\t}\n\n\treturn\n}\n\n// IsSetActivityId returns true if ActivityId is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetActivityId() bool {\n\treturn v != nil && v.ActivityId != nil\n}\n\n// GetActivityType returns the value of ActivityType if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\n\treturn\n}\n\n// IsSetActivityType returns true if ActivityType is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetActivityType() bool {\n\treturn v != nil && v.ActivityType != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetScheduleToCloseTimeoutSeconds returns the value of ScheduleToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToCloseTimeoutSeconds returns true if ScheduleToCloseTimeoutSeconds is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetScheduleToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToCloseTimeoutSeconds != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\n// GetStartToCloseTimeoutSeconds returns the value of StartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToCloseTimeoutSeconds returns true if StartToCloseTimeoutSeconds is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.StartToCloseTimeoutSeconds != nil\n}\n\n// GetHeartbeatTimeoutSeconds returns the value of HeartbeatTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatTimeoutSeconds returns true if HeartbeatTimeoutSeconds is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetHeartbeatTimeoutSeconds() bool {\n\treturn v != nil && v.HeartbeatTimeoutSeconds != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetRequestLocalDispatch returns the value of RequestLocalDispatch if it is set or its\n// zero value if it is unset.\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetRequestLocalDispatch() (o bool) {\n\tif v != nil && v.RequestLocalDispatch != nil {\n\t\treturn *v.RequestLocalDispatch\n\t}\n\n\treturn\n}\n\n// IsSetRequestLocalDispatch returns true if RequestLocalDispatch is not nil.\nfunc (v *ScheduleActivityTaskDecisionAttributes) IsSetRequestLocalDispatch() bool {\n\treturn v != nil && v.RequestLocalDispatch != nil\n}\n\ntype SearchAttributes struct {\n\tIndexedFields map[string][]byte `json:\"indexedFields,omitempty\"`\n}\n\n// ToWire translates a SearchAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SearchAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IndexedFields != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Binary_MapItemList(v.IndexedFields)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SearchAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SearchAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SearchAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SearchAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.IndexedFields, err = _Map_String_Binary_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SearchAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SearchAttributes struct could not be encoded.\nfunc (v *SearchAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IndexedFields != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Binary_Encode(v.IndexedFields, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SearchAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SearchAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SearchAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TMap:\n\t\t\tv.IndexedFields, err = _Map_String_Binary_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SearchAttributes\n// struct.\nfunc (v *SearchAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IndexedFields != nil {\n\t\tfields[i] = fmt.Sprintf(\"IndexedFields: %v\", v.IndexedFields)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SearchAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SearchAttributes match the\n// provided SearchAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SearchAttributes) Equals(rhs *SearchAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IndexedFields == nil && rhs.IndexedFields == nil) || (v.IndexedFields != nil && rhs.IndexedFields != nil && _Map_String_Binary_Equals(v.IndexedFields, rhs.IndexedFields))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SearchAttributes.\nfunc (v *SearchAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IndexedFields != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"indexedFields\", (_Map_String_Binary_Zapper)(v.IndexedFields)))\n\t}\n\treturn err\n}\n\n// GetIndexedFields returns the value of IndexedFields if it is set or its\n// zero value if it is unset.\nfunc (v *SearchAttributes) GetIndexedFields() (o map[string][]byte) {\n\tif v != nil && v.IndexedFields != nil {\n\t\treturn v.IndexedFields\n\t}\n\n\treturn\n}\n\n// IsSetIndexedFields returns true if IndexedFields is not nil.\nfunc (v *SearchAttributes) IsSetIndexedFields() bool {\n\treturn v != nil && v.IndexedFields != nil\n}\n\ntype ServiceBusyError struct {\n\tMessage string  `json:\"message,required\"`\n\tReason  *string `json:\"reason,omitempty\"`\n}\n\n// ToWire translates a ServiceBusyError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ServiceBusyError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 2, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ServiceBusyError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ServiceBusyError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ServiceBusyError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ServiceBusyError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of ServiceBusyError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ServiceBusyError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ServiceBusyError struct could not be encoded.\nfunc (v *ServiceBusyError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ServiceBusyError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ServiceBusyError struct could not be generated from the wire\n// representation.\nfunc (v *ServiceBusyError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of ServiceBusyError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ServiceBusyError\n// struct.\nfunc (v *ServiceBusyError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ServiceBusyError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*ServiceBusyError) ErrorName() string {\n\treturn \"ServiceBusyError\"\n}\n\n// Equals returns true if all the fields of this ServiceBusyError match the\n// provided ServiceBusyError.\n//\n// This function performs a deep comparison.\nfunc (v *ServiceBusyError) Equals(rhs *ServiceBusyError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ServiceBusyError.\nfunc (v *ServiceBusyError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *ServiceBusyError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *ServiceBusyError) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *ServiceBusyError) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\nfunc (v *ServiceBusyError) Error() string {\n\treturn v.String()\n}\n\ntype SignalExternalWorkflowExecutionDecisionAttributes struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tExecution         *WorkflowExecution `json:\"execution,omitempty\"`\n\tSignalName        *string            `json:\"signalName,omitempty\"`\n\tInput             []byte             `json:\"input,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly *bool              `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a SignalExternalWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SignalExternalWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalExternalWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalExternalWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalExternalWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SignalExternalWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalExternalWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalExternalWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalExternalWorkflowExecutionDecisionAttributes match the\n// provided SignalExternalWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) Equals(rhs *SignalExternalWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalExternalWorkflowExecutionDecisionAttributes.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype SignalExternalWorkflowExecutionFailedCause int32\n\nconst (\n\tSignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution SignalExternalWorkflowExecutionFailedCause = 0\n\tSignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted         SignalExternalWorkflowExecutionFailedCause = 1\n)\n\n// SignalExternalWorkflowExecutionFailedCause_Values returns all recognized values of SignalExternalWorkflowExecutionFailedCause.\nfunc SignalExternalWorkflowExecutionFailedCause_Values() []SignalExternalWorkflowExecutionFailedCause {\n\treturn []SignalExternalWorkflowExecutionFailedCause{\n\t\tSignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution,\n\t\tSignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted,\n\t}\n}\n\n// UnmarshalText tries to decode SignalExternalWorkflowExecutionFailedCause from a byte slice\n// containing its name.\n//\n//\tvar v SignalExternalWorkflowExecutionFailedCause\n//\terr := v.UnmarshalText([]byte(\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"))\nfunc (v *SignalExternalWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\":\n\t\t*v = SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*v = SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"SignalExternalWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*v = SignalExternalWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes SignalExternalWorkflowExecutionFailedCause to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v SignalExternalWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"), nil\n\tcase 1:\n\t\treturn []byte(\"WORKFLOW_ALREADY_COMPLETED\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalExternalWorkflowExecutionFailedCause.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v SignalExternalWorkflowExecutionFailedCause) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"WORKFLOW_ALREADY_COMPLETED\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v SignalExternalWorkflowExecutionFailedCause) Ptr() *SignalExternalWorkflowExecutionFailedCause {\n\treturn &v\n}\n\n// Encode encodes SignalExternalWorkflowExecutionFailedCause directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v SignalExternalWorkflowExecutionFailedCause\n//\treturn v.Encode(sWriter)\nfunc (v SignalExternalWorkflowExecutionFailedCause) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates SignalExternalWorkflowExecutionFailedCause into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v SignalExternalWorkflowExecutionFailedCause) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes SignalExternalWorkflowExecutionFailedCause from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return SignalExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\n//\tvar v SignalExternalWorkflowExecutionFailedCause\n//\tif err := v.FromWire(x); err != nil {\n//\t  return SignalExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *SignalExternalWorkflowExecutionFailedCause) FromWire(w wire.Value) error {\n\t*v = (SignalExternalWorkflowExecutionFailedCause)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded SignalExternalWorkflowExecutionFailedCause directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v SignalExternalWorkflowExecutionFailedCause\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return SignalExternalWorkflowExecutionFailedCause(0), err\n//\t}\n//\treturn v, nil\nfunc (v *SignalExternalWorkflowExecutionFailedCause) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (SignalExternalWorkflowExecutionFailedCause)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of SignalExternalWorkflowExecutionFailedCause.\nfunc (v SignalExternalWorkflowExecutionFailedCause) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"\n\tcase 1:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\t}\n\treturn fmt.Sprintf(\"SignalExternalWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// Equals returns true if this SignalExternalWorkflowExecutionFailedCause value matches the provided\n// value.\nfunc (v SignalExternalWorkflowExecutionFailedCause) Equals(rhs SignalExternalWorkflowExecutionFailedCause) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes SignalExternalWorkflowExecutionFailedCause into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v SignalExternalWorkflowExecutionFailedCause) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"WORKFLOW_ALREADY_COMPLETED\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode SignalExternalWorkflowExecutionFailedCause from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *SignalExternalWorkflowExecutionFailedCause) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"SignalExternalWorkflowExecutionFailedCause\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"SignalExternalWorkflowExecutionFailedCause\")\n\t\t}\n\t\t*v = (SignalExternalWorkflowExecutionFailedCause)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"SignalExternalWorkflowExecutionFailedCause\")\n\t}\n}\n\ntype SignalExternalWorkflowExecutionFailedEventAttributes struct {\n\tCause                        *SignalExternalWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventId *int64                                      `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       *string                                     `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution                          `json:\"workflowExecution,omitempty\"`\n\tInitiatedEventId             *int64                                      `json:\"initiatedEventId,omitempty\"`\n\tControl                      []byte                                      `json:\"control,omitempty\"`\n}\n\n// ToWire translates a SignalExternalWorkflowExecutionFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalExternalWorkflowExecutionFailedCause_Read(w wire.Value) (SignalExternalWorkflowExecutionFailedCause, error) {\n\tvar v SignalExternalWorkflowExecutionFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a SignalExternalWorkflowExecutionFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalExternalWorkflowExecutionFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalExternalWorkflowExecutionFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x SignalExternalWorkflowExecutionFailedCause\n\t\t\t\tx, err = _SignalExternalWorkflowExecutionFailedCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalExternalWorkflowExecutionFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionFailedEventAttributes struct could not be encoded.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalExternalWorkflowExecutionFailedCause_Decode(sr stream.Reader) (SignalExternalWorkflowExecutionFailedCause, error) {\n\tvar v SignalExternalWorkflowExecutionFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a SignalExternalWorkflowExecutionFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x SignalExternalWorkflowExecutionFailedCause\n\t\t\tx, err = _SignalExternalWorkflowExecutionFailedCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalExternalWorkflowExecutionFailedEventAttributes\n// struct.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalExternalWorkflowExecutionFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _SignalExternalWorkflowExecutionFailedCause_EqualsPtr(lhs, rhs *SignalExternalWorkflowExecutionFailedCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this SignalExternalWorkflowExecutionFailedEventAttributes match the\n// provided SignalExternalWorkflowExecutionFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) Equals(rhs *SignalExternalWorkflowExecutionFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_SignalExternalWorkflowExecutionFailedCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalExternalWorkflowExecutionFailedEventAttributes.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetCause() (o SignalExternalWorkflowExecutionFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype SignalExternalWorkflowExecutionInitiatedEventAttributes struct {\n\tDecisionTaskCompletedEventId *int64             `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tSignalName                   *string            `json:\"signalName,omitempty\"`\n\tInput                        []byte             `json:\"input,omitempty\"`\n\tControl                      []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly            *bool              `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// ToWire translates a SignalExternalWorkflowExecutionInitiatedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.ChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SignalExternalWorkflowExecutionInitiatedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalExternalWorkflowExecutionInitiatedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalExternalWorkflowExecutionInitiatedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.ChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalExternalWorkflowExecutionInitiatedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionInitiatedEventAttributes struct could not be encoded.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.ChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SignalExternalWorkflowExecutionInitiatedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalExternalWorkflowExecutionInitiatedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.ChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalExternalWorkflowExecutionInitiatedEventAttributes\n// struct.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChildWorkflowOnly: %v\", *(v.ChildWorkflowOnly))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalExternalWorkflowExecutionInitiatedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalExternalWorkflowExecutionInitiatedEventAttributes match the\n// provided SignalExternalWorkflowExecutionInitiatedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) Equals(rhs *SignalExternalWorkflowExecutionInitiatedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.ChildWorkflowOnly, rhs.ChildWorkflowOnly) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalExternalWorkflowExecutionInitiatedEventAttributes.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.ChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"childWorkflowOnly\", *v.ChildWorkflowOnly)\n\t}\n\treturn err\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetChildWorkflowOnly returns the value of ChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.ChildWorkflowOnly != nil {\n\t\treturn *v.ChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetChildWorkflowOnly returns true if ChildWorkflowOnly is not nil.\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) IsSetChildWorkflowOnly() bool {\n\treturn v != nil && v.ChildWorkflowOnly != nil\n}\n\ntype SignalWithStartWorkflowExecutionAsyncRequest struct {\n\tRequest *SignalWithStartWorkflowExecutionRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a SignalWithStartWorkflowExecutionAsyncRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Read(w wire.Value) (*SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v SignalWithStartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a SignalWithStartWorkflowExecutionAsyncRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWithStartWorkflowExecutionAsyncRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWithStartWorkflowExecutionAsyncRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _SignalWithStartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalWithStartWorkflowExecutionAsyncRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionAsyncRequest struct could not be encoded.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _SignalWithStartWorkflowExecutionRequest_Decode(sr stream.Reader) (*SignalWithStartWorkflowExecutionRequest, error) {\n\tvar v SignalWithStartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a SignalWithStartWorkflowExecutionAsyncRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionAsyncRequest struct could not be generated from the wire\n// representation.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _SignalWithStartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWithStartWorkflowExecutionAsyncRequest\n// struct.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalWithStartWorkflowExecutionAsyncRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalWithStartWorkflowExecutionAsyncRequest match the\n// provided SignalWithStartWorkflowExecutionAsyncRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) Equals(rhs *SignalWithStartWorkflowExecutionAsyncRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWithStartWorkflowExecutionAsyncRequest.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) GetRequest() (o *SignalWithStartWorkflowExecutionRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *SignalWithStartWorkflowExecutionAsyncRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype SignalWithStartWorkflowExecutionAsyncResponse struct {\n}\n\n// ToWire translates a SignalWithStartWorkflowExecutionAsyncResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SignalWithStartWorkflowExecutionAsyncResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWithStartWorkflowExecutionAsyncResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWithStartWorkflowExecutionAsyncResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalWithStartWorkflowExecutionAsyncResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionAsyncResponse struct could not be encoded.\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SignalWithStartWorkflowExecutionAsyncResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionAsyncResponse struct could not be generated from the wire\n// representation.\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWithStartWorkflowExecutionAsyncResponse\n// struct.\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"SignalWithStartWorkflowExecutionAsyncResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalWithStartWorkflowExecutionAsyncResponse match the\n// provided SignalWithStartWorkflowExecutionAsyncResponse.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) Equals(rhs *SignalWithStartWorkflowExecutionAsyncResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWithStartWorkflowExecutionAsyncResponse.\nfunc (v *SignalWithStartWorkflowExecutionAsyncResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype SignalWithStartWorkflowExecutionRequest struct {\n\tDomain                              *string                       `json:\"domain,omitempty\"`\n\tWorkflowId                          *string                       `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tIdentity                            *string                       `json:\"identity,omitempty\"`\n\tRequestId                           *string                       `json:\"requestId,omitempty\"`\n\tWorkflowIdReusePolicy               *WorkflowIdReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tSignalName                          *string                       `json:\"signalName,omitempty\"`\n\tSignalInput                         []byte                        `json:\"signalInput,omitempty\"`\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimestamp                 *int64                        `json:\"firstRunAtTimestamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a SignalWithStartWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWithStartWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [23]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tw, err = v.WorkflowIdReusePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.SignalInput != nil {\n\t\tw, err = wire.NewValueBinary(v.SignalInput), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 161, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.DelayStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.JitterStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstRunAtTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 200, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 210, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 220, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _WorkflowIdReusePolicy_Read(w wire.Value) (WorkflowIdReusePolicy, error) {\n\tvar v WorkflowIdReusePolicy\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a SignalWithStartWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWithStartWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWithStartWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWithStartWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowIdReusePolicy\n\t\t\t\tx, err = _WorkflowIdReusePolicy_Read(field.Value)\n\t\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.SignalInput, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 161:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DelayStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.JitterStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 200:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 210:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 220:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalWithStartWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionRequest struct could not be encoded.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowIdReusePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalInput != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.SignalInput); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 161, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DelayStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DelayStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.JitterStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.JitterStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstRunAtTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 200, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstRunAtTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 210, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 220, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _WorkflowIdReusePolicy_Decode(sr stream.Reader) (WorkflowIdReusePolicy, error) {\n\tvar v WorkflowIdReusePolicy\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a SignalWithStartWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWithStartWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowIdReusePolicy\n\t\t\tx, err = _WorkflowIdReusePolicy_Decode(sr)\n\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tv.SignalInput, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 161 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DelayStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.JitterStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 200 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 210 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 220 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWithStartWorkflowExecutionRequest\n// struct.\nfunc (v *SignalWithStartWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [23]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowIdReusePolicy: %v\", *(v.WorkflowIdReusePolicy))\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.SignalInput != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalInput: %v\", v.SignalInput)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"DelayStartSeconds: %v\", *(v.DelayStartSeconds))\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"JitterStartSeconds: %v\", *(v.JitterStartSeconds))\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstRunAtTimestamp: %v\", *(v.FirstRunAtTimestamp))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalWithStartWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _WorkflowIdReusePolicy_EqualsPtr(lhs, rhs *WorkflowIdReusePolicy) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this SignalWithStartWorkflowExecutionRequest match the\n// provided SignalWithStartWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWithStartWorkflowExecutionRequest) Equals(rhs *SignalWithStartWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_WorkflowIdReusePolicy_EqualsPtr(v.WorkflowIdReusePolicy, rhs.WorkflowIdReusePolicy) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.SignalInput == nil && rhs.SignalInput == nil) || (v.SignalInput != nil && rhs.SignalInput != nil && bytes.Equal(v.SignalInput, rhs.SignalInput))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DelayStartSeconds, rhs.DelayStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.JitterStartSeconds, rhs.JitterStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstRunAtTimestamp, rhs.FirstRunAtTimestamp) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWithStartWorkflowExecutionRequest.\nfunc (v *SignalWithStartWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowIdReusePolicy\", *v.WorkflowIdReusePolicy))\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.SignalInput != nil {\n\t\tenc.AddString(\"signalInput\", base64.StdEncoding.EncodeToString(v.SignalInput))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tenc.AddInt32(\"delayStartSeconds\", *v.DelayStartSeconds)\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tenc.AddInt32(\"jitterStartSeconds\", *v.JitterStartSeconds)\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tenc.AddInt64(\"firstRunAtTimestamp\", *v.FirstRunAtTimestamp)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetWorkflowIdReusePolicy returns the value of WorkflowIdReusePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowIdReusePolicy() (o WorkflowIdReusePolicy) {\n\tif v != nil && v.WorkflowIdReusePolicy != nil {\n\t\treturn *v.WorkflowIdReusePolicy\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowIdReusePolicy returns true if WorkflowIdReusePolicy is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetWorkflowIdReusePolicy() bool {\n\treturn v != nil && v.WorkflowIdReusePolicy != nil\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetSignalInput returns the value of SignalInput if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSignalInput() (o []byte) {\n\tif v != nil && v.SignalInput != nil {\n\t\treturn v.SignalInput\n\t}\n\n\treturn\n}\n\n// IsSetSignalInput returns true if SignalInput is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetSignalInput() bool {\n\treturn v != nil && v.SignalInput != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetDelayStartSeconds returns the value of DelayStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetDelayStartSeconds() (o int32) {\n\tif v != nil && v.DelayStartSeconds != nil {\n\t\treturn *v.DelayStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetDelayStartSeconds returns true if DelayStartSeconds is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetDelayStartSeconds() bool {\n\treturn v != nil && v.DelayStartSeconds != nil\n}\n\n// GetJitterStartSeconds returns the value of JitterStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetJitterStartSeconds returns true if JitterStartSeconds is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetJitterStartSeconds() bool {\n\treturn v != nil && v.JitterStartSeconds != nil\n}\n\n// GetFirstRunAtTimestamp returns the value of FirstRunAtTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetFirstRunAtTimestamp() (o int64) {\n\tif v != nil && v.FirstRunAtTimestamp != nil {\n\t\treturn *v.FirstRunAtTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetFirstRunAtTimestamp returns true if FirstRunAtTimestamp is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetFirstRunAtTimestamp() bool {\n\treturn v != nil && v.FirstRunAtTimestamp != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *SignalWithStartWorkflowExecutionRequest) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype SignalWorkflowExecutionRequest struct {\n\tDomain            *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tSignalName        *string            `json:\"signalName,omitempty\"`\n\tInput             []byte             `json:\"input,omitempty\"`\n\tIdentity          *string            `json:\"identity,omitempty\"`\n\tRequestId         *string            `json:\"requestId,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n}\n\n// ToWire translates a SignalWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SignalWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalWorkflowExecutionRequest struct could not be encoded.\nfunc (v *SignalWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SignalWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *SignalWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalWorkflowExecutionRequest\n// struct.\nfunc (v *SignalWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalWorkflowExecutionRequest match the\n// provided SignalWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *SignalWorkflowExecutionRequest) Equals(rhs *SignalWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalWorkflowExecutionRequest.\nfunc (v *SignalWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalWorkflowExecutionRequest) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalWorkflowExecutionRequest) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype StartChildWorkflowExecutionDecisionAttributes struct {\n\tDomain                              *string                       `json:\"domain,omitempty\"`\n\tWorkflowId                          *string                       `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tParentClosePolicy                   *ParentClosePolicy            `json:\"parentClosePolicy,omitempty\"`\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tWorkflowIdReusePolicy               *WorkflowIdReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a StartChildWorkflowExecutionDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [17]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tw, err = v.ParentClosePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 81, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tw, err = v.WorkflowIdReusePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartChildWorkflowExecutionDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartChildWorkflowExecutionDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartChildWorkflowExecutionDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 81:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ParentClosePolicy\n\t\t\t\tx, err = _ParentClosePolicy_Read(field.Value)\n\t\t\t\tv.ParentClosePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowIdReusePolicy\n\t\t\t\tx, err = _WorkflowIdReusePolicy_Read(field.Value)\n\t\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartChildWorkflowExecutionDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionDecisionAttributes struct could not be encoded.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentClosePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 81, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentClosePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowIdReusePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartChildWorkflowExecutionDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 81 && fh.Type == wire.TI32:\n\t\t\tvar x ParentClosePolicy\n\t\t\tx, err = _ParentClosePolicy_Decode(sr)\n\t\t\tv.ParentClosePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowIdReusePolicy\n\t\t\tx, err = _WorkflowIdReusePolicy_Decode(sr)\n\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartChildWorkflowExecutionDecisionAttributes\n// struct.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [17]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentClosePolicy: %v\", *(v.ParentClosePolicy))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowIdReusePolicy: %v\", *(v.WorkflowIdReusePolicy))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartChildWorkflowExecutionDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartChildWorkflowExecutionDecisionAttributes match the\n// provided StartChildWorkflowExecutionDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) Equals(rhs *StartChildWorkflowExecutionDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_ParentClosePolicy_EqualsPtr(v.ParentClosePolicy, rhs.ParentClosePolicy) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_WorkflowIdReusePolicy_EqualsPtr(v.WorkflowIdReusePolicy, rhs.WorkflowIdReusePolicy) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartChildWorkflowExecutionDecisionAttributes.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentClosePolicy\", *v.ParentClosePolicy))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowIdReusePolicy\", *v.WorkflowIdReusePolicy))\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetParentClosePolicy returns the value of ParentClosePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\n\treturn\n}\n\n// IsSetParentClosePolicy returns true if ParentClosePolicy is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetParentClosePolicy() bool {\n\treturn v != nil && v.ParentClosePolicy != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetWorkflowIdReusePolicy returns the value of WorkflowIdReusePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetWorkflowIdReusePolicy() (o WorkflowIdReusePolicy) {\n\tif v != nil && v.WorkflowIdReusePolicy != nil {\n\t\treturn *v.WorkflowIdReusePolicy\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowIdReusePolicy returns true if WorkflowIdReusePolicy is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetWorkflowIdReusePolicy() bool {\n\treturn v != nil && v.WorkflowIdReusePolicy != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype StartChildWorkflowExecutionFailedEventAttributes struct {\n\tDomain                       *string                            `json:\"domain,omitempty\"`\n\tWorkflowId                   *string                            `json:\"workflowId,omitempty\"`\n\tWorkflowType                 *WorkflowType                      `json:\"workflowType,omitempty\"`\n\tCause                        *ChildWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tControl                      []byte                             `json:\"control,omitempty\"`\n\tInitiatedEventId             *int64                             `json:\"initiatedEventId,omitempty\"`\n\tDecisionTaskCompletedEventId *int64                             `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a StartChildWorkflowExecutionFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tw, err = v.Cause.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ChildWorkflowExecutionFailedCause_Read(w wire.Value) (ChildWorkflowExecutionFailedCause, error) {\n\tvar v ChildWorkflowExecutionFailedCause\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a StartChildWorkflowExecutionFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartChildWorkflowExecutionFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartChildWorkflowExecutionFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ChildWorkflowExecutionFailedCause\n\t\t\t\tx, err = _ChildWorkflowExecutionFailedCause_Read(field.Value)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartChildWorkflowExecutionFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionFailedEventAttributes struct could not be encoded.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Cause.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ChildWorkflowExecutionFailedCause_Decode(sr stream.Reader) (ChildWorkflowExecutionFailedCause, error) {\n\tvar v ChildWorkflowExecutionFailedCause\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a StartChildWorkflowExecutionFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x ChildWorkflowExecutionFailedCause\n\t\t\tx, err = _ChildWorkflowExecutionFailedCause_Decode(sr)\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartChildWorkflowExecutionFailedEventAttributes\n// struct.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventId: %v\", *(v.InitiatedEventId))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartChildWorkflowExecutionFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _ChildWorkflowExecutionFailedCause_EqualsPtr(lhs, rhs *ChildWorkflowExecutionFailedCause) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this StartChildWorkflowExecutionFailedEventAttributes match the\n// provided StartChildWorkflowExecutionFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) Equals(rhs *StartChildWorkflowExecutionFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_ChildWorkflowExecutionFailedCause_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventId, rhs.InitiatedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartChildWorkflowExecutionFailedEventAttributes.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.Cause != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cause\", *v.Cause))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.InitiatedEventId != nil {\n\t\tenc.AddInt64(\"initiatedEventId\", *v.InitiatedEventId)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetCause() (o ChildWorkflowExecutionFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetInitiatedEventId returns the value of InitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetInitiatedEventId() (o int64) {\n\tif v != nil && v.InitiatedEventId != nil {\n\t\treturn *v.InitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventId returns true if InitiatedEventId is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetInitiatedEventId() bool {\n\treturn v != nil && v.InitiatedEventId != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype StartChildWorkflowExecutionInitiatedEventAttributes struct {\n\tDomain                              *string                       `json:\"domain,omitempty\"`\n\tWorkflowId                          *string                       `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tParentClosePolicy                   *ParentClosePolicy            `json:\"parentClosePolicy,omitempty\"`\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tDecisionTaskCompletedEventId        *int64                        `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tWorkflowIdReusePolicy               *WorkflowIdReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimestamp                 *int64                        `json:\"firstRunAtTimestamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a StartChildWorkflowExecutionInitiatedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [21]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tw, err = v.ParentClosePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 81, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tw, err = v.WorkflowIdReusePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.DelayStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.JitterStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstRunAtTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 200, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 210, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartChildWorkflowExecutionInitiatedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartChildWorkflowExecutionInitiatedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartChildWorkflowExecutionInitiatedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 81:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ParentClosePolicy\n\t\t\t\tx, err = _ParentClosePolicy_Read(field.Value)\n\t\t\t\tv.ParentClosePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowIdReusePolicy\n\t\t\t\tx, err = _WorkflowIdReusePolicy_Read(field.Value)\n\t\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DelayStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.JitterStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 200:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 210:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartChildWorkflowExecutionInitiatedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionInitiatedEventAttributes struct could not be encoded.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentClosePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 81, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentClosePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowIdReusePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DelayStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DelayStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.JitterStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.JitterStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstRunAtTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstRunAtTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 200, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 210, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartChildWorkflowExecutionInitiatedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartChildWorkflowExecutionInitiatedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 81 && fh.Type == wire.TI32:\n\t\t\tvar x ParentClosePolicy\n\t\t\tx, err = _ParentClosePolicy_Decode(sr)\n\t\t\tv.ParentClosePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowIdReusePolicy\n\t\t\tx, err = _WorkflowIdReusePolicy_Decode(sr)\n\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DelayStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.JitterStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 200 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 210 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartChildWorkflowExecutionInitiatedEventAttributes\n// struct.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [21]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentClosePolicy: %v\", *(v.ParentClosePolicy))\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowIdReusePolicy: %v\", *(v.WorkflowIdReusePolicy))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"DelayStartSeconds: %v\", *(v.DelayStartSeconds))\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"JitterStartSeconds: %v\", *(v.JitterStartSeconds))\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstRunAtTimestamp: %v\", *(v.FirstRunAtTimestamp))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartChildWorkflowExecutionInitiatedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartChildWorkflowExecutionInitiatedEventAttributes match the\n// provided StartChildWorkflowExecutionInitiatedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) Equals(rhs *StartChildWorkflowExecutionInitiatedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_ParentClosePolicy_EqualsPtr(v.ParentClosePolicy, rhs.ParentClosePolicy) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_WorkflowIdReusePolicy_EqualsPtr(v.WorkflowIdReusePolicy, rhs.WorkflowIdReusePolicy) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DelayStartSeconds, rhs.DelayStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.JitterStartSeconds, rhs.JitterStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstRunAtTimestamp, rhs.FirstRunAtTimestamp) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartChildWorkflowExecutionInitiatedEventAttributes.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentClosePolicy\", *v.ParentClosePolicy))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowIdReusePolicy\", *v.WorkflowIdReusePolicy))\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tenc.AddInt32(\"delayStartSeconds\", *v.DelayStartSeconds)\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tenc.AddInt32(\"jitterStartSeconds\", *v.JitterStartSeconds)\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tenc.AddInt64(\"firstRunAtTimestamp\", *v.FirstRunAtTimestamp)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetParentClosePolicy returns the value of ParentClosePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\n\treturn\n}\n\n// IsSetParentClosePolicy returns true if ParentClosePolicy is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetParentClosePolicy() bool {\n\treturn v != nil && v.ParentClosePolicy != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetWorkflowIdReusePolicy returns the value of WorkflowIdReusePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetWorkflowIdReusePolicy() (o WorkflowIdReusePolicy) {\n\tif v != nil && v.WorkflowIdReusePolicy != nil {\n\t\treturn *v.WorkflowIdReusePolicy\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowIdReusePolicy returns true if WorkflowIdReusePolicy is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetWorkflowIdReusePolicy() bool {\n\treturn v != nil && v.WorkflowIdReusePolicy != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetDelayStartSeconds returns the value of DelayStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetDelayStartSeconds() (o int32) {\n\tif v != nil && v.DelayStartSeconds != nil {\n\t\treturn *v.DelayStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetDelayStartSeconds returns true if DelayStartSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetDelayStartSeconds() bool {\n\treturn v != nil && v.DelayStartSeconds != nil\n}\n\n// GetJitterStartSeconds returns the value of JitterStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetJitterStartSeconds returns true if JitterStartSeconds is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetJitterStartSeconds() bool {\n\treturn v != nil && v.JitterStartSeconds != nil\n}\n\n// GetFirstRunAtTimestamp returns the value of FirstRunAtTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetFirstRunAtTimestamp() (o int64) {\n\tif v != nil && v.FirstRunAtTimestamp != nil {\n\t\treturn *v.FirstRunAtTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetFirstRunAtTimestamp returns true if FirstRunAtTimestamp is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetFirstRunAtTimestamp() bool {\n\treturn v != nil && v.FirstRunAtTimestamp != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype StartTimeFilter struct {\n\tEarliestTime *int64 `json:\"earliestTime,omitempty\"`\n\tLatestTime   *int64 `json:\"latestTime,omitempty\"`\n}\n\n// ToWire translates a StartTimeFilter struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartTimeFilter) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.EarliestTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.EarliestTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.LatestTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.LatestTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartTimeFilter struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartTimeFilter struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartTimeFilter\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartTimeFilter) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EarliestTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LatestTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartTimeFilter struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartTimeFilter struct could not be encoded.\nfunc (v *StartTimeFilter) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EarliestTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EarliestTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LatestTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LatestTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartTimeFilter struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartTimeFilter struct could not be generated from the wire\n// representation.\nfunc (v *StartTimeFilter) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EarliestTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LatestTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartTimeFilter\n// struct.\nfunc (v *StartTimeFilter) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.EarliestTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"EarliestTime: %v\", *(v.EarliestTime))\n\t\ti++\n\t}\n\tif v.LatestTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"LatestTime: %v\", *(v.LatestTime))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartTimeFilter{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartTimeFilter match the\n// provided StartTimeFilter.\n//\n// This function performs a deep comparison.\nfunc (v *StartTimeFilter) Equals(rhs *StartTimeFilter) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EarliestTime, rhs.EarliestTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LatestTime, rhs.LatestTime) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartTimeFilter.\nfunc (v *StartTimeFilter) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.EarliestTime != nil {\n\t\tenc.AddInt64(\"earliestTime\", *v.EarliestTime)\n\t}\n\tif v.LatestTime != nil {\n\t\tenc.AddInt64(\"latestTime\", *v.LatestTime)\n\t}\n\treturn err\n}\n\n// GetEarliestTime returns the value of EarliestTime if it is set or its\n// zero value if it is unset.\nfunc (v *StartTimeFilter) GetEarliestTime() (o int64) {\n\tif v != nil && v.EarliestTime != nil {\n\t\treturn *v.EarliestTime\n\t}\n\n\treturn\n}\n\n// IsSetEarliestTime returns true if EarliestTime is not nil.\nfunc (v *StartTimeFilter) IsSetEarliestTime() bool {\n\treturn v != nil && v.EarliestTime != nil\n}\n\n// GetLatestTime returns the value of LatestTime if it is set or its\n// zero value if it is unset.\nfunc (v *StartTimeFilter) GetLatestTime() (o int64) {\n\tif v != nil && v.LatestTime != nil {\n\t\treturn *v.LatestTime\n\t}\n\n\treturn\n}\n\n// IsSetLatestTime returns true if LatestTime is not nil.\nfunc (v *StartTimeFilter) IsSetLatestTime() bool {\n\treturn v != nil && v.LatestTime != nil\n}\n\ntype StartTimerDecisionAttributes struct {\n\tTimerId                   *string `json:\"timerId,omitempty\"`\n\tStartToFireTimeoutSeconds *int64  `json:\"startToFireTimeoutSeconds,omitempty\"`\n}\n\n// ToWire translates a StartTimerDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartTimerDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartToFireTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartTimerDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartTimerDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartTimerDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartTimerDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartToFireTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartTimerDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartTimerDecisionAttributes struct could not be encoded.\nfunc (v *StartTimerDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartToFireTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartTimerDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartTimerDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *StartTimerDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartToFireTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartTimerDecisionAttributes\n// struct.\nfunc (v *StartTimerDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToFireTimeoutSeconds: %v\", *(v.StartToFireTimeoutSeconds))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartTimerDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartTimerDecisionAttributes match the\n// provided StartTimerDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *StartTimerDecisionAttributes) Equals(rhs *StartTimerDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartToFireTimeoutSeconds, rhs.StartToFireTimeoutSeconds) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartTimerDecisionAttributes.\nfunc (v *StartTimerDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tenc.AddInt64(\"startToFireTimeoutSeconds\", *v.StartToFireTimeoutSeconds)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *StartTimerDecisionAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *StartTimerDecisionAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\n// GetStartToFireTimeoutSeconds returns the value of StartToFireTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartTimerDecisionAttributes) GetStartToFireTimeoutSeconds() (o int64) {\n\tif v != nil && v.StartToFireTimeoutSeconds != nil {\n\t\treturn *v.StartToFireTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToFireTimeoutSeconds returns true if StartToFireTimeoutSeconds is not nil.\nfunc (v *StartTimerDecisionAttributes) IsSetStartToFireTimeoutSeconds() bool {\n\treturn v != nil && v.StartToFireTimeoutSeconds != nil\n}\n\ntype StartWorkflowExecutionAsyncRequest struct {\n\tRequest *StartWorkflowExecutionRequest `json:\"request,omitempty\"`\n}\n\n// ToWire translates a StartWorkflowExecutionAsyncRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartWorkflowExecutionAsyncRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Request != nil {\n\t\tw, err = v.Request.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _StartWorkflowExecutionRequest_Read(w wire.Value) (*StartWorkflowExecutionRequest, error) {\n\tvar v StartWorkflowExecutionRequest\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a StartWorkflowExecutionAsyncRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartWorkflowExecutionAsyncRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartWorkflowExecutionAsyncRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartWorkflowExecutionAsyncRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Request, err = _StartWorkflowExecutionRequest_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartWorkflowExecutionAsyncRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartWorkflowExecutionAsyncRequest struct could not be encoded.\nfunc (v *StartWorkflowExecutionAsyncRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Request != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Request.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _StartWorkflowExecutionRequest_Decode(sr stream.Reader) (*StartWorkflowExecutionRequest, error) {\n\tvar v StartWorkflowExecutionRequest\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a StartWorkflowExecutionAsyncRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartWorkflowExecutionAsyncRequest struct could not be generated from the wire\n// representation.\nfunc (v *StartWorkflowExecutionAsyncRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Request, err = _StartWorkflowExecutionRequest_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartWorkflowExecutionAsyncRequest\n// struct.\nfunc (v *StartWorkflowExecutionAsyncRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Request != nil {\n\t\tfields[i] = fmt.Sprintf(\"Request: %v\", v.Request)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartWorkflowExecutionAsyncRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartWorkflowExecutionAsyncRequest match the\n// provided StartWorkflowExecutionAsyncRequest.\n//\n// This function performs a deep comparison.\nfunc (v *StartWorkflowExecutionAsyncRequest) Equals(rhs *StartWorkflowExecutionAsyncRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Request == nil && rhs.Request == nil) || (v.Request != nil && rhs.Request != nil && v.Request.Equals(rhs.Request))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartWorkflowExecutionAsyncRequest.\nfunc (v *StartWorkflowExecutionAsyncRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Request != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"request\", v.Request))\n\t}\n\treturn err\n}\n\n// GetRequest returns the value of Request if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionAsyncRequest) GetRequest() (o *StartWorkflowExecutionRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\n\treturn\n}\n\n// IsSetRequest returns true if Request is not nil.\nfunc (v *StartWorkflowExecutionAsyncRequest) IsSetRequest() bool {\n\treturn v != nil && v.Request != nil\n}\n\ntype StartWorkflowExecutionAsyncResponse struct {\n}\n\n// ToWire translates a StartWorkflowExecutionAsyncResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartWorkflowExecutionAsyncResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartWorkflowExecutionAsyncResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartWorkflowExecutionAsyncResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartWorkflowExecutionAsyncResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartWorkflowExecutionAsyncResponse) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartWorkflowExecutionAsyncResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartWorkflowExecutionAsyncResponse struct could not be encoded.\nfunc (v *StartWorkflowExecutionAsyncResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartWorkflowExecutionAsyncResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartWorkflowExecutionAsyncResponse struct could not be generated from the wire\n// representation.\nfunc (v *StartWorkflowExecutionAsyncResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartWorkflowExecutionAsyncResponse\n// struct.\nfunc (v *StartWorkflowExecutionAsyncResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"StartWorkflowExecutionAsyncResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartWorkflowExecutionAsyncResponse match the\n// provided StartWorkflowExecutionAsyncResponse.\n//\n// This function performs a deep comparison.\nfunc (v *StartWorkflowExecutionAsyncResponse) Equals(rhs *StartWorkflowExecutionAsyncResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartWorkflowExecutionAsyncResponse.\nfunc (v *StartWorkflowExecutionAsyncResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype StartWorkflowExecutionRequest struct {\n\tDomain                              *string                       `json:\"domain,omitempty\"`\n\tWorkflowId                          *string                       `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tIdentity                            *string                       `json:\"identity,omitempty\"`\n\tRequestId                           *string                       `json:\"requestId,omitempty\"`\n\tWorkflowIdReusePolicy               *WorkflowIdReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimestamp                 *int64                        `json:\"firstRunAtTimestamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a StartWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [20]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tw, err = v.WorkflowIdReusePolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 141, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.DelayStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.JitterStartSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstRunAtTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 200, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowIdReusePolicy\n\t\t\t\tx, err = _WorkflowIdReusePolicy_Read(field.Value)\n\t\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 141:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DelayStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.JitterStartSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 200:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartWorkflowExecutionRequest struct could not be encoded.\nfunc (v *StartWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowIdReusePolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 141, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DelayStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DelayStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.JitterStartSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.JitterStartSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstRunAtTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstRunAtTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 200, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *StartWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowIdReusePolicy\n\t\t\tx, err = _WorkflowIdReusePolicy_Decode(sr)\n\t\t\tv.WorkflowIdReusePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 141 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DelayStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.JitterStartSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstRunAtTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 200 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartWorkflowExecutionRequest\n// struct.\nfunc (v *StartWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [20]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowIdReusePolicy: %v\", *(v.WorkflowIdReusePolicy))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"DelayStartSeconds: %v\", *(v.DelayStartSeconds))\n\t\ti++\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"JitterStartSeconds: %v\", *(v.JitterStartSeconds))\n\t\ti++\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstRunAtTimestamp: %v\", *(v.FirstRunAtTimestamp))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartWorkflowExecutionRequest match the\n// provided StartWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *StartWorkflowExecutionRequest) Equals(rhs *StartWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_WorkflowIdReusePolicy_EqualsPtr(v.WorkflowIdReusePolicy, rhs.WorkflowIdReusePolicy) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DelayStartSeconds, rhs.DelayStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.JitterStartSeconds, rhs.JitterStartSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstRunAtTimestamp, rhs.FirstRunAtTimestamp) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartWorkflowExecutionRequest.\nfunc (v *StartWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.WorkflowIdReusePolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowIdReusePolicy\", *v.WorkflowIdReusePolicy))\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.DelayStartSeconds != nil {\n\t\tenc.AddInt32(\"delayStartSeconds\", *v.DelayStartSeconds)\n\t}\n\tif v.JitterStartSeconds != nil {\n\t\tenc.AddInt32(\"jitterStartSeconds\", *v.JitterStartSeconds)\n\t}\n\tif v.FirstRunAtTimestamp != nil {\n\t\tenc.AddInt64(\"firstRunAtTimestamp\", *v.FirstRunAtTimestamp)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetWorkflowIdReusePolicy returns the value of WorkflowIdReusePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetWorkflowIdReusePolicy() (o WorkflowIdReusePolicy) {\n\tif v != nil && v.WorkflowIdReusePolicy != nil {\n\t\treturn *v.WorkflowIdReusePolicy\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowIdReusePolicy returns true if WorkflowIdReusePolicy is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetWorkflowIdReusePolicy() bool {\n\treturn v != nil && v.WorkflowIdReusePolicy != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetDelayStartSeconds returns the value of DelayStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetDelayStartSeconds() (o int32) {\n\tif v != nil && v.DelayStartSeconds != nil {\n\t\treturn *v.DelayStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetDelayStartSeconds returns true if DelayStartSeconds is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetDelayStartSeconds() bool {\n\treturn v != nil && v.DelayStartSeconds != nil\n}\n\n// GetJitterStartSeconds returns the value of JitterStartSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\n\treturn\n}\n\n// IsSetJitterStartSeconds returns true if JitterStartSeconds is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetJitterStartSeconds() bool {\n\treturn v != nil && v.JitterStartSeconds != nil\n}\n\n// GetFirstRunAtTimestamp returns the value of FirstRunAtTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetFirstRunAtTimestamp() (o int64) {\n\tif v != nil && v.FirstRunAtTimestamp != nil {\n\t\treturn *v.FirstRunAtTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetFirstRunAtTimestamp returns true if FirstRunAtTimestamp is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetFirstRunAtTimestamp() bool {\n\treturn v != nil && v.FirstRunAtTimestamp != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionRequest) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *StartWorkflowExecutionRequest) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype StartWorkflowExecutionResponse struct {\n\tRunId *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a StartWorkflowExecutionResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StartWorkflowExecutionResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StartWorkflowExecutionResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StartWorkflowExecutionResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StartWorkflowExecutionResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StartWorkflowExecutionResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StartWorkflowExecutionResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StartWorkflowExecutionResponse struct could not be encoded.\nfunc (v *StartWorkflowExecutionResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StartWorkflowExecutionResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StartWorkflowExecutionResponse struct could not be generated from the wire\n// representation.\nfunc (v *StartWorkflowExecutionResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StartWorkflowExecutionResponse\n// struct.\nfunc (v *StartWorkflowExecutionResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StartWorkflowExecutionResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StartWorkflowExecutionResponse match the\n// provided StartWorkflowExecutionResponse.\n//\n// This function performs a deep comparison.\nfunc (v *StartWorkflowExecutionResponse) Equals(rhs *StartWorkflowExecutionResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StartWorkflowExecutionResponse.\nfunc (v *StartWorkflowExecutionResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *StartWorkflowExecutionResponse) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *StartWorkflowExecutionResponse) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\ntype StickyExecutionAttributes struct {\n\tWorkerTaskList                *TaskList `json:\"workerTaskList,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32    `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n}\n\n// ToWire translates a StickyExecutionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StickyExecutionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkerTaskList != nil {\n\t\tw, err = v.WorkerTaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StickyExecutionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StickyExecutionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StickyExecutionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StickyExecutionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkerTaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StickyExecutionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StickyExecutionAttributes struct could not be encoded.\nfunc (v *StickyExecutionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkerTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkerTaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StickyExecutionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StickyExecutionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *StickyExecutionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.WorkerTaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StickyExecutionAttributes\n// struct.\nfunc (v *StickyExecutionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.WorkerTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkerTaskList: %v\", v.WorkerTaskList)\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"StickyExecutionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this StickyExecutionAttributes match the\n// provided StickyExecutionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *StickyExecutionAttributes) Equals(rhs *StickyExecutionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.WorkerTaskList == nil && rhs.WorkerTaskList == nil) || (v.WorkerTaskList != nil && rhs.WorkerTaskList != nil && v.WorkerTaskList.Equals(rhs.WorkerTaskList))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StickyExecutionAttributes.\nfunc (v *StickyExecutionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkerTaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workerTaskList\", v.WorkerTaskList))\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\treturn err\n}\n\n// GetWorkerTaskList returns the value of WorkerTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *StickyExecutionAttributes) GetWorkerTaskList() (o *TaskList) {\n\tif v != nil && v.WorkerTaskList != nil {\n\t\treturn v.WorkerTaskList\n\t}\n\n\treturn\n}\n\n// IsSetWorkerTaskList returns true if WorkerTaskList is not nil.\nfunc (v *StickyExecutionAttributes) IsSetWorkerTaskList() bool {\n\treturn v != nil && v.WorkerTaskList != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *StickyExecutionAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *StickyExecutionAttributes) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\ntype StickyWorkerUnavailableError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a StickyWorkerUnavailableError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *StickyWorkerUnavailableError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a StickyWorkerUnavailableError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a StickyWorkerUnavailableError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v StickyWorkerUnavailableError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *StickyWorkerUnavailableError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of StickyWorkerUnavailableError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a StickyWorkerUnavailableError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a StickyWorkerUnavailableError struct could not be encoded.\nfunc (v *StickyWorkerUnavailableError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a StickyWorkerUnavailableError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a StickyWorkerUnavailableError struct could not be generated from the wire\n// representation.\nfunc (v *StickyWorkerUnavailableError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of StickyWorkerUnavailableError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a StickyWorkerUnavailableError\n// struct.\nfunc (v *StickyWorkerUnavailableError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"StickyWorkerUnavailableError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*StickyWorkerUnavailableError) ErrorName() string {\n\treturn \"StickyWorkerUnavailableError\"\n}\n\n// Equals returns true if all the fields of this StickyWorkerUnavailableError match the\n// provided StickyWorkerUnavailableError.\n//\n// This function performs a deep comparison.\nfunc (v *StickyWorkerUnavailableError) Equals(rhs *StickyWorkerUnavailableError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of StickyWorkerUnavailableError.\nfunc (v *StickyWorkerUnavailableError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *StickyWorkerUnavailableError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *StickyWorkerUnavailableError) Error() string {\n\treturn v.String()\n}\n\ntype SupportedClientVersions struct {\n\tGoSdk   *string `json:\"goSdk,omitempty\"`\n\tJavaSdk *string `json:\"javaSdk,omitempty\"`\n}\n\n// ToWire translates a SupportedClientVersions struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SupportedClientVersions) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.GoSdk != nil {\n\t\tw, err = wire.NewValueString(*(v.GoSdk)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.JavaSdk != nil {\n\t\tw, err = wire.NewValueString(*(v.JavaSdk)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SupportedClientVersions struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SupportedClientVersions struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SupportedClientVersions\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SupportedClientVersions) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.GoSdk = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.JavaSdk = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SupportedClientVersions struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SupportedClientVersions struct could not be encoded.\nfunc (v *SupportedClientVersions) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.GoSdk != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.GoSdk)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.JavaSdk != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.JavaSdk)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SupportedClientVersions struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SupportedClientVersions struct could not be generated from the wire\n// representation.\nfunc (v *SupportedClientVersions) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.GoSdk = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.JavaSdk = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SupportedClientVersions\n// struct.\nfunc (v *SupportedClientVersions) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.GoSdk != nil {\n\t\tfields[i] = fmt.Sprintf(\"GoSdk: %v\", *(v.GoSdk))\n\t\ti++\n\t}\n\tif v.JavaSdk != nil {\n\t\tfields[i] = fmt.Sprintf(\"JavaSdk: %v\", *(v.JavaSdk))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SupportedClientVersions{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SupportedClientVersions match the\n// provided SupportedClientVersions.\n//\n// This function performs a deep comparison.\nfunc (v *SupportedClientVersions) Equals(rhs *SupportedClientVersions) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.GoSdk, rhs.GoSdk) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.JavaSdk, rhs.JavaSdk) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SupportedClientVersions.\nfunc (v *SupportedClientVersions) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.GoSdk != nil {\n\t\tenc.AddString(\"goSdk\", *v.GoSdk)\n\t}\n\tif v.JavaSdk != nil {\n\t\tenc.AddString(\"javaSdk\", *v.JavaSdk)\n\t}\n\treturn err\n}\n\n// GetGoSdk returns the value of GoSdk if it is set or its\n// zero value if it is unset.\nfunc (v *SupportedClientVersions) GetGoSdk() (o string) {\n\tif v != nil && v.GoSdk != nil {\n\t\treturn *v.GoSdk\n\t}\n\n\treturn\n}\n\n// IsSetGoSdk returns true if GoSdk is not nil.\nfunc (v *SupportedClientVersions) IsSetGoSdk() bool {\n\treturn v != nil && v.GoSdk != nil\n}\n\n// GetJavaSdk returns the value of JavaSdk if it is set or its\n// zero value if it is unset.\nfunc (v *SupportedClientVersions) GetJavaSdk() (o string) {\n\tif v != nil && v.JavaSdk != nil {\n\t\treturn *v.JavaSdk\n\t}\n\n\treturn\n}\n\n// IsSetJavaSdk returns true if JavaSdk is not nil.\nfunc (v *SupportedClientVersions) IsSetJavaSdk() bool {\n\treturn v != nil && v.JavaSdk != nil\n}\n\ntype TaskIDBlock struct {\n\tStartID *int64 `json:\"startID,omitempty\"`\n\tEndID   *int64 `json:\"endID,omitempty\"`\n}\n\n// ToWire translates a TaskIDBlock struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskIDBlock) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StartID != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.EndID != nil {\n\t\tw, err = wire.NewValueI64(*(v.EndID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskIDBlock struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskIDBlock struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskIDBlock\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskIDBlock) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EndID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskIDBlock struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskIDBlock struct could not be encoded.\nfunc (v *TaskIDBlock) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StartID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EndID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EndID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskIDBlock struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskIDBlock struct could not be generated from the wire\n// representation.\nfunc (v *TaskIDBlock) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EndID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskIDBlock\n// struct.\nfunc (v *TaskIDBlock) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.StartID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartID: %v\", *(v.StartID))\n\t\ti++\n\t}\n\tif v.EndID != nil {\n\t\tfields[i] = fmt.Sprintf(\"EndID: %v\", *(v.EndID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskIDBlock{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskIDBlock match the\n// provided TaskIDBlock.\n//\n// This function performs a deep comparison.\nfunc (v *TaskIDBlock) Equals(rhs *TaskIDBlock) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartID, rhs.StartID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EndID, rhs.EndID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskIDBlock.\nfunc (v *TaskIDBlock) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StartID != nil {\n\t\tenc.AddInt64(\"startID\", *v.StartID)\n\t}\n\tif v.EndID != nil {\n\t\tenc.AddInt64(\"endID\", *v.EndID)\n\t}\n\treturn err\n}\n\n// GetStartID returns the value of StartID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskIDBlock) GetStartID() (o int64) {\n\tif v != nil && v.StartID != nil {\n\t\treturn *v.StartID\n\t}\n\n\treturn\n}\n\n// IsSetStartID returns true if StartID is not nil.\nfunc (v *TaskIDBlock) IsSetStartID() bool {\n\treturn v != nil && v.StartID != nil\n}\n\n// GetEndID returns the value of EndID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskIDBlock) GetEndID() (o int64) {\n\tif v != nil && v.EndID != nil {\n\t\treturn *v.EndID\n\t}\n\n\treturn\n}\n\n// IsSetEndID returns true if EndID is not nil.\nfunc (v *TaskIDBlock) IsSetEndID() bool {\n\treturn v != nil && v.EndID != nil\n}\n\ntype TaskKey struct {\n\tScheduledTimeNano *int64 `json:\"scheduledTimeNano,omitempty\"`\n\tTaskID            *int64 `json:\"taskID,omitempty\"`\n}\n\n// ToWire translates a TaskKey struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskKey) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledTimeNano != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimeNano)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskKey struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskKey struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskKey\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskKey) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimeNano = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskKey struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskKey struct could not be encoded.\nfunc (v *TaskKey) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledTimeNano != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimeNano)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskKey struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskKey struct could not be generated from the wire\n// representation.\nfunc (v *TaskKey) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimeNano = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskKey\n// struct.\nfunc (v *TaskKey) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ScheduledTimeNano != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimeNano: %v\", *(v.ScheduledTimeNano))\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskKey{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskKey match the\n// provided TaskKey.\n//\n// This function performs a deep comparison.\nfunc (v *TaskKey) Equals(rhs *TaskKey) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimeNano, rhs.ScheduledTimeNano) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskKey.\nfunc (v *TaskKey) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledTimeNano != nil {\n\t\tenc.AddInt64(\"scheduledTimeNano\", *v.ScheduledTimeNano)\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\treturn err\n}\n\n// GetScheduledTimeNano returns the value of ScheduledTimeNano if it is set or its\n// zero value if it is unset.\nfunc (v *TaskKey) GetScheduledTimeNano() (o int64) {\n\tif v != nil && v.ScheduledTimeNano != nil {\n\t\treturn *v.ScheduledTimeNano\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimeNano returns true if ScheduledTimeNano is not nil.\nfunc (v *TaskKey) IsSetScheduledTimeNano() bool {\n\treturn v != nil && v.ScheduledTimeNano != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskKey) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *TaskKey) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\ntype TaskList struct {\n\tName *string       `json:\"name,omitempty\"`\n\tKind *TaskListKind `json:\"kind,omitempty\"`\n}\n\n// ToWire translates a TaskList struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskList) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Kind != nil {\n\t\tw, err = v.Kind.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListKind_Read(w wire.Value) (TaskListKind, error) {\n\tvar v TaskListKind\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a TaskList struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskList struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskList\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskList) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TaskListKind\n\t\t\t\tx, err = _TaskListKind_Read(field.Value)\n\t\t\t\tv.Kind = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskList struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskList struct could not be encoded.\nfunc (v *TaskList) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Kind != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Kind.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListKind_Decode(sr stream.Reader) (TaskListKind, error) {\n\tvar v TaskListKind\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a TaskList struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskList struct could not be generated from the wire\n// representation.\nfunc (v *TaskList) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x TaskListKind\n\t\t\tx, err = _TaskListKind_Decode(sr)\n\t\t\tv.Kind = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskList\n// struct.\nfunc (v *TaskList) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Kind != nil {\n\t\tfields[i] = fmt.Sprintf(\"Kind: %v\", *(v.Kind))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskList{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _TaskListKind_EqualsPtr(lhs, rhs *TaskListKind) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this TaskList match the\n// provided TaskList.\n//\n// This function performs a deep comparison.\nfunc (v *TaskList) Equals(rhs *TaskList) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_TaskListKind_EqualsPtr(v.Kind, rhs.Kind) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskList.\nfunc (v *TaskList) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Kind != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"kind\", *v.Kind))\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *TaskList) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *TaskList) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetKind returns the value of Kind if it is set or its\n// zero value if it is unset.\nfunc (v *TaskList) GetKind() (o TaskListKind) {\n\tif v != nil && v.Kind != nil {\n\t\treturn *v.Kind\n\t}\n\n\treturn\n}\n\n// IsSetKind returns true if Kind is not nil.\nfunc (v *TaskList) IsSetKind() bool {\n\treturn v != nil && v.Kind != nil\n}\n\ntype TaskListKind int32\n\nconst (\n\tTaskListKindNormal    TaskListKind = 0\n\tTaskListKindSticky    TaskListKind = 1\n\tTaskListKindEphemeral TaskListKind = 2\n)\n\n// TaskListKind_Values returns all recognized values of TaskListKind.\nfunc TaskListKind_Values() []TaskListKind {\n\treturn []TaskListKind{\n\t\tTaskListKindNormal,\n\t\tTaskListKindSticky,\n\t\tTaskListKindEphemeral,\n\t}\n}\n\n// UnmarshalText tries to decode TaskListKind from a byte slice\n// containing its name.\n//\n//\tvar v TaskListKind\n//\terr := v.UnmarshalText([]byte(\"NORMAL\"))\nfunc (v *TaskListKind) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"NORMAL\":\n\t\t*v = TaskListKindNormal\n\t\treturn nil\n\tcase \"STICKY\":\n\t\t*v = TaskListKindSticky\n\t\treturn nil\n\tcase \"EPHEMERAL\":\n\t\t*v = TaskListKindEphemeral\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskListKind\", err)\n\t\t}\n\t\t*v = TaskListKind(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskListKind to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v TaskListKind) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"NORMAL\"), nil\n\tcase 1:\n\t\treturn []byte(\"STICKY\"), nil\n\tcase 2:\n\t\treturn []byte(\"EPHEMERAL\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListKind.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v TaskListKind) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"NORMAL\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"STICKY\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"EPHEMERAL\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v TaskListKind) Ptr() *TaskListKind {\n\treturn &v\n}\n\n// Encode encodes TaskListKind directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v TaskListKind\n//\treturn v.Encode(sWriter)\nfunc (v TaskListKind) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates TaskListKind into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v TaskListKind) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes TaskListKind from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return TaskListKind(0), err\n//\t}\n//\n//\tvar v TaskListKind\n//\tif err := v.FromWire(x); err != nil {\n//\t  return TaskListKind(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskListKind) FromWire(w wire.Value) error {\n\t*v = (TaskListKind)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded TaskListKind directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v TaskListKind\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return TaskListKind(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskListKind) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (TaskListKind)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of TaskListKind.\nfunc (v TaskListKind) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"NORMAL\"\n\tcase 1:\n\t\treturn \"STICKY\"\n\tcase 2:\n\t\treturn \"EPHEMERAL\"\n\t}\n\treturn fmt.Sprintf(\"TaskListKind(%d)\", w)\n}\n\n// Equals returns true if this TaskListKind value matches the provided\n// value.\nfunc (v TaskListKind) Equals(rhs TaskListKind) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes TaskListKind into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v TaskListKind) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"NORMAL\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"STICKY\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"EPHEMERAL\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode TaskListKind from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *TaskListKind) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"TaskListKind\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"TaskListKind\")\n\t\t}\n\t\t*v = (TaskListKind)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"TaskListKind\")\n\t}\n}\n\ntype TaskListMetadata struct {\n\tMaxTasksPerSecond *float64 `json:\"maxTasksPerSecond,omitempty\"`\n}\n\n// ToWire translates a TaskListMetadata struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListMetadata) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.MaxTasksPerSecond != nil {\n\t\tw, err = wire.NewValueDouble(*(v.MaxTasksPerSecond)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskListMetadata struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListMetadata struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListMetadata\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListMetadata) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.MaxTasksPerSecond = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskListMetadata struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListMetadata struct could not be encoded.\nfunc (v *TaskListMetadata) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.MaxTasksPerSecond != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.MaxTasksPerSecond)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskListMetadata struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListMetadata struct could not be generated from the wire\n// representation.\nfunc (v *TaskListMetadata) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.MaxTasksPerSecond = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListMetadata\n// struct.\nfunc (v *TaskListMetadata) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.MaxTasksPerSecond != nil {\n\t\tfields[i] = fmt.Sprintf(\"MaxTasksPerSecond: %v\", *(v.MaxTasksPerSecond))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListMetadata{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskListMetadata match the\n// provided TaskListMetadata.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListMetadata) Equals(rhs *TaskListMetadata) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.MaxTasksPerSecond, rhs.MaxTasksPerSecond) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListMetadata.\nfunc (v *TaskListMetadata) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.MaxTasksPerSecond != nil {\n\t\tenc.AddFloat64(\"maxTasksPerSecond\", *v.MaxTasksPerSecond)\n\t}\n\treturn err\n}\n\n// GetMaxTasksPerSecond returns the value of MaxTasksPerSecond if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListMetadata) GetMaxTasksPerSecond() (o float64) {\n\tif v != nil && v.MaxTasksPerSecond != nil {\n\t\treturn *v.MaxTasksPerSecond\n\t}\n\n\treturn\n}\n\n// IsSetMaxTasksPerSecond returns true if MaxTasksPerSecond is not nil.\nfunc (v *TaskListMetadata) IsSetMaxTasksPerSecond() bool {\n\treturn v != nil && v.MaxTasksPerSecond != nil\n}\n\ntype TaskListNotOwnedByHostError struct {\n\tOwnedByIdentity string `json:\"ownedByIdentity,required\"`\n\tMyIdentity      string `json:\"myIdentity,required\"`\n\tTasklistName    string `json:\"tasklistName,required\"`\n}\n\n// ToWire translates a TaskListNotOwnedByHostError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListNotOwnedByHostError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.OwnedByIdentity), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.MyIdentity), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 2, Value: w}\n\ti++\n\n\tw, err = wire.NewValueString(v.TasklistName), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 3, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskListNotOwnedByHostError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListNotOwnedByHostError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListNotOwnedByHostError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListNotOwnedByHostError) FromWire(w wire.Value) error {\n\tvar err error\n\n\townedByIdentityIsSet := false\n\tmyIdentityIsSet := false\n\ttasklistNameIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.OwnedByIdentity, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\townedByIdentityIsSet = true\n\t\t\t}\n\t\tcase 2:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.MyIdentity, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmyIdentityIsSet = true\n\t\t\t}\n\t\tcase 3:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TasklistName, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\ttasklistNameIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !ownedByIdentityIsSet {\n\t\treturn errors.New(\"field OwnedByIdentity of TaskListNotOwnedByHostError is required\")\n\t}\n\n\tif !myIdentityIsSet {\n\t\treturn errors.New(\"field MyIdentity of TaskListNotOwnedByHostError is required\")\n\t}\n\n\tif !tasklistNameIsSet {\n\t\treturn errors.New(\"field TasklistName of TaskListNotOwnedByHostError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskListNotOwnedByHostError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListNotOwnedByHostError struct could not be encoded.\nfunc (v *TaskListNotOwnedByHostError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.OwnedByIdentity); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 2, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.MyIdentity); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 3, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.TasklistName); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskListNotOwnedByHostError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListNotOwnedByHostError struct could not be generated from the wire\n// representation.\nfunc (v *TaskListNotOwnedByHostError) Decode(sr stream.Reader) error {\n\n\townedByIdentityIsSet := false\n\tmyIdentityIsSet := false\n\ttasklistNameIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.OwnedByIdentity, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\townedByIdentityIsSet = true\n\t\tcase fh.ID == 2 && fh.Type == wire.TBinary:\n\t\t\tv.MyIdentity, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmyIdentityIsSet = true\n\t\tcase fh.ID == 3 && fh.Type == wire.TBinary:\n\t\t\tv.TasklistName, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\ttasklistNameIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !ownedByIdentityIsSet {\n\t\treturn errors.New(\"field OwnedByIdentity of TaskListNotOwnedByHostError is required\")\n\t}\n\n\tif !myIdentityIsSet {\n\t\treturn errors.New(\"field MyIdentity of TaskListNotOwnedByHostError is required\")\n\t}\n\n\tif !tasklistNameIsSet {\n\t\treturn errors.New(\"field TasklistName of TaskListNotOwnedByHostError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListNotOwnedByHostError\n// struct.\nfunc (v *TaskListNotOwnedByHostError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"OwnedByIdentity: %v\", v.OwnedByIdentity)\n\ti++\n\tfields[i] = fmt.Sprintf(\"MyIdentity: %v\", v.MyIdentity)\n\ti++\n\tfields[i] = fmt.Sprintf(\"TasklistName: %v\", v.TasklistName)\n\ti++\n\n\treturn fmt.Sprintf(\"TaskListNotOwnedByHostError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*TaskListNotOwnedByHostError) ErrorName() string {\n\treturn \"TaskListNotOwnedByHostError\"\n}\n\n// Equals returns true if all the fields of this TaskListNotOwnedByHostError match the\n// provided TaskListNotOwnedByHostError.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListNotOwnedByHostError) Equals(rhs *TaskListNotOwnedByHostError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.OwnedByIdentity == rhs.OwnedByIdentity) {\n\t\treturn false\n\t}\n\tif !(v.MyIdentity == rhs.MyIdentity) {\n\t\treturn false\n\t}\n\tif !(v.TasklistName == rhs.TasklistName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListNotOwnedByHostError.\nfunc (v *TaskListNotOwnedByHostError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"ownedByIdentity\", v.OwnedByIdentity)\n\tenc.AddString(\"myIdentity\", v.MyIdentity)\n\tenc.AddString(\"tasklistName\", v.TasklistName)\n\treturn err\n}\n\n// GetOwnedByIdentity returns the value of OwnedByIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListNotOwnedByHostError) GetOwnedByIdentity() (o string) {\n\tif v != nil {\n\t\to = v.OwnedByIdentity\n\t}\n\treturn\n}\n\n// GetMyIdentity returns the value of MyIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListNotOwnedByHostError) GetMyIdentity() (o string) {\n\tif v != nil {\n\t\to = v.MyIdentity\n\t}\n\treturn\n}\n\n// GetTasklistName returns the value of TasklistName if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListNotOwnedByHostError) GetTasklistName() (o string) {\n\tif v != nil {\n\t\to = v.TasklistName\n\t}\n\treturn\n}\n\nfunc (v *TaskListNotOwnedByHostError) Error() string {\n\treturn v.String()\n}\n\ntype TaskListPartitionMetadata struct {\n\tKey           *string `json:\"key,omitempty\"`\n\tOwnerHostName *string `json:\"ownerHostName,omitempty\"`\n}\n\n// ToWire translates a TaskListPartitionMetadata struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListPartitionMetadata) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Key != nil {\n\t\tw, err = wire.NewValueString(*(v.Key)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.OwnerHostName != nil {\n\t\tw, err = wire.NewValueString(*(v.OwnerHostName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskListPartitionMetadata struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListPartitionMetadata struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListPartitionMetadata\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListPartitionMetadata) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Key = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OwnerHostName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskListPartitionMetadata struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListPartitionMetadata struct could not be encoded.\nfunc (v *TaskListPartitionMetadata) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Key != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Key)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OwnerHostName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OwnerHostName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskListPartitionMetadata struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListPartitionMetadata struct could not be generated from the wire\n// representation.\nfunc (v *TaskListPartitionMetadata) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Key = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OwnerHostName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListPartitionMetadata\n// struct.\nfunc (v *TaskListPartitionMetadata) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Key != nil {\n\t\tfields[i] = fmt.Sprintf(\"Key: %v\", *(v.Key))\n\t\ti++\n\t}\n\tif v.OwnerHostName != nil {\n\t\tfields[i] = fmt.Sprintf(\"OwnerHostName: %v\", *(v.OwnerHostName))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListPartitionMetadata{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskListPartitionMetadata match the\n// provided TaskListPartitionMetadata.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListPartitionMetadata) Equals(rhs *TaskListPartitionMetadata) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Key, rhs.Key) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OwnerHostName, rhs.OwnerHostName) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListPartitionMetadata.\nfunc (v *TaskListPartitionMetadata) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Key != nil {\n\t\tenc.AddString(\"key\", *v.Key)\n\t}\n\tif v.OwnerHostName != nil {\n\t\tenc.AddString(\"ownerHostName\", *v.OwnerHostName)\n\t}\n\treturn err\n}\n\n// GetKey returns the value of Key if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionMetadata) GetKey() (o string) {\n\tif v != nil && v.Key != nil {\n\t\treturn *v.Key\n\t}\n\n\treturn\n}\n\n// IsSetKey returns true if Key is not nil.\nfunc (v *TaskListPartitionMetadata) IsSetKey() bool {\n\treturn v != nil && v.Key != nil\n}\n\n// GetOwnerHostName returns the value of OwnerHostName if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionMetadata) GetOwnerHostName() (o string) {\n\tif v != nil && v.OwnerHostName != nil {\n\t\treturn *v.OwnerHostName\n\t}\n\n\treturn\n}\n\n// IsSetOwnerHostName returns true if OwnerHostName is not nil.\nfunc (v *TaskListPartitionMetadata) IsSetOwnerHostName() bool {\n\treturn v != nil && v.OwnerHostName != nil\n}\n\ntype TaskListStatus struct {\n\tBacklogCountHint      *int64                            `json:\"backlogCountHint,omitempty\"`\n\tReadLevel             *int64                            `json:\"readLevel,omitempty\"`\n\tAckLevel              *int64                            `json:\"ackLevel,omitempty\"`\n\tRatePerSecond         *float64                          `json:\"ratePerSecond,omitempty\"`\n\tTaskIDBlock           *TaskIDBlock                      `json:\"taskIDBlock,omitempty\"`\n\tIsolationGroupMetrics map[string]*IsolationGroupMetrics `json:\"isolationGroupMetrics,omitempty\"`\n\tNewTasksPerSecond     *float64                          `json:\"newTasksPerSecond,omitempty\"`\n\tEmpty                 *bool                             `json:\"empty,omitempty\"`\n}\n\ntype _Map_String_IsolationGroupMetrics_MapItemList map[string]*IsolationGroupMetrics\n\nfunc (m _Map_String_IsolationGroupMetrics_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*IsolationGroupMetrics', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_IsolationGroupMetrics_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_IsolationGroupMetrics_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_IsolationGroupMetrics_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_String_IsolationGroupMetrics_MapItemList) Close() {}\n\n// ToWire translates a TaskListStatus struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListStatus) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [8]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BacklogCountHint != nil {\n\t\tw, err = wire.NewValueI64(*(v.BacklogCountHint)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ReadLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.ReadLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.AckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RatePerSecond != nil {\n\t\tw, err = wire.NewValueDouble(*(v.RatePerSecond)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\tif v.TaskIDBlock != nil {\n\t\tw, err = v.TaskIDBlock.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroupMetrics != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_IsolationGroupMetrics_MapItemList(v.IsolationGroupMetrics)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.NewTasksPerSecond != nil {\n\t\tw, err = wire.NewValueDouble(*(v.NewTasksPerSecond)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.Empty != nil {\n\t\tw, err = wire.NewValueBool(*(v.Empty)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskIDBlock_Read(w wire.Value) (*TaskIDBlock, error) {\n\tvar v TaskIDBlock\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _IsolationGroupMetrics_Read(w wire.Value) (*IsolationGroupMetrics, error) {\n\tvar v IsolationGroupMetrics\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_String_IsolationGroupMetrics_Read(m wire.MapItemList) (map[string]*IsolationGroupMetrics, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]*IsolationGroupMetrics, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _IsolationGroupMetrics_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a TaskListStatus struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListStatus struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListStatus) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.BacklogCountHint = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ReadLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.AckLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.RatePerSecond = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskIDBlock, err = _TaskIDBlock_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.IsolationGroupMetrics, err = _Map_String_IsolationGroupMetrics_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.NewTasksPerSecond = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.Empty = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_IsolationGroupMetrics_Encode(val map[string]*IsolationGroupMetrics, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string]*IsolationGroupMetrics', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a TaskListStatus struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListStatus struct could not be encoded.\nfunc (v *TaskListStatus) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BacklogCountHint != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.BacklogCountHint)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReadLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ReadLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.AckLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RatePerSecond != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.RatePerSecond)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskIDBlock != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskIDBlock.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroupMetrics != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_IsolationGroupMetrics_Encode(v.IsolationGroupMetrics, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewTasksPerSecond != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.NewTasksPerSecond)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Empty != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.Empty)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskIDBlock_Decode(sr stream.Reader) (*TaskIDBlock, error) {\n\tvar v TaskIDBlock\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _IsolationGroupMetrics_Decode(sr stream.Reader) (*IsolationGroupMetrics, error) {\n\tvar v IsolationGroupMetrics\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_String_IsolationGroupMetrics_Decode(sr stream.Reader) (map[string]*IsolationGroupMetrics, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]*IsolationGroupMetrics, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _IsolationGroupMetrics_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a TaskListStatus struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListStatus struct could not be generated from the wire\n// representation.\nfunc (v *TaskListStatus) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.BacklogCountHint = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ReadLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.AckLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.RatePerSecond = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.TaskIDBlock, err = _TaskIDBlock_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TMap:\n\t\t\tv.IsolationGroupMetrics, err = _Map_String_IsolationGroupMetrics_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.NewTasksPerSecond = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.Empty = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListStatus\n// struct.\nfunc (v *TaskListStatus) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [8]string\n\ti := 0\n\tif v.BacklogCountHint != nil {\n\t\tfields[i] = fmt.Sprintf(\"BacklogCountHint: %v\", *(v.BacklogCountHint))\n\t\ti++\n\t}\n\tif v.ReadLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReadLevel: %v\", *(v.ReadLevel))\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"AckLevel: %v\", *(v.AckLevel))\n\t\ti++\n\t}\n\tif v.RatePerSecond != nil {\n\t\tfields[i] = fmt.Sprintf(\"RatePerSecond: %v\", *(v.RatePerSecond))\n\t\ti++\n\t}\n\tif v.TaskIDBlock != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskIDBlock: %v\", v.TaskIDBlock)\n\t\ti++\n\t}\n\tif v.IsolationGroupMetrics != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroupMetrics: %v\", v.IsolationGroupMetrics)\n\t\ti++\n\t}\n\tif v.NewTasksPerSecond != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewTasksPerSecond: %v\", *(v.NewTasksPerSecond))\n\t\ti++\n\t}\n\tif v.Empty != nil {\n\t\tfields[i] = fmt.Sprintf(\"Empty: %v\", *(v.Empty))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListStatus{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_IsolationGroupMetrics_Equals(lhs, rhs map[string]*IsolationGroupMetrics) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this TaskListStatus match the\n// provided TaskListStatus.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListStatus) Equals(rhs *TaskListStatus) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.BacklogCountHint, rhs.BacklogCountHint) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ReadLevel, rhs.ReadLevel) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.AckLevel, rhs.AckLevel) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.RatePerSecond, rhs.RatePerSecond) {\n\t\treturn false\n\t}\n\tif !((v.TaskIDBlock == nil && rhs.TaskIDBlock == nil) || (v.TaskIDBlock != nil && rhs.TaskIDBlock != nil && v.TaskIDBlock.Equals(rhs.TaskIDBlock))) {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroupMetrics == nil && rhs.IsolationGroupMetrics == nil) || (v.IsolationGroupMetrics != nil && rhs.IsolationGroupMetrics != nil && _Map_String_IsolationGroupMetrics_Equals(v.IsolationGroupMetrics, rhs.IsolationGroupMetrics))) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.NewTasksPerSecond, rhs.NewTasksPerSecond) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.Empty, rhs.Empty) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_IsolationGroupMetrics_Zapper map[string]*IsolationGroupMetrics\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_IsolationGroupMetrics_Zapper.\nfunc (m _Map_String_IsolationGroupMetrics_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AddObject((string)(k), v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListStatus.\nfunc (v *TaskListStatus) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BacklogCountHint != nil {\n\t\tenc.AddInt64(\"backlogCountHint\", *v.BacklogCountHint)\n\t}\n\tif v.ReadLevel != nil {\n\t\tenc.AddInt64(\"readLevel\", *v.ReadLevel)\n\t}\n\tif v.AckLevel != nil {\n\t\tenc.AddInt64(\"ackLevel\", *v.AckLevel)\n\t}\n\tif v.RatePerSecond != nil {\n\t\tenc.AddFloat64(\"ratePerSecond\", *v.RatePerSecond)\n\t}\n\tif v.TaskIDBlock != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskIDBlock\", v.TaskIDBlock))\n\t}\n\tif v.IsolationGroupMetrics != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"isolationGroupMetrics\", (_Map_String_IsolationGroupMetrics_Zapper)(v.IsolationGroupMetrics)))\n\t}\n\tif v.NewTasksPerSecond != nil {\n\t\tenc.AddFloat64(\"newTasksPerSecond\", *v.NewTasksPerSecond)\n\t}\n\tif v.Empty != nil {\n\t\tenc.AddBool(\"empty\", *v.Empty)\n\t}\n\treturn err\n}\n\n// GetBacklogCountHint returns the value of BacklogCountHint if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetBacklogCountHint() (o int64) {\n\tif v != nil && v.BacklogCountHint != nil {\n\t\treturn *v.BacklogCountHint\n\t}\n\n\treturn\n}\n\n// IsSetBacklogCountHint returns true if BacklogCountHint is not nil.\nfunc (v *TaskListStatus) IsSetBacklogCountHint() bool {\n\treturn v != nil && v.BacklogCountHint != nil\n}\n\n// GetReadLevel returns the value of ReadLevel if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetReadLevel() (o int64) {\n\tif v != nil && v.ReadLevel != nil {\n\t\treturn *v.ReadLevel\n\t}\n\n\treturn\n}\n\n// IsSetReadLevel returns true if ReadLevel is not nil.\nfunc (v *TaskListStatus) IsSetReadLevel() bool {\n\treturn v != nil && v.ReadLevel != nil\n}\n\n// GetAckLevel returns the value of AckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetAckLevel() (o int64) {\n\tif v != nil && v.AckLevel != nil {\n\t\treturn *v.AckLevel\n\t}\n\n\treturn\n}\n\n// IsSetAckLevel returns true if AckLevel is not nil.\nfunc (v *TaskListStatus) IsSetAckLevel() bool {\n\treturn v != nil && v.AckLevel != nil\n}\n\n// GetRatePerSecond returns the value of RatePerSecond if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetRatePerSecond() (o float64) {\n\tif v != nil && v.RatePerSecond != nil {\n\t\treturn *v.RatePerSecond\n\t}\n\n\treturn\n}\n\n// IsSetRatePerSecond returns true if RatePerSecond is not nil.\nfunc (v *TaskListStatus) IsSetRatePerSecond() bool {\n\treturn v != nil && v.RatePerSecond != nil\n}\n\n// GetTaskIDBlock returns the value of TaskIDBlock if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetTaskIDBlock() (o *TaskIDBlock) {\n\tif v != nil && v.TaskIDBlock != nil {\n\t\treturn v.TaskIDBlock\n\t}\n\n\treturn\n}\n\n// IsSetTaskIDBlock returns true if TaskIDBlock is not nil.\nfunc (v *TaskListStatus) IsSetTaskIDBlock() bool {\n\treturn v != nil && v.TaskIDBlock != nil\n}\n\n// GetIsolationGroupMetrics returns the value of IsolationGroupMetrics if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetIsolationGroupMetrics() (o map[string]*IsolationGroupMetrics) {\n\tif v != nil && v.IsolationGroupMetrics != nil {\n\t\treturn v.IsolationGroupMetrics\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroupMetrics returns true if IsolationGroupMetrics is not nil.\nfunc (v *TaskListStatus) IsSetIsolationGroupMetrics() bool {\n\treturn v != nil && v.IsolationGroupMetrics != nil\n}\n\n// GetNewTasksPerSecond returns the value of NewTasksPerSecond if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetNewTasksPerSecond() (o float64) {\n\tif v != nil && v.NewTasksPerSecond != nil {\n\t\treturn *v.NewTasksPerSecond\n\t}\n\n\treturn\n}\n\n// IsSetNewTasksPerSecond returns true if NewTasksPerSecond is not nil.\nfunc (v *TaskListStatus) IsSetNewTasksPerSecond() bool {\n\treturn v != nil && v.NewTasksPerSecond != nil\n}\n\n// GetEmpty returns the value of Empty if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListStatus) GetEmpty() (o bool) {\n\tif v != nil && v.Empty != nil {\n\t\treturn *v.Empty\n\t}\n\n\treturn\n}\n\n// IsSetEmpty returns true if Empty is not nil.\nfunc (v *TaskListStatus) IsSetEmpty() bool {\n\treturn v != nil && v.Empty != nil\n}\n\ntype TaskListType int32\n\nconst (\n\tTaskListTypeDecision TaskListType = 0\n\tTaskListTypeActivity TaskListType = 1\n)\n\n// TaskListType_Values returns all recognized values of TaskListType.\nfunc TaskListType_Values() []TaskListType {\n\treturn []TaskListType{\n\t\tTaskListTypeDecision,\n\t\tTaskListTypeActivity,\n\t}\n}\n\n// UnmarshalText tries to decode TaskListType from a byte slice\n// containing its name.\n//\n//\tvar v TaskListType\n//\terr := v.UnmarshalText([]byte(\"Decision\"))\nfunc (v *TaskListType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"Decision\":\n\t\t*v = TaskListTypeDecision\n\t\treturn nil\n\tcase \"Activity\":\n\t\t*v = TaskListTypeActivity\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskListType\", err)\n\t\t}\n\t\t*v = TaskListType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskListType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v TaskListType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"Decision\"), nil\n\tcase 1:\n\t\treturn []byte(\"Activity\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v TaskListType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"Decision\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"Activity\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v TaskListType) Ptr() *TaskListType {\n\treturn &v\n}\n\n// Encode encodes TaskListType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v TaskListType\n//\treturn v.Encode(sWriter)\nfunc (v TaskListType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates TaskListType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v TaskListType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes TaskListType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return TaskListType(0), err\n//\t}\n//\n//\tvar v TaskListType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return TaskListType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskListType) FromWire(w wire.Value) error {\n\t*v = (TaskListType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded TaskListType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v TaskListType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return TaskListType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TaskListType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (TaskListType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of TaskListType.\nfunc (v TaskListType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Decision\"\n\tcase 1:\n\t\treturn \"Activity\"\n\t}\n\treturn fmt.Sprintf(\"TaskListType(%d)\", w)\n}\n\n// Equals returns true if this TaskListType value matches the provided\n// value.\nfunc (v TaskListType) Equals(rhs TaskListType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes TaskListType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v TaskListType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"Decision\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"Activity\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode TaskListType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *TaskListType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"TaskListType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"TaskListType\")\n\t\t}\n\t\t*v = (TaskListType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"TaskListType\")\n\t}\n}\n\ntype TaskRange struct {\n\tInclusiveMin *TaskKey `json:\"inclusiveMin,omitempty\"`\n\tExclusiveMax *TaskKey `json:\"exclusiveMax,omitempty\"`\n}\n\n// ToWire translates a TaskRange struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskRange) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.InclusiveMin != nil {\n\t\tw, err = v.InclusiveMin.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExclusiveMax != nil {\n\t\tw, err = v.ExclusiveMax.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskRange struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskRange struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskRange\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskRange) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.InclusiveMin, err = _TaskKey_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExclusiveMax, err = _TaskKey_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskRange struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskRange struct could not be encoded.\nfunc (v *TaskRange) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.InclusiveMin != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.InclusiveMin.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExclusiveMax != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExclusiveMax.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskRange struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskRange struct could not be generated from the wire\n// representation.\nfunc (v *TaskRange) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.InclusiveMin, err = _TaskKey_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.ExclusiveMax, err = _TaskKey_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskRange\n// struct.\nfunc (v *TaskRange) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.InclusiveMin != nil {\n\t\tfields[i] = fmt.Sprintf(\"InclusiveMin: %v\", v.InclusiveMin)\n\t\ti++\n\t}\n\tif v.ExclusiveMax != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExclusiveMax: %v\", v.ExclusiveMax)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskRange{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskRange match the\n// provided TaskRange.\n//\n// This function performs a deep comparison.\nfunc (v *TaskRange) Equals(rhs *TaskRange) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.InclusiveMin == nil && rhs.InclusiveMin == nil) || (v.InclusiveMin != nil && rhs.InclusiveMin != nil && v.InclusiveMin.Equals(rhs.InclusiveMin))) {\n\t\treturn false\n\t}\n\tif !((v.ExclusiveMax == nil && rhs.ExclusiveMax == nil) || (v.ExclusiveMax != nil && rhs.ExclusiveMax != nil && v.ExclusiveMax.Equals(rhs.ExclusiveMax))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskRange.\nfunc (v *TaskRange) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.InclusiveMin != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"inclusiveMin\", v.InclusiveMin))\n\t}\n\tif v.ExclusiveMax != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"exclusiveMax\", v.ExclusiveMax))\n\t}\n\treturn err\n}\n\n// GetInclusiveMin returns the value of InclusiveMin if it is set or its\n// zero value if it is unset.\nfunc (v *TaskRange) GetInclusiveMin() (o *TaskKey) {\n\tif v != nil && v.InclusiveMin != nil {\n\t\treturn v.InclusiveMin\n\t}\n\n\treturn\n}\n\n// IsSetInclusiveMin returns true if InclusiveMin is not nil.\nfunc (v *TaskRange) IsSetInclusiveMin() bool {\n\treturn v != nil && v.InclusiveMin != nil\n}\n\n// GetExclusiveMax returns the value of ExclusiveMax if it is set or its\n// zero value if it is unset.\nfunc (v *TaskRange) GetExclusiveMax() (o *TaskKey) {\n\tif v != nil && v.ExclusiveMax != nil {\n\t\treturn v.ExclusiveMax\n\t}\n\n\treturn\n}\n\n// IsSetExclusiveMax returns true if ExclusiveMax is not nil.\nfunc (v *TaskRange) IsSetExclusiveMax() bool {\n\treturn v != nil && v.ExclusiveMax != nil\n}\n\ntype TerminateWorkflowExecutionRequest struct {\n\tDomain              *string            `json:\"domain,omitempty\"`\n\tWorkflowExecution   *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tReason              *string            `json:\"reason,omitempty\"`\n\tDetails             []byte             `json:\"details,omitempty\"`\n\tIdentity            *string            `json:\"identity,omitempty\"`\n\tFirstExecutionRunID *string            `json:\"firstExecutionRunID,omitempty\"`\n}\n\n// ToWire translates a TerminateWorkflowExecutionRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TerminateWorkflowExecutionRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Domain != nil {\n\t\tw, err = wire.NewValueString(*(v.Domain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tw, err = v.WorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tw, err = wire.NewValueString(*(v.FirstExecutionRunID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TerminateWorkflowExecutionRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TerminateWorkflowExecutionRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TerminateWorkflowExecutionRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TerminateWorkflowExecutionRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Domain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FirstExecutionRunID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TerminateWorkflowExecutionRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TerminateWorkflowExecutionRequest struct could not be encoded.\nfunc (v *TerminateWorkflowExecutionRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Domain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Domain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstExecutionRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FirstExecutionRunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TerminateWorkflowExecutionRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TerminateWorkflowExecutionRequest struct could not be generated from the wire\n// representation.\nfunc (v *TerminateWorkflowExecutionRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Domain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FirstExecutionRunID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TerminateWorkflowExecutionRequest\n// struct.\nfunc (v *TerminateWorkflowExecutionRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Domain != nil {\n\t\tfields[i] = fmt.Sprintf(\"Domain: %v\", *(v.Domain))\n\t\ti++\n\t}\n\tif v.WorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowExecution: %v\", v.WorkflowExecution)\n\t\ti++\n\t}\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstExecutionRunID: %v\", *(v.FirstExecutionRunID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TerminateWorkflowExecutionRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TerminateWorkflowExecutionRequest match the\n// provided TerminateWorkflowExecutionRequest.\n//\n// This function performs a deep comparison.\nfunc (v *TerminateWorkflowExecutionRequest) Equals(rhs *TerminateWorkflowExecutionRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Domain, rhs.Domain) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowExecution == nil && rhs.WorkflowExecution == nil) || (v.WorkflowExecution != nil && rhs.WorkflowExecution != nil && v.WorkflowExecution.Equals(rhs.WorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FirstExecutionRunID, rhs.FirstExecutionRunID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TerminateWorkflowExecutionRequest.\nfunc (v *TerminateWorkflowExecutionRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Domain != nil {\n\t\tenc.AddString(\"domain\", *v.Domain)\n\t}\n\tif v.WorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowExecution\", v.WorkflowExecution))\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tenc.AddString(\"firstExecutionRunID\", *v.FirstExecutionRunID)\n\t}\n\treturn err\n}\n\n// GetDomain returns the value of Domain if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\n\treturn\n}\n\n// IsSetDomain returns true if Domain is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetDomain() bool {\n\treturn v != nil && v.Domain != nil\n}\n\n// GetWorkflowExecution returns the value of WorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowExecution returns true if WorkflowExecution is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetWorkflowExecution() bool {\n\treturn v != nil && v.WorkflowExecution != nil\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetFirstExecutionRunID returns the value of FirstExecutionRunID if it is set or its\n// zero value if it is unset.\nfunc (v *TerminateWorkflowExecutionRequest) GetFirstExecutionRunID() (o string) {\n\tif v != nil && v.FirstExecutionRunID != nil {\n\t\treturn *v.FirstExecutionRunID\n\t}\n\n\treturn\n}\n\n// IsSetFirstExecutionRunID returns true if FirstExecutionRunID is not nil.\nfunc (v *TerminateWorkflowExecutionRequest) IsSetFirstExecutionRunID() bool {\n\treturn v != nil && v.FirstExecutionRunID != nil\n}\n\ntype TimeoutType int32\n\nconst (\n\tTimeoutTypeStartToClose    TimeoutType = 0\n\tTimeoutTypeScheduleToStart TimeoutType = 1\n\tTimeoutTypeScheduleToClose TimeoutType = 2\n\tTimeoutTypeHeartbeat       TimeoutType = 3\n)\n\n// TimeoutType_Values returns all recognized values of TimeoutType.\nfunc TimeoutType_Values() []TimeoutType {\n\treturn []TimeoutType{\n\t\tTimeoutTypeStartToClose,\n\t\tTimeoutTypeScheduleToStart,\n\t\tTimeoutTypeScheduleToClose,\n\t\tTimeoutTypeHeartbeat,\n\t}\n}\n\n// UnmarshalText tries to decode TimeoutType from a byte slice\n// containing its name.\n//\n//\tvar v TimeoutType\n//\terr := v.UnmarshalText([]byte(\"START_TO_CLOSE\"))\nfunc (v *TimeoutType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"START_TO_CLOSE\":\n\t\t*v = TimeoutTypeStartToClose\n\t\treturn nil\n\tcase \"SCHEDULE_TO_START\":\n\t\t*v = TimeoutTypeScheduleToStart\n\t\treturn nil\n\tcase \"SCHEDULE_TO_CLOSE\":\n\t\t*v = TimeoutTypeScheduleToClose\n\t\treturn nil\n\tcase \"HEARTBEAT\":\n\t\t*v = TimeoutTypeHeartbeat\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TimeoutType\", err)\n\t\t}\n\t\t*v = TimeoutType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TimeoutType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v TimeoutType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"START_TO_CLOSE\"), nil\n\tcase 1:\n\t\treturn []byte(\"SCHEDULE_TO_START\"), nil\n\tcase 2:\n\t\treturn []byte(\"SCHEDULE_TO_CLOSE\"), nil\n\tcase 3:\n\t\treturn []byte(\"HEARTBEAT\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimeoutType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v TimeoutType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"START_TO_CLOSE\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"SCHEDULE_TO_START\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"SCHEDULE_TO_CLOSE\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"HEARTBEAT\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v TimeoutType) Ptr() *TimeoutType {\n\treturn &v\n}\n\n// Encode encodes TimeoutType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v TimeoutType\n//\treturn v.Encode(sWriter)\nfunc (v TimeoutType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates TimeoutType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v TimeoutType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes TimeoutType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return TimeoutType(0), err\n//\t}\n//\n//\tvar v TimeoutType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return TimeoutType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TimeoutType) FromWire(w wire.Value) error {\n\t*v = (TimeoutType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded TimeoutType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v TimeoutType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return TimeoutType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *TimeoutType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (TimeoutType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of TimeoutType.\nfunc (v TimeoutType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"START_TO_CLOSE\"\n\tcase 1:\n\t\treturn \"SCHEDULE_TO_START\"\n\tcase 2:\n\t\treturn \"SCHEDULE_TO_CLOSE\"\n\tcase 3:\n\t\treturn \"HEARTBEAT\"\n\t}\n\treturn fmt.Sprintf(\"TimeoutType(%d)\", w)\n}\n\n// Equals returns true if this TimeoutType value matches the provided\n// value.\nfunc (v TimeoutType) Equals(rhs TimeoutType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes TimeoutType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v TimeoutType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"START_TO_CLOSE\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"SCHEDULE_TO_START\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"SCHEDULE_TO_CLOSE\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"HEARTBEAT\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode TimeoutType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *TimeoutType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"TimeoutType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"TimeoutType\")\n\t\t}\n\t\t*v = (TimeoutType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"TimeoutType\")\n\t}\n}\n\ntype TimerCanceledEventAttributes struct {\n\tTimerId                      *string `json:\"timerId,omitempty\"`\n\tStartedEventId               *int64  `json:\"startedEventId,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tIdentity                     *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a TimerCanceledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerCanceledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerCanceledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerCanceledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerCanceledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerCanceledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerCanceledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerCanceledEventAttributes struct could not be encoded.\nfunc (v *TimerCanceledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerCanceledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerCanceledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *TimerCanceledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerCanceledEventAttributes\n// struct.\nfunc (v *TimerCanceledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerCanceledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerCanceledEventAttributes match the\n// provided TimerCanceledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *TimerCanceledEventAttributes) Equals(rhs *TimerCanceledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerCanceledEventAttributes.\nfunc (v *TimerCanceledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerCanceledEventAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *TimerCanceledEventAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerCanceledEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *TimerCanceledEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerCanceledEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *TimerCanceledEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *TimerCanceledEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *TimerCanceledEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype TimerFiredEventAttributes struct {\n\tTimerId        *string `json:\"timerId,omitempty\"`\n\tStartedEventId *int64  `json:\"startedEventId,omitempty\"`\n}\n\n// ToWire translates a TimerFiredEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerFiredEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerFiredEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerFiredEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerFiredEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerFiredEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerFiredEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerFiredEventAttributes struct could not be encoded.\nfunc (v *TimerFiredEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerFiredEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerFiredEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *TimerFiredEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerFiredEventAttributes\n// struct.\nfunc (v *TimerFiredEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\tif v.StartedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventId: %v\", *(v.StartedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerFiredEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerFiredEventAttributes match the\n// provided TimerFiredEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *TimerFiredEventAttributes) Equals(rhs *TimerFiredEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedEventId, rhs.StartedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerFiredEventAttributes.\nfunc (v *TimerFiredEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\tif v.StartedEventId != nil {\n\t\tenc.AddInt64(\"startedEventId\", *v.StartedEventId)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerFiredEventAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *TimerFiredEventAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\n// GetStartedEventId returns the value of StartedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerFiredEventAttributes) GetStartedEventId() (o int64) {\n\tif v != nil && v.StartedEventId != nil {\n\t\treturn *v.StartedEventId\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventId returns true if StartedEventId is not nil.\nfunc (v *TimerFiredEventAttributes) IsSetStartedEventId() bool {\n\treturn v != nil && v.StartedEventId != nil\n}\n\ntype TimerStartedEventAttributes struct {\n\tTimerId                      *string `json:\"timerId,omitempty\"`\n\tStartToFireTimeoutSeconds    *int64  `json:\"startToFireTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a TimerStartedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerStartedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimerId != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartToFireTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerStartedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerStartedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerStartedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerStartedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartToFireTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerStartedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerStartedEventAttributes struct could not be encoded.\nfunc (v *TimerStartedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimerId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartToFireTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerStartedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerStartedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *TimerStartedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartToFireTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerStartedEventAttributes\n// struct.\nfunc (v *TimerStartedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TimerId != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerId: %v\", *(v.TimerId))\n\t\ti++\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToFireTimeoutSeconds: %v\", *(v.StartToFireTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerStartedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerStartedEventAttributes match the\n// provided TimerStartedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *TimerStartedEventAttributes) Equals(rhs *TimerStartedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerId, rhs.TimerId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartToFireTimeoutSeconds, rhs.StartToFireTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerStartedEventAttributes.\nfunc (v *TimerStartedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimerId != nil {\n\t\tenc.AddString(\"timerId\", *v.TimerId)\n\t}\n\tif v.StartToFireTimeoutSeconds != nil {\n\t\tenc.AddInt64(\"startToFireTimeoutSeconds\", *v.StartToFireTimeoutSeconds)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetTimerId returns the value of TimerId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerStartedEventAttributes) GetTimerId() (o string) {\n\tif v != nil && v.TimerId != nil {\n\t\treturn *v.TimerId\n\t}\n\n\treturn\n}\n\n// IsSetTimerId returns true if TimerId is not nil.\nfunc (v *TimerStartedEventAttributes) IsSetTimerId() bool {\n\treturn v != nil && v.TimerId != nil\n}\n\n// GetStartToFireTimeoutSeconds returns the value of StartToFireTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *TimerStartedEventAttributes) GetStartToFireTimeoutSeconds() (o int64) {\n\tif v != nil && v.StartToFireTimeoutSeconds != nil {\n\t\treturn *v.StartToFireTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToFireTimeoutSeconds returns true if StartToFireTimeoutSeconds is not nil.\nfunc (v *TimerStartedEventAttributes) IsSetStartToFireTimeoutSeconds() bool {\n\treturn v != nil && v.StartToFireTimeoutSeconds != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *TimerStartedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *TimerStartedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype TransientDecisionInfo struct {\n\tScheduledEvent *HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedEvent   *HistoryEvent `json:\"startedEvent,omitempty\"`\n}\n\n// ToWire translates a TransientDecisionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TransientDecisionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ScheduledEvent != nil {\n\t\tw, err = v.ScheduledEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tw, err = v.StartedEvent.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TransientDecisionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TransientDecisionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TransientDecisionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TransientDecisionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ScheduledEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.StartedEvent, err = _HistoryEvent_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TransientDecisionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TransientDecisionInfo struct could not be encoded.\nfunc (v *TransientDecisionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ScheduledEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ScheduledEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.StartedEvent.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TransientDecisionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TransientDecisionInfo struct could not be generated from the wire\n// representation.\nfunc (v *TransientDecisionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.ScheduledEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.StartedEvent, err = _HistoryEvent_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TransientDecisionInfo\n// struct.\nfunc (v *TransientDecisionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.ScheduledEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEvent: %v\", v.ScheduledEvent)\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEvent: %v\", v.StartedEvent)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TransientDecisionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TransientDecisionInfo match the\n// provided TransientDecisionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TransientDecisionInfo) Equals(rhs *TransientDecisionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ScheduledEvent == nil && rhs.ScheduledEvent == nil) || (v.ScheduledEvent != nil && rhs.ScheduledEvent != nil && v.ScheduledEvent.Equals(rhs.ScheduledEvent))) {\n\t\treturn false\n\t}\n\tif !((v.StartedEvent == nil && rhs.StartedEvent == nil) || (v.StartedEvent != nil && rhs.StartedEvent != nil && v.StartedEvent.Equals(rhs.StartedEvent))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TransientDecisionInfo.\nfunc (v *TransientDecisionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ScheduledEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"scheduledEvent\", v.ScheduledEvent))\n\t}\n\tif v.StartedEvent != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"startedEvent\", v.StartedEvent))\n\t}\n\treturn err\n}\n\n// GetScheduledEvent returns the value of ScheduledEvent if it is set or its\n// zero value if it is unset.\nfunc (v *TransientDecisionInfo) GetScheduledEvent() (o *HistoryEvent) {\n\tif v != nil && v.ScheduledEvent != nil {\n\t\treturn v.ScheduledEvent\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEvent returns true if ScheduledEvent is not nil.\nfunc (v *TransientDecisionInfo) IsSetScheduledEvent() bool {\n\treturn v != nil && v.ScheduledEvent != nil\n}\n\n// GetStartedEvent returns the value of StartedEvent if it is set or its\n// zero value if it is unset.\nfunc (v *TransientDecisionInfo) GetStartedEvent() (o *HistoryEvent) {\n\tif v != nil && v.StartedEvent != nil {\n\t\treturn v.StartedEvent\n\t}\n\n\treturn\n}\n\n// IsSetStartedEvent returns true if StartedEvent is not nil.\nfunc (v *TransientDecisionInfo) IsSetStartedEvent() bool {\n\treturn v != nil && v.StartedEvent != nil\n}\n\ntype UniversalPredicateAttributes struct {\n}\n\n// ToWire translates a UniversalPredicateAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UniversalPredicateAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [0]wire.Field\n\t\ti      int = 0\n\t)\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UniversalPredicateAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UniversalPredicateAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UniversalPredicateAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UniversalPredicateAttributes) FromWire(w wire.Value) error {\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UniversalPredicateAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UniversalPredicateAttributes struct could not be encoded.\nfunc (v *UniversalPredicateAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UniversalPredicateAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UniversalPredicateAttributes struct could not be generated from the wire\n// representation.\nfunc (v *UniversalPredicateAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UniversalPredicateAttributes\n// struct.\nfunc (v *UniversalPredicateAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [0]string\n\ti := 0\n\n\treturn fmt.Sprintf(\"UniversalPredicateAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UniversalPredicateAttributes match the\n// provided UniversalPredicateAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *UniversalPredicateAttributes) Equals(rhs *UniversalPredicateAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UniversalPredicateAttributes.\nfunc (v *UniversalPredicateAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn err\n}\n\ntype UpdateDomainInfo struct {\n\tDescription *string           `json:\"description,omitempty\"`\n\tOwnerEmail  *string           `json:\"ownerEmail,omitempty\"`\n\tData        map[string]string `json:\"data,omitempty\"`\n}\n\n// ToWire translates a UpdateDomainInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Description != nil {\n\t\tw, err = wire.NewValueString(*(v.Description)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tw, err = wire.NewValueString(*(v.OwnerEmail)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.Data)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Description = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OwnerEmail = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Data, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainInfo struct could not be encoded.\nfunc (v *UpdateDomainInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Description != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Description)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OwnerEmail != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OwnerEmail)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.Data, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainInfo struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Description = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OwnerEmail = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TMap:\n\t\t\tv.Data, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainInfo\n// struct.\nfunc (v *UpdateDomainInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Description != nil {\n\t\tfields[i] = fmt.Sprintf(\"Description: %v\", *(v.Description))\n\t\ti++\n\t}\n\tif v.OwnerEmail != nil {\n\t\tfields[i] = fmt.Sprintf(\"OwnerEmail: %v\", *(v.OwnerEmail))\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDomainInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainInfo match the\n// provided UpdateDomainInfo.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainInfo) Equals(rhs *UpdateDomainInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Description, rhs.Description) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OwnerEmail, rhs.OwnerEmail) {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && _Map_String_String_Equals(v.Data, rhs.Data))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainInfo.\nfunc (v *UpdateDomainInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Description != nil {\n\t\tenc.AddString(\"description\", *v.Description)\n\t}\n\tif v.OwnerEmail != nil {\n\t\tenc.AddString(\"ownerEmail\", *v.OwnerEmail)\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", (_Map_String_String_Zapper)(v.Data)))\n\t}\n\treturn err\n}\n\n// GetDescription returns the value of Description if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainInfo) GetDescription() (o string) {\n\tif v != nil && v.Description != nil {\n\t\treturn *v.Description\n\t}\n\n\treturn\n}\n\n// IsSetDescription returns true if Description is not nil.\nfunc (v *UpdateDomainInfo) IsSetDescription() bool {\n\treturn v != nil && v.Description != nil\n}\n\n// GetOwnerEmail returns the value of OwnerEmail if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainInfo) GetOwnerEmail() (o string) {\n\tif v != nil && v.OwnerEmail != nil {\n\t\treturn *v.OwnerEmail\n\t}\n\n\treturn\n}\n\n// IsSetOwnerEmail returns true if OwnerEmail is not nil.\nfunc (v *UpdateDomainInfo) IsSetOwnerEmail() bool {\n\treturn v != nil && v.OwnerEmail != nil\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainInfo) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *UpdateDomainInfo) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\ntype UpdateDomainRequest struct {\n\tName                     *string                         `json:\"name,omitempty\"`\n\tUpdatedInfo              *UpdateDomainInfo               `json:\"updatedInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tSecurityToken            *string                         `json:\"securityToken,omitempty\"`\n\tDeleteBadBinary          *string                         `json:\"deleteBadBinary,omitempty\"`\n\tFailoverTimeoutInSeconds *int32                          `json:\"failoverTimeoutInSeconds,omitempty\"`\n}\n\n// ToWire translates a UpdateDomainRequest struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainRequest) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [7]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.UpdatedInfo != nil {\n\t\tw, err = v.UpdatedInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tw, err = v.ReplicationConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tw, err = wire.NewValueString(*(v.SecurityToken)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.DeleteBadBinary != nil {\n\t\tw, err = wire.NewValueString(*(v.DeleteBadBinary)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverTimeoutInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.FailoverTimeoutInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _UpdateDomainInfo_Read(w wire.Value) (*UpdateDomainInfo, error) {\n\tvar v UpdateDomainInfo\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a UpdateDomainRequest struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainRequest struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainRequest\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainRequest) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.UpdatedInfo, err = _UpdateDomainInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _DomainConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SecurityToken = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DeleteBadBinary = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.FailoverTimeoutInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainRequest struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainRequest struct could not be encoded.\nfunc (v *UpdateDomainRequest) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UpdatedInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.UpdatedInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicationConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SecurityToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SecurityToken)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DeleteBadBinary != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DeleteBadBinary)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverTimeoutInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.FailoverTimeoutInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _UpdateDomainInfo_Decode(sr stream.Reader) (*UpdateDomainInfo, error) {\n\tvar v UpdateDomainInfo\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a UpdateDomainRequest struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainRequest struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainRequest) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.UpdatedInfo, err = _UpdateDomainInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _DomainConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SecurityToken = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DeleteBadBinary = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.FailoverTimeoutInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainRequest\n// struct.\nfunc (v *UpdateDomainRequest) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [7]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.UpdatedInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpdatedInfo: %v\", v.UpdatedInfo)\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationConfiguration: %v\", v.ReplicationConfiguration)\n\t\ti++\n\t}\n\tif v.SecurityToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"SecurityToken: %v\", *(v.SecurityToken))\n\t\ti++\n\t}\n\tif v.DeleteBadBinary != nil {\n\t\tfields[i] = fmt.Sprintf(\"DeleteBadBinary: %v\", *(v.DeleteBadBinary))\n\t\ti++\n\t}\n\tif v.FailoverTimeoutInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverTimeoutInSeconds: %v\", *(v.FailoverTimeoutInSeconds))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDomainRequest{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainRequest match the\n// provided UpdateDomainRequest.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainRequest) Equals(rhs *UpdateDomainRequest) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !((v.UpdatedInfo == nil && rhs.UpdatedInfo == nil) || (v.UpdatedInfo != nil && rhs.UpdatedInfo != nil && v.UpdatedInfo.Equals(rhs.UpdatedInfo))) {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationConfiguration == nil && rhs.ReplicationConfiguration == nil) || (v.ReplicationConfiguration != nil && rhs.ReplicationConfiguration != nil && v.ReplicationConfiguration.Equals(rhs.ReplicationConfiguration))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SecurityToken, rhs.SecurityToken) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DeleteBadBinary, rhs.DeleteBadBinary) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.FailoverTimeoutInSeconds, rhs.FailoverTimeoutInSeconds) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainRequest.\nfunc (v *UpdateDomainRequest) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.UpdatedInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"updatedInfo\", v.UpdatedInfo))\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationConfiguration\", v.ReplicationConfiguration))\n\t}\n\tif v.SecurityToken != nil {\n\t\tenc.AddString(\"securityToken\", *v.SecurityToken)\n\t}\n\tif v.DeleteBadBinary != nil {\n\t\tenc.AddString(\"deleteBadBinary\", *v.DeleteBadBinary)\n\t}\n\tif v.FailoverTimeoutInSeconds != nil {\n\t\tenc.AddInt32(\"failoverTimeoutInSeconds\", *v.FailoverTimeoutInSeconds)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *UpdateDomainRequest) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetUpdatedInfo returns the value of UpdatedInfo if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetUpdatedInfo() (o *UpdateDomainInfo) {\n\tif v != nil && v.UpdatedInfo != nil {\n\t\treturn v.UpdatedInfo\n\t}\n\n\treturn\n}\n\n// IsSetUpdatedInfo returns true if UpdatedInfo is not nil.\nfunc (v *UpdateDomainRequest) IsSetUpdatedInfo() bool {\n\treturn v != nil && v.UpdatedInfo != nil\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetConfiguration() (o *DomainConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *UpdateDomainRequest) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\n// GetReplicationConfiguration returns the value of ReplicationConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetReplicationConfiguration() (o *DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfiguration != nil {\n\t\treturn v.ReplicationConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetReplicationConfiguration returns true if ReplicationConfiguration is not nil.\nfunc (v *UpdateDomainRequest) IsSetReplicationConfiguration() bool {\n\treturn v != nil && v.ReplicationConfiguration != nil\n}\n\n// GetSecurityToken returns the value of SecurityToken if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetSecurityToken() (o string) {\n\tif v != nil && v.SecurityToken != nil {\n\t\treturn *v.SecurityToken\n\t}\n\n\treturn\n}\n\n// IsSetSecurityToken returns true if SecurityToken is not nil.\nfunc (v *UpdateDomainRequest) IsSetSecurityToken() bool {\n\treturn v != nil && v.SecurityToken != nil\n}\n\n// GetDeleteBadBinary returns the value of DeleteBadBinary if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetDeleteBadBinary() (o string) {\n\tif v != nil && v.DeleteBadBinary != nil {\n\t\treturn *v.DeleteBadBinary\n\t}\n\n\treturn\n}\n\n// IsSetDeleteBadBinary returns true if DeleteBadBinary is not nil.\nfunc (v *UpdateDomainRequest) IsSetDeleteBadBinary() bool {\n\treturn v != nil && v.DeleteBadBinary != nil\n}\n\n// GetFailoverTimeoutInSeconds returns the value of FailoverTimeoutInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainRequest) GetFailoverTimeoutInSeconds() (o int32) {\n\tif v != nil && v.FailoverTimeoutInSeconds != nil {\n\t\treturn *v.FailoverTimeoutInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetFailoverTimeoutInSeconds returns true if FailoverTimeoutInSeconds is not nil.\nfunc (v *UpdateDomainRequest) IsSetFailoverTimeoutInSeconds() bool {\n\treturn v != nil && v.FailoverTimeoutInSeconds != nil\n}\n\ntype UpdateDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          *int64                          `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           *bool                           `json:\"isGlobalDomain,omitempty\"`\n}\n\n// ToWire translates a UpdateDomainResponse struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpdateDomainResponse) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainInfo != nil {\n\t\tw, err = v.DomainInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tw, err = v.Configuration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tw, err = v.ReplicationConfiguration.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsGlobalDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpdateDomainResponse struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpdateDomainResponse struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpdateDomainResponse\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpdateDomainResponse) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.DomainInfo, err = _DomainInfo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Configuration, err = _DomainConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsGlobalDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpdateDomainResponse struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpdateDomainResponse struct could not be encoded.\nfunc (v *UpdateDomainResponse) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.DomainInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Configuration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Configuration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ReplicationConfiguration.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsGlobalDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsGlobalDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpdateDomainResponse struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpdateDomainResponse struct could not be generated from the wire\n// representation.\nfunc (v *UpdateDomainResponse) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.DomainInfo, err = _DomainInfo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Configuration, err = _DomainConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ReplicationConfiguration, err = _DomainReplicationConfiguration_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsGlobalDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpdateDomainResponse\n// struct.\nfunc (v *UpdateDomainResponse) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.DomainInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainInfo: %v\", v.DomainInfo)\n\t\ti++\n\t}\n\tif v.Configuration != nil {\n\t\tfields[i] = fmt.Sprintf(\"Configuration: %v\", v.Configuration)\n\t\ti++\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationConfiguration: %v\", v.ReplicationConfiguration)\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsGlobalDomain: %v\", *(v.IsGlobalDomain))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpdateDomainResponse{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpdateDomainResponse match the\n// provided UpdateDomainResponse.\n//\n// This function performs a deep comparison.\nfunc (v *UpdateDomainResponse) Equals(rhs *UpdateDomainResponse) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainInfo == nil && rhs.DomainInfo == nil) || (v.DomainInfo != nil && rhs.DomainInfo != nil && v.DomainInfo.Equals(rhs.DomainInfo))) {\n\t\treturn false\n\t}\n\tif !((v.Configuration == nil && rhs.Configuration == nil) || (v.Configuration != nil && rhs.Configuration != nil && v.Configuration.Equals(rhs.Configuration))) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationConfiguration == nil && rhs.ReplicationConfiguration == nil) || (v.ReplicationConfiguration != nil && rhs.ReplicationConfiguration != nil && v.ReplicationConfiguration.Equals(rhs.ReplicationConfiguration))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsGlobalDomain, rhs.IsGlobalDomain) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpdateDomainResponse.\nfunc (v *UpdateDomainResponse) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"domainInfo\", v.DomainInfo))\n\t}\n\tif v.Configuration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"configuration\", v.Configuration))\n\t}\n\tif v.ReplicationConfiguration != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationConfiguration\", v.ReplicationConfiguration))\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.IsGlobalDomain != nil {\n\t\tenc.AddBool(\"isGlobalDomain\", *v.IsGlobalDomain)\n\t}\n\treturn err\n}\n\n// GetDomainInfo returns the value of DomainInfo if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\n\treturn\n}\n\n// IsSetDomainInfo returns true if DomainInfo is not nil.\nfunc (v *UpdateDomainResponse) IsSetDomainInfo() bool {\n\treturn v != nil && v.DomainInfo != nil\n}\n\n// GetConfiguration returns the value of Configuration if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainResponse) GetConfiguration() (o *DomainConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\n\treturn\n}\n\n// IsSetConfiguration returns true if Configuration is not nil.\nfunc (v *UpdateDomainResponse) IsSetConfiguration() bool {\n\treturn v != nil && v.Configuration != nil\n}\n\n// GetReplicationConfiguration returns the value of ReplicationConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainResponse) GetReplicationConfiguration() (o *DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfiguration != nil {\n\t\treturn v.ReplicationConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetReplicationConfiguration returns true if ReplicationConfiguration is not nil.\nfunc (v *UpdateDomainResponse) IsSetReplicationConfiguration() bool {\n\treturn v != nil && v.ReplicationConfiguration != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *UpdateDomainResponse) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetIsGlobalDomain returns the value of IsGlobalDomain if it is set or its\n// zero value if it is unset.\nfunc (v *UpdateDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil && v.IsGlobalDomain != nil {\n\t\treturn *v.IsGlobalDomain\n\t}\n\n\treturn\n}\n\n// IsSetIsGlobalDomain returns true if IsGlobalDomain is not nil.\nfunc (v *UpdateDomainResponse) IsSetIsGlobalDomain() bool {\n\treturn v != nil && v.IsGlobalDomain != nil\n}\n\ntype UpsertWorkflowSearchAttributesDecisionAttributes struct {\n\tSearchAttributes *SearchAttributes `json:\"searchAttributes,omitempty\"`\n}\n\n// ToWire translates a UpsertWorkflowSearchAttributesDecisionAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpsertWorkflowSearchAttributesDecisionAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpsertWorkflowSearchAttributesDecisionAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpsertWorkflowSearchAttributesDecisionAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpsertWorkflowSearchAttributesDecisionAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpsertWorkflowSearchAttributesDecisionAttributes struct could not be encoded.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpsertWorkflowSearchAttributesDecisionAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpsertWorkflowSearchAttributesDecisionAttributes struct could not be generated from the wire\n// representation.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpsertWorkflowSearchAttributesDecisionAttributes\n// struct.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpsertWorkflowSearchAttributesDecisionAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpsertWorkflowSearchAttributesDecisionAttributes match the\n// provided UpsertWorkflowSearchAttributesDecisionAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) Equals(rhs *UpsertWorkflowSearchAttributesDecisionAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpsertWorkflowSearchAttributesDecisionAttributes.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\treturn err\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\ntype UpsertWorkflowSearchAttributesEventAttributes struct {\n\tDecisionTaskCompletedEventId *int64            `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tSearchAttributes             *SearchAttributes `json:\"searchAttributes,omitempty\"`\n}\n\n// ToWire translates a UpsertWorkflowSearchAttributesEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a UpsertWorkflowSearchAttributesEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a UpsertWorkflowSearchAttributesEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v UpsertWorkflowSearchAttributesEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a UpsertWorkflowSearchAttributesEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a UpsertWorkflowSearchAttributesEventAttributes struct could not be encoded.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a UpsertWorkflowSearchAttributesEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a UpsertWorkflowSearchAttributesEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a UpsertWorkflowSearchAttributesEventAttributes\n// struct.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"UpsertWorkflowSearchAttributesEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this UpsertWorkflowSearchAttributesEventAttributes match the\n// provided UpsertWorkflowSearchAttributesEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) Equals(rhs *UpsertWorkflowSearchAttributesEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of UpsertWorkflowSearchAttributesEventAttributes.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\treturn err\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\ntype VersionHistories struct {\n\tCurrentVersionHistoryIndex *int32            `json:\"currentVersionHistoryIndex,omitempty\"`\n\tHistories                  []*VersionHistory `json:\"histories,omitempty\"`\n}\n\ntype _List_VersionHistory_ValueList []*VersionHistory\n\nfunc (v _List_VersionHistory_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VersionHistory', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_VersionHistory_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_VersionHistory_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_VersionHistory_ValueList) Close() {}\n\n// ToWire translates a VersionHistories struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *VersionHistories) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CurrentVersionHistoryIndex != nil {\n\t\tw, err = wire.NewValueI32(*(v.CurrentVersionHistoryIndex)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Histories != nil {\n\t\tw, err = wire.NewValueList(_List_VersionHistory_ValueList(v.Histories)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistory_Read(w wire.Value) (*VersionHistory, error) {\n\tvar v VersionHistory\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_VersionHistory_Read(l wire.ValueList) ([]*VersionHistory, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*VersionHistory, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _VersionHistory_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a VersionHistories struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a VersionHistories struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v VersionHistories\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *VersionHistories) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.CurrentVersionHistoryIndex = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Histories, err = _List_VersionHistory_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_VersionHistory_Encode(val []*VersionHistory, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VersionHistory', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a VersionHistories struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a VersionHistories struct could not be encoded.\nfunc (v *VersionHistories) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CurrentVersionHistoryIndex != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.CurrentVersionHistoryIndex)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Histories != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_VersionHistory_Encode(v.Histories, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistory_Decode(sr stream.Reader) (*VersionHistory, error) {\n\tvar v VersionHistory\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_VersionHistory_Decode(sr stream.Reader) ([]*VersionHistory, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*VersionHistory, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _VersionHistory_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a VersionHistories struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a VersionHistories struct could not be generated from the wire\n// representation.\nfunc (v *VersionHistories) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.CurrentVersionHistoryIndex = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Histories, err = _List_VersionHistory_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a VersionHistories\n// struct.\nfunc (v *VersionHistories) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.CurrentVersionHistoryIndex != nil {\n\t\tfields[i] = fmt.Sprintf(\"CurrentVersionHistoryIndex: %v\", *(v.CurrentVersionHistoryIndex))\n\t\ti++\n\t}\n\tif v.Histories != nil {\n\t\tfields[i] = fmt.Sprintf(\"Histories: %v\", v.Histories)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"VersionHistories{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_VersionHistory_Equals(lhs, rhs []*VersionHistory) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this VersionHistories match the\n// provided VersionHistories.\n//\n// This function performs a deep comparison.\nfunc (v *VersionHistories) Equals(rhs *VersionHistories) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.CurrentVersionHistoryIndex, rhs.CurrentVersionHistoryIndex) {\n\t\treturn false\n\t}\n\tif !((v.Histories == nil && rhs.Histories == nil) || (v.Histories != nil && rhs.Histories != nil && _List_VersionHistory_Equals(v.Histories, rhs.Histories))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_VersionHistory_Zapper []*VersionHistory\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_VersionHistory_Zapper.\nfunc (l _List_VersionHistory_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VersionHistories.\nfunc (v *VersionHistories) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CurrentVersionHistoryIndex != nil {\n\t\tenc.AddInt32(\"currentVersionHistoryIndex\", *v.CurrentVersionHistoryIndex)\n\t}\n\tif v.Histories != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"histories\", (_List_VersionHistory_Zapper)(v.Histories)))\n\t}\n\treturn err\n}\n\n// GetCurrentVersionHistoryIndex returns the value of CurrentVersionHistoryIndex if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistories) GetCurrentVersionHistoryIndex() (o int32) {\n\tif v != nil && v.CurrentVersionHistoryIndex != nil {\n\t\treturn *v.CurrentVersionHistoryIndex\n\t}\n\n\treturn\n}\n\n// IsSetCurrentVersionHistoryIndex returns true if CurrentVersionHistoryIndex is not nil.\nfunc (v *VersionHistories) IsSetCurrentVersionHistoryIndex() bool {\n\treturn v != nil && v.CurrentVersionHistoryIndex != nil\n}\n\n// GetHistories returns the value of Histories if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistories) GetHistories() (o []*VersionHistory) {\n\tif v != nil && v.Histories != nil {\n\t\treturn v.Histories\n\t}\n\n\treturn\n}\n\n// IsSetHistories returns true if Histories is not nil.\nfunc (v *VersionHistories) IsSetHistories() bool {\n\treturn v != nil && v.Histories != nil\n}\n\ntype VersionHistory struct {\n\tBranchToken []byte                `json:\"branchToken,omitempty\"`\n\tItems       []*VersionHistoryItem `json:\"items,omitempty\"`\n}\n\ntype _List_VersionHistoryItem_ValueList []*VersionHistoryItem\n\nfunc (v _List_VersionHistoryItem_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_VersionHistoryItem_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_VersionHistoryItem_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_VersionHistoryItem_ValueList) Close() {}\n\n// ToWire translates a VersionHistory struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *VersionHistory) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.BranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.BranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Items != nil {\n\t\tw, err = wire.NewValueList(_List_VersionHistoryItem_ValueList(v.Items)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VersionHistoryItem_Read(w wire.Value) (*VersionHistoryItem, error) {\n\tvar v VersionHistoryItem\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_VersionHistoryItem_Read(l wire.ValueList) ([]*VersionHistoryItem, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*VersionHistoryItem, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _VersionHistoryItem_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a VersionHistory struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a VersionHistory struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v VersionHistory\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *VersionHistory) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Items, err = _List_VersionHistoryItem_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_VersionHistoryItem_Encode(val []*VersionHistoryItem, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VersionHistoryItem', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a VersionHistory struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a VersionHistory struct could not be encoded.\nfunc (v *VersionHistory) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.BranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Items != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_VersionHistoryItem_Encode(v.Items, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VersionHistoryItem_Decode(sr stream.Reader) (*VersionHistoryItem, error) {\n\tvar v VersionHistoryItem\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_VersionHistoryItem_Decode(sr stream.Reader) ([]*VersionHistoryItem, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*VersionHistoryItem, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _VersionHistoryItem_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a VersionHistory struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a VersionHistory struct could not be generated from the wire\n// representation.\nfunc (v *VersionHistory) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.BranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TList:\n\t\t\tv.Items, err = _List_VersionHistoryItem_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a VersionHistory\n// struct.\nfunc (v *VersionHistory) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.BranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchToken: %v\", v.BranchToken)\n\t\ti++\n\t}\n\tif v.Items != nil {\n\t\tfields[i] = fmt.Sprintf(\"Items: %v\", v.Items)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"VersionHistory{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_VersionHistoryItem_Equals(lhs, rhs []*VersionHistoryItem) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this VersionHistory match the\n// provided VersionHistory.\n//\n// This function performs a deep comparison.\nfunc (v *VersionHistory) Equals(rhs *VersionHistory) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.BranchToken == nil && rhs.BranchToken == nil) || (v.BranchToken != nil && rhs.BranchToken != nil && bytes.Equal(v.BranchToken, rhs.BranchToken))) {\n\t\treturn false\n\t}\n\tif !((v.Items == nil && rhs.Items == nil) || (v.Items != nil && rhs.Items != nil && _List_VersionHistoryItem_Equals(v.Items, rhs.Items))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_VersionHistoryItem_Zapper []*VersionHistoryItem\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_VersionHistoryItem_Zapper.\nfunc (l _List_VersionHistoryItem_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VersionHistory.\nfunc (v *VersionHistory) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.BranchToken != nil {\n\t\tenc.AddString(\"branchToken\", base64.StdEncoding.EncodeToString(v.BranchToken))\n\t}\n\tif v.Items != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"items\", (_List_VersionHistoryItem_Zapper)(v.Items)))\n\t}\n\treturn err\n}\n\n// GetBranchToken returns the value of BranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistory) GetBranchToken() (o []byte) {\n\tif v != nil && v.BranchToken != nil {\n\t\treturn v.BranchToken\n\t}\n\n\treturn\n}\n\n// IsSetBranchToken returns true if BranchToken is not nil.\nfunc (v *VersionHistory) IsSetBranchToken() bool {\n\treturn v != nil && v.BranchToken != nil\n}\n\n// GetItems returns the value of Items if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistory) GetItems() (o []*VersionHistoryItem) {\n\tif v != nil && v.Items != nil {\n\t\treturn v.Items\n\t}\n\n\treturn\n}\n\n// IsSetItems returns true if Items is not nil.\nfunc (v *VersionHistory) IsSetItems() bool {\n\treturn v != nil && v.Items != nil\n}\n\ntype VersionHistoryItem struct {\n\tEventID *int64 `json:\"eventID,omitempty\"`\n\tVersion *int64 `json:\"version,omitempty\"`\n}\n\n// ToWire translates a VersionHistoryItem struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *VersionHistoryItem) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.EventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.EventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a VersionHistoryItem struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a VersionHistoryItem struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v VersionHistoryItem\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *VersionHistoryItem) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a VersionHistoryItem struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a VersionHistoryItem struct could not be encoded.\nfunc (v *VersionHistoryItem) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.EventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a VersionHistoryItem struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a VersionHistoryItem struct could not be generated from the wire\n// representation.\nfunc (v *VersionHistoryItem) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a VersionHistoryItem\n// struct.\nfunc (v *VersionHistoryItem) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.EventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventID: %v\", *(v.EventID))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"VersionHistoryItem{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this VersionHistoryItem match the\n// provided VersionHistoryItem.\n//\n// This function performs a deep comparison.\nfunc (v *VersionHistoryItem) Equals(rhs *VersionHistoryItem) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EventID, rhs.EventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VersionHistoryItem.\nfunc (v *VersionHistoryItem) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.EventID != nil {\n\t\tenc.AddInt64(\"eventID\", *v.EventID)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\treturn err\n}\n\n// GetEventID returns the value of EventID if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistoryItem) GetEventID() (o int64) {\n\tif v != nil && v.EventID != nil {\n\t\treturn *v.EventID\n\t}\n\n\treturn\n}\n\n// IsSetEventID returns true if EventID is not nil.\nfunc (v *VersionHistoryItem) IsSetEventID() bool {\n\treturn v != nil && v.EventID != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *VersionHistoryItem) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *VersionHistoryItem) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\ntype VirtualQueueState struct {\n\tVirtualSliceStates []*VirtualSliceState `json:\"virtualSliceStates,omitempty\"`\n}\n\ntype _List_VirtualSliceState_ValueList []*VirtualSliceState\n\nfunc (v _List_VirtualSliceState_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VirtualSliceState', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_VirtualSliceState_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_VirtualSliceState_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_VirtualSliceState_ValueList) Close() {}\n\n// ToWire translates a VirtualQueueState struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *VirtualQueueState) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.VirtualSliceStates != nil {\n\t\tw, err = wire.NewValueList(_List_VirtualSliceState_ValueList(v.VirtualSliceStates)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _VirtualSliceState_Read(w wire.Value) (*VirtualSliceState, error) {\n\tvar v VirtualSliceState\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_VirtualSliceState_Read(l wire.ValueList) ([]*VirtualSliceState, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*VirtualSliceState, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _VirtualSliceState_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a VirtualQueueState struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a VirtualQueueState struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v VirtualQueueState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *VirtualQueueState) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.VirtualSliceStates, err = _List_VirtualSliceState_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_VirtualSliceState_Encode(val []*VirtualSliceState, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*VirtualSliceState', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a VirtualQueueState struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a VirtualQueueState struct could not be encoded.\nfunc (v *VirtualQueueState) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.VirtualSliceStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_VirtualSliceState_Encode(v.VirtualSliceStates, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _VirtualSliceState_Decode(sr stream.Reader) (*VirtualSliceState, error) {\n\tvar v VirtualSliceState\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_VirtualSliceState_Decode(sr stream.Reader) ([]*VirtualSliceState, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*VirtualSliceState, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _VirtualSliceState_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a VirtualQueueState struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a VirtualQueueState struct could not be generated from the wire\n// representation.\nfunc (v *VirtualQueueState) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.VirtualSliceStates, err = _List_VirtualSliceState_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a VirtualQueueState\n// struct.\nfunc (v *VirtualQueueState) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.VirtualSliceStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"VirtualSliceStates: %v\", v.VirtualSliceStates)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"VirtualQueueState{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_VirtualSliceState_Equals(lhs, rhs []*VirtualSliceState) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this VirtualQueueState match the\n// provided VirtualQueueState.\n//\n// This function performs a deep comparison.\nfunc (v *VirtualQueueState) Equals(rhs *VirtualQueueState) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.VirtualSliceStates == nil && rhs.VirtualSliceStates == nil) || (v.VirtualSliceStates != nil && rhs.VirtualSliceStates != nil && _List_VirtualSliceState_Equals(v.VirtualSliceStates, rhs.VirtualSliceStates))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_VirtualSliceState_Zapper []*VirtualSliceState\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_VirtualSliceState_Zapper.\nfunc (l _List_VirtualSliceState_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VirtualQueueState.\nfunc (v *VirtualQueueState) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.VirtualSliceStates != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"virtualSliceStates\", (_List_VirtualSliceState_Zapper)(v.VirtualSliceStates)))\n\t}\n\treturn err\n}\n\n// GetVirtualSliceStates returns the value of VirtualSliceStates if it is set or its\n// zero value if it is unset.\nfunc (v *VirtualQueueState) GetVirtualSliceStates() (o []*VirtualSliceState) {\n\tif v != nil && v.VirtualSliceStates != nil {\n\t\treturn v.VirtualSliceStates\n\t}\n\n\treturn\n}\n\n// IsSetVirtualSliceStates returns true if VirtualSliceStates is not nil.\nfunc (v *VirtualQueueState) IsSetVirtualSliceStates() bool {\n\treturn v != nil && v.VirtualSliceStates != nil\n}\n\ntype VirtualSliceState struct {\n\tTaskRange *TaskRange `json:\"taskRange,omitempty\"`\n\tPredicate *Predicate `json:\"predicate,omitempty\"`\n}\n\n// ToWire translates a VirtualSliceState struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *VirtualSliceState) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskRange != nil {\n\t\tw, err = v.TaskRange.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Predicate != nil {\n\t\tw, err = v.Predicate.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskRange_Read(w wire.Value) (*TaskRange, error) {\n\tvar v TaskRange\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Predicate_Read(w wire.Value) (*Predicate, error) {\n\tvar v Predicate\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a VirtualSliceState struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a VirtualSliceState struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v VirtualSliceState\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *VirtualSliceState) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskRange, err = _TaskRange_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Predicate, err = _Predicate_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a VirtualSliceState struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a VirtualSliceState struct could not be encoded.\nfunc (v *VirtualSliceState) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskRange != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskRange.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Predicate != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Predicate.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskRange_Decode(sr stream.Reader) (*TaskRange, error) {\n\tvar v TaskRange\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Predicate_Decode(sr stream.Reader) (*Predicate, error) {\n\tvar v Predicate\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a VirtualSliceState struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a VirtualSliceState struct could not be generated from the wire\n// representation.\nfunc (v *VirtualSliceState) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.TaskRange, err = _TaskRange_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Predicate, err = _Predicate_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a VirtualSliceState\n// struct.\nfunc (v *VirtualSliceState) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.TaskRange != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskRange: %v\", v.TaskRange)\n\t\ti++\n\t}\n\tif v.Predicate != nil {\n\t\tfields[i] = fmt.Sprintf(\"Predicate: %v\", v.Predicate)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"VirtualSliceState{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this VirtualSliceState match the\n// provided VirtualSliceState.\n//\n// This function performs a deep comparison.\nfunc (v *VirtualSliceState) Equals(rhs *VirtualSliceState) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskRange == nil && rhs.TaskRange == nil) || (v.TaskRange != nil && rhs.TaskRange != nil && v.TaskRange.Equals(rhs.TaskRange))) {\n\t\treturn false\n\t}\n\tif !((v.Predicate == nil && rhs.Predicate == nil) || (v.Predicate != nil && rhs.Predicate != nil && v.Predicate.Equals(rhs.Predicate))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of VirtualSliceState.\nfunc (v *VirtualSliceState) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskRange != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskRange\", v.TaskRange))\n\t}\n\tif v.Predicate != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"predicate\", v.Predicate))\n\t}\n\treturn err\n}\n\n// GetTaskRange returns the value of TaskRange if it is set or its\n// zero value if it is unset.\nfunc (v *VirtualSliceState) GetTaskRange() (o *TaskRange) {\n\tif v != nil && v.TaskRange != nil {\n\t\treturn v.TaskRange\n\t}\n\n\treturn\n}\n\n// IsSetTaskRange returns true if TaskRange is not nil.\nfunc (v *VirtualSliceState) IsSetTaskRange() bool {\n\treturn v != nil && v.TaskRange != nil\n}\n\n// GetPredicate returns the value of Predicate if it is set or its\n// zero value if it is unset.\nfunc (v *VirtualSliceState) GetPredicate() (o *Predicate) {\n\tif v != nil && v.Predicate != nil {\n\t\treturn v.Predicate\n\t}\n\n\treturn\n}\n\n// IsSetPredicate returns true if Predicate is not nil.\nfunc (v *VirtualSliceState) IsSetPredicate() bool {\n\treturn v != nil && v.Predicate != nil\n}\n\ntype WorkerVersionInfo struct {\n\tImpl           *string `json:\"impl,omitempty\"`\n\tFeatureVersion *string `json:\"featureVersion,omitempty\"`\n}\n\n// ToWire translates a WorkerVersionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkerVersionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Impl != nil {\n\t\tw, err = wire.NewValueString(*(v.Impl)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.FeatureVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.FeatureVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkerVersionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkerVersionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkerVersionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkerVersionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Impl = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FeatureVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkerVersionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkerVersionInfo struct could not be encoded.\nfunc (v *WorkerVersionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Impl != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Impl)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FeatureVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FeatureVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkerVersionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkerVersionInfo struct could not be generated from the wire\n// representation.\nfunc (v *WorkerVersionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Impl = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FeatureVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkerVersionInfo\n// struct.\nfunc (v *WorkerVersionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Impl != nil {\n\t\tfields[i] = fmt.Sprintf(\"Impl: %v\", *(v.Impl))\n\t\ti++\n\t}\n\tif v.FeatureVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FeatureVersion: %v\", *(v.FeatureVersion))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkerVersionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkerVersionInfo match the\n// provided WorkerVersionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *WorkerVersionInfo) Equals(rhs *WorkerVersionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Impl, rhs.Impl) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FeatureVersion, rhs.FeatureVersion) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkerVersionInfo.\nfunc (v *WorkerVersionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Impl != nil {\n\t\tenc.AddString(\"impl\", *v.Impl)\n\t}\n\tif v.FeatureVersion != nil {\n\t\tenc.AddString(\"featureVersion\", *v.FeatureVersion)\n\t}\n\treturn err\n}\n\n// GetImpl returns the value of Impl if it is set or its\n// zero value if it is unset.\nfunc (v *WorkerVersionInfo) GetImpl() (o string) {\n\tif v != nil && v.Impl != nil {\n\t\treturn *v.Impl\n\t}\n\n\treturn\n}\n\n// IsSetImpl returns true if Impl is not nil.\nfunc (v *WorkerVersionInfo) IsSetImpl() bool {\n\treturn v != nil && v.Impl != nil\n}\n\n// GetFeatureVersion returns the value of FeatureVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkerVersionInfo) GetFeatureVersion() (o string) {\n\tif v != nil && v.FeatureVersion != nil {\n\t\treturn *v.FeatureVersion\n\t}\n\n\treturn\n}\n\n// IsSetFeatureVersion returns true if FeatureVersion is not nil.\nfunc (v *WorkerVersionInfo) IsSetFeatureVersion() bool {\n\treturn v != nil && v.FeatureVersion != nil\n}\n\ntype WorkflowExecution struct {\n\tWorkflowId *string `json:\"workflowId,omitempty\"`\n\tRunId      *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecution struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecution) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecution struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecution struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecution\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecution) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecution struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecution struct could not be encoded.\nfunc (v *WorkflowExecution) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecution struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecution struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecution) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecution\n// struct.\nfunc (v *WorkflowExecution) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecution{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecution match the\n// provided WorkflowExecution.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecution) Equals(rhs *WorkflowExecution) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecution.\nfunc (v *WorkflowExecution) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecution) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *WorkflowExecution) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecution) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *WorkflowExecution) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\ntype WorkflowExecutionAlreadyCompletedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ToWire translates a WorkflowExecutionAlreadyCompletedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionAlreadyCompletedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tw, err = wire.NewValueString(v.Message), error(nil)\n\tif err != nil {\n\t\treturn w, err\n\t}\n\tfields[i] = wire.Field{ID: 1, Value: w}\n\ti++\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionAlreadyCompletedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionAlreadyCompletedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionAlreadyCompletedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionAlreadyCompletedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tmessageIsSet := false\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 1:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Message, err = field.Value.GetString(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tmessageIsSet = true\n\t\t\t}\n\t\t}\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of WorkflowExecutionAlreadyCompletedError is required\")\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionAlreadyCompletedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionAlreadyCompletedError struct could not be encoded.\nfunc (v *WorkflowExecutionAlreadyCompletedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 1, Type: wire.TBinary}); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteString(v.Message); err != nil {\n\t\treturn err\n\t}\n\tif err := sw.WriteFieldEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionAlreadyCompletedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionAlreadyCompletedError struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionAlreadyCompletedError) Decode(sr stream.Reader) error {\n\n\tmessageIsSet := false\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 1 && fh.Type == wire.TBinary:\n\t\t\tv.Message, err = sr.ReadString()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmessageIsSet = true\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\tif !messageIsSet {\n\t\treturn errors.New(\"field Message of WorkflowExecutionAlreadyCompletedError is required\")\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionAlreadyCompletedError\n// struct.\nfunc (v *WorkflowExecutionAlreadyCompletedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tfields[i] = fmt.Sprintf(\"Message: %v\", v.Message)\n\ti++\n\n\treturn fmt.Sprintf(\"WorkflowExecutionAlreadyCompletedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*WorkflowExecutionAlreadyCompletedError) ErrorName() string {\n\treturn \"WorkflowExecutionAlreadyCompletedError\"\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionAlreadyCompletedError match the\n// provided WorkflowExecutionAlreadyCompletedError.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionAlreadyCompletedError) Equals(rhs *WorkflowExecutionAlreadyCompletedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !(v.Message == rhs.Message) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionAlreadyCompletedError.\nfunc (v *WorkflowExecutionAlreadyCompletedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tenc.AddString(\"message\", v.Message)\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionAlreadyCompletedError) GetMessage() (o string) {\n\tif v != nil {\n\t\to = v.Message\n\t}\n\treturn\n}\n\nfunc (v *WorkflowExecutionAlreadyCompletedError) Error() string {\n\treturn v.String()\n}\n\ntype WorkflowExecutionAlreadyStartedError struct {\n\tMessage        *string `json:\"message,omitempty\"`\n\tStartRequestId *string `json:\"startRequestId,omitempty\"`\n\tRunId          *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionAlreadyStartedError struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionAlreadyStartedError) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Message != nil {\n\t\tw, err = wire.NewValueString(*(v.Message)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartRequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.StartRequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionAlreadyStartedError struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionAlreadyStartedError struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionAlreadyStartedError\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionAlreadyStartedError) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Message = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartRequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionAlreadyStartedError struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionAlreadyStartedError struct could not be encoded.\nfunc (v *WorkflowExecutionAlreadyStartedError) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Message != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Message)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartRequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartRequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionAlreadyStartedError struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionAlreadyStartedError struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionAlreadyStartedError) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Message = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartRequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionAlreadyStartedError\n// struct.\nfunc (v *WorkflowExecutionAlreadyStartedError) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Message != nil {\n\t\tfields[i] = fmt.Sprintf(\"Message: %v\", *(v.Message))\n\t\ti++\n\t}\n\tif v.StartRequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartRequestId: %v\", *(v.StartRequestId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionAlreadyStartedError{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// ErrorName is the name of this type as defined in the Thrift\n// file.\nfunc (*WorkflowExecutionAlreadyStartedError) ErrorName() string {\n\treturn \"WorkflowExecutionAlreadyStartedError\"\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionAlreadyStartedError match the\n// provided WorkflowExecutionAlreadyStartedError.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionAlreadyStartedError) Equals(rhs *WorkflowExecutionAlreadyStartedError) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Message, rhs.Message) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartRequestId, rhs.StartRequestId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionAlreadyStartedError.\nfunc (v *WorkflowExecutionAlreadyStartedError) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Message != nil {\n\t\tenc.AddString(\"message\", *v.Message)\n\t}\n\tif v.StartRequestId != nil {\n\t\tenc.AddString(\"startRequestId\", *v.StartRequestId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetMessage returns the value of Message if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionAlreadyStartedError) GetMessage() (o string) {\n\tif v != nil && v.Message != nil {\n\t\treturn *v.Message\n\t}\n\n\treturn\n}\n\n// IsSetMessage returns true if Message is not nil.\nfunc (v *WorkflowExecutionAlreadyStartedError) IsSetMessage() bool {\n\treturn v != nil && v.Message != nil\n}\n\n// GetStartRequestId returns the value of StartRequestId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionAlreadyStartedError) GetStartRequestId() (o string) {\n\tif v != nil && v.StartRequestId != nil {\n\t\treturn *v.StartRequestId\n\t}\n\n\treturn\n}\n\n// IsSetStartRequestId returns true if StartRequestId is not nil.\nfunc (v *WorkflowExecutionAlreadyStartedError) IsSetStartRequestId() bool {\n\treturn v != nil && v.StartRequestId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionAlreadyStartedError) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *WorkflowExecutionAlreadyStartedError) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\nfunc (v *WorkflowExecutionAlreadyStartedError) Error() string {\n\treturn v.String()\n}\n\ntype WorkflowExecutionCancelRequestedEventAttributes struct {\n\tCause                     *string            `json:\"cause,omitempty\"`\n\tExternalInitiatedEventId  *int64             `json:\"externalInitiatedEventId,omitempty\"`\n\tExternalWorkflowExecution *WorkflowExecution `json:\"externalWorkflowExecution,omitempty\"`\n\tIdentity                  *string            `json:\"identity,omitempty\"`\n\tRequestId                 *string            `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionCancelRequestedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Cause != nil {\n\t\tw, err = wire.NewValueString(*(v.Cause)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExternalInitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tw, err = v.ExternalWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionCancelRequestedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionCancelRequestedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionCancelRequestedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Cause = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExternalInitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionCancelRequestedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionCancelRequestedEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Cause != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Cause)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalInitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExternalInitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExternalWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExternalWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionCancelRequestedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionCancelRequestedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Cause = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExternalInitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.ExternalWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionCancelRequestedEventAttributes\n// struct.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Cause != nil {\n\t\tfields[i] = fmt.Sprintf(\"Cause: %v\", *(v.Cause))\n\t\ti++\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalInitiatedEventId: %v\", *(v.ExternalInitiatedEventId))\n\t\ti++\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExternalWorkflowExecution: %v\", v.ExternalWorkflowExecution)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionCancelRequestedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionCancelRequestedEventAttributes match the\n// provided WorkflowExecutionCancelRequestedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) Equals(rhs *WorkflowExecutionCancelRequestedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Cause, rhs.Cause) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExternalInitiatedEventId, rhs.ExternalInitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.ExternalWorkflowExecution == nil && rhs.ExternalWorkflowExecution == nil) || (v.ExternalWorkflowExecution != nil && rhs.ExternalWorkflowExecution != nil && v.ExternalWorkflowExecution.Equals(rhs.ExternalWorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionCancelRequestedEventAttributes.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Cause != nil {\n\t\tenc.AddString(\"cause\", *v.Cause)\n\t}\n\tif v.ExternalInitiatedEventId != nil {\n\t\tenc.AddInt64(\"externalInitiatedEventId\", *v.ExternalInitiatedEventId)\n\t}\n\tif v.ExternalWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"externalWorkflowExecution\", v.ExternalWorkflowExecution))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetCause returns the value of Cause if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) GetCause() (o string) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\n\treturn\n}\n\n// IsSetCause returns true if Cause is not nil.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) IsSetCause() bool {\n\treturn v != nil && v.Cause != nil\n}\n\n// GetExternalInitiatedEventId returns the value of ExternalInitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) GetExternalInitiatedEventId() (o int64) {\n\tif v != nil && v.ExternalInitiatedEventId != nil {\n\t\treturn *v.ExternalInitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetExternalInitiatedEventId returns true if ExternalInitiatedEventId is not nil.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) IsSetExternalInitiatedEventId() bool {\n\treturn v != nil && v.ExternalInitiatedEventId != nil\n}\n\n// GetExternalWorkflowExecution returns the value of ExternalWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) GetExternalWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetExternalWorkflowExecution returns true if ExternalWorkflowExecution is not nil.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) IsSetExternalWorkflowExecution() bool {\n\treturn v != nil && v.ExternalWorkflowExecution != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype WorkflowExecutionCanceledEventAttributes struct {\n\tDecisionTaskCompletedEventId *int64 `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDetails                      []byte `json:\"details,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionCanceledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionCanceledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionCanceledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionCanceledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionCanceledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionCanceledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionCanceledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionCanceledEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionCanceledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionCanceledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionCanceledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionCanceledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionCanceledEventAttributes\n// struct.\nfunc (v *WorkflowExecutionCanceledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionCanceledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionCanceledEventAttributes match the\n// provided WorkflowExecutionCanceledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionCanceledEventAttributes) Equals(rhs *WorkflowExecutionCanceledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionCanceledEventAttributes.\nfunc (v *WorkflowExecutionCanceledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\treturn err\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCanceledEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *WorkflowExecutionCanceledEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCanceledEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *WorkflowExecutionCanceledEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\ntype WorkflowExecutionCloseStatus int32\n\nconst (\n\tWorkflowExecutionCloseStatusCompleted      WorkflowExecutionCloseStatus = 0\n\tWorkflowExecutionCloseStatusFailed         WorkflowExecutionCloseStatus = 1\n\tWorkflowExecutionCloseStatusCanceled       WorkflowExecutionCloseStatus = 2\n\tWorkflowExecutionCloseStatusTerminated     WorkflowExecutionCloseStatus = 3\n\tWorkflowExecutionCloseStatusContinuedAsNew WorkflowExecutionCloseStatus = 4\n\tWorkflowExecutionCloseStatusTimedOut       WorkflowExecutionCloseStatus = 5\n)\n\n// WorkflowExecutionCloseStatus_Values returns all recognized values of WorkflowExecutionCloseStatus.\nfunc WorkflowExecutionCloseStatus_Values() []WorkflowExecutionCloseStatus {\n\treturn []WorkflowExecutionCloseStatus{\n\t\tWorkflowExecutionCloseStatusCompleted,\n\t\tWorkflowExecutionCloseStatusFailed,\n\t\tWorkflowExecutionCloseStatusCanceled,\n\t\tWorkflowExecutionCloseStatusTerminated,\n\t\tWorkflowExecutionCloseStatusContinuedAsNew,\n\t\tWorkflowExecutionCloseStatusTimedOut,\n\t}\n}\n\n// UnmarshalText tries to decode WorkflowExecutionCloseStatus from a byte slice\n// containing its name.\n//\n//\tvar v WorkflowExecutionCloseStatus\n//\terr := v.UnmarshalText([]byte(\"COMPLETED\"))\nfunc (v *WorkflowExecutionCloseStatus) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"COMPLETED\":\n\t\t*v = WorkflowExecutionCloseStatusCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*v = WorkflowExecutionCloseStatusFailed\n\t\treturn nil\n\tcase \"CANCELED\":\n\t\t*v = WorkflowExecutionCloseStatusCanceled\n\t\treturn nil\n\tcase \"TERMINATED\":\n\t\t*v = WorkflowExecutionCloseStatusTerminated\n\t\treturn nil\n\tcase \"CONTINUED_AS_NEW\":\n\t\t*v = WorkflowExecutionCloseStatusContinuedAsNew\n\t\treturn nil\n\tcase \"TIMED_OUT\":\n\t\t*v = WorkflowExecutionCloseStatusTimedOut\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowExecutionCloseStatus\", err)\n\t\t}\n\t\t*v = WorkflowExecutionCloseStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowExecutionCloseStatus to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v WorkflowExecutionCloseStatus) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"COMPLETED\"), nil\n\tcase 1:\n\t\treturn []byte(\"FAILED\"), nil\n\tcase 2:\n\t\treturn []byte(\"CANCELED\"), nil\n\tcase 3:\n\t\treturn []byte(\"TERMINATED\"), nil\n\tcase 4:\n\t\treturn []byte(\"CONTINUED_AS_NEW\"), nil\n\tcase 5:\n\t\treturn []byte(\"TIMED_OUT\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionCloseStatus.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v WorkflowExecutionCloseStatus) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"COMPLETED\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"FAILED\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"CANCELED\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"TERMINATED\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"CONTINUED_AS_NEW\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"TIMED_OUT\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v WorkflowExecutionCloseStatus) Ptr() *WorkflowExecutionCloseStatus {\n\treturn &v\n}\n\n// Encode encodes WorkflowExecutionCloseStatus directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v WorkflowExecutionCloseStatus\n//\treturn v.Encode(sWriter)\nfunc (v WorkflowExecutionCloseStatus) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates WorkflowExecutionCloseStatus into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v WorkflowExecutionCloseStatus) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes WorkflowExecutionCloseStatus from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return WorkflowExecutionCloseStatus(0), err\n//\t}\n//\n//\tvar v WorkflowExecutionCloseStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return WorkflowExecutionCloseStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowExecutionCloseStatus) FromWire(w wire.Value) error {\n\t*v = (WorkflowExecutionCloseStatus)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded WorkflowExecutionCloseStatus directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v WorkflowExecutionCloseStatus\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return WorkflowExecutionCloseStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowExecutionCloseStatus) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (WorkflowExecutionCloseStatus)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of WorkflowExecutionCloseStatus.\nfunc (v WorkflowExecutionCloseStatus) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"COMPLETED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\tcase 2:\n\t\treturn \"CANCELED\"\n\tcase 3:\n\t\treturn \"TERMINATED\"\n\tcase 4:\n\t\treturn \"CONTINUED_AS_NEW\"\n\tcase 5:\n\t\treturn \"TIMED_OUT\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowExecutionCloseStatus(%d)\", w)\n}\n\n// Equals returns true if this WorkflowExecutionCloseStatus value matches the provided\n// value.\nfunc (v WorkflowExecutionCloseStatus) Equals(rhs WorkflowExecutionCloseStatus) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes WorkflowExecutionCloseStatus into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v WorkflowExecutionCloseStatus) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"COMPLETED\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"FAILED\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"CANCELED\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"TERMINATED\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"CONTINUED_AS_NEW\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"TIMED_OUT\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode WorkflowExecutionCloseStatus from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *WorkflowExecutionCloseStatus) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"WorkflowExecutionCloseStatus\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"WorkflowExecutionCloseStatus\")\n\t\t}\n\t\t*v = (WorkflowExecutionCloseStatus)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"WorkflowExecutionCloseStatus\")\n\t}\n}\n\ntype WorkflowExecutionCompletedEventAttributes struct {\n\tResult                       []byte `json:\"result,omitempty\"`\n\tDecisionTaskCompletedEventId *int64 `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionCompletedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionCompletedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Result != nil {\n\t\tw, err = wire.NewValueBinary(v.Result), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionCompletedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionCompletedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionCompletedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionCompletedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Result, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionCompletedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionCompletedEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionCompletedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Result != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionCompletedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionCompletedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionCompletedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.Result, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionCompletedEventAttributes\n// struct.\nfunc (v *WorkflowExecutionCompletedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.Result != nil {\n\t\tfields[i] = fmt.Sprintf(\"Result: %v\", v.Result)\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionCompletedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionCompletedEventAttributes match the\n// provided WorkflowExecutionCompletedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionCompletedEventAttributes) Equals(rhs *WorkflowExecutionCompletedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Result == nil && rhs.Result == nil) || (v.Result != nil && rhs.Result != nil && bytes.Equal(v.Result, rhs.Result))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionCompletedEventAttributes.\nfunc (v *WorkflowExecutionCompletedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Result != nil {\n\t\tenc.AddString(\"result\", base64.StdEncoding.EncodeToString(v.Result))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetResult returns the value of Result if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCompletedEventAttributes) GetResult() (o []byte) {\n\tif v != nil && v.Result != nil {\n\t\treturn v.Result\n\t}\n\n\treturn\n}\n\n// IsSetResult returns true if Result is not nil.\nfunc (v *WorkflowExecutionCompletedEventAttributes) IsSetResult() bool {\n\treturn v != nil && v.Result != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionCompletedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *WorkflowExecutionCompletedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype WorkflowExecutionConfiguration struct {\n\tTaskList                            *TaskList `json:\"taskList,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32    `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32    `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionConfiguration struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionConfiguration) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionConfiguration struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionConfiguration struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionConfiguration\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionConfiguration) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionConfiguration struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionConfiguration struct could not be encoded.\nfunc (v *WorkflowExecutionConfiguration) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionConfiguration struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionConfiguration struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionConfiguration) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionConfiguration\n// struct.\nfunc (v *WorkflowExecutionConfiguration) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionConfiguration{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionConfiguration match the\n// provided WorkflowExecutionConfiguration.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionConfiguration) Equals(rhs *WorkflowExecutionConfiguration) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionConfiguration.\nfunc (v *WorkflowExecutionConfiguration) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\treturn err\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionConfiguration) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowExecutionConfiguration) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionConfiguration) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionConfiguration) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionConfiguration) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionConfiguration) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\ntype WorkflowExecutionContinuedAsNewEventAttributes struct {\n\tNewExecutionRunId                   *string                       `json:\"newExecutionRunId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventId        *int64                        `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tBackoffStartIntervalInSeconds       *int32                        `json:\"backoffStartIntervalInSeconds,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tFailureReason                       *string                       `json:\"failureReason,omitempty\"`\n\tFailureDetails                      []byte                        `json:\"failureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionContinuedAsNewEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [17]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.NewExecutionRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.NewExecutionRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.BackoffStartIntervalInSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tw, err = v.Initiator.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.FailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.FailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.FailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.FailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tw, err = wire.NewValueBinary(v.LastCompletionResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionContinuedAsNewEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionContinuedAsNewEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionContinuedAsNewEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.NewExecutionRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.BackoffStartIntervalInSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ContinueAsNewInitiator\n\t\t\t\tx, err = _ContinueAsNewInitiator_Read(field.Value)\n\t\t\t\tv.Initiator = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.FailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastCompletionResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionContinuedAsNewEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionContinuedAsNewEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.NewExecutionRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.NewExecutionRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.BackoffStartIntervalInSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Initiator != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Initiator.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.FailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastCompletionResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastCompletionResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionContinuedAsNewEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionContinuedAsNewEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.NewExecutionRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.BackoffStartIntervalInSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI32:\n\t\t\tvar x ContinueAsNewInitiator\n\t\t\tx, err = _ContinueAsNewInitiator_Decode(sr)\n\t\t\tv.Initiator = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tv.FailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tv.LastCompletionResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionContinuedAsNewEventAttributes\n// struct.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [17]string\n\ti := 0\n\tif v.NewExecutionRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewExecutionRunId: %v\", *(v.NewExecutionRunId))\n\t\ti++\n\t}\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"BackoffStartIntervalInSeconds: %v\", *(v.BackoffStartIntervalInSeconds))\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tfields[i] = fmt.Sprintf(\"Initiator: %v\", *(v.Initiator))\n\t\ti++\n\t}\n\tif v.FailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailureReason: %v\", *(v.FailureReason))\n\t\ti++\n\t}\n\tif v.FailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailureDetails: %v\", v.FailureDetails)\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastCompletionResult: %v\", v.LastCompletionResult)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionContinuedAsNewEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionContinuedAsNewEventAttributes match the\n// provided WorkflowExecutionContinuedAsNewEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) Equals(rhs *WorkflowExecutionContinuedAsNewEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.NewExecutionRunId, rhs.NewExecutionRunId) {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.BackoffStartIntervalInSeconds, rhs.BackoffStartIntervalInSeconds) {\n\t\treturn false\n\t}\n\tif !_ContinueAsNewInitiator_EqualsPtr(v.Initiator, rhs.Initiator) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FailureReason, rhs.FailureReason) {\n\t\treturn false\n\t}\n\tif !((v.FailureDetails == nil && rhs.FailureDetails == nil) || (v.FailureDetails != nil && rhs.FailureDetails != nil && bytes.Equal(v.FailureDetails, rhs.FailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.LastCompletionResult == nil && rhs.LastCompletionResult == nil) || (v.LastCompletionResult != nil && rhs.LastCompletionResult != nil && bytes.Equal(v.LastCompletionResult, rhs.LastCompletionResult))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionContinuedAsNewEventAttributes.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.NewExecutionRunId != nil {\n\t\tenc.AddString(\"newExecutionRunId\", *v.NewExecutionRunId)\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\tif v.BackoffStartIntervalInSeconds != nil {\n\t\tenc.AddInt32(\"backoffStartIntervalInSeconds\", *v.BackoffStartIntervalInSeconds)\n\t}\n\tif v.Initiator != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"initiator\", *v.Initiator))\n\t}\n\tif v.FailureReason != nil {\n\t\tenc.AddString(\"failureReason\", *v.FailureReason)\n\t}\n\tif v.FailureDetails != nil {\n\t\tenc.AddString(\"failureDetails\", base64.StdEncoding.EncodeToString(v.FailureDetails))\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tenc.AddString(\"lastCompletionResult\", base64.StdEncoding.EncodeToString(v.LastCompletionResult))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetNewExecutionRunId returns the value of NewExecutionRunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetNewExecutionRunId() (o string) {\n\tif v != nil && v.NewExecutionRunId != nil {\n\t\treturn *v.NewExecutionRunId\n\t}\n\n\treturn\n}\n\n// IsSetNewExecutionRunId returns true if NewExecutionRunId is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetNewExecutionRunId() bool {\n\treturn v != nil && v.NewExecutionRunId != nil\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\n// GetBackoffStartIntervalInSeconds returns the value of BackoffStartIntervalInSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetBackoffStartIntervalInSeconds() (o int32) {\n\tif v != nil && v.BackoffStartIntervalInSeconds != nil {\n\t\treturn *v.BackoffStartIntervalInSeconds\n\t}\n\n\treturn\n}\n\n// IsSetBackoffStartIntervalInSeconds returns true if BackoffStartIntervalInSeconds is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetBackoffStartIntervalInSeconds() bool {\n\treturn v != nil && v.BackoffStartIntervalInSeconds != nil\n}\n\n// GetInitiator returns the value of Initiator if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\n\treturn\n}\n\n// IsSetInitiator returns true if Initiator is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetInitiator() bool {\n\treturn v != nil && v.Initiator != nil\n}\n\n// GetFailureReason returns the value of FailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetFailureReason() (o string) {\n\tif v != nil && v.FailureReason != nil {\n\t\treturn *v.FailureReason\n\t}\n\n\treturn\n}\n\n// IsSetFailureReason returns true if FailureReason is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetFailureReason() bool {\n\treturn v != nil && v.FailureReason != nil\n}\n\n// GetFailureDetails returns the value of FailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetFailureDetails() (o []byte) {\n\tif v != nil && v.FailureDetails != nil {\n\t\treturn v.FailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetFailureDetails returns true if FailureDetails is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetFailureDetails() bool {\n\treturn v != nil && v.FailureDetails != nil\n}\n\n// GetLastCompletionResult returns the value of LastCompletionResult if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetLastCompletionResult() (o []byte) {\n\tif v != nil && v.LastCompletionResult != nil {\n\t\treturn v.LastCompletionResult\n\t}\n\n\treturn\n}\n\n// IsSetLastCompletionResult returns true if LastCompletionResult is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetLastCompletionResult() bool {\n\treturn v != nil && v.LastCompletionResult != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype WorkflowExecutionFailedEventAttributes struct {\n\tReason                       *string `json:\"reason,omitempty\"`\n\tDetails                      []byte  `json:\"details,omitempty\"`\n\tDecisionTaskCompletedEventId *int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionFailedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionFailedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionTaskCompletedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionFailedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionFailedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionFailedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionFailedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionFailedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionFailedEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionFailedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionTaskCompletedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionFailedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionFailedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionFailedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionTaskCompletedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionFailedEventAttributes\n// struct.\nfunc (v *WorkflowExecutionFailedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskCompletedEventId: %v\", *(v.DecisionTaskCompletedEventId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionFailedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionFailedEventAttributes match the\n// provided WorkflowExecutionFailedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionFailedEventAttributes) Equals(rhs *WorkflowExecutionFailedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionTaskCompletedEventId, rhs.DecisionTaskCompletedEventId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionFailedEventAttributes.\nfunc (v *WorkflowExecutionFailedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.DecisionTaskCompletedEventId != nil {\n\t\tenc.AddInt64(\"decisionTaskCompletedEventId\", *v.DecisionTaskCompletedEventId)\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionFailedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *WorkflowExecutionFailedEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionFailedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *WorkflowExecutionFailedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetDecisionTaskCompletedEventId returns the value of DecisionTaskCompletedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionFailedEventAttributes) GetDecisionTaskCompletedEventId() (o int64) {\n\tif v != nil && v.DecisionTaskCompletedEventId != nil {\n\t\treturn *v.DecisionTaskCompletedEventId\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskCompletedEventId returns true if DecisionTaskCompletedEventId is not nil.\nfunc (v *WorkflowExecutionFailedEventAttributes) IsSetDecisionTaskCompletedEventId() bool {\n\treturn v != nil && v.DecisionTaskCompletedEventId != nil\n}\n\ntype WorkflowExecutionFilter struct {\n\tWorkflowId *string `json:\"workflowId,omitempty\"`\n\tRunId      *string `json:\"runId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionFilter struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionFilter) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowId != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tw, err = wire.NewValueString(*(v.RunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionFilter struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionFilter struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionFilter\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionFilter) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionFilter struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionFilter struct could not be encoded.\nfunc (v *WorkflowExecutionFilter) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionFilter struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionFilter struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionFilter) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionFilter\n// struct.\nfunc (v *WorkflowExecutionFilter) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.WorkflowId != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowId: %v\", *(v.WorkflowId))\n\t\ti++\n\t}\n\tif v.RunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunId: %v\", *(v.RunId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionFilter{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionFilter match the\n// provided WorkflowExecutionFilter.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionFilter) Equals(rhs *WorkflowExecutionFilter) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowId, rhs.WorkflowId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RunId, rhs.RunId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionFilter.\nfunc (v *WorkflowExecutionFilter) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowId != nil {\n\t\tenc.AddString(\"workflowId\", *v.WorkflowId)\n\t}\n\tif v.RunId != nil {\n\t\tenc.AddString(\"runId\", *v.RunId)\n\t}\n\treturn err\n}\n\n// GetWorkflowId returns the value of WorkflowId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionFilter) GetWorkflowId() (o string) {\n\tif v != nil && v.WorkflowId != nil {\n\t\treturn *v.WorkflowId\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowId returns true if WorkflowId is not nil.\nfunc (v *WorkflowExecutionFilter) IsSetWorkflowId() bool {\n\treturn v != nil && v.WorkflowId != nil\n}\n\n// GetRunId returns the value of RunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionFilter) GetRunId() (o string) {\n\tif v != nil && v.RunId != nil {\n\t\treturn *v.RunId\n\t}\n\n\treturn\n}\n\n// IsSetRunId returns true if RunId is not nil.\nfunc (v *WorkflowExecutionFilter) IsSetRunId() bool {\n\treturn v != nil && v.RunId != nil\n}\n\ntype WorkflowExecutionInfo struct {\n\tExecution                    *WorkflowExecution            `json:\"execution,omitempty\"`\n\tType                         *WorkflowType                 `json:\"type,omitempty\"`\n\tStartTime                    *int64                        `json:\"startTime,omitempty\"`\n\tCloseTime                    *int64                        `json:\"closeTime,omitempty\"`\n\tCloseStatus                  *WorkflowExecutionCloseStatus `json:\"closeStatus,omitempty\"`\n\tHistoryLength                *int64                        `json:\"historyLength,omitempty\"`\n\tParentDomainId               *string                       `json:\"parentDomainId,omitempty\"`\n\tParentDomainName             *string                       `json:\"parentDomainName,omitempty\"`\n\tParentInitatedId             *int64                        `json:\"parentInitatedId,omitempty\"`\n\tParentExecution              *WorkflowExecution            `json:\"parentExecution,omitempty\"`\n\tExecutionTime                *int64                        `json:\"executionTime,omitempty\"`\n\tMemo                         *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes             *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tAutoResetPoints              *ResetPoints                  `json:\"autoResetPoints,omitempty\"`\n\tTaskList                     *string                       `json:\"taskList,omitempty\"`\n\tTaskListInfo                 *TaskList                     `json:\"taskListInfo,omitempty\"`\n\tIsCron                       *bool                         `json:\"isCron,omitempty\"`\n\tUpdateTime                   *int64                        `json:\"updateTime,omitempty\"`\n\tPartitionConfig              map[string]string             `json:\"partitionConfig,omitempty\"`\n\tCronOverlapPolicy            *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n\tCronSchedule                 *string                       `json:\"cronSchedule,omitempty\"`\n\tExecutionStatus              *WorkflowExecutionStatus      `json:\"executionStatus,omitempty\"`\n\tScheduledExecutionTime       *int64                        `json:\"scheduledExecutionTime,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [24]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Execution != nil {\n\t\tw, err = v.Execution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.CloseTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.CloseTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tw, err = v.CloseStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryLength != nil {\n\t\tw, err = wire.NewValueI64(*(v.HistoryLength)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.ParentDomainId != nil {\n\t\tw, err = wire.NewValueString(*(v.ParentDomainId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.ParentDomainName != nil {\n\t\tw, err = wire.NewValueString(*(v.ParentDomainName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 71, Value: w}\n\t\ti++\n\t}\n\tif v.ParentInitatedId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ParentInitatedId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 72, Value: w}\n\t\ti++\n\t}\n\tif v.ParentExecution != nil {\n\t\tw, err = v.ParentExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExecutionTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 101, Value: w}\n\t\ti++\n\t}\n\tif v.AutoResetPoints != nil {\n\t\tw, err = v.AutoResetPoints.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListInfo != nil {\n\t\tw, err = v.TaskListInfo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 121, Value: w}\n\t\ti++\n\t}\n\tif v.IsCron != nil {\n\t\tw, err = wire.NewValueBool(*(v.IsCron)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.UpdateTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.UpdateTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStatus != nil {\n\t\tw, err = v.ExecutionStatus.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 190, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledExecutionTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledExecutionTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 200, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _ResetPoints_Read(w wire.Value) (*ResetPoints, error) {\n\tvar v ResetPoints\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionStatus_Read(w wire.Value) (WorkflowExecutionStatus, error) {\n\tvar v WorkflowExecutionStatus\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a WorkflowExecutionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Execution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Type, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CloseTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\t\tx, err = _WorkflowExecutionCloseStatus_Read(field.Value)\n\t\t\t\tv.CloseStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.HistoryLength = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ParentDomainId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 71:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ParentDomainName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 72:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ParentInitatedId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ParentExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExecutionTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 101:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AutoResetPoints, err = _ResetPoints_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 121:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskListInfo, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.IsCron = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.UpdateTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 190:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x WorkflowExecutionStatus\n\t\t\t\tx, err = _WorkflowExecutionStatus_Read(field.Value)\n\t\t\t\tv.ExecutionStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 200:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledExecutionTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionInfo struct could not be encoded.\nfunc (v *WorkflowExecutionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Execution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Execution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CloseTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CloseTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CloseStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CloseStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryLength != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.HistoryLength)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentDomainId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ParentDomainId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentDomainName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 71, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ParentDomainName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentInitatedId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 72, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ParentInitatedId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExecutionTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 101, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoResetPoints != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AutoResetPoints.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListInfo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 121, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListInfo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsCron != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.IsCron)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UpdateTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.UpdateTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 190, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ExecutionStatus.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledExecutionTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 200, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledExecutionTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _ResetPoints_Decode(sr stream.Reader) (*ResetPoints, error) {\n\tvar v ResetPoints\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _WorkflowExecutionStatus_Decode(sr stream.Reader) (WorkflowExecutionStatus, error) {\n\tvar v WorkflowExecutionStatus\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a WorkflowExecutionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionInfo struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.Execution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.Type, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CloseTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowExecutionCloseStatus\n\t\t\tx, err = _WorkflowExecutionCloseStatus_Decode(sr)\n\t\t\tv.CloseStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.HistoryLength = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ParentDomainId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 71 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ParentDomainName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 72 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ParentInitatedId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TStruct:\n\t\t\tv.ParentExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExecutionTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 101 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TStruct:\n\t\t\tv.AutoResetPoints, err = _ResetPoints_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 121 && fh.Type == wire.TStruct:\n\t\t\tv.TaskListInfo, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.IsCron = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.UpdateTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 190 && fh.Type == wire.TI32:\n\t\t\tvar x WorkflowExecutionStatus\n\t\t\tx, err = _WorkflowExecutionStatus_Decode(sr)\n\t\t\tv.ExecutionStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 200 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledExecutionTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionInfo\n// struct.\nfunc (v *WorkflowExecutionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [24]string\n\ti := 0\n\tif v.Execution != nil {\n\t\tfields[i] = fmt.Sprintf(\"Execution: %v\", v.Execution)\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", v.Type)\n\t\ti++\n\t}\n\tif v.StartTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartTime: %v\", *(v.StartTime))\n\t\ti++\n\t}\n\tif v.CloseTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"CloseTime: %v\", *(v.CloseTime))\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"CloseStatus: %v\", *(v.CloseStatus))\n\t\ti++\n\t}\n\tif v.HistoryLength != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryLength: %v\", *(v.HistoryLength))\n\t\ti++\n\t}\n\tif v.ParentDomainId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentDomainId: %v\", *(v.ParentDomainId))\n\t\ti++\n\t}\n\tif v.ParentDomainName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentDomainName: %v\", *(v.ParentDomainName))\n\t\ti++\n\t}\n\tif v.ParentInitatedId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentInitatedId: %v\", *(v.ParentInitatedId))\n\t\ti++\n\t}\n\tif v.ParentExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentExecution: %v\", v.ParentExecution)\n\t\ti++\n\t}\n\tif v.ExecutionTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionTime: %v\", *(v.ExecutionTime))\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.AutoResetPoints != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoResetPoints: %v\", v.AutoResetPoints)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\tif v.TaskListInfo != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListInfo: %v\", v.TaskListInfo)\n\t\ti++\n\t}\n\tif v.IsCron != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsCron: %v\", *(v.IsCron))\n\t\ti++\n\t}\n\tif v.UpdateTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpdateTime: %v\", *(v.UpdateTime))\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.ExecutionStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStatus: %v\", *(v.ExecutionStatus))\n\t\ti++\n\t}\n\tif v.ScheduledExecutionTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledExecutionTime: %v\", *(v.ScheduledExecutionTime))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _WorkflowExecutionStatus_EqualsPtr(lhs, rhs *WorkflowExecutionStatus) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionInfo match the\n// provided WorkflowExecutionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionInfo) Equals(rhs *WorkflowExecutionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.Execution == nil && rhs.Execution == nil) || (v.Execution != nil && rhs.Execution != nil && v.Execution.Equals(rhs.Execution))) {\n\t\treturn false\n\t}\n\tif !((v.Type == nil && rhs.Type == nil) || (v.Type != nil && rhs.Type != nil && v.Type.Equals(rhs.Type))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartTime, rhs.StartTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CloseTime, rhs.CloseTime) {\n\t\treturn false\n\t}\n\tif !_WorkflowExecutionCloseStatus_EqualsPtr(v.CloseStatus, rhs.CloseStatus) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.HistoryLength, rhs.HistoryLength) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ParentDomainId, rhs.ParentDomainId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ParentDomainName, rhs.ParentDomainName) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ParentInitatedId, rhs.ParentInitatedId) {\n\t\treturn false\n\t}\n\tif !((v.ParentExecution == nil && rhs.ParentExecution == nil) || (v.ParentExecution != nil && rhs.ParentExecution != nil && v.ParentExecution.Equals(rhs.ParentExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExecutionTime, rhs.ExecutionTime) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.AutoResetPoints == nil && rhs.AutoResetPoints == nil) || (v.AutoResetPoints != nil && rhs.AutoResetPoints != nil && v.AutoResetPoints.Equals(rhs.AutoResetPoints))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\tif !((v.TaskListInfo == nil && rhs.TaskListInfo == nil) || (v.TaskListInfo != nil && rhs.TaskListInfo != nil && v.TaskListInfo.Equals(rhs.TaskListInfo))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.IsCron, rhs.IsCron) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.UpdateTime, rhs.UpdateTime) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !_WorkflowExecutionStatus_EqualsPtr(v.ExecutionStatus, rhs.ExecutionStatus) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledExecutionTime, rhs.ScheduledExecutionTime) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionInfo.\nfunc (v *WorkflowExecutionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Execution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"execution\", v.Execution))\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", v.Type))\n\t}\n\tif v.StartTime != nil {\n\t\tenc.AddInt64(\"startTime\", *v.StartTime)\n\t}\n\tif v.CloseTime != nil {\n\t\tenc.AddInt64(\"closeTime\", *v.CloseTime)\n\t}\n\tif v.CloseStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"closeStatus\", *v.CloseStatus))\n\t}\n\tif v.HistoryLength != nil {\n\t\tenc.AddInt64(\"historyLength\", *v.HistoryLength)\n\t}\n\tif v.ParentDomainId != nil {\n\t\tenc.AddString(\"parentDomainId\", *v.ParentDomainId)\n\t}\n\tif v.ParentDomainName != nil {\n\t\tenc.AddString(\"parentDomainName\", *v.ParentDomainName)\n\t}\n\tif v.ParentInitatedId != nil {\n\t\tenc.AddInt64(\"parentInitatedId\", *v.ParentInitatedId)\n\t}\n\tif v.ParentExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentExecution\", v.ParentExecution))\n\t}\n\tif v.ExecutionTime != nil {\n\t\tenc.AddInt64(\"executionTime\", *v.ExecutionTime)\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.AutoResetPoints != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"autoResetPoints\", v.AutoResetPoints))\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\tif v.TaskListInfo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListInfo\", v.TaskListInfo))\n\t}\n\tif v.IsCron != nil {\n\t\tenc.AddBool(\"isCron\", *v.IsCron)\n\t}\n\tif v.UpdateTime != nil {\n\t\tenc.AddInt64(\"updateTime\", *v.UpdateTime)\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.ExecutionStatus != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"executionStatus\", *v.ExecutionStatus))\n\t}\n\tif v.ScheduledExecutionTime != nil {\n\t\tenc.AddInt64(\"scheduledExecutionTime\", *v.ScheduledExecutionTime)\n\t}\n\treturn err\n}\n\n// GetExecution returns the value of Execution if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\n\treturn\n}\n\n// IsSetExecution returns true if Execution is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetExecution() bool {\n\treturn v != nil && v.Execution != nil\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetType() (o *WorkflowType) {\n\tif v != nil && v.Type != nil {\n\t\treturn v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetStartTime returns the value of StartTime if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetStartTime() (o int64) {\n\tif v != nil && v.StartTime != nil {\n\t\treturn *v.StartTime\n\t}\n\n\treturn\n}\n\n// IsSetStartTime returns true if StartTime is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetStartTime() bool {\n\treturn v != nil && v.StartTime != nil\n}\n\n// GetCloseTime returns the value of CloseTime if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCloseTime() (o int64) {\n\tif v != nil && v.CloseTime != nil {\n\t\treturn *v.CloseTime\n\t}\n\n\treturn\n}\n\n// IsSetCloseTime returns true if CloseTime is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCloseTime() bool {\n\treturn v != nil && v.CloseTime != nil\n}\n\n// GetCloseStatus returns the value of CloseStatus if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCloseStatus() (o WorkflowExecutionCloseStatus) {\n\tif v != nil && v.CloseStatus != nil {\n\t\treturn *v.CloseStatus\n\t}\n\n\treturn\n}\n\n// IsSetCloseStatus returns true if CloseStatus is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCloseStatus() bool {\n\treturn v != nil && v.CloseStatus != nil\n}\n\n// GetHistoryLength returns the value of HistoryLength if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetHistoryLength() (o int64) {\n\tif v != nil && v.HistoryLength != nil {\n\t\treturn *v.HistoryLength\n\t}\n\n\treturn\n}\n\n// IsSetHistoryLength returns true if HistoryLength is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetHistoryLength() bool {\n\treturn v != nil && v.HistoryLength != nil\n}\n\n// GetParentDomainId returns the value of ParentDomainId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentDomainId() (o string) {\n\tif v != nil && v.ParentDomainId != nil {\n\t\treturn *v.ParentDomainId\n\t}\n\n\treturn\n}\n\n// IsSetParentDomainId returns true if ParentDomainId is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentDomainId() bool {\n\treturn v != nil && v.ParentDomainId != nil\n}\n\n// GetParentDomainName returns the value of ParentDomainName if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentDomainName() (o string) {\n\tif v != nil && v.ParentDomainName != nil {\n\t\treturn *v.ParentDomainName\n\t}\n\n\treturn\n}\n\n// IsSetParentDomainName returns true if ParentDomainName is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentDomainName() bool {\n\treturn v != nil && v.ParentDomainName != nil\n}\n\n// GetParentInitatedId returns the value of ParentInitatedId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentInitatedId() (o int64) {\n\tif v != nil && v.ParentInitatedId != nil {\n\t\treturn *v.ParentInitatedId\n\t}\n\n\treturn\n}\n\n// IsSetParentInitatedId returns true if ParentInitatedId is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentInitatedId() bool {\n\treturn v != nil && v.ParentInitatedId != nil\n}\n\n// GetParentExecution returns the value of ParentExecution if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentExecution() (o *WorkflowExecution) {\n\tif v != nil && v.ParentExecution != nil {\n\t\treturn v.ParentExecution\n\t}\n\n\treturn\n}\n\n// IsSetParentExecution returns true if ParentExecution is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentExecution() bool {\n\treturn v != nil && v.ParentExecution != nil\n}\n\n// GetExecutionTime returns the value of ExecutionTime if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetExecutionTime() (o int64) {\n\tif v != nil && v.ExecutionTime != nil {\n\t\treturn *v.ExecutionTime\n\t}\n\n\treturn\n}\n\n// IsSetExecutionTime returns true if ExecutionTime is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetExecutionTime() bool {\n\treturn v != nil && v.ExecutionTime != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetAutoResetPoints returns the value of AutoResetPoints if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetAutoResetPoints() (o *ResetPoints) {\n\tif v != nil && v.AutoResetPoints != nil {\n\t\treturn v.AutoResetPoints\n\t}\n\n\treturn\n}\n\n// IsSetAutoResetPoints returns true if AutoResetPoints is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetAutoResetPoints() bool {\n\treturn v != nil && v.AutoResetPoints != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTaskListInfo returns the value of TaskListInfo if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetTaskListInfo() (o *TaskList) {\n\tif v != nil && v.TaskListInfo != nil {\n\t\treturn v.TaskListInfo\n\t}\n\n\treturn\n}\n\n// IsSetTaskListInfo returns true if TaskListInfo is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetTaskListInfo() bool {\n\treturn v != nil && v.TaskListInfo != nil\n}\n\n// GetIsCron returns the value of IsCron if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetIsCron() (o bool) {\n\tif v != nil && v.IsCron != nil {\n\t\treturn *v.IsCron\n\t}\n\n\treturn\n}\n\n// IsSetIsCron returns true if IsCron is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetIsCron() bool {\n\treturn v != nil && v.IsCron != nil\n}\n\n// GetUpdateTime returns the value of UpdateTime if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetUpdateTime() (o int64) {\n\tif v != nil && v.UpdateTime != nil {\n\t\treturn *v.UpdateTime\n\t}\n\n\treturn\n}\n\n// IsSetUpdateTime returns true if UpdateTime is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetUpdateTime() bool {\n\treturn v != nil && v.UpdateTime != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetExecutionStatus returns the value of ExecutionStatus if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetExecutionStatus() (o WorkflowExecutionStatus) {\n\tif v != nil && v.ExecutionStatus != nil {\n\t\treturn *v.ExecutionStatus\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStatus returns true if ExecutionStatus is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetExecutionStatus() bool {\n\treturn v != nil && v.ExecutionStatus != nil\n}\n\n// GetScheduledExecutionTime returns the value of ScheduledExecutionTime if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetScheduledExecutionTime() (o int64) {\n\tif v != nil && v.ScheduledExecutionTime != nil {\n\t\treturn *v.ScheduledExecutionTime\n\t}\n\n\treturn\n}\n\n// IsSetScheduledExecutionTime returns true if ScheduledExecutionTime is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetScheduledExecutionTime() bool {\n\treturn v != nil && v.ScheduledExecutionTime != nil\n}\n\ntype WorkflowExecutionSignaledEventAttributes struct {\n\tSignalName *string `json:\"signalName,omitempty\"`\n\tInput      []byte  `json:\"input,omitempty\"`\n\tIdentity   *string `json:\"identity,omitempty\"`\n\tRequestId  *string `json:\"requestId,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionSignaledEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionSignaledEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.SignalName != nil {\n\t\tw, err = wire.NewValueString(*(v.SignalName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionSignaledEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionSignaledEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionSignaledEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionSignaledEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.SignalName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionSignaledEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionSignaledEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionSignaledEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.SignalName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.SignalName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionSignaledEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionSignaledEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionSignaledEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.SignalName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionSignaledEventAttributes\n// struct.\nfunc (v *WorkflowExecutionSignaledEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.SignalName != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalName: %v\", *(v.SignalName))\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionSignaledEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionSignaledEventAttributes match the\n// provided WorkflowExecutionSignaledEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionSignaledEventAttributes) Equals(rhs *WorkflowExecutionSignaledEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.SignalName, rhs.SignalName) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionSignaledEventAttributes.\nfunc (v *WorkflowExecutionSignaledEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.SignalName != nil {\n\t\tenc.AddString(\"signalName\", *v.SignalName)\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\treturn err\n}\n\n// GetSignalName returns the value of SignalName if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetSignalName() (o string) {\n\tif v != nil && v.SignalName != nil {\n\t\treturn *v.SignalName\n\t}\n\n\treturn\n}\n\n// IsSetSignalName returns true if SignalName is not nil.\nfunc (v *WorkflowExecutionSignaledEventAttributes) IsSetSignalName() bool {\n\treturn v != nil && v.SignalName != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *WorkflowExecutionSignaledEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *WorkflowExecutionSignaledEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *WorkflowExecutionSignaledEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\ntype WorkflowExecutionStartedEventAttributes struct {\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tParentWorkflowDomain                *string                       `json:\"parentWorkflowDomain,omitempty\"`\n\tParentWorkflowExecution             *WorkflowExecution            `json:\"parentWorkflowExecution,omitempty\"`\n\tParentInitiatedEventId              *int64                        `json:\"parentInitiatedEventId,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tContinuedExecutionRunId             *string                       `json:\"continuedExecutionRunId,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tContinuedFailureReason              *string                       `json:\"continuedFailureReason,omitempty\"`\n\tContinuedFailureDetails             []byte                        `json:\"continuedFailureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tOriginalExecutionRunId              *string                       `json:\"originalExecutionRunId,omitempty\"`\n\tIdentity                            *string                       `json:\"identity,omitempty\"`\n\tFirstExecutionRunId                 *string                       `json:\"firstExecutionRunId,omitempty\"`\n\tFirstScheduledTimeNano              *int64                        `json:\"firstScheduledTimeNano,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tAttempt                             *int32                        `json:\"attempt,omitempty\"`\n\tExpirationTimestamp                 *int64                        `json:\"expirationTimestamp,omitempty\"`\n\tCronSchedule                        *string                       `json:\"cronSchedule,omitempty\"`\n\tFirstDecisionTaskBackoffSeconds     *int32                        `json:\"firstDecisionTaskBackoffSeconds,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tPrevAutoResetPoints                 *ResetPoints                  `json:\"prevAutoResetPoints,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tPartitionConfig                     map[string]string             `json:\"partitionConfig,omitempty\"`\n\tRequestId                           *string                       `json:\"requestId,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionStartedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionStartedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [30]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowType != nil {\n\t\tw, err = v.WorkflowType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ParentWorkflowDomain != nil {\n\t\tw, err = wire.NewValueString(*(v.ParentWorkflowDomain)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ParentWorkflowExecution != nil {\n\t\tw, err = v.ParentWorkflowExecution.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.ParentInitiatedEventId != nil {\n\t\tw, err = wire.NewValueI64(*(v.ParentInitiatedEventId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = v.TaskList.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ExecutionStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.TaskStartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.ContinuedExecutionRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.ContinuedExecutionRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 54, Value: w}\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tw, err = v.Initiator.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.ContinuedFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.ContinuedFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 57, Value: w}\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tw, err = wire.NewValueBinary(v.LastCompletionResult), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 58, Value: w}\n\t\ti++\n\t}\n\tif v.OriginalExecutionRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.OriginalExecutionRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 59, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.FirstExecutionRunId != nil {\n\t\tw, err = wire.NewValueString(*(v.FirstExecutionRunId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 61, Value: w}\n\t\ti++\n\t}\n\tif v.FirstScheduledTimeNano != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstScheduledTimeNano)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 62, Value: w}\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tw, err = v.RetryPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpirationTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.FirstDecisionTaskBackoffSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = v.Memo.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = v.SearchAttributes.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 121, Value: w}\n\t\ti++\n\t}\n\tif v.PrevAutoResetPoints != nil {\n\t\tw, err = v.PrevAutoResetPoints.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 140, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 150, Value: w}\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestId)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 160, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 170, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = v.ActiveClusterSelectionPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 180, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionStartedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionStartedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionStartedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionStartedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.WorkflowType, err = _WorkflowType_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ParentWorkflowDomain = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ParentWorkflowExecution, err = _WorkflowExecution_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ParentInitiatedEventId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.TaskList, err = _TaskList_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 54:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ContinuedExecutionRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x ContinueAsNewInitiator\n\t\t\t\tx, err = _ContinueAsNewInitiator_Read(field.Value)\n\t\t\t\tv.Initiator = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ContinuedFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 57:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ContinuedFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 58:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.LastCompletionResult, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 59:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OriginalExecutionRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 61:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.FirstExecutionRunId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 62:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstScheduledTimeNano = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.RetryPolicy, err = _RetryPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpirationTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.FirstDecisionTaskBackoffSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Memo, err = _Memo_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 121:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.SearchAttributes, err = _SearchAttributes_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.PrevAutoResetPoints, err = _ResetPoints_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 140:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 150:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 160:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestId = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 170:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 180:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionStartedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionStartedEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionStartedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.WorkflowType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentWorkflowDomain != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ParentWorkflowDomain)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentWorkflowExecution != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ParentWorkflowExecution.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentInitiatedEventId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ParentInitiatedEventId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskList.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ExecutionStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TaskStartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinuedExecutionRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 54, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ContinuedExecutionRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Initiator != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Initiator.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinuedFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ContinuedFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ContinuedFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 57, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ContinuedFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastCompletionResult != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 58, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.LastCompletionResult); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OriginalExecutionRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 59, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OriginalExecutionRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstExecutionRunId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 61, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.FirstExecutionRunId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstScheduledTimeNano != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 62, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstScheduledTimeNano)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.RetryPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpirationTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpirationTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.FirstDecisionTaskBackoffSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Memo.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 121, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.SearchAttributes.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PrevAutoResetPoints != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.PrevAutoResetPoints.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 140, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 150, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestId != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 160, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestId)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 170, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 180, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ActiveClusterSelectionPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionStartedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionStartedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionStartedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TStruct:\n\t\t\tv.WorkflowType, err = _WorkflowType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ParentWorkflowDomain = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TStruct:\n\t\t\tv.ParentWorkflowExecution, err = _WorkflowExecution_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ParentInitiatedEventId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TStruct:\n\t\t\tv.TaskList, err = _TaskList_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ExecutionStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TaskStartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 54 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ContinuedExecutionRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TI32:\n\t\t\tvar x ContinueAsNewInitiator\n\t\t\tx, err = _ContinueAsNewInitiator_Decode(sr)\n\t\t\tv.Initiator = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ContinuedFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 57 && fh.Type == wire.TBinary:\n\t\t\tv.ContinuedFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 58 && fh.Type == wire.TBinary:\n\t\t\tv.LastCompletionResult, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 59 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OriginalExecutionRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 61 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.FirstExecutionRunId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 62 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstScheduledTimeNano = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TStruct:\n\t\t\tv.RetryPolicy, err = _RetryPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpirationTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.FirstDecisionTaskBackoffSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TStruct:\n\t\t\tv.Memo, err = _Memo_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 121 && fh.Type == wire.TStruct:\n\t\t\tv.SearchAttributes, err = _SearchAttributes_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TStruct:\n\t\t\tv.PrevAutoResetPoints, err = _ResetPoints_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 140 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 150 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 160 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestId = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 170 && fh.Type == wire.TI32:\n\t\t\tvar x CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 180 && fh.Type == wire.TStruct:\n\t\t\tv.ActiveClusterSelectionPolicy, err = _ActiveClusterSelectionPolicy_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionStartedEventAttributes\n// struct.\nfunc (v *WorkflowExecutionStartedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [30]string\n\ti := 0\n\tif v.WorkflowType != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowType: %v\", v.WorkflowType)\n\t\ti++\n\t}\n\tif v.ParentWorkflowDomain != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentWorkflowDomain: %v\", *(v.ParentWorkflowDomain))\n\t\ti++\n\t}\n\tif v.ParentWorkflowExecution != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentWorkflowExecution: %v\", v.ParentWorkflowExecution)\n\t\ti++\n\t}\n\tif v.ParentInitiatedEventId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentInitiatedEventId: %v\", *(v.ParentInitiatedEventId))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", v.TaskList)\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionStartToCloseTimeoutSeconds: %v\", *(v.ExecutionStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskStartToCloseTimeoutSeconds: %v\", *(v.TaskStartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ContinuedExecutionRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinuedExecutionRunId: %v\", *(v.ContinuedExecutionRunId))\n\t\ti++\n\t}\n\tif v.Initiator != nil {\n\t\tfields[i] = fmt.Sprintf(\"Initiator: %v\", *(v.Initiator))\n\t\ti++\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinuedFailureReason: %v\", *(v.ContinuedFailureReason))\n\t\ti++\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"ContinuedFailureDetails: %v\", v.ContinuedFailureDetails)\n\t\ti++\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastCompletionResult: %v\", v.LastCompletionResult)\n\t\ti++\n\t}\n\tif v.OriginalExecutionRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"OriginalExecutionRunId: %v\", *(v.OriginalExecutionRunId))\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\tif v.FirstExecutionRunId != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstExecutionRunId: %v\", *(v.FirstExecutionRunId))\n\t\ti++\n\t}\n\tif v.FirstScheduledTimeNano != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstScheduledTimeNano: %v\", *(v.FirstScheduledTimeNano))\n\t\ti++\n\t}\n\tif v.RetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryPolicy: %v\", v.RetryPolicy)\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpirationTimestamp: %v\", *(v.ExpirationTimestamp))\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstDecisionTaskBackoffSeconds: %v\", *(v.FirstDecisionTaskBackoffSeconds))\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.PrevAutoResetPoints != nil {\n\t\tfields[i] = fmt.Sprintf(\"PrevAutoResetPoints: %v\", v.PrevAutoResetPoints)\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\tif v.RequestId != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestId: %v\", *(v.RequestId))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionStartedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionStartedEventAttributes match the\n// provided WorkflowExecutionStartedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionStartedEventAttributes) Equals(rhs *WorkflowExecutionStartedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.WorkflowType == nil && rhs.WorkflowType == nil) || (v.WorkflowType != nil && rhs.WorkflowType != nil && v.WorkflowType.Equals(rhs.WorkflowType))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ParentWorkflowDomain, rhs.ParentWorkflowDomain) {\n\t\treturn false\n\t}\n\tif !((v.ParentWorkflowExecution == nil && rhs.ParentWorkflowExecution == nil) || (v.ParentWorkflowExecution != nil && rhs.ParentWorkflowExecution != nil && v.ParentWorkflowExecution.Equals(rhs.ParentWorkflowExecution))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ParentInitiatedEventId, rhs.ParentInitiatedEventId) {\n\t\treturn false\n\t}\n\tif !((v.TaskList == nil && rhs.TaskList == nil) || (v.TaskList != nil && rhs.TaskList != nil && v.TaskList.Equals(rhs.TaskList))) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ExecutionStartToCloseTimeoutSeconds, rhs.ExecutionStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TaskStartToCloseTimeoutSeconds, rhs.TaskStartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ContinuedExecutionRunId, rhs.ContinuedExecutionRunId) {\n\t\treturn false\n\t}\n\tif !_ContinueAsNewInitiator_EqualsPtr(v.Initiator, rhs.Initiator) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ContinuedFailureReason, rhs.ContinuedFailureReason) {\n\t\treturn false\n\t}\n\tif !((v.ContinuedFailureDetails == nil && rhs.ContinuedFailureDetails == nil) || (v.ContinuedFailureDetails != nil && rhs.ContinuedFailureDetails != nil && bytes.Equal(v.ContinuedFailureDetails, rhs.ContinuedFailureDetails))) {\n\t\treturn false\n\t}\n\tif !((v.LastCompletionResult == nil && rhs.LastCompletionResult == nil) || (v.LastCompletionResult != nil && rhs.LastCompletionResult != nil && bytes.Equal(v.LastCompletionResult, rhs.LastCompletionResult))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OriginalExecutionRunId, rhs.OriginalExecutionRunId) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.FirstExecutionRunId, rhs.FirstExecutionRunId) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstScheduledTimeNano, rhs.FirstScheduledTimeNano) {\n\t\treturn false\n\t}\n\tif !((v.RetryPolicy == nil && rhs.RetryPolicy == nil) || (v.RetryPolicy != nil && rhs.RetryPolicy != nil && v.RetryPolicy.Equals(rhs.RetryPolicy))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpirationTimestamp, rhs.ExpirationTimestamp) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.FirstDecisionTaskBackoffSeconds, rhs.FirstDecisionTaskBackoffSeconds) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && v.Memo.Equals(rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && v.SearchAttributes.Equals(rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.PrevAutoResetPoints == nil && rhs.PrevAutoResetPoints == nil) || (v.PrevAutoResetPoints != nil && rhs.PrevAutoResetPoints != nil && v.PrevAutoResetPoints.Equals(rhs.PrevAutoResetPoints))) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestId, rhs.RequestId) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && v.ActiveClusterSelectionPolicy.Equals(rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionStartedEventAttributes.\nfunc (v *WorkflowExecutionStartedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"workflowType\", v.WorkflowType))\n\t}\n\tif v.ParentWorkflowDomain != nil {\n\t\tenc.AddString(\"parentWorkflowDomain\", *v.ParentWorkflowDomain)\n\t}\n\tif v.ParentWorkflowExecution != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"parentWorkflowExecution\", v.ParentWorkflowExecution))\n\t}\n\tif v.ParentInitiatedEventId != nil {\n\t\tenc.AddInt64(\"parentInitiatedEventId\", *v.ParentInitiatedEventId)\n\t}\n\tif v.TaskList != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskList\", v.TaskList))\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"executionStartToCloseTimeoutSeconds\", *v.ExecutionStartToCloseTimeoutSeconds)\n\t}\n\tif v.TaskStartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"taskStartToCloseTimeoutSeconds\", *v.TaskStartToCloseTimeoutSeconds)\n\t}\n\tif v.ContinuedExecutionRunId != nil {\n\t\tenc.AddString(\"continuedExecutionRunId\", *v.ContinuedExecutionRunId)\n\t}\n\tif v.Initiator != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"initiator\", *v.Initiator))\n\t}\n\tif v.ContinuedFailureReason != nil {\n\t\tenc.AddString(\"continuedFailureReason\", *v.ContinuedFailureReason)\n\t}\n\tif v.ContinuedFailureDetails != nil {\n\t\tenc.AddString(\"continuedFailureDetails\", base64.StdEncoding.EncodeToString(v.ContinuedFailureDetails))\n\t}\n\tif v.LastCompletionResult != nil {\n\t\tenc.AddString(\"lastCompletionResult\", base64.StdEncoding.EncodeToString(v.LastCompletionResult))\n\t}\n\tif v.OriginalExecutionRunId != nil {\n\t\tenc.AddString(\"originalExecutionRunId\", *v.OriginalExecutionRunId)\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\tif v.FirstExecutionRunId != nil {\n\t\tenc.AddString(\"firstExecutionRunId\", *v.FirstExecutionRunId)\n\t}\n\tif v.FirstScheduledTimeNano != nil {\n\t\tenc.AddInt64(\"firstScheduledTimeNano\", *v.FirstScheduledTimeNano)\n\t}\n\tif v.RetryPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"retryPolicy\", v.RetryPolicy))\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.ExpirationTimestamp != nil {\n\t\tenc.AddInt64(\"expirationTimestamp\", *v.ExpirationTimestamp)\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.FirstDecisionTaskBackoffSeconds != nil {\n\t\tenc.AddInt32(\"firstDecisionTaskBackoffSeconds\", *v.FirstDecisionTaskBackoffSeconds)\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", v.Memo))\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", v.SearchAttributes))\n\t}\n\tif v.PrevAutoResetPoints != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"prevAutoResetPoints\", v.PrevAutoResetPoints))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\tif v.RequestId != nil {\n\t\tenc.AddString(\"requestId\", *v.RequestId)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"activeClusterSelectionPolicy\", v.ActiveClusterSelectionPolicy))\n\t}\n\treturn err\n}\n\n// GetWorkflowType returns the value of WorkflowType if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowType returns true if WorkflowType is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetWorkflowType() bool {\n\treturn v != nil && v.WorkflowType != nil\n}\n\n// GetParentWorkflowDomain returns the value of ParentWorkflowDomain if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetParentWorkflowDomain() (o string) {\n\tif v != nil && v.ParentWorkflowDomain != nil {\n\t\treturn *v.ParentWorkflowDomain\n\t}\n\n\treturn\n}\n\n// IsSetParentWorkflowDomain returns true if ParentWorkflowDomain is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetParentWorkflowDomain() bool {\n\treturn v != nil && v.ParentWorkflowDomain != nil\n}\n\n// GetParentWorkflowExecution returns the value of ParentWorkflowExecution if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetParentWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.ParentWorkflowExecution != nil {\n\t\treturn v.ParentWorkflowExecution\n\t}\n\n\treturn\n}\n\n// IsSetParentWorkflowExecution returns true if ParentWorkflowExecution is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetParentWorkflowExecution() bool {\n\treturn v != nil && v.ParentWorkflowExecution != nil\n}\n\n// GetParentInitiatedEventId returns the value of ParentInitiatedEventId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetParentInitiatedEventId() (o int64) {\n\tif v != nil && v.ParentInitiatedEventId != nil {\n\t\treturn *v.ParentInitiatedEventId\n\t}\n\n\treturn\n}\n\n// IsSetParentInitiatedEventId returns true if ParentInitiatedEventId is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetParentInitiatedEventId() bool {\n\treturn v != nil && v.ParentInitiatedEventId != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetExecutionStartToCloseTimeoutSeconds returns the value of ExecutionStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetExecutionStartToCloseTimeoutSeconds returns true if ExecutionStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetExecutionStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil\n}\n\n// GetTaskStartToCloseTimeoutSeconds returns the value of TaskStartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetTaskStartToCloseTimeoutSeconds returns true if TaskStartToCloseTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetTaskStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.TaskStartToCloseTimeoutSeconds != nil\n}\n\n// GetContinuedExecutionRunId returns the value of ContinuedExecutionRunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetContinuedExecutionRunId() (o string) {\n\tif v != nil && v.ContinuedExecutionRunId != nil {\n\t\treturn *v.ContinuedExecutionRunId\n\t}\n\n\treturn\n}\n\n// IsSetContinuedExecutionRunId returns true if ContinuedExecutionRunId is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetContinuedExecutionRunId() bool {\n\treturn v != nil && v.ContinuedExecutionRunId != nil\n}\n\n// GetInitiator returns the value of Initiator if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\n\treturn\n}\n\n// IsSetInitiator returns true if Initiator is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetInitiator() bool {\n\treturn v != nil && v.Initiator != nil\n}\n\n// GetContinuedFailureReason returns the value of ContinuedFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetContinuedFailureReason() (o string) {\n\tif v != nil && v.ContinuedFailureReason != nil {\n\t\treturn *v.ContinuedFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetContinuedFailureReason returns true if ContinuedFailureReason is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetContinuedFailureReason() bool {\n\treturn v != nil && v.ContinuedFailureReason != nil\n}\n\n// GetContinuedFailureDetails returns the value of ContinuedFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetContinuedFailureDetails() (o []byte) {\n\tif v != nil && v.ContinuedFailureDetails != nil {\n\t\treturn v.ContinuedFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetContinuedFailureDetails returns true if ContinuedFailureDetails is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetContinuedFailureDetails() bool {\n\treturn v != nil && v.ContinuedFailureDetails != nil\n}\n\n// GetLastCompletionResult returns the value of LastCompletionResult if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetLastCompletionResult() (o []byte) {\n\tif v != nil && v.LastCompletionResult != nil {\n\t\treturn v.LastCompletionResult\n\t}\n\n\treturn\n}\n\n// IsSetLastCompletionResult returns true if LastCompletionResult is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetLastCompletionResult() bool {\n\treturn v != nil && v.LastCompletionResult != nil\n}\n\n// GetOriginalExecutionRunId returns the value of OriginalExecutionRunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetOriginalExecutionRunId() (o string) {\n\tif v != nil && v.OriginalExecutionRunId != nil {\n\t\treturn *v.OriginalExecutionRunId\n\t}\n\n\treturn\n}\n\n// IsSetOriginalExecutionRunId returns true if OriginalExecutionRunId is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetOriginalExecutionRunId() bool {\n\treturn v != nil && v.OriginalExecutionRunId != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\n// GetFirstExecutionRunId returns the value of FirstExecutionRunId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstExecutionRunId() (o string) {\n\tif v != nil && v.FirstExecutionRunId != nil {\n\t\treturn *v.FirstExecutionRunId\n\t}\n\n\treturn\n}\n\n// IsSetFirstExecutionRunId returns true if FirstExecutionRunId is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetFirstExecutionRunId() bool {\n\treturn v != nil && v.FirstExecutionRunId != nil\n}\n\n// GetFirstScheduledTimeNano returns the value of FirstScheduledTimeNano if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstScheduledTimeNano() (o int64) {\n\tif v != nil && v.FirstScheduledTimeNano != nil {\n\t\treturn *v.FirstScheduledTimeNano\n\t}\n\n\treturn\n}\n\n// IsSetFirstScheduledTimeNano returns true if FirstScheduledTimeNano is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetFirstScheduledTimeNano() bool {\n\treturn v != nil && v.FirstScheduledTimeNano != nil\n}\n\n// GetRetryPolicy returns the value of RetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetRetryPolicy() (o *RetryPolicy) {\n\tif v != nil && v.RetryPolicy != nil {\n\t\treturn v.RetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetRetryPolicy returns true if RetryPolicy is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetRetryPolicy() bool {\n\treturn v != nil && v.RetryPolicy != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetExpirationTimestamp returns the value of ExpirationTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetExpirationTimestamp() (o int64) {\n\tif v != nil && v.ExpirationTimestamp != nil {\n\t\treturn *v.ExpirationTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetExpirationTimestamp returns true if ExpirationTimestamp is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetExpirationTimestamp() bool {\n\treturn v != nil && v.ExpirationTimestamp != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetFirstDecisionTaskBackoffSeconds returns the value of FirstDecisionTaskBackoffSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstDecisionTaskBackoffSeconds() (o int32) {\n\tif v != nil && v.FirstDecisionTaskBackoffSeconds != nil {\n\t\treturn *v.FirstDecisionTaskBackoffSeconds\n\t}\n\n\treturn\n}\n\n// IsSetFirstDecisionTaskBackoffSeconds returns true if FirstDecisionTaskBackoffSeconds is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetFirstDecisionTaskBackoffSeconds() bool {\n\treturn v != nil && v.FirstDecisionTaskBackoffSeconds != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetPrevAutoResetPoints returns the value of PrevAutoResetPoints if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetPrevAutoResetPoints() (o *ResetPoints) {\n\tif v != nil && v.PrevAutoResetPoints != nil {\n\t\treturn v.PrevAutoResetPoints\n\t}\n\n\treturn\n}\n\n// IsSetPrevAutoResetPoints returns true if PrevAutoResetPoints is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetPrevAutoResetPoints() bool {\n\treturn v != nil && v.PrevAutoResetPoints != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetHeader() (o *Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\n// GetRequestId returns the value of RequestId if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetRequestId() (o string) {\n\tif v != nil && v.RequestId != nil {\n\t\treturn *v.RequestId\n\t}\n\n\treturn\n}\n\n// IsSetRequestId returns true if RequestId is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetRequestId() bool {\n\treturn v != nil && v.RequestId != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionStartedEventAttributes) GetActiveClusterSelectionPolicy() (o *ActiveClusterSelectionPolicy) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *WorkflowExecutionStartedEventAttributes) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\ntype WorkflowExecutionStatus int32\n\nconst (\n\tWorkflowExecutionStatusPending        WorkflowExecutionStatus = 0\n\tWorkflowExecutionStatusStarted        WorkflowExecutionStatus = 1\n\tWorkflowExecutionStatusCompleted      WorkflowExecutionStatus = 2\n\tWorkflowExecutionStatusFailed         WorkflowExecutionStatus = 3\n\tWorkflowExecutionStatusCanceled       WorkflowExecutionStatus = 4\n\tWorkflowExecutionStatusTerminated     WorkflowExecutionStatus = 5\n\tWorkflowExecutionStatusContinuedAsNew WorkflowExecutionStatus = 6\n\tWorkflowExecutionStatusTimedOut       WorkflowExecutionStatus = 7\n)\n\n// WorkflowExecutionStatus_Values returns all recognized values of WorkflowExecutionStatus.\nfunc WorkflowExecutionStatus_Values() []WorkflowExecutionStatus {\n\treturn []WorkflowExecutionStatus{\n\t\tWorkflowExecutionStatusPending,\n\t\tWorkflowExecutionStatusStarted,\n\t\tWorkflowExecutionStatusCompleted,\n\t\tWorkflowExecutionStatusFailed,\n\t\tWorkflowExecutionStatusCanceled,\n\t\tWorkflowExecutionStatusTerminated,\n\t\tWorkflowExecutionStatusContinuedAsNew,\n\t\tWorkflowExecutionStatusTimedOut,\n\t}\n}\n\n// UnmarshalText tries to decode WorkflowExecutionStatus from a byte slice\n// containing its name.\n//\n//\tvar v WorkflowExecutionStatus\n//\terr := v.UnmarshalText([]byte(\"PENDING\"))\nfunc (v *WorkflowExecutionStatus) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"PENDING\":\n\t\t*v = WorkflowExecutionStatusPending\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*v = WorkflowExecutionStatusStarted\n\t\treturn nil\n\tcase \"COMPLETED\":\n\t\t*v = WorkflowExecutionStatusCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*v = WorkflowExecutionStatusFailed\n\t\treturn nil\n\tcase \"CANCELED\":\n\t\t*v = WorkflowExecutionStatusCanceled\n\t\treturn nil\n\tcase \"TERMINATED\":\n\t\t*v = WorkflowExecutionStatusTerminated\n\t\treturn nil\n\tcase \"CONTINUED_AS_NEW\":\n\t\t*v = WorkflowExecutionStatusContinuedAsNew\n\t\treturn nil\n\tcase \"TIMED_OUT\":\n\t\t*v = WorkflowExecutionStatusTimedOut\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowExecutionStatus\", err)\n\t\t}\n\t\t*v = WorkflowExecutionStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowExecutionStatus to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v WorkflowExecutionStatus) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"PENDING\"), nil\n\tcase 1:\n\t\treturn []byte(\"STARTED\"), nil\n\tcase 2:\n\t\treturn []byte(\"COMPLETED\"), nil\n\tcase 3:\n\t\treturn []byte(\"FAILED\"), nil\n\tcase 4:\n\t\treturn []byte(\"CANCELED\"), nil\n\tcase 5:\n\t\treturn []byte(\"TERMINATED\"), nil\n\tcase 6:\n\t\treturn []byte(\"CONTINUED_AS_NEW\"), nil\n\tcase 7:\n\t\treturn []byte(\"TIMED_OUT\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionStatus.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v WorkflowExecutionStatus) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"PENDING\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"STARTED\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"COMPLETED\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"FAILED\")\n\tcase 4:\n\t\tenc.AddString(\"name\", \"CANCELED\")\n\tcase 5:\n\t\tenc.AddString(\"name\", \"TERMINATED\")\n\tcase 6:\n\t\tenc.AddString(\"name\", \"CONTINUED_AS_NEW\")\n\tcase 7:\n\t\tenc.AddString(\"name\", \"TIMED_OUT\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v WorkflowExecutionStatus) Ptr() *WorkflowExecutionStatus {\n\treturn &v\n}\n\n// Encode encodes WorkflowExecutionStatus directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v WorkflowExecutionStatus\n//\treturn v.Encode(sWriter)\nfunc (v WorkflowExecutionStatus) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates WorkflowExecutionStatus into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v WorkflowExecutionStatus) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes WorkflowExecutionStatus from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return WorkflowExecutionStatus(0), err\n//\t}\n//\n//\tvar v WorkflowExecutionStatus\n//\tif err := v.FromWire(x); err != nil {\n//\t  return WorkflowExecutionStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowExecutionStatus) FromWire(w wire.Value) error {\n\t*v = (WorkflowExecutionStatus)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded WorkflowExecutionStatus directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v WorkflowExecutionStatus\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return WorkflowExecutionStatus(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowExecutionStatus) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (WorkflowExecutionStatus)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of WorkflowExecutionStatus.\nfunc (v WorkflowExecutionStatus) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"PENDING\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\tcase 2:\n\t\treturn \"COMPLETED\"\n\tcase 3:\n\t\treturn \"FAILED\"\n\tcase 4:\n\t\treturn \"CANCELED\"\n\tcase 5:\n\t\treturn \"TERMINATED\"\n\tcase 6:\n\t\treturn \"CONTINUED_AS_NEW\"\n\tcase 7:\n\t\treturn \"TIMED_OUT\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowExecutionStatus(%d)\", w)\n}\n\n// Equals returns true if this WorkflowExecutionStatus value matches the provided\n// value.\nfunc (v WorkflowExecutionStatus) Equals(rhs WorkflowExecutionStatus) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes WorkflowExecutionStatus into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v WorkflowExecutionStatus) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"PENDING\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"STARTED\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"COMPLETED\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"FAILED\\\"\"), nil\n\tcase 4:\n\t\treturn ([]byte)(\"\\\"CANCELED\\\"\"), nil\n\tcase 5:\n\t\treturn ([]byte)(\"\\\"TERMINATED\\\"\"), nil\n\tcase 6:\n\t\treturn ([]byte)(\"\\\"CONTINUED_AS_NEW\\\"\"), nil\n\tcase 7:\n\t\treturn ([]byte)(\"\\\"TIMED_OUT\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode WorkflowExecutionStatus from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *WorkflowExecutionStatus) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"WorkflowExecutionStatus\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"WorkflowExecutionStatus\")\n\t\t}\n\t\t*v = (WorkflowExecutionStatus)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"WorkflowExecutionStatus\")\n\t}\n}\n\ntype WorkflowExecutionTerminatedEventAttributes struct {\n\tReason   *string `json:\"reason,omitempty\"`\n\tDetails  []byte  `json:\"details,omitempty\"`\n\tIdentity *string `json:\"identity,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionTerminatedEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionTerminatedEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Reason != nil {\n\t\tw, err = wire.NewValueString(*(v.Reason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tw, err = wire.NewValueBinary(v.Details), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tw, err = wire.NewValueString(*(v.Identity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionTerminatedEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionTerminatedEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionTerminatedEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionTerminatedEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Reason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Details, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Identity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionTerminatedEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionTerminatedEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Reason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Reason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Details != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Details); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Identity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Identity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionTerminatedEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionTerminatedEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Reason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Details, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Identity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionTerminatedEventAttributes\n// struct.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Reason != nil {\n\t\tfields[i] = fmt.Sprintf(\"Reason: %v\", *(v.Reason))\n\t\ti++\n\t}\n\tif v.Details != nil {\n\t\tfields[i] = fmt.Sprintf(\"Details: %v\", v.Details)\n\t\ti++\n\t}\n\tif v.Identity != nil {\n\t\tfields[i] = fmt.Sprintf(\"Identity: %v\", *(v.Identity))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionTerminatedEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionTerminatedEventAttributes match the\n// provided WorkflowExecutionTerminatedEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) Equals(rhs *WorkflowExecutionTerminatedEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Reason, rhs.Reason) {\n\t\treturn false\n\t}\n\tif !((v.Details == nil && rhs.Details == nil) || (v.Details != nil && rhs.Details != nil && bytes.Equal(v.Details, rhs.Details))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Identity, rhs.Identity) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionTerminatedEventAttributes.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Reason != nil {\n\t\tenc.AddString(\"reason\", *v.Reason)\n\t}\n\tif v.Details != nil {\n\t\tenc.AddString(\"details\", base64.StdEncoding.EncodeToString(v.Details))\n\t}\n\tif v.Identity != nil {\n\t\tenc.AddString(\"identity\", *v.Identity)\n\t}\n\treturn err\n}\n\n// GetReason returns the value of Reason if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\n\treturn\n}\n\n// IsSetReason returns true if Reason is not nil.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) IsSetReason() bool {\n\treturn v != nil && v.Reason != nil\n}\n\n// GetDetails returns the value of Details if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\n\treturn\n}\n\n// IsSetDetails returns true if Details is not nil.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) IsSetDetails() bool {\n\treturn v != nil && v.Details != nil\n}\n\n// GetIdentity returns the value of Identity if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) GetIdentity() (o string) {\n\tif v != nil && v.Identity != nil {\n\t\treturn *v.Identity\n\t}\n\n\treturn\n}\n\n// IsSetIdentity returns true if Identity is not nil.\nfunc (v *WorkflowExecutionTerminatedEventAttributes) IsSetIdentity() bool {\n\treturn v != nil && v.Identity != nil\n}\n\ntype WorkflowExecutionTimedOutEventAttributes struct {\n\tTimeoutType *TimeoutType `json:\"timeoutType,omitempty\"`\n}\n\n// ToWire translates a WorkflowExecutionTimedOutEventAttributes struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionTimedOutEventAttributes) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TimeoutType != nil {\n\t\tw, err = v.TimeoutType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowExecutionTimedOutEventAttributes struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionTimedOutEventAttributes struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionTimedOutEventAttributes\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionTimedOutEventAttributes) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x TimeoutType\n\t\t\t\tx, err = _TimeoutType_Read(field.Value)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowExecutionTimedOutEventAttributes struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionTimedOutEventAttributes struct could not be encoded.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TimeoutType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowExecutionTimedOutEventAttributes struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionTimedOutEventAttributes struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x TimeoutType\n\t\t\tx, err = _TimeoutType_Decode(sr)\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionTimedOutEventAttributes\n// struct.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionTimedOutEventAttributes{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionTimedOutEventAttributes match the\n// provided WorkflowExecutionTimedOutEventAttributes.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) Equals(rhs *WorkflowExecutionTimedOutEventAttributes) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_TimeoutType_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionTimedOutEventAttributes.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TimeoutType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"timeoutType\", *v.TimeoutType))\n\t}\n\treturn err\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *WorkflowExecutionTimedOutEventAttributes) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\ntype WorkflowIdReusePolicy int32\n\nconst (\n\tWorkflowIdReusePolicyAllowDuplicateFailedOnly WorkflowIdReusePolicy = 0\n\tWorkflowIdReusePolicyAllowDuplicate           WorkflowIdReusePolicy = 1\n\tWorkflowIdReusePolicyRejectDuplicate          WorkflowIdReusePolicy = 2\n\tWorkflowIdReusePolicyTerminateIfRunning       WorkflowIdReusePolicy = 3\n)\n\n// WorkflowIdReusePolicy_Values returns all recognized values of WorkflowIdReusePolicy.\nfunc WorkflowIdReusePolicy_Values() []WorkflowIdReusePolicy {\n\treturn []WorkflowIdReusePolicy{\n\t\tWorkflowIdReusePolicyAllowDuplicateFailedOnly,\n\t\tWorkflowIdReusePolicyAllowDuplicate,\n\t\tWorkflowIdReusePolicyRejectDuplicate,\n\t\tWorkflowIdReusePolicyTerminateIfRunning,\n\t}\n}\n\n// UnmarshalText tries to decode WorkflowIdReusePolicy from a byte slice\n// containing its name.\n//\n//\tvar v WorkflowIdReusePolicy\n//\terr := v.UnmarshalText([]byte(\"AllowDuplicateFailedOnly\"))\nfunc (v *WorkflowIdReusePolicy) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"AllowDuplicateFailedOnly\":\n\t\t*v = WorkflowIdReusePolicyAllowDuplicateFailedOnly\n\t\treturn nil\n\tcase \"AllowDuplicate\":\n\t\t*v = WorkflowIdReusePolicyAllowDuplicate\n\t\treturn nil\n\tcase \"RejectDuplicate\":\n\t\t*v = WorkflowIdReusePolicyRejectDuplicate\n\t\treturn nil\n\tcase \"TerminateIfRunning\":\n\t\t*v = WorkflowIdReusePolicyTerminateIfRunning\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowIdReusePolicy\", err)\n\t\t}\n\t\t*v = WorkflowIdReusePolicy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowIdReusePolicy to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v WorkflowIdReusePolicy) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"AllowDuplicateFailedOnly\"), nil\n\tcase 1:\n\t\treturn []byte(\"AllowDuplicate\"), nil\n\tcase 2:\n\t\treturn []byte(\"RejectDuplicate\"), nil\n\tcase 3:\n\t\treturn []byte(\"TerminateIfRunning\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowIdReusePolicy.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v WorkflowIdReusePolicy) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"AllowDuplicateFailedOnly\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"AllowDuplicate\")\n\tcase 2:\n\t\tenc.AddString(\"name\", \"RejectDuplicate\")\n\tcase 3:\n\t\tenc.AddString(\"name\", \"TerminateIfRunning\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v WorkflowIdReusePolicy) Ptr() *WorkflowIdReusePolicy {\n\treturn &v\n}\n\n// Encode encodes WorkflowIdReusePolicy directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v WorkflowIdReusePolicy\n//\treturn v.Encode(sWriter)\nfunc (v WorkflowIdReusePolicy) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates WorkflowIdReusePolicy into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v WorkflowIdReusePolicy) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes WorkflowIdReusePolicy from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return WorkflowIdReusePolicy(0), err\n//\t}\n//\n//\tvar v WorkflowIdReusePolicy\n//\tif err := v.FromWire(x); err != nil {\n//\t  return WorkflowIdReusePolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowIdReusePolicy) FromWire(w wire.Value) error {\n\t*v = (WorkflowIdReusePolicy)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded WorkflowIdReusePolicy directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v WorkflowIdReusePolicy\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return WorkflowIdReusePolicy(0), err\n//\t}\n//\treturn v, nil\nfunc (v *WorkflowIdReusePolicy) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (WorkflowIdReusePolicy)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of WorkflowIdReusePolicy.\nfunc (v WorkflowIdReusePolicy) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"AllowDuplicateFailedOnly\"\n\tcase 1:\n\t\treturn \"AllowDuplicate\"\n\tcase 2:\n\t\treturn \"RejectDuplicate\"\n\tcase 3:\n\t\treturn \"TerminateIfRunning\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowIdReusePolicy(%d)\", w)\n}\n\n// Equals returns true if this WorkflowIdReusePolicy value matches the provided\n// value.\nfunc (v WorkflowIdReusePolicy) Equals(rhs WorkflowIdReusePolicy) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes WorkflowIdReusePolicy into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v WorkflowIdReusePolicy) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"AllowDuplicateFailedOnly\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"AllowDuplicate\\\"\"), nil\n\tcase 2:\n\t\treturn ([]byte)(\"\\\"RejectDuplicate\\\"\"), nil\n\tcase 3:\n\t\treturn ([]byte)(\"\\\"TerminateIfRunning\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode WorkflowIdReusePolicy from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *WorkflowIdReusePolicy) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"WorkflowIdReusePolicy\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"WorkflowIdReusePolicy\")\n\t\t}\n\t\t*v = (WorkflowIdReusePolicy)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"WorkflowIdReusePolicy\")\n\t}\n}\n\ntype WorkflowQuery struct {\n\tQueryType *string `json:\"queryType,omitempty\"`\n\tQueryArgs []byte  `json:\"queryArgs,omitempty\"`\n}\n\n// ToWire translates a WorkflowQuery struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowQuery) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [2]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.QueryType != nil {\n\t\tw, err = wire.NewValueString(*(v.QueryType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.QueryArgs != nil {\n\t\tw, err = wire.NewValueBinary(v.QueryArgs), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowQuery struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowQuery struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowQuery\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowQuery) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.QueryType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.QueryArgs, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowQuery struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowQuery struct could not be encoded.\nfunc (v *WorkflowQuery) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.QueryType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.QueryType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueryArgs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.QueryArgs); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowQuery struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowQuery struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowQuery) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.QueryType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.QueryArgs, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowQuery\n// struct.\nfunc (v *WorkflowQuery) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [2]string\n\ti := 0\n\tif v.QueryType != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryType: %v\", *(v.QueryType))\n\t\ti++\n\t}\n\tif v.QueryArgs != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueryArgs: %v\", v.QueryArgs)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowQuery{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowQuery match the\n// provided WorkflowQuery.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowQuery) Equals(rhs *WorkflowQuery) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.QueryType, rhs.QueryType) {\n\t\treturn false\n\t}\n\tif !((v.QueryArgs == nil && rhs.QueryArgs == nil) || (v.QueryArgs != nil && rhs.QueryArgs != nil && bytes.Equal(v.QueryArgs, rhs.QueryArgs))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowQuery.\nfunc (v *WorkflowQuery) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.QueryType != nil {\n\t\tenc.AddString(\"queryType\", *v.QueryType)\n\t}\n\tif v.QueryArgs != nil {\n\t\tenc.AddString(\"queryArgs\", base64.StdEncoding.EncodeToString(v.QueryArgs))\n\t}\n\treturn err\n}\n\n// GetQueryType returns the value of QueryType if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowQuery) GetQueryType() (o string) {\n\tif v != nil && v.QueryType != nil {\n\t\treturn *v.QueryType\n\t}\n\n\treturn\n}\n\n// IsSetQueryType returns true if QueryType is not nil.\nfunc (v *WorkflowQuery) IsSetQueryType() bool {\n\treturn v != nil && v.QueryType != nil\n}\n\n// GetQueryArgs returns the value of QueryArgs if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowQuery) GetQueryArgs() (o []byte) {\n\tif v != nil && v.QueryArgs != nil {\n\t\treturn v.QueryArgs\n\t}\n\n\treturn\n}\n\n// IsSetQueryArgs returns true if QueryArgs is not nil.\nfunc (v *WorkflowQuery) IsSetQueryArgs() bool {\n\treturn v != nil && v.QueryArgs != nil\n}\n\ntype WorkflowQueryResult struct {\n\tResultType   *QueryResultType `json:\"resultType,omitempty\"`\n\tAnswer       []byte           `json:\"answer,omitempty\"`\n\tErrorMessage *string          `json:\"errorMessage,omitempty\"`\n}\n\n// ToWire translates a WorkflowQueryResult struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowQueryResult) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ResultType != nil {\n\t\tw, err = v.ResultType.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Answer != nil {\n\t\tw, err = wire.NewValueBinary(v.Answer), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ErrorMessage != nil {\n\t\tw, err = wire.NewValueString(*(v.ErrorMessage)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _QueryResultType_Read(w wire.Value) (QueryResultType, error) {\n\tvar v QueryResultType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a WorkflowQueryResult struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowQueryResult struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowQueryResult\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowQueryResult) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x QueryResultType\n\t\t\t\tx, err = _QueryResultType_Read(field.Value)\n\t\t\t\tv.ResultType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Answer, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ErrorMessage = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowQueryResult struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowQueryResult struct could not be encoded.\nfunc (v *WorkflowQueryResult) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ResultType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.ResultType.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Answer != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Answer); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ErrorMessage != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ErrorMessage)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _QueryResultType_Decode(sr stream.Reader) (QueryResultType, error) {\n\tvar v QueryResultType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a WorkflowQueryResult struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowQueryResult struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowQueryResult) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x QueryResultType\n\t\t\tx, err = _QueryResultType_Decode(sr)\n\t\t\tv.ResultType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.Answer, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ErrorMessage = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowQueryResult\n// struct.\nfunc (v *WorkflowQueryResult) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.ResultType != nil {\n\t\tfields[i] = fmt.Sprintf(\"ResultType: %v\", *(v.ResultType))\n\t\ti++\n\t}\n\tif v.Answer != nil {\n\t\tfields[i] = fmt.Sprintf(\"Answer: %v\", v.Answer)\n\t\ti++\n\t}\n\tif v.ErrorMessage != nil {\n\t\tfields[i] = fmt.Sprintf(\"ErrorMessage: %v\", *(v.ErrorMessage))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowQueryResult{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _QueryResultType_EqualsPtr(lhs, rhs *QueryResultType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this WorkflowQueryResult match the\n// provided WorkflowQueryResult.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowQueryResult) Equals(rhs *WorkflowQueryResult) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_QueryResultType_EqualsPtr(v.ResultType, rhs.ResultType) {\n\t\treturn false\n\t}\n\tif !((v.Answer == nil && rhs.Answer == nil) || (v.Answer != nil && rhs.Answer != nil && bytes.Equal(v.Answer, rhs.Answer))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ErrorMessage, rhs.ErrorMessage) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowQueryResult.\nfunc (v *WorkflowQueryResult) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ResultType != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"resultType\", *v.ResultType))\n\t}\n\tif v.Answer != nil {\n\t\tenc.AddString(\"answer\", base64.StdEncoding.EncodeToString(v.Answer))\n\t}\n\tif v.ErrorMessage != nil {\n\t\tenc.AddString(\"errorMessage\", *v.ErrorMessage)\n\t}\n\treturn err\n}\n\n// GetResultType returns the value of ResultType if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowQueryResult) GetResultType() (o QueryResultType) {\n\tif v != nil && v.ResultType != nil {\n\t\treturn *v.ResultType\n\t}\n\n\treturn\n}\n\n// IsSetResultType returns true if ResultType is not nil.\nfunc (v *WorkflowQueryResult) IsSetResultType() bool {\n\treturn v != nil && v.ResultType != nil\n}\n\n// GetAnswer returns the value of Answer if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowQueryResult) GetAnswer() (o []byte) {\n\tif v != nil && v.Answer != nil {\n\t\treturn v.Answer\n\t}\n\n\treturn\n}\n\n// IsSetAnswer returns true if Answer is not nil.\nfunc (v *WorkflowQueryResult) IsSetAnswer() bool {\n\treturn v != nil && v.Answer != nil\n}\n\n// GetErrorMessage returns the value of ErrorMessage if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowQueryResult) GetErrorMessage() (o string) {\n\tif v != nil && v.ErrorMessage != nil {\n\t\treturn *v.ErrorMessage\n\t}\n\n\treturn\n}\n\n// IsSetErrorMessage returns true if ErrorMessage is not nil.\nfunc (v *WorkflowQueryResult) IsSetErrorMessage() bool {\n\treturn v != nil && v.ErrorMessage != nil\n}\n\ntype WorkflowType struct {\n\tName *string `json:\"name,omitempty\"`\n}\n\n// ToWire translates a WorkflowType struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowType) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowType struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowType struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowType) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowType struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowType struct could not be encoded.\nfunc (v *WorkflowType) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowType struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowType struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowType) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowType\n// struct.\nfunc (v *WorkflowType) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowType{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowType match the\n// provided WorkflowType.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowType) Equals(rhs *WorkflowType) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowType.\nfunc (v *WorkflowType) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowType) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *WorkflowType) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\ntype WorkflowTypeFilter struct {\n\tName *string `json:\"name,omitempty\"`\n}\n\n// ToWire translates a WorkflowTypeFilter struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowTypeFilter) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a WorkflowTypeFilter struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowTypeFilter struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowTypeFilter\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowTypeFilter) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a WorkflowTypeFilter struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowTypeFilter struct could not be encoded.\nfunc (v *WorkflowTypeFilter) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a WorkflowTypeFilter struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowTypeFilter struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowTypeFilter) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowTypeFilter\n// struct.\nfunc (v *WorkflowTypeFilter) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowTypeFilter{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this WorkflowTypeFilter match the\n// provided WorkflowTypeFilter.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowTypeFilter) Equals(rhs *WorkflowTypeFilter) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowTypeFilter.\nfunc (v *WorkflowTypeFilter) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowTypeFilter) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *WorkflowTypeFilter) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"shared\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/shared\",\n\tFilePath: \"shared.thrift\",\n\tSHA1:     \"5a8ec0270fb416939964d113fdb6059662176745\",\n\tRaw:      rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence\\n\\nexception BadRequestError {\\n  1: required string message\\n}\\n\\nexception InternalServiceError {\\n  1: required string message\\n}\\n\\nexception InternalDataInconsistencyError {\\n  1: required string message\\n}\\n\\nexception DomainAlreadyExistsError {\\n  1: required string message\\n}\\n\\nexception WorkflowExecutionAlreadyStartedError {\\n  10: optional string message\\n  20: optional string startRequestId\\n  30: optional string runId\\n}\\n\\nexception WorkflowExecutionAlreadyCompletedError {\\n  1: required string message\\n}\\n\\nexception EntityNotExistsError {\\n  1: required string message\\n  2: optional string currentCluster\\n  3: optional string activeCluster\\n  4: required list<string> activeClusters // todo(david.porter) remove as its disused\\n}\\n\\nexception ServiceBusyError {\\n  1: required string message\\n  2: optional string reason\\n}\\n\\nexception CancellationAlreadyRequestedError {\\n  1: required string message\\n}\\n\\nexception QueryFailedError {\\n  1: required string message\\n}\\n\\nexception DomainNotActiveError {\\n  1: required string message\\n  2: required string domainName\\n  3: required string currentCluster\\n  4: required string activeCluster\\n  5: required list<string> activeClusters // todo (david.porter) remove this field as it's disused\\n}\\n\\nexception LimitExceededError {\\n  1: required string message\\n}\\n\\nexception AccessDeniedError {\\n  1: required string message\\n}\\n\\nexception RetryTaskV2Error {\\n  1: required string message\\n  2: optional string domainId\\n  3: optional string workflowId\\n  4: optional string runId\\n  5: optional i64 (js.type = \\\"Long\\\") startEventId\\n  6: optional i64 (js.type = \\\"Long\\\") startEventVersion\\n  7: optional i64 (js.type = \\\"Long\\\") endEventId\\n  8: optional i64 (js.type = \\\"Long\\\") endEventVersion\\n}\\n\\nexception ClientVersionNotSupportedError {\\n  1: required string featureVersion\\n  2: required string clientImpl\\n  3: required string supportedVersions\\n}\\n\\nexception FeatureNotEnabledError {\\n  1: required string featureFlag\\n}\\n\\nexception CurrentBranchChangedError {\\n  10: required string message\\n  20: required binary currentBranchToken\\n}\\n\\nexception RemoteSyncMatchedError {\\n  10: required string message\\n}\\n\\nexception StickyWorkerUnavailableError {\\n  1: required string message\\n}\\n\\nexception TaskListNotOwnedByHostError {\\n    1: required string ownedByIdentity\\n    2: required string myIdentity\\n    3: required string tasklistName\\n}\\n\\nenum WorkflowIdReusePolicy {\\n  /*\\n   * allow start a workflow execution using the same workflow ID,\\n   * when workflow not running, and the last execution close state is in\\n   * [terminated, cancelled, timeouted, failed].\\n   */\\n  AllowDuplicateFailedOnly,\\n  /*\\n   * allow start a workflow execution using the same workflow ID,\\n   * when workflow not running.\\n   */\\n  AllowDuplicate,\\n  /*\\n   * do not allow start a workflow execution using the same workflow ID at all\\n   */\\n  RejectDuplicate,\\n  /*\\n   * if a workflow is running using the same workflow ID, terminate it and start a new one\\n   */\\n  TerminateIfRunning,\\n}\\n\\nenum DomainStatus {\\n  REGISTERED,\\n  DEPRECATED,\\n  DELETED,\\n}\\n\\nenum TimeoutType {\\n  START_TO_CLOSE,\\n  SCHEDULE_TO_START,\\n  SCHEDULE_TO_CLOSE,\\n  HEARTBEAT,\\n}\\n\\nenum ParentClosePolicy {\\n\\tABANDON,\\n\\tREQUEST_CANCEL,\\n\\tTERMINATE,\\n}\\n\\n\\n// whenever this list of decision is changed\\n// do change the mutableStateBuilder.go\\n// function shouldBufferEvent\\n// to make sure wo do the correct event ordering\\nenum DecisionType {\\n  ScheduleActivityTask,\\n  RequestCancelActivityTask,\\n  StartTimer,\\n  CompleteWorkflowExecution,\\n  FailWorkflowExecution,\\n  CancelTimer,\\n  CancelWorkflowExecution,\\n  RequestCancelExternalWorkflowExecution,\\n  RecordMarker,\\n  ContinueAsNewWorkflowExecution,\\n  StartChildWorkflowExecution,\\n  SignalExternalWorkflowExecution,\\n  UpsertWorkflowSearchAttributes,\\n}\\n\\nenum EventType {\\n  WorkflowExecutionStarted,\\n  WorkflowExecutionCompleted,\\n  WorkflowExecutionFailed,\\n  WorkflowExecutionTimedOut,\\n  DecisionTaskScheduled,\\n  DecisionTaskStarted,\\n  DecisionTaskCompleted,\\n  DecisionTaskTimedOut\\n  DecisionTaskFailed,\\n  ActivityTaskScheduled,\\n  ActivityTaskStarted,\\n  ActivityTaskCompleted,\\n  ActivityTaskFailed,\\n  ActivityTaskTimedOut,\\n  ActivityTaskCancelRequested,\\n  RequestCancelActivityTaskFailed,\\n  ActivityTaskCanceled,\\n  TimerStarted,\\n  TimerFired,\\n  CancelTimerFailed,\\n  TimerCanceled,\\n  WorkflowExecutionCancelRequested,\\n  WorkflowExecutionCanceled,\\n  RequestCancelExternalWorkflowExecutionInitiated,\\n  RequestCancelExternalWorkflowExecutionFailed,\\n  ExternalWorkflowExecutionCancelRequested,\\n  MarkerRecorded,\\n  WorkflowExecutionSignaled,\\n  WorkflowExecutionTerminated,\\n  WorkflowExecutionContinuedAsNew,\\n  StartChildWorkflowExecutionInitiated,\\n  StartChildWorkflowExecutionFailed,\\n  ChildWorkflowExecutionStarted,\\n  ChildWorkflowExecutionCompleted,\\n  ChildWorkflowExecutionFailed,\\n  ChildWorkflowExecutionCanceled,\\n  ChildWorkflowExecutionTimedOut,\\n  ChildWorkflowExecutionTerminated,\\n  SignalExternalWorkflowExecutionInitiated,\\n  SignalExternalWorkflowExecutionFailed,\\n  ExternalWorkflowExecutionSignaled,\\n  UpsertWorkflowSearchAttributes,\\n}\\n\\nenum DecisionTaskFailedCause {\\n  UNHANDLED_DECISION,\\n  BAD_SCHEDULE_ACTIVITY_ATTRIBUTES,\\n  BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES,\\n  BAD_START_TIMER_ATTRIBUTES,\\n  BAD_CANCEL_TIMER_ATTRIBUTES,\\n  BAD_RECORD_MARKER_ATTRIBUTES,\\n  BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES,\\n  BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES,\\n  BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES,\\n  BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES,\\n  BAD_CONTINUE_AS_NEW_ATTRIBUTES,\\n  START_TIMER_DUPLICATE_ID,\\n  RESET_STICKY_TASKLIST,\\n  WORKFLOW_WORKER_UNHANDLED_FAILURE,\\n  BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES,\\n  BAD_START_CHILD_EXECUTION_ATTRIBUTES,\\n  FORCE_CLOSE_DECISION,\\n  FAILOVER_CLOSE_DECISION,\\n  BAD_SIGNAL_INPUT_SIZE,\\n  RESET_WORKFLOW,\\n  BAD_BINARY,\\n  SCHEDULE_ACTIVITY_DUPLICATE_ID,\\n  BAD_SEARCH_ATTRIBUTES,\\n}\\n\\nenum DecisionTaskTimedOutCause {\\n  TIMEOUT,\\n  RESET,\\n}\\n\\nenum CancelExternalWorkflowExecutionFailedCause {\\n  UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\\n  WORKFLOW_ALREADY_COMPLETED,\\n}\\n\\nenum SignalExternalWorkflowExecutionFailedCause {\\n  UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\\n  WORKFLOW_ALREADY_COMPLETED,\\n}\\n\\nenum ChildWorkflowExecutionFailedCause {\\n  WORKFLOW_ALREADY_RUNNING,\\n}\\n\\n// TODO: when migrating to gRPC, add a running / none status,\\n//  currently, customer is using null / nil as an indication\\n//  that workflow is still running\\nenum WorkflowExecutionCloseStatus {\\n  COMPLETED,\\n  FAILED,\\n  CANCELED,\\n  TERMINATED,\\n  CONTINUED_AS_NEW,\\n  TIMED_OUT,\\n}\\n\\nenum WorkflowExecutionStatus {\\n  PENDING,\\n  STARTED,\\n  COMPLETED,\\n  FAILED,\\n  CANCELED,\\n  TERMINATED,\\n  CONTINUED_AS_NEW,\\n  TIMED_OUT,\\n}\\n\\nenum QueryTaskCompletedType {\\n  COMPLETED,\\n  FAILED,\\n}\\n\\nenum QueryResultType {\\n  ANSWERED,\\n  FAILED,\\n}\\n\\nenum PendingActivityState {\\n  SCHEDULED,\\n  STARTED,\\n  CANCEL_REQUESTED,\\n}\\n\\nenum PendingDecisionState {\\n  SCHEDULED,\\n  STARTED,\\n}\\n\\nenum HistoryEventFilterType {\\n  ALL_EVENT,\\n  CLOSE_EVENT,\\n}\\n\\nenum TaskListKind {\\n  NORMAL,\\n  STICKY,\\n  EPHEMERAL,\\n}\\n\\nenum ArchivalStatus {\\n  DISABLED,\\n  ENABLED,\\n}\\n\\nenum CronOverlapPolicy {\\n  SKIPPED,\\n  BUFFERONE,\\n}\\n\\nenum IndexedValueType {\\n  STRING,\\n  KEYWORD,\\n  INT,\\n  DOUBLE,\\n  BOOL,\\n  DATETIME,\\n}\\n\\nstruct Header {\\n    10: optional map<string, binary> fields\\n}\\n\\nstruct WorkflowType {\\n  10: optional string name\\n}\\n\\nstruct ActivityType {\\n  10: optional string name\\n}\\n\\nstruct TaskList {\\n  10: optional string name\\n  20: optional TaskListKind kind\\n}\\n\\nenum EncodingType {\\n  ThriftRW,\\n  JSON,\\n}\\n\\nenum QueryRejectCondition {\\n  // NOT_OPEN indicates that query should be rejected if workflow is not open\\n  NOT_OPEN\\n  // NOT_COMPLETED_CLEANLY indicates that query should be rejected if workflow did not complete cleanly\\n  NOT_COMPLETED_CLEANLY\\n}\\n\\nenum QueryConsistencyLevel {\\n  // EVENTUAL indicates that query should be eventually consistent\\n  EVENTUAL\\n  // STRONG indicates that any events that came before query should be reflected in workflow state before running query\\n  STRONG\\n}\\n\\nstruct DataBlob {\\n  10: optional EncodingType EncodingType\\n  20: optional binary Data\\n}\\n\\nstruct TaskListMetadata {\\n  10: optional double maxTasksPerSecond\\n}\\n\\nstruct WorkflowExecution {\\n  10: optional string workflowId\\n  20: optional string runId\\n}\\n\\nstruct Memo {\\n  10: optional map<string,binary> fields\\n}\\n\\nstruct SearchAttributes {\\n  10: optional map<string,binary> indexedFields\\n}\\n\\nstruct WorkerVersionInfo {\\n  10: optional string impl\\n  20: optional string featureVersion\\n}\\n\\nstruct WorkflowExecutionInfo {\\n  10: optional WorkflowExecution execution\\n  20: optional WorkflowType type\\n  30: optional i64 (js.type = \\\"Long\\\") startTime\\n  40: optional i64 (js.type = \\\"Long\\\") closeTime\\n  50: optional WorkflowExecutionCloseStatus closeStatus\\n  60: optional i64 (js.type = \\\"Long\\\") historyLength\\n  70: optional string parentDomainId\\n  71: optional string parentDomainName\\n  72: optional i64 parentInitatedId\\n  80: optional WorkflowExecution parentExecution\\n  90: optional i64 (js.type = \\\"Long\\\") executionTime\\n  100: optional Memo memo\\n  101: optional SearchAttributes searchAttributes\\n  110: optional ResetPoints autoResetPoints\\n  120: optional string taskList\\n  121: optional TaskList taskListInfo\\n  130: optional bool isCron\\n  140: optional i64 (js.type = \\\"Long\\\") updateTime\\n  150: optional map<string, string> partitionConfig\\n  160: optional CronOverlapPolicy cronOverlapPolicy\\n  170: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n  180: optional string cronSchedule\\n  190: optional WorkflowExecutionStatus executionStatus\\n  200: optional i64 (js.type = \\\"Long\\\") scheduledExecutionTime\\n}\\n\\nstruct WorkflowExecutionConfiguration {\\n  10: optional TaskList taskList\\n  20: optional i32 executionStartToCloseTimeoutSeconds\\n  30: optional i32 taskStartToCloseTimeoutSeconds\\n//  40: optional ChildPolicy childPolicy -- Removed but reserve the IDL order number\\n}\\n\\nstruct TransientDecisionInfo {\\n  10: optional HistoryEvent scheduledEvent\\n  20: optional HistoryEvent startedEvent\\n}\\n\\nstruct ScheduleActivityTaskDecisionAttributes {\\n  10: optional string activityId\\n  20: optional ActivityType activityType\\n  25: optional string domain\\n  30: optional TaskList taskList\\n  40: optional binary input\\n  45: optional i32 scheduleToCloseTimeoutSeconds\\n  50: optional i32 scheduleToStartTimeoutSeconds\\n  55: optional i32 startToCloseTimeoutSeconds\\n  60: optional i32 heartbeatTimeoutSeconds\\n  70: optional RetryPolicy retryPolicy\\n  80: optional Header header\\n  90: optional bool requestLocalDispatch\\n}\\n\\nstruct ActivityLocalDispatchInfo{\\n  10: optional string activityId\\n  20: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  30: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  40: optional i64 (js.type = \\\"Long\\\") scheduledTimestampOfThisAttempt\\n  50: optional binary taskToken\\n}\\n\\nstruct RequestCancelActivityTaskDecisionAttributes {\\n  10: optional string activityId\\n}\\n\\nstruct StartTimerDecisionAttributes {\\n  10: optional string timerId\\n  20: optional i64 (js.type = \\\"Long\\\") startToFireTimeoutSeconds\\n}\\n\\nstruct CompleteWorkflowExecutionDecisionAttributes {\\n  10: optional binary result\\n}\\n\\nstruct FailWorkflowExecutionDecisionAttributes {\\n  10: optional string reason\\n  20: optional binary details\\n}\\n\\nstruct CancelTimerDecisionAttributes {\\n  10: optional string timerId\\n}\\n\\nstruct CancelWorkflowExecutionDecisionAttributes {\\n  10: optional binary details\\n}\\n\\nstruct RequestCancelExternalWorkflowExecutionDecisionAttributes {\\n  10: optional string domain\\n  20: optional string workflowId\\n  30: optional string runId\\n  40: optional binary control\\n  50: optional bool childWorkflowOnly\\n}\\n\\nstruct SignalExternalWorkflowExecutionDecisionAttributes {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n  30: optional string signalName\\n  40: optional binary input\\n  50: optional binary control\\n  60: optional bool childWorkflowOnly\\n}\\n\\nstruct UpsertWorkflowSearchAttributesDecisionAttributes {\\n  10: optional SearchAttributes searchAttributes\\n}\\n\\nstruct RecordMarkerDecisionAttributes {\\n  10: optional string markerName\\n  20: optional binary details\\n  30: optional Header header\\n}\\n\\nstruct ContinueAsNewWorkflowExecutionDecisionAttributes {\\n  10: optional WorkflowType workflowType\\n  20: optional TaskList taskList\\n  30: optional binary input\\n  40: optional i32 executionStartToCloseTimeoutSeconds\\n  50: optional i32 taskStartToCloseTimeoutSeconds\\n  60: optional i32 backoffStartIntervalInSeconds\\n  70: optional RetryPolicy retryPolicy\\n  80: optional ContinueAsNewInitiator initiator\\n  90: optional string failureReason\\n  100: optional binary failureDetails\\n  110: optional binary lastCompletionResult\\n  120: optional string cronSchedule\\n  130: optional Header header\\n  140: optional Memo memo\\n  150: optional SearchAttributes searchAttributes\\n  160: optional i32 jitterStartSeconds\\n  170: optional CronOverlapPolicy cronOverlapPolicy\\n  180: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct StartChildWorkflowExecutionDecisionAttributes {\\n  10: optional string domain\\n  20: optional string workflowId\\n  30: optional WorkflowType workflowType\\n  40: optional TaskList taskList\\n  50: optional binary input\\n  60: optional i32 executionStartToCloseTimeoutSeconds\\n  70: optional i32 taskStartToCloseTimeoutSeconds\\n//  80: optional ChildPolicy childPolicy -- Removed but reserve the IDL order number\\n  81: optional ParentClosePolicy parentClosePolicy\\n  90: optional binary control\\n  100: optional WorkflowIdReusePolicy workflowIdReusePolicy\\n  110: optional RetryPolicy retryPolicy\\n  120: optional string cronSchedule\\n  130: optional Header header\\n  140: optional Memo memo\\n  150: optional SearchAttributes searchAttributes\\n  160: optional CronOverlapPolicy cronOverlapPolicy\\n  170: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct Decision {\\n  10:  optional DecisionType decisionType\\n  20:  optional ScheduleActivityTaskDecisionAttributes scheduleActivityTaskDecisionAttributes\\n  25:  optional StartTimerDecisionAttributes startTimerDecisionAttributes\\n  30:  optional CompleteWorkflowExecutionDecisionAttributes completeWorkflowExecutionDecisionAttributes\\n  35:  optional FailWorkflowExecutionDecisionAttributes failWorkflowExecutionDecisionAttributes\\n  40:  optional RequestCancelActivityTaskDecisionAttributes requestCancelActivityTaskDecisionAttributes\\n  50:  optional CancelTimerDecisionAttributes cancelTimerDecisionAttributes\\n  60:  optional CancelWorkflowExecutionDecisionAttributes cancelWorkflowExecutionDecisionAttributes\\n  70:  optional RequestCancelExternalWorkflowExecutionDecisionAttributes requestCancelExternalWorkflowExecutionDecisionAttributes\\n  80:  optional RecordMarkerDecisionAttributes recordMarkerDecisionAttributes\\n  90:  optional ContinueAsNewWorkflowExecutionDecisionAttributes continueAsNewWorkflowExecutionDecisionAttributes\\n  100: optional StartChildWorkflowExecutionDecisionAttributes startChildWorkflowExecutionDecisionAttributes\\n  110: optional SignalExternalWorkflowExecutionDecisionAttributes signalExternalWorkflowExecutionDecisionAttributes\\n  120: optional UpsertWorkflowSearchAttributesDecisionAttributes upsertWorkflowSearchAttributesDecisionAttributes\\n}\\n\\nstruct WorkflowExecutionStartedEventAttributes {\\n  10: optional WorkflowType workflowType\\n  12: optional string parentWorkflowDomain\\n  14: optional WorkflowExecution parentWorkflowExecution\\n  16: optional i64 (js.type = \\\"Long\\\") parentInitiatedEventId\\n  20: optional TaskList taskList\\n  30: optional binary input\\n  40: optional i32 executionStartToCloseTimeoutSeconds\\n  50: optional i32 taskStartToCloseTimeoutSeconds\\n//  52: optional ChildPolicy childPolicy -- Removed but reserve the IDL order number\\n  54: optional string continuedExecutionRunId\\n  55: optional ContinueAsNewInitiator initiator\\n  56: optional string continuedFailureReason\\n  57: optional binary continuedFailureDetails\\n  58: optional binary lastCompletionResult\\n  59: optional string originalExecutionRunId // This is the runID when the WorkflowExecutionStarted event is written\\n  60: optional string identity\\n  61: optional string firstExecutionRunId // This is the very first runID along the chain of ContinueAsNew and Reset.\\n  62: optional i64 (js.type = \\\"Long\\\") firstScheduledTimeNano\\n  70: optional RetryPolicy retryPolicy\\n  80: optional i32 attempt\\n  90: optional i64 (js.type = \\\"Long\\\") expirationTimestamp\\n  100: optional string cronSchedule\\n  110: optional i32 firstDecisionTaskBackoffSeconds\\n  120: optional Memo memo\\n  121: optional SearchAttributes searchAttributes\\n  130: optional ResetPoints prevAutoResetPoints\\n  140: optional Header header\\n  150: optional map<string, string> partitionConfig\\n  160: optional string requestId\\n  170: optional CronOverlapPolicy cronOverlapPolicy\\n  180: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct ResetPoints{\\n  10: optional list<ResetPointInfo> points\\n}\\n\\n struct ResetPointInfo{\\n  10: optional string binaryChecksum\\n  20: optional string runId\\n  30: optional i64 firstDecisionCompletedId\\n  40: optional i64 (js.type = \\\"Long\\\") createdTimeNano\\n  50: optional i64 (js.type = \\\"Long\\\") expiringTimeNano //the time that the run is deleted due to retention\\n  60: optional bool resettable                         // false if the resset point has pending childWFs/reqCancels/signalExternals.\\n}\\n\\nstruct WorkflowExecutionCompletedEventAttributes {\\n  10: optional binary result\\n  20: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct WorkflowExecutionFailedEventAttributes {\\n  10: optional string reason\\n  20: optional binary details\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct WorkflowExecutionTimedOutEventAttributes {\\n  10: optional TimeoutType timeoutType\\n}\\n\\nenum ContinueAsNewInitiator {\\n  Decider,\\n  RetryPolicy,\\n  CronSchedule,\\n}\\n\\nstruct WorkflowExecutionContinuedAsNewEventAttributes {\\n  10: optional string newExecutionRunId\\n  20: optional WorkflowType workflowType\\n  30: optional TaskList taskList\\n  40: optional binary input\\n  50: optional i32 executionStartToCloseTimeoutSeconds\\n  60: optional i32 taskStartToCloseTimeoutSeconds\\n  70: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  80: optional i32 backoffStartIntervalInSeconds\\n  90: optional ContinueAsNewInitiator initiator\\n  100: optional string failureReason\\n  110: optional binary failureDetails\\n  120: optional binary lastCompletionResult\\n  130: optional Header header\\n  140: optional Memo memo\\n  150: optional SearchAttributes searchAttributes\\n  160: optional CronOverlapPolicy cronOverlapPolicy\\n  170: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct DecisionTaskScheduledEventAttributes {\\n  10: optional TaskList taskList\\n  20: optional i32 startToCloseTimeoutSeconds\\n  30: optional i64 (js.type = \\\"Long\\\") attempt\\n}\\n\\nstruct DecisionTaskStartedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  20: optional string identity\\n  30: optional string requestId\\n}\\n\\nstruct DecisionTaskCompletedEventAttributes {\\n  10: optional binary executionContext\\n  20: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  30: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  40: optional string identity\\n  50: optional string binaryChecksum\\n}\\n\\nstruct DecisionTaskTimedOutEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  20: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  30: optional TimeoutType timeoutType\\n  // for reset workflow\\n  40: optional string baseRunId\\n  50: optional string newRunId\\n  60: optional i64 (js.type = \\\"Long\\\") forkEventVersion\\n  70: optional string reason\\n  80: optional DecisionTaskTimedOutCause cause\\n  90: optional string requestId\\n}\\n\\nstruct DecisionTaskFailedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  20: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  30: optional DecisionTaskFailedCause cause\\n  35: optional binary details\\n  40: optional string identity\\n  50: optional string reason\\n  // for reset workflow\\n  60: optional string baseRunId\\n  70: optional string newRunId\\n  80: optional i64 (js.type = \\\"Long\\\") forkEventVersion\\n  90: optional string binaryChecksum\\n  100: optional string requestId\\n}\\n\\nstruct ActivityTaskScheduledEventAttributes {\\n  10: optional string activityId\\n  20: optional ActivityType activityType\\n  25: optional string domain\\n  30: optional TaskList taskList\\n  40: optional binary input\\n  45: optional i32 scheduleToCloseTimeoutSeconds\\n  50: optional i32 scheduleToStartTimeoutSeconds\\n  55: optional i32 startToCloseTimeoutSeconds\\n  60: optional i32 heartbeatTimeoutSeconds\\n  90: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  110: optional RetryPolicy retryPolicy\\n  120: optional Header header\\n}\\n\\nstruct ActivityTaskStartedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  20: optional string identity\\n  30: optional string requestId\\n  40: optional i32 attempt\\n  50: optional string lastFailureReason\\n  60: optional binary lastFailureDetails\\n}\\n\\nstruct ActivityTaskCompletedEventAttributes {\\n  10: optional binary result\\n  20: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  30: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  40: optional string identity\\n}\\n\\nstruct ActivityTaskFailedEventAttributes {\\n  10: optional string reason\\n  20: optional binary details\\n  30: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  40: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  50: optional string identity\\n}\\n\\nstruct ActivityTaskTimedOutEventAttributes {\\n  05: optional binary details\\n  10: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  20: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  30: optional TimeoutType timeoutType\\n  // For retry activity, it may have a failure before timeout. It's important to keep those information for debug.\\n  // Client can also provide the info for making next decision\\n  40: optional string lastFailureReason\\n  50: optional binary lastFailureDetails\\n}\\n\\nstruct ActivityTaskCancelRequestedEventAttributes {\\n  10: optional string activityId\\n  20: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct RequestCancelActivityTaskFailedEventAttributes{\\n  10: optional string activityId\\n  20: optional string cause\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct ActivityTaskCanceledEventAttributes {\\n  10: optional binary details\\n  20: optional i64 (js.type = \\\"Long\\\") latestCancelRequestedEventId\\n  30: optional i64 (js.type = \\\"Long\\\") scheduledEventId\\n  40: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  50: optional string identity\\n}\\n\\nstruct TimerStartedEventAttributes {\\n  10: optional string timerId\\n  20: optional i64 (js.type = \\\"Long\\\") startToFireTimeoutSeconds\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct TimerFiredEventAttributes {\\n  10: optional string timerId\\n  20: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct TimerCanceledEventAttributes {\\n  10: optional string timerId\\n  20: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  40: optional string identity\\n}\\n\\nstruct CancelTimerFailedEventAttributes {\\n  10: optional string timerId\\n  20: optional string cause\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  40: optional string identity\\n}\\n\\nstruct WorkflowExecutionCancelRequestedEventAttributes {\\n  10: optional string cause\\n  20: optional i64 (js.type = \\\"Long\\\") externalInitiatedEventId\\n  30: optional WorkflowExecution externalWorkflowExecution\\n  40: optional string identity\\n  50: optional string requestId\\n}\\n\\nstruct WorkflowExecutionCanceledEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  20: optional binary details\\n}\\n\\nstruct MarkerRecordedEventAttributes {\\n  10: optional string markerName\\n  20: optional binary details\\n  30: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  40: optional Header header\\n}\\n\\nstruct WorkflowExecutionSignaledEventAttributes {\\n  10: optional string signalName\\n  20: optional binary input\\n  30: optional string identity\\n  40: optional string requestId\\n}\\n\\nstruct WorkflowExecutionTerminatedEventAttributes {\\n  10: optional string reason\\n  20: optional binary details\\n  30: optional string identity\\n}\\n\\nstruct RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional binary control\\n  50: optional bool childWorkflowOnly\\n}\\n\\nstruct RequestCancelExternalWorkflowExecutionFailedEventAttributes {\\n  10: optional CancelExternalWorkflowExecutionFailedCause cause\\n  20: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  30: optional string domain\\n  40: optional WorkflowExecution workflowExecution\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  60: optional binary control\\n}\\n\\nstruct ExternalWorkflowExecutionCancelRequestedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n}\\n\\nstruct SignalExternalWorkflowExecutionInitiatedEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional string signalName\\n  50: optional binary input\\n  60: optional binary control\\n  70: optional bool childWorkflowOnly\\n}\\n\\nstruct SignalExternalWorkflowExecutionFailedEventAttributes {\\n  10: optional SignalExternalWorkflowExecutionFailedCause cause\\n  20: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  30: optional string domain\\n  40: optional WorkflowExecution workflowExecution\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  60: optional binary control\\n}\\n\\nstruct ExternalWorkflowExecutionSignaledEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional binary control\\n}\\n\\nstruct UpsertWorkflowSearchAttributesEventAttributes {\\n  10: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  20: optional SearchAttributes searchAttributes\\n}\\n\\nstruct StartChildWorkflowExecutionInitiatedEventAttributes {\\n  10:  optional string domain\\n  20:  optional string workflowId\\n  30:  optional WorkflowType workflowType\\n  40:  optional TaskList taskList\\n  50:  optional binary input\\n  60:  optional i32 executionStartToCloseTimeoutSeconds\\n  70:  optional i32 taskStartToCloseTimeoutSeconds\\n//  80:  optional ChildPolicy childPolicy -- Removed but reserve the IDL order number\\n  81:  optional ParentClosePolicy parentClosePolicy\\n  90:  optional binary control\\n  100: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n  110: optional WorkflowIdReusePolicy workflowIdReusePolicy\\n  120: optional RetryPolicy retryPolicy\\n  130: optional string cronSchedule\\n  140: optional Header header\\n  150: optional Memo memo\\n  160: optional SearchAttributes searchAttributes\\n  170: optional i32 delayStartSeconds\\n  180: optional i32 jitterStartSeconds\\n  190: optional i64 (js.type = \\\"Long\\\") firstRunAtTimestamp\\n  200: optional CronOverlapPolicy cronOverlapPolicy\\n  210: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct StartChildWorkflowExecutionFailedEventAttributes {\\n  10: optional string domain\\n  20: optional string workflowId\\n  30: optional WorkflowType workflowType\\n  40: optional ChildWorkflowExecutionFailedCause cause\\n  50: optional binary control\\n  60: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  70: optional i64 (js.type = \\\"Long\\\") decisionTaskCompletedEventId\\n}\\n\\nstruct ChildWorkflowExecutionStartedEventAttributes {\\n  10: optional string domain\\n  20: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional WorkflowType workflowType\\n  50: optional Header header\\n}\\n\\nstruct ChildWorkflowExecutionCompletedEventAttributes {\\n  10: optional binary result\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional WorkflowType workflowType\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  60: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct ChildWorkflowExecutionFailedEventAttributes {\\n  10: optional string reason\\n  20: optional binary details\\n  30: optional string domain\\n  40: optional WorkflowExecution workflowExecution\\n  50: optional WorkflowType workflowType\\n  60: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  70: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct ChildWorkflowExecutionCanceledEventAttributes {\\n  10: optional binary details\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional WorkflowType workflowType\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  60: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct ChildWorkflowExecutionTimedOutEventAttributes {\\n  10: optional TimeoutType timeoutType\\n  20: optional string domain\\n  30: optional WorkflowExecution workflowExecution\\n  40: optional WorkflowType workflowType\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  60: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct ChildWorkflowExecutionTerminatedEventAttributes {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional WorkflowType workflowType\\n  40: optional i64 (js.type = \\\"Long\\\") initiatedEventId\\n  50: optional i64 (js.type = \\\"Long\\\") startedEventId\\n}\\n\\nstruct HistoryEvent {\\n  10:  optional i64 (js.type = \\\"Long\\\") eventId\\n  20:  optional i64 (js.type = \\\"Long\\\") timestamp\\n  30:  optional EventType eventType\\n  35:  optional i64 (js.type = \\\"Long\\\") version\\n  36:  optional i64 (js.type = \\\"Long\\\") taskId\\n  40:  optional WorkflowExecutionStartedEventAttributes workflowExecutionStartedEventAttributes\\n  50:  optional WorkflowExecutionCompletedEventAttributes workflowExecutionCompletedEventAttributes\\n  60:  optional WorkflowExecutionFailedEventAttributes workflowExecutionFailedEventAttributes\\n  70:  optional WorkflowExecutionTimedOutEventAttributes workflowExecutionTimedOutEventAttributes\\n  80:  optional DecisionTaskScheduledEventAttributes decisionTaskScheduledEventAttributes\\n  90:  optional DecisionTaskStartedEventAttributes decisionTaskStartedEventAttributes\\n  100: optional DecisionTaskCompletedEventAttributes decisionTaskCompletedEventAttributes\\n  110: optional DecisionTaskTimedOutEventAttributes decisionTaskTimedOutEventAttributes\\n  120: optional DecisionTaskFailedEventAttributes decisionTaskFailedEventAttributes\\n  130: optional ActivityTaskScheduledEventAttributes activityTaskScheduledEventAttributes\\n  140: optional ActivityTaskStartedEventAttributes activityTaskStartedEventAttributes\\n  150: optional ActivityTaskCompletedEventAttributes activityTaskCompletedEventAttributes\\n  160: optional ActivityTaskFailedEventAttributes activityTaskFailedEventAttributes\\n  170: optional ActivityTaskTimedOutEventAttributes activityTaskTimedOutEventAttributes\\n  180: optional TimerStartedEventAttributes timerStartedEventAttributes\\n  190: optional TimerFiredEventAttributes timerFiredEventAttributes\\n  200: optional ActivityTaskCancelRequestedEventAttributes activityTaskCancelRequestedEventAttributes\\n  210: optional RequestCancelActivityTaskFailedEventAttributes requestCancelActivityTaskFailedEventAttributes\\n  220: optional ActivityTaskCanceledEventAttributes activityTaskCanceledEventAttributes\\n  230: optional TimerCanceledEventAttributes timerCanceledEventAttributes\\n  240: optional CancelTimerFailedEventAttributes cancelTimerFailedEventAttributes\\n  250: optional MarkerRecordedEventAttributes markerRecordedEventAttributes\\n  260: optional WorkflowExecutionSignaledEventAttributes workflowExecutionSignaledEventAttributes\\n  270: optional WorkflowExecutionTerminatedEventAttributes workflowExecutionTerminatedEventAttributes\\n  280: optional WorkflowExecutionCancelRequestedEventAttributes workflowExecutionCancelRequestedEventAttributes\\n  290: optional WorkflowExecutionCanceledEventAttributes workflowExecutionCanceledEventAttributes\\n  300: optional RequestCancelExternalWorkflowExecutionInitiatedEventAttributes requestCancelExternalWorkflowExecutionInitiatedEventAttributes\\n  310: optional RequestCancelExternalWorkflowExecutionFailedEventAttributes requestCancelExternalWorkflowExecutionFailedEventAttributes\\n  320: optional ExternalWorkflowExecutionCancelRequestedEventAttributes externalWorkflowExecutionCancelRequestedEventAttributes\\n  330: optional WorkflowExecutionContinuedAsNewEventAttributes workflowExecutionContinuedAsNewEventAttributes\\n  340: optional StartChildWorkflowExecutionInitiatedEventAttributes startChildWorkflowExecutionInitiatedEventAttributes\\n  350: optional StartChildWorkflowExecutionFailedEventAttributes startChildWorkflowExecutionFailedEventAttributes\\n  360: optional ChildWorkflowExecutionStartedEventAttributes childWorkflowExecutionStartedEventAttributes\\n  370: optional ChildWorkflowExecutionCompletedEventAttributes childWorkflowExecutionCompletedEventAttributes\\n  380: optional ChildWorkflowExecutionFailedEventAttributes childWorkflowExecutionFailedEventAttributes\\n  390: optional ChildWorkflowExecutionCanceledEventAttributes childWorkflowExecutionCanceledEventAttributes\\n  400: optional ChildWorkflowExecutionTimedOutEventAttributes childWorkflowExecutionTimedOutEventAttributes\\n  410: optional ChildWorkflowExecutionTerminatedEventAttributes childWorkflowExecutionTerminatedEventAttributes\\n  420: optional SignalExternalWorkflowExecutionInitiatedEventAttributes signalExternalWorkflowExecutionInitiatedEventAttributes\\n  430: optional SignalExternalWorkflowExecutionFailedEventAttributes signalExternalWorkflowExecutionFailedEventAttributes\\n  440: optional ExternalWorkflowExecutionSignaledEventAttributes externalWorkflowExecutionSignaledEventAttributes\\n  450: optional UpsertWorkflowSearchAttributesEventAttributes upsertWorkflowSearchAttributesEventAttributes\\n}\\n\\nstruct History {\\n  10: optional list<HistoryEvent> events\\n}\\n\\nstruct WorkflowExecutionFilter {\\n  10: optional string workflowId\\n  20: optional string runId\\n}\\n\\nstruct WorkflowTypeFilter {\\n  10: optional string name\\n}\\n\\nstruct StartTimeFilter {\\n  10: optional i64 (js.type = \\\"Long\\\") earliestTime\\n  20: optional i64 (js.type = \\\"Long\\\") latestTime\\n}\\n\\nstruct DomainInfo {\\n  10: optional string name\\n  20: optional DomainStatus status\\n  30: optional string description\\n  40: optional string ownerEmail\\n  // A key-value map for any customized purpose\\n  50: optional map<string,string> data\\n  60: optional string uuid\\n}\\n\\nstruct DomainConfiguration {\\n  10: optional i32 workflowExecutionRetentionPeriodInDays\\n  20: optional bool emitMetric\\n  60: optional IsolationGroupConfiguration isolationgroups\\n  70: optional BadBinaries badBinaries\\n  80: optional ArchivalStatus historyArchivalStatus\\n  90: optional string historyArchivalURI\\n  100: optional ArchivalStatus visibilityArchivalStatus\\n  110: optional string visibilityArchivalURI\\n  120: optional AsyncWorkflowConfiguration AsyncWorkflowConfiguration\\n}\\n\\nstruct FailoverInfo {\\n    10: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n    20: optional i64 (js.type = \\\"Long\\\") failoverStartTimestamp\\n    30: optional i64 (js.type = \\\"Long\\\") failoverExpireTimestamp\\n    40: optional i32 completedShardCount\\n    50: optional list<i32> pendingShards\\n}\\n\\nstruct BadBinaries{\\n  10: optional map<string, BadBinaryInfo> binaries\\n}\\n\\nstruct BadBinaryInfo{\\n  10: optional string reason\\n  20: optional string operator\\n  30: optional i64 (js.type = \\\"Long\\\") createdTimeNano\\n}\\n\\nstruct UpdateDomainInfo {\\n  10: optional string description\\n  20: optional string ownerEmail\\n  // A key-value map for any customized purpose\\n  30: optional map<string,string> data\\n}\\n\\nstruct ClusterReplicationConfiguration {\\n 10: optional string clusterName\\n}\\n\\nstruct DomainReplicationConfiguration {\\n // activeClusterName is the name of the active cluster for active-passive domain\\n 10: optional string activeClusterName\\n\\n //  clusters is list of all active and passive clusters of domain\\n 20: optional list<ClusterReplicationConfiguration> clusters\\n\\n // activeClusters contains active cluster(s) information for active-active domain\\n 30: optional ActiveClusters activeClusters\\n}\\n\\n// ClusterAttributeScope is a mapping of the cluster atribute to the scope's\\n// current stae and failover version, indicating how recently the change was made\\nstruct ClusterAttributeScope {\\n  10: optional map<string, ActiveClusterInfo> clusterAttributes;\\n}\\n\\n// activeClustersByClusterAttribute is a map of whatever subdivision of the domain chosen\\n// to active cluster info for active-active domains. The key refers to the type of\\n// cluster attribute and the value refers to its cluster mappings.\\n//\\n// For example, a request to update the domain for two locations\\n//\\n// UpdateDomainRequest{\\n//    ReplicationConfiguration: {\\n//       ActiveClusters: {\\n//           ActiveClustersByClusterAttribute: {\\n//             \\\"location\\\": ClusterAttributeScope{\\n//                   \\\"Tokyo\\\": {ActiveClusterInfo: \\\"cluster0, FailoverVersion: 123},\\n//                   \\\"Morocco\\\": {ActiveClusterInfo: \\\"cluster1\\\", FailoverVersion: 100},\\n//             }\\n//          }\\n//       }\\n//    }\\n//  }\\nstruct ActiveClusters {\\n  10: optional map<string, ActiveClusterInfo> activeClustersByRegion // todo (david.porter) remove this as it's no longer used\\n  11: optional map<string, ClusterAttributeScope> activeClustersByClusterAttribute\\n}\\n\\n// ActiveClusterInfo contains the configuration of active-active domain's active\\n// cluster & failover version for a specific region\\nstruct ActiveClusterInfo {\\n  10: optional string activeClusterName\\n  20: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n}\\n\\nstruct RegisterDomainRequest {\\n  10: optional string name\\n  20: optional string description\\n  30: optional string ownerEmail\\n  40: optional i32 workflowExecutionRetentionPeriodInDays\\n  50: optional bool emitMetric = true\\n  60: optional list<ClusterReplicationConfiguration> clusters\\n  70: optional string activeClusterName\\n  // todo (david.porter) remove this field as it's not going to be used\\n  75: optional map<string, string> activeClustersByRegion\\n  // activeClusters is a map of cluster-attribute name to active cluster name for active-active domain\\n  76: optional ActiveClusters activeClusters\\n  // A key-value map for any customized purpose\\n  80: optional map<string,string> data\\n  90: optional string securityToken\\n  120: optional bool isGlobalDomain\\n  130: optional ArchivalStatus historyArchivalStatus\\n  140: optional string historyArchivalURI\\n  150: optional ArchivalStatus visibilityArchivalStatus\\n  160: optional string visibilityArchivalURI\\n}\\n\\nstruct ListDomainsRequest {\\n  10: optional i32 pageSize\\n  20: optional binary nextPageToken\\n}\\n\\nstruct ListDomainsResponse {\\n  10: optional list<DescribeDomainResponse> domains\\n  20: optional binary nextPageToken\\n}\\n\\nstruct DescribeDomainRequest {\\n  10: optional string name\\n  20: optional string uuid\\n}\\n\\nstruct DescribeDomainResponse {\\n  10: optional DomainInfo domainInfo\\n  20: optional DomainConfiguration configuration\\n  30: optional DomainReplicationConfiguration replicationConfiguration\\n  40: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n  50: optional bool isGlobalDomain\\n  60: optional FailoverInfo failoverInfo\\n}\\n\\nstruct UpdateDomainRequest {\\n 10: optional string name\\n 20: optional UpdateDomainInfo updatedInfo\\n 30: optional DomainConfiguration configuration\\n 40: optional DomainReplicationConfiguration replicationConfiguration\\n 50: optional string securityToken\\n 60: optional string deleteBadBinary\\n 70: optional i32 failoverTimeoutInSeconds\\n}\\n\\nstruct UpdateDomainResponse {\\n  10: optional DomainInfo domainInfo\\n  20: optional DomainConfiguration configuration\\n  30: optional DomainReplicationConfiguration replicationConfiguration\\n  40: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n  50: optional bool isGlobalDomain\\n}\\n\\nstruct FailoverDomainRequest {\\n 10: optional string domainName\\n 20: optional string domainActiveClusterName\\n // only applicable to active-active domains where\\n // specific cluster-attributes are being failed over\\n 30: optional ActiveClusters activeClusters\\n // user-requested addition \\\"reason\\\" variable created to increase transparency around failovers\\n 40: optional string reason\\n}\\n\\nstruct FailoverDomainResponse {\\n  10: optional DomainInfo domainInfo\\n  20: optional DomainConfiguration configuration\\n  30: optional DomainReplicationConfiguration replicationConfiguration\\n  40: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n  50: optional bool isGlobalDomain\\n}\\n\\nstruct DeprecateDomainRequest {\\n 10: optional string name\\n 20: optional string securityToken\\n}\\n\\nstruct DeleteDomainRequest {\\n 10: optional string name\\n 20: optional string securityToken\\n}\\n\\nstruct ListFailoverHistoryRequest {\\n  // ListFailoverHistoryRequestFilters specifies the filters to apply to the request.\\n  // If not provided all failover events will be returned.\\n  10: optional ListFailoverHistoryRequestFilters filters\\n  // PaginationOptions will be used to paginate the results.\\n  // If not provided the first 5 events will be returned.\\n  20: optional PaginationOptions pagination\\n}\\n\\n// ListFailoverHistoryRequestFilters is used to filter the failover history.\\n// It will be extended with additional filters (e.g ClusterAttributes) as the active-active feature is developed.\\nstruct ListFailoverHistoryRequestFilters {\\n  // domain_id is the id of the domain to list failover history for.\\n  10: optional string domainID\\n}\\n\\nstruct ListFailoverHistoryResponse {\\n  10: optional list<FailoverEvent> failoverEvents\\n  // next_page_token can be passed in a subsequent request to fetch the next set of events.\\n  20: optional binary nextPageToken\\n}\\n\\nstruct FailoverEvent {\\n  // id of the failover event\\n  // Can be passed with the created time to fetch a specific event.\\n  10: optional string id\\n  // created_time is the time the failover event was created.\\n  // Can be passed with the ID to fetch a specific event.\\n  20: optional i64 (js.type = \\\"Long\\\") createdTime\\n  30: optional FailoverType failoverType\\n  40: optional list<ClusterFailover> clusterFailovers\\n}\\n\\nstruct ClusterFailover {\\n  10: optional ActiveClusterInfo fromCluster\\n  20: optional ActiveClusterInfo toCluster\\n  // cluster_attribute is the scope and name for the attribute that was failed over.\\n  // If the cluster_attribute is not defined this failover can be assumed to be the default ActiveCluster.\\n  30: optional ClusterAttribute clusterAttribute\\n}\\n\\nstruct StartWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional string workflowId\\n  30: optional WorkflowType workflowType\\n  40: optional TaskList taskList\\n  50: optional binary input\\n  60: optional i32 executionStartToCloseTimeoutSeconds\\n  70: optional i32 taskStartToCloseTimeoutSeconds\\n  80: optional string identity\\n  90: optional string requestId\\n  100: optional WorkflowIdReusePolicy workflowIdReusePolicy\\n//  110: optional ChildPolicy childPolicy -- Removed but reserve the IDL order number\\n  120: optional RetryPolicy retryPolicy\\n  130: optional string cronSchedule\\n  140: optional Memo memo\\n  141: optional SearchAttributes searchAttributes\\n  150: optional Header header\\n  160: optional i32 delayStartSeconds\\n  170: optional i32 jitterStartSeconds\\n  180: optional i64 (js.type = \\\"Long\\\") firstRunAtTimestamp\\n  190: optional CronOverlapPolicy cronOverlapPolicy\\n  200: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct StartWorkflowExecutionResponse {\\n  10: optional string runId\\n}\\n\\nstruct StartWorkflowExecutionAsyncRequest {\\n  10: optional StartWorkflowExecutionRequest request\\n}\\n\\nstruct StartWorkflowExecutionAsyncResponse {\\n}\\n\\nstruct RestartWorkflowExecutionResponse {\\n  10: optional string runId\\n}\\n\\nstruct DiagnoseWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string identity\\n}\\n\\nstruct DiagnoseWorkflowExecutionResponse {\\n  10: optional string domain\\n  20: optional WorkflowExecution diagnosticWorkflowExecution\\n}\\n\\nstruct PollForDecisionTaskRequest {\\n  10: optional string domain\\n  20: optional TaskList taskList\\n  30: optional string identity\\n  40: optional string binaryChecksum\\n}\\n\\nstruct PollForDecisionTaskResponse {\\n  10: optional binary taskToken\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional WorkflowType workflowType\\n  40: optional i64 (js.type = \\\"Long\\\") previousStartedEventId\\n  50: optional i64 (js.type = \\\"Long\\\") startedEventId\\n  51: optional i64 (js.type = 'Long') attempt\\n  54: optional i64 (js.type = \\\"Long\\\") backlogCountHint\\n  60: optional History history\\n  70: optional binary nextPageToken\\n  80: optional WorkflowQuery query\\n  90: optional TaskList WorkflowExecutionTaskList\\n  100: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  110: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  120: optional map<string, WorkflowQuery> queries\\n  130: optional i64 (js.type = 'Long') nextEventId\\n  140: optional i64 (js.type = 'Long') totalHistoryBytes\\n  150: optional AutoConfigHint autoConfigHint\\n}\\n\\nstruct StickyExecutionAttributes {\\n  10: optional TaskList workerTaskList\\n  20: optional i32 scheduleToStartTimeoutSeconds\\n}\\n\\nstruct RespondDecisionTaskCompletedRequest {\\n  10: optional binary taskToken\\n  20: optional list<Decision> decisions\\n  30: optional binary executionContext\\n  40: optional string identity\\n  50: optional StickyExecutionAttributes stickyAttributes\\n  60: optional bool returnNewDecisionTask\\n  70: optional bool forceCreateNewDecisionTask\\n  80: optional string binaryChecksum\\n  90: optional map<string, WorkflowQueryResult> queryResults\\n}\\n\\nstruct RespondDecisionTaskCompletedResponse {\\n  10: optional PollForDecisionTaskResponse decisionTask\\n  20: optional map<string,ActivityLocalDispatchInfo> activitiesToDispatchLocally\\n}\\n\\nstruct RespondDecisionTaskFailedRequest {\\n  10: optional binary taskToken\\n  20: optional DecisionTaskFailedCause cause\\n  30: optional binary details\\n  40: optional string identity\\n  50: optional string binaryChecksum\\n}\\n\\nstruct PollForActivityTaskRequest {\\n  10: optional string domain\\n  20: optional TaskList taskList\\n  30: optional string identity\\n  40: optional TaskListMetadata taskListMetadata\\n}\\n\\nstruct PollForActivityTaskResponse {\\n  10:  optional binary taskToken\\n  20:  optional WorkflowExecution workflowExecution\\n  30:  optional string activityId\\n  40:  optional ActivityType activityType\\n  50:  optional binary input\\n  70:  optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  80:  optional i32 scheduleToCloseTimeoutSeconds\\n  90:  optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  100: optional i32 startToCloseTimeoutSeconds\\n  110: optional i32 heartbeatTimeoutSeconds\\n  120: optional i32 attempt\\n  130: optional i64 (js.type = \\\"Long\\\") scheduledTimestampOfThisAttempt\\n  140: optional binary heartbeatDetails\\n  150: optional WorkflowType workflowType\\n  160: optional string workflowDomain\\n  170: optional Header header\\n  180: optional AutoConfigHint autoConfigHint\\n}\\n\\nstruct RecordActivityTaskHeartbeatRequest {\\n  10: optional binary taskToken\\n  20: optional binary details\\n  30: optional string identity\\n}\\n\\nstruct RecordActivityTaskHeartbeatByIDRequest {\\n  10: optional string domain\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional string activityID\\n  50: optional binary details\\n  60: optional string identity\\n}\\n\\nstruct RecordActivityTaskHeartbeatResponse {\\n  10: optional bool cancelRequested\\n}\\n\\nstruct RespondActivityTaskCompletedRequest {\\n  10: optional binary taskToken\\n  20: optional binary result\\n  30: optional string identity\\n}\\n\\nstruct RespondActivityTaskFailedRequest {\\n  10: optional binary taskToken\\n  20: optional string reason\\n  30: optional binary details\\n  40: optional string identity\\n}\\n\\nstruct RespondActivityTaskCanceledRequest {\\n  10: optional binary taskToken\\n  20: optional binary details\\n  30: optional string identity\\n}\\n\\nstruct RespondActivityTaskCompletedByIDRequest {\\n  10: optional string domain\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional string activityID\\n  50: optional binary result\\n  60: optional string identity\\n}\\n\\nstruct RespondActivityTaskFailedByIDRequest {\\n  10: optional string domain\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional string activityID\\n  50: optional string reason\\n  60: optional binary details\\n  70: optional string identity\\n}\\n\\nstruct RespondActivityTaskCanceledByIDRequest {\\n  10: optional string domain\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional string activityID\\n  50: optional binary details\\n  60: optional string identity\\n}\\n\\nstruct RequestCancelWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string identity\\n  40: optional string requestId\\n  50: optional string cause\\n  60: optional string firstExecutionRunID\\n}\\n\\nstruct GetWorkflowExecutionHistoryRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n  30: optional i32 maximumPageSize\\n  40: optional binary nextPageToken\\n  50: optional bool waitForNewEvent\\n  60: optional HistoryEventFilterType HistoryEventFilterType\\n  70: optional bool skipArchival\\n  80: optional QueryConsistencyLevel queryConsistencyLevel\\n}\\n\\nstruct GetWorkflowExecutionHistoryResponse {\\n  10: optional History history\\n  11: optional list<DataBlob> rawHistory\\n  20: optional binary nextPageToken\\n  30: optional bool archived\\n}\\n\\nstruct SignalWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string signalName\\n  40: optional binary input\\n  50: optional string identity\\n  60: optional string requestId\\n  70: optional binary control\\n}\\n\\nstruct SignalWithStartWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional string workflowId\\n  30: optional WorkflowType workflowType\\n  40: optional TaskList taskList\\n  50: optional binary input\\n  60: optional i32 executionStartToCloseTimeoutSeconds\\n  70: optional i32 taskStartToCloseTimeoutSeconds\\n  80: optional string identity\\n  90: optional string requestId\\n  100: optional WorkflowIdReusePolicy workflowIdReusePolicy\\n  110: optional string signalName\\n  120: optional binary signalInput\\n  130: optional binary control\\n  140: optional RetryPolicy retryPolicy\\n  150: optional string cronSchedule\\n  160: optional Memo memo\\n  161: optional SearchAttributes searchAttributes\\n  170: optional Header header\\n  180: optional i32 delayStartSeconds\\n  190: optional i32 jitterStartSeconds\\n  200: optional i64 (js.type = \\\"Long\\\") firstRunAtTimestamp\\n  210: optional CronOverlapPolicy cronOverlapPolicy\\n  220: optional ActiveClusterSelectionPolicy activeClusterSelectionPolicy\\n}\\n\\nstruct SignalWithStartWorkflowExecutionAsyncRequest {\\n  10: optional SignalWithStartWorkflowExecutionRequest request\\n}\\n\\nstruct SignalWithStartWorkflowExecutionAsyncResponse {\\n}\\n\\nstruct RestartWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string reason\\n  40: optional string identity\\n}\\nstruct TerminateWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string reason\\n  40: optional binary details\\n  50: optional string identity\\n  60: optional string firstExecutionRunID\\n}\\n\\nstruct ResetWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional string reason\\n  40: optional i64 (js.type = \\\"Long\\\") decisionFinishEventId\\n  50: optional string requestId\\n  60: optional bool skipSignalReapply\\n}\\n\\nstruct ResetWorkflowExecutionResponse {\\n  10: optional string runId\\n}\\n\\nstruct ListOpenWorkflowExecutionsRequest {\\n  10: optional string domain\\n  20: optional i32 maximumPageSize\\n  30: optional binary nextPageToken\\n  40: optional StartTimeFilter StartTimeFilter\\n  50: optional WorkflowExecutionFilter executionFilter\\n  60: optional WorkflowTypeFilter typeFilter\\n}\\n\\nstruct ListOpenWorkflowExecutionsResponse {\\n  10: optional list<WorkflowExecutionInfo> executions\\n  20: optional binary nextPageToken\\n}\\n\\nstruct ListClosedWorkflowExecutionsRequest {\\n  10: optional string domain\\n  20: optional i32 maximumPageSize\\n  30: optional binary nextPageToken\\n  40: optional StartTimeFilter StartTimeFilter\\n  50: optional WorkflowExecutionFilter executionFilter\\n  60: optional WorkflowTypeFilter typeFilter\\n  70: optional WorkflowExecutionCloseStatus statusFilter\\n}\\n\\nstruct ListClosedWorkflowExecutionsResponse {\\n  10: optional list<WorkflowExecutionInfo> executions\\n  20: optional binary nextPageToken\\n}\\n\\nstruct ListWorkflowExecutionsRequest {\\n  10: optional string domain\\n  20: optional i32 pageSize\\n  30: optional binary nextPageToken\\n  40: optional string query\\n}\\n\\nstruct ListWorkflowExecutionsResponse {\\n  10: optional list<WorkflowExecutionInfo> executions\\n  20: optional binary nextPageToken\\n}\\n\\nstruct ListArchivedWorkflowExecutionsRequest {\\n  10: optional string domain\\n  20: optional i32 pageSize\\n  30: optional binary nextPageToken\\n  40: optional string query\\n}\\n\\nstruct ListArchivedWorkflowExecutionsResponse {\\n  10: optional list<WorkflowExecutionInfo> executions\\n  20: optional binary nextPageToken\\n}\\n\\nstruct CountWorkflowExecutionsRequest {\\n  10: optional string domain\\n  20: optional string query\\n}\\n\\nstruct CountWorkflowExecutionsResponse {\\n  10: optional i64 count\\n}\\n\\nstruct GetSearchAttributesResponse {\\n  10: optional map<string, IndexedValueType> keys\\n}\\n\\nstruct QueryWorkflowRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n  30: optional WorkflowQuery query\\n  // QueryRejectCondition can used to reject the query if workflow state does not satisify condition\\n  40: optional QueryRejectCondition queryRejectCondition\\n  50: optional QueryConsistencyLevel queryConsistencyLevel\\n}\\n\\nstruct QueryRejected {\\n  10: optional WorkflowExecutionCloseStatus closeStatus\\n}\\n\\nstruct QueryWorkflowResponse {\\n  10: optional binary queryResult\\n  20: optional QueryRejected queryRejected\\n}\\n\\nstruct WorkflowQuery {\\n  10: optional string queryType\\n  20: optional binary queryArgs\\n}\\n\\nstruct ResetStickyTaskListRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n}\\n\\nstruct ResetStickyTaskListResponse {\\n    // The reason to keep this response is to allow returning\\n    // information in the future.\\n}\\n\\nstruct RespondQueryTaskCompletedRequest {\\n  10: optional binary taskToken\\n  20: optional QueryTaskCompletedType completedType\\n  30: optional binary queryResult\\n  40: optional string errorMessage\\n  50: optional WorkerVersionInfo workerVersionInfo\\n}\\n\\nstruct WorkflowQueryResult {\\n  10: optional QueryResultType resultType\\n  20: optional binary answer\\n  30: optional string errorMessage\\n}\\n\\nstruct DescribeWorkflowExecutionRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n  30: optional QueryConsistencyLevel queryConsistencyLevel\\n}\\n\\nstruct PendingActivityInfo {\\n  10: optional string activityID\\n  20: optional ActivityType activityType\\n  30: optional PendingActivityState state\\n  40: optional binary heartbeatDetails\\n  50: optional i64 (js.type = \\\"Long\\\") lastHeartbeatTimestamp\\n  60: optional i64 (js.type = \\\"Long\\\") lastStartedTimestamp\\n  70: optional i32 attempt\\n  80: optional i32 maximumAttempts\\n  90: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  100: optional i64 (js.type = \\\"Long\\\") expirationTimestamp\\n  110: optional string lastFailureReason\\n  120: optional string lastWorkerIdentity\\n  130: optional binary lastFailureDetails\\n  140: optional string startedWorkerIdentity\\n  150: optional i64 (js.type = \\\"Long\\\") scheduleID\\n}\\n\\nstruct PendingDecisionInfo {\\n  10: optional PendingDecisionState state\\n  20: optional i64 (js.type = \\\"Long\\\") scheduledTimestamp\\n  30: optional i64 (js.type = \\\"Long\\\") startedTimestamp\\n  40: optional i64 attempt\\n  50: optional i64 (js.type = \\\"Long\\\") originalScheduledTimestamp\\n  60: optional i64 (js.type = \\\"Long\\\") scheduleID\\n}\\n\\nstruct PendingChildExecutionInfo {\\n  1: optional string domain\\n  10: optional string workflowID\\n  20: optional string runID\\n  30: optional string workflowTypName\\n  40: optional i64 (js.type = \\\"Long\\\") initiatedID\\n  50: optional ParentClosePolicy parentClosePolicy\\n}\\n\\nstruct DescribeWorkflowExecutionResponse {\\n  10: optional WorkflowExecutionConfiguration executionConfiguration\\n  20: optional WorkflowExecutionInfo workflowExecutionInfo\\n  30: optional list<PendingActivityInfo> pendingActivities\\n  40: optional list<PendingChildExecutionInfo> pendingChildren\\n  50: optional PendingDecisionInfo pendingDecision\\n}\\n\\nstruct DescribeTaskListRequest {\\n  10: optional string domain\\n  20: optional TaskList taskList\\n  30: optional TaskListType taskListType\\n  40: optional bool includeTaskListStatus\\n}\\n\\nstruct DescribeTaskListResponse {\\n  10: optional list<PollerInfo> pollers\\n  20: optional TaskListStatus taskListStatus\\n  // The TaskList being described\\n  30: optional TaskList taskList\\n}\\n\\nstruct GetTaskListsByDomainRequest {\\n  10: optional string domainName\\n}\\n\\nstruct GetTaskListsByDomainResponse {\\n  10: optional map<string,DescribeTaskListResponse> decisionTaskListMap\\n  20: optional map<string,DescribeTaskListResponse> activityTaskListMap\\n}\\n\\nstruct ListTaskListPartitionsRequest {\\n  10: optional string domain\\n  20: optional TaskList taskList\\n}\\n\\nstruct TaskListPartitionMetadata {\\n  10: optional string key\\n  20: optional string ownerHostName\\n}\\n\\nstruct ListTaskListPartitionsResponse {\\n  10: optional list<TaskListPartitionMetadata> activityTaskListPartitions\\n  20: optional list<TaskListPartitionMetadata> decisionTaskListPartitions\\n}\\n\\nstruct IsolationGroupMetrics {\\n  10: optional double newTasksPerSecond\\n  20: optional i64 (js.type = \\\"Long\\\") pollerCount\\n}\\n\\nstruct TaskListStatus {\\n  10: optional i64 (js.type = \\\"Long\\\") backlogCountHint\\n  20: optional i64 (js.type = \\\"Long\\\") readLevel\\n  30: optional i64 (js.type = \\\"Long\\\") ackLevel\\n  35: optional double ratePerSecond\\n  40: optional TaskIDBlock taskIDBlock\\n  50: optional map<string, IsolationGroupMetrics> isolationGroupMetrics\\n  60: optional double newTasksPerSecond\\n  70: optional bool empty\\n}\\n\\nstruct TaskIDBlock {\\n  10: optional i64 (js.type = \\\"Long\\\")  startID\\n  20: optional i64 (js.type = \\\"Long\\\")  endID\\n}\\n\\n//At least one of the parameters needs to be provided\\nstruct DescribeHistoryHostRequest {\\n  10: optional string               hostAddress //ip:port\\n  20: optional i32                  shardIdForHost\\n  30: optional WorkflowExecution    executionForHost\\n}\\n\\nstruct RemoveTaskRequest {\\n  10: optional i32                      shardID\\n  20: optional i32                      type\\n  30: optional i64 (js.type = \\\"Long\\\")   taskID\\n  40: optional i64 (js.type = \\\"Long\\\")   visibilityTimestamp\\n  50: optional string                   clusterName\\n}\\n\\nstruct CloseShardRequest {\\n  10: optional i32               shardID\\n}\\n\\nstruct ResetQueueRequest {\\n  10: optional i32    shardID\\n  20: optional string clusterName\\n  30: optional i32    type\\n}\\n\\nstruct DescribeQueueRequest {\\n  10: optional i32    shardID\\n  20: optional string clusterName\\n  30: optional i32    type\\n}\\n\\nstruct DescribeQueueResponse {\\n  10: optional list<string> processingQueueStates\\n}\\n\\nstruct DescribeShardDistributionRequest {\\n  10: optional i32 pageSize\\n  20: optional i32 pageID\\n}\\n\\nstruct DescribeShardDistributionResponse {\\n  10: optional i32              numberOfShards\\n\\n  // ShardID to Address (ip:port) map\\n  20: optional map<i32, string> shards\\n}\\n\\nstruct DescribeHistoryHostResponse{\\n  10: optional i32                  numberOfShards\\n  20: optional list<i32>            shardIDs\\n  30: optional DomainCacheInfo      domainCache\\n  40: optional string               shardControllerStatus\\n  50: optional string               address\\n}\\n\\nstruct DomainCacheInfo{\\n  10: optional i64 numOfItemsInCacheByID\\n  20: optional i64 numOfItemsInCacheByName\\n}\\n\\nenum TaskListType {\\n  /*\\n   * Decision type of tasklist\\n   */\\n  Decision,\\n  /*\\n   * Activity type of tasklist\\n   */\\n  Activity,\\n}\\n\\nstruct PollerInfo {\\n  // Unix Nano\\n  10: optional i64 (js.type = \\\"Long\\\")  lastAccessTime\\n  20: optional string identity\\n  30: optional double ratePerSecond\\n}\\n\\nstruct RetryPolicy {\\n  // Interval of the first retry. If coefficient is 1.0 then it is used for all retries.\\n  10: optional i32 initialIntervalInSeconds\\n\\n  // Coefficient used to calculate the next retry interval.\\n  // The next retry interval is previous interval multiplied by the coefficient.\\n  // Must be 1 or larger.\\n  20: optional double backoffCoefficient\\n\\n  // Maximum interval between retries. Exponential backoff leads to interval increase.\\n  // This value is the cap of the increase. Default is 100x of initial interval.\\n  30: optional i32 maximumIntervalInSeconds\\n\\n  // Maximum number of attempts. When exceeded the retries stop even if not expired yet.\\n  // Must be 1 or bigger. Default is unlimited.\\n  40: optional i32 maximumAttempts\\n\\n  // Non-Retriable errors. Will stop retrying if error matches this list.\\n  50: optional list<string> nonRetriableErrorReasons\\n\\n  // Expiration time for the whole retry process.\\n  60: optional i32 expirationIntervalInSeconds\\n}\\n\\n// HistoryBranchRange represents a piece of range for a branch.\\nstruct HistoryBranchRange{\\n  // branchID of original branch forked from\\n  10: optional string branchID\\n  // beinning node for the range, inclusive\\n  20: optional i64 beginNodeID\\n  // ending node for the range, exclusive\\n  30: optional i64 endNodeID\\n}\\n\\n// For history persistence to serialize/deserialize branch details\\nstruct HistoryBranch{\\n  10: optional string treeID\\n  20: optional string branchID\\n  30: optional list<HistoryBranchRange> ancestors\\n}\\n\\n// VersionHistoryItem contains signal eventID and the corresponding version\\nstruct VersionHistoryItem{\\n  10: optional i64 (js.type = \\\"Long\\\") eventID\\n  20: optional i64 (js.type = \\\"Long\\\") version\\n}\\n\\n// VersionHistory contains the version history of a branch\\nstruct VersionHistory{\\n  10: optional binary branchToken\\n  20: optional list<VersionHistoryItem> items\\n}\\n\\n// VersionHistories contains all version histories from all branches\\nstruct VersionHistories{\\n  10: optional i32 currentVersionHistoryIndex\\n  20: optional list<VersionHistory> histories\\n}\\n\\n// ReapplyEventsRequest is the request for reapply events API\\nstruct ReapplyEventsRequest{\\n  10: optional string domainName\\n  20: optional WorkflowExecution workflowExecution\\n  30: optional DataBlob events\\n}\\n\\n// SupportedClientVersions contains the support versions for client library\\nstruct SupportedClientVersions{\\n  10: optional string goSdk\\n  20: optional string javaSdk\\n}\\n\\n// ClusterInfo contains information about cadence cluster\\nstruct ClusterInfo{\\n  10: optional SupportedClientVersions supportedClientVersions\\n}\\n\\nstruct RefreshWorkflowTasksRequest {\\n  10: optional string domain\\n  20: optional WorkflowExecution execution\\n}\\n\\n// DEPRECATED: use proto definition instead\\nstruct FeatureFlags {\\n\\t10: optional bool WorkflowExecutionAlreadyCompletedErrorEnabled\\n  20: optional bool AutoForwardingEnabled\\n}\\n\\nenum CrossClusterTaskType {\\n  StartChildExecution\\n  CancelExecution\\n  SignalExecution\\n  RecordChildWorkflowExecutionComplete\\n  ApplyParentClosePolicy\\n}\\n\\nenum CrossClusterTaskFailedCause {\\n  DOMAIN_NOT_ACTIVE\\n  DOMAIN_NOT_EXISTS\\n  WORKFLOW_ALREADY_RUNNING\\n  WORKFLOW_NOT_EXISTS\\n  WORKFLOW_ALREADY_COMPLETED\\n  UNCATEGORIZED\\n}\\n\\nenum GetTaskFailedCause {\\n  SERVICE_BUSY\\n  TIMEOUT\\n  SHARD_OWNERSHIP_LOST\\n  UNCATEGORIZED\\n}\\n\\nstruct CrossClusterTaskInfo {\\n  10: optional string domainID\\n  20: optional string workflowID\\n  30: optional string runID\\n  40: optional CrossClusterTaskType taskType\\n  50: optional i16 taskState\\n  60: optional i64 (js.type = \\\"Long\\\") taskID\\n  70: optional i64 (js.type = \\\"Long\\\") visibilityTimestamp\\n}\\n\\nstruct CrossClusterStartChildExecutionRequestAttributes {\\n  10: optional string targetDomainID\\n  20: optional string requestID\\n  30: optional i64 (js.type = \\\"Long\\\") initiatedEventID\\n  40: optional StartChildWorkflowExecutionInitiatedEventAttributes initiatedEventAttributes\\n  // targetRunID is for scheduling first decision task\\n  // targetWorkflowID is available in initiatedEventAttributes\\n  50: optional string targetRunID\\n  60: optional map<string, string> partitionConfig\\n}\\n\\nstruct CrossClusterStartChildExecutionResponseAttributes {\\n  10: optional string runID\\n}\\n\\nstruct CrossClusterCancelExecutionRequestAttributes {\\n  10: optional string targetDomainID\\n  20: optional string targetWorkflowID\\n  30: optional string targetRunID\\n  40: optional string requestID\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventID\\n  60: optional bool childWorkflowOnly\\n}\\n\\nstruct CrossClusterCancelExecutionResponseAttributes {\\n}\\n\\nstruct CrossClusterSignalExecutionRequestAttributes {\\n  10: optional string targetDomainID\\n  20: optional string targetWorkflowID\\n  30: optional string targetRunID\\n  40: optional string requestID\\n  50: optional i64 (js.type = \\\"Long\\\") initiatedEventID\\n  60: optional bool childWorkflowOnly\\n  70: optional string signalName\\n  80: optional binary signalInput\\n  90: optional binary control\\n}\\n\\nstruct CrossClusterSignalExecutionResponseAttributes {\\n}\\n\\nstruct CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes {\\n  10: optional string targetDomainID\\n  20: optional string targetWorkflowID\\n  30: optional string targetRunID\\n  40: optional i64 (js.type = \\\"Long\\\") initiatedEventID\\n  50: optional HistoryEvent completionEvent\\n}\\n\\nstruct CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes {\\n}\\n\\nstruct ApplyParentClosePolicyAttributes {\\n  10: optional string childDomainID\\n  20: optional string childWorkflowID\\n  30: optional string childRunID\\n  40: optional ParentClosePolicy parentClosePolicy\\n}\\n\\nstruct ApplyParentClosePolicyStatus {\\n  10: optional bool completed\\n  20: optional CrossClusterTaskFailedCause failedCause\\n}\\n\\nstruct ApplyParentClosePolicyRequest {\\n  10: optional ApplyParentClosePolicyAttributes child\\n  20: optional ApplyParentClosePolicyStatus status\\n}\\n\\nstruct CrossClusterApplyParentClosePolicyRequestAttributes {\\n  10: optional list<ApplyParentClosePolicyRequest> children\\n}\\n\\nstruct ApplyParentClosePolicyResult {\\n  10: optional ApplyParentClosePolicyAttributes child\\n  20: optional CrossClusterTaskFailedCause failedCause\\n}\\n\\nstruct CrossClusterApplyParentClosePolicyResponseAttributes {\\n  10: optional list<ApplyParentClosePolicyResult> childrenStatus\\n}\\n\\nstruct CrossClusterTaskRequest {\\n  10: optional CrossClusterTaskInfo taskInfo\\n  20: optional CrossClusterStartChildExecutionRequestAttributes startChildExecutionAttributes\\n  30: optional CrossClusterCancelExecutionRequestAttributes cancelExecutionAttributes\\n  40: optional CrossClusterSignalExecutionRequestAttributes signalExecutionAttributes\\n  50: optional CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes recordChildWorkflowExecutionCompleteAttributes\\n  60: optional CrossClusterApplyParentClosePolicyRequestAttributes applyParentClosePolicyAttributes\\n}\\n\\nstruct CrossClusterTaskResponse {\\n  10: optional i64 (js.type = \\\"Long\\\") taskID\\n  20: optional CrossClusterTaskType taskType\\n  30: optional i16 taskState\\n  40: optional CrossClusterTaskFailedCause failedCause\\n  50: optional CrossClusterStartChildExecutionResponseAttributes startChildExecutionAttributes\\n  60: optional CrossClusterCancelExecutionResponseAttributes cancelExecutionAttributes\\n  70: optional CrossClusterSignalExecutionResponseAttributes signalExecutionAttributes\\n  80: optional CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes recordChildWorkflowExecutionCompleteAttributes\\n  90: optional CrossClusterApplyParentClosePolicyResponseAttributes applyParentClosePolicyAttributes\\n}\\n\\nstruct GetCrossClusterTasksRequest {\\n  10: optional list<i32> shardIDs\\n  20: optional string targetCluster\\n}\\n\\nstruct GetCrossClusterTasksResponse {\\n  10: optional map<i32, list<CrossClusterTaskRequest>> tasksByShard\\n  20: optional map<i32, GetTaskFailedCause> failedCauseByShard\\n}\\n\\nstruct RespondCrossClusterTasksCompletedRequest {\\n  10: optional i32 shardID\\n  20: optional string targetCluster\\n  30: optional list<CrossClusterTaskResponse> taskResponses\\n  40: optional bool fetchNewTasks\\n}\\n\\nstruct RespondCrossClusterTasksCompletedResponse {\\n  10: optional list<CrossClusterTaskRequest> tasks\\n}\\n\\nenum IsolationGroupState {\\n  INVALID,\\n  HEALTHY,\\n  DRAINED,\\n}\\n\\nstruct IsolationGroupPartition {\\n  10: optional string name\\n  20: optional IsolationGroupState state\\n}\\n\\nstruct IsolationGroupConfiguration {\\n  10: optional list<IsolationGroupPartition> isolationGroups\\n}\\n\\nstruct AsyncWorkflowConfiguration {\\n  10: optional bool enabled\\n  // PredefinedQueueName is the name of the predefined queue in cadence server config's asyncWorkflowQueues\\n  20: optional string predefinedQueueName\\n  // queueType is the type of the queue if predefined_queue_name is not used\\n  30: optional string queueType\\n  // queueConfig is the configuration for the queue if predefined_queue_name is not used\\n  40: optional DataBlob queueConfig\\n}\\n\\n/**\\n* Any is a logical duplicate of google.protobuf.Any.\\n*\\n* The intent of the type is the same, but it is not intended to be directly\\n* compatible with google.protobuf.Any or any Thrift equivalent - this blob is\\n* RPC-type agnostic by design (as the underlying data may be transported over\\n* proto or thrift), and the data-bytes may be in any encoding.\\n*\\n* This is intentionally different from DataBlob, which supports only a handful\\n* of known encodings so it can be interpreted everywhere.  Any supports literally\\n* any contents, and needs to be considered opaque until it is given to something\\n* that is expecting it.\\n*\\n* See ValueType to interpret the contents.\\n**/\\nstruct Any {\\n  // Type-string describing value's contents, and intentionally avoiding the\\n  // name \\\"type\\\" as it is often a special term.\\n  // This should usually be a hard-coded string of some kind.\\n  10: optional string ValueType\\n  // Arbitrarily-encoded bytes, to be deserialized by a runtime implementation.\\n  // The contents are described by ValueType.\\n  20: optional binary Value\\n}\\n\\nstruct AutoConfigHint {\\n  10: optional bool enableAutoConfig\\n  20: optional i64 pollerWaitTimeInMs\\n}\\n\\nstruct QueueState {\\n  10: optional map<i64, VirtualQueueState> virtualQueueStates\\n  20: optional TaskKey exclusiveMaxReadLevel\\n}\\n\\nstruct VirtualQueueState {\\n  10: optional list<VirtualSliceState> virtualSliceStates\\n}\\n\\nstruct VirtualSliceState {\\n  10: optional TaskRange taskRange\\n  20: optional Predicate predicate\\n}\\n\\nstruct TaskRange {\\n  10: optional TaskKey inclusiveMin\\n  20: optional TaskKey exclusiveMax\\n}\\n\\nstruct TaskKey {\\n  10: optional i64 scheduledTimeNano\\n  20: optional i64 taskID\\n}\\n\\n// ActiveClusterSelectionPolicy is for active-active domains, it serves as a means to select\\n// the active cluster, by specifying the attribute by which to divide the workflows\\n// in that domain.\\nstruct ActiveClusterSelectionPolicy {\\n  1: optional ClusterAttribute clusterAttribute\\n\\n  10: optional ActiveClusterSelectionStrategy strategy // todo (david.porter) remove these as they're not used anymore\\n  20: optional string stickyRegion                     // todo (david.porter) remove these as they're not used anymore\\n  30: optional string externalEntityType               // todo (david.porter) remove these as they're not used anymore\\n  40: optional string externalEntityKey                // todo (david.porter) remove these as they're not used anymore\\n}\\n\\n// ClusterAttribute is used for subdividing workflows in a domain into their active\\n// and passive clusters. Examples of this might be 'region' and 'cluster1' as\\n// respective region and scope fields.\\n//\\n// for example, a workflow may specify this in it's start request:\\n//\\n//   StartWorkflowRequest{\\n//     ActiveClusterSelectionPolicy: {\\n//       ClusterAttribute: {\\n//            Scope: \\\"cityID\\\",\\n//            Name: \\\"Lisbon\\\"\\n//        }\\n//     }\\n//   }\\n//\\n// and this means that this workflow will be associate with the domain's cluster attribute 'Lisbon',\\n// be active in the cluster that has Lisbon active and\\n// failover when that cluster-attribute is set to failover.\\nstruct ClusterAttribute {\\n  1: optional string scope\\n  2: optional string name\\n}\\n\\n// FailoverType describes how a failover operation will be performed.\\nenum FailoverType {\\n  INVALID\\n  FORCE\\n  GRACEFUL\\n}\\n\\n// PaginationOptions provides common options for paginated RPCs.\\nstruct PaginationOptions {\\n  // page_size configures the number of results to be returned as part of each page\\n  10: optional i32 pageSize\\n  // next_page_token should be provided from a previous response to fetch the next page.\\n  // if empty, the first page will be returned.\\n  20: optional binary nextPageToken\\n}\\n\\n// todo (david.porter) Remove this, as it's no longer needed\\n// with the active/active configuration we have\\nenum ActiveClusterSelectionStrategy {\\n  REGION_STICKY,\\n  EXTERNAL_ENTITY,\\n}\\n\\nenum PredicateType {\\n  Universal,\\n  Empty,\\n  DomainID,\\n}\\n\\nstruct UniversalPredicateAttributes {}\\n\\nstruct EmptyPredicateAttributes {}\\n\\nstruct DomainIDPredicateAttributes {\\n  10: optional list<string> domainIDs\\n  20: optional bool isExclusive\\n}\\n\\nstruct Predicate {\\n  10: optional PredicateType predicateType\\n  20: optional UniversalPredicateAttributes universalPredicateAttributes\\n  30: optional EmptyPredicateAttributes emptyPredicateAttributes\\n  40: optional DomainIDPredicateAttributes domainIDPredicateAttributes\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/shared/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage shared\n\nimport yarpcerrors \"go.uber.org/yarpc/yarpcerrors\"\n\n// YARPCErrorCode returns nil for AccessDeniedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *AccessDeniedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for AccessDeniedError.\nfunc (e *AccessDeniedError) YARPCErrorName() string { return \"AccessDeniedError\" }\n\n// YARPCErrorCode returns nil for BadRequestError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *BadRequestError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for BadRequestError.\nfunc (e *BadRequestError) YARPCErrorName() string { return \"BadRequestError\" }\n\n// YARPCErrorCode returns nil for CancellationAlreadyRequestedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *CancellationAlreadyRequestedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for CancellationAlreadyRequestedError.\nfunc (e *CancellationAlreadyRequestedError) YARPCErrorName() string {\n\treturn \"CancellationAlreadyRequestedError\"\n}\n\n// YARPCErrorCode returns nil for ClientVersionNotSupportedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *ClientVersionNotSupportedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for ClientVersionNotSupportedError.\nfunc (e *ClientVersionNotSupportedError) YARPCErrorName() string {\n\treturn \"ClientVersionNotSupportedError\"\n}\n\n// YARPCErrorCode returns nil for CurrentBranchChangedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *CurrentBranchChangedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for CurrentBranchChangedError.\nfunc (e *CurrentBranchChangedError) YARPCErrorName() string { return \"CurrentBranchChangedError\" }\n\n// YARPCErrorCode returns nil for DomainAlreadyExistsError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *DomainAlreadyExistsError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for DomainAlreadyExistsError.\nfunc (e *DomainAlreadyExistsError) YARPCErrorName() string { return \"DomainAlreadyExistsError\" }\n\n// YARPCErrorCode returns nil for DomainNotActiveError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *DomainNotActiveError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for DomainNotActiveError.\nfunc (e *DomainNotActiveError) YARPCErrorName() string { return \"DomainNotActiveError\" }\n\n// YARPCErrorCode returns nil for EntityNotExistsError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *EntityNotExistsError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for EntityNotExistsError.\nfunc (e *EntityNotExistsError) YARPCErrorName() string { return \"EntityNotExistsError\" }\n\n// YARPCErrorCode returns nil for FeatureNotEnabledError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *FeatureNotEnabledError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for FeatureNotEnabledError.\nfunc (e *FeatureNotEnabledError) YARPCErrorName() string { return \"FeatureNotEnabledError\" }\n\n// YARPCErrorCode returns nil for InternalDataInconsistencyError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *InternalDataInconsistencyError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for InternalDataInconsistencyError.\nfunc (e *InternalDataInconsistencyError) YARPCErrorName() string {\n\treturn \"InternalDataInconsistencyError\"\n}\n\n// YARPCErrorCode returns nil for InternalServiceError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *InternalServiceError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for InternalServiceError.\nfunc (e *InternalServiceError) YARPCErrorName() string { return \"InternalServiceError\" }\n\n// YARPCErrorCode returns nil for LimitExceededError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *LimitExceededError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for LimitExceededError.\nfunc (e *LimitExceededError) YARPCErrorName() string { return \"LimitExceededError\" }\n\n// YARPCErrorCode returns nil for QueryFailedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *QueryFailedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for QueryFailedError.\nfunc (e *QueryFailedError) YARPCErrorName() string { return \"QueryFailedError\" }\n\n// YARPCErrorCode returns nil for RemoteSyncMatchedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *RemoteSyncMatchedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for RemoteSyncMatchedError.\nfunc (e *RemoteSyncMatchedError) YARPCErrorName() string { return \"RemoteSyncMatchedError\" }\n\n// YARPCErrorCode returns nil for RetryTaskV2Error.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *RetryTaskV2Error) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for RetryTaskV2Error.\nfunc (e *RetryTaskV2Error) YARPCErrorName() string { return \"RetryTaskV2Error\" }\n\n// YARPCErrorCode returns nil for ServiceBusyError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *ServiceBusyError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for ServiceBusyError.\nfunc (e *ServiceBusyError) YARPCErrorName() string { return \"ServiceBusyError\" }\n\n// YARPCErrorCode returns nil for StickyWorkerUnavailableError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *StickyWorkerUnavailableError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for StickyWorkerUnavailableError.\nfunc (e *StickyWorkerUnavailableError) YARPCErrorName() string { return \"StickyWorkerUnavailableError\" }\n\n// YARPCErrorCode returns nil for TaskListNotOwnedByHostError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *TaskListNotOwnedByHostError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for TaskListNotOwnedByHostError.\nfunc (e *TaskListNotOwnedByHostError) YARPCErrorName() string { return \"TaskListNotOwnedByHostError\" }\n\n// YARPCErrorCode returns nil for WorkflowExecutionAlreadyCompletedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *WorkflowExecutionAlreadyCompletedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for WorkflowExecutionAlreadyCompletedError.\nfunc (e *WorkflowExecutionAlreadyCompletedError) YARPCErrorName() string {\n\treturn \"WorkflowExecutionAlreadyCompletedError\"\n}\n\n// YARPCErrorCode returns nil for WorkflowExecutionAlreadyStartedError.\n//\n// This is derived from the rpc.code annotation on the Thrift exception.\nfunc (e *WorkflowExecutionAlreadyStartedError) YARPCErrorCode() *yarpcerrors.Code {\n\n\treturn nil\n}\n\n// Name is the error name for WorkflowExecutionAlreadyStartedError.\nfunc (e *WorkflowExecutionAlreadyStartedError) YARPCErrorName() string {\n\treturn \"WorkflowExecutionAlreadyStartedError\"\n}\n"
  },
  {
    "path": ".gen/go/sqlblobs/sqlblobs.go",
    "content": "// Code generated by thriftrw v1.29.2. DO NOT EDIT.\n// @generated\n\npackage sqlblobs\n\nimport (\n\tbytes \"bytes\"\n\tbase64 \"encoding/base64\"\n\tjson \"encoding/json\"\n\tfmt \"fmt\"\n\tmath \"math\"\n\tstrconv \"strconv\"\n\tstrings \"strings\"\n\n\tmultierr \"go.uber.org/multierr\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\tthriftreflect \"go.uber.org/thriftrw/thriftreflect\"\n\twire \"go.uber.org/thriftrw/wire\"\n\tzapcore \"go.uber.org/zap/zapcore\"\n\n\tshared \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype ActivityInfo struct {\n\tVersion                       *int64               `json:\"version,omitempty\"`\n\tScheduledEventBatchID         *int64               `json:\"scheduledEventBatchID,omitempty\"`\n\tScheduledEvent                []byte               `json:\"scheduledEvent,omitempty\"`\n\tScheduledEventEncoding        *string              `json:\"scheduledEventEncoding,omitempty\"`\n\tScheduledTimeNanos            *int64               `json:\"scheduledTimeNanos,omitempty\"`\n\tStartedID                     *int64               `json:\"startedID,omitempty\"`\n\tStartedEvent                  []byte               `json:\"startedEvent,omitempty\"`\n\tStartedEventEncoding          *string              `json:\"startedEventEncoding,omitempty\"`\n\tStartedTimeNanos              *int64               `json:\"startedTimeNanos,omitempty\"`\n\tActivityID                    *string              `json:\"activityID,omitempty\"`\n\tRequestID                     *string              `json:\"requestID,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32               `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tScheduleToCloseTimeoutSeconds *int32               `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tStartToCloseTimeoutSeconds    *int32               `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds       *int32               `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tCancelRequested               *bool                `json:\"cancelRequested,omitempty\"`\n\tCancelRequestID               *int64               `json:\"cancelRequestID,omitempty\"`\n\tTimerTaskStatus               *int32               `json:\"timerTaskStatus,omitempty\"`\n\tAttempt                       *int32               `json:\"attempt,omitempty\"`\n\tTaskList                      *string              `json:\"taskList,omitempty\"`\n\tTaskListKind                  *shared.TaskListKind `json:\"taskListKind,omitempty\"`\n\tStartedIdentity               *string              `json:\"startedIdentity,omitempty\"`\n\tHasRetryPolicy                *bool                `json:\"hasRetryPolicy,omitempty\"`\n\tRetryInitialIntervalSeconds   *int32               `json:\"retryInitialIntervalSeconds,omitempty\"`\n\tRetryMaximumIntervalSeconds   *int32               `json:\"retryMaximumIntervalSeconds,omitempty\"`\n\tRetryMaximumAttempts          *int32               `json:\"retryMaximumAttempts,omitempty\"`\n\tRetryExpirationTimeNanos      *int64               `json:\"retryExpirationTimeNanos,omitempty\"`\n\tRetryBackoffCoefficient       *float64             `json:\"retryBackoffCoefficient,omitempty\"`\n\tRetryNonRetryableErrors       []string             `json:\"retryNonRetryableErrors,omitempty\"`\n\tRetryLastFailureReason        *string              `json:\"retryLastFailureReason,omitempty\"`\n\tRetryLastWorkerIdentity       *string              `json:\"retryLastWorkerIdentity,omitempty\"`\n\tRetryLastFailureDetails       []byte               `json:\"retryLastFailureDetails,omitempty\"`\n}\n\ntype _List_String_ValueList []string\n\nfunc (v _List_String_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tw, err := wire.NewValueString(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_String_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_String_ValueList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_List_String_ValueList) Close() {}\n\n// ToWire translates a ActivityInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ActivityInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [32]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventBatchID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledEventBatchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEvent != nil {\n\t\tw, err = wire.NewValueBinary(v.ScheduledEvent), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledEventEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.ScheduledEventEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tw, err = wire.NewValueBinary(v.StartedEvent), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.StartedEventEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.StartedTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tw, err = wire.NewValueString(*(v.ActivityID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToStartTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 32, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.ScheduleToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.StartToCloseTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.HeartbeatTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequested != nil {\n\t\tw, err = wire.NewValueBool(*(v.CancelRequested)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tw, err = wire.NewValueI64(*(v.CancelRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 42, Value: w}\n\t\ti++\n\t}\n\tif v.TimerTaskStatus != nil {\n\t\tw, err = wire.NewValueI32(*(v.TimerTaskStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 44, Value: w}\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tw, err = wire.NewValueI32(*(v.Attempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 46, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 48, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListKind != nil {\n\t\tw, err = v.TaskListKind.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 49, Value: w}\n\t\ti++\n\t}\n\tif v.StartedIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.StartedIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tw, err = wire.NewValueBool(*(v.HasRetryPolicy)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 52, Value: w}\n\t\ti++\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryInitialIntervalSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 54, Value: w}\n\t\ti++\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryMaximumIntervalSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryMaximumAttempts)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 58, Value: w}\n\t\ti++\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.RetryExpirationTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tw, err = wire.NewValueDouble(*(v.RetryBackoffCoefficient)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 62, Value: w}\n\t\ti++\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.RetryNonRetryableErrors)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 64, Value: w}\n\t\ti++\n\t}\n\tif v.RetryLastFailureReason != nil {\n\t\tw, err = wire.NewValueString(*(v.RetryLastFailureReason)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 66, Value: w}\n\t\ti++\n\t}\n\tif v.RetryLastWorkerIdentity != nil {\n\t\tw, err = wire.NewValueString(*(v.RetryLastWorkerIdentity)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 68, Value: w}\n\t\ti++\n\t}\n\tif v.RetryLastFailureDetails != nil {\n\t\tw, err = wire.NewValueBinary(v.RetryLastFailureDetails), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListKind_Read(w wire.Value) (shared.TaskListKind, error) {\n\tvar v shared.TaskListKind\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _List_String_Read(l wire.ValueList) ([]string, error) {\n\tif l.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make([]string, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ActivityInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ActivityInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ActivityInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ActivityInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledEventBatchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ScheduledEvent, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ScheduledEventEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.StartedEvent, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartedEventEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActivityID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 32:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.CancelRequested = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 42:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CancelRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 44:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.TimerTaskStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 46:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Attempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 48:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 49:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x shared.TaskListKind\n\t\t\t\tx, err = _TaskListKind_Read(field.Value)\n\t\t\t\tv.TaskListKind = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartedIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 52:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.HasRetryPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 54:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryInitialIntervalSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryMaximumIntervalSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 58:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryMaximumAttempts = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.RetryExpirationTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 62:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.RetryBackoffCoefficient = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 64:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.RetryNonRetryableErrors, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 66:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RetryLastFailureReason = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 68:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RetryLastWorkerIdentity = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.RetryLastFailureDetails, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_String_Encode(val []string, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TBinary,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a ActivityInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ActivityInfo struct could not be encoded.\nfunc (v *ActivityInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventBatchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledEventBatchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ScheduledEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledEventEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ScheduledEventEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.StartedEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartedEventEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActivityID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActivityID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 32, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToStartTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ScheduleToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StartToCloseTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.HeartbeatTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequested != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.CancelRequested)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 42, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CancelRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerTaskStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 44, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.TimerTaskStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Attempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 46, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Attempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 48, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListKind != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 49, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListKind.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartedIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HasRetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 52, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.HasRetryPolicy)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 54, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryInitialIntervalSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryMaximumIntervalSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryMaximumAttempts != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 58, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryMaximumAttempts)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.RetryExpirationTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryBackoffCoefficient != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 62, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.RetryBackoffCoefficient)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryNonRetryableErrors != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 64, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.RetryNonRetryableErrors, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryLastFailureReason != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 66, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RetryLastFailureReason)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryLastWorkerIdentity != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 68, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RetryLastWorkerIdentity)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryLastFailureDetails != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.RetryLastFailureDetails); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListKind_Decode(sr stream.Reader) (shared.TaskListKind, error) {\n\tvar v shared.TaskListKind\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _List_String_Decode(sr stream.Reader) ([]string, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TBinary {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]string, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ActivityInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ActivityInfo struct could not be generated from the wire\n// representation.\nfunc (v *ActivityInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledEventBatchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tv.ScheduledEvent, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ScheduledEventEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TBinary:\n\t\t\tv.StartedEvent, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartedEventEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActivityID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 32 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToStartTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ScheduleToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StartToCloseTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.HeartbeatTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.CancelRequested = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 42 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CancelRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 44 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.TimerTaskStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 46 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Attempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 48 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 49 && fh.Type == wire.TI32:\n\t\t\tvar x shared.TaskListKind\n\t\t\tx, err = _TaskListKind_Decode(sr)\n\t\t\tv.TaskListKind = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartedIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 52 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.HasRetryPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 54 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryInitialIntervalSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryMaximumIntervalSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 58 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryMaximumAttempts = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.RetryExpirationTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 62 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.RetryBackoffCoefficient = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 64 && fh.Type == wire.TList:\n\t\t\tv.RetryNonRetryableErrors, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 66 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RetryLastFailureReason = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 68 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RetryLastWorkerIdentity = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBinary:\n\t\t\tv.RetryLastFailureDetails, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ActivityInfo\n// struct.\nfunc (v *ActivityInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [32]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.ScheduledEventBatchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventBatchID: %v\", *(v.ScheduledEventBatchID))\n\t\ti++\n\t}\n\tif v.ScheduledEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEvent: %v\", v.ScheduledEvent)\n\t\ti++\n\t}\n\tif v.ScheduledEventEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledEventEncoding: %v\", *(v.ScheduledEventEncoding))\n\t\ti++\n\t}\n\tif v.ScheduledTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledTimeNanos: %v\", *(v.ScheduledTimeNanos))\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedID: %v\", *(v.StartedID))\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEvent: %v\", v.StartedEvent)\n\t\ti++\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventEncoding: %v\", *(v.StartedEventEncoding))\n\t\ti++\n\t}\n\tif v.StartedTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedTimeNanos: %v\", *(v.StartedTimeNanos))\n\t\ti++\n\t}\n\tif v.ActivityID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActivityID: %v\", *(v.ActivityID))\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestID: %v\", *(v.RequestID))\n\t\ti++\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToStartTimeoutSeconds: %v\", *(v.ScheduleToStartTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleToCloseTimeoutSeconds: %v\", *(v.ScheduleToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartToCloseTimeoutSeconds: %v\", *(v.StartToCloseTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"HeartbeatTimeoutSeconds: %v\", *(v.HeartbeatTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.CancelRequested != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequested: %v\", *(v.CancelRequested))\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequestID: %v\", *(v.CancelRequestID))\n\t\ti++\n\t}\n\tif v.TimerTaskStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerTaskStatus: %v\", *(v.TimerTaskStatus))\n\t\ti++\n\t}\n\tif v.Attempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"Attempt: %v\", *(v.Attempt))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\tif v.TaskListKind != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListKind: %v\", *(v.TaskListKind))\n\t\ti++\n\t}\n\tif v.StartedIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedIdentity: %v\", *(v.StartedIdentity))\n\t\ti++\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"HasRetryPolicy: %v\", *(v.HasRetryPolicy))\n\t\ti++\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryInitialIntervalSeconds: %v\", *(v.RetryInitialIntervalSeconds))\n\t\ti++\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryMaximumIntervalSeconds: %v\", *(v.RetryMaximumIntervalSeconds))\n\t\ti++\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryMaximumAttempts: %v\", *(v.RetryMaximumAttempts))\n\t\ti++\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryExpirationTimeNanos: %v\", *(v.RetryExpirationTimeNanos))\n\t\ti++\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryBackoffCoefficient: %v\", *(v.RetryBackoffCoefficient))\n\t\ti++\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryNonRetryableErrors: %v\", v.RetryNonRetryableErrors)\n\t\ti++\n\t}\n\tif v.RetryLastFailureReason != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryLastFailureReason: %v\", *(v.RetryLastFailureReason))\n\t\ti++\n\t}\n\tif v.RetryLastWorkerIdentity != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryLastWorkerIdentity: %v\", *(v.RetryLastWorkerIdentity))\n\t\ti++\n\t}\n\tif v.RetryLastFailureDetails != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryLastFailureDetails: %v\", v.RetryLastFailureDetails)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ActivityInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I64_EqualsPtr(lhs, rhs *int64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _String_EqualsPtr(lhs, rhs *string) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _I32_EqualsPtr(lhs, rhs *int32) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Bool_EqualsPtr(lhs, rhs *bool) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _TaskListKind_EqualsPtr(lhs, rhs *shared.TaskListKind) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Double_EqualsPtr(lhs, rhs *float64) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _List_String_Equals(lhs, rhs []string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this ActivityInfo match the\n// provided ActivityInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ActivityInfo) Equals(rhs *ActivityInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledEventBatchID, rhs.ScheduledEventBatchID) {\n\t\treturn false\n\t}\n\tif !((v.ScheduledEvent == nil && rhs.ScheduledEvent == nil) || (v.ScheduledEvent != nil && rhs.ScheduledEvent != nil && bytes.Equal(v.ScheduledEvent, rhs.ScheduledEvent))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ScheduledEventEncoding, rhs.ScheduledEventEncoding) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledTimeNanos, rhs.ScheduledTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedID, rhs.StartedID) {\n\t\treturn false\n\t}\n\tif !((v.StartedEvent == nil && rhs.StartedEvent == nil) || (v.StartedEvent != nil && rhs.StartedEvent != nil && bytes.Equal(v.StartedEvent, rhs.StartedEvent))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartedEventEncoding, rhs.StartedEventEncoding) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedTimeNanos, rhs.StartedTimeNanos) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActivityID, rhs.ActivityID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestID, rhs.RequestID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToStartTimeoutSeconds, rhs.ScheduleToStartTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ScheduleToCloseTimeoutSeconds, rhs.ScheduleToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StartToCloseTimeoutSeconds, rhs.StartToCloseTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.HeartbeatTimeoutSeconds, rhs.HeartbeatTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.CancelRequested, rhs.CancelRequested) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CancelRequestID, rhs.CancelRequestID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.TimerTaskStatus, rhs.TimerTaskStatus) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Attempt, rhs.Attempt) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\tif !_TaskListKind_EqualsPtr(v.TaskListKind, rhs.TaskListKind) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartedIdentity, rhs.StartedIdentity) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.HasRetryPolicy, rhs.HasRetryPolicy) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryInitialIntervalSeconds, rhs.RetryInitialIntervalSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryMaximumIntervalSeconds, rhs.RetryMaximumIntervalSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryMaximumAttempts, rhs.RetryMaximumAttempts) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.RetryExpirationTimeNanos, rhs.RetryExpirationTimeNanos) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.RetryBackoffCoefficient, rhs.RetryBackoffCoefficient) {\n\t\treturn false\n\t}\n\tif !((v.RetryNonRetryableErrors == nil && rhs.RetryNonRetryableErrors == nil) || (v.RetryNonRetryableErrors != nil && rhs.RetryNonRetryableErrors != nil && _List_String_Equals(v.RetryNonRetryableErrors, rhs.RetryNonRetryableErrors))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RetryLastFailureReason, rhs.RetryLastFailureReason) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RetryLastWorkerIdentity, rhs.RetryLastWorkerIdentity) {\n\t\treturn false\n\t}\n\tif !((v.RetryLastFailureDetails == nil && rhs.RetryLastFailureDetails == nil) || (v.RetryLastFailureDetails != nil && rhs.RetryLastFailureDetails != nil && bytes.Equal(v.RetryLastFailureDetails, rhs.RetryLastFailureDetails))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_String_Zapper []string\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_String_Zapper.\nfunc (l _List_String_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\tenc.AppendString(v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ActivityInfo.\nfunc (v *ActivityInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.ScheduledEventBatchID != nil {\n\t\tenc.AddInt64(\"scheduledEventBatchID\", *v.ScheduledEventBatchID)\n\t}\n\tif v.ScheduledEvent != nil {\n\t\tenc.AddString(\"scheduledEvent\", base64.StdEncoding.EncodeToString(v.ScheduledEvent))\n\t}\n\tif v.ScheduledEventEncoding != nil {\n\t\tenc.AddString(\"scheduledEventEncoding\", *v.ScheduledEventEncoding)\n\t}\n\tif v.ScheduledTimeNanos != nil {\n\t\tenc.AddInt64(\"scheduledTimeNanos\", *v.ScheduledTimeNanos)\n\t}\n\tif v.StartedID != nil {\n\t\tenc.AddInt64(\"startedID\", *v.StartedID)\n\t}\n\tif v.StartedEvent != nil {\n\t\tenc.AddString(\"startedEvent\", base64.StdEncoding.EncodeToString(v.StartedEvent))\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tenc.AddString(\"startedEventEncoding\", *v.StartedEventEncoding)\n\t}\n\tif v.StartedTimeNanos != nil {\n\t\tenc.AddInt64(\"startedTimeNanos\", *v.StartedTimeNanos)\n\t}\n\tif v.ActivityID != nil {\n\t\tenc.AddString(\"activityID\", *v.ActivityID)\n\t}\n\tif v.RequestID != nil {\n\t\tenc.AddString(\"requestID\", *v.RequestID)\n\t}\n\tif v.ScheduleToStartTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToStartTimeoutSeconds\", *v.ScheduleToStartTimeoutSeconds)\n\t}\n\tif v.ScheduleToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"scheduleToCloseTimeoutSeconds\", *v.ScheduleToCloseTimeoutSeconds)\n\t}\n\tif v.StartToCloseTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"startToCloseTimeoutSeconds\", *v.StartToCloseTimeoutSeconds)\n\t}\n\tif v.HeartbeatTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"heartbeatTimeoutSeconds\", *v.HeartbeatTimeoutSeconds)\n\t}\n\tif v.CancelRequested != nil {\n\t\tenc.AddBool(\"cancelRequested\", *v.CancelRequested)\n\t}\n\tif v.CancelRequestID != nil {\n\t\tenc.AddInt64(\"cancelRequestID\", *v.CancelRequestID)\n\t}\n\tif v.TimerTaskStatus != nil {\n\t\tenc.AddInt32(\"timerTaskStatus\", *v.TimerTaskStatus)\n\t}\n\tif v.Attempt != nil {\n\t\tenc.AddInt32(\"attempt\", *v.Attempt)\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\tif v.TaskListKind != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListKind\", *v.TaskListKind))\n\t}\n\tif v.StartedIdentity != nil {\n\t\tenc.AddString(\"startedIdentity\", *v.StartedIdentity)\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tenc.AddBool(\"hasRetryPolicy\", *v.HasRetryPolicy)\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tenc.AddInt32(\"retryInitialIntervalSeconds\", *v.RetryInitialIntervalSeconds)\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tenc.AddInt32(\"retryMaximumIntervalSeconds\", *v.RetryMaximumIntervalSeconds)\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tenc.AddInt32(\"retryMaximumAttempts\", *v.RetryMaximumAttempts)\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tenc.AddInt64(\"retryExpirationTimeNanos\", *v.RetryExpirationTimeNanos)\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tenc.AddFloat64(\"retryBackoffCoefficient\", *v.RetryBackoffCoefficient)\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"retryNonRetryableErrors\", (_List_String_Zapper)(v.RetryNonRetryableErrors)))\n\t}\n\tif v.RetryLastFailureReason != nil {\n\t\tenc.AddString(\"retryLastFailureReason\", *v.RetryLastFailureReason)\n\t}\n\tif v.RetryLastWorkerIdentity != nil {\n\t\tenc.AddString(\"retryLastWorkerIdentity\", *v.RetryLastWorkerIdentity)\n\t}\n\tif v.RetryLastFailureDetails != nil {\n\t\tenc.AddString(\"retryLastFailureDetails\", base64.StdEncoding.EncodeToString(v.RetryLastFailureDetails))\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *ActivityInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetScheduledEventBatchID returns the value of ScheduledEventBatchID if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduledEventBatchID() (o int64) {\n\tif v != nil && v.ScheduledEventBatchID != nil {\n\t\treturn *v.ScheduledEventBatchID\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventBatchID returns true if ScheduledEventBatchID is not nil.\nfunc (v *ActivityInfo) IsSetScheduledEventBatchID() bool {\n\treturn v != nil && v.ScheduledEventBatchID != nil\n}\n\n// GetScheduledEvent returns the value of ScheduledEvent if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduledEvent() (o []byte) {\n\tif v != nil && v.ScheduledEvent != nil {\n\t\treturn v.ScheduledEvent\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEvent returns true if ScheduledEvent is not nil.\nfunc (v *ActivityInfo) IsSetScheduledEvent() bool {\n\treturn v != nil && v.ScheduledEvent != nil\n}\n\n// GetScheduledEventEncoding returns the value of ScheduledEventEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduledEventEncoding() (o string) {\n\tif v != nil && v.ScheduledEventEncoding != nil {\n\t\treturn *v.ScheduledEventEncoding\n\t}\n\n\treturn\n}\n\n// IsSetScheduledEventEncoding returns true if ScheduledEventEncoding is not nil.\nfunc (v *ActivityInfo) IsSetScheduledEventEncoding() bool {\n\treturn v != nil && v.ScheduledEventEncoding != nil\n}\n\n// GetScheduledTimeNanos returns the value of ScheduledTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduledTimeNanos() (o int64) {\n\tif v != nil && v.ScheduledTimeNanos != nil {\n\t\treturn *v.ScheduledTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetScheduledTimeNanos returns true if ScheduledTimeNanos is not nil.\nfunc (v *ActivityInfo) IsSetScheduledTimeNanos() bool {\n\treturn v != nil && v.ScheduledTimeNanos != nil\n}\n\n// GetStartedID returns the value of StartedID if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartedID() (o int64) {\n\tif v != nil && v.StartedID != nil {\n\t\treturn *v.StartedID\n\t}\n\n\treturn\n}\n\n// IsSetStartedID returns true if StartedID is not nil.\nfunc (v *ActivityInfo) IsSetStartedID() bool {\n\treturn v != nil && v.StartedID != nil\n}\n\n// GetStartedEvent returns the value of StartedEvent if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartedEvent() (o []byte) {\n\tif v != nil && v.StartedEvent != nil {\n\t\treturn v.StartedEvent\n\t}\n\n\treturn\n}\n\n// IsSetStartedEvent returns true if StartedEvent is not nil.\nfunc (v *ActivityInfo) IsSetStartedEvent() bool {\n\treturn v != nil && v.StartedEvent != nil\n}\n\n// GetStartedEventEncoding returns the value of StartedEventEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartedEventEncoding() (o string) {\n\tif v != nil && v.StartedEventEncoding != nil {\n\t\treturn *v.StartedEventEncoding\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventEncoding returns true if StartedEventEncoding is not nil.\nfunc (v *ActivityInfo) IsSetStartedEventEncoding() bool {\n\treturn v != nil && v.StartedEventEncoding != nil\n}\n\n// GetStartedTimeNanos returns the value of StartedTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartedTimeNanos() (o int64) {\n\tif v != nil && v.StartedTimeNanos != nil {\n\t\treturn *v.StartedTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetStartedTimeNanos returns true if StartedTimeNanos is not nil.\nfunc (v *ActivityInfo) IsSetStartedTimeNanos() bool {\n\treturn v != nil && v.StartedTimeNanos != nil\n}\n\n// GetActivityID returns the value of ActivityID if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetActivityID() (o string) {\n\tif v != nil && v.ActivityID != nil {\n\t\treturn *v.ActivityID\n\t}\n\n\treturn\n}\n\n// IsSetActivityID returns true if ActivityID is not nil.\nfunc (v *ActivityInfo) IsSetActivityID() bool {\n\treturn v != nil && v.ActivityID != nil\n}\n\n// GetRequestID returns the value of RequestID if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRequestID() (o string) {\n\tif v != nil && v.RequestID != nil {\n\t\treturn *v.RequestID\n\t}\n\n\treturn\n}\n\n// IsSetRequestID returns true if RequestID is not nil.\nfunc (v *ActivityInfo) IsSetRequestID() bool {\n\treturn v != nil && v.RequestID != nil\n}\n\n// GetScheduleToStartTimeoutSeconds returns the value of ScheduleToStartTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToStartTimeoutSeconds returns true if ScheduleToStartTimeoutSeconds is not nil.\nfunc (v *ActivityInfo) IsSetScheduleToStartTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToStartTimeoutSeconds != nil\n}\n\n// GetScheduleToCloseTimeoutSeconds returns the value of ScheduleToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetScheduleToCloseTimeoutSeconds returns true if ScheduleToCloseTimeoutSeconds is not nil.\nfunc (v *ActivityInfo) IsSetScheduleToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.ScheduleToCloseTimeoutSeconds != nil\n}\n\n// GetStartToCloseTimeoutSeconds returns the value of StartToCloseTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetStartToCloseTimeoutSeconds returns true if StartToCloseTimeoutSeconds is not nil.\nfunc (v *ActivityInfo) IsSetStartToCloseTimeoutSeconds() bool {\n\treturn v != nil && v.StartToCloseTimeoutSeconds != nil\n}\n\n// GetHeartbeatTimeoutSeconds returns the value of HeartbeatTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetHeartbeatTimeoutSeconds returns true if HeartbeatTimeoutSeconds is not nil.\nfunc (v *ActivityInfo) IsSetHeartbeatTimeoutSeconds() bool {\n\treturn v != nil && v.HeartbeatTimeoutSeconds != nil\n}\n\n// GetCancelRequested returns the value of CancelRequested if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetCancelRequested() (o bool) {\n\tif v != nil && v.CancelRequested != nil {\n\t\treturn *v.CancelRequested\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequested returns true if CancelRequested is not nil.\nfunc (v *ActivityInfo) IsSetCancelRequested() bool {\n\treturn v != nil && v.CancelRequested != nil\n}\n\n// GetCancelRequestID returns the value of CancelRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetCancelRequestID() (o int64) {\n\tif v != nil && v.CancelRequestID != nil {\n\t\treturn *v.CancelRequestID\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequestID returns true if CancelRequestID is not nil.\nfunc (v *ActivityInfo) IsSetCancelRequestID() bool {\n\treturn v != nil && v.CancelRequestID != nil\n}\n\n// GetTimerTaskStatus returns the value of TimerTaskStatus if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetTimerTaskStatus() (o int32) {\n\tif v != nil && v.TimerTaskStatus != nil {\n\t\treturn *v.TimerTaskStatus\n\t}\n\n\treturn\n}\n\n// IsSetTimerTaskStatus returns true if TimerTaskStatus is not nil.\nfunc (v *ActivityInfo) IsSetTimerTaskStatus() bool {\n\treturn v != nil && v.TimerTaskStatus != nil\n}\n\n// GetAttempt returns the value of Attempt if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetAttempt() (o int32) {\n\tif v != nil && v.Attempt != nil {\n\t\treturn *v.Attempt\n\t}\n\n\treturn\n}\n\n// IsSetAttempt returns true if Attempt is not nil.\nfunc (v *ActivityInfo) IsSetAttempt() bool {\n\treturn v != nil && v.Attempt != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *ActivityInfo) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTaskListKind returns the value of TaskListKind if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetTaskListKind() (o shared.TaskListKind) {\n\tif v != nil && v.TaskListKind != nil {\n\t\treturn *v.TaskListKind\n\t}\n\n\treturn\n}\n\n// IsSetTaskListKind returns true if TaskListKind is not nil.\nfunc (v *ActivityInfo) IsSetTaskListKind() bool {\n\treturn v != nil && v.TaskListKind != nil\n}\n\n// GetStartedIdentity returns the value of StartedIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetStartedIdentity() (o string) {\n\tif v != nil && v.StartedIdentity != nil {\n\t\treturn *v.StartedIdentity\n\t}\n\n\treturn\n}\n\n// IsSetStartedIdentity returns true if StartedIdentity is not nil.\nfunc (v *ActivityInfo) IsSetStartedIdentity() bool {\n\treturn v != nil && v.StartedIdentity != nil\n}\n\n// GetHasRetryPolicy returns the value of HasRetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetHasRetryPolicy() (o bool) {\n\tif v != nil && v.HasRetryPolicy != nil {\n\t\treturn *v.HasRetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetHasRetryPolicy returns true if HasRetryPolicy is not nil.\nfunc (v *ActivityInfo) IsSetHasRetryPolicy() bool {\n\treturn v != nil && v.HasRetryPolicy != nil\n}\n\n// GetRetryInitialIntervalSeconds returns the value of RetryInitialIntervalSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryInitialIntervalSeconds() (o int32) {\n\tif v != nil && v.RetryInitialIntervalSeconds != nil {\n\t\treturn *v.RetryInitialIntervalSeconds\n\t}\n\n\treturn\n}\n\n// IsSetRetryInitialIntervalSeconds returns true if RetryInitialIntervalSeconds is not nil.\nfunc (v *ActivityInfo) IsSetRetryInitialIntervalSeconds() bool {\n\treturn v != nil && v.RetryInitialIntervalSeconds != nil\n}\n\n// GetRetryMaximumIntervalSeconds returns the value of RetryMaximumIntervalSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryMaximumIntervalSeconds() (o int32) {\n\tif v != nil && v.RetryMaximumIntervalSeconds != nil {\n\t\treturn *v.RetryMaximumIntervalSeconds\n\t}\n\n\treturn\n}\n\n// IsSetRetryMaximumIntervalSeconds returns true if RetryMaximumIntervalSeconds is not nil.\nfunc (v *ActivityInfo) IsSetRetryMaximumIntervalSeconds() bool {\n\treturn v != nil && v.RetryMaximumIntervalSeconds != nil\n}\n\n// GetRetryMaximumAttempts returns the value of RetryMaximumAttempts if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryMaximumAttempts() (o int32) {\n\tif v != nil && v.RetryMaximumAttempts != nil {\n\t\treturn *v.RetryMaximumAttempts\n\t}\n\n\treturn\n}\n\n// IsSetRetryMaximumAttempts returns true if RetryMaximumAttempts is not nil.\nfunc (v *ActivityInfo) IsSetRetryMaximumAttempts() bool {\n\treturn v != nil && v.RetryMaximumAttempts != nil\n}\n\n// GetRetryExpirationTimeNanos returns the value of RetryExpirationTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryExpirationTimeNanos() (o int64) {\n\tif v != nil && v.RetryExpirationTimeNanos != nil {\n\t\treturn *v.RetryExpirationTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetRetryExpirationTimeNanos returns true if RetryExpirationTimeNanos is not nil.\nfunc (v *ActivityInfo) IsSetRetryExpirationTimeNanos() bool {\n\treturn v != nil && v.RetryExpirationTimeNanos != nil\n}\n\n// GetRetryBackoffCoefficient returns the value of RetryBackoffCoefficient if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryBackoffCoefficient() (o float64) {\n\tif v != nil && v.RetryBackoffCoefficient != nil {\n\t\treturn *v.RetryBackoffCoefficient\n\t}\n\n\treturn\n}\n\n// IsSetRetryBackoffCoefficient returns true if RetryBackoffCoefficient is not nil.\nfunc (v *ActivityInfo) IsSetRetryBackoffCoefficient() bool {\n\treturn v != nil && v.RetryBackoffCoefficient != nil\n}\n\n// GetRetryNonRetryableErrors returns the value of RetryNonRetryableErrors if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryNonRetryableErrors() (o []string) {\n\tif v != nil && v.RetryNonRetryableErrors != nil {\n\t\treturn v.RetryNonRetryableErrors\n\t}\n\n\treturn\n}\n\n// IsSetRetryNonRetryableErrors returns true if RetryNonRetryableErrors is not nil.\nfunc (v *ActivityInfo) IsSetRetryNonRetryableErrors() bool {\n\treturn v != nil && v.RetryNonRetryableErrors != nil\n}\n\n// GetRetryLastFailureReason returns the value of RetryLastFailureReason if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryLastFailureReason() (o string) {\n\tif v != nil && v.RetryLastFailureReason != nil {\n\t\treturn *v.RetryLastFailureReason\n\t}\n\n\treturn\n}\n\n// IsSetRetryLastFailureReason returns true if RetryLastFailureReason is not nil.\nfunc (v *ActivityInfo) IsSetRetryLastFailureReason() bool {\n\treturn v != nil && v.RetryLastFailureReason != nil\n}\n\n// GetRetryLastWorkerIdentity returns the value of RetryLastWorkerIdentity if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryLastWorkerIdentity() (o string) {\n\tif v != nil && v.RetryLastWorkerIdentity != nil {\n\t\treturn *v.RetryLastWorkerIdentity\n\t}\n\n\treturn\n}\n\n// IsSetRetryLastWorkerIdentity returns true if RetryLastWorkerIdentity is not nil.\nfunc (v *ActivityInfo) IsSetRetryLastWorkerIdentity() bool {\n\treturn v != nil && v.RetryLastWorkerIdentity != nil\n}\n\n// GetRetryLastFailureDetails returns the value of RetryLastFailureDetails if it is set or its\n// zero value if it is unset.\nfunc (v *ActivityInfo) GetRetryLastFailureDetails() (o []byte) {\n\tif v != nil && v.RetryLastFailureDetails != nil {\n\t\treturn v.RetryLastFailureDetails\n\t}\n\n\treturn\n}\n\n// IsSetRetryLastFailureDetails returns true if RetryLastFailureDetails is not nil.\nfunc (v *ActivityInfo) IsSetRetryLastFailureDetails() bool {\n\treturn v != nil && v.RetryLastFailureDetails != nil\n}\n\ntype AsyncRequestMessage struct {\n\tPartitionKey *string           `json:\"partitionKey,omitempty\"`\n\tType         *AsyncRequestType `json:\"type,omitempty\"`\n\tHeader       *shared.Header    `json:\"header,omitempty\"`\n\tEncoding     *string           `json:\"encoding,omitempty\"`\n\tPayload      []byte            `json:\"payload,omitempty\"`\n}\n\n// ToWire translates a AsyncRequestMessage struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *AsyncRequestMessage) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.PartitionKey != nil {\n\t\tw, err = wire.NewValueString(*(v.PartitionKey)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tw, err = v.Type.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tw, err = v.Header.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.Encoding != nil {\n\t\tw, err = wire.NewValueString(*(v.Encoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.Payload != nil {\n\t\tw, err = wire.NewValueBinary(v.Payload), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _AsyncRequestType_Read(w wire.Value) (AsyncRequestType, error) {\n\tvar v AsyncRequestType\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\nfunc _Header_Read(w wire.Value) (*shared.Header, error) {\n\tvar v shared.Header\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a AsyncRequestMessage struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a AsyncRequestMessage struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v AsyncRequestMessage\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *AsyncRequestMessage) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PartitionKey = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x AsyncRequestType\n\t\t\t\tx, err = _AsyncRequestType_Read(field.Value)\n\t\t\t\tv.Type = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.Header, err = _Header_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Encoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Payload, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a AsyncRequestMessage struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a AsyncRequestMessage struct could not be encoded.\nfunc (v *AsyncRequestMessage) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.PartitionKey != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PartitionKey)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Type != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Type.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Header != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Header.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Encoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Encoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Payload != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Payload); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _AsyncRequestType_Decode(sr stream.Reader) (AsyncRequestType, error) {\n\tvar v AsyncRequestType\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\nfunc _Header_Decode(sr stream.Reader) (*shared.Header, error) {\n\tvar v shared.Header\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a AsyncRequestMessage struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a AsyncRequestMessage struct could not be generated from the wire\n// representation.\nfunc (v *AsyncRequestMessage) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PartitionKey = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI32:\n\t\t\tvar x AsyncRequestType\n\t\t\tx, err = _AsyncRequestType_Decode(sr)\n\t\t\tv.Type = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TStruct:\n\t\t\tv.Header, err = _Header_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Encoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TBinary:\n\t\t\tv.Payload, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a AsyncRequestMessage\n// struct.\nfunc (v *AsyncRequestMessage) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.PartitionKey != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionKey: %v\", *(v.PartitionKey))\n\t\ti++\n\t}\n\tif v.Type != nil {\n\t\tfields[i] = fmt.Sprintf(\"Type: %v\", *(v.Type))\n\t\ti++\n\t}\n\tif v.Header != nil {\n\t\tfields[i] = fmt.Sprintf(\"Header: %v\", v.Header)\n\t\ti++\n\t}\n\tif v.Encoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"Encoding: %v\", *(v.Encoding))\n\t\ti++\n\t}\n\tif v.Payload != nil {\n\t\tfields[i] = fmt.Sprintf(\"Payload: %v\", v.Payload)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"AsyncRequestMessage{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _AsyncRequestType_EqualsPtr(lhs, rhs *AsyncRequestType) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this AsyncRequestMessage match the\n// provided AsyncRequestMessage.\n//\n// This function performs a deep comparison.\nfunc (v *AsyncRequestMessage) Equals(rhs *AsyncRequestMessage) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PartitionKey, rhs.PartitionKey) {\n\t\treturn false\n\t}\n\tif !_AsyncRequestType_EqualsPtr(v.Type, rhs.Type) {\n\t\treturn false\n\t}\n\tif !((v.Header == nil && rhs.Header == nil) || (v.Header != nil && rhs.Header != nil && v.Header.Equals(rhs.Header))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Encoding, rhs.Encoding) {\n\t\treturn false\n\t}\n\tif !((v.Payload == nil && rhs.Payload == nil) || (v.Payload != nil && rhs.Payload != nil && bytes.Equal(v.Payload, rhs.Payload))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AsyncRequestMessage.\nfunc (v *AsyncRequestMessage) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.PartitionKey != nil {\n\t\tenc.AddString(\"partitionKey\", *v.PartitionKey)\n\t}\n\tif v.Type != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"type\", *v.Type))\n\t}\n\tif v.Header != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"header\", v.Header))\n\t}\n\tif v.Encoding != nil {\n\t\tenc.AddString(\"encoding\", *v.Encoding)\n\t}\n\tif v.Payload != nil {\n\t\tenc.AddString(\"payload\", base64.StdEncoding.EncodeToString(v.Payload))\n\t}\n\treturn err\n}\n\n// GetPartitionKey returns the value of PartitionKey if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncRequestMessage) GetPartitionKey() (o string) {\n\tif v != nil && v.PartitionKey != nil {\n\t\treturn *v.PartitionKey\n\t}\n\n\treturn\n}\n\n// IsSetPartitionKey returns true if PartitionKey is not nil.\nfunc (v *AsyncRequestMessage) IsSetPartitionKey() bool {\n\treturn v != nil && v.PartitionKey != nil\n}\n\n// GetType returns the value of Type if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncRequestMessage) GetType() (o AsyncRequestType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\n\treturn\n}\n\n// IsSetType returns true if Type is not nil.\nfunc (v *AsyncRequestMessage) IsSetType() bool {\n\treturn v != nil && v.Type != nil\n}\n\n// GetHeader returns the value of Header if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncRequestMessage) GetHeader() (o *shared.Header) {\n\tif v != nil && v.Header != nil {\n\t\treturn v.Header\n\t}\n\n\treturn\n}\n\n// IsSetHeader returns true if Header is not nil.\nfunc (v *AsyncRequestMessage) IsSetHeader() bool {\n\treturn v != nil && v.Header != nil\n}\n\n// GetEncoding returns the value of Encoding if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncRequestMessage) GetEncoding() (o string) {\n\tif v != nil && v.Encoding != nil {\n\t\treturn *v.Encoding\n\t}\n\n\treturn\n}\n\n// IsSetEncoding returns true if Encoding is not nil.\nfunc (v *AsyncRequestMessage) IsSetEncoding() bool {\n\treturn v != nil && v.Encoding != nil\n}\n\n// GetPayload returns the value of Payload if it is set or its\n// zero value if it is unset.\nfunc (v *AsyncRequestMessage) GetPayload() (o []byte) {\n\tif v != nil && v.Payload != nil {\n\t\treturn v.Payload\n\t}\n\n\treturn\n}\n\n// IsSetPayload returns true if Payload is not nil.\nfunc (v *AsyncRequestMessage) IsSetPayload() bool {\n\treturn v != nil && v.Payload != nil\n}\n\ntype AsyncRequestType int32\n\nconst (\n\tAsyncRequestTypeStartWorkflowExecutionAsyncRequest           AsyncRequestType = 0\n\tAsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest AsyncRequestType = 1\n)\n\n// AsyncRequestType_Values returns all recognized values of AsyncRequestType.\nfunc AsyncRequestType_Values() []AsyncRequestType {\n\treturn []AsyncRequestType{\n\t\tAsyncRequestTypeStartWorkflowExecutionAsyncRequest,\n\t\tAsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest,\n\t}\n}\n\n// UnmarshalText tries to decode AsyncRequestType from a byte slice\n// containing its name.\n//\n//\tvar v AsyncRequestType\n//\terr := v.UnmarshalText([]byte(\"StartWorkflowExecutionAsyncRequest\"))\nfunc (v *AsyncRequestType) UnmarshalText(value []byte) error {\n\tswitch s := string(value); s {\n\tcase \"StartWorkflowExecutionAsyncRequest\":\n\t\t*v = AsyncRequestTypeStartWorkflowExecutionAsyncRequest\n\t\treturn nil\n\tcase \"SignalWithStartWorkflowExecutionAsyncRequest\":\n\t\t*v = AsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"AsyncRequestType\", err)\n\t\t}\n\t\t*v = AsyncRequestType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes AsyncRequestType to text.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements the TextMarshaler interface.\nfunc (v AsyncRequestType) MarshalText() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn []byte(\"StartWorkflowExecutionAsyncRequest\"), nil\n\tcase 1:\n\t\treturn []byte(\"SignalWithStartWorkflowExecutionAsyncRequest\"), nil\n\t}\n\treturn []byte(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of AsyncRequestType.\n// Enums are logged as objects, where the value is logged with key \"value\", and\n// if this value's name is known, the name is logged with key \"name\".\nfunc (v AsyncRequestType) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddInt32(\"value\", int32(v))\n\tswitch int32(v) {\n\tcase 0:\n\t\tenc.AddString(\"name\", \"StartWorkflowExecutionAsyncRequest\")\n\tcase 1:\n\t\tenc.AddString(\"name\", \"SignalWithStartWorkflowExecutionAsyncRequest\")\n\t}\n\treturn nil\n}\n\n// Ptr returns a pointer to this enum value.\nfunc (v AsyncRequestType) Ptr() *AsyncRequestType {\n\treturn &v\n}\n\n// Encode encodes AsyncRequestType directly to bytes.\n//\n//\tsWriter := BinaryStreamer.Writer(writer)\n//\n//\tvar v AsyncRequestType\n//\treturn v.Encode(sWriter)\nfunc (v AsyncRequestType) Encode(sw stream.Writer) error {\n\treturn sw.WriteInt32(int32(v))\n}\n\n// ToWire translates AsyncRequestType into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// Enums are represented as 32-bit integers over the wire.\nfunc (v AsyncRequestType) ToWire() (wire.Value, error) {\n\treturn wire.NewValueI32(int32(v)), nil\n}\n\n// FromWire deserializes AsyncRequestType from its Thrift-level\n// representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TI32)\n//\tif err != nil {\n//\t  return AsyncRequestType(0), err\n//\t}\n//\n//\tvar v AsyncRequestType\n//\tif err := v.FromWire(x); err != nil {\n//\t  return AsyncRequestType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *AsyncRequestType) FromWire(w wire.Value) error {\n\t*v = (AsyncRequestType)(w.GetI32())\n\treturn nil\n}\n\n// Decode reads off the encoded AsyncRequestType directly off of the wire.\n//\n//\tsReader := BinaryStreamer.Reader(reader)\n//\n//\tvar v AsyncRequestType\n//\tif err := v.Decode(sReader); err != nil {\n//\t  return AsyncRequestType(0), err\n//\t}\n//\treturn v, nil\nfunc (v *AsyncRequestType) Decode(sr stream.Reader) error {\n\ti, err := sr.ReadInt32()\n\tif err != nil {\n\t\treturn err\n\t}\n\t*v = (AsyncRequestType)(i)\n\treturn nil\n}\n\n// String returns a readable string representation of AsyncRequestType.\nfunc (v AsyncRequestType) String() string {\n\tw := int32(v)\n\tswitch w {\n\tcase 0:\n\t\treturn \"StartWorkflowExecutionAsyncRequest\"\n\tcase 1:\n\t\treturn \"SignalWithStartWorkflowExecutionAsyncRequest\"\n\t}\n\treturn fmt.Sprintf(\"AsyncRequestType(%d)\", w)\n}\n\n// Equals returns true if this AsyncRequestType value matches the provided\n// value.\nfunc (v AsyncRequestType) Equals(rhs AsyncRequestType) bool {\n\treturn v == rhs\n}\n\n// MarshalJSON serializes AsyncRequestType into JSON.\n//\n// If the enum value is recognized, its name is returned.\n// Otherwise, its integer value is returned.\n//\n// This implements json.Marshaler.\nfunc (v AsyncRequestType) MarshalJSON() ([]byte, error) {\n\tswitch int32(v) {\n\tcase 0:\n\t\treturn ([]byte)(\"\\\"StartWorkflowExecutionAsyncRequest\\\"\"), nil\n\tcase 1:\n\t\treturn ([]byte)(\"\\\"SignalWithStartWorkflowExecutionAsyncRequest\\\"\"), nil\n\t}\n\treturn ([]byte)(strconv.FormatInt(int64(v), 10)), nil\n}\n\n// UnmarshalJSON attempts to decode AsyncRequestType from its JSON\n// representation.\n//\n// This implementation supports both, numeric and string inputs. If a\n// string is provided, it must be a known enum name.\n//\n// This implements json.Unmarshaler.\nfunc (v *AsyncRequestType) UnmarshalJSON(text []byte) error {\n\td := json.NewDecoder(bytes.NewReader(text))\n\td.UseNumber()\n\tt, err := d.Token()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch w := t.(type) {\n\tcase json.Number:\n\t\tx, err := w.Int64()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif x > math.MaxInt32 {\n\t\t\treturn fmt.Errorf(\"enum overflow from JSON %q for %q\", text, \"AsyncRequestType\")\n\t\t}\n\t\tif x < math.MinInt32 {\n\t\t\treturn fmt.Errorf(\"enum underflow from JSON %q for %q\", text, \"AsyncRequestType\")\n\t\t}\n\t\t*v = (AsyncRequestType)(x)\n\t\treturn nil\n\tcase string:\n\t\treturn v.UnmarshalText([]byte(w))\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid JSON value %q (%T) to unmarshal into %q\", t, t, \"AsyncRequestType\")\n\t}\n}\n\ntype ChildExecutionInfo struct {\n\tVersion                *int64  `json:\"version,omitempty\"`\n\tInitiatedEventBatchID  *int64  `json:\"initiatedEventBatchID,omitempty\"`\n\tStartedID              *int64  `json:\"startedID,omitempty\"`\n\tInitiatedEvent         []byte  `json:\"initiatedEvent,omitempty\"`\n\tInitiatedEventEncoding *string `json:\"initiatedEventEncoding,omitempty\"`\n\tStartedWorkflowID      *string `json:\"startedWorkflowID,omitempty\"`\n\tStartedRunID           []byte  `json:\"startedRunID,omitempty\"`\n\tStartedEvent           []byte  `json:\"startedEvent,omitempty\"`\n\tStartedEventEncoding   *string `json:\"startedEventEncoding,omitempty\"`\n\tCreateRequestID        *string `json:\"createRequestID,omitempty\"`\n\tDomainID               *string `json:\"domainID,omitempty\"`\n\tDomainName             *string `json:\"domainName,omitempty\"`\n\tWorkflowTypeName       *string `json:\"workflowTypeName,omitempty\"`\n\tParentClosePolicy      *int32  `json:\"parentClosePolicy,omitempty\"`\n}\n\n// ToWire translates a ChildExecutionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ChildExecutionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [14]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventBatchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEvent != nil {\n\t\tw, err = wire.NewValueBinary(v.InitiatedEvent), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.InitiatedEventEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.StartedWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.StartedWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.StartedRunID != nil {\n\t\tw, err = wire.NewValueBinary(v.StartedRunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tw, err = wire.NewValueBinary(v.StartedEvent), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.StartedEventEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.CreateRequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.CreateRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 29, Value: w}\n\t\ti++\n\t}\n\tif v.DomainName != nil {\n\t\tw, err = wire.NewValueString(*(v.DomainName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowTypeName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 32, Value: w}\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tw, err = wire.NewValueI32(*(v.ParentClosePolicy)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 35, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ChildExecutionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ChildExecutionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ChildExecutionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ChildExecutionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventBatchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.InitiatedEvent, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.InitiatedEventEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartedWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.StartedRunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.StartedEvent, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StartedEventEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CreateRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 29:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DomainName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 32:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowTypeName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 35:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.ParentClosePolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ChildExecutionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ChildExecutionInfo struct could not be encoded.\nfunc (v *ChildExecutionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventBatchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventBatchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.InitiatedEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.InitiatedEventEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartedWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.StartedRunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.StartedEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedEventEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StartedEventEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreateRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CreateRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 29, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DomainName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowTypeName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 32, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowTypeName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentClosePolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 35, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.ParentClosePolicy)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ChildExecutionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ChildExecutionInfo struct could not be generated from the wire\n// representation.\nfunc (v *ChildExecutionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventBatchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TBinary:\n\t\t\tv.InitiatedEvent, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.InitiatedEventEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartedWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TBinary:\n\t\t\tv.StartedRunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TBinary:\n\t\t\tv.StartedEvent, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StartedEventEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CreateRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 29 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DomainName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 32 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowTypeName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 35 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.ParentClosePolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ChildExecutionInfo\n// struct.\nfunc (v *ChildExecutionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [14]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventBatchID: %v\", *(v.InitiatedEventBatchID))\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedID: %v\", *(v.StartedID))\n\t\ti++\n\t}\n\tif v.InitiatedEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEvent: %v\", v.InitiatedEvent)\n\t\ti++\n\t}\n\tif v.InitiatedEventEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventEncoding: %v\", *(v.InitiatedEventEncoding))\n\t\ti++\n\t}\n\tif v.StartedWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedWorkflowID: %v\", *(v.StartedWorkflowID))\n\t\ti++\n\t}\n\tif v.StartedRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedRunID: %v\", v.StartedRunID)\n\t\ti++\n\t}\n\tif v.StartedEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEvent: %v\", v.StartedEvent)\n\t\ti++\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedEventEncoding: %v\", *(v.StartedEventEncoding))\n\t\ti++\n\t}\n\tif v.CreateRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreateRequestID: %v\", *(v.CreateRequestID))\n\t\ti++\n\t}\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", *(v.DomainID))\n\t\ti++\n\t}\n\tif v.DomainName != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainName: %v\", *(v.DomainName))\n\t\ti++\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowTypeName: %v\", *(v.WorkflowTypeName))\n\t\ti++\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentClosePolicy: %v\", *(v.ParentClosePolicy))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ChildExecutionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ChildExecutionInfo match the\n// provided ChildExecutionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ChildExecutionInfo) Equals(rhs *ChildExecutionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventBatchID, rhs.InitiatedEventBatchID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedID, rhs.StartedID) {\n\t\treturn false\n\t}\n\tif !((v.InitiatedEvent == nil && rhs.InitiatedEvent == nil) || (v.InitiatedEvent != nil && rhs.InitiatedEvent != nil && bytes.Equal(v.InitiatedEvent, rhs.InitiatedEvent))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.InitiatedEventEncoding, rhs.InitiatedEventEncoding) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartedWorkflowID, rhs.StartedWorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.StartedRunID == nil && rhs.StartedRunID == nil) || (v.StartedRunID != nil && rhs.StartedRunID != nil && bytes.Equal(v.StartedRunID, rhs.StartedRunID))) {\n\t\treturn false\n\t}\n\tif !((v.StartedEvent == nil && rhs.StartedEvent == nil) || (v.StartedEvent != nil && rhs.StartedEvent != nil && bytes.Equal(v.StartedEvent, rhs.StartedEvent))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StartedEventEncoding, rhs.StartedEventEncoding) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CreateRequestID, rhs.CreateRequestID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainID, rhs.DomainID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DomainName, rhs.DomainName) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowTypeName, rhs.WorkflowTypeName) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.ParentClosePolicy, rhs.ParentClosePolicy) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ChildExecutionInfo.\nfunc (v *ChildExecutionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tenc.AddInt64(\"initiatedEventBatchID\", *v.InitiatedEventBatchID)\n\t}\n\tif v.StartedID != nil {\n\t\tenc.AddInt64(\"startedID\", *v.StartedID)\n\t}\n\tif v.InitiatedEvent != nil {\n\t\tenc.AddString(\"initiatedEvent\", base64.StdEncoding.EncodeToString(v.InitiatedEvent))\n\t}\n\tif v.InitiatedEventEncoding != nil {\n\t\tenc.AddString(\"initiatedEventEncoding\", *v.InitiatedEventEncoding)\n\t}\n\tif v.StartedWorkflowID != nil {\n\t\tenc.AddString(\"startedWorkflowID\", *v.StartedWorkflowID)\n\t}\n\tif v.StartedRunID != nil {\n\t\tenc.AddString(\"startedRunID\", base64.StdEncoding.EncodeToString(v.StartedRunID))\n\t}\n\tif v.StartedEvent != nil {\n\t\tenc.AddString(\"startedEvent\", base64.StdEncoding.EncodeToString(v.StartedEvent))\n\t}\n\tif v.StartedEventEncoding != nil {\n\t\tenc.AddString(\"startedEventEncoding\", *v.StartedEventEncoding)\n\t}\n\tif v.CreateRequestID != nil {\n\t\tenc.AddString(\"createRequestID\", *v.CreateRequestID)\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", *v.DomainID)\n\t}\n\tif v.DomainName != nil {\n\t\tenc.AddString(\"domainName\", *v.DomainName)\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tenc.AddString(\"workflowTypeName\", *v.WorkflowTypeName)\n\t}\n\tif v.ParentClosePolicy != nil {\n\t\tenc.AddInt32(\"parentClosePolicy\", *v.ParentClosePolicy)\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *ChildExecutionInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetInitiatedEventBatchID returns the value of InitiatedEventBatchID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetInitiatedEventBatchID() (o int64) {\n\tif v != nil && v.InitiatedEventBatchID != nil {\n\t\treturn *v.InitiatedEventBatchID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventBatchID returns true if InitiatedEventBatchID is not nil.\nfunc (v *ChildExecutionInfo) IsSetInitiatedEventBatchID() bool {\n\treturn v != nil && v.InitiatedEventBatchID != nil\n}\n\n// GetStartedID returns the value of StartedID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetStartedID() (o int64) {\n\tif v != nil && v.StartedID != nil {\n\t\treturn *v.StartedID\n\t}\n\n\treturn\n}\n\n// IsSetStartedID returns true if StartedID is not nil.\nfunc (v *ChildExecutionInfo) IsSetStartedID() bool {\n\treturn v != nil && v.StartedID != nil\n}\n\n// GetInitiatedEvent returns the value of InitiatedEvent if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetInitiatedEvent() (o []byte) {\n\tif v != nil && v.InitiatedEvent != nil {\n\t\treturn v.InitiatedEvent\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEvent returns true if InitiatedEvent is not nil.\nfunc (v *ChildExecutionInfo) IsSetInitiatedEvent() bool {\n\treturn v != nil && v.InitiatedEvent != nil\n}\n\n// GetInitiatedEventEncoding returns the value of InitiatedEventEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetInitiatedEventEncoding() (o string) {\n\tif v != nil && v.InitiatedEventEncoding != nil {\n\t\treturn *v.InitiatedEventEncoding\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventEncoding returns true if InitiatedEventEncoding is not nil.\nfunc (v *ChildExecutionInfo) IsSetInitiatedEventEncoding() bool {\n\treturn v != nil && v.InitiatedEventEncoding != nil\n}\n\n// GetStartedWorkflowID returns the value of StartedWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetStartedWorkflowID() (o string) {\n\tif v != nil && v.StartedWorkflowID != nil {\n\t\treturn *v.StartedWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetStartedWorkflowID returns true if StartedWorkflowID is not nil.\nfunc (v *ChildExecutionInfo) IsSetStartedWorkflowID() bool {\n\treturn v != nil && v.StartedWorkflowID != nil\n}\n\n// GetStartedRunID returns the value of StartedRunID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetStartedRunID() (o []byte) {\n\tif v != nil && v.StartedRunID != nil {\n\t\treturn v.StartedRunID\n\t}\n\n\treturn\n}\n\n// IsSetStartedRunID returns true if StartedRunID is not nil.\nfunc (v *ChildExecutionInfo) IsSetStartedRunID() bool {\n\treturn v != nil && v.StartedRunID != nil\n}\n\n// GetStartedEvent returns the value of StartedEvent if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetStartedEvent() (o []byte) {\n\tif v != nil && v.StartedEvent != nil {\n\t\treturn v.StartedEvent\n\t}\n\n\treturn\n}\n\n// IsSetStartedEvent returns true if StartedEvent is not nil.\nfunc (v *ChildExecutionInfo) IsSetStartedEvent() bool {\n\treturn v != nil && v.StartedEvent != nil\n}\n\n// GetStartedEventEncoding returns the value of StartedEventEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetStartedEventEncoding() (o string) {\n\tif v != nil && v.StartedEventEncoding != nil {\n\t\treturn *v.StartedEventEncoding\n\t}\n\n\treturn\n}\n\n// IsSetStartedEventEncoding returns true if StartedEventEncoding is not nil.\nfunc (v *ChildExecutionInfo) IsSetStartedEventEncoding() bool {\n\treturn v != nil && v.StartedEventEncoding != nil\n}\n\n// GetCreateRequestID returns the value of CreateRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetCreateRequestID() (o string) {\n\tif v != nil && v.CreateRequestID != nil {\n\t\treturn *v.CreateRequestID\n\t}\n\n\treturn\n}\n\n// IsSetCreateRequestID returns true if CreateRequestID is not nil.\nfunc (v *ChildExecutionInfo) IsSetCreateRequestID() bool {\n\treturn v != nil && v.CreateRequestID != nil\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetDomainID() (o string) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn *v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *ChildExecutionInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetDomainName returns the value of DomainName if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetDomainName() (o string) {\n\tif v != nil && v.DomainName != nil {\n\t\treturn *v.DomainName\n\t}\n\n\treturn\n}\n\n// IsSetDomainName returns true if DomainName is not nil.\nfunc (v *ChildExecutionInfo) IsSetDomainName() bool {\n\treturn v != nil && v.DomainName != nil\n}\n\n// GetWorkflowTypeName returns the value of WorkflowTypeName if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetWorkflowTypeName() (o string) {\n\tif v != nil && v.WorkflowTypeName != nil {\n\t\treturn *v.WorkflowTypeName\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowTypeName returns true if WorkflowTypeName is not nil.\nfunc (v *ChildExecutionInfo) IsSetWorkflowTypeName() bool {\n\treturn v != nil && v.WorkflowTypeName != nil\n}\n\n// GetParentClosePolicy returns the value of ParentClosePolicy if it is set or its\n// zero value if it is unset.\nfunc (v *ChildExecutionInfo) GetParentClosePolicy() (o int32) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\n\treturn\n}\n\n// IsSetParentClosePolicy returns true if ParentClosePolicy is not nil.\nfunc (v *ChildExecutionInfo) IsSetParentClosePolicy() bool {\n\treturn v != nil && v.ParentClosePolicy != nil\n}\n\ntype DomainInfo struct {\n\tName                                 *string           `json:\"name,omitempty\"`\n\tDescription                          *string           `json:\"description,omitempty\"`\n\tOwner                                *string           `json:\"owner,omitempty\"`\n\tStatus                               *int32            `json:\"status,omitempty\"`\n\tRetentionDays                        *int16            `json:\"retentionDays,omitempty\"`\n\tEmitMetric                           *bool             `json:\"emitMetric,omitempty\"`\n\tArchivalBucket                       *string           `json:\"archivalBucket,omitempty\"`\n\tArchivalStatus                       *int16            `json:\"archivalStatus,omitempty\"`\n\tConfigVersion                        *int64            `json:\"configVersion,omitempty\"`\n\tNotificationVersion                  *int64            `json:\"notificationVersion,omitempty\"`\n\tFailoverNotificationVersion          *int64            `json:\"failoverNotificationVersion,omitempty\"`\n\tFailoverVersion                      *int64            `json:\"failoverVersion,omitempty\"`\n\tActiveClusterName                    *string           `json:\"activeClusterName,omitempty\"`\n\tClusters                             []string          `json:\"clusters,omitempty\"`\n\tData                                 map[string]string `json:\"data,omitempty\"`\n\tBadBinaries                          []byte            `json:\"badBinaries,omitempty\"`\n\tBadBinariesEncoding                  *string           `json:\"badBinariesEncoding,omitempty\"`\n\tHistoryArchivalStatus                *int16            `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                   *string           `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus             *int16            `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                *string           `json:\"visibilityArchivalURI,omitempty\"`\n\tFailoverEndTime                      *int64            `json:\"failoverEndTime,omitempty\"`\n\tPreviousFailoverVersion              *int64            `json:\"previousFailoverVersion,omitempty\"`\n\tLastUpdatedTime                      *int64            `json:\"lastUpdatedTime,omitempty\"`\n\tIsolationGroupsConfiguration         []byte            `json:\"isolationGroupsConfiguration,omitempty\"`\n\tIsolationGroupsConfigurationEncoding *string           `json:\"isolationGroupsConfigurationEncoding,omitempty\"`\n\tAsyncWorkflowConfiguration           []byte            `json:\"asyncWorkflowConfiguration,omitempty\"`\n\tAsyncWorkflowConfigurationEncoding   *string           `json:\"asyncWorkflowConfigurationEncoding,omitempty\"`\n\tActiveClustersConfiguration          []byte            `json:\"activeClustersConfiguration,omitempty\"`\n\tActiveClustersConfigurationEncoding  *string           `json:\"activeClustersConfigurationEncoding,omitempty\"`\n}\n\ntype _Map_String_String_MapItemList map[string]string\n\nfunc (m _Map_String_String_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueString(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_String_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_String_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_String_MapItemList) Close() {}\n\n// ToWire translates a DomainInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *DomainInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [30]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tw, err = wire.NewValueString(*(v.Description)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tw, err = wire.NewValueString(*(v.Owner)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tw, err = wire.NewValueI32(*(v.Status)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.RetentionDays != nil {\n\t\tw, err = wire.NewValueI16(*(v.RetentionDays)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.EmitMetric != nil {\n\t\tw, err = wire.NewValueBool(*(v.EmitMetric)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ArchivalBucket != nil {\n\t\tw, err = wire.NewValueString(*(v.ArchivalBucket)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.ArchivalStatus != nil {\n\t\tw, err = wire.NewValueI16(*(v.ArchivalStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.ConfigVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.ConfigVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.NotificationVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.NotificationVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverNotificationVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverNotificationVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 32, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClusterName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.Clusters)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.Data)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\tif v.BadBinaries != nil {\n\t\tw, err = wire.NewValueBinary(v.BadBinaries), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 39, Value: w}\n\t\ti++\n\t}\n\tif v.BadBinariesEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.BadBinariesEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tw, err = wire.NewValueI16(*(v.HistoryArchivalStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 42, Value: w}\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.HistoryArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 44, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tw, err = wire.NewValueI16(*(v.VisibilityArchivalStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 46, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tw, err = wire.NewValueString(*(v.VisibilityArchivalURI)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 48, Value: w}\n\t\ti++\n\t}\n\tif v.FailoverEndTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.FailoverEndTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.PreviousFailoverVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 52, Value: w}\n\t\ti++\n\t}\n\tif v.LastUpdatedTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastUpdatedTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 54, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroupsConfiguration != nil {\n\t\tw, err = wire.NewValueBinary(v.IsolationGroupsConfiguration), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.IsolationGroupsConfigurationEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.IsolationGroupsConfigurationEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 58, Value: w}\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tw, err = wire.NewValueBinary(v.AsyncWorkflowConfiguration), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfigurationEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.AsyncWorkflowConfigurationEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 62, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClustersConfiguration != nil {\n\t\tw, err = wire.NewValueBinary(v.ActiveClustersConfiguration), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 64, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClustersConfigurationEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClustersConfigurationEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 66, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_String_String_Read(m wire.MapItemList) (map[string]string, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]string, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a DomainInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a DomainInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v DomainInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *DomainInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Description = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Owner = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.Status = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.RetentionDays = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.EmitMetric = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ArchivalBucket = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.ArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ConfigVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NotificationVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverNotificationVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 32:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClusterName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Clusters, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Data, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 39:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BadBinaries, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.BadBinariesEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 42:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.HistoryArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 44:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.HistoryArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 46:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 48:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.VisibilityArchivalURI = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FailoverEndTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 52:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.PreviousFailoverVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 54:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastUpdatedTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.IsolationGroupsConfiguration, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 58:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.IsolationGroupsConfigurationEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.AsyncWorkflowConfiguration, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 62:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.AsyncWorkflowConfigurationEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 64:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ActiveClustersConfiguration, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 66:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClustersConfigurationEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_String_Encode(val map[string]string, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a DomainInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a DomainInfo struct could not be encoded.\nfunc (v *DomainInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Description != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Description)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Owner != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Owner)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Status != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.Status)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetentionDays != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.RetentionDays)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EmitMetric != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.EmitMetric)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ArchivalBucket != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ArchivalBucket)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.ArchivalStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ConfigVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ConfigVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NotificationVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NotificationVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverNotificationVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverNotificationVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 32, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClusterName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Clusters != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.Clusters, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Data != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.Data, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadBinaries != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 39, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BadBinaries); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BadBinariesEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.BadBinariesEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 42, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.HistoryArchivalStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistoryArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 44, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.HistoryArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 46, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.VisibilityArchivalStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityArchivalURI != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 48, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.VisibilityArchivalURI)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FailoverEndTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FailoverEndTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PreviousFailoverVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 52, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.PreviousFailoverVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastUpdatedTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 54, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastUpdatedTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroupsConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.IsolationGroupsConfiguration); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.IsolationGroupsConfigurationEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 58, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.IsolationGroupsConfigurationEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.AsyncWorkflowConfiguration); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AsyncWorkflowConfigurationEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 62, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.AsyncWorkflowConfigurationEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClustersConfiguration != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 64, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ActiveClustersConfiguration); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClustersConfigurationEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 66, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClustersConfigurationEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_String_String_Decode(sr stream.Reader) (map[string]string, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]string, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a DomainInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a DomainInfo struct could not be generated from the wire\n// representation.\nfunc (v *DomainInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Description = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Owner = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.Status = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.RetentionDays = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.EmitMetric = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ArchivalBucket = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.ArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ConfigVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NotificationVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverNotificationVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 32 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClusterName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TList:\n\t\t\tv.Clusters, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TMap:\n\t\t\tv.Data, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 39 && fh.Type == wire.TBinary:\n\t\t\tv.BadBinaries, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.BadBinariesEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 42 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.HistoryArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 44 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.HistoryArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 46 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.VisibilityArchivalStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 48 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.VisibilityArchivalURI = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FailoverEndTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 52 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.PreviousFailoverVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 54 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastUpdatedTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TBinary:\n\t\t\tv.IsolationGroupsConfiguration, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 58 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.IsolationGroupsConfigurationEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.AsyncWorkflowConfiguration, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 62 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.AsyncWorkflowConfigurationEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 64 && fh.Type == wire.TBinary:\n\t\t\tv.ActiveClustersConfiguration, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 66 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClustersConfigurationEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a DomainInfo\n// struct.\nfunc (v *DomainInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [30]string\n\ti := 0\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Description != nil {\n\t\tfields[i] = fmt.Sprintf(\"Description: %v\", *(v.Description))\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tfields[i] = fmt.Sprintf(\"Owner: %v\", *(v.Owner))\n\t\ti++\n\t}\n\tif v.Status != nil {\n\t\tfields[i] = fmt.Sprintf(\"Status: %v\", *(v.Status))\n\t\ti++\n\t}\n\tif v.RetentionDays != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetentionDays: %v\", *(v.RetentionDays))\n\t\ti++\n\t}\n\tif v.EmitMetric != nil {\n\t\tfields[i] = fmt.Sprintf(\"EmitMetric: %v\", *(v.EmitMetric))\n\t\ti++\n\t}\n\tif v.ArchivalBucket != nil {\n\t\tfields[i] = fmt.Sprintf(\"ArchivalBucket: %v\", *(v.ArchivalBucket))\n\t\ti++\n\t}\n\tif v.ArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"ArchivalStatus: %v\", *(v.ArchivalStatus))\n\t\ti++\n\t}\n\tif v.ConfigVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ConfigVersion: %v\", *(v.ConfigVersion))\n\t\ti++\n\t}\n\tif v.NotificationVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"NotificationVersion: %v\", *(v.NotificationVersion))\n\t\ti++\n\t}\n\tif v.FailoverNotificationVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverNotificationVersion: %v\", *(v.FailoverNotificationVersion))\n\t\ti++\n\t}\n\tif v.FailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverVersion: %v\", *(v.FailoverVersion))\n\t\ti++\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterName: %v\", *(v.ActiveClusterName))\n\t\ti++\n\t}\n\tif v.Clusters != nil {\n\t\tfields[i] = fmt.Sprintf(\"Clusters: %v\", v.Clusters)\n\t\ti++\n\t}\n\tif v.Data != nil {\n\t\tfields[i] = fmt.Sprintf(\"Data: %v\", v.Data)\n\t\ti++\n\t}\n\tif v.BadBinaries != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadBinaries: %v\", v.BadBinaries)\n\t\ti++\n\t}\n\tif v.BadBinariesEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"BadBinariesEncoding: %v\", *(v.BadBinariesEncoding))\n\t\ti++\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalStatus: %v\", *(v.HistoryArchivalStatus))\n\t\ti++\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistoryArchivalURI: %v\", *(v.HistoryArchivalURI))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalStatus: %v\", *(v.VisibilityArchivalStatus))\n\t\ti++\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityArchivalURI: %v\", *(v.VisibilityArchivalURI))\n\t\ti++\n\t}\n\tif v.FailoverEndTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"FailoverEndTime: %v\", *(v.FailoverEndTime))\n\t\ti++\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"PreviousFailoverVersion: %v\", *(v.PreviousFailoverVersion))\n\t\ti++\n\t}\n\tif v.LastUpdatedTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastUpdatedTime: %v\", *(v.LastUpdatedTime))\n\t\ti++\n\t}\n\tif v.IsolationGroupsConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroupsConfiguration: %v\", v.IsolationGroupsConfiguration)\n\t\ti++\n\t}\n\tif v.IsolationGroupsConfigurationEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroupsConfigurationEncoding: %v\", *(v.IsolationGroupsConfigurationEncoding))\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"AsyncWorkflowConfiguration: %v\", v.AsyncWorkflowConfiguration)\n\t\ti++\n\t}\n\tif v.AsyncWorkflowConfigurationEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"AsyncWorkflowConfigurationEncoding: %v\", *(v.AsyncWorkflowConfigurationEncoding))\n\t\ti++\n\t}\n\tif v.ActiveClustersConfiguration != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClustersConfiguration: %v\", v.ActiveClustersConfiguration)\n\t\ti++\n\t}\n\tif v.ActiveClustersConfigurationEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClustersConfigurationEncoding: %v\", *(v.ActiveClustersConfigurationEncoding))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"DomainInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _I16_EqualsPtr(lhs, rhs *int16) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn (x == y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\nfunc _Map_String_String_Equals(lhs, rhs map[string]string) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this DomainInfo match the\n// provided DomainInfo.\n//\n// This function performs a deep comparison.\nfunc (v *DomainInfo) Equals(rhs *DomainInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Description, rhs.Description) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Owner, rhs.Owner) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.Status, rhs.Status) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.RetentionDays, rhs.RetentionDays) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.EmitMetric, rhs.EmitMetric) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ArchivalBucket, rhs.ArchivalBucket) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.ArchivalStatus, rhs.ArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ConfigVersion, rhs.ConfigVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NotificationVersion, rhs.NotificationVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverNotificationVersion, rhs.FailoverNotificationVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverVersion, rhs.FailoverVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClusterName, rhs.ActiveClusterName) {\n\t\treturn false\n\t}\n\tif !((v.Clusters == nil && rhs.Clusters == nil) || (v.Clusters != nil && rhs.Clusters != nil && _List_String_Equals(v.Clusters, rhs.Clusters))) {\n\t\treturn false\n\t}\n\tif !((v.Data == nil && rhs.Data == nil) || (v.Data != nil && rhs.Data != nil && _Map_String_String_Equals(v.Data, rhs.Data))) {\n\t\treturn false\n\t}\n\tif !((v.BadBinaries == nil && rhs.BadBinaries == nil) || (v.BadBinaries != nil && rhs.BadBinaries != nil && bytes.Equal(v.BadBinaries, rhs.BadBinaries))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.BadBinariesEncoding, rhs.BadBinariesEncoding) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.HistoryArchivalStatus, rhs.HistoryArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.HistoryArchivalURI, rhs.HistoryArchivalURI) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.VisibilityArchivalStatus, rhs.VisibilityArchivalStatus) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.VisibilityArchivalURI, rhs.VisibilityArchivalURI) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FailoverEndTime, rhs.FailoverEndTime) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.PreviousFailoverVersion, rhs.PreviousFailoverVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastUpdatedTime, rhs.LastUpdatedTime) {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroupsConfiguration == nil && rhs.IsolationGroupsConfiguration == nil) || (v.IsolationGroupsConfiguration != nil && rhs.IsolationGroupsConfiguration != nil && bytes.Equal(v.IsolationGroupsConfiguration, rhs.IsolationGroupsConfiguration))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.IsolationGroupsConfigurationEncoding, rhs.IsolationGroupsConfigurationEncoding) {\n\t\treturn false\n\t}\n\tif !((v.AsyncWorkflowConfiguration == nil && rhs.AsyncWorkflowConfiguration == nil) || (v.AsyncWorkflowConfiguration != nil && rhs.AsyncWorkflowConfiguration != nil && bytes.Equal(v.AsyncWorkflowConfiguration, rhs.AsyncWorkflowConfiguration))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.AsyncWorkflowConfigurationEncoding, rhs.AsyncWorkflowConfigurationEncoding) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClustersConfiguration == nil && rhs.ActiveClustersConfiguration == nil) || (v.ActiveClustersConfiguration != nil && rhs.ActiveClustersConfiguration != nil && bytes.Equal(v.ActiveClustersConfiguration, rhs.ActiveClustersConfiguration))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClustersConfigurationEncoding, rhs.ActiveClustersConfigurationEncoding) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_String_Zapper map[string]string\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_String_Zapper.\nfunc (m _Map_String_String_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), v)\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of DomainInfo.\nfunc (v *DomainInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Description != nil {\n\t\tenc.AddString(\"description\", *v.Description)\n\t}\n\tif v.Owner != nil {\n\t\tenc.AddString(\"owner\", *v.Owner)\n\t}\n\tif v.Status != nil {\n\t\tenc.AddInt32(\"status\", *v.Status)\n\t}\n\tif v.RetentionDays != nil {\n\t\tenc.AddInt16(\"retentionDays\", *v.RetentionDays)\n\t}\n\tif v.EmitMetric != nil {\n\t\tenc.AddBool(\"emitMetric\", *v.EmitMetric)\n\t}\n\tif v.ArchivalBucket != nil {\n\t\tenc.AddString(\"archivalBucket\", *v.ArchivalBucket)\n\t}\n\tif v.ArchivalStatus != nil {\n\t\tenc.AddInt16(\"archivalStatus\", *v.ArchivalStatus)\n\t}\n\tif v.ConfigVersion != nil {\n\t\tenc.AddInt64(\"configVersion\", *v.ConfigVersion)\n\t}\n\tif v.NotificationVersion != nil {\n\t\tenc.AddInt64(\"notificationVersion\", *v.NotificationVersion)\n\t}\n\tif v.FailoverNotificationVersion != nil {\n\t\tenc.AddInt64(\"failoverNotificationVersion\", *v.FailoverNotificationVersion)\n\t}\n\tif v.FailoverVersion != nil {\n\t\tenc.AddInt64(\"failoverVersion\", *v.FailoverVersion)\n\t}\n\tif v.ActiveClusterName != nil {\n\t\tenc.AddString(\"activeClusterName\", *v.ActiveClusterName)\n\t}\n\tif v.Clusters != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"clusters\", (_List_String_Zapper)(v.Clusters)))\n\t}\n\tif v.Data != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"data\", (_Map_String_String_Zapper)(v.Data)))\n\t}\n\tif v.BadBinaries != nil {\n\t\tenc.AddString(\"badBinaries\", base64.StdEncoding.EncodeToString(v.BadBinaries))\n\t}\n\tif v.BadBinariesEncoding != nil {\n\t\tenc.AddString(\"badBinariesEncoding\", *v.BadBinariesEncoding)\n\t}\n\tif v.HistoryArchivalStatus != nil {\n\t\tenc.AddInt16(\"historyArchivalStatus\", *v.HistoryArchivalStatus)\n\t}\n\tif v.HistoryArchivalURI != nil {\n\t\tenc.AddString(\"historyArchivalURI\", *v.HistoryArchivalURI)\n\t}\n\tif v.VisibilityArchivalStatus != nil {\n\t\tenc.AddInt16(\"visibilityArchivalStatus\", *v.VisibilityArchivalStatus)\n\t}\n\tif v.VisibilityArchivalURI != nil {\n\t\tenc.AddString(\"visibilityArchivalURI\", *v.VisibilityArchivalURI)\n\t}\n\tif v.FailoverEndTime != nil {\n\t\tenc.AddInt64(\"failoverEndTime\", *v.FailoverEndTime)\n\t}\n\tif v.PreviousFailoverVersion != nil {\n\t\tenc.AddInt64(\"previousFailoverVersion\", *v.PreviousFailoverVersion)\n\t}\n\tif v.LastUpdatedTime != nil {\n\t\tenc.AddInt64(\"lastUpdatedTime\", *v.LastUpdatedTime)\n\t}\n\tif v.IsolationGroupsConfiguration != nil {\n\t\tenc.AddString(\"isolationGroupsConfiguration\", base64.StdEncoding.EncodeToString(v.IsolationGroupsConfiguration))\n\t}\n\tif v.IsolationGroupsConfigurationEncoding != nil {\n\t\tenc.AddString(\"isolationGroupsConfigurationEncoding\", *v.IsolationGroupsConfigurationEncoding)\n\t}\n\tif v.AsyncWorkflowConfiguration != nil {\n\t\tenc.AddString(\"asyncWorkflowConfiguration\", base64.StdEncoding.EncodeToString(v.AsyncWorkflowConfiguration))\n\t}\n\tif v.AsyncWorkflowConfigurationEncoding != nil {\n\t\tenc.AddString(\"asyncWorkflowConfigurationEncoding\", *v.AsyncWorkflowConfigurationEncoding)\n\t}\n\tif v.ActiveClustersConfiguration != nil {\n\t\tenc.AddString(\"activeClustersConfiguration\", base64.StdEncoding.EncodeToString(v.ActiveClustersConfiguration))\n\t}\n\tif v.ActiveClustersConfigurationEncoding != nil {\n\t\tenc.AddString(\"activeClustersConfigurationEncoding\", *v.ActiveClustersConfigurationEncoding)\n\t}\n\treturn err\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *DomainInfo) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetDescription returns the value of Description if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetDescription() (o string) {\n\tif v != nil && v.Description != nil {\n\t\treturn *v.Description\n\t}\n\n\treturn\n}\n\n// IsSetDescription returns true if Description is not nil.\nfunc (v *DomainInfo) IsSetDescription() bool {\n\treturn v != nil && v.Description != nil\n}\n\n// GetOwner returns the value of Owner if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetOwner() (o string) {\n\tif v != nil && v.Owner != nil {\n\t\treturn *v.Owner\n\t}\n\n\treturn\n}\n\n// IsSetOwner returns true if Owner is not nil.\nfunc (v *DomainInfo) IsSetOwner() bool {\n\treturn v != nil && v.Owner != nil\n}\n\n// GetStatus returns the value of Status if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetStatus() (o int32) {\n\tif v != nil && v.Status != nil {\n\t\treturn *v.Status\n\t}\n\n\treturn\n}\n\n// IsSetStatus returns true if Status is not nil.\nfunc (v *DomainInfo) IsSetStatus() bool {\n\treturn v != nil && v.Status != nil\n}\n\n// GetRetentionDays returns the value of RetentionDays if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetRetentionDays() (o int16) {\n\tif v != nil && v.RetentionDays != nil {\n\t\treturn *v.RetentionDays\n\t}\n\n\treturn\n}\n\n// IsSetRetentionDays returns true if RetentionDays is not nil.\nfunc (v *DomainInfo) IsSetRetentionDays() bool {\n\treturn v != nil && v.RetentionDays != nil\n}\n\n// GetEmitMetric returns the value of EmitMetric if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetEmitMetric() (o bool) {\n\tif v != nil && v.EmitMetric != nil {\n\t\treturn *v.EmitMetric\n\t}\n\n\treturn\n}\n\n// IsSetEmitMetric returns true if EmitMetric is not nil.\nfunc (v *DomainInfo) IsSetEmitMetric() bool {\n\treturn v != nil && v.EmitMetric != nil\n}\n\n// GetArchivalBucket returns the value of ArchivalBucket if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetArchivalBucket() (o string) {\n\tif v != nil && v.ArchivalBucket != nil {\n\t\treturn *v.ArchivalBucket\n\t}\n\n\treturn\n}\n\n// IsSetArchivalBucket returns true if ArchivalBucket is not nil.\nfunc (v *DomainInfo) IsSetArchivalBucket() bool {\n\treturn v != nil && v.ArchivalBucket != nil\n}\n\n// GetArchivalStatus returns the value of ArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetArchivalStatus() (o int16) {\n\tif v != nil && v.ArchivalStatus != nil {\n\t\treturn *v.ArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetArchivalStatus returns true if ArchivalStatus is not nil.\nfunc (v *DomainInfo) IsSetArchivalStatus() bool {\n\treturn v != nil && v.ArchivalStatus != nil\n}\n\n// GetConfigVersion returns the value of ConfigVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetConfigVersion() (o int64) {\n\tif v != nil && v.ConfigVersion != nil {\n\t\treturn *v.ConfigVersion\n\t}\n\n\treturn\n}\n\n// IsSetConfigVersion returns true if ConfigVersion is not nil.\nfunc (v *DomainInfo) IsSetConfigVersion() bool {\n\treturn v != nil && v.ConfigVersion != nil\n}\n\n// GetNotificationVersion returns the value of NotificationVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetNotificationVersion() (o int64) {\n\tif v != nil && v.NotificationVersion != nil {\n\t\treturn *v.NotificationVersion\n\t}\n\n\treturn\n}\n\n// IsSetNotificationVersion returns true if NotificationVersion is not nil.\nfunc (v *DomainInfo) IsSetNotificationVersion() bool {\n\treturn v != nil && v.NotificationVersion != nil\n}\n\n// GetFailoverNotificationVersion returns the value of FailoverNotificationVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetFailoverNotificationVersion() (o int64) {\n\tif v != nil && v.FailoverNotificationVersion != nil {\n\t\treturn *v.FailoverNotificationVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverNotificationVersion returns true if FailoverNotificationVersion is not nil.\nfunc (v *DomainInfo) IsSetFailoverNotificationVersion() bool {\n\treturn v != nil && v.FailoverNotificationVersion != nil\n}\n\n// GetFailoverVersion returns the value of FailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetFailoverVersion() (o int64) {\n\tif v != nil && v.FailoverVersion != nil {\n\t\treturn *v.FailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetFailoverVersion returns true if FailoverVersion is not nil.\nfunc (v *DomainInfo) IsSetFailoverVersion() bool {\n\treturn v != nil && v.FailoverVersion != nil\n}\n\n// GetActiveClusterName returns the value of ActiveClusterName if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetActiveClusterName() (o string) {\n\tif v != nil && v.ActiveClusterName != nil {\n\t\treturn *v.ActiveClusterName\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterName returns true if ActiveClusterName is not nil.\nfunc (v *DomainInfo) IsSetActiveClusterName() bool {\n\treturn v != nil && v.ActiveClusterName != nil\n}\n\n// GetClusters returns the value of Clusters if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetClusters() (o []string) {\n\tif v != nil && v.Clusters != nil {\n\t\treturn v.Clusters\n\t}\n\n\treturn\n}\n\n// IsSetClusters returns true if Clusters is not nil.\nfunc (v *DomainInfo) IsSetClusters() bool {\n\treturn v != nil && v.Clusters != nil\n}\n\n// GetData returns the value of Data if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\n\treturn\n}\n\n// IsSetData returns true if Data is not nil.\nfunc (v *DomainInfo) IsSetData() bool {\n\treturn v != nil && v.Data != nil\n}\n\n// GetBadBinaries returns the value of BadBinaries if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetBadBinaries() (o []byte) {\n\tif v != nil && v.BadBinaries != nil {\n\t\treturn v.BadBinaries\n\t}\n\n\treturn\n}\n\n// IsSetBadBinaries returns true if BadBinaries is not nil.\nfunc (v *DomainInfo) IsSetBadBinaries() bool {\n\treturn v != nil && v.BadBinaries != nil\n}\n\n// GetBadBinariesEncoding returns the value of BadBinariesEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetBadBinariesEncoding() (o string) {\n\tif v != nil && v.BadBinariesEncoding != nil {\n\t\treturn *v.BadBinariesEncoding\n\t}\n\n\treturn\n}\n\n// IsSetBadBinariesEncoding returns true if BadBinariesEncoding is not nil.\nfunc (v *DomainInfo) IsSetBadBinariesEncoding() bool {\n\treturn v != nil && v.BadBinariesEncoding != nil\n}\n\n// GetHistoryArchivalStatus returns the value of HistoryArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetHistoryArchivalStatus() (o int16) {\n\tif v != nil && v.HistoryArchivalStatus != nil {\n\t\treturn *v.HistoryArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalStatus returns true if HistoryArchivalStatus is not nil.\nfunc (v *DomainInfo) IsSetHistoryArchivalStatus() bool {\n\treturn v != nil && v.HistoryArchivalStatus != nil\n}\n\n// GetHistoryArchivalURI returns the value of HistoryArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetHistoryArchivalURI() (o string) {\n\tif v != nil && v.HistoryArchivalURI != nil {\n\t\treturn *v.HistoryArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetHistoryArchivalURI returns true if HistoryArchivalURI is not nil.\nfunc (v *DomainInfo) IsSetHistoryArchivalURI() bool {\n\treturn v != nil && v.HistoryArchivalURI != nil\n}\n\n// GetVisibilityArchivalStatus returns the value of VisibilityArchivalStatus if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetVisibilityArchivalStatus() (o int16) {\n\tif v != nil && v.VisibilityArchivalStatus != nil {\n\t\treturn *v.VisibilityArchivalStatus\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalStatus returns true if VisibilityArchivalStatus is not nil.\nfunc (v *DomainInfo) IsSetVisibilityArchivalStatus() bool {\n\treturn v != nil && v.VisibilityArchivalStatus != nil\n}\n\n// GetVisibilityArchivalURI returns the value of VisibilityArchivalURI if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetVisibilityArchivalURI() (o string) {\n\tif v != nil && v.VisibilityArchivalURI != nil {\n\t\treturn *v.VisibilityArchivalURI\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityArchivalURI returns true if VisibilityArchivalURI is not nil.\nfunc (v *DomainInfo) IsSetVisibilityArchivalURI() bool {\n\treturn v != nil && v.VisibilityArchivalURI != nil\n}\n\n// GetFailoverEndTime returns the value of FailoverEndTime if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetFailoverEndTime() (o int64) {\n\tif v != nil && v.FailoverEndTime != nil {\n\t\treturn *v.FailoverEndTime\n\t}\n\n\treturn\n}\n\n// IsSetFailoverEndTime returns true if FailoverEndTime is not nil.\nfunc (v *DomainInfo) IsSetFailoverEndTime() bool {\n\treturn v != nil && v.FailoverEndTime != nil\n}\n\n// GetPreviousFailoverVersion returns the value of PreviousFailoverVersion if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetPreviousFailoverVersion() (o int64) {\n\tif v != nil && v.PreviousFailoverVersion != nil {\n\t\treturn *v.PreviousFailoverVersion\n\t}\n\n\treturn\n}\n\n// IsSetPreviousFailoverVersion returns true if PreviousFailoverVersion is not nil.\nfunc (v *DomainInfo) IsSetPreviousFailoverVersion() bool {\n\treturn v != nil && v.PreviousFailoverVersion != nil\n}\n\n// GetLastUpdatedTime returns the value of LastUpdatedTime if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetLastUpdatedTime() (o int64) {\n\tif v != nil && v.LastUpdatedTime != nil {\n\t\treturn *v.LastUpdatedTime\n\t}\n\n\treturn\n}\n\n// IsSetLastUpdatedTime returns true if LastUpdatedTime is not nil.\nfunc (v *DomainInfo) IsSetLastUpdatedTime() bool {\n\treturn v != nil && v.LastUpdatedTime != nil\n}\n\n// GetIsolationGroupsConfiguration returns the value of IsolationGroupsConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetIsolationGroupsConfiguration() (o []byte) {\n\tif v != nil && v.IsolationGroupsConfiguration != nil {\n\t\treturn v.IsolationGroupsConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroupsConfiguration returns true if IsolationGroupsConfiguration is not nil.\nfunc (v *DomainInfo) IsSetIsolationGroupsConfiguration() bool {\n\treturn v != nil && v.IsolationGroupsConfiguration != nil\n}\n\n// GetIsolationGroupsConfigurationEncoding returns the value of IsolationGroupsConfigurationEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetIsolationGroupsConfigurationEncoding() (o string) {\n\tif v != nil && v.IsolationGroupsConfigurationEncoding != nil {\n\t\treturn *v.IsolationGroupsConfigurationEncoding\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroupsConfigurationEncoding returns true if IsolationGroupsConfigurationEncoding is not nil.\nfunc (v *DomainInfo) IsSetIsolationGroupsConfigurationEncoding() bool {\n\treturn v != nil && v.IsolationGroupsConfigurationEncoding != nil\n}\n\n// GetAsyncWorkflowConfiguration returns the value of AsyncWorkflowConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetAsyncWorkflowConfiguration() (o []byte) {\n\tif v != nil && v.AsyncWorkflowConfiguration != nil {\n\t\treturn v.AsyncWorkflowConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetAsyncWorkflowConfiguration returns true if AsyncWorkflowConfiguration is not nil.\nfunc (v *DomainInfo) IsSetAsyncWorkflowConfiguration() bool {\n\treturn v != nil && v.AsyncWorkflowConfiguration != nil\n}\n\n// GetAsyncWorkflowConfigurationEncoding returns the value of AsyncWorkflowConfigurationEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetAsyncWorkflowConfigurationEncoding() (o string) {\n\tif v != nil && v.AsyncWorkflowConfigurationEncoding != nil {\n\t\treturn *v.AsyncWorkflowConfigurationEncoding\n\t}\n\n\treturn\n}\n\n// IsSetAsyncWorkflowConfigurationEncoding returns true if AsyncWorkflowConfigurationEncoding is not nil.\nfunc (v *DomainInfo) IsSetAsyncWorkflowConfigurationEncoding() bool {\n\treturn v != nil && v.AsyncWorkflowConfigurationEncoding != nil\n}\n\n// GetActiveClustersConfiguration returns the value of ActiveClustersConfiguration if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetActiveClustersConfiguration() (o []byte) {\n\tif v != nil && v.ActiveClustersConfiguration != nil {\n\t\treturn v.ActiveClustersConfiguration\n\t}\n\n\treturn\n}\n\n// IsSetActiveClustersConfiguration returns true if ActiveClustersConfiguration is not nil.\nfunc (v *DomainInfo) IsSetActiveClustersConfiguration() bool {\n\treturn v != nil && v.ActiveClustersConfiguration != nil\n}\n\n// GetActiveClustersConfigurationEncoding returns the value of ActiveClustersConfigurationEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *DomainInfo) GetActiveClustersConfigurationEncoding() (o string) {\n\tif v != nil && v.ActiveClustersConfigurationEncoding != nil {\n\t\treturn *v.ActiveClustersConfigurationEncoding\n\t}\n\n\treturn\n}\n\n// IsSetActiveClustersConfigurationEncoding returns true if ActiveClustersConfigurationEncoding is not nil.\nfunc (v *DomainInfo) IsSetActiveClustersConfigurationEncoding() bool {\n\treturn v != nil && v.ActiveClustersConfigurationEncoding != nil\n}\n\ntype HistoryTreeInfo struct {\n\tCreatedTimeNanos *int64                       `json:\"createdTimeNanos,omitempty\"`\n\tAncestors        []*shared.HistoryBranchRange `json:\"ancestors,omitempty\"`\n\tInfo             *string                      `json:\"info,omitempty\"`\n}\n\ntype _List_HistoryBranchRange_ValueList []*shared.HistoryBranchRange\n\nfunc (v _List_HistoryBranchRange_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.HistoryBranchRange', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_HistoryBranchRange_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_HistoryBranchRange_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_HistoryBranchRange_ValueList) Close() {}\n\n// ToWire translates a HistoryTreeInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *HistoryTreeInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.CreatedTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreatedTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.Ancestors != nil {\n\t\tw, err = wire.NewValueList(_List_HistoryBranchRange_ValueList(v.Ancestors)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.Info != nil {\n\t\tw, err = wire.NewValueString(*(v.Info)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _HistoryBranchRange_Read(w wire.Value) (*shared.HistoryBranchRange, error) {\n\tvar v shared.HistoryBranchRange\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_HistoryBranchRange_Read(l wire.ValueList) ([]*shared.HistoryBranchRange, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*shared.HistoryBranchRange, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _HistoryBranchRange_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a HistoryTreeInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a HistoryTreeInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v HistoryTreeInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *HistoryTreeInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreatedTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.Ancestors, err = _List_HistoryBranchRange_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Info = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_HistoryBranchRange_Encode(val []*shared.HistoryBranchRange, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*shared.HistoryBranchRange', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a HistoryTreeInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a HistoryTreeInfo struct could not be encoded.\nfunc (v *HistoryTreeInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.CreatedTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreatedTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Ancestors != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_HistoryBranchRange_Encode(v.Ancestors, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Info != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Info)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _HistoryBranchRange_Decode(sr stream.Reader) (*shared.HistoryBranchRange, error) {\n\tvar v shared.HistoryBranchRange\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_HistoryBranchRange_Decode(sr stream.Reader) ([]*shared.HistoryBranchRange, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*shared.HistoryBranchRange, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _HistoryBranchRange_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a HistoryTreeInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a HistoryTreeInfo struct could not be generated from the wire\n// representation.\nfunc (v *HistoryTreeInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreatedTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TList:\n\t\t\tv.Ancestors, err = _List_HistoryBranchRange_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Info = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a HistoryTreeInfo\n// struct.\nfunc (v *HistoryTreeInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.CreatedTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreatedTimeNanos: %v\", *(v.CreatedTimeNanos))\n\t\ti++\n\t}\n\tif v.Ancestors != nil {\n\t\tfields[i] = fmt.Sprintf(\"Ancestors: %v\", v.Ancestors)\n\t\ti++\n\t}\n\tif v.Info != nil {\n\t\tfields[i] = fmt.Sprintf(\"Info: %v\", *(v.Info))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"HistoryTreeInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_HistoryBranchRange_Equals(lhs, rhs []*shared.HistoryBranchRange) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this HistoryTreeInfo match the\n// provided HistoryTreeInfo.\n//\n// This function performs a deep comparison.\nfunc (v *HistoryTreeInfo) Equals(rhs *HistoryTreeInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreatedTimeNanos, rhs.CreatedTimeNanos) {\n\t\treturn false\n\t}\n\tif !((v.Ancestors == nil && rhs.Ancestors == nil) || (v.Ancestors != nil && rhs.Ancestors != nil && _List_HistoryBranchRange_Equals(v.Ancestors, rhs.Ancestors))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Info, rhs.Info) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_HistoryBranchRange_Zapper []*shared.HistoryBranchRange\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_HistoryBranchRange_Zapper.\nfunc (l _List_HistoryBranchRange_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of HistoryTreeInfo.\nfunc (v *HistoryTreeInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.CreatedTimeNanos != nil {\n\t\tenc.AddInt64(\"createdTimeNanos\", *v.CreatedTimeNanos)\n\t}\n\tif v.Ancestors != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"ancestors\", (_List_HistoryBranchRange_Zapper)(v.Ancestors)))\n\t}\n\tif v.Info != nil {\n\t\tenc.AddString(\"info\", *v.Info)\n\t}\n\treturn err\n}\n\n// GetCreatedTimeNanos returns the value of CreatedTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTreeInfo) GetCreatedTimeNanos() (o int64) {\n\tif v != nil && v.CreatedTimeNanos != nil {\n\t\treturn *v.CreatedTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetCreatedTimeNanos returns true if CreatedTimeNanos is not nil.\nfunc (v *HistoryTreeInfo) IsSetCreatedTimeNanos() bool {\n\treturn v != nil && v.CreatedTimeNanos != nil\n}\n\n// GetAncestors returns the value of Ancestors if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTreeInfo) GetAncestors() (o []*shared.HistoryBranchRange) {\n\tif v != nil && v.Ancestors != nil {\n\t\treturn v.Ancestors\n\t}\n\n\treturn\n}\n\n// IsSetAncestors returns true if Ancestors is not nil.\nfunc (v *HistoryTreeInfo) IsSetAncestors() bool {\n\treturn v != nil && v.Ancestors != nil\n}\n\n// GetInfo returns the value of Info if it is set or its\n// zero value if it is unset.\nfunc (v *HistoryTreeInfo) GetInfo() (o string) {\n\tif v != nil && v.Info != nil {\n\t\treturn *v.Info\n\t}\n\n\treturn\n}\n\n// IsSetInfo returns true if Info is not nil.\nfunc (v *HistoryTreeInfo) IsSetInfo() bool {\n\treturn v != nil && v.Info != nil\n}\n\ntype ReplicationTaskInfo struct {\n\tDomainID                []byte  `json:\"domainID,omitempty\"`\n\tWorkflowID              *string `json:\"workflowID,omitempty\"`\n\tRunID                   []byte  `json:\"runID,omitempty\"`\n\tTaskType                *int16  `json:\"taskType,omitempty\"`\n\tVersion                 *int64  `json:\"version,omitempty\"`\n\tFirstEventID            *int64  `json:\"firstEventID,omitempty\"`\n\tNextEventID             *int64  `json:\"nextEventID,omitempty\"`\n\tScheduledID             *int64  `json:\"scheduledID,omitempty\"`\n\tEventStoreVersion       *int32  `json:\"eventStoreVersion,omitempty\"`\n\tNewRunEventStoreVersion *int32  `json:\"newRunEventStoreVersion,omitempty\"`\n\tBranchToken             []byte  `json:\"branch_token,omitempty\"`\n\tNewRunBranchToken       []byte  `json:\"newRunBranchToken,omitempty\"`\n\tCreationTime            *int64  `json:\"creationTime,omitempty\"`\n}\n\n// ToWire translates a ReplicationTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ReplicationTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [13]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueBinary(v.DomainID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueBinary(v.RunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.FirstEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.FirstEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.NextEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduledID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduledID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.EventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunEventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.NewRunEventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.BranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.NewRunBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.NewRunBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreationTime)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a ReplicationTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ReplicationTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ReplicationTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ReplicationTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.DomainID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.RunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.FirstEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.NextEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduledID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.EventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.NewRunEventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.BranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.NewRunBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreationTime = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a ReplicationTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ReplicationTaskInfo struct could not be encoded.\nfunc (v *ReplicationTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.DomainID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.RunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.FirstEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NextEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.NextEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduledID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduledID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.EventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunEventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.NewRunEventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.BranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.BranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NewRunBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.NewRunBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreationTime != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreationTime)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a ReplicationTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ReplicationTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *ReplicationTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.DomainID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tv.RunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.FirstEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.NextEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduledID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.EventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.NewRunEventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TBinary:\n\t\t\tv.BranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TBinary:\n\t\t\tv.NewRunBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreationTime = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ReplicationTaskInfo\n// struct.\nfunc (v *ReplicationTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [13]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", v.DomainID)\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", v.RunID)\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.FirstEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstEventID: %v\", *(v.FirstEventID))\n\t\ti++\n\t}\n\tif v.NextEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"NextEventID: %v\", *(v.NextEventID))\n\t\ti++\n\t}\n\tif v.ScheduledID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduledID: %v\", *(v.ScheduledID))\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventStoreVersion: %v\", *(v.EventStoreVersion))\n\t\ti++\n\t}\n\tif v.NewRunEventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunEventStoreVersion: %v\", *(v.NewRunEventStoreVersion))\n\t\ti++\n\t}\n\tif v.BranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"BranchToken: %v\", v.BranchToken)\n\t\ti++\n\t}\n\tif v.NewRunBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"NewRunBranchToken: %v\", v.NewRunBranchToken)\n\t\ti++\n\t}\n\tif v.CreationTime != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreationTime: %v\", *(v.CreationTime))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ReplicationTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this ReplicationTaskInfo match the\n// provided ReplicationTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ReplicationTaskInfo) Equals(rhs *ReplicationTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainID == nil && rhs.DomainID == nil) || (v.DomainID != nil && rhs.DomainID != nil && bytes.Equal(v.DomainID, rhs.DomainID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.RunID == nil && rhs.RunID == nil) || (v.RunID != nil && rhs.RunID != nil && bytes.Equal(v.RunID, rhs.RunID))) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.FirstEventID, rhs.FirstEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.NextEventID, rhs.NextEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduledID, rhs.ScheduledID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.EventStoreVersion, rhs.EventStoreVersion) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.NewRunEventStoreVersion, rhs.NewRunEventStoreVersion) {\n\t\treturn false\n\t}\n\tif !((v.BranchToken == nil && rhs.BranchToken == nil) || (v.BranchToken != nil && rhs.BranchToken != nil && bytes.Equal(v.BranchToken, rhs.BranchToken))) {\n\t\treturn false\n\t}\n\tif !((v.NewRunBranchToken == nil && rhs.NewRunBranchToken == nil) || (v.NewRunBranchToken != nil && rhs.NewRunBranchToken != nil && bytes.Equal(v.NewRunBranchToken, rhs.NewRunBranchToken))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreationTime, rhs.CreationTime) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ReplicationTaskInfo.\nfunc (v *ReplicationTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", base64.StdEncoding.EncodeToString(v.DomainID))\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", base64.StdEncoding.EncodeToString(v.RunID))\n\t}\n\tif v.TaskType != nil {\n\t\tenc.AddInt16(\"taskType\", *v.TaskType)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.FirstEventID != nil {\n\t\tenc.AddInt64(\"firstEventID\", *v.FirstEventID)\n\t}\n\tif v.NextEventID != nil {\n\t\tenc.AddInt64(\"nextEventID\", *v.NextEventID)\n\t}\n\tif v.ScheduledID != nil {\n\t\tenc.AddInt64(\"scheduledID\", *v.ScheduledID)\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tenc.AddInt32(\"eventStoreVersion\", *v.EventStoreVersion)\n\t}\n\tif v.NewRunEventStoreVersion != nil {\n\t\tenc.AddInt32(\"newRunEventStoreVersion\", *v.NewRunEventStoreVersion)\n\t}\n\tif v.BranchToken != nil {\n\t\tenc.AddString(\"branch_token\", base64.StdEncoding.EncodeToString(v.BranchToken))\n\t}\n\tif v.NewRunBranchToken != nil {\n\t\tenc.AddString(\"newRunBranchToken\", base64.StdEncoding.EncodeToString(v.NewRunBranchToken))\n\t}\n\tif v.CreationTime != nil {\n\t\tenc.AddInt64(\"creationTime\", *v.CreationTime)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetDomainID() (o []byte) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetRunID() (o []byte) {\n\tif v != nil && v.RunID != nil {\n\t\treturn v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetTaskType() (o int16) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *ReplicationTaskInfo) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *ReplicationTaskInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetFirstEventID returns the value of FirstEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetFirstEventID() (o int64) {\n\tif v != nil && v.FirstEventID != nil {\n\t\treturn *v.FirstEventID\n\t}\n\n\treturn\n}\n\n// IsSetFirstEventID returns true if FirstEventID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetFirstEventID() bool {\n\treturn v != nil && v.FirstEventID != nil\n}\n\n// GetNextEventID returns the value of NextEventID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetNextEventID() (o int64) {\n\tif v != nil && v.NextEventID != nil {\n\t\treturn *v.NextEventID\n\t}\n\n\treturn\n}\n\n// IsSetNextEventID returns true if NextEventID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetNextEventID() bool {\n\treturn v != nil && v.NextEventID != nil\n}\n\n// GetScheduledID returns the value of ScheduledID if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetScheduledID() (o int64) {\n\tif v != nil && v.ScheduledID != nil {\n\t\treturn *v.ScheduledID\n\t}\n\n\treturn\n}\n\n// IsSetScheduledID returns true if ScheduledID is not nil.\nfunc (v *ReplicationTaskInfo) IsSetScheduledID() bool {\n\treturn v != nil && v.ScheduledID != nil\n}\n\n// GetEventStoreVersion returns the value of EventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetEventStoreVersion() (o int32) {\n\tif v != nil && v.EventStoreVersion != nil {\n\t\treturn *v.EventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetEventStoreVersion returns true if EventStoreVersion is not nil.\nfunc (v *ReplicationTaskInfo) IsSetEventStoreVersion() bool {\n\treturn v != nil && v.EventStoreVersion != nil\n}\n\n// GetNewRunEventStoreVersion returns the value of NewRunEventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetNewRunEventStoreVersion() (o int32) {\n\tif v != nil && v.NewRunEventStoreVersion != nil {\n\t\treturn *v.NewRunEventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetNewRunEventStoreVersion returns true if NewRunEventStoreVersion is not nil.\nfunc (v *ReplicationTaskInfo) IsSetNewRunEventStoreVersion() bool {\n\treturn v != nil && v.NewRunEventStoreVersion != nil\n}\n\n// GetBranchToken returns the value of BranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetBranchToken() (o []byte) {\n\tif v != nil && v.BranchToken != nil {\n\t\treturn v.BranchToken\n\t}\n\n\treturn\n}\n\n// IsSetBranchToken returns true if BranchToken is not nil.\nfunc (v *ReplicationTaskInfo) IsSetBranchToken() bool {\n\treturn v != nil && v.BranchToken != nil\n}\n\n// GetNewRunBranchToken returns the value of NewRunBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetNewRunBranchToken() (o []byte) {\n\tif v != nil && v.NewRunBranchToken != nil {\n\t\treturn v.NewRunBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetNewRunBranchToken returns true if NewRunBranchToken is not nil.\nfunc (v *ReplicationTaskInfo) IsSetNewRunBranchToken() bool {\n\treturn v != nil && v.NewRunBranchToken != nil\n}\n\n// GetCreationTime returns the value of CreationTime if it is set or its\n// zero value if it is unset.\nfunc (v *ReplicationTaskInfo) GetCreationTime() (o int64) {\n\tif v != nil && v.CreationTime != nil {\n\t\treturn *v.CreationTime\n\t}\n\n\treturn\n}\n\n// IsSetCreationTime returns true if CreationTime is not nil.\nfunc (v *ReplicationTaskInfo) IsSetCreationTime() bool {\n\treturn v != nil && v.CreationTime != nil\n}\n\ntype RequestCancelInfo struct {\n\tVersion               *int64  `json:\"version,omitempty\"`\n\tInitiatedEventBatchID *int64  `json:\"initiatedEventBatchID,omitempty\"`\n\tCancelRequestID       *string `json:\"cancelRequestID,omitempty\"`\n}\n\n// ToWire translates a RequestCancelInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *RequestCancelInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventBatchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.CancelRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a RequestCancelInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a RequestCancelInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v RequestCancelInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *RequestCancelInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventBatchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CancelRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a RequestCancelInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a RequestCancelInfo struct could not be encoded.\nfunc (v *RequestCancelInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventBatchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventBatchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CancelRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a RequestCancelInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a RequestCancelInfo struct could not be generated from the wire\n// representation.\nfunc (v *RequestCancelInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventBatchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CancelRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a RequestCancelInfo\n// struct.\nfunc (v *RequestCancelInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventBatchID: %v\", *(v.InitiatedEventBatchID))\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequestID: %v\", *(v.CancelRequestID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"RequestCancelInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this RequestCancelInfo match the\n// provided RequestCancelInfo.\n//\n// This function performs a deep comparison.\nfunc (v *RequestCancelInfo) Equals(rhs *RequestCancelInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventBatchID, rhs.InitiatedEventBatchID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CancelRequestID, rhs.CancelRequestID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of RequestCancelInfo.\nfunc (v *RequestCancelInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tenc.AddInt64(\"initiatedEventBatchID\", *v.InitiatedEventBatchID)\n\t}\n\tif v.CancelRequestID != nil {\n\t\tenc.AddString(\"cancelRequestID\", *v.CancelRequestID)\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *RequestCancelInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetInitiatedEventBatchID returns the value of InitiatedEventBatchID if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelInfo) GetInitiatedEventBatchID() (o int64) {\n\tif v != nil && v.InitiatedEventBatchID != nil {\n\t\treturn *v.InitiatedEventBatchID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventBatchID returns true if InitiatedEventBatchID is not nil.\nfunc (v *RequestCancelInfo) IsSetInitiatedEventBatchID() bool {\n\treturn v != nil && v.InitiatedEventBatchID != nil\n}\n\n// GetCancelRequestID returns the value of CancelRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *RequestCancelInfo) GetCancelRequestID() (o string) {\n\tif v != nil && v.CancelRequestID != nil {\n\t\treturn *v.CancelRequestID\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequestID returns true if CancelRequestID is not nil.\nfunc (v *RequestCancelInfo) IsSetCancelRequestID() bool {\n\treturn v != nil && v.CancelRequestID != nil\n}\n\ntype ShardInfo struct {\n\tStolenSinceRenew                          *int32                       `json:\"stolenSinceRenew,omitempty\"`\n\tUpdatedAtNanos                            *int64                       `json:\"updatedAtNanos,omitempty\"`\n\tReplicationAckLevel                       *int64                       `json:\"replicationAckLevel,omitempty\"`\n\tTransferAckLevel                          *int64                       `json:\"transferAckLevel,omitempty\"`\n\tTimerAckLevelNanos                        *int64                       `json:\"timerAckLevelNanos,omitempty\"`\n\tDomainNotificationVersion                 *int64                       `json:\"domainNotificationVersion,omitempty\"`\n\tClusterTransferAckLevel                   map[string]int64             `json:\"clusterTransferAckLevel,omitempty\"`\n\tClusterTimerAckLevel                      map[string]int64             `json:\"clusterTimerAckLevel,omitempty\"`\n\tOwner                                     *string                      `json:\"owner,omitempty\"`\n\tClusterReplicationLevel                   map[string]int64             `json:\"clusterReplicationLevel,omitempty\"`\n\tPendingFailoverMarkers                    []byte                       `json:\"pendingFailoverMarkers,omitempty\"`\n\tPendingFailoverMarkersEncoding            *string                      `json:\"pendingFailoverMarkersEncoding,omitempty\"`\n\tReplicationDlqAckLevel                    map[string]int64             `json:\"replicationDlqAckLevel,omitempty\"`\n\tTransferProcessingQueueStates             []byte                       `json:\"transferProcessingQueueStates,omitempty\"`\n\tTransferProcessingQueueStatesEncoding     *string                      `json:\"transferProcessingQueueStatesEncoding,omitempty\"`\n\tTimerProcessingQueueStates                []byte                       `json:\"timerProcessingQueueStates,omitempty\"`\n\tTimerProcessingQueueStatesEncoding        *string                      `json:\"timerProcessingQueueStatesEncoding,omitempty\"`\n\tCrossClusterProcessingQueueStates         []byte                       `json:\"crossClusterProcessingQueueStates,omitempty\"`\n\tCrossClusterProcessingQueueStatesEncoding *string                      `json:\"crossClusterProcessingQueueStatesEncoding,omitempty\"`\n\tQueueStates                               map[int32]*shared.QueueState `json:\"queueStates,omitempty\"`\n}\n\ntype _Map_String_I64_MapItemList map[string]int64\n\nfunc (m _Map_String_I64_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueI64(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_I64_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_I64_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_I64_MapItemList) ValueType() wire.Type {\n\treturn wire.TI64\n}\n\nfunc (_Map_String_I64_MapItemList) Close() {}\n\ntype _Map_I32_QueueState_MapItemList map[int32]*shared.QueueState\n\nfunc (m _Map_I32_QueueState_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*shared.QueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_QueueState_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_QueueState_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_QueueState_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_I32_QueueState_MapItemList) Close() {}\n\n// ToWire translates a ShardInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *ShardInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [20]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.StolenSinceRenew != nil {\n\t\tw, err = wire.NewValueI32(*(v.StolenSinceRenew)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.UpdatedAtNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.UpdatedAtNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationAckLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.ReplicationAckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.TransferAckLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.TransferAckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.TimerAckLevelNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.TimerAckLevelNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.DomainNotificationVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.DomainNotificationVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterTransferAckLevel != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_I64_MapItemList(v.ClusterTransferAckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterTimerAckLevel != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_I64_MapItemList(v.ClusterTimerAckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tw, err = wire.NewValueString(*(v.Owner)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\tif v.ClusterReplicationLevel != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_I64_MapItemList(v.ClusterReplicationLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 40, Value: w}\n\t\ti++\n\t}\n\tif v.PendingFailoverMarkers != nil {\n\t\tw, err = wire.NewValueBinary(v.PendingFailoverMarkers), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 42, Value: w}\n\t\ti++\n\t}\n\tif v.PendingFailoverMarkersEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.PendingFailoverMarkersEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 44, Value: w}\n\t\ti++\n\t}\n\tif v.ReplicationDlqAckLevel != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_I64_MapItemList(v.ReplicationDlqAckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 46, Value: w}\n\t\ti++\n\t}\n\tif v.TransferProcessingQueueStates != nil {\n\t\tw, err = wire.NewValueBinary(v.TransferProcessingQueueStates), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.TransferProcessingQueueStatesEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.TransferProcessingQueueStatesEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 51, Value: w}\n\t\ti++\n\t}\n\tif v.TimerProcessingQueueStates != nil {\n\t\tw, err = wire.NewValueBinary(v.TimerProcessingQueueStates), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 55, Value: w}\n\t\ti++\n\t}\n\tif v.TimerProcessingQueueStatesEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.TimerProcessingQueueStatesEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.CrossClusterProcessingQueueStates != nil {\n\t\tw, err = wire.NewValueBinary(v.CrossClusterProcessingQueueStates), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.CrossClusterProcessingQueueStatesEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.CrossClusterProcessingQueueStatesEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 61, Value: w}\n\t\ti++\n\t}\n\tif v.QueueStates != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_QueueState_MapItemList(v.QueueStates)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 64, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_String_I64_Read(m wire.MapItemList) (map[string]int64, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TI64 {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string]int64, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetI64(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _QueueState_Read(w wire.Value) (*shared.QueueState, error) {\n\tvar v shared.QueueState\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_I32_QueueState_Read(m wire.MapItemList) (map[int32]*shared.QueueState, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32]*shared.QueueState, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _QueueState_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a ShardInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a ShardInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v ShardInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *ShardInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.StolenSinceRenew = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.UpdatedAtNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ReplicationAckLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TransferAckLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TimerAckLevelNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DomainNotificationVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ClusterTransferAckLevel, err = _Map_String_I64_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ClusterTimerAckLevel, err = _Map_String_I64_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Owner = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 40:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ClusterReplicationLevel, err = _Map_String_I64_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 42:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.PendingFailoverMarkers, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 44:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.PendingFailoverMarkersEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 46:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ReplicationDlqAckLevel, err = _Map_String_I64_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TransferProcessingQueueStates, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 51:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TransferProcessingQueueStatesEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 55:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TimerProcessingQueueStates, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TimerProcessingQueueStatesEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CrossClusterProcessingQueueStates, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 61:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CrossClusterProcessingQueueStatesEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 64:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.QueueStates, err = _Map_I32_QueueState_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_I64_Encode(val map[string]int64, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TI64,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\nfunc _Map_I32_QueueState_Encode(val map[int32]*shared.QueueState, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*shared.QueueState', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a ShardInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a ShardInfo struct could not be encoded.\nfunc (v *ShardInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.StolenSinceRenew != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.StolenSinceRenew)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.UpdatedAtNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.UpdatedAtNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationAckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ReplicationAckLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TransferAckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TransferAckLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerAckLevelNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TimerAckLevelNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DomainNotificationVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DomainNotificationVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterTransferAckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_I64_Encode(v.ClusterTransferAckLevel, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterTimerAckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_I64_Encode(v.ClusterTimerAckLevel, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Owner != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Owner)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClusterReplicationLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 40, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_I64_Encode(v.ClusterReplicationLevel, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingFailoverMarkers != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 42, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.PendingFailoverMarkers); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PendingFailoverMarkersEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 44, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.PendingFailoverMarkersEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReplicationDlqAckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 46, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_I64_Encode(v.ReplicationDlqAckLevel, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TransferProcessingQueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TransferProcessingQueueStates); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TransferProcessingQueueStatesEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 51, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TransferProcessingQueueStatesEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerProcessingQueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 55, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TimerProcessingQueueStates); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimerProcessingQueueStatesEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TimerProcessingQueueStatesEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CrossClusterProcessingQueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CrossClusterProcessingQueueStates); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CrossClusterProcessingQueueStatesEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 61, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CrossClusterProcessingQueueStatesEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.QueueStates != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 64, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_QueueState_Encode(v.QueueStates, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_String_I64_Decode(sr stream.Reader) (map[string]int64, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TI64 {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string]int64, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadInt64()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _QueueState_Decode(sr stream.Reader) (*shared.QueueState, error) {\n\tvar v shared.QueueState\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_I32_QueueState_Decode(sr stream.Reader) (map[int32]*shared.QueueState, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32]*shared.QueueState, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _QueueState_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a ShardInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a ShardInfo struct could not be generated from the wire\n// representation.\nfunc (v *ShardInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.StolenSinceRenew = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.UpdatedAtNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ReplicationAckLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TransferAckLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TimerAckLevelNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DomainNotificationVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TMap:\n\t\t\tv.ClusterTransferAckLevel, err = _Map_String_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TMap:\n\t\t\tv.ClusterTimerAckLevel, err = _Map_String_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Owner = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 40 && fh.Type == wire.TMap:\n\t\t\tv.ClusterReplicationLevel, err = _Map_String_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 42 && fh.Type == wire.TBinary:\n\t\t\tv.PendingFailoverMarkers, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 44 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.PendingFailoverMarkersEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 46 && fh.Type == wire.TMap:\n\t\t\tv.ReplicationDlqAckLevel, err = _Map_String_I64_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TBinary:\n\t\t\tv.TransferProcessingQueueStates, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 51 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TransferProcessingQueueStatesEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 55 && fh.Type == wire.TBinary:\n\t\t\tv.TimerProcessingQueueStates, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TimerProcessingQueueStatesEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TBinary:\n\t\t\tv.CrossClusterProcessingQueueStates, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 61 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CrossClusterProcessingQueueStatesEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 64 && fh.Type == wire.TMap:\n\t\t\tv.QueueStates, err = _Map_I32_QueueState_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a ShardInfo\n// struct.\nfunc (v *ShardInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [20]string\n\ti := 0\n\tif v.StolenSinceRenew != nil {\n\t\tfields[i] = fmt.Sprintf(\"StolenSinceRenew: %v\", *(v.StolenSinceRenew))\n\t\ti++\n\t}\n\tif v.UpdatedAtNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"UpdatedAtNanos: %v\", *(v.UpdatedAtNanos))\n\t\ti++\n\t}\n\tif v.ReplicationAckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationAckLevel: %v\", *(v.ReplicationAckLevel))\n\t\ti++\n\t}\n\tif v.TransferAckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"TransferAckLevel: %v\", *(v.TransferAckLevel))\n\t\ti++\n\t}\n\tif v.TimerAckLevelNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerAckLevelNanos: %v\", *(v.TimerAckLevelNanos))\n\t\ti++\n\t}\n\tif v.DomainNotificationVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainNotificationVersion: %v\", *(v.DomainNotificationVersion))\n\t\ti++\n\t}\n\tif v.ClusterTransferAckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterTransferAckLevel: %v\", v.ClusterTransferAckLevel)\n\t\ti++\n\t}\n\tif v.ClusterTimerAckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterTimerAckLevel: %v\", v.ClusterTimerAckLevel)\n\t\ti++\n\t}\n\tif v.Owner != nil {\n\t\tfields[i] = fmt.Sprintf(\"Owner: %v\", *(v.Owner))\n\t\ti++\n\t}\n\tif v.ClusterReplicationLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClusterReplicationLevel: %v\", v.ClusterReplicationLevel)\n\t\ti++\n\t}\n\tif v.PendingFailoverMarkers != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingFailoverMarkers: %v\", v.PendingFailoverMarkers)\n\t\ti++\n\t}\n\tif v.PendingFailoverMarkersEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"PendingFailoverMarkersEncoding: %v\", *(v.PendingFailoverMarkersEncoding))\n\t\ti++\n\t}\n\tif v.ReplicationDlqAckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReplicationDlqAckLevel: %v\", v.ReplicationDlqAckLevel)\n\t\ti++\n\t}\n\tif v.TransferProcessingQueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"TransferProcessingQueueStates: %v\", v.TransferProcessingQueueStates)\n\t\ti++\n\t}\n\tif v.TransferProcessingQueueStatesEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"TransferProcessingQueueStatesEncoding: %v\", *(v.TransferProcessingQueueStatesEncoding))\n\t\ti++\n\t}\n\tif v.TimerProcessingQueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerProcessingQueueStates: %v\", v.TimerProcessingQueueStates)\n\t\ti++\n\t}\n\tif v.TimerProcessingQueueStatesEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimerProcessingQueueStatesEncoding: %v\", *(v.TimerProcessingQueueStatesEncoding))\n\t\ti++\n\t}\n\tif v.CrossClusterProcessingQueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"CrossClusterProcessingQueueStates: %v\", v.CrossClusterProcessingQueueStates)\n\t\ti++\n\t}\n\tif v.CrossClusterProcessingQueueStatesEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"CrossClusterProcessingQueueStatesEncoding: %v\", *(v.CrossClusterProcessingQueueStatesEncoding))\n\t\ti++\n\t}\n\tif v.QueueStates != nil {\n\t\tfields[i] = fmt.Sprintf(\"QueueStates: %v\", v.QueueStates)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"ShardInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_I64_Equals(lhs, rhs map[string]int64) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !(lv == rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _Map_I32_QueueState_Equals(lhs, rhs map[int32]*shared.QueueState) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this ShardInfo match the\n// provided ShardInfo.\n//\n// This function performs a deep comparison.\nfunc (v *ShardInfo) Equals(rhs *ShardInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.StolenSinceRenew, rhs.StolenSinceRenew) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.UpdatedAtNanos, rhs.UpdatedAtNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ReplicationAckLevel, rhs.ReplicationAckLevel) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TransferAckLevel, rhs.TransferAckLevel) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TimerAckLevelNanos, rhs.TimerAckLevelNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DomainNotificationVersion, rhs.DomainNotificationVersion) {\n\t\treturn false\n\t}\n\tif !((v.ClusterTransferAckLevel == nil && rhs.ClusterTransferAckLevel == nil) || (v.ClusterTransferAckLevel != nil && rhs.ClusterTransferAckLevel != nil && _Map_String_I64_Equals(v.ClusterTransferAckLevel, rhs.ClusterTransferAckLevel))) {\n\t\treturn false\n\t}\n\tif !((v.ClusterTimerAckLevel == nil && rhs.ClusterTimerAckLevel == nil) || (v.ClusterTimerAckLevel != nil && rhs.ClusterTimerAckLevel != nil && _Map_String_I64_Equals(v.ClusterTimerAckLevel, rhs.ClusterTimerAckLevel))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Owner, rhs.Owner) {\n\t\treturn false\n\t}\n\tif !((v.ClusterReplicationLevel == nil && rhs.ClusterReplicationLevel == nil) || (v.ClusterReplicationLevel != nil && rhs.ClusterReplicationLevel != nil && _Map_String_I64_Equals(v.ClusterReplicationLevel, rhs.ClusterReplicationLevel))) {\n\t\treturn false\n\t}\n\tif !((v.PendingFailoverMarkers == nil && rhs.PendingFailoverMarkers == nil) || (v.PendingFailoverMarkers != nil && rhs.PendingFailoverMarkers != nil && bytes.Equal(v.PendingFailoverMarkers, rhs.PendingFailoverMarkers))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.PendingFailoverMarkersEncoding, rhs.PendingFailoverMarkersEncoding) {\n\t\treturn false\n\t}\n\tif !((v.ReplicationDlqAckLevel == nil && rhs.ReplicationDlqAckLevel == nil) || (v.ReplicationDlqAckLevel != nil && rhs.ReplicationDlqAckLevel != nil && _Map_String_I64_Equals(v.ReplicationDlqAckLevel, rhs.ReplicationDlqAckLevel))) {\n\t\treturn false\n\t}\n\tif !((v.TransferProcessingQueueStates == nil && rhs.TransferProcessingQueueStates == nil) || (v.TransferProcessingQueueStates != nil && rhs.TransferProcessingQueueStates != nil && bytes.Equal(v.TransferProcessingQueueStates, rhs.TransferProcessingQueueStates))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TransferProcessingQueueStatesEncoding, rhs.TransferProcessingQueueStatesEncoding) {\n\t\treturn false\n\t}\n\tif !((v.TimerProcessingQueueStates == nil && rhs.TimerProcessingQueueStates == nil) || (v.TimerProcessingQueueStates != nil && rhs.TimerProcessingQueueStates != nil && bytes.Equal(v.TimerProcessingQueueStates, rhs.TimerProcessingQueueStates))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TimerProcessingQueueStatesEncoding, rhs.TimerProcessingQueueStatesEncoding) {\n\t\treturn false\n\t}\n\tif !((v.CrossClusterProcessingQueueStates == nil && rhs.CrossClusterProcessingQueueStates == nil) || (v.CrossClusterProcessingQueueStates != nil && rhs.CrossClusterProcessingQueueStates != nil && bytes.Equal(v.CrossClusterProcessingQueueStates, rhs.CrossClusterProcessingQueueStates))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CrossClusterProcessingQueueStatesEncoding, rhs.CrossClusterProcessingQueueStatesEncoding) {\n\t\treturn false\n\t}\n\tif !((v.QueueStates == nil && rhs.QueueStates == nil) || (v.QueueStates != nil && rhs.QueueStates != nil && _Map_I32_QueueState_Equals(v.QueueStates, rhs.QueueStates))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_I64_Zapper map[string]int64\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_I64_Zapper.\nfunc (m _Map_String_I64_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddInt64((string)(k), v)\n\t}\n\treturn err\n}\n\ntype _Map_I32_QueueState_Item_Zapper struct {\n\tKey   int32\n\tValue *shared.QueueState\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_QueueState_Item_Zapper.\nfunc (v _Map_I32_QueueState_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\treturn err\n}\n\ntype _Map_I32_QueueState_Zapper map[int32]*shared.QueueState\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_QueueState_Zapper.\nfunc (m _Map_I32_QueueState_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_QueueState_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of ShardInfo.\nfunc (v *ShardInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.StolenSinceRenew != nil {\n\t\tenc.AddInt32(\"stolenSinceRenew\", *v.StolenSinceRenew)\n\t}\n\tif v.UpdatedAtNanos != nil {\n\t\tenc.AddInt64(\"updatedAtNanos\", *v.UpdatedAtNanos)\n\t}\n\tif v.ReplicationAckLevel != nil {\n\t\tenc.AddInt64(\"replicationAckLevel\", *v.ReplicationAckLevel)\n\t}\n\tif v.TransferAckLevel != nil {\n\t\tenc.AddInt64(\"transferAckLevel\", *v.TransferAckLevel)\n\t}\n\tif v.TimerAckLevelNanos != nil {\n\t\tenc.AddInt64(\"timerAckLevelNanos\", *v.TimerAckLevelNanos)\n\t}\n\tif v.DomainNotificationVersion != nil {\n\t\tenc.AddInt64(\"domainNotificationVersion\", *v.DomainNotificationVersion)\n\t}\n\tif v.ClusterTransferAckLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterTransferAckLevel\", (_Map_String_I64_Zapper)(v.ClusterTransferAckLevel)))\n\t}\n\tif v.ClusterTimerAckLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterTimerAckLevel\", (_Map_String_I64_Zapper)(v.ClusterTimerAckLevel)))\n\t}\n\tif v.Owner != nil {\n\t\tenc.AddString(\"owner\", *v.Owner)\n\t}\n\tif v.ClusterReplicationLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"clusterReplicationLevel\", (_Map_String_I64_Zapper)(v.ClusterReplicationLevel)))\n\t}\n\tif v.PendingFailoverMarkers != nil {\n\t\tenc.AddString(\"pendingFailoverMarkers\", base64.StdEncoding.EncodeToString(v.PendingFailoverMarkers))\n\t}\n\tif v.PendingFailoverMarkersEncoding != nil {\n\t\tenc.AddString(\"pendingFailoverMarkersEncoding\", *v.PendingFailoverMarkersEncoding)\n\t}\n\tif v.ReplicationDlqAckLevel != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"replicationDlqAckLevel\", (_Map_String_I64_Zapper)(v.ReplicationDlqAckLevel)))\n\t}\n\tif v.TransferProcessingQueueStates != nil {\n\t\tenc.AddString(\"transferProcessingQueueStates\", base64.StdEncoding.EncodeToString(v.TransferProcessingQueueStates))\n\t}\n\tif v.TransferProcessingQueueStatesEncoding != nil {\n\t\tenc.AddString(\"transferProcessingQueueStatesEncoding\", *v.TransferProcessingQueueStatesEncoding)\n\t}\n\tif v.TimerProcessingQueueStates != nil {\n\t\tenc.AddString(\"timerProcessingQueueStates\", base64.StdEncoding.EncodeToString(v.TimerProcessingQueueStates))\n\t}\n\tif v.TimerProcessingQueueStatesEncoding != nil {\n\t\tenc.AddString(\"timerProcessingQueueStatesEncoding\", *v.TimerProcessingQueueStatesEncoding)\n\t}\n\tif v.CrossClusterProcessingQueueStates != nil {\n\t\tenc.AddString(\"crossClusterProcessingQueueStates\", base64.StdEncoding.EncodeToString(v.CrossClusterProcessingQueueStates))\n\t}\n\tif v.CrossClusterProcessingQueueStatesEncoding != nil {\n\t\tenc.AddString(\"crossClusterProcessingQueueStatesEncoding\", *v.CrossClusterProcessingQueueStatesEncoding)\n\t}\n\tif v.QueueStates != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"queueStates\", (_Map_I32_QueueState_Zapper)(v.QueueStates)))\n\t}\n\treturn err\n}\n\n// GetStolenSinceRenew returns the value of StolenSinceRenew if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetStolenSinceRenew() (o int32) {\n\tif v != nil && v.StolenSinceRenew != nil {\n\t\treturn *v.StolenSinceRenew\n\t}\n\n\treturn\n}\n\n// IsSetStolenSinceRenew returns true if StolenSinceRenew is not nil.\nfunc (v *ShardInfo) IsSetStolenSinceRenew() bool {\n\treturn v != nil && v.StolenSinceRenew != nil\n}\n\n// GetUpdatedAtNanos returns the value of UpdatedAtNanos if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetUpdatedAtNanos() (o int64) {\n\tif v != nil && v.UpdatedAtNanos != nil {\n\t\treturn *v.UpdatedAtNanos\n\t}\n\n\treturn\n}\n\n// IsSetUpdatedAtNanos returns true if UpdatedAtNanos is not nil.\nfunc (v *ShardInfo) IsSetUpdatedAtNanos() bool {\n\treturn v != nil && v.UpdatedAtNanos != nil\n}\n\n// GetReplicationAckLevel returns the value of ReplicationAckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetReplicationAckLevel() (o int64) {\n\tif v != nil && v.ReplicationAckLevel != nil {\n\t\treturn *v.ReplicationAckLevel\n\t}\n\n\treturn\n}\n\n// IsSetReplicationAckLevel returns true if ReplicationAckLevel is not nil.\nfunc (v *ShardInfo) IsSetReplicationAckLevel() bool {\n\treturn v != nil && v.ReplicationAckLevel != nil\n}\n\n// GetTransferAckLevel returns the value of TransferAckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTransferAckLevel() (o int64) {\n\tif v != nil && v.TransferAckLevel != nil {\n\t\treturn *v.TransferAckLevel\n\t}\n\n\treturn\n}\n\n// IsSetTransferAckLevel returns true if TransferAckLevel is not nil.\nfunc (v *ShardInfo) IsSetTransferAckLevel() bool {\n\treturn v != nil && v.TransferAckLevel != nil\n}\n\n// GetTimerAckLevelNanos returns the value of TimerAckLevelNanos if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTimerAckLevelNanos() (o int64) {\n\tif v != nil && v.TimerAckLevelNanos != nil {\n\t\treturn *v.TimerAckLevelNanos\n\t}\n\n\treturn\n}\n\n// IsSetTimerAckLevelNanos returns true if TimerAckLevelNanos is not nil.\nfunc (v *ShardInfo) IsSetTimerAckLevelNanos() bool {\n\treturn v != nil && v.TimerAckLevelNanos != nil\n}\n\n// GetDomainNotificationVersion returns the value of DomainNotificationVersion if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetDomainNotificationVersion() (o int64) {\n\tif v != nil && v.DomainNotificationVersion != nil {\n\t\treturn *v.DomainNotificationVersion\n\t}\n\n\treturn\n}\n\n// IsSetDomainNotificationVersion returns true if DomainNotificationVersion is not nil.\nfunc (v *ShardInfo) IsSetDomainNotificationVersion() bool {\n\treturn v != nil && v.DomainNotificationVersion != nil\n}\n\n// GetClusterTransferAckLevel returns the value of ClusterTransferAckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetClusterTransferAckLevel() (o map[string]int64) {\n\tif v != nil && v.ClusterTransferAckLevel != nil {\n\t\treturn v.ClusterTransferAckLevel\n\t}\n\n\treturn\n}\n\n// IsSetClusterTransferAckLevel returns true if ClusterTransferAckLevel is not nil.\nfunc (v *ShardInfo) IsSetClusterTransferAckLevel() bool {\n\treturn v != nil && v.ClusterTransferAckLevel != nil\n}\n\n// GetClusterTimerAckLevel returns the value of ClusterTimerAckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetClusterTimerAckLevel() (o map[string]int64) {\n\tif v != nil && v.ClusterTimerAckLevel != nil {\n\t\treturn v.ClusterTimerAckLevel\n\t}\n\n\treturn\n}\n\n// IsSetClusterTimerAckLevel returns true if ClusterTimerAckLevel is not nil.\nfunc (v *ShardInfo) IsSetClusterTimerAckLevel() bool {\n\treturn v != nil && v.ClusterTimerAckLevel != nil\n}\n\n// GetOwner returns the value of Owner if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetOwner() (o string) {\n\tif v != nil && v.Owner != nil {\n\t\treturn *v.Owner\n\t}\n\n\treturn\n}\n\n// IsSetOwner returns true if Owner is not nil.\nfunc (v *ShardInfo) IsSetOwner() bool {\n\treturn v != nil && v.Owner != nil\n}\n\n// GetClusterReplicationLevel returns the value of ClusterReplicationLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetClusterReplicationLevel() (o map[string]int64) {\n\tif v != nil && v.ClusterReplicationLevel != nil {\n\t\treturn v.ClusterReplicationLevel\n\t}\n\n\treturn\n}\n\n// IsSetClusterReplicationLevel returns true if ClusterReplicationLevel is not nil.\nfunc (v *ShardInfo) IsSetClusterReplicationLevel() bool {\n\treturn v != nil && v.ClusterReplicationLevel != nil\n}\n\n// GetPendingFailoverMarkers returns the value of PendingFailoverMarkers if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetPendingFailoverMarkers() (o []byte) {\n\tif v != nil && v.PendingFailoverMarkers != nil {\n\t\treturn v.PendingFailoverMarkers\n\t}\n\n\treturn\n}\n\n// IsSetPendingFailoverMarkers returns true if PendingFailoverMarkers is not nil.\nfunc (v *ShardInfo) IsSetPendingFailoverMarkers() bool {\n\treturn v != nil && v.PendingFailoverMarkers != nil\n}\n\n// GetPendingFailoverMarkersEncoding returns the value of PendingFailoverMarkersEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetPendingFailoverMarkersEncoding() (o string) {\n\tif v != nil && v.PendingFailoverMarkersEncoding != nil {\n\t\treturn *v.PendingFailoverMarkersEncoding\n\t}\n\n\treturn\n}\n\n// IsSetPendingFailoverMarkersEncoding returns true if PendingFailoverMarkersEncoding is not nil.\nfunc (v *ShardInfo) IsSetPendingFailoverMarkersEncoding() bool {\n\treturn v != nil && v.PendingFailoverMarkersEncoding != nil\n}\n\n// GetReplicationDlqAckLevel returns the value of ReplicationDlqAckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetReplicationDlqAckLevel() (o map[string]int64) {\n\tif v != nil && v.ReplicationDlqAckLevel != nil {\n\t\treturn v.ReplicationDlqAckLevel\n\t}\n\n\treturn\n}\n\n// IsSetReplicationDlqAckLevel returns true if ReplicationDlqAckLevel is not nil.\nfunc (v *ShardInfo) IsSetReplicationDlqAckLevel() bool {\n\treturn v != nil && v.ReplicationDlqAckLevel != nil\n}\n\n// GetTransferProcessingQueueStates returns the value of TransferProcessingQueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTransferProcessingQueueStates() (o []byte) {\n\tif v != nil && v.TransferProcessingQueueStates != nil {\n\t\treturn v.TransferProcessingQueueStates\n\t}\n\n\treturn\n}\n\n// IsSetTransferProcessingQueueStates returns true if TransferProcessingQueueStates is not nil.\nfunc (v *ShardInfo) IsSetTransferProcessingQueueStates() bool {\n\treturn v != nil && v.TransferProcessingQueueStates != nil\n}\n\n// GetTransferProcessingQueueStatesEncoding returns the value of TransferProcessingQueueStatesEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTransferProcessingQueueStatesEncoding() (o string) {\n\tif v != nil && v.TransferProcessingQueueStatesEncoding != nil {\n\t\treturn *v.TransferProcessingQueueStatesEncoding\n\t}\n\n\treturn\n}\n\n// IsSetTransferProcessingQueueStatesEncoding returns true if TransferProcessingQueueStatesEncoding is not nil.\nfunc (v *ShardInfo) IsSetTransferProcessingQueueStatesEncoding() bool {\n\treturn v != nil && v.TransferProcessingQueueStatesEncoding != nil\n}\n\n// GetTimerProcessingQueueStates returns the value of TimerProcessingQueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTimerProcessingQueueStates() (o []byte) {\n\tif v != nil && v.TimerProcessingQueueStates != nil {\n\t\treturn v.TimerProcessingQueueStates\n\t}\n\n\treturn\n}\n\n// IsSetTimerProcessingQueueStates returns true if TimerProcessingQueueStates is not nil.\nfunc (v *ShardInfo) IsSetTimerProcessingQueueStates() bool {\n\treturn v != nil && v.TimerProcessingQueueStates != nil\n}\n\n// GetTimerProcessingQueueStatesEncoding returns the value of TimerProcessingQueueStatesEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetTimerProcessingQueueStatesEncoding() (o string) {\n\tif v != nil && v.TimerProcessingQueueStatesEncoding != nil {\n\t\treturn *v.TimerProcessingQueueStatesEncoding\n\t}\n\n\treturn\n}\n\n// IsSetTimerProcessingQueueStatesEncoding returns true if TimerProcessingQueueStatesEncoding is not nil.\nfunc (v *ShardInfo) IsSetTimerProcessingQueueStatesEncoding() bool {\n\treturn v != nil && v.TimerProcessingQueueStatesEncoding != nil\n}\n\n// GetCrossClusterProcessingQueueStates returns the value of CrossClusterProcessingQueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetCrossClusterProcessingQueueStates() (o []byte) {\n\tif v != nil && v.CrossClusterProcessingQueueStates != nil {\n\t\treturn v.CrossClusterProcessingQueueStates\n\t}\n\n\treturn\n}\n\n// IsSetCrossClusterProcessingQueueStates returns true if CrossClusterProcessingQueueStates is not nil.\nfunc (v *ShardInfo) IsSetCrossClusterProcessingQueueStates() bool {\n\treturn v != nil && v.CrossClusterProcessingQueueStates != nil\n}\n\n// GetCrossClusterProcessingQueueStatesEncoding returns the value of CrossClusterProcessingQueueStatesEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetCrossClusterProcessingQueueStatesEncoding() (o string) {\n\tif v != nil && v.CrossClusterProcessingQueueStatesEncoding != nil {\n\t\treturn *v.CrossClusterProcessingQueueStatesEncoding\n\t}\n\n\treturn\n}\n\n// IsSetCrossClusterProcessingQueueStatesEncoding returns true if CrossClusterProcessingQueueStatesEncoding is not nil.\nfunc (v *ShardInfo) IsSetCrossClusterProcessingQueueStatesEncoding() bool {\n\treturn v != nil && v.CrossClusterProcessingQueueStatesEncoding != nil\n}\n\n// GetQueueStates returns the value of QueueStates if it is set or its\n// zero value if it is unset.\nfunc (v *ShardInfo) GetQueueStates() (o map[int32]*shared.QueueState) {\n\tif v != nil && v.QueueStates != nil {\n\t\treturn v.QueueStates\n\t}\n\n\treturn\n}\n\n// IsSetQueueStates returns true if QueueStates is not nil.\nfunc (v *ShardInfo) IsSetQueueStates() bool {\n\treturn v != nil && v.QueueStates != nil\n}\n\ntype SignalInfo struct {\n\tVersion               *int64  `json:\"version,omitempty\"`\n\tInitiatedEventBatchID *int64  `json:\"initiatedEventBatchID,omitempty\"`\n\tRequestID             *string `json:\"requestID,omitempty\"`\n\tName                  *string `json:\"name,omitempty\"`\n\tInput                 []byte  `json:\"input,omitempty\"`\n\tControl               []byte  `json:\"control,omitempty\"`\n}\n\n// ToWire translates a SignalInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *SignalInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedEventBatchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.RequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.Name != nil {\n\t\tw, err = wire.NewValueString(*(v.Name)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tw, err = wire.NewValueBinary(v.Input), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tw, err = wire.NewValueBinary(v.Control), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a SignalInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a SignalInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v SignalInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *SignalInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedEventBatchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.RequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.Name = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Input, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Control, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a SignalInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a SignalInfo struct could not be encoded.\nfunc (v *SignalInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedEventBatchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedEventBatchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.RequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Name != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.Name)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Input != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Input); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Control != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Control); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a SignalInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a SignalInfo struct could not be generated from the wire\n// representation.\nfunc (v *SignalInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedEventBatchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.RequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.Name = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TBinary:\n\t\t\tv.Input, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TBinary:\n\t\t\tv.Control, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a SignalInfo\n// struct.\nfunc (v *SignalInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedEventBatchID: %v\", *(v.InitiatedEventBatchID))\n\t\ti++\n\t}\n\tif v.RequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RequestID: %v\", *(v.RequestID))\n\t\ti++\n\t}\n\tif v.Name != nil {\n\t\tfields[i] = fmt.Sprintf(\"Name: %v\", *(v.Name))\n\t\ti++\n\t}\n\tif v.Input != nil {\n\t\tfields[i] = fmt.Sprintf(\"Input: %v\", v.Input)\n\t\ti++\n\t}\n\tif v.Control != nil {\n\t\tfields[i] = fmt.Sprintf(\"Control: %v\", v.Control)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"SignalInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this SignalInfo match the\n// provided SignalInfo.\n//\n// This function performs a deep comparison.\nfunc (v *SignalInfo) Equals(rhs *SignalInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedEventBatchID, rhs.InitiatedEventBatchID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.RequestID, rhs.RequestID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.Name, rhs.Name) {\n\t\treturn false\n\t}\n\tif !((v.Input == nil && rhs.Input == nil) || (v.Input != nil && rhs.Input != nil && bytes.Equal(v.Input, rhs.Input))) {\n\t\treturn false\n\t}\n\tif !((v.Control == nil && rhs.Control == nil) || (v.Control != nil && rhs.Control != nil && bytes.Equal(v.Control, rhs.Control))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of SignalInfo.\nfunc (v *SignalInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.InitiatedEventBatchID != nil {\n\t\tenc.AddInt64(\"initiatedEventBatchID\", *v.InitiatedEventBatchID)\n\t}\n\tif v.RequestID != nil {\n\t\tenc.AddString(\"requestID\", *v.RequestID)\n\t}\n\tif v.Name != nil {\n\t\tenc.AddString(\"name\", *v.Name)\n\t}\n\tif v.Input != nil {\n\t\tenc.AddString(\"input\", base64.StdEncoding.EncodeToString(v.Input))\n\t}\n\tif v.Control != nil {\n\t\tenc.AddString(\"control\", base64.StdEncoding.EncodeToString(v.Control))\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *SignalInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetInitiatedEventBatchID returns the value of InitiatedEventBatchID if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetInitiatedEventBatchID() (o int64) {\n\tif v != nil && v.InitiatedEventBatchID != nil {\n\t\treturn *v.InitiatedEventBatchID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedEventBatchID returns true if InitiatedEventBatchID is not nil.\nfunc (v *SignalInfo) IsSetInitiatedEventBatchID() bool {\n\treturn v != nil && v.InitiatedEventBatchID != nil\n}\n\n// GetRequestID returns the value of RequestID if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetRequestID() (o string) {\n\tif v != nil && v.RequestID != nil {\n\t\treturn *v.RequestID\n\t}\n\n\treturn\n}\n\n// IsSetRequestID returns true if RequestID is not nil.\nfunc (v *SignalInfo) IsSetRequestID() bool {\n\treturn v != nil && v.RequestID != nil\n}\n\n// GetName returns the value of Name if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\n\treturn\n}\n\n// IsSetName returns true if Name is not nil.\nfunc (v *SignalInfo) IsSetName() bool {\n\treturn v != nil && v.Name != nil\n}\n\n// GetInput returns the value of Input if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\n\treturn\n}\n\n// IsSetInput returns true if Input is not nil.\nfunc (v *SignalInfo) IsSetInput() bool {\n\treturn v != nil && v.Input != nil\n}\n\n// GetControl returns the value of Control if it is set or its\n// zero value if it is unset.\nfunc (v *SignalInfo) GetControl() (o []byte) {\n\tif v != nil && v.Control != nil {\n\t\treturn v.Control\n\t}\n\n\treturn\n}\n\n// IsSetControl returns true if Control is not nil.\nfunc (v *SignalInfo) IsSetControl() bool {\n\treturn v != nil && v.Control != nil\n}\n\ntype TaskInfo struct {\n\tWorkflowID       *string           `json:\"workflowID,omitempty\"`\n\tRunID            []byte            `json:\"runID,omitempty\"`\n\tScheduleID       *int64            `json:\"scheduleID,omitempty\"`\n\tExpiryTimeNanos  *int64            `json:\"expiryTimeNanos,omitempty\"`\n\tCreatedTimeNanos *int64            `json:\"createdTimeNanos,omitempty\"`\n\tPartitionConfig  map[string]string `json:\"partitionConfig,omitempty\"`\n}\n\n// ToWire translates a TaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [6]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueBinary(v.RunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 13, Value: w}\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpiryTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.CreatedTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.CreatedTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 15, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 17, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.RunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 13:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpiryTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 15:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CreatedTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 17:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskInfo struct could not be encoded.\nfunc (v *TaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.RunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 13, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpiryTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpiryTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreatedTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 15, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CreatedTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 17, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *TaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tv.RunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 13 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpiryTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 15 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CreatedTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 17 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskInfo\n// struct.\nfunc (v *TaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [6]string\n\ti := 0\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", v.RunID)\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleID: %v\", *(v.ScheduleID))\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpiryTimeNanos: %v\", *(v.ExpiryTimeNanos))\n\t\ti++\n\t}\n\tif v.CreatedTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreatedTimeNanos: %v\", *(v.CreatedTimeNanos))\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskInfo match the\n// provided TaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TaskInfo) Equals(rhs *TaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.RunID == nil && rhs.RunID == nil) || (v.RunID != nil && rhs.RunID != nil && bytes.Equal(v.RunID, rhs.RunID))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleID, rhs.ScheduleID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpiryTimeNanos, rhs.ExpiryTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CreatedTimeNanos, rhs.CreatedTimeNanos) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskInfo.\nfunc (v *TaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", base64.StdEncoding.EncodeToString(v.RunID))\n\t}\n\tif v.ScheduleID != nil {\n\t\tenc.AddInt64(\"scheduleID\", *v.ScheduleID)\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tenc.AddInt64(\"expiryTimeNanos\", *v.ExpiryTimeNanos)\n\t}\n\tif v.CreatedTimeNanos != nil {\n\t\tenc.AddInt64(\"createdTimeNanos\", *v.CreatedTimeNanos)\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\treturn err\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *TaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetRunID() (o []byte) {\n\tif v != nil && v.RunID != nil {\n\t\treturn v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *TaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetScheduleID returns the value of ScheduleID if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetScheduleID() (o int64) {\n\tif v != nil && v.ScheduleID != nil {\n\t\treturn *v.ScheduleID\n\t}\n\n\treturn\n}\n\n// IsSetScheduleID returns true if ScheduleID is not nil.\nfunc (v *TaskInfo) IsSetScheduleID() bool {\n\treturn v != nil && v.ScheduleID != nil\n}\n\n// GetExpiryTimeNanos returns the value of ExpiryTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetExpiryTimeNanos() (o int64) {\n\tif v != nil && v.ExpiryTimeNanos != nil {\n\t\treturn *v.ExpiryTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetExpiryTimeNanos returns true if ExpiryTimeNanos is not nil.\nfunc (v *TaskInfo) IsSetExpiryTimeNanos() bool {\n\treturn v != nil && v.ExpiryTimeNanos != nil\n}\n\n// GetCreatedTimeNanos returns the value of CreatedTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetCreatedTimeNanos() (o int64) {\n\tif v != nil && v.CreatedTimeNanos != nil {\n\t\treturn *v.CreatedTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetCreatedTimeNanos returns true if CreatedTimeNanos is not nil.\nfunc (v *TaskInfo) IsSetCreatedTimeNanos() bool {\n\treturn v != nil && v.CreatedTimeNanos != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *TaskInfo) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *TaskInfo) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\ntype TaskListInfo struct {\n\tKind                    *int16                   `json:\"kind,omitempty\"`\n\tAckLevel                *int64                   `json:\"ackLevel,omitempty\"`\n\tExpiryTimeNanos         *int64                   `json:\"expiryTimeNanos,omitempty\"`\n\tLastUpdatedNanos        *int64                   `json:\"lastUpdatedNanos,omitempty\"`\n\tAdaptivePartitionConfig *TaskListPartitionConfig `json:\"adaptivePartitionConfig,omitempty\"`\n}\n\n// ToWire translates a TaskListInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Kind != nil {\n\t\tw, err = wire.NewValueI16(*(v.Kind)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tw, err = wire.NewValueI64(*(v.AckLevel)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpiryTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.LastUpdatedNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastUpdatedNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.AdaptivePartitionConfig != nil {\n\t\tw, err = v.AdaptivePartitionConfig.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListPartitionConfig_Read(w wire.Value) (*TaskListPartitionConfig, error) {\n\tvar v TaskListPartitionConfig\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\n// FromWire deserializes a TaskListInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.Kind = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.AckLevel = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpiryTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastUpdatedNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TStruct {\n\t\t\t\tv.AdaptivePartitionConfig, err = _TaskListPartitionConfig_Read(field.Value)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskListInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListInfo struct could not be encoded.\nfunc (v *TaskListInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Kind != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.Kind)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AckLevel != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.AckLevel)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpiryTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpiryTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastUpdatedNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastUpdatedNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AdaptivePartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TStruct}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.AdaptivePartitionConfig.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListPartitionConfig_Decode(sr stream.Reader) (*TaskListPartitionConfig, error) {\n\tvar v TaskListPartitionConfig\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\n// Decode deserializes a TaskListInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListInfo struct could not be generated from the wire\n// representation.\nfunc (v *TaskListInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.Kind = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.AckLevel = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpiryTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastUpdatedNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TStruct:\n\t\t\tv.AdaptivePartitionConfig, err = _TaskListPartitionConfig_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListInfo\n// struct.\nfunc (v *TaskListInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Kind != nil {\n\t\tfields[i] = fmt.Sprintf(\"Kind: %v\", *(v.Kind))\n\t\ti++\n\t}\n\tif v.AckLevel != nil {\n\t\tfields[i] = fmt.Sprintf(\"AckLevel: %v\", *(v.AckLevel))\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpiryTimeNanos: %v\", *(v.ExpiryTimeNanos))\n\t\ti++\n\t}\n\tif v.LastUpdatedNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastUpdatedNanos: %v\", *(v.LastUpdatedNanos))\n\t\ti++\n\t}\n\tif v.AdaptivePartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"AdaptivePartitionConfig: %v\", v.AdaptivePartitionConfig)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskListInfo match the\n// provided TaskListInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListInfo) Equals(rhs *TaskListInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.Kind, rhs.Kind) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.AckLevel, rhs.AckLevel) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpiryTimeNanos, rhs.ExpiryTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastUpdatedNanos, rhs.LastUpdatedNanos) {\n\t\treturn false\n\t}\n\tif !((v.AdaptivePartitionConfig == nil && rhs.AdaptivePartitionConfig == nil) || (v.AdaptivePartitionConfig != nil && rhs.AdaptivePartitionConfig != nil && v.AdaptivePartitionConfig.Equals(rhs.AdaptivePartitionConfig))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListInfo.\nfunc (v *TaskListInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Kind != nil {\n\t\tenc.AddInt16(\"kind\", *v.Kind)\n\t}\n\tif v.AckLevel != nil {\n\t\tenc.AddInt64(\"ackLevel\", *v.AckLevel)\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tenc.AddInt64(\"expiryTimeNanos\", *v.ExpiryTimeNanos)\n\t}\n\tif v.LastUpdatedNanos != nil {\n\t\tenc.AddInt64(\"lastUpdatedNanos\", *v.LastUpdatedNanos)\n\t}\n\tif v.AdaptivePartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"adaptivePartitionConfig\", v.AdaptivePartitionConfig))\n\t}\n\treturn err\n}\n\n// GetKind returns the value of Kind if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListInfo) GetKind() (o int16) {\n\tif v != nil && v.Kind != nil {\n\t\treturn *v.Kind\n\t}\n\n\treturn\n}\n\n// IsSetKind returns true if Kind is not nil.\nfunc (v *TaskListInfo) IsSetKind() bool {\n\treturn v != nil && v.Kind != nil\n}\n\n// GetAckLevel returns the value of AckLevel if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListInfo) GetAckLevel() (o int64) {\n\tif v != nil && v.AckLevel != nil {\n\t\treturn *v.AckLevel\n\t}\n\n\treturn\n}\n\n// IsSetAckLevel returns true if AckLevel is not nil.\nfunc (v *TaskListInfo) IsSetAckLevel() bool {\n\treturn v != nil && v.AckLevel != nil\n}\n\n// GetExpiryTimeNanos returns the value of ExpiryTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListInfo) GetExpiryTimeNanos() (o int64) {\n\tif v != nil && v.ExpiryTimeNanos != nil {\n\t\treturn *v.ExpiryTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetExpiryTimeNanos returns true if ExpiryTimeNanos is not nil.\nfunc (v *TaskListInfo) IsSetExpiryTimeNanos() bool {\n\treturn v != nil && v.ExpiryTimeNanos != nil\n}\n\n// GetLastUpdatedNanos returns the value of LastUpdatedNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListInfo) GetLastUpdatedNanos() (o int64) {\n\tif v != nil && v.LastUpdatedNanos != nil {\n\t\treturn *v.LastUpdatedNanos\n\t}\n\n\treturn\n}\n\n// IsSetLastUpdatedNanos returns true if LastUpdatedNanos is not nil.\nfunc (v *TaskListInfo) IsSetLastUpdatedNanos() bool {\n\treturn v != nil && v.LastUpdatedNanos != nil\n}\n\n// GetAdaptivePartitionConfig returns the value of AdaptivePartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListInfo) GetAdaptivePartitionConfig() (o *TaskListPartitionConfig) {\n\tif v != nil && v.AdaptivePartitionConfig != nil {\n\t\treturn v.AdaptivePartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetAdaptivePartitionConfig returns true if AdaptivePartitionConfig is not nil.\nfunc (v *TaskListInfo) IsSetAdaptivePartitionConfig() bool {\n\treturn v != nil && v.AdaptivePartitionConfig != nil\n}\n\ntype TaskListPartition struct {\n\tIsolationGroups []string `json:\"isolationGroups,omitempty\"`\n}\n\n// ToWire translates a TaskListPartition struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListPartition) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.IsolationGroups != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.IsolationGroups)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TaskListPartition struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListPartition struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListPartition\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListPartition) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.IsolationGroups, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TaskListPartition struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListPartition struct could not be encoded.\nfunc (v *TaskListPartition) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.IsolationGroups != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.IsolationGroups, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TaskListPartition struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListPartition struct could not be generated from the wire\n// representation.\nfunc (v *TaskListPartition) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.IsolationGroups, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListPartition\n// struct.\nfunc (v *TaskListPartition) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.IsolationGroups != nil {\n\t\tfields[i] = fmt.Sprintf(\"IsolationGroups: %v\", v.IsolationGroups)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListPartition{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TaskListPartition match the\n// provided TaskListPartition.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListPartition) Equals(rhs *TaskListPartition) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.IsolationGroups == nil && rhs.IsolationGroups == nil) || (v.IsolationGroups != nil && rhs.IsolationGroups != nil && _List_String_Equals(v.IsolationGroups, rhs.IsolationGroups))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListPartition.\nfunc (v *TaskListPartition) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.IsolationGroups != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"isolationGroups\", (_List_String_Zapper)(v.IsolationGroups)))\n\t}\n\treturn err\n}\n\n// GetIsolationGroups returns the value of IsolationGroups if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartition) GetIsolationGroups() (o []string) {\n\tif v != nil && v.IsolationGroups != nil {\n\t\treturn v.IsolationGroups\n\t}\n\n\treturn\n}\n\n// IsSetIsolationGroups returns true if IsolationGroups is not nil.\nfunc (v *TaskListPartition) IsSetIsolationGroups() bool {\n\treturn v != nil && v.IsolationGroups != nil\n}\n\ntype TaskListPartitionConfig struct {\n\tVersion            *int64                       `json:\"version,omitempty\"`\n\tNumReadPartitions  *int32                       `json:\"numReadPartitions,omitempty\"`\n\tNumWritePartitions *int32                       `json:\"numWritePartitions,omitempty\"`\n\tReadPartitions     map[int32]*TaskListPartition `json:\"readPartitions,omitempty\"`\n\tWritePartitions    map[int32]*TaskListPartition `json:\"writePartitions,omitempty\"`\n}\n\ntype _Map_I32_TaskListPartition_MapItemList map[int32]*TaskListPartition\n\nfunc (m _Map_I32_TaskListPartition_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*TaskListPartition', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueI32(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := v.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_I32_TaskListPartition_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_I32_TaskListPartition_MapItemList) KeyType() wire.Type {\n\treturn wire.TI32\n}\n\nfunc (_Map_I32_TaskListPartition_MapItemList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_Map_I32_TaskListPartition_MapItemList) Close() {}\n\n// ToWire translates a TaskListPartitionConfig struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TaskListPartitionConfig) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [5]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.NumReadPartitions != nil {\n\t\tw, err = wire.NewValueI32(*(v.NumReadPartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.NumWritePartitions != nil {\n\t\tw, err = wire.NewValueI32(*(v.NumWritePartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.ReadPartitions != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_TaskListPartition_MapItemList(v.ReadPartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.WritePartitions != nil {\n\t\tw, err = wire.NewValueMap(_Map_I32_TaskListPartition_MapItemList(v.WritePartitions)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TaskListPartition_Read(w wire.Value) (*TaskListPartition, error) {\n\tvar v TaskListPartition\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _Map_I32_TaskListPartition_Read(m wire.MapItemList) (map[int32]*TaskListPartition, error) {\n\tif m.KeyType() != wire.TI32 {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[int32]*TaskListPartition, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetI32(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := _TaskListPartition_Read(x.Value)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a TaskListPartitionConfig struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TaskListPartitionConfig struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TaskListPartitionConfig\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TaskListPartitionConfig) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.NumReadPartitions = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.NumWritePartitions = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.ReadPartitions, err = _Map_I32_TaskListPartition_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.WritePartitions, err = _Map_I32_TaskListPartition_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_I32_TaskListPartition_Encode(val map[int32]*TaskListPartition, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TI32,\n\t\tValueType: wire.TStruct,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[int32]*TaskListPartition', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteInt32(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a TaskListPartitionConfig struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TaskListPartitionConfig struct could not be encoded.\nfunc (v *TaskListPartitionConfig) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NumReadPartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.NumReadPartitions)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.NumWritePartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.NumWritePartitions)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ReadPartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_TaskListPartition_Encode(v.ReadPartitions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WritePartitions != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_I32_TaskListPartition_Encode(v.WritePartitions, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TaskListPartition_Decode(sr stream.Reader) (*TaskListPartition, error) {\n\tvar v TaskListPartition\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _Map_I32_TaskListPartition_Decode(sr stream.Reader) (map[int32]*TaskListPartition, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TI32 || mh.ValueType != wire.TStruct {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[int32]*TaskListPartition, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadInt32()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := _TaskListPartition_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a TaskListPartitionConfig struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TaskListPartitionConfig struct could not be generated from the wire\n// representation.\nfunc (v *TaskListPartitionConfig) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.NumReadPartitions = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.NumWritePartitions = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TMap:\n\t\t\tv.ReadPartitions, err = _Map_I32_TaskListPartition_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TMap:\n\t\t\tv.WritePartitions, err = _Map_I32_TaskListPartition_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TaskListPartitionConfig\n// struct.\nfunc (v *TaskListPartitionConfig) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [5]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.NumReadPartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumReadPartitions: %v\", *(v.NumReadPartitions))\n\t\ti++\n\t}\n\tif v.NumWritePartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"NumWritePartitions: %v\", *(v.NumWritePartitions))\n\t\ti++\n\t}\n\tif v.ReadPartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"ReadPartitions: %v\", v.ReadPartitions)\n\t\ti++\n\t}\n\tif v.WritePartitions != nil {\n\t\tfields[i] = fmt.Sprintf(\"WritePartitions: %v\", v.WritePartitions)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TaskListPartitionConfig{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_I32_TaskListPartition_Equals(lhs, rhs map[int32]*TaskListPartition) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// Equals returns true if all the fields of this TaskListPartitionConfig match the\n// provided TaskListPartitionConfig.\n//\n// This function performs a deep comparison.\nfunc (v *TaskListPartitionConfig) Equals(rhs *TaskListPartitionConfig) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.NumReadPartitions, rhs.NumReadPartitions) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.NumWritePartitions, rhs.NumWritePartitions) {\n\t\treturn false\n\t}\n\tif !((v.ReadPartitions == nil && rhs.ReadPartitions == nil) || (v.ReadPartitions != nil && rhs.ReadPartitions != nil && _Map_I32_TaskListPartition_Equals(v.ReadPartitions, rhs.ReadPartitions))) {\n\t\treturn false\n\t}\n\tif !((v.WritePartitions == nil && rhs.WritePartitions == nil) || (v.WritePartitions != nil && rhs.WritePartitions != nil && _Map_I32_TaskListPartition_Equals(v.WritePartitions, rhs.WritePartitions))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_I32_TaskListPartition_Item_Zapper struct {\n\tKey   int32\n\tValue *TaskListPartition\n}\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_TaskListPartition_Item_Zapper.\nfunc (v _Map_I32_TaskListPartition_Item_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tenc.AddInt32(\"key\", v.Key)\n\terr = multierr.Append(err, enc.AddObject(\"value\", v.Value))\n\treturn err\n}\n\ntype _Map_I32_TaskListPartition_Zapper map[int32]*TaskListPartition\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Map_I32_TaskListPartition_Zapper.\nfunc (m _Map_I32_TaskListPartition_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor k, v := range m {\n\t\terr = multierr.Append(err, enc.AppendObject(_Map_I32_TaskListPartition_Item_Zapper{Key: k, Value: v}))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TaskListPartitionConfig.\nfunc (v *TaskListPartitionConfig) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.NumReadPartitions != nil {\n\t\tenc.AddInt32(\"numReadPartitions\", *v.NumReadPartitions)\n\t}\n\tif v.NumWritePartitions != nil {\n\t\tenc.AddInt32(\"numWritePartitions\", *v.NumWritePartitions)\n\t}\n\tif v.ReadPartitions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"readPartitions\", (_Map_I32_TaskListPartition_Zapper)(v.ReadPartitions)))\n\t}\n\tif v.WritePartitions != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"writePartitions\", (_Map_I32_TaskListPartition_Zapper)(v.WritePartitions)))\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionConfig) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *TaskListPartitionConfig) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetNumReadPartitions returns the value of NumReadPartitions if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionConfig) GetNumReadPartitions() (o int32) {\n\tif v != nil && v.NumReadPartitions != nil {\n\t\treturn *v.NumReadPartitions\n\t}\n\n\treturn\n}\n\n// IsSetNumReadPartitions returns true if NumReadPartitions is not nil.\nfunc (v *TaskListPartitionConfig) IsSetNumReadPartitions() bool {\n\treturn v != nil && v.NumReadPartitions != nil\n}\n\n// GetNumWritePartitions returns the value of NumWritePartitions if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionConfig) GetNumWritePartitions() (o int32) {\n\tif v != nil && v.NumWritePartitions != nil {\n\t\treturn *v.NumWritePartitions\n\t}\n\n\treturn\n}\n\n// IsSetNumWritePartitions returns true if NumWritePartitions is not nil.\nfunc (v *TaskListPartitionConfig) IsSetNumWritePartitions() bool {\n\treturn v != nil && v.NumWritePartitions != nil\n}\n\n// GetReadPartitions returns the value of ReadPartitions if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionConfig) GetReadPartitions() (o map[int32]*TaskListPartition) {\n\tif v != nil && v.ReadPartitions != nil {\n\t\treturn v.ReadPartitions\n\t}\n\n\treturn\n}\n\n// IsSetReadPartitions returns true if ReadPartitions is not nil.\nfunc (v *TaskListPartitionConfig) IsSetReadPartitions() bool {\n\treturn v != nil && v.ReadPartitions != nil\n}\n\n// GetWritePartitions returns the value of WritePartitions if it is set or its\n// zero value if it is unset.\nfunc (v *TaskListPartitionConfig) GetWritePartitions() (o map[int32]*TaskListPartition) {\n\tif v != nil && v.WritePartitions != nil {\n\t\treturn v.WritePartitions\n\t}\n\n\treturn\n}\n\n// IsSetWritePartitions returns true if WritePartitions is not nil.\nfunc (v *TaskListPartitionConfig) IsSetWritePartitions() bool {\n\treturn v != nil && v.WritePartitions != nil\n}\n\ntype TimerInfo struct {\n\tVersion         *int64 `json:\"version,omitempty\"`\n\tStartedID       *int64 `json:\"startedID,omitempty\"`\n\tExpiryTimeNanos *int64 `json:\"expiryTimeNanos,omitempty\"`\n\tTaskID          *int64 `json:\"taskID,omitempty\"`\n}\n\n// ToWire translates a TimerInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [4]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.ExpiryTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ExpiryTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerInfo struct could not be encoded.\nfunc (v *TimerInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExpiryTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ExpiryTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerInfo struct could not be generated from the wire\n// representation.\nfunc (v *TimerInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ExpiryTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerInfo\n// struct.\nfunc (v *TimerInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [4]string\n\ti := 0\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.StartedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartedID: %v\", *(v.StartedID))\n\t\ti++\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExpiryTimeNanos: %v\", *(v.ExpiryTimeNanos))\n\t\ti++\n\t}\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerInfo match the\n// provided TimerInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TimerInfo) Equals(rhs *TimerInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartedID, rhs.StartedID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ExpiryTimeNanos, rhs.ExpiryTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerInfo.\nfunc (v *TimerInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.StartedID != nil {\n\t\tenc.AddInt64(\"startedID\", *v.StartedID)\n\t}\n\tif v.ExpiryTimeNanos != nil {\n\t\tenc.AddInt64(\"expiryTimeNanos\", *v.ExpiryTimeNanos)\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\treturn err\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *TimerInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *TimerInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetStartedID returns the value of StartedID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerInfo) GetStartedID() (o int64) {\n\tif v != nil && v.StartedID != nil {\n\t\treturn *v.StartedID\n\t}\n\n\treturn\n}\n\n// IsSetStartedID returns true if StartedID is not nil.\nfunc (v *TimerInfo) IsSetStartedID() bool {\n\treturn v != nil && v.StartedID != nil\n}\n\n// GetExpiryTimeNanos returns the value of ExpiryTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TimerInfo) GetExpiryTimeNanos() (o int64) {\n\tif v != nil && v.ExpiryTimeNanos != nil {\n\t\treturn *v.ExpiryTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetExpiryTimeNanos returns true if ExpiryTimeNanos is not nil.\nfunc (v *TimerInfo) IsSetExpiryTimeNanos() bool {\n\treturn v != nil && v.ExpiryTimeNanos != nil\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerInfo) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *TimerInfo) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\ntype TimerReference struct {\n\tTaskID              *int64 `json:\"taskID,omitempty\"`\n\tVisibilityTimestamp *int64 `json:\"visibilityTimestamp,omitempty\"`\n\tTimeoutType         *int16 `json:\"TimeoutType,omitempty\"`\n}\n\n// ToWire translates a TimerReference struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerReference) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [3]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.TaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.TaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tw, err = wire.NewValueI64(*(v.VisibilityTimestamp)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 11, Value: w}\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TimeoutType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 13, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerReference struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerReference struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerReference\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerReference) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.TaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 11:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.VisibilityTimestamp = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 13:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerReference struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerReference struct could not be encoded.\nfunc (v *TimerReference) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.TaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.TaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityTimestamp != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 11, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.VisibilityTimestamp)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 13, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TimeoutType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerReference struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerReference struct could not be generated from the wire\n// representation.\nfunc (v *TimerReference) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.TaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 11 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.VisibilityTimestamp = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 13 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerReference\n// struct.\nfunc (v *TimerReference) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [3]string\n\ti := 0\n\tif v.TaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskID: %v\", *(v.TaskID))\n\t\ti++\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityTimestamp: %v\", *(v.VisibilityTimestamp))\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerReference{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerReference match the\n// provided TimerReference.\n//\n// This function performs a deep comparison.\nfunc (v *TimerReference) Equals(rhs *TimerReference) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.TaskID, rhs.TaskID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.VisibilityTimestamp, rhs.VisibilityTimestamp) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerReference.\nfunc (v *TimerReference) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.TaskID != nil {\n\t\tenc.AddInt64(\"taskID\", *v.TaskID)\n\t}\n\tif v.VisibilityTimestamp != nil {\n\t\tenc.AddInt64(\"visibilityTimestamp\", *v.VisibilityTimestamp)\n\t}\n\tif v.TimeoutType != nil {\n\t\tenc.AddInt16(\"TimeoutType\", *v.TimeoutType)\n\t}\n\treturn err\n}\n\n// GetTaskID returns the value of TaskID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerReference) GetTaskID() (o int64) {\n\tif v != nil && v.TaskID != nil {\n\t\treturn *v.TaskID\n\t}\n\n\treturn\n}\n\n// IsSetTaskID returns true if TaskID is not nil.\nfunc (v *TimerReference) IsSetTaskID() bool {\n\treturn v != nil && v.TaskID != nil\n}\n\n// GetVisibilityTimestamp returns the value of VisibilityTimestamp if it is set or its\n// zero value if it is unset.\nfunc (v *TimerReference) GetVisibilityTimestamp() (o int64) {\n\tif v != nil && v.VisibilityTimestamp != nil {\n\t\treturn *v.VisibilityTimestamp\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityTimestamp returns true if VisibilityTimestamp is not nil.\nfunc (v *TimerReference) IsSetVisibilityTimestamp() bool {\n\treturn v != nil && v.VisibilityTimestamp != nil\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *TimerReference) GetTimeoutType() (o int16) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *TimerReference) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\ntype TimerTaskInfo struct {\n\tDomainID        []byte  `json:\"domainID,omitempty\"`\n\tWorkflowID      *string `json:\"workflowID,omitempty\"`\n\tRunID           []byte  `json:\"runID,omitempty\"`\n\tTaskType        *int16  `json:\"taskType,omitempty\"`\n\tTimeoutType     *int16  `json:\"timeoutType,omitempty\"`\n\tVersion         *int64  `json:\"version,omitempty\"`\n\tScheduleAttempt *int64  `json:\"scheduleAttempt,omitempty\"`\n\tEventID         *int64  `json:\"eventID,omitempty\"`\n\tTaskList        *string `json:\"taskList,omitempty\"`\n}\n\n// ToWire translates a TimerTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TimerTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [9]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueBinary(v.DomainID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueBinary(v.RunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TimeoutType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.EventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.EventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\n// FromWire deserializes a TimerTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TimerTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TimerTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TimerTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.DomainID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.RunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TimeoutType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.EventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Encode serializes a TimerTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TimerTaskInfo struct could not be encoded.\nfunc (v *TimerTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.DomainID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.RunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TimeoutType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TimeoutType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.EventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\n// Decode deserializes a TimerTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TimerTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *TimerTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.DomainID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tv.RunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TimeoutType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.EventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TimerTaskInfo\n// struct.\nfunc (v *TimerTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [9]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", v.DomainID)\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", v.RunID)\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.TimeoutType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TimeoutType: %v\", *(v.TimeoutType))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.ScheduleAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleAttempt: %v\", *(v.ScheduleAttempt))\n\t\ti++\n\t}\n\tif v.EventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventID: %v\", *(v.EventID))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TimerTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\n// Equals returns true if all the fields of this TimerTaskInfo match the\n// provided TimerTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TimerTaskInfo) Equals(rhs *TimerTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainID == nil && rhs.DomainID == nil) || (v.DomainID != nil && rhs.DomainID != nil && bytes.Equal(v.DomainID, rhs.DomainID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.RunID == nil && rhs.RunID == nil) || (v.RunID != nil && rhs.RunID != nil && bytes.Equal(v.RunID, rhs.RunID))) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TimeoutType, rhs.TimeoutType) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleAttempt, rhs.ScheduleAttempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.EventID, rhs.EventID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TimerTaskInfo.\nfunc (v *TimerTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", base64.StdEncoding.EncodeToString(v.DomainID))\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", base64.StdEncoding.EncodeToString(v.RunID))\n\t}\n\tif v.TaskType != nil {\n\t\tenc.AddInt16(\"taskType\", *v.TaskType)\n\t}\n\tif v.TimeoutType != nil {\n\t\tenc.AddInt16(\"timeoutType\", *v.TimeoutType)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.ScheduleAttempt != nil {\n\t\tenc.AddInt64(\"scheduleAttempt\", *v.ScheduleAttempt)\n\t}\n\tif v.EventID != nil {\n\t\tenc.AddInt64(\"eventID\", *v.EventID)\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetDomainID() (o []byte) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *TimerTaskInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *TimerTaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetRunID() (o []byte) {\n\tif v != nil && v.RunID != nil {\n\t\treturn v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *TimerTaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetTaskType() (o int16) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *TimerTaskInfo) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetTimeoutType returns the value of TimeoutType if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetTimeoutType() (o int16) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\n\treturn\n}\n\n// IsSetTimeoutType returns true if TimeoutType is not nil.\nfunc (v *TimerTaskInfo) IsSetTimeoutType() bool {\n\treturn v != nil && v.TimeoutType != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *TimerTaskInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetScheduleAttempt returns the value of ScheduleAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetScheduleAttempt() (o int64) {\n\tif v != nil && v.ScheduleAttempt != nil {\n\t\treturn *v.ScheduleAttempt\n\t}\n\n\treturn\n}\n\n// IsSetScheduleAttempt returns true if ScheduleAttempt is not nil.\nfunc (v *TimerTaskInfo) IsSetScheduleAttempt() bool {\n\treturn v != nil && v.ScheduleAttempt != nil\n}\n\n// GetEventID returns the value of EventID if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetEventID() (o int64) {\n\tif v != nil && v.EventID != nil {\n\t\treturn *v.EventID\n\t}\n\n\treturn\n}\n\n// IsSetEventID returns true if EventID is not nil.\nfunc (v *TimerTaskInfo) IsSetEventID() bool {\n\treturn v != nil && v.EventID != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *TimerTaskInfo) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *TimerTaskInfo) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\ntype TransferTaskInfo struct {\n\tDomainID                 []byte               `json:\"domainID,omitempty\"`\n\tWorkflowID               *string              `json:\"workflowID,omitempty\"`\n\tRunID                    []byte               `json:\"runID,omitempty\"`\n\tTaskType                 *int16               `json:\"taskType,omitempty\"`\n\tTargetDomainID           []byte               `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID         *string              `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID              []byte               `json:\"targetRunID,omitempty\"`\n\tTaskList                 *string              `json:\"taskList,omitempty\"`\n\tTargetChildWorkflowOnly  *bool                `json:\"targetChildWorkflowOnly,omitempty\"`\n\tScheduleID               *int64               `json:\"scheduleID,omitempty\"`\n\tVersion                  *int64               `json:\"version,omitempty\"`\n\tVisibilityTimestampNanos *int64               `json:\"visibilityTimestampNanos,omitempty\"`\n\tTargetDomainIDs          [][]byte             `json:\"targetDomainIDs,omitempty\"`\n\tOriginalTaskList         *string              `json:\"originalTaskList,omitempty\"`\n\tOriginalTaskListKind     *shared.TaskListKind `json:\"originalTaskListKind,omitempty\"`\n}\n\ntype _Set_Binary_sliceType_ValueList [][]byte\n\nfunc (v _Set_Binary_sliceType_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor _, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid set '[]byte': contains nil value\")\n\t\t}\n\t\tw, err := wire.NewValueBinary(x), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif err := f(w); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _Set_Binary_sliceType_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_Set_Binary_sliceType_ValueList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Set_Binary_sliceType_ValueList) Close() {}\n\n// ToWire translates a TransferTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *TransferTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [15]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.DomainID != nil {\n\t\tw, err = wire.NewValueBinary(v.DomainID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tw, err = wire.NewValueBinary(v.RunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tw, err = wire.NewValueI16(*(v.TaskType)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.TargetDomainID != nil {\n\t\tw, err = wire.NewValueBinary(v.TargetDomainID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.TargetWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tw, err = wire.NewValueBinary(v.TargetRunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.TargetChildWorkflowOnly != nil {\n\t\tw, err = wire.NewValueBool(*(v.TargetChildWorkflowOnly)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tw, err = wire.NewValueI64(*(v.ScheduleID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tw, err = wire.NewValueI64(*(v.Version)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.VisibilityTimestampNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.VisibilityTimestampNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 32, Value: w}\n\t\ti++\n\t}\n\tif v.TargetDomainIDs != nil {\n\t\tw, err = wire.NewValueSet(_Set_Binary_sliceType_ValueList(v.TargetDomainIDs)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.OriginalTaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.OriginalTaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.OriginalTaskListKind != nil {\n\t\tw, err = v.OriginalTaskListKind.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Set_Binary_sliceType_Read(s wire.ValueList) ([][]byte, error) {\n\tif s.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make([][]byte, 0, s.Size())\n\terr := s.ForEach(func(x wire.Value) error {\n\t\ti, err := x.GetBinary(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\ts.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a TransferTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a TransferTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v TransferTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *TransferTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.DomainID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.RunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI16 {\n\t\t\t\tvar x int16\n\t\t\t\tx, err = field.Value.GetI16(), error(nil)\n\t\t\t\tv.TaskType = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TargetDomainID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TargetWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.TargetRunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.TargetChildWorkflowOnly = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.ScheduleID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.Version = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 32:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.VisibilityTimestampNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TSet {\n\t\t\t\tv.TargetDomainIDs, err = _Set_Binary_sliceType_Read(field.Value.GetSet())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.OriginalTaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x shared.TaskListKind\n\t\t\t\tx, err = _TaskListKind_Read(field.Value)\n\t\t\t\tv.OriginalTaskListKind = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Set_Binary_sliceType_Encode(val [][]byte, sw stream.Writer) error {\n\n\tsh := stream.SetHeader{\n\t\tType:   wire.TBinary,\n\t\tLength: len(val),\n\t}\n\n\tif err := sw.WriteSetBegin(sh); err != nil {\n\t\treturn err\n\t}\n\n\tfor _, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid set '[]byte': contains nil value\")\n\t\t}\n\n\t\tif err := sw.WriteBinary(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteSetEnd()\n}\n\n// Encode serializes a TransferTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a TransferTaskInfo struct could not be encoded.\nfunc (v *TransferTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.DomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.DomainID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.RunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskType != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI16}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt16(*(v.TaskType)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TargetDomainID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TargetWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.TargetRunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetChildWorkflowOnly != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.TargetChildWorkflowOnly)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ScheduleID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.ScheduleID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Version != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.Version)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VisibilityTimestampNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 32, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.VisibilityTimestampNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TargetDomainIDs != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TSet}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Set_Binary_sliceType_Encode(v.TargetDomainIDs, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OriginalTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.OriginalTaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.OriginalTaskListKind != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.OriginalTaskListKind.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Set_Binary_sliceType_Decode(sr stream.Reader) ([][]byte, error) {\n\tsh, err := sr.ReadSetBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif sh.Type != wire.TBinary {\n\t\tfor i := 0; i < sh.Length; i++ {\n\t\t\tif err := sr.Skip(sh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadSetEnd()\n\t}\n\n\to := make([][]byte, 0, sh.Length)\n\tfor i := 0; i < sh.Length; i++ {\n\t\tv, err := sr.ReadBinary()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadSetEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a TransferTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a TransferTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *TransferTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.DomainID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tv.RunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI16:\n\t\t\tvar x int16\n\t\t\tx, err = sr.ReadInt16()\n\t\t\tv.TaskType = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TBinary:\n\t\t\tv.TargetDomainID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TargetWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TBinary:\n\t\t\tv.TargetRunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.TargetChildWorkflowOnly = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.ScheduleID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.Version = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 32 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.VisibilityTimestampNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TSet:\n\t\t\tv.TargetDomainIDs, err = _Set_Binary_sliceType_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.OriginalTaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TI32:\n\t\t\tvar x shared.TaskListKind\n\t\t\tx, err = _TaskListKind_Decode(sr)\n\t\t\tv.OriginalTaskListKind = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a TransferTaskInfo\n// struct.\nfunc (v *TransferTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [15]string\n\ti := 0\n\tif v.DomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DomainID: %v\", v.DomainID)\n\t\ti++\n\t}\n\tif v.WorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowID: %v\", *(v.WorkflowID))\n\t\ti++\n\t}\n\tif v.RunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"RunID: %v\", v.RunID)\n\t\ti++\n\t}\n\tif v.TaskType != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskType: %v\", *(v.TaskType))\n\t\ti++\n\t}\n\tif v.TargetDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainID: %v\", v.TargetDomainID)\n\t\ti++\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetWorkflowID: %v\", *(v.TargetWorkflowID))\n\t\ti++\n\t}\n\tif v.TargetRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetRunID: %v\", v.TargetRunID)\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\tif v.TargetChildWorkflowOnly != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetChildWorkflowOnly: %v\", *(v.TargetChildWorkflowOnly))\n\t\ti++\n\t}\n\tif v.ScheduleID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ScheduleID: %v\", *(v.ScheduleID))\n\t\ti++\n\t}\n\tif v.Version != nil {\n\t\tfields[i] = fmt.Sprintf(\"Version: %v\", *(v.Version))\n\t\ti++\n\t}\n\tif v.VisibilityTimestampNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"VisibilityTimestampNanos: %v\", *(v.VisibilityTimestampNanos))\n\t\ti++\n\t}\n\tif v.TargetDomainIDs != nil {\n\t\tfields[i] = fmt.Sprintf(\"TargetDomainIDs: %v\", v.TargetDomainIDs)\n\t\ti++\n\t}\n\tif v.OriginalTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"OriginalTaskList: %v\", *(v.OriginalTaskList))\n\t\ti++\n\t}\n\tif v.OriginalTaskListKind != nil {\n\t\tfields[i] = fmt.Sprintf(\"OriginalTaskListKind: %v\", *(v.OriginalTaskListKind))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"TransferTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Set_Binary_sliceType_Equals(lhs, rhs [][]byte) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor _, x := range lhs {\n\t\tok := false\n\t\tfor _, y := range rhs {\n\t\t\tif bytes.Equal(x, y) {\n\t\t\t\tok = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this TransferTaskInfo match the\n// provided TransferTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *TransferTaskInfo) Equals(rhs *TransferTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.DomainID == nil && rhs.DomainID == nil) || (v.DomainID != nil && rhs.DomainID != nil && bytes.Equal(v.DomainID, rhs.DomainID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowID, rhs.WorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.RunID == nil && rhs.RunID == nil) || (v.RunID != nil && rhs.RunID != nil && bytes.Equal(v.RunID, rhs.RunID))) {\n\t\treturn false\n\t}\n\tif !_I16_EqualsPtr(v.TaskType, rhs.TaskType) {\n\t\treturn false\n\t}\n\tif !((v.TargetDomainID == nil && rhs.TargetDomainID == nil) || (v.TargetDomainID != nil && rhs.TargetDomainID != nil && bytes.Equal(v.TargetDomainID, rhs.TargetDomainID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TargetWorkflowID, rhs.TargetWorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.TargetRunID == nil && rhs.TargetRunID == nil) || (v.TargetRunID != nil && rhs.TargetRunID != nil && bytes.Equal(v.TargetRunID, rhs.TargetRunID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.TargetChildWorkflowOnly, rhs.TargetChildWorkflowOnly) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.ScheduleID, rhs.ScheduleID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.Version, rhs.Version) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.VisibilityTimestampNanos, rhs.VisibilityTimestampNanos) {\n\t\treturn false\n\t}\n\tif !((v.TargetDomainIDs == nil && rhs.TargetDomainIDs == nil) || (v.TargetDomainIDs != nil && rhs.TargetDomainIDs != nil && _Set_Binary_sliceType_Equals(v.TargetDomainIDs, rhs.TargetDomainIDs))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.OriginalTaskList, rhs.OriginalTaskList) {\n\t\treturn false\n\t}\n\tif !_TaskListKind_EqualsPtr(v.OriginalTaskListKind, rhs.OriginalTaskListKind) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Set_Binary_sliceType_Zapper [][]byte\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _Set_Binary_sliceType_Zapper.\nfunc (s _Set_Binary_sliceType_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range s {\n\t\tenc.AppendString(base64.StdEncoding.EncodeToString(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of TransferTaskInfo.\nfunc (v *TransferTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.DomainID != nil {\n\t\tenc.AddString(\"domainID\", base64.StdEncoding.EncodeToString(v.DomainID))\n\t}\n\tif v.WorkflowID != nil {\n\t\tenc.AddString(\"workflowID\", *v.WorkflowID)\n\t}\n\tif v.RunID != nil {\n\t\tenc.AddString(\"runID\", base64.StdEncoding.EncodeToString(v.RunID))\n\t}\n\tif v.TaskType != nil {\n\t\tenc.AddInt16(\"taskType\", *v.TaskType)\n\t}\n\tif v.TargetDomainID != nil {\n\t\tenc.AddString(\"targetDomainID\", base64.StdEncoding.EncodeToString(v.TargetDomainID))\n\t}\n\tif v.TargetWorkflowID != nil {\n\t\tenc.AddString(\"targetWorkflowID\", *v.TargetWorkflowID)\n\t}\n\tif v.TargetRunID != nil {\n\t\tenc.AddString(\"targetRunID\", base64.StdEncoding.EncodeToString(v.TargetRunID))\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\tif v.TargetChildWorkflowOnly != nil {\n\t\tenc.AddBool(\"targetChildWorkflowOnly\", *v.TargetChildWorkflowOnly)\n\t}\n\tif v.ScheduleID != nil {\n\t\tenc.AddInt64(\"scheduleID\", *v.ScheduleID)\n\t}\n\tif v.Version != nil {\n\t\tenc.AddInt64(\"version\", *v.Version)\n\t}\n\tif v.VisibilityTimestampNanos != nil {\n\t\tenc.AddInt64(\"visibilityTimestampNanos\", *v.VisibilityTimestampNanos)\n\t}\n\tif v.TargetDomainIDs != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"targetDomainIDs\", (_Set_Binary_sliceType_Zapper)(v.TargetDomainIDs)))\n\t}\n\tif v.OriginalTaskList != nil {\n\t\tenc.AddString(\"originalTaskList\", *v.OriginalTaskList)\n\t}\n\tif v.OriginalTaskListKind != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"originalTaskListKind\", *v.OriginalTaskListKind))\n\t}\n\treturn err\n}\n\n// GetDomainID returns the value of DomainID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetDomainID() (o []byte) {\n\tif v != nil && v.DomainID != nil {\n\t\treturn v.DomainID\n\t}\n\n\treturn\n}\n\n// IsSetDomainID returns true if DomainID is not nil.\nfunc (v *TransferTaskInfo) IsSetDomainID() bool {\n\treturn v != nil && v.DomainID != nil\n}\n\n// GetWorkflowID returns the value of WorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil && v.WorkflowID != nil {\n\t\treturn *v.WorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowID returns true if WorkflowID is not nil.\nfunc (v *TransferTaskInfo) IsSetWorkflowID() bool {\n\treturn v != nil && v.WorkflowID != nil\n}\n\n// GetRunID returns the value of RunID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetRunID() (o []byte) {\n\tif v != nil && v.RunID != nil {\n\t\treturn v.RunID\n\t}\n\n\treturn\n}\n\n// IsSetRunID returns true if RunID is not nil.\nfunc (v *TransferTaskInfo) IsSetRunID() bool {\n\treturn v != nil && v.RunID != nil\n}\n\n// GetTaskType returns the value of TaskType if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTaskType() (o int16) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\n\treturn\n}\n\n// IsSetTaskType returns true if TaskType is not nil.\nfunc (v *TransferTaskInfo) IsSetTaskType() bool {\n\treturn v != nil && v.TaskType != nil\n}\n\n// GetTargetDomainID returns the value of TargetDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTargetDomainID() (o []byte) {\n\tif v != nil && v.TargetDomainID != nil {\n\t\treturn v.TargetDomainID\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainID returns true if TargetDomainID is not nil.\nfunc (v *TransferTaskInfo) IsSetTargetDomainID() bool {\n\treturn v != nil && v.TargetDomainID != nil\n}\n\n// GetTargetWorkflowID returns the value of TargetWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTargetWorkflowID() (o string) {\n\tif v != nil && v.TargetWorkflowID != nil {\n\t\treturn *v.TargetWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetTargetWorkflowID returns true if TargetWorkflowID is not nil.\nfunc (v *TransferTaskInfo) IsSetTargetWorkflowID() bool {\n\treturn v != nil && v.TargetWorkflowID != nil\n}\n\n// GetTargetRunID returns the value of TargetRunID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTargetRunID() (o []byte) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn v.TargetRunID\n\t}\n\n\treturn\n}\n\n// IsSetTargetRunID returns true if TargetRunID is not nil.\nfunc (v *TransferTaskInfo) IsSetTargetRunID() bool {\n\treturn v != nil && v.TargetRunID != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *TransferTaskInfo) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTargetChildWorkflowOnly returns the value of TargetChildWorkflowOnly if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTargetChildWorkflowOnly() (o bool) {\n\tif v != nil && v.TargetChildWorkflowOnly != nil {\n\t\treturn *v.TargetChildWorkflowOnly\n\t}\n\n\treturn\n}\n\n// IsSetTargetChildWorkflowOnly returns true if TargetChildWorkflowOnly is not nil.\nfunc (v *TransferTaskInfo) IsSetTargetChildWorkflowOnly() bool {\n\treturn v != nil && v.TargetChildWorkflowOnly != nil\n}\n\n// GetScheduleID returns the value of ScheduleID if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetScheduleID() (o int64) {\n\tif v != nil && v.ScheduleID != nil {\n\t\treturn *v.ScheduleID\n\t}\n\n\treturn\n}\n\n// IsSetScheduleID returns true if ScheduleID is not nil.\nfunc (v *TransferTaskInfo) IsSetScheduleID() bool {\n\treturn v != nil && v.ScheduleID != nil\n}\n\n// GetVersion returns the value of Version if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetVersion() (o int64) {\n\tif v != nil && v.Version != nil {\n\t\treturn *v.Version\n\t}\n\n\treturn\n}\n\n// IsSetVersion returns true if Version is not nil.\nfunc (v *TransferTaskInfo) IsSetVersion() bool {\n\treturn v != nil && v.Version != nil\n}\n\n// GetVisibilityTimestampNanos returns the value of VisibilityTimestampNanos if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetVisibilityTimestampNanos() (o int64) {\n\tif v != nil && v.VisibilityTimestampNanos != nil {\n\t\treturn *v.VisibilityTimestampNanos\n\t}\n\n\treturn\n}\n\n// IsSetVisibilityTimestampNanos returns true if VisibilityTimestampNanos is not nil.\nfunc (v *TransferTaskInfo) IsSetVisibilityTimestampNanos() bool {\n\treturn v != nil && v.VisibilityTimestampNanos != nil\n}\n\n// GetTargetDomainIDs returns the value of TargetDomainIDs if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetTargetDomainIDs() (o [][]byte) {\n\tif v != nil && v.TargetDomainIDs != nil {\n\t\treturn v.TargetDomainIDs\n\t}\n\n\treturn\n}\n\n// IsSetTargetDomainIDs returns true if TargetDomainIDs is not nil.\nfunc (v *TransferTaskInfo) IsSetTargetDomainIDs() bool {\n\treturn v != nil && v.TargetDomainIDs != nil\n}\n\n// GetOriginalTaskList returns the value of OriginalTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetOriginalTaskList() (o string) {\n\tif v != nil && v.OriginalTaskList != nil {\n\t\treturn *v.OriginalTaskList\n\t}\n\n\treturn\n}\n\n// IsSetOriginalTaskList returns true if OriginalTaskList is not nil.\nfunc (v *TransferTaskInfo) IsSetOriginalTaskList() bool {\n\treturn v != nil && v.OriginalTaskList != nil\n}\n\n// GetOriginalTaskListKind returns the value of OriginalTaskListKind if it is set or its\n// zero value if it is unset.\nfunc (v *TransferTaskInfo) GetOriginalTaskListKind() (o shared.TaskListKind) {\n\tif v != nil && v.OriginalTaskListKind != nil {\n\t\treturn *v.OriginalTaskListKind\n\t}\n\n\treturn\n}\n\n// IsSetOriginalTaskListKind returns true if OriginalTaskListKind is not nil.\nfunc (v *TransferTaskInfo) IsSetOriginalTaskListKind() bool {\n\treturn v != nil && v.OriginalTaskListKind != nil\n}\n\ntype WorkflowExecutionInfo struct {\n\tParentDomainID                          []byte                    `json:\"parentDomainID,omitempty\"`\n\tParentWorkflowID                        *string                   `json:\"parentWorkflowID,omitempty\"`\n\tParentRunID                             []byte                    `json:\"parentRunID,omitempty\"`\n\tInitiatedID                             *int64                    `json:\"initiatedID,omitempty\"`\n\tCompletionEventBatchID                  *int64                    `json:\"completionEventBatchID,omitempty\"`\n\tCompletionEvent                         []byte                    `json:\"completionEvent,omitempty\"`\n\tCompletionEventEncoding                 *string                   `json:\"completionEventEncoding,omitempty\"`\n\tTaskList                                *string                   `json:\"taskList,omitempty\"`\n\tTaskListKind                            *shared.TaskListKind      `json:\"taskListKind,omitempty\"`\n\tWorkflowTypeName                        *string                   `json:\"workflowTypeName,omitempty\"`\n\tWorkflowTimeoutSeconds                  *int32                    `json:\"workflowTimeoutSeconds,omitempty\"`\n\tDecisionTaskTimeoutSeconds              *int32                    `json:\"decisionTaskTimeoutSeconds,omitempty\"`\n\tExecutionContext                        []byte                    `json:\"executionContext,omitempty\"`\n\tState                                   *int32                    `json:\"state,omitempty\"`\n\tCloseStatus                             *int32                    `json:\"closeStatus,omitempty\"`\n\tStartVersion                            *int64                    `json:\"startVersion,omitempty\"`\n\tLastWriteEventID                        *int64                    `json:\"lastWriteEventID,omitempty\"`\n\tLastEventTaskID                         *int64                    `json:\"lastEventTaskID,omitempty\"`\n\tLastFirstEventID                        *int64                    `json:\"lastFirstEventID,omitempty\"`\n\tLastProcessedEvent                      *int64                    `json:\"lastProcessedEvent,omitempty\"`\n\tStartTimeNanos                          *int64                    `json:\"startTimeNanos,omitempty\"`\n\tLastUpdatedTimeNanos                    *int64                    `json:\"lastUpdatedTimeNanos,omitempty\"`\n\tDecisionVersion                         *int64                    `json:\"decisionVersion,omitempty\"`\n\tDecisionScheduleID                      *int64                    `json:\"decisionScheduleID,omitempty\"`\n\tDecisionStartedID                       *int64                    `json:\"decisionStartedID,omitempty\"`\n\tDecisionTimeout                         *int32                    `json:\"decisionTimeout,omitempty\"`\n\tDecisionAttempt                         *int64                    `json:\"decisionAttempt,omitempty\"`\n\tDecisionStartedTimestampNanos           *int64                    `json:\"decisionStartedTimestampNanos,omitempty\"`\n\tDecisionScheduledTimestampNanos         *int64                    `json:\"decisionScheduledTimestampNanos,omitempty\"`\n\tCancelRequested                         *bool                     `json:\"cancelRequested,omitempty\"`\n\tDecisionOriginalScheduledTimestampNanos *int64                    `json:\"decisionOriginalScheduledTimestampNanos,omitempty\"`\n\tCreateRequestID                         *string                   `json:\"createRequestID,omitempty\"`\n\tDecisionRequestID                       *string                   `json:\"decisionRequestID,omitempty\"`\n\tCancelRequestID                         *string                   `json:\"cancelRequestID,omitempty\"`\n\tStickyTaskList                          *string                   `json:\"stickyTaskList,omitempty\"`\n\tStickyScheduleToStartTimeout            *int64                    `json:\"stickyScheduleToStartTimeout,omitempty\"`\n\tRetryAttempt                            *int64                    `json:\"retryAttempt,omitempty\"`\n\tRetryInitialIntervalSeconds             *int32                    `json:\"retryInitialIntervalSeconds,omitempty\"`\n\tRetryMaximumIntervalSeconds             *int32                    `json:\"retryMaximumIntervalSeconds,omitempty\"`\n\tRetryMaximumAttempts                    *int32                    `json:\"retryMaximumAttempts,omitempty\"`\n\tRetryExpirationSeconds                  *int32                    `json:\"retryExpirationSeconds,omitempty\"`\n\tRetryBackoffCoefficient                 *float64                  `json:\"retryBackoffCoefficient,omitempty\"`\n\tRetryExpirationTimeNanos                *int64                    `json:\"retryExpirationTimeNanos,omitempty\"`\n\tRetryNonRetryableErrors                 []string                  `json:\"retryNonRetryableErrors,omitempty\"`\n\tHasRetryPolicy                          *bool                     `json:\"hasRetryPolicy,omitempty\"`\n\tCronSchedule                            *string                   `json:\"cronSchedule,omitempty\"`\n\tEventStoreVersion                       *int32                    `json:\"eventStoreVersion,omitempty\"`\n\tEventBranchToken                        []byte                    `json:\"eventBranchToken,omitempty\"`\n\tSignalCount                             *int64                    `json:\"signalCount,omitempty\"`\n\tHistorySize                             *int64                    `json:\"historySize,omitempty\"`\n\tClientLibraryVersion                    *string                   `json:\"clientLibraryVersion,omitempty\"`\n\tClientFeatureVersion                    *string                   `json:\"clientFeatureVersion,omitempty\"`\n\tClientImpl                              *string                   `json:\"clientImpl,omitempty\"`\n\tAutoResetPoints                         []byte                    `json:\"autoResetPoints,omitempty\"`\n\tAutoResetPointsEncoding                 *string                   `json:\"autoResetPointsEncoding,omitempty\"`\n\tSearchAttributes                        map[string][]byte         `json:\"searchAttributes,omitempty\"`\n\tMemo                                    map[string][]byte         `json:\"memo,omitempty\"`\n\tVersionHistories                        []byte                    `json:\"versionHistories,omitempty\"`\n\tVersionHistoriesEncoding                *string                   `json:\"versionHistoriesEncoding,omitempty\"`\n\tFirstExecutionRunID                     []byte                    `json:\"firstExecutionRunID,omitempty\"`\n\tPartitionConfig                         map[string]string         `json:\"partitionConfig,omitempty\"`\n\tChecksum                                []byte                    `json:\"checksum,omitempty\"`\n\tChecksumEncoding                        *string                   `json:\"checksumEncoding,omitempty\"`\n\tCronOverlapPolicy                       *shared.CronOverlapPolicy `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy            []byte                    `json:\"activeClusterSelectionPolicy,omitempty\"`\n\tActiveClusterSelectionPolicyEncoding    *string                   `json:\"activeClusterSelectionPolicyEncoding,omitempty\"`\n}\n\ntype _Map_String_Binary_MapItemList map[string][]byte\n\nfunc (m _Map_String_Binary_MapItemList) ForEach(f func(wire.MapItem) error) error {\n\tfor k, v := range m {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]byte', key [%v]: value is nil\", k)\n\t\t}\n\t\tkw, err := wire.NewValueString(k), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tvw, err := wire.NewValueBinary(v), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(wire.MapItem{Key: kw, Value: vw})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m _Map_String_Binary_MapItemList) Size() int {\n\treturn len(m)\n}\n\nfunc (_Map_String_Binary_MapItemList) KeyType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Binary_MapItemList) ValueType() wire.Type {\n\treturn wire.TBinary\n}\n\nfunc (_Map_String_Binary_MapItemList) Close() {}\n\n// ToWire translates a WorkflowExecutionInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowExecutionInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [66]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.ParentDomainID != nil {\n\t\tw, err = wire.NewValueBinary(v.ParentDomainID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\tif v.ParentWorkflowID != nil {\n\t\tw, err = wire.NewValueString(*(v.ParentWorkflowID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 12, Value: w}\n\t\ti++\n\t}\n\tif v.ParentRunID != nil {\n\t\tw, err = wire.NewValueBinary(v.ParentRunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 14, Value: w}\n\t\ti++\n\t}\n\tif v.InitiatedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.InitiatedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 16, Value: w}\n\t\ti++\n\t}\n\tif v.CompletionEventBatchID != nil {\n\t\tw, err = wire.NewValueI64(*(v.CompletionEventBatchID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 18, Value: w}\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tw, err = wire.NewValueBinary(v.CompletionEvent), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 20, Value: w}\n\t\ti++\n\t}\n\tif v.CompletionEventEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.CompletionEventEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 22, Value: w}\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.TaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 24, Value: w}\n\t\ti++\n\t}\n\tif v.TaskListKind != nil {\n\t\tw, err = v.TaskListKind.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 25, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tw, err = wire.NewValueString(*(v.WorkflowTypeName)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 26, Value: w}\n\t\ti++\n\t}\n\tif v.WorkflowTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.WorkflowTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 28, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTaskTimeoutSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.DecisionTaskTimeoutSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 30, Value: w}\n\t\ti++\n\t}\n\tif v.ExecutionContext != nil {\n\t\tw, err = wire.NewValueBinary(v.ExecutionContext), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 32, Value: w}\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tw, err = wire.NewValueI32(*(v.State)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 34, Value: w}\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tw, err = wire.NewValueI32(*(v.CloseStatus)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 36, Value: w}\n\t\ti++\n\t}\n\tif v.StartVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 38, Value: w}\n\t\ti++\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastWriteEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 44, Value: w}\n\t\ti++\n\t}\n\tif v.LastEventTaskID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastEventTaskID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 48, Value: w}\n\t\ti++\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastFirstEventID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 50, Value: w}\n\t\ti++\n\t}\n\tif v.LastProcessedEvent != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastProcessedEvent)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 52, Value: w}\n\t\ti++\n\t}\n\tif v.StartTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.StartTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 54, Value: w}\n\t\ti++\n\t}\n\tif v.LastUpdatedTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.LastUpdatedTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 56, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionVersion != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 58, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionScheduleID != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionScheduleID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 60, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionStartedID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 62, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionTimeout != nil {\n\t\tw, err = wire.NewValueI32(*(v.DecisionTimeout)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 64, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 66, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionStartedTimestampNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionStartedTimestampNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 68, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionScheduledTimestampNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionScheduledTimestampNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 69, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequested != nil {\n\t\tw, err = wire.NewValueBool(*(v.CancelRequested)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 70, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionOriginalScheduledTimestampNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.DecisionOriginalScheduledTimestampNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 71, Value: w}\n\t\ti++\n\t}\n\tif v.CreateRequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.CreateRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 72, Value: w}\n\t\ti++\n\t}\n\tif v.DecisionRequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.DecisionRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 74, Value: w}\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tw, err = wire.NewValueString(*(v.CancelRequestID)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 76, Value: w}\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tw, err = wire.NewValueString(*(v.StickyTaskList)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 78, Value: w}\n\t\ti++\n\t}\n\tif v.StickyScheduleToStartTimeout != nil {\n\t\tw, err = wire.NewValueI64(*(v.StickyScheduleToStartTimeout)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 80, Value: w}\n\t\ti++\n\t}\n\tif v.RetryAttempt != nil {\n\t\tw, err = wire.NewValueI64(*(v.RetryAttempt)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 82, Value: w}\n\t\ti++\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryInitialIntervalSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 84, Value: w}\n\t\ti++\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryMaximumIntervalSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 86, Value: w}\n\t\ti++\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryMaximumAttempts)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 88, Value: w}\n\t\ti++\n\t}\n\tif v.RetryExpirationSeconds != nil {\n\t\tw, err = wire.NewValueI32(*(v.RetryExpirationSeconds)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 90, Value: w}\n\t\ti++\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tw, err = wire.NewValueDouble(*(v.RetryBackoffCoefficient)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 92, Value: w}\n\t\ti++\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tw, err = wire.NewValueI64(*(v.RetryExpirationTimeNanos)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 94, Value: w}\n\t\ti++\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\tw, err = wire.NewValueList(_List_String_ValueList(v.RetryNonRetryableErrors)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 96, Value: w}\n\t\ti++\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tw, err = wire.NewValueBool(*(v.HasRetryPolicy)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 98, Value: w}\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tw, err = wire.NewValueString(*(v.CronSchedule)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 100, Value: w}\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tw, err = wire.NewValueI32(*(v.EventStoreVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 102, Value: w}\n\t\ti++\n\t}\n\tif v.EventBranchToken != nil {\n\t\tw, err = wire.NewValueBinary(v.EventBranchToken), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 104, Value: w}\n\t\ti++\n\t}\n\tif v.SignalCount != nil {\n\t\tw, err = wire.NewValueI64(*(v.SignalCount)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 106, Value: w}\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tw, err = wire.NewValueI64(*(v.HistorySize)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 108, Value: w}\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientLibraryVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 110, Value: w}\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientFeatureVersion)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 112, Value: w}\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tw, err = wire.NewValueString(*(v.ClientImpl)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 114, Value: w}\n\t\ti++\n\t}\n\tif v.AutoResetPoints != nil {\n\t\tw, err = wire.NewValueBinary(v.AutoResetPoints), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 115, Value: w}\n\t\ti++\n\t}\n\tif v.AutoResetPointsEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.AutoResetPointsEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 116, Value: w}\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Binary_MapItemList(v.SearchAttributes)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 118, Value: w}\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_Binary_MapItemList(v.Memo)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 120, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tw, err = wire.NewValueBinary(v.VersionHistories), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 122, Value: w}\n\t\ti++\n\t}\n\tif v.VersionHistoriesEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.VersionHistoriesEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 124, Value: w}\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tw, err = wire.NewValueBinary(v.FirstExecutionRunID), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 126, Value: w}\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tw, err = wire.NewValueMap(_Map_String_String_MapItemList(v.PartitionConfig)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 128, Value: w}\n\t\ti++\n\t}\n\tif v.Checksum != nil {\n\t\tw, err = wire.NewValueBinary(v.Checksum), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 130, Value: w}\n\t\ti++\n\t}\n\tif v.ChecksumEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.ChecksumEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 132, Value: w}\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tw, err = v.CronOverlapPolicy.ToWire()\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 134, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tw, err = wire.NewValueBinary(v.ActiveClusterSelectionPolicy), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 137, Value: w}\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicyEncoding != nil {\n\t\tw, err = wire.NewValueString(*(v.ActiveClusterSelectionPolicyEncoding)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 138, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _Map_String_Binary_Read(m wire.MapItemList) (map[string][]byte, error) {\n\tif m.KeyType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\tif m.ValueType() != wire.TBinary {\n\t\treturn nil, nil\n\t}\n\n\to := make(map[string][]byte, m.Size())\n\terr := m.ForEach(func(x wire.MapItem) error {\n\t\tk, err := x.Key.GetString(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tv, err := x.Value.GetBinary(), error(nil)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\to[k] = v\n\t\treturn nil\n\t})\n\tm.Close()\n\treturn o, err\n}\n\nfunc _CronOverlapPolicy_Read(w wire.Value) (shared.CronOverlapPolicy, error) {\n\tvar v shared.CronOverlapPolicy\n\terr := v.FromWire(w)\n\treturn v, err\n}\n\n// FromWire deserializes a WorkflowExecutionInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowExecutionInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowExecutionInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowExecutionInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ParentDomainID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 12:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ParentWorkflowID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 14:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ParentRunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 16:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.InitiatedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 18:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.CompletionEventBatchID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 20:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.CompletionEvent, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 22:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CompletionEventEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 24:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.TaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 25:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x shared.TaskListKind\n\t\t\t\tx, err = _TaskListKind_Read(field.Value)\n\t\t\t\tv.TaskListKind = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 26:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.WorkflowTypeName = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 28:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.WorkflowTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 30:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DecisionTaskTimeoutSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 32:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ExecutionContext, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 34:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.State = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 36:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.CloseStatus = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 38:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 44:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastWriteEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 48:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastEventTaskID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 50:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastFirstEventID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 52:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastProcessedEvent = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 54:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StartTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 56:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.LastUpdatedTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 58:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 60:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionScheduleID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 62:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionStartedID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 64:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.DecisionTimeout = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 66:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 68:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionStartedTimestampNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 69:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionScheduledTimestampNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 70:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.CancelRequested = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 71:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.DecisionOriginalScheduledTimestampNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 72:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CreateRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 74:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.DecisionRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 76:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CancelRequestID = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 78:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.StickyTaskList = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 80:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.StickyScheduleToStartTimeout = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 82:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.RetryAttempt = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 84:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryInitialIntervalSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 86:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryMaximumIntervalSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 88:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryMaximumAttempts = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 90:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.RetryExpirationSeconds = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 92:\n\t\t\tif field.Value.Type() == wire.TDouble {\n\t\t\t\tvar x float64\n\t\t\t\tx, err = field.Value.GetDouble(), error(nil)\n\t\t\t\tv.RetryBackoffCoefficient = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 94:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.RetryExpirationTimeNanos = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 96:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.RetryNonRetryableErrors, err = _List_String_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 98:\n\t\t\tif field.Value.Type() == wire.TBool {\n\t\t\t\tvar x bool\n\t\t\t\tx, err = field.Value.GetBool(), error(nil)\n\t\t\t\tv.HasRetryPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 100:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.CronSchedule = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 102:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x int32\n\t\t\t\tx, err = field.Value.GetI32(), error(nil)\n\t\t\t\tv.EventStoreVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 104:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.EventBranchToken, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 106:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.SignalCount = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 108:\n\t\t\tif field.Value.Type() == wire.TI64 {\n\t\t\t\tvar x int64\n\t\t\t\tx, err = field.Value.GetI64(), error(nil)\n\t\t\t\tv.HistorySize = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 110:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientLibraryVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 112:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientFeatureVersion = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 114:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ClientImpl = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 115:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.AutoResetPoints, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 116:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.AutoResetPointsEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 118:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.SearchAttributes, err = _Map_String_Binary_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 120:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.Memo, err = _Map_String_Binary_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 122:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.VersionHistories, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 124:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.VersionHistoriesEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 126:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.FirstExecutionRunID, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 128:\n\t\t\tif field.Value.Type() == wire.TMap {\n\t\t\t\tv.PartitionConfig, err = _Map_String_String_Read(field.Value.GetMap())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 130:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.Checksum, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 132:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ChecksumEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 134:\n\t\t\tif field.Value.Type() == wire.TI32 {\n\t\t\t\tvar x shared.CronOverlapPolicy\n\t\t\t\tx, err = _CronOverlapPolicy_Read(field.Value)\n\t\t\t\tv.CronOverlapPolicy = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 137:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tv.ActiveClusterSelectionPolicy, err = field.Value.GetBinary(), error(nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\tcase 138:\n\t\t\tif field.Value.Type() == wire.TBinary {\n\t\t\t\tvar x string\n\t\t\t\tx, err = field.Value.GetString(), error(nil)\n\t\t\t\tv.ActiveClusterSelectionPolicyEncoding = &x\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _Map_String_Binary_Encode(val map[string][]byte, sw stream.Writer) error {\n\n\tmh := stream.MapHeader{\n\t\tKeyType:   wire.TBinary,\n\t\tValueType: wire.TBinary,\n\t\tLength:    len(val),\n\t}\n\tif err := sw.WriteMapBegin(mh); err != nil {\n\t\treturn err\n\t}\n\n\tfor k, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid map 'map[string][]byte', key [%v]: value is nil\", k)\n\t\t}\n\t\tif err := sw.WriteString(k); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteMapEnd()\n}\n\n// Encode serializes a WorkflowExecutionInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowExecutionInfo struct could not be encoded.\nfunc (v *WorkflowExecutionInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.ParentDomainID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ParentDomainID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentWorkflowID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 12, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ParentWorkflowID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ParentRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 14, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ParentRunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.InitiatedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 16, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.InitiatedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletionEventBatchID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 18, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.CompletionEventBatchID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletionEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 20, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.CompletionEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CompletionEventEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 22, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CompletionEventEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 24, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.TaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.TaskListKind != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 25, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.TaskListKind.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowTypeName != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 26, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.WorkflowTypeName)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.WorkflowTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 28, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.WorkflowTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTaskTimeoutSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 30, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DecisionTaskTimeoutSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ExecutionContext != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 32, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ExecutionContext); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.State != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 34, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.State)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CloseStatus != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 36, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.CloseStatus)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 38, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastWriteEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 44, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastWriteEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastEventTaskID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 48, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastEventTaskID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastFirstEventID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 50, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastFirstEventID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastProcessedEvent != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 52, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastProcessedEvent)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StartTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 54, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StartTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.LastUpdatedTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 56, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.LastUpdatedTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 58, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionScheduleID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 60, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionScheduleID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionStartedID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 62, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionStartedID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionTimeout != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 64, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.DecisionTimeout)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 66, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionStartedTimestampNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 68, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionStartedTimestampNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionScheduledTimestampNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 69, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionScheduledTimestampNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequested != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 70, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.CancelRequested)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionOriginalScheduledTimestampNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 71, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.DecisionOriginalScheduledTimestampNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CreateRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 72, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CreateRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.DecisionRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 74, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.DecisionRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CancelRequestID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 76, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CancelRequestID)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyTaskList != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 78, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.StickyTaskList)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.StickyScheduleToStartTimeout != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 80, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.StickyScheduleToStartTimeout)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryAttempt != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 82, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.RetryAttempt)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 84, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryInitialIntervalSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 86, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryMaximumIntervalSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryMaximumAttempts != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 88, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryMaximumAttempts)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryExpirationSeconds != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 90, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.RetryExpirationSeconds)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryBackoffCoefficient != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 92, Type: wire.TDouble}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteDouble(*(v.RetryBackoffCoefficient)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 94, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.RetryExpirationTimeNanos)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.RetryNonRetryableErrors != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 96, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_String_Encode(v.RetryNonRetryableErrors, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HasRetryPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 98, Type: wire.TBool}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBool(*(v.HasRetryPolicy)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronSchedule != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 100, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.CronSchedule)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventStoreVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 102, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt32(*(v.EventStoreVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.EventBranchToken != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 104, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.EventBranchToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SignalCount != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 106, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.SignalCount)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.HistorySize != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 108, Type: wire.TI64}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteInt64(*(v.HistorySize)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientLibraryVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 110, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientLibraryVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientFeatureVersion != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 112, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientFeatureVersion)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ClientImpl != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 114, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ClientImpl)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoResetPoints != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 115, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.AutoResetPoints); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.AutoResetPointsEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 116, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.AutoResetPointsEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.SearchAttributes != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 118, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Binary_Encode(v.SearchAttributes, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Memo != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 120, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_Binary_Encode(v.Memo, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistories != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 122, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.VersionHistories); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.VersionHistoriesEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 124, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.VersionHistoriesEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.FirstExecutionRunID != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 126, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.FirstExecutionRunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.PartitionConfig != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 128, Type: wire.TMap}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _Map_String_String_Encode(v.PartitionConfig, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.Checksum != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 130, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.Checksum); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ChecksumEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 132, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ChecksumEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.CronOverlapPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 134, Type: wire.TI32}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := v.CronOverlapPolicy.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 137, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteBinary(v.ActiveClusterSelectionPolicy); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif v.ActiveClusterSelectionPolicyEncoding != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 138, Type: wire.TBinary}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteString(*(v.ActiveClusterSelectionPolicyEncoding)); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _Map_String_Binary_Decode(sr stream.Reader) (map[string][]byte, error) {\n\tmh, err := sr.ReadMapBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif mh.KeyType != wire.TBinary || mh.ValueType != wire.TBinary {\n\t\tfor i := 0; i < mh.Length; i++ {\n\t\t\tif err := sr.Skip(mh.KeyType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := sr.Skip(mh.ValueType); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadMapEnd()\n\t}\n\n\to := make(map[string][]byte, mh.Length)\n\tfor i := 0; i < mh.Length; i++ {\n\t\tk, err := sr.ReadString()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tv, err := sr.ReadBinary()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\to[k] = v\n\t}\n\n\tif err = sr.ReadMapEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\nfunc _CronOverlapPolicy_Decode(sr stream.Reader) (shared.CronOverlapPolicy, error) {\n\tvar v shared.CronOverlapPolicy\n\terr := v.Decode(sr)\n\treturn v, err\n}\n\n// Decode deserializes a WorkflowExecutionInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowExecutionInfo struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowExecutionInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TBinary:\n\t\t\tv.ParentDomainID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 12 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ParentWorkflowID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 14 && fh.Type == wire.TBinary:\n\t\t\tv.ParentRunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 16 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.InitiatedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 18 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.CompletionEventBatchID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 20 && fh.Type == wire.TBinary:\n\t\t\tv.CompletionEvent, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 22 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CompletionEventEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 24 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.TaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 25 && fh.Type == wire.TI32:\n\t\t\tvar x shared.TaskListKind\n\t\t\tx, err = _TaskListKind_Decode(sr)\n\t\t\tv.TaskListKind = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 26 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.WorkflowTypeName = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 28 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.WorkflowTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 30 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DecisionTaskTimeoutSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 32 && fh.Type == wire.TBinary:\n\t\t\tv.ExecutionContext, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 34 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.State = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 36 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.CloseStatus = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 38 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 44 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastWriteEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 48 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastEventTaskID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 50 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastFirstEventID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 52 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastProcessedEvent = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 54 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StartTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 56 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.LastUpdatedTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 58 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 60 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionScheduleID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 62 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionStartedID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 64 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.DecisionTimeout = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 66 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 68 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionStartedTimestampNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 69 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionScheduledTimestampNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 70 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.CancelRequested = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 71 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.DecisionOriginalScheduledTimestampNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 72 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CreateRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 74 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.DecisionRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 76 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CancelRequestID = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 78 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.StickyTaskList = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 80 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.StickyScheduleToStartTimeout = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 82 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.RetryAttempt = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 84 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryInitialIntervalSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 86 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryMaximumIntervalSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 88 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryMaximumAttempts = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 90 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.RetryExpirationSeconds = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 92 && fh.Type == wire.TDouble:\n\t\t\tvar x float64\n\t\t\tx, err = sr.ReadDouble()\n\t\t\tv.RetryBackoffCoefficient = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 94 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.RetryExpirationTimeNanos = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 96 && fh.Type == wire.TList:\n\t\t\tv.RetryNonRetryableErrors, err = _List_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 98 && fh.Type == wire.TBool:\n\t\t\tvar x bool\n\t\t\tx, err = sr.ReadBool()\n\t\t\tv.HasRetryPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 100 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.CronSchedule = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 102 && fh.Type == wire.TI32:\n\t\t\tvar x int32\n\t\t\tx, err = sr.ReadInt32()\n\t\t\tv.EventStoreVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 104 && fh.Type == wire.TBinary:\n\t\t\tv.EventBranchToken, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 106 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.SignalCount = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 108 && fh.Type == wire.TI64:\n\t\t\tvar x int64\n\t\t\tx, err = sr.ReadInt64()\n\t\t\tv.HistorySize = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 110 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientLibraryVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 112 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientFeatureVersion = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 114 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ClientImpl = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 115 && fh.Type == wire.TBinary:\n\t\t\tv.AutoResetPoints, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 116 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.AutoResetPointsEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 118 && fh.Type == wire.TMap:\n\t\t\tv.SearchAttributes, err = _Map_String_Binary_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 120 && fh.Type == wire.TMap:\n\t\t\tv.Memo, err = _Map_String_Binary_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 122 && fh.Type == wire.TBinary:\n\t\t\tv.VersionHistories, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 124 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.VersionHistoriesEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 126 && fh.Type == wire.TBinary:\n\t\t\tv.FirstExecutionRunID, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 128 && fh.Type == wire.TMap:\n\t\t\tv.PartitionConfig, err = _Map_String_String_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 130 && fh.Type == wire.TBinary:\n\t\t\tv.Checksum, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 132 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ChecksumEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 134 && fh.Type == wire.TI32:\n\t\t\tvar x shared.CronOverlapPolicy\n\t\t\tx, err = _CronOverlapPolicy_Decode(sr)\n\t\t\tv.CronOverlapPolicy = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 137 && fh.Type == wire.TBinary:\n\t\t\tv.ActiveClusterSelectionPolicy, err = sr.ReadBinary()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tcase fh.ID == 138 && fh.Type == wire.TBinary:\n\t\t\tvar x string\n\t\t\tx, err = sr.ReadString()\n\t\t\tv.ActiveClusterSelectionPolicyEncoding = &x\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowExecutionInfo\n// struct.\nfunc (v *WorkflowExecutionInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [66]string\n\ti := 0\n\tif v.ParentDomainID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentDomainID: %v\", v.ParentDomainID)\n\t\ti++\n\t}\n\tif v.ParentWorkflowID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentWorkflowID: %v\", *(v.ParentWorkflowID))\n\t\ti++\n\t}\n\tif v.ParentRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"ParentRunID: %v\", v.ParentRunID)\n\t\ti++\n\t}\n\tif v.InitiatedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"InitiatedID: %v\", *(v.InitiatedID))\n\t\ti++\n\t}\n\tif v.CompletionEventBatchID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionEventBatchID: %v\", *(v.CompletionEventBatchID))\n\t\ti++\n\t}\n\tif v.CompletionEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionEvent: %v\", v.CompletionEvent)\n\t\ti++\n\t}\n\tif v.CompletionEventEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"CompletionEventEncoding: %v\", *(v.CompletionEventEncoding))\n\t\ti++\n\t}\n\tif v.TaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskList: %v\", *(v.TaskList))\n\t\ti++\n\t}\n\tif v.TaskListKind != nil {\n\t\tfields[i] = fmt.Sprintf(\"TaskListKind: %v\", *(v.TaskListKind))\n\t\ti++\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowTypeName: %v\", *(v.WorkflowTypeName))\n\t\ti++\n\t}\n\tif v.WorkflowTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"WorkflowTimeoutSeconds: %v\", *(v.WorkflowTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.DecisionTaskTimeoutSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTaskTimeoutSeconds: %v\", *(v.DecisionTaskTimeoutSeconds))\n\t\ti++\n\t}\n\tif v.ExecutionContext != nil {\n\t\tfields[i] = fmt.Sprintf(\"ExecutionContext: %v\", v.ExecutionContext)\n\t\ti++\n\t}\n\tif v.State != nil {\n\t\tfields[i] = fmt.Sprintf(\"State: %v\", *(v.State))\n\t\ti++\n\t}\n\tif v.CloseStatus != nil {\n\t\tfields[i] = fmt.Sprintf(\"CloseStatus: %v\", *(v.CloseStatus))\n\t\ti++\n\t}\n\tif v.StartVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartVersion: %v\", *(v.StartVersion))\n\t\ti++\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastWriteEventID: %v\", *(v.LastWriteEventID))\n\t\ti++\n\t}\n\tif v.LastEventTaskID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastEventTaskID: %v\", *(v.LastEventTaskID))\n\t\ti++\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastFirstEventID: %v\", *(v.LastFirstEventID))\n\t\ti++\n\t}\n\tif v.LastProcessedEvent != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastProcessedEvent: %v\", *(v.LastProcessedEvent))\n\t\ti++\n\t}\n\tif v.StartTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"StartTimeNanos: %v\", *(v.StartTimeNanos))\n\t\ti++\n\t}\n\tif v.LastUpdatedTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"LastUpdatedTimeNanos: %v\", *(v.LastUpdatedTimeNanos))\n\t\ti++\n\t}\n\tif v.DecisionVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionVersion: %v\", *(v.DecisionVersion))\n\t\ti++\n\t}\n\tif v.DecisionScheduleID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionScheduleID: %v\", *(v.DecisionScheduleID))\n\t\ti++\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionStartedID: %v\", *(v.DecisionStartedID))\n\t\ti++\n\t}\n\tif v.DecisionTimeout != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionTimeout: %v\", *(v.DecisionTimeout))\n\t\ti++\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionAttempt: %v\", *(v.DecisionAttempt))\n\t\ti++\n\t}\n\tif v.DecisionStartedTimestampNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionStartedTimestampNanos: %v\", *(v.DecisionStartedTimestampNanos))\n\t\ti++\n\t}\n\tif v.DecisionScheduledTimestampNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionScheduledTimestampNanos: %v\", *(v.DecisionScheduledTimestampNanos))\n\t\ti++\n\t}\n\tif v.CancelRequested != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequested: %v\", *(v.CancelRequested))\n\t\ti++\n\t}\n\tif v.DecisionOriginalScheduledTimestampNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionOriginalScheduledTimestampNanos: %v\", *(v.DecisionOriginalScheduledTimestampNanos))\n\t\ti++\n\t}\n\tif v.CreateRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CreateRequestID: %v\", *(v.CreateRequestID))\n\t\ti++\n\t}\n\tif v.DecisionRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"DecisionRequestID: %v\", *(v.DecisionRequestID))\n\t\ti++\n\t}\n\tif v.CancelRequestID != nil {\n\t\tfields[i] = fmt.Sprintf(\"CancelRequestID: %v\", *(v.CancelRequestID))\n\t\ti++\n\t}\n\tif v.StickyTaskList != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyTaskList: %v\", *(v.StickyTaskList))\n\t\ti++\n\t}\n\tif v.StickyScheduleToStartTimeout != nil {\n\t\tfields[i] = fmt.Sprintf(\"StickyScheduleToStartTimeout: %v\", *(v.StickyScheduleToStartTimeout))\n\t\ti++\n\t}\n\tif v.RetryAttempt != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryAttempt: %v\", *(v.RetryAttempt))\n\t\ti++\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryInitialIntervalSeconds: %v\", *(v.RetryInitialIntervalSeconds))\n\t\ti++\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryMaximumIntervalSeconds: %v\", *(v.RetryMaximumIntervalSeconds))\n\t\ti++\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryMaximumAttempts: %v\", *(v.RetryMaximumAttempts))\n\t\ti++\n\t}\n\tif v.RetryExpirationSeconds != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryExpirationSeconds: %v\", *(v.RetryExpirationSeconds))\n\t\ti++\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryBackoffCoefficient: %v\", *(v.RetryBackoffCoefficient))\n\t\ti++\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryExpirationTimeNanos: %v\", *(v.RetryExpirationTimeNanos))\n\t\ti++\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\tfields[i] = fmt.Sprintf(\"RetryNonRetryableErrors: %v\", v.RetryNonRetryableErrors)\n\t\ti++\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"HasRetryPolicy: %v\", *(v.HasRetryPolicy))\n\t\ti++\n\t}\n\tif v.CronSchedule != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronSchedule: %v\", *(v.CronSchedule))\n\t\ti++\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventStoreVersion: %v\", *(v.EventStoreVersion))\n\t\ti++\n\t}\n\tif v.EventBranchToken != nil {\n\t\tfields[i] = fmt.Sprintf(\"EventBranchToken: %v\", v.EventBranchToken)\n\t\ti++\n\t}\n\tif v.SignalCount != nil {\n\t\tfields[i] = fmt.Sprintf(\"SignalCount: %v\", *(v.SignalCount))\n\t\ti++\n\t}\n\tif v.HistorySize != nil {\n\t\tfields[i] = fmt.Sprintf(\"HistorySize: %v\", *(v.HistorySize))\n\t\ti++\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientLibraryVersion: %v\", *(v.ClientLibraryVersion))\n\t\ti++\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientFeatureVersion: %v\", *(v.ClientFeatureVersion))\n\t\ti++\n\t}\n\tif v.ClientImpl != nil {\n\t\tfields[i] = fmt.Sprintf(\"ClientImpl: %v\", *(v.ClientImpl))\n\t\ti++\n\t}\n\tif v.AutoResetPoints != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoResetPoints: %v\", v.AutoResetPoints)\n\t\ti++\n\t}\n\tif v.AutoResetPointsEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"AutoResetPointsEncoding: %v\", *(v.AutoResetPointsEncoding))\n\t\ti++\n\t}\n\tif v.SearchAttributes != nil {\n\t\tfields[i] = fmt.Sprintf(\"SearchAttributes: %v\", v.SearchAttributes)\n\t\ti++\n\t}\n\tif v.Memo != nil {\n\t\tfields[i] = fmt.Sprintf(\"Memo: %v\", v.Memo)\n\t\ti++\n\t}\n\tif v.VersionHistories != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistories: %v\", v.VersionHistories)\n\t\ti++\n\t}\n\tif v.VersionHistoriesEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"VersionHistoriesEncoding: %v\", *(v.VersionHistoriesEncoding))\n\t\ti++\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tfields[i] = fmt.Sprintf(\"FirstExecutionRunID: %v\", v.FirstExecutionRunID)\n\t\ti++\n\t}\n\tif v.PartitionConfig != nil {\n\t\tfields[i] = fmt.Sprintf(\"PartitionConfig: %v\", v.PartitionConfig)\n\t\ti++\n\t}\n\tif v.Checksum != nil {\n\t\tfields[i] = fmt.Sprintf(\"Checksum: %v\", v.Checksum)\n\t\ti++\n\t}\n\tif v.ChecksumEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"ChecksumEncoding: %v\", *(v.ChecksumEncoding))\n\t\ti++\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"CronOverlapPolicy: %v\", *(v.CronOverlapPolicy))\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicy: %v\", v.ActiveClusterSelectionPolicy)\n\t\ti++\n\t}\n\tif v.ActiveClusterSelectionPolicyEncoding != nil {\n\t\tfields[i] = fmt.Sprintf(\"ActiveClusterSelectionPolicyEncoding: %v\", *(v.ActiveClusterSelectionPolicyEncoding))\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowExecutionInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _Map_String_Binary_Equals(lhs, rhs map[string][]byte) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor lk, lv := range lhs {\n\t\trv, ok := rhs[lk]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\t\tif !bytes.Equal(lv, rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc _CronOverlapPolicy_EqualsPtr(lhs, rhs *shared.CronOverlapPolicy) bool {\n\tif lhs != nil && rhs != nil {\n\n\t\tx := *lhs\n\t\ty := *rhs\n\t\treturn x.Equals(y)\n\t}\n\treturn lhs == nil && rhs == nil\n}\n\n// Equals returns true if all the fields of this WorkflowExecutionInfo match the\n// provided WorkflowExecutionInfo.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowExecutionInfo) Equals(rhs *WorkflowExecutionInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.ParentDomainID == nil && rhs.ParentDomainID == nil) || (v.ParentDomainID != nil && rhs.ParentDomainID != nil && bytes.Equal(v.ParentDomainID, rhs.ParentDomainID))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ParentWorkflowID, rhs.ParentWorkflowID) {\n\t\treturn false\n\t}\n\tif !((v.ParentRunID == nil && rhs.ParentRunID == nil) || (v.ParentRunID != nil && rhs.ParentRunID != nil && bytes.Equal(v.ParentRunID, rhs.ParentRunID))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.InitiatedID, rhs.InitiatedID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.CompletionEventBatchID, rhs.CompletionEventBatchID) {\n\t\treturn false\n\t}\n\tif !((v.CompletionEvent == nil && rhs.CompletionEvent == nil) || (v.CompletionEvent != nil && rhs.CompletionEvent != nil && bytes.Equal(v.CompletionEvent, rhs.CompletionEvent))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CompletionEventEncoding, rhs.CompletionEventEncoding) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.TaskList, rhs.TaskList) {\n\t\treturn false\n\t}\n\tif !_TaskListKind_EqualsPtr(v.TaskListKind, rhs.TaskListKind) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.WorkflowTypeName, rhs.WorkflowTypeName) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.WorkflowTimeoutSeconds, rhs.WorkflowTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DecisionTaskTimeoutSeconds, rhs.DecisionTaskTimeoutSeconds) {\n\t\treturn false\n\t}\n\tif !((v.ExecutionContext == nil && rhs.ExecutionContext == nil) || (v.ExecutionContext != nil && rhs.ExecutionContext != nil && bytes.Equal(v.ExecutionContext, rhs.ExecutionContext))) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.State, rhs.State) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.CloseStatus, rhs.CloseStatus) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartVersion, rhs.StartVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastWriteEventID, rhs.LastWriteEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastEventTaskID, rhs.LastEventTaskID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastFirstEventID, rhs.LastFirstEventID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastProcessedEvent, rhs.LastProcessedEvent) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StartTimeNanos, rhs.StartTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.LastUpdatedTimeNanos, rhs.LastUpdatedTimeNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionVersion, rhs.DecisionVersion) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionScheduleID, rhs.DecisionScheduleID) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionStartedID, rhs.DecisionStartedID) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.DecisionTimeout, rhs.DecisionTimeout) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionAttempt, rhs.DecisionAttempt) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionStartedTimestampNanos, rhs.DecisionStartedTimestampNanos) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionScheduledTimestampNanos, rhs.DecisionScheduledTimestampNanos) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.CancelRequested, rhs.CancelRequested) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.DecisionOriginalScheduledTimestampNanos, rhs.DecisionOriginalScheduledTimestampNanos) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CreateRequestID, rhs.CreateRequestID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.DecisionRequestID, rhs.DecisionRequestID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CancelRequestID, rhs.CancelRequestID) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.StickyTaskList, rhs.StickyTaskList) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.StickyScheduleToStartTimeout, rhs.StickyScheduleToStartTimeout) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.RetryAttempt, rhs.RetryAttempt) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryInitialIntervalSeconds, rhs.RetryInitialIntervalSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryMaximumIntervalSeconds, rhs.RetryMaximumIntervalSeconds) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryMaximumAttempts, rhs.RetryMaximumAttempts) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.RetryExpirationSeconds, rhs.RetryExpirationSeconds) {\n\t\treturn false\n\t}\n\tif !_Double_EqualsPtr(v.RetryBackoffCoefficient, rhs.RetryBackoffCoefficient) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.RetryExpirationTimeNanos, rhs.RetryExpirationTimeNanos) {\n\t\treturn false\n\t}\n\tif !((v.RetryNonRetryableErrors == nil && rhs.RetryNonRetryableErrors == nil) || (v.RetryNonRetryableErrors != nil && rhs.RetryNonRetryableErrors != nil && _List_String_Equals(v.RetryNonRetryableErrors, rhs.RetryNonRetryableErrors))) {\n\t\treturn false\n\t}\n\tif !_Bool_EqualsPtr(v.HasRetryPolicy, rhs.HasRetryPolicy) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.CronSchedule, rhs.CronSchedule) {\n\t\treturn false\n\t}\n\tif !_I32_EqualsPtr(v.EventStoreVersion, rhs.EventStoreVersion) {\n\t\treturn false\n\t}\n\tif !((v.EventBranchToken == nil && rhs.EventBranchToken == nil) || (v.EventBranchToken != nil && rhs.EventBranchToken != nil && bytes.Equal(v.EventBranchToken, rhs.EventBranchToken))) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.SignalCount, rhs.SignalCount) {\n\t\treturn false\n\t}\n\tif !_I64_EqualsPtr(v.HistorySize, rhs.HistorySize) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientLibraryVersion, rhs.ClientLibraryVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientFeatureVersion, rhs.ClientFeatureVersion) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ClientImpl, rhs.ClientImpl) {\n\t\treturn false\n\t}\n\tif !((v.AutoResetPoints == nil && rhs.AutoResetPoints == nil) || (v.AutoResetPoints != nil && rhs.AutoResetPoints != nil && bytes.Equal(v.AutoResetPoints, rhs.AutoResetPoints))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.AutoResetPointsEncoding, rhs.AutoResetPointsEncoding) {\n\t\treturn false\n\t}\n\tif !((v.SearchAttributes == nil && rhs.SearchAttributes == nil) || (v.SearchAttributes != nil && rhs.SearchAttributes != nil && _Map_String_Binary_Equals(v.SearchAttributes, rhs.SearchAttributes))) {\n\t\treturn false\n\t}\n\tif !((v.Memo == nil && rhs.Memo == nil) || (v.Memo != nil && rhs.Memo != nil && _Map_String_Binary_Equals(v.Memo, rhs.Memo))) {\n\t\treturn false\n\t}\n\tif !((v.VersionHistories == nil && rhs.VersionHistories == nil) || (v.VersionHistories != nil && rhs.VersionHistories != nil && bytes.Equal(v.VersionHistories, rhs.VersionHistories))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.VersionHistoriesEncoding, rhs.VersionHistoriesEncoding) {\n\t\treturn false\n\t}\n\tif !((v.FirstExecutionRunID == nil && rhs.FirstExecutionRunID == nil) || (v.FirstExecutionRunID != nil && rhs.FirstExecutionRunID != nil && bytes.Equal(v.FirstExecutionRunID, rhs.FirstExecutionRunID))) {\n\t\treturn false\n\t}\n\tif !((v.PartitionConfig == nil && rhs.PartitionConfig == nil) || (v.PartitionConfig != nil && rhs.PartitionConfig != nil && _Map_String_String_Equals(v.PartitionConfig, rhs.PartitionConfig))) {\n\t\treturn false\n\t}\n\tif !((v.Checksum == nil && rhs.Checksum == nil) || (v.Checksum != nil && rhs.Checksum != nil && bytes.Equal(v.Checksum, rhs.Checksum))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ChecksumEncoding, rhs.ChecksumEncoding) {\n\t\treturn false\n\t}\n\tif !_CronOverlapPolicy_EqualsPtr(v.CronOverlapPolicy, rhs.CronOverlapPolicy) {\n\t\treturn false\n\t}\n\tif !((v.ActiveClusterSelectionPolicy == nil && rhs.ActiveClusterSelectionPolicy == nil) || (v.ActiveClusterSelectionPolicy != nil && rhs.ActiveClusterSelectionPolicy != nil && bytes.Equal(v.ActiveClusterSelectionPolicy, rhs.ActiveClusterSelectionPolicy))) {\n\t\treturn false\n\t}\n\tif !_String_EqualsPtr(v.ActiveClusterSelectionPolicyEncoding, rhs.ActiveClusterSelectionPolicyEncoding) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _Map_String_Binary_Zapper map[string][]byte\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of _Map_String_Binary_Zapper.\nfunc (m _Map_String_Binary_Zapper) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tfor k, v := range m {\n\t\tenc.AddString((string)(k), base64.StdEncoding.EncodeToString(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowExecutionInfo.\nfunc (v *WorkflowExecutionInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.ParentDomainID != nil {\n\t\tenc.AddString(\"parentDomainID\", base64.StdEncoding.EncodeToString(v.ParentDomainID))\n\t}\n\tif v.ParentWorkflowID != nil {\n\t\tenc.AddString(\"parentWorkflowID\", *v.ParentWorkflowID)\n\t}\n\tif v.ParentRunID != nil {\n\t\tenc.AddString(\"parentRunID\", base64.StdEncoding.EncodeToString(v.ParentRunID))\n\t}\n\tif v.InitiatedID != nil {\n\t\tenc.AddInt64(\"initiatedID\", *v.InitiatedID)\n\t}\n\tif v.CompletionEventBatchID != nil {\n\t\tenc.AddInt64(\"completionEventBatchID\", *v.CompletionEventBatchID)\n\t}\n\tif v.CompletionEvent != nil {\n\t\tenc.AddString(\"completionEvent\", base64.StdEncoding.EncodeToString(v.CompletionEvent))\n\t}\n\tif v.CompletionEventEncoding != nil {\n\t\tenc.AddString(\"completionEventEncoding\", *v.CompletionEventEncoding)\n\t}\n\tif v.TaskList != nil {\n\t\tenc.AddString(\"taskList\", *v.TaskList)\n\t}\n\tif v.TaskListKind != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"taskListKind\", *v.TaskListKind))\n\t}\n\tif v.WorkflowTypeName != nil {\n\t\tenc.AddString(\"workflowTypeName\", *v.WorkflowTypeName)\n\t}\n\tif v.WorkflowTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"workflowTimeoutSeconds\", *v.WorkflowTimeoutSeconds)\n\t}\n\tif v.DecisionTaskTimeoutSeconds != nil {\n\t\tenc.AddInt32(\"decisionTaskTimeoutSeconds\", *v.DecisionTaskTimeoutSeconds)\n\t}\n\tif v.ExecutionContext != nil {\n\t\tenc.AddString(\"executionContext\", base64.StdEncoding.EncodeToString(v.ExecutionContext))\n\t}\n\tif v.State != nil {\n\t\tenc.AddInt32(\"state\", *v.State)\n\t}\n\tif v.CloseStatus != nil {\n\t\tenc.AddInt32(\"closeStatus\", *v.CloseStatus)\n\t}\n\tif v.StartVersion != nil {\n\t\tenc.AddInt64(\"startVersion\", *v.StartVersion)\n\t}\n\tif v.LastWriteEventID != nil {\n\t\tenc.AddInt64(\"lastWriteEventID\", *v.LastWriteEventID)\n\t}\n\tif v.LastEventTaskID != nil {\n\t\tenc.AddInt64(\"lastEventTaskID\", *v.LastEventTaskID)\n\t}\n\tif v.LastFirstEventID != nil {\n\t\tenc.AddInt64(\"lastFirstEventID\", *v.LastFirstEventID)\n\t}\n\tif v.LastProcessedEvent != nil {\n\t\tenc.AddInt64(\"lastProcessedEvent\", *v.LastProcessedEvent)\n\t}\n\tif v.StartTimeNanos != nil {\n\t\tenc.AddInt64(\"startTimeNanos\", *v.StartTimeNanos)\n\t}\n\tif v.LastUpdatedTimeNanos != nil {\n\t\tenc.AddInt64(\"lastUpdatedTimeNanos\", *v.LastUpdatedTimeNanos)\n\t}\n\tif v.DecisionVersion != nil {\n\t\tenc.AddInt64(\"decisionVersion\", *v.DecisionVersion)\n\t}\n\tif v.DecisionScheduleID != nil {\n\t\tenc.AddInt64(\"decisionScheduleID\", *v.DecisionScheduleID)\n\t}\n\tif v.DecisionStartedID != nil {\n\t\tenc.AddInt64(\"decisionStartedID\", *v.DecisionStartedID)\n\t}\n\tif v.DecisionTimeout != nil {\n\t\tenc.AddInt32(\"decisionTimeout\", *v.DecisionTimeout)\n\t}\n\tif v.DecisionAttempt != nil {\n\t\tenc.AddInt64(\"decisionAttempt\", *v.DecisionAttempt)\n\t}\n\tif v.DecisionStartedTimestampNanos != nil {\n\t\tenc.AddInt64(\"decisionStartedTimestampNanos\", *v.DecisionStartedTimestampNanos)\n\t}\n\tif v.DecisionScheduledTimestampNanos != nil {\n\t\tenc.AddInt64(\"decisionScheduledTimestampNanos\", *v.DecisionScheduledTimestampNanos)\n\t}\n\tif v.CancelRequested != nil {\n\t\tenc.AddBool(\"cancelRequested\", *v.CancelRequested)\n\t}\n\tif v.DecisionOriginalScheduledTimestampNanos != nil {\n\t\tenc.AddInt64(\"decisionOriginalScheduledTimestampNanos\", *v.DecisionOriginalScheduledTimestampNanos)\n\t}\n\tif v.CreateRequestID != nil {\n\t\tenc.AddString(\"createRequestID\", *v.CreateRequestID)\n\t}\n\tif v.DecisionRequestID != nil {\n\t\tenc.AddString(\"decisionRequestID\", *v.DecisionRequestID)\n\t}\n\tif v.CancelRequestID != nil {\n\t\tenc.AddString(\"cancelRequestID\", *v.CancelRequestID)\n\t}\n\tif v.StickyTaskList != nil {\n\t\tenc.AddString(\"stickyTaskList\", *v.StickyTaskList)\n\t}\n\tif v.StickyScheduleToStartTimeout != nil {\n\t\tenc.AddInt64(\"stickyScheduleToStartTimeout\", *v.StickyScheduleToStartTimeout)\n\t}\n\tif v.RetryAttempt != nil {\n\t\tenc.AddInt64(\"retryAttempt\", *v.RetryAttempt)\n\t}\n\tif v.RetryInitialIntervalSeconds != nil {\n\t\tenc.AddInt32(\"retryInitialIntervalSeconds\", *v.RetryInitialIntervalSeconds)\n\t}\n\tif v.RetryMaximumIntervalSeconds != nil {\n\t\tenc.AddInt32(\"retryMaximumIntervalSeconds\", *v.RetryMaximumIntervalSeconds)\n\t}\n\tif v.RetryMaximumAttempts != nil {\n\t\tenc.AddInt32(\"retryMaximumAttempts\", *v.RetryMaximumAttempts)\n\t}\n\tif v.RetryExpirationSeconds != nil {\n\t\tenc.AddInt32(\"retryExpirationSeconds\", *v.RetryExpirationSeconds)\n\t}\n\tif v.RetryBackoffCoefficient != nil {\n\t\tenc.AddFloat64(\"retryBackoffCoefficient\", *v.RetryBackoffCoefficient)\n\t}\n\tif v.RetryExpirationTimeNanos != nil {\n\t\tenc.AddInt64(\"retryExpirationTimeNanos\", *v.RetryExpirationTimeNanos)\n\t}\n\tif v.RetryNonRetryableErrors != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"retryNonRetryableErrors\", (_List_String_Zapper)(v.RetryNonRetryableErrors)))\n\t}\n\tif v.HasRetryPolicy != nil {\n\t\tenc.AddBool(\"hasRetryPolicy\", *v.HasRetryPolicy)\n\t}\n\tif v.CronSchedule != nil {\n\t\tenc.AddString(\"cronSchedule\", *v.CronSchedule)\n\t}\n\tif v.EventStoreVersion != nil {\n\t\tenc.AddInt32(\"eventStoreVersion\", *v.EventStoreVersion)\n\t}\n\tif v.EventBranchToken != nil {\n\t\tenc.AddString(\"eventBranchToken\", base64.StdEncoding.EncodeToString(v.EventBranchToken))\n\t}\n\tif v.SignalCount != nil {\n\t\tenc.AddInt64(\"signalCount\", *v.SignalCount)\n\t}\n\tif v.HistorySize != nil {\n\t\tenc.AddInt64(\"historySize\", *v.HistorySize)\n\t}\n\tif v.ClientLibraryVersion != nil {\n\t\tenc.AddString(\"clientLibraryVersion\", *v.ClientLibraryVersion)\n\t}\n\tif v.ClientFeatureVersion != nil {\n\t\tenc.AddString(\"clientFeatureVersion\", *v.ClientFeatureVersion)\n\t}\n\tif v.ClientImpl != nil {\n\t\tenc.AddString(\"clientImpl\", *v.ClientImpl)\n\t}\n\tif v.AutoResetPoints != nil {\n\t\tenc.AddString(\"autoResetPoints\", base64.StdEncoding.EncodeToString(v.AutoResetPoints))\n\t}\n\tif v.AutoResetPointsEncoding != nil {\n\t\tenc.AddString(\"autoResetPointsEncoding\", *v.AutoResetPointsEncoding)\n\t}\n\tif v.SearchAttributes != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"searchAttributes\", (_Map_String_Binary_Zapper)(v.SearchAttributes)))\n\t}\n\tif v.Memo != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"memo\", (_Map_String_Binary_Zapper)(v.Memo)))\n\t}\n\tif v.VersionHistories != nil {\n\t\tenc.AddString(\"versionHistories\", base64.StdEncoding.EncodeToString(v.VersionHistories))\n\t}\n\tif v.VersionHistoriesEncoding != nil {\n\t\tenc.AddString(\"versionHistoriesEncoding\", *v.VersionHistoriesEncoding)\n\t}\n\tif v.FirstExecutionRunID != nil {\n\t\tenc.AddString(\"firstExecutionRunID\", base64.StdEncoding.EncodeToString(v.FirstExecutionRunID))\n\t}\n\tif v.PartitionConfig != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"partitionConfig\", (_Map_String_String_Zapper)(v.PartitionConfig)))\n\t}\n\tif v.Checksum != nil {\n\t\tenc.AddString(\"checksum\", base64.StdEncoding.EncodeToString(v.Checksum))\n\t}\n\tif v.ChecksumEncoding != nil {\n\t\tenc.AddString(\"checksumEncoding\", *v.ChecksumEncoding)\n\t}\n\tif v.CronOverlapPolicy != nil {\n\t\terr = multierr.Append(err, enc.AddObject(\"cronOverlapPolicy\", *v.CronOverlapPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicy != nil {\n\t\tenc.AddString(\"activeClusterSelectionPolicy\", base64.StdEncoding.EncodeToString(v.ActiveClusterSelectionPolicy))\n\t}\n\tif v.ActiveClusterSelectionPolicyEncoding != nil {\n\t\tenc.AddString(\"activeClusterSelectionPolicyEncoding\", *v.ActiveClusterSelectionPolicyEncoding)\n\t}\n\treturn err\n}\n\n// GetParentDomainID returns the value of ParentDomainID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentDomainID() (o []byte) {\n\tif v != nil && v.ParentDomainID != nil {\n\t\treturn v.ParentDomainID\n\t}\n\n\treturn\n}\n\n// IsSetParentDomainID returns true if ParentDomainID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentDomainID() bool {\n\treturn v != nil && v.ParentDomainID != nil\n}\n\n// GetParentWorkflowID returns the value of ParentWorkflowID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentWorkflowID() (o string) {\n\tif v != nil && v.ParentWorkflowID != nil {\n\t\treturn *v.ParentWorkflowID\n\t}\n\n\treturn\n}\n\n// IsSetParentWorkflowID returns true if ParentWorkflowID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentWorkflowID() bool {\n\treturn v != nil && v.ParentWorkflowID != nil\n}\n\n// GetParentRunID returns the value of ParentRunID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetParentRunID() (o []byte) {\n\tif v != nil && v.ParentRunID != nil {\n\t\treturn v.ParentRunID\n\t}\n\n\treturn\n}\n\n// IsSetParentRunID returns true if ParentRunID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetParentRunID() bool {\n\treturn v != nil && v.ParentRunID != nil\n}\n\n// GetInitiatedID returns the value of InitiatedID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetInitiatedID() (o int64) {\n\tif v != nil && v.InitiatedID != nil {\n\t\treturn *v.InitiatedID\n\t}\n\n\treturn\n}\n\n// IsSetInitiatedID returns true if InitiatedID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetInitiatedID() bool {\n\treturn v != nil && v.InitiatedID != nil\n}\n\n// GetCompletionEventBatchID returns the value of CompletionEventBatchID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCompletionEventBatchID() (o int64) {\n\tif v != nil && v.CompletionEventBatchID != nil {\n\t\treturn *v.CompletionEventBatchID\n\t}\n\n\treturn\n}\n\n// IsSetCompletionEventBatchID returns true if CompletionEventBatchID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCompletionEventBatchID() bool {\n\treturn v != nil && v.CompletionEventBatchID != nil\n}\n\n// GetCompletionEvent returns the value of CompletionEvent if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCompletionEvent() (o []byte) {\n\tif v != nil && v.CompletionEvent != nil {\n\t\treturn v.CompletionEvent\n\t}\n\n\treturn\n}\n\n// IsSetCompletionEvent returns true if CompletionEvent is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCompletionEvent() bool {\n\treturn v != nil && v.CompletionEvent != nil\n}\n\n// GetCompletionEventEncoding returns the value of CompletionEventEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCompletionEventEncoding() (o string) {\n\tif v != nil && v.CompletionEventEncoding != nil {\n\t\treturn *v.CompletionEventEncoding\n\t}\n\n\treturn\n}\n\n// IsSetCompletionEventEncoding returns true if CompletionEventEncoding is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCompletionEventEncoding() bool {\n\treturn v != nil && v.CompletionEventEncoding != nil\n}\n\n// GetTaskList returns the value of TaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetTaskList() (o string) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn *v.TaskList\n\t}\n\n\treturn\n}\n\n// IsSetTaskList returns true if TaskList is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetTaskList() bool {\n\treturn v != nil && v.TaskList != nil\n}\n\n// GetTaskListKind returns the value of TaskListKind if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetTaskListKind() (o shared.TaskListKind) {\n\tif v != nil && v.TaskListKind != nil {\n\t\treturn *v.TaskListKind\n\t}\n\n\treturn\n}\n\n// IsSetTaskListKind returns true if TaskListKind is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetTaskListKind() bool {\n\treturn v != nil && v.TaskListKind != nil\n}\n\n// GetWorkflowTypeName returns the value of WorkflowTypeName if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetWorkflowTypeName() (o string) {\n\tif v != nil && v.WorkflowTypeName != nil {\n\t\treturn *v.WorkflowTypeName\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowTypeName returns true if WorkflowTypeName is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetWorkflowTypeName() bool {\n\treturn v != nil && v.WorkflowTypeName != nil\n}\n\n// GetWorkflowTimeoutSeconds returns the value of WorkflowTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetWorkflowTimeoutSeconds() (o int32) {\n\tif v != nil && v.WorkflowTimeoutSeconds != nil {\n\t\treturn *v.WorkflowTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetWorkflowTimeoutSeconds returns true if WorkflowTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetWorkflowTimeoutSeconds() bool {\n\treturn v != nil && v.WorkflowTimeoutSeconds != nil\n}\n\n// GetDecisionTaskTimeoutSeconds returns the value of DecisionTaskTimeoutSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionTaskTimeoutSeconds() (o int32) {\n\tif v != nil && v.DecisionTaskTimeoutSeconds != nil {\n\t\treturn *v.DecisionTaskTimeoutSeconds\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTaskTimeoutSeconds returns true if DecisionTaskTimeoutSeconds is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionTaskTimeoutSeconds() bool {\n\treturn v != nil && v.DecisionTaskTimeoutSeconds != nil\n}\n\n// GetExecutionContext returns the value of ExecutionContext if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetExecutionContext() (o []byte) {\n\tif v != nil && v.ExecutionContext != nil {\n\t\treturn v.ExecutionContext\n\t}\n\n\treturn\n}\n\n// IsSetExecutionContext returns true if ExecutionContext is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetExecutionContext() bool {\n\treturn v != nil && v.ExecutionContext != nil\n}\n\n// GetState returns the value of State if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetState() (o int32) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\n\treturn\n}\n\n// IsSetState returns true if State is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetState() bool {\n\treturn v != nil && v.State != nil\n}\n\n// GetCloseStatus returns the value of CloseStatus if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCloseStatus() (o int32) {\n\tif v != nil && v.CloseStatus != nil {\n\t\treturn *v.CloseStatus\n\t}\n\n\treturn\n}\n\n// IsSetCloseStatus returns true if CloseStatus is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCloseStatus() bool {\n\treturn v != nil && v.CloseStatus != nil\n}\n\n// GetStartVersion returns the value of StartVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetStartVersion() (o int64) {\n\tif v != nil && v.StartVersion != nil {\n\t\treturn *v.StartVersion\n\t}\n\n\treturn\n}\n\n// IsSetStartVersion returns true if StartVersion is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetStartVersion() bool {\n\treturn v != nil && v.StartVersion != nil\n}\n\n// GetLastWriteEventID returns the value of LastWriteEventID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetLastWriteEventID() (o int64) {\n\tif v != nil && v.LastWriteEventID != nil {\n\t\treturn *v.LastWriteEventID\n\t}\n\n\treturn\n}\n\n// IsSetLastWriteEventID returns true if LastWriteEventID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetLastWriteEventID() bool {\n\treturn v != nil && v.LastWriteEventID != nil\n}\n\n// GetLastEventTaskID returns the value of LastEventTaskID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetLastEventTaskID() (o int64) {\n\tif v != nil && v.LastEventTaskID != nil {\n\t\treturn *v.LastEventTaskID\n\t}\n\n\treturn\n}\n\n// IsSetLastEventTaskID returns true if LastEventTaskID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetLastEventTaskID() bool {\n\treturn v != nil && v.LastEventTaskID != nil\n}\n\n// GetLastFirstEventID returns the value of LastFirstEventID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetLastFirstEventID() (o int64) {\n\tif v != nil && v.LastFirstEventID != nil {\n\t\treturn *v.LastFirstEventID\n\t}\n\n\treturn\n}\n\n// IsSetLastFirstEventID returns true if LastFirstEventID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetLastFirstEventID() bool {\n\treturn v != nil && v.LastFirstEventID != nil\n}\n\n// GetLastProcessedEvent returns the value of LastProcessedEvent if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetLastProcessedEvent() (o int64) {\n\tif v != nil && v.LastProcessedEvent != nil {\n\t\treturn *v.LastProcessedEvent\n\t}\n\n\treturn\n}\n\n// IsSetLastProcessedEvent returns true if LastProcessedEvent is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetLastProcessedEvent() bool {\n\treturn v != nil && v.LastProcessedEvent != nil\n}\n\n// GetStartTimeNanos returns the value of StartTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetStartTimeNanos() (o int64) {\n\tif v != nil && v.StartTimeNanos != nil {\n\t\treturn *v.StartTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetStartTimeNanos returns true if StartTimeNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetStartTimeNanos() bool {\n\treturn v != nil && v.StartTimeNanos != nil\n}\n\n// GetLastUpdatedTimeNanos returns the value of LastUpdatedTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetLastUpdatedTimeNanos() (o int64) {\n\tif v != nil && v.LastUpdatedTimeNanos != nil {\n\t\treturn *v.LastUpdatedTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetLastUpdatedTimeNanos returns true if LastUpdatedTimeNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetLastUpdatedTimeNanos() bool {\n\treturn v != nil && v.LastUpdatedTimeNanos != nil\n}\n\n// GetDecisionVersion returns the value of DecisionVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionVersion() (o int64) {\n\tif v != nil && v.DecisionVersion != nil {\n\t\treturn *v.DecisionVersion\n\t}\n\n\treturn\n}\n\n// IsSetDecisionVersion returns true if DecisionVersion is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionVersion() bool {\n\treturn v != nil && v.DecisionVersion != nil\n}\n\n// GetDecisionScheduleID returns the value of DecisionScheduleID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionScheduleID() (o int64) {\n\tif v != nil && v.DecisionScheduleID != nil {\n\t\treturn *v.DecisionScheduleID\n\t}\n\n\treturn\n}\n\n// IsSetDecisionScheduleID returns true if DecisionScheduleID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionScheduleID() bool {\n\treturn v != nil && v.DecisionScheduleID != nil\n}\n\n// GetDecisionStartedID returns the value of DecisionStartedID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionStartedID() (o int64) {\n\tif v != nil && v.DecisionStartedID != nil {\n\t\treturn *v.DecisionStartedID\n\t}\n\n\treturn\n}\n\n// IsSetDecisionStartedID returns true if DecisionStartedID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionStartedID() bool {\n\treturn v != nil && v.DecisionStartedID != nil\n}\n\n// GetDecisionTimeout returns the value of DecisionTimeout if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionTimeout() (o int32) {\n\tif v != nil && v.DecisionTimeout != nil {\n\t\treturn *v.DecisionTimeout\n\t}\n\n\treturn\n}\n\n// IsSetDecisionTimeout returns true if DecisionTimeout is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionTimeout() bool {\n\treturn v != nil && v.DecisionTimeout != nil\n}\n\n// GetDecisionAttempt returns the value of DecisionAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionAttempt() (o int64) {\n\tif v != nil && v.DecisionAttempt != nil {\n\t\treturn *v.DecisionAttempt\n\t}\n\n\treturn\n}\n\n// IsSetDecisionAttempt returns true if DecisionAttempt is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionAttempt() bool {\n\treturn v != nil && v.DecisionAttempt != nil\n}\n\n// GetDecisionStartedTimestampNanos returns the value of DecisionStartedTimestampNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionStartedTimestampNanos() (o int64) {\n\tif v != nil && v.DecisionStartedTimestampNanos != nil {\n\t\treturn *v.DecisionStartedTimestampNanos\n\t}\n\n\treturn\n}\n\n// IsSetDecisionStartedTimestampNanos returns true if DecisionStartedTimestampNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionStartedTimestampNanos() bool {\n\treturn v != nil && v.DecisionStartedTimestampNanos != nil\n}\n\n// GetDecisionScheduledTimestampNanos returns the value of DecisionScheduledTimestampNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionScheduledTimestampNanos() (o int64) {\n\tif v != nil && v.DecisionScheduledTimestampNanos != nil {\n\t\treturn *v.DecisionScheduledTimestampNanos\n\t}\n\n\treturn\n}\n\n// IsSetDecisionScheduledTimestampNanos returns true if DecisionScheduledTimestampNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionScheduledTimestampNanos() bool {\n\treturn v != nil && v.DecisionScheduledTimestampNanos != nil\n}\n\n// GetCancelRequested returns the value of CancelRequested if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCancelRequested() (o bool) {\n\tif v != nil && v.CancelRequested != nil {\n\t\treturn *v.CancelRequested\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequested returns true if CancelRequested is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCancelRequested() bool {\n\treturn v != nil && v.CancelRequested != nil\n}\n\n// GetDecisionOriginalScheduledTimestampNanos returns the value of DecisionOriginalScheduledTimestampNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionOriginalScheduledTimestampNanos() (o int64) {\n\tif v != nil && v.DecisionOriginalScheduledTimestampNanos != nil {\n\t\treturn *v.DecisionOriginalScheduledTimestampNanos\n\t}\n\n\treturn\n}\n\n// IsSetDecisionOriginalScheduledTimestampNanos returns true if DecisionOriginalScheduledTimestampNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionOriginalScheduledTimestampNanos() bool {\n\treturn v != nil && v.DecisionOriginalScheduledTimestampNanos != nil\n}\n\n// GetCreateRequestID returns the value of CreateRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCreateRequestID() (o string) {\n\tif v != nil && v.CreateRequestID != nil {\n\t\treturn *v.CreateRequestID\n\t}\n\n\treturn\n}\n\n// IsSetCreateRequestID returns true if CreateRequestID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCreateRequestID() bool {\n\treturn v != nil && v.CreateRequestID != nil\n}\n\n// GetDecisionRequestID returns the value of DecisionRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetDecisionRequestID() (o string) {\n\tif v != nil && v.DecisionRequestID != nil {\n\t\treturn *v.DecisionRequestID\n\t}\n\n\treturn\n}\n\n// IsSetDecisionRequestID returns true if DecisionRequestID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetDecisionRequestID() bool {\n\treturn v != nil && v.DecisionRequestID != nil\n}\n\n// GetCancelRequestID returns the value of CancelRequestID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCancelRequestID() (o string) {\n\tif v != nil && v.CancelRequestID != nil {\n\t\treturn *v.CancelRequestID\n\t}\n\n\treturn\n}\n\n// IsSetCancelRequestID returns true if CancelRequestID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCancelRequestID() bool {\n\treturn v != nil && v.CancelRequestID != nil\n}\n\n// GetStickyTaskList returns the value of StickyTaskList if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetStickyTaskList() (o string) {\n\tif v != nil && v.StickyTaskList != nil {\n\t\treturn *v.StickyTaskList\n\t}\n\n\treturn\n}\n\n// IsSetStickyTaskList returns true if StickyTaskList is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetStickyTaskList() bool {\n\treturn v != nil && v.StickyTaskList != nil\n}\n\n// GetStickyScheduleToStartTimeout returns the value of StickyScheduleToStartTimeout if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetStickyScheduleToStartTimeout() (o int64) {\n\tif v != nil && v.StickyScheduleToStartTimeout != nil {\n\t\treturn *v.StickyScheduleToStartTimeout\n\t}\n\n\treturn\n}\n\n// IsSetStickyScheduleToStartTimeout returns true if StickyScheduleToStartTimeout is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetStickyScheduleToStartTimeout() bool {\n\treturn v != nil && v.StickyScheduleToStartTimeout != nil\n}\n\n// GetRetryAttempt returns the value of RetryAttempt if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryAttempt() (o int64) {\n\tif v != nil && v.RetryAttempt != nil {\n\t\treturn *v.RetryAttempt\n\t}\n\n\treturn\n}\n\n// IsSetRetryAttempt returns true if RetryAttempt is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryAttempt() bool {\n\treturn v != nil && v.RetryAttempt != nil\n}\n\n// GetRetryInitialIntervalSeconds returns the value of RetryInitialIntervalSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryInitialIntervalSeconds() (o int32) {\n\tif v != nil && v.RetryInitialIntervalSeconds != nil {\n\t\treturn *v.RetryInitialIntervalSeconds\n\t}\n\n\treturn\n}\n\n// IsSetRetryInitialIntervalSeconds returns true if RetryInitialIntervalSeconds is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryInitialIntervalSeconds() bool {\n\treturn v != nil && v.RetryInitialIntervalSeconds != nil\n}\n\n// GetRetryMaximumIntervalSeconds returns the value of RetryMaximumIntervalSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryMaximumIntervalSeconds() (o int32) {\n\tif v != nil && v.RetryMaximumIntervalSeconds != nil {\n\t\treturn *v.RetryMaximumIntervalSeconds\n\t}\n\n\treturn\n}\n\n// IsSetRetryMaximumIntervalSeconds returns true if RetryMaximumIntervalSeconds is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryMaximumIntervalSeconds() bool {\n\treturn v != nil && v.RetryMaximumIntervalSeconds != nil\n}\n\n// GetRetryMaximumAttempts returns the value of RetryMaximumAttempts if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryMaximumAttempts() (o int32) {\n\tif v != nil && v.RetryMaximumAttempts != nil {\n\t\treturn *v.RetryMaximumAttempts\n\t}\n\n\treturn\n}\n\n// IsSetRetryMaximumAttempts returns true if RetryMaximumAttempts is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryMaximumAttempts() bool {\n\treturn v != nil && v.RetryMaximumAttempts != nil\n}\n\n// GetRetryExpirationSeconds returns the value of RetryExpirationSeconds if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryExpirationSeconds() (o int32) {\n\tif v != nil && v.RetryExpirationSeconds != nil {\n\t\treturn *v.RetryExpirationSeconds\n\t}\n\n\treturn\n}\n\n// IsSetRetryExpirationSeconds returns true if RetryExpirationSeconds is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryExpirationSeconds() bool {\n\treturn v != nil && v.RetryExpirationSeconds != nil\n}\n\n// GetRetryBackoffCoefficient returns the value of RetryBackoffCoefficient if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryBackoffCoefficient() (o float64) {\n\tif v != nil && v.RetryBackoffCoefficient != nil {\n\t\treturn *v.RetryBackoffCoefficient\n\t}\n\n\treturn\n}\n\n// IsSetRetryBackoffCoefficient returns true if RetryBackoffCoefficient is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryBackoffCoefficient() bool {\n\treturn v != nil && v.RetryBackoffCoefficient != nil\n}\n\n// GetRetryExpirationTimeNanos returns the value of RetryExpirationTimeNanos if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryExpirationTimeNanos() (o int64) {\n\tif v != nil && v.RetryExpirationTimeNanos != nil {\n\t\treturn *v.RetryExpirationTimeNanos\n\t}\n\n\treturn\n}\n\n// IsSetRetryExpirationTimeNanos returns true if RetryExpirationTimeNanos is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryExpirationTimeNanos() bool {\n\treturn v != nil && v.RetryExpirationTimeNanos != nil\n}\n\n// GetRetryNonRetryableErrors returns the value of RetryNonRetryableErrors if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetRetryNonRetryableErrors() (o []string) {\n\tif v != nil && v.RetryNonRetryableErrors != nil {\n\t\treturn v.RetryNonRetryableErrors\n\t}\n\n\treturn\n}\n\n// IsSetRetryNonRetryableErrors returns true if RetryNonRetryableErrors is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetRetryNonRetryableErrors() bool {\n\treturn v != nil && v.RetryNonRetryableErrors != nil\n}\n\n// GetHasRetryPolicy returns the value of HasRetryPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetHasRetryPolicy() (o bool) {\n\tif v != nil && v.HasRetryPolicy != nil {\n\t\treturn *v.HasRetryPolicy\n\t}\n\n\treturn\n}\n\n// IsSetHasRetryPolicy returns true if HasRetryPolicy is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetHasRetryPolicy() bool {\n\treturn v != nil && v.HasRetryPolicy != nil\n}\n\n// GetCronSchedule returns the value of CronSchedule if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\n\treturn\n}\n\n// IsSetCronSchedule returns true if CronSchedule is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCronSchedule() bool {\n\treturn v != nil && v.CronSchedule != nil\n}\n\n// GetEventStoreVersion returns the value of EventStoreVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetEventStoreVersion() (o int32) {\n\tif v != nil && v.EventStoreVersion != nil {\n\t\treturn *v.EventStoreVersion\n\t}\n\n\treturn\n}\n\n// IsSetEventStoreVersion returns true if EventStoreVersion is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetEventStoreVersion() bool {\n\treturn v != nil && v.EventStoreVersion != nil\n}\n\n// GetEventBranchToken returns the value of EventBranchToken if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetEventBranchToken() (o []byte) {\n\tif v != nil && v.EventBranchToken != nil {\n\t\treturn v.EventBranchToken\n\t}\n\n\treturn\n}\n\n// IsSetEventBranchToken returns true if EventBranchToken is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetEventBranchToken() bool {\n\treturn v != nil && v.EventBranchToken != nil\n}\n\n// GetSignalCount returns the value of SignalCount if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetSignalCount() (o int64) {\n\tif v != nil && v.SignalCount != nil {\n\t\treturn *v.SignalCount\n\t}\n\n\treturn\n}\n\n// IsSetSignalCount returns true if SignalCount is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetSignalCount() bool {\n\treturn v != nil && v.SignalCount != nil\n}\n\n// GetHistorySize returns the value of HistorySize if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetHistorySize() (o int64) {\n\tif v != nil && v.HistorySize != nil {\n\t\treturn *v.HistorySize\n\t}\n\n\treturn\n}\n\n// IsSetHistorySize returns true if HistorySize is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetHistorySize() bool {\n\treturn v != nil && v.HistorySize != nil\n}\n\n// GetClientLibraryVersion returns the value of ClientLibraryVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetClientLibraryVersion() (o string) {\n\tif v != nil && v.ClientLibraryVersion != nil {\n\t\treturn *v.ClientLibraryVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientLibraryVersion returns true if ClientLibraryVersion is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetClientLibraryVersion() bool {\n\treturn v != nil && v.ClientLibraryVersion != nil\n}\n\n// GetClientFeatureVersion returns the value of ClientFeatureVersion if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetClientFeatureVersion() (o string) {\n\tif v != nil && v.ClientFeatureVersion != nil {\n\t\treturn *v.ClientFeatureVersion\n\t}\n\n\treturn\n}\n\n// IsSetClientFeatureVersion returns true if ClientFeatureVersion is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetClientFeatureVersion() bool {\n\treturn v != nil && v.ClientFeatureVersion != nil\n}\n\n// GetClientImpl returns the value of ClientImpl if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetClientImpl() (o string) {\n\tif v != nil && v.ClientImpl != nil {\n\t\treturn *v.ClientImpl\n\t}\n\n\treturn\n}\n\n// IsSetClientImpl returns true if ClientImpl is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetClientImpl() bool {\n\treturn v != nil && v.ClientImpl != nil\n}\n\n// GetAutoResetPoints returns the value of AutoResetPoints if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetAutoResetPoints() (o []byte) {\n\tif v != nil && v.AutoResetPoints != nil {\n\t\treturn v.AutoResetPoints\n\t}\n\n\treturn\n}\n\n// IsSetAutoResetPoints returns true if AutoResetPoints is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetAutoResetPoints() bool {\n\treturn v != nil && v.AutoResetPoints != nil\n}\n\n// GetAutoResetPointsEncoding returns the value of AutoResetPointsEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetAutoResetPointsEncoding() (o string) {\n\tif v != nil && v.AutoResetPointsEncoding != nil {\n\t\treturn *v.AutoResetPointsEncoding\n\t}\n\n\treturn\n}\n\n// IsSetAutoResetPointsEncoding returns true if AutoResetPointsEncoding is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetAutoResetPointsEncoding() bool {\n\treturn v != nil && v.AutoResetPointsEncoding != nil\n}\n\n// GetSearchAttributes returns the value of SearchAttributes if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetSearchAttributes() (o map[string][]byte) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\n\treturn\n}\n\n// IsSetSearchAttributes returns true if SearchAttributes is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetSearchAttributes() bool {\n\treturn v != nil && v.SearchAttributes != nil\n}\n\n// GetMemo returns the value of Memo if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetMemo() (o map[string][]byte) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\n\treturn\n}\n\n// IsSetMemo returns true if Memo is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetMemo() bool {\n\treturn v != nil && v.Memo != nil\n}\n\n// GetVersionHistories returns the value of VersionHistories if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetVersionHistories() (o []byte) {\n\tif v != nil && v.VersionHistories != nil {\n\t\treturn v.VersionHistories\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistories returns true if VersionHistories is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetVersionHistories() bool {\n\treturn v != nil && v.VersionHistories != nil\n}\n\n// GetVersionHistoriesEncoding returns the value of VersionHistoriesEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetVersionHistoriesEncoding() (o string) {\n\tif v != nil && v.VersionHistoriesEncoding != nil {\n\t\treturn *v.VersionHistoriesEncoding\n\t}\n\n\treturn\n}\n\n// IsSetVersionHistoriesEncoding returns true if VersionHistoriesEncoding is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetVersionHistoriesEncoding() bool {\n\treturn v != nil && v.VersionHistoriesEncoding != nil\n}\n\n// GetFirstExecutionRunID returns the value of FirstExecutionRunID if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetFirstExecutionRunID() (o []byte) {\n\tif v != nil && v.FirstExecutionRunID != nil {\n\t\treturn v.FirstExecutionRunID\n\t}\n\n\treturn\n}\n\n// IsSetFirstExecutionRunID returns true if FirstExecutionRunID is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetFirstExecutionRunID() bool {\n\treturn v != nil && v.FirstExecutionRunID != nil\n}\n\n// GetPartitionConfig returns the value of PartitionConfig if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\n\treturn\n}\n\n// IsSetPartitionConfig returns true if PartitionConfig is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetPartitionConfig() bool {\n\treturn v != nil && v.PartitionConfig != nil\n}\n\n// GetChecksum returns the value of Checksum if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetChecksum() (o []byte) {\n\tif v != nil && v.Checksum != nil {\n\t\treturn v.Checksum\n\t}\n\n\treturn\n}\n\n// IsSetChecksum returns true if Checksum is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetChecksum() bool {\n\treturn v != nil && v.Checksum != nil\n}\n\n// GetChecksumEncoding returns the value of ChecksumEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetChecksumEncoding() (o string) {\n\tif v != nil && v.ChecksumEncoding != nil {\n\t\treturn *v.ChecksumEncoding\n\t}\n\n\treturn\n}\n\n// IsSetChecksumEncoding returns true if ChecksumEncoding is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetChecksumEncoding() bool {\n\treturn v != nil && v.ChecksumEncoding != nil\n}\n\n// GetCronOverlapPolicy returns the value of CronOverlapPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetCronOverlapPolicy() (o shared.CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\n\treturn\n}\n\n// IsSetCronOverlapPolicy returns true if CronOverlapPolicy is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetCronOverlapPolicy() bool {\n\treturn v != nil && v.CronOverlapPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicy returns the value of ActiveClusterSelectionPolicy if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetActiveClusterSelectionPolicy() (o []byte) {\n\tif v != nil && v.ActiveClusterSelectionPolicy != nil {\n\t\treturn v.ActiveClusterSelectionPolicy\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicy returns true if ActiveClusterSelectionPolicy is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetActiveClusterSelectionPolicy() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicy != nil\n}\n\n// GetActiveClusterSelectionPolicyEncoding returns the value of ActiveClusterSelectionPolicyEncoding if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowExecutionInfo) GetActiveClusterSelectionPolicyEncoding() (o string) {\n\tif v != nil && v.ActiveClusterSelectionPolicyEncoding != nil {\n\t\treturn *v.ActiveClusterSelectionPolicyEncoding\n\t}\n\n\treturn\n}\n\n// IsSetActiveClusterSelectionPolicyEncoding returns true if ActiveClusterSelectionPolicyEncoding is not nil.\nfunc (v *WorkflowExecutionInfo) IsSetActiveClusterSelectionPolicyEncoding() bool {\n\treturn v != nil && v.ActiveClusterSelectionPolicyEncoding != nil\n}\n\ntype WorkflowTimerTaskInfo struct {\n\tReferences []*TimerReference `json:\"references,omitempty\"`\n}\n\ntype _List_TimerReference_ValueList []*TimerReference\n\nfunc (v _List_TimerReference_ValueList) ForEach(f func(wire.Value) error) error {\n\tfor i, x := range v {\n\t\tif x == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*TimerReference', index [%v]: value is nil\", i)\n\t\t}\n\t\tw, err := x.ToWire()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = f(w)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v _List_TimerReference_ValueList) Size() int {\n\treturn len(v)\n}\n\nfunc (_List_TimerReference_ValueList) ValueType() wire.Type {\n\treturn wire.TStruct\n}\n\nfunc (_List_TimerReference_ValueList) Close() {}\n\n// ToWire translates a WorkflowTimerTaskInfo struct into a Thrift-level intermediate\n// representation. This intermediate representation may be serialized\n// into bytes using a ThriftRW protocol implementation.\n//\n// An error is returned if the struct or any of its fields failed to\n// validate.\n//\n//\tx, err := v.ToWire()\n//\tif err != nil {\n//\t  return err\n//\t}\n//\n//\tif err := binaryProtocol.Encode(x, writer); err != nil {\n//\t  return err\n//\t}\nfunc (v *WorkflowTimerTaskInfo) ToWire() (wire.Value, error) {\n\tvar (\n\t\tfields [1]wire.Field\n\t\ti      int = 0\n\t\tw      wire.Value\n\t\terr    error\n\t)\n\n\tif v.References != nil {\n\t\tw, err = wire.NewValueList(_List_TimerReference_ValueList(v.References)), error(nil)\n\t\tif err != nil {\n\t\t\treturn w, err\n\t\t}\n\t\tfields[i] = wire.Field{ID: 10, Value: w}\n\t\ti++\n\t}\n\n\treturn wire.NewValueStruct(wire.Struct{Fields: fields[:i]}), nil\n}\n\nfunc _TimerReference_Read(w wire.Value) (*TimerReference, error) {\n\tvar v TimerReference\n\terr := v.FromWire(w)\n\treturn &v, err\n}\n\nfunc _List_TimerReference_Read(l wire.ValueList) ([]*TimerReference, error) {\n\tif l.ValueType() != wire.TStruct {\n\t\treturn nil, nil\n\t}\n\n\to := make([]*TimerReference, 0, l.Size())\n\terr := l.ForEach(func(x wire.Value) error {\n\t\ti, err := _TimerReference_Read(x)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\to = append(o, i)\n\t\treturn nil\n\t})\n\tl.Close()\n\treturn o, err\n}\n\n// FromWire deserializes a WorkflowTimerTaskInfo struct from its Thrift-level\n// representation. The Thrift-level representation may be obtained\n// from a ThriftRW protocol implementation.\n//\n// An error is returned if we were unable to build a WorkflowTimerTaskInfo struct\n// from the provided intermediate representation.\n//\n//\tx, err := binaryProtocol.Decode(reader, wire.TStruct)\n//\tif err != nil {\n//\t  return nil, err\n//\t}\n//\n//\tvar v WorkflowTimerTaskInfo\n//\tif err := v.FromWire(x); err != nil {\n//\t  return nil, err\n//\t}\n//\treturn &v, nil\nfunc (v *WorkflowTimerTaskInfo) FromWire(w wire.Value) error {\n\tvar err error\n\n\tfor _, field := range w.GetStruct().Fields {\n\t\tswitch field.ID {\n\t\tcase 10:\n\t\t\tif field.Value.Type() == wire.TList {\n\t\t\t\tv.References, err = _List_TimerReference_Read(field.Value.GetList())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc _List_TimerReference_Encode(val []*TimerReference, sw stream.Writer) error {\n\n\tlh := stream.ListHeader{\n\t\tType:   wire.TStruct,\n\t\tLength: len(val),\n\t}\n\tif err := sw.WriteListBegin(lh); err != nil {\n\t\treturn err\n\t}\n\n\tfor i, v := range val {\n\t\tif v == nil {\n\t\t\treturn fmt.Errorf(\"invalid list '[]*TimerReference', index [%v]: value is nil\", i)\n\t\t}\n\t\tif err := v.Encode(sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn sw.WriteListEnd()\n}\n\n// Encode serializes a WorkflowTimerTaskInfo struct directly into bytes, without going\n// through an intermediary type.\n//\n// An error is returned if a WorkflowTimerTaskInfo struct could not be encoded.\nfunc (v *WorkflowTimerTaskInfo) Encode(sw stream.Writer) error {\n\tif err := sw.WriteStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tif v.References != nil {\n\t\tif err := sw.WriteFieldBegin(stream.FieldHeader{ID: 10, Type: wire.TList}); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := _List_TimerReference_Encode(v.References, sw); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := sw.WriteFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn sw.WriteStructEnd()\n}\n\nfunc _TimerReference_Decode(sr stream.Reader) (*TimerReference, error) {\n\tvar v TimerReference\n\terr := v.Decode(sr)\n\treturn &v, err\n}\n\nfunc _List_TimerReference_Decode(sr stream.Reader) ([]*TimerReference, error) {\n\tlh, err := sr.ReadListBegin()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif lh.Type != wire.TStruct {\n\t\tfor i := 0; i < lh.Length; i++ {\n\t\t\tif err := sr.Skip(lh.Type); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, sr.ReadListEnd()\n\t}\n\n\to := make([]*TimerReference, 0, lh.Length)\n\tfor i := 0; i < lh.Length; i++ {\n\t\tv, err := _TimerReference_Decode(sr)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\to = append(o, v)\n\t}\n\n\tif err = sr.ReadListEnd(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn o, err\n}\n\n// Decode deserializes a WorkflowTimerTaskInfo struct directly from its Thrift-level\n// representation, without going through an intemediary type.\n//\n// An error is returned if a WorkflowTimerTaskInfo struct could not be generated from the wire\n// representation.\nfunc (v *WorkflowTimerTaskInfo) Decode(sr stream.Reader) error {\n\n\tif err := sr.ReadStructBegin(); err != nil {\n\t\treturn err\n\t}\n\n\tfh, ok, err := sr.ReadFieldBegin()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor ok {\n\t\tswitch {\n\t\tcase fh.ID == 10 && fh.Type == wire.TList:\n\t\t\tv.References, err = _List_TimerReference_Decode(sr)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\tdefault:\n\t\t\tif err := sr.Skip(fh.Type); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tif err := sr.ReadFieldEnd(); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif fh, ok, err = sr.ReadFieldBegin(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := sr.ReadStructEnd(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// String returns a readable string representation of a WorkflowTimerTaskInfo\n// struct.\nfunc (v *WorkflowTimerTaskInfo) String() string {\n\tif v == nil {\n\t\treturn \"<nil>\"\n\t}\n\n\tvar fields [1]string\n\ti := 0\n\tif v.References != nil {\n\t\tfields[i] = fmt.Sprintf(\"References: %v\", v.References)\n\t\ti++\n\t}\n\n\treturn fmt.Sprintf(\"WorkflowTimerTaskInfo{%v}\", strings.Join(fields[:i], \", \"))\n}\n\nfunc _List_TimerReference_Equals(lhs, rhs []*TimerReference) bool {\n\tif len(lhs) != len(rhs) {\n\t\treturn false\n\t}\n\n\tfor i, lv := range lhs {\n\t\trv := rhs[i]\n\t\tif !lv.Equals(rv) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\n// Equals returns true if all the fields of this WorkflowTimerTaskInfo match the\n// provided WorkflowTimerTaskInfo.\n//\n// This function performs a deep comparison.\nfunc (v *WorkflowTimerTaskInfo) Equals(rhs *WorkflowTimerTaskInfo) bool {\n\tif v == nil {\n\t\treturn rhs == nil\n\t} else if rhs == nil {\n\t\treturn false\n\t}\n\tif !((v.References == nil && rhs.References == nil) || (v.References != nil && rhs.References != nil && _List_TimerReference_Equals(v.References, rhs.References))) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\ntype _List_TimerReference_Zapper []*TimerReference\n\n// MarshalLogArray implements zapcore.ArrayMarshaler, enabling\n// fast logging of _List_TimerReference_Zapper.\nfunc (l _List_TimerReference_Zapper) MarshalLogArray(enc zapcore.ArrayEncoder) (err error) {\n\tfor _, v := range l {\n\t\terr = multierr.Append(err, enc.AppendObject(v))\n\t}\n\treturn err\n}\n\n// MarshalLogObject implements zapcore.ObjectMarshaler, enabling\n// fast logging of WorkflowTimerTaskInfo.\nfunc (v *WorkflowTimerTaskInfo) MarshalLogObject(enc zapcore.ObjectEncoder) (err error) {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tif v.References != nil {\n\t\terr = multierr.Append(err, enc.AddArray(\"references\", (_List_TimerReference_Zapper)(v.References)))\n\t}\n\treturn err\n}\n\n// GetReferences returns the value of References if it is set or its\n// zero value if it is unset.\nfunc (v *WorkflowTimerTaskInfo) GetReferences() (o []*TimerReference) {\n\tif v != nil && v.References != nil {\n\t\treturn v.References\n\t}\n\n\treturn\n}\n\n// IsSetReferences returns true if References is not nil.\nfunc (v *WorkflowTimerTaskInfo) IsSetReferences() bool {\n\treturn v != nil && v.References != nil\n}\n\n// ThriftModule represents the IDL file used to generate this package.\nvar ThriftModule = &thriftreflect.ThriftModule{\n\tName:     \"sqlblobs\",\n\tPackage:  \"github.com/uber/cadence/.gen/go/sqlblobs\",\n\tFilePath: \"sqlblobs.thrift\",\n\tSHA1:     \"c688f6d76c90c9e402bf15d124f588c7dee2bee3\",\n\tIncludes: []*thriftreflect.ThriftModule{\n\t\tshared.ThriftModule,\n\t},\n\tRaw: rawIDL,\n}\n\nconst rawIDL = \"// Copyright (c) 2017 Uber Technologies, Inc.\\n//\\n// Permission is hereby granted, free of charge, to any person obtaining a copy\\n// of this software and associated documentation files (the \\\"Software\\\"), to deal\\n// in the Software without restriction, including without limitation the rights\\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\\n// copies of the Software, and to permit persons to whom the Software is\\n// furnished to do so, subject to the following conditions:\\n//\\n// The above copyright notice and this permission notice shall be included in\\n// all copies or substantial portions of the Software.\\n//\\n// THE SOFTWARE IS PROVIDED \\\"AS IS\\\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\\n// THE SOFTWARE.\\n\\nnamespace java com.uber.cadence.sqlblobs\\n\\ninclude \\\"shared.thrift\\\"\\n\\nstruct ShardInfo {\\n  10: optional i32 stolenSinceRenew\\n  12: optional i64 (js.type = \\\"Long\\\") updatedAtNanos\\n  14: optional i64 (js.type = \\\"Long\\\") replicationAckLevel\\n  16: optional i64 (js.type = \\\"Long\\\") transferAckLevel\\n  18: optional i64 (js.type = \\\"Long\\\") timerAckLevelNanos\\n  24: optional i64 (js.type = \\\"Long\\\") domainNotificationVersion\\n  34: optional map<string, i64> clusterTransferAckLevel\\n  36: optional map<string, i64> clusterTimerAckLevel\\n  38: optional string owner\\n  40: optional map<string, i64> clusterReplicationLevel\\n  42: optional binary pendingFailoverMarkers\\n  44: optional string pendingFailoverMarkersEncoding\\n  46: optional map<string, i64> replicationDlqAckLevel\\n  50: optional binary transferProcessingQueueStates\\n  51: optional string transferProcessingQueueStatesEncoding\\n  55: optional binary timerProcessingQueueStates\\n  56: optional string timerProcessingQueueStatesEncoding\\n  60: optional binary crossClusterProcessingQueueStates\\n  61: optional string crossClusterProcessingQueueStatesEncoding\\n  64: optional map<i32, shared.QueueState> queueStates\\n}\\n\\nstruct DomainInfo {\\n  10: optional string name\\n  12: optional string description\\n  14: optional string owner\\n  16: optional i32 status\\n  18: optional i16 retentionDays\\n  20: optional bool emitMetric\\n  22: optional string archivalBucket\\n  24: optional i16 archivalStatus\\n  26: optional i64 (js.type = \\\"Long\\\") configVersion\\n  28: optional i64 (js.type = \\\"Long\\\") notificationVersion\\n  30: optional i64 (js.type = \\\"Long\\\") failoverNotificationVersion\\n  32: optional i64 (js.type = \\\"Long\\\") failoverVersion\\n  34: optional string activeClusterName\\n  36: optional list<string> clusters\\n  38: optional map<string, string> data\\n  39: optional binary badBinaries\\n  40: optional string badBinariesEncoding\\n  42: optional i16 historyArchivalStatus\\n  44: optional string historyArchivalURI\\n  46: optional i16 visibilityArchivalStatus\\n  48: optional string visibilityArchivalURI\\n  50: optional i64 (js.type = \\\"Long\\\") failoverEndTime\\n  52: optional i64 (js.type = \\\"Long\\\") previousFailoverVersion\\n  54: optional i64 (js.type = \\\"Long\\\") lastUpdatedTime\\n  56: optional binary isolationGroupsConfiguration\\n  58: optional string isolationGroupsConfigurationEncoding\\n  60: optional binary asyncWorkflowConfiguration\\n  62: optional string asyncWorkflowConfigurationEncoding\\n  64: optional binary activeClustersConfiguration\\n  66: optional string activeClustersConfigurationEncoding\\n}\\n\\nstruct HistoryTreeInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") createdTimeNanos // For fork operation to prevent race condition of leaking event data when forking branches fail. Also can be used for clean up leaked data\\n  12: optional list<shared.HistoryBranchRange> ancestors\\n  14: optional string info // For lookup back to workflow during debugging, also background cleanup when fork operation cannot finish self cleanup due to crash.\\n}\\n\\nstruct WorkflowExecutionInfo {\\n  10: optional binary parentDomainID\\n  12: optional string parentWorkflowID\\n  14: optional binary parentRunID\\n  16: optional i64 (js.type = \\\"Long\\\") initiatedID\\n  18: optional i64 (js.type = \\\"Long\\\") completionEventBatchID\\n  20: optional binary completionEvent\\n  22: optional string completionEventEncoding\\n  24: optional string taskList\\n  25: optional shared.TaskListKind taskListKind\\n  26: optional string workflowTypeName\\n  28: optional i32 workflowTimeoutSeconds\\n  30: optional i32 decisionTaskTimeoutSeconds\\n  32: optional binary executionContext\\n  34: optional i32 state\\n  36: optional i32 closeStatus\\n  38: optional i64 (js.type = \\\"Long\\\") startVersion\\n  44: optional i64 (js.type = \\\"Long\\\") lastWriteEventID\\n  48: optional i64 (js.type = \\\"Long\\\") lastEventTaskID\\n  50: optional i64 (js.type = \\\"Long\\\") lastFirstEventID\\n  52: optional i64 (js.type = \\\"Long\\\") lastProcessedEvent\\n  54: optional i64 (js.type = \\\"Long\\\") startTimeNanos\\n  56: optional i64 (js.type = \\\"Long\\\") lastUpdatedTimeNanos\\n  58: optional i64 (js.type = \\\"Long\\\") decisionVersion\\n  60: optional i64 (js.type = \\\"Long\\\") decisionScheduleID\\n  62: optional i64 (js.type = \\\"Long\\\") decisionStartedID\\n  64: optional i32 decisionTimeout\\n  66: optional i64 (js.type = \\\"Long\\\") decisionAttempt\\n  68: optional i64 (js.type = \\\"Long\\\") decisionStartedTimestampNanos\\n  69: optional i64 (js.type = \\\"Long\\\") decisionScheduledTimestampNanos\\n  70: optional bool cancelRequested\\n  71: optional i64 (js.type = \\\"Long\\\") decisionOriginalScheduledTimestampNanos\\n  72: optional string createRequestID\\n  74: optional string decisionRequestID\\n  76: optional string cancelRequestID\\n  78: optional string stickyTaskList\\n  80: optional i64 (js.type = \\\"Long\\\") stickyScheduleToStartTimeout\\n  82: optional i64 (js.type = \\\"Long\\\") retryAttempt\\n  84: optional i32 retryInitialIntervalSeconds\\n  86: optional i32 retryMaximumIntervalSeconds\\n  88: optional i32 retryMaximumAttempts\\n  90: optional i32 retryExpirationSeconds\\n  92: optional double retryBackoffCoefficient\\n  94: optional i64 (js.type = \\\"Long\\\") retryExpirationTimeNanos\\n  96: optional list<string> retryNonRetryableErrors\\n  98: optional bool hasRetryPolicy\\n  100: optional string cronSchedule\\n  102: optional i32 eventStoreVersion\\n  104: optional binary eventBranchToken\\n  106: optional i64 (js.type = \\\"Long\\\") signalCount\\n  108: optional i64 (js.type = \\\"Long\\\") historySize\\n  110: optional string clientLibraryVersion\\n  112: optional string clientFeatureVersion\\n  114: optional string clientImpl\\n  115: optional binary autoResetPoints\\n  116: optional string autoResetPointsEncoding\\n  118: optional map<string, binary> searchAttributes\\n  120: optional map<string, binary> memo\\n  122: optional binary versionHistories\\n  124: optional string versionHistoriesEncoding\\n  126: optional binary firstExecutionRunID\\n  128: optional map<string, string> partitionConfig\\n  130: optional binary checksum\\n  132: optional string checksumEncoding\\n  134: optional shared.CronOverlapPolicy cronOverlapPolicy\\n  137: optional binary activeClusterSelectionPolicy\\n  138: optional string activeClusterSelectionPolicyEncoding\\n}\\n\\nstruct ActivityInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  12: optional i64 (js.type = \\\"Long\\\") scheduledEventBatchID\\n  14: optional binary scheduledEvent\\n  16: optional string scheduledEventEncoding\\n  18: optional i64 (js.type = \\\"Long\\\") scheduledTimeNanos\\n  20: optional i64 (js.type = \\\"Long\\\") startedID\\n  22: optional binary startedEvent\\n  24: optional string startedEventEncoding\\n  26: optional i64 (js.type = \\\"Long\\\") startedTimeNanos\\n  28: optional string activityID\\n  30: optional string requestID\\n  32: optional i32 scheduleToStartTimeoutSeconds\\n  34: optional i32 scheduleToCloseTimeoutSeconds\\n  36: optional i32 startToCloseTimeoutSeconds\\n  38: optional i32 heartbeatTimeoutSeconds\\n  40: optional bool cancelRequested\\n  42: optional i64 (js.type = \\\"Long\\\") cancelRequestID\\n  44: optional i32 timerTaskStatus\\n  46: optional i32 attempt\\n  48: optional string taskList\\n  49: optional shared.TaskListKind taskListKind\\n  50: optional string startedIdentity\\n  52: optional bool hasRetryPolicy\\n  54: optional i32 retryInitialIntervalSeconds\\n  56: optional i32 retryMaximumIntervalSeconds\\n  58: optional i32 retryMaximumAttempts\\n  60: optional i64 (js.type = \\\"Long\\\") retryExpirationTimeNanos\\n  62: optional double retryBackoffCoefficient\\n  64: optional list<string> retryNonRetryableErrors\\n  66: optional string retryLastFailureReason\\n  68: optional string retryLastWorkerIdentity\\n  70: optional binary retryLastFailureDetails\\n}\\n\\nstruct ChildExecutionInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  12: optional i64 (js.type = \\\"Long\\\") initiatedEventBatchID\\n  14: optional i64 (js.type = \\\"Long\\\") startedID\\n  16: optional binary initiatedEvent\\n  18: optional string initiatedEventEncoding\\n  20: optional string startedWorkflowID\\n  22: optional binary startedRunID\\n  24: optional binary startedEvent\\n  26: optional string startedEventEncoding\\n  28: optional string createRequestID\\n  29: optional string domainID\\n  30: optional string domainName // deprecated\\n  32: optional string workflowTypeName\\n  35: optional i32 parentClosePolicy\\n}\\n\\nstruct SignalInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  11: optional i64 (js.type = \\\"Long\\\") initiatedEventBatchID\\n  12: optional string requestID\\n  14: optional string name\\n  16: optional binary input\\n  18: optional binary control\\n}\\n\\nstruct RequestCancelInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  11: optional i64 (js.type = \\\"Long\\\") initiatedEventBatchID\\n  12: optional string cancelRequestID\\n}\\n\\nstruct TimerInfo {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  12: optional i64 (js.type = \\\"Long\\\") startedID\\n  14: optional i64 (js.type = \\\"Long\\\") expiryTimeNanos\\n  // TaskID is a misleading variable, it actually serves\\n  // the purpose of indicating whether a timer task is\\n  // generated for this timer info\\n  16: optional i64 (js.type = \\\"Long\\\") taskID\\n}\\n\\nstruct TaskInfo {\\n  10: optional string workflowID\\n  12: optional binary runID\\n  13: optional i64 (js.type = \\\"Long\\\") scheduleID\\n  14: optional i64 (js.type = \\\"Long\\\") expiryTimeNanos\\n  15: optional i64 (js.type = \\\"Long\\\") createdTimeNanos\\n  17: optional map<string, string> partitionConfig\\n}\\n\\nstruct TaskListPartition {\\n    10: optional list<string> isolationGroups\\n}\\n\\nstruct TaskListPartitionConfig {\\n  10: optional i64 (js.type = \\\"Long\\\") version\\n  12: optional i32 numReadPartitions\\n  14: optional i32 numWritePartitions\\n  16: optional map<i32, TaskListPartition> readPartitions\\n  18: optional map<i32, TaskListPartition> writePartitions\\n}\\n\\nstruct TaskListInfo {\\n  10: optional i16 kind // {Normal, Sticky}\\n  12: optional i64 (js.type = \\\"Long\\\") ackLevel\\n  14: optional i64 (js.type = \\\"Long\\\") expiryTimeNanos\\n  16: optional i64 (js.type = \\\"Long\\\") lastUpdatedNanos\\n  18: optional TaskListPartitionConfig adaptivePartitionConfig\\n}\\n\\nstruct TransferTaskInfo {\\n  10: optional binary domainID\\n  12: optional string workflowID\\n  14: optional binary runID\\n  16: optional i16 taskType\\n  18: optional binary targetDomainID\\n  20: optional string targetWorkflowID\\n  22: optional binary targetRunID\\n  24: optional string taskList\\n  26: optional bool targetChildWorkflowOnly\\n  28: optional i64 (js.type = \\\"Long\\\") scheduleID\\n  30: optional i64 (js.type = \\\"Long\\\") version\\n  32: optional i64 (js.type = \\\"Long\\\") visibilityTimestampNanos\\n  34: optional set<binary> targetDomainIDs\\n  36: optional string originalTaskList\\n  38: optional shared.TaskListKind originalTaskListKind\\n}\\n\\nstruct TimerTaskInfo {\\n  10: optional binary domainID\\n  12: optional string workflowID\\n  14: optional binary runID\\n  16: optional i16 taskType\\n  18: optional i16 timeoutType\\n  20: optional i64 (js.type = \\\"Long\\\") version\\n  22: optional i64 (js.type = \\\"Long\\\") scheduleAttempt\\n  24: optional i64 (js.type = \\\"Long\\\") eventID\\n  26: optional string taskList\\n}\\n\\nstruct ReplicationTaskInfo {\\n  10: optional binary domainID\\n  12: optional string workflowID\\n  14: optional binary runID\\n  16: optional i16 taskType\\n  18: optional i64 (js.type = \\\"Long\\\") version\\n  20: optional i64 (js.type = \\\"Long\\\") firstEventID\\n  22: optional i64 (js.type = \\\"Long\\\") nextEventID\\n  24: optional i64 (js.type = \\\"Long\\\") scheduledID\\n  26: optional i32 eventStoreVersion\\n  28: optional i32 newRunEventStoreVersion\\n  30: optional binary branch_token\\n  34: optional binary newRunBranchToken\\n  38: optional i64 (js.type = \\\"Long\\\") creationTime\\n}\\n\\nenum AsyncRequestType {\\n  StartWorkflowExecutionAsyncRequest\\n  SignalWithStartWorkflowExecutionAsyncRequest\\n}\\n\\nstruct AsyncRequestMessage {\\n  10: optional string partitionKey\\n  12: optional AsyncRequestType type\\n  14: optional shared.Header header\\n  16: optional string encoding\\n  18: optional binary payload\\n}\\n\\n// a substruct on the executions record which is intended to be used to track\\n// timers and other records for debugging and cleanup\\nstruct WorkflowTimerTaskInfo {\\n    10: optional list<TimerReference> references\\n}\\n\\nstruct TimerReference {\\n    // Primary Keys. Always required\\n    // a reference to the the execution table task_id\\n    10: optional i64 taskID\\n    // a reference to the execution table visibility_ts\\n    11: optional i64 (js.type = \\\"Long\\\") visibilityTimestamp\\n\\n    // Reference fields:\\n    // for workflow timer values, the type of timeout\\n    13: optional i16 TimeoutType\\n}\\n\"\n"
  },
  {
    "path": ".gen/go/sqlblobs/types_yarpc.go",
    "content": "// Code generated by thriftrw-plugin-yarpc\n// @generated\n\npackage sqlblobs\n"
  },
  {
    "path": ".gen/proto/history/v1/service.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/history/v1/service.proto\n\npackage historyv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n\ttypes \"github.com/gogo/protobuf/types\"\n\tv11 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\tv12 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype StartWorkflowExecutionRequest struct {\n\tRequest                  *v1.StartWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId                 string                            `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tParentExecutionInfo      *v1.ParentExecutionInfo           `protobuf:\"bytes,3,opt,name=parent_execution_info,json=parentExecutionInfo,proto3\" json:\"parent_execution_info,omitempty\"`\n\tAttempt                  int32                             `protobuf:\"varint,4,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tExpirationTime           *types.Timestamp                  `protobuf:\"bytes,5,opt,name=expiration_time,json=expirationTime,proto3\" json:\"expiration_time,omitempty\"`\n\tContinueAsNewInitiator   v1.ContinueAsNewInitiator         `protobuf:\"varint,6,opt,name=continue_as_new_initiator,json=continueAsNewInitiator,proto3,enum=uber.cadence.api.v1.ContinueAsNewInitiator\" json:\"continue_as_new_initiator,omitempty\"`\n\tContinuedFailure         *v1.Failure                       `protobuf:\"bytes,7,opt,name=continued_failure,json=continuedFailure,proto3\" json:\"continued_failure,omitempty\"`\n\tLastCompletionResult     *v1.Payload                       `protobuf:\"bytes,8,opt,name=last_completion_result,json=lastCompletionResult,proto3\" json:\"last_completion_result,omitempty\"`\n\tFirstDecisionTaskBackoff *types.Duration                   `protobuf:\"bytes,9,opt,name=first_decision_task_backoff,json=firstDecisionTaskBackoff,proto3\" json:\"first_decision_task_backoff,omitempty\"`\n\tPartitionConfig          map[string]string                 `protobuf:\"bytes,10,rep,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral     struct{}                          `json:\"-\"`\n\tXXX_unrecognized         []byte                            `json:\"-\"`\n\tXXX_sizecache            int32                             `json:\"-\"`\n}\n\nfunc (m *StartWorkflowExecutionRequest) Reset()         { *m = StartWorkflowExecutionRequest{} }\nfunc (m *StartWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*StartWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*StartWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{0}\n}\nfunc (m *StartWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *StartWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_StartWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *StartWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_StartWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *StartWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *StartWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_StartWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_StartWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *StartWorkflowExecutionRequest) GetRequest() *v1.StartWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetParentExecutionInfo() *v1.ParentExecutionInfo {\n\tif m != nil {\n\t\treturn m.ParentExecutionInfo\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetExpirationTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ExpirationTime\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetContinueAsNewInitiator() v1.ContinueAsNewInitiator {\n\tif m != nil {\n\t\treturn m.ContinueAsNewInitiator\n\t}\n\treturn v1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetContinuedFailure() *v1.Failure {\n\tif m != nil {\n\t\treturn m.ContinuedFailure\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetLastCompletionResult() *v1.Payload {\n\tif m != nil {\n\t\treturn m.LastCompletionResult\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetFirstDecisionTaskBackoff() *types.Duration {\n\tif m != nil {\n\t\treturn m.FirstDecisionTaskBackoff\n\t}\n\treturn nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) GetPartitionConfig() map[string]string {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype StartWorkflowExecutionResponse struct {\n\tRunId                string   `protobuf:\"bytes,1,opt,name=run_id,json=runId,proto3\" json:\"run_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *StartWorkflowExecutionResponse) Reset()         { *m = StartWorkflowExecutionResponse{} }\nfunc (m *StartWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*StartWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*StartWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{1}\n}\nfunc (m *StartWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *StartWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_StartWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *StartWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_StartWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *StartWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *StartWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_StartWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_StartWorkflowExecutionResponse proto.InternalMessageInfo\n\nfunc (m *StartWorkflowExecutionResponse) GetRunId() string {\n\tif m != nil {\n\t\treturn m.RunId\n\t}\n\treturn \"\"\n}\n\ntype SignalWorkflowExecutionRequest struct {\n\tRequest  *v1.SignalWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId string                             `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\t// workflow execution that requests this signal, for making sure\n\t// the workflow being signaled is actually a child of the workflow\n\t// making the request\n\tExternalWorkflowExecution *v1.WorkflowExecution `protobuf:\"bytes,3,opt,name=external_workflow_execution,json=externalWorkflowExecution,proto3\" json:\"external_workflow_execution,omitempty\"`\n\tChildWorkflowOnly         bool                  `protobuf:\"varint,4,opt,name=child_workflow_only,json=childWorkflowOnly,proto3\" json:\"child_workflow_only,omitempty\"`\n\tXXX_NoUnkeyedLiteral      struct{}              `json:\"-\"`\n\tXXX_unrecognized          []byte                `json:\"-\"`\n\tXXX_sizecache             int32                 `json:\"-\"`\n}\n\nfunc (m *SignalWorkflowExecutionRequest) Reset()         { *m = SignalWorkflowExecutionRequest{} }\nfunc (m *SignalWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*SignalWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*SignalWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{2}\n}\nfunc (m *SignalWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SignalWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SignalWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SignalWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SignalWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *SignalWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SignalWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_SignalWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SignalWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *SignalWorkflowExecutionRequest) GetRequest() *v1.SignalWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *SignalWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *SignalWorkflowExecutionRequest) GetExternalWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.ExternalWorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *SignalWorkflowExecutionRequest) GetChildWorkflowOnly() bool {\n\tif m != nil {\n\t\treturn m.ChildWorkflowOnly\n\t}\n\treturn false\n}\n\ntype SignalWorkflowExecutionResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *SignalWorkflowExecutionResponse) Reset()         { *m = SignalWorkflowExecutionResponse{} }\nfunc (m *SignalWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*SignalWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*SignalWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{3}\n}\nfunc (m *SignalWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SignalWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SignalWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SignalWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SignalWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *SignalWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SignalWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_SignalWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SignalWorkflowExecutionResponse proto.InternalMessageInfo\n\ntype SignalWithStartWorkflowExecutionRequest struct {\n\tRequest              *v1.SignalWithStartWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                                      `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tPartitionConfig      map[string]string                           `protobuf:\"bytes,3,rep,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}                                    `json:\"-\"`\n\tXXX_unrecognized     []byte                                      `json:\"-\"`\n\tXXX_sizecache        int32                                       `json:\"-\"`\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) Reset() {\n\t*m = SignalWithStartWorkflowExecutionRequest{}\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*SignalWithStartWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*SignalWithStartWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{4}\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SignalWithStartWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SignalWithStartWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_SignalWithStartWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SignalWithStartWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) GetRequest() *v1.SignalWithStartWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) GetPartitionConfig() map[string]string {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype SignalWithStartWorkflowExecutionResponse struct {\n\tRunId                string   `protobuf:\"bytes,1,opt,name=run_id,json=runId,proto3\" json:\"run_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) Reset() {\n\t*m = SignalWithStartWorkflowExecutionResponse{}\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*SignalWithStartWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*SignalWithStartWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{5}\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SignalWithStartWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SignalWithStartWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_SignalWithStartWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SignalWithStartWorkflowExecutionResponse proto.InternalMessageInfo\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) GetRunId() string {\n\tif m != nil {\n\t\treturn m.RunId\n\t}\n\treturn \"\"\n}\n\ntype ResetWorkflowExecutionRequest struct {\n\tRequest              *v1.ResetWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                            `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                          `json:\"-\"`\n\tXXX_unrecognized     []byte                            `json:\"-\"`\n\tXXX_sizecache        int32                             `json:\"-\"`\n}\n\nfunc (m *ResetWorkflowExecutionRequest) Reset()         { *m = ResetWorkflowExecutionRequest{} }\nfunc (m *ResetWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ResetWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*ResetWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{6}\n}\nfunc (m *ResetWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *ResetWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *ResetWorkflowExecutionRequest) GetRequest() *v1.ResetWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *ResetWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype ResetWorkflowExecutionResponse struct {\n\tRunId                string   `protobuf:\"bytes,1,opt,name=run_id,json=runId,proto3\" json:\"run_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ResetWorkflowExecutionResponse) Reset()         { *m = ResetWorkflowExecutionResponse{} }\nfunc (m *ResetWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ResetWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*ResetWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{7}\n}\nfunc (m *ResetWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *ResetWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetWorkflowExecutionResponse proto.InternalMessageInfo\n\nfunc (m *ResetWorkflowExecutionResponse) GetRunId() string {\n\tif m != nil {\n\t\treturn m.RunId\n\t}\n\treturn \"\"\n}\n\ntype TerminateWorkflowExecutionRequest struct {\n\tRequest  *v1.TerminateWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId string                                `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\t// workflow execution that requests this termination, for making sure\n\t// the workflow being terminated is actually a child of the workflow\n\t// making the request\n\tExternalWorkflowExecution *v1.WorkflowExecution `protobuf:\"bytes,3,opt,name=external_workflow_execution,json=externalWorkflowExecution,proto3\" json:\"external_workflow_execution,omitempty\"`\n\tChildWorkflowOnly         bool                  `protobuf:\"varint,4,opt,name=child_workflow_only,json=childWorkflowOnly,proto3\" json:\"child_workflow_only,omitempty\"`\n\tXXX_NoUnkeyedLiteral      struct{}              `json:\"-\"`\n\tXXX_unrecognized          []byte                `json:\"-\"`\n\tXXX_sizecache             int32                 `json:\"-\"`\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) Reset()         { *m = TerminateWorkflowExecutionRequest{} }\nfunc (m *TerminateWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*TerminateWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*TerminateWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{8}\n}\nfunc (m *TerminateWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TerminateWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TerminateWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TerminateWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TerminateWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *TerminateWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TerminateWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_TerminateWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TerminateWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *TerminateWorkflowExecutionRequest) GetRequest() *v1.TerminateWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) GetExternalWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.ExternalWorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) GetChildWorkflowOnly() bool {\n\tif m != nil {\n\t\treturn m.ChildWorkflowOnly\n\t}\n\treturn false\n}\n\ntype TerminateWorkflowExecutionResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *TerminateWorkflowExecutionResponse) Reset()         { *m = TerminateWorkflowExecutionResponse{} }\nfunc (m *TerminateWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*TerminateWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*TerminateWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{9}\n}\nfunc (m *TerminateWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TerminateWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TerminateWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TerminateWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TerminateWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *TerminateWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TerminateWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_TerminateWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TerminateWorkflowExecutionResponse proto.InternalMessageInfo\n\ntype DescribeWorkflowExecutionRequest struct {\n\tRequest              *v1.DescribeWorkflowExecutionRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                               `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                             `json:\"-\"`\n\tXXX_unrecognized     []byte                               `json:\"-\"`\n\tXXX_sizecache        int32                                `json:\"-\"`\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) Reset()         { *m = DescribeWorkflowExecutionRequest{} }\nfunc (m *DescribeWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*DescribeWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{10}\n}\nfunc (m *DescribeWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *DescribeWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *DescribeWorkflowExecutionRequest) GetRequest() *v1.DescribeWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype DescribeWorkflowExecutionResponse struct {\n\tExecutionConfiguration *v1.WorkflowExecutionConfiguration `protobuf:\"bytes,1,opt,name=execution_configuration,json=executionConfiguration,proto3\" json:\"execution_configuration,omitempty\"`\n\tWorkflowExecutionInfo  *v1.WorkflowExecutionInfo          `protobuf:\"bytes,2,opt,name=workflow_execution_info,json=workflowExecutionInfo,proto3\" json:\"workflow_execution_info,omitempty\"`\n\tPendingActivities      []*v1.PendingActivityInfo          `protobuf:\"bytes,3,rep,name=pending_activities,json=pendingActivities,proto3\" json:\"pending_activities,omitempty\"`\n\tPendingChildren        []*v1.PendingChildExecutionInfo    `protobuf:\"bytes,4,rep,name=pending_children,json=pendingChildren,proto3\" json:\"pending_children,omitempty\"`\n\tPendingDecision        *v1.PendingDecisionInfo            `protobuf:\"bytes,5,opt,name=pending_decision,json=pendingDecision,proto3\" json:\"pending_decision,omitempty\"`\n\tXXX_NoUnkeyedLiteral   struct{}                           `json:\"-\"`\n\tXXX_unrecognized       []byte                             `json:\"-\"`\n\tXXX_sizecache          int32                              `json:\"-\"`\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) Reset()         { *m = DescribeWorkflowExecutionResponse{} }\nfunc (m *DescribeWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*DescribeWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{11}\n}\nfunc (m *DescribeWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *DescribeWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeWorkflowExecutionResponse proto.InternalMessageInfo\n\nfunc (m *DescribeWorkflowExecutionResponse) GetExecutionConfiguration() *v1.WorkflowExecutionConfiguration {\n\tif m != nil {\n\t\treturn m.ExecutionConfiguration\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) GetWorkflowExecutionInfo() *v1.WorkflowExecutionInfo {\n\tif m != nil {\n\t\treturn m.WorkflowExecutionInfo\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) GetPendingActivities() []*v1.PendingActivityInfo {\n\tif m != nil {\n\t\treturn m.PendingActivities\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) GetPendingChildren() []*v1.PendingChildExecutionInfo {\n\tif m != nil {\n\t\treturn m.PendingChildren\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) GetPendingDecision() *v1.PendingDecisionInfo {\n\tif m != nil {\n\t\treturn m.PendingDecision\n\t}\n\treturn nil\n}\n\ntype QueryWorkflowRequest struct {\n\tRequest              *v1.QueryWorkflowRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                   `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                 `json:\"-\"`\n\tXXX_unrecognized     []byte                   `json:\"-\"`\n\tXXX_sizecache        int32                    `json:\"-\"`\n}\n\nfunc (m *QueryWorkflowRequest) Reset()         { *m = QueryWorkflowRequest{} }\nfunc (m *QueryWorkflowRequest) String() string { return proto.CompactTextString(m) }\nfunc (*QueryWorkflowRequest) ProtoMessage()    {}\nfunc (*QueryWorkflowRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{12}\n}\nfunc (m *QueryWorkflowRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *QueryWorkflowRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_QueryWorkflowRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *QueryWorkflowRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_QueryWorkflowRequest.Merge(m, src)\n}\nfunc (m *QueryWorkflowRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *QueryWorkflowRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_QueryWorkflowRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_QueryWorkflowRequest proto.InternalMessageInfo\n\nfunc (m *QueryWorkflowRequest) GetRequest() *v1.QueryWorkflowRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype QueryWorkflowResponse struct {\n\tQueryResult          *v1.Payload       `protobuf:\"bytes,1,opt,name=query_result,json=queryResult,proto3\" json:\"query_result,omitempty\"`\n\tQueryRejected        *v1.QueryRejected `protobuf:\"bytes,2,opt,name=query_rejected,json=queryRejected,proto3\" json:\"query_rejected,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}          `json:\"-\"`\n\tXXX_unrecognized     []byte            `json:\"-\"`\n\tXXX_sizecache        int32             `json:\"-\"`\n}\n\nfunc (m *QueryWorkflowResponse) Reset()         { *m = QueryWorkflowResponse{} }\nfunc (m *QueryWorkflowResponse) String() string { return proto.CompactTextString(m) }\nfunc (*QueryWorkflowResponse) ProtoMessage()    {}\nfunc (*QueryWorkflowResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{13}\n}\nfunc (m *QueryWorkflowResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *QueryWorkflowResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_QueryWorkflowResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *QueryWorkflowResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_QueryWorkflowResponse.Merge(m, src)\n}\nfunc (m *QueryWorkflowResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *QueryWorkflowResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_QueryWorkflowResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_QueryWorkflowResponse proto.InternalMessageInfo\n\nfunc (m *QueryWorkflowResponse) GetQueryResult() *v1.Payload {\n\tif m != nil {\n\t\treturn m.QueryResult\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowResponse) GetQueryRejected() *v1.QueryRejected {\n\tif m != nil {\n\t\treturn m.QueryRejected\n\t}\n\treturn nil\n}\n\ntype ResetStickyTaskListRequest struct {\n\tRequest              *v1.ResetStickyTaskListRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                         `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                       `json:\"-\"`\n\tXXX_unrecognized     []byte                         `json:\"-\"`\n\tXXX_sizecache        int32                          `json:\"-\"`\n}\n\nfunc (m *ResetStickyTaskListRequest) Reset()         { *m = ResetStickyTaskListRequest{} }\nfunc (m *ResetStickyTaskListRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ResetStickyTaskListRequest) ProtoMessage()    {}\nfunc (*ResetStickyTaskListRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{14}\n}\nfunc (m *ResetStickyTaskListRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetStickyTaskListRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetStickyTaskListRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetStickyTaskListRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetStickyTaskListRequest.Merge(m, src)\n}\nfunc (m *ResetStickyTaskListRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetStickyTaskListRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetStickyTaskListRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetStickyTaskListRequest proto.InternalMessageInfo\n\nfunc (m *ResetStickyTaskListRequest) GetRequest() *v1.ResetStickyTaskListRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *ResetStickyTaskListRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype ResetStickyTaskListResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ResetStickyTaskListResponse) Reset()         { *m = ResetStickyTaskListResponse{} }\nfunc (m *ResetStickyTaskListResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ResetStickyTaskListResponse) ProtoMessage()    {}\nfunc (*ResetStickyTaskListResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{15}\n}\nfunc (m *ResetStickyTaskListResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetStickyTaskListResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetStickyTaskListResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetStickyTaskListResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetStickyTaskListResponse.Merge(m, src)\n}\nfunc (m *ResetStickyTaskListResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetStickyTaskListResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetStickyTaskListResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetStickyTaskListResponse proto.InternalMessageInfo\n\ntype GetMutableStateRequest struct {\n\tDomainId             string                  `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution   `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tExpectedNextEventId  int64                   `protobuf:\"varint,3,opt,name=expected_next_event_id,json=expectedNextEventId,proto3\" json:\"expected_next_event_id,omitempty\"`\n\tCurrentBranchToken   []byte                  `protobuf:\"bytes,4,opt,name=current_branch_token,json=currentBranchToken,proto3\" json:\"current_branch_token,omitempty\"`\n\tVersionHistoryItem   *v11.VersionHistoryItem `protobuf:\"bytes,5,opt,name=version_history_item,json=versionHistoryItem,proto3\" json:\"version_history_item,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                `json:\"-\"`\n\tXXX_unrecognized     []byte                  `json:\"-\"`\n\tXXX_sizecache        int32                   `json:\"-\"`\n}\n\nfunc (m *GetMutableStateRequest) Reset()         { *m = GetMutableStateRequest{} }\nfunc (m *GetMutableStateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetMutableStateRequest) ProtoMessage()    {}\nfunc (*GetMutableStateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{16}\n}\nfunc (m *GetMutableStateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetMutableStateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetMutableStateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetMutableStateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetMutableStateRequest.Merge(m, src)\n}\nfunc (m *GetMutableStateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetMutableStateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetMutableStateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetMutableStateRequest proto.InternalMessageInfo\n\nfunc (m *GetMutableStateRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetMutableStateRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateRequest) GetExpectedNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.ExpectedNextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *GetMutableStateRequest) GetCurrentBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.CurrentBranchToken\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateRequest) GetVersionHistoryItem() *v11.VersionHistoryItem {\n\tif m != nil {\n\t\treturn m.VersionHistoryItem\n\t}\n\treturn nil\n}\n\ntype GetMutableStateResponse struct {\n\tWorkflowExecution                    *v1.WorkflowExecution           `protobuf:\"bytes,1,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tWorkflowType                         *v1.WorkflowType                `protobuf:\"bytes,2,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tNextEventId                          int64                           `protobuf:\"varint,3,opt,name=next_event_id,json=nextEventId,proto3\" json:\"next_event_id,omitempty\"`\n\tPreviousStartedEventId               *types.Int64Value               `protobuf:\"bytes,4,opt,name=previous_started_event_id,json=previousStartedEventId,proto3\" json:\"previous_started_event_id,omitempty\"`\n\tLastFirstEventId                     int64                           `protobuf:\"varint,5,opt,name=last_first_event_id,json=lastFirstEventId,proto3\" json:\"last_first_event_id,omitempty\"`\n\tTaskList                             *v1.TaskList                    `protobuf:\"bytes,6,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tStickyTaskList                       *v1.TaskList                    `protobuf:\"bytes,7,opt,name=sticky_task_list,json=stickyTaskList,proto3\" json:\"sticky_task_list,omitempty\"`\n\tClientLibraryVersion                 string                          `protobuf:\"bytes,8,opt,name=client_library_version,json=clientLibraryVersion,proto3\" json:\"client_library_version,omitempty\"`\n\tClientFeatureVersion                 string                          `protobuf:\"bytes,9,opt,name=client_feature_version,json=clientFeatureVersion,proto3\" json:\"client_feature_version,omitempty\"`\n\tClientImpl                           string                          `protobuf:\"bytes,10,opt,name=client_impl,json=clientImpl,proto3\" json:\"client_impl,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *types.Duration                 `protobuf:\"bytes,11,opt,name=sticky_task_list_schedule_to_start_timeout,json=stickyTaskListScheduleToStartTimeout,proto3\" json:\"sticky_task_list_schedule_to_start_timeout,omitempty\"`\n\tEventStoreVersion                    int32                           `protobuf:\"varint,12,opt,name=event_store_version,json=eventStoreVersion,proto3\" json:\"event_store_version,omitempty\"`\n\tCurrentBranchToken                   []byte                          `protobuf:\"bytes,13,opt,name=current_branch_token,json=currentBranchToken,proto3\" json:\"current_branch_token,omitempty\"`\n\tWorkflowState                        v12.WorkflowState               `protobuf:\"varint,14,opt,name=workflow_state,json=workflowState,proto3,enum=uber.cadence.shared.v1.WorkflowState\" json:\"workflow_state,omitempty\"`\n\tWorkflowCloseState                   v1.WorkflowExecutionCloseStatus `protobuf:\"varint,15,opt,name=workflow_close_state,json=workflowCloseState,proto3,enum=uber.cadence.api.v1.WorkflowExecutionCloseStatus\" json:\"workflow_close_state,omitempty\"`\n\tVersionHistories                     *v12.VersionHistories           `protobuf:\"bytes,16,opt,name=version_histories,json=versionHistories,proto3\" json:\"version_histories,omitempty\"`\n\tIsStickyTaskListEnabled              bool                            `protobuf:\"varint,17,opt,name=is_sticky_task_list_enabled,json=isStickyTaskListEnabled,proto3\" json:\"is_sticky_task_list_enabled,omitempty\"`\n\tHistorySize                          int64                           `protobuf:\"varint,18,opt,name=history_size,json=historySize,proto3\" json:\"history_size,omitempty\"`\n\tXXX_NoUnkeyedLiteral                 struct{}                        `json:\"-\"`\n\tXXX_unrecognized                     []byte                          `json:\"-\"`\n\tXXX_sizecache                        int32                           `json:\"-\"`\n}\n\nfunc (m *GetMutableStateResponse) Reset()         { *m = GetMutableStateResponse{} }\nfunc (m *GetMutableStateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetMutableStateResponse) ProtoMessage()    {}\nfunc (*GetMutableStateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{17}\n}\nfunc (m *GetMutableStateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetMutableStateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetMutableStateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetMutableStateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetMutableStateResponse.Merge(m, src)\n}\nfunc (m *GetMutableStateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetMutableStateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetMutableStateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetMutableStateResponse proto.InternalMessageInfo\n\nfunc (m *GetMutableStateResponse) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.NextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *GetMutableStateResponse) GetPreviousStartedEventId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.PreviousStartedEventId\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetLastFirstEventId() int64 {\n\tif m != nil {\n\t\treturn m.LastFirstEventId\n\t}\n\treturn 0\n}\n\nfunc (m *GetMutableStateResponse) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetStickyTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.StickyTaskList\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetClientLibraryVersion() string {\n\tif m != nil {\n\t\treturn m.ClientLibraryVersion\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetMutableStateResponse) GetClientFeatureVersion() string {\n\tif m != nil {\n\t\treturn m.ClientFeatureVersion\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetMutableStateResponse) GetClientImpl() string {\n\tif m != nil {\n\t\treturn m.ClientImpl\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetMutableStateResponse) GetStickyTaskListScheduleToStartTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.StickyTaskListScheduleToStartTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetEventStoreVersion() int32 {\n\tif m != nil {\n\t\treturn m.EventStoreVersion\n\t}\n\treturn 0\n}\n\nfunc (m *GetMutableStateResponse) GetCurrentBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.CurrentBranchToken\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetWorkflowState() v12.WorkflowState {\n\tif m != nil {\n\t\treturn m.WorkflowState\n\t}\n\treturn v12.WorkflowState_WORKFLOW_STATE_INVALID\n}\n\nfunc (m *GetMutableStateResponse) GetWorkflowCloseState() v1.WorkflowExecutionCloseStatus {\n\tif m != nil {\n\t\treturn m.WorkflowCloseState\n\t}\n\treturn v1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID\n}\n\nfunc (m *GetMutableStateResponse) GetVersionHistories() *v12.VersionHistories {\n\tif m != nil {\n\t\treturn m.VersionHistories\n\t}\n\treturn nil\n}\n\nfunc (m *GetMutableStateResponse) GetIsStickyTaskListEnabled() bool {\n\tif m != nil {\n\t\treturn m.IsStickyTaskListEnabled\n\t}\n\treturn false\n}\n\nfunc (m *GetMutableStateResponse) GetHistorySize() int64 {\n\tif m != nil {\n\t\treturn m.HistorySize\n\t}\n\treturn 0\n}\n\ntype PollMutableStateRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tExpectedNextEventId  int64                 `protobuf:\"varint,3,opt,name=expected_next_event_id,json=expectedNextEventId,proto3\" json:\"expected_next_event_id,omitempty\"`\n\tCurrentBranchToken   []byte                `protobuf:\"bytes,4,opt,name=current_branch_token,json=currentBranchToken,proto3\" json:\"current_branch_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *PollMutableStateRequest) Reset()         { *m = PollMutableStateRequest{} }\nfunc (m *PollMutableStateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*PollMutableStateRequest) ProtoMessage()    {}\nfunc (*PollMutableStateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{18}\n}\nfunc (m *PollMutableStateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollMutableStateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollMutableStateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollMutableStateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollMutableStateRequest.Merge(m, src)\n}\nfunc (m *PollMutableStateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollMutableStateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollMutableStateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollMutableStateRequest proto.InternalMessageInfo\n\nfunc (m *PollMutableStateRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollMutableStateRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateRequest) GetExpectedNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.ExpectedNextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *PollMutableStateRequest) GetCurrentBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.CurrentBranchToken\n\t}\n\treturn nil\n}\n\ntype PollMutableStateResponse struct {\n\tWorkflowExecution                    *v1.WorkflowExecution           `protobuf:\"bytes,1,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tWorkflowType                         *v1.WorkflowType                `protobuf:\"bytes,2,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tNextEventId                          int64                           `protobuf:\"varint,3,opt,name=next_event_id,json=nextEventId,proto3\" json:\"next_event_id,omitempty\"`\n\tPreviousStartedEventId               *types.Int64Value               `protobuf:\"bytes,4,opt,name=previous_started_event_id,json=previousStartedEventId,proto3\" json:\"previous_started_event_id,omitempty\"`\n\tLastFirstEventId                     int64                           `protobuf:\"varint,5,opt,name=last_first_event_id,json=lastFirstEventId,proto3\" json:\"last_first_event_id,omitempty\"`\n\tTaskList                             *v1.TaskList                    `protobuf:\"bytes,6,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tStickyTaskList                       *v1.TaskList                    `protobuf:\"bytes,7,opt,name=sticky_task_list,json=stickyTaskList,proto3\" json:\"sticky_task_list,omitempty\"`\n\tClientLibraryVersion                 string                          `protobuf:\"bytes,8,opt,name=client_library_version,json=clientLibraryVersion,proto3\" json:\"client_library_version,omitempty\"`\n\tClientFeatureVersion                 string                          `protobuf:\"bytes,9,opt,name=client_feature_version,json=clientFeatureVersion,proto3\" json:\"client_feature_version,omitempty\"`\n\tClientImpl                           string                          `protobuf:\"bytes,10,opt,name=client_impl,json=clientImpl,proto3\" json:\"client_impl,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *types.Duration                 `protobuf:\"bytes,11,opt,name=sticky_task_list_schedule_to_start_timeout,json=stickyTaskListScheduleToStartTimeout,proto3\" json:\"sticky_task_list_schedule_to_start_timeout,omitempty\"`\n\tCurrentBranchToken                   []byte                          `protobuf:\"bytes,12,opt,name=current_branch_token,json=currentBranchToken,proto3\" json:\"current_branch_token,omitempty\"`\n\tVersionHistories                     *v12.VersionHistories           `protobuf:\"bytes,13,opt,name=version_histories,json=versionHistories,proto3\" json:\"version_histories,omitempty\"`\n\tWorkflowState                        v12.WorkflowState               `protobuf:\"varint,14,opt,name=workflow_state,json=workflowState,proto3,enum=uber.cadence.shared.v1.WorkflowState\" json:\"workflow_state,omitempty\"`\n\tWorkflowCloseState                   v1.WorkflowExecutionCloseStatus `protobuf:\"varint,15,opt,name=workflow_close_state,json=workflowCloseState,proto3,enum=uber.cadence.api.v1.WorkflowExecutionCloseStatus\" json:\"workflow_close_state,omitempty\"`\n\tXXX_NoUnkeyedLiteral                 struct{}                        `json:\"-\"`\n\tXXX_unrecognized                     []byte                          `json:\"-\"`\n\tXXX_sizecache                        int32                           `json:\"-\"`\n}\n\nfunc (m *PollMutableStateResponse) Reset()         { *m = PollMutableStateResponse{} }\nfunc (m *PollMutableStateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*PollMutableStateResponse) ProtoMessage()    {}\nfunc (*PollMutableStateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{19}\n}\nfunc (m *PollMutableStateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollMutableStateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollMutableStateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollMutableStateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollMutableStateResponse.Merge(m, src)\n}\nfunc (m *PollMutableStateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollMutableStateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollMutableStateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollMutableStateResponse proto.InternalMessageInfo\n\nfunc (m *PollMutableStateResponse) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.NextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *PollMutableStateResponse) GetPreviousStartedEventId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.PreviousStartedEventId\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetLastFirstEventId() int64 {\n\tif m != nil {\n\t\treturn m.LastFirstEventId\n\t}\n\treturn 0\n}\n\nfunc (m *PollMutableStateResponse) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetStickyTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.StickyTaskList\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetClientLibraryVersion() string {\n\tif m != nil {\n\t\treturn m.ClientLibraryVersion\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollMutableStateResponse) GetClientFeatureVersion() string {\n\tif m != nil {\n\t\treturn m.ClientFeatureVersion\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollMutableStateResponse) GetClientImpl() string {\n\tif m != nil {\n\t\treturn m.ClientImpl\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollMutableStateResponse) GetStickyTaskListScheduleToStartTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.StickyTaskListScheduleToStartTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetCurrentBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.CurrentBranchToken\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetVersionHistories() *v12.VersionHistories {\n\tif m != nil {\n\t\treturn m.VersionHistories\n\t}\n\treturn nil\n}\n\nfunc (m *PollMutableStateResponse) GetWorkflowState() v12.WorkflowState {\n\tif m != nil {\n\t\treturn m.WorkflowState\n\t}\n\treturn v12.WorkflowState_WORKFLOW_STATE_INVALID\n}\n\nfunc (m *PollMutableStateResponse) GetWorkflowCloseState() v1.WorkflowExecutionCloseStatus {\n\tif m != nil {\n\t\treturn m.WorkflowCloseState\n\t}\n\treturn v1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID\n}\n\ntype RecordDecisionTaskStartedRequest struct {\n\tDomainId          string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tScheduleId        int64                 `protobuf:\"varint,3,opt,name=schedule_id,json=scheduleId,proto3\" json:\"schedule_id,omitempty\"`\n\tTaskId            int64                 `protobuf:\"varint,4,opt,name=task_id,json=taskId,proto3\" json:\"task_id,omitempty\"`\n\t// Unique id of each poll request. Used to ensure at most once delivery of tasks.\n\tRequestId            string                         `protobuf:\"bytes,5,opt,name=request_id,json=requestId,proto3\" json:\"request_id,omitempty\"`\n\tPollRequest          *v1.PollForDecisionTaskRequest `protobuf:\"bytes,6,opt,name=poll_request,json=pollRequest,proto3\" json:\"poll_request,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                       `json:\"-\"`\n\tXXX_unrecognized     []byte                         `json:\"-\"`\n\tXXX_sizecache        int32                          `json:\"-\"`\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) Reset()         { *m = RecordDecisionTaskStartedRequest{} }\nfunc (m *RecordDecisionTaskStartedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RecordDecisionTaskStartedRequest) ProtoMessage()    {}\nfunc (*RecordDecisionTaskStartedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{20}\n}\nfunc (m *RecordDecisionTaskStartedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordDecisionTaskStartedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordDecisionTaskStartedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordDecisionTaskStartedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordDecisionTaskStartedRequest.Merge(m, src)\n}\nfunc (m *RecordDecisionTaskStartedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordDecisionTaskStartedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordDecisionTaskStartedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordDecisionTaskStartedRequest proto.InternalMessageInfo\n\nfunc (m *RecordDecisionTaskStartedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) GetScheduleId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduleId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) GetTaskId() int64 {\n\tif m != nil {\n\t\treturn m.TaskId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) GetRequestId() string {\n\tif m != nil {\n\t\treturn m.RequestId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) GetPollRequest() *v1.PollForDecisionTaskRequest {\n\tif m != nil {\n\t\treturn m.PollRequest\n\t}\n\treturn nil\n}\n\ntype RecordDecisionTaskStartedResponse struct {\n\tWorkflowType              *v1.WorkflowType             `protobuf:\"bytes,1,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tPreviousStartedEventId    *types.Int64Value            `protobuf:\"bytes,2,opt,name=previous_started_event_id,json=previousStartedEventId,proto3\" json:\"previous_started_event_id,omitempty\"`\n\tScheduledEventId          int64                        `protobuf:\"varint,3,opt,name=scheduled_event_id,json=scheduledEventId,proto3\" json:\"scheduled_event_id,omitempty\"`\n\tStartedEventId            int64                        `protobuf:\"varint,4,opt,name=started_event_id,json=startedEventId,proto3\" json:\"started_event_id,omitempty\"`\n\tNextEventId               int64                        `protobuf:\"varint,5,opt,name=next_event_id,json=nextEventId,proto3\" json:\"next_event_id,omitempty\"`\n\tAttempt                   int32                        `protobuf:\"varint,6,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tStickyExecutionEnabled    bool                         `protobuf:\"varint,7,opt,name=sticky_execution_enabled,json=stickyExecutionEnabled,proto3\" json:\"sticky_execution_enabled,omitempty\"`\n\tDecisionInfo              *v12.TransientDecisionInfo   `protobuf:\"bytes,8,opt,name=decision_info,json=decisionInfo,proto3\" json:\"decision_info,omitempty\"`\n\tWorkflowExecutionTaskList *v1.TaskList                 `protobuf:\"bytes,9,opt,name=workflow_execution_task_list,json=workflowExecutionTaskList,proto3\" json:\"workflow_execution_task_list,omitempty\"`\n\tEventStoreVersion         int32                        `protobuf:\"varint,10,opt,name=event_store_version,json=eventStoreVersion,proto3\" json:\"event_store_version,omitempty\"`\n\tBranchToken               []byte                       `protobuf:\"bytes,11,opt,name=branch_token,json=branchToken,proto3\" json:\"branch_token,omitempty\"`\n\tScheduledTime             *types.Timestamp             `protobuf:\"bytes,12,opt,name=scheduled_time,json=scheduledTime,proto3\" json:\"scheduled_time,omitempty\"`\n\tStartedTime               *types.Timestamp             `protobuf:\"bytes,13,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tQueries                   map[string]*v1.WorkflowQuery `protobuf:\"bytes,14,rep,name=queries,proto3\" json:\"queries,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tHistorySize               int64                        `protobuf:\"varint,15,opt,name=history_size,json=historySize,proto3\" json:\"history_size,omitempty\"`\n\tXXX_NoUnkeyedLiteral      struct{}                     `json:\"-\"`\n\tXXX_unrecognized          []byte                       `json:\"-\"`\n\tXXX_sizecache             int32                        `json:\"-\"`\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) Reset()         { *m = RecordDecisionTaskStartedResponse{} }\nfunc (m *RecordDecisionTaskStartedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RecordDecisionTaskStartedResponse) ProtoMessage()    {}\nfunc (*RecordDecisionTaskStartedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{21}\n}\nfunc (m *RecordDecisionTaskStartedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordDecisionTaskStartedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordDecisionTaskStartedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordDecisionTaskStartedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordDecisionTaskStartedResponse.Merge(m, src)\n}\nfunc (m *RecordDecisionTaskStartedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordDecisionTaskStartedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordDecisionTaskStartedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordDecisionTaskStartedResponse proto.InternalMessageInfo\n\nfunc (m *RecordDecisionTaskStartedResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetPreviousStartedEventId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.PreviousStartedEventId\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetScheduledEventId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduledEventId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetStartedEventId() int64 {\n\tif m != nil {\n\t\treturn m.StartedEventId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.NextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetStickyExecutionEnabled() bool {\n\tif m != nil {\n\t\treturn m.StickyExecutionEnabled\n\t}\n\treturn false\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetDecisionInfo() *v12.TransientDecisionInfo {\n\tif m != nil {\n\t\treturn m.DecisionInfo\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetWorkflowExecutionTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.WorkflowExecutionTaskList\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetEventStoreVersion() int32 {\n\tif m != nil {\n\t\treturn m.EventStoreVersion\n\t}\n\treturn 0\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.BranchToken\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetScheduledTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTime\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetQueries() map[string]*v1.WorkflowQuery {\n\tif m != nil {\n\t\treturn m.Queries\n\t}\n\treturn nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) GetHistorySize() int64 {\n\tif m != nil {\n\t\treturn m.HistorySize\n\t}\n\treturn 0\n}\n\ntype RecordActivityTaskStartedRequest struct {\n\tDomainId          string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tScheduleId        int64                 `protobuf:\"varint,3,opt,name=schedule_id,json=scheduleId,proto3\" json:\"schedule_id,omitempty\"`\n\tTaskId            int64                 `protobuf:\"varint,4,opt,name=task_id,json=taskId,proto3\" json:\"task_id,omitempty\"`\n\t// Unique id of each poll request. Used to ensure at most once delivery of tasks.\n\tRequestId            string                         `protobuf:\"bytes,5,opt,name=request_id,json=requestId,proto3\" json:\"request_id,omitempty\"`\n\tPollRequest          *v1.PollForActivityTaskRequest `protobuf:\"bytes,6,opt,name=poll_request,json=pollRequest,proto3\" json:\"poll_request,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                       `json:\"-\"`\n\tXXX_unrecognized     []byte                         `json:\"-\"`\n\tXXX_sizecache        int32                          `json:\"-\"`\n}\n\nfunc (m *RecordActivityTaskStartedRequest) Reset()         { *m = RecordActivityTaskStartedRequest{} }\nfunc (m *RecordActivityTaskStartedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RecordActivityTaskStartedRequest) ProtoMessage()    {}\nfunc (*RecordActivityTaskStartedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{22}\n}\nfunc (m *RecordActivityTaskStartedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordActivityTaskStartedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordActivityTaskStartedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordActivityTaskStartedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordActivityTaskStartedRequest.Merge(m, src)\n}\nfunc (m *RecordActivityTaskStartedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordActivityTaskStartedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordActivityTaskStartedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordActivityTaskStartedRequest proto.InternalMessageInfo\n\nfunc (m *RecordActivityTaskStartedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RecordActivityTaskStartedRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedRequest) GetScheduleId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduleId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordActivityTaskStartedRequest) GetTaskId() int64 {\n\tif m != nil {\n\t\treturn m.TaskId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordActivityTaskStartedRequest) GetRequestId() string {\n\tif m != nil {\n\t\treturn m.RequestId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RecordActivityTaskStartedRequest) GetPollRequest() *v1.PollForActivityTaskRequest {\n\tif m != nil {\n\t\treturn m.PollRequest\n\t}\n\treturn nil\n}\n\ntype RecordActivityTaskStartedResponse struct {\n\tScheduledEvent             *v1.HistoryEvent `protobuf:\"bytes,1,opt,name=scheduled_event,json=scheduledEvent,proto3\" json:\"scheduled_event,omitempty\"`\n\tStartedTime                *types.Timestamp `protobuf:\"bytes,2,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tAttempt                    int32            `protobuf:\"varint,3,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tScheduledTimeOfThisAttempt *types.Timestamp `protobuf:\"bytes,4,opt,name=scheduled_time_of_this_attempt,json=scheduledTimeOfThisAttempt,proto3\" json:\"scheduled_time_of_this_attempt,omitempty\"`\n\tHeartbeatDetails           *v1.Payload      `protobuf:\"bytes,5,opt,name=heartbeat_details,json=heartbeatDetails,proto3\" json:\"heartbeat_details,omitempty\"`\n\tWorkflowType               *v1.WorkflowType `protobuf:\"bytes,6,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tWorkflowDomain             string           `protobuf:\"bytes,7,opt,name=workflow_domain,json=workflowDomain,proto3\" json:\"workflow_domain,omitempty\"`\n\tXXX_NoUnkeyedLiteral       struct{}         `json:\"-\"`\n\tXXX_unrecognized           []byte           `json:\"-\"`\n\tXXX_sizecache              int32            `json:\"-\"`\n}\n\nfunc (m *RecordActivityTaskStartedResponse) Reset()         { *m = RecordActivityTaskStartedResponse{} }\nfunc (m *RecordActivityTaskStartedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RecordActivityTaskStartedResponse) ProtoMessage()    {}\nfunc (*RecordActivityTaskStartedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{23}\n}\nfunc (m *RecordActivityTaskStartedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordActivityTaskStartedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordActivityTaskStartedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordActivityTaskStartedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordActivityTaskStartedResponse.Merge(m, src)\n}\nfunc (m *RecordActivityTaskStartedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordActivityTaskStartedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordActivityTaskStartedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordActivityTaskStartedResponse proto.InternalMessageInfo\n\nfunc (m *RecordActivityTaskStartedResponse) GetScheduledEvent() *v1.HistoryEvent {\n\tif m != nil {\n\t\treturn m.ScheduledEvent\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetScheduledTimeOfThisAttempt() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTimeOfThisAttempt\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetHeartbeatDetails() *v1.Payload {\n\tif m != nil {\n\t\treturn m.HeartbeatDetails\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) GetWorkflowDomain() string {\n\tif m != nil {\n\t\treturn m.WorkflowDomain\n\t}\n\treturn \"\"\n}\n\ntype RespondDecisionTaskCompletedRequest struct {\n\tRequest              *v1.RespondDecisionTaskCompletedRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                                  `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                                `json:\"-\"`\n\tXXX_unrecognized     []byte                                  `json:\"-\"`\n\tXXX_sizecache        int32                                   `json:\"-\"`\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) Reset()         { *m = RespondDecisionTaskCompletedRequest{} }\nfunc (m *RespondDecisionTaskCompletedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondDecisionTaskCompletedRequest) ProtoMessage()    {}\nfunc (*RespondDecisionTaskCompletedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{24}\n}\nfunc (m *RespondDecisionTaskCompletedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondDecisionTaskCompletedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondDecisionTaskCompletedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondDecisionTaskCompletedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondDecisionTaskCompletedRequest.Merge(m, src)\n}\nfunc (m *RespondDecisionTaskCompletedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondDecisionTaskCompletedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondDecisionTaskCompletedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondDecisionTaskCompletedRequest proto.InternalMessageInfo\n\nfunc (m *RespondDecisionTaskCompletedRequest) GetRequest() *v1.RespondDecisionTaskCompletedRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RespondDecisionTaskCompletedResponse struct {\n\tStartedResponse             *RecordDecisionTaskStartedResponse       `protobuf:\"bytes,1,opt,name=started_response,json=startedResponse,proto3\" json:\"started_response,omitempty\"`\n\tActivitiesToDispatchLocally map[string]*v1.ActivityLocalDispatchInfo `protobuf:\"bytes,2,rep,name=activities_to_dispatch_locally,json=activitiesToDispatchLocally,proto3\" json:\"activities_to_dispatch_locally,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral        struct{}                                 `json:\"-\"`\n\tXXX_unrecognized            []byte                                   `json:\"-\"`\n\tXXX_sizecache               int32                                    `json:\"-\"`\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) Reset()         { *m = RespondDecisionTaskCompletedResponse{} }\nfunc (m *RespondDecisionTaskCompletedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondDecisionTaskCompletedResponse) ProtoMessage()    {}\nfunc (*RespondDecisionTaskCompletedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{25}\n}\nfunc (m *RespondDecisionTaskCompletedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondDecisionTaskCompletedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondDecisionTaskCompletedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondDecisionTaskCompletedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondDecisionTaskCompletedResponse.Merge(m, src)\n}\nfunc (m *RespondDecisionTaskCompletedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondDecisionTaskCompletedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondDecisionTaskCompletedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondDecisionTaskCompletedResponse proto.InternalMessageInfo\n\nfunc (m *RespondDecisionTaskCompletedResponse) GetStartedResponse() *RecordDecisionTaskStartedResponse {\n\tif m != nil {\n\t\treturn m.StartedResponse\n\t}\n\treturn nil\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) GetActivitiesToDispatchLocally() map[string]*v1.ActivityLocalDispatchInfo {\n\tif m != nil {\n\t\treturn m.ActivitiesToDispatchLocally\n\t}\n\treturn nil\n}\n\ntype RespondDecisionTaskFailedRequest struct {\n\tRequest              *v1.RespondDecisionTaskFailedRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                               `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                             `json:\"-\"`\n\tXXX_unrecognized     []byte                               `json:\"-\"`\n\tXXX_sizecache        int32                                `json:\"-\"`\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) Reset()         { *m = RespondDecisionTaskFailedRequest{} }\nfunc (m *RespondDecisionTaskFailedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondDecisionTaskFailedRequest) ProtoMessage()    {}\nfunc (*RespondDecisionTaskFailedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{26}\n}\nfunc (m *RespondDecisionTaskFailedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondDecisionTaskFailedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondDecisionTaskFailedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondDecisionTaskFailedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondDecisionTaskFailedRequest.Merge(m, src)\n}\nfunc (m *RespondDecisionTaskFailedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondDecisionTaskFailedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondDecisionTaskFailedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondDecisionTaskFailedRequest proto.InternalMessageInfo\n\nfunc (m *RespondDecisionTaskFailedRequest) GetRequest() *v1.RespondDecisionTaskFailedRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RespondDecisionTaskFailedResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RespondDecisionTaskFailedResponse) Reset()         { *m = RespondDecisionTaskFailedResponse{} }\nfunc (m *RespondDecisionTaskFailedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondDecisionTaskFailedResponse) ProtoMessage()    {}\nfunc (*RespondDecisionTaskFailedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{27}\n}\nfunc (m *RespondDecisionTaskFailedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondDecisionTaskFailedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondDecisionTaskFailedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondDecisionTaskFailedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondDecisionTaskFailedResponse.Merge(m, src)\n}\nfunc (m *RespondDecisionTaskFailedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondDecisionTaskFailedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondDecisionTaskFailedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondDecisionTaskFailedResponse proto.InternalMessageInfo\n\ntype RecordActivityTaskHeartbeatRequest struct {\n\tRequest              *v1.RecordActivityTaskHeartbeatRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                                 `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                               `json:\"-\"`\n\tXXX_unrecognized     []byte                                 `json:\"-\"`\n\tXXX_sizecache        int32                                  `json:\"-\"`\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) Reset()         { *m = RecordActivityTaskHeartbeatRequest{} }\nfunc (m *RecordActivityTaskHeartbeatRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RecordActivityTaskHeartbeatRequest) ProtoMessage()    {}\nfunc (*RecordActivityTaskHeartbeatRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{28}\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordActivityTaskHeartbeatRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordActivityTaskHeartbeatRequest.Merge(m, src)\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordActivityTaskHeartbeatRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordActivityTaskHeartbeatRequest proto.InternalMessageInfo\n\nfunc (m *RecordActivityTaskHeartbeatRequest) GetRequest() *v1.RecordActivityTaskHeartbeatRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RecordActivityTaskHeartbeatResponse struct {\n\tCancelRequested      bool     `protobuf:\"varint,1,opt,name=cancel_requested,json=cancelRequested,proto3\" json:\"cancel_requested,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RecordActivityTaskHeartbeatResponse) Reset()         { *m = RecordActivityTaskHeartbeatResponse{} }\nfunc (m *RecordActivityTaskHeartbeatResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RecordActivityTaskHeartbeatResponse) ProtoMessage()    {}\nfunc (*RecordActivityTaskHeartbeatResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{29}\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordActivityTaskHeartbeatResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordActivityTaskHeartbeatResponse.Merge(m, src)\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordActivityTaskHeartbeatResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordActivityTaskHeartbeatResponse proto.InternalMessageInfo\n\nfunc (m *RecordActivityTaskHeartbeatResponse) GetCancelRequested() bool {\n\tif m != nil {\n\t\treturn m.CancelRequested\n\t}\n\treturn false\n}\n\ntype RespondActivityTaskCompletedRequest struct {\n\tRequest              *v1.RespondActivityTaskCompletedRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                                  `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                                `json:\"-\"`\n\tXXX_unrecognized     []byte                                  `json:\"-\"`\n\tXXX_sizecache        int32                                   `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) Reset()         { *m = RespondActivityTaskCompletedRequest{} }\nfunc (m *RespondActivityTaskCompletedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskCompletedRequest) ProtoMessage()    {}\nfunc (*RespondActivityTaskCompletedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{30}\n}\nfunc (m *RespondActivityTaskCompletedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskCompletedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskCompletedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskCompletedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskCompletedRequest.Merge(m, src)\n}\nfunc (m *RespondActivityTaskCompletedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskCompletedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskCompletedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskCompletedRequest proto.InternalMessageInfo\n\nfunc (m *RespondActivityTaskCompletedRequest) GetRequest() *v1.RespondActivityTaskCompletedRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RespondActivityTaskCompletedResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskCompletedResponse) Reset()         { *m = RespondActivityTaskCompletedResponse{} }\nfunc (m *RespondActivityTaskCompletedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskCompletedResponse) ProtoMessage()    {}\nfunc (*RespondActivityTaskCompletedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{31}\n}\nfunc (m *RespondActivityTaskCompletedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskCompletedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskCompletedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskCompletedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskCompletedResponse.Merge(m, src)\n}\nfunc (m *RespondActivityTaskCompletedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskCompletedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskCompletedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskCompletedResponse proto.InternalMessageInfo\n\ntype RespondActivityTaskFailedRequest struct {\n\tRequest              *v1.RespondActivityTaskFailedRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                               `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                             `json:\"-\"`\n\tXXX_unrecognized     []byte                               `json:\"-\"`\n\tXXX_sizecache        int32                                `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskFailedRequest) Reset()         { *m = RespondActivityTaskFailedRequest{} }\nfunc (m *RespondActivityTaskFailedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskFailedRequest) ProtoMessage()    {}\nfunc (*RespondActivityTaskFailedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{32}\n}\nfunc (m *RespondActivityTaskFailedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskFailedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskFailedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskFailedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskFailedRequest.Merge(m, src)\n}\nfunc (m *RespondActivityTaskFailedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskFailedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskFailedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskFailedRequest proto.InternalMessageInfo\n\nfunc (m *RespondActivityTaskFailedRequest) GetRequest() *v1.RespondActivityTaskFailedRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondActivityTaskFailedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RespondActivityTaskFailedResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskFailedResponse) Reset()         { *m = RespondActivityTaskFailedResponse{} }\nfunc (m *RespondActivityTaskFailedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskFailedResponse) ProtoMessage()    {}\nfunc (*RespondActivityTaskFailedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{33}\n}\nfunc (m *RespondActivityTaskFailedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskFailedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskFailedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskFailedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskFailedResponse.Merge(m, src)\n}\nfunc (m *RespondActivityTaskFailedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskFailedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskFailedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskFailedResponse proto.InternalMessageInfo\n\ntype RespondActivityTaskCanceledRequest struct {\n\tRequest              *v1.RespondActivityTaskCanceledRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                                 `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                               `json:\"-\"`\n\tXXX_unrecognized     []byte                                 `json:\"-\"`\n\tXXX_sizecache        int32                                  `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) Reset()         { *m = RespondActivityTaskCanceledRequest{} }\nfunc (m *RespondActivityTaskCanceledRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskCanceledRequest) ProtoMessage()    {}\nfunc (*RespondActivityTaskCanceledRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{34}\n}\nfunc (m *RespondActivityTaskCanceledRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskCanceledRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskCanceledRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskCanceledRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskCanceledRequest.Merge(m, src)\n}\nfunc (m *RespondActivityTaskCanceledRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskCanceledRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskCanceledRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskCanceledRequest proto.InternalMessageInfo\n\nfunc (m *RespondActivityTaskCanceledRequest) GetRequest() *v1.RespondActivityTaskCanceledRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype RespondActivityTaskCanceledResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RespondActivityTaskCanceledResponse) Reset()         { *m = RespondActivityTaskCanceledResponse{} }\nfunc (m *RespondActivityTaskCanceledResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondActivityTaskCanceledResponse) ProtoMessage()    {}\nfunc (*RespondActivityTaskCanceledResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{35}\n}\nfunc (m *RespondActivityTaskCanceledResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondActivityTaskCanceledResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondActivityTaskCanceledResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondActivityTaskCanceledResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondActivityTaskCanceledResponse.Merge(m, src)\n}\nfunc (m *RespondActivityTaskCanceledResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondActivityTaskCanceledResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondActivityTaskCanceledResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondActivityTaskCanceledResponse proto.InternalMessageInfo\n\ntype RemoveSignalMutableStateRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tRequestId            string                `protobuf:\"bytes,3,opt,name=request_id,json=requestId,proto3\" json:\"request_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *RemoveSignalMutableStateRequest) Reset()         { *m = RemoveSignalMutableStateRequest{} }\nfunc (m *RemoveSignalMutableStateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RemoveSignalMutableStateRequest) ProtoMessage()    {}\nfunc (*RemoveSignalMutableStateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{36}\n}\nfunc (m *RemoveSignalMutableStateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RemoveSignalMutableStateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RemoveSignalMutableStateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RemoveSignalMutableStateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RemoveSignalMutableStateRequest.Merge(m, src)\n}\nfunc (m *RemoveSignalMutableStateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RemoveSignalMutableStateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RemoveSignalMutableStateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RemoveSignalMutableStateRequest proto.InternalMessageInfo\n\nfunc (m *RemoveSignalMutableStateRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RemoveSignalMutableStateRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RemoveSignalMutableStateRequest) GetRequestId() string {\n\tif m != nil {\n\t\treturn m.RequestId\n\t}\n\treturn \"\"\n}\n\ntype RemoveSignalMutableStateResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RemoveSignalMutableStateResponse) Reset()         { *m = RemoveSignalMutableStateResponse{} }\nfunc (m *RemoveSignalMutableStateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RemoveSignalMutableStateResponse) ProtoMessage()    {}\nfunc (*RemoveSignalMutableStateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{37}\n}\nfunc (m *RemoveSignalMutableStateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RemoveSignalMutableStateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RemoveSignalMutableStateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RemoveSignalMutableStateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RemoveSignalMutableStateResponse.Merge(m, src)\n}\nfunc (m *RemoveSignalMutableStateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RemoveSignalMutableStateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RemoveSignalMutableStateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RemoveSignalMutableStateResponse proto.InternalMessageInfo\n\ntype RequestCancelWorkflowExecutionRequest struct {\n\tDomainId      string                                    `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tCancelRequest *v1.RequestCancelWorkflowExecutionRequest `protobuf:\"bytes,2,opt,name=cancel_request,json=cancelRequest,proto3\" json:\"cancel_request,omitempty\"`\n\t// workflow execution that requests this cancellation, for making sure\n\t// the workflow being cancelled is actually a child of the workflow\n\t// making the request\n\tExternalExecutionInfo *v1.ExternalExecutionInfo `protobuf:\"bytes,3,opt,name=external_execution_info,json=externalExecutionInfo,proto3\" json:\"external_execution_info,omitempty\"`\n\tChildWorkflowOnly     bool                      `protobuf:\"varint,4,opt,name=child_workflow_only,json=childWorkflowOnly,proto3\" json:\"child_workflow_only,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{}                  `json:\"-\"`\n\tXXX_unrecognized      []byte                    `json:\"-\"`\n\tXXX_sizecache         int32                     `json:\"-\"`\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) Reset()         { *m = RequestCancelWorkflowExecutionRequest{} }\nfunc (m *RequestCancelWorkflowExecutionRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RequestCancelWorkflowExecutionRequest) ProtoMessage()    {}\nfunc (*RequestCancelWorkflowExecutionRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{38}\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RequestCancelWorkflowExecutionRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RequestCancelWorkflowExecutionRequest.Merge(m, src)\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RequestCancelWorkflowExecutionRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RequestCancelWorkflowExecutionRequest proto.InternalMessageInfo\n\nfunc (m *RequestCancelWorkflowExecutionRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) GetCancelRequest() *v1.RequestCancelWorkflowExecutionRequest {\n\tif m != nil {\n\t\treturn m.CancelRequest\n\t}\n\treturn nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) GetExternalExecutionInfo() *v1.ExternalExecutionInfo {\n\tif m != nil {\n\t\treturn m.ExternalExecutionInfo\n\t}\n\treturn nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) GetChildWorkflowOnly() bool {\n\tif m != nil {\n\t\treturn m.ChildWorkflowOnly\n\t}\n\treturn false\n}\n\ntype RequestCancelWorkflowExecutionResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RequestCancelWorkflowExecutionResponse) Reset() {\n\t*m = RequestCancelWorkflowExecutionResponse{}\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RequestCancelWorkflowExecutionResponse) ProtoMessage()    {}\nfunc (*RequestCancelWorkflowExecutionResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{39}\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RequestCancelWorkflowExecutionResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RequestCancelWorkflowExecutionResponse.Merge(m, src)\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RequestCancelWorkflowExecutionResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RequestCancelWorkflowExecutionResponse proto.InternalMessageInfo\n\ntype ScheduleDecisionTaskRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tIsFirstDecision      bool                  `protobuf:\"varint,3,opt,name=is_first_decision,json=isFirstDecision,proto3\" json:\"is_first_decision,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *ScheduleDecisionTaskRequest) Reset()         { *m = ScheduleDecisionTaskRequest{} }\nfunc (m *ScheduleDecisionTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ScheduleDecisionTaskRequest) ProtoMessage()    {}\nfunc (*ScheduleDecisionTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{40}\n}\nfunc (m *ScheduleDecisionTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ScheduleDecisionTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ScheduleDecisionTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ScheduleDecisionTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ScheduleDecisionTaskRequest.Merge(m, src)\n}\nfunc (m *ScheduleDecisionTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ScheduleDecisionTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ScheduleDecisionTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ScheduleDecisionTaskRequest proto.InternalMessageInfo\n\nfunc (m *ScheduleDecisionTaskRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *ScheduleDecisionTaskRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *ScheduleDecisionTaskRequest) GetIsFirstDecision() bool {\n\tif m != nil {\n\t\treturn m.IsFirstDecision\n\t}\n\treturn false\n}\n\ntype ScheduleDecisionTaskResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ScheduleDecisionTaskResponse) Reset()         { *m = ScheduleDecisionTaskResponse{} }\nfunc (m *ScheduleDecisionTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ScheduleDecisionTaskResponse) ProtoMessage()    {}\nfunc (*ScheduleDecisionTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{41}\n}\nfunc (m *ScheduleDecisionTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ScheduleDecisionTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ScheduleDecisionTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ScheduleDecisionTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ScheduleDecisionTaskResponse.Merge(m, src)\n}\nfunc (m *ScheduleDecisionTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ScheduleDecisionTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ScheduleDecisionTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ScheduleDecisionTaskResponse proto.InternalMessageInfo\n\n// RecordChildExecutionCompletedRequest is used for reporting the completion of child execution to parent workflow\n// execution which started it.  When a child execution is completed it creates this request and calls the\n// RecordChildExecutionCompleted API with the workflowExecution of parent.  It also sets the completedExecution of the\n// child as it could potentially be different than the ChildExecutionStartedEvent of parent in the situation when\n// child creates multiple runs through ContinueAsNew before finally completing.\ntype RecordChildExecutionCompletedRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tInitiatedId          int64                 `protobuf:\"varint,3,opt,name=initiated_id,json=initiatedId,proto3\" json:\"initiated_id,omitempty\"`\n\tCompletedExecution   *v1.WorkflowExecution `protobuf:\"bytes,4,opt,name=completed_execution,json=completedExecution,proto3\" json:\"completed_execution,omitempty\"`\n\tCompletionEvent      *v1.HistoryEvent      `protobuf:\"bytes,5,opt,name=completion_event,json=completionEvent,proto3\" json:\"completion_event,omitempty\"`\n\tStartedId            int64                 `protobuf:\"varint,6,opt,name=started_id,json=startedId,proto3\" json:\"started_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) Reset()         { *m = RecordChildExecutionCompletedRequest{} }\nfunc (m *RecordChildExecutionCompletedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RecordChildExecutionCompletedRequest) ProtoMessage()    {}\nfunc (*RecordChildExecutionCompletedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{42}\n}\nfunc (m *RecordChildExecutionCompletedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordChildExecutionCompletedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordChildExecutionCompletedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordChildExecutionCompletedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordChildExecutionCompletedRequest.Merge(m, src)\n}\nfunc (m *RecordChildExecutionCompletedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordChildExecutionCompletedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordChildExecutionCompletedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordChildExecutionCompletedRequest proto.InternalMessageInfo\n\nfunc (m *RecordChildExecutionCompletedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) GetInitiatedId() int64 {\n\tif m != nil {\n\t\treturn m.InitiatedId\n\t}\n\treturn 0\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) GetCompletedExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.CompletedExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) GetCompletionEvent() *v1.HistoryEvent {\n\tif m != nil {\n\t\treturn m.CompletionEvent\n\t}\n\treturn nil\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) GetStartedId() int64 {\n\tif m != nil {\n\t\treturn m.StartedId\n\t}\n\treturn 0\n}\n\ntype RecordChildExecutionCompletedResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RecordChildExecutionCompletedResponse) Reset()         { *m = RecordChildExecutionCompletedResponse{} }\nfunc (m *RecordChildExecutionCompletedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RecordChildExecutionCompletedResponse) ProtoMessage()    {}\nfunc (*RecordChildExecutionCompletedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{43}\n}\nfunc (m *RecordChildExecutionCompletedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RecordChildExecutionCompletedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RecordChildExecutionCompletedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RecordChildExecutionCompletedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RecordChildExecutionCompletedResponse.Merge(m, src)\n}\nfunc (m *RecordChildExecutionCompletedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RecordChildExecutionCompletedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RecordChildExecutionCompletedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RecordChildExecutionCompletedResponse proto.InternalMessageInfo\n\ntype ReplicateEventsV2Request struct {\n\tDomainId            string                    `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution   *v1.WorkflowExecution     `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tVersionHistoryItems []*v11.VersionHistoryItem `protobuf:\"bytes,3,rep,name=version_history_items,json=versionHistoryItems,proto3\" json:\"version_history_items,omitempty\"`\n\tEvents              *v1.DataBlob              `protobuf:\"bytes,4,opt,name=events,proto3\" json:\"events,omitempty\"`\n\t// New run events does not need version history since there is no prior events.\n\tNewRunEvents         *v1.DataBlob `protobuf:\"bytes,5,opt,name=new_run_events,json=newRunEvents,proto3\" json:\"new_run_events,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}     `json:\"-\"`\n\tXXX_unrecognized     []byte       `json:\"-\"`\n\tXXX_sizecache        int32        `json:\"-\"`\n}\n\nfunc (m *ReplicateEventsV2Request) Reset()         { *m = ReplicateEventsV2Request{} }\nfunc (m *ReplicateEventsV2Request) String() string { return proto.CompactTextString(m) }\nfunc (*ReplicateEventsV2Request) ProtoMessage()    {}\nfunc (*ReplicateEventsV2Request) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{44}\n}\nfunc (m *ReplicateEventsV2Request) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReplicateEventsV2Request) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReplicateEventsV2Request.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReplicateEventsV2Request) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReplicateEventsV2Request.Merge(m, src)\n}\nfunc (m *ReplicateEventsV2Request) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReplicateEventsV2Request) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReplicateEventsV2Request.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReplicateEventsV2Request proto.InternalMessageInfo\n\nfunc (m *ReplicateEventsV2Request) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *ReplicateEventsV2Request) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *ReplicateEventsV2Request) GetVersionHistoryItems() []*v11.VersionHistoryItem {\n\tif m != nil {\n\t\treturn m.VersionHistoryItems\n\t}\n\treturn nil\n}\n\nfunc (m *ReplicateEventsV2Request) GetEvents() *v1.DataBlob {\n\tif m != nil {\n\t\treturn m.Events\n\t}\n\treturn nil\n}\n\nfunc (m *ReplicateEventsV2Request) GetNewRunEvents() *v1.DataBlob {\n\tif m != nil {\n\t\treturn m.NewRunEvents\n\t}\n\treturn nil\n}\n\ntype ReplicateEventsV2Response struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ReplicateEventsV2Response) Reset()         { *m = ReplicateEventsV2Response{} }\nfunc (m *ReplicateEventsV2Response) String() string { return proto.CompactTextString(m) }\nfunc (*ReplicateEventsV2Response) ProtoMessage()    {}\nfunc (*ReplicateEventsV2Response) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{45}\n}\nfunc (m *ReplicateEventsV2Response) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReplicateEventsV2Response) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReplicateEventsV2Response.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReplicateEventsV2Response) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReplicateEventsV2Response.Merge(m, src)\n}\nfunc (m *ReplicateEventsV2Response) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReplicateEventsV2Response) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReplicateEventsV2Response.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReplicateEventsV2Response proto.InternalMessageInfo\n\ntype SyncShardStatusRequest struct {\n\tSourceCluster        string           `protobuf:\"bytes,1,opt,name=source_cluster,json=sourceCluster,proto3\" json:\"source_cluster,omitempty\"`\n\tShardId              int32            `protobuf:\"varint,2,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tTime                 *types.Timestamp `protobuf:\"bytes,3,opt,name=time,proto3\" json:\"time,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}         `json:\"-\"`\n\tXXX_unrecognized     []byte           `json:\"-\"`\n\tXXX_sizecache        int32            `json:\"-\"`\n}\n\nfunc (m *SyncShardStatusRequest) Reset()         { *m = SyncShardStatusRequest{} }\nfunc (m *SyncShardStatusRequest) String() string { return proto.CompactTextString(m) }\nfunc (*SyncShardStatusRequest) ProtoMessage()    {}\nfunc (*SyncShardStatusRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{46}\n}\nfunc (m *SyncShardStatusRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SyncShardStatusRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SyncShardStatusRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SyncShardStatusRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SyncShardStatusRequest.Merge(m, src)\n}\nfunc (m *SyncShardStatusRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SyncShardStatusRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_SyncShardStatusRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SyncShardStatusRequest proto.InternalMessageInfo\n\nfunc (m *SyncShardStatusRequest) GetSourceCluster() string {\n\tif m != nil {\n\t\treturn m.SourceCluster\n\t}\n\treturn \"\"\n}\n\nfunc (m *SyncShardStatusRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *SyncShardStatusRequest) GetTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.Time\n\t}\n\treturn nil\n}\n\ntype SyncShardStatusResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *SyncShardStatusResponse) Reset()         { *m = SyncShardStatusResponse{} }\nfunc (m *SyncShardStatusResponse) String() string { return proto.CompactTextString(m) }\nfunc (*SyncShardStatusResponse) ProtoMessage()    {}\nfunc (*SyncShardStatusResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{47}\n}\nfunc (m *SyncShardStatusResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SyncShardStatusResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SyncShardStatusResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SyncShardStatusResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SyncShardStatusResponse.Merge(m, src)\n}\nfunc (m *SyncShardStatusResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SyncShardStatusResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_SyncShardStatusResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SyncShardStatusResponse proto.InternalMessageInfo\n\ntype SyncActivityRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tVersion              int64                 `protobuf:\"varint,3,opt,name=version,proto3\" json:\"version,omitempty\"`\n\tScheduledId          int64                 `protobuf:\"varint,4,opt,name=scheduled_id,json=scheduledId,proto3\" json:\"scheduled_id,omitempty\"`\n\tScheduledTime        *types.Timestamp      `protobuf:\"bytes,5,opt,name=scheduled_time,json=scheduledTime,proto3\" json:\"scheduled_time,omitempty\"`\n\tStartedId            int64                 `protobuf:\"varint,6,opt,name=started_id,json=startedId,proto3\" json:\"started_id,omitempty\"`\n\tStartedTime          *types.Timestamp      `protobuf:\"bytes,7,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tLastHeartbeatTime    *types.Timestamp      `protobuf:\"bytes,8,opt,name=last_heartbeat_time,json=lastHeartbeatTime,proto3\" json:\"last_heartbeat_time,omitempty\"`\n\tDetails              *v1.Payload           `protobuf:\"bytes,9,opt,name=details,proto3\" json:\"details,omitempty\"`\n\tAttempt              int32                 `protobuf:\"varint,10,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tLastFailure          *v1.Failure           `protobuf:\"bytes,11,opt,name=last_failure,json=lastFailure,proto3\" json:\"last_failure,omitempty\"`\n\tLastWorkerIdentity   string                `protobuf:\"bytes,12,opt,name=last_worker_identity,json=lastWorkerIdentity,proto3\" json:\"last_worker_identity,omitempty\"`\n\tVersionHistory       *v11.VersionHistory   `protobuf:\"bytes,13,opt,name=version_history,json=versionHistory,proto3\" json:\"version_history,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *SyncActivityRequest) Reset()         { *m = SyncActivityRequest{} }\nfunc (m *SyncActivityRequest) String() string { return proto.CompactTextString(m) }\nfunc (*SyncActivityRequest) ProtoMessage()    {}\nfunc (*SyncActivityRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{48}\n}\nfunc (m *SyncActivityRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SyncActivityRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SyncActivityRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SyncActivityRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SyncActivityRequest.Merge(m, src)\n}\nfunc (m *SyncActivityRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SyncActivityRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_SyncActivityRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SyncActivityRequest proto.InternalMessageInfo\n\nfunc (m *SyncActivityRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *SyncActivityRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetVersion() int64 {\n\tif m != nil {\n\t\treturn m.Version\n\t}\n\treturn 0\n}\n\nfunc (m *SyncActivityRequest) GetScheduledId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduledId\n\t}\n\treturn 0\n}\n\nfunc (m *SyncActivityRequest) GetScheduledTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTime\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetStartedId() int64 {\n\tif m != nil {\n\t\treturn m.StartedId\n\t}\n\treturn 0\n}\n\nfunc (m *SyncActivityRequest) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetLastHeartbeatTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.LastHeartbeatTime\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetDetails() *v1.Payload {\n\tif m != nil {\n\t\treturn m.Details\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *SyncActivityRequest) GetLastFailure() *v1.Failure {\n\tif m != nil {\n\t\treturn m.LastFailure\n\t}\n\treturn nil\n}\n\nfunc (m *SyncActivityRequest) GetLastWorkerIdentity() string {\n\tif m != nil {\n\t\treturn m.LastWorkerIdentity\n\t}\n\treturn \"\"\n}\n\nfunc (m *SyncActivityRequest) GetVersionHistory() *v11.VersionHistory {\n\tif m != nil {\n\t\treturn m.VersionHistory\n\t}\n\treturn nil\n}\n\ntype SyncActivityResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *SyncActivityResponse) Reset()         { *m = SyncActivityResponse{} }\nfunc (m *SyncActivityResponse) String() string { return proto.CompactTextString(m) }\nfunc (*SyncActivityResponse) ProtoMessage()    {}\nfunc (*SyncActivityResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{49}\n}\nfunc (m *SyncActivityResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *SyncActivityResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_SyncActivityResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *SyncActivityResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_SyncActivityResponse.Merge(m, src)\n}\nfunc (m *SyncActivityResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *SyncActivityResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_SyncActivityResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_SyncActivityResponse proto.InternalMessageInfo\n\ntype DescribeMutableStateRequest struct {\n\tDomainId             string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *DescribeMutableStateRequest) Reset()         { *m = DescribeMutableStateRequest{} }\nfunc (m *DescribeMutableStateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeMutableStateRequest) ProtoMessage()    {}\nfunc (*DescribeMutableStateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{50}\n}\nfunc (m *DescribeMutableStateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeMutableStateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeMutableStateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeMutableStateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeMutableStateRequest.Merge(m, src)\n}\nfunc (m *DescribeMutableStateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeMutableStateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeMutableStateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeMutableStateRequest proto.InternalMessageInfo\n\nfunc (m *DescribeMutableStateRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *DescribeMutableStateRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\ntype DescribeMutableStateResponse struct {\n\tMutableStateInCache    string   `protobuf:\"bytes,1,opt,name=mutable_state_in_cache,json=mutableStateInCache,proto3\" json:\"mutable_state_in_cache,omitempty\"`\n\tMutableStateInDatabase string   `protobuf:\"bytes,2,opt,name=mutable_state_in_database,json=mutableStateInDatabase,proto3\" json:\"mutable_state_in_database,omitempty\"`\n\tXXX_NoUnkeyedLiteral   struct{} `json:\"-\"`\n\tXXX_unrecognized       []byte   `json:\"-\"`\n\tXXX_sizecache          int32    `json:\"-\"`\n}\n\nfunc (m *DescribeMutableStateResponse) Reset()         { *m = DescribeMutableStateResponse{} }\nfunc (m *DescribeMutableStateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeMutableStateResponse) ProtoMessage()    {}\nfunc (*DescribeMutableStateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{51}\n}\nfunc (m *DescribeMutableStateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeMutableStateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeMutableStateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeMutableStateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeMutableStateResponse.Merge(m, src)\n}\nfunc (m *DescribeMutableStateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeMutableStateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeMutableStateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeMutableStateResponse proto.InternalMessageInfo\n\nfunc (m *DescribeMutableStateResponse) GetMutableStateInCache() string {\n\tif m != nil {\n\t\treturn m.MutableStateInCache\n\t}\n\treturn \"\"\n}\n\nfunc (m *DescribeMutableStateResponse) GetMutableStateInDatabase() string {\n\tif m != nil {\n\t\treturn m.MutableStateInDatabase\n\t}\n\treturn \"\"\n}\n\ntype DescribeHistoryHostRequest struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *DescribeHistoryHostRequest) Reset()         { *m = DescribeHistoryHostRequest{} }\nfunc (m *DescribeHistoryHostRequest) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeHistoryHostRequest) ProtoMessage()    {}\nfunc (*DescribeHistoryHostRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{52}\n}\nfunc (m *DescribeHistoryHostRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeHistoryHostRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeHistoryHostRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeHistoryHostRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeHistoryHostRequest.Merge(m, src)\n}\nfunc (m *DescribeHistoryHostRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeHistoryHostRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeHistoryHostRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeHistoryHostRequest proto.InternalMessageInfo\n\ntype DescribeHistoryHostResponse struct {\n\tNumberOfShards        int32                `protobuf:\"varint,1,opt,name=number_of_shards,json=numberOfShards,proto3\" json:\"number_of_shards,omitempty\"`\n\tShardIds              []int32              `protobuf:\"varint,2,rep,packed,name=shard_ids,json=shardIds,proto3\" json:\"shard_ids,omitempty\"`\n\tDomainCache           *v11.DomainCacheInfo `protobuf:\"bytes,3,opt,name=domain_cache,json=domainCache,proto3\" json:\"domain_cache,omitempty\"`\n\tShardControllerStatus string               `protobuf:\"bytes,4,opt,name=shard_controller_status,json=shardControllerStatus,proto3\" json:\"shard_controller_status,omitempty\"`\n\tAddress               string               `protobuf:\"bytes,5,opt,name=address,proto3\" json:\"address,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{}             `json:\"-\"`\n\tXXX_unrecognized      []byte               `json:\"-\"`\n\tXXX_sizecache         int32                `json:\"-\"`\n}\n\nfunc (m *DescribeHistoryHostResponse) Reset()         { *m = DescribeHistoryHostResponse{} }\nfunc (m *DescribeHistoryHostResponse) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeHistoryHostResponse) ProtoMessage()    {}\nfunc (*DescribeHistoryHostResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{53}\n}\nfunc (m *DescribeHistoryHostResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeHistoryHostResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeHistoryHostResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeHistoryHostResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeHistoryHostResponse.Merge(m, src)\n}\nfunc (m *DescribeHistoryHostResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeHistoryHostResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeHistoryHostResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeHistoryHostResponse proto.InternalMessageInfo\n\nfunc (m *DescribeHistoryHostResponse) GetNumberOfShards() int32 {\n\tif m != nil {\n\t\treturn m.NumberOfShards\n\t}\n\treturn 0\n}\n\nfunc (m *DescribeHistoryHostResponse) GetShardIds() []int32 {\n\tif m != nil {\n\t\treturn m.ShardIds\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeHistoryHostResponse) GetDomainCache() *v11.DomainCacheInfo {\n\tif m != nil {\n\t\treturn m.DomainCache\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeHistoryHostResponse) GetShardControllerStatus() string {\n\tif m != nil {\n\t\treturn m.ShardControllerStatus\n\t}\n\treturn \"\"\n}\n\nfunc (m *DescribeHistoryHostResponse) GetAddress() string {\n\tif m != nil {\n\t\treturn m.Address\n\t}\n\treturn \"\"\n}\n\ntype CloseShardRequest struct {\n\tShardId              int32    `protobuf:\"varint,1,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *CloseShardRequest) Reset()         { *m = CloseShardRequest{} }\nfunc (m *CloseShardRequest) String() string { return proto.CompactTextString(m) }\nfunc (*CloseShardRequest) ProtoMessage()    {}\nfunc (*CloseShardRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{54}\n}\nfunc (m *CloseShardRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CloseShardRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CloseShardRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CloseShardRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CloseShardRequest.Merge(m, src)\n}\nfunc (m *CloseShardRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CloseShardRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_CloseShardRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CloseShardRequest proto.InternalMessageInfo\n\nfunc (m *CloseShardRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\ntype CloseShardResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *CloseShardResponse) Reset()         { *m = CloseShardResponse{} }\nfunc (m *CloseShardResponse) String() string { return proto.CompactTextString(m) }\nfunc (*CloseShardResponse) ProtoMessage()    {}\nfunc (*CloseShardResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{55}\n}\nfunc (m *CloseShardResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CloseShardResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CloseShardResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CloseShardResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CloseShardResponse.Merge(m, src)\n}\nfunc (m *CloseShardResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CloseShardResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_CloseShardResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CloseShardResponse proto.InternalMessageInfo\n\ntype RemoveTaskRequest struct {\n\tShardId              int32            `protobuf:\"varint,1,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tTaskType             v11.TaskType     `protobuf:\"varint,2,opt,name=task_type,json=taskType,proto3,enum=uber.cadence.admin.v1.TaskType\" json:\"task_type,omitempty\"`\n\tTaskId               int64            `protobuf:\"varint,3,opt,name=task_id,json=taskId,proto3\" json:\"task_id,omitempty\"`\n\tVisibilityTime       *types.Timestamp `protobuf:\"bytes,4,opt,name=visibility_time,json=visibilityTime,proto3\" json:\"visibility_time,omitempty\"`\n\tClusterName          string           `protobuf:\"bytes,5,opt,name=cluster_name,json=clusterName,proto3\" json:\"cluster_name,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}         `json:\"-\"`\n\tXXX_unrecognized     []byte           `json:\"-\"`\n\tXXX_sizecache        int32            `json:\"-\"`\n}\n\nfunc (m *RemoveTaskRequest) Reset()         { *m = RemoveTaskRequest{} }\nfunc (m *RemoveTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RemoveTaskRequest) ProtoMessage()    {}\nfunc (*RemoveTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{56}\n}\nfunc (m *RemoveTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RemoveTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RemoveTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RemoveTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RemoveTaskRequest.Merge(m, src)\n}\nfunc (m *RemoveTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RemoveTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RemoveTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RemoveTaskRequest proto.InternalMessageInfo\n\nfunc (m *RemoveTaskRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *RemoveTaskRequest) GetTaskType() v11.TaskType {\n\tif m != nil {\n\t\treturn m.TaskType\n\t}\n\treturn v11.TaskType_TASK_TYPE_INVALID\n}\n\nfunc (m *RemoveTaskRequest) GetTaskId() int64 {\n\tif m != nil {\n\t\treturn m.TaskId\n\t}\n\treturn 0\n}\n\nfunc (m *RemoveTaskRequest) GetVisibilityTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.VisibilityTime\n\t}\n\treturn nil\n}\n\nfunc (m *RemoveTaskRequest) GetClusterName() string {\n\tif m != nil {\n\t\treturn m.ClusterName\n\t}\n\treturn \"\"\n}\n\ntype RemoveTaskResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RemoveTaskResponse) Reset()         { *m = RemoveTaskResponse{} }\nfunc (m *RemoveTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RemoveTaskResponse) ProtoMessage()    {}\nfunc (*RemoveTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{57}\n}\nfunc (m *RemoveTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RemoveTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RemoveTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RemoveTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RemoveTaskResponse.Merge(m, src)\n}\nfunc (m *RemoveTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RemoveTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RemoveTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RemoveTaskResponse proto.InternalMessageInfo\n\ntype ResetQueueRequest struct {\n\tShardId              int32        `protobuf:\"varint,1,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tClusterName          string       `protobuf:\"bytes,2,opt,name=cluster_name,json=clusterName,proto3\" json:\"cluster_name,omitempty\"`\n\tTaskType             v11.TaskType `protobuf:\"varint,3,opt,name=task_type,json=taskType,proto3,enum=uber.cadence.admin.v1.TaskType\" json:\"task_type,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}     `json:\"-\"`\n\tXXX_unrecognized     []byte       `json:\"-\"`\n\tXXX_sizecache        int32        `json:\"-\"`\n}\n\nfunc (m *ResetQueueRequest) Reset()         { *m = ResetQueueRequest{} }\nfunc (m *ResetQueueRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ResetQueueRequest) ProtoMessage()    {}\nfunc (*ResetQueueRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{58}\n}\nfunc (m *ResetQueueRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetQueueRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetQueueRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetQueueRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetQueueRequest.Merge(m, src)\n}\nfunc (m *ResetQueueRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetQueueRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetQueueRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetQueueRequest proto.InternalMessageInfo\n\nfunc (m *ResetQueueRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *ResetQueueRequest) GetClusterName() string {\n\tif m != nil {\n\t\treturn m.ClusterName\n\t}\n\treturn \"\"\n}\n\nfunc (m *ResetQueueRequest) GetTaskType() v11.TaskType {\n\tif m != nil {\n\t\treturn m.TaskType\n\t}\n\treturn v11.TaskType_TASK_TYPE_INVALID\n}\n\ntype ResetQueueResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ResetQueueResponse) Reset()         { *m = ResetQueueResponse{} }\nfunc (m *ResetQueueResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ResetQueueResponse) ProtoMessage()    {}\nfunc (*ResetQueueResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{59}\n}\nfunc (m *ResetQueueResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ResetQueueResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ResetQueueResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ResetQueueResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ResetQueueResponse.Merge(m, src)\n}\nfunc (m *ResetQueueResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ResetQueueResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ResetQueueResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ResetQueueResponse proto.InternalMessageInfo\n\ntype DescribeQueueRequest struct {\n\tShardId              int32        `protobuf:\"varint,1,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tClusterName          string       `protobuf:\"bytes,2,opt,name=cluster_name,json=clusterName,proto3\" json:\"cluster_name,omitempty\"`\n\tTaskType             v11.TaskType `protobuf:\"varint,3,opt,name=task_type,json=taskType,proto3,enum=uber.cadence.admin.v1.TaskType\" json:\"task_type,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}     `json:\"-\"`\n\tXXX_unrecognized     []byte       `json:\"-\"`\n\tXXX_sizecache        int32        `json:\"-\"`\n}\n\nfunc (m *DescribeQueueRequest) Reset()         { *m = DescribeQueueRequest{} }\nfunc (m *DescribeQueueRequest) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeQueueRequest) ProtoMessage()    {}\nfunc (*DescribeQueueRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{60}\n}\nfunc (m *DescribeQueueRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeQueueRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeQueueRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeQueueRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeQueueRequest.Merge(m, src)\n}\nfunc (m *DescribeQueueRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeQueueRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeQueueRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeQueueRequest proto.InternalMessageInfo\n\nfunc (m *DescribeQueueRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *DescribeQueueRequest) GetClusterName() string {\n\tif m != nil {\n\t\treturn m.ClusterName\n\t}\n\treturn \"\"\n}\n\nfunc (m *DescribeQueueRequest) GetTaskType() v11.TaskType {\n\tif m != nil {\n\t\treturn m.TaskType\n\t}\n\treturn v11.TaskType_TASK_TYPE_INVALID\n}\n\ntype DescribeQueueResponse struct {\n\tProcessingQueueStates []string `protobuf:\"bytes,1,rep,name=processing_queue_states,json=processingQueueStates,proto3\" json:\"processing_queue_states,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{} `json:\"-\"`\n\tXXX_unrecognized      []byte   `json:\"-\"`\n\tXXX_sizecache         int32    `json:\"-\"`\n}\n\nfunc (m *DescribeQueueResponse) Reset()         { *m = DescribeQueueResponse{} }\nfunc (m *DescribeQueueResponse) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeQueueResponse) ProtoMessage()    {}\nfunc (*DescribeQueueResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{61}\n}\nfunc (m *DescribeQueueResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeQueueResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeQueueResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeQueueResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeQueueResponse.Merge(m, src)\n}\nfunc (m *DescribeQueueResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeQueueResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeQueueResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeQueueResponse proto.InternalMessageInfo\n\nfunc (m *DescribeQueueResponse) GetProcessingQueueStates() []string {\n\tif m != nil {\n\t\treturn m.ProcessingQueueStates\n\t}\n\treturn nil\n}\n\ntype GetReplicationMessagesRequest struct {\n\tTokens               []*v11.ReplicationToken `protobuf:\"bytes,1,rep,name=tokens,proto3\" json:\"tokens,omitempty\"`\n\tClusterName          string                  `protobuf:\"bytes,2,opt,name=cluster_name,json=clusterName,proto3\" json:\"cluster_name,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                `json:\"-\"`\n\tXXX_unrecognized     []byte                  `json:\"-\"`\n\tXXX_sizecache        int32                   `json:\"-\"`\n}\n\nfunc (m *GetReplicationMessagesRequest) Reset()         { *m = GetReplicationMessagesRequest{} }\nfunc (m *GetReplicationMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetReplicationMessagesRequest) ProtoMessage()    {}\nfunc (*GetReplicationMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{62}\n}\nfunc (m *GetReplicationMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetReplicationMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetReplicationMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetReplicationMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetReplicationMessagesRequest.Merge(m, src)\n}\nfunc (m *GetReplicationMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetReplicationMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetReplicationMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetReplicationMessagesRequest proto.InternalMessageInfo\n\nfunc (m *GetReplicationMessagesRequest) GetTokens() []*v11.ReplicationToken {\n\tif m != nil {\n\t\treturn m.Tokens\n\t}\n\treturn nil\n}\n\nfunc (m *GetReplicationMessagesRequest) GetClusterName() string {\n\tif m != nil {\n\t\treturn m.ClusterName\n\t}\n\treturn \"\"\n}\n\ntype GetReplicationMessagesResponse struct {\n\tShardMessages        map[int32]*v11.ReplicationMessages `protobuf:\"bytes,1,rep,name=shard_messages,json=shardMessages,proto3\" json:\"shard_messages,omitempty\" protobuf_key:\"varint,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}                           `json:\"-\"`\n\tXXX_unrecognized     []byte                             `json:\"-\"`\n\tXXX_sizecache        int32                              `json:\"-\"`\n}\n\nfunc (m *GetReplicationMessagesResponse) Reset()         { *m = GetReplicationMessagesResponse{} }\nfunc (m *GetReplicationMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetReplicationMessagesResponse) ProtoMessage()    {}\nfunc (*GetReplicationMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{63}\n}\nfunc (m *GetReplicationMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetReplicationMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetReplicationMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetReplicationMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetReplicationMessagesResponse.Merge(m, src)\n}\nfunc (m *GetReplicationMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetReplicationMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetReplicationMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetReplicationMessagesResponse proto.InternalMessageInfo\n\nfunc (m *GetReplicationMessagesResponse) GetShardMessages() map[int32]*v11.ReplicationMessages {\n\tif m != nil {\n\t\treturn m.ShardMessages\n\t}\n\treturn nil\n}\n\ntype GetDLQReplicationMessagesRequest struct {\n\tTaskInfos            []*v11.ReplicationTaskInfo `protobuf:\"bytes,1,rep,name=task_infos,json=taskInfos,proto3\" json:\"task_infos,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                   `json:\"-\"`\n\tXXX_unrecognized     []byte                     `json:\"-\"`\n\tXXX_sizecache        int32                      `json:\"-\"`\n}\n\nfunc (m *GetDLQReplicationMessagesRequest) Reset()         { *m = GetDLQReplicationMessagesRequest{} }\nfunc (m *GetDLQReplicationMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetDLQReplicationMessagesRequest) ProtoMessage()    {}\nfunc (*GetDLQReplicationMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{64}\n}\nfunc (m *GetDLQReplicationMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetDLQReplicationMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetDLQReplicationMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetDLQReplicationMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetDLQReplicationMessagesRequest.Merge(m, src)\n}\nfunc (m *GetDLQReplicationMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetDLQReplicationMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetDLQReplicationMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetDLQReplicationMessagesRequest proto.InternalMessageInfo\n\nfunc (m *GetDLQReplicationMessagesRequest) GetTaskInfos() []*v11.ReplicationTaskInfo {\n\tif m != nil {\n\t\treturn m.TaskInfos\n\t}\n\treturn nil\n}\n\ntype GetDLQReplicationMessagesResponse struct {\n\tReplicationTasks     []*v11.ReplicationTask `protobuf:\"bytes,1,rep,name=replication_tasks,json=replicationTasks,proto3\" json:\"replication_tasks,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}               `json:\"-\"`\n\tXXX_unrecognized     []byte                 `json:\"-\"`\n\tXXX_sizecache        int32                  `json:\"-\"`\n}\n\nfunc (m *GetDLQReplicationMessagesResponse) Reset()         { *m = GetDLQReplicationMessagesResponse{} }\nfunc (m *GetDLQReplicationMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetDLQReplicationMessagesResponse) ProtoMessage()    {}\nfunc (*GetDLQReplicationMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{65}\n}\nfunc (m *GetDLQReplicationMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetDLQReplicationMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetDLQReplicationMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetDLQReplicationMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetDLQReplicationMessagesResponse.Merge(m, src)\n}\nfunc (m *GetDLQReplicationMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetDLQReplicationMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetDLQReplicationMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetDLQReplicationMessagesResponse proto.InternalMessageInfo\n\nfunc (m *GetDLQReplicationMessagesResponse) GetReplicationTasks() []*v11.ReplicationTask {\n\tif m != nil {\n\t\treturn m.ReplicationTasks\n\t}\n\treturn nil\n}\n\ntype ReapplyEventsRequest struct {\n\tDomain               string                `protobuf:\"bytes,1,opt,name=domain,proto3\" json:\"domain,omitempty\"`\n\tDomainId             string                `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,3,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tEvents               *v1.DataBlob          `protobuf:\"bytes,4,opt,name=events,proto3\" json:\"events,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *ReapplyEventsRequest) Reset()         { *m = ReapplyEventsRequest{} }\nfunc (m *ReapplyEventsRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ReapplyEventsRequest) ProtoMessage()    {}\nfunc (*ReapplyEventsRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{66}\n}\nfunc (m *ReapplyEventsRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReapplyEventsRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReapplyEventsRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReapplyEventsRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReapplyEventsRequest.Merge(m, src)\n}\nfunc (m *ReapplyEventsRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReapplyEventsRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReapplyEventsRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReapplyEventsRequest proto.InternalMessageInfo\n\nfunc (m *ReapplyEventsRequest) GetDomain() string {\n\tif m != nil {\n\t\treturn m.Domain\n\t}\n\treturn \"\"\n}\n\nfunc (m *ReapplyEventsRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *ReapplyEventsRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *ReapplyEventsRequest) GetEvents() *v1.DataBlob {\n\tif m != nil {\n\t\treturn m.Events\n\t}\n\treturn nil\n}\n\ntype ReapplyEventsResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ReapplyEventsResponse) Reset()         { *m = ReapplyEventsResponse{} }\nfunc (m *ReapplyEventsResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ReapplyEventsResponse) ProtoMessage()    {}\nfunc (*ReapplyEventsResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{67}\n}\nfunc (m *ReapplyEventsResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReapplyEventsResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReapplyEventsResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReapplyEventsResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReapplyEventsResponse.Merge(m, src)\n}\nfunc (m *ReapplyEventsResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReapplyEventsResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReapplyEventsResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReapplyEventsResponse proto.InternalMessageInfo\n\ntype RefreshWorkflowTasksRequest struct {\n\tDomain               string                `protobuf:\"bytes,1,opt,name=domain,proto3\" json:\"domain,omitempty\"`\n\tDomainId             string                `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,3,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *RefreshWorkflowTasksRequest) Reset()         { *m = RefreshWorkflowTasksRequest{} }\nfunc (m *RefreshWorkflowTasksRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RefreshWorkflowTasksRequest) ProtoMessage()    {}\nfunc (*RefreshWorkflowTasksRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{68}\n}\nfunc (m *RefreshWorkflowTasksRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RefreshWorkflowTasksRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RefreshWorkflowTasksRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RefreshWorkflowTasksRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RefreshWorkflowTasksRequest.Merge(m, src)\n}\nfunc (m *RefreshWorkflowTasksRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RefreshWorkflowTasksRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RefreshWorkflowTasksRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RefreshWorkflowTasksRequest proto.InternalMessageInfo\n\nfunc (m *RefreshWorkflowTasksRequest) GetDomain() string {\n\tif m != nil {\n\t\treturn m.Domain\n\t}\n\treturn \"\"\n}\n\nfunc (m *RefreshWorkflowTasksRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RefreshWorkflowTasksRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\ntype RefreshWorkflowTasksResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RefreshWorkflowTasksResponse) Reset()         { *m = RefreshWorkflowTasksResponse{} }\nfunc (m *RefreshWorkflowTasksResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RefreshWorkflowTasksResponse) ProtoMessage()    {}\nfunc (*RefreshWorkflowTasksResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{69}\n}\nfunc (m *RefreshWorkflowTasksResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RefreshWorkflowTasksResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RefreshWorkflowTasksResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RefreshWorkflowTasksResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RefreshWorkflowTasksResponse.Merge(m, src)\n}\nfunc (m *RefreshWorkflowTasksResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RefreshWorkflowTasksResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RefreshWorkflowTasksResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RefreshWorkflowTasksResponse proto.InternalMessageInfo\n\ntype CountDLQMessagesRequest struct {\n\tForceFetch           bool     `protobuf:\"varint,1,opt,name=forceFetch,proto3\" json:\"forceFetch,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *CountDLQMessagesRequest) Reset()         { *m = CountDLQMessagesRequest{} }\nfunc (m *CountDLQMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*CountDLQMessagesRequest) ProtoMessage()    {}\nfunc (*CountDLQMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{70}\n}\nfunc (m *CountDLQMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CountDLQMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CountDLQMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CountDLQMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CountDLQMessagesRequest.Merge(m, src)\n}\nfunc (m *CountDLQMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CountDLQMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_CountDLQMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CountDLQMessagesRequest proto.InternalMessageInfo\n\nfunc (m *CountDLQMessagesRequest) GetForceFetch() bool {\n\tif m != nil {\n\t\treturn m.ForceFetch\n\t}\n\treturn false\n}\n\ntype CountDLQMessagesResponse struct {\n\tEntries              []*v11.HistoryDLQCountEntry `protobuf:\"bytes,1,rep,name=entries,proto3\" json:\"entries,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *CountDLQMessagesResponse) Reset()         { *m = CountDLQMessagesResponse{} }\nfunc (m *CountDLQMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*CountDLQMessagesResponse) ProtoMessage()    {}\nfunc (*CountDLQMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{71}\n}\nfunc (m *CountDLQMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CountDLQMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CountDLQMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CountDLQMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CountDLQMessagesResponse.Merge(m, src)\n}\nfunc (m *CountDLQMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CountDLQMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_CountDLQMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CountDLQMessagesResponse proto.InternalMessageInfo\n\nfunc (m *CountDLQMessagesResponse) GetEntries() []*v11.HistoryDLQCountEntry {\n\tif m != nil {\n\t\treturn m.Entries\n\t}\n\treturn nil\n}\n\ntype ReadDLQMessagesRequest struct {\n\tType                  v11.DLQType       `protobuf:\"varint,1,opt,name=type,proto3,enum=uber.cadence.admin.v1.DLQType\" json:\"type,omitempty\"`\n\tShardId               int32             `protobuf:\"varint,2,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tSourceCluster         string            `protobuf:\"bytes,3,opt,name=source_cluster,json=sourceCluster,proto3\" json:\"source_cluster,omitempty\"`\n\tInclusiveEndMessageId *types.Int64Value `protobuf:\"bytes,4,opt,name=inclusive_end_message_id,json=inclusiveEndMessageId,proto3\" json:\"inclusive_end_message_id,omitempty\"`\n\tPageSize              int32             `protobuf:\"varint,5,opt,name=page_size,json=pageSize,proto3\" json:\"page_size,omitempty\"`\n\tNextPageToken         []byte            `protobuf:\"bytes,6,opt,name=next_page_token,json=nextPageToken,proto3\" json:\"next_page_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{}          `json:\"-\"`\n\tXXX_unrecognized      []byte            `json:\"-\"`\n\tXXX_sizecache         int32             `json:\"-\"`\n}\n\nfunc (m *ReadDLQMessagesRequest) Reset()         { *m = ReadDLQMessagesRequest{} }\nfunc (m *ReadDLQMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ReadDLQMessagesRequest) ProtoMessage()    {}\nfunc (*ReadDLQMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{72}\n}\nfunc (m *ReadDLQMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReadDLQMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReadDLQMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReadDLQMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReadDLQMessagesRequest.Merge(m, src)\n}\nfunc (m *ReadDLQMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReadDLQMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReadDLQMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReadDLQMessagesRequest proto.InternalMessageInfo\n\nfunc (m *ReadDLQMessagesRequest) GetType() v11.DLQType {\n\tif m != nil {\n\t\treturn m.Type\n\t}\n\treturn v11.DLQType_DLQ_TYPE_INVALID\n}\n\nfunc (m *ReadDLQMessagesRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *ReadDLQMessagesRequest) GetSourceCluster() string {\n\tif m != nil {\n\t\treturn m.SourceCluster\n\t}\n\treturn \"\"\n}\n\nfunc (m *ReadDLQMessagesRequest) GetInclusiveEndMessageId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.InclusiveEndMessageId\n\t}\n\treturn nil\n}\n\nfunc (m *ReadDLQMessagesRequest) GetPageSize() int32 {\n\tif m != nil {\n\t\treturn m.PageSize\n\t}\n\treturn 0\n}\n\nfunc (m *ReadDLQMessagesRequest) GetNextPageToken() []byte {\n\tif m != nil {\n\t\treturn m.NextPageToken\n\t}\n\treturn nil\n}\n\ntype ReadDLQMessagesResponse struct {\n\tType                 v11.DLQType                `protobuf:\"varint,1,opt,name=type,proto3,enum=uber.cadence.admin.v1.DLQType\" json:\"type,omitempty\"`\n\tReplicationTasks     []*v11.ReplicationTask     `protobuf:\"bytes,2,rep,name=replication_tasks,json=replicationTasks,proto3\" json:\"replication_tasks,omitempty\"`\n\tReplicationTasksInfo []*v11.ReplicationTaskInfo `protobuf:\"bytes,3,rep,name=replication_tasks_info,json=replicationTasksInfo,proto3\" json:\"replication_tasks_info,omitempty\"`\n\tNextPageToken        []byte                     `protobuf:\"bytes,4,opt,name=next_page_token,json=nextPageToken,proto3\" json:\"next_page_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                   `json:\"-\"`\n\tXXX_unrecognized     []byte                     `json:\"-\"`\n\tXXX_sizecache        int32                      `json:\"-\"`\n}\n\nfunc (m *ReadDLQMessagesResponse) Reset()         { *m = ReadDLQMessagesResponse{} }\nfunc (m *ReadDLQMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ReadDLQMessagesResponse) ProtoMessage()    {}\nfunc (*ReadDLQMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{73}\n}\nfunc (m *ReadDLQMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ReadDLQMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ReadDLQMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ReadDLQMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ReadDLQMessagesResponse.Merge(m, src)\n}\nfunc (m *ReadDLQMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ReadDLQMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ReadDLQMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ReadDLQMessagesResponse proto.InternalMessageInfo\n\nfunc (m *ReadDLQMessagesResponse) GetType() v11.DLQType {\n\tif m != nil {\n\t\treturn m.Type\n\t}\n\treturn v11.DLQType_DLQ_TYPE_INVALID\n}\n\nfunc (m *ReadDLQMessagesResponse) GetReplicationTasks() []*v11.ReplicationTask {\n\tif m != nil {\n\t\treturn m.ReplicationTasks\n\t}\n\treturn nil\n}\n\nfunc (m *ReadDLQMessagesResponse) GetReplicationTasksInfo() []*v11.ReplicationTaskInfo {\n\tif m != nil {\n\t\treturn m.ReplicationTasksInfo\n\t}\n\treturn nil\n}\n\nfunc (m *ReadDLQMessagesResponse) GetNextPageToken() []byte {\n\tif m != nil {\n\t\treturn m.NextPageToken\n\t}\n\treturn nil\n}\n\ntype PurgeDLQMessagesRequest struct {\n\tType                  v11.DLQType       `protobuf:\"varint,1,opt,name=type,proto3,enum=uber.cadence.admin.v1.DLQType\" json:\"type,omitempty\"`\n\tShardId               int32             `protobuf:\"varint,2,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tSourceCluster         string            `protobuf:\"bytes,3,opt,name=source_cluster,json=sourceCluster,proto3\" json:\"source_cluster,omitempty\"`\n\tInclusiveEndMessageId *types.Int64Value `protobuf:\"bytes,4,opt,name=inclusive_end_message_id,json=inclusiveEndMessageId,proto3\" json:\"inclusive_end_message_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{}          `json:\"-\"`\n\tXXX_unrecognized      []byte            `json:\"-\"`\n\tXXX_sizecache         int32             `json:\"-\"`\n}\n\nfunc (m *PurgeDLQMessagesRequest) Reset()         { *m = PurgeDLQMessagesRequest{} }\nfunc (m *PurgeDLQMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*PurgeDLQMessagesRequest) ProtoMessage()    {}\nfunc (*PurgeDLQMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{74}\n}\nfunc (m *PurgeDLQMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PurgeDLQMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PurgeDLQMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PurgeDLQMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PurgeDLQMessagesRequest.Merge(m, src)\n}\nfunc (m *PurgeDLQMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PurgeDLQMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_PurgeDLQMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PurgeDLQMessagesRequest proto.InternalMessageInfo\n\nfunc (m *PurgeDLQMessagesRequest) GetType() v11.DLQType {\n\tif m != nil {\n\t\treturn m.Type\n\t}\n\treturn v11.DLQType_DLQ_TYPE_INVALID\n}\n\nfunc (m *PurgeDLQMessagesRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *PurgeDLQMessagesRequest) GetSourceCluster() string {\n\tif m != nil {\n\t\treturn m.SourceCluster\n\t}\n\treturn \"\"\n}\n\nfunc (m *PurgeDLQMessagesRequest) GetInclusiveEndMessageId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.InclusiveEndMessageId\n\t}\n\treturn nil\n}\n\ntype PurgeDLQMessagesResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *PurgeDLQMessagesResponse) Reset()         { *m = PurgeDLQMessagesResponse{} }\nfunc (m *PurgeDLQMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*PurgeDLQMessagesResponse) ProtoMessage()    {}\nfunc (*PurgeDLQMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{75}\n}\nfunc (m *PurgeDLQMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PurgeDLQMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PurgeDLQMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PurgeDLQMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PurgeDLQMessagesResponse.Merge(m, src)\n}\nfunc (m *PurgeDLQMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PurgeDLQMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_PurgeDLQMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PurgeDLQMessagesResponse proto.InternalMessageInfo\n\ntype MergeDLQMessagesRequest struct {\n\tType                  v11.DLQType       `protobuf:\"varint,1,opt,name=type,proto3,enum=uber.cadence.admin.v1.DLQType\" json:\"type,omitempty\"`\n\tShardId               int32             `protobuf:\"varint,2,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tSourceCluster         string            `protobuf:\"bytes,3,opt,name=source_cluster,json=sourceCluster,proto3\" json:\"source_cluster,omitempty\"`\n\tInclusiveEndMessageId *types.Int64Value `protobuf:\"bytes,4,opt,name=inclusive_end_message_id,json=inclusiveEndMessageId,proto3\" json:\"inclusive_end_message_id,omitempty\"`\n\tPageSize              int32             `protobuf:\"varint,5,opt,name=page_size,json=pageSize,proto3\" json:\"page_size,omitempty\"`\n\tNextPageToken         []byte            `protobuf:\"bytes,6,opt,name=next_page_token,json=nextPageToken,proto3\" json:\"next_page_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral  struct{}          `json:\"-\"`\n\tXXX_unrecognized      []byte            `json:\"-\"`\n\tXXX_sizecache         int32             `json:\"-\"`\n}\n\nfunc (m *MergeDLQMessagesRequest) Reset()         { *m = MergeDLQMessagesRequest{} }\nfunc (m *MergeDLQMessagesRequest) String() string { return proto.CompactTextString(m) }\nfunc (*MergeDLQMessagesRequest) ProtoMessage()    {}\nfunc (*MergeDLQMessagesRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{76}\n}\nfunc (m *MergeDLQMessagesRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *MergeDLQMessagesRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_MergeDLQMessagesRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *MergeDLQMessagesRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_MergeDLQMessagesRequest.Merge(m, src)\n}\nfunc (m *MergeDLQMessagesRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *MergeDLQMessagesRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_MergeDLQMessagesRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_MergeDLQMessagesRequest proto.InternalMessageInfo\n\nfunc (m *MergeDLQMessagesRequest) GetType() v11.DLQType {\n\tif m != nil {\n\t\treturn m.Type\n\t}\n\treturn v11.DLQType_DLQ_TYPE_INVALID\n}\n\nfunc (m *MergeDLQMessagesRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *MergeDLQMessagesRequest) GetSourceCluster() string {\n\tif m != nil {\n\t\treturn m.SourceCluster\n\t}\n\treturn \"\"\n}\n\nfunc (m *MergeDLQMessagesRequest) GetInclusiveEndMessageId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.InclusiveEndMessageId\n\t}\n\treturn nil\n}\n\nfunc (m *MergeDLQMessagesRequest) GetPageSize() int32 {\n\tif m != nil {\n\t\treturn m.PageSize\n\t}\n\treturn 0\n}\n\nfunc (m *MergeDLQMessagesRequest) GetNextPageToken() []byte {\n\tif m != nil {\n\t\treturn m.NextPageToken\n\t}\n\treturn nil\n}\n\ntype MergeDLQMessagesResponse struct {\n\tNextPageToken        []byte   `protobuf:\"bytes,1,opt,name=next_page_token,json=nextPageToken,proto3\" json:\"next_page_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *MergeDLQMessagesResponse) Reset()         { *m = MergeDLQMessagesResponse{} }\nfunc (m *MergeDLQMessagesResponse) String() string { return proto.CompactTextString(m) }\nfunc (*MergeDLQMessagesResponse) ProtoMessage()    {}\nfunc (*MergeDLQMessagesResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{77}\n}\nfunc (m *MergeDLQMessagesResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *MergeDLQMessagesResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_MergeDLQMessagesResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *MergeDLQMessagesResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_MergeDLQMessagesResponse.Merge(m, src)\n}\nfunc (m *MergeDLQMessagesResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *MergeDLQMessagesResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_MergeDLQMessagesResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_MergeDLQMessagesResponse proto.InternalMessageInfo\n\nfunc (m *MergeDLQMessagesResponse) GetNextPageToken() []byte {\n\tif m != nil {\n\t\treturn m.NextPageToken\n\t}\n\treturn nil\n}\n\ntype NotifyFailoverMarkersRequest struct {\n\tFailoverMarkerTokens []*v11.FailoverMarkerToken `protobuf:\"bytes,1,rep,name=failover_marker_tokens,json=failoverMarkerTokens,proto3\" json:\"failover_marker_tokens,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                   `json:\"-\"`\n\tXXX_unrecognized     []byte                     `json:\"-\"`\n\tXXX_sizecache        int32                      `json:\"-\"`\n}\n\nfunc (m *NotifyFailoverMarkersRequest) Reset()         { *m = NotifyFailoverMarkersRequest{} }\nfunc (m *NotifyFailoverMarkersRequest) String() string { return proto.CompactTextString(m) }\nfunc (*NotifyFailoverMarkersRequest) ProtoMessage()    {}\nfunc (*NotifyFailoverMarkersRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{78}\n}\nfunc (m *NotifyFailoverMarkersRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *NotifyFailoverMarkersRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_NotifyFailoverMarkersRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *NotifyFailoverMarkersRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_NotifyFailoverMarkersRequest.Merge(m, src)\n}\nfunc (m *NotifyFailoverMarkersRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *NotifyFailoverMarkersRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_NotifyFailoverMarkersRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_NotifyFailoverMarkersRequest proto.InternalMessageInfo\n\nfunc (m *NotifyFailoverMarkersRequest) GetFailoverMarkerTokens() []*v11.FailoverMarkerToken {\n\tif m != nil {\n\t\treturn m.FailoverMarkerTokens\n\t}\n\treturn nil\n}\n\ntype NotifyFailoverMarkersResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *NotifyFailoverMarkersResponse) Reset()         { *m = NotifyFailoverMarkersResponse{} }\nfunc (m *NotifyFailoverMarkersResponse) String() string { return proto.CompactTextString(m) }\nfunc (*NotifyFailoverMarkersResponse) ProtoMessage()    {}\nfunc (*NotifyFailoverMarkersResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{79}\n}\nfunc (m *NotifyFailoverMarkersResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *NotifyFailoverMarkersResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_NotifyFailoverMarkersResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *NotifyFailoverMarkersResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_NotifyFailoverMarkersResponse.Merge(m, src)\n}\nfunc (m *NotifyFailoverMarkersResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *NotifyFailoverMarkersResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_NotifyFailoverMarkersResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_NotifyFailoverMarkersResponse proto.InternalMessageInfo\n\ntype GetCrossClusterTasksRequest struct {\n\tShardIds             []int32  `protobuf:\"varint,1,rep,packed,name=shard_ids,json=shardIds,proto3\" json:\"shard_ids,omitempty\"`\n\tTargetCluster        string   `protobuf:\"bytes,2,opt,name=target_cluster,json=targetCluster,proto3\" json:\"target_cluster,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *GetCrossClusterTasksRequest) Reset()         { *m = GetCrossClusterTasksRequest{} }\nfunc (m *GetCrossClusterTasksRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetCrossClusterTasksRequest) ProtoMessage()    {}\nfunc (*GetCrossClusterTasksRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{80}\n}\nfunc (m *GetCrossClusterTasksRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetCrossClusterTasksRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetCrossClusterTasksRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetCrossClusterTasksRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetCrossClusterTasksRequest.Merge(m, src)\n}\nfunc (m *GetCrossClusterTasksRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetCrossClusterTasksRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetCrossClusterTasksRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetCrossClusterTasksRequest proto.InternalMessageInfo\n\nfunc (m *GetCrossClusterTasksRequest) GetShardIds() []int32 {\n\tif m != nil {\n\t\treturn m.ShardIds\n\t}\n\treturn nil\n}\n\nfunc (m *GetCrossClusterTasksRequest) GetTargetCluster() string {\n\tif m != nil {\n\t\treturn m.TargetCluster\n\t}\n\treturn \"\"\n}\n\ntype GetCrossClusterTasksResponse struct {\n\tTasksByShard         map[int32]*v11.CrossClusterTaskRequests `protobuf:\"bytes,1,rep,name=tasks_by_shard,json=tasksByShard,proto3\" json:\"tasks_by_shard,omitempty\" protobuf_key:\"varint,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tFailedCauseByShard   map[int32]v11.GetTaskFailedCause        `protobuf:\"bytes,2,rep,name=failed_cause_by_shard,json=failedCauseByShard,proto3\" json:\"failed_cause_by_shard,omitempty\" protobuf_key:\"varint,1,opt,name=key,proto3\" protobuf_val:\"varint,2,opt,name=value,proto3,enum=uber.cadence.admin.v1.GetTaskFailedCause\"`\n\tXXX_NoUnkeyedLiteral struct{}                                `json:\"-\"`\n\tXXX_unrecognized     []byte                                  `json:\"-\"`\n\tXXX_sizecache        int32                                   `json:\"-\"`\n}\n\nfunc (m *GetCrossClusterTasksResponse) Reset()         { *m = GetCrossClusterTasksResponse{} }\nfunc (m *GetCrossClusterTasksResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetCrossClusterTasksResponse) ProtoMessage()    {}\nfunc (*GetCrossClusterTasksResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{81}\n}\nfunc (m *GetCrossClusterTasksResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetCrossClusterTasksResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetCrossClusterTasksResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetCrossClusterTasksResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetCrossClusterTasksResponse.Merge(m, src)\n}\nfunc (m *GetCrossClusterTasksResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetCrossClusterTasksResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetCrossClusterTasksResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetCrossClusterTasksResponse proto.InternalMessageInfo\n\nfunc (m *GetCrossClusterTasksResponse) GetTasksByShard() map[int32]*v11.CrossClusterTaskRequests {\n\tif m != nil {\n\t\treturn m.TasksByShard\n\t}\n\treturn nil\n}\n\nfunc (m *GetCrossClusterTasksResponse) GetFailedCauseByShard() map[int32]v11.GetTaskFailedCause {\n\tif m != nil {\n\t\treturn m.FailedCauseByShard\n\t}\n\treturn nil\n}\n\ntype RespondCrossClusterTasksCompletedRequest struct {\n\tShardId              int32                           `protobuf:\"varint,1,opt,name=shard_id,json=shardId,proto3\" json:\"shard_id,omitempty\"`\n\tTargetCluster        string                          `protobuf:\"bytes,2,opt,name=target_cluster,json=targetCluster,proto3\" json:\"target_cluster,omitempty\"`\n\tTaskResponses        []*v11.CrossClusterTaskResponse `protobuf:\"bytes,3,rep,name=task_responses,json=taskResponses,proto3\" json:\"task_responses,omitempty\"`\n\tFetchNewTasks        bool                            `protobuf:\"varint,4,opt,name=fetchNewTasks,proto3\" json:\"fetchNewTasks,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                        `json:\"-\"`\n\tXXX_unrecognized     []byte                          `json:\"-\"`\n\tXXX_sizecache        int32                           `json:\"-\"`\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) Reset() {\n\t*m = RespondCrossClusterTasksCompletedRequest{}\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondCrossClusterTasksCompletedRequest) ProtoMessage()    {}\nfunc (*RespondCrossClusterTasksCompletedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{82}\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondCrossClusterTasksCompletedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondCrossClusterTasksCompletedRequest.Merge(m, src)\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondCrossClusterTasksCompletedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondCrossClusterTasksCompletedRequest proto.InternalMessageInfo\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) GetShardId() int32 {\n\tif m != nil {\n\t\treturn m.ShardId\n\t}\n\treturn 0\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) GetTargetCluster() string {\n\tif m != nil {\n\t\treturn m.TargetCluster\n\t}\n\treturn \"\"\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) GetTaskResponses() []*v11.CrossClusterTaskResponse {\n\tif m != nil {\n\t\treturn m.TaskResponses\n\t}\n\treturn nil\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) GetFetchNewTasks() bool {\n\tif m != nil {\n\t\treturn m.FetchNewTasks\n\t}\n\treturn false\n}\n\ntype RespondCrossClusterTasksCompletedResponse struct {\n\tTasks                *v11.CrossClusterTaskRequests `protobuf:\"bytes,1,opt,name=tasks,proto3\" json:\"tasks,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                      `json:\"-\"`\n\tXXX_unrecognized     []byte                        `json:\"-\"`\n\tXXX_sizecache        int32                         `json:\"-\"`\n}\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) Reset() {\n\t*m = RespondCrossClusterTasksCompletedResponse{}\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) String() string {\n\treturn proto.CompactTextString(m)\n}\nfunc (*RespondCrossClusterTasksCompletedResponse) ProtoMessage() {}\nfunc (*RespondCrossClusterTasksCompletedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{83}\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondCrossClusterTasksCompletedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondCrossClusterTasksCompletedResponse.Merge(m, src)\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondCrossClusterTasksCompletedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondCrossClusterTasksCompletedResponse proto.InternalMessageInfo\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) GetTasks() *v11.CrossClusterTaskRequests {\n\tif m != nil {\n\t\treturn m.Tasks\n\t}\n\treturn nil\n}\n\ntype GetFailoverInfoRequest struct {\n\tDomainId             string   `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *GetFailoverInfoRequest) Reset()         { *m = GetFailoverInfoRequest{} }\nfunc (m *GetFailoverInfoRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetFailoverInfoRequest) ProtoMessage()    {}\nfunc (*GetFailoverInfoRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{84}\n}\nfunc (m *GetFailoverInfoRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetFailoverInfoRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetFailoverInfoRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetFailoverInfoRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetFailoverInfoRequest.Merge(m, src)\n}\nfunc (m *GetFailoverInfoRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetFailoverInfoRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetFailoverInfoRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetFailoverInfoRequest proto.InternalMessageInfo\n\nfunc (m *GetFailoverInfoRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype GetFailoverInfoResponse struct {\n\tCompletedShardCount  int32    `protobuf:\"varint,1,opt,name=completed_shard_count,json=completedShardCount,proto3\" json:\"completed_shard_count,omitempty\"`\n\tPendingShards        []int32  `protobuf:\"varint,2,rep,packed,name=pending_shards,json=pendingShards,proto3\" json:\"pending_shards,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *GetFailoverInfoResponse) Reset()         { *m = GetFailoverInfoResponse{} }\nfunc (m *GetFailoverInfoResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetFailoverInfoResponse) ProtoMessage()    {}\nfunc (*GetFailoverInfoResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{85}\n}\nfunc (m *GetFailoverInfoResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetFailoverInfoResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetFailoverInfoResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetFailoverInfoResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetFailoverInfoResponse.Merge(m, src)\n}\nfunc (m *GetFailoverInfoResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetFailoverInfoResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetFailoverInfoResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetFailoverInfoResponse proto.InternalMessageInfo\n\nfunc (m *GetFailoverInfoResponse) GetCompletedShardCount() int32 {\n\tif m != nil {\n\t\treturn m.CompletedShardCount\n\t}\n\treturn 0\n}\n\nfunc (m *GetFailoverInfoResponse) GetPendingShards() []int32 {\n\tif m != nil {\n\t\treturn m.PendingShards\n\t}\n\treturn nil\n}\n\ntype RatelimitUpdateRequest struct {\n\t// impl-specific data.\n\t// likely some simple top-level keys and then either:\n\t// - map<ratelimit-key-string, something>\n\t// - list<something>\n\t//\n\t// this is a single blob rather than a collection to save on\n\t// repeated serialization of the type name, and to allow impls\n\t// to choose whatever structures are most-convenient for them.\n\tData                 *v12.Any `protobuf:\"bytes,1,opt,name=data,proto3\" json:\"data,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RatelimitUpdateRequest) Reset()         { *m = RatelimitUpdateRequest{} }\nfunc (m *RatelimitUpdateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RatelimitUpdateRequest) ProtoMessage()    {}\nfunc (*RatelimitUpdateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{86}\n}\nfunc (m *RatelimitUpdateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RatelimitUpdateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RatelimitUpdateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RatelimitUpdateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RatelimitUpdateRequest.Merge(m, src)\n}\nfunc (m *RatelimitUpdateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RatelimitUpdateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RatelimitUpdateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RatelimitUpdateRequest proto.InternalMessageInfo\n\nfunc (m *RatelimitUpdateRequest) GetData() *v12.Any {\n\tif m != nil {\n\t\treturn m.Data\n\t}\n\treturn nil\n}\n\ntype RatelimitUpdateResponse struct {\n\t// impl-specific data.\n\t//\n\t// likely some simple top-level keys and then either:\n\t// - map<ratelimit-key-string, something>\n\t// - list<something>\n\t//\n\t// this is a single blob rather than a collection to save on\n\t// repeated serialization of the type name, and to allow impls\n\t// to choose whatever structures are most-convenient for them.\n\tData                 *v12.Any `protobuf:\"bytes,1,opt,name=data,proto3\" json:\"data,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RatelimitUpdateResponse) Reset()         { *m = RatelimitUpdateResponse{} }\nfunc (m *RatelimitUpdateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RatelimitUpdateResponse) ProtoMessage()    {}\nfunc (*RatelimitUpdateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_fee8ff76963a38ed, []int{87}\n}\nfunc (m *RatelimitUpdateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RatelimitUpdateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RatelimitUpdateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RatelimitUpdateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RatelimitUpdateResponse.Merge(m, src)\n}\nfunc (m *RatelimitUpdateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RatelimitUpdateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RatelimitUpdateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RatelimitUpdateResponse proto.InternalMessageInfo\n\nfunc (m *RatelimitUpdateResponse) GetData() *v12.Any {\n\tif m != nil {\n\t\treturn m.Data\n\t}\n\treturn nil\n}\n\nfunc init() {\n\tproto.RegisterType((*StartWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.StartWorkflowExecutionRequest\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.history.v1.StartWorkflowExecutionRequest.PartitionConfigEntry\")\n\tproto.RegisterType((*StartWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.StartWorkflowExecutionResponse\")\n\tproto.RegisterType((*SignalWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.SignalWorkflowExecutionRequest\")\n\tproto.RegisterType((*SignalWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.SignalWorkflowExecutionResponse\")\n\tproto.RegisterType((*SignalWithStartWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.SignalWithStartWorkflowExecutionRequest\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.history.v1.SignalWithStartWorkflowExecutionRequest.PartitionConfigEntry\")\n\tproto.RegisterType((*SignalWithStartWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.SignalWithStartWorkflowExecutionResponse\")\n\tproto.RegisterType((*ResetWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.ResetWorkflowExecutionRequest\")\n\tproto.RegisterType((*ResetWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.ResetWorkflowExecutionResponse\")\n\tproto.RegisterType((*TerminateWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.TerminateWorkflowExecutionRequest\")\n\tproto.RegisterType((*TerminateWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.TerminateWorkflowExecutionResponse\")\n\tproto.RegisterType((*DescribeWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.DescribeWorkflowExecutionRequest\")\n\tproto.RegisterType((*DescribeWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.DescribeWorkflowExecutionResponse\")\n\tproto.RegisterType((*QueryWorkflowRequest)(nil), \"uber.cadence.history.v1.QueryWorkflowRequest\")\n\tproto.RegisterType((*QueryWorkflowResponse)(nil), \"uber.cadence.history.v1.QueryWorkflowResponse\")\n\tproto.RegisterType((*ResetStickyTaskListRequest)(nil), \"uber.cadence.history.v1.ResetStickyTaskListRequest\")\n\tproto.RegisterType((*ResetStickyTaskListResponse)(nil), \"uber.cadence.history.v1.ResetStickyTaskListResponse\")\n\tproto.RegisterType((*GetMutableStateRequest)(nil), \"uber.cadence.history.v1.GetMutableStateRequest\")\n\tproto.RegisterType((*GetMutableStateResponse)(nil), \"uber.cadence.history.v1.GetMutableStateResponse\")\n\tproto.RegisterType((*PollMutableStateRequest)(nil), \"uber.cadence.history.v1.PollMutableStateRequest\")\n\tproto.RegisterType((*PollMutableStateResponse)(nil), \"uber.cadence.history.v1.PollMutableStateResponse\")\n\tproto.RegisterType((*RecordDecisionTaskStartedRequest)(nil), \"uber.cadence.history.v1.RecordDecisionTaskStartedRequest\")\n\tproto.RegisterType((*RecordDecisionTaskStartedResponse)(nil), \"uber.cadence.history.v1.RecordDecisionTaskStartedResponse\")\n\tproto.RegisterMapType((map[string]*v1.WorkflowQuery)(nil), \"uber.cadence.history.v1.RecordDecisionTaskStartedResponse.QueriesEntry\")\n\tproto.RegisterType((*RecordActivityTaskStartedRequest)(nil), \"uber.cadence.history.v1.RecordActivityTaskStartedRequest\")\n\tproto.RegisterType((*RecordActivityTaskStartedResponse)(nil), \"uber.cadence.history.v1.RecordActivityTaskStartedResponse\")\n\tproto.RegisterType((*RespondDecisionTaskCompletedRequest)(nil), \"uber.cadence.history.v1.RespondDecisionTaskCompletedRequest\")\n\tproto.RegisterType((*RespondDecisionTaskCompletedResponse)(nil), \"uber.cadence.history.v1.RespondDecisionTaskCompletedResponse\")\n\tproto.RegisterMapType((map[string]*v1.ActivityLocalDispatchInfo)(nil), \"uber.cadence.history.v1.RespondDecisionTaskCompletedResponse.ActivitiesToDispatchLocallyEntry\")\n\tproto.RegisterType((*RespondDecisionTaskFailedRequest)(nil), \"uber.cadence.history.v1.RespondDecisionTaskFailedRequest\")\n\tproto.RegisterType((*RespondDecisionTaskFailedResponse)(nil), \"uber.cadence.history.v1.RespondDecisionTaskFailedResponse\")\n\tproto.RegisterType((*RecordActivityTaskHeartbeatRequest)(nil), \"uber.cadence.history.v1.RecordActivityTaskHeartbeatRequest\")\n\tproto.RegisterType((*RecordActivityTaskHeartbeatResponse)(nil), \"uber.cadence.history.v1.RecordActivityTaskHeartbeatResponse\")\n\tproto.RegisterType((*RespondActivityTaskCompletedRequest)(nil), \"uber.cadence.history.v1.RespondActivityTaskCompletedRequest\")\n\tproto.RegisterType((*RespondActivityTaskCompletedResponse)(nil), \"uber.cadence.history.v1.RespondActivityTaskCompletedResponse\")\n\tproto.RegisterType((*RespondActivityTaskFailedRequest)(nil), \"uber.cadence.history.v1.RespondActivityTaskFailedRequest\")\n\tproto.RegisterType((*RespondActivityTaskFailedResponse)(nil), \"uber.cadence.history.v1.RespondActivityTaskFailedResponse\")\n\tproto.RegisterType((*RespondActivityTaskCanceledRequest)(nil), \"uber.cadence.history.v1.RespondActivityTaskCanceledRequest\")\n\tproto.RegisterType((*RespondActivityTaskCanceledResponse)(nil), \"uber.cadence.history.v1.RespondActivityTaskCanceledResponse\")\n\tproto.RegisterType((*RemoveSignalMutableStateRequest)(nil), \"uber.cadence.history.v1.RemoveSignalMutableStateRequest\")\n\tproto.RegisterType((*RemoveSignalMutableStateResponse)(nil), \"uber.cadence.history.v1.RemoveSignalMutableStateResponse\")\n\tproto.RegisterType((*RequestCancelWorkflowExecutionRequest)(nil), \"uber.cadence.history.v1.RequestCancelWorkflowExecutionRequest\")\n\tproto.RegisterType((*RequestCancelWorkflowExecutionResponse)(nil), \"uber.cadence.history.v1.RequestCancelWorkflowExecutionResponse\")\n\tproto.RegisterType((*ScheduleDecisionTaskRequest)(nil), \"uber.cadence.history.v1.ScheduleDecisionTaskRequest\")\n\tproto.RegisterType((*ScheduleDecisionTaskResponse)(nil), \"uber.cadence.history.v1.ScheduleDecisionTaskResponse\")\n\tproto.RegisterType((*RecordChildExecutionCompletedRequest)(nil), \"uber.cadence.history.v1.RecordChildExecutionCompletedRequest\")\n\tproto.RegisterType((*RecordChildExecutionCompletedResponse)(nil), \"uber.cadence.history.v1.RecordChildExecutionCompletedResponse\")\n\tproto.RegisterType((*ReplicateEventsV2Request)(nil), \"uber.cadence.history.v1.ReplicateEventsV2Request\")\n\tproto.RegisterType((*ReplicateEventsV2Response)(nil), \"uber.cadence.history.v1.ReplicateEventsV2Response\")\n\tproto.RegisterType((*SyncShardStatusRequest)(nil), \"uber.cadence.history.v1.SyncShardStatusRequest\")\n\tproto.RegisterType((*SyncShardStatusResponse)(nil), \"uber.cadence.history.v1.SyncShardStatusResponse\")\n\tproto.RegisterType((*SyncActivityRequest)(nil), \"uber.cadence.history.v1.SyncActivityRequest\")\n\tproto.RegisterType((*SyncActivityResponse)(nil), \"uber.cadence.history.v1.SyncActivityResponse\")\n\tproto.RegisterType((*DescribeMutableStateRequest)(nil), \"uber.cadence.history.v1.DescribeMutableStateRequest\")\n\tproto.RegisterType((*DescribeMutableStateResponse)(nil), \"uber.cadence.history.v1.DescribeMutableStateResponse\")\n\tproto.RegisterType((*DescribeHistoryHostRequest)(nil), \"uber.cadence.history.v1.DescribeHistoryHostRequest\")\n\tproto.RegisterType((*DescribeHistoryHostResponse)(nil), \"uber.cadence.history.v1.DescribeHistoryHostResponse\")\n\tproto.RegisterType((*CloseShardRequest)(nil), \"uber.cadence.history.v1.CloseShardRequest\")\n\tproto.RegisterType((*CloseShardResponse)(nil), \"uber.cadence.history.v1.CloseShardResponse\")\n\tproto.RegisterType((*RemoveTaskRequest)(nil), \"uber.cadence.history.v1.RemoveTaskRequest\")\n\tproto.RegisterType((*RemoveTaskResponse)(nil), \"uber.cadence.history.v1.RemoveTaskResponse\")\n\tproto.RegisterType((*ResetQueueRequest)(nil), \"uber.cadence.history.v1.ResetQueueRequest\")\n\tproto.RegisterType((*ResetQueueResponse)(nil), \"uber.cadence.history.v1.ResetQueueResponse\")\n\tproto.RegisterType((*DescribeQueueRequest)(nil), \"uber.cadence.history.v1.DescribeQueueRequest\")\n\tproto.RegisterType((*DescribeQueueResponse)(nil), \"uber.cadence.history.v1.DescribeQueueResponse\")\n\tproto.RegisterType((*GetReplicationMessagesRequest)(nil), \"uber.cadence.history.v1.GetReplicationMessagesRequest\")\n\tproto.RegisterType((*GetReplicationMessagesResponse)(nil), \"uber.cadence.history.v1.GetReplicationMessagesResponse\")\n\tproto.RegisterMapType((map[int32]*v11.ReplicationMessages)(nil), \"uber.cadence.history.v1.GetReplicationMessagesResponse.ShardMessagesEntry\")\n\tproto.RegisterType((*GetDLQReplicationMessagesRequest)(nil), \"uber.cadence.history.v1.GetDLQReplicationMessagesRequest\")\n\tproto.RegisterType((*GetDLQReplicationMessagesResponse)(nil), \"uber.cadence.history.v1.GetDLQReplicationMessagesResponse\")\n\tproto.RegisterType((*ReapplyEventsRequest)(nil), \"uber.cadence.history.v1.ReapplyEventsRequest\")\n\tproto.RegisterType((*ReapplyEventsResponse)(nil), \"uber.cadence.history.v1.ReapplyEventsResponse\")\n\tproto.RegisterType((*RefreshWorkflowTasksRequest)(nil), \"uber.cadence.history.v1.RefreshWorkflowTasksRequest\")\n\tproto.RegisterType((*RefreshWorkflowTasksResponse)(nil), \"uber.cadence.history.v1.RefreshWorkflowTasksResponse\")\n\tproto.RegisterType((*CountDLQMessagesRequest)(nil), \"uber.cadence.history.v1.CountDLQMessagesRequest\")\n\tproto.RegisterType((*CountDLQMessagesResponse)(nil), \"uber.cadence.history.v1.CountDLQMessagesResponse\")\n\tproto.RegisterType((*ReadDLQMessagesRequest)(nil), \"uber.cadence.history.v1.ReadDLQMessagesRequest\")\n\tproto.RegisterType((*ReadDLQMessagesResponse)(nil), \"uber.cadence.history.v1.ReadDLQMessagesResponse\")\n\tproto.RegisterType((*PurgeDLQMessagesRequest)(nil), \"uber.cadence.history.v1.PurgeDLQMessagesRequest\")\n\tproto.RegisterType((*PurgeDLQMessagesResponse)(nil), \"uber.cadence.history.v1.PurgeDLQMessagesResponse\")\n\tproto.RegisterType((*MergeDLQMessagesRequest)(nil), \"uber.cadence.history.v1.MergeDLQMessagesRequest\")\n\tproto.RegisterType((*MergeDLQMessagesResponse)(nil), \"uber.cadence.history.v1.MergeDLQMessagesResponse\")\n\tproto.RegisterType((*NotifyFailoverMarkersRequest)(nil), \"uber.cadence.history.v1.NotifyFailoverMarkersRequest\")\n\tproto.RegisterType((*NotifyFailoverMarkersResponse)(nil), \"uber.cadence.history.v1.NotifyFailoverMarkersResponse\")\n\tproto.RegisterType((*GetCrossClusterTasksRequest)(nil), \"uber.cadence.history.v1.GetCrossClusterTasksRequest\")\n\tproto.RegisterType((*GetCrossClusterTasksResponse)(nil), \"uber.cadence.history.v1.GetCrossClusterTasksResponse\")\n\tproto.RegisterMapType((map[int32]v11.GetTaskFailedCause)(nil), \"uber.cadence.history.v1.GetCrossClusterTasksResponse.FailedCauseByShardEntry\")\n\tproto.RegisterMapType((map[int32]*v11.CrossClusterTaskRequests)(nil), \"uber.cadence.history.v1.GetCrossClusterTasksResponse.TasksByShardEntry\")\n\tproto.RegisterType((*RespondCrossClusterTasksCompletedRequest)(nil), \"uber.cadence.history.v1.RespondCrossClusterTasksCompletedRequest\")\n\tproto.RegisterType((*RespondCrossClusterTasksCompletedResponse)(nil), \"uber.cadence.history.v1.RespondCrossClusterTasksCompletedResponse\")\n\tproto.RegisterType((*GetFailoverInfoRequest)(nil), \"uber.cadence.history.v1.GetFailoverInfoRequest\")\n\tproto.RegisterType((*GetFailoverInfoResponse)(nil), \"uber.cadence.history.v1.GetFailoverInfoResponse\")\n\tproto.RegisterType((*RatelimitUpdateRequest)(nil), \"uber.cadence.history.v1.RatelimitUpdateRequest\")\n\tproto.RegisterType((*RatelimitUpdateResponse)(nil), \"uber.cadence.history.v1.RatelimitUpdateResponse\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/history/v1/service.proto\", fileDescriptor_fee8ff76963a38ed)\n}\n\nvar fileDescriptor_fee8ff76963a38ed = []byte{\n\t// 4981 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x7c, 0x4d, 0x6c, 0x1c, 0x47,\n\t0x76, 0x30, 0x7a, 0x46, 0xfc, 0x7b, 0x24, 0x87, 0x64, 0x89, 0x3f, 0xc3, 0xa1, 0x44, 0x91, 0x6d,\n\t0xcb, 0xa2, 0xe5, 0xf5, 0xd0, 0xa2, 0x6c, 0x49, 0x96, 0xe5, 0xd5, 0x4a, 0xa4, 0x24, 0x8f, 0x3f,\n\t0xfd, 0x36, 0x69, 0xf9, 0xcb, 0x9f, 0x7b, 0x9b, 0xd3, 0x35, 0x64, 0x47, 0x3d, 0xdd, 0xa3, 0xee,\n\t0x1e, 0x52, 0xf4, 0x21, 0x70, 0xe2, 0x20, 0x40, 0x16, 0x41, 0x76, 0xb3, 0x48, 0x82, 0x00, 0x01,\n\t0x02, 0x04, 0x1b, 0x60, 0xb1, 0x46, 0x6e, 0x09, 0x90, 0x43, 0x92, 0x53, 0x2e, 0x7b, 0xdc, 0x6b,\n\t0x6e, 0x81, 0xb1, 0x7b, 0x48, 0x80, 0xdc, 0xf6, 0x1c, 0x04, 0xf5, 0xd3, 0x3d, 0xfd, 0x53, 0x5d,\n\t0x3d, 0x43, 0x06, 0x91, 0xd7, 0xf1, 0x6d, 0xba, 0xaa, 0xde, 0xab, 0x57, 0xaf, 0xde, 0x7b, 0xfd,\n\t0xfe, 0x7a, 0xe0, 0x7c, 0x77, 0x17, 0x7b, 0xeb, 0x4d, 0xc3, 0xc4, 0x4e, 0x13, 0xaf, 0xef, 0x5b,\n\t0x7e, 0xe0, 0x7a, 0x47, 0xeb, 0x07, 0x97, 0xd6, 0x7d, 0xec, 0x1d, 0x58, 0x4d, 0x5c, 0xef, 0x78,\n\t0x6e, 0xe0, 0xa2, 0x05, 0xb2, 0xac, 0xce, 0x97, 0xd5, 0xf9, 0xb2, 0xfa, 0xc1, 0xa5, 0xda, 0xf2,\n\t0x9e, 0xeb, 0xee, 0xd9, 0x78, 0x9d, 0x2e, 0xdb, 0xed, 0xb6, 0xd6, 0xcd, 0xae, 0x67, 0x04, 0x96,\n\t0xeb, 0x30, 0xc0, 0xda, 0xb9, 0xf4, 0x7c, 0x60, 0xb5, 0xb1, 0x1f, 0x18, 0xed, 0x0e, 0x5f, 0x90,\n\t0x41, 0x70, 0xe8, 0x19, 0x9d, 0x0e, 0xf6, 0x7c, 0x3e, 0xbf, 0x92, 0x20, 0xd0, 0xe8, 0x58, 0x84,\n\t0xb8, 0xa6, 0xdb, 0x6e, 0x47, 0x5b, 0xac, 0x8a, 0x56, 0x84, 0x24, 0x72, 0x2a, 0x44, 0x4b, 0x9e,\n\t0x77, 0x71, 0xb4, 0x40, 0x15, 0x2d, 0x08, 0x0c, 0xff, 0x99, 0x6d, 0xf9, 0x81, 0x6c, 0xcd, 0xa1,\n\t0xeb, 0x3d, 0x6b, 0xd9, 0xee, 0x21, 0x5f, 0x73, 0x51, 0xb4, 0x86, 0xb3, 0x52, 0x4f, 0xad, 0x5d,\n\t0x2b, 0x5a, 0x8b, 0x3d, 0xbe, 0xf2, 0x95, 0xe4, 0x4a, 0xb3, 0x6d, 0x39, 0x94, 0x0b, 0x76, 0xd7,\n\t0x0f, 0x8a, 0x16, 0x25, 0x19, 0xb1, 0x2a, 0x5e, 0xf4, 0xbc, 0x8b, 0xbb, 0xfc, 0xaa, 0x6b, 0x17,\n\t0xc4, 0x4b, 0x3c, 0xdc, 0xb1, 0xad, 0x66, 0xfc, 0x6a, 0x93, 0x37, 0xe3, 0xef, 0x1b, 0x1e, 0x36,\n\t0xc9, 0x4a, 0xc3, 0x09, 0x77, 0x7b, 0x35, 0x67, 0x45, 0x92, 0xa6, 0xf3, 0x39, 0xab, 0x92, 0xec,\n\t0x52, 0x7f, 0x3e, 0x0c, 0x67, 0xb7, 0x03, 0xc3, 0x0b, 0x3e, 0xe6, 0xe3, 0x77, 0x5e, 0xe0, 0x66,\n\t0x97, 0xd0, 0xa3, 0xe1, 0xe7, 0x5d, 0xec, 0x07, 0xe8, 0x3e, 0x8c, 0x78, 0xec, 0x67, 0x55, 0x59,\n\t0x51, 0xd6, 0xc6, 0x37, 0x36, 0xea, 0x09, 0xb1, 0x35, 0x3a, 0x56, 0xfd, 0xe0, 0x52, 0x5d, 0x8a,\n\t0x44, 0x0b, 0x51, 0xa0, 0x25, 0x18, 0x33, 0xdd, 0xb6, 0x61, 0x39, 0xba, 0x65, 0x56, 0x4b, 0x2b,\n\t0xca, 0xda, 0x98, 0x36, 0xca, 0x06, 0x1a, 0x26, 0xfa, 0x4d, 0x98, 0xeb, 0x18, 0x1e, 0x76, 0x02,\n\t0x1d, 0x87, 0x08, 0x74, 0xcb, 0x69, 0xb9, 0xd5, 0x32, 0xdd, 0x78, 0x4d, 0xb8, 0xf1, 0x63, 0x0a,\n\t0x11, 0xed, 0xd8, 0x70, 0x5a, 0xae, 0x76, 0xba, 0x93, 0x1d, 0x44, 0x55, 0x18, 0x31, 0x82, 0x00,\n\t0xb7, 0x3b, 0x41, 0xf5, 0xd4, 0x8a, 0xb2, 0x36, 0xa4, 0x85, 0x8f, 0x68, 0x13, 0xa6, 0xf0, 0x8b,\n\t0x8e, 0xc5, 0x54, 0x4c, 0x27, 0xba, 0x54, 0x1d, 0xa2, 0x3b, 0xd6, 0xea, 0x4c, 0x8f, 0xea, 0xa1,\n\t0x1e, 0xd5, 0x77, 0x42, 0x45, 0xd3, 0x2a, 0x3d, 0x10, 0x32, 0x88, 0x5a, 0xb0, 0xd8, 0x74, 0x9d,\n\t0xc0, 0x72, 0xba, 0x58, 0x37, 0x7c, 0xdd, 0xc1, 0x87, 0xba, 0xe5, 0x58, 0x81, 0x65, 0x04, 0xae,\n\t0x57, 0x1d, 0x5e, 0x51, 0xd6, 0x2a, 0x1b, 0x6f, 0x08, 0x0f, 0xb0, 0xc9, 0xa1, 0x6e, 0xf9, 0x0f,\n\t0xf1, 0x61, 0x23, 0x04, 0xd1, 0xe6, 0x9b, 0xc2, 0x71, 0xd4, 0x80, 0x99, 0x70, 0xc6, 0xd4, 0x5b,\n\t0x86, 0x65, 0x77, 0x3d, 0x5c, 0x1d, 0xa1, 0xe4, 0x9e, 0x11, 0xe2, 0xbf, 0xcb, 0xd6, 0x68, 0xd3,\n\t0x11, 0x18, 0x1f, 0x41, 0x1a, 0xcc, 0xdb, 0x86, 0x1f, 0xe8, 0x4d, 0xb7, 0xdd, 0xb1, 0x31, 0x3d,\n\t0xbc, 0x87, 0xfd, 0xae, 0x1d, 0x54, 0x47, 0x25, 0xf8, 0x1e, 0x1b, 0x47, 0xb6, 0x6b, 0x98, 0xda,\n\t0x2c, 0x81, 0xdd, 0x8c, 0x40, 0x35, 0x0a, 0x89, 0xfe, 0x3f, 0x2c, 0xb5, 0x2c, 0xcf, 0x0f, 0x74,\n\t0x13, 0x37, 0x2d, 0x9f, 0xf2, 0xd3, 0xf0, 0x9f, 0xe9, 0xbb, 0x46, 0xf3, 0x99, 0xdb, 0x6a, 0x55,\n\t0xc7, 0x28, 0xe2, 0xc5, 0x0c, 0x5f, 0xb7, 0xb8, 0x81, 0xd3, 0xaa, 0x14, 0x7a, 0x8b, 0x03, 0xef,\n\t0x18, 0xfe, 0xb3, 0xdb, 0x0c, 0x14, 0x1d, 0xc0, 0x74, 0xc7, 0xf0, 0x02, 0x8b, 0xd2, 0xd9, 0x74,\n\t0x9d, 0x96, 0xb5, 0x57, 0x85, 0x95, 0xf2, 0xda, 0xf8, 0xc6, 0xff, 0xab, 0xe7, 0x18, 0x52, 0xb9,\n\t0x54, 0x12, 0xd1, 0x61, 0xe8, 0x36, 0x29, 0xb6, 0x3b, 0x4e, 0xe0, 0x1d, 0x69, 0x53, 0x9d, 0xe4,\n\t0x68, 0xed, 0x36, 0xcc, 0x8a, 0x16, 0xa2, 0x69, 0x28, 0x3f, 0xc3, 0x47, 0x54, 0x29, 0xc6, 0x34,\n\t0xf2, 0x13, 0xcd, 0xc2, 0xd0, 0x81, 0x61, 0x77, 0x31, 0x17, 0x6c, 0xf6, 0x70, 0xbd, 0x74, 0x4d,\n\t0x51, 0xaf, 0xc2, 0x72, 0x1e, 0x29, 0x7e, 0xc7, 0x75, 0x7c, 0x8c, 0xe6, 0x60, 0xd8, 0xeb, 0x52,\n\t0xad, 0x60, 0x08, 0x87, 0xbc, 0xae, 0xd3, 0x30, 0xd5, 0xbf, 0x29, 0xc1, 0xf2, 0xb6, 0xb5, 0xe7,\n\t0x18, 0x76, 0xae, 0x82, 0x3e, 0x48, 0x2b, 0xe8, 0x65, 0xb1, 0x82, 0x4a, 0xb1, 0xf4, 0xa9, 0xa1,\n\t0x2d, 0x58, 0xc2, 0x2f, 0x02, 0xec, 0x39, 0x86, 0x1d, 0x19, 0xde, 0x9e, 0xb2, 0x72, 0x3d, 0x7d,\n\t0x4d, 0xb8, 0x7f, 0x76, 0xe7, 0xc5, 0x10, 0x55, 0x66, 0x0a, 0xd5, 0xe1, 0x74, 0x73, 0xdf, 0xb2,\n\t0xcd, 0xde, 0x26, 0xae, 0x63, 0x1f, 0x51, 0xbd, 0x1d, 0xd5, 0x66, 0xe8, 0x54, 0x08, 0xf4, 0xc8,\n\t0xb1, 0x8f, 0xd4, 0x55, 0x38, 0x97, 0x7b, 0x3e, 0xc6, 0x60, 0xf5, 0x17, 0x25, 0xb8, 0xc0, 0xd7,\n\t0x58, 0xc1, 0xbe, 0xdc, 0xe6, 0x3d, 0x4d, 0xb3, 0xf4, 0x86, 0x8c, 0xa5, 0x45, 0xe8, 0xfa, 0xe4,\n\t0xed, 0x67, 0x8a, 0x40, 0xc0, 0xcb, 0x54, 0xc0, 0x3f, 0xca, 0x17, 0xf0, 0xfe, 0x48, 0xf8, 0x5f,\n\t0x14, 0xf5, 0x5b, 0xb0, 0x56, 0x4c, 0x94, 0x5c, 0xe8, 0xbf, 0xa7, 0xc0, 0x59, 0x0d, 0xfb, 0xf8,\n\t0xc4, 0x2f, 0x25, 0x29, 0x92, 0xfe, 0xae, 0x85, 0xa8, 0x6e, 0x1e, 0x1a, 0xf9, 0x29, 0xbe, 0x28,\n\t0xc1, 0xea, 0x0e, 0xf6, 0xda, 0x96, 0x63, 0x04, 0x38, 0xf7, 0x24, 0x8f, 0xd3, 0x27, 0xb9, 0x22,\n\t0x3c, 0x49, 0x21, 0xa2, 0x5f, 0x71, 0x05, 0x7e, 0x15, 0x54, 0xd9, 0x11, 0xb9, 0x0e, 0xff, 0x40,\n\t0x81, 0x95, 0x2d, 0xec, 0x37, 0x3d, 0x6b, 0x37, 0x9f, 0xa3, 0x8f, 0xd2, 0x1c, 0x7d, 0x47, 0x78,\n\t0x9c, 0x22, 0x3c, 0x7d, 0x8a, 0xc7, 0x7f, 0x95, 0x61, 0x55, 0x82, 0x8a, 0x8b, 0x88, 0x0d, 0x0b,\n\t0x3d, 0x97, 0x86, 0xa9, 0x36, 0x7f, 0xe1, 0x49, 0x6d, 0x76, 0x06, 0xe1, 0x66, 0x1c, 0x54, 0x9b,\n\t0xc7, 0xc2, 0x71, 0xb4, 0x0b, 0x0b, 0xd9, 0xbb, 0x65, 0x9e, 0x54, 0x89, 0xee, 0x76, 0xb1, 0xbf,\n\t0xdd, 0xa8, 0x2f, 0x35, 0x77, 0x28, 0x1a, 0x46, 0x1f, 0x03, 0xea, 0x60, 0xc7, 0xb4, 0x9c, 0x3d,\n\t0xdd, 0x68, 0x06, 0xd6, 0x81, 0x15, 0x58, 0xd8, 0xe7, 0xe6, 0x2a, 0xc7, 0x51, 0x63, 0xcb, 0x6f,\n\t0xb1, 0xd5, 0x47, 0x14, 0xf9, 0x4c, 0x27, 0x31, 0x68, 0x61, 0x1f, 0xfd, 0x1a, 0x4c, 0x87, 0x88,\n\t0xa9, 0x98, 0x78, 0xd8, 0xa9, 0x9e, 0xa2, 0x68, 0xeb, 0x32, 0xb4, 0x9b, 0x64, 0x6d, 0x92, 0xf2,\n\t0xa9, 0x4e, 0x6c, 0xca, 0xc3, 0x0e, 0xda, 0xee, 0xa1, 0x0e, 0xbd, 0x13, 0xee, 0xe8, 0x49, 0x29,\n\t0x0e, 0x9d, 0x91, 0x04, 0xd2, 0x70, 0x50, 0x7d, 0x01, 0xb3, 0x4f, 0x48, 0xcc, 0x13, 0x72, 0x2f,\n\t0x14, 0xc3, 0xcd, 0xb4, 0x18, 0xbe, 0x2e, 0xdc, 0x43, 0x04, 0xdb, 0xa7, 0xe8, 0xfd, 0x48, 0x81,\n\t0xb9, 0x14, 0x38, 0x17, 0xb7, 0x9b, 0x30, 0x41, 0xe3, 0xb0, 0xd0, 0x9d, 0x53, 0xfa, 0x70, 0xe7,\n\t0xc6, 0x29, 0x04, 0xf7, 0xe2, 0x1a, 0x50, 0x09, 0x11, 0xfc, 0x36, 0x6e, 0x06, 0xd8, 0xe4, 0x82,\n\t0xa3, 0xe6, 0x9f, 0x41, 0xe3, 0x2b, 0xb5, 0xc9, 0xe7, 0xf1, 0x47, 0xf5, 0xf7, 0x15, 0xa8, 0x51,\n\t0x03, 0xba, 0x1d, 0x58, 0xcd, 0x67, 0x47, 0xc4, 0xa3, 0xbb, 0x6f, 0xf9, 0x41, 0xc8, 0xa6, 0x46,\n\t0x9a, 0x4d, 0xeb, 0xf9, 0x96, 0x5c, 0x88, 0xa1, 0x4f, 0x66, 0x9d, 0x85, 0x25, 0x21, 0x0e, 0x6e,\n\t0x59, 0x7e, 0x56, 0x82, 0xf9, 0x7b, 0x38, 0x78, 0xd0, 0x0d, 0x8c, 0x5d, 0x1b, 0x6f, 0x07, 0x46,\n\t0x80, 0x35, 0x11, 0x5a, 0x25, 0x65, 0x4f, 0x3f, 0x02, 0x24, 0x30, 0xa3, 0xa5, 0x81, 0xcc, 0xe8,\n\t0x4c, 0x46, 0xc3, 0xd0, 0x65, 0x98, 0xc7, 0x2f, 0x3a, 0x94, 0x81, 0xba, 0x83, 0x5f, 0x04, 0x3a,\n\t0x3e, 0x20, 0x61, 0x91, 0x65, 0x52, 0x0b, 0x5d, 0xd6, 0x4e, 0x87, 0xb3, 0x0f, 0xf1, 0x8b, 0xe0,\n\t0x0e, 0x99, 0x6b, 0x98, 0xe8, 0x2d, 0x98, 0x6d, 0x76, 0x3d, 0x1a, 0x3f, 0xed, 0x7a, 0x86, 0xd3,\n\t0xdc, 0xd7, 0x03, 0xf7, 0x19, 0xd5, 0x1e, 0x65, 0x6d, 0x42, 0x43, 0x7c, 0xee, 0x36, 0x9d, 0xda,\n\t0x21, 0x33, 0xe8, 0x37, 0x60, 0xf6, 0x00, 0x7b, 0xd4, 0x4b, 0xe7, 0x3e, 0x85, 0x6e, 0x05, 0xb8,\n\t0xcd, 0x95, 0x22, 0x2d, 0xb0, 0x24, 0x68, 0x25, 0x27, 0x78, 0xca, 0x40, 0x3e, 0x60, 0x10, 0x8d,\n\t0x00, 0xb7, 0x35, 0x74, 0x90, 0x19, 0x53, 0xff, 0x61, 0x0c, 0x16, 0x32, 0x2c, 0xe5, 0x02, 0x2a,\n\t0x66, 0x9b, 0x72, 0x52, 0xb6, 0xdd, 0x85, 0xc9, 0x08, 0x6d, 0x70, 0xd4, 0xc1, 0xfc, 0x22, 0x56,\n\t0xa5, 0x18, 0x77, 0x8e, 0x3a, 0x58, 0x9b, 0x38, 0x8c, 0x3d, 0x21, 0x15, 0x26, 0x45, 0x5c, 0x1f,\n\t0x77, 0x62, 0xdc, 0x7e, 0x0a, 0x8b, 0x1d, 0x0f, 0x1f, 0x58, 0x6e, 0xd7, 0xd7, 0x7d, 0xe2, 0xe6,\n\t0x60, 0xb3, 0xb7, 0xfe, 0x14, 0xdd, 0x77, 0x29, 0x13, 0xe6, 0x34, 0x9c, 0xe0, 0xca, 0xdb, 0x4f,\n\t0x89, 0xaf, 0xa4, 0xcd, 0x87, 0xd0, 0xdb, 0x0c, 0x38, 0xc4, 0xfb, 0x26, 0x9c, 0xa6, 0x41, 0x19,\n\t0x8b, 0xa2, 0x22, 0x8c, 0x43, 0x94, 0x82, 0x69, 0x32, 0x75, 0x97, 0xcc, 0x84, 0xcb, 0xaf, 0xc3,\n\t0x18, 0x0d, 0xb0, 0x6c, 0xcb, 0x0f, 0x68, 0x98, 0x39, 0xbe, 0x71, 0x56, 0xec, 0x41, 0x84, 0x22,\n\t0x3f, 0x1a, 0xf0, 0x5f, 0xe8, 0x1e, 0x4c, 0xfb, 0x54, 0x1d, 0xf4, 0x1e, 0x8a, 0x91, 0x7e, 0x50,\n\t0x54, 0xfc, 0x84, 0x16, 0xa1, 0xb7, 0x61, 0xbe, 0x69, 0x5b, 0x84, 0x52, 0xdb, 0xda, 0xf5, 0x0c,\n\t0xef, 0x48, 0xe7, 0xf2, 0x40, 0x03, 0xc9, 0x31, 0x6d, 0x96, 0xcd, 0xde, 0x67, 0x93, 0x5c, 0x7e,\n\t0x62, 0x50, 0x2d, 0x6c, 0x04, 0x5d, 0x0f, 0x47, 0x50, 0x63, 0x71, 0xa8, 0xbb, 0x6c, 0x32, 0x84,\n\t0x3a, 0x07, 0xe3, 0x1c, 0xca, 0x6a, 0x77, 0xec, 0x2a, 0xd0, 0xa5, 0xc0, 0x86, 0x1a, 0xed, 0x8e,\n\t0x8d, 0x7c, 0xb8, 0x98, 0x3e, 0x95, 0xee, 0x37, 0xf7, 0xb1, 0xd9, 0xb5, 0xb1, 0x1e, 0xb8, 0xec,\n\t0xb2, 0x68, 0x94, 0xef, 0x76, 0x83, 0xea, 0x78, 0x51, 0x40, 0xfa, 0x6a, 0xf2, 0xac, 0xdb, 0x1c,\n\t0xd3, 0x8e, 0x4b, 0xef, 0x6d, 0x87, 0xa1, 0x21, 0xfe, 0x0e, 0xbb, 0x2a, 0x22, 0xff, 0xbd, 0x83,\n\t0x4c, 0xd0, 0x44, 0xc3, 0x0c, 0x9d, 0xda, 0x26, 0x33, 0xe1, 0x29, 0xf2, 0x74, 0x75, 0x32, 0x57,\n\t0x57, 0xef, 0x43, 0x25, 0x92, 0x6d, 0x9f, 0x28, 0x53, 0xb5, 0x42, 0x93, 0x0a, 0xe7, 0x93, 0x57,\n\t0xc5, 0x32, 0x3d, 0x71, 0xf9, 0x66, 0x9a, 0x17, 0x29, 0x06, 0x7d, 0x44, 0x4d, 0x98, 0x8d, 0xb0,\n\t0x35, 0x6d, 0xd7, 0xc7, 0x1c, 0xe7, 0x14, 0xc5, 0x79, 0xa9, 0x4f, 0x6f, 0x84, 0x00, 0x12, 0x7c,\n\t0x5d, 0x5f, 0x8b, 0xf4, 0x39, 0x1a, 0x24, 0x5a, 0x3e, 0x93, 0x34, 0x2f, 0xc4, 0x45, 0x98, 0x16,\n\t0xbd, 0x70, 0x7b, 0x54, 0x27, 0x8c, 0x8b, 0x85, 0x7d, 0x6d, 0xfa, 0x20, 0x35, 0x82, 0x6e, 0xc0,\n\t0x92, 0x45, 0x74, 0x2e, 0x75, 0xc7, 0xd8, 0x21, 0x76, 0xc6, 0xac, 0xce, 0x50, 0x1f, 0x73, 0xc1,\n\t0xf2, 0x93, 0xa6, 0xfe, 0x0e, 0x9b, 0x46, 0xab, 0x30, 0x11, 0xda, 0x3a, 0xdf, 0xfa, 0x14, 0x57,\n\t0x11, 0x53, 0x6d, 0x3e, 0xb6, 0x6d, 0x7d, 0x8a, 0xd5, 0x5f, 0x2a, 0xb0, 0xf0, 0xd8, 0xb5, 0xed,\n\t0xff, 0x5b, 0x6f, 0x03, 0xf5, 0xc7, 0xa3, 0x50, 0xcd, 0x1e, 0xfb, 0x1b, 0x8b, 0xfd, 0x8d, 0xc5,\n\t0xfe, 0x3a, 0x5a, 0xec, 0x3c, 0xfd, 0x98, 0xc8, 0xb5, 0xc0, 0x42, 0x73, 0x36, 0x79, 0x62, 0x73,\n\t0xf6, 0xab, 0x67, 0xd8, 0xd5, 0x7f, 0x29, 0xc1, 0x8a, 0x86, 0x9b, 0xae, 0x67, 0xc6, 0x13, 0xb5,\n\t0x5c, 0x2d, 0x5e, 0xa6, 0xa5, 0x3c, 0x07, 0xe3, 0x91, 0xe0, 0x44, 0x46, 0x00, 0xc2, 0xa1, 0x86,\n\t0x89, 0x16, 0x60, 0x84, 0xca, 0x18, 0xd7, 0xf8, 0xb2, 0x36, 0x4c, 0x1e, 0x1b, 0x26, 0x3a, 0x0b,\n\t0xc0, 0xe3, 0x88, 0x50, 0x77, 0xc7, 0xb4, 0x31, 0x3e, 0xd2, 0x30, 0x91, 0x06, 0x13, 0x1d, 0xd7,\n\t0xb6, 0xf5, 0x30, 0x56, 0x19, 0x96, 0xc4, 0x2a, 0xc4, 0x86, 0xde, 0x75, 0xbd, 0x38, 0x6b, 0xc2,\n\t0x58, 0x65, 0x9c, 0x20, 0xe1, 0x0f, 0xea, 0xef, 0x8d, 0xc2, 0xaa, 0x84, 0x8b, 0xdc, 0xf0, 0x66,\n\t0x2c, 0xa4, 0x72, 0x3c, 0x0b, 0x29, 0xb5, 0x7e, 0xa5, 0xe3, 0x5b, 0xbf, 0x6f, 0x01, 0x0a, 0xf9,\n\t0x6b, 0xa6, 0xcd, 0xef, 0x74, 0x34, 0x13, 0xae, 0x5e, 0x23, 0x06, 0x4c, 0x60, 0x7a, 0xcb, 0xc4,\n\t0x42, 0x25, 0xf0, 0x66, 0x2c, 0xfa, 0x50, 0xd6, 0xa2, 0xc7, 0x4a, 0x3a, 0xc3, 0xc9, 0x92, 0xce,\n\t0x35, 0xa8, 0x72, 0x93, 0xd2, 0x4b, 0x80, 0x84, 0x0e, 0xc2, 0x08, 0x75, 0x10, 0xe6, 0xd9, 0x7c,\n\t0x24, 0x3b, 0xa1, 0x7f, 0xa0, 0xc1, 0x64, 0x54, 0xba, 0xa0, 0x29, 0x13, 0x56, 0x0b, 0x79, 0x33,\n\t0x4f, 0x1b, 0x77, 0x3c, 0xc3, 0xf1, 0x89, 0x29, 0x4b, 0xa4, 0x09, 0x26, 0xcc, 0xd8, 0x13, 0xfa,\n\t0x04, 0xce, 0x08, 0x12, 0x32, 0x3d, 0x13, 0x3e, 0xd6, 0x8f, 0x09, 0x5f, 0xcc, 0x88, 0x7b, 0x64,\n\t0xcd, 0x73, 0xbc, 0x4f, 0xc8, 0xf3, 0x3e, 0x57, 0x61, 0x22, 0x61, 0xf3, 0xc6, 0xa9, 0xcd, 0x1b,\n\t0xdf, 0x8d, 0x19, 0xbb, 0x5b, 0x50, 0xe9, 0x5d, 0x2b, 0x2d, 0x89, 0x4d, 0x14, 0x96, 0xc4, 0x26,\n\t0x23, 0x08, 0x5a, 0x11, 0x7b, 0x1f, 0x26, 0xc2, 0xbb, 0xa6, 0x08, 0x26, 0x0b, 0x11, 0x8c, 0xf3,\n\t0xf5, 0x14, 0xdc, 0x80, 0x91, 0xe7, 0x5d, 0x4c, 0x8d, 0x6c, 0x85, 0xe6, 0x7f, 0xee, 0xe5, 0x66,\n\t0xc1, 0x0b, 0xb5, 0x88, 0xa6, 0x28, 0x2c, 0xec, 0xb3, 0xbc, 0x77, 0x88, 0x37, 0xe3, 0x0b, 0x4e,\n\t0x65, 0x7c, 0xc1, 0xda, 0x27, 0x30, 0x11, 0x87, 0x15, 0xa4, 0xc2, 0xaf, 0xc5, 0x53, 0xe1, 0x79,\n\t0x29, 0x92, 0x50, 0x31, 0x59, 0xaa, 0x24, 0x96, 0x2e, 0xef, 0x99, 0xd2, 0x30, 0x31, 0xf6, 0x8d,\n\t0x29, 0xcd, 0x98, 0xd2, 0x38, 0x6b, 0x84, 0xa6, 0xf4, 0xe7, 0xe5, 0xd0, 0x94, 0x0a, 0xb9, 0xc8,\n\t0x4d, 0xe9, 0x87, 0x30, 0x95, 0x32, 0x55, 0x52, 0x63, 0xca, 0x93, 0x19, 0xd4, 0xd8, 0x68, 0x95,\n\t0xa4, 0x29, 0xcb, 0x08, 0x77, 0x69, 0x30, 0xe1, 0x8e, 0x59, 0xae, 0x72, 0xd2, 0x72, 0x7d, 0x02,\n\t0xcb, 0x49, 0xc5, 0xd3, 0xdd, 0x96, 0x1e, 0xec, 0x5b, 0xbe, 0x1e, 0xaf, 0x5e, 0xcb, 0xb7, 0xaa,\n\t0x25, 0x14, 0xf1, 0x51, 0x6b, 0x67, 0xdf, 0xf2, 0x6f, 0x71, 0xfc, 0x0d, 0x98, 0xd9, 0xc7, 0x86,\n\t0x17, 0xec, 0x62, 0x23, 0xd0, 0x4d, 0x1c, 0x18, 0x96, 0xed, 0xf3, 0x84, 0x8f, 0x3c, 0x41, 0x38,\n\t0x1d, 0x81, 0x6d, 0x31, 0xa8, 0xec, 0xab, 0x69, 0xf8, 0x78, 0xaf, 0xa6, 0x0b, 0x30, 0x15, 0xe1,\n\t0x61, 0x62, 0x4d, 0x6d, 0xf4, 0x98, 0x16, 0x39, 0x46, 0x5b, 0x74, 0x54, 0xfd, 0x73, 0x05, 0x5e,\n\t0x61, 0xb7, 0x99, 0x50, 0x76, 0x5e, 0x84, 0xee, 0xe9, 0x8b, 0x96, 0x4e, 0x2a, 0x5e, 0xcb, 0x4b,\n\t0x2a, 0x16, 0xa1, 0xea, 0x33, 0xbb, 0xf8, 0x77, 0x65, 0x78, 0x55, 0x8e, 0x8d, 0x8b, 0x20, 0xee,\n\t0xbd, 0xff, 0x3c, 0x3e, 0xc6, 0x49, 0xbc, 0x7e, 0x7c, 0xeb, 0xa6, 0x4d, 0xf9, 0x29, 0x49, 0xff,\n\t0x91, 0x02, 0xcb, 0xbd, 0xb4, 0x3c, 0xf1, 0xa1, 0x4d, 0xcb, 0xef, 0x18, 0x41, 0x73, 0x5f, 0xb7,\n\t0xdd, 0xa6, 0x61, 0xdb, 0x47, 0xd5, 0x12, 0xb5, 0xa9, 0x9f, 0x48, 0x76, 0x2d, 0x3e, 0x4e, 0xbd,\n\t0x97, 0xb7, 0xdf, 0x71, 0xb7, 0xf8, 0x0e, 0xf7, 0xd9, 0x06, 0xcc, 0xd4, 0x2e, 0x19, 0xf9, 0x2b,\n\t0x6a, 0xbf, 0x03, 0x2b, 0x45, 0x08, 0x04, 0xf6, 0x76, 0x2b, 0x69, 0x6f, 0xc5, 0x55, 0x81, 0xd0,\n\t0x0c, 0x50, 0x5c, 0x21, 0x62, 0xfa, 0x66, 0x8e, 0xd9, 0xde, 0x1f, 0x28, 0xc4, 0xf6, 0x66, 0x8e,\n\t0x79, 0xd7, 0xb0, 0xec, 0x9e, 0x2c, 0xf5, 0x59, 0x4e, 0x2a, 0xc2, 0xd3, 0xa7, 0x20, 0xbd, 0x42,\n\t0xec, 0x58, 0x2e, 0x26, 0x9e, 0xac, 0xfe, 0x53, 0x05, 0xd4, 0xac, 0xb5, 0xfb, 0x20, 0x54, 0xcf,\n\t0x90, 0xf2, 0x27, 0x69, 0xca, 0xaf, 0xe6, 0x50, 0x5e, 0x84, 0xa9, 0x4f, 0xda, 0x1f, 0x13, 0xe5,\n\t0x94, 0xe0, 0xe2, 0xb2, 0xf9, 0x3a, 0x4c, 0x37, 0x0d, 0xa7, 0x89, 0xa3, 0x37, 0x00, 0x66, 0xef,\n\t0xb4, 0x51, 0x6d, 0x8a, 0x8d, 0x6b, 0xe1, 0x70, 0x5c, 0xdf, 0xe3, 0x38, 0x4f, 0xa8, 0xef, 0x32,\n\t0x54, 0x7d, 0x1e, 0xf5, 0xb5, 0x48, 0xdd, 0x73, 0x90, 0xc5, 0x0a, 0x96, 0x82, 0x85, 0x27, 0x91,\n\t0xb0, 0x5c, 0x3c, 0x03, 0x4b, 0x98, 0x08, 0x53, 0x42, 0xc2, 0xb2, 0x07, 0xa4, 0xf7, 0xd3, 0xa3,\n\t0xbc, 0x6f, 0x09, 0x2b, 0xc2, 0xd4, 0x27, 0xed, 0xe7, 0xc5, 0xe2, 0x10, 0xe1, 0xe2, 0xd4, 0xff,\n\t0xbd, 0x02, 0xe7, 0x34, 0xdc, 0x76, 0x0f, 0x30, 0xeb, 0x44, 0xf8, 0xaa, 0xe4, 0xf1, 0x92, 0x8e,\n\t0x51, 0x39, 0xe5, 0x18, 0xa9, 0x2a, 0x91, 0x95, 0x3c, 0xaa, 0xf9, 0xd1, 0xfe, 0xb1, 0x04, 0xe7,\n\t0xf9, 0x11, 0xd8, 0xb1, 0x73, 0xcb, 0xe0, 0xd2, 0x03, 0x1a, 0x50, 0x49, 0xea, 0x20, 0x3f, 0xdc,\n\t0xf5, 0x9c, 0xfb, 0xeb, 0x63, 0x43, 0x6d, 0x32, 0xa1, 0xbd, 0x68, 0x17, 0x16, 0xa2, 0x4e, 0x03,\n\t0x61, 0x3b, 0x9f, 0xb8, 0x08, 0x7d, 0x87, 0xc3, 0xa4, 0x8a, 0xd0, 0x58, 0x34, 0x3c, 0x70, 0x97,\n\t0xc1, 0x1a, 0xbc, 0x56, 0x74, 0x16, 0xce, 0xe7, 0x7f, 0x56, 0x60, 0x29, 0x4c, 0x1c, 0x09, 0x02,\n\t0xf9, 0x97, 0x22, 0x3e, 0x17, 0x61, 0xc6, 0xf2, 0xf5, 0x64, 0x77, 0x1d, 0xe5, 0xe5, 0xa8, 0x36,\n\t0x65, 0xf9, 0x77, 0xe3, 0x7d, 0x73, 0xea, 0x32, 0x9c, 0x11, 0x93, 0xcf, 0xcf, 0xf7, 0x39, 0x75,\n\t0x58, 0x88, 0xb1, 0x4e, 0x16, 0xce, 0x33, 0xa6, 0xf5, 0x65, 0x1c, 0x74, 0x15, 0x26, 0x78, 0xeb,\n\t0x24, 0x36, 0x63, 0xb9, 0xdc, 0x68, 0xac, 0x61, 0xa2, 0x8f, 0xe1, 0x74, 0x33, 0x24, 0x35, 0xb6,\n\t0xf5, 0xa9, 0x81, 0xb6, 0x46, 0x11, 0x8a, 0xde, 0xde, 0xf7, 0x61, 0x3a, 0xd6, 0x0e, 0xc9, 0x82,\n\t0x84, 0xa1, 0x7e, 0x83, 0x84, 0xa9, 0x1e, 0x28, 0x8b, 0x12, 0xce, 0x02, 0x84, 0xee, 0x9e, 0x65,\n\t0x52, 0xf7, 0xb8, 0xac, 0x8d, 0xf1, 0x91, 0x86, 0xa9, 0x5e, 0x20, 0xca, 0x2c, 0xbd, 0x04, 0x7e,\n\t0x5d, 0xff, 0x5e, 0x82, 0xaa, 0xc6, 0x7b, 0x85, 0x31, 0x45, 0xed, 0x3f, 0xdd, 0x78, 0x99, 0x57,\n\t0xf4, 0x5b, 0x30, 0x27, 0xaa, 0x1c, 0x87, 0x1d, 0x20, 0x03, 0x94, 0x8e, 0x4f, 0x67, 0x4b, 0xc7,\n\t0x3e, 0x7a, 0x07, 0x86, 0x29, 0xeb, 0x7d, 0x7e, 0xa3, 0xe2, 0xd4, 0xc8, 0x96, 0x11, 0x18, 0xb7,\n\t0x6d, 0x77, 0x57, 0xe3, 0x8b, 0xd1, 0x26, 0x54, 0x1c, 0x7c, 0xa8, 0x7b, 0x5d, 0x7e, 0x73, 0x61,\n\t0x60, 0x53, 0x00, 0x3e, 0xe1, 0xe0, 0x43, 0xad, 0xcb, 0xae, 0xcc, 0x57, 0x97, 0x60, 0x51, 0xc0,\n\t0x6a, 0x7e, 0x11, 0xdf, 0x53, 0x60, 0x7e, 0xfb, 0xc8, 0x69, 0x6e, 0xef, 0x1b, 0x9e, 0xc9, 0x33,\n\t0xa4, 0xfc, 0x1a, 0xce, 0x43, 0xc5, 0x77, 0xbb, 0x5e, 0x13, 0xeb, 0xbc, 0x85, 0x9c, 0xdf, 0xc5,\n\t0x24, 0x1b, 0xdd, 0x64, 0x83, 0x68, 0x11, 0x46, 0x7d, 0x02, 0x1c, 0xbe, 0xdf, 0x86, 0xb4, 0x11,\n\t0xfa, 0xdc, 0x30, 0x51, 0x1d, 0x4e, 0xd1, 0x58, 0xb2, 0x5c, 0x18, 0xe0, 0xd1, 0x75, 0xea, 0x22,\n\t0x2c, 0x64, 0x68, 0xe1, 0x74, 0xfe, 0x74, 0x08, 0x4e, 0x93, 0xb9, 0xf0, 0x3d, 0xf9, 0x32, 0x65,\n\t0xa5, 0x0a, 0x23, 0x61, 0x46, 0x8a, 0x69, 0x72, 0xf8, 0x48, 0x14, 0xbd, 0x17, 0xeb, 0x46, 0x79,\n\t0x84, 0x28, 0xef, 0x40, 0x78, 0x92, 0xcd, 0x43, 0x0d, 0x0d, 0x9a, 0x87, 0x92, 0x2b, 0x61, 0x26,\n\t0x92, 0x1f, 0x19, 0x2c, 0x92, 0xff, 0x90, 0x57, 0x7f, 0x7a, 0x41, 0x35, 0xc5, 0x32, 0x5a, 0x88,\n\t0x65, 0x86, 0x80, 0x45, 0xee, 0x31, 0xc5, 0x75, 0x05, 0x46, 0xc2, 0x88, 0x7c, 0xac, 0x8f, 0x88,\n\t0x3c, 0x5c, 0x1c, 0xcf, 0x26, 0x40, 0x32, 0x9b, 0x70, 0x13, 0x26, 0x58, 0x6d, 0x8a, 0x37, 0x8a,\n\t0x8f, 0xf7, 0xd1, 0x28, 0x3e, 0x4e, 0x4b, 0x56, 0xbc, 0x47, 0xfc, 0x2d, 0xa0, 0x7d, 0xde, 0xfc,\n\t0xd3, 0x09, 0xdd, 0x32, 0xb1, 0x13, 0x58, 0xc1, 0x11, 0xcd, 0x06, 0x8e, 0x69, 0x88, 0xcc, 0x7d,\n\t0x4c, 0xa7, 0x1a, 0x7c, 0x06, 0x3d, 0x84, 0xa9, 0x94, 0x69, 0xe0, 0x99, 0xbf, 0xf3, 0x7d, 0x19,\n\t0x05, 0xad, 0x92, 0x34, 0x08, 0xea, 0x3c, 0xcc, 0x26, 0x25, 0x99, 0x8b, 0xf8, 0x9f, 0x28, 0xb0,\n\t0x14, 0x76, 0xde, 0x7d, 0x45, 0x3c, 0x3c, 0xf5, 0x8f, 0x15, 0x38, 0x23, 0xa6, 0x89, 0x07, 0x3f,\n\t0x97, 0x61, 0xbe, 0xcd, 0xc6, 0x59, 0x5d, 0x46, 0xb7, 0x1c, 0xbd, 0x69, 0x34, 0xf7, 0x31, 0xa7,\n\t0xf0, 0x74, 0x3b, 0x06, 0xd5, 0x70, 0x36, 0xc9, 0x14, 0x7a, 0x17, 0x16, 0x33, 0x40, 0xa6, 0x11,\n\t0x18, 0xbb, 0x86, 0x1f, 0x36, 0xe0, 0xce, 0x27, 0xe1, 0xb6, 0xf8, 0xac, 0x7a, 0x06, 0x6a, 0x21,\n\t0x3d, 0x9c, 0x9f, 0x1f, 0xb8, 0x51, 0xeb, 0x94, 0xfa, 0xbb, 0xa5, 0x1e, 0x0b, 0x13, 0xd3, 0x9c,\n\t0xda, 0x35, 0x98, 0x76, 0xba, 0xed, 0x5d, 0xec, 0xe9, 0x6e, 0x4b, 0xa7, 0x56, 0xca, 0xa7, 0x74,\n\t0x0e, 0x69, 0x15, 0x36, 0xfe, 0xa8, 0x45, 0x8d, 0x8f, 0x4f, 0x98, 0x1d, 0x5a, 0x35, 0x9f, 0xa6,\n\t0x16, 0x86, 0xb4, 0x51, 0x6e, 0xd6, 0x7c, 0xd4, 0x80, 0x09, 0x7e, 0x13, 0xec, 0xa8, 0xe2, 0x2e,\n\t0xd3, 0x50, 0x1c, 0x58, 0xae, 0x87, 0x9e, 0x9c, 0xfa, 0x7e, 0xe3, 0x66, 0x6f, 0x00, 0x5d, 0x81,\n\t0x05, 0xb6, 0x4f, 0xd3, 0x75, 0x02, 0xcf, 0xb5, 0x6d, 0xec, 0x51, 0x9e, 0x74, 0xd9, 0x9b, 0x62,\n\t0x4c, 0x9b, 0xa3, 0xd3, 0x9b, 0xd1, 0x2c, 0xb3, 0x8b, 0x54, 0x43, 0x4c, 0xd3, 0xc3, 0xbe, 0xcf,\n\t0x13, 0x92, 0xe1, 0xa3, 0x5a, 0x87, 0x19, 0x56, 0xd9, 0x22, 0x70, 0xa1, 0xec, 0xc4, 0x8d, 0xb4,\n\t0x92, 0x30, 0xd2, 0xea, 0x2c, 0xa0, 0xf8, 0x7a, 0x2e, 0x8c, 0xff, 0xa9, 0xc0, 0x0c, 0x73, 0xde,\n\t0xe3, 0x5e, 0x62, 0x3e, 0x1a, 0x74, 0x83, 0x57, 0x81, 0xa3, 0xa2, 0x77, 0x65, 0xe3, 0x5c, 0x0e,\n\t0x43, 0x08, 0x46, 0x9a, 0x35, 0xa3, 0x75, 0x60, 0x9a, 0x31, 0x8b, 0xe5, 0x5e, 0xcb, 0x89, 0xdc,\n\t0xeb, 0x26, 0x4c, 0x1d, 0x58, 0xbe, 0xb5, 0x6b, 0xd9, 0x56, 0x70, 0xc4, 0x2c, 0x51, 0x71, 0xba,\n\t0xb0, 0xd2, 0x03, 0xa1, 0x66, 0x68, 0x15, 0x26, 0xf8, 0x2b, 0x4c, 0x77, 0x0c, 0x6e, 0x71, 0xc7,\n\t0xb4, 0x71, 0x3e, 0xf6, 0xd0, 0x68, 0x63, 0xc2, 0x85, 0xf8, 0x71, 0x39, 0x17, 0xbe, 0x4f, 0xb9,\n\t0xe0, 0xe3, 0xe0, 0x49, 0x17, 0x77, 0x71, 0x1f, 0x5c, 0x48, 0xef, 0x54, 0xca, 0xec, 0x94, 0x64,\n\t0x54, 0x79, 0x40, 0x46, 0x31, 0x3a, 0x7b, 0x04, 0x71, 0x3a, 0x7f, 0xa8, 0xc0, 0x6c, 0x28, 0xf7,\n\t0x5f, 0x19, 0x52, 0x1f, 0xc1, 0x5c, 0x8a, 0x26, 0xae, 0x85, 0x57, 0x60, 0xa1, 0xe3, 0xb9, 0x4d,\n\t0xec, 0xfb, 0x96, 0xb3, 0xa7, 0xd3, 0xaf, 0xca, 0x98, 0x1d, 0x20, 0xca, 0x58, 0x26, 0x32, 0xdf,\n\t0x9b, 0xa6, 0x90, 0xd4, 0x08, 0xf8, 0xea, 0xe7, 0x0a, 0x9c, 0xbd, 0x87, 0x03, 0xad, 0xf7, 0x8d,\n\t0xd9, 0x03, 0xec, 0xfb, 0xc6, 0x1e, 0x8e, 0x5c, 0x96, 0x9b, 0x30, 0x4c, 0x0b, 0x40, 0x0c, 0xd1,\n\t0xf8, 0xc6, 0x85, 0x1c, 0x6a, 0x63, 0x28, 0x68, 0x75, 0x48, 0xe3, 0x60, 0x7d, 0x30, 0x85, 0xd8,\n\t0x98, 0xe5, 0x3c, 0x2a, 0xf8, 0x01, 0x9f, 0x43, 0x85, 0x71, 0xbd, 0xcd, 0x67, 0x38, 0x39, 0x1f,\n\t0xe6, 0x26, 0x27, 0xe5, 0x08, 0xeb, 0x54, 0x37, 0xc3, 0x51, 0x96, 0x88, 0x9c, 0xf4, 0xe3, 0x63,\n\t0x35, 0x1b, 0x50, 0x76, 0x51, 0x3c, 0xd9, 0x38, 0xc4, 0x92, 0x8d, 0xdf, 0x49, 0x26, 0x1b, 0x2f,\n\t0x16, 0x33, 0x28, 0x22, 0x26, 0x96, 0x68, 0x6c, 0xc3, 0xca, 0x3d, 0x1c, 0x6c, 0xdd, 0x7f, 0x22,\n\t0xb9, 0x8b, 0x06, 0x00, 0x53, 0x69, 0xa7, 0xe5, 0x86, 0x0c, 0xe8, 0x63, 0x3b, 0x22, 0x48, 0xd4,\n\t0x4c, 0x52, 0xd1, 0x23, 0xbf, 0x7c, 0xf5, 0x05, 0xac, 0x4a, 0xb6, 0xe3, 0x4c, 0xdf, 0x86, 0x99,\n\t0xd8, 0xd7, 0x87, 0xb4, 0x18, 0x19, 0x6e, 0xfb, 0x5a, 0x7f, 0xdb, 0x6a, 0xd3, 0x5e, 0x72, 0xc0,\n\t0x57, 0xff, 0x55, 0x81, 0x59, 0x0d, 0x1b, 0x9d, 0x8e, 0xcd, 0x22, 0xa2, 0xe8, 0x74, 0xf3, 0x30,\n\t0xcc, 0x33, 0xfb, 0xec, 0x3d, 0xc7, 0x9f, 0xe4, 0x1f, 0x2b, 0x88, 0x5f, 0xd2, 0xe5, 0x93, 0xfa,\n\t0xa3, 0xc7, 0x0b, 0x2e, 0xd4, 0x05, 0x98, 0x4b, 0x1d, 0x8d, 0x5b, 0x93, 0x9f, 0x28, 0xb0, 0xa4,\n\t0xe1, 0x96, 0x87, 0xfd, 0xfd, 0xa8, 0xc8, 0x41, 0xb8, 0xf1, 0x15, 0x3c, 0xbb, 0xba, 0x0c, 0x67,\n\t0xc4, 0xa4, 0xf2, 0xb3, 0xbc, 0x0b, 0x0b, 0x9b, 0x6e, 0xd7, 0x21, 0xc2, 0x93, 0x16, 0xd0, 0x65,\n\t0x80, 0x96, 0xeb, 0x35, 0xf1, 0x5d, 0x1c, 0x34, 0xf7, 0x79, 0xc6, 0x36, 0x36, 0xa2, 0x1a, 0x50,\n\t0xcd, 0x82, 0x72, 0x61, 0xbb, 0x03, 0x23, 0xd8, 0x09, 0x68, 0x2d, 0x97, 0x89, 0xd8, 0x1b, 0x39,\n\t0x22, 0xc6, 0xbd, 0x90, 0xad, 0xfb, 0x4f, 0x28, 0x2e, 0x5e, 0xaf, 0xe5, 0xb0, 0xea, 0x4f, 0x4a,\n\t0x30, 0xaf, 0x61, 0xc3, 0x14, 0x50, 0xb7, 0x01, 0xa7, 0xa2, 0xee, 0x88, 0xca, 0xc6, 0x72, 0x9e,\n\t0x6f, 0x71, 0xff, 0x09, 0xb5, 0xba, 0x74, 0xad, 0x2c, 0x14, 0xcb, 0x06, 0x73, 0x65, 0x51, 0x30,\n\t0xb7, 0x03, 0x55, 0xcb, 0x21, 0x2b, 0xac, 0x03, 0xac, 0x63, 0x27, 0xb2, 0x60, 0x7d, 0x76, 0x94,\n\t0xcd, 0x45, 0xc0, 0x77, 0x9c, 0xd0, 0x14, 0x35, 0x4c, 0x22, 0x18, 0x1d, 0x82, 0x84, 0xd6, 0xa4,\n\t0x87, 0x28, 0x61, 0xa3, 0x64, 0x60, 0xdb, 0xfa, 0x14, 0xa3, 0xd7, 0x60, 0x8a, 0xf6, 0x45, 0xd0,\n\t0x15, 0xac, 0x7c, 0x3f, 0x4c, 0xcb, 0xf7, 0xb4, 0x5d, 0xe2, 0xb1, 0xb1, 0x87, 0x59, 0x37, 0xdf,\n\t0xdf, 0x96, 0x60, 0x21, 0xc3, 0x2b, 0x7e, 0x1d, 0xc7, 0x61, 0x96, 0xd0, 0x5e, 0x94, 0x4e, 0x66,\n\t0x2f, 0xd0, 0x77, 0x61, 0x3e, 0x83, 0x34, 0xcc, 0x11, 0x0e, 0x6a, 0x00, 0x67, 0xd3, 0xd8, 0x69,\n\t0x8a, 0x50, 0xc0, 0xae, 0x53, 0x22, 0x76, 0xfd, 0x42, 0x81, 0x85, 0xc7, 0x5d, 0x6f, 0x0f, 0x7f,\n\t0xbd, 0x65, 0x4b, 0xad, 0x41, 0x35, 0x7b, 0x4c, 0xae, 0xfc, 0x5f, 0x94, 0x60, 0xe1, 0x01, 0xfe,\n\t0xda, 0xf3, 0xe0, 0x7f, 0x46, 0xbf, 0x6e, 0x43, 0x35, 0xcb, 0x2b, 0xae, 0x5f, 0x02, 0x1c, 0x8a,\n\t0x08, 0xc7, 0x67, 0x0a, 0x9c, 0x79, 0xe8, 0x06, 0x56, 0xeb, 0x88, 0x84, 0xdb, 0xee, 0x01, 0xf6,\n\t0x1e, 0x18, 0x24, 0x96, 0x8e, 0xb8, 0xfe, 0x5d, 0x98, 0x6f, 0xf1, 0x19, 0xbd, 0x4d, 0xa7, 0xf4,\n\t0x84, 0xc3, 0x96, 0xa7, 0x1f, 0x49, 0x74, 0xcc, 0x67, 0x9b, 0x6d, 0x65, 0x07, 0x7d, 0xf5, 0x1c,\n\t0x9c, 0xcd, 0xa1, 0x80, 0x0b, 0x85, 0x01, 0x4b, 0xf7, 0x70, 0xb0, 0xe9, 0xb9, 0xbe, 0xcf, 0x6f,\n\t0x25, 0xf1, 0x72, 0x4b, 0x04, 0x7e, 0x4a, 0x2a, 0xf0, 0x3b, 0x0f, 0x95, 0xc0, 0xf0, 0xf6, 0x70,\n\t0x10, 0xdd, 0x32, 0x7b, 0xcd, 0x4d, 0xb2, 0x51, 0x8e, 0x4f, 0xfd, 0x65, 0x19, 0xce, 0x88, 0xf7,\n\t0xe0, 0xfc, 0x6c, 0x13, 0x3c, 0xc4, 0x34, 0xec, 0x1e, 0xb1, 0x30, 0x94, 0x1f, 0xff, 0x9e, 0xcc,\n\t0x41, 0xcc, 0x45, 0x47, 0x9d, 0x6f, 0xff, 0xf6, 0x11, 0x75, 0x00, 0xd9, 0x1b, 0x66, 0x22, 0x88,\n\t0x0d, 0xa1, 0xcf, 0x14, 0x98, 0x6b, 0xd1, 0x82, 0x98, 0xde, 0x34, 0xba, 0x3e, 0xee, 0x6d, 0xcb,\n\t0xec, 0xdd, 0x83, 0xe3, 0x6d, 0xcb, 0x6a, 0x6c, 0x9b, 0x04, 0x63, 0x62, 0x73, 0xd4, 0xca, 0x4c,\n\t0xd4, 0x3a, 0x30, 0x93, 0xa1, 0x52, 0xe0, 0x9e, 0xde, 0x49, 0xba, 0xa7, 0xeb, 0x39, 0xe2, 0x90,\n\t0xa6, 0x89, 0x5f, 0x5e, 0xdc, 0x47, 0xad, 0x75, 0x60, 0x21, 0x87, 0x40, 0xc1, 0xbe, 0x37, 0xe3,\n\t0xfb, 0x56, 0x72, 0xd3, 0xbd, 0xf7, 0x70, 0xd0, 0x2b, 0x2e, 0x52, 0xbc, 0x71, 0xaf, 0xf8, 0x3f,\n\t0x14, 0x58, 0xe3, 0xe5, 0xbc, 0x0c, 0xd3, 0x32, 0x75, 0x08, 0x49, 0x64, 0xd6, 0x9f, 0x94, 0xa1,\n\t0xa7, 0x4c, 0x88, 0xa2, 0xbe, 0x8b, 0x30, 0x57, 0xdd, 0x3f, 0xd3, 0x78, 0xb7, 0xc5, 0x64, 0x10,\n\t0x7b, 0xf2, 0xd1, 0xab, 0x30, 0xd9, 0x22, 0x0e, 0xd0, 0x43, 0xcc, 0x7c, 0x29, 0x5e, 0x7e, 0x4a,\n\t0x0e, 0xaa, 0x1e, 0xbc, 0xde, 0xc7, 0x59, 0x23, 0x77, 0x69, 0x28, 0xf4, 0xc7, 0x8f, 0x77, 0xad,\n\t0x14, 0x5a, 0x7d, 0x87, 0x7e, 0xd3, 0x16, 0x2a, 0x36, 0x7d, 0x49, 0xf6, 0x91, 0x1b, 0x53, 0x03,\n\t0xfa, 0xdd, 0x56, 0x12, 0x2c, 0x72, 0x1c, 0xe6, 0x7a, 0x65, 0x97, 0x30, 0x11, 0xd3, 0xe5, 0x7d,\n\t0x54, 0x43, 0x5a, 0xaf, 0x26, 0xb3, 0xcd, 0xb2, 0x30, 0x5d, 0x87, 0xe6, 0xc5, 0xc3, 0xaf, 0x2e,\n\t0x79, 0x0a, 0x89, 0xe5, 0x87, 0x26, 0xf9, 0x28, 0xcb, 0x20, 0xa9, 0x0d, 0x98, 0xd7, 0x8c, 0x00,\n\t0xdb, 0x56, 0xdb, 0x0a, 0x3e, 0xea, 0x98, 0xb1, 0x44, 0xde, 0x3a, 0x9c, 0x32, 0x8d, 0xc0, 0xe0,\n\t0xcc, 0x58, 0xca, 0x6b, 0xc4, 0xbc, 0xe5, 0x1c, 0x69, 0x74, 0xa1, 0xfa, 0x21, 0x2c, 0x64, 0x50,\n\t0xf1, 0x03, 0x0c, 0x8a, 0x6b, 0xe3, 0x9f, 0xea, 0x00, 0xdc, 0x29, 0xbd, 0xf5, 0xb8, 0x81, 0xfe,\n\t0x50, 0x81, 0x79, 0xf1, 0x47, 0xed, 0xe8, 0xca, 0xf1, 0xfe, 0x85, 0xa2, 0x76, 0x75, 0x60, 0x38,\n\t0x7e, 0x96, 0x3f, 0x52, 0x60, 0x21, 0xe7, 0x5f, 0x0f, 0xd0, 0xd5, 0xa2, 0x7f, 0x0c, 0xc8, 0xa3,\n\t0xe6, 0xda, 0xe0, 0x80, 0x9c, 0x9c, 0x1f, 0x2b, 0xb0, 0x52, 0xf4, 0xe5, 0x3f, 0xfa, 0xce, 0x49,\n\t0xff, 0xc9, 0xa0, 0x76, 0xeb, 0x04, 0x18, 0x38, 0xa5, 0xe4, 0x12, 0xc5, 0xdf, 0xf4, 0x4b, 0x2e,\n\t0x51, 0xfa, 0x5f, 0x02, 0x92, 0x4b, 0x2c, 0xf8, 0xf3, 0x80, 0x3f, 0x53, 0xa0, 0x96, 0xff, 0xe5,\n\t0x3b, 0xca, 0xef, 0x0a, 0x2b, 0xfc, 0x47, 0x80, 0xda, 0x7b, 0xc7, 0x82, 0xe5, 0x74, 0xfd, 0x50,\n\t0x81, 0xc5, 0xdc, 0xef, 0xda, 0xd1, 0xbb, 0xb9, 0xa8, 0x8b, 0x3e, 0xab, 0xaf, 0x5d, 0x3f, 0x0e,\n\t0x28, 0x27, 0xca, 0x81, 0xc9, 0xc4, 0x07, 0xcf, 0xe8, 0xcd, 0x5c, 0x64, 0xa2, 0xef, 0xaa, 0x6b,\n\t0xf5, 0x7e, 0x97, 0xf3, 0xfd, 0x3e, 0x53, 0xe0, 0xb4, 0xe0, 0xab, 0x61, 0x74, 0x59, 0x7e, 0xdb,\n\t0xc2, 0xef, 0x94, 0x6b, 0x6f, 0x0f, 0x06, 0xc4, 0x49, 0x08, 0x60, 0x2a, 0xf5, 0x11, 0x2d, 0x5a,\n\t0x97, 0xb9, 0x1f, 0x82, 0x4a, 0x48, 0xed, 0xad, 0xfe, 0x01, 0xf8, 0xae, 0x87, 0x30, 0x9d, 0xfe,\n\t0x12, 0x0c, 0xe5, 0x63, 0xc9, 0xf9, 0x56, 0xae, 0x76, 0x69, 0x00, 0x88, 0x98, 0xd8, 0xe5, 0xf6,\n\t0x3b, 0x4a, 0xc4, 0xae, 0xe8, 0x6b, 0x94, 0xda, 0x09, 0xda, 0x2b, 0xd1, 0x5f, 0x2a, 0x70, 0x46,\n\t0xd6, 0x0e, 0x89, 0x6e, 0x1c, 0xb3, 0x8b, 0x92, 0x91, 0xf6, 0xfe, 0x89, 0x7a, 0x30, 0x39, 0xcb,\n\t0x72, 0x7a, 0x06, 0xa5, 0x2c, 0x93, 0x77, 0x2c, 0x4a, 0x59, 0x56, 0xd0, 0xa2, 0x18, 0xbb, 0x47,\n\t0x41, 0x43, 0x76, 0xe1, 0x3d, 0xe6, 0xb7, 0xc2, 0x17, 0xde, 0xa3, 0xac, 0xff, 0x3b, 0x76, 0x8f,\n\t0xc2, 0xb6, 0xbd, 0xe2, 0x7b, 0x94, 0xb5, 0x0e, 0x16, 0xdf, 0xa3, 0xb4, 0x57, 0x30, 0x7e, 0x8f,\n\t0xd9, 0xce, 0xbc, 0xe2, 0x7b, 0xcc, 0xed, 0x0b, 0x2c, 0xbe, 0xc7, 0xfc, 0x46, 0x40, 0xf4, 0x17,\n\t0x34, 0xb7, 0x99, 0xdb, 0x72, 0x87, 0xde, 0x1b, 0xe8, 0xcc, 0xc9, 0xa6, 0xbf, 0xda, 0x8d, 0xe3,\n\t0x01, 0x27, 0x48, 0xcb, 0xed, 0x37, 0x95, 0x92, 0x56, 0xd4, 0xf1, 0x2a, 0x25, 0xad, 0xb8, 0xc5,\n\t0xf5, 0xaf, 0x15, 0x58, 0x96, 0x37, 0x9a, 0xa1, 0x6f, 0x4b, 0x36, 0xe8, 0xa3, 0xdb, 0xae, 0x76,\n\t0xf3, 0xd8, 0xf0, 0x9c, 0xc6, 0xef, 0x2b, 0x50, 0xcd, 0x6b, 0x37, 0x44, 0xd7, 0x24, 0xd8, 0xa5,\n\t0x7d, 0x95, 0xb5, 0x77, 0x8f, 0x01, 0xc9, 0x29, 0xfa, 0x5c, 0x81, 0x59, 0x51, 0xd3, 0x1a, 0xca,\n\t0x7f, 0x73, 0x4a, 0x5a, 0xf4, 0x6a, 0xef, 0x0c, 0x08, 0xc5, 0xa9, 0xf8, 0x2b, 0xfa, 0xe7, 0x53,\n\t0x92, 0xa6, 0x2c, 0xf4, 0x7e, 0x81, 0x6c, 0xc8, 0x3b, 0xea, 0x6a, 0xdf, 0x3e, 0x2e, 0x38, 0x27,\n\t0xf0, 0x53, 0x98, 0xc9, 0xf4, 0x27, 0xa1, 0x4b, 0x12, 0xa4, 0xe2, 0xb6, 0xb1, 0xda, 0xc6, 0x20,\n\t0x20, 0x3d, 0x6f, 0x24, 0xd5, 0x71, 0x24, 0xf1, 0x46, 0xc4, 0x7d, 0x52, 0x12, 0x6f, 0x24, 0xa7,\n\t0x99, 0x09, 0x3d, 0x83, 0x89, 0x78, 0x07, 0x08, 0xfa, 0x96, 0x14, 0x43, 0xaa, 0xe5, 0xa9, 0xf6,\n\t0x66, 0x9f, 0xab, 0x63, 0x52, 0x28, 0x6a, 0xe1, 0x90, 0x48, 0xa1, 0xa4, 0x0b, 0x45, 0x22, 0x85,\n\t0xd2, 0x3e, 0x11, 0xe2, 0x79, 0x0a, 0x3a, 0x33, 0x24, 0x9e, 0x67, 0x7e, 0x9b, 0x47, 0xed, 0xed,\n\t0xc1, 0x80, 0xa2, 0x4f, 0x55, 0xa0, 0xd7, 0xe8, 0x80, 0x2e, 0xe6, 0xe2, 0xc8, 0x74, 0x4f, 0xd4,\n\t0xde, 0xe8, 0x6b, 0x6d, 0x6f, 0x9b, 0x5e, 0x27, 0x81, 0x64, 0x9b, 0x4c, 0x77, 0x85, 0x64, 0x9b,\n\t0x6c, 0x6b, 0x02, 0xdb, 0x26, 0x6c, 0x04, 0x90, 0x6e, 0x93, 0x6a, 0x5f, 0x90, 0x6e, 0x93, 0xee,\n\t0x2c, 0x20, 0x11, 0x4a, 0xa2, 0x88, 0x2f, 0x89, 0x50, 0x44, 0x0d, 0x08, 0x92, 0x08, 0x45, 0xdc,\n\t0x1b, 0x40, 0x42, 0x59, 0x71, 0x31, 0x5c, 0x12, 0xca, 0x4a, 0x9b, 0x02, 0x24, 0xa1, 0x6c, 0x41,\n\t0x19, 0x9f, 0x38, 0x30, 0xb9, 0x75, 0x67, 0x89, 0x03, 0x53, 0x54, 0x1a, 0x97, 0x38, 0x30, 0xc5,\n\t0x65, 0x6e, 0x07, 0x26, 0x13, 0x55, 0x5b, 0xc9, 0x85, 0x88, 0x0a, 0xd7, 0x92, 0x0b, 0x11, 0x16,\n\t0x83, 0xa9, 0xf9, 0x10, 0x55, 0x58, 0x91, 0x2c, 0xfc, 0xcb, 0xad, 0x1d, 0x4b, 0xcc, 0x87, 0xac,\n\t0x8c, 0x4b, 0xe2, 0xb7, 0x74, 0x2d, 0x56, 0x12, 0xbf, 0xe5, 0x54, 0x7c, 0x25, 0xf1, 0x5b, 0x6e,\n\t0xa1, 0x37, 0x80, 0xa9, 0x54, 0xd1, 0x51, 0xf2, 0x82, 0x10, 0x97, 0x72, 0x25, 0x2f, 0x88, 0xbc,\n\t0x7a, 0x26, 0x09, 0x57, 0x53, 0x45, 0x2d, 0x59, 0xb8, 0x2a, 0x2e, 0xf3, 0xc9, 0xc2, 0xd5, 0x9c,\n\t0x8a, 0x19, 0xd9, 0x38, 0x5d, 0x04, 0x92, 0x6c, 0x9c, 0x53, 0x5b, 0x93, 0x6c, 0x9c, 0x5b, 0x61,\n\t0xfa, 0x03, 0x05, 0xe6, 0x84, 0x75, 0x1b, 0x94, 0x2f, 0x31, 0xb2, 0x4a, 0x53, 0xed, 0xca, 0xa0,\n\t0x60, 0x31, 0x79, 0x17, 0x55, 0x3d, 0x24, 0xf2, 0x2e, 0x29, 0x27, 0x49, 0xe4, 0x5d, 0x5a, 0x20,\n\t0xfa, 0x42, 0x89, 0xbe, 0x6a, 0xca, 0x4f, 0xaf, 0xa3, 0x5b, 0x45, 0xf1, 0x46, 0x61, 0x19, 0xa2,\n\t0x76, 0xfb, 0x24, 0x28, 0x12, 0x29, 0x9d, 0x78, 0x7e, 0x5d, 0x9e, 0xd2, 0x11, 0x24, 0xf0, 0xe5,\n\t0x29, 0x1d, 0x61, 0xea, 0x9e, 0x68, 0x66, 0x32, 0x29, 0x2e, 0xd3, 0x4c, 0x61, 0x26, 0x5e, 0xa6,\n\t0x99, 0xe2, 0x7c, 0xfb, 0xed, 0x3b, 0x3f, 0xfd, 0x72, 0x59, 0xf9, 0xd9, 0x97, 0xcb, 0xca, 0xbf,\n\t0x7d, 0xb9, 0xac, 0xfc, 0xfa, 0xd5, 0x3d, 0x2b, 0xd8, 0xef, 0xee, 0xd6, 0x9b, 0x6e, 0x7b, 0x3d,\n\t0xf1, 0xff, 0xe4, 0xf5, 0x3d, 0xec, 0xb0, 0x3f, 0xab, 0x8f, 0xfd, 0x5b, 0xfe, 0x7b, 0xfc, 0xe7,\n\t0xc1, 0xa5, 0xdd, 0x61, 0x3a, 0x77, 0xf9, 0xbf, 0x03, 0x00, 0x00, 0xff, 0xff, 0xc2, 0x7a, 0xca,\n\t0x50, 0x59, 0x5f, 0x00, 0x00,\n}\n\nfunc (m *StartWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *StartWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *StartWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k := range m.PartitionConfig {\n\t\t\tv := m.PartitionConfig[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x52\n\t\t}\n\t}\n\tif m.FirstDecisionTaskBackoff != nil {\n\t\t{\n\t\t\tsize, err := m.FirstDecisionTaskBackoff.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif m.LastCompletionResult != nil {\n\t\t{\n\t\t\tsize, err := m.LastCompletionResult.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.ContinuedFailure != nil {\n\t\t{\n\t\t\tsize, err := m.ContinuedFailure.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.ContinueAsNewInitiator != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ContinueAsNewInitiator))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.ExpirationTime != nil {\n\t\t{\n\t\t\tsize, err := m.ExpirationTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ParentExecutionInfo != nil {\n\t\t{\n\t\t\tsize, err := m.ParentExecutionInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *StartWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *StartWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *StartWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.RunId) > 0 {\n\t\ti -= len(m.RunId)\n\t\tcopy(dAtA[i:], m.RunId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RunId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SignalWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SignalWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SignalWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ChildWorkflowOnly {\n\t\ti--\n\t\tif m.ChildWorkflowOnly {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ExternalWorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.ExternalWorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SignalWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SignalWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SignalWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k := range m.PartitionConfig {\n\t\t\tv := m.PartitionConfig[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.RunId) > 0 {\n\t\ti -= len(m.RunId)\n\t\tcopy(dAtA[i:], m.RunId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RunId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.RunId) > 0 {\n\t\ti -= len(m.RunId)\n\t\tcopy(dAtA[i:], m.RunId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RunId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ChildWorkflowOnly {\n\t\ti--\n\t\tif m.ChildWorkflowOnly {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ExternalWorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.ExternalWorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *TerminateWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TerminateWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TerminateWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PendingDecision != nil {\n\t\t{\n\t\t\tsize, err := m.PendingDecision.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif len(m.PendingChildren) > 0 {\n\t\tfor iNdEx := len(m.PendingChildren) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.PendingChildren[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x22\n\t\t}\n\t}\n\tif len(m.PendingActivities) > 0 {\n\t\tfor iNdEx := len(m.PendingActivities) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.PendingActivities[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif m.WorkflowExecutionInfo != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecutionInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ExecutionConfiguration != nil {\n\t\t{\n\t\t\tsize, err := m.ExecutionConfiguration.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *QueryWorkflowRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *QueryWorkflowRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *QueryWorkflowRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *QueryWorkflowResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *QueryWorkflowResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *QueryWorkflowResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.QueryRejected != nil {\n\t\t{\n\t\t\tsize, err := m.QueryRejected.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.QueryResult != nil {\n\t\t{\n\t\t\tsize, err := m.QueryResult.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetStickyTaskListRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetStickyTaskListRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetStickyTaskListRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetStickyTaskListResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetStickyTaskListResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetStickyTaskListResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetMutableStateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetMutableStateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetMutableStateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.VersionHistoryItem != nil {\n\t\t{\n\t\t\tsize, err := m.VersionHistoryItem.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif len(m.CurrentBranchToken) > 0 {\n\t\ti -= len(m.CurrentBranchToken)\n\t\tcopy(dAtA[i:], m.CurrentBranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.CurrentBranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.ExpectedNextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ExpectedNextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetMutableStateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetMutableStateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetMutableStateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.HistorySize != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.HistorySize))\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x90\n\t}\n\tif m.IsStickyTaskListEnabled {\n\t\ti--\n\t\tif m.IsStickyTaskListEnabled {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x88\n\t}\n\tif m.VersionHistories != nil {\n\t\t{\n\t\t\tsize, err := m.VersionHistories.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x82\n\t}\n\tif m.WorkflowCloseState != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.WorkflowCloseState))\n\t\ti--\n\t\tdAtA[i] = 0x78\n\t}\n\tif m.WorkflowState != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.WorkflowState))\n\t\ti--\n\t\tdAtA[i] = 0x70\n\t}\n\tif len(m.CurrentBranchToken) > 0 {\n\t\ti -= len(m.CurrentBranchToken)\n\t\tcopy(dAtA[i:], m.CurrentBranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.CurrentBranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x6a\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.EventStoreVersion))\n\t\ti--\n\t\tdAtA[i] = 0x60\n\t}\n\tif m.StickyTaskListScheduleToStartTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.StickyTaskListScheduleToStartTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x5a\n\t}\n\tif len(m.ClientImpl) > 0 {\n\t\ti -= len(m.ClientImpl)\n\t\tcopy(dAtA[i:], m.ClientImpl)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientImpl)))\n\t\ti--\n\t\tdAtA[i] = 0x52\n\t}\n\tif len(m.ClientFeatureVersion) > 0 {\n\t\ti -= len(m.ClientFeatureVersion)\n\t\tcopy(dAtA[i:], m.ClientFeatureVersion)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientFeatureVersion)))\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif len(m.ClientLibraryVersion) > 0 {\n\t\ti -= len(m.ClientLibraryVersion)\n\t\tcopy(dAtA[i:], m.ClientLibraryVersion)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientLibraryVersion)))\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.StickyTaskList != nil {\n\t\t{\n\t\t\tsize, err := m.StickyTaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.LastFirstEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.LastFirstEventId))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\t{\n\t\t\tsize, err := m.PreviousStartedEventId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.NextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollMutableStateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollMutableStateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollMutableStateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.CurrentBranchToken) > 0 {\n\t\ti -= len(m.CurrentBranchToken)\n\t\tcopy(dAtA[i:], m.CurrentBranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.CurrentBranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.ExpectedNextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ExpectedNextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollMutableStateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollMutableStateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollMutableStateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.WorkflowCloseState != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.WorkflowCloseState))\n\t\ti--\n\t\tdAtA[i] = 0x78\n\t}\n\tif m.WorkflowState != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.WorkflowState))\n\t\ti--\n\t\tdAtA[i] = 0x70\n\t}\n\tif m.VersionHistories != nil {\n\t\t{\n\t\t\tsize, err := m.VersionHistories.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x6a\n\t}\n\tif len(m.CurrentBranchToken) > 0 {\n\t\ti -= len(m.CurrentBranchToken)\n\t\tcopy(dAtA[i:], m.CurrentBranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.CurrentBranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x62\n\t}\n\tif m.StickyTaskListScheduleToStartTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.StickyTaskListScheduleToStartTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x5a\n\t}\n\tif len(m.ClientImpl) > 0 {\n\t\ti -= len(m.ClientImpl)\n\t\tcopy(dAtA[i:], m.ClientImpl)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientImpl)))\n\t\ti--\n\t\tdAtA[i] = 0x52\n\t}\n\tif len(m.ClientFeatureVersion) > 0 {\n\t\ti -= len(m.ClientFeatureVersion)\n\t\tcopy(dAtA[i:], m.ClientFeatureVersion)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientFeatureVersion)))\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif len(m.ClientLibraryVersion) > 0 {\n\t\ti -= len(m.ClientLibraryVersion)\n\t\tcopy(dAtA[i:], m.ClientLibraryVersion)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClientLibraryVersion)))\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.StickyTaskList != nil {\n\t\t{\n\t\t\tsize, err := m.StickyTaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.LastFirstEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.LastFirstEventId))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\t{\n\t\t\tsize, err := m.PreviousStartedEventId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.NextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PollRequest != nil {\n\t\t{\n\t\t\tsize, err := m.PollRequest.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif len(m.RequestId) > 0 {\n\t\ti -= len(m.RequestId)\n\t\tcopy(dAtA[i:], m.RequestId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RequestId)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.TaskId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskId))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ScheduleId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduleId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.HistorySize != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.HistorySize))\n\t\ti--\n\t\tdAtA[i] = 0x78\n\t}\n\tif len(m.Queries) > 0 {\n\t\tfor k := range m.Queries {\n\t\t\tv := m.Queries[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x72\n\t\t}\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x6a\n\t}\n\tif m.ScheduledTime != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x62\n\t}\n\tif len(m.BranchToken) > 0 {\n\t\ti -= len(m.BranchToken)\n\t\tcopy(dAtA[i:], m.BranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.BranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x5a\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.EventStoreVersion))\n\t\ti--\n\t\tdAtA[i] = 0x50\n\t}\n\tif m.WorkflowExecutionTaskList != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecutionTaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif m.DecisionInfo != nil {\n\t\t{\n\t\t\tsize, err := m.DecisionInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.StickyExecutionEnabled {\n\t\ti--\n\t\tif m.StickyExecutionEnabled {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x38\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.NextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.StartedEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.StartedEventId))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ScheduledEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduledEventId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\t{\n\t\t\tsize, err := m.PreviousStartedEventId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordActivityTaskStartedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordActivityTaskStartedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordActivityTaskStartedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PollRequest != nil {\n\t\t{\n\t\t\tsize, err := m.PollRequest.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif len(m.RequestId) > 0 {\n\t\ti -= len(m.RequestId)\n\t\tcopy(dAtA[i:], m.RequestId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RequestId)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.TaskId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskId))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ScheduleId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduleId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordActivityTaskStartedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordActivityTaskStartedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.WorkflowDomain) > 0 {\n\t\ti -= len(m.WorkflowDomain)\n\t\tcopy(dAtA[i:], m.WorkflowDomain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.WorkflowDomain)))\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\t{\n\t\t\tsize, err := m.HeartbeatDetails.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTimeOfThisAttempt.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ScheduledEvent != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ActivitiesToDispatchLocally) > 0 {\n\t\tfor k := range m.ActivitiesToDispatchLocally {\n\t\t\tv := m.ActivitiesToDispatchLocally[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif m.StartedResponse != nil {\n\t\t{\n\t\t\tsize, err := m.StartedResponse.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondDecisionTaskFailedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondDecisionTaskFailedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondDecisionTaskFailedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordActivityTaskHeartbeatResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordActivityTaskHeartbeatResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordActivityTaskHeartbeatResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.CancelRequested {\n\t\ti--\n\t\tif m.CancelRequested {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskCompletedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskCompletedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskCompletedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskFailedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskFailedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskFailedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskFailedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskFailedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskFailedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondActivityTaskCanceledResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondActivityTaskCanceledResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondActivityTaskCanceledResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RemoveSignalMutableStateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RemoveSignalMutableStateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RemoveSignalMutableStateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.RequestId) > 0 {\n\t\ti -= len(m.RequestId)\n\t\tcopy(dAtA[i:], m.RequestId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.RequestId)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RemoveSignalMutableStateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RemoveSignalMutableStateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RemoveSignalMutableStateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ChildWorkflowOnly {\n\t\ti--\n\t\tif m.ChildWorkflowOnly {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.ExternalExecutionInfo != nil {\n\t\t{\n\t\t\tsize, err := m.ExternalExecutionInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.CancelRequest != nil {\n\t\t{\n\t\t\tsize, err := m.CancelRequest.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RequestCancelWorkflowExecutionResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RequestCancelWorkflowExecutionResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ScheduleDecisionTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ScheduleDecisionTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ScheduleDecisionTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.IsFirstDecision {\n\t\ti--\n\t\tif m.IsFirstDecision {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ScheduleDecisionTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ScheduleDecisionTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ScheduleDecisionTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.StartedId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.StartedId))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.CompletionEvent != nil {\n\t\t{\n\t\t\tsize, err := m.CompletionEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.CompletedExecution != nil {\n\t\t{\n\t\t\tsize, err := m.CompletedExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.InitiatedId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.InitiatedId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RecordChildExecutionCompletedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RecordChildExecutionCompletedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RecordChildExecutionCompletedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReplicateEventsV2Request) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReplicateEventsV2Request) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReplicateEventsV2Request) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.NewRunEvents != nil {\n\t\t{\n\t\t\tsize, err := m.NewRunEvents.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.Events != nil {\n\t\t{\n\t\t\tsize, err := m.Events.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.VersionHistoryItems) > 0 {\n\t\tfor iNdEx := len(m.VersionHistoryItems) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.VersionHistoryItems[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReplicateEventsV2Response) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReplicateEventsV2Response) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReplicateEventsV2Response) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SyncShardStatusRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SyncShardStatusRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SyncShardStatusRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Time != nil {\n\t\t{\n\t\t\tsize, err := m.Time.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif len(m.SourceCluster) > 0 {\n\t\ti -= len(m.SourceCluster)\n\t\tcopy(dAtA[i:], m.SourceCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.SourceCluster)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SyncShardStatusResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SyncShardStatusResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SyncShardStatusResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SyncActivityRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SyncActivityRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SyncActivityRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.VersionHistory != nil {\n\t\t{\n\t\t\tsize, err := m.VersionHistory.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x6a\n\t}\n\tif len(m.LastWorkerIdentity) > 0 {\n\t\ti -= len(m.LastWorkerIdentity)\n\t\tcopy(dAtA[i:], m.LastWorkerIdentity)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.LastWorkerIdentity)))\n\t\ti--\n\t\tdAtA[i] = 0x62\n\t}\n\tif m.LastFailure != nil {\n\t\t{\n\t\t\tsize, err := m.LastFailure.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x5a\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x50\n\t}\n\tif m.Details != nil {\n\t\t{\n\t\t\tsize, err := m.Details.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif m.LastHeartbeatTime != nil {\n\t\t{\n\t\t\tsize, err := m.LastHeartbeatTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.StartedId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.StartedId))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.ScheduledTime != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.ScheduledId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduledId))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.Version != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Version))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *SyncActivityResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *SyncActivityResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *SyncActivityResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeMutableStateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeMutableStateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeMutableStateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeMutableStateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeMutableStateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeMutableStateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.MutableStateInDatabase) > 0 {\n\t\ti -= len(m.MutableStateInDatabase)\n\t\tcopy(dAtA[i:], m.MutableStateInDatabase)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.MutableStateInDatabase)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.MutableStateInCache) > 0 {\n\t\ti -= len(m.MutableStateInCache)\n\t\tcopy(dAtA[i:], m.MutableStateInCache)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.MutableStateInCache)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeHistoryHostRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeHistoryHostRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeHistoryHostRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeHistoryHostResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeHistoryHostResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeHistoryHostResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Address) > 0 {\n\t\ti -= len(m.Address)\n\t\tcopy(dAtA[i:], m.Address)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Address)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif len(m.ShardControllerStatus) > 0 {\n\t\ti -= len(m.ShardControllerStatus)\n\t\tcopy(dAtA[i:], m.ShardControllerStatus)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ShardControllerStatus)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.DomainCache != nil {\n\t\t{\n\t\t\tsize, err := m.DomainCache.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.ShardIds) > 0 {\n\t\tdAtA83 := make([]byte, len(m.ShardIds)*10)\n\t\tvar j82 int\n\t\tfor _, num1 := range m.ShardIds {\n\t\t\tnum := uint64(num1)\n\t\t\tfor num >= 1<<7 {\n\t\t\t\tdAtA83[j82] = uint8(uint64(num)&0x7f | 0x80)\n\t\t\t\tnum >>= 7\n\t\t\t\tj82++\n\t\t\t}\n\t\t\tdAtA83[j82] = uint8(num)\n\t\t\tj82++\n\t\t}\n\t\ti -= j82\n\t\tcopy(dAtA[i:], dAtA83[:j82])\n\t\ti = encodeVarintService(dAtA, i, uint64(j82))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.NumberOfShards != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NumberOfShards))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CloseShardRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CloseShardRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CloseShardRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CloseShardResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CloseShardResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CloseShardResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RemoveTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RemoveTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RemoveTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ClusterName) > 0 {\n\t\ti -= len(m.ClusterName)\n\t\tcopy(dAtA[i:], m.ClusterName)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClusterName)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.VisibilityTime != nil {\n\t\t{\n\t\t\tsize, err := m.VisibilityTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskId))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.TaskType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskType))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RemoveTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RemoveTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RemoveTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetQueueRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetQueueRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetQueueRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.TaskType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskType))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif len(m.ClusterName) > 0 {\n\t\ti -= len(m.ClusterName)\n\t\tcopy(dAtA[i:], m.ClusterName)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClusterName)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ResetQueueResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ResetQueueResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ResetQueueResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeQueueRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeQueueRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeQueueRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.TaskType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskType))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif len(m.ClusterName) > 0 {\n\t\ti -= len(m.ClusterName)\n\t\tcopy(dAtA[i:], m.ClusterName)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClusterName)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeQueueResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeQueueResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeQueueResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ProcessingQueueStates) > 0 {\n\t\tfor iNdEx := len(m.ProcessingQueueStates) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\ti -= len(m.ProcessingQueueStates[iNdEx])\n\t\t\tcopy(dAtA[i:], m.ProcessingQueueStates[iNdEx])\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ProcessingQueueStates[iNdEx])))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetReplicationMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetReplicationMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetReplicationMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ClusterName) > 0 {\n\t\ti -= len(m.ClusterName)\n\t\tcopy(dAtA[i:], m.ClusterName)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ClusterName)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Tokens) > 0 {\n\t\tfor iNdEx := len(m.Tokens) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Tokens[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetReplicationMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetReplicationMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetReplicationMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ShardMessages) > 0 {\n\t\tfor k := range m.ShardMessages {\n\t\t\tv := m.ShardMessages[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti = encodeVarintService(dAtA, i, uint64(k))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetDLQReplicationMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetDLQReplicationMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetDLQReplicationMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.TaskInfos) > 0 {\n\t\tfor iNdEx := len(m.TaskInfos) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.TaskInfos[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetDLQReplicationMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetDLQReplicationMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetDLQReplicationMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ReplicationTasks) > 0 {\n\t\tfor iNdEx := len(m.ReplicationTasks) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.ReplicationTasks[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReapplyEventsRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReapplyEventsRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReapplyEventsRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Events != nil {\n\t\t{\n\t\t\tsize, err := m.Events.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Domain) > 0 {\n\t\ti -= len(m.Domain)\n\t\tcopy(dAtA[i:], m.Domain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Domain)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReapplyEventsResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReapplyEventsResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReapplyEventsResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RefreshWorkflowTasksRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RefreshWorkflowTasksRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RefreshWorkflowTasksRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Domain) > 0 {\n\t\ti -= len(m.Domain)\n\t\tcopy(dAtA[i:], m.Domain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Domain)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RefreshWorkflowTasksResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RefreshWorkflowTasksResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RefreshWorkflowTasksResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CountDLQMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CountDLQMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CountDLQMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ForceFetch {\n\t\ti--\n\t\tif m.ForceFetch {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CountDLQMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CountDLQMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CountDLQMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Entries) > 0 {\n\t\tfor iNdEx := len(m.Entries) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Entries[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReadDLQMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReadDLQMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReadDLQMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.NextPageToken) > 0 {\n\t\ti -= len(m.NextPageToken)\n\t\tcopy(dAtA[i:], m.NextPageToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.NextPageToken)))\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.PageSize != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.PageSize))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\t{\n\t\t\tsize, err := m.InclusiveEndMessageId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.SourceCluster) > 0 {\n\t\ti -= len(m.SourceCluster)\n\t\tcopy(dAtA[i:], m.SourceCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.SourceCluster)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif m.Type != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Type))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ReadDLQMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ReadDLQMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ReadDLQMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.NextPageToken) > 0 {\n\t\ti -= len(m.NextPageToken)\n\t\tcopy(dAtA[i:], m.NextPageToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.NextPageToken)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.ReplicationTasksInfo) > 0 {\n\t\tfor iNdEx := len(m.ReplicationTasksInfo) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.ReplicationTasksInfo[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif len(m.ReplicationTasks) > 0 {\n\t\tfor iNdEx := len(m.ReplicationTasks) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.ReplicationTasks[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif m.Type != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Type))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PurgeDLQMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PurgeDLQMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PurgeDLQMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\t{\n\t\t\tsize, err := m.InclusiveEndMessageId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.SourceCluster) > 0 {\n\t\ti -= len(m.SourceCluster)\n\t\tcopy(dAtA[i:], m.SourceCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.SourceCluster)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif m.Type != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Type))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PurgeDLQMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PurgeDLQMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PurgeDLQMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *MergeDLQMessagesRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *MergeDLQMessagesRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *MergeDLQMessagesRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.NextPageToken) > 0 {\n\t\ti -= len(m.NextPageToken)\n\t\tcopy(dAtA[i:], m.NextPageToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.NextPageToken)))\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.PageSize != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.PageSize))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\t{\n\t\t\tsize, err := m.InclusiveEndMessageId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.SourceCluster) > 0 {\n\t\ti -= len(m.SourceCluster)\n\t\tcopy(dAtA[i:], m.SourceCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.SourceCluster)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif m.Type != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Type))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *MergeDLQMessagesResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *MergeDLQMessagesResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *MergeDLQMessagesResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.NextPageToken) > 0 {\n\t\ti -= len(m.NextPageToken)\n\t\tcopy(dAtA[i:], m.NextPageToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.NextPageToken)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *NotifyFailoverMarkersRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *NotifyFailoverMarkersRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *NotifyFailoverMarkersRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.FailoverMarkerTokens) > 0 {\n\t\tfor iNdEx := len(m.FailoverMarkerTokens) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.FailoverMarkerTokens[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *NotifyFailoverMarkersResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *NotifyFailoverMarkersResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *NotifyFailoverMarkersResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetCrossClusterTasksRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetCrossClusterTasksRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetCrossClusterTasksRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.TargetCluster) > 0 {\n\t\ti -= len(m.TargetCluster)\n\t\tcopy(dAtA[i:], m.TargetCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.TargetCluster)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.ShardIds) > 0 {\n\t\tdAtA93 := make([]byte, len(m.ShardIds)*10)\n\t\tvar j92 int\n\t\tfor _, num1 := range m.ShardIds {\n\t\t\tnum := uint64(num1)\n\t\t\tfor num >= 1<<7 {\n\t\t\t\tdAtA93[j92] = uint8(uint64(num)&0x7f | 0x80)\n\t\t\t\tnum >>= 7\n\t\t\t\tj92++\n\t\t\t}\n\t\t\tdAtA93[j92] = uint8(num)\n\t\t\tj92++\n\t\t}\n\t\ti -= j92\n\t\tcopy(dAtA[i:], dAtA93[:j92])\n\t\ti = encodeVarintService(dAtA, i, uint64(j92))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetCrossClusterTasksResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetCrossClusterTasksResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetCrossClusterTasksResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.FailedCauseByShard) > 0 {\n\t\tfor k := range m.FailedCauseByShard {\n\t\t\tv := m.FailedCauseByShard[k]\n\t\t\tbaseI := i\n\t\t\ti = encodeVarintService(dAtA, i, uint64(v))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x10\n\t\t\ti = encodeVarintService(dAtA, i, uint64(k))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif len(m.TasksByShard) > 0 {\n\t\tfor k := range m.TasksByShard {\n\t\t\tv := m.TasksByShard[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti = encodeVarintService(dAtA, i, uint64(k))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.FetchNewTasks {\n\t\ti--\n\t\tif m.FetchNewTasks {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif len(m.TaskResponses) > 0 {\n\t\tfor iNdEx := len(m.TaskResponses) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.TaskResponses[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif len(m.TargetCluster) > 0 {\n\t\ti -= len(m.TargetCluster)\n\t\tcopy(dAtA[i:], m.TargetCluster)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.TargetCluster)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ShardId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ShardId))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Tasks != nil {\n\t\t{\n\t\t\tsize, err := m.Tasks.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetFailoverInfoRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetFailoverInfoRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetFailoverInfoRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetFailoverInfoResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetFailoverInfoResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetFailoverInfoResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.PendingShards) > 0 {\n\t\tdAtA97 := make([]byte, len(m.PendingShards)*10)\n\t\tvar j96 int\n\t\tfor _, num1 := range m.PendingShards {\n\t\t\tnum := uint64(num1)\n\t\t\tfor num >= 1<<7 {\n\t\t\t\tdAtA97[j96] = uint8(uint64(num)&0x7f | 0x80)\n\t\t\t\tnum >>= 7\n\t\t\t\tj96++\n\t\t\t}\n\t\t\tdAtA97[j96] = uint8(num)\n\t\t\tj96++\n\t\t}\n\t\ti -= j96\n\t\tcopy(dAtA[i:], dAtA97[:j96])\n\t\ti = encodeVarintService(dAtA, i, uint64(j96))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.CompletedShardCount != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.CompletedShardCount))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RatelimitUpdateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RatelimitUpdateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RatelimitUpdateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Data != nil {\n\t\t{\n\t\t\tsize, err := m.Data.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RatelimitUpdateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RatelimitUpdateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RatelimitUpdateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Data != nil {\n\t\t{\n\t\t\tsize, err := m.Data.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintService(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovService(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *StartWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ParentExecutionInfo != nil {\n\t\tl = m.ParentExecutionInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.ExpirationTime != nil {\n\t\tl = m.ExpirationTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ContinueAsNewInitiator != 0 {\n\t\tn += 1 + sovService(uint64(m.ContinueAsNewInitiator))\n\t}\n\tif m.ContinuedFailure != nil {\n\t\tl = m.ContinuedFailure.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.LastCompletionResult != nil {\n\t\tl = m.LastCompletionResult.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.FirstDecisionTaskBackoff != nil {\n\t\tl = m.FirstDecisionTaskBackoff.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k, v := range m.PartitionConfig {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *StartWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.RunId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SignalWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ExternalWorkflowExecution != nil {\n\t\tl = m.ExternalWorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ChildWorkflowOnly {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SignalWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SignalWithStartWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k, v := range m.PartitionConfig {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SignalWithStartWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.RunId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.RunId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *TerminateWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ExternalWorkflowExecution != nil {\n\t\tl = m.ExternalWorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ChildWorkflowOnly {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *TerminateWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ExecutionConfiguration != nil {\n\t\tl = m.ExecutionConfiguration.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecutionInfo != nil {\n\t\tl = m.WorkflowExecutionInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.PendingActivities) > 0 {\n\t\tfor _, e := range m.PendingActivities {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif len(m.PendingChildren) > 0 {\n\t\tfor _, e := range m.PendingChildren {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.PendingDecision != nil {\n\t\tl = m.PendingDecision.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *QueryWorkflowRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *QueryWorkflowResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.QueryResult != nil {\n\t\tl = m.QueryResult.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.QueryRejected != nil {\n\t\tl = m.QueryRejected.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetStickyTaskListRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetStickyTaskListResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetMutableStateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ExpectedNextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.ExpectedNextEventId))\n\t}\n\tl = len(m.CurrentBranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.VersionHistoryItem != nil {\n\t\tl = m.VersionHistoryItem.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetMutableStateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.NextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.NextEventId))\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\tl = m.PreviousStartedEventId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.LastFirstEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.LastFirstEventId))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StickyTaskList != nil {\n\t\tl = m.StickyTaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientLibraryVersion)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientFeatureVersion)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientImpl)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StickyTaskListScheduleToStartTimeout != nil {\n\t\tl = m.StickyTaskListScheduleToStartTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\tn += 1 + sovService(uint64(m.EventStoreVersion))\n\t}\n\tl = len(m.CurrentBranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowState != 0 {\n\t\tn += 1 + sovService(uint64(m.WorkflowState))\n\t}\n\tif m.WorkflowCloseState != 0 {\n\t\tn += 1 + sovService(uint64(m.WorkflowCloseState))\n\t}\n\tif m.VersionHistories != nil {\n\t\tl = m.VersionHistories.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.IsStickyTaskListEnabled {\n\t\tn += 3\n\t}\n\tif m.HistorySize != 0 {\n\t\tn += 2 + sovService(uint64(m.HistorySize))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollMutableStateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ExpectedNextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.ExpectedNextEventId))\n\t}\n\tl = len(m.CurrentBranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollMutableStateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.NextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.NextEventId))\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\tl = m.PreviousStartedEventId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.LastFirstEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.LastFirstEventId))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StickyTaskList != nil {\n\t\tl = m.StickyTaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientLibraryVersion)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientFeatureVersion)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClientImpl)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StickyTaskListScheduleToStartTimeout != nil {\n\t\tl = m.StickyTaskListScheduleToStartTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.CurrentBranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.VersionHistories != nil {\n\t\tl = m.VersionHistories.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowState != 0 {\n\t\tn += 1 + sovService(uint64(m.WorkflowState))\n\t}\n\tif m.WorkflowCloseState != 0 {\n\t\tn += 1 + sovService(uint64(m.WorkflowCloseState))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordDecisionTaskStartedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduleId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduleId))\n\t}\n\tif m.TaskId != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskId))\n\t}\n\tl = len(m.RequestId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PollRequest != nil {\n\t\tl = m.PollRequest.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordDecisionTaskStartedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\tl = m.PreviousStartedEventId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduledEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduledEventId))\n\t}\n\tif m.StartedEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.StartedEventId))\n\t}\n\tif m.NextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.NextEventId))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.StickyExecutionEnabled {\n\t\tn += 2\n\t}\n\tif m.DecisionInfo != nil {\n\t\tl = m.DecisionInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecutionTaskList != nil {\n\t\tl = m.WorkflowExecutionTaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\tn += 1 + sovService(uint64(m.EventStoreVersion))\n\t}\n\tl = len(m.BranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduledTime != nil {\n\t\tl = m.ScheduledTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.Queries) > 0 {\n\t\tfor k, v := range m.Queries {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.HistorySize != 0 {\n\t\tn += 1 + sovService(uint64(m.HistorySize))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordActivityTaskStartedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduleId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduleId))\n\t}\n\tif m.TaskId != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskId))\n\t}\n\tl = len(m.RequestId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PollRequest != nil {\n\t\tl = m.PollRequest.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordActivityTaskStartedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ScheduledEvent != nil {\n\t\tl = m.ScheduledEvent.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\tl = m.ScheduledTimeOfThisAttempt.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\tl = m.HeartbeatDetails.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.WorkflowDomain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondDecisionTaskCompletedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondDecisionTaskCompletedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.StartedResponse != nil {\n\t\tl = m.StartedResponse.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.ActivitiesToDispatchLocally) > 0 {\n\t\tfor k, v := range m.ActivitiesToDispatchLocally {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondDecisionTaskFailedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondDecisionTaskFailedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordActivityTaskHeartbeatRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordActivityTaskHeartbeatResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.CancelRequested {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskCompletedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskCompletedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskFailedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskFailedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskCanceledRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondActivityTaskCanceledResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RemoveSignalMutableStateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.RequestId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RemoveSignalMutableStateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RequestCancelWorkflowExecutionRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.CancelRequest != nil {\n\t\tl = m.CancelRequest.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ExternalExecutionInfo != nil {\n\t\tl = m.ExternalExecutionInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ChildWorkflowOnly {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RequestCancelWorkflowExecutionResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ScheduleDecisionTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.IsFirstDecision {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ScheduleDecisionTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordChildExecutionCompletedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.InitiatedId != 0 {\n\t\tn += 1 + sovService(uint64(m.InitiatedId))\n\t}\n\tif m.CompletedExecution != nil {\n\t\tl = m.CompletedExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.CompletionEvent != nil {\n\t\tl = m.CompletionEvent.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedId != 0 {\n\t\tn += 1 + sovService(uint64(m.StartedId))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RecordChildExecutionCompletedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReplicateEventsV2Request) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.VersionHistoryItems) > 0 {\n\t\tfor _, e := range m.VersionHistoryItems {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.Events != nil {\n\t\tl = m.Events.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.NewRunEvents != nil {\n\t\tl = m.NewRunEvents.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReplicateEventsV2Response) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SyncShardStatusRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.SourceCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tif m.Time != nil {\n\t\tl = m.Time.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SyncShardStatusResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SyncActivityRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Version != 0 {\n\t\tn += 1 + sovService(uint64(m.Version))\n\t}\n\tif m.ScheduledId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduledId))\n\t}\n\tif m.ScheduledTime != nil {\n\t\tl = m.ScheduledTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedId != 0 {\n\t\tn += 1 + sovService(uint64(m.StartedId))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.LastHeartbeatTime != nil {\n\t\tl = m.LastHeartbeatTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Details != nil {\n\t\tl = m.Details.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.LastFailure != nil {\n\t\tl = m.LastFailure.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.LastWorkerIdentity)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.VersionHistory != nil {\n\t\tl = m.VersionHistory.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *SyncActivityResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeMutableStateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeMutableStateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.MutableStateInCache)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.MutableStateInDatabase)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeHistoryHostRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeHistoryHostResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.NumberOfShards != 0 {\n\t\tn += 1 + sovService(uint64(m.NumberOfShards))\n\t}\n\tif len(m.ShardIds) > 0 {\n\t\tl = 0\n\t\tfor _, e := range m.ShardIds {\n\t\t\tl += sovService(uint64(e))\n\t\t}\n\t\tn += 1 + sovService(uint64(l)) + l\n\t}\n\tif m.DomainCache != nil {\n\t\tl = m.DomainCache.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ShardControllerStatus)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.Address)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CloseShardRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CloseShardResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RemoveTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tif m.TaskType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskType))\n\t}\n\tif m.TaskId != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskId))\n\t}\n\tif m.VisibilityTime != nil {\n\t\tl = m.VisibilityTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ClusterName)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RemoveTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetQueueRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.ClusterName)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskType))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ResetQueueResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeQueueRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.ClusterName)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskType))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeQueueResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ProcessingQueueStates) > 0 {\n\t\tfor _, s := range m.ProcessingQueueStates {\n\t\t\tl = len(s)\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetReplicationMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.Tokens) > 0 {\n\t\tfor _, e := range m.Tokens {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tl = len(m.ClusterName)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetReplicationMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ShardMessages) > 0 {\n\t\tfor k, v := range m.ShardMessages {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + sovService(uint64(k)) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetDLQReplicationMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.TaskInfos) > 0 {\n\t\tfor _, e := range m.TaskInfos {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetDLQReplicationMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ReplicationTasks) > 0 {\n\t\tfor _, e := range m.ReplicationTasks {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReapplyEventsRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Domain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Events != nil {\n\t\tl = m.Events.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReapplyEventsResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RefreshWorkflowTasksRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Domain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RefreshWorkflowTasksResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CountDLQMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ForceFetch {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CountDLQMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.Entries) > 0 {\n\t\tfor _, e := range m.Entries {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReadDLQMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Type != 0 {\n\t\tn += 1 + sovService(uint64(m.Type))\n\t}\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.SourceCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\tl = m.InclusiveEndMessageId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PageSize != 0 {\n\t\tn += 1 + sovService(uint64(m.PageSize))\n\t}\n\tl = len(m.NextPageToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ReadDLQMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Type != 0 {\n\t\tn += 1 + sovService(uint64(m.Type))\n\t}\n\tif len(m.ReplicationTasks) > 0 {\n\t\tfor _, e := range m.ReplicationTasks {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif len(m.ReplicationTasksInfo) > 0 {\n\t\tfor _, e := range m.ReplicationTasksInfo {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tl = len(m.NextPageToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PurgeDLQMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Type != 0 {\n\t\tn += 1 + sovService(uint64(m.Type))\n\t}\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.SourceCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\tl = m.InclusiveEndMessageId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PurgeDLQMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *MergeDLQMessagesRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Type != 0 {\n\t\tn += 1 + sovService(uint64(m.Type))\n\t}\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.SourceCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.InclusiveEndMessageId != nil {\n\t\tl = m.InclusiveEndMessageId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PageSize != 0 {\n\t\tn += 1 + sovService(uint64(m.PageSize))\n\t}\n\tl = len(m.NextPageToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *MergeDLQMessagesResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.NextPageToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *NotifyFailoverMarkersRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.FailoverMarkerTokens) > 0 {\n\t\tfor _, e := range m.FailoverMarkerTokens {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *NotifyFailoverMarkersResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetCrossClusterTasksRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ShardIds) > 0 {\n\t\tl = 0\n\t\tfor _, e := range m.ShardIds {\n\t\t\tl += sovService(uint64(e))\n\t\t}\n\t\tn += 1 + sovService(uint64(l)) + l\n\t}\n\tl = len(m.TargetCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetCrossClusterTasksResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.TasksByShard) > 0 {\n\t\tfor k, v := range m.TasksByShard {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + sovService(uint64(k)) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif len(m.FailedCauseByShard) > 0 {\n\t\tfor k, v := range m.FailedCauseByShard {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + sovService(uint64(k)) + 1 + sovService(uint64(v))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondCrossClusterTasksCompletedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ShardId != 0 {\n\t\tn += 1 + sovService(uint64(m.ShardId))\n\t}\n\tl = len(m.TargetCluster)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.TaskResponses) > 0 {\n\t\tfor _, e := range m.TaskResponses {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.FetchNewTasks {\n\t\tn += 2\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondCrossClusterTasksCompletedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Tasks != nil {\n\t\tl = m.Tasks.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetFailoverInfoRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetFailoverInfoResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.CompletedShardCount != 0 {\n\t\tn += 1 + sovService(uint64(m.CompletedShardCount))\n\t}\n\tif len(m.PendingShards) > 0 {\n\t\tl = 0\n\t\tfor _, e := range m.PendingShards {\n\t\t\tl += sovService(uint64(e))\n\t\t}\n\t\tn += 1 + sovService(uint64(l)) + l\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RatelimitUpdateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Data != nil {\n\t\tl = m.Data.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RatelimitUpdateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Data != nil {\n\t\tl = m.Data.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovService(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozService(x uint64) (n int) {\n\treturn sovService(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *StartWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: StartWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: StartWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.StartWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ParentExecutionInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ParentExecutionInfo == nil {\n\t\t\t\tm.ParentExecutionInfo = &v1.ParentExecutionInfo{}\n\t\t\t}\n\t\t\tif err := m.ParentExecutionInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExpirationTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ExpirationTime == nil {\n\t\t\t\tm.ExpirationTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ExpirationTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ContinueAsNewInitiator\", wireType)\n\t\t\t}\n\t\t\tm.ContinueAsNewInitiator = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ContinueAsNewInitiator |= v1.ContinueAsNewInitiator(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ContinuedFailure\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ContinuedFailure == nil {\n\t\t\t\tm.ContinuedFailure = &v1.Failure{}\n\t\t\t}\n\t\t\tif err := m.ContinuedFailure.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastCompletionResult\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.LastCompletionResult == nil {\n\t\t\t\tm.LastCompletionResult = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.LastCompletionResult.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field FirstDecisionTaskBackoff\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.FirstDecisionTaskBackoff == nil {\n\t\t\t\tm.FirstDecisionTaskBackoff = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.FirstDecisionTaskBackoff.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.PartitionConfig[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *StartWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: StartWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: StartWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RunId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RunId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SignalWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.SignalWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExternalWorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ExternalWorkflowExecution == nil {\n\t\t\t\tm.ExternalWorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.ExternalWorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ChildWorkflowOnly\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ChildWorkflowOnly = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SignalWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SignalWithStartWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWithStartWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWithStartWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.SignalWithStartWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.PartitionConfig[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SignalWithStartWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWithStartWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SignalWithStartWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RunId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RunId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.ResetWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RunId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RunId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *TerminateWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TerminateWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TerminateWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.TerminateWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExternalWorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ExternalWorkflowExecution == nil {\n\t\t\t\tm.ExternalWorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.ExternalWorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ChildWorkflowOnly\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ChildWorkflowOnly = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *TerminateWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TerminateWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TerminateWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.DescribeWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExecutionConfiguration\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ExecutionConfiguration == nil {\n\t\t\t\tm.ExecutionConfiguration = &v1.WorkflowExecutionConfiguration{}\n\t\t\t}\n\t\t\tif err := m.ExecutionConfiguration.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecutionInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecutionInfo == nil {\n\t\t\t\tm.WorkflowExecutionInfo = &v1.WorkflowExecutionInfo{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecutionInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PendingActivities\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.PendingActivities = append(m.PendingActivities, &v1.PendingActivityInfo{})\n\t\t\tif err := m.PendingActivities[len(m.PendingActivities)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PendingChildren\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.PendingChildren = append(m.PendingChildren, &v1.PendingChildExecutionInfo{})\n\t\t\tif err := m.PendingChildren[len(m.PendingChildren)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PendingDecision\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PendingDecision == nil {\n\t\t\t\tm.PendingDecision = &v1.PendingDecisionInfo{}\n\t\t\t}\n\t\t\tif err := m.PendingDecision.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *QueryWorkflowRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.QueryWorkflowRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *QueryWorkflowResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field QueryResult\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.QueryResult == nil {\n\t\t\t\tm.QueryResult = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.QueryResult.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field QueryRejected\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.QueryRejected == nil {\n\t\t\t\tm.QueryRejected = &v1.QueryRejected{}\n\t\t\t}\n\t\t\tif err := m.QueryRejected.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetStickyTaskListRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetStickyTaskListRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetStickyTaskListRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.ResetStickyTaskListRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetStickyTaskListResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetStickyTaskListResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetStickyTaskListResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetMutableStateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetMutableStateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetMutableStateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExpectedNextEventId\", wireType)\n\t\t\t}\n\t\t\tm.ExpectedNextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ExpectedNextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentBranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.CurrentBranchToken = append(m.CurrentBranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.CurrentBranchToken == nil {\n\t\t\t\tm.CurrentBranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VersionHistoryItem\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.VersionHistoryItem == nil {\n\t\t\t\tm.VersionHistoryItem = &v11.VersionHistoryItem{}\n\t\t\t}\n\t\t\tif err := m.VersionHistoryItem.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetMutableStateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetMutableStateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetMutableStateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextEventId\", wireType)\n\t\t\t}\n\t\t\tm.NextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PreviousStartedEventId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PreviousStartedEventId == nil {\n\t\t\t\tm.PreviousStartedEventId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.PreviousStartedEventId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastFirstEventId\", wireType)\n\t\t\t}\n\t\t\tm.LastFirstEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.LastFirstEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyTaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StickyTaskList == nil {\n\t\t\t\tm.StickyTaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.StickyTaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientLibraryVersion\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientLibraryVersion = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientFeatureVersion\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientFeatureVersion = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientImpl\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientImpl = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 11:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyTaskListScheduleToStartTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StickyTaskListScheduleToStartTimeout == nil {\n\t\t\t\tm.StickyTaskListScheduleToStartTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.StickyTaskListScheduleToStartTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 12:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field EventStoreVersion\", wireType)\n\t\t\t}\n\t\t\tm.EventStoreVersion = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.EventStoreVersion |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 13:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentBranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.CurrentBranchToken = append(m.CurrentBranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.CurrentBranchToken == nil {\n\t\t\t\tm.CurrentBranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 14:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowState\", wireType)\n\t\t\t}\n\t\t\tm.WorkflowState = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.WorkflowState |= v12.WorkflowState(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 15:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowCloseState\", wireType)\n\t\t\t}\n\t\t\tm.WorkflowCloseState = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.WorkflowCloseState |= v1.WorkflowExecutionCloseStatus(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 16:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VersionHistories\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.VersionHistories == nil {\n\t\t\t\tm.VersionHistories = &v12.VersionHistories{}\n\t\t\t}\n\t\t\tif err := m.VersionHistories.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 17:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IsStickyTaskListEnabled\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.IsStickyTaskListEnabled = bool(v != 0)\n\t\tcase 18:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HistorySize\", wireType)\n\t\t\t}\n\t\t\tm.HistorySize = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.HistorySize |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollMutableStateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollMutableStateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollMutableStateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExpectedNextEventId\", wireType)\n\t\t\t}\n\t\t\tm.ExpectedNextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ExpectedNextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentBranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.CurrentBranchToken = append(m.CurrentBranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.CurrentBranchToken == nil {\n\t\t\t\tm.CurrentBranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollMutableStateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollMutableStateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollMutableStateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextEventId\", wireType)\n\t\t\t}\n\t\t\tm.NextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PreviousStartedEventId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PreviousStartedEventId == nil {\n\t\t\t\tm.PreviousStartedEventId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.PreviousStartedEventId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastFirstEventId\", wireType)\n\t\t\t}\n\t\t\tm.LastFirstEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.LastFirstEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyTaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StickyTaskList == nil {\n\t\t\t\tm.StickyTaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.StickyTaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientLibraryVersion\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientLibraryVersion = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientFeatureVersion\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientFeatureVersion = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClientImpl\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClientImpl = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 11:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyTaskListScheduleToStartTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StickyTaskListScheduleToStartTimeout == nil {\n\t\t\t\tm.StickyTaskListScheduleToStartTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.StickyTaskListScheduleToStartTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 12:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentBranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.CurrentBranchToken = append(m.CurrentBranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.CurrentBranchToken == nil {\n\t\t\t\tm.CurrentBranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 13:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VersionHistories\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.VersionHistories == nil {\n\t\t\t\tm.VersionHistories = &v12.VersionHistories{}\n\t\t\t}\n\t\t\tif err := m.VersionHistories.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 14:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowState\", wireType)\n\t\t\t}\n\t\t\tm.WorkflowState = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.WorkflowState |= v12.WorkflowState(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 15:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowCloseState\", wireType)\n\t\t\t}\n\t\t\tm.WorkflowCloseState = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.WorkflowCloseState |= v1.WorkflowExecutionCloseStatus(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordDecisionTaskStartedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordDecisionTaskStartedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordDecisionTaskStartedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduleId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduleId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskId\", wireType)\n\t\t\t}\n\t\t\tm.TaskId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RequestId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RequestId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PollRequest\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PollRequest == nil {\n\t\t\t\tm.PollRequest = &v1.PollForDecisionTaskRequest{}\n\t\t\t}\n\t\t\tif err := m.PollRequest.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordDecisionTaskStartedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordDecisionTaskStartedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordDecisionTaskStartedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PreviousStartedEventId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PreviousStartedEventId == nil {\n\t\t\t\tm.PreviousStartedEventId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.PreviousStartedEventId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledEventId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduledEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduledEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedEventId\", wireType)\n\t\t\t}\n\t\t\tm.StartedEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.StartedEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextEventId\", wireType)\n\t\t\t}\n\t\t\tm.NextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 7:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyExecutionEnabled\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.StickyExecutionEnabled = bool(v != 0)\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DecisionInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.DecisionInfo == nil {\n\t\t\t\tm.DecisionInfo = &v12.TransientDecisionInfo{}\n\t\t\t}\n\t\t\tif err := m.DecisionInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecutionTaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecutionTaskList == nil {\n\t\t\t\tm.WorkflowExecutionTaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecutionTaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field EventStoreVersion\", wireType)\n\t\t\t}\n\t\t\tm.EventStoreVersion = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.EventStoreVersion |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 11:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.BranchToken = append(m.BranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.BranchToken == nil {\n\t\t\t\tm.BranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 12:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTime == nil {\n\t\t\t\tm.ScheduledTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 13:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 14:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Queries\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Queries == nil {\n\t\t\t\tm.Queries = make(map[string]*v1.WorkflowQuery)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *v1.WorkflowQuery\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &v1.WorkflowQuery{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Queries[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 15:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HistorySize\", wireType)\n\t\t\t}\n\t\t\tm.HistorySize = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.HistorySize |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordActivityTaskStartedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskStartedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskStartedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduleId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduleId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskId\", wireType)\n\t\t\t}\n\t\t\tm.TaskId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RequestId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RequestId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PollRequest\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PollRequest == nil {\n\t\t\t\tm.PollRequest = &v1.PollForActivityTaskRequest{}\n\t\t\t}\n\t\t\tif err := m.PollRequest.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordActivityTaskStartedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskStartedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskStartedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledEvent == nil {\n\t\t\t\tm.ScheduledEvent = &v1.HistoryEvent{}\n\t\t\t}\n\t\t\tif err := m.ScheduledEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTimeOfThisAttempt\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTimeOfThisAttempt == nil {\n\t\t\t\tm.ScheduledTimeOfThisAttempt = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTimeOfThisAttempt.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HeartbeatDetails\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.HeartbeatDetails == nil {\n\t\t\t\tm.HeartbeatDetails = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.HeartbeatDetails.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowDomain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.WorkflowDomain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondDecisionTaskCompletedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskCompletedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskCompletedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondDecisionTaskCompletedRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondDecisionTaskCompletedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskCompletedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskCompletedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedResponse\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedResponse == nil {\n\t\t\t\tm.StartedResponse = &RecordDecisionTaskStartedResponse{}\n\t\t\t}\n\t\t\tif err := m.StartedResponse.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivitiesToDispatchLocally\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ActivitiesToDispatchLocally == nil {\n\t\t\t\tm.ActivitiesToDispatchLocally = make(map[string]*v1.ActivityLocalDispatchInfo)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *v1.ActivityLocalDispatchInfo\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &v1.ActivityLocalDispatchInfo{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ActivitiesToDispatchLocally[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondDecisionTaskFailedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskFailedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskFailedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondDecisionTaskFailedRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondDecisionTaskFailedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskFailedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondDecisionTaskFailedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordActivityTaskHeartbeatRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskHeartbeatRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskHeartbeatRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RecordActivityTaskHeartbeatRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordActivityTaskHeartbeatResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskHeartbeatResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordActivityTaskHeartbeatResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CancelRequested\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.CancelRequested = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskCompletedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCompletedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCompletedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondActivityTaskCompletedRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskCompletedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCompletedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCompletedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskFailedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskFailedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskFailedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondActivityTaskFailedRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskFailedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskFailedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskFailedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskCanceledRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCanceledRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCanceledRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondActivityTaskCanceledRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondActivityTaskCanceledResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCanceledResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondActivityTaskCanceledResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RemoveSignalMutableStateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveSignalMutableStateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveSignalMutableStateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RequestId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.RequestId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RemoveSignalMutableStateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveSignalMutableStateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveSignalMutableStateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RequestCancelWorkflowExecutionRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RequestCancelWorkflowExecutionRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RequestCancelWorkflowExecutionRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CancelRequest\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.CancelRequest == nil {\n\t\t\t\tm.CancelRequest = &v1.RequestCancelWorkflowExecutionRequest{}\n\t\t\t}\n\t\t\tif err := m.CancelRequest.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExternalExecutionInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ExternalExecutionInfo == nil {\n\t\t\t\tm.ExternalExecutionInfo = &v1.ExternalExecutionInfo{}\n\t\t\t}\n\t\t\tif err := m.ExternalExecutionInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ChildWorkflowOnly\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ChildWorkflowOnly = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RequestCancelWorkflowExecutionResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RequestCancelWorkflowExecutionResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RequestCancelWorkflowExecutionResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ScheduleDecisionTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ScheduleDecisionTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ScheduleDecisionTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IsFirstDecision\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.IsFirstDecision = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ScheduleDecisionTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ScheduleDecisionTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ScheduleDecisionTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordChildExecutionCompletedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordChildExecutionCompletedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordChildExecutionCompletedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field InitiatedId\", wireType)\n\t\t\t}\n\t\t\tm.InitiatedId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.InitiatedId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CompletedExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.CompletedExecution == nil {\n\t\t\t\tm.CompletedExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.CompletedExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CompletionEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.CompletionEvent == nil {\n\t\t\t\tm.CompletionEvent = &v1.HistoryEvent{}\n\t\t\t}\n\t\t\tif err := m.CompletionEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedId\", wireType)\n\t\t\t}\n\t\t\tm.StartedId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.StartedId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RecordChildExecutionCompletedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RecordChildExecutionCompletedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RecordChildExecutionCompletedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReplicateEventsV2Request) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReplicateEventsV2Request: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReplicateEventsV2Request: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VersionHistoryItems\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.VersionHistoryItems = append(m.VersionHistoryItems, &v11.VersionHistoryItem{})\n\t\t\tif err := m.VersionHistoryItems[len(m.VersionHistoryItems)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Events\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Events == nil {\n\t\t\t\tm.Events = &v1.DataBlob{}\n\t\t\t}\n\t\t\tif err := m.Events.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NewRunEvents\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.NewRunEvents == nil {\n\t\t\t\tm.NewRunEvents = &v1.DataBlob{}\n\t\t\t}\n\t\t\tif err := m.NewRunEvents.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReplicateEventsV2Response) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReplicateEventsV2Response: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReplicateEventsV2Response: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SyncShardStatusRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SyncShardStatusRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SyncShardStatusRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field SourceCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.SourceCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Time\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Time == nil {\n\t\t\t\tm.Time = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.Time.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SyncShardStatusResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SyncShardStatusResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SyncShardStatusResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SyncActivityRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SyncActivityRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SyncActivityRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Version\", wireType)\n\t\t\t}\n\t\t\tm.Version = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Version |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduledId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduledId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTime == nil {\n\t\t\t\tm.ScheduledTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedId\", wireType)\n\t\t\t}\n\t\t\tm.StartedId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.StartedId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastHeartbeatTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.LastHeartbeatTime == nil {\n\t\t\t\tm.LastHeartbeatTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.LastHeartbeatTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Details\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Details == nil {\n\t\t\t\tm.Details = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.Details.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 11:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastFailure\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.LastFailure == nil {\n\t\t\t\tm.LastFailure = &v1.Failure{}\n\t\t\t}\n\t\t\tif err := m.LastFailure.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 12:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LastWorkerIdentity\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.LastWorkerIdentity = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 13:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VersionHistory\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.VersionHistory == nil {\n\t\t\t\tm.VersionHistory = &v11.VersionHistory{}\n\t\t\t}\n\t\t\tif err := m.VersionHistory.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *SyncActivityResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: SyncActivityResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: SyncActivityResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeMutableStateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeMutableStateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeMutableStateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeMutableStateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeMutableStateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeMutableStateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field MutableStateInCache\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.MutableStateInCache = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field MutableStateInDatabase\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.MutableStateInDatabase = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeHistoryHostRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeHistoryHostRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeHistoryHostRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeHistoryHostResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeHistoryHostResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeHistoryHostResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NumberOfShards\", wireType)\n\t\t\t}\n\t\t\tm.NumberOfShards = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NumberOfShards |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType == 0 {\n\t\t\t\tvar v int32\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tm.ShardIds = append(m.ShardIds, v)\n\t\t\t} else if wireType == 2 {\n\t\t\t\tvar packedLen int\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tpackedLen |= int(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif packedLen < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tpostIndex := iNdEx + packedLen\n\t\t\t\tif postIndex < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tif postIndex > l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tvar elementCount int\n\t\t\t\tvar count int\n\t\t\t\tfor _, integer := range dAtA[iNdEx:postIndex] {\n\t\t\t\t\tif integer < 128 {\n\t\t\t\t\t\tcount++\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\telementCount = count\n\t\t\t\tif elementCount != 0 && len(m.ShardIds) == 0 {\n\t\t\t\t\tm.ShardIds = make([]int32, 0, elementCount)\n\t\t\t\t}\n\t\t\t\tfor iNdEx < postIndex {\n\t\t\t\t\tvar v int32\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tm.ShardIds = append(m.ShardIds, v)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardIds\", wireType)\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainCache\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.DomainCache == nil {\n\t\t\t\tm.DomainCache = &v11.DomainCacheInfo{}\n\t\t\t}\n\t\t\tif err := m.DomainCache.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardControllerStatus\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardControllerStatus = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Address\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Address = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CloseShardRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CloseShardRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CloseShardRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CloseShardResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CloseShardResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CloseShardResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RemoveTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskType\", wireType)\n\t\t\t}\n\t\t\tm.TaskType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskType |= v11.TaskType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskId\", wireType)\n\t\t\t}\n\t\t\tm.TaskId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field VisibilityTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.VisibilityTime == nil {\n\t\t\t\tm.VisibilityTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.VisibilityTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClusterName\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClusterName = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RemoveTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RemoveTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetQueueRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetQueueRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetQueueRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClusterName\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClusterName = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskType\", wireType)\n\t\t\t}\n\t\t\tm.TaskType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskType |= v11.TaskType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ResetQueueResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ResetQueueResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ResetQueueResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeQueueRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeQueueRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeQueueRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClusterName\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClusterName = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskType\", wireType)\n\t\t\t}\n\t\t\tm.TaskType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskType |= v11.TaskType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeQueueResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeQueueResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeQueueResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ProcessingQueueStates\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ProcessingQueueStates = append(m.ProcessingQueueStates, string(dAtA[iNdEx:postIndex]))\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetReplicationMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetReplicationMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetReplicationMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Tokens\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Tokens = append(m.Tokens, &v11.ReplicationToken{})\n\t\t\tif err := m.Tokens[len(m.Tokens)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ClusterName\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ClusterName = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetReplicationMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetReplicationMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetReplicationMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardMessages\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ShardMessages == nil {\n\t\t\t\tm.ShardMessages = make(map[int32]*v11.ReplicationMessages)\n\t\t\t}\n\t\t\tvar mapkey int32\n\t\t\tvar mapvalue *v11.ReplicationMessages\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapkey |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &v11.ReplicationMessages{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ShardMessages[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetDLQReplicationMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetDLQReplicationMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetDLQReplicationMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskInfos\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskInfos = append(m.TaskInfos, &v11.ReplicationTaskInfo{})\n\t\t\tif err := m.TaskInfos[len(m.TaskInfos)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetDLQReplicationMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetDLQReplicationMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetDLQReplicationMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ReplicationTasks\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ReplicationTasks = append(m.ReplicationTasks, &v11.ReplicationTask{})\n\t\t\tif err := m.ReplicationTasks[len(m.ReplicationTasks)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReapplyEventsRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReapplyEventsRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReapplyEventsRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Domain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Domain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Events\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Events == nil {\n\t\t\t\tm.Events = &v1.DataBlob{}\n\t\t\t}\n\t\t\tif err := m.Events.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReapplyEventsResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReapplyEventsResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReapplyEventsResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RefreshWorkflowTasksRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshWorkflowTasksRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshWorkflowTasksRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Domain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Domain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RefreshWorkflowTasksResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshWorkflowTasksResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshWorkflowTasksResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CountDLQMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CountDLQMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CountDLQMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForceFetch\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ForceFetch = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CountDLQMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CountDLQMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CountDLQMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Entries\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Entries = append(m.Entries, &v11.HistoryDLQCountEntry{})\n\t\t\tif err := m.Entries[len(m.Entries)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReadDLQMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReadDLQMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReadDLQMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Type\", wireType)\n\t\t\t}\n\t\t\tm.Type = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Type |= v11.DLQType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field SourceCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.SourceCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field InclusiveEndMessageId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.InclusiveEndMessageId == nil {\n\t\t\t\tm.InclusiveEndMessageId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.InclusiveEndMessageId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PageSize\", wireType)\n\t\t\t}\n\t\t\tm.PageSize = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.PageSize |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextPageToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.NextPageToken = append(m.NextPageToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.NextPageToken == nil {\n\t\t\t\tm.NextPageToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ReadDLQMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ReadDLQMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ReadDLQMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Type\", wireType)\n\t\t\t}\n\t\t\tm.Type = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Type |= v11.DLQType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ReplicationTasks\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ReplicationTasks = append(m.ReplicationTasks, &v11.ReplicationTask{})\n\t\t\tif err := m.ReplicationTasks[len(m.ReplicationTasks)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ReplicationTasksInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ReplicationTasksInfo = append(m.ReplicationTasksInfo, &v11.ReplicationTaskInfo{})\n\t\t\tif err := m.ReplicationTasksInfo[len(m.ReplicationTasksInfo)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextPageToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.NextPageToken = append(m.NextPageToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.NextPageToken == nil {\n\t\t\t\tm.NextPageToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PurgeDLQMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PurgeDLQMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PurgeDLQMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Type\", wireType)\n\t\t\t}\n\t\t\tm.Type = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Type |= v11.DLQType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field SourceCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.SourceCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field InclusiveEndMessageId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.InclusiveEndMessageId == nil {\n\t\t\t\tm.InclusiveEndMessageId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.InclusiveEndMessageId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PurgeDLQMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PurgeDLQMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PurgeDLQMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *MergeDLQMessagesRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: MergeDLQMessagesRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: MergeDLQMessagesRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Type\", wireType)\n\t\t\t}\n\t\t\tm.Type = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Type |= v11.DLQType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field SourceCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.SourceCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field InclusiveEndMessageId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.InclusiveEndMessageId == nil {\n\t\t\t\tm.InclusiveEndMessageId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.InclusiveEndMessageId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PageSize\", wireType)\n\t\t\t}\n\t\t\tm.PageSize = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.PageSize |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextPageToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.NextPageToken = append(m.NextPageToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.NextPageToken == nil {\n\t\t\t\tm.NextPageToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *MergeDLQMessagesResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: MergeDLQMessagesResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: MergeDLQMessagesResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextPageToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.NextPageToken = append(m.NextPageToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.NextPageToken == nil {\n\t\t\t\tm.NextPageToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *NotifyFailoverMarkersRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: NotifyFailoverMarkersRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: NotifyFailoverMarkersRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field FailoverMarkerTokens\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.FailoverMarkerTokens = append(m.FailoverMarkerTokens, &v11.FailoverMarkerToken{})\n\t\t\tif err := m.FailoverMarkerTokens[len(m.FailoverMarkerTokens)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *NotifyFailoverMarkersResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: NotifyFailoverMarkersResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: NotifyFailoverMarkersResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetCrossClusterTasksRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetCrossClusterTasksRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetCrossClusterTasksRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType == 0 {\n\t\t\t\tvar v int32\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tm.ShardIds = append(m.ShardIds, v)\n\t\t\t} else if wireType == 2 {\n\t\t\t\tvar packedLen int\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tpackedLen |= int(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif packedLen < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tpostIndex := iNdEx + packedLen\n\t\t\t\tif postIndex < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tif postIndex > l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tvar elementCount int\n\t\t\t\tvar count int\n\t\t\t\tfor _, integer := range dAtA[iNdEx:postIndex] {\n\t\t\t\t\tif integer < 128 {\n\t\t\t\t\t\tcount++\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\telementCount = count\n\t\t\t\tif elementCount != 0 && len(m.ShardIds) == 0 {\n\t\t\t\t\tm.ShardIds = make([]int32, 0, elementCount)\n\t\t\t\t}\n\t\t\t\tfor iNdEx < postIndex {\n\t\t\t\t\tvar v int32\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tm.ShardIds = append(m.ShardIds, v)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardIds\", wireType)\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TargetCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TargetCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetCrossClusterTasksResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetCrossClusterTasksResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetCrossClusterTasksResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TasksByShard\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TasksByShard == nil {\n\t\t\t\tm.TasksByShard = make(map[int32]*v11.CrossClusterTaskRequests)\n\t\t\t}\n\t\t\tvar mapkey int32\n\t\t\tvar mapvalue *v11.CrossClusterTaskRequests\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapkey |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &v11.CrossClusterTaskRequests{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.TasksByShard[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field FailedCauseByShard\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.FailedCauseByShard == nil {\n\t\t\t\tm.FailedCauseByShard = make(map[int32]v11.GetTaskFailedCause)\n\t\t\t}\n\t\t\tvar mapkey int32\n\t\t\tvar mapvalue v11.GetTaskFailedCause\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapkey |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapvalue |= v11.GetTaskFailedCause(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.FailedCauseByShard[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondCrossClusterTasksCompletedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondCrossClusterTasksCompletedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondCrossClusterTasksCompletedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardId\", wireType)\n\t\t\t}\n\t\t\tm.ShardId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ShardId |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TargetCluster\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TargetCluster = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskResponses\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskResponses = append(m.TaskResponses, &v11.CrossClusterTaskResponse{})\n\t\t\tif err := m.TaskResponses[len(m.TaskResponses)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field FetchNewTasks\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.FetchNewTasks = bool(v != 0)\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondCrossClusterTasksCompletedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondCrossClusterTasksCompletedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondCrossClusterTasksCompletedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Tasks\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Tasks == nil {\n\t\t\t\tm.Tasks = &v11.CrossClusterTaskRequests{}\n\t\t\t}\n\t\t\tif err := m.Tasks.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetFailoverInfoRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetFailoverInfoRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetFailoverInfoRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetFailoverInfoResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetFailoverInfoResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetFailoverInfoResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CompletedShardCount\", wireType)\n\t\t\t}\n\t\t\tm.CompletedShardCount = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.CompletedShardCount |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType == 0 {\n\t\t\t\tvar v int32\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tm.PendingShards = append(m.PendingShards, v)\n\t\t\t} else if wireType == 2 {\n\t\t\t\tvar packedLen int\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\tpackedLen |= int(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif packedLen < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tpostIndex := iNdEx + packedLen\n\t\t\t\tif postIndex < 0 {\n\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t}\n\t\t\t\tif postIndex > l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tvar elementCount int\n\t\t\t\tvar count int\n\t\t\t\tfor _, integer := range dAtA[iNdEx:postIndex] {\n\t\t\t\t\tif integer < 128 {\n\t\t\t\t\t\tcount++\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\telementCount = count\n\t\t\t\tif elementCount != 0 && len(m.PendingShards) == 0 {\n\t\t\t\t\tm.PendingShards = make([]int32, 0, elementCount)\n\t\t\t\t}\n\t\t\t\tfor iNdEx < postIndex {\n\t\t\t\t\tvar v int32\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tv |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tm.PendingShards = append(m.PendingShards, v)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PendingShards\", wireType)\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RatelimitUpdateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RatelimitUpdateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RatelimitUpdateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Data\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Data == nil {\n\t\t\t\tm.Data = &v12.Any{}\n\t\t\t}\n\t\t\tif err := m.Data.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RatelimitUpdateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RatelimitUpdateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RatelimitUpdateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Data\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Data == nil {\n\t\t\t\tm.Data = &v12.Any{}\n\t\t\t}\n\t\t\tif err := m.Data.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipService(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthService\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupService\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthService\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthService        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowService          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupService = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/history/v1/service.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/history/v1/service.proto\n\npackage historyv1\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"reflect\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/gogo/protobuf/proto\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/api/x/restriction\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/encoding/protobuf/reflection\"\n)\n\nvar _ = ioutil.NopCloser\n\n// HistoryAPIYARPCClient is the YARPC client-side interface for the HistoryAPI service.\ntype HistoryAPIYARPCClient interface {\n\tStartWorkflowExecution(context.Context, *StartWorkflowExecutionRequest, ...yarpc.CallOption) (*StartWorkflowExecutionResponse, error)\n\tSignalWorkflowExecution(context.Context, *SignalWorkflowExecutionRequest, ...yarpc.CallOption) (*SignalWorkflowExecutionResponse, error)\n\tSignalWithStartWorkflowExecution(context.Context, *SignalWithStartWorkflowExecutionRequest, ...yarpc.CallOption) (*SignalWithStartWorkflowExecutionResponse, error)\n\tResetWorkflowExecution(context.Context, *ResetWorkflowExecutionRequest, ...yarpc.CallOption) (*ResetWorkflowExecutionResponse, error)\n\tTerminateWorkflowExecution(context.Context, *TerminateWorkflowExecutionRequest, ...yarpc.CallOption) (*TerminateWorkflowExecutionResponse, error)\n\tDescribeWorkflowExecution(context.Context, *DescribeWorkflowExecutionRequest, ...yarpc.CallOption) (*DescribeWorkflowExecutionResponse, error)\n\tQueryWorkflow(context.Context, *QueryWorkflowRequest, ...yarpc.CallOption) (*QueryWorkflowResponse, error)\n\tResetStickyTaskList(context.Context, *ResetStickyTaskListRequest, ...yarpc.CallOption) (*ResetStickyTaskListResponse, error)\n\tGetMutableState(context.Context, *GetMutableStateRequest, ...yarpc.CallOption) (*GetMutableStateResponse, error)\n\tPollMutableState(context.Context, *PollMutableStateRequest, ...yarpc.CallOption) (*PollMutableStateResponse, error)\n\tRecordDecisionTaskStarted(context.Context, *RecordDecisionTaskStartedRequest, ...yarpc.CallOption) (*RecordDecisionTaskStartedResponse, error)\n\tRespondDecisionTaskCompleted(context.Context, *RespondDecisionTaskCompletedRequest, ...yarpc.CallOption) (*RespondDecisionTaskCompletedResponse, error)\n\tRespondDecisionTaskFailed(context.Context, *RespondDecisionTaskFailedRequest, ...yarpc.CallOption) (*RespondDecisionTaskFailedResponse, error)\n\tRecordActivityTaskStarted(context.Context, *RecordActivityTaskStartedRequest, ...yarpc.CallOption) (*RecordActivityTaskStartedResponse, error)\n\tRespondActivityTaskCompleted(context.Context, *RespondActivityTaskCompletedRequest, ...yarpc.CallOption) (*RespondActivityTaskCompletedResponse, error)\n\tRespondActivityTaskFailed(context.Context, *RespondActivityTaskFailedRequest, ...yarpc.CallOption) (*RespondActivityTaskFailedResponse, error)\n\tRespondActivityTaskCanceled(context.Context, *RespondActivityTaskCanceledRequest, ...yarpc.CallOption) (*RespondActivityTaskCanceledResponse, error)\n\tRecordActivityTaskHeartbeat(context.Context, *RecordActivityTaskHeartbeatRequest, ...yarpc.CallOption) (*RecordActivityTaskHeartbeatResponse, error)\n\tRequestCancelWorkflowExecution(context.Context, *RequestCancelWorkflowExecutionRequest, ...yarpc.CallOption) (*RequestCancelWorkflowExecutionResponse, error)\n\tRemoveSignalMutableState(context.Context, *RemoveSignalMutableStateRequest, ...yarpc.CallOption) (*RemoveSignalMutableStateResponse, error)\n\tScheduleDecisionTask(context.Context, *ScheduleDecisionTaskRequest, ...yarpc.CallOption) (*ScheduleDecisionTaskResponse, error)\n\tRecordChildExecutionCompleted(context.Context, *RecordChildExecutionCompletedRequest, ...yarpc.CallOption) (*RecordChildExecutionCompletedResponse, error)\n\tReplicateEventsV2(context.Context, *ReplicateEventsV2Request, ...yarpc.CallOption) (*ReplicateEventsV2Response, error)\n\tSyncShardStatus(context.Context, *SyncShardStatusRequest, ...yarpc.CallOption) (*SyncShardStatusResponse, error)\n\tSyncActivity(context.Context, *SyncActivityRequest, ...yarpc.CallOption) (*SyncActivityResponse, error)\n\tDescribeMutableState(context.Context, *DescribeMutableStateRequest, ...yarpc.CallOption) (*DescribeMutableStateResponse, error)\n\tDescribeHistoryHost(context.Context, *DescribeHistoryHostRequest, ...yarpc.CallOption) (*DescribeHistoryHostResponse, error)\n\tCloseShard(context.Context, *CloseShardRequest, ...yarpc.CallOption) (*CloseShardResponse, error)\n\tRemoveTask(context.Context, *RemoveTaskRequest, ...yarpc.CallOption) (*RemoveTaskResponse, error)\n\tResetQueue(context.Context, *ResetQueueRequest, ...yarpc.CallOption) (*ResetQueueResponse, error)\n\tDescribeQueue(context.Context, *DescribeQueueRequest, ...yarpc.CallOption) (*DescribeQueueResponse, error)\n\tGetReplicationMessages(context.Context, *GetReplicationMessagesRequest, ...yarpc.CallOption) (*GetReplicationMessagesResponse, error)\n\tGetDLQReplicationMessages(context.Context, *GetDLQReplicationMessagesRequest, ...yarpc.CallOption) (*GetDLQReplicationMessagesResponse, error)\n\tReapplyEvents(context.Context, *ReapplyEventsRequest, ...yarpc.CallOption) (*ReapplyEventsResponse, error)\n\tRefreshWorkflowTasks(context.Context, *RefreshWorkflowTasksRequest, ...yarpc.CallOption) (*RefreshWorkflowTasksResponse, error)\n\tCountDLQMessages(context.Context, *CountDLQMessagesRequest, ...yarpc.CallOption) (*CountDLQMessagesResponse, error)\n\tReadDLQMessages(context.Context, *ReadDLQMessagesRequest, ...yarpc.CallOption) (*ReadDLQMessagesResponse, error)\n\tPurgeDLQMessages(context.Context, *PurgeDLQMessagesRequest, ...yarpc.CallOption) (*PurgeDLQMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *MergeDLQMessagesRequest, ...yarpc.CallOption) (*MergeDLQMessagesResponse, error)\n\tNotifyFailoverMarkers(context.Context, *NotifyFailoverMarkersRequest, ...yarpc.CallOption) (*NotifyFailoverMarkersResponse, error)\n\tGetCrossClusterTasks(context.Context, *GetCrossClusterTasksRequest, ...yarpc.CallOption) (*GetCrossClusterTasksResponse, error)\n\tRespondCrossClusterTasksCompleted(context.Context, *RespondCrossClusterTasksCompletedRequest, ...yarpc.CallOption) (*RespondCrossClusterTasksCompletedResponse, error)\n\tGetFailoverInfo(context.Context, *GetFailoverInfoRequest, ...yarpc.CallOption) (*GetFailoverInfoResponse, error)\n\tRatelimitUpdate(context.Context, *RatelimitUpdateRequest, ...yarpc.CallOption) (*RatelimitUpdateResponse, error)\n}\n\nfunc newHistoryAPIYARPCClient(clientConfig transport.ClientConfig, anyResolver jsonpb.AnyResolver, options ...protobuf.ClientOption) HistoryAPIYARPCClient {\n\treturn &_HistoryAPIYARPCCaller{protobuf.NewStreamClient(\n\t\tprotobuf.ClientParams{\n\t\t\tServiceName:  \"uber.cadence.history.v1.HistoryAPI\",\n\t\t\tClientConfig: clientConfig,\n\t\t\tAnyResolver:  anyResolver,\n\t\t\tOptions:      options,\n\t\t},\n\t)}\n}\n\n// NewHistoryAPIYARPCClient builds a new YARPC client for the HistoryAPI service.\nfunc NewHistoryAPIYARPCClient(clientConfig transport.ClientConfig, options ...protobuf.ClientOption) HistoryAPIYARPCClient {\n\treturn newHistoryAPIYARPCClient(clientConfig, nil, options...)\n}\n\n// HistoryAPIYARPCServer is the YARPC server-side interface for the HistoryAPI service.\ntype HistoryAPIYARPCServer interface {\n\tStartWorkflowExecution(context.Context, *StartWorkflowExecutionRequest) (*StartWorkflowExecutionResponse, error)\n\tSignalWorkflowExecution(context.Context, *SignalWorkflowExecutionRequest) (*SignalWorkflowExecutionResponse, error)\n\tSignalWithStartWorkflowExecution(context.Context, *SignalWithStartWorkflowExecutionRequest) (*SignalWithStartWorkflowExecutionResponse, error)\n\tResetWorkflowExecution(context.Context, *ResetWorkflowExecutionRequest) (*ResetWorkflowExecutionResponse, error)\n\tTerminateWorkflowExecution(context.Context, *TerminateWorkflowExecutionRequest) (*TerminateWorkflowExecutionResponse, error)\n\tDescribeWorkflowExecution(context.Context, *DescribeWorkflowExecutionRequest) (*DescribeWorkflowExecutionResponse, error)\n\tQueryWorkflow(context.Context, *QueryWorkflowRequest) (*QueryWorkflowResponse, error)\n\tResetStickyTaskList(context.Context, *ResetStickyTaskListRequest) (*ResetStickyTaskListResponse, error)\n\tGetMutableState(context.Context, *GetMutableStateRequest) (*GetMutableStateResponse, error)\n\tPollMutableState(context.Context, *PollMutableStateRequest) (*PollMutableStateResponse, error)\n\tRecordDecisionTaskStarted(context.Context, *RecordDecisionTaskStartedRequest) (*RecordDecisionTaskStartedResponse, error)\n\tRespondDecisionTaskCompleted(context.Context, *RespondDecisionTaskCompletedRequest) (*RespondDecisionTaskCompletedResponse, error)\n\tRespondDecisionTaskFailed(context.Context, *RespondDecisionTaskFailedRequest) (*RespondDecisionTaskFailedResponse, error)\n\tRecordActivityTaskStarted(context.Context, *RecordActivityTaskStartedRequest) (*RecordActivityTaskStartedResponse, error)\n\tRespondActivityTaskCompleted(context.Context, *RespondActivityTaskCompletedRequest) (*RespondActivityTaskCompletedResponse, error)\n\tRespondActivityTaskFailed(context.Context, *RespondActivityTaskFailedRequest) (*RespondActivityTaskFailedResponse, error)\n\tRespondActivityTaskCanceled(context.Context, *RespondActivityTaskCanceledRequest) (*RespondActivityTaskCanceledResponse, error)\n\tRecordActivityTaskHeartbeat(context.Context, *RecordActivityTaskHeartbeatRequest) (*RecordActivityTaskHeartbeatResponse, error)\n\tRequestCancelWorkflowExecution(context.Context, *RequestCancelWorkflowExecutionRequest) (*RequestCancelWorkflowExecutionResponse, error)\n\tRemoveSignalMutableState(context.Context, *RemoveSignalMutableStateRequest) (*RemoveSignalMutableStateResponse, error)\n\tScheduleDecisionTask(context.Context, *ScheduleDecisionTaskRequest) (*ScheduleDecisionTaskResponse, error)\n\tRecordChildExecutionCompleted(context.Context, *RecordChildExecutionCompletedRequest) (*RecordChildExecutionCompletedResponse, error)\n\tReplicateEventsV2(context.Context, *ReplicateEventsV2Request) (*ReplicateEventsV2Response, error)\n\tSyncShardStatus(context.Context, *SyncShardStatusRequest) (*SyncShardStatusResponse, error)\n\tSyncActivity(context.Context, *SyncActivityRequest) (*SyncActivityResponse, error)\n\tDescribeMutableState(context.Context, *DescribeMutableStateRequest) (*DescribeMutableStateResponse, error)\n\tDescribeHistoryHost(context.Context, *DescribeHistoryHostRequest) (*DescribeHistoryHostResponse, error)\n\tCloseShard(context.Context, *CloseShardRequest) (*CloseShardResponse, error)\n\tRemoveTask(context.Context, *RemoveTaskRequest) (*RemoveTaskResponse, error)\n\tResetQueue(context.Context, *ResetQueueRequest) (*ResetQueueResponse, error)\n\tDescribeQueue(context.Context, *DescribeQueueRequest) (*DescribeQueueResponse, error)\n\tGetReplicationMessages(context.Context, *GetReplicationMessagesRequest) (*GetReplicationMessagesResponse, error)\n\tGetDLQReplicationMessages(context.Context, *GetDLQReplicationMessagesRequest) (*GetDLQReplicationMessagesResponse, error)\n\tReapplyEvents(context.Context, *ReapplyEventsRequest) (*ReapplyEventsResponse, error)\n\tRefreshWorkflowTasks(context.Context, *RefreshWorkflowTasksRequest) (*RefreshWorkflowTasksResponse, error)\n\tCountDLQMessages(context.Context, *CountDLQMessagesRequest) (*CountDLQMessagesResponse, error)\n\tReadDLQMessages(context.Context, *ReadDLQMessagesRequest) (*ReadDLQMessagesResponse, error)\n\tPurgeDLQMessages(context.Context, *PurgeDLQMessagesRequest) (*PurgeDLQMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *MergeDLQMessagesRequest) (*MergeDLQMessagesResponse, error)\n\tNotifyFailoverMarkers(context.Context, *NotifyFailoverMarkersRequest) (*NotifyFailoverMarkersResponse, error)\n\tGetCrossClusterTasks(context.Context, *GetCrossClusterTasksRequest) (*GetCrossClusterTasksResponse, error)\n\tRespondCrossClusterTasksCompleted(context.Context, *RespondCrossClusterTasksCompletedRequest) (*RespondCrossClusterTasksCompletedResponse, error)\n\tGetFailoverInfo(context.Context, *GetFailoverInfoRequest) (*GetFailoverInfoResponse, error)\n\tRatelimitUpdate(context.Context, *RatelimitUpdateRequest) (*RatelimitUpdateResponse, error)\n}\n\ntype buildHistoryAPIYARPCProceduresParams struct {\n\tServer      HistoryAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver\n}\n\nfunc buildHistoryAPIYARPCProcedures(params buildHistoryAPIYARPCProceduresParams) []transport.Procedure {\n\thandler := &_HistoryAPIYARPCHandler{params.Server}\n\treturn protobuf.BuildProcedures(\n\t\tprotobuf.BuildProceduresParams{\n\t\t\tServiceName: \"uber.cadence.history.v1.HistoryAPI\",\n\t\t\tUnaryHandlerParams: []protobuf.BuildProceduresUnaryHandlerParams{\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"StartWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.StartWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceStartWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"SignalWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.SignalWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceSignalWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"SignalWithStartWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.SignalWithStartWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ResetWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ResetWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceResetWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"TerminateWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.TerminateWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceTerminateWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"DescribeWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.DescribeWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceDescribeWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"QueryWorkflow\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.QueryWorkflow,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceQueryWorkflowYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ResetStickyTaskList\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ResetStickyTaskList,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceResetStickyTaskListYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetMutableState\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetMutableState,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceGetMutableStateYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"PollMutableState\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.PollMutableState,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServicePollMutableStateYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RecordDecisionTaskStarted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RecordDecisionTaskStarted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRecordDecisionTaskStartedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondDecisionTaskCompleted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondDecisionTaskCompleted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondDecisionTaskCompletedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondDecisionTaskFailed\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondDecisionTaskFailed,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondDecisionTaskFailedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RecordActivityTaskStarted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RecordActivityTaskStarted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRecordActivityTaskStartedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondActivityTaskCompleted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondActivityTaskCompleted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondActivityTaskCompletedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondActivityTaskFailed\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondActivityTaskFailed,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondActivityTaskFailedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondActivityTaskCanceled\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondActivityTaskCanceled,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondActivityTaskCanceledYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RecordActivityTaskHeartbeat\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RecordActivityTaskHeartbeat,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRecordActivityTaskHeartbeatYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RequestCancelWorkflowExecution\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RequestCancelWorkflowExecution,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRequestCancelWorkflowExecutionYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RemoveSignalMutableState\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RemoveSignalMutableState,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRemoveSignalMutableStateYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ScheduleDecisionTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ScheduleDecisionTask,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceScheduleDecisionTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RecordChildExecutionCompleted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RecordChildExecutionCompleted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRecordChildExecutionCompletedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ReplicateEventsV2\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ReplicateEventsV2,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceReplicateEventsV2YARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"SyncShardStatus\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.SyncShardStatus,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceSyncShardStatusYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"SyncActivity\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.SyncActivity,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceSyncActivityYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"DescribeMutableState\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.DescribeMutableState,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceDescribeMutableStateYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"DescribeHistoryHost\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.DescribeHistoryHost,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceDescribeHistoryHostYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"CloseShard\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.CloseShard,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceCloseShardYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RemoveTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RemoveTask,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRemoveTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ResetQueue\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ResetQueue,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceResetQueueYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"DescribeQueue\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.DescribeQueue,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceDescribeQueueYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetReplicationMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetReplicationMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceGetReplicationMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetDLQReplicationMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetDLQReplicationMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceGetDLQReplicationMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ReapplyEvents\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ReapplyEvents,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceReapplyEventsYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RefreshWorkflowTasks\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RefreshWorkflowTasks,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRefreshWorkflowTasksYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"CountDLQMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.CountDLQMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceCountDLQMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ReadDLQMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ReadDLQMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceReadDLQMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"PurgeDLQMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.PurgeDLQMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServicePurgeDLQMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"MergeDLQMessages\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.MergeDLQMessages,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceMergeDLQMessagesYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"NotifyFailoverMarkers\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.NotifyFailoverMarkers,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceNotifyFailoverMarkersYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetCrossClusterTasks\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetCrossClusterTasks,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceGetCrossClusterTasksYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondCrossClusterTasksCompleted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondCrossClusterTasksCompleted,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetFailoverInfo\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetFailoverInfo,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceGetFailoverInfoYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RatelimitUpdate\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RatelimitUpdate,\n\t\t\t\t\t\t\tNewRequest:  newHistoryAPIServiceRatelimitUpdateYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tOnewayHandlerParams: []protobuf.BuildProceduresOnewayHandlerParams{},\n\t\t\tStreamHandlerParams: []protobuf.BuildProceduresStreamHandlerParams{},\n\t\t},\n\t)\n}\n\n// BuildHistoryAPIYARPCProcedures prepares an implementation of the HistoryAPI service for YARPC registration.\nfunc BuildHistoryAPIYARPCProcedures(server HistoryAPIYARPCServer) []transport.Procedure {\n\treturn buildHistoryAPIYARPCProcedures(buildHistoryAPIYARPCProceduresParams{Server: server})\n}\n\n// FxHistoryAPIYARPCClientParams defines the input\n// for NewFxHistoryAPIYARPCClient. It provides the\n// paramaters to get a HistoryAPIYARPCClient in an\n// Fx application.\ntype FxHistoryAPIYARPCClientParams struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tAnyResolver jsonpb.AnyResolver  `name:\"yarpcfx\" optional:\"true\"`\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// FxHistoryAPIYARPCClientResult defines the output\n// of NewFxHistoryAPIYARPCClient. It provides a\n// HistoryAPIYARPCClient to an Fx application.\ntype FxHistoryAPIYARPCClientResult struct {\n\tfx.Out\n\n\tClient HistoryAPIYARPCClient\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// NewFxHistoryAPIYARPCClient provides a HistoryAPIYARPCClient\n// to an Fx application using the given name for routing.\n//\n//\tfx.Provide(\n//\t  historyv1.NewFxHistoryAPIYARPCClient(\"service-name\"),\n//\t  ...\n//\t)\nfunc NewFxHistoryAPIYARPCClient(name string, options ...protobuf.ClientOption) interface{} {\n\treturn func(params FxHistoryAPIYARPCClientParams) FxHistoryAPIYARPCClientResult {\n\t\tcc := params.Provider.ClientConfig(name)\n\n\t\tif params.Restriction != nil {\n\t\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok {\n\t\t\t\tif err := params.Restriction.Check(protobuf.Encoding, namer.TransportName()); err != nil {\n\t\t\t\t\tpanic(err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn FxHistoryAPIYARPCClientResult{\n\t\t\tClient: newHistoryAPIYARPCClient(cc, params.AnyResolver, options...),\n\t\t}\n\t}\n}\n\n// FxHistoryAPIYARPCProceduresParams defines the input\n// for NewFxHistoryAPIYARPCProcedures. It provides the\n// paramaters to get HistoryAPIYARPCServer procedures in an\n// Fx application.\ntype FxHistoryAPIYARPCProceduresParams struct {\n\tfx.In\n\n\tServer      HistoryAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver `name:\"yarpcfx\" optional:\"true\"`\n}\n\n// FxHistoryAPIYARPCProceduresResult defines the output\n// of NewFxHistoryAPIYARPCProcedures. It provides\n// HistoryAPIYARPCServer procedures to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group.\n// Dig 1.2 or newer must be used for this feature to work.\ntype FxHistoryAPIYARPCProceduresResult struct {\n\tfx.Out\n\n\tProcedures     []transport.Procedure `group:\"yarpcfx\"`\n\tReflectionMeta reflection.ServerMeta `group:\"yarpcfx\"`\n}\n\n// NewFxHistoryAPIYARPCProcedures provides HistoryAPIYARPCServer procedures to an Fx application.\n// It expects a HistoryAPIYARPCServer to be present in the container.\n//\n//\tfx.Provide(\n//\t  historyv1.NewFxHistoryAPIYARPCProcedures(),\n//\t  ...\n//\t)\nfunc NewFxHistoryAPIYARPCProcedures() interface{} {\n\treturn func(params FxHistoryAPIYARPCProceduresParams) FxHistoryAPIYARPCProceduresResult {\n\t\treturn FxHistoryAPIYARPCProceduresResult{\n\t\t\tProcedures: buildHistoryAPIYARPCProcedures(buildHistoryAPIYARPCProceduresParams{\n\t\t\t\tServer:      params.Server,\n\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t}),\n\t\t\tReflectionMeta: HistoryAPIReflectionMeta,\n\t\t}\n\t}\n}\n\n// HistoryAPIReflectionMeta is the reflection server metadata\n// required for using the gRPC reflection protocol with YARPC.\n//\n// See https://github.com/grpc/grpc/blob/master/doc/server-reflection.md.\nvar HistoryAPIReflectionMeta = reflection.ServerMeta{\n\tServiceName:     \"uber.cadence.history.v1.HistoryAPI\",\n\tFileDescriptors: yarpcFileDescriptorClosurefee8ff76963a38ed,\n}\n\ntype _HistoryAPIYARPCCaller struct {\n\tstreamClient protobuf.StreamClient\n}\n\nfunc (c *_HistoryAPIYARPCCaller) StartWorkflowExecution(ctx context.Context, request *StartWorkflowExecutionRequest, options ...yarpc.CallOption) (*StartWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"StartWorkflowExecution\", request, newHistoryAPIServiceStartWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*StartWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceStartWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) SignalWorkflowExecution(ctx context.Context, request *SignalWorkflowExecutionRequest, options ...yarpc.CallOption) (*SignalWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"SignalWorkflowExecution\", request, newHistoryAPIServiceSignalWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*SignalWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSignalWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) SignalWithStartWorkflowExecution(ctx context.Context, request *SignalWithStartWorkflowExecutionRequest, options ...yarpc.CallOption) (*SignalWithStartWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"SignalWithStartWorkflowExecution\", request, newHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*SignalWithStartWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ResetWorkflowExecution(ctx context.Context, request *ResetWorkflowExecutionRequest, options ...yarpc.CallOption) (*ResetWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ResetWorkflowExecution\", request, newHistoryAPIServiceResetWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ResetWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) TerminateWorkflowExecution(ctx context.Context, request *TerminateWorkflowExecutionRequest, options ...yarpc.CallOption) (*TerminateWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"TerminateWorkflowExecution\", request, newHistoryAPIServiceTerminateWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*TerminateWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceTerminateWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) DescribeWorkflowExecution(ctx context.Context, request *DescribeWorkflowExecutionRequest, options ...yarpc.CallOption) (*DescribeWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"DescribeWorkflowExecution\", request, newHistoryAPIServiceDescribeWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*DescribeWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) QueryWorkflow(ctx context.Context, request *QueryWorkflowRequest, options ...yarpc.CallOption) (*QueryWorkflowResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"QueryWorkflow\", request, newHistoryAPIServiceQueryWorkflowYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*QueryWorkflowResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceQueryWorkflowYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ResetStickyTaskList(ctx context.Context, request *ResetStickyTaskListRequest, options ...yarpc.CallOption) (*ResetStickyTaskListResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ResetStickyTaskList\", request, newHistoryAPIServiceResetStickyTaskListYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ResetStickyTaskListResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetStickyTaskListYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) GetMutableState(ctx context.Context, request *GetMutableStateRequest, options ...yarpc.CallOption) (*GetMutableStateResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetMutableState\", request, newHistoryAPIServiceGetMutableStateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetMutableStateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetMutableStateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) PollMutableState(ctx context.Context, request *PollMutableStateRequest, options ...yarpc.CallOption) (*PollMutableStateResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"PollMutableState\", request, newHistoryAPIServicePollMutableStateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*PollMutableStateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServicePollMutableStateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RecordDecisionTaskStarted(ctx context.Context, request *RecordDecisionTaskStartedRequest, options ...yarpc.CallOption) (*RecordDecisionTaskStartedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RecordDecisionTaskStarted\", request, newHistoryAPIServiceRecordDecisionTaskStartedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RecordDecisionTaskStartedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordDecisionTaskStartedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondDecisionTaskCompleted(ctx context.Context, request *RespondDecisionTaskCompletedRequest, options ...yarpc.CallOption) (*RespondDecisionTaskCompletedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondDecisionTaskCompleted\", request, newHistoryAPIServiceRespondDecisionTaskCompletedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondDecisionTaskCompletedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondDecisionTaskCompletedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondDecisionTaskFailed(ctx context.Context, request *RespondDecisionTaskFailedRequest, options ...yarpc.CallOption) (*RespondDecisionTaskFailedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondDecisionTaskFailed\", request, newHistoryAPIServiceRespondDecisionTaskFailedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondDecisionTaskFailedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondDecisionTaskFailedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RecordActivityTaskStarted(ctx context.Context, request *RecordActivityTaskStartedRequest, options ...yarpc.CallOption) (*RecordActivityTaskStartedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RecordActivityTaskStarted\", request, newHistoryAPIServiceRecordActivityTaskStartedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RecordActivityTaskStartedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordActivityTaskStartedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondActivityTaskCompleted(ctx context.Context, request *RespondActivityTaskCompletedRequest, options ...yarpc.CallOption) (*RespondActivityTaskCompletedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondActivityTaskCompleted\", request, newHistoryAPIServiceRespondActivityTaskCompletedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondActivityTaskCompletedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskCompletedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondActivityTaskFailed(ctx context.Context, request *RespondActivityTaskFailedRequest, options ...yarpc.CallOption) (*RespondActivityTaskFailedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondActivityTaskFailed\", request, newHistoryAPIServiceRespondActivityTaskFailedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondActivityTaskFailedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskFailedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondActivityTaskCanceled(ctx context.Context, request *RespondActivityTaskCanceledRequest, options ...yarpc.CallOption) (*RespondActivityTaskCanceledResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondActivityTaskCanceled\", request, newHistoryAPIServiceRespondActivityTaskCanceledYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondActivityTaskCanceledResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskCanceledYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RecordActivityTaskHeartbeat(ctx context.Context, request *RecordActivityTaskHeartbeatRequest, options ...yarpc.CallOption) (*RecordActivityTaskHeartbeatResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RecordActivityTaskHeartbeat\", request, newHistoryAPIServiceRecordActivityTaskHeartbeatYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RecordActivityTaskHeartbeatResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordActivityTaskHeartbeatYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RequestCancelWorkflowExecution(ctx context.Context, request *RequestCancelWorkflowExecutionRequest, options ...yarpc.CallOption) (*RequestCancelWorkflowExecutionResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RequestCancelWorkflowExecution\", request, newHistoryAPIServiceRequestCancelWorkflowExecutionYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RequestCancelWorkflowExecutionResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRequestCancelWorkflowExecutionYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RemoveSignalMutableState(ctx context.Context, request *RemoveSignalMutableStateRequest, options ...yarpc.CallOption) (*RemoveSignalMutableStateResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RemoveSignalMutableState\", request, newHistoryAPIServiceRemoveSignalMutableStateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RemoveSignalMutableStateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRemoveSignalMutableStateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ScheduleDecisionTask(ctx context.Context, request *ScheduleDecisionTaskRequest, options ...yarpc.CallOption) (*ScheduleDecisionTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ScheduleDecisionTask\", request, newHistoryAPIServiceScheduleDecisionTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ScheduleDecisionTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceScheduleDecisionTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RecordChildExecutionCompleted(ctx context.Context, request *RecordChildExecutionCompletedRequest, options ...yarpc.CallOption) (*RecordChildExecutionCompletedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RecordChildExecutionCompleted\", request, newHistoryAPIServiceRecordChildExecutionCompletedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RecordChildExecutionCompletedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordChildExecutionCompletedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ReplicateEventsV2(ctx context.Context, request *ReplicateEventsV2Request, options ...yarpc.CallOption) (*ReplicateEventsV2Response, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ReplicateEventsV2\", request, newHistoryAPIServiceReplicateEventsV2YARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ReplicateEventsV2Response)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReplicateEventsV2YARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) SyncShardStatus(ctx context.Context, request *SyncShardStatusRequest, options ...yarpc.CallOption) (*SyncShardStatusResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"SyncShardStatus\", request, newHistoryAPIServiceSyncShardStatusYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*SyncShardStatusResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSyncShardStatusYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) SyncActivity(ctx context.Context, request *SyncActivityRequest, options ...yarpc.CallOption) (*SyncActivityResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"SyncActivity\", request, newHistoryAPIServiceSyncActivityYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*SyncActivityResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSyncActivityYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) DescribeMutableState(ctx context.Context, request *DescribeMutableStateRequest, options ...yarpc.CallOption) (*DescribeMutableStateResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"DescribeMutableState\", request, newHistoryAPIServiceDescribeMutableStateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*DescribeMutableStateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeMutableStateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) DescribeHistoryHost(ctx context.Context, request *DescribeHistoryHostRequest, options ...yarpc.CallOption) (*DescribeHistoryHostResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"DescribeHistoryHost\", request, newHistoryAPIServiceDescribeHistoryHostYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*DescribeHistoryHostResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeHistoryHostYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) CloseShard(ctx context.Context, request *CloseShardRequest, options ...yarpc.CallOption) (*CloseShardResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"CloseShard\", request, newHistoryAPIServiceCloseShardYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*CloseShardResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceCloseShardYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RemoveTask(ctx context.Context, request *RemoveTaskRequest, options ...yarpc.CallOption) (*RemoveTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RemoveTask\", request, newHistoryAPIServiceRemoveTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RemoveTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRemoveTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ResetQueue(ctx context.Context, request *ResetQueueRequest, options ...yarpc.CallOption) (*ResetQueueResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ResetQueue\", request, newHistoryAPIServiceResetQueueYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ResetQueueResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetQueueYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) DescribeQueue(ctx context.Context, request *DescribeQueueRequest, options ...yarpc.CallOption) (*DescribeQueueResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"DescribeQueue\", request, newHistoryAPIServiceDescribeQueueYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*DescribeQueueResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeQueueYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) GetReplicationMessages(ctx context.Context, request *GetReplicationMessagesRequest, options ...yarpc.CallOption) (*GetReplicationMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetReplicationMessages\", request, newHistoryAPIServiceGetReplicationMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetReplicationMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetReplicationMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) GetDLQReplicationMessages(ctx context.Context, request *GetDLQReplicationMessagesRequest, options ...yarpc.CallOption) (*GetDLQReplicationMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetDLQReplicationMessages\", request, newHistoryAPIServiceGetDLQReplicationMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetDLQReplicationMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetDLQReplicationMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ReapplyEvents(ctx context.Context, request *ReapplyEventsRequest, options ...yarpc.CallOption) (*ReapplyEventsResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ReapplyEvents\", request, newHistoryAPIServiceReapplyEventsYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ReapplyEventsResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReapplyEventsYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RefreshWorkflowTasks(ctx context.Context, request *RefreshWorkflowTasksRequest, options ...yarpc.CallOption) (*RefreshWorkflowTasksResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RefreshWorkflowTasks\", request, newHistoryAPIServiceRefreshWorkflowTasksYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RefreshWorkflowTasksResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRefreshWorkflowTasksYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) CountDLQMessages(ctx context.Context, request *CountDLQMessagesRequest, options ...yarpc.CallOption) (*CountDLQMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"CountDLQMessages\", request, newHistoryAPIServiceCountDLQMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*CountDLQMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceCountDLQMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) ReadDLQMessages(ctx context.Context, request *ReadDLQMessagesRequest, options ...yarpc.CallOption) (*ReadDLQMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ReadDLQMessages\", request, newHistoryAPIServiceReadDLQMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ReadDLQMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReadDLQMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) PurgeDLQMessages(ctx context.Context, request *PurgeDLQMessagesRequest, options ...yarpc.CallOption) (*PurgeDLQMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"PurgeDLQMessages\", request, newHistoryAPIServicePurgeDLQMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*PurgeDLQMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServicePurgeDLQMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) MergeDLQMessages(ctx context.Context, request *MergeDLQMessagesRequest, options ...yarpc.CallOption) (*MergeDLQMessagesResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"MergeDLQMessages\", request, newHistoryAPIServiceMergeDLQMessagesYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*MergeDLQMessagesResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceMergeDLQMessagesYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) NotifyFailoverMarkers(ctx context.Context, request *NotifyFailoverMarkersRequest, options ...yarpc.CallOption) (*NotifyFailoverMarkersResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"NotifyFailoverMarkers\", request, newHistoryAPIServiceNotifyFailoverMarkersYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*NotifyFailoverMarkersResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceNotifyFailoverMarkersYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) GetCrossClusterTasks(ctx context.Context, request *GetCrossClusterTasksRequest, options ...yarpc.CallOption) (*GetCrossClusterTasksResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetCrossClusterTasks\", request, newHistoryAPIServiceGetCrossClusterTasksYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetCrossClusterTasksResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetCrossClusterTasksYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RespondCrossClusterTasksCompleted(ctx context.Context, request *RespondCrossClusterTasksCompletedRequest, options ...yarpc.CallOption) (*RespondCrossClusterTasksCompletedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondCrossClusterTasksCompleted\", request, newHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondCrossClusterTasksCompletedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) GetFailoverInfo(ctx context.Context, request *GetFailoverInfoRequest, options ...yarpc.CallOption) (*GetFailoverInfoResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetFailoverInfo\", request, newHistoryAPIServiceGetFailoverInfoYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetFailoverInfoResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetFailoverInfoYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_HistoryAPIYARPCCaller) RatelimitUpdate(ctx context.Context, request *RatelimitUpdateRequest, options ...yarpc.CallOption) (*RatelimitUpdateResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RatelimitUpdate\", request, newHistoryAPIServiceRatelimitUpdateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RatelimitUpdateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRatelimitUpdateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\ntype _HistoryAPIYARPCHandler struct {\n\tserver HistoryAPIYARPCServer\n}\n\nfunc (h *_HistoryAPIYARPCHandler) StartWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *StartWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*StartWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceStartWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.StartWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) SignalWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *SignalWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*SignalWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSignalWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.SignalWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) SignalWithStartWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *SignalWithStartWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*SignalWithStartWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.SignalWithStartWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ResetWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ResetWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ResetWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ResetWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) TerminateWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *TerminateWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*TerminateWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceTerminateWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.TerminateWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) DescribeWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *DescribeWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*DescribeWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.DescribeWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) QueryWorkflow(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *QueryWorkflowRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*QueryWorkflowRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceQueryWorkflowYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.QueryWorkflow(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ResetStickyTaskList(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ResetStickyTaskListRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ResetStickyTaskListRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetStickyTaskListYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ResetStickyTaskList(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) GetMutableState(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetMutableStateRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetMutableStateRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetMutableStateYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetMutableState(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) PollMutableState(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *PollMutableStateRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*PollMutableStateRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServicePollMutableStateYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.PollMutableState(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RecordDecisionTaskStarted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RecordDecisionTaskStartedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RecordDecisionTaskStartedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordDecisionTaskStartedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RecordDecisionTaskStarted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondDecisionTaskCompleted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondDecisionTaskCompletedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondDecisionTaskCompletedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondDecisionTaskCompletedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondDecisionTaskCompleted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondDecisionTaskFailed(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondDecisionTaskFailedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondDecisionTaskFailedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondDecisionTaskFailedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondDecisionTaskFailed(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RecordActivityTaskStarted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RecordActivityTaskStartedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RecordActivityTaskStartedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordActivityTaskStartedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RecordActivityTaskStarted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondActivityTaskCompleted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondActivityTaskCompletedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondActivityTaskCompletedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskCompletedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondActivityTaskCompleted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondActivityTaskFailed(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondActivityTaskFailedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondActivityTaskFailedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskFailedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondActivityTaskFailed(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondActivityTaskCanceled(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondActivityTaskCanceledRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondActivityTaskCanceledRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondActivityTaskCanceledYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondActivityTaskCanceled(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RecordActivityTaskHeartbeat(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RecordActivityTaskHeartbeatRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RecordActivityTaskHeartbeatRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordActivityTaskHeartbeatYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RecordActivityTaskHeartbeat(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RequestCancelWorkflowExecution(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RequestCancelWorkflowExecutionRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RequestCancelWorkflowExecutionRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRequestCancelWorkflowExecutionYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RequestCancelWorkflowExecution(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RemoveSignalMutableState(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RemoveSignalMutableStateRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RemoveSignalMutableStateRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRemoveSignalMutableStateYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RemoveSignalMutableState(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ScheduleDecisionTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ScheduleDecisionTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ScheduleDecisionTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceScheduleDecisionTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ScheduleDecisionTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RecordChildExecutionCompleted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RecordChildExecutionCompletedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RecordChildExecutionCompletedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRecordChildExecutionCompletedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RecordChildExecutionCompleted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ReplicateEventsV2(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ReplicateEventsV2Request\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ReplicateEventsV2Request)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReplicateEventsV2YARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ReplicateEventsV2(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) SyncShardStatus(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *SyncShardStatusRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*SyncShardStatusRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSyncShardStatusYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.SyncShardStatus(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) SyncActivity(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *SyncActivityRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*SyncActivityRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceSyncActivityYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.SyncActivity(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) DescribeMutableState(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *DescribeMutableStateRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*DescribeMutableStateRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeMutableStateYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.DescribeMutableState(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) DescribeHistoryHost(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *DescribeHistoryHostRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*DescribeHistoryHostRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeHistoryHostYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.DescribeHistoryHost(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) CloseShard(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *CloseShardRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*CloseShardRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceCloseShardYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.CloseShard(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RemoveTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RemoveTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RemoveTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRemoveTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RemoveTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ResetQueue(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ResetQueueRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ResetQueueRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceResetQueueYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ResetQueue(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) DescribeQueue(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *DescribeQueueRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*DescribeQueueRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceDescribeQueueYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.DescribeQueue(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) GetReplicationMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetReplicationMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetReplicationMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetReplicationMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetReplicationMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) GetDLQReplicationMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetDLQReplicationMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetDLQReplicationMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetDLQReplicationMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetDLQReplicationMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ReapplyEvents(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ReapplyEventsRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ReapplyEventsRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReapplyEventsYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ReapplyEvents(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RefreshWorkflowTasks(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RefreshWorkflowTasksRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RefreshWorkflowTasksRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRefreshWorkflowTasksYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RefreshWorkflowTasks(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) CountDLQMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *CountDLQMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*CountDLQMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceCountDLQMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.CountDLQMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) ReadDLQMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ReadDLQMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ReadDLQMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceReadDLQMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ReadDLQMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) PurgeDLQMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *PurgeDLQMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*PurgeDLQMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServicePurgeDLQMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.PurgeDLQMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) MergeDLQMessages(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *MergeDLQMessagesRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*MergeDLQMessagesRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceMergeDLQMessagesYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.MergeDLQMessages(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) NotifyFailoverMarkers(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *NotifyFailoverMarkersRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*NotifyFailoverMarkersRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceNotifyFailoverMarkersYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.NotifyFailoverMarkers(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) GetCrossClusterTasks(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetCrossClusterTasksRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetCrossClusterTasksRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetCrossClusterTasksYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetCrossClusterTasks(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RespondCrossClusterTasksCompleted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondCrossClusterTasksCompletedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondCrossClusterTasksCompletedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondCrossClusterTasksCompleted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) GetFailoverInfo(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetFailoverInfoRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetFailoverInfoRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceGetFailoverInfoYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetFailoverInfo(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_HistoryAPIYARPCHandler) RatelimitUpdate(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RatelimitUpdateRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RatelimitUpdateRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyHistoryAPIServiceRatelimitUpdateYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RatelimitUpdate(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc newHistoryAPIServiceStartWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &StartWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceStartWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &StartWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceSignalWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &SignalWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceSignalWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &SignalWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &SignalWithStartWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &SignalWithStartWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceResetWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &ResetWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceResetWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &ResetWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceTerminateWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &TerminateWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceTerminateWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &TerminateWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceDescribeWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &DescribeWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceDescribeWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &DescribeWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceQueryWorkflowYARPCRequest() proto.Message {\n\treturn &QueryWorkflowRequest{}\n}\n\nfunc newHistoryAPIServiceQueryWorkflowYARPCResponse() proto.Message {\n\treturn &QueryWorkflowResponse{}\n}\n\nfunc newHistoryAPIServiceResetStickyTaskListYARPCRequest() proto.Message {\n\treturn &ResetStickyTaskListRequest{}\n}\n\nfunc newHistoryAPIServiceResetStickyTaskListYARPCResponse() proto.Message {\n\treturn &ResetStickyTaskListResponse{}\n}\n\nfunc newHistoryAPIServiceGetMutableStateYARPCRequest() proto.Message {\n\treturn &GetMutableStateRequest{}\n}\n\nfunc newHistoryAPIServiceGetMutableStateYARPCResponse() proto.Message {\n\treturn &GetMutableStateResponse{}\n}\n\nfunc newHistoryAPIServicePollMutableStateYARPCRequest() proto.Message {\n\treturn &PollMutableStateRequest{}\n}\n\nfunc newHistoryAPIServicePollMutableStateYARPCResponse() proto.Message {\n\treturn &PollMutableStateResponse{}\n}\n\nfunc newHistoryAPIServiceRecordDecisionTaskStartedYARPCRequest() proto.Message {\n\treturn &RecordDecisionTaskStartedRequest{}\n}\n\nfunc newHistoryAPIServiceRecordDecisionTaskStartedYARPCResponse() proto.Message {\n\treturn &RecordDecisionTaskStartedResponse{}\n}\n\nfunc newHistoryAPIServiceRespondDecisionTaskCompletedYARPCRequest() proto.Message {\n\treturn &RespondDecisionTaskCompletedRequest{}\n}\n\nfunc newHistoryAPIServiceRespondDecisionTaskCompletedYARPCResponse() proto.Message {\n\treturn &RespondDecisionTaskCompletedResponse{}\n}\n\nfunc newHistoryAPIServiceRespondDecisionTaskFailedYARPCRequest() proto.Message {\n\treturn &RespondDecisionTaskFailedRequest{}\n}\n\nfunc newHistoryAPIServiceRespondDecisionTaskFailedYARPCResponse() proto.Message {\n\treturn &RespondDecisionTaskFailedResponse{}\n}\n\nfunc newHistoryAPIServiceRecordActivityTaskStartedYARPCRequest() proto.Message {\n\treturn &RecordActivityTaskStartedRequest{}\n}\n\nfunc newHistoryAPIServiceRecordActivityTaskStartedYARPCResponse() proto.Message {\n\treturn &RecordActivityTaskStartedResponse{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskCompletedYARPCRequest() proto.Message {\n\treturn &RespondActivityTaskCompletedRequest{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskCompletedYARPCResponse() proto.Message {\n\treturn &RespondActivityTaskCompletedResponse{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskFailedYARPCRequest() proto.Message {\n\treturn &RespondActivityTaskFailedRequest{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskFailedYARPCResponse() proto.Message {\n\treturn &RespondActivityTaskFailedResponse{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskCanceledYARPCRequest() proto.Message {\n\treturn &RespondActivityTaskCanceledRequest{}\n}\n\nfunc newHistoryAPIServiceRespondActivityTaskCanceledYARPCResponse() proto.Message {\n\treturn &RespondActivityTaskCanceledResponse{}\n}\n\nfunc newHistoryAPIServiceRecordActivityTaskHeartbeatYARPCRequest() proto.Message {\n\treturn &RecordActivityTaskHeartbeatRequest{}\n}\n\nfunc newHistoryAPIServiceRecordActivityTaskHeartbeatYARPCResponse() proto.Message {\n\treturn &RecordActivityTaskHeartbeatResponse{}\n}\n\nfunc newHistoryAPIServiceRequestCancelWorkflowExecutionYARPCRequest() proto.Message {\n\treturn &RequestCancelWorkflowExecutionRequest{}\n}\n\nfunc newHistoryAPIServiceRequestCancelWorkflowExecutionYARPCResponse() proto.Message {\n\treturn &RequestCancelWorkflowExecutionResponse{}\n}\n\nfunc newHistoryAPIServiceRemoveSignalMutableStateYARPCRequest() proto.Message {\n\treturn &RemoveSignalMutableStateRequest{}\n}\n\nfunc newHistoryAPIServiceRemoveSignalMutableStateYARPCResponse() proto.Message {\n\treturn &RemoveSignalMutableStateResponse{}\n}\n\nfunc newHistoryAPIServiceScheduleDecisionTaskYARPCRequest() proto.Message {\n\treturn &ScheduleDecisionTaskRequest{}\n}\n\nfunc newHistoryAPIServiceScheduleDecisionTaskYARPCResponse() proto.Message {\n\treturn &ScheduleDecisionTaskResponse{}\n}\n\nfunc newHistoryAPIServiceRecordChildExecutionCompletedYARPCRequest() proto.Message {\n\treturn &RecordChildExecutionCompletedRequest{}\n}\n\nfunc newHistoryAPIServiceRecordChildExecutionCompletedYARPCResponse() proto.Message {\n\treturn &RecordChildExecutionCompletedResponse{}\n}\n\nfunc newHistoryAPIServiceReplicateEventsV2YARPCRequest() proto.Message {\n\treturn &ReplicateEventsV2Request{}\n}\n\nfunc newHistoryAPIServiceReplicateEventsV2YARPCResponse() proto.Message {\n\treturn &ReplicateEventsV2Response{}\n}\n\nfunc newHistoryAPIServiceSyncShardStatusYARPCRequest() proto.Message {\n\treturn &SyncShardStatusRequest{}\n}\n\nfunc newHistoryAPIServiceSyncShardStatusYARPCResponse() proto.Message {\n\treturn &SyncShardStatusResponse{}\n}\n\nfunc newHistoryAPIServiceSyncActivityYARPCRequest() proto.Message {\n\treturn &SyncActivityRequest{}\n}\n\nfunc newHistoryAPIServiceSyncActivityYARPCResponse() proto.Message {\n\treturn &SyncActivityResponse{}\n}\n\nfunc newHistoryAPIServiceDescribeMutableStateYARPCRequest() proto.Message {\n\treturn &DescribeMutableStateRequest{}\n}\n\nfunc newHistoryAPIServiceDescribeMutableStateYARPCResponse() proto.Message {\n\treturn &DescribeMutableStateResponse{}\n}\n\nfunc newHistoryAPIServiceDescribeHistoryHostYARPCRequest() proto.Message {\n\treturn &DescribeHistoryHostRequest{}\n}\n\nfunc newHistoryAPIServiceDescribeHistoryHostYARPCResponse() proto.Message {\n\treturn &DescribeHistoryHostResponse{}\n}\n\nfunc newHistoryAPIServiceCloseShardYARPCRequest() proto.Message {\n\treturn &CloseShardRequest{}\n}\n\nfunc newHistoryAPIServiceCloseShardYARPCResponse() proto.Message {\n\treturn &CloseShardResponse{}\n}\n\nfunc newHistoryAPIServiceRemoveTaskYARPCRequest() proto.Message {\n\treturn &RemoveTaskRequest{}\n}\n\nfunc newHistoryAPIServiceRemoveTaskYARPCResponse() proto.Message {\n\treturn &RemoveTaskResponse{}\n}\n\nfunc newHistoryAPIServiceResetQueueYARPCRequest() proto.Message {\n\treturn &ResetQueueRequest{}\n}\n\nfunc newHistoryAPIServiceResetQueueYARPCResponse() proto.Message {\n\treturn &ResetQueueResponse{}\n}\n\nfunc newHistoryAPIServiceDescribeQueueYARPCRequest() proto.Message {\n\treturn &DescribeQueueRequest{}\n}\n\nfunc newHistoryAPIServiceDescribeQueueYARPCResponse() proto.Message {\n\treturn &DescribeQueueResponse{}\n}\n\nfunc newHistoryAPIServiceGetReplicationMessagesYARPCRequest() proto.Message {\n\treturn &GetReplicationMessagesRequest{}\n}\n\nfunc newHistoryAPIServiceGetReplicationMessagesYARPCResponse() proto.Message {\n\treturn &GetReplicationMessagesResponse{}\n}\n\nfunc newHistoryAPIServiceGetDLQReplicationMessagesYARPCRequest() proto.Message {\n\treturn &GetDLQReplicationMessagesRequest{}\n}\n\nfunc newHistoryAPIServiceGetDLQReplicationMessagesYARPCResponse() proto.Message {\n\treturn &GetDLQReplicationMessagesResponse{}\n}\n\nfunc newHistoryAPIServiceReapplyEventsYARPCRequest() proto.Message {\n\treturn &ReapplyEventsRequest{}\n}\n\nfunc newHistoryAPIServiceReapplyEventsYARPCResponse() proto.Message {\n\treturn &ReapplyEventsResponse{}\n}\n\nfunc newHistoryAPIServiceRefreshWorkflowTasksYARPCRequest() proto.Message {\n\treturn &RefreshWorkflowTasksRequest{}\n}\n\nfunc newHistoryAPIServiceRefreshWorkflowTasksYARPCResponse() proto.Message {\n\treturn &RefreshWorkflowTasksResponse{}\n}\n\nfunc newHistoryAPIServiceCountDLQMessagesYARPCRequest() proto.Message {\n\treturn &CountDLQMessagesRequest{}\n}\n\nfunc newHistoryAPIServiceCountDLQMessagesYARPCResponse() proto.Message {\n\treturn &CountDLQMessagesResponse{}\n}\n\nfunc newHistoryAPIServiceReadDLQMessagesYARPCRequest() proto.Message {\n\treturn &ReadDLQMessagesRequest{}\n}\n\nfunc newHistoryAPIServiceReadDLQMessagesYARPCResponse() proto.Message {\n\treturn &ReadDLQMessagesResponse{}\n}\n\nfunc newHistoryAPIServicePurgeDLQMessagesYARPCRequest() proto.Message {\n\treturn &PurgeDLQMessagesRequest{}\n}\n\nfunc newHistoryAPIServicePurgeDLQMessagesYARPCResponse() proto.Message {\n\treturn &PurgeDLQMessagesResponse{}\n}\n\nfunc newHistoryAPIServiceMergeDLQMessagesYARPCRequest() proto.Message {\n\treturn &MergeDLQMessagesRequest{}\n}\n\nfunc newHistoryAPIServiceMergeDLQMessagesYARPCResponse() proto.Message {\n\treturn &MergeDLQMessagesResponse{}\n}\n\nfunc newHistoryAPIServiceNotifyFailoverMarkersYARPCRequest() proto.Message {\n\treturn &NotifyFailoverMarkersRequest{}\n}\n\nfunc newHistoryAPIServiceNotifyFailoverMarkersYARPCResponse() proto.Message {\n\treturn &NotifyFailoverMarkersResponse{}\n}\n\nfunc newHistoryAPIServiceGetCrossClusterTasksYARPCRequest() proto.Message {\n\treturn &GetCrossClusterTasksRequest{}\n}\n\nfunc newHistoryAPIServiceGetCrossClusterTasksYARPCResponse() proto.Message {\n\treturn &GetCrossClusterTasksResponse{}\n}\n\nfunc newHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCRequest() proto.Message {\n\treturn &RespondCrossClusterTasksCompletedRequest{}\n}\n\nfunc newHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCResponse() proto.Message {\n\treturn &RespondCrossClusterTasksCompletedResponse{}\n}\n\nfunc newHistoryAPIServiceGetFailoverInfoYARPCRequest() proto.Message {\n\treturn &GetFailoverInfoRequest{}\n}\n\nfunc newHistoryAPIServiceGetFailoverInfoYARPCResponse() proto.Message {\n\treturn &GetFailoverInfoResponse{}\n}\n\nfunc newHistoryAPIServiceRatelimitUpdateYARPCRequest() proto.Message {\n\treturn &RatelimitUpdateRequest{}\n}\n\nfunc newHistoryAPIServiceRatelimitUpdateYARPCResponse() proto.Message {\n\treturn &RatelimitUpdateResponse{}\n}\n\nvar (\n\temptyHistoryAPIServiceStartWorkflowExecutionYARPCRequest             = &StartWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceStartWorkflowExecutionYARPCResponse            = &StartWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceSignalWorkflowExecutionYARPCRequest            = &SignalWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceSignalWorkflowExecutionYARPCResponse           = &SignalWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCRequest   = &SignalWithStartWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceSignalWithStartWorkflowExecutionYARPCResponse  = &SignalWithStartWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceResetWorkflowExecutionYARPCRequest             = &ResetWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceResetWorkflowExecutionYARPCResponse            = &ResetWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceTerminateWorkflowExecutionYARPCRequest         = &TerminateWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceTerminateWorkflowExecutionYARPCResponse        = &TerminateWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceDescribeWorkflowExecutionYARPCRequest          = &DescribeWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceDescribeWorkflowExecutionYARPCResponse         = &DescribeWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceQueryWorkflowYARPCRequest                      = &QueryWorkflowRequest{}\n\temptyHistoryAPIServiceQueryWorkflowYARPCResponse                     = &QueryWorkflowResponse{}\n\temptyHistoryAPIServiceResetStickyTaskListYARPCRequest                = &ResetStickyTaskListRequest{}\n\temptyHistoryAPIServiceResetStickyTaskListYARPCResponse               = &ResetStickyTaskListResponse{}\n\temptyHistoryAPIServiceGetMutableStateYARPCRequest                    = &GetMutableStateRequest{}\n\temptyHistoryAPIServiceGetMutableStateYARPCResponse                   = &GetMutableStateResponse{}\n\temptyHistoryAPIServicePollMutableStateYARPCRequest                   = &PollMutableStateRequest{}\n\temptyHistoryAPIServicePollMutableStateYARPCResponse                  = &PollMutableStateResponse{}\n\temptyHistoryAPIServiceRecordDecisionTaskStartedYARPCRequest          = &RecordDecisionTaskStartedRequest{}\n\temptyHistoryAPIServiceRecordDecisionTaskStartedYARPCResponse         = &RecordDecisionTaskStartedResponse{}\n\temptyHistoryAPIServiceRespondDecisionTaskCompletedYARPCRequest       = &RespondDecisionTaskCompletedRequest{}\n\temptyHistoryAPIServiceRespondDecisionTaskCompletedYARPCResponse      = &RespondDecisionTaskCompletedResponse{}\n\temptyHistoryAPIServiceRespondDecisionTaskFailedYARPCRequest          = &RespondDecisionTaskFailedRequest{}\n\temptyHistoryAPIServiceRespondDecisionTaskFailedYARPCResponse         = &RespondDecisionTaskFailedResponse{}\n\temptyHistoryAPIServiceRecordActivityTaskStartedYARPCRequest          = &RecordActivityTaskStartedRequest{}\n\temptyHistoryAPIServiceRecordActivityTaskStartedYARPCResponse         = &RecordActivityTaskStartedResponse{}\n\temptyHistoryAPIServiceRespondActivityTaskCompletedYARPCRequest       = &RespondActivityTaskCompletedRequest{}\n\temptyHistoryAPIServiceRespondActivityTaskCompletedYARPCResponse      = &RespondActivityTaskCompletedResponse{}\n\temptyHistoryAPIServiceRespondActivityTaskFailedYARPCRequest          = &RespondActivityTaskFailedRequest{}\n\temptyHistoryAPIServiceRespondActivityTaskFailedYARPCResponse         = &RespondActivityTaskFailedResponse{}\n\temptyHistoryAPIServiceRespondActivityTaskCanceledYARPCRequest        = &RespondActivityTaskCanceledRequest{}\n\temptyHistoryAPIServiceRespondActivityTaskCanceledYARPCResponse       = &RespondActivityTaskCanceledResponse{}\n\temptyHistoryAPIServiceRecordActivityTaskHeartbeatYARPCRequest        = &RecordActivityTaskHeartbeatRequest{}\n\temptyHistoryAPIServiceRecordActivityTaskHeartbeatYARPCResponse       = &RecordActivityTaskHeartbeatResponse{}\n\temptyHistoryAPIServiceRequestCancelWorkflowExecutionYARPCRequest     = &RequestCancelWorkflowExecutionRequest{}\n\temptyHistoryAPIServiceRequestCancelWorkflowExecutionYARPCResponse    = &RequestCancelWorkflowExecutionResponse{}\n\temptyHistoryAPIServiceRemoveSignalMutableStateYARPCRequest           = &RemoveSignalMutableStateRequest{}\n\temptyHistoryAPIServiceRemoveSignalMutableStateYARPCResponse          = &RemoveSignalMutableStateResponse{}\n\temptyHistoryAPIServiceScheduleDecisionTaskYARPCRequest               = &ScheduleDecisionTaskRequest{}\n\temptyHistoryAPIServiceScheduleDecisionTaskYARPCResponse              = &ScheduleDecisionTaskResponse{}\n\temptyHistoryAPIServiceRecordChildExecutionCompletedYARPCRequest      = &RecordChildExecutionCompletedRequest{}\n\temptyHistoryAPIServiceRecordChildExecutionCompletedYARPCResponse     = &RecordChildExecutionCompletedResponse{}\n\temptyHistoryAPIServiceReplicateEventsV2YARPCRequest                  = &ReplicateEventsV2Request{}\n\temptyHistoryAPIServiceReplicateEventsV2YARPCResponse                 = &ReplicateEventsV2Response{}\n\temptyHistoryAPIServiceSyncShardStatusYARPCRequest                    = &SyncShardStatusRequest{}\n\temptyHistoryAPIServiceSyncShardStatusYARPCResponse                   = &SyncShardStatusResponse{}\n\temptyHistoryAPIServiceSyncActivityYARPCRequest                       = &SyncActivityRequest{}\n\temptyHistoryAPIServiceSyncActivityYARPCResponse                      = &SyncActivityResponse{}\n\temptyHistoryAPIServiceDescribeMutableStateYARPCRequest               = &DescribeMutableStateRequest{}\n\temptyHistoryAPIServiceDescribeMutableStateYARPCResponse              = &DescribeMutableStateResponse{}\n\temptyHistoryAPIServiceDescribeHistoryHostYARPCRequest                = &DescribeHistoryHostRequest{}\n\temptyHistoryAPIServiceDescribeHistoryHostYARPCResponse               = &DescribeHistoryHostResponse{}\n\temptyHistoryAPIServiceCloseShardYARPCRequest                         = &CloseShardRequest{}\n\temptyHistoryAPIServiceCloseShardYARPCResponse                        = &CloseShardResponse{}\n\temptyHistoryAPIServiceRemoveTaskYARPCRequest                         = &RemoveTaskRequest{}\n\temptyHistoryAPIServiceRemoveTaskYARPCResponse                        = &RemoveTaskResponse{}\n\temptyHistoryAPIServiceResetQueueYARPCRequest                         = &ResetQueueRequest{}\n\temptyHistoryAPIServiceResetQueueYARPCResponse                        = &ResetQueueResponse{}\n\temptyHistoryAPIServiceDescribeQueueYARPCRequest                      = &DescribeQueueRequest{}\n\temptyHistoryAPIServiceDescribeQueueYARPCResponse                     = &DescribeQueueResponse{}\n\temptyHistoryAPIServiceGetReplicationMessagesYARPCRequest             = &GetReplicationMessagesRequest{}\n\temptyHistoryAPIServiceGetReplicationMessagesYARPCResponse            = &GetReplicationMessagesResponse{}\n\temptyHistoryAPIServiceGetDLQReplicationMessagesYARPCRequest          = &GetDLQReplicationMessagesRequest{}\n\temptyHistoryAPIServiceGetDLQReplicationMessagesYARPCResponse         = &GetDLQReplicationMessagesResponse{}\n\temptyHistoryAPIServiceReapplyEventsYARPCRequest                      = &ReapplyEventsRequest{}\n\temptyHistoryAPIServiceReapplyEventsYARPCResponse                     = &ReapplyEventsResponse{}\n\temptyHistoryAPIServiceRefreshWorkflowTasksYARPCRequest               = &RefreshWorkflowTasksRequest{}\n\temptyHistoryAPIServiceRefreshWorkflowTasksYARPCResponse              = &RefreshWorkflowTasksResponse{}\n\temptyHistoryAPIServiceCountDLQMessagesYARPCRequest                   = &CountDLQMessagesRequest{}\n\temptyHistoryAPIServiceCountDLQMessagesYARPCResponse                  = &CountDLQMessagesResponse{}\n\temptyHistoryAPIServiceReadDLQMessagesYARPCRequest                    = &ReadDLQMessagesRequest{}\n\temptyHistoryAPIServiceReadDLQMessagesYARPCResponse                   = &ReadDLQMessagesResponse{}\n\temptyHistoryAPIServicePurgeDLQMessagesYARPCRequest                   = &PurgeDLQMessagesRequest{}\n\temptyHistoryAPIServicePurgeDLQMessagesYARPCResponse                  = &PurgeDLQMessagesResponse{}\n\temptyHistoryAPIServiceMergeDLQMessagesYARPCRequest                   = &MergeDLQMessagesRequest{}\n\temptyHistoryAPIServiceMergeDLQMessagesYARPCResponse                  = &MergeDLQMessagesResponse{}\n\temptyHistoryAPIServiceNotifyFailoverMarkersYARPCRequest              = &NotifyFailoverMarkersRequest{}\n\temptyHistoryAPIServiceNotifyFailoverMarkersYARPCResponse             = &NotifyFailoverMarkersResponse{}\n\temptyHistoryAPIServiceGetCrossClusterTasksYARPCRequest               = &GetCrossClusterTasksRequest{}\n\temptyHistoryAPIServiceGetCrossClusterTasksYARPCResponse              = &GetCrossClusterTasksResponse{}\n\temptyHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCRequest  = &RespondCrossClusterTasksCompletedRequest{}\n\temptyHistoryAPIServiceRespondCrossClusterTasksCompletedYARPCResponse = &RespondCrossClusterTasksCompletedResponse{}\n\temptyHistoryAPIServiceGetFailoverInfoYARPCRequest                    = &GetFailoverInfoRequest{}\n\temptyHistoryAPIServiceGetFailoverInfoYARPCResponse                   = &GetFailoverInfoResponse{}\n\temptyHistoryAPIServiceRatelimitUpdateYARPCRequest                    = &RatelimitUpdateRequest{}\n\temptyHistoryAPIServiceRatelimitUpdateYARPCResponse                   = &RatelimitUpdateResponse{}\n)\n\nvar yarpcFileDescriptorClosurefee8ff76963a38ed = [][]byte{\n\t// uber/cadence/history/v1/service.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x3c, 0x4b, 0x6c, 0x1c, 0x47,\n\t\t0x76, 0xe8, 0x19, 0xf1, 0xf7, 0x48, 0x0e, 0xc9, 0x12, 0x3f, 0xc3, 0xa1, 0x44, 0x91, 0xbd, 0x96,\n\t\t0x44, 0xcb, 0xeb, 0xa1, 0x45, 0xd9, 0xfa, 0x59, 0x5e, 0xad, 0x44, 0x4a, 0xf2, 0x38, 0xfa, 0x36,\n\t\t0x69, 0x39, 0x5f, 0xf7, 0x36, 0xa7, 0x6b, 0xc8, 0x8e, 0x7a, 0xba, 0x47, 0xdd, 0x3d, 0xa4, 0xe8,\n\t\t0x43, 0xe0, 0xc4, 0x41, 0x80, 0x2c, 0x82, 0xec, 0x66, 0x91, 0x04, 0x01, 0x02, 0x04, 0x08, 0x36,\n\t\t0xc0, 0x62, 0x8d, 0xdc, 0x12, 0x20, 0x87, 0x24, 0xa7, 0x5c, 0x72, 0xcc, 0x35, 0xf7, 0xdd, 0x43,\n\t\t0x02, 0xe4, 0xb6, 0xe7, 0x20, 0xa8, 0x4f, 0xf7, 0xf4, 0xa7, 0xba, 0x7a, 0x86, 0x0c, 0x22, 0xaf,\n\t\t0xe3, 0xdb, 0x74, 0x55, 0xbd, 0x57, 0xaf, 0x5e, 0xbd, 0xf7, 0xfa, 0xfd, 0x7a, 0xe0, 0x7c, 0x77,\n\t\t0x17, 0x7b, 0xeb, 0x4d, 0xc3, 0xc4, 0x4e, 0x13, 0xaf, 0xef, 0x5b, 0x7e, 0xe0, 0x7a, 0x47, 0xeb,\n\t\t0x07, 0x97, 0xd7, 0x7d, 0xec, 0x1d, 0x58, 0x4d, 0x5c, 0xef, 0x78, 0x6e, 0xe0, 0xa2, 0x05, 0xb2,\n\t\t0xac, 0xce, 0x97, 0xd5, 0xf9, 0xb2, 0xfa, 0xc1, 0xe5, 0xda, 0xf2, 0x9e, 0xeb, 0xee, 0xd9, 0x78,\n\t\t0x9d, 0x2e, 0xdb, 0xed, 0xb6, 0xd6, 0xcd, 0xae, 0x67, 0x04, 0x96, 0xeb, 0x30, 0xc0, 0xda, 0xb9,\n\t\t0xf4, 0x7c, 0x60, 0xb5, 0xb1, 0x1f, 0x18, 0xed, 0x0e, 0x5f, 0x90, 0x41, 0x70, 0xe8, 0x19, 0x9d,\n\t\t0x0e, 0xf6, 0x7c, 0x3e, 0xbf, 0x92, 0x20, 0xd0, 0xe8, 0x58, 0x84, 0xb8, 0xa6, 0xdb, 0x6e, 0x47,\n\t\t0x5b, 0xac, 0x8a, 0x56, 0x84, 0x24, 0x72, 0x2a, 0x44, 0x4b, 0x5e, 0x76, 0x71, 0xb4, 0x40, 0x15,\n\t\t0x2d, 0x08, 0x0c, 0xff, 0x85, 0x6d, 0xf9, 0x81, 0x6c, 0xcd, 0xa1, 0xeb, 0xbd, 0x68, 0xd9, 0xee,\n\t\t0x21, 0x5f, 0x73, 0x49, 0xb4, 0x86, 0xb3, 0x52, 0x4f, 0xad, 0x5d, 0x2b, 0x5a, 0x8b, 0x3d, 0xbe,\n\t\t0xf2, 0x5b, 0xc9, 0x95, 0x66, 0xdb, 0x72, 0x28, 0x17, 0xec, 0xae, 0x1f, 0x14, 0x2d, 0x4a, 0x32,\n\t\t0x62, 0x55, 0xbc, 0xe8, 0x65, 0x17, 0x77, 0xf9, 0x55, 0xd7, 0x2e, 0x8a, 0x97, 0x78, 0xb8, 0x63,\n\t\t0x5b, 0xcd, 0xf8, 0xd5, 0x26, 0x6f, 0xc6, 0xdf, 0x37, 0x3c, 0x6c, 0x92, 0x95, 0x86, 0x13, 0xee,\n\t\t0xf6, 0x46, 0xce, 0x8a, 0x24, 0x4d, 0xe7, 0x73, 0x56, 0x25, 0xd9, 0xa5, 0xfe, 0x6c, 0x18, 0xce,\n\t\t0x6e, 0x07, 0x86, 0x17, 0x7c, 0xc2, 0xc7, 0xef, 0xbd, 0xc2, 0xcd, 0x2e, 0xa1, 0x47, 0xc3, 0x2f,\n\t\t0xbb, 0xd8, 0x0f, 0xd0, 0x43, 0x18, 0xf1, 0xd8, 0xcf, 0xaa, 0xb2, 0xa2, 0xac, 0x8d, 0x6f, 0x6c,\n\t\t0xd4, 0x13, 0x62, 0x6b, 0x74, 0xac, 0xfa, 0xc1, 0xe5, 0xba, 0x14, 0x89, 0x16, 0xa2, 0x40, 0x4b,\n\t\t0x30, 0x66, 0xba, 0x6d, 0xc3, 0x72, 0x74, 0xcb, 0xac, 0x96, 0x56, 0x94, 0xb5, 0x31, 0x6d, 0x94,\n\t\t0x0d, 0x34, 0x4c, 0xf4, 0x9b, 0x30, 0xd7, 0x31, 0x3c, 0xec, 0x04, 0x3a, 0x0e, 0x11, 0xe8, 0x96,\n\t\t0xd3, 0x72, 0xab, 0x65, 0xba, 0xf1, 0x9a, 0x70, 0xe3, 0xa7, 0x14, 0x22, 0xda, 0xb1, 0xe1, 0xb4,\n\t\t0x5c, 0xed, 0x74, 0x27, 0x3b, 0x88, 0xaa, 0x30, 0x62, 0x04, 0x01, 0x6e, 0x77, 0x82, 0xea, 0xa9,\n\t\t0x15, 0x65, 0x6d, 0x48, 0x0b, 0x1f, 0xd1, 0x26, 0x4c, 0xe1, 0x57, 0x1d, 0x8b, 0xa9, 0x98, 0x4e,\n\t\t0x74, 0xa9, 0x3a, 0x44, 0x77, 0xac, 0xd5, 0x99, 0x1e, 0xd5, 0x43, 0x3d, 0xaa, 0xef, 0x84, 0x8a,\n\t\t0xa6, 0x55, 0x7a, 0x20, 0x64, 0x10, 0xb5, 0x60, 0xb1, 0xe9, 0x3a, 0x81, 0xe5, 0x74, 0xb1, 0x6e,\n\t\t0xf8, 0xba, 0x83, 0x0f, 0x75, 0xcb, 0xb1, 0x02, 0xcb, 0x08, 0x5c, 0xaf, 0x3a, 0xbc, 0xa2, 0xac,\n\t\t0x55, 0x36, 0xde, 0x12, 0x1e, 0x60, 0x93, 0x43, 0xdd, 0xf1, 0x1f, 0xe3, 0xc3, 0x46, 0x08, 0xa2,\n\t\t0xcd, 0x37, 0x85, 0xe3, 0xa8, 0x01, 0x33, 0xe1, 0x8c, 0xa9, 0xb7, 0x0c, 0xcb, 0xee, 0x7a, 0xb8,\n\t\t0x3a, 0x42, 0xc9, 0x3d, 0x23, 0xc4, 0x7f, 0x9f, 0xad, 0xd1, 0xa6, 0x23, 0x30, 0x3e, 0x82, 0x34,\n\t\t0x98, 0xb7, 0x0d, 0x3f, 0xd0, 0x9b, 0x6e, 0xbb, 0x63, 0x63, 0x7a, 0x78, 0x0f, 0xfb, 0x5d, 0x3b,\n\t\t0xa8, 0x8e, 0x4a, 0xf0, 0x3d, 0x35, 0x8e, 0x6c, 0xd7, 0x30, 0xb5, 0x59, 0x02, 0xbb, 0x19, 0x81,\n\t\t0x6a, 0x14, 0x12, 0xfd, 0x2a, 0x2c, 0xb5, 0x2c, 0xcf, 0x0f, 0x74, 0x13, 0x37, 0x2d, 0x9f, 0xf2,\n\t\t0xd3, 0xf0, 0x5f, 0xe8, 0xbb, 0x46, 0xf3, 0x85, 0xdb, 0x6a, 0x55, 0xc7, 0x28, 0xe2, 0xc5, 0x0c,\n\t\t0x5f, 0xb7, 0xb8, 0x81, 0xd3, 0xaa, 0x14, 0x7a, 0x8b, 0x03, 0xef, 0x18, 0xfe, 0x8b, 0xbb, 0x0c,\n\t\t0x14, 0x1d, 0xc0, 0x74, 0xc7, 0xf0, 0x02, 0x8b, 0xd2, 0xd9, 0x74, 0x9d, 0x96, 0xb5, 0x57, 0x85,\n\t\t0x95, 0xf2, 0xda, 0xf8, 0xc6, 0xaf, 0xd4, 0x73, 0x0c, 0xa9, 0x5c, 0x2a, 0x89, 0xe8, 0x30, 0x74,\n\t\t0x9b, 0x14, 0xdb, 0x3d, 0x27, 0xf0, 0x8e, 0xb4, 0xa9, 0x4e, 0x72, 0xb4, 0x76, 0x17, 0x66, 0x45,\n\t\t0x0b, 0xd1, 0x34, 0x94, 0x5f, 0xe0, 0x23, 0xaa, 0x14, 0x63, 0x1a, 0xf9, 0x89, 0x66, 0x61, 0xe8,\n\t\t0xc0, 0xb0, 0xbb, 0x98, 0x0b, 0x36, 0x7b, 0xb8, 0x59, 0xba, 0xae, 0xa8, 0xd7, 0x60, 0x39, 0x8f,\n\t\t0x14, 0xbf, 0xe3, 0x3a, 0x3e, 0x46, 0x73, 0x30, 0xec, 0x75, 0xa9, 0x56, 0x30, 0x84, 0x43, 0x5e,\n\t\t0xd7, 0x69, 0x98, 0xea, 0xdf, 0x94, 0x60, 0x79, 0xdb, 0xda, 0x73, 0x0c, 0x3b, 0x57, 0x41, 0x1f,\n\t\t0xa5, 0x15, 0xf4, 0x8a, 0x58, 0x41, 0xa5, 0x58, 0xfa, 0xd4, 0xd0, 0x16, 0x2c, 0xe1, 0x57, 0x01,\n\t\t0xf6, 0x1c, 0xc3, 0x8e, 0x0c, 0x6f, 0x4f, 0x59, 0xb9, 0x9e, 0x5e, 0x10, 0xee, 0x9f, 0xdd, 0x79,\n\t\t0x31, 0x44, 0x95, 0x99, 0x42, 0x75, 0x38, 0xdd, 0xdc, 0xb7, 0x6c, 0xb3, 0xb7, 0x89, 0xeb, 0xd8,\n\t\t0x47, 0x54, 0x6f, 0x47, 0xb5, 0x19, 0x3a, 0x15, 0x02, 0x3d, 0x71, 0xec, 0x23, 0x75, 0x15, 0xce,\n\t\t0xe5, 0x9e, 0x8f, 0x31, 0x58, 0xfd, 0x79, 0x09, 0x2e, 0xf2, 0x35, 0x56, 0xb0, 0x2f, 0xb7, 0x79,\n\t\t0xcf, 0xd3, 0x2c, 0xbd, 0x25, 0x63, 0x69, 0x11, 0xba, 0x3e, 0x79, 0xfb, 0xb9, 0x22, 0x10, 0xf0,\n\t\t0x32, 0x15, 0xf0, 0x8f, 0xf3, 0x05, 0xbc, 0x3f, 0x12, 0xfe, 0x0f, 0x45, 0xfd, 0x0e, 0xac, 0x15,\n\t\t0x13, 0x25, 0x17, 0xfa, 0xef, 0x2b, 0x70, 0x56, 0xc3, 0x3e, 0x3e, 0xf1, 0x4b, 0x49, 0x8a, 0xa4,\n\t\t0xbf, 0x6b, 0x21, 0xaa, 0x9b, 0x87, 0x46, 0x7e, 0x8a, 0x2f, 0x4b, 0xb0, 0xba, 0x83, 0xbd, 0xb6,\n\t\t0xe5, 0x18, 0x01, 0xce, 0x3d, 0xc9, 0xd3, 0xf4, 0x49, 0xae, 0x0a, 0x4f, 0x52, 0x88, 0xe8, 0x97,\n\t\t0x5c, 0x81, 0xdf, 0x00, 0x55, 0x76, 0x44, 0xae, 0xc3, 0x3f, 0x54, 0x60, 0x65, 0x0b, 0xfb, 0x4d,\n\t\t0xcf, 0xda, 0xcd, 0xe7, 0xe8, 0x93, 0x34, 0x47, 0xdf, 0x13, 0x1e, 0xa7, 0x08, 0x4f, 0x9f, 0xe2,\n\t\t0xf1, 0xdf, 0x65, 0x58, 0x95, 0xa0, 0xe2, 0x22, 0x62, 0xc3, 0x42, 0xcf, 0xa5, 0x61, 0xaa, 0xcd,\n\t\t0x5f, 0x78, 0x52, 0x9b, 0x9d, 0x41, 0xb8, 0x19, 0x07, 0xd5, 0xe6, 0xb1, 0x70, 0x1c, 0xed, 0xc2,\n\t\t0x42, 0xf6, 0x6e, 0x99, 0x27, 0x55, 0xa2, 0xbb, 0x5d, 0xea, 0x6f, 0x37, 0xea, 0x4b, 0xcd, 0x1d,\n\t\t0x8a, 0x86, 0xd1, 0x27, 0x80, 0x3a, 0xd8, 0x31, 0x2d, 0x67, 0x4f, 0x37, 0x9a, 0x81, 0x75, 0x60,\n\t\t0x05, 0x16, 0xf6, 0xb9, 0xb9, 0xca, 0x71, 0xd4, 0xd8, 0xf2, 0x3b, 0x6c, 0xf5, 0x11, 0x45, 0x3e,\n\t\t0xd3, 0x49, 0x0c, 0x5a, 0xd8, 0x47, 0xbf, 0x06, 0xd3, 0x21, 0x62, 0x2a, 0x26, 0x1e, 0x76, 0xaa,\n\t\t0xa7, 0x28, 0xda, 0xba, 0x0c, 0xed, 0x26, 0x59, 0x9b, 0xa4, 0x7c, 0xaa, 0x13, 0x9b, 0xf2, 0xb0,\n\t\t0x83, 0xb6, 0x7b, 0xa8, 0x43, 0xef, 0x84, 0x3b, 0x7a, 0x52, 0x8a, 0x43, 0x67, 0x24, 0x81, 0x34,\n\t\t0x1c, 0x54, 0x5f, 0xc1, 0xec, 0x33, 0x12, 0xf3, 0x84, 0xdc, 0x0b, 0xc5, 0x70, 0x33, 0x2d, 0x86,\n\t\t0x6f, 0x0a, 0xf7, 0x10, 0xc1, 0xf6, 0x29, 0x7a, 0x3f, 0x56, 0x60, 0x2e, 0x05, 0xce, 0xc5, 0xed,\n\t\t0x36, 0x4c, 0xd0, 0x38, 0x2c, 0x74, 0xe7, 0x94, 0x3e, 0xdc, 0xb9, 0x71, 0x0a, 0xc1, 0xbd, 0xb8,\n\t\t0x06, 0x54, 0x42, 0x04, 0xbf, 0x8d, 0x9b, 0x01, 0x36, 0xb9, 0xe0, 0xa8, 0xf9, 0x67, 0xd0, 0xf8,\n\t\t0x4a, 0x6d, 0xf2, 0x65, 0xfc, 0x51, 0xfd, 0x7d, 0x05, 0x6a, 0xd4, 0x80, 0x6e, 0x07, 0x56, 0xf3,\n\t\t0xc5, 0x11, 0xf1, 0xe8, 0x1e, 0x5a, 0x7e, 0x10, 0xb2, 0xa9, 0x91, 0x66, 0xd3, 0x7a, 0xbe, 0x25,\n\t\t0x17, 0x62, 0xe8, 0x93, 0x59, 0x67, 0x61, 0x49, 0x88, 0x83, 0x5b, 0x96, 0x7f, 0x2b, 0xc1, 0xfc,\n\t\t0x03, 0x1c, 0x3c, 0xea, 0x06, 0xc6, 0xae, 0x8d, 0xb7, 0x03, 0x23, 0xc0, 0x9a, 0x08, 0xad, 0x92,\n\t\t0xb2, 0xa7, 0x1f, 0x03, 0x12, 0x98, 0xd1, 0xd2, 0x40, 0x66, 0x74, 0x26, 0xa3, 0x61, 0xe8, 0x0a,\n\t\t0xcc, 0xe3, 0x57, 0x1d, 0xca, 0x40, 0xdd, 0xc1, 0xaf, 0x02, 0x1d, 0x1f, 0x90, 0xb0, 0xc8, 0x32,\n\t\t0xa9, 0x85, 0x2e, 0x6b, 0xa7, 0xc3, 0xd9, 0xc7, 0xf8, 0x55, 0x70, 0x8f, 0xcc, 0x35, 0x4c, 0xf4,\n\t\t0x0e, 0xcc, 0x36, 0xbb, 0x1e, 0x8d, 0x9f, 0x76, 0x3d, 0xc3, 0x69, 0xee, 0xeb, 0x81, 0xfb, 0x82,\n\t\t0x6a, 0x8f, 0xb2, 0x36, 0xa1, 0x21, 0x3e, 0x77, 0x97, 0x4e, 0xed, 0x90, 0x19, 0xf4, 0x1b, 0x30,\n\t\t0x7b, 0x80, 0x3d, 0xea, 0xa5, 0x73, 0x9f, 0x42, 0xb7, 0x02, 0xdc, 0xe6, 0x4a, 0x91, 0x16, 0x58,\n\t\t0x12, 0xb4, 0x92, 0x13, 0x3c, 0x67, 0x20, 0x1f, 0x32, 0x88, 0x46, 0x80, 0xdb, 0x1a, 0x3a, 0xc8,\n\t\t0x8c, 0xa9, 0xff, 0x30, 0x06, 0x0b, 0x19, 0x96, 0x72, 0x01, 0x15, 0xb3, 0x4d, 0x39, 0x29, 0xdb,\n\t\t0xee, 0xc3, 0x64, 0x84, 0x36, 0x38, 0xea, 0x60, 0x7e, 0x11, 0xab, 0x52, 0x8c, 0x3b, 0x47, 0x1d,\n\t\t0xac, 0x4d, 0x1c, 0xc6, 0x9e, 0x90, 0x0a, 0x93, 0x22, 0xae, 0x8f, 0x3b, 0x31, 0x6e, 0x3f, 0x87,\n\t\t0xc5, 0x8e, 0x87, 0x0f, 0x2c, 0xb7, 0xeb, 0xeb, 0x3e, 0x71, 0x73, 0xb0, 0xd9, 0x5b, 0x7f, 0x8a,\n\t\t0xee, 0xbb, 0x94, 0x09, 0x73, 0x1a, 0x4e, 0x70, 0xf5, 0xdd, 0xe7, 0xc4, 0x57, 0xd2, 0xe6, 0x43,\n\t\t0xe8, 0x6d, 0x06, 0x1c, 0xe2, 0x7d, 0x1b, 0x4e, 0xd3, 0xa0, 0x8c, 0x45, 0x51, 0x11, 0xc6, 0x21,\n\t\t0x4a, 0xc1, 0x34, 0x99, 0xba, 0x4f, 0x66, 0xc2, 0xe5, 0x37, 0x61, 0x8c, 0x06, 0x58, 0xb6, 0xe5,\n\t\t0x07, 0x34, 0xcc, 0x1c, 0xdf, 0x38, 0x2b, 0xf6, 0x20, 0x42, 0x91, 0x1f, 0x0d, 0xf8, 0x2f, 0xf4,\n\t\t0x00, 0xa6, 0x7d, 0xaa, 0x0e, 0x7a, 0x0f, 0xc5, 0x48, 0x3f, 0x28, 0x2a, 0x7e, 0x42, 0x8b, 0xd0,\n\t\t0xbb, 0x30, 0xdf, 0xb4, 0x2d, 0x42, 0xa9, 0x6d, 0xed, 0x7a, 0x86, 0x77, 0xa4, 0x73, 0x79, 0xa0,\n\t\t0x81, 0xe4, 0x98, 0x36, 0xcb, 0x66, 0x1f, 0xb2, 0x49, 0x2e, 0x3f, 0x31, 0xa8, 0x16, 0x36, 0x82,\n\t\t0xae, 0x87, 0x23, 0xa8, 0xb1, 0x38, 0xd4, 0x7d, 0x36, 0x19, 0x42, 0x9d, 0x83, 0x71, 0x0e, 0x65,\n\t\t0xb5, 0x3b, 0x76, 0x15, 0xe8, 0x52, 0x60, 0x43, 0x8d, 0x76, 0xc7, 0x46, 0x3e, 0x5c, 0x4a, 0x9f,\n\t\t0x4a, 0xf7, 0x9b, 0xfb, 0xd8, 0xec, 0xda, 0x58, 0x0f, 0x5c, 0x76, 0x59, 0x34, 0xca, 0x77, 0xbb,\n\t\t0x41, 0x75, 0xbc, 0x28, 0x20, 0x7d, 0x23, 0x79, 0xd6, 0x6d, 0x8e, 0x69, 0xc7, 0xa5, 0xf7, 0xb6,\n\t\t0xc3, 0xd0, 0x10, 0x7f, 0x87, 0x5d, 0x15, 0x91, 0xff, 0xde, 0x41, 0x26, 0x68, 0xa2, 0x61, 0x86,\n\t\t0x4e, 0x6d, 0x93, 0x99, 0xf0, 0x14, 0x79, 0xba, 0x3a, 0x99, 0xab, 0xab, 0x0f, 0xa1, 0x12, 0xc9,\n\t\t0xb6, 0x4f, 0x94, 0xa9, 0x5a, 0xa1, 0x49, 0x85, 0xf3, 0xc9, 0xab, 0x62, 0x99, 0x9e, 0xb8, 0x7c,\n\t\t0x33, 0xcd, 0x8b, 0x14, 0x83, 0x3e, 0xa2, 0x26, 0xcc, 0x46, 0xd8, 0x9a, 0xb6, 0xeb, 0x63, 0x8e,\n\t\t0x73, 0x8a, 0xe2, 0xbc, 0xdc, 0xa7, 0x37, 0x42, 0x00, 0x09, 0xbe, 0xae, 0xaf, 0x45, 0xfa, 0x1c,\n\t\t0x0d, 0x12, 0x2d, 0x9f, 0x49, 0x9a, 0x17, 0xe2, 0x22, 0x4c, 0x8b, 0x5e, 0xb8, 0x3d, 0xaa, 0x13,\n\t\t0xc6, 0xc5, 0xc2, 0xbe, 0x36, 0x7d, 0x90, 0x1a, 0x41, 0xb7, 0x60, 0xc9, 0x22, 0x3a, 0x97, 0xba,\n\t\t0x63, 0xec, 0x10, 0x3b, 0x63, 0x56, 0x67, 0xa8, 0x8f, 0xb9, 0x60, 0xf9, 0x49, 0x53, 0x7f, 0x8f,\n\t\t0x4d, 0xa3, 0x55, 0x98, 0x08, 0x6d, 0x9d, 0x6f, 0x7d, 0x86, 0xab, 0x88, 0xa9, 0x36, 0x1f, 0xdb,\n\t\t0xb6, 0x3e, 0xc3, 0xea, 0x2f, 0x14, 0x58, 0x78, 0xea, 0xda, 0xf6, 0xff, 0xaf, 0xb7, 0x81, 0xfa,\n\t\t0x93, 0x51, 0xa8, 0x66, 0x8f, 0xfd, 0x8d, 0xc5, 0xfe, 0xc6, 0x62, 0x7f, 0x1d, 0x2d, 0x76, 0x9e,\n\t\t0x7e, 0x4c, 0xe4, 0x5a, 0x60, 0xa1, 0x39, 0x9b, 0x3c, 0xb1, 0x39, 0xfb, 0xe5, 0x33, 0xec, 0xea,\n\t\t0xbf, 0x94, 0x60, 0x45, 0xc3, 0x4d, 0xd7, 0x33, 0xe3, 0x89, 0x5a, 0xae, 0x16, 0xaf, 0xd3, 0x52,\n\t\t0x9e, 0x83, 0xf1, 0x48, 0x70, 0x22, 0x23, 0x00, 0xe1, 0x50, 0xc3, 0x44, 0x0b, 0x30, 0x42, 0x65,\n\t\t0x8c, 0x6b, 0x7c, 0x59, 0x1b, 0x26, 0x8f, 0x0d, 0x13, 0x9d, 0x05, 0xe0, 0x71, 0x44, 0xa8, 0xbb,\n\t\t0x63, 0xda, 0x18, 0x1f, 0x69, 0x98, 0x48, 0x83, 0x89, 0x8e, 0x6b, 0xdb, 0x7a, 0x18, 0xab, 0x0c,\n\t\t0x4b, 0x62, 0x15, 0x62, 0x43, 0xef, 0xbb, 0x5e, 0x9c, 0x35, 0x61, 0xac, 0x32, 0x4e, 0x90, 0xf0,\n\t\t0x07, 0xf5, 0xf7, 0x46, 0x61, 0x55, 0xc2, 0x45, 0x6e, 0x78, 0x33, 0x16, 0x52, 0x39, 0x9e, 0x85,\n\t\t0x94, 0x5a, 0xbf, 0xd2, 0xf1, 0xad, 0xdf, 0xb7, 0x01, 0x85, 0xfc, 0x35, 0xd3, 0xe6, 0x77, 0x3a,\n\t\t0x9a, 0x09, 0x57, 0xaf, 0x11, 0x03, 0x26, 0x30, 0xbd, 0x65, 0x62, 0xa1, 0x12, 0x78, 0x33, 0x16,\n\t\t0x7d, 0x28, 0x6b, 0xd1, 0x63, 0x25, 0x9d, 0xe1, 0x64, 0x49, 0xe7, 0x3a, 0x54, 0xb9, 0x49, 0xe9,\n\t\t0x25, 0x40, 0x42, 0x07, 0x61, 0x84, 0x3a, 0x08, 0xf3, 0x6c, 0x3e, 0x92, 0x9d, 0xd0, 0x3f, 0xd0,\n\t\t0x60, 0x32, 0x2a, 0x5d, 0xd0, 0x94, 0x09, 0xab, 0x85, 0xbc, 0x9d, 0xa7, 0x8d, 0x3b, 0x9e, 0xe1,\n\t\t0xf8, 0xc4, 0x94, 0x25, 0xd2, 0x04, 0x13, 0x66, 0xec, 0x09, 0x7d, 0x0a, 0x67, 0x04, 0x09, 0x99,\n\t\t0x9e, 0x09, 0x1f, 0xeb, 0xc7, 0x84, 0x2f, 0x66, 0xc4, 0x3d, 0xb2, 0xe6, 0x39, 0xde, 0x27, 0xe4,\n\t\t0x79, 0x9f, 0xab, 0x30, 0x91, 0xb0, 0x79, 0xe3, 0xd4, 0xe6, 0x8d, 0xef, 0xc6, 0x8c, 0xdd, 0x1d,\n\t\t0xa8, 0xf4, 0xae, 0x95, 0x96, 0xc4, 0x26, 0x0a, 0x4b, 0x62, 0x93, 0x11, 0x04, 0xad, 0x88, 0x7d,\n\t\t0x00, 0x13, 0xe1, 0x5d, 0x53, 0x04, 0x93, 0x85, 0x08, 0xc6, 0xf9, 0x7a, 0x0a, 0x6e, 0xc0, 0xc8,\n\t\t0xcb, 0x2e, 0xa6, 0x46, 0xb6, 0x42, 0xf3, 0x3f, 0x0f, 0x72, 0xb3, 0xe0, 0x85, 0x5a, 0x44, 0x53,\n\t\t0x14, 0x16, 0xf6, 0x59, 0xde, 0x3b, 0xc4, 0x9b, 0xf1, 0x05, 0xa7, 0x32, 0xbe, 0x60, 0xed, 0x53,\n\t\t0x98, 0x88, 0xc3, 0x0a, 0x52, 0xe1, 0xd7, 0xe3, 0xa9, 0xf0, 0xbc, 0x14, 0x49, 0xa8, 0x98, 0x2c,\n\t\t0x55, 0x12, 0x4b, 0x97, 0xf7, 0x4c, 0x69, 0x98, 0x18, 0xfb, 0xc6, 0x94, 0x66, 0x4c, 0x69, 0x9c,\n\t\t0x35, 0x42, 0x53, 0xfa, 0xb3, 0x72, 0x68, 0x4a, 0x85, 0x5c, 0xe4, 0xa6, 0xf4, 0x23, 0x98, 0x4a,\n\t\t0x99, 0x2a, 0xa9, 0x31, 0xe5, 0xc9, 0x0c, 0x6a, 0x6c, 0xb4, 0x4a, 0xd2, 0x94, 0x65, 0x84, 0xbb,\n\t\t0x34, 0x98, 0x70, 0xc7, 0x2c, 0x57, 0x39, 0x69, 0xb9, 0x3e, 0x85, 0xe5, 0xa4, 0xe2, 0xe9, 0x6e,\n\t\t0x4b, 0x0f, 0xf6, 0x2d, 0x5f, 0x8f, 0x57, 0xaf, 0xe5, 0x5b, 0xd5, 0x12, 0x8a, 0xf8, 0xa4, 0xb5,\n\t\t0xb3, 0x6f, 0xf9, 0x77, 0x38, 0xfe, 0x06, 0xcc, 0xec, 0x63, 0xc3, 0x0b, 0x76, 0xb1, 0x11, 0xe8,\n\t\t0x26, 0x0e, 0x0c, 0xcb, 0xf6, 0x79, 0xc2, 0x47, 0x9e, 0x20, 0x9c, 0x8e, 0xc0, 0xb6, 0x18, 0x54,\n\t\t0xf6, 0xd5, 0x34, 0x7c, 0xbc, 0x57, 0xd3, 0x45, 0x98, 0x8a, 0xf0, 0x30, 0xb1, 0xa6, 0x36, 0x7a,\n\t\t0x4c, 0x8b, 0x1c, 0xa3, 0x2d, 0x3a, 0xaa, 0xfe, 0xb9, 0x02, 0xdf, 0x62, 0xb7, 0x99, 0x50, 0x76,\n\t\t0x5e, 0x84, 0xee, 0xe9, 0x8b, 0x96, 0x4e, 0x2a, 0x5e, 0xcf, 0x4b, 0x2a, 0x16, 0xa1, 0xea, 0x33,\n\t\t0xbb, 0xf8, 0x77, 0x65, 0x78, 0x43, 0x8e, 0x8d, 0x8b, 0x20, 0xee, 0xbd, 0xff, 0x3c, 0x3e, 0xc6,\n\t\t0x49, 0xbc, 0x79, 0x7c, 0xeb, 0xa6, 0x4d, 0xf9, 0x29, 0x49, 0xff, 0xb1, 0x02, 0xcb, 0xbd, 0xb4,\n\t\t0x3c, 0xf1, 0xa1, 0x4d, 0xcb, 0xef, 0x18, 0x41, 0x73, 0x5f, 0xb7, 0xdd, 0xa6, 0x61, 0xdb, 0x47,\n\t\t0xd5, 0x12, 0xb5, 0xa9, 0x9f, 0x4a, 0x76, 0x2d, 0x3e, 0x4e, 0xbd, 0x97, 0xb7, 0xdf, 0x71, 0xb7,\n\t\t0xf8, 0x0e, 0x0f, 0xd9, 0x06, 0xcc, 0xd4, 0x2e, 0x19, 0xf9, 0x2b, 0x6a, 0xbf, 0x03, 0x2b, 0x45,\n\t\t0x08, 0x04, 0xf6, 0x76, 0x2b, 0x69, 0x6f, 0xc5, 0x55, 0x81, 0xd0, 0x0c, 0x50, 0x5c, 0x21, 0x62,\n\t\t0xfa, 0x66, 0x8e, 0xd9, 0xde, 0x1f, 0x2a, 0xc4, 0xf6, 0x66, 0x8e, 0x79, 0xdf, 0xb0, 0xec, 0x9e,\n\t\t0x2c, 0xf5, 0x59, 0x4e, 0x2a, 0xc2, 0xd3, 0xa7, 0x20, 0x7d, 0x8b, 0xd8, 0xb1, 0x5c, 0x4c, 0x3c,\n\t\t0x59, 0xfd, 0xa7, 0x0a, 0xa8, 0x59, 0x6b, 0xf7, 0x61, 0xa8, 0x9e, 0x21, 0xe5, 0xcf, 0xd2, 0x94,\n\t\t0x5f, 0xcb, 0xa1, 0xbc, 0x08, 0x53, 0x9f, 0xb4, 0x3f, 0x25, 0xca, 0x29, 0xc1, 0xc5, 0x65, 0xf3,\n\t\t0x4d, 0x98, 0x6e, 0x1a, 0x4e, 0x13, 0x47, 0x6f, 0x00, 0xcc, 0xde, 0x69, 0xa3, 0xda, 0x14, 0x1b,\n\t\t0xd7, 0xc2, 0xe1, 0xb8, 0xbe, 0xc7, 0x71, 0x9e, 0x50, 0xdf, 0x65, 0xa8, 0xfa, 0x3c, 0xea, 0x85,\n\t\t0x48, 0xdd, 0x73, 0x90, 0xc5, 0x0a, 0x96, 0x82, 0x85, 0x27, 0x91, 0xb0, 0x5c, 0x3c, 0x03, 0x4b,\n\t\t0x98, 0x08, 0x53, 0x42, 0xc2, 0xb2, 0x07, 0xa4, 0xf7, 0xd3, 0xa3, 0xbc, 0x6f, 0x09, 0x2b, 0xc2,\n\t\t0xd4, 0x27, 0xed, 0xe7, 0xc5, 0xe2, 0x10, 0xe1, 0xe2, 0xd4, 0xff, 0xbd, 0x02, 0xe7, 0x34, 0xdc,\n\t\t0x76, 0x0f, 0x30, 0xeb, 0x44, 0xf8, 0xaa, 0xe4, 0xf1, 0x92, 0x8e, 0x51, 0x39, 0xe5, 0x18, 0xa9,\n\t\t0x2a, 0x91, 0x95, 0x3c, 0xaa, 0xf9, 0xd1, 0xfe, 0xb1, 0x04, 0xe7, 0xf9, 0x11, 0xd8, 0xb1, 0x73,\n\t\t0xcb, 0xe0, 0xd2, 0x03, 0x1a, 0x50, 0x49, 0xea, 0x20, 0x3f, 0xdc, 0xcd, 0x9c, 0xfb, 0xeb, 0x63,\n\t\t0x43, 0x6d, 0x32, 0xa1, 0xbd, 0x68, 0x17, 0x16, 0xa2, 0x4e, 0x03, 0x61, 0x3b, 0x9f, 0xb8, 0x08,\n\t\t0x7d, 0x8f, 0xc3, 0xa4, 0x8a, 0xd0, 0x58, 0x34, 0x3c, 0x70, 0x97, 0xc1, 0x1a, 0x5c, 0x28, 0x3a,\n\t\t0x0b, 0xe7, 0xf3, 0x3f, 0x2b, 0xb0, 0x14, 0x26, 0x8e, 0x04, 0x81, 0xfc, 0x6b, 0x11, 0x9f, 0x4b,\n\t\t0x30, 0x63, 0xf9, 0x7a, 0xb2, 0xbb, 0x8e, 0xf2, 0x72, 0x54, 0x9b, 0xb2, 0xfc, 0xfb, 0xf1, 0xbe,\n\t\t0x39, 0x75, 0x19, 0xce, 0x88, 0xc9, 0xe7, 0xe7, 0xfb, 0x82, 0x3a, 0x2c, 0xc4, 0x58, 0x27, 0x0b,\n\t\t0xe7, 0x19, 0xd3, 0xfa, 0x3a, 0x0e, 0xba, 0x0a, 0x13, 0xbc, 0x75, 0x12, 0x9b, 0xb1, 0x5c, 0x6e,\n\t\t0x34, 0xd6, 0x30, 0xd1, 0x27, 0x70, 0xba, 0x19, 0x92, 0x1a, 0xdb, 0xfa, 0xd4, 0x40, 0x5b, 0xa3,\n\t\t0x08, 0x45, 0x6f, 0xef, 0x87, 0x30, 0x1d, 0x6b, 0x87, 0x64, 0x41, 0xc2, 0x50, 0xbf, 0x41, 0xc2,\n\t\t0x54, 0x0f, 0x94, 0x45, 0x09, 0x67, 0x01, 0x42, 0x77, 0xcf, 0x32, 0xa9, 0x7b, 0x5c, 0xd6, 0xc6,\n\t\t0xf8, 0x48, 0xc3, 0x54, 0x2f, 0x12, 0x65, 0x96, 0x5e, 0x02, 0xbf, 0xae, 0xff, 0x28, 0x41, 0x55,\n\t\t0xe3, 0xbd, 0xc2, 0x98, 0xa2, 0xf6, 0x9f, 0x6f, 0xbc, 0xce, 0x2b, 0xfa, 0x2d, 0x98, 0x13, 0x55,\n\t\t0x8e, 0xc3, 0x0e, 0x90, 0x01, 0x4a, 0xc7, 0xa7, 0xb3, 0xa5, 0x63, 0x1f, 0xbd, 0x07, 0xc3, 0x94,\n\t\t0xf5, 0x3e, 0xbf, 0x51, 0x71, 0x6a, 0x64, 0xcb, 0x08, 0x8c, 0xbb, 0xb6, 0xbb, 0xab, 0xf1, 0xc5,\n\t\t0x68, 0x13, 0x2a, 0x0e, 0x3e, 0xd4, 0xbd, 0x2e, 0xbf, 0xb9, 0x30, 0xb0, 0x29, 0x00, 0x9f, 0x70,\n\t\t0xf0, 0xa1, 0xd6, 0x65, 0x57, 0xe6, 0xab, 0x4b, 0xb0, 0x28, 0x60, 0x35, 0xbf, 0x88, 0xef, 0x2b,\n\t\t0x30, 0xbf, 0x7d, 0xe4, 0x34, 0xb7, 0xf7, 0x0d, 0xcf, 0xe4, 0x19, 0x52, 0x7e, 0x0d, 0xe7, 0xa1,\n\t\t0xe2, 0xbb, 0x5d, 0xaf, 0x89, 0x75, 0xde, 0x42, 0xce, 0xef, 0x62, 0x92, 0x8d, 0x6e, 0xb2, 0x41,\n\t\t0xb4, 0x08, 0xa3, 0x3e, 0x01, 0x0e, 0xdf, 0x6f, 0x43, 0xda, 0x08, 0x7d, 0x6e, 0x98, 0xa8, 0x0e,\n\t\t0xa7, 0x68, 0x2c, 0x59, 0x2e, 0x0c, 0xf0, 0xe8, 0x3a, 0x75, 0x11, 0x16, 0x32, 0xb4, 0x70, 0x3a,\n\t\t0xff, 0x75, 0x08, 0x4e, 0x93, 0xb9, 0xf0, 0x3d, 0xf9, 0x3a, 0x65, 0xa5, 0x0a, 0x23, 0x61, 0x46,\n\t\t0x8a, 0x69, 0x72, 0xf8, 0x48, 0x14, 0xbd, 0x17, 0xeb, 0x46, 0x79, 0x84, 0x28, 0xef, 0x40, 0x78,\n\t\t0x92, 0xcd, 0x43, 0x0d, 0x0d, 0x9a, 0x87, 0x92, 0x2b, 0x61, 0x26, 0x92, 0x1f, 0x19, 0x2c, 0x92,\n\t\t0xff, 0x88, 0x57, 0x7f, 0x7a, 0x41, 0x35, 0xc5, 0x32, 0x5a, 0x88, 0x65, 0x86, 0x80, 0x45, 0xee,\n\t\t0x31, 0xc5, 0x75, 0x15, 0x46, 0xc2, 0x88, 0x7c, 0xac, 0x8f, 0x88, 0x3c, 0x5c, 0x1c, 0xcf, 0x26,\n\t\t0x40, 0x32, 0x9b, 0x70, 0x1b, 0x26, 0x58, 0x6d, 0x8a, 0x37, 0x8a, 0x8f, 0xf7, 0xd1, 0x28, 0x3e,\n\t\t0x4e, 0x4b, 0x56, 0xbc, 0x47, 0xfc, 0x1d, 0xa0, 0x7d, 0xde, 0xfc, 0xd3, 0x09, 0xdd, 0x32, 0xb1,\n\t\t0x13, 0x58, 0xc1, 0x11, 0xcd, 0x06, 0x8e, 0x69, 0x88, 0xcc, 0x7d, 0x42, 0xa7, 0x1a, 0x7c, 0x06,\n\t\t0x3d, 0x86, 0xa9, 0x94, 0x69, 0xe0, 0x99, 0xbf, 0xf3, 0x7d, 0x19, 0x05, 0xad, 0x92, 0x34, 0x08,\n\t\t0xea, 0x3c, 0xcc, 0x26, 0x25, 0x99, 0x8b, 0xf8, 0x9f, 0x28, 0xb0, 0x14, 0x76, 0xde, 0x7d, 0x45,\n\t\t0x3c, 0x3c, 0xf5, 0x8f, 0x15, 0x38, 0x23, 0xa6, 0x89, 0x07, 0x3f, 0x57, 0x60, 0xbe, 0xcd, 0xc6,\n\t\t0x59, 0x5d, 0x46, 0xb7, 0x1c, 0xbd, 0x69, 0x34, 0xf7, 0x31, 0xa7, 0xf0, 0x74, 0x3b, 0x06, 0xd5,\n\t\t0x70, 0x36, 0xc9, 0x14, 0xba, 0x01, 0x8b, 0x19, 0x20, 0xd3, 0x08, 0x8c, 0x5d, 0xc3, 0x0f, 0x1b,\n\t\t0x70, 0xe7, 0x93, 0x70, 0x5b, 0x7c, 0x56, 0x3d, 0x03, 0xb5, 0x90, 0x1e, 0xce, 0xcf, 0x0f, 0xdd,\n\t\t0xa8, 0x75, 0x4a, 0xfd, 0xdd, 0x52, 0x8f, 0x85, 0x89, 0x69, 0x4e, 0xed, 0x1a, 0x4c, 0x3b, 0xdd,\n\t\t0xf6, 0x2e, 0xf6, 0x74, 0xb7, 0xa5, 0x53, 0x2b, 0xe5, 0x53, 0x3a, 0x87, 0xb4, 0x0a, 0x1b, 0x7f,\n\t\t0xd2, 0xa2, 0xc6, 0xc7, 0x27, 0xcc, 0x0e, 0xad, 0x9a, 0x4f, 0x53, 0x0b, 0x43, 0xda, 0x28, 0x37,\n\t\t0x6b, 0x3e, 0x6a, 0xc0, 0x04, 0xbf, 0x09, 0x76, 0x54, 0x71, 0x97, 0x69, 0x28, 0x0e, 0x2c, 0xd7,\n\t\t0x43, 0x4f, 0x4e, 0x7d, 0xbf, 0x71, 0xb3, 0x37, 0x80, 0xae, 0xc2, 0x02, 0xdb, 0xa7, 0xe9, 0x3a,\n\t\t0x81, 0xe7, 0xda, 0x36, 0xf6, 0x28, 0x4f, 0xba, 0xec, 0x4d, 0x31, 0xa6, 0xcd, 0xd1, 0xe9, 0xcd,\n\t\t0x68, 0x96, 0xd9, 0x45, 0xaa, 0x21, 0xa6, 0xe9, 0x61, 0xdf, 0xe7, 0x09, 0xc9, 0xf0, 0x51, 0xad,\n\t\t0xc3, 0x0c, 0xab, 0x6c, 0x11, 0xb8, 0x50, 0x76, 0xe2, 0x46, 0x5a, 0x49, 0x18, 0x69, 0x75, 0x16,\n\t\t0x50, 0x7c, 0x3d, 0x17, 0xc6, 0xff, 0x52, 0x60, 0x86, 0x39, 0xef, 0x71, 0x2f, 0x31, 0x1f, 0x0d,\n\t\t0xba, 0xc5, 0xab, 0xc0, 0x51, 0xd1, 0xbb, 0xb2, 0x71, 0x2e, 0x87, 0x21, 0x04, 0x23, 0xcd, 0x9a,\n\t\t0xd1, 0x3a, 0x30, 0xcd, 0x98, 0xc5, 0x72, 0xaf, 0xe5, 0x44, 0xee, 0x75, 0x13, 0xa6, 0x0e, 0x2c,\n\t\t0xdf, 0xda, 0xb5, 0x6c, 0x2b, 0x38, 0x62, 0x96, 0xa8, 0x38, 0x5d, 0x58, 0xe9, 0x81, 0x50, 0x33,\n\t\t0xb4, 0x0a, 0x13, 0xfc, 0x15, 0xa6, 0x3b, 0x06, 0xb7, 0xb8, 0x63, 0xda, 0x38, 0x1f, 0x7b, 0x6c,\n\t\t0xb4, 0x31, 0xe1, 0x42, 0xfc, 0xb8, 0x9c, 0x0b, 0x3f, 0xa0, 0x5c, 0xf0, 0x71, 0xf0, 0xac, 0x8b,\n\t\t0xbb, 0xb8, 0x0f, 0x2e, 0xa4, 0x77, 0x2a, 0x65, 0x76, 0x4a, 0x32, 0xaa, 0x3c, 0x20, 0xa3, 0x18,\n\t\t0x9d, 0x3d, 0x82, 0x38, 0x9d, 0x3f, 0x52, 0x60, 0x36, 0x94, 0xfb, 0xaf, 0x0c, 0xa9, 0x4f, 0x60,\n\t\t0x2e, 0x45, 0x13, 0xd7, 0xc2, 0xab, 0xb0, 0xd0, 0xf1, 0xdc, 0x26, 0xf6, 0x7d, 0xcb, 0xd9, 0xd3,\n\t\t0xe9, 0x57, 0x65, 0xcc, 0x0e, 0x10, 0x65, 0x2c, 0x13, 0x99, 0xef, 0x4d, 0x53, 0x48, 0x6a, 0x04,\n\t\t0x7c, 0xf5, 0x0b, 0x05, 0xce, 0x3e, 0xc0, 0x81, 0xd6, 0xfb, 0xc6, 0xec, 0x11, 0xf6, 0x7d, 0x63,\n\t\t0x0f, 0x47, 0x2e, 0xcb, 0x6d, 0x18, 0xa6, 0x05, 0x20, 0x86, 0x68, 0x7c, 0xe3, 0x62, 0x0e, 0xb5,\n\t\t0x31, 0x14, 0xb4, 0x3a, 0xa4, 0x71, 0xb0, 0x3e, 0x98, 0x42, 0x6c, 0xcc, 0x72, 0x1e, 0x15, 0xfc,\n\t\t0x80, 0x2f, 0xa1, 0xc2, 0xb8, 0xde, 0xe6, 0x33, 0x9c, 0x9c, 0x8f, 0x72, 0x93, 0x93, 0x72, 0x84,\n\t\t0x75, 0xaa, 0x9b, 0xe1, 0x28, 0x4b, 0x44, 0x4e, 0xfa, 0xf1, 0xb1, 0x9a, 0x0d, 0x28, 0xbb, 0x28,\n\t\t0x9e, 0x6c, 0x1c, 0x62, 0xc9, 0xc6, 0xef, 0x26, 0x93, 0x8d, 0x97, 0x8a, 0x19, 0x14, 0x11, 0x13,\n\t\t0x4b, 0x34, 0xb6, 0x61, 0xe5, 0x01, 0x0e, 0xb6, 0x1e, 0x3e, 0x93, 0xdc, 0x45, 0x03, 0x80, 0xa9,\n\t\t0xb4, 0xd3, 0x72, 0x43, 0x06, 0xf4, 0xb1, 0x1d, 0x11, 0x24, 0x6a, 0x26, 0xa9, 0xe8, 0x91, 0x5f,\n\t\t0xbe, 0xfa, 0x0a, 0x56, 0x25, 0xdb, 0x71, 0xa6, 0x6f, 0xc3, 0x4c, 0xec, 0xeb, 0x43, 0x5a, 0x8c,\n\t\t0x0c, 0xb7, 0xbd, 0xd0, 0xdf, 0xb6, 0xda, 0xb4, 0x97, 0x1c, 0xf0, 0xd5, 0x7f, 0x57, 0x60, 0x56,\n\t\t0xc3, 0x46, 0xa7, 0x63, 0xb3, 0x88, 0x28, 0x3a, 0xdd, 0x3c, 0x0c, 0xf3, 0xcc, 0x3e, 0x7b, 0xcf,\n\t\t0xf1, 0x27, 0xf9, 0xc7, 0x0a, 0xe2, 0x97, 0x74, 0xf9, 0xa4, 0xfe, 0xe8, 0xf1, 0x82, 0x0b, 0x75,\n\t\t0x01, 0xe6, 0x52, 0x47, 0xe3, 0xd6, 0xe4, 0xa7, 0x0a, 0x2c, 0x69, 0xb8, 0xe5, 0x61, 0x7f, 0x3f,\n\t\t0x2a, 0x72, 0x10, 0x6e, 0x7c, 0x05, 0xcf, 0xae, 0x2e, 0xc3, 0x19, 0x31, 0xa9, 0xfc, 0x2c, 0x37,\n\t\t0x60, 0x61, 0xd3, 0xed, 0x3a, 0x44, 0x78, 0xd2, 0x02, 0xba, 0x0c, 0xd0, 0x72, 0xbd, 0x26, 0xbe,\n\t\t0x8f, 0x83, 0xe6, 0x3e, 0xcf, 0xd8, 0xc6, 0x46, 0x54, 0x03, 0xaa, 0x59, 0x50, 0x2e, 0x6c, 0xf7,\n\t\t0x60, 0x04, 0x3b, 0x01, 0xad, 0xe5, 0x32, 0x11, 0x7b, 0x2b, 0x47, 0xc4, 0xb8, 0x17, 0xb2, 0xf5,\n\t\t0xf0, 0x19, 0xc5, 0xc5, 0xeb, 0xb5, 0x1c, 0x56, 0xfd, 0x69, 0x09, 0xe6, 0x35, 0x6c, 0x98, 0x02,\n\t\t0xea, 0x36, 0xe0, 0x54, 0xd4, 0x1d, 0x51, 0xd9, 0x58, 0xce, 0xf3, 0x2d, 0x1e, 0x3e, 0xa3, 0x56,\n\t\t0x97, 0xae, 0x95, 0x85, 0x62, 0xd9, 0x60, 0xae, 0x2c, 0x0a, 0xe6, 0x76, 0xa0, 0x6a, 0x39, 0x64,\n\t\t0x85, 0x75, 0x80, 0x75, 0xec, 0x44, 0x16, 0xac, 0xcf, 0x8e, 0xb2, 0xb9, 0x08, 0xf8, 0x9e, 0x13,\n\t\t0x9a, 0xa2, 0x86, 0x49, 0x04, 0xa3, 0x43, 0x90, 0xd0, 0x9a, 0xf4, 0x10, 0x25, 0x6c, 0x94, 0x0c,\n\t\t0x6c, 0x5b, 0x9f, 0x61, 0x74, 0x01, 0xa6, 0x68, 0x5f, 0x04, 0x5d, 0xc1, 0xca, 0xf7, 0xc3, 0xb4,\n\t\t0x7c, 0x4f, 0xdb, 0x25, 0x9e, 0x1a, 0x7b, 0x98, 0x75, 0xf3, 0xfd, 0x6d, 0x09, 0x16, 0x32, 0xbc,\n\t\t0xe2, 0xd7, 0x71, 0x1c, 0x66, 0x09, 0xed, 0x45, 0xe9, 0x64, 0xf6, 0x02, 0x7d, 0x0f, 0xe6, 0x33,\n\t\t0x48, 0xc3, 0x1c, 0xe1, 0xa0, 0x06, 0x70, 0x36, 0x8d, 0x9d, 0xa6, 0x08, 0x05, 0xec, 0x3a, 0x25,\n\t\t0x62, 0xd7, 0xcf, 0x15, 0x58, 0x78, 0xda, 0xf5, 0xf6, 0xf0, 0xd7, 0x5b, 0xb6, 0xd4, 0x1a, 0x54,\n\t\t0xb3, 0xc7, 0xe4, 0xca, 0xff, 0x65, 0x09, 0x16, 0x1e, 0xe1, 0xaf, 0x3d, 0x0f, 0xfe, 0x77, 0xf4,\n\t\t0xeb, 0x2e, 0x54, 0xb3, 0xbc, 0xe2, 0xfa, 0x25, 0xc0, 0xa1, 0x88, 0x70, 0x7c, 0xae, 0xc0, 0x99,\n\t\t0xc7, 0x6e, 0x60, 0xb5, 0x8e, 0x48, 0xb8, 0xed, 0x1e, 0x60, 0xef, 0x91, 0x41, 0x62, 0xe9, 0x88,\n\t\t0xeb, 0xdf, 0x83, 0xf9, 0x16, 0x9f, 0xd1, 0xdb, 0x74, 0x4a, 0x4f, 0x38, 0x6c, 0x79, 0xfa, 0x91,\n\t\t0x44, 0xc7, 0x7c, 0xb6, 0xd9, 0x56, 0x76, 0xd0, 0x57, 0xcf, 0xc1, 0xd9, 0x1c, 0x0a, 0xb8, 0x50,\n\t\t0x18, 0xb0, 0xf4, 0x00, 0x07, 0x9b, 0x9e, 0xeb, 0xfb, 0xfc, 0x56, 0x12, 0x2f, 0xb7, 0x44, 0xe0,\n\t\t0xa7, 0xa4, 0x02, 0xbf, 0xf3, 0x50, 0x09, 0x0c, 0x6f, 0x0f, 0x07, 0xd1, 0x2d, 0xb3, 0xd7, 0xdc,\n\t\t0x24, 0x1b, 0xe5, 0xf8, 0xd4, 0x5f, 0x94, 0xe1, 0x8c, 0x78, 0x0f, 0xce, 0xcf, 0x36, 0xc1, 0x43,\n\t\t0x4c, 0xc3, 0xee, 0x11, 0x0b, 0x43, 0xf9, 0xf1, 0x1f, 0xc8, 0x1c, 0xc4, 0x5c, 0x74, 0xd4, 0xf9,\n\t\t0xf6, 0xef, 0x1e, 0x51, 0x07, 0x90, 0xbd, 0x61, 0x26, 0x82, 0xd8, 0x10, 0xfa, 0x5c, 0x81, 0xb9,\n\t\t0x16, 0x2d, 0x88, 0xe9, 0x4d, 0xa3, 0xeb, 0xe3, 0xde, 0xb6, 0xcc, 0xde, 0x3d, 0x3a, 0xde, 0xb6,\n\t\t0xac, 0xc6, 0xb6, 0x49, 0x30, 0x26, 0x36, 0x47, 0xad, 0xcc, 0x44, 0xad, 0x03, 0x33, 0x19, 0x2a,\n\t\t0x05, 0xee, 0xe9, 0xbd, 0xa4, 0x7b, 0xba, 0x9e, 0x23, 0x0e, 0x69, 0x9a, 0xf8, 0xe5, 0xc5, 0x7d,\n\t\t0xd4, 0x5a, 0x07, 0x16, 0x72, 0x08, 0x14, 0xec, 0x7b, 0x3b, 0xbe, 0x6f, 0x25, 0x37, 0xdd, 0xfb,\n\t\t0x00, 0x07, 0xbd, 0xe2, 0x22, 0xc5, 0x1b, 0xf7, 0x8a, 0xff, 0x53, 0x81, 0x35, 0x5e, 0xce, 0xcb,\n\t\t0x30, 0x2d, 0x53, 0x87, 0x90, 0x44, 0x66, 0xfd, 0x49, 0x19, 0x7a, 0xce, 0x84, 0x28, 0xea, 0xbb,\n\t\t0x08, 0x73, 0xd5, 0xfd, 0x33, 0x8d, 0x77, 0x5b, 0x4c, 0x06, 0xb1, 0x27, 0x1f, 0xbd, 0x01, 0x93,\n\t\t0x2d, 0xe2, 0x00, 0x3d, 0xc6, 0xcc, 0x97, 0xe2, 0xe5, 0xa7, 0xe4, 0xa0, 0xea, 0xc1, 0x9b, 0x7d,\n\t\t0x9c, 0x35, 0x72, 0x97, 0x86, 0x42, 0x7f, 0xfc, 0x78, 0xd7, 0x4a, 0xa1, 0xd5, 0xf7, 0xe8, 0x37,\n\t\t0x6d, 0xa1, 0x62, 0xd3, 0x97, 0x64, 0x1f, 0xb9, 0x31, 0x35, 0xa0, 0xdf, 0x6d, 0x25, 0xc1, 0x22,\n\t\t0xc7, 0x61, 0xae, 0x57, 0x76, 0x09, 0x13, 0x31, 0x5d, 0xde, 0x47, 0x35, 0xa4, 0xf5, 0x6a, 0x32,\n\t\t0xdb, 0x2c, 0x0b, 0xd3, 0x75, 0x68, 0x5e, 0x3c, 0xfc, 0xea, 0x92, 0xa7, 0x90, 0x58, 0x7e, 0x68,\n\t\t0x92, 0x8f, 0xb2, 0x0c, 0x92, 0xda, 0x80, 0x79, 0xcd, 0x08, 0xb0, 0x6d, 0xb5, 0xad, 0xe0, 0xe3,\n\t\t0x8e, 0x19, 0x4b, 0xe4, 0xad, 0xc3, 0x29, 0xd3, 0x08, 0x0c, 0xce, 0x8c, 0xa5, 0xbc, 0x46, 0xcc,\n\t\t0x3b, 0xce, 0x91, 0x46, 0x17, 0xaa, 0x1f, 0xc1, 0x42, 0x06, 0x15, 0x3f, 0xc0, 0xa0, 0xb8, 0x36,\n\t\t0xfe, 0xa9, 0x0e, 0xc0, 0x9d, 0xd2, 0x3b, 0x4f, 0x1b, 0xe8, 0x0f, 0x15, 0x98, 0x17, 0x7f, 0xd4,\n\t\t0x8e, 0xae, 0x1e, 0xef, 0x5f, 0x28, 0x6a, 0xd7, 0x06, 0x86, 0xe3, 0x67, 0xf9, 0x23, 0x05, 0x16,\n\t\t0x72, 0xfe, 0xf5, 0x00, 0x5d, 0x2b, 0xfa, 0xc7, 0x80, 0x3c, 0x6a, 0xae, 0x0f, 0x0e, 0xc8, 0xc9,\n\t\t0xf9, 0x89, 0x02, 0x2b, 0x45, 0x5f, 0xfe, 0xa3, 0xef, 0x9e, 0xf4, 0x9f, 0x0c, 0x6a, 0x77, 0x4e,\n\t\t0x80, 0x81, 0x53, 0x4a, 0x2e, 0x51, 0xfc, 0x4d, 0xbf, 0xe4, 0x12, 0xa5, 0xff, 0x25, 0x20, 0xb9,\n\t\t0xc4, 0x82, 0x3f, 0x0f, 0xf8, 0x33, 0x05, 0x6a, 0xf9, 0x5f, 0xbe, 0xa3, 0xfc, 0xae, 0xb0, 0xc2,\n\t\t0x7f, 0x04, 0xa8, 0xbd, 0x7f, 0x2c, 0x58, 0x4e, 0xd7, 0x8f, 0x14, 0x58, 0xcc, 0xfd, 0xae, 0x1d,\n\t\t0xdd, 0xc8, 0x45, 0x5d, 0xf4, 0x59, 0x7d, 0xed, 0xe6, 0x71, 0x40, 0x39, 0x51, 0x0e, 0x4c, 0x26,\n\t\t0x3e, 0x78, 0x46, 0x6f, 0xe7, 0x22, 0x13, 0x7d, 0x57, 0x5d, 0xab, 0xf7, 0xbb, 0x9c, 0xef, 0xf7,\n\t\t0xb9, 0x02, 0xa7, 0x05, 0x5f, 0x0d, 0xa3, 0x2b, 0xf2, 0xdb, 0x16, 0x7e, 0xa7, 0x5c, 0x7b, 0x77,\n\t\t0x30, 0x20, 0x4e, 0x42, 0x00, 0x53, 0xa9, 0x8f, 0x68, 0xd1, 0xba, 0xcc, 0xfd, 0x10, 0x54, 0x42,\n\t\t0x6a, 0xef, 0xf4, 0x0f, 0xc0, 0x77, 0x3d, 0x84, 0xe9, 0xf4, 0x97, 0x60, 0x28, 0x1f, 0x4b, 0xce,\n\t\t0xb7, 0x72, 0xb5, 0xcb, 0x03, 0x40, 0xc4, 0xc4, 0x2e, 0xb7, 0xdf, 0x51, 0x22, 0x76, 0x45, 0x5f,\n\t\t0xa3, 0xd4, 0x4e, 0xd0, 0x5e, 0x89, 0xfe, 0x52, 0x81, 0x33, 0xb2, 0x76, 0x48, 0x74, 0xeb, 0x98,\n\t\t0x5d, 0x94, 0x8c, 0xb4, 0x0f, 0x4e, 0xd4, 0x83, 0xc9, 0x59, 0x96, 0xd3, 0x33, 0x28, 0x65, 0x99,\n\t\t0xbc, 0x63, 0x51, 0xca, 0xb2, 0x82, 0x16, 0xc5, 0xd8, 0x3d, 0x0a, 0x1a, 0xb2, 0x0b, 0xef, 0x31,\n\t\t0xbf, 0x15, 0xbe, 0xf0, 0x1e, 0x65, 0xfd, 0xdf, 0xb1, 0x7b, 0x14, 0xb6, 0xed, 0x15, 0xdf, 0xa3,\n\t\t0xac, 0x75, 0xb0, 0xf8, 0x1e, 0xa5, 0xbd, 0x82, 0xf1, 0x7b, 0xcc, 0x76, 0xe6, 0x15, 0xdf, 0x63,\n\t\t0x6e, 0x5f, 0x60, 0xf1, 0x3d, 0xe6, 0x37, 0x02, 0xa2, 0xbf, 0xa0, 0xb9, 0xcd, 0xdc, 0x96, 0x3b,\n\t\t0xf4, 0xfe, 0x40, 0x67, 0x4e, 0x36, 0xfd, 0xd5, 0x6e, 0x1d, 0x0f, 0x38, 0x41, 0x5a, 0x6e, 0xbf,\n\t\t0xa9, 0x94, 0xb4, 0xa2, 0x8e, 0x57, 0x29, 0x69, 0xc5, 0x2d, 0xae, 0x7f, 0xad, 0xc0, 0xb2, 0xbc,\n\t\t0xd1, 0x0c, 0x7d, 0x47, 0xb2, 0x41, 0x1f, 0xdd, 0x76, 0xb5, 0xdb, 0xc7, 0x86, 0xe7, 0x34, 0xfe,\n\t\t0x40, 0x81, 0x6a, 0x5e, 0xbb, 0x21, 0xba, 0x2e, 0xc1, 0x2e, 0xed, 0xab, 0xac, 0xdd, 0x38, 0x06,\n\t\t0x24, 0xa7, 0xe8, 0x0b, 0x05, 0x66, 0x45, 0x4d, 0x6b, 0x28, 0xff, 0xcd, 0x29, 0x69, 0xd1, 0xab,\n\t\t0xbd, 0x37, 0x20, 0x14, 0xa7, 0xe2, 0xaf, 0xe8, 0x9f, 0x4f, 0x49, 0x9a, 0xb2, 0xd0, 0x07, 0x05,\n\t\t0xb2, 0x21, 0xef, 0xa8, 0xab, 0x7d, 0xe7, 0xb8, 0xe0, 0x9c, 0xc0, 0xcf, 0x60, 0x26, 0xd3, 0x9f,\n\t\t0x84, 0x2e, 0x4b, 0x90, 0x8a, 0xdb, 0xc6, 0x6a, 0x1b, 0x83, 0x80, 0xf4, 0xbc, 0x91, 0x54, 0xc7,\n\t\t0x91, 0xc4, 0x1b, 0x11, 0xf7, 0x49, 0x49, 0xbc, 0x91, 0x9c, 0x66, 0x26, 0xf4, 0x02, 0x26, 0xe2,\n\t\t0x1d, 0x20, 0xe8, 0xdb, 0x52, 0x0c, 0xa9, 0x96, 0xa7, 0xda, 0xdb, 0x7d, 0xae, 0x8e, 0x49, 0xa1,\n\t\t0xa8, 0x85, 0x43, 0x22, 0x85, 0x92, 0x2e, 0x14, 0x89, 0x14, 0x4a, 0xfb, 0x44, 0x88, 0xe7, 0x29,\n\t\t0xe8, 0xcc, 0x90, 0x78, 0x9e, 0xf9, 0x6d, 0x1e, 0xb5, 0x77, 0x07, 0x03, 0x8a, 0x3e, 0x55, 0x81,\n\t\t0x5e, 0xa3, 0x03, 0xba, 0x94, 0x8b, 0x23, 0xd3, 0x3d, 0x51, 0x7b, 0xab, 0xaf, 0xb5, 0xbd, 0x6d,\n\t\t0x7a, 0x9d, 0x04, 0x92, 0x6d, 0x32, 0xdd, 0x15, 0x92, 0x6d, 0xb2, 0xad, 0x09, 0x6c, 0x9b, 0xb0,\n\t\t0x11, 0x40, 0xba, 0x4d, 0xaa, 0x7d, 0x41, 0xba, 0x4d, 0xba, 0xb3, 0x80, 0x44, 0x28, 0x89, 0x22,\n\t\t0xbe, 0x24, 0x42, 0x11, 0x35, 0x20, 0x48, 0x22, 0x14, 0x71, 0x6f, 0x00, 0x09, 0x65, 0xc5, 0xc5,\n\t\t0x70, 0x49, 0x28, 0x2b, 0x6d, 0x0a, 0x90, 0x84, 0xb2, 0x05, 0x65, 0x7c, 0xe2, 0xc0, 0xe4, 0xd6,\n\t\t0x9d, 0x25, 0x0e, 0x4c, 0x51, 0x69, 0x5c, 0xe2, 0xc0, 0x14, 0x97, 0xb9, 0x1d, 0x98, 0x4c, 0x54,\n\t\t0x6d, 0x25, 0x17, 0x22, 0x2a, 0x5c, 0x4b, 0x2e, 0x44, 0x58, 0x0c, 0xa6, 0xe6, 0x43, 0x54, 0x61,\n\t\t0x45, 0xb2, 0xf0, 0x2f, 0xb7, 0x76, 0x2c, 0x31, 0x1f, 0xb2, 0x32, 0x2e, 0x89, 0xdf, 0xd2, 0xb5,\n\t\t0x58, 0x49, 0xfc, 0x96, 0x53, 0xf1, 0x95, 0xc4, 0x6f, 0xb9, 0x85, 0xde, 0x00, 0xa6, 0x52, 0x45,\n\t\t0x47, 0xc9, 0x0b, 0x42, 0x5c, 0xca, 0x95, 0xbc, 0x20, 0xf2, 0xea, 0x99, 0x24, 0x5c, 0x4d, 0x15,\n\t\t0xb5, 0x64, 0xe1, 0xaa, 0xb8, 0xcc, 0x27, 0x0b, 0x57, 0x73, 0x2a, 0x66, 0x64, 0xe3, 0x74, 0x11,\n\t\t0x48, 0xb2, 0x71, 0x4e, 0x6d, 0x4d, 0xb2, 0x71, 0x6e, 0x85, 0xe9, 0x0f, 0x14, 0x98, 0x13, 0xd6,\n\t\t0x6d, 0x50, 0xbe, 0xc4, 0xc8, 0x2a, 0x4d, 0xb5, 0xab, 0x83, 0x82, 0xc5, 0xe4, 0x5d, 0x54, 0xf5,\n\t\t0x90, 0xc8, 0xbb, 0xa4, 0x9c, 0x24, 0x91, 0x77, 0x69, 0x81, 0xe8, 0x4b, 0x25, 0xfa, 0xaa, 0x29,\n\t\t0x3f, 0xbd, 0x8e, 0xee, 0x14, 0xc5, 0x1b, 0x85, 0x65, 0x88, 0xda, 0xdd, 0x93, 0xa0, 0x48, 0xa4,\n\t\t0x74, 0xe2, 0xf9, 0x75, 0x79, 0x4a, 0x47, 0x90, 0xc0, 0x97, 0xa7, 0x74, 0x84, 0xa9, 0x7b, 0xa2,\n\t\t0x99, 0xc9, 0xa4, 0xb8, 0x4c, 0x33, 0x85, 0x99, 0x78, 0x99, 0x66, 0x8a, 0xf3, 0xed, 0x77, 0x6f,\n\t\t0xfc, 0xfa, 0xb5, 0x3d, 0x2b, 0xd8, 0xef, 0xee, 0xd6, 0x9b, 0x6e, 0x7b, 0x3d, 0xf1, 0x9f, 0xe4,\n\t\t0xf5, 0x3d, 0xec, 0xb0, 0x3f, 0xa8, 0x8f, 0xfd, 0x43, 0xfe, 0xfb, 0xfc, 0xe7, 0xc1, 0xe5, 0xdd,\n\t\t0x61, 0x3a, 0x77, 0xe5, 0x7f, 0x02, 0x00, 0x00, 0xff, 0xff, 0x55, 0xea, 0x8b, 0xf1, 0x4d, 0x5f,\n\t\t0x00, 0x00,\n\t},\n\t// google/protobuf/duration.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x4f, 0x29, 0x2d, 0x4a,\n\t\t0x2c, 0xc9, 0xcc, 0xcf, 0xd3, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0x56,\n\t\t0x5c, 0x1c, 0x2e, 0x50, 0x25, 0x42, 0x12, 0x5c, 0xec, 0xc5, 0xa9, 0xc9, 0xf9, 0x79, 0x29, 0xc5,\n\t\t0x12, 0x8c, 0x0a, 0x8c, 0x1a, 0xcc, 0x41, 0x30, 0xae, 0x90, 0x08, 0x17, 0x6b, 0x5e, 0x62, 0x5e,\n\t\t0x7e, 0xb1, 0x04, 0x93, 0x02, 0xa3, 0x06, 0x6b, 0x10, 0x84, 0xe3, 0xd4, 0xcc, 0xc8, 0x25, 0x9c,\n\t\t0x9c, 0x9f, 0xab, 0x87, 0x66, 0xa6, 0x13, 0x2f, 0xcc, 0xc4, 0x00, 0x90, 0x48, 0x00, 0x63, 0x94,\n\t\t0x21, 0x54, 0x45, 0x7a, 0x7e, 0x4e, 0x62, 0x5e, 0xba, 0x5e, 0x7e, 0x51, 0x3a, 0xc2, 0x81, 0x25,\n\t\t0x95, 0x05, 0xa9, 0xc5, 0xfa, 0xd9, 0x79, 0xf9, 0xe5, 0x79, 0x70, 0xc7, 0x16, 0x24, 0xfd, 0x60,\n\t\t0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce, 0x1d, 0xa2, 0x39, 0x00, 0xaa,\n\t\t0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4, 0x35, 0x89, 0x0d, 0x6c, 0x94,\n\t\t0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0xef, 0x8a, 0xb4, 0xc3, 0xfb, 0x00, 0x00, 0x00,\n\t},\n\t// google/protobuf/timestamp.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4f, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0xc9, 0xcc, 0x4d,\n\t\t0x2d, 0x2e, 0x49, 0xcc, 0x2d, 0xd0, 0x03, 0x0b, 0x09, 0xf1, 0x43, 0x14, 0xe8, 0xc1, 0x14, 0x28,\n\t\t0x59, 0x73, 0x71, 0x86, 0xc0, 0xd4, 0x08, 0x49, 0x70, 0xb1, 0x17, 0xa7, 0x26, 0xe7, 0xe7, 0xa5,\n\t\t0x14, 0x4b, 0x30, 0x2a, 0x30, 0x6a, 0x30, 0x07, 0xc1, 0xb8, 0x42, 0x22, 0x5c, 0xac, 0x79, 0x89,\n\t\t0x79, 0xf9, 0xc5, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0xac, 0x41, 0x10, 0x8e, 0x53, 0x2b, 0x23, 0x97,\n\t\t0x70, 0x72, 0x7e, 0xae, 0x1e, 0x9a, 0xa1, 0x4e, 0x7c, 0x70, 0x23, 0x03, 0x40, 0x42, 0x01, 0x8c,\n\t\t0x51, 0x46, 0x50, 0x25, 0xe9, 0xf9, 0x39, 0x89, 0x79, 0xe9, 0x7a, 0xf9, 0x45, 0xe9, 0x48, 0x6e,\n\t\t0xac, 0x2c, 0x48, 0x2d, 0xd6, 0xcf, 0xce, 0xcb, 0x2f, 0xcf, 0x43, 0xb8, 0xb7, 0x20, 0xe9, 0x07,\n\t\t0x23, 0xe3, 0x22, 0x26, 0x66, 0xf7, 0x00, 0xa7, 0x55, 0x4c, 0x72, 0xee, 0x10, 0xdd, 0x01, 0x50,\n\t\t0x2d, 0x7a, 0xe1, 0xa9, 0x39, 0x39, 0xde, 0x20, 0x0d, 0x21, 0x20, 0xbd, 0x49, 0x6c, 0x60, 0xb3,\n\t\t0x8c, 0x01, 0x01, 0x00, 0x00, 0xff, 0xff, 0xae, 0x65, 0xce, 0x7d, 0xff, 0x00, 0x00, 0x00,\n\t},\n\t// google/protobuf/wrappers.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0x2f, 0x4a, 0x2c,\n\t\t0x28, 0x48, 0x2d, 0x2a, 0xd6, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0xca,\n\t\t0x5c, 0xdc, 0x2e, 0xf9, 0xa5, 0x49, 0x39, 0xa9, 0x61, 0x89, 0x39, 0xa5, 0xa9, 0x42, 0x22, 0x5c,\n\t\t0xac, 0x65, 0x20, 0x86, 0x04, 0xa3, 0x02, 0xa3, 0x06, 0x63, 0x10, 0x84, 0xa3, 0xa4, 0xc4, 0xc5,\n\t\t0xe5, 0x96, 0x93, 0x9f, 0x58, 0x82, 0x45, 0x0d, 0x13, 0x92, 0x1a, 0xcf, 0xbc, 0x12, 0x33, 0x13,\n\t\t0x2c, 0x6a, 0x98, 0x61, 0x6a, 0x94, 0xb9, 0xb8, 0x43, 0x71, 0x29, 0x62, 0x41, 0x35, 0xc8, 0xd8,\n\t\t0x08, 0x8b, 0x1a, 0x56, 0x34, 0x83, 0xb0, 0x2a, 0xe2, 0x85, 0x29, 0x52, 0xe4, 0xe2, 0x74, 0xca,\n\t\t0xcf, 0xcf, 0xc1, 0xa2, 0x84, 0x03, 0xc9, 0x9c, 0xe0, 0x92, 0xa2, 0xcc, 0xbc, 0x74, 0x2c, 0x8a,\n\t\t0x38, 0x91, 0x1c, 0xe4, 0x54, 0x59, 0x92, 0x5a, 0x8c, 0x45, 0x0d, 0x0f, 0x54, 0x8d, 0x53, 0x33,\n\t\t0x23, 0x97, 0x70, 0x72, 0x7e, 0xae, 0x1e, 0x5a, 0xf0, 0x3a, 0xf1, 0x86, 0x43, 0xc3, 0x3f, 0x00,\n\t\t0x24, 0x12, 0xc0, 0x18, 0x65, 0x08, 0x55, 0x91, 0x9e, 0x9f, 0x93, 0x98, 0x97, 0xae, 0x97, 0x5f,\n\t\t0x94, 0x8e, 0x88, 0xab, 0x92, 0xca, 0x82, 0xd4, 0x62, 0xfd, 0xec, 0xbc, 0xfc, 0xf2, 0x3c, 0x78,\n\t\t0xbc, 0x15, 0x24, 0xfd, 0x60, 0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce,\n\t\t0x1d, 0xa2, 0x39, 0x00, 0xaa, 0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4,\n\t\t0x35, 0x89, 0x0d, 0x6c, 0x94, 0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0x3c, 0x92, 0x48, 0x30, 0x06,\n\t\t0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/common.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x57, 0xdd, 0x72, 0x22, 0xc7,\n\t\t0x15, 0xf6, 0xc0, 0xa2, 0x9f, 0x03, 0xbb, 0x42, 0xad, 0xfd, 0x61, 0xb5, 0x5e, 0xaf, 0x16, 0x97,\n\t\t0x63, 0x79, 0x2b, 0x86, 0x88, 0x4d, 0x52, 0x2e, 0x3b, 0x4e, 0x82, 0xd0, 0x48, 0x9a, 0x5d, 0x02,\n\t\t0xa4, 0x99, 0x95, 0xac, 0xa4, 0xca, 0x53, 0xcd, 0x4c, 0x83, 0x3b, 0x0c, 0xd3, 0x93, 0x99, 0x1e,\n\t\t0x56, 0xf8, 0x22, 0x95, 0xcb, 0xe4, 0x26, 0x8f, 0x90, 0x8b, 0xbc, 0x48, 0x1e, 0x20, 0x97, 0x79,\n\t\t0x97, 0x5c, 0xa7, 0xba, 0xa7, 0x07, 0x81, 0xc2, 0x1a, 0x5f, 0xa4, 0x7c, 0x47, 0x9f, 0xf3, 0x7d,\n\t\t0xa7, 0xbf, 0xd3, 0xdd, 0xe7, 0x1c, 0x06, 0x0e, 0x92, 0x01, 0x8d, 0xea, 0x2e, 0xf1, 0x68, 0xe0,\n\t\t0xd2, 0x3a, 0x09, 0x59, 0x7d, 0x7a, 0x54, 0x77, 0xf9, 0x64, 0xc2, 0x83, 0x5a, 0x18, 0x71, 0xc1,\n\t\t0xd1, 0x9e, 0x44, 0xd4, 0x34, 0xa2, 0x46, 0x42, 0x56, 0x9b, 0x1e, 0xed, 0x7f, 0x30, 0xe2, 0x7c,\n\t\t0xe4, 0xd3, 0xba, 0x82, 0x0c, 0x92, 0x61, 0xdd, 0x4b, 0x22, 0x22, 0x58, 0x46, 0xaa, 0xbe, 0x86,\n\t\t0xdd, 0x4b, 0x1e, 0x8d, 0x87, 0x3e, 0x7f, 0x6b, 0x5e, 0x53, 0x37, 0x91, 0x2e, 0xf4, 0x0c, 0x8a,\n\t\t0x6f, 0xb5, 0xd1, 0x61, 0x5e, 0xc5, 0x38, 0x30, 0x0e, 0xb7, 0x31, 0x64, 0x26, 0xcb, 0x43, 0x0f,\n\t\t0x60, 0x23, 0x4a, 0x02, 0xe9, 0xcb, 0x29, 0x5f, 0x21, 0x4a, 0x02, 0xcb, 0xab, 0x56, 0xa1, 0x94,\n\t\t0x05, 0xb3, 0x67, 0x21, 0x45, 0x08, 0xee, 0x04, 0x64, 0x42, 0x75, 0x00, 0xf5, 0x5b, 0x62, 0x9a,\n\t\t0xae, 0x60, 0x53, 0x26, 0x66, 0xef, 0xc4, 0x3c, 0x85, 0xcd, 0x1e, 0x99, 0xf9, 0x9c, 0x78, 0xd2,\n\t\t0xed, 0x11, 0x41, 0x94, 0xbb, 0x84, 0xd5, 0xef, 0xea, 0x17, 0xb0, 0x79, 0x4a, 0x98, 0x9f, 0x44,\n\t\t0x14, 0x3d, 0x84, 0x8d, 0x88, 0x92, 0x98, 0x07, 0x9a, 0xaf, 0x57, 0xa8, 0x02, 0x9b, 0x1e, 0x15,\n\t\t0x84, 0xf9, 0xb1, 0x52, 0x58, 0xc2, 0xd9, 0xb2, 0xfa, 0x77, 0x03, 0xee, 0xfc, 0x86, 0x4e, 0x38,\n\t\t0xfa, 0x12, 0x36, 0x86, 0x8c, 0xfa, 0x5e, 0x5c, 0x31, 0x0e, 0xf2, 0x87, 0xc5, 0xc6, 0x47, 0xb5,\n\t\t0x15, 0xe7, 0x57, 0x93, 0xd0, 0xda, 0xa9, 0xc2, 0x99, 0x81, 0x88, 0x66, 0x58, 0x93, 0xf6, 0x2f,\n\t\t0xa1, 0xb8, 0x60, 0x46, 0x65, 0xc8, 0x8f, 0xe9, 0x4c, 0xab, 0x90, 0x3f, 0x51, 0x03, 0x0a, 0x53,\n\t\t0xe2, 0x27, 0x54, 0x09, 0x28, 0x36, 0xde, 0x5f, 0x19, 0x5e, 0xa7, 0x89, 0x53, 0xe8, 0xe7, 0xb9,\n\t\t0xcf, 0x8c, 0xea, 0x3f, 0x0c, 0xd8, 0x38, 0xa7, 0xc4, 0xa3, 0x11, 0xfa, 0xd5, 0x2d, 0x89, 0x1f,\n\t\t0xaf, 0x8c, 0x91, 0x82, 0x7f, 0x58, 0x91, 0xff, 0x36, 0xa0, 0xdc, 0xa7, 0x24, 0x72, 0xbf, 0x69,\n\t\t0x0a, 0x11, 0xb1, 0x41, 0x22, 0x68, 0x8c, 0x1c, 0xb8, 0xc7, 0x02, 0x8f, 0x5e, 0x53, 0xcf, 0x59,\n\t\t0x92, 0xfd, 0xd9, 0xca, 0xa8, 0xb7, 0xe9, 0x35, 0x2b, 0xe5, 0x2e, 0xe6, 0x71, 0x97, 0x2d, 0xda,\n\t\t0xf6, 0xbf, 0x06, 0xf4, 0xbf, 0xa0, 0xff, 0x63, 0x56, 0x43, 0xd8, 0x3a, 0x21, 0x82, 0x1c, 0xfb,\n\t\t0x7c, 0x80, 0x4e, 0xe1, 0x2e, 0x0d, 0x5c, 0xee, 0xb1, 0x60, 0xe4, 0x88, 0x59, 0x98, 0x3e, 0xd0,\n\t\t0x7b, 0x8d, 0xe7, 0x2b, 0x63, 0x99, 0x1a, 0x29, 0x5f, 0x34, 0x2e, 0xd1, 0x85, 0xd5, 0xfc, 0x01,\n\t\t0xe7, 0x16, 0x1e, 0x70, 0x2f, 0x2d, 0x3a, 0x1a, 0x5d, 0xd0, 0x28, 0x66, 0x3c, 0xb0, 0x82, 0x21,\n\t\t0x97, 0x40, 0x36, 0x09, 0xfd, 0xac, 0x10, 0xe4, 0x6f, 0xf4, 0x31, 0xec, 0x0c, 0x29, 0x11, 0x49,\n\t\t0x44, 0x9d, 0x69, 0x0a, 0xd5, 0x05, 0x77, 0x4f, 0x9b, 0x75, 0x80, 0xea, 0x6b, 0x78, 0xd4, 0x4f,\n\t\t0xc2, 0x90, 0x47, 0x82, 0x7a, 0x2d, 0x9f, 0xd1, 0x40, 0x68, 0x4f, 0x2c, 0x6b, 0x75, 0xc4, 0x9d,\n\t\t0xd8, 0x1b, 0xeb, 0xc8, 0x85, 0x11, 0xef, 0x7b, 0x63, 0xf4, 0x18, 0xb6, 0xfe, 0x40, 0xa6, 0x44,\n\t\t0x39, 0xd2, 0x98, 0x9b, 0x72, 0xdd, 0xf7, 0xc6, 0xd5, 0x3f, 0xe7, 0xa1, 0x88, 0xa9, 0x88, 0x66,\n\t\t0x3d, 0xee, 0x33, 0x77, 0x86, 0x4e, 0xa0, 0xcc, 0x02, 0x26, 0x18, 0xf1, 0x1d, 0x16, 0x08, 0x1a,\n\t\t0x4d, 0x49, 0xaa, 0xb2, 0xd8, 0x78, 0x5c, 0x4b, 0xdb, 0x4b, 0x2d, 0x6b, 0x2f, 0xb5, 0x13, 0xdd,\n\t\t0x5e, 0xf0, 0x8e, 0xa6, 0x58, 0x9a, 0x81, 0xea, 0xb0, 0x37, 0x20, 0xee, 0x98, 0x0f, 0x87, 0x8e,\n\t\t0xcb, 0xe9, 0x70, 0xc8, 0x5c, 0x29, 0x53, 0xed, 0x6d, 0x60, 0xa4, 0x5d, 0xad, 0x1b, 0x8f, 0xdc,\n\t\t0x76, 0x42, 0xae, 0xd9, 0x24, 0x99, 0xdc, 0x6c, 0x9b, 0x5f, 0xbb, 0xad, 0xa6, 0xcc, 0xb7, 0xfd,\n\t\t0xe4, 0x26, 0x0a, 0x11, 0x82, 0x4e, 0x42, 0x11, 0x57, 0xee, 0x1c, 0x18, 0x87, 0x85, 0x39, 0xb4,\n\t\t0xa9, 0xcd, 0xe8, 0x4b, 0x78, 0x12, 0xf0, 0xc0, 0x89, 0x64, 0xea, 0x64, 0xe0, 0x53, 0x87, 0x46,\n\t\t0x11, 0x8f, 0x9c, 0xb4, 0xa5, 0xc4, 0x95, 0xc2, 0x41, 0xfe, 0x70, 0x1b, 0x57, 0x02, 0x1e, 0xe0,\n\t\t0x0c, 0x61, 0x4a, 0x00, 0x4e, 0xfd, 0xe8, 0x15, 0xec, 0xd1, 0xeb, 0x90, 0xa5, 0x42, 0x6e, 0x24,\n\t\t0x6f, 0xac, 0x93, 0x8c, 0x6e, 0x58, 0x99, 0xea, 0xea, 0x04, 0x1e, 0x59, 0x31, 0xf7, 0x95, 0xf1,\n\t\t0x2c, 0xe2, 0x49, 0xd8, 0x23, 0x91, 0x60, 0xaa, 0x39, 0xaf, 0x68, 0x98, 0xe8, 0x97, 0x50, 0x88,\n\t\t0x05, 0x11, 0xe9, 0x83, 0xbf, 0xd7, 0x38, 0x5c, 0xf9, 0x48, 0x97, 0x03, 0xf6, 0x25, 0x1e, 0xa7,\n\t\t0xb4, 0xea, 0x14, 0x9e, 0x2c, 0x7b, 0x5b, 0x3c, 0x18, 0xb2, 0x91, 0x56, 0x88, 0x2e, 0xa1, 0xcc,\n\t\t0x32, 0xb7, 0x33, 0x92, 0xfe, 0xac, 0xb4, 0x7f, 0xfc, 0x3d, 0x76, 0x9a, 0x4b, 0xc7, 0x3b, 0x6c,\n\t\t0xc9, 0x11, 0x57, 0xff, 0x65, 0xc0, 0x7e, 0x33, 0x9e, 0x05, 0x6e, 0x36, 0x36, 0x96, 0xf7, 0xad,\n\t\t0xc0, 0x26, 0x0d, 0xe4, 0x39, 0xa7, 0x33, 0x68, 0x0b, 0x67, 0x4b, 0xd4, 0x80, 0x07, 0x61, 0x44,\n\t\t0x3d, 0x3a, 0x64, 0x01, 0xf5, 0x9c, 0x3f, 0x26, 0x34, 0xa1, 0x8e, 0x3a, 0x95, 0xf4, 0x29, 0xef,\n\t\t0xdd, 0x38, 0x7f, 0x2b, 0x7d, 0x1d, 0x79, 0x48, 0x4f, 0x01, 0x52, 0xa0, 0x2a, 0xe7, 0xbc, 0x02,\n\t\t0x6e, 0x2b, 0x8b, 0x2a, 0xd4, 0x5f, 0x43, 0x29, 0x75, 0xbb, 0x4a, 0x83, 0x7a, 0x24, 0xc5, 0xc6,\n\t\t0xd3, 0x95, 0x09, 0x66, 0x5d, 0x02, 0x17, 0x15, 0x25, 0x55, 0x5d, 0xfd, 0x4f, 0x1e, 0xde, 0x57,\n\t\t0xb3, 0x8d, 0xb6, 0xfc, 0x24, 0x16, 0x34, 0xea, 0x53, 0x9f, 0xba, 0x32, 0x13, 0x5d, 0x48, 0x7d,\n\t\t0xd8, 0x8a, 0x45, 0x44, 0x04, 0x1d, 0xcd, 0x74, 0x3b, 0x79, 0xb9, 0x32, 0xfc, 0xea, 0x20, 0x7d,\n\t\t0x4d, 0x3d, 0xce, 0x55, 0x0c, 0x3c, 0x0f, 0x84, 0xfe, 0x62, 0xc0, 0x87, 0x44, 0x11, 0x1c, 0x37,\n\t\t0x65, 0x38, 0xb1, 0x60, 0xee, 0x78, 0xe6, 0x44, 0x74, 0x24, 0x2f, 0x4c, 0xe7, 0x93, 0xf6, 0xc2,\n\t\t0x9f, 0x7e, 0x8f, 0x0d, 0x15, 0x1b, 0x2b, 0x72, 0x9a, 0x99, 0xdc, 0xf1, 0xfc, 0x3d, 0xfc, 0x8c,\n\t\t0x7c, 0x37, 0x0c, 0xfd, 0xcd, 0x80, 0x8f, 0x6e, 0x49, 0xa1, 0xd7, 0x82, 0x46, 0x01, 0xf1, 0x1d,\n\t\t0x1a, 0x08, 0x26, 0x66, 0x99, 0x98, 0xb4, 0x8e, 0x7f, 0xbe, 0x5e, 0x8c, 0xa9, 0xf9, 0xa6, 0xa2,\n\t\t0x2f, 0xc9, 0x79, 0x4e, 0xd6, 0x01, 0x11, 0x86, 0xdd, 0x4c, 0x08, 0xc9, 0x06, 0x8d, 0xbe, 0xd8,\n\t\t0xd5, 0xe3, 0x5e, 0x07, 0x9b, 0x4f, 0x25, 0x5c, 0x76, 0x6f, 0x59, 0x8e, 0x77, 0x61, 0x27, 0x3b,\n\t\t0x7b, 0x9d, 0x4d, 0xf5, 0x17, 0x50, 0xbe, 0x4d, 0x44, 0xf7, 0xa1, 0x10, 0xbb, 0x3c, 0xcc, 0xea,\n\t\t0x34, 0x5d, 0xcc, 0x8b, 0x37, 0xb7, 0xf0, 0x6f, 0xe7, 0x15, 0x3c, 0x5b, 0x73, 0xfe, 0xe8, 0x43,\n\t\t0xb8, 0xbb, 0x74, 0xa7, 0x3a, 0x68, 0x29, 0x5e, 0x80, 0x7e, 0x9e, 0xab, 0x18, 0xd5, 0xbf, 0x1a,\n\t\t0xf0, 0x7c, 0xed, 0xf9, 0xa1, 0x9f, 0xc0, 0xfd, 0xdb, 0xf7, 0x32, 0x1f, 0x71, 0xdb, 0xb2, 0x1f,\n\t\t0x2d, 0x72, 0x54, 0x71, 0xd4, 0x64, 0x6f, 0x5b, 0x66, 0xc8, 0x99, 0x9b, 0xa6, 0xb1, 0xbb, 0x4c,\n\t\t0x78, 0x4d, 0x67, 0x4a, 0xcb, 0x57, 0xb0, 0xdb, 0x23, 0x23, 0x16, 0xa8, 0x5a, 0xee, 0x86, 0x42,\n\t\t0x4d, 0xa3, 0x27, 0xb0, 0x1d, 0x92, 0x11, 0x75, 0x62, 0xf6, 0x6d, 0xba, 0x5f, 0x01, 0x6f, 0x49,\n\t\t0x43, 0x9f, 0x7d, 0x4b, 0xd1, 0x8f, 0x60, 0x27, 0xa0, 0xd7, 0xc2, 0x51, 0x08, 0xc1, 0xc7, 0x34,\n\t\t0xd0, 0x63, 0xf3, 0xae, 0x34, 0xf7, 0xc8, 0x88, 0xda, 0xd2, 0xf8, 0xe2, 0x2d, 0x94, 0x16, 0x27,\n\t\t0x2e, 0x7a, 0x0c, 0x0f, 0xcc, 0x4e, 0xab, 0x7b, 0x62, 0x75, 0xce, 0x1c, 0xfb, 0xaa, 0x67, 0x3a,\n\t\t0x56, 0xe7, 0xa2, 0xd9, 0xb6, 0x4e, 0xca, 0xef, 0xa1, 0x7d, 0x78, 0xb8, 0xec, 0xb2, 0xcf, 0xb1,\n\t\t0x75, 0x6a, 0xe3, 0xcb, 0xb2, 0x81, 0x1e, 0x02, 0x5a, 0xf6, 0xbd, 0xea, 0x77, 0x3b, 0xe5, 0x1c,\n\t\t0xaa, 0xc0, 0xfd, 0x65, 0x7b, 0x0f, 0x77, 0xed, 0xee, 0xcb, 0x72, 0xfe, 0xc5, 0x9f, 0x60, 0x6f,\n\t\t0x45, 0x17, 0x45, 0xcf, 0xe1, 0xa9, 0xd5, 0xef, 0xb6, 0x9b, 0xb6, 0xd5, 0xed, 0x38, 0x67, 0xb8,\n\t\t0xfb, 0xa6, 0xe7, 0xf4, 0xed, 0xa6, 0xbd, 0xa8, 0xe3, 0x9d, 0x90, 0x73, 0xb3, 0xd9, 0xb6, 0xcf,\n\t\t0xaf, 0xca, 0xc6, 0xbb, 0x21, 0x27, 0xb8, 0x69, 0x75, 0xcc, 0x93, 0x72, 0xee, 0xc5, 0x3f, 0x0d,\n\t\t0xf8, 0xe0, 0xbb, 0x9b, 0x03, 0xfa, 0x14, 0x3e, 0x69, 0xb6, 0x6c, 0xeb, 0xc2, 0x74, 0x5a, 0xed,\n\t\t0x37, 0x7d, 0xdb, 0xc4, 0x4e, 0xdf, 0x6c, 0x9b, 0x2d, 0x15, 0xb4, 0x6f, 0xe3, 0xa6, 0x6d, 0x9e,\n\t\t0x5d, 0x2d, 0xe8, 0x7a, 0x09, 0xf5, 0xf5, 0x70, 0x6c, 0x9e, 0xa5, 0x6b, 0xab, 0xf5, 0x5a, 0x2a,\n\t\t0xfd, 0x19, 0x1c, 0xad, 0x27, 0x99, 0x5f, 0xd9, 0x26, 0xee, 0x34, 0xdb, 0x8e, 0xd9, 0xb1, 0x2d,\n\t\t0xfb, 0xaa, 0x9c, 0xdb, 0xcf, 0x55, 0x8c, 0x17, 0x5f, 0x43, 0x49, 0xfe, 0x77, 0xe7, 0x53, 0x1a,\n\t\t0x65, 0x57, 0x77, 0xda, 0xb4, 0xda, 0xdd, 0x0b, 0x13, 0xdf, 0xbe, 0xba, 0x47, 0xb0, 0xb7, 0xec,\n\t\t0x3a, 0xed, 0xe2, 0x96, 0x59, 0x36, 0xe4, 0x9d, 0x2e, 0x3b, 0xce, 0x70, 0xb3, 0x65, 0x9e, 0xbe,\n\t\t0x69, 0x97, 0x73, 0xc7, 0xbf, 0x87, 0x47, 0x2e, 0x9f, 0xac, 0xaa, 0xed, 0xe3, 0x62, 0x4b, 0x7d,\n\t\t0x2d, 0xf5, 0xe4, 0x00, 0xee, 0x19, 0xbf, 0x3b, 0x1a, 0x31, 0xf1, 0x4d, 0x32, 0xa8, 0xb9, 0x7c,\n\t\t0x52, 0x5f, 0xfc, 0xb6, 0xfa, 0x94, 0x79, 0x7e, 0x7d, 0xc4, 0xd3, 0x2f, 0x26, 0xfd, 0xa1, 0xf5,\n\t\t0x05, 0x09, 0xd9, 0xf4, 0x68, 0xb0, 0xa1, 0x6c, 0x2f, 0xff, 0x1b, 0x00, 0x00, 0xff, 0xff, 0x5e,\n\t\t0xe2, 0xc9, 0x06, 0x8c, 0x0d, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x5c, 0x5d, 0x6c, 0x1c, 0x49,\n\t\t0xb5, 0xde, 0x9e, 0xb1, 0xc7, 0x9e, 0x33, 0x8e, 0x3d, 0x2e, 0x3b, 0x8e, 0xff, 0x12, 0x3b, 0x93,\n\t\t0x6c, 0xe2, 0x75, 0x9c, 0x71, 0xe2, 0x64, 0x93, 0x4d, 0xb2, 0x3f, 0xd7, 0x76, 0x6c, 0xed, 0x48,\n\t\t0xbe, 0x49, 0x6e, 0xc7, 0xc9, 0xde, 0x7b, 0xb5, 0xd2, 0xd0, 0xee, 0x2e, 0xc7, 0x8d, 0x67, 0xba,\n\t\t0x67, 0xbb, 0x6b, 0x3c, 0x31, 0x82, 0x27, 0x1e, 0x90, 0x10, 0x2b, 0x58, 0xad, 0x90, 0x58, 0x09,\n\t\t0x04, 0x42, 0x02, 0xb1, 0xfc, 0x68, 0x11, 0x08, 0xf1, 0xf7, 0x02, 0x48, 0x68, 0x91, 0x40, 0x0b,\n\t\t0x4f, 0xbc, 0xf0, 0x84, 0x84, 0x10, 0xfb, 0xc6, 0x03, 0xcb, 0x33, 0xea, 0xea, 0xea, 0x99, 0xe9,\n\t\t0x9e, 0xaa, 0xfe, 0x19, 0x3b, 0x59, 0xd0, 0xe6, 0x6d, 0xba, 0xfb, 0x9c, 0xd3, 0x5f, 0x55, 0x9d,\n\t\t0x73, 0xea, 0xd4, 0x39, 0xa7, 0x07, 0x4e, 0xd6, 0xb7, 0xb0, 0xb5, 0xa8, 0x2a, 0x1a, 0x36, 0x54,\n\t\t0xbc, 0xa8, 0xd4, 0xf4, 0xc5, 0xbd, 0x8b, 0x8b, 0x3b, 0xba, 0x4d, 0x4c, 0x6b, 0xbf, 0x58, 0xb3,\n\t\t0x4c, 0x62, 0xa2, 0x11, 0x87, 0xa4, 0xc8, 0x48, 0x8a, 0x4a, 0x4d, 0x2f, 0xee, 0x5d, 0x9c, 0x3c,\n\t\t0xf1, 0xc0, 0x34, 0x1f, 0x54, 0xf0, 0x22, 0x25, 0xd9, 0xaa, 0x6f, 0x2f, 0x6a, 0x75, 0x4b, 0x21,\n\t\t0xba, 0x69, 0xb8, 0x4c, 0x93, 0x33, 0xc1, 0xe7, 0x44, 0xaf, 0x62, 0x9b, 0x28, 0xd5, 0x1a, 0x23,\n\t\t0x98, 0xe5, 0xbd, 0x58, 0x35, 0xab, 0xd5, 0xa6, 0x88, 0x02, 0x8f, 0x82, 0x28, 0xf6, 0x6e, 0x45,\n\t\t0xb7, 0x49, 0x18, 0x4d, 0xc3, 0xb4, 0x76, 0xb7, 0x2b, 0x66, 0xc3, 0xa5, 0x29, 0xdc, 0x84, 0xbe,\n\t\t0x97, 0xdd, 0x01, 0xa1, 0x6b, 0x90, 0xc1, 0x7b, 0xd8, 0x20, 0xf6, 0xb8, 0x34, 0x9b, 0x9e, 0xcb,\n\t\t0x2d, 0x9d, 0x2c, 0x72, 0xc6, 0x56, 0x64, 0xd4, 0x6b, 0x0e, 0xa5, 0xcc, 0x18, 0x0a, 0xef, 0x5f,\n\t\t0x85, 0x81, 0xf6, 0x07, 0x68, 0x02, 0xfa, 0xe9, 0xa3, 0xb2, 0xae, 0x8d, 0x4b, 0xb3, 0xd2, 0x5c,\n\t\t0x5a, 0xee, 0xa3, 0xd7, 0x25, 0x0d, 0x5d, 0x03, 0x70, 0x1f, 0x39, 0x83, 0x1e, 0x4f, 0xcd, 0x4a,\n\t\t0x73, 0xb9, 0xa5, 0xc9, 0xa2, 0x3b, 0x23, 0x45, 0x6f, 0x46, 0x8a, 0x9b, 0xde, 0x8c, 0xc8, 0x59,\n\t\t0x4a, 0xed, 0x5c, 0xa3, 0x71, 0xe8, 0xdb, 0xc3, 0x96, 0xad, 0x9b, 0xc6, 0x78, 0xda, 0x15, 0xca,\n\t\t0x2e, 0xd1, 0x31, 0xe8, 0x73, 0x06, 0xef, 0xbc, 0xae, 0x87, 0x3e, 0xc9, 0x38, 0x97, 0x25, 0x0d,\n\t\t0x7d, 0x59, 0x82, 0x73, 0xde, 0x90, 0xcb, 0xf8, 0x21, 0x56, 0xeb, 0xce, 0x3a, 0x94, 0x6d, 0xa2,\n\t\t0x58, 0x04, 0x6b, 0x65, 0x17, 0x89, 0x42, 0x88, 0xa5, 0x6f, 0xd5, 0x09, 0xb6, 0xc7, 0x7b, 0x29,\n\t\t0x9e, 0xe7, 0xb9, 0x43, 0x7f, 0x85, 0xc9, 0x59, 0xf3, 0xc4, 0xdc, 0x75, 0xa5, 0xd0, 0x21, 0x2f,\n\t\t0x37, 0x65, 0xbc, 0xfc, 0x94, 0x7c, 0xb6, 0x11, 0x8f, 0x14, 0x7d, 0x5d, 0x82, 0xf3, 0x1c, 0x78,\n\t\t0xaa, 0x59, 0xad, 0x55, 0x30, 0x17, 0x60, 0x86, 0x02, 0x7c, 0x31, 0x1e, 0xc0, 0x55, 0x4f, 0x4e,\n\t\t0x27, 0xc4, 0x67, 0x1a, 0x71, 0x89, 0xd1, 0x5b, 0x12, 0xcc, 0x73, 0x40, 0x6e, 0x2b, 0x7a, 0x85,\n\t\t0x87, 0xb0, 0x8f, 0x22, 0xbc, 0x11, 0x0f, 0xe1, 0x3a, 0x15, 0xd2, 0x09, 0xef, 0x4c, 0x23, 0x16,\n\t\t0x25, 0xfa, 0x1a, 0x7f, 0x02, 0x1d, 0xdd, 0xd2, 0xca, 0x66, 0x9d, 0x74, 0xc2, 0xeb, 0xa7, 0xf0,\n\t\t0x5e, 0x88, 0x07, 0xcf, 0x51, 0x3b, 0xed, 0x76, 0x9d, 0x74, 0x02, 0x9c, 0x6b, 0xc4, 0xa4, 0x45,\n\t\t0x6f, 0x4a, 0x30, 0xa7, 0x61, 0x55, 0xb7, 0x29, 0x30, 0x47, 0x4b, 0x6d, 0x75, 0x07, 0x6b, 0x75,\n\t\t0xee, 0xe4, 0x65, 0x29, 0xba, 0x6b, 0x5c, 0x74, 0x37, 0x99, 0x90, 0x4d, 0xc5, 0xde, 0xbd, 0xeb,\n\t\t0x89, 0xe8, 0x44, 0x76, 0x5a, 0x8b, 0x41, 0x87, 0x5e, 0x97, 0xe0, 0x4c, 0x00, 0x95, 0xc8, 0x26,\n\t\t0x80, 0x62, 0xba, 0x1a, 0x8d, 0x49, 0x64, 0x0e, 0x05, 0x2d, 0x92, 0x8a, 0x33, 0x4b, 0x21, 0x46,\n\t\t0x90, 0x8b, 0x39, 0x4b, 0x21, 0xfa, 0xef, 0x9b, 0x25, 0xa1, 0xea, 0xbf, 0xd1, 0x81, 0x2a, 0x44,\n\t\t0xb3, 0x06, 0x28, 0xaa, 0xe7, 0x22, 0x51, 0x89, 0x95, 0xea, 0x94, 0x16, 0x4d, 0x86, 0x3e, 0x2b,\n\t\t0xc1, 0xd3, 0x7e, 0x4c, 0x22, 0x4b, 0x3c, 0x42, 0x01, 0x5d, 0x89, 0x04, 0x24, 0x32, 0xc2, 0x93,\n\t\t0x5a, 0x14, 0x11, 0x5d, 0x36, 0x45, 0x25, 0xfa, 0x9e, 0x4e, 0xf6, 0x23, 0x95, 0x7b, 0x30, 0x64,\n\t\t0xd9, 0x96, 0x99, 0x90, 0x28, 0xe5, 0x56, 0x62, 0xd0, 0x51, 0xe5, 0x0e, 0xa0, 0x12, 0x29, 0xf7,\n\t\t0x50, 0x88, 0x72, 0xfb, 0x30, 0x09, 0x95, 0x5b, 0x89, 0xa4, 0xe2, 0xcc, 0x52, 0x88, 0x72, 0xe7,\n\t\t0x63, 0xce, 0x52, 0x98, 0x72, 0x2b, 0x31, 0xe8, 0xa8, 0x22, 0xf9, 0x51, 0x89, 0x14, 0x69, 0x38,\n\t\t0x44, 0x91, 0xda, 0x21, 0x09, 0x15, 0x49, 0x89, 0x22, 0xa2, 0x96, 0xe6, 0x07, 0x13, 0x62, 0x69,\n\t\t0x28, 0xc4, 0xd2, 0xda, 0xf1, 0x84, 0x58, 0x9a, 0x12, 0x4d, 0x86, 0x1a, 0x70, 0xc2, 0x01, 0x61,\n\t\t0x89, 0xb5, 0x67, 0x84, 0x02, 0xb9, 0xc0, 0x05, 0xe2, 0x48, 0xb5, 0x84, 0x6a, 0x33, 0x45, 0xc4,\n\t\t0x8f, 0xd1, 0x6b, 0x30, 0xed, 0xbe, 0x78, 0x5b, 0xb7, 0x78, 0xaf, 0x1d, 0xa5, 0xaf, 0x2d, 0x8a,\n\t\t0x5f, 0xbb, 0xee, 0xf0, 0x75, 0xbe, 0x74, 0x82, 0x88, 0x1e, 0xa2, 0x6f, 0x4a, 0xb0, 0x18, 0x50,\n\t\t0x51, 0xc5, 0x50, 0x71, 0xa5, 0x6c, 0xe1, 0xd7, 0xea, 0xd8, 0xe6, 0x8e, 0xfe, 0x28, 0x85, 0xf1,\n\t\t0x52, 0xb4, 0xa6, 0x52, 0x49, 0xb2, 0x27, 0xa8, 0x13, 0xd7, 0xbc, 0x12, 0x9b, 0x1a, 0xfd, 0x50,\n\t\t0x82, 0xcb, 0x0c, 0x93, 0x07, 0x31, 0x9e, 0x12, 0x8f, 0x51, 0xb4, 0xab, 0x5c, 0xb4, 0xec, 0x6d,\n\t\t0xee, 0xab, 0xe3, 0x68, 0x74, 0xd1, 0x4a, 0xc4, 0x81, 0xbe, 0x20, 0xc1, 0x59, 0xde, 0xf4, 0xf2,\n\t\t0x80, 0x1e, 0x8b, 0xa9, 0xdd, 0xab, 0x4c, 0x42, 0x84, 0x76, 0x0b, 0xc8, 0xd0, 0x27, 0x60, 0xc6,\n\t\t0x55, 0x32, 0x31, 0x92, 0x71, 0x8a, 0xe4, 0xa2, 0x58, 0xcf, 0xc4, 0x10, 0x5c, 0x05, 0x16, 0xbd,\n\t\t0xfb, 0x33, 0x12, 0x9c, 0x66, 0x8b, 0xc7, 0x14, 0x5d, 0xb0, 0x68, 0x13, 0x14, 0xc1, 0xb3, 0x5c,\n\t\t0x04, 0xae, 0x70, 0x57, 0xdf, 0x05, 0xcb, 0x34, 0xab, 0x46, 0xd0, 0xa0, 0x4f, 0xc1, 0x6c, 0x55,\n\t\t0xb1, 0x76, 0xb1, 0x55, 0xb6, 0xb0, 0x6a, 0x5a, 0x1a, 0x0f, 0xc4, 0x24, 0x05, 0xb1, 0xc4, 0x05,\n\t\t0xf1, 0xdf, 0x94, 0x59, 0x66, 0xbc, 0x9d, 0x08, 0x8e, 0x57, 0xc3, 0x08, 0xd0, 0x57, 0x25, 0x58,\n\t\t0xe0, 0x9d, 0x4f, 0xf4, 0x07, 0x86, 0xc2, 0x9d, 0x90, 0xa9, 0x24, 0xe1, 0xeb, 0x5d, 0x26, 0x26,\n\t\t0x4e, 0xf8, 0x2a, 0xa0, 0x45, 0xdf, 0x90, 0xa0, 0xc8, 0x8b, 0xb0, 0xb1, 0x55, 0xd5, 0x0d, 0x85,\n\t\t0xeb, 0x17, 0xa6, 0x43, 0xfc, 0x42, 0x67, 0x88, 0xdd, 0x14, 0xc4, 0xf1, 0x0b, 0x8d, 0xd8, 0xd4,\n\t\t0xe8, 0x47, 0x12, 0x5c, 0xe6, 0x1d, 0xa5, 0x22, 0xbd, 0xd8, 0x71, 0x8a, 0xf6, 0x66, 0xcc, 0x13,\n\t\t0x55, 0x94, 0x2b, 0x5b, 0x6c, 0x24, 0x63, 0x11, 0x69, 0x80, 0xd8, 0x28, 0x4f, 0x24, 0xd1, 0x00,\n\t\t0xb1, 0x81, 0xce, 0x35, 0x62, 0xd2, 0xa2, 0xbf, 0x48, 0xb0, 0x16, 0xf0, 0xb8, 0xf8, 0x21, 0xc1,\n\t\t0x96, 0xa1, 0x54, 0xca, 0x1c, 0xe4, 0xba, 0xa1, 0x13, 0x9d, 0xaf, 0x18, 0x33, 0x14, 0xfa, 0xdd,\n\t\t0x68, 0x17, 0xbc, 0xc6, 0xe4, 0x77, 0x8c, 0xa7, 0xe4, 0x09, 0xef, 0x1c, 0xd0, 0x8b, 0xd6, 0x81,\n\t\t0x24, 0xa0, 0x3f, 0x49, 0xb0, 0x92, 0x60, 0x98, 0x22, 0x8f, 0x35, 0x4b, 0xc7, 0x78, 0xe7, 0x00,\n\t\t0x63, 0x14, 0x39, 0xb3, 0x1b, 0x56, 0xf7, 0xec, 0xe8, 0x3d, 0x09, 0x5e, 0x08, 0x1b, 0x4e, 0xb4,\n\t\t0x9d, 0x9c, 0xa4, 0x03, 0xdb, 0xe0, 0x0e, 0x4c, 0x08, 0x26, 0xd2, 0x5e, 0xae, 0xe2, 0xee, 0x58,\n\t\t0x69, 0x1c, 0xc0, 0x4d, 0x9d, 0x18, 0x44, 0x37, 0xea, 0x58, 0x2b, 0x2b, 0x76, 0xd9, 0xc0, 0x8d,\n\t\t0xce, 0x71, 0x14, 0x42, 0xe2, 0x00, 0x4e, 0x06, 0x85, 0x89, 0x5b, 0xb6, 0x6f, 0xe1, 0x06, 0x27,\n\t\t0x0e, 0x68, 0x24, 0xe2, 0x40, 0xbf, 0x92, 0xe0, 0x1a, 0x8d, 0x26, 0xcb, 0xea, 0x8e, 0x5e, 0xd1,\n\t\t0x12, 0xda, 0xcf, 0x29, 0x0a, 0xfd, 0x65, 0x2e, 0x74, 0x1a, 0x4a, 0xae, 0x3a, 0x42, 0x93, 0x18,\n\t\t0xcd, 0x25, 0x3b, 0x39, 0x1b, 0xfa, 0xa9, 0x04, 0x57, 0x22, 0x06, 0x21, 0xb2, 0x8e, 0xd3, 0x74,\n\t\t0x04, 0x6b, 0x49, 0x47, 0x20, 0x32, 0x89, 0x0b, 0x76, 0x42, 0x1e, 0xf4, 0x5d, 0x09, 0x2e, 0x0a,\n\t\t0x51, 0x0b, 0xe3, 0xfc, 0xa7, 0x29, 0xec, 0x65, 0x7e, 0x18, 0xc2, 0x7d, 0xbb, 0x30, 0xf0, 0x5f,\n\t\t0x50, 0x13, 0xd0, 0xa3, 0x1f, 0x48, 0x70, 0x49, 0x08, 0x37, 0xe4, 0x10, 0x79, 0x26, 0x44, 0xc9,\n\t\t0xf9, 0x80, 0x43, 0x8e, 0x93, 0x45, 0x35, 0x11, 0x07, 0x7a, 0x5b, 0x82, 0x0b, 0x89, 0x35, 0xe3,\n\t\t0x2c, 0x45, 0xfc, 0x5f, 0x09, 0x10, 0x8b, 0x94, 0xe2, 0x9c, 0x9a, 0x40, 0x1f, 0xde, 0x91, 0x60,\n\t\t0x49, 0x3c, 0xc1, 0xc2, 0x4d, 0x78, 0x8e, 0xa2, 0x5d, 0x49, 0x32, 0xbf, 0xc2, 0x9d, 0xf8, 0xbc,\n\t\t0x9a, 0x84, 0x01, 0x7d, 0x3f, 0x4c, 0x25, 0x42, 0x0e, 0xcd, 0xcf, 0x24, 0x86, 0x2c, 0x3e, 0x3e,\n\t\t0x0b, 0x20, 0x8b, 0x0e, 0xd2, 0x4e, 0x6c, 0x26, 0x86, 0x1c, 0x12, 0x49, 0xce, 0x87, 0xc4, 0x66,\n\t\t0x02, 0xcc, 0x21, 0xe1, 0xe4, 0xa2, 0x9a, 0x8c, 0x85, 0x6e, 0x9a, 0x6e, 0x28, 0xde, 0x6d, 0xc4,\n\t\t0x73, 0x2e, 0x64, 0xd3, 0x74, 0x23, 0xee, 0x6e, 0x42, 0x9d, 0xab, 0x76, 0x77, 0xac, 0xe8, 0xd7,\n\t\t0x12, 0x5c, 0x8f, 0x31, 0x20, 0x91, 0x8d, 0x2e, 0xd0, 0xd1, 0x94, 0xba, 0x19, 0x8d, 0xc8, 0x58,\n\t\t0x2f, 0xdb, 0x5d, 0xf0, 0xa1, 0x9f, 0x48, 0xf0, 0x6c, 0xd8, 0x00, 0xc4, 0xe7, 0xa7, 0xf3, 0x21,\n\t\t0x1b, 0x90, 0x10, 0x84, 0xf8, 0x1c, 0x75, 0x01, 0x27, 0xe4, 0xa1, 0x0e, 0xa7, 0x5e, 0xb3, 0xb1,\n\t\t0x45, 0x5a, 0xc0, 0x6d, 0xac, 0x58, 0xea, 0x4e, 0x1b, 0xcc, 0x4e, 0xdc, 0xc5, 0x10, 0xeb, 0xbd,\n\t\t0x47, 0xc5, 0x79, 0x08, 0xee, 0x52, 0x61, 0xad, 0x37, 0x72, 0xac, 0xb7, 0x9e, 0x84, 0x61, 0x65,\n\t\t0x00, 0xa0, 0x05, 0xa4, 0xf0, 0xe7, 0x21, 0x38, 0x1b, 0x77, 0xf7, 0x5a, 0x87, 0x23, 0xcd, 0x31,\n\t\t0x92, 0xfd, 0x1a, 0xa6, 0xb5, 0x40, 0x51, 0x65, 0xd1, 0x13, 0xba, 0xb9, 0x5f, 0xc3, 0xf2, 0x40,\n\t\t0xa3, 0xed, 0x0a, 0xbd, 0x0a, 0x47, 0x6b, 0x8a, 0xe5, 0xcc, 0x48, 0xbb, 0xd1, 0x6d, 0x9b, 0xac,\n\t\t0x7c, 0x38, 0xc7, 0x95, 0x77, 0x87, 0x72, 0xb4, 0xd9, 0xc4, 0xb6, 0x29, 0x8f, 0xd4, 0x3a, 0x6f,\n\t\t0xa2, 0xeb, 0x90, 0xa5, 0x19, 0x99, 0x8a, 0x6e, 0x13, 0x5a, 0x58, 0xcc, 0x2d, 0x1d, 0xe7, 0xa7,\n\t\t0x3c, 0x14, 0x7b, 0x77, 0x43, 0xb7, 0x89, 0xdc, 0x4f, 0xd8, 0x2f, 0xb4, 0x04, 0xbd, 0xba, 0x51,\n\t\t0xab, 0x13, 0x5a, 0x76, 0xcc, 0x2d, 0x4d, 0x0b, 0x90, 0xec, 0x57, 0x4c, 0x45, 0x93, 0x5d, 0x52,\n\t\t0xa4, 0xc0, 0x6c, 0x20, 0xe4, 0x28, 0x13, 0xb3, 0xac, 0x56, 0x4c, 0x1b, 0x53, 0xff, 0x6d, 0xd6,\n\t\t0x09, 0xab, 0x43, 0x4e, 0x74, 0xd4, 0x45, 0x6f, 0xb2, 0x4a, 0xb2, 0x3c, 0x8d, 0x7d, 0x73, 0xbf,\n\t\t0x69, 0xae, 0x3a, 0xfc, 0x9b, 0x2e, 0x3b, 0x7a, 0x05, 0xa6, 0x5a, 0x69, 0xef, 0x4e, 0xe9, 0x99,\n\t\t0x28, 0xe9, 0xc7, 0x88, 0x97, 0xcc, 0x0e, 0x08, 0xbe, 0x01, 0x93, 0xad, 0x08, 0xbb, 0x35, 0x0a,\n\t\t0xab, 0x6e, 0x94, 0x75, 0x8d, 0x96, 0xfe, 0xb2, 0xf2, 0xb1, 0x26, 0x45, 0x73, 0x9e, 0xe5, 0xba,\n\t\t0x51, 0xd2, 0x50, 0x09, 0xb2, 0xcc, 0x55, 0x9a, 0x16, 0xad, 0xc3, 0x0d, 0x2e, 0x9d, 0xe3, 0xbb,\n\t\t0x76, 0x26, 0x80, 0x86, 0xd0, 0x25, 0x8f, 0x45, 0x6e, 0x71, 0xa3, 0x12, 0x0c, 0xb7, 0x70, 0x38,\n\t\t0xee, 0xaa, 0x6e, 0x61, 0x56, 0x3c, 0xe3, 0xaf, 0xc1, 0xba, 0x4b, 0x23, 0xe7, 0x9b, 0x6c, 0xec,\n\t\t0x0e, 0x92, 0x61, 0xac, 0xa2, 0x38, 0x67, 0x3e, 0x37, 0x9c, 0xa1, 0xc3, 0xc1, 0x76, 0xbd, 0x42,\n\t\t0x58, 0xe1, 0x2b, 0x7c, 0x4d, 0x47, 0x1d, 0xde, 0xd5, 0x26, 0xab, 0x4c, 0x39, 0xd1, 0x35, 0x98,\n\t\t0x30, 0x2d, 0xfd, 0x81, 0xee, 0x3a, 0xda, 0xc0, 0x2c, 0xe5, 0xe8, 0x2c, 0x8d, 0x79, 0x04, 0x81,\n\t\t0x49, 0x9a, 0x84, 0x7e, 0x5d, 0xc3, 0x06, 0xd1, 0xc9, 0x3e, 0xad, 0x28, 0x65, 0xe5, 0xe6, 0x35,\n\t\t0xba, 0x04, 0x63, 0xdb, 0xba, 0x65, 0x93, 0x4e, 0x99, 0x47, 0x28, 0xe5, 0x08, 0x7d, 0x1a, 0x10,\n\t\t0xb8, 0x0a, 0x03, 0x16, 0x26, 0xd6, 0x7e, 0xb9, 0x66, 0x56, 0x74, 0x75, 0x9f, 0x55, 0x61, 0x66,\n\t\t0x05, 0x07, 0x54, 0x62, 0xed, 0xdf, 0xa1, 0x74, 0x72, 0xce, 0x6a, 0x5d, 0xa0, 0x71, 0xe8, 0x53,\n\t\t0x08, 0xc1, 0xd5, 0x1a, 0xa1, 0x15, 0x93, 0x5e, 0xd9, 0xbb, 0x44, 0xab, 0x30, 0x84, 0x1f, 0xd6,\n\t\t0x74, 0x57, 0x71, 0xdc, 0xa2, 0x7e, 0x3e, 0xb2, 0xa8, 0x3f, 0xd8, 0x62, 0xa1, 0x95, 0xfd, 0x53,\n\t\t0x70, 0x44, 0xb5, 0x1c, 0x6b, 0x60, 0x15, 0x1d, 0x5a, 0x71, 0xc8, 0xca, 0x03, 0xce, 0x4d, 0xaf,\n\t\t0xca, 0x83, 0xfe, 0x17, 0xa6, 0xdc, 0xd1, 0xfb, 0xab, 0x5f, 0x5b, 0x8a, 0xba, 0x6b, 0x6e, 0x6f,\n\t\t0xb3, 0xa2, 0x40, 0x88, 0x52, 0x8f, 0x53, 0xee, 0xf6, 0xc2, 0xd7, 0x8a, 0xcb, 0x8a, 0xce, 0x43,\n\t\t0x4f, 0x15, 0x57, 0x4d, 0x96, 0xce, 0x9f, 0xe0, 0x27, 0xfa, 0x70, 0xd5, 0x94, 0x29, 0x19, 0x92,\n\t\t0x61, 0xb8, 0xc3, 0x63, 0xb3, 0x9c, 0xfc, 0xd3, 0xfc, 0xbd, 0x31, 0xe0, 0x61, 0xe5, 0xbc, 0x1d,\n\t\t0xb8, 0x83, 0xee, 0xc1, 0x58, 0xcd, 0xc2, 0x7b, 0x65, 0xa5, 0x4e, 0x4c, 0x47, 0xff, 0x30, 0x29,\n\t\t0xd7, 0x4c, 0xdd, 0x20, 0x5e, 0x96, 0x5d, 0xb4, 0x5e, 0x36, 0x26, 0x77, 0x28, 0x9d, 0x3c, 0xe2,\n\t\t0xf0, 0x2f, 0xd7, 0x89, 0xd9, 0x76, 0x13, 0x5d, 0x82, 0xcc, 0x0e, 0x56, 0x34, 0x6c, 0xb1, 0xf4,\n\t\t0xf7, 0x14, 0xbf, 0xa9, 0x83, 0x92, 0xc8, 0x8c, 0x14, 0x6d, 0xc0, 0xa8, 0x3b, 0xd1, 0xad, 0x5a,\n\t\t0x1e, 0x5d, 0xd7, 0x63, 0x91, 0xeb, 0x8a, 0x28, 0x5f, 0xb3, 0x2e, 0x47, 0xd7, 0xf6, 0x93, 0x90,\n\t\t0xaf, 0x29, 0x16, 0xd1, 0xbd, 0xe3, 0xf9, 0xb6, 0xfe, 0x60, 0x7c, 0x9c, 0x76, 0x98, 0xfc, 0xcf,\n\t\t0x41, 0xda, 0x2c, 0x1c, 0xff, 0xee, 0x0a, 0x5d, 0xa5, 0x32, 0xd7, 0x0c, 0x62, 0xed, 0xcb, 0x43,\n\t\t0x35, 0xff, 0x5d, 0x74, 0x1c, 0xc0, 0x4b, 0xea, 0xe8, 0x1a, 0x4d, 0x27, 0x67, 0xe5, 0x2c, 0xbb,\n\t\t0x53, 0xd2, 0xd0, 0x7d, 0x18, 0xa1, 0x8a, 0x67, 0xee, 0x61, 0xab, 0xa2, 0xd4, 0x3c, 0x1b, 0x99,\n\t\t0xa4, 0xce, 0xe9, 0x0c, 0xdf, 0x39, 0x59, 0xa6, 0x71, 0xdb, 0x25, 0x67, 0x96, 0x32, 0xac, 0x06,\n\t\t0x6f, 0xa1, 0x87, 0x30, 0x43, 0x73, 0xf0, 0xb8, 0xac, 0x56, 0xea, 0x36, 0xc1, 0x56, 0xd9, 0xc6,\n\t\t0x15, 0xac, 0xd2, 0x39, 0x60, 0xef, 0x98, 0x0a, 0x49, 0xae, 0xd3, 0x34, 0x3f, 0x5e, 0x75, 0x59,\n\t\t0xef, 0x7a, 0x9c, 0xec, 0x75, 0xd3, 0x4a, 0xc8, 0xd3, 0xc9, 0x15, 0x18, 0xe5, 0xcd, 0x0c, 0xca,\n\t\t0x43, 0x7a, 0x17, 0xef, 0xd3, 0x1d, 0x38, 0x2b, 0x3b, 0x3f, 0xd1, 0x28, 0xf4, 0xee, 0x29, 0x95,\n\t\t0xba, 0xdb, 0x84, 0x93, 0x95, 0xdd, 0x8b, 0xeb, 0xa9, 0xe7, 0xa4, 0xc2, 0xdb, 0x12, 0x3c, 0x13,\n\t\t0xff, 0xb8, 0x77, 0x19, 0x32, 0xcc, 0x61, 0x4a, 0x31, 0x1c, 0x26, 0xa3, 0x45, 0xeb, 0x30, 0x1b,\n\t\t0x5e, 0xef, 0xd7, 0x35, 0x0a, 0x2c, 0x2d, 0x4f, 0x8b, 0x4b, 0xf5, 0x25, 0xad, 0xf0, 0x2d, 0x09,\n\t\t0xce, 0xc4, 0x8c, 0x1a, 0xaf, 0x40, 0x9f, 0xb7, 0x55, 0x48, 0x31, 0xb6, 0x0a, 0x8f, 0xf8, 0xd0,\n\t\t0xa0, 0x9a, 0x30, 0x17, 0xfb, 0xc8, 0xb4, 0x0a, 0x03, 0x6c, 0xb7, 0x6e, 0x45, 0x4e, 0x83, 0x02,\n\t\t0x2f, 0xc0, 0x36, 0x67, 0x1a, 0x38, 0xe5, 0x48, 0xeb, 0xa2, 0xf0, 0x3b, 0x09, 0x4e, 0xc7, 0xe9,\n\t\t0x1a, 0xf1, 0x87, 0x40, 0x52, 0xb2, 0x10, 0xe8, 0x16, 0x8c, 0x09, 0xc2, 0x8c, 0x54, 0x94, 0x47,\n\t\t0x1e, 0xb1, 0x39, 0x21, 0x46, 0xdb, 0x56, 0x93, 0xf6, 0x6d, 0x35, 0x85, 0xd7, 0x25, 0x28, 0x44,\n\t\t0x37, 0x9c, 0xa0, 0x05, 0x40, 0xc1, 0x26, 0x84, 0x66, 0x1b, 0x5a, 0xde, 0xf6, 0x4d, 0x41, 0x60,\n\t\t0xbf, 0x4d, 0x05, 0xf6, 0x5b, 0xbf, 0xf3, 0x48, 0x07, 0x9c, 0x47, 0xe1, 0xef, 0x81, 0xe9, 0x15,\n\t\t0x5a, 0x48, 0x32, 0x44, 0x73, 0x90, 0xf7, 0x27, 0xa2, 0x9a, 0xea, 0x35, 0x68, 0xb7, 0x8d, 0x38,\n\t\t0x80, 0x3d, 0x1d, 0xc0, 0x7e, 0x16, 0x86, 0xb6, 0x74, 0x43, 0xb1, 0xf6, 0xcb, 0xea, 0x0e, 0x56,\n\t\t0x77, 0xed, 0x7a, 0x95, 0xc6, 0xa8, 0x59, 0x79, 0xd0, 0xbd, 0xbd, 0xca, 0xee, 0xa2, 0x73, 0x30,\n\t\t0xec, 0x4f, 0x9f, 0xe2, 0x87, 0x6e, 0xfc, 0x39, 0x20, 0xe7, 0x71, 0x7b, 0x56, 0x13, 0x3f, 0x24,\n\t\t0x85, 0xef, 0xa4, 0xe1, 0x54, 0x8c, 0x5e, 0x96, 0x47, 0x36, 0xe2, 0xa0, 0x59, 0xa4, 0xbb, 0x30,\n\t\t0x0b, 0x74, 0x02, 0x72, 0x5b, 0x8a, 0x8d, 0xbd, 0xd8, 0xc9, 0x9d, 0x96, 0xac, 0x73, 0xcb, 0x8d,\n\t\t0x98, 0xa6, 0x01, 0x0c, 0xdc, 0xf0, 0x1e, 0xf7, 0xba, 0x13, 0x6b, 0xe0, 0x86, 0xfb, 0x74, 0x01,\n\t\t0xd0, 0xb6, 0x69, 0xed, 0x32, 0xa4, 0x5e, 0x43, 0x62, 0xc6, 0x1d, 0x9a, 0xf3, 0x84, 0x62, 0xbd,\n\t\t0xcf, 0x3a, 0x13, 0xc7, 0x1c, 0xe7, 0xa8, 0xd8, 0xa6, 0xc1, 0x82, 0x63, 0x76, 0x85, 0x6e, 0x42,\n\t\t0xaf, 0xaa, 0xd4, 0x6d, 0xcc, 0xe2, 0xe0, 0x62, 0xec, 0xae, 0xa1, 0x55, 0x87, 0x4b, 0x76, 0x99,\n\t\t0x03, 0x0a, 0x9a, 0x0d, 0x2a, 0xe8, 0xbb, 0x69, 0x38, 0x19, 0xd9, 0xe8, 0xf3, 0xc8, 0xd6, 0x6a,\n\t\t0xc5, 0x1b, 0xa2, 0xbb, 0x48, 0x0b, 0x31, 0xfb, 0x90, 0x7c, 0x03, 0x6c, 0x73, 0xd9, 0x3d, 0x49,\n\t\t0x5c, 0x76, 0xbb, 0x65, 0xf4, 0x06, 0x2c, 0x23, 0xb0, 0xfc, 0x99, 0xf0, 0xe5, 0xef, 0x8b, 0xb5,\n\t\t0xfc, 0xfd, 0x82, 0xe5, 0xe7, 0x58, 0x61, 0x96, 0x6b, 0x85, 0xfe, 0x95, 0x84, 0xe0, 0x4a, 0x7e,\n\t\t0x25, 0x03, 0xa7, 0xe3, 0xb4, 0x48, 0xa1, 0x19, 0xc8, 0x35, 0xfb, 0x0c, 0xd8, 0x2a, 0x66, 0x65,\n\t\t0xf0, 0x6e, 0x95, 0x34, 0xe7, 0x4c, 0xde, 0x6a, 0x44, 0x70, 0x4c, 0x28, 0x15, 0x72, 0x26, 0x6f,\n\t\t0xbe, 0x92, 0x9e, 0xc9, 0x95, 0xb6, 0x2b, 0x47, 0xb1, 0x35, 0xb3, 0xaa, 0xe8, 0x06, 0xf3, 0x3c,\n\t\t0xec, 0xca, 0xbf, 0x95, 0xf4, 0x74, 0x79, 0x9a, 0xce, 0xc4, 0x3f, 0x4d, 0x6f, 0xc2, 0x84, 0xa7,\n\t\t0xa3, 0x9d, 0x3b, 0x50, 0x5f, 0xd4, 0x0e, 0x34, 0xe6, 0xf1, 0x06, 0x36, 0xa1, 0x80, 0x54, 0xb6,\n\t\t0xc1, 0x31, 0xa9, 0xfd, 0x09, 0xa4, 0xba, 0x87, 0x68, 0x26, 0x55, 0xbc, 0x55, 0x66, 0xbb, 0xda,\n\t\t0x2a, 0xd7, 0x61, 0x78, 0x07, 0x2b, 0x16, 0xd9, 0xc2, 0x4a, 0x0b, 0x1d, 0x44, 0x89, 0xca, 0x37,\n\t\t0x79, 0x5a, 0x72, 0xa2, 0x03, 0x9c, 0x5c, 0x74, 0x80, 0xd3, 0x71, 0xd4, 0x1c, 0xe8, 0xe6, 0xa8,\n\t\t0xd9, 0x3a, 0xb2, 0x1c, 0x89, 0x7d, 0x64, 0x29, 0xfc, 0x4d, 0x82, 0x42, 0x74, 0xbb, 0xde, 0x63,\n\t\t0x0b, 0x0d, 0xda, 0x83, 0x98, 0x1e, 0xff, 0x79, 0xf9, 0x25, 0x18, 0xa0, 0xe9, 0x06, 0xcf, 0xad,\n\t\t0xf5, 0xc6, 0x70, 0x6b, 0x39, 0x87, 0x83, 0x5d, 0x14, 0xfe, 0x20, 0xf9, 0x5d, 0xc1, 0x21, 0xc7,\n\t\t0xe5, 0xfc, 0x29, 0x4a, 0x25, 0xd8, 0x0d, 0xd2, 0x91, 0xb1, 0x4a, 0x8f, 0x7f, 0x32, 0x0b, 0xbf,\n\t\t0x97, 0xe0, 0x64, 0x74, 0x0f, 0x55, 0xb7, 0xe1, 0xfb, 0x87, 0x31, 0xa2, 0x9f, 0xa7, 0xe0, 0x54,\n\t\t0x8c, 0x4e, 0x44, 0x67, 0x4c, 0x1a, 0x26, 0x8a, 0x5e, 0xb1, 0x63, 0x2d, 0x92, 0x47, 0xfc, 0xc8,\n\t\t0xc6, 0x14, 0x8c, 0xaf, 0x7a, 0xba, 0x89, 0xaf, 0x0e, 0xac, 0xe2, 0x5f, 0x94, 0x60, 0x3e, 0x7e,\n\t\t0x03, 0x61, 0x9c, 0x3d, 0xef, 0x70, 0x0e, 0x70, 0xef, 0x48, 0x90, 0xb0, 0x55, 0x30, 0x1a, 0xdb,\n\t\t0xa8, 0x17, 0x25, 0xb1, 0x53, 0xb8, 0x1b, 0xf7, 0xc4, 0x41, 0x9c, 0x8e, 0x81, 0xf8, 0xad, 0x80,\n\t\t0x1e, 0x8a, 0x8a, 0x8a, 0xdd, 0xea, 0xe1, 0x3a, 0xcc, 0x56, 0x14, 0xd2, 0xd6, 0x32, 0x13, 0x6c,\n\t\t0x20, 0x69, 0xcd, 0xac, 0x4b, 0xc7, 0x5b, 0x4a, 0x37, 0xaa, 0xe2, 0xe8, 0x73, 0x3a, 0x81, 0x3e,\n\t\t0xf7, 0x44, 0xda, 0x68, 0x20, 0x0e, 0x2c, 0xbc, 0x27, 0xc1, 0x54, 0x48, 0x93, 0x2e, 0x9a, 0x80,\n\t\t0x7e, 0xb7, 0x39, 0xb1, 0xb9, 0x6e, 0x7d, 0xf4, 0xba, 0xa4, 0xa1, 0x0d, 0x38, 0xda, 0xdc, 0xc8,\n\t\t0xb7, 0x75, 0x2b, 0xc1, 0x91, 0x17, 0xb1, 0x7d, 0x7c, 0x5d, 0xb7, 0x70, 0x92, 0xed, 0x37, 0xce,\n\t\t0x62, 0x7f, 0x0c, 0x26, 0x84, 0xdd, 0xbf, 0x61, 0xa3, 0x89, 0x1d, 0xd2, 0x17, 0xde, 0x95, 0x60,\n\t\t0x3a, 0xac, 0xf1, 0xf3, 0x50, 0xde, 0x72, 0x58, 0xf3, 0x11, 0xea, 0xa0, 0x7f, 0x2c, 0xc1, 0x6c,\n\t\t0x54, 0x03, 0x69, 0xd8, 0x68, 0x1e, 0xa9, 0xd9, 0x86, 0x6f, 0x96, 0x59, 0x48, 0xd8, 0xa7, 0x84,\n\t\t0x16, 0x61, 0x94, 0xb6, 0x42, 0x05, 0xab, 0x06, 0xee, 0x98, 0x86, 0x0d, 0xdc, 0x08, 0xd4, 0x0c,\n\t\t0x3a, 0x0a, 0x77, 0xa9, 0xee, 0x0a, 0x77, 0x4f, 0x4a, 0x6b, 0xf1, 0x4b, 0x6b, 0x71, 0x74, 0xa7,\n\t\t0x2f, 0x86, 0xee, 0xdc, 0x86, 0x31, 0x56, 0x12, 0x61, 0x18, 0x75, 0x83, 0x60, 0x6b, 0x4f, 0xa9,\n\t\t0x44, 0x9f, 0x5b, 0x46, 0x19, 0x23, 0x85, 0x57, 0x62, 0x6c, 0xfe, 0xb2, 0x5d, 0xf6, 0x40, 0x65,\n\t\t0xbb, 0xb6, 0x10, 0x0e, 0x92, 0x84, 0x70, 0xe2, 0x1a, 0x5d, 0xae, 0xeb, 0x1a, 0x5d, 0xeb, 0x9c,\n\t\t0x31, 0x10, 0xbf, 0x34, 0xe2, 0x55, 0x8a, 0x8e, 0x1c, 0xa0, 0x52, 0x34, 0x78, 0xb0, 0x4a, 0x91,\n\t\t0xa0, 0x64, 0x31, 0xf4, 0x18, 0x4a, 0x16, 0xf9, 0x47, 0x52, 0xb2, 0x28, 0xfc, 0x55, 0x82, 0xc5,\n\t\t0xa4, 0xed, 0x9f, 0x4d, 0xff, 0x2b, 0xb5, 0xfb, 0xdf, 0xb0, 0x13, 0xdb, 0x16, 0x1c, 0x6b, 0xb6,\n\t\t0x8c, 0x04, 0xda, 0x08, 0x5c, 0xcf, 0x34, 0x1f, 0xda, 0x14, 0xe2, 0x6f, 0x24, 0x38, 0x8a, 0x79,\n\t\t0xb7, 0x03, 0xa7, 0xc2, 0x9e, 0x60, 0x16, 0xe7, 0xdb, 0x12, 0xa7, 0x02, 0x20, 0xda, 0x4a, 0xe3,\n\t\t0xf8, 0x03, 0x29, 0x86, 0x3f, 0x68, 0x0b, 0xed, 0x52, 0x09, 0x42, 0xbb, 0xc2, 0x07, 0x12, 0x1c,\n\t\t0x0f, 0xfd, 0xba, 0xc1, 0x89, 0x6d, 0xd9, 0xb7, 0x13, 0x86, 0x52, 0xf5, 0x56, 0x02, 0xdc, 0x5b,\n\t\t0xb7, 0x94, 0x2a, 0xee, 0xf6, 0xd5, 0x87, 0xb6, 0x8d, 0xb6, 0x4c, 0xbc, 0x27, 0x7e, 0x2a, 0xe1,\n\t\t0x67, 0xbc, 0x45, 0x12, 0x75, 0xf3, 0xcc, 0x40, 0x8e, 0xf5, 0x53, 0xb5, 0x4f, 0x81, 0x7b, 0x8b,\n\t\t0x4e, 0x41, 0x73, 0x17, 0x4b, 0xc5, 0xdf, 0xc5, 0xc2, 0xd2, 0xfa, 0x11, 0x1a, 0xf6, 0x25, 0x09,\n\t\t0xe6, 0x13, 0x34, 0xb8, 0xb5, 0xb2, 0xd3, 0x92, 0x2f, 0x3b, 0xdd, 0xed, 0xc2, 0x85, 0x20, 0x2f,\n\t\t0xfc, 0x32, 0x05, 0x2f, 0x1e, 0xac, 0xc9, 0xff, 0xd0, 0x4c, 0xa2, 0x95, 0xbb, 0x4c, 0xf9, 0x72,\n\t\t0x97, 0xf7, 0x00, 0x75, 0x36, 0x93, 0x31, 0xef, 0x70, 0x26, 0x5e, 0xb1, 0x5a, 0x1e, 0xee, 0xe8,\n\t\t0x08, 0x47, 0xe3, 0xd0, 0xa7, 0x9a, 0x06, 0xb1, 0xcc, 0x0a, 0x5d, 0xb0, 0x01, 0xd9, 0xbb, 0x44,\n\t\t0x45, 0x18, 0x09, 0xf4, 0x45, 0x9a, 0x46, 0xc5, 0x3d, 0xa9, 0xf4, 0xcb, 0xc3, 0xbe, 0x76, 0xc5,\n\t\t0xdb, 0x46, 0x65, 0xbf, 0xf0, 0x66, 0x1a, 0x6e, 0x1c, 0xe0, 0x23, 0x02, 0x74, 0xaf, 0xdd, 0x6b,\n\t\t0x0e, 0x0a, 0x3e, 0xd1, 0x89, 0x25, 0xd9, 0x97, 0xa5, 0x3f, 0xa4, 0xf3, 0xb5, 0x30, 0xa7, 0xcc,\n\t\t0x5f, 0x97, 0x9e, 0x83, 0xae, 0xcb, 0x02, 0xa0, 0x60, 0xeb, 0x26, 0xab, 0xf7, 0xa4, 0xe5, 0xbc,\n\t\t0xee, 0x53, 0x42, 0x37, 0xa5, 0xe7, 0xad, 0x62, 0xc6, 0xb7, 0x8a, 0x85, 0x3f, 0x4a, 0x70, 0xb5,\n\t\t0xcb, 0x2f, 0x20, 0x04, 0x18, 0x24, 0x01, 0x86, 0xc7, 0xab, 0xb8, 0x85, 0xcf, 0xa7, 0xe1, 0x6a,\n\t\t0x97, 0x5d, 0xaa, 0xff, 0xa9, 0xb6, 0x1a, 0x70, 0xe8, 0x3d, 0x62, 0x87, 0xde, 0x1b, 0xdf, 0xa1,\n\t\t0x0b, 0x55, 0x47, 0xe4, 0x00, 0xfa, 0x44, 0x0e, 0xe0, 0x73, 0x69, 0xb8, 0xdc, 0x4d, 0xa7, 0x6d,\n\t\t0x3c, 0xcb, 0x8f, 0x25, 0xf9, 0x89, 0xe5, 0xb7, 0x2c, 0xff, 0x7d, 0x09, 0x2e, 0x24, 0xed, 0x1a,\n\t\t0xfe, 0xb7, 0x36, 0x79, 0xf1, 0x5e, 0x55, 0xf8, 0xad, 0x04, 0xe7, 0x13, 0x75, 0x1a, 0x1f, 0x9a,\n\t\t0x0b, 0xe0, 0x9e, 0xa2, 0x52, 0x07, 0x3a, 0x45, 0x15, 0xbe, 0x97, 0x83, 0x4b, 0x5d, 0x7c, 0x32,\n\t\t0xd5, 0xb6, 0x1c, 0x92, 0x6f, 0x39, 0x66, 0x20, 0xd7, 0x5c, 0x0e, 0xa6, 0xf3, 0x59, 0x19, 0xbc,\n\t\t0x5b, 0xbc, 0x94, 0x4a, 0xfa, 0x10, 0x52, 0x2a, 0xdd, 0xd6, 0x57, 0x7b, 0x0f, 0x37, 0xa5, 0x92,\n\t\t0x79, 0xa4, 0x29, 0x95, 0xbe, 0xae, 0x53, 0x2a, 0xf7, 0x81, 0x35, 0x7c, 0x33, 0x89, 0xec, 0x18,\n\t\t0xdb, 0x1f, 0x72, 0x54, 0x76, 0xbb, 0xc6, 0xa9, 0x14, 0xef, 0xa8, 0x5c, 0x0b, 0xde, 0x6a, 0x37,\n\t\t0x92, 0xac, 0xdf, 0x9f, 0xc7, 0x51, 0x79, 0x88, 0xa1, 0xf2, 0x2a, 0x8c, 0xb7, 0xa9, 0x53, 0xd9,\n\t\t0xc2, 0xf5, 0x16, 0xfc, 0x1c, 0x85, 0x3f, 0x1f, 0xaa, 0x38, 0x25, 0x4d, 0x76, 0x58, 0xd8, 0x10,\n\t\t0x8e, 0x36, 0x78, 0xb7, 0x3b, 0xca, 0xb5, 0x47, 0xba, 0x29, 0xd7, 0x76, 0xb4, 0xee, 0x0e, 0x72,\n\t\t0x5a, 0x77, 0x5b, 0x07, 0xb1, 0xa1, 0xe4, 0xb9, 0x96, 0xfc, 0x01, 0x72, 0x2d, 0xc3, 0x07, 0xcb,\n\t\t0xb5, 0x5c, 0x87, 0x9c, 0x86, 0x2b, 0xca, 0xbe, 0xab, 0x9a, 0xd1, 0x2d, 0xc6, 0x40, 0xa9, 0xa9,\n\t\t0x2a, 0xa2, 0xe7, 0x61, 0xe0, 0xe3, 0x3a, 0x21, 0xde, 0xdf, 0x87, 0x34, 0x9b, 0x8b, 0x85, 0xcc,\n\t\t0x39, 0x97, 0xbc, 0xc9, 0xed, 0xf6, 0xe0, 0x5a, 0x75, 0xa3, 0xac, 0x10, 0xd6, 0x5e, 0x1c, 0xd6,\n\t\t0x7b, 0x0b, 0x94, 0x5e, 0xae, 0x1b, 0xcb, 0x44, 0x94, 0x23, 0x3a, 0xfa, 0x18, 0x72, 0x44, 0x63,\n\t\t0x8f, 0x26, 0x47, 0xf4, 0x46, 0x1a, 0x2e, 0x24, 0xfd, 0x40, 0xf4, 0xc3, 0x77, 0xd6, 0x1b, 0x5e,\n\t\t0xd4, 0xe5, 0xd6, 0x51, 0xaf, 0x24, 0xfe, 0xba, 0xd1, 0x17, 0x6c, 0xb5, 0xb9, 0x9d, 0x5e, 0xbf,\n\t\t0xdb, 0xe1, 0x87, 0x14, 0x19, 0x41, 0x48, 0x71, 0x48, 0x99, 0xe6, 0xc2, 0x6f, 0x52, 0xb0, 0x90,\n\t\t0xe4, 0xeb, 0x57, 0xe1, 0x7a, 0xf0, 0x63, 0x99, 0xd4, 0x41, 0x63, 0x99, 0xc3, 0x5a, 0x45, 0xfe,\n\t\t0xec, 0xf6, 0x08, 0x66, 0xb7, 0xe5, 0xeb, 0x7a, 0xe3, 0x27, 0x9d, 0x3e, 0x48, 0x41, 0xc2, 0xef,\n\t\t0x72, 0x3f, 0x1a, 0x93, 0xc9, 0x2b, 0x1a, 0xf6, 0x72, 0x8b, 0x86, 0xad, 0x6e, 0x97, 0x4c, 0xfc,\n\t\t0x6e, 0x97, 0xc2, 0x3f, 0x52, 0x70, 0xee, 0x30, 0x3c, 0xca, 0x47, 0x74, 0xd2, 0xdb, 0xea, 0x39,\n\t\t0x99, 0x04, 0xf5, 0x9c, 0xc2, 0x3f, 0x53, 0x70, 0x3e, 0xd1, 0x67, 0xd2, 0x4f, 0x26, 0xbe, 0x63,\n\t\t0xe2, 0xbd, 0x04, 0x6d, 0x26, 0x49, 0x52, 0xff, 0xd3, 0x69, 0xd1, 0xc4, 0x8b, 0x3a, 0x94, 0x9e,\n\t\t0x4c, 0x7c, 0x68, 0x83, 0x54, 0xa6, 0x9b, 0xef, 0x32, 0x7e, 0x91, 0x82, 0xc5, 0x84, 0x9f, 0xaf,\n\t\t0x3f, 0x59, 0x07, 0xdf, 0x3a, 0xcc, 0x13, 0x18, 0xa2, 0x3f, 0xd7, 0xf5, 0x0a, 0xc1, 0x16, 0x7d,\n\t\t0xd5, 0x71, 0x98, 0x58, 0xbb, 0xbf, 0x76, 0x6b, 0xb3, 0xbc, 0x5e, 0xda, 0xd8, 0x5c, 0x93, 0xcb,\n\t\t0x9b, 0xff, 0x77, 0x67, 0xad, 0x5c, 0xba, 0x75, 0x7f, 0x79, 0xa3, 0x74, 0x33, 0xff, 0x14, 0x9a,\n\t\t0x81, 0xa9, 0xce, 0xc7, 0xcb, 0x1b, 0x1b, 0x65, 0x7a, 0x37, 0x2f, 0xa1, 0x93, 0x70, 0xbc, 0x93,\n\t\t0x60, 0x75, 0xe3, 0xf6, 0xdd, 0x35, 0x46, 0x92, 0x5a, 0x79, 0x15, 0x8e, 0xa9, 0x66, 0x95, 0x37,\n\t\t0x07, 0x2b, 0xde, 0x1f, 0x20, 0xdf, 0x71, 0x22, 0xf3, 0x3b, 0xd2, 0xff, 0x5f, 0x7c, 0xa0, 0x93,\n\t\t0x9d, 0xfa, 0x56, 0x51, 0x35, 0xab, 0x8b, 0xed, 0x7f, 0xc4, 0x7c, 0x5e, 0xd7, 0x2a, 0x8b, 0x0f,\n\t\t0x4c, 0xf7, 0xcf, 0x9f, 0xd9, 0xbf, 0x32, 0xdf, 0x50, 0x6a, 0xfa, 0xde, 0xc5, 0xad, 0x0c, 0xbd,\n\t\t0x77, 0xe9, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0xf3, 0xc4, 0x77, 0x00, 0x78, 0x5a, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/tasklist.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xb4, 0x56, 0xdb, 0x6e, 0xdb, 0x36,\n\t\t0x18, 0x9e, 0x7c, 0x68, 0x9d, 0xdf, 0x4d, 0xa2, 0xb2, 0x49, 0x63, 0xbb, 0xed, 0xe6, 0xfa, 0xa2,\n\t\t0xc8, 0x8a, 0x4d, 0x46, 0xb2, 0x0d, 0x18, 0xb6, 0xa1, 0xab, 0x13, 0x1b, 0xad, 0x10, 0x27, 0x35,\n\t\t0x64, 0xb5, 0x43, 0x07, 0x0c, 0x02, 0x2d, 0xb1, 0x0e, 0x67, 0x49, 0x14, 0x44, 0xca, 0xae, 0x6f,\n\t\t0xf6, 0x18, 0xbb, 0xdb, 0x8b, 0xec, 0x1d, 0xf6, 0x4e, 0x03, 0x29, 0xd9, 0xf5, 0x41, 0x0d, 0xd6,\n\t\t0x8b, 0xdd, 0x99, 0xff, 0xc7, 0x8f, 0xdf, 0x7f, 0xb6, 0xa0, 0x95, 0x8c, 0x48, 0xdc, 0x76, 0xb1,\n\t\t0x47, 0x42, 0x97, 0xb4, 0x71, 0x44, 0xdb, 0xd3, 0x93, 0xb6, 0xc0, 0x7c, 0xe2, 0x53, 0x2e, 0x8c,\n\t\t0x28, 0x66, 0x82, 0xa1, 0x7b, 0xf2, 0x8e, 0x91, 0xdd, 0x31, 0x70, 0x44, 0x8d, 0xe9, 0x49, 0xe3,\n\t\t0xf3, 0x31, 0x63, 0x63, 0x9f, 0xb4, 0xd5, 0x95, 0x51, 0xf2, 0xae, 0xed, 0x25, 0x31, 0x16, 0x94,\n\t\t0x85, 0x29, 0xa9, 0xf1, 0xc5, 0x26, 0x2e, 0x68, 0x40, 0xb8, 0xc0, 0x41, 0x94, 0x5d, 0xd8, 0x7a,\n\t\t0x60, 0x16, 0xe3, 0x28, 0x22, 0x31, 0x4f, 0xf1, 0xd6, 0x6b, 0xa8, 0xd8, 0x98, 0x4f, 0xfa, 0x94,\n\t\t0x0b, 0x84, 0xa0, 0x14, 0xe2, 0x80, 0xd4, 0xb4, 0xa6, 0x76, 0xbc, 0x63, 0xa9, 0xdf, 0xe8, 0x3b,\n\t\t0x28, 0x4d, 0x68, 0xe8, 0xd5, 0x0a, 0x4d, 0xed, 0x78, 0xef, 0xf4, 0xb1, 0x91, 0xe3, 0xa4, 0xb1,\n\t\t0x78, 0xe0, 0x82, 0x86, 0x9e, 0xa5, 0xae, 0xb7, 0x30, 0xe8, 0x0b, 0xeb, 0x25, 0x11, 0xd8, 0xc3,\n\t\t0x02, 0xa3, 0x4b, 0x38, 0x08, 0xf0, 0x7b, 0x47, 0x86, 0xcd, 0x9d, 0x88, 0xc4, 0x0e, 0x27, 0x2e,\n\t\t0x0b, 0x3d, 0x25, 0x57, 0x3d, 0x7d, 0x68, 0xa4, 0x9e, 0x1a, 0x0b, 0x4f, 0x8d, 0x2e, 0x4b, 0x46,\n\t\t0x3e, 0x79, 0x83, 0xfd, 0x84, 0x58, 0x77, 0x03, 0xfc, 0x5e, 0x3e, 0xc8, 0x07, 0x24, 0x1e, 0x2a,\n\t\t0x5a, 0xeb, 0x35, 0xd4, 0x17, 0x12, 0x03, 0x1c, 0x0b, 0x2a, 0xb3, 0xb2, 0xd4, 0xd2, 0xa1, 0x38,\n\t\t0x21, 0xf3, 0x2c, 0x12, 0xf9, 0x13, 0x3d, 0x81, 0x7d, 0x36, 0x0b, 0x49, 0xec, 0x5c, 0x33, 0x2e,\n\t\t0x1c, 0x15, 0x67, 0x41, 0xa1, 0xbb, 0xca, 0xfc, 0x92, 0x71, 0x71, 0x85, 0x03, 0xd2, 0x9a, 0xc0,\n\t\t0xa1, 0xc9, 0x99, 0xaf, 0x92, 0xfc, 0x22, 0x66, 0x49, 0x74, 0x49, 0x44, 0x4c, 0x5d, 0x8e, 0xda,\n\t\t0x70, 0x10, 0x92, 0x59, 0xbe, 0xfb, 0x9a, 0x75, 0x37, 0x24, 0xb3, 0x75, 0x07, 0xd1, 0x63, 0xb8,\n\t\t0x13, 0x31, 0xdf, 0x27, 0xb1, 0xe3, 0xb2, 0x24, 0x14, 0x4a, 0xae, 0x68, 0x55, 0x53, 0xdb, 0xb9,\n\t\t0x34, 0xb5, 0xfe, 0x2a, 0xc1, 0xde, 0x22, 0x88, 0xa1, 0xc0, 0x22, 0xe1, 0xe8, 0x2b, 0x40, 0x23,\n\t\t0xec, 0x4e, 0x7c, 0x36, 0x4e, 0x69, 0xce, 0x35, 0x0d, 0x85, 0x12, 0x29, 0x5a, 0x7a, 0x86, 0x28,\n\t\t0xf2, 0x4b, 0x1a, 0x0a, 0xf4, 0x08, 0x20, 0x26, 0xd8, 0x73, 0x7c, 0x32, 0x25, 0x7e, 0xa6, 0xb0,\n\t\t0x23, 0x2d, 0x7d, 0x69, 0x40, 0x0f, 0x60, 0x07, 0xbb, 0x93, 0x0c, 0x2d, 0x2a, 0xb4, 0x82, 0xdd,\n\t\t0x49, 0x0a, 0x3e, 0x81, 0xfd, 0x18, 0x0b, 0xb2, 0x1a, 0x4b, 0x49, 0xc5, 0xb2, 0x2b, 0xcd, 0x1f,\n\t\t0xe2, 0xe8, 0xc2, 0xae, 0x0c, 0xda, 0xa1, 0x9e, 0x33, 0xf2, 0x99, 0x3b, 0xa9, 0x95, 0x55, 0xc1,\n\t\t0x9a, 0x1f, 0xed, 0x05, 0xb3, 0x7b, 0x26, 0xef, 0x59, 0x55, 0x49, 0x33, 0x3d, 0x75, 0x40, 0x53,\n\t\t0x38, 0xa2, 0x8b, 0xbc, 0x3a, 0x63, 0x99, 0x58, 0x27, 0x48, 0x33, 0x5b, 0xbb, 0xd5, 0x2c, 0x1e,\n\t\t0x57, 0x4f, 0x9f, 0xdd, 0xd8, 0x5b, 0x69, 0x76, 0x8c, 0xdc, 0xd2, 0xf4, 0x42, 0x11, 0xcf, 0xad,\n\t\t0x43, 0xfa, 0x49, 0x65, 0xbb, 0xfd, 0xb1, 0xb2, 0x1d, 0x40, 0x99, 0x04, 0x91, 0x98, 0xd7, 0x2a,\n\t\t0x4d, 0xed, 0xb8, 0x62, 0xa5, 0x87, 0x86, 0x80, 0xc6, 0xc7, 0xb5, 0x73, 0xda, 0xed, 0x39, 0x94,\n\t\t0xa7, 0xb2, 0x73, 0x55, 0x4d, 0xaa, 0xa7, 0x4f, 0x73, 0x83, 0xcb, 0x7d, 0xd1, 0x4a, 0x89, 0x3f,\n\t\t0x14, 0xbe, 0xd7, 0x5a, 0x3f, 0x43, 0x75, 0x25, 0xa1, 0xa8, 0x0e, 0x15, 0x2e, 0x70, 0x2c, 0x1c,\n\t\t0xea, 0x65, 0x1d, 0x71, 0x5b, 0x9d, 0x4d, 0x0f, 0x1d, 0xc2, 0x2d, 0x12, 0x7a, 0x12, 0x48, 0x9b,\n\t\t0xa0, 0x4c, 0x42, 0xcf, 0xf4, 0x5a, 0x7f, 0x6a, 0x00, 0x03, 0xd5, 0x70, 0x66, 0xf8, 0x8e, 0xa1,\n\t\t0x2e, 0xe8, 0x3e, 0xe6, 0xc2, 0xc1, 0xae, 0x4b, 0x38, 0x77, 0xe4, 0xb2, 0xc8, 0xc6, 0xaf, 0xb1,\n\t\t0x35, 0x7e, 0xf6, 0x62, 0x93, 0x58, 0x7b, 0x92, 0xd3, 0x51, 0x14, 0x69, 0x44, 0x0d, 0xa8, 0x50,\n\t\t0x8f, 0x84, 0x82, 0x8a, 0x79, 0x36, 0x43, 0xcb, 0x73, 0x5e, 0x53, 0x15, 0x73, 0x9a, 0xaa, 0xf5,\n\t\t0xb7, 0x06, 0xf5, 0xa1, 0xa0, 0xee, 0x64, 0xde, 0x7b, 0x4f, 0xdc, 0x44, 0x26, 0xa1, 0x23, 0x44,\n\t\t0x4c, 0x47, 0x89, 0x20, 0x1c, 0xbd, 0x00, 0x7d, 0xc6, 0xe2, 0x09, 0x89, 0x55, 0xdd, 0x1c, 0xb9,\n\t\t0x25, 0x33, 0x3f, 0x1f, 0xdd, 0xd8, 0x25, 0xd6, 0x5e, 0x4a, 0x5b, 0xae, 0x34, 0x1b, 0xea, 0xdc,\n\t\t0xbd, 0x26, 0x5e, 0xe2, 0x13, 0x47, 0x30, 0x27, 0xcd, 0x9e, 0x0c, 0x9b, 0x25, 0x22, 0x2b, 0x4d,\n\t\t0x7d, 0x7b, 0xf1, 0x64, 0x3b, 0xd6, 0xba, 0xbf, 0xe0, 0xda, 0x6c, 0x28, 0x99, 0x76, 0x4a, 0x6c,\n\t\t0x3d, 0x83, 0xbb, 0x5b, 0xab, 0x07, 0x7d, 0x09, 0xfa, 0x46, 0x83, 0xf3, 0x9a, 0xd6, 0x2c, 0x1e,\n\t\t0xef, 0x58, 0xfb, 0xeb, 0x9d, 0xc9, 0x5b, 0xff, 0x94, 0xe0, 0x68, 0xeb, 0x81, 0x73, 0x16, 0xbe,\n\t\t0xa3, 0x63, 0x54, 0x83, 0xdb, 0x53, 0x12, 0x73, 0xca, 0xc2, 0x45, 0x89, 0xb3, 0x23, 0x3a, 0x85,\n\t\t0x7b, 0x61, 0x12, 0x38, 0x6a, 0xde, 0xa3, 0x05, 0x8b, 0xab, 0x28, 0xca, 0x67, 0x85, 0x9a, 0x6c,\n\t\t0xe6, 0x24, 0xb0, 0x08, 0xf6, 0x96, 0x4f, 0x72, 0xf4, 0x2d, 0x1c, 0x48, 0xce, 0x2c, 0xa6, 0xb2,\n\t\t0x26, 0x1f, 0x48, 0xc5, 0x25, 0x09, 0x85, 0x49, 0xf0, 0x8b, 0x84, 0x57, 0x58, 0x14, 0xf6, 0x37,\n\t\t0x55, 0x4a, 0x6a, 0x46, 0x9f, 0xdf, 0x98, 0xfd, 0x8d, 0x50, 0x8c, 0x75, 0x5f, 0xd2, 0x29, 0xdd,\n\t\t0x8b, 0xd7, 0x1d, 0xf4, 0x41, 0xdf, 0x72, 0xae, 0xac, 0xb4, 0x3a, 0x9f, 0xa4, 0xb5, 0x11, 0x42,\n\t\t0x2a, 0xb6, 0x3f, 0x5b, 0xb7, 0x36, 0x28, 0xdc, 0xcb, 0x71, 0x6a, 0x75, 0x7c, 0xcb, 0xe9, 0xf8,\n\t\t0xfe, 0xb4, 0x3e, 0xbe, 0x4f, 0xfe, 0x9b, 0x2f, 0x2b, 0xa3, 0xdb, 0xf8, 0x1d, 0x0e, 0xf2, 0x7c,\n\t\t0xfa, 0x3f, 0xb4, 0x9e, 0xfe, 0x01, 0x77, 0x56, 0xff, 0x83, 0x51, 0x03, 0xee, 0xdb, 0x9d, 0xe1,\n\t\t0x85, 0xd3, 0x37, 0x87, 0xb6, 0x73, 0x61, 0x5e, 0x75, 0x1d, 0xf3, 0xea, 0x4d, 0xa7, 0x6f, 0x76,\n\t\t0xf5, 0xcf, 0x50, 0x1d, 0x0e, 0x37, 0xb0, 0xab, 0x57, 0xd6, 0x65, 0xa7, 0xaf, 0x6b, 0x39, 0xd0,\n\t\t0xd0, 0x36, 0xcf, 0x2f, 0xde, 0xea, 0x05, 0xf4, 0x10, 0x6a, 0x1b, 0x50, 0x6f, 0xf0, 0xb2, 0x77,\n\t\t0xd9, 0xb3, 0x3a, 0x7d, 0xbd, 0xf8, 0xd4, 0xfb, 0xa0, 0x6f, 0xcf, 0x23, 0xb2, 0xae, 0x6f, 0xbf,\n\t\t0x1d, 0xf4, 0x56, 0xf4, 0x1f, 0xc0, 0xd1, 0x06, 0xd6, 0xed, 0x9d, 0x9b, 0x43, 0xf3, 0xd5, 0x95,\n\t\t0xae, 0xe5, 0x80, 0x9d, 0x73, 0xdb, 0x7c, 0x63, 0xda, 0x6f, 0xf5, 0xc2, 0xd9, 0x6f, 0x70, 0xe4,\n\t\t0xb2, 0x20, 0x2f, 0x3b, 0x67, 0xbb, 0xcb, 0xf4, 0xc8, 0x19, 0x1e, 0x68, 0xbf, 0x9e, 0x8c, 0xa9,\n\t\t0xb8, 0x4e, 0x46, 0x86, 0xcb, 0x82, 0xf6, 0xea, 0xb7, 0xd7, 0xd7, 0xd4, 0xf3, 0xdb, 0x63, 0x96,\n\t\t0x7e, 0x0e, 0x65, 0x1f, 0x62, 0x3f, 0xe2, 0x88, 0x4e, 0x4f, 0x46, 0xb7, 0x94, 0xed, 0x9b, 0x7f,\n\t\t0x03, 0x00, 0x00, 0xff, 0xff, 0xea, 0xcc, 0x39, 0x04, 0xac, 0x09, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xc4, 0x5a, 0xcd, 0x72, 0xdb, 0xc8,\n\t\t0xb5, 0xbe, 0x20, 0x25, 0x59, 0x3a, 0xa4, 0x24, 0xa8, 0x25, 0x59, 0xf4, 0xcf, 0xd8, 0xb2, 0x66,\n\t\t0xec, 0x91, 0x79, 0xc7, 0xd2, 0xc8, 0xf3, 0x3f, 0xbe, 0x73, 0x1d, 0x08, 0x84, 0x6c, 0xd8, 0x34,\n\t\t0xc8, 0x34, 0x41, 0x6b, 0x34, 0x95, 0x04, 0x05, 0x91, 0x2d, 0x09, 0x31, 0x09, 0xb0, 0x80, 0xa6,\n\t\t0x6d, 0xed, 0x53, 0x95, 0x6c, 0x93, 0xd5, 0x54, 0x56, 0x79, 0x80, 0x54, 0xa5, 0x52, 0x59, 0x64,\n\t\t0x95, 0xca, 0x13, 0x64, 0x9b, 0x57, 0x48, 0xe5, 0x2d, 0x52, 0xdd, 0x68, 0x90, 0x00, 0x09, 0x12,\n\t\t0x74, 0x52, 0x35, 0xd9, 0x09, 0xa7, 0xbf, 0xef, 0xf4, 0xe9, 0xf3, 0xd7, 0x07, 0x10, 0x61, 0xa7,\n\t\t0x7f, 0x4a, 0xfc, 0xfd, 0x96, 0xdd, 0x26, 0x6e, 0x8b, 0xec, 0xdb, 0x3d, 0x67, 0xff, 0xf5, 0xc1,\n\t\t0xfe, 0x1b, 0xcf, 0x7f, 0x75, 0xd6, 0xf1, 0xde, 0xec, 0xf5, 0x7c, 0x8f, 0x7a, 0x68, 0x9d, 0x61,\n\t\t0xf6, 0x04, 0x66, 0xcf, 0xee, 0x39, 0x7b, 0xaf, 0x0f, 0xae, 0xdf, 0x3a, 0xf7, 0xbc, 0xf3, 0x0e,\n\t\t0xd9, 0xe7, 0x90, 0xd3, 0xfe, 0xd9, 0x7e, 0xbb, 0xef, 0xdb, 0xd4, 0xf1, 0xdc, 0x90, 0x74, 0xfd,\n\t\t0xf6, 0xe8, 0x3a, 0x75, 0xba, 0x24, 0xa0, 0x76, 0xb7, 0x27, 0x00, 0xdb, 0x69, 0x3b, 0xb7, 0xbc,\n\t\t0x6e, 0x77, 0xa0, 0x22, 0xd5, 0x36, 0x6a, 0x07, 0xaf, 0x3a, 0x4e, 0x40, 0x43, 0xcc, 0xce, 0xf7,\n\t\t0x45, 0xd8, 0x3c, 0x16, 0xe6, 0x6a, 0x6f, 0x49, 0xab, 0xcf, 0x4c, 0xd0, 0xdd, 0x33, 0x0f, 0x35,\n\t\t0x01, 0x45, 0xe7, 0xb0, 0x48, 0xb4, 0x52, 0x92, 0xb6, 0xa5, 0xdd, 0xc2, 0xc3, 0x7b, 0x7b, 0x29,\n\t\t0x47, 0xda, 0x1b, 0xd3, 0x83, 0xd7, 0xde, 0x8c, 0x8a, 0xd0, 0x67, 0x30, 0x47, 0x2f, 0x7b, 0xa4,\n\t\t0x94, 0xe3, 0x8a, 0xee, 0x4c, 0x55, 0x64, 0x5e, 0xf6, 0x08, 0xe6, 0x70, 0xf4, 0x15, 0x40, 0x40,\n\t\t0x6d, 0x9f, 0x5a, 0xcc, 0x0d, 0xa5, 0x3c, 0x27, 0x5f, 0xdf, 0x0b, 0x7d, 0xb4, 0x17, 0xf9, 0x68,\n\t\t0xcf, 0x8c, 0x7c, 0x84, 0x97, 0x38, 0x9a, 0x3d, 0x33, 0x6a, 0xab, 0xe3, 0x05, 0x24, 0xa4, 0xce,\n\t\t0x65, 0x53, 0x39, 0x9a, 0x53, 0x4d, 0x28, 0x86, 0xd4, 0x80, 0xda, 0xb4, 0x1f, 0x94, 0xe6, 0xb7,\n\t\t0xa5, 0xdd, 0x95, 0x87, 0x07, 0xb3, 0x9d, 0x5e, 0x65, 0xcc, 0x06, 0x27, 0xe2, 0x42, 0x6b, 0xf8,\n\t\t0x80, 0xee, 0xc2, 0xca, 0x85, 0x13, 0x50, 0xcf, 0xbf, 0xb4, 0x3a, 0xc4, 0x3d, 0xa7, 0x17, 0xa5,\n\t\t0x85, 0x6d, 0x69, 0x37, 0x8f, 0x97, 0x85, 0xb4, 0xca, 0x85, 0xe8, 0x27, 0xb0, 0xd9, 0xb3, 0x7d,\n\t\t0xe2, 0xd2, 0xa1, 0xfb, 0x2d, 0xc7, 0x3d, 0xf3, 0x4a, 0x57, 0xf8, 0x11, 0x76, 0x53, 0xad, 0xa8,\n\t\t0x73, 0x46, 0x22, 0x92, 0x78, 0xbd, 0x37, 0x2e, 0x44, 0x0a, 0xac, 0x0c, 0xd5, 0x72, 0xcf, 0x2c,\n\t\t0x66, 0x7a, 0x66, 0x79, 0xc0, 0xe0, 0xde, 0x79, 0x00, 0x73, 0x5d, 0xd2, 0xf5, 0x4a, 0x4b, 0x9c,\n\t\t0x78, 0x2d, 0xd5, 0x9e, 0x17, 0xa4, 0xeb, 0x61, 0x0e, 0x43, 0x18, 0xd6, 0x02, 0x62, 0xfb, 0xad,\n\t\t0x0b, 0xcb, 0xa6, 0xd4, 0x77, 0x4e, 0xfb, 0x94, 0x04, 0x25, 0xe0, 0xdc, 0xbb, 0xa9, 0xdc, 0x06,\n\t\t0x47, 0x2b, 0x03, 0x30, 0x96, 0x83, 0x11, 0x09, 0xaa, 0xc2, 0x9a, 0xdd, 0xa7, 0x9e, 0xe5, 0x93,\n\t\t0x80, 0x50, 0xab, 0xe7, 0x39, 0x2e, 0x0d, 0x4a, 0x05, 0xae, 0x73, 0x3b, 0x55, 0x27, 0x66, 0xc0,\n\t\t0x3a, 0xc7, 0xe1, 0x55, 0x46, 0x8d, 0x09, 0xd0, 0x0d, 0x58, 0x62, 0xe5, 0x61, 0xb1, 0xfa, 0x28,\n\t\t0x15, 0xb7, 0xa5, 0xdd, 0x25, 0xbc, 0xc8, 0x04, 0x55, 0x27, 0xa0, 0x48, 0x85, 0x95, 0xc1, 0x62,\n\t\t0x18, 0x87, 0x35, 0xbe, 0xcf, 0x7b, 0xa9, 0xfb, 0x98, 0x82, 0x86, 0x8b, 0x91, 0x02, 0xee, 0xf5,\n\t\t0x2d, 0xb8, 0xe2, 0x04, 0x56, 0xcb, 0xf7, 0xdc, 0xd2, 0xf2, 0xb6, 0xb4, 0xbb, 0x88, 0x17, 0x9c,\n\t\t0x40, 0xf5, 0x3d, 0x17, 0x3d, 0x82, 0x42, 0xbf, 0xd7, 0xb6, 0xa9, 0xc8, 0xd2, 0x95, 0xcc, 0x58,\n\t\t0x40, 0x08, 0xe7, 0x81, 0xf8, 0x39, 0xc8, 0x3d, 0xdb, 0xa7, 0x0e, 0x8f, 0x65, 0xcb, 0x73, 0xcf,\n\t\t0x9c, 0xf3, 0xd2, 0xea, 0x76, 0x7e, 0xb7, 0xf0, 0xf0, 0xf1, 0x6c, 0xa9, 0xca, 0x6c, 0x63, 0xa9,\n\t\t0x13, 0xaa, 0x50, 0xb9, 0x06, 0xcd, 0xa5, 0xfe, 0x25, 0x5e, 0xed, 0x25, 0xa5, 0xe8, 0x25, 0xac,\n\t\t0x33, 0xf3, 0x2d, 0xef, 0x35, 0xf1, 0x3b, 0x76, 0xcf, 0xea, 0x79, 0x1d, 0xa7, 0x75, 0x59, 0x92,\n\t\t0x79, 0x65, 0xa4, 0xf7, 0x05, 0x76, 0xc0, 0x5a, 0x08, 0xaf, 0x73, 0x34, 0x5e, 0x6b, 0x8d, 0x8a,\n\t\t0xd0, 0x5b, 0xb8, 0x6d, 0xb7, 0xa8, 0xf3, 0x9a, 0x58, 0xad, 0x4e, 0x3f, 0xa0, 0xc4, 0xb7, 0x02,\n\t\t0xd2, 0x21, 0x2d, 0x7e, 0x24, 0xb1, 0x07, 0xe2, 0x4e, 0x49, 0xaf, 0x3e, 0x85, 0x73, 0xd5, 0x90,\n\t\t0xda, 0x88, 0x98, 0x62, 0xbb, 0x9b, 0xf6, 0x94, 0x55, 0xf4, 0x3e, 0x2c, 0xf3, 0x13, 0x05, 0xad,\n\t\t0x0b, 0xd2, 0xee, 0x77, 0x48, 0x69, 0x9d, 0x47, 0xbe, 0xc8, 0x84, 0x0d, 0x21, 0x43, 0xc7, 0x20,\n\t\t0x0f, 0xcb, 0x45, 0x74, 0x83, 0x0d, 0x7e, 0xe6, 0x8f, 0x66, 0x73, 0xb1, 0x68, 0x04, 0xab, 0x24,\n\t\t0x29, 0x40, 0x26, 0x94, 0xa2, 0x8d, 0xdb, 0xd6, 0x48, 0x45, 0x6e, 0x66, 0x66, 0xc1, 0xd5, 0x01,\n\t\t0x57, 0x8b, 0x97, 0xe6, 0xf5, 0x43, 0xd8, 0x48, 0x0b, 0x27, 0x92, 0x21, 0xff, 0x8a, 0x5c, 0xf2,\n\t\t0x2e, 0xbe, 0x84, 0xd9, 0x9f, 0x68, 0x03, 0xe6, 0x5f, 0xdb, 0x9d, 0x7e, 0xd8, 0x90, 0x97, 0x70,\n\t\t0xf8, 0xf0, 0x75, 0xee, 0x4b, 0x69, 0xe7, 0xfb, 0x1c, 0xdc, 0x1a, 0x6f, 0x6a, 0x5c, 0x99, 0xb8,\n\t\t0xaa, 0xd0, 0xd7, 0xf1, 0x82, 0x91, 0x66, 0x29, 0x87, 0x61, 0x3d, 0xd9, 0xb0, 0x9d, 0xf0, 0x28,\n\t\t0xeb, 0xed, 0x9e, 0x35, 0xec, 0xd4, 0x5e, 0x9f, 0x8a, 0x4b, 0xe2, 0xda, 0x98, 0x03, 0x2a, 0xc2,\n\t\t0x00, 0x7c, 0x33, 0xee, 0x4e, 0x9f, 0x9a, 0x9e, 0x1a, 0xf5, 0x6e, 0xaf, 0x4f, 0xd1, 0x31, 0xdc,\n\t\t0xe0, 0xe6, 0x4d, 0xd0, 0x9e, 0xcf, 0xd2, 0xbe, 0xc5, 0xd8, 0x29, 0x8a, 0x77, 0xfe, 0x26, 0xc1,\n\t\t0x7a, 0x4a, 0xa7, 0x65, 0x0d, 0xa4, 0xed, 0x75, 0x6d, 0xc7, 0xb5, 0x9c, 0xb6, 0x70, 0xf2, 0x62,\n\t\t0x28, 0xd0, 0xdb, 0xe8, 0x36, 0x14, 0xc4, 0xa2, 0x6b, 0x77, 0x23, 0x7f, 0x43, 0x28, 0x32, 0xec,\n\t\t0x2e, 0x99, 0x70, 0xe3, 0xe6, 0xff, 0xd3, 0x1b, 0xf7, 0x0e, 0x14, 0x1d, 0xd7, 0xa1, 0x8e, 0x4d,\n\t\t0x49, 0x9b, 0xd9, 0x35, 0xc7, 0x2f, 0x9b, 0xc2, 0x40, 0xa6, 0xb7, 0x77, 0x7e, 0x2d, 0xc1, 0xa6,\n\t\t0xf6, 0x96, 0x12, 0xdf, 0xb5, 0x3b, 0x3f, 0xc8, 0x14, 0x30, 0x6a, 0x53, 0x6e, 0xdc, 0xa6, 0x3f,\n\t\t0x2f, 0xc0, 0x7a, 0x9d, 0xb8, 0x6d, 0xc7, 0x3d, 0xe7, 0xc5, 0xed, 0xd0, 0x4b, 0x6e, 0xd1, 0x6d,\n\t\t0x28, 0xd8, 0xe2, 0x79, 0xe8, 0x65, 0x88, 0x44, 0x7a, 0x1b, 0x1d, 0xc1, 0xf2, 0x00, 0x90, 0x39,\n\t\t0x6a, 0x44, 0xaa, 0xf9, 0xa8, 0x51, 0xb4, 0x63, 0x4f, 0xe8, 0x31, 0xcc, 0xb3, 0x42, 0x0f, 0xa7,\n\t\t0x8d, 0x95, 0x87, 0xf7, 0xd3, 0xef, 0xdb, 0xa4, 0x85, 0xac, 0xa8, 0x09, 0x0e, 0x79, 0x48, 0x87,\n\t\t0xb5, 0x0b, 0x62, 0xfb, 0xf4, 0x94, 0xd8, 0xd4, 0x6a, 0x13, 0x6a, 0x3b, 0x9d, 0x40, 0xcc, 0x1f,\n\t\t0x37, 0x27, 0x5c, 0xde, 0x97, 0x1d, 0xcf, 0x6e, 0x63, 0x79, 0x40, 0xab, 0x84, 0x2c, 0xf4, 0x0c,\n\t\t0xd6, 0x3b, 0x76, 0x40, 0xad, 0xa1, 0x3e, 0xde, 0x20, 0xe6, 0x33, 0x1b, 0xc4, 0x1a, 0xa3, 0x3d,\n\t\t0x8d, 0x58, 0xfc, 0xb6, 0x38, 0x02, 0x2e, 0x0c, 0xab, 0x82, 0xb4, 0x43, 0x4d, 0x0b, 0x99, 0x9a,\n\t\t0x56, 0x19, 0xa9, 0x11, 0x72, 0xb8, 0x9e, 0x12, 0x5c, 0xb1, 0x29, 0x25, 0xdd, 0x1e, 0xe5, 0x13,\n\t\t0xc9, 0x3c, 0x8e, 0x1e, 0xd1, 0x7d, 0x90, 0xbb, 0xf6, 0x5b, 0xa7, 0xdb, 0xef, 0x5a, 0x42, 0x14,\n\t\t0xf0, 0xe9, 0x62, 0x1e, 0xaf, 0x0a, 0xb9, 0x22, 0xc4, 0x6c, 0x0c, 0x19, 0xb6, 0x3f, 0x6e, 0xc9,\n\t\t0x52, 0xf6, 0x18, 0x32, 0x60, 0x70, 0x3b, 0x54, 0x58, 0x25, 0x6f, 0x7b, 0x4e, 0x58, 0xb3, 0xa1,\n\t\t0x0e, 0xc8, 0xd4, 0xb1, 0x32, 0xa4, 0x70, 0x25, 0x8f, 0xa1, 0xc8, 0x9d, 0x72, 0x66, 0x3b, 0x9d,\n\t\t0xbe, 0x4f, 0xc4, 0x0c, 0x91, 0x1e, 0xa6, 0xa3, 0x10, 0x83, 0x0b, 0x8c, 0x21, 0x1e, 0xd0, 0xc7,\n\t\t0xb0, 0xc1, 0x15, 0xb0, 0x5c, 0x27, 0xbe, 0xe5, 0xb4, 0x89, 0x4b, 0x1d, 0x7a, 0x29, 0xc6, 0x08,\n\t\t0xc4, 0xd6, 0x8e, 0xf9, 0x92, 0x2e, 0x56, 0xd0, 0xe7, 0xb0, 0x15, 0x85, 0x60, 0x94, 0xb4, 0xcc,\n\t\t0x49, 0x9b, 0x62, 0x79, 0x84, 0x77, 0x1b, 0x0a, 0x91, 0x03, 0x58, 0x01, 0xac, 0xf0, 0xd2, 0x81,\n\t\t0x48, 0xa4, 0xb7, 0x77, 0xfe, 0x94, 0x83, 0x6b, 0x22, 0x2f, 0xd5, 0x0b, 0xa7, 0xd3, 0xfe, 0x41,\n\t\t0x2a, 0xfa, 0xa3, 0x98, 0x5a, 0x56, 0x75, 0xf1, 0x26, 0x27, 0xbf, 0x89, 0x0d, 0xf4, 0xbc, 0xd5,\n\t\t0x8d, 0xd6, 0x7f, 0x7e, 0xac, 0xfe, 0xd9, 0xa0, 0x21, 0xc6, 0xdf, 0xb0, 0x6b, 0x8b, 0x21, 0x60,\n\t\t0x6e, 0xca, 0xa0, 0x11, 0xb6, 0x64, 0xde, 0xa9, 0xa3, 0x41, 0xa3, 0x37, 0x2a, 0x42, 0x57, 0x61,\n\t\t0x21, 0xec, 0xb9, 0xbc, 0x7a, 0x96, 0xb0, 0x78, 0xda, 0xf9, 0x47, 0x6e, 0xd0, 0x6f, 0x2a, 0xa4,\n\t\t0xe5, 0x04, 0x91, 0xbf, 0x06, 0x6d, 0x40, 0xca, 0x6e, 0x03, 0x11, 0x31, 0xd1, 0x06, 0xc6, 0x53,\n\t\t0x3c, 0xf7, 0xae, 0x29, 0xfe, 0x0d, 0x14, 0x13, 0xd5, 0x9a, 0xfd, 0xfe, 0x53, 0x08, 0xd2, 0x2b,\n\t\t0x75, 0x2e, 0x59, 0xa9, 0x18, 0xb6, 0x3c, 0xdf, 0x39, 0x77, 0x5c, 0xbb, 0x63, 0x8d, 0x18, 0x99,\n\t\t0xdd, 0x5b, 0x36, 0x23, 0x6a, 0x23, 0x61, 0xec, 0x48, 0x7e, 0x2e, 0x8c, 0xe5, 0xe7, 0x5f, 0x72,\n\t\t0x70, 0x2d, 0x6a, 0x98, 0x55, 0xaf, 0x65, 0x77, 0x2a, 0x4e, 0xd0, 0xb3, 0x69, 0xeb, 0x62, 0xb6,\n\t\t0xfe, 0xfe, 0xdf, 0xf7, 0xe7, 0xcf, 0xe0, 0x56, 0xd2, 0x02, 0xcb, 0x3b, 0xb3, 0xe8, 0x85, 0x13,\n\t\t0x58, 0x71, 0x37, 0x4f, 0x57, 0x78, 0x3d, 0x61, 0x51, 0xed, 0xcc, 0xbc, 0x70, 0x02, 0xd1, 0x15,\n\t\t0xd1, 0x7b, 0x00, 0x7c, 0x6e, 0xa1, 0xde, 0x2b, 0x12, 0xa6, 0x69, 0x11, 0xf3, 0x41, 0xcb, 0x64,\n\t\t0x82, 0x9d, 0x67, 0x50, 0x88, 0xbf, 0xb5, 0x3c, 0x82, 0x05, 0xf1, 0xe2, 0x23, 0xf1, 0x99, 0xff,\n\t\t0xfd, 0x8c, 0x17, 0x1f, 0xfe, 0x4e, 0x28, 0x28, 0x3b, 0x7f, 0xc8, 0xc1, 0x4a, 0x72, 0x09, 0x7d,\n\t\t0x08, 0xab, 0xa7, 0x8e, 0x6b, 0xfb, 0x97, 0x56, 0xeb, 0x82, 0xb4, 0x5e, 0x05, 0xfd, 0xae, 0x08,\n\t\t0xc2, 0x4a, 0x28, 0x56, 0x85, 0x14, 0x6d, 0xc2, 0x82, 0xdf, 0x77, 0xa3, 0xeb, 0x7b, 0x09, 0xcf,\n\t\t0xfb, 0x7d, 0x36, 0xe7, 0x7c, 0x03, 0x37, 0xce, 0x1c, 0x3f, 0x60, 0x57, 0x5e, 0x58, 0x0d, 0x56,\n\t\t0xcb, 0xeb, 0xf6, 0x3a, 0x24, 0x51, 0xea, 0x25, 0x0e, 0x89, 0xea, 0x45, 0x8d, 0x00, 0x9c, 0x5e,\n\t\t0x6c, 0xf9, 0xc4, 0x1e, 0xc4, 0x26, 0xdb, 0x95, 0x05, 0x81, 0x17, 0x8d, 0x7c, 0x99, 0xb7, 0x76,\n\t\t0xc7, 0x3d, 0x9f, 0x35, 0x8f, 0x8b, 0x11, 0x81, 0x2b, 0xb8, 0x05, 0xc0, 0xdf, 0x26, 0xa9, 0x7d,\n\t\t0xda, 0x09, 0xef, 0xc5, 0x45, 0x1c, 0x93, 0x94, 0xff, 0x28, 0xc1, 0x46, 0xda, 0xad, 0x8f, 0x76,\n\t\t0xe0, 0x56, 0x5d, 0x33, 0x2a, 0xba, 0xf1, 0xc4, 0x52, 0x54, 0x53, 0x7f, 0xa9, 0x9b, 0x27, 0x56,\n\t\t0xc3, 0x54, 0x4c, 0xcd, 0xd2, 0x8d, 0x97, 0x4a, 0x55, 0xaf, 0xc8, 0xff, 0x83, 0x3e, 0x80, 0xed,\n\t\t0x09, 0x98, 0x86, 0xfa, 0x54, 0xab, 0x34, 0xab, 0x5a, 0x45, 0x96, 0xa6, 0x68, 0x6a, 0x98, 0x0a,\n\t\t0x36, 0xb5, 0x8a, 0x9c, 0x43, 0xff, 0x0b, 0x1f, 0x4e, 0xc0, 0xa8, 0x8a, 0xa1, 0x6a, 0x55, 0x0b,\n\t\t0x6b, 0x3f, 0x6e, 0x6a, 0x0d, 0x06, 0xce, 0x97, 0x7f, 0x31, 0xb4, 0x39, 0xd1, 0xa2, 0xe2, 0x3b,\n\t\t0x55, 0x34, 0x55, 0x6f, 0xe8, 0x35, 0x63, 0x9a, 0xcd, 0x23, 0x98, 0x09, 0x36, 0x8f, 0xa2, 0x22,\n\t\t0x9b, 0xcb, 0xbf, 0xcc, 0x0d, 0x3f, 0x36, 0xe9, 0x6d, 0x4c, 0xfa, 0x83, 0xa6, 0xfc, 0x01, 0x6c,\n\t\t0x1f, 0xd7, 0xf0, 0xf3, 0xa3, 0x6a, 0xed, 0xd8, 0xd2, 0x2b, 0x16, 0xd6, 0x9a, 0x0d, 0xcd, 0xaa,\n\t\t0xd7, 0xaa, 0xba, 0x7a, 0x12, 0xb3, 0xe4, 0x4b, 0xf8, 0x74, 0x22, 0x4a, 0xa9, 0x32, 0x69, 0xa5,\n\t\t0x59, 0xaf, 0xea, 0x2a, 0xdb, 0xf5, 0x48, 0xd1, 0xab, 0x5a, 0xc5, 0xaa, 0x19, 0xd5, 0x13, 0x59,\n\t\t0x42, 0x1f, 0xc1, 0xee, 0xac, 0x4c, 0x39, 0x87, 0x1e, 0xc0, 0xfd, 0x89, 0x68, 0xac, 0x3d, 0xd3,\n\t\t0x54, 0x33, 0x06, 0xcf, 0xa3, 0x03, 0x78, 0x30, 0x11, 0x6e, 0x6a, 0xf8, 0x85, 0x6e, 0x70, 0x87,\n\t\t0x1e, 0x59, 0xb8, 0x69, 0x18, 0xba, 0xf1, 0x44, 0x9e, 0x2b, 0x5f, 0xc2, 0xda, 0xd8, 0x5b, 0x31,\n\t\t0xba, 0x0d, 0x37, 0x54, 0x5c, 0x33, 0xac, 0xda, 0x4b, 0x0d, 0x57, 0x95, 0xfa, 0xf8, 0xf9, 0x27,\n\t\t0x00, 0x1a, 0xcf, 0xf5, 0x7a, 0x3d, 0x0a, 0x42, 0x1a, 0xe0, 0xb0, 0x79, 0x74, 0xa4, 0x61, 0xab,\n\t\t0x66, 0x68, 0x72, 0xae, 0xfc, 0x3b, 0x09, 0xd6, 0xc6, 0x2e, 0x4a, 0xa6, 0xba, 0xae, 0x60, 0xcd,\n\t\t0x30, 0x2d, 0xb5, 0x5a, 0x4b, 0xf3, 0xfd, 0x04, 0x80, 0x72, 0xa8, 0x18, 0x95, 0x9a, 0x21, 0x4b,\n\t\t0xe8, 0x1e, 0xec, 0xa4, 0x01, 0x44, 0x1a, 0x8a, 0xac, 0x94, 0x73, 0xe8, 0x0e, 0xbc, 0x97, 0x86,\n\t\t0x1b, 0x38, 0x4a, 0xce, 0x97, 0xff, 0x99, 0x83, 0x9b, 0xd3, 0x3e, 0xa7, 0xb1, 0xe4, 0x1f, 0x78,\n\t\t0x5c, 0xfb, 0x56, 0x53, 0x9b, 0x26, 0x4b, 0xb7, 0x50, 0x1f, 0x4b, 0xba, 0x66, 0x23, 0x66, 0x79,\n\t\t0x3c, 0x9a, 0x13, 0xc0, 0x6a, 0xed, 0x45, 0xbd, 0xaa, 0x99, 0xdc, 0x87, 0x65, 0xb8, 0x97, 0x05,\n\t\t0x0f, 0x73, 0x4b, 0xce, 0x25, 0xd2, 0x6a, 0x92, 0x6a, 0x7e, 0x6e, 0x56, 0x85, 0x68, 0x0f, 0xca,\n\t\t0x59, 0xe8, 0x81, 0x17, 0x2a, 0xf2, 0x1c, 0xfa, 0x14, 0x3e, 0xce, 0x36, 0xdc, 0x30, 0x75, 0xa3,\n\t\t0xa9, 0x55, 0x2c, 0xa5, 0x61, 0x19, 0xda, 0xb1, 0x3c, 0x3f, 0xcb, 0x71, 0x4d, 0xfd, 0x05, 0x2b,\n\t\t0x8d, 0xa6, 0x29, 0x2f, 0x94, 0x7f, 0x95, 0x87, 0xad, 0x09, 0x1f, 0x2b, 0xd0, 0x5d, 0xb8, 0x93,\n\t\t0xa2, 0x6a, 0xcc, 0xc1, 0x53, 0x61, 0xa2, 0x29, 0xc8, 0xd2, 0x74, 0xd8, 0xb0, 0xb1, 0x7d, 0x08,\n\t\t0xef, 0x4f, 0x86, 0x0d, 0x03, 0x95, 0x4f, 0xf4, 0x8c, 0x31, 0xa0, 0x08, 0xd1, 0x1c, 0x4b, 0xcb,\n\t\t0x29, 0xea, 0xa2, 0xe0, 0xcc, 0xa3, 0x5d, 0xf8, 0x60, 0x32, 0x2e, 0x16, 0x96, 0x85, 0x09, 0x61,\n\t\t0x9c, 0x14, 0x90, 0x2b, 0xd3, 0x0f, 0x34, 0x0c, 0xc5, 0x62, 0xf9, 0xaf, 0x12, 0x5c, 0x55, 0x3d,\n\t\t0x97, 0x3a, 0x6e, 0x9f, 0x28, 0x81, 0x41, 0xde, 0xe8, 0xe1, 0x38, 0xec, 0xf9, 0xcc, 0x77, 0x91,\n\t\t0x66, 0xa1, 0xd8, 0xd2, 0x0d, 0xdd, 0xd4, 0x15, 0xb3, 0x86, 0x93, 0x91, 0x98, 0x0c, 0x63, 0x6d,\n\t\t0xb9, 0xa2, 0xe1, 0x30, 0xc5, 0x27, 0xc3, 0xb0, 0x66, 0xe2, 0x13, 0x51, 0x95, 0xe1, 0x3d, 0x33,\n\t\t0x19, 0xcb, 0x9b, 0x4d, 0x74, 0x0b, 0xc8, 0xf9, 0xf2, 0xef, 0x25, 0x28, 0x88, 0x6f, 0x24, 0xfc,\n\t\t0x15, 0xba, 0x04, 0x1b, 0xec, 0x80, 0xb5, 0xa6, 0x69, 0x99, 0x27, 0x75, 0x2d, 0xd9, 0x4e, 0x12,\n\t\t0x2b, 0x3c, 0xfe, 0x96, 0x59, 0x0b, 0x13, 0x35, 0x6c, 0x65, 0x49, 0x80, 0xd8, 0x85, 0x61, 0x38,\n\t\t0x58, 0xce, 0x4d, 0xc5, 0x84, 0x7a, 0xf2, 0xe8, 0x3a, 0x5c, 0x4d, 0x60, 0x9e, 0x6a, 0x0a, 0x36,\n\t\t0x0f, 0x35, 0xc5, 0x94, 0xe7, 0xca, 0xbf, 0x95, 0xe0, 0x5a, 0x74, 0x1f, 0x9a, 0x6c, 0xbc, 0x72,\n\t\t0xba, 0xa4, 0x5d, 0xeb, 0x53, 0xd5, 0xee, 0x07, 0x04, 0xdd, 0x87, 0xbb, 0x83, 0x9b, 0xcc, 0x54,\n\t\t0x1a, 0xcf, 0x87, 0xb1, 0xb2, 0x54, 0x85, 0xb5, 0xf8, 0xe1, 0x69, 0x32, 0xa1, 0xc2, 0x04, 0x59,\n\t\t0x62, 0xd9, 0x30, 0x1d, 0x8a, 0xb5, 0x86, 0x66, 0xca, 0xb9, 0xf2, 0xdf, 0x0b, 0xb0, 0x15, 0x37,\n\t\t0x8e, 0xbd, 0x68, 0x92, 0x76, 0x68, 0xda, 0x3d, 0xd8, 0x49, 0x2a, 0x11, 0xb7, 0xdd, 0xa8, 0x5d,\n\t\t0x07, 0xf0, 0x60, 0x0a, 0xae, 0x69, 0x3c, 0x55, 0x8c, 0x0a, 0x7b, 0x8e, 0x40, 0xb2, 0x84, 0x1e,\n\t\t0xc3, 0xa3, 0x29, 0x94, 0x43, 0xa5, 0x32, 0xf4, 0xf2, 0x60, 0xee, 0x50, 0x4c, 0x13, 0xeb, 0x87,\n\t\t0x4d, 0x53, 0x6b, 0xc8, 0x39, 0xa4, 0x81, 0x92, 0xa1, 0x20, 0x79, 0x25, 0xa4, 0xaa, 0xc9, 0xa3,\n\t\t0xaf, 0xe0, 0xb3, 0x2c, 0x3b, 0xc2, 0x94, 0xd1, 0x5f, 0x68, 0x38, 0x4e, 0x9d, 0x43, 0x5f, 0xc3,\n\t\t0xe7, 0x19, 0x54, 0xb1, 0xf3, 0x18, 0x77, 0x1e, 0x3d, 0x82, 0x2f, 0x32, 0xad, 0x57, 0x6b, 0xb8,\n\t\t0x62, 0xbd, 0x50, 0xf0, 0xf3, 0x24, 0x79, 0x01, 0xe9, 0xa0, 0x65, 0x6d, 0x2c, 0xfa, 0x97, 0x95,\n\t\t0xd2, 0x11, 0x62, 0xaa, 0xae, 0xcc, 0xe0, 0x45, 0x26, 0xc8, 0x50, 0xb3, 0x88, 0x9e, 0x80, 0x3a,\n\t\t0x9b, 0x2b, 0xa6, 0x2b, 0x5a, 0x42, 0xdf, 0x82, 0xf9, 0x6e, 0x51, 0xd5, 0xbe, 0x35, 0x35, 0x6c,\n\t\t0x28, 0x59, 0x9a, 0x01, 0x7d, 0x03, 0x5f, 0x65, 0x3a, 0x2d, 0xd9, 0x7f, 0x62, 0xf4, 0x02, 0xfa,\n\t\t0x02, 0x3e, 0x99, 0x42, 0x8f, 0xe7, 0xc8, 0x70, 0x36, 0xd4, 0x2b, 0x72, 0x11, 0x7d, 0x06, 0x07,\n\t\t0x53, 0x88, 0xbc, 0x0a, 0xad, 0x86, 0xa9, 0xab, 0xcf, 0x4f, 0xc2, 0xe5, 0xaa, 0xde, 0x30, 0xe5,\n\t\t0x65, 0xf4, 0x23, 0xf8, 0xbf, 0x29, 0xb4, 0xc1, 0x61, 0xd9, 0x1f, 0x1a, 0x8e, 0x95, 0x18, 0x83,\n\t\t0x35, 0xb1, 0x26, 0xaf, 0xcc, 0x10, 0x93, 0x86, 0xfe, 0x24, 0xdb, 0x73, 0xab, 0x48, 0x85, 0xc7,\n\t\t0x33, 0x95, 0x88, 0xfa, 0x54, 0xaf, 0x56, 0xd2, 0x95, 0xc8, 0xe8, 0x13, 0xd8, 0x9f, 0xa2, 0xe4,\n\t\t0xa8, 0x86, 0x55, 0x4d, 0x0c, 0x0f, 0x83, 0x26, 0xb1, 0x86, 0x3e, 0x87, 0x87, 0xd3, 0x48, 0x8a,\n\t\t0x5e, 0x65, 0x13, 0xe8, 0x28, 0x0f, 0xb1, 0x89, 0x66, 0xb6, 0xa3, 0xeb, 0x46, 0xbd, 0x69, 0x5a,\n\t\t0x0d, 0xfd, 0x3b, 0x4d, 0x5e, 0x67, 0x13, 0x4d, 0x66, 0xa4, 0x22, 0x5f, 0xc9, 0x1b, 0xe3, 0xcd,\n\t\t0x78, 0x6c, 0x93, 0x43, 0xdd, 0x50, 0xf0, 0x89, 0xbc, 0x99, 0x91, 0x7b, 0xe3, 0x8d, 0x2e, 0x91,\n\t\t0x42, 0x57, 0x67, 0x39, 0x8e, 0xa6, 0x60, 0xf5, 0x69, 0xdc, 0xe3, 0x5b, 0xec, 0xd6, 0xb9, 0xc3,\n\t\t0xbf, 0xcb, 0x8d, 0x8d, 0x5d, 0xf1, 0x16, 0x7f, 0x00, 0x0f, 0xc2, 0xb8, 0xa5, 0x64, 0xc1, 0x84,\n\t\t0x6e, 0x7f, 0x08, 0xff, 0x3f, 0x1b, 0x65, 0xb0, 0xae, 0x54, 0xb1, 0xa6, 0x54, 0x4e, 0x06, 0x2f,\n\t\t0x26, 0x52, 0xf9, 0x37, 0x39, 0x28, 0xab, 0xb6, 0xdb, 0x22, 0x9d, 0xe8, 0xff, 0x01, 0x53, 0xad,\n\t\t0x7c, 0x04, 0x5f, 0xcc, 0x50, 0xef, 0x13, 0xec, 0x3d, 0x86, 0xc6, 0xbb, 0x92, 0x9b, 0xc6, 0x73,\n\t\t0xa3, 0x76, 0x6c, 0x4c, 0x23, 0xc8, 0x12, 0x32, 0xe0, 0xd9, 0xbb, 0x2a, 0x1e, 0x73, 0xc9, 0x70,\n\t\t0xd2, 0xcc, 0x71, 0xa7, 0x34, 0x9c, 0x73, 0xfe, 0xcf, 0x91, 0xd9, 0x9c, 0x22, 0xd2, 0xf8, 0xdf,\n\t\t0x73, 0xca, 0xbb, 0x92, 0x67, 0x76, 0xca, 0xbb, 0x2a, 0x9e, 0xe6, 0x94, 0xc3, 0x9f, 0xc2, 0x56,\n\t\t0xcb, 0xeb, 0xa6, 0x7d, 0x6b, 0x3a, 0x5c, 0x8e, 0xdc, 0x53, 0xf7, 0x3d, 0xea, 0xd5, 0xa5, 0xef,\n\t\t0x0e, 0xce, 0x1d, 0x7a, 0xd1, 0x3f, 0xdd, 0x6b, 0x79, 0xdd, 0xfd, 0xf8, 0x8f, 0x52, 0x1e, 0x38,\n\t\t0xed, 0xce, 0xfe, 0xb9, 0x17, 0xfe, 0xc8, 0x45, 0xfc, 0x42, 0xe5, 0x91, 0xdd, 0x73, 0x5e, 0x1f,\n\t\t0x9c, 0x2e, 0x70, 0xd9, 0x27, 0xff, 0x0a, 0x00, 0x00, 0xff, 0xff, 0x00, 0x6a, 0x76, 0xd7, 0x61,\n\t\t0x23, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/query.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x53, 0xdf, 0x6f, 0x93, 0x50,\n\t\t0x18, 0x95, 0x9a, 0x2c, 0xd9, 0xb7, 0x55, 0xc9, 0x9d, 0xc6, 0xda, 0xec, 0x47, 0xd3, 0xed, 0x61,\n\t\t0x69, 0x14, 0xec, 0xf4, 0x6d, 0x4f, 0x8c, 0x5e, 0x0d, 0x86, 0x01, 0x03, 0xda, 0xa5, 0x7b, 0x21,\n\t\t0x94, 0x5e, 0x2b, 0x8e, 0x72, 0xf1, 0x5e, 0x68, 0xed, 0x3f, 0xe0, 0xbb, 0x7f, 0x8d, 0xff, 0x9e,\n\t\t0x81, 0x52, 0x5b, 0x2d, 0x33, 0xbe, 0x7d, 0x9c, 0xef, 0x1c, 0xce, 0x39, 0xb9, 0xf9, 0xe0, 0x24,\n\t\t0x1b, 0x11, 0x26, 0x07, 0xfe, 0x98, 0xc4, 0x01, 0x91, 0xfd, 0x24, 0x94, 0x67, 0x5d, 0xf9, 0x6b,\n\t\t0x46, 0xd8, 0x42, 0x4a, 0x18, 0x4d, 0x29, 0x3a, 0xc8, 0x09, 0x52, 0x49, 0x90, 0xfc, 0x24, 0x94,\n\t\t0x66, 0xdd, 0x66, 0xab, 0x4a, 0x15, 0xd0, 0xe9, 0x94, 0xc6, 0x4b, 0x59, 0xb3, 0x5d, 0xc5, 0x98,\n\t\t0x53, 0x76, 0xff, 0x29, 0xa2, 0xf3, 0x25, 0xa7, 0x7d, 0x0f, 0xf5, 0xdb, 0x12, 0xb9, 0xc9, 0x1d,\n\t\t0xd1, 0x11, 0x40, 0x61, 0xed, 0xa5, 0x8b, 0x84, 0x34, 0x84, 0x96, 0x70, 0xbe, 0x6b, 0xef, 0x16,\n\t\t0x88, 0xbb, 0x48, 0x08, 0xba, 0x5c, 0xad, 0x7d, 0x36, 0xe1, 0x8d, 0x5a, 0x4b, 0x38, 0xdf, 0xbb,\n\t\t0x38, 0x94, 0x2a, 0xf2, 0x49, 0x96, 0xbf, 0x88, 0xa8, 0x3f, 0x2e, 0xc5, 0x0a, 0x9b, 0xf0, 0xf6,\n\t\t0x4f, 0x01, 0x0e, 0xfe, 0x70, 0xb3, 0x09, 0xcf, 0xa2, 0x14, 0x61, 0xd8, 0x63, 0xc5, 0xb4, 0x36,\n\t\t0x7d, 0x72, 0x71, 0x56, 0xf9, 0xd7, 0x0d, 0x59, 0x9e, 0xc7, 0x06, 0xf6, 0x7b, 0x46, 0xef, 0x60,\n\t\t0xc7, 0x8f, 0xf9, 0x9c, 0xb0, 0xff, 0xca, 0x55, 0x72, 0xd1, 0x29, 0xd4, 0x09, 0x63, 0x94, 0x79,\n\t\t0x53, 0xc2, 0xb9, 0x3f, 0x21, 0x8d, 0xc7, 0x45, 0xe7, 0xfd, 0x02, 0xbc, 0x5e, 0x62, 0x6d, 0x02,\n\t\t0xf5, 0xd2, 0xf9, 0x0b, 0x09, 0x52, 0x32, 0x46, 0x2e, 0xec, 0x07, 0x11, 0xe5, 0xc4, 0xe3, 0xa9,\n\t\t0x9f, 0x66, 0xbc, 0xcc, 0xdc, 0xad, 0x74, 0x5c, 0x55, 0xc6, 0xdf, 0x48, 0x90, 0xa5, 0x21, 0x8d,\n\t\t0xd5, 0x5c, 0xe9, 0x14, 0x42, 0x7b, 0x2f, 0x58, 0x7f, 0x74, 0x62, 0x78, 0xfa, 0x57, 0x41, 0x74,\n\t\t0x04, 0x2f, 0x6f, 0xfa, 0xd8, 0x1e, 0x7a, 0x36, 0x76, 0xfa, 0xba, 0xeb, 0xb9, 0x43, 0x0b, 0x7b,\n\t\t0x9a, 0x31, 0x50, 0x74, 0xad, 0x27, 0x3e, 0x42, 0xc7, 0xd0, 0xdc, 0x5e, 0x2b, 0x86, 0x73, 0x8b,\n\t\t0x6d, 0xdc, 0x13, 0x05, 0x74, 0x08, 0x8d, 0xed, 0xfd, 0x7b, 0x45, 0xd3, 0x71, 0x4f, 0xac, 0x75,\n\t\t0x7e, 0x08, 0xf0, 0x6c, 0xa3, 0x97, 0x4a, 0xe3, 0x71, 0x98, 0x07, 0x44, 0x6d, 0x38, 0x5e, 0xc9,\n\t\t0x3e, 0x62, 0xd5, 0xf5, 0x54, 0xd3, 0xe8, 0x69, 0xae, 0x66, 0x1a, 0x1b, 0xd6, 0xa7, 0x70, 0xf2,\n\t\t0x00, 0xc7, 0x30, 0x5d, 0xcf, 0xb4, 0xb0, 0x21, 0x0a, 0xe8, 0x0d, 0xbc, 0xfa, 0x07, 0x49, 0x35,\n\t\t0xaf, 0x2d, 0x1d, 0xbb, 0xb8, 0xe7, 0xa9, 0x3a, 0x56, 0x0c, 0x7d, 0x28, 0xd6, 0x3a, 0xdf, 0x05,\n\t\t0x78, 0x5e, 0x64, 0x52, 0x69, 0xcc, 0x43, 0x9e, 0x92, 0x38, 0x58, 0xe8, 0x64, 0x46, 0xa2, 0xb5,\n\t\t0xa1, 0x6a, 0x1a, 0x8e, 0xe6, 0xb8, 0xd8, 0x50, 0x87, 0x9e, 0x8e, 0x07, 0x58, 0xdf, 0x48, 0x75,\n\t\t0x06, 0xad, 0x87, 0x48, 0x78, 0x80, 0x0d, 0xb7, 0xaf, 0xe8, 0xa2, 0xb0, 0xee, 0xb7, 0xcd, 0x72,\n\t\t0x5c, 0xdb, 0x34, 0x3e, 0x88, 0xb5, 0xab, 0x3b, 0x78, 0x11, 0xd0, 0x69, 0xd5, 0x8b, 0x5e, 0x41,\n\t\t0x11, 0xd0, 0xca, 0x2f, 0xc8, 0x12, 0xee, 0xba, 0x93, 0x30, 0xfd, 0x9c, 0x8d, 0xa4, 0x80, 0x4e,\n\t\t0xe5, 0xcd, 0x93, 0x7b, 0x1d, 0x8e, 0x23, 0x79, 0x42, 0xe5, 0xe2, 0xd2, 0xca, 0xfb, 0xbb, 0xf4,\n\t\t0x93, 0x70, 0xd6, 0x1d, 0xed, 0x14, 0xd8, 0xdb, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0xbd, 0x69,\n\t\t0x28, 0x5b, 0xfb, 0x03, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/service_workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xdc, 0x5a, 0x5b, 0x6f, 0xdc, 0xc6,\n\t\t0xf5, 0x07, 0x57, 0xf7, 0xb3, 0x92, 0x2c, 0x8d, 0x6e, 0xeb, 0x95, 0x75, 0xa3, 0xe3, 0xfc, 0xf5,\n\t\t0x77, 0xec, 0x55, 0x25, 0xc5, 0x97, 0x38, 0x69, 0x03, 0x79, 0x6d, 0x39, 0x2a, 0xec, 0x54, 0xa5,\n\t\t0x54, 0x1b, 0xed, 0x0b, 0x31, 0x22, 0x67, 0x57, 0x63, 0x71, 0x49, 0x6a, 0x38, 0x94, 0xbc, 0xc9,\n\t\t0x43, 0xd1, 0x22, 0x48, 0x81, 0xa2, 0x2d, 0xfa, 0xd8, 0x02, 0x05, 0xfa, 0xd0, 0x87, 0x3c, 0x35,\n\t\t0x28, 0xd0, 0xa7, 0xbe, 0x17, 0xfd, 0x1e, 0xfd, 0x04, 0x79, 0xea, 0x6b, 0x50, 0x70, 0x66, 0xb8,\n\t\t0x37, 0x91, 0xdc, 0x95, 0x94, 0xc0, 0x4e, 0xdf, 0x76, 0x66, 0xce, 0xef, 0xcc, 0x99, 0x73, 0x9b,\n\t\t0x33, 0x67, 0x09, 0x37, 0xc3, 0x03, 0xc2, 0xd6, 0x2c, 0x6c, 0x13, 0xd7, 0x22, 0x6b, 0xd8, 0xa7,\n\t\t0x6b, 0x27, 0xeb, 0x6b, 0x01, 0x61, 0x27, 0xd4, 0x22, 0xe6, 0xa9, 0xc7, 0x8e, 0x2a, 0x8e, 0x77,\n\t\t0x5a, 0xf2, 0x99, 0xc7, 0x3d, 0x34, 0x15, 0xd1, 0x96, 0x14, 0x6d, 0x09, 0xfb, 0xb4, 0x74, 0xb2,\n\t\t0x5e, 0x5c, 0xac, 0x7a, 0x5e, 0xd5, 0x21, 0x6b, 0x82, 0xe4, 0x20, 0xac, 0xac, 0xd9, 0x21, 0xc3,\n\t\t0x9c, 0x7a, 0xae, 0x04, 0x15, 0x97, 0x3a, 0xd7, 0x39, 0xad, 0x91, 0x80, 0xe3, 0x9a, 0xaf, 0x08,\n\t\t0x96, 0x93, 0x24, 0xb0, 0xbc, 0x5a, 0xad, 0xc1, 0x62, 0x25, 0x89, 0xe2, 0x90, 0x06, 0xdc, 0x63,\n\t\t0xf5, 0x78, 0x97, 0x24, 0x92, 0xe3, 0x90, 0x34, 0x08, 0xf4, 0x24, 0x02, 0x8e, 0x83, 0x23, 0x87,\n\t\t0x06, 0x3c, 0x8b, 0xa6, 0x5d, 0x07, 0xfa, 0x3f, 0x35, 0x58, 0x32, 0x22, 0xf9, 0x19, 0x7f, 0xa1,\n\t\t0x56, 0x1e, 0xbf, 0x22, 0x56, 0x18, 0x9d, 0xd8, 0x20, 0xc7, 0x21, 0x09, 0x38, 0x9a, 0x85, 0x41,\n\t\t0xdb, 0xab, 0x61, 0xea, 0x16, 0xb4, 0x65, 0x6d, 0x75, 0xc4, 0x50, 0x23, 0xf4, 0x13, 0x40, 0x31,\n\t\t0x37, 0x93, 0xc4, 0xa0, 0x42, 0x6e, 0x59, 0x5b, 0xcd, 0x6f, 0xbc, 0x5d, 0x4a, 0x50, 0x6e, 0xe9,\n\t\t0xec, 0x16, 0x93, 0xa7, 0x9d, 0x53, 0xa8, 0x08, 0xc3, 0xd4, 0x26, 0x2e, 0xa7, 0xbc, 0x5e, 0xe8,\n\t\t0x13, 0x1b, 0x36, 0xc6, 0x91, 0x28, 0x8c, 0xe0, 0xc0, 0x73, 0x0b, 0xfd, 0x52, 0x14, 0x39, 0xd2,\n\t\t0xff, 0xaa, 0xc1, 0xf2, 0x23, 0x8a, 0xab, 0xae, 0x17, 0x90, 0xef, 0xc0, 0x39, 0xf4, 0x2f, 0x34,\n\t\t0x58, 0xc9, 0x90, 0x37, 0xf0, 0x3d, 0x37, 0x20, 0xa9, 0x02, 0xbf, 0x84, 0x05, 0x5b, 0x82, 0x39,\n\t\t0xb5, 0xcc, 0x4b, 0xcb, 0x3e, 0xdf, 0x64, 0x76, 0x66, 0x51, 0xff, 0x03, 0xc0, 0xc2, 0xde, 0x85,\n\t\t0xdc, 0x63, 0x09, 0xf2, 0x0d, 0xd1, 0xa8, 0x2d, 0x64, 0x1a, 0x31, 0x20, 0x9e, 0xda, 0xb1, 0xd1,\n\t\t0x36, 0x8c, 0x35, 0x08, 0x78, 0xdd, 0x27, 0x42, 0x4b, 0xf9, 0x8d, 0x95, 0x4c, 0xb1, 0xf7, 0xeb,\n\t\t0x3e, 0x31, 0x46, 0x4f, 0x5b, 0x46, 0xe8, 0x01, 0x8c, 0x44, 0x9e, 0x6f, 0x46, 0xae, 0x2f, 0xfc,\n\t\t0x22, 0xbf, 0xb1, 0x90, 0xc8, 0x63, 0x1f, 0x07, 0x47, 0x4f, 0x69, 0xc0, 0x8d, 0x61, 0xae, 0x7e,\n\t\t0xa1, 0x0d, 0x18, 0xa0, 0xae, 0x1f, 0xf2, 0xc2, 0x80, 0xc0, 0x5d, 0x4b, 0xc4, 0xed, 0xe2, 0xba,\n\t\t0xe3, 0x61, 0xdb, 0x90, 0xa4, 0x08, 0xc3, 0x72, 0x43, 0xd5, 0xa6, 0x08, 0x1d, 0x93, 0x7b, 0xa6,\n\t\t0xe5, 0x78, 0x01, 0x31, 0xa3, 0x6c, 0xe0, 0x85, 0xbc, 0x30, 0x28, 0xd8, 0x5d, 0x2d, 0xc9, 0x6c,\n\t\t0x51, 0x8a, 0xb3, 0x45, 0xe9, 0x91, 0xca, 0x26, 0xc6, 0xb5, 0x06, 0x0b, 0xa1, 0xdd, 0x7d, 0xaf,\n\t\t0x1c, 0xe1, 0xf7, 0x25, 0x1c, 0xbd, 0x80, 0x79, 0x71, 0xa4, 0x14, 0xee, 0x43, 0xdd, 0xb8, 0xcf,\n\t\t0x45, 0xe8, 0x24, 0xc6, 0xad, 0x4e, 0x39, 0xdc, 0x11, 0x5c, 0x0b, 0x00, 0x4c, 0xda, 0x34, 0xb2,\n\t\t0xd7, 0x88, 0x58, 0x1d, 0x51, 0x33, 0x3b, 0x36, 0xb2, 0xa0, 0xd0, 0x62, 0x4f, 0x93, 0x91, 0x30,\n\t\t0x20, 0xa6, 0xef, 0x39, 0xd4, 0xaa, 0x17, 0x60, 0x59, 0x5b, 0x1d, 0xdf, 0xb8, 0x99, 0x69, 0xb9,\n\t\t0x1d, 0xdb, 0x88, 0x20, 0xbb, 0x02, 0x61, 0xcc, 0x9c, 0x26, 0x4d, 0xa3, 0x32, 0x8c, 0x32, 0xc2,\n\t\t0x59, 0x3d, 0x66, 0x9c, 0x17, 0x27, 0x5d, 0x4e, 0x64, 0x6c, 0x44, 0x84, 0x8a, 0x5d, 0x9e, 0x35,\n\t\t0x07, 0xe8, 0x3a, 0x8c, 0x59, 0x2c, 0xb2, 0x8d, 0x75, 0x48, 0xec, 0xd0, 0x21, 0x85, 0x51, 0x71,\n\t\t0x96, 0xd1, 0x68, 0x72, 0x4f, 0xcd, 0xa1, 0xdb, 0xd0, 0x5f, 0x23, 0x35, 0xaf, 0x30, 0xa6, 0x74,\n\t\t0x99, 0xb4, 0xc3, 0x33, 0x52, 0xf3, 0x0c, 0x41, 0x86, 0x0c, 0x98, 0x0c, 0x08, 0x66, 0xd6, 0xa1,\n\t\t0x89, 0x39, 0x67, 0xf4, 0x20, 0xe4, 0x24, 0x28, 0x8c, 0x0b, 0xec, 0x8d, 0x44, 0xec, 0x9e, 0xa0,\n\t\t0xde, 0x6a, 0x10, 0x1b, 0x13, 0x41, 0xc7, 0x0c, 0xda, 0x84, 0xc1, 0x43, 0x82, 0x6d, 0xc2, 0x0a,\n\t\t0x57, 0x04, 0xa3, 0xf9, 0x44, 0x46, 0x1f, 0x09, 0x12, 0x43, 0x91, 0xa2, 0x07, 0x90, 0xb7, 0x89,\n\t\t0x83, 0xeb, 0xd2, 0x37, 0x0a, 0x13, 0xdd, 0x5c, 0x01, 0x04, 0xb5, 0xf0, 0x05, 0xf4, 0x01, 0x8c,\n\t\t0xbe, 0xa4, 0x9c, 0x13, 0xa6, 0xc0, 0x93, 0xdd, 0xc0, 0x79, 0x49, 0xde, 0x40, 0x57, 0x28, 0x0b,\n\t\t0xb8, 0xc9, 0x42, 0xd7, 0xc4, 0xbc, 0x80, 0x04, 0xba, 0x78, 0x06, 0xbd, 0x1f, 0xdf, 0x88, 0x06,\n\t\t0x08, 0x7a, 0x23, 0x74, 0xb7, 0x38, 0x7a, 0x0e, 0x53, 0xc2, 0x28, 0xde, 0x09, 0x61, 0x0e, 0xf6,\n\t\t0x63, 0x03, 0x4f, 0x09, 0xcf, 0x49, 0x4e, 0x55, 0x65, 0xe6, 0xb9, 0x3f, 0x92, 0xe4, 0xca, 0xcc,\n\t\t0x93, 0x56, 0xe7, 0x14, 0x7a, 0x05, 0x4b, 0xd8, 0xe2, 0xf4, 0x84, 0x98, 0x96, 0x13, 0x06, 0xe2,\n\t\t0x6c, 0xc4, 0x21, 0x96, 0x08, 0x4e, 0xb5, 0xc7, 0xb4, 0x10, 0x74, 0x3d, 0x71, 0x8f, 0x2d, 0x81,\n\t\t0x2d, 0x4b, 0xe8, 0x5e, 0x8c, 0x54, 0xdb, 0x5d, 0xc3, 0x19, 0xab, 0xfa, 0x3d, 0x58, 0x4c, 0xcb,\n\t\t0x8c, 0x2a, 0x81, 0xcf, 0xc0, 0x60, 0xa4, 0x2b, 0x6a, 0xab, 0xd4, 0x38, 0xc0, 0x42, 0x77, 0xc7,\n\t\t0xd6, 0x19, 0xe8, 0xc9, 0xc0, 0xad, 0xa0, 0xee, 0x5a, 0x71, 0x5e, 0x7d, 0x0a, 0x43, 0x2a, 0xf8,\n\t\t0x04, 0x3a, 0xbf, 0xb1, 0x91, 0xec, 0x67, 0x59, 0xc9, 0xd9, 0x88, 0x59, 0xe8, 0x37, 0xe0, 0x7a,\n\t\t0xe6, 0x9e, 0x52, 0x62, 0xfd, 0x3d, 0x58, 0x4e, 0x2f, 0x07, 0xb2, 0x4f, 0xf5, 0xaf, 0x1c, 0x2c,\n\t\t0xee, 0xd1, 0xaa, 0x8b, 0x9d, 0xef, 0x42, 0x25, 0xd1, 0x9e, 0xec, 0xfa, 0x3b, 0x93, 0xdd, 0x12,\n\t\t0xe4, 0x03, 0x71, 0x16, 0xd3, 0xc5, 0x35, 0x22, 0x6e, 0x87, 0x11, 0x03, 0xe4, 0xd4, 0xc7, 0xb8,\n\t\t0x46, 0xd0, 0x87, 0x30, 0xaa, 0x08, 0xe4, 0xfd, 0x31, 0xd8, 0xc3, 0xfd, 0xa1, 0x58, 0xee, 0x88,\n\t\t0x5b, 0xa4, 0x00, 0x43, 0x96, 0xe7, 0x72, 0xe6, 0x39, 0x22, 0x9d, 0x8f, 0x1a, 0xf1, 0x50, 0x5f,\n\t\t0x81, 0xa5, 0x54, 0x3d, 0x2a, 0x33, 0x7d, 0xad, 0xc1, 0xff, 0x29, 0x1a, 0xca, 0x0f, 0xb3, 0xef,\n\t\t0xe7, 0x17, 0x30, 0x26, 0xaf, 0x91, 0xcb, 0x7b, 0xd3, 0xa8, 0x60, 0x14, 0x33, 0xee, 0xd0, 0x51,\n\t\t0xae, 0xab, 0x8e, 0xfa, 0x2e, 0xa1, 0xa3, 0xfe, 0x76, 0x1d, 0x6d, 0xc1, 0x6a, 0xf7, 0xf3, 0x67,\n\t\t0xfb, 0xeb, 0xe7, 0x1a, 0xdc, 0xea, 0xc6, 0xa3, 0x2d, 0x20, 0x9f, 0x77, 0x06, 0xe4, 0x07, 0xc9,\n\t\t0x2a, 0xec, 0xcd, 0x2e, 0xcd, 0xd0, 0x5c, 0x83, 0xdb, 0x3d, 0xca, 0xa1, 0xac, 0xff, 0x65, 0x0e,\n\t\t0x16, 0x0c, 0x12, 0x90, 0x37, 0xa6, 0x64, 0x6f, 0x96, 0xe5, 0x7d, 0xad, 0x65, 0x39, 0xba, 0x07,\n\t\t0x05, 0x9b, 0x58, 0x34, 0x88, 0x72, 0x71, 0x85, 0xba, 0x34, 0x38, 0x34, 0xc9, 0x09, 0x71, 0x1b,\n\t\t0x21, 0xd7, 0x67, 0xcc, 0xc4, 0xeb, 0xdb, 0x62, 0xf9, 0x71, 0xb4, 0xba, 0x63, 0x77, 0x44, 0xe7,\n\t\t0x40, 0x67, 0x74, 0x96, 0x60, 0x2a, 0x38, 0xa2, 0xbe, 0xa9, 0xbc, 0x8b, 0x11, 0xec, 0xfb, 0x4e,\n\t\t0x5d, 0xc4, 0xe0, 0xb0, 0x31, 0x19, 0x2d, 0x49, 0x85, 0x1a, 0x72, 0x21, 0xca, 0xd4, 0x69, 0xfa,\n\t\t0xca, 0xf6, 0x91, 0x3f, 0xe5, 0xe0, 0x86, 0xd2, 0x69, 0x19, 0xbb, 0x16, 0xf9, 0x5f, 0x48, 0x6d,\n\t\t0xd3, 0x30, 0x60, 0xe1, 0x30, 0x88, 0x93, 0x9a, 0x1c, 0xa0, 0x4d, 0x98, 0x95, 0x97, 0x7b, 0xb3,\n\t\t0xb4, 0x55, 0x0a, 0x19, 0x14, 0x64, 0x53, 0x62, 0xb5, 0x29, 0x93, 0x50, 0xcf, 0x2a, 0xbc, 0xdd,\n\t\t0x4d, 0x3b, 0xca, 0x65, 0xff, 0x9e, 0x83, 0x95, 0x7d, 0xc2, 0x6a, 0xd4, 0xc5, 0x9c, 0xbc, 0xe9,\n\t\t0x6e, 0x7b, 0x17, 0x86, 0x6c, 0xc2, 0x31, 0x75, 0x02, 0xf5, 0x9c, 0xc8, 0x4e, 0x59, 0x31, 0x71,\n\t\t0x9b, 0x51, 0x06, 0x3a, 0x8c, 0x72, 0x21, 0xfd, 0xbe, 0x05, 0x7a, 0x96, 0xd2, 0x94, 0x6e, 0xff,\n\t\t0x13, 0x3d, 0x7e, 0x49, 0x60, 0x31, 0x7a, 0xf0, 0xc6, 0xa8, 0xf6, 0x00, 0xe6, 0x44, 0xbb, 0xc2,\n\t\t0xb4, 0x3c, 0x37, 0xa0, 0x01, 0x27, 0xae, 0x55, 0x37, 0x1d, 0x72, 0x42, 0x1c, 0xa1, 0xeb, 0xb4,\n\t\t0xb7, 0xc2, 0x8f, 0x23, 0x4c, 0xb9, 0x09, 0x79, 0x1a, 0x21, 0x8c, 0x99, 0xe3, 0xa4, 0x69, 0xfd,\n\t\t0xeb, 0x3e, 0x58, 0xc9, 0x38, 0xb7, 0x8a, 0x6c, 0x07, 0xe6, 0x9a, 0x2a, 0xb7, 0x3c, 0xb7, 0x42,\n\t\t0xab, 0xaa, 0xba, 0x55, 0x59, 0x7c, 0xb3, 0xb7, 0x53, 0x96, 0x5b, 0xa1, 0xc6, 0x2c, 0x49, 0x9c,\n\t\t0x8f, 0xce, 0x7d, 0x56, 0x9d, 0x26, 0x75, 0x2b, 0x9e, 0xd2, 0xe9, 0xcd, 0xde, 0x76, 0xdb, 0x71,\n\t\t0x2b, 0x5e, 0xf3, 0x8d, 0xd4, 0x36, 0x8d, 0x5e, 0x00, 0xf2, 0x89, 0x6b, 0x53, 0xb7, 0x6a, 0x8a,\n\t\t0xfa, 0x94, 0x72, 0x4a, 0x82, 0x42, 0xdf, 0x72, 0xdf, 0x6a, 0x7e, 0x63, 0x35, 0xd9, 0x53, 0x25,\n\t\t0xf9, 0x96, 0xa4, 0xae, 0x0b, 0xe6, 0x93, 0x7e, 0xdb, 0x24, 0x25, 0x01, 0xfa, 0x29, 0x4c, 0xc4,\n\t\t0x8c, 0xad, 0x43, 0xea, 0xd8, 0x8c, 0xb8, 0x85, 0x7e, 0xc1, 0xb6, 0x94, 0xc5, 0xb6, 0x1c, 0xd1,\n\t\t0xb6, 0x4b, 0x7e, 0xc5, 0x6f, 0x59, 0x62, 0xc4, 0x45, 0x7b, 0x4d, 0xd6, 0x71, 0xc6, 0x57, 0x4f,\n\t\t0xee, 0x4c, 0x89, 0x1f, 0x29, 0xda, 0x36, 0xa6, 0xf1, 0xa4, 0xfe, 0x59, 0x1f, 0x4c, 0x0b, 0x8f,\n\t\t0x89, 0xd5, 0xf7, 0x9a, 0x9c, 0xfd, 0x3e, 0x0c, 0x08, 0x0f, 0x55, 0x05, 0x8e, 0x9e, 0xc9, 0x49,\n\t\t0x08, 0x6c, 0x48, 0x00, 0x32, 0x61, 0x56, 0x86, 0x09, 0x23, 0x2f, 0x89, 0xc5, 0x23, 0xff, 0xb4,\n\t\t0xa9, 0x10, 0xaa, 0x5f, 0x44, 0xc9, 0xff, 0xa7, 0x47, 0x89, 0x21, 0x10, 0xe5, 0x18, 0x60, 0x4c,\n\t\t0x1f, 0x27, 0xcc, 0x66, 0xc5, 0xe1, 0xc0, 0x37, 0x15, 0x87, 0x7f, 0xd1, 0x60, 0xa6, 0xc3, 0x0c,\n\t\t0x2a, 0xf6, 0x3e, 0x84, 0xd1, 0xf8, 0x78, 0x41, 0xe8, 0xc4, 0x65, 0x53, 0x97, 0x02, 0x50, 0x9d,\n\t\t0x23, 0x02, 0xa0, 0x1d, 0x18, 0x6f, 0xd5, 0x0f, 0xb1, 0x95, 0xb1, 0xf4, 0x6e, 0x7a, 0x21, 0xb6,\n\t\t0x31, 0x76, 0xdc, 0x3a, 0xd4, 0xbf, 0xd2, 0x60, 0x2e, 0xce, 0x16, 0x8d, 0x46, 0x50, 0x17, 0x7f,\n\t\t0x69, 0xeb, 0x2c, 0xe5, 0xce, 0xd7, 0x59, 0x7a, 0x02, 0xe3, 0x0d, 0x6c, 0xb3, 0xbd, 0x35, 0x9e,\n\t\t0xd2, 0xde, 0x8a, 0x19, 0xc8, 0xf6, 0x16, 0x6f, 0x19, 0x45, 0x45, 0x14, 0x75, 0x2d, 0x27, 0xb4,\n\t\t0x89, 0xd9, 0x64, 0x18, 0x70, 0xcc, 0x43, 0x79, 0x3d, 0x0d, 0x1b, 0x33, 0x6a, 0x3d, 0x66, 0xb2,\n\t\t0x27, 0x16, 0xf5, 0x7f, 0xe4, 0xa0, 0x70, 0xf6, 0xc4, 0xca, 0x34, 0xef, 0xc1, 0x90, 0xef, 0x39,\n\t\t0x0e, 0x61, 0x41, 0x41, 0x13, 0x21, 0xbe, 0x94, 0x6c, 0x15, 0x41, 0x23, 0xc2, 0x2f, 0xa6, 0x47,\n\t\t0xcf, 0x60, 0xe2, 0x8c, 0x20, 0x52, 0x39, 0xd7, 0x33, 0xcf, 0x26, 0xc5, 0x32, 0xc6, 0x79, 0xdb,\n\t\t0x18, 0xbd, 0x80, 0x09, 0x1f, 0x33, 0x4e, 0x5b, 0x12, 0xb4, 0x0a, 0xa4, 0x5b, 0x99, 0xec, 0x76,\n\t\t0x63, 0x90, 0xcc, 0xc0, 0xc6, 0x15, 0xbf, 0x7d, 0xe2, 0x32, 0x7d, 0x41, 0xfd, 0x0e, 0xcc, 0x3f,\n\t\t0x21, 0x3c, 0x5e, 0x08, 0x1e, 0xd6, 0x1f, 0x09, 0x8f, 0xe8, 0xe2, 0x30, 0xfa, 0xef, 0xfa, 0xe1,\n\t\t0x5a, 0x32, 0x4e, 0xa9, 0xfd, 0xe7, 0x30, 0xdb, 0xa8, 0x88, 0x9b, 0x4a, 0xac, 0x61, 0x5f, 0x59,\n\t\t0xe1, 0x87, 0x89, 0x02, 0x66, 0xb1, 0x2c, 0xc5, 0xe9, 0x30, 0xa6, 0x78, 0x86, 0xfd, 0xc7, 0x2e,\n\t\t0x67, 0x75, 0x63, 0xca, 0x3e, 0xbb, 0x12, 0x09, 0xa0, 0x2e, 0x8d, 0x7a, 0x87, 0x00, 0xb9, 0x8b,\n\t\t0x0a, 0x10, 0x5f, 0x2b, 0x67, 0x05, 0xc0, 0x67, 0x57, 0x8a, 0x61, 0xe4, 0x94, 0xc9, 0x12, 0xa3,\n\t\t0x09, 0xe8, 0x3b, 0x22, 0x75, 0xa5, 0xd3, 0xe8, 0x27, 0x2a, 0xc3, 0xc0, 0x09, 0x76, 0x42, 0xa2,\n\t\t0x1c, 0xec, 0x76, 0xa2, 0x74, 0x69, 0x4e, 0x6e, 0x48, 0xec, 0x83, 0xdc, 0x7d, 0x2d, 0xda, 0x36,\n\t\t0x4d, 0xce, 0x6f, 0x71, 0x5b, 0x3d, 0x80, 0x05, 0x11, 0xc8, 0x9d, 0x3e, 0x1b, 0x7c, 0x8b, 0xa9,\n\t\t0x47, 0xff, 0x3c, 0x07, 0x8b, 0x69, 0xbb, 0x2a, 0x3f, 0x3c, 0x86, 0x85, 0x04, 0x37, 0x68, 0x44,\n\t\t0x50, 0x9c, 0x14, 0x4a, 0xbd, 0x45, 0xe0, 0x33, 0xc2, 0xb1, 0x8d, 0x39, 0x36, 0x8a, 0x9d, 0x16,\n\t\t0x6f, 0x6e, 0x1d, 0x6d, 0x99, 0xe0, 0xfa, 0x2d, 0x5b, 0xe6, 0x2e, 0xb6, 0x65, 0xa7, 0x97, 0x37,\n\t\t0xb7, 0xd4, 0xe7, 0x60, 0xe6, 0x09, 0xe1, 0xaa, 0x7d, 0x27, 0x92, 0x98, 0x7a, 0x72, 0xff, 0x52,\n\t\t0x83, 0xd9, 0xce, 0x15, 0xa5, 0x99, 0x43, 0xb8, 0x1a, 0x84, 0xbe, 0xef, 0x31, 0x4e, 0x6c, 0xd3,\n\t\t0x72, 0x68, 0xf4, 0x5c, 0x3d, 0x21, 0x2c, 0x50, 0x5a, 0x49, 0xcf, 0x4b, 0x7b, 0x31, 0xaa, 0x2c,\n\t\t0x40, 0xcf, 0x15, 0xc6, 0x98, 0x0b, 0x92, 0x17, 0xf4, 0xaf, 0xfa, 0x40, 0x7f, 0x92, 0xf0, 0x28,\n\t\t0xfd, 0x48, 0xfe, 0x15, 0xf8, 0x9a, 0x8a, 0x99, 0x79, 0x18, 0xf1, 0x71, 0x95, 0x98, 0x01, 0xfd,\n\t\t0x44, 0x5e, 0x59, 0x03, 0xc6, 0x70, 0x34, 0xb1, 0x47, 0x3f, 0x21, 0xe8, 0x6d, 0xb8, 0xe2, 0x92,\n\t\t0x57, 0x91, 0xd5, 0xaa, 0xc4, 0xe4, 0xde, 0x11, 0x71, 0x55, 0x63, 0x66, 0x2c, 0x9a, 0xde, 0xc5,\n\t\t0x55, 0xb2, 0x1f, 0x4d, 0xa2, 0x77, 0x00, 0x9d, 0x62, 0xca, 0xcd, 0x8a, 0xc7, 0x4c, 0x97, 0x9c,\n\t\t0xca, 0x57, 0xbf, 0xa8, 0x38, 0x86, 0x8d, 0x2b, 0xd1, 0xca, 0xb6, 0xc7, 0x3e, 0x26, 0xa7, 0xe2,\n\t\t0xb9, 0x8f, 0x4c, 0xb8, 0xaa, 0xfe, 0xfd, 0x54, 0xdd, 0x81, 0x0a, 0x75, 0x38, 0x61, 0xf2, 0xd2,\n\t\t0x1c, 0x14, 0x97, 0xe6, 0x5b, 0x89, 0xe7, 0x11, 0xf0, 0x6d, 0x41, 0x2c, 0xee, 0xcd, 0x59, 0xc5,\n\t\t0xa6, 0x63, 0x1e, 0x5d, 0x87, 0x31, 0xd1, 0x2e, 0xc0, 0xcc, 0x3a, 0xa4, 0x27, 0x58, 0x36, 0xdc,\n\t\t0x86, 0x8d, 0xd1, 0x68, 0x72, 0x4b, 0xcd, 0x65, 0x55, 0x4a, 0xc3, 0xdf, 0x54, 0xa5, 0xf4, 0x6f,\n\t\t0x0d, 0xae, 0x67, 0x5a, 0x5c, 0xf9, 0xe0, 0x5d, 0x18, 0x52, 0x47, 0xc9, 0x2c, 0x99, 0x62, 0x58,\n\t\t0x4c, 0x8c, 0x7e, 0x00, 0x79, 0x86, 0x4f, 0xcd, 0x18, 0x2b, 0x03, 0x2a, 0x39, 0x6d, 0x3c, 0xc2,\n\t\t0x1c, 0x3f, 0x74, 0xbc, 0x03, 0x03, 0x18, 0x3e, 0x55, 0x8c, 0x92, 0xcc, 0xdb, 0x97, 0x64, 0xde,\n\t\t0x22, 0x0c, 0x4b, 0x5d, 0x12, 0x5b, 0x95, 0x20, 0x8d, 0xb1, 0xfe, 0x37, 0x0d, 0x46, 0xb7, 0x09,\n\t\t0xe6, 0x21, 0x23, 0xdb, 0x0e, 0xae, 0x06, 0x88, 0xc2, 0x46, 0xc2, 0x93, 0x08, 0x3b, 0x8c, 0x60,\n\t\t0x3b, 0xd2, 0x76, 0xcd, 0x77, 0x48, 0x14, 0x6b, 0x84, 0x31, 0x8f, 0x99, 0xc4, 0xc5, 0x07, 0x0e,\n\t\t0x91, 0x6d, 0x98, 0x61, 0xe3, 0xf6, 0xd9, 0x4e, 0x98, 0xc4, 0x95, 0x63, 0xd8, 0xe3, 0x08, 0xf5,\n\t\t0x58, 0x82, 0xd0, 0x1d, 0x98, 0xc5, 0x21, 0xf7, 0x2a, 0x1e, 0x3b, 0xc5, 0x4c, 0x3c, 0x36, 0x62,\n\t\t0x76, 0x39, 0x59, 0x28, 0xb5, 0xaf, 0x2a, 0x98, 0xfe, 0x1b, 0x0d, 0xe6, 0x0d, 0x52, 0x61, 0x24,\n\t\t0x38, 0x6c, 0xfc, 0xcd, 0x88, 0x83, 0xa3, 0xe0, 0xf5, 0x44, 0xa0, 0xbe, 0x08, 0xd7, 0x92, 0xa5,\n\t\t0x91, 0xde, 0xb1, 0xf1, 0xc5, 0x14, 0xe4, 0xe3, 0x95, 0xad, 0xdd, 0x1d, 0xf4, 0x2b, 0x0d, 0x0a,\n\t\t0x69, 0x4d, 0x7b, 0xf4, 0x6e, 0xca, 0x5f, 0x67, 0x99, 0x7f, 0xf9, 0x17, 0xef, 0x9c, 0x13, 0xa5,\n\t\t0xfc, 0xf6, 0x17, 0x1a, 0xcc, 0x26, 0x37, 0x30, 0xd1, 0x05, 0xda, 0xcd, 0xc5, 0xcd, 0x73, 0x61,\n\t\t0x94, 0x0c, 0xbf, 0xd7, 0x60, 0x3e, 0xa3, 0x89, 0x8a, 0xee, 0x9d, 0x83, 0x69, 0x6b, 0xfb, 0xb7,\n\t\t0x78, 0xff, 0xfc, 0x40, 0x25, 0xd2, 0x67, 0x1a, 0xcc, 0xa5, 0x74, 0xf4, 0xd1, 0x66, 0x56, 0x0f,\n\t\t0x39, 0x4d, 0x31, 0xef, 0x9e, 0x0f, 0xa4, 0xc4, 0xf8, 0xb3, 0x06, 0xcb, 0xdd, 0x1a, 0xcd, 0xe8,\n\t\t0x52, 0x3d, 0xed, 0xe2, 0xf7, 0x2f, 0x88, 0x56, 0x12, 0x7e, 0xa9, 0xc1, 0x8d, 0x9e, 0x5a, 0xe1,\n\t\t0x68, 0xeb, 0x42, 0x1b, 0xb5, 0xd9, 0xf3, 0xe1, 0x65, 0x58, 0xb4, 0x38, 0x7c, 0x72, 0x67, 0x39,\n\t\t0xc5, 0xe1, 0x33, 0xdb, 0xf6, 0x29, 0x0e, 0xdf, 0xa5, 0x75, 0xfd, 0x47, 0x0d, 0x16, 0xb3, 0xbb,\n\t\t0xb0, 0xe8, 0x41, 0x0a, 0xdf, 0x1e, 0x1a, 0xdb, 0xc5, 0xf7, 0x2f, 0x84, 0x55, 0xb2, 0xfd, 0x56,\n\t\t0x83, 0x62, 0x7a, 0x07, 0x13, 0xdd, 0x4d, 0xae, 0xf5, 0xba, 0xf5, 0x89, 0x8b, 0xf7, 0xce, 0x8d,\n\t\t0x53, 0xf2, 0xfc, 0x5a, 0x83, 0xab, 0xa9, 0x2d, 0x43, 0x74, 0x27, 0xb3, 0xcc, 0x4f, 0x95, 0xe6,\n\t\t0xee, 0x79, 0x61, 0x4a, 0x98, 0x0a, 0x8c, 0xb5, 0xb5, 0x4d, 0x50, 0x46, 0xb7, 0xa7, 0xa3, 0xc3,\n\t\t0x55, 0xbc, 0xd9, 0x0b, 0xa9, 0xda, 0xc7, 0x83, 0x89, 0xce, 0xa7, 0x0a, 0xba, 0xd5, 0xe3, 0x8b,\n\t\t0x46, 0xee, 0x76, 0xbe, 0xf7, 0x0f, 0xfa, 0x14, 0xa6, 0x93, 0x1e, 0x8c, 0xe8, 0x7b, 0xe7, 0x78,\n\t\t0x5b, 0xca, 0x8d, 0xd7, 0xcf, 0xfd, 0x1a, 0x15, 0x21, 0x99, 0xfc, 0xf8, 0x49, 0x09, 0xc9, 0xcc,\n\t\t0xf7, 0x59, 0x4a, 0x48, 0x76, 0x79, 0x5d, 0x51, 0x18, 0x6f, 0x7f, 0x5d, 0xa0, 0x9b, 0x69, 0x07,\n\t\t0x39, 0xfb, 0x38, 0x29, 0xbe, 0xd3, 0x13, 0x6d, 0xcb, 0x75, 0x97, 0x51, 0x52, 0xa6, 0x5c, 0x77,\n\t\t0xdd, 0x9f, 0x1d, 0x29, 0xd7, 0x5d, 0x2f, 0xd5, 0xeb, 0xa7, 0x30, 0x9d, 0x54, 0xbf, 0xa4, 0x98,\n\t\t0x3f, 0xa3, 0xf0, 0x4a, 0x31, 0x7f, 0x56, 0x71, 0x24, 0x23, 0x3c, 0xed, 0xcb, 0xba, 0xb4, 0x08,\n\t\t0xef, 0xf2, 0xe5, 0x60, 0x5a, 0x84, 0x77, 0xfb, 0x80, 0xef, 0xa1, 0x0d, 0x73, 0x96, 0x57, 0x4b,\n\t\t0x02, 0x3f, 0x9c, 0x8e, 0x51, 0x7b, 0xf2, 0xe3, 0xd4, 0x5d, 0xe6, 0x71, 0x6f, 0x57, 0xfb, 0xd9,\n\t\t0x7a, 0x95, 0xf2, 0xc3, 0xf0, 0xa0, 0x64, 0x79, 0xb5, 0xb5, 0xd6, 0xef, 0x37, 0x6f, 0x53, 0xdb,\n\t\t0x59, 0xab, 0x7a, 0xf2, 0xbb, 0x53, 0xf5, 0x31, 0xe7, 0xfb, 0xd8, 0xa7, 0x27, 0xeb, 0x07, 0x83,\n\t\t0x62, 0x6e, 0xf3, 0xbf, 0x01, 0x00, 0x00, 0xff, 0xff, 0xbd, 0x51, 0x4a, 0x5f, 0xfc, 0x2a, 0x00,\n\t\t0x00,\n\t},\n\t// uber/cadence/api/v1/service_worker.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe4, 0x5a, 0x5b, 0x6f, 0xdb, 0xc8,\n\t\t0xf5, 0x07, 0x7d, 0xf7, 0xf1, 0x4d, 0x1e, 0xe3, 0xef, 0xbf, 0xac, 0xdc, 0x1c, 0xa5, 0x49, 0xbc,\n\t\t0x6d, 0x2a, 0x37, 0xde, 0x6d, 0x36, 0x9b, 0x6c, 0x8a, 0xfa, 0x12, 0x23, 0x2e, 0x36, 0x5b, 0x2f,\n\t\t0xa3, 0x4d, 0x80, 0x2d, 0x10, 0x62, 0x44, 0x8e, 0xac, 0x81, 0x29, 0x8e, 0x42, 0x0e, 0xa5, 0xe8,\n\t\t0xa5, 0x0f, 0x7d, 0xdc, 0xf6, 0xa1, 0x40, 0xd1, 0xed, 0x4b, 0x81, 0x3c, 0xf7, 0x03, 0xf4, 0xcb,\n\t\t0xb4, 0x4f, 0x7d, 0xee, 0x67, 0x28, 0x50, 0x70, 0x66, 0x48, 0x51, 0xd2, 0x90, 0xba, 0xb4, 0x80,\n\t\t0x17, 0xe8, 0x9b, 0x39, 0xfc, 0x9d, 0xc3, 0x33, 0xe7, 0x77, 0xe6, 0xcc, 0x6f, 0xc6, 0x82, 0xbd,\n\t\t0xb0, 0x46, 0xfc, 0x7d, 0x1b, 0x3b, 0xc4, 0xb3, 0xc9, 0x3e, 0x6e, 0xd1, 0xfd, 0xf6, 0xc3, 0xfd,\n\t\t0x80, 0xf8, 0x6d, 0x6a, 0x13, 0xab, 0xc3, 0xfc, 0x4b, 0xe2, 0x57, 0x5a, 0x3e, 0xe3, 0x0c, 0x6d,\n\t\t0x45, 0xc8, 0x8a, 0x42, 0x56, 0x70, 0x8b, 0x56, 0xda, 0x0f, 0x4b, 0x37, 0x2f, 0x18, 0xbb, 0x70,\n\t\t0xc9, 0xbe, 0x80, 0xd4, 0xc2, 0xfa, 0xbe, 0x13, 0xfa, 0x98, 0x53, 0xe6, 0x49, 0xa3, 0xd2, 0xad,\n\t\t0xc1, 0xf7, 0x9c, 0x36, 0x49, 0xc0, 0x71, 0xb3, 0xa5, 0x00, 0x43, 0x0e, 0x3a, 0x3e, 0x6e, 0xb5,\n\t\t0x88, 0x1f, 0xa8, 0xf7, 0xbb, 0xba, 0xf8, 0x6c, 0xd6, 0x6c, 0x26, 0x9f, 0x28, 0xeb, 0x10, 0x0e,\n\t\t0xb1, 0x69, 0xd0, 0x0b, 0xe3, 0xb6, 0x0e, 0xd3, 0xa0, 0x01, 0x67, 0x7e, 0x37, 0x8e, 0x54, 0x07,\n\t\t0x79, 0x17, 0x92, 0x04, 0xa0, 0xfd, 0x0e, 0xc7, 0xc1, 0xa5, 0x4b, 0x03, 0x9e, 0x87, 0x89, 0xb2,\n\t\t0x58, 0x77, 0x59, 0x47, 0x62, 0xca, 0x7f, 0x35, 0xa0, 0x74, 0xce, 0x5c, 0xf7, 0x94, 0xf9, 0x27,\n\t\t0x2a, 0xca, 0x2a, 0x0e, 0x2e, 0x4d, 0xf2, 0x2e, 0x24, 0x01, 0x47, 0xdb, 0xb0, 0xe0, 0xb0, 0x26,\n\t\t0xa6, 0x5e, 0xd1, 0xd8, 0x35, 0xf6, 0x96, 0x4d, 0xf5, 0x84, 0x9e, 0xc0, 0x72, 0xf4, 0x31, 0x2b,\n\t\t0xfa, 0x5a, 0x71, 0x66, 0xd7, 0xd8, 0x5b, 0x39, 0xb8, 0x51, 0xd1, 0x50, 0x52, 0x89, 0x9c, 0x7d,\n\t\t0x41, 0x03, 0x6e, 0x2e, 0x71, 0xf5, 0x17, 0x2a, 0xc1, 0x12, 0x75, 0x88, 0xc7, 0x29, 0xef, 0x16,\n\t\t0x67, 0x85, 0xd7, 0xe4, 0x19, 0xdd, 0x87, 0x8d, 0x1a, 0xf5, 0xb0, 0xdf, 0xb5, 0xec, 0x06, 0xb1,\n\t\t0x2f, 0x83, 0xb0, 0x59, 0x9c, 0x13, 0x90, 0x75, 0x39, 0x7c, 0xac, 0x46, 0xcb, 0xff, 0x5a, 0x82,\n\t\t0x6b, 0xda, 0xb8, 0x83, 0x16, 0xf3, 0x02, 0x82, 0x6e, 0x00, 0x88, 0x00, 0x39, 0xbb, 0x24, 0x32,\n\t\t0xf8, 0x55, 0x53, 0x84, 0x5c, 0x8d, 0x06, 0xd0, 0xd7, 0x80, 0xe2, 0x44, 0x58, 0xe4, 0x3d, 0xb1,\n\t\t0xc3, 0xa8, 0x4a, 0xd4, 0x44, 0xee, 0x69, 0x27, 0xf2, 0x46, 0xc1, 0x9f, 0xc7, 0x68, 0x73, 0xb3,\n\t\t0x33, 0x38, 0x84, 0x4e, 0x61, 0x2d, 0x71, 0xcb, 0xbb, 0x2d, 0x22, 0xe6, 0xb7, 0x72, 0x70, 0x3b,\n\t\t0xd7, 0x63, 0xb5, 0xdb, 0x22, 0xe6, 0x6a, 0x27, 0xf5, 0x84, 0x5e, 0xc3, 0x4e, 0xcb, 0x27, 0x6d,\n\t\t0xca, 0xc2, 0xc0, 0x0a, 0x38, 0xf6, 0x39, 0x71, 0x2c, 0xd2, 0x26, 0x1e, 0xb7, 0xa8, 0x23, 0x12,\n\t\t0xb2, 0x72, 0x70, 0xad, 0x22, 0x6b, 0xb5, 0x12, 0xd7, 0x6a, 0xe5, 0xcc, 0xe3, 0x8f, 0x3e, 0x79,\n\t\t0x8d, 0xdd, 0x90, 0x98, 0xdb, 0xb1, 0xf5, 0x2b, 0x69, 0xfc, 0x3c, 0xb2, 0x3d, 0x73, 0xd0, 0x1e,\n\t\t0x14, 0x86, 0xdc, 0xcd, 0xef, 0x1a, 0x7b, 0xb3, 0xe6, 0x7a, 0xd0, 0x8f, 0x2c, 0xc2, 0x22, 0xe6,\n\t\t0x9c, 0x34, 0x5b, 0xbc, 0xb8, 0x20, 0x00, 0xf1, 0x23, 0x7a, 0x00, 0xa8, 0x86, 0xed, 0x4b, 0x97,\n\t\t0x5d, 0x58, 0x36, 0x0b, 0x3d, 0x6e, 0x35, 0xa8, 0xc7, 0x8b, 0x8b, 0x02, 0x54, 0x50, 0x6f, 0x8e,\n\t\t0xa3, 0x17, 0x2f, 0xa8, 0xc7, 0xd1, 0x23, 0x58, 0x54, 0x95, 0x5d, 0x5c, 0x12, 0x71, 0x5f, 0xd7,\n\t\t0xe6, 0xe2, 0x85, 0xc4, 0x98, 0x31, 0x18, 0xdd, 0x83, 0x0d, 0x8f, 0xbc, 0xe7, 0x56, 0x0b, 0x5f,\n\t\t0x10, 0x45, 0xe2, 0xb2, 0x20, 0x71, 0x2d, 0x1a, 0x3e, 0xc7, 0x17, 0x44, 0x12, 0xf9, 0x18, 0xe6,\n\t\t0xc5, 0xb2, 0x28, 0x82, 0xf0, 0x5e, 0xce, 0xcd, 0xf4, 0x57, 0x11, 0xd2, 0x94, 0x06, 0xe8, 0x2d,\n\t\t0x5c, 0x1f, 0x2e, 0x01, 0xab, 0x57, 0xd5, 0x2b, 0xe3, 0x54, 0xf5, 0xce, 0x50, 0x0d, 0xc4, 0xaf,\n\t\t0xd0, 0x21, 0xac, 0x07, 0x76, 0x83, 0x38, 0xa1, 0x4b, 0x1c, 0x2b, 0x6a, 0x34, 0xc5, 0x55, 0xe1,\n\t\t0xb1, 0x34, 0x44, 0x5c, 0x35, 0xee, 0x42, 0xe6, 0x5a, 0x62, 0x11, 0x8d, 0xa1, 0x67, 0xb0, 0x1a,\n\t\t0xd3, 0x25, 0x1c, 0xac, 0x8d, 0x74, 0xb0, 0xa2, 0xf0, 0xc2, 0xfc, 0x0d, 0x2c, 0x46, 0x53, 0xa5,\n\t\t0x24, 0x28, 0xae, 0xef, 0xce, 0xee, 0xad, 0x1c, 0x3c, 0xd3, 0x4e, 0x26, 0x67, 0x19, 0x55, 0xbe,\n\t\t0x92, 0xf6, 0xcf, 0x3d, 0x1e, 0x91, 0xa3, 0xbc, 0xa1, 0x32, 0x08, 0x16, 0x7a, 0x35, 0xb4, 0x21,\n\t\t0xd8, 0x5f, 0x89, 0x06, 0xe3, 0x02, 0xaa, 0xc0, 0x16, 0x67, 0x1c, 0xbb, 0x96, 0x62, 0xd4, 0xaa,\n\t\t0x75, 0x39, 0x09, 0x8a, 0x05, 0x81, 0xdc, 0x14, 0xaf, 0x14, 0xe9, 0x47, 0xd1, 0x0b, 0xf4, 0x12,\n\t\t0x0a, 0x38, 0xe4, 0xcc, 0xb2, 0x99, 0x57, 0xa7, 0x17, 0xb2, 0xa8, 0x36, 0xc5, 0x7c, 0xef, 0x68,\n\t\t0xa3, 0x3e, 0x0c, 0x39, 0x3b, 0x16, 0xd8, 0xa8, 0xce, 0xcc, 0x75, 0xdc, 0xf7, 0x5c, 0x7a, 0x0b,\n\t\t0xab, 0xe9, 0xd8, 0x51, 0x01, 0x66, 0x2f, 0x49, 0x57, 0x75, 0xb1, 0xe8, 0xcf, 0xa8, 0x72, 0xda,\n\t\t0xd1, 0x62, 0x51, 0xab, 0x7e, 0xac, 0xca, 0x11, 0x06, 0x4f, 0x66, 0x1e, 0x1b, 0xe5, 0xbf, 0xcc,\n\t\t0xc3, 0x1d, 0x99, 0x25, 0x27, 0x9d, 0xb8, 0x63, 0xd6, 0x6c, 0xb9, 0x84, 0x13, 0x27, 0x6e, 0xa0,\n\t\t0x23, 0xfa, 0xd0, 0x53, 0x58, 0x8e, 0x37, 0x87, 0xa0, 0x38, 0x23, 0x48, 0xd2, 0x57, 0x5c, 0xfc,\n\t\t0x11, 0xb3, 0x87, 0x47, 0x3f, 0x82, 0xcd, 0x5e, 0xe1, 0xda, 0xcc, 0xe3, 0xe4, 0x3d, 0x17, 0x1d,\n\t\t0x67, 0xd5, 0x2c, 0x24, 0x2f, 0x8e, 0xe5, 0x78, 0x5f, 0xd7, 0x9d, 0x1b, 0xe8, 0xba, 0xbf, 0x82,\n\t\t0xcd, 0x80, 0x53, 0xfb, 0xb2, 0x6b, 0x61, 0xce, 0x7d, 0x5a, 0x0b, 0x23, 0xa6, 0xe6, 0x45, 0x5a,\n\t\t0x2a, 0xda, 0x68, 0x5e, 0x09, 0x74, 0x52, 0xf3, 0x87, 0x89, 0x95, 0x59, 0x90, 0x8e, 0x7a, 0x23,\n\t\t0xe8, 0x53, 0x28, 0xfa, 0x84, 0x87, 0xbe, 0x67, 0x79, 0xa4, 0x63, 0xc5, 0xd1, 0x8b, 0x85, 0x26,\n\t\t0x5a, 0xcb, 0x92, 0xf9, 0x7f, 0xf2, 0xfd, 0x97, 0xa4, 0x93, 0x4e, 0x25, 0x3a, 0x82, 0x9b, 0x75,\n\t\t0xe6, 0xdb, 0xc4, 0xb2, 0x7d, 0x82, 0x39, 0xd1, 0x98, 0x2f, 0x0a, 0xf3, 0x92, 0x40, 0x1d, 0x0b,\n\t\t0xd0, 0xa0, 0x0f, 0xcd, 0x7e, 0xb2, 0xa4, 0xdb, 0x4f, 0x10, 0x83, 0x35, 0xd1, 0x16, 0x2c, 0x9f,\n\t\t0x04, 0xa1, 0xcb, 0x83, 0xe2, 0xb2, 0x20, 0xe3, 0x17, 0xda, 0xe9, 0x8f, 0x41, 0x7c, 0x45, 0x56,\n\t\t0x8c, 0x74, 0x26, 0x97, 0xcf, 0xea, 0xbb, 0xd4, 0x50, 0x89, 0xc2, 0xe6, 0x10, 0x44, 0x53, 0xa5,\n\t\t0x3f, 0xeb, 0xaf, 0xd2, 0xbd, 0x31, 0xaa, 0x54, 0x38, 0x4c, 0xd7, 0xea, 0x87, 0x59, 0xf8, 0x41,\n\t\t0x7e, 0xc8, 0x6a, 0xd3, 0xfc, 0x1a, 0xd6, 0xfa, 0x13, 0x6c, 0x88, 0x8f, 0xfe, 0x64, 0xd2, 0xb6,\n\t\t0x61, 0xae, 0x3a, 0x69, 0x12, 0x3e, 0x18, 0x70, 0x13, 0xdb, 0x9c, 0xb6, 0x29, 0xa7, 0x24, 0xb0,\n\t\t0x38, 0xb3, 0x1c, 0x1a, 0xb4, 0x30, 0xb7, 0x1b, 0x96, 0xcb, 0x6c, 0xec, 0xba, 0x5d, 0x55, 0xfa,\n\t\t0xdf, 0x4c, 0x91, 0x6d, 0xd5, 0xa8, 0x0e, 0x13, 0xff, 0x55, 0x76, 0xa2, 0xbc, 0x7f, 0x21, 0x9d,\n\t\t0xcb, 0xec, 0x5f, 0xc3, 0xd9, 0x88, 0xd2, 0xaf, 0x61, 0x77, 0x94, 0x03, 0x0d, 0x37, 0x27, 0xfd,\n\t\t0xdc, 0xe8, 0x97, 0x8a, 0xf2, 0xdb, 0x15, 0xbe, 0x62, 0xc7, 0x67, 0x5e, 0x9d, 0xa5, 0x19, 0xfa,\n\t\t0xcd, 0x0c, 0xec, 0x6a, 0xa6, 0x79, 0x8a, 0xa9, 0x3b, 0x76, 0x2b, 0x39, 0x82, 0x79, 0x1b, 0x87,\n\t\t0x81, 0x8c, 0x66, 0xfd, 0xe0, 0x41, 0x6e, 0x1b, 0xe9, 0x79, 0x3f, 0x8e, 0x6c, 0x4c, 0x69, 0x1a,\n\t\t0xed, 0xd6, 0x0e, 0xe1, 0x98, 0xba, 0x81, 0x52, 0x2e, 0xfa, 0xdd, 0xfa, 0x1c, 0x77, 0x5d, 0x86,\n\t\t0x1d, 0x33, 0x06, 0xe7, 0x36, 0x17, 0xcd, 0x12, 0x9c, 0xd7, 0x4a, 0xba, 0x3b, 0x70, 0x3b, 0x27,\n\t\t0x07, 0x92, 0xe7, 0xf2, 0x3f, 0x7a, 0x7a, 0x35, 0xce, 0xec, 0x55, 0xea, 0xd5, 0x57, 0x80, 0x12,\n\t\t0xbf, 0x56, 0x93, 0x70, 0xec, 0x60, 0x8e, 0x95, 0x42, 0xbb, 0x9b, 0xfb, 0x81, 0x97, 0x0a, 0x6c,\n\t\t0x16, 0xf8, 0xc0, 0x48, 0xf9, 0x6f, 0x3d, 0x6d, 0xdb, 0x3f, 0xc7, 0x2b, 0xd5, 0xb6, 0xb7, 0x60,\n\t\t0x45, 0x2d, 0xa1, 0x6e, 0xb4, 0xe5, 0xcb, 0x4c, 0x40, 0x3c, 0x74, 0xe6, 0x44, 0xe2, 0x37, 0x01,\n\t\t0x08, 0xf1, 0x3b, 0x97, 0x23, 0x7e, 0x93, 0x89, 0x09, 0xf1, 0x8b, 0x53, 0x4f, 0xe8, 0x00, 0xe6,\n\t\t0xa9, 0xd7, 0x0a, 0xb9, 0xda, 0x81, 0xf2, 0x4b, 0x50, 0x42, 0x35, 0x62, 0x6b, 0xe1, 0x3f, 0x15,\n\t\t0x5b, 0x8b, 0x93, 0x89, 0xad, 0x2a, 0xec, 0xc4, 0xfe, 0xa2, 0x0e, 0x67, 0xbb, 0x2c, 0x20, 0xc2,\n\t\t0x11, 0x0b, 0xb9, 0x92, 0xbe, 0x3b, 0x43, 0xbe, 0x4e, 0xd4, 0xf9, 0xd4, 0xdc, 0x8e, 0x6d, 0xab,\n\t\t0xec, 0x38, 0xb2, 0xac, 0x4a, 0x43, 0xf4, 0x25, 0x6c, 0x8b, 0x8f, 0x0c, 0xbb, 0x5c, 0x1e, 0xe5,\n\t\t0x72, 0x4b, 0x18, 0x0e, 0xf8, 0x3b, 0x85, 0xcd, 0x06, 0xc1, 0x3e, 0xaf, 0x11, 0xcc, 0x13, 0x57,\n\t\t0x30, 0xca, 0x55, 0x21, 0xb1, 0x89, 0xfd, 0xa4, 0x8e, 0x07, 0x91, 0x4e, 0x9e, 0xef, 0x1d, 0x0f,\n\t\t0xde, 0xc2, 0xcd, 0x7e, 0x26, 0x2c, 0x56, 0xb7, 0x78, 0x83, 0x06, 0x56, 0x6c, 0x30, 0x5a, 0x06,\n\t\t0x97, 0xfa, 0x98, 0xf9, 0x65, 0xbd, 0xda, 0xa0, 0xc1, 0xa1, 0xf2, 0x7f, 0x96, 0x9e, 0x41, 0xdc,\n\t\t0xac, 0xd6, 0xc6, 0xa8, 0x94, 0xde, 0x24, 0x4e, 0x54, 0xd7, 0x1a, 0x3a, 0xad, 0xad, 0x4f, 0x77,\n\t\t0x5a, 0xbb, 0x0f, 0x1b, 0x89, 0x1f, 0xd5, 0x7d, 0x36, 0x64, 0x87, 0x8b, 0x87, 0x4f, 0x64, 0x17,\n\t\t0xfa, 0x18, 0x16, 0x1a, 0x04, 0x3b, 0xc4, 0x17, 0x32, 0x38, 0x3a, 0xc3, 0x69, 0xcf, 0x42, 0x02,\n\t\t0x62, 0x2a, 0xe8, 0x7f, 0x59, 0x18, 0x97, 0xbf, 0x33, 0x12, 0xe1, 0x9a, 0x6e, 0x2e, 0x93, 0x0a,\n\t\t0xd7, 0x4f, 0x60, 0x41, 0x2a, 0x25, 0xd5, 0x58, 0xf2, 0x73, 0xaf, 0xb0, 0x79, 0xad, 0xb4, 0x7c,\n\t\t0x2f, 0x11, 0x29, 0x19, 0x71, 0xa9, 0x1d, 0xe0, 0xb7, 0x33, 0x70, 0x3f, 0x0f, 0x78, 0xd4, 0x3d,\n\t\t0x3b, 0x19, 0xb5, 0x1d, 0x5c, 0x55, 0x8b, 0xec, 0x65, 0x6d, 0x6e, 0xca, 0xac, 0xcd, 0x0f, 0x64,\n\t\t0xed, 0x87, 0xb0, 0x37, 0x3a, 0x19, 0x2a, 0x73, 0x7f, 0x34, 0x12, 0x95, 0x91, 0x06, 0x4f, 0xa4,\n\t\t0x32, 0x1e, 0xc1, 0x62, 0x1d, 0x53, 0x37, 0xf4, 0x49, 0x2e, 0xf1, 0xa7, 0x12, 0x63, 0xc6, 0xe0,\n\t\t0x5c, 0xe6, 0x7b, 0x1b, 0xbf, 0x2e, 0x2c, 0x15, 0xfc, 0xb7, 0x33, 0xda, 0xfa, 0x90, 0xa8, 0xef,\n\t\t0x33, 0xe7, 0xa9, 0x8c, 0xcd, 0x4d, 0x9b, 0xb1, 0x41, 0xd6, 0xef, 0xc3, 0xdd, 0x11, 0xb9, 0x50,\n\t\t0x59, 0xfb, 0x93, 0x01, 0x65, 0x5d, 0x7d, 0x60, 0xcf, 0x26, 0x13, 0x91, 0x1e, 0x77, 0xda, 0x99,\n\t\t0x69, 0x65, 0xe1, 0x20, 0xe9, 0x77, 0xf5, 0x6d, 0x28, 0x09, 0x4c, 0x4d, 0xe0, 0x77, 0x33, 0x70,\n\t\t0x2f, 0x07, 0xf7, 0x3d, 0x27, 0x3e, 0xce, 0xda, 0xdc, 0xb4, 0x59, 0x1b, 0x24, 0xfe, 0x23, 0x7d,\n\t\t0xef, 0xeb, 0xcb, 0x46, 0x1f, 0xf5, 0x36, 0xf3, 0xfb, 0xa0, 0x2f, 0xe2, 0x4d, 0xf0, 0x0a, 0xa9,\n\t\t0x3f, 0x8f, 0xa8, 0xcf, 0x09, 0x4c, 0xc9, 0xdc, 0x8f, 0xa0, 0x60, 0x8b, 0x89, 0x59, 0xbe, 0x8c,\n\t\t0x95, 0x38, 0x22, 0xbe, 0x25, 0x73, 0x43, 0x8e, 0x9b, 0xf1, 0xb0, 0xaa, 0x92, 0x4c, 0x97, 0xff,\n\t\t0x6b, 0x55, 0x52, 0x8d, 0xaa, 0x64, 0x44, 0x36, 0x26, 0x4f, 0xf2, 0xdf, 0x7b, 0xdb, 0x87, 0xb8,\n\t\t0x68, 0x98, 0x46, 0x36, 0xfc, 0x7c, 0x40, 0x36, 0x8c, 0x7f, 0x9f, 0x11, 0x6f, 0x86, 0xaf, 0x61,\n\t\t0x4b, 0xfe, 0x23, 0xc8, 0x6a, 0x13, 0x5f, 0xdc, 0x54, 0x50, 0xaf, 0xce, 0xd4, 0x71, 0x35, 0x9b,\n\t\t0x28, 0xe2, 0xbf, 0x96, 0x70, 0x71, 0xf4, 0xde, 0xec, 0x0c, 0x0e, 0xa5, 0x36, 0x21, 0xdd, 0xe4,\n\t\t0x62, 0xed, 0x61, 0x40, 0xc9, 0x24, 0x01, 0xe1, 0xf2, 0x02, 0x2c, 0x39, 0x2c, 0x5e, 0x49, 0x6d,\n\t\t0x95, 0x6f, 0xc0, 0x35, 0x6d, 0x30, 0x2a, 0x58, 0x1f, 0xd6, 0xfb, 0xb5, 0x20, 0x7a, 0x00, 0x88,\n\t\t0x78, 0xb8, 0xe6, 0x12, 0x2b, 0xa5, 0x28, 0x15, 0xdd, 0x05, 0xf9, 0xa6, 0x67, 0x81, 0x0e, 0x60,\n\t\t0xbb, 0xc5, 0x5c, 0x97, 0xf8, 0x56, 0x07, 0x53, 0x79, 0x5a, 0xb0, 0xa8, 0x67, 0x35, 0x65, 0x27,\n\t\t0x98, 0x35, 0x91, 0x7c, 0xfb, 0x06, 0x53, 0x71, 0x2c, 0x38, 0xf3, 0x5e, 0x06, 0x07, 0xff, 0xdc,\n\t\t0x80, 0x65, 0x99, 0xee, 0xc3, 0xf3, 0x33, 0xf4, 0x1e, 0xb6, 0x34, 0xb7, 0x44, 0x68, 0x7f, 0xfc,\n\t\t0xfb, 0x24, 0x91, 0xd7, 0xd2, 0xc4, 0x17, 0x50, 0xe8, 0x0f, 0x06, 0x5c, 0xcf, 0xbb, 0x37, 0x42,\n\t\t0x8f, 0xa7, 0xbd, 0xd8, 0x2b, 0x7d, 0x36, 0xf5, 0x25, 0x15, 0xfa, 0xd6, 0x80, 0x9d, 0xcc, 0x2b,\n\t\t0x0e, 0xf4, 0xd3, 0x71, 0x1d, 0xf7, 0x09, 0xb6, 0xd2, 0xa3, 0x49, 0xcd, 0x54, 0x30, 0x3d, 0x72,\n\t\t0xd2, 0x5d, 0x22, 0x9f, 0x1c, 0xcd, 0x95, 0x4b, 0x3e, 0x39, 0xda, 0xfb, 0x8b, 0x14, 0x39, 0x5a,\n\t\t0xd1, 0x9a, 0x4f, 0x4e, 0xde, 0xa9, 0x25, 0x9f, 0x9c, 0xdc, 0x73, 0x05, 0xfa, 0xa0, 0x57, 0xc7,\n\t\t0x7d, 0x52, 0x1a, 0x7d, 0x3e, 0xb1, 0xff, 0xd4, 0xde, 0x53, 0x7a, 0x36, 0xa5, 0xf5, 0x70, 0xf9,\n\t\t0x0c, 0xcb, 0xbe, 0xfc, 0xf2, 0xc9, 0xd4, 0xfb, 0xf9, 0xe5, 0x93, 0xad, 0xc7, 0xd1, 0x77, 0x06,\n\t\t0xdc, 0xc8, 0xd5, 0xa0, 0xe8, 0xb3, 0xc9, 0x3c, 0xa7, 0x13, 0xf5, 0x64, 0x1a, 0x53, 0x15, 0xd8,\n\t\t0xef, 0x0d, 0xd1, 0x16, 0xb3, 0x34, 0x12, 0xfa, 0x74, 0x6c, 0x12, 0xfa, 0x45, 0x72, 0xe9, 0xf1,\n\t\t0xe4, 0x86, 0x2a, 0xa4, 0x3f, 0x1b, 0x70, 0x6b, 0x84, 0x6c, 0x43, 0x4f, 0x27, 0xf5, 0x9e, 0xce,\n\t\t0xd7, 0xe7, 0xd3, 0x19, 0xf7, 0x65, 0x2c, 0x53, 0x2f, 0x64, 0x66, 0x6c, 0x94, 0xb6, 0xcc, 0xcc,\n\t\t0xd8, 0x68, 0xed, 0x27, 0x33, 0x96, 0x2b, 0x61, 0x32, 0x33, 0x36, 0x8e, 0x0c, 0xcc, 0xcc, 0xd8,\n\t\t0x78, 0xaa, 0x29, 0xb5, 0x12, 0x87, 0xd5, 0x42, 0xfe, 0x4a, 0xcc, 0x94, 0x4e, 0xf9, 0x2b, 0x31,\n\t\t0x5b, 0x94, 0x44, 0x8d, 0x5c, 0x23, 0x03, 0x32, 0x1a, 0x79, 0xb6, 0x7a, 0xc9, 0x68, 0xe4, 0x39,\n\t\t0x0a, 0xe3, 0xa8, 0x06, 0xff, 0x6f, 0xb3, 0xa6, 0xce, 0xec, 0x08, 0x49, 0x15, 0xf0, 0x4a, 0xfe,\n\t\t0x76, 0xe7, 0xdc, 0x67, 0x9c, 0x9d, 0x1b, 0xdf, 0x3c, 0xbc, 0xa0, 0xbc, 0x11, 0xd6, 0x2a, 0x36,\n\t\t0x6b, 0xee, 0xa7, 0x7f, 0x9c, 0xf2, 0x63, 0xea, 0xb8, 0xfb, 0x17, 0x4c, 0xfe, 0xee, 0x46, 0xfd,\n\t\t0x52, 0xe5, 0x29, 0x6e, 0xd1, 0xf6, 0xc3, 0xda, 0x82, 0x18, 0xfb, 0xf8, 0xdf, 0x01, 0x00, 0x00,\n\t\t0xff, 0xff, 0x83, 0x33, 0x80, 0xed, 0x1b, 0x24, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/decision.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xdc, 0x59, 0xdf, 0x6e, 0xd4, 0xc6,\n\t\t0x17, 0xfe, 0x39, 0x7f, 0x36, 0xbb, 0x27, 0x1b, 0x20, 0x13, 0x08, 0x09, 0x04, 0x12, 0xf6, 0xa7,\n\t\t0x42, 0x21, 0xca, 0x6e, 0x12, 0x28, 0x42, 0x80, 0x50, 0xc9, 0x42, 0x44, 0x24, 0x08, 0x91, 0x13,\n\t\t0x40, 0xaa, 0x54, 0x59, 0x93, 0xf1, 0x24, 0x99, 0xc6, 0xeb, 0xd9, 0x8e, 0xc7, 0x09, 0x5b, 0xa9,\n\t\t0x12, 0x57, 0x6d, 0x6f, 0xfa, 0x00, 0x95, 0x7a, 0xd5, 0x9b, 0xb6, 0x37, 0xed, 0x6d, 0xab, 0x5e,\n\t\t0xf5, 0x11, 0xfa, 0x2c, 0x7d, 0x81, 0xca, 0xe3, 0xb1, 0x77, 0xb3, 0xf1, 0x7a, 0xed, 0x40, 0xb9,\n\t\t0xe8, 0x5d, 0x3c, 0x3e, 0xe7, 0x9b, 0xcf, 0x73, 0xce, 0x7e, 0xe7, 0xb3, 0x03, 0x15, 0x7f, 0x9b,\n\t\t0x8a, 0x1a, 0xc1, 0x36, 0x75, 0x09, 0xad, 0xe1, 0x26, 0xab, 0x1d, 0x2c, 0xd5, 0x6c, 0x4a, 0x98,\n\t\t0xc7, 0xb8, 0x5b, 0x6d, 0x0a, 0x2e, 0x39, 0x9a, 0x08, 0x62, 0xaa, 0x3a, 0xa6, 0x8a, 0x9b, 0xac,\n\t\t0x7a, 0xb0, 0x74, 0xe1, 0xf2, 0x2e, 0xe7, 0xbb, 0x0e, 0xad, 0xa9, 0x90, 0x6d, 0x7f, 0xa7, 0x66,\n\t\t0xfb, 0x02, 0xcb, 0x38, 0xe9, 0xc2, 0x5c, 0x12, 0x30, 0xe1, 0x8d, 0x46, 0x1c, 0x91, 0xb8, 0xb5,\n\t\t0xc4, 0xde, 0xbe, 0xc3, 0x3c, 0x99, 0x16, 0x73, 0xc8, 0xc5, 0xfe, 0x8e, 0xc3, 0x0f, 0xc3, 0x98,\n\t\t0xca, 0x37, 0xe3, 0x50, 0x7c, 0xa4, 0x19, 0xa3, 0xef, 0x0c, 0xb8, 0xe1, 0x91, 0x3d, 0x6a, 0xfb,\n\t\t0x0e, 0xb5, 0x30, 0x91, 0xec, 0x80, 0xc9, 0x96, 0x15, 0xa0, 0x5a, 0xd1, 0x53, 0x59, 0x58, 0x4a,\n\t\t0xc1, 0xb6, 0x7d, 0x49, 0xbd, 0x29, 0x63, 0xce, 0xf8, 0x70, 0x74, 0xf9, 0x5e, 0x35, 0xe1, 0x09,\n\t\t0xab, 0x9b, 0x1a, 0xe6, 0xa1, 0x46, 0xd9, 0xc2, 0xde, 0x7e, 0xb4, 0xcf, 0xc3, 0x18, 0xe2, 0xc9,\n\t\t0xff, 0xcc, 0xab, 0x5e, 0xa6, 0x48, 0xf4, 0x05, 0xcc, 0x7a, 0x12, 0x0b, 0x69, 0x49, 0xd6, 0xa0,\n\t\t0x22, 0x91, 0xcf, 0x80, 0xe2, 0xb3, 0x94, 0xcc, 0x27, 0xc8, 0xdd, 0x0a, 0x52, 0x13, 0x59, 0xcc,\n\t\t0x78, 0x29, 0xf7, 0xd1, 0x4f, 0x06, 0x04, 0xa7, 0xdf, 0x74, 0xa8, 0xa4, 0x56, 0x74, 0x80, 0x16,\n\t\t0x7d, 0x4d, 0x89, 0x1f, 0x14, 0x2d, 0x91, 0xcc, 0xa0, 0x22, 0xf3, 0x71, 0x22, 0x99, 0xba, 0xc6,\n\t\t0x7a, 0xa5, 0xa1, 0x1e, 0x47, 0x48, 0x89, 0xdc, 0xe6, 0x49, 0xf6, 0x70, 0xf4, 0xbd, 0x01, 0xf3,\n\t\t0x3b, 0x98, 0x39, 0x59, 0x69, 0x0e, 0x29, 0x9a, 0xf7, 0x13, 0x69, 0xae, 0x62, 0xe6, 0x64, 0xa3,\n\t\t0x78, 0x6d, 0x27, 0x5b, 0x28, 0xfa, 0xd9, 0x80, 0x45, 0x41, 0x3f, 0xf7, 0xa9, 0x27, 0x2d, 0x82,\n\t\t0x5d, 0x42, 0x9d, 0x0c, 0x7d, 0x36, 0x9c, 0x72, 0x94, 0x66, 0x08, 0x56, 0x57, 0x58, 0x7d, 0x9b,\n\t\t0x6d, 0x5e, 0x64, 0x0f, 0x47, 0x5f, 0xc2, 0x9c, 0xa6, 0xd8, 0xbb, 0xe5, 0x0a, 0x8a, 0xda, 0x72,\n\t\t0x72, 0x95, 0x55, 0x72, 0xef, 0x9e, 0xbb, 0x44, 0xd2, 0x02, 0xd0, 0x0f, 0x06, 0x2c, 0xe8, 0xfd,\n\t\t0x33, 0xd6, 0x72, 0x44, 0x91, 0x79, 0x90, 0x42, 0x26, 0x5b, 0x35, 0xaf, 0x93, 0xac, 0xc1, 0xe8,\n\t\t0x2f, 0x03, 0x1e, 0x74, 0xd5, 0x93, 0xbe, 0x96, 0x54, 0xb8, 0x38, 0x33, 0xeb, 0xa2, 0x62, 0xfd,\n\t\t0xac, 0x7f, 0x75, 0x1f, 0x6b, 0xe0, 0x6c, 0x0f, 0x71, 0x47, 0x9c, 0x30, 0x17, 0xbd, 0x31, 0xe0,\n\t\t0x8a, 0xa0, 0x84, 0x0b, 0xdb, 0x6a, 0x60, 0xb1, 0xdf, 0xa3, 0xf2, 0x25, 0x45, 0xfb, 0x66, 0x0f,\n\t\t0xda, 0x41, 0xf6, 0x33, 0x95, 0x9c, 0x48, 0xee, 0xb2, 0x48, 0x8d, 0x40, 0xbf, 0x1b, 0x70, 0x9b,\n\t\t0x70, 0x57, 0x32, 0xd7, 0xa7, 0x16, 0xf6, 0x2c, 0x97, 0x1e, 0x66, 0x3d, 0x4e, 0x50, 0xbc, 0x1e,\n\t\t0xf7, 0xd0, 0x9d, 0x10, 0xf2, 0xa1, 0xb7, 0x4e, 0x0f, 0xb3, 0x1d, 0xe3, 0x22, 0xc9, 0x99, 0x83,\n\t\t0x7e, 0x35, 0x60, 0x39, 0x54, 0x6a, 0xb2, 0xc7, 0x1c, 0x3b, 0x2b, 0xef, 0x51, 0xc5, 0x7b, 0xa5,\n\t\t0xb7, 0x78, 0xd7, 0x03, 0xb4, 0x6c, 0xa4, 0x17, 0xbc, 0x3c, 0x09, 0xe8, 0x0f, 0x03, 0x6e, 0x7b,\n\t\t0x6c, 0x37, 0xe8, 0xd9, 0xbc, 0xcd, 0x5b, 0x56, 0xac, 0x57, 0x93, 0x59, 0x2b, 0xc8, 0x7c, 0x5d,\n\t\t0xbb, 0xe4, 0xe5, 0x4d, 0x42, 0xbf, 0x19, 0xf0, 0x91, 0xdf, 0xf4, 0xa8, 0x90, 0x6d, 0xd2, 0x1e,\n\t\t0xc5, 0x82, 0xec, 0x75, 0x10, 0x4d, 0x24, 0x3f, 0x96, 0xd2, 0x2a, 0x2f, 0x14, 0x62, 0xb4, 0xff,\n\t\t0xa6, 0xc2, 0x6b, 0x6f, 0x9a, 0xdc, 0x2a, 0x7e, 0xce, 0x9c, 0x95, 0x32, 0x40, 0x9b, 0x4e, 0xe5,\n\t\t0xdb, 0x02, 0x5c, 0xcd, 0x66, 0x1b, 0xd0, 0x2c, 0x8c, 0xc6, 0x63, 0x83, 0xd9, 0xca, 0x88, 0x94,\n\t\t0x4c, 0x88, 0x96, 0xd6, 0x6c, 0xb4, 0x0a, 0x63, 0xed, 0xb9, 0xd2, 0x6a, 0x52, 0xed, 0x0d, 0xae,\n\t\t0x24, 0x3e, 0x6b, 0xbc, 0x59, 0xab, 0x49, 0xcd, 0x32, 0xee, 0xb8, 0x42, 0x93, 0x50, 0xb0, 0x79,\n\t\t0x03, 0x33, 0x57, 0xcd, 0xf3, 0x92, 0xa9, 0xaf, 0xd0, 0x5d, 0x28, 0xa9, 0x71, 0x15, 0xb8, 0x2d,\n\t\t0x3d, 0x43, 0x2f, 0x25, 0x62, 0x07, 0x0f, 0xf0, 0x94, 0x79, 0xd2, 0x2c, 0x4a, 0xfd, 0x17, 0x5a,\n\t\t0x86, 0x61, 0xe6, 0x36, 0x7d, 0xa9, 0xe7, 0xda, 0x4c, 0x62, 0xde, 0x06, 0x6e, 0x39, 0x1c, 0xdb,\n\t\t0x66, 0x18, 0x8a, 0xb6, 0x60, 0x3a, 0x36, 0x66, 0x92, 0x5b, 0xc4, 0xe1, 0x1e, 0x55, 0x63, 0x89,\n\t\t0xfb, 0x52, 0x0f, 0xa1, 0xe9, 0x6a, 0x68, 0x2a, 0xab, 0x91, 0xa9, 0xac, 0x3e, 0xd2, 0xa6, 0xd2,\n\t\t0x9c, 0x8c, 0x72, 0xb7, 0x78, 0x3d, 0xc8, 0xdc, 0x0a, 0x13, 0xbb, 0x51, 0xdb, 0xfe, 0x2a, 0x40,\n\t\t0x1d, 0xc9, 0x81, 0x1a, 0xbb, 0xab, 0x00, 0x75, 0x1d, 0x26, 0x35, 0x52, 0x37, 0xd1, 0x62, 0x3f,\n\t\t0xc8, 0x89, 0xd0, 0x86, 0x1d, 0x65, 0xb9, 0x0a, 0xe3, 0x7b, 0x14, 0x0b, 0xb9, 0x4d, 0x71, 0x9b,\n\t\t0x5d, 0xa9, 0x1f, 0xd4, 0x99, 0x38, 0x27, 0xc2, 0xa9, 0x43, 0x59, 0x50, 0x29, 0x5a, 0x56, 0x93,\n\t\t0x3b, 0x8c, 0xb4, 0xb4, 0xe2, 0xcc, 0xf5, 0x50, 0x70, 0x29, 0x5a, 0x1b, 0x2a, 0xce, 0x1c, 0x15,\n\t\t0xed, 0x0b, 0x74, 0x13, 0x0a, 0x7b, 0x14, 0xdb, 0x54, 0xe8, 0x9f, 0xfe, 0xc5, 0xc4, 0xf4, 0x27,\n\t\t0x2a, 0xc4, 0xd4, 0xa1, 0xe8, 0x16, 0x4c, 0x46, 0x43, 0xd2, 0xe1, 0x04, 0x3b, 0x96, 0xcd, 0xbc,\n\t\t0x26, 0x96, 0x64, 0x4f, 0xfd, 0x04, 0x8b, 0xe6, 0x59, 0x7d, 0xf7, 0x69, 0x70, 0xf3, 0x91, 0xbe,\n\t\t0x57, 0xf9, 0xda, 0x80, 0x99, 0x34, 0xdb, 0x8a, 0xa6, 0xa1, 0x18, 0x3a, 0x93, 0xf8, 0x27, 0x30,\n\t\t0xa2, 0xae, 0xd7, 0x6c, 0xf4, 0x14, 0xce, 0xc5, 0x35, 0xd8, 0x61, 0xa2, 0x5d, 0x82, 0x81, 0x7e,\n\t\t0xe7, 0x86, 0x74, 0x09, 0x56, 0x99, 0x88, 0x2a, 0x50, 0x21, 0x30, 0x9f, 0xc3, 0xb2, 0xa2, 0x5b,\n\t\t0x50, 0x10, 0xd4, 0xf3, 0x1d, 0xa9, 0xdf, 0x10, 0xd2, 0x3b, 0x5c, 0xc7, 0x56, 0x30, 0x5c, 0xcb,\n\t\t0x68, 0x38, 0xd1, 0x6d, 0x18, 0x09, 0x0c, 0xa7, 0x2f, 0x68, 0xea, 0x0e, 0xab, 0x61, 0x8c, 0x19,\n\t\t0x05, 0x57, 0xd6, 0x61, 0x3e, 0x87, 0x5f, 0xec, 0xab, 0x32, 0x95, 0xbb, 0x70, 0x29, 0xd5, 0xe4,\n\t\t0xa5, 0x54, 0xa8, 0x42, 0xe0, 0x7a, 0x66, 0x4f, 0x16, 0x3c, 0xb0, 0x4d, 0x25, 0x66, 0x8e, 0x97,\n\t\t0xe9, 0x48, 0xa3, 0xe0, 0xca, 0xdf, 0x06, 0xdc, 0x39, 0xa9, 0x87, 0xea, 0xd0, 0x3e, 0xe3, 0x88,\n\t\t0xf6, 0xbd, 0x00, 0x74, 0x7c, 0x3a, 0xea, 0xc6, 0xba, 0x9a, 0xc8, 0xeb, 0xd8, 0x6e, 0xe6, 0xf8,\n\t\t0x61, 0xf7, 0x12, 0x9a, 0x82, 0x91, 0xc0, 0x6b, 0x08, 0xee, 0x28, 0xad, 0x2d, 0x9b, 0xd1, 0x25,\n\t\t0xaa, 0xc2, 0x44, 0x97, 0x95, 0xe0, 0xae, 0xd3, 0x52, 0xb2, 0x5b, 0x34, 0xc7, 0x49, 0xe7, 0x98,\n\t\t0x7f, 0xee, 0x3a, 0xad, 0xca, 0x2f, 0x06, 0x5c, 0x4e, 0xb7, 0x60, 0x41, 0x69, 0xb5, 0xb7, 0x73,\n\t\t0x71, 0x83, 0x46, 0xa5, 0x0d, 0x97, 0xd6, 0x71, 0x83, 0x76, 0x9e, 0xf8, 0x40, 0x8e, 0x13, 0xef,\n\t\t0xd0, 0x87, 0xc1, 0xcc, 0xfa, 0x50, 0x79, 0x03, 0xb0, 0x98, 0xd7, 0x9b, 0x05, 0x23, 0x2e, 0x3e,\n\t\t0x0f, 0x35, 0xe2, 0x8c, 0x94, 0x11, 0x17, 0x01, 0x86, 0x23, 0xee, 0xb0, 0xe3, 0xea, 0xe8, 0x28,\n\t\t0x1b, 0x38, 0xe1, 0x28, 0x1b, 0xcc, 0x3e, 0xca, 0x30, 0xcc, 0xb5, 0x3d, 0x55, 0x8f, 0x41, 0x31,\n\t\t0xd4, 0x4f, 0xa5, 0x66, 0x62, 0x88, 0xcd, 0x84, 0x89, 0xf1, 0x0a, 0x2e, 0xaa, 0x47, 0xea, 0x81,\n\t\t0x3e, 0xdc, 0x0f, 0xfd, 0x7c, 0x90, 0x9d, 0x04, 0xfc, 0x1c, 0x26, 0xb7, 0x31, 0xd9, 0xe7, 0x3b,\n\t\t0x3b, 0x1a, 0x9b, 0xb9, 0x92, 0x8a, 0x03, 0xec, 0xf4, 0x9f, 0xc1, 0x67, 0x75, 0xa2, 0x82, 0x5d,\n\t\t0xd3, 0x69, 0xc7, 0x66, 0xd2, 0xc8, 0x49, 0x66, 0xd2, 0x1a, 0x94, 0x98, 0xcb, 0x24, 0xc3, 0x92,\n\t\t0x0b, 0x35, 0x63, 0x4f, 0x2d, 0xcf, 0xf7, 0xf7, 0xff, 0x6b, 0x51, 0x8a, 0xd9, 0xce, 0xee, 0x54,\n\t\t0xd6, 0x52, 0x0e, 0x65, 0x45, 0x26, 0x4c, 0x3a, 0x38, 0x78, 0x07, 0x0c, 0xc7, 0x44, 0x50, 0x5a,\n\t\t0x3d, 0x02, 0x20, 0x43, 0x67, 0x9c, 0x0d, 0x72, 0xeb, 0x71, 0xaa, 0xa9, 0x32, 0xd1, 0xff, 0x61,\n\t\t0x8c, 0x88, 0xa0, 0x47, 0xb4, 0xcd, 0x50, 0x03, 0xbb, 0x64, 0x96, 0x83, 0xc5, 0xc8, 0x27, 0x9e,\n\t\t0x6c, 0x1e, 0x2f, 0xc0, 0x50, 0x83, 0x36, 0xb8, 0x36, 0xc0, 0xd3, 0x89, 0x29, 0xcf, 0x68, 0x83,\n\t\t0x9b, 0x2a, 0x0c, 0x99, 0x30, 0x7e, 0xcc, 0x50, 0x4f, 0x9d, 0x52, 0xb9, 0x1f, 0x24, 0x3b, 0xff,\n\t\t0x2e, 0xeb, 0x6b, 0x9e, 0xf1, 0xba, 0x56, 0xd0, 0x7d, 0x28, 0x7f, 0xc6, 0xa4, 0xa4, 0x22, 0x6c,\n\t\t0xa4, 0xa9, 0xd3, 0xfd, 0xfa, 0x67, 0x34, 0x0c, 0x57, 0xed, 0x83, 0x5e, 0xc2, 0x84, 0x3a, 0x1a,\n\t\t0x7e, 0x40, 0x85, 0x83, 0x9b, 0x51, 0xf7, 0x9c, 0x51, 0xb5, 0x4f, 0xd6, 0xe0, 0xba, 0xe0, 0xee,\n\t\t0xf3, 0x30, 0x5c, 0xf7, 0xd0, 0x38, 0xe9, 0x5e, 0x42, 0xaf, 0x61, 0x56, 0x8d, 0x37, 0x6a, 0x11,\n\t\t0xc7, 0xf7, 0x14, 0x3b, 0xea, 0x50, 0xa2, 0xea, 0xa9, 0xf7, 0x18, 0x4f, 0xf9, 0xc8, 0xa6, 0xe6,\n\t\t0x29, 0xad, 0x87, 0xa9, 0x9b, 0x51, 0xa6, 0xde, 0x6e, 0x06, 0xa7, 0xdc, 0xad, 0xfc, 0x58, 0x82,\n\t\t0x85, 0x5c, 0xaf, 0x79, 0x3d, 0xc7, 0xd3, 0x2c, 0x8c, 0xc6, 0xba, 0xc8, 0x6c, 0xa5, 0x68, 0x25,\n\t\t0x13, 0xa2, 0xa5, 0xf0, 0xdd, 0xe0, 0xa8, 0x70, 0x0e, 0xbe, 0x03, 0xe1, 0x7c, 0x0f, 0xef, 0x00,\n\t\t0x59, 0x84, 0xb3, 0xf0, 0xaf, 0x0a, 0xe7, 0xc8, 0x89, 0x85, 0xf3, 0x25, 0x4c, 0x34, 0xb1, 0xa0,\n\t\t0xae, 0xd4, 0x88, 0xba, 0x99, 0x8a, 0x29, 0x0d, 0xbb, 0xa1, 0xe2, 0x15, 0x4a, 0xd4, 0xb0, 0xcd,\n\t\t0xee, 0xa5, 0x4e, 0xd3, 0x50, 0x3a, 0x6a, 0x1a, 0x08, 0x4c, 0x75, 0xb4, 0x81, 0x25, 0xa8, 0xdf,\n\t\t0xde, 0x16, 0xd4, 0xb6, 0x37, 0x52, 0x0b, 0xbe, 0x66, 0x9b, 0x41, 0x8a, 0xde, 0xfa, 0xdc, 0x61,\n\t\t0xd2, 0xf2, 0xbb, 0x79, 0xa5, 0x38, 0xa6, 0x73, 0xe5, 0x54, 0x9d, 0x1b, 0xcb, 0xaf, 0x73, 0xa7,\n\t\t0xde, 0x42, 0xe7, 0x4e, 0xbf, 0x9d, 0xce, 0xfd, 0xf7, 0x94, 0xea, 0xcf, 0x01, 0x58, 0xca, 0xfd,\n\t\t0x69, 0xe7, 0x7d, 0x9b, 0xe9, 0x59, 0x18, 0xd5, 0x5f, 0xb4, 0x94, 0xbf, 0x0d, 0x3f, 0x5e, 0x40,\n\t\t0xb8, 0xa4, 0xfc, 0x6d, 0x2c, 0x40, 0x43, 0xd9, 0x05, 0xa8, 0xe3, 0xc7, 0x36, 0x9c, 0xc9, 0xa1,\n\t\t0x17, 0x7a, 0x39, 0xf4, 0xaf, 0x0c, 0x58, 0xcc, 0xfb, 0x85, 0x29, 0xb9, 0x3d, 0x8d, 0xb7, 0x6a,\n\t\t0xcf, 0x95, 0x4f, 0xe1, 0x3c, 0xe1, 0x8d, 0xa4, 0xec, 0x95, 0xb1, 0x88, 0xc2, 0x46, 0x20, 0x73,\n\t\t0x1b, 0xc6, 0x27, 0x4b, 0xbb, 0x4c, 0xee, 0xf9, 0xdb, 0x55, 0xc2, 0x1b, 0xb5, 0xce, 0xff, 0xac,\n\t\t0x2d, 0x30, 0xdb, 0xa9, 0xed, 0xf2, 0xf0, 0x9f, 0x79, 0xfa, 0xdf, 0x6c, 0xf7, 0x70, 0x93, 0x1d,\n\t\t0x2c, 0x6d, 0x17, 0xd4, 0xda, 0xcd, 0x7f, 0x02, 0x00, 0x00, 0xff, 0xff, 0x3a, 0x34, 0x7d, 0x59,\n\t\t0x29, 0x1c, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/cluster.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x53, 0x6d, 0x8b, 0xd3, 0x40,\n\t\t0x10, 0x26, 0x17, 0x6b, 0xdb, 0xe9, 0xe1, 0x9d, 0x8b, 0x87, 0x45, 0x11, 0x6b, 0x04, 0xa9, 0x88,\n\t\t0x09, 0x3d, 0x39, 0xc4, 0x17, 0x44, 0x7a, 0x2a, 0xf6, 0x83, 0x2f, 0xac, 0xdf, 0xfc, 0x12, 0x36,\n\t\t0x9b, 0x69, 0xbb, 0x5c, 0x76, 0xf7, 0xd8, 0x6c, 0x0a, 0x05, 0xff, 0x82, 0xff, 0x47, 0xff, 0x9d,\n\t\t0x64, 0x37, 0x09, 0x27, 0xf6, 0xf0, 0xbe, 0xcd, 0xcc, 0x3e, 0xcf, 0xb3, 0xcf, 0xcc, 0x30, 0xf0,\n\t\t0xb0, 0xca, 0xd0, 0x24, 0x9c, 0xe5, 0xa8, 0x38, 0x26, 0x2c, 0x97, 0x42, 0x25, 0x9b, 0x59, 0xc2,\n\t\t0x8b, 0xaa, 0xb4, 0x68, 0xe2, 0x73, 0xa3, 0xad, 0x26, 0x47, 0x35, 0x28, 0x6e, 0x40, 0xb1, 0x03,\n\t\t0xc5, 0x9b, 0x59, 0xf4, 0x08, 0x06, 0x1f, 0x75, 0x69, 0x17, 0x6a, 0xa9, 0xc9, 0x1d, 0x18, 0x88,\n\t\t0x1c, 0x95, 0x15, 0x76, 0x3b, 0x0e, 0x26, 0xc1, 0x74, 0x48, 0xbb, 0x3c, 0xfa, 0x01, 0x03, 0x2a,\n\t\t0xd4, 0xca, 0xe1, 0x08, 0x5c, 0x33, 0xba, 0xc0, 0x06, 0xe3, 0x62, 0xf2, 0x00, 0xf6, 0x25, 0xca,\n\t\t0x0c, 0x4d, 0xca, 0x75, 0xa5, 0xec, 0x78, 0x6f, 0x12, 0x4c, 0x7b, 0x74, 0xe4, 0x6b, 0xa7, 0x75,\n\t\t0x89, 0xbc, 0x80, 0xbe, 0x4f, 0xcb, 0x71, 0x38, 0x09, 0xa7, 0xa3, 0xe3, 0xfb, 0xf1, 0x4e, 0x4f,\n\t\t0x71, 0x6b, 0x88, 0xb6, 0xf8, 0xe8, 0x57, 0x00, 0x37, 0x3e, 0xf9, 0x78, 0x2d, 0xce, 0x9d, 0x89,\n\t\t0x39, 0xec, 0xf3, 0xca, 0x18, 0x54, 0x36, 0x5d, 0xeb, 0xd2, 0x3a, 0x33, 0x57, 0x90, 0x1c, 0x35,\n\t\t0xa4, 0xba, 0x40, 0x9e, 0xc0, 0x4d, 0x83, 0x8c, 0xaf, 0x59, 0x56, 0x60, 0xda, 0x7a, 0xdb, 0x9b,\n\t\t0x84, 0xd3, 0x21, 0x3d, 0xec, 0x1e, 0x9a, 0x7f, 0xc9, 0x09, 0xf4, 0x8c, 0x50, 0xab, 0xff, 0x99,\n\t\t0x6f, 0xa7, 0x44, 0x3d, 0x3a, 0xfa, 0x19, 0xc0, 0xc1, 0x3b, 0x2d, 0x99, 0x50, 0xa7, 0x8c, 0xaf,\n\t\t0xd1, 0x79, 0x7f, 0x09, 0x77, 0x55, 0x25, 0x53, 0xbd, 0x4c, 0x85, 0x45, 0x59, 0xa6, 0x42, 0xa5,\n\t\t0xbc, 0x7e, 0x4c, 0xb3, 0x6d, 0x2a, 0x72, 0xd7, 0x4a, 0x48, 0x8f, 0x54, 0x25, 0xbf, 0x2c, 0x17,\n\t\t0x35, 0x60, 0xe1, 0xb9, 0xf3, 0xed, 0x22, 0x27, 0x6f, 0xe0, 0xde, 0xa5, 0x5c, 0xc5, 0x24, 0xba,\n\t\t0xc9, 0x87, 0xf4, 0xf6, 0x0e, 0xf6, 0x67, 0x26, 0x31, 0x7a, 0x0d, 0xe4, 0x2b, 0x9a, 0x52, 0x94,\n\t\t0xb6, 0xf6, 0xfd, 0x0d, 0xad, 0x15, 0x6a, 0x45, 0x0e, 0x21, 0x3c, 0xc3, 0x76, 0xeb, 0x75, 0x48,\n\t\t0x6e, 0x41, 0x6f, 0xc3, 0x8a, 0xca, 0xeb, 0x0d, 0xa9, 0x4f, 0xa2, 0xb7, 0x7f, 0xb1, 0x3f, 0x20,\n\t\t0xb3, 0x95, 0xc1, 0x1d, 0xec, 0x31, 0xf4, 0x51, 0xd5, 0xd3, 0xcb, 0x1d, 0x7f, 0x40, 0xdb, 0x34,\n\t\t0xfa, 0x1d, 0xc0, 0xc1, 0x05, 0x09, 0x37, 0x8f, 0x31, 0xf4, 0x33, 0xc6, 0xcf, 0x50, 0xe5, 0x8d,\n\t\t0x46, 0x9b, 0x92, 0xf7, 0x30, 0x28, 0xbd, 0x45, 0xbf, 0x98, 0xd1, 0xf1, 0xe3, 0x4b, 0xe6, 0xfe,\n\t\t0x6f, 0x53, 0xb4, 0xa3, 0xd6, 0x32, 0x4b, 0xef, 0xb5, 0x5d, 0xdf, 0x15, 0x64, 0x9a, 0xee, 0x68,\n\t\t0x47, 0x9d, 0x3f, 0xff, 0x7e, 0xb2, 0x12, 0x76, 0x5d, 0x65, 0x31, 0xd7, 0x32, 0xb9, 0x78, 0x75,\n\t\t0x4f, 0x45, 0x5e, 0x24, 0x2b, 0x9d, 0xb8, 0x5b, 0xeb, 0x4e, 0xf0, 0x95, 0x0b, 0x36, 0xb3, 0xec,\n\t\t0xba, 0xab, 0x3f, 0xfb, 0x13, 0x00, 0x00, 0xff, 0xff, 0xaf, 0x49, 0xfd, 0x6d, 0xaa, 0x03, 0x00,\n\t\t0x00,\n\t},\n\t// uber/cadence/admin/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2e, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x4f, 0x4c, 0xc9, 0xcd, 0xcc, 0xd3, 0x2f,\n\t\t0x33, 0xd4, 0xcf, 0xc8, 0x2c, 0x2e, 0xc9, 0x2f, 0xaa, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17,\n\t\t0x12, 0x05, 0x29, 0xd2, 0x83, 0x2a, 0xd2, 0x03, 0x2b, 0xd2, 0x2b, 0x33, 0x54, 0xf2, 0xe4, 0x12,\n\t\t0x0a, 0x4b, 0x2d, 0x2a, 0xce, 0xcc, 0xcf, 0xf3, 0x80, 0x28, 0xf7, 0x2c, 0x49, 0xcd, 0x15, 0x92,\n\t\t0xe4, 0xe2, 0x48, 0x2d, 0x4b, 0xcd, 0x2b, 0x89, 0xcf, 0x4c, 0x91, 0x60, 0x54, 0x60, 0xd4, 0x60,\n\t\t0x0e, 0x62, 0x07, 0xf3, 0x3d, 0x53, 0x84, 0x24, 0xb8, 0xd8, 0xcb, 0x20, 0x1a, 0x24, 0x98, 0x20,\n\t\t0x32, 0x50, 0xae, 0x52, 0x09, 0x17, 0x1f, 0xaa, 0x51, 0x42, 0x8a, 0x5c, 0x3c, 0x49, 0x45, 0x89,\n\t\t0x79, 0xc9, 0x19, 0xf1, 0x25, 0xf9, 0xd9, 0xa9, 0x79, 0x60, 0xa3, 0x78, 0x82, 0xb8, 0x21, 0x62,\n\t\t0x21, 0x20, 0x21, 0x21, 0x7b, 0x2e, 0xd6, 0xcc, 0x92, 0xd4, 0xdc, 0x62, 0x09, 0x26, 0x05, 0x66,\n\t\t0x0d, 0x6e, 0x23, 0x4d, 0x3d, 0xac, 0xce, 0xd4, 0xc3, 0x74, 0x63, 0x10, 0x44, 0x9f, 0x93, 0x79,\n\t\t0x94, 0x69, 0x7a, 0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x72, 0x48, 0xe8,\n\t\t0x66, 0xa6, 0xe4, 0xe8, 0xa7, 0xe7, 0xeb, 0x83, 0xfd, 0x0f, 0x0f, 0x16, 0x6b, 0x30, 0xa3, 0xcc,\n\t\t0x30, 0x89, 0x0d, 0x2c, 0x6e, 0x0c, 0x08, 0x00, 0x00, 0xff, 0xff, 0x44, 0x14, 0xd7, 0xd4, 0x3e,\n\t\t0x01, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/queue.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x59, 0x4f, 0x6f, 0xdb, 0xc8,\n\t\t0x15, 0x37, 0x25, 0xcb, 0x96, 0x9f, 0x1d, 0x9b, 0x9e, 0x24, 0x8d, 0xe2, 0xc4, 0x88, 0xa2, 0x05,\n\t\t0x12, 0xc1, 0x9b, 0x48, 0xb5, 0xbd, 0x41, 0x16, 0xbb, 0x45, 0x5b, 0x9a, 0x62, 0x2c, 0xae, 0x65,\n\t\t0x49, 0x18, 0x52, 0x49, 0xbc, 0x58, 0x80, 0xa0, 0xa9, 0xb1, 0x43, 0x84, 0x22, 0xb5, 0x24, 0xe5,\n\t\t0x54, 0x1f, 0xa0, 0xd9, 0x7b, 0xb7, 0xd7, 0x5e, 0x8a, 0xde, 0x0a, 0xf4, 0xd4, 0x1e, 0x8a, 0x5e,\n\t\t0x0b, 0xf4, 0xd4, 0x6b, 0x7b, 0xd8, 0x2f, 0xd1, 0x8f, 0x50, 0x70, 0x48, 0x4a, 0x14, 0x45, 0x59,\n\t\t0x74, 0xe2, 0x02, 0x3d, 0xec, 0x4d, 0x7a, 0xfc, 0xcd, 0xe3, 0xef, 0xfd, 0x99, 0xdf, 0xbc, 0x91,\n\t\t0xe0, 0xe1, 0xe0, 0x94, 0xd8, 0x55, 0x4d, 0xed, 0x12, 0x53, 0x23, 0x55, 0xb5, 0xdb, 0xd3, 0xcd,\n\t\t0xea, 0xc5, 0x6e, 0xf5, 0xdb, 0x01, 0x19, 0x90, 0x4a, 0xdf, 0xb6, 0x5c, 0x0b, 0xdd, 0xf6, 0x20,\n\t\t0x95, 0x00, 0x52, 0xa1, 0x90, 0xca, 0xc5, 0xee, 0xd6, 0x83, 0x73, 0xcb, 0x3a, 0x37, 0x48, 0x95,\n\t\t0x82, 0x4e, 0x07, 0x67, 0x55, 0x57, 0xef, 0x11, 0xc7, 0x55, 0x7b, 0x7d, 0x7f, 0xdd, 0x56, 0x71,\n\t\t0xd2, 0x75, 0x5f, 0xf7, 0x1c, 0x6b, 0x56, 0xaf, 0x67, 0x99, 0x01, 0xe2, 0x61, 0x12, 0xe2, 0x8d,\n\t\t0xee, 0xb8, 0x96, 0x3d, 0x0c, 0x20, 0xa5, 0x24, 0xc8, 0x3b, 0xcb, 0x7e, 0x7b, 0x66, 0x58, 0xef,\n\t\t0x7c, 0x4c, 0xe9, 0x87, 0x0c, 0xdc, 0xe2, 0x6d, 0xcb, 0x71, 0x78, 0x63, 0xe0, 0xb8, 0xc4, 0x96,\n\t\t0x55, 0xe7, 0xad, 0x68, 0x9e, 0x59, 0xe8, 0x1e, 0xac, 0x74, 0xad, 0x9e, 0xaa, 0x9b, 0x8a, 0xde,\n\t\t0x2d, 0x30, 0x45, 0xa6, 0xbc, 0x82, 0xf3, 0xbe, 0x41, 0xec, 0xa2, 0x0e, 0xa0, 0xd0, 0x8f, 0x42,\n\t\t0x7e, 0x45, 0xb4, 0x81, 0xab, 0x5b, 0x66, 0x21, 0x53, 0x64, 0xca, 0xab, 0x7b, 0x8f, 0x2a, 0x93,\n\t\t0x31, 0xf7, 0xf5, 0xca, 0xc5, 0x6e, 0xe5, 0x55, 0x00, 0x17, 0x42, 0x34, 0xde, 0x7c, 0x17, 0x37,\n\t\t0xa1, 0x3a, 0xac, 0xb8, 0xaa, 0xf3, 0x56, 0x71, 0x87, 0x7d, 0x52, 0xc8, 0x16, 0x99, 0xf2, 0xfa,\n\t\t0xde, 0xa7, 0x95, 0xc4, 0x0c, 0x56, 0xe2, 0x9c, 0xe5, 0x61, 0x9f, 0xe0, 0xbc, 0x1b, 0x7c, 0x42,\n\t\t0xdb, 0x00, 0xd4, 0x93, 0xe3, 0xaa, 0x2e, 0x29, 0x2c, 0x16, 0x99, 0x72, 0x0e, 0x53, 0xdf, 0x92,\n\t\t0x67, 0x40, 0x77, 0x60, 0x99, 0x3e, 0xd6, 0xbb, 0x85, 0x5c, 0x91, 0x29, 0x67, 0xf1, 0x92, 0xf7,\n\t\t0x55, 0xec, 0xa2, 0x63, 0xb8, 0x75, 0xa1, 0x3b, 0xfa, 0xa9, 0x6e, 0xe8, 0xee, 0x50, 0x19, 0x55,\n\t\t0xa5, 0xb0, 0x44, 0x43, 0xdb, 0xaa, 0xf8, 0x75, 0xab, 0x84, 0x75, 0xab, 0xc8, 0x21, 0x02, 0xdf,\n\t\t0x1c, 0xaf, 0x1b, 0x19, 0x4b, 0x7f, 0x58, 0x84, 0x9f, 0x46, 0x99, 0x4a, 0xae, 0x6a, 0xbb, 0xfc,\n\t\t0x1b, 0xdd, 0xe8, 0x8e, 0xf3, 0x40, 0xbe, 0x1d, 0x10, 0xc7, 0xe5, 0x5c, 0xd7, 0xd6, 0x4f, 0x07,\n\t\t0x2e, 0x71, 0x50, 0x19, 0x58, 0x57, 0xb5, 0xcf, 0x89, 0xab, 0xc4, 0x0b, 0xb0, 0xee, 0xdb, 0x6b,\n\t\t0x61, 0x19, 0xb6, 0x01, 0x6c, 0x7f, 0xb9, 0x87, 0xc9, 0x50, 0xcc, 0x4a, 0x60, 0x11, 0xbb, 0xe8,\n\t\t0x09, 0x20, 0xdd, 0xd4, 0x5d, 0x5d, 0x75, 0x49, 0x57, 0x21, 0x17, 0xc4, 0xa4, 0xb0, 0x2c, 0x0d,\n\t\t0x98, 0x1d, 0x3d, 0x11, 0xbc, 0x07, 0x62, 0x17, 0xbd, 0x67, 0x60, 0x2b, 0x0e, 0x57, 0x47, 0xac,\n\t\t0x68, 0x0e, 0x57, 0xf7, 0xea, 0x89, 0xc5, 0x1d, 0x87, 0x35, 0x55, 0x66, 0x71, 0xe2, 0x35, 0xe3,\n\t\t0x28, 0x71, 0x41, 0x9f, 0xf1, 0x04, 0x95, 0xe0, 0x46, 0x10, 0xbf, 0x3d, 0x30, 0xc3, 0x12, 0xad,\n\t\t0xe0, 0x55, 0xdf, 0x88, 0x07, 0x5e, 0xe4, 0xdf, 0x31, 0xc0, 0xf6, 0x55, 0xdb, 0xd5, 0xbd, 0x77,\n\t\t0x28, 0x9a, 0x65, 0x9e, 0xe9, 0xe7, 0x85, 0xa5, 0x62, 0xb6, 0xbc, 0xba, 0xf7, 0x4d, 0x8a, 0x8e,\n\t\t0x49, 0x53, 0x87, 0x4a, 0x3b, 0xf4, 0xcf, 0x53, 0xf7, 0x82, 0xe9, 0xda, 0x43, 0xbc, 0xd1, 0x9f,\n\t\t0xb4, 0x6e, 0x1d, 0xc0, 0xad, 0x24, 0x20, 0x62, 0x21, 0xfb, 0x96, 0x0c, 0x83, 0xc2, 0x79, 0x1f,\n\t\t0xd1, 0x2d, 0xc8, 0x5d, 0xa8, 0xc6, 0x80, 0x04, 0x85, 0xf2, 0xbf, 0x7c, 0x91, 0xf9, 0x9c, 0x29,\n\t\t0x7d, 0x05, 0xbb, 0x73, 0xd9, 0x39, 0x7d, 0xcb, 0x74, 0x48, 0x24, 0x4d, 0xb7, 0x61, 0x29, 0xc8,\n\t\t0x8f, 0xff, 0x8e, 0x9c, 0xed, 0x65, 0xa6, 0xf4, 0xd7, 0x0c, 0x3c, 0x89, 0x3a, 0xe3, 0x55, 0x53,\n\t\t0x23, 0xc6, 0xb5, 0xb4, 0xdb, 0x29, 0xdc, 0x0d, 0x90, 0x1f, 0xbd, 0xf9, 0xef, 0xf8, 0x8e, 0xa6,\n\t\t0x1e, 0xc4, 0x5a, 0x3a, 0x9b, 0xae, 0xa5, 0x17, 0x67, 0xb4, 0x74, 0x05, 0x6e, 0x6a, 0x5e, 0x1a,\n\t\t0xc7, 0x7c, 0x2d, 0xd3, 0x18, 0xd2, 0x7e, 0xca, 0xe3, 0x4d, 0x2d, 0xda, 0xb0, 0x2d, 0xd3, 0x18,\n\t\t0x96, 0xaa, 0xf0, 0xf4, 0xd2, 0xd4, 0xc5, 0x6b, 0x50, 0xfa, 0x4b, 0x76, 0x32, 0xd9, 0x92, 0x7e,\n\t\t0x6e, 0xaa, 0x3f, 0x26, 0x3b, 0x4d, 0xb2, 0xd1, 0x03, 0x58, 0x75, 0x68, 0xba, 0x14, 0x53, 0xed,\n\t\t0x11, 0xaa, 0xb0, 0x2b, 0x18, 0x7c, 0x53, 0x53, 0xed, 0x11, 0xf4, 0x0b, 0x58, 0x0b, 0x00, 0xba,\n\t\t0xd9, 0x1f, 0xb8, 0x85, 0x65, 0x1a, 0xf4, 0xfd, 0xc4, 0xa0, 0xdb, 0xea, 0xd0, 0xb0, 0xd4, 0x2e,\n\t\t0x0e, 0x5c, 0x8a, 0xde, 0x02, 0x54, 0x80, 0x65, 0xcd, 0x32, 0x5d, 0xdb, 0x32, 0x0a, 0xf9, 0x22,\n\t\t0x53, 0x5e, 0xc3, 0xe1, 0xd7, 0x78, 0xa1, 0xa7, 0xca, 0x36, 0x55, 0xe8, 0x7f, 0x66, 0x80, 0x8b,\n\t\t0xae, 0xc0, 0x44, 0xb3, 0xec, 0x6e, 0xb2, 0xe4, 0xf1, 0x56, 0xaf, 0x6f, 0x10, 0x97, 0xfc, 0xbf,\n\t\t0x57, 0xff, 0x6a, 0xc7, 0x43, 0x03, 0x58, 0xcd, 0x0f, 0xcc, 0x53, 0x5c, 0x0a, 0x0f, 0xce, 0x84,\n\t\t0x87, 0x89, 0x44, 0xea, 0xfe, 0x28, 0x42, 0x97, 0xe3, 0x8d, 0xf1, 0x52, 0x6a, 0x28, 0xd5, 0xe0,\n\t\t0xe0, 0xea, 0xe9, 0x9c, 0xaa, 0xca, 0x7f, 0x18, 0x28, 0x72, 0xfd, 0xbe, 0x31, 0x6c, 0xab, 0x36,\n\t\t0x31, 0x5d, 0xde, 0xb0, 0x1c, 0xd2, 0xb6, 0x0c, 0x5d, 0x1b, 0x46, 0x92, 0xfe, 0x08, 0x36, 0xfc,\n\t\t0xbe, 0x8c, 0xe7, 0xfc, 0x06, 0x35, 0x8f, 0x52, 0xbe, 0x03, 0x9b, 0xb1, 0xfe, 0x1d, 0x9d, 0xa9,\n\t\t0x1b, 0x13, 0xdd, 0x2b, 0x76, 0x51, 0x11, 0xd6, 0x7c, 0x6c, 0xa0, 0xc0, 0xfe, 0xd6, 0x01, 0x6a,\n\t\t0xf3, 0x0f, 0xa8, 0x97, 0x70, 0xb3, 0x4f, 0x49, 0x29, 0x9a, 0xc7, 0x4a, 0xe9, 0x53, 0x5a, 0x34,\n\t\t0x63, 0xeb, 0x33, 0x4a, 0x37, 0x15, 0x04, 0xde, 0xec, 0xc7, 0x4d, 0xa5, 0xef, 0x19, 0xb8, 0x9f,\n\t\t0x1c, 0xb2, 0x37, 0xd9, 0x0c, 0x1c, 0x74, 0x1f, 0x56, 0x82, 0x64, 0x13, 0x3f, 0xd0, 0x3c, 0x1e,\n\t\t0x1b, 0x50, 0x07, 0xd6, 0xce, 0x54, 0xdd, 0x20, 0x5d, 0x45, 0x53, 0x07, 0x8e, 0x7f, 0x14, 0xad,\n\t\t0xef, 0xed, 0xa5, 0x1c, 0xb2, 0x5e, 0xd0, 0xa5, 0xbc, 0xb7, 0x12, 0xaf, 0x9e, 0x8d, 0xbf, 0x94,\n\t\t0xfe, 0xc6, 0xc0, 0x76, 0x32, 0xab, 0x60, 0x13, 0xa0, 0x63, 0xc8, 0xd1, 0xec, 0x50, 0x4a, 0xab,\n\t\t0x7b, 0xcf, 0x67, 0xbc, 0x71, 0x5e, 0x35, 0xb1, 0xef, 0x05, 0x1d, 0xc1, 0x92, 0x43, 0xe3, 0x0d,\n\t\t0x36, 0xc3, 0xfe, 0x95, 0xfc, 0xf9, 0xa9, 0xc2, 0x81, 0x8b, 0xd2, 0x77, 0x0c, 0xec, 0x47, 0x43,\n\t\t0xbd, 0x34, 0x92, 0x48, 0x67, 0xb5, 0x21, 0x4f, 0xd9, 0xd8, 0xc4, 0x2c, 0x30, 0x74, 0xf6, 0xf8,\n\t\t0xec, 0x4a, 0x34, 0x02, 0x8f, 0x78, 0xe4, 0xa5, 0xf4, 0xf7, 0x99, 0xd5, 0xc5, 0xc4, 0x19, 0x18,\n\t\t0xd7, 0x9e, 0xc6, 0xff, 0x51, 0x3b, 0xfc, 0x96, 0x81, 0xcf, 0xd2, 0x24, 0x74, 0x6a, 0xa6, 0xf9,\n\t\t0x26, 0xd8, 0xab, 0x36, 0x31, 0x95, 0xa0, 0xbe, 0x7e, 0x62, 0xf7, 0xaf, 0x98, 0x58, 0x2f, 0x59,\n\t\t0x78, 0x3d, 0xf4, 0xe5, 0xd7, 0xbb, 0xf4, 0xbb, 0x65, 0xb8, 0x13, 0x8f, 0x21, 0xec, 0xcf, 0xf0,\n\t\t0xea, 0xa1, 0x9b, 0x67, 0x56, 0x90, 0xdc, 0xb4, 0x57, 0x0f, 0xef, 0xba, 0xe4, 0x5f, 0x3d, 0xe8,\n\t\t0xc5, 0xe9, 0x37, 0x0c, 0x14, 0x1d, 0x6f, 0x82, 0x53, 0x7c, 0x89, 0x18, 0xe9, 0x76, 0x74, 0x9a,\n\t\t0xf6, 0xbb, 0xf6, 0xf0, 0x9a, 0x46, 0xd5, 0xfa, 0x02, 0xde, 0x76, 0xa6, 0x71, 0x91, 0xc4, 0xfe,\n\t\t0x9a, 0x81, 0x7b, 0x1a, 0x1d, 0x67, 0x92, 0xf9, 0x64, 0x29, 0x1f, 0x3e, 0x05, 0x9f, 0x79, 0xf3,\n\t\t0x64, 0x7d, 0x01, 0xdf, 0xd5, 0x26, 0x31, 0x31, 0x1e, 0xc1, 0xa1, 0x9e, 0xc8, 0x63, 0x31, 0x35,\n\t\t0x8f, 0x79, 0xa3, 0x96, 0xc7, 0xc3, 0x99, 0xc4, 0x44, 0x78, 0xfc, 0x8b, 0x81, 0x2f, 0x6d, 0x7a,\n\t\t0xe8, 0x28, 0x31, 0xd1, 0x1f, 0xd3, 0x0a, 0x55, 0x53, 0x09, 0x67, 0xa4, 0x08, 0xcf, 0x1c, 0xe5,\n\t\t0xf9, 0x3a, 0x05, 0xcf, 0x0f, 0x9a, 0x14, 0xea, 0x0b, 0xf8, 0x99, 0xfd, 0x41, 0x23, 0xc6, 0x1f,\n\t\t0x19, 0x78, 0xa2, 0x7a, 0x9b, 0x42, 0x49, 0x38, 0x7e, 0x92, 0x22, 0xf1, 0x6f, 0xb6, 0x5f, 0xa5,\n\t\t0x88, 0x24, 0xa5, 0x2c, 0xd6, 0x17, 0xf0, 0x63, 0x35, 0x1d, 0xf4, 0x60, 0x0d, 0x60, 0x4c, 0xa5,\n\t\t0xf4, 0xe7, 0x3c, 0x14, 0xa6, 0xf7, 0xa7, 0x2f, 0x12, 0xd1, 0x2b, 0x3b, 0x33, 0x71, 0x65, 0x9f,\n\t\t0xf8, 0xd1, 0x20, 0x73, 0x7d, 0x3f, 0x1a, 0x64, 0xe3, 0x3f, 0x1a, 0xc4, 0xc5, 0x72, 0xf1, 0x5a,\n\t\t0xc4, 0x12, 0x7d, 0x9f, 0x46, 0x2f, 0x72, 0x89, 0xb7, 0xef, 0xd4, 0x7a, 0x11, 0x17, 0xda, 0xf9,\n\t\t0x82, 0xf1, 0x7e, 0x8e, 0x60, 0xf8, 0x6d, 0x53, 0xfb, 0x10, 0xc1, 0x48, 0x20, 0x73, 0x89, 0x62,\n\t\t0xbc, 0x9f, 0xa3, 0x18, 0xcb, 0xa9, 0x89, 0xcc, 0x9d, 0xf2, 0x2f, 0x97, 0x8c, 0x7f, 0x7f, 0xa4,\n\t\t0x64, 0xe4, 0x29, 0xd1, 0x93, 0x6b, 0x93, 0x8c, 0x04, 0xf6, 0x1f, 0xa8, 0x19, 0x7f, 0x62, 0xe0,\n\t\t0xe9, 0x65, 0x9a, 0xe1, 0xbf, 0x29, 0x1a, 0xcb, 0x0a, 0x8d, 0xe5, 0xe8, 0x23, 0x44, 0x23, 0x81,\n\t\t0x7d, 0x59, 0x4d, 0x89, 0x8d, 0xc9, 0x86, 0x95, 0xa4, 0x1a, 0x34, 0x46, 0x07, 0x49, 0x70, 0x83,\n\t\t0x6e, 0xe9, 0xa0, 0x1e, 0xe1, 0x38, 0x51, 0x49, 0xb9, 0x69, 0xc3, 0x09, 0x6d, 0xcd, 0x8d, 0x38,\n\t\t0xdd, 0x79, 0xcf, 0x40, 0x3e, 0x94, 0x0f, 0x74, 0x1b, 0x36, 0x65, 0x4e, 0x3a, 0x52, 0xe4, 0x93,\n\t\t0xb6, 0xa0, 0x88, 0xcd, 0x97, 0x5c, 0x43, 0xac, 0xb1, 0x0b, 0xe8, 0x27, 0x80, 0xc6, 0x66, 0x19,\n\t\t0x73, 0x4d, 0xe9, 0x85, 0x80, 0x59, 0x06, 0xdd, 0x84, 0x8d, 0x88, 0x5d, 0x3c, 0x16, 0x30, 0x9b,\n\t\t0x41, 0x77, 0xe1, 0xf6, 0xd8, 0x88, 0x85, 0x76, 0x43, 0xe4, 0x39, 0x59, 0x6c, 0x35, 0xd9, 0x2c,\n\t\t0xba, 0x07, 0x77, 0xc6, 0x8f, 0x78, 0xdc, 0x92, 0x24, 0x85, 0x6f, 0x74, 0x24, 0x59, 0xc0, 0xec,\n\t\t0xe2, 0xce, 0x3f, 0x12, 0x7e, 0xbc, 0xa5, 0xa4, 0x3e, 0x81, 0x07, 0x13, 0x58, 0x25, 0x89, 0xe2,\n\t\t0x2e, 0x3c, 0x9d, 0x05, 0x92, 0x64, 0x0e, 0xcb, 0x0a, 0x5f, 0x17, 0x1b, 0x35, 0x45, 0x78, 0x2d,\n\t\t0xf0, 0x1d, 0xca, 0x86, 0x41, 0x4f, 0xa0, 0x3c, 0x6b, 0x09, 0xcf, 0x35, 0x79, 0xa1, 0x11, 0x41,\n\t\t0x67, 0x2e, 0x43, 0x4b, 0xe2, 0x61, 0x93, 0x8b, 0xa2, 0xb3, 0xa8, 0x06, 0xbf, 0x9c, 0x85, 0xc6,\n\t\t0x02, 0xdf, 0xc2, 0xb5, 0x80, 0xcf, 0xab, 0x16, 0x3e, 0x3a, 0x6a, 0xb4, 0x5e, 0x8d, 0x17, 0x2b,\n\t\t0x7c, 0xeb, 0xb8, 0xdd, 0x10, 0x64, 0x81, 0x5d, 0x44, 0xcf, 0x60, 0x77, 0x96, 0x17, 0xae, 0xdd,\n\t\t0x6e, 0x9c, 0x28, 0x6d, 0x0e, 0x0b, 0x4d, 0x59, 0xe1, 0x1b, 0x2d, 0x49, 0x50, 0xda, 0xad, 0x86,\n\t\t0xc8, 0x9f, 0xb0, 0xb9, 0x9d, 0xdf, 0x67, 0xe1, 0xde, 0x25, 0x8a, 0x8d, 0x3e, 0x85, 0xc7, 0x09,\n\t\t0x6e, 0x5f, 0x70, 0x62, 0x43, 0xa8, 0x29, 0x3c, 0xd7, 0x91, 0xa2, 0x89, 0x4d, 0xe6, 0x30, 0x01,\n\t\t0xae, 0xb5, 0x8e, 0x39, 0xb1, 0xa9, 0x34, 0x5b, 0xb2, 0xc2, 0xf1, 0xb2, 0xf8, 0x52, 0x60, 0x99,\n\t\t0x2b, 0x2e, 0x13, 0x5e, 0x8b, 0x92, 0x2c, 0xb1, 0x19, 0xf4, 0x33, 0xf8, 0x7c, 0xde, 0x32, 0x2f,\n\t\t0x65, 0x2f, 0xbc, 0x94, 0x71, 0x0d, 0x2c, 0x70, 0xb5, 0x13, 0x05, 0x77, 0x9a, 0x4d, 0xb1, 0x79,\n\t\t0xc8, 0x66, 0xd1, 0x73, 0xd8, 0x4f, 0xbd, 0x3a, 0xf2, 0xda, 0x45, 0xf4, 0x73, 0xf8, 0xe2, 0xca,\n\t\t0xaf, 0x0d, 0xeb, 0x54, 0x63, 0x73, 0x33, 0xba, 0x6f, 0x62, 0x7d, 0xa7, 0xc9, 0x73, 0xb2, 0x70,\n\t\t0xd8, 0xc2, 0xe2, 0xd7, 0x42, 0x8d, 0x5d, 0xda, 0xf9, 0x81, 0x01, 0x74, 0x48, 0xdc, 0x78, 0x6d,\n\t\t0x1e, 0xc2, 0xf6, 0xa1, 0x20, 0x5f, 0x5a, 0x91, 0x47, 0x50, 0x4a, 0x86, 0x48, 0x02, 0x7e, 0x29,\n\t\t0xf2, 0x82, 0x72, 0xd0, 0x91, 0x4e, 0x58, 0x66, 0xb6, 0x2b, 0x6f, 0xa7, 0xb6, 0x3a, 0x32, 0x9b,\n\t\t0x41, 0x15, 0xd8, 0x99, 0xe1, 0xaa, 0xce, 0xe1, 0x9a, 0xd2, 0x7a, 0xd5, 0x14, 0xb0, 0x54, 0x17,\n\t\t0xdb, 0x4a, 0xa3, 0x25, 0xc9, 0x6c, 0x16, 0x3d, 0x86, 0x4f, 0x92, 0xf1, 0x93, 0xd1, 0x2d, 0x1e,\n\t\t0x3c, 0xff, 0xfa, 0xd9, 0xb9, 0xee, 0xbe, 0x19, 0x9c, 0x56, 0x34, 0xab, 0x57, 0x8d, 0xfe, 0x75,\n\t\t0xf3, 0x54, 0xef, 0x1a, 0xd5, 0x73, 0xcb, 0xff, 0xb7, 0x68, 0xf4, 0x3f, 0xd3, 0x97, 0xf4, 0xc3,\n\t\t0xc5, 0xee, 0xe9, 0x12, 0xb5, 0xef, 0xff, 0x37, 0x00, 0x00, 0xff, 0xff, 0x4b, 0xe2, 0x06, 0x07,\n\t\t0x8f, 0x1a, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/replication.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xc4, 0x58, 0xcd, 0x6f, 0xdb, 0xc6,\n\t\t0x12, 0x0f, 0x25, 0xeb, 0xc3, 0x63, 0x59, 0xa2, 0xd7, 0x4e, 0xac, 0xd8, 0x31, 0x9e, 0xa2, 0x7c,\n\t\t0x39, 0x0e, 0x9e, 0xf4, 0xec, 0xbc, 0xbc, 0x4f, 0x3c, 0x04, 0x8c, 0x25, 0x3f, 0xb3, 0xf1, 0xe7,\n\t\t0x8a, 0x71, 0xea, 0x02, 0x05, 0x41, 0x93, 0x6b, 0x8b, 0xb0, 0x44, 0x0a, 0xe4, 0x4a, 0x8e, 0x6e,\n\t\t0xed, 0xa5, 0xa7, 0x02, 0xbd, 0xf4, 0xd6, 0x63, 0xff, 0x8d, 0x9e, 0x7b, 0x6d, 0xff, 0xa4, 0x82,\n\t\t0xbb, 0x4b, 0x49, 0x94, 0x28, 0xc5, 0x41, 0x0e, 0xbd, 0x69, 0x67, 0x7e, 0x33, 0xb3, 0x3b, 0x3b,\n\t\t0xfb, 0x9b, 0x11, 0xe1, 0x59, 0xf7, 0x82, 0x78, 0x55, 0xd3, 0xb0, 0x88, 0x63, 0x92, 0xaa, 0x61,\n\t\t0xb5, 0x6d, 0xa7, 0xda, 0xdb, 0xae, 0x7a, 0xa4, 0xd3, 0xb2, 0x4d, 0x83, 0xda, 0xae, 0x53, 0xe9,\n\t\t0x78, 0x2e, 0x75, 0xd1, 0xdd, 0x00, 0x58, 0x11, 0xc0, 0x0a, 0x03, 0x56, 0x7a, 0xdb, 0x6b, 0x7f,\n\t\t0xb9, 0x72, 0xdd, 0xab, 0x16, 0xa9, 0x32, 0xd0, 0x45, 0xf7, 0xb2, 0x4a, 0xed, 0x36, 0xf1, 0xa9,\n\t\t0xd1, 0xee, 0x70, 0xbb, 0xb5, 0x52, 0x34, 0x40, 0xc7, 0x0e, 0xdc, 0x9b, 0x6e, 0xbb, 0x1d, 0x7a,\n\t\t0x8e, 0x47, 0x58, 0x6e, 0xdb, 0xb0, 0x43, 0xc4, 0xa3, 0xf8, 0x4d, 0x36, 0x6d, 0x9f, 0xba, 0x5e,\n\t\t0x9f, 0x83, 0xca, 0x3f, 0x26, 0x60, 0x19, 0x0f, 0xb7, 0x7d, 0x48, 0x7c, 0xdf, 0xb8, 0x22, 0x3e,\n\t\t0x6a, 0xc0, 0xd2, 0xc8, 0x69, 0x74, 0x6a, 0xf8, 0xd7, 0x7e, 0x51, 0x2a, 0x25, 0x37, 0x17, 0x76,\n\t\t0x9e, 0x56, 0x62, 0x0f, 0x55, 0x19, 0x71, 0xa3, 0x19, 0xfe, 0x35, 0x96, 0xbd, 0xa8, 0xc0, 0x47,\n\t\t0xff, 0x86, 0xfb, 0x2d, 0xc3, 0xa7, 0xba, 0x47, 0xa8, 0x67, 0x93, 0x1e, 0xb1, 0xf4, 0x36, 0x8f,\n\t\t0xa7, 0xdb, 0x56, 0x31, 0x51, 0x92, 0x36, 0x93, 0xf8, 0x5e, 0x00, 0xc0, 0xa1, 0x5e, 0x6c, 0x47,\n\t\t0xb5, 0xd0, 0x7d, 0xc8, 0x36, 0x0d, 0x5f, 0x6f, 0xbb, 0x1e, 0x29, 0x26, 0x4b, 0xd2, 0x66, 0x16,\n\t\t0x67, 0x9a, 0x86, 0x7f, 0xe8, 0x7a, 0x04, 0x61, 0x58, 0xf2, 0xfb, 0x8e, 0xa9, 0xfb, 0x4d, 0xc3,\n\t\t0xb3, 0x74, 0x9f, 0x1a, 0xb4, 0xeb, 0x17, 0xe7, 0x4a, 0xd2, 0x8c, 0xad, 0x36, 0xfa, 0x8e, 0xd9,\n\t\t0x08, 0xe0, 0x0d, 0x86, 0xc6, 0x05, 0x3f, 0x2a, 0x28, 0xff, 0x90, 0x86, 0xc2, 0xd8, 0x79, 0xd0,\n\t\t0xff, 0x61, 0x3e, 0x48, 0x83, 0x4e, 0xfb, 0x1d, 0x52, 0x94, 0x4a, 0xd2, 0x66, 0x7e, 0x67, 0xeb,\n\t\t0x76, 0xa9, 0xd0, 0xfa, 0x1d, 0x82, 0xb3, 0x54, 0xfc, 0x42, 0x8f, 0x21, 0xef, 0xbb, 0x5d, 0xcf,\n\t\t0x24, 0x2c, 0xad, 0xc3, 0xb3, 0xe7, 0xb8, 0x34, 0xb0, 0x50, 0x2d, 0xf4, 0x1a, 0x16, 0x4d, 0x8f,\n\t\t0x88, 0xf4, 0xdb, 0x6d, 0x7e, 0xec, 0x85, 0x9d, 0xb5, 0x0a, 0xaf, 0x9d, 0x4a, 0x58, 0x3b, 0x15,\n\t\t0x2d, 0xac, 0x1d, 0x9c, 0x0b, 0x0d, 0x02, 0x11, 0x32, 0xe1, 0x1e, 0xaf, 0x07, 0x1e, 0xc6, 0xa0,\n\t\t0xd4, 0xb3, 0x2f, 0xba, 0x94, 0x84, 0xc9, 0x79, 0x31, 0x65, 0xf3, 0x35, 0x66, 0x14, 0xec, 0x42,\n\t\t0x19, 0x98, 0xec, 0xdf, 0xc1, 0x2b, 0x56, 0x8c, 0x1c, 0x7d, 0x23, 0xc1, 0xc3, 0x89, 0xec, 0x4f,\n\t\t0x04, 0x4c, 0xb1, 0x80, 0x7f, 0xbf, 0xdd, 0x6d, 0x4c, 0x44, 0xde, 0xf0, 0x67, 0x01, 0x50, 0x0f,\n\t\t0x18, 0x40, 0x37, 0x4c, 0x6a, 0xf7, 0x6c, 0xda, 0x9f, 0x88, 0x9e, 0x66, 0xd1, 0xb7, 0x67, 0x44,\n\t\t0x57, 0x84, 0xe9, 0x44, 0xe8, 0x35, 0x7f, 0xaa, 0x16, 0xb5, 0x61, 0x4d, 0xbc, 0x25, 0x1e, 0xb1,\n\t\t0xb7, 0x33, 0x1a, 0x34, 0xc3, 0x82, 0x56, 0xa6, 0x04, 0xdd, 0xe7, 0x86, 0x81, 0xc7, 0xb3, 0x9d,\n\t\t0x48, 0xc4, 0xd5, 0x66, 0xbc, 0x0a, 0xb9, 0xb0, 0x76, 0x69, 0xd8, 0x2d, 0xb7, 0x47, 0x3c, 0xbd,\n\t\t0x6d, 0x78, 0xd7, 0xc4, 0x1b, 0x0d, 0x97, 0x65, 0xe1, 0xaa, 0x53, 0xc2, 0xed, 0x09, 0xc3, 0x43,\n\t\t0x66, 0x17, 0x89, 0x57, 0xbc, 0x9c, 0xa2, 0x7b, 0x93, 0x03, 0x18, 0x06, 0x28, 0xff, 0x92, 0x80,\n\t\t0x95, 0xb8, 0xca, 0x40, 0xa7, 0x20, 0x8b, 0x32, 0x73, 0x3b, 0xc4, 0x63, 0xe5, 0x27, 0x5e, 0xc7,\n\t\t0xd3, 0x99, 0x05, 0x76, 0x1c, 0xa2, 0x71, 0xc1, 0x8a, 0x0a, 0x50, 0x1e, 0x12, 0xe2, 0x51, 0xcc,\n\t\t0xe3, 0x84, 0x6d, 0xa1, 0x97, 0x90, 0xe6, 0x10, 0xf1, 0x06, 0xd6, 0xc7, 0x1c, 0x77, 0xec, 0xa1,\n\t\t0x5b, 0x2c, 0xa0, 0xe8, 0x09, 0xe4, 0x4d, 0xd7, 0xb9, 0xb4, 0xaf, 0xf4, 0x1e, 0xf1, 0xfc, 0x60,\n\t\t0x57, 0x73, 0xec, 0x95, 0x2d, 0x72, 0xe9, 0x19, 0x17, 0xa2, 0xe7, 0x20, 0x0f, 0xd2, 0x1a, 0x02,\n\t\t0x53, 0x0c, 0x58, 0x08, 0xe5, 0x21, 0xf4, 0x3f, 0x70, 0xbf, 0xe3, 0x91, 0x9e, 0xed, 0x76, 0x7d,\n\t\t0x7d, 0xc2, 0x26, 0xcd, 0x6c, 0x56, 0x43, 0xc0, 0x5e, 0xd4, 0xb6, 0xfc, 0x93, 0x04, 0x1b, 0x33,\n\t\t0xeb, 0x3c, 0xd8, 0xaf, 0x60, 0x05, 0xb3, 0xd5, 0xf5, 0x29, 0xf1, 0x58, 0x16, 0xe7, 0xf1, 0x22,\n\t\t0x97, 0xee, 0x72, 0x61, 0x40, 0x84, 0xfc, 0xa9, 0x89, 0x0c, 0xa5, 0x70, 0x86, 0xad, 0x55, 0x0b,\n\t\t0xfd, 0x0b, 0xe6, 0x07, 0x7d, 0xe4, 0x16, 0x6c, 0x31, 0x04, 0x97, 0x7f, 0x4b, 0xc1, 0xda, 0xf4,\n\t\t0x77, 0x80, 0xd6, 0x61, 0x5e, 0x5c, 0xb1, 0x6d, 0x89, 0x5d, 0x65, 0xb9, 0x40, 0xb5, 0xd0, 0x3b,\n\t\t0x40, 0x37, 0xae, 0x77, 0x7d, 0xd9, 0x72, 0x6f, 0x74, 0xf2, 0x81, 0x98, 0x5d, 0x56, 0x01, 0x89,\n\t\t0x58, 0xfe, 0xe5, 0x17, 0xf5, 0x5e, 0xc0, 0xeb, 0x21, 0x1a, 0x2f, 0xdd, 0x8c, 0x8b, 0x50, 0x11,\n\t\t0x32, 0x61, 0x6a, 0x93, 0x2c, 0xb5, 0xe1, 0x12, 0x3d, 0x84, 0x9c, 0x6f, 0x36, 0x89, 0xd5, 0x6d,\n\t\t0x11, 0x96, 0x05, 0x7e, 0xad, 0x0b, 0x03, 0x99, 0x6a, 0x21, 0x05, 0xf2, 0x43, 0x08, 0x23, 0xcf,\n\t\t0xd4, 0x47, 0xd3, 0xb1, 0x38, 0xb0, 0x60, 0xec, 0xb9, 0x01, 0xe0, 0x53, 0xc3, 0xa3, 0x3c, 0x06,\n\t\t0xbf, 0xdd, 0x79, 0x21, 0x51, 0x2d, 0xf4, 0x3f, 0xc8, 0x85, 0x6a, 0xe6, 0x3f, 0xf3, 0x51, 0xff,\n\t\t0x0b, 0x02, 0xcf, 0xbc, 0x7f, 0x01, 0xcb, 0xac, 0x13, 0x36, 0x89, 0xe1, 0xd1, 0x0b, 0x62, 0x50,\n\t\t0xee, 0x25, 0xfb, 0x51, 0x2f, 0x4b, 0x81, 0xd9, 0x7e, 0x68, 0xc5, 0x7c, 0xfd, 0x03, 0x32, 0x16,\n\t\t0xa1, 0x86, 0xdd, 0xf2, 0x8b, 0xf3, 0xcc, 0xfe, 0x41, 0x6c, 0xd6, 0x4f, 0x8c, 0x7e, 0xcb, 0x35,\n\t\t0x2c, 0x1c, 0x82, 0x83, 0x0c, 0x1b, 0x94, 0x92, 0x76, 0x87, 0x16, 0x81, 0x17, 0x92, 0x58, 0xa2,\n\t\t0xd7, 0x90, 0x63, 0xbb, 0x0b, 0x8a, 0xbc, 0xeb, 0x91, 0xe2, 0xc2, 0x0c, 0xb7, 0x7b, 0x1c, 0x83,\n\t\t0x17, 0x02, 0x0b, 0xb1, 0x40, 0x7f, 0x83, 0x15, 0xe6, 0x20, 0xb8, 0x56, 0xe2, 0xe9, 0xb6, 0x45,\n\t\t0x1c, 0x6a, 0xd3, 0x7e, 0x31, 0xc7, 0x6a, 0x07, 0x05, 0xba, 0xf7, 0x4c, 0xa5, 0x0a, 0x0d, 0x3a,\n\t\t0x82, 0x82, 0xb8, 0x5f, 0x5d, 0x10, 0x60, 0x71, 0x91, 0x45, 0x7d, 0x32, 0x85, 0x44, 0xc4, 0xc3,\n\t\t0x12, 0x44, 0x8a, 0xf3, 0xbd, 0xc8, 0xba, 0xfc, 0x6d, 0x12, 0x56, 0xa7, 0x90, 0x2c, 0x5a, 0x85,\n\t\t0x4c, 0xd8, 0x78, 0x25, 0x76, 0xaf, 0x69, 0xca, 0x5b, 0x6e, 0xa4, 0xce, 0x13, 0xb7, 0xaa, 0xf3,\n\t\t0xe4, 0xe7, 0xd6, 0xf9, 0xd7, 0x70, 0x77, 0xec, 0xe0, 0xba, 0x4d, 0x49, 0x3b, 0x68, 0xd2, 0xc1,\n\t\t0xb0, 0xf5, 0xfc, 0x56, 0xc7, 0x57, 0x29, 0x69, 0xe3, 0xe5, 0xde, 0x84, 0xcc, 0x47, 0xaf, 0x20,\n\t\t0x4d, 0x7a, 0xc4, 0xa1, 0x61, 0x0f, 0xde, 0x88, 0xa7, 0x4e, 0x83, 0x1a, 0x6f, 0x5a, 0xee, 0x05,\n\t\t0x16, 0x60, 0xb4, 0x0b, 0x79, 0x87, 0xdc, 0xe8, 0x5e, 0xd7, 0xd1, 0x85, 0x79, 0xfa, 0x36, 0xe6,\n\t\t0x39, 0x87, 0xdc, 0xe0, 0xae, 0x53, 0x67, 0x26, 0xe5, 0x9f, 0x25, 0x28, 0x4e, 0xeb, 0x3c, 0xb3,\n\t\t0x39, 0x25, 0x8e, 0x94, 0x13, 0xf1, 0xa4, 0xfc, 0xb9, 0x63, 0x52, 0xf9, 0x7b, 0x09, 0x96, 0xa3,\n\t\t0xbb, 0xd4, 0xdc, 0x6b, 0xe2, 0x04, 0x1b, 0x0c, 0x89, 0x96, 0x4f, 0xbe, 0x29, 0x9c, 0x15, 0x4c,\n\t\t0xeb, 0xa3, 0x2f, 0xa1, 0x30, 0xd6, 0x8c, 0x05, 0xe3, 0x7d, 0x6a, 0x07, 0xc6, 0xf9, 0x68, 0xff,\n\t\t0x2d, 0xff, 0x1a, 0x1d, 0xc8, 0xd9, 0x30, 0xe8, 0x5c, 0xba, 0x7f, 0x0a, 0x07, 0xaf, 0x8f, 0x4e,\n\t\t0xbc, 0x49, 0xc6, 0x11, 0xc3, 0x29, 0x76, 0xe4, 0x15, 0xcd, 0x45, 0x5e, 0xd1, 0x08, 0x73, 0xa7,\n\t\t0xa2, 0xcc, 0xfd, 0x18, 0xf2, 0x97, 0xb6, 0xe7, 0x53, 0x5e, 0x53, 0x43, 0x5e, 0xcd, 0x31, 0x29,\n\t\t0xab, 0x1a, 0xd5, 0x42, 0x65, 0x58, 0x74, 0xc8, 0x87, 0x11, 0x50, 0x86, 0x13, 0x7c, 0x20, 0x0c,\n\t\t0x31, 0xe3, 0x3d, 0x20, 0x3b, 0xd1, 0x03, 0x82, 0xea, 0x93, 0x47, 0x13, 0xc9, 0x2e, 0x75, 0xb4,\n\t\t0x7b, 0x4a, 0xd1, 0xee, 0xf9, 0x19, 0x7f, 0x4e, 0x42, 0xd3, 0x8e, 0xe7, 0x9a, 0xc4, 0xf7, 0xa3,\n\t\t0xa6, 0xc9, 0xa1, 0xe9, 0x49, 0xa8, 0x1f, 0x98, 0x96, 0xdf, 0x42, 0x61, 0x6c, 0x2c, 0x88, 0xb6,\n\t\t0x71, 0xe9, 0x53, 0xda, 0xb8, 0x03, 0x2b, 0xe2, 0xf1, 0xd7, 0x0e, 0x4e, 0x77, 0xdd, 0xae, 0x43,\n\t\t0xeb, 0x0e, 0xf5, 0xfa, 0x68, 0x05, 0x52, 0x66, 0xb0, 0x12, 0x74, 0xc7, 0x17, 0xb3, 0x26, 0x89,\n\t\t0xc9, 0x59, 0x24, 0x19, 0x33, 0x8b, 0x6c, 0xfd, 0x3e, 0x59, 0xab, 0xac, 0x34, 0x1e, 0xc2, 0x06,\n\t\t0xae, 0x9f, 0x1c, 0xa8, 0xbb, 0x8a, 0xa6, 0x1e, 0x1f, 0xe9, 0x9a, 0xd2, 0x78, 0xab, 0x6b, 0xe7,\n\t\t0x27, 0x75, 0x5d, 0x3d, 0x3a, 0x53, 0x0e, 0xd4, 0x9a, 0x7c, 0x07, 0x95, 0xe0, 0x41, 0x3c, 0xa4,\n\t\t0x76, 0x7c, 0xa8, 0xa8, 0x47, 0xb2, 0x34, 0xdd, 0xc9, 0xbe, 0xda, 0xd0, 0x8e, 0xf1, 0xb9, 0x9c,\n\t\t0x40, 0x2f, 0xe0, 0x59, 0x3c, 0xa4, 0x71, 0x7e, 0xb4, 0xab, 0x37, 0xf6, 0x15, 0x5c, 0xd3, 0x1b,\n\t\t0x9a, 0xa2, 0xbd, 0x6b, 0xc8, 0x49, 0xf4, 0x0c, 0x1e, 0xcd, 0x00, 0x2b, 0xbb, 0x9a, 0x7a, 0xa6,\n\t\t0x6a, 0xe7, 0xf2, 0x1c, 0xda, 0x82, 0xa7, 0x33, 0x03, 0xeb, 0x87, 0x75, 0x4d, 0xa9, 0x29, 0x9a,\n\t\t0x22, 0xa7, 0xd0, 0x63, 0x28, 0xcd, 0xc6, 0x9e, 0xed, 0xc8, 0x69, 0xf4, 0x1c, 0x9e, 0xc4, 0xa3,\n\t\t0xf6, 0x14, 0xf5, 0xe0, 0xf8, 0xac, 0x8e, 0xf5, 0x43, 0x05, 0xbf, 0xad, 0x63, 0x39, 0xb3, 0xf5,\n\t\t0x9d, 0x04, 0x85, 0xb1, 0xf9, 0x18, 0x3d, 0x80, 0x22, 0xcf, 0x8a, 0x7e, 0x7c, 0x52, 0xc7, 0xdc,\n\t\t0xc7, 0x30, 0x93, 0xeb, 0xb0, 0x3a, 0xa1, 0xdd, 0xc5, 0x75, 0x45, 0xab, 0xcb, 0x52, 0xac, 0xf2,\n\t\t0xdd, 0x49, 0x2d, 0x50, 0x26, 0x62, 0x95, 0xb5, 0xfa, 0x41, 0x5d, 0xab, 0xcb, 0xc9, 0xad, 0x23,\n\t\t0xc8, 0xd4, 0x0e, 0x4e, 0xd9, 0x75, 0xae, 0x80, 0x5c, 0x3b, 0x38, 0x1d, 0xbf, 0xc1, 0x22, 0xac,\n\t\t0x0c, 0xa4, 0x23, 0xa7, 0x93, 0x25, 0xb4, 0x0c, 0x85, 0x81, 0x46, 0x5c, 0x67, 0xe2, 0xcd, 0x3f,\n\t\t0xbf, 0x7a, 0x75, 0x65, 0xd3, 0x66, 0xf7, 0xa2, 0x62, 0xba, 0xed, 0xea, 0xe8, 0xa7, 0x89, 0xbf,\n\t\t0xda, 0x56, 0xab, 0x7a, 0xe5, 0xf2, 0x8f, 0x21, 0x83, 0xef, 0x14, 0xff, 0x65, 0x3f, 0x7a, 0xdb,\n\t\t0x17, 0x69, 0x26, 0x7f, 0xf9, 0x47, 0x00, 0x00, 0x00, 0xff, 0xff, 0x02, 0x9c, 0xe5, 0x90, 0x74,\n\t\t0x11, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/domain.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xb4, 0x57, 0xdd, 0x72, 0xdb, 0x44,\n\t\t0x14, 0x46, 0xce, 0x0f, 0xe9, 0xb1, 0xe3, 0x38, 0x9b, 0xa4, 0x51, 0x0d, 0x43, 0x5d, 0x97, 0x32,\n\t\t0xa6, 0x33, 0xc8, 0x4d, 0x60, 0xa0, 0xe5, 0x67, 0x06, 0x3b, 0x56, 0x8b, 0x99, 0x10, 0x32, 0xb2,\n\t\t0x1b, 0x66, 0xe8, 0x85, 0x66, 0x2d, 0xad, 0xed, 0xa5, 0xb2, 0x56, 0xb3, 0x5a, 0xbb, 0xf5, 0x15,\n\t\t0x0c, 0x3c, 0x05, 0x2f, 0xc0, 0x43, 0x70, 0xc9, 0x4b, 0x71, 0xcb, 0x68, 0xb5, 0x52, 0x2c, 0x5b,\n\t\t0xf9, 0xb9, 0xe1, 0x6e, 0xf7, 0xfc, 0x7c, 0xe7, 0xec, 0xd9, 0xef, 0x9c, 0x95, 0xa0, 0x36, 0x1d,\n\t\t0x10, 0xde, 0x74, 0xb0, 0x4b, 0x7c, 0x87, 0x34, 0x71, 0x40, 0x9b, 0xb3, 0xa3, 0xa6, 0xcb, 0x26,\n\t\t0x98, 0xfa, 0x46, 0xc0, 0x99, 0x60, 0x68, 0x2f, 0xb2, 0x30, 0x94, 0x85, 0x81, 0x03, 0x6a, 0xcc,\n\t\t0x8e, 0xaa, 0x1f, 0x8c, 0x18, 0x1b, 0x79, 0xa4, 0x29, 0x4d, 0x06, 0xd3, 0x61, 0xd3, 0x9d, 0x72,\n\t\t0x2c, 0x28, 0x53, 0x4e, 0xd5, 0xfb, 0xcb, 0x7a, 0x41, 0x27, 0x24, 0x14, 0x78, 0x12, 0x28, 0x83,\n\t\t0xdc, 0xb8, 0x0e, 0x9b, 0x4c, 0x12, 0x88, 0xfa, 0x5f, 0x00, 0x9b, 0x1d, 0x99, 0x08, 0x2a, 0x43,\n\t\t0x81, 0xba, 0xba, 0x56, 0xd3, 0x1a, 0x77, 0xac, 0x02, 0x75, 0x11, 0x82, 0x75, 0x1f, 0x4f, 0x88,\n\t\t0x5e, 0x90, 0x12, 0xb9, 0x46, 0xcf, 0x60, 0x33, 0x14, 0x58, 0x4c, 0x43, 0x7d, 0xad, 0xa6, 0x35,\n\t\t0xca, 0xc7, 0x0f, 0x8c, 0x9c, 0xbc, 0x8d, 0x18, 0xb0, 0x27, 0x0d, 0x2d, 0xe5, 0x80, 0x6a, 0x50,\n\t\t0x74, 0x49, 0xe8, 0x70, 0x1a, 0x44, 0x27, 0xd0, 0xd7, 0x25, 0xea, 0xa2, 0x08, 0xdd, 0x87, 0x22,\n\t\t0x7b, 0xe3, 0x13, 0x6e, 0x93, 0x09, 0xa6, 0x9e, 0xbe, 0x21, 0x2d, 0x40, 0x8a, 0xcc, 0x48, 0x82,\n\t\t0x9e, 0xc1, 0xba, 0x8b, 0x05, 0xd6, 0x37, 0x6b, 0x6b, 0x8d, 0xe2, 0xf1, 0xa3, 0x6b, 0x62, 0x1b,\n\t\t0x1d, 0x2c, 0xb0, 0xe9, 0x0b, 0x3e, 0xb7, 0xa4, 0x0b, 0x1a, 0xc3, 0xc3, 0x37, 0x8c, 0xbf, 0x1e,\n\t\t0x7a, 0xec, 0x8d, 0x4d, 0xde, 0x12, 0x67, 0x1a, 0x45, 0xb4, 0x39, 0x11, 0xc4, 0x97, 0xab, 0x80,\n\t\t0x70, 0xca, 0x5c, 0xfd, 0xdd, 0x9a, 0xd6, 0x28, 0x1e, 0xdf, 0x33, 0xe2, 0xc2, 0x1a, 0x49, 0x61,\n\t\t0x8d, 0x8e, 0x2a, 0xbc, 0x55, 0x4b, 0x50, 0xcc, 0x04, 0xc4, 0x4a, 0x30, 0xce, 0x25, 0x04, 0x3a,\n\t\t0x81, 0xd2, 0x00, 0xbb, 0xf6, 0x80, 0xfa, 0x98, 0x53, 0x12, 0xea, 0x5b, 0x12, 0xb2, 0x96, 0x9b,\n\t\t0x6c, 0x1b, 0xbb, 0x6d, 0x65, 0x67, 0x15, 0x07, 0x97, 0x1b, 0xf4, 0x0a, 0x0e, 0xc7, 0x34, 0x14,\n\t\t0x8c, 0xcf, 0x6d, 0xcc, 0x9d, 0x31, 0x9d, 0x61, 0xcf, 0x56, 0x85, 0xbf, 0x23, 0x0b, 0xff, 0x30,\n\t\t0x17, 0xaf, 0xa5, 0x6c, 0x55, 0xe9, 0x0f, 0x14, 0x46, 0x56, 0x8c, 0x9e, 0xc0, 0xfe, 0x0a, 0xf8,\n\t\t0x94, 0x53, 0x1d, 0x64, 0xc1, 0xd1, 0x92, 0xd3, 0x4b, 0x4e, 0x11, 0x86, 0xea, 0x8c, 0x86, 0x74,\n\t\t0x40, 0x3d, 0x2a, 0x56, 0x33, 0x2a, 0xde, 0x3e, 0x23, 0xfd, 0x12, 0x66, 0x29, 0xa9, 0xcf, 0xe1,\n\t\t0x30, 0x2f, 0x44, 0x94, 0x57, 0x49, 0xe6, 0x75, 0xb0, 0xea, 0x1a, 0xa5, 0x66, 0xc0, 0x1e, 0x76,\n\t\t0x04, 0x9d, 0x11, 0xdb, 0xf1, 0xa6, 0xa1, 0x20, 0xdc, 0x96, 0xa4, 0xdd, 0x96, 0x3e, 0xbb, 0xb1,\n\t\t0xea, 0x24, 0xd6, 0x9c, 0x45, 0x0c, 0x3e, 0x87, 0x2d, 0x65, 0x18, 0xea, 0x65, 0xc9, 0xa3, 0xcf,\n\t\t0x72, 0x13, 0x57, 0x3e, 0x16, 0x09, 0x3c, 0xea, 0xc8, 0xbb, 0x3f, 0x61, 0xfe, 0x90, 0x8e, 0x12,\n\t\t0x22, 0xa4, 0x28, 0xe8, 0x63, 0xa8, 0x0c, 0x31, 0xf5, 0xd8, 0x8c, 0x70, 0x7b, 0x46, 0x78, 0x18,\n\t\t0xb1, 0x7b, 0xa7, 0xa6, 0x35, 0xd6, 0xac, 0x9d, 0x44, 0x7e, 0x11, 0x8b, 0x51, 0x03, 0x2a, 0x34,\n\t\t0xb4, 0x47, 0x1e, 0x1b, 0x60, 0xcf, 0x8e, 0xfb, 0x5f, 0xaf, 0xd4, 0xb4, 0xc6, 0x96, 0x55, 0xa6,\n\t\t0xe1, 0x0b, 0x29, 0x56, 0xcd, 0xf8, 0x1c, 0xb6, 0x53, 0x50, 0xea, 0x0f, 0x99, 0xbe, 0x2b, 0x69,\n\t\t0x94, 0xdf, 0x6f, 0xcf, 0x95, 0x65, 0xd7, 0x1f, 0x32, 0xab, 0x34, 0x5c, 0xd8, 0xa1, 0x57, 0x51,\n\t\t0x44, 0xe6, 0xc9, 0x9c, 0xed, 0x11, 0x67, 0xd3, 0x20, 0xd4, 0x91, 0x84, 0x7a, 0x92, 0x0b, 0xd5,\n\t\t0x4d, 0x8c, 0x5f, 0x44, 0xb6, 0xd9, 0x23, 0xef, 0xd0, 0x8c, 0x32, 0x44, 0x0e, 0x1c, 0xe0, 0x70,\n\t\t0xee, 0x3b, 0x76, 0xda, 0x5a, 0x8e, 0x74, 0xd0, 0xf7, 0x64, 0x84, 0x66, 0x3e, 0x23, 0x22, 0x8f,\n\t\t0x9f, 0x94, 0x43, 0x36, 0xc0, 0x1e, 0x5e, 0xd5, 0xa1, 0x53, 0xd8, 0xc9, 0x5e, 0x70, 0xa8, 0xef,\n\t\t0x4b, 0xf8, 0x2b, 0x08, 0xb7, 0x78, 0xe3, 0xa1, 0x55, 0xce, 0x30, 0x20, 0xac, 0x7e, 0x01, 0x77,\n\t\t0xd2, 0xd1, 0x80, 0x2a, 0xb0, 0xf6, 0x9a, 0xcc, 0xd5, 0xc8, 0x8b, 0x96, 0x68, 0x1f, 0x36, 0x66,\n\t\t0xd8, 0x9b, 0x26, 0x43, 0x2f, 0xde, 0x7c, 0x59, 0x78, 0xaa, 0xd5, 0x3b, 0x70, 0xff, 0x06, 0x4a,\n\t\t0xa0, 0x07, 0x50, 0xca, 0x70, 0x30, 0xc6, 0x2d, 0x3a, 0x97, 0xec, 0xab, 0xff, 0xad, 0x41, 0x71,\n\t\t0xa1, 0xe9, 0xd1, 0xf7, 0xb0, 0x95, 0x0e, 0x0a, 0x4d, 0xb2, 0xd1, 0xb8, 0x69, 0x50, 0x18, 0xc9,\n\t\t0x22, 0x1e, 0x6f, 0xa9, 0x7f, 0xd5, 0x86, 0xed, 0x8c, 0x2a, 0xe7, 0x78, 0x4f, 0x17, 0x8f, 0x57,\n\t\t0x3c, 0xae, 0x5f, 0x1b, 0x6b, 0x2e, 0xe9, 0xb4, 0x50, 0x82, 0xdf, 0x35, 0xd8, 0xce, 0x28, 0xd1,\n\t\t0x5d, 0xd8, 0xe4, 0x04, 0x87, 0xcc, 0x57, 0x41, 0xd4, 0x0e, 0x55, 0x61, 0x8b, 0x05, 0x84, 0x63,\n\t\t0xc1, 0xb8, 0xaa, 0x64, 0xba, 0x47, 0xdf, 0x40, 0xc9, 0xe1, 0x04, 0x0b, 0xe2, 0xda, 0xd1, 0x73,\n\t\t0x25, 0x1f, 0x92, 0xe2, 0x71, 0x75, 0x65, 0xe4, 0xf6, 0x93, 0xb7, 0xcc, 0x2a, 0x2a, 0xfb, 0x48,\n\t\t0x52, 0xff, 0xa7, 0x00, 0xa5, 0x45, 0xbe, 0xe7, 0xb6, 0x9f, 0x96, 0xdf, 0x7e, 0x7d, 0xd0, 0x53,\n\t\t0xd3, 0x50, 0x60, 0x2e, 0xec, 0xf4, 0xc1, 0x54, 0x15, 0xb9, 0x2e, 0x8d, 0xbb, 0x89, 0x6f, 0x2f,\n\t\t0x72, 0x4d, 0xe5, 0xe8, 0x02, 0xee, 0xa5, 0xa8, 0xe4, 0x6d, 0x40, 0x39, 0x59, 0x80, 0xbd, 0xf9,\n\t\t0x74, 0x87, 0x89, 0xb3, 0x29, 0x7d, 0x2f, 0x71, 0x8f, 0xe1, 0xc0, 0x61, 0x93, 0xc0, 0x23, 0x51,\n\t\t0xa9, 0xc2, 0x31, 0xe6, 0xae, 0xed, 0xb0, 0xa9, 0x2f, 0xe4, 0xd3, 0xb9, 0x61, 0xed, 0xa5, 0xca,\n\t\t0x5e, 0xa4, 0x3b, 0x89, 0x54, 0xe8, 0x11, 0x94, 0x03, 0xe2, 0xbb, 0xd4, 0x1f, 0xc5, 0x1e, 0xa1,\n\t\t0xbe, 0x51, 0x5b, 0x6b, 0x6c, 0x58, 0xdb, 0x4a, 0x2a, 0x4d, 0xc3, 0xfa, 0x1f, 0xeb, 0x50, 0xce,\n\t\t0x36, 0x0a, 0x1a, 0xc3, 0x2e, 0x27, 0xa3, 0x68, 0x4a, 0x08, 0x96, 0x74, 0x9a, 0xa2, 0xe4, 0xd3,\n\t\t0x5b, 0x34, 0x9a, 0x61, 0x49, 0xe7, 0x3e, 0x53, 0x02, 0xc9, 0xc0, 0x76, 0x41, 0xd7, 0xac, 0x1d,\n\t\t0x9e, 0xd5, 0xa0, 0x3f, 0x35, 0xf8, 0x70, 0xa9, 0xa3, 0xed, 0xc1, 0x3c, 0x1d, 0xdf, 0x58, 0x08,\n\t\t0x4e, 0x07, 0x53, 0x11, 0x91, 0x34, 0x8a, 0xde, 0xbd, 0x4d, 0xf4, 0xec, 0xb6, 0x3d, 0x57, 0xab,\n\t\t0x56, 0x82, 0x15, 0xf7, 0x4a, 0x0d, 0xdf, 0x60, 0x56, 0xfd, 0x05, 0xf6, 0xf3, 0x0e, 0x92, 0xd3,\n\t\t0x4a, 0x5f, 0x67, 0x5b, 0xe9, 0xa3, 0x9b, 0xb3, 0x5c, 0x6a, 0xa7, 0xea, 0xaf, 0xf0, 0xe8, 0x56,\n\t\t0x69, 0xe7, 0x04, 0xff, 0x36, 0x1b, 0xfc, 0xf1, 0x75, 0x2f, 0x58, 0x0a, 0xd6, 0x73, 0x58, 0x40,\n\t\t0x16, 0xfb, 0xf9, 0x5f, 0x0d, 0x0e, 0x72, 0x8d, 0x50, 0x00, 0x68, 0xe5, 0x3a, 0x92, 0x01, 0xd5,\n\t\t0xba, 0x7d, 0xb0, 0x15, 0xa9, 0x9a, 0x59, 0xbb, 0xce, 0xb2, 0xbc, 0xea, 0xc1, 0xdd, 0x7c, 0xe3,\n\t\t0xff, 0xa3, 0xf4, 0x75, 0x1f, 0x76, 0x57, 0xf4, 0x57, 0x7d, 0x49, 0x68, 0x57, 0x7d, 0x49, 0xe4,\n\t\t0x0d, 0x9e, 0x42, 0xee, 0xe0, 0x79, 0xfc, 0x9b, 0x06, 0xa5, 0xc5, 0x8f, 0x62, 0x74, 0x0f, 0x0e,\n\t\t0x3a, 0x3f, 0xfe, 0xd0, 0xea, 0x9e, 0xd9, 0xbd, 0x7e, 0xab, 0xff, 0xb2, 0x67, 0x77, 0xcf, 0x2e,\n\t\t0x5a, 0xa7, 0xdd, 0x4e, 0xe5, 0x1d, 0xf4, 0x3e, 0xe8, 0x59, 0x95, 0x65, 0xbe, 0xe8, 0xf6, 0xfa,\n\t\t0xa6, 0x65, 0x76, 0x2a, 0xda, 0xaa, 0xb6, 0x63, 0x9e, 0x5b, 0xe6, 0x49, 0xab, 0x6f, 0x76, 0x2a,\n\t\t0x85, 0x55, 0xd8, 0x8e, 0x79, 0x6a, 0x46, 0xaa, 0xb5, 0xc7, 0x63, 0x28, 0x2f, 0x7d, 0x71, 0xbd,\n\t\t0x07, 0x87, 0x2d, 0xeb, 0xe4, 0xbb, 0xee, 0x45, 0xeb, 0x34, 0x37, 0x8b, 0x65, 0x65, 0xa7, 0xdb,\n\t\t0x6b, 0xb5, 0x4f, 0x65, 0x16, 0x39, 0xae, 0xe6, 0x59, 0xac, 0x2c, 0xb4, 0x5f, 0xc1, 0xa1, 0xc3,\n\t\t0x26, 0x79, 0x97, 0xd2, 0x2e, 0xc6, 0x45, 0x38, 0x8f, 0xa6, 0xe0, 0xb9, 0xf6, 0xf3, 0xd1, 0x88,\n\t\t0x8a, 0xf1, 0x74, 0x60, 0x38, 0x6c, 0xd2, 0x5c, 0xfc, 0x53, 0xf9, 0x84, 0xba, 0x5e, 0x73, 0xc4,\n\t\t0xe2, 0xff, 0x1a, 0xf5, 0xdb, 0xf2, 0x15, 0x0e, 0xe8, 0xec, 0x68, 0xb0, 0x29, 0x65, 0x9f, 0xfe,\n\t\t0x17, 0x00, 0x00, 0xff, 0xff, 0x35, 0x67, 0x87, 0x48, 0x52, 0x0d, 0x00, 0x00,\n\t},\n\t// uber/cadence/shared/v1/any.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x28, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x4f, 0xcc, 0xab, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x12, 0x03, 0xa9,\n\t\t0xd0, 0x83, 0xaa, 0xd0, 0x83, 0xa8, 0xd0, 0x2b, 0x33, 0x54, 0xb2, 0xe2, 0x62, 0x76, 0xcc, 0xab,\n\t\t0x14, 0x92, 0xe5, 0xe2, 0x2a, 0x4b, 0xcc, 0x29, 0x4d, 0x8d, 0x2f, 0xa9, 0x2c, 0x48, 0x95, 0x60,\n\t\t0x54, 0x60, 0xd4, 0xe0, 0x0c, 0xe2, 0x04, 0x8b, 0x84, 0x54, 0x16, 0xa4, 0x0a, 0x89, 0x70, 0xb1,\n\t\t0x82, 0x39, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0x3c, 0x41, 0x10, 0x8e, 0x93, 0x79, 0x94, 0x69, 0x7a,\n\t\t0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x8a, 0x13, 0xf4, 0xd2, 0x53, 0xf3,\n\t\t0xf4, 0xc1, 0x36, 0x23, 0x5c, 0x63, 0x0d, 0x61, 0x95, 0x19, 0x26, 0xb1, 0x81, 0x65, 0x8c, 0x01,\n\t\t0x01, 0x00, 0x00, 0xff, 0xff, 0x4e, 0x0a, 0xa2, 0x2e, 0xb7, 0x00, 0x00, 0x00,\n\t},\n\t// uber/cadence/shared/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0x4f, 0x4b, 0xfb, 0x30,\n\t\t0x18, 0xc7, 0xe9, 0x7e, 0xfc, 0x04, 0x33, 0xff, 0x51, 0x50, 0x46, 0x41, 0xd8, 0xa6, 0xc2, 0x4e,\n\t\t0x09, 0x9d, 0x88, 0x07, 0x4f, 0xfe, 0xc5, 0x79, 0x2c, 0xe2, 0xc1, 0x4b, 0x49, 0x93, 0xc7, 0x35,\n\t\t0xe0, 0x92, 0x92, 0xa4, 0xc1, 0xbd, 0x15, 0xdf, 0x82, 0x6f, 0x52, 0xd2, 0xd6, 0x8d, 0xb8, 0x8b,\n\t\t0xb7, 0x3e, 0x3c, 0x9f, 0xcf, 0x97, 0x6f, 0x9f, 0xa0, 0xd3, 0xba, 0x00, 0x4d, 0x18, 0xe5, 0x20,\n\t\t0x19, 0x10, 0x53, 0x52, 0x0d, 0x9c, 0xb8, 0x94, 0x94, 0xc2, 0x58, 0xa5, 0x97, 0xb8, 0xd2, 0xca,\n\t\t0xaa, 0xf8, 0xc8, 0x53, 0xb8, 0xa3, 0x70, 0x4b, 0x61, 0x97, 0x26, 0xa3, 0xc0, 0xa6, 0x95, 0xd8,\n\t\t0x50, 0x93, 0x93, 0x10, 0xe1, 0x0b, 0x21, 0x37, 0xa0, 0xf1, 0x57, 0x84, 0x0e, 0x9f, 0x35, 0x95,\n\t\t0x46, 0x80, 0xb4, 0x77, 0xc0, 0x84, 0x11, 0x4a, 0xce, 0xe4, 0x9b, 0x8a, 0x9f, 0xd0, 0xbe, 0x61,\n\t\t0x25, 0xf0, 0xfa, 0x1d, 0x78, 0x0e, 0x0e, 0xa4, 0x1d, 0x44, 0xc3, 0x68, 0xd2, 0x9f, 0x8e, 0x70,\n\t\t0xd0, 0x89, 0x56, 0x02, 0xbb, 0x14, 0x3f, 0xb6, 0xb1, 0xf7, 0x1e, 0xcc, 0xf6, 0x56, 0x66, 0x33,\n\t\t0xc7, 0x0f, 0x68, 0xd7, 0x58, 0xaa, 0xed, 0x2a, 0xa9, 0xf7, 0xd7, 0xa4, 0x9d, 0xce, 0x6b, 0xa6,\n\t\t0xf1, 0x67, 0x84, 0x0e, 0x5e, 0x40, 0xfb, 0x8e, 0x2d, 0x25, 0xc0, 0xc4, 0xd7, 0xe8, 0x98, 0xd5,\n\t\t0x5a, 0x83, 0xb4, 0xb9, 0x6b, 0x77, 0x79, 0xf7, 0x8f, 0xb9, 0x90, 0x1c, 0x3e, 0x9a, 0xda, 0xff,\n\t\t0xb3, 0xa4, 0x83, 0x02, 0x7f, 0x39, 0xf3, 0x44, 0x7c, 0x8b, 0xb6, 0xcb, 0x9f, 0xbc, 0x41, 0x6f,\n\t\t0xf8, 0x6f, 0xd2, 0x9f, 0x9e, 0xfd, 0xea, 0xe6, 0xcf, 0xe7, 0xdb, 0x85, 0x7a, 0xb6, 0xf6, 0x6e,\n\t\t0x2e, 0x5f, 0x2f, 0xe6, 0xc2, 0x96, 0x75, 0x81, 0x99, 0x5a, 0x90, 0xe0, 0xf8, 0x78, 0x0e, 0x92,\n\t\t0x34, 0x07, 0x5f, 0x3f, 0xf4, 0x55, 0xfb, 0xe5, 0xd2, 0x62, 0xab, 0xd9, 0x9c, 0x7f, 0x07, 0x00,\n\t\t0x00, 0xff, 0xff, 0xc6, 0xf4, 0xa6, 0x9c, 0x12, 0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/shared/v1/workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x2f, 0xcf, 0x2f, 0xca, 0x4e, 0xcb, 0xc9, 0x2f, 0xd7, 0x2b, 0x28, 0xca, 0x2f,\n\t\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0xba,\n\t\t0xcc, 0xc8, 0xc5, 0x1b, 0x0e, 0x55, 0x1a, 0x5c, 0x92, 0x58, 0x92, 0x2a, 0x24, 0xc5, 0x25, 0x16,\n\t\t0xee, 0x1f, 0xe4, 0xed, 0xe6, 0xe3, 0x1f, 0x1e, 0x1f, 0x1c, 0xe2, 0x18, 0xe2, 0x1a, 0xef, 0xe9,\n\t\t0x17, 0xe6, 0xe8, 0xe3, 0xe9, 0x22, 0xc0, 0x80, 0x45, 0xce, 0x39, 0xc8, 0xd5, 0x31, 0xc4, 0xd5,\n\t\t0x45, 0x80, 0x11, 0x8b, 0x5c, 0x50, 0xa8, 0x9f, 0x9f, 0xa7, 0x9f, 0xbb, 0x00, 0x93, 0x90, 0x0c,\n\t\t0x97, 0x04, 0xba, 0x3e, 0x7f, 0xdf, 0x00, 0x1f, 0x57, 0x90, 0x4e, 0x66, 0x21, 0x49, 0x2e, 0x51,\n\t\t0x34, 0xd9, 0x28, 0x7f, 0x5f, 0x27, 0x4f, 0x57, 0x01, 0x16, 0x21, 0x71, 0x2e, 0x61, 0x34, 0xa9,\n\t\t0x30, 0x7f, 0x4f, 0x17, 0x01, 0x56, 0xac, 0x26, 0x06, 0x05, 0x85, 0x06, 0x80, 0x4c, 0x64, 0x73,\n\t\t0x32, 0x8f, 0x32, 0x4d, 0xcf, 0x2c, 0xc9, 0x28, 0x4d, 0xd2, 0x4b, 0xce, 0xcf, 0xd5, 0x47, 0x09,\n\t\t0x21, 0xbd, 0xf4, 0xd4, 0x3c, 0x7d, 0x70, 0x98, 0x20, 0x02, 0xcb, 0x1a, 0xc2, 0x2a, 0x33, 0x4c,\n\t\t0x62, 0x03, 0xcb, 0x18, 0x03, 0x02, 0x00, 0x00, 0xff, 0xff, 0xff, 0xc3, 0x37, 0x89, 0x56, 0x01,\n\t\t0x00, 0x00,\n\t},\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(clientConfig transport.ClientConfig, structField reflect.StructField) HistoryAPIYARPCClient {\n\t\t\treturn NewHistoryAPIYARPCClient(clientConfig, protobuf.ClientBuilderOptions(clientConfig, structField)...)\n\t\t},\n\t)\n}\n"
  },
  {
    "path": ".gen/proto/indexer/v1/messages.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/indexer/v1/messages.proto\n\npackage indexerv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype MessageType int32\n\nconst (\n\tMessageType_MESSAGE_TYPE_INVALID MessageType = 0\n\tMessageType_MESSAGE_TYPE_INDEX   MessageType = 1\n\tMessageType_MESSAGE_TYPE_DELETE  MessageType = 2\n)\n\nvar MessageType_name = map[int32]string{\n\t0: \"MESSAGE_TYPE_INVALID\",\n\t1: \"MESSAGE_TYPE_INDEX\",\n\t2: \"MESSAGE_TYPE_DELETE\",\n}\n\nvar MessageType_value = map[string]int32{\n\t\"MESSAGE_TYPE_INVALID\": 0,\n\t\"MESSAGE_TYPE_INDEX\":   1,\n\t\"MESSAGE_TYPE_DELETE\":  2,\n}\n\nfunc (x MessageType) String() string {\n\treturn proto.EnumName(MessageType_name, int32(x))\n}\n\nfunc (MessageType) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_60256a432328b016, []int{0}\n}\n\ntype Message struct {\n\tMessageType          MessageType           `protobuf:\"varint,1,opt,name=message_type,json=messageType,proto3,enum=uber.cadence.indexer.v1.MessageType\" json:\"message_type,omitempty\"`\n\tDomainId             string                `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution `protobuf:\"bytes,3,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tVersion              int64                 `protobuf:\"varint,4,opt,name=version,proto3\" json:\"version,omitempty\"`\n\tFields               map[string]*Field     `protobuf:\"bytes,5,rep,name=fields,proto3\" json:\"fields,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}              `json:\"-\"`\n\tXXX_unrecognized     []byte                `json:\"-\"`\n\tXXX_sizecache        int32                 `json:\"-\"`\n}\n\nfunc (m *Message) Reset()         { *m = Message{} }\nfunc (m *Message) String() string { return proto.CompactTextString(m) }\nfunc (*Message) ProtoMessage()    {}\nfunc (*Message) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_60256a432328b016, []int{0}\n}\nfunc (m *Message) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *Message) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_Message.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *Message) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_Message.Merge(m, src)\n}\nfunc (m *Message) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *Message) XXX_DiscardUnknown() {\n\txxx_messageInfo_Message.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_Message proto.InternalMessageInfo\n\nfunc (m *Message) GetMessageType() MessageType {\n\tif m != nil {\n\t\treturn m.MessageType\n\t}\n\treturn MessageType_MESSAGE_TYPE_INVALID\n}\n\nfunc (m *Message) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *Message) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *Message) GetVersion() int64 {\n\tif m != nil {\n\t\treturn m.Version\n\t}\n\treturn 0\n}\n\nfunc (m *Message) GetFields() map[string]*Field {\n\tif m != nil {\n\t\treturn m.Fields\n\t}\n\treturn nil\n}\n\ntype Field struct {\n\t// Types that are valid to be assigned to Data:\n\t//\t*Field_StringData\n\t//\t*Field_IntData\n\t//\t*Field_BoolData\n\t//\t*Field_BinaryData\n\tData                 isField_Data `protobuf_oneof:\"data\"`\n\tXXX_NoUnkeyedLiteral struct{}     `json:\"-\"`\n\tXXX_unrecognized     []byte       `json:\"-\"`\n\tXXX_sizecache        int32        `json:\"-\"`\n}\n\nfunc (m *Field) Reset()         { *m = Field{} }\nfunc (m *Field) String() string { return proto.CompactTextString(m) }\nfunc (*Field) ProtoMessage()    {}\nfunc (*Field) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_60256a432328b016, []int{1}\n}\nfunc (m *Field) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *Field) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_Field.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *Field) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_Field.Merge(m, src)\n}\nfunc (m *Field) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *Field) XXX_DiscardUnknown() {\n\txxx_messageInfo_Field.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_Field proto.InternalMessageInfo\n\ntype isField_Data interface {\n\tisField_Data()\n\tMarshalTo([]byte) (int, error)\n\tSize() int\n}\n\ntype Field_StringData struct {\n\tStringData string `protobuf:\"bytes,1,opt,name=string_data,json=stringData,proto3,oneof\" json:\"string_data,omitempty\"`\n}\ntype Field_IntData struct {\n\tIntData int64 `protobuf:\"varint,2,opt,name=int_data,json=intData,proto3,oneof\" json:\"int_data,omitempty\"`\n}\ntype Field_BoolData struct {\n\tBoolData bool `protobuf:\"varint,3,opt,name=bool_data,json=boolData,proto3,oneof\" json:\"bool_data,omitempty\"`\n}\ntype Field_BinaryData struct {\n\tBinaryData []byte `protobuf:\"bytes,4,opt,name=binary_data,json=binaryData,proto3,oneof\" json:\"binary_data,omitempty\"`\n}\n\nfunc (*Field_StringData) isField_Data() {}\nfunc (*Field_IntData) isField_Data()    {}\nfunc (*Field_BoolData) isField_Data()   {}\nfunc (*Field_BinaryData) isField_Data() {}\n\nfunc (m *Field) GetData() isField_Data {\n\tif m != nil {\n\t\treturn m.Data\n\t}\n\treturn nil\n}\n\nfunc (m *Field) GetStringData() string {\n\tif x, ok := m.GetData().(*Field_StringData); ok {\n\t\treturn x.StringData\n\t}\n\treturn \"\"\n}\n\nfunc (m *Field) GetIntData() int64 {\n\tif x, ok := m.GetData().(*Field_IntData); ok {\n\t\treturn x.IntData\n\t}\n\treturn 0\n}\n\nfunc (m *Field) GetBoolData() bool {\n\tif x, ok := m.GetData().(*Field_BoolData); ok {\n\t\treturn x.BoolData\n\t}\n\treturn false\n}\n\nfunc (m *Field) GetBinaryData() []byte {\n\tif x, ok := m.GetData().(*Field_BinaryData); ok {\n\t\treturn x.BinaryData\n\t}\n\treturn nil\n}\n\n// XXX_OneofWrappers is for the internal use of the proto package.\nfunc (*Field) XXX_OneofWrappers() []interface{} {\n\treturn []interface{}{\n\t\t(*Field_StringData)(nil),\n\t\t(*Field_IntData)(nil),\n\t\t(*Field_BoolData)(nil),\n\t\t(*Field_BinaryData)(nil),\n\t}\n}\n\nfunc init() {\n\tproto.RegisterEnum(\"uber.cadence.indexer.v1.MessageType\", MessageType_name, MessageType_value)\n\tproto.RegisterType((*Message)(nil), \"uber.cadence.indexer.v1.Message\")\n\tproto.RegisterMapType((map[string]*Field)(nil), \"uber.cadence.indexer.v1.Message.FieldsEntry\")\n\tproto.RegisterType((*Field)(nil), \"uber.cadence.indexer.v1.Field\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/indexer/v1/messages.proto\", fileDescriptor_60256a432328b016)\n}\n\nvar fileDescriptor_60256a432328b016 = []byte{\n\t// 488 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x84, 0x92, 0xdf, 0x6a, 0x13, 0x41,\n\t0x14, 0xc6, 0x33, 0xd9, 0xe6, 0xdf, 0xd9, 0x22, 0x71, 0x14, 0xbb, 0xb4, 0x18, 0xb6, 0x45, 0x4a,\n\t0x10, 0xd9, 0x25, 0x51, 0x50, 0xf4, 0xaa, 0x25, 0x6b, 0x13, 0x68, 0x45, 0xa6, 0x51, 0x5b, 0x6f,\n\t0x96, 0x49, 0x76, 0x1a, 0x87, 0x66, 0x67, 0xc2, 0xee, 0x24, 0xe9, 0x5e, 0xfa, 0x08, 0xbe, 0x95,\n\t0x97, 0x3e, 0x82, 0xe4, 0x49, 0x64, 0x76, 0xb6, 0x98, 0x14, 0x4a, 0xef, 0x66, 0xbe, 0xef, 0x77,\n\t0xbe, 0xc3, 0x9c, 0x39, 0x70, 0x38, 0x1f, 0xb1, 0xc4, 0x1f, 0xd3, 0x88, 0x89, 0x31, 0xf3, 0xb9,\n\t0x88, 0xd8, 0x0d, 0x4b, 0xfc, 0x45, 0xc7, 0x8f, 0x59, 0x9a, 0xd2, 0x09, 0x4b, 0xbd, 0x59, 0x22,\n\t0x95, 0xc4, 0x3b, 0x9a, 0xf3, 0x0a, 0xce, 0x2b, 0x38, 0x6f, 0xd1, 0xd9, 0x75, 0x37, 0x02, 0xe8,\n\t0x8c, 0xeb, 0xe2, 0xb1, 0x8c, 0x63, 0x29, 0x4c, 0xe9, 0xc1, 0x4f, 0x0b, 0x6a, 0x67, 0x26, 0x0d,\n\t0x9f, 0xc0, 0x76, 0x11, 0x1c, 0xaa, 0x6c, 0xc6, 0x1c, 0xe4, 0xa2, 0xf6, 0xa3, 0xee, 0x0b, 0xef,\n\t0x9e, 0x74, 0xaf, 0xa8, 0x1b, 0x66, 0x33, 0x46, 0xec, 0xf8, 0xff, 0x05, 0xef, 0x41, 0x23, 0x92,\n\t0x31, 0xe5, 0x22, 0xe4, 0x91, 0x53, 0x76, 0x51, 0xbb, 0x41, 0xea, 0x46, 0x18, 0x44, 0xf8, 0x0b,\n\t0xe0, 0xa5, 0x4c, 0xae, 0xaf, 0xa6, 0x72, 0x19, 0xb2, 0x1b, 0x36, 0x9e, 0x2b, 0x2e, 0x85, 0x63,\n\t0xb9, 0xa8, 0x6d, 0x77, 0x0f, 0x37, 0x7b, 0xd1, 0x19, 0xd7, 0x7d, 0xbe, 0x15, 0x78, 0x70, 0x4b,\n\t0x93, 0xc7, 0xcb, 0xbb, 0x12, 0x76, 0xa0, 0xb6, 0x60, 0x49, 0xaa, 0xb3, 0xb6, 0x5c, 0xd4, 0xb6,\n\t0xc8, 0xed, 0x15, 0xf7, 0xa0, 0x7a, 0xc5, 0xd9, 0x34, 0x4a, 0x9d, 0x8a, 0x6b, 0xb5, 0xed, 0xee,\n\t0xab, 0x87, 0x1e, 0xe4, 0x7d, 0xcc, 0xf1, 0x40, 0xa8, 0x24, 0x23, 0x45, 0xed, 0xee, 0x25, 0xd8,\n\t0x6b, 0x32, 0x6e, 0x82, 0x75, 0xcd, 0xb2, 0x7c, 0x44, 0x0d, 0xa2, 0x8f, 0xf8, 0x0d, 0x54, 0x16,\n\t0x74, 0x3a, 0x67, 0xf9, 0x83, 0xed, 0x6e, 0xeb, 0xde, 0x2e, 0x79, 0x0c, 0x31, 0xf0, 0xfb, 0xf2,\n\t0x3b, 0x74, 0xf0, 0x0b, 0x41, 0x25, 0x17, 0xf1, 0x3e, 0xd8, 0xa9, 0x4a, 0xb8, 0x98, 0x84, 0x11,\n\t0x55, 0xd4, 0xa4, 0xf7, 0x4b, 0x04, 0x8c, 0xd8, 0xa3, 0x8a, 0xe2, 0x3d, 0xa8, 0x73, 0xa1, 0x8c,\n\t0xaf, 0x3b, 0x59, 0xfd, 0x12, 0xa9, 0x71, 0xa1, 0x72, 0xf3, 0x39, 0x34, 0x46, 0x52, 0x4e, 0x8d,\n\t0xab, 0x47, 0x5a, 0xef, 0x97, 0x48, 0x5d, 0x4b, 0xb9, 0xbd, 0x0f, 0xf6, 0x88, 0x0b, 0x9a, 0x64,\n\t0x06, 0xd0, 0x73, 0xda, 0xd6, 0xf1, 0x46, 0xd4, 0xc8, 0x71, 0x15, 0xb6, 0xb4, 0xf7, 0xf2, 0x02,\n\t0xec, 0xb5, 0xef, 0xc5, 0x0e, 0x3c, 0x3d, 0x0b, 0xce, 0xcf, 0x8f, 0x4e, 0x82, 0x70, 0x78, 0xf9,\n\t0x39, 0x08, 0x07, 0x9f, 0xbe, 0x1e, 0x9d, 0x0e, 0x7a, 0xcd, 0x12, 0x7e, 0x06, 0xf8, 0x8e, 0xd3,\n\t0x0b, 0x2e, 0x9a, 0x08, 0xef, 0xc0, 0x93, 0x0d, 0xbd, 0x17, 0x9c, 0x06, 0xc3, 0xa0, 0x59, 0x3e,\n\t0x0e, 0x7e, 0xaf, 0x5a, 0xe8, 0xcf, 0xaa, 0x85, 0xfe, 0xae, 0x5a, 0xe8, 0xfb, 0xdb, 0x09, 0x57,\n\t0x3f, 0xe6, 0x23, 0x6f, 0x2c, 0x63, 0x7f, 0x63, 0x59, 0xbd, 0x09, 0x13, 0x7e, 0xbe, 0xa3, 0x6b,\n\t0x8b, 0xff, 0xa1, 0x38, 0x2e, 0x3a, 0xa3, 0x6a, 0xee, 0xbd, 0xfe, 0x17, 0x00, 0x00, 0xff, 0xff,\n\t0x94, 0x56, 0xcb, 0x00, 0x24, 0x03, 0x00, 0x00,\n}\n\nfunc (m *Message) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *Message) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Message) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Fields) > 0 {\n\t\tfor k := range m.Fields {\n\t\t\tv := m.Fields[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintMessages(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintMessages(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintMessages(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x2a\n\t\t}\n\t}\n\tif m.Version != 0 {\n\t\ti = encodeVarintMessages(dAtA, i, uint64(m.Version))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintMessages(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintMessages(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.MessageType != 0 {\n\t\ti = encodeVarintMessages(dAtA, i, uint64(m.MessageType))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *Field) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *Field) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Field) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Data != nil {\n\t\t{\n\t\t\tsize := m.Data.Size()\n\t\t\ti -= size\n\t\t\tif _, err := m.Data.MarshalTo(dAtA[i:]); err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *Field_StringData) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Field_StringData) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\ti -= len(m.StringData)\n\tcopy(dAtA[i:], m.StringData)\n\ti = encodeVarintMessages(dAtA, i, uint64(len(m.StringData)))\n\ti--\n\tdAtA[i] = 0xa\n\treturn len(dAtA) - i, nil\n}\nfunc (m *Field_IntData) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Field_IntData) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\ti = encodeVarintMessages(dAtA, i, uint64(m.IntData))\n\ti--\n\tdAtA[i] = 0x10\n\treturn len(dAtA) - i, nil\n}\nfunc (m *Field_BoolData) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Field_BoolData) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\ti--\n\tif m.BoolData {\n\t\tdAtA[i] = 1\n\t} else {\n\t\tdAtA[i] = 0\n\t}\n\ti--\n\tdAtA[i] = 0x18\n\treturn len(dAtA) - i, nil\n}\nfunc (m *Field_BinaryData) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Field_BinaryData) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\tif m.BinaryData != nil {\n\t\ti -= len(m.BinaryData)\n\t\tcopy(dAtA[i:], m.BinaryData)\n\t\ti = encodeVarintMessages(dAtA, i, uint64(len(m.BinaryData)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\treturn len(dAtA) - i, nil\n}\nfunc encodeVarintMessages(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovMessages(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *Message) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.MessageType != 0 {\n\t\tn += 1 + sovMessages(uint64(m.MessageType))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovMessages(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovMessages(uint64(l))\n\t}\n\tif m.Version != 0 {\n\t\tn += 1 + sovMessages(uint64(m.Version))\n\t}\n\tif len(m.Fields) > 0 {\n\t\tfor k, v := range m.Fields {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovMessages(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovMessages(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovMessages(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *Field) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Data != nil {\n\t\tn += m.Data.Size()\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *Field_StringData) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.StringData)\n\tn += 1 + l + sovMessages(uint64(l))\n\treturn n\n}\nfunc (m *Field_IntData) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tn += 1 + sovMessages(uint64(m.IntData))\n\treturn n\n}\nfunc (m *Field_BoolData) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tn += 2\n\treturn n\n}\nfunc (m *Field_BinaryData) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.BinaryData != nil {\n\t\tl = len(m.BinaryData)\n\t\tn += 1 + l + sovMessages(uint64(l))\n\t}\n\treturn n\n}\n\nfunc sovMessages(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozMessages(x uint64) (n int) {\n\treturn sovMessages(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *Message) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: Message: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: Message: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field MessageType\", wireType)\n\t\t\t}\n\t\t\tm.MessageType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.MessageType |= MessageType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Version\", wireType)\n\t\t\t}\n\t\t\tm.Version = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Version |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Fields\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Fields == nil {\n\t\t\t\tm.Fields = make(map[string]*Field)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *Field\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &Field{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipMessages(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Fields[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipMessages(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *Field) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: Field: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: Field: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StringData\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Data = &Field_StringData{string(dAtA[iNdEx:postIndex])}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IntData\", wireType)\n\t\t\t}\n\t\t\tvar v int64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Data = &Field_IntData{v}\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BoolData\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tb := bool(v != 0)\n\t\t\tm.Data = &Field_BoolData{b}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BinaryData\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tv := make([]byte, postIndex-iNdEx)\n\t\t\tcopy(v, dAtA[iNdEx:postIndex])\n\t\t\tm.Data = &Field_BinaryData{v}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipMessages(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipMessages(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowMessages\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowMessages\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthMessages\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupMessages\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthMessages\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthMessages        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowMessages          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupMessages = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/indexer/v1/messages.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/indexer/v1/messages.proto\n\npackage indexerv1\n\nvar yarpcFileDescriptorClosure60256a432328b016 = [][]byte{\n\t// uber/cadence/indexer/v1/messages.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x84, 0x92, 0x5f, 0x6b, 0xdb, 0x30,\n\t\t0x14, 0xc5, 0xe3, 0xba, 0xf9, 0x77, 0x5d, 0x46, 0xa6, 0x8d, 0xd5, 0xb4, 0x6c, 0xb8, 0x65, 0x94,\n\t\t0x30, 0x86, 0x4d, 0xb2, 0xc1, 0xfe, 0x3d, 0xb5, 0xd8, 0x6b, 0x02, 0xed, 0x18, 0x6a, 0xb6, 0xb5,\n\t\t0x7b, 0x31, 0xb2, 0xad, 0x66, 0xa2, 0xb1, 0x64, 0x6c, 0xc5, 0xa9, 0x1f, 0xf7, 0x11, 0xf6, 0x8d,\n\t\t0x87, 0x2c, 0x97, 0x25, 0x85, 0xb2, 0x37, 0xe9, 0x9c, 0xdf, 0x3d, 0x17, 0x5d, 0x5d, 0x38, 0x5a,\n\t\t0x46, 0x34, 0xf7, 0x62, 0x92, 0x50, 0x1e, 0x53, 0x8f, 0xf1, 0x84, 0xde, 0xd2, 0xdc, 0x2b, 0x47,\n\t\t0x5e, 0x4a, 0x8b, 0x82, 0xcc, 0x69, 0xe1, 0x66, 0xb9, 0x90, 0x02, 0xed, 0x2a, 0xce, 0x6d, 0x38,\n\t\t0xb7, 0xe1, 0xdc, 0x72, 0xb4, 0xe7, 0x6c, 0x04, 0x90, 0x8c, 0xa9, 0xe2, 0x58, 0xa4, 0xa9, 0xe0,\n\t\t0xba, 0xf4, 0xf0, 0xb7, 0x09, 0xdd, 0x73, 0x9d, 0x86, 0x4e, 0x61, 0xa7, 0x09, 0x0e, 0x65, 0x95,\n\t\t0x51, 0xdb, 0x70, 0x8c, 0xe1, 0xa3, 0xf1, 0x4b, 0xf7, 0x81, 0x74, 0xb7, 0xa9, 0x9b, 0x55, 0x19,\n\t\t0xc5, 0x56, 0xfa, 0xef, 0x82, 0xf6, 0xa1, 0x9f, 0x88, 0x94, 0x30, 0x1e, 0xb2, 0xc4, 0xde, 0x72,\n\t\t0x8c, 0x61, 0x1f, 0xf7, 0xb4, 0x30, 0x4d, 0xd0, 0x37, 0x40, 0x2b, 0x91, 0xdf, 0x5c, 0x2f, 0xc4,\n\t\t0x2a, 0xa4, 0xb7, 0x34, 0x5e, 0x4a, 0x26, 0xb8, 0x6d, 0x3a, 0xc6, 0xd0, 0x1a, 0x1f, 0x6d, 0xf6,\n\t\t0x22, 0x19, 0x53, 0x7d, 0x7e, 0x34, 0x78, 0x70, 0x47, 0xe3, 0xc7, 0xab, 0xfb, 0x12, 0xb2, 0xa1,\n\t\t0x5b, 0xd2, 0xbc, 0x50, 0x59, 0xdb, 0x8e, 0x31, 0x34, 0xf1, 0xdd, 0x15, 0xf9, 0xd0, 0xb9, 0x66,\n\t\t0x74, 0x91, 0x14, 0x76, 0xdb, 0x31, 0x87, 0xd6, 0xf8, 0xf5, 0xff, 0x1e, 0xe4, 0x7e, 0xae, 0xf1,\n\t\t0x80, 0xcb, 0xbc, 0xc2, 0x4d, 0xed, 0xde, 0x15, 0x58, 0x6b, 0x32, 0x1a, 0x80, 0x79, 0x43, 0xab,\n\t\t0x7a, 0x44, 0x7d, 0xac, 0x8e, 0xe8, 0x2d, 0xb4, 0x4b, 0xb2, 0x58, 0xd2, 0xfa, 0xc1, 0xd6, 0xf8,\n\t\t0xc5, 0x83, 0x5d, 0xea, 0x18, 0xac, 0xe1, 0x8f, 0x5b, 0xef, 0x8d, 0xc3, 0x3f, 0x06, 0xb4, 0x6b,\n\t\t0x11, 0x1d, 0x80, 0x55, 0xc8, 0x9c, 0xf1, 0x79, 0x98, 0x10, 0x49, 0x74, 0xfa, 0xa4, 0x85, 0x41,\n\t\t0x8b, 0x3e, 0x91, 0x04, 0xed, 0x43, 0x8f, 0x71, 0xa9, 0x7d, 0xd5, 0xc9, 0x9c, 0xb4, 0x70, 0x97,\n\t\t0x71, 0x59, 0x9b, 0xcf, 0xa1, 0x1f, 0x09, 0xb1, 0xd0, 0xae, 0x1a, 0x69, 0x6f, 0xd2, 0xc2, 0x3d,\n\t\t0x25, 0xd5, 0xf6, 0x01, 0x58, 0x11, 0xe3, 0x24, 0xaf, 0x34, 0xa0, 0xe6, 0xb4, 0xa3, 0xe2, 0xb5,\n\t\t0xa8, 0x90, 0x93, 0x0e, 0x6c, 0x2b, 0xef, 0xd5, 0x25, 0x58, 0x6b, 0xdf, 0x8b, 0x6c, 0x78, 0x7a,\n\t\t0x1e, 0x5c, 0x5c, 0x1c, 0x9f, 0x06, 0xe1, 0xec, 0xea, 0x6b, 0x10, 0x4e, 0xbf, 0x7c, 0x3f, 0x3e,\n\t\t0x9b, 0xfa, 0x83, 0x16, 0x7a, 0x06, 0xe8, 0x9e, 0xe3, 0x07, 0x97, 0x03, 0x03, 0xed, 0xc2, 0x93,\n\t\t0x0d, 0xdd, 0x0f, 0xce, 0x82, 0x59, 0x30, 0xd8, 0x3a, 0xf9, 0xf0, 0xf3, 0xdd, 0x9c, 0xc9, 0x5f,\n\t\t0xcb, 0xc8, 0x8d, 0x45, 0xea, 0x6d, 0x2c, 0xa8, 0x3b, 0xa7, 0xdc, 0xab, 0xf7, 0x72, 0x6d, 0xd9,\n\t\t0x3f, 0x35, 0xc7, 0x72, 0x14, 0x75, 0x6a, 0xef, 0xcd, 0xdf, 0x00, 0x00, 0x00, 0xff, 0xff, 0xc0,\n\t\t0x8b, 0x53, 0x80, 0x18, 0x03, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/common.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x57, 0xdd, 0x72, 0x22, 0xc7,\n\t\t0x15, 0xf6, 0xc0, 0xa2, 0x9f, 0x03, 0xbb, 0x42, 0xad, 0xfd, 0x61, 0xb5, 0x5e, 0xaf, 0x16, 0x97,\n\t\t0x63, 0x79, 0x2b, 0x86, 0x88, 0x4d, 0x52, 0x2e, 0x3b, 0x4e, 0x82, 0xd0, 0x48, 0x9a, 0x5d, 0x02,\n\t\t0xa4, 0x99, 0x95, 0xac, 0xa4, 0xca, 0x53, 0xcd, 0x4c, 0x83, 0x3b, 0x0c, 0xd3, 0x93, 0x99, 0x1e,\n\t\t0x56, 0xf8, 0x22, 0x95, 0xcb, 0xe4, 0x26, 0x8f, 0x90, 0x8b, 0xbc, 0x48, 0x1e, 0x20, 0x97, 0x79,\n\t\t0x97, 0x5c, 0xa7, 0xba, 0xa7, 0x07, 0x81, 0xc2, 0x1a, 0x5f, 0xa4, 0x7c, 0x47, 0x9f, 0xf3, 0x7d,\n\t\t0xa7, 0xbf, 0xd3, 0xdd, 0xe7, 0x1c, 0x06, 0x0e, 0x92, 0x01, 0x8d, 0xea, 0x2e, 0xf1, 0x68, 0xe0,\n\t\t0xd2, 0x3a, 0x09, 0x59, 0x7d, 0x7a, 0x54, 0x77, 0xf9, 0x64, 0xc2, 0x83, 0x5a, 0x18, 0x71, 0xc1,\n\t\t0xd1, 0x9e, 0x44, 0xd4, 0x34, 0xa2, 0x46, 0x42, 0x56, 0x9b, 0x1e, 0xed, 0x7f, 0x30, 0xe2, 0x7c,\n\t\t0xe4, 0xd3, 0xba, 0x82, 0x0c, 0x92, 0x61, 0xdd, 0x4b, 0x22, 0x22, 0x58, 0x46, 0xaa, 0xbe, 0x86,\n\t\t0xdd, 0x4b, 0x1e, 0x8d, 0x87, 0x3e, 0x7f, 0x6b, 0x5e, 0x53, 0x37, 0x91, 0x2e, 0xf4, 0x0c, 0x8a,\n\t\t0x6f, 0xb5, 0xd1, 0x61, 0x5e, 0xc5, 0x38, 0x30, 0x0e, 0xb7, 0x31, 0x64, 0x26, 0xcb, 0x43, 0x0f,\n\t\t0x60, 0x23, 0x4a, 0x02, 0xe9, 0xcb, 0x29, 0x5f, 0x21, 0x4a, 0x02, 0xcb, 0xab, 0x56, 0xa1, 0x94,\n\t\t0x05, 0xb3, 0x67, 0x21, 0x45, 0x08, 0xee, 0x04, 0x64, 0x42, 0x75, 0x00, 0xf5, 0x5b, 0x62, 0x9a,\n\t\t0xae, 0x60, 0x53, 0x26, 0x66, 0xef, 0xc4, 0x3c, 0x85, 0xcd, 0x1e, 0x99, 0xf9, 0x9c, 0x78, 0xd2,\n\t\t0xed, 0x11, 0x41, 0x94, 0xbb, 0x84, 0xd5, 0xef, 0xea, 0x17, 0xb0, 0x79, 0x4a, 0x98, 0x9f, 0x44,\n\t\t0x14, 0x3d, 0x84, 0x8d, 0x88, 0x92, 0x98, 0x07, 0x9a, 0xaf, 0x57, 0xa8, 0x02, 0x9b, 0x1e, 0x15,\n\t\t0x84, 0xf9, 0xb1, 0x52, 0x58, 0xc2, 0xd9, 0xb2, 0xfa, 0x77, 0x03, 0xee, 0xfc, 0x86, 0x4e, 0x38,\n\t\t0xfa, 0x12, 0x36, 0x86, 0x8c, 0xfa, 0x5e, 0x5c, 0x31, 0x0e, 0xf2, 0x87, 0xc5, 0xc6, 0x47, 0xb5,\n\t\t0x15, 0xe7, 0x57, 0x93, 0xd0, 0xda, 0xa9, 0xc2, 0x99, 0x81, 0x88, 0x66, 0x58, 0x93, 0xf6, 0x2f,\n\t\t0xa1, 0xb8, 0x60, 0x46, 0x65, 0xc8, 0x8f, 0xe9, 0x4c, 0xab, 0x90, 0x3f, 0x51, 0x03, 0x0a, 0x53,\n\t\t0xe2, 0x27, 0x54, 0x09, 0x28, 0x36, 0xde, 0x5f, 0x19, 0x5e, 0xa7, 0x89, 0x53, 0xe8, 0xe7, 0xb9,\n\t\t0xcf, 0x8c, 0xea, 0x3f, 0x0c, 0xd8, 0x38, 0xa7, 0xc4, 0xa3, 0x11, 0xfa, 0xd5, 0x2d, 0x89, 0x1f,\n\t\t0xaf, 0x8c, 0x91, 0x82, 0x7f, 0x58, 0x91, 0xff, 0x36, 0xa0, 0xdc, 0xa7, 0x24, 0x72, 0xbf, 0x69,\n\t\t0x0a, 0x11, 0xb1, 0x41, 0x22, 0x68, 0x8c, 0x1c, 0xb8, 0xc7, 0x02, 0x8f, 0x5e, 0x53, 0xcf, 0x59,\n\t\t0x92, 0xfd, 0xd9, 0xca, 0xa8, 0xb7, 0xe9, 0x35, 0x2b, 0xe5, 0x2e, 0xe6, 0x71, 0x97, 0x2d, 0xda,\n\t\t0xf6, 0xbf, 0x06, 0xf4, 0xbf, 0xa0, 0xff, 0x63, 0x56, 0x43, 0xd8, 0x3a, 0x21, 0x82, 0x1c, 0xfb,\n\t\t0x7c, 0x80, 0x4e, 0xe1, 0x2e, 0x0d, 0x5c, 0xee, 0xb1, 0x60, 0xe4, 0x88, 0x59, 0x98, 0x3e, 0xd0,\n\t\t0x7b, 0x8d, 0xe7, 0x2b, 0x63, 0x99, 0x1a, 0x29, 0x5f, 0x34, 0x2e, 0xd1, 0x85, 0xd5, 0xfc, 0x01,\n\t\t0xe7, 0x16, 0x1e, 0x70, 0x2f, 0x2d, 0x3a, 0x1a, 0x5d, 0xd0, 0x28, 0x66, 0x3c, 0xb0, 0x82, 0x21,\n\t\t0x97, 0x40, 0x36, 0x09, 0xfd, 0xac, 0x10, 0xe4, 0x6f, 0xf4, 0x31, 0xec, 0x0c, 0x29, 0x11, 0x49,\n\t\t0x44, 0x9d, 0x69, 0x0a, 0xd5, 0x05, 0x77, 0x4f, 0x9b, 0x75, 0x80, 0xea, 0x6b, 0x78, 0xd4, 0x4f,\n\t\t0xc2, 0x90, 0x47, 0x82, 0x7a, 0x2d, 0x9f, 0xd1, 0x40, 0x68, 0x4f, 0x2c, 0x6b, 0x75, 0xc4, 0x9d,\n\t\t0xd8, 0x1b, 0xeb, 0xc8, 0x85, 0x11, 0xef, 0x7b, 0x63, 0xf4, 0x18, 0xb6, 0xfe, 0x40, 0xa6, 0x44,\n\t\t0x39, 0xd2, 0x98, 0x9b, 0x72, 0xdd, 0xf7, 0xc6, 0xd5, 0x3f, 0xe7, 0xa1, 0x88, 0xa9, 0x88, 0x66,\n\t\t0x3d, 0xee, 0x33, 0x77, 0x86, 0x4e, 0xa0, 0xcc, 0x02, 0x26, 0x18, 0xf1, 0x1d, 0x16, 0x08, 0x1a,\n\t\t0x4d, 0x49, 0xaa, 0xb2, 0xd8, 0x78, 0x5c, 0x4b, 0xdb, 0x4b, 0x2d, 0x6b, 0x2f, 0xb5, 0x13, 0xdd,\n\t\t0x5e, 0xf0, 0x8e, 0xa6, 0x58, 0x9a, 0x81, 0xea, 0xb0, 0x37, 0x20, 0xee, 0x98, 0x0f, 0x87, 0x8e,\n\t\t0xcb, 0xe9, 0x70, 0xc8, 0x5c, 0x29, 0x53, 0xed, 0x6d, 0x60, 0xa4, 0x5d, 0xad, 0x1b, 0x8f, 0xdc,\n\t\t0x76, 0x42, 0xae, 0xd9, 0x24, 0x99, 0xdc, 0x6c, 0x9b, 0x5f, 0xbb, 0xad, 0xa6, 0xcc, 0xb7, 0xfd,\n\t\t0xe4, 0x26, 0x0a, 0x11, 0x82, 0x4e, 0x42, 0x11, 0x57, 0xee, 0x1c, 0x18, 0x87, 0x85, 0x39, 0xb4,\n\t\t0xa9, 0xcd, 0xe8, 0x4b, 0x78, 0x12, 0xf0, 0xc0, 0x89, 0x64, 0xea, 0x64, 0xe0, 0x53, 0x87, 0x46,\n\t\t0x11, 0x8f, 0x9c, 0xb4, 0xa5, 0xc4, 0x95, 0xc2, 0x41, 0xfe, 0x70, 0x1b, 0x57, 0x02, 0x1e, 0xe0,\n\t\t0x0c, 0x61, 0x4a, 0x00, 0x4e, 0xfd, 0xe8, 0x15, 0xec, 0xd1, 0xeb, 0x90, 0xa5, 0x42, 0x6e, 0x24,\n\t\t0x6f, 0xac, 0x93, 0x8c, 0x6e, 0x58, 0x99, 0xea, 0xea, 0x04, 0x1e, 0x59, 0x31, 0xf7, 0x95, 0xf1,\n\t\t0x2c, 0xe2, 0x49, 0xd8, 0x23, 0x91, 0x60, 0xaa, 0x39, 0xaf, 0x68, 0x98, 0xe8, 0x97, 0x50, 0x88,\n\t\t0x05, 0x11, 0xe9, 0x83, 0xbf, 0xd7, 0x38, 0x5c, 0xf9, 0x48, 0x97, 0x03, 0xf6, 0x25, 0x1e, 0xa7,\n\t\t0xb4, 0xea, 0x14, 0x9e, 0x2c, 0x7b, 0x5b, 0x3c, 0x18, 0xb2, 0x91, 0x56, 0x88, 0x2e, 0xa1, 0xcc,\n\t\t0x32, 0xb7, 0x33, 0x92, 0xfe, 0xac, 0xb4, 0x7f, 0xfc, 0x3d, 0x76, 0x9a, 0x4b, 0xc7, 0x3b, 0x6c,\n\t\t0xc9, 0x11, 0x57, 0xff, 0x65, 0xc0, 0x7e, 0x33, 0x9e, 0x05, 0x6e, 0x36, 0x36, 0x96, 0xf7, 0xad,\n\t\t0xc0, 0x26, 0x0d, 0xe4, 0x39, 0xa7, 0x33, 0x68, 0x0b, 0x67, 0x4b, 0xd4, 0x80, 0x07, 0x61, 0x44,\n\t\t0x3d, 0x3a, 0x64, 0x01, 0xf5, 0x9c, 0x3f, 0x26, 0x34, 0xa1, 0x8e, 0x3a, 0x95, 0xf4, 0x29, 0xef,\n\t\t0xdd, 0x38, 0x7f, 0x2b, 0x7d, 0x1d, 0x79, 0x48, 0x4f, 0x01, 0x52, 0xa0, 0x2a, 0xe7, 0xbc, 0x02,\n\t\t0x6e, 0x2b, 0x8b, 0x2a, 0xd4, 0x5f, 0x43, 0x29, 0x75, 0xbb, 0x4a, 0x83, 0x7a, 0x24, 0xc5, 0xc6,\n\t\t0xd3, 0x95, 0x09, 0x66, 0x5d, 0x02, 0x17, 0x15, 0x25, 0x55, 0x5d, 0xfd, 0x4f, 0x1e, 0xde, 0x57,\n\t\t0xb3, 0x8d, 0xb6, 0xfc, 0x24, 0x16, 0x34, 0xea, 0x53, 0x9f, 0xba, 0x32, 0x13, 0x5d, 0x48, 0x7d,\n\t\t0xd8, 0x8a, 0x45, 0x44, 0x04, 0x1d, 0xcd, 0x74, 0x3b, 0x79, 0xb9, 0x32, 0xfc, 0xea, 0x20, 0x7d,\n\t\t0x4d, 0x3d, 0xce, 0x55, 0x0c, 0x3c, 0x0f, 0x84, 0xfe, 0x62, 0xc0, 0x87, 0x44, 0x11, 0x1c, 0x37,\n\t\t0x65, 0x38, 0xb1, 0x60, 0xee, 0x78, 0xe6, 0x44, 0x74, 0x24, 0x2f, 0x4c, 0xe7, 0x93, 0xf6, 0xc2,\n\t\t0x9f, 0x7e, 0x8f, 0x0d, 0x15, 0x1b, 0x2b, 0x72, 0x9a, 0x99, 0xdc, 0xf1, 0xfc, 0x3d, 0xfc, 0x8c,\n\t\t0x7c, 0x37, 0x0c, 0xfd, 0xcd, 0x80, 0x8f, 0x6e, 0x49, 0xa1, 0xd7, 0x82, 0x46, 0x01, 0xf1, 0x1d,\n\t\t0x1a, 0x08, 0x26, 0x66, 0x99, 0x98, 0xb4, 0x8e, 0x7f, 0xbe, 0x5e, 0x8c, 0xa9, 0xf9, 0xa6, 0xa2,\n\t\t0x2f, 0xc9, 0x79, 0x4e, 0xd6, 0x01, 0x11, 0x86, 0xdd, 0x4c, 0x08, 0xc9, 0x06, 0x8d, 0xbe, 0xd8,\n\t\t0xd5, 0xe3, 0x5e, 0x07, 0x9b, 0x4f, 0x25, 0x5c, 0x76, 0x6f, 0x59, 0x8e, 0x77, 0x61, 0x27, 0x3b,\n\t\t0x7b, 0x9d, 0x4d, 0xf5, 0x17, 0x50, 0xbe, 0x4d, 0x44, 0xf7, 0xa1, 0x10, 0xbb, 0x3c, 0xcc, 0xea,\n\t\t0x34, 0x5d, 0xcc, 0x8b, 0x37, 0xb7, 0xf0, 0x6f, 0xe7, 0x15, 0x3c, 0x5b, 0x73, 0xfe, 0xe8, 0x43,\n\t\t0xb8, 0xbb, 0x74, 0xa7, 0x3a, 0x68, 0x29, 0x5e, 0x80, 0x7e, 0x9e, 0xab, 0x18, 0xd5, 0xbf, 0x1a,\n\t\t0xf0, 0x7c, 0xed, 0xf9, 0xa1, 0x9f, 0xc0, 0xfd, 0xdb, 0xf7, 0x32, 0x1f, 0x71, 0xdb, 0xb2, 0x1f,\n\t\t0x2d, 0x72, 0x54, 0x71, 0xd4, 0x64, 0x6f, 0x5b, 0x66, 0xc8, 0x99, 0x9b, 0xa6, 0xb1, 0xbb, 0x4c,\n\t\t0x78, 0x4d, 0x67, 0x4a, 0xcb, 0x57, 0xb0, 0xdb, 0x23, 0x23, 0x16, 0xa8, 0x5a, 0xee, 0x86, 0x42,\n\t\t0x4d, 0xa3, 0x27, 0xb0, 0x1d, 0x92, 0x11, 0x75, 0x62, 0xf6, 0x6d, 0xba, 0x5f, 0x01, 0x6f, 0x49,\n\t\t0x43, 0x9f, 0x7d, 0x4b, 0xd1, 0x8f, 0x60, 0x27, 0xa0, 0xd7, 0xc2, 0x51, 0x08, 0xc1, 0xc7, 0x34,\n\t\t0xd0, 0x63, 0xf3, 0xae, 0x34, 0xf7, 0xc8, 0x88, 0xda, 0xd2, 0xf8, 0xe2, 0x2d, 0x94, 0x16, 0x27,\n\t\t0x2e, 0x7a, 0x0c, 0x0f, 0xcc, 0x4e, 0xab, 0x7b, 0x62, 0x75, 0xce, 0x1c, 0xfb, 0xaa, 0x67, 0x3a,\n\t\t0x56, 0xe7, 0xa2, 0xd9, 0xb6, 0x4e, 0xca, 0xef, 0xa1, 0x7d, 0x78, 0xb8, 0xec, 0xb2, 0xcf, 0xb1,\n\t\t0x75, 0x6a, 0xe3, 0xcb, 0xb2, 0x81, 0x1e, 0x02, 0x5a, 0xf6, 0xbd, 0xea, 0x77, 0x3b, 0xe5, 0x1c,\n\t\t0xaa, 0xc0, 0xfd, 0x65, 0x7b, 0x0f, 0x77, 0xed, 0xee, 0xcb, 0x72, 0xfe, 0xc5, 0x9f, 0x60, 0x6f,\n\t\t0x45, 0x17, 0x45, 0xcf, 0xe1, 0xa9, 0xd5, 0xef, 0xb6, 0x9b, 0xb6, 0xd5, 0xed, 0x38, 0x67, 0xb8,\n\t\t0xfb, 0xa6, 0xe7, 0xf4, 0xed, 0xa6, 0xbd, 0xa8, 0xe3, 0x9d, 0x90, 0x73, 0xb3, 0xd9, 0xb6, 0xcf,\n\t\t0xaf, 0xca, 0xc6, 0xbb, 0x21, 0x27, 0xb8, 0x69, 0x75, 0xcc, 0x93, 0x72, 0xee, 0xc5, 0x3f, 0x0d,\n\t\t0xf8, 0xe0, 0xbb, 0x9b, 0x03, 0xfa, 0x14, 0x3e, 0x69, 0xb6, 0x6c, 0xeb, 0xc2, 0x74, 0x5a, 0xed,\n\t\t0x37, 0x7d, 0xdb, 0xc4, 0x4e, 0xdf, 0x6c, 0x9b, 0x2d, 0x15, 0xb4, 0x6f, 0xe3, 0xa6, 0x6d, 0x9e,\n\t\t0x5d, 0x2d, 0xe8, 0x7a, 0x09, 0xf5, 0xf5, 0x70, 0x6c, 0x9e, 0xa5, 0x6b, 0xab, 0xf5, 0x5a, 0x2a,\n\t\t0xfd, 0x19, 0x1c, 0xad, 0x27, 0x99, 0x5f, 0xd9, 0x26, 0xee, 0x34, 0xdb, 0x8e, 0xd9, 0xb1, 0x2d,\n\t\t0xfb, 0xaa, 0x9c, 0xdb, 0xcf, 0x55, 0x8c, 0x17, 0x5f, 0x43, 0x49, 0xfe, 0x77, 0xe7, 0x53, 0x1a,\n\t\t0x65, 0x57, 0x77, 0xda, 0xb4, 0xda, 0xdd, 0x0b, 0x13, 0xdf, 0xbe, 0xba, 0x47, 0xb0, 0xb7, 0xec,\n\t\t0x3a, 0xed, 0xe2, 0x96, 0x59, 0x36, 0xe4, 0x9d, 0x2e, 0x3b, 0xce, 0x70, 0xb3, 0x65, 0x9e, 0xbe,\n\t\t0x69, 0x97, 0x73, 0xc7, 0xbf, 0x87, 0x47, 0x2e, 0x9f, 0xac, 0xaa, 0xed, 0xe3, 0x62, 0x4b, 0x7d,\n\t\t0x2d, 0xf5, 0xe4, 0x00, 0xee, 0x19, 0xbf, 0x3b, 0x1a, 0x31, 0xf1, 0x4d, 0x32, 0xa8, 0xb9, 0x7c,\n\t\t0x52, 0x5f, 0xfc, 0xb6, 0xfa, 0x94, 0x79, 0x7e, 0x7d, 0xc4, 0xd3, 0x2f, 0x26, 0xfd, 0xa1, 0xf5,\n\t\t0x05, 0x09, 0xd9, 0xf4, 0x68, 0xb0, 0xa1, 0x6c, 0x2f, 0xff, 0x1b, 0x00, 0x00, 0xff, 0xff, 0x5e,\n\t\t0xe2, 0xc9, 0x06, 0x8c, 0x0d, 0x00, 0x00,\n\t},\n\t// google/protobuf/duration.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x4f, 0x29, 0x2d, 0x4a,\n\t\t0x2c, 0xc9, 0xcc, 0xcf, 0xd3, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0x56,\n\t\t0x5c, 0x1c, 0x2e, 0x50, 0x25, 0x42, 0x12, 0x5c, 0xec, 0xc5, 0xa9, 0xc9, 0xf9, 0x79, 0x29, 0xc5,\n\t\t0x12, 0x8c, 0x0a, 0x8c, 0x1a, 0xcc, 0x41, 0x30, 0xae, 0x90, 0x08, 0x17, 0x6b, 0x5e, 0x62, 0x5e,\n\t\t0x7e, 0xb1, 0x04, 0x93, 0x02, 0xa3, 0x06, 0x6b, 0x10, 0x84, 0xe3, 0xd4, 0xcc, 0xc8, 0x25, 0x9c,\n\t\t0x9c, 0x9f, 0xab, 0x87, 0x66, 0xa6, 0x13, 0x2f, 0xcc, 0xc4, 0x00, 0x90, 0x48, 0x00, 0x63, 0x94,\n\t\t0x21, 0x54, 0x45, 0x7a, 0x7e, 0x4e, 0x62, 0x5e, 0xba, 0x5e, 0x7e, 0x51, 0x3a, 0xc2, 0x81, 0x25,\n\t\t0x95, 0x05, 0xa9, 0xc5, 0xfa, 0xd9, 0x79, 0xf9, 0xe5, 0x79, 0x70, 0xc7, 0x16, 0x24, 0xfd, 0x60,\n\t\t0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce, 0x1d, 0xa2, 0x39, 0x00, 0xaa,\n\t\t0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4, 0x35, 0x89, 0x0d, 0x6c, 0x94,\n\t\t0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0xef, 0x8a, 0xb4, 0xc3, 0xfb, 0x00, 0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gen/proto/matching/v1/service.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/matching/v1/service.proto\n\npackage matchingv1\n\nimport (\n\tencoding_binary \"encoding/binary\"\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n\ttypes \"github.com/gogo/protobuf/types\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\tv11 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype TaskListPartition struct {\n\tIsolationGroups      []string `protobuf:\"bytes,1,rep,name=isolation_groups,json=isolationGroups,proto3\" json:\"isolation_groups,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *TaskListPartition) Reset()         { *m = TaskListPartition{} }\nfunc (m *TaskListPartition) String() string { return proto.CompactTextString(m) }\nfunc (*TaskListPartition) ProtoMessage()    {}\nfunc (*TaskListPartition) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{0}\n}\nfunc (m *TaskListPartition) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TaskListPartition) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TaskListPartition.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TaskListPartition) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TaskListPartition.Merge(m, src)\n}\nfunc (m *TaskListPartition) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TaskListPartition) XXX_DiscardUnknown() {\n\txxx_messageInfo_TaskListPartition.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TaskListPartition proto.InternalMessageInfo\n\nfunc (m *TaskListPartition) GetIsolationGroups() []string {\n\tif m != nil {\n\t\treturn m.IsolationGroups\n\t}\n\treturn nil\n}\n\ntype TaskListPartitionConfig struct {\n\tVersion              int64                        `protobuf:\"varint,1,opt,name=version,proto3\" json:\"version,omitempty\"`\n\tNumReadPartitions    int32                        `protobuf:\"varint,2,opt,name=num_read_partitions,json=numReadPartitions,proto3\" json:\"num_read_partitions,omitempty\"`    // Deprecated: Do not use.\n\tNumWritePartitions   int32                        `protobuf:\"varint,3,opt,name=num_write_partitions,json=numWritePartitions,proto3\" json:\"num_write_partitions,omitempty\"` // Deprecated: Do not use.\n\tReadPartitions       map[int32]*TaskListPartition `protobuf:\"bytes,4,rep,name=read_partitions,json=readPartitions,proto3\" json:\"read_partitions,omitempty\" protobuf_key:\"varint,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tWritePartitions      map[int32]*TaskListPartition `protobuf:\"bytes,5,rep,name=write_partitions,json=writePartitions,proto3\" json:\"write_partitions,omitempty\" protobuf_key:\"varint,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}                     `json:\"-\"`\n\tXXX_unrecognized     []byte                       `json:\"-\"`\n\tXXX_sizecache        int32                        `json:\"-\"`\n}\n\nfunc (m *TaskListPartitionConfig) Reset()         { *m = TaskListPartitionConfig{} }\nfunc (m *TaskListPartitionConfig) String() string { return proto.CompactTextString(m) }\nfunc (*TaskListPartitionConfig) ProtoMessage()    {}\nfunc (*TaskListPartitionConfig) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{1}\n}\nfunc (m *TaskListPartitionConfig) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TaskListPartitionConfig) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TaskListPartitionConfig.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TaskListPartitionConfig) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TaskListPartitionConfig.Merge(m, src)\n}\nfunc (m *TaskListPartitionConfig) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TaskListPartitionConfig) XXX_DiscardUnknown() {\n\txxx_messageInfo_TaskListPartitionConfig.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TaskListPartitionConfig proto.InternalMessageInfo\n\nfunc (m *TaskListPartitionConfig) GetVersion() int64 {\n\tif m != nil {\n\t\treturn m.Version\n\t}\n\treturn 0\n}\n\n// Deprecated: Do not use.\nfunc (m *TaskListPartitionConfig) GetNumReadPartitions() int32 {\n\tif m != nil {\n\t\treturn m.NumReadPartitions\n\t}\n\treturn 0\n}\n\n// Deprecated: Do not use.\nfunc (m *TaskListPartitionConfig) GetNumWritePartitions() int32 {\n\tif m != nil {\n\t\treturn m.NumWritePartitions\n\t}\n\treturn 0\n}\n\nfunc (m *TaskListPartitionConfig) GetReadPartitions() map[int32]*TaskListPartition {\n\tif m != nil {\n\t\treturn m.ReadPartitions\n\t}\n\treturn nil\n}\n\nfunc (m *TaskListPartitionConfig) GetWritePartitions() map[int32]*TaskListPartition {\n\tif m != nil {\n\t\treturn m.WritePartitions\n\t}\n\treturn nil\n}\n\ntype LoadBalancerHints struct {\n\tBacklogCount         int64    `protobuf:\"varint,1,opt,name=backlog_count,json=backlogCount,proto3\" json:\"backlog_count,omitempty\"`\n\tRatePerSecond        float64  `protobuf:\"fixed64,2,opt,name=rate_per_second,json=ratePerSecond,proto3\" json:\"rate_per_second,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *LoadBalancerHints) Reset()         { *m = LoadBalancerHints{} }\nfunc (m *LoadBalancerHints) String() string { return proto.CompactTextString(m) }\nfunc (*LoadBalancerHints) ProtoMessage()    {}\nfunc (*LoadBalancerHints) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{2}\n}\nfunc (m *LoadBalancerHints) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *LoadBalancerHints) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_LoadBalancerHints.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *LoadBalancerHints) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_LoadBalancerHints.Merge(m, src)\n}\nfunc (m *LoadBalancerHints) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *LoadBalancerHints) XXX_DiscardUnknown() {\n\txxx_messageInfo_LoadBalancerHints.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_LoadBalancerHints proto.InternalMessageInfo\n\nfunc (m *LoadBalancerHints) GetBacklogCount() int64 {\n\tif m != nil {\n\t\treturn m.BacklogCount\n\t}\n\treturn 0\n}\n\nfunc (m *LoadBalancerHints) GetRatePerSecond() float64 {\n\tif m != nil {\n\t\treturn m.RatePerSecond\n\t}\n\treturn 0\n}\n\ntype PollForDecisionTaskRequest struct {\n\tRequest              *v1.PollForDecisionTaskRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                         `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tPollerId             string                         `protobuf:\"bytes,3,opt,name=poller_id,json=pollerId,proto3\" json:\"poller_id,omitempty\"`\n\tForwardedFrom        string                         `protobuf:\"bytes,4,opt,name=forwarded_from,json=forwardedFrom,proto3\" json:\"forwarded_from,omitempty\"`\n\tIsolationGroup       string                         `protobuf:\"bytes,5,opt,name=isolation_group,json=isolationGroup,proto3\" json:\"isolation_group,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                       `json:\"-\"`\n\tXXX_unrecognized     []byte                         `json:\"-\"`\n\tXXX_sizecache        int32                          `json:\"-\"`\n}\n\nfunc (m *PollForDecisionTaskRequest) Reset()         { *m = PollForDecisionTaskRequest{} }\nfunc (m *PollForDecisionTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*PollForDecisionTaskRequest) ProtoMessage()    {}\nfunc (*PollForDecisionTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{3}\n}\nfunc (m *PollForDecisionTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollForDecisionTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollForDecisionTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollForDecisionTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollForDecisionTaskRequest.Merge(m, src)\n}\nfunc (m *PollForDecisionTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollForDecisionTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollForDecisionTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollForDecisionTaskRequest proto.InternalMessageInfo\n\nfunc (m *PollForDecisionTaskRequest) GetRequest() *v1.PollForDecisionTaskRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForDecisionTaskRequest) GetPollerId() string {\n\tif m != nil {\n\t\treturn m.PollerId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForDecisionTaskRequest) GetForwardedFrom() string {\n\tif m != nil {\n\t\treturn m.ForwardedFrom\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForDecisionTaskRequest) GetIsolationGroup() string {\n\tif m != nil {\n\t\treturn m.IsolationGroup\n\t}\n\treturn \"\"\n}\n\ntype PollForDecisionTaskResponse struct {\n\tTaskToken                 []byte                       `protobuf:\"bytes,1,opt,name=task_token,json=taskToken,proto3\" json:\"task_token,omitempty\"`\n\tWorkflowExecution         *v1.WorkflowExecution        `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tWorkflowType              *v1.WorkflowType             `protobuf:\"bytes,3,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tPreviousStartedEventId    *types.Int64Value            `protobuf:\"bytes,4,opt,name=previous_started_event_id,json=previousStartedEventId,proto3\" json:\"previous_started_event_id,omitempty\"`\n\tStartedEventId            int64                        `protobuf:\"varint,5,opt,name=started_event_id,json=startedEventId,proto3\" json:\"started_event_id,omitempty\"`\n\tAttempt                   int32                        `protobuf:\"varint,6,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tNextEventId               int64                        `protobuf:\"varint,7,opt,name=next_event_id,json=nextEventId,proto3\" json:\"next_event_id,omitempty\"`\n\tBacklogCountHint          int64                        `protobuf:\"varint,8,opt,name=backlog_count_hint,json=backlogCountHint,proto3\" json:\"backlog_count_hint,omitempty\"`\n\tStickyExecutionEnabled    bool                         `protobuf:\"varint,9,opt,name=sticky_execution_enabled,json=stickyExecutionEnabled,proto3\" json:\"sticky_execution_enabled,omitempty\"`\n\tQuery                     *v1.WorkflowQuery            `protobuf:\"bytes,10,opt,name=query,proto3\" json:\"query,omitempty\"`\n\tDecisionInfo              *v11.TransientDecisionInfo   `protobuf:\"bytes,11,opt,name=decision_info,json=decisionInfo,proto3\" json:\"decision_info,omitempty\"`\n\tWorkflowExecutionTaskList *v1.TaskList                 `protobuf:\"bytes,12,opt,name=workflow_execution_task_list,json=workflowExecutionTaskList,proto3\" json:\"workflow_execution_task_list,omitempty\"`\n\tEventStoreVersion         int32                        `protobuf:\"varint,13,opt,name=event_store_version,json=eventStoreVersion,proto3\" json:\"event_store_version,omitempty\"`\n\tBranchToken               []byte                       `protobuf:\"bytes,14,opt,name=branch_token,json=branchToken,proto3\" json:\"branch_token,omitempty\"`\n\tScheduledTime             *types.Timestamp             `protobuf:\"bytes,15,opt,name=scheduled_time,json=scheduledTime,proto3\" json:\"scheduled_time,omitempty\"`\n\tStartedTime               *types.Timestamp             `protobuf:\"bytes,16,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tQueries                   map[string]*v1.WorkflowQuery `protobuf:\"bytes,17,rep,name=queries,proto3\" json:\"queries,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tTotalHistoryBytes         int64                        `protobuf:\"varint,18,opt,name=total_history_bytes,json=totalHistoryBytes,proto3\" json:\"total_history_bytes,omitempty\"`\n\tPartitionConfig           *TaskListPartitionConfig     `protobuf:\"bytes,19,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tLoadBalancerHints         *LoadBalancerHints           `protobuf:\"bytes,20,opt,name=load_balancer_hints,json=loadBalancerHints,proto3\" json:\"load_balancer_hints,omitempty\"`\n\tAutoConfigHint            *v1.AutoConfigHint           `protobuf:\"bytes,21,opt,name=auto_config_hint,json=autoConfigHint,proto3\" json:\"auto_config_hint,omitempty\"`\n\tXXX_NoUnkeyedLiteral      struct{}                     `json:\"-\"`\n\tXXX_unrecognized          []byte                       `json:\"-\"`\n\tXXX_sizecache             int32                        `json:\"-\"`\n}\n\nfunc (m *PollForDecisionTaskResponse) Reset()         { *m = PollForDecisionTaskResponse{} }\nfunc (m *PollForDecisionTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*PollForDecisionTaskResponse) ProtoMessage()    {}\nfunc (*PollForDecisionTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{4}\n}\nfunc (m *PollForDecisionTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollForDecisionTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollForDecisionTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollForDecisionTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollForDecisionTaskResponse.Merge(m, src)\n}\nfunc (m *PollForDecisionTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollForDecisionTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollForDecisionTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollForDecisionTaskResponse proto.InternalMessageInfo\n\nfunc (m *PollForDecisionTaskResponse) GetTaskToken() []byte {\n\tif m != nil {\n\t\treturn m.TaskToken\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetPreviousStartedEventId() *types.Int64Value {\n\tif m != nil {\n\t\treturn m.PreviousStartedEventId\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetStartedEventId() int64 {\n\tif m != nil {\n\t\treturn m.StartedEventId\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetNextEventId() int64 {\n\tif m != nil {\n\t\treturn m.NextEventId\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetBacklogCountHint() int64 {\n\tif m != nil {\n\t\treturn m.BacklogCountHint\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetStickyExecutionEnabled() bool {\n\tif m != nil {\n\t\treturn m.StickyExecutionEnabled\n\t}\n\treturn false\n}\n\nfunc (m *PollForDecisionTaskResponse) GetQuery() *v1.WorkflowQuery {\n\tif m != nil {\n\t\treturn m.Query\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetDecisionInfo() *v11.TransientDecisionInfo {\n\tif m != nil {\n\t\treturn m.DecisionInfo\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetWorkflowExecutionTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.WorkflowExecutionTaskList\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetEventStoreVersion() int32 {\n\tif m != nil {\n\t\treturn m.EventStoreVersion\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.BranchToken\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetScheduledTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTime\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetQueries() map[string]*v1.WorkflowQuery {\n\tif m != nil {\n\t\treturn m.Queries\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetTotalHistoryBytes() int64 {\n\tif m != nil {\n\t\treturn m.TotalHistoryBytes\n\t}\n\treturn 0\n}\n\nfunc (m *PollForDecisionTaskResponse) GetPartitionConfig() *TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetLoadBalancerHints() *LoadBalancerHints {\n\tif m != nil {\n\t\treturn m.LoadBalancerHints\n\t}\n\treturn nil\n}\n\nfunc (m *PollForDecisionTaskResponse) GetAutoConfigHint() *v1.AutoConfigHint {\n\tif m != nil {\n\t\treturn m.AutoConfigHint\n\t}\n\treturn nil\n}\n\ntype PollForActivityTaskRequest struct {\n\tRequest              *v1.PollForActivityTaskRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                         `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tPollerId             string                         `protobuf:\"bytes,3,opt,name=poller_id,json=pollerId,proto3\" json:\"poller_id,omitempty\"`\n\tForwardedFrom        string                         `protobuf:\"bytes,4,opt,name=forwarded_from,json=forwardedFrom,proto3\" json:\"forwarded_from,omitempty\"`\n\tIsolationGroup       string                         `protobuf:\"bytes,5,opt,name=isolation_group,json=isolationGroup,proto3\" json:\"isolation_group,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                       `json:\"-\"`\n\tXXX_unrecognized     []byte                         `json:\"-\"`\n\tXXX_sizecache        int32                          `json:\"-\"`\n}\n\nfunc (m *PollForActivityTaskRequest) Reset()         { *m = PollForActivityTaskRequest{} }\nfunc (m *PollForActivityTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*PollForActivityTaskRequest) ProtoMessage()    {}\nfunc (*PollForActivityTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{5}\n}\nfunc (m *PollForActivityTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollForActivityTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollForActivityTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollForActivityTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollForActivityTaskRequest.Merge(m, src)\n}\nfunc (m *PollForActivityTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollForActivityTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollForActivityTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollForActivityTaskRequest proto.InternalMessageInfo\n\nfunc (m *PollForActivityTaskRequest) GetRequest() *v1.PollForActivityTaskRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForActivityTaskRequest) GetPollerId() string {\n\tif m != nil {\n\t\treturn m.PollerId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForActivityTaskRequest) GetForwardedFrom() string {\n\tif m != nil {\n\t\treturn m.ForwardedFrom\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForActivityTaskRequest) GetIsolationGroup() string {\n\tif m != nil {\n\t\treturn m.IsolationGroup\n\t}\n\treturn \"\"\n}\n\ntype PollForActivityTaskResponse struct {\n\tTaskToken                  []byte                   `protobuf:\"bytes,1,opt,name=task_token,json=taskToken,proto3\" json:\"task_token,omitempty\"`\n\tWorkflowExecution          *v1.WorkflowExecution    `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tActivityId                 string                   `protobuf:\"bytes,3,opt,name=activity_id,json=activityId,proto3\" json:\"activity_id,omitempty\"`\n\tActivityType               *v1.ActivityType         `protobuf:\"bytes,4,opt,name=activity_type,json=activityType,proto3\" json:\"activity_type,omitempty\"`\n\tInput                      *v1.Payload              `protobuf:\"bytes,5,opt,name=input,proto3\" json:\"input,omitempty\"`\n\tScheduledTime              *types.Timestamp         `protobuf:\"bytes,6,opt,name=scheduled_time,json=scheduledTime,proto3\" json:\"scheduled_time,omitempty\"`\n\tStartedTime                *types.Timestamp         `protobuf:\"bytes,7,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tScheduleToCloseTimeout     *types.Duration          `protobuf:\"bytes,8,opt,name=schedule_to_close_timeout,json=scheduleToCloseTimeout,proto3\" json:\"schedule_to_close_timeout,omitempty\"`\n\tStartToCloseTimeout        *types.Duration          `protobuf:\"bytes,9,opt,name=start_to_close_timeout,json=startToCloseTimeout,proto3\" json:\"start_to_close_timeout,omitempty\"`\n\tHeartbeatTimeout           *types.Duration          `protobuf:\"bytes,10,opt,name=heartbeat_timeout,json=heartbeatTimeout,proto3\" json:\"heartbeat_timeout,omitempty\"`\n\tAttempt                    int32                    `protobuf:\"varint,11,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tScheduledTimeOfThisAttempt *types.Timestamp         `protobuf:\"bytes,12,opt,name=scheduled_time_of_this_attempt,json=scheduledTimeOfThisAttempt,proto3\" json:\"scheduled_time_of_this_attempt,omitempty\"`\n\tHeartbeatDetails           *v1.Payload              `protobuf:\"bytes,13,opt,name=heartbeat_details,json=heartbeatDetails,proto3\" json:\"heartbeat_details,omitempty\"`\n\tWorkflowType               *v1.WorkflowType         `protobuf:\"bytes,14,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tWorkflowDomain             string                   `protobuf:\"bytes,15,opt,name=workflow_domain,json=workflowDomain,proto3\" json:\"workflow_domain,omitempty\"`\n\tHeader                     *v1.Header               `protobuf:\"bytes,16,opt,name=header,proto3\" json:\"header,omitempty\"`\n\tLoadBalancerHints          *LoadBalancerHints       `protobuf:\"bytes,17,opt,name=load_balancer_hints,json=loadBalancerHints,proto3\" json:\"load_balancer_hints,omitempty\"`\n\tPartitionConfig            *TaskListPartitionConfig `protobuf:\"bytes,19,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tAutoConfigHint             *v1.AutoConfigHint       `protobuf:\"bytes,20,opt,name=auto_config_hint,json=autoConfigHint,proto3\" json:\"auto_config_hint,omitempty\"`\n\tXXX_NoUnkeyedLiteral       struct{}                 `json:\"-\"`\n\tXXX_unrecognized           []byte                   `json:\"-\"`\n\tXXX_sizecache              int32                    `json:\"-\"`\n}\n\nfunc (m *PollForActivityTaskResponse) Reset()         { *m = PollForActivityTaskResponse{} }\nfunc (m *PollForActivityTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*PollForActivityTaskResponse) ProtoMessage()    {}\nfunc (*PollForActivityTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{6}\n}\nfunc (m *PollForActivityTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PollForActivityTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PollForActivityTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PollForActivityTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PollForActivityTaskResponse.Merge(m, src)\n}\nfunc (m *PollForActivityTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PollForActivityTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_PollForActivityTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PollForActivityTaskResponse proto.InternalMessageInfo\n\nfunc (m *PollForActivityTaskResponse) GetTaskToken() []byte {\n\tif m != nil {\n\t\treturn m.TaskToken\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetActivityId() string {\n\tif m != nil {\n\t\treturn m.ActivityId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForActivityTaskResponse) GetActivityType() *v1.ActivityType {\n\tif m != nil {\n\t\treturn m.ActivityType\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetInput() *v1.Payload {\n\tif m != nil {\n\t\treturn m.Input\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetScheduledTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTime\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetScheduleToCloseTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.ScheduleToCloseTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetStartToCloseTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.StartToCloseTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetHeartbeatTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.HeartbeatTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *PollForActivityTaskResponse) GetScheduledTimeOfThisAttempt() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTimeOfThisAttempt\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetHeartbeatDetails() *v1.Payload {\n\tif m != nil {\n\t\treturn m.HeartbeatDetails\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetWorkflowDomain() string {\n\tif m != nil {\n\t\treturn m.WorkflowDomain\n\t}\n\treturn \"\"\n}\n\nfunc (m *PollForActivityTaskResponse) GetHeader() *v1.Header {\n\tif m != nil {\n\t\treturn m.Header\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetLoadBalancerHints() *LoadBalancerHints {\n\tif m != nil {\n\t\treturn m.LoadBalancerHints\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetPartitionConfig() *TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\nfunc (m *PollForActivityTaskResponse) GetAutoConfigHint() *v1.AutoConfigHint {\n\tif m != nil {\n\t\treturn m.AutoConfigHint\n\t}\n\treturn nil\n}\n\ntype AddDecisionTaskRequest struct {\n\tDomainId               string                `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution      *v1.WorkflowExecution `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tTaskList               *v1.TaskList          `protobuf:\"bytes,3,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tScheduleId             int64                 `protobuf:\"varint,4,opt,name=schedule_id,json=scheduleId,proto3\" json:\"schedule_id,omitempty\"`\n\tScheduleToStartTimeout *types.Duration       `protobuf:\"bytes,5,opt,name=schedule_to_start_timeout,json=scheduleToStartTimeout,proto3\" json:\"schedule_to_start_timeout,omitempty\"`\n\tSource                 v11.TaskSource        `protobuf:\"varint,6,opt,name=source,proto3,enum=uber.cadence.shared.v1.TaskSource\" json:\"source,omitempty\"`\n\tForwardedFrom          string                `protobuf:\"bytes,7,opt,name=forwarded_from,json=forwardedFrom,proto3\" json:\"forwarded_from,omitempty\"`\n\tPartitionConfig        map[string]string     `protobuf:\"bytes,8,rep,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral   struct{}              `json:\"-\"`\n\tXXX_unrecognized       []byte                `json:\"-\"`\n\tXXX_sizecache          int32                 `json:\"-\"`\n}\n\nfunc (m *AddDecisionTaskRequest) Reset()         { *m = AddDecisionTaskRequest{} }\nfunc (m *AddDecisionTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*AddDecisionTaskRequest) ProtoMessage()    {}\nfunc (*AddDecisionTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{7}\n}\nfunc (m *AddDecisionTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *AddDecisionTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_AddDecisionTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *AddDecisionTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_AddDecisionTaskRequest.Merge(m, src)\n}\nfunc (m *AddDecisionTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *AddDecisionTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_AddDecisionTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_AddDecisionTaskRequest proto.InternalMessageInfo\n\nfunc (m *AddDecisionTaskRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *AddDecisionTaskRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *AddDecisionTaskRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *AddDecisionTaskRequest) GetScheduleId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduleId\n\t}\n\treturn 0\n}\n\nfunc (m *AddDecisionTaskRequest) GetScheduleToStartTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.ScheduleToStartTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *AddDecisionTaskRequest) GetSource() v11.TaskSource {\n\tif m != nil {\n\t\treturn m.Source\n\t}\n\treturn v11.TaskSource_TASK_SOURCE_INVALID\n}\n\nfunc (m *AddDecisionTaskRequest) GetForwardedFrom() string {\n\tif m != nil {\n\t\treturn m.ForwardedFrom\n\t}\n\treturn \"\"\n}\n\nfunc (m *AddDecisionTaskRequest) GetPartitionConfig() map[string]string {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype AddDecisionTaskResponse struct {\n\tPartitionConfig      *TaskListPartitionConfig `protobuf:\"bytes,1,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                 `json:\"-\"`\n\tXXX_unrecognized     []byte                   `json:\"-\"`\n\tXXX_sizecache        int32                    `json:\"-\"`\n}\n\nfunc (m *AddDecisionTaskResponse) Reset()         { *m = AddDecisionTaskResponse{} }\nfunc (m *AddDecisionTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*AddDecisionTaskResponse) ProtoMessage()    {}\nfunc (*AddDecisionTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{8}\n}\nfunc (m *AddDecisionTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *AddDecisionTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_AddDecisionTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *AddDecisionTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_AddDecisionTaskResponse.Merge(m, src)\n}\nfunc (m *AddDecisionTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *AddDecisionTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_AddDecisionTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_AddDecisionTaskResponse proto.InternalMessageInfo\n\nfunc (m *AddDecisionTaskResponse) GetPartitionConfig() *TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype AddActivityTaskRequest struct {\n\tDomainId                 string                    `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution        *v1.WorkflowExecution     `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tSourceDomainId           string                    `protobuf:\"bytes,3,opt,name=source_domain_id,json=sourceDomainId,proto3\" json:\"source_domain_id,omitempty\"`\n\tTaskList                 *v1.TaskList              `protobuf:\"bytes,4,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tScheduleId               int64                     `protobuf:\"varint,5,opt,name=schedule_id,json=scheduleId,proto3\" json:\"schedule_id,omitempty\"`\n\tScheduleToStartTimeout   *types.Duration           `protobuf:\"bytes,6,opt,name=schedule_to_start_timeout,json=scheduleToStartTimeout,proto3\" json:\"schedule_to_start_timeout,omitempty\"`\n\tSource                   v11.TaskSource            `protobuf:\"varint,7,opt,name=source,proto3,enum=uber.cadence.shared.v1.TaskSource\" json:\"source,omitempty\"`\n\tForwardedFrom            string                    `protobuf:\"bytes,8,opt,name=forwarded_from,json=forwardedFrom,proto3\" json:\"forwarded_from,omitempty\"`\n\tActivityTaskDispatchInfo *ActivityTaskDispatchInfo `protobuf:\"bytes,9,opt,name=activityTaskDispatchInfo,proto3\" json:\"activityTaskDispatchInfo,omitempty\"`\n\tPartitionConfig          map[string]string         `protobuf:\"bytes,10,rep,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral     struct{}                  `json:\"-\"`\n\tXXX_unrecognized         []byte                    `json:\"-\"`\n\tXXX_sizecache            int32                     `json:\"-\"`\n}\n\nfunc (m *AddActivityTaskRequest) Reset()         { *m = AddActivityTaskRequest{} }\nfunc (m *AddActivityTaskRequest) String() string { return proto.CompactTextString(m) }\nfunc (*AddActivityTaskRequest) ProtoMessage()    {}\nfunc (*AddActivityTaskRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{9}\n}\nfunc (m *AddActivityTaskRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *AddActivityTaskRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_AddActivityTaskRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *AddActivityTaskRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_AddActivityTaskRequest.Merge(m, src)\n}\nfunc (m *AddActivityTaskRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *AddActivityTaskRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_AddActivityTaskRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_AddActivityTaskRequest proto.InternalMessageInfo\n\nfunc (m *AddActivityTaskRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *AddActivityTaskRequest) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *AddActivityTaskRequest) GetSourceDomainId() string {\n\tif m != nil {\n\t\treturn m.SourceDomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *AddActivityTaskRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *AddActivityTaskRequest) GetScheduleId() int64 {\n\tif m != nil {\n\t\treturn m.ScheduleId\n\t}\n\treturn 0\n}\n\nfunc (m *AddActivityTaskRequest) GetScheduleToStartTimeout() *types.Duration {\n\tif m != nil {\n\t\treturn m.ScheduleToStartTimeout\n\t}\n\treturn nil\n}\n\nfunc (m *AddActivityTaskRequest) GetSource() v11.TaskSource {\n\tif m != nil {\n\t\treturn m.Source\n\t}\n\treturn v11.TaskSource_TASK_SOURCE_INVALID\n}\n\nfunc (m *AddActivityTaskRequest) GetForwardedFrom() string {\n\tif m != nil {\n\t\treturn m.ForwardedFrom\n\t}\n\treturn \"\"\n}\n\nfunc (m *AddActivityTaskRequest) GetActivityTaskDispatchInfo() *ActivityTaskDispatchInfo {\n\tif m != nil {\n\t\treturn m.ActivityTaskDispatchInfo\n\t}\n\treturn nil\n}\n\nfunc (m *AddActivityTaskRequest) GetPartitionConfig() map[string]string {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype ActivityTaskDispatchInfo struct {\n\tScheduledEvent             *v1.HistoryEvent `protobuf:\"bytes,1,opt,name=scheduled_event,json=scheduledEvent,proto3\" json:\"scheduled_event,omitempty\"`\n\tStartedTime                *types.Timestamp `protobuf:\"bytes,2,opt,name=started_time,json=startedTime,proto3\" json:\"started_time,omitempty\"`\n\tAttempt                    int32            `protobuf:\"varint,3,opt,name=attempt,proto3\" json:\"attempt,omitempty\"`\n\tScheduledTimeOfThisAttempt *types.Timestamp `protobuf:\"bytes,4,opt,name=scheduled_time_of_this_attempt,json=scheduledTimeOfThisAttempt,proto3\" json:\"scheduled_time_of_this_attempt,omitempty\"`\n\tHeartbeatDetails           *v1.Payload      `protobuf:\"bytes,5,opt,name=heartbeat_details,json=heartbeatDetails,proto3\" json:\"heartbeat_details,omitempty\"`\n\tWorkflowType               *v1.WorkflowType `protobuf:\"bytes,6,opt,name=workflow_type,json=workflowType,proto3\" json:\"workflow_type,omitempty\"`\n\tWorkflowDomain             string           `protobuf:\"bytes,7,opt,name=workflow_domain,json=workflowDomain,proto3\" json:\"workflow_domain,omitempty\"`\n\tXXX_NoUnkeyedLiteral       struct{}         `json:\"-\"`\n\tXXX_unrecognized           []byte           `json:\"-\"`\n\tXXX_sizecache              int32            `json:\"-\"`\n}\n\nfunc (m *ActivityTaskDispatchInfo) Reset()         { *m = ActivityTaskDispatchInfo{} }\nfunc (m *ActivityTaskDispatchInfo) String() string { return proto.CompactTextString(m) }\nfunc (*ActivityTaskDispatchInfo) ProtoMessage()    {}\nfunc (*ActivityTaskDispatchInfo) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{10}\n}\nfunc (m *ActivityTaskDispatchInfo) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ActivityTaskDispatchInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ActivityTaskDispatchInfo.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ActivityTaskDispatchInfo) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ActivityTaskDispatchInfo.Merge(m, src)\n}\nfunc (m *ActivityTaskDispatchInfo) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ActivityTaskDispatchInfo) XXX_DiscardUnknown() {\n\txxx_messageInfo_ActivityTaskDispatchInfo.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ActivityTaskDispatchInfo proto.InternalMessageInfo\n\nfunc (m *ActivityTaskDispatchInfo) GetScheduledEvent() *v1.HistoryEvent {\n\tif m != nil {\n\t\treturn m.ScheduledEvent\n\t}\n\treturn nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetStartedTime() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.StartedTime\n\t}\n\treturn nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetAttempt() int32 {\n\tif m != nil {\n\t\treturn m.Attempt\n\t}\n\treturn 0\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetScheduledTimeOfThisAttempt() *types.Timestamp {\n\tif m != nil {\n\t\treturn m.ScheduledTimeOfThisAttempt\n\t}\n\treturn nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetHeartbeatDetails() *v1.Payload {\n\tif m != nil {\n\t\treturn m.HeartbeatDetails\n\t}\n\treturn nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetWorkflowType() *v1.WorkflowType {\n\tif m != nil {\n\t\treturn m.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) GetWorkflowDomain() string {\n\tif m != nil {\n\t\treturn m.WorkflowDomain\n\t}\n\treturn \"\"\n}\n\ntype AddActivityTaskResponse struct {\n\tPartitionConfig      *TaskListPartitionConfig `protobuf:\"bytes,1,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                 `json:\"-\"`\n\tXXX_unrecognized     []byte                   `json:\"-\"`\n\tXXX_sizecache        int32                    `json:\"-\"`\n}\n\nfunc (m *AddActivityTaskResponse) Reset()         { *m = AddActivityTaskResponse{} }\nfunc (m *AddActivityTaskResponse) String() string { return proto.CompactTextString(m) }\nfunc (*AddActivityTaskResponse) ProtoMessage()    {}\nfunc (*AddActivityTaskResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{11}\n}\nfunc (m *AddActivityTaskResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *AddActivityTaskResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_AddActivityTaskResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *AddActivityTaskResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_AddActivityTaskResponse.Merge(m, src)\n}\nfunc (m *AddActivityTaskResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *AddActivityTaskResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_AddActivityTaskResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_AddActivityTaskResponse proto.InternalMessageInfo\n\nfunc (m *AddActivityTaskResponse) GetPartitionConfig() *TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype QueryWorkflowRequest struct {\n\tRequest              *v1.QueryWorkflowRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                   `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tTaskList             *v1.TaskList             `protobuf:\"bytes,3,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tForwardedFrom        string                   `protobuf:\"bytes,4,opt,name=forwarded_from,json=forwardedFrom,proto3\" json:\"forwarded_from,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                 `json:\"-\"`\n\tXXX_unrecognized     []byte                   `json:\"-\"`\n\tXXX_sizecache        int32                    `json:\"-\"`\n}\n\nfunc (m *QueryWorkflowRequest) Reset()         { *m = QueryWorkflowRequest{} }\nfunc (m *QueryWorkflowRequest) String() string { return proto.CompactTextString(m) }\nfunc (*QueryWorkflowRequest) ProtoMessage()    {}\nfunc (*QueryWorkflowRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{12}\n}\nfunc (m *QueryWorkflowRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *QueryWorkflowRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_QueryWorkflowRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *QueryWorkflowRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_QueryWorkflowRequest.Merge(m, src)\n}\nfunc (m *QueryWorkflowRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *QueryWorkflowRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_QueryWorkflowRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_QueryWorkflowRequest proto.InternalMessageInfo\n\nfunc (m *QueryWorkflowRequest) GetRequest() *v1.QueryWorkflowRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *QueryWorkflowRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowRequest) GetForwardedFrom() string {\n\tif m != nil {\n\t\treturn m.ForwardedFrom\n\t}\n\treturn \"\"\n}\n\ntype QueryWorkflowResponse struct {\n\tQueryResult          *v1.Payload                 `protobuf:\"bytes,1,opt,name=query_result,json=queryResult,proto3\" json:\"query_result,omitempty\"`\n\tQueryRejected        *v1.QueryRejected           `protobuf:\"bytes,2,opt,name=query_rejected,json=queryRejected,proto3\" json:\"query_rejected,omitempty\"`\n\tPartitionConfig      *v1.TaskListPartitionConfig `protobuf:\"bytes,3,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *QueryWorkflowResponse) Reset()         { *m = QueryWorkflowResponse{} }\nfunc (m *QueryWorkflowResponse) String() string { return proto.CompactTextString(m) }\nfunc (*QueryWorkflowResponse) ProtoMessage()    {}\nfunc (*QueryWorkflowResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{13}\n}\nfunc (m *QueryWorkflowResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *QueryWorkflowResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_QueryWorkflowResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *QueryWorkflowResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_QueryWorkflowResponse.Merge(m, src)\n}\nfunc (m *QueryWorkflowResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *QueryWorkflowResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_QueryWorkflowResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_QueryWorkflowResponse proto.InternalMessageInfo\n\nfunc (m *QueryWorkflowResponse) GetQueryResult() *v1.Payload {\n\tif m != nil {\n\t\treturn m.QueryResult\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowResponse) GetQueryRejected() *v1.QueryRejected {\n\tif m != nil {\n\t\treturn m.QueryRejected\n\t}\n\treturn nil\n}\n\nfunc (m *QueryWorkflowResponse) GetPartitionConfig() *v1.TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype RespondQueryTaskCompletedRequest struct {\n\tRequest              *v1.RespondQueryTaskCompletedRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                               `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tTaskList             *v1.TaskList                         `protobuf:\"bytes,3,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tTaskId               string                               `protobuf:\"bytes,4,opt,name=task_id,json=taskId,proto3\" json:\"task_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                             `json:\"-\"`\n\tXXX_unrecognized     []byte                               `json:\"-\"`\n\tXXX_sizecache        int32                                `json:\"-\"`\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) Reset()         { *m = RespondQueryTaskCompletedRequest{} }\nfunc (m *RespondQueryTaskCompletedRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RespondQueryTaskCompletedRequest) ProtoMessage()    {}\nfunc (*RespondQueryTaskCompletedRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{14}\n}\nfunc (m *RespondQueryTaskCompletedRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondQueryTaskCompletedRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondQueryTaskCompletedRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondQueryTaskCompletedRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondQueryTaskCompletedRequest.Merge(m, src)\n}\nfunc (m *RespondQueryTaskCompletedRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondQueryTaskCompletedRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondQueryTaskCompletedRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondQueryTaskCompletedRequest proto.InternalMessageInfo\n\nfunc (m *RespondQueryTaskCompletedRequest) GetRequest() *v1.RespondQueryTaskCompletedRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) GetTaskId() string {\n\tif m != nil {\n\t\treturn m.TaskId\n\t}\n\treturn \"\"\n}\n\ntype RespondQueryTaskCompletedResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RespondQueryTaskCompletedResponse) Reset()         { *m = RespondQueryTaskCompletedResponse{} }\nfunc (m *RespondQueryTaskCompletedResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RespondQueryTaskCompletedResponse) ProtoMessage()    {}\nfunc (*RespondQueryTaskCompletedResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{15}\n}\nfunc (m *RespondQueryTaskCompletedResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RespondQueryTaskCompletedResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RespondQueryTaskCompletedResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RespondQueryTaskCompletedResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RespondQueryTaskCompletedResponse.Merge(m, src)\n}\nfunc (m *RespondQueryTaskCompletedResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RespondQueryTaskCompletedResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RespondQueryTaskCompletedResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RespondQueryTaskCompletedResponse proto.InternalMessageInfo\n\ntype CancelOutstandingPollRequest struct {\n\tDomainId             string          `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tPollerId             string          `protobuf:\"bytes,2,opt,name=poller_id,json=pollerId,proto3\" json:\"poller_id,omitempty\"`\n\tTaskListType         v1.TaskListType `protobuf:\"varint,3,opt,name=task_list_type,json=taskListType,proto3,enum=uber.cadence.api.v1.TaskListType\" json:\"task_list_type,omitempty\"`\n\tTaskList             *v1.TaskList    `protobuf:\"bytes,4,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}        `json:\"-\"`\n\tXXX_unrecognized     []byte          `json:\"-\"`\n\tXXX_sizecache        int32           `json:\"-\"`\n}\n\nfunc (m *CancelOutstandingPollRequest) Reset()         { *m = CancelOutstandingPollRequest{} }\nfunc (m *CancelOutstandingPollRequest) String() string { return proto.CompactTextString(m) }\nfunc (*CancelOutstandingPollRequest) ProtoMessage()    {}\nfunc (*CancelOutstandingPollRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{16}\n}\nfunc (m *CancelOutstandingPollRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CancelOutstandingPollRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CancelOutstandingPollRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CancelOutstandingPollRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CancelOutstandingPollRequest.Merge(m, src)\n}\nfunc (m *CancelOutstandingPollRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CancelOutstandingPollRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_CancelOutstandingPollRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CancelOutstandingPollRequest proto.InternalMessageInfo\n\nfunc (m *CancelOutstandingPollRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *CancelOutstandingPollRequest) GetPollerId() string {\n\tif m != nil {\n\t\treturn m.PollerId\n\t}\n\treturn \"\"\n}\n\nfunc (m *CancelOutstandingPollRequest) GetTaskListType() v1.TaskListType {\n\tif m != nil {\n\t\treturn m.TaskListType\n\t}\n\treturn v1.TaskListType_TASK_LIST_TYPE_INVALID\n}\n\nfunc (m *CancelOutstandingPollRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\ntype CancelOutstandingPollResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *CancelOutstandingPollResponse) Reset()         { *m = CancelOutstandingPollResponse{} }\nfunc (m *CancelOutstandingPollResponse) String() string { return proto.CompactTextString(m) }\nfunc (*CancelOutstandingPollResponse) ProtoMessage()    {}\nfunc (*CancelOutstandingPollResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{17}\n}\nfunc (m *CancelOutstandingPollResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CancelOutstandingPollResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CancelOutstandingPollResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CancelOutstandingPollResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CancelOutstandingPollResponse.Merge(m, src)\n}\nfunc (m *CancelOutstandingPollResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CancelOutstandingPollResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_CancelOutstandingPollResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CancelOutstandingPollResponse proto.InternalMessageInfo\n\ntype DescribeTaskListRequest struct {\n\tRequest              *v1.DescribeTaskListRequest `protobuf:\"bytes,1,opt,name=request,proto3\" json:\"request,omitempty\"`\n\tDomainId             string                      `protobuf:\"bytes,2,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *DescribeTaskListRequest) Reset()         { *m = DescribeTaskListRequest{} }\nfunc (m *DescribeTaskListRequest) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeTaskListRequest) ProtoMessage()    {}\nfunc (*DescribeTaskListRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{18}\n}\nfunc (m *DescribeTaskListRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeTaskListRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeTaskListRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeTaskListRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeTaskListRequest.Merge(m, src)\n}\nfunc (m *DescribeTaskListRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeTaskListRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeTaskListRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeTaskListRequest proto.InternalMessageInfo\n\nfunc (m *DescribeTaskListRequest) GetRequest() *v1.DescribeTaskListRequest {\n\tif m != nil {\n\t\treturn m.Request\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeTaskListRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\ntype DescribeTaskListResponse struct {\n\tPollers              []*v1.PollerInfo            `protobuf:\"bytes,1,rep,name=pollers,proto3\" json:\"pollers,omitempty\"`\n\tTaskListStatus       *v1.TaskListStatus          `protobuf:\"bytes,2,opt,name=task_list_status,json=taskListStatus,proto3\" json:\"task_list_status,omitempty\"`\n\tPartitionConfig      *v1.TaskListPartitionConfig `protobuf:\"bytes,3,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tTaskList             *v1.TaskList                `protobuf:\"bytes,4,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *DescribeTaskListResponse) Reset()         { *m = DescribeTaskListResponse{} }\nfunc (m *DescribeTaskListResponse) String() string { return proto.CompactTextString(m) }\nfunc (*DescribeTaskListResponse) ProtoMessage()    {}\nfunc (*DescribeTaskListResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{19}\n}\nfunc (m *DescribeTaskListResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *DescribeTaskListResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_DescribeTaskListResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *DescribeTaskListResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_DescribeTaskListResponse.Merge(m, src)\n}\nfunc (m *DescribeTaskListResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *DescribeTaskListResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_DescribeTaskListResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_DescribeTaskListResponse proto.InternalMessageInfo\n\nfunc (m *DescribeTaskListResponse) GetPollers() []*v1.PollerInfo {\n\tif m != nil {\n\t\treturn m.Pollers\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeTaskListResponse) GetTaskListStatus() *v1.TaskListStatus {\n\tif m != nil {\n\t\treturn m.TaskListStatus\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeTaskListResponse) GetPartitionConfig() *v1.TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\nfunc (m *DescribeTaskListResponse) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\ntype ListTaskListPartitionsRequest struct {\n\tDomain               string       `protobuf:\"bytes,1,opt,name=domain,proto3\" json:\"domain,omitempty\"`\n\tTaskList             *v1.TaskList `protobuf:\"bytes,2,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}     `json:\"-\"`\n\tXXX_unrecognized     []byte       `json:\"-\"`\n\tXXX_sizecache        int32        `json:\"-\"`\n}\n\nfunc (m *ListTaskListPartitionsRequest) Reset()         { *m = ListTaskListPartitionsRequest{} }\nfunc (m *ListTaskListPartitionsRequest) String() string { return proto.CompactTextString(m) }\nfunc (*ListTaskListPartitionsRequest) ProtoMessage()    {}\nfunc (*ListTaskListPartitionsRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{20}\n}\nfunc (m *ListTaskListPartitionsRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ListTaskListPartitionsRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ListTaskListPartitionsRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ListTaskListPartitionsRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ListTaskListPartitionsRequest.Merge(m, src)\n}\nfunc (m *ListTaskListPartitionsRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ListTaskListPartitionsRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_ListTaskListPartitionsRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ListTaskListPartitionsRequest proto.InternalMessageInfo\n\nfunc (m *ListTaskListPartitionsRequest) GetDomain() string {\n\tif m != nil {\n\t\treturn m.Domain\n\t}\n\treturn \"\"\n}\n\nfunc (m *ListTaskListPartitionsRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\ntype ListTaskListPartitionsResponse struct {\n\tActivityTaskListPartitions []*v1.TaskListPartitionMetadata `protobuf:\"bytes,1,rep,name=activity_task_list_partitions,json=activityTaskListPartitions,proto3\" json:\"activity_task_list_partitions,omitempty\"`\n\tDecisionTaskListPartitions []*v1.TaskListPartitionMetadata `protobuf:\"bytes,2,rep,name=decision_task_list_partitions,json=decisionTaskListPartitions,proto3\" json:\"decision_task_list_partitions,omitempty\"`\n\tXXX_NoUnkeyedLiteral       struct{}                        `json:\"-\"`\n\tXXX_unrecognized           []byte                          `json:\"-\"`\n\tXXX_sizecache              int32                           `json:\"-\"`\n}\n\nfunc (m *ListTaskListPartitionsResponse) Reset()         { *m = ListTaskListPartitionsResponse{} }\nfunc (m *ListTaskListPartitionsResponse) String() string { return proto.CompactTextString(m) }\nfunc (*ListTaskListPartitionsResponse) ProtoMessage()    {}\nfunc (*ListTaskListPartitionsResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{21}\n}\nfunc (m *ListTaskListPartitionsResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ListTaskListPartitionsResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ListTaskListPartitionsResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ListTaskListPartitionsResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ListTaskListPartitionsResponse.Merge(m, src)\n}\nfunc (m *ListTaskListPartitionsResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ListTaskListPartitionsResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_ListTaskListPartitionsResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ListTaskListPartitionsResponse proto.InternalMessageInfo\n\nfunc (m *ListTaskListPartitionsResponse) GetActivityTaskListPartitions() []*v1.TaskListPartitionMetadata {\n\tif m != nil {\n\t\treturn m.ActivityTaskListPartitions\n\t}\n\treturn nil\n}\n\nfunc (m *ListTaskListPartitionsResponse) GetDecisionTaskListPartitions() []*v1.TaskListPartitionMetadata {\n\tif m != nil {\n\t\treturn m.DecisionTaskListPartitions\n\t}\n\treturn nil\n}\n\ntype GetTaskListsByDomainRequest struct {\n\tDomain               string   `protobuf:\"bytes,1,opt,name=domain,proto3\" json:\"domain,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *GetTaskListsByDomainRequest) Reset()         { *m = GetTaskListsByDomainRequest{} }\nfunc (m *GetTaskListsByDomainRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetTaskListsByDomainRequest) ProtoMessage()    {}\nfunc (*GetTaskListsByDomainRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{22}\n}\nfunc (m *GetTaskListsByDomainRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetTaskListsByDomainRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetTaskListsByDomainRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetTaskListsByDomainRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetTaskListsByDomainRequest.Merge(m, src)\n}\nfunc (m *GetTaskListsByDomainRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetTaskListsByDomainRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetTaskListsByDomainRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetTaskListsByDomainRequest proto.InternalMessageInfo\n\nfunc (m *GetTaskListsByDomainRequest) GetDomain() string {\n\tif m != nil {\n\t\treturn m.Domain\n\t}\n\treturn \"\"\n}\n\ntype GetTaskListsByDomainResponse struct {\n\tDecisionTaskListMap  map[string]*DescribeTaskListResponse `protobuf:\"bytes,1,rep,name=decision_task_list_map,json=decisionTaskListMap,proto3\" json:\"decision_task_list_map,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tActivityTaskListMap  map[string]*DescribeTaskListResponse `protobuf:\"bytes,2,rep,name=activity_task_list_map,json=activityTaskListMap,proto3\" json:\"activity_task_list_map,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}                             `json:\"-\"`\n\tXXX_unrecognized     []byte                               `json:\"-\"`\n\tXXX_sizecache        int32                                `json:\"-\"`\n}\n\nfunc (m *GetTaskListsByDomainResponse) Reset()         { *m = GetTaskListsByDomainResponse{} }\nfunc (m *GetTaskListsByDomainResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetTaskListsByDomainResponse) ProtoMessage()    {}\nfunc (*GetTaskListsByDomainResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{23}\n}\nfunc (m *GetTaskListsByDomainResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetTaskListsByDomainResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetTaskListsByDomainResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetTaskListsByDomainResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetTaskListsByDomainResponse.Merge(m, src)\n}\nfunc (m *GetTaskListsByDomainResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetTaskListsByDomainResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetTaskListsByDomainResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetTaskListsByDomainResponse proto.InternalMessageInfo\n\nfunc (m *GetTaskListsByDomainResponse) GetDecisionTaskListMap() map[string]*DescribeTaskListResponse {\n\tif m != nil {\n\t\treturn m.DecisionTaskListMap\n\t}\n\treturn nil\n}\n\nfunc (m *GetTaskListsByDomainResponse) GetActivityTaskListMap() map[string]*DescribeTaskListResponse {\n\tif m != nil {\n\t\treturn m.ActivityTaskListMap\n\t}\n\treturn nil\n}\n\ntype UpdateTaskListPartitionConfigRequest struct {\n\tDomainId             string                      `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tTaskList             *v1.TaskList                `protobuf:\"bytes,2,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tTaskListType         v1.TaskListType             `protobuf:\"varint,3,opt,name=task_list_type,json=taskListType,proto3,enum=uber.cadence.api.v1.TaskListType\" json:\"task_list_type,omitempty\"`\n\tPartitionConfig      *v1.TaskListPartitionConfig `protobuf:\"bytes,4,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) Reset()         { *m = UpdateTaskListPartitionConfigRequest{} }\nfunc (m *UpdateTaskListPartitionConfigRequest) String() string { return proto.CompactTextString(m) }\nfunc (*UpdateTaskListPartitionConfigRequest) ProtoMessage()    {}\nfunc (*UpdateTaskListPartitionConfigRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{24}\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_UpdateTaskListPartitionConfigRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_UpdateTaskListPartitionConfigRequest.Merge(m, src)\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_UpdateTaskListPartitionConfigRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_UpdateTaskListPartitionConfigRequest proto.InternalMessageInfo\n\nfunc (m *UpdateTaskListPartitionConfigRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) GetTaskListType() v1.TaskListType {\n\tif m != nil {\n\t\treturn m.TaskListType\n\t}\n\treturn v1.TaskListType_TASK_LIST_TYPE_INVALID\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) GetPartitionConfig() *v1.TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype UpdateTaskListPartitionConfigResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *UpdateTaskListPartitionConfigResponse) Reset()         { *m = UpdateTaskListPartitionConfigResponse{} }\nfunc (m *UpdateTaskListPartitionConfigResponse) String() string { return proto.CompactTextString(m) }\nfunc (*UpdateTaskListPartitionConfigResponse) ProtoMessage()    {}\nfunc (*UpdateTaskListPartitionConfigResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{25}\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_UpdateTaskListPartitionConfigResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_UpdateTaskListPartitionConfigResponse.Merge(m, src)\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_UpdateTaskListPartitionConfigResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_UpdateTaskListPartitionConfigResponse proto.InternalMessageInfo\n\ntype RefreshTaskListPartitionConfigRequest struct {\n\tDomainId             string                      `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tTaskList             *v1.TaskList                `protobuf:\"bytes,2,opt,name=task_list,json=taskList,proto3\" json:\"task_list,omitempty\"`\n\tTaskListType         v1.TaskListType             `protobuf:\"varint,3,opt,name=task_list_type,json=taskListType,proto3,enum=uber.cadence.api.v1.TaskListType\" json:\"task_list_type,omitempty\"`\n\tPartitionConfig      *v1.TaskListPartitionConfig `protobuf:\"bytes,4,opt,name=partition_config,json=partitionConfig,proto3\" json:\"partition_config,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) Reset()         { *m = RefreshTaskListPartitionConfigRequest{} }\nfunc (m *RefreshTaskListPartitionConfigRequest) String() string { return proto.CompactTextString(m) }\nfunc (*RefreshTaskListPartitionConfigRequest) ProtoMessage()    {}\nfunc (*RefreshTaskListPartitionConfigRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{26}\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RefreshTaskListPartitionConfigRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RefreshTaskListPartitionConfigRequest.Merge(m, src)\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_RefreshTaskListPartitionConfigRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RefreshTaskListPartitionConfigRequest proto.InternalMessageInfo\n\nfunc (m *RefreshTaskListPartitionConfigRequest) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) GetTaskList() *v1.TaskList {\n\tif m != nil {\n\t\treturn m.TaskList\n\t}\n\treturn nil\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) GetTaskListType() v1.TaskListType {\n\tif m != nil {\n\t\treturn m.TaskListType\n\t}\n\treturn v1.TaskListType_TASK_LIST_TYPE_INVALID\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) GetPartitionConfig() *v1.TaskListPartitionConfig {\n\tif m != nil {\n\t\treturn m.PartitionConfig\n\t}\n\treturn nil\n}\n\ntype RefreshTaskListPartitionConfigResponse struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RefreshTaskListPartitionConfigResponse) Reset() {\n\t*m = RefreshTaskListPartitionConfigResponse{}\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) String() string { return proto.CompactTextString(m) }\nfunc (*RefreshTaskListPartitionConfigResponse) ProtoMessage()    {}\nfunc (*RefreshTaskListPartitionConfigResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_826e827d3aabf7fc, []int{27}\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RefreshTaskListPartitionConfigResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RefreshTaskListPartitionConfigResponse.Merge(m, src)\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_RefreshTaskListPartitionConfigResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RefreshTaskListPartitionConfigResponse proto.InternalMessageInfo\n\nfunc init() {\n\tproto.RegisterType((*TaskListPartition)(nil), \"uber.cadence.matching.v1.TaskListPartition\")\n\tproto.RegisterType((*TaskListPartitionConfig)(nil), \"uber.cadence.matching.v1.TaskListPartitionConfig\")\n\tproto.RegisterMapType((map[int32]*TaskListPartition)(nil), \"uber.cadence.matching.v1.TaskListPartitionConfig.ReadPartitionsEntry\")\n\tproto.RegisterMapType((map[int32]*TaskListPartition)(nil), \"uber.cadence.matching.v1.TaskListPartitionConfig.WritePartitionsEntry\")\n\tproto.RegisterType((*LoadBalancerHints)(nil), \"uber.cadence.matching.v1.LoadBalancerHints\")\n\tproto.RegisterType((*PollForDecisionTaskRequest)(nil), \"uber.cadence.matching.v1.PollForDecisionTaskRequest\")\n\tproto.RegisterType((*PollForDecisionTaskResponse)(nil), \"uber.cadence.matching.v1.PollForDecisionTaskResponse\")\n\tproto.RegisterMapType((map[string]*v1.WorkflowQuery)(nil), \"uber.cadence.matching.v1.PollForDecisionTaskResponse.QueriesEntry\")\n\tproto.RegisterType((*PollForActivityTaskRequest)(nil), \"uber.cadence.matching.v1.PollForActivityTaskRequest\")\n\tproto.RegisterType((*PollForActivityTaskResponse)(nil), \"uber.cadence.matching.v1.PollForActivityTaskResponse\")\n\tproto.RegisterType((*AddDecisionTaskRequest)(nil), \"uber.cadence.matching.v1.AddDecisionTaskRequest\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.matching.v1.AddDecisionTaskRequest.PartitionConfigEntry\")\n\tproto.RegisterType((*AddDecisionTaskResponse)(nil), \"uber.cadence.matching.v1.AddDecisionTaskResponse\")\n\tproto.RegisterType((*AddActivityTaskRequest)(nil), \"uber.cadence.matching.v1.AddActivityTaskRequest\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.matching.v1.AddActivityTaskRequest.PartitionConfigEntry\")\n\tproto.RegisterType((*ActivityTaskDispatchInfo)(nil), \"uber.cadence.matching.v1.ActivityTaskDispatchInfo\")\n\tproto.RegisterType((*AddActivityTaskResponse)(nil), \"uber.cadence.matching.v1.AddActivityTaskResponse\")\n\tproto.RegisterType((*QueryWorkflowRequest)(nil), \"uber.cadence.matching.v1.QueryWorkflowRequest\")\n\tproto.RegisterType((*QueryWorkflowResponse)(nil), \"uber.cadence.matching.v1.QueryWorkflowResponse\")\n\tproto.RegisterType((*RespondQueryTaskCompletedRequest)(nil), \"uber.cadence.matching.v1.RespondQueryTaskCompletedRequest\")\n\tproto.RegisterType((*RespondQueryTaskCompletedResponse)(nil), \"uber.cadence.matching.v1.RespondQueryTaskCompletedResponse\")\n\tproto.RegisterType((*CancelOutstandingPollRequest)(nil), \"uber.cadence.matching.v1.CancelOutstandingPollRequest\")\n\tproto.RegisterType((*CancelOutstandingPollResponse)(nil), \"uber.cadence.matching.v1.CancelOutstandingPollResponse\")\n\tproto.RegisterType((*DescribeTaskListRequest)(nil), \"uber.cadence.matching.v1.DescribeTaskListRequest\")\n\tproto.RegisterType((*DescribeTaskListResponse)(nil), \"uber.cadence.matching.v1.DescribeTaskListResponse\")\n\tproto.RegisterType((*ListTaskListPartitionsRequest)(nil), \"uber.cadence.matching.v1.ListTaskListPartitionsRequest\")\n\tproto.RegisterType((*ListTaskListPartitionsResponse)(nil), \"uber.cadence.matching.v1.ListTaskListPartitionsResponse\")\n\tproto.RegisterType((*GetTaskListsByDomainRequest)(nil), \"uber.cadence.matching.v1.GetTaskListsByDomainRequest\")\n\tproto.RegisterType((*GetTaskListsByDomainResponse)(nil), \"uber.cadence.matching.v1.GetTaskListsByDomainResponse\")\n\tproto.RegisterMapType((map[string]*DescribeTaskListResponse)(nil), \"uber.cadence.matching.v1.GetTaskListsByDomainResponse.ActivityTaskListMapEntry\")\n\tproto.RegisterMapType((map[string]*DescribeTaskListResponse)(nil), \"uber.cadence.matching.v1.GetTaskListsByDomainResponse.DecisionTaskListMapEntry\")\n\tproto.RegisterType((*UpdateTaskListPartitionConfigRequest)(nil), \"uber.cadence.matching.v1.UpdateTaskListPartitionConfigRequest\")\n\tproto.RegisterType((*UpdateTaskListPartitionConfigResponse)(nil), \"uber.cadence.matching.v1.UpdateTaskListPartitionConfigResponse\")\n\tproto.RegisterType((*RefreshTaskListPartitionConfigRequest)(nil), \"uber.cadence.matching.v1.RefreshTaskListPartitionConfigRequest\")\n\tproto.RegisterType((*RefreshTaskListPartitionConfigResponse)(nil), \"uber.cadence.matching.v1.RefreshTaskListPartitionConfigResponse\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/matching/v1/service.proto\", fileDescriptor_826e827d3aabf7fc)\n}\n\nvar fileDescriptor_826e827d3aabf7fc = []byte{\n\t// 2528 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x5a, 0xcb, 0x6f, 0x1c, 0x49,\n\t0x19, 0x57, 0x8f, 0x3d, 0x7e, 0x7c, 0x63, 0x8f, 0xed, 0xb2, 0xd7, 0xe9, 0x4c, 0x62, 0xc7, 0x99,\n\t0x6c, 0xb2, 0x5e, 0x58, 0xc6, 0x6b, 0xef, 0x26, 0x64, 0xb3, 0x62, 0x83, 0x1f, 0x71, 0x32, 0x68,\n\t0x43, 0xb2, 0x1d, 0x6f, 0x22, 0xc1, 0x2a, 0x4d, 0x79, 0xba, 0xec, 0x69, 0xdc, 0xd3, 0xdd, 0xe9,\n\t0xae, 0xb6, 0x77, 0xf6, 0xc0, 0x01, 0x01, 0x42, 0xe2, 0x0a, 0x77, 0x5e, 0xff, 0x04, 0x17, 0xce,\n\t0x1c, 0x39, 0x22, 0xad, 0x90, 0x20, 0x12, 0x7f, 0x00, 0x48, 0xdc, 0x38, 0xa0, 0x7a, 0xf4, 0x4c,\n\t0xf7, 0x4c, 0xf5, 0x3c, 0x6c, 0x27, 0xcb, 0x81, 0x9b, 0xbb, 0xea, 0x7b, 0xd5, 0xf7, 0xfa, 0x7d,\n\t0x55, 0x63, 0xb8, 0x11, 0xed, 0x93, 0x60, 0xad, 0x86, 0x2d, 0xe2, 0xd6, 0xc8, 0x5a, 0x03, 0xd3,\n\t0x5a, 0xdd, 0x76, 0x0f, 0xd7, 0x8e, 0xd7, 0xd7, 0x42, 0x12, 0x1c, 0xdb, 0x35, 0x52, 0xf1, 0x03,\n\t0x8f, 0x7a, 0x48, 0x67, 0x74, 0x15, 0x49, 0x57, 0x89, 0xe9, 0x2a, 0xc7, 0xeb, 0xa5, 0xe5, 0x43,\n\t0xcf, 0x3b, 0x74, 0xc8, 0x1a, 0xa7, 0xdb, 0x8f, 0x0e, 0xd6, 0xac, 0x28, 0xc0, 0xd4, 0xf6, 0x5c,\n\t0xc1, 0x59, 0xba, 0xd2, 0xb9, 0x4f, 0xed, 0x06, 0x09, 0x29, 0x6e, 0xf8, 0x92, 0xa0, 0x4b, 0xc0,\n\t0x49, 0x80, 0x7d, 0x9f, 0x04, 0xa1, 0xdc, 0x5f, 0x49, 0x99, 0x88, 0x7d, 0x9b, 0x59, 0x57, 0xf3,\n\t0x1a, 0x8d, 0xb6, 0x0a, 0x15, 0xc5, 0x8b, 0x88, 0x04, 0x4d, 0x49, 0x50, 0x56, 0x11, 0x50, 0x1c,\n\t0x1e, 0x39, 0x76, 0x48, 0x25, 0xcd, 0xaa, 0x8a, 0x46, 0x3a, 0xc1, 0x3c, 0xf1, 0x82, 0x23, 0x12,\n\t0x48, 0xca, 0xaf, 0xf5, 0xa3, 0x3c, 0x70, 0xbc, 0x13, 0x49, 0x7b, 0x55, 0x45, 0x5b, 0xb7, 0x43,\n\t0xea, 0xb5, 0x8c, 0x7b, 0x33, 0x45, 0x12, 0xd6, 0x71, 0x40, 0xac, 0x6e, 0xaa, 0xeb, 0x19, 0x54,\n\t0xe9, 0x53, 0x94, 0x3f, 0x82, 0xb9, 0x3d, 0x1c, 0x1e, 0x7d, 0x6c, 0x87, 0xf4, 0x31, 0x0e, 0xa8,\n\t0xcd, 0x02, 0x81, 0xde, 0x86, 0x59, 0x3b, 0xf4, 0x1c, 0x1e, 0x15, 0xf3, 0x30, 0xf0, 0x22, 0x3f,\n\t0xd4, 0xb5, 0x95, 0x91, 0xd5, 0x49, 0x63, 0xa6, 0xb5, 0x7e, 0x9f, 0x2f, 0x97, 0xff, 0x3e, 0x0a,\n\t0x17, 0xba, 0x04, 0x6c, 0x7b, 0xee, 0x81, 0x7d, 0x88, 0x74, 0x18, 0x3f, 0x26, 0x41, 0x68, 0x7b,\n\t0xae, 0xae, 0xad, 0x68, 0xab, 0x23, 0x46, 0xfc, 0x89, 0x36, 0x60, 0xde, 0x8d, 0x1a, 0x66, 0x40,\n\t0xb0, 0x65, 0xfa, 0x31, 0x57, 0xa8, 0xe7, 0x56, 0xb4, 0xd5, 0xfc, 0x56, 0x4e, 0xd7, 0x8c, 0x39,\n\t0x37, 0x6a, 0x18, 0x04, 0x5b, 0x2d, 0x91, 0x21, 0x7a, 0x1f, 0x16, 0x18, 0xcf, 0x49, 0x60, 0x53,\n\t0x92, 0x64, 0x1a, 0x69, 0x31, 0x21, 0x37, 0x6a, 0x3c, 0x63, 0xdb, 0x09, 0x2e, 0x17, 0x66, 0x3a,\n\t0xb5, 0x8c, 0xae, 0x8c, 0xac, 0x16, 0x36, 0xee, 0x55, 0xb2, 0x32, 0xb4, 0x92, 0x71, 0x9e, 0x4a,\n\t0xda, 0xa0, 0x7b, 0x2e, 0x0d, 0x9a, 0x46, 0x31, 0x48, 0x5b, 0xf9, 0x02, 0x66, 0xbb, 0x2c, 0xcc,\n\t0x73, 0x85, 0xbb, 0xc3, 0x2b, 0xec, 0x38, 0x8c, 0xd0, 0x38, 0x73, 0x92, 0x5e, 0x2d, 0xb9, 0x30,\n\t0xaf, 0xb0, 0x0c, 0xcd, 0xc2, 0xc8, 0x11, 0x69, 0x72, 0xcf, 0xe7, 0x0d, 0xf6, 0x27, 0xda, 0x84,\n\t0xfc, 0x31, 0x76, 0x22, 0xc2, 0xfd, 0x5c, 0xd8, 0xf8, 0xfa, 0x10, 0x06, 0x19, 0x82, 0xf3, 0x4e,\n\t0xee, 0xb6, 0x56, 0xf2, 0x60, 0x41, 0x65, 0xd8, 0x2b, 0x53, 0x58, 0xfe, 0x01, 0xcc, 0x7d, 0xec,\n\t0x61, 0x6b, 0x0b, 0x3b, 0xd8, 0xad, 0x91, 0xe0, 0x81, 0xed, 0xd2, 0x10, 0x5d, 0x83, 0xe9, 0x7d,\n\t0x5c, 0x3b, 0x72, 0xbc, 0x43, 0xb3, 0xe6, 0x45, 0x2e, 0x95, 0x29, 0x36, 0x25, 0x17, 0xb7, 0xd9,\n\t0x1a, 0xba, 0x01, 0x33, 0x01, 0x66, 0xc1, 0x20, 0x81, 0x19, 0x92, 0x9a, 0xe7, 0x5a, 0xdc, 0x14,\n\t0xcd, 0x98, 0x66, 0xcb, 0x8f, 0x49, 0xf0, 0x84, 0x2f, 0x96, 0xff, 0xa9, 0x41, 0xe9, 0xb1, 0xe7,\n\t0x38, 0xbb, 0x5e, 0xb0, 0x43, 0x6a, 0x36, 0xcb, 0x51, 0x66, 0x91, 0x41, 0x5e, 0x44, 0x24, 0xa4,\n\t0xa8, 0x0a, 0xe3, 0x81, 0xf8, 0x93, 0x6b, 0x29, 0x6c, 0xac, 0xa5, 0x4f, 0x82, 0x7d, 0x9b, 0x1d,\n\t0x22, 0x5b, 0x82, 0x11, 0xf3, 0xa3, 0x4b, 0x30, 0x69, 0x79, 0x0d, 0x6c, 0xbb, 0xa6, 0x2d, 0x6c,\n\t0x99, 0x34, 0x26, 0xc4, 0x42, 0xd5, 0x62, 0x9b, 0xbe, 0xe7, 0x38, 0x24, 0x60, 0x9b, 0x23, 0x62,\n\t0x53, 0x2c, 0x54, 0x2d, 0x74, 0x1d, 0x8a, 0x07, 0x5e, 0x70, 0x82, 0x03, 0x8b, 0x58, 0xe6, 0x41,\n\t0xe0, 0x35, 0xf4, 0x51, 0x4e, 0x31, 0xdd, 0x5a, 0xdd, 0x0d, 0xbc, 0x06, 0x7a, 0x0b, 0x66, 0x3a,\n\t0x6a, 0x57, 0xcf, 0x73, 0xba, 0x62, 0xba, 0x74, 0xcb, 0x7f, 0x2c, 0xc0, 0x25, 0xa5, 0xc5, 0xa1,\n\t0xef, 0xb9, 0x21, 0x41, 0x4b, 0x00, 0xac, 0x57, 0x98, 0xd4, 0x3b, 0x22, 0xa2, 0x80, 0xa7, 0x8c,\n\t0x49, 0xb6, 0xb2, 0xc7, 0x16, 0xd0, 0xa7, 0x80, 0xe2, 0xd6, 0x65, 0x92, 0xcf, 0x49, 0x2d, 0x62,\n\t0x92, 0x65, 0xa0, 0x6f, 0x28, 0xdd, 0xf3, 0x4c, 0x92, 0xdf, 0x8b, 0xa9, 0x8d, 0xb9, 0x93, 0xce,\n\t0x25, 0xb4, 0x0b, 0xd3, 0x2d, 0xb1, 0xb4, 0xe9, 0x13, 0xee, 0x86, 0xc2, 0xc6, 0xd5, 0x9e, 0x12,\n\t0xf7, 0x9a, 0x3e, 0x31, 0xa6, 0x4e, 0x12, 0x5f, 0xe8, 0x29, 0x5c, 0xf4, 0x03, 0x72, 0x6c, 0x7b,\n\t0x51, 0x68, 0x86, 0x14, 0x07, 0x94, 0x58, 0x26, 0x39, 0x26, 0x2e, 0x65, 0xae, 0x1d, 0xe5, 0x32,\n\t0x2f, 0x55, 0x04, 0x90, 0x54, 0x62, 0x20, 0xa9, 0x54, 0x5d, 0x7a, 0xeb, 0xfd, 0xa7, 0x2c, 0xef,\n\t0x8c, 0xc5, 0x98, 0xfb, 0x89, 0x60, 0xbe, 0xc7, 0x78, 0xab, 0x16, 0x5a, 0x85, 0xd9, 0x2e, 0x71,\n\t0x79, 0x9e, 0x79, 0xc5, 0x30, 0x4d, 0xa9, 0xc3, 0x38, 0xa6, 0x94, 0x34, 0x7c, 0xaa, 0x8f, 0xf1,\n\t0x92, 0x88, 0x3f, 0x51, 0x19, 0xa6, 0x5d, 0xf2, 0x39, 0x6d, 0x0b, 0x18, 0xe7, 0x02, 0x0a, 0x6c,\n\t0x31, 0xe6, 0x7e, 0x07, 0x50, 0x2a, 0xbd, 0xcd, 0xba, 0xed, 0x52, 0x7d, 0x82, 0x13, 0xce, 0x26,\n\t0x73, 0x9c, 0x55, 0x03, 0xba, 0x0d, 0x7a, 0x48, 0xed, 0xda, 0x51, 0xb3, 0x1d, 0x0a, 0x93, 0xb8,\n\t0x78, 0xdf, 0x21, 0x96, 0x3e, 0xb9, 0xa2, 0xad, 0x4e, 0x18, 0x8b, 0x62, 0xbf, 0xe5, 0xe8, 0x7b,\n\t0x62, 0x17, 0xdd, 0x86, 0x3c, 0x07, 0x3e, 0x1d, 0xb8, 0x4f, 0xca, 0x3d, 0xfd, 0xfc, 0x09, 0xa3,\n\t0x34, 0x04, 0x03, 0x32, 0x60, 0xda, 0x92, 0x79, 0x63, 0xda, 0xee, 0x81, 0xa7, 0x17, 0xb8, 0x84,\n\t0x6f, 0xa4, 0x25, 0x08, 0xe0, 0xe1, 0x25, 0x1e, 0x60, 0x37, 0xb4, 0x89, 0x4b, 0xe3, 0x6c, 0xab,\n\t0xba, 0x07, 0x9e, 0x31, 0x65, 0x25, 0xbe, 0xd0, 0x73, 0xb8, 0xdc, 0x9d, 0x54, 0x26, 0x4f, 0x43,\n\t0x86, 0x59, 0xfa, 0x14, 0x57, 0xb1, 0xa4, 0x34, 0x32, 0x6e, 0x21, 0xc6, 0xc5, 0xae, 0xac, 0x8a,\n\t0xb7, 0x50, 0x05, 0xe6, 0x85, 0xd3, 0x19, 0x52, 0x12, 0x33, 0x46, 0xa7, 0x69, 0x1e, 0x9f, 0x39,\n\t0xbe, 0xf5, 0x84, 0xed, 0x3c, 0x95, 0x38, 0x75, 0x15, 0xa6, 0xf6, 0x03, 0xec, 0xd6, 0xea, 0xb2,\n\t0x0a, 0x8a, 0xbc, 0x0a, 0x0a, 0x62, 0x4d, 0xd4, 0xc1, 0x26, 0x14, 0xc3, 0x5a, 0x9d, 0x58, 0x91,\n\t0x43, 0x2c, 0x93, 0x8d, 0x2a, 0xfa, 0x0c, 0x37, 0xb2, 0xd4, 0x95, 0x5d, 0x7b, 0xf1, 0x1c, 0x63,\n\t0x4c, 0xb7, 0x38, 0xd8, 0x1a, 0xfa, 0x16, 0x4c, 0xc5, 0x39, 0xc5, 0x05, 0xcc, 0xf6, 0x15, 0x50,\n\t0x90, 0xf4, 0x9c, 0xfd, 0x33, 0x18, 0x67, 0x11, 0xb1, 0x49, 0xa8, 0xcf, 0x71, 0xa4, 0xd9, 0xca,\n\t0xee, 0xb3, 0x3d, 0x0a, 0xbe, 0xf2, 0x89, 0x10, 0x22, 0x50, 0x26, 0x16, 0xc9, 0x5c, 0x46, 0x3d,\n\t0x8a, 0x1d, 0x53, 0x8e, 0x17, 0xe6, 0x7e, 0x93, 0x92, 0x50, 0x47, 0x3c, 0x13, 0xe7, 0xf8, 0xd6,\n\t0x03, 0xb1, 0xb3, 0xc5, 0x36, 0xd0, 0x67, 0x30, 0xdb, 0x82, 0x3e, 0xb3, 0xc6, 0x71, 0x4c, 0x9f,\n\t0xe7, 0x07, 0x5a, 0x1f, 0x1a, 0x00, 0x8d, 0x19, 0xbf, 0x63, 0xa4, 0xf8, 0x3e, 0xcc, 0x3b, 0x1e,\n\t0xb6, 0xcc, 0x7d, 0x89, 0x05, 0xbc, 0x2c, 0x42, 0x7d, 0xa1, 0x1f, 0xbe, 0x74, 0xe1, 0x87, 0x31,\n\t0xe7, 0x74, 0x41, 0xca, 0x43, 0x98, 0xc5, 0x11, 0xf5, 0xa4, 0xd5, 0xa2, 0xe2, 0xde, 0xe0, 0x92,\n\t0xaf, 0x29, 0x33, 0x6e, 0x33, 0xa2, 0x9e, 0xb0, 0x8b, 0xf1, 0x1b, 0x45, 0x9c, 0xfa, 0x2e, 0x3d,\n\t0x87, 0xa9, 0xa4, 0x4b, 0x93, 0xf8, 0x38, 0x29, 0xf0, 0xf1, 0x76, 0x1a, 0x1f, 0x07, 0x2a, 0xbe,\n\t0x36, 0x2c, 0x26, 0x40, 0x6b, 0xb3, 0x46, 0xed, 0x63, 0x9b, 0x36, 0x4f, 0x0f, 0x5a, 0x0a, 0x09,\n\t0xff, 0x8b, 0xa0, 0xf5, 0x2b, 0x68, 0x81, 0x56, 0xda, 0xe2, 0xaf, 0x14, 0xb4, 0xae, 0x40, 0x01,\n\t0x4b, 0x6b, 0xda, 0x4e, 0x80, 0x78, 0xa9, 0x6a, 0x31, 0x54, 0x6b, 0x11, 0x70, 0x54, 0x1b, 0xed,\n\t0x81, 0x6a, 0xad, 0x83, 0x71, 0x54, 0xc3, 0x89, 0x2f, 0xb4, 0x01, 0x79, 0xdb, 0xf5, 0x23, 0xca,\n\t0xbd, 0x53, 0xd8, 0xb8, 0xac, 0x8e, 0x28, 0x6e, 0xb2, 0xdc, 0x36, 0x04, 0xa9, 0xa2, 0x41, 0x8d,\n\t0x9d, 0xb5, 0x41, 0x8d, 0x0f, 0xd7, 0xa0, 0xf6, 0xe0, 0x62, 0x2c, 0xcf, 0x64, 0xe5, 0xe5, 0x78,\n\t0x21, 0xe1, 0x82, 0xbc, 0x48, 0x40, 0x5a, 0x61, 0xe3, 0x62, 0x97, 0xac, 0x1d, 0x79, 0x2b, 0x34,\n\t0x16, 0x63, 0xde, 0x3d, 0x6f, 0x9b, 0x71, 0xee, 0x09, 0x46, 0xf4, 0x5d, 0x58, 0xe4, 0x4a, 0xba,\n\t0x45, 0x4e, 0xf6, 0x13, 0x39, 0xcf, 0x19, 0x3b, 0xe4, 0xed, 0xc2, 0x5c, 0x9d, 0xe0, 0x80, 0xee,\n\t0x13, 0x4c, 0x5b, 0xa2, 0xa0, 0x9f, 0xa8, 0xd9, 0x16, 0x4f, 0x2c, 0x27, 0x81, 0xfb, 0x85, 0x34,\n\t0xee, 0x3f, 0x87, 0xe5, 0x74, 0x24, 0x4c, 0xef, 0xc0, 0xa4, 0x75, 0x3b, 0x34, 0x63, 0x86, 0xa9,\n\t0xbe, 0x8e, 0x2d, 0xa5, 0x22, 0xf3, 0xe8, 0x60, 0xaf, 0x6e, 0x87, 0x9b, 0x52, 0x7e, 0x35, 0x79,\n\t0x02, 0x8b, 0x50, 0x6c, 0x3b, 0x21, 0xc7, 0xb6, 0x7e, 0x99, 0xd2, 0x3e, 0xc4, 0x8e, 0xe0, 0xea,\n\t0x1e, 0xc3, 0x8a, 0xa7, 0x1b, 0xc3, 0xde, 0x82, 0x99, 0x96, 0x1c, 0xd1, 0x31, 0x38, 0x3c, 0x4e,\n\t0x1a, 0xc5, 0x78, 0x79, 0x87, 0xaf, 0xa2, 0xf7, 0x60, 0xac, 0x4e, 0xb0, 0x45, 0x02, 0x89, 0x7e,\n\t0x97, 0x94, 0x9a, 0x1e, 0x70, 0x12, 0x43, 0x92, 0x66, 0xa1, 0xc1, 0xdc, 0xb9, 0xa0, 0xc1, 0xab,\n\t0x05, 0x32, 0x15, 0xd6, 0x2c, 0x9c, 0x1a, 0x6b, 0xca, 0x7f, 0x19, 0x85, 0xc5, 0x4d, 0xcb, 0x52,\n\t0x5d, 0x5e, 0x52, 0xcd, 0x5b, 0xeb, 0x68, 0xde, 0xaf, 0xa8, 0x21, 0xde, 0x81, 0xc9, 0xf6, 0xd0,\n\t0x36, 0x32, 0xc8, 0xd0, 0x36, 0x41, 0xe3, 0x19, 0xed, 0x0a, 0x14, 0x5a, 0xdd, 0x42, 0xce, 0xea,\n\t0x23, 0x06, 0xc4, 0x4b, 0x55, 0xab, 0xb3, 0x9d, 0xc8, 0x26, 0x20, 0x0b, 0x36, 0x3f, 0x44, 0x3b,\n\t0xe1, 0xa3, 0x7d, 0x5c, 0xb6, 0x77, 0x60, 0x2c, 0xf4, 0xa2, 0xa0, 0x26, 0xda, 0x63, 0xb1, 0x13,\n\t0x8c, 0x13, 0x73, 0x2c, 0x0e, 0x8f, 0x9e, 0x70, 0x4a, 0x43, 0x72, 0x28, 0x50, 0x6e, 0x5c, 0x85,\n\t0x72, 0xbe, 0x22, 0xa3, 0x26, 0xfa, 0x3d, 0x46, 0xa8, 0xa3, 0x5a, 0xe9, 0x48, 0x30, 0xf9, 0x34,\n\t0xd0, 0x91, 0x65, 0xa5, 0x2d, 0x58, 0x50, 0x11, 0x2a, 0x46, 0x91, 0x85, 0xe4, 0x28, 0x32, 0x99,\n\t0x1c, 0x33, 0x4e, 0xe0, 0x42, 0x97, 0x0d, 0x12, 0x6d, 0x55, 0x25, 0xa2, 0x9d, 0x57, 0x89, 0x94,\n\t0xff, 0x95, 0xe7, 0x39, 0xad, 0x9a, 0x6d, 0xbe, 0x8a, 0x9c, 0x66, 0x37, 0x3f, 0x1e, 0x6e, 0xb3,\n\t0xad, 0x5a, 0x20, 0x7d, 0x51, 0xac, 0xef, 0xc4, 0x06, 0xa4, 0xb2, 0x7f, 0xf4, 0x4c, 0xd9, 0x9f,\n\t0x1f, 0x2e, 0xfb, 0xc7, 0xce, 0x9e, 0xfd, 0xe3, 0xe7, 0x90, 0xfd, 0x13, 0xaa, 0xec, 0x77, 0x41,\n\t0xc7, 0x89, 0x50, 0xee, 0xd8, 0xa1, 0xcf, 0xb2, 0x82, 0xdd, 0xfb, 0x24, 0x62, 0x6f, 0xf4, 0xa8,\n\t0x82, 0x0c, 0x4e, 0x23, 0x53, 0xa6, 0xb2, 0xda, 0x60, 0x80, 0x6a, 0x53, 0xe4, 0xdb, 0x6b, 0xac,\n\t0xb6, 0x2f, 0x47, 0x40, 0xcf, 0x3a, 0x2c, 0xfa, 0x0e, 0xcc, 0xb4, 0x07, 0x08, 0x7e, 0x5b, 0x95,\n\t0xe5, 0xa6, 0xc6, 0x65, 0x79, 0x2f, 0xe3, 0x4f, 0x0a, 0x46, 0x7b, 0x08, 0xe4, 0xdf, 0x5d, 0x33,\n\t0x5d, 0x6e, 0xb8, 0x99, 0x2e, 0x31, 0xe5, 0x8c, 0x0c, 0x3b, 0xe5, 0x8c, 0x9e, 0xff, 0x94, 0x93,\n\t0x3f, 0x9f, 0x29, 0x67, 0xec, 0xdc, 0xa6, 0x9c, 0x71, 0xd5, 0x94, 0x23, 0x7b, 0xa9, 0xf2, 0xe6,\n\t0xf2, 0x6a, 0x7b, 0xe9, 0x97, 0x1a, 0x2c, 0xf0, 0x0b, 0x64, 0x7c, 0x8a, 0xb8, 0x93, 0x6e, 0x77,\n\t0xde, 0x12, 0xdf, 0x56, 0x1e, 0x5e, 0xc5, 0x3b, 0xe0, 0xfd, 0xf0, 0x2c, 0xb3, 0xc0, 0x60, 0xd7,\n\t0xc7, 0xf2, 0x7f, 0x34, 0x78, 0xa3, 0xc3, 0x42, 0xe9, 0xd5, 0xbb, 0x30, 0xc5, 0x5f, 0xab, 0xcc,\n\t0x80, 0x84, 0x91, 0x13, 0x9f, 0xb1, 0x77, 0x9e, 0x14, 0x38, 0x87, 0xc1, 0x19, 0x50, 0x15, 0x8a,\n\t0xb1, 0x80, 0x1f, 0x92, 0x1a, 0x25, 0x56, 0xcf, 0xbb, 0xba, 0xb8, 0xa3, 0x4b, 0x4a, 0x63, 0xfa,\n\t0x45, 0xf2, 0x13, 0x3d, 0x53, 0x44, 0x58, 0xf8, 0xe3, 0x9d, 0x9e, 0xfe, 0xe8, 0x1b, 0xdc, 0x7f,\n\t0x68, 0xb0, 0x22, 0x4e, 0x6c, 0x71, 0x03, 0x18, 0xe3, 0xb6, 0xd7, 0xf0, 0x1d, 0xc2, 0xac, 0x90,\n\t0x31, 0x7a, 0xd4, 0x19, 0xe8, 0x9b, 0x4a, 0xa5, 0xfd, 0xe4, 0xbc, 0x86, 0xa0, 0x5f, 0x80, 0x71,\n\t0xce, 0x2b, 0x87, 0xbf, 0x49, 0x63, 0x8c, 0x7d, 0x56, 0xad, 0xf2, 0x35, 0xb8, 0xda, 0xc3, 0x3c,\n\t0x11, 0xf1, 0xf2, 0x5f, 0x35, 0xb8, 0xbc, 0xcd, 0xc6, 0x78, 0xe7, 0x51, 0x44, 0x43, 0x8a, 0x5d,\n\t0xcb, 0x76, 0x0f, 0x1f, 0x7b, 0x8e, 0x33, 0xd0, 0xec, 0x90, 0x7a, 0xcc, 0xc8, 0x75, 0x3c, 0x66,\n\t0xdc, 0x87, 0x62, 0xeb, 0x50, 0xed, 0xc7, 0xe9, 0x62, 0x46, 0xbf, 0x88, 0x4f, 0x26, 0xfa, 0x05,\n\t0x4d, 0x7c, 0x9d, 0x65, 0x40, 0x28, 0x5f, 0x81, 0xa5, 0x8c, 0xe3, 0x49, 0x07, 0xfc, 0x08, 0x2e,\n\t0xec, 0x90, 0xb0, 0x16, 0xd8, 0xfb, 0xa4, 0xc5, 0x2e, 0x8f, 0xbe, 0xdb, 0x99, 0x03, 0xea, 0xc4,\n\t0xcb, 0x60, 0x1f, 0x2c, 0xf4, 0xe5, 0x3f, 0xe4, 0x40, 0xef, 0x96, 0x20, 0xeb, 0xf1, 0x03, 0x18,\n\t0x17, 0xee, 0x14, 0x3f, 0x28, 0x16, 0x36, 0xae, 0x64, 0x3e, 0x4a, 0x91, 0x80, 0x03, 0x7c, 0x4c,\n\t0xcf, 0x6e, 0x4c, 0x6d, 0xef, 0x87, 0x14, 0xd3, 0x28, 0x94, 0xb5, 0x78, 0xad, 0xa7, 0xef, 0x9e,\n\t0x70, 0x52, 0xa3, 0x48, 0x53, 0xdf, 0xaf, 0xac, 0x1a, 0xcf, 0x14, 0xdc, 0x10, 0x96, 0x78, 0x92,\n\t0x74, 0xea, 0x0a, 0xe3, 0x08, 0x2e, 0xc2, 0x98, 0x04, 0x18, 0x91, 0xb9, 0xf2, 0x2b, 0xad, 0x34,\n\t0x37, 0x9c, 0xd2, 0x9f, 0xe5, 0x60, 0x39, 0x4b, 0xab, 0x0c, 0xdb, 0x0b, 0x58, 0x6a, 0xbf, 0x5f,\n\t0xb5, 0x82, 0x90, 0xf8, 0x89, 0x53, 0x04, 0xb3, 0x32, 0x98, 0xe7, 0x1e, 0x12, 0x8a, 0x2d, 0x4c,\n\t0xb1, 0x51, 0x4a, 0x0e, 0x6f, 0x69, 0xd5, 0x4c, 0x65, 0xeb, 0xe7, 0x05, 0xa5, 0xca, 0xdc, 0xe9,\n\t0x54, 0x5a, 0x89, 0x8b, 0x4c, 0x5a, 0x65, 0xf9, 0x26, 0x5c, 0xba, 0x4f, 0x5a, 0x6e, 0x08, 0xb7,\n\t0x9a, 0x02, 0xb5, 0xfb, 0xf8, 0xbe, 0xfc, 0xfb, 0x51, 0xb8, 0xac, 0xe6, 0x93, 0xde, 0xfb, 0x89,\n\t0x06, 0x8b, 0x8a, 0xb3, 0x34, 0xb0, 0x2f, 0xfd, 0xf6, 0x28, 0x1b, 0xe1, 0x7b, 0x09, 0xae, 0xec,\n\t0x74, 0x9c, 0xe5, 0x21, 0xf6, 0xc5, 0x68, 0x3a, 0x6f, 0x75, 0xef, 0x70, 0x33, 0x14, 0x51, 0x64,\n\t0x66, 0xe4, 0xce, 0x64, 0xc6, 0x66, 0x47, 0x14, 0xdb, 0x66, 0xe0, 0xee, 0x9d, 0xd2, 0x17, 0xac,\n\t0x3d, 0xa8, 0xed, 0x56, 0x4c, 0xca, 0x0f, 0xd2, 0x4f, 0xe4, 0x3d, 0xae, 0x08, 0x59, 0x3d, 0x27,\n\t0xf9, 0xd3, 0xf5, 0x17, 0xe9, 0xe1, 0xfa, 0x75, 0xea, 0x2e, 0xff, 0x26, 0x07, 0x6f, 0x7e, 0xea,\n\t0x5b, 0x98, 0x92, 0xac, 0x56, 0x32, 0x08, 0x40, 0x9d, 0xa1, 0xd0, 0xcf, 0x0f, 0xbf, 0x54, 0xbd,\n\t0x73, 0xf4, 0x3c, 0x26, 0x99, 0xb7, 0xe0, 0x7a, 0x1f, 0x17, 0x49, 0x90, 0xfb, 0x6d, 0x0e, 0xae,\n\t0x1b, 0xe4, 0x20, 0x20, 0x61, 0xfd, 0xff, 0xde, 0xcc, 0xf2, 0xe6, 0x2a, 0xdc, 0xe8, 0xe7, 0x23,\n\t0xe1, 0xce, 0x8d, 0x7f, 0x4f, 0x41, 0xe1, 0xa1, 0xcc, 0xe7, 0xcd, 0xc7, 0x55, 0xf4, 0x63, 0x0d,\n\t0xe6, 0x15, 0x3f, 0x15, 0xa2, 0xf7, 0x87, 0xfc, 0x65, 0x91, 0x87, 0xa0, 0x74, 0xf3, 0x54, 0xbf,\n\t0x47, 0x26, 0x8d, 0x48, 0x16, 0xed, 0x00, 0x46, 0x28, 0xae, 0xf0, 0x03, 0x18, 0xa1, 0xbc, 0x96,\n\t0x1d, 0xc3, 0x4c, 0xc7, 0xeb, 0x17, 0x7a, 0x77, 0xd8, 0xc7, 0xba, 0xd2, 0xfa, 0x10, 0x1c, 0x29,\n\t0xbd, 0xa9, 0x73, 0xbf, 0x3b, 0xec, 0xb3, 0x45, 0x1f, 0xbd, 0xca, 0xf3, 0xfa, 0x30, 0x9d, 0xba,\n\t0x49, 0xa1, 0x4a, 0xb6, 0x0c, 0xd5, 0xa5, 0xb0, 0xb4, 0x36, 0x30, 0xbd, 0xd4, 0xf8, 0x4b, 0x0d,\n\t0x2e, 0x66, 0x8e, 0xf5, 0xe8, 0x4e, 0xb6, 0xb8, 0x7e, 0x57, 0x95, 0xd2, 0x87, 0xa7, 0xe2, 0x95,\n\t0x66, 0xfd, 0x5c, 0x83, 0x37, 0x94, 0x83, 0x36, 0xba, 0x95, 0x2d, 0xb6, 0xd7, 0xc5, 0xa3, 0xf4,\n\t0xcd, 0xa1, 0xf9, 0xa4, 0x29, 0x4d, 0x98, 0xed, 0x04, 0x18, 0xb4, 0x3e, 0x0c, 0x18, 0x09, 0xfd,\n\t0xa7, 0xc0, 0x2f, 0xf4, 0x0b, 0x0d, 0x16, 0xd5, 0xb3, 0x21, 0xea, 0x71, 0x9c, 0x9e, 0x33, 0x6c,\n\t0xe9, 0xf6, 0xf0, 0x8c, 0xd2, 0x9a, 0x9f, 0x6a, 0xb0, 0xa0, 0x9a, 0x44, 0xd0, 0xcd, 0x61, 0x27,\n\t0x17, 0x61, 0xc9, 0xad, 0xd3, 0x0d, 0x3c, 0xe8, 0xd7, 0x1a, 0x2c, 0xf5, 0xc4, 0x29, 0xf4, 0x51,\n\t0xb6, 0xe4, 0x41, 0x66, 0x80, 0xd2, 0xdd, 0x53, 0xf3, 0x4b, 0x13, 0x7f, 0xa7, 0xc1, 0x72, 0xef,\n\t0xe6, 0x8f, 0xee, 0xf6, 0x2a, 0x8f, 0x01, 0xa0, 0xb5, 0xf4, 0xed, 0xd3, 0x0b, 0x10, 0x56, 0x6e,\n\t0xdd, 0xff, 0xd3, 0xcb, 0x65, 0xed, 0xcf, 0x2f, 0x97, 0xb5, 0xbf, 0xbd, 0x5c, 0xd6, 0xbe, 0xf7,\n\t0xc1, 0xa1, 0x4d, 0xeb, 0xd1, 0x7e, 0xa5, 0xe6, 0x35, 0xd6, 0x52, 0xff, 0xbd, 0x5a, 0x39, 0x24,\n\t0xae, 0xf8, 0x77, 0xdf, 0xe4, 0x7f, 0x1c, 0x7f, 0x18, 0xff, 0x7d, 0xbc, 0xbe, 0x3f, 0xc6, 0x77,\n\t0xdf, 0xfb, 0x6f, 0x00, 0x00, 0x00, 0xff, 0xff, 0x48, 0x50, 0xe3, 0xd9, 0x9f, 0x2c, 0x00, 0x00,\n}\n\nfunc (m *TaskListPartition) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TaskListPartition) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TaskListPartition) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.IsolationGroups) > 0 {\n\t\tfor iNdEx := len(m.IsolationGroups) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\ti -= len(m.IsolationGroups[iNdEx])\n\t\t\tcopy(dAtA[i:], m.IsolationGroups[iNdEx])\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(m.IsolationGroups[iNdEx])))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *TaskListPartitionConfig) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TaskListPartitionConfig) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TaskListPartitionConfig) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.WritePartitions) > 0 {\n\t\tfor k := range m.WritePartitions {\n\t\t\tv := m.WritePartitions[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti = encodeVarintService(dAtA, i, uint64(k))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x2a\n\t\t}\n\t}\n\tif len(m.ReadPartitions) > 0 {\n\t\tfor k := range m.ReadPartitions {\n\t\t\tv := m.ReadPartitions[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti = encodeVarintService(dAtA, i, uint64(k))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x22\n\t\t}\n\t}\n\tif m.NumWritePartitions != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NumWritePartitions))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.NumReadPartitions != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NumReadPartitions))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif m.Version != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Version))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *LoadBalancerHints) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *LoadBalancerHints) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *LoadBalancerHints) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.RatePerSecond != 0 {\n\t\ti -= 8\n\t\tencoding_binary.LittleEndian.PutUint64(dAtA[i:], uint64(math.Float64bits(float64(m.RatePerSecond))))\n\t\ti--\n\t\tdAtA[i] = 0x11\n\t}\n\tif m.BacklogCount != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.BacklogCount))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollForDecisionTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollForDecisionTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollForDecisionTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.IsolationGroup) > 0 {\n\t\ti -= len(m.IsolationGroup)\n\t\tcopy(dAtA[i:], m.IsolationGroup)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.IsolationGroup)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif len(m.ForwardedFrom) > 0 {\n\t\ti -= len(m.ForwardedFrom)\n\t\tcopy(dAtA[i:], m.ForwardedFrom)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ForwardedFrom)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.PollerId) > 0 {\n\t\ti -= len(m.PollerId)\n\t\tcopy(dAtA[i:], m.PollerId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.PollerId)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollForDecisionTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollForDecisionTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollForDecisionTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.AutoConfigHint != nil {\n\t\t{\n\t\t\tsize, err := m.AutoConfigHint.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0xaa\n\t}\n\tif m.LoadBalancerHints != nil {\n\t\t{\n\t\t\tsize, err := m.LoadBalancerHints.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0xa2\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x9a\n\t}\n\tif m.TotalHistoryBytes != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TotalHistoryBytes))\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x90\n\t}\n\tif len(m.Queries) > 0 {\n\t\tfor k := range m.Queries {\n\t\t\tv := m.Queries[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1\n\t\t\ti--\n\t\t\tdAtA[i] = 0x8a\n\t\t}\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x82\n\t}\n\tif m.ScheduledTime != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x7a\n\t}\n\tif len(m.BranchToken) > 0 {\n\t\ti -= len(m.BranchToken)\n\t\tcopy(dAtA[i:], m.BranchToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.BranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0x72\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.EventStoreVersion))\n\t\ti--\n\t\tdAtA[i] = 0x68\n\t}\n\tif m.WorkflowExecutionTaskList != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecutionTaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x62\n\t}\n\tif m.DecisionInfo != nil {\n\t\t{\n\t\t\tsize, err := m.DecisionInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x5a\n\t}\n\tif m.Query != nil {\n\t\t{\n\t\t\tsize, err := m.Query.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x52\n\t}\n\tif m.StickyExecutionEnabled {\n\t\ti--\n\t\tif m.StickyExecutionEnabled {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x48\n\t}\n\tif m.BacklogCountHint != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.BacklogCountHint))\n\t\ti--\n\t\tdAtA[i] = 0x40\n\t}\n\tif m.NextEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.NextEventId))\n\t\ti--\n\t\tdAtA[i] = 0x38\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.StartedEventId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.StartedEventId))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\t{\n\t\t\tsize, err := m.PreviousStartedEventId.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.TaskToken) > 0 {\n\t\ti -= len(m.TaskToken)\n\t\tcopy(dAtA[i:], m.TaskToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.TaskToken)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollForActivityTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollForActivityTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollForActivityTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.IsolationGroup) > 0 {\n\t\ti -= len(m.IsolationGroup)\n\t\tcopy(dAtA[i:], m.IsolationGroup)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.IsolationGroup)))\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif len(m.ForwardedFrom) > 0 {\n\t\ti -= len(m.ForwardedFrom)\n\t\tcopy(dAtA[i:], m.ForwardedFrom)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ForwardedFrom)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.PollerId) > 0 {\n\t\ti -= len(m.PollerId)\n\t\tcopy(dAtA[i:], m.PollerId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.PollerId)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PollForActivityTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PollForActivityTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PollForActivityTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.AutoConfigHint != nil {\n\t\t{\n\t\t\tsize, err := m.AutoConfigHint.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0xa2\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x9a\n\t}\n\tif m.LoadBalancerHints != nil {\n\t\t{\n\t\t\tsize, err := m.LoadBalancerHints.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x8a\n\t}\n\tif m.Header != nil {\n\t\t{\n\t\t\tsize, err := m.Header.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1\n\t\ti--\n\t\tdAtA[i] = 0x82\n\t}\n\tif len(m.WorkflowDomain) > 0 {\n\t\ti -= len(m.WorkflowDomain)\n\t\tcopy(dAtA[i:], m.WorkflowDomain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.WorkflowDomain)))\n\t\ti--\n\t\tdAtA[i] = 0x7a\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x72\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\t{\n\t\t\tsize, err := m.HeartbeatDetails.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x6a\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTimeOfThisAttempt.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x62\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x58\n\t}\n\tif m.HeartbeatTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.HeartbeatTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x52\n\t}\n\tif m.StartToCloseTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.StartToCloseTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif m.ScheduleToCloseTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduleToCloseTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.ScheduledTime != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.Input != nil {\n\t\t{\n\t\t\tsize, err := m.Input.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.ActivityType != nil {\n\t\t{\n\t\t\tsize, err := m.ActivityType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.ActivityId) > 0 {\n\t\ti -= len(m.ActivityId)\n\t\tcopy(dAtA[i:], m.ActivityId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ActivityId)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.TaskToken) > 0 {\n\t\ti -= len(m.TaskToken)\n\t\tcopy(dAtA[i:], m.TaskToken)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.TaskToken)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *AddDecisionTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *AddDecisionTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *AddDecisionTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k := range m.PartitionConfig {\n\t\t\tv := m.PartitionConfig[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x42\n\t\t}\n\t}\n\tif len(m.ForwardedFrom) > 0 {\n\t\ti -= len(m.ForwardedFrom)\n\t\tcopy(dAtA[i:], m.ForwardedFrom)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ForwardedFrom)))\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.Source != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Source))\n\t\ti--\n\t\tdAtA[i] = 0x30\n\t}\n\tif m.ScheduleToStartTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduleToStartTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.ScheduleId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduleId))\n\t\ti--\n\t\tdAtA[i] = 0x20\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *AddDecisionTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *AddDecisionTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *AddDecisionTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *AddActivityTaskRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *AddActivityTaskRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *AddActivityTaskRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k := range m.PartitionConfig {\n\t\t\tv := m.PartitionConfig[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x52\n\t\t}\n\t}\n\tif m.ActivityTaskDispatchInfo != nil {\n\t\t{\n\t\t\tsize, err := m.ActivityTaskDispatchInfo.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x4a\n\t}\n\tif len(m.ForwardedFrom) > 0 {\n\t\ti -= len(m.ForwardedFrom)\n\t\tcopy(dAtA[i:], m.ForwardedFrom)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ForwardedFrom)))\n\t\ti--\n\t\tdAtA[i] = 0x42\n\t}\n\tif m.Source != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Source))\n\t\ti--\n\t\tdAtA[i] = 0x38\n\t}\n\tif m.ScheduleToStartTimeout != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduleToStartTimeout.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.ScheduleId != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.ScheduleId))\n\t\ti--\n\t\tdAtA[i] = 0x28\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif len(m.SourceDomainId) > 0 {\n\t\ti -= len(m.SourceDomainId)\n\t\tcopy(dAtA[i:], m.SourceDomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.SourceDomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ActivityTaskDispatchInfo) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ActivityTaskDispatchInfo) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.WorkflowDomain) > 0 {\n\t\ti -= len(m.WorkflowDomain)\n\t\tcopy(dAtA[i:], m.WorkflowDomain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.WorkflowDomain)))\n\t\ti--\n\t\tdAtA[i] = 0x3a\n\t}\n\tif m.WorkflowType != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowType.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x32\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\t{\n\t\t\tsize, err := m.HeartbeatDetails.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x2a\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledTimeOfThisAttempt.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.Attempt != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.Attempt))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.StartedTime != nil {\n\t\t{\n\t\t\tsize, err := m.StartedTime.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ScheduledEvent != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *AddActivityTaskResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *AddActivityTaskResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *AddActivityTaskResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *QueryWorkflowRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *QueryWorkflowRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *QueryWorkflowRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ForwardedFrom) > 0 {\n\t\ti -= len(m.ForwardedFrom)\n\t\tcopy(dAtA[i:], m.ForwardedFrom)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ForwardedFrom)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *QueryWorkflowResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *QueryWorkflowResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *QueryWorkflowResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.QueryRejected != nil {\n\t\t{\n\t\t\tsize, err := m.QueryRejected.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.QueryResult != nil {\n\t\t{\n\t\t\tsize, err := m.QueryResult.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.TaskId) > 0 {\n\t\ti -= len(m.TaskId)\n\t\tcopy(dAtA[i:], m.TaskId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.TaskId)))\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RespondQueryTaskCompletedResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RespondQueryTaskCompletedResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RespondQueryTaskCompletedResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CancelOutstandingPollRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CancelOutstandingPollRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CancelOutstandingPollRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskListType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskListType))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif len(m.PollerId) > 0 {\n\t\ti -= len(m.PollerId)\n\t\tcopy(dAtA[i:], m.PollerId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.PollerId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *CancelOutstandingPollResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CancelOutstandingPollResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CancelOutstandingPollResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeTaskListRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeTaskListRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeTaskListRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.Request != nil {\n\t\t{\n\t\t\tsize, err := m.Request.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *DescribeTaskListResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *DescribeTaskListResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *DescribeTaskListResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.TaskListStatus != nil {\n\t\t{\n\t\t\tsize, err := m.TaskListStatus.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Pollers) > 0 {\n\t\tfor iNdEx := len(m.Pollers) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Pollers[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ListTaskListPartitionsRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ListTaskListPartitionsRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ListTaskListPartitionsRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Domain) > 0 {\n\t\ti -= len(m.Domain)\n\t\tcopy(dAtA[i:], m.Domain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Domain)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ListTaskListPartitionsResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ListTaskListPartitionsResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ListTaskListPartitionsResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.DecisionTaskListPartitions) > 0 {\n\t\tfor iNdEx := len(m.DecisionTaskListPartitions) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.DecisionTaskListPartitions[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif len(m.ActivityTaskListPartitions) > 0 {\n\t\tfor iNdEx := len(m.ActivityTaskListPartitions) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.ActivityTaskListPartitions[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetTaskListsByDomainRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetTaskListsByDomainRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetTaskListsByDomainRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Domain) > 0 {\n\t\ti -= len(m.Domain)\n\t\tcopy(dAtA[i:], m.Domain)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Domain)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetTaskListsByDomainResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetTaskListsByDomainResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetTaskListsByDomainResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ActivityTaskListMap) > 0 {\n\t\tfor k := range m.ActivityTaskListMap {\n\t\t\tv := m.ActivityTaskListMap[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif len(m.DecisionTaskListMap) > 0 {\n\t\tfor k := range m.DecisionTaskListMap {\n\t\t\tv := m.DecisionTaskListMap[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskListType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskListType))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *UpdateTaskListPartitionConfigResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *UpdateTaskListPartitionConfigResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *UpdateTaskListPartitionConfigResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.PartitionConfig != nil {\n\t\t{\n\t\t\tsize, err := m.PartitionConfig.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.TaskListType != 0 {\n\t\ti = encodeVarintService(dAtA, i, uint64(m.TaskListType))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif m.TaskList != nil {\n\t\t{\n\t\t\tsize, err := m.TaskList.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RefreshTaskListPartitionConfigResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RefreshTaskListPartitionConfigResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RefreshTaskListPartitionConfigResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintService(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovService(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *TaskListPartition) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.IsolationGroups) > 0 {\n\t\tfor _, s := range m.IsolationGroups {\n\t\t\tl = len(s)\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *TaskListPartitionConfig) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Version != 0 {\n\t\tn += 1 + sovService(uint64(m.Version))\n\t}\n\tif m.NumReadPartitions != 0 {\n\t\tn += 1 + sovService(uint64(m.NumReadPartitions))\n\t}\n\tif m.NumWritePartitions != 0 {\n\t\tn += 1 + sovService(uint64(m.NumWritePartitions))\n\t}\n\tif len(m.ReadPartitions) > 0 {\n\t\tfor k, v := range m.ReadPartitions {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + sovService(uint64(k)) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif len(m.WritePartitions) > 0 {\n\t\tfor k, v := range m.WritePartitions {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + sovService(uint64(k)) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *LoadBalancerHints) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.BacklogCount != 0 {\n\t\tn += 1 + sovService(uint64(m.BacklogCount))\n\t}\n\tif m.RatePerSecond != 0 {\n\t\tn += 9\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollForDecisionTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.PollerId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ForwardedFrom)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.IsolationGroup)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollForDecisionTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.TaskToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PreviousStartedEventId != nil {\n\t\tl = m.PreviousStartedEventId.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.StartedEventId))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.NextEventId != 0 {\n\t\tn += 1 + sovService(uint64(m.NextEventId))\n\t}\n\tif m.BacklogCountHint != 0 {\n\t\tn += 1 + sovService(uint64(m.BacklogCountHint))\n\t}\n\tif m.StickyExecutionEnabled {\n\t\tn += 2\n\t}\n\tif m.Query != nil {\n\t\tl = m.Query.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.DecisionInfo != nil {\n\t\tl = m.DecisionInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecutionTaskList != nil {\n\t\tl = m.WorkflowExecutionTaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.EventStoreVersion != 0 {\n\t\tn += 1 + sovService(uint64(m.EventStoreVersion))\n\t}\n\tl = len(m.BranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduledTime != nil {\n\t\tl = m.ScheduledTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif len(m.Queries) > 0 {\n\t\tfor k, v := range m.Queries {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 2 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.TotalHistoryBytes != 0 {\n\t\tn += 2 + sovService(uint64(m.TotalHistoryBytes))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.LoadBalancerHints != nil {\n\t\tl = m.LoadBalancerHints.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.AutoConfigHint != nil {\n\t\tl = m.AutoConfigHint.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollForActivityTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.PollerId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ForwardedFrom)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.IsolationGroup)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PollForActivityTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.TaskToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ActivityId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ActivityType != nil {\n\t\tl = m.ActivityType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Input != nil {\n\t\tl = m.Input.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduledTime != nil {\n\t\tl = m.ScheduledTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduleToCloseTimeout != nil {\n\t\tl = m.ScheduleToCloseTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartToCloseTimeout != nil {\n\t\tl = m.StartToCloseTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.HeartbeatTimeout != nil {\n\t\tl = m.HeartbeatTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\tl = m.ScheduledTimeOfThisAttempt.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\tl = m.HeartbeatDetails.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.WorkflowDomain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Header != nil {\n\t\tl = m.Header.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.LoadBalancerHints != nil {\n\t\tl = m.LoadBalancerHints.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.AutoConfigHint != nil {\n\t\tl = m.AutoConfigHint.Size()\n\t\tn += 2 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *AddDecisionTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduleId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduleId))\n\t}\n\tif m.ScheduleToStartTimeout != nil {\n\t\tl = m.ScheduleToStartTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Source != 0 {\n\t\tn += 1 + sovService(uint64(m.Source))\n\t}\n\tl = len(m.ForwardedFrom)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k, v := range m.PartitionConfig {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *AddDecisionTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *AddActivityTaskRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.SourceDomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ScheduleId != 0 {\n\t\tn += 1 + sovService(uint64(m.ScheduleId))\n\t}\n\tif m.ScheduleToStartTimeout != nil {\n\t\tl = m.ScheduleToStartTimeout.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Source != 0 {\n\t\tn += 1 + sovService(uint64(m.Source))\n\t}\n\tl = len(m.ForwardedFrom)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.ActivityTaskDispatchInfo != nil {\n\t\tl = m.ActivityTaskDispatchInfo.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.PartitionConfig) > 0 {\n\t\tfor k, v := range m.PartitionConfig {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ActivityTaskDispatchInfo) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ScheduledEvent != nil {\n\t\tl = m.ScheduledEvent.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.StartedTime != nil {\n\t\tl = m.StartedTime.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.Attempt != 0 {\n\t\tn += 1 + sovService(uint64(m.Attempt))\n\t}\n\tif m.ScheduledTimeOfThisAttempt != nil {\n\t\tl = m.ScheduledTimeOfThisAttempt.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.HeartbeatDetails != nil {\n\t\tl = m.HeartbeatDetails.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.WorkflowType != nil {\n\t\tl = m.WorkflowType.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.WorkflowDomain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *AddActivityTaskResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *QueryWorkflowRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ForwardedFrom)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *QueryWorkflowResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.QueryResult != nil {\n\t\tl = m.QueryResult.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.QueryRejected != nil {\n\t\tl = m.QueryRejected.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondQueryTaskCompletedRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.TaskId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RespondQueryTaskCompletedResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CancelOutstandingPollRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.PollerId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskListType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskListType))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *CancelOutstandingPollResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeTaskListRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Request != nil {\n\t\tl = m.Request.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *DescribeTaskListResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.Pollers) > 0 {\n\t\tfor _, e := range m.Pollers {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.TaskListStatus != nil {\n\t\tl = m.TaskListStatus.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ListTaskListPartitionsRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Domain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ListTaskListPartitionsResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ActivityTaskListPartitions) > 0 {\n\t\tfor _, e := range m.ActivityTaskListPartitions {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif len(m.DecisionTaskListPartitions) > 0 {\n\t\tfor _, e := range m.DecisionTaskListPartitions {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetTaskListsByDomainRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Domain)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetTaskListsByDomainResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.DecisionTaskListMap) > 0 {\n\t\tfor k, v := range m.DecisionTaskListMap {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif len(m.ActivityTaskListMap) > 0 {\n\t\tfor k, v := range m.ActivityTaskListMap {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovService(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *UpdateTaskListPartitionConfigRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskListType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskListType))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *UpdateTaskListPartitionConfigResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RefreshTaskListPartitionConfigRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskList != nil {\n\t\tl = m.TaskList.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.TaskListType != 0 {\n\t\tn += 1 + sovService(uint64(m.TaskListType))\n\t}\n\tif m.PartitionConfig != nil {\n\t\tl = m.PartitionConfig.Size()\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RefreshTaskListPartitionConfigResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovService(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozService(x uint64) (n int) {\n\treturn sovService(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *TaskListPartition) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListPartition: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListPartition: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IsolationGroups\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.IsolationGroups = append(m.IsolationGroups, string(dAtA[iNdEx:postIndex]))\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *TaskListPartitionConfig) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListPartitionConfig: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListPartitionConfig: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Version\", wireType)\n\t\t\t}\n\t\t\tm.Version = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Version |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NumReadPartitions\", wireType)\n\t\t\t}\n\t\t\tm.NumReadPartitions = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NumReadPartitions |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NumWritePartitions\", wireType)\n\t\t\t}\n\t\t\tm.NumWritePartitions = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NumWritePartitions |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ReadPartitions\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ReadPartitions == nil {\n\t\t\t\tm.ReadPartitions = make(map[int32]*TaskListPartition)\n\t\t\t}\n\t\t\tvar mapkey int32\n\t\t\tvar mapvalue *TaskListPartition\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapkey |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &TaskListPartition{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ReadPartitions[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WritePartitions\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WritePartitions == nil {\n\t\t\t\tm.WritePartitions = make(map[int32]*TaskListPartition)\n\t\t\t}\n\t\t\tvar mapkey int32\n\t\t\tvar mapvalue *TaskListPartition\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapkey |= int32(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &TaskListPartition{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.WritePartitions[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *LoadBalancerHints) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: LoadBalancerHints: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: LoadBalancerHints: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BacklogCount\", wireType)\n\t\t\t}\n\t\t\tm.BacklogCount = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.BacklogCount |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 1 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field RatePerSecond\", wireType)\n\t\t\t}\n\t\t\tvar v uint64\n\t\t\tif (iNdEx + 8) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tv = uint64(encoding_binary.LittleEndian.Uint64(dAtA[iNdEx:]))\n\t\t\tiNdEx += 8\n\t\t\tm.RatePerSecond = float64(math.Float64frombits(v))\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollForDecisionTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollForDecisionTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollForDecisionTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.PollForDecisionTaskRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PollerId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.PollerId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForwardedFrom\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ForwardedFrom = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IsolationGroup\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.IsolationGroup = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollForDecisionTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollForDecisionTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollForDecisionTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskToken = append(m.TaskToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.TaskToken == nil {\n\t\t\t\tm.TaskToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PreviousStartedEventId\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PreviousStartedEventId == nil {\n\t\t\t\tm.PreviousStartedEventId = &types.Int64Value{}\n\t\t\t}\n\t\t\tif err := m.PreviousStartedEventId.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedEventId\", wireType)\n\t\t\t}\n\t\t\tm.StartedEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.StartedEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 7:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field NextEventId\", wireType)\n\t\t\t}\n\t\t\tm.NextEventId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.NextEventId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 8:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BacklogCountHint\", wireType)\n\t\t\t}\n\t\t\tm.BacklogCountHint = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.BacklogCountHint |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 9:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StickyExecutionEnabled\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.StickyExecutionEnabled = bool(v != 0)\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Query\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Query == nil {\n\t\t\t\tm.Query = &v1.WorkflowQuery{}\n\t\t\t}\n\t\t\tif err := m.Query.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 11:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DecisionInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.DecisionInfo == nil {\n\t\t\t\tm.DecisionInfo = &v11.TransientDecisionInfo{}\n\t\t\t}\n\t\t\tif err := m.DecisionInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 12:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecutionTaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecutionTaskList == nil {\n\t\t\t\tm.WorkflowExecutionTaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecutionTaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 13:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field EventStoreVersion\", wireType)\n\t\t\t}\n\t\t\tm.EventStoreVersion = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.EventStoreVersion |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 14:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field BranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.BranchToken = append(m.BranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.BranchToken == nil {\n\t\t\t\tm.BranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 15:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTime == nil {\n\t\t\t\tm.ScheduledTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 16:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 17:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Queries\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Queries == nil {\n\t\t\t\tm.Queries = make(map[string]*v1.WorkflowQuery)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *v1.WorkflowQuery\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &v1.WorkflowQuery{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Queries[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 18:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TotalHistoryBytes\", wireType)\n\t\t\t}\n\t\t\tm.TotalHistoryBytes = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TotalHistoryBytes |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 19:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 20:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LoadBalancerHints\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.LoadBalancerHints == nil {\n\t\t\t\tm.LoadBalancerHints = &LoadBalancerHints{}\n\t\t\t}\n\t\t\tif err := m.LoadBalancerHints.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 21:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field AutoConfigHint\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.AutoConfigHint == nil {\n\t\t\t\tm.AutoConfigHint = &v1.AutoConfigHint{}\n\t\t\t}\n\t\t\tif err := m.AutoConfigHint.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollForActivityTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollForActivityTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollForActivityTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.PollForActivityTaskRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PollerId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.PollerId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForwardedFrom\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ForwardedFrom = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field IsolationGroup\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.IsolationGroup = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PollForActivityTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PollForActivityTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PollForActivityTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskToken = append(m.TaskToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.TaskToken == nil {\n\t\t\t\tm.TaskToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivityId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ActivityId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivityType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ActivityType == nil {\n\t\t\t\tm.ActivityType = &v1.ActivityType{}\n\t\t\t}\n\t\t\tif err := m.ActivityType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Input\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Input == nil {\n\t\t\t\tm.Input = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.Input.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTime == nil {\n\t\t\t\tm.ScheduledTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleToCloseTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduleToCloseTimeout == nil {\n\t\t\t\tm.ScheduleToCloseTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.ScheduleToCloseTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartToCloseTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartToCloseTimeout == nil {\n\t\t\t\tm.StartToCloseTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.StartToCloseTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HeartbeatTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.HeartbeatTimeout == nil {\n\t\t\t\tm.HeartbeatTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.HeartbeatTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 11:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 12:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTimeOfThisAttempt\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTimeOfThisAttempt == nil {\n\t\t\t\tm.ScheduledTimeOfThisAttempt = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTimeOfThisAttempt.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 13:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HeartbeatDetails\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.HeartbeatDetails == nil {\n\t\t\t\tm.HeartbeatDetails = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.HeartbeatDetails.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 14:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 15:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowDomain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.WorkflowDomain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 16:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Header\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Header == nil {\n\t\t\t\tm.Header = &v1.Header{}\n\t\t\t}\n\t\t\tif err := m.Header.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 17:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field LoadBalancerHints\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.LoadBalancerHints == nil {\n\t\t\t\tm.LoadBalancerHints = &LoadBalancerHints{}\n\t\t\t}\n\t\t\tif err := m.LoadBalancerHints.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 19:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 20:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field AutoConfigHint\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.AutoConfigHint == nil {\n\t\t\t\tm.AutoConfigHint = &v1.AutoConfigHint{}\n\t\t\t}\n\t\t\tif err := m.AutoConfigHint.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *AddDecisionTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: AddDecisionTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: AddDecisionTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduleId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduleId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleToStartTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduleToStartTimeout == nil {\n\t\t\t\tm.ScheduleToStartTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.ScheduleToStartTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Source\", wireType)\n\t\t\t}\n\t\t\tm.Source = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Source |= v11.TaskSource(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForwardedFrom\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ForwardedFrom = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.PartitionConfig[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *AddDecisionTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: AddDecisionTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: AddDecisionTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *AddActivityTaskRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: AddActivityTaskRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: AddActivityTaskRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field SourceDomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.SourceDomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleId\", wireType)\n\t\t\t}\n\t\t\tm.ScheduleId = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.ScheduleId |= int64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduleToStartTimeout\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduleToStartTimeout == nil {\n\t\t\t\tm.ScheduleToStartTimeout = &types.Duration{}\n\t\t\t}\n\t\t\tif err := m.ScheduleToStartTimeout.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Source\", wireType)\n\t\t\t}\n\t\t\tm.Source = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Source |= v11.TaskSource(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 8:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForwardedFrom\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ForwardedFrom = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 9:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivityTaskDispatchInfo\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ActivityTaskDispatchInfo == nil {\n\t\t\t\tm.ActivityTaskDispatchInfo = &ActivityTaskDispatchInfo{}\n\t\t\t}\n\t\t\tif err := m.ActivityTaskDispatchInfo.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 10:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.PartitionConfig[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ActivityTaskDispatchInfo) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ActivityTaskDispatchInfo: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ActivityTaskDispatchInfo: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledEvent == nil {\n\t\t\t\tm.ScheduledEvent = &v1.HistoryEvent{}\n\t\t\t}\n\t\t\tif err := m.ScheduledEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedTime\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedTime == nil {\n\t\t\t\tm.StartedTime = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.StartedTime.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Attempt\", wireType)\n\t\t\t}\n\t\t\tm.Attempt = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Attempt |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledTimeOfThisAttempt\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledTimeOfThisAttempt == nil {\n\t\t\t\tm.ScheduledTimeOfThisAttempt = &types.Timestamp{}\n\t\t\t}\n\t\t\tif err := m.ScheduledTimeOfThisAttempt.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field HeartbeatDetails\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.HeartbeatDetails == nil {\n\t\t\t\tm.HeartbeatDetails = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.HeartbeatDetails.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 6:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowType\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowType == nil {\n\t\t\t\tm.WorkflowType = &v1.WorkflowType{}\n\t\t\t}\n\t\t\tif err := m.WorkflowType.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 7:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowDomain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.WorkflowDomain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *AddActivityTaskResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: AddActivityTaskResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: AddActivityTaskResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *QueryWorkflowRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.QueryWorkflowRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ForwardedFrom\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ForwardedFrom = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *QueryWorkflowResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: QueryWorkflowResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field QueryResult\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.QueryResult == nil {\n\t\t\t\tm.QueryResult = &v1.Payload{}\n\t\t\t}\n\t\t\tif err := m.QueryResult.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field QueryRejected\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.QueryRejected == nil {\n\t\t\t\tm.QueryRejected = &v1.QueryRejected{}\n\t\t\t}\n\t\t\tif err := m.QueryRejected.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &v1.TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondQueryTaskCompletedRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondQueryTaskCompletedRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondQueryTaskCompletedRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.RespondQueryTaskCompletedRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RespondQueryTaskCompletedResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RespondQueryTaskCompletedResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RespondQueryTaskCompletedResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CancelOutstandingPollRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CancelOutstandingPollRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CancelOutstandingPollRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PollerId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.PollerId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskListType\", wireType)\n\t\t\t}\n\t\t\tm.TaskListType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskListType |= v1.TaskListType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *CancelOutstandingPollResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CancelOutstandingPollResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CancelOutstandingPollResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeTaskListRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeTaskListRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeTaskListRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Request\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Request == nil {\n\t\t\t\tm.Request = &v1.DescribeTaskListRequest{}\n\t\t\t}\n\t\t\tif err := m.Request.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *DescribeTaskListResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeTaskListResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: DescribeTaskListResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Pollers\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Pollers = append(m.Pollers, &v1.PollerInfo{})\n\t\t\tif err := m.Pollers[len(m.Pollers)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskListStatus\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskListStatus == nil {\n\t\t\t\tm.TaskListStatus = &v1.TaskListStatus{}\n\t\t\t}\n\t\t\tif err := m.TaskListStatus.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &v1.TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ListTaskListPartitionsRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ListTaskListPartitionsRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ListTaskListPartitionsRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Domain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Domain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ListTaskListPartitionsResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ListTaskListPartitionsResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ListTaskListPartitionsResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivityTaskListPartitions\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ActivityTaskListPartitions = append(m.ActivityTaskListPartitions, &v1.TaskListPartitionMetadata{})\n\t\t\tif err := m.ActivityTaskListPartitions[len(m.ActivityTaskListPartitions)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DecisionTaskListPartitions\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DecisionTaskListPartitions = append(m.DecisionTaskListPartitions, &v1.TaskListPartitionMetadata{})\n\t\t\tif err := m.DecisionTaskListPartitions[len(m.DecisionTaskListPartitions)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetTaskListsByDomainRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetTaskListsByDomainRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetTaskListsByDomainRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Domain\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Domain = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetTaskListsByDomainResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetTaskListsByDomainResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetTaskListsByDomainResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DecisionTaskListMap\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.DecisionTaskListMap == nil {\n\t\t\t\tm.DecisionTaskListMap = make(map[string]*DescribeTaskListResponse)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *DescribeTaskListResponse\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &DescribeTaskListResponse{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.DecisionTaskListMap[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ActivityTaskListMap\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ActivityTaskListMap == nil {\n\t\t\t\tm.ActivityTaskListMap = make(map[string]*DescribeTaskListResponse)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *DescribeTaskListResponse\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &DescribeTaskListResponse{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ActivityTaskListMap[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *UpdateTaskListPartitionConfigRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: UpdateTaskListPartitionConfigRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: UpdateTaskListPartitionConfigRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskListType\", wireType)\n\t\t\t}\n\t\t\tm.TaskListType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskListType |= v1.TaskListType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &v1.TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *UpdateTaskListPartitionConfigResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: UpdateTaskListPartitionConfigResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: UpdateTaskListPartitionConfigResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RefreshTaskListPartitionConfigRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshTaskListPartitionConfigRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshTaskListPartitionConfigRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskList\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.TaskList == nil {\n\t\t\t\tm.TaskList = &v1.TaskList{}\n\t\t\t}\n\t\t\tif err := m.TaskList.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskListType\", wireType)\n\t\t\t}\n\t\t\tm.TaskListType = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.TaskListType |= v1.TaskListType(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field PartitionConfig\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.PartitionConfig == nil {\n\t\t\t\tm.PartitionConfig = &v1.TaskListPartitionConfig{}\n\t\t\t}\n\t\t\tif err := m.PartitionConfig.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RefreshTaskListPartitionConfigResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshTaskListPartitionConfigResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RefreshTaskListPartitionConfigResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipService(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthService\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupService\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthService\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthService        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowService          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupService = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/matching/v1/service.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/matching/v1/service.proto\n\npackage matchingv1\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"reflect\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/gogo/protobuf/proto\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/api/x/restriction\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/encoding/protobuf/reflection\"\n)\n\nvar _ = ioutil.NopCloser\n\n// MatchingAPIYARPCClient is the YARPC client-side interface for the MatchingAPI service.\ntype MatchingAPIYARPCClient interface {\n\tPollForDecisionTask(context.Context, *PollForDecisionTaskRequest, ...yarpc.CallOption) (*PollForDecisionTaskResponse, error)\n\tPollForActivityTask(context.Context, *PollForActivityTaskRequest, ...yarpc.CallOption) (*PollForActivityTaskResponse, error)\n\tAddDecisionTask(context.Context, *AddDecisionTaskRequest, ...yarpc.CallOption) (*AddDecisionTaskResponse, error)\n\tAddActivityTask(context.Context, *AddActivityTaskRequest, ...yarpc.CallOption) (*AddActivityTaskResponse, error)\n\tQueryWorkflow(context.Context, *QueryWorkflowRequest, ...yarpc.CallOption) (*QueryWorkflowResponse, error)\n\tRespondQueryTaskCompleted(context.Context, *RespondQueryTaskCompletedRequest, ...yarpc.CallOption) (*RespondQueryTaskCompletedResponse, error)\n\tCancelOutstandingPoll(context.Context, *CancelOutstandingPollRequest, ...yarpc.CallOption) (*CancelOutstandingPollResponse, error)\n\tDescribeTaskList(context.Context, *DescribeTaskListRequest, ...yarpc.CallOption) (*DescribeTaskListResponse, error)\n\tListTaskListPartitions(context.Context, *ListTaskListPartitionsRequest, ...yarpc.CallOption) (*ListTaskListPartitionsResponse, error)\n\tGetTaskListsByDomain(context.Context, *GetTaskListsByDomainRequest, ...yarpc.CallOption) (*GetTaskListsByDomainResponse, error)\n\tUpdateTaskListPartitionConfig(context.Context, *UpdateTaskListPartitionConfigRequest, ...yarpc.CallOption) (*UpdateTaskListPartitionConfigResponse, error)\n\tRefreshTaskListPartitionConfig(context.Context, *RefreshTaskListPartitionConfigRequest, ...yarpc.CallOption) (*RefreshTaskListPartitionConfigResponse, error)\n}\n\nfunc newMatchingAPIYARPCClient(clientConfig transport.ClientConfig, anyResolver jsonpb.AnyResolver, options ...protobuf.ClientOption) MatchingAPIYARPCClient {\n\treturn &_MatchingAPIYARPCCaller{protobuf.NewStreamClient(\n\t\tprotobuf.ClientParams{\n\t\t\tServiceName:  \"uber.cadence.matching.v1.MatchingAPI\",\n\t\t\tClientConfig: clientConfig,\n\t\t\tAnyResolver:  anyResolver,\n\t\t\tOptions:      options,\n\t\t},\n\t)}\n}\n\n// NewMatchingAPIYARPCClient builds a new YARPC client for the MatchingAPI service.\nfunc NewMatchingAPIYARPCClient(clientConfig transport.ClientConfig, options ...protobuf.ClientOption) MatchingAPIYARPCClient {\n\treturn newMatchingAPIYARPCClient(clientConfig, nil, options...)\n}\n\n// MatchingAPIYARPCServer is the YARPC server-side interface for the MatchingAPI service.\ntype MatchingAPIYARPCServer interface {\n\tPollForDecisionTask(context.Context, *PollForDecisionTaskRequest) (*PollForDecisionTaskResponse, error)\n\tPollForActivityTask(context.Context, *PollForActivityTaskRequest) (*PollForActivityTaskResponse, error)\n\tAddDecisionTask(context.Context, *AddDecisionTaskRequest) (*AddDecisionTaskResponse, error)\n\tAddActivityTask(context.Context, *AddActivityTaskRequest) (*AddActivityTaskResponse, error)\n\tQueryWorkflow(context.Context, *QueryWorkflowRequest) (*QueryWorkflowResponse, error)\n\tRespondQueryTaskCompleted(context.Context, *RespondQueryTaskCompletedRequest) (*RespondQueryTaskCompletedResponse, error)\n\tCancelOutstandingPoll(context.Context, *CancelOutstandingPollRequest) (*CancelOutstandingPollResponse, error)\n\tDescribeTaskList(context.Context, *DescribeTaskListRequest) (*DescribeTaskListResponse, error)\n\tListTaskListPartitions(context.Context, *ListTaskListPartitionsRequest) (*ListTaskListPartitionsResponse, error)\n\tGetTaskListsByDomain(context.Context, *GetTaskListsByDomainRequest) (*GetTaskListsByDomainResponse, error)\n\tUpdateTaskListPartitionConfig(context.Context, *UpdateTaskListPartitionConfigRequest) (*UpdateTaskListPartitionConfigResponse, error)\n\tRefreshTaskListPartitionConfig(context.Context, *RefreshTaskListPartitionConfigRequest) (*RefreshTaskListPartitionConfigResponse, error)\n}\n\ntype buildMatchingAPIYARPCProceduresParams struct {\n\tServer      MatchingAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver\n}\n\nfunc buildMatchingAPIYARPCProcedures(params buildMatchingAPIYARPCProceduresParams) []transport.Procedure {\n\thandler := &_MatchingAPIYARPCHandler{params.Server}\n\treturn protobuf.BuildProcedures(\n\t\tprotobuf.BuildProceduresParams{\n\t\t\tServiceName: \"uber.cadence.matching.v1.MatchingAPI\",\n\t\t\tUnaryHandlerParams: []protobuf.BuildProceduresUnaryHandlerParams{\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"PollForDecisionTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.PollForDecisionTask,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServicePollForDecisionTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"PollForActivityTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.PollForActivityTask,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServicePollForActivityTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"AddDecisionTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.AddDecisionTask,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceAddDecisionTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"AddActivityTask\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.AddActivityTask,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceAddActivityTaskYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"QueryWorkflow\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.QueryWorkflow,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceQueryWorkflowYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RespondQueryTaskCompleted\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RespondQueryTaskCompleted,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceRespondQueryTaskCompletedYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"CancelOutstandingPoll\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.CancelOutstandingPoll,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceCancelOutstandingPollYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"DescribeTaskList\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.DescribeTaskList,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceDescribeTaskListYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"ListTaskListPartitions\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.ListTaskListPartitions,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceListTaskListPartitionsYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetTaskListsByDomain\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetTaskListsByDomain,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceGetTaskListsByDomainYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"UpdateTaskListPartitionConfig\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.UpdateTaskListPartitionConfig,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceUpdateTaskListPartitionConfigYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"RefreshTaskListPartitionConfig\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.RefreshTaskListPartitionConfig,\n\t\t\t\t\t\t\tNewRequest:  newMatchingAPIServiceRefreshTaskListPartitionConfigYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tOnewayHandlerParams: []protobuf.BuildProceduresOnewayHandlerParams{},\n\t\t\tStreamHandlerParams: []protobuf.BuildProceduresStreamHandlerParams{},\n\t\t},\n\t)\n}\n\n// BuildMatchingAPIYARPCProcedures prepares an implementation of the MatchingAPI service for YARPC registration.\nfunc BuildMatchingAPIYARPCProcedures(server MatchingAPIYARPCServer) []transport.Procedure {\n\treturn buildMatchingAPIYARPCProcedures(buildMatchingAPIYARPCProceduresParams{Server: server})\n}\n\n// FxMatchingAPIYARPCClientParams defines the input\n// for NewFxMatchingAPIYARPCClient. It provides the\n// paramaters to get a MatchingAPIYARPCClient in an\n// Fx application.\ntype FxMatchingAPIYARPCClientParams struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tAnyResolver jsonpb.AnyResolver  `name:\"yarpcfx\" optional:\"true\"`\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// FxMatchingAPIYARPCClientResult defines the output\n// of NewFxMatchingAPIYARPCClient. It provides a\n// MatchingAPIYARPCClient to an Fx application.\ntype FxMatchingAPIYARPCClientResult struct {\n\tfx.Out\n\n\tClient MatchingAPIYARPCClient\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// NewFxMatchingAPIYARPCClient provides a MatchingAPIYARPCClient\n// to an Fx application using the given name for routing.\n//\n//\tfx.Provide(\n//\t  matchingv1.NewFxMatchingAPIYARPCClient(\"service-name\"),\n//\t  ...\n//\t)\nfunc NewFxMatchingAPIYARPCClient(name string, options ...protobuf.ClientOption) interface{} {\n\treturn func(params FxMatchingAPIYARPCClientParams) FxMatchingAPIYARPCClientResult {\n\t\tcc := params.Provider.ClientConfig(name)\n\n\t\tif params.Restriction != nil {\n\t\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok {\n\t\t\t\tif err := params.Restriction.Check(protobuf.Encoding, namer.TransportName()); err != nil {\n\t\t\t\t\tpanic(err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn FxMatchingAPIYARPCClientResult{\n\t\t\tClient: newMatchingAPIYARPCClient(cc, params.AnyResolver, options...),\n\t\t}\n\t}\n}\n\n// FxMatchingAPIYARPCProceduresParams defines the input\n// for NewFxMatchingAPIYARPCProcedures. It provides the\n// paramaters to get MatchingAPIYARPCServer procedures in an\n// Fx application.\ntype FxMatchingAPIYARPCProceduresParams struct {\n\tfx.In\n\n\tServer      MatchingAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver `name:\"yarpcfx\" optional:\"true\"`\n}\n\n// FxMatchingAPIYARPCProceduresResult defines the output\n// of NewFxMatchingAPIYARPCProcedures. It provides\n// MatchingAPIYARPCServer procedures to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group.\n// Dig 1.2 or newer must be used for this feature to work.\ntype FxMatchingAPIYARPCProceduresResult struct {\n\tfx.Out\n\n\tProcedures     []transport.Procedure `group:\"yarpcfx\"`\n\tReflectionMeta reflection.ServerMeta `group:\"yarpcfx\"`\n}\n\n// NewFxMatchingAPIYARPCProcedures provides MatchingAPIYARPCServer procedures to an Fx application.\n// It expects a MatchingAPIYARPCServer to be present in the container.\n//\n//\tfx.Provide(\n//\t  matchingv1.NewFxMatchingAPIYARPCProcedures(),\n//\t  ...\n//\t)\nfunc NewFxMatchingAPIYARPCProcedures() interface{} {\n\treturn func(params FxMatchingAPIYARPCProceduresParams) FxMatchingAPIYARPCProceduresResult {\n\t\treturn FxMatchingAPIYARPCProceduresResult{\n\t\t\tProcedures: buildMatchingAPIYARPCProcedures(buildMatchingAPIYARPCProceduresParams{\n\t\t\t\tServer:      params.Server,\n\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t}),\n\t\t\tReflectionMeta: MatchingAPIReflectionMeta,\n\t\t}\n\t}\n}\n\n// MatchingAPIReflectionMeta is the reflection server metadata\n// required for using the gRPC reflection protocol with YARPC.\n//\n// See https://github.com/grpc/grpc/blob/master/doc/server-reflection.md.\nvar MatchingAPIReflectionMeta = reflection.ServerMeta{\n\tServiceName:     \"uber.cadence.matching.v1.MatchingAPI\",\n\tFileDescriptors: yarpcFileDescriptorClosure826e827d3aabf7fc,\n}\n\ntype _MatchingAPIYARPCCaller struct {\n\tstreamClient protobuf.StreamClient\n}\n\nfunc (c *_MatchingAPIYARPCCaller) PollForDecisionTask(ctx context.Context, request *PollForDecisionTaskRequest, options ...yarpc.CallOption) (*PollForDecisionTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"PollForDecisionTask\", request, newMatchingAPIServicePollForDecisionTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*PollForDecisionTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServicePollForDecisionTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) PollForActivityTask(ctx context.Context, request *PollForActivityTaskRequest, options ...yarpc.CallOption) (*PollForActivityTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"PollForActivityTask\", request, newMatchingAPIServicePollForActivityTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*PollForActivityTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServicePollForActivityTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) AddDecisionTask(ctx context.Context, request *AddDecisionTaskRequest, options ...yarpc.CallOption) (*AddDecisionTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"AddDecisionTask\", request, newMatchingAPIServiceAddDecisionTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*AddDecisionTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceAddDecisionTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) AddActivityTask(ctx context.Context, request *AddActivityTaskRequest, options ...yarpc.CallOption) (*AddActivityTaskResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"AddActivityTask\", request, newMatchingAPIServiceAddActivityTaskYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*AddActivityTaskResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceAddActivityTaskYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) QueryWorkflow(ctx context.Context, request *QueryWorkflowRequest, options ...yarpc.CallOption) (*QueryWorkflowResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"QueryWorkflow\", request, newMatchingAPIServiceQueryWorkflowYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*QueryWorkflowResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceQueryWorkflowYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) RespondQueryTaskCompleted(ctx context.Context, request *RespondQueryTaskCompletedRequest, options ...yarpc.CallOption) (*RespondQueryTaskCompletedResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RespondQueryTaskCompleted\", request, newMatchingAPIServiceRespondQueryTaskCompletedYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RespondQueryTaskCompletedResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceRespondQueryTaskCompletedYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) CancelOutstandingPoll(ctx context.Context, request *CancelOutstandingPollRequest, options ...yarpc.CallOption) (*CancelOutstandingPollResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"CancelOutstandingPoll\", request, newMatchingAPIServiceCancelOutstandingPollYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*CancelOutstandingPollResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceCancelOutstandingPollYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) DescribeTaskList(ctx context.Context, request *DescribeTaskListRequest, options ...yarpc.CallOption) (*DescribeTaskListResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"DescribeTaskList\", request, newMatchingAPIServiceDescribeTaskListYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*DescribeTaskListResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceDescribeTaskListYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) ListTaskListPartitions(ctx context.Context, request *ListTaskListPartitionsRequest, options ...yarpc.CallOption) (*ListTaskListPartitionsResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"ListTaskListPartitions\", request, newMatchingAPIServiceListTaskListPartitionsYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*ListTaskListPartitionsResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceListTaskListPartitionsYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) GetTaskListsByDomain(ctx context.Context, request *GetTaskListsByDomainRequest, options ...yarpc.CallOption) (*GetTaskListsByDomainResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetTaskListsByDomain\", request, newMatchingAPIServiceGetTaskListsByDomainYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetTaskListsByDomainResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceGetTaskListsByDomainYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) UpdateTaskListPartitionConfig(ctx context.Context, request *UpdateTaskListPartitionConfigRequest, options ...yarpc.CallOption) (*UpdateTaskListPartitionConfigResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"UpdateTaskListPartitionConfig\", request, newMatchingAPIServiceUpdateTaskListPartitionConfigYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*UpdateTaskListPartitionConfigResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceUpdateTaskListPartitionConfigYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_MatchingAPIYARPCCaller) RefreshTaskListPartitionConfig(ctx context.Context, request *RefreshTaskListPartitionConfigRequest, options ...yarpc.CallOption) (*RefreshTaskListPartitionConfigResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"RefreshTaskListPartitionConfig\", request, newMatchingAPIServiceRefreshTaskListPartitionConfigYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*RefreshTaskListPartitionConfigResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceRefreshTaskListPartitionConfigYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\ntype _MatchingAPIYARPCHandler struct {\n\tserver MatchingAPIYARPCServer\n}\n\nfunc (h *_MatchingAPIYARPCHandler) PollForDecisionTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *PollForDecisionTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*PollForDecisionTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServicePollForDecisionTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.PollForDecisionTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) PollForActivityTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *PollForActivityTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*PollForActivityTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServicePollForActivityTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.PollForActivityTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) AddDecisionTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *AddDecisionTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*AddDecisionTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceAddDecisionTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.AddDecisionTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) AddActivityTask(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *AddActivityTaskRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*AddActivityTaskRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceAddActivityTaskYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.AddActivityTask(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) QueryWorkflow(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *QueryWorkflowRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*QueryWorkflowRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceQueryWorkflowYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.QueryWorkflow(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) RespondQueryTaskCompleted(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RespondQueryTaskCompletedRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RespondQueryTaskCompletedRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceRespondQueryTaskCompletedYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RespondQueryTaskCompleted(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) CancelOutstandingPoll(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *CancelOutstandingPollRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*CancelOutstandingPollRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceCancelOutstandingPollYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.CancelOutstandingPoll(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) DescribeTaskList(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *DescribeTaskListRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*DescribeTaskListRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceDescribeTaskListYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.DescribeTaskList(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) ListTaskListPartitions(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *ListTaskListPartitionsRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*ListTaskListPartitionsRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceListTaskListPartitionsYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.ListTaskListPartitions(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) GetTaskListsByDomain(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetTaskListsByDomainRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetTaskListsByDomainRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceGetTaskListsByDomainYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetTaskListsByDomain(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) UpdateTaskListPartitionConfig(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *UpdateTaskListPartitionConfigRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*UpdateTaskListPartitionConfigRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceUpdateTaskListPartitionConfigYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.UpdateTaskListPartitionConfig(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_MatchingAPIYARPCHandler) RefreshTaskListPartitionConfig(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *RefreshTaskListPartitionConfigRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*RefreshTaskListPartitionConfigRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyMatchingAPIServiceRefreshTaskListPartitionConfigYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.RefreshTaskListPartitionConfig(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc newMatchingAPIServicePollForDecisionTaskYARPCRequest() proto.Message {\n\treturn &PollForDecisionTaskRequest{}\n}\n\nfunc newMatchingAPIServicePollForDecisionTaskYARPCResponse() proto.Message {\n\treturn &PollForDecisionTaskResponse{}\n}\n\nfunc newMatchingAPIServicePollForActivityTaskYARPCRequest() proto.Message {\n\treturn &PollForActivityTaskRequest{}\n}\n\nfunc newMatchingAPIServicePollForActivityTaskYARPCResponse() proto.Message {\n\treturn &PollForActivityTaskResponse{}\n}\n\nfunc newMatchingAPIServiceAddDecisionTaskYARPCRequest() proto.Message {\n\treturn &AddDecisionTaskRequest{}\n}\n\nfunc newMatchingAPIServiceAddDecisionTaskYARPCResponse() proto.Message {\n\treturn &AddDecisionTaskResponse{}\n}\n\nfunc newMatchingAPIServiceAddActivityTaskYARPCRequest() proto.Message {\n\treturn &AddActivityTaskRequest{}\n}\n\nfunc newMatchingAPIServiceAddActivityTaskYARPCResponse() proto.Message {\n\treturn &AddActivityTaskResponse{}\n}\n\nfunc newMatchingAPIServiceQueryWorkflowYARPCRequest() proto.Message {\n\treturn &QueryWorkflowRequest{}\n}\n\nfunc newMatchingAPIServiceQueryWorkflowYARPCResponse() proto.Message {\n\treturn &QueryWorkflowResponse{}\n}\n\nfunc newMatchingAPIServiceRespondQueryTaskCompletedYARPCRequest() proto.Message {\n\treturn &RespondQueryTaskCompletedRequest{}\n}\n\nfunc newMatchingAPIServiceRespondQueryTaskCompletedYARPCResponse() proto.Message {\n\treturn &RespondQueryTaskCompletedResponse{}\n}\n\nfunc newMatchingAPIServiceCancelOutstandingPollYARPCRequest() proto.Message {\n\treturn &CancelOutstandingPollRequest{}\n}\n\nfunc newMatchingAPIServiceCancelOutstandingPollYARPCResponse() proto.Message {\n\treturn &CancelOutstandingPollResponse{}\n}\n\nfunc newMatchingAPIServiceDescribeTaskListYARPCRequest() proto.Message {\n\treturn &DescribeTaskListRequest{}\n}\n\nfunc newMatchingAPIServiceDescribeTaskListYARPCResponse() proto.Message {\n\treturn &DescribeTaskListResponse{}\n}\n\nfunc newMatchingAPIServiceListTaskListPartitionsYARPCRequest() proto.Message {\n\treturn &ListTaskListPartitionsRequest{}\n}\n\nfunc newMatchingAPIServiceListTaskListPartitionsYARPCResponse() proto.Message {\n\treturn &ListTaskListPartitionsResponse{}\n}\n\nfunc newMatchingAPIServiceGetTaskListsByDomainYARPCRequest() proto.Message {\n\treturn &GetTaskListsByDomainRequest{}\n}\n\nfunc newMatchingAPIServiceGetTaskListsByDomainYARPCResponse() proto.Message {\n\treturn &GetTaskListsByDomainResponse{}\n}\n\nfunc newMatchingAPIServiceUpdateTaskListPartitionConfigYARPCRequest() proto.Message {\n\treturn &UpdateTaskListPartitionConfigRequest{}\n}\n\nfunc newMatchingAPIServiceUpdateTaskListPartitionConfigYARPCResponse() proto.Message {\n\treturn &UpdateTaskListPartitionConfigResponse{}\n}\n\nfunc newMatchingAPIServiceRefreshTaskListPartitionConfigYARPCRequest() proto.Message {\n\treturn &RefreshTaskListPartitionConfigRequest{}\n}\n\nfunc newMatchingAPIServiceRefreshTaskListPartitionConfigYARPCResponse() proto.Message {\n\treturn &RefreshTaskListPartitionConfigResponse{}\n}\n\nvar (\n\temptyMatchingAPIServicePollForDecisionTaskYARPCRequest             = &PollForDecisionTaskRequest{}\n\temptyMatchingAPIServicePollForDecisionTaskYARPCResponse            = &PollForDecisionTaskResponse{}\n\temptyMatchingAPIServicePollForActivityTaskYARPCRequest             = &PollForActivityTaskRequest{}\n\temptyMatchingAPIServicePollForActivityTaskYARPCResponse            = &PollForActivityTaskResponse{}\n\temptyMatchingAPIServiceAddDecisionTaskYARPCRequest                 = &AddDecisionTaskRequest{}\n\temptyMatchingAPIServiceAddDecisionTaskYARPCResponse                = &AddDecisionTaskResponse{}\n\temptyMatchingAPIServiceAddActivityTaskYARPCRequest                 = &AddActivityTaskRequest{}\n\temptyMatchingAPIServiceAddActivityTaskYARPCResponse                = &AddActivityTaskResponse{}\n\temptyMatchingAPIServiceQueryWorkflowYARPCRequest                   = &QueryWorkflowRequest{}\n\temptyMatchingAPIServiceQueryWorkflowYARPCResponse                  = &QueryWorkflowResponse{}\n\temptyMatchingAPIServiceRespondQueryTaskCompletedYARPCRequest       = &RespondQueryTaskCompletedRequest{}\n\temptyMatchingAPIServiceRespondQueryTaskCompletedYARPCResponse      = &RespondQueryTaskCompletedResponse{}\n\temptyMatchingAPIServiceCancelOutstandingPollYARPCRequest           = &CancelOutstandingPollRequest{}\n\temptyMatchingAPIServiceCancelOutstandingPollYARPCResponse          = &CancelOutstandingPollResponse{}\n\temptyMatchingAPIServiceDescribeTaskListYARPCRequest                = &DescribeTaskListRequest{}\n\temptyMatchingAPIServiceDescribeTaskListYARPCResponse               = &DescribeTaskListResponse{}\n\temptyMatchingAPIServiceListTaskListPartitionsYARPCRequest          = &ListTaskListPartitionsRequest{}\n\temptyMatchingAPIServiceListTaskListPartitionsYARPCResponse         = &ListTaskListPartitionsResponse{}\n\temptyMatchingAPIServiceGetTaskListsByDomainYARPCRequest            = &GetTaskListsByDomainRequest{}\n\temptyMatchingAPIServiceGetTaskListsByDomainYARPCResponse           = &GetTaskListsByDomainResponse{}\n\temptyMatchingAPIServiceUpdateTaskListPartitionConfigYARPCRequest   = &UpdateTaskListPartitionConfigRequest{}\n\temptyMatchingAPIServiceUpdateTaskListPartitionConfigYARPCResponse  = &UpdateTaskListPartitionConfigResponse{}\n\temptyMatchingAPIServiceRefreshTaskListPartitionConfigYARPCRequest  = &RefreshTaskListPartitionConfigRequest{}\n\temptyMatchingAPIServiceRefreshTaskListPartitionConfigYARPCResponse = &RefreshTaskListPartitionConfigResponse{}\n)\n\nvar yarpcFileDescriptorClosure826e827d3aabf7fc = [][]byte{\n\t// uber/cadence/matching/v1/service.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x5a, 0x49, 0x73, 0xdc, 0xc6,\n\t\t0xf5, 0x2f, 0x0c, 0x39, 0x5c, 0xde, 0x90, 0x43, 0xb2, 0x49, 0x53, 0xd0, 0x48, 0x94, 0xa8, 0x91,\n\t\t0x25, 0xd3, 0xff, 0xbf, 0x33, 0x34, 0x69, 0x49, 0x91, 0xa5, 0x8a, 0x15, 0x2e, 0xa2, 0x34, 0x29,\n\t\t0x2b, 0x92, 0x21, 0x5a, 0xaa, 0x4a, 0x5c, 0x42, 0x9a, 0x83, 0x26, 0x07, 0x21, 0x06, 0x80, 0x80,\n\t\t0x06, 0x69, 0xfa, 0x90, 0x43, 0x2a, 0x49, 0xa5, 0x2a, 0xd7, 0xe4, 0x9e, 0xed, 0x4b, 0xe4, 0x92,\n\t\t0xcf, 0x91, 0x2a, 0x57, 0x0e, 0x39, 0xe4, 0x03, 0x24, 0x55, 0xb9, 0xe5, 0x90, 0xea, 0x05, 0x33,\n\t\t0x00, 0xa6, 0x31, 0x0b, 0x49, 0xc9, 0x39, 0xe4, 0x46, 0x74, 0xbf, 0xad, 0xdf, 0xf6, 0x7b, 0xdd,\n\t\t0x43, 0xb8, 0x19, 0xed, 0x91, 0x60, 0xb5, 0x81, 0x2d, 0xe2, 0x36, 0xc8, 0x6a, 0x0b, 0xd3, 0x46,\n\t\t0xd3, 0x76, 0x0f, 0x56, 0x8f, 0xd6, 0x56, 0x43, 0x12, 0x1c, 0xd9, 0x0d, 0x52, 0xf3, 0x03, 0x8f,\n\t\t0x7a, 0x48, 0x67, 0x74, 0x35, 0x49, 0x57, 0x8b, 0xe9, 0x6a, 0x47, 0x6b, 0x95, 0x2b, 0x07, 0x9e,\n\t\t0x77, 0xe0, 0x90, 0x55, 0x4e, 0xb7, 0x17, 0xed, 0xaf, 0x5a, 0x51, 0x80, 0xa9, 0xed, 0xb9, 0x82,\n\t\t0xb3, 0x72, 0x35, 0xbb, 0x4f, 0xed, 0x16, 0x09, 0x29, 0x6e, 0xf9, 0x92, 0xa0, 0x4b, 0xc0, 0x71,\n\t\t0x80, 0x7d, 0x9f, 0x04, 0xa1, 0xdc, 0x5f, 0x4e, 0x99, 0x88, 0x7d, 0x9b, 0x59, 0xd7, 0xf0, 0x5a,\n\t\t0xad, 0x8e, 0x0a, 0x15, 0xc5, 0xeb, 0x88, 0x04, 0x27, 0x92, 0xa0, 0xaa, 0x22, 0xa0, 0x38, 0x3c,\n\t\t0x74, 0xec, 0x90, 0x4a, 0x9a, 0x15, 0x15, 0x8d, 0x74, 0x82, 0x79, 0xec, 0x05, 0x87, 0x24, 0x90,\n\t\t0x94, 0xff, 0xd7, 0x8f, 0x72, 0xdf, 0xf1, 0x8e, 0x25, 0xed, 0x35, 0x15, 0x6d, 0xd3, 0x0e, 0xa9,\n\t\t0xd7, 0x36, 0xee, 0xdd, 0x14, 0x49, 0xd8, 0xc4, 0x01, 0xb1, 0xba, 0xa9, 0x6e, 0xe4, 0x50, 0xa5,\n\t\t0x4f, 0x51, 0xfd, 0x04, 0xe6, 0x76, 0x71, 0x78, 0xf8, 0xa9, 0x1d, 0xd2, 0x67, 0x38, 0xa0, 0x36,\n\t\t0x0b, 0x04, 0x7a, 0x1f, 0x66, 0xed, 0xd0, 0x73, 0x78, 0x54, 0xcc, 0x83, 0xc0, 0x8b, 0xfc, 0x50,\n\t\t0xd7, 0x96, 0x47, 0x56, 0x26, 0x8d, 0x99, 0xf6, 0xfa, 0x23, 0xbe, 0x5c, 0xfd, 0xdb, 0x28, 0x5c,\n\t\t0xe8, 0x12, 0xb0, 0xe5, 0xb9, 0xfb, 0xf6, 0x01, 0xd2, 0x61, 0xfc, 0x88, 0x04, 0xa1, 0xed, 0xb9,\n\t\t0xba, 0xb6, 0xac, 0xad, 0x8c, 0x18, 0xf1, 0x27, 0x5a, 0x87, 0x79, 0x37, 0x6a, 0x99, 0x01, 0xc1,\n\t\t0x96, 0xe9, 0xc7, 0x5c, 0xa1, 0x5e, 0x58, 0xd6, 0x56, 0x8a, 0x9b, 0x05, 0x5d, 0x33, 0xe6, 0xdc,\n\t\t0xa8, 0x65, 0x10, 0x6c, 0xb5, 0x45, 0x86, 0xe8, 0x16, 0x2c, 0x30, 0x9e, 0xe3, 0xc0, 0xa6, 0x24,\n\t\t0xc9, 0x34, 0xd2, 0x66, 0x42, 0x6e, 0xd4, 0x7a, 0xc9, 0xb6, 0x13, 0x5c, 0x2e, 0xcc, 0x64, 0xb5,\n\t\t0x8c, 0x2e, 0x8f, 0xac, 0x94, 0xd6, 0x1f, 0xd6, 0xf2, 0x32, 0xb4, 0x96, 0x73, 0x9e, 0x5a, 0xda,\n\t\t0xa0, 0x87, 0x2e, 0x0d, 0x4e, 0x8c, 0x72, 0x90, 0xb6, 0xf2, 0x35, 0xcc, 0x76, 0x59, 0x58, 0xe4,\n\t\t0x0a, 0x77, 0x86, 0x57, 0x98, 0x39, 0x8c, 0xd0, 0x38, 0x73, 0x9c, 0x5e, 0xad, 0xb8, 0x30, 0xaf,\n\t\t0xb0, 0x0c, 0xcd, 0xc2, 0xc8, 0x21, 0x39, 0xe1, 0x9e, 0x2f, 0x1a, 0xec, 0x4f, 0xb4, 0x01, 0xc5,\n\t\t0x23, 0xec, 0x44, 0x84, 0xfb, 0xb9, 0xb4, 0xfe, 0xff, 0x43, 0x18, 0x64, 0x08, 0xce, 0x7b, 0x85,\n\t\t0xbb, 0x5a, 0xc5, 0x83, 0x05, 0x95, 0x61, 0x6f, 0x4c, 0x61, 0xf5, 0x47, 0x30, 0xf7, 0xa9, 0x87,\n\t\t0xad, 0x4d, 0xec, 0x60, 0xb7, 0x41, 0x82, 0xc7, 0xb6, 0x4b, 0x43, 0x74, 0x1d, 0xa6, 0xf7, 0x70,\n\t\t0xe3, 0xd0, 0xf1, 0x0e, 0xcc, 0x86, 0x17, 0xb9, 0x54, 0xa6, 0xd8, 0x94, 0x5c, 0xdc, 0x62, 0x6b,\n\t\t0xe8, 0x26, 0xcc, 0x04, 0x98, 0x05, 0x83, 0x04, 0x66, 0x48, 0x1a, 0x9e, 0x6b, 0x71, 0x53, 0x34,\n\t\t0x63, 0x9a, 0x2d, 0x3f, 0x23, 0xc1, 0x73, 0xbe, 0x58, 0xfd, 0x87, 0x06, 0x95, 0x67, 0x9e, 0xe3,\n\t\t0xec, 0x78, 0xc1, 0x36, 0x69, 0xd8, 0x2c, 0x47, 0x99, 0x45, 0x06, 0x79, 0x1d, 0x91, 0x90, 0xa2,\n\t\t0x3a, 0x8c, 0x07, 0xe2, 0x4f, 0xae, 0xa5, 0xb4, 0xbe, 0x9a, 0x3e, 0x09, 0xf6, 0x6d, 0x76, 0x88,\n\t\t0x7c, 0x09, 0x46, 0xcc, 0x8f, 0x2e, 0xc1, 0xa4, 0xe5, 0xb5, 0xb0, 0xed, 0x9a, 0xb6, 0xb0, 0x65,\n\t\t0xd2, 0x98, 0x10, 0x0b, 0x75, 0x8b, 0x6d, 0xfa, 0x9e, 0xe3, 0x90, 0x80, 0x6d, 0x8e, 0x88, 0x4d,\n\t\t0xb1, 0x50, 0xb7, 0xd0, 0x0d, 0x28, 0xef, 0x7b, 0xc1, 0x31, 0x0e, 0x2c, 0x62, 0x99, 0xfb, 0x81,\n\t\t0xd7, 0xd2, 0x47, 0x39, 0xc5, 0x74, 0x7b, 0x75, 0x27, 0xf0, 0x5a, 0xe8, 0x3d, 0x98, 0xc9, 0xd4,\n\t\t0xae, 0x5e, 0xe4, 0x74, 0xe5, 0x74, 0xe9, 0x56, 0xff, 0x5c, 0x82, 0x4b, 0x4a, 0x8b, 0x43, 0xdf,\n\t\t0x73, 0x43, 0x82, 0x96, 0x00, 0x58, 0xaf, 0x30, 0xa9, 0x77, 0x48, 0x44, 0x01, 0x4f, 0x19, 0x93,\n\t\t0x6c, 0x65, 0x97, 0x2d, 0xa0, 0xcf, 0x01, 0xc5, 0xad, 0xcb, 0x24, 0x5f, 0x92, 0x46, 0xc4, 0x24,\n\t\t0xcb, 0x40, 0xdf, 0x54, 0xba, 0xe7, 0xa5, 0x24, 0x7f, 0x18, 0x53, 0x1b, 0x73, 0xc7, 0xd9, 0x25,\n\t\t0xb4, 0x03, 0xd3, 0x6d, 0xb1, 0xf4, 0xc4, 0x27, 0xdc, 0x0d, 0xa5, 0xf5, 0x6b, 0x3d, 0x25, 0xee,\n\t\t0x9e, 0xf8, 0xc4, 0x98, 0x3a, 0x4e, 0x7c, 0xa1, 0x17, 0x70, 0xd1, 0x0f, 0xc8, 0x91, 0xed, 0x45,\n\t\t0xa1, 0x19, 0x52, 0x1c, 0x50, 0x62, 0x99, 0xe4, 0x88, 0xb8, 0x94, 0xb9, 0x76, 0x94, 0xcb, 0xbc,\n\t\t0x54, 0x13, 0x40, 0x52, 0x8b, 0x81, 0xa4, 0x56, 0x77, 0xe9, 0x9d, 0x5b, 0x2f, 0x58, 0xde, 0x19,\n\t\t0x8b, 0x31, 0xf7, 0x73, 0xc1, 0xfc, 0x90, 0xf1, 0xd6, 0x2d, 0xb4, 0x02, 0xb3, 0x5d, 0xe2, 0x8a,\n\t\t0x3c, 0xf3, 0xca, 0x61, 0x9a, 0x52, 0x87, 0x71, 0x4c, 0x29, 0x69, 0xf9, 0x54, 0x1f, 0xe3, 0x25,\n\t\t0x11, 0x7f, 0xa2, 0x2a, 0x4c, 0xbb, 0xe4, 0x4b, 0xda, 0x11, 0x30, 0xce, 0x05, 0x94, 0xd8, 0x62,\n\t\t0xcc, 0xfd, 0x01, 0xa0, 0x54, 0x7a, 0x9b, 0x4d, 0xdb, 0xa5, 0xfa, 0x04, 0x27, 0x9c, 0x4d, 0xe6,\n\t\t0x38, 0xab, 0x06, 0x74, 0x17, 0xf4, 0x90, 0xda, 0x8d, 0xc3, 0x93, 0x4e, 0x28, 0x4c, 0xe2, 0xe2,\n\t\t0x3d, 0x87, 0x58, 0xfa, 0xe4, 0xb2, 0xb6, 0x32, 0x61, 0x2c, 0x8a, 0xfd, 0xb6, 0xa3, 0x1f, 0x8a,\n\t\t0x5d, 0x74, 0x17, 0x8a, 0x1c, 0xf8, 0x74, 0xe0, 0x3e, 0xa9, 0xf6, 0xf4, 0xf3, 0x67, 0x8c, 0xd2,\n\t\t0x10, 0x0c, 0xc8, 0x80, 0x69, 0x4b, 0xe6, 0x8d, 0x69, 0xbb, 0xfb, 0x9e, 0x5e, 0xe2, 0x12, 0xbe,\n\t\t0x95, 0x96, 0x20, 0x80, 0x87, 0x97, 0x78, 0x80, 0xdd, 0xd0, 0x26, 0x2e, 0x8d, 0xb3, 0xad, 0xee,\n\t\t0xee, 0x7b, 0xc6, 0x94, 0x95, 0xf8, 0x42, 0xaf, 0xe0, 0x72, 0x77, 0x52, 0x99, 0x3c, 0x0d, 0x19,\n\t\t0x66, 0xe9, 0x53, 0x5c, 0xc5, 0x92, 0xd2, 0xc8, 0xb8, 0x85, 0x18, 0x17, 0xbb, 0xb2, 0x2a, 0xde,\n\t\t0x42, 0x35, 0x98, 0x17, 0x4e, 0x67, 0x48, 0x49, 0xcc, 0x18, 0x9d, 0xa6, 0x79, 0x7c, 0xe6, 0xf8,\n\t\t0xd6, 0x73, 0xb6, 0xf3, 0x42, 0xe2, 0xd4, 0x35, 0x98, 0xda, 0x0b, 0xb0, 0xdb, 0x68, 0xca, 0x2a,\n\t\t0x28, 0xf3, 0x2a, 0x28, 0x89, 0x35, 0x51, 0x07, 0x1b, 0x50, 0x0e, 0x1b, 0x4d, 0x62, 0x45, 0x0e,\n\t\t0xb1, 0x4c, 0x36, 0xaa, 0xe8, 0x33, 0xdc, 0xc8, 0x4a, 0x57, 0x76, 0xed, 0xc6, 0x73, 0x8c, 0x31,\n\t\t0xdd, 0xe6, 0x60, 0x6b, 0xe8, 0x3b, 0x30, 0x15, 0xe7, 0x14, 0x17, 0x30, 0xdb, 0x57, 0x40, 0x49,\n\t\t0xd2, 0x73, 0xf6, 0x2f, 0x60, 0x9c, 0x45, 0xc4, 0x26, 0xa1, 0x3e, 0xc7, 0x91, 0x66, 0x33, 0xbf,\n\t\t0xcf, 0xf6, 0x28, 0xf8, 0xda, 0x67, 0x42, 0x88, 0x40, 0x99, 0x58, 0x24, 0x73, 0x19, 0xf5, 0x28,\n\t\t0x76, 0x4c, 0x39, 0x5e, 0x98, 0x7b, 0x27, 0x94, 0x84, 0x3a, 0xe2, 0x99, 0x38, 0xc7, 0xb7, 0x1e,\n\t\t0x8b, 0x9d, 0x4d, 0xb6, 0x81, 0xbe, 0x80, 0xd9, 0x36, 0xf4, 0x99, 0x0d, 0x8e, 0x63, 0xfa, 0x3c,\n\t\t0x3f, 0xd0, 0xda, 0xd0, 0x00, 0x68, 0xcc, 0xf8, 0x99, 0x91, 0xe2, 0x87, 0x30, 0xef, 0x78, 0xd8,\n\t\t0x32, 0xf7, 0x24, 0x16, 0xf0, 0xb2, 0x08, 0xf5, 0x85, 0x7e, 0xf8, 0xd2, 0x85, 0x1f, 0xc6, 0x9c,\n\t\t0xd3, 0x05, 0x29, 0x4f, 0x60, 0x16, 0x47, 0xd4, 0x93, 0x56, 0x8b, 0x8a, 0x7b, 0x87, 0x4b, 0xbe,\n\t\t0xae, 0xcc, 0xb8, 0x8d, 0x88, 0x7a, 0xc2, 0x2e, 0xc6, 0x6f, 0x94, 0x71, 0xea, 0xbb, 0xf2, 0x0a,\n\t\t0xa6, 0x92, 0x2e, 0x4d, 0xe2, 0xe3, 0xa4, 0xc0, 0xc7, 0xbb, 0x69, 0x7c, 0x1c, 0xa8, 0xf8, 0x3a,\n\t\t0xb0, 0x98, 0x00, 0xad, 0x8d, 0x06, 0xb5, 0x8f, 0x6c, 0x7a, 0x72, 0x7a, 0xd0, 0x52, 0x48, 0xf8,\n\t\t0x6f, 0x04, 0xad, 0xdf, 0x40, 0x1b, 0xb4, 0xd2, 0x16, 0x7f, 0xa3, 0xa0, 0x75, 0x15, 0x4a, 0x58,\n\t\t0x5a, 0xd3, 0x71, 0x02, 0xc4, 0x4b, 0x75, 0x8b, 0xa1, 0x5a, 0x9b, 0x80, 0xa3, 0xda, 0x68, 0x0f,\n\t\t0x54, 0x6b, 0x1f, 0x8c, 0xa3, 0x1a, 0x4e, 0x7c, 0xa1, 0x75, 0x28, 0xda, 0xae, 0x1f, 0x51, 0xee,\n\t\t0x9d, 0xd2, 0xfa, 0x65, 0x75, 0x44, 0xf1, 0x09, 0xcb, 0x6d, 0x43, 0x90, 0x2a, 0x1a, 0xd4, 0xd8,\n\t\t0x59, 0x1b, 0xd4, 0xf8, 0x70, 0x0d, 0x6a, 0x17, 0x2e, 0xc6, 0xf2, 0x4c, 0x56, 0x5e, 0x8e, 0x17,\n\t\t0x12, 0x2e, 0xc8, 0x8b, 0x04, 0xa4, 0x95, 0xd6, 0x2f, 0x76, 0xc9, 0xda, 0x96, 0xb7, 0x42, 0x63,\n\t\t0x31, 0xe6, 0xdd, 0xf5, 0xb6, 0x18, 0xe7, 0xae, 0x60, 0x44, 0xdf, 0x87, 0x45, 0xae, 0xa4, 0x5b,\n\t\t0xe4, 0x64, 0x3f, 0x91, 0xf3, 0x9c, 0x31, 0x23, 0x6f, 0x07, 0xe6, 0x9a, 0x04, 0x07, 0x74, 0x8f,\n\t\t0x60, 0xda, 0x16, 0x05, 0xfd, 0x44, 0xcd, 0xb6, 0x79, 0x62, 0x39, 0x09, 0xdc, 0x2f, 0xa5, 0x71,\n\t\t0xff, 0x15, 0x5c, 0x49, 0x47, 0xc2, 0xf4, 0xf6, 0x4d, 0xda, 0xb4, 0x43, 0x33, 0x66, 0x98, 0xea,\n\t\t0xeb, 0xd8, 0x4a, 0x2a, 0x32, 0x4f, 0xf7, 0x77, 0x9b, 0x76, 0xb8, 0x21, 0xe5, 0xd7, 0x93, 0x27,\n\t\t0xb0, 0x08, 0xc5, 0xb6, 0x13, 0x72, 0x6c, 0xeb, 0x97, 0x29, 0x9d, 0x43, 0x6c, 0x0b, 0xae, 0xee,\n\t\t0x31, 0xac, 0x7c, 0xba, 0x31, 0xec, 0x3d, 0x98, 0x69, 0xcb, 0x11, 0x1d, 0x83, 0xc3, 0xe3, 0xa4,\n\t\t0x51, 0x8e, 0x97, 0xb7, 0xf9, 0x2a, 0xfa, 0x08, 0xc6, 0x9a, 0x04, 0x5b, 0x24, 0x90, 0xe8, 0x77,\n\t\t0x49, 0xa9, 0xe9, 0x31, 0x27, 0x31, 0x24, 0x69, 0x1e, 0x1a, 0xcc, 0x9d, 0x0b, 0x1a, 0xbc, 0x59,\n\t\t0x20, 0x53, 0x61, 0xcd, 0xc2, 0xa9, 0xb1, 0xa6, 0xfa, 0x97, 0x51, 0x58, 0xdc, 0xb0, 0x2c, 0xd5,\n\t\t0xe5, 0x25, 0xd5, 0xbc, 0xb5, 0x4c, 0xf3, 0x7e, 0x43, 0x0d, 0xf1, 0x1e, 0x4c, 0x76, 0x86, 0xb6,\n\t\t0x91, 0x41, 0x86, 0xb6, 0x09, 0x1a, 0xcf, 0x68, 0x57, 0xa1, 0xd4, 0xee, 0x16, 0x72, 0x56, 0x1f,\n\t\t0x31, 0x20, 0x5e, 0xaa, 0x5b, 0xd9, 0x76, 0x22, 0x9b, 0x80, 0x2c, 0xd8, 0xe2, 0x10, 0xed, 0x84,\n\t\t0x8f, 0xf6, 0x71, 0xd9, 0xde, 0x83, 0xb1, 0xd0, 0x8b, 0x82, 0x86, 0x68, 0x8f, 0xe5, 0x2c, 0x18,\n\t\t0x27, 0xe6, 0x58, 0x1c, 0x1e, 0x3e, 0xe7, 0x94, 0x86, 0xe4, 0x50, 0xa0, 0xdc, 0xb8, 0x0a, 0xe5,\n\t\t0x7c, 0x45, 0x46, 0x4d, 0xf4, 0x7b, 0x8c, 0x50, 0x47, 0xb5, 0x96, 0x49, 0x30, 0xf9, 0x34, 0x90,\n\t\t0xc9, 0xb2, 0xca, 0x26, 0x2c, 0xa8, 0x08, 0x15, 0xa3, 0xc8, 0x42, 0x72, 0x14, 0x99, 0x4c, 0x8e,\n\t\t0x19, 0xc7, 0x70, 0xa1, 0xcb, 0x06, 0x89, 0xb6, 0xaa, 0x12, 0xd1, 0xce, 0xab, 0x44, 0xaa, 0xff,\n\t\t0x2c, 0xf2, 0x9c, 0x56, 0xcd, 0x36, 0xdf, 0x44, 0x4e, 0xb3, 0x9b, 0x1f, 0x0f, 0xb7, 0xd9, 0x51,\n\t\t0x2d, 0x90, 0xbe, 0x2c, 0xd6, 0xb7, 0x63, 0x03, 0x52, 0xd9, 0x3f, 0x7a, 0xa6, 0xec, 0x2f, 0x0e,\n\t\t0x97, 0xfd, 0x63, 0x67, 0xcf, 0xfe, 0xf1, 0x73, 0xc8, 0xfe, 0x09, 0x55, 0xf6, 0xbb, 0xa0, 0xe3,\n\t\t0x44, 0x28, 0xb7, 0xed, 0xd0, 0x67, 0x59, 0xc1, 0xee, 0x7d, 0x12, 0xb1, 0xd7, 0x7b, 0x54, 0x41,\n\t\t0x0e, 0xa7, 0x91, 0x2b, 0x53, 0x59, 0x6d, 0x30, 0x40, 0xb5, 0x29, 0xf2, 0xed, 0x2d, 0x56, 0xdb,\n\t\t0xd7, 0x23, 0xa0, 0xe7, 0x1d, 0x16, 0x7d, 0x0f, 0x66, 0x3a, 0x03, 0x04, 0xbf, 0xad, 0xca, 0x72,\n\t\t0x53, 0xe3, 0xb2, 0xbc, 0x97, 0xf1, 0x27, 0x05, 0xa3, 0x33, 0x04, 0xf2, 0xef, 0xae, 0x99, 0xae,\n\t\t0x30, 0xdc, 0x4c, 0x97, 0x98, 0x72, 0x46, 0x86, 0x9d, 0x72, 0x46, 0xcf, 0x7f, 0xca, 0x29, 0x9e,\n\t\t0xcf, 0x94, 0x33, 0x76, 0x6e, 0x53, 0xce, 0xb8, 0x6a, 0xca, 0x91, 0xbd, 0x54, 0x79, 0x73, 0x79,\n\t\t0xb3, 0xbd, 0xf4, 0x6b, 0x0d, 0x16, 0xf8, 0x05, 0x32, 0x3e, 0x45, 0xdc, 0x49, 0xb7, 0xb2, 0xb7,\n\t\t0xc4, 0xf7, 0x95, 0x87, 0x57, 0xf1, 0x0e, 0x78, 0x3f, 0x3c, 0xcb, 0x2c, 0x30, 0xd8, 0xf5, 0xb1,\n\t\t0xfa, 0x6f, 0x0d, 0xde, 0xc9, 0x58, 0x28, 0xbd, 0xfa, 0x00, 0xa6, 0xf8, 0x6b, 0x95, 0x19, 0x90,\n\t\t0x30, 0x72, 0xe2, 0x33, 0xf6, 0xce, 0x93, 0x12, 0xe7, 0x30, 0x38, 0x03, 0xaa, 0x43, 0x39, 0x16,\n\t\t0xf0, 0x63, 0xd2, 0xa0, 0xc4, 0xea, 0x79, 0x57, 0x17, 0x77, 0x74, 0x49, 0x69, 0x4c, 0xbf, 0x4e,\n\t\t0x7e, 0xa2, 0x97, 0x8a, 0x08, 0x0b, 0x7f, 0x7c, 0xd0, 0xd3, 0x1f, 0x7d, 0x83, 0xfb, 0x77, 0x0d,\n\t\t0x96, 0xc5, 0x89, 0x2d, 0x6e, 0x00, 0x63, 0xdc, 0xf2, 0x5a, 0xbe, 0x43, 0x98, 0x15, 0x32, 0x46,\n\t\t0x4f, 0xb3, 0x81, 0xbe, 0xad, 0x54, 0xda, 0x4f, 0xce, 0x5b, 0x08, 0xfa, 0x05, 0x18, 0xe7, 0xbc,\n\t\t0x72, 0xf8, 0x9b, 0x34, 0xc6, 0xd8, 0x67, 0xdd, 0xaa, 0x5e, 0x87, 0x6b, 0x3d, 0xcc, 0x13, 0x11,\n\t\t0xaf, 0xfe, 0x55, 0x83, 0xcb, 0x5b, 0x6c, 0x8c, 0x77, 0x9e, 0x46, 0x34, 0xa4, 0xd8, 0xb5, 0x6c,\n\t\t0xf7, 0xe0, 0x99, 0xe7, 0x38, 0x03, 0xcd, 0x0e, 0xa9, 0xc7, 0x8c, 0x42, 0xe6, 0x31, 0xe3, 0x11,\n\t\t0x94, 0xdb, 0x87, 0xea, 0x3c, 0x4e, 0x97, 0x73, 0xfa, 0x45, 0x7c, 0x32, 0xd1, 0x2f, 0x68, 0xe2,\n\t\t0xeb, 0x2c, 0x03, 0x42, 0xf5, 0x2a, 0x2c, 0xe5, 0x1c, 0x4f, 0x3a, 0xe0, 0x27, 0x70, 0x61, 0x9b,\n\t\t0x84, 0x8d, 0xc0, 0xde, 0x23, 0x6d, 0x76, 0x79, 0xf4, 0x9d, 0x6c, 0x0e, 0xa8, 0x13, 0x2f, 0x87,\n\t\t0x7d, 0xb0, 0xd0, 0x57, 0xff, 0x54, 0x00, 0xbd, 0x5b, 0x82, 0xac, 0xc7, 0x8f, 0x61, 0x5c, 0xb8,\n\t\t0x53, 0xfc, 0xa0, 0x58, 0x5a, 0xbf, 0x9a, 0xfb, 0x28, 0x45, 0x02, 0x0e, 0xf0, 0x31, 0x3d, 0xbb,\n\t\t0x31, 0x75, 0xbc, 0x1f, 0x52, 0x4c, 0xa3, 0x50, 0xd6, 0xe2, 0xf5, 0x9e, 0xbe, 0x7b, 0xce, 0x49,\n\t\t0x8d, 0x32, 0x4d, 0x7d, 0xbf, 0xb1, 0x6a, 0x3c, 0x53, 0x70, 0x43, 0x58, 0xe2, 0x49, 0x92, 0xd5,\n\t\t0x15, 0xc6, 0x11, 0x5c, 0x84, 0x31, 0x09, 0x30, 0x22, 0x73, 0xe5, 0x57, 0x5a, 0x69, 0x61, 0x38,\n\t\t0xa5, 0xbf, 0x28, 0xc0, 0x95, 0x3c, 0xad, 0x32, 0x6c, 0xaf, 0x61, 0xa9, 0xf3, 0x7e, 0xd5, 0x0e,\n\t\t0x42, 0xe2, 0x27, 0x4e, 0x11, 0xcc, 0xda, 0x60, 0x9e, 0x7b, 0x42, 0x28, 0xb6, 0x30, 0xc5, 0x46,\n\t\t0x25, 0x39, 0xbc, 0xa5, 0x55, 0x33, 0x95, 0xed, 0x9f, 0x17, 0x94, 0x2a, 0x0b, 0xa7, 0x53, 0x69,\n\t\t0x25, 0x2e, 0x32, 0x69, 0x95, 0xd5, 0xdb, 0x70, 0xe9, 0x11, 0x69, 0xbb, 0x21, 0xdc, 0x3c, 0x11,\n\t\t0xa8, 0xdd, 0xc7, 0xf7, 0xd5, 0x3f, 0x8e, 0xc2, 0x65, 0x35, 0x9f, 0xf4, 0xde, 0xcf, 0x34, 0x58,\n\t\t0x54, 0x9c, 0xa5, 0x85, 0x7d, 0xe9, 0xb7, 0xa7, 0xf9, 0x08, 0xdf, 0x4b, 0x70, 0x6d, 0x3b, 0x73,\n\t\t0x96, 0x27, 0xd8, 0x17, 0xa3, 0xe9, 0xbc, 0xd5, 0xbd, 0xc3, 0xcd, 0x50, 0x44, 0x91, 0x99, 0x51,\n\t\t0x38, 0x93, 0x19, 0x1b, 0x99, 0x28, 0x76, 0xcc, 0xc0, 0xdd, 0x3b, 0x95, 0xaf, 0x58, 0x7b, 0x50,\n\t\t0xdb, 0xad, 0x98, 0x94, 0x1f, 0xa7, 0x9f, 0xc8, 0x7b, 0x5c, 0x11, 0xf2, 0x7a, 0x4e, 0xf2, 0xa7,\n\t\t0xeb, 0xaf, 0xd2, 0xc3, 0xf5, 0xdb, 0xd4, 0x5d, 0xfd, 0x5d, 0x01, 0xde, 0xfd, 0xdc, 0xb7, 0x30,\n\t\t0x25, 0x79, 0xad, 0x64, 0x10, 0x80, 0x3a, 0x43, 0xa1, 0x9f, 0x1f, 0x7e, 0xa9, 0x7a, 0xe7, 0xe8,\n\t\t0x79, 0x4c, 0x32, 0xef, 0xc1, 0x8d, 0x3e, 0x2e, 0x92, 0x20, 0xf7, 0xfb, 0x02, 0xdc, 0x30, 0xc8,\n\t\t0x7e, 0x40, 0xc2, 0xe6, 0xff, 0xbc, 0x99, 0xe7, 0xcd, 0x15, 0xb8, 0xd9, 0xcf, 0x47, 0xc2, 0x9d,\n\t\t0xeb, 0xff, 0x9a, 0x82, 0xd2, 0x13, 0x99, 0xcf, 0x1b, 0xcf, 0xea, 0xe8, 0xa7, 0x1a, 0xcc, 0x2b,\n\t\t0x7e, 0x2a, 0x44, 0xb7, 0x86, 0xfc, 0x65, 0x91, 0x87, 0xa0, 0x72, 0xfb, 0x54, 0xbf, 0x47, 0x26,\n\t\t0x8d, 0x48, 0x16, 0xed, 0x00, 0x46, 0x28, 0xae, 0xf0, 0x03, 0x18, 0xa1, 0xbc, 0x96, 0x1d, 0xc1,\n\t\t0x4c, 0xe6, 0xf5, 0x0b, 0x7d, 0x38, 0xec, 0x63, 0x5d, 0x65, 0x6d, 0x08, 0x8e, 0x94, 0xde, 0xd4,\n\t\t0xb9, 0x3f, 0x1c, 0xf6, 0xd9, 0xa2, 0x8f, 0x5e, 0xe5, 0x79, 0x7d, 0x98, 0x4e, 0xdd, 0xa4, 0x50,\n\t\t0x2d, 0x5f, 0x86, 0xea, 0x52, 0x58, 0x59, 0x1d, 0x98, 0x5e, 0x6a, 0xfc, 0xb5, 0x06, 0x17, 0x73,\n\t\t0xc7, 0x7a, 0x74, 0x2f, 0x5f, 0x5c, 0xbf, 0xab, 0x4a, 0xe5, 0xfe, 0xa9, 0x78, 0xa5, 0x59, 0xbf,\n\t\t0xd4, 0xe0, 0x1d, 0xe5, 0xa0, 0x8d, 0xee, 0xe4, 0x8b, 0xed, 0x75, 0xf1, 0xa8, 0x7c, 0x7b, 0x68,\n\t\t0x3e, 0x69, 0xca, 0x09, 0xcc, 0x66, 0x01, 0x06, 0xad, 0x0d, 0x03, 0x46, 0x42, 0xff, 0x29, 0xf0,\n\t\t0x0b, 0xfd, 0x4a, 0x83, 0x45, 0xf5, 0x6c, 0x88, 0x7a, 0x1c, 0xa7, 0xe7, 0x0c, 0x5b, 0xb9, 0x3b,\n\t\t0x3c, 0xa3, 0xb4, 0xe6, 0xe7, 0x1a, 0x2c, 0xa8, 0x26, 0x11, 0x74, 0x7b, 0xd8, 0xc9, 0x45, 0x58,\n\t\t0x72, 0xe7, 0x74, 0x03, 0x0f, 0xfa, 0xad, 0x06, 0x4b, 0x3d, 0x71, 0x0a, 0x7d, 0x92, 0x2f, 0x79,\n\t\t0x90, 0x19, 0xa0, 0xf2, 0xe0, 0xd4, 0xfc, 0xd2, 0xc4, 0x3f, 0x68, 0x70, 0xa5, 0x77, 0xf3, 0x47,\n\t\t0x0f, 0x7a, 0x95, 0xc7, 0x00, 0xd0, 0x5a, 0xf9, 0xee, 0xe9, 0x05, 0x08, 0x2b, 0x37, 0xef, 0xff,\n\t\t0xe0, 0xe3, 0x03, 0x9b, 0x36, 0xa3, 0xbd, 0x5a, 0xc3, 0x6b, 0xad, 0xa6, 0xfe, 0x63, 0xb5, 0x76,\n\t\t0x40, 0x5c, 0xf1, 0x2f, 0xbe, 0xc9, 0xff, 0x32, 0xbe, 0x1f, 0xff, 0x7d, 0xb4, 0xb6, 0x37, 0xc6,\n\t\t0x77, 0x3f, 0xfa, 0x4f, 0x00, 0x00, 0x00, 0xff, 0xff, 0x7a, 0xe2, 0xa5, 0xb6, 0x93, 0x2c, 0x00,\n\t\t0x00,\n\t},\n\t// google/protobuf/duration.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x4f, 0x29, 0x2d, 0x4a,\n\t\t0x2c, 0xc9, 0xcc, 0xcf, 0xd3, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0x56,\n\t\t0x5c, 0x1c, 0x2e, 0x50, 0x25, 0x42, 0x12, 0x5c, 0xec, 0xc5, 0xa9, 0xc9, 0xf9, 0x79, 0x29, 0xc5,\n\t\t0x12, 0x8c, 0x0a, 0x8c, 0x1a, 0xcc, 0x41, 0x30, 0xae, 0x90, 0x08, 0x17, 0x6b, 0x5e, 0x62, 0x5e,\n\t\t0x7e, 0xb1, 0x04, 0x93, 0x02, 0xa3, 0x06, 0x6b, 0x10, 0x84, 0xe3, 0xd4, 0xcc, 0xc8, 0x25, 0x9c,\n\t\t0x9c, 0x9f, 0xab, 0x87, 0x66, 0xa6, 0x13, 0x2f, 0xcc, 0xc4, 0x00, 0x90, 0x48, 0x00, 0x63, 0x94,\n\t\t0x21, 0x54, 0x45, 0x7a, 0x7e, 0x4e, 0x62, 0x5e, 0xba, 0x5e, 0x7e, 0x51, 0x3a, 0xc2, 0x81, 0x25,\n\t\t0x95, 0x05, 0xa9, 0xc5, 0xfa, 0xd9, 0x79, 0xf9, 0xe5, 0x79, 0x70, 0xc7, 0x16, 0x24, 0xfd, 0x60,\n\t\t0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce, 0x1d, 0xa2, 0x39, 0x00, 0xaa,\n\t\t0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4, 0x35, 0x89, 0x0d, 0x6c, 0x94,\n\t\t0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0xef, 0x8a, 0xb4, 0xc3, 0xfb, 0x00, 0x00, 0x00,\n\t},\n\t// google/protobuf/timestamp.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4f, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0xc9, 0xcc, 0x4d,\n\t\t0x2d, 0x2e, 0x49, 0xcc, 0x2d, 0xd0, 0x03, 0x0b, 0x09, 0xf1, 0x43, 0x14, 0xe8, 0xc1, 0x14, 0x28,\n\t\t0x59, 0x73, 0x71, 0x86, 0xc0, 0xd4, 0x08, 0x49, 0x70, 0xb1, 0x17, 0xa7, 0x26, 0xe7, 0xe7, 0xa5,\n\t\t0x14, 0x4b, 0x30, 0x2a, 0x30, 0x6a, 0x30, 0x07, 0xc1, 0xb8, 0x42, 0x22, 0x5c, 0xac, 0x79, 0x89,\n\t\t0x79, 0xf9, 0xc5, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0xac, 0x41, 0x10, 0x8e, 0x53, 0x2b, 0x23, 0x97,\n\t\t0x70, 0x72, 0x7e, 0xae, 0x1e, 0x9a, 0xa1, 0x4e, 0x7c, 0x70, 0x23, 0x03, 0x40, 0x42, 0x01, 0x8c,\n\t\t0x51, 0x46, 0x50, 0x25, 0xe9, 0xf9, 0x39, 0x89, 0x79, 0xe9, 0x7a, 0xf9, 0x45, 0xe9, 0x48, 0x6e,\n\t\t0xac, 0x2c, 0x48, 0x2d, 0xd6, 0xcf, 0xce, 0xcb, 0x2f, 0xcf, 0x43, 0xb8, 0xb7, 0x20, 0xe9, 0x07,\n\t\t0x23, 0xe3, 0x22, 0x26, 0x66, 0xf7, 0x00, 0xa7, 0x55, 0x4c, 0x72, 0xee, 0x10, 0xdd, 0x01, 0x50,\n\t\t0x2d, 0x7a, 0xe1, 0xa9, 0x39, 0x39, 0xde, 0x20, 0x0d, 0x21, 0x20, 0xbd, 0x49, 0x6c, 0x60, 0xb3,\n\t\t0x8c, 0x01, 0x01, 0x00, 0x00, 0xff, 0xff, 0xae, 0x65, 0xce, 0x7d, 0xff, 0x00, 0x00, 0x00,\n\t},\n\t// google/protobuf/wrappers.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0x2f, 0x4a, 0x2c,\n\t\t0x28, 0x48, 0x2d, 0x2a, 0xd6, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0xca,\n\t\t0x5c, 0xdc, 0x2e, 0xf9, 0xa5, 0x49, 0x39, 0xa9, 0x61, 0x89, 0x39, 0xa5, 0xa9, 0x42, 0x22, 0x5c,\n\t\t0xac, 0x65, 0x20, 0x86, 0x04, 0xa3, 0x02, 0xa3, 0x06, 0x63, 0x10, 0x84, 0xa3, 0xa4, 0xc4, 0xc5,\n\t\t0xe5, 0x96, 0x93, 0x9f, 0x58, 0x82, 0x45, 0x0d, 0x13, 0x92, 0x1a, 0xcf, 0xbc, 0x12, 0x33, 0x13,\n\t\t0x2c, 0x6a, 0x98, 0x61, 0x6a, 0x94, 0xb9, 0xb8, 0x43, 0x71, 0x29, 0x62, 0x41, 0x35, 0xc8, 0xd8,\n\t\t0x08, 0x8b, 0x1a, 0x56, 0x34, 0x83, 0xb0, 0x2a, 0xe2, 0x85, 0x29, 0x52, 0xe4, 0xe2, 0x74, 0xca,\n\t\t0xcf, 0xcf, 0xc1, 0xa2, 0x84, 0x03, 0xc9, 0x9c, 0xe0, 0x92, 0xa2, 0xcc, 0xbc, 0x74, 0x2c, 0x8a,\n\t\t0x38, 0x91, 0x1c, 0xe4, 0x54, 0x59, 0x92, 0x5a, 0x8c, 0x45, 0x0d, 0x0f, 0x54, 0x8d, 0x53, 0x33,\n\t\t0x23, 0x97, 0x70, 0x72, 0x7e, 0xae, 0x1e, 0x5a, 0xf0, 0x3a, 0xf1, 0x86, 0x43, 0xc3, 0x3f, 0x00,\n\t\t0x24, 0x12, 0xc0, 0x18, 0x65, 0x08, 0x55, 0x91, 0x9e, 0x9f, 0x93, 0x98, 0x97, 0xae, 0x97, 0x5f,\n\t\t0x94, 0x8e, 0x88, 0xab, 0x92, 0xca, 0x82, 0xd4, 0x62, 0xfd, 0xec, 0xbc, 0xfc, 0xf2, 0x3c, 0x78,\n\t\t0xbc, 0x15, 0x24, 0xfd, 0x60, 0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce,\n\t\t0x1d, 0xa2, 0x39, 0x00, 0xaa, 0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4,\n\t\t0x35, 0x89, 0x0d, 0x6c, 0x94, 0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0x3c, 0x92, 0x48, 0x30, 0x06,\n\t\t0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/common.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x57, 0xdd, 0x72, 0x22, 0xc7,\n\t\t0x15, 0xf6, 0xc0, 0xa2, 0x9f, 0x03, 0xbb, 0x42, 0xad, 0xfd, 0x61, 0xb5, 0x5e, 0xaf, 0x16, 0x97,\n\t\t0x63, 0x79, 0x2b, 0x86, 0x88, 0x4d, 0x52, 0x2e, 0x3b, 0x4e, 0x82, 0xd0, 0x48, 0x9a, 0x5d, 0x02,\n\t\t0xa4, 0x99, 0x95, 0xac, 0xa4, 0xca, 0x53, 0xcd, 0x4c, 0x83, 0x3b, 0x0c, 0xd3, 0x93, 0x99, 0x1e,\n\t\t0x56, 0xf8, 0x22, 0x95, 0xcb, 0xe4, 0x26, 0x8f, 0x90, 0x8b, 0xbc, 0x48, 0x1e, 0x20, 0x97, 0x79,\n\t\t0x97, 0x5c, 0xa7, 0xba, 0xa7, 0x07, 0x81, 0xc2, 0x1a, 0x5f, 0xa4, 0x7c, 0x47, 0x9f, 0xf3, 0x7d,\n\t\t0xa7, 0xbf, 0xd3, 0xdd, 0xe7, 0x1c, 0x06, 0x0e, 0x92, 0x01, 0x8d, 0xea, 0x2e, 0xf1, 0x68, 0xe0,\n\t\t0xd2, 0x3a, 0x09, 0x59, 0x7d, 0x7a, 0x54, 0x77, 0xf9, 0x64, 0xc2, 0x83, 0x5a, 0x18, 0x71, 0xc1,\n\t\t0xd1, 0x9e, 0x44, 0xd4, 0x34, 0xa2, 0x46, 0x42, 0x56, 0x9b, 0x1e, 0xed, 0x7f, 0x30, 0xe2, 0x7c,\n\t\t0xe4, 0xd3, 0xba, 0x82, 0x0c, 0x92, 0x61, 0xdd, 0x4b, 0x22, 0x22, 0x58, 0x46, 0xaa, 0xbe, 0x86,\n\t\t0xdd, 0x4b, 0x1e, 0x8d, 0x87, 0x3e, 0x7f, 0x6b, 0x5e, 0x53, 0x37, 0x91, 0x2e, 0xf4, 0x0c, 0x8a,\n\t\t0x6f, 0xb5, 0xd1, 0x61, 0x5e, 0xc5, 0x38, 0x30, 0x0e, 0xb7, 0x31, 0x64, 0x26, 0xcb, 0x43, 0x0f,\n\t\t0x60, 0x23, 0x4a, 0x02, 0xe9, 0xcb, 0x29, 0x5f, 0x21, 0x4a, 0x02, 0xcb, 0xab, 0x56, 0xa1, 0x94,\n\t\t0x05, 0xb3, 0x67, 0x21, 0x45, 0x08, 0xee, 0x04, 0x64, 0x42, 0x75, 0x00, 0xf5, 0x5b, 0x62, 0x9a,\n\t\t0xae, 0x60, 0x53, 0x26, 0x66, 0xef, 0xc4, 0x3c, 0x85, 0xcd, 0x1e, 0x99, 0xf9, 0x9c, 0x78, 0xd2,\n\t\t0xed, 0x11, 0x41, 0x94, 0xbb, 0x84, 0xd5, 0xef, 0xea, 0x17, 0xb0, 0x79, 0x4a, 0x98, 0x9f, 0x44,\n\t\t0x14, 0x3d, 0x84, 0x8d, 0x88, 0x92, 0x98, 0x07, 0x9a, 0xaf, 0x57, 0xa8, 0x02, 0x9b, 0x1e, 0x15,\n\t\t0x84, 0xf9, 0xb1, 0x52, 0x58, 0xc2, 0xd9, 0xb2, 0xfa, 0x77, 0x03, 0xee, 0xfc, 0x86, 0x4e, 0x38,\n\t\t0xfa, 0x12, 0x36, 0x86, 0x8c, 0xfa, 0x5e, 0x5c, 0x31, 0x0e, 0xf2, 0x87, 0xc5, 0xc6, 0x47, 0xb5,\n\t\t0x15, 0xe7, 0x57, 0x93, 0xd0, 0xda, 0xa9, 0xc2, 0x99, 0x81, 0x88, 0x66, 0x58, 0x93, 0xf6, 0x2f,\n\t\t0xa1, 0xb8, 0x60, 0x46, 0x65, 0xc8, 0x8f, 0xe9, 0x4c, 0xab, 0x90, 0x3f, 0x51, 0x03, 0x0a, 0x53,\n\t\t0xe2, 0x27, 0x54, 0x09, 0x28, 0x36, 0xde, 0x5f, 0x19, 0x5e, 0xa7, 0x89, 0x53, 0xe8, 0xe7, 0xb9,\n\t\t0xcf, 0x8c, 0xea, 0x3f, 0x0c, 0xd8, 0x38, 0xa7, 0xc4, 0xa3, 0x11, 0xfa, 0xd5, 0x2d, 0x89, 0x1f,\n\t\t0xaf, 0x8c, 0x91, 0x82, 0x7f, 0x58, 0x91, 0xff, 0x36, 0xa0, 0xdc, 0xa7, 0x24, 0x72, 0xbf, 0x69,\n\t\t0x0a, 0x11, 0xb1, 0x41, 0x22, 0x68, 0x8c, 0x1c, 0xb8, 0xc7, 0x02, 0x8f, 0x5e, 0x53, 0xcf, 0x59,\n\t\t0x92, 0xfd, 0xd9, 0xca, 0xa8, 0xb7, 0xe9, 0x35, 0x2b, 0xe5, 0x2e, 0xe6, 0x71, 0x97, 0x2d, 0xda,\n\t\t0xf6, 0xbf, 0x06, 0xf4, 0xbf, 0xa0, 0xff, 0x63, 0x56, 0x43, 0xd8, 0x3a, 0x21, 0x82, 0x1c, 0xfb,\n\t\t0x7c, 0x80, 0x4e, 0xe1, 0x2e, 0x0d, 0x5c, 0xee, 0xb1, 0x60, 0xe4, 0x88, 0x59, 0x98, 0x3e, 0xd0,\n\t\t0x7b, 0x8d, 0xe7, 0x2b, 0x63, 0x99, 0x1a, 0x29, 0x5f, 0x34, 0x2e, 0xd1, 0x85, 0xd5, 0xfc, 0x01,\n\t\t0xe7, 0x16, 0x1e, 0x70, 0x2f, 0x2d, 0x3a, 0x1a, 0x5d, 0xd0, 0x28, 0x66, 0x3c, 0xb0, 0x82, 0x21,\n\t\t0x97, 0x40, 0x36, 0x09, 0xfd, 0xac, 0x10, 0xe4, 0x6f, 0xf4, 0x31, 0xec, 0x0c, 0x29, 0x11, 0x49,\n\t\t0x44, 0x9d, 0x69, 0x0a, 0xd5, 0x05, 0x77, 0x4f, 0x9b, 0x75, 0x80, 0xea, 0x6b, 0x78, 0xd4, 0x4f,\n\t\t0xc2, 0x90, 0x47, 0x82, 0x7a, 0x2d, 0x9f, 0xd1, 0x40, 0x68, 0x4f, 0x2c, 0x6b, 0x75, 0xc4, 0x9d,\n\t\t0xd8, 0x1b, 0xeb, 0xc8, 0x85, 0x11, 0xef, 0x7b, 0x63, 0xf4, 0x18, 0xb6, 0xfe, 0x40, 0xa6, 0x44,\n\t\t0x39, 0xd2, 0x98, 0x9b, 0x72, 0xdd, 0xf7, 0xc6, 0xd5, 0x3f, 0xe7, 0xa1, 0x88, 0xa9, 0x88, 0x66,\n\t\t0x3d, 0xee, 0x33, 0x77, 0x86, 0x4e, 0xa0, 0xcc, 0x02, 0x26, 0x18, 0xf1, 0x1d, 0x16, 0x08, 0x1a,\n\t\t0x4d, 0x49, 0xaa, 0xb2, 0xd8, 0x78, 0x5c, 0x4b, 0xdb, 0x4b, 0x2d, 0x6b, 0x2f, 0xb5, 0x13, 0xdd,\n\t\t0x5e, 0xf0, 0x8e, 0xa6, 0x58, 0x9a, 0x81, 0xea, 0xb0, 0x37, 0x20, 0xee, 0x98, 0x0f, 0x87, 0x8e,\n\t\t0xcb, 0xe9, 0x70, 0xc8, 0x5c, 0x29, 0x53, 0xed, 0x6d, 0x60, 0xa4, 0x5d, 0xad, 0x1b, 0x8f, 0xdc,\n\t\t0x76, 0x42, 0xae, 0xd9, 0x24, 0x99, 0xdc, 0x6c, 0x9b, 0x5f, 0xbb, 0xad, 0xa6, 0xcc, 0xb7, 0xfd,\n\t\t0xe4, 0x26, 0x0a, 0x11, 0x82, 0x4e, 0x42, 0x11, 0x57, 0xee, 0x1c, 0x18, 0x87, 0x85, 0x39, 0xb4,\n\t\t0xa9, 0xcd, 0xe8, 0x4b, 0x78, 0x12, 0xf0, 0xc0, 0x89, 0x64, 0xea, 0x64, 0xe0, 0x53, 0x87, 0x46,\n\t\t0x11, 0x8f, 0x9c, 0xb4, 0xa5, 0xc4, 0x95, 0xc2, 0x41, 0xfe, 0x70, 0x1b, 0x57, 0x02, 0x1e, 0xe0,\n\t\t0x0c, 0x61, 0x4a, 0x00, 0x4e, 0xfd, 0xe8, 0x15, 0xec, 0xd1, 0xeb, 0x90, 0xa5, 0x42, 0x6e, 0x24,\n\t\t0x6f, 0xac, 0x93, 0x8c, 0x6e, 0x58, 0x99, 0xea, 0xea, 0x04, 0x1e, 0x59, 0x31, 0xf7, 0x95, 0xf1,\n\t\t0x2c, 0xe2, 0x49, 0xd8, 0x23, 0x91, 0x60, 0xaa, 0x39, 0xaf, 0x68, 0x98, 0xe8, 0x97, 0x50, 0x88,\n\t\t0x05, 0x11, 0xe9, 0x83, 0xbf, 0xd7, 0x38, 0x5c, 0xf9, 0x48, 0x97, 0x03, 0xf6, 0x25, 0x1e, 0xa7,\n\t\t0xb4, 0xea, 0x14, 0x9e, 0x2c, 0x7b, 0x5b, 0x3c, 0x18, 0xb2, 0x91, 0x56, 0x88, 0x2e, 0xa1, 0xcc,\n\t\t0x32, 0xb7, 0x33, 0x92, 0xfe, 0xac, 0xb4, 0x7f, 0xfc, 0x3d, 0x76, 0x9a, 0x4b, 0xc7, 0x3b, 0x6c,\n\t\t0xc9, 0x11, 0x57, 0xff, 0x65, 0xc0, 0x7e, 0x33, 0x9e, 0x05, 0x6e, 0x36, 0x36, 0x96, 0xf7, 0xad,\n\t\t0xc0, 0x26, 0x0d, 0xe4, 0x39, 0xa7, 0x33, 0x68, 0x0b, 0x67, 0x4b, 0xd4, 0x80, 0x07, 0x61, 0x44,\n\t\t0x3d, 0x3a, 0x64, 0x01, 0xf5, 0x9c, 0x3f, 0x26, 0x34, 0xa1, 0x8e, 0x3a, 0x95, 0xf4, 0x29, 0xef,\n\t\t0xdd, 0x38, 0x7f, 0x2b, 0x7d, 0x1d, 0x79, 0x48, 0x4f, 0x01, 0x52, 0xa0, 0x2a, 0xe7, 0xbc, 0x02,\n\t\t0x6e, 0x2b, 0x8b, 0x2a, 0xd4, 0x5f, 0x43, 0x29, 0x75, 0xbb, 0x4a, 0x83, 0x7a, 0x24, 0xc5, 0xc6,\n\t\t0xd3, 0x95, 0x09, 0x66, 0x5d, 0x02, 0x17, 0x15, 0x25, 0x55, 0x5d, 0xfd, 0x4f, 0x1e, 0xde, 0x57,\n\t\t0xb3, 0x8d, 0xb6, 0xfc, 0x24, 0x16, 0x34, 0xea, 0x53, 0x9f, 0xba, 0x32, 0x13, 0x5d, 0x48, 0x7d,\n\t\t0xd8, 0x8a, 0x45, 0x44, 0x04, 0x1d, 0xcd, 0x74, 0x3b, 0x79, 0xb9, 0x32, 0xfc, 0xea, 0x20, 0x7d,\n\t\t0x4d, 0x3d, 0xce, 0x55, 0x0c, 0x3c, 0x0f, 0x84, 0xfe, 0x62, 0xc0, 0x87, 0x44, 0x11, 0x1c, 0x37,\n\t\t0x65, 0x38, 0xb1, 0x60, 0xee, 0x78, 0xe6, 0x44, 0x74, 0x24, 0x2f, 0x4c, 0xe7, 0x93, 0xf6, 0xc2,\n\t\t0x9f, 0x7e, 0x8f, 0x0d, 0x15, 0x1b, 0x2b, 0x72, 0x9a, 0x99, 0xdc, 0xf1, 0xfc, 0x3d, 0xfc, 0x8c,\n\t\t0x7c, 0x37, 0x0c, 0xfd, 0xcd, 0x80, 0x8f, 0x6e, 0x49, 0xa1, 0xd7, 0x82, 0x46, 0x01, 0xf1, 0x1d,\n\t\t0x1a, 0x08, 0x26, 0x66, 0x99, 0x98, 0xb4, 0x8e, 0x7f, 0xbe, 0x5e, 0x8c, 0xa9, 0xf9, 0xa6, 0xa2,\n\t\t0x2f, 0xc9, 0x79, 0x4e, 0xd6, 0x01, 0x11, 0x86, 0xdd, 0x4c, 0x08, 0xc9, 0x06, 0x8d, 0xbe, 0xd8,\n\t\t0xd5, 0xe3, 0x5e, 0x07, 0x9b, 0x4f, 0x25, 0x5c, 0x76, 0x6f, 0x59, 0x8e, 0x77, 0x61, 0x27, 0x3b,\n\t\t0x7b, 0x9d, 0x4d, 0xf5, 0x17, 0x50, 0xbe, 0x4d, 0x44, 0xf7, 0xa1, 0x10, 0xbb, 0x3c, 0xcc, 0xea,\n\t\t0x34, 0x5d, 0xcc, 0x8b, 0x37, 0xb7, 0xf0, 0x6f, 0xe7, 0x15, 0x3c, 0x5b, 0x73, 0xfe, 0xe8, 0x43,\n\t\t0xb8, 0xbb, 0x74, 0xa7, 0x3a, 0x68, 0x29, 0x5e, 0x80, 0x7e, 0x9e, 0xab, 0x18, 0xd5, 0xbf, 0x1a,\n\t\t0xf0, 0x7c, 0xed, 0xf9, 0xa1, 0x9f, 0xc0, 0xfd, 0xdb, 0xf7, 0x32, 0x1f, 0x71, 0xdb, 0xb2, 0x1f,\n\t\t0x2d, 0x72, 0x54, 0x71, 0xd4, 0x64, 0x6f, 0x5b, 0x66, 0xc8, 0x99, 0x9b, 0xa6, 0xb1, 0xbb, 0x4c,\n\t\t0x78, 0x4d, 0x67, 0x4a, 0xcb, 0x57, 0xb0, 0xdb, 0x23, 0x23, 0x16, 0xa8, 0x5a, 0xee, 0x86, 0x42,\n\t\t0x4d, 0xa3, 0x27, 0xb0, 0x1d, 0x92, 0x11, 0x75, 0x62, 0xf6, 0x6d, 0xba, 0x5f, 0x01, 0x6f, 0x49,\n\t\t0x43, 0x9f, 0x7d, 0x4b, 0xd1, 0x8f, 0x60, 0x27, 0xa0, 0xd7, 0xc2, 0x51, 0x08, 0xc1, 0xc7, 0x34,\n\t\t0xd0, 0x63, 0xf3, 0xae, 0x34, 0xf7, 0xc8, 0x88, 0xda, 0xd2, 0xf8, 0xe2, 0x2d, 0x94, 0x16, 0x27,\n\t\t0x2e, 0x7a, 0x0c, 0x0f, 0xcc, 0x4e, 0xab, 0x7b, 0x62, 0x75, 0xce, 0x1c, 0xfb, 0xaa, 0x67, 0x3a,\n\t\t0x56, 0xe7, 0xa2, 0xd9, 0xb6, 0x4e, 0xca, 0xef, 0xa1, 0x7d, 0x78, 0xb8, 0xec, 0xb2, 0xcf, 0xb1,\n\t\t0x75, 0x6a, 0xe3, 0xcb, 0xb2, 0x81, 0x1e, 0x02, 0x5a, 0xf6, 0xbd, 0xea, 0x77, 0x3b, 0xe5, 0x1c,\n\t\t0xaa, 0xc0, 0xfd, 0x65, 0x7b, 0x0f, 0x77, 0xed, 0xee, 0xcb, 0x72, 0xfe, 0xc5, 0x9f, 0x60, 0x6f,\n\t\t0x45, 0x17, 0x45, 0xcf, 0xe1, 0xa9, 0xd5, 0xef, 0xb6, 0x9b, 0xb6, 0xd5, 0xed, 0x38, 0x67, 0xb8,\n\t\t0xfb, 0xa6, 0xe7, 0xf4, 0xed, 0xa6, 0xbd, 0xa8, 0xe3, 0x9d, 0x90, 0x73, 0xb3, 0xd9, 0xb6, 0xcf,\n\t\t0xaf, 0xca, 0xc6, 0xbb, 0x21, 0x27, 0xb8, 0x69, 0x75, 0xcc, 0x93, 0x72, 0xee, 0xc5, 0x3f, 0x0d,\n\t\t0xf8, 0xe0, 0xbb, 0x9b, 0x03, 0xfa, 0x14, 0x3e, 0x69, 0xb6, 0x6c, 0xeb, 0xc2, 0x74, 0x5a, 0xed,\n\t\t0x37, 0x7d, 0xdb, 0xc4, 0x4e, 0xdf, 0x6c, 0x9b, 0x2d, 0x15, 0xb4, 0x6f, 0xe3, 0xa6, 0x6d, 0x9e,\n\t\t0x5d, 0x2d, 0xe8, 0x7a, 0x09, 0xf5, 0xf5, 0x70, 0x6c, 0x9e, 0xa5, 0x6b, 0xab, 0xf5, 0x5a, 0x2a,\n\t\t0xfd, 0x19, 0x1c, 0xad, 0x27, 0x99, 0x5f, 0xd9, 0x26, 0xee, 0x34, 0xdb, 0x8e, 0xd9, 0xb1, 0x2d,\n\t\t0xfb, 0xaa, 0x9c, 0xdb, 0xcf, 0x55, 0x8c, 0x17, 0x5f, 0x43, 0x49, 0xfe, 0x77, 0xe7, 0x53, 0x1a,\n\t\t0x65, 0x57, 0x77, 0xda, 0xb4, 0xda, 0xdd, 0x0b, 0x13, 0xdf, 0xbe, 0xba, 0x47, 0xb0, 0xb7, 0xec,\n\t\t0x3a, 0xed, 0xe2, 0x96, 0x59, 0x36, 0xe4, 0x9d, 0x2e, 0x3b, 0xce, 0x70, 0xb3, 0x65, 0x9e, 0xbe,\n\t\t0x69, 0x97, 0x73, 0xc7, 0xbf, 0x87, 0x47, 0x2e, 0x9f, 0xac, 0xaa, 0xed, 0xe3, 0x62, 0x4b, 0x7d,\n\t\t0x2d, 0xf5, 0xe4, 0x00, 0xee, 0x19, 0xbf, 0x3b, 0x1a, 0x31, 0xf1, 0x4d, 0x32, 0xa8, 0xb9, 0x7c,\n\t\t0x52, 0x5f, 0xfc, 0xb6, 0xfa, 0x94, 0x79, 0x7e, 0x7d, 0xc4, 0xd3, 0x2f, 0x26, 0xfd, 0xa1, 0xf5,\n\t\t0x05, 0x09, 0xd9, 0xf4, 0x68, 0xb0, 0xa1, 0x6c, 0x2f, 0xff, 0x1b, 0x00, 0x00, 0xff, 0xff, 0x5e,\n\t\t0xe2, 0xc9, 0x06, 0x8c, 0x0d, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/query.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x53, 0xdf, 0x6f, 0x93, 0x50,\n\t\t0x18, 0x95, 0x9a, 0x2c, 0xd9, 0xb7, 0x55, 0xc9, 0x9d, 0xc6, 0xda, 0xec, 0x47, 0xd3, 0xed, 0x61,\n\t\t0x69, 0x14, 0xec, 0xf4, 0x6d, 0x4f, 0x8c, 0x5e, 0x0d, 0x86, 0x01, 0x03, 0xda, 0xa5, 0x7b, 0x21,\n\t\t0x94, 0x5e, 0x2b, 0x8e, 0x72, 0xf1, 0x5e, 0x68, 0xed, 0x3f, 0xe0, 0xbb, 0x7f, 0x8d, 0xff, 0x9e,\n\t\t0x81, 0x52, 0x5b, 0x2d, 0x33, 0xbe, 0x7d, 0x9c, 0xef, 0x1c, 0xce, 0x39, 0xb9, 0xf9, 0xe0, 0x24,\n\t\t0x1b, 0x11, 0x26, 0x07, 0xfe, 0x98, 0xc4, 0x01, 0x91, 0xfd, 0x24, 0x94, 0x67, 0x5d, 0xf9, 0x6b,\n\t\t0x46, 0xd8, 0x42, 0x4a, 0x18, 0x4d, 0x29, 0x3a, 0xc8, 0x09, 0x52, 0x49, 0x90, 0xfc, 0x24, 0x94,\n\t\t0x66, 0xdd, 0x66, 0xab, 0x4a, 0x15, 0xd0, 0xe9, 0x94, 0xc6, 0x4b, 0x59, 0xb3, 0x5d, 0xc5, 0x98,\n\t\t0x53, 0x76, 0xff, 0x29, 0xa2, 0xf3, 0x25, 0xa7, 0x7d, 0x0f, 0xf5, 0xdb, 0x12, 0xb9, 0xc9, 0x1d,\n\t\t0xd1, 0x11, 0x40, 0x61, 0xed, 0xa5, 0x8b, 0x84, 0x34, 0x84, 0x96, 0x70, 0xbe, 0x6b, 0xef, 0x16,\n\t\t0x88, 0xbb, 0x48, 0x08, 0xba, 0x5c, 0xad, 0x7d, 0x36, 0xe1, 0x8d, 0x5a, 0x4b, 0x38, 0xdf, 0xbb,\n\t\t0x38, 0x94, 0x2a, 0xf2, 0x49, 0x96, 0xbf, 0x88, 0xa8, 0x3f, 0x2e, 0xc5, 0x0a, 0x9b, 0xf0, 0xf6,\n\t\t0x4f, 0x01, 0x0e, 0xfe, 0x70, 0xb3, 0x09, 0xcf, 0xa2, 0x14, 0x61, 0xd8, 0x63, 0xc5, 0xb4, 0x36,\n\t\t0x7d, 0x72, 0x71, 0x56, 0xf9, 0xd7, 0x0d, 0x59, 0x9e, 0xc7, 0x06, 0xf6, 0x7b, 0x46, 0xef, 0x60,\n\t\t0xc7, 0x8f, 0xf9, 0x9c, 0xb0, 0xff, 0xca, 0x55, 0x72, 0xd1, 0x29, 0xd4, 0x09, 0x63, 0x94, 0x79,\n\t\t0x53, 0xc2, 0xb9, 0x3f, 0x21, 0x8d, 0xc7, 0x45, 0xe7, 0xfd, 0x02, 0xbc, 0x5e, 0x62, 0x6d, 0x02,\n\t\t0xf5, 0xd2, 0xf9, 0x0b, 0x09, 0x52, 0x32, 0x46, 0x2e, 0xec, 0x07, 0x11, 0xe5, 0xc4, 0xe3, 0xa9,\n\t\t0x9f, 0x66, 0xbc, 0xcc, 0xdc, 0xad, 0x74, 0x5c, 0x55, 0xc6, 0xdf, 0x48, 0x90, 0xa5, 0x21, 0x8d,\n\t\t0xd5, 0x5c, 0xe9, 0x14, 0x42, 0x7b, 0x2f, 0x58, 0x7f, 0x74, 0x62, 0x78, 0xfa, 0x57, 0x41, 0x74,\n\t\t0x04, 0x2f, 0x6f, 0xfa, 0xd8, 0x1e, 0x7a, 0x36, 0x76, 0xfa, 0xba, 0xeb, 0xb9, 0x43, 0x0b, 0x7b,\n\t\t0x9a, 0x31, 0x50, 0x74, 0xad, 0x27, 0x3e, 0x42, 0xc7, 0xd0, 0xdc, 0x5e, 0x2b, 0x86, 0x73, 0x8b,\n\t\t0x6d, 0xdc, 0x13, 0x05, 0x74, 0x08, 0x8d, 0xed, 0xfd, 0x7b, 0x45, 0xd3, 0x71, 0x4f, 0xac, 0x75,\n\t\t0x7e, 0x08, 0xf0, 0x6c, 0xa3, 0x97, 0x4a, 0xe3, 0x71, 0x98, 0x07, 0x44, 0x6d, 0x38, 0x5e, 0xc9,\n\t\t0x3e, 0x62, 0xd5, 0xf5, 0x54, 0xd3, 0xe8, 0x69, 0xae, 0x66, 0x1a, 0x1b, 0xd6, 0xa7, 0x70, 0xf2,\n\t\t0x00, 0xc7, 0x30, 0x5d, 0xcf, 0xb4, 0xb0, 0x21, 0x0a, 0xe8, 0x0d, 0xbc, 0xfa, 0x07, 0x49, 0x35,\n\t\t0xaf, 0x2d, 0x1d, 0xbb, 0xb8, 0xe7, 0xa9, 0x3a, 0x56, 0x0c, 0x7d, 0x28, 0xd6, 0x3a, 0xdf, 0x05,\n\t\t0x78, 0x5e, 0x64, 0x52, 0x69, 0xcc, 0x43, 0x9e, 0x92, 0x38, 0x58, 0xe8, 0x64, 0x46, 0xa2, 0xb5,\n\t\t0xa1, 0x6a, 0x1a, 0x8e, 0xe6, 0xb8, 0xd8, 0x50, 0x87, 0x9e, 0x8e, 0x07, 0x58, 0xdf, 0x48, 0x75,\n\t\t0x06, 0xad, 0x87, 0x48, 0x78, 0x80, 0x0d, 0xb7, 0xaf, 0xe8, 0xa2, 0xb0, 0xee, 0xb7, 0xcd, 0x72,\n\t\t0x5c, 0xdb, 0x34, 0x3e, 0x88, 0xb5, 0xab, 0x3b, 0x78, 0x11, 0xd0, 0x69, 0xd5, 0x8b, 0x5e, 0x41,\n\t\t0x11, 0xd0, 0xca, 0x2f, 0xc8, 0x12, 0xee, 0xba, 0x93, 0x30, 0xfd, 0x9c, 0x8d, 0xa4, 0x80, 0x4e,\n\t\t0xe5, 0xcd, 0x93, 0x7b, 0x1d, 0x8e, 0x23, 0x79, 0x42, 0xe5, 0xe2, 0xd2, 0xca, 0xfb, 0xbb, 0xf4,\n\t\t0x93, 0x70, 0xd6, 0x1d, 0xed, 0x14, 0xd8, 0xdb, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0xbd, 0x69,\n\t\t0x28, 0x5b, 0xfb, 0x03, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xc4, 0x5a, 0xcd, 0x72, 0xdb, 0xc8,\n\t\t0xb5, 0xbe, 0x20, 0x25, 0x59, 0x3a, 0xa4, 0x24, 0xa8, 0x25, 0x59, 0xf4, 0xcf, 0xd8, 0xb2, 0x66,\n\t\t0xec, 0x91, 0x79, 0xc7, 0xd2, 0xc8, 0xf3, 0x3f, 0xbe, 0x73, 0x1d, 0x08, 0x84, 0x6c, 0xd8, 0x34,\n\t\t0xc8, 0x34, 0x41, 0x6b, 0x34, 0x95, 0x04, 0x05, 0x91, 0x2d, 0x09, 0x31, 0x09, 0xb0, 0x80, 0xa6,\n\t\t0x6d, 0xed, 0x53, 0x95, 0x6c, 0x93, 0xd5, 0x54, 0x56, 0x79, 0x80, 0x54, 0xa5, 0x52, 0x59, 0x64,\n\t\t0x95, 0xca, 0x13, 0x64, 0x9b, 0x57, 0x48, 0xe5, 0x2d, 0x52, 0xdd, 0x68, 0x90, 0x00, 0x09, 0x12,\n\t\t0x74, 0x52, 0x35, 0xd9, 0x09, 0xa7, 0xbf, 0xef, 0xf4, 0xe9, 0xf3, 0xd7, 0x07, 0x10, 0x61, 0xa7,\n\t\t0x7f, 0x4a, 0xfc, 0xfd, 0x96, 0xdd, 0x26, 0x6e, 0x8b, 0xec, 0xdb, 0x3d, 0x67, 0xff, 0xf5, 0xc1,\n\t\t0xfe, 0x1b, 0xcf, 0x7f, 0x75, 0xd6, 0xf1, 0xde, 0xec, 0xf5, 0x7c, 0x8f, 0x7a, 0x68, 0x9d, 0x61,\n\t\t0xf6, 0x04, 0x66, 0xcf, 0xee, 0x39, 0x7b, 0xaf, 0x0f, 0xae, 0xdf, 0x3a, 0xf7, 0xbc, 0xf3, 0x0e,\n\t\t0xd9, 0xe7, 0x90, 0xd3, 0xfe, 0xd9, 0x7e, 0xbb, 0xef, 0xdb, 0xd4, 0xf1, 0xdc, 0x90, 0x74, 0xfd,\n\t\t0xf6, 0xe8, 0x3a, 0x75, 0xba, 0x24, 0xa0, 0x76, 0xb7, 0x27, 0x00, 0xdb, 0x69, 0x3b, 0xb7, 0xbc,\n\t\t0x6e, 0x77, 0xa0, 0x22, 0xd5, 0x36, 0x6a, 0x07, 0xaf, 0x3a, 0x4e, 0x40, 0x43, 0xcc, 0xce, 0xf7,\n\t\t0x45, 0xd8, 0x3c, 0x16, 0xe6, 0x6a, 0x6f, 0x49, 0xab, 0xcf, 0x4c, 0xd0, 0xdd, 0x33, 0x0f, 0x35,\n\t\t0x01, 0x45, 0xe7, 0xb0, 0x48, 0xb4, 0x52, 0x92, 0xb6, 0xa5, 0xdd, 0xc2, 0xc3, 0x7b, 0x7b, 0x29,\n\t\t0x47, 0xda, 0x1b, 0xd3, 0x83, 0xd7, 0xde, 0x8c, 0x8a, 0xd0, 0x67, 0x30, 0x47, 0x2f, 0x7b, 0xa4,\n\t\t0x94, 0xe3, 0x8a, 0xee, 0x4c, 0x55, 0x64, 0x5e, 0xf6, 0x08, 0xe6, 0x70, 0xf4, 0x15, 0x40, 0x40,\n\t\t0x6d, 0x9f, 0x5a, 0xcc, 0x0d, 0xa5, 0x3c, 0x27, 0x5f, 0xdf, 0x0b, 0x7d, 0xb4, 0x17, 0xf9, 0x68,\n\t\t0xcf, 0x8c, 0x7c, 0x84, 0x97, 0x38, 0x9a, 0x3d, 0x33, 0x6a, 0xab, 0xe3, 0x05, 0x24, 0xa4, 0xce,\n\t\t0x65, 0x53, 0x39, 0x9a, 0x53, 0x4d, 0x28, 0x86, 0xd4, 0x80, 0xda, 0xb4, 0x1f, 0x94, 0xe6, 0xb7,\n\t\t0xa5, 0xdd, 0x95, 0x87, 0x07, 0xb3, 0x9d, 0x5e, 0x65, 0xcc, 0x06, 0x27, 0xe2, 0x42, 0x6b, 0xf8,\n\t\t0x80, 0xee, 0xc2, 0xca, 0x85, 0x13, 0x50, 0xcf, 0xbf, 0xb4, 0x3a, 0xc4, 0x3d, 0xa7, 0x17, 0xa5,\n\t\t0x85, 0x6d, 0x69, 0x37, 0x8f, 0x97, 0x85, 0xb4, 0xca, 0x85, 0xe8, 0x27, 0xb0, 0xd9, 0xb3, 0x7d,\n\t\t0xe2, 0xd2, 0xa1, 0xfb, 0x2d, 0xc7, 0x3d, 0xf3, 0x4a, 0x57, 0xf8, 0x11, 0x76, 0x53, 0xad, 0xa8,\n\t\t0x73, 0x46, 0x22, 0x92, 0x78, 0xbd, 0x37, 0x2e, 0x44, 0x0a, 0xac, 0x0c, 0xd5, 0x72, 0xcf, 0x2c,\n\t\t0x66, 0x7a, 0x66, 0x79, 0xc0, 0xe0, 0xde, 0x79, 0x00, 0x73, 0x5d, 0xd2, 0xf5, 0x4a, 0x4b, 0x9c,\n\t\t0x78, 0x2d, 0xd5, 0x9e, 0x17, 0xa4, 0xeb, 0x61, 0x0e, 0x43, 0x18, 0xd6, 0x02, 0x62, 0xfb, 0xad,\n\t\t0x0b, 0xcb, 0xa6, 0xd4, 0x77, 0x4e, 0xfb, 0x94, 0x04, 0x25, 0xe0, 0xdc, 0xbb, 0xa9, 0xdc, 0x06,\n\t\t0x47, 0x2b, 0x03, 0x30, 0x96, 0x83, 0x11, 0x09, 0xaa, 0xc2, 0x9a, 0xdd, 0xa7, 0x9e, 0xe5, 0x93,\n\t\t0x80, 0x50, 0xab, 0xe7, 0x39, 0x2e, 0x0d, 0x4a, 0x05, 0xae, 0x73, 0x3b, 0x55, 0x27, 0x66, 0xc0,\n\t\t0x3a, 0xc7, 0xe1, 0x55, 0x46, 0x8d, 0x09, 0xd0, 0x0d, 0x58, 0x62, 0xe5, 0x61, 0xb1, 0xfa, 0x28,\n\t\t0x15, 0xb7, 0xa5, 0xdd, 0x25, 0xbc, 0xc8, 0x04, 0x55, 0x27, 0xa0, 0x48, 0x85, 0x95, 0xc1, 0x62,\n\t\t0x18, 0x87, 0x35, 0xbe, 0xcf, 0x7b, 0xa9, 0xfb, 0x98, 0x82, 0x86, 0x8b, 0x91, 0x02, 0xee, 0xf5,\n\t\t0x2d, 0xb8, 0xe2, 0x04, 0x56, 0xcb, 0xf7, 0xdc, 0xd2, 0xf2, 0xb6, 0xb4, 0xbb, 0x88, 0x17, 0x9c,\n\t\t0x40, 0xf5, 0x3d, 0x17, 0x3d, 0x82, 0x42, 0xbf, 0xd7, 0xb6, 0xa9, 0xc8, 0xd2, 0x95, 0xcc, 0x58,\n\t\t0x40, 0x08, 0xe7, 0x81, 0xf8, 0x39, 0xc8, 0x3d, 0xdb, 0xa7, 0x0e, 0x8f, 0x65, 0xcb, 0x73, 0xcf,\n\t\t0x9c, 0xf3, 0xd2, 0xea, 0x76, 0x7e, 0xb7, 0xf0, 0xf0, 0xf1, 0x6c, 0xa9, 0xca, 0x6c, 0x63, 0xa9,\n\t\t0x13, 0xaa, 0x50, 0xb9, 0x06, 0xcd, 0xa5, 0xfe, 0x25, 0x5e, 0xed, 0x25, 0xa5, 0xe8, 0x25, 0xac,\n\t\t0x33, 0xf3, 0x2d, 0xef, 0x35, 0xf1, 0x3b, 0x76, 0xcf, 0xea, 0x79, 0x1d, 0xa7, 0x75, 0x59, 0x92,\n\t\t0x79, 0x65, 0xa4, 0xf7, 0x05, 0x76, 0xc0, 0x5a, 0x08, 0xaf, 0x73, 0x34, 0x5e, 0x6b, 0x8d, 0x8a,\n\t\t0xd0, 0x5b, 0xb8, 0x6d, 0xb7, 0xa8, 0xf3, 0x9a, 0x58, 0xad, 0x4e, 0x3f, 0xa0, 0xc4, 0xb7, 0x02,\n\t\t0xd2, 0x21, 0x2d, 0x7e, 0x24, 0xb1, 0x07, 0xe2, 0x4e, 0x49, 0xaf, 0x3e, 0x85, 0x73, 0xd5, 0x90,\n\t\t0xda, 0x88, 0x98, 0x62, 0xbb, 0x9b, 0xf6, 0x94, 0x55, 0xf4, 0x3e, 0x2c, 0xf3, 0x13, 0x05, 0xad,\n\t\t0x0b, 0xd2, 0xee, 0x77, 0x48, 0x69, 0x9d, 0x47, 0xbe, 0xc8, 0x84, 0x0d, 0x21, 0x43, 0xc7, 0x20,\n\t\t0x0f, 0xcb, 0x45, 0x74, 0x83, 0x0d, 0x7e, 0xe6, 0x8f, 0x66, 0x73, 0xb1, 0x68, 0x04, 0xab, 0x24,\n\t\t0x29, 0x40, 0x26, 0x94, 0xa2, 0x8d, 0xdb, 0xd6, 0x48, 0x45, 0x6e, 0x66, 0x66, 0xc1, 0xd5, 0x01,\n\t\t0x57, 0x8b, 0x97, 0xe6, 0xf5, 0x43, 0xd8, 0x48, 0x0b, 0x27, 0x92, 0x21, 0xff, 0x8a, 0x5c, 0xf2,\n\t\t0x2e, 0xbe, 0x84, 0xd9, 0x9f, 0x68, 0x03, 0xe6, 0x5f, 0xdb, 0x9d, 0x7e, 0xd8, 0x90, 0x97, 0x70,\n\t\t0xf8, 0xf0, 0x75, 0xee, 0x4b, 0x69, 0xe7, 0xfb, 0x1c, 0xdc, 0x1a, 0x6f, 0x6a, 0x5c, 0x99, 0xb8,\n\t\t0xaa, 0xd0, 0xd7, 0xf1, 0x82, 0x91, 0x66, 0x29, 0x87, 0x61, 0x3d, 0xd9, 0xb0, 0x9d, 0xf0, 0x28,\n\t\t0xeb, 0xed, 0x9e, 0x35, 0xec, 0xd4, 0x5e, 0x9f, 0x8a, 0x4b, 0xe2, 0xda, 0x98, 0x03, 0x2a, 0xc2,\n\t\t0x00, 0x7c, 0x33, 0xee, 0x4e, 0x9f, 0x9a, 0x9e, 0x1a, 0xf5, 0x6e, 0xaf, 0x4f, 0xd1, 0x31, 0xdc,\n\t\t0xe0, 0xe6, 0x4d, 0xd0, 0x9e, 0xcf, 0xd2, 0xbe, 0xc5, 0xd8, 0x29, 0x8a, 0x77, 0xfe, 0x26, 0xc1,\n\t\t0x7a, 0x4a, 0xa7, 0x65, 0x0d, 0xa4, 0xed, 0x75, 0x6d, 0xc7, 0xb5, 0x9c, 0xb6, 0x70, 0xf2, 0x62,\n\t\t0x28, 0xd0, 0xdb, 0xe8, 0x36, 0x14, 0xc4, 0xa2, 0x6b, 0x77, 0x23, 0x7f, 0x43, 0x28, 0x32, 0xec,\n\t\t0x2e, 0x99, 0x70, 0xe3, 0xe6, 0xff, 0xd3, 0x1b, 0xf7, 0x0e, 0x14, 0x1d, 0xd7, 0xa1, 0x8e, 0x4d,\n\t\t0x49, 0x9b, 0xd9, 0x35, 0xc7, 0x2f, 0x9b, 0xc2, 0x40, 0xa6, 0xb7, 0x77, 0x7e, 0x2d, 0xc1, 0xa6,\n\t\t0xf6, 0x96, 0x12, 0xdf, 0xb5, 0x3b, 0x3f, 0xc8, 0x14, 0x30, 0x6a, 0x53, 0x6e, 0xdc, 0xa6, 0x3f,\n\t\t0x2f, 0xc0, 0x7a, 0x9d, 0xb8, 0x6d, 0xc7, 0x3d, 0xe7, 0xc5, 0xed, 0xd0, 0x4b, 0x6e, 0xd1, 0x6d,\n\t\t0x28, 0xd8, 0xe2, 0x79, 0xe8, 0x65, 0x88, 0x44, 0x7a, 0x1b, 0x1d, 0xc1, 0xf2, 0x00, 0x90, 0x39,\n\t\t0x6a, 0x44, 0xaa, 0xf9, 0xa8, 0x51, 0xb4, 0x63, 0x4f, 0xe8, 0x31, 0xcc, 0xb3, 0x42, 0x0f, 0xa7,\n\t\t0x8d, 0x95, 0x87, 0xf7, 0xd3, 0xef, 0xdb, 0xa4, 0x85, 0xac, 0xa8, 0x09, 0x0e, 0x79, 0x48, 0x87,\n\t\t0xb5, 0x0b, 0x62, 0xfb, 0xf4, 0x94, 0xd8, 0xd4, 0x6a, 0x13, 0x6a, 0x3b, 0x9d, 0x40, 0xcc, 0x1f,\n\t\t0x37, 0x27, 0x5c, 0xde, 0x97, 0x1d, 0xcf, 0x6e, 0x63, 0x79, 0x40, 0xab, 0x84, 0x2c, 0xf4, 0x0c,\n\t\t0xd6, 0x3b, 0x76, 0x40, 0xad, 0xa1, 0x3e, 0xde, 0x20, 0xe6, 0x33, 0x1b, 0xc4, 0x1a, 0xa3, 0x3d,\n\t\t0x8d, 0x58, 0xfc, 0xb6, 0x38, 0x02, 0x2e, 0x0c, 0xab, 0x82, 0xb4, 0x43, 0x4d, 0x0b, 0x99, 0x9a,\n\t\t0x56, 0x19, 0xa9, 0x11, 0x72, 0xb8, 0x9e, 0x12, 0x5c, 0xb1, 0x29, 0x25, 0xdd, 0x1e, 0xe5, 0x13,\n\t\t0xc9, 0x3c, 0x8e, 0x1e, 0xd1, 0x7d, 0x90, 0xbb, 0xf6, 0x5b, 0xa7, 0xdb, 0xef, 0x5a, 0x42, 0x14,\n\t\t0xf0, 0xe9, 0x62, 0x1e, 0xaf, 0x0a, 0xb9, 0x22, 0xc4, 0x6c, 0x0c, 0x19, 0xb6, 0x3f, 0x6e, 0xc9,\n\t\t0x52, 0xf6, 0x18, 0x32, 0x60, 0x70, 0x3b, 0x54, 0x58, 0x25, 0x6f, 0x7b, 0x4e, 0x58, 0xb3, 0xa1,\n\t\t0x0e, 0xc8, 0xd4, 0xb1, 0x32, 0xa4, 0x70, 0x25, 0x8f, 0xa1, 0xc8, 0x9d, 0x72, 0x66, 0x3b, 0x9d,\n\t\t0xbe, 0x4f, 0xc4, 0x0c, 0x91, 0x1e, 0xa6, 0xa3, 0x10, 0x83, 0x0b, 0x8c, 0x21, 0x1e, 0xd0, 0xc7,\n\t\t0xb0, 0xc1, 0x15, 0xb0, 0x5c, 0x27, 0xbe, 0xe5, 0xb4, 0x89, 0x4b, 0x1d, 0x7a, 0x29, 0xc6, 0x08,\n\t\t0xc4, 0xd6, 0x8e, 0xf9, 0x92, 0x2e, 0x56, 0xd0, 0xe7, 0xb0, 0x15, 0x85, 0x60, 0x94, 0xb4, 0xcc,\n\t\t0x49, 0x9b, 0x62, 0x79, 0x84, 0x77, 0x1b, 0x0a, 0x91, 0x03, 0x58, 0x01, 0xac, 0xf0, 0xd2, 0x81,\n\t\t0x48, 0xa4, 0xb7, 0x77, 0xfe, 0x94, 0x83, 0x6b, 0x22, 0x2f, 0xd5, 0x0b, 0xa7, 0xd3, 0xfe, 0x41,\n\t\t0x2a, 0xfa, 0xa3, 0x98, 0x5a, 0x56, 0x75, 0xf1, 0x26, 0x27, 0xbf, 0x89, 0x0d, 0xf4, 0xbc, 0xd5,\n\t\t0x8d, 0xd6, 0x7f, 0x7e, 0xac, 0xfe, 0xd9, 0xa0, 0x21, 0xc6, 0xdf, 0xb0, 0x6b, 0x8b, 0x21, 0x60,\n\t\t0x6e, 0xca, 0xa0, 0x11, 0xb6, 0x64, 0xde, 0xa9, 0xa3, 0x41, 0xa3, 0x37, 0x2a, 0x42, 0x57, 0x61,\n\t\t0x21, 0xec, 0xb9, 0xbc, 0x7a, 0x96, 0xb0, 0x78, 0xda, 0xf9, 0x47, 0x6e, 0xd0, 0x6f, 0x2a, 0xa4,\n\t\t0xe5, 0x04, 0x91, 0xbf, 0x06, 0x6d, 0x40, 0xca, 0x6e, 0x03, 0x11, 0x31, 0xd1, 0x06, 0xc6, 0x53,\n\t\t0x3c, 0xf7, 0xae, 0x29, 0xfe, 0x0d, 0x14, 0x13, 0xd5, 0x9a, 0xfd, 0xfe, 0x53, 0x08, 0xd2, 0x2b,\n\t\t0x75, 0x2e, 0x59, 0xa9, 0x18, 0xb6, 0x3c, 0xdf, 0x39, 0x77, 0x5c, 0xbb, 0x63, 0x8d, 0x18, 0x99,\n\t\t0xdd, 0x5b, 0x36, 0x23, 0x6a, 0x23, 0x61, 0xec, 0x48, 0x7e, 0x2e, 0x8c, 0xe5, 0xe7, 0x5f, 0x72,\n\t\t0x70, 0x2d, 0x6a, 0x98, 0x55, 0xaf, 0x65, 0x77, 0x2a, 0x4e, 0xd0, 0xb3, 0x69, 0xeb, 0x62, 0xb6,\n\t\t0xfe, 0xfe, 0xdf, 0xf7, 0xe7, 0xcf, 0xe0, 0x56, 0xd2, 0x02, 0xcb, 0x3b, 0xb3, 0xe8, 0x85, 0x13,\n\t\t0x58, 0x71, 0x37, 0x4f, 0x57, 0x78, 0x3d, 0x61, 0x51, 0xed, 0xcc, 0xbc, 0x70, 0x02, 0xd1, 0x15,\n\t\t0xd1, 0x7b, 0x00, 0x7c, 0x6e, 0xa1, 0xde, 0x2b, 0x12, 0xa6, 0x69, 0x11, 0xf3, 0x41, 0xcb, 0x64,\n\t\t0x82, 0x9d, 0x67, 0x50, 0x88, 0xbf, 0xb5, 0x3c, 0x82, 0x05, 0xf1, 0xe2, 0x23, 0xf1, 0x99, 0xff,\n\t\t0xfd, 0x8c, 0x17, 0x1f, 0xfe, 0x4e, 0x28, 0x28, 0x3b, 0x7f, 0xc8, 0xc1, 0x4a, 0x72, 0x09, 0x7d,\n\t\t0x08, 0xab, 0xa7, 0x8e, 0x6b, 0xfb, 0x97, 0x56, 0xeb, 0x82, 0xb4, 0x5e, 0x05, 0xfd, 0xae, 0x08,\n\t\t0xc2, 0x4a, 0x28, 0x56, 0x85, 0x14, 0x6d, 0xc2, 0x82, 0xdf, 0x77, 0xa3, 0xeb, 0x7b, 0x09, 0xcf,\n\t\t0xfb, 0x7d, 0x36, 0xe7, 0x7c, 0x03, 0x37, 0xce, 0x1c, 0x3f, 0x60, 0x57, 0x5e, 0x58, 0x0d, 0x56,\n\t\t0xcb, 0xeb, 0xf6, 0x3a, 0x24, 0x51, 0xea, 0x25, 0x0e, 0x89, 0xea, 0x45, 0x8d, 0x00, 0x9c, 0x5e,\n\t\t0x6c, 0xf9, 0xc4, 0x1e, 0xc4, 0x26, 0xdb, 0x95, 0x05, 0x81, 0x17, 0x8d, 0x7c, 0x99, 0xb7, 0x76,\n\t\t0xc7, 0x3d, 0x9f, 0x35, 0x8f, 0x8b, 0x11, 0x81, 0x2b, 0xb8, 0x05, 0xc0, 0xdf, 0x26, 0xa9, 0x7d,\n\t\t0xda, 0x09, 0xef, 0xc5, 0x45, 0x1c, 0x93, 0x94, 0xff, 0x28, 0xc1, 0x46, 0xda, 0xad, 0x8f, 0x76,\n\t\t0xe0, 0x56, 0x5d, 0x33, 0x2a, 0xba, 0xf1, 0xc4, 0x52, 0x54, 0x53, 0x7f, 0xa9, 0x9b, 0x27, 0x56,\n\t\t0xc3, 0x54, 0x4c, 0xcd, 0xd2, 0x8d, 0x97, 0x4a, 0x55, 0xaf, 0xc8, 0xff, 0x83, 0x3e, 0x80, 0xed,\n\t\t0x09, 0x98, 0x86, 0xfa, 0x54, 0xab, 0x34, 0xab, 0x5a, 0x45, 0x96, 0xa6, 0x68, 0x6a, 0x98, 0x0a,\n\t\t0x36, 0xb5, 0x8a, 0x9c, 0x43, 0xff, 0x0b, 0x1f, 0x4e, 0xc0, 0xa8, 0x8a, 0xa1, 0x6a, 0x55, 0x0b,\n\t\t0x6b, 0x3f, 0x6e, 0x6a, 0x0d, 0x06, 0xce, 0x97, 0x7f, 0x31, 0xb4, 0x39, 0xd1, 0xa2, 0xe2, 0x3b,\n\t\t0x55, 0x34, 0x55, 0x6f, 0xe8, 0x35, 0x63, 0x9a, 0xcd, 0x23, 0x98, 0x09, 0x36, 0x8f, 0xa2, 0x22,\n\t\t0x9b, 0xcb, 0xbf, 0xcc, 0x0d, 0x3f, 0x36, 0xe9, 0x6d, 0x4c, 0xfa, 0x83, 0xa6, 0xfc, 0x01, 0x6c,\n\t\t0x1f, 0xd7, 0xf0, 0xf3, 0xa3, 0x6a, 0xed, 0xd8, 0xd2, 0x2b, 0x16, 0xd6, 0x9a, 0x0d, 0xcd, 0xaa,\n\t\t0xd7, 0xaa, 0xba, 0x7a, 0x12, 0xb3, 0xe4, 0x4b, 0xf8, 0x74, 0x22, 0x4a, 0xa9, 0x32, 0x69, 0xa5,\n\t\t0x59, 0xaf, 0xea, 0x2a, 0xdb, 0xf5, 0x48, 0xd1, 0xab, 0x5a, 0xc5, 0xaa, 0x19, 0xd5, 0x13, 0x59,\n\t\t0x42, 0x1f, 0xc1, 0xee, 0xac, 0x4c, 0x39, 0x87, 0x1e, 0xc0, 0xfd, 0x89, 0x68, 0xac, 0x3d, 0xd3,\n\t\t0x54, 0x33, 0x06, 0xcf, 0xa3, 0x03, 0x78, 0x30, 0x11, 0x6e, 0x6a, 0xf8, 0x85, 0x6e, 0x70, 0x87,\n\t\t0x1e, 0x59, 0xb8, 0x69, 0x18, 0xba, 0xf1, 0x44, 0x9e, 0x2b, 0x5f, 0xc2, 0xda, 0xd8, 0x5b, 0x31,\n\t\t0xba, 0x0d, 0x37, 0x54, 0x5c, 0x33, 0xac, 0xda, 0x4b, 0x0d, 0x57, 0x95, 0xfa, 0xf8, 0xf9, 0x27,\n\t\t0x00, 0x1a, 0xcf, 0xf5, 0x7a, 0x3d, 0x0a, 0x42, 0x1a, 0xe0, 0xb0, 0x79, 0x74, 0xa4, 0x61, 0xab,\n\t\t0x66, 0x68, 0x72, 0xae, 0xfc, 0x3b, 0x09, 0xd6, 0xc6, 0x2e, 0x4a, 0xa6, 0xba, 0xae, 0x60, 0xcd,\n\t\t0x30, 0x2d, 0xb5, 0x5a, 0x4b, 0xf3, 0xfd, 0x04, 0x80, 0x72, 0xa8, 0x18, 0x95, 0x9a, 0x21, 0x4b,\n\t\t0xe8, 0x1e, 0xec, 0xa4, 0x01, 0x44, 0x1a, 0x8a, 0xac, 0x94, 0x73, 0xe8, 0x0e, 0xbc, 0x97, 0x86,\n\t\t0x1b, 0x38, 0x4a, 0xce, 0x97, 0xff, 0x99, 0x83, 0x9b, 0xd3, 0x3e, 0xa7, 0xb1, 0xe4, 0x1f, 0x78,\n\t\t0x5c, 0xfb, 0x56, 0x53, 0x9b, 0x26, 0x4b, 0xb7, 0x50, 0x1f, 0x4b, 0xba, 0x66, 0x23, 0x66, 0x79,\n\t\t0x3c, 0x9a, 0x13, 0xc0, 0x6a, 0xed, 0x45, 0xbd, 0xaa, 0x99, 0xdc, 0x87, 0x65, 0xb8, 0x97, 0x05,\n\t\t0x0f, 0x73, 0x4b, 0xce, 0x25, 0xd2, 0x6a, 0x92, 0x6a, 0x7e, 0x6e, 0x56, 0x85, 0x68, 0x0f, 0xca,\n\t\t0x59, 0xe8, 0x81, 0x17, 0x2a, 0xf2, 0x1c, 0xfa, 0x14, 0x3e, 0xce, 0x36, 0xdc, 0x30, 0x75, 0xa3,\n\t\t0xa9, 0x55, 0x2c, 0xa5, 0x61, 0x19, 0xda, 0xb1, 0x3c, 0x3f, 0xcb, 0x71, 0x4d, 0xfd, 0x05, 0x2b,\n\t\t0x8d, 0xa6, 0x29, 0x2f, 0x94, 0x7f, 0x95, 0x87, 0xad, 0x09, 0x1f, 0x2b, 0xd0, 0x5d, 0xb8, 0x93,\n\t\t0xa2, 0x6a, 0xcc, 0xc1, 0x53, 0x61, 0xa2, 0x29, 0xc8, 0xd2, 0x74, 0xd8, 0xb0, 0xb1, 0x7d, 0x08,\n\t\t0xef, 0x4f, 0x86, 0x0d, 0x03, 0x95, 0x4f, 0xf4, 0x8c, 0x31, 0xa0, 0x08, 0xd1, 0x1c, 0x4b, 0xcb,\n\t\t0x29, 0xea, 0xa2, 0xe0, 0xcc, 0xa3, 0x5d, 0xf8, 0x60, 0x32, 0x2e, 0x16, 0x96, 0x85, 0x09, 0x61,\n\t\t0x9c, 0x14, 0x90, 0x2b, 0xd3, 0x0f, 0x34, 0x0c, 0xc5, 0x62, 0xf9, 0xaf, 0x12, 0x5c, 0x55, 0x3d,\n\t\t0x97, 0x3a, 0x6e, 0x9f, 0x28, 0x81, 0x41, 0xde, 0xe8, 0xe1, 0x38, 0xec, 0xf9, 0xcc, 0x77, 0x91,\n\t\t0x66, 0xa1, 0xd8, 0xd2, 0x0d, 0xdd, 0xd4, 0x15, 0xb3, 0x86, 0x93, 0x91, 0x98, 0x0c, 0x63, 0x6d,\n\t\t0xb9, 0xa2, 0xe1, 0x30, 0xc5, 0x27, 0xc3, 0xb0, 0x66, 0xe2, 0x13, 0x51, 0x95, 0xe1, 0x3d, 0x33,\n\t\t0x19, 0xcb, 0x9b, 0x4d, 0x74, 0x0b, 0xc8, 0xf9, 0xf2, 0xef, 0x25, 0x28, 0x88, 0x6f, 0x24, 0xfc,\n\t\t0x15, 0xba, 0x04, 0x1b, 0xec, 0x80, 0xb5, 0xa6, 0x69, 0x99, 0x27, 0x75, 0x2d, 0xd9, 0x4e, 0x12,\n\t\t0x2b, 0x3c, 0xfe, 0x96, 0x59, 0x0b, 0x13, 0x35, 0x6c, 0x65, 0x49, 0x80, 0xd8, 0x85, 0x61, 0x38,\n\t\t0x58, 0xce, 0x4d, 0xc5, 0x84, 0x7a, 0xf2, 0xe8, 0x3a, 0x5c, 0x4d, 0x60, 0x9e, 0x6a, 0x0a, 0x36,\n\t\t0x0f, 0x35, 0xc5, 0x94, 0xe7, 0xca, 0xbf, 0x95, 0xe0, 0x5a, 0x74, 0x1f, 0x9a, 0x6c, 0xbc, 0x72,\n\t\t0xba, 0xa4, 0x5d, 0xeb, 0x53, 0xd5, 0xee, 0x07, 0x04, 0xdd, 0x87, 0xbb, 0x83, 0x9b, 0xcc, 0x54,\n\t\t0x1a, 0xcf, 0x87, 0xb1, 0xb2, 0x54, 0x85, 0xb5, 0xf8, 0xe1, 0x69, 0x32, 0xa1, 0xc2, 0x04, 0x59,\n\t\t0x62, 0xd9, 0x30, 0x1d, 0x8a, 0xb5, 0x86, 0x66, 0xca, 0xb9, 0xf2, 0xdf, 0x0b, 0xb0, 0x15, 0x37,\n\t\t0x8e, 0xbd, 0x68, 0x92, 0x76, 0x68, 0xda, 0x3d, 0xd8, 0x49, 0x2a, 0x11, 0xb7, 0xdd, 0xa8, 0x5d,\n\t\t0x07, 0xf0, 0x60, 0x0a, 0xae, 0x69, 0x3c, 0x55, 0x8c, 0x0a, 0x7b, 0x8e, 0x40, 0xb2, 0x84, 0x1e,\n\t\t0xc3, 0xa3, 0x29, 0x94, 0x43, 0xa5, 0x32, 0xf4, 0xf2, 0x60, 0xee, 0x50, 0x4c, 0x13, 0xeb, 0x87,\n\t\t0x4d, 0x53, 0x6b, 0xc8, 0x39, 0xa4, 0x81, 0x92, 0xa1, 0x20, 0x79, 0x25, 0xa4, 0xaa, 0xc9, 0xa3,\n\t\t0xaf, 0xe0, 0xb3, 0x2c, 0x3b, 0xc2, 0x94, 0xd1, 0x5f, 0x68, 0x38, 0x4e, 0x9d, 0x43, 0x5f, 0xc3,\n\t\t0xe7, 0x19, 0x54, 0xb1, 0xf3, 0x18, 0x77, 0x1e, 0x3d, 0x82, 0x2f, 0x32, 0xad, 0x57, 0x6b, 0xb8,\n\t\t0x62, 0xbd, 0x50, 0xf0, 0xf3, 0x24, 0x79, 0x01, 0xe9, 0xa0, 0x65, 0x6d, 0x2c, 0xfa, 0x97, 0x95,\n\t\t0xd2, 0x11, 0x62, 0xaa, 0xae, 0xcc, 0xe0, 0x45, 0x26, 0xc8, 0x50, 0xb3, 0x88, 0x9e, 0x80, 0x3a,\n\t\t0x9b, 0x2b, 0xa6, 0x2b, 0x5a, 0x42, 0xdf, 0x82, 0xf9, 0x6e, 0x51, 0xd5, 0xbe, 0x35, 0x35, 0x6c,\n\t\t0x28, 0x59, 0x9a, 0x01, 0x7d, 0x03, 0x5f, 0x65, 0x3a, 0x2d, 0xd9, 0x7f, 0x62, 0xf4, 0x02, 0xfa,\n\t\t0x02, 0x3e, 0x99, 0x42, 0x8f, 0xe7, 0xc8, 0x70, 0x36, 0xd4, 0x2b, 0x72, 0x11, 0x7d, 0x06, 0x07,\n\t\t0x53, 0x88, 0xbc, 0x0a, 0xad, 0x86, 0xa9, 0xab, 0xcf, 0x4f, 0xc2, 0xe5, 0xaa, 0xde, 0x30, 0xe5,\n\t\t0x65, 0xf4, 0x23, 0xf8, 0xbf, 0x29, 0xb4, 0xc1, 0x61, 0xd9, 0x1f, 0x1a, 0x8e, 0x95, 0x18, 0x83,\n\t\t0x35, 0xb1, 0x26, 0xaf, 0xcc, 0x10, 0x93, 0x86, 0xfe, 0x24, 0xdb, 0x73, 0xab, 0x48, 0x85, 0xc7,\n\t\t0x33, 0x95, 0x88, 0xfa, 0x54, 0xaf, 0x56, 0xd2, 0x95, 0xc8, 0xe8, 0x13, 0xd8, 0x9f, 0xa2, 0xe4,\n\t\t0xa8, 0x86, 0x55, 0x4d, 0x0c, 0x0f, 0x83, 0x26, 0xb1, 0x86, 0x3e, 0x87, 0x87, 0xd3, 0x48, 0x8a,\n\t\t0x5e, 0x65, 0x13, 0xe8, 0x28, 0x0f, 0xb1, 0x89, 0x66, 0xb6, 0xa3, 0xeb, 0x46, 0xbd, 0x69, 0x5a,\n\t\t0x0d, 0xfd, 0x3b, 0x4d, 0x5e, 0x67, 0x13, 0x4d, 0x66, 0xa4, 0x22, 0x5f, 0xc9, 0x1b, 0xe3, 0xcd,\n\t\t0x78, 0x6c, 0x93, 0x43, 0xdd, 0x50, 0xf0, 0x89, 0xbc, 0x99, 0x91, 0x7b, 0xe3, 0x8d, 0x2e, 0x91,\n\t\t0x42, 0x57, 0x67, 0x39, 0x8e, 0xa6, 0x60, 0xf5, 0x69, 0xdc, 0xe3, 0x5b, 0xec, 0xd6, 0xb9, 0xc3,\n\t\t0xbf, 0xcb, 0x8d, 0x8d, 0x5d, 0xf1, 0x16, 0x7f, 0x00, 0x0f, 0xc2, 0xb8, 0xa5, 0x64, 0xc1, 0x84,\n\t\t0x6e, 0x7f, 0x08, 0xff, 0x3f, 0x1b, 0x65, 0xb0, 0xae, 0x54, 0xb1, 0xa6, 0x54, 0x4e, 0x06, 0x2f,\n\t\t0x26, 0x52, 0xf9, 0x37, 0x39, 0x28, 0xab, 0xb6, 0xdb, 0x22, 0x9d, 0xe8, 0xff, 0x01, 0x53, 0xad,\n\t\t0x7c, 0x04, 0x5f, 0xcc, 0x50, 0xef, 0x13, 0xec, 0x3d, 0x86, 0xc6, 0xbb, 0x92, 0x9b, 0xc6, 0x73,\n\t\t0xa3, 0x76, 0x6c, 0x4c, 0x23, 0xc8, 0x12, 0x32, 0xe0, 0xd9, 0xbb, 0x2a, 0x1e, 0x73, 0xc9, 0x70,\n\t\t0xd2, 0xcc, 0x71, 0xa7, 0x34, 0x9c, 0x73, 0xfe, 0xcf, 0x91, 0xd9, 0x9c, 0x22, 0xd2, 0xf8, 0xdf,\n\t\t0x73, 0xca, 0xbb, 0x92, 0x67, 0x76, 0xca, 0xbb, 0x2a, 0x9e, 0xe6, 0x94, 0xc3, 0x9f, 0xc2, 0x56,\n\t\t0xcb, 0xeb, 0xa6, 0x7d, 0x6b, 0x3a, 0x5c, 0x8e, 0xdc, 0x53, 0xf7, 0x3d, 0xea, 0xd5, 0xa5, 0xef,\n\t\t0x0e, 0xce, 0x1d, 0x7a, 0xd1, 0x3f, 0xdd, 0x6b, 0x79, 0xdd, 0xfd, 0xf8, 0x8f, 0x52, 0x1e, 0x38,\n\t\t0xed, 0xce, 0xfe, 0xb9, 0x17, 0xfe, 0xc8, 0x45, 0xfc, 0x42, 0xe5, 0x91, 0xdd, 0x73, 0x5e, 0x1f,\n\t\t0x9c, 0x2e, 0x70, 0xd9, 0x27, 0xff, 0x0a, 0x00, 0x00, 0xff, 0xff, 0x00, 0x6a, 0x76, 0xd7, 0x61,\n\t\t0x23, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/tasklist.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xb4, 0x56, 0xdb, 0x6e, 0xdb, 0x36,\n\t\t0x18, 0x9e, 0x7c, 0x68, 0x9d, 0xdf, 0x4d, 0xa2, 0xb2, 0x49, 0x63, 0xbb, 0xed, 0xe6, 0xfa, 0xa2,\n\t\t0xc8, 0x8a, 0x4d, 0x46, 0xb2, 0x0d, 0x18, 0xb6, 0xa1, 0xab, 0x13, 0x1b, 0xad, 0x10, 0x27, 0x35,\n\t\t0x64, 0xb5, 0x43, 0x07, 0x0c, 0x02, 0x2d, 0xb1, 0x0e, 0x67, 0x49, 0x14, 0x44, 0xca, 0xae, 0x6f,\n\t\t0xf6, 0x18, 0xbb, 0xdb, 0x8b, 0xec, 0x1d, 0xf6, 0x4e, 0x03, 0x29, 0xd9, 0xf5, 0x41, 0x0d, 0xd6,\n\t\t0x8b, 0xdd, 0x99, 0xff, 0xc7, 0x8f, 0xdf, 0x7f, 0xb6, 0xa0, 0x95, 0x8c, 0x48, 0xdc, 0x76, 0xb1,\n\t\t0x47, 0x42, 0x97, 0xb4, 0x71, 0x44, 0xdb, 0xd3, 0x93, 0xb6, 0xc0, 0x7c, 0xe2, 0x53, 0x2e, 0x8c,\n\t\t0x28, 0x66, 0x82, 0xa1, 0x7b, 0xf2, 0x8e, 0x91, 0xdd, 0x31, 0x70, 0x44, 0x8d, 0xe9, 0x49, 0xe3,\n\t\t0xf3, 0x31, 0x63, 0x63, 0x9f, 0xb4, 0xd5, 0x95, 0x51, 0xf2, 0xae, 0xed, 0x25, 0x31, 0x16, 0x94,\n\t\t0x85, 0x29, 0xa9, 0xf1, 0xc5, 0x26, 0x2e, 0x68, 0x40, 0xb8, 0xc0, 0x41, 0x94, 0x5d, 0xd8, 0x7a,\n\t\t0x60, 0x16, 0xe3, 0x28, 0x22, 0x31, 0x4f, 0xf1, 0xd6, 0x6b, 0xa8, 0xd8, 0x98, 0x4f, 0xfa, 0x94,\n\t\t0x0b, 0x84, 0xa0, 0x14, 0xe2, 0x80, 0xd4, 0xb4, 0xa6, 0x76, 0xbc, 0x63, 0xa9, 0xdf, 0xe8, 0x3b,\n\t\t0x28, 0x4d, 0x68, 0xe8, 0xd5, 0x0a, 0x4d, 0xed, 0x78, 0xef, 0xf4, 0xb1, 0x91, 0xe3, 0xa4, 0xb1,\n\t\t0x78, 0xe0, 0x82, 0x86, 0x9e, 0xa5, 0xae, 0xb7, 0x30, 0xe8, 0x0b, 0xeb, 0x25, 0x11, 0xd8, 0xc3,\n\t\t0x02, 0xa3, 0x4b, 0x38, 0x08, 0xf0, 0x7b, 0x47, 0x86, 0xcd, 0x9d, 0x88, 0xc4, 0x0e, 0x27, 0x2e,\n\t\t0x0b, 0x3d, 0x25, 0x57, 0x3d, 0x7d, 0x68, 0xa4, 0x9e, 0x1a, 0x0b, 0x4f, 0x8d, 0x2e, 0x4b, 0x46,\n\t\t0x3e, 0x79, 0x83, 0xfd, 0x84, 0x58, 0x77, 0x03, 0xfc, 0x5e, 0x3e, 0xc8, 0x07, 0x24, 0x1e, 0x2a,\n\t\t0x5a, 0xeb, 0x35, 0xd4, 0x17, 0x12, 0x03, 0x1c, 0x0b, 0x2a, 0xb3, 0xb2, 0xd4, 0xd2, 0xa1, 0x38,\n\t\t0x21, 0xf3, 0x2c, 0x12, 0xf9, 0x13, 0x3d, 0x81, 0x7d, 0x36, 0x0b, 0x49, 0xec, 0x5c, 0x33, 0x2e,\n\t\t0x1c, 0x15, 0x67, 0x41, 0xa1, 0xbb, 0xca, 0xfc, 0x92, 0x71, 0x71, 0x85, 0x03, 0xd2, 0x9a, 0xc0,\n\t\t0xa1, 0xc9, 0x99, 0xaf, 0x92, 0xfc, 0x22, 0x66, 0x49, 0x74, 0x49, 0x44, 0x4c, 0x5d, 0x8e, 0xda,\n\t\t0x70, 0x10, 0x92, 0x59, 0xbe, 0xfb, 0x9a, 0x75, 0x37, 0x24, 0xb3, 0x75, 0x07, 0xd1, 0x63, 0xb8,\n\t\t0x13, 0x31, 0xdf, 0x27, 0xb1, 0xe3, 0xb2, 0x24, 0x14, 0x4a, 0xae, 0x68, 0x55, 0x53, 0xdb, 0xb9,\n\t\t0x34, 0xb5, 0xfe, 0x2a, 0xc1, 0xde, 0x22, 0x88, 0xa1, 0xc0, 0x22, 0xe1, 0xe8, 0x2b, 0x40, 0x23,\n\t\t0xec, 0x4e, 0x7c, 0x36, 0x4e, 0x69, 0xce, 0x35, 0x0d, 0x85, 0x12, 0x29, 0x5a, 0x7a, 0x86, 0x28,\n\t\t0xf2, 0x4b, 0x1a, 0x0a, 0xf4, 0x08, 0x20, 0x26, 0xd8, 0x73, 0x7c, 0x32, 0x25, 0x7e, 0xa6, 0xb0,\n\t\t0x23, 0x2d, 0x7d, 0x69, 0x40, 0x0f, 0x60, 0x07, 0xbb, 0x93, 0x0c, 0x2d, 0x2a, 0xb4, 0x82, 0xdd,\n\t\t0x49, 0x0a, 0x3e, 0x81, 0xfd, 0x18, 0x0b, 0xb2, 0x1a, 0x4b, 0x49, 0xc5, 0xb2, 0x2b, 0xcd, 0x1f,\n\t\t0xe2, 0xe8, 0xc2, 0xae, 0x0c, 0xda, 0xa1, 0x9e, 0x33, 0xf2, 0x99, 0x3b, 0xa9, 0x95, 0x55, 0xc1,\n\t\t0x9a, 0x1f, 0xed, 0x05, 0xb3, 0x7b, 0x26, 0xef, 0x59, 0x55, 0x49, 0x33, 0x3d, 0x75, 0x40, 0x53,\n\t\t0x38, 0xa2, 0x8b, 0xbc, 0x3a, 0x63, 0x99, 0x58, 0x27, 0x48, 0x33, 0x5b, 0xbb, 0xd5, 0x2c, 0x1e,\n\t\t0x57, 0x4f, 0x9f, 0xdd, 0xd8, 0x5b, 0x69, 0x76, 0x8c, 0xdc, 0xd2, 0xf4, 0x42, 0x11, 0xcf, 0xad,\n\t\t0x43, 0xfa, 0x49, 0x65, 0xbb, 0xfd, 0xb1, 0xb2, 0x1d, 0x40, 0x99, 0x04, 0x91, 0x98, 0xd7, 0x2a,\n\t\t0x4d, 0xed, 0xb8, 0x62, 0xa5, 0x87, 0x86, 0x80, 0xc6, 0xc7, 0xb5, 0x73, 0xda, 0xed, 0x39, 0x94,\n\t\t0xa7, 0xb2, 0x73, 0x55, 0x4d, 0xaa, 0xa7, 0x4f, 0x73, 0x83, 0xcb, 0x7d, 0xd1, 0x4a, 0x89, 0x3f,\n\t\t0x14, 0xbe, 0xd7, 0x5a, 0x3f, 0x43, 0x75, 0x25, 0xa1, 0xa8, 0x0e, 0x15, 0x2e, 0x70, 0x2c, 0x1c,\n\t\t0xea, 0x65, 0x1d, 0x71, 0x5b, 0x9d, 0x4d, 0x0f, 0x1d, 0xc2, 0x2d, 0x12, 0x7a, 0x12, 0x48, 0x9b,\n\t\t0xa0, 0x4c, 0x42, 0xcf, 0xf4, 0x5a, 0x7f, 0x6a, 0x00, 0x03, 0xd5, 0x70, 0x66, 0xf8, 0x8e, 0xa1,\n\t\t0x2e, 0xe8, 0x3e, 0xe6, 0xc2, 0xc1, 0xae, 0x4b, 0x38, 0x77, 0xe4, 0xb2, 0xc8, 0xc6, 0xaf, 0xb1,\n\t\t0x35, 0x7e, 0xf6, 0x62, 0x93, 0x58, 0x7b, 0x92, 0xd3, 0x51, 0x14, 0x69, 0x44, 0x0d, 0xa8, 0x50,\n\t\t0x8f, 0x84, 0x82, 0x8a, 0x79, 0x36, 0x43, 0xcb, 0x73, 0x5e, 0x53, 0x15, 0x73, 0x9a, 0xaa, 0xf5,\n\t\t0xb7, 0x06, 0xf5, 0xa1, 0xa0, 0xee, 0x64, 0xde, 0x7b, 0x4f, 0xdc, 0x44, 0x26, 0xa1, 0x23, 0x44,\n\t\t0x4c, 0x47, 0x89, 0x20, 0x1c, 0xbd, 0x00, 0x7d, 0xc6, 0xe2, 0x09, 0x89, 0x55, 0xdd, 0x1c, 0xb9,\n\t\t0x25, 0x33, 0x3f, 0x1f, 0xdd, 0xd8, 0x25, 0xd6, 0x5e, 0x4a, 0x5b, 0xae, 0x34, 0x1b, 0xea, 0xdc,\n\t\t0xbd, 0x26, 0x5e, 0xe2, 0x13, 0x47, 0x30, 0x27, 0xcd, 0x9e, 0x0c, 0x9b, 0x25, 0x22, 0x2b, 0x4d,\n\t\t0x7d, 0x7b, 0xf1, 0x64, 0x3b, 0xd6, 0xba, 0xbf, 0xe0, 0xda, 0x6c, 0x28, 0x99, 0x76, 0x4a, 0x6c,\n\t\t0x3d, 0x83, 0xbb, 0x5b, 0xab, 0x07, 0x7d, 0x09, 0xfa, 0x46, 0x83, 0xf3, 0x9a, 0xd6, 0x2c, 0x1e,\n\t\t0xef, 0x58, 0xfb, 0xeb, 0x9d, 0xc9, 0x5b, 0xff, 0x94, 0xe0, 0x68, 0xeb, 0x81, 0x73, 0x16, 0xbe,\n\t\t0xa3, 0x63, 0x54, 0x83, 0xdb, 0x53, 0x12, 0x73, 0xca, 0xc2, 0x45, 0x89, 0xb3, 0x23, 0x3a, 0x85,\n\t\t0x7b, 0x61, 0x12, 0x38, 0x6a, 0xde, 0xa3, 0x05, 0x8b, 0xab, 0x28, 0xca, 0x67, 0x85, 0x9a, 0x6c,\n\t\t0xe6, 0x24, 0xb0, 0x08, 0xf6, 0x96, 0x4f, 0x72, 0xf4, 0x2d, 0x1c, 0x48, 0xce, 0x2c, 0xa6, 0xb2,\n\t\t0x26, 0x1f, 0x48, 0xc5, 0x25, 0x09, 0x85, 0x49, 0xf0, 0x8b, 0x84, 0x57, 0x58, 0x14, 0xf6, 0x37,\n\t\t0x55, 0x4a, 0x6a, 0x46, 0x9f, 0xdf, 0x98, 0xfd, 0x8d, 0x50, 0x8c, 0x75, 0x5f, 0xd2, 0x29, 0xdd,\n\t\t0x8b, 0xd7, 0x1d, 0xf4, 0x41, 0xdf, 0x72, 0xae, 0xac, 0xb4, 0x3a, 0x9f, 0xa4, 0xb5, 0x11, 0x42,\n\t\t0x2a, 0xb6, 0x3f, 0x5b, 0xb7, 0x36, 0x28, 0xdc, 0xcb, 0x71, 0x6a, 0x75, 0x7c, 0xcb, 0xe9, 0xf8,\n\t\t0xfe, 0xb4, 0x3e, 0xbe, 0x4f, 0xfe, 0x9b, 0x2f, 0x2b, 0xa3, 0xdb, 0xf8, 0x1d, 0x0e, 0xf2, 0x7c,\n\t\t0xfa, 0x3f, 0xb4, 0x9e, 0xfe, 0x01, 0x77, 0x56, 0xff, 0x83, 0x51, 0x03, 0xee, 0xdb, 0x9d, 0xe1,\n\t\t0x85, 0xd3, 0x37, 0x87, 0xb6, 0x73, 0x61, 0x5e, 0x75, 0x1d, 0xf3, 0xea, 0x4d, 0xa7, 0x6f, 0x76,\n\t\t0xf5, 0xcf, 0x50, 0x1d, 0x0e, 0x37, 0xb0, 0xab, 0x57, 0xd6, 0x65, 0xa7, 0xaf, 0x6b, 0x39, 0xd0,\n\t\t0xd0, 0x36, 0xcf, 0x2f, 0xde, 0xea, 0x05, 0xf4, 0x10, 0x6a, 0x1b, 0x50, 0x6f, 0xf0, 0xb2, 0x77,\n\t\t0xd9, 0xb3, 0x3a, 0x7d, 0xbd, 0xf8, 0xd4, 0xfb, 0xa0, 0x6f, 0xcf, 0x23, 0xb2, 0xae, 0x6f, 0xbf,\n\t\t0x1d, 0xf4, 0x56, 0xf4, 0x1f, 0xc0, 0xd1, 0x06, 0xd6, 0xed, 0x9d, 0x9b, 0x43, 0xf3, 0xd5, 0x95,\n\t\t0xae, 0xe5, 0x80, 0x9d, 0x73, 0xdb, 0x7c, 0x63, 0xda, 0x6f, 0xf5, 0xc2, 0xd9, 0x6f, 0x70, 0xe4,\n\t\t0xb2, 0x20, 0x2f, 0x3b, 0x67, 0xbb, 0xcb, 0xf4, 0xc8, 0x19, 0x1e, 0x68, 0xbf, 0x9e, 0x8c, 0xa9,\n\t\t0xb8, 0x4e, 0x46, 0x86, 0xcb, 0x82, 0xf6, 0xea, 0xb7, 0xd7, 0xd7, 0xd4, 0xf3, 0xdb, 0x63, 0x96,\n\t\t0x7e, 0x0e, 0x65, 0x1f, 0x62, 0x3f, 0xe2, 0x88, 0x4e, 0x4f, 0x46, 0xb7, 0x94, 0xed, 0x9b, 0x7f,\n\t\t0x03, 0x00, 0x00, 0xff, 0xff, 0xea, 0xcc, 0x39, 0x04, 0xac, 0x09, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/service_worker.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe4, 0x5a, 0x5b, 0x6f, 0xdb, 0xc8,\n\t\t0xf5, 0x07, 0x7d, 0xf7, 0xf1, 0x4d, 0x1e, 0xe3, 0xef, 0xbf, 0xac, 0xdc, 0x1c, 0xa5, 0x49, 0xbc,\n\t\t0x6d, 0x2a, 0x37, 0xde, 0x6d, 0x36, 0x9b, 0x6c, 0x8a, 0xfa, 0x12, 0x23, 0x2e, 0x36, 0x5b, 0x2f,\n\t\t0xa3, 0x4d, 0x80, 0x2d, 0x10, 0x62, 0x44, 0x8e, 0xac, 0x81, 0x29, 0x8e, 0x42, 0x0e, 0xa5, 0xe8,\n\t\t0xa5, 0x0f, 0x7d, 0xdc, 0xf6, 0xa1, 0x40, 0xd1, 0xed, 0x4b, 0x81, 0x3c, 0xf7, 0x03, 0xf4, 0xcb,\n\t\t0xb4, 0x4f, 0x7d, 0xee, 0x67, 0x28, 0x50, 0x70, 0x66, 0x48, 0x51, 0xd2, 0x90, 0xba, 0xb4, 0x80,\n\t\t0x17, 0xe8, 0x9b, 0x39, 0xfc, 0x9d, 0xc3, 0x33, 0xe7, 0x77, 0xe6, 0xcc, 0x6f, 0xc6, 0x82, 0xbd,\n\t\t0xb0, 0x46, 0xfc, 0x7d, 0x1b, 0x3b, 0xc4, 0xb3, 0xc9, 0x3e, 0x6e, 0xd1, 0xfd, 0xf6, 0xc3, 0xfd,\n\t\t0x80, 0xf8, 0x6d, 0x6a, 0x13, 0xab, 0xc3, 0xfc, 0x4b, 0xe2, 0x57, 0x5a, 0x3e, 0xe3, 0x0c, 0x6d,\n\t\t0x45, 0xc8, 0x8a, 0x42, 0x56, 0x70, 0x8b, 0x56, 0xda, 0x0f, 0x4b, 0x37, 0x2f, 0x18, 0xbb, 0x70,\n\t\t0xc9, 0xbe, 0x80, 0xd4, 0xc2, 0xfa, 0xbe, 0x13, 0xfa, 0x98, 0x53, 0xe6, 0x49, 0xa3, 0xd2, 0xad,\n\t\t0xc1, 0xf7, 0x9c, 0x36, 0x49, 0xc0, 0x71, 0xb3, 0xa5, 0x00, 0x43, 0x0e, 0x3a, 0x3e, 0x6e, 0xb5,\n\t\t0x88, 0x1f, 0xa8, 0xf7, 0xbb, 0xba, 0xf8, 0x6c, 0xd6, 0x6c, 0x26, 0x9f, 0x28, 0xeb, 0x10, 0x0e,\n\t\t0xb1, 0x69, 0xd0, 0x0b, 0xe3, 0xb6, 0x0e, 0xd3, 0xa0, 0x01, 0x67, 0x7e, 0x37, 0x8e, 0x54, 0x07,\n\t\t0x79, 0x17, 0x92, 0x04, 0xa0, 0xfd, 0x0e, 0xc7, 0xc1, 0xa5, 0x4b, 0x03, 0x9e, 0x87, 0x89, 0xb2,\n\t\t0x58, 0x77, 0x59, 0x47, 0x62, 0xca, 0x7f, 0x35, 0xa0, 0x74, 0xce, 0x5c, 0xf7, 0x94, 0xf9, 0x27,\n\t\t0x2a, 0xca, 0x2a, 0x0e, 0x2e, 0x4d, 0xf2, 0x2e, 0x24, 0x01, 0x47, 0xdb, 0xb0, 0xe0, 0xb0, 0x26,\n\t\t0xa6, 0x5e, 0xd1, 0xd8, 0x35, 0xf6, 0x96, 0x4d, 0xf5, 0x84, 0x9e, 0xc0, 0x72, 0xf4, 0x31, 0x2b,\n\t\t0xfa, 0x5a, 0x71, 0x66, 0xd7, 0xd8, 0x5b, 0x39, 0xb8, 0x51, 0xd1, 0x50, 0x52, 0x89, 0x9c, 0x7d,\n\t\t0x41, 0x03, 0x6e, 0x2e, 0x71, 0xf5, 0x17, 0x2a, 0xc1, 0x12, 0x75, 0x88, 0xc7, 0x29, 0xef, 0x16,\n\t\t0x67, 0x85, 0xd7, 0xe4, 0x19, 0xdd, 0x87, 0x8d, 0x1a, 0xf5, 0xb0, 0xdf, 0xb5, 0xec, 0x06, 0xb1,\n\t\t0x2f, 0x83, 0xb0, 0x59, 0x9c, 0x13, 0x90, 0x75, 0x39, 0x7c, 0xac, 0x46, 0xcb, 0xff, 0x5a, 0x82,\n\t\t0x6b, 0xda, 0xb8, 0x83, 0x16, 0xf3, 0x02, 0x82, 0x6e, 0x00, 0x88, 0x00, 0x39, 0xbb, 0x24, 0x32,\n\t\t0xf8, 0x55, 0x53, 0x84, 0x5c, 0x8d, 0x06, 0xd0, 0xd7, 0x80, 0xe2, 0x44, 0x58, 0xe4, 0x3d, 0xb1,\n\t\t0xc3, 0xa8, 0x4a, 0xd4, 0x44, 0xee, 0x69, 0x27, 0xf2, 0x46, 0xc1, 0x9f, 0xc7, 0x68, 0x73, 0xb3,\n\t\t0x33, 0x38, 0x84, 0x4e, 0x61, 0x2d, 0x71, 0xcb, 0xbb, 0x2d, 0x22, 0xe6, 0xb7, 0x72, 0x70, 0x3b,\n\t\t0xd7, 0x63, 0xb5, 0xdb, 0x22, 0xe6, 0x6a, 0x27, 0xf5, 0x84, 0x5e, 0xc3, 0x4e, 0xcb, 0x27, 0x6d,\n\t\t0xca, 0xc2, 0xc0, 0x0a, 0x38, 0xf6, 0x39, 0x71, 0x2c, 0xd2, 0x26, 0x1e, 0xb7, 0xa8, 0x23, 0x12,\n\t\t0xb2, 0x72, 0x70, 0xad, 0x22, 0x6b, 0xb5, 0x12, 0xd7, 0x6a, 0xe5, 0xcc, 0xe3, 0x8f, 0x3e, 0x79,\n\t\t0x8d, 0xdd, 0x90, 0x98, 0xdb, 0xb1, 0xf5, 0x2b, 0x69, 0xfc, 0x3c, 0xb2, 0x3d, 0x73, 0xd0, 0x1e,\n\t\t0x14, 0x86, 0xdc, 0xcd, 0xef, 0x1a, 0x7b, 0xb3, 0xe6, 0x7a, 0xd0, 0x8f, 0x2c, 0xc2, 0x22, 0xe6,\n\t\t0x9c, 0x34, 0x5b, 0xbc, 0xb8, 0x20, 0x00, 0xf1, 0x23, 0x7a, 0x00, 0xa8, 0x86, 0xed, 0x4b, 0x97,\n\t\t0x5d, 0x58, 0x36, 0x0b, 0x3d, 0x6e, 0x35, 0xa8, 0xc7, 0x8b, 0x8b, 0x02, 0x54, 0x50, 0x6f, 0x8e,\n\t\t0xa3, 0x17, 0x2f, 0xa8, 0xc7, 0xd1, 0x23, 0x58, 0x54, 0x95, 0x5d, 0x5c, 0x12, 0x71, 0x5f, 0xd7,\n\t\t0xe6, 0xe2, 0x85, 0xc4, 0x98, 0x31, 0x18, 0xdd, 0x83, 0x0d, 0x8f, 0xbc, 0xe7, 0x56, 0x0b, 0x5f,\n\t\t0x10, 0x45, 0xe2, 0xb2, 0x20, 0x71, 0x2d, 0x1a, 0x3e, 0xc7, 0x17, 0x44, 0x12, 0xf9, 0x18, 0xe6,\n\t\t0xc5, 0xb2, 0x28, 0x82, 0xf0, 0x5e, 0xce, 0xcd, 0xf4, 0x57, 0x11, 0xd2, 0x94, 0x06, 0xe8, 0x2d,\n\t\t0x5c, 0x1f, 0x2e, 0x01, 0xab, 0x57, 0xd5, 0x2b, 0xe3, 0x54, 0xf5, 0xce, 0x50, 0x0d, 0xc4, 0xaf,\n\t\t0xd0, 0x21, 0xac, 0x07, 0x76, 0x83, 0x38, 0xa1, 0x4b, 0x1c, 0x2b, 0x6a, 0x34, 0xc5, 0x55, 0xe1,\n\t\t0xb1, 0x34, 0x44, 0x5c, 0x35, 0xee, 0x42, 0xe6, 0x5a, 0x62, 0x11, 0x8d, 0xa1, 0x67, 0xb0, 0x1a,\n\t\t0xd3, 0x25, 0x1c, 0xac, 0x8d, 0x74, 0xb0, 0xa2, 0xf0, 0xc2, 0xfc, 0x0d, 0x2c, 0x46, 0x53, 0xa5,\n\t\t0x24, 0x28, 0xae, 0xef, 0xce, 0xee, 0xad, 0x1c, 0x3c, 0xd3, 0x4e, 0x26, 0x67, 0x19, 0x55, 0xbe,\n\t\t0x92, 0xf6, 0xcf, 0x3d, 0x1e, 0x91, 0xa3, 0xbc, 0xa1, 0x32, 0x08, 0x16, 0x7a, 0x35, 0xb4, 0x21,\n\t\t0xd8, 0x5f, 0x89, 0x06, 0xe3, 0x02, 0xaa, 0xc0, 0x16, 0x67, 0x1c, 0xbb, 0x96, 0x62, 0xd4, 0xaa,\n\t\t0x75, 0x39, 0x09, 0x8a, 0x05, 0x81, 0xdc, 0x14, 0xaf, 0x14, 0xe9, 0x47, 0xd1, 0x0b, 0xf4, 0x12,\n\t\t0x0a, 0x38, 0xe4, 0xcc, 0xb2, 0x99, 0x57, 0xa7, 0x17, 0xb2, 0xa8, 0x36, 0xc5, 0x7c, 0xef, 0x68,\n\t\t0xa3, 0x3e, 0x0c, 0x39, 0x3b, 0x16, 0xd8, 0xa8, 0xce, 0xcc, 0x75, 0xdc, 0xf7, 0x5c, 0x7a, 0x0b,\n\t\t0xab, 0xe9, 0xd8, 0x51, 0x01, 0x66, 0x2f, 0x49, 0x57, 0x75, 0xb1, 0xe8, 0xcf, 0xa8, 0x72, 0xda,\n\t\t0xd1, 0x62, 0x51, 0xab, 0x7e, 0xac, 0xca, 0x11, 0x06, 0x4f, 0x66, 0x1e, 0x1b, 0xe5, 0xbf, 0xcc,\n\t\t0xc3, 0x1d, 0x99, 0x25, 0x27, 0x9d, 0xb8, 0x63, 0xd6, 0x6c, 0xb9, 0x84, 0x13, 0x27, 0x6e, 0xa0,\n\t\t0x23, 0xfa, 0xd0, 0x53, 0x58, 0x8e, 0x37, 0x87, 0xa0, 0x38, 0x23, 0x48, 0xd2, 0x57, 0x5c, 0xfc,\n\t\t0x11, 0xb3, 0x87, 0x47, 0x3f, 0x82, 0xcd, 0x5e, 0xe1, 0xda, 0xcc, 0xe3, 0xe4, 0x3d, 0x17, 0x1d,\n\t\t0x67, 0xd5, 0x2c, 0x24, 0x2f, 0x8e, 0xe5, 0x78, 0x5f, 0xd7, 0x9d, 0x1b, 0xe8, 0xba, 0xbf, 0x82,\n\t\t0xcd, 0x80, 0x53, 0xfb, 0xb2, 0x6b, 0x61, 0xce, 0x7d, 0x5a, 0x0b, 0x23, 0xa6, 0xe6, 0x45, 0x5a,\n\t\t0x2a, 0xda, 0x68, 0x5e, 0x09, 0x74, 0x52, 0xf3, 0x87, 0x89, 0x95, 0x59, 0x90, 0x8e, 0x7a, 0x23,\n\t\t0xe8, 0x53, 0x28, 0xfa, 0x84, 0x87, 0xbe, 0x67, 0x79, 0xa4, 0x63, 0xc5, 0xd1, 0x8b, 0x85, 0x26,\n\t\t0x5a, 0xcb, 0x92, 0xf9, 0x7f, 0xf2, 0xfd, 0x97, 0xa4, 0x93, 0x4e, 0x25, 0x3a, 0x82, 0x9b, 0x75,\n\t\t0xe6, 0xdb, 0xc4, 0xb2, 0x7d, 0x82, 0x39, 0xd1, 0x98, 0x2f, 0x0a, 0xf3, 0x92, 0x40, 0x1d, 0x0b,\n\t\t0xd0, 0xa0, 0x0f, 0xcd, 0x7e, 0xb2, 0xa4, 0xdb, 0x4f, 0x10, 0x83, 0x35, 0xd1, 0x16, 0x2c, 0x9f,\n\t\t0x04, 0xa1, 0xcb, 0x83, 0xe2, 0xb2, 0x20, 0xe3, 0x17, 0xda, 0xe9, 0x8f, 0x41, 0x7c, 0x45, 0x56,\n\t\t0x8c, 0x74, 0x26, 0x97, 0xcf, 0xea, 0xbb, 0xd4, 0x50, 0x89, 0xc2, 0xe6, 0x10, 0x44, 0x53, 0xa5,\n\t\t0x3f, 0xeb, 0xaf, 0xd2, 0xbd, 0x31, 0xaa, 0x54, 0x38, 0x4c, 0xd7, 0xea, 0x87, 0x59, 0xf8, 0x41,\n\t\t0x7e, 0xc8, 0x6a, 0xd3, 0xfc, 0x1a, 0xd6, 0xfa, 0x13, 0x6c, 0x88, 0x8f, 0xfe, 0x64, 0xd2, 0xb6,\n\t\t0x61, 0xae, 0x3a, 0x69, 0x12, 0x3e, 0x18, 0x70, 0x13, 0xdb, 0x9c, 0xb6, 0x29, 0xa7, 0x24, 0xb0,\n\t\t0x38, 0xb3, 0x1c, 0x1a, 0xb4, 0x30, 0xb7, 0x1b, 0x96, 0xcb, 0x6c, 0xec, 0xba, 0x5d, 0x55, 0xfa,\n\t\t0xdf, 0x4c, 0x91, 0x6d, 0xd5, 0xa8, 0x0e, 0x13, 0xff, 0x55, 0x76, 0xa2, 0xbc, 0x7f, 0x21, 0x9d,\n\t\t0xcb, 0xec, 0x5f, 0xc3, 0xd9, 0x88, 0xd2, 0xaf, 0x61, 0x77, 0x94, 0x03, 0x0d, 0x37, 0x27, 0xfd,\n\t\t0xdc, 0xe8, 0x97, 0x8a, 0xf2, 0xdb, 0x15, 0xbe, 0x62, 0xc7, 0x67, 0x5e, 0x9d, 0xa5, 0x19, 0xfa,\n\t\t0xcd, 0x0c, 0xec, 0x6a, 0xa6, 0x79, 0x8a, 0xa9, 0x3b, 0x76, 0x2b, 0x39, 0x82, 0x79, 0x1b, 0x87,\n\t\t0x81, 0x8c, 0x66, 0xfd, 0xe0, 0x41, 0x6e, 0x1b, 0xe9, 0x79, 0x3f, 0x8e, 0x6c, 0x4c, 0x69, 0x1a,\n\t\t0xed, 0xd6, 0x0e, 0xe1, 0x98, 0xba, 0x81, 0x52, 0x2e, 0xfa, 0xdd, 0xfa, 0x1c, 0x77, 0x5d, 0x86,\n\t\t0x1d, 0x33, 0x06, 0xe7, 0x36, 0x17, 0xcd, 0x12, 0x9c, 0xd7, 0x4a, 0xba, 0x3b, 0x70, 0x3b, 0x27,\n\t\t0x07, 0x92, 0xe7, 0xf2, 0x3f, 0x7a, 0x7a, 0x35, 0xce, 0xec, 0x55, 0xea, 0xd5, 0x57, 0x80, 0x12,\n\t\t0xbf, 0x56, 0x93, 0x70, 0xec, 0x60, 0x8e, 0x95, 0x42, 0xbb, 0x9b, 0xfb, 0x81, 0x97, 0x0a, 0x6c,\n\t\t0x16, 0xf8, 0xc0, 0x48, 0xf9, 0x6f, 0x3d, 0x6d, 0xdb, 0x3f, 0xc7, 0x2b, 0xd5, 0xb6, 0xb7, 0x60,\n\t\t0x45, 0x2d, 0xa1, 0x6e, 0xb4, 0xe5, 0xcb, 0x4c, 0x40, 0x3c, 0x74, 0xe6, 0x44, 0xe2, 0x37, 0x01,\n\t\t0x08, 0xf1, 0x3b, 0x97, 0x23, 0x7e, 0x93, 0x89, 0x09, 0xf1, 0x8b, 0x53, 0x4f, 0xe8, 0x00, 0xe6,\n\t\t0xa9, 0xd7, 0x0a, 0xb9, 0xda, 0x81, 0xf2, 0x4b, 0x50, 0x42, 0x35, 0x62, 0x6b, 0xe1, 0x3f, 0x15,\n\t\t0x5b, 0x8b, 0x93, 0x89, 0xad, 0x2a, 0xec, 0xc4, 0xfe, 0xa2, 0x0e, 0x67, 0xbb, 0x2c, 0x20, 0xc2,\n\t\t0x11, 0x0b, 0xb9, 0x92, 0xbe, 0x3b, 0x43, 0xbe, 0x4e, 0xd4, 0xf9, 0xd4, 0xdc, 0x8e, 0x6d, 0xab,\n\t\t0xec, 0x38, 0xb2, 0xac, 0x4a, 0x43, 0xf4, 0x25, 0x6c, 0x8b, 0x8f, 0x0c, 0xbb, 0x5c, 0x1e, 0xe5,\n\t\t0x72, 0x4b, 0x18, 0x0e, 0xf8, 0x3b, 0x85, 0xcd, 0x06, 0xc1, 0x3e, 0xaf, 0x11, 0xcc, 0x13, 0x57,\n\t\t0x30, 0xca, 0x55, 0x21, 0xb1, 0x89, 0xfd, 0xa4, 0x8e, 0x07, 0x91, 0x4e, 0x9e, 0xef, 0x1d, 0x0f,\n\t\t0xde, 0xc2, 0xcd, 0x7e, 0x26, 0x2c, 0x56, 0xb7, 0x78, 0x83, 0x06, 0x56, 0x6c, 0x30, 0x5a, 0x06,\n\t\t0x97, 0xfa, 0x98, 0xf9, 0x65, 0xbd, 0xda, 0xa0, 0xc1, 0xa1, 0xf2, 0x7f, 0x96, 0x9e, 0x41, 0xdc,\n\t\t0xac, 0xd6, 0xc6, 0xa8, 0x94, 0xde, 0x24, 0x4e, 0x54, 0xd7, 0x1a, 0x3a, 0xad, 0xad, 0x4f, 0x77,\n\t\t0x5a, 0xbb, 0x0f, 0x1b, 0x89, 0x1f, 0xd5, 0x7d, 0x36, 0x64, 0x87, 0x8b, 0x87, 0x4f, 0x64, 0x17,\n\t\t0xfa, 0x18, 0x16, 0x1a, 0x04, 0x3b, 0xc4, 0x17, 0x32, 0x38, 0x3a, 0xc3, 0x69, 0xcf, 0x42, 0x02,\n\t\t0x62, 0x2a, 0xe8, 0x7f, 0x59, 0x18, 0x97, 0xbf, 0x33, 0x12, 0xe1, 0x9a, 0x6e, 0x2e, 0x93, 0x0a,\n\t\t0xd7, 0x4f, 0x60, 0x41, 0x2a, 0x25, 0xd5, 0x58, 0xf2, 0x73, 0xaf, 0xb0, 0x79, 0xad, 0xb4, 0x7c,\n\t\t0x2f, 0x11, 0x29, 0x19, 0x71, 0xa9, 0x1d, 0xe0, 0xb7, 0x33, 0x70, 0x3f, 0x0f, 0x78, 0xd4, 0x3d,\n\t\t0x3b, 0x19, 0xb5, 0x1d, 0x5c, 0x55, 0x8b, 0xec, 0x65, 0x6d, 0x6e, 0xca, 0xac, 0xcd, 0x0f, 0x64,\n\t\t0xed, 0x87, 0xb0, 0x37, 0x3a, 0x19, 0x2a, 0x73, 0x7f, 0x34, 0x12, 0x95, 0x91, 0x06, 0x4f, 0xa4,\n\t\t0x32, 0x1e, 0xc1, 0x62, 0x1d, 0x53, 0x37, 0xf4, 0x49, 0x2e, 0xf1, 0xa7, 0x12, 0x63, 0xc6, 0xe0,\n\t\t0x5c, 0xe6, 0x7b, 0x1b, 0xbf, 0x2e, 0x2c, 0x15, 0xfc, 0xb7, 0x33, 0xda, 0xfa, 0x90, 0xa8, 0xef,\n\t\t0x33, 0xe7, 0xa9, 0x8c, 0xcd, 0x4d, 0x9b, 0xb1, 0x41, 0xd6, 0xef, 0xc3, 0xdd, 0x11, 0xb9, 0x50,\n\t\t0x59, 0xfb, 0x93, 0x01, 0x65, 0x5d, 0x7d, 0x60, 0xcf, 0x26, 0x13, 0x91, 0x1e, 0x77, 0xda, 0x99,\n\t\t0x69, 0x65, 0xe1, 0x20, 0xe9, 0x77, 0xf5, 0x6d, 0x28, 0x09, 0x4c, 0x4d, 0xe0, 0x77, 0x33, 0x70,\n\t\t0x2f, 0x07, 0xf7, 0x3d, 0x27, 0x3e, 0xce, 0xda, 0xdc, 0xb4, 0x59, 0x1b, 0x24, 0xfe, 0x23, 0x7d,\n\t\t0xef, 0xeb, 0xcb, 0x46, 0x1f, 0xf5, 0x36, 0xf3, 0xfb, 0xa0, 0x2f, 0xe2, 0x4d, 0xf0, 0x0a, 0xa9,\n\t\t0x3f, 0x8f, 0xa8, 0xcf, 0x09, 0x4c, 0xc9, 0xdc, 0x8f, 0xa0, 0x60, 0x8b, 0x89, 0x59, 0xbe, 0x8c,\n\t\t0x95, 0x38, 0x22, 0xbe, 0x25, 0x73, 0x43, 0x8e, 0x9b, 0xf1, 0xb0, 0xaa, 0x92, 0x4c, 0x97, 0xff,\n\t\t0x6b, 0x55, 0x52, 0x8d, 0xaa, 0x64, 0x44, 0x36, 0x26, 0x4f, 0xf2, 0xdf, 0x7b, 0xdb, 0x87, 0xb8,\n\t\t0x68, 0x98, 0x46, 0x36, 0xfc, 0x7c, 0x40, 0x36, 0x8c, 0x7f, 0x9f, 0x11, 0x6f, 0x86, 0xaf, 0x61,\n\t\t0x4b, 0xfe, 0x23, 0xc8, 0x6a, 0x13, 0x5f, 0xdc, 0x54, 0x50, 0xaf, 0xce, 0xd4, 0x71, 0x35, 0x9b,\n\t\t0x28, 0xe2, 0xbf, 0x96, 0x70, 0x71, 0xf4, 0xde, 0xec, 0x0c, 0x0e, 0xa5, 0x36, 0x21, 0xdd, 0xe4,\n\t\t0x62, 0xed, 0x61, 0x40, 0xc9, 0x24, 0x01, 0xe1, 0xf2, 0x02, 0x2c, 0x39, 0x2c, 0x5e, 0x49, 0x6d,\n\t\t0x95, 0x6f, 0xc0, 0x35, 0x6d, 0x30, 0x2a, 0x58, 0x1f, 0xd6, 0xfb, 0xb5, 0x20, 0x7a, 0x00, 0x88,\n\t\t0x78, 0xb8, 0xe6, 0x12, 0x2b, 0xa5, 0x28, 0x15, 0xdd, 0x05, 0xf9, 0xa6, 0x67, 0x81, 0x0e, 0x60,\n\t\t0xbb, 0xc5, 0x5c, 0x97, 0xf8, 0x56, 0x07, 0x53, 0x79, 0x5a, 0xb0, 0xa8, 0x67, 0x35, 0x65, 0x27,\n\t\t0x98, 0x35, 0x91, 0x7c, 0xfb, 0x06, 0x53, 0x71, 0x2c, 0x38, 0xf3, 0x5e, 0x06, 0x07, 0xff, 0xdc,\n\t\t0x80, 0x65, 0x99, 0xee, 0xc3, 0xf3, 0x33, 0xf4, 0x1e, 0xb6, 0x34, 0xb7, 0x44, 0x68, 0x7f, 0xfc,\n\t\t0xfb, 0x24, 0x91, 0xd7, 0xd2, 0xc4, 0x17, 0x50, 0xe8, 0x0f, 0x06, 0x5c, 0xcf, 0xbb, 0x37, 0x42,\n\t\t0x8f, 0xa7, 0xbd, 0xd8, 0x2b, 0x7d, 0x36, 0xf5, 0x25, 0x15, 0xfa, 0xd6, 0x80, 0x9d, 0xcc, 0x2b,\n\t\t0x0e, 0xf4, 0xd3, 0x71, 0x1d, 0xf7, 0x09, 0xb6, 0xd2, 0xa3, 0x49, 0xcd, 0x54, 0x30, 0x3d, 0x72,\n\t\t0xd2, 0x5d, 0x22, 0x9f, 0x1c, 0xcd, 0x95, 0x4b, 0x3e, 0x39, 0xda, 0xfb, 0x8b, 0x14, 0x39, 0x5a,\n\t\t0xd1, 0x9a, 0x4f, 0x4e, 0xde, 0xa9, 0x25, 0x9f, 0x9c, 0xdc, 0x73, 0x05, 0xfa, 0xa0, 0x57, 0xc7,\n\t\t0x7d, 0x52, 0x1a, 0x7d, 0x3e, 0xb1, 0xff, 0xd4, 0xde, 0x53, 0x7a, 0x36, 0xa5, 0xf5, 0x70, 0xf9,\n\t\t0x0c, 0xcb, 0xbe, 0xfc, 0xf2, 0xc9, 0xd4, 0xfb, 0xf9, 0xe5, 0x93, 0xad, 0xc7, 0xd1, 0x77, 0x06,\n\t\t0xdc, 0xc8, 0xd5, 0xa0, 0xe8, 0xb3, 0xc9, 0x3c, 0xa7, 0x13, 0xf5, 0x64, 0x1a, 0x53, 0x15, 0xd8,\n\t\t0xef, 0x0d, 0xd1, 0x16, 0xb3, 0x34, 0x12, 0xfa, 0x74, 0x6c, 0x12, 0xfa, 0x45, 0x72, 0xe9, 0xf1,\n\t\t0xe4, 0x86, 0x2a, 0xa4, 0x3f, 0x1b, 0x70, 0x6b, 0x84, 0x6c, 0x43, 0x4f, 0x27, 0xf5, 0x9e, 0xce,\n\t\t0xd7, 0xe7, 0xd3, 0x19, 0xf7, 0x65, 0x2c, 0x53, 0x2f, 0x64, 0x66, 0x6c, 0x94, 0xb6, 0xcc, 0xcc,\n\t\t0xd8, 0x68, 0xed, 0x27, 0x33, 0x96, 0x2b, 0x61, 0x32, 0x33, 0x36, 0x8e, 0x0c, 0xcc, 0xcc, 0xd8,\n\t\t0x78, 0xaa, 0x29, 0xb5, 0x12, 0x87, 0xd5, 0x42, 0xfe, 0x4a, 0xcc, 0x94, 0x4e, 0xf9, 0x2b, 0x31,\n\t\t0x5b, 0x94, 0x44, 0x8d, 0x5c, 0x23, 0x03, 0x32, 0x1a, 0x79, 0xb6, 0x7a, 0xc9, 0x68, 0xe4, 0x39,\n\t\t0x0a, 0xe3, 0xa8, 0x06, 0xff, 0x6f, 0xb3, 0xa6, 0xce, 0xec, 0x08, 0x49, 0x15, 0xf0, 0x4a, 0xfe,\n\t\t0x76, 0xe7, 0xdc, 0x67, 0x9c, 0x9d, 0x1b, 0xdf, 0x3c, 0xbc, 0xa0, 0xbc, 0x11, 0xd6, 0x2a, 0x36,\n\t\t0x6b, 0xee, 0xa7, 0x7f, 0x9c, 0xf2, 0x63, 0xea, 0xb8, 0xfb, 0x17, 0x4c, 0xfe, 0xee, 0x46, 0xfd,\n\t\t0x52, 0xe5, 0x29, 0x6e, 0xd1, 0xf6, 0xc3, 0xda, 0x82, 0x18, 0xfb, 0xf8, 0xdf, 0x01, 0x00, 0x00,\n\t\t0xff, 0xff, 0x83, 0x33, 0x80, 0xed, 0x1b, 0x24, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/decision.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xdc, 0x59, 0xdf, 0x6e, 0xd4, 0xc6,\n\t\t0x17, 0xfe, 0x39, 0x7f, 0x36, 0xbb, 0x27, 0x1b, 0x20, 0x13, 0x08, 0x09, 0x04, 0x12, 0xf6, 0xa7,\n\t\t0x42, 0x21, 0xca, 0x6e, 0x12, 0x28, 0x42, 0x80, 0x50, 0xc9, 0x42, 0x44, 0x24, 0x08, 0x91, 0x13,\n\t\t0x40, 0xaa, 0x54, 0x59, 0x93, 0xf1, 0x24, 0x99, 0xc6, 0xeb, 0xd9, 0x8e, 0xc7, 0x09, 0x5b, 0xa9,\n\t\t0x12, 0x57, 0x6d, 0x6f, 0xfa, 0x00, 0x95, 0x7a, 0xd5, 0x9b, 0xb6, 0x37, 0xed, 0x6d, 0xab, 0x5e,\n\t\t0xf5, 0x11, 0xfa, 0x2c, 0x7d, 0x81, 0xca, 0xe3, 0xb1, 0x77, 0xb3, 0xf1, 0x7a, 0xed, 0x40, 0xb9,\n\t\t0xe8, 0x5d, 0x3c, 0x3e, 0xe7, 0x9b, 0xcf, 0x73, 0xce, 0x7e, 0xe7, 0xb3, 0x03, 0x15, 0x7f, 0x9b,\n\t\t0x8a, 0x1a, 0xc1, 0x36, 0x75, 0x09, 0xad, 0xe1, 0x26, 0xab, 0x1d, 0x2c, 0xd5, 0x6c, 0x4a, 0x98,\n\t\t0xc7, 0xb8, 0x5b, 0x6d, 0x0a, 0x2e, 0x39, 0x9a, 0x08, 0x62, 0xaa, 0x3a, 0xa6, 0x8a, 0x9b, 0xac,\n\t\t0x7a, 0xb0, 0x74, 0xe1, 0xf2, 0x2e, 0xe7, 0xbb, 0x0e, 0xad, 0xa9, 0x90, 0x6d, 0x7f, 0xa7, 0x66,\n\t\t0xfb, 0x02, 0xcb, 0x38, 0xe9, 0xc2, 0x5c, 0x12, 0x30, 0xe1, 0x8d, 0x46, 0x1c, 0x91, 0xb8, 0xb5,\n\t\t0xc4, 0xde, 0xbe, 0xc3, 0x3c, 0x99, 0x16, 0x73, 0xc8, 0xc5, 0xfe, 0x8e, 0xc3, 0x0f, 0xc3, 0x98,\n\t\t0xca, 0x37, 0xe3, 0x50, 0x7c, 0xa4, 0x19, 0xa3, 0xef, 0x0c, 0xb8, 0xe1, 0x91, 0x3d, 0x6a, 0xfb,\n\t\t0x0e, 0xb5, 0x30, 0x91, 0xec, 0x80, 0xc9, 0x96, 0x15, 0xa0, 0x5a, 0xd1, 0x53, 0x59, 0x58, 0x4a,\n\t\t0xc1, 0xb6, 0x7d, 0x49, 0xbd, 0x29, 0x63, 0xce, 0xf8, 0x70, 0x74, 0xf9, 0x5e, 0x35, 0xe1, 0x09,\n\t\t0xab, 0x9b, 0x1a, 0xe6, 0xa1, 0x46, 0xd9, 0xc2, 0xde, 0x7e, 0xb4, 0xcf, 0xc3, 0x18, 0xe2, 0xc9,\n\t\t0xff, 0xcc, 0xab, 0x5e, 0xa6, 0x48, 0xf4, 0x05, 0xcc, 0x7a, 0x12, 0x0b, 0x69, 0x49, 0xd6, 0xa0,\n\t\t0x22, 0x91, 0xcf, 0x80, 0xe2, 0xb3, 0x94, 0xcc, 0x27, 0xc8, 0xdd, 0x0a, 0x52, 0x13, 0x59, 0xcc,\n\t\t0x78, 0x29, 0xf7, 0xd1, 0x4f, 0x06, 0x04, 0xa7, 0xdf, 0x74, 0xa8, 0xa4, 0x56, 0x74, 0x80, 0x16,\n\t\t0x7d, 0x4d, 0x89, 0x1f, 0x14, 0x2d, 0x91, 0xcc, 0xa0, 0x22, 0xf3, 0x71, 0x22, 0x99, 0xba, 0xc6,\n\t\t0x7a, 0xa5, 0xa1, 0x1e, 0x47, 0x48, 0x89, 0xdc, 0xe6, 0x49, 0xf6, 0x70, 0xf4, 0xbd, 0x01, 0xf3,\n\t\t0x3b, 0x98, 0x39, 0x59, 0x69, 0x0e, 0x29, 0x9a, 0xf7, 0x13, 0x69, 0xae, 0x62, 0xe6, 0x64, 0xa3,\n\t\t0x78, 0x6d, 0x27, 0x5b, 0x28, 0xfa, 0xd9, 0x80, 0x45, 0x41, 0x3f, 0xf7, 0xa9, 0x27, 0x2d, 0x82,\n\t\t0x5d, 0x42, 0x9d, 0x0c, 0x7d, 0x36, 0x9c, 0x72, 0x94, 0x66, 0x08, 0x56, 0x57, 0x58, 0x7d, 0x9b,\n\t\t0x6d, 0x5e, 0x64, 0x0f, 0x47, 0x5f, 0xc2, 0x9c, 0xa6, 0xd8, 0xbb, 0xe5, 0x0a, 0x8a, 0xda, 0x72,\n\t\t0x72, 0x95, 0x55, 0x72, 0xef, 0x9e, 0xbb, 0x44, 0xd2, 0x02, 0xd0, 0x0f, 0x06, 0x2c, 0xe8, 0xfd,\n\t\t0x33, 0xd6, 0x72, 0x44, 0x91, 0x79, 0x90, 0x42, 0x26, 0x5b, 0x35, 0xaf, 0x93, 0xac, 0xc1, 0xe8,\n\t\t0x2f, 0x03, 0x1e, 0x74, 0xd5, 0x93, 0xbe, 0x96, 0x54, 0xb8, 0x38, 0x33, 0xeb, 0xa2, 0x62, 0xfd,\n\t\t0xac, 0x7f, 0x75, 0x1f, 0x6b, 0xe0, 0x6c, 0x0f, 0x71, 0x47, 0x9c, 0x30, 0x17, 0xbd, 0x31, 0xe0,\n\t\t0x8a, 0xa0, 0x84, 0x0b, 0xdb, 0x6a, 0x60, 0xb1, 0xdf, 0xa3, 0xf2, 0x25, 0x45, 0xfb, 0x66, 0x0f,\n\t\t0xda, 0x41, 0xf6, 0x33, 0x95, 0x9c, 0x48, 0xee, 0xb2, 0x48, 0x8d, 0x40, 0xbf, 0x1b, 0x70, 0x9b,\n\t\t0x70, 0x57, 0x32, 0xd7, 0xa7, 0x16, 0xf6, 0x2c, 0x97, 0x1e, 0x66, 0x3d, 0x4e, 0x50, 0xbc, 0x1e,\n\t\t0xf7, 0xd0, 0x9d, 0x10, 0xf2, 0xa1, 0xb7, 0x4e, 0x0f, 0xb3, 0x1d, 0xe3, 0x22, 0xc9, 0x99, 0x83,\n\t\t0x7e, 0x35, 0x60, 0x39, 0x54, 0x6a, 0xb2, 0xc7, 0x1c, 0x3b, 0x2b, 0xef, 0x51, 0xc5, 0x7b, 0xa5,\n\t\t0xb7, 0x78, 0xd7, 0x03, 0xb4, 0x6c, 0xa4, 0x17, 0xbc, 0x3c, 0x09, 0xe8, 0x0f, 0x03, 0x6e, 0x7b,\n\t\t0x6c, 0x37, 0xe8, 0xd9, 0xbc, 0xcd, 0x5b, 0x56, 0xac, 0x57, 0x93, 0x59, 0x2b, 0xc8, 0x7c, 0x5d,\n\t\t0xbb, 0xe4, 0xe5, 0x4d, 0x42, 0xbf, 0x19, 0xf0, 0x91, 0xdf, 0xf4, 0xa8, 0x90, 0x6d, 0xd2, 0x1e,\n\t\t0xc5, 0x82, 0xec, 0x75, 0x10, 0x4d, 0x24, 0x3f, 0x96, 0xd2, 0x2a, 0x2f, 0x14, 0x62, 0xb4, 0xff,\n\t\t0xa6, 0xc2, 0x6b, 0x6f, 0x9a, 0xdc, 0x2a, 0x7e, 0xce, 0x9c, 0x95, 0x32, 0x40, 0x9b, 0x4e, 0xe5,\n\t\t0xdb, 0x02, 0x5c, 0xcd, 0x66, 0x1b, 0xd0, 0x2c, 0x8c, 0xc6, 0x63, 0x83, 0xd9, 0xca, 0x88, 0x94,\n\t\t0x4c, 0x88, 0x96, 0xd6, 0x6c, 0xb4, 0x0a, 0x63, 0xed, 0xb9, 0xd2, 0x6a, 0x52, 0xed, 0x0d, 0xae,\n\t\t0x24, 0x3e, 0x6b, 0xbc, 0x59, 0xab, 0x49, 0xcd, 0x32, 0xee, 0xb8, 0x42, 0x93, 0x50, 0xb0, 0x79,\n\t\t0x03, 0x33, 0x57, 0xcd, 0xf3, 0x92, 0xa9, 0xaf, 0xd0, 0x5d, 0x28, 0xa9, 0x71, 0x15, 0xb8, 0x2d,\n\t\t0x3d, 0x43, 0x2f, 0x25, 0x62, 0x07, 0x0f, 0xf0, 0x94, 0x79, 0xd2, 0x2c, 0x4a, 0xfd, 0x17, 0x5a,\n\t\t0x86, 0x61, 0xe6, 0x36, 0x7d, 0xa9, 0xe7, 0xda, 0x4c, 0x62, 0xde, 0x06, 0x6e, 0x39, 0x1c, 0xdb,\n\t\t0x66, 0x18, 0x8a, 0xb6, 0x60, 0x3a, 0x36, 0x66, 0x92, 0x5b, 0xc4, 0xe1, 0x1e, 0x55, 0x63, 0x89,\n\t\t0xfb, 0x52, 0x0f, 0xa1, 0xe9, 0x6a, 0x68, 0x2a, 0xab, 0x91, 0xa9, 0xac, 0x3e, 0xd2, 0xa6, 0xd2,\n\t\t0x9c, 0x8c, 0x72, 0xb7, 0x78, 0x3d, 0xc8, 0xdc, 0x0a, 0x13, 0xbb, 0x51, 0xdb, 0xfe, 0x2a, 0x40,\n\t\t0x1d, 0xc9, 0x81, 0x1a, 0xbb, 0xab, 0x00, 0x75, 0x1d, 0x26, 0x35, 0x52, 0x37, 0xd1, 0x62, 0x3f,\n\t\t0xc8, 0x89, 0xd0, 0x86, 0x1d, 0x65, 0xb9, 0x0a, 0xe3, 0x7b, 0x14, 0x0b, 0xb9, 0x4d, 0x71, 0x9b,\n\t\t0x5d, 0xa9, 0x1f, 0xd4, 0x99, 0x38, 0x27, 0xc2, 0xa9, 0x43, 0x59, 0x50, 0x29, 0x5a, 0x56, 0x93,\n\t\t0x3b, 0x8c, 0xb4, 0xb4, 0xe2, 0xcc, 0xf5, 0x50, 0x70, 0x29, 0x5a, 0x1b, 0x2a, 0xce, 0x1c, 0x15,\n\t\t0xed, 0x0b, 0x74, 0x13, 0x0a, 0x7b, 0x14, 0xdb, 0x54, 0xe8, 0x9f, 0xfe, 0xc5, 0xc4, 0xf4, 0x27,\n\t\t0x2a, 0xc4, 0xd4, 0xa1, 0xe8, 0x16, 0x4c, 0x46, 0x43, 0xd2, 0xe1, 0x04, 0x3b, 0x96, 0xcd, 0xbc,\n\t\t0x26, 0x96, 0x64, 0x4f, 0xfd, 0x04, 0x8b, 0xe6, 0x59, 0x7d, 0xf7, 0x69, 0x70, 0xf3, 0x91, 0xbe,\n\t\t0x57, 0xf9, 0xda, 0x80, 0x99, 0x34, 0xdb, 0x8a, 0xa6, 0xa1, 0x18, 0x3a, 0x93, 0xf8, 0x27, 0x30,\n\t\t0xa2, 0xae, 0xd7, 0x6c, 0xf4, 0x14, 0xce, 0xc5, 0x35, 0xd8, 0x61, 0xa2, 0x5d, 0x82, 0x81, 0x7e,\n\t\t0xe7, 0x86, 0x74, 0x09, 0x56, 0x99, 0x88, 0x2a, 0x50, 0x21, 0x30, 0x9f, 0xc3, 0xb2, 0xa2, 0x5b,\n\t\t0x50, 0x10, 0xd4, 0xf3, 0x1d, 0xa9, 0xdf, 0x10, 0xd2, 0x3b, 0x5c, 0xc7, 0x56, 0x30, 0x5c, 0xcb,\n\t\t0x68, 0x38, 0xd1, 0x6d, 0x18, 0x09, 0x0c, 0xa7, 0x2f, 0x68, 0xea, 0x0e, 0xab, 0x61, 0x8c, 0x19,\n\t\t0x05, 0x57, 0xd6, 0x61, 0x3e, 0x87, 0x5f, 0xec, 0xab, 0x32, 0x95, 0xbb, 0x70, 0x29, 0xd5, 0xe4,\n\t\t0xa5, 0x54, 0xa8, 0x42, 0xe0, 0x7a, 0x66, 0x4f, 0x16, 0x3c, 0xb0, 0x4d, 0x25, 0x66, 0x8e, 0x97,\n\t\t0xe9, 0x48, 0xa3, 0xe0, 0xca, 0xdf, 0x06, 0xdc, 0x39, 0xa9, 0x87, 0xea, 0xd0, 0x3e, 0xe3, 0x88,\n\t\t0xf6, 0xbd, 0x00, 0x74, 0x7c, 0x3a, 0xea, 0xc6, 0xba, 0x9a, 0xc8, 0xeb, 0xd8, 0x6e, 0xe6, 0xf8,\n\t\t0x61, 0xf7, 0x12, 0x9a, 0x82, 0x91, 0xc0, 0x6b, 0x08, 0xee, 0x28, 0xad, 0x2d, 0x9b, 0xd1, 0x25,\n\t\t0xaa, 0xc2, 0x44, 0x97, 0x95, 0xe0, 0xae, 0xd3, 0x52, 0xb2, 0x5b, 0x34, 0xc7, 0x49, 0xe7, 0x98,\n\t\t0x7f, 0xee, 0x3a, 0xad, 0xca, 0x2f, 0x06, 0x5c, 0x4e, 0xb7, 0x60, 0x41, 0x69, 0xb5, 0xb7, 0x73,\n\t\t0x71, 0x83, 0x46, 0xa5, 0x0d, 0x97, 0xd6, 0x71, 0x83, 0x76, 0x9e, 0xf8, 0x40, 0x8e, 0x13, 0xef,\n\t\t0xd0, 0x87, 0xc1, 0xcc, 0xfa, 0x50, 0x79, 0x03, 0xb0, 0x98, 0xd7, 0x9b, 0x05, 0x23, 0x2e, 0x3e,\n\t\t0x0f, 0x35, 0xe2, 0x8c, 0x94, 0x11, 0x17, 0x01, 0x86, 0x23, 0xee, 0xb0, 0xe3, 0xea, 0xe8, 0x28,\n\t\t0x1b, 0x38, 0xe1, 0x28, 0x1b, 0xcc, 0x3e, 0xca, 0x30, 0xcc, 0xb5, 0x3d, 0x55, 0x8f, 0x41, 0x31,\n\t\t0xd4, 0x4f, 0xa5, 0x66, 0x62, 0x88, 0xcd, 0x84, 0x89, 0xf1, 0x0a, 0x2e, 0xaa, 0x47, 0xea, 0x81,\n\t\t0x3e, 0xdc, 0x0f, 0xfd, 0x7c, 0x90, 0x9d, 0x04, 0xfc, 0x1c, 0x26, 0xb7, 0x31, 0xd9, 0xe7, 0x3b,\n\t\t0x3b, 0x1a, 0x9b, 0xb9, 0x92, 0x8a, 0x03, 0xec, 0xf4, 0x9f, 0xc1, 0x67, 0x75, 0xa2, 0x82, 0x5d,\n\t\t0xd3, 0x69, 0xc7, 0x66, 0xd2, 0xc8, 0x49, 0x66, 0xd2, 0x1a, 0x94, 0x98, 0xcb, 0x24, 0xc3, 0x92,\n\t\t0x0b, 0x35, 0x63, 0x4f, 0x2d, 0xcf, 0xf7, 0xf7, 0xff, 0x6b, 0x51, 0x8a, 0xd9, 0xce, 0xee, 0x54,\n\t\t0xd6, 0x52, 0x0e, 0x65, 0x45, 0x26, 0x4c, 0x3a, 0x38, 0x78, 0x07, 0x0c, 0xc7, 0x44, 0x50, 0x5a,\n\t\t0x3d, 0x02, 0x20, 0x43, 0x67, 0x9c, 0x0d, 0x72, 0xeb, 0x71, 0xaa, 0xa9, 0x32, 0xd1, 0xff, 0x61,\n\t\t0x8c, 0x88, 0xa0, 0x47, 0xb4, 0xcd, 0x50, 0x03, 0xbb, 0x64, 0x96, 0x83, 0xc5, 0xc8, 0x27, 0x9e,\n\t\t0x6c, 0x1e, 0x2f, 0xc0, 0x50, 0x83, 0x36, 0xb8, 0x36, 0xc0, 0xd3, 0x89, 0x29, 0xcf, 0x68, 0x83,\n\t\t0x9b, 0x2a, 0x0c, 0x99, 0x30, 0x7e, 0xcc, 0x50, 0x4f, 0x9d, 0x52, 0xb9, 0x1f, 0x24, 0x3b, 0xff,\n\t\t0x2e, 0xeb, 0x6b, 0x9e, 0xf1, 0xba, 0x56, 0xd0, 0x7d, 0x28, 0x7f, 0xc6, 0xa4, 0xa4, 0x22, 0x6c,\n\t\t0xa4, 0xa9, 0xd3, 0xfd, 0xfa, 0x67, 0x34, 0x0c, 0x57, 0xed, 0x83, 0x5e, 0xc2, 0x84, 0x3a, 0x1a,\n\t\t0x7e, 0x40, 0x85, 0x83, 0x9b, 0x51, 0xf7, 0x9c, 0x51, 0xb5, 0x4f, 0xd6, 0xe0, 0xba, 0xe0, 0xee,\n\t\t0xf3, 0x30, 0x5c, 0xf7, 0xd0, 0x38, 0xe9, 0x5e, 0x42, 0xaf, 0x61, 0x56, 0x8d, 0x37, 0x6a, 0x11,\n\t\t0xc7, 0xf7, 0x14, 0x3b, 0xea, 0x50, 0xa2, 0xea, 0xa9, 0xf7, 0x18, 0x4f, 0xf9, 0xc8, 0xa6, 0xe6,\n\t\t0x29, 0xad, 0x87, 0xa9, 0x9b, 0x51, 0xa6, 0xde, 0x6e, 0x06, 0xa7, 0xdc, 0xad, 0xfc, 0x58, 0x82,\n\t\t0x85, 0x5c, 0xaf, 0x79, 0x3d, 0xc7, 0xd3, 0x2c, 0x8c, 0xc6, 0xba, 0xc8, 0x6c, 0xa5, 0x68, 0x25,\n\t\t0x13, 0xa2, 0xa5, 0xf0, 0xdd, 0xe0, 0xa8, 0x70, 0x0e, 0xbe, 0x03, 0xe1, 0x7c, 0x0f, 0xef, 0x00,\n\t\t0x59, 0x84, 0xb3, 0xf0, 0xaf, 0x0a, 0xe7, 0xc8, 0x89, 0x85, 0xf3, 0x25, 0x4c, 0x34, 0xb1, 0xa0,\n\t\t0xae, 0xd4, 0x88, 0xba, 0x99, 0x8a, 0x29, 0x0d, 0xbb, 0xa1, 0xe2, 0x15, 0x4a, 0xd4, 0xb0, 0xcd,\n\t\t0xee, 0xa5, 0x4e, 0xd3, 0x50, 0x3a, 0x6a, 0x1a, 0x08, 0x4c, 0x75, 0xb4, 0x81, 0x25, 0xa8, 0xdf,\n\t\t0xde, 0x16, 0xd4, 0xb6, 0x37, 0x52, 0x0b, 0xbe, 0x66, 0x9b, 0x41, 0x8a, 0xde, 0xfa, 0xdc, 0x61,\n\t\t0xd2, 0xf2, 0xbb, 0x79, 0xa5, 0x38, 0xa6, 0x73, 0xe5, 0x54, 0x9d, 0x1b, 0xcb, 0xaf, 0x73, 0xa7,\n\t\t0xde, 0x42, 0xe7, 0x4e, 0xbf, 0x9d, 0xce, 0xfd, 0xf7, 0x94, 0xea, 0xcf, 0x01, 0x58, 0xca, 0xfd,\n\t\t0x69, 0xe7, 0x7d, 0x9b, 0xe9, 0x59, 0x18, 0xd5, 0x5f, 0xb4, 0x94, 0xbf, 0x0d, 0x3f, 0x5e, 0x40,\n\t\t0xb8, 0xa4, 0xfc, 0x6d, 0x2c, 0x40, 0x43, 0xd9, 0x05, 0xa8, 0xe3, 0xc7, 0x36, 0x9c, 0xc9, 0xa1,\n\t\t0x17, 0x7a, 0x39, 0xf4, 0xaf, 0x0c, 0x58, 0xcc, 0xfb, 0x85, 0x29, 0xb9, 0x3d, 0x8d, 0xb7, 0x6a,\n\t\t0xcf, 0x95, 0x4f, 0xe1, 0x3c, 0xe1, 0x8d, 0xa4, 0xec, 0x95, 0xb1, 0x88, 0xc2, 0x46, 0x20, 0x73,\n\t\t0x1b, 0xc6, 0x27, 0x4b, 0xbb, 0x4c, 0xee, 0xf9, 0xdb, 0x55, 0xc2, 0x1b, 0xb5, 0xce, 0xff, 0xac,\n\t\t0x2d, 0x30, 0xdb, 0xa9, 0xed, 0xf2, 0xf0, 0x9f, 0x79, 0xfa, 0xdf, 0x6c, 0xf7, 0x70, 0x93, 0x1d,\n\t\t0x2c, 0x6d, 0x17, 0xd4, 0xda, 0xcd, 0x7f, 0x02, 0x00, 0x00, 0xff, 0xff, 0x3a, 0x34, 0x7d, 0x59,\n\t\t0x29, 0x1c, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x5c, 0x5d, 0x6c, 0x1c, 0x49,\n\t\t0xb5, 0xde, 0x9e, 0xb1, 0xc7, 0x9e, 0x33, 0x8e, 0x3d, 0x2e, 0x3b, 0x8e, 0xff, 0x12, 0x3b, 0x93,\n\t\t0x6c, 0xe2, 0x75, 0x9c, 0x71, 0xe2, 0x64, 0x93, 0x4d, 0xb2, 0x3f, 0xd7, 0x76, 0x6c, 0xed, 0x48,\n\t\t0xbe, 0x49, 0x6e, 0xc7, 0xc9, 0xde, 0x7b, 0xb5, 0xd2, 0xd0, 0xee, 0x2e, 0xc7, 0x8d, 0x67, 0xba,\n\t\t0x67, 0xbb, 0x6b, 0x3c, 0x31, 0x82, 0x27, 0x1e, 0x90, 0x10, 0x2b, 0x58, 0xad, 0x90, 0x58, 0x09,\n\t\t0x04, 0x42, 0x02, 0xb1, 0xfc, 0x68, 0x11, 0x08, 0xf1, 0xf7, 0x02, 0x48, 0x68, 0x91, 0x40, 0x0b,\n\t\t0x4f, 0xbc, 0xf0, 0x84, 0x84, 0x10, 0xfb, 0xc6, 0x03, 0xcb, 0x33, 0xea, 0xea, 0xea, 0x99, 0xe9,\n\t\t0x9e, 0xaa, 0xfe, 0x19, 0x3b, 0x59, 0xd0, 0xe6, 0x6d, 0xba, 0xfb, 0x9c, 0xd3, 0x5f, 0x55, 0x9d,\n\t\t0x73, 0xea, 0xd4, 0x39, 0xa7, 0x07, 0x4e, 0xd6, 0xb7, 0xb0, 0xb5, 0xa8, 0x2a, 0x1a, 0x36, 0x54,\n\t\t0xbc, 0xa8, 0xd4, 0xf4, 0xc5, 0xbd, 0x8b, 0x8b, 0x3b, 0xba, 0x4d, 0x4c, 0x6b, 0xbf, 0x58, 0xb3,\n\t\t0x4c, 0x62, 0xa2, 0x11, 0x87, 0xa4, 0xc8, 0x48, 0x8a, 0x4a, 0x4d, 0x2f, 0xee, 0x5d, 0x9c, 0x3c,\n\t\t0xf1, 0xc0, 0x34, 0x1f, 0x54, 0xf0, 0x22, 0x25, 0xd9, 0xaa, 0x6f, 0x2f, 0x6a, 0x75, 0x4b, 0x21,\n\t\t0xba, 0x69, 0xb8, 0x4c, 0x93, 0x33, 0xc1, 0xe7, 0x44, 0xaf, 0x62, 0x9b, 0x28, 0xd5, 0x1a, 0x23,\n\t\t0x98, 0xe5, 0xbd, 0x58, 0x35, 0xab, 0xd5, 0xa6, 0x88, 0x02, 0x8f, 0x82, 0x28, 0xf6, 0x6e, 0x45,\n\t\t0xb7, 0x49, 0x18, 0x4d, 0xc3, 0xb4, 0x76, 0xb7, 0x2b, 0x66, 0xc3, 0xa5, 0x29, 0xdc, 0x84, 0xbe,\n\t\t0x97, 0xdd, 0x01, 0xa1, 0x6b, 0x90, 0xc1, 0x7b, 0xd8, 0x20, 0xf6, 0xb8, 0x34, 0x9b, 0x9e, 0xcb,\n\t\t0x2d, 0x9d, 0x2c, 0x72, 0xc6, 0x56, 0x64, 0xd4, 0x6b, 0x0e, 0xa5, 0xcc, 0x18, 0x0a, 0xef, 0x5f,\n\t\t0x85, 0x81, 0xf6, 0x07, 0x68, 0x02, 0xfa, 0xe9, 0xa3, 0xb2, 0xae, 0x8d, 0x4b, 0xb3, 0xd2, 0x5c,\n\t\t0x5a, 0xee, 0xa3, 0xd7, 0x25, 0x0d, 0x5d, 0x03, 0x70, 0x1f, 0x39, 0x83, 0x1e, 0x4f, 0xcd, 0x4a,\n\t\t0x73, 0xb9, 0xa5, 0xc9, 0xa2, 0x3b, 0x23, 0x45, 0x6f, 0x46, 0x8a, 0x9b, 0xde, 0x8c, 0xc8, 0x59,\n\t\t0x4a, 0xed, 0x5c, 0xa3, 0x71, 0xe8, 0xdb, 0xc3, 0x96, 0xad, 0x9b, 0xc6, 0x78, 0xda, 0x15, 0xca,\n\t\t0x2e, 0xd1, 0x31, 0xe8, 0x73, 0x06, 0xef, 0xbc, 0xae, 0x87, 0x3e, 0xc9, 0x38, 0x97, 0x25, 0x0d,\n\t\t0x7d, 0x59, 0x82, 0x73, 0xde, 0x90, 0xcb, 0xf8, 0x21, 0x56, 0xeb, 0xce, 0x3a, 0x94, 0x6d, 0xa2,\n\t\t0x58, 0x04, 0x6b, 0x65, 0x17, 0x89, 0x42, 0x88, 0xa5, 0x6f, 0xd5, 0x09, 0xb6, 0xc7, 0x7b, 0x29,\n\t\t0x9e, 0xe7, 0xb9, 0x43, 0x7f, 0x85, 0xc9, 0x59, 0xf3, 0xc4, 0xdc, 0x75, 0xa5, 0xd0, 0x21, 0x2f,\n\t\t0x37, 0x65, 0xbc, 0xfc, 0x94, 0x7c, 0xb6, 0x11, 0x8f, 0x14, 0x7d, 0x5d, 0x82, 0xf3, 0x1c, 0x78,\n\t\t0xaa, 0x59, 0xad, 0x55, 0x30, 0x17, 0x60, 0x86, 0x02, 0x7c, 0x31, 0x1e, 0xc0, 0x55, 0x4f, 0x4e,\n\t\t0x27, 0xc4, 0x67, 0x1a, 0x71, 0x89, 0xd1, 0x5b, 0x12, 0xcc, 0x73, 0x40, 0x6e, 0x2b, 0x7a, 0x85,\n\t\t0x87, 0xb0, 0x8f, 0x22, 0xbc, 0x11, 0x0f, 0xe1, 0x3a, 0x15, 0xd2, 0x09, 0xef, 0x4c, 0x23, 0x16,\n\t\t0x25, 0xfa, 0x1a, 0x7f, 0x02, 0x1d, 0xdd, 0xd2, 0xca, 0x66, 0x9d, 0x74, 0xc2, 0xeb, 0xa7, 0xf0,\n\t\t0x5e, 0x88, 0x07, 0xcf, 0x51, 0x3b, 0xed, 0x76, 0x9d, 0x74, 0x02, 0x9c, 0x6b, 0xc4, 0xa4, 0x45,\n\t\t0x6f, 0x4a, 0x30, 0xa7, 0x61, 0x55, 0xb7, 0x29, 0x30, 0x47, 0x4b, 0x6d, 0x75, 0x07, 0x6b, 0x75,\n\t\t0xee, 0xe4, 0x65, 0x29, 0xba, 0x6b, 0x5c, 0x74, 0x37, 0x99, 0x90, 0x4d, 0xc5, 0xde, 0xbd, 0xeb,\n\t\t0x89, 0xe8, 0x44, 0x76, 0x5a, 0x8b, 0x41, 0x87, 0x5e, 0x97, 0xe0, 0x4c, 0x00, 0x95, 0xc8, 0x26,\n\t\t0x80, 0x62, 0xba, 0x1a, 0x8d, 0x49, 0x64, 0x0e, 0x05, 0x2d, 0x92, 0x8a, 0x33, 0x4b, 0x21, 0x46,\n\t\t0x90, 0x8b, 0x39, 0x4b, 0x21, 0xfa, 0xef, 0x9b, 0x25, 0xa1, 0xea, 0xbf, 0xd1, 0x81, 0x2a, 0x44,\n\t\t0xb3, 0x06, 0x28, 0xaa, 0xe7, 0x22, 0x51, 0x89, 0x95, 0xea, 0x94, 0x16, 0x4d, 0x86, 0x3e, 0x2b,\n\t\t0xc1, 0xd3, 0x7e, 0x4c, 0x22, 0x4b, 0x3c, 0x42, 0x01, 0x5d, 0x89, 0x04, 0x24, 0x32, 0xc2, 0x93,\n\t\t0x5a, 0x14, 0x11, 0x5d, 0x36, 0x45, 0x25, 0xfa, 0x9e, 0x4e, 0xf6, 0x23, 0x95, 0x7b, 0x30, 0x64,\n\t\t0xd9, 0x96, 0x99, 0x90, 0x28, 0xe5, 0x56, 0x62, 0xd0, 0x51, 0xe5, 0x0e, 0xa0, 0x12, 0x29, 0xf7,\n\t\t0x50, 0x88, 0x72, 0xfb, 0x30, 0x09, 0x95, 0x5b, 0x89, 0xa4, 0xe2, 0xcc, 0x52, 0x88, 0x72, 0xe7,\n\t\t0x63, 0xce, 0x52, 0x98, 0x72, 0x2b, 0x31, 0xe8, 0xa8, 0x22, 0xf9, 0x51, 0x89, 0x14, 0x69, 0x38,\n\t\t0x44, 0x91, 0xda, 0x21, 0x09, 0x15, 0x49, 0x89, 0x22, 0xa2, 0x96, 0xe6, 0x07, 0x13, 0x62, 0x69,\n\t\t0x28, 0xc4, 0xd2, 0xda, 0xf1, 0x84, 0x58, 0x9a, 0x12, 0x4d, 0x86, 0x1a, 0x70, 0xc2, 0x01, 0x61,\n\t\t0x89, 0xb5, 0x67, 0x84, 0x02, 0xb9, 0xc0, 0x05, 0xe2, 0x48, 0xb5, 0x84, 0x6a, 0x33, 0x45, 0xc4,\n\t\t0x8f, 0xd1, 0x6b, 0x30, 0xed, 0xbe, 0x78, 0x5b, 0xb7, 0x78, 0xaf, 0x1d, 0xa5, 0xaf, 0x2d, 0x8a,\n\t\t0x5f, 0xbb, 0xee, 0xf0, 0x75, 0xbe, 0x74, 0x82, 0x88, 0x1e, 0xa2, 0x6f, 0x4a, 0xb0, 0x18, 0x50,\n\t\t0x51, 0xc5, 0x50, 0x71, 0xa5, 0x6c, 0xe1, 0xd7, 0xea, 0xd8, 0xe6, 0x8e, 0xfe, 0x28, 0x85, 0xf1,\n\t\t0x52, 0xb4, 0xa6, 0x52, 0x49, 0xb2, 0x27, 0xa8, 0x13, 0xd7, 0xbc, 0x12, 0x9b, 0x1a, 0xfd, 0x50,\n\t\t0x82, 0xcb, 0x0c, 0x93, 0x07, 0x31, 0x9e, 0x12, 0x8f, 0x51, 0xb4, 0xab, 0x5c, 0xb4, 0xec, 0x6d,\n\t\t0xee, 0xab, 0xe3, 0x68, 0x74, 0xd1, 0x4a, 0xc4, 0x81, 0xbe, 0x20, 0xc1, 0x59, 0xde, 0xf4, 0xf2,\n\t\t0x80, 0x1e, 0x8b, 0xa9, 0xdd, 0xab, 0x4c, 0x42, 0x84, 0x76, 0x0b, 0xc8, 0xd0, 0x27, 0x60, 0xc6,\n\t\t0x55, 0x32, 0x31, 0x92, 0x71, 0x8a, 0xe4, 0xa2, 0x58, 0xcf, 0xc4, 0x10, 0x5c, 0x05, 0x16, 0xbd,\n\t\t0xfb, 0x33, 0x12, 0x9c, 0x66, 0x8b, 0xc7, 0x14, 0x5d, 0xb0, 0x68, 0x13, 0x14, 0xc1, 0xb3, 0x5c,\n\t\t0x04, 0xae, 0x70, 0x57, 0xdf, 0x05, 0xcb, 0x34, 0xab, 0x46, 0xd0, 0xa0, 0x4f, 0xc1, 0x6c, 0x55,\n\t\t0xb1, 0x76, 0xb1, 0x55, 0xb6, 0xb0, 0x6a, 0x5a, 0x1a, 0x0f, 0xc4, 0x24, 0x05, 0xb1, 0xc4, 0x05,\n\t\t0xf1, 0xdf, 0x94, 0x59, 0x66, 0xbc, 0x9d, 0x08, 0x8e, 0x57, 0xc3, 0x08, 0xd0, 0x57, 0x25, 0x58,\n\t\t0xe0, 0x9d, 0x4f, 0xf4, 0x07, 0x86, 0xc2, 0x9d, 0x90, 0xa9, 0x24, 0xe1, 0xeb, 0x5d, 0x26, 0x26,\n\t\t0x4e, 0xf8, 0x2a, 0xa0, 0x45, 0xdf, 0x90, 0xa0, 0xc8, 0x8b, 0xb0, 0xb1, 0x55, 0xd5, 0x0d, 0x85,\n\t\t0xeb, 0x17, 0xa6, 0x43, 0xfc, 0x42, 0x67, 0x88, 0xdd, 0x14, 0xc4, 0xf1, 0x0b, 0x8d, 0xd8, 0xd4,\n\t\t0xe8, 0x47, 0x12, 0x5c, 0xe6, 0x1d, 0xa5, 0x22, 0xbd, 0xd8, 0x71, 0x8a, 0xf6, 0x66, 0xcc, 0x13,\n\t\t0x55, 0x94, 0x2b, 0x5b, 0x6c, 0x24, 0x63, 0x11, 0x69, 0x80, 0xd8, 0x28, 0x4f, 0x24, 0xd1, 0x00,\n\t\t0xb1, 0x81, 0xce, 0x35, 0x62, 0xd2, 0xa2, 0xbf, 0x48, 0xb0, 0x16, 0xf0, 0xb8, 0xf8, 0x21, 0xc1,\n\t\t0x96, 0xa1, 0x54, 0xca, 0x1c, 0xe4, 0xba, 0xa1, 0x13, 0x9d, 0xaf, 0x18, 0x33, 0x14, 0xfa, 0xdd,\n\t\t0x68, 0x17, 0xbc, 0xc6, 0xe4, 0x77, 0x8c, 0xa7, 0xe4, 0x09, 0xef, 0x1c, 0xd0, 0x8b, 0xd6, 0x81,\n\t\t0x24, 0xa0, 0x3f, 0x49, 0xb0, 0x92, 0x60, 0x98, 0x22, 0x8f, 0x35, 0x4b, 0xc7, 0x78, 0xe7, 0x00,\n\t\t0x63, 0x14, 0x39, 0xb3, 0x1b, 0x56, 0xf7, 0xec, 0xe8, 0x3d, 0x09, 0x5e, 0x08, 0x1b, 0x4e, 0xb4,\n\t\t0x9d, 0x9c, 0xa4, 0x03, 0xdb, 0xe0, 0x0e, 0x4c, 0x08, 0x26, 0xd2, 0x5e, 0xae, 0xe2, 0xee, 0x58,\n\t\t0x69, 0x1c, 0xc0, 0x4d, 0x9d, 0x18, 0x44, 0x37, 0xea, 0x58, 0x2b, 0x2b, 0x76, 0xd9, 0xc0, 0x8d,\n\t\t0xce, 0x71, 0x14, 0x42, 0xe2, 0x00, 0x4e, 0x06, 0x85, 0x89, 0x5b, 0xb6, 0x6f, 0xe1, 0x06, 0x27,\n\t\t0x0e, 0x68, 0x24, 0xe2, 0x40, 0xbf, 0x92, 0xe0, 0x1a, 0x8d, 0x26, 0xcb, 0xea, 0x8e, 0x5e, 0xd1,\n\t\t0x12, 0xda, 0xcf, 0x29, 0x0a, 0xfd, 0x65, 0x2e, 0x74, 0x1a, 0x4a, 0xae, 0x3a, 0x42, 0x93, 0x18,\n\t\t0xcd, 0x25, 0x3b, 0x39, 0x1b, 0xfa, 0xa9, 0x04, 0x57, 0x22, 0x06, 0x21, 0xb2, 0x8e, 0xd3, 0x74,\n\t\t0x04, 0x6b, 0x49, 0x47, 0x20, 0x32, 0x89, 0x0b, 0x76, 0x42, 0x1e, 0xf4, 0x5d, 0x09, 0x2e, 0x0a,\n\t\t0x51, 0x0b, 0xe3, 0xfc, 0xa7, 0x29, 0xec, 0x65, 0x7e, 0x18, 0xc2, 0x7d, 0xbb, 0x30, 0xf0, 0x5f,\n\t\t0x50, 0x13, 0xd0, 0xa3, 0x1f, 0x48, 0x70, 0x49, 0x08, 0x37, 0xe4, 0x10, 0x79, 0x26, 0x44, 0xc9,\n\t\t0xf9, 0x80, 0x43, 0x8e, 0x93, 0x45, 0x35, 0x11, 0x07, 0x7a, 0x5b, 0x82, 0x0b, 0x89, 0x35, 0xe3,\n\t\t0x2c, 0x45, 0xfc, 0x5f, 0x09, 0x10, 0x8b, 0x94, 0xe2, 0x9c, 0x9a, 0x40, 0x1f, 0xde, 0x91, 0x60,\n\t\t0x49, 0x3c, 0xc1, 0xc2, 0x4d, 0x78, 0x8e, 0xa2, 0x5d, 0x49, 0x32, 0xbf, 0xc2, 0x9d, 0xf8, 0xbc,\n\t\t0x9a, 0x84, 0x01, 0x7d, 0x3f, 0x4c, 0x25, 0x42, 0x0e, 0xcd, 0xcf, 0x24, 0x86, 0x2c, 0x3e, 0x3e,\n\t\t0x0b, 0x20, 0x8b, 0x0e, 0xd2, 0x4e, 0x6c, 0x26, 0x86, 0x1c, 0x12, 0x49, 0xce, 0x87, 0xc4, 0x66,\n\t\t0x02, 0xcc, 0x21, 0xe1, 0xe4, 0xa2, 0x9a, 0x8c, 0x85, 0x6e, 0x9a, 0x6e, 0x28, 0xde, 0x6d, 0xc4,\n\t\t0x73, 0x2e, 0x64, 0xd3, 0x74, 0x23, 0xee, 0x6e, 0x42, 0x9d, 0xab, 0x76, 0x77, 0xac, 0xe8, 0xd7,\n\t\t0x12, 0x5c, 0x8f, 0x31, 0x20, 0x91, 0x8d, 0x2e, 0xd0, 0xd1, 0x94, 0xba, 0x19, 0x8d, 0xc8, 0x58,\n\t\t0x2f, 0xdb, 0x5d, 0xf0, 0xa1, 0x9f, 0x48, 0xf0, 0x6c, 0xd8, 0x00, 0xc4, 0xe7, 0xa7, 0xf3, 0x21,\n\t\t0x1b, 0x90, 0x10, 0x84, 0xf8, 0x1c, 0x75, 0x01, 0x27, 0xe4, 0xa1, 0x0e, 0xa7, 0x5e, 0xb3, 0xb1,\n\t\t0x45, 0x5a, 0xc0, 0x6d, 0xac, 0x58, 0xea, 0x4e, 0x1b, 0xcc, 0x4e, 0xdc, 0xc5, 0x10, 0xeb, 0xbd,\n\t\t0x47, 0xc5, 0x79, 0x08, 0xee, 0x52, 0x61, 0xad, 0x37, 0x72, 0xac, 0xb7, 0x9e, 0x84, 0x61, 0x65,\n\t\t0x00, 0xa0, 0x05, 0xa4, 0xf0, 0xe7, 0x21, 0x38, 0x1b, 0x77, 0xf7, 0x5a, 0x87, 0x23, 0xcd, 0x31,\n\t\t0x92, 0xfd, 0x1a, 0xa6, 0xb5, 0x40, 0x51, 0x65, 0xd1, 0x13, 0xba, 0xb9, 0x5f, 0xc3, 0xf2, 0x40,\n\t\t0xa3, 0xed, 0x0a, 0xbd, 0x0a, 0x47, 0x6b, 0x8a, 0xe5, 0xcc, 0x48, 0xbb, 0xd1, 0x6d, 0x9b, 0xac,\n\t\t0x7c, 0x38, 0xc7, 0x95, 0x77, 0x87, 0x72, 0xb4, 0xd9, 0xc4, 0xb6, 0x29, 0x8f, 0xd4, 0x3a, 0x6f,\n\t\t0xa2, 0xeb, 0x90, 0xa5, 0x19, 0x99, 0x8a, 0x6e, 0x13, 0x5a, 0x58, 0xcc, 0x2d, 0x1d, 0xe7, 0xa7,\n\t\t0x3c, 0x14, 0x7b, 0x77, 0x43, 0xb7, 0x89, 0xdc, 0x4f, 0xd8, 0x2f, 0xb4, 0x04, 0xbd, 0xba, 0x51,\n\t\t0xab, 0x13, 0x5a, 0x76, 0xcc, 0x2d, 0x4d, 0x0b, 0x90, 0xec, 0x57, 0x4c, 0x45, 0x93, 0x5d, 0x52,\n\t\t0xa4, 0xc0, 0x6c, 0x20, 0xe4, 0x28, 0x13, 0xb3, 0xac, 0x56, 0x4c, 0x1b, 0x53, 0xff, 0x6d, 0xd6,\n\t\t0x09, 0xab, 0x43, 0x4e, 0x74, 0xd4, 0x45, 0x6f, 0xb2, 0x4a, 0xb2, 0x3c, 0x8d, 0x7d, 0x73, 0xbf,\n\t\t0x69, 0xae, 0x3a, 0xfc, 0x9b, 0x2e, 0x3b, 0x7a, 0x05, 0xa6, 0x5a, 0x69, 0xef, 0x4e, 0xe9, 0x99,\n\t\t0x28, 0xe9, 0xc7, 0x88, 0x97, 0xcc, 0x0e, 0x08, 0xbe, 0x01, 0x93, 0xad, 0x08, 0xbb, 0x35, 0x0a,\n\t\t0xab, 0x6e, 0x94, 0x75, 0x8d, 0x96, 0xfe, 0xb2, 0xf2, 0xb1, 0x26, 0x45, 0x73, 0x9e, 0xe5, 0xba,\n\t\t0x51, 0xd2, 0x50, 0x09, 0xb2, 0xcc, 0x55, 0x9a, 0x16, 0xad, 0xc3, 0x0d, 0x2e, 0x9d, 0xe3, 0xbb,\n\t\t0x76, 0x26, 0x80, 0x86, 0xd0, 0x25, 0x8f, 0x45, 0x6e, 0x71, 0xa3, 0x12, 0x0c, 0xb7, 0x70, 0x38,\n\t\t0xee, 0xaa, 0x6e, 0x61, 0x56, 0x3c, 0xe3, 0xaf, 0xc1, 0xba, 0x4b, 0x23, 0xe7, 0x9b, 0x6c, 0xec,\n\t\t0x0e, 0x92, 0x61, 0xac, 0xa2, 0x38, 0x67, 0x3e, 0x37, 0x9c, 0xa1, 0xc3, 0xc1, 0x76, 0xbd, 0x42,\n\t\t0x58, 0xe1, 0x2b, 0x7c, 0x4d, 0x47, 0x1d, 0xde, 0xd5, 0x26, 0xab, 0x4c, 0x39, 0xd1, 0x35, 0x98,\n\t\t0x30, 0x2d, 0xfd, 0x81, 0xee, 0x3a, 0xda, 0xc0, 0x2c, 0xe5, 0xe8, 0x2c, 0x8d, 0x79, 0x04, 0x81,\n\t\t0x49, 0x9a, 0x84, 0x7e, 0x5d, 0xc3, 0x06, 0xd1, 0xc9, 0x3e, 0xad, 0x28, 0x65, 0xe5, 0xe6, 0x35,\n\t\t0xba, 0x04, 0x63, 0xdb, 0xba, 0x65, 0x93, 0x4e, 0x99, 0x47, 0x28, 0xe5, 0x08, 0x7d, 0x1a, 0x10,\n\t\t0xb8, 0x0a, 0x03, 0x16, 0x26, 0xd6, 0x7e, 0xb9, 0x66, 0x56, 0x74, 0x75, 0x9f, 0x55, 0x61, 0x66,\n\t\t0x05, 0x07, 0x54, 0x62, 0xed, 0xdf, 0xa1, 0x74, 0x72, 0xce, 0x6a, 0x5d, 0xa0, 0x71, 0xe8, 0x53,\n\t\t0x08, 0xc1, 0xd5, 0x1a, 0xa1, 0x15, 0x93, 0x5e, 0xd9, 0xbb, 0x44, 0xab, 0x30, 0x84, 0x1f, 0xd6,\n\t\t0x74, 0x57, 0x71, 0xdc, 0xa2, 0x7e, 0x3e, 0xb2, 0xa8, 0x3f, 0xd8, 0x62, 0xa1, 0x95, 0xfd, 0x53,\n\t\t0x70, 0x44, 0xb5, 0x1c, 0x6b, 0x60, 0x15, 0x1d, 0x5a, 0x71, 0xc8, 0xca, 0x03, 0xce, 0x4d, 0xaf,\n\t\t0xca, 0x83, 0xfe, 0x17, 0xa6, 0xdc, 0xd1, 0xfb, 0xab, 0x5f, 0x5b, 0x8a, 0xba, 0x6b, 0x6e, 0x6f,\n\t\t0xb3, 0xa2, 0x40, 0x88, 0x52, 0x8f, 0x53, 0xee, 0xf6, 0xc2, 0xd7, 0x8a, 0xcb, 0x8a, 0xce, 0x43,\n\t\t0x4f, 0x15, 0x57, 0x4d, 0x96, 0xce, 0x9f, 0xe0, 0x27, 0xfa, 0x70, 0xd5, 0x94, 0x29, 0x19, 0x92,\n\t\t0x61, 0xb8, 0xc3, 0x63, 0xb3, 0x9c, 0xfc, 0xd3, 0xfc, 0xbd, 0x31, 0xe0, 0x61, 0xe5, 0xbc, 0x1d,\n\t\t0xb8, 0x83, 0xee, 0xc1, 0x58, 0xcd, 0xc2, 0x7b, 0x65, 0xa5, 0x4e, 0x4c, 0x47, 0xff, 0x30, 0x29,\n\t\t0xd7, 0x4c, 0xdd, 0x20, 0x5e, 0x96, 0x5d, 0xb4, 0x5e, 0x36, 0x26, 0x77, 0x28, 0x9d, 0x3c, 0xe2,\n\t\t0xf0, 0x2f, 0xd7, 0x89, 0xd9, 0x76, 0x13, 0x5d, 0x82, 0xcc, 0x0e, 0x56, 0x34, 0x6c, 0xb1, 0xf4,\n\t\t0xf7, 0x14, 0xbf, 0xa9, 0x83, 0x92, 0xc8, 0x8c, 0x14, 0x6d, 0xc0, 0xa8, 0x3b, 0xd1, 0xad, 0x5a,\n\t\t0x1e, 0x5d, 0xd7, 0x63, 0x91, 0xeb, 0x8a, 0x28, 0x5f, 0xb3, 0x2e, 0x47, 0xd7, 0xf6, 0x93, 0x90,\n\t\t0xaf, 0x29, 0x16, 0xd1, 0xbd, 0xe3, 0xf9, 0xb6, 0xfe, 0x60, 0x7c, 0x9c, 0x76, 0x98, 0xfc, 0xcf,\n\t\t0x41, 0xda, 0x2c, 0x1c, 0xff, 0xee, 0x0a, 0x5d, 0xa5, 0x32, 0xd7, 0x0c, 0x62, 0xed, 0xcb, 0x43,\n\t\t0x35, 0xff, 0x5d, 0x74, 0x1c, 0xc0, 0x4b, 0xea, 0xe8, 0x1a, 0x4d, 0x27, 0x67, 0xe5, 0x2c, 0xbb,\n\t\t0x53, 0xd2, 0xd0, 0x7d, 0x18, 0xa1, 0x8a, 0x67, 0xee, 0x61, 0xab, 0xa2, 0xd4, 0x3c, 0x1b, 0x99,\n\t\t0xa4, 0xce, 0xe9, 0x0c, 0xdf, 0x39, 0x59, 0xa6, 0x71, 0xdb, 0x25, 0x67, 0x96, 0x32, 0xac, 0x06,\n\t\t0x6f, 0xa1, 0x87, 0x30, 0x43, 0x73, 0xf0, 0xb8, 0xac, 0x56, 0xea, 0x36, 0xc1, 0x56, 0xd9, 0xc6,\n\t\t0x15, 0xac, 0xd2, 0x39, 0x60, 0xef, 0x98, 0x0a, 0x49, 0xae, 0xd3, 0x34, 0x3f, 0x5e, 0x75, 0x59,\n\t\t0xef, 0x7a, 0x9c, 0xec, 0x75, 0xd3, 0x4a, 0xc8, 0xd3, 0xc9, 0x15, 0x18, 0xe5, 0xcd, 0x0c, 0xca,\n\t\t0x43, 0x7a, 0x17, 0xef, 0xd3, 0x1d, 0x38, 0x2b, 0x3b, 0x3f, 0xd1, 0x28, 0xf4, 0xee, 0x29, 0x95,\n\t\t0xba, 0xdb, 0x84, 0x93, 0x95, 0xdd, 0x8b, 0xeb, 0xa9, 0xe7, 0xa4, 0xc2, 0xdb, 0x12, 0x3c, 0x13,\n\t\t0xff, 0xb8, 0x77, 0x19, 0x32, 0xcc, 0x61, 0x4a, 0x31, 0x1c, 0x26, 0xa3, 0x45, 0xeb, 0x30, 0x1b,\n\t\t0x5e, 0xef, 0xd7, 0x35, 0x0a, 0x2c, 0x2d, 0x4f, 0x8b, 0x4b, 0xf5, 0x25, 0xad, 0xf0, 0x2d, 0x09,\n\t\t0xce, 0xc4, 0x8c, 0x1a, 0xaf, 0x40, 0x9f, 0xb7, 0x55, 0x48, 0x31, 0xb6, 0x0a, 0x8f, 0xf8, 0xd0,\n\t\t0xa0, 0x9a, 0x30, 0x17, 0xfb, 0xc8, 0xb4, 0x0a, 0x03, 0x6c, 0xb7, 0x6e, 0x45, 0x4e, 0x83, 0x02,\n\t\t0x2f, 0xc0, 0x36, 0x67, 0x1a, 0x38, 0xe5, 0x48, 0xeb, 0xa2, 0xf0, 0x3b, 0x09, 0x4e, 0xc7, 0xe9,\n\t\t0x1a, 0xf1, 0x87, 0x40, 0x52, 0xb2, 0x10, 0xe8, 0x16, 0x8c, 0x09, 0xc2, 0x8c, 0x54, 0x94, 0x47,\n\t\t0x1e, 0xb1, 0x39, 0x21, 0x46, 0xdb, 0x56, 0x93, 0xf6, 0x6d, 0x35, 0x85, 0xd7, 0x25, 0x28, 0x44,\n\t\t0x37, 0x9c, 0xa0, 0x05, 0x40, 0xc1, 0x26, 0x84, 0x66, 0x1b, 0x5a, 0xde, 0xf6, 0x4d, 0x41, 0x60,\n\t\t0xbf, 0x4d, 0x05, 0xf6, 0x5b, 0xbf, 0xf3, 0x48, 0x07, 0x9c, 0x47, 0xe1, 0xef, 0x81, 0xe9, 0x15,\n\t\t0x5a, 0x48, 0x32, 0x44, 0x73, 0x90, 0xf7, 0x27, 0xa2, 0x9a, 0xea, 0x35, 0x68, 0xb7, 0x8d, 0x38,\n\t\t0x80, 0x3d, 0x1d, 0xc0, 0x7e, 0x16, 0x86, 0xb6, 0x74, 0x43, 0xb1, 0xf6, 0xcb, 0xea, 0x0e, 0x56,\n\t\t0x77, 0xed, 0x7a, 0x95, 0xc6, 0xa8, 0x59, 0x79, 0xd0, 0xbd, 0xbd, 0xca, 0xee, 0xa2, 0x73, 0x30,\n\t\t0xec, 0x4f, 0x9f, 0xe2, 0x87, 0x6e, 0xfc, 0x39, 0x20, 0xe7, 0x71, 0x7b, 0x56, 0x13, 0x3f, 0x24,\n\t\t0x85, 0xef, 0xa4, 0xe1, 0x54, 0x8c, 0x5e, 0x96, 0x47, 0x36, 0xe2, 0xa0, 0x59, 0xa4, 0xbb, 0x30,\n\t\t0x0b, 0x74, 0x02, 0x72, 0x5b, 0x8a, 0x8d, 0xbd, 0xd8, 0xc9, 0x9d, 0x96, 0xac, 0x73, 0xcb, 0x8d,\n\t\t0x98, 0xa6, 0x01, 0x0c, 0xdc, 0xf0, 0x1e, 0xf7, 0xba, 0x13, 0x6b, 0xe0, 0x86, 0xfb, 0x74, 0x01,\n\t\t0xd0, 0xb6, 0x69, 0xed, 0x32, 0xa4, 0x5e, 0x43, 0x62, 0xc6, 0x1d, 0x9a, 0xf3, 0x84, 0x62, 0xbd,\n\t\t0xcf, 0x3a, 0x13, 0xc7, 0x1c, 0xe7, 0xa8, 0xd8, 0xa6, 0xc1, 0x82, 0x63, 0x76, 0x85, 0x6e, 0x42,\n\t\t0xaf, 0xaa, 0xd4, 0x6d, 0xcc, 0xe2, 0xe0, 0x62, 0xec, 0xae, 0xa1, 0x55, 0x87, 0x4b, 0x76, 0x99,\n\t\t0x03, 0x0a, 0x9a, 0x0d, 0x2a, 0xe8, 0xbb, 0x69, 0x38, 0x19, 0xd9, 0xe8, 0xf3, 0xc8, 0xd6, 0x6a,\n\t\t0xc5, 0x1b, 0xa2, 0xbb, 0x48, 0x0b, 0x31, 0xfb, 0x90, 0x7c, 0x03, 0x6c, 0x73, 0xd9, 0x3d, 0x49,\n\t\t0x5c, 0x76, 0xbb, 0x65, 0xf4, 0x06, 0x2c, 0x23, 0xb0, 0xfc, 0x99, 0xf0, 0xe5, 0xef, 0x8b, 0xb5,\n\t\t0xfc, 0xfd, 0x82, 0xe5, 0xe7, 0x58, 0x61, 0x96, 0x6b, 0x85, 0xfe, 0x95, 0x84, 0xe0, 0x4a, 0x7e,\n\t\t0x25, 0x03, 0xa7, 0xe3, 0xb4, 0x48, 0xa1, 0x19, 0xc8, 0x35, 0xfb, 0x0c, 0xd8, 0x2a, 0x66, 0x65,\n\t\t0xf0, 0x6e, 0x95, 0x34, 0xe7, 0x4c, 0xde, 0x6a, 0x44, 0x70, 0x4c, 0x28, 0x15, 0x72, 0x26, 0x6f,\n\t\t0xbe, 0x92, 0x9e, 0xc9, 0x95, 0xb6, 0x2b, 0x47, 0xb1, 0x35, 0xb3, 0xaa, 0xe8, 0x06, 0xf3, 0x3c,\n\t\t0xec, 0xca, 0xbf, 0x95, 0xf4, 0x74, 0x79, 0x9a, 0xce, 0xc4, 0x3f, 0x4d, 0x6f, 0xc2, 0x84, 0xa7,\n\t\t0xa3, 0x9d, 0x3b, 0x50, 0x5f, 0xd4, 0x0e, 0x34, 0xe6, 0xf1, 0x06, 0x36, 0xa1, 0x80, 0x54, 0xb6,\n\t\t0xc1, 0x31, 0xa9, 0xfd, 0x09, 0xa4, 0xba, 0x87, 0x68, 0x26, 0x55, 0xbc, 0x55, 0x66, 0xbb, 0xda,\n\t\t0x2a, 0xd7, 0x61, 0x78, 0x07, 0x2b, 0x16, 0xd9, 0xc2, 0x4a, 0x0b, 0x1d, 0x44, 0x89, 0xca, 0x37,\n\t\t0x79, 0x5a, 0x72, 0xa2, 0x03, 0x9c, 0x5c, 0x74, 0x80, 0xd3, 0x71, 0xd4, 0x1c, 0xe8, 0xe6, 0xa8,\n\t\t0xd9, 0x3a, 0xb2, 0x1c, 0x89, 0x7d, 0x64, 0x29, 0xfc, 0x4d, 0x82, 0x42, 0x74, 0xbb, 0xde, 0x63,\n\t\t0x0b, 0x0d, 0xda, 0x83, 0x98, 0x1e, 0xff, 0x79, 0xf9, 0x25, 0x18, 0xa0, 0xe9, 0x06, 0xcf, 0xad,\n\t\t0xf5, 0xc6, 0x70, 0x6b, 0x39, 0x87, 0x83, 0x5d, 0x14, 0xfe, 0x20, 0xf9, 0x5d, 0xc1, 0x21, 0xc7,\n\t\t0xe5, 0xfc, 0x29, 0x4a, 0x25, 0xd8, 0x0d, 0xd2, 0x91, 0xb1, 0x4a, 0x8f, 0x7f, 0x32, 0x0b, 0xbf,\n\t\t0x97, 0xe0, 0x64, 0x74, 0x0f, 0x55, 0xb7, 0xe1, 0xfb, 0x87, 0x31, 0xa2, 0x9f, 0xa7, 0xe0, 0x54,\n\t\t0x8c, 0x4e, 0x44, 0x67, 0x4c, 0x1a, 0x26, 0x8a, 0x5e, 0xb1, 0x63, 0x2d, 0x92, 0x47, 0xfc, 0xc8,\n\t\t0xc6, 0x14, 0x8c, 0xaf, 0x7a, 0xba, 0x89, 0xaf, 0x0e, 0xac, 0xe2, 0x5f, 0x94, 0x60, 0x3e, 0x7e,\n\t\t0x03, 0x61, 0x9c, 0x3d, 0xef, 0x70, 0x0e, 0x70, 0xef, 0x48, 0x90, 0xb0, 0x55, 0x30, 0x1a, 0xdb,\n\t\t0xa8, 0x17, 0x25, 0xb1, 0x53, 0xb8, 0x1b, 0xf7, 0xc4, 0x41, 0x9c, 0x8e, 0x81, 0xf8, 0xad, 0x80,\n\t\t0x1e, 0x8a, 0x8a, 0x8a, 0xdd, 0xea, 0xe1, 0x3a, 0xcc, 0x56, 0x14, 0xd2, 0xd6, 0x32, 0x13, 0x6c,\n\t\t0x20, 0x69, 0xcd, 0xac, 0x4b, 0xc7, 0x5b, 0x4a, 0x37, 0xaa, 0xe2, 0xe8, 0x73, 0x3a, 0x81, 0x3e,\n\t\t0xf7, 0x44, 0xda, 0x68, 0x20, 0x0e, 0x2c, 0xbc, 0x27, 0xc1, 0x54, 0x48, 0x93, 0x2e, 0x9a, 0x80,\n\t\t0x7e, 0xb7, 0x39, 0xb1, 0xb9, 0x6e, 0x7d, 0xf4, 0xba, 0xa4, 0xa1, 0x0d, 0x38, 0xda, 0xdc, 0xc8,\n\t\t0xb7, 0x75, 0x2b, 0xc1, 0x91, 0x17, 0xb1, 0x7d, 0x7c, 0x5d, 0xb7, 0x70, 0x92, 0xed, 0x37, 0xce,\n\t\t0x62, 0x7f, 0x0c, 0x26, 0x84, 0xdd, 0xbf, 0x61, 0xa3, 0x89, 0x1d, 0xd2, 0x17, 0xde, 0x95, 0x60,\n\t\t0x3a, 0xac, 0xf1, 0xf3, 0x50, 0xde, 0x72, 0x58, 0xf3, 0x11, 0xea, 0xa0, 0x7f, 0x2c, 0xc1, 0x6c,\n\t\t0x54, 0x03, 0x69, 0xd8, 0x68, 0x1e, 0xa9, 0xd9, 0x86, 0x6f, 0x96, 0x59, 0x48, 0xd8, 0xa7, 0x84,\n\t\t0x16, 0x61, 0x94, 0xb6, 0x42, 0x05, 0xab, 0x06, 0xee, 0x98, 0x86, 0x0d, 0xdc, 0x08, 0xd4, 0x0c,\n\t\t0x3a, 0x0a, 0x77, 0xa9, 0xee, 0x0a, 0x77, 0x4f, 0x4a, 0x6b, 0xf1, 0x4b, 0x6b, 0x71, 0x74, 0xa7,\n\t\t0x2f, 0x86, 0xee, 0xdc, 0x86, 0x31, 0x56, 0x12, 0x61, 0x18, 0x75, 0x83, 0x60, 0x6b, 0x4f, 0xa9,\n\t\t0x44, 0x9f, 0x5b, 0x46, 0x19, 0x23, 0x85, 0x57, 0x62, 0x6c, 0xfe, 0xb2, 0x5d, 0xf6, 0x40, 0x65,\n\t\t0xbb, 0xb6, 0x10, 0x0e, 0x92, 0x84, 0x70, 0xe2, 0x1a, 0x5d, 0xae, 0xeb, 0x1a, 0x5d, 0xeb, 0x9c,\n\t\t0x31, 0x10, 0xbf, 0x34, 0xe2, 0x55, 0x8a, 0x8e, 0x1c, 0xa0, 0x52, 0x34, 0x78, 0xb0, 0x4a, 0x91,\n\t\t0xa0, 0x64, 0x31, 0xf4, 0x18, 0x4a, 0x16, 0xf9, 0x47, 0x52, 0xb2, 0x28, 0xfc, 0x55, 0x82, 0xc5,\n\t\t0xa4, 0xed, 0x9f, 0x4d, 0xff, 0x2b, 0xb5, 0xfb, 0xdf, 0xb0, 0x13, 0xdb, 0x16, 0x1c, 0x6b, 0xb6,\n\t\t0x8c, 0x04, 0xda, 0x08, 0x5c, 0xcf, 0x34, 0x1f, 0xda, 0x14, 0xe2, 0x6f, 0x24, 0x38, 0x8a, 0x79,\n\t\t0xb7, 0x03, 0xa7, 0xc2, 0x9e, 0x60, 0x16, 0xe7, 0xdb, 0x12, 0xa7, 0x02, 0x20, 0xda, 0x4a, 0xe3,\n\t\t0xf8, 0x03, 0x29, 0x86, 0x3f, 0x68, 0x0b, 0xed, 0x52, 0x09, 0x42, 0xbb, 0xc2, 0x07, 0x12, 0x1c,\n\t\t0x0f, 0xfd, 0xba, 0xc1, 0x89, 0x6d, 0xd9, 0xb7, 0x13, 0x86, 0x52, 0xf5, 0x56, 0x02, 0xdc, 0x5b,\n\t\t0xb7, 0x94, 0x2a, 0xee, 0xf6, 0xd5, 0x87, 0xb6, 0x8d, 0xb6, 0x4c, 0xbc, 0x27, 0x7e, 0x2a, 0xe1,\n\t\t0x67, 0xbc, 0x45, 0x12, 0x75, 0xf3, 0xcc, 0x40, 0x8e, 0xf5, 0x53, 0xb5, 0x4f, 0x81, 0x7b, 0x8b,\n\t\t0x4e, 0x41, 0x73, 0x17, 0x4b, 0xc5, 0xdf, 0xc5, 0xc2, 0xd2, 0xfa, 0x11, 0x1a, 0xf6, 0x25, 0x09,\n\t\t0xe6, 0x13, 0x34, 0xb8, 0xb5, 0xb2, 0xd3, 0x92, 0x2f, 0x3b, 0xdd, 0xed, 0xc2, 0x85, 0x20, 0x2f,\n\t\t0xfc, 0x32, 0x05, 0x2f, 0x1e, 0xac, 0xc9, 0xff, 0xd0, 0x4c, 0xa2, 0x95, 0xbb, 0x4c, 0xf9, 0x72,\n\t\t0x97, 0xf7, 0x00, 0x75, 0x36, 0x93, 0x31, 0xef, 0x70, 0x26, 0x5e, 0xb1, 0x5a, 0x1e, 0xee, 0xe8,\n\t\t0x08, 0x47, 0xe3, 0xd0, 0xa7, 0x9a, 0x06, 0xb1, 0xcc, 0x0a, 0x5d, 0xb0, 0x01, 0xd9, 0xbb, 0x44,\n\t\t0x45, 0x18, 0x09, 0xf4, 0x45, 0x9a, 0x46, 0xc5, 0x3d, 0xa9, 0xf4, 0xcb, 0xc3, 0xbe, 0x76, 0xc5,\n\t\t0xdb, 0x46, 0x65, 0xbf, 0xf0, 0x66, 0x1a, 0x6e, 0x1c, 0xe0, 0x23, 0x02, 0x74, 0xaf, 0xdd, 0x6b,\n\t\t0x0e, 0x0a, 0x3e, 0xd1, 0x89, 0x25, 0xd9, 0x97, 0xa5, 0x3f, 0xa4, 0xf3, 0xb5, 0x30, 0xa7, 0xcc,\n\t\t0x5f, 0x97, 0x9e, 0x83, 0xae, 0xcb, 0x02, 0xa0, 0x60, 0xeb, 0x26, 0xab, 0xf7, 0xa4, 0xe5, 0xbc,\n\t\t0xee, 0x53, 0x42, 0x37, 0xa5, 0xe7, 0xad, 0x62, 0xc6, 0xb7, 0x8a, 0x85, 0x3f, 0x4a, 0x70, 0xb5,\n\t\t0xcb, 0x2f, 0x20, 0x04, 0x18, 0x24, 0x01, 0x86, 0xc7, 0xab, 0xb8, 0x85, 0xcf, 0xa7, 0xe1, 0x6a,\n\t\t0x97, 0x5d, 0xaa, 0xff, 0xa9, 0xb6, 0x1a, 0x70, 0xe8, 0x3d, 0x62, 0x87, 0xde, 0x1b, 0xdf, 0xa1,\n\t\t0x0b, 0x55, 0x47, 0xe4, 0x00, 0xfa, 0x44, 0x0e, 0xe0, 0x73, 0x69, 0xb8, 0xdc, 0x4d, 0xa7, 0x6d,\n\t\t0x3c, 0xcb, 0x8f, 0x25, 0xf9, 0x89, 0xe5, 0xb7, 0x2c, 0xff, 0x7d, 0x09, 0x2e, 0x24, 0xed, 0x1a,\n\t\t0xfe, 0xb7, 0x36, 0x79, 0xf1, 0x5e, 0x55, 0xf8, 0xad, 0x04, 0xe7, 0x13, 0x75, 0x1a, 0x1f, 0x9a,\n\t\t0x0b, 0xe0, 0x9e, 0xa2, 0x52, 0x07, 0x3a, 0x45, 0x15, 0xbe, 0x97, 0x83, 0x4b, 0x5d, 0x7c, 0x32,\n\t\t0xd5, 0xb6, 0x1c, 0x92, 0x6f, 0x39, 0x66, 0x20, 0xd7, 0x5c, 0x0e, 0xa6, 0xf3, 0x59, 0x19, 0xbc,\n\t\t0x5b, 0xbc, 0x94, 0x4a, 0xfa, 0x10, 0x52, 0x2a, 0xdd, 0xd6, 0x57, 0x7b, 0x0f, 0x37, 0xa5, 0x92,\n\t\t0x79, 0xa4, 0x29, 0x95, 0xbe, 0xae, 0x53, 0x2a, 0xf7, 0x81, 0x35, 0x7c, 0x33, 0x89, 0xec, 0x18,\n\t\t0xdb, 0x1f, 0x72, 0x54, 0x76, 0xbb, 0xc6, 0xa9, 0x14, 0xef, 0xa8, 0x5c, 0x0b, 0xde, 0x6a, 0x37,\n\t\t0x92, 0xac, 0xdf, 0x9f, 0xc7, 0x51, 0x79, 0x88, 0xa1, 0xf2, 0x2a, 0x8c, 0xb7, 0xa9, 0x53, 0xd9,\n\t\t0xc2, 0xf5, 0x16, 0xfc, 0x1c, 0x85, 0x3f, 0x1f, 0xaa, 0x38, 0x25, 0x4d, 0x76, 0x58, 0xd8, 0x10,\n\t\t0x8e, 0x36, 0x78, 0xb7, 0x3b, 0xca, 0xb5, 0x47, 0xba, 0x29, 0xd7, 0x76, 0xb4, 0xee, 0x0e, 0x72,\n\t\t0x5a, 0x77, 0x5b, 0x07, 0xb1, 0xa1, 0xe4, 0xb9, 0x96, 0xfc, 0x01, 0x72, 0x2d, 0xc3, 0x07, 0xcb,\n\t\t0xb5, 0x5c, 0x87, 0x9c, 0x86, 0x2b, 0xca, 0xbe, 0xab, 0x9a, 0xd1, 0x2d, 0xc6, 0x40, 0xa9, 0xa9,\n\t\t0x2a, 0xa2, 0xe7, 0x61, 0xe0, 0xe3, 0x3a, 0x21, 0xde, 0xdf, 0x87, 0x34, 0x9b, 0x8b, 0x85, 0xcc,\n\t\t0x39, 0x97, 0xbc, 0xc9, 0xed, 0xf6, 0xe0, 0x5a, 0x75, 0xa3, 0xac, 0x10, 0xd6, 0x5e, 0x1c, 0xd6,\n\t\t0x7b, 0x0b, 0x94, 0x5e, 0xae, 0x1b, 0xcb, 0x44, 0x94, 0x23, 0x3a, 0xfa, 0x18, 0x72, 0x44, 0x63,\n\t\t0x8f, 0x26, 0x47, 0xf4, 0x46, 0x1a, 0x2e, 0x24, 0xfd, 0x40, 0xf4, 0xc3, 0x77, 0xd6, 0x1b, 0x5e,\n\t\t0xd4, 0xe5, 0xd6, 0x51, 0xaf, 0x24, 0xfe, 0xba, 0xd1, 0x17, 0x6c, 0xb5, 0xb9, 0x9d, 0x5e, 0xbf,\n\t\t0xdb, 0xe1, 0x87, 0x14, 0x19, 0x41, 0x48, 0x71, 0x48, 0x99, 0xe6, 0xc2, 0x6f, 0x52, 0xb0, 0x90,\n\t\t0xe4, 0xeb, 0x57, 0xe1, 0x7a, 0xf0, 0x63, 0x99, 0xd4, 0x41, 0x63, 0x99, 0xc3, 0x5a, 0x45, 0xfe,\n\t\t0xec, 0xf6, 0x08, 0x66, 0xb7, 0xe5, 0xeb, 0x7a, 0xe3, 0x27, 0x9d, 0x3e, 0x48, 0x41, 0xc2, 0xef,\n\t\t0x72, 0x3f, 0x1a, 0x93, 0xc9, 0x2b, 0x1a, 0xf6, 0x72, 0x8b, 0x86, 0xad, 0x6e, 0x97, 0x4c, 0xfc,\n\t\t0x6e, 0x97, 0xc2, 0x3f, 0x52, 0x70, 0xee, 0x30, 0x3c, 0xca, 0x47, 0x74, 0xd2, 0xdb, 0xea, 0x39,\n\t\t0x99, 0x04, 0xf5, 0x9c, 0xc2, 0x3f, 0x53, 0x70, 0x3e, 0xd1, 0x67, 0xd2, 0x4f, 0x26, 0xbe, 0x63,\n\t\t0xe2, 0xbd, 0x04, 0x6d, 0x26, 0x49, 0x52, 0xff, 0xd3, 0x69, 0xd1, 0xc4, 0x8b, 0x3a, 0x94, 0x9e,\n\t\t0x4c, 0x7c, 0x68, 0x83, 0x54, 0xa6, 0x9b, 0xef, 0x32, 0x7e, 0x91, 0x82, 0xc5, 0x84, 0x9f, 0xaf,\n\t\t0x3f, 0x59, 0x07, 0xdf, 0x3a, 0xcc, 0x13, 0x18, 0xa2, 0x3f, 0xd7, 0xf5, 0x0a, 0xc1, 0x16, 0x7d,\n\t\t0xd5, 0x71, 0x98, 0x58, 0xbb, 0xbf, 0x76, 0x6b, 0xb3, 0xbc, 0x5e, 0xda, 0xd8, 0x5c, 0x93, 0xcb,\n\t\t0x9b, 0xff, 0x77, 0x67, 0xad, 0x5c, 0xba, 0x75, 0x7f, 0x79, 0xa3, 0x74, 0x33, 0xff, 0x14, 0x9a,\n\t\t0x81, 0xa9, 0xce, 0xc7, 0xcb, 0x1b, 0x1b, 0x65, 0x7a, 0x37, 0x2f, 0xa1, 0x93, 0x70, 0xbc, 0x93,\n\t\t0x60, 0x75, 0xe3, 0xf6, 0xdd, 0x35, 0x46, 0x92, 0x5a, 0x79, 0x15, 0x8e, 0xa9, 0x66, 0x95, 0x37,\n\t\t0x07, 0x2b, 0xde, 0x1f, 0x20, 0xdf, 0x71, 0x22, 0xf3, 0x3b, 0xd2, 0xff, 0x5f, 0x7c, 0xa0, 0x93,\n\t\t0x9d, 0xfa, 0x56, 0x51, 0x35, 0xab, 0x8b, 0xed, 0x7f, 0xc4, 0x7c, 0x5e, 0xd7, 0x2a, 0x8b, 0x0f,\n\t\t0x4c, 0xf7, 0xcf, 0x9f, 0xd9, 0xbf, 0x32, 0xdf, 0x50, 0x6a, 0xfa, 0xde, 0xc5, 0xad, 0x0c, 0xbd,\n\t\t0x77, 0xe9, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0xf3, 0xc4, 0x77, 0x00, 0x78, 0x5a, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/service_workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xdc, 0x5a, 0x5b, 0x6f, 0xdc, 0xc6,\n\t\t0xf5, 0x07, 0x57, 0xf7, 0xb3, 0x92, 0x2c, 0x8d, 0x6e, 0xeb, 0x95, 0x75, 0xa3, 0xe3, 0xfc, 0xf5,\n\t\t0x77, 0xec, 0x55, 0x25, 0xc5, 0x97, 0x38, 0x69, 0x03, 0x79, 0x6d, 0x39, 0x2a, 0xec, 0x54, 0xa5,\n\t\t0x54, 0x1b, 0xed, 0x0b, 0x31, 0x22, 0x67, 0x57, 0x63, 0x71, 0x49, 0x6a, 0x38, 0x94, 0xbc, 0xc9,\n\t\t0x43, 0xd1, 0x22, 0x48, 0x81, 0xa2, 0x2d, 0xfa, 0xd8, 0x02, 0x05, 0xfa, 0xd0, 0x87, 0x3c, 0x35,\n\t\t0x28, 0xd0, 0xa7, 0xbe, 0x17, 0xfd, 0x1e, 0xfd, 0x04, 0x79, 0xea, 0x6b, 0x50, 0x70, 0x66, 0xb8,\n\t\t0x37, 0x91, 0xdc, 0x95, 0x94, 0xc0, 0x4e, 0xdf, 0x76, 0x66, 0xce, 0xef, 0xcc, 0x99, 0x73, 0x9b,\n\t\t0x33, 0x67, 0x09, 0x37, 0xc3, 0x03, 0xc2, 0xd6, 0x2c, 0x6c, 0x13, 0xd7, 0x22, 0x6b, 0xd8, 0xa7,\n\t\t0x6b, 0x27, 0xeb, 0x6b, 0x01, 0x61, 0x27, 0xd4, 0x22, 0xe6, 0xa9, 0xc7, 0x8e, 0x2a, 0x8e, 0x77,\n\t\t0x5a, 0xf2, 0x99, 0xc7, 0x3d, 0x34, 0x15, 0xd1, 0x96, 0x14, 0x6d, 0x09, 0xfb, 0xb4, 0x74, 0xb2,\n\t\t0x5e, 0x5c, 0xac, 0x7a, 0x5e, 0xd5, 0x21, 0x6b, 0x82, 0xe4, 0x20, 0xac, 0xac, 0xd9, 0x21, 0xc3,\n\t\t0x9c, 0x7a, 0xae, 0x04, 0x15, 0x97, 0x3a, 0xd7, 0x39, 0xad, 0x91, 0x80, 0xe3, 0x9a, 0xaf, 0x08,\n\t\t0x96, 0x93, 0x24, 0xb0, 0xbc, 0x5a, 0xad, 0xc1, 0x62, 0x25, 0x89, 0xe2, 0x90, 0x06, 0xdc, 0x63,\n\t\t0xf5, 0x78, 0x97, 0x24, 0x92, 0xe3, 0x90, 0x34, 0x08, 0xf4, 0x24, 0x02, 0x8e, 0x83, 0x23, 0x87,\n\t\t0x06, 0x3c, 0x8b, 0xa6, 0x5d, 0x07, 0xfa, 0x3f, 0x35, 0x58, 0x32, 0x22, 0xf9, 0x19, 0x7f, 0xa1,\n\t\t0x56, 0x1e, 0xbf, 0x22, 0x56, 0x18, 0x9d, 0xd8, 0x20, 0xc7, 0x21, 0x09, 0x38, 0x9a, 0x85, 0x41,\n\t\t0xdb, 0xab, 0x61, 0xea, 0x16, 0xb4, 0x65, 0x6d, 0x75, 0xc4, 0x50, 0x23, 0xf4, 0x13, 0x40, 0x31,\n\t\t0x37, 0x93, 0xc4, 0xa0, 0x42, 0x6e, 0x59, 0x5b, 0xcd, 0x6f, 0xbc, 0x5d, 0x4a, 0x50, 0x6e, 0xe9,\n\t\t0xec, 0x16, 0x93, 0xa7, 0x9d, 0x53, 0xa8, 0x08, 0xc3, 0xd4, 0x26, 0x2e, 0xa7, 0xbc, 0x5e, 0xe8,\n\t\t0x13, 0x1b, 0x36, 0xc6, 0x91, 0x28, 0x8c, 0xe0, 0xc0, 0x73, 0x0b, 0xfd, 0x52, 0x14, 0x39, 0xd2,\n\t\t0xff, 0xaa, 0xc1, 0xf2, 0x23, 0x8a, 0xab, 0xae, 0x17, 0x90, 0xef, 0xc0, 0x39, 0xf4, 0x2f, 0x34,\n\t\t0x58, 0xc9, 0x90, 0x37, 0xf0, 0x3d, 0x37, 0x20, 0xa9, 0x02, 0xbf, 0x84, 0x05, 0x5b, 0x82, 0x39,\n\t\t0xb5, 0xcc, 0x4b, 0xcb, 0x3e, 0xdf, 0x64, 0x76, 0x66, 0x51, 0xff, 0x03, 0xc0, 0xc2, 0xde, 0x85,\n\t\t0xdc, 0x63, 0x09, 0xf2, 0x0d, 0xd1, 0xa8, 0x2d, 0x64, 0x1a, 0x31, 0x20, 0x9e, 0xda, 0xb1, 0xd1,\n\t\t0x36, 0x8c, 0x35, 0x08, 0x78, 0xdd, 0x27, 0x42, 0x4b, 0xf9, 0x8d, 0x95, 0x4c, 0xb1, 0xf7, 0xeb,\n\t\t0x3e, 0x31, 0x46, 0x4f, 0x5b, 0x46, 0xe8, 0x01, 0x8c, 0x44, 0x9e, 0x6f, 0x46, 0xae, 0x2f, 0xfc,\n\t\t0x22, 0xbf, 0xb1, 0x90, 0xc8, 0x63, 0x1f, 0x07, 0x47, 0x4f, 0x69, 0xc0, 0x8d, 0x61, 0xae, 0x7e,\n\t\t0xa1, 0x0d, 0x18, 0xa0, 0xae, 0x1f, 0xf2, 0xc2, 0x80, 0xc0, 0x5d, 0x4b, 0xc4, 0xed, 0xe2, 0xba,\n\t\t0xe3, 0x61, 0xdb, 0x90, 0xa4, 0x08, 0xc3, 0x72, 0x43, 0xd5, 0xa6, 0x08, 0x1d, 0x93, 0x7b, 0xa6,\n\t\t0xe5, 0x78, 0x01, 0x31, 0xa3, 0x6c, 0xe0, 0x85, 0xbc, 0x30, 0x28, 0xd8, 0x5d, 0x2d, 0xc9, 0x6c,\n\t\t0x51, 0x8a, 0xb3, 0x45, 0xe9, 0x91, 0xca, 0x26, 0xc6, 0xb5, 0x06, 0x0b, 0xa1, 0xdd, 0x7d, 0xaf,\n\t\t0x1c, 0xe1, 0xf7, 0x25, 0x1c, 0xbd, 0x80, 0x79, 0x71, 0xa4, 0x14, 0xee, 0x43, 0xdd, 0xb8, 0xcf,\n\t\t0x45, 0xe8, 0x24, 0xc6, 0xad, 0x4e, 0x39, 0xdc, 0x11, 0x5c, 0x0b, 0x00, 0x4c, 0xda, 0x34, 0xb2,\n\t\t0xd7, 0x88, 0x58, 0x1d, 0x51, 0x33, 0x3b, 0x36, 0xb2, 0xa0, 0xd0, 0x62, 0x4f, 0x93, 0x91, 0x30,\n\t\t0x20, 0xa6, 0xef, 0x39, 0xd4, 0xaa, 0x17, 0x60, 0x59, 0x5b, 0x1d, 0xdf, 0xb8, 0x99, 0x69, 0xb9,\n\t\t0x1d, 0xdb, 0x88, 0x20, 0xbb, 0x02, 0x61, 0xcc, 0x9c, 0x26, 0x4d, 0xa3, 0x32, 0x8c, 0x32, 0xc2,\n\t\t0x59, 0x3d, 0x66, 0x9c, 0x17, 0x27, 0x5d, 0x4e, 0x64, 0x6c, 0x44, 0x84, 0x8a, 0x5d, 0x9e, 0x35,\n\t\t0x07, 0xe8, 0x3a, 0x8c, 0x59, 0x2c, 0xb2, 0x8d, 0x75, 0x48, 0xec, 0xd0, 0x21, 0x85, 0x51, 0x71,\n\t\t0x96, 0xd1, 0x68, 0x72, 0x4f, 0xcd, 0xa1, 0xdb, 0xd0, 0x5f, 0x23, 0x35, 0xaf, 0x30, 0xa6, 0x74,\n\t\t0x99, 0xb4, 0xc3, 0x33, 0x52, 0xf3, 0x0c, 0x41, 0x86, 0x0c, 0x98, 0x0c, 0x08, 0x66, 0xd6, 0xa1,\n\t\t0x89, 0x39, 0x67, 0xf4, 0x20, 0xe4, 0x24, 0x28, 0x8c, 0x0b, 0xec, 0x8d, 0x44, 0xec, 0x9e, 0xa0,\n\t\t0xde, 0x6a, 0x10, 0x1b, 0x13, 0x41, 0xc7, 0x0c, 0xda, 0x84, 0xc1, 0x43, 0x82, 0x6d, 0xc2, 0x0a,\n\t\t0x57, 0x04, 0xa3, 0xf9, 0x44, 0x46, 0x1f, 0x09, 0x12, 0x43, 0x91, 0xa2, 0x07, 0x90, 0xb7, 0x89,\n\t\t0x83, 0xeb, 0xd2, 0x37, 0x0a, 0x13, 0xdd, 0x5c, 0x01, 0x04, 0xb5, 0xf0, 0x05, 0xf4, 0x01, 0x8c,\n\t\t0xbe, 0xa4, 0x9c, 0x13, 0xa6, 0xc0, 0x93, 0xdd, 0xc0, 0x79, 0x49, 0xde, 0x40, 0x57, 0x28, 0x0b,\n\t\t0xb8, 0xc9, 0x42, 0xd7, 0xc4, 0xbc, 0x80, 0x04, 0xba, 0x78, 0x06, 0xbd, 0x1f, 0xdf, 0x88, 0x06,\n\t\t0x08, 0x7a, 0x23, 0x74, 0xb7, 0x38, 0x7a, 0x0e, 0x53, 0xc2, 0x28, 0xde, 0x09, 0x61, 0x0e, 0xf6,\n\t\t0x63, 0x03, 0x4f, 0x09, 0xcf, 0x49, 0x4e, 0x55, 0x65, 0xe6, 0xb9, 0x3f, 0x92, 0xe4, 0xca, 0xcc,\n\t\t0x93, 0x56, 0xe7, 0x14, 0x7a, 0x05, 0x4b, 0xd8, 0xe2, 0xf4, 0x84, 0x98, 0x96, 0x13, 0x06, 0xe2,\n\t\t0x6c, 0xc4, 0x21, 0x96, 0x08, 0x4e, 0xb5, 0xc7, 0xb4, 0x10, 0x74, 0x3d, 0x71, 0x8f, 0x2d, 0x81,\n\t\t0x2d, 0x4b, 0xe8, 0x5e, 0x8c, 0x54, 0xdb, 0x5d, 0xc3, 0x19, 0xab, 0xfa, 0x3d, 0x58, 0x4c, 0xcb,\n\t\t0x8c, 0x2a, 0x81, 0xcf, 0xc0, 0x60, 0xa4, 0x2b, 0x6a, 0xab, 0xd4, 0x38, 0xc0, 0x42, 0x77, 0xc7,\n\t\t0xd6, 0x19, 0xe8, 0xc9, 0xc0, 0xad, 0xa0, 0xee, 0x5a, 0x71, 0x5e, 0x7d, 0x0a, 0x43, 0x2a, 0xf8,\n\t\t0x04, 0x3a, 0xbf, 0xb1, 0x91, 0xec, 0x67, 0x59, 0xc9, 0xd9, 0x88, 0x59, 0xe8, 0x37, 0xe0, 0x7a,\n\t\t0xe6, 0x9e, 0x52, 0x62, 0xfd, 0x3d, 0x58, 0x4e, 0x2f, 0x07, 0xb2, 0x4f, 0xf5, 0xaf, 0x1c, 0x2c,\n\t\t0xee, 0xd1, 0xaa, 0x8b, 0x9d, 0xef, 0x42, 0x25, 0xd1, 0x9e, 0xec, 0xfa, 0x3b, 0x93, 0xdd, 0x12,\n\t\t0xe4, 0x03, 0x71, 0x16, 0xd3, 0xc5, 0x35, 0x22, 0x6e, 0x87, 0x11, 0x03, 0xe4, 0xd4, 0xc7, 0xb8,\n\t\t0x46, 0xd0, 0x87, 0x30, 0xaa, 0x08, 0xe4, 0xfd, 0x31, 0xd8, 0xc3, 0xfd, 0xa1, 0x58, 0xee, 0x88,\n\t\t0x5b, 0xa4, 0x00, 0x43, 0x96, 0xe7, 0x72, 0xe6, 0x39, 0x22, 0x9d, 0x8f, 0x1a, 0xf1, 0x50, 0x5f,\n\t\t0x81, 0xa5, 0x54, 0x3d, 0x2a, 0x33, 0x7d, 0xad, 0xc1, 0xff, 0x29, 0x1a, 0xca, 0x0f, 0xb3, 0xef,\n\t\t0xe7, 0x17, 0x30, 0x26, 0xaf, 0x91, 0xcb, 0x7b, 0xd3, 0xa8, 0x60, 0x14, 0x33, 0xee, 0xd0, 0x51,\n\t\t0xae, 0xab, 0x8e, 0xfa, 0x2e, 0xa1, 0xa3, 0xfe, 0x76, 0x1d, 0x6d, 0xc1, 0x6a, 0xf7, 0xf3, 0x67,\n\t\t0xfb, 0xeb, 0xe7, 0x1a, 0xdc, 0xea, 0xc6, 0xa3, 0x2d, 0x20, 0x9f, 0x77, 0x06, 0xe4, 0x07, 0xc9,\n\t\t0x2a, 0xec, 0xcd, 0x2e, 0xcd, 0xd0, 0x5c, 0x83, 0xdb, 0x3d, 0xca, 0xa1, 0xac, 0xff, 0x65, 0x0e,\n\t\t0x16, 0x0c, 0x12, 0x90, 0x37, 0xa6, 0x64, 0x6f, 0x96, 0xe5, 0x7d, 0xad, 0x65, 0x39, 0xba, 0x07,\n\t\t0x05, 0x9b, 0x58, 0x34, 0x88, 0x72, 0x71, 0x85, 0xba, 0x34, 0x38, 0x34, 0xc9, 0x09, 0x71, 0x1b,\n\t\t0x21, 0xd7, 0x67, 0xcc, 0xc4, 0xeb, 0xdb, 0x62, 0xf9, 0x71, 0xb4, 0xba, 0x63, 0x77, 0x44, 0xe7,\n\t\t0x40, 0x67, 0x74, 0x96, 0x60, 0x2a, 0x38, 0xa2, 0xbe, 0xa9, 0xbc, 0x8b, 0x11, 0xec, 0xfb, 0x4e,\n\t\t0x5d, 0xc4, 0xe0, 0xb0, 0x31, 0x19, 0x2d, 0x49, 0x85, 0x1a, 0x72, 0x21, 0xca, 0xd4, 0x69, 0xfa,\n\t\t0xca, 0xf6, 0x91, 0x3f, 0xe5, 0xe0, 0x86, 0xd2, 0x69, 0x19, 0xbb, 0x16, 0xf9, 0x5f, 0x48, 0x6d,\n\t\t0xd3, 0x30, 0x60, 0xe1, 0x30, 0x88, 0x93, 0x9a, 0x1c, 0xa0, 0x4d, 0x98, 0x95, 0x97, 0x7b, 0xb3,\n\t\t0xb4, 0x55, 0x0a, 0x19, 0x14, 0x64, 0x53, 0x62, 0xb5, 0x29, 0x93, 0x50, 0xcf, 0x2a, 0xbc, 0xdd,\n\t\t0x4d, 0x3b, 0xca, 0x65, 0xff, 0x9e, 0x83, 0x95, 0x7d, 0xc2, 0x6a, 0xd4, 0xc5, 0x9c, 0xbc, 0xe9,\n\t\t0x6e, 0x7b, 0x17, 0x86, 0x6c, 0xc2, 0x31, 0x75, 0x02, 0xf5, 0x9c, 0xc8, 0x4e, 0x59, 0x31, 0x71,\n\t\t0x9b, 0x51, 0x06, 0x3a, 0x8c, 0x72, 0x21, 0xfd, 0xbe, 0x05, 0x7a, 0x96, 0xd2, 0x94, 0x6e, 0xff,\n\t\t0x13, 0x3d, 0x7e, 0x49, 0x60, 0x31, 0x7a, 0xf0, 0xc6, 0xa8, 0xf6, 0x00, 0xe6, 0x44, 0xbb, 0xc2,\n\t\t0xb4, 0x3c, 0x37, 0xa0, 0x01, 0x27, 0xae, 0x55, 0x37, 0x1d, 0x72, 0x42, 0x1c, 0xa1, 0xeb, 0xb4,\n\t\t0xb7, 0xc2, 0x8f, 0x23, 0x4c, 0xb9, 0x09, 0x79, 0x1a, 0x21, 0x8c, 0x99, 0xe3, 0xa4, 0x69, 0xfd,\n\t\t0xeb, 0x3e, 0x58, 0xc9, 0x38, 0xb7, 0x8a, 0x6c, 0x07, 0xe6, 0x9a, 0x2a, 0xb7, 0x3c, 0xb7, 0x42,\n\t\t0xab, 0xaa, 0xba, 0x55, 0x59, 0x7c, 0xb3, 0xb7, 0x53, 0x96, 0x5b, 0xa1, 0xc6, 0x2c, 0x49, 0x9c,\n\t\t0x8f, 0xce, 0x7d, 0x56, 0x9d, 0x26, 0x75, 0x2b, 0x9e, 0xd2, 0xe9, 0xcd, 0xde, 0x76, 0xdb, 0x71,\n\t\t0x2b, 0x5e, 0xf3, 0x8d, 0xd4, 0x36, 0x8d, 0x5e, 0x00, 0xf2, 0x89, 0x6b, 0x53, 0xb7, 0x6a, 0x8a,\n\t\t0xfa, 0x94, 0x72, 0x4a, 0x82, 0x42, 0xdf, 0x72, 0xdf, 0x6a, 0x7e, 0x63, 0x35, 0xd9, 0x53, 0x25,\n\t\t0xf9, 0x96, 0xa4, 0xae, 0x0b, 0xe6, 0x93, 0x7e, 0xdb, 0x24, 0x25, 0x01, 0xfa, 0x29, 0x4c, 0xc4,\n\t\t0x8c, 0xad, 0x43, 0xea, 0xd8, 0x8c, 0xb8, 0x85, 0x7e, 0xc1, 0xb6, 0x94, 0xc5, 0xb6, 0x1c, 0xd1,\n\t\t0xb6, 0x4b, 0x7e, 0xc5, 0x6f, 0x59, 0x62, 0xc4, 0x45, 0x7b, 0x4d, 0xd6, 0x71, 0xc6, 0x57, 0x4f,\n\t\t0xee, 0x4c, 0x89, 0x1f, 0x29, 0xda, 0x36, 0xa6, 0xf1, 0xa4, 0xfe, 0x59, 0x1f, 0x4c, 0x0b, 0x8f,\n\t\t0x89, 0xd5, 0xf7, 0x9a, 0x9c, 0xfd, 0x3e, 0x0c, 0x08, 0x0f, 0x55, 0x05, 0x8e, 0x9e, 0xc9, 0x49,\n\t\t0x08, 0x6c, 0x48, 0x00, 0x32, 0x61, 0x56, 0x86, 0x09, 0x23, 0x2f, 0x89, 0xc5, 0x23, 0xff, 0xb4,\n\t\t0xa9, 0x10, 0xaa, 0x5f, 0x44, 0xc9, 0xff, 0xa7, 0x47, 0x89, 0x21, 0x10, 0xe5, 0x18, 0x60, 0x4c,\n\t\t0x1f, 0x27, 0xcc, 0x66, 0xc5, 0xe1, 0xc0, 0x37, 0x15, 0x87, 0x7f, 0xd1, 0x60, 0xa6, 0xc3, 0x0c,\n\t\t0x2a, 0xf6, 0x3e, 0x84, 0xd1, 0xf8, 0x78, 0x41, 0xe8, 0xc4, 0x65, 0x53, 0x97, 0x02, 0x50, 0x9d,\n\t\t0x23, 0x02, 0xa0, 0x1d, 0x18, 0x6f, 0xd5, 0x0f, 0xb1, 0x95, 0xb1, 0xf4, 0x6e, 0x7a, 0x21, 0xb6,\n\t\t0x31, 0x76, 0xdc, 0x3a, 0xd4, 0xbf, 0xd2, 0x60, 0x2e, 0xce, 0x16, 0x8d, 0x46, 0x50, 0x17, 0x7f,\n\t\t0x69, 0xeb, 0x2c, 0xe5, 0xce, 0xd7, 0x59, 0x7a, 0x02, 0xe3, 0x0d, 0x6c, 0xb3, 0xbd, 0x35, 0x9e,\n\t\t0xd2, 0xde, 0x8a, 0x19, 0xc8, 0xf6, 0x16, 0x6f, 0x19, 0x45, 0x45, 0x14, 0x75, 0x2d, 0x27, 0xb4,\n\t\t0x89, 0xd9, 0x64, 0x18, 0x70, 0xcc, 0x43, 0x79, 0x3d, 0x0d, 0x1b, 0x33, 0x6a, 0x3d, 0x66, 0xb2,\n\t\t0x27, 0x16, 0xf5, 0x7f, 0xe4, 0xa0, 0x70, 0xf6, 0xc4, 0xca, 0x34, 0xef, 0xc1, 0x90, 0xef, 0x39,\n\t\t0x0e, 0x61, 0x41, 0x41, 0x13, 0x21, 0xbe, 0x94, 0x6c, 0x15, 0x41, 0x23, 0xc2, 0x2f, 0xa6, 0x47,\n\t\t0xcf, 0x60, 0xe2, 0x8c, 0x20, 0x52, 0x39, 0xd7, 0x33, 0xcf, 0x26, 0xc5, 0x32, 0xc6, 0x79, 0xdb,\n\t\t0x18, 0xbd, 0x80, 0x09, 0x1f, 0x33, 0x4e, 0x5b, 0x12, 0xb4, 0x0a, 0xa4, 0x5b, 0x99, 0xec, 0x76,\n\t\t0x63, 0x90, 0xcc, 0xc0, 0xc6, 0x15, 0xbf, 0x7d, 0xe2, 0x32, 0x7d, 0x41, 0xfd, 0x0e, 0xcc, 0x3f,\n\t\t0x21, 0x3c, 0x5e, 0x08, 0x1e, 0xd6, 0x1f, 0x09, 0x8f, 0xe8, 0xe2, 0x30, 0xfa, 0xef, 0xfa, 0xe1,\n\t\t0x5a, 0x32, 0x4e, 0xa9, 0xfd, 0xe7, 0x30, 0xdb, 0xa8, 0x88, 0x9b, 0x4a, 0xac, 0x61, 0x5f, 0x59,\n\t\t0xe1, 0x87, 0x89, 0x02, 0x66, 0xb1, 0x2c, 0xc5, 0xe9, 0x30, 0xa6, 0x78, 0x86, 0xfd, 0xc7, 0x2e,\n\t\t0x67, 0x75, 0x63, 0xca, 0x3e, 0xbb, 0x12, 0x09, 0xa0, 0x2e, 0x8d, 0x7a, 0x87, 0x00, 0xb9, 0x8b,\n\t\t0x0a, 0x10, 0x5f, 0x2b, 0x67, 0x05, 0xc0, 0x67, 0x57, 0x8a, 0x61, 0xe4, 0x94, 0xc9, 0x12, 0xa3,\n\t\t0x09, 0xe8, 0x3b, 0x22, 0x75, 0xa5, 0xd3, 0xe8, 0x27, 0x2a, 0xc3, 0xc0, 0x09, 0x76, 0x42, 0xa2,\n\t\t0x1c, 0xec, 0x76, 0xa2, 0x74, 0x69, 0x4e, 0x6e, 0x48, 0xec, 0x83, 0xdc, 0x7d, 0x2d, 0xda, 0x36,\n\t\t0x4d, 0xce, 0x6f, 0x71, 0x5b, 0x3d, 0x80, 0x05, 0x11, 0xc8, 0x9d, 0x3e, 0x1b, 0x7c, 0x8b, 0xa9,\n\t\t0x47, 0xff, 0x3c, 0x07, 0x8b, 0x69, 0xbb, 0x2a, 0x3f, 0x3c, 0x86, 0x85, 0x04, 0x37, 0x68, 0x44,\n\t\t0x50, 0x9c, 0x14, 0x4a, 0xbd, 0x45, 0xe0, 0x33, 0xc2, 0xb1, 0x8d, 0x39, 0x36, 0x8a, 0x9d, 0x16,\n\t\t0x6f, 0x6e, 0x1d, 0x6d, 0x99, 0xe0, 0xfa, 0x2d, 0x5b, 0xe6, 0x2e, 0xb6, 0x65, 0xa7, 0x97, 0x37,\n\t\t0xb7, 0xd4, 0xe7, 0x60, 0xe6, 0x09, 0xe1, 0xaa, 0x7d, 0x27, 0x92, 0x98, 0x7a, 0x72, 0xff, 0x52,\n\t\t0x83, 0xd9, 0xce, 0x15, 0xa5, 0x99, 0x43, 0xb8, 0x1a, 0x84, 0xbe, 0xef, 0x31, 0x4e, 0x6c, 0xd3,\n\t\t0x72, 0x68, 0xf4, 0x5c, 0x3d, 0x21, 0x2c, 0x50, 0x5a, 0x49, 0xcf, 0x4b, 0x7b, 0x31, 0xaa, 0x2c,\n\t\t0x40, 0xcf, 0x15, 0xc6, 0x98, 0x0b, 0x92, 0x17, 0xf4, 0xaf, 0xfa, 0x40, 0x7f, 0x92, 0xf0, 0x28,\n\t\t0xfd, 0x48, 0xfe, 0x15, 0xf8, 0x9a, 0x8a, 0x99, 0x79, 0x18, 0xf1, 0x71, 0x95, 0x98, 0x01, 0xfd,\n\t\t0x44, 0x5e, 0x59, 0x03, 0xc6, 0x70, 0x34, 0xb1, 0x47, 0x3f, 0x21, 0xe8, 0x6d, 0xb8, 0xe2, 0x92,\n\t\t0x57, 0x91, 0xd5, 0xaa, 0xc4, 0xe4, 0xde, 0x11, 0x71, 0x55, 0x63, 0x66, 0x2c, 0x9a, 0xde, 0xc5,\n\t\t0x55, 0xb2, 0x1f, 0x4d, 0xa2, 0x77, 0x00, 0x9d, 0x62, 0xca, 0xcd, 0x8a, 0xc7, 0x4c, 0x97, 0x9c,\n\t\t0xca, 0x57, 0xbf, 0xa8, 0x38, 0x86, 0x8d, 0x2b, 0xd1, 0xca, 0xb6, 0xc7, 0x3e, 0x26, 0xa7, 0xe2,\n\t\t0xb9, 0x8f, 0x4c, 0xb8, 0xaa, 0xfe, 0xfd, 0x54, 0xdd, 0x81, 0x0a, 0x75, 0x38, 0x61, 0xf2, 0xd2,\n\t\t0x1c, 0x14, 0x97, 0xe6, 0x5b, 0x89, 0xe7, 0x11, 0xf0, 0x6d, 0x41, 0x2c, 0xee, 0xcd, 0x59, 0xc5,\n\t\t0xa6, 0x63, 0x1e, 0x5d, 0x87, 0x31, 0xd1, 0x2e, 0xc0, 0xcc, 0x3a, 0xa4, 0x27, 0x58, 0x36, 0xdc,\n\t\t0x86, 0x8d, 0xd1, 0x68, 0x72, 0x4b, 0xcd, 0x65, 0x55, 0x4a, 0xc3, 0xdf, 0x54, 0xa5, 0xf4, 0x6f,\n\t\t0x0d, 0xae, 0x67, 0x5a, 0x5c, 0xf9, 0xe0, 0x5d, 0x18, 0x52, 0x47, 0xc9, 0x2c, 0x99, 0x62, 0x58,\n\t\t0x4c, 0x8c, 0x7e, 0x00, 0x79, 0x86, 0x4f, 0xcd, 0x18, 0x2b, 0x03, 0x2a, 0x39, 0x6d, 0x3c, 0xc2,\n\t\t0x1c, 0x3f, 0x74, 0xbc, 0x03, 0x03, 0x18, 0x3e, 0x55, 0x8c, 0x92, 0xcc, 0xdb, 0x97, 0x64, 0xde,\n\t\t0x22, 0x0c, 0x4b, 0x5d, 0x12, 0x5b, 0x95, 0x20, 0x8d, 0xb1, 0xfe, 0x37, 0x0d, 0x46, 0xb7, 0x09,\n\t\t0xe6, 0x21, 0x23, 0xdb, 0x0e, 0xae, 0x06, 0x88, 0xc2, 0x46, 0xc2, 0x93, 0x08, 0x3b, 0x8c, 0x60,\n\t\t0x3b, 0xd2, 0x76, 0xcd, 0x77, 0x48, 0x14, 0x6b, 0x84, 0x31, 0x8f, 0x99, 0xc4, 0xc5, 0x07, 0x0e,\n\t\t0x91, 0x6d, 0x98, 0x61, 0xe3, 0xf6, 0xd9, 0x4e, 0x98, 0xc4, 0x95, 0x63, 0xd8, 0xe3, 0x08, 0xf5,\n\t\t0x58, 0x82, 0xd0, 0x1d, 0x98, 0xc5, 0x21, 0xf7, 0x2a, 0x1e, 0x3b, 0xc5, 0x4c, 0x3c, 0x36, 0x62,\n\t\t0x76, 0x39, 0x59, 0x28, 0xb5, 0xaf, 0x2a, 0x98, 0xfe, 0x1b, 0x0d, 0xe6, 0x0d, 0x52, 0x61, 0x24,\n\t\t0x38, 0x6c, 0xfc, 0xcd, 0x88, 0x83, 0xa3, 0xe0, 0xf5, 0x44, 0xa0, 0xbe, 0x08, 0xd7, 0x92, 0xa5,\n\t\t0x91, 0xde, 0xb1, 0xf1, 0xc5, 0x14, 0xe4, 0xe3, 0x95, 0xad, 0xdd, 0x1d, 0xf4, 0x2b, 0x0d, 0x0a,\n\t\t0x69, 0x4d, 0x7b, 0xf4, 0x6e, 0xca, 0x5f, 0x67, 0x99, 0x7f, 0xf9, 0x17, 0xef, 0x9c, 0x13, 0xa5,\n\t\t0xfc, 0xf6, 0x17, 0x1a, 0xcc, 0x26, 0x37, 0x30, 0xd1, 0x05, 0xda, 0xcd, 0xc5, 0xcd, 0x73, 0x61,\n\t\t0x94, 0x0c, 0xbf, 0xd7, 0x60, 0x3e, 0xa3, 0x89, 0x8a, 0xee, 0x9d, 0x83, 0x69, 0x6b, 0xfb, 0xb7,\n\t\t0x78, 0xff, 0xfc, 0x40, 0x25, 0xd2, 0x67, 0x1a, 0xcc, 0xa5, 0x74, 0xf4, 0xd1, 0x66, 0x56, 0x0f,\n\t\t0x39, 0x4d, 0x31, 0xef, 0x9e, 0x0f, 0xa4, 0xc4, 0xf8, 0xb3, 0x06, 0xcb, 0xdd, 0x1a, 0xcd, 0xe8,\n\t\t0x52, 0x3d, 0xed, 0xe2, 0xf7, 0x2f, 0x88, 0x56, 0x12, 0x7e, 0xa9, 0xc1, 0x8d, 0x9e, 0x5a, 0xe1,\n\t\t0x68, 0xeb, 0x42, 0x1b, 0xb5, 0xd9, 0xf3, 0xe1, 0x65, 0x58, 0xb4, 0x38, 0x7c, 0x72, 0x67, 0x39,\n\t\t0xc5, 0xe1, 0x33, 0xdb, 0xf6, 0x29, 0x0e, 0xdf, 0xa5, 0x75, 0xfd, 0x47, 0x0d, 0x16, 0xb3, 0xbb,\n\t\t0xb0, 0xe8, 0x41, 0x0a, 0xdf, 0x1e, 0x1a, 0xdb, 0xc5, 0xf7, 0x2f, 0x84, 0x55, 0xb2, 0xfd, 0x56,\n\t\t0x83, 0x62, 0x7a, 0x07, 0x13, 0xdd, 0x4d, 0xae, 0xf5, 0xba, 0xf5, 0x89, 0x8b, 0xf7, 0xce, 0x8d,\n\t\t0x53, 0xf2, 0xfc, 0x5a, 0x83, 0xab, 0xa9, 0x2d, 0x43, 0x74, 0x27, 0xb3, 0xcc, 0x4f, 0x95, 0xe6,\n\t\t0xee, 0x79, 0x61, 0x4a, 0x98, 0x0a, 0x8c, 0xb5, 0xb5, 0x4d, 0x50, 0x46, 0xb7, 0xa7, 0xa3, 0xc3,\n\t\t0x55, 0xbc, 0xd9, 0x0b, 0xa9, 0xda, 0xc7, 0x83, 0x89, 0xce, 0xa7, 0x0a, 0xba, 0xd5, 0xe3, 0x8b,\n\t\t0x46, 0xee, 0x76, 0xbe, 0xf7, 0x0f, 0xfa, 0x14, 0xa6, 0x93, 0x1e, 0x8c, 0xe8, 0x7b, 0xe7, 0x78,\n\t\t0x5b, 0xca, 0x8d, 0xd7, 0xcf, 0xfd, 0x1a, 0x15, 0x21, 0x99, 0xfc, 0xf8, 0x49, 0x09, 0xc9, 0xcc,\n\t\t0xf7, 0x59, 0x4a, 0x48, 0x76, 0x79, 0x5d, 0x51, 0x18, 0x6f, 0x7f, 0x5d, 0xa0, 0x9b, 0x69, 0x07,\n\t\t0x39, 0xfb, 0x38, 0x29, 0xbe, 0xd3, 0x13, 0x6d, 0xcb, 0x75, 0x97, 0x51, 0x52, 0xa6, 0x5c, 0x77,\n\t\t0xdd, 0x9f, 0x1d, 0x29, 0xd7, 0x5d, 0x2f, 0xd5, 0xeb, 0xa7, 0x30, 0x9d, 0x54, 0xbf, 0xa4, 0x98,\n\t\t0x3f, 0xa3, 0xf0, 0x4a, 0x31, 0x7f, 0x56, 0x71, 0x24, 0x23, 0x3c, 0xed, 0xcb, 0xba, 0xb4, 0x08,\n\t\t0xef, 0xf2, 0xe5, 0x60, 0x5a, 0x84, 0x77, 0xfb, 0x80, 0xef, 0xa1, 0x0d, 0x73, 0x96, 0x57, 0x4b,\n\t\t0x02, 0x3f, 0x9c, 0x8e, 0x51, 0x7b, 0xf2, 0xe3, 0xd4, 0x5d, 0xe6, 0x71, 0x6f, 0x57, 0xfb, 0xd9,\n\t\t0x7a, 0x95, 0xf2, 0xc3, 0xf0, 0xa0, 0x64, 0x79, 0xb5, 0xb5, 0xd6, 0xef, 0x37, 0x6f, 0x53, 0xdb,\n\t\t0x59, 0xab, 0x7a, 0xf2, 0xbb, 0x53, 0xf5, 0x31, 0xe7, 0xfb, 0xd8, 0xa7, 0x27, 0xeb, 0x07, 0x83,\n\t\t0x62, 0x6e, 0xf3, 0xbf, 0x01, 0x00, 0x00, 0xff, 0xff, 0xbd, 0x51, 0x4a, 0x5f, 0xfc, 0x2a, 0x00,\n\t\t0x00,\n\t},\n\t// uber/cadence/shared/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0x4f, 0x4b, 0xfb, 0x30,\n\t\t0x18, 0xc7, 0xe9, 0x7e, 0xfc, 0x04, 0x33, 0xff, 0x51, 0x50, 0x46, 0x41, 0xd8, 0xa6, 0xc2, 0x4e,\n\t\t0x09, 0x9d, 0x88, 0x07, 0x4f, 0xfe, 0xc5, 0x79, 0x2c, 0xe2, 0xc1, 0x4b, 0x49, 0x93, 0xc7, 0x35,\n\t\t0xe0, 0x92, 0x92, 0xa4, 0xc1, 0xbd, 0x15, 0xdf, 0x82, 0x6f, 0x52, 0xd2, 0xd6, 0x8d, 0xb8, 0x8b,\n\t\t0xb7, 0x3e, 0x3c, 0x9f, 0xcf, 0x97, 0x6f, 0x9f, 0xa0, 0xd3, 0xba, 0x00, 0x4d, 0x18, 0xe5, 0x20,\n\t\t0x19, 0x10, 0x53, 0x52, 0x0d, 0x9c, 0xb8, 0x94, 0x94, 0xc2, 0x58, 0xa5, 0x97, 0xb8, 0xd2, 0xca,\n\t\t0xaa, 0xf8, 0xc8, 0x53, 0xb8, 0xa3, 0x70, 0x4b, 0x61, 0x97, 0x26, 0xa3, 0xc0, 0xa6, 0x95, 0xd8,\n\t\t0x50, 0x93, 0x93, 0x10, 0xe1, 0x0b, 0x21, 0x37, 0xa0, 0xf1, 0x57, 0x84, 0x0e, 0x9f, 0x35, 0x95,\n\t\t0x46, 0x80, 0xb4, 0x77, 0xc0, 0x84, 0x11, 0x4a, 0xce, 0xe4, 0x9b, 0x8a, 0x9f, 0xd0, 0xbe, 0x61,\n\t\t0x25, 0xf0, 0xfa, 0x1d, 0x78, 0x0e, 0x0e, 0xa4, 0x1d, 0x44, 0xc3, 0x68, 0xd2, 0x9f, 0x8e, 0x70,\n\t\t0xd0, 0x89, 0x56, 0x02, 0xbb, 0x14, 0x3f, 0xb6, 0xb1, 0xf7, 0x1e, 0xcc, 0xf6, 0x56, 0x66, 0x33,\n\t\t0xc7, 0x0f, 0x68, 0xd7, 0x58, 0xaa, 0xed, 0x2a, 0xa9, 0xf7, 0xd7, 0xa4, 0x9d, 0xce, 0x6b, 0xa6,\n\t\t0xf1, 0x67, 0x84, 0x0e, 0x5e, 0x40, 0xfb, 0x8e, 0x2d, 0x25, 0xc0, 0xc4, 0xd7, 0xe8, 0x98, 0xd5,\n\t\t0x5a, 0x83, 0xb4, 0xb9, 0x6b, 0x77, 0x79, 0xf7, 0x8f, 0xb9, 0x90, 0x1c, 0x3e, 0x9a, 0xda, 0xff,\n\t\t0xb3, 0xa4, 0x83, 0x02, 0x7f, 0x39, 0xf3, 0x44, 0x7c, 0x8b, 0xb6, 0xcb, 0x9f, 0xbc, 0x41, 0x6f,\n\t\t0xf8, 0x6f, 0xd2, 0x9f, 0x9e, 0xfd, 0xea, 0xe6, 0xcf, 0xe7, 0xdb, 0x85, 0x7a, 0xb6, 0xf6, 0x6e,\n\t\t0x2e, 0x5f, 0x2f, 0xe6, 0xc2, 0x96, 0x75, 0x81, 0x99, 0x5a, 0x90, 0xe0, 0xf8, 0x78, 0x0e, 0x92,\n\t\t0x34, 0x07, 0x5f, 0x3f, 0xf4, 0x55, 0xfb, 0xe5, 0xd2, 0x62, 0xab, 0xd9, 0x9c, 0x7f, 0x07, 0x00,\n\t\t0x00, 0xff, 0xff, 0xc6, 0xf4, 0xa6, 0x9c, 0x12, 0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2e, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x4f, 0x4c, 0xc9, 0xcd, 0xcc, 0xd3, 0x2f,\n\t\t0x33, 0xd4, 0xcf, 0xc8, 0x2c, 0x2e, 0xc9, 0x2f, 0xaa, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17,\n\t\t0x12, 0x05, 0x29, 0xd2, 0x83, 0x2a, 0xd2, 0x03, 0x2b, 0xd2, 0x2b, 0x33, 0x54, 0xf2, 0xe4, 0x12,\n\t\t0x0a, 0x4b, 0x2d, 0x2a, 0xce, 0xcc, 0xcf, 0xf3, 0x80, 0x28, 0xf7, 0x2c, 0x49, 0xcd, 0x15, 0x92,\n\t\t0xe4, 0xe2, 0x48, 0x2d, 0x4b, 0xcd, 0x2b, 0x89, 0xcf, 0x4c, 0x91, 0x60, 0x54, 0x60, 0xd4, 0x60,\n\t\t0x0e, 0x62, 0x07, 0xf3, 0x3d, 0x53, 0x84, 0x24, 0xb8, 0xd8, 0xcb, 0x20, 0x1a, 0x24, 0x98, 0x20,\n\t\t0x32, 0x50, 0xae, 0x52, 0x09, 0x17, 0x1f, 0xaa, 0x51, 0x42, 0x8a, 0x5c, 0x3c, 0x49, 0x45, 0x89,\n\t\t0x79, 0xc9, 0x19, 0xf1, 0x25, 0xf9, 0xd9, 0xa9, 0x79, 0x60, 0xa3, 0x78, 0x82, 0xb8, 0x21, 0x62,\n\t\t0x21, 0x20, 0x21, 0x21, 0x7b, 0x2e, 0xd6, 0xcc, 0x92, 0xd4, 0xdc, 0x62, 0x09, 0x26, 0x05, 0x66,\n\t\t0x0d, 0x6e, 0x23, 0x4d, 0x3d, 0xac, 0xce, 0xd4, 0xc3, 0x74, 0x63, 0x10, 0x44, 0x9f, 0x93, 0x79,\n\t\t0x94, 0x69, 0x7a, 0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x72, 0x48, 0xe8,\n\t\t0x66, 0xa6, 0xe4, 0xe8, 0xa7, 0xe7, 0xeb, 0x83, 0xfd, 0x0f, 0x0f, 0x16, 0x6b, 0x30, 0xa3, 0xcc,\n\t\t0x30, 0x89, 0x0d, 0x2c, 0x6e, 0x0c, 0x08, 0x00, 0x00, 0xff, 0xff, 0x44, 0x14, 0xd7, 0xd4, 0x3e,\n\t\t0x01, 0x00, 0x00,\n\t},\n\t// uber/cadence/shared/v1/tasklist.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x2f, 0x49, 0x2c, 0xce, 0xce, 0xc9, 0x2c, 0x2e, 0xd1, 0x2b, 0x28, 0xca, 0x2f,\n\t\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0x8a,\n\t\t0xe2, 0xe2, 0x0a, 0x49, 0x2c, 0xce, 0x0e, 0xce, 0x2f, 0x2d, 0x4a, 0x4e, 0x15, 0x12, 0xe7, 0x12,\n\t\t0x0e, 0x71, 0x0c, 0xf6, 0x8e, 0x0f, 0xf6, 0x0f, 0x0d, 0x72, 0x76, 0x8d, 0xf7, 0xf4, 0x0b, 0x73,\n\t\t0xf4, 0xf1, 0x74, 0x11, 0x60, 0x40, 0x97, 0xf0, 0xf0, 0x0c, 0x0e, 0xf1, 0x0f, 0x8a, 0x14, 0x60,\n\t\t0x14, 0x92, 0xe2, 0x12, 0x43, 0x96, 0x70, 0x71, 0x8a, 0x77, 0x72, 0x74, 0xf6, 0xf6, 0xf1, 0x77,\n\t\t0x17, 0x60, 0x72, 0x32, 0x8f, 0x32, 0x4d, 0xcf, 0x2c, 0xc9, 0x28, 0x4d, 0xd2, 0x4b, 0xce, 0xcf,\n\t\t0xd5, 0x47, 0x71, 0xa7, 0x5e, 0x7a, 0x6a, 0x9e, 0x3e, 0xd8, 0x65, 0x08, 0x27, 0x5b, 0x43, 0x58,\n\t\t0x65, 0x86, 0x49, 0x6c, 0x60, 0x19, 0x63, 0x40, 0x00, 0x00, 0x00, 0xff, 0xff, 0x1f, 0x73, 0x06,\n\t\t0x1c, 0xdc, 0x00, 0x00, 0x00,\n\t},\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(clientConfig transport.ClientConfig, structField reflect.StructField) MatchingAPIYARPCClient {\n\t\t\treturn NewMatchingAPIYARPCClient(clientConfig, protobuf.ClientBuilderOptions(clientConfig, structField)...)\n\t\t},\n\t)\n}\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/canary.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/canary.proto\n\npackage sharddistributorv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype PingRequest struct {\n\tShardKey             string   `protobuf:\"bytes,1,opt,name=shard_key,json=shardKey,proto3\" json:\"shard_key,omitempty\"`\n\tNamespace            string   `protobuf:\"bytes,2,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *PingRequest) Reset()         { *m = PingRequest{} }\nfunc (m *PingRequest) String() string { return proto.CompactTextString(m) }\nfunc (*PingRequest) ProtoMessage()    {}\nfunc (*PingRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_03b11524fa4f4f94, []int{0}\n}\nfunc (m *PingRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PingRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PingRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PingRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PingRequest.Merge(m, src)\n}\nfunc (m *PingRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PingRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_PingRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PingRequest proto.InternalMessageInfo\n\nfunc (m *PingRequest) GetShardKey() string {\n\tif m != nil {\n\t\treturn m.ShardKey\n\t}\n\treturn \"\"\n}\n\nfunc (m *PingRequest) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\ntype PingResponse struct {\n\tExecutorId           string   `protobuf:\"bytes,1,opt,name=executor_id,json=executorId,proto3\" json:\"executor_id,omitempty\"`\n\tOwnsShard            bool     `protobuf:\"varint,2,opt,name=owns_shard,json=ownsShard,proto3\" json:\"owns_shard,omitempty\"`\n\tShardKey             string   `protobuf:\"bytes,3,opt,name=shard_key,json=shardKey,proto3\" json:\"shard_key,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *PingResponse) Reset()         { *m = PingResponse{} }\nfunc (m *PingResponse) String() string { return proto.CompactTextString(m) }\nfunc (*PingResponse) ProtoMessage()    {}\nfunc (*PingResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_03b11524fa4f4f94, []int{1}\n}\nfunc (m *PingResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *PingResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_PingResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *PingResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_PingResponse.Merge(m, src)\n}\nfunc (m *PingResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *PingResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_PingResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_PingResponse proto.InternalMessageInfo\n\nfunc (m *PingResponse) GetExecutorId() string {\n\tif m != nil {\n\t\treturn m.ExecutorId\n\t}\n\treturn \"\"\n}\n\nfunc (m *PingResponse) GetOwnsShard() bool {\n\tif m != nil {\n\t\treturn m.OwnsShard\n\t}\n\treturn false\n}\n\nfunc (m *PingResponse) GetShardKey() string {\n\tif m != nil {\n\t\treturn m.ShardKey\n\t}\n\treturn \"\"\n}\n\nfunc init() {\n\tproto.RegisterType((*PingRequest)(nil), \"uber.cadence.sharddistributor.v1.PingRequest\")\n\tproto.RegisterType((*PingResponse)(nil), \"uber.cadence.sharddistributor.v1.PingResponse\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/sharddistributor/v1/canary.proto\", fileDescriptor_03b11524fa4f4f94)\n}\n\nvar fileDescriptor_03b11524fa4f4f94 = []byte{\n\t// 290 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0xbb, 0x4e, 0xc3, 0x30,\n\t0x14, 0x86, 0x65, 0x40, 0xa8, 0x39, 0x65, 0xf2, 0x54, 0x71, 0x29, 0xa5, 0x13, 0x4b, 0x6d, 0x15,\n\t0x46, 0x26, 0x6e, 0x12, 0x11, 0x4b, 0x15, 0x06, 0x24, 0x96, 0xc8, 0x71, 0x8e, 0xd2, 0xa8, 0xaa,\n\t0x1d, 0xec, 0x24, 0x90, 0x57, 0xe0, 0xc9, 0x18, 0x79, 0x04, 0x94, 0x27, 0x41, 0x71, 0x83, 0x4a,\n\t0x2a, 0x24, 0x58, 0x3f, 0x7f, 0xfa, 0x7d, 0xce, 0xf9, 0x61, 0x52, 0x44, 0x68, 0xb8, 0x14, 0x31,\n\t0x2a, 0x89, 0xdc, 0xce, 0x85, 0x89, 0xe3, 0xd4, 0xe6, 0x26, 0x8d, 0x8a, 0x5c, 0x1b, 0x5e, 0x4e,\n\t0xb9, 0x14, 0x4a, 0x98, 0x8a, 0x65, 0x46, 0xe7, 0x9a, 0x8e, 0x1a, 0x9d, 0xb5, 0x3a, 0xdb, 0xd4,\n\t0x59, 0x39, 0x1d, 0xdf, 0x41, 0x7f, 0x96, 0xaa, 0x24, 0xc0, 0xe7, 0x02, 0x6d, 0x4e, 0x0f, 0xc0,\n\t0x73, 0x56, 0xb8, 0xc0, 0x6a, 0x40, 0x46, 0xe4, 0xd4, 0x0b, 0x7a, 0x0e, 0xdc, 0x63, 0x45, 0x0f,\n\t0xc1, 0x53, 0x62, 0x89, 0x36, 0x13, 0x12, 0x07, 0x5b, 0xee, 0x71, 0x0d, 0xc6, 0x0b, 0xd8, 0x5b,\n\t0x25, 0xd9, 0x4c, 0x2b, 0x8b, 0xf4, 0x18, 0xfa, 0xf8, 0x8a, 0xb2, 0xf9, 0x28, 0x4c, 0xe3, 0x36,\n\t0x0c, 0xbe, 0x91, 0x1f, 0xd3, 0x23, 0x00, 0xfd, 0xa2, 0x6c, 0xe8, 0xf2, 0x5d, 0x5e, 0x2f, 0xf0,\n\t0x1a, 0xf2, 0xd0, 0x80, 0xee, 0x28, 0xdb, 0xdd, 0x51, 0xce, 0xde, 0x08, 0x9c, 0x38, 0xed, 0x66,\n\t0xbd, 0xce, 0x6d, 0x1b, 0x7d, 0xed, 0x2e, 0x70, 0x39, 0xf3, 0x29, 0xc2, 0x4e, 0x33, 0x12, 0x9d,\n\t0xb0, 0xbf, 0xee, 0xc0, 0x7e, 0x1c, 0x61, 0x9f, 0xfd, 0x57, 0x5f, 0x6d, 0x7a, 0xf5, 0xf8, 0x5e,\n\t0x0f, 0xc9, 0x47, 0x3d, 0x24, 0x9f, 0xf5, 0x90, 0x3c, 0xf9, 0x49, 0x9a, 0xcf, 0x8b, 0x88, 0x49,\n\t0xbd, 0xe4, 0x9d, 0xb6, 0x58, 0x82, 0x8a, 0xbb, 0x5e, 0x7e, 0x2b, 0xee, 0x62, 0x93, 0x95, 0xd3,\n\t0x68, 0xd7, 0xd9, 0xe7, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0x9d, 0xd8, 0xdd, 0xdd, 0xf6, 0x01,\n\t0x00, 0x00,\n}\n\nfunc (m *PingRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PingRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PingRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintCanary(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.ShardKey) > 0 {\n\t\ti -= len(m.ShardKey)\n\t\tcopy(dAtA[i:], m.ShardKey)\n\t\ti = encodeVarintCanary(dAtA, i, uint64(len(m.ShardKey)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *PingResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *PingResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *PingResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ShardKey) > 0 {\n\t\ti -= len(m.ShardKey)\n\t\tcopy(dAtA[i:], m.ShardKey)\n\t\ti = encodeVarintCanary(dAtA, i, uint64(len(m.ShardKey)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.OwnsShard {\n\t\ti--\n\t\tif m.OwnsShard {\n\t\t\tdAtA[i] = 1\n\t\t} else {\n\t\t\tdAtA[i] = 0\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif len(m.ExecutorId) > 0 {\n\t\ti -= len(m.ExecutorId)\n\t\tcopy(dAtA[i:], m.ExecutorId)\n\t\ti = encodeVarintCanary(dAtA, i, uint64(len(m.ExecutorId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintCanary(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovCanary(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *PingRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ShardKey)\n\tif l > 0 {\n\t\tn += 1 + l + sovCanary(uint64(l))\n\t}\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovCanary(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *PingResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ExecutorId)\n\tif l > 0 {\n\t\tn += 1 + l + sovCanary(uint64(l))\n\t}\n\tif m.OwnsShard {\n\t\tn += 2\n\t}\n\tl = len(m.ShardKey)\n\tif l > 0 {\n\t\tn += 1 + l + sovCanary(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovCanary(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozCanary(x uint64) (n int) {\n\treturn sovCanary(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *PingRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PingRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PingRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardKey\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardKey = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipCanary(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *PingResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: PingResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: PingResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExecutorId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ExecutorId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field OwnsShard\", wireType)\n\t\t\t}\n\t\t\tvar v int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tv |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.OwnsShard = bool(v != 0)\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardKey\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardKey = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipCanary(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipCanary(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowCanary\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowCanary\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthCanary\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupCanary\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthCanary\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthCanary        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowCanary          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupCanary = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/canary.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/canary.proto\n\npackage sharddistributorv1\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"reflect\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/gogo/protobuf/proto\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/api/x/restriction\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/encoding/protobuf/reflection\"\n)\n\nvar _ = ioutil.NopCloser\n\n// ShardDistributorExecutorCanaryAPIYARPCClient is the YARPC client-side interface for the ShardDistributorExecutorCanaryAPI service.\ntype ShardDistributorExecutorCanaryAPIYARPCClient interface {\n\tPing(context.Context, *PingRequest, ...yarpc.CallOption) (*PingResponse, error)\n}\n\nfunc newShardDistributorExecutorCanaryAPIYARPCClient(clientConfig transport.ClientConfig, anyResolver jsonpb.AnyResolver, options ...protobuf.ClientOption) ShardDistributorExecutorCanaryAPIYARPCClient {\n\treturn &_ShardDistributorExecutorCanaryAPIYARPCCaller{protobuf.NewStreamClient(\n\t\tprotobuf.ClientParams{\n\t\t\tServiceName:  \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorCanaryAPI\",\n\t\t\tClientConfig: clientConfig,\n\t\t\tAnyResolver:  anyResolver,\n\t\t\tOptions:      options,\n\t\t},\n\t)}\n}\n\n// NewShardDistributorExecutorCanaryAPIYARPCClient builds a new YARPC client for the ShardDistributorExecutorCanaryAPI service.\nfunc NewShardDistributorExecutorCanaryAPIYARPCClient(clientConfig transport.ClientConfig, options ...protobuf.ClientOption) ShardDistributorExecutorCanaryAPIYARPCClient {\n\treturn newShardDistributorExecutorCanaryAPIYARPCClient(clientConfig, nil, options...)\n}\n\n// ShardDistributorExecutorCanaryAPIYARPCServer is the YARPC server-side interface for the ShardDistributorExecutorCanaryAPI service.\ntype ShardDistributorExecutorCanaryAPIYARPCServer interface {\n\tPing(context.Context, *PingRequest) (*PingResponse, error)\n}\n\ntype buildShardDistributorExecutorCanaryAPIYARPCProceduresParams struct {\n\tServer      ShardDistributorExecutorCanaryAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver\n}\n\nfunc buildShardDistributorExecutorCanaryAPIYARPCProcedures(params buildShardDistributorExecutorCanaryAPIYARPCProceduresParams) []transport.Procedure {\n\thandler := &_ShardDistributorExecutorCanaryAPIYARPCHandler{params.Server}\n\treturn protobuf.BuildProcedures(\n\t\tprotobuf.BuildProceduresParams{\n\t\t\tServiceName: \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorCanaryAPI\",\n\t\t\tUnaryHandlerParams: []protobuf.BuildProceduresUnaryHandlerParams{\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"Ping\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.Ping,\n\t\t\t\t\t\t\tNewRequest:  newShardDistributorExecutorCanaryAPIServicePingYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tOnewayHandlerParams: []protobuf.BuildProceduresOnewayHandlerParams{},\n\t\t\tStreamHandlerParams: []protobuf.BuildProceduresStreamHandlerParams{},\n\t\t},\n\t)\n}\n\n// BuildShardDistributorExecutorCanaryAPIYARPCProcedures prepares an implementation of the ShardDistributorExecutorCanaryAPI service for YARPC registration.\nfunc BuildShardDistributorExecutorCanaryAPIYARPCProcedures(server ShardDistributorExecutorCanaryAPIYARPCServer) []transport.Procedure {\n\treturn buildShardDistributorExecutorCanaryAPIYARPCProcedures(buildShardDistributorExecutorCanaryAPIYARPCProceduresParams{Server: server})\n}\n\n// FxShardDistributorExecutorCanaryAPIYARPCClientParams defines the input\n// for NewFxShardDistributorExecutorCanaryAPIYARPCClient. It provides the\n// paramaters to get a ShardDistributorExecutorCanaryAPIYARPCClient in an\n// Fx application.\ntype FxShardDistributorExecutorCanaryAPIYARPCClientParams struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tAnyResolver jsonpb.AnyResolver  `name:\"yarpcfx\" optional:\"true\"`\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// FxShardDistributorExecutorCanaryAPIYARPCClientResult defines the output\n// of NewFxShardDistributorExecutorCanaryAPIYARPCClient. It provides a\n// ShardDistributorExecutorCanaryAPIYARPCClient to an Fx application.\ntype FxShardDistributorExecutorCanaryAPIYARPCClientResult struct {\n\tfx.Out\n\n\tClient ShardDistributorExecutorCanaryAPIYARPCClient\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// NewFxShardDistributorExecutorCanaryAPIYARPCClient provides a ShardDistributorExecutorCanaryAPIYARPCClient\n// to an Fx application using the given name for routing.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorExecutorCanaryAPIYARPCClient(\"service-name\"),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorExecutorCanaryAPIYARPCClient(name string, options ...protobuf.ClientOption) interface{} {\n\treturn func(params FxShardDistributorExecutorCanaryAPIYARPCClientParams) FxShardDistributorExecutorCanaryAPIYARPCClientResult {\n\t\tcc := params.Provider.ClientConfig(name)\n\n\t\tif params.Restriction != nil {\n\t\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok {\n\t\t\t\tif err := params.Restriction.Check(protobuf.Encoding, namer.TransportName()); err != nil {\n\t\t\t\t\tpanic(err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn FxShardDistributorExecutorCanaryAPIYARPCClientResult{\n\t\t\tClient: newShardDistributorExecutorCanaryAPIYARPCClient(cc, params.AnyResolver, options...),\n\t\t}\n\t}\n}\n\n// FxShardDistributorExecutorCanaryAPIYARPCProceduresParams defines the input\n// for NewFxShardDistributorExecutorCanaryAPIYARPCProcedures. It provides the\n// paramaters to get ShardDistributorExecutorCanaryAPIYARPCServer procedures in an\n// Fx application.\ntype FxShardDistributorExecutorCanaryAPIYARPCProceduresParams struct {\n\tfx.In\n\n\tServer      ShardDistributorExecutorCanaryAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver `name:\"yarpcfx\" optional:\"true\"`\n}\n\n// FxShardDistributorExecutorCanaryAPIYARPCProceduresResult defines the output\n// of NewFxShardDistributorExecutorCanaryAPIYARPCProcedures. It provides\n// ShardDistributorExecutorCanaryAPIYARPCServer procedures to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group.\n// Dig 1.2 or newer must be used for this feature to work.\ntype FxShardDistributorExecutorCanaryAPIYARPCProceduresResult struct {\n\tfx.Out\n\n\tProcedures     []transport.Procedure `group:\"yarpcfx\"`\n\tReflectionMeta reflection.ServerMeta `group:\"yarpcfx\"`\n}\n\n// NewFxShardDistributorExecutorCanaryAPIYARPCProcedures provides ShardDistributorExecutorCanaryAPIYARPCServer procedures to an Fx application.\n// It expects a ShardDistributorExecutorCanaryAPIYARPCServer to be present in the container.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorExecutorCanaryAPIYARPCProcedures(),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorExecutorCanaryAPIYARPCProcedures() interface{} {\n\treturn func(params FxShardDistributorExecutorCanaryAPIYARPCProceduresParams) FxShardDistributorExecutorCanaryAPIYARPCProceduresResult {\n\t\treturn FxShardDistributorExecutorCanaryAPIYARPCProceduresResult{\n\t\t\tProcedures: buildShardDistributorExecutorCanaryAPIYARPCProcedures(buildShardDistributorExecutorCanaryAPIYARPCProceduresParams{\n\t\t\t\tServer:      params.Server,\n\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t}),\n\t\t\tReflectionMeta: ShardDistributorExecutorCanaryAPIReflectionMeta,\n\t\t}\n\t}\n}\n\n// ShardDistributorExecutorCanaryAPIReflectionMeta is the reflection server metadata\n// required for using the gRPC reflection protocol with YARPC.\n//\n// See https://github.com/grpc/grpc/blob/master/doc/server-reflection.md.\nvar ShardDistributorExecutorCanaryAPIReflectionMeta = reflection.ServerMeta{\n\tServiceName:     \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorCanaryAPI\",\n\tFileDescriptors: yarpcFileDescriptorClosure03b11524fa4f4f94,\n}\n\ntype _ShardDistributorExecutorCanaryAPIYARPCCaller struct {\n\tstreamClient protobuf.StreamClient\n}\n\nfunc (c *_ShardDistributorExecutorCanaryAPIYARPCCaller) Ping(ctx context.Context, request *PingRequest, options ...yarpc.CallOption) (*PingResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"Ping\", request, newShardDistributorExecutorCanaryAPIServicePingYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*PingResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyShardDistributorExecutorCanaryAPIServicePingYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\ntype _ShardDistributorExecutorCanaryAPIYARPCHandler struct {\n\tserver ShardDistributorExecutorCanaryAPIYARPCServer\n}\n\nfunc (h *_ShardDistributorExecutorCanaryAPIYARPCHandler) Ping(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *PingRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*PingRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyShardDistributorExecutorCanaryAPIServicePingYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.Ping(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc newShardDistributorExecutorCanaryAPIServicePingYARPCRequest() proto.Message {\n\treturn &PingRequest{}\n}\n\nfunc newShardDistributorExecutorCanaryAPIServicePingYARPCResponse() proto.Message {\n\treturn &PingResponse{}\n}\n\nvar (\n\temptyShardDistributorExecutorCanaryAPIServicePingYARPCRequest  = &PingRequest{}\n\temptyShardDistributorExecutorCanaryAPIServicePingYARPCResponse = &PingResponse{}\n)\n\nvar yarpcFileDescriptorClosure03b11524fa4f4f94 = [][]byte{\n\t// uber/cadence/sharddistributor/v1/canary.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0xbd, 0x4e, 0xf3, 0x30,\n\t\t0x14, 0x86, 0x95, 0xef, 0x43, 0xa8, 0x39, 0x65, 0xf2, 0x54, 0xf1, 0x23, 0x4a, 0x27, 0x96, 0xda,\n\t\t0x0a, 0x8c, 0x4c, 0xfc, 0x49, 0x44, 0x5d, 0xaa, 0xb0, 0xb1, 0x44, 0x8e, 0x73, 0x94, 0x5a, 0x55,\n\t\t0xed, 0x60, 0x3b, 0x81, 0xdc, 0x02, 0x57, 0x8d, 0xec, 0x06, 0x95, 0x54, 0x48, 0xb0, 0x3e, 0x7e,\n\t\t0xf4, 0xfa, 0x9c, 0xf3, 0xc2, 0xbc, 0x29, 0xd0, 0x30, 0xc1, 0x4b, 0x54, 0x02, 0x99, 0x5d, 0x71,\n\t\t0x53, 0x96, 0xd2, 0x3a, 0x23, 0x8b, 0xc6, 0x69, 0xc3, 0xda, 0x84, 0x09, 0xae, 0xb8, 0xe9, 0x68,\n\t\t0x6d, 0xb4, 0xd3, 0x64, 0xea, 0x75, 0xda, 0xeb, 0x74, 0x5f, 0xa7, 0x6d, 0x32, 0x7b, 0x82, 0xf1,\n\t\t0x52, 0xaa, 0x2a, 0xc3, 0xd7, 0x06, 0xad, 0x23, 0x27, 0x10, 0x07, 0x2b, 0x5f, 0x63, 0x37, 0x89,\n\t\t0xa6, 0xd1, 0x65, 0x9c, 0x8d, 0x02, 0x58, 0x60, 0x47, 0x4e, 0x21, 0x56, 0x7c, 0x83, 0xb6, 0xe6,\n\t\t0x02, 0x27, 0xff, 0xc2, 0xe3, 0x0e, 0xcc, 0xd6, 0x70, 0xb4, 0x4d, 0xb2, 0xb5, 0x56, 0x16, 0xc9,\n\t\t0x39, 0x8c, 0xf1, 0x1d, 0x85, 0xff, 0x28, 0x97, 0x65, 0x1f, 0x06, 0x5f, 0x28, 0x2d, 0xc9, 0x19,\n\t\t0x80, 0x7e, 0x53, 0x36, 0x0f, 0xf9, 0x21, 0x6f, 0x94, 0xc5, 0x9e, 0x3c, 0x7b, 0x30, 0x1c, 0xe5,\n\t\t0xff, 0x70, 0x94, 0xab, 0x8f, 0x08, 0x2e, 0x82, 0xf6, 0xb0, 0x5b, 0xe7, 0xb1, 0x8f, 0xbe, 0x0f,\n\t\t0x17, 0xb8, 0x5d, 0xa6, 0x04, 0xe1, 0xc0, 0x8f, 0x44, 0xe6, 0xf4, 0xb7, 0x3b, 0xd0, 0x6f, 0x47,\n\t\t0x38, 0xa6, 0x7f, 0xd5, 0xb7, 0x9b, 0xde, 0x2d, 0x5e, 0xd2, 0x4a, 0xba, 0x55, 0x53, 0x50, 0xa1,\n\t\t0x37, 0x6c, 0xd0, 0x10, 0xad, 0x50, 0xb1, 0xd0, 0xc5, 0x4f, 0x65, 0xdd, 0xec, 0xb3, 0x36, 0x29,\n\t\t0x0e, 0x83, 0x7d, 0xfd, 0x19, 0x00, 0x00, 0xff, 0xff, 0xc5, 0x61, 0x98, 0x91, 0xea, 0x01, 0x00,\n\t\t0x00,\n\t},\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(clientConfig transport.ClientConfig, structField reflect.StructField) ShardDistributorExecutorCanaryAPIYARPCClient {\n\t\t\treturn NewShardDistributorExecutorCanaryAPIYARPCClient(clientConfig, protobuf.ClientBuilderOptions(clientConfig, structField)...)\n\t\t},\n\t)\n}\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/executor.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/executor.proto\n\npackage sharddistributorv1\n\nimport (\n\tencoding_binary \"encoding/binary\"\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype ExecutorStatus int32\n\nconst (\n\tExecutorStatus_EXECUTOR_STATUS_INVALID  ExecutorStatus = 0\n\tExecutorStatus_EXECUTOR_STATUS_ACTIVE   ExecutorStatus = 1\n\tExecutorStatus_EXECUTOR_STATUS_DRAINING ExecutorStatus = 2\n\tExecutorStatus_EXECUTOR_STATUS_DRAINED  ExecutorStatus = 3\n)\n\nvar ExecutorStatus_name = map[int32]string{\n\t0: \"EXECUTOR_STATUS_INVALID\",\n\t1: \"EXECUTOR_STATUS_ACTIVE\",\n\t2: \"EXECUTOR_STATUS_DRAINING\",\n\t3: \"EXECUTOR_STATUS_DRAINED\",\n}\n\nvar ExecutorStatus_value = map[string]int32{\n\t\"EXECUTOR_STATUS_INVALID\":  0,\n\t\"EXECUTOR_STATUS_ACTIVE\":   1,\n\t\"EXECUTOR_STATUS_DRAINING\": 2,\n\t\"EXECUTOR_STATUS_DRAINED\":  3,\n}\n\nfunc (x ExecutorStatus) String() string {\n\treturn proto.EnumName(ExecutorStatus_name, int32(x))\n}\n\nfunc (ExecutorStatus) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{0}\n}\n\n// We only have one status for now, but when adding\n// graceful handover, we will need to add more statuses.\n// We do not need an \"inactive\" status, as we will not include\n// inactive shards in the heartbeat request.\ntype ShardStatus int32\n\nconst (\n\tShardStatus_SHARD_STATUS_INVALID ShardStatus = 0\n\tShardStatus_SHARD_STATUS_READY   ShardStatus = 1\n\tShardStatus_SHARD_STATUS_DONE    ShardStatus = 2\n)\n\nvar ShardStatus_name = map[int32]string{\n\t0: \"SHARD_STATUS_INVALID\",\n\t1: \"SHARD_STATUS_READY\",\n\t2: \"SHARD_STATUS_DONE\",\n}\n\nvar ShardStatus_value = map[string]int32{\n\t\"SHARD_STATUS_INVALID\": 0,\n\t\"SHARD_STATUS_READY\":   1,\n\t\"SHARD_STATUS_DONE\":    2,\n}\n\nfunc (x ShardStatus) String() string {\n\treturn proto.EnumName(ShardStatus_name, int32(x))\n}\n\nfunc (ShardStatus) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{1}\n}\n\n// We only have one status for now, but when adding\n// graceful handover, we will need to add more statuses.\n// We do not need an \"inactive\" status, as we will not include\n// inactive shards in the heartbeat request.\ntype AssignmentStatus int32\n\nconst (\n\tAssignmentStatus_ASSIGNMENT_STATUS_INVALID AssignmentStatus = 0\n\tAssignmentStatus_ASSIGNMENT_STATUS_READY   AssignmentStatus = 1\n)\n\nvar AssignmentStatus_name = map[int32]string{\n\t0: \"ASSIGNMENT_STATUS_INVALID\",\n\t1: \"ASSIGNMENT_STATUS_READY\",\n}\n\nvar AssignmentStatus_value = map[string]int32{\n\t\"ASSIGNMENT_STATUS_INVALID\": 0,\n\t\"ASSIGNMENT_STATUS_READY\":   1,\n}\n\nfunc (x AssignmentStatus) String() string {\n\treturn proto.EnumName(AssignmentStatus_name, int32(x))\n}\n\nfunc (AssignmentStatus) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{2}\n}\n\n// We handle  the migration steps from SD side\ntype MigrationMode int32\n\nconst (\n\tMigrationMode_MIGRATION_MODE_INVALID                  MigrationMode = 0\n\tMigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH        MigrationMode = 1\n\tMigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW MigrationMode = 2\n\tMigrationMode_MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH  MigrationMode = 3\n\tMigrationMode_MIGRATION_MODE_ONBOARDED                MigrationMode = 4\n)\n\nvar MigrationMode_name = map[int32]string{\n\t0: \"MIGRATION_MODE_INVALID\",\n\t1: \"MIGRATION_MODE_LOCAL_PASSTHROUGH\",\n\t2: \"MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW\",\n\t3: \"MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH\",\n\t4: \"MIGRATION_MODE_ONBOARDED\",\n}\n\nvar MigrationMode_value = map[string]int32{\n\t\"MIGRATION_MODE_INVALID\":                  0,\n\t\"MIGRATION_MODE_LOCAL_PASSTHROUGH\":        1,\n\t\"MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW\": 2,\n\t\"MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH\":  3,\n\t\"MIGRATION_MODE_ONBOARDED\":                4,\n}\n\nfunc (x MigrationMode) String() string {\n\treturn proto.EnumName(MigrationMode_name, int32(x))\n}\n\nfunc (MigrationMode) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{3}\n}\n\ntype HeartbeatRequest struct {\n\tNamespace            string                        `protobuf:\"bytes,1,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tExecutorId           string                        `protobuf:\"bytes,2,opt,name=executor_id,json=executorId,proto3\" json:\"executor_id,omitempty\"`\n\tStatus               ExecutorStatus                `protobuf:\"varint,3,opt,name=status,proto3,enum=uber.cadence.sharddistributor.v1.ExecutorStatus\" json:\"status,omitempty\"`\n\tShardStatusReports   map[string]*ShardStatusReport `protobuf:\"bytes,4,rep,name=shard_status_reports,json=shardStatusReports,proto3\" json:\"shard_status_reports,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tMetadata             map[string]string             `protobuf:\"bytes,5,rep,name=metadata,proto3\" json:\"metadata,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}                      `json:\"-\"`\n\tXXX_unrecognized     []byte                        `json:\"-\"`\n\tXXX_sizecache        int32                         `json:\"-\"`\n}\n\nfunc (m *HeartbeatRequest) Reset()         { *m = HeartbeatRequest{} }\nfunc (m *HeartbeatRequest) String() string { return proto.CompactTextString(m) }\nfunc (*HeartbeatRequest) ProtoMessage()    {}\nfunc (*HeartbeatRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{0}\n}\nfunc (m *HeartbeatRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *HeartbeatRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_HeartbeatRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *HeartbeatRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_HeartbeatRequest.Merge(m, src)\n}\nfunc (m *HeartbeatRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *HeartbeatRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_HeartbeatRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_HeartbeatRequest proto.InternalMessageInfo\n\nfunc (m *HeartbeatRequest) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\nfunc (m *HeartbeatRequest) GetExecutorId() string {\n\tif m != nil {\n\t\treturn m.ExecutorId\n\t}\n\treturn \"\"\n}\n\nfunc (m *HeartbeatRequest) GetStatus() ExecutorStatus {\n\tif m != nil {\n\t\treturn m.Status\n\t}\n\treturn ExecutorStatus_EXECUTOR_STATUS_INVALID\n}\n\nfunc (m *HeartbeatRequest) GetShardStatusReports() map[string]*ShardStatusReport {\n\tif m != nil {\n\t\treturn m.ShardStatusReports\n\t}\n\treturn nil\n}\n\nfunc (m *HeartbeatRequest) GetMetadata() map[string]string {\n\tif m != nil {\n\t\treturn m.Metadata\n\t}\n\treturn nil\n}\n\ntype ShardStatusReport struct {\n\tStatus               ShardStatus `protobuf:\"varint,1,opt,name=status,proto3,enum=uber.cadence.sharddistributor.v1.ShardStatus\" json:\"status,omitempty\"`\n\tShardLoad            float64     `protobuf:\"fixed64,2,opt,name=shard_load,json=shardLoad,proto3\" json:\"shard_load,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}    `json:\"-\"`\n\tXXX_unrecognized     []byte      `json:\"-\"`\n\tXXX_sizecache        int32       `json:\"-\"`\n}\n\nfunc (m *ShardStatusReport) Reset()         { *m = ShardStatusReport{} }\nfunc (m *ShardStatusReport) String() string { return proto.CompactTextString(m) }\nfunc (*ShardStatusReport) ProtoMessage()    {}\nfunc (*ShardStatusReport) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{1}\n}\nfunc (m *ShardStatusReport) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ShardStatusReport) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ShardStatusReport.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ShardStatusReport) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ShardStatusReport.Merge(m, src)\n}\nfunc (m *ShardStatusReport) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ShardStatusReport) XXX_DiscardUnknown() {\n\txxx_messageInfo_ShardStatusReport.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ShardStatusReport proto.InternalMessageInfo\n\nfunc (m *ShardStatusReport) GetStatus() ShardStatus {\n\tif m != nil {\n\t\treturn m.Status\n\t}\n\treturn ShardStatus_SHARD_STATUS_INVALID\n}\n\nfunc (m *ShardStatusReport) GetShardLoad() float64 {\n\tif m != nil {\n\t\treturn m.ShardLoad\n\t}\n\treturn 0\n}\n\ntype HeartbeatResponse struct {\n\tShardAssignments     map[string]*ShardAssignment `protobuf:\"bytes,1,rep,name=shard_assignments,json=shardAssignments,proto3\" json:\"shard_assignments,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tMigrationMode        MigrationMode               `protobuf:\"varint,2,opt,name=migration_mode,json=migrationMode,proto3,enum=uber.cadence.sharddistributor.v1.MigrationMode\" json:\"migration_mode,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                    `json:\"-\"`\n\tXXX_unrecognized     []byte                      `json:\"-\"`\n\tXXX_sizecache        int32                       `json:\"-\"`\n}\n\nfunc (m *HeartbeatResponse) Reset()         { *m = HeartbeatResponse{} }\nfunc (m *HeartbeatResponse) String() string { return proto.CompactTextString(m) }\nfunc (*HeartbeatResponse) ProtoMessage()    {}\nfunc (*HeartbeatResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{2}\n}\nfunc (m *HeartbeatResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *HeartbeatResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_HeartbeatResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *HeartbeatResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_HeartbeatResponse.Merge(m, src)\n}\nfunc (m *HeartbeatResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *HeartbeatResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_HeartbeatResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_HeartbeatResponse proto.InternalMessageInfo\n\nfunc (m *HeartbeatResponse) GetShardAssignments() map[string]*ShardAssignment {\n\tif m != nil {\n\t\treturn m.ShardAssignments\n\t}\n\treturn nil\n}\n\nfunc (m *HeartbeatResponse) GetMigrationMode() MigrationMode {\n\tif m != nil {\n\t\treturn m.MigrationMode\n\t}\n\treturn MigrationMode_MIGRATION_MODE_INVALID\n}\n\ntype ShardAssignment struct {\n\tStatus               AssignmentStatus `protobuf:\"varint,1,opt,name=status,proto3,enum=uber.cadence.sharddistributor.v1.AssignmentStatus\" json:\"status,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}         `json:\"-\"`\n\tXXX_unrecognized     []byte           `json:\"-\"`\n\tXXX_sizecache        int32            `json:\"-\"`\n}\n\nfunc (m *ShardAssignment) Reset()         { *m = ShardAssignment{} }\nfunc (m *ShardAssignment) String() string { return proto.CompactTextString(m) }\nfunc (*ShardAssignment) ProtoMessage()    {}\nfunc (*ShardAssignment) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_5aab034437d08cca, []int{3}\n}\nfunc (m *ShardAssignment) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ShardAssignment) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ShardAssignment.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ShardAssignment) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ShardAssignment.Merge(m, src)\n}\nfunc (m *ShardAssignment) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ShardAssignment) XXX_DiscardUnknown() {\n\txxx_messageInfo_ShardAssignment.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ShardAssignment proto.InternalMessageInfo\n\nfunc (m *ShardAssignment) GetStatus() AssignmentStatus {\n\tif m != nil {\n\t\treturn m.Status\n\t}\n\treturn AssignmentStatus_ASSIGNMENT_STATUS_INVALID\n}\n\nfunc init() {\n\tproto.RegisterEnum(\"uber.cadence.sharddistributor.v1.ExecutorStatus\", ExecutorStatus_name, ExecutorStatus_value)\n\tproto.RegisterEnum(\"uber.cadence.sharddistributor.v1.ShardStatus\", ShardStatus_name, ShardStatus_value)\n\tproto.RegisterEnum(\"uber.cadence.sharddistributor.v1.AssignmentStatus\", AssignmentStatus_name, AssignmentStatus_value)\n\tproto.RegisterEnum(\"uber.cadence.sharddistributor.v1.MigrationMode\", MigrationMode_name, MigrationMode_value)\n\tproto.RegisterType((*HeartbeatRequest)(nil), \"uber.cadence.sharddistributor.v1.HeartbeatRequest\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.sharddistributor.v1.HeartbeatRequest.MetadataEntry\")\n\tproto.RegisterMapType((map[string]*ShardStatusReport)(nil), \"uber.cadence.sharddistributor.v1.HeartbeatRequest.ShardStatusReportsEntry\")\n\tproto.RegisterType((*ShardStatusReport)(nil), \"uber.cadence.sharddistributor.v1.ShardStatusReport\")\n\tproto.RegisterType((*HeartbeatResponse)(nil), \"uber.cadence.sharddistributor.v1.HeartbeatResponse\")\n\tproto.RegisterMapType((map[string]*ShardAssignment)(nil), \"uber.cadence.sharddistributor.v1.HeartbeatResponse.ShardAssignmentsEntry\")\n\tproto.RegisterType((*ShardAssignment)(nil), \"uber.cadence.sharddistributor.v1.ShardAssignment\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/sharddistributor/v1/executor.proto\", fileDescriptor_5aab034437d08cca)\n}\n\nvar fileDescriptor_5aab034437d08cca = []byte{\n\t// 754 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x9c, 0x55, 0xcd, 0x6e, 0xda, 0x4a,\n\t0x14, 0xbe, 0x86, 0x24, 0xba, 0x1c, 0x14, 0xae, 0x19, 0xe5, 0x87, 0x4b, 0x7e, 0x2e, 0x8a, 0xae,\n\t0xda, 0x88, 0xaa, 0xa6, 0x90, 0x4d, 0xd5, 0x6c, 0xea, 0xc4, 0x16, 0x38, 0x02, 0x3b, 0x1a, 0x1b,\n\t0xd2, 0x56, 0xad, 0xac, 0x01, 0x8f, 0x08, 0x6a, 0xb0, 0xa9, 0x3d, 0xa0, 0xa6, 0xea, 0xb2, 0x6f,\n\t0xd0, 0x17, 0xe9, 0x63, 0x64, 0xd9, 0x7d, 0x37, 0x55, 0x9e, 0xa4, 0xb2, 0x0d, 0x01, 0x1b, 0x2a,\n\t0x9a, 0xec, 0xec, 0x73, 0xce, 0xf7, 0x7d, 0xc7, 0xe7, 0x3b, 0xe3, 0x81, 0xd2, 0xb0, 0x4d, 0xdd,\n\t0x52, 0x87, 0x58, 0xd4, 0xee, 0xd0, 0x92, 0x77, 0x49, 0x5c, 0xcb, 0xea, 0x79, 0xcc, 0xed, 0xb5,\n\t0x87, 0xcc, 0x71, 0x4b, 0xa3, 0x72, 0x89, 0x7e, 0xa4, 0x1d, 0xff, 0x59, 0x18, 0xb8, 0x0e, 0x73,\n\t0x50, 0xc1, 0x07, 0x08, 0x63, 0x80, 0x10, 0x07, 0x08, 0xa3, 0xf2, 0xc1, 0xb7, 0x15, 0xe0, 0x6b,\n\t0x94, 0xb8, 0xac, 0x4d, 0x09, 0xc3, 0xf4, 0xc3, 0x90, 0x7a, 0x0c, 0xed, 0x42, 0xca, 0x26, 0x7d,\n\t0xea, 0x0d, 0x48, 0x87, 0xe6, 0xb8, 0x02, 0x77, 0x98, 0xc2, 0xd3, 0x00, 0xfa, 0x0f, 0xd2, 0x13,\n\t0x19, 0xb3, 0x67, 0xe5, 0x12, 0x41, 0x1e, 0x26, 0x21, 0xc5, 0x42, 0x35, 0x58, 0xf3, 0x18, 0x61,\n\t0x43, 0x2f, 0x97, 0x2c, 0x70, 0x87, 0x99, 0xca, 0x33, 0x61, 0x59, 0x1b, 0x82, 0x3c, 0x46, 0xeb,\n\t0x01, 0x0e, 0x8f, 0xf1, 0xe8, 0x33, 0x6c, 0x04, 0xd5, 0x66, 0xf8, 0x6e, 0xba, 0x74, 0xe0, 0xb8,\n\t0xcc, 0xcb, 0xad, 0x14, 0x92, 0x87, 0xe9, 0xca, 0xd9, 0x72, 0xde, 0xf8, 0xa7, 0x09, 0xba, 0x5f,\n\t0x34, 0x56, 0x09, 0xc9, 0x64, 0x9b, 0xb9, 0xd7, 0x18, 0x79, 0x73, 0x09, 0xf4, 0x16, 0xfe, 0xee,\n\t0x53, 0x46, 0x2c, 0xc2, 0x48, 0x6e, 0x35, 0x50, 0x7c, 0xf9, 0x00, 0xc5, 0xc6, 0x98, 0x22, 0xd4,\n\t0xb9, 0x63, 0xcc, 0x7f, 0x82, 0xed, 0xdf, 0x34, 0x83, 0x78, 0x48, 0xbe, 0xa7, 0xd7, 0xe3, 0xc9,\n\t0xfb, 0x8f, 0x48, 0x81, 0xd5, 0x11, 0xb9, 0x1a, 0xd2, 0x60, 0xda, 0xe9, 0xca, 0xd1, 0xf2, 0x3e,\n\t0xe6, 0xb8, 0x71, 0xc8, 0xf0, 0x22, 0xf1, 0x9c, 0xcb, 0x1f, 0xc3, 0x7a, 0xa4, 0xad, 0x05, 0x8a,\n\t0x1b, 0xb3, 0x8a, 0xa9, 0x19, 0xf0, 0xc1, 0x35, 0x64, 0xe7, 0xc8, 0x91, 0x7c, 0xe7, 0x39, 0x17,\n\t0x78, 0xfe, 0xf4, 0x7e, 0x1d, 0x4e, 0x0c, 0xdf, 0x03, 0x08, 0x0d, 0xbf, 0x72, 0x48, 0xb8, 0x5a,\n\t0x1c, 0x4e, 0x05, 0x91, 0xba, 0x43, 0xac, 0x83, 0x1f, 0x09, 0xc8, 0xce, 0x0c, 0xd8, 0x1b, 0x38,\n\t0xb6, 0x47, 0xd1, 0x08, 0xb2, 0x21, 0x88, 0x78, 0x5e, 0xaf, 0x6b, 0xf7, 0xa9, 0xcd, 0xfc, 0x36,\n\t0x7c, 0xc3, 0x94, 0x7b, 0x19, 0x16, 0xf2, 0x85, 0x8d, 0x89, 0x53, 0xae, 0xd0, 0x39, 0xde, 0x8b,\n\t0x85, 0x51, 0x0b, 0x32, 0xfd, 0x5e, 0xd7, 0x25, 0xac, 0xe7, 0xd8, 0x66, 0xdf, 0xb1, 0xc2, 0x59,\n\t0x65, 0x2a, 0xa5, 0xe5, 0xa2, 0x8d, 0x09, 0xae, 0xe1, 0x58, 0x14, 0xaf, 0xf7, 0x67, 0x5f, 0xf3,\n\t0x23, 0xd8, 0x5c, 0xd8, 0xc2, 0x02, 0x97, 0xaa, 0xd1, 0xbd, 0x28, 0xff, 0xe1, 0xd4, 0xa7, 0xcc,\n\t0xb3, 0xc6, 0xbe, 0x83, 0x7f, 0x62, 0x59, 0x74, 0x16, 0xb3, 0xb5, 0xb2, 0x5c, 0x60, 0x8a, 0x8e,\n\t0x7a, 0x5b, 0xfc, 0xc2, 0x41, 0x26, 0x7a, 0xce, 0xd1, 0x0e, 0x6c, 0xcb, 0xaf, 0xe4, 0xd3, 0xa6,\n\t0xa1, 0x61, 0x53, 0x37, 0x44, 0xa3, 0xa9, 0x9b, 0x8a, 0xda, 0x12, 0xeb, 0x8a, 0xc4, 0xff, 0x85,\n\t0xf2, 0xb0, 0x15, 0x4f, 0x8a, 0xa7, 0x86, 0xd2, 0x92, 0x79, 0x0e, 0xed, 0x42, 0x2e, 0x9e, 0x93,\n\t0xb0, 0xa8, 0xa8, 0x8a, 0x5a, 0xe5, 0x13, 0x8b, 0x68, 0x83, 0xac, 0x2c, 0xf1, 0xc9, 0x62, 0x0b,\n\t0xd2, 0x33, 0x9b, 0x87, 0x72, 0xb0, 0xa1, 0xd7, 0x44, 0x2c, 0xcd, 0xeb, 0x6f, 0x01, 0x8a, 0x64,\n\t0xb0, 0x2c, 0x4a, 0xaf, 0x79, 0x0e, 0x6d, 0x42, 0x36, 0x12, 0x97, 0x34, 0x55, 0xe6, 0x13, 0x45,\n\t0x15, 0xf8, 0xf8, 0xa7, 0xa3, 0x3d, 0xf8, 0x57, 0xd4, 0x75, 0xa5, 0xaa, 0x36, 0x64, 0xd5, 0x98,\n\t0x57, 0xd8, 0x81, 0xed, 0xf9, 0xf4, 0x58, 0xa6, 0x78, 0xc3, 0xc1, 0x7a, 0x64, 0x4d, 0xfc, 0x81,\n\t0x34, 0x94, 0x2a, 0x16, 0x0d, 0x45, 0x53, 0xcd, 0x86, 0x26, 0xc9, 0x33, 0x54, 0xff, 0x43, 0x21,\n\t0x96, 0xab, 0x6b, 0xa7, 0x62, 0xdd, 0x3c, 0x17, 0x75, 0xdd, 0xa8, 0x61, 0xad, 0x59, 0xad, 0xf1,\n\t0x1c, 0x7a, 0x02, 0x8f, 0x97, 0x55, 0x99, 0x7a, 0x4d, 0x94, 0xb4, 0x0b, 0x3e, 0x81, 0x8a, 0xf0,\n\t0x28, 0x56, 0x2c, 0x29, 0xba, 0x81, 0x95, 0x93, 0xa6, 0x21, 0x4b, 0x11, 0xe2, 0xa4, 0xef, 0x47,\n\t0xac, 0x56, 0x53, 0x4f, 0x34, 0x11, 0x4b, 0xb2, 0xc4, 0xaf, 0x54, 0xbe, 0x72, 0xb0, 0x13, 0xcc,\n\t0x5c, 0x9a, 0xae, 0xca, 0x64, 0x13, 0xc4, 0x73, 0x05, 0x31, 0x48, 0xdd, 0x9d, 0x42, 0x54, 0xb9,\n\t0xff, 0x3f, 0x36, 0x7f, 0xf4, 0x80, 0x63, 0x7e, 0x72, 0x71, 0x73, 0xbb, 0xcf, 0x7d, 0xbf, 0xdd,\n\t0xe7, 0x7e, 0xde, 0xee, 0x73, 0x6f, 0x94, 0x6e, 0x8f, 0x5d, 0x0e, 0xdb, 0x42, 0xc7, 0xe9, 0x47,\n\t0xaf, 0x59, 0xa1, 0x4b, 0xed, 0x52, 0x70, 0x9d, 0x2e, 0xba, 0x71, 0x8f, 0xe3, 0xb1, 0x51, 0xb9,\n\t0xbd, 0x16, 0x54, 0x1f, 0xfd, 0x0a, 0x00, 0x00, 0xff, 0xff, 0x33, 0x53, 0x4b, 0xe1, 0xaf, 0x07,\n\t0x00, 0x00,\n}\n\nfunc (m *HeartbeatRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *HeartbeatRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *HeartbeatRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k := range m.Metadata {\n\t\t\tv := m.Metadata[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x2a\n\t\t}\n\t}\n\tif len(m.ShardStatusReports) > 0 {\n\t\tfor k := range m.ShardStatusReports {\n\t\t\tv := m.ShardStatusReports[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x22\n\t\t}\n\t}\n\tif m.Status != 0 {\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(m.Status))\n\t\ti--\n\t\tdAtA[i] = 0x18\n\t}\n\tif len(m.ExecutorId) > 0 {\n\t\ti -= len(m.ExecutorId)\n\t\tcopy(dAtA[i:], m.ExecutorId)\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(m.ExecutorId)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ShardStatusReport) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ShardStatusReport) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ShardStatusReport) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.ShardLoad != 0 {\n\t\ti -= 8\n\t\tencoding_binary.LittleEndian.PutUint64(dAtA[i:], uint64(math.Float64bits(float64(m.ShardLoad))))\n\t\ti--\n\t\tdAtA[i] = 0x11\n\t}\n\tif m.Status != 0 {\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(m.Status))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *HeartbeatResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *HeartbeatResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *HeartbeatResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.MigrationMode != 0 {\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(m.MigrationMode))\n\t\ti--\n\t\tdAtA[i] = 0x10\n\t}\n\tif len(m.ShardAssignments) > 0 {\n\t\tfor k := range m.ShardAssignments {\n\t\t\tv := m.ShardAssignments[k]\n\t\t\tbaseI := i\n\t\t\tif v != nil {\n\t\t\t\t{\n\t\t\t\t\tsize, err := v.MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn 0, err\n\t\t\t\t\t}\n\t\t\t\t\ti -= size\n\t\t\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(size))\n\t\t\t\t}\n\t\t\t\ti--\n\t\t\t\tdAtA[i] = 0x12\n\t\t\t}\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintExecutor(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ShardAssignment) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ShardAssignment) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ShardAssignment) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.Status != 0 {\n\t\ti = encodeVarintExecutor(dAtA, i, uint64(m.Status))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintExecutor(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovExecutor(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *HeartbeatRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovExecutor(uint64(l))\n\t}\n\tl = len(m.ExecutorId)\n\tif l > 0 {\n\t\tn += 1 + l + sovExecutor(uint64(l))\n\t}\n\tif m.Status != 0 {\n\t\tn += 1 + sovExecutor(uint64(m.Status))\n\t}\n\tif len(m.ShardStatusReports) > 0 {\n\t\tfor k, v := range m.ShardStatusReports {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovExecutor(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovExecutor(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovExecutor(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k, v := range m.Metadata {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovExecutor(uint64(len(k))) + 1 + len(v) + sovExecutor(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovExecutor(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ShardStatusReport) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Status != 0 {\n\t\tn += 1 + sovExecutor(uint64(m.Status))\n\t}\n\tif m.ShardLoad != 0 {\n\t\tn += 9\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *HeartbeatResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.ShardAssignments) > 0 {\n\t\tfor k, v := range m.ShardAssignments {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tl = 0\n\t\t\tif v != nil {\n\t\t\t\tl = v.Size()\n\t\t\t\tl += 1 + sovExecutor(uint64(l))\n\t\t\t}\n\t\t\tmapEntrySize := 1 + len(k) + sovExecutor(uint64(len(k))) + l\n\t\t\tn += mapEntrySize + 1 + sovExecutor(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.MigrationMode != 0 {\n\t\tn += 1 + sovExecutor(uint64(m.MigrationMode))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ShardAssignment) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.Status != 0 {\n\t\tn += 1 + sovExecutor(uint64(m.Status))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovExecutor(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozExecutor(x uint64) (n int) {\n\treturn sovExecutor(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *HeartbeatRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: HeartbeatRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: HeartbeatRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExecutorId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ExecutorId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Status\", wireType)\n\t\t\t}\n\t\t\tm.Status = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Status |= ExecutorStatus(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardStatusReports\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ShardStatusReports == nil {\n\t\t\t\tm.ShardStatusReports = make(map[string]*ShardStatusReport)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *ShardStatusReport\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &ShardStatusReport{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ShardStatusReports[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 5:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Metadata\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Metadata == nil {\n\t\t\t\tm.Metadata = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Metadata[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ShardStatusReport) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ShardStatusReport: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ShardStatusReport: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Status\", wireType)\n\t\t\t}\n\t\t\tm.Status = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Status |= ShardStatus(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 1 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardLoad\", wireType)\n\t\t\t}\n\t\t\tvar v uint64\n\t\t\tif (iNdEx + 8) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tv = uint64(encoding_binary.LittleEndian.Uint64(dAtA[iNdEx:]))\n\t\t\tiNdEx += 8\n\t\t\tm.ShardLoad = float64(math.Float64frombits(v))\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *HeartbeatResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: HeartbeatResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: HeartbeatResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardAssignments\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ShardAssignments == nil {\n\t\t\t\tm.ShardAssignments = make(map[string]*ShardAssignment)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue *ShardAssignment\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar mapmsglen int\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tmapmsglen |= int(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tif mapmsglen < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tpostmsgIndex := iNdEx + mapmsglen\n\t\t\t\t\tif postmsgIndex < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif postmsgIndex > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = &ShardAssignment{}\n\t\t\t\t\tif err := mapvalue.Unmarshal(dAtA[iNdEx:postmsgIndex]); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx = postmsgIndex\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.ShardAssignments[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field MigrationMode\", wireType)\n\t\t\t}\n\t\t\tm.MigrationMode = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.MigrationMode |= MigrationMode(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ShardAssignment) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ShardAssignment: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ShardAssignment: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Status\", wireType)\n\t\t\t}\n\t\t\tm.Status = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.Status |= AssignmentStatus(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipExecutor(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipExecutor(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowExecutor\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowExecutor\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthExecutor\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupExecutor\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthExecutor\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthExecutor        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowExecutor          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupExecutor = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/executor.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/executor.proto\n\npackage sharddistributorv1\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"reflect\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/gogo/protobuf/proto\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/api/x/restriction\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/encoding/protobuf/reflection\"\n)\n\nvar _ = ioutil.NopCloser\n\n// ShardDistributorExecutorAPIYARPCClient is the YARPC client-side interface for the ShardDistributorExecutorAPI service.\ntype ShardDistributorExecutorAPIYARPCClient interface {\n\tHeartbeat(context.Context, *HeartbeatRequest, ...yarpc.CallOption) (*HeartbeatResponse, error)\n}\n\nfunc newShardDistributorExecutorAPIYARPCClient(clientConfig transport.ClientConfig, anyResolver jsonpb.AnyResolver, options ...protobuf.ClientOption) ShardDistributorExecutorAPIYARPCClient {\n\treturn &_ShardDistributorExecutorAPIYARPCCaller{protobuf.NewStreamClient(\n\t\tprotobuf.ClientParams{\n\t\t\tServiceName:  \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorAPI\",\n\t\t\tClientConfig: clientConfig,\n\t\t\tAnyResolver:  anyResolver,\n\t\t\tOptions:      options,\n\t\t},\n\t)}\n}\n\n// NewShardDistributorExecutorAPIYARPCClient builds a new YARPC client for the ShardDistributorExecutorAPI service.\nfunc NewShardDistributorExecutorAPIYARPCClient(clientConfig transport.ClientConfig, options ...protobuf.ClientOption) ShardDistributorExecutorAPIYARPCClient {\n\treturn newShardDistributorExecutorAPIYARPCClient(clientConfig, nil, options...)\n}\n\n// ShardDistributorExecutorAPIYARPCServer is the YARPC server-side interface for the ShardDistributorExecutorAPI service.\ntype ShardDistributorExecutorAPIYARPCServer interface {\n\tHeartbeat(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error)\n}\n\ntype buildShardDistributorExecutorAPIYARPCProceduresParams struct {\n\tServer      ShardDistributorExecutorAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver\n}\n\nfunc buildShardDistributorExecutorAPIYARPCProcedures(params buildShardDistributorExecutorAPIYARPCProceduresParams) []transport.Procedure {\n\thandler := &_ShardDistributorExecutorAPIYARPCHandler{params.Server}\n\treturn protobuf.BuildProcedures(\n\t\tprotobuf.BuildProceduresParams{\n\t\t\tServiceName: \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorAPI\",\n\t\t\tUnaryHandlerParams: []protobuf.BuildProceduresUnaryHandlerParams{\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"Heartbeat\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.Heartbeat,\n\t\t\t\t\t\t\tNewRequest:  newShardDistributorExecutorAPIServiceHeartbeatYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tOnewayHandlerParams: []protobuf.BuildProceduresOnewayHandlerParams{},\n\t\t\tStreamHandlerParams: []protobuf.BuildProceduresStreamHandlerParams{},\n\t\t},\n\t)\n}\n\n// BuildShardDistributorExecutorAPIYARPCProcedures prepares an implementation of the ShardDistributorExecutorAPI service for YARPC registration.\nfunc BuildShardDistributorExecutorAPIYARPCProcedures(server ShardDistributorExecutorAPIYARPCServer) []transport.Procedure {\n\treturn buildShardDistributorExecutorAPIYARPCProcedures(buildShardDistributorExecutorAPIYARPCProceduresParams{Server: server})\n}\n\n// FxShardDistributorExecutorAPIYARPCClientParams defines the input\n// for NewFxShardDistributorExecutorAPIYARPCClient. It provides the\n// paramaters to get a ShardDistributorExecutorAPIYARPCClient in an\n// Fx application.\ntype FxShardDistributorExecutorAPIYARPCClientParams struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tAnyResolver jsonpb.AnyResolver  `name:\"yarpcfx\" optional:\"true\"`\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// FxShardDistributorExecutorAPIYARPCClientResult defines the output\n// of NewFxShardDistributorExecutorAPIYARPCClient. It provides a\n// ShardDistributorExecutorAPIYARPCClient to an Fx application.\ntype FxShardDistributorExecutorAPIYARPCClientResult struct {\n\tfx.Out\n\n\tClient ShardDistributorExecutorAPIYARPCClient\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// NewFxShardDistributorExecutorAPIYARPCClient provides a ShardDistributorExecutorAPIYARPCClient\n// to an Fx application using the given name for routing.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorExecutorAPIYARPCClient(\"service-name\"),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorExecutorAPIYARPCClient(name string, options ...protobuf.ClientOption) interface{} {\n\treturn func(params FxShardDistributorExecutorAPIYARPCClientParams) FxShardDistributorExecutorAPIYARPCClientResult {\n\t\tcc := params.Provider.ClientConfig(name)\n\n\t\tif params.Restriction != nil {\n\t\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok {\n\t\t\t\tif err := params.Restriction.Check(protobuf.Encoding, namer.TransportName()); err != nil {\n\t\t\t\t\tpanic(err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn FxShardDistributorExecutorAPIYARPCClientResult{\n\t\t\tClient: newShardDistributorExecutorAPIYARPCClient(cc, params.AnyResolver, options...),\n\t\t}\n\t}\n}\n\n// FxShardDistributorExecutorAPIYARPCProceduresParams defines the input\n// for NewFxShardDistributorExecutorAPIYARPCProcedures. It provides the\n// paramaters to get ShardDistributorExecutorAPIYARPCServer procedures in an\n// Fx application.\ntype FxShardDistributorExecutorAPIYARPCProceduresParams struct {\n\tfx.In\n\n\tServer      ShardDistributorExecutorAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver `name:\"yarpcfx\" optional:\"true\"`\n}\n\n// FxShardDistributorExecutorAPIYARPCProceduresResult defines the output\n// of NewFxShardDistributorExecutorAPIYARPCProcedures. It provides\n// ShardDistributorExecutorAPIYARPCServer procedures to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group.\n// Dig 1.2 or newer must be used for this feature to work.\ntype FxShardDistributorExecutorAPIYARPCProceduresResult struct {\n\tfx.Out\n\n\tProcedures     []transport.Procedure `group:\"yarpcfx\"`\n\tReflectionMeta reflection.ServerMeta `group:\"yarpcfx\"`\n}\n\n// NewFxShardDistributorExecutorAPIYARPCProcedures provides ShardDistributorExecutorAPIYARPCServer procedures to an Fx application.\n// It expects a ShardDistributorExecutorAPIYARPCServer to be present in the container.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorExecutorAPIYARPCProcedures(),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorExecutorAPIYARPCProcedures() interface{} {\n\treturn func(params FxShardDistributorExecutorAPIYARPCProceduresParams) FxShardDistributorExecutorAPIYARPCProceduresResult {\n\t\treturn FxShardDistributorExecutorAPIYARPCProceduresResult{\n\t\t\tProcedures: buildShardDistributorExecutorAPIYARPCProcedures(buildShardDistributorExecutorAPIYARPCProceduresParams{\n\t\t\t\tServer:      params.Server,\n\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t}),\n\t\t\tReflectionMeta: ShardDistributorExecutorAPIReflectionMeta,\n\t\t}\n\t}\n}\n\n// ShardDistributorExecutorAPIReflectionMeta is the reflection server metadata\n// required for using the gRPC reflection protocol with YARPC.\n//\n// See https://github.com/grpc/grpc/blob/master/doc/server-reflection.md.\nvar ShardDistributorExecutorAPIReflectionMeta = reflection.ServerMeta{\n\tServiceName:     \"uber.cadence.sharddistributor.v1.ShardDistributorExecutorAPI\",\n\tFileDescriptors: yarpcFileDescriptorClosure5aab034437d08cca,\n}\n\ntype _ShardDistributorExecutorAPIYARPCCaller struct {\n\tstreamClient protobuf.StreamClient\n}\n\nfunc (c *_ShardDistributorExecutorAPIYARPCCaller) Heartbeat(ctx context.Context, request *HeartbeatRequest, options ...yarpc.CallOption) (*HeartbeatResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"Heartbeat\", request, newShardDistributorExecutorAPIServiceHeartbeatYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*HeartbeatResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyShardDistributorExecutorAPIServiceHeartbeatYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\ntype _ShardDistributorExecutorAPIYARPCHandler struct {\n\tserver ShardDistributorExecutorAPIYARPCServer\n}\n\nfunc (h *_ShardDistributorExecutorAPIYARPCHandler) Heartbeat(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *HeartbeatRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*HeartbeatRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyShardDistributorExecutorAPIServiceHeartbeatYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.Heartbeat(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc newShardDistributorExecutorAPIServiceHeartbeatYARPCRequest() proto.Message {\n\treturn &HeartbeatRequest{}\n}\n\nfunc newShardDistributorExecutorAPIServiceHeartbeatYARPCResponse() proto.Message {\n\treturn &HeartbeatResponse{}\n}\n\nvar (\n\temptyShardDistributorExecutorAPIServiceHeartbeatYARPCRequest  = &HeartbeatRequest{}\n\temptyShardDistributorExecutorAPIServiceHeartbeatYARPCResponse = &HeartbeatResponse{}\n)\n\nvar yarpcFileDescriptorClosure5aab034437d08cca = [][]byte{\n\t// uber/cadence/sharddistributor/v1/executor.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x9c, 0x55, 0xdd, 0x6e, 0xda, 0x48,\n\t\t0x14, 0x5e, 0x43, 0x12, 0x2d, 0x07, 0x85, 0x35, 0xa3, 0xfc, 0xb0, 0x24, 0xd1, 0xa2, 0x68, 0xb5,\n\t\t0x1b, 0x51, 0xd5, 0x14, 0x72, 0x53, 0x35, 0x37, 0x75, 0x62, 0x0b, 0x26, 0x05, 0x3b, 0x1a, 0x1b,\n\t\t0xfa, 0xa3, 0x56, 0xd6, 0x80, 0x47, 0x04, 0x35, 0xd8, 0xd4, 0x1e, 0x50, 0x53, 0xf5, 0xb2, 0x6f,\n\t\t0xd0, 0x17, 0xe9, 0x63, 0xf4, 0x1d, 0xfa, 0x32, 0x95, 0x6d, 0x08, 0xd8, 0x50, 0xd1, 0xe4, 0xce,\n\t\t0x3e, 0xe7, 0x7c, 0xdf, 0x77, 0x7c, 0xbe, 0x33, 0x1e, 0xa8, 0x8c, 0xbb, 0xcc, 0xab, 0xf4, 0xa8,\n\t\t0xcd, 0x9c, 0x1e, 0xab, 0xf8, 0xd7, 0xd4, 0xb3, 0xed, 0x81, 0xcf, 0xbd, 0x41, 0x77, 0xcc, 0x5d,\n\t\t0xaf, 0x32, 0xa9, 0x56, 0xd8, 0x47, 0xd6, 0x0b, 0x9e, 0xa5, 0x91, 0xe7, 0x72, 0x17, 0x95, 0x02,\n\t\t0x80, 0x34, 0x05, 0x48, 0x49, 0x80, 0x34, 0xa9, 0x1e, 0x7f, 0xdb, 0x00, 0xb1, 0xc1, 0xa8, 0xc7,\n\t\t0xbb, 0x8c, 0x72, 0xc2, 0x3e, 0x8c, 0x99, 0xcf, 0xd1, 0x21, 0x64, 0x1c, 0x3a, 0x64, 0xfe, 0x88,\n\t\t0xf6, 0x58, 0x41, 0x28, 0x09, 0x27, 0x19, 0x32, 0x0f, 0xa0, 0x7f, 0x20, 0x3b, 0x93, 0xb1, 0x06,\n\t\t0x76, 0x21, 0x15, 0xe6, 0x61, 0x16, 0xc2, 0x36, 0x6a, 0xc0, 0x96, 0xcf, 0x29, 0x1f, 0xfb, 0x85,\n\t\t0x74, 0x49, 0x38, 0xc9, 0xd5, 0x9e, 0x48, 0xeb, 0xda, 0x90, 0xd4, 0x29, 0xda, 0x08, 0x71, 0x64,\n\t\t0x8a, 0x47, 0x9f, 0x61, 0x27, 0xac, 0xb6, 0xa2, 0x77, 0xcb, 0x63, 0x23, 0xd7, 0xe3, 0x7e, 0x61,\n\t\t0xa3, 0x94, 0x3e, 0xc9, 0xd6, 0x2e, 0xd7, 0xf3, 0x26, 0x3f, 0x4d, 0x32, 0x82, 0xa2, 0xa9, 0x4a,\n\t\t0x44, 0xa6, 0x3a, 0xdc, 0xbb, 0x25, 0xc8, 0x5f, 0x4a, 0xa0, 0xb7, 0xf0, 0xe7, 0x90, 0x71, 0x6a,\n\t\t0x53, 0x4e, 0x0b, 0x9b, 0xa1, 0xe2, 0xf3, 0x07, 0x28, 0xb6, 0xa6, 0x14, 0x91, 0xce, 0x1d, 0x63,\n\t\t0xf1, 0x13, 0xec, 0xff, 0xa2, 0x19, 0x24, 0x42, 0xfa, 0x3d, 0xbb, 0x9d, 0x4e, 0x3e, 0x78, 0x44,\n\t\t0x18, 0x36, 0x27, 0xf4, 0x66, 0xcc, 0xc2, 0x69, 0x67, 0x6b, 0xa7, 0xeb, 0xfb, 0x58, 0xe2, 0x26,\n\t\t0x11, 0xc3, 0xb3, 0xd4, 0x53, 0xa1, 0x78, 0x06, 0xdb, 0xb1, 0xb6, 0x56, 0x28, 0xee, 0x2c, 0x2a,\n\t\t0x66, 0x16, 0xc0, 0xc7, 0xb7, 0x90, 0x5f, 0x22, 0x47, 0xea, 0x9d, 0xe7, 0x42, 0xe8, 0xf9, 0xe3,\n\t\t0xfb, 0x75, 0x38, 0x33, 0xfc, 0x08, 0x20, 0x32, 0xfc, 0xc6, 0xa5, 0xd1, 0x6a, 0x09, 0x24, 0x13,\n\t\t0x46, 0x9a, 0x2e, 0xb5, 0x8f, 0x7f, 0xa4, 0x20, 0xbf, 0x30, 0x60, 0x7f, 0xe4, 0x3a, 0x3e, 0x43,\n\t\t0x13, 0xc8, 0x47, 0x20, 0xea, 0xfb, 0x83, 0xbe, 0x33, 0x64, 0x0e, 0x0f, 0xda, 0x08, 0x0c, 0xc3,\n\t\t0xf7, 0x32, 0x2c, 0xe2, 0x8b, 0x1a, 0x93, 0xe7, 0x5c, 0x91, 0x73, 0xa2, 0x9f, 0x08, 0xa3, 0x0e,\n\t\t0xe4, 0x86, 0x83, 0xbe, 0x47, 0xf9, 0xc0, 0x75, 0xac, 0xa1, 0x6b, 0x47, 0xb3, 0xca, 0xd5, 0x2a,\n\t\t0xeb, 0x45, 0x5b, 0x33, 0x5c, 0xcb, 0xb5, 0x19, 0xd9, 0x1e, 0x2e, 0xbe, 0x16, 0x27, 0xb0, 0xbb,\n\t\t0xb2, 0x85, 0x15, 0x2e, 0xd5, 0xe3, 0x7b, 0x51, 0xfd, 0xcd, 0xa9, 0xcf, 0x99, 0x17, 0x8d, 0x7d,\n\t\t0x07, 0x7f, 0x25, 0xb2, 0xe8, 0x32, 0x61, 0x6b, 0x6d, 0xbd, 0xc0, 0x1c, 0x1d, 0xf7, 0xb6, 0xfc,\n\t\t0x45, 0x80, 0x5c, 0xfc, 0x9c, 0xa3, 0x03, 0xd8, 0x57, 0x5f, 0xa9, 0x17, 0x6d, 0x53, 0x27, 0x96,\n\t\t0x61, 0xca, 0x66, 0xdb, 0xb0, 0xb0, 0xd6, 0x91, 0x9b, 0x58, 0x11, 0xff, 0x40, 0x45, 0xd8, 0x4b,\n\t\t0x26, 0xe5, 0x0b, 0x13, 0x77, 0x54, 0x51, 0x40, 0x87, 0x50, 0x48, 0xe6, 0x14, 0x22, 0x63, 0x0d,\n\t\t0x6b, 0x75, 0x31, 0xb5, 0x8a, 0x36, 0xcc, 0xaa, 0x8a, 0x98, 0x2e, 0x77, 0x20, 0xbb, 0xb0, 0x79,\n\t\t0xa8, 0x00, 0x3b, 0x46, 0x43, 0x26, 0xca, 0xb2, 0xfe, 0x1e, 0xa0, 0x58, 0x86, 0xa8, 0xb2, 0xf2,\n\t\t0x5a, 0x14, 0xd0, 0x2e, 0xe4, 0x63, 0x71, 0x45, 0xd7, 0x54, 0x31, 0x55, 0xd6, 0x40, 0x4c, 0x7e,\n\t\t0x3a, 0x3a, 0x82, 0xbf, 0x65, 0xc3, 0xc0, 0x75, 0xad, 0xa5, 0x6a, 0xe6, 0xb2, 0xc2, 0x01, 0xec,\n\t\t0x2f, 0xa7, 0xa7, 0x32, 0xe5, 0xef, 0x02, 0x6c, 0xc7, 0xd6, 0x24, 0x18, 0x48, 0x0b, 0xd7, 0x89,\n\t\t0x6c, 0x62, 0x5d, 0xb3, 0x5a, 0xba, 0xa2, 0x2e, 0x50, 0xfd, 0x0b, 0xa5, 0x44, 0xae, 0xa9, 0x5f,\n\t\t0xc8, 0x4d, 0xeb, 0x4a, 0x36, 0x0c, 0xb3, 0x41, 0xf4, 0x76, 0xbd, 0x21, 0x0a, 0xe8, 0x11, 0xfc,\n\t\t0xbf, 0xae, 0xca, 0x32, 0x1a, 0xb2, 0xa2, 0xbf, 0x14, 0x53, 0xa8, 0x0c, 0xff, 0x25, 0x8a, 0x15,\n\t\t0x6c, 0x98, 0x04, 0x9f, 0xb7, 0x4d, 0x55, 0x89, 0x11, 0xa7, 0x03, 0x3f, 0x12, 0xb5, 0xba, 0x76,\n\t\t0xae, 0xcb, 0x44, 0x51, 0x15, 0x71, 0xa3, 0xf6, 0x55, 0x80, 0x83, 0x70, 0xe6, 0xca, 0x7c, 0x55,\n\t\t0x66, 0x9b, 0x20, 0x5f, 0x61, 0xc4, 0x21, 0x73, 0x77, 0x0a, 0x51, 0xed, 0xfe, 0xff, 0xd8, 0xe2,\n\t\t0xe9, 0x03, 0x8e, 0xf9, 0xf9, 0x8b, 0x37, 0xb8, 0x3f, 0xe0, 0xd7, 0xe3, 0xae, 0xd4, 0x73, 0x87,\n\t\t0xf1, 0xab, 0x55, 0xea, 0x33, 0xa7, 0x12, 0x5e, 0xa1, 0xab, 0x6e, 0xd9, 0xb3, 0x64, 0x6c, 0x52,\n\t\t0xed, 0x6e, 0x85, 0xd5, 0xa7, 0x3f, 0x03, 0x00, 0x00, 0xff, 0xff, 0x7a, 0x3d, 0x1d, 0x21, 0xa3,\n\t\t0x07, 0x00, 0x00,\n\t},\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(clientConfig transport.ClientConfig, structField reflect.StructField) ShardDistributorExecutorAPIYARPCClient {\n\t\t\treturn NewShardDistributorExecutorAPIYARPCClient(clientConfig, protobuf.ClientBuilderOptions(clientConfig, structField)...)\n\t\t},\n\t)\n}\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/service.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/service.proto\n\npackage sharddistributorv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype GetShardOwnerRequest struct {\n\tShardKey             string   `protobuf:\"bytes,1,opt,name=shard_key,json=shardKey,proto3\" json:\"shard_key,omitempty\"`\n\tNamespace            string   `protobuf:\"bytes,2,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *GetShardOwnerRequest) Reset()         { *m = GetShardOwnerRequest{} }\nfunc (m *GetShardOwnerRequest) String() string { return proto.CompactTextString(m) }\nfunc (*GetShardOwnerRequest) ProtoMessage()    {}\nfunc (*GetShardOwnerRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{0}\n}\nfunc (m *GetShardOwnerRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetShardOwnerRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetShardOwnerRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetShardOwnerRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetShardOwnerRequest.Merge(m, src)\n}\nfunc (m *GetShardOwnerRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetShardOwnerRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetShardOwnerRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetShardOwnerRequest proto.InternalMessageInfo\n\nfunc (m *GetShardOwnerRequest) GetShardKey() string {\n\tif m != nil {\n\t\treturn m.ShardKey\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetShardOwnerRequest) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\ntype GetShardOwnerResponse struct {\n\tOwner                string            `protobuf:\"bytes,1,opt,name=owner,proto3\" json:\"owner,omitempty\"`\n\tNamespace            string            `protobuf:\"bytes,2,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tMetadata             map[string]string `protobuf:\"bytes,3,rep,name=metadata,proto3\" json:\"metadata,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tXXX_NoUnkeyedLiteral struct{}          `json:\"-\"`\n\tXXX_unrecognized     []byte            `json:\"-\"`\n\tXXX_sizecache        int32             `json:\"-\"`\n}\n\nfunc (m *GetShardOwnerResponse) Reset()         { *m = GetShardOwnerResponse{} }\nfunc (m *GetShardOwnerResponse) String() string { return proto.CompactTextString(m) }\nfunc (*GetShardOwnerResponse) ProtoMessage()    {}\nfunc (*GetShardOwnerResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{1}\n}\nfunc (m *GetShardOwnerResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *GetShardOwnerResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_GetShardOwnerResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *GetShardOwnerResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_GetShardOwnerResponse.Merge(m, src)\n}\nfunc (m *GetShardOwnerResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *GetShardOwnerResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_GetShardOwnerResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_GetShardOwnerResponse proto.InternalMessageInfo\n\nfunc (m *GetShardOwnerResponse) GetOwner() string {\n\tif m != nil {\n\t\treturn m.Owner\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetShardOwnerResponse) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\nfunc (m *GetShardOwnerResponse) GetMetadata() map[string]string {\n\tif m != nil {\n\t\treturn m.Metadata\n\t}\n\treturn nil\n}\n\ntype NamespaceNotFoundError struct {\n\tNamespace            string   `protobuf:\"bytes,1,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *NamespaceNotFoundError) Reset()         { *m = NamespaceNotFoundError{} }\nfunc (m *NamespaceNotFoundError) String() string { return proto.CompactTextString(m) }\nfunc (*NamespaceNotFoundError) ProtoMessage()    {}\nfunc (*NamespaceNotFoundError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{2}\n}\nfunc (m *NamespaceNotFoundError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *NamespaceNotFoundError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_NamespaceNotFoundError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *NamespaceNotFoundError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_NamespaceNotFoundError.Merge(m, src)\n}\nfunc (m *NamespaceNotFoundError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *NamespaceNotFoundError) XXX_DiscardUnknown() {\n\txxx_messageInfo_NamespaceNotFoundError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_NamespaceNotFoundError proto.InternalMessageInfo\n\nfunc (m *NamespaceNotFoundError) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\ntype ShardNotFoundError struct {\n\tNamespace            string   `protobuf:\"bytes,1,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tShardKey             string   `protobuf:\"bytes,2,opt,name=shard_key,json=shardKey,proto3\" json:\"shard_key,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ShardNotFoundError) Reset()         { *m = ShardNotFoundError{} }\nfunc (m *ShardNotFoundError) String() string { return proto.CompactTextString(m) }\nfunc (*ShardNotFoundError) ProtoMessage()    {}\nfunc (*ShardNotFoundError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{3}\n}\nfunc (m *ShardNotFoundError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ShardNotFoundError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ShardNotFoundError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ShardNotFoundError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ShardNotFoundError.Merge(m, src)\n}\nfunc (m *ShardNotFoundError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ShardNotFoundError) XXX_DiscardUnknown() {\n\txxx_messageInfo_ShardNotFoundError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ShardNotFoundError proto.InternalMessageInfo\n\nfunc (m *ShardNotFoundError) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\nfunc (m *ShardNotFoundError) GetShardKey() string {\n\tif m != nil {\n\t\treturn m.ShardKey\n\t}\n\treturn \"\"\n}\n\ntype WatchNamespaceStateRequest struct {\n\tNamespace            string   `protobuf:\"bytes,1,opt,name=namespace,proto3\" json:\"namespace,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *WatchNamespaceStateRequest) Reset()         { *m = WatchNamespaceStateRequest{} }\nfunc (m *WatchNamespaceStateRequest) String() string { return proto.CompactTextString(m) }\nfunc (*WatchNamespaceStateRequest) ProtoMessage()    {}\nfunc (*WatchNamespaceStateRequest) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{4}\n}\nfunc (m *WatchNamespaceStateRequest) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *WatchNamespaceStateRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_WatchNamespaceStateRequest.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *WatchNamespaceStateRequest) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_WatchNamespaceStateRequest.Merge(m, src)\n}\nfunc (m *WatchNamespaceStateRequest) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *WatchNamespaceStateRequest) XXX_DiscardUnknown() {\n\txxx_messageInfo_WatchNamespaceStateRequest.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_WatchNamespaceStateRequest proto.InternalMessageInfo\n\nfunc (m *WatchNamespaceStateRequest) GetNamespace() string {\n\tif m != nil {\n\t\treturn m.Namespace\n\t}\n\treturn \"\"\n}\n\ntype WatchNamespaceStateResponse struct {\n\t// Executor information with assigned shards\n\tExecutors            []*ExecutorInfo `protobuf:\"bytes,1,rep,name=executors,proto3\" json:\"executors,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}        `json:\"-\"`\n\tXXX_unrecognized     []byte          `json:\"-\"`\n\tXXX_sizecache        int32           `json:\"-\"`\n}\n\nfunc (m *WatchNamespaceStateResponse) Reset()         { *m = WatchNamespaceStateResponse{} }\nfunc (m *WatchNamespaceStateResponse) String() string { return proto.CompactTextString(m) }\nfunc (*WatchNamespaceStateResponse) ProtoMessage()    {}\nfunc (*WatchNamespaceStateResponse) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{5}\n}\nfunc (m *WatchNamespaceStateResponse) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *WatchNamespaceStateResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_WatchNamespaceStateResponse.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *WatchNamespaceStateResponse) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_WatchNamespaceStateResponse.Merge(m, src)\n}\nfunc (m *WatchNamespaceStateResponse) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *WatchNamespaceStateResponse) XXX_DiscardUnknown() {\n\txxx_messageInfo_WatchNamespaceStateResponse.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_WatchNamespaceStateResponse proto.InternalMessageInfo\n\nfunc (m *WatchNamespaceStateResponse) GetExecutors() []*ExecutorInfo {\n\tif m != nil {\n\t\treturn m.Executors\n\t}\n\treturn nil\n}\n\ntype ExecutorInfo struct {\n\tExecutorId           string            `protobuf:\"bytes,1,opt,name=executor_id,json=executorId,proto3\" json:\"executor_id,omitempty\"`\n\tMetadata             map[string]string `protobuf:\"bytes,2,rep,name=metadata,proto3\" json:\"metadata,omitempty\" protobuf_key:\"bytes,1,opt,name=key,proto3\" protobuf_val:\"bytes,2,opt,name=value,proto3\"`\n\tShards               []*Shard          `protobuf:\"bytes,3,rep,name=shards,proto3\" json:\"shards,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}          `json:\"-\"`\n\tXXX_unrecognized     []byte            `json:\"-\"`\n\tXXX_sizecache        int32             `json:\"-\"`\n}\n\nfunc (m *ExecutorInfo) Reset()         { *m = ExecutorInfo{} }\nfunc (m *ExecutorInfo) String() string { return proto.CompactTextString(m) }\nfunc (*ExecutorInfo) ProtoMessage()    {}\nfunc (*ExecutorInfo) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{6}\n}\nfunc (m *ExecutorInfo) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ExecutorInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ExecutorInfo.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ExecutorInfo) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ExecutorInfo.Merge(m, src)\n}\nfunc (m *ExecutorInfo) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ExecutorInfo) XXX_DiscardUnknown() {\n\txxx_messageInfo_ExecutorInfo.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ExecutorInfo proto.InternalMessageInfo\n\nfunc (m *ExecutorInfo) GetExecutorId() string {\n\tif m != nil {\n\t\treturn m.ExecutorId\n\t}\n\treturn \"\"\n}\n\nfunc (m *ExecutorInfo) GetMetadata() map[string]string {\n\tif m != nil {\n\t\treturn m.Metadata\n\t}\n\treturn nil\n}\n\nfunc (m *ExecutorInfo) GetShards() []*Shard {\n\tif m != nil {\n\t\treturn m.Shards\n\t}\n\treturn nil\n}\n\ntype Shard struct {\n\tShardKey             string   `protobuf:\"bytes,1,opt,name=shard_key,json=shardKey,proto3\" json:\"shard_key,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *Shard) Reset()         { *m = Shard{} }\nfunc (m *Shard) String() string { return proto.CompactTextString(m) }\nfunc (*Shard) ProtoMessage()    {}\nfunc (*Shard) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0055bfd59dff1f95, []int{7}\n}\nfunc (m *Shard) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *Shard) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_Shard.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *Shard) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_Shard.Merge(m, src)\n}\nfunc (m *Shard) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *Shard) XXX_DiscardUnknown() {\n\txxx_messageInfo_Shard.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_Shard proto.InternalMessageInfo\n\nfunc (m *Shard) GetShardKey() string {\n\tif m != nil {\n\t\treturn m.ShardKey\n\t}\n\treturn \"\"\n}\n\nfunc init() {\n\tproto.RegisterType((*GetShardOwnerRequest)(nil), \"uber.cadence.sharddistributor.v1.GetShardOwnerRequest\")\n\tproto.RegisterType((*GetShardOwnerResponse)(nil), \"uber.cadence.sharddistributor.v1.GetShardOwnerResponse\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.sharddistributor.v1.GetShardOwnerResponse.MetadataEntry\")\n\tproto.RegisterType((*NamespaceNotFoundError)(nil), \"uber.cadence.sharddistributor.v1.NamespaceNotFoundError\")\n\tproto.RegisterType((*ShardNotFoundError)(nil), \"uber.cadence.sharddistributor.v1.ShardNotFoundError\")\n\tproto.RegisterType((*WatchNamespaceStateRequest)(nil), \"uber.cadence.sharddistributor.v1.WatchNamespaceStateRequest\")\n\tproto.RegisterType((*WatchNamespaceStateResponse)(nil), \"uber.cadence.sharddistributor.v1.WatchNamespaceStateResponse\")\n\tproto.RegisterType((*ExecutorInfo)(nil), \"uber.cadence.sharddistributor.v1.ExecutorInfo\")\n\tproto.RegisterMapType((map[string]string)(nil), \"uber.cadence.sharddistributor.v1.ExecutorInfo.MetadataEntry\")\n\tproto.RegisterType((*Shard)(nil), \"uber.cadence.sharddistributor.v1.Shard\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/sharddistributor/v1/service.proto\", fileDescriptor_0055bfd59dff1f95)\n}\n\nvar fileDescriptor_0055bfd59dff1f95 = []byte{\n\t// 501 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xa4, 0x54, 0xd1, 0x6e, 0xd3, 0x30,\n\t0x14, 0x95, 0x53, 0x6d, 0x5a, 0xef, 0x98, 0x84, 0xbc, 0x81, 0xaa, 0x0e, 0x95, 0x2a, 0x42, 0x62,\n\t0x4f, 0x0e, 0x1d, 0xd2, 0x40, 0x1b, 0x08, 0x81, 0x28, 0xa8, 0x02, 0x36, 0xc8, 0x1e, 0x86, 0x78,\n\t0x99, 0xdc, 0xe4, 0xb2, 0x46, 0xa3, 0x71, 0xb1, 0x9d, 0x40, 0xdf, 0x78, 0xe3, 0x07, 0xf8, 0x0f,\n\t0x7e, 0x83, 0x47, 0x3e, 0x01, 0xf5, 0x95, 0x9f, 0x40, 0x71, 0x9d, 0x75, 0x89, 0xba, 0x96, 0x6e,\n\t0x6f, 0xc9, 0xbd, 0xf7, 0x1c, 0x1f, 0xfb, 0x1c, 0x1b, 0x58, 0xd2, 0x45, 0xe9, 0x05, 0x3c, 0xc4,\n\t0x38, 0x40, 0x4f, 0xf5, 0xb8, 0x0c, 0xc3, 0x48, 0x69, 0x19, 0x75, 0x13, 0x2d, 0xa4, 0x97, 0xb6,\n\t0x3c, 0x85, 0x32, 0x8d, 0x02, 0x64, 0x03, 0x29, 0xb4, 0xa0, 0xcd, 0x6c, 0x9e, 0xd9, 0x79, 0x56,\n\t0x9e, 0x67, 0x69, 0xcb, 0x7d, 0x07, 0x1b, 0x2f, 0x51, 0x1f, 0x66, 0x9d, 0x83, 0x2f, 0x31, 0x4a,\n\t0x1f, 0x3f, 0x27, 0xa8, 0x34, 0xdd, 0x84, 0xaa, 0x19, 0x3f, 0x3e, 0xc5, 0x61, 0x8d, 0x34, 0xc9,\n\t0x56, 0xd5, 0x5f, 0x31, 0x85, 0x57, 0x38, 0xa4, 0xb7, 0xa0, 0x1a, 0xf3, 0x3e, 0xaa, 0x01, 0x0f,\n\t0xb0, 0xe6, 0x98, 0xe6, 0xa4, 0xe0, 0xfe, 0x25, 0x70, 0xa3, 0xc4, 0xa9, 0x06, 0x22, 0x56, 0x48,\n\t0x37, 0x60, 0x49, 0x64, 0x05, 0x4b, 0x38, 0xfe, 0x99, 0xcd, 0x46, 0x39, 0xac, 0xf4, 0x51, 0xf3,\n\t0x90, 0x6b, 0x5e, 0xab, 0x34, 0x2b, 0x5b, 0xab, 0xdb, 0x6d, 0x36, 0x6f, 0x57, 0x6c, 0xea, 0xf2,\n\t0xec, 0x8d, 0xe5, 0x69, 0xc7, 0x5a, 0x0e, 0xfd, 0x33, 0xda, 0xfa, 0x1e, 0xac, 0x15, 0x5a, 0xf4,\n\t0x3a, 0x54, 0x26, 0xdb, 0xce, 0x3e, 0x33, 0xe5, 0x29, 0xff, 0x94, 0xe4, 0xfa, 0xc6, 0x3f, 0xbb,\n\t0xce, 0x43, 0xe2, 0xee, 0xc0, 0xcd, 0xfd, 0x5c, 0xec, 0xbe, 0xd0, 0x2f, 0x44, 0x12, 0x87, 0x6d,\n\t0x29, 0x45, 0x69, 0x5f, 0xa4, 0x7c, 0x4a, 0x07, 0x40, 0x8d, 0xc4, 0x05, 0x30, 0x45, 0x53, 0x9c,\n\t0xa2, 0x29, 0xee, 0x2e, 0xd4, 0x8f, 0xb8, 0x0e, 0x7a, 0x67, 0x6a, 0x0e, 0x35, 0xd7, 0x98, 0xfb,\n\t0x39, 0x5b, 0xcc, 0x29, 0x6c, 0x4e, 0xc5, 0x5a, 0xdf, 0x5e, 0x43, 0x15, 0xbf, 0x62, 0x90, 0x9d,\n\t0xae, 0xaa, 0x11, 0x63, 0x02, 0x9b, 0x6f, 0x42, 0xdb, 0x42, 0x3a, 0xf1, 0x47, 0xe1, 0x4f, 0x08,\n\t0xdc, 0xef, 0x0e, 0x5c, 0x3b, 0xdf, 0xa3, 0xb7, 0x61, 0x35, 0xef, 0x1e, 0x47, 0xa1, 0x55, 0x07,\n\t0x79, 0xa9, 0x13, 0xd2, 0xf7, 0xe7, 0x32, 0xe0, 0x98, 0xe5, 0x1f, 0x2d, 0xb6, 0xfc, 0x45, 0xd6,\n\t0xd3, 0x27, 0xb0, 0x6c, 0xb0, 0xca, 0x66, 0xeb, 0xee, 0x7c, 0x5e, 0xe3, 0x9a, 0x6f, 0x61, 0x57,\n\t0xcb, 0xce, 0x1d, 0x58, 0x32, 0x6c, 0x33, 0x6f, 0xdb, 0xf6, 0x4f, 0x07, 0xd6, 0xcd, 0xd8, 0xf3,\n\t0x89, 0x90, 0xa7, 0x6f, 0x3b, 0xf4, 0x1b, 0x81, 0xb5, 0x42, 0xd0, 0xe9, 0xce, 0xc2, 0x37, 0xc3,\n\t0x84, 0xa3, 0xfe, 0xe0, 0x92, 0x37, 0x8a, 0xfe, 0x20, 0xb0, 0x3e, 0x25, 0x38, 0xf4, 0x3f, 0xec,\n\t0xb9, 0x38, 0xab, 0xf5, 0xc7, 0x97, 0x44, 0x8f, 0x45, 0xdd, 0x23, 0xcf, 0x8e, 0x7e, 0x8d, 0x1a,\n\t0xe4, 0xf7, 0xa8, 0x41, 0xfe, 0x8c, 0x1a, 0xe4, 0x43, 0xe7, 0x24, 0xd2, 0xbd, 0xa4, 0xcb, 0x02,\n\t0xd1, 0xf7, 0x0a, 0xef, 0x27, 0x3b, 0xc1, 0xd8, 0x33, 0x0f, 0xe5, 0xb4, 0xa7, 0x74, 0xaf, 0x5c,\n\t0x4b, 0x5b, 0xdd, 0x65, 0x33, 0x7d, 0xff, 0x5f, 0x00, 0x00, 0x00, 0xff, 0xff, 0xcc, 0xf7, 0x0c,\n\t0x43, 0x88, 0x05, 0x00, 0x00,\n}\n\nfunc (m *GetShardOwnerRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetShardOwnerRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetShardOwnerRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.ShardKey) > 0 {\n\t\ti -= len(m.ShardKey)\n\t\tcopy(dAtA[i:], m.ShardKey)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ShardKey)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *GetShardOwnerResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *GetShardOwnerResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *GetShardOwnerResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k := range m.Metadata {\n\t\t\tv := m.Metadata[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Owner) > 0 {\n\t\ti -= len(m.Owner)\n\t\tcopy(dAtA[i:], m.Owner)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Owner)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *NamespaceNotFoundError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *NamespaceNotFoundError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *NamespaceNotFoundError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ShardNotFoundError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ShardNotFoundError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ShardNotFoundError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ShardKey) > 0 {\n\t\ti -= len(m.ShardKey)\n\t\tcopy(dAtA[i:], m.ShardKey)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ShardKey)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *WatchNamespaceStateRequest) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *WatchNamespaceStateRequest) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *WatchNamespaceStateRequest) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Namespace) > 0 {\n\t\ti -= len(m.Namespace)\n\t\tcopy(dAtA[i:], m.Namespace)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.Namespace)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *WatchNamespaceStateResponse) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *WatchNamespaceStateResponse) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *WatchNamespaceStateResponse) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Executors) > 0 {\n\t\tfor iNdEx := len(m.Executors) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Executors[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t}\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ExecutorInfo) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ExecutorInfo) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ExecutorInfo) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Shards) > 0 {\n\t\tfor iNdEx := len(m.Shards) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Shards[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintService(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x1a\n\t\t}\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k := range m.Metadata {\n\t\t\tv := m.Metadata[k]\n\t\t\tbaseI := i\n\t\t\ti -= len(v)\n\t\t\tcopy(dAtA[i:], v)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(v)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t\ti -= len(k)\n\t\t\tcopy(dAtA[i:], k)\n\t\t\ti = encodeVarintService(dAtA, i, uint64(len(k)))\n\t\t\ti--\n\t\t\tdAtA[i] = 0xa\n\t\t\ti = encodeVarintService(dAtA, i, uint64(baseI-i))\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif len(m.ExecutorId) > 0 {\n\t\ti -= len(m.ExecutorId)\n\t\tcopy(dAtA[i:], m.ExecutorId)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ExecutorId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *Shard) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *Shard) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Shard) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.ShardKey) > 0 {\n\t\ti -= len(m.ShardKey)\n\t\tcopy(dAtA[i:], m.ShardKey)\n\t\ti = encodeVarintService(dAtA, i, uint64(len(m.ShardKey)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintService(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovService(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *GetShardOwnerRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ShardKey)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *GetShardOwnerResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Owner)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k, v := range m.Metadata {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *NamespaceNotFoundError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ShardNotFoundError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tl = len(m.ShardKey)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *WatchNamespaceStateRequest) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Namespace)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *WatchNamespaceStateResponse) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif len(m.Executors) > 0 {\n\t\tfor _, e := range m.Executors {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ExecutorInfo) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ExecutorId)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif len(m.Metadata) > 0 {\n\t\tfor k, v := range m.Metadata {\n\t\t\t_ = k\n\t\t\t_ = v\n\t\t\tmapEntrySize := 1 + len(k) + sovService(uint64(len(k))) + 1 + len(v) + sovService(uint64(len(v)))\n\t\t\tn += mapEntrySize + 1 + sovService(uint64(mapEntrySize))\n\t\t}\n\t}\n\tif len(m.Shards) > 0 {\n\t\tfor _, e := range m.Shards {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovService(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *Shard) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ShardKey)\n\tif l > 0 {\n\t\tn += 1 + l + sovService(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovService(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozService(x uint64) (n int) {\n\treturn sovService(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *GetShardOwnerRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetShardOwnerRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetShardOwnerRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardKey\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardKey = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *GetShardOwnerResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: GetShardOwnerResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: GetShardOwnerResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Owner\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Owner = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Metadata\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Metadata == nil {\n\t\t\t\tm.Metadata = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Metadata[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *NamespaceNotFoundError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: NamespaceNotFoundError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: NamespaceNotFoundError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ShardNotFoundError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ShardNotFoundError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ShardNotFoundError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardKey\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardKey = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *WatchNamespaceStateRequest) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: WatchNamespaceStateRequest: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: WatchNamespaceStateRequest: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Namespace\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Namespace = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *WatchNamespaceStateResponse) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: WatchNamespaceStateResponse: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: WatchNamespaceStateResponse: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Executors\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Executors = append(m.Executors, &ExecutorInfo{})\n\t\t\tif err := m.Executors[len(m.Executors)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ExecutorInfo) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ExecutorInfo: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ExecutorInfo: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ExecutorId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ExecutorId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Metadata\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.Metadata == nil {\n\t\t\t\tm.Metadata = make(map[string]string)\n\t\t\t}\n\t\t\tvar mapkey string\n\t\t\tvar mapvalue string\n\t\t\tfor iNdEx < postIndex {\n\t\t\t\tentryPreIndex := iNdEx\n\t\t\t\tvar wire uint64\n\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t}\n\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\tiNdEx++\n\t\t\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfieldNum := int32(wire >> 3)\n\t\t\t\tif fieldNum == 1 {\n\t\t\t\t\tvar stringLenmapkey uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapkey |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapkey := int(stringLenmapkey)\n\t\t\t\t\tif intStringLenmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapkey := iNdEx + intStringLenmapkey\n\t\t\t\t\tif postStringIndexmapkey < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapkey > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapkey = string(dAtA[iNdEx:postStringIndexmapkey])\n\t\t\t\t\tiNdEx = postStringIndexmapkey\n\t\t\t\t} else if fieldNum == 2 {\n\t\t\t\t\tvar stringLenmapvalue uint64\n\t\t\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\t\t\tif shift >= 64 {\n\t\t\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif iNdEx >= l {\n\t\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t\t}\n\t\t\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\t\t\tiNdEx++\n\t\t\t\t\t\tstringLenmapvalue |= uint64(b&0x7F) << shift\n\t\t\t\t\t\tif b < 0x80 {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tintStringLenmapvalue := int(stringLenmapvalue)\n\t\t\t\t\tif intStringLenmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tpostStringIndexmapvalue := iNdEx + intStringLenmapvalue\n\t\t\t\t\tif postStringIndexmapvalue < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif postStringIndexmapvalue > l {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tmapvalue = string(dAtA[iNdEx:postStringIndexmapvalue])\n\t\t\t\t\tiNdEx = postStringIndexmapvalue\n\t\t\t\t} else {\n\t\t\t\t\tiNdEx = entryPreIndex\n\t\t\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\t\t\treturn ErrInvalidLengthService\n\t\t\t\t\t}\n\t\t\t\t\tif (iNdEx + skippy) > postIndex {\n\t\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t\t}\n\t\t\t\t\tiNdEx += skippy\n\t\t\t\t}\n\t\t\t}\n\t\t\tm.Metadata[mapkey] = mapvalue\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Shards\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Shards = append(m.Shards, &Shard{})\n\t\t\tif err := m.Shards[len(m.Shards)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *Shard) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: Shard: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: Shard: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ShardKey\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ShardKey = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipService(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthService\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipService(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowService\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthService\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupService\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthService\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthService        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowService          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupService = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/sharddistributor/v1/service.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/sharddistributor/v1/service.proto\n\npackage sharddistributorv1\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"reflect\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/gogo/protobuf/proto\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/api/x/restriction\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/encoding/protobuf/reflection\"\n)\n\nvar _ = ioutil.NopCloser\n\n// ShardDistributorAPIYARPCClient is the YARPC client-side interface for the ShardDistributorAPI service.\ntype ShardDistributorAPIYARPCClient interface {\n\tGetShardOwner(context.Context, *GetShardOwnerRequest, ...yarpc.CallOption) (*GetShardOwnerResponse, error)\n\tWatchNamespaceState(context.Context, *WatchNamespaceStateRequest, ...yarpc.CallOption) (ShardDistributorAPIServiceWatchNamespaceStateYARPCClient, error)\n}\n\n// ShardDistributorAPIServiceWatchNamespaceStateYARPCClient receives WatchNamespaceStateResponses, returning io.EOF when the stream is complete.\ntype ShardDistributorAPIServiceWatchNamespaceStateYARPCClient interface {\n\tContext() context.Context\n\tRecv(...yarpc.StreamOption) (*WatchNamespaceStateResponse, error)\n\tCloseSend(...yarpc.StreamOption) error\n}\n\nfunc newShardDistributorAPIYARPCClient(clientConfig transport.ClientConfig, anyResolver jsonpb.AnyResolver, options ...protobuf.ClientOption) ShardDistributorAPIYARPCClient {\n\treturn &_ShardDistributorAPIYARPCCaller{protobuf.NewStreamClient(\n\t\tprotobuf.ClientParams{\n\t\t\tServiceName:  \"uber.cadence.sharddistributor.v1.ShardDistributorAPI\",\n\t\t\tClientConfig: clientConfig,\n\t\t\tAnyResolver:  anyResolver,\n\t\t\tOptions:      options,\n\t\t},\n\t)}\n}\n\n// NewShardDistributorAPIYARPCClient builds a new YARPC client for the ShardDistributorAPI service.\nfunc NewShardDistributorAPIYARPCClient(clientConfig transport.ClientConfig, options ...protobuf.ClientOption) ShardDistributorAPIYARPCClient {\n\treturn newShardDistributorAPIYARPCClient(clientConfig, nil, options...)\n}\n\n// ShardDistributorAPIYARPCServer is the YARPC server-side interface for the ShardDistributorAPI service.\ntype ShardDistributorAPIYARPCServer interface {\n\tGetShardOwner(context.Context, *GetShardOwnerRequest) (*GetShardOwnerResponse, error)\n\tWatchNamespaceState(*WatchNamespaceStateRequest, ShardDistributorAPIServiceWatchNamespaceStateYARPCServer) error\n}\n\n// ShardDistributorAPIServiceWatchNamespaceStateYARPCServer sends WatchNamespaceStateResponses.\ntype ShardDistributorAPIServiceWatchNamespaceStateYARPCServer interface {\n\tContext() context.Context\n\tSend(*WatchNamespaceStateResponse, ...yarpc.StreamOption) error\n}\n\ntype buildShardDistributorAPIYARPCProceduresParams struct {\n\tServer      ShardDistributorAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver\n}\n\nfunc buildShardDistributorAPIYARPCProcedures(params buildShardDistributorAPIYARPCProceduresParams) []transport.Procedure {\n\thandler := &_ShardDistributorAPIYARPCHandler{params.Server}\n\treturn protobuf.BuildProcedures(\n\t\tprotobuf.BuildProceduresParams{\n\t\t\tServiceName: \"uber.cadence.sharddistributor.v1.ShardDistributorAPI\",\n\t\t\tUnaryHandlerParams: []protobuf.BuildProceduresUnaryHandlerParams{\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"GetShardOwner\",\n\t\t\t\t\tHandler: protobuf.NewUnaryHandler(\n\t\t\t\t\t\tprotobuf.UnaryHandlerParams{\n\t\t\t\t\t\t\tHandle:      handler.GetShardOwner,\n\t\t\t\t\t\t\tNewRequest:  newShardDistributorAPIServiceGetShardOwnerYARPCRequest,\n\t\t\t\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tOnewayHandlerParams: []protobuf.BuildProceduresOnewayHandlerParams{},\n\t\t\tStreamHandlerParams: []protobuf.BuildProceduresStreamHandlerParams{\n\n\t\t\t\t{\n\t\t\t\t\tMethodName: \"WatchNamespaceState\",\n\t\t\t\t\tHandler: protobuf.NewStreamHandler(\n\t\t\t\t\t\tprotobuf.StreamHandlerParams{\n\t\t\t\t\t\t\tHandle: handler.WatchNamespaceState,\n\t\t\t\t\t\t},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t)\n}\n\n// BuildShardDistributorAPIYARPCProcedures prepares an implementation of the ShardDistributorAPI service for YARPC registration.\nfunc BuildShardDistributorAPIYARPCProcedures(server ShardDistributorAPIYARPCServer) []transport.Procedure {\n\treturn buildShardDistributorAPIYARPCProcedures(buildShardDistributorAPIYARPCProceduresParams{Server: server})\n}\n\n// FxShardDistributorAPIYARPCClientParams defines the input\n// for NewFxShardDistributorAPIYARPCClient. It provides the\n// paramaters to get a ShardDistributorAPIYARPCClient in an\n// Fx application.\ntype FxShardDistributorAPIYARPCClientParams struct {\n\tfx.In\n\n\tProvider    yarpc.ClientConfig\n\tAnyResolver jsonpb.AnyResolver  `name:\"yarpcfx\" optional:\"true\"`\n\tRestriction restriction.Checker `optional:\"true\"`\n}\n\n// FxShardDistributorAPIYARPCClientResult defines the output\n// of NewFxShardDistributorAPIYARPCClient. It provides a\n// ShardDistributorAPIYARPCClient to an Fx application.\ntype FxShardDistributorAPIYARPCClientResult struct {\n\tfx.Out\n\n\tClient ShardDistributorAPIYARPCClient\n\n\t// We are using an fx.Out struct here instead of just returning a client\n\t// so that we can add more values or add named versions of the client in\n\t// the future without breaking any existing code.\n}\n\n// NewFxShardDistributorAPIYARPCClient provides a ShardDistributorAPIYARPCClient\n// to an Fx application using the given name for routing.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorAPIYARPCClient(\"service-name\"),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorAPIYARPCClient(name string, options ...protobuf.ClientOption) interface{} {\n\treturn func(params FxShardDistributorAPIYARPCClientParams) FxShardDistributorAPIYARPCClientResult {\n\t\tcc := params.Provider.ClientConfig(name)\n\n\t\tif params.Restriction != nil {\n\t\t\tif namer, ok := cc.GetUnaryOutbound().(transport.Namer); ok {\n\t\t\t\tif err := params.Restriction.Check(protobuf.Encoding, namer.TransportName()); err != nil {\n\t\t\t\t\tpanic(err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn FxShardDistributorAPIYARPCClientResult{\n\t\t\tClient: newShardDistributorAPIYARPCClient(cc, params.AnyResolver, options...),\n\t\t}\n\t}\n}\n\n// FxShardDistributorAPIYARPCProceduresParams defines the input\n// for NewFxShardDistributorAPIYARPCProcedures. It provides the\n// paramaters to get ShardDistributorAPIYARPCServer procedures in an\n// Fx application.\ntype FxShardDistributorAPIYARPCProceduresParams struct {\n\tfx.In\n\n\tServer      ShardDistributorAPIYARPCServer\n\tAnyResolver jsonpb.AnyResolver `name:\"yarpcfx\" optional:\"true\"`\n}\n\n// FxShardDistributorAPIYARPCProceduresResult defines the output\n// of NewFxShardDistributorAPIYARPCProcedures. It provides\n// ShardDistributorAPIYARPCServer procedures to an Fx application.\n//\n// The procedures are provided to the \"yarpcfx\" value group.\n// Dig 1.2 or newer must be used for this feature to work.\ntype FxShardDistributorAPIYARPCProceduresResult struct {\n\tfx.Out\n\n\tProcedures     []transport.Procedure `group:\"yarpcfx\"`\n\tReflectionMeta reflection.ServerMeta `group:\"yarpcfx\"`\n}\n\n// NewFxShardDistributorAPIYARPCProcedures provides ShardDistributorAPIYARPCServer procedures to an Fx application.\n// It expects a ShardDistributorAPIYARPCServer to be present in the container.\n//\n//\tfx.Provide(\n//\t  sharddistributorv1.NewFxShardDistributorAPIYARPCProcedures(),\n//\t  ...\n//\t)\nfunc NewFxShardDistributorAPIYARPCProcedures() interface{} {\n\treturn func(params FxShardDistributorAPIYARPCProceduresParams) FxShardDistributorAPIYARPCProceduresResult {\n\t\treturn FxShardDistributorAPIYARPCProceduresResult{\n\t\t\tProcedures: buildShardDistributorAPIYARPCProcedures(buildShardDistributorAPIYARPCProceduresParams{\n\t\t\t\tServer:      params.Server,\n\t\t\t\tAnyResolver: params.AnyResolver,\n\t\t\t}),\n\t\t\tReflectionMeta: ShardDistributorAPIReflectionMeta,\n\t\t}\n\t}\n}\n\n// ShardDistributorAPIReflectionMeta is the reflection server metadata\n// required for using the gRPC reflection protocol with YARPC.\n//\n// See https://github.com/grpc/grpc/blob/master/doc/server-reflection.md.\nvar ShardDistributorAPIReflectionMeta = reflection.ServerMeta{\n\tServiceName:     \"uber.cadence.sharddistributor.v1.ShardDistributorAPI\",\n\tFileDescriptors: yarpcFileDescriptorClosure0055bfd59dff1f95,\n}\n\ntype _ShardDistributorAPIYARPCCaller struct {\n\tstreamClient protobuf.StreamClient\n}\n\nfunc (c *_ShardDistributorAPIYARPCCaller) GetShardOwner(ctx context.Context, request *GetShardOwnerRequest, options ...yarpc.CallOption) (*GetShardOwnerResponse, error) {\n\tresponseMessage, err := c.streamClient.Call(ctx, \"GetShardOwner\", request, newShardDistributorAPIServiceGetShardOwnerYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*GetShardOwnerResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyShardDistributorAPIServiceGetShardOwnerYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_ShardDistributorAPIYARPCCaller) WatchNamespaceState(ctx context.Context, request *WatchNamespaceStateRequest, options ...yarpc.CallOption) (ShardDistributorAPIServiceWatchNamespaceStateYARPCClient, error) {\n\tstream, err := c.streamClient.CallStream(ctx, \"WatchNamespaceState\", options...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif err := stream.Send(request); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &_ShardDistributorAPIServiceWatchNamespaceStateYARPCClient{stream: stream}, nil\n}\n\ntype _ShardDistributorAPIYARPCHandler struct {\n\tserver ShardDistributorAPIYARPCServer\n}\n\nfunc (h *_ShardDistributorAPIYARPCHandler) GetShardOwner(ctx context.Context, requestMessage proto.Message) (proto.Message, error) {\n\tvar request *GetShardOwnerRequest\n\tvar ok bool\n\tif requestMessage != nil {\n\t\trequest, ok = requestMessage.(*GetShardOwnerRequest)\n\t\tif !ok {\n\t\t\treturn nil, protobuf.CastError(emptyShardDistributorAPIServiceGetShardOwnerYARPCRequest, requestMessage)\n\t\t}\n\t}\n\tresponse, err := h.server.GetShardOwner(ctx, request)\n\tif response == nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (h *_ShardDistributorAPIYARPCHandler) WatchNamespaceState(serverStream *protobuf.ServerStream) error {\n\trequestMessage, err := serverStream.Receive(newShardDistributorAPIServiceWatchNamespaceStateYARPCRequest)\n\tif requestMessage == nil {\n\t\treturn err\n\t}\n\n\trequest, ok := requestMessage.(*WatchNamespaceStateRequest)\n\tif !ok {\n\t\treturn protobuf.CastError(emptyShardDistributorAPIServiceWatchNamespaceStateYARPCRequest, requestMessage)\n\t}\n\treturn h.server.WatchNamespaceState(request, &_ShardDistributorAPIServiceWatchNamespaceStateYARPCServer{serverStream: serverStream})\n}\n\ntype _ShardDistributorAPIServiceWatchNamespaceStateYARPCClient struct {\n\tstream *protobuf.ClientStream\n}\n\nfunc (c *_ShardDistributorAPIServiceWatchNamespaceStateYARPCClient) Context() context.Context {\n\treturn c.stream.Context()\n}\n\nfunc (c *_ShardDistributorAPIServiceWatchNamespaceStateYARPCClient) Recv(options ...yarpc.StreamOption) (*WatchNamespaceStateResponse, error) {\n\tresponseMessage, err := c.stream.Receive(newShardDistributorAPIServiceWatchNamespaceStateYARPCResponse, options...)\n\tif responseMessage == nil {\n\t\treturn nil, err\n\t}\n\tresponse, ok := responseMessage.(*WatchNamespaceStateResponse)\n\tif !ok {\n\t\treturn nil, protobuf.CastError(emptyShardDistributorAPIServiceWatchNamespaceStateYARPCResponse, responseMessage)\n\t}\n\treturn response, err\n}\n\nfunc (c *_ShardDistributorAPIServiceWatchNamespaceStateYARPCClient) CloseSend(options ...yarpc.StreamOption) error {\n\treturn c.stream.Close(options...)\n}\n\ntype _ShardDistributorAPIServiceWatchNamespaceStateYARPCServer struct {\n\tserverStream *protobuf.ServerStream\n}\n\nfunc (s *_ShardDistributorAPIServiceWatchNamespaceStateYARPCServer) Context() context.Context {\n\treturn s.serverStream.Context()\n}\n\nfunc (s *_ShardDistributorAPIServiceWatchNamespaceStateYARPCServer) Send(response *WatchNamespaceStateResponse, options ...yarpc.StreamOption) error {\n\treturn s.serverStream.Send(response, options...)\n}\n\nfunc newShardDistributorAPIServiceGetShardOwnerYARPCRequest() proto.Message {\n\treturn &GetShardOwnerRequest{}\n}\n\nfunc newShardDistributorAPIServiceGetShardOwnerYARPCResponse() proto.Message {\n\treturn &GetShardOwnerResponse{}\n}\n\nfunc newShardDistributorAPIServiceWatchNamespaceStateYARPCRequest() proto.Message {\n\treturn &WatchNamespaceStateRequest{}\n}\n\nfunc newShardDistributorAPIServiceWatchNamespaceStateYARPCResponse() proto.Message {\n\treturn &WatchNamespaceStateResponse{}\n}\n\nvar (\n\temptyShardDistributorAPIServiceGetShardOwnerYARPCRequest        = &GetShardOwnerRequest{}\n\temptyShardDistributorAPIServiceGetShardOwnerYARPCResponse       = &GetShardOwnerResponse{}\n\temptyShardDistributorAPIServiceWatchNamespaceStateYARPCRequest  = &WatchNamespaceStateRequest{}\n\temptyShardDistributorAPIServiceWatchNamespaceStateYARPCResponse = &WatchNamespaceStateResponse{}\n)\n\nvar yarpcFileDescriptorClosure0055bfd59dff1f95 = [][]byte{\n\t// uber/cadence/sharddistributor/v1/service.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xa4, 0x54, 0xd1, 0x6e, 0xd3, 0x30,\n\t\t0x14, 0x95, 0x53, 0x6d, 0x5a, 0xef, 0x98, 0x84, 0xbc, 0x81, 0xaa, 0x0e, 0x89, 0x2a, 0x42, 0x62,\n\t\t0x4f, 0x0e, 0x1d, 0xd2, 0x40, 0x1b, 0x08, 0x81, 0x28, 0xa8, 0x1a, 0x6c, 0x90, 0x3d, 0x80, 0x78,\n\t\t0x99, 0xdc, 0xe4, 0xb2, 0x46, 0xa3, 0x76, 0xb1, 0x9d, 0x40, 0xdf, 0x78, 0xe3, 0x07, 0xf8, 0x0f,\n\t\t0x3e, 0x8c, 0x9f, 0x40, 0x71, 0x9d, 0x75, 0x89, 0xb2, 0x96, 0x8e, 0xb7, 0xe4, 0xde, 0x7b, 0x8e,\n\t\t0x8f, 0x7d, 0x8e, 0x0d, 0x2c, 0x1d, 0xa0, 0x0a, 0x22, 0x1e, 0xa3, 0x88, 0x30, 0xd0, 0x43, 0xae,\n\t\t0xe2, 0x38, 0xd1, 0x46, 0x25, 0x83, 0xd4, 0x48, 0x15, 0x64, 0xdd, 0x40, 0xa3, 0xca, 0x92, 0x08,\n\t\t0xd9, 0x58, 0x49, 0x23, 0x69, 0x27, 0x9f, 0x67, 0x6e, 0x9e, 0x55, 0xe7, 0x59, 0xd6, 0xf5, 0xdf,\n\t\t0xc3, 0xd6, 0x6b, 0x34, 0x27, 0x79, 0xe7, 0xf8, 0x9b, 0x40, 0x15, 0xe2, 0xd7, 0x14, 0xb5, 0xa1,\n\t\t0xdb, 0xd0, 0xb4, 0xe3, 0xa7, 0xe7, 0x38, 0x69, 0x91, 0x0e, 0xd9, 0x69, 0x86, 0x6b, 0xb6, 0x70,\n\t\t0x88, 0x13, 0x7a, 0x07, 0x9a, 0x82, 0x8f, 0x50, 0x8f, 0x79, 0x84, 0x2d, 0xcf, 0x36, 0x67, 0x05,\n\t\t0xff, 0x0f, 0x81, 0x5b, 0x15, 0x4e, 0x3d, 0x96, 0x42, 0x23, 0xdd, 0x82, 0x15, 0x99, 0x17, 0x1c,\n\t\t0xe1, 0xf4, 0x67, 0x3e, 0x1b, 0xe5, 0xb0, 0x36, 0x42, 0xc3, 0x63, 0x6e, 0x78, 0xab, 0xd1, 0x69,\n\t\t0xec, 0xac, 0xef, 0xf6, 0xd8, 0xa2, 0x5d, 0xb1, 0xda, 0xe5, 0xd9, 0x5b, 0xc7, 0xd3, 0x13, 0x46,\n\t\t0x4d, 0xc2, 0x0b, 0xda, 0xf6, 0x01, 0x6c, 0x94, 0x5a, 0xf4, 0x26, 0x34, 0x66, 0xdb, 0xce, 0x3f,\n\t\t0x73, 0xe5, 0x19, 0xff, 0x92, 0x16, 0xfa, 0xa6, 0x3f, 0xfb, 0xde, 0x63, 0xe2, 0xef, 0xc1, 0xed,\n\t\t0xa3, 0x42, 0xec, 0x91, 0x34, 0xaf, 0x64, 0x2a, 0xe2, 0x9e, 0x52, 0xb2, 0xb2, 0x2f, 0x52, 0x3d,\n\t\t0xa5, 0x63, 0xa0, 0x56, 0xe2, 0x12, 0x98, 0xb2, 0x29, 0x5e, 0xd9, 0x14, 0x7f, 0x1f, 0xda, 0x1f,\n\t\t0xb8, 0x89, 0x86, 0x17, 0x6a, 0x4e, 0x0c, 0x37, 0x58, 0xf8, 0x39, 0x5f, 0xcc, 0x39, 0x6c, 0xd7,\n\t\t0x62, 0x9d, 0x6f, 0x6f, 0xa0, 0x89, 0xdf, 0x31, 0xca, 0x4f, 0x57, 0xb7, 0x88, 0x35, 0x81, 0x2d,\n\t\t0x36, 0xa1, 0xe7, 0x20, 0x7d, 0xf1, 0x59, 0x86, 0x33, 0x02, 0xff, 0xa7, 0x07, 0x37, 0x2e, 0xf7,\n\t\t0xe8, 0x5d, 0x58, 0x2f, 0xba, 0xa7, 0x49, 0xec, 0xd4, 0x41, 0x51, 0xea, 0xc7, 0xf4, 0xe3, 0xa5,\n\t\t0x0c, 0x78, 0x76, 0xf9, 0x27, 0xcb, 0x2d, 0x7f, 0x95, 0xf5, 0xf4, 0x19, 0xac, 0x5a, 0xac, 0x76,\n\t\t0xd9, 0xba, 0xbf, 0x98, 0xd7, 0xba, 0x16, 0x3a, 0xd8, 0xff, 0x65, 0xe7, 0x1e, 0xac, 0x58, 0xb6,\n\t\t0xb9, 0xb7, 0x6d, 0xf7, 0xb7, 0x07, 0x9b, 0x76, 0xec, 0xe5, 0x4c, 0xc8, 0xf3, 0x77, 0x7d, 0xfa,\n\t\t0x83, 0xc0, 0x46, 0x29, 0xe8, 0x74, 0x6f, 0xe9, 0x9b, 0x61, 0xc3, 0xd1, 0x7e, 0x74, 0xcd, 0x1b,\n\t\t0x45, 0x7f, 0x11, 0xd8, 0xac, 0x09, 0x0e, 0xfd, 0x07, 0x7b, 0xae, 0xce, 0x6a, 0xfb, 0xe9, 0x35,\n\t\t0xd1, 0x53, 0x51, 0x0f, 0xc8, 0x8b, 0xc3, 0x4f, 0xfd, 0xb3, 0xc4, 0x0c, 0xd3, 0x01, 0x8b, 0xe4,\n\t\t0x28, 0x28, 0xbd, 0x99, 0xec, 0x0c, 0x45, 0x60, 0x1f, 0xc7, 0xba, 0xe7, 0xf3, 0xa0, 0x5a, 0xcb,\n\t\t0xba, 0x83, 0x55, 0x3b, 0xfd, 0xf0, 0x6f, 0x00, 0x00, 0x00, 0xff, 0xff, 0xfd, 0x73, 0x0d, 0x95,\n\t\t0x7c, 0x05, 0x00, 0x00,\n\t},\n}\n\nfunc init() {\n\tyarpc.RegisterClientBuilder(\n\t\tfunc(clientConfig transport.ClientConfig, structField reflect.StructField) ShardDistributorAPIYARPCClient {\n\t\t\treturn NewShardDistributorAPIYARPCClient(clientConfig, protobuf.ClientBuilderOptions(clientConfig, structField)...)\n\t\t},\n\t)\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/any.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/shared/v1/any.proto\n\npackage sharedv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\n// Any is a logical duplicate of google.protobuf.Any, but is intentionally\n// breaking compatibility because it will not be directly used as a\n// google.protobuf.Any in our high-level RPC mappers.\n//\n// The intent of the type is the same, but it is not intended to be directly\n// compatible with google.protobuf.Any - this blob is RPC-type agnostic by\n// design (as the underlying data may be transported over proto or thrift), and\n// the data-bytes may or may not be proto-encoded.\n//\n// This is intentionally different from DataBlob, which supports only a handful\n// of known encodings so it can be interpreted everywhere.  Any supports literally\n// any contents, and needs to be considered opaque until it is given to something\n// that is expecting it.\n//\n// See value_type to interpret the contents.\ntype Any struct {\n\t// Type-string describing value's contents, and intentionally avoiding the\n\t// name \"type\" as it is often a special term.\n\t// This should usually be a hard-coded string of some kind.\n\tValueType string `protobuf:\"bytes,1,opt,name=value_type,json=valueType,proto3\" json:\"value_type,omitempty\"`\n\t// Arbitrarily-encoded bytes, to be deserialized by a runtime implementation.\n\t// The contents are described by value_type.\n\tValue                []byte   `protobuf:\"bytes,2,opt,name=value,proto3\" json:\"value,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *Any) Reset()         { *m = Any{} }\nfunc (m *Any) String() string { return proto.CompactTextString(m) }\nfunc (*Any) ProtoMessage()    {}\nfunc (*Any) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3ed449e8097bca22, []int{0}\n}\nfunc (m *Any) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *Any) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_Any.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *Any) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_Any.Merge(m, src)\n}\nfunc (m *Any) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *Any) XXX_DiscardUnknown() {\n\txxx_messageInfo_Any.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_Any proto.InternalMessageInfo\n\nfunc (m *Any) GetValueType() string {\n\tif m != nil {\n\t\treturn m.ValueType\n\t}\n\treturn \"\"\n}\n\nfunc (m *Any) GetValue() []byte {\n\tif m != nil {\n\t\treturn m.Value\n\t}\n\treturn nil\n}\n\nfunc init() {\n\tproto.RegisterType((*Any)(nil), \"uber.cadence.shared.v1.Any\")\n}\n\nfunc init() { proto.RegisterFile(\"uber/cadence/shared/v1/any.proto\", fileDescriptor_3ed449e8097bca22) }\n\nvar fileDescriptor_3ed449e8097bca22 = []byte{\n\t// 170 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x28, 0x4d, 0x4a, 0x2d,\n\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t0x2f, 0x33, 0xd4, 0x4f, 0xcc, 0xab, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x12, 0x03, 0xa9,\n\t0xd0, 0x83, 0xaa, 0xd0, 0x83, 0xa8, 0xd0, 0x2b, 0x33, 0x54, 0xb2, 0xe2, 0x62, 0x76, 0xcc, 0xab,\n\t0x14, 0x92, 0xe5, 0xe2, 0x2a, 0x4b, 0xcc, 0x29, 0x4d, 0x8d, 0x2f, 0xa9, 0x2c, 0x48, 0x95, 0x60,\n\t0x54, 0x60, 0xd4, 0xe0, 0x0c, 0xe2, 0x04, 0x8b, 0x84, 0x54, 0x16, 0xa4, 0x0a, 0x89, 0x70, 0xb1,\n\t0x82, 0x39, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0x3c, 0x41, 0x10, 0x8e, 0x93, 0xf3, 0x89, 0x47, 0x72,\n\t0x8c, 0x17, 0x1e, 0xc9, 0x31, 0x3e, 0x78, 0x24, 0xc7, 0x18, 0x65, 0x9a, 0x9e, 0x59, 0x92, 0x51,\n\t0x9a, 0xa4, 0x97, 0x9c, 0x9f, 0xab, 0x8f, 0xe2, 0x1c, 0xbd, 0xf4, 0xd4, 0x3c, 0x7d, 0xb0, 0x2b,\n\t0x10, 0x2e, 0xb3, 0x86, 0xb0, 0xca, 0x0c, 0x93, 0xd8, 0xc0, 0x32, 0xc6, 0x80, 0x00, 0x00, 0x00,\n\t0xff, 0xff, 0x29, 0x98, 0x69, 0x21, 0xc3, 0x00, 0x00, 0x00,\n}\n\nfunc (m *Any) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *Any) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *Any) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Value) > 0 {\n\t\ti -= len(m.Value)\n\t\tcopy(dAtA[i:], m.Value)\n\t\ti = encodeVarintAny(dAtA, i, uint64(len(m.Value)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.ValueType) > 0 {\n\t\ti -= len(m.ValueType)\n\t\tcopy(dAtA[i:], m.ValueType)\n\t\ti = encodeVarintAny(dAtA, i, uint64(len(m.ValueType)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintAny(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovAny(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *Any) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.ValueType)\n\tif l > 0 {\n\t\tn += 1 + l + sovAny(uint64(l))\n\t}\n\tl = len(m.Value)\n\tif l > 0 {\n\t\tn += 1 + l + sovAny(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovAny(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozAny(x uint64) (n int) {\n\treturn sovAny(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *Any) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowAny\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: Any: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: Any: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ValueType\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowAny\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthAny\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthAny\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.ValueType = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Value\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowAny\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthAny\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthAny\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Value = append(m.Value[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.Value == nil {\n\t\t\t\tm.Value = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipAny(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthAny\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipAny(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowAny\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowAny\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowAny\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthAny\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupAny\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthAny\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthAny        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowAny          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupAny = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/shared/v1/any.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/shared/v1/any.proto\n\npackage sharedv1\n\nvar yarpcFileDescriptorClosure3ed449e8097bca22 = [][]byte{\n\t// uber/cadence/shared/v1/any.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x28, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x4f, 0xcc, 0xab, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x12, 0x03, 0xa9,\n\t\t0xd0, 0x83, 0xaa, 0xd0, 0x83, 0xa8, 0xd0, 0x2b, 0x33, 0x54, 0xb2, 0xe2, 0x62, 0x76, 0xcc, 0xab,\n\t\t0x14, 0x92, 0xe5, 0xe2, 0x2a, 0x4b, 0xcc, 0x29, 0x4d, 0x8d, 0x2f, 0xa9, 0x2c, 0x48, 0x95, 0x60,\n\t\t0x54, 0x60, 0xd4, 0xe0, 0x0c, 0xe2, 0x04, 0x8b, 0x84, 0x54, 0x16, 0xa4, 0x0a, 0x89, 0x70, 0xb1,\n\t\t0x82, 0x39, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0x3c, 0x41, 0x10, 0x8e, 0x93, 0x79, 0x94, 0x69, 0x7a,\n\t\t0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x8a, 0x13, 0xf4, 0xd2, 0x53, 0xf3,\n\t\t0xf4, 0xc1, 0x36, 0x23, 0x5c, 0x63, 0x0d, 0x61, 0x95, 0x19, 0x26, 0xb1, 0x81, 0x65, 0x8c, 0x01,\n\t\t0x01, 0x00, 0x00, 0xff, 0xff, 0x4e, 0x0a, 0xa2, 0x2e, 0xb7, 0x00, 0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/error.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/shared/v1/error.proto\n\npackage sharedv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n\tv11 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype CurrentBranchChangedError struct {\n\tCurrentBranchToken   []byte   `protobuf:\"bytes,1,opt,name=current_branch_token,json=currentBranchToken,proto3\" json:\"current_branch_token,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *CurrentBranchChangedError) Reset()         { *m = CurrentBranchChangedError{} }\nfunc (m *CurrentBranchChangedError) String() string { return proto.CompactTextString(m) }\nfunc (*CurrentBranchChangedError) ProtoMessage()    {}\nfunc (*CurrentBranchChangedError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{0}\n}\nfunc (m *CurrentBranchChangedError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *CurrentBranchChangedError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_CurrentBranchChangedError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *CurrentBranchChangedError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_CurrentBranchChangedError.Merge(m, src)\n}\nfunc (m *CurrentBranchChangedError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *CurrentBranchChangedError) XXX_DiscardUnknown() {\n\txxx_messageInfo_CurrentBranchChangedError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_CurrentBranchChangedError proto.InternalMessageInfo\n\nfunc (m *CurrentBranchChangedError) GetCurrentBranchToken() []byte {\n\tif m != nil {\n\t\treturn m.CurrentBranchToken\n\t}\n\treturn nil\n}\n\ntype InternalDataInconsistencyError struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *InternalDataInconsistencyError) Reset()         { *m = InternalDataInconsistencyError{} }\nfunc (m *InternalDataInconsistencyError) String() string { return proto.CompactTextString(m) }\nfunc (*InternalDataInconsistencyError) ProtoMessage()    {}\nfunc (*InternalDataInconsistencyError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{1}\n}\nfunc (m *InternalDataInconsistencyError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *InternalDataInconsistencyError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_InternalDataInconsistencyError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *InternalDataInconsistencyError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_InternalDataInconsistencyError.Merge(m, src)\n}\nfunc (m *InternalDataInconsistencyError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *InternalDataInconsistencyError) XXX_DiscardUnknown() {\n\txxx_messageInfo_InternalDataInconsistencyError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_InternalDataInconsistencyError proto.InternalMessageInfo\n\ntype EventAlreadyStartedError struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *EventAlreadyStartedError) Reset()         { *m = EventAlreadyStartedError{} }\nfunc (m *EventAlreadyStartedError) String() string { return proto.CompactTextString(m) }\nfunc (*EventAlreadyStartedError) ProtoMessage()    {}\nfunc (*EventAlreadyStartedError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{2}\n}\nfunc (m *EventAlreadyStartedError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *EventAlreadyStartedError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_EventAlreadyStartedError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *EventAlreadyStartedError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_EventAlreadyStartedError.Merge(m, src)\n}\nfunc (m *EventAlreadyStartedError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *EventAlreadyStartedError) XXX_DiscardUnknown() {\n\txxx_messageInfo_EventAlreadyStartedError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_EventAlreadyStartedError proto.InternalMessageInfo\n\ntype RemoteSyncMatchedError struct {\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *RemoteSyncMatchedError) Reset()         { *m = RemoteSyncMatchedError{} }\nfunc (m *RemoteSyncMatchedError) String() string { return proto.CompactTextString(m) }\nfunc (*RemoteSyncMatchedError) ProtoMessage()    {}\nfunc (*RemoteSyncMatchedError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{3}\n}\nfunc (m *RemoteSyncMatchedError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RemoteSyncMatchedError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RemoteSyncMatchedError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RemoteSyncMatchedError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RemoteSyncMatchedError.Merge(m, src)\n}\nfunc (m *RemoteSyncMatchedError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RemoteSyncMatchedError) XXX_DiscardUnknown() {\n\txxx_messageInfo_RemoteSyncMatchedError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RemoteSyncMatchedError proto.InternalMessageInfo\n\ntype RetryTaskV2Error struct {\n\tDomainId             string                  `protobuf:\"bytes,1,opt,name=domain_id,json=domainId,proto3\" json:\"domain_id,omitempty\"`\n\tWorkflowExecution    *v1.WorkflowExecution   `protobuf:\"bytes,2,opt,name=workflow_execution,json=workflowExecution,proto3\" json:\"workflow_execution,omitempty\"`\n\tStartEvent           *v11.VersionHistoryItem `protobuf:\"bytes,3,opt,name=start_event,json=startEvent,proto3\" json:\"start_event,omitempty\"`\n\tEndEvent             *v11.VersionHistoryItem `protobuf:\"bytes,4,opt,name=end_event,json=endEvent,proto3\" json:\"end_event,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}                `json:\"-\"`\n\tXXX_unrecognized     []byte                  `json:\"-\"`\n\tXXX_sizecache        int32                   `json:\"-\"`\n}\n\nfunc (m *RetryTaskV2Error) Reset()         { *m = RetryTaskV2Error{} }\nfunc (m *RetryTaskV2Error) String() string { return proto.CompactTextString(m) }\nfunc (*RetryTaskV2Error) ProtoMessage()    {}\nfunc (*RetryTaskV2Error) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{4}\n}\nfunc (m *RetryTaskV2Error) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *RetryTaskV2Error) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_RetryTaskV2Error.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *RetryTaskV2Error) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_RetryTaskV2Error.Merge(m, src)\n}\nfunc (m *RetryTaskV2Error) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *RetryTaskV2Error) XXX_DiscardUnknown() {\n\txxx_messageInfo_RetryTaskV2Error.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_RetryTaskV2Error proto.InternalMessageInfo\n\nfunc (m *RetryTaskV2Error) GetDomainId() string {\n\tif m != nil {\n\t\treturn m.DomainId\n\t}\n\treturn \"\"\n}\n\nfunc (m *RetryTaskV2Error) GetWorkflowExecution() *v1.WorkflowExecution {\n\tif m != nil {\n\t\treturn m.WorkflowExecution\n\t}\n\treturn nil\n}\n\nfunc (m *RetryTaskV2Error) GetStartEvent() *v11.VersionHistoryItem {\n\tif m != nil {\n\t\treturn m.StartEvent\n\t}\n\treturn nil\n}\n\nfunc (m *RetryTaskV2Error) GetEndEvent() *v11.VersionHistoryItem {\n\tif m != nil {\n\t\treturn m.EndEvent\n\t}\n\treturn nil\n}\n\ntype ShardOwnershipLostError struct {\n\tOwner                string   `protobuf:\"bytes,1,opt,name=owner,proto3\" json:\"owner,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *ShardOwnershipLostError) Reset()         { *m = ShardOwnershipLostError{} }\nfunc (m *ShardOwnershipLostError) String() string { return proto.CompactTextString(m) }\nfunc (*ShardOwnershipLostError) ProtoMessage()    {}\nfunc (*ShardOwnershipLostError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{5}\n}\nfunc (m *ShardOwnershipLostError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *ShardOwnershipLostError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_ShardOwnershipLostError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *ShardOwnershipLostError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_ShardOwnershipLostError.Merge(m, src)\n}\nfunc (m *ShardOwnershipLostError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *ShardOwnershipLostError) XXX_DiscardUnknown() {\n\txxx_messageInfo_ShardOwnershipLostError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_ShardOwnershipLostError proto.InternalMessageInfo\n\nfunc (m *ShardOwnershipLostError) GetOwner() string {\n\tif m != nil {\n\t\treturn m.Owner\n\t}\n\treturn \"\"\n}\n\ntype TaskListNotOwnedByHostError struct {\n\tOwnedByIdentity      string   `protobuf:\"bytes,1,opt,name=owned_by_identity,json=ownedByIdentity,proto3\" json:\"owned_by_identity,omitempty\"`\n\tMyIdentity           string   `protobuf:\"bytes,2,opt,name=my_identity,json=myIdentity,proto3\" json:\"my_identity,omitempty\"`\n\tTaskListName         string   `protobuf:\"bytes,3,opt,name=task_list_name,json=taskListName,proto3\" json:\"task_list_name,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{} `json:\"-\"`\n\tXXX_unrecognized     []byte   `json:\"-\"`\n\tXXX_sizecache        int32    `json:\"-\"`\n}\n\nfunc (m *TaskListNotOwnedByHostError) Reset()         { *m = TaskListNotOwnedByHostError{} }\nfunc (m *TaskListNotOwnedByHostError) String() string { return proto.CompactTextString(m) }\nfunc (*TaskListNotOwnedByHostError) ProtoMessage()    {}\nfunc (*TaskListNotOwnedByHostError) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_3688ca0fd170c8f9, []int{6}\n}\nfunc (m *TaskListNotOwnedByHostError) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TaskListNotOwnedByHostError) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TaskListNotOwnedByHostError.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TaskListNotOwnedByHostError) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TaskListNotOwnedByHostError.Merge(m, src)\n}\nfunc (m *TaskListNotOwnedByHostError) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TaskListNotOwnedByHostError) XXX_DiscardUnknown() {\n\txxx_messageInfo_TaskListNotOwnedByHostError.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TaskListNotOwnedByHostError proto.InternalMessageInfo\n\nfunc (m *TaskListNotOwnedByHostError) GetOwnedByIdentity() string {\n\tif m != nil {\n\t\treturn m.OwnedByIdentity\n\t}\n\treturn \"\"\n}\n\nfunc (m *TaskListNotOwnedByHostError) GetMyIdentity() string {\n\tif m != nil {\n\t\treturn m.MyIdentity\n\t}\n\treturn \"\"\n}\n\nfunc (m *TaskListNotOwnedByHostError) GetTaskListName() string {\n\tif m != nil {\n\t\treturn m.TaskListName\n\t}\n\treturn \"\"\n}\n\nfunc init() {\n\tproto.RegisterType((*CurrentBranchChangedError)(nil), \"uber.cadence.shared.v1.CurrentBranchChangedError\")\n\tproto.RegisterType((*InternalDataInconsistencyError)(nil), \"uber.cadence.shared.v1.InternalDataInconsistencyError\")\n\tproto.RegisterType((*EventAlreadyStartedError)(nil), \"uber.cadence.shared.v1.EventAlreadyStartedError\")\n\tproto.RegisterType((*RemoteSyncMatchedError)(nil), \"uber.cadence.shared.v1.RemoteSyncMatchedError\")\n\tproto.RegisterType((*RetryTaskV2Error)(nil), \"uber.cadence.shared.v1.RetryTaskV2Error\")\n\tproto.RegisterType((*ShardOwnershipLostError)(nil), \"uber.cadence.shared.v1.ShardOwnershipLostError\")\n\tproto.RegisterType((*TaskListNotOwnedByHostError)(nil), \"uber.cadence.shared.v1.TaskListNotOwnedByHostError\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/shared/v1/error.proto\", fileDescriptor_3688ca0fd170c8f9)\n}\n\nvar fileDescriptor_3688ca0fd170c8f9 = []byte{\n\t// 514 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x94, 0x93, 0xcd, 0x6e, 0xd3, 0x40,\n\t0x14, 0x85, 0xe5, 0xf2, 0xa3, 0x66, 0x5a, 0x01, 0xb5, 0xaa, 0x12, 0x5a, 0x29, 0x44, 0x06, 0xa1,\n\t0xc2, 0xc2, 0x26, 0x45, 0xac, 0x58, 0x91, 0x10, 0xd4, 0xa0, 0x16, 0x24, 0xa7, 0x14, 0x89, 0x8d,\n\t0x35, 0x99, 0xb9, 0xc4, 0xa3, 0xc4, 0x77, 0xa2, 0x99, 0x9b, 0x04, 0xbf, 0x05, 0x6b, 0x9e, 0x88,\n\t0x25, 0x8f, 0x80, 0xf2, 0x24, 0x68, 0x3c, 0x0e, 0x6d, 0xba, 0x63, 0x67, 0xdf, 0xfb, 0x9d, 0xe3,\n\t0xe3, 0xa3, 0x19, 0x16, 0xcd, 0x47, 0x60, 0x12, 0xc1, 0x25, 0xa0, 0x80, 0xc4, 0xe6, 0xdc, 0x80,\n\t0x4c, 0x16, 0x9d, 0x04, 0x8c, 0xd1, 0x26, 0x9e, 0x19, 0x4d, 0x3a, 0x3c, 0x70, 0x4c, 0x5c, 0x33,\n\t0xb1, 0x67, 0xe2, 0x45, 0xe7, 0xb0, 0xbd, 0xa1, 0xe5, 0x33, 0xe5, 0x84, 0x42, 0x17, 0x85, 0x46,\n\t0xaf, 0x3c, 0x7c, 0xb2, 0x49, 0xc8, 0x42, 0xa1, 0x63, 0x72, 0x65, 0x49, 0x9b, 0xd2, 0x43, 0xd1,\n\t0x39, 0x7b, 0xd4, 0x9b, 0x1b, 0x03, 0x48, 0x5d, 0xc3, 0x51, 0xe4, 0xbd, 0x9c, 0xe3, 0x18, 0x64,\n\t0xdf, 0x25, 0x08, 0x5f, 0xb2, 0x7d, 0xe1, 0x97, 0xd9, 0xa8, 0xda, 0x66, 0xa4, 0x27, 0x80, 0xcd,\n\t0xa0, 0x1d, 0x1c, 0xef, 0xa6, 0xa1, 0xb8, 0x2e, 0xbc, 0x70, 0x9b, 0xa8, 0xcd, 0x5a, 0x03, 0x24,\n\t0x30, 0xc8, 0xa7, 0xef, 0x38, 0xf1, 0x01, 0x0a, 0x8d, 0x56, 0x59, 0x02, 0x14, 0x65, 0xe5, 0x19,\n\t0x1d, 0xb2, 0x66, 0x7f, 0x01, 0x48, 0x6f, 0xa7, 0x06, 0xb8, 0x2c, 0x87, 0xc4, 0x0d, 0xd5, 0xdf,\n\t0x8b, 0x9a, 0xec, 0x20, 0x85, 0x42, 0x13, 0x0c, 0x4b, 0x14, 0xe7, 0x9c, 0x44, 0xbe, 0xde, 0xfc,\n\t0xdc, 0x62, 0x0f, 0x52, 0x20, 0x53, 0x5e, 0x70, 0x3b, 0xb9, 0x3c, 0xf1, 0xf1, 0x8e, 0x58, 0x43,\n\t0xea, 0x82, 0x2b, 0xcc, 0x94, 0xac, 0x32, 0x35, 0xd2, 0x6d, 0x3f, 0x18, 0xc8, 0xf0, 0x33, 0x0b,\n\t0x97, 0xda, 0x4c, 0xbe, 0x4d, 0xf5, 0x32, 0x83, 0xef, 0x20, 0xe6, 0xa4, 0x34, 0x36, 0xb7, 0xda,\n\t0xc1, 0xf1, 0xce, 0xc9, 0xb3, 0x78, 0xa3, 0x54, 0x3e, 0x53, 0xf1, 0xa2, 0x13, 0x7f, 0xa9, 0xf1,\n\t0xfe, 0x9a, 0x4e, 0xf7, 0x96, 0x37, 0x47, 0xe1, 0x07, 0xb6, 0x63, 0x5d, 0xe4, 0x0c, 0xdc, 0x4f,\n\t0x34, 0x6f, 0x55, 0x7e, 0xcf, 0x6f, 0xf8, 0xb9, 0xaa, 0x9d, 0xe3, 0x25, 0x18, 0xab, 0x34, 0x9e,\n\t0xfa, 0xc6, 0x07, 0x04, 0x45, 0xca, 0x2a, 0x75, 0xd5, 0x40, 0xf8, 0x9e, 0x35, 0x00, 0x65, 0xed,\n\t0x74, 0xfb, 0x7f, 0x9d, 0xb6, 0x01, 0x65, 0xe5, 0x13, 0x25, 0xec, 0xe1, 0x30, 0xe7, 0x46, 0x7e,\n\t0x5a, 0x22, 0x18, 0x9b, 0xab, 0xd9, 0x99, 0xb6, 0xe4, 0x2b, 0xda, 0x67, 0x77, 0xb4, 0x9b, 0xd6,\n\t0xf5, 0xf8, 0x97, 0xe8, 0x47, 0xc0, 0x8e, 0x5c, 0x91, 0x67, 0xca, 0xd2, 0x47, 0x4d, 0x4e, 0x27,\n\t0xbb, 0xe5, 0xe9, 0x3f, 0xd5, 0x0b, 0xb6, 0xe7, 0x40, 0x99, 0x8d, 0xca, 0x4c, 0x49, 0x40, 0x52,\n\t0x54, 0xd6, 0x0e, 0xf7, 0xb5, 0x87, 0x07, 0xf5, 0x38, 0x7c, 0xcc, 0x76, 0x8a, 0x6b, 0xd4, 0x56,\n\t0x45, 0xb1, 0xe2, 0x0a, 0x78, 0xca, 0xee, 0x11, 0xb7, 0x93, 0x6c, 0xaa, 0x2c, 0x65, 0xc8, 0x0b,\n\t0xa8, 0x4a, 0x6b, 0xa4, 0xbb, 0xb4, 0x4e, 0xc0, 0x0b, 0xe8, 0xf6, 0x7e, 0xad, 0x5a, 0xc1, 0xef,\n\t0x55, 0x2b, 0xf8, 0xb3, 0x6a, 0x05, 0x5f, 0x5f, 0x8f, 0x15, 0xe5, 0xf3, 0x51, 0x2c, 0x74, 0x91,\n\t0x6c, 0x9c, 0xe2, 0x78, 0x0c, 0x98, 0x54, 0x27, 0xf7, 0xea, 0xba, 0xbc, 0xf1, 0x4f, 0x8b, 0xce,\n\t0xe8, 0x6e, 0xb5, 0x79, 0xf5, 0x37, 0x00, 0x00, 0xff, 0xff, 0x3d, 0x40, 0x81, 0x71, 0x58, 0x03,\n\t0x00, 0x00,\n}\n\nfunc (m *CurrentBranchChangedError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *CurrentBranchChangedError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *CurrentBranchChangedError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.CurrentBranchToken) > 0 {\n\t\ti -= len(m.CurrentBranchToken)\n\t\tcopy(dAtA[i:], m.CurrentBranchToken)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.CurrentBranchToken)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *InternalDataInconsistencyError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *InternalDataInconsistencyError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *InternalDataInconsistencyError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *EventAlreadyStartedError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *EventAlreadyStartedError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *EventAlreadyStartedError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RemoteSyncMatchedError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RemoteSyncMatchedError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RemoteSyncMatchedError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *RetryTaskV2Error) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *RetryTaskV2Error) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *RetryTaskV2Error) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.EndEvent != nil {\n\t\t{\n\t\t\tsize, err := m.EndEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintError(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x22\n\t}\n\tif m.StartEvent != nil {\n\t\t{\n\t\t\tsize, err := m.StartEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintError(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif m.WorkflowExecution != nil {\n\t\t{\n\t\t\tsize, err := m.WorkflowExecution.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintError(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.DomainId) > 0 {\n\t\ti -= len(m.DomainId)\n\t\tcopy(dAtA[i:], m.DomainId)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.DomainId)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *ShardOwnershipLostError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *ShardOwnershipLostError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *ShardOwnershipLostError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Owner) > 0 {\n\t\ti -= len(m.Owner)\n\t\tcopy(dAtA[i:], m.Owner)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.Owner)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *TaskListNotOwnedByHostError) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TaskListNotOwnedByHostError) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TaskListNotOwnedByHostError) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.TaskListName) > 0 {\n\t\ti -= len(m.TaskListName)\n\t\tcopy(dAtA[i:], m.TaskListName)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.TaskListName)))\n\t\ti--\n\t\tdAtA[i] = 0x1a\n\t}\n\tif len(m.MyIdentity) > 0 {\n\t\ti -= len(m.MyIdentity)\n\t\tcopy(dAtA[i:], m.MyIdentity)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.MyIdentity)))\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif len(m.OwnedByIdentity) > 0 {\n\t\ti -= len(m.OwnedByIdentity)\n\t\tcopy(dAtA[i:], m.OwnedByIdentity)\n\t\ti = encodeVarintError(dAtA, i, uint64(len(m.OwnedByIdentity)))\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintError(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovError(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *CurrentBranchChangedError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.CurrentBranchToken)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *InternalDataInconsistencyError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *EventAlreadyStartedError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RemoteSyncMatchedError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *RetryTaskV2Error) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.DomainId)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.WorkflowExecution != nil {\n\t\tl = m.WorkflowExecution.Size()\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.StartEvent != nil {\n\t\tl = m.StartEvent.Size()\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.EndEvent != nil {\n\t\tl = m.EndEvent.Size()\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *ShardOwnershipLostError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.Owner)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *TaskListNotOwnedByHostError) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tl = len(m.OwnedByIdentity)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tl = len(m.MyIdentity)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tl = len(m.TaskListName)\n\tif l > 0 {\n\t\tn += 1 + l + sovError(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovError(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozError(x uint64) (n int) {\n\treturn sovError(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *CurrentBranchChangedError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: CurrentBranchChangedError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: CurrentBranchChangedError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentBranchToken\", wireType)\n\t\t\t}\n\t\t\tvar byteLen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tbyteLen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif byteLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + byteLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.CurrentBranchToken = append(m.CurrentBranchToken[:0], dAtA[iNdEx:postIndex]...)\n\t\t\tif m.CurrentBranchToken == nil {\n\t\t\t\tm.CurrentBranchToken = []byte{}\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *InternalDataInconsistencyError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: InternalDataInconsistencyError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: InternalDataInconsistencyError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *EventAlreadyStartedError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: EventAlreadyStartedError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: EventAlreadyStartedError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RemoteSyncMatchedError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RemoteSyncMatchedError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RemoteSyncMatchedError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *RetryTaskV2Error) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: RetryTaskV2Error: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: RetryTaskV2Error: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field DomainId\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.DomainId = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field WorkflowExecution\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.WorkflowExecution == nil {\n\t\t\t\tm.WorkflowExecution = &v1.WorkflowExecution{}\n\t\t\t}\n\t\t\tif err := m.WorkflowExecution.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartEvent == nil {\n\t\t\t\tm.StartEvent = &v11.VersionHistoryItem{}\n\t\t\t}\n\t\t\tif err := m.StartEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 4:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field EndEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.EndEvent == nil {\n\t\t\t\tm.EndEvent = &v11.VersionHistoryItem{}\n\t\t\t}\n\t\t\tif err := m.EndEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *ShardOwnershipLostError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: ShardOwnershipLostError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: ShardOwnershipLostError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Owner\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Owner = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *TaskListNotOwnedByHostError) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListNotOwnedByHostError: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TaskListNotOwnedByHostError: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field OwnedByIdentity\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.OwnedByIdentity = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field MyIdentity\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.MyIdentity = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tcase 3:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field TaskListName\", wireType)\n\t\t\t}\n\t\t\tvar stringLen uint64\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tstringLen |= uint64(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tintStringLen := int(stringLen)\n\t\t\tif intStringLen < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tpostIndex := iNdEx + intStringLen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.TaskListName = string(dAtA[iNdEx:postIndex])\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipError(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthError\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipError(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowError\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowError\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthError\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupError\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthError\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthError        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowError          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupError = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/shared/v1/error.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/shared/v1/error.proto\n\npackage sharedv1\n\nvar yarpcFileDescriptorClosure3688ca0fd170c8f9 = [][]byte{\n\t// uber/cadence/shared/v1/error.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x94, 0x93, 0x5f, 0x6f, 0xd3, 0x30,\n\t\t0x14, 0xc5, 0xd5, 0xf2, 0x47, 0xab, 0x3b, 0x01, 0x8b, 0xa6, 0x51, 0x3a, 0x09, 0xaa, 0x80, 0xd0,\n\t\t0xe0, 0x21, 0xa1, 0x43, 0x88, 0x07, 0x9e, 0xe8, 0x28, 0x5a, 0xd1, 0x06, 0x52, 0x3a, 0x86, 0xc4,\n\t\t0x4b, 0xe4, 0xda, 0x97, 0xc6, 0x6a, 0x73, 0x5d, 0xd9, 0xb7, 0x2d, 0xf9, 0x16, 0x3c, 0xf3, 0x69,\n\t\t0x91, 0xed, 0x94, 0xad, 0x7b, 0xdb, 0x5b, 0x72, 0xef, 0xef, 0x9c, 0x9c, 0x1c, 0xd9, 0x2c, 0x5e,\n\t\t0x4e, 0xc0, 0xa4, 0x82, 0x4b, 0x40, 0x01, 0xa9, 0x2d, 0xb8, 0x01, 0x99, 0xae, 0xfa, 0x29, 0x18,\n\t\t0xa3, 0x4d, 0xb2, 0x30, 0x9a, 0x74, 0x74, 0xe0, 0x98, 0xa4, 0x66, 0x92, 0xc0, 0x24, 0xab, 0x7e,\n\t\t0xb7, 0xb7, 0xa5, 0xe5, 0x0b, 0xe5, 0x84, 0x42, 0x97, 0xa5, 0xc6, 0xa0, 0xec, 0x3e, 0xdf, 0x26,\n\t\t0x64, 0xa9, 0xd0, 0x31, 0x85, 0xb2, 0xa4, 0x4d, 0x15, 0xa0, 0xf8, 0x9c, 0x3d, 0x39, 0x59, 0x1a,\n\t\t0x03, 0x48, 0x03, 0xc3, 0x51, 0x14, 0x27, 0x05, 0xc7, 0x29, 0xc8, 0xa1, 0x4b, 0x10, 0xbd, 0x61,\n\t\t0xfb, 0x22, 0x2c, 0xf3, 0x89, 0xdf, 0xe6, 0xa4, 0x67, 0x80, 0x9d, 0x46, 0xaf, 0x71, 0xb4, 0x9b,\n\t\t0x45, 0xe2, 0xba, 0xf0, 0xc2, 0x6d, 0xe2, 0x1e, 0x7b, 0x3a, 0x42, 0x02, 0x83, 0x7c, 0xfe, 0x89,\n\t\t0x13, 0x1f, 0xa1, 0xd0, 0x68, 0x95, 0x25, 0x40, 0x51, 0x79, 0xcf, 0xb8, 0xcb, 0x3a, 0xc3, 0x15,\n\t\t0x20, 0x7d, 0x9c, 0x1b, 0xe0, 0xb2, 0x1a, 0x13, 0x37, 0x54, 0x7f, 0x2f, 0xee, 0xb0, 0x83, 0x0c,\n\t\t0x4a, 0x4d, 0x30, 0xae, 0x50, 0x9c, 0x73, 0x12, 0xc5, 0x66, 0xf3, 0xb7, 0xc9, 0x1e, 0x65, 0x40,\n\t\t0xa6, 0xba, 0xe0, 0x76, 0x76, 0x79, 0x1c, 0xe2, 0x1d, 0xb2, 0x96, 0xd4, 0x25, 0x57, 0x98, 0x2b,\n\t\t0xe9, 0x33, 0xb5, 0xb2, 0x9d, 0x30, 0x18, 0xc9, 0xe8, 0x3b, 0x8b, 0xd6, 0xda, 0xcc, 0x7e, 0xcd,\n\t\t0xf5, 0x3a, 0x87, 0xdf, 0x20, 0x96, 0xa4, 0x34, 0x76, 0x9a, 0xbd, 0xc6, 0x51, 0xfb, 0xf8, 0x65,\n\t\t0xb2, 0x55, 0x2a, 0x5f, 0xa8, 0x64, 0xd5, 0x4f, 0x7e, 0xd4, 0xf8, 0x70, 0x43, 0x67, 0x7b, 0xeb,\n\t\t0x9b, 0xa3, 0xe8, 0x0b, 0x6b, 0x5b, 0x17, 0x39, 0x07, 0xf7, 0x13, 0x9d, 0x3b, 0xde, 0xef, 0xd5,\n\t\t0x0d, 0x3f, 0x57, 0xb5, 0x73, 0xbc, 0x04, 0x63, 0x95, 0xc6, 0xd3, 0xd0, 0xf8, 0x88, 0xa0, 0xcc,\n\t\t0x98, 0x57, 0xfb, 0x06, 0xa2, 0xcf, 0xac, 0x05, 0x28, 0x6b, 0xa7, 0xbb, 0xb7, 0x75, 0xda, 0x01,\n\t\t0x94, 0xde, 0x27, 0x4e, 0xd9, 0xe3, 0x71, 0xc1, 0x8d, 0xfc, 0xb6, 0x46, 0x30, 0xb6, 0x50, 0x8b,\n\t\t0x33, 0x6d, 0x29, 0x54, 0xb4, 0xcf, 0xee, 0x69, 0x37, 0xad, 0xeb, 0x09, 0x2f, 0xf1, 0x9f, 0x06,\n\t\t0x3b, 0x74, 0x45, 0x9e, 0x29, 0x4b, 0x5f, 0x35, 0x39, 0x9d, 0x1c, 0x54, 0xa7, 0xff, 0x55, 0xaf,\n\t\t0xd9, 0x9e, 0x03, 0x65, 0x3e, 0xa9, 0x72, 0x25, 0x01, 0x49, 0x51, 0x55, 0x3b, 0x3c, 0xd4, 0x01,\n\t\t0x1e, 0xd5, 0xe3, 0xe8, 0x19, 0x6b, 0x97, 0xd7, 0xa8, 0xa6, 0xa7, 0x58, 0x79, 0x05, 0xbc, 0x60,\n\t\t0x0f, 0x88, 0xdb, 0x59, 0x3e, 0x57, 0x96, 0x72, 0xe4, 0x25, 0xf8, 0xd2, 0x5a, 0xd9, 0x2e, 0x6d,\n\t\t0x12, 0xf0, 0x12, 0x06, 0xef, 0x7f, 0xbe, 0x9b, 0x2a, 0x2a, 0x96, 0x93, 0x44, 0xe8, 0x32, 0xdd,\n\t\t0x3a, 0xb9, 0xc9, 0x14, 0x30, 0xf5, 0xa7, 0xf5, 0xea, 0x8a, 0x7c, 0x08, 0x4f, 0xab, 0xfe, 0xe4,\n\t\t0xbe, 0xdf, 0xbc, 0xfd, 0x17, 0x00, 0x00, 0xff, 0xff, 0x28, 0x15, 0x25, 0x92, 0x4c, 0x03, 0x00,\n\t\t0x00,\n\t},\n\t// uber/cadence/api/v1/common.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x57, 0xdd, 0x72, 0x22, 0xc7,\n\t\t0x15, 0xf6, 0xc0, 0xa2, 0x9f, 0x03, 0xbb, 0x42, 0xad, 0xfd, 0x61, 0xb5, 0x5e, 0xaf, 0x16, 0x97,\n\t\t0x63, 0x79, 0x2b, 0x86, 0x88, 0x4d, 0x52, 0x2e, 0x3b, 0x4e, 0x82, 0xd0, 0x48, 0x9a, 0x5d, 0x02,\n\t\t0xa4, 0x99, 0x95, 0xac, 0xa4, 0xca, 0x53, 0xcd, 0x4c, 0x83, 0x3b, 0x0c, 0xd3, 0x93, 0x99, 0x1e,\n\t\t0x56, 0xf8, 0x22, 0x95, 0xcb, 0xe4, 0x26, 0x8f, 0x90, 0x8b, 0xbc, 0x48, 0x1e, 0x20, 0x97, 0x79,\n\t\t0x97, 0x5c, 0xa7, 0xba, 0xa7, 0x07, 0x81, 0xc2, 0x1a, 0x5f, 0xa4, 0x7c, 0x47, 0x9f, 0xf3, 0x7d,\n\t\t0xa7, 0xbf, 0xd3, 0xdd, 0xe7, 0x1c, 0x06, 0x0e, 0x92, 0x01, 0x8d, 0xea, 0x2e, 0xf1, 0x68, 0xe0,\n\t\t0xd2, 0x3a, 0x09, 0x59, 0x7d, 0x7a, 0x54, 0x77, 0xf9, 0x64, 0xc2, 0x83, 0x5a, 0x18, 0x71, 0xc1,\n\t\t0xd1, 0x9e, 0x44, 0xd4, 0x34, 0xa2, 0x46, 0x42, 0x56, 0x9b, 0x1e, 0xed, 0x7f, 0x30, 0xe2, 0x7c,\n\t\t0xe4, 0xd3, 0xba, 0x82, 0x0c, 0x92, 0x61, 0xdd, 0x4b, 0x22, 0x22, 0x58, 0x46, 0xaa, 0xbe, 0x86,\n\t\t0xdd, 0x4b, 0x1e, 0x8d, 0x87, 0x3e, 0x7f, 0x6b, 0x5e, 0x53, 0x37, 0x91, 0x2e, 0xf4, 0x0c, 0x8a,\n\t\t0x6f, 0xb5, 0xd1, 0x61, 0x5e, 0xc5, 0x38, 0x30, 0x0e, 0xb7, 0x31, 0x64, 0x26, 0xcb, 0x43, 0x0f,\n\t\t0x60, 0x23, 0x4a, 0x02, 0xe9, 0xcb, 0x29, 0x5f, 0x21, 0x4a, 0x02, 0xcb, 0xab, 0x56, 0xa1, 0x94,\n\t\t0x05, 0xb3, 0x67, 0x21, 0x45, 0x08, 0xee, 0x04, 0x64, 0x42, 0x75, 0x00, 0xf5, 0x5b, 0x62, 0x9a,\n\t\t0xae, 0x60, 0x53, 0x26, 0x66, 0xef, 0xc4, 0x3c, 0x85, 0xcd, 0x1e, 0x99, 0xf9, 0x9c, 0x78, 0xd2,\n\t\t0xed, 0x11, 0x41, 0x94, 0xbb, 0x84, 0xd5, 0xef, 0xea, 0x17, 0xb0, 0x79, 0x4a, 0x98, 0x9f, 0x44,\n\t\t0x14, 0x3d, 0x84, 0x8d, 0x88, 0x92, 0x98, 0x07, 0x9a, 0xaf, 0x57, 0xa8, 0x02, 0x9b, 0x1e, 0x15,\n\t\t0x84, 0xf9, 0xb1, 0x52, 0x58, 0xc2, 0xd9, 0xb2, 0xfa, 0x77, 0x03, 0xee, 0xfc, 0x86, 0x4e, 0x38,\n\t\t0xfa, 0x12, 0x36, 0x86, 0x8c, 0xfa, 0x5e, 0x5c, 0x31, 0x0e, 0xf2, 0x87, 0xc5, 0xc6, 0x47, 0xb5,\n\t\t0x15, 0xe7, 0x57, 0x93, 0xd0, 0xda, 0xa9, 0xc2, 0x99, 0x81, 0x88, 0x66, 0x58, 0x93, 0xf6, 0x2f,\n\t\t0xa1, 0xb8, 0x60, 0x46, 0x65, 0xc8, 0x8f, 0xe9, 0x4c, 0xab, 0x90, 0x3f, 0x51, 0x03, 0x0a, 0x53,\n\t\t0xe2, 0x27, 0x54, 0x09, 0x28, 0x36, 0xde, 0x5f, 0x19, 0x5e, 0xa7, 0x89, 0x53, 0xe8, 0xe7, 0xb9,\n\t\t0xcf, 0x8c, 0xea, 0x3f, 0x0c, 0xd8, 0x38, 0xa7, 0xc4, 0xa3, 0x11, 0xfa, 0xd5, 0x2d, 0x89, 0x1f,\n\t\t0xaf, 0x8c, 0x91, 0x82, 0x7f, 0x58, 0x91, 0xff, 0x36, 0xa0, 0xdc, 0xa7, 0x24, 0x72, 0xbf, 0x69,\n\t\t0x0a, 0x11, 0xb1, 0x41, 0x22, 0x68, 0x8c, 0x1c, 0xb8, 0xc7, 0x02, 0x8f, 0x5e, 0x53, 0xcf, 0x59,\n\t\t0x92, 0xfd, 0xd9, 0xca, 0xa8, 0xb7, 0xe9, 0x35, 0x2b, 0xe5, 0x2e, 0xe6, 0x71, 0x97, 0x2d, 0xda,\n\t\t0xf6, 0xbf, 0x06, 0xf4, 0xbf, 0xa0, 0xff, 0x63, 0x56, 0x43, 0xd8, 0x3a, 0x21, 0x82, 0x1c, 0xfb,\n\t\t0x7c, 0x80, 0x4e, 0xe1, 0x2e, 0x0d, 0x5c, 0xee, 0xb1, 0x60, 0xe4, 0x88, 0x59, 0x98, 0x3e, 0xd0,\n\t\t0x7b, 0x8d, 0xe7, 0x2b, 0x63, 0x99, 0x1a, 0x29, 0x5f, 0x34, 0x2e, 0xd1, 0x85, 0xd5, 0xfc, 0x01,\n\t\t0xe7, 0x16, 0x1e, 0x70, 0x2f, 0x2d, 0x3a, 0x1a, 0x5d, 0xd0, 0x28, 0x66, 0x3c, 0xb0, 0x82, 0x21,\n\t\t0x97, 0x40, 0x36, 0x09, 0xfd, 0xac, 0x10, 0xe4, 0x6f, 0xf4, 0x31, 0xec, 0x0c, 0x29, 0x11, 0x49,\n\t\t0x44, 0x9d, 0x69, 0x0a, 0xd5, 0x05, 0x77, 0x4f, 0x9b, 0x75, 0x80, 0xea, 0x6b, 0x78, 0xd4, 0x4f,\n\t\t0xc2, 0x90, 0x47, 0x82, 0x7a, 0x2d, 0x9f, 0xd1, 0x40, 0x68, 0x4f, 0x2c, 0x6b, 0x75, 0xc4, 0x9d,\n\t\t0xd8, 0x1b, 0xeb, 0xc8, 0x85, 0x11, 0xef, 0x7b, 0x63, 0xf4, 0x18, 0xb6, 0xfe, 0x40, 0xa6, 0x44,\n\t\t0x39, 0xd2, 0x98, 0x9b, 0x72, 0xdd, 0xf7, 0xc6, 0xd5, 0x3f, 0xe7, 0xa1, 0x88, 0xa9, 0x88, 0x66,\n\t\t0x3d, 0xee, 0x33, 0x77, 0x86, 0x4e, 0xa0, 0xcc, 0x02, 0x26, 0x18, 0xf1, 0x1d, 0x16, 0x08, 0x1a,\n\t\t0x4d, 0x49, 0xaa, 0xb2, 0xd8, 0x78, 0x5c, 0x4b, 0xdb, 0x4b, 0x2d, 0x6b, 0x2f, 0xb5, 0x13, 0xdd,\n\t\t0x5e, 0xf0, 0x8e, 0xa6, 0x58, 0x9a, 0x81, 0xea, 0xb0, 0x37, 0x20, 0xee, 0x98, 0x0f, 0x87, 0x8e,\n\t\t0xcb, 0xe9, 0x70, 0xc8, 0x5c, 0x29, 0x53, 0xed, 0x6d, 0x60, 0xa4, 0x5d, 0xad, 0x1b, 0x8f, 0xdc,\n\t\t0x76, 0x42, 0xae, 0xd9, 0x24, 0x99, 0xdc, 0x6c, 0x9b, 0x5f, 0xbb, 0xad, 0xa6, 0xcc, 0xb7, 0xfd,\n\t\t0xe4, 0x26, 0x0a, 0x11, 0x82, 0x4e, 0x42, 0x11, 0x57, 0xee, 0x1c, 0x18, 0x87, 0x85, 0x39, 0xb4,\n\t\t0xa9, 0xcd, 0xe8, 0x4b, 0x78, 0x12, 0xf0, 0xc0, 0x89, 0x64, 0xea, 0x64, 0xe0, 0x53, 0x87, 0x46,\n\t\t0x11, 0x8f, 0x9c, 0xb4, 0xa5, 0xc4, 0x95, 0xc2, 0x41, 0xfe, 0x70, 0x1b, 0x57, 0x02, 0x1e, 0xe0,\n\t\t0x0c, 0x61, 0x4a, 0x00, 0x4e, 0xfd, 0xe8, 0x15, 0xec, 0xd1, 0xeb, 0x90, 0xa5, 0x42, 0x6e, 0x24,\n\t\t0x6f, 0xac, 0x93, 0x8c, 0x6e, 0x58, 0x99, 0xea, 0xea, 0x04, 0x1e, 0x59, 0x31, 0xf7, 0x95, 0xf1,\n\t\t0x2c, 0xe2, 0x49, 0xd8, 0x23, 0x91, 0x60, 0xaa, 0x39, 0xaf, 0x68, 0x98, 0xe8, 0x97, 0x50, 0x88,\n\t\t0x05, 0x11, 0xe9, 0x83, 0xbf, 0xd7, 0x38, 0x5c, 0xf9, 0x48, 0x97, 0x03, 0xf6, 0x25, 0x1e, 0xa7,\n\t\t0xb4, 0xea, 0x14, 0x9e, 0x2c, 0x7b, 0x5b, 0x3c, 0x18, 0xb2, 0x91, 0x56, 0x88, 0x2e, 0xa1, 0xcc,\n\t\t0x32, 0xb7, 0x33, 0x92, 0xfe, 0xac, 0xb4, 0x7f, 0xfc, 0x3d, 0x76, 0x9a, 0x4b, 0xc7, 0x3b, 0x6c,\n\t\t0xc9, 0x11, 0x57, 0xff, 0x65, 0xc0, 0x7e, 0x33, 0x9e, 0x05, 0x6e, 0x36, 0x36, 0x96, 0xf7, 0xad,\n\t\t0xc0, 0x26, 0x0d, 0xe4, 0x39, 0xa7, 0x33, 0x68, 0x0b, 0x67, 0x4b, 0xd4, 0x80, 0x07, 0x61, 0x44,\n\t\t0x3d, 0x3a, 0x64, 0x01, 0xf5, 0x9c, 0x3f, 0x26, 0x34, 0xa1, 0x8e, 0x3a, 0x95, 0xf4, 0x29, 0xef,\n\t\t0xdd, 0x38, 0x7f, 0x2b, 0x7d, 0x1d, 0x79, 0x48, 0x4f, 0x01, 0x52, 0xa0, 0x2a, 0xe7, 0xbc, 0x02,\n\t\t0x6e, 0x2b, 0x8b, 0x2a, 0xd4, 0x5f, 0x43, 0x29, 0x75, 0xbb, 0x4a, 0x83, 0x7a, 0x24, 0xc5, 0xc6,\n\t\t0xd3, 0x95, 0x09, 0x66, 0x5d, 0x02, 0x17, 0x15, 0x25, 0x55, 0x5d, 0xfd, 0x4f, 0x1e, 0xde, 0x57,\n\t\t0xb3, 0x8d, 0xb6, 0xfc, 0x24, 0x16, 0x34, 0xea, 0x53, 0x9f, 0xba, 0x32, 0x13, 0x5d, 0x48, 0x7d,\n\t\t0xd8, 0x8a, 0x45, 0x44, 0x04, 0x1d, 0xcd, 0x74, 0x3b, 0x79, 0xb9, 0x32, 0xfc, 0xea, 0x20, 0x7d,\n\t\t0x4d, 0x3d, 0xce, 0x55, 0x0c, 0x3c, 0x0f, 0x84, 0xfe, 0x62, 0xc0, 0x87, 0x44, 0x11, 0x1c, 0x37,\n\t\t0x65, 0x38, 0xb1, 0x60, 0xee, 0x78, 0xe6, 0x44, 0x74, 0x24, 0x2f, 0x4c, 0xe7, 0x93, 0xf6, 0xc2,\n\t\t0x9f, 0x7e, 0x8f, 0x0d, 0x15, 0x1b, 0x2b, 0x72, 0x9a, 0x99, 0xdc, 0xf1, 0xfc, 0x3d, 0xfc, 0x8c,\n\t\t0x7c, 0x37, 0x0c, 0xfd, 0xcd, 0x80, 0x8f, 0x6e, 0x49, 0xa1, 0xd7, 0x82, 0x46, 0x01, 0xf1, 0x1d,\n\t\t0x1a, 0x08, 0x26, 0x66, 0x99, 0x98, 0xb4, 0x8e, 0x7f, 0xbe, 0x5e, 0x8c, 0xa9, 0xf9, 0xa6, 0xa2,\n\t\t0x2f, 0xc9, 0x79, 0x4e, 0xd6, 0x01, 0x11, 0x86, 0xdd, 0x4c, 0x08, 0xc9, 0x06, 0x8d, 0xbe, 0xd8,\n\t\t0xd5, 0xe3, 0x5e, 0x07, 0x9b, 0x4f, 0x25, 0x5c, 0x76, 0x6f, 0x59, 0x8e, 0x77, 0x61, 0x27, 0x3b,\n\t\t0x7b, 0x9d, 0x4d, 0xf5, 0x17, 0x50, 0xbe, 0x4d, 0x44, 0xf7, 0xa1, 0x10, 0xbb, 0x3c, 0xcc, 0xea,\n\t\t0x34, 0x5d, 0xcc, 0x8b, 0x37, 0xb7, 0xf0, 0x6f, 0xe7, 0x15, 0x3c, 0x5b, 0x73, 0xfe, 0xe8, 0x43,\n\t\t0xb8, 0xbb, 0x74, 0xa7, 0x3a, 0x68, 0x29, 0x5e, 0x80, 0x7e, 0x9e, 0xab, 0x18, 0xd5, 0xbf, 0x1a,\n\t\t0xf0, 0x7c, 0xed, 0xf9, 0xa1, 0x9f, 0xc0, 0xfd, 0xdb, 0xf7, 0x32, 0x1f, 0x71, 0xdb, 0xb2, 0x1f,\n\t\t0x2d, 0x72, 0x54, 0x71, 0xd4, 0x64, 0x6f, 0x5b, 0x66, 0xc8, 0x99, 0x9b, 0xa6, 0xb1, 0xbb, 0x4c,\n\t\t0x78, 0x4d, 0x67, 0x4a, 0xcb, 0x57, 0xb0, 0xdb, 0x23, 0x23, 0x16, 0xa8, 0x5a, 0xee, 0x86, 0x42,\n\t\t0x4d, 0xa3, 0x27, 0xb0, 0x1d, 0x92, 0x11, 0x75, 0x62, 0xf6, 0x6d, 0xba, 0x5f, 0x01, 0x6f, 0x49,\n\t\t0x43, 0x9f, 0x7d, 0x4b, 0xd1, 0x8f, 0x60, 0x27, 0xa0, 0xd7, 0xc2, 0x51, 0x08, 0xc1, 0xc7, 0x34,\n\t\t0xd0, 0x63, 0xf3, 0xae, 0x34, 0xf7, 0xc8, 0x88, 0xda, 0xd2, 0xf8, 0xe2, 0x2d, 0x94, 0x16, 0x27,\n\t\t0x2e, 0x7a, 0x0c, 0x0f, 0xcc, 0x4e, 0xab, 0x7b, 0x62, 0x75, 0xce, 0x1c, 0xfb, 0xaa, 0x67, 0x3a,\n\t\t0x56, 0xe7, 0xa2, 0xd9, 0xb6, 0x4e, 0xca, 0xef, 0xa1, 0x7d, 0x78, 0xb8, 0xec, 0xb2, 0xcf, 0xb1,\n\t\t0x75, 0x6a, 0xe3, 0xcb, 0xb2, 0x81, 0x1e, 0x02, 0x5a, 0xf6, 0xbd, 0xea, 0x77, 0x3b, 0xe5, 0x1c,\n\t\t0xaa, 0xc0, 0xfd, 0x65, 0x7b, 0x0f, 0x77, 0xed, 0xee, 0xcb, 0x72, 0xfe, 0xc5, 0x9f, 0x60, 0x6f,\n\t\t0x45, 0x17, 0x45, 0xcf, 0xe1, 0xa9, 0xd5, 0xef, 0xb6, 0x9b, 0xb6, 0xd5, 0xed, 0x38, 0x67, 0xb8,\n\t\t0xfb, 0xa6, 0xe7, 0xf4, 0xed, 0xa6, 0xbd, 0xa8, 0xe3, 0x9d, 0x90, 0x73, 0xb3, 0xd9, 0xb6, 0xcf,\n\t\t0xaf, 0xca, 0xc6, 0xbb, 0x21, 0x27, 0xb8, 0x69, 0x75, 0xcc, 0x93, 0x72, 0xee, 0xc5, 0x3f, 0x0d,\n\t\t0xf8, 0xe0, 0xbb, 0x9b, 0x03, 0xfa, 0x14, 0x3e, 0x69, 0xb6, 0x6c, 0xeb, 0xc2, 0x74, 0x5a, 0xed,\n\t\t0x37, 0x7d, 0xdb, 0xc4, 0x4e, 0xdf, 0x6c, 0x9b, 0x2d, 0x15, 0xb4, 0x6f, 0xe3, 0xa6, 0x6d, 0x9e,\n\t\t0x5d, 0x2d, 0xe8, 0x7a, 0x09, 0xf5, 0xf5, 0x70, 0x6c, 0x9e, 0xa5, 0x6b, 0xab, 0xf5, 0x5a, 0x2a,\n\t\t0xfd, 0x19, 0x1c, 0xad, 0x27, 0x99, 0x5f, 0xd9, 0x26, 0xee, 0x34, 0xdb, 0x8e, 0xd9, 0xb1, 0x2d,\n\t\t0xfb, 0xaa, 0x9c, 0xdb, 0xcf, 0x55, 0x8c, 0x17, 0x5f, 0x43, 0x49, 0xfe, 0x77, 0xe7, 0x53, 0x1a,\n\t\t0x65, 0x57, 0x77, 0xda, 0xb4, 0xda, 0xdd, 0x0b, 0x13, 0xdf, 0xbe, 0xba, 0x47, 0xb0, 0xb7, 0xec,\n\t\t0x3a, 0xed, 0xe2, 0x96, 0x59, 0x36, 0xe4, 0x9d, 0x2e, 0x3b, 0xce, 0x70, 0xb3, 0x65, 0x9e, 0xbe,\n\t\t0x69, 0x97, 0x73, 0xc7, 0xbf, 0x87, 0x47, 0x2e, 0x9f, 0xac, 0xaa, 0xed, 0xe3, 0x62, 0x4b, 0x7d,\n\t\t0x2d, 0xf5, 0xe4, 0x00, 0xee, 0x19, 0xbf, 0x3b, 0x1a, 0x31, 0xf1, 0x4d, 0x32, 0xa8, 0xb9, 0x7c,\n\t\t0x52, 0x5f, 0xfc, 0xb6, 0xfa, 0x94, 0x79, 0x7e, 0x7d, 0xc4, 0xd3, 0x2f, 0x26, 0xfd, 0xa1, 0xf5,\n\t\t0x05, 0x09, 0xd9, 0xf4, 0x68, 0xb0, 0xa1, 0x6c, 0x2f, 0xff, 0x1b, 0x00, 0x00, 0xff, 0xff, 0x5e,\n\t\t0xe2, 0xc9, 0x06, 0x8c, 0x0d, 0x00, 0x00,\n\t},\n\t// google/protobuf/duration.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x4f, 0x29, 0x2d, 0x4a,\n\t\t0x2c, 0xc9, 0xcc, 0xcf, 0xd3, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0x56,\n\t\t0x5c, 0x1c, 0x2e, 0x50, 0x25, 0x42, 0x12, 0x5c, 0xec, 0xc5, 0xa9, 0xc9, 0xf9, 0x79, 0x29, 0xc5,\n\t\t0x12, 0x8c, 0x0a, 0x8c, 0x1a, 0xcc, 0x41, 0x30, 0xae, 0x90, 0x08, 0x17, 0x6b, 0x5e, 0x62, 0x5e,\n\t\t0x7e, 0xb1, 0x04, 0x93, 0x02, 0xa3, 0x06, 0x6b, 0x10, 0x84, 0xe3, 0xd4, 0xcc, 0xc8, 0x25, 0x9c,\n\t\t0x9c, 0x9f, 0xab, 0x87, 0x66, 0xa6, 0x13, 0x2f, 0xcc, 0xc4, 0x00, 0x90, 0x48, 0x00, 0x63, 0x94,\n\t\t0x21, 0x54, 0x45, 0x7a, 0x7e, 0x4e, 0x62, 0x5e, 0xba, 0x5e, 0x7e, 0x51, 0x3a, 0xc2, 0x81, 0x25,\n\t\t0x95, 0x05, 0xa9, 0xc5, 0xfa, 0xd9, 0x79, 0xf9, 0xe5, 0x79, 0x70, 0xc7, 0x16, 0x24, 0xfd, 0x60,\n\t\t0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce, 0x1d, 0xa2, 0x39, 0x00, 0xaa,\n\t\t0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4, 0x35, 0x89, 0x0d, 0x6c, 0x94,\n\t\t0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0xef, 0x8a, 0xb4, 0xc3, 0xfb, 0x00, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2e, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x4f, 0x4c, 0xc9, 0xcd, 0xcc, 0xd3, 0x2f,\n\t\t0x33, 0xd4, 0xcf, 0xc8, 0x2c, 0x2e, 0xc9, 0x2f, 0xaa, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17,\n\t\t0x12, 0x05, 0x29, 0xd2, 0x83, 0x2a, 0xd2, 0x03, 0x2b, 0xd2, 0x2b, 0x33, 0x54, 0xf2, 0xe4, 0x12,\n\t\t0x0a, 0x4b, 0x2d, 0x2a, 0xce, 0xcc, 0xcf, 0xf3, 0x80, 0x28, 0xf7, 0x2c, 0x49, 0xcd, 0x15, 0x92,\n\t\t0xe4, 0xe2, 0x48, 0x2d, 0x4b, 0xcd, 0x2b, 0x89, 0xcf, 0x4c, 0x91, 0x60, 0x54, 0x60, 0xd4, 0x60,\n\t\t0x0e, 0x62, 0x07, 0xf3, 0x3d, 0x53, 0x84, 0x24, 0xb8, 0xd8, 0xcb, 0x20, 0x1a, 0x24, 0x98, 0x20,\n\t\t0x32, 0x50, 0xae, 0x52, 0x09, 0x17, 0x1f, 0xaa, 0x51, 0x42, 0x8a, 0x5c, 0x3c, 0x49, 0x45, 0x89,\n\t\t0x79, 0xc9, 0x19, 0xf1, 0x25, 0xf9, 0xd9, 0xa9, 0x79, 0x60, 0xa3, 0x78, 0x82, 0xb8, 0x21, 0x62,\n\t\t0x21, 0x20, 0x21, 0x21, 0x7b, 0x2e, 0xd6, 0xcc, 0x92, 0xd4, 0xdc, 0x62, 0x09, 0x26, 0x05, 0x66,\n\t\t0x0d, 0x6e, 0x23, 0x4d, 0x3d, 0xac, 0xce, 0xd4, 0xc3, 0x74, 0x63, 0x10, 0x44, 0x9f, 0x93, 0x79,\n\t\t0x94, 0x69, 0x7a, 0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x72, 0x48, 0xe8,\n\t\t0x66, 0xa6, 0xe4, 0xe8, 0xa7, 0xe7, 0xeb, 0x83, 0xfd, 0x0f, 0x0f, 0x16, 0x6b, 0x30, 0xa3, 0xcc,\n\t\t0x30, 0x89, 0x0d, 0x2c, 0x6e, 0x0c, 0x08, 0x00, 0x00, 0xff, 0xff, 0x44, 0x14, 0xd7, 0xd4, 0x3e,\n\t\t0x01, 0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/history.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/shared/v1/history.proto\n\npackage sharedv1\n\nimport (\n\tfmt \"fmt\"\n\tio \"io\"\n\tmath \"math\"\n\tmath_bits \"math/bits\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n\tv11 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype TransientDecisionInfo struct {\n\tScheduledEvent       *v1.HistoryEvent `protobuf:\"bytes,1,opt,name=scheduled_event,json=scheduledEvent,proto3\" json:\"scheduled_event,omitempty\"`\n\tStartedEvent         *v1.HistoryEvent `protobuf:\"bytes,2,opt,name=started_event,json=startedEvent,proto3\" json:\"started_event,omitempty\"`\n\tXXX_NoUnkeyedLiteral struct{}         `json:\"-\"`\n\tXXX_unrecognized     []byte           `json:\"-\"`\n\tXXX_sizecache        int32            `json:\"-\"`\n}\n\nfunc (m *TransientDecisionInfo) Reset()         { *m = TransientDecisionInfo{} }\nfunc (m *TransientDecisionInfo) String() string { return proto.CompactTextString(m) }\nfunc (*TransientDecisionInfo) ProtoMessage()    {}\nfunc (*TransientDecisionInfo) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0370c4177fcc3ee8, []int{0}\n}\nfunc (m *TransientDecisionInfo) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *TransientDecisionInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_TransientDecisionInfo.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *TransientDecisionInfo) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_TransientDecisionInfo.Merge(m, src)\n}\nfunc (m *TransientDecisionInfo) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *TransientDecisionInfo) XXX_DiscardUnknown() {\n\txxx_messageInfo_TransientDecisionInfo.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_TransientDecisionInfo proto.InternalMessageInfo\n\nfunc (m *TransientDecisionInfo) GetScheduledEvent() *v1.HistoryEvent {\n\tif m != nil {\n\t\treturn m.ScheduledEvent\n\t}\n\treturn nil\n}\n\nfunc (m *TransientDecisionInfo) GetStartedEvent() *v1.HistoryEvent {\n\tif m != nil {\n\t\treturn m.StartedEvent\n\t}\n\treturn nil\n}\n\n// VersionHistories contains all version histories from all branches.\ntype VersionHistories struct {\n\tCurrentVersionHistoryIndex int32                 `protobuf:\"varint,1,opt,name=current_version_history_index,json=currentVersionHistoryIndex,proto3\" json:\"current_version_history_index,omitempty\"`\n\tHistories                  []*v11.VersionHistory `protobuf:\"bytes,2,rep,name=histories,proto3\" json:\"histories,omitempty\"`\n\tXXX_NoUnkeyedLiteral       struct{}              `json:\"-\"`\n\tXXX_unrecognized           []byte                `json:\"-\"`\n\tXXX_sizecache              int32                 `json:\"-\"`\n}\n\nfunc (m *VersionHistories) Reset()         { *m = VersionHistories{} }\nfunc (m *VersionHistories) String() string { return proto.CompactTextString(m) }\nfunc (*VersionHistories) ProtoMessage()    {}\nfunc (*VersionHistories) Descriptor() ([]byte, []int) {\n\treturn fileDescriptor_0370c4177fcc3ee8, []int{1}\n}\nfunc (m *VersionHistories) XXX_Unmarshal(b []byte) error {\n\treturn m.Unmarshal(b)\n}\nfunc (m *VersionHistories) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {\n\tif deterministic {\n\t\treturn xxx_messageInfo_VersionHistories.Marshal(b, m, deterministic)\n\t} else {\n\t\tb = b[:cap(b)]\n\t\tn, err := m.MarshalToSizedBuffer(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn b[:n], nil\n\t}\n}\nfunc (m *VersionHistories) XXX_Merge(src proto.Message) {\n\txxx_messageInfo_VersionHistories.Merge(m, src)\n}\nfunc (m *VersionHistories) XXX_Size() int {\n\treturn m.Size()\n}\nfunc (m *VersionHistories) XXX_DiscardUnknown() {\n\txxx_messageInfo_VersionHistories.DiscardUnknown(m)\n}\n\nvar xxx_messageInfo_VersionHistories proto.InternalMessageInfo\n\nfunc (m *VersionHistories) GetCurrentVersionHistoryIndex() int32 {\n\tif m != nil {\n\t\treturn m.CurrentVersionHistoryIndex\n\t}\n\treturn 0\n}\n\nfunc (m *VersionHistories) GetHistories() []*v11.VersionHistory {\n\tif m != nil {\n\t\treturn m.Histories\n\t}\n\treturn nil\n}\n\nfunc init() {\n\tproto.RegisterType((*TransientDecisionInfo)(nil), \"uber.cadence.shared.v1.TransientDecisionInfo\")\n\tproto.RegisterType((*VersionHistories)(nil), \"uber.cadence.shared.v1.VersionHistories\")\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/shared/v1/history.proto\", fileDescriptor_0370c4177fcc3ee8)\n}\n\nvar fileDescriptor_0370c4177fcc3ee8 = []byte{\n\t// 314 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0x4f, 0x4a, 0x03, 0x31,\n\t0x14, 0xc6, 0x49, 0x45, 0xc1, 0xd4, 0x7f, 0x0c, 0x28, 0xa5, 0x60, 0x69, 0xab, 0x42, 0x57, 0x19,\n\t0xa6, 0xe2, 0xca, 0x95, 0x56, 0xc5, 0xba, 0x2c, 0xe2, 0xc2, 0xcd, 0x90, 0x26, 0xcf, 0x4e, 0xc0,\n\t0x26, 0x25, 0xc9, 0x04, 0x7b, 0x15, 0xaf, 0xe0, 0x45, 0x5c, 0x7a, 0x04, 0xe9, 0x49, 0x24, 0x33,\n\t0x63, 0x4b, 0xec, 0xc6, 0xdd, 0x84, 0xf7, 0xfb, 0x7d, 0x7c, 0xf3, 0x1e, 0x3e, 0xcd, 0xc7, 0xa0,\n\t0x63, 0x46, 0x39, 0x48, 0x06, 0xb1, 0xc9, 0xa8, 0x06, 0x1e, 0xbb, 0x24, 0xce, 0x84, 0xb1, 0x4a,\n\t0xcf, 0xc9, 0x4c, 0x2b, 0xab, 0xa2, 0x23, 0x4f, 0x91, 0x8a, 0x22, 0x25, 0x45, 0x5c, 0xd2, 0xec,\n\t0x04, 0x36, 0x9d, 0x89, 0x35, 0xb5, 0x79, 0x12, 0x22, 0x7c, 0x2a, 0xe4, 0x1a, 0xd4, 0xfd, 0x40,\n\t0xf8, 0xf0, 0x51, 0x53, 0x69, 0x04, 0x48, 0x7b, 0x03, 0x4c, 0x18, 0xa1, 0xe4, 0x50, 0xbe, 0xa8,\n\t0xe8, 0x01, 0xef, 0x1b, 0x96, 0x01, 0xcf, 0x5f, 0x81, 0xa7, 0xe0, 0x40, 0xda, 0x06, 0x6a, 0xa3,\n\t0x5e, 0xbd, 0xdf, 0x21, 0x41, 0x27, 0x3a, 0x13, 0xc4, 0x25, 0xe4, 0xbe, 0x8c, 0xbd, 0xf5, 0xe0,\n\t0x68, 0x6f, 0x69, 0x16, 0xef, 0xe8, 0x0e, 0xef, 0x1a, 0x4b, 0xb5, 0x5d, 0x26, 0xd5, 0xfe, 0x9b,\n\t0xb4, 0x53, 0x79, 0xc5, 0xab, 0xfb, 0x8e, 0xf0, 0xc1, 0x13, 0x68, 0xdf, 0xb1, 0xa4, 0x04, 0x98,\n\t0xe8, 0x0a, 0x1f, 0xb3, 0x5c, 0x6b, 0x90, 0x36, 0x75, 0xe5, 0x2c, 0xad, 0xfe, 0x31, 0x15, 0x92,\n\t0xc3, 0x5b, 0x51, 0x7b, 0x73, 0xd4, 0xac, 0xa0, 0xc0, 0x9f, 0x0f, 0x3d, 0x11, 0x0d, 0xf0, 0x76,\n\t0xf6, 0x9b, 0xd7, 0xa8, 0xb5, 0x37, 0x7a, 0xf5, 0xfe, 0xd9, 0x9f, 0x6e, 0x7e, 0x7d, 0xbe, 0x5d,\n\t0xa8, 0x8f, 0x56, 0xde, 0xf5, 0xe0, 0x73, 0xd1, 0x42, 0x5f, 0x8b, 0x16, 0xfa, 0x5e, 0xb4, 0xd0,\n\t0xf3, 0xc5, 0x44, 0xd8, 0x2c, 0x1f, 0x13, 0xa6, 0xa6, 0x71, 0x70, 0x08, 0x32, 0x01, 0x19, 0x17,\n\t0xcb, 0x5f, 0x1d, 0xfd, 0xb2, 0xfc, 0x72, 0xc9, 0x78, 0xab, 0x98, 0x9c, 0xff, 0x04, 0x00, 0x00,\n\t0xff, 0xff, 0x3c, 0x85, 0x6e, 0xf3, 0x1e, 0x02, 0x00, 0x00,\n}\n\nfunc (m *TransientDecisionInfo) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *TransientDecisionInfo) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *TransientDecisionInfo) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif m.StartedEvent != nil {\n\t\t{\n\t\t\tsize, err := m.StartedEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintHistory(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0x12\n\t}\n\tif m.ScheduledEvent != nil {\n\t\t{\n\t\t\tsize, err := m.ScheduledEvent.MarshalToSizedBuffer(dAtA[:i])\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t\ti -= size\n\t\t\ti = encodeVarintHistory(dAtA, i, uint64(size))\n\t\t}\n\t\ti--\n\t\tdAtA[i] = 0xa\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc (m *VersionHistories) Marshal() (dAtA []byte, err error) {\n\tsize := m.Size()\n\tdAtA = make([]byte, size)\n\tn, err := m.MarshalToSizedBuffer(dAtA[:size])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn dAtA[:n], nil\n}\n\nfunc (m *VersionHistories) MarshalTo(dAtA []byte) (int, error) {\n\tsize := m.Size()\n\treturn m.MarshalToSizedBuffer(dAtA[:size])\n}\n\nfunc (m *VersionHistories) MarshalToSizedBuffer(dAtA []byte) (int, error) {\n\ti := len(dAtA)\n\t_ = i\n\tvar l int\n\t_ = l\n\tif m.XXX_unrecognized != nil {\n\t\ti -= len(m.XXX_unrecognized)\n\t\tcopy(dAtA[i:], m.XXX_unrecognized)\n\t}\n\tif len(m.Histories) > 0 {\n\t\tfor iNdEx := len(m.Histories) - 1; iNdEx >= 0; iNdEx-- {\n\t\t\t{\n\t\t\t\tsize, err := m.Histories[iNdEx].MarshalToSizedBuffer(dAtA[:i])\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn 0, err\n\t\t\t\t}\n\t\t\t\ti -= size\n\t\t\t\ti = encodeVarintHistory(dAtA, i, uint64(size))\n\t\t\t}\n\t\t\ti--\n\t\t\tdAtA[i] = 0x12\n\t\t}\n\t}\n\tif m.CurrentVersionHistoryIndex != 0 {\n\t\ti = encodeVarintHistory(dAtA, i, uint64(m.CurrentVersionHistoryIndex))\n\t\ti--\n\t\tdAtA[i] = 0x8\n\t}\n\treturn len(dAtA) - i, nil\n}\n\nfunc encodeVarintHistory(dAtA []byte, offset int, v uint64) int {\n\toffset -= sovHistory(v)\n\tbase := offset\n\tfor v >= 1<<7 {\n\t\tdAtA[offset] = uint8(v&0x7f | 0x80)\n\t\tv >>= 7\n\t\toffset++\n\t}\n\tdAtA[offset] = uint8(v)\n\treturn base\n}\nfunc (m *TransientDecisionInfo) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.ScheduledEvent != nil {\n\t\tl = m.ScheduledEvent.Size()\n\t\tn += 1 + l + sovHistory(uint64(l))\n\t}\n\tif m.StartedEvent != nil {\n\t\tl = m.StartedEvent.Size()\n\t\tn += 1 + l + sovHistory(uint64(l))\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc (m *VersionHistories) Size() (n int) {\n\tif m == nil {\n\t\treturn 0\n\t}\n\tvar l int\n\t_ = l\n\tif m.CurrentVersionHistoryIndex != 0 {\n\t\tn += 1 + sovHistory(uint64(m.CurrentVersionHistoryIndex))\n\t}\n\tif len(m.Histories) > 0 {\n\t\tfor _, e := range m.Histories {\n\t\t\tl = e.Size()\n\t\t\tn += 1 + l + sovHistory(uint64(l))\n\t\t}\n\t}\n\tif m.XXX_unrecognized != nil {\n\t\tn += len(m.XXX_unrecognized)\n\t}\n\treturn n\n}\n\nfunc sovHistory(x uint64) (n int) {\n\treturn (math_bits.Len64(x|1) + 6) / 7\n}\nfunc sozHistory(x uint64) (n int) {\n\treturn sovHistory(uint64((x << 1) ^ uint64((int64(x) >> 63))))\n}\nfunc (m *TransientDecisionInfo) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: TransientDecisionInfo: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: TransientDecisionInfo: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field ScheduledEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.ScheduledEvent == nil {\n\t\t\t\tm.ScheduledEvent = &v1.HistoryEvent{}\n\t\t\t}\n\t\t\tif err := m.ScheduledEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field StartedEvent\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tif m.StartedEvent == nil {\n\t\t\t\tm.StartedEvent = &v1.HistoryEvent{}\n\t\t\t}\n\t\t\tif err := m.StartedEvent.Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipHistory(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc (m *VersionHistories) Unmarshal(dAtA []byte) error {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tfor iNdEx < l {\n\t\tpreIndex := iNdEx\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= uint64(b&0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfieldNum := int32(wire >> 3)\n\t\twireType := int(wire & 0x7)\n\t\tif wireType == 4 {\n\t\t\treturn fmt.Errorf(\"proto: VersionHistories: wiretype end group for non-group\")\n\t\t}\n\t\tif fieldNum <= 0 {\n\t\t\treturn fmt.Errorf(\"proto: VersionHistories: illegal tag %d (wire type %d)\", fieldNum, wire)\n\t\t}\n\t\tswitch fieldNum {\n\t\tcase 1:\n\t\t\tif wireType != 0 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field CurrentVersionHistoryIndex\", wireType)\n\t\t\t}\n\t\t\tm.CurrentVersionHistoryIndex = 0\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tm.CurrentVersionHistoryIndex |= int32(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 2:\n\t\t\tif wireType != 2 {\n\t\t\t\treturn fmt.Errorf(\"proto: wrong wireType = %d for field Histories\", wireType)\n\t\t\t}\n\t\t\tvar msglen int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tmsglen |= int(b&0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif msglen < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tpostIndex := iNdEx + msglen\n\t\t\tif postIndex < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tif postIndex > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.Histories = append(m.Histories, &v11.VersionHistory{})\n\t\t\tif err := m.Histories[len(m.Histories)-1].Unmarshal(dAtA[iNdEx:postIndex]); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tiNdEx = postIndex\n\t\tdefault:\n\t\t\tiNdEx = preIndex\n\t\t\tskippy, err := skipHistory(dAtA[iNdEx:])\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif (skippy < 0) || (iNdEx+skippy) < 0 {\n\t\t\t\treturn ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tif (iNdEx + skippy) > l {\n\t\t\t\treturn io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tm.XXX_unrecognized = append(m.XXX_unrecognized, dAtA[iNdEx:iNdEx+skippy]...)\n\t\t\tiNdEx += skippy\n\t\t}\n\t}\n\n\tif iNdEx > l {\n\t\treturn io.ErrUnexpectedEOF\n\t}\n\treturn nil\n}\nfunc skipHistory(dAtA []byte) (n int, err error) {\n\tl := len(dAtA)\n\tiNdEx := 0\n\tdepth := 0\n\tfor iNdEx < l {\n\t\tvar wire uint64\n\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\tif shift >= 64 {\n\t\t\t\treturn 0, ErrIntOverflowHistory\n\t\t\t}\n\t\t\tif iNdEx >= l {\n\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t}\n\t\t\tb := dAtA[iNdEx]\n\t\t\tiNdEx++\n\t\t\twire |= (uint64(b) & 0x7F) << shift\n\t\t\tif b < 0x80 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\twireType := int(wire & 0x7)\n\t\tswitch wireType {\n\t\tcase 0:\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tiNdEx++\n\t\t\t\tif dAtA[iNdEx-1] < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\tcase 1:\n\t\t\tiNdEx += 8\n\t\tcase 2:\n\t\t\tvar length int\n\t\t\tfor shift := uint(0); ; shift += 7 {\n\t\t\t\tif shift >= 64 {\n\t\t\t\t\treturn 0, ErrIntOverflowHistory\n\t\t\t\t}\n\t\t\t\tif iNdEx >= l {\n\t\t\t\t\treturn 0, io.ErrUnexpectedEOF\n\t\t\t\t}\n\t\t\t\tb := dAtA[iNdEx]\n\t\t\t\tiNdEx++\n\t\t\t\tlength |= (int(b) & 0x7F) << shift\n\t\t\t\tif b < 0x80 {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif length < 0 {\n\t\t\t\treturn 0, ErrInvalidLengthHistory\n\t\t\t}\n\t\t\tiNdEx += length\n\t\tcase 3:\n\t\t\tdepth++\n\t\tcase 4:\n\t\t\tif depth == 0 {\n\t\t\t\treturn 0, ErrUnexpectedEndOfGroupHistory\n\t\t\t}\n\t\t\tdepth--\n\t\tcase 5:\n\t\t\tiNdEx += 4\n\t\tdefault:\n\t\t\treturn 0, fmt.Errorf(\"proto: illegal wireType %d\", wireType)\n\t\t}\n\t\tif iNdEx < 0 {\n\t\t\treturn 0, ErrInvalidLengthHistory\n\t\t}\n\t\tif depth == 0 {\n\t\t\treturn iNdEx, nil\n\t\t}\n\t}\n\treturn 0, io.ErrUnexpectedEOF\n}\n\nvar (\n\tErrInvalidLengthHistory        = fmt.Errorf(\"proto: negative length found during unmarshaling\")\n\tErrIntOverflowHistory          = fmt.Errorf(\"proto: integer overflow\")\n\tErrUnexpectedEndOfGroupHistory = fmt.Errorf(\"proto: unexpected end of group\")\n)\n"
  },
  {
    "path": ".gen/proto/shared/v1/history.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/shared/v1/history.proto\n\npackage sharedv1\n\nvar yarpcFileDescriptorClosure0370c4177fcc3ee8 = [][]byte{\n\t// uber/cadence/shared/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0x4f, 0x4b, 0xfb, 0x30,\n\t\t0x18, 0xc7, 0xe9, 0x7e, 0xfc, 0x04, 0x33, 0xff, 0x51, 0x50, 0x46, 0x41, 0xd8, 0xa6, 0xc2, 0x4e,\n\t\t0x09, 0x9d, 0x88, 0x07, 0x4f, 0xfe, 0xc5, 0x79, 0x2c, 0xe2, 0xc1, 0x4b, 0x49, 0x93, 0xc7, 0x35,\n\t\t0xe0, 0x92, 0x92, 0xa4, 0xc1, 0xbd, 0x15, 0xdf, 0x82, 0x6f, 0x52, 0xd2, 0xd6, 0x8d, 0xb8, 0x8b,\n\t\t0xb7, 0x3e, 0x3c, 0x9f, 0xcf, 0x97, 0x6f, 0x9f, 0xa0, 0xd3, 0xba, 0x00, 0x4d, 0x18, 0xe5, 0x20,\n\t\t0x19, 0x10, 0x53, 0x52, 0x0d, 0x9c, 0xb8, 0x94, 0x94, 0xc2, 0x58, 0xa5, 0x97, 0xb8, 0xd2, 0xca,\n\t\t0xaa, 0xf8, 0xc8, 0x53, 0xb8, 0xa3, 0x70, 0x4b, 0x61, 0x97, 0x26, 0xa3, 0xc0, 0xa6, 0x95, 0xd8,\n\t\t0x50, 0x93, 0x93, 0x10, 0xe1, 0x0b, 0x21, 0x37, 0xa0, 0xf1, 0x57, 0x84, 0x0e, 0x9f, 0x35, 0x95,\n\t\t0x46, 0x80, 0xb4, 0x77, 0xc0, 0x84, 0x11, 0x4a, 0xce, 0xe4, 0x9b, 0x8a, 0x9f, 0xd0, 0xbe, 0x61,\n\t\t0x25, 0xf0, 0xfa, 0x1d, 0x78, 0x0e, 0x0e, 0xa4, 0x1d, 0x44, 0xc3, 0x68, 0xd2, 0x9f, 0x8e, 0x70,\n\t\t0xd0, 0x89, 0x56, 0x02, 0xbb, 0x14, 0x3f, 0xb6, 0xb1, 0xf7, 0x1e, 0xcc, 0xf6, 0x56, 0x66, 0x33,\n\t\t0xc7, 0x0f, 0x68, 0xd7, 0x58, 0xaa, 0xed, 0x2a, 0xa9, 0xf7, 0xd7, 0xa4, 0x9d, 0xce, 0x6b, 0xa6,\n\t\t0xf1, 0x67, 0x84, 0x0e, 0x5e, 0x40, 0xfb, 0x8e, 0x2d, 0x25, 0xc0, 0xc4, 0xd7, 0xe8, 0x98, 0xd5,\n\t\t0x5a, 0x83, 0xb4, 0xb9, 0x6b, 0x77, 0x79, 0xf7, 0x8f, 0xb9, 0x90, 0x1c, 0x3e, 0x9a, 0xda, 0xff,\n\t\t0xb3, 0xa4, 0x83, 0x02, 0x7f, 0x39, 0xf3, 0x44, 0x7c, 0x8b, 0xb6, 0xcb, 0x9f, 0xbc, 0x41, 0x6f,\n\t\t0xf8, 0x6f, 0xd2, 0x9f, 0x9e, 0xfd, 0xea, 0xe6, 0xcf, 0xe7, 0xdb, 0x85, 0x7a, 0xb6, 0xf6, 0x6e,\n\t\t0x2e, 0x5f, 0x2f, 0xe6, 0xc2, 0x96, 0x75, 0x81, 0x99, 0x5a, 0x90, 0xe0, 0xf8, 0x78, 0x0e, 0x92,\n\t\t0x34, 0x07, 0x5f, 0x3f, 0xf4, 0x55, 0xfb, 0xe5, 0xd2, 0x62, 0xab, 0xd9, 0x9c, 0x7f, 0x07, 0x00,\n\t\t0x00, 0xff, 0xff, 0xc6, 0xf4, 0xa6, 0x9c, 0x12, 0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x5c, 0x5d, 0x6c, 0x1c, 0x49,\n\t\t0xb5, 0xde, 0x9e, 0xb1, 0xc7, 0x9e, 0x33, 0x8e, 0x3d, 0x2e, 0x3b, 0x8e, 0xff, 0x12, 0x3b, 0x93,\n\t\t0x6c, 0xe2, 0x75, 0x9c, 0x71, 0xe2, 0x64, 0x93, 0x4d, 0xb2, 0x3f, 0xd7, 0x76, 0x6c, 0xed, 0x48,\n\t\t0xbe, 0x49, 0x6e, 0xc7, 0xc9, 0xde, 0x7b, 0xb5, 0xd2, 0xd0, 0xee, 0x2e, 0xc7, 0x8d, 0x67, 0xba,\n\t\t0x67, 0xbb, 0x6b, 0x3c, 0x31, 0x82, 0x27, 0x1e, 0x90, 0x10, 0x2b, 0x58, 0xad, 0x90, 0x58, 0x09,\n\t\t0x04, 0x42, 0x02, 0xb1, 0xfc, 0x68, 0x11, 0x08, 0xf1, 0xf7, 0x02, 0x48, 0x68, 0x91, 0x40, 0x0b,\n\t\t0x4f, 0xbc, 0xf0, 0x84, 0x84, 0x10, 0xfb, 0xc6, 0x03, 0xcb, 0x33, 0xea, 0xea, 0xea, 0x99, 0xe9,\n\t\t0x9e, 0xaa, 0xfe, 0x19, 0x3b, 0x59, 0xd0, 0xe6, 0x6d, 0xba, 0xfb, 0x9c, 0xd3, 0x5f, 0x55, 0x9d,\n\t\t0x73, 0xea, 0xd4, 0x39, 0xa7, 0x07, 0x4e, 0xd6, 0xb7, 0xb0, 0xb5, 0xa8, 0x2a, 0x1a, 0x36, 0x54,\n\t\t0xbc, 0xa8, 0xd4, 0xf4, 0xc5, 0xbd, 0x8b, 0x8b, 0x3b, 0xba, 0x4d, 0x4c, 0x6b, 0xbf, 0x58, 0xb3,\n\t\t0x4c, 0x62, 0xa2, 0x11, 0x87, 0xa4, 0xc8, 0x48, 0x8a, 0x4a, 0x4d, 0x2f, 0xee, 0x5d, 0x9c, 0x3c,\n\t\t0xf1, 0xc0, 0x34, 0x1f, 0x54, 0xf0, 0x22, 0x25, 0xd9, 0xaa, 0x6f, 0x2f, 0x6a, 0x75, 0x4b, 0x21,\n\t\t0xba, 0x69, 0xb8, 0x4c, 0x93, 0x33, 0xc1, 0xe7, 0x44, 0xaf, 0x62, 0x9b, 0x28, 0xd5, 0x1a, 0x23,\n\t\t0x98, 0xe5, 0xbd, 0x58, 0x35, 0xab, 0xd5, 0xa6, 0x88, 0x02, 0x8f, 0x82, 0x28, 0xf6, 0x6e, 0x45,\n\t\t0xb7, 0x49, 0x18, 0x4d, 0xc3, 0xb4, 0x76, 0xb7, 0x2b, 0x66, 0xc3, 0xa5, 0x29, 0xdc, 0x84, 0xbe,\n\t\t0x97, 0xdd, 0x01, 0xa1, 0x6b, 0x90, 0xc1, 0x7b, 0xd8, 0x20, 0xf6, 0xb8, 0x34, 0x9b, 0x9e, 0xcb,\n\t\t0x2d, 0x9d, 0x2c, 0x72, 0xc6, 0x56, 0x64, 0xd4, 0x6b, 0x0e, 0xa5, 0xcc, 0x18, 0x0a, 0xef, 0x5f,\n\t\t0x85, 0x81, 0xf6, 0x07, 0x68, 0x02, 0xfa, 0xe9, 0xa3, 0xb2, 0xae, 0x8d, 0x4b, 0xb3, 0xd2, 0x5c,\n\t\t0x5a, 0xee, 0xa3, 0xd7, 0x25, 0x0d, 0x5d, 0x03, 0x70, 0x1f, 0x39, 0x83, 0x1e, 0x4f, 0xcd, 0x4a,\n\t\t0x73, 0xb9, 0xa5, 0xc9, 0xa2, 0x3b, 0x23, 0x45, 0x6f, 0x46, 0x8a, 0x9b, 0xde, 0x8c, 0xc8, 0x59,\n\t\t0x4a, 0xed, 0x5c, 0xa3, 0x71, 0xe8, 0xdb, 0xc3, 0x96, 0xad, 0x9b, 0xc6, 0x78, 0xda, 0x15, 0xca,\n\t\t0x2e, 0xd1, 0x31, 0xe8, 0x73, 0x06, 0xef, 0xbc, 0xae, 0x87, 0x3e, 0xc9, 0x38, 0x97, 0x25, 0x0d,\n\t\t0x7d, 0x59, 0x82, 0x73, 0xde, 0x90, 0xcb, 0xf8, 0x21, 0x56, 0xeb, 0xce, 0x3a, 0x94, 0x6d, 0xa2,\n\t\t0x58, 0x04, 0x6b, 0x65, 0x17, 0x89, 0x42, 0x88, 0xa5, 0x6f, 0xd5, 0x09, 0xb6, 0xc7, 0x7b, 0x29,\n\t\t0x9e, 0xe7, 0xb9, 0x43, 0x7f, 0x85, 0xc9, 0x59, 0xf3, 0xc4, 0xdc, 0x75, 0xa5, 0xd0, 0x21, 0x2f,\n\t\t0x37, 0x65, 0xbc, 0xfc, 0x94, 0x7c, 0xb6, 0x11, 0x8f, 0x14, 0x7d, 0x5d, 0x82, 0xf3, 0x1c, 0x78,\n\t\t0xaa, 0x59, 0xad, 0x55, 0x30, 0x17, 0x60, 0x86, 0x02, 0x7c, 0x31, 0x1e, 0xc0, 0x55, 0x4f, 0x4e,\n\t\t0x27, 0xc4, 0x67, 0x1a, 0x71, 0x89, 0xd1, 0x5b, 0x12, 0xcc, 0x73, 0x40, 0x6e, 0x2b, 0x7a, 0x85,\n\t\t0x87, 0xb0, 0x8f, 0x22, 0xbc, 0x11, 0x0f, 0xe1, 0x3a, 0x15, 0xd2, 0x09, 0xef, 0x4c, 0x23, 0x16,\n\t\t0x25, 0xfa, 0x1a, 0x7f, 0x02, 0x1d, 0xdd, 0xd2, 0xca, 0x66, 0x9d, 0x74, 0xc2, 0xeb, 0xa7, 0xf0,\n\t\t0x5e, 0x88, 0x07, 0xcf, 0x51, 0x3b, 0xed, 0x76, 0x9d, 0x74, 0x02, 0x9c, 0x6b, 0xc4, 0xa4, 0x45,\n\t\t0x6f, 0x4a, 0x30, 0xa7, 0x61, 0x55, 0xb7, 0x29, 0x30, 0x47, 0x4b, 0x6d, 0x75, 0x07, 0x6b, 0x75,\n\t\t0xee, 0xe4, 0x65, 0x29, 0xba, 0x6b, 0x5c, 0x74, 0x37, 0x99, 0x90, 0x4d, 0xc5, 0xde, 0xbd, 0xeb,\n\t\t0x89, 0xe8, 0x44, 0x76, 0x5a, 0x8b, 0x41, 0x87, 0x5e, 0x97, 0xe0, 0x4c, 0x00, 0x95, 0xc8, 0x26,\n\t\t0x80, 0x62, 0xba, 0x1a, 0x8d, 0x49, 0x64, 0x0e, 0x05, 0x2d, 0x92, 0x8a, 0x33, 0x4b, 0x21, 0x46,\n\t\t0x90, 0x8b, 0x39, 0x4b, 0x21, 0xfa, 0xef, 0x9b, 0x25, 0xa1, 0xea, 0xbf, 0xd1, 0x81, 0x2a, 0x44,\n\t\t0xb3, 0x06, 0x28, 0xaa, 0xe7, 0x22, 0x51, 0x89, 0x95, 0xea, 0x94, 0x16, 0x4d, 0x86, 0x3e, 0x2b,\n\t\t0xc1, 0xd3, 0x7e, 0x4c, 0x22, 0x4b, 0x3c, 0x42, 0x01, 0x5d, 0x89, 0x04, 0x24, 0x32, 0xc2, 0x93,\n\t\t0x5a, 0x14, 0x11, 0x5d, 0x36, 0x45, 0x25, 0xfa, 0x9e, 0x4e, 0xf6, 0x23, 0x95, 0x7b, 0x30, 0x64,\n\t\t0xd9, 0x96, 0x99, 0x90, 0x28, 0xe5, 0x56, 0x62, 0xd0, 0x51, 0xe5, 0x0e, 0xa0, 0x12, 0x29, 0xf7,\n\t\t0x50, 0x88, 0x72, 0xfb, 0x30, 0x09, 0x95, 0x5b, 0x89, 0xa4, 0xe2, 0xcc, 0x52, 0x88, 0x72, 0xe7,\n\t\t0x63, 0xce, 0x52, 0x98, 0x72, 0x2b, 0x31, 0xe8, 0xa8, 0x22, 0xf9, 0x51, 0x89, 0x14, 0x69, 0x38,\n\t\t0x44, 0x91, 0xda, 0x21, 0x09, 0x15, 0x49, 0x89, 0x22, 0xa2, 0x96, 0xe6, 0x07, 0x13, 0x62, 0x69,\n\t\t0x28, 0xc4, 0xd2, 0xda, 0xf1, 0x84, 0x58, 0x9a, 0x12, 0x4d, 0x86, 0x1a, 0x70, 0xc2, 0x01, 0x61,\n\t\t0x89, 0xb5, 0x67, 0x84, 0x02, 0xb9, 0xc0, 0x05, 0xe2, 0x48, 0xb5, 0x84, 0x6a, 0x33, 0x45, 0xc4,\n\t\t0x8f, 0xd1, 0x6b, 0x30, 0xed, 0xbe, 0x78, 0x5b, 0xb7, 0x78, 0xaf, 0x1d, 0xa5, 0xaf, 0x2d, 0x8a,\n\t\t0x5f, 0xbb, 0xee, 0xf0, 0x75, 0xbe, 0x74, 0x82, 0x88, 0x1e, 0xa2, 0x6f, 0x4a, 0xb0, 0x18, 0x50,\n\t\t0x51, 0xc5, 0x50, 0x71, 0xa5, 0x6c, 0xe1, 0xd7, 0xea, 0xd8, 0xe6, 0x8e, 0xfe, 0x28, 0x85, 0xf1,\n\t\t0x52, 0xb4, 0xa6, 0x52, 0x49, 0xb2, 0x27, 0xa8, 0x13, 0xd7, 0xbc, 0x12, 0x9b, 0x1a, 0xfd, 0x50,\n\t\t0x82, 0xcb, 0x0c, 0x93, 0x07, 0x31, 0x9e, 0x12, 0x8f, 0x51, 0xb4, 0xab, 0x5c, 0xb4, 0xec, 0x6d,\n\t\t0xee, 0xab, 0xe3, 0x68, 0x74, 0xd1, 0x4a, 0xc4, 0x81, 0xbe, 0x20, 0xc1, 0x59, 0xde, 0xf4, 0xf2,\n\t\t0x80, 0x1e, 0x8b, 0xa9, 0xdd, 0xab, 0x4c, 0x42, 0x84, 0x76, 0x0b, 0xc8, 0xd0, 0x27, 0x60, 0xc6,\n\t\t0x55, 0x32, 0x31, 0x92, 0x71, 0x8a, 0xe4, 0xa2, 0x58, 0xcf, 0xc4, 0x10, 0x5c, 0x05, 0x16, 0xbd,\n\t\t0xfb, 0x33, 0x12, 0x9c, 0x66, 0x8b, 0xc7, 0x14, 0x5d, 0xb0, 0x68, 0x13, 0x14, 0xc1, 0xb3, 0x5c,\n\t\t0x04, 0xae, 0x70, 0x57, 0xdf, 0x05, 0xcb, 0x34, 0xab, 0x46, 0xd0, 0xa0, 0x4f, 0xc1, 0x6c, 0x55,\n\t\t0xb1, 0x76, 0xb1, 0x55, 0xb6, 0xb0, 0x6a, 0x5a, 0x1a, 0x0f, 0xc4, 0x24, 0x05, 0xb1, 0xc4, 0x05,\n\t\t0xf1, 0xdf, 0x94, 0x59, 0x66, 0xbc, 0x9d, 0x08, 0x8e, 0x57, 0xc3, 0x08, 0xd0, 0x57, 0x25, 0x58,\n\t\t0xe0, 0x9d, 0x4f, 0xf4, 0x07, 0x86, 0xc2, 0x9d, 0x90, 0xa9, 0x24, 0xe1, 0xeb, 0x5d, 0x26, 0x26,\n\t\t0x4e, 0xf8, 0x2a, 0xa0, 0x45, 0xdf, 0x90, 0xa0, 0xc8, 0x8b, 0xb0, 0xb1, 0x55, 0xd5, 0x0d, 0x85,\n\t\t0xeb, 0x17, 0xa6, 0x43, 0xfc, 0x42, 0x67, 0x88, 0xdd, 0x14, 0xc4, 0xf1, 0x0b, 0x8d, 0xd8, 0xd4,\n\t\t0xe8, 0x47, 0x12, 0x5c, 0xe6, 0x1d, 0xa5, 0x22, 0xbd, 0xd8, 0x71, 0x8a, 0xf6, 0x66, 0xcc, 0x13,\n\t\t0x55, 0x94, 0x2b, 0x5b, 0x6c, 0x24, 0x63, 0x11, 0x69, 0x80, 0xd8, 0x28, 0x4f, 0x24, 0xd1, 0x00,\n\t\t0xb1, 0x81, 0xce, 0x35, 0x62, 0xd2, 0xa2, 0xbf, 0x48, 0xb0, 0x16, 0xf0, 0xb8, 0xf8, 0x21, 0xc1,\n\t\t0x96, 0xa1, 0x54, 0xca, 0x1c, 0xe4, 0xba, 0xa1, 0x13, 0x9d, 0xaf, 0x18, 0x33, 0x14, 0xfa, 0xdd,\n\t\t0x68, 0x17, 0xbc, 0xc6, 0xe4, 0x77, 0x8c, 0xa7, 0xe4, 0x09, 0xef, 0x1c, 0xd0, 0x8b, 0xd6, 0x81,\n\t\t0x24, 0xa0, 0x3f, 0x49, 0xb0, 0x92, 0x60, 0x98, 0x22, 0x8f, 0x35, 0x4b, 0xc7, 0x78, 0xe7, 0x00,\n\t\t0x63, 0x14, 0x39, 0xb3, 0x1b, 0x56, 0xf7, 0xec, 0xe8, 0x3d, 0x09, 0x5e, 0x08, 0x1b, 0x4e, 0xb4,\n\t\t0x9d, 0x9c, 0xa4, 0x03, 0xdb, 0xe0, 0x0e, 0x4c, 0x08, 0x26, 0xd2, 0x5e, 0xae, 0xe2, 0xee, 0x58,\n\t\t0x69, 0x1c, 0xc0, 0x4d, 0x9d, 0x18, 0x44, 0x37, 0xea, 0x58, 0x2b, 0x2b, 0x76, 0xd9, 0xc0, 0x8d,\n\t\t0xce, 0x71, 0x14, 0x42, 0xe2, 0x00, 0x4e, 0x06, 0x85, 0x89, 0x5b, 0xb6, 0x6f, 0xe1, 0x06, 0x27,\n\t\t0x0e, 0x68, 0x24, 0xe2, 0x40, 0xbf, 0x92, 0xe0, 0x1a, 0x8d, 0x26, 0xcb, 0xea, 0x8e, 0x5e, 0xd1,\n\t\t0x12, 0xda, 0xcf, 0x29, 0x0a, 0xfd, 0x65, 0x2e, 0x74, 0x1a, 0x4a, 0xae, 0x3a, 0x42, 0x93, 0x18,\n\t\t0xcd, 0x25, 0x3b, 0x39, 0x1b, 0xfa, 0xa9, 0x04, 0x57, 0x22, 0x06, 0x21, 0xb2, 0x8e, 0xd3, 0x74,\n\t\t0x04, 0x6b, 0x49, 0x47, 0x20, 0x32, 0x89, 0x0b, 0x76, 0x42, 0x1e, 0xf4, 0x5d, 0x09, 0x2e, 0x0a,\n\t\t0x51, 0x0b, 0xe3, 0xfc, 0xa7, 0x29, 0xec, 0x65, 0x7e, 0x18, 0xc2, 0x7d, 0xbb, 0x30, 0xf0, 0x5f,\n\t\t0x50, 0x13, 0xd0, 0xa3, 0x1f, 0x48, 0x70, 0x49, 0x08, 0x37, 0xe4, 0x10, 0x79, 0x26, 0x44, 0xc9,\n\t\t0xf9, 0x80, 0x43, 0x8e, 0x93, 0x45, 0x35, 0x11, 0x07, 0x7a, 0x5b, 0x82, 0x0b, 0x89, 0x35, 0xe3,\n\t\t0x2c, 0x45, 0xfc, 0x5f, 0x09, 0x10, 0x8b, 0x94, 0xe2, 0x9c, 0x9a, 0x40, 0x1f, 0xde, 0x91, 0x60,\n\t\t0x49, 0x3c, 0xc1, 0xc2, 0x4d, 0x78, 0x8e, 0xa2, 0x5d, 0x49, 0x32, 0xbf, 0xc2, 0x9d, 0xf8, 0xbc,\n\t\t0x9a, 0x84, 0x01, 0x7d, 0x3f, 0x4c, 0x25, 0x42, 0x0e, 0xcd, 0xcf, 0x24, 0x86, 0x2c, 0x3e, 0x3e,\n\t\t0x0b, 0x20, 0x8b, 0x0e, 0xd2, 0x4e, 0x6c, 0x26, 0x86, 0x1c, 0x12, 0x49, 0xce, 0x87, 0xc4, 0x66,\n\t\t0x02, 0xcc, 0x21, 0xe1, 0xe4, 0xa2, 0x9a, 0x8c, 0x85, 0x6e, 0x9a, 0x6e, 0x28, 0xde, 0x6d, 0xc4,\n\t\t0x73, 0x2e, 0x64, 0xd3, 0x74, 0x23, 0xee, 0x6e, 0x42, 0x9d, 0xab, 0x76, 0x77, 0xac, 0xe8, 0xd7,\n\t\t0x12, 0x5c, 0x8f, 0x31, 0x20, 0x91, 0x8d, 0x2e, 0xd0, 0xd1, 0x94, 0xba, 0x19, 0x8d, 0xc8, 0x58,\n\t\t0x2f, 0xdb, 0x5d, 0xf0, 0xa1, 0x9f, 0x48, 0xf0, 0x6c, 0xd8, 0x00, 0xc4, 0xe7, 0xa7, 0xf3, 0x21,\n\t\t0x1b, 0x90, 0x10, 0x84, 0xf8, 0x1c, 0x75, 0x01, 0x27, 0xe4, 0xa1, 0x0e, 0xa7, 0x5e, 0xb3, 0xb1,\n\t\t0x45, 0x5a, 0xc0, 0x6d, 0xac, 0x58, 0xea, 0x4e, 0x1b, 0xcc, 0x4e, 0xdc, 0xc5, 0x10, 0xeb, 0xbd,\n\t\t0x47, 0xc5, 0x79, 0x08, 0xee, 0x52, 0x61, 0xad, 0x37, 0x72, 0xac, 0xb7, 0x9e, 0x84, 0x61, 0x65,\n\t\t0x00, 0xa0, 0x05, 0xa4, 0xf0, 0xe7, 0x21, 0x38, 0x1b, 0x77, 0xf7, 0x5a, 0x87, 0x23, 0xcd, 0x31,\n\t\t0x92, 0xfd, 0x1a, 0xa6, 0xb5, 0x40, 0x51, 0x65, 0xd1, 0x13, 0xba, 0xb9, 0x5f, 0xc3, 0xf2, 0x40,\n\t\t0xa3, 0xed, 0x0a, 0xbd, 0x0a, 0x47, 0x6b, 0x8a, 0xe5, 0xcc, 0x48, 0xbb, 0xd1, 0x6d, 0x9b, 0xac,\n\t\t0x7c, 0x38, 0xc7, 0x95, 0x77, 0x87, 0x72, 0xb4, 0xd9, 0xc4, 0xb6, 0x29, 0x8f, 0xd4, 0x3a, 0x6f,\n\t\t0xa2, 0xeb, 0x90, 0xa5, 0x19, 0x99, 0x8a, 0x6e, 0x13, 0x5a, 0x58, 0xcc, 0x2d, 0x1d, 0xe7, 0xa7,\n\t\t0x3c, 0x14, 0x7b, 0x77, 0x43, 0xb7, 0x89, 0xdc, 0x4f, 0xd8, 0x2f, 0xb4, 0x04, 0xbd, 0xba, 0x51,\n\t\t0xab, 0x13, 0x5a, 0x76, 0xcc, 0x2d, 0x4d, 0x0b, 0x90, 0xec, 0x57, 0x4c, 0x45, 0x93, 0x5d, 0x52,\n\t\t0xa4, 0xc0, 0x6c, 0x20, 0xe4, 0x28, 0x13, 0xb3, 0xac, 0x56, 0x4c, 0x1b, 0x53, 0xff, 0x6d, 0xd6,\n\t\t0x09, 0xab, 0x43, 0x4e, 0x74, 0xd4, 0x45, 0x6f, 0xb2, 0x4a, 0xb2, 0x3c, 0x8d, 0x7d, 0x73, 0xbf,\n\t\t0x69, 0xae, 0x3a, 0xfc, 0x9b, 0x2e, 0x3b, 0x7a, 0x05, 0xa6, 0x5a, 0x69, 0xef, 0x4e, 0xe9, 0x99,\n\t\t0x28, 0xe9, 0xc7, 0x88, 0x97, 0xcc, 0x0e, 0x08, 0xbe, 0x01, 0x93, 0xad, 0x08, 0xbb, 0x35, 0x0a,\n\t\t0xab, 0x6e, 0x94, 0x75, 0x8d, 0x96, 0xfe, 0xb2, 0xf2, 0xb1, 0x26, 0x45, 0x73, 0x9e, 0xe5, 0xba,\n\t\t0x51, 0xd2, 0x50, 0x09, 0xb2, 0xcc, 0x55, 0x9a, 0x16, 0xad, 0xc3, 0x0d, 0x2e, 0x9d, 0xe3, 0xbb,\n\t\t0x76, 0x26, 0x80, 0x86, 0xd0, 0x25, 0x8f, 0x45, 0x6e, 0x71, 0xa3, 0x12, 0x0c, 0xb7, 0x70, 0x38,\n\t\t0xee, 0xaa, 0x6e, 0x61, 0x56, 0x3c, 0xe3, 0xaf, 0xc1, 0xba, 0x4b, 0x23, 0xe7, 0x9b, 0x6c, 0xec,\n\t\t0x0e, 0x92, 0x61, 0xac, 0xa2, 0x38, 0x67, 0x3e, 0x37, 0x9c, 0xa1, 0xc3, 0xc1, 0x76, 0xbd, 0x42,\n\t\t0x58, 0xe1, 0x2b, 0x7c, 0x4d, 0x47, 0x1d, 0xde, 0xd5, 0x26, 0xab, 0x4c, 0x39, 0xd1, 0x35, 0x98,\n\t\t0x30, 0x2d, 0xfd, 0x81, 0xee, 0x3a, 0xda, 0xc0, 0x2c, 0xe5, 0xe8, 0x2c, 0x8d, 0x79, 0x04, 0x81,\n\t\t0x49, 0x9a, 0x84, 0x7e, 0x5d, 0xc3, 0x06, 0xd1, 0xc9, 0x3e, 0xad, 0x28, 0x65, 0xe5, 0xe6, 0x35,\n\t\t0xba, 0x04, 0x63, 0xdb, 0xba, 0x65, 0x93, 0x4e, 0x99, 0x47, 0x28, 0xe5, 0x08, 0x7d, 0x1a, 0x10,\n\t\t0xb8, 0x0a, 0x03, 0x16, 0x26, 0xd6, 0x7e, 0xb9, 0x66, 0x56, 0x74, 0x75, 0x9f, 0x55, 0x61, 0x66,\n\t\t0x05, 0x07, 0x54, 0x62, 0xed, 0xdf, 0xa1, 0x74, 0x72, 0xce, 0x6a, 0x5d, 0xa0, 0x71, 0xe8, 0x53,\n\t\t0x08, 0xc1, 0xd5, 0x1a, 0xa1, 0x15, 0x93, 0x5e, 0xd9, 0xbb, 0x44, 0xab, 0x30, 0x84, 0x1f, 0xd6,\n\t\t0x74, 0x57, 0x71, 0xdc, 0xa2, 0x7e, 0x3e, 0xb2, 0xa8, 0x3f, 0xd8, 0x62, 0xa1, 0x95, 0xfd, 0x53,\n\t\t0x70, 0x44, 0xb5, 0x1c, 0x6b, 0x60, 0x15, 0x1d, 0x5a, 0x71, 0xc8, 0xca, 0x03, 0xce, 0x4d, 0xaf,\n\t\t0xca, 0x83, 0xfe, 0x17, 0xa6, 0xdc, 0xd1, 0xfb, 0xab, 0x5f, 0x5b, 0x8a, 0xba, 0x6b, 0x6e, 0x6f,\n\t\t0xb3, 0xa2, 0x40, 0x88, 0x52, 0x8f, 0x53, 0xee, 0xf6, 0xc2, 0xd7, 0x8a, 0xcb, 0x8a, 0xce, 0x43,\n\t\t0x4f, 0x15, 0x57, 0x4d, 0x96, 0xce, 0x9f, 0xe0, 0x27, 0xfa, 0x70, 0xd5, 0x94, 0x29, 0x19, 0x92,\n\t\t0x61, 0xb8, 0xc3, 0x63, 0xb3, 0x9c, 0xfc, 0xd3, 0xfc, 0xbd, 0x31, 0xe0, 0x61, 0xe5, 0xbc, 0x1d,\n\t\t0xb8, 0x83, 0xee, 0xc1, 0x58, 0xcd, 0xc2, 0x7b, 0x65, 0xa5, 0x4e, 0x4c, 0x47, 0xff, 0x30, 0x29,\n\t\t0xd7, 0x4c, 0xdd, 0x20, 0x5e, 0x96, 0x5d, 0xb4, 0x5e, 0x36, 0x26, 0x77, 0x28, 0x9d, 0x3c, 0xe2,\n\t\t0xf0, 0x2f, 0xd7, 0x89, 0xd9, 0x76, 0x13, 0x5d, 0x82, 0xcc, 0x0e, 0x56, 0x34, 0x6c, 0xb1, 0xf4,\n\t\t0xf7, 0x14, 0xbf, 0xa9, 0x83, 0x92, 0xc8, 0x8c, 0x14, 0x6d, 0xc0, 0xa8, 0x3b, 0xd1, 0xad, 0x5a,\n\t\t0x1e, 0x5d, 0xd7, 0x63, 0x91, 0xeb, 0x8a, 0x28, 0x5f, 0xb3, 0x2e, 0x47, 0xd7, 0xf6, 0x93, 0x90,\n\t\t0xaf, 0x29, 0x16, 0xd1, 0xbd, 0xe3, 0xf9, 0xb6, 0xfe, 0x60, 0x7c, 0x9c, 0x76, 0x98, 0xfc, 0xcf,\n\t\t0x41, 0xda, 0x2c, 0x1c, 0xff, 0xee, 0x0a, 0x5d, 0xa5, 0x32, 0xd7, 0x0c, 0x62, 0xed, 0xcb, 0x43,\n\t\t0x35, 0xff, 0x5d, 0x74, 0x1c, 0xc0, 0x4b, 0xea, 0xe8, 0x1a, 0x4d, 0x27, 0x67, 0xe5, 0x2c, 0xbb,\n\t\t0x53, 0xd2, 0xd0, 0x7d, 0x18, 0xa1, 0x8a, 0x67, 0xee, 0x61, 0xab, 0xa2, 0xd4, 0x3c, 0x1b, 0x99,\n\t\t0xa4, 0xce, 0xe9, 0x0c, 0xdf, 0x39, 0x59, 0xa6, 0x71, 0xdb, 0x25, 0x67, 0x96, 0x32, 0xac, 0x06,\n\t\t0x6f, 0xa1, 0x87, 0x30, 0x43, 0x73, 0xf0, 0xb8, 0xac, 0x56, 0xea, 0x36, 0xc1, 0x56, 0xd9, 0xc6,\n\t\t0x15, 0xac, 0xd2, 0x39, 0x60, 0xef, 0x98, 0x0a, 0x49, 0xae, 0xd3, 0x34, 0x3f, 0x5e, 0x75, 0x59,\n\t\t0xef, 0x7a, 0x9c, 0xec, 0x75, 0xd3, 0x4a, 0xc8, 0xd3, 0xc9, 0x15, 0x18, 0xe5, 0xcd, 0x0c, 0xca,\n\t\t0x43, 0x7a, 0x17, 0xef, 0xd3, 0x1d, 0x38, 0x2b, 0x3b, 0x3f, 0xd1, 0x28, 0xf4, 0xee, 0x29, 0x95,\n\t\t0xba, 0xdb, 0x84, 0x93, 0x95, 0xdd, 0x8b, 0xeb, 0xa9, 0xe7, 0xa4, 0xc2, 0xdb, 0x12, 0x3c, 0x13,\n\t\t0xff, 0xb8, 0x77, 0x19, 0x32, 0xcc, 0x61, 0x4a, 0x31, 0x1c, 0x26, 0xa3, 0x45, 0xeb, 0x30, 0x1b,\n\t\t0x5e, 0xef, 0xd7, 0x35, 0x0a, 0x2c, 0x2d, 0x4f, 0x8b, 0x4b, 0xf5, 0x25, 0xad, 0xf0, 0x2d, 0x09,\n\t\t0xce, 0xc4, 0x8c, 0x1a, 0xaf, 0x40, 0x9f, 0xb7, 0x55, 0x48, 0x31, 0xb6, 0x0a, 0x8f, 0xf8, 0xd0,\n\t\t0xa0, 0x9a, 0x30, 0x17, 0xfb, 0xc8, 0xb4, 0x0a, 0x03, 0x6c, 0xb7, 0x6e, 0x45, 0x4e, 0x83, 0x02,\n\t\t0x2f, 0xc0, 0x36, 0x67, 0x1a, 0x38, 0xe5, 0x48, 0xeb, 0xa2, 0xf0, 0x3b, 0x09, 0x4e, 0xc7, 0xe9,\n\t\t0x1a, 0xf1, 0x87, 0x40, 0x52, 0xb2, 0x10, 0xe8, 0x16, 0x8c, 0x09, 0xc2, 0x8c, 0x54, 0x94, 0x47,\n\t\t0x1e, 0xb1, 0x39, 0x21, 0x46, 0xdb, 0x56, 0x93, 0xf6, 0x6d, 0x35, 0x85, 0xd7, 0x25, 0x28, 0x44,\n\t\t0x37, 0x9c, 0xa0, 0x05, 0x40, 0xc1, 0x26, 0x84, 0x66, 0x1b, 0x5a, 0xde, 0xf6, 0x4d, 0x41, 0x60,\n\t\t0xbf, 0x4d, 0x05, 0xf6, 0x5b, 0xbf, 0xf3, 0x48, 0x07, 0x9c, 0x47, 0xe1, 0xef, 0x81, 0xe9, 0x15,\n\t\t0x5a, 0x48, 0x32, 0x44, 0x73, 0x90, 0xf7, 0x27, 0xa2, 0x9a, 0xea, 0x35, 0x68, 0xb7, 0x8d, 0x38,\n\t\t0x80, 0x3d, 0x1d, 0xc0, 0x7e, 0x16, 0x86, 0xb6, 0x74, 0x43, 0xb1, 0xf6, 0xcb, 0xea, 0x0e, 0x56,\n\t\t0x77, 0xed, 0x7a, 0x95, 0xc6, 0xa8, 0x59, 0x79, 0xd0, 0xbd, 0xbd, 0xca, 0xee, 0xa2, 0x73, 0x30,\n\t\t0xec, 0x4f, 0x9f, 0xe2, 0x87, 0x6e, 0xfc, 0x39, 0x20, 0xe7, 0x71, 0x7b, 0x56, 0x13, 0x3f, 0x24,\n\t\t0x85, 0xef, 0xa4, 0xe1, 0x54, 0x8c, 0x5e, 0x96, 0x47, 0x36, 0xe2, 0xa0, 0x59, 0xa4, 0xbb, 0x30,\n\t\t0x0b, 0x74, 0x02, 0x72, 0x5b, 0x8a, 0x8d, 0xbd, 0xd8, 0xc9, 0x9d, 0x96, 0xac, 0x73, 0xcb, 0x8d,\n\t\t0x98, 0xa6, 0x01, 0x0c, 0xdc, 0xf0, 0x1e, 0xf7, 0xba, 0x13, 0x6b, 0xe0, 0x86, 0xfb, 0x74, 0x01,\n\t\t0xd0, 0xb6, 0x69, 0xed, 0x32, 0xa4, 0x5e, 0x43, 0x62, 0xc6, 0x1d, 0x9a, 0xf3, 0x84, 0x62, 0xbd,\n\t\t0xcf, 0x3a, 0x13, 0xc7, 0x1c, 0xe7, 0xa8, 0xd8, 0xa6, 0xc1, 0x82, 0x63, 0x76, 0x85, 0x6e, 0x42,\n\t\t0xaf, 0xaa, 0xd4, 0x6d, 0xcc, 0xe2, 0xe0, 0x62, 0xec, 0xae, 0xa1, 0x55, 0x87, 0x4b, 0x76, 0x99,\n\t\t0x03, 0x0a, 0x9a, 0x0d, 0x2a, 0xe8, 0xbb, 0x69, 0x38, 0x19, 0xd9, 0xe8, 0xf3, 0xc8, 0xd6, 0x6a,\n\t\t0xc5, 0x1b, 0xa2, 0xbb, 0x48, 0x0b, 0x31, 0xfb, 0x90, 0x7c, 0x03, 0x6c, 0x73, 0xd9, 0x3d, 0x49,\n\t\t0x5c, 0x76, 0xbb, 0x65, 0xf4, 0x06, 0x2c, 0x23, 0xb0, 0xfc, 0x99, 0xf0, 0xe5, 0xef, 0x8b, 0xb5,\n\t\t0xfc, 0xfd, 0x82, 0xe5, 0xe7, 0x58, 0x61, 0x96, 0x6b, 0x85, 0xfe, 0x95, 0x84, 0xe0, 0x4a, 0x7e,\n\t\t0x25, 0x03, 0xa7, 0xe3, 0xb4, 0x48, 0xa1, 0x19, 0xc8, 0x35, 0xfb, 0x0c, 0xd8, 0x2a, 0x66, 0x65,\n\t\t0xf0, 0x6e, 0x95, 0x34, 0xe7, 0x4c, 0xde, 0x6a, 0x44, 0x70, 0x4c, 0x28, 0x15, 0x72, 0x26, 0x6f,\n\t\t0xbe, 0x92, 0x9e, 0xc9, 0x95, 0xb6, 0x2b, 0x47, 0xb1, 0x35, 0xb3, 0xaa, 0xe8, 0x06, 0xf3, 0x3c,\n\t\t0xec, 0xca, 0xbf, 0x95, 0xf4, 0x74, 0x79, 0x9a, 0xce, 0xc4, 0x3f, 0x4d, 0x6f, 0xc2, 0x84, 0xa7,\n\t\t0xa3, 0x9d, 0x3b, 0x50, 0x5f, 0xd4, 0x0e, 0x34, 0xe6, 0xf1, 0x06, 0x36, 0xa1, 0x80, 0x54, 0xb6,\n\t\t0xc1, 0x31, 0xa9, 0xfd, 0x09, 0xa4, 0xba, 0x87, 0x68, 0x26, 0x55, 0xbc, 0x55, 0x66, 0xbb, 0xda,\n\t\t0x2a, 0xd7, 0x61, 0x78, 0x07, 0x2b, 0x16, 0xd9, 0xc2, 0x4a, 0x0b, 0x1d, 0x44, 0x89, 0xca, 0x37,\n\t\t0x79, 0x5a, 0x72, 0xa2, 0x03, 0x9c, 0x5c, 0x74, 0x80, 0xd3, 0x71, 0xd4, 0x1c, 0xe8, 0xe6, 0xa8,\n\t\t0xd9, 0x3a, 0xb2, 0x1c, 0x89, 0x7d, 0x64, 0x29, 0xfc, 0x4d, 0x82, 0x42, 0x74, 0xbb, 0xde, 0x63,\n\t\t0x0b, 0x0d, 0xda, 0x83, 0x98, 0x1e, 0xff, 0x79, 0xf9, 0x25, 0x18, 0xa0, 0xe9, 0x06, 0xcf, 0xad,\n\t\t0xf5, 0xc6, 0x70, 0x6b, 0x39, 0x87, 0x83, 0x5d, 0x14, 0xfe, 0x20, 0xf9, 0x5d, 0xc1, 0x21, 0xc7,\n\t\t0xe5, 0xfc, 0x29, 0x4a, 0x25, 0xd8, 0x0d, 0xd2, 0x91, 0xb1, 0x4a, 0x8f, 0x7f, 0x32, 0x0b, 0xbf,\n\t\t0x97, 0xe0, 0x64, 0x74, 0x0f, 0x55, 0xb7, 0xe1, 0xfb, 0x87, 0x31, 0xa2, 0x9f, 0xa7, 0xe0, 0x54,\n\t\t0x8c, 0x4e, 0x44, 0x67, 0x4c, 0x1a, 0x26, 0x8a, 0x5e, 0xb1, 0x63, 0x2d, 0x92, 0x47, 0xfc, 0xc8,\n\t\t0xc6, 0x14, 0x8c, 0xaf, 0x7a, 0xba, 0x89, 0xaf, 0x0e, 0xac, 0xe2, 0x5f, 0x94, 0x60, 0x3e, 0x7e,\n\t\t0x03, 0x61, 0x9c, 0x3d, 0xef, 0x70, 0x0e, 0x70, 0xef, 0x48, 0x90, 0xb0, 0x55, 0x30, 0x1a, 0xdb,\n\t\t0xa8, 0x17, 0x25, 0xb1, 0x53, 0xb8, 0x1b, 0xf7, 0xc4, 0x41, 0x9c, 0x8e, 0x81, 0xf8, 0xad, 0x80,\n\t\t0x1e, 0x8a, 0x8a, 0x8a, 0xdd, 0xea, 0xe1, 0x3a, 0xcc, 0x56, 0x14, 0xd2, 0xd6, 0x32, 0x13, 0x6c,\n\t\t0x20, 0x69, 0xcd, 0xac, 0x4b, 0xc7, 0x5b, 0x4a, 0x37, 0xaa, 0xe2, 0xe8, 0x73, 0x3a, 0x81, 0x3e,\n\t\t0xf7, 0x44, 0xda, 0x68, 0x20, 0x0e, 0x2c, 0xbc, 0x27, 0xc1, 0x54, 0x48, 0x93, 0x2e, 0x9a, 0x80,\n\t\t0x7e, 0xb7, 0x39, 0xb1, 0xb9, 0x6e, 0x7d, 0xf4, 0xba, 0xa4, 0xa1, 0x0d, 0x38, 0xda, 0xdc, 0xc8,\n\t\t0xb7, 0x75, 0x2b, 0xc1, 0x91, 0x17, 0xb1, 0x7d, 0x7c, 0x5d, 0xb7, 0x70, 0x92, 0xed, 0x37, 0xce,\n\t\t0x62, 0x7f, 0x0c, 0x26, 0x84, 0xdd, 0xbf, 0x61, 0xa3, 0x89, 0x1d, 0xd2, 0x17, 0xde, 0x95, 0x60,\n\t\t0x3a, 0xac, 0xf1, 0xf3, 0x50, 0xde, 0x72, 0x58, 0xf3, 0x11, 0xea, 0xa0, 0x7f, 0x2c, 0xc1, 0x6c,\n\t\t0x54, 0x03, 0x69, 0xd8, 0x68, 0x1e, 0xa9, 0xd9, 0x86, 0x6f, 0x96, 0x59, 0x48, 0xd8, 0xa7, 0x84,\n\t\t0x16, 0x61, 0x94, 0xb6, 0x42, 0x05, 0xab, 0x06, 0xee, 0x98, 0x86, 0x0d, 0xdc, 0x08, 0xd4, 0x0c,\n\t\t0x3a, 0x0a, 0x77, 0xa9, 0xee, 0x0a, 0x77, 0x4f, 0x4a, 0x6b, 0xf1, 0x4b, 0x6b, 0x71, 0x74, 0xa7,\n\t\t0x2f, 0x86, 0xee, 0xdc, 0x86, 0x31, 0x56, 0x12, 0x61, 0x18, 0x75, 0x83, 0x60, 0x6b, 0x4f, 0xa9,\n\t\t0x44, 0x9f, 0x5b, 0x46, 0x19, 0x23, 0x85, 0x57, 0x62, 0x6c, 0xfe, 0xb2, 0x5d, 0xf6, 0x40, 0x65,\n\t\t0xbb, 0xb6, 0x10, 0x0e, 0x92, 0x84, 0x70, 0xe2, 0x1a, 0x5d, 0xae, 0xeb, 0x1a, 0x5d, 0xeb, 0x9c,\n\t\t0x31, 0x10, 0xbf, 0x34, 0xe2, 0x55, 0x8a, 0x8e, 0x1c, 0xa0, 0x52, 0x34, 0x78, 0xb0, 0x4a, 0x91,\n\t\t0xa0, 0x64, 0x31, 0xf4, 0x18, 0x4a, 0x16, 0xf9, 0x47, 0x52, 0xb2, 0x28, 0xfc, 0x55, 0x82, 0xc5,\n\t\t0xa4, 0xed, 0x9f, 0x4d, 0xff, 0x2b, 0xb5, 0xfb, 0xdf, 0xb0, 0x13, 0xdb, 0x16, 0x1c, 0x6b, 0xb6,\n\t\t0x8c, 0x04, 0xda, 0x08, 0x5c, 0xcf, 0x34, 0x1f, 0xda, 0x14, 0xe2, 0x6f, 0x24, 0x38, 0x8a, 0x79,\n\t\t0xb7, 0x03, 0xa7, 0xc2, 0x9e, 0x60, 0x16, 0xe7, 0xdb, 0x12, 0xa7, 0x02, 0x20, 0xda, 0x4a, 0xe3,\n\t\t0xf8, 0x03, 0x29, 0x86, 0x3f, 0x68, 0x0b, 0xed, 0x52, 0x09, 0x42, 0xbb, 0xc2, 0x07, 0x12, 0x1c,\n\t\t0x0f, 0xfd, 0xba, 0xc1, 0x89, 0x6d, 0xd9, 0xb7, 0x13, 0x86, 0x52, 0xf5, 0x56, 0x02, 0xdc, 0x5b,\n\t\t0xb7, 0x94, 0x2a, 0xee, 0xf6, 0xd5, 0x87, 0xb6, 0x8d, 0xb6, 0x4c, 0xbc, 0x27, 0x7e, 0x2a, 0xe1,\n\t\t0x67, 0xbc, 0x45, 0x12, 0x75, 0xf3, 0xcc, 0x40, 0x8e, 0xf5, 0x53, 0xb5, 0x4f, 0x81, 0x7b, 0x8b,\n\t\t0x4e, 0x41, 0x73, 0x17, 0x4b, 0xc5, 0xdf, 0xc5, 0xc2, 0xd2, 0xfa, 0x11, 0x1a, 0xf6, 0x25, 0x09,\n\t\t0xe6, 0x13, 0x34, 0xb8, 0xb5, 0xb2, 0xd3, 0x92, 0x2f, 0x3b, 0xdd, 0xed, 0xc2, 0x85, 0x20, 0x2f,\n\t\t0xfc, 0x32, 0x05, 0x2f, 0x1e, 0xac, 0xc9, 0xff, 0xd0, 0x4c, 0xa2, 0x95, 0xbb, 0x4c, 0xf9, 0x72,\n\t\t0x97, 0xf7, 0x00, 0x75, 0x36, 0x93, 0x31, 0xef, 0x70, 0x26, 0x5e, 0xb1, 0x5a, 0x1e, 0xee, 0xe8,\n\t\t0x08, 0x47, 0xe3, 0xd0, 0xa7, 0x9a, 0x06, 0xb1, 0xcc, 0x0a, 0x5d, 0xb0, 0x01, 0xd9, 0xbb, 0x44,\n\t\t0x45, 0x18, 0x09, 0xf4, 0x45, 0x9a, 0x46, 0xc5, 0x3d, 0xa9, 0xf4, 0xcb, 0xc3, 0xbe, 0x76, 0xc5,\n\t\t0xdb, 0x46, 0x65, 0xbf, 0xf0, 0x66, 0x1a, 0x6e, 0x1c, 0xe0, 0x23, 0x02, 0x74, 0xaf, 0xdd, 0x6b,\n\t\t0x0e, 0x0a, 0x3e, 0xd1, 0x89, 0x25, 0xd9, 0x97, 0xa5, 0x3f, 0xa4, 0xf3, 0xb5, 0x30, 0xa7, 0xcc,\n\t\t0x5f, 0x97, 0x9e, 0x83, 0xae, 0xcb, 0x02, 0xa0, 0x60, 0xeb, 0x26, 0xab, 0xf7, 0xa4, 0xe5, 0xbc,\n\t\t0xee, 0x53, 0x42, 0x37, 0xa5, 0xe7, 0xad, 0x62, 0xc6, 0xb7, 0x8a, 0x85, 0x3f, 0x4a, 0x70, 0xb5,\n\t\t0xcb, 0x2f, 0x20, 0x04, 0x18, 0x24, 0x01, 0x86, 0xc7, 0xab, 0xb8, 0x85, 0xcf, 0xa7, 0xe1, 0x6a,\n\t\t0x97, 0x5d, 0xaa, 0xff, 0xa9, 0xb6, 0x1a, 0x70, 0xe8, 0x3d, 0x62, 0x87, 0xde, 0x1b, 0xdf, 0xa1,\n\t\t0x0b, 0x55, 0x47, 0xe4, 0x00, 0xfa, 0x44, 0x0e, 0xe0, 0x73, 0x69, 0xb8, 0xdc, 0x4d, 0xa7, 0x6d,\n\t\t0x3c, 0xcb, 0x8f, 0x25, 0xf9, 0x89, 0xe5, 0xb7, 0x2c, 0xff, 0x7d, 0x09, 0x2e, 0x24, 0xed, 0x1a,\n\t\t0xfe, 0xb7, 0x36, 0x79, 0xf1, 0x5e, 0x55, 0xf8, 0xad, 0x04, 0xe7, 0x13, 0x75, 0x1a, 0x1f, 0x9a,\n\t\t0x0b, 0xe0, 0x9e, 0xa2, 0x52, 0x07, 0x3a, 0x45, 0x15, 0xbe, 0x97, 0x83, 0x4b, 0x5d, 0x7c, 0x32,\n\t\t0xd5, 0xb6, 0x1c, 0x92, 0x6f, 0x39, 0x66, 0x20, 0xd7, 0x5c, 0x0e, 0xa6, 0xf3, 0x59, 0x19, 0xbc,\n\t\t0x5b, 0xbc, 0x94, 0x4a, 0xfa, 0x10, 0x52, 0x2a, 0xdd, 0xd6, 0x57, 0x7b, 0x0f, 0x37, 0xa5, 0x92,\n\t\t0x79, 0xa4, 0x29, 0x95, 0xbe, 0xae, 0x53, 0x2a, 0xf7, 0x81, 0x35, 0x7c, 0x33, 0x89, 0xec, 0x18,\n\t\t0xdb, 0x1f, 0x72, 0x54, 0x76, 0xbb, 0xc6, 0xa9, 0x14, 0xef, 0xa8, 0x5c, 0x0b, 0xde, 0x6a, 0x37,\n\t\t0x92, 0xac, 0xdf, 0x9f, 0xc7, 0x51, 0x79, 0x88, 0xa1, 0xf2, 0x2a, 0x8c, 0xb7, 0xa9, 0x53, 0xd9,\n\t\t0xc2, 0xf5, 0x16, 0xfc, 0x1c, 0x85, 0x3f, 0x1f, 0xaa, 0x38, 0x25, 0x4d, 0x76, 0x58, 0xd8, 0x10,\n\t\t0x8e, 0x36, 0x78, 0xb7, 0x3b, 0xca, 0xb5, 0x47, 0xba, 0x29, 0xd7, 0x76, 0xb4, 0xee, 0x0e, 0x72,\n\t\t0x5a, 0x77, 0x5b, 0x07, 0xb1, 0xa1, 0xe4, 0xb9, 0x96, 0xfc, 0x01, 0x72, 0x2d, 0xc3, 0x07, 0xcb,\n\t\t0xb5, 0x5c, 0x87, 0x9c, 0x86, 0x2b, 0xca, 0xbe, 0xab, 0x9a, 0xd1, 0x2d, 0xc6, 0x40, 0xa9, 0xa9,\n\t\t0x2a, 0xa2, 0xe7, 0x61, 0xe0, 0xe3, 0x3a, 0x21, 0xde, 0xdf, 0x87, 0x34, 0x9b, 0x8b, 0x85, 0xcc,\n\t\t0x39, 0x97, 0xbc, 0xc9, 0xed, 0xf6, 0xe0, 0x5a, 0x75, 0xa3, 0xac, 0x10, 0xd6, 0x5e, 0x1c, 0xd6,\n\t\t0x7b, 0x0b, 0x94, 0x5e, 0xae, 0x1b, 0xcb, 0x44, 0x94, 0x23, 0x3a, 0xfa, 0x18, 0x72, 0x44, 0x63,\n\t\t0x8f, 0x26, 0x47, 0xf4, 0x46, 0x1a, 0x2e, 0x24, 0xfd, 0x40, 0xf4, 0xc3, 0x77, 0xd6, 0x1b, 0x5e,\n\t\t0xd4, 0xe5, 0xd6, 0x51, 0xaf, 0x24, 0xfe, 0xba, 0xd1, 0x17, 0x6c, 0xb5, 0xb9, 0x9d, 0x5e, 0xbf,\n\t\t0xdb, 0xe1, 0x87, 0x14, 0x19, 0x41, 0x48, 0x71, 0x48, 0x99, 0xe6, 0xc2, 0x6f, 0x52, 0xb0, 0x90,\n\t\t0xe4, 0xeb, 0x57, 0xe1, 0x7a, 0xf0, 0x63, 0x99, 0xd4, 0x41, 0x63, 0x99, 0xc3, 0x5a, 0x45, 0xfe,\n\t\t0xec, 0xf6, 0x08, 0x66, 0xb7, 0xe5, 0xeb, 0x7a, 0xe3, 0x27, 0x9d, 0x3e, 0x48, 0x41, 0xc2, 0xef,\n\t\t0x72, 0x3f, 0x1a, 0x93, 0xc9, 0x2b, 0x1a, 0xf6, 0x72, 0x8b, 0x86, 0xad, 0x6e, 0x97, 0x4c, 0xfc,\n\t\t0x6e, 0x97, 0xc2, 0x3f, 0x52, 0x70, 0xee, 0x30, 0x3c, 0xca, 0x47, 0x74, 0xd2, 0xdb, 0xea, 0x39,\n\t\t0x99, 0x04, 0xf5, 0x9c, 0xc2, 0x3f, 0x53, 0x70, 0x3e, 0xd1, 0x67, 0xd2, 0x4f, 0x26, 0xbe, 0x63,\n\t\t0xe2, 0xbd, 0x04, 0x6d, 0x26, 0x49, 0x52, 0xff, 0xd3, 0x69, 0xd1, 0xc4, 0x8b, 0x3a, 0x94, 0x9e,\n\t\t0x4c, 0x7c, 0x68, 0x83, 0x54, 0xa6, 0x9b, 0xef, 0x32, 0x7e, 0x91, 0x82, 0xc5, 0x84, 0x9f, 0xaf,\n\t\t0x3f, 0x59, 0x07, 0xdf, 0x3a, 0xcc, 0x13, 0x18, 0xa2, 0x3f, 0xd7, 0xf5, 0x0a, 0xc1, 0x16, 0x7d,\n\t\t0xd5, 0x71, 0x98, 0x58, 0xbb, 0xbf, 0x76, 0x6b, 0xb3, 0xbc, 0x5e, 0xda, 0xd8, 0x5c, 0x93, 0xcb,\n\t\t0x9b, 0xff, 0x77, 0x67, 0xad, 0x5c, 0xba, 0x75, 0x7f, 0x79, 0xa3, 0x74, 0x33, 0xff, 0x14, 0x9a,\n\t\t0x81, 0xa9, 0xce, 0xc7, 0xcb, 0x1b, 0x1b, 0x65, 0x7a, 0x37, 0x2f, 0xa1, 0x93, 0x70, 0xbc, 0x93,\n\t\t0x60, 0x75, 0xe3, 0xf6, 0xdd, 0x35, 0x46, 0x92, 0x5a, 0x79, 0x15, 0x8e, 0xa9, 0x66, 0x95, 0x37,\n\t\t0x07, 0x2b, 0xde, 0x1f, 0x20, 0xdf, 0x71, 0x22, 0xf3, 0x3b, 0xd2, 0xff, 0x5f, 0x7c, 0xa0, 0x93,\n\t\t0x9d, 0xfa, 0x56, 0x51, 0x35, 0xab, 0x8b, 0xed, 0x7f, 0xc4, 0x7c, 0x5e, 0xd7, 0x2a, 0x8b, 0x0f,\n\t\t0x4c, 0xf7, 0xcf, 0x9f, 0xd9, 0xbf, 0x32, 0xdf, 0x50, 0x6a, 0xfa, 0xde, 0xc5, 0xad, 0x0c, 0xbd,\n\t\t0x77, 0xe9, 0x5f, 0x01, 0x00, 0x00, 0xff, 0xff, 0xf3, 0xc4, 0x77, 0x00, 0x78, 0x5a, 0x00, 0x00,\n\t},\n\t// google/protobuf/duration.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x4f, 0x29, 0x2d, 0x4a,\n\t\t0x2c, 0xc9, 0xcc, 0xcf, 0xd3, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0x56,\n\t\t0x5c, 0x1c, 0x2e, 0x50, 0x25, 0x42, 0x12, 0x5c, 0xec, 0xc5, 0xa9, 0xc9, 0xf9, 0x79, 0x29, 0xc5,\n\t\t0x12, 0x8c, 0x0a, 0x8c, 0x1a, 0xcc, 0x41, 0x30, 0xae, 0x90, 0x08, 0x17, 0x6b, 0x5e, 0x62, 0x5e,\n\t\t0x7e, 0xb1, 0x04, 0x93, 0x02, 0xa3, 0x06, 0x6b, 0x10, 0x84, 0xe3, 0xd4, 0xcc, 0xc8, 0x25, 0x9c,\n\t\t0x9c, 0x9f, 0xab, 0x87, 0x66, 0xa6, 0x13, 0x2f, 0xcc, 0xc4, 0x00, 0x90, 0x48, 0x00, 0x63, 0x94,\n\t\t0x21, 0x54, 0x45, 0x7a, 0x7e, 0x4e, 0x62, 0x5e, 0xba, 0x5e, 0x7e, 0x51, 0x3a, 0xc2, 0x81, 0x25,\n\t\t0x95, 0x05, 0xa9, 0xc5, 0xfa, 0xd9, 0x79, 0xf9, 0xe5, 0x79, 0x70, 0xc7, 0x16, 0x24, 0xfd, 0x60,\n\t\t0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce, 0x1d, 0xa2, 0x39, 0x00, 0xaa,\n\t\t0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4, 0x35, 0x89, 0x0d, 0x6c, 0x94,\n\t\t0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0xef, 0x8a, 0xb4, 0xc3, 0xfb, 0x00, 0x00, 0x00,\n\t},\n\t// google/protobuf/timestamp.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4f, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0xc9, 0xcc, 0x4d,\n\t\t0x2d, 0x2e, 0x49, 0xcc, 0x2d, 0xd0, 0x03, 0x0b, 0x09, 0xf1, 0x43, 0x14, 0xe8, 0xc1, 0x14, 0x28,\n\t\t0x59, 0x73, 0x71, 0x86, 0xc0, 0xd4, 0x08, 0x49, 0x70, 0xb1, 0x17, 0xa7, 0x26, 0xe7, 0xe7, 0xa5,\n\t\t0x14, 0x4b, 0x30, 0x2a, 0x30, 0x6a, 0x30, 0x07, 0xc1, 0xb8, 0x42, 0x22, 0x5c, 0xac, 0x79, 0x89,\n\t\t0x79, 0xf9, 0xc5, 0x12, 0x4c, 0x0a, 0x8c, 0x1a, 0xac, 0x41, 0x10, 0x8e, 0x53, 0x2b, 0x23, 0x97,\n\t\t0x70, 0x72, 0x7e, 0xae, 0x1e, 0x9a, 0xa1, 0x4e, 0x7c, 0x70, 0x23, 0x03, 0x40, 0x42, 0x01, 0x8c,\n\t\t0x51, 0x46, 0x50, 0x25, 0xe9, 0xf9, 0x39, 0x89, 0x79, 0xe9, 0x7a, 0xf9, 0x45, 0xe9, 0x48, 0x6e,\n\t\t0xac, 0x2c, 0x48, 0x2d, 0xd6, 0xcf, 0xce, 0xcb, 0x2f, 0xcf, 0x43, 0xb8, 0xb7, 0x20, 0xe9, 0x07,\n\t\t0x23, 0xe3, 0x22, 0x26, 0x66, 0xf7, 0x00, 0xa7, 0x55, 0x4c, 0x72, 0xee, 0x10, 0xdd, 0x01, 0x50,\n\t\t0x2d, 0x7a, 0xe1, 0xa9, 0x39, 0x39, 0xde, 0x20, 0x0d, 0x21, 0x20, 0xbd, 0x49, 0x6c, 0x60, 0xb3,\n\t\t0x8c, 0x01, 0x01, 0x00, 0x00, 0xff, 0xff, 0xae, 0x65, 0xce, 0x7d, 0xff, 0x00, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/common.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x57, 0xdd, 0x72, 0x22, 0xc7,\n\t\t0x15, 0xf6, 0xc0, 0xa2, 0x9f, 0x03, 0xbb, 0x42, 0xad, 0xfd, 0x61, 0xb5, 0x5e, 0xaf, 0x16, 0x97,\n\t\t0x63, 0x79, 0x2b, 0x86, 0x88, 0x4d, 0x52, 0x2e, 0x3b, 0x4e, 0x82, 0xd0, 0x48, 0x9a, 0x5d, 0x02,\n\t\t0xa4, 0x99, 0x95, 0xac, 0xa4, 0xca, 0x53, 0xcd, 0x4c, 0x83, 0x3b, 0x0c, 0xd3, 0x93, 0x99, 0x1e,\n\t\t0x56, 0xf8, 0x22, 0x95, 0xcb, 0xe4, 0x26, 0x8f, 0x90, 0x8b, 0xbc, 0x48, 0x1e, 0x20, 0x97, 0x79,\n\t\t0x97, 0x5c, 0xa7, 0xba, 0xa7, 0x07, 0x81, 0xc2, 0x1a, 0x5f, 0xa4, 0x7c, 0x47, 0x9f, 0xf3, 0x7d,\n\t\t0xa7, 0xbf, 0xd3, 0xdd, 0xe7, 0x1c, 0x06, 0x0e, 0x92, 0x01, 0x8d, 0xea, 0x2e, 0xf1, 0x68, 0xe0,\n\t\t0xd2, 0x3a, 0x09, 0x59, 0x7d, 0x7a, 0x54, 0x77, 0xf9, 0x64, 0xc2, 0x83, 0x5a, 0x18, 0x71, 0xc1,\n\t\t0xd1, 0x9e, 0x44, 0xd4, 0x34, 0xa2, 0x46, 0x42, 0x56, 0x9b, 0x1e, 0xed, 0x7f, 0x30, 0xe2, 0x7c,\n\t\t0xe4, 0xd3, 0xba, 0x82, 0x0c, 0x92, 0x61, 0xdd, 0x4b, 0x22, 0x22, 0x58, 0x46, 0xaa, 0xbe, 0x86,\n\t\t0xdd, 0x4b, 0x1e, 0x8d, 0x87, 0x3e, 0x7f, 0x6b, 0x5e, 0x53, 0x37, 0x91, 0x2e, 0xf4, 0x0c, 0x8a,\n\t\t0x6f, 0xb5, 0xd1, 0x61, 0x5e, 0xc5, 0x38, 0x30, 0x0e, 0xb7, 0x31, 0x64, 0x26, 0xcb, 0x43, 0x0f,\n\t\t0x60, 0x23, 0x4a, 0x02, 0xe9, 0xcb, 0x29, 0x5f, 0x21, 0x4a, 0x02, 0xcb, 0xab, 0x56, 0xa1, 0x94,\n\t\t0x05, 0xb3, 0x67, 0x21, 0x45, 0x08, 0xee, 0x04, 0x64, 0x42, 0x75, 0x00, 0xf5, 0x5b, 0x62, 0x9a,\n\t\t0xae, 0x60, 0x53, 0x26, 0x66, 0xef, 0xc4, 0x3c, 0x85, 0xcd, 0x1e, 0x99, 0xf9, 0x9c, 0x78, 0xd2,\n\t\t0xed, 0x11, 0x41, 0x94, 0xbb, 0x84, 0xd5, 0xef, 0xea, 0x17, 0xb0, 0x79, 0x4a, 0x98, 0x9f, 0x44,\n\t\t0x14, 0x3d, 0x84, 0x8d, 0x88, 0x92, 0x98, 0x07, 0x9a, 0xaf, 0x57, 0xa8, 0x02, 0x9b, 0x1e, 0x15,\n\t\t0x84, 0xf9, 0xb1, 0x52, 0x58, 0xc2, 0xd9, 0xb2, 0xfa, 0x77, 0x03, 0xee, 0xfc, 0x86, 0x4e, 0x38,\n\t\t0xfa, 0x12, 0x36, 0x86, 0x8c, 0xfa, 0x5e, 0x5c, 0x31, 0x0e, 0xf2, 0x87, 0xc5, 0xc6, 0x47, 0xb5,\n\t\t0x15, 0xe7, 0x57, 0x93, 0xd0, 0xda, 0xa9, 0xc2, 0x99, 0x81, 0x88, 0x66, 0x58, 0x93, 0xf6, 0x2f,\n\t\t0xa1, 0xb8, 0x60, 0x46, 0x65, 0xc8, 0x8f, 0xe9, 0x4c, 0xab, 0x90, 0x3f, 0x51, 0x03, 0x0a, 0x53,\n\t\t0xe2, 0x27, 0x54, 0x09, 0x28, 0x36, 0xde, 0x5f, 0x19, 0x5e, 0xa7, 0x89, 0x53, 0xe8, 0xe7, 0xb9,\n\t\t0xcf, 0x8c, 0xea, 0x3f, 0x0c, 0xd8, 0x38, 0xa7, 0xc4, 0xa3, 0x11, 0xfa, 0xd5, 0x2d, 0x89, 0x1f,\n\t\t0xaf, 0x8c, 0x91, 0x82, 0x7f, 0x58, 0x91, 0xff, 0x36, 0xa0, 0xdc, 0xa7, 0x24, 0x72, 0xbf, 0x69,\n\t\t0x0a, 0x11, 0xb1, 0x41, 0x22, 0x68, 0x8c, 0x1c, 0xb8, 0xc7, 0x02, 0x8f, 0x5e, 0x53, 0xcf, 0x59,\n\t\t0x92, 0xfd, 0xd9, 0xca, 0xa8, 0xb7, 0xe9, 0x35, 0x2b, 0xe5, 0x2e, 0xe6, 0x71, 0x97, 0x2d, 0xda,\n\t\t0xf6, 0xbf, 0x06, 0xf4, 0xbf, 0xa0, 0xff, 0x63, 0x56, 0x43, 0xd8, 0x3a, 0x21, 0x82, 0x1c, 0xfb,\n\t\t0x7c, 0x80, 0x4e, 0xe1, 0x2e, 0x0d, 0x5c, 0xee, 0xb1, 0x60, 0xe4, 0x88, 0x59, 0x98, 0x3e, 0xd0,\n\t\t0x7b, 0x8d, 0xe7, 0x2b, 0x63, 0x99, 0x1a, 0x29, 0x5f, 0x34, 0x2e, 0xd1, 0x85, 0xd5, 0xfc, 0x01,\n\t\t0xe7, 0x16, 0x1e, 0x70, 0x2f, 0x2d, 0x3a, 0x1a, 0x5d, 0xd0, 0x28, 0x66, 0x3c, 0xb0, 0x82, 0x21,\n\t\t0x97, 0x40, 0x36, 0x09, 0xfd, 0xac, 0x10, 0xe4, 0x6f, 0xf4, 0x31, 0xec, 0x0c, 0x29, 0x11, 0x49,\n\t\t0x44, 0x9d, 0x69, 0x0a, 0xd5, 0x05, 0x77, 0x4f, 0x9b, 0x75, 0x80, 0xea, 0x6b, 0x78, 0xd4, 0x4f,\n\t\t0xc2, 0x90, 0x47, 0x82, 0x7a, 0x2d, 0x9f, 0xd1, 0x40, 0x68, 0x4f, 0x2c, 0x6b, 0x75, 0xc4, 0x9d,\n\t\t0xd8, 0x1b, 0xeb, 0xc8, 0x85, 0x11, 0xef, 0x7b, 0x63, 0xf4, 0x18, 0xb6, 0xfe, 0x40, 0xa6, 0x44,\n\t\t0x39, 0xd2, 0x98, 0x9b, 0x72, 0xdd, 0xf7, 0xc6, 0xd5, 0x3f, 0xe7, 0xa1, 0x88, 0xa9, 0x88, 0x66,\n\t\t0x3d, 0xee, 0x33, 0x77, 0x86, 0x4e, 0xa0, 0xcc, 0x02, 0x26, 0x18, 0xf1, 0x1d, 0x16, 0x08, 0x1a,\n\t\t0x4d, 0x49, 0xaa, 0xb2, 0xd8, 0x78, 0x5c, 0x4b, 0xdb, 0x4b, 0x2d, 0x6b, 0x2f, 0xb5, 0x13, 0xdd,\n\t\t0x5e, 0xf0, 0x8e, 0xa6, 0x58, 0x9a, 0x81, 0xea, 0xb0, 0x37, 0x20, 0xee, 0x98, 0x0f, 0x87, 0x8e,\n\t\t0xcb, 0xe9, 0x70, 0xc8, 0x5c, 0x29, 0x53, 0xed, 0x6d, 0x60, 0xa4, 0x5d, 0xad, 0x1b, 0x8f, 0xdc,\n\t\t0x76, 0x42, 0xae, 0xd9, 0x24, 0x99, 0xdc, 0x6c, 0x9b, 0x5f, 0xbb, 0xad, 0xa6, 0xcc, 0xb7, 0xfd,\n\t\t0xe4, 0x26, 0x0a, 0x11, 0x82, 0x4e, 0x42, 0x11, 0x57, 0xee, 0x1c, 0x18, 0x87, 0x85, 0x39, 0xb4,\n\t\t0xa9, 0xcd, 0xe8, 0x4b, 0x78, 0x12, 0xf0, 0xc0, 0x89, 0x64, 0xea, 0x64, 0xe0, 0x53, 0x87, 0x46,\n\t\t0x11, 0x8f, 0x9c, 0xb4, 0xa5, 0xc4, 0x95, 0xc2, 0x41, 0xfe, 0x70, 0x1b, 0x57, 0x02, 0x1e, 0xe0,\n\t\t0x0c, 0x61, 0x4a, 0x00, 0x4e, 0xfd, 0xe8, 0x15, 0xec, 0xd1, 0xeb, 0x90, 0xa5, 0x42, 0x6e, 0x24,\n\t\t0x6f, 0xac, 0x93, 0x8c, 0x6e, 0x58, 0x99, 0xea, 0xea, 0x04, 0x1e, 0x59, 0x31, 0xf7, 0x95, 0xf1,\n\t\t0x2c, 0xe2, 0x49, 0xd8, 0x23, 0x91, 0x60, 0xaa, 0x39, 0xaf, 0x68, 0x98, 0xe8, 0x97, 0x50, 0x88,\n\t\t0x05, 0x11, 0xe9, 0x83, 0xbf, 0xd7, 0x38, 0x5c, 0xf9, 0x48, 0x97, 0x03, 0xf6, 0x25, 0x1e, 0xa7,\n\t\t0xb4, 0xea, 0x14, 0x9e, 0x2c, 0x7b, 0x5b, 0x3c, 0x18, 0xb2, 0x91, 0x56, 0x88, 0x2e, 0xa1, 0xcc,\n\t\t0x32, 0xb7, 0x33, 0x92, 0xfe, 0xac, 0xb4, 0x7f, 0xfc, 0x3d, 0x76, 0x9a, 0x4b, 0xc7, 0x3b, 0x6c,\n\t\t0xc9, 0x11, 0x57, 0xff, 0x65, 0xc0, 0x7e, 0x33, 0x9e, 0x05, 0x6e, 0x36, 0x36, 0x96, 0xf7, 0xad,\n\t\t0xc0, 0x26, 0x0d, 0xe4, 0x39, 0xa7, 0x33, 0x68, 0x0b, 0x67, 0x4b, 0xd4, 0x80, 0x07, 0x61, 0x44,\n\t\t0x3d, 0x3a, 0x64, 0x01, 0xf5, 0x9c, 0x3f, 0x26, 0x34, 0xa1, 0x8e, 0x3a, 0x95, 0xf4, 0x29, 0xef,\n\t\t0xdd, 0x38, 0x7f, 0x2b, 0x7d, 0x1d, 0x79, 0x48, 0x4f, 0x01, 0x52, 0xa0, 0x2a, 0xe7, 0xbc, 0x02,\n\t\t0x6e, 0x2b, 0x8b, 0x2a, 0xd4, 0x5f, 0x43, 0x29, 0x75, 0xbb, 0x4a, 0x83, 0x7a, 0x24, 0xc5, 0xc6,\n\t\t0xd3, 0x95, 0x09, 0x66, 0x5d, 0x02, 0x17, 0x15, 0x25, 0x55, 0x5d, 0xfd, 0x4f, 0x1e, 0xde, 0x57,\n\t\t0xb3, 0x8d, 0xb6, 0xfc, 0x24, 0x16, 0x34, 0xea, 0x53, 0x9f, 0xba, 0x32, 0x13, 0x5d, 0x48, 0x7d,\n\t\t0xd8, 0x8a, 0x45, 0x44, 0x04, 0x1d, 0xcd, 0x74, 0x3b, 0x79, 0xb9, 0x32, 0xfc, 0xea, 0x20, 0x7d,\n\t\t0x4d, 0x3d, 0xce, 0x55, 0x0c, 0x3c, 0x0f, 0x84, 0xfe, 0x62, 0xc0, 0x87, 0x44, 0x11, 0x1c, 0x37,\n\t\t0x65, 0x38, 0xb1, 0x60, 0xee, 0x78, 0xe6, 0x44, 0x74, 0x24, 0x2f, 0x4c, 0xe7, 0x93, 0xf6, 0xc2,\n\t\t0x9f, 0x7e, 0x8f, 0x0d, 0x15, 0x1b, 0x2b, 0x72, 0x9a, 0x99, 0xdc, 0xf1, 0xfc, 0x3d, 0xfc, 0x8c,\n\t\t0x7c, 0x37, 0x0c, 0xfd, 0xcd, 0x80, 0x8f, 0x6e, 0x49, 0xa1, 0xd7, 0x82, 0x46, 0x01, 0xf1, 0x1d,\n\t\t0x1a, 0x08, 0x26, 0x66, 0x99, 0x98, 0xb4, 0x8e, 0x7f, 0xbe, 0x5e, 0x8c, 0xa9, 0xf9, 0xa6, 0xa2,\n\t\t0x2f, 0xc9, 0x79, 0x4e, 0xd6, 0x01, 0x11, 0x86, 0xdd, 0x4c, 0x08, 0xc9, 0x06, 0x8d, 0xbe, 0xd8,\n\t\t0xd5, 0xe3, 0x5e, 0x07, 0x9b, 0x4f, 0x25, 0x5c, 0x76, 0x6f, 0x59, 0x8e, 0x77, 0x61, 0x27, 0x3b,\n\t\t0x7b, 0x9d, 0x4d, 0xf5, 0x17, 0x50, 0xbe, 0x4d, 0x44, 0xf7, 0xa1, 0x10, 0xbb, 0x3c, 0xcc, 0xea,\n\t\t0x34, 0x5d, 0xcc, 0x8b, 0x37, 0xb7, 0xf0, 0x6f, 0xe7, 0x15, 0x3c, 0x5b, 0x73, 0xfe, 0xe8, 0x43,\n\t\t0xb8, 0xbb, 0x74, 0xa7, 0x3a, 0x68, 0x29, 0x5e, 0x80, 0x7e, 0x9e, 0xab, 0x18, 0xd5, 0xbf, 0x1a,\n\t\t0xf0, 0x7c, 0xed, 0xf9, 0xa1, 0x9f, 0xc0, 0xfd, 0xdb, 0xf7, 0x32, 0x1f, 0x71, 0xdb, 0xb2, 0x1f,\n\t\t0x2d, 0x72, 0x54, 0x71, 0xd4, 0x64, 0x6f, 0x5b, 0x66, 0xc8, 0x99, 0x9b, 0xa6, 0xb1, 0xbb, 0x4c,\n\t\t0x78, 0x4d, 0x67, 0x4a, 0xcb, 0x57, 0xb0, 0xdb, 0x23, 0x23, 0x16, 0xa8, 0x5a, 0xee, 0x86, 0x42,\n\t\t0x4d, 0xa3, 0x27, 0xb0, 0x1d, 0x92, 0x11, 0x75, 0x62, 0xf6, 0x6d, 0xba, 0x5f, 0x01, 0x6f, 0x49,\n\t\t0x43, 0x9f, 0x7d, 0x4b, 0xd1, 0x8f, 0x60, 0x27, 0xa0, 0xd7, 0xc2, 0x51, 0x08, 0xc1, 0xc7, 0x34,\n\t\t0xd0, 0x63, 0xf3, 0xae, 0x34, 0xf7, 0xc8, 0x88, 0xda, 0xd2, 0xf8, 0xe2, 0x2d, 0x94, 0x16, 0x27,\n\t\t0x2e, 0x7a, 0x0c, 0x0f, 0xcc, 0x4e, 0xab, 0x7b, 0x62, 0x75, 0xce, 0x1c, 0xfb, 0xaa, 0x67, 0x3a,\n\t\t0x56, 0xe7, 0xa2, 0xd9, 0xb6, 0x4e, 0xca, 0xef, 0xa1, 0x7d, 0x78, 0xb8, 0xec, 0xb2, 0xcf, 0xb1,\n\t\t0x75, 0x6a, 0xe3, 0xcb, 0xb2, 0x81, 0x1e, 0x02, 0x5a, 0xf6, 0xbd, 0xea, 0x77, 0x3b, 0xe5, 0x1c,\n\t\t0xaa, 0xc0, 0xfd, 0x65, 0x7b, 0x0f, 0x77, 0xed, 0xee, 0xcb, 0x72, 0xfe, 0xc5, 0x9f, 0x60, 0x6f,\n\t\t0x45, 0x17, 0x45, 0xcf, 0xe1, 0xa9, 0xd5, 0xef, 0xb6, 0x9b, 0xb6, 0xd5, 0xed, 0x38, 0x67, 0xb8,\n\t\t0xfb, 0xa6, 0xe7, 0xf4, 0xed, 0xa6, 0xbd, 0xa8, 0xe3, 0x9d, 0x90, 0x73, 0xb3, 0xd9, 0xb6, 0xcf,\n\t\t0xaf, 0xca, 0xc6, 0xbb, 0x21, 0x27, 0xb8, 0x69, 0x75, 0xcc, 0x93, 0x72, 0xee, 0xc5, 0x3f, 0x0d,\n\t\t0xf8, 0xe0, 0xbb, 0x9b, 0x03, 0xfa, 0x14, 0x3e, 0x69, 0xb6, 0x6c, 0xeb, 0xc2, 0x74, 0x5a, 0xed,\n\t\t0x37, 0x7d, 0xdb, 0xc4, 0x4e, 0xdf, 0x6c, 0x9b, 0x2d, 0x15, 0xb4, 0x6f, 0xe3, 0xa6, 0x6d, 0x9e,\n\t\t0x5d, 0x2d, 0xe8, 0x7a, 0x09, 0xf5, 0xf5, 0x70, 0x6c, 0x9e, 0xa5, 0x6b, 0xab, 0xf5, 0x5a, 0x2a,\n\t\t0xfd, 0x19, 0x1c, 0xad, 0x27, 0x99, 0x5f, 0xd9, 0x26, 0xee, 0x34, 0xdb, 0x8e, 0xd9, 0xb1, 0x2d,\n\t\t0xfb, 0xaa, 0x9c, 0xdb, 0xcf, 0x55, 0x8c, 0x17, 0x5f, 0x43, 0x49, 0xfe, 0x77, 0xe7, 0x53, 0x1a,\n\t\t0x65, 0x57, 0x77, 0xda, 0xb4, 0xda, 0xdd, 0x0b, 0x13, 0xdf, 0xbe, 0xba, 0x47, 0xb0, 0xb7, 0xec,\n\t\t0x3a, 0xed, 0xe2, 0x96, 0x59, 0x36, 0xe4, 0x9d, 0x2e, 0x3b, 0xce, 0x70, 0xb3, 0x65, 0x9e, 0xbe,\n\t\t0x69, 0x97, 0x73, 0xc7, 0xbf, 0x87, 0x47, 0x2e, 0x9f, 0xac, 0xaa, 0xed, 0xe3, 0x62, 0x4b, 0x7d,\n\t\t0x2d, 0xf5, 0xe4, 0x00, 0xee, 0x19, 0xbf, 0x3b, 0x1a, 0x31, 0xf1, 0x4d, 0x32, 0xa8, 0xb9, 0x7c,\n\t\t0x52, 0x5f, 0xfc, 0xb6, 0xfa, 0x94, 0x79, 0x7e, 0x7d, 0xc4, 0xd3, 0x2f, 0x26, 0xfd, 0xa1, 0xf5,\n\t\t0x05, 0x09, 0xd9, 0xf4, 0x68, 0xb0, 0xa1, 0x6c, 0x2f, 0xff, 0x1b, 0x00, 0x00, 0xff, 0xff, 0x5e,\n\t\t0xe2, 0xc9, 0x06, 0x8c, 0x0d, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/tasklist.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xb4, 0x56, 0xdb, 0x6e, 0xdb, 0x36,\n\t\t0x18, 0x9e, 0x7c, 0x68, 0x9d, 0xdf, 0x4d, 0xa2, 0xb2, 0x49, 0x63, 0xbb, 0xed, 0xe6, 0xfa, 0xa2,\n\t\t0xc8, 0x8a, 0x4d, 0x46, 0xb2, 0x0d, 0x18, 0xb6, 0xa1, 0xab, 0x13, 0x1b, 0xad, 0x10, 0x27, 0x35,\n\t\t0x64, 0xb5, 0x43, 0x07, 0x0c, 0x02, 0x2d, 0xb1, 0x0e, 0x67, 0x49, 0x14, 0x44, 0xca, 0xae, 0x6f,\n\t\t0xf6, 0x18, 0xbb, 0xdb, 0x8b, 0xec, 0x1d, 0xf6, 0x4e, 0x03, 0x29, 0xd9, 0xf5, 0x41, 0x0d, 0xd6,\n\t\t0x8b, 0xdd, 0x99, 0xff, 0xc7, 0x8f, 0xdf, 0x7f, 0xb6, 0xa0, 0x95, 0x8c, 0x48, 0xdc, 0x76, 0xb1,\n\t\t0x47, 0x42, 0x97, 0xb4, 0x71, 0x44, 0xdb, 0xd3, 0x93, 0xb6, 0xc0, 0x7c, 0xe2, 0x53, 0x2e, 0x8c,\n\t\t0x28, 0x66, 0x82, 0xa1, 0x7b, 0xf2, 0x8e, 0x91, 0xdd, 0x31, 0x70, 0x44, 0x8d, 0xe9, 0x49, 0xe3,\n\t\t0xf3, 0x31, 0x63, 0x63, 0x9f, 0xb4, 0xd5, 0x95, 0x51, 0xf2, 0xae, 0xed, 0x25, 0x31, 0x16, 0x94,\n\t\t0x85, 0x29, 0xa9, 0xf1, 0xc5, 0x26, 0x2e, 0x68, 0x40, 0xb8, 0xc0, 0x41, 0x94, 0x5d, 0xd8, 0x7a,\n\t\t0x60, 0x16, 0xe3, 0x28, 0x22, 0x31, 0x4f, 0xf1, 0xd6, 0x6b, 0xa8, 0xd8, 0x98, 0x4f, 0xfa, 0x94,\n\t\t0x0b, 0x84, 0xa0, 0x14, 0xe2, 0x80, 0xd4, 0xb4, 0xa6, 0x76, 0xbc, 0x63, 0xa9, 0xdf, 0xe8, 0x3b,\n\t\t0x28, 0x4d, 0x68, 0xe8, 0xd5, 0x0a, 0x4d, 0xed, 0x78, 0xef, 0xf4, 0xb1, 0x91, 0xe3, 0xa4, 0xb1,\n\t\t0x78, 0xe0, 0x82, 0x86, 0x9e, 0xa5, 0xae, 0xb7, 0x30, 0xe8, 0x0b, 0xeb, 0x25, 0x11, 0xd8, 0xc3,\n\t\t0x02, 0xa3, 0x4b, 0x38, 0x08, 0xf0, 0x7b, 0x47, 0x86, 0xcd, 0x9d, 0x88, 0xc4, 0x0e, 0x27, 0x2e,\n\t\t0x0b, 0x3d, 0x25, 0x57, 0x3d, 0x7d, 0x68, 0xa4, 0x9e, 0x1a, 0x0b, 0x4f, 0x8d, 0x2e, 0x4b, 0x46,\n\t\t0x3e, 0x79, 0x83, 0xfd, 0x84, 0x58, 0x77, 0x03, 0xfc, 0x5e, 0x3e, 0xc8, 0x07, 0x24, 0x1e, 0x2a,\n\t\t0x5a, 0xeb, 0x35, 0xd4, 0x17, 0x12, 0x03, 0x1c, 0x0b, 0x2a, 0xb3, 0xb2, 0xd4, 0xd2, 0xa1, 0x38,\n\t\t0x21, 0xf3, 0x2c, 0x12, 0xf9, 0x13, 0x3d, 0x81, 0x7d, 0x36, 0x0b, 0x49, 0xec, 0x5c, 0x33, 0x2e,\n\t\t0x1c, 0x15, 0x67, 0x41, 0xa1, 0xbb, 0xca, 0xfc, 0x92, 0x71, 0x71, 0x85, 0x03, 0xd2, 0x9a, 0xc0,\n\t\t0xa1, 0xc9, 0x99, 0xaf, 0x92, 0xfc, 0x22, 0x66, 0x49, 0x74, 0x49, 0x44, 0x4c, 0x5d, 0x8e, 0xda,\n\t\t0x70, 0x10, 0x92, 0x59, 0xbe, 0xfb, 0x9a, 0x75, 0x37, 0x24, 0xb3, 0x75, 0x07, 0xd1, 0x63, 0xb8,\n\t\t0x13, 0x31, 0xdf, 0x27, 0xb1, 0xe3, 0xb2, 0x24, 0x14, 0x4a, 0xae, 0x68, 0x55, 0x53, 0xdb, 0xb9,\n\t\t0x34, 0xb5, 0xfe, 0x2a, 0xc1, 0xde, 0x22, 0x88, 0xa1, 0xc0, 0x22, 0xe1, 0xe8, 0x2b, 0x40, 0x23,\n\t\t0xec, 0x4e, 0x7c, 0x36, 0x4e, 0x69, 0xce, 0x35, 0x0d, 0x85, 0x12, 0x29, 0x5a, 0x7a, 0x86, 0x28,\n\t\t0xf2, 0x4b, 0x1a, 0x0a, 0xf4, 0x08, 0x20, 0x26, 0xd8, 0x73, 0x7c, 0x32, 0x25, 0x7e, 0xa6, 0xb0,\n\t\t0x23, 0x2d, 0x7d, 0x69, 0x40, 0x0f, 0x60, 0x07, 0xbb, 0x93, 0x0c, 0x2d, 0x2a, 0xb4, 0x82, 0xdd,\n\t\t0x49, 0x0a, 0x3e, 0x81, 0xfd, 0x18, 0x0b, 0xb2, 0x1a, 0x4b, 0x49, 0xc5, 0xb2, 0x2b, 0xcd, 0x1f,\n\t\t0xe2, 0xe8, 0xc2, 0xae, 0x0c, 0xda, 0xa1, 0x9e, 0x33, 0xf2, 0x99, 0x3b, 0xa9, 0x95, 0x55, 0xc1,\n\t\t0x9a, 0x1f, 0xed, 0x05, 0xb3, 0x7b, 0x26, 0xef, 0x59, 0x55, 0x49, 0x33, 0x3d, 0x75, 0x40, 0x53,\n\t\t0x38, 0xa2, 0x8b, 0xbc, 0x3a, 0x63, 0x99, 0x58, 0x27, 0x48, 0x33, 0x5b, 0xbb, 0xd5, 0x2c, 0x1e,\n\t\t0x57, 0x4f, 0x9f, 0xdd, 0xd8, 0x5b, 0x69, 0x76, 0x8c, 0xdc, 0xd2, 0xf4, 0x42, 0x11, 0xcf, 0xad,\n\t\t0x43, 0xfa, 0x49, 0x65, 0xbb, 0xfd, 0xb1, 0xb2, 0x1d, 0x40, 0x99, 0x04, 0x91, 0x98, 0xd7, 0x2a,\n\t\t0x4d, 0xed, 0xb8, 0x62, 0xa5, 0x87, 0x86, 0x80, 0xc6, 0xc7, 0xb5, 0x73, 0xda, 0xed, 0x39, 0x94,\n\t\t0xa7, 0xb2, 0x73, 0x55, 0x4d, 0xaa, 0xa7, 0x4f, 0x73, 0x83, 0xcb, 0x7d, 0xd1, 0x4a, 0x89, 0x3f,\n\t\t0x14, 0xbe, 0xd7, 0x5a, 0x3f, 0x43, 0x75, 0x25, 0xa1, 0xa8, 0x0e, 0x15, 0x2e, 0x70, 0x2c, 0x1c,\n\t\t0xea, 0x65, 0x1d, 0x71, 0x5b, 0x9d, 0x4d, 0x0f, 0x1d, 0xc2, 0x2d, 0x12, 0x7a, 0x12, 0x48, 0x9b,\n\t\t0xa0, 0x4c, 0x42, 0xcf, 0xf4, 0x5a, 0x7f, 0x6a, 0x00, 0x03, 0xd5, 0x70, 0x66, 0xf8, 0x8e, 0xa1,\n\t\t0x2e, 0xe8, 0x3e, 0xe6, 0xc2, 0xc1, 0xae, 0x4b, 0x38, 0x77, 0xe4, 0xb2, 0xc8, 0xc6, 0xaf, 0xb1,\n\t\t0x35, 0x7e, 0xf6, 0x62, 0x93, 0x58, 0x7b, 0x92, 0xd3, 0x51, 0x14, 0x69, 0x44, 0x0d, 0xa8, 0x50,\n\t\t0x8f, 0x84, 0x82, 0x8a, 0x79, 0x36, 0x43, 0xcb, 0x73, 0x5e, 0x53, 0x15, 0x73, 0x9a, 0xaa, 0xf5,\n\t\t0xb7, 0x06, 0xf5, 0xa1, 0xa0, 0xee, 0x64, 0xde, 0x7b, 0x4f, 0xdc, 0x44, 0x26, 0xa1, 0x23, 0x44,\n\t\t0x4c, 0x47, 0x89, 0x20, 0x1c, 0xbd, 0x00, 0x7d, 0xc6, 0xe2, 0x09, 0x89, 0x55, 0xdd, 0x1c, 0xb9,\n\t\t0x25, 0x33, 0x3f, 0x1f, 0xdd, 0xd8, 0x25, 0xd6, 0x5e, 0x4a, 0x5b, 0xae, 0x34, 0x1b, 0xea, 0xdc,\n\t\t0xbd, 0x26, 0x5e, 0xe2, 0x13, 0x47, 0x30, 0x27, 0xcd, 0x9e, 0x0c, 0x9b, 0x25, 0x22, 0x2b, 0x4d,\n\t\t0x7d, 0x7b, 0xf1, 0x64, 0x3b, 0xd6, 0xba, 0xbf, 0xe0, 0xda, 0x6c, 0x28, 0x99, 0x76, 0x4a, 0x6c,\n\t\t0x3d, 0x83, 0xbb, 0x5b, 0xab, 0x07, 0x7d, 0x09, 0xfa, 0x46, 0x83, 0xf3, 0x9a, 0xd6, 0x2c, 0x1e,\n\t\t0xef, 0x58, 0xfb, 0xeb, 0x9d, 0xc9, 0x5b, 0xff, 0x94, 0xe0, 0x68, 0xeb, 0x81, 0x73, 0x16, 0xbe,\n\t\t0xa3, 0x63, 0x54, 0x83, 0xdb, 0x53, 0x12, 0x73, 0xca, 0xc2, 0x45, 0x89, 0xb3, 0x23, 0x3a, 0x85,\n\t\t0x7b, 0x61, 0x12, 0x38, 0x6a, 0xde, 0xa3, 0x05, 0x8b, 0xab, 0x28, 0xca, 0x67, 0x85, 0x9a, 0x6c,\n\t\t0xe6, 0x24, 0xb0, 0x08, 0xf6, 0x96, 0x4f, 0x72, 0xf4, 0x2d, 0x1c, 0x48, 0xce, 0x2c, 0xa6, 0xb2,\n\t\t0x26, 0x1f, 0x48, 0xc5, 0x25, 0x09, 0x85, 0x49, 0xf0, 0x8b, 0x84, 0x57, 0x58, 0x14, 0xf6, 0x37,\n\t\t0x55, 0x4a, 0x6a, 0x46, 0x9f, 0xdf, 0x98, 0xfd, 0x8d, 0x50, 0x8c, 0x75, 0x5f, 0xd2, 0x29, 0xdd,\n\t\t0x8b, 0xd7, 0x1d, 0xf4, 0x41, 0xdf, 0x72, 0xae, 0xac, 0xb4, 0x3a, 0x9f, 0xa4, 0xb5, 0x11, 0x42,\n\t\t0x2a, 0xb6, 0x3f, 0x5b, 0xb7, 0x36, 0x28, 0xdc, 0xcb, 0x71, 0x6a, 0x75, 0x7c, 0xcb, 0xe9, 0xf8,\n\t\t0xfe, 0xb4, 0x3e, 0xbe, 0x4f, 0xfe, 0x9b, 0x2f, 0x2b, 0xa3, 0xdb, 0xf8, 0x1d, 0x0e, 0xf2, 0x7c,\n\t\t0xfa, 0x3f, 0xb4, 0x9e, 0xfe, 0x01, 0x77, 0x56, 0xff, 0x83, 0x51, 0x03, 0xee, 0xdb, 0x9d, 0xe1,\n\t\t0x85, 0xd3, 0x37, 0x87, 0xb6, 0x73, 0x61, 0x5e, 0x75, 0x1d, 0xf3, 0xea, 0x4d, 0xa7, 0x6f, 0x76,\n\t\t0xf5, 0xcf, 0x50, 0x1d, 0x0e, 0x37, 0xb0, 0xab, 0x57, 0xd6, 0x65, 0xa7, 0xaf, 0x6b, 0x39, 0xd0,\n\t\t0xd0, 0x36, 0xcf, 0x2f, 0xde, 0xea, 0x05, 0xf4, 0x10, 0x6a, 0x1b, 0x50, 0x6f, 0xf0, 0xb2, 0x77,\n\t\t0xd9, 0xb3, 0x3a, 0x7d, 0xbd, 0xf8, 0xd4, 0xfb, 0xa0, 0x6f, 0xcf, 0x23, 0xb2, 0xae, 0x6f, 0xbf,\n\t\t0x1d, 0xf4, 0x56, 0xf4, 0x1f, 0xc0, 0xd1, 0x06, 0xd6, 0xed, 0x9d, 0x9b, 0x43, 0xf3, 0xd5, 0x95,\n\t\t0xae, 0xe5, 0x80, 0x9d, 0x73, 0xdb, 0x7c, 0x63, 0xda, 0x6f, 0xf5, 0xc2, 0xd9, 0x6f, 0x70, 0xe4,\n\t\t0xb2, 0x20, 0x2f, 0x3b, 0x67, 0xbb, 0xcb, 0xf4, 0xc8, 0x19, 0x1e, 0x68, 0xbf, 0x9e, 0x8c, 0xa9,\n\t\t0xb8, 0x4e, 0x46, 0x86, 0xcb, 0x82, 0xf6, 0xea, 0xb7, 0xd7, 0xd7, 0xd4, 0xf3, 0xdb, 0x63, 0x96,\n\t\t0x7e, 0x0e, 0x65, 0x1f, 0x62, 0x3f, 0xe2, 0x88, 0x4e, 0x4f, 0x46, 0xb7, 0x94, 0xed, 0x9b, 0x7f,\n\t\t0x03, 0x00, 0x00, 0xff, 0xff, 0xea, 0xcc, 0x39, 0x04, 0xac, 0x09, 0x00, 0x00,\n\t},\n\t// google/protobuf/wrappers.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4b, 0xcf, 0xcf, 0x4f,\n\t\t0xcf, 0x49, 0xd5, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x4f, 0x2a, 0x4d, 0xd3, 0x2f, 0x2f, 0x4a, 0x2c,\n\t\t0x28, 0x48, 0x2d, 0x2a, 0xd6, 0x03, 0x8b, 0x08, 0xf1, 0x43, 0xe4, 0xf5, 0x60, 0xf2, 0x4a, 0xca,\n\t\t0x5c, 0xdc, 0x2e, 0xf9, 0xa5, 0x49, 0x39, 0xa9, 0x61, 0x89, 0x39, 0xa5, 0xa9, 0x42, 0x22, 0x5c,\n\t\t0xac, 0x65, 0x20, 0x86, 0x04, 0xa3, 0x02, 0xa3, 0x06, 0x63, 0x10, 0x84, 0xa3, 0xa4, 0xc4, 0xc5,\n\t\t0xe5, 0x96, 0x93, 0x9f, 0x58, 0x82, 0x45, 0x0d, 0x13, 0x92, 0x1a, 0xcf, 0xbc, 0x12, 0x33, 0x13,\n\t\t0x2c, 0x6a, 0x98, 0x61, 0x6a, 0x94, 0xb9, 0xb8, 0x43, 0x71, 0x29, 0x62, 0x41, 0x35, 0xc8, 0xd8,\n\t\t0x08, 0x8b, 0x1a, 0x56, 0x34, 0x83, 0xb0, 0x2a, 0xe2, 0x85, 0x29, 0x52, 0xe4, 0xe2, 0x74, 0xca,\n\t\t0xcf, 0xcf, 0xc1, 0xa2, 0x84, 0x03, 0xc9, 0x9c, 0xe0, 0x92, 0xa2, 0xcc, 0xbc, 0x74, 0x2c, 0x8a,\n\t\t0x38, 0x91, 0x1c, 0xe4, 0x54, 0x59, 0x92, 0x5a, 0x8c, 0x45, 0x0d, 0x0f, 0x54, 0x8d, 0x53, 0x33,\n\t\t0x23, 0x97, 0x70, 0x72, 0x7e, 0xae, 0x1e, 0x5a, 0xf0, 0x3a, 0xf1, 0x86, 0x43, 0xc3, 0x3f, 0x00,\n\t\t0x24, 0x12, 0xc0, 0x18, 0x65, 0x08, 0x55, 0x91, 0x9e, 0x9f, 0x93, 0x98, 0x97, 0xae, 0x97, 0x5f,\n\t\t0x94, 0x8e, 0x88, 0xab, 0x92, 0xca, 0x82, 0xd4, 0x62, 0xfd, 0xec, 0xbc, 0xfc, 0xf2, 0x3c, 0x78,\n\t\t0xbc, 0x15, 0x24, 0xfd, 0x60, 0x64, 0x5c, 0xc4, 0xc4, 0xec, 0x1e, 0xe0, 0xb4, 0x8a, 0x49, 0xce,\n\t\t0x1d, 0xa2, 0x39, 0x00, 0xaa, 0x43, 0x2f, 0x3c, 0x35, 0x27, 0xc7, 0x1b, 0xa4, 0x3e, 0x04, 0xa4,\n\t\t0x35, 0x89, 0x0d, 0x6c, 0x94, 0x31, 0x20, 0x00, 0x00, 0xff, 0xff, 0x3c, 0x92, 0x48, 0x30, 0x06,\n\t\t0x02, 0x00, 0x00,\n\t},\n\t// uber/cadence/api/v1/workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xc4, 0x5a, 0xcd, 0x72, 0xdb, 0xc8,\n\t\t0xb5, 0xbe, 0x20, 0x25, 0x59, 0x3a, 0xa4, 0x24, 0xa8, 0x25, 0x59, 0xf4, 0xcf, 0xd8, 0xb2, 0x66,\n\t\t0xec, 0x91, 0x79, 0xc7, 0xd2, 0xc8, 0xf3, 0x3f, 0xbe, 0x73, 0x1d, 0x08, 0x84, 0x6c, 0xd8, 0x34,\n\t\t0xc8, 0x34, 0x41, 0x6b, 0x34, 0x95, 0x04, 0x05, 0x91, 0x2d, 0x09, 0x31, 0x09, 0xb0, 0x80, 0xa6,\n\t\t0x6d, 0xed, 0x53, 0x95, 0x6c, 0x93, 0xd5, 0x54, 0x56, 0x79, 0x80, 0x54, 0xa5, 0x52, 0x59, 0x64,\n\t\t0x95, 0xca, 0x13, 0x64, 0x9b, 0x57, 0x48, 0xe5, 0x2d, 0x52, 0xdd, 0x68, 0x90, 0x00, 0x09, 0x12,\n\t\t0x74, 0x52, 0x35, 0xd9, 0x09, 0xa7, 0xbf, 0xef, 0xf4, 0xe9, 0xf3, 0xd7, 0x07, 0x10, 0x61, 0xa7,\n\t\t0x7f, 0x4a, 0xfc, 0xfd, 0x96, 0xdd, 0x26, 0x6e, 0x8b, 0xec, 0xdb, 0x3d, 0x67, 0xff, 0xf5, 0xc1,\n\t\t0xfe, 0x1b, 0xcf, 0x7f, 0x75, 0xd6, 0xf1, 0xde, 0xec, 0xf5, 0x7c, 0x8f, 0x7a, 0x68, 0x9d, 0x61,\n\t\t0xf6, 0x04, 0x66, 0xcf, 0xee, 0x39, 0x7b, 0xaf, 0x0f, 0xae, 0xdf, 0x3a, 0xf7, 0xbc, 0xf3, 0x0e,\n\t\t0xd9, 0xe7, 0x90, 0xd3, 0xfe, 0xd9, 0x7e, 0xbb, 0xef, 0xdb, 0xd4, 0xf1, 0xdc, 0x90, 0x74, 0xfd,\n\t\t0xf6, 0xe8, 0x3a, 0x75, 0xba, 0x24, 0xa0, 0x76, 0xb7, 0x27, 0x00, 0xdb, 0x69, 0x3b, 0xb7, 0xbc,\n\t\t0x6e, 0x77, 0xa0, 0x22, 0xd5, 0x36, 0x6a, 0x07, 0xaf, 0x3a, 0x4e, 0x40, 0x43, 0xcc, 0xce, 0xf7,\n\t\t0x45, 0xd8, 0x3c, 0x16, 0xe6, 0x6a, 0x6f, 0x49, 0xab, 0xcf, 0x4c, 0xd0, 0xdd, 0x33, 0x0f, 0x35,\n\t\t0x01, 0x45, 0xe7, 0xb0, 0x48, 0xb4, 0x52, 0x92, 0xb6, 0xa5, 0xdd, 0xc2, 0xc3, 0x7b, 0x7b, 0x29,\n\t\t0x47, 0xda, 0x1b, 0xd3, 0x83, 0xd7, 0xde, 0x8c, 0x8a, 0xd0, 0x67, 0x30, 0x47, 0x2f, 0x7b, 0xa4,\n\t\t0x94, 0xe3, 0x8a, 0xee, 0x4c, 0x55, 0x64, 0x5e, 0xf6, 0x08, 0xe6, 0x70, 0xf4, 0x15, 0x40, 0x40,\n\t\t0x6d, 0x9f, 0x5a, 0xcc, 0x0d, 0xa5, 0x3c, 0x27, 0x5f, 0xdf, 0x0b, 0x7d, 0xb4, 0x17, 0xf9, 0x68,\n\t\t0xcf, 0x8c, 0x7c, 0x84, 0x97, 0x38, 0x9a, 0x3d, 0x33, 0x6a, 0xab, 0xe3, 0x05, 0x24, 0xa4, 0xce,\n\t\t0x65, 0x53, 0x39, 0x9a, 0x53, 0x4d, 0x28, 0x86, 0xd4, 0x80, 0xda, 0xb4, 0x1f, 0x94, 0xe6, 0xb7,\n\t\t0xa5, 0xdd, 0x95, 0x87, 0x07, 0xb3, 0x9d, 0x5e, 0x65, 0xcc, 0x06, 0x27, 0xe2, 0x42, 0x6b, 0xf8,\n\t\t0x80, 0xee, 0xc2, 0xca, 0x85, 0x13, 0x50, 0xcf, 0xbf, 0xb4, 0x3a, 0xc4, 0x3d, 0xa7, 0x17, 0xa5,\n\t\t0x85, 0x6d, 0x69, 0x37, 0x8f, 0x97, 0x85, 0xb4, 0xca, 0x85, 0xe8, 0x27, 0xb0, 0xd9, 0xb3, 0x7d,\n\t\t0xe2, 0xd2, 0xa1, 0xfb, 0x2d, 0xc7, 0x3d, 0xf3, 0x4a, 0x57, 0xf8, 0x11, 0x76, 0x53, 0xad, 0xa8,\n\t\t0x73, 0x46, 0x22, 0x92, 0x78, 0xbd, 0x37, 0x2e, 0x44, 0x0a, 0xac, 0x0c, 0xd5, 0x72, 0xcf, 0x2c,\n\t\t0x66, 0x7a, 0x66, 0x79, 0xc0, 0xe0, 0xde, 0x79, 0x00, 0x73, 0x5d, 0xd2, 0xf5, 0x4a, 0x4b, 0x9c,\n\t\t0x78, 0x2d, 0xd5, 0x9e, 0x17, 0xa4, 0xeb, 0x61, 0x0e, 0x43, 0x18, 0xd6, 0x02, 0x62, 0xfb, 0xad,\n\t\t0x0b, 0xcb, 0xa6, 0xd4, 0x77, 0x4e, 0xfb, 0x94, 0x04, 0x25, 0xe0, 0xdc, 0xbb, 0xa9, 0xdc, 0x06,\n\t\t0x47, 0x2b, 0x03, 0x30, 0x96, 0x83, 0x11, 0x09, 0xaa, 0xc2, 0x9a, 0xdd, 0xa7, 0x9e, 0xe5, 0x93,\n\t\t0x80, 0x50, 0xab, 0xe7, 0x39, 0x2e, 0x0d, 0x4a, 0x05, 0xae, 0x73, 0x3b, 0x55, 0x27, 0x66, 0xc0,\n\t\t0x3a, 0xc7, 0xe1, 0x55, 0x46, 0x8d, 0x09, 0xd0, 0x0d, 0x58, 0x62, 0xe5, 0x61, 0xb1, 0xfa, 0x28,\n\t\t0x15, 0xb7, 0xa5, 0xdd, 0x25, 0xbc, 0xc8, 0x04, 0x55, 0x27, 0xa0, 0x48, 0x85, 0x95, 0xc1, 0x62,\n\t\t0x18, 0x87, 0x35, 0xbe, 0xcf, 0x7b, 0xa9, 0xfb, 0x98, 0x82, 0x86, 0x8b, 0x91, 0x02, 0xee, 0xf5,\n\t\t0x2d, 0xb8, 0xe2, 0x04, 0x56, 0xcb, 0xf7, 0xdc, 0xd2, 0xf2, 0xb6, 0xb4, 0xbb, 0x88, 0x17, 0x9c,\n\t\t0x40, 0xf5, 0x3d, 0x17, 0x3d, 0x82, 0x42, 0xbf, 0xd7, 0xb6, 0xa9, 0xc8, 0xd2, 0x95, 0xcc, 0x58,\n\t\t0x40, 0x08, 0xe7, 0x81, 0xf8, 0x39, 0xc8, 0x3d, 0xdb, 0xa7, 0x0e, 0x8f, 0x65, 0xcb, 0x73, 0xcf,\n\t\t0x9c, 0xf3, 0xd2, 0xea, 0x76, 0x7e, 0xb7, 0xf0, 0xf0, 0xf1, 0x6c, 0xa9, 0xca, 0x6c, 0x63, 0xa9,\n\t\t0x13, 0xaa, 0x50, 0xb9, 0x06, 0xcd, 0xa5, 0xfe, 0x25, 0x5e, 0xed, 0x25, 0xa5, 0xe8, 0x25, 0xac,\n\t\t0x33, 0xf3, 0x2d, 0xef, 0x35, 0xf1, 0x3b, 0x76, 0xcf, 0xea, 0x79, 0x1d, 0xa7, 0x75, 0x59, 0x92,\n\t\t0x79, 0x65, 0xa4, 0xf7, 0x05, 0x76, 0xc0, 0x5a, 0x08, 0xaf, 0x73, 0x34, 0x5e, 0x6b, 0x8d, 0x8a,\n\t\t0xd0, 0x5b, 0xb8, 0x6d, 0xb7, 0xa8, 0xf3, 0x9a, 0x58, 0xad, 0x4e, 0x3f, 0xa0, 0xc4, 0xb7, 0x02,\n\t\t0xd2, 0x21, 0x2d, 0x7e, 0x24, 0xb1, 0x07, 0xe2, 0x4e, 0x49, 0xaf, 0x3e, 0x85, 0x73, 0xd5, 0x90,\n\t\t0xda, 0x88, 0x98, 0x62, 0xbb, 0x9b, 0xf6, 0x94, 0x55, 0xf4, 0x3e, 0x2c, 0xf3, 0x13, 0x05, 0xad,\n\t\t0x0b, 0xd2, 0xee, 0x77, 0x48, 0x69, 0x9d, 0x47, 0xbe, 0xc8, 0x84, 0x0d, 0x21, 0x43, 0xc7, 0x20,\n\t\t0x0f, 0xcb, 0x45, 0x74, 0x83, 0x0d, 0x7e, 0xe6, 0x8f, 0x66, 0x73, 0xb1, 0x68, 0x04, 0xab, 0x24,\n\t\t0x29, 0x40, 0x26, 0x94, 0xa2, 0x8d, 0xdb, 0xd6, 0x48, 0x45, 0x6e, 0x66, 0x66, 0xc1, 0xd5, 0x01,\n\t\t0x57, 0x8b, 0x97, 0xe6, 0xf5, 0x43, 0xd8, 0x48, 0x0b, 0x27, 0x92, 0x21, 0xff, 0x8a, 0x5c, 0xf2,\n\t\t0x2e, 0xbe, 0x84, 0xd9, 0x9f, 0x68, 0x03, 0xe6, 0x5f, 0xdb, 0x9d, 0x7e, 0xd8, 0x90, 0x97, 0x70,\n\t\t0xf8, 0xf0, 0x75, 0xee, 0x4b, 0x69, 0xe7, 0xfb, 0x1c, 0xdc, 0x1a, 0x6f, 0x6a, 0x5c, 0x99, 0xb8,\n\t\t0xaa, 0xd0, 0xd7, 0xf1, 0x82, 0x91, 0x66, 0x29, 0x87, 0x61, 0x3d, 0xd9, 0xb0, 0x9d, 0xf0, 0x28,\n\t\t0xeb, 0xed, 0x9e, 0x35, 0xec, 0xd4, 0x5e, 0x9f, 0x8a, 0x4b, 0xe2, 0xda, 0x98, 0x03, 0x2a, 0xc2,\n\t\t0x00, 0x7c, 0x33, 0xee, 0x4e, 0x9f, 0x9a, 0x9e, 0x1a, 0xf5, 0x6e, 0xaf, 0x4f, 0xd1, 0x31, 0xdc,\n\t\t0xe0, 0xe6, 0x4d, 0xd0, 0x9e, 0xcf, 0xd2, 0xbe, 0xc5, 0xd8, 0x29, 0x8a, 0x77, 0xfe, 0x26, 0xc1,\n\t\t0x7a, 0x4a, 0xa7, 0x65, 0x0d, 0xa4, 0xed, 0x75, 0x6d, 0xc7, 0xb5, 0x9c, 0xb6, 0x70, 0xf2, 0x62,\n\t\t0x28, 0xd0, 0xdb, 0xe8, 0x36, 0x14, 0xc4, 0xa2, 0x6b, 0x77, 0x23, 0x7f, 0x43, 0x28, 0x32, 0xec,\n\t\t0x2e, 0x99, 0x70, 0xe3, 0xe6, 0xff, 0xd3, 0x1b, 0xf7, 0x0e, 0x14, 0x1d, 0xd7, 0xa1, 0x8e, 0x4d,\n\t\t0x49, 0x9b, 0xd9, 0x35, 0xc7, 0x2f, 0x9b, 0xc2, 0x40, 0xa6, 0xb7, 0x77, 0x7e, 0x2d, 0xc1, 0xa6,\n\t\t0xf6, 0x96, 0x12, 0xdf, 0xb5, 0x3b, 0x3f, 0xc8, 0x14, 0x30, 0x6a, 0x53, 0x6e, 0xdc, 0xa6, 0x3f,\n\t\t0x2f, 0xc0, 0x7a, 0x9d, 0xb8, 0x6d, 0xc7, 0x3d, 0xe7, 0xc5, 0xed, 0xd0, 0x4b, 0x6e, 0xd1, 0x6d,\n\t\t0x28, 0xd8, 0xe2, 0x79, 0xe8, 0x65, 0x88, 0x44, 0x7a, 0x1b, 0x1d, 0xc1, 0xf2, 0x00, 0x90, 0x39,\n\t\t0x6a, 0x44, 0xaa, 0xf9, 0xa8, 0x51, 0xb4, 0x63, 0x4f, 0xe8, 0x31, 0xcc, 0xb3, 0x42, 0x0f, 0xa7,\n\t\t0x8d, 0x95, 0x87, 0xf7, 0xd3, 0xef, 0xdb, 0xa4, 0x85, 0xac, 0xa8, 0x09, 0x0e, 0x79, 0x48, 0x87,\n\t\t0xb5, 0x0b, 0x62, 0xfb, 0xf4, 0x94, 0xd8, 0xd4, 0x6a, 0x13, 0x6a, 0x3b, 0x9d, 0x40, 0xcc, 0x1f,\n\t\t0x37, 0x27, 0x5c, 0xde, 0x97, 0x1d, 0xcf, 0x6e, 0x63, 0x79, 0x40, 0xab, 0x84, 0x2c, 0xf4, 0x0c,\n\t\t0xd6, 0x3b, 0x76, 0x40, 0xad, 0xa1, 0x3e, 0xde, 0x20, 0xe6, 0x33, 0x1b, 0xc4, 0x1a, 0xa3, 0x3d,\n\t\t0x8d, 0x58, 0xfc, 0xb6, 0x38, 0x02, 0x2e, 0x0c, 0xab, 0x82, 0xb4, 0x43, 0x4d, 0x0b, 0x99, 0x9a,\n\t\t0x56, 0x19, 0xa9, 0x11, 0x72, 0xb8, 0x9e, 0x12, 0x5c, 0xb1, 0x29, 0x25, 0xdd, 0x1e, 0xe5, 0x13,\n\t\t0xc9, 0x3c, 0x8e, 0x1e, 0xd1, 0x7d, 0x90, 0xbb, 0xf6, 0x5b, 0xa7, 0xdb, 0xef, 0x5a, 0x42, 0x14,\n\t\t0xf0, 0xe9, 0x62, 0x1e, 0xaf, 0x0a, 0xb9, 0x22, 0xc4, 0x6c, 0x0c, 0x19, 0xb6, 0x3f, 0x6e, 0xc9,\n\t\t0x52, 0xf6, 0x18, 0x32, 0x60, 0x70, 0x3b, 0x54, 0x58, 0x25, 0x6f, 0x7b, 0x4e, 0x58, 0xb3, 0xa1,\n\t\t0x0e, 0xc8, 0xd4, 0xb1, 0x32, 0xa4, 0x70, 0x25, 0x8f, 0xa1, 0xc8, 0x9d, 0x72, 0x66, 0x3b, 0x9d,\n\t\t0xbe, 0x4f, 0xc4, 0x0c, 0x91, 0x1e, 0xa6, 0xa3, 0x10, 0x83, 0x0b, 0x8c, 0x21, 0x1e, 0xd0, 0xc7,\n\t\t0xb0, 0xc1, 0x15, 0xb0, 0x5c, 0x27, 0xbe, 0xe5, 0xb4, 0x89, 0x4b, 0x1d, 0x7a, 0x29, 0xc6, 0x08,\n\t\t0xc4, 0xd6, 0x8e, 0xf9, 0x92, 0x2e, 0x56, 0xd0, 0xe7, 0xb0, 0x15, 0x85, 0x60, 0x94, 0xb4, 0xcc,\n\t\t0x49, 0x9b, 0x62, 0x79, 0x84, 0x77, 0x1b, 0x0a, 0x91, 0x03, 0x58, 0x01, 0xac, 0xf0, 0xd2, 0x81,\n\t\t0x48, 0xa4, 0xb7, 0x77, 0xfe, 0x94, 0x83, 0x6b, 0x22, 0x2f, 0xd5, 0x0b, 0xa7, 0xd3, 0xfe, 0x41,\n\t\t0x2a, 0xfa, 0xa3, 0x98, 0x5a, 0x56, 0x75, 0xf1, 0x26, 0x27, 0xbf, 0x89, 0x0d, 0xf4, 0xbc, 0xd5,\n\t\t0x8d, 0xd6, 0x7f, 0x7e, 0xac, 0xfe, 0xd9, 0xa0, 0x21, 0xc6, 0xdf, 0xb0, 0x6b, 0x8b, 0x21, 0x60,\n\t\t0x6e, 0xca, 0xa0, 0x11, 0xb6, 0x64, 0xde, 0xa9, 0xa3, 0x41, 0xa3, 0x37, 0x2a, 0x42, 0x57, 0x61,\n\t\t0x21, 0xec, 0xb9, 0xbc, 0x7a, 0x96, 0xb0, 0x78, 0xda, 0xf9, 0x47, 0x6e, 0xd0, 0x6f, 0x2a, 0xa4,\n\t\t0xe5, 0x04, 0x91, 0xbf, 0x06, 0x6d, 0x40, 0xca, 0x6e, 0x03, 0x11, 0x31, 0xd1, 0x06, 0xc6, 0x53,\n\t\t0x3c, 0xf7, 0xae, 0x29, 0xfe, 0x0d, 0x14, 0x13, 0xd5, 0x9a, 0xfd, 0xfe, 0x53, 0x08, 0xd2, 0x2b,\n\t\t0x75, 0x2e, 0x59, 0xa9, 0x18, 0xb6, 0x3c, 0xdf, 0x39, 0x77, 0x5c, 0xbb, 0x63, 0x8d, 0x18, 0x99,\n\t\t0xdd, 0x5b, 0x36, 0x23, 0x6a, 0x23, 0x61, 0xec, 0x48, 0x7e, 0x2e, 0x8c, 0xe5, 0xe7, 0x5f, 0x72,\n\t\t0x70, 0x2d, 0x6a, 0x98, 0x55, 0xaf, 0x65, 0x77, 0x2a, 0x4e, 0xd0, 0xb3, 0x69, 0xeb, 0x62, 0xb6,\n\t\t0xfe, 0xfe, 0xdf, 0xf7, 0xe7, 0xcf, 0xe0, 0x56, 0xd2, 0x02, 0xcb, 0x3b, 0xb3, 0xe8, 0x85, 0x13,\n\t\t0x58, 0x71, 0x37, 0x4f, 0x57, 0x78, 0x3d, 0x61, 0x51, 0xed, 0xcc, 0xbc, 0x70, 0x02, 0xd1, 0x15,\n\t\t0xd1, 0x7b, 0x00, 0x7c, 0x6e, 0xa1, 0xde, 0x2b, 0x12, 0xa6, 0x69, 0x11, 0xf3, 0x41, 0xcb, 0x64,\n\t\t0x82, 0x9d, 0x67, 0x50, 0x88, 0xbf, 0xb5, 0x3c, 0x82, 0x05, 0xf1, 0xe2, 0x23, 0xf1, 0x99, 0xff,\n\t\t0xfd, 0x8c, 0x17, 0x1f, 0xfe, 0x4e, 0x28, 0x28, 0x3b, 0x7f, 0xc8, 0xc1, 0x4a, 0x72, 0x09, 0x7d,\n\t\t0x08, 0xab, 0xa7, 0x8e, 0x6b, 0xfb, 0x97, 0x56, 0xeb, 0x82, 0xb4, 0x5e, 0x05, 0xfd, 0xae, 0x08,\n\t\t0xc2, 0x4a, 0x28, 0x56, 0x85, 0x14, 0x6d, 0xc2, 0x82, 0xdf, 0x77, 0xa3, 0xeb, 0x7b, 0x09, 0xcf,\n\t\t0xfb, 0x7d, 0x36, 0xe7, 0x7c, 0x03, 0x37, 0xce, 0x1c, 0x3f, 0x60, 0x57, 0x5e, 0x58, 0x0d, 0x56,\n\t\t0xcb, 0xeb, 0xf6, 0x3a, 0x24, 0x51, 0xea, 0x25, 0x0e, 0x89, 0xea, 0x45, 0x8d, 0x00, 0x9c, 0x5e,\n\t\t0x6c, 0xf9, 0xc4, 0x1e, 0xc4, 0x26, 0xdb, 0x95, 0x05, 0x81, 0x17, 0x8d, 0x7c, 0x99, 0xb7, 0x76,\n\t\t0xc7, 0x3d, 0x9f, 0x35, 0x8f, 0x8b, 0x11, 0x81, 0x2b, 0xb8, 0x05, 0xc0, 0xdf, 0x26, 0xa9, 0x7d,\n\t\t0xda, 0x09, 0xef, 0xc5, 0x45, 0x1c, 0x93, 0x94, 0xff, 0x28, 0xc1, 0x46, 0xda, 0xad, 0x8f, 0x76,\n\t\t0xe0, 0x56, 0x5d, 0x33, 0x2a, 0xba, 0xf1, 0xc4, 0x52, 0x54, 0x53, 0x7f, 0xa9, 0x9b, 0x27, 0x56,\n\t\t0xc3, 0x54, 0x4c, 0xcd, 0xd2, 0x8d, 0x97, 0x4a, 0x55, 0xaf, 0xc8, 0xff, 0x83, 0x3e, 0x80, 0xed,\n\t\t0x09, 0x98, 0x86, 0xfa, 0x54, 0xab, 0x34, 0xab, 0x5a, 0x45, 0x96, 0xa6, 0x68, 0x6a, 0x98, 0x0a,\n\t\t0x36, 0xb5, 0x8a, 0x9c, 0x43, 0xff, 0x0b, 0x1f, 0x4e, 0xc0, 0xa8, 0x8a, 0xa1, 0x6a, 0x55, 0x0b,\n\t\t0x6b, 0x3f, 0x6e, 0x6a, 0x0d, 0x06, 0xce, 0x97, 0x7f, 0x31, 0xb4, 0x39, 0xd1, 0xa2, 0xe2, 0x3b,\n\t\t0x55, 0x34, 0x55, 0x6f, 0xe8, 0x35, 0x63, 0x9a, 0xcd, 0x23, 0x98, 0x09, 0x36, 0x8f, 0xa2, 0x22,\n\t\t0x9b, 0xcb, 0xbf, 0xcc, 0x0d, 0x3f, 0x36, 0xe9, 0x6d, 0x4c, 0xfa, 0x83, 0xa6, 0xfc, 0x01, 0x6c,\n\t\t0x1f, 0xd7, 0xf0, 0xf3, 0xa3, 0x6a, 0xed, 0xd8, 0xd2, 0x2b, 0x16, 0xd6, 0x9a, 0x0d, 0xcd, 0xaa,\n\t\t0xd7, 0xaa, 0xba, 0x7a, 0x12, 0xb3, 0xe4, 0x4b, 0xf8, 0x74, 0x22, 0x4a, 0xa9, 0x32, 0x69, 0xa5,\n\t\t0x59, 0xaf, 0xea, 0x2a, 0xdb, 0xf5, 0x48, 0xd1, 0xab, 0x5a, 0xc5, 0xaa, 0x19, 0xd5, 0x13, 0x59,\n\t\t0x42, 0x1f, 0xc1, 0xee, 0xac, 0x4c, 0x39, 0x87, 0x1e, 0xc0, 0xfd, 0x89, 0x68, 0xac, 0x3d, 0xd3,\n\t\t0x54, 0x33, 0x06, 0xcf, 0xa3, 0x03, 0x78, 0x30, 0x11, 0x6e, 0x6a, 0xf8, 0x85, 0x6e, 0x70, 0x87,\n\t\t0x1e, 0x59, 0xb8, 0x69, 0x18, 0xba, 0xf1, 0x44, 0x9e, 0x2b, 0x5f, 0xc2, 0xda, 0xd8, 0x5b, 0x31,\n\t\t0xba, 0x0d, 0x37, 0x54, 0x5c, 0x33, 0xac, 0xda, 0x4b, 0x0d, 0x57, 0x95, 0xfa, 0xf8, 0xf9, 0x27,\n\t\t0x00, 0x1a, 0xcf, 0xf5, 0x7a, 0x3d, 0x0a, 0x42, 0x1a, 0xe0, 0xb0, 0x79, 0x74, 0xa4, 0x61, 0xab,\n\t\t0x66, 0x68, 0x72, 0xae, 0xfc, 0x3b, 0x09, 0xd6, 0xc6, 0x2e, 0x4a, 0xa6, 0xba, 0xae, 0x60, 0xcd,\n\t\t0x30, 0x2d, 0xb5, 0x5a, 0x4b, 0xf3, 0xfd, 0x04, 0x80, 0x72, 0xa8, 0x18, 0x95, 0x9a, 0x21, 0x4b,\n\t\t0xe8, 0x1e, 0xec, 0xa4, 0x01, 0x44, 0x1a, 0x8a, 0xac, 0x94, 0x73, 0xe8, 0x0e, 0xbc, 0x97, 0x86,\n\t\t0x1b, 0x38, 0x4a, 0xce, 0x97, 0xff, 0x99, 0x83, 0x9b, 0xd3, 0x3e, 0xa7, 0xb1, 0xe4, 0x1f, 0x78,\n\t\t0x5c, 0xfb, 0x56, 0x53, 0x9b, 0x26, 0x4b, 0xb7, 0x50, 0x1f, 0x4b, 0xba, 0x66, 0x23, 0x66, 0x79,\n\t\t0x3c, 0x9a, 0x13, 0xc0, 0x6a, 0xed, 0x45, 0xbd, 0xaa, 0x99, 0xdc, 0x87, 0x65, 0xb8, 0x97, 0x05,\n\t\t0x0f, 0x73, 0x4b, 0xce, 0x25, 0xd2, 0x6a, 0x92, 0x6a, 0x7e, 0x6e, 0x56, 0x85, 0x68, 0x0f, 0xca,\n\t\t0x59, 0xe8, 0x81, 0x17, 0x2a, 0xf2, 0x1c, 0xfa, 0x14, 0x3e, 0xce, 0x36, 0xdc, 0x30, 0x75, 0xa3,\n\t\t0xa9, 0x55, 0x2c, 0xa5, 0x61, 0x19, 0xda, 0xb1, 0x3c, 0x3f, 0xcb, 0x71, 0x4d, 0xfd, 0x05, 0x2b,\n\t\t0x8d, 0xa6, 0x29, 0x2f, 0x94, 0x7f, 0x95, 0x87, 0xad, 0x09, 0x1f, 0x2b, 0xd0, 0x5d, 0xb8, 0x93,\n\t\t0xa2, 0x6a, 0xcc, 0xc1, 0x53, 0x61, 0xa2, 0x29, 0xc8, 0xd2, 0x74, 0xd8, 0xb0, 0xb1, 0x7d, 0x08,\n\t\t0xef, 0x4f, 0x86, 0x0d, 0x03, 0x95, 0x4f, 0xf4, 0x8c, 0x31, 0xa0, 0x08, 0xd1, 0x1c, 0x4b, 0xcb,\n\t\t0x29, 0xea, 0xa2, 0xe0, 0xcc, 0xa3, 0x5d, 0xf8, 0x60, 0x32, 0x2e, 0x16, 0x96, 0x85, 0x09, 0x61,\n\t\t0x9c, 0x14, 0x90, 0x2b, 0xd3, 0x0f, 0x34, 0x0c, 0xc5, 0x62, 0xf9, 0xaf, 0x12, 0x5c, 0x55, 0x3d,\n\t\t0x97, 0x3a, 0x6e, 0x9f, 0x28, 0x81, 0x41, 0xde, 0xe8, 0xe1, 0x38, 0xec, 0xf9, 0xcc, 0x77, 0x91,\n\t\t0x66, 0xa1, 0xd8, 0xd2, 0x0d, 0xdd, 0xd4, 0x15, 0xb3, 0x86, 0x93, 0x91, 0x98, 0x0c, 0x63, 0x6d,\n\t\t0xb9, 0xa2, 0xe1, 0x30, 0xc5, 0x27, 0xc3, 0xb0, 0x66, 0xe2, 0x13, 0x51, 0x95, 0xe1, 0x3d, 0x33,\n\t\t0x19, 0xcb, 0x9b, 0x4d, 0x74, 0x0b, 0xc8, 0xf9, 0xf2, 0xef, 0x25, 0x28, 0x88, 0x6f, 0x24, 0xfc,\n\t\t0x15, 0xba, 0x04, 0x1b, 0xec, 0x80, 0xb5, 0xa6, 0x69, 0x99, 0x27, 0x75, 0x2d, 0xd9, 0x4e, 0x12,\n\t\t0x2b, 0x3c, 0xfe, 0x96, 0x59, 0x0b, 0x13, 0x35, 0x6c, 0x65, 0x49, 0x80, 0xd8, 0x85, 0x61, 0x38,\n\t\t0x58, 0xce, 0x4d, 0xc5, 0x84, 0x7a, 0xf2, 0xe8, 0x3a, 0x5c, 0x4d, 0x60, 0x9e, 0x6a, 0x0a, 0x36,\n\t\t0x0f, 0x35, 0xc5, 0x94, 0xe7, 0xca, 0xbf, 0x95, 0xe0, 0x5a, 0x74, 0x1f, 0x9a, 0x6c, 0xbc, 0x72,\n\t\t0xba, 0xa4, 0x5d, 0xeb, 0x53, 0xd5, 0xee, 0x07, 0x04, 0xdd, 0x87, 0xbb, 0x83, 0x9b, 0xcc, 0x54,\n\t\t0x1a, 0xcf, 0x87, 0xb1, 0xb2, 0x54, 0x85, 0xb5, 0xf8, 0xe1, 0x69, 0x32, 0xa1, 0xc2, 0x04, 0x59,\n\t\t0x62, 0xd9, 0x30, 0x1d, 0x8a, 0xb5, 0x86, 0x66, 0xca, 0xb9, 0xf2, 0xdf, 0x0b, 0xb0, 0x15, 0x37,\n\t\t0x8e, 0xbd, 0x68, 0x92, 0x76, 0x68, 0xda, 0x3d, 0xd8, 0x49, 0x2a, 0x11, 0xb7, 0xdd, 0xa8, 0x5d,\n\t\t0x07, 0xf0, 0x60, 0x0a, 0xae, 0x69, 0x3c, 0x55, 0x8c, 0x0a, 0x7b, 0x8e, 0x40, 0xb2, 0x84, 0x1e,\n\t\t0xc3, 0xa3, 0x29, 0x94, 0x43, 0xa5, 0x32, 0xf4, 0xf2, 0x60, 0xee, 0x50, 0x4c, 0x13, 0xeb, 0x87,\n\t\t0x4d, 0x53, 0x6b, 0xc8, 0x39, 0xa4, 0x81, 0x92, 0xa1, 0x20, 0x79, 0x25, 0xa4, 0xaa, 0xc9, 0xa3,\n\t\t0xaf, 0xe0, 0xb3, 0x2c, 0x3b, 0xc2, 0x94, 0xd1, 0x5f, 0x68, 0x38, 0x4e, 0x9d, 0x43, 0x5f, 0xc3,\n\t\t0xe7, 0x19, 0x54, 0xb1, 0xf3, 0x18, 0x77, 0x1e, 0x3d, 0x82, 0x2f, 0x32, 0xad, 0x57, 0x6b, 0xb8,\n\t\t0x62, 0xbd, 0x50, 0xf0, 0xf3, 0x24, 0x79, 0x01, 0xe9, 0xa0, 0x65, 0x6d, 0x2c, 0xfa, 0x97, 0x95,\n\t\t0xd2, 0x11, 0x62, 0xaa, 0xae, 0xcc, 0xe0, 0x45, 0x26, 0xc8, 0x50, 0xb3, 0x88, 0x9e, 0x80, 0x3a,\n\t\t0x9b, 0x2b, 0xa6, 0x2b, 0x5a, 0x42, 0xdf, 0x82, 0xf9, 0x6e, 0x51, 0xd5, 0xbe, 0x35, 0x35, 0x6c,\n\t\t0x28, 0x59, 0x9a, 0x01, 0x7d, 0x03, 0x5f, 0x65, 0x3a, 0x2d, 0xd9, 0x7f, 0x62, 0xf4, 0x02, 0xfa,\n\t\t0x02, 0x3e, 0x99, 0x42, 0x8f, 0xe7, 0xc8, 0x70, 0x36, 0xd4, 0x2b, 0x72, 0x11, 0x7d, 0x06, 0x07,\n\t\t0x53, 0x88, 0xbc, 0x0a, 0xad, 0x86, 0xa9, 0xab, 0xcf, 0x4f, 0xc2, 0xe5, 0xaa, 0xde, 0x30, 0xe5,\n\t\t0x65, 0xf4, 0x23, 0xf8, 0xbf, 0x29, 0xb4, 0xc1, 0x61, 0xd9, 0x1f, 0x1a, 0x8e, 0x95, 0x18, 0x83,\n\t\t0x35, 0xb1, 0x26, 0xaf, 0xcc, 0x10, 0x93, 0x86, 0xfe, 0x24, 0xdb, 0x73, 0xab, 0x48, 0x85, 0xc7,\n\t\t0x33, 0x95, 0x88, 0xfa, 0x54, 0xaf, 0x56, 0xd2, 0x95, 0xc8, 0xe8, 0x13, 0xd8, 0x9f, 0xa2, 0xe4,\n\t\t0xa8, 0x86, 0x55, 0x4d, 0x0c, 0x0f, 0x83, 0x26, 0xb1, 0x86, 0x3e, 0x87, 0x87, 0xd3, 0x48, 0x8a,\n\t\t0x5e, 0x65, 0x13, 0xe8, 0x28, 0x0f, 0xb1, 0x89, 0x66, 0xb6, 0xa3, 0xeb, 0x46, 0xbd, 0x69, 0x5a,\n\t\t0x0d, 0xfd, 0x3b, 0x4d, 0x5e, 0x67, 0x13, 0x4d, 0x66, 0xa4, 0x22, 0x5f, 0xc9, 0x1b, 0xe3, 0xcd,\n\t\t0x78, 0x6c, 0x93, 0x43, 0xdd, 0x50, 0xf0, 0x89, 0xbc, 0x99, 0x91, 0x7b, 0xe3, 0x8d, 0x2e, 0x91,\n\t\t0x42, 0x57, 0x67, 0x39, 0x8e, 0xa6, 0x60, 0xf5, 0x69, 0xdc, 0xe3, 0x5b, 0xec, 0xd6, 0xb9, 0xc3,\n\t\t0xbf, 0xcb, 0x8d, 0x8d, 0x5d, 0xf1, 0x16, 0x7f, 0x00, 0x0f, 0xc2, 0xb8, 0xa5, 0x64, 0xc1, 0x84,\n\t\t0x6e, 0x7f, 0x08, 0xff, 0x3f, 0x1b, 0x65, 0xb0, 0xae, 0x54, 0xb1, 0xa6, 0x54, 0x4e, 0x06, 0x2f,\n\t\t0x26, 0x52, 0xf9, 0x37, 0x39, 0x28, 0xab, 0xb6, 0xdb, 0x22, 0x9d, 0xe8, 0xff, 0x01, 0x53, 0xad,\n\t\t0x7c, 0x04, 0x5f, 0xcc, 0x50, 0xef, 0x13, 0xec, 0x3d, 0x86, 0xc6, 0xbb, 0x92, 0x9b, 0xc6, 0x73,\n\t\t0xa3, 0x76, 0x6c, 0x4c, 0x23, 0xc8, 0x12, 0x32, 0xe0, 0xd9, 0xbb, 0x2a, 0x1e, 0x73, 0xc9, 0x70,\n\t\t0xd2, 0xcc, 0x71, 0xa7, 0x34, 0x9c, 0x73, 0xfe, 0xcf, 0x91, 0xd9, 0x9c, 0x22, 0xd2, 0xf8, 0xdf,\n\t\t0x73, 0xca, 0xbb, 0x92, 0x67, 0x76, 0xca, 0xbb, 0x2a, 0x9e, 0xe6, 0x94, 0xc3, 0x9f, 0xc2, 0x56,\n\t\t0xcb, 0xeb, 0xa6, 0x7d, 0x6b, 0x3a, 0x5c, 0x8e, 0xdc, 0x53, 0xf7, 0x3d, 0xea, 0xd5, 0xa5, 0xef,\n\t\t0x0e, 0xce, 0x1d, 0x7a, 0xd1, 0x3f, 0xdd, 0x6b, 0x79, 0xdd, 0xfd, 0xf8, 0x8f, 0x52, 0x1e, 0x38,\n\t\t0xed, 0xce, 0xfe, 0xb9, 0x17, 0xfe, 0xc8, 0x45, 0xfc, 0x42, 0xe5, 0x91, 0xdd, 0x73, 0x5e, 0x1f,\n\t\t0x9c, 0x2e, 0x70, 0xd9, 0x27, 0xff, 0x0a, 0x00, 0x00, 0xff, 0xff, 0x00, 0x6a, 0x76, 0xd7, 0x61,\n\t\t0x23, 0x00, 0x00,\n\t},\n\t// uber/cadence/admin/v1/history.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2e, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x4f, 0x4c, 0xc9, 0xcd, 0xcc, 0xd3, 0x2f,\n\t\t0x33, 0xd4, 0xcf, 0xc8, 0x2c, 0x2e, 0xc9, 0x2f, 0xaa, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17,\n\t\t0x12, 0x05, 0x29, 0xd2, 0x83, 0x2a, 0xd2, 0x03, 0x2b, 0xd2, 0x2b, 0x33, 0x54, 0xf2, 0xe4, 0x12,\n\t\t0x0a, 0x4b, 0x2d, 0x2a, 0xce, 0xcc, 0xcf, 0xf3, 0x80, 0x28, 0xf7, 0x2c, 0x49, 0xcd, 0x15, 0x92,\n\t\t0xe4, 0xe2, 0x48, 0x2d, 0x4b, 0xcd, 0x2b, 0x89, 0xcf, 0x4c, 0x91, 0x60, 0x54, 0x60, 0xd4, 0x60,\n\t\t0x0e, 0x62, 0x07, 0xf3, 0x3d, 0x53, 0x84, 0x24, 0xb8, 0xd8, 0xcb, 0x20, 0x1a, 0x24, 0x98, 0x20,\n\t\t0x32, 0x50, 0xae, 0x52, 0x09, 0x17, 0x1f, 0xaa, 0x51, 0x42, 0x8a, 0x5c, 0x3c, 0x49, 0x45, 0x89,\n\t\t0x79, 0xc9, 0x19, 0xf1, 0x25, 0xf9, 0xd9, 0xa9, 0x79, 0x60, 0xa3, 0x78, 0x82, 0xb8, 0x21, 0x62,\n\t\t0x21, 0x20, 0x21, 0x21, 0x7b, 0x2e, 0xd6, 0xcc, 0x92, 0xd4, 0xdc, 0x62, 0x09, 0x26, 0x05, 0x66,\n\t\t0x0d, 0x6e, 0x23, 0x4d, 0x3d, 0xac, 0xce, 0xd4, 0xc3, 0x74, 0x63, 0x10, 0x44, 0x9f, 0x93, 0x79,\n\t\t0x94, 0x69, 0x7a, 0x66, 0x49, 0x46, 0x69, 0x92, 0x5e, 0x72, 0x7e, 0xae, 0x3e, 0x72, 0x48, 0xe8,\n\t\t0x66, 0xa6, 0xe4, 0xe8, 0xa7, 0xe7, 0xeb, 0x83, 0xfd, 0x0f, 0x0f, 0x16, 0x6b, 0x30, 0xa3, 0xcc,\n\t\t0x30, 0x89, 0x0d, 0x2c, 0x6e, 0x0c, 0x08, 0x00, 0x00, 0xff, 0xff, 0x44, 0x14, 0xd7, 0xd4, 0x3e,\n\t\t0x01, 0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/tasklist.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/shared/v1/tasklist.proto\n\npackage sharedv1\n\nimport (\n\tfmt \"fmt\"\n\tmath \"math\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\n// TaskSource is the source from which a task was produced.\ntype TaskSource int32\n\nconst (\n\tTaskSource_TASK_SOURCE_INVALID TaskSource = 0\n\t// Task produced by history service.\n\tTaskSource_TASK_SOURCE_HISTORY TaskSource = 1\n\t// Task produced from matching db backlog.\n\tTaskSource_TASK_SOURCE_DB_BACKLOG TaskSource = 2\n)\n\nvar TaskSource_name = map[int32]string{\n\t0: \"TASK_SOURCE_INVALID\",\n\t1: \"TASK_SOURCE_HISTORY\",\n\t2: \"TASK_SOURCE_DB_BACKLOG\",\n}\n\nvar TaskSource_value = map[string]int32{\n\t\"TASK_SOURCE_INVALID\":    0,\n\t\"TASK_SOURCE_HISTORY\":    1,\n\t\"TASK_SOURCE_DB_BACKLOG\": 2,\n}\n\nfunc (x TaskSource) String() string {\n\treturn proto.EnumName(TaskSource_name, int32(x))\n}\n\nfunc (TaskSource) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_e97215b1a3d16713, []int{0}\n}\n\nfunc init() {\n\tproto.RegisterEnum(\"uber.cadence.shared.v1.TaskSource\", TaskSource_name, TaskSource_value)\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/shared/v1/tasklist.proto\", fileDescriptor_e97215b1a3d16713)\n}\n\nvar fileDescriptor_e97215b1a3d16713 = []byte{\n\t// 193 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t0x2f, 0x33, 0xd4, 0x2f, 0x49, 0x2c, 0xce, 0xce, 0xc9, 0x2c, 0x2e, 0xd1, 0x2b, 0x28, 0xca, 0x2f,\n\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0x8a,\n\t0xe2, 0xe2, 0x0a, 0x49, 0x2c, 0xce, 0x0e, 0xce, 0x2f, 0x2d, 0x4a, 0x4e, 0x15, 0x12, 0xe7, 0x12,\n\t0x0e, 0x71, 0x0c, 0xf6, 0x8e, 0x0f, 0xf6, 0x0f, 0x0d, 0x72, 0x76, 0x8d, 0xf7, 0xf4, 0x0b, 0x73,\n\t0xf4, 0xf1, 0x74, 0x11, 0x60, 0x40, 0x97, 0xf0, 0xf0, 0x0c, 0x0e, 0xf1, 0x0f, 0x8a, 0x14, 0x60,\n\t0x14, 0x92, 0xe2, 0x12, 0x43, 0x96, 0x70, 0x71, 0x8a, 0x77, 0x72, 0x74, 0xf6, 0xf6, 0xf1, 0x77,\n\t0x17, 0x60, 0x72, 0x72, 0x3e, 0xf1, 0x48, 0x8e, 0xf1, 0xc2, 0x23, 0x39, 0xc6, 0x07, 0x8f, 0xe4,\n\t0x18, 0xa3, 0x4c, 0xd3, 0x33, 0x4b, 0x32, 0x4a, 0x93, 0xf4, 0x92, 0xf3, 0x73, 0xf5, 0x51, 0xdc,\n\t0xac, 0x97, 0x9e, 0x9a, 0xa7, 0x0f, 0x76, 0x25, 0xc2, 0xf9, 0xd6, 0x10, 0x56, 0x99, 0x61, 0x12,\n\t0x1b, 0x58, 0xc6, 0x18, 0x10, 0x00, 0x00, 0xff, 0xff, 0xc3, 0xa0, 0xac, 0xa3, 0xe8, 0x00, 0x00,\n\t0x00,\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/tasklist.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/shared/v1/tasklist.proto\n\npackage sharedv1\n\nvar yarpcFileDescriptorClosuree97215b1a3d16713 = [][]byte{\n\t// uber/cadence/shared/v1/tasklist.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x2f, 0x49, 0x2c, 0xce, 0xce, 0xc9, 0x2c, 0x2e, 0xd1, 0x2b, 0x28, 0xca, 0x2f,\n\t\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0x8a,\n\t\t0xe2, 0xe2, 0x0a, 0x49, 0x2c, 0xce, 0x0e, 0xce, 0x2f, 0x2d, 0x4a, 0x4e, 0x15, 0x12, 0xe7, 0x12,\n\t\t0x0e, 0x71, 0x0c, 0xf6, 0x8e, 0x0f, 0xf6, 0x0f, 0x0d, 0x72, 0x76, 0x8d, 0xf7, 0xf4, 0x0b, 0x73,\n\t\t0xf4, 0xf1, 0x74, 0x11, 0x60, 0x40, 0x97, 0xf0, 0xf0, 0x0c, 0x0e, 0xf1, 0x0f, 0x8a, 0x14, 0x60,\n\t\t0x14, 0x92, 0xe2, 0x12, 0x43, 0x96, 0x70, 0x71, 0x8a, 0x77, 0x72, 0x74, 0xf6, 0xf6, 0xf1, 0x77,\n\t\t0x17, 0x60, 0x72, 0x32, 0x8f, 0x32, 0x4d, 0xcf, 0x2c, 0xc9, 0x28, 0x4d, 0xd2, 0x4b, 0xce, 0xcf,\n\t\t0xd5, 0x47, 0x71, 0xa7, 0x5e, 0x7a, 0x6a, 0x9e, 0x3e, 0xd8, 0x65, 0x08, 0x27, 0x5b, 0x43, 0x58,\n\t\t0x65, 0x86, 0x49, 0x6c, 0x60, 0x19, 0x63, 0x40, 0x00, 0x00, 0x00, 0xff, 0xff, 0x1f, 0x73, 0x06,\n\t\t0x1c, 0xdc, 0x00, 0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/workflow.pb.go",
    "content": "// Code generated by protoc-gen-gogo. DO NOT EDIT.\n// source: uber/cadence/shared/v1/workflow.proto\n\npackage sharedv1\n\nimport (\n\tfmt \"fmt\"\n\tmath \"math\"\n\n\tproto \"github.com/gogo/protobuf/proto\"\n)\n\n// Reference imports to suppress errors if they are not otherwise used.\nvar _ = proto.Marshal\nvar _ = fmt.Errorf\nvar _ = math.Inf\n\n// This is a compile-time assertion to ensure that this generated file\n// is compatible with the proto package it is being compiled against.\n// A compilation error at this line likely means your copy of the\n// proto package needs to be updated.\nconst _ = proto.GoGoProtoPackageIsVersion3 // please upgrade the proto package\n\ntype WorkflowState int32\n\nconst (\n\tWorkflowState_WORKFLOW_STATE_INVALID   WorkflowState = 0\n\tWorkflowState_WORKFLOW_STATE_CREATED   WorkflowState = 1\n\tWorkflowState_WORKFLOW_STATE_RUNNING   WorkflowState = 2\n\tWorkflowState_WORKFLOW_STATE_COMPLETED WorkflowState = 3\n\tWorkflowState_WORKFLOW_STATE_ZOMBIE    WorkflowState = 4\n\tWorkflowState_WORKFLOW_STATE_VOID      WorkflowState = 5\n\tWorkflowState_WORKFLOW_STATE_CORRUPTED WorkflowState = 6\n)\n\nvar WorkflowState_name = map[int32]string{\n\t0: \"WORKFLOW_STATE_INVALID\",\n\t1: \"WORKFLOW_STATE_CREATED\",\n\t2: \"WORKFLOW_STATE_RUNNING\",\n\t3: \"WORKFLOW_STATE_COMPLETED\",\n\t4: \"WORKFLOW_STATE_ZOMBIE\",\n\t5: \"WORKFLOW_STATE_VOID\",\n\t6: \"WORKFLOW_STATE_CORRUPTED\",\n}\n\nvar WorkflowState_value = map[string]int32{\n\t\"WORKFLOW_STATE_INVALID\":   0,\n\t\"WORKFLOW_STATE_CREATED\":   1,\n\t\"WORKFLOW_STATE_RUNNING\":   2,\n\t\"WORKFLOW_STATE_COMPLETED\": 3,\n\t\"WORKFLOW_STATE_ZOMBIE\":    4,\n\t\"WORKFLOW_STATE_VOID\":      5,\n\t\"WORKFLOW_STATE_CORRUPTED\": 6,\n}\n\nfunc (x WorkflowState) String() string {\n\treturn proto.EnumName(WorkflowState_name, int32(x))\n}\n\nfunc (WorkflowState) EnumDescriptor() ([]byte, []int) {\n\treturn fileDescriptor_7ca73ea33aecbb95, []int{0}\n}\n\nfunc init() {\n\tproto.RegisterEnum(\"uber.cadence.shared.v1.WorkflowState\", WorkflowState_name, WorkflowState_value)\n}\n\nfunc init() {\n\tproto.RegisterFile(\"uber/cadence/shared/v1/workflow.proto\", fileDescriptor_7ca73ea33aecbb95)\n}\n\nvar fileDescriptor_7ca73ea33aecbb95 = []byte{\n\t// 239 bytes of a gzipped FileDescriptorProto\n\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t0x2f, 0x33, 0xd4, 0x2f, 0xcf, 0x2f, 0xca, 0x4e, 0xcb, 0xc9, 0x2f, 0xd7, 0x2b, 0x28, 0xca, 0x2f,\n\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0xba,\n\t0xcc, 0xc8, 0xc5, 0x1b, 0x0e, 0x55, 0x1a, 0x5c, 0x92, 0x58, 0x92, 0x2a, 0x24, 0xc5, 0x25, 0x16,\n\t0xee, 0x1f, 0xe4, 0xed, 0xe6, 0xe3, 0x1f, 0x1e, 0x1f, 0x1c, 0xe2, 0x18, 0xe2, 0x1a, 0xef, 0xe9,\n\t0x17, 0xe6, 0xe8, 0xe3, 0xe9, 0x22, 0xc0, 0x80, 0x45, 0xce, 0x39, 0xc8, 0xd5, 0x31, 0xc4, 0xd5,\n\t0x45, 0x80, 0x11, 0x8b, 0x5c, 0x50, 0xa8, 0x9f, 0x9f, 0xa7, 0x9f, 0xbb, 0x00, 0x93, 0x90, 0x0c,\n\t0x97, 0x04, 0xba, 0x3e, 0x7f, 0xdf, 0x00, 0x1f, 0x57, 0x90, 0x4e, 0x66, 0x21, 0x49, 0x2e, 0x51,\n\t0x34, 0xd9, 0x28, 0x7f, 0x5f, 0x27, 0x4f, 0x57, 0x01, 0x16, 0x21, 0x71, 0x2e, 0x61, 0x34, 0xa9,\n\t0x30, 0x7f, 0x4f, 0x17, 0x01, 0x56, 0xac, 0x26, 0x06, 0x05, 0x85, 0x06, 0x80, 0x4c, 0x64, 0x73,\n\t0x72, 0x3e, 0xf1, 0x48, 0x8e, 0xf1, 0xc2, 0x23, 0x39, 0xc6, 0x07, 0x8f, 0xe4, 0x18, 0xa3, 0x4c,\n\t0xd3, 0x33, 0x4b, 0x32, 0x4a, 0x93, 0xf4, 0x92, 0xf3, 0x73, 0xf5, 0x51, 0x42, 0x4b, 0x2f, 0x3d,\n\t0x35, 0x4f, 0x1f, 0x1c, 0x3e, 0x88, 0x80, 0xb3, 0x86, 0xb0, 0xca, 0x0c, 0x93, 0xd8, 0xc0, 0x32,\n\t0xc6, 0x80, 0x00, 0x00, 0x00, 0xff, 0xff, 0xdf, 0x33, 0x81, 0xb3, 0x62, 0x01, 0x00, 0x00,\n}\n"
  },
  {
    "path": ".gen/proto/shared/v1/workflow.pb.yarpc.go",
    "content": "// Code generated by protoc-gen-yarpc-go. DO NOT EDIT.\n// source: uber/cadence/shared/v1/workflow.proto\n\npackage sharedv1\n\nvar yarpcFileDescriptorClosure7ca73ea33aecbb95 = [][]byte{\n\t// uber/cadence/shared/v1/workflow.proto\n\t[]byte{\n\t\t0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x52, 0x2d, 0x4d, 0x4a, 0x2d,\n\t\t0xd2, 0x4f, 0x4e, 0x4c, 0x49, 0xcd, 0x4b, 0x4e, 0xd5, 0x2f, 0xce, 0x48, 0x2c, 0x4a, 0x4d, 0xd1,\n\t\t0x2f, 0x33, 0xd4, 0x2f, 0xcf, 0x2f, 0xca, 0x4e, 0xcb, 0xc9, 0x2f, 0xd7, 0x2b, 0x28, 0xca, 0x2f,\n\t\t0xc9, 0x17, 0x12, 0x03, 0x29, 0xd3, 0x83, 0x2a, 0xd3, 0x83, 0x28, 0xd3, 0x2b, 0x33, 0xd4, 0xba,\n\t\t0xcc, 0xc8, 0xc5, 0x1b, 0x0e, 0x55, 0x1a, 0x5c, 0x92, 0x58, 0x92, 0x2a, 0x24, 0xc5, 0x25, 0x16,\n\t\t0xee, 0x1f, 0xe4, 0xed, 0xe6, 0xe3, 0x1f, 0x1e, 0x1f, 0x1c, 0xe2, 0x18, 0xe2, 0x1a, 0xef, 0xe9,\n\t\t0x17, 0xe6, 0xe8, 0xe3, 0xe9, 0x22, 0xc0, 0x80, 0x45, 0xce, 0x39, 0xc8, 0xd5, 0x31, 0xc4, 0xd5,\n\t\t0x45, 0x80, 0x11, 0x8b, 0x5c, 0x50, 0xa8, 0x9f, 0x9f, 0xa7, 0x9f, 0xbb, 0x00, 0x93, 0x90, 0x0c,\n\t\t0x97, 0x04, 0xba, 0x3e, 0x7f, 0xdf, 0x00, 0x1f, 0x57, 0x90, 0x4e, 0x66, 0x21, 0x49, 0x2e, 0x51,\n\t\t0x34, 0xd9, 0x28, 0x7f, 0x5f, 0x27, 0x4f, 0x57, 0x01, 0x16, 0x21, 0x71, 0x2e, 0x61, 0x34, 0xa9,\n\t\t0x30, 0x7f, 0x4f, 0x17, 0x01, 0x56, 0xac, 0x26, 0x06, 0x05, 0x85, 0x06, 0x80, 0x4c, 0x64, 0x73,\n\t\t0x32, 0x8f, 0x32, 0x4d, 0xcf, 0x2c, 0xc9, 0x28, 0x4d, 0xd2, 0x4b, 0xce, 0xcf, 0xd5, 0x47, 0x09,\n\t\t0x21, 0xbd, 0xf4, 0xd4, 0x3c, 0x7d, 0x70, 0x98, 0x20, 0x02, 0xcb, 0x1a, 0xc2, 0x2a, 0x33, 0x4c,\n\t\t0x62, 0x03, 0xcb, 0x18, 0x03, 0x02, 0x00, 0x00, 0xff, 0xff, 0xff, 0xc3, 0x37, 0x89, 0x56, 0x01,\n\t\t0x00, 0x00,\n\t},\n}\n"
  },
  {
    "path": ".gitar/rules/assign-maintainer-reviewer.md",
    "content": "---\ntitle: Assign Maintainer Reviewer for External Contributors\ndescription: Automatically assigns a maintainer to review PRs from non-maintainers\nwhen: Pull request is opened\nactions:\n  - Assign reviewer in round-robin rotation\n---\n\n# Auto-assign Maintainer Reviewer\n\nWhen a pull request is opened by someone not listed in `.github/CODEOWNERS`, assign @natemort plus one reviewer from the following list in round-robin rotation:\n\n- @c-warren\n- @fimanishi \n- @neil-xie\n- @zawadzkidiana\n- @shijiesheng\n- @abhishekj720 \n\nYou must modify the assignees field on the Pull Request - being a reviewer is not sufficient.\n\nDo not assign a reviewer if the PR author is already a maintainer listed in CODEOWNERS.\n"
  },
  {
    "path": ".gitar/rules/issue-linking-required.md",
    "content": "---\ntitle: GitHub Issue Linking Requirement\ndescription: Ensures PRs link to cadence-workflow org issues with smart exemptions\nwhen: PR is opened or updated\nactions: Check for issue link and report requirement status\n---\n\n# GitHub Issue Linking Requirement\n\nThis rule ensures that pull requests link to GitHub issues from the cadence-workflow organization, providing better tracking, context, and historical linkage between code changes and their motivations.\n\n## Overview\n\nWhen evaluating a pull request:\n\n1. **Check skip conditions first** - Some PRs are exempt from this requirement\n2. **Search for issue links** - Look for references in the PR description body\n3. **Validate organization** - Ensure links point to cadence-workflow org\n4. **Report status** - Approve silently if valid link found, or provide helpful feedback if missing\n\n## Skip Conditions\n\nThe issue linking requirement is automatically skipped when **ANY** of these conditions are true:\n\n### 1. Small Changes (< 50 lines)\n- Count total lines changed: additions + deletions\n- PRs with fewer than 50 total lines changed are exempt\n- Example: Quick typo fixes, minor documentation updates\n\n### 2. Maintenance Commits\nPRs with titles starting with these conventional commit prefixes are exempt:\n- `docs:` - Documentation changes\n- `chore:` - Maintenance tasks, dependency updates\n- `ci:` - CI/CD configuration changes\n- `style:` - Code formatting or style changes\n- `revert:` - Pull Request Reverts\n\nExamples:\n- ✅ Skip: `docs: Update installation instructions`\n- ✅ Skip: `chore: Bump dependency versions`\n- ✅ Skip: `ci: Add new workflow for release`\n- ✅ Skip: `style: Apply consistent formatting`\n- ❌ Don't skip: `feat: Add new authentication method`\n- ❌ Don't skip: `fix: Resolve memory leak in worker`\n\n### 3. Bot-Authored PRs\n- PRs where the author username ends with `[bot]`\n- Examples: `dependabot[bot]`, `renovate[bot]`, `github-actions[bot]`\n\n## Issue Link Detection\n\nSearch the **PR description body only** for issue references in these formats:\n\n### Accepted Formats\n\n1. **Same repository short format:**\n   - `#123`\n   - Example: \"Fixes #456\"\n   - Example: \"Related to #789\"\n\n2. **Cross-repository format:**\n   - `cadence-workflow/other-repo#123`\n   - Example: \"Addresses cadence-workflow/web#45\"\n   - Must reference the `cadence-workflow` organization\n\n3. **Full URL format:**\n   - `https://github.com/cadence-workflow/cadence/issues/123`\n   - `https://github.com/cadence-workflow/other-repo/issues/456`\n   - Must be from the `cadence-workflow` organization\n\n### Not Accepted\n\n- ❌ Issues from other organizations: `other-org/repo#123`\n- ❌ Issue links in code comments or file content\n- ❌ Issue links in commit messages (must be in PR description)\n\n## Organization Validation\n\n**CRITICAL**: All issue links must reference the `cadence-workflow` organization.\n\nHow to validate:\n- Short format (`#123`) is implicitly within cadence-workflow if the PR is in a cadence-workflow repo\n- Cross-repo format must explicitly include `cadence-workflow/`\n- Full URLs must contain `github.com/cadence-workflow/`\n\nExamples:\n- ✅ Valid: `#123` (in cadence-workflow repo)\n- ✅ Valid: `cadence-workflow/web#45`\n- ✅ Valid: `https://github.com/cadence-workflow/cadence/issues/789`\n- ❌ Invalid: `external-org/repo#123`\n- ❌ Invalid: `https://github.com/other-org/repo/issues/123`\n\n## Validation Logic\n\n### Step 1: Check Skip Conditions\n\n```\n1. Get PR diff stats\n2. Calculate total_lines = additions + deletions\n3. If total_lines < 50, skip and approve\n4. Get PR title\n5. If title starts with \"docs:\", \"chore:\", \"ci:\", or \"style:\", skip and approve\n6. Get PR author username\n7. If author ends with \"[bot]\", skip and approve\n```\n\n### Step 2: Search for Issue Links\n\n```\n1. Get PR description body\n2. Search for patterns:\n   - #\\d+\n   - cadence-workflow/[\\w-]+#\\d+\n   - https://github.com/cadence-workflow/[\\w-]+/issues/\\d+\n3. For each match, extract organization name\n4. Validate organization is \"cadence-workflow\"\n```\n\n### Step 3: Report Status\n\nReport as part of all rules. \n\n## Edge Cases\n\n### Multiple Issue Links\n- If the PR description contains multiple valid issue links, the rule passes\n- Use the first valid link found for reporting purposes\n\n### Issue Links in Code\n- Issue links in code comments or file content do not count\n- Only links in the PR description body are evaluated\n\n### Mixed Formats\n- A PR can reference issues in multiple formats\n- The rule passes if ANY format contains a valid cadence-workflow issue link\n\n### Case Sensitivity\n- Organization name matching is case-insensitive\n- `cadence-workflow`, `Cadence-Workflow`, and `CADENCE-WORKFLOW` are all valid\n\n## Integration Notes\n\nThis rule is complementary to the existing `pr-description-quality.md` rule:\n- **Description quality**: Checks content structure and completeness\n- **Issue linking**: Checks for specific requirement (issue reference)\n- Both rules can run on the same PR without conflicts\n- Both provide guidance without blocking PRs\n\n"
  },
  {
    "path": ".gitar/rules/pr-description-quality.md",
    "content": "---\ntitle: PR Description Quality Standards\ndescription: Ensures PR descriptions meet Cadence quality criteria using guidance from PR template\nwhen: PR description is created or updated\nactions: Read PR template for guidance then report requirement status\n---\n\n# PR Description Quality Standards\n\nWhen evaluating a pull request description:\n\n1. **Read the PR template guidance** at `.github/pull_request_guidance.md` to understand the expected guidance for each section\n2. Apply that guidance to evaluate the current PR description\n3. Provide recommendations for how to improve the description. \n\n## Core Principle: Why Not How\n\nFrom https://cbea.ms/git-commit/#why-not-how:\n- **\"A diff shows WHAT changed, but only the description can explain WHY\"**\n- Focus on: the problem being solved, the reasoning behind the solution, context\n- The code itself documents HOW - the PR description documents WHY\n\n## Evaluation Criteria\n\n### Required Sections (must exist with substantive content per PR template guidance)\n\n1. **What changed?**\n   - 1-2 line summary of WHAT changed technically\n   - Focus on key modification, not implementation details\n   - Template has good/bad examples\n\n2. **Why?**\n   - Full context and motivation\n   - What problem does this solve? What's the use case?\n   - What's the impact if we don't make this change?\n   - **CRITICAL**: Technical rationale required for ALL changes (not just large ones)\n   - Must explain WHY this implementation approach was chosen\n\n3. **How did you test it?**\n   - Concrete, copyable commands with exact invocations\n   - ✅ GOOD: `go test -v ./common/types/mapper/proto -run TestFailoverDomainRequest`\n   - ❌ BAD: \"Tested locally\" or \"See tests/foo_test.go\"\n   - For integration tests: setup steps + commands\n   - For canary: which canary, environment, results\n\n4. **Potential risks**\n   - Backward/forward compatibility concerns?\n   - Performance impact? What could break? Safe to rollback?\n\n5. **Release notes**\n   - If this completes a user-facing feature, describe it\n   - Skip for: incremental work, internal refactors, partial implementations\n\n6. **Documentation Changes**\n   - Config changes affecting operation-guide?\n   - New features needing user docs?\n   - Only mark N/A if certain no docs affected\n\n### Quality Checks\n\n- **Skip obvious things** - Don't flag items clear from folder structure\n- **Skip trivial refactors** - Minor formatting/style changes don't need deep rationale\n- **Don't check automated items** - Issue links, CI, linting are automated\n\n## FORBIDDEN - Never Include\n\n- ❌ \"Issues Found\", \"Testing Evidence Quality\", \"Documentation Reasoning\", \"Summary\" sections\n- ❌ \"Note:\" paragraphs or explanatory text outside recommendations\n- ❌ Grouping recommendations by type\n\n## Section Names (Use EXACT Brackets)\n\n- **[What changed?]**\n- **[Why?]**\n- **[How did you test it?]**\n- **[Potential risks]**\n- **[Release notes]**\n- **[Documentation Changes]**\n"
  },
  {
    "path": ".gitattributes",
    "content": ".gen linguist-generated=true\n"
  },
  {
    "path": ".github/CODEOWNERS",
    "content": "# This is a comment.\n# Each line is a file pattern followed by one or more owners.\n# Order is important; the last matching pattern takes the most\n# precedence.\n\n* @adhityamamallan @macrotim @Assem-Uber @abhishekj720 @fimanishi @bowenxia @timl3136 @eleonoradgr @gazi-yestemirova @arzonus @c-warren @natemort @Shaddoll @neil-xie @davidporter-id-au @shijiesheng @jakobht @sankari165 @dkrotx @demirkayaender\n"
  },
  {
    "path": ".github/ISSUE_TEMPLATE/bug_report.md",
    "content": "---\nname: Bug report\nabout: Create a report to help us improve\ntitle: ''\nlabels: bug\nassignees: ''\n\n---\n### Description\nA clear and concise description of what the bug is.\n\n### Steps to Reproduce / How to Trigger\nProvide a minimal set of steps or a script to reproduce the issue.\n\n### Expected Behavior\nWhat you expected to happen.\n\n### Actual Behavior\nWhat actually happened.\n\n### Logs / Screenshots\nRelevant logs, stack traces, workflow history, or screenshots.\n\n### Environment\n- **Cadence server version**: Server version or build SHA\n- **Cadence SDK language and version** (if applicable): Language and SDK version\n- **Cadence web version** (if applicable): Web UI version\n- **DB & version**: Database type and version\n- **Scale**: Cluster size, workload volume, or other scale details\n"
  },
  {
    "path": ".github/ISSUE_TEMPLATE/feature_request.md",
    "content": "---\nname: Feature request\nabout: Suggest an idea for this project\ntitle: ''\nlabels: feature\nassignees: ''\n\n---\n\nUse this template for feature, improvement, or refractoring requests.\n\n### Description\nWhat you want to happen and why.\n\n### Is this a breaking change?\n- [ ] Yes\n- [ ] No\n\n**If yes, explain why:**\n\n\n### Scope of the feature (server, specific client, all clients)\nClarify where this applies.\n"
  },
  {
    "path": ".github/ISSUE_TEMPLATE/help_support.md",
    "content": "---\nname: QA/Help/Support\nabout: Please use [Discussion](https://github.com/cadence-workflow/cadence/discussions) or [StackOverflow](https://stackoverflow.com/questions/tagged/cadence-workflow) for QA/Help/Support\ntitle: ''\nlabels: ''\nassignees: ''\n\n---\n\nPlease use [Discussion](https://github.com/cadence-workflow/cadence/discussions) or [StackOverflow](https://stackoverflow.com/questions/tagged/cadence-workflow) for QA/Help/Support.\nDo NOT use issue for this.\n"
  },
  {
    "path": ".github/dco.yml",
    "content": "require:\n  members: false\n"
  },
  {
    "path": ".github/pull_request_guidance.md",
    "content": "<!-- 1-2 line summary of WHAT changed technically:\n- Always link the relevant projects GitHub issue, unless it is a minor bugfix\n- Good: \"Modified FailoverDomain mapper to allow null ActiveClusterName #320\"\n- Bad: \"added nil check\" -->\n**What changed?**\n\n\n<!-- Your goal is to provide all the required context for a future maintainer \nto understand the reasons for making this change (see https://cbea.ms/git-commit/#why-not-how).\nHow did this work previously (and what was wrong with it)? What has changed, and why did you solve it \nthis way?\n- Good: \"Active-active domains have independent cluster attributes per region. Previously,\n  modifying cluster attributes required spedifying the default ActiveClusterName which\n  updates the global domain default. This prevents operators from updating regional\n  configurations without affecting the primary cluster designation. This change allows\n  attribute updates to be independent of active cluster selection.\"\n- Bad: \"Improves domain handling\" -->\n**Why?**\n\n\n<!-- Include specific test commands and setup. Please include the exact commands such that\nanother maintainer or contributor can reproduce the test steps taken. \n- e.g Unit test commands with exact invocation\n  `go test -v ./common/types/mapper/proto -run TestFailoverDomainRequest`\n- For integration tests include setup steps and test commands\n  Example: \"Started local server with `./cadence start`, then ran `make test_e2e`\"\n- For local simulation testing include setup steps for the server and how you ran the tests\n- Good: Full commands that reviewers can copy-paste to verify\n- Bad: \"Tested locally\" or \"Added tests\" -->\n**How did you test it?**\n\n\n<!-- If there are risks that the release engineer should know about document them here. \nFor example:\n- Has an API/IDL been modified? Is it backwards/forwards compatible? If not, what are the repecussions? \n- Has a schema change been introduced? Is it possible to roll back?\n- Has a feature flag been re-used for a new purpose? \n- Is there a potential performance concern? Is the change modifying core task processing logic? \n- If truly N/A, you can mark it as such -->\n**Potential risks**\n\n\n<!-- If this PR completes a user facing feature or changes functionality add release notes here.\nYour release notes should allow a user and the release engineer to understand the changes with little context.\nAlways ensure that the description contains a link to the relevant GitHub issue. -->\n**Release notes**\n\n\n<!-- Consider whether this change requires documentation updates in the Cadence-Docs repo\n- If yes: mention what needs updating (or link to docs PR in cadence-docs repo)\n- If in doubt, add a note about potential doc needs\n- Only mark N/A if you're certain no docs are affected -->\n**Documentation Changes**\n\n\n"
  },
  {
    "path": ".github/pull_request_template.md",
    "content": "<!-- If you are new to contributing or want a refresher, please read ./pull_request_guidance.md -->\n**What changed?**\n\n\n**Why?**\n\n\n**How did you test it?**\n\n\n**Potential risks**\n\n\n**Release notes**\n\n\n**Documentation Changes**\n\n"
  },
  {
    "path": ".github/workflows/breaking_change_pr_template.md",
    "content": "**Detailed Description**\n[In-depth description of the changes made to the schema or interfaces, specifying new fields, removed fields, or modified data structures]\n\n**Impact Analysis**\n- **Backward Compatibility**: [Analysis of backward compatibility]\n- **Forward Compatibility**: [Analysis of forward compatibility]\n\n**Testing Plan**\n- **Unit Tests**: [Do we have unit test covering the change?]\n- **Persistence Tests**: [If the change is related to a data type which is persisted, do we have persistence tests covering the change?]\n- **Integration Tests**: [Do we have integration test covering the change?]\n- **Compatibility Tests**: [Have we done tests to test the backward and forward compatibility?]\n\n**Rollout Plan**\n- What is the rollout plan?\n- Does the order of deployment matter?\n- Is it safe to rollback? Does the order of rollback matter?\n- Is there a kill switch to mitigate the impact immediately?\n"
  },
  {
    "path": ".github/workflows/breaking_change_reminder.yml",
    "content": "name: Workflow for Breaking Change Reminder\non:\n  pull_request:\n    paths:\n      - schema/**\n      - client/frontend/interface.go\n      - client/history/interface.go\n      - client/matching/interface.go\n      - client/admin/interface.go\n\njobs:\n  breaking-change-pr-template-reminder:\n    runs-on: ubuntu-latest\n\n    steps:\n    - name: Checkout\n      uses: actions/checkout@v4\n      with:\n        fetch-depth: 0\n\n    - name: Fail if PR description is missing breaking change template\n      if: steps.pr-changes.outputs.changes != '[]'\n      run: |\n        PR_NUMBER=${{ github.event.pull_request.number }}\n        PR_URL=\"https://api.github.com/repos/${{ github.repository }}/pulls/${PR_NUMBER}\"\n        BODY=$(curl $PR_URL | jq '.body')\n        CHECKLIST=(\n          \"Detailed Description\"\n          \"Impact Analysis\"\n          \"Testing Plan\"\n          \"Rollout Plan\"\n        )\n        TEMPLATE=$(cat .github/workflows/breaking_change_pr_template.md)\n\n        for i in \"${CHECKLIST[@]}\"; do\n            if [[ \"$BODY\" == *\"$i\"* ]]; then\n                continue\n            else\n                echo \"Potential breaking changes detected! Please update the PR description to include following template:\"\n                echo \"---\"\n                echo \"$TEMPLATE\"\n                echo \"---\"\n                exit 1\n            fi\n        done\n"
  },
  {
    "path": ".github/workflows/ci-checks.yml",
    "content": "name: CI Checks\non: \n  push:\n  pull_request:\n\njobs:\n  idl-submodule-points-to-master:\n    name: IDL submodule points to master\n    runs-on: ubuntu-latest # uses ubuntu as runner\n    \n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true # ensures git submodules are intialized and updated\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Check IDL submodule status (must point to master)\n        run: make .idl-status\n\n  golang-unit-test:\n    name: Golang unit test\n    runs-on: ubuntu-latest\n    \n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n      \n      - name: Run unit test\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run unit-test bash -c \"make .just-build && make cover_profile && ./scripts/github_actions/gen_coverage_metadata.sh .build/coverage/metadata.txt\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-unit-test-coverage\n          path: |\n            .build/coverage/*.out\n            .build/coverage/metadata.txt\n\n\n  golangci-lint-validate-code-is-clean:\n    name: Golangci lint validate code is clean\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run golint\n        run: docker compose -f docker/github_actions/docker-compose.yml run coverage-report bash -c \"./scripts/github_actions/golint.sh\"\n\n\n  golang-integration-test-with-cassandra:\n    name: Golang integration test with cassandra\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for cassandra\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-cassandra bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-integration-coverage\n          path: .build/coverage/*.out\n\n  golang-integration-test-with-cassandra-running-history-queue-v2:\n    name: Golang integration test with running history queue v2\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for cassandra running history queue v2\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-cassandra-queue-v2 bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-running-history-queue-v2-integration-coverage\n          path: .build/coverage/*.out\n\n  golang-integration-test-with-cassandra-running-history-queue-v2-alert:\n    name: Golang integration test with running history queue v2 with alert\n    runs-on: ubuntu-latest\n    continue-on-error: true\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for cassandra running history queue v2 with alert\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-cassandra-queue-v2-alert bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-running-history-queue-v2-alert-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-test-with-cassandra-and-elasticsearch-v7:\n    name: Golang integration test with cassandra and elasticsearch v7\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for cassandra and elasticsearch v7\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose-es7.yml run integration-test-cassandra bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-and-elasticsearch-v7-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-test-with-cassandra-and-pinot:\n    name: Golang integration test with cassandra and pinot\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration test with cassandra and pinot\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose-pinot.yml run integration-test-cassandra-pinot bash -c \"mkdir -p .build/coverage && make .just-build && go test -timeout 600s -run ^TestPinotIntegrationSuite -tags pinotintegration -count 1 -v -coverprofile=.build/coverage/pinot.out github.com/uber/cadence/host\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-pinot-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-test-with-cassandra-with-opensearch-v2:\n    name: Golang integration test with cassandra with opensearch v2\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for cassandra and opensearch v2\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose-opensearch2.yml run integration-test-cassandra bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-and-opensearch-v2-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-ndc-test-with-cassandra:\n    name: Golang integration ndc test with cassandra\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run ndc profile for cassandra\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-ndc-cassandra bash -c \"make .just-build && make cover_ndc_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-cassandra-ndc-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-test-with-mysql:\n    name: Golang integration test with mysql\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for mysql\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-mysql bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-mysql-integration-coverage\n          path: .build/coverage/*.out\n          \n\n  golang-integration-ndc-test-with-mysql:\n    name: Golang integration ndc test with mysql\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run ndc profile for mysql\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-ndc-mysql bash -c \"make .just-build && make cover_ndc_profile\"\n      \n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-mysql-ndc-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-test-with-postgres:\n    name: Golang integration test with postgres\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n          fetch-depth: 0 # get full history for branch checking\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for postgres\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-postgres bash -c \"make .just-build && make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-postgres-integration-coverage\n          path: .build/coverage/*.out\n             \n\n  golang-integration-test-with-sqlite:\n    name: Golang integration test with sqlite\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration profile for sqlite\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-sqlite bash -c \"make cover_integration_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-sqlite-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-integration-ndc-test-with-postgres:\n    name: Golang integration ndc test with postgres\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run ndc profile for postgres\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-ndc-postgres bash -c \"make .just-build && make cover_ndc_profile\"\n\n      - name: Upload coverage artifacts\n        if: always()\n        uses: actions/upload-artifact@v4\n        with:\n          name: go-postgres-ndc-integration-coverage\n          path: .build/coverage/*.out\n\n\n  golang-async-wf-integration-test-with-kafka:\n    name: Golang async wf integration test with kafka\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run async wf integration test with kafka\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-async-wf bash -c \"make .just-build && go test -timeout 60s -run ^TestAsyncWFIntegrationSuite -tags asyncwfintegration -count 1 -v github.com/uber/cadence/host\"\n\n  golang-integration-test-with-etcd:\n    if: github.event_name == 'pull_request'\n    name: Golang integration test with etcd\n    runs-on: ubuntu-latest\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run integration test with etcd\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 30\n          command: |\n            docker compose -f docker/github_actions/docker-compose.yml run integration-test-with-etcd bash -c \"make .just-build && make integration_tests_etcd\"\n"
  },
  {
    "path": ".github/workflows/codecov-on-master.yml",
    "content": "name: Workflow for Codecov integration\non: push\n\njobs:\n  codecov:\n    if: github.repository == 'cadence-workflow/cadence'\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          fetch-depth: 0\n      - name: Set up Go\n        uses: actions/setup-go@v4\n        with:\n          go-version: 1.23.4\n          cache: false\n      - name: Download dependencies\n        run: go mod download\n      - name: Test and generate coverage report\n        run: make cover_profile\n      - name: Upload coverage reports to Codecov\n        uses: codecov/codecov-action@v4.5.0 # https://github.com/codecov/codecov-action\n        with:\n          file: .build/coverage/unit_cover.out\n          exclude: ./\n          token: ${{ secrets.CODECOV_TOKEN }}\n          slug: cadence-workflow/cadence\n"
  },
  {
    "path": ".github/workflows/codecov-on-pr.yml",
    "content": "name: Workflow for Codecov integration\non: pull_request\n\njobs:\n  codecov:\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          fetch-depth: 0\n\n      - name: Set up Go\n        uses: actions/setup-go@v4\n        with:\n          go-version: 1.23.4\n          cache: false\n\n      - name: Download dependencies\n        run: go mod download\n\n      - name: Test and generate coverage report\n        run: make cover_profile\n\n      - name: Upload coverage reports to Codecov\n        uses: codecov/codecov-action@v4.5.0 # https://github.com/codecov/codecov-action\n        with:\n          file: .build/coverage/unit_cover.out\n          exclude: ./\n          token: ${{ secrets.CODECOV_TOKEN }}\n          slug: cadence-workflow/cadence\n"
  },
  {
    "path": ".github/workflows/docker_publish.yml",
    "content": "name: Docker Build & Push\n\non:\n  pull_request:\n  push:\n    branches:\n      - master\n  release:\n    types:\n      - released\n\n\njobs:\n  # This job exposes image_tag and push_enabled variables for other jobs.\n  # Other jobs depends on this. see needs: meta in other jobs.\n  meta:\n    if: github.repository == 'cadence-workflow/cadence'\n    runs-on: ubuntu-latest\n    outputs:\n      image_tag: ${{ steps.determine-image-tag.outputs.image_tag }}\n      push_enabled: ${{ steps.determine-push.outputs.enabled }}\n    steps:\n      # this step determines the image tag based on the event that triggered the workflow.\n      # it sets the image_tag output variable to either \"master\" or the release version.\n      - name: Determine image tag\n        id: determine-image-tag\n        run: |\n          if [[ \"${{ github.event_name }}\" == \"push\" ]]; then\n            echo \"image_tag=master\" >> $GITHUB_OUTPUT\n          elif [[ \"${{ github.event_name }}\" == \"release\" ]]; then\n            echo \"image_tag=${{ github.event.release.tag_name }}\" >> $GITHUB_OUTPUT\n          else\n            echo \"image_tag=test\" >> $GITHUB_OUTPUT\n          fi\n\n      # This step determines if push is enabled based on the event that triggered the workflow. Push is disabled on pull requests.\n      - name: Determine if push is enabled\n        id: determine-push\n        run: |\n          if [[ \"${{ github.event_name }}\" == \"pull_request\" ]]; then\n            echo \"enabled=false\" >> $GITHUB_OUTPUT\n          else\n            echo \"enabled=true\" >> $GITHUB_OUTPUT\n          fi\n\n  push_server_image:\n    name: Push server image to Docker Hub\n    needs: meta\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Login to Docker Hub\n        if: ${{ needs.meta.outputs.push_enabled == 'true' }}\n        uses: docker/login-action@v3\n        with:\n          username: ${{ secrets.CADENCE_SERVER_DOCKERHUB_USERNAME }}\n          password: ${{ secrets.CADENCE_SERVER_DOCKERHUB_TOKEN }}\n      - name: Set up QEMU\n        uses: docker/setup-qemu-action@v3\n      - name: Set up Docker Buildx\n        uses: docker/setup-buildx-action@v3\n      - name: Build and push\n        uses: docker/build-push-action@v6\n        with:\n          context: .\n          platforms: linux/amd64, linux/arm64\n          build-args: TARGET=server\n          push: ${{ needs.meta.outputs.push_enabled == 'true' }}\n          tags: ubercadence/server:${{ needs.meta.outputs.image_tag }}\n      - name: Validate tag\n        run: |\n          echo \"ubercadence/server:${{ needs.meta.outputs.image_tag }}\"\n\n  push_server_auto_setup_image:\n    name: Push auto-setup image to Docker Hub\n    needs: meta\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Login to Docker Hub\n        if: ${{ needs.meta.outputs.push_enabled == 'true' }}\n        uses: docker/login-action@v3\n        with:\n          username: ${{ secrets.CADENCE_SERVER_DOCKERHUB_USERNAME }}\n          password: ${{ secrets.CADENCE_SERVER_DOCKERHUB_TOKEN }}\n      - name: Set up QEMU\n        uses: docker/setup-qemu-action@v3\n      - name: Set up Docker Buildx\n        uses: docker/setup-buildx-action@v3\n      - name: Build and push\n        uses: docker/build-push-action@v6\n        with:\n          context: .\n          platforms: linux/amd64, linux/arm64\n          build-args: TARGET=auto-setup\n          push: ${{ needs.meta.outputs.push_enabled == 'true' }}\n          tags: ubercadence/server:${{ needs.meta.outputs.image_tag }}-auto-setup\n      - name: Validate tag\n        run: |\n          echo \"ubercadence/server:${{ needs.meta.outputs.image_tag }}-auto-setup\"\n\n\n  push_cli_image:\n    name: Push CLI image to Docker Hub\n    needs: meta\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Login to Docker Hub\n        uses: docker/login-action@v3\n        if: ${{ needs.meta.outputs.push_enabled == 'true' }}\n        with:\n          username: ${{ secrets.CADENCE_SERVER_DOCKERHUB_USERNAME }}\n          password: ${{ secrets.CADENCE_SERVER_DOCKERHUB_TOKEN }}\n      - name: Set up QEMU\n        uses: docker/setup-qemu-action@v3\n      - name: Set up Docker Buildx\n        uses: docker/setup-buildx-action@v3\n      - name: Build and push\n        uses: docker/build-push-action@v6\n        with:\n          context: .\n          platforms: linux/amd64, linux/arm64\n          build-args: TARGET=cli\n          push: ${{ needs.meta.outputs.push_enabled == 'true' }}\n          tags: ubercadence/cli:${{ needs.meta.outputs.image_tag }}\n      - name: Validate tag\n        run: |\n          echo \"ubercadence/cli:${{ needs.meta.outputs.image_tag }}\"\n\n  push_bench_image:\n    name: Push bench image to Docker Hub on master pushes\n    needs: meta\n    runs-on: ubuntu-latest\n    if: ${{ needs.meta.outputs.image_tag == 'master' }}\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Login to Docker Hub\n        uses: docker/login-action@v3\n        if: ${{ needs.meta.outputs.push_enabled == 'true' }}\n        with:\n          username: ${{ secrets.CADENCE_SERVER_DOCKERHUB_USERNAME }}\n          password: ${{ secrets.CADENCE_SERVER_DOCKERHUB_TOKEN }}\n      - name: Set up QEMU\n        uses: docker/setup-qemu-action@v3\n      - name: Set up Docker Buildx\n        uses: docker/setup-buildx-action@v3\n      - name: Build and push\n        uses: docker/build-push-action@v6\n        with:\n          context: .\n          platforms: linux/amd64, linux/arm64\n          build-args: TARGET=bench\n          tags: ubercadence/bench:master\n          push: ${{ needs.meta.outputs.push_enabled == 'true' }}\n\n  push_canary_image:\n    name: Push canary image to Docker Hub on master pushes\n    needs: meta\n    runs-on: ubuntu-latest\n    if: ${{ needs.meta.outputs.image_tag == 'master' }}\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Login to Docker Hub\n        uses: docker/login-action@v3\n        if: ${{ needs.meta.outputs.push_enabled == 'true' }}\n        with:\n          username: ${{ secrets.CADENCE_SERVER_DOCKERHUB_USERNAME }}\n          password: ${{ secrets.CADENCE_SERVER_DOCKERHUB_TOKEN }}\n      - name: Set up QEMU\n        uses: docker/setup-qemu-action@v3\n      - name: Set up Docker Buildx\n        uses: docker/setup-buildx-action@v3\n      - name: Build and push\n        uses: docker/build-push-action@v6\n        with:\n          context: .\n          platforms: linux/amd64, linux/arm64\n          build-args: TARGET=canary\n          tags: ubercadence/canary:master\n          push: ${{ needs.meta.outputs.push_enabled == 'true' }}\n"
  },
  {
    "path": ".github/workflows/issue-validation.yml",
    "content": "name: Issue Validation\n\non:\n  issues:\n    types:\n      - opened\n      - edited\n      - labeled\n      - unlabeled\n      - reopened\n\npermissions:\n  issues: write\n\njobs:\n  validate-issue:\n    runs-on: ubuntu-latest\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n      - name: Validate issue template requirements\n        uses: actions/github-script@v7\n        with:\n          script: |\n            const fs = require(\"fs\");\n            const path = require(\"path\");\n\n            const issue = context.payload.issue;\n            const issueNumber = issue.number;\n            const owner = context.repo.owner;\n            const repo = context.repo.repo;\n\n            const TYPE_LABELS = [\"bug\", \"feature\", \"improvement\", \"refractoring\"];\n            const BUG_LABEL = \"bug\";\n            const ENHANCEMENT_LABELS = [\"feature\", \"improvement\", \"refractoring\"];\n            const NEEDS_INFO_LABEL = \"needs-info\";\n\n            const FALLBACK_BUG_REQUIRED = [\n              \"Description\",\n              \"Steps to Reproduce / How to Trigger\",\n              \"Expected Behavior\",\n              \"Actual Behavior\",\n              \"Logs / Screenshots\",\n              \"Environment\",\n            ];\n\n            const FALLBACK_ENHANCEMENT_REQUIRED = [\n              \"Description\",\n              \"Is this a breaking change?\",\n              \"Scope of the feature (server, specific client, all clients)\",\n            ];\n\n            const normalize = (title) =>\n              title\n                .toLowerCase()\n                .replace(/[^\\w\\s]+/g, \" \")\n                .replace(/\\s+/g, \" \")\n                .trim();\n\n            const labelNames = (issue.labels || [])\n              .map((label) => (typeof label === \"string\" ? label : label.name))\n              .filter(Boolean)\n              .map((label) => label.toLowerCase());\n\n            let hasTypeLabel = TYPE_LABELS.some((label) => labelNames.includes(label));\n            let isBug = labelNames.includes(BUG_LABEL);\n            let isEnhancement = ENHANCEMENT_LABELS.some((label) => labelNames.includes(label));\n            let missingLabel = !hasTypeLabel;\n\n            // Detect issue type from template usage and auto-apply label\n            const detectIssueType = (body) => {\n              if (!body) return null;\n              const normalizedBody = body.toLowerCase();\n              \n              // Check for bug template sections\n              if (normalizedBody.includes(\"steps to reproduce\") || \n                  normalizedBody.includes(\"expected behavior\") || \n                  normalizedBody.includes(\"actual behavior\")) {\n                return \"bug\";\n              }\n              \n              // Check for feature template sections\n              if (normalizedBody.includes(\"breaking change\") || \n                  normalizedBody.includes(\"scope of the feature\")) {\n                return \"feature\";\n              }\n              \n              return null;\n            };\n\n            const detectedType = detectIssueType(issue.body);\n            \n            // Auto-apply type label if missing but detected\n            if (missingLabel && detectedType) {\n              await github.rest.issues.addLabels({\n                owner,\n                repo,\n                issue_number: issueNumber,\n                labels: [detectedType],\n              });\n              // Refresh label check\n              labelNames.push(detectedType);\n              hasTypeLabel = TYPE_LABELS.some((label) => labelNames.includes(label));\n              isBug = labelNames.includes(BUG_LABEL);\n              isEnhancement = ENHANCEMENT_LABELS.some((label) => labelNames.includes(label));\n              missingLabel = !hasTypeLabel;\n            }\n\n            const readTemplateHeadings = (fileName) => {\n              try {\n                const templatePath = path.join(\n                  process.env.GITHUB_WORKSPACE || process.cwd(),\n                  \".github\",\n                  \"ISSUE_TEMPLATE\",\n                  fileName\n                );\n                const contents = fs.readFileSync(templatePath, \"utf8\");\n                const headings = [];\n                for (const line of contents.split(/\\r?\\n/)) {\n                  const match = line.match(/^###\\s+(.+?)\\s*$/);\n                  if (match?.[1]) {\n                    headings.push(match[1].trim());\n                  }\n                }\n                return headings.length > 0 ? headings : null;\n              } catch {\n                return null;\n              }\n            };\n\n            const BUG_REQUIRED =\n              readTemplateHeadings(\"bug_report.md\") || FALLBACK_BUG_REQUIRED;\n            const ENHANCEMENT_REQUIRED =\n              readTemplateHeadings(\"feature_request.md\") ||\n              FALLBACK_ENHANCEMENT_REQUIRED;\n\n            const requiredSections = isBug\n              ? BUG_REQUIRED\n              : isEnhancement\n                ? ENHANCEMENT_REQUIRED\n                : [];\n\n            const parseSections = (body) => {\n              const sections = new Map();\n              if (!body) {\n                return sections;\n              }\n              const lines = body.split(/\\r?\\n/);\n              let currentKey = null;\n              let buffer = [];\n\n              const flush = () => {\n                if (currentKey) {\n                  sections.set(currentKey, buffer.join(\"\\n\").trim());\n                }\n                buffer = [];\n              };\n\n              for (const line of lines) {\n                const trimmed = line.trim();\n                const headingMatch = trimmed.match(/^#{1,6}\\s*(.+?)\\s*$/);\n                const boldMatch = trimmed.match(/^\\*\\*(.+?)\\*\\*$/);\n                const colonMatch = trimmed.match(/^(.+?):\\s*$/);\n                const title = headingMatch?.[1] || boldMatch?.[1] || colonMatch?.[1];\n\n                if (title) {\n                  flush();\n                  currentKey = normalize(title);\n                  continue;\n                }\n                if (currentKey) {\n                  buffer.push(line);\n                }\n              }\n\n              flush();\n              return sections;\n            };\n\n            const hasMeaningfulContent = (content) => {\n              if (!content) {\n                return false;\n              }\n              const cleaned = content\n                .replace(/<!--[\\s\\S]*?-->/g, \"\")\n                .replace(/[\\s*-]+/g, \" \")\n                .trim()\n                .toLowerCase();\n              if (!cleaned) {\n                return false;\n              }\n              return ![\"n/a\", \"na\", \"none\", \"tbd\", \"todo\"].includes(cleaned);\n            };\n\n            const sections = parseSections(issue.body || \"\");\n            const missingSections = requiredSections.filter((section) => {\n              const content = sections.get(normalize(section));\n              return !hasMeaningfulContent(content);\n            });\n\n            const ensureNeedsInfoLabelExists = async () => {\n              try {\n                await github.rest.issues.getLabel({\n                  owner,\n                  repo,\n                  name: NEEDS_INFO_LABEL,\n                });\n              } catch (error) {\n                if (error.status !== 404) {\n                  throw error;\n                }\n                await github.rest.issues.createLabel({\n                  owner,\n                  repo,\n                  name: NEEDS_INFO_LABEL,\n                  color: \"f9d0c4\",\n                  description: \"Needs additional information from the reporter\",\n                });\n              }\n            };\n\n            const findExistingBotComment = async () => {\n              const comments = await github.paginate(\n                github.rest.issues.listComments,\n                { owner, repo, issue_number: issueNumber, per_page: 100 }\n              );\n              return comments.find(\n                (comment) =>\n                  comment.user?.type === \"Bot\" &&\n                  comment.body?.includes(\"<!-- issue-template-check -->\")\n              );\n            };\n\n            const updateOrCreateComment = async (body) => {\n              const existing = await findExistingBotComment();\n              if (existing) {\n                await github.rest.issues.updateComment({\n                  owner,\n                  repo,\n                  comment_id: existing.id,\n                  body,\n                });\n                return;\n              }\n              await github.rest.issues.createComment({\n                owner,\n                repo,\n                issue_number: issueNumber,\n                body,\n              });\n            };\n\n            const deleteExistingComment = async () => {\n              const existing = await findExistingBotComment();\n              if (!existing) {\n                return;\n              }\n              await github.rest.issues.deleteComment({\n                owner,\n                repo,\n                comment_id: existing.id,\n              });\n            };\n\n            if (missingLabel || missingSections.length > 0) {\n              await ensureNeedsInfoLabelExists();\n              if (!labelNames.includes(NEEDS_INFO_LABEL)) {\n                await github.rest.issues.addLabels({\n                  owner,\n                  repo,\n                  issue_number: issueNumber,\n                  labels: [NEEDS_INFO_LABEL],\n                });\n              }\n\n              const templateLines = [\n                \"Issue templates:\",\n                `- Bug report: https://github.com/${owner}/${repo}/issues/new?template=bug_report.md`,\n                `- Feature request (use for feature, improvement, refractoring): https://github.com/${owner}/${repo}/issues/new?template=feature_request.md`,\n              ].join(\"\\n\");\n\n              const missingLabelLine = missingLabel\n                ? `- Add one label from: ${TYPE_LABELS.join(\", \")}`\n                : null;\n\n              const missingSectionsLines =\n                missingSections.length > 0\n                  ? [\n                      \"Missing required sections:\",\n                      ...missingSections.map((section) => `- ${section}`),\n                    ].join(\"\\n\")\n                  : null;\n\n              const messageParts = [\n                \"<!-- issue-template-check -->\",\n                \"Thanks for opening this issue. To help us triage it, please provide the missing details:\",\n                missingLabelLine,\n                missingSectionsLines,\n                templateLines,\n              ].filter(Boolean);\n\n              await updateOrCreateComment(messageParts.join(\"\\n\"));\n              return;\n            }\n\n            if (labelNames.includes(NEEDS_INFO_LABEL)) {\n              await github.rest.issues.removeLabel({\n                owner,\n                repo,\n                issue_number: issueNumber,\n                name: NEEDS_INFO_LABEL,\n              });\n            }\n\n            await deleteExistingComment();\n"
  },
  {
    "path": ".github/workflows/replication-simulation.yml",
    "content": "# This workflow runs replication simulaton scenarios via GithubActions matrix strategy.\n# Scenarios are defined in the `simulation/replication/testdata/replication_simulation_<scenario>.yaml` files.\nname: Replication Simulation\non:\n  push:\n  pull_request:\n\njobs:\n  replication-simulation:\n    name: Replication Simulation (${{ matrix.scenario }})\n    runs-on: ubuntu-latest\n    strategy:\n      fail-fast: false\n      matrix:\n        scenario:\n          - activeactive\n          - activeactive_child\n          - activeactive_invalid_cluster_attribute\n          - activeactive_same_wfid\n          - activeactive_same_wfid_signalwithstart\n          - activeactive_same_wfid_signalwithstart_delayed\n          - activeactive_start_terminateifrunning\n          - activeactive_cron\n          - activeactive_regional_failover\n          - activeactive_regional_failover_start_same_wfid\n          - activeactive_regional_failover_start_same_wfid_2\n          - activepassive_to_activeactive\n          - budget_manager\n          - clusterredirection\n          - default\n\n    steps:\n      - name: Checkout\n        uses: actions/checkout@v4\n        with:\n          submodules: true\n\n      - name: Setup Go environment\n        uses: actions/setup-go@v5\n        with:\n          go-version: '1.23.4'\n\n      - name: Run simulation\n        uses: nick-fields/retry@v3\n        with:\n          max_attempts: 2\n          timeout_minutes: 20\n          command: |\n            ./simulation/replication/run.sh --scenario ${{ matrix.scenario }}\n\n      - name: Upload test logs\n        uses: actions/upload-artifact@v4\n        with:\n          name: replication-${{ matrix.scenario }}-test.log\n          path: ./test.log\n"
  },
  {
    "path": ".github/workflows/semantic-pr.yml",
    "content": "name: Semantic Pull Request\n\non:\n  pull_request:\n    types:\n      - opened\n      - edited\n      - synchronize\n\njobs:\n  semantic-pr:\n    name: Validate PR title follows conventional commit format\n    runs-on: ubuntu-latest\n    # TODO: Remove this once we commit to conventional commits\n    continue-on-error: true\n\n    steps:\n      - name: Validate PR title\n        id: lint_pr_title\n        uses: amannn/action-semantic-pull-request@v6.1.1\n        with:\n          # Allow standard conventional commit types\n          types: |\n            fix\n            feat\n            docs\n            style\n            refactor\n            perf\n            test\n            chore\n            ci\n            build\n            revert\n          # TODO: Remove this once we've decided on scopes\n          requireScope: false\n          # Skip validation for certain labels if needed\n          ignoreLabels: |\n            skip-commit-format\n        env:\n          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n\n"
  },
  {
    "path": ".gitignore",
    "content": "*.out\n*.test\n*.xml\n*.swp\n.idea/\n*.iml\n*.cov\n*.html\n.gobincache/\n.tmp/\n.vscode/\n.build/\n.bin/\n.cursor/\n/vendor\n/_vendor-*\n/cadence\n.DS_Store\n*.log\ntest\ntest_xdc\ntest_e2e\ntest_eventsV2\ntest_eventsV2_xdc\nmatching-simulator-output/\nhistory-simulator-output/\nreplication-simulator-output/\n**/Dockerfile.local*\n**/Dockerfile.light*\ndockerize\n*.crt\n\n# Executables produced by cadence repo\n/cadence\n/cadence-server\n/cadence-canary\n/cadence-bench\n/cadence-cassandra-tool\n/cadence-sql-tool\n/cadence-releaser\n/sharddistributor-canary\n\n# SQLite databases\ncadence.db*\ncadence_visibility.db*\n"
  },
  {
    "path": ".gitmodules",
    "content": "[submodule \"idls\"]\n\tpath = idls\n\turl = https://github.com/cadence-workflow/cadence-idl.git\n\tbranch = master\n"
  },
  {
    "path": "ADOPTERS.md",
    "content": "# ADOPTERS\n\nThis document lists organizations, companies, and individuals that are using Cadence in production or evaluation environments.\nIf you are using Cadence, please consider adding yourself to this list by opening a pull request or contacting the maintainers. See more instructions at the end of this page.\nYour support helps us understand our user community and prioritize improvements.\n\n## Production Users\n\n![using.png](./logos/Uber.jpg)\n\n### Uber Technologies\n\nUber has over 2000 use cases (Cadence domains) powered by Cadence. Here are some examples:\n\n- **Infrastructure**: service & config rollout, CD pipelines, health checks & rollbacks, DB maintenance, bad node detection / replacement, compaction, DB backup / restore, migration orchestration, ...\n- **ML / AI**: Agent orchestration, model training, data pipelines, fare estimation, ...\n- **Finance**: payments, tips, disputes, bank verification, account onboarding, promotions, discounts, partnerships, gifting, billing / invoicing, loyalty points, ...\n- **Product**: earner eligibility, onboarding, periodic document & compliance checks, document processing, vehicle management, inventory / catalog management, image processing, ads, repeat orders, vouchers, spending tracking, notifications, analytics, trip reports, ...\n\nThere are 20+ environments running Cadence where some of them hosts 400+ domains.\n\n![using.png](./logos/netapp-logo.jpg)\n\n### NetApp\n\nAt [NetApp](https://www.netapp.com) we use Cadence as a key component of the Instaclustr platform’s control plane architecture. Our automated maintenance system relies on Cadence to orchestrate and schedule maintenance across tens of thousands of hosts in our managed fleet. By adopting Cadence we’ve been able to offer both our engineers and customers high-throughput, fault-tolerant execution of critical fleet operations.\n\nNetApp is the intelligent data infrastructure company, providing solutions to manage any data, for any application, anywhere it is needed. Through the [Instaclustr](https://www.instaclustr.com) platform we offer fully managed Cadence, including automated provisioning of supporting infrastructure such as Apache Cassandra, Apache Kafka, and OpenSearch.\n\n![using.png](./logos/DoorDash-logo.jpg)\n\n### DoorDash, Inc\nDoorDash uses Cadence across a wide range of product and infrastructure use cases, including ETA, Fulfillment, Order Management, Storage, Catalog, Photos, Ads and more.\nDoorDash’ use cases take advantage of Cadence’s orchestration for long running, fault tolerant workflows.\n\n## Evaluating / Early Adopters\n\n## How to Add Your Organization\n\n- Fork this repository.\n- Edit this file (ADOPTERS.md).\n- Add your organization under the relevant section.\n\n### Adding Your Logo\n\n- If you would like your organization’s logo to appear on the Cadence website, repositories and documentation:\n- Add your logo file (preferably in JPG/JPEG format) to the /logos directory in the repository.\n- Name the file using lowercase letters and hyphens (for example: acme-corp.svg).\n- Keep your image dimensions less than 300 pixels in width and 200 pixels in height.\n- Update ADOPTERS.md to include your logo and link to your organization’s website.\n- Logos should be submitted only by verified representatives or with permission from the organization.\n\nOpen a pull request with the title: Add \\<Your Organization\\> to ADOPTERS.md.\n"
  },
  {
    "path": "CHANGELOG.md",
    "content": "# Changelog\nAll notable changes to this project will be documented in this file.\n\nThe format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),\nand this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).\n\nYou can find a list of previous releases on the [github releases](https://github.com/cadence-workflow/cadence/releases) page.\n\n## [Note]\nThere's a new opt-in feature for autoscale of tasklist partitions. It's optional but recommended for large scale use cases. Please refer to [tasklist-partition-config.md](https://github.com/cadence-workflow/cadence/blob/master/docs/migration/tasklist-partition-config.md) for additional details on the migration and its rationale.\n\n## [1.3.3] - 2025-08-06\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.3.3) for details\n\n## [1.3.2] - 2025-07-03\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.3.2) for details\n\n## [1.3.1] - 2025-06-11\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.3.1) for details\n\n## [1.3.0] - 2025-05-14\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.3.0) for details\n\n## [1.2.18] - 2025-04-03\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.18) for details\n\n## [1.2.17] - 2025-03-05\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.17) for details\n\n## [1.2.16] - 2025-02-19\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.16) for details\n\n## [1.2.15] - 2025-01-22\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.15) for details\n\n## [1.2.14] - 2024-11-13\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.14) for details\n\n## [1.2.13] - 2024-09-25\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.13) for details\n\n## [1.2.12] - 2024-08-19\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.12) for details\n\n## [1.2.11] - 2024-07-10\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.11) for details\n\n## [1.2.10] - 2024-06-04\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.10) for details\n\n## [1.2.9] - 2024-05-01\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.9) for details\n\n## [1.2.8] - 2024-03-26\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.8) for details\n\n## [1.2.7] - 2024-02-09\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.2.7) for details\n### Upgrade notes\nCadence repo now has multiple submodules,\nthe split and explanation in [PR](https://github.com/cadence-workflow/cadence/pull/5609).\n\nIn principle, \"plugins\" are \"optional\" and we should not be forcing all optional dependencies on all users of any of Cadence.\nSplitting dependencies into choose-your-own-adventure submodules is simply good library design for the ecosystem, and it's something we should be doing more of.\n\n## [1.2.6] - 2023-12-14\n### Added\n- Added range query support for Pinot json index (#5426)\n- Implemented GetTaskListSize method at persistence layer (#5442, #5447)\n- Added a framework for the Task validator service (#5446)\n- Added nit comments describing the Update workflow cycle (#5432)\n- Added log user query param (#5437)\n- Added CODEOWNERS file (#5453)\n- Added a function to evict all elements older than the cache TTL (#5464)\n\n### Fixed\n- Fixed workflow replication for reset workflow (#5412)\n- Fixed visibility mode for admin when use Pinot visibility (#5441)\n- Fixed workflow started metric (#5443)\n- Fixed timer-fixer, unfortunately broken in 1.2.5 (#5433)\n- Fixed confusing comment in matching handler (#5450)\n\n### Changed\n- Cassandra version is changed from 3.11 to 4.1.3 (#5461)\n  - If your machine already has ubercadence/server:master-auto-setup image then you need to repull so it works with latest docker-compose*.yml files\n- Move dynamic ratelimiter to its own file (#5451)\n- Create and use a limiter struct instead of just passing a function (#5454)\n- Dynamic ratelimiter factories (#5455)\n- Update github action for image publishing to released (#5460)\n- Update matching to emit metric for tasklist backlog size (#5448)\n- Change variable name from SecondsSinceEpoch into EventTimeMs (#5463)\n\n### Removed\n- Get rid of noisy task adding failure log in matching service (#5445)\n\n## [1.2.5] - 2023-11-01\n\n### Caution:\n\nPrefer 1.2.4 or 1.2.6 if you have enabled the timer fixer.\nThese were broken by #5361, and fixed by #5433.\n\nBy default this fixer is _disabled_, so the version has not been retracted.\n\nIf you have already upgraded to 1.2.5, downgrading or using 1.2.6 should restore the timer fixer.\nOr apply #5433 as a local patch.\n\n### Added\n- Scanner / Fixer changes (#5361)\n  - Stale-workflow detection and cleanup added to shardscanner, disabled by default.\n  - New dynamic config to better control scanner and fixer, particularly for concrete executions.\n  - Documentation about how scanner/fixer work and how to control them, see [the scanner readme.md](service/worker/scanner/README.md)\n    - This also includes example config to enable the new fixer.\n- MigrationChecker interface to expose migration CLI (#5424)\n- Added Pinot as new visibility store option (#5201)\n  - Added pinot visibility triple manager to provide options to write to both ES and Pinot.\n  - Added pinotVisibilityStore and pinotClient to support CRUD operations for Pinot.\n  - Added pinot integration test to set up Pinot test cluster and test Pinot functionality.\n\n### Fixed\n- Fix CreateWorkflowModeContinueAsNew for SQL (#5413)\n- Fix CLI count&list workflows error message (#5417)\n- Hotfix for async matching for isolation-group redirection (#5423)\n- Fix closeStatus for --format flag (#5422)\n\n### Upgrade notes\n- Any \"concrete execution\" or \"timers\" fixers run on upgrade may be missing the new config and those activities will have no invariants for a single run.  Later runs will work normally. (#5361)\n\n## [1.2.4] - 2023-09-27\n### Added\nImplement config store for MySQL (#5403)\nImplement config store for PostgresSQL (#5405)\n\n### Fixed\nRemove database check for config store tests (#5401)\nFix persistence tests setup (#5402)\nRetract v1.2.3 (#5406)\n\n## [1.2.3] - 2023-09-15\n### Added\nExpose workflow history size and count to client (#5392)\n\n### Fixed\n[cadence-cli] fix typo in input flag for parallelism (#5397)\n\n### Changed\nUpdate config store client to support SQL database (#5395)\nScaffold config store for sql plugins (#5396)\nImprove poller detection for isolation (#5399)\n\n## [1.2.2] - 2023-08-29\n### Added\nAdded a update workflow execution count metric for RI (#5386)\n\n### Fixed\nFixed nil pointer issue in domain migration command rendering (#5378)\n\n### Changed\nPass partition config and isolation group to history/matching even if isolation is disabled (#5385)\n\n## [1.2.1] - 2023-08-24\n### Added\nAdded guardrail to on scalable tasklist number of write partitions (#5331)\nAdded Opensearch2 client with bulk API shared between clients (#5241)\nAdded Filters method to dynamic config Key (#5346)\nAdded domain migration validator command (#5369, #5374)\nAdded docker-compose with http API enabled (#5358)\nAdded support to enable TLS for HTTP inbounds (#5381)\n\n### Fixed\n\n### Changed\nUpdates to partition config middleware (#5334)\nAllow to configure HTTP settings using template (#5329)\nCleanup the unused watchdog code (#5096)\nUpgrade yarpc to v1.70.3 (#5341)\nupgrade mysql (#5345)\nchange mysql schema folder v57 to v8 (#5356)\nRemove duplicated line (#5328)\nFill IsCron for proto of WorkflowExecutionInfo (#5366)\nUpdate idls and ensure thrift fields roundtrip (#5365)\nGo version bump (#5367)\nUpdate Dockerfile with a proper Go version and bump alpine version (#5371)\nStartWorkflowExecution: validate RequestID before calling history (#5359)\nUpdate list of available frontend HTTP endpoints (#5338)\nSet a minimum timeout for async task dispatch based on rate limit (#5382)\nValidate search-attribute-key so keys are fine in advanced search (#5340)\n\n## [1.2.0] - 2023-08-24\n### Added\nAdded more logs around unsupported version for consistent query (#5287)\nAdded datasource and dashboards to Grafana in Docker (#5207)\nAdded metric for isolation task matching (#5288)\nAdded local build instructions to Readme (#5299)\nAdded examples for reset-batch and batch commands in help usage section (#5302)\nRecord current worker Identity on Pending activity tasks (#5307)\nSupport invoking RPC using HTTP and JSON (#5305)\nAdded Worklfow start metric (#5289)\nAdded count of workflows indicator when running the listall command and waiting it to complete (#5309)\nAdded option to pass consistency level in the cassandra schema version validation (#5327)\n\n### Fixed\nFixed rendering for isolation-groups in the CLI (#5285)\nFixed SearchAfter usage (#5311)\nFixed garbage collection logic for matching tasks (#5355)\nDo not make poller crash if isolation group is invalid (#5372)\n\n### Changed\nRemoved unused metric (#5292)\nShow error message if requesting workflow does not exist (#5300)\nParse JWT flags when creating a context (#5296)\nSet ReplicatorCacheCapacity to 0 (#5301)\nShow explicit message if command is not found (#5298)\nIDL update to include new field in PendingActivityInfo (#5306)\nSet 12.4 version for postgres containers (#5326)\nExtract EventColorFunction from ColorEvent (#5321)\nUpdate consistency level for cassandra visibility (#5330)\nImprove async-matching performance for isolation (#5363)\n\n## [1.1.0] - 2023-08-24\n### Added\nAdded metrics for delete workflow execution on a shard level (#5126)\nAdded overall persistence count for shardid (#5134)\nAdded Scanner to purge deprecated domain workflows (#5125)\nAdded domain status check in taskfilter (#5140)\nAdded usage for InactiveDomain Invariant (#5144, #5213)\nAdded request body into Attributes for auditing purpose with PII (#5151)\nAdded remaining persistence metrics that goes to a shard #5142\nAdded consistent query pershard metric (#5143, #5170)\nAdded logging with workflow/domain tags (#5159)\nAdded ShardID to valid attributes (#5161)\nAdded Pinot docker files, table config and schema (#5163)\nAdded Canary TLS support (#5086)\nAdded a small test to catch issues with deadlocks (#5171)\nAdded large workflow hot shard detection (#5166)\nAdded thin ES clients (#5162, #5217)\nAdded generic ES query building utilities (#5168)\nAdded Physical sharding for NoSQL-based persistence (#5187)\nAdded tasklist traffic metrics for non-sticky and non-forwarded tasklists (#5218, #5235)\nAdded default value for shard_id and update_time in mysql (#5246)\nAdded default value for shard_id and update_time in postgres (#5259)\nAdded Hostname tag into metrics and other services (#5245)\nAdded Domain name validation (#5250)\nAdded isolation-group types (#5260)\nAdded Dynamic-config type (#5261)\nAdded isolation groups to persistence (#5270)\nAdded helper function to store/retrieve partition config from context (#5195)\nAdded domain handler changes for isolation group (#5274)\nAdded isolation-group and partition libraries (#5262)\nAdded resource implmentation (#5278)\nAdded Admin API for zonal-isolation drain commands (#5282)\nAdded cli operations for the zonal-isolation drains (#5283)\nAdded metric for isolation task matching (#5288)\nAdded some more coverage to isolation-group state handler (#5304)\n\n### Fixed\nRemoved circular dependencies between matching components (#5111)\nFixed InsertTasks query for Cassandra (#5119)\nFixed prometheus frontend label inconsistencies (#5087)\nFixed samples documentation (#5088)\nFixed type validation in configstore DC client value updating (#5110)\nFixed the config-store handling for not-found errors (#5203)\nFixed docker (#5244)\nFixed thrift mapper for DomainConfiguration (#5268)\nFixed panic while parsing workflows with timeouts (#5267)\nFixed sticky tasklist isolation (#5308)\nFixed SearchAfter usage (#5311)\nFixed isolation groups domain drains (#5315)\n\n### Changed\nIndexer: refactor ES processor (#5100)\nMoved sample logger into persistence metric client for cleaness (#5129)\nES: do not set _type when using Bulk API for v7 client (#5104)\nES: single interface for different ES/OpenSearch versions (#5158)\nES: reduce code duplication (#5137)\nUpdated README.md (#5064, #5251)\nSet poll interval for filebased dynamic config if not set (#5160)\nUpgraded Golang base image to 1.18 to remediate CVEs (#5035)\nRefactor matching integration test (#5182)\nMerge Activity and Decision matching tests (#5186)\nUpdate idls version (#5200)\nAllow registering search attributes without Advance Visibility enabled (#5185)\nScaffold config store for SQL (#5239)\nBench: possibility to pass frontend address using env (#5113)\nRemove tcheck dependency (#5247)\nUpdate internal type to adopt idl change (#5253)\nUpdate persistence layer to adopt idl update for isolation (#5254)\nUpdate history to persist partition config (#5257)\nUpgrades IDL to include isolation-groups (#5258)\nMake ESClient fields public (#5269)\nRemove unneeded file (#5276)\nUpdated frontend to adopt draining isolation groups (#5225) (#5281)\nUpdated matching to support tasklist isolation (#5280)\nBumping version to 1.1.0 (#5284)\nImprovements on tasklist isolation (#5291)\nTask processing panic handling (#5294)\nUpdate ClusterNameForFailoverVersion to return error (#5293)\nFeature/isolation group library logging improvements (#5295)\nIncrease number of forward tokens for isolation tasklists (#5310)\nSeperate token pools for tasklists with isolation (#5314)\nDisable isolation for sticky tasklist (#5319)\nChange default value of AsyncTaskDispatchTimeout (#5320)\n\n## [1.0.0] - 2023-04-26\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v1.0.0)\n\n## [0.23.1] - 2021-11-18\nSee [Release Note](https://github.com/cadence-workflow/cadence/releases/tag/v0.23.1)\n\n## [0.21.3] - 2021-07-17\n### Added\n- Added GRPC support. Cadence server will accept requests on both TChannel and GRPC. With dynamic config flag `system.enableGRPCOutbound` it will also switch to GRPC communication internally between server components.\n\n### Fixed\n- This change contains breaking change on user config. The masterClusterName config key is deprecated and is replaced with primaryClusterName key. (#4185)\n\n### Changed\n- Bump CLI version to v0.19.0\n- Change `--connect-attributes` in `cadence-sql-tool` from URL encoding to the format of k1=v1,k2=v2...\n- Change `--domain_data` in `cadence domain update/register` from the format of k1:v1,k2:v2... to the format of k1=v1,k2=v2...\n- Deprecate local domains. All new domains should be created as global domains.\n- Server will deep merge configuration files when loading. No need to copy the whole config section, specify only those fields that needs overriding. (#4165)\n\n## [0.18.0] - 2021-01-22\n\n## [0.16.1] - 2021-01-21\n\n## [0.17.0] - 2021-01-13\n\n## [0.16.0] - 2020-12-10\n"
  },
  {
    "path": "CLAUDE.md",
    "content": "# Development Guidelines\n\nThis document contains critical information about working with this codebase.\n\n## Core Rules\n\n- NEVER ever mention a `co-authored-by` or similar aspects. In particular, never mention the tool used to create the commit message or PR.\n\n## Development Commands\n\n```bash\nmake bins   # build all binaries\nmake build  # compile-check all packages + test files (no codegen, no test execution)\nmake pr     # pre-PR: tidy → go-generate → fmt → lint\nmake pr GEN_DIR=service/history  # scoped codegen (faster for single package)\nmake test   # all unit tests (excludes host/ integration tests)\nmake test_e2e  # end-to-end integration tests in host/ (requires Docker dependencies running)\nmake lint   # lint only\nmake go-generate  # regenerate mocks, enums, wrapper files\n\ngo test -race -run TestFoo ./path/to/pkg/...  # run a specific test\n```\n\n## Gotchas\n\n- **`idls/` submodule**: run `git submodule update --init --recursive` after checkout or all codegen and build targets fail silently.\n- **Generated files**: `*_generated.go` and `*_mock.go` are produced by `make go-generate`.\n  Edit the source `.tmpl` or interface file, then regenerate — never edit generated files directly.\n- **`make pr` is required before every PR**: it runs tidy → go-generate → fmt → lint in sequence.\n  If CI shows unexpected diffs in generated files, you forgot `make pr`. Prefer to use `make pr GEN_DIR=<package>` for faster iteration.\n- **Go workspace gotcha**: `go build ./...` and `go test ./...` only cover the root module.\n  Use `make bins` and `make test` for full coverage. Use `make tidy` (not `go mod tidy`).\n- **IDL local testing**: To test local IDL changes before pushing, add `replace github.com/uber/cadence-idl => ./idls` to the bottom of `go.mod`. Remove before committing.\n- **Submodule drift**: `git submodule update --init --recursive` fixes not just post-checkout failures but also mid-development build errors after upstream IDL changes.\n- **SQLite for quick local dev**: No Docker required. Run `make install-schema-sqlite` then `./cadence-server --zone sqlite start` for the fastest path to a running local server.\n\n## Coding Best Practices\n\n- **Testing**:\n  - Prefer table-tests; plain Go tests for trivially simple cases.\n  - Do **not** write new suite-style tests (`testify/suite`) — legacy, maintain only.\n  - Do **not** use `github.com/stretchr/testify` mocks — use `github.com/uber-go/mock`.\n  - All new tests should be either plain Go tests or table-tests.\n  - Round-trip test all mappers: `ToX(FromX(item)) == item`. Fuzz-test mappers following the pattern in `common/types/mapper/proto/api_test.go` and `schedule_test.go`.\n\n- **Types**:\n  - Never use IDL code (`.gen/go/` or `.gen/proto/`) directly in service logic.\n  - Map to `common/types` or `common/persistence` types via mappers in `common/types/mapper/`.\n  - Files in `.gen/` are generated from IDL — do not edit manually.\n\n## Pull Request Guidelines\n\nPRs must follow the template in `.github/pull_request_guidance.md`.\n\nPR titles must use Conventional Commits format: `<type>(<optional scope>): <description>`.\nValid types are defined in `.github/workflows/semantic-pr.yml`.\nExample: `feat(history): add retry logic for shard takeover`.\n\n## Development\n\nFor database setup, schema installation, and server start options: [CONTRIBUTING.md](CONTRIBUTING.md).\n\n## Repository layout\nA Cadence server cluster is composed of four different services: Frontend, Matching, History and Worker(system).\nHere's what's in each top-level directory in this repository:\n\n* **bench/** : Benchmark and load test suite for stress-testing a Cadence cluster\n* **canary/** : The test code that needs to run periodically to ensure Cadence is healthy\n* **client/** : Client wrappers to let the four different services talk to each other\n* **cmd/** : The main function to build binaries for servers and CLI tools\n* **common/** : Basically contains all the rest of the code in Cadence server, the names of the sub folder are the topics of the packages\n* **config/** : Sample configuration files\n* **docker/** : Code/scripts to build docker images\n* **docs/** : Documentation\n* **environment/** : Test helpers for reading environment variables (DB host/port, etc.) used by integration tests\n* **host/** : End-to-end integration tests\n* **idls/** : Git submodule with Thrift and Protobuf IDL definitions; source for all generated code under `.gen/`\n* **internal/** : Internal Go tool dependencies (blank imports to pin codegen tools in `go.mod`)\n* **proto/** : Protobuf definitions for internal persistence blobs and public Cadence APIs\n* **schema/** : Versioned persistence schema for Cassandra/MySQL/Postgres/ElasticSearch\n* **scripts/** : Scripts for CI build\n* **service/** : Contains four sub-folders dedicated to each of the four services (frontend, history, matching, worker)\n* **simulation/** : Black-box simulation tests that spin up a local Docker cluster and validate complex multi-component scenarios\n* **tools/** : CLI tools for Cadence workflows and also schema updates for persistence"
  },
  {
    "path": "CONTRIBUTING.md",
    "content": "# Developing Cadence\n\nThis doc is intended for contributors to Cadence backend. Thanks for considering to contribute ❤️\n\n> 📚 **New to contributing to Cadence?** Check out our [Contributing Guide](https://cadenceworkflow.io/community/how-to-contribute/getting-started) for an overview of the contribution process across all Cadence repositories. This document contains cadence backend specific setup and development instructions.\n\nOnce you go through the rest of this doc and get familiar with local development setup, take a look at the list of issues labeled with\n[good first issue](https://github.com/cadence-workflow/cadence/labels/good%20first%20issue).\nThese issues are a great way to start contributing to Cadence. Later when you are more familiar with Cadence, look at issues with\n[up-for-grabs](https://github.com/cadence-workflow/cadence/labels/up-for-grabs).\n\nJoin our community on the CNCF Slack workspace at [cloud-native.slack.com](https://communityinviter.com/apps/cloud-native/cncf) in the **#cadence-users** channel to reach out and discuss issues with the team.\n\n## Development Environment\nBelow are the instructions of how to set up a development Environment.\n\n### 1. Building Environment\n\n* Golang. Install on OS X with.\n```\nbrew install go\n```\n* Make sure set PATH to include bin path of GO so that other executables like thriftrw can be found.\n```bash\n# check it first\necho $GOPATH\n# append to PATH\nPATH=$PATH:$GOPATH/bin\n# to confirm, run\necho $PATH\n```\n\n* Download golang dependencies.\n```bash\ngo mod download\n```\n\nAfter check out and go to the Cadence repo, compile the `cadence` service and helper tools without running test:\n\n```bash\ngit submodule update --init --recursive\n\nmake bins\n```\n\nYou should be able to get all the binaries of this repo:\n* cadence-server: the server binary\n* cadence: the CLI binary\n* cadence-cassandra-tool: the Cassandra schema tools\n* cadence-sql-tool: the SQL schema tools(for now only supports MySQL and Postgres)\n* cadence-canary: the canary test binary\n* cadence-bench: the benchmark test binary\n\n\n:warning: Note:\n\nIf running into any compiling issue\n>1. For proto/thrift errors, run `git submodule update --init --recursive` to fix\n>2. Make sure you upgrade to the latest stable version of Golang.\n>3. Check if this document is outdated by comparing with the building steps in [Dockerfile](https://github.com/cadence-workflow/cadence/blob/master/Dockerfile)\n\n### 2. Setup Dependency\nNOTE: you may skip this section if you want to use SQLite or you have installed the dependencies in any other ways, for example, using homebrew.\n\nCadence's core data model can be running with different persistence storages, including Cassandra,MySQL and Postgres.\nPlease refer to [persistence documentation](https://github.com/cadence-workflow/cadence/blob/master/docs/persistence.md) if you want to learn more.\nCadence's visibility data model can be running with either Cassandra/MySQL/Postgres database, or ElasticSearch+Kafka. The latter provides [advanced visibility feature](./docs/visibility-on-elasticsearch.md)\n\nWe recommend to use [docker-compose](https://docs.docker.com/compose/) to start those dependencies:\n\n* If you want to start Cassandra dependency, use `./docker/dev/cassandra.yml`:\n```\ndocker compose -f ./docker/dev/cassandra.yml up\n```\nYou will use `CTRL+C` to stop it. Then `docker compose -f ./docker/dev/cassandra.yml down` to clean up the resources.\n\nOr to run in the background\n```\ndocker compose -f ./docker/dev/cassandra.yml up -d\n```\nAlso use `docker compose -f ./docker/dev/cassandra.yml down` to stop and clean up the resources.\n\n* Alternatively, use `./docker/dev/mysql.yml` for MySQL dependency. (MySQL has been updated from 5.7 to 8.0)\n* Alternatively, use `./docker/dev/postgres.yml` for PostgreSQL dependency\n* Alternatively, use `./docker/dev/cassandra-esv7-kafka.yml` for Cassandra, ElasticSearch(v7) and Kafka/ZooKeeper dependencies\n* Alternatively, use `./docker/dev/mysql-esv7-kafka.yml` for MySQL, ElasticSearch(v7) and Kafka/ZooKeeper dependencies\n* Alternatively, use `./docker/dev/cassandra-opensearch-kafka.yml` for Cassandra, OpenSearch(compatible with ElasticSearch v7) and Kafka/ZooKeeper dependencies\n* Alternatively, use `./docker/dev/mongo-esv7-kafka.yml` for MongoDB, ElasticSearch(v7) and Kafka/ZooKeeper dependencies\n\n### 3. Schema installation\nBased on the above dependency setup, you also need to install the schemas.\n\n* If you use SQLite then run `make install-schema-sqlite` to install SQLite schemas\n* If you use `cassandra.yml` then run `make install-schema` to install Cassandra schemas\n* If you use `cassandra-esv7-kafka.yml` then run `make install-schema && make install-schema-es-v7` to install Cassandra & ElasticSearch schemas\n* If you use `cassandra-opensearch-kafka.yml` then run `make install-schema && make install-schema-es-opensearch` to install Cassandra & OpenSearch schemas\n* If you use `mysql.yml` then run `install-schema-mysql` to install MySQL schemas\n* If you use `postgres.yml` then run `install-schema-postgres` to install Postgres schemas\n* `mysql-esv7-kafka.yml` can be used for single MySQL + ElasticSearch or multiple MySQL + ElasticSearch mode\n  * for single MySQL: run `install-schema-mysql && make install-schema-es-v7`\n  * for multiple MySQL: run `make install-schema-multiple-mysql` which will install schemas for 4 mysql databases and ElasticSearch\n\n:warning: Note:\n>If you use `cassandra-esv7-kafka.yml` and start server before `make install-schema-es-v7`, ElasticSearch may create a wrong index on demand.\nYou will have to delete the wrong index and then run the `make install-schema-es-v7` again. To delete the wrong index:\n```\ncurl -X DELETE \"http://127.0.0.1:9200/cadence-visibility-dev\"\n```\n\n### 4. Run\nOnce you have done all above, try running the local binaries:\n\n\nThen you will be able to run a basic local Cadence server for development.\n\n  * If you use SQLite, then run `./cadence-server --zone sqlite start`, which load , which will load `config/development.yaml` + `config/development_sqlite.yaml` as config\n  * If you use `cassandra.yml`, then run `./cadence-server start`, which will load `config/development.yaml` as config\n  * If you use `mysql.yml` then run `./cadence-server --zone mysql start`, which will load `config/development.yaml` + `config/development_mysql.yaml` as config\n  * If you use `postgres.yml` then run `./cadence-server --zone postgres start` , which will load `config/development.yaml` + `config/development_postgres.yaml` as config\n  * If you use `cassandra-esv7-kafka.yml` then run `./cadence-server --zone es_v7 start`, which will load `config/development.yaml` + `config/development_es_v7.yaml` as config\n  * If you use `cassandra-opensearch-kafka.yml` then run `./cadence-server --zone es_opensearch start` , which will load `config/development.yaml` + `config/development_es_opensearch.yaml` as config\n  * If you use `mysql-esv7-kafka.yaml`\n    * To run with multiple MySQL : `./cadence-server --zone multiple_mysql start`, which will load `config/development.yaml` + `config/development_multiple_mysql.yaml` as config\n\nThen register a domain:\n```\n./cadence --do samples-domain domain register\n```\n\n### Sample Repo\nThe sample code is available in the [Sample repo](https://github.com/cadence-workflow/cadence-samples)\n\nThen run a helloworld from [Go Client Sample](https://github.com/cadence-workflow/cadence-samples) or [Java Client Sample](https://github.com/cadence-workflow/cadence-java-samples)\n\n```\nmake bins\n```\nwill build all the samples.\n\nThen\n```\n./bin/helloworld -m worker &\n```\nwill start a worker for helloworld workflow.\nYou will see like:\n```\n$./bin/helloworld -m worker &\n[1] 16520\n2021-09-24T21:07:03.242-0700\tINFO\tcommon/sample_helper.go:109\tLogger created.\n2021-09-24T21:07:03.243-0700\tDEBUG\tcommon/factory.go:151\tCreating RPC dispatcher outbound\t{\"ServiceName\": \"cadence-frontend\", \"HostPort\": \"127.0.0.1:7933\"}\n2021-09-24T21:07:03.250-0700\tINFO\tcommon/sample_helper.go:161\tDomain successfully registered.\t{\"Domain\": \"samples-domain\"}\n2021-09-24T21:07:03.291-0700\tINFO\tinternal/internal_worker.go:833\tStarted Workflow Worker\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\"}\n2021-09-24T21:07:03.300-0700\tINFO\tinternal/internal_worker.go:858\tStarted Activity Worker\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\"}\n```\nThen\n```\n./bin/helloworld\n```\nto start a helloworld workflow.\nYou will see the result like :\n```\n$./bin/helloworld\n2021-09-24T21:07:06.220-0700\tINFO\tcommon/sample_helper.go:109\tLogger created.\n2021-09-24T21:07:06.220-0700\tDEBUG\tcommon/factory.go:151\tCreating RPC dispatcher outbound\t{\"ServiceName\": \"cadence-frontend\", \"HostPort\": \"127.0.0.1:7933\"}\n2021-09-24T21:07:06.226-0700\tINFO\tcommon/sample_helper.go:161\tDomain successfully registered.\t{\"Domain\": \"samples-domain\"}\n2021-09-24T21:07:06.272-0700\tINFO\tcommon/sample_helper.go:195\tStarted Workflow\t{\"WorkflowID\": \"helloworld_75cf142b-c0de-407e-9115-1d33e9b7551a\", \"RunID\": \"98a229b8-8fdd-4d1f-bf41-df00fb06f441\"}\n2021-09-24T21:07:06.347-0700\tINFO\thelloworld/helloworld_workflow.go:31\thelloworld workflow started\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\", \"WorkflowType\": \"helloWorldWorkflow\", \"WorkflowID\": \"helloworld_75cf142b-c0de-407e-9115-1d33e9b7551a\", \"RunID\": \"98a229b8-8fdd-4d1f-bf41-df00fb06f441\"}\n2021-09-24T21:07:06.347-0700\tDEBUG\tinternal/internal_event_handlers.go:489\tExecuteActivity\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\", \"WorkflowType\": \"helloWorldWorkflow\", \"WorkflowID\": \"helloworld_75cf142b-c0de-407e-9115-1d33e9b7551a\", \"RunID\": \"98a229b8-8fdd-4d1f-bf41-df00fb06f441\", \"ActivityID\": \"0\", \"ActivityType\": \"main.helloWorldActivity\"}\n2021-09-24T21:07:06.437-0700\tINFO\thelloworld/helloworld_workflow.go:62\thelloworld activity started\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\", \"ActivityID\": \"0\", \"ActivityType\": \"main.helloWorldActivity\", \"WorkflowType\": \"helloWorldWorkflow\", \"WorkflowID\": \"helloworld_75cf142b-c0de-407e-9115-1d33e9b7551a\", \"RunID\": \"98a229b8-8fdd-4d1f-bf41-df00fb06f441\"}\n2021-09-24T21:07:06.513-0700\tINFO\thelloworld/helloworld_workflow.go:55\tWorkflow completed.\t{\"Domain\": \"samples-domain\", \"TaskList\": \"helloWorldGroup\", \"WorkerID\": \"16520@IT-USA-25920@helloWorldGroup\", \"WorkflowType\": \"helloWorldWorkflow\", \"WorkflowID\": \"helloworld_75cf142b-c0de-407e-9115-1d33e9b7551a\", \"RunID\": \"98a229b8-8fdd-4d1f-bf41-df00fb06f441\", \"Result\": \"Hello Cadence!\"}\n```\n\nSee [instructions](service/worker/README.md) for setting up replication(XDC).\n\n## Repository layout\nA Cadence server cluster is composed of four different services: Frontend, Matching, History and Worker(system).\nHere's what's in each top-level directory in this repository:\n\n* **canary/** : The test code that needs to run periodically to ensure Cadence is healthy\n* **client/** : Client wrappers to let the four different services talk to each other\n* **cmd/** : The main function to build binaries for servers and CLI tools\n* **config/** : Sample configuration files\n* **docker/** : Code/scripts to build docker images\n* **docs/** : Documentation\n* **host/** : End-to-end integration tests\n* **schema/** : Versioned persistence schema for Cassandra/MySQL/Postgres/ElasticSearch\n* **scripts/** : Scripts for CI build\n* **tools/** : CLI tools for Cadence workflows and also schema updates for persistence\n* **service/** : Contains four sub-folders that dedicated for each of the four services\n* **common/** :  Basically contains all the rest of the code in Cadence server, the names of the sub folder are the topics of the packages\n\nIf looking at the Git history of the file/package, there should be some engineers focusing on each area/package/component that you are working on.\nYou will probably get code review from them. Don't hesitate to ask for some early feedback or help on Slack.\n\n## Testing & Debug\n\nTo run all the package tests use the below command.\nThis will run all the tests excluding end-to-end integration test in host/ package):\n\n```bash\nmake test\n```\n:warning: Note:\n> You will see some test failures because of errors connecting to MySQL/Postgres if only Cassandra is up. This is okay if you don't write any code related to persistence layer.\n\nTo run all end-to-end integration tests in **host/** package:\n```bash\nmake test_e2e\n```\n\nTo debug a specific test case when you see some failure, you can trigger it from an IDE, or use the command\n```\ngo test -v <path> -run <TestSuite> -testify.m <TestSpercificTaskName>\n# example:\ngo test -v github.com/uber/cadence/common/persistence/persistence-tests -run TestVisibilitySamplingSuite -testify.m TestListClosedWorkflowExecutions\n```\n\n## IDL Changes\n\nIf you make changes in the idls submodule and want to test them locally, you can easily do that by using go mod to use the local idls directory instead of github.com/uber/cadence-idl. Temporarily add the following to the bottom of go.mod:\n\n```replace github.com/uber/cadence-idl => ./idls```\n\nOr alternatively, push your idl changes to a branch in your own fork of idl repo and pull that specific commit to idl submodule. Then apply the go.mod change mentioned above.\n\n### Using IDL Changes\n\nOnce your [Pull Request](#pull-requests) has been successfully merged you can update cadence to use the new version:\n\n```bash\n# Update the IDLs directory\ngit submodule foreach git pull origin master\n# Update go to use the latest idl package\ngo get github.com/uber/cadence-idl@latest\n```\n\n## Pull Requests\nAfter all the preparation you are about to write code and make a Pull Request for the issue.\n\n### When to open a PR\nYou have a few options for choosing when to submit:\n\n* You can open a PR with an initial prototype with \"Draft\" option or with \"WIP\"(work in progress) in the title. This is useful when want to get some early feedback.\n\n* PR is supposed to be or near production ready. You should have fixed all styling, adding new tests if possible, and verified the change doesn't break any existing tests.\n\n* For small changes where the approach seems obvious, you can open a PR with what you believe to be production-ready or near-production-ready code. As you get more experience with how we develop code, you'll find that more PRs will begin falling into this category.\n\n### Commit Messages And Titles of Pull Requests\n\nOvercommit adds some requirements to your commit messages. At Uber, we follow the\n[Chris Beams](http://chris.beams.io/posts/git-commit/) guide to writing git\ncommit messages. Read it, follow it, learn it, love it.\n\nAll commit messages are from the titles of your pull requests. So make sure follow the rules when titling them.\nPlease don't use very generic titles like \"bug fixes\".\n\nAll PR titles should start with UPPER case.\n\nExamples:\n\n- [Make sync activity retry multiple times before fetch history from remote](https://github.com/cadence-workflow/cadence/pull/1379)\n- [Enable archival config per domain](https://github.com/cadence-workflow/cadence/pull/1351)\n\n#### Code Format and Licence headers checking\n\nThe project has strict rule about Golang format, import ordering and license declarations. You have to run\n```bash\nmake pr\n```\nwhich will take care of formatting for you.\n\n\n### Code review\nWe take code reviews very seriously at Cadence. Please don't be deterred if you feel like you've received some hefty feedback. That's completely normal and expected—and, if you're an external contributor, we very much appreciate your contribution!\n\nIn this repository in particular, merging a PR means accepting responsibility for maintaining that code for, quite possibly, the lifetime of Cadence. To take on that reponsibility, we need to ensure that meets our strict standards for production-ready code.\n\nNo one is expected to write perfect code on the first try. That's why we have code reviews in the first place!\n\nAlso, don't be embarrassed when your review points out syntax errors, stray whitespace, typos, and missing docstrings! That's why we have reviews. These properties are meant to guide you in your final scan.\n\n### Addressing feedback\nIf someone leaves line comments on your PR without leaving a top-level \"looks good to me\" (LGTM), it means they feel you should address their line comments before merging.\n\nYou should respond to all unresolved comments whenever you push a new revision or before you merge.\n\nAlso, as you gain confidence in Go, you'll find that some of the nitpicky style feedback you get does not make for obviously better code. Don't be afraid to stick to your guns and push back. Much of coding style is subjective.\n\n### Merging\nExternal contributors: you don't need to worry about this section. We'll merge your PR as soon as you've addressed all review feedback(you will get at least one approval) and pipeline runs are all successful(meaning all tests are passing).\n"
  },
  {
    "path": "Dockerfile",
    "content": "ARG TARGET=server\n\n# Can be used in case a proxy is necessary\nARG GOPROXY\n\n# Build Cadence binaries\nFROM golang:1.23.4-alpine3.21 AS builder\n\nARG RELEASE_VERSION\n\nRUN apk add --update --no-cache ca-certificates make git curl mercurial unzip bash\n\nWORKDIR /cadence\n\n# Making sure that dependency is not touched\nENV GOFLAGS=\"-mod=readonly\"\n\n# Copy go mod dependencies and try to share the module download cache\nCOPY go.* ./\nCOPY cmd/server/go.* ./cmd/server/\nCOPY common/archiver/gcloud/go.* ./common/archiver/gcloud/\n# go.work means this downloads everything, not just the top module\nRUN go mod download\n\nCOPY . .\nRUN rm -fr .bin .build idls\n\nENV CADENCE_RELEASE_VERSION=$RELEASE_VERSION\n\n# don't do anything fancy, just build.  must be run separately, before building things.\nRUN make .just-build\nRUN CGO_ENABLED=0 make cadence-cassandra-tool cadence-sql-tool cadence cadence-server cadence-bench cadence-canary\n\n\n# Download dockerize\nFROM alpine:3.18 AS dockerize\n\n# appears to require `docker buildx` or an explicit `--platform` at build time\nARG TARGETARCH\n\nRUN apk add --no-cache openssl\n\nENV DOCKERIZE_VERSION=v0.9.3\nRUN wget https://github.com/jwilder/dockerize/releases/download/$DOCKERIZE_VERSION/dockerize-linux-$TARGETARCH-$DOCKERIZE_VERSION.tar.gz \\\n    && tar -C /usr/local/bin -xzvf dockerize-linux-$TARGETARCH-$DOCKERIZE_VERSION.tar.gz \\\n    && rm dockerize-linux-$TARGETARCH-$DOCKERIZE_VERSION.tar.gz \\\n    && echo \"**** fix for host id mapping error ****\" \\\n    && chown root:root /usr/local/bin/dockerize\n\n\n# Alpine base image\nFROM alpine:3.18 AS alpine\n\nRUN apk add --update --no-cache ca-certificates tzdata bash curl\n\n# set up nsswitch.conf for Go's \"netgo\" implementation\n# https://github.com/gliderlabs/docker-alpine/issues/367#issuecomment-424546457\nRUN [ -e /etc/nsswitch.conf ] && grep '^hosts: files dns' /etc/nsswitch.conf\n\nSHELL [\"/bin/bash\", \"-c\"]\n\n\n# Cadence server\nFROM alpine AS cadence-server\n\nENV CADENCE_HOME=/etc/cadence\nRUN mkdir -p /etc/cadence\n\nCOPY --from=dockerize /usr/local/bin/dockerize /usr/local/bin\nCOPY --from=builder /cadence/cadence-cassandra-tool /usr/local/bin\nCOPY --from=builder /cadence/cadence-sql-tool /usr/local/bin\nCOPY --from=builder /cadence/cadence /usr/local/bin\nCOPY --from=builder /cadence/cadence-server /usr/local/bin\nCOPY --from=builder /cadence/schema /etc/cadence/schema\n\nCOPY docker/entrypoint.sh /docker-entrypoint.sh\nCOPY config/dynamicconfig /etc/cadence/config/dynamicconfig\nCOPY config/credentials /etc/cadence/config/credentials\nCOPY docker/config_template.yaml /etc/cadence/config\nCOPY docker/start-cadence.sh /start-cadence.sh\n\nWORKDIR /etc/cadence\n\nENV SERVICES=\"history,matching,frontend,worker\"\n\nEXPOSE 7933 7934 7935 7939\nENTRYPOINT [\"/docker-entrypoint.sh\"]\nCMD /start-cadence.sh\n\n\n# All-in-one Cadence server (~450mb)\nFROM cadence-server AS cadence-auto-setup\n\nRUN apk add --update --no-cache ca-certificates py3-pip mysql-client\nRUN pip3 install setuptools wheel && pip3 install cqlsh && cqlsh --version\n\nCOPY docker/start.sh /start.sh\nCOPY docker/domain /etc/cadence/domain\n\nCMD /start.sh\n\n# Cadence CLI\nFROM alpine AS cadence-cli\n\nCOPY --from=builder /cadence/cadence /usr/local/bin\n\nENTRYPOINT [\"cadence\"]\n\n# Cadence Canary\nFROM alpine AS cadence-canary\n\nCOPY --from=builder /cadence/cadence-canary /usr/local/bin\nCOPY --from=builder /cadence/cadence /usr/local/bin\n\nCMD [\"/usr/local/bin/cadence-canary\", \"--root\", \"/etc/cadence-canary\", \"start\"]\n\n# Cadence Bench\nFROM alpine AS cadence-bench\n\nCOPY --from=builder /cadence/cadence-bench /usr/local/bin\nCOPY --from=builder /cadence/cadence /usr/local/bin\n\nCMD [\"/usr/local/bin/cadence-bench\", \"--root\", \"/etc/cadence-bench\", \"start\"]\n\n# Final image\nFROM cadence-${TARGET}\n"
  },
  {
    "path": "LICENSE",
    "content": "                                 Apache License\n                           Version 2.0, January 2004\n                        http://www.apache.org/licenses/\n\n   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION\n\n   1. Definitions.\n\n      \"License\" shall mean the terms and conditions for use, reproduction,\n      and distribution as defined by Sections 1 through 9 of this document.\n\n      \"Licensor\" shall mean the copyright owner or entity authorized by\n      the copyright owner that is granting the License.\n\n      \"Legal Entity\" shall mean the union of the acting entity and all\n      other entities that control, are controlled by, or are under common\n      control with that entity. For the purposes of this definition,\n      \"control\" means (i) the power, direct or indirect, to cause the\n      direction or management of such entity, whether by contract or\n      otherwise, or (ii) ownership of fifty percent (50%) or more of the\n      outstanding shares, or (iii) beneficial ownership of such entity.\n\n      \"You\" (or \"Your\") shall mean an individual or Legal Entity\n      exercising permissions granted by this License.\n\n      \"Source\" form shall mean the preferred form for making modifications,\n      including but not limited to software source code, documentation\n      source, and configuration files.\n\n      \"Object\" form shall mean any form resulting from mechanical\n      transformation or translation of a Source form, including but\n      not limited to compiled object code, generated documentation,\n      and conversions to other media types.\n\n      \"Work\" shall mean the work of authorship, whether in Source or\n      Object form, made available under the License, as indicated by a\n      copyright notice that is included in or attached to the work\n      (an example is provided in the Appendix below).\n\n      \"Derivative Works\" shall mean any work, whether in Source or Object\n      form, that is based on (or derived from) the Work and for which the\n      editorial revisions, annotations, elaborations, or other modifications\n      represent, as a whole, an original work of authorship. For the purposes\n      of this License, Derivative Works shall not include works that remain\n      separable from, or merely link (or bind by name) to the interfaces of,\n      the Work and Derivative Works thereof.\n\n      \"Contribution\" shall mean any work of authorship, including\n      the original version of the Work and any modifications or additions\n      to that Work or Derivative Works thereof, that is intentionally\n      submitted to Licensor for inclusion in the Work by the copyright owner\n      or by an individual or Legal Entity authorized to submit on behalf of\n      the copyright owner. For the purposes of this definition, \"submitted\"\n      means any form of electronic, verbal, or written communication sent\n      to the Licensor or its representatives, including but not limited to\n      communication on electronic mailing lists, source code control systems,\n      and issue tracking systems that are managed by, or on behalf of, the\n      Licensor for the purpose of discussing and improving the Work, but\n      excluding communication that is conspicuously marked or otherwise\n      designated in writing by the copyright owner as \"Not a Contribution.\"\n\n      \"Contributor\" shall mean Licensor and any individual or Legal Entity\n      on behalf of whom a Contribution has been received by Licensor and\n      subsequently incorporated within the Work.\n\n   2. Grant of Copyright License. Subject to the terms and conditions of\n      this License, each Contributor hereby grants to You a perpetual,\n      worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n      copyright license to reproduce, prepare Derivative Works of,\n      publicly display, publicly perform, sublicense, and distribute the\n      Work and such Derivative Works in Source or Object form.\n\n   3. Grant of Patent License. Subject to the terms and conditions of\n      this License, each Contributor hereby grants to You a perpetual,\n      worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n      (except as stated in this section) patent license to make, have made,\n      use, offer to sell, sell, import, and otherwise transfer the Work,\n      where such license applies only to those patent claims licensable\n      by such Contributor that are necessarily infringed by their\n      Contribution(s) alone or by combination of their Contribution(s)\n      with the Work to which such Contribution(s) was submitted. If You\n      institute patent litigation against any entity (including a\n      cross-claim or counterclaim in a lawsuit) alleging that the Work\n      or a Contribution incorporated within the Work constitutes direct\n      or contributory patent infringement, then any patent licenses\n      granted to You under this License for that Work shall terminate\n      as of the date such litigation is filed.\n\n   4. Redistribution. You may reproduce and distribute copies of the\n      Work or Derivative Works thereof in any medium, with or without\n      modifications, and in Source or Object form, provided that You\n      meet the following conditions:\n\n      (a) You must give any other recipients of the Work or\n          Derivative Works a copy of this License; and\n\n      (b) You must cause any modified files to carry prominent notices\n          stating that You changed the files; and\n\n      (c) You must retain, in the Source form of any Derivative Works\n          that You distribute, all copyright, patent, trademark, and\n          attribution notices from the Source form of the Work,\n          excluding those notices that do not pertain to any part of\n          the Derivative Works; and\n\n      (d) If the Work includes a \"NOTICE\" text file as part of its\n          distribution, then any Derivative Works that You distribute must\n          include a readable copy of the attribution notices contained\n          within such NOTICE file, excluding those notices that do not\n          pertain to any part of the Derivative Works, in at least one\n          of the following places: within a NOTICE text file distributed\n          as part of the Derivative Works; within the Source form or\n          documentation, if provided along with the Derivative Works; or,\n          within a display generated by the Derivative Works, if and\n          wherever such third-party notices normally appear. The contents\n          of the NOTICE file are for informational purposes only and\n          do not modify the License. You may add Your own attribution\n          notices within Derivative Works that You distribute, alongside\n          or as an addendum to the NOTICE text from the Work, provided\n          that such additional attribution notices cannot be construed\n          as modifying the License.\n\n      You may add Your own copyright statement to Your modifications and\n      may provide additional or different license terms and conditions\n      for use, reproduction, or distribution of Your modifications, or\n      for any such Derivative Works as a whole, provided Your use,\n      reproduction, and distribution of the Work otherwise complies with\n      the conditions stated in this License.\n\n   5. Submission of Contributions. Unless You explicitly state otherwise,\n      any Contribution intentionally submitted for inclusion in the Work\n      by You to the Licensor shall be under the terms and conditions of\n      this License, without any additional terms or conditions.\n      Notwithstanding the above, nothing herein shall supersede or modify\n      the terms of any separate license agreement you may have executed\n      with Licensor regarding such Contributions.\n\n   6. Trademarks. This License does not grant permission to use the trade\n      names, trademarks, service marks, or product names of the Licensor,\n      except as required for reasonable and customary use in describing the\n      origin of the Work and reproducing the content of the NOTICE file.\n\n   7. Disclaimer of Warranty. Unless required by applicable law or\n      agreed to in writing, Licensor provides the Work (and each\n      Contributor provides its Contributions) on an \"AS IS\" BASIS,\n      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or\n      implied, including, without limitation, any warranties or conditions\n      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A\n      PARTICULAR PURPOSE. You are solely responsible for determining the\n      appropriateness of using or redistributing the Work and assume any\n      risks associated with Your exercise of permissions under this License.\n\n   8. Limitation of Liability. In no event and under no legal theory,\n      whether in tort (including negligence), contract, or otherwise,\n      unless required by applicable law (such as deliberate and grossly\n      negligent acts) or agreed to in writing, shall any Contributor be\n      liable to You for damages, including any direct, indirect, special,\n      incidental, or consequential damages of any character arising as a\n      result of this License or out of the use or inability to use the\n      Work (including but not limited to damages for loss of goodwill,\n      work stoppage, computer failure or malfunction, or any and all\n      other commercial damages or losses), even if such Contributor\n      has been advised of the possibility of such damages.\n\n   9. Accepting Warranty or Additional Liability. While redistributing\n      the Work or Derivative Works thereof, You may choose to offer,\n      and charge a fee for, acceptance of support, warranty, indemnity,\n      or other liability obligations and/or rights consistent with this\n      License. However, in accepting such obligations, You may act only\n      on Your own behalf and on Your sole responsibility, not on behalf\n      of any other Contributor, and only if You agree to indemnify,\n      defend, and hold each Contributor harmless for any liability\n      incurred by, or claims asserted against, such Contributor by reason\n      of your accepting any such warranty or additional liability.\n\n   END OF TERMS AND CONDITIONS\n\n   APPENDIX: How to apply the Apache License to your work.\n\n      To apply the Apache License to your work, attach the following\n      boilerplate notice, with the fields enclosed by brackets \"[]\"\n      replaced with your own identifying information. (Don't include\n      the brackets!)  The text should be enclosed in the appropriate\n      comment syntax for the file format. We also recommend that a\n      file or class name and description of purpose be included on the\n      same \"printed page\" as the copyright notice for easier\n      identification within third-party archives.\n\n   Copyright [yyyy] [name of copyright owner]\n\n   Licensed under the Apache License, Version 2.0 (the \"License\");\n   you may not use this file except in compliance with the License.\n   You may obtain a copy of the License at\n\n       http://www.apache.org/licenses/LICENSE-2.0\n\n   Unless required by applicable law or agreed to in writing, software\n   distributed under the License is distributed on an \"AS IS\" BASIS,\n   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n   See the License for the specific language governing permissions and\n   limitations under the License."
  },
  {
    "path": "MAINTAINERS.md",
    "content": "# Maintainers\n\n- Ender Demirkaya (https://github.com/demirkayaender)\n- Jan Kisel (https://github.com/dkrotx)\n\n- Abhishek Jha (https://github.com/abhishekj720)\n- Adhitya Mamallan (https://github.com/adhityamamallan)\n- Anshuman Pandey (https://github.com/pandeyanshuman)\n- Assem Hafez (https://github.com/Assem-Hafez)\n- Bowen Xiao (https://github.com/bowenxia)\n- Chris Warren (https://github.com/c-warren)\n- Diana Zawadzki (https://github.com/zawadzkidiana)\n- David Porter (https://github.com/davidporter-id-au)\n- Eleonora Di Gregorio (https://github.com/eleonoradgr)\n- Felipe Imanishi (https://github.com/fimanishi)\n- Gaziza Yestemirova (https://github.com/gazi-yestemirova)\n- Jakob Haahr Taankvist (https://github.com/jakobht)\n- Kevin Burns (https://github.com/Bueller87)\n- Nate Mortensen (https://github.com/natemort)\n- Neil Xie (https://github.com/neil-xie)\n- Shijie Sheng (https://github.com/shijiesheng)\n- Tim Chan (https://github.com/macrotim)\n- Tim Li (https://github.com/timl3136)\n- Vsevolod Kaloshin (https://github.com/arzonus)\n- Zijian Chen (https://github.com/Shaddoll)\n"
  },
  {
    "path": "Makefile",
    "content": "# get rid of default behaviors, they're just noise\nMAKEFLAGS += --no-builtin-rules\n.SUFFIXES:\n\n# pipefail is very easy to forget about, and sadly not the default.\n# this changes that, and makes our target scripts more strict.\n#\n# make sure to include all this if overriding on the CLI, e.g.:\n# \tDO:      make SHELL='bash -eoux pipefail'\n# \tDO NOT:  make SHELL='bash -x'\n# as otherwise you will ignore all errors.\nSHELL = /bin/bash -e -u -o pipefail\n\ndefault: help\n\n# ###########################################\n#                TL;DR DOCS:\n# ###########################################\n# - Targets should never, EVER be *actual source files*.\n#   Always use book-keeping files in $(BUILD).\n#   Otherwise e.g. changing git branches could confuse Make about what it needs to do.\n# - Similarly, prerequisites should be those book-keeping files,\n#   not source files that are prerequisites for book-keeping.\n#   e.g. depend on .build/fmt, not $(ALL_SRC), and not both.\n# - Be strict and explicit about prerequisites / order of execution / etc.\n# - Test your changes with `-j 27 --output-sync` or something!\n# - Test your changes with `make -d ...`!  It should be reasonable!\n\n# temporary build products and book-keeping targets that are always good to / safe to clean.\nBUILD := .build\n# bins that are `make clean` friendly, i.e. they build quickly and do not require new downloads.\n# in particular this should include goimports, as it changes based on which version of go compiles it,\n# and few know to do more than `make clean`.\nBIN := $(BUILD)/bin\n# relatively stable build products, e.g. tools.\n# usually unnecessary to clean, and may require downloads to restore, so this folder is not automatically cleaned.\nSTABLE_BIN := .bin\n\n# toolchain version we all use.\n# this export ensures it is a precise version rather than a minimum.\n# lint step ensures this matches other files.\nGOWORK_TOOLCHAIN := $(word 2,$(shell grep 'toolchain' go.work))\nexport GOTOOLCHAIN ?= $(GOWORK_TOOLCHAIN)\nifneq ($(GOTOOLCHAIN),$(GOWORK_TOOLCHAIN))\n# this can be useful for trying new/old versions, so don't block it\n$(warning warning: your Go toolchain is explicitly set to GOTOOLCHAIN=$(GOTOOLCHAIN), ignoring go.work version $(GOWORK_TOOLCHAIN)...)\nendif\n\n# ====================================\n# book-keeping files that are used to control sequencing.\n#\n# you should use these as prerequisites in almost all cases, not the source files themselves.\n# these are defined in roughly the reverse order that they are executed, for easier reading.\n#\n# recipes and any other prerequisites are defined only once, further below.\n# ====================================\n\n# all bins depend on: $(BUILD)/lint\n# note that vars that do not yet exist are empty, so any prerequisites defined below are ineffective here.\n$(BUILD)/lint: $(BUILD)/fmt # lint will fail if fmt fails, so fmt first\n$(BUILD)/proto-lint:\n$(BUILD)/gomod-lint:\n$(BUILD)/goversion-lint:\n$(BUILD)/fmt: $(BUILD)/codegen # formatting must occur only after all other go-file-modifications are done\n# $(BUILD)/copyright\n# $(BUILD)/copyright: $(BUILD)/codegen # must add copyright to generated code, sometimes needs re-formatting\n$(BUILD)/codegen: $(BUILD)/thrift $(BUILD)/protoc\n$(BUILD)/thrift: $(BUILD)/go_mod_check\n$(BUILD)/protoc: $(BUILD)/go_mod_check\n$(BUILD)/go_mod_check:\n\n# ====================================\n# helper vars\n# ====================================\n\n# a literal space value, for makefile purposes.\n# the full \"trailing # one space after $(null)\" is necessary for correct behavior,\n# and this strategy works in both new and old versions of make, `SPACE +=` does not.\nnull  :=\nSPACE := $(null) #\nCOMMA := ,\n\n# set a V=1 env var for verbose output. V=0 (or unset) disables.\n# this is used to make two verbose flags:\n# - $Q, to replace ALL @ use, so CI can be reliably verbose\n# - $(verbose), to forward verbosity flags to commands via `$(if $(verbose),-v)` or similar\n#\n# SHELL='bash -x' is useful too, but can be more confusing to understand.\nV ?= 0\nifneq (0,$(V))\nverbose := 1\nQ :=\nelse\nverbose :=\nQ := @\nendif\n\n# and enforce ^ that rule: grep the makefile for line-starting @ use, error if any exist.\n# limit to one match because multiple look too weird.\n_BAD_AT_USE=$(shell grep -n -m1 '^\\s*@' $(MAKEFILE_LIST))\nifneq (,$(_BAD_AT_USE))\n$(warning Makefile cannot use @ to silence commands, use $$Q instead:)\n$(warning found on line $(_BAD_AT_USE))\n$(error fix that line and try again)\nendif\n\n# M1 macs may need to switch back to x86, until arm releases are available\nEMULATE_X86 =\nifeq ($(shell uname -sm),Darwin arm64)\nEMULATE_X86 = arch -x86_64\nendif\n\nPROJECT_ROOT = github.com/uber/cadence\n\n# helper for executing bins that need other bins, just `$(BIN_PATH) the_command ...`\n# I'd recommend not exporting this in general, to reduce the chance of accidentally using non-versioned tools.\nBIN_PATH := PATH=\"$(abspath $(BIN)):$(abspath $(STABLE_BIN)):$$PATH\"\n\n# automatically gather all source files that currently exist.\n# works by ignoring everything in the parens (and does not descend into matching folders) due to `-prune`,\n# and everything else goes to the other side of the `-o` branch, which is `-print`ed.\n# this is dramatically faster than a `find . | grep -v vendor` pipeline, and scales far better.\nFRESH_ALL_SRC = $(shell \\\n\tfind . \\\n\t\\( \\\n\t\t-path './vendor/*' \\\n\t\t-o -path './idls/*' \\\n\t\t-o -path './.build/*' \\\n\t\t-o -path './.bin/*' \\\n\t\t-o -path './.git/*' \\\n\t\t-o -path './.worktrees/*' \\\n\t\\) \\\n\t-prune \\\n\t-o -name '*.go' \\\n\t-type f \\\n\t-print \\\n)\n# most things can use a cached copy, e.g. all dependencies.\n# this will not include any files that are created during a `make` run, e.g. via protoc,\n# but that generally should not matter (e.g. dependencies are computed at parse time, so it\n# won't affect behavior either way - choose the fast option).\n#\n# if you require a fully up-to-date list, e.g. for shell commands, use FRESH_ALL_SRC instead.\nALL_SRC := $(FRESH_ALL_SRC)\n# as lint ignores generated code, it can use the cached copy in all cases\nLINT_SRC := $(filter-out %_test.go ./.gen/%, $(ALL_SRC))\n\n# ====================================\n# $(BIN) targets\n# ====================================\n\n# downloads and builds a go-gettable tool, versioned by go.mod, and installs\n# it into the build folder, named the same as the last portion of the URL.\n#\n# unfortunately go.work and `go list -modfile=sub/module/go.mod` seem to interact badly,\n# and some versions complain about duplicates, while others simply block it outright.\n# the good news is that you can just drop that and `cd` to the folder and it works.\ndefine go_build_tool\n$Q echo \"building $(or $(2), $(notdir $(1))) from internal/tools/go.mod...\"\n$Q cd internal/tools; GOTOOLCHAIN=auto go build -mod=readonly -o ../../$(BIN)/$(or $(2), $(notdir $(1))) $(1)\nendef\n\n# same as go_build_tool, but uses our main module file, not the tools one.\n# this is necessary / useful for tools that we are already importing in the repo, e.g. yarpc.\n# versions here are checked to make sure the tools version matches the service version.\n#\n# this is an imperfect check as it only checks the top-level version.\n# checks of other packages are handled in $(BUILD)/go_mod_check\ndefine go_mod_build_tool\n$Q echo \"building $(or $(2), $(notdir $(1))) from go.mod...\"\n$Q ./scripts/check-gomod-version.sh $(1) $(if $(verbose),-v)\n$Q GOTOOLCHAIN=auto go build -mod=readonly -o $(BIN)/$(or $(2), $(notdir $(1))) $(1)\nendef\n\n# utility target.\n# use as an order-only prerequisite for targets that do not implicitly create these folders.\n$(BIN) $(BUILD) $(STABLE_BIN):\n\t$Q mkdir -p $@\n\n$(BIN)/thriftrw: go.mod go.work\n\t$(call go_mod_build_tool,go.uber.org/thriftrw)\n\n$(BIN)/thriftrw-plugin-yarpc: go.mod go.work\n\t$(call go_mod_build_tool,go.uber.org/yarpc/encoding/thrift/thriftrw-plugin-yarpc)\n\n$(BIN)/mockgen: internal/tools/go.mod go.work\n\t$(call go_build_tool,go.uber.org/mock/mockgen)\n\n$(BIN)/mockery: internal/tools/go.mod go.work\n\t$(call go_build_tool,github.com/vektra/mockery/v2,mockery)\n\n$(BIN)/enumer: internal/tools/go.mod go.work\n\t$(call go_build_tool,github.com/dmarkham/enumer)\n\n# organizes imports and reformats\n$(BIN)/gci: internal/tools/go.mod go.work\n\t$(call go_build_tool,github.com/daixiang0/gci)\n\n# removes unused imports and reformats\n$(BIN)/goimports: internal/tools/go.mod go.work\n\t$(call go_build_tool,golang.org/x/tools/cmd/goimports)\n\n$(BIN)/gowrap: go.mod go.work\n\t$(call go_build_tool,github.com/hexdigest/gowrap/cmd/gowrap)\n\n$(BIN)/revive: internal/tools/go.mod go.work\n\t$(call go_build_tool,github.com/mgechev/revive)\n\n$(BIN)/nilaway: internal/tools/go.mod go.work\n\t$(call go_build_tool,go.uber.org/nilaway/cmd/nilaway,nilaway)\n\n$(BIN)/protoc-gen-gogofast: go.mod go.work | $(BIN)\n\t$(call go_mod_build_tool,github.com/gogo/protobuf/protoc-gen-gogofast)\n\n$(BIN)/protoc-gen-yarpc-go: go.mod go.work | $(BIN)\n\t$(call go_mod_build_tool,go.uber.org/yarpc/encoding/protobuf/protoc-gen-yarpc-go)\n\n$(BUILD)/go_mod_check: go.mod internal/tools/go.mod go.work\n\t$Q # generated == used is occasionally important for gomock / mock libs in general.  this is not a definite problem if violated though.\n\t$Q ./scripts/check-gomod-version.sh github.com/golang/mock/gomock $(if $(verbose),-v)\n\t$Q touch $@\n\n# copyright header checker/writer.  only requires stdlib, so no other dependencies are needed.\n# $(BIN)/copyright: cmd/tools/copyright/licensegen.go\n# \t$Q go build -o $@ ./cmd/tools/copyright/licensegen.go\n\n# https://docs.buf.build/\n# changing BUF_VERSION will automatically download and use the specified version.\nBUF_VERSION = 1.47.2\nOS = $(shell uname -s)\nARCH = $(shell $(EMULATE_X86) uname -m)\nBUF_URL = https://github.com/bufbuild/buf/releases/download/v$(BUF_VERSION)/buf-$(OS)-$(shell uname -m)\n# use BUF_VERSION_BIN as a bin prerequisite, not \"buf\", so the correct version will be used.\n# otherwise this must be a .PHONY rule, or the buf bin / symlink could become out of date.\nBUF_VERSION_BIN = buf-$(BUF_VERSION)\n$(STABLE_BIN)/$(BUF_VERSION_BIN): | $(STABLE_BIN)\n\t$Q echo \"downloading buf $(BUF_VERSION)\"\n\t$Q curl -sSL $(BUF_URL) -o $@\n\t$Q chmod +x $@\n\n# https://www.grpc.io/docs/languages/go/quickstart/\n# protoc-gen-gogofast (yarpc) are versioned via tools.go + go.mod (built above) and will be rebuilt as needed.\n# changing PROTOC_VERSION will automatically download and use the specified version\nPROTOC_VERSION = 3.14.0\nPROTOC_URL = https://github.com/protocolbuffers/protobuf/releases/download/v$(PROTOC_VERSION)/protoc-$(PROTOC_VERSION)-$(subst Darwin,osx,$(OS))-$(ARCH).zip\n# the zip contains an /include folder that we need to use to learn the well-known types\nPROTOC_UNZIP_DIR = $(STABLE_BIN)/protoc-$(PROTOC_VERSION)-zip\n# use PROTOC_VERSION_BIN as a bin prerequisite, not \"protoc\", so the correct version will be used.\n# otherwise this must be a .PHONY rule, or the buf bin / symlink could become out of date.\nPROTOC_VERSION_BIN = protoc-$(PROTOC_VERSION)\n$(STABLE_BIN)/$(PROTOC_VERSION_BIN): | $(STABLE_BIN)\n\t$Q echo \"downloading protoc $(PROTOC_VERSION): $(PROTOC_URL)\"\n\t$Q # recover from partial success\n\t$Q rm -rf $(STABLE_BIN)/protoc.zip $(PROTOC_UNZIP_DIR)\n\t$Q # download, unzip, copy to a normal location\n\t$Q curl -sSL $(PROTOC_URL) -o $(STABLE_BIN)/protoc.zip\n\t$Q unzip -q $(STABLE_BIN)/protoc.zip -d $(PROTOC_UNZIP_DIR)\n\t$Q cp $(PROTOC_UNZIP_DIR)/bin/protoc $@\n\n# checks that the idl submodule points to a commit on master, and that it matches the go module (which must be a pseudo version).\n# this is only used in an explicit CI step, because it's expected to fail when developing.\n#\n# `git ls-tree HEAD idls` is selected because this only cares about the committed/checked-out target,\n# not whatever the current status is, because only the committed value will exist for others.\n#\n# and last but not least: this avoids using `go` to make this check take only a couple seconds in CI,\n# so the whole docker container doesn't have to be prepared.\n.idl-status:\n\t$(Q) cd idls && \\\n\t\tSUBMODULE_COMMIT=$$(git rev-parse HEAD) && \\\n\t\tBRANCH_INFO=$$(git branch -r --contains \"$$SUBMODULE_COMMIT\" | head -n1) && \\\n\t\tif ! git branch -r --contains \"$$SUBMODULE_COMMIT\" | grep -q \"origin/master\"; then \\\n\t\t\techo \"Error: Submodule commit $$SUBMODULE_COMMIT belongs to $$BRANCH_INFO, not to master branch\" && \\\n\t\t\texit 1; \\\n\t\tfi\n\t$(Q) idlsha=$$(git ls-tree HEAD idls | awk '{print substr($$3,0,12)}') && \\\n\t\tgosha=$$(grep github.com/uber/cadence-idl go.mod | tr '-' '\\n' | tail -n1) && \\\n\t\tif [[ \"$$idlsha\" != \"$$gosha\" ]]; then \\\n\t\t\techo \"IDL submodule sha ($$idlsha) does not match go module sha ($$gosha).\" >&2 && \\\n\t\t\techo \"Make sure the IDL PR has been merged, and this PR is updated, before merging here.\" >&2 && \\\n\t\t\texit 1; \\\n\t\tfi\n\n\n# ====================================\n# Codegen targets\n# ====================================\n\n# IDL submodule must be populated, or files will not exist -> prerequisites will be wrong -> build will fail.\n# Because it must exist before the makefile is parsed, this cannot be done automatically as part of a build.\n# Instead: call this func in targets that require the submodule to exist, so that target will not be built.\n#\n# THRIFT_FILES is just an easy identifier for \"the submodule has files\", others would work fine as well.\ndefine ensure_idl_submodule\n$(if $(THRIFT_FILES),,$(error idls/ submodule must exist, or build will fail.  Run `git submodule update --init` and try again))\nendef\n\n# codegen is done when thrift and protoc are done\n$(BUILD)/codegen: $(BUILD)/thrift $(BUILD)/protoc | $(BUILD)\n\t$Q touch $@\n\nTHRIFT_FILES := $(shell find idls -name '*.thrift')\n# book-keeping targets to build.  one per thrift file.\n# idls/thrift/thing.thrift -> .build/thing.thrift\n# the reverse is done in the recipe.\nTHRIFT_GEN := $(subst idls/thrift/,.build/,$(THRIFT_FILES))\n\n# thrift is done when all sub-thrifts are done\n$(BUILD)/thrift: $(THRIFT_GEN) | $(BUILD)\n\t$(call ensure_idl_submodule)\n\t$Q touch $@\n\n# how to generate each thrift book-keeping file.\n#\n# note that each generated file depends on ALL thrift files - this is necessary because they can import each other.\n# as --no-recurse is specified, these can be done in parallel, since output files will not overwrite each other.\n$(THRIFT_GEN): $(THRIFT_FILES) $(BIN)/thriftrw $(BIN)/thriftrw-plugin-yarpc | $(BUILD)\n\t$Q echo 'thriftrw for $(subst .build/,idls/thrift/,$@)...'\n\t$Q $(BIN_PATH) $(BIN)/thriftrw \\\n\t\t--plugin=yarpc \\\n\t\t--pkg-prefix=$(PROJECT_ROOT)/.gen/go \\\n\t\t--out=.gen/go \\\n\t\t--no-recurse \\\n\t\t$(subst .build/,idls/thrift/,$@)\n\t$Q touch $@\n\nPROTO_ROOT := proto\n# output location is defined by `option go_package` in the proto files, all must stay in sync with this\nPROTO_OUT := .gen/proto\nPROTO_FILES = $(shell find -L ./$(PROTO_ROOT) -name \"*.proto\" | grep -v \"persistenceblobs\" | grep -v public)\nPROTO_DIRS = $(sort $(dir $(PROTO_FILES)))\n\n# protoc splits proto files into directories, otherwise protoc-gen-gogofast is complaining about inconsistent package\n# import paths due to multiple packages being compiled at once.\n#\n# After compilation files are moved to final location, as plugins adds additional path based on proto package.\n$(BUILD)/protoc: $(PROTO_FILES) $(STABLE_BIN)/$(PROTOC_VERSION_BIN) $(BIN)/protoc-gen-gogofast $(BIN)/protoc-gen-yarpc-go | $(BUILD)\n\t$(call ensure_idl_submodule)\n\t$Q mkdir -p $(PROTO_OUT)\n\t$Q echo \"protoc...\"\n\t$Q chmod +x $(STABLE_BIN)/$(PROTOC_VERSION_BIN)\n\t$Q $(foreach PROTO_DIR,$(PROTO_DIRS),$(EMULATE_X86) $(STABLE_BIN)/$(PROTOC_VERSION_BIN) \\\n\t\t--plugin $(BIN)/protoc-gen-gogofast \\\n\t\t--plugin $(BIN)/protoc-gen-yarpc-go \\\n\t\t-I=$(PROTO_ROOT)/public \\\n\t\t-I=$(PROTO_ROOT)/internal \\\n\t\t-I=$(PROTOC_UNZIP_DIR)/include \\\n\t\t--gogofast_out=Mgoogle/protobuf/duration.proto=github.com/gogo/protobuf/types,Mgoogle/protobuf/field_mask.proto=github.com/gogo/protobuf/types,Mgoogle/protobuf/timestamp.proto=github.com/gogo/protobuf/types,Mgoogle/protobuf/wrappers.proto=github.com/gogo/protobuf/types,paths=source_relative:$(PROTO_OUT) \\\n\t\t--yarpc-go_out=$(PROTO_OUT) \\\n\t\t$$(find $(PROTO_DIR) -name '*.proto');\\\n\t)\n\t$Q # This directory exists for local/github_actions but not for docker builds.\n\t$Q if [ -d \"$(PROTO_OUT)/uber/cadence\" ]; then \\\n\t\tcp -R $(PROTO_OUT)/uber/cadence/* $(PROTO_OUT)/; \\\n\t\trm -r $(PROTO_OUT)/uber; \\\n\t   fi\n\t$Q touch $@\n\n# ====================================\n# Rule-breaking targets intended ONLY for special cases with no good alternatives.\n# ====================================\n\n# used to bypass checks when building binaries.\n# this is primarily intended for docker image builds, but can be used to skip things locally.\n.PHONY: .just-build\n.just-build: | $(BUILD)\n\ttouch $(BUILD)/just-build\n\n# ====================================\n# other intermediates\n# ====================================\n\n$(BUILD)/proto-lint: $(PROTO_FILES) $(STABLE_BIN)/$(BUF_VERSION_BIN) | $(BUILD)\n\t$Q cd $(PROTO_ROOT) && ../$(STABLE_BIN)/$(BUF_VERSION_BIN) lint\n\t$Q touch $@\n\n# lints that go modules are as expected, e.g. parent does not import submodule.\n# tool builds that need to be in sync with the parent are partially checked through go_mod_build_tool, but should probably be checked here too\n$(BUILD)/gomod-lint: go.mod internal/tools/go.mod common/archiver/gcloud/go.mod | $(BUILD)\n\t$Q echo \"checking for direct submodule dependencies in root go.mod...\"\n\t$Q ( \\\n\t\tMAIN_MODULE=$$(grep \"^module \" go.mod | awk '{print $$2}'); \\\n\t\tSUBMODULES=$$(find . -type f -name \"go.mod\" -not -path \"./go.mod\" -not -path \"./idls/*\" -not -path \"./.worktrees/*\" -exec dirname {} \\; | sed 's|^\\./||'); \\\n\t\tfor submodule in $$SUBMODULES; do \\\n\t\t\tsubmodule_path=\"$$MAIN_MODULE/$$submodule\"; \\\n\t\t\tif grep -q \"$$submodule_path\" go.mod; then \\\n\t\t\t\techo \"ERROR: Root go.mod directly depends on submodule: $$submodule_path\" >&2; \\\n\t\t\t\texit 1; \\\n\t\t\tfi; \\\n\t\t\techo \"✓ No direct dependency on $$submodule\"; \\\n\t\tdone; \\\n\t)\n\t$Q touch $@\n\n# note that LINT_SRC is fairly fake as a prerequisite.\n# it's a coarse \"you probably don't need to re-lint\" filter, nothing more.\n$(BUILD)/code-lint: $(LINT_SRC) $(BIN)/revive $(BIN)/nilaway | $(BUILD)\n\t$Q echo \"lint...\"\n\t$Q # non-optional vet checks.  unfortunately these are not currently included in `go test`'s default behavior.\n\t$Q go vet -copylocks ./... ./common/archiver/gcloud/...\n\t$Q $(BIN)/revive -config revive.toml -exclude './vendor/...' -exclude './.gen/...' -formatter stylish ./...\n\t$Q # look for go files with \"//comments\", and ignore \"//go:build\"-style directives (\"grep -n\" shows \"file:line: //go:build\" so the regex is a bit complex)\n\t$Q bad=\"$$(find . -type f -name '*.go' -not -path './idls/*' -not -path './.worktrees/*' | xargs grep -n -E '^\\s*//\\S' | grep -E -v '^[^:]+:[^:]+:\\s*//[a-z]+:[a-z]+' || true)\"; \\\n\t\tif [ -n \"$$bad\" ]; then \\\n\t\t  echo \"$$bad\" >&2; \\\n\t\t  echo 'non-directive comments must have a space after the \"//\"' >&2; \\\n\t\t  exit 1; \\\n\t\tfi\n\t$Q echo \"nilaway check...\"\n\t$Q GOTOOLCHAIN=go1.24.0 $(BIN)/nilaway -test=false github.com/uber/cadence/common/types/mapper/...\n\t$Q touch $@\n\n$(BUILD)/goversion-lint: go.work Dockerfile docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n\t$Q echo \"checking go version...\"\n\t$Q # intentionally using go.work toolchain, as GOTOOLCHAIN is user-overridable\n\t$Q ./scripts/check-go-toolchain.sh $(GOWORK_TOOLCHAIN)\n\t$Q touch $@\n\n# fmt and copyright are mutually cyclic with their inputs, so if a copyright header is modified:\n# - copyright -> makes changes\n# - fmt sees changes -> makes changes\n# - now copyright thinks it needs to run again (but does nothing)\n# - which means fmt needs to run again (but does nothing)\n# and now after two passes it's finally stable, because they stopped making changes.\n#\n# this is not fatal, we can just run 2x.\n# to be fancier though, we can detect when *both* are run, and re-touch the book-keeping files to prevent the second run.\n# this STRICTLY REQUIRES that `copyright` and `fmt` are mutually stable, and that copyright runs before fmt.\n# if either changes, this will need to change.\nMAYBE_TOUCH_COPYRIGHT=\n\n# use FRESH_ALL_SRC so it won't miss any generated files produced earlier.\n$(BUILD)/fmt: $(ALL_SRC) $(BIN)/goimports $(BIN)/gci | $(BUILD)\n\t$Q echo \"removing unused imports...\"\n\t$Q # goimports thrashes on internal/tools, sadly.  just hide it.\n\t$Q echo $(filter-out ./internal/tools/tools.go,$(FRESH_ALL_SRC)) | xargs $(BIN)/goimports -w\n\t$Q echo \"grouping imports...\"\n\t$Q echo $(FRESH_ALL_SRC) | xargs $(BIN)/gci write --section standard --section 'Prefix(github.com/uber/cadence/)' --section default --section blank\n\t$Q touch $@\n# \t$Q $(MAYBE_TOUCH_COPYRIGHT)\n\n# $(BUILD)/copyright: $(ALL_SRC) $(BIN)/copyright | $(BUILD)\n# \t$(BIN)/copyright --verifyOnly\n# \t$Q $(eval MAYBE_TOUCH_COPYRIGHT=touch $@)\n# \t$Q touch $@\n\n# ====================================\n# developer-oriented targets\n#\n# many of these share logic with other intermediates, but are useful to make .PHONY for output on demand.\n# as the Makefile is fast, it's reasonable to just delete the book-keeping file recursively make.\n# this way the effort is shared with future `make` runs.\n# ====================================\n\n# \"re-make\" a target by deleting and re-building book-keeping target(s).\n# the + is necessary for parallelism flags to be propagated\ndefine remake\n$Q rm -f $(addprefix $(BUILD)/,$(1))\n$Q +$(MAKE) --no-print-directory $(addprefix $(BUILD)/,$(1))\nendef\n\n.PHONY: lint fmt copyright pr\n\n# useful to actually re-run to get output again.\n# reuse the intermediates for simplicity and consistency.\nlint: ## (Re)run the linter\n\t$(call remake,proto-lint gomod-lint code-lint goversion-lint)\n\n# intentionally not re-making, it's a bit slow and it's clear when it's unnecessary\nfmt: $(BUILD)/fmt ## Run `gofmt` / organize imports / etc\n\n# not identical to the intermediate target, but does provide the same codegen (or more).\n# copyright: $(BIN)/copyright | $(BUILD) ## Update copyright headers\n# \t$(BIN)/copyright\n# \t$Q touch $(BUILD)/copyright\n\ndefine make_quietly\n$Q echo \"make $1...\"\n$Q output=$$(mktemp); $(MAKE) $1 > $$output 2>&1 || ( cat $$output; echo -e '\\nfailed `make $1`, check output above' >&2; exit 1)\nendef\n\noverride GEN_DIR := $(patsubst %/,%,$(strip $(GEN_DIR)))\nGO_GENERATE_SCOPE ?= $(if $(GEN_DIR),./$(GEN_DIR)/...,./...)\nGO_GENERATE_MAKE_ARG = $(if $(GEN_DIR),GEN_DIR=$(GEN_DIR),)\n\n# pre-PR target to build and refresh everything\npr: ## Redo all codegen and basic checks, to ensure your PR will be able to run tests. Optional: GEN_DIR=path/to/package to scope go-generate\n\t$Q $(if $(verbose),$(MAKE) tidy,$(call make_quietly,tidy))\n\t$Q $(if $(verbose),$(MAKE) go-generate $(GO_GENERATE_MAKE_ARG),$(call make_quietly,go-generate $(GO_GENERATE_MAKE_ARG)))\n\t$Q $(if $(verbose),$(MAKE) fmt,$(call make_quietly,fmt))\n\t$Q $(if $(verbose),$(MAKE) lint,$(call make_quietly,lint))\n# \t$Q $(if $(verbose),$(MAKE) copyright,$(call make_quietly,copyright))\n\n# ====================================\n# binaries to build\n# ====================================\nGOOS ?= $(shell go env GOOS)\nGOARCH ?= $(shell go env GOARCH)\n\n# normally, depend on lint, so a full build and check and codegen runs.\n# docker builds though must *not* do this, and need to rely entirely on committed code.\nifeq (,$(wildcard $(BUILD)/just-build))\nBINS_DEPEND_ON := $(BUILD)/lint\nelse\nBINS_DEPEND_ON :=\n$(warning !!!!! lint and codegen disabled, validations skipped !!!!!)\nendif\n\nBINS =\nTOOLS =\n\nBINS  += cadence-cassandra-tool\nTOOLS += cadence-cassandra-tool\ncadence-cassandra-tool: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-cassandra-tool with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/tools/cassandra/main.go\n\nBINS  += cadence-sql-tool\nTOOLS += cadence-sql-tool\ncadence-sql-tool: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-sql-tool with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/tools/sql/main.go\n\nBINS  += cadence\nTOOLS += cadence\ncadence: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/tools/cli/main.go\n\nBINS += cadence-server\ncadence-server: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-server with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/server/main.go\n\nBINS += cadence-canary\ncadence-canary: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-canary with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/canary/main.go\n\nBINS += sharddistributor-canary\nsharddistributor-canary: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling sharddistributor-canary with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/sharddistributor-canary/main.go\n\nBINS += cadence-bench\ncadence-bench: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-bench with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/bench/main.go\n\n\nBINS  += cadence-releaser\ncadence-releaser: $(BINS_DEPEND_ON)\n\t$Q echo \"compiling cadence-releaser with OS: $(GOOS), ARCH: $(GOARCH)\"\n\t$Q ./scripts/build-with-ldflags.sh -o $@ cmd/tools/releaser/releaser.go\n\n.PHONY: go-generate bins tools release clean\n\nbins: $(BINS) ## Build all binaries, and any fast codegen needed (does not refresh wrappers or mocks)\n\ntools: $(TOOLS)\n\ngo-generate: $(BIN)/mockgen $(BIN)/enumer $(BIN)/mockery  $(BIN)/gowrap ## Run `go generate` to regen mocks, enums, etc\n\t$Q echo \"running go generate $(GO_GENERATE_SCOPE), this takes a minute or more...\"\n\t$Q # add our bins to PATH so `go generate` can find them\n\t$Q $(BIN_PATH) go generate $(if $(verbose),-v) $(GO_GENERATE_SCOPE)\n\t$Q $(MAKE) --no-print-directory fmt\n# \t$Q echo \"updating copyright headers\"\n# \t$Q $(MAKE) --no-print-directory copyright\n\nrelease: ## Re-generate generated code and run tests\n\t$(MAKE) --no-print-directory go-generate\n\t$(MAKE) --no-print-directory test\n\nbuild: ## `go build` all packages and tests (a quick compile check only, skips all other steps)\n\t$Q echo 'Building all packages and submodules...'\n\t$Q go build ./...\n\t$Q cd common/archiver/gcloud; go build ./...\n\t$Q cd cmd/server; go build ./...\n\t$Q # \"tests\" by building and then running `true`, and hides test-success output\n\t$Q echo 'Building all tests (~5x slower)...'\n\t$Q # intentionally not -race due to !race build tags\n\t$Q go test -exec /usr/bin/true ./... >/dev/null\n\t$Q cd common/archiver/gcloud; go test -exec /usr/bin/true ./... >/dev/null\n\t$Q cd cmd/server; go test -exec /usr/bin/true ./... >/dev/null\n\ntidy: ## `go mod tidy` all packages\n\t$Q # tidy in dependency order\n\t$Q go mod tidy\n\t$Q cd common/archiver/gcloud; go mod tidy || (echo \"failed to tidy gcloud plugin, try manually copying go.mod contents into common/archiver/gcloud/go.mod and rerunning\" >&2; exit 1)\n\t$Q cd cmd/server; go mod tidy || (echo \"failed to tidy main server module, try manually copying go.mod and common/archiver/gcloud/go.mod contents into cmd/server/go.mod and rerunning\" >&2; exit 1)\n\nclean: ## Clean build products and SQLite database\n\trm -f $(BINS)\n\trm -Rf $(BUILD)\n\trm *.db\n\t$(if \\\n\t\t$(wildcard $(STABLE_BIN)/*), \\\n\t\t$(warning usually-stable build tools still exist, delete the $(STABLE_BIN) folder to rebuild them),)\n\n# v----- not yet cleaned up -----v\n\n.PHONY: git-submodules test bins build clean cover help\n\nTOOLS_CMD_ROOT=./cmd/tools\nINTEG_TEST_ROOT=./host\nINTEG_TEST_DIR=host\nINTEG_TEST_XDC_ROOT=./host/xdc\nINTEG_TEST_XDC_DIR=hostxdc\nINTEG_TEST_NDC_ROOT=./host/ndc\nINTEG_TEST_NDC_DIR=hostndc\n\n# INTEG_TEST_ROOT requires to use test flags that are not used in other sub directories\n# go test require all test package support the flags that are used in the test\n# So INTEG_TEST_DIRS contains all test directories of INTEG_TEST_ROOT that are not using test flags\nINTEG_TEST_DIRS=$$(go list $(INTEG_TEST_ROOT)/... | grep -v $(INTEG_TEST_XDC_ROOT) | grep -v $(INTEG_TEST_NDC_ROOT) | grep -v $(INTEG_TEST_ROOT)$$)\n\n# Opt out folders that shouldn't be run as part of unit tests such as integration tests, simulations.\n# Syntax: \"folder1% folder2%\" # space separated list of folders to opt out\nOPT_OUT_TEST_FOLDERS=./simulation%\n\nTEST_TIMEOUT ?= 20m\nTEST_ARG ?= -race $(if $(verbose),-v) -timeout $(TEST_TIMEOUT)\n\n# TODO to be consistent, use nosql as PERSISTENCE_TYPE and cassandra PERSISTENCE_PLUGIN\n# file names like integ_cassandra__cover should become integ_nosql_cassandra_cover\n# for https://github.com/uber/cadence/issues/3514\nPERSISTENCE_TYPE ?= cassandra\nTEST_RUN_COUNT ?= 1\nifdef TEST_TAG\noverride TEST_TAG := -tags $(TEST_TAG)\nendif\n\n# all directories with *_test.go files in them (exclude host/xdc)\nTEST_DIRS := $(filter-out $(INTEG_TEST_XDC_ROOT)%, $(sort $(dir $(filter %_test.go,$(ALL_SRC)))))\n# all tests other than end-to-end integration test fall into the pkg_test category.\n# ?= allows passing specific (space-separated) dirs for faster testing\nPKG_TEST_DIRS ?= $(filter-out $(INTEG_TEST_ROOT)% $(OPT_OUT_TEST_FOLDERS), $(TEST_DIRS))\n\n# Code coverage output files\nCOVER_ROOT                      := $(BUILD)/coverage\nUNIT_COVER_FILE                 := $(COVER_ROOT)/unit_cover.out\n\nINTEG_COVER_FILE                := $(COVER_ROOT)/integ_$(PERSISTENCE_TYPE)_$(PERSISTENCE_PLUGIN)_cover.out\nINTEG_COVER_FILE_CASS           := $(COVER_ROOT)/integ_cassandra__cover.out\nINTEG_COVER_FILE_MYSQL          := $(COVER_ROOT)/integ_sql_mysql_cover.out\nINTEG_COVER_FILE_POSTGRES       := $(COVER_ROOT)/integ_sql_postgres_cover.out\nINTEG_COVER_FILE_ETCD           := $(COVER_ROOT)/integ_etcd_cover.out\n\nINTEG_NDC_COVER_FILE            := $(COVER_ROOT)/integ_ndc_$(PERSISTENCE_TYPE)_$(PERSISTENCE_PLUGIN)_cover.out\nINTEG_NDC_COVER_FILE_CASS       := $(COVER_ROOT)/integ_ndc_cassandra__cover.out\nINTEG_NDC_COVER_FILE_MYSQL      := $(COVER_ROOT)/integ_ndc_sql_mysql_cover.out\nINTEG_NDC_COVER_FILE_POSTGRES   := $(COVER_ROOT)/integ_ndc_sql_postgres_cover.out\n\n# Need the following option to have integration tests\n# count towards coverage. godoc below:\n# -coverpkg pkg1,pkg2,pkg3\n#   Apply coverage analysis in each test to the given list of packages.\n#   The default is for each test to analyze only the package being tested.\n#   Packages are specified as import paths.\nCOVER_PKGS = client common host service tools\n# pkg -> pkg/... -> github.com/uber/cadence/pkg/... -> join with commas\nGOCOVERPKG_ARG := -coverpkg=\"$(subst $(SPACE),$(COMMA),$(addprefix $(PROJECT_ROOT)/,$(addsuffix /...,$(COVER_PKGS))))\"\n\n# iterates over a list of dirs and runs go test on each one, collecting errors as it runs.\n# this is primarily written because it's a verbose bit of boilerplate, until we switch to `go test ./...` where possible.\n# CAUTION: when changing to `go test ./...`, note that this DOES NOT test submodules.  Those must be run separately.\ndefine looptest\n$Q FAIL=\"\"; for dir in $1; do \\\n\tgo test $(TEST_ARG) -coverprofile=$@ \"$$dir\" $(TEST_TAG) 2>&1 | tee -a test.log || FAIL=\"$$FAIL $$dir\"; \\\ndone; test -z \"$$FAIL\" || (echo \"Failed packages; $$FAIL\"; exit 1)\nendef\n\ntest: ## Build and run all tests locally\n\t$Q rm -f test\n\t$Q rm -f test.log\n\t$Q echo Running special test cases without race detector:\n\t$Q go test -v ./cmd/server/cadence/\n\t$Q $(call looptest,$(PKG_TEST_DIRS))\n\ntest_dirs:\n\techo $(PKG_TEST_DIRS)\n\ntest_e2e:\n\t$Q rm -f test\n\t$Q rm -f test.log\n\t$Q $(call looptest,$(INTEG_TEST_ROOT))\n\n# need to run end-to-end xdc tests with race detector off because of ringpop bug causing data race issue\ntest_e2e_xdc:\n\t$Q rm -f test\n\t$Q rm -f test.log\n\t$Q $(call looptest,$(INTEG_TEST_XDC_ROOT))\n\ncover_profile:\n\t$Q mkdir -p $(BUILD)\n\t$Q mkdir -p $(COVER_ROOT)\n\t$Q echo \"mode: atomic\" > $(UNIT_COVER_FILE)\n\n\t$Q echo Running special test cases without race detector:\n\t$Q go test ./cmd/server/cadence/\n\t$Q echo Running package tests:\n\t$Q go test $(PKG_TEST_DIRS) $(TEST_ARG) -coverprofile=$(UNIT_COVER_FILE)\n\ncover_integration_profile:\n\t$Q mkdir -p $(BUILD)\n\t$Q mkdir -p $(COVER_ROOT)\n\t$Q echo \"mode: atomic\" > $(INTEG_COVER_FILE)\n\n\t$Q echo Running integration test with $(PERSISTENCE_TYPE) $(PERSISTENCE_PLUGIN)\n\t$Q mkdir -p $(BUILD)/$(INTEG_TEST_DIR)\n\t$Q time go test $(INTEG_TEST_DIRS) $(TEST_ARG) $(TEST_TAG) $(GOCOVERPKG_ARG) -coverprofile=$(BUILD)/$(INTEG_TEST_DIR)/coverage.out || exit 1;\n\t$Q cat $(BUILD)/$(INTEG_TEST_DIR)/coverage.out | grep -v \"^mode: \\w\\+\" >> $(INTEG_COVER_FILE)\n\t$Q time go test $(INTEG_TEST_ROOT) $(TEST_ARG) $(TEST_TAG) -persistenceType=$(PERSISTENCE_TYPE) -sqlPluginName=$(PERSISTENCE_PLUGIN) $(GOCOVERPKG_ARG) -coverprofile=$(BUILD)/$(INTEG_TEST_DIR)/coverage.out || exit 1;\n\t$Q cat $(BUILD)/$(INTEG_TEST_DIR)/coverage.out | grep -v \"^mode: \\w\\+\" >> $(INTEG_COVER_FILE)\n\nintegration_tests_etcd:\n\t$Q mkdir -p $(BUILD)\n\t$Q mkdir -p $(COVER_ROOT)\n\t$Q echo \"mode: atomic\" > $(INTEG_COVER_FILE_ETCD)\n\n\t$Q echo \"Running integration tests with etcd\"\n\t$Q mkdir -p $(BUILD)/$(INTEG_TEST_DIR)\n\t$Q (ETCD_TEST_DIRS=$$(find . -name \"*_test.go\" -exec grep -l \"testhelper.SetupStoreTestCluster\\|testflags.RequireEtcd\" {} \\; | xargs -n1 dirname | sort | uniq); \\\n\t\techo \"Found etcd test directories:\"; \\\n\t\techo \"$$ETCD_TEST_DIRS\"; \\\n\t\techo \"Using ETCD_ENDPOINTS='$(ETCD_ENDPOINTS)'\"; \\\n\t\tETCD=1 ETCD_ENDPOINTS=\"$(ETCD_ENDPOINTS)\" time go test $$ETCD_TEST_DIRS $(TEST_ARG) $(TEST_TAG) -coverprofile=$(BUILD)/$(INTEG_TEST_DIR)/coverage.out || exit 1)\n\t$Q cat $(BUILD)/$(INTEG_TEST_DIR)/coverage.out | grep -v \"^mode: \\w\\+\" >> $(INTEG_COVER_FILE_ETCD)\n\ncover_ndc_profile:\n\t$Q mkdir -p $(BUILD)\n\t$Q mkdir -p $(COVER_ROOT)\n\t$Q echo \"mode: atomic\" > $(INTEG_NDC_COVER_FILE)\n\n\t$Q echo Running integration test for 3+ dc with $(PERSISTENCE_TYPE) $(PERSISTENCE_PLUGIN)\n\t$Q mkdir -p $(BUILD)/$(INTEG_TEST_NDC_DIR)\n\t$Q time go test -timeout $(TEST_TIMEOUT) $(INTEG_TEST_NDC_ROOT) $(TEST_TAG) -persistenceType=$(PERSISTENCE_TYPE) -sqlPluginName=$(PERSISTENCE_PLUGIN) $(GOCOVERPKG_ARG) -coverprofile=$(BUILD)/$(INTEG_TEST_NDC_DIR)/coverage.out -count=$(TEST_RUN_COUNT) || exit 1;\n\t$Q cat $(BUILD)/$(INTEG_TEST_NDC_DIR)/coverage.out | grep -v \"^mode: \\w\\+\" | grep -v \"mode: set\" >> $(INTEG_NDC_COVER_FILE)\n\n$(COVER_ROOT)/cover.out: $(UNIT_COVER_FILE) $(INTEG_COVER_FILE_CASS) $(INTEG_COVER_FILE_MYSQL) $(INTEG_COVER_FILE_POSTGRES) $(INTEG_NDC_COVER_FILE_CASS) $(INTEG_NDC_COVER_FILE_MYSQL) $(INTEG_NDC_COVER_FILE_POSTGRES)\n\t$Q echo \"mode: atomic\" > $(COVER_ROOT)/cover.out\n\tcat $(UNIT_COVER_FILE) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_COVER_FILE_CASS) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_COVER_FILE_MYSQL) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_COVER_FILE_POSTGRES) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_NDC_COVER_FILE_CASS) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_NDC_COVER_FILE_MYSQL) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_NDC_COVER_FILE_POSTGRES) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\tcat $(INTEG_COVER_FILE_ETCD) | grep -v \"^mode: \\w\\+\" | grep -vP \".gen|_generated|[Mm]ock[s]?\" >> $(COVER_ROOT)/cover.out\n\ncover: $(COVER_ROOT)/cover.out\n\tgo tool cover -html=$(COVER_ROOT)/cover.out;\n\ninstall-schema: cadence-cassandra-tool\n\t$Q echo installing schema\n\t./cadence-cassandra-tool create -k cadence --rf 1\n\t./cadence-cassandra-tool -k cadence setup-schema -v 0.0\n\t./cadence-cassandra-tool -k cadence update-schema -d ./schema/cassandra/cadence/versioned\n\t./cadence-cassandra-tool create -k cadence_visibility --rf 1\n\t./cadence-cassandra-tool -k cadence_visibility setup-schema -v 0.0\n\t./cadence-cassandra-tool -k cadence_visibility update-schema -d ./schema/cassandra/visibility/versioned\n\t$Q echo installed schema\n\ninstall-schema-mysql: cadence-sql-tool\n\t./cadence-sql-tool --user root --pw cadence create --db cadence\n\t./cadence-sql-tool --user root --pw cadence --db cadence setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence update-schema -d ./schema/mysql/v8/cadence/versioned\n\t./cadence-sql-tool --user root --pw cadence create --db cadence_visibility\n\t./cadence-sql-tool --user root --pw cadence --db cadence_visibility setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence_visibility update-schema -d ./schema/mysql/v8/visibility/versioned\n\ninstall-schema-multiple-mysql: cadence-sql-tool install-schema-es-v7\n\t./cadence-sql-tool --user root --pw cadence create --db cadence0\n\t./cadence-sql-tool --user root --pw cadence --db cadence0 setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence0 update-schema -d ./schema/mysql/v8/cadence/versioned\n\t./cadence-sql-tool --user root --pw cadence create --db cadence1\n\t./cadence-sql-tool --user root --pw cadence --db cadence1 setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence1 update-schema -d ./schema/mysql/v8/cadence/versioned\n\t./cadence-sql-tool --user root --pw cadence create --db cadence2\n\t./cadence-sql-tool --user root --pw cadence --db cadence2 setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence2 update-schema -d ./schema/mysql/v8/cadence/versioned\n\t./cadence-sql-tool --user root --pw cadence create --db cadence3\n\t./cadence-sql-tool --user root --pw cadence --db cadence3 setup-schema -v 0.0\n\t./cadence-sql-tool --user root --pw cadence --db cadence3 update-schema -d ./schema/mysql/v8/cadence/versioned\n\ninstall-schema-postgres: cadence-sql-tool\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres create --db cadence\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres --db cadence setup -v 0.0\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres --db cadence update-schema -d ./schema/postgres/cadence/versioned\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres create --db cadence_visibility\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres --db cadence_visibility setup-schema -v 0.0\n\t./cadence-sql-tool -p 5432 -u postgres -pw cadence --pl postgres --db cadence_visibility update-schema -d ./schema/postgres/visibility/versioned\n\ninstall-schema-sqlite: cadence-sql-tool\n\t./cadence-sql-tool -pl sqlite --db cadence.db setup -v 0.0\n\t./cadence-sql-tool -pl sqlite --db cadence.db update-schema -d ./schema/sqlite/cadence/versioned\n\t./cadence-sql-tool -pl sqlite --db cadence_visibility.db setup -v 0.0\n\t./cadence-sql-tool -pl sqlite --db cadence_visibility.db update-schema -d ./schema/sqlite/visibility/versioned\n\ninstall-schema-es-v7:\n\tcurl -X PUT \"http://127.0.0.1:9200/_template/cadence-visibility-template\" -H 'Content-Type: application/json' -d @./schema/elasticsearch/v7/visibility/index_template.json\n\tcurl -X PUT \"http://127.0.0.1:9200/cadence-visibility-dev\"\n\ninstall-schema-es-v6:\n\tcurl -X PUT \"http://127.0.0.1:9200/_template/cadence-visibility-template\" -H 'Content-Type: application/json' -d @./schema/elasticsearch/v6/visibility/index_template.json\n\tcurl -X PUT \"http://127.0.0.1:9200/cadence-visibility-dev\"\n\ninstall-schema-es-opensearch:\n\tcurl -X PUT \"http://127.0.0.1:9200/_template/cadence-visibility-template\" -H 'Content-Type: application/json' -d @./schema/elasticsearch/os2/visibility/index_template.json\n\tcurl -X PUT \"http://127.0.0.1:9200/cadence-visibility-dev\"\n\nstart: bins\n\t./cadence-server start\n\ninstall-schema-xdc: cadence-cassandra-tool\n\t$Q echo Setting up cadence_cluster0 key space\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_cluster0 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster0 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster0 update-schema -d ./schema/cassandra/cadence/versioned\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_visibility_cluster0 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster0 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster0 update-schema -d ./schema/cassandra/visibility/versioned\n\n\t$Q echo Setting up cadence_cluster1 key space\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_cluster1 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster1 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster1 update-schema -d ./schema/cassandra/cadence/versioned\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_visibility_cluster1 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster1 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster1 update-schema -d ./schema/cassandra/visibility/versioned\n\n\t$Q echo Setting up cadence_cluster2 key space\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_cluster2 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster2 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_cluster2 update-schema -d ./schema/cassandra/cadence/versioned\n\t./cadence-cassandra-tool --ep 127.0.0.1 create -k cadence_visibility_cluster2 --rf 1\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster2 setup-schema -v 0.0\n\t./cadence-cassandra-tool --ep 127.0.0.1 -k cadence_visibility_cluster2 update-schema -d ./schema/cassandra/visibility/versioned\n\nstart-xdc-cluster0: cadence-server\n\t./cadence-server --zone xdc_cluster0 start\n\nstart-xdc-cluster1: cadence-server\n\t./cadence-server --zone xdc_cluster1 start\n\nstart-xdc-cluster2: cadence-server\n\t./cadence-server --zone xdc_cluster2 start\n\nstart-canary: cadence-canary\n\t./cadence-canary start\n\nstart-sharddistributor-canary: sharddistributor-canary\n\t./sharddistributor-canary start\n\nstart-bench: cadence-bench\n\t./cadence-bench start\n\nstart-mysql: cadence-server\n\t./cadence-server --zone mysql start\n\nstart-postgres: cadence-server\n\t./cadence-server --zone postgres start\n\n# broken up into multiple += so I can interleave comments.\n# this all becomes a single line of output.\n# you must not use single-quotes within the string in this var.\nJQ_DEPS_AGE = jq '\n# only deal with things with updates\nJQ_DEPS_AGE += select(.Update)\n# allow additional filtering, e.g. DEPS_FILTER='$(JQ_DEPS_ONLY_DIRECT)'\nJQ_DEPS_AGE += $(DEPS_FILTER)\n# add \"days between current version and latest version\"\nJQ_DEPS_AGE += | . + {Age:(((.Update.Time | fromdate) - (.Time | fromdate))/60/60/24 | floor)}\n# add \"days between latest version and now\"\nJQ_DEPS_AGE += | . + {Available:((now - (.Update.Time | fromdate))/60/60/24 | floor)}\n# 123 days: library \told_version -> new_version\nJQ_DEPS_AGE += | ([.Age, .Available] | max | tostring) + \" days: \" + .Path + \"  \\t\" + .Version + \" -> \" + .Update.Version\nJQ_DEPS_AGE += '\n# remove surrounding quotes from output\nJQ_DEPS_AGE += --raw-output\n\n# exclude `\"Indirect\": true` dependencies.  direct ones have no \"Indirect\" key at all.\nJQ_DEPS_ONLY_DIRECT = | select(has(\"Indirect\") | not)\n\ndeps: ## Check for dependency updates, for things that are directly imported\n\t$Q make --no-print-directory DEPS_FILTER='$(JQ_DEPS_ONLY_DIRECT)' deps-all\n\ndeps-all: ## Check for all dependency updates\n\t$Q go list -u -m -json all \\\n\t\t| $(JQ_DEPS_AGE) \\\n\t\t| sort -n\n\nhelp: ## Prints a help message showing any specially-commented targets\n\t$Q # print the high-value ones first, so they're more visible.  the \".....\" prefixes match the shell coloring chars\n\t$Q cat $(MAKEFILE_LIST) | grep -e \"^[a-zA-Z_\\-]*:.* ## .*\" | awk 'BEGIN {FS = \":.*? ## \"}; {printf \"\\033[36m%-20s\\033[0m %s\\n\", $$1, $$2}' | sort | grep -E '^.....(help|pr|bins)\\b'\n\t$Q echo '-----------------------------------'\n\t$Q cat $(MAKEFILE_LIST) | grep -e \"^[a-zA-Z_\\-]*:.* ## .*\" | awk 'BEGIN {FS = \":.*? ## \"}; {printf \"\\033[36m%-20s\\033[0m %s\\n\", $$1, $$2}' | sort | grep -vE '^.....(help|pr|bins)\\b'\n"
  },
  {
    "path": "NOTICE",
    "content": "Copyright (c) 2025 Uber Technologies, Inc.\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\n of this software and associated documentation files (the \"Software\"), to deal\n in the Software without restriction, including without limitation the rights\n to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n copies of the Software, and to permit persons to whom the Software is\n furnished to do so, subject to the following conditions:\n\n The above copyright notice and this permission notice shall be included in\n all copies or substantial portions of the Software.\n\n THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n THE SOFTWARE.\n"
  },
  {
    "path": "PROPOSALS.md",
    "content": "# Proposing Changes to Cadence\n\n## Introduction\n\nThe design process for changes to Cadence is modeled on the [proposal process used by the M3 project](https://github.com/m3db/proposal).\n\n## Process\n\n- [Create an issue](https://github.com/cadence-workflow/cadence/issues/new) describing the proposal.\n\n- Like any GitHub issue, a proposal issue is followed by an initial discussion about the suggestion. For Proposal issues:\n\n  - The goal of the initial discussion is to reach agreement on the next step: (1) accept, (2) decline, or (3) ask for a design doc.\n  - The discussion is expected to be resolved in a timely manner.\n  - If the author wants to write a design doc, then they can write one.\n  - A lack of agreement means the author should write a design doc.\n\n- If a Proposal issue leads to a design doc:\n\n  - The design doc should be presented as a Google Doc and must follow [the template](https://docs.google.com/document/d/1hpWpy5MB5l8uXfnl23lebhx-dvo79vb1jFUOyAtIHJw/edit?usp=sharing).\n  - The design doc should be linked from the opened GitHub issue.\n  - The design doc should only allow edit access to authors of the document.\n  - Do not create the document from a corporate G Suite account. If you want to edit from a corporate G Suite account then first create the document from a personal Google account and grant edit access to your corporate G Suite account.\n  - Comment access should also be accessible by the public. Make sure this is the case by clicking \"Share\" and \"Get shareable link\" ensuring you select \"Anyone with the link can comment\".\n\n- Once comments and revisions on the design doc wind down, there is a final discussion about the proposal.\n\n  - The goal of the final discussion is to reach agreement on the next step: (1) accept or (2) decline.\n  - The discussion is expected to be resolved in a timely manner.\n\n- Once the design doc is agreed upon, the author shall export the contents of the Google Doc as a Markdown document and submit a PR to add it to the proposal repository.\n\n  - Authors can use [gdocs2md](https://github.com/mangini/gdocs2md) to convert the document but any cleanup required must be done by hand.\n  - The design doc should be checked in at `docs/design/XYZ-shortname.md`, where `XYZ` is the GitHub issue number and `shortname` is a short name (a few dash-separated words at most).\n  - The design doc should be added to the list of existing proposals in `docs/design/index.md`\n\n- The author (and/or other contributors) do the work as described by the \"Implementation\" section of the proposal.\n\n---\n\nThis project is released under the [Apache License, Version 2.0](LICENSE).\n"
  },
  {
    "path": "README.md",
    "content": "# Cadence\n[![Build Status](https://github.com/cadence-workflow/cadence/actions/workflows/ci-checks.yml/badge.svg)](https://github.com/cadence-workflow/cadence/actions/workflows/ci-checks.yml)\n[![Coverage](https://codecov.io/gh/cadence-workflow/cadence/graph/badge.svg?token=7SD244ImNF)](https://codecov.io/gh/cadence-workflow/cadence)\n[![Slack Status](https://img.shields.io/badge/slack-join_chat-white.svg?logo=slack&style=social)](https://communityinviter.com/apps/cloud-native/cncf)\n[![Github release](https://img.shields.io/github/v/release/cadence-workflow/cadence.svg)](https://github.com/cadence-workflow/cadence/releases)\n[![License](https://img.shields.io/github/license/cadence-workflow/cadence.svg)](http://www.apache.org/licenses/LICENSE-2.0)\n\nCadence Workflow is an open-source platform since 2017 for building and running scalable, fault-tolerant, and long-running workflows. This repository contains the core orchestration engine and tools including CLI, schema managment, benchmark and canary.\n\n\n## Getting Started\n\nCadence backend consists of multiple services, a database (Cassandra/MySQL/PostgreSQL) and optionally Kafka+Elasticsearch.\nAs a user, you need a worker which contains your workflow implementation.\nOnce you have Cadence backend and worker(s) running, you can trigger workflows by using SDKs or via CLI.\n\n1. Start cadence backend components locally\n\n```\ndocker compose -f docker/docker-compose.yml up\n```\n\n2. Run the Samples\n\nTry out the sample recipes for [Go](https://github.com/cadence-workflow/cadence-samples) or [Java](https://github.com/cadence-workflow/cadence-java-samples).\n\n3. Visit UI\n\nVisit http://localhost:8088 to check workflow histories and detailed traces.\n\n\n### Client Libraries\nYou can implement your workflows with one of our client libraries:\n- [Official Cadence Go SDK](https://github.com/cadence-workflow/cadence-go-client)\n- [Official Cadence Java SDK](https://github.com/cadence-workflow/cadence-java-client)\nThere are also unofficial [Python](https://github.com/firdaus/cadence-python) and [Ruby](https://github.com/coinbase/cadence-ruby) SDKs developed by the community.\n\nYou can also use [iWF](https://github.com/indeedeng/iwf) as a DSL framework on top of Cadence.\n\n### CLI\n\nCadence CLI can be used to operate workflows, tasklist, domain and even the clusters.\n\nYou can use the following ways to install Cadence CLI:\n* Use brew to install CLI: `brew install cadence-workflow`\n  * Follow the [instructions](https://github.com/cadence-workflow/cadence/discussions/4457) if you need to install older versions of CLI via homebrew. Usually this is only needed when you are running a server of a too old version.\n* Use docker image for CLI: `docker run --rm ubercadence/cli:<releaseVersion>`  or `docker run --rm ubercadence/cli:master ` . Be sure to update your image when you want to try new features: `docker pull ubercadence/cli:master `\n* Build the CLI binary yourself, check out the repo and run `make cadence` to build all tools. See [CONTRIBUTING](CONTRIBUTING.md) for prerequisite of make command.\n* Build the CLI image yourself, see [instructions](docker/README.md#diy-building-an-image-for-any-tag-or-branch)\n\nCadence CLI is a powerful tool. The commands are organized by tabs. E.g. `workflow`->`batch`->`start`, or `admin`->`workflow`->`describe`.\n\nPlease read the [documentation](https://cadenceworkflow.io/docs/cli/#documentation) and always try out `--help` on any tab to learn & explore.\n\n### UI\n\nTry out [Cadence Web UI](https://github.com/cadence-workflow/cadence-web) to view your workflows on Cadence.\n(This is already available at localhost:8088 if you run Cadence with docker compose)\n\n\n### Other binaries in this repo\n\n#### Bench/stress test workflow tools\nSee [bench documentation](./bench/README.md).\n\n#### Periodical feature health check workflow tools(aka Canary)\nSee [canary documentation](./canary/README.md).\n\n#### Schema tools for SQL and Cassandra\nThe tools are for [manual setup or upgrading database schema](docs/persistence.md)\n\n  * If server runs with Cassandra, Use [Cadence Cassandra tool](tools/cassandra/README.md)\n  * If server runs with SQL database, Use [Cadence SQL tool](tools/sql/README.md)\n\nThe easiest way to get the schema tool is via homebrew.\n\n`brew install cadence-workflow` also includes `cadence-sql-tool` and `cadence-cassandra-tool`.\n * The schema files are located at `/usr/local/etc/cadence/schema/`.\n * To upgrade, make sure you remove the old ElasticSearch schema first: `mv /usr/local/etc/cadence/schema/elasticsearch /usr/local/etc/cadence/schema/elasticsearch.old && brew upgrade cadence-workflow`. Otherwise ElasticSearch schemas may not be able to get updated.\n * Follow the [instructions](https://github.com/cadence-workflow/cadence/discussions/4457) if you need to install older versions of schema tools via homebrew.\n However, easier way is to use new versions of schema tools with old versions of schemas.\n All you need is to check out the older version of schemas from this repo. Run `git checkout v0.21.3` to get the v0.21.3 schemas in [the schema folder](/schema).\n\n\n## Contributing\n\nWe'd love your help in making Cadence great. Please review our [contribution guide](CONTRIBUTING.md).\n\nIf you'd like to propose a new feature, first join the [CNCF Slack workspace](https://communityinviter.com/apps/cloud-native/cncf) in the **#cadence-users** channel to start a discussion.\n\nPlease visit our [documentation](https://cadenceworkflow.io/docs/operation-guide/) site for production/cluster setup.\n\n\n### Learning Resources\nSee Maxim's talk at [Data@Scale Conference](https://atscaleconference.com/videos/cadence-microservice-architecture-beyond-requestreply) for an architectural overview of Cadence.\n\nVisit [cadenceworkflow.io](https://cadenceworkflow.io) to learn more about Cadence. Join us in [Cadence Documentation](https://github.com/cadence-workflow/Cadence-Docs) project. Feel free to raise an Issue or Pull Request there.\n\n### Community\n* [Github Discussion](https://github.com/cadence-workflow/cadence/discussions)\n  * Best for Q&A, support/help, general discusion, and annoucement\n* [Github Issues](https://github.com/cadence-workflow/cadence/issues)\n  * Best for reporting bugs and feature requests\n* [StackOverflow](https://stackoverflow.com/questions/tagged/cadence-workflow)\n  * Best for Q&A and general discusion\n* [Slack](https://communityinviter.com/apps/cloud-native/cncf) - Join **#cadence-users** channel on CNCF Slack\n  * Best for contributing/development discussion\n\n\n## Stars over time\n[![Stargazers over time](https://starchart.cc/uber/cadence.svg?variant=adaptive)](https://starchart.cc/uber/cadence)\n\n\n## License\n\nApache 2.0 License, please see [LICENSE](https://github.com/cadence-workflow/cadence/blob/master/LICENSE) for details.\n"
  },
  {
    "path": "RELEASES.md",
    "content": "# Releases upgrade instruction\n\n## Upgrade to 0.16 and above\n\n**TL;DR:** If your Cadence service is running on or above 0.14.0 and you do not have workflows running for more than 6 months. It is safe to upgrade without any operations.\n\nIf your cluster has open workflows for more than 6 months, please download and run the following CLI command prior to the upgrades.\n\nPrior to release 0.16, the workflow state structure contains a field called replication state. Instead, we introduce a new field 'version histories' \nto manage the replication. In release 0.16, this field 'replication state' has been removed from the code path.\nFrom release 0.14, the field 'replication state' has been deprecated. If you are upgrading the service from version below 0.14.0 to 0.16.0 or \nyou may have workflows running for more than 6 months, please consider follow the instruction to see if you have any unsupported workflows.\n\n## Detect unsupported open workflows\n\nRun the following command based on your database.\n\nCassandra:\n\n`cadence admin db unsupported-workflow --db_type=cassandra --db_address <db address> --db_port <port number> --username=<username> --password=<password> --keyspace <keyspace> --lower_shard_bound=<the start shard id> --upper_shard_bound=<the end shard id> --rps <scan rps> --output_filename ./cadence_scan`\n\nMySQL/Postgres:\n\n`cadence admin db unsupported-workflow --db_type=<mysql/postgres> --db_address <db address> --db_port <port number> --username=<username> --password=<password> --db_name <database name> --lower_shard_bound=<the start shard id> --upper_shard_bound=<the end shard id> --rps <scan rps> --output_filename ./cadence_scan`\n\n**Note:** This CLI is a long-running process to scan the database for unsupported workflows.\nIf you have TLS configurations or use customized encoding/decoding type. Please use\n\n`cadence adm db unsupported-workflow --help`\n\nto configurate the correct connection.\n\n\nAfter the CLI completes, a list of unsupported workflows will be listed in the file `cadence_scan`.\n\nFor example:\n\n`cadence --address <host>:<port> --domain <ee7316ab-2e42-4f3c-86ee-344a4299791c> workflow reset --wid helloworld_7f2fa1fe-594e-44ad-a9e2-7ac7c5f97861 --rid e79ce972-b657-48a4-bba9-e2f2ac6424e2 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'`\n\n## Reset the workflow \n\nUse reset CLI to reset these unsupported workflow. This operation is to migrate these unsupported workflow data.\n\n1. replace the host and port with your cadence host address and port number.\n2. replace the domain uuid with the domain name.\nTo find the domain name, you can use `cadence --address <host>:<port> domain describe --domain_id=<domain uuid>`.\n\nAfter replacing the fields with the correct values, you can copy/paste to run the CLI. This is going to reset those workflows to the last decision task completed event.\n\n**Note:** If your workflows has outstanding child workflows, you have to wait for these child workflows completion or terminate the workflows.\n"
  },
  {
    "path": "bench/README.md",
    "content": "Cadence Bench Tests\n===================\n\nThis README describes how to set up Cadence bench, different types of bench loads, and how to start the load.\n\nSetup\n-----------\n### Cadence server\n\nBench suite is running against a Cadence server/cluster. See [documentation](https://cadenceworkflow.io/docs/operation-guide/setup/) for Cadence server cluster setup.\n\nNote that only the Basic bench test don't require Advanced Visibility. \n \nOther advanced bench tests requires Cadence server with Advanced Visibility. \n\nFor local env you can run it through:\n- Docker: Instructions for running Cadence server through docker can be found in `docker/README.md`. Either `docker-compose-es-v7.yml` or `docker-compose-es.yml` can be used to start the server.\n- Build from source: Please check [CONTRIBUTING](/CONTRIBUTING.md) for how to build and run Cadence server from source. Please also make sure Kafka and ElasticSearch are running before starting the server with `./cadence-server --zone es start`. If ElasticSearch v7 is used, change the value for `--zone` flag to `es_v7`.\n\nSee more [documentation here](https://cadenceworkflow.io/docs/concepts/search-workflows/).\n\n### Bench Workers\n:warning: NOTE: Starting this bench worker will not automatically start a bench test. Next two sections will cover how to start and configure it.\n\nDifferent ways of start the bench workers:\n\n#### 1. Use docker image `ubercadence/cadence-bench:master`\n\nYou can [pre-built docker-compose file](../docker/docker-compose-bench.yml) to run against local server\nIn the `docker/` directory, run:\n```\ndocker compose -f docker-compose-bench.yml up\n```\nYou can modify [the bench worker config](../docker/config/bench/development.yaml) to run against a prod server cluster. \n\nOr may run it with Kubernetes, for [example](https://github.com/longquanzheng/cadence-lab/blob/master/eks/bench-deployment.yaml). \n\n\nNOTE: Similar to server/CLI images, the `master` image will be built and published automatically by Github on every commit onto the `master` branch.\nTo use a different image than `master` tag.  See [docker hub](https://hub.docker.com/repository/docker/ubercadence/cadence-bench) for all the images.     \n\n#### 2.  Build & Run the binary \n\nIn the project root, build cadence bench binary:\n   ```\n   make cadence-bench\n   ```\n\nThen start bench worker:\n   ```\n   ./cadence-bench start\n   ```\nBy default, it will load [the configuration in `config/bench/development.yaml`](../config/bench/development.yaml). \nRun `./cadence-bench -h` for details to understand the start options of how to change the loading directory if needed. \n\nWorker Configurations\n----------------------\nBench workers configuration contains two parts:\n- **Bench**: this part controls the client side, including the bench service name, which domains bench workers are responsible for and how many taskLists each domain should use.\n```yaml \nbench:\n  name: \"cadence-bench\" # bench name\n  domains: [\"cadence-bench\", \"cadence-bench-sync\", \"cadence-bench-batch\"] # it will start workers on all those domains(also try to register if not exists) \n  numTaskLists: 3 # it will start workers listening on cadence-bench-tl-0, cadence-bench-tl-1,  cadence-bench-tl-2\n``` \n1. Bench workers will only poll from task lists whose name start with `cadence-bench-tl-`. If in the configuration, `numTaskLists` is specified to be 2, then workers will only listen to `cadence-bench-tl-0` and `cadence-bench-tl-1`. So make sure you use a valid task list name when starting the bench load.\n2. When starting bench workers, it will try to register a **local domain with archival feature disabled** for each domain name listed in the configuration, if not already exists. If your want to test the performance of global domains and/or archival feature, please register the domains first before starting the worker.\n\n- **Cadence**: this control how bench worker should talk to Cadence server, which includes the server's service name and address.\n```yaml\ncadence:\n  service: \"cadence-frontend\" # frontend service name\n  host: \"127.0.0.1:7833\" # frontend address\n  #metrics: ... # optional detailed client side metrics like workflow latency  \n```\n- **Metrics**: metrics configuration. Similar to server metric emitter, only M3/Statsd/Prometheus is supported. \n- **Log**: logging configuration.  Similar to server logging configuration. \n\n\nBench Load Types\n-----------\nThis section briefly describes the purpose of each bench load and provides a sample command for running the load. Detailed descriptions for each test's configuration can be found in `bench/lib/config.go`\n\nPlease note that all load configurations in `config/bench` is for only local development and illustration purpose, it does not reflect the actual capability of Cadence server.\n\n### Basic\n:warning: NOTE: This is the only bench test which doesn't require advanced visibility feature on the server. Make sure you set `useBasicVisibilityValidation` to true if run with basic(db) visibility.  \nAlso basicVisibilityValidation requires only one test load run in the same domain. This is because of the limitation of basic visibility now allow using workflowType and status filters at the same time.  \n\nAs the name suggests, this load tests the basic case of load testing. \nYou will start a `launchWorkflow` which will execute some `launchActivities` to start `stressWorkflows`. Then the stressWorkflows running activities in sequential/parallel.\nOnce all stressWorkflows are started, launchWorkflow will wait stressWorkflows timeout + buffer time(default to 5 mins) before checking the status of all test workflows. \n\nTwo criteria must be met to pass the verification:\n1. No open workflows(this means server may lose some tasks and not able to close the stressWorkflows)\n2. Failed/timeouted workflows <= threshold(totalLaunchCount * failureThreshold )\n\nThe basic load can also be run in \"panic\" mode by setting `\"panicStressWorkflow\": true,` to test if server can handle large number of panic workflows (which can be caused by a bad worker deployment).\n\nSample configuration can be found in `config/bench/basic.json` and `config/ben/basic_panic.json`. To start the test, a sample command can be\n```\ncadence --do <domain> wf start --tl cadence-bench-tl-0 --wt basic-load-test-workflow --dt 30 --et 3600 --if config/bench/basic.json\n```\n\n`<domain>` needs to be one of the domains in bench config (by default  ./config/bench/development.yaml), e.g. `cadence-bench`. \n\nThen wait for the bench test result. \n```\n$cadence --do cadence-bench wf ob -w a2813321-a1bd-40c6-934f-88ad0ded6037\nProgress:\n  1, 2021-08-20T11:49:14-07:00, WorkflowExecutionStarted\n  2, 2021-08-20T11:49:14-07:00, DecisionTaskScheduled\n...\n...\n  20, 2021-08-20T11:59:24-07:00, DecisionTaskStarted\n  21, 2021-08-20T11:59:24-07:00, DecisionTaskCompleted\n  22, 2021-08-20T11:59:24-07:00, WorkflowExecutionCompleted\n\nResult:\n  Run Time: 26 seconds\n  Status: COMPLETED\n  Output: \"TEST PASSED. Details report: timeoutCount: 0, failedCount: 0, openCount:0, launchCount: 100, maxThreshold:1\"\n\n```\n\nThe output/error result shows whether the test passes with detailed report.\n\nConfiguration of basic load type. The config is passed as the launch workflow input parameter using a JSON file. \n \n```yaml\n# configuration for launch workflow\nuseBasicVisibilityValidation:   use basic(db based) visibility to verify the stress workflows, default false which requires advanced visibility on the server\ntotalLaunchCount\t: total number of stressWorkflows that started by the launchWorkflow\nwaitTimeBufferInSeconds : buffer time in addition of ExecutionStartToCloseTimeoutInSeconds to wait for stressWorkflows before verification, default 300(5 minutes)\nroutineCount\t: number of in-parallel launch activities that started by launchWorkflow, to start the stressWorkflows\nfailureThreshold\t: the threshold of failed stressWorkflow for deciding whether or not the whole testSuite failed.\nmaxLauncherActivityRetryCount   : the max retry on launcher activity to start stress workflows, default: 5\ncontextTimeoutInSeconds\t: RPC timeout inside activities(e.g. starting a stressWorkflow) default 3s\n\n# configuration for stress workflow\nexecutionStartToCloseTimeoutInSeconds\t: StartToCloseTimeout of stressWorkflow, default 5m\nchainSequence\t: number of steps in the stressWorkflow\nconcurrentCount\t: number of in-parallel activity(dummy activity only echo data) in a step of the stressWorkflow\npayloadSizeBytes\t: payloadSize of echo data in the dummy activity\nminCadenceSleepInSeconds\t: control sleep time between two steps in the stressWorkflow, actual sleep time = random(min,max), default: 0\nmaxCadenceSleepInSeconds\t: control sleep time between two steps in the stressWorkflow, actual sleep time = random(min,max), default: 0\npanicStressWorkflow\t: if true, stressWorkflow will always panic, default false\n``` \n\n### Cancellation\nThe load tests the StartWorkflowExecution and CancelWorkflowExecution sync API, and validates the number of cancelled workflows and if there's any open workflow.\n\nSample configuration can be found in `config/bench/cancellation.json` and it can be started with\n```\ncadence --do <domain> wf start --tl cadence-bench-tl-0 --wt cancellation-load-test-workflow --dt 30 --et 3600 --if config/bench/cancellation.json \n```\n\n### Signal\nThe load tests the SignalWorkflowExecution and SignalWithStartWorkflowExecution sync API, and validates the latency of signaling, the number of successfully completed workflows and if there's any open workflow.\n\nSample configuration can be found in `config/bench/signal.json` and it can be started with\n```\ncadence --do cadence-bench wf start --tl cadence-bench-tl-0 --wt signal-load-test-workflow --dt 30 --et 3600 --if config/bench/signal.json  \n```\n\n\n### Concurrent Execution\nThe purpose of this load is to test when a workflow schedules a large number of activities or child workflows in a single decision batch, whether server can properly throttle the processing of this workflow without affecting the execution of workflows in other domains. It will also check if the delayed period is within limit or not and fail the test if it takes too long.\n\nA typical usage will be run this load and another load for testing sync APIs (for example, basic, cancellation or signal) in two different test suites/domains (so that they are run in parallel in two domains). Apply proper task processing throttling configuration to the domain that is running the concurrent execution test and see if tests in the other domain can still pass or not. \n\nSample configuration can be found in `config/bench/concurrent_execution.json` and it can be started with\n```\ncadence --do <domain> wf start --tl cadence-bench-tl-0 --wt concurrent-execution-test-workflow --dt 30 --et 3600 --if config/bench/concurrent_execution.json\n```\n\n### Timer\nThis load tests if Cadence server can properly handle the case when one domain fires a large number of timers in a short period of time. Ideally timer from that domain should be throttled and delayed without affecting workflows in other domains. It will also check if the delayed period is within limit or not and fail the test if the timer latency is too high.\n\nTypical usage is the same as the concurrent execution load above. Run it in parallel with another sync API test and see if the other test can pass or not.\n\nSample configuration can be found in `config/bench/timer.json` and it can be started with\n```\ncadence --do <domain> wf start --tl cadence-bench-tl-0 --wt timer-load-test-workflow --dt 30 --et 3600 --if config/bench/timer.json \n```\n\n### Cron: Run all the workloads as a TestSuite\n\n:warning: NOTE: This requires a search attribute named `Passed` as boolean type. This search attribute should have been added to the [ES schema](/schema/elasticsearch). \nmake sure the dynamic config also have [this search attribute (`frontend.validSearchAttributes`)](/config/dynamicconfig/development_es.yaml), so that Cadence server can recognize it.\n* Validate `Passed` has been successfully added in the dynamic config:\n   ```\n   cadence cluster get-search-attr\n   ```\n   \n`Cron` itself is not a test. It is responsible for running all other tests in parallel or sequential according a cron schedule. \n\nTests in `Cron` are divided to into multiple test suites. Tests in different test suites will be run in parallel, while tests within a test suite will be run in a random sequential order. Different test suites can also be run in different domains, which provides a way for testing the multi-tenant performance of Cadence server. \n\nOn the completion of each test, `Cron` will be signaled with the result of the test, which can be queried through:\n```\ncadence --do <domain> wf query --wid <workflowID of the Cron workflow> --qt test-results\n```\nThis command will show the result of all completed tests.\n\nWhen all tests complete, `Cron` will update the value of the `Passed` search attribute accordingly. `Passed` will be set to `true` only when all tests have passed, and `false` otherwise. Since the last event for cron workflow is always WorkflowContinuedAsNew, this search attribute can be used to tell whether one run of `Cron` is successful or not. You can see the search attribute value by adding `--psa` flag to workflow list commands when listing `Cron` runs.\n\nA sample cron configuration is in `config/bench/cron.json`, and it can be started with\n```\ncadence --do <domain> wf start --tl cadence-bench-tl-0 --wt cron-test-workflow --dt 30 --et 7200 --if config/bench/cron.json\n```\n"
  },
  {
    "path": "bench/lib/client.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n)\n\nconst workflowRetentionDays = 1\n\n// CadenceClient is an abstraction on top of\n// the cadence library client that serves as\n// a union of all the client interfaces that\n// the library exposes\ntype CadenceClient struct {\n\tclient.Client\n\t// domainClient only exposes domain API\n\tclient.DomainClient\n\t// this is the service needed to start the workers\n\tService workflowserviceclient.Interface\n}\n\n// CreateDomain creates a cadence domain with the given name and description\n// if the domain already exist, this method silently returns success\nfunc (client CadenceClient) CreateDomain(name string, desc string, owner string) error {\n\temitMetric := true\n\tisGlobalDomain := false\n\tretention := int32(workflowRetentionDays)\n\treq := &shared.RegisterDomainRequest{\n\t\tName:                                   &name,\n\t\tDescription:                            &desc,\n\t\tOwnerEmail:                             &owner,\n\t\tWorkflowExecutionRetentionPeriodInDays: &retention,\n\t\tEmitMetric:                             &emitMetric,\n\t\tIsGlobalDomain:                         &isGlobalDomain,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\tdefer cancel()\n\terr := client.Register(ctx, req)\n\tif err != nil {\n\t\tif _, ok := err.(*shared.DomainAlreadyExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// NewCadenceClients builds a CadenceClient for each domain from the runtimeContext\nfunc NewCadenceClients(runtime *RuntimeContext) (map[string]CadenceClient, error) {\n\tcadenceClients := make(map[string]CadenceClient)\n\tfor _, domain := range runtime.Bench.Domains {\n\t\tclient, err := NewCadenceClientForDomain(runtime, domain)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tcadenceClients[domain] = client\n\t}\n\n\treturn cadenceClients, nil\n}\n\n// NewCadenceClientForDomain builds a CadenceClient for a specified domain based on runtimeContext\nfunc NewCadenceClientForDomain(\n\truntime *RuntimeContext,\n\tdomain string,\n) (CadenceClient, error) {\n\n\ttransport := grpc.NewTransport(\n\t\tgrpc.ServiceName(runtime.Bench.Name),\n\t)\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: runtime.Bench.Name,\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\truntime.Cadence.ServiceName: {Unary: transport.NewSingleOutbound(runtime.Cadence.HostNameAndPort)},\n\t\t},\n\t})\n\n\tif err := dispatcher.Start(); err != nil {\n\t\tdispatcher.Stop()\n\t\treturn CadenceClient{}, fmt.Errorf(\"failed to create outbound transport channel: %v\", err)\n\t}\n\n\tvar cadenceClient CadenceClient\n\tcadenceClient.Service = workflowserviceclient.New(dispatcher.ClientConfig(runtime.Cadence.ServiceName))\n\tcadenceClient.Client = client.NewClient(\n\t\tcadenceClient.Service,\n\t\tdomain,\n\t\t&client.Options{\n\t\t\tMetricsScope: runtime.Metrics,\n\t\t},\n\t)\n\tcadenceClient.DomainClient = client.NewDomainClient(\n\t\tcadenceClient.Service,\n\t\t&client.Options{\n\t\t\tMetricsScope: runtime.Metrics,\n\t\t},\n\t)\n\n\treturn cadenceClient, nil\n}\n"
  },
  {
    "path": "bench/lib/config.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nconst (\n\t// EnvKeyRoot the environment variable key for runtime root dir\n\tEnvKeyRoot = \"CADENCE_BENCH_ROOT\"\n\t// EnvKeyConfigDir the environment variable key for config dir\n\tEnvKeyConfigDir = \"CADENCE_BENCH_CONFIG_DIR\"\n\t// EnvKeyEnvironment is the environment variable key for environment\n\tEnvKeyEnvironment = \"CADENCE_BENCH_ENVIRONMENT\"\n\t// EnvKeyAvailabilityZone is the environment variable key for AZ\n\tEnvKeyAvailabilityZone = \"CADENCE_BENCH_AVAILABILITY_ZONE\"\n)\n\ntype (\n\t// Config contains the configuration for cadence bench\n\tConfig struct {\n\t\tBench   Bench          `yaml:\"bench\"`\n\t\tCadence Cadence        `yaml:\"cadence\"`\n\t\tLog     config.Logger  `yaml:\"log\"`\n\t\tMetrics config.Metrics `yaml:\"metrics\"`\n\t}\n\n\t// Cadence contains the configuration for cadence service\n\tCadence struct {\n\t\tServiceName     string `yaml:\"service\"`\n\t\tHostNameAndPort string `yaml:\"host\"`\n\t}\n\n\t// Bench contains the configuration for bench tests\n\tBench struct {\n\t\tName         string   `yaml:\"name\"`\n\t\tDomains      []string `yaml:\"domains\"`\n\t\tNumTaskLists int      `yaml:\"numTaskLists\"`\n\t}\n\n\t// TODO: add comment for each config field\n\n\t// CronTestConfig contains the configuration for running a set of\n\t// testsuites in parallel based on a cron schedule\n\tCronTestConfig struct {\n\t\tTestSuites []TestSuiteConfig `yaml:\"testSuites\"`\n\t}\n\n\t// TestSuiteConfig contains the configration for running a set of\n\t// tests sequentially\n\tTestSuiteConfig struct {\n\t\tName    string                 `yaml:\"name\"`\n\t\tDomain  string                 `yaml:\"domain\"`\n\t\tConfigs []AggregatedTestConfig `yaml:\"configs\"`\n\t}\n\n\t// AggregatedTestConfig contains the configration for a single test\n\tAggregatedTestConfig struct {\n\t\tName             string                    `yaml:\"name\"`\n\t\tDescription      string                    `yaml:\"description\"`\n\t\tTimeoutInSeconds int32                     `yaml:\"timeoutInSeconds\"`\n\t\tBasic            *BasicTestConfig          `yaml:\"basic\"`\n\t\tSignal           *SignalTestConfig         `yaml:\"signal\"`\n\t\tTimer            *TimerTestConfig          `yaml:\"timer\"`\n\t\tConcurrentExec   *ConcurrentExecTestConfig `yaml:\"concurrentExec\"`\n\t\tCancellation     *CancellationTestConfig   `yaml:\"cancellation\"`\n\t}\n\n\t// BasicTestConfig contains the configuration for running the Basic test scenario\n\tBasicTestConfig struct {\n\t\t// Launch workflow config\n\t\tUseBasicVisibilityValidation  bool    `yaml:\"useBasicVisibilityValidation\"`  // use basic(db based) visibility to verify the stress workflows, default false which requires advanced visibility on the server\n\t\tTotalLaunchCount              int     `yaml:\"totalLaunchCount\"`              // total number of stressWorkflows that started by the launchWorkflow\n\t\tRoutineCount                  int     `yaml:\"routineCount\"`                  // number of in-parallel launch activities that started by launchWorkflow, to start the stressWorkflows\n\t\tFailureThreshold              float64 `yaml:\"failureThreshold\"`              // the threshold of failed stressWorkflow for deciding whether or not the whole testSuite failed.\n\t\tMaxLauncherActivityRetryCount int     `yaml:\"maxLauncherActivityRetryCount\"` // the max retry on launcher activity to start stress workflows, default: 5\n\t\tContextTimeoutInSeconds       int     `yaml:\"contextTimeoutInSeconds\"`       // RPC timeout inside activities(e.g. starting a stressWorkflow) default 3s\n\t\tWaitTimeBufferInSeconds       int     `yaml:\"waitTimeBufferInSeconds\"`       // buffer time in addition of ExecutionStartToCloseTimeoutInSeconds to wait for stressWorkflows before verification, default 300(5 minutes)\n\t\t// Stress workflow config\n\t\tExecutionStartToCloseTimeoutInSeconds int  `yaml:\"executionStartToCloseTimeoutInSeconds\"` // StartToCloseTimeout of stressWorkflow, default 5m\n\t\tChainSequence                         int  `yaml:\"chainSequence\"`                         // number of steps in the stressWorkflow\n\t\tConcurrentCount                       int  `yaml:\"concurrentCount\"`                       // number of in-parallel activity(dummy activity only echo data) in a step of the stressWorkflow\n\t\tPayloadSizeBytes                      int  `yaml:\"payloadSizeBytes\"`                      // payloadSize of echo data in the dummy activity\n\t\tMinCadenceSleepInSeconds              int  `yaml:\"minCadenceSleepInSeconds\"`              // control sleep time between two steps in the stressWorkflow, actual sleep time = random(min,max), default: 0\n\t\tMaxCadenceSleepInSeconds              int  `yaml:\"maxCadenceSleepInSeconds\"`              // control sleep time between two steps in the stressWorkflow, actual sleep time = random(min,max), default: 0\n\t\tPanicStressWorkflow                   bool `yaml:\"panicStressWorkflow\"`                   // if true, stressWorkflow will always panic, default false\n\t}\n\n\t// SignalTestConfig is the parameters for signalLoadTestWorkflow\n\tSignalTestConfig struct {\n\t\t// LoaderCount defines how many loader activities\n\t\tLoaderCount int `yaml:\"loaderCount\"`\n\t\t// LoadTestWorkflowCount defines how many load test workflow in total\n\t\tLoadTestWorkflowCount int `yaml:\"loadTestWorkflowCount\"`\n\t\t// SignalCount is the number of signals per workflow\n\t\tSignalCount int `yaml:\"signalCount\"`\n\t\t// SignalDataSize is the size of signal data\n\t\tSignalDataSize int `yaml:\"signalDataSize\"`\n\t\t// RateLimit is per loader rate limit to hit cadence server\n\t\tRateLimit                         int     `yaml:\"rateLimit\"`\n\t\tWorkflowExecutionTimeoutInSeconds int     `yaml:\"workflowExecutionTimeoutInSeconds\"`\n\t\tDecisionTaskTimeoutInSeconds      int     `yaml:\"decisionTaskTimeoutInSeconds\"`\n\t\tFailureThreshold                  float64 `yaml:\"failureThreshold\"`\n\t\tProcessSignalWorkflowConfig\n\t}\n\n\t// ProcessSignalWorkflowConfig is the parameters to process signal workflow\n\tProcessSignalWorkflowConfig struct {\n\t\t// CampaignCount is the number of local activities to be executed\n\t\tCampaignCount int `yaml:\"campaignCount\"`\n\t\t// ActionRate is probability that local activities result in actual action\n\t\tActionRate float64 `yaml:\"actionRate\"`\n\t\t// Local activity failure rate\n\t\tFailureRate float64 `yaml:\"failureRate\"`\n\t\t// SignalCount before continue as new\n\t\tSignalBeforeContinueAsNew int   `yaml:\"signalBeforeContinueAsNew\"`\n\t\tEnableRollingWindow       bool  `yaml:\"enableRollingWindow\"`\n\t\tScheduleTimeNano          int64 `yaml:\"scheduleTimeNano\"`\n\t\tMaxSignalDelayInSeconds   int   `yaml:\"maxSignalDelayInSeconds\"`\n\t\tMaxSignalDelayCount       int   `yaml:\"maxSignalDelayCount\"`\n\t}\n\n\t// TimerTestConfig contains the config for running timer bench test\n\tTimerTestConfig struct {\n\t\t// TotalTimerCount is the total number of timers to fire\n\t\tTotalTimerCount int `yaml:\"totalTimerCount\"`\n\n\t\t// TimerPerWorkflow is the number of timers in each workflow\n\t\t// workflow will continue execution and complete when the first timer fires\n\t\t// Set this number larger than one to test no-op timer case\n\t\t// TotalTimerCount / TimerPerWorkflow = total number of workflows\n\t\tTimerPerWorkflow int `yaml:\"timerPerWorkflow\"`\n\n\t\t// ShortestTimerDurationInSeconds after test start, the first timer will fire\n\t\tShortestTimerDurationInSeconds int `yaml:\"shortestTimerDurationInSeconds\"`\n\n\t\t// LongestTimerDurationInSeconds after test start, the last timer will fire\n\t\tLongestTimerDurationInSeconds int `yaml:\"longestTimerDurationInSeconds\"`\n\n\t\t// MaxTimerLatencyInSeconds specifies the maximum latency for the first timer in the workflow\n\t\t// if a timer's latency is larger than this value, that timer will be considered as failed\n\t\t// if > 1% timer fire beyond this threshold, the test will fail\n\t\tMaxTimerLatencyInSeconds int `yaml:\"maxTimerLatencyInSeconds\"`\n\n\t\t// TimerTimeoutInSeconds specifies the duration beyond which a timer is considered as lost and fail the test\n\t\tTimerTimeoutInSeconds int `yaml:\"timerTimeoutInSeconds\"`\n\n\t\t// RoutineCount is the number of goroutines used for starting workflows\n\t\t// approx. RPS = 10 * RoutineCount\n\t\t// # of workflows = TotalTimerCount / TimerPerWorkflow\n\t\t// please make sure ShortestTimerDurationInSeconds > # of workflows / RPS, so that timer starts firing\n\t\t// after all workflows has been started.\n\t\t// please also make sure test timeout > LongestTimerDurationInSeconds + TimerTimeoutInSeconds\n\t\tRoutineCount int `yaml:\"routineCount\"`\n\t}\n\n\t// ConcurrentExecTestConfig contains the config for running concurrent execution test\n\tConcurrentExecTestConfig struct {\n\t\t// TotalBatches is the total number of batches\n\t\tTotalBatches int `yaml:\"totalBatches\"`\n\n\t\t// Concurrency specifies the number of batches that will be run concurrently\n\t\tConcurrency int `yaml:\"concurrency\"`\n\n\t\t// BatchType specifies the type of batch, can be either \"activity\" or \"childWorkflow\", case insensitive\n\t\tBatchType string `yaml:\"batchType\"`\n\n\t\t// BatchSize specifies the number of activities or childWorkflows scheduled in a single decision batch\n\t\tBatchSize int `yaml:\"batchSize\"`\n\n\t\t// BatchPeriodInSeconds specifies the time interval between two set of batches (each set has #concurrency batches)\n\t\tBatchPeriodInSeconds int `yaml:\"batchPeriodInSeconds\"`\n\n\t\t// BatchMaxLatencyInSeconds specifies the max latency for scheduling/starting the activity or childWorkflow.\n\t\t// if latency is higher than this number, the corresponding activity or childWorkflow will be\n\t\t// considered as failed. This bench test is considered as success if:\n\t\t// avg(succeed activity or childWorkflow / batchSize) >= 0.99\n\t\t// If any of the activity or childWorkflow returns an error (for example, execution takes longer than BatchTimeoutInSeconds),\n\t\t// the bench test will fail immediately\n\t\tBatchMaxLatencyInSeconds int `yaml:\"batchMaxLatencyInSeconds\"`\n\n\t\t// BatchTimeoutInSeconds specifies the timeout for each batch execution\n\t\tBatchTimeoutInSeconds int `yaml:\"batchTimeoutInSeconds\"`\n\t}\n\n\t// CancellationTestConfig contains the config for running workflow cancellation test\n\tCancellationTestConfig struct {\n\t\t// TotalLaunchCount is the total number of workflow to start\n\t\t// note: make sure TotalLaunchCount mod Concurrency = 0 otherwise the validation for the test will fail\n\t\tTotalLaunchCount int `yaml:\"totalLaunchCount\"`\n\n\t\t// Concurrency specifies the concurrency for start and cancel workflow execution\n\t\t// approx. RPS = 10 * concurrency, approx. duration = totalLaunchCount / RPS\n\t\tConcurrency int `yaml:\"concurrency\"`\n\n\t\t// ContextTimeoutInSeconds specifies the context timeout for start and cancel workflow execution call\n\t\t// default: 3s\n\t\tContextTimeoutInSeconds int `yaml:\"contextTimeoutInSeconds\"`\n\t}\n)\n\nfunc (c *Config) Validate() error {\n\tif len(c.Bench.Name) == 0 {\n\t\treturn errors.New(\"missing value for bench service name\")\n\t}\n\tif len(c.Bench.Domains) == 0 {\n\t\treturn errors.New(\"missing value for domains property\")\n\t}\n\tif c.Bench.NumTaskLists == 0 {\n\t\treturn errors.New(\"number of taskLists can not be 0\")\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "bench/lib/config_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype ConfigTestSuite struct {\n\tsuite.Suite\n}\n\nfunc TestConfigTestSuite(t *testing.T) {\n\tsuite.Run(t, new(ConfigTestSuite))\n}\n\nfunc (s *ConfigTestSuite) TestValidate() {\n\ttestCases := []func(*Config){\n\t\tfunc(c *Config) { c.Bench.Name = \"\" },\n\t\tfunc(c *Config) { c.Bench.Domains = []string{} },\n\t\tfunc(c *Config) { c.Bench.NumTaskLists = 0 },\n\t}\n\n\tfor _, tc := range testCases {\n\t\tconfig := s.buildConfig()\n\t\ttc(&config)\n\t\ts.Error(config.Validate())\n\t}\n}\n\nfunc (s *ConfigTestSuite) buildConfig() Config {\n\treturn Config{\n\t\tBench: Bench{\n\t\t\tName:         \"cadence-bench\",\n\t\t\tDomains:      []string{\"cadence-bench\"},\n\t\t\tNumTaskLists: 1,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "bench/lib/context.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\nimport (\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\nconst (\n\tdefaultCadenceLocalHostPort = \"127.0.0.1:7933\"\n\tdefaultCadenceServiceName   = \"cadence-frontend\"\n)\n\n// ContextKey is an alias for string, used as context key\ntype ContextKey string\n\nconst (\n\t// CtxKeyRuntimeContext is the name of the context key whose value is the RuntimeContext\n\tCtxKeyRuntimeContext = ContextKey(\"ctxKeyRuntimeCtx\")\n\n\t// CtxKeyCadenceClient is the name of the context key for the cadence client this cadence worker listens to\n\tCtxKeyCadenceClient = ContextKey(\"ctxKeyCadenceClient\")\n)\n\n// RuntimeContext contains all of the context information\n// needed at cadence bench runtime\ntype RuntimeContext struct {\n\tBench   Bench\n\tCadence Cadence\n\tLogger  *zap.Logger\n\tMetrics tally.Scope\n}\n\n// NewRuntimeContext builds a runtime context from the config\nfunc NewRuntimeContext(cfg *Config) (*RuntimeContext, error) {\n\tlogger, err := cfg.Log.NewZapLogger()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmetricsScope := cfg.Metrics.NewScope(log.NewLogger(logger), cfg.Bench.Name)\n\n\tif cfg.Cadence.ServiceName == \"\" {\n\t\tcfg.Cadence.ServiceName = defaultCadenceServiceName\n\t}\n\n\tif cfg.Cadence.HostNameAndPort == \"\" {\n\t\tcfg.Cadence.HostNameAndPort = defaultCadenceLocalHostPort\n\t}\n\n\treturn &RuntimeContext{\n\t\tBench:   cfg.Bench,\n\t\tCadence: cfg.Cadence,\n\t\tLogger:  logger,\n\t\tMetrics: metricsScope,\n\t}, nil\n}\n"
  },
  {
    "path": "bench/lib/metrics.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\n// counters go here\nconst (\n\tstartedCount           = \"started\"\n\tFailedCount            = \"failed\"\n\tsuccessCount           = \"succeeded\"\n\terrTimeoutCount        = \"errors.timeout\"\n\terrIncompatibleVersion = \"errors.incompatibleversion\"\n)\n\n// latency metrics go here\nconst (\n\tlatency      = \"latency\"\n\tstartLatency = \"latency.schedule-to-start\"\n)\n\nconst workflowVersion = 1\nconst workflowChangeID = \"initial\"\n\n// WorkflowMetricsProfile is the state that's needed to\n// record success/failed and latency metrics at the end\n// of a workflow\ntype WorkflowMetricsProfile struct {\n\tctx            workflow.Context\n\tstartTimestamp int64\n\tScope          tally.Scope\n}\n\n// BeginWorkflow executes the common steps involved in all the workflow functions\n// It checks for workflow task version compatibility and also records the execution\n// in m3. This function must be the first call in every workflow function\n// Returns metrics scope on success, error on failure\nfunc BeginWorkflow(ctx workflow.Context, wfType string, scheduledTimeNanos int64) (*WorkflowMetricsProfile, error) {\n\tprofile := recordWorkflowStart(ctx, wfType, scheduledTimeNanos)\n\tif err := checkWFVersionCompatibility(ctx); err != nil {\n\t\tprofile.Scope.Counter(errIncompatibleVersion).Inc(1)\n\t\treturn nil, err\n\t}\n\treturn profile, nil\n}\n\n// End records the elapsed time and reports the latency,\n// success, failed counts to m3\nfunc (profile *WorkflowMetricsProfile) End(err error) error {\n\tnow := workflow.Now(profile.ctx).UnixNano()\n\telapsed := time.Duration(now - profile.startTimestamp)\n\treturn recordWorkflowEnd(profile.Scope, elapsed, err)\n}\n\n// RecordActivityStart emits metrics at the beginning of an activity function\nfunc RecordActivityStart(\n\tscope tally.Scope,\n\tname string,\n\tscheduledTimeNanos int64,\n) (tally.Scope, tally.Stopwatch) {\n\tscope = scope.Tagged(map[string]string{\"operation\": name})\n\telapsed := max(0, time.Now().UnixNano()-scheduledTimeNanos)\n\tscope.Timer(startLatency).Record(time.Duration(elapsed))\n\tscope.Counter(startedCount).Inc(1)\n\tsw := scope.Timer(latency).Start()\n\treturn scope, sw\n}\n\n// RecordActivityEnd emits metrics at the end of an activity function\nfunc RecordActivityEnd(\n\tscope tally.Scope,\n\tsw tally.Stopwatch,\n\terr error,\n) {\n\tsw.Stop()\n\tif err != nil {\n\t\tscope.Counter(FailedCount).Inc(1)\n\t\treturn\n\t}\n\tscope.Counter(successCount).Inc(1)\n}\n\n// workflowMetricScope creates and returns a child metric scope with tags\n// that identify the current workflow type\nfunc workflowMetricScope(\n\tctx workflow.Context,\n\twfType string,\n) tally.Scope {\n\tparent := workflow.GetMetricsScope(ctx)\n\treturn parent.Tagged(map[string]string{\"operation\": wfType})\n}\n\n// recordWorkflowStart emits metrics at the beginning of a workflow function\nfunc recordWorkflowStart(\n\tctx workflow.Context,\n\twfType string,\n\tscheduledTimeNanos int64,\n) *WorkflowMetricsProfile {\n\tnow := workflow.Now(ctx).UnixNano()\n\tscope := workflowMetricScope(ctx, wfType)\n\telapsed := max(0, now-scheduledTimeNanos)\n\tscope.Timer(startLatency).Record(time.Duration(elapsed))\n\tscope.Counter(startedCount).Inc(1)\n\treturn &WorkflowMetricsProfile{\n\t\tctx:            ctx,\n\t\tstartTimestamp: now,\n\t\tScope:          scope,\n\t}\n}\n\n// recordWorkflowEnd emits metrics at the end of a workflow function\nfunc recordWorkflowEnd(\n\tscope tally.Scope,\n\telapsed time.Duration,\n\terr error,\n) error {\n\tscope.Timer(latency).Record(elapsed)\n\tif err == nil {\n\t\tscope.Counter(successCount).Inc(1)\n\t\treturn err\n\t}\n\tscope.Counter(FailedCount).Inc(1)\n\tif _, ok := err.(*workflow.TimeoutError); ok {\n\t\tscope.Counter(errTimeoutCount).Inc(1)\n\t}\n\treturn err\n}\n\n// checkWFVersionCompatibility takes a workflow.Context param and\n// validates that the workflow task currently being handled\n// is compatible with this version of the bench - this method\n// MUST only be called within a workflow function and it MUST\n// be the first line in the workflow function\n// Returns an error if the version is incompatible\nfunc checkWFVersionCompatibility(ctx workflow.Context) error {\n\tversion := workflow.GetVersion(ctx, workflowChangeID, workflowVersion, workflowVersion)\n\tif version != workflowVersion {\n\t\tworkflow.GetLogger(ctx).Error(\"workflow version mismatch\",\n\t\t\tzap.Int(\"want\", int(workflowVersion)), zap.Int(\"got\", int(version)))\n\t\treturn fmt.Errorf(\"workflow version mismatch, want=%v, got=%v\", workflowVersion, version)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "bench/lib/types.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lib\n\ntype (\n\t// Runnable is an interface for anything that exposes a Run method\n\tRunnable interface {\n\t\tRun() error\n\t}\n)\n"
  },
  {
    "path": "bench/load/basic/launchWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage basic\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n\tc \"github.com/uber/cadence/common\"\n)\n\nconst (\n\t// TestName is the test name for basic test\n\tTestName = \"basic\"\n\n\t// LauncherWorkflowName is the workflow name for launching basic load test\n\tLauncherWorkflowName = \"basic-load-test-workflow\"\n\t// LauncherLaunchActivityName is the activity name for launching stress load test\n\tLauncherLaunchActivityName = \"basic-load-test-launch-activity\"\n\t// LauncherVerifyActivityName is the verification activity name\n\tLauncherVerifyActivityName = \"basic-load-test-verify-activity\"\n\n\tdefaultStressWorkflowStartToCloseTimeout = 5 * time.Minute // default 5m may not be not enough for long running workflow\n\tdefaultStressWorkflowWaitTimeBuffer      = 5 * time.Minute // time buffer of waiting for stressWorkflow execution. Hence the actual waiting time is  stressWorkflowStartToCloseTimeout + defaultStressWorkflowWaitTimeBuffer\n\tdefaultMaxLauncherActivityRetryCount     = 5               // number of retry for launcher activity\n)\n\ntype (\n\tlauncherActivityParams struct {\n\t\tRoutineID int                 // the ID of the launchActivity\n\t\tCount     int                 // stressWorkflows to start per launchActivity\n\t\tConfig    lib.BasicTestConfig // config of this load test\n\t}\n\n\tverifyActivityParams struct {\n\t\tWorkflowStartTime            int64\n\t\tListWorkflowPageSize         int32\n\t\tUseBasicVisibilityValidation bool\n\t}\n\n\tverifyActivityResult struct {\n\t\tOpenCount    int\n\t\tTimeoutCount int\n\t\tFailedCount  int\n\t}\n)\n\n// RegisterLauncher registers workflows and activities for basic load launching\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(launcherWorkflow, workflow.RegisterOptions{Name: LauncherWorkflowName})\n\tw.RegisterActivityWithOptions(launcherActivity, activity.RegisterOptions{Name: LauncherLaunchActivityName})\n\tw.RegisterActivityWithOptions(verifyResultActivity, activity.RegisterOptions{Name: LauncherVerifyActivityName})\n}\n\nfunc launcherWorkflow(ctx workflow.Context, config lib.BasicTestConfig) (string, error) {\n\tlogger := workflow.GetLogger(ctx).With(zap.String(\"Test\", TestName))\n\tif config.MaxLauncherActivityRetryCount == 0 {\n\t\tconfig.MaxLauncherActivityRetryCount = defaultMaxLauncherActivityRetryCount\n\t}\n\tif config.ContextTimeoutInSeconds == 0 {\n\t\tconfig.ContextTimeoutInSeconds = int(common.DefaultContextTimeout / time.Second)\n\t}\n\tif config.ExecutionStartToCloseTimeoutInSeconds == 0 {\n\t\tconfig.ExecutionStartToCloseTimeoutInSeconds = int(defaultStressWorkflowStartToCloseTimeout / time.Second)\n\t}\n\tif config.WaitTimeBufferInSeconds == 0 {\n\t\tconfig.WaitTimeBufferInSeconds = int(defaultStressWorkflowWaitTimeBuffer / time.Second)\n\t}\n\n\tworkflowPerActivity := config.TotalLaunchCount / config.RoutineCount\n\tworkflowTimeout := workflow.GetInfo(ctx).ExecutionStartToCloseTimeoutSeconds\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Duration(workflowTimeout) * time.Second,\n\t\tHeartbeatTimeout:       20 * time.Second,\n\t\tRetryPolicy: &workflow.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tBackoffCoefficient: 1,\n\t\t\tMaximumInterval:    time.Second,\n\t\t\tExpirationInterval: time.Second * time.Duration(workflowTimeout+1) * time.Second * time.Duration(config.MaxLauncherActivityRetryCount),\n\t\t\tMaximumAttempts:    int32(config.MaxLauncherActivityRetryCount),\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\tstartTime := workflow.Now(ctx)\n\n\tfutures := make([]workflow.Future, 0, config.RoutineCount)\n\tfor i := 0; i < config.RoutineCount; i++ {\n\t\tactInput := launcherActivityParams{\n\t\t\tRoutineID: i,\n\t\t\tCount:     workflowPerActivity,\n\t\t\tConfig:    config,\n\t\t}\n\t\tf := workflow.ExecuteActivity(\n\t\t\tctx,\n\t\t\tLauncherLaunchActivityName,\n\t\t\tactInput,\n\t\t)\n\t\tfutures = append(futures, f)\n\t}\n\n\tfor _, f := range futures {\n\t\terr := f.Get(ctx, nil)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t}\n\n\tworkflowWaitTime := time.Duration(config.ExecutionStartToCloseTimeoutInSeconds+config.WaitTimeBufferInSeconds) * time.Second\n\tlogger.Info(fmt.Sprintf(\"%v stressWorkflows are launched, now waiting for %v ...\", config.TotalLaunchCount, workflowWaitTime))\n\tif err := workflow.Sleep(ctx, workflowWaitTime); err != nil {\n\t\treturn \"\", fmt.Errorf(\"launcher workflow sleep failed: %v\", err)\n\t}\n\n\tmaxTolerantFailure := int32(float64(config.TotalLaunchCount) * config.FailureThreshold)\n\tactInput := verifyActivityParams{\n\t\tWorkflowStartTime:            startTime.UnixNano(),\n\t\tListWorkflowPageSize:         maxTolerantFailure + 10,\n\t\tUseBasicVisibilityValidation: config.UseBasicVisibilityValidation,\n\t}\n\tactOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tBackoffCoefficient: 2,\n\t\t\tMaximumAttempts:    5,\n\t\t},\n\t}\n\tvar result verifyActivityResult\n\tctx = workflow.WithActivityOptions(ctx, actOptions)\n\tif err := workflow.ExecuteActivity(\n\t\tctx,\n\t\tLauncherVerifyActivityName,\n\t\tactInput,\n\t).Get(ctx, &result); err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// Passing criteria:\n\t//    1. No open workflows(this means server may lose some tasks and not able to close the stressWorkflows)\n\t//    2. Failed workflows <= threshold\n\tpassed := (result.TimeoutCount+result.FailedCount) <= int(maxTolerantFailure) && result.OpenCount == 0\n\n\tdetailResult := fmt.Sprintf(\"Details report: timeoutCount: %v, failedCount: %v, openCount:%v, launchCount: %v, maxThreshold:%v\",\n\t\tresult.TimeoutCount, result.FailedCount, result.OpenCount, config.TotalLaunchCount, maxTolerantFailure)\n\tif passed {\n\t\treturn fmt.Sprintf(\"TEST PASSED. %v\", detailResult), nil\n\t}\n\treturn \"\", fmt.Errorf(\"TEST FAILED. %v\", detailResult)\n}\n\nfunc launcherActivity(ctx context.Context, params launcherActivityParams) error {\n\tinfo := activity.GetInfo(ctx)\n\tlogger := activity.GetLogger(ctx).With(zap.String(\"Test\", TestName))\n\n\tvar lastStartedID int\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\terr := activity.GetHeartbeatDetails(ctx, &lastStartedID)\n\t\tif err != nil {\n\t\t\tlogger.Error(\"failed to resume from last  checkpoint...start from beginning...\")\n\t\t}\n\t\tlogger.Info(\"resume from last checkpoint\", zap.Int(\"checkpoint\", lastStartedID))\n\t}\n\n\tlogger.Info(\"Start activity routineID\", zap.Int(\"RoutineID\", params.RoutineID))\n\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\trc := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tnumTaskList := rc.Bench.NumTaskLists\n\tbasicTestConfig := params.Config\n\n\tstressWorkflowTimeout := time.Duration(basicTestConfig.ExecutionStartToCloseTimeoutInSeconds) * time.Second\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tExecutionStartToCloseTimeout: stressWorkflowTimeout,\n\t}\n\n\tvar sleepTime time.Duration\n\tif basicTestConfig.MinCadenceSleepInSeconds > 0 && basicTestConfig.MaxCadenceSleepInSeconds > 0 {\n\t\tminSleep := time.Duration(basicTestConfig.MinCadenceSleepInSeconds) * time.Second\n\t\tmaxSleep := time.Duration(basicTestConfig.MaxCadenceSleepInSeconds) * time.Second\n\t\tjitter := rand.Float64() * float64(maxSleep-minSleep)\n\t\tsleepTime = minSleep + time.Duration(int64(jitter))\n\t}\n\tstressWorkflowInput := WorkflowParams{\n\t\tChainSequence:    basicTestConfig.ChainSequence,\n\t\tConcurrentCount:  basicTestConfig.ConcurrentCount,\n\t\tPayloadSizeBytes: basicTestConfig.PayloadSizeBytes,\n\t\tPanicWorkflow:    basicTestConfig.PanicStressWorkflow,\n\t\tCadenceSleep:     sleepTime,\n\t}\n\n\tfor startedID := lastStartedID; startedID < params.Count; startedID++ {\n\t\tstressWorkflowInput.TaskListNumber = rand.Intn(numTaskList)\n\n\t\tworkflowOptions.ID = fmt.Sprintf(\"%v-%d-%d\", info.WorkflowExecution.ID, params.RoutineID, startedID)\n\t\tworkflowOptions.TaskList = common.GetTaskListName(stressWorkflowInput.TaskListNumber)\n\n\t\tstartWorkflowContext, cancelF := context.WithTimeout(context.Background(), time.Duration(basicTestConfig.ContextTimeoutInSeconds)*time.Second)\n\t\twe, err := cc.StartWorkflow(startWorkflowContext, workflowOptions, stressWorkflowName, stressWorkflowInput)\n\t\tcancelF()\n\t\tif err == nil {\n\t\t\tlogger.Debug(\"Created Workflow successfully\", zap.String(\"WorkflowID\", we.ID), zap.String(\"RunID\", we.RunID))\n\t\t} else {\n\t\t\tif cadence.IsWorkflowExecutionAlreadyStartedError(err) {\n\t\t\t\tlogger.Debug(\"Workflow already started in previous activity attempt\")\n\t\t\t} else {\n\t\t\t\tlogger.Error(\"Failed to start workflow execution\", zap.Error(err))\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tactivity.RecordHeartbeat(ctx, startedID)\n\t\tjitter := time.Duration(75 + rand.Intn(25))\n\t\ttime.Sleep(jitter * time.Millisecond)\n\t}\n\n\tlogger.Info(\"finish running launcher activity\", zap.Int(\"StartedCount\", params.Count))\n\treturn nil\n}\n\nfunc verifyResultActivity(\n\tctx context.Context,\n\tparams verifyActivityParams,\n) (verifyActivityResult, error) {\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\tinfo := activity.GetInfo(ctx)\n\n\tvar opens, timeouts, faileds int\n\tvar err error\n\n\tif params.UseBasicVisibilityValidation {\n\t\topens, timeouts, faileds, err = verifyByBasicVisibility(ctx, params, cc, info)\n\t} else {\n\t\topens, timeouts, faileds, err = verifyByAdvancedVisibility(ctx, params, cc, info)\n\t}\n\n\treturn verifyActivityResult{\n\t\tOpenCount:    opens,\n\t\tFailedCount:  faileds,\n\t\tTimeoutCount: timeouts,\n\t}, err\n}\n\nfunc verifyByAdvancedVisibility(ctx context.Context, params verifyActivityParams, cc lib.CadenceClient, info activity.Info) (opens, timeouts, faileds int, err error) {\n\topenWorkflowQuery := \"WorkflowType='%v' and StartTime > %v and CloseTime = missing\"\n\ttimeoutWorkflowQuery := \"WorkflowType='%v' and CloseStatus=5 and StartTime > %v and CloseTime < %v\"\n\tfailedWorkflowQuery := \"WorkflowType='%v' and CloseStatus=1 and StartTime > %v and CloseTime < %v\"\n\n\tquery := fmt.Sprintf(\n\t\topenWorkflowQuery,\n\t\tstressWorkflowName,\n\t\tparams.WorkflowStartTime)\n\trequest := &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: c.StringPtr(info.WorkflowDomain),\n\t\tQuery:  &query,\n\t}\n\tresp, err := cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn\n\t}\n\topens = int(resp.GetCount())\n\n\tquery = fmt.Sprintf(\n\t\ttimeoutWorkflowQuery,\n\t\tstressWorkflowName,\n\t\tparams.WorkflowStartTime,\n\t\ttime.Now().UnixNano())\n\trequest = &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: c.StringPtr(info.WorkflowDomain),\n\t\tQuery:  &query,\n\t}\n\tresp, err = cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn\n\t}\n\ttimeouts = int(resp.GetCount())\n\n\tquery = fmt.Sprintf(\n\t\tfailedWorkflowQuery,\n\t\tstressWorkflowName,\n\t\tparams.WorkflowStartTime,\n\t\ttime.Now().UnixNano())\n\trequest = &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: c.StringPtr(info.WorkflowDomain),\n\t\tQuery:  &query,\n\t}\n\tresp, err = cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn\n\t}\n\tfaileds = int(resp.GetCount())\n\treturn\n}\n\nfunc verifyByBasicVisibility(ctx context.Context, params verifyActivityParams, cc lib.CadenceClient, info activity.Info) (opens, timeouts, faileds int, err error) {\n\t// verify if any open workflow\n\tlistOpenWorkflowRequest := &shared.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          c.StringPtr(info.WorkflowDomain),\n\t\tMaximumPageSize: &params.ListWorkflowPageSize,\n\t\tStartTimeFilter: &shared.StartTimeFilter{\n\t\t\tEarliestTime: c.Int64Ptr(params.WorkflowStartTime),\n\t\t\tLatestTime:   c.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tTypeFilter: &shared.WorkflowTypeFilter{\n\t\t\tName: c.StringPtr(stressWorkflowName),\n\t\t},\n\t}\n\topenWfs, err := cc.ListOpenWorkflow(ctx, listOpenWorkflowRequest)\n\tif err != nil {\n\t\treturn\n\t}\n\n\tif len(openWfs.Executions) > 0 {\n\t\topens = len(openWfs.Executions)\n\t}\n\n\t// verify if any failed workflow\n\tcloseStatus := shared.WorkflowExecutionCloseStatusFailed\n\tlistWorkflowRequest := &shared.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          c.StringPtr(info.WorkflowDomain),\n\t\tMaximumPageSize: &params.ListWorkflowPageSize,\n\t\tStartTimeFilter: &shared.StartTimeFilter{\n\t\t\tEarliestTime: c.Int64Ptr(params.WorkflowStartTime),\n\t\t\tLatestTime:   c.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tStatusFilter: &closeStatus,\n\t}\n\twfs, err := cc.ListClosedWorkflow(ctx, listWorkflowRequest)\n\tif err != nil {\n\t\treturn\n\t}\n\n\tif len(wfs.Executions) > 0 {\n\t\tfaileds = len(wfs.Executions)\n\t}\n\n\t// verify if any timeouted workflow\n\tcloseStatus = shared.WorkflowExecutionCloseStatusTimedOut\n\tlistWorkflowRequest.StatusFilter = &closeStatus\n\twfs, err = cc.ListClosedWorkflow(ctx, listWorkflowRequest)\n\tif err != nil {\n\t\treturn\n\t}\n\n\tif len(wfs.Executions) > 0 {\n\t\ttimeouts = len(wfs.Executions)\n\t}\n\treturn\n}\n"
  },
  {
    "path": "bench/load/basic/stressWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage basic\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\tstressWorkflowName = \"basicStressWorkflow\"\n)\n\ntype (\n\t// WorkflowParams inputs to workflow.\n\tWorkflowParams struct {\n\t\tChainSequence    int\n\t\tConcurrentCount  int\n\t\tTaskListNumber   int\n\t\tPayloadSizeBytes int\n\t\tCadenceSleep     time.Duration\n\t\tPanicWorkflow    bool\n\t}\n)\n\n// RegisterWorker registers workflows and activities for basic load\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(stressWorkflowExecute, workflow.RegisterOptions{Name: stressWorkflowName})\n}\n\nfunc stressWorkflowExecute(ctx workflow.Context, workflowInput WorkflowParams) error {\n\n\tif workflowInput.PanicWorkflow {\n\t\tpanic(\"panic workflow load test.\")\n\t}\n\n\tactivityParams := common.EchoActivityParams{\n\t\tPayload: make([]byte, workflowInput.PayloadSizeBytes),\n\t}\n\n\tao := workflow.ActivityOptions{\n\t\tTaskList:               common.GetTaskListName(workflowInput.TaskListNumber),\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tHeartbeatTimeout:       20 * time.Second,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tfor i := 0; i < workflowInput.ChainSequence; i++ {\n\t\tselector := workflow.NewSelector(ctx)\n\t\tvar activityErr error\n\t\tfor j := 0; j < workflowInput.ConcurrentCount; j++ {\n\t\t\tselector.AddFuture(workflow.ExecuteActivity(ctx, common.EchoActivityName, activityParams), func(f workflow.Future) {\n\t\t\t\terr := f.Get(ctx, nil)\n\t\t\t\tif err != nil {\n\t\t\t\t\tworkflow.GetLogger(ctx).Error(\"basic test stress workflow echo activity execution failed\", zap.Error(err))\n\t\t\t\t\tactivityErr = err\n\t\t\t\t}\n\t\t\t})\n\t\t}\n\n\t\tfor i := 0; i < workflowInput.ConcurrentCount; i++ {\n\t\t\tselector.Select(ctx) // this will wait for one branch\n\t\t\tif activityErr != nil {\n\t\t\t\treturn fmt.Errorf(\"echo activity execution failed: %v\", activityErr)\n\t\t\t}\n\t\t}\n\n\t\tif workflowInput.CadenceSleep > 0 {\n\t\t\tif err := workflow.Sleep(ctx, workflowInput.CadenceSleep); err != nil {\n\t\t\t\tworkflow.GetLogger(ctx).Error(\"cadence.Sleep() returned error\", zap.Error(err))\n\t\t\t\treturn fmt.Errorf(\"stress workflow sleep failed: %v\", err)\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "bench/load/cancellation/workflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cancellation\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\t// TestName is the test name for cancellation test\n\tTestName = \"cancellation\"\n\n\t// LauncherWorkflowName is the workflow name for launching cancellation load test\n\tLauncherWorkflowName = \"cancellation-load-test-workflow\"\n)\n\nconst (\n\tsleepWorkflowName = \"cancellation-sleep-workflow\"\n\n\twaitDurationInMilliSeconds       = 75\n\twaitDurationJitterInMilliSeconds = 50\n\n\tminActivityTimeout = time.Minute\n\n\tdefaultDurationBeforeCancellation = 10 * time.Second\n\tdefaultWorkflowSleepDuration      = 20 * time.Second\n\tdefaultDurationBeforeValidation   = 30 * time.Second\n)\n\ntype (\n\tlauncherActivityResult struct {\n\t\tStartAvailability  float64\n\t\tCancelAvailability float64\n\t}\n\n\tstartWorkflowProgress struct {\n\t\tTotalStartWorkflowCall     int\n\t\tSucceededStartWorkflowCall int\n\t\tWorkflowStarted            int\n\t\tNextStartID                int\n\t}\n)\n\n// RegisterLauncher registers workflows for launching cancellation load\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(launcherWorkflow, workflow.RegisterOptions{Name: LauncherWorkflowName})\n\tw.RegisterActivity(launcherActivity)\n\tw.RegisterActivity(validationActivity)\n}\n\n// RegisterWorker registers workflows for cancellation test\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(sleepWorkflow, workflow.RegisterOptions{Name: sleepWorkflowName})\n}\n\nfunc launcherWorkflow(\n\tctx workflow.Context,\n\tconfig lib.CancellationTestConfig,\n) error {\n\tworkflowPerActivity := config.TotalLaunchCount / config.Concurrency\n\n\tactivityStartToCloseTimeout := time.Duration(workflowPerActivity/(1000/waitDurationInMilliSeconds)) * time.Second * common.DefaultMaxRetryCount\n\tif activityStartToCloseTimeout < minActivityTimeout {\n\t\tactivityStartToCloseTimeout = minActivityTimeout\n\t}\n\n\tlauncherActivityOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    activityStartToCloseTimeout,\n\t\tHeartbeatTimeout:       30 * time.Second,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tBackoffCoefficient: 2,\n\t\t\tMaximumAttempts:    10,\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, launcherActivityOptions)\n\n\tstartTime := workflow.Now(ctx)\n\n\tfutures := make([]workflow.Future, 0, config.Concurrency)\n\tfor i := 0; i < config.Concurrency; i++ {\n\t\tfuture := workflow.ExecuteActivity(ctx, launcherActivity, config)\n\t\tfutures = append(futures, future)\n\t}\n\n\tavgStartAvailability := 0.0\n\tavgCancelAvailability := 0.0\n\tfor _, future := range futures {\n\t\tvar result launcherActivityResult\n\t\tif err := future.Get(ctx, &result); err != nil {\n\t\t\treturn fmt.Errorf(\"launcherActivity failed: %v\", err)\n\t\t}\n\t\tavgStartAvailability += result.StartAvailability\n\t\tavgCancelAvailability += result.CancelAvailability\n\t}\n\tavgStartAvailability /= float64(config.Concurrency)\n\tavgCancelAvailability /= float64(config.Concurrency)\n\n\t// validate availability\n\tif avgStartAvailability < common.DefaultAvailabilityThreshold {\n\t\treturn fmt.Errorf(\"startWorkflow availability too low, required: %v, actual: %v\", common.DefaultAvailabilityThreshold, avgStartAvailability)\n\t}\n\tif avgCancelAvailability < common.DefaultAvailabilityThreshold {\n\t\treturn fmt.Errorf(\"cancelWorkflow availability too low, required: %v, actual: %v\", common.DefaultAvailabilityThreshold, avgCancelAvailability)\n\t}\n\n\t// validate if there's stuck workflow using visibility records\n\n\t// give the system some time to propagate ES records and wait for workflow that failed to cancel to finish\n\tif err := workflow.Sleep(ctx, defaultDurationBeforeValidation); err != nil {\n\t\treturn fmt.Errorf(\"launcher workflow sleep failed: %v\", err)\n\t}\n\n\tvalidationActivityOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2,\n\t\t\tMaximumAttempts:          5,\n\t\t\tNonRetriableErrorReasons: []string{common.ErrReasonValidationFailed},\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, validationActivityOptions)\n\t// move startTime backward by 10 secs to account for the time drift between worker and cadence hosts if any\n\treturn workflow.ExecuteActivity(ctx, validationActivity, config, startTime.Add(-time.Second*10).UnixNano()).Get(ctx, nil)\n}\n\nfunc launcherActivity(\n\tctx context.Context,\n\tconfig lib.CancellationTestConfig,\n) (launcherActivityResult, error) {\n\tif config.ContextTimeoutInSeconds == 0 {\n\t\tconfig.ContextTimeoutInSeconds = int(common.DefaultContextTimeout.Seconds())\n\t}\n\n\tstartRPS := 1000 / waitDurationInMilliSeconds\n\tworkflowIDChan := make(chan string, startRPS*int(defaultDurationBeforeCancellation.Seconds()))\n\n\tvar startAvailability float64\n\tgo func() {\n\t\tstartAvailability = startWorkflow(ctx, &config, workflowIDChan)\n\t}()\n\n\ttime.Sleep(defaultDurationBeforeCancellation)\n\n\tcancelAvailability := cancelWorkflow(ctx, &config, workflowIDChan)\n\n\treturn launcherActivityResult{\n\t\tStartAvailability:  startAvailability,\n\t\tCancelAvailability: cancelAvailability,\n\t}, nil\n}\n\nfunc startWorkflow(\n\tctx context.Context,\n\tconfig *lib.CancellationTestConfig,\n\tworkflowIDChan chan string,\n) float64 {\n\tdefer close(workflowIDChan)\n\n\tlogger := activity.GetLogger(ctx)\n\n\tvar progress startWorkflowProgress\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\tif err := activity.GetHeartbeatDetails(ctx, &progress); err != nil {\n\t\t\t// explicitly reset value to 0 if failed to get details, in case the implementation changed the value\n\t\t\tlogger.Error(\"Failed to get activity heartbeat details\", zap.Error(err))\n\t\t\tprogress = startWorkflowProgress{\n\t\t\t\tTotalStartWorkflowCall:     0,\n\t\t\t\tSucceededStartWorkflowCall: 0,\n\t\t\t\tWorkflowStarted:            0,\n\t\t\t\tNextStartID:                0,\n\t\t\t}\n\t\t}\n\t}\n\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\trc := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tnumTaskList := rc.Bench.NumTaskLists\n\n\tfor i := progress.NextStartID; i < config.TotalLaunchCount/config.Concurrency; i++ {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\t// unable to start specified # of workflows in time.\n\t\t\t// Workflow will receive timeout error, the value we return here is irrelevant.\n\t\t\tlogger.Error(\"Failed to start all workflows in time\", zap.Any(\"progress\", progress))\n\t\t\treturn 0\n\t\tdefault:\n\t\t}\n\n\t\tstartWorkflowOptions := client.StartWorkflowOptions{\n\t\t\tExecutionStartToCloseTimeout: 2 * defaultWorkflowSleepDuration,\n\t\t\tID:                           fmt.Sprintf(\"%s-%s\", TestName, uuid.New()),\n\t\t\tTaskList:                     common.GetTaskListName(rand.Intn(numTaskList)),\n\t\t}\n\n\t\tif err := common.RetryOp(func() error {\n\t\t\tprogress.TotalStartWorkflowCall++\n\n\t\t\tstartCtx, cancel := context.WithTimeout(ctx, time.Duration(config.ContextTimeoutInSeconds)*time.Second)\n\t\t\twe, err := cc.StartWorkflow(startCtx, startWorkflowOptions, sleepWorkflowName, defaultWorkflowSleepDuration)\n\t\t\tcancel()\n\n\t\t\tif err == nil || cadence.IsWorkflowExecutionAlreadyStartedError(err) {\n\t\t\t\tworkflowIDChan <- we.ID\n\t\t\t\tprogress.SucceededStartWorkflowCall++\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tif common.IsServiceBusyError(err) {\n\t\t\t\t// do not count service busy as failure\n\t\t\t\tprogress.SucceededStartWorkflowCall++\n\t\t\t}\n\n\t\t\tlogger.Error(\"Failed to start workflow execution\", zap.Error(err))\n\t\t\treturn err\n\t\t}, nil); err == nil {\n\t\t\tprogress.WorkflowStarted++\n\t\t}\n\n\t\t// successfully started the workflow or gave up after several retries\n\t\tprogress.NextStartID++\n\t\tactivity.RecordHeartbeat(ctx, progress)\n\t\ttime.Sleep(time.Duration(waitDurationInMilliSeconds+rand.Intn(waitDurationJitterInMilliSeconds)) * time.Millisecond)\n\t}\n\n\tavailability := float64(progress.SucceededStartWorkflowCall) / float64(progress.TotalStartWorkflowCall)\n\tlogger.Info(\"Completed start workflow\", zap.Float64(\"availability\", availability), zap.Int(\"workflow-started\", progress.WorkflowStarted))\n\treturn availability\n}\n\nfunc cancelWorkflow(\n\tctx context.Context,\n\tconfig *lib.CancellationTestConfig,\n\tworkflowIDChan chan string,\n) float64 {\n\tlogger := activity.GetLogger(ctx)\n\n\ttotalCancelWorkflowCall := 0\n\tsucceededCancelWorkflowCall := 0\n\tworkflowCancelled := 0\n\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\t// unable to start specified # of workflows in time.\n\t\t\t// Workflow will receive timeout error, the value we return here is irrelevant.\n\t\t\tlogger.Error(\"Failed to cancel all workflows in time\")\n\t\t\treturn 0\n\t\tcase workflowID := <-workflowIDChan:\n\t\t\tif workflowID == \"\" {\n\t\t\t\t// channel has closed, all workflow cancelled\n\t\t\t\tavailability := float64(succeededCancelWorkflowCall) / float64(totalCancelWorkflowCall)\n\t\t\t\tlogger.Info(\"Completed cancel workflow\", zap.Float64(\"availability\", availability), zap.Int(\"workflow-cancelled\", workflowCancelled))\n\t\t\t\treturn availability\n\t\t\t}\n\n\t\t\tif err := common.RetryOp(func() error {\n\t\t\t\ttotalCancelWorkflowCall++\n\n\t\t\t\tcancelCtx, cancel := context.WithTimeout(ctx, time.Duration(config.ContextTimeoutInSeconds)*time.Second)\n\t\t\t\terr := cc.CancelWorkflow(cancelCtx, workflowID, \"\")\n\t\t\t\tcancel()\n\n\t\t\t\tif err == nil || common.IsCancellationAlreadyRequestedError(err) || common.IsEntityNotExistsError(err) {\n\t\t\t\t\tsucceededCancelWorkflowCall++\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t\tif common.IsServiceBusyError(err) {\n\t\t\t\t\t// do not count service busy as failure\n\t\t\t\t\tsucceededCancelWorkflowCall++\n\t\t\t\t}\n\n\t\t\t\tlogger.Error(\"Failed to cancel workflow execution\", zap.Error(err))\n\t\t\t\treturn err\n\t\t\t}, nil); err == nil {\n\t\t\t\tworkflowCancelled++\n\t\t\t}\n\n\t\t\ttime.Sleep(time.Duration(waitDurationInMilliSeconds+rand.Intn(waitDurationJitterInMilliSeconds)) * time.Millisecond)\n\t\t}\n\t}\n}\n\nfunc validationActivity(\n\tctx context.Context,\n\tconfig *lib.CancellationTestConfig,\n\ttestStartTimeNanos int64,\n) error {\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\n\tdomain := activity.GetInfo(ctx).WorkflowDomain\n\n\tquery := fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v\", sleepWorkflowName, testStartTimeNanos)\n\trequest := &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: &domain,\n\t\tQuery:  &query,\n\t}\n\n\t// 1. check if enough workflows are started\n\tresp, err := cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttotalLaunchCount := resp.GetCount()\n\tif totalLaunchCount < int64(config.TotalLaunchCount) {\n\t\treturn cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Expected to start %v workflow, actual started: %v\", config.TotalLaunchCount, totalLaunchCount))\n\t}\n\n\t// 2. check if all workflows are closed\n\tquery = fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v and CloseTime != missing\", sleepWorkflowName, testStartTimeNanos)\n\trequest.Query = &query\n\tresp, err = cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.GetCount() != totalLaunchCount {\n\t\treturn cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Not all workflows are closed, started: %v, closed: %v\", totalLaunchCount, resp.GetCount()))\n\t}\n\n\t// TODO: uncomment the following check after cancellation test is rewritten.\n\t// currently if the launcherActivity failed in the middle (e.g. heartbeat timeout),\n\t// all the workflowID in memory will be lost and those workflows can't be cancelled.\n\n\t// 3. check if enough workflows are cancelled\n\t// query = fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v and CloseStatus = 2\", sleepWorkflowName, startTimeNanos)\n\t// request.Query = &query\n\t// resp, err = cc.CountWorkflow(ctx, request)\n\t// if err != nil {\n\t// \treturn err\n\t// }\n\t// if resp.GetCount() < int64(float64(totalLaunchCount)*common.DefaultAvailabilityThreshold) {\n\t// \treturn cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Cancelled workflow count too low, started: %v, cancelled: %v\", totalLaunchCount, resp.GetCount()))\n\t// }\n\n\treturn nil\n}\n\nfunc sleepWorkflow(\n\tctx workflow.Context,\n\tsleepDuration time.Duration,\n) error {\n\treturn workflow.Sleep(ctx, sleepDuration)\n}\n"
  },
  {
    "path": "bench/load/common/activities.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n)\n\ntype (\n\t// EchoActivityParams is the paramer for echoActivity\n\tEchoActivityParams struct {\n\t\tPayload []byte\n\t}\n)\n\nconst (\n\t// EchoActivityName is the name of echoActivity\n\tEchoActivityName = \"echoActivity\"\n)\n\n// RegisterWorker registers common activities\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterActivityWithOptions(echoActivity, activity.RegisterOptions{Name: EchoActivityName})\n}\n\nfunc echoActivity(ctx context.Context, activityParams EchoActivityParams) ([]byte, error) {\n\treturn activityParams.Payload, nil\n}\n"
  },
  {
    "path": "bench/load/common/constants.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport \"time\"\n\nconst (\n\t// DefaultAvailabilityThreshold is the default threshold for availability\n\tDefaultAvailabilityThreshold = 0.99\n\n\t// DefaultContextTimeout is the default context timeout for RPC calls\n\tDefaultContextTimeout = 3 * time.Second\n)\n\nconst (\n\t// DefaultMaxRetryCount is the default max retry count\n\tDefaultMaxRetryCount = 5\n\t// DefaultRetryBackoffDuration is the default backoff duration for retry\n\tDefaultRetryBackoffDuration = 50 * time.Millisecond\n)\n\nconst (\n\t// ErrReasonValidationFailed is the failure reason for validation failure\n\tErrReasonValidationFailed = \"validation failed\"\n\t// ErrReasonWorkflowNotExist is the error reason for workflow not exist\n\tErrReasonWorkflowNotExist = \"workflow not exist\"\n)\n"
  },
  {
    "path": "bench/load/common/helpers.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n)\n\nconst (\n\ttaskListPrefix = \"cadence-bench-tl\"\n)\n\n// GetTaskListName returns the task list name for the given task list id\nfunc GetTaskListName(taskListNumber int) string {\n\treturn fmt.Sprintf(\"%s-%v\", taskListPrefix, taskListNumber)\n}\n\n// GetActivityServiceConfig returns the service config from activity context\n// Returns nil if the context does not contain the service config\nfunc GetActivityServiceConfig(ctx context.Context) *lib.RuntimeContext {\n\tval := ctx.Value(lib.CtxKeyRuntimeContext)\n\tif val == nil {\n\t\treturn nil\n\t}\n\treturn val.(*lib.RuntimeContext)\n}\n\n// IsServiceBusyError returns if the err is a ServiceBusyError\nfunc IsServiceBusyError(err error) bool {\n\t_, ok := err.(*shared.ServiceBusyError)\n\treturn ok\n}\n\n// IsCancellationAlreadyRequestedError returns if the err is a CancellationAlreadyRequestedError\nfunc IsCancellationAlreadyRequestedError(err error) bool {\n\t_, ok := err.(*shared.CancellationAlreadyRequestedError)\n\treturn ok\n}\n\n// IsEntityNotExistsError returns if the err is a EntityNotExistsError\nfunc IsEntityNotExistsError(err error) bool {\n\t_, ok := err.(*shared.EntityNotExistsError)\n\treturn ok\n}\n\n// IsNonRetryableError return true if the err is considered non-retryable\nfunc IsNonRetryableError(err error) bool {\n\tif err == context.DeadlineExceeded || err == context.Canceled {\n\t\treturn true\n\t}\n\n\tif IsEntityNotExistsError(err) {\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// RetryOp retries an operation based on the default retry policy\nfunc RetryOp(\n\top func() error,\n\tisNonRetryableError func(error) bool,\n) error {\n\tvar err error\n\tfor retryCount := 0; retryCount < DefaultMaxRetryCount; retryCount++ {\n\t\tif err = op(); err == nil {\n\t\t\treturn nil\n\t\t}\n\n\t\tif isNonRetryableError != nil && isNonRetryableError(err) {\n\t\t\treturn err\n\t\t}\n\n\t\tif retryCount < DefaultMaxRetryCount-1 {\n\t\t\ttime.Sleep(DefaultRetryBackoffDuration)\n\t\t}\n\t}\n\n\treturn err\n}\n"
  },
  {
    "path": "bench/load/concurrentexec/batchWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage concurrentexec\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\tbatchWorkflowName = \"concurrent-execution-batch-workflow\"\n)\n\nconst (\n\tbatchTypeActivity      = \"activity\"\n\tbatchTypeChildWorkflow = \"childworkflow\"\n\n\tmaxSleepTimeInSeconds = 10\n)\n\n// RegisterWorker registers workflows and activities for concurrent execution test\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(batchWorkflow, workflow.RegisterOptions{Name: batchWorkflowName})\n\tw.RegisterWorkflow(concurrentChildWorkflow)\n\tw.RegisterActivity(concurrentActivity)\n}\n\nfunc batchWorkflow(\n\tctx workflow.Context,\n\tconfig lib.ConcurrentExecTestConfig,\n\tscheduledTimeNanos int64,\n) (float64, error) {\n\tprofile, err := lib.BeginWorkflow(ctx, batchWorkflowName, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\tnow := workflow.Now(ctx).UnixNano()\n\n\tbatchType := strings.ToLower(strings.TrimSpace(config.BatchType))\n\n\tvar numTaskList int\n\tif numTaskList, err = getTaskListNum(ctx); err != nil {\n\t\treturn 0, profile.End(fmt.Errorf(\"failed to getTaskListNum, error: %v\", err))\n\t}\n\n\tfutures := make([]workflow.Future, 0, config.BatchSize)\n\tfor i := 0; i != config.BatchSize; i++ {\n\t\tswitch batchType {\n\t\tcase batchTypeActivity:\n\t\t\tfutures = append(futures, scheduleActivity(ctx, numTaskList, config.BatchTimeoutInSeconds, now))\n\t\tcase batchTypeChildWorkflow:\n\t\t\tfutures = append(futures, scheduleChildWorkflow(ctx, numTaskList, config.BatchTimeoutInSeconds, now))\n\t\tdefault:\n\t\t\treturn 0, profile.End(fmt.Errorf(\"unknown batch type: %v\", batchType))\n\t\t}\n\t}\n\n\tscheduledInTime := 0\n\tmaxScheduleLatency := time.Duration(config.BatchMaxLatencyInSeconds) * time.Second\n\tfor _, future := range futures {\n\t\tvar scheduleLatency time.Duration\n\t\tif err := future.Get(ctx, &scheduleLatency); err != nil {\n\t\t\treturn 0, profile.End(fmt.Errorf(\"childworkflow/activity failed, error: %v\", err))\n\t\t}\n\t\tif scheduleLatency < maxScheduleLatency {\n\t\t\tscheduledInTime++\n\t\t}\n\t}\n\n\treturn float64(scheduledInTime) / float64(config.BatchSize), profile.End(nil)\n}\n\nfunc scheduleActivity(\n\tctx workflow.Context,\n\tnumTaskList int,\n\ttimeoutInSeconds int,\n\tscheduledTimeNanos int64,\n) workflow.Future {\n\tao := workflow.ActivityOptions{\n\t\tTaskList:               common.GetTaskListName(rand.Intn(numTaskList)),\n\t\tScheduleToStartTimeout: time.Duration(timeoutInSeconds) * time.Second,\n\t\tStartToCloseTimeout:    2 * maxSleepTimeInSeconds * time.Second,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tExpirationInterval: time.Duration(timeoutInSeconds) * time.Second,\n\t\t},\n\t}\n\tactivityCtx := workflow.WithActivityOptions(ctx, ao)\n\treturn workflow.ExecuteActivity(activityCtx, concurrentActivity, scheduledTimeNanos, maxSleepTimeInSeconds)\n}\n\nfunc scheduleChildWorkflow(\n\tctx workflow.Context,\n\tnumTaskList int,\n\ttimeoutInSeconds int,\n\tscheduledTimeNanos int64,\n) workflow.Future {\n\tcwo := workflow.ChildWorkflowOptions{\n\t\tTaskList:                     common.GetTaskListName(rand.Intn(numTaskList)),\n\t\tExecutionStartToCloseTimeout: time.Duration(timeoutInSeconds) * time.Second,\n\t\tTaskStartToCloseTimeout:      time.Second * 10,\n\t\tParentClosePolicy:            client.ParentClosePolicyTerminate,\n\t}\n\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\treturn workflow.ExecuteChildWorkflow(childCtx, concurrentChildWorkflow, scheduledTimeNanos, maxSleepTimeInSeconds)\n}\n\nfunc concurrentActivity(\n\tctx context.Context,\n\tscheduledTimeNanos int64,\n\tmaxSleepTimeInSeconds int,\n) (time.Duration, error) {\n\tvar latency time.Duration\n\tif activity.GetInfo(ctx).Attempt == 0 {\n\t\tlatency = time.Since(time.Unix(0, scheduledTimeNanos))\n\t}\n\n\ttime.Sleep(time.Duration(rand.Intn(maxSleepTimeInSeconds)) * time.Second)\n\treturn latency, nil\n}\n\nfunc concurrentChildWorkflow(\n\tctx workflow.Context,\n\tscheduledTimeNanos int64,\n\tmaxSleepTimeInSeconds int,\n) (time.Duration, error) {\n\tlatency := workflow.Now(ctx).Sub(time.Unix(0, scheduledTimeNanos))\n\n\tvar sleepTimeInSeconds int\n\tworkflow.SideEffect(ctx, func(_ workflow.Context) interface{} {\n\t\treturn rand.Intn(maxSleepTimeInSeconds)\n\t}).Get(&sleepTimeInSeconds)\n\n\tif err := workflow.Sleep(ctx, time.Duration(sleepTimeInSeconds)*time.Second); err != nil {\n\t\treturn 0, err\n\t}\n\treturn latency, nil\n}\n\nfunc getTaskListNum(ctx workflow.Context) (int, error) {\n\tvar numTaskList int\n\tlao := workflow.LocalActivityOptions{\n\t\tScheduleToCloseTimeout: 5 * time.Second,\n\t}\n\tlaCtx := workflow.WithLocalActivityOptions(ctx, lao)\n\tif err := workflow.ExecuteLocalActivity(laCtx, func(ctx context.Context) (int, error) {\n\t\treturn ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext).Bench.NumTaskLists, nil\n\t}).Get(laCtx, &numTaskList); err != nil {\n\t\treturn 0, err\n\t}\n\treturn numTaskList, nil\n}\n"
  },
  {
    "path": "bench/load/concurrentexec/launchWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage concurrentexec\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\t// TestName is the test name for concurrent execution bench test\n\tTestName = \"concurrent-execution\"\n\n\t// LauncherWorkflowName is the workflow name for launching concurrent execution bench test\n\tLauncherWorkflowName = \"concurrent-execution-test-workflow\"\n)\n\ntype (\n\tbatchResult struct {\n\t\tscore     float64\n\t\terrString string\n\t}\n)\n\n// RegisterLauncher registers workflows for launching concurrent execution load\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(concurrentExecTestWorkflow, workflow.RegisterOptions{Name: LauncherWorkflowName})\n}\n\nfunc concurrentExecTestWorkflow(\n\tctx workflow.Context,\n\tconfig lib.ConcurrentExecTestConfig,\n) (float64, error) {\n\tbatchCompletionCh := workflow.NewChannel(ctx)\n\n\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\tstartBatchWorkflow(ctx, config, batchCompletionCh)\n\t})\n\n\ttotalScore := 0.0\n\tfor i := 0; i != config.TotalBatches; i++ {\n\t\tvar result batchResult\n\t\tbatchCompletionCh.Receive(ctx, &result)\n\t\tif len(result.errString) != 0 {\n\t\t\treturn 0, fmt.Errorf(\"batch workflow execution failed: %v\", result.errString)\n\t\t}\n\t\ttotalScore += result.score\n\t}\n\n\tavgScore := totalScore / float64(config.TotalBatches)\n\tif avgScore < common.DefaultAvailabilityThreshold {\n\t\treturn 0, fmt.Errorf(\"batch workflow score too low, expected: %v, actual: %v\", common.DefaultAvailabilityThreshold, avgScore)\n\t}\n\n\treturn avgScore, nil\n}\n\nfunc startBatchWorkflow(\n\tctx workflow.Context,\n\tconfig lib.ConcurrentExecTestConfig,\n\tbatchCompletionCh workflow.Channel,\n) {\n\tvar numTaskList int\n\tvar err error\n\tif numTaskList, err = getTaskListNum(ctx); err != nil {\n\t\tbatchCompletionCh.Send(ctx, batchResult{\n\t\t\terrString: \"Failed to getTaskListNum, error: \" + err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tnumConcurrentBatches := config.TotalBatches / config.Concurrency\n\tfor i := 0; i != numConcurrentBatches; i++ {\n\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\tstartConcurrentBatches(ctx, config, batchCompletionCh, i*config.Concurrency, numTaskList)\n\t\t})\n\n\t\tif i != numConcurrentBatches-1 {\n\t\t\ttimer := workflow.NewTimer(ctx, time.Duration(config.BatchPeriodInSeconds)*time.Second)\n\t\t\tif err := timer.Get(ctx, nil); err != nil {\n\t\t\t\tbatchCompletionCh.Send(ctx, batchResult{\n\t\t\t\t\terrString: \"Failed to start batch workflow, error: \" + err.Error(),\n\t\t\t\t})\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc startConcurrentBatches(\n\tctx workflow.Context,\n\tconfig lib.ConcurrentExecTestConfig,\n\tbatchCompletionCh workflow.Channel,\n\tbatchStartIdx int,\n\tnumTaskList int,\n) {\n\tparentWorkflowID := workflow.GetInfo(ctx).WorkflowExecution.ID\n\tnow := workflow.Now(ctx).UnixNano()\n\n\tchildFutures := make([]workflow.Future, 0, config.Concurrency)\n\tfor batchIdx := batchStartIdx; batchIdx != batchStartIdx+config.Concurrency; batchIdx++ {\n\t\tcwo := workflow.ChildWorkflowOptions{\n\t\t\tWorkflowID:                   parentWorkflowID + \"-batch-\" + strconv.Itoa(batchIdx),\n\t\t\tTaskList:                     common.GetTaskListName(rand.Intn(numTaskList)),\n\t\t\tExecutionStartToCloseTimeout: time.Duration(config.BatchTimeoutInSeconds) * time.Second,\n\t\t\tTaskStartToCloseTimeout:      time.Minute,\n\t\t\tParentClosePolicy:            client.ParentClosePolicyTerminate,\n\t\t\tWorkflowIDReusePolicy:        client.WorkflowIDReusePolicyAllowDuplicate,\n\t\t}\n\t\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\t\tchildFuture := workflow.ExecuteChildWorkflow(childCtx, batchWorkflowName, config, now)\n\t\tchildFutures = append(childFutures, childFuture)\n\t}\n\n\tfor _, childFuture := range childFutures {\n\t\tvar batchScore float64\n\t\tif err := childFuture.Get(ctx, &batchScore); err != nil {\n\t\t\tbatchCompletionCh.Send(ctx, batchResult{\n\t\t\t\terrString: \"Batch workflow failed, error: \" + err.Error(),\n\t\t\t})\n\t\t} else {\n\t\t\tbatchCompletionCh.Send(ctx, batchResult{\n\t\t\t\tscore: batchScore,\n\t\t\t})\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "bench/load/cron/workflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cron\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/basic\"\n\t\"github.com/uber/cadence/bench/load/cancellation\"\n\t\"github.com/uber/cadence/bench/load/common\"\n\t\"github.com/uber/cadence/bench/load/concurrentexec\"\n\t\"github.com/uber/cadence/bench/load/signal\"\n\t\"github.com/uber/cadence/bench/load/timer\"\n)\n\nconst (\n\t// TestName is the test name for cron bench test\n\tTestName = \"cron\"\n)\n\nconst (\n\tcronWorkflowName         = \"cron-test-workflow\"\n\tcronLauncherWorkflowName = \"cron-launcher-workflow\"\n\n\tqueryTypeTestResults = \"test-results\"\n\n\ttestResultSignalName = \"test-result\"\n\n\ttestPassedBoolSearchAttribute = \"Passed\"\n)\n\ntype (\n\tworkflowExecution struct {\n\t\tWorkflowID string\n\t\tRunID      string\n\t}\n\n\ttestResult struct {\n\t\tName        string\n\t\tDescription string\n\t\tTestStatus  string\n\t\tDetails     string\n\t}\n)\n\nconst (\n\ttestStatusPassed = \"Passed\"\n\ttestStatusFailed = \"Failed\"\n)\n\n// RegisterLauncher registers workflows for cron load launching\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(cronWorkflow, workflow.RegisterOptions{Name: cronWorkflowName})\n\tw.RegisterWorkflowWithOptions(launcherWorkflow, workflow.RegisterOptions{Name: cronLauncherWorkflowName})\n\tw.RegisterActivity(launcherActivity)\n\tw.RegisterActivity(signalResultActivity)\n}\n\nfunc cronWorkflow(\n\tctx workflow.Context,\n\tconfig lib.CronTestConfig,\n) error {\n\tnow := workflow.Now(ctx).UnixNano()\n\tprofile, err := lib.BeginWorkflow(ctx, cronWorkflowName, now)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tworkflowTimeoutInSeconds := workflow.GetInfo(ctx).ExecutionStartToCloseTimeoutSeconds\n\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tBackoffCoefficient: 2,\n\t\t\tExpirationInterval: time.Duration(workflowTimeoutInSeconds) * time.Second, // retry the activity until workflow timeout\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\ttotalTests := 0\n\tfutures := make([]workflow.Future, 0, len(config.TestSuites))\n\tfor _, testSuite := range config.TestSuites {\n\t\ttotalTests += len(testSuite.Configs)\n\t\ttestSuiteName := testSuite.Name\n\t\tfuture := workflow.ExecuteActivity(ctx, launcherActivity, testSuite.Domain, testSuiteName, testSuite.Configs, workflowTimeoutInSeconds)\n\t\tfutures = append(futures, future)\n\t}\n\n\tfor _, future := range futures {\n\t\tif err := future.Get(ctx, nil); err != nil {\n\t\t\treturn profile.End(err)\n\t\t}\n\t}\n\n\ttestResults := make([]testResult, 0, totalTests)\n\t_ = workflow.SetQueryHandler(ctx, queryTypeTestResults, func() ([]testResult, error) {\n\t\treturn testResults, nil\n\t})\n\n\ttestPassed := true\n\ttestResultCh := workflow.GetSignalChannel(ctx, testResultSignalName)\n\tfor len(testResults) != totalTests {\n\t\tvar result testResult\n\t\ttestResultCh.Receive(ctx, &result)\n\t\ttestResults = append(testResults, result)\n\n\t\tif result.TestStatus == testStatusFailed {\n\t\t\ttestPassed = false\n\t\t}\n\t}\n\n\terr = workflow.UpsertSearchAttributes(ctx, map[string]interface{}{testPassedBoolSearchAttribute: testPassed})\n\n\treturn profile.End(err)\n}\n\nfunc launcherActivity(\n\tctx context.Context,\n\tdomain string,\n\ttestSuiteName string,\n\ttestConfigs []lib.AggregatedTestConfig,\n\ttimeoutInSeconds int32,\n) error {\n\tlogger := activity.GetLogger(ctx)\n\n\trc := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tcc, err := lib.NewCadenceClientForDomain(rc, domain)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to create cadence client\", zap.String(\"domain\", domain))\n\t\treturn err\n\t}\n\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tID:                              \"cron-launcher-\" + testSuiteName,\n\t\tTaskList:                        common.GetTaskListName(0), // default to use the tasklist with ID 0\n\t\tExecutionStartToCloseTimeout:    time.Duration(timeoutInSeconds) * time.Second,\n\t\tDecisionTaskStartToCloseTimeout: time.Minute,\n\t\tWorkflowIDReusePolicy:           client.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n\n\tactivityInfo := activity.GetInfo(ctx)\n\n\t_ = common.RetryOp(func() error {\n\t\t_, err = cc.StartWorkflow(\n\t\t\tctx,\n\t\t\tworkflowOptions,\n\t\t\tcronLauncherWorkflowName,\n\t\t\tactivityInfo.WorkflowDomain,\n\t\t\tworkflowExecution{\n\t\t\t\tWorkflowID: activityInfo.WorkflowExecution.ID,\n\t\t\t\tRunID:      activityInfo.WorkflowExecution.RunID,\n\t\t\t},\n\t\t\ttestSuiteName,\n\t\t\ttestConfigs,\n\t\t\ttime.Now().UnixNano(),\n\t\t)\n\t\tif err == nil || cadence.IsWorkflowExecutionAlreadyStartedError(err) {\n\t\t\t// TODO: it's possible that the launcher workflow is started by another cron\n\t\t\tlogger.Info(\"Started test suite\", zap.String(\"test-suite\", testSuiteName))\n\t\t\treturn nil\n\t\t}\n\n\t\tlogger.Error(\"Failed to start test suite\", zap.String(\"test-suite\", testSuiteName))\n\t\treturn err\n\t}, nil)\n\n\treturn err\n}\n\nfunc launcherWorkflow(\n\tctx workflow.Context,\n\tparentDomain string,\n\tparentExecution workflowExecution,\n\ttestSuiteName string,\n\ttestConfigs []lib.AggregatedTestConfig,\n\tscheduledTimeNanos int64,\n) ([]testResult, error) {\n\tprofile, err := lib.BeginWorkflow(ctx, cronLauncherWorkflowName, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttestTimeout := workflow.GetInfo(ctx).ExecutionStartToCloseTimeoutSeconds\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2,\n\t\t\tExpirationInterval:       time.Duration(testTimeout) * time.Second, // retry the signal activity until workflow timeout\n\t\t\tNonRetriableErrorReasons: []string{common.ErrReasonWorkflowNotExist},\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tnumTests := len(testConfigs)\n\ttestResults := make([]testResult, 0, numTests)\n\n\t// run all tests in random order\n\tvar testOrder []int\n\terr = workflow.SideEffect(ctx, func(ctx workflow.Context) interface{} {\n\t\treturn rand.Perm(len(testConfigs))\n\t}).Get(&testOrder)\n\tif err != nil {\n\t\tfor idx := range testConfigs {\n\t\t\tresult := testResult{\n\t\t\t\tName:       testConfigs[idx].Name,\n\t\t\t\tTestStatus: testStatusFailed,\n\t\t\t\tDetails:    fmt.Sprintf(\"Failed to generate randomized test order: %v\", err.Error()),\n\t\t\t}\n\t\t\ttestResults = append(testResults, result)\n\t\t\tif err := workflow.ExecuteActivity(ctx, signalResultActivity, parentDomain, parentExecution, result).Get(ctx, nil); err != nil {\n\t\t\t\tworkflow.GetLogger(ctx).Error(\"Failed to signal test result\", zap.Error(err))\n\t\t\t\treturn nil, profile.End(err)\n\t\t\t}\n\t\t}\n\t\treturn testResults, nil\n\t}\n\n\tfor _, idx := range testOrder {\n\t\ttestConfig := testConfigs[idx]\n\t\tcwo := workflow.ChildWorkflowOptions{\n\t\t\tWorkflowID:                   testSuiteName + \"-\" + testConfig.Name + \"-\" + strconv.Itoa(idx),\n\t\t\tTaskList:                     common.GetTaskListName(0), // default to use the tasklist with ID 0\n\t\t\tExecutionStartToCloseTimeout: time.Duration(testConfig.TimeoutInSeconds) * time.Second,\n\t\t\tTaskStartToCloseTimeout:      time.Minute,\n\t\t\tParentClosePolicy:            client.ParentClosePolicyTerminate,\n\t\t\tWorkflowIDReusePolicy:        client.WorkflowIDReusePolicyAllowDuplicate,\n\t\t}\n\t\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\t\tvar childFuture workflow.Future\n\t\tswitch testConfig.Name {\n\t\tcase basic.TestName:\n\t\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, basic.LauncherWorkflowName, *testConfig.Basic)\n\t\tcase signal.TestName:\n\t\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, signal.LauncherWorkflowName, *testConfig.Signal)\n\t\tcase timer.TestName:\n\t\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, timer.LauncherWorkflowName, *testConfig.Timer)\n\t\tcase concurrentexec.TestName:\n\t\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, concurrentexec.LauncherWorkflowName, *testConfig.ConcurrentExec)\n\t\tcase cancellation.TestName:\n\t\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, cancellation.LauncherWorkflowName, *testConfig.Cancellation)\n\t\tdefault:\n\t\t\tworkflow.GetLogger(ctx).Error(\"Unknown test name\", zap.String(\"test-name\", testConfig.Name))\n\t\t}\n\n\t\tresult := testResult{\n\t\t\tName:        testSuiteName + \"::\" + testConfig.Name,\n\t\t\tDescription: testConfig.Description,\n\t\t}\n\t\tif childFuture == nil {\n\t\t\tresult.TestStatus = testStatusFailed\n\t\t\tresult.Details = \"Unknown test\"\n\t\t} else if err := childFuture.Get(childCtx, nil); err != nil {\n\t\t\tresult.TestStatus = testStatusFailed\n\t\t\tresult.Details = err.Error()\n\t\t\tif customErr, ok := err.(*cadence.CustomError); ok {\n\t\t\t\tvar detailStr string\n\t\t\t\tif err := customErr.Details(&detailStr); err == nil {\n\t\t\t\t\tresult.Details += \": \" + detailStr\n\t\t\t\t}\n\t\t\t}\n\t\t} else {\n\t\t\tresult.TestStatus = testStatusPassed\n\t\t}\n\t\ttestResults = append(testResults, result)\n\n\t\tif err := workflow.ExecuteActivity(ctx, signalResultActivity, parentDomain, parentExecution, result).Get(ctx, nil); err != nil {\n\t\t\tworkflow.GetLogger(ctx).Error(\"Failed to signal test result\", zap.String(\"test-name\", testConfig.Name), zap.Error(err))\n\t\t\treturn nil, profile.End(err)\n\t\t}\n\t}\n\n\treturn testResults, profile.End(nil)\n}\n\nfunc signalResultActivity(\n\tctx context.Context,\n\ttargetDomain string,\n\ttargetWorkflowExecution workflowExecution,\n\tresult testResult,\n) error {\n\tlogger := activity.GetLogger(ctx)\n\n\trc := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tcc, err := lib.NewCadenceClientForDomain(rc, targetDomain)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to create cadence client\", zap.String(\"domain\", targetDomain))\n\t\treturn err\n\t}\n\n\tif err = common.RetryOp(func() error {\n\t\treturn cc.SignalWorkflow(\n\t\t\tctx,\n\t\t\ttargetWorkflowExecution.WorkflowID,\n\t\t\ttargetWorkflowExecution.RunID,\n\t\t\ttestResultSignalName,\n\t\t\tresult,\n\t\t)\n\t}, common.IsNonRetryableError); err != nil {\n\t\tlogger.Error(\"Failed to signal test result back to workflow\",\n\t\t\tzap.String(\"domain\", targetDomain),\n\t\t\tzap.String(\"workflowID\", targetWorkflowExecution.WorkflowID),\n\t\t\tzap.String(\"workflowID\", targetWorkflowExecution.RunID),\n\t\t\tzap.Error(err),\n\t\t)\n\t\treturn cadence.NewCustomError(common.ErrReasonWorkflowNotExist, err.Error())\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "bench/load/signal/workflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage signal\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n\tc \"github.com/uber/cadence/common\"\n)\n\nconst (\n\t// TestName is the test name\n\tTestName = \"signal\"\n\n\t// LauncherWorkflowName is the workflow name for launching load test\n\tLauncherWorkflowName = \"signal-load-test-workflow\"\n\n\tloadTestActivityName      = \"loadTestActivity\"\n\tverifyResultActivityName  = \"verifyResultActivity\"\n\tprocessSignalWorkflowName = \"processSignalWorkflow\"\n\tstressTestSignalName      = \"stressSignalName\"\n\tprocessSignalActivityName = \"processSignalActivity\"\n\texitSignalNumber          = -1\n\tfailedSignalWorkflowQuery = \"WorkflowType='%v' and CloseStatus=1 and StartTime > %v and CloseTime < %v\"\n)\n\nvar (\n\tmaxPageSize = int32(10)\n)\n\n// RegisterLauncher registers workflows and activities for load launching\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(loadTestWorkflow, workflow.RegisterOptions{Name: LauncherWorkflowName})\n\tw.RegisterActivityWithOptions(loadTestActivity, activity.RegisterOptions{Name: loadTestActivityName})\n\tw.RegisterActivityWithOptions(verifyResultActivity, activity.RegisterOptions{Name: verifyResultActivityName})\n}\n\n// RegisterWorker registers workflows and activities for sync api load\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(processSignalWorkflow, workflow.RegisterOptions{Name: processSignalWorkflowName})\n\tw.RegisterActivityWithOptions(processSignalActivity, activity.RegisterOptions{Name: processSignalActivityName})\n}\n\ntype (\n\tloadTestActivityParams struct {\n\t\tWorkflowExecutionTimeoutInSeconds int\n\t\tDecisionTaskTimeoutInSeconds      int\n\t\tCampaignCount                     int\n\t\tActionRate                        float64\n\t\tFailureRate                       float64\n\t\tStartingWorkflowID                int\n\t\tBatchWorkflowCount                int\n\t\tSignalCount                       int\n\t\tSignalDataSize                    int\n\t\tRateLimit                         int\n\t\tSignalCountBeforeContinueAsNew    int\n\t\tEnableRollingWindow               bool\n\t\tMaxSignalDelayInSeconds           int\n\t\tMaxSignalDelayCount               int\n\t}\n\n\tsignalActivityResult struct {\n\t\tSucceedCount int\n\t\tFailureCount int\n\t}\n\n\tsignalEvent struct {\n\t\tSignalNumber int\n\t\tData         []byte\n\t\tTimestamp    int64\n\t}\n\n\tverifyActivityParams struct {\n\t\tFailedWorkflowCount int64\n\t\tWorkflowStartTime   int64\n\t}\n)\n\n// loadTestWorkflow sends signals to the stress workflow\nfunc loadTestWorkflow(ctx workflow.Context, params lib.SignalTestConfig) error {\n\tinfo := workflow.GetInfo(ctx)\n\tstartTime := workflow.Now(ctx)\n\texpiration := time.Duration(info.ExecutionStartToCloseTimeoutSeconds) * time.Second\n\tprofile, err := lib.BeginWorkflow(ctx, LauncherWorkflowName, startTime.UnixNano())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tretryPolicy := &cadence.RetryPolicy{\n\t\tInitialInterval:          time.Second * 5,\n\t\tBackoffCoefficient:       1, // always backoff 5s\n\t\tMaximumInterval:          time.Second * 5,\n\t\tExpirationInterval:       expiration,\n\t\tMaximumAttempts:          2000,\n\t\tNonRetriableErrorReasons: []string{common.ErrReasonValidationFailed},\n\t}\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: expiration,\n\t\tStartToCloseTimeout:    expiration,\n\t\tHeartbeatTimeout:       time.Second * 30,\n\t\tRetryPolicy:            retryPolicy,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tbatchSize := params.LoadTestWorkflowCount / params.LoaderCount\n\tdoneCh := workflow.NewChannel(ctx)\n\tfor i := 0; i < params.LoaderCount; i++ {\n\t\tloaderParams := loadTestActivityParams{\n\t\t\tStartingWorkflowID:                i * batchSize,\n\t\t\tBatchWorkflowCount:                batchSize,\n\t\t\tWorkflowExecutionTimeoutInSeconds: params.WorkflowExecutionTimeoutInSeconds,\n\t\t\tDecisionTaskTimeoutInSeconds:      params.DecisionTaskTimeoutInSeconds,\n\t\t\tCampaignCount:                     params.CampaignCount,\n\t\t\tActionRate:                        params.ActionRate,\n\t\t\tFailureRate:                       params.FailureRate,\n\t\t\tRateLimit:                         params.RateLimit,\n\t\t\tSignalCount:                       params.SignalCount,\n\t\t\tSignalDataSize:                    params.SignalDataSize,\n\t\t\tSignalCountBeforeContinueAsNew:    params.SignalBeforeContinueAsNew,\n\t\t\tEnableRollingWindow:               params.EnableRollingWindow,\n\t\t\tMaxSignalDelayInSeconds:           params.MaxSignalDelayInSeconds,\n\t\t\tMaxSignalDelayCount:               params.MaxSignalDelayCount,\n\t\t}\n\n\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\tvar activityResult signalActivityResult\n\t\t\terr := workflow.ExecuteActivity(ctx, loadTestActivityName, loaderParams).Get(ctx, &activityResult)\n\t\t\tif err != nil {\n\t\t\t\tworkflow.GetLogger(ctx).Info(\"signal LoadTestActivity failed\", zap.Error(err))\n\t\t\t} else {\n\t\t\t\tworkflow.GetLogger(ctx).Info(\"signal LoadTestActivity completed\")\n\t\t\t}\n\n\t\t\tdoneCh.Send(ctx, \"done\")\n\t\t})\n\t}\n\n\tfor i := 0; i < params.LoaderCount; i++ {\n\t\tdoneCh.Receive(ctx, nil)\n\t}\n\n\tif err := workflow.Sleep(ctx, time.Minute*5); err != nil {\n\t\tprofile.End(err)\n\t\treturn err\n\t}\n\n\tactInput := verifyActivityParams{\n\t\tFailedWorkflowCount: int64(float64(params.LoadTestWorkflowCount) * params.FailureThreshold),\n\t\tWorkflowStartTime:   startTime.UnixNano(),\n\t}\n\terr = workflow.ExecuteActivity(ctx, verifyResultActivityName, actInput).Get(ctx, nil)\n\treturn profile.End(err)\n}\n\nfunc loadTestActivity(ctx context.Context, params loadTestActivityParams) (signalActivityResult, error) {\n\tinfo := activity.GetInfo(ctx)\n\tlogger := activity.GetLogger(ctx)\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\trc := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tnumTaskList := rc.Bench.NumTaskLists\n\tloaderID := info.WorkflowExecution.ID\n\tlimiter := rate.NewLimiter(rate.Limit(params.RateLimit), 1)\n\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tExecutionStartToCloseTimeout:    time.Second * time.Duration(params.WorkflowExecutionTimeoutInSeconds),\n\t\tDecisionTaskStartToCloseTimeout: time.Second * time.Duration(params.DecisionTaskTimeoutInSeconds),\n\t\tWorkflowIDReusePolicy:           client.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n\n\twfParams := lib.ProcessSignalWorkflowConfig{\n\t\tCampaignCount:             params.CampaignCount,\n\t\tActionRate:                params.ActionRate,\n\t\tFailureRate:               params.FailureRate,\n\t\tSignalBeforeContinueAsNew: params.SignalCountBeforeContinueAsNew,\n\t\tMaxSignalDelayInSeconds:   params.MaxSignalDelayInSeconds,\n\t\tMaxSignalDelayCount:       params.MaxSignalDelayCount,\n\t}\n\n\tdata := make([]byte, params.SignalDataSize)\n\tfor i := 0; i < len(data); i++ {\n\t\tdata[i] = 'a'\n\t}\n\ttotalSignalCount := params.SignalCount * params.BatchWorkflowCount\n\tsucceedCount, failedCount := 0, 0\n\twfIDs := make(map[string]struct{})\n\n\tbatchStartID := 0\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\tvar finishedID int\n\t\tif err := activity.GetHeartbeatDetails(ctx, &finishedID); err == nil {\n\t\t\tbatchStartID = finishedID + 1\n\t\t\tlogger.Info(\"recover from failed attempt\", zap.Int(\"FinishedID\", finishedID))\n\t\t}\n\t}\n\n\tlogger.Info(\"start sending signals\",\n\t\tzap.Int32(\"Attempt\", info.Attempt),\n\t\tzap.Int(\"StartSignalCount\", batchStartID),\n\t\tzap.Int(\"TotalSignalCount\", totalSignalCount),\n\t\tzap.Int(\"BatchWorkflowCount\", params.BatchWorkflowCount))\n\n\twid := -1\n\tfor i := batchStartID; i < totalSignalCount; i++ {\n\t\trandTaskListID := rand.Intn(numTaskList)\n\t\tworkflowOptions.TaskList = common.GetTaskListName(randTaskListID)\n\n\t\tif wid == -1 || rand.Float64() > 0.1 {\n\t\t\t// 10% chance we are going to send signal to the same workflow as last time, this is to simulate multiple\n\t\t\t// signals to same workflow within short time.\n\t\t\twid = params.StartingWorkflowID + int(rand.Int31n(int32(params.BatchWorkflowCount)))\n\n\t\t\t/*\n\t\t\t\tIf we have large number of target workflow set:\n\t\t\t\t\tWe want to simulate the real scenario where signals coming to a small set of users instead of randomly to any\n\t\t\t\t\tuser in the entire user set. This set of active user is moving around the glob as day time moves. So we want to\n\t\t\t\t\tsimulate here is that, we define a 1 hour window around current time, and only select random user from this\n\t\t\t\t\twindow.\n\t\t\t*/\n\t\t\tif params.EnableRollingWindow {\n\t\t\t\twid = params.StartingWorkflowID + getRandomID(params.BatchWorkflowCount, time.Now())\n\t\t\t}\n\t\t}\n\n\t\tworkflowID := getStressWorkflowID(loaderID, wid)\n\t\twfIDs[workflowID] = struct{}{}\n\t\tsignal := signalEvent{SignalNumber: i, Data: data, Timestamp: time.Now().UnixNano()}\n\t\tif err := limiter.Wait(ctx); err != nil {\n\t\t\tif ctx.Err() != nil {\n\t\t\t\treturn signalActivityResult{}, ctx.Err()\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\t\twfParams.ScheduleTimeNano = time.Now().UnixNano()\n\t\t_, err := cc.SignalWithStartWorkflow(ctx, workflowID, stressTestSignalName, signal, workflowOptions, processSignalWorkflowName, wfParams)\n\t\tif err == nil {\n\t\t\tlogger.Debug(\"SignalWithStartWorkflow succeed\", zap.String(\"workflowID\", workflowID))\n\t\t\tsucceedCount++\n\t\t} else {\n\t\t\tlogger.Error(\"SignalWithStartWorkflow failed\", zap.Error(err))\n\t\t\tfailedCount++\n\t\t}\n\t\tactivity.RecordHeartbeat(ctx, i)\n\t\tif ctx.Err() != nil {\n\t\t\treturn signalActivityResult{}, ctx.Err()\n\t\t}\n\t}\n\n\t// send last signal to notify the target workflow to exit\n\texitStartID := 0\n\tif batchStartID > totalSignalCount {\n\t\t// this means previous attempt failed after sending all normal signals, and failed while sending exit signal.\n\t\texitStartID = batchStartID - totalSignalCount\n\t}\n\tfor i := exitStartID; i < params.BatchWorkflowCount; i++ {\n\t\twid := params.StartingWorkflowID + i\n\t\tworkflowID := getStressWorkflowID(loaderID, wid)\n\t\tsignal := signalEvent{SignalNumber: -1, Data: data}\n\t\tif err := limiter.Wait(ctx); err != nil {\n\t\t\tif ctx.Err() != nil {\n\t\t\t\treturn signalActivityResult{}, ctx.Err()\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\t\terr := cc.SignalWorkflow(context.Background(), workflowID, \"\", stressTestSignalName, signal)\n\t\tif err == nil {\n\t\t\tlogger.Debug(\"SignalWorkflow succeed\", zap.String(\"workflowID\", workflowID))\n\t\t\tsucceedCount++\n\t\t} else {\n\t\t\tlogger.Error(\"SignalWorkflow failed\", zap.Error(err))\n\t\t\tfailedCount++\n\t\t}\n\t\tactivity.RecordHeartbeat(ctx, totalSignalCount+i)\n\t\tif ctx.Err() != nil {\n\t\t\treturn signalActivityResult{}, ctx.Err()\n\t\t}\n\t}\n\n\treturn signalActivityResult{SucceedCount: succeedCount, FailureCount: failedCount}, nil\n}\n\nfunc verifyResultActivity(ctx context.Context, params verifyActivityParams) error {\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\tinfo := activity.GetInfo(ctx)\n\t// step 1. verify if any open workflow\n\tlistWorkflowRequest := &shared.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          c.StringPtr(info.WorkflowDomain),\n\t\tMaximumPageSize: &maxPageSize,\n\t\tStartTimeFilter: &shared.StartTimeFilter{\n\t\t\tEarliestTime: c.Int64Ptr(params.WorkflowStartTime),\n\t\t\tLatestTime:   c.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tTypeFilter: &shared.WorkflowTypeFilter{\n\t\t\tName: c.StringPtr(processSignalWorkflowName),\n\t\t},\n\t}\n\topenWorkflow, err := cc.ListOpenWorkflow(ctx, listWorkflowRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif len(openWorkflow.Executions) > 0 {\n\t\treturn cadence.NewCustomError(\n\t\t\tcommon.ErrReasonValidationFailed,\n\t\t\t\"found open workflows after signal load test completed\",\n\t\t)\n\t}\n\n\t// step 2: check failed workflow reason\n\tquery := fmt.Sprintf(\n\t\tfailedSignalWorkflowQuery,\n\t\tprocessSignalWorkflowName,\n\t\tparams.WorkflowStartTime,\n\t\ttime.Now().UnixNano())\n\trequest := &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: c.StringPtr(info.WorkflowDomain),\n\t\tQuery:  &query,\n\t}\n\tresp, err := cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.GetCount() > params.FailedWorkflowCount {\n\t\treturn cadence.NewCustomError(\n\t\t\tcommon.ErrReasonValidationFailed,\n\t\t\t\"found failed workflows after signal load test completed\",\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc getStressWorkflowID(loaderID string, wid int) string {\n\treturn fmt.Sprintf(\"%v-sync-stress-%v\", loaderID, wid)\n}\n\nfunc processSignalWorkflow(ctx workflow.Context, params lib.ProcessSignalWorkflowConfig) error {\n\tlogger := workflow.GetLogger(ctx)\n\tinfo := workflow.GetInfo(ctx)\n\tprofile, err := lib.BeginWorkflow(ctx, processSignalWorkflowName, params.ScheduleTimeNano)\n\tif err != nil {\n\t\treturn err\n\t}\n\tch := workflow.GetSignalChannel(ctx, stressTestSignalName)\n\ttotalSigCount := 0\n\tsignalDelayCount := 0\nmain_loop:\n\tfor {\n\t\tvar signal signalEvent\n\t\tch.Receive(ctx, &signal)\n\t\ttotalSigCount++\n\t\tif signal.SignalNumber == exitSignalNumber {\n\t\t\tbreak main_loop\n\t\t}\n\n\t\terr := processSignal(ctx, signal, params)\n\t\tif err != nil {\n\t\t\t// log the error and continue\n\t\t\tlogger.Error(\"sync api bench test process signal failed.\")\n\t\t}\n\n\t\tbatchSigCount := 1\n\t\tfor ch.ReceiveAsync(&signal) {\n\t\t\ttotalSigCount++\n\t\t\tbatchSigCount++\n\t\t\tif signal.SignalNumber == exitSignalNumber {\n\t\t\t\tbreak main_loop\n\t\t\t}\n\t\t\tif workflow.Now(ctx).Sub(time.Unix(0, signal.Timestamp)) > time.Duration(params.MaxSignalDelayInSeconds)*time.Second {\n\t\t\t\tsignalDelayCount++\n\t\t\t\tif signalDelayCount > params.MaxSignalDelayCount {\n\t\t\t\t\treturn fmt.Errorf(fmt.Sprintf(\"received %v signals are longer than %v seconds.\", params.MaxSignalDelayCount, params.MaxSignalDelayInSeconds))\n\t\t\t\t}\n\t\t\t}\n\n\t\t\terr := processSignal(ctx, signal, params)\n\t\t\tif err != nil {\n\t\t\t\t// log the error and continue\n\t\t\t\tlogger.Error(\"sync stress workflow process signal failed.\")\n\t\t\t}\n\n\t\t\tif batchSigCount >= 5 {\n\t\t\t\tlogger.Info(\"force sleep 1s\", zap.String(\"WorkflowID\", info.WorkflowExecution.ID), zap.String(\"RunID\", info.WorkflowExecution.RunID))\n\t\t\t\terr := workflow.Sleep(ctx, time.Second)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.Error(\"Failed sleep\", zap.Error(err))\n\t\t\t\t}\n\t\t\t\tbreak // continue main_loop\n\t\t\t}\n\t\t}\n\n\t\tif totalSigCount >= params.SignalBeforeContinueAsNew {\n\t\t\tlogger.Info(\"ContinueAsNew\", zap.Int(\"ProcessedSignalCount\", totalSigCount))\n\t\t\tprofile.End(nil)\n\t\t\tparams.ScheduleTimeNano = workflow.Now(ctx).UnixNano()\n\t\t\treturn workflow.NewContinueAsNewError(ctx, processSignalWorkflowName, params)\n\t\t}\n\t}\n\n\tlogger.Info(\"sync stress workflow completed\")\n\tprofile.End(nil)\n\treturn nil\n}\n\nfunc processSignal(ctx workflow.Context, signal signalEvent, params lib.ProcessSignalWorkflowConfig) error {\n\tretryPolicy := &cadence.RetryPolicy{\n\t\tInitialInterval:    time.Second,\n\t\tBackoffCoefficient: 3,\n\t\tMaximumInterval:    time.Minute,\n\t\tExpirationInterval: time.Minute * 5,\n\t\tMaximumAttempts:    5,\n\t}\n\n\tlao := workflow.LocalActivityOptions{\n\t\tScheduleToCloseTimeout: time.Second * 3,\n\t\tRetryPolicy:            retryPolicy,\n\t}\n\tctx = workflow.WithLocalActivityOptions(ctx, lao)\n\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Hour,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy:            retryPolicy,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tvar actionFutures []workflow.Future\n\tvar trueConditions []string\n\tfor i := 0; i < params.CampaignCount; i++ {\n\t\tvar conditionMeet bool\n\t\terr := workflow.ExecuteLocalActivity(ctx, checkCondition, params.ActionRate, params.FailureRate).Get(ctx, &conditionMeet)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif conditionMeet {\n\t\t\tf := workflow.ExecuteActivity(ctx, processSignalActivityName, signal, params.FailureRate, workflow.Now(ctx).UnixNano())\n\t\t\tactionFutures = append(actionFutures, f)\n\t\t\ttrueConditions = append(trueConditions, strconv.Itoa(i))\n\t\t}\n\t}\n\n\tfor _, f := range actionFutures {\n\t\tvar actionResult string\n\t\terr := f.Get(ctx, &actionResult)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tworkflow.GetLogger(ctx).Info(\"Processed signal\", zap.Int(\"SignalNumber\", signal.SignalNumber), zap.Int(\"ActionCount\", len(trueConditions)))\n\treturn nil\n}\n\n// checkCondition is a local activity to check condition, it returns true if action needs to be taken\nfunc checkCondition(\n\tctx context.Context,\n\tactionRate float64,\n\tfailureRate float64,\n) (bool, error) {\n\n\tinfo := activity.GetInfo(ctx)\n\tlogger := activity.GetLogger(ctx)\n\tif info.Attempt > 0 {\n\t\tfailureRate += 0.1 * float64(info.Attempt) // increase failure rate for retry attempt\n\t\tlogger.Info(\"Retry attempt\", zap.Int32(\"Attempt\", info.Attempt), zap.Float64(\"FailureRate\", failureRate))\n\t}\n\tif rand.Float64() < failureRate {\n\n\t\treturn false, errors.New(\"failed by rand\")\n\t}\n\treturn rand.Float64() < actionRate, nil\n}\n\n// processSignalActivity is a regular activity simulate actual action\nfunc processSignalActivity(\n\tctx context.Context,\n\tevt signalEvent,\n\tfailureRate float64,\n\tscheduleTimeNano int64,\n) (err error) {\n\n\tinfo := activity.GetInfo(ctx)\n\tlogger := activity.GetLogger(ctx)\n\tsvcConfig := common.GetActivityServiceConfig(ctx)\n\tscope := svcConfig.Metrics\n\tif scope == nil {\n\t\tpanic(\"metrics client is not set\")\n\t}\n\tscope, stopWatch := lib.RecordActivityStart(scope, processSignalActivityName, scheduleTimeNano)\n\tdefer lib.RecordActivityEnd(scope, stopWatch, err)\n\n\tif info.Attempt > 0 {\n\t\tfailureRate += 0.1 * float64(info.Attempt) // incrase failure rate for retry attempt\n\t\tlogger.Info(\"Retry attempt\", zap.Int32(\"Attempt\", info.Attempt), zap.Float64(\"FailureRate\", failureRate))\n\t}\n\tif rand.Float64() < failureRate {\n\t\treturn errors.New(\"failed by rand\")\n\t}\n\n\treturn nil\n}\n\nvar secondsInADay = 24 * 60 * 60\n\nfunc getRandomID(totalCount int, now time.Time) int {\n\tr := rand.Float64()\n\twindowSize := totalCount / 24 // window size\n\tseconds := now.Second() + now.Minute()*60 + now.Hour()*3600\n\n\tbeginOfWindowID := int64(seconds-1800) * int64(totalCount) / int64(secondsInADay)\n\twid := int(beginOfWindowID + int64(r*float64(windowSize)))\n\twid = (wid + totalCount) % totalCount\n\n\treturn wid\n}\n"
  },
  {
    "path": "bench/load/timer/launchWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timer\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\t// TestName is the test name for timer test\n\tTestName = \"timer\"\n\n\t// LauncherWorkflowName is the workflow name for launching timer test\n\tLauncherWorkflowName = \"timer-load-test-workflow\"\n)\n\ntype (\n\tlauncherActivityProgress struct {\n\t\tWorkflowStarted int\n\t\tNextStartID     int\n\t}\n)\n\n// RegisterLauncher registers workflows and activities for timer load launching\nfunc RegisterLauncher(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(launcherWorkflow, workflow.RegisterOptions{Name: LauncherWorkflowName})\n\tw.RegisterActivity(launcherActivity)\n\tw.RegisterActivity(validationActivity)\n}\n\nfunc launcherWorkflow(ctx workflow.Context, config lib.TimerTestConfig) (float64, error) {\n\ttestTimeout := time.Duration(workflow.GetInfo(ctx).ExecutionStartToCloseTimeoutSeconds) * time.Second\n\n\tif testTimeout <= time.Duration(config.LongestTimerDurationInSeconds+config.TimerTimeoutInSeconds)*time.Second {\n\t\treturn 0, cadence.NewCustomError(\"Test timeout too short, need to be longer than LongestTimerDuration + TimerTimeout\")\n\t}\n\n\ttotalLaunchCount := config.TotalTimerCount / config.TimerPerWorkflow\n\tworkflowsPerActivity := totalLaunchCount / config.RoutineCount\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Hour,\n\t\tStartToCloseTimeout:    testTimeout,\n\t\tHeartbeatTimeout:       20 * time.Second,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    time.Second,\n\t\t\tBackoffCoefficient: 2,\n\t\t\tMaximumAttempts:    10,\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tstartTime := workflow.Now(ctx)\n\tfutures := make([]workflow.Future, 0, config.RoutineCount)\n\tfor i := 0; i != config.RoutineCount; i++ {\n\t\tfutures = append(futures, workflow.ExecuteActivity(ctx, launcherActivity, i, workflowsPerActivity, startTime, config))\n\t}\n\n\tvar totalStartedWorkflow int\n\tfor _, future := range futures {\n\t\tvar workflowStarted int\n\t\tif err := future.Get(ctx, &workflowStarted); err != nil {\n\t\t\treturn 0, err\n\t\t}\n\n\t\ttotalStartedWorkflow += workflowStarted\n\t}\n\n\t// wait until the timer fires and timerWorkflow completes\n\twaitDuration := startTime.Add(time.Duration(config.LongestTimerDurationInSeconds+config.TimerTimeoutInSeconds) * time.Second).Sub(workflow.Now(ctx))\n\tif err := workflow.NewTimer(ctx, waitDuration).Get(ctx, nil); err != nil {\n\t\treturn 0, err\n\t}\n\n\t// start the validation phase\n\n\t// 1. check if enough workflow are started\n\tif float64(totalStartedWorkflow) < common.DefaultAvailabilityThreshold*float64(totalLaunchCount) {\n\t\treturn 0, cadence.NewCustomError(\n\t\t\tcommon.ErrReasonValidationFailed,\n\t\t\tfmt.Sprintf(\"Too few workflows are started. Expected: %v, actual: %v\", totalLaunchCount, totalStartedWorkflow),\n\t\t)\n\t}\n\n\t// 2. run an activity to check the status of started workflow by querying visibility records\n\t// this checks 3 things:\n\t//     (1) number of workflow in visibility matches the one returned from launcher activity\n\t//     (2) if all started workflow has closed (meaning fire has fired and no workflow stuck).\n\t//     (3) how many timers are fired within the MaxTimerLatency limit\n\tvalidationActivityOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2,\n\t\t\tMaximumAttempts:          5,\n\t\t\tNonRetriableErrorReasons: []string{common.ErrReasonValidationFailed},\n\t\t},\n\t}\n\tctx = workflow.WithActivityOptions(ctx, validationActivityOptions)\n\n\tvar firedInTimeRatio float64\n\tif err := workflow.ExecuteActivity(ctx, validationActivity, startTime.Add(-10*time.Second).UnixNano(), totalStartedWorkflow).Get(ctx, &firedInTimeRatio); err != nil {\n\t\treturn 0, err\n\t}\n\n\treturn firedInTimeRatio, nil\n}\n\nfunc launcherActivity(\n\tctx context.Context,\n\troutineID int,\n\tlaunchCount int,\n\tstartTime time.Time,\n\tconfig lib.TimerTestConfig,\n) (int, error) {\n\tlogger := activity.GetLogger(ctx)\n\tlogger.Info(\"Start launcher activity\", zap.Int(\"routineID\", routineID))\n\tcadenceClient := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\truntimeContext := ctx.Value(lib.CtxKeyRuntimeContext).(*lib.RuntimeContext)\n\tnumTaskList := runtimeContext.Bench.NumTaskLists\n\tinput := WorkflowParams{\n\t\tTimerCount:           config.TimerPerWorkflow,\n\t\tEarliesTimerFireTime: startTime.Add(time.Duration(config.ShortestTimerDurationInSeconds) * time.Second),\n\t\tLatestTimerFireTime:  startTime.Add(time.Duration(config.LongestTimerDurationInSeconds) * time.Second),\n\t\tMaxTimerLatency:      time.Duration(config.MaxTimerLatencyInSeconds) * time.Second,\n\t}\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tExecutionStartToCloseTimeout:    72 * time.Hour,\n\t\tDecisionTaskStartToCloseTimeout: 5 * time.Minute,\n\t}\n\n\tvar progress launcherActivityProgress\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\tif err := activity.GetHeartbeatDetails(ctx, &progress); err != nil {\n\t\t\tlogger.Error(\"Failed to get activity heartbeat details\", zap.Int(\"routineID\", routineID), zap.Error(err))\n\t\t\tprogress = launcherActivityProgress{\n\t\t\t\tWorkflowStarted: 0,\n\t\t\t\tNextStartID:     0,\n\t\t\t}\n\t\t} else {\n\t\t\tlogger.Info(\"Successfully loaded activity progress\", zap.Int(\"routineID\", routineID), zap.Any(\"progress\", progress))\n\t\t}\n\t}\n\n\tfor i := progress.NextStartID; i < launchCount; i++ {\n\t\tworkflowOptions.ID = fmt.Sprintf(\"%s-%d-%s\", TestName, routineID, uuid.New())\n\t\tworkflowOptions.TaskList = common.GetTaskListName(rand.Intn(numTaskList))\n\n\t\t_ = common.RetryOp(func() error {\n\t\t\tstartCtx, cancel := context.WithTimeout(ctx, common.DefaultContextTimeout)\n\t\t\t_, err := cadenceClient.StartWorkflow(startCtx, workflowOptions, timerWorkflowName, input)\n\t\t\tcancel()\n\n\t\t\tif err == nil || cadence.IsWorkflowExecutionAlreadyStartedError(err) {\n\t\t\t\tprogress.WorkflowStarted++\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tlogger.Error(\"Failed to start workflow execution\", zap.Error(err))\n\t\t\treturn err\n\t\t}, nil)\n\n\t\tprogress.NextStartID++\n\t\tactivity.RecordHeartbeat(ctx, progress)\n\t\ttime.Sleep(time.Duration(75+rand.Intn(50)) * time.Millisecond)\n\t}\n\n\tlogger.Info(\"Completed launcher workflow\", zap.Int(\"routineID\", routineID))\n\treturn progress.WorkflowStarted, nil\n}\n\nfunc validationActivity(\n\tctx context.Context,\n\tstartTimeNanos int64,\n\tstartedWorkflow int,\n) (float64, error) {\n\tcc := ctx.Value(lib.CtxKeyCadenceClient).(lib.CadenceClient)\n\n\tdomain := activity.GetInfo(ctx).WorkflowDomain\n\n\t// 1. check if enough workflows are started\n\tquery := fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v\", timerWorkflowName, startTimeNanos)\n\trequest := &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: &domain,\n\t\tQuery:  &query,\n\t}\n\tresp, err := cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tactualLaunchedCount := resp.GetCount()\n\tif actualLaunchedCount < int64(startedWorkflow) {\n\t\treturn 0, cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Expected to start %v workflow, actual started: %v\", startedWorkflow, actualLaunchedCount))\n\t}\n\n\t// 2. check if timer has fired and workflow has closed\n\tquery = fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v and CloseTime != missing\", timerWorkflowName, startTimeNanos)\n\trequest.Query = &query\n\tresp, err = cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tif resp.GetCount() != actualLaunchedCount {\n\t\treturn 0, cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Not all workflows are closed, started: %v, closed: %v\", actualLaunchedCount, resp.GetCount()))\n\t}\n\n\t// 3. check # of timers that are fired within max latency\n\tquery = fmt.Sprintf(\"WorkflowType = '%s' and StartTime > %v and CloseStatus = 0\", timerWorkflowName, startTimeNanos)\n\trequest.Query = &query\n\tresp, err = cc.CountWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tfiredInTime := resp.GetCount()\n\tif firedInTime < int64(float64(actualLaunchedCount)*common.DefaultAvailabilityThreshold) {\n\t\treturn 0, cadence.NewCustomError(common.ErrReasonValidationFailed, fmt.Sprintf(\"Too few timers fired within latency limit, expected: %v, actual: %v\", actualLaunchedCount, firedInTime))\n\t}\n\n\treturn float64(firedInTime) / float64(actualLaunchedCount), nil\n}\n"
  },
  {
    "path": "bench/load/timer/timerWorkflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timer\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/bench/load/common\"\n)\n\nconst (\n\ttimerWorkflowName = \"timerWorkflow\"\n)\n\ntype (\n\t// WorkflowParams inputs to workflow.\n\tWorkflowParams struct {\n\t\tTimerCount           int\n\t\tEarliesTimerFireTime time.Time\n\t\tLatestTimerFireTime  time.Time\n\t\tMaxTimerLatency      time.Duration\n\t}\n)\n\n// RegisterWorker registers workflows and activities for timer load\nfunc RegisterWorker(w worker.Worker) {\n\tw.RegisterWorkflowWithOptions(timerWorkflow, workflow.RegisterOptions{Name: timerWorkflowName})\n}\n\nfunc timerWorkflow(ctx workflow.Context, workflowInput WorkflowParams) error {\n\tnow := workflow.Now(ctx)\n\tshortestTimerDuration := workflowInput.EarliesTimerFireTime.Sub(now)\n\ttimerFireWindowInSeconds := int64(workflowInput.LatestTimerFireTime.Sub(workflowInput.EarliesTimerFireTime).Seconds())\n\n\tselector := workflow.NewSelector(ctx)\n\texpectedFireTime := time.Unix(0, math.MaxInt64)\n\tfor i := 0; i != workflowInput.TimerCount; i++ {\n\t\ttimerDuration := shortestTimerDuration\n\t\tif timerFireWindowInSeconds > 0 {\n\t\t\tvar randDurationNano int64\n\t\t\tworkflow.SideEffect(ctx, func(ctx workflow.Context) interface{} {\n\t\t\t\treturn rand.Int63n(timerFireWindowInSeconds)\n\t\t\t}).Get(&randDurationNano)\n\n\t\t\ttimerDuration += time.Duration(randDurationNano) * time.Second\n\t\t}\n\n\t\tif timerDuration < 0 {\n\t\t\ttimerDuration = time.Second\n\t\t}\n\n\t\tfireTime := now.Add(timerDuration)\n\t\tif fireTime.Before(expectedFireTime) {\n\t\t\texpectedFireTime = fireTime\n\t\t}\n\n\t\tf := workflow.NewTimer(ctx, timerDuration)\n\t\tselector.AddFuture(f, func(_ workflow.Future) {})\n\t}\n\tselector.Select(ctx)\n\n\ttimerLatency := workflow.Now(ctx).Sub(expectedFireTime)\n\tif timerLatency > workflowInput.MaxTimerLatency {\n\t\treturn cadence.NewCustomError(\"timer latency too high\", fmt.Sprintf(\"expectedLatency: %v, actual latency: %v\", workflowInput.MaxTimerLatency, timerLatency))\n\t}\n\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Hour,\n\t\tStartToCloseTimeout:    time.Hour,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\treturn workflow.ExecuteActivity(ctx, common.EchoActivityName, common.EchoActivityParams{}).Get(ctx, nil)\n}\n"
  },
  {
    "path": "bench/worker.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage bench\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/bench/load/basic\"\n\t\"github.com/uber/cadence/bench/load/cancellation\"\n\t\"github.com/uber/cadence/bench/load/common\"\n\t\"github.com/uber/cadence/bench/load/concurrentexec\"\n\t\"github.com/uber/cadence/bench/load/cron\"\n\t\"github.com/uber/cadence/bench/load/signal\"\n\t\"github.com/uber/cadence/bench/load/timer\"\n)\n\ntype (\n\tloadTestWorker struct {\n\t\tclients map[string]lib.CadenceClient\n\t\truntime *lib.RuntimeContext\n\n\t\tshutdownWG sync.WaitGroup\n\t}\n)\n\nconst (\n\tdecisionWorkerConcurrency = 20\n\tactivityWorkerConcurrency = 5000\n\tstickyWorkflowCacheSize   = 100000\n)\n\n// NewWorker builds and returns a new instance of load test worker\nfunc NewWorker(cfg *lib.Config) (lib.Runnable, error) {\n\trc, err := lib.NewRuntimeContext(cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tclients, err := lib.NewCadenceClients(rc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &loadTestWorker{\n\t\truntime: rc,\n\t\tclients: clients,\n\t}, nil\n}\n\n// Run runs the worker\nfunc (w *loadTestWorker) Run() error {\n\tif err := w.createDomains(); err != nil {\n\t\treturn fmt.Errorf(\"creating bench domain: %w\", err)\n\t}\n\n\tworker.SetStickyWorkflowCacheSize(stickyWorkflowCacheSize)\n\tw.runtime.Metrics.Counter(\"worker.restarts\").Inc(1)\n\tfor _, domainName := range w.runtime.Bench.Domains {\n\t\tfor i := 0; i < w.runtime.Bench.NumTaskLists; i++ {\n\t\t\tw.shutdownWG.Add(1)\n\t\t\tgo w.runWorker(domainName, common.GetTaskListName(i))\n\t\t}\n\t}\n\n\tw.shutdownWG.Wait()\n\treturn nil\n}\n\nfunc (w *loadTestWorker) createDomains() error {\n\tfor _, domainName := range w.runtime.Bench.Domains {\n\t\tdesc := \"Domain for running cadence load test\"\n\t\towner := \"cadence-bench\"\n\t\tif err := w.clients[domainName].CreateDomain(domainName, desc, owner); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (w *loadTestWorker) runWorker(\n\tdomainName string,\n\ttaskList string,\n) {\n\tdefer w.shutdownWG.Done()\n\n\topts := w.newWorkerOptions(domainName)\n\tworker := worker.New(w.clients[domainName].Service, domainName, taskList, opts)\n\tregisterWorkers(worker)\n\tregisterLaunchers(worker)\n\n\tif err := worker.Run(); err != nil {\n\t\tw.runtime.Logger.Error(\"Failed to start worker\", zap.String(\"Domain\", domainName), zap.String(\"TaskList\", taskList), zap.Error(err))\n\t}\n}\n\nfunc (w *loadTestWorker) newWorkerOptions(domainName string) worker.Options {\n\treturn worker.Options{\n\t\tLogger:                    w.runtime.Logger,\n\t\tMetricsScope:              w.runtime.Metrics,\n\t\tBackgroundActivityContext: w.newActivityContext(domainName),\n\t\t// TODO: do we need these limits?\n\t\tMaxConcurrentActivityExecutionSize:     activityWorkerConcurrency,\n\t\tMaxConcurrentDecisionTaskExecutionSize: decisionWorkerConcurrency,\n\t}\n}\n\nfunc (w *loadTestWorker) newActivityContext(domainName string) context.Context {\n\tctx := context.Background()\n\tctx = context.WithValue(ctx, lib.CtxKeyCadenceClient, w.clients[domainName])\n\treturn context.WithValue(ctx, lib.CtxKeyRuntimeContext, w.runtime)\n}\n\nfunc registerWorkers(w worker.Worker) {\n\tcommon.RegisterWorker(w)\n\tbasic.RegisterWorker(w)\n\tsignal.RegisterWorker(w)\n\ttimer.RegisterWorker(w)\n\tconcurrentexec.RegisterWorker(w)\n\tcancellation.RegisterWorker(w)\n}\n\nfunc registerLaunchers(w worker.Worker) {\n\tcron.RegisterLauncher(w)\n\tsignal.RegisterLauncher(w)\n\tbasic.RegisterLauncher(w)\n\ttimer.RegisterLauncher(w)\n\tconcurrentexec.RegisterLauncher(w)\n\tcancellation.RegisterLauncher(w)\n}\n"
  },
  {
    "path": "canary/README.md",
    "content": "# Periodical feature health check workflow tools(aka Canary)\n\nThis README describes how to set up Cadence canary, different types of canary test cases, and how to start the canary.\n\nSetup\n-----------\n## Prerequisite: Cadence server\n\nCanary test suite is running against a Cadence server/cluster. See [documentation](https://cadenceworkflow.io/docs/operation-guide/setup/) for Cadence server cluster setup.\n\nNote that some tests require features like [Advanced Visibility]((https://cadenceworkflow.io/docs/concepts/search-workflows/).) and [History Archival](https://cadenceworkflow.io/docs/concepts/archival/).\n\nFor local server env you can run it through:\n- Docker: Instructions for running Cadence server through docker can be found in `docker/README.md`. Either `docker-compose-es-v7.yml` or `docker-compose-es.yml` can be used to start the server.\n- Build from source: Please check [CONTRIBUTING](/CONTRIBUTING.md) for how to build and run Cadence server from source. Please also make sure Kafka and ElasticSearch are running before starting the server with `./cadence-server --zone es start`. If ElasticSearch v7 is used, change the value for `--zone` flag to `es_v7`.\n\n## Run canary\n\nDifferent ways of start the canary:\n\n### 1. Use docker image `ubercadence/cadence-canary:master`\n\nYou can [pre-built docker-compose file](../docker/docker-compose-canary.yml) to run against local server\nIn the `docker/` directory, run:\n```\ndocker compose -f docker-compose-canary.yml up\n```\n\nThis will start the canary worker and also the cron canary.\n\nYou can modify [the canary worker config](../docker/config/canary/development.yaml) to run against a prod server cluster:\n* Use a different mode to start canary worker only for testing\n* Update the config to use Thrift/gRPC for communication\n* Use a different image than `master` tag.  See [docker hub](https://hub.docker.com/repository/docker/ubercadence/cadence-canary) for all the images.\nSimilar to server/CLI images, the `master` image will be built and published automatically by Github on every commit onto the `master` branch.\n\n### 2.  Build & Run\n\nIn the project root, build cadence canary binary:\n ```\n make cadence-canary\n ```\n\nThen start canary worker & cron:\n ```\n ./cadence-canary start\n ```\nThis is essentially the same as\n```\n ./cadence-canary start -mode all\n ```\n\nBy default, it will load [the configuration in `config/canary/development.yaml`](../config/canary/development.yaml).\nRun `./cadence-canary -h` for details to understand the start options of how to change the loading directory if needed.\n\nTo start the worker only for manual testing certain cases:\n ```\n  ./cadence-canary start -mode worker\n  ```\n\n### 3. Monitoring\n\nIn production, it's recommended to monitor the result of this canary. You can use [the workflow success metric](https://github.com/cadence-workflow/cadence/blob/9336ed963ca1b5e0df7206312aa5236433e04fd9/service/history/execution/context_util.go#L138)\nemitted by cadence history service `workflow_success`. To monitor all the canary test cases, use `workflowType` of `workflow.sanity`.\n\nConfigurations\n----------------------\nCanary workers configuration contains two parts:\n- **Canary**: this part controls which domains canary workers are responsible for what tests the sanity workflow will exclude.\n```yaml\ncanary:\n  domains: [\"cadence-canary\"] # it will start workers on all those domains(also try to register if not exists)\n  excludes: [\"workflow.searchAttributes\", \"workflow.batch\", \"workflow.archival.visibility\", \"workflow.archival.history\"] # it will exclude the three test cases. If archival is not enabled, you should exclude \"workflow.archival.visibility\" and\"workflow.archival.history\". If advanced visibility is not enabled, you should exclude \"workflow.searchAttributes\" and \"workflow.batch\". Otherwise canary will fail on those test cases.\n  cron:\n    cronSchedule: \"@every 30s\" #the schedule of cron canary, default to \"@every 30s\"\n    cronExecutionTimeout: 18m #the timeout of each run of the cron execution, default to 18 minutes\n    startJobTimeout: 9m #the timeout of each run of the sanity test suite, default to 9 minutes\n```\nAn exception here is `HistoryArchival` and `VisibilityArchival` test cases will always use `canary-archival-domain` domain.\n\n- **Cadence**: this control how canary worker should talk to Cadence server, which includes the server's service name and address.\n```yaml\ncadence:\n  service: \"cadence-frontend\" # frontend service name\n  address: \"127.0.0.1:7833\" # frontend address\n  #host: \"127.0.0.1:7933\" # replace address with host if using Thrift for compatibility\n  #tlsCaFile: \"path/to/file\" # give file path to TLS CA file if TLS is enabled on the Cadence server\n  #metrics: ... # optional detailed client side metrics like workflow latency. But for monitoring, simply use server side metrics `workflow_success` is enough.\n```\n- **Metrics**: metrics configuration. Similar to server metric emitter, only M3/Statsd/Prometheus is supported.\n- **Log**: logging configuration.  Similar to server logging configuration.\n\nCanary Test Cases & Starter\n----------------------\n\n### Cron Canary (periodically running the Sanity/starter suite)\n\nThe Cron workflow is not a test case. It's a top-level workflow to kick off the Sanity suite(described below) periodically.\nTo start the cron canary:\n```\n ./cadence-canary start -mode cronCanary\n ```\n\nFor local development, you can also start the cron canary workflows along with the worker:\n```\n ./cadence-canary start -m all\n ```\n\nThe Cron Schedule is from the Configuration.\nHowever, changing the schedule requires you manually terminate the existing cron workflow to take into effect.\nIt can be [improved](https://github.com/cadence-workflow/cadence/issues/4469) in the future.\n\nThe workflowID is fixed: `\"cadence.canary.cron\"`\n\n### Sanity suite (Starter for all test cases)\nThe sanity workflow is test suite workflow. It will kick off a bunch of childWorkflows for all the test to verify that Cadence server is operating correctly.\n\nAn error result of the sanity workflow indicates at least one of the test case fails.\n\nYou can start the sanity workflow as one-off run:\n```\ncadence --do <the domain you configured> workflow start --tl canary-task-queue --et 1200 --wt workflow.sanity -i 0\n```\n\nOr using the Cron Canary mentioned above to manage it.\n\n\nThen observe the progress:\n```\ncadence --do cadence-canary workflow ob -w <...workflowID form the start command output>\n```\n\nNOTE 1:\n* tasklist(tl) is fixed to `canary-task-queue`\n* execution timeout(et) is recommended to 20 minutes(`1200` seconds) but you can adjust it\n* the only required input is the scheduled unix timestamp, and `0` will uses the workflow starting time\n\n\nNOTE 2: This is the workflow that you should monitor for alerting.\nYou can use [the workflow success metric](https://github.com/cadence-workflow/cadence/blob/9336ed963ca1b5e0df7206312aa5236433e04fd9/service/history/execution/context_util.go#L138)\nemitted by cadence history service `workflow_success`. To monitor all the canary test cases use `workflowType` of `workflow.sanity`.\n\n\nNOTE 3: This is [the list of the test cases](./sanity.go) that it will start all supported test cases by default if no excludes are configured.\nYou can find [the workflow names of the tests cases in this file](./const.go) if you want to manually start certain test cases.\n\n\n### Echo\nEcho workflow tests the very basic workflow functionality. It executes an activity to return some output and verifies it as the workflow result.\n\nTo manually start an `Echo` test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.echo -i 0\n```\nThen observe the progress:\n```\ncadence --do cadence-canary workflow ob -w <...workflowID form the start command output>\n```\n\nYou can use these command for all other test cases listed below.\n\n### Signal\nSignal workflow tests the signal feature.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.signal -i 0\n```\n\n### Visibility\nVisibility workflow tests the basic visibility feature. No advanced visibility needed, but advanced visibility should also support it.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.visibility -i 0\n```\n\n### SearchAttributes\nSearchAttributes workflow tests the advanced visibility feature. Make sure advanced visibility feature is configured on the server. Otherwise, it should be excluded from the sanity test suite/case.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.searchAttributes -i 0\n```\n\n### ConcurrentExec\nConcurrentExec workflow tests executing activities concurrently.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.concurrent-execution -i 0\n```\n\n### Query\nQuery workflow tests the Query feature.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.query -i 0\n```\n\n### Timeout\nTimeout workflow make sure the activity timeout is enforced.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.timeout -i 0\n```\n\n### LocalActivity\nLocalActivity workflow tests the local activity feature.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.localactivity -i 0\n```\n\n### Cancellation\nCancellation workflowt tests cancellation feature.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.cancellation -i 0\n```\n\n### Retry\nRetry workflow tests activity retry policy.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.retry -i 0\n```\n\n### Reset\nReset workflow tests reset feature.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.reset -i 0\n```\n\n### HistoryArchival\nHistoryArchival tests history archival feature. Make sure history archival feature is configured on the server. Otherwise, it should be excluded from the sanity test suite/case.\n\nThis test case always uses `canary-archival-domain` domain.\n\nTo manually start one run of this test case:\n```\ncadence --do canary-archival-domain workflow start --tl canary-task-queue --et 10 --wt workflow.timeout -i 0\n```\n\n### VisibilityArchival\nVisibilityArchival tests visibility archival feature. Make sure visibility feature is configured on the server. Otherwise, it should be excluded from the sanity test suite/case.\n\nThis test case always uses `canary-archival-domain` domain.\n\nTo manually start one run of this test case:\n```\ncadence --do canary-archival-domain workflow start --tl canary-task-queue --et 10 --wt workflow.timeout -i 0\n```\n\n### Batch\nBatch workflow tests the batch job feature. Make sure advanced visibility feature is configured on the server. Otherwise, it should be excluded from the sanity test suite/case.\n\nTo manually start one run of this test case:\n```\ncadence --do <> workflow start --tl canary-task-queue --et 10 --wt workflow.batch -i 0\n```\n"
  },
  {
    "path": "canary/batch.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\nfunc init() {\n\tregisterWorkflow(batchWorkflow, wfTypeBatch)\n\tregisterWorkflow(batchWorkflowParent, wfTypeBatchParent)\n\tregisterWorkflow(batchWorkflowChild, wfTypeBatchChild)\n\n\tregisterActivity(verifyBatchActivity, activityTypeVerifyBatch)\n\tregisterActivity(startBatchWorkflow, activityTypeStartBatch)\n}\n\nconst (\n\t// TODO: to get rid of them:\n\t//  after batch job has an API, we should use the API: https://github.com/uber/cadence/issues/2225\n\tsysBatchWFTypeName        = \"cadence-sys-batch-workflow\"\n\tsystemBatcherTaskListName = \"cadence-sys-batcher-tasklist\"\n\n\t// there are two level, so totally 5*5 + 5 == 30 descendants\n\t// default batch RPS is 50, so it will takes ~1 seconds to terminate all\n\tnumChildrenPerLevel = 5\n)\n\ntype (\n\t// BatchParams is from server repo\n\t// TODO: to get rid of it:\n\t//  after batch job has an API, we should use the API: https://github.com/uber/cadence/issues/2225\n\tBatchParams struct {\n\t\tDomainName string\n\t\tQuery      string\n\t\tReason     string\n\t\tBatchType  string\n\t}\n)\n\nfunc batchWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tprofile, err := beginWorkflow(ctx, wfTypeBatch, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tcwo := workflow.ChildWorkflowOptions{\n\t\tExecutionStartToCloseTimeout: childWorkflowTimeout,\n\t\tTaskStartToCloseTimeout:      decisionTaskTimeout,\n\t}\n\n\tctx = workflow.WithChildOptions(ctx, cwo)\n\tvar fs []workflow.ChildWorkflowFuture\n\tfor i := 0; i < numChildrenPerLevel; i++ {\n\t\tf := workflow.ExecuteChildWorkflow(ctx, wfTypeBatchParent, scheduledTimeNanos)\n\t\tfs = append(fs, f)\n\t}\n\t// waiting for all workflow started\n\tfor i := 0; i < numChildrenPerLevel; i++ {\n\t\terr := fs[i].GetChildWorkflowExecution().Get(ctx, nil)\n\t\tif err != nil {\n\t\t\treturn profile.end(err)\n\t\t}\n\t}\n\n\t// waiting for visibility\n\tif err := workflow.Sleep(ctx, time.Second*2); err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tretryPolicy := &cadence.RetryPolicy{\n\t\tInitialInterval:    time.Second,\n\t\tBackoffCoefficient: 2,\n\t\tMaximumInterval:    time.Second * 12,\n\t\tExpirationInterval: time.Second * 3,\n\t\tMaximumAttempts:    4,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tScheduleToStartTimeout: scheduleToStartTimeout,\n\t\tStartToCloseTimeout:    activityTaskTimeout,\n\t\tRetryPolicy:            retryPolicy,\n\t})\n\n\tstartTime := workflow.Now(ctx).Format(time.RFC3339)\n\terr = workflow.ExecuteActivity(ctx, activityTypeStartBatch, domain, startTime).Get(ctx, nil)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\t// waiting for visibility\n\tif err := workflow.Sleep(ctx, time.Second*2); err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\terr = workflow.ExecuteActivity(ctx, activityTypeVerifyBatch, domain, startTime).Get(ctx, nil)\n\n\treturn profile.end(err)\n}\n\nfunc batchWorkflowParent(ctx workflow.Context, scheduledTimeNanos int64) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeBatchParent, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tcwo := workflow.ChildWorkflowOptions{\n\t\tExecutionStartToCloseTimeout: childWorkflowTimeout,\n\t\tTaskStartToCloseTimeout:      decisionTaskTimeout,\n\t}\n\n\tctx = workflow.WithChildOptions(ctx, cwo)\n\tvar fs []workflow.ChildWorkflowFuture\n\tfor i := 0; i < numChildrenPerLevel; i++ {\n\t\tf := workflow.ExecuteChildWorkflow(ctx, wfTypeBatchChild, scheduledTimeNanos)\n\t\tfs = append(fs, f)\n\t}\n\t// waiting for all workflow to finish\n\tfor i := 0; i < numChildrenPerLevel; i++ {\n\t\terr := fs[i].Get(ctx, nil)\n\t\tif err != nil {\n\t\t\treturn profile.end(err)\n\t\t}\n\t}\n\n\treturn profile.end(err)\n}\n\nfunc batchWorkflowChild(ctx workflow.Context, scheduledTimeNanos int64) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeBatchChild, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := workflow.Sleep(ctx, time.Hour); err != nil {\n\t\treturn profile.end(err)\n\t}\n\treturn profile.end(err)\n}\n\nfunc startBatchWorkflow(ctx context.Context, domain, startTime string) error {\n\tsdkClient := getContextValue(ctx, ctxKeyActivityBatcherClient).(*activityContext).cadence\n\n\tparams := BatchParams{\n\t\tDomainName: domain,\n\t\tQuery:      \"WorkflowType = '\" + wfTypeBatchParent + \"' AND CloseTime = missing AND StartTime <'\" + startTime + \"' \",\n\t\tReason:     \"batch canary\",\n\t\tBatchType:  \"terminate\",\n\t}\n\n\toptions := client.StartWorkflowOptions{\n\t\tExecutionStartToCloseTimeout:    childWorkflowTimeout,\n\t\tDecisionTaskStartToCloseTimeout: decisionTaskTimeout,\n\t\tTaskList:                        systemBatcherTaskListName,\n\t\tSearchAttributes: map[string]interface{}{\n\t\t\t\"CustomDomain\": domain,\n\t\t\t\"Operator\":     \"admin\",\n\t\t},\n\t}\n\n\trun, err := sdkClient.ExecuteWorkflow(ctx, options, sysBatchWFTypeName, params)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = run.Get(ctx, nil)\n\n\treturn err\n}\n\nfunc verifyBatchActivity(ctx context.Context, domain, startTime string) error {\n\tsvClient := getActivityContext(ctx).cadence.Service\n\n\tq1 := \"WorkflowType = '\" + wfTypeBatchParent + \"' AND CloseTime = missing  AND StartTime <'\" + startTime + \"' \"\n\tresp, err := svClient.CountWorkflowExecutions(ctx, &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: &domain,\n\t\tQuery:  &q1,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.GetCount() > 0 {\n\t\treturn fmt.Errorf(\"still seeing open workflows for %v , %v, %v\", wfTypeBatchParent, q1, resp.GetCount())\n\t}\n\n\tq2 := \"WorkflowType = '\" + wfTypeBatchChild + \"' AND CloseTime = missing  AND StartTime <'\" + startTime + \"' \"\n\tresp, err = svClient.CountWorkflowExecutions(ctx, &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: &domain,\n\t\tQuery:  &q2,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.GetCount() > 0 {\n\t\treturn fmt.Errorf(\"still seeing open workflows for %v, %v, %v\", wfTypeBatchChild, q2, resp.GetCount())\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "canary/canary.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype (\n\t// Runnable is an interface for anything that exposes a Run method\n\tRunnable interface {\n\t\tRun(mode string) error\n\t}\n\n\tcanaryImpl struct {\n\t\tcanaryClient           cadenceClient\n\t\tcanaryDomain           string\n\t\tcrossClusterDestClient *cadenceClient\n\t\tarchivalClient         cadenceClient\n\t\tsystemClient           cadenceClient\n\t\tbatcherClient          cadenceClient\n\t\truntime                *RuntimeContext\n\t\tcanaryConfig           *Canary\n\t}\n\n\tactivityContext struct {\n\t\tcadence cadenceClient\n\t}\n)\n\ntype contextKey string\n\nconst (\n\t// this context key should be the same as the one defined in\n\t// internal_worker.go, cadence go client\n\ttestTagsContextKey contextKey = \"cadence-testTags\"\n)\n\n// new returns a new instance of Canary runnable\nfunc newCanary(domain string, rc *RuntimeContext, canaryConfig *Canary) Runnable {\n\tcanaryClient := newCadenceClient(domain, rc)\n\tarchivalClient := newCadenceClient(archivalDomain, rc)\n\tsystemClient := newCadenceClient(constants.SystemLocalDomainName, rc)\n\tbatcherClient := newCadenceClient(constants.BatcherLocalDomainName, rc)\n\tvar xClusterDest cadenceClient\n\tif canaryConfig.CrossClusterTestMode == CrossClusterCanaryModeFull {\n\t\txClusterDest = newCadenceClient(deriveCanaryDomain(domain), rc)\n\t}\n\treturn &canaryImpl{\n\t\tcanaryClient:           canaryClient,\n\t\tcanaryDomain:           domain,\n\t\tcrossClusterDestClient: &xClusterDest,\n\t\tarchivalClient:         archivalClient,\n\t\tsystemClient:           systemClient,\n\t\tbatcherClient:          batcherClient,\n\t\truntime:                rc,\n\t\tcanaryConfig:           canaryConfig,\n\t}\n}\n\n// Run runs the canary\nfunc (c *canaryImpl) Run(mode string) error {\n\tif mode != ModeCronCanary && mode != ModeAll && mode != ModeWorker {\n\t\treturn fmt.Errorf(\"wrong mode to start canary\")\n\t}\n\tvar err error\n\tlog := c.runtime.logger\n\n\tif err = c.createDomain(); err != nil {\n\t\tlog.Error(\"createDomain failed\", zap.Error(err))\n\t\treturn err\n\t}\n\n\tif err = c.createArchivalDomain(); err != nil {\n\t\tlog.Error(\"createArchivalDomain failed\", zap.Error(err))\n\t\treturn err\n\t}\n\n\tif mode == ModeAll || mode == ModeCronCanary {\n\t\t// start the initial cron workflow\n\t\tc.startCronWorkflow()\n\t}\n\n\tif mode == ModeAll || mode == ModeWorker {\n\t\terr = c.startWorker()\n\t\tif err != nil {\n\t\t\tlog.Error(\"start worker failed\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (c *canaryImpl) startWorker() error {\n\tc.runtime.logger.Info(\"starting canary worker...\")\n\n\toptions := worker.Options{\n\t\tLogger:                             c.runtime.logger,\n\t\tMetricsScope:                       c.runtime.metrics,\n\t\tBackgroundActivityContext:          c.newActivityContext(),\n\t\tMaxConcurrentActivityExecutionSize: activityWorkerMaxExecutors,\n\t\tTracer:                             opentracing.GlobalTracer(),\n\t}\n\n\tarchivalWorker := worker.New(c.archivalClient.Service, archivalDomain, archivalTaskListName, options)\n\tdefer archivalWorker.Stop()\n\tif err := archivalWorker.Start(); err != nil {\n\t\treturn err\n\t}\n\tcanaryWorker := worker.New(c.canaryClient.Service, c.canaryDomain, taskListName, options)\n\tif c.canaryConfig.CrossClusterTestMode == CrossClusterCanaryModeFull {\n\t\tgo worker.New(c.canaryClient.Service, c.canaryDomain, crossClusterSrcTasklist, options).Run()\n\t\tgo worker.New(c.crossClusterDestClient.Service, c.getCrossClusterTargetDomain(), crossClusterDestTasklist, options).Run()\n\t}\n\treturn canaryWorker.Run()\n}\n\nfunc (c *canaryImpl) startCronWorkflow() {\n\tc.runtime.logger.Info(\"starting canary cron workflow...\")\n\twfID := \"cadence.canary.cron\"\n\topts := newWorkflowOptions(wfID, c.canaryConfig.Cron.CronExecutionTimeout)\n\topts.CronSchedule = c.canaryConfig.Cron.CronSchedule\n\n\t// create the cron workflow span\n\tctx := context.Background()\n\tspan := opentracing.StartSpan(\"start-cron-workflow-span\")\n\tdefer span.Finish()\n\tctx = opentracing.ContextWithSpan(ctx, span)\n\tc.registerMethods()\n\t_, err := c.canaryClient.StartWorkflow(ctx, opts, cronWorkflow, wfTypeSanity)\n\tif err != nil {\n\t\t// TODO: improvement: compare the cron schedule to decide whether or not terminating the current one\n\t\t// https://github.com/uber/cadence/issues/4469\n\t\tif _, ok := err.(*shared.WorkflowExecutionAlreadyStartedError); !ok {\n\t\t\tc.runtime.logger.Error(\"error starting cron workflow\", zap.Error(err))\n\t\t} else {\n\t\t\tc.runtime.logger.Info(\"cron workflow already started, you may need to terminate and restart if cron schedule is changed...\")\n\t\t}\n\t}\n}\n\n// newActivityContext builds an activity context containing\n// logger, metricsClient and cadenceClient\nfunc (c *canaryImpl) newActivityContext() context.Context {\n\tctx := context.WithValue(context.Background(), ctxKeyActivityRuntime, &activityContext{cadence: c.canaryClient})\n\tctx = context.WithValue(ctx, ctxKeyActivityArchivalRuntime, &activityContext{cadence: c.archivalClient})\n\tctx = context.WithValue(ctx, ctxKeyActivitySystemClient, &activityContext{cadence: c.systemClient})\n\tctx = context.WithValue(ctx, ctxKeyActivityBatcherClient, &activityContext{cadence: c.batcherClient})\n\tctx = context.WithValue(ctx, ctxKeyConfig, c.canaryConfig)\n\treturn overrideWorkerOptions(ctx)\n}\n\nfunc (c *canaryImpl) createDomain() error {\n\tname := c.canaryDomain\n\tdesc := \"Domain for running cadence canary workflows\"\n\towner := \"cadence-canary\"\n\tarchivalStatus := shared.ArchivalStatusDisabled\n\terr := c.canaryClient.createDomain(name, desc, owner, &archivalStatus, true, c.canaryConfig.CanaryDomainClusters)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif c.canaryConfig.CrossClusterTestMode == CrossClusterCanaryModeFull {\n\t\terr := c.crossClusterDestClient.createDomain(\n\t\t\tderiveCanaryDomain(c.canaryDomain),\n\t\t\t\"cross cluster canary dest domain\",\n\t\t\towner,\n\t\t\t&archivalStatus,\n\t\t\ttrue,\n\t\t\tc.canaryConfig.CanaryDomainClusters,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to setup cross-cluster domain on canary domain registration: %w\", err)\n\t\t}\n\t}\n\treturn nil\n}\n\n// registers workflow methods and activities\nfunc (c *canaryImpl) registerMethods() {\n\tregisterWorkflow(c.crossClusterParentWf, wfTypeCrossClusterParent)\n\tregisterWorkflow(c.crossClusterChildWf, wfTypeCrossClusterChild)\n\tregisterActivity(c.crossClusterSampleActivity, activityTypeCrossCluster)\n\tregisterActivity(c.failoverDestDomainActivity, activityTypeCrossClusterFailover)\n}\n\nfunc (c *canaryImpl) createArchivalDomain() error {\n\tname := archivalDomain\n\tdesc := \"Domain used by cadence canary workflows to verify archival\"\n\towner := \"cadence-canary\"\n\tarchivalStatus := shared.ArchivalStatusEnabled\n\treturn c.archivalClient.createDomain(name, desc, owner, &archivalStatus, true, c.canaryConfig.CanaryDomainClusters)\n}\n\nfunc (c *canaryImpl) getCrossClusterTargetDomain() string {\n\treturn deriveCanaryDomain(c.canaryDomain)\n}\n\n// Override worker options to create large number of pollers to improve the chances of activities getting sync matched\n//\n//nolint:unused\nfunc overrideWorkerOptions(ctx context.Context) context.Context {\n\toptionsOverride := make(map[string]map[string]string)\n\toptionsOverride[\"worker-options\"] = map[string]string{\n\t\t\"ConcurrentPollRoutineSize\": \"20\",\n\t}\n\n\treturn context.WithValue(ctx, testTagsContextKey, optionsOverride)\n}\n\nfunc deriveCanaryDomain(domain string) string {\n\treturn fmt.Sprintf(\"%s-cross-cluster\", domain)\n}\n"
  },
  {
    "path": "canary/cancellation.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\tsleepDuration = time.Second * 10\n)\n\nfunc init() {\n\tregisterWorkflow(cancellationWorkflow, wfTypeCancellation)\n\tregisterWorkflow(cancellationExternalWorkflow, wfTypeCancellationExternal)\n\tregisterActivity(cancellationActivity, activityTypeCancellation)\n\tregisterActivity(cancellationChildActivity, activityTypeCancellationChild)\n}\n\n// cancellationWorkflow is the workflow implementation to test for cancellation of workflows\nfunc cancellationWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tprofile, err := beginWorkflow(ctx, wfTypeCancellation, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// this test cancel external workflow\n\tactivityCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\terr = workflow.ExecuteActivity(activityCtx, activityTypeCancellation, workflow.Now(ctx).UnixNano()).Get(activityCtx, nil)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Info(\"cancel workflow failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\t// this test cancellation of child workflow, inside workflow\n\tcwo := newChildWorkflowOptions(domain, wfTypeCancellationExternal+\"-child\")\n\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\tchildCtx, cancel := workflow.WithCancel(childCtx)\n\tchildFuture := workflow.ExecuteChildWorkflow(childCtx, wfTypeCancellationExternal, workflow.Now(ctx).UnixNano(), sleepDuration)\n\n\tif err := workflow.Sleep(ctx, 1*time.Second); err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tcancel()\n\terr = childFuture.Get(childCtx, nil)\n\tif err == nil {\n\t\tmsg := \"cancellationWorkflow failed: child workflow not cancelled\"\n\t\tworkflow.GetLogger(ctx).Info(msg)\n\t\treturn profile.end(errors.New(msg))\n\t}\n\tif _, ok := err.(*cadence.CanceledError); !ok {\n\t\tworkflow.GetLogger(ctx).Info(\"cancellationWorkflow failed: child workflow return non CanceledError\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\t// this test cancellation of child workflow, outside workflow\n\tactivityCancelChildTestFn := func(useRunID bool) error {\n\t\tcwo := newChildWorkflowOptions(domain, wfTypeCancellationExternal+\"-child-outside\")\n\t\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\t\tchildFuture := workflow.ExecuteChildWorkflow(childCtx, wfTypeCancellationExternal, workflow.Now(ctx).UnixNano(), sleepDuration)\n\t\tchildExecution := &workflow.Execution{}\n\t\terr := childFuture.GetChildWorkflowExecution().Get(childCtx, childExecution)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !useRunID {\n\t\t\tchildExecution.RunID = \"\"\n\t\t}\n\t\t// starts a activity to cancel this child workflow\n\t\tactivityCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\t\terr = workflow.ExecuteActivity(activityCtx, activityTypeCancellationChild, workflow.Now(ctx).UnixNano(), childExecution).Get(activityCtx, nil)\n\t\tif err != nil {\n\t\t\tworkflow.GetLogger(ctx).Info(\"cancel child workflow from activity failed\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\terr = childFuture.Get(childCtx, nil)\n\t\tif err == nil {\n\t\t\tmsg := \"cancellationWorkflow failed: child workflow not cancelled\"\n\t\t\tworkflow.GetLogger(ctx).Info(msg)\n\t\t\treturn errors.New(msg)\n\t\t}\n\t\tif _, ok := err.(*cadence.CanceledError); !ok {\n\t\t\tworkflow.GetLogger(ctx).Info(\"cancellationWorkflow failed: child workflow return non CanceledError\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\terr = activityCancelChildTestFn(false)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\terr = activityCancelChildTestFn(true)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tdecisionCancelTestFn := func(useRunID bool) error {\n\t\tworkflowIDSuffix := \"-with-run-ID\"\n\t\tif !useRunID {\n\t\t\tworkflowIDSuffix = \"-without-run-ID\"\n\t\t}\n\t\tcwo = newChildWorkflowOptions(domain, wfTypeCancellationExternal+workflowIDSuffix)\n\t\tchildCtx = workflow.WithChildOptions(ctx, cwo)\n\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, wfTypeCancellationExternal, workflow.Now(ctx).UnixNano(), sleepDuration)\n\t\tchildExecution := &workflow.Execution{}\n\t\terr = childFuture.GetChildWorkflowExecution().Get(childCtx, childExecution)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !useRunID {\n\t\t\tchildExecution.RunID = \"\"\n\t\t}\n\t\tcancellatinFuture := workflow.RequestCancelExternalWorkflow(ctx, childExecution.ID, childExecution.RunID)\n\t\terr = cancellatinFuture.Get(ctx, nil)\n\t\tif err != nil {\n\t\t\tworkflow.GetLogger(ctx).Info(\"cancellationWorkflow failed: fail to send cancellation to child workflow\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\terr = childFuture.Get(childCtx, nil)\n\t\tif err == nil {\n\t\t\tmsg := \"cancellationWorkflow failed: child workflow not cancelled\"\n\t\t\tworkflow.GetLogger(ctx).Info(msg)\n\t\t\treturn errors.New(msg)\n\t\t}\n\t\tif _, ok := err.(*cadence.CanceledError); !ok {\n\t\t\tworkflow.GetLogger(ctx).Info(\"cancellationWorkflow failed: child workflow return non CanceledError\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\terr = decisionCancelTestFn(false)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\terr = decisionCancelTestFn(true)\n\treturn profile.end(err)\n}\n\n// cancellationActivity is the activity implementation to test for cancellation of non child workflow, using API\nfunc cancellationActivity(ctx context.Context, scheduledTimeNanos int64) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeCancellation, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\tapiCancelTestFn := func(useRunID bool) error {\n\t\topts := newWorkflowOptions(wfTypeCancellationExternal, childWorkflowTimeout)\n\t\tworkflowRun, err := client.ExecuteWorkflow(context.Background(), opts, wfTypeCancellationExternal, scheduledTimeNanos, sleepDuration)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\trunID := \"\"\n\t\tif useRunID {\n\t\t\trunID = workflowRun.GetRunID()\n\t\t}\n\t\terr = client.CancelWorkflow(context.Background(), wfTypeCancellationExternal, runID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = workflowRun.Get(ctx, nil)\n\t\tif err == nil {\n\t\t\treturn errors.New(\"cancellationWorkflow failed: non child workflow not cancelled\")\n\t\t}\n\t\tif _, ok := err.(*cadence.CanceledError); !ok {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\tif err := apiCancelTestFn(false); err != nil {\n\t\treturn err\n\t}\n\treturn apiCancelTestFn(true)\n}\n\n// cancellationChildActivity is the activity implementation to test for cancellation of non child workflow, using API\nfunc cancellationChildActivity(ctx context.Context, scheduledTimeNanos int64, execution workflow.Execution) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeCancellationChild, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\treturn client.CancelWorkflow(context.Background(), execution.ID, execution.RunID)\n}\n\n// cancellationExternalWorkflow is the workflow implementation to test for cancellation of non child workflow\nfunc cancellationExternalWorkflow(ctx workflow.Context, scheduledTimeNanos int64, sleepDuration time.Duration) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeCancellationExternal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\terr = workflow.Sleep(ctx, sleepDuration)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "canary/client.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\n// cadenceClient is an abstraction on top of\n// the cadence library client that serves as\n// a union of all the client interfaces that\n// the library exposes\ntype cadenceClient struct {\n\tDomain string\n\tclient.Client\n\t// domainClient only exposes domain API\n\tclient.DomainClient\n\t// this is the service needed to start the workers\n\tService workflowserviceclient.Interface\n}\n\n// createDomain creates a cadence domain with the given name and description\n// if the domain already exist, this method silently returns success\nfunc (client *cadenceClient) createDomain(name string, desc string, owner string, archivalStatus *shared.ArchivalStatus, isGlobalDomain bool, clusters []string) error {\n\temitMetric := true\n\tretention := int32(workflowRetentionDays)\n\tif archivalStatus != nil && *archivalStatus == shared.ArchivalStatusEnabled {\n\t\tretention = int32(0)\n\t}\n\tvar clusterCfg []*shared.ClusterReplicationConfiguration\n\n\tfor i := range clusters {\n\t\tclusterCfg = append(clusterCfg, &shared.ClusterReplicationConfiguration{\n\t\t\tClusterName: &clusters[i],\n\t\t})\n\t}\n\n\treq := &shared.RegisterDomainRequest{\n\t\tName:                                   &name,\n\t\tDescription:                            &desc,\n\t\tOwnerEmail:                             &owner,\n\t\tClusters:                               clusterCfg,\n\t\tWorkflowExecutionRetentionPeriodInDays: &retention,\n\t\tEmitMetric:                             &emitMetric,\n\t\tHistoryArchivalStatus:                  archivalStatus,\n\t\tVisibilityArchivalStatus:               archivalStatus,\n\t\tIsGlobalDomain:                         &isGlobalDomain,\n\t}\n\terr := client.Register(context.Background(), req)\n\tif err != nil {\n\t\tif _, ok := err.(*shared.DomainAlreadyExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// newCadenceClient builds a cadenceClient from the runtimeContext\nfunc newCadenceClient(domain string, runtime *RuntimeContext) cadenceClient {\n\ttracer := opentracing.GlobalTracer()\n\tcclient := client.NewClient(\n\t\truntime.service,\n\t\tdomain,\n\t\t&client.Options{\n\t\t\tMetricsScope: runtime.metrics,\n\t\t\tTracer:       tracer,\n\t\t},\n\t)\n\tdomainClient := client.NewDomainClient(\n\t\truntime.service,\n\t\t&client.Options{\n\t\t\tMetricsScope: runtime.metrics,\n\t\t\tTracer:       tracer,\n\t\t},\n\t)\n\treturn cadenceClient{\n\t\tDomain:       domain,\n\t\tClient:       cclient,\n\t\tDomainClient: domainClient,\n\t\tService:      runtime.service,\n\t}\n}\n\n// newWorkflowOptions builds workflowOptions with defaults for everything except startToCloseTimeout\nfunc newWorkflowOptions(id string, executionTimeout time.Duration) client.StartWorkflowOptions {\n\treturn client.StartWorkflowOptions{\n\t\tID:                              id,\n\t\tTaskList:                        taskListName,\n\t\tExecutionStartToCloseTimeout:    executionTimeout,\n\t\tDecisionTaskStartToCloseTimeout: decisionTaskTimeout,\n\t\tWorkflowIDReusePolicy:           client.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n}\n\n// newActivityOptions builds and returns activityOptions with reasonable defaults\nfunc newActivityOptions() workflow.ActivityOptions {\n\treturn workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tStartToCloseTimeout:    activityTaskTimeout,\n\t\tScheduleToStartTimeout: scheduleToStartTimeout,\n\t\tScheduleToCloseTimeout: scheduleToStartTimeout + activityTaskTimeout,\n\t}\n}\n\n// newChildWorkflowOptions builds and returns childWorkflowOptions for given domain\nfunc newChildWorkflowOptions(domain string, wfID string) workflow.ChildWorkflowOptions {\n\treturn workflow.ChildWorkflowOptions{\n\t\tDomain:                       domain,\n\t\tWorkflowID:                   wfID,\n\t\tTaskList:                     taskListName,\n\t\tExecutionStartToCloseTimeout: childWorkflowTimeout,\n\t\tTaskStartToCloseTimeout:      decisionTaskTimeout,\n\t\tWorkflowIDReusePolicy:        client.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n}\n\n// registerWorkflow registers a workflow function with a given friendly name\nfunc registerWorkflow(workflowFunc interface{}, name string) {\n\topts := workflow.RegisterOptions{Name: name}\n\tworkflow.RegisterWithOptions(workflowFunc, opts)\n}\n\n// registerActivity registers an activity function with a given friendly name\nfunc registerActivity(activityFunc interface{}, name string) {\n\topts := activity.RegisterOptions{Name: name}\n\tactivity.RegisterWithOptions(activityFunc, opts)\n}\n"
  },
  {
    "path": "canary/common.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nfunc absDurationDiff(d1, d2 time.Duration) time.Duration {\n\tif d1 > d2 {\n\t\treturn d1 - d2\n\t}\n\treturn d2 - d1\n}\n\nfunc stringPtr(v string) *string {\n\treturn &v\n}\n\nfunc int32Ptr(v int32) *int32 {\n\treturn &v\n}\n\n// getContextValue retrieves and returns the value corresponding\n// to the given key - panics if the key does not exist\nfunc getContextValue(ctx context.Context, key contextKey) interface{} {\n\tvalue := ctx.Value(key)\n\tif value == nil {\n\t\tpanic(\"ctx.Value(\" + key + \") returned nil\")\n\t}\n\treturn value\n}\n\n// getActivityContext retrieves and returns the activity context from the\n// global context passed to the activity\nfunc getActivityContext(ctx context.Context) *activityContext {\n\treturn getContextValue(ctx, ctxKeyActivityRuntime).(*activityContext)\n}\n\n// getActivityArchivalContext retrieves and returns the activity archival context from the\n// global context passed to the activity\nfunc getActivityArchivalContext(ctx context.Context) *activityContext {\n\treturn getContextValue(ctx, ctxKeyActivityArchivalRuntime).(*activityContext)\n}\n\n// checkWFVersionCompatibility takes a workflow.Context param and\n// validates that the workflow task currently being handled\n// is compatible with this version of the canary - this method\n// MUST only be called within a workflow function and it MUST\n// be the first line in the workflow function\n// Returns an error if the version is incompatible\nfunc checkWFVersionCompatibility(ctx workflow.Context) error {\n\tversion := workflow.GetVersion(ctx, workflowChangeID, workflowVersion, workflowVersion)\n\tif version != workflowVersion {\n\t\tworkflow.GetLogger(ctx).Error(\"workflow version mismatch\",\n\t\t\tzap.Int(\"want\", int(workflowVersion)), zap.Int(\"got\", int(version)))\n\t\treturn fmt.Errorf(\"workflow version mismatch, want=%v, got=%v\", workflowVersion, version)\n\t}\n\treturn nil\n}\n\n// beginWorkflow executes the common steps involved in all the workflow functions\n// It checks for workflow task version compatibility and also records the execution\n// in m3. This function must be the first call in every workflow function\n// Returns metrics scope on success, error on failure\nfunc beginWorkflow(ctx workflow.Context, wfType string, scheduledTimeNanos int64) (*workflowMetricsProfile, error) {\n\tprofile := recordWorkflowStart(ctx, wfType, scheduledTimeNanos)\n\tif err := checkWFVersionCompatibility(ctx); err != nil {\n\t\tprofile.scope.Counter(errIncompatibleVersion).Inc(1)\n\t\treturn nil, profile.end(err)\n\t}\n\treturn profile, nil\n}\n\nfunc concat(first string, second string) string {\n\treturn first + \"/\" + second\n}\n\nfunc getScheduledTimeFromInputIfNonZero(ctx workflow.Context, nanos int64) int64 {\n\tif nanos == 0 {\n\t\treturn workflow.Now(ctx).UnixNano()\n\t}\n\treturn nanos\n}\n"
  },
  {
    "path": "canary/concurrentExec.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\t// to force cuncurrent execution of activity and pagination on history events\n\t// we need to make this number larger than the default page size, which is 1000\n\tnumConcurrentExec   = 25\n\ttotalConcurrentExec = 250\n)\n\nfunc init() {\n\tregisterWorkflow(concurrentExecWorkflow, wfTypeConcurrentExec)\n\tregisterActivity(concurrentExecActivity, activityTypeConcurrentExec)\n}\n\n// concurrentExecWorkflow is the workflow implementation to test\n// 1. client side events pagination when reconstructing workflow state\n// 2. concurrent execution of activities\nfunc concurrentExecWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeConcurrentExec, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tselector := workflow.NewSelector(ctx)\n\terrors := make([]error, totalConcurrentExec)\n\n\tdoActivity := func(index int) {\n\t\tnow := workflow.Now(ctx).UnixNano()\n\t\tactivityCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\n\t\tfuture := workflow.ExecuteActivity(activityCtx, activityTypeConcurrentExec, now)\n\t\tselector.AddFuture(future, func(f workflow.Future) {\n\t\t\t// do not care about the return value\n\t\t\terrors[index] = f.Get(activityCtx, nil)\n\t\t})\n\t}\n\n\tfor index := 0; index < numConcurrentExec; index++ {\n\t\tdoActivity(index)\n\t}\n\n\tfor index := numConcurrentExec; index < totalConcurrentExec; index++ {\n\t\tselector.Select(ctx)\n\t\tdoActivity(index)\n\t}\n\n\tfor index := 0; index < numConcurrentExec; index++ {\n\t\tselector.Select(ctx)\n\t}\n\n\tfor _, err := range errors {\n\t\tif err != nil {\n\t\t\tworkflow.GetLogger(ctx).Info(\"concurrentExecActivity failed\", zap.Error(err))\n\t\t\treturn profile.end(err)\n\t\t}\n\t}\n\n\treturn profile.end(nil)\n}\n\n// concurrentExecActivity is the activity implementation for concurrent execution test\nfunc concurrentExecActivity(ctx context.Context, scheduledTimeNanos int64) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeConcurrentExec, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\t// canary test do not require any actual work to be done, so just return\n\treturn nil\n}\n"
  },
  {
    "path": "canary/config.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nconst (\n\t// EnvKeyRoot the environment variable key for runtime root dir\n\tEnvKeyRoot = \"CADENCE_CANARY_ROOT\"\n\t// EnvKeyConfigDir the environment variable key for config dir\n\tEnvKeyConfigDir = \"CADENCE_CANARY_CONFIG_DIR\"\n\t// EnvKeyEnvironment is the environment variable key for environment\n\tEnvKeyEnvironment = \"CADENCE_CANARY_ENVIRONMENT\"\n\t// EnvKeyAvailabilityZone is the environment variable key for AZ\n\tEnvKeyAvailabilityZone = \"CADENCE_CANARY_AVAILABILITY_ZONE\"\n\t// EnvKeyMode is the environment variable key for Mode\n\tEnvKeyMode = \"CADENCE_CANARY_MODE\"\n)\n\nconst (\n\t// CadenceServiceName is the default service name for cadence frontend\n\tCadenceServiceName = \"cadence-frontend\"\n\t// CanaryServiceName is the default service name for cadence canary\n\tCanaryServiceName = \"cadence-canary\"\n\t// CrossClusterCanaryModeFull is a canary testing mode which tests all permutations of\n\t// the cross-cluster/domain feature\n\tCrossClusterCanaryModeFull = \"test-all\"\n)\n\ntype (\n\t// Config contains the configurable yaml\n\t// properties for the canary runtime\n\tConfig struct {\n\t\tCanary  Canary         `yaml:\"canary\"`\n\t\tCadence Cadence        `yaml:\"cadence\"`\n\t\tLog     config.Logger  `yaml:\"log\"`\n\t\tMetrics config.Metrics `yaml:\"metrics\"`\n\t}\n\n\t// Canary contains the configuration for canary tests\n\tCanary struct {\n\t\tCrossClusterTestMode string   `yaml:\"crossClusterTestMode\"`\n\t\tCanaryDomainClusters []string `yaml:\"canaryDomainClusters\"` // the clusters to set for each domain\n\t\tDomains              []string `yaml:\"domains\"`\n\t\tExcludes             []string `yaml:\"excludes\"`\n\t\tCron                 Cron     `yaml:\"cron\"`\n\t}\n\n\t// Cron contains configuration for the cron workflow for canary\n\tCron struct {\n\t\tCronSchedule         string        `yaml:\"cronSchedule\"`         // default to \"@every 30s\"\n\t\tCronExecutionTimeout time.Duration `yaml:\"cronExecutionTimeout\"` // default to 18 minutes\n\t\tStartJobTimeout      time.Duration `yaml:\"startJobTimeout\"`      // default to 9 minutes\n\t}\n\n\t// Cadence contains the configuration for cadence service\n\tCadence struct {\n\t\tServiceName string `yaml:\"service\"`\n\t\t// support Thrift for backward compatibility. It will be ignored if host (gRPC) is used.\n\t\tThriftHostNameAndPort string `yaml:\"host\"`\n\t\t// gRPC host name and port\n\t\tGRPCHostNameAndPort string `yaml:\"address\"`\n\t\t// TLS cert file if TLS is enabled on the Cadence server\n\t\tTLSCAFile string `yaml:\"tlsCaFile\"`\n\t}\n)\n\n// Validate validates canary configration\nfunc (c *Config) Validate() error {\n\tif len(c.Canary.Domains) == 0 {\n\t\treturn errors.New(\"missing value for domains property\")\n\t}\n\treturn nil\n}\n\n// RuntimeContext contains all the context\n// information needed to run the canary\ntype RuntimeContext struct {\n\tlogger  *zap.Logger\n\tmetrics tally.Scope\n\tservice workflowserviceclient.Interface\n}\n\n// NewRuntimeContext builds a runtime context from the config\nfunc NewRuntimeContext(\n\tlogger *zap.Logger,\n\tscope tally.Scope,\n\tservice workflowserviceclient.Interface,\n) *RuntimeContext {\n\treturn &RuntimeContext{\n\t\tlogger:  logger,\n\t\tmetrics: scope,\n\t\tservice: service,\n\t}\n}\n"
  },
  {
    "path": "canary/const.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n)\n\n// global constants\nconst (\n\tworkflowRetentionDays         = int32(1)\n\tactivityWorkerMaxExecutors    = 256\n\tscheduleToStartTimeout        = 3 * time.Minute\n\tdecisionTaskTimeout           = 10 * time.Second\n\tactivityTaskTimeout           = 3 * time.Minute\n\tchildWorkflowTimeout          = 6 * time.Minute\n\ttaskListName                  = \"canary-task-queue\"\n\tcrossClusterSrcTasklist       = \"cross-cluster-src-tasklist\"\n\tcrossClusterDestTasklist      = \"cross-cluster-dest-tasklist\"\n\tctxKeyActivityRuntime         = contextKey(\"runtime\")\n\tctxKeyActivityArchivalRuntime = contextKey(\"runtime-archival\")\n\tctxKeyActivitySystemClient    = contextKey(\"system-client\")\n\tctxKeyActivityBatcherClient   = contextKey(\"batcher-client\")\n\tctxKeyConfig                  = contextKey(\"runtime-config\")\n\tarchivalDomain                = \"canary-archival-domain\"\n\tarchivalTaskListName          = \"canary-archival-task-queue\"\n)\n\n// canary running modes\nconst (\n\tModeAll        = \"all\"\n\tModeWorker     = \"worker\"\n\tModeCronCanary = \"cronCanary\"\n)\n\n// workflowVersion represents the current version of every single\n// workflow function in this canary. Every workflow function verifies\n// that the decision task it is executing is compatible with this version\n// Bump this version whenever a backward incompatible change for any workflow\n// also see beingWorkflow function\nconst workflowVersion = workflow.Version(3)\nconst workflowChangeID = \"initial version\"\n\n// wfType/activityType refers to the friendly short names given to\n// workflows and activities - at the time of registration, these names\n// will be used to associate with a workflow or activity function\nconst (\n\twfTypeCron                 = \"workflow.cron\"\n\twfTypeSanity               = \"workflow.sanity\"\n\twfTypeEcho                 = \"workflow.echo\"\n\twfTypeSignal               = \"workflow.signal\"\n\twfTypeSignalExternal       = \"workflow.signal.external\"\n\twfTypeVisibility           = \"workflow.visibility\"\n\twfTypeSearchAttributes     = \"workflow.searchAttributes\"\n\twfTypeConcurrentExec       = \"workflow.concurrent-execution\"\n\twfTypeQuery                = \"workflow.query\"\n\twfTypeTimeout              = \"workflow.timeout\"\n\twfTypeLocalActivity        = \"workflow.localactivity\"\n\twfTypeCancellation         = \"workflow.cancellation\"\n\twfTypeCancellationExternal = \"workflow.cancellation.external\"\n\twfTypeRetry                = \"workflow.retry\"\n\twfTypeResetBase            = \"workflow.reset.base\"\n\twfTypeReset                = \"workflow.reset\"\n\twfTypeHistoryArchival      = \"workflow.archival.history\"\n\twfTypeVisibilityArchival   = \"workflow.archival.visibility\"\n\twfTypeArchivalExternal     = \"workflow.archival.external\"\n\twfTypeBatch                = \"workflow.batch\"\n\twfTypeCrossClusterParent   = \"workflow.CrossCluster.parent\"\n\twfTypeCrossClusterChild    = \"workflow.CrossCluster.child\"\n\twfTypeBatchParent          = \"workflow.batch.parent\"\n\twfTypeBatchChild           = \"workflow.batch.child\"\n\n\tactivityTypeEcho                 = \"activity.echo\"\n\tactivityTypeCron                 = \"activity.cron\"\n\tactivityTypeSignal               = \"activity.signal\"\n\tactivityTypeVisibility           = \"activity.visibility\"\n\tactivityTypeSearchAttributes     = \"activity.searchAttributes\"\n\tactivityTypeConcurrentExec       = \"activity.concurrent-execution\"\n\tactivityTypeQuery1               = \"activity.query1\"\n\tactivityTypeQuery2               = \"activity.query2\"\n\tactivityTypeTimeout              = \"activity.timeout\"\n\tactivityTypeCancellation         = \"activity.cancellation\"\n\tactivityTypeCancellationChild    = \"activity.cancellation.child\"\n\tactivityTypeRetryOnTimeout       = \"activity.retry-on-timeout\"\n\tactivityTypeRetryOnFailure       = \"activity.retry-on-failure\"\n\tactivityTypeTriggerReset         = \"activity.reset.trigger\"\n\tactivityTypeVerifyReset          = \"activity.reset.verify\"\n\tactivityTypeResetBase            = \"activity.reset.base\"\n\tactivityTypeHistoryArchival      = \"activity.archival.history\"\n\tactivityTypeVisibilityArchival   = \"activity.archival.visibility\"\n\tactivityTypeLargeResult          = \"activity.largeResult\"\n\tactivityTypeVerifyBatch          = \"activity.batch.verify\"\n\tactivityTypeStartBatch           = \"activity.batch.start.batch\"\n\tactivityTypeCrossCluster         = \"activity.crosscluster.sample\"\n\tactivityTypeCrossClusterFailover = \"activity.crosscluster.failover\"\n)\n"
  },
  {
    "path": "canary/cron.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nvar (\n\tcronSleepTime = time.Second * 5\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(cronWorkflow, workflow.RegisterOptions{Name: wfTypeCron})\n\tactivity.Register(cronActivity)\n}\n\n// cronWorkflow is a generic cron workflow implementation that takes as input\n// a tuple - (jobName, JobFrequency) where jobName refers to a workflow name\n// and job frequency refers to the rate at which the job must be scheduled\n// there are several assumptions that this cron makes:\n// - jobFrequency is any value between 1-59 seconds\n// - jobName refers to a workflow function with a well-defined function signature\n// - every instance of job completes within 10 mins\nfunc cronWorkflow(\n\tctx workflow.Context,\n\tjobName string) error {\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tprofile, err := beginWorkflow(ctx, wfTypeCron, workflow.Now(ctx).UnixNano())\n\taCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\n\tstartTime := workflow.Now(ctx).UnixNano()\n\tworkflow.ExecuteActivity(aCtx, cronActivity, startTime, domain, jobName)\n\n\tworkflow.Sleep(ctx, cronSleepTime)\n\telapsed := time.Duration(workflow.Now(ctx).UnixNano() - startTime)\n\tprofile.scope.Timer(timerDriftLatency).Record(absDurationDiff(elapsed, cronSleepTime))\n\n\treturn err\n}\n\n// cronActivity starts root canary workflows at a pre-defined frequency\n// this activity exits automatically a minute after it is scheduled\nfunc cronActivity(\n\tctx context.Context,\n\tscheduledTimeNanos int64,\n\tdomain string,\n\tjobName string) error {\n\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeCron, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tcadenceClient := getActivityContext(ctx).cadence\n\tlogger := activity.GetLogger(ctx)\n\tparentID := activity.GetInfo(ctx).WorkflowExecution.ID\n\n\tjobID := fmt.Sprintf(\"%s-%s-%v\", parentID, jobName, time.Now().Format(time.RFC3339))\n\tconfig := getContextValue(ctx, ctxKeyConfig).(*Canary)\n\twf, err := startJob(&cadenceClient, scope, jobID, jobName, domain, config)\n\tif err != nil {\n\t\tlogger.Error(\"cronActivity: failed to start job\", zap.Error(err))\n\t\tif isDomainNotActiveErr(err) {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\tlogger.Info(\"cronActivity: started new job\",\n\t\t\tzap.String(\"wfID\", jobID), zap.String(\"runID\", wf.RunID))\n\t}\n\n\treturn nil\n}\n\nfunc startJob(\n\tcadenceClient *cadenceClient,\n\tscope tally.Scope,\n\tjobID string,\n\tjobName string,\n\tdomain string,\n\tconfig *Canary) (*workflow.Execution, error) {\n\n\tscope.Counter(startWorkflowCount).Inc(1)\n\tsw := scope.Timer(startWorkflowLatency).Start()\n\tdefer sw.Stop()\n\n\t// start off a workflow span\n\tctx := context.Background()\n\tspan := opentracing.StartSpan(fmt.Sprintf(\"start-%v\", jobName))\n\tdefer span.Finish()\n\tctx = opentracing.ContextWithSpan(ctx, span)\n\n\topts := newWorkflowOptions(jobID, config.Cron.StartJobTimeout)\n\twf, err := cadenceClient.StartWorkflow(ctx, opts, jobName, time.Now().UnixNano(), domain)\n\tif err != nil {\n\t\tscope.Counter(startWorkflowFailureCount).Inc(1)\n\t\tswitch err.(type) {\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\tscope.Counter(startWorkflowAlreadyStartedCount).Inc(1)\n\t\tcase *shared.DomainNotActiveError:\n\t\t\tscope.Counter(startWorkflowDomainNotActiveCount).Inc(1)\n\t\t}\n\t\treturn nil, err\n\t}\n\tscope.Counter(startWorkflowSuccessCount).Inc(1)\n\treturn wf, err\n}\n\nfunc isDomainNotActiveErr(err error) bool {\n\t_, ok := err.(*shared.DomainNotActiveError)\n\treturn ok\n}\n"
  },
  {
    "path": "canary/crosscluster.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\tshared \"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\n// some sample data to be passed around\ntype Data struct {\n\tVal        string\n\tIterations int\n}\n\nfunc (c canaryImpl) crossClusterParentWf(ctx workflow.Context) error {\n\t// first try launching a child workflow in another cluster, active in another region\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"starting child workflow in domain1, cluster 1\")\n\tctx1 := workflow.WithChildOptions(ctx, workflow.ChildWorkflowOptions{\n\t\tDomain:                       c.getCrossClusterTargetDomain(),\n\t\tWorkflowID:                   \"wf-domain-1-\" + uuid.New().String(),\n\t\tTaskList:                     crossClusterDestTasklist,\n\t\tExecutionStartToCloseTimeout: 1 * time.Minute,\n\t})\n\terr := workflow.ExecuteChildWorkflow(ctx1, wfTypeCrossClusterChild, Data{Val: \"test\"}).Get(ctx1, nil)\n\tif err != nil {\n\t\tlogger.Error(\"got error executing child workflow\", zap.Error(err))\n\t\treturn err\n\t}\n\tlogger.Info(\"test success - Cross-cluster cross-domain workflow completed\", zap.Any(\"return-value\", nil))\n\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    4 * time.Minute,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\terr = workflow.ExecuteActivity(ctx, c.failoverDestDomainActivity).Get(ctx, nil)\n\tif err != nil {\n\t\tlogger.Error(\"error during cross-cluster failover\", zap.Error(err))\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c canaryImpl) crossClusterChildWf(ctx workflow.Context, args Data) error {\n\tif args.Val != \"test\" {\n\t\tpanic(\"wf1 did not receive expected args\")\n\t}\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    4 * time.Minute,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"workflow child wf starting activity.\")\n\terr := workflow.ExecuteActivity(ctx, activityTypeCrossCluster).Get(ctx, nil)\n\tif err != nil {\n\t\tlogger.Error(\"activity error\", zap.Error(err))\n\t}\n\t// make the workflow do a loop\n\tif args.Iterations == 0 {\n\t\tlogger.Info(\"continuing as new\")\n\t\treturn workflow.NewContinueAsNewError(ctx, c.crossClusterChildWf, Data{\n\t\t\tVal:        args.Val,\n\t\t\tIterations: args.Iterations + 1,\n\t\t})\n\t}\n\tlogger.Info(\"workflow child wf completed.\")\n\treturn nil\n}\n\nfunc (c canaryImpl) crossClusterSampleActivity(ctx context.Context) (string, error) {\n\tlogger := activity.GetLogger(ctx)\n\tlogger.Info(\"activity 1 - running\")\n\treturn \"Hello - activity 1\", nil\n}\n\n// fails over the Dest domain so that both types of cross cluster operations can be tested\nfunc (c canaryImpl) failoverDestDomainActivity(ctx context.Context) error {\n\tlogger := activity.GetLogger(ctx)\n\tsrcReplicationInfo, err := getReplicationInfo(ctx, c.canaryClient, c.canaryDomain)\n\tfmt.Println(\"Src replication info\", srcReplicationInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdestReplicationInfo, err := getReplicationInfo(ctx, *c.crossClusterDestClient, c.getCrossClusterTargetDomain())\n\tfmt.Println(\"dest replication info\", srcReplicationInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\tsrcActiveCluster := srcReplicationInfo.GetActiveClusterName()\n\tdestActiveCluster := destReplicationInfo.GetActiveClusterName()\n\n\tcurrentlyInSameCluster := srcActiveCluster == destActiveCluster\n\t// don't do this every iteration so as to not get rate-limited\n\tsmallProbability := time.Now().Second() <= 10\n\ttargetDomain := c.getCrossClusterTargetDomain()\n\n\tif currentlyInSameCluster && smallProbability {\n\t\t// do a failover to move the dest to a different cluster\n\t\t// thereby making the next iteration of the canary be cross-cluster AND cross domain\n\t\t// this operation is probably nondeterministic, but that's ok in this instance\n\t\tfor _, cluster := range destReplicationInfo.GetClusters() {\n\t\t\tif cluster.GetClusterName() != srcActiveCluster {\n\t\t\t\terr = c.crossClusterDestClient.DomainClient.Update(ctx,\n\t\t\t\t\t&shared.UpdateDomainRequest{\n\t\t\t\t\t\tName:                     &targetDomain,\n\t\t\t\t\t\tReplicationConfiguration: &shared.DomainReplicationConfiguration{ActiveClusterName: cluster.ClusterName},\n\t\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn fmt.Errorf(\"failed to update destination domain for failover in crossdomain canary test: cluster trying to move dest domain to %s, error: %w \", cluster.GetClusterName(), err)\n\t\t\t\t}\n\t\t\t\tlogger.Info(\"failed over cross-cluster domain to \", zap.String(\"cluster\", cluster.GetClusterName()))\n\t\t\t\treturn nil\n\t\t\t}\n\t\t}\n\t} else if smallProbability {\n\t\t// else they're not currently in the same cluster, so fail the destination domain back to the same cluster\n\t\t// so the next iteration of the cross cluster canary test will be just cross domain\n\t\terr = c.crossClusterDestClient.DomainClient.Update(ctx,\n\t\t\t&shared.UpdateDomainRequest{\n\t\t\t\tReplicationConfiguration: &shared.DomainReplicationConfiguration{ActiveClusterName: &srcActiveCluster},\n\t\t\t\tName:                     &targetDomain,\n\t\t\t})\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to update destination domain for failover in crossdomain canary test - in this instance the dest back to the same cluster as the source: %w\", err)\n\t\t}\n\t\tlogger.Info(\"failed over cross-cluster domain to \", zap.String(\"cluster\", srcActiveCluster))\n\t}\n\treturn nil\n}\n\nfunc getReplicationInfo(ctx context.Context, client cadenceClient, domain string) (*shared.DomainReplicationConfiguration, error) {\n\tres, err := client.DomainClient.Describe(ctx, domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn res.GetReplicationConfiguration(), nil\n}\n"
  },
  {
    "path": "canary/echo.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math\"\n\t\"reflect\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\n// echoInput is the input to echoActivity\ntype echoInput struct {\n\tIntVal       int64\n\tIntPtrVal    *int64\n\tFloatVal     float64\n\tStringVal    string\n\tStringPtrVal *string\n\tSliceVal     []string\n\tMapVal       map[string]string\n}\n\n// echoOutput is the output from echoActivity\ntype echoOutput echoInput\n\nfunc init() {\n\tregisterWorkflow(echoWorkflow, wfTypeEcho)\n\tregisterActivity(echoActivity, activityTypeEcho)\n}\n\n// echoWorkflow is a workflow implementation which simply executes an\n// activity that echoes back the input as output\nfunc echoWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeEcho, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tinput := newEchoInput()\n\tnow := workflow.Now(ctx).UnixNano()\n\taCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\tfuture := workflow.ExecuteActivity(aCtx, activityTypeEcho, now, input)\n\n\tvar output echoOutput\n\tif err := future.Get(aCtx, &output); err != nil {\n\t\tworkflow.GetLogger(ctx).Info(\"echoActivity failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tif !reflect.DeepEqual(output, echoOutput(*input)) {\n\t\tworkflow.GetLogger(ctx).Sugar().Error(\"echoActivity returned wrong output\", output, input)\n\t\treturn profile.end(errors.New(\"echoActivity returned invalid output\"))\n\t}\n\n\treturn profile.end(nil)\n}\n\n// echoActivity is an activity implementation that simply returns the input as output\n// it also validates that the passed in input has the exepected values\nfunc echoActivity(ctx context.Context, scheduledTimeNanos int64, input *echoInput) (echoOutput, error) {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeEcho, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\texpected := newEchoInput()\n\tif !reflect.DeepEqual(expected, input) {\n\t\terr = errors.New(\"equality check failed\")\n\t\treturn echoOutput{}, err\n\t}\n\treturn echoOutput(*input), nil\n}\n\nfunc newEchoInput() *echoInput {\n\tintVal := int64(math.MaxInt64)\n\tstringVal := \"canary_echo_test\"\n\treturn &echoInput{\n\t\tIntVal:       intVal,\n\t\tIntPtrVal:    &intVal,\n\t\tFloatVal:     math.MaxFloat64,\n\t\tStringVal:    stringVal,\n\t\tStringPtrVal: &stringVal,\n\t\tSliceVal:     []string{\"canary\", \".\", \"echoWorkflow\"},\n\t\tMapVal:       map[string]string{\"us-east-1\": \"dca1a\", \"us-west-1\": \"sjc1a\"},\n\t}\n}\n"
  },
  {
    "path": "canary/historyArchival.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\tnumHistoryArchivals = 5\n\tresultSize          = 1024 // 1KB\n)\n\nfunc init() {\n\tregisterWorkflow(historyArchivalWorkflow, wfTypeHistoryArchival)\n\tregisterActivity(historyArchivalActivity, activityTypeHistoryArchival)\n\tregisterWorkflow(archivalExternalWorkflow, wfTypeArchivalExternal)\n\tregisterActivity(largeResultActivity, activityTypeLargeResult)\n}\n\nfunc historyArchivalWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeHistoryArchival, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\tch := workflow.NewBufferedChannel(ctx, numHistoryArchivals)\n\tfor i := 0; i < numHistoryArchivals; i++ {\n\t\tworkflow.Go(ctx, func(ctx2 workflow.Context) {\n\t\t\taCtx := workflow.WithActivityOptions(ctx2, newActivityOptions())\n\t\t\terr := workflow.ExecuteActivity(aCtx, activityTypeHistoryArchival, workflow.Now(ctx2).UnixNano()).Get(aCtx, nil)\n\t\t\terrStr := \"\"\n\t\t\tif err != nil {\n\t\t\t\terrStr = err.Error()\n\t\t\t}\n\t\t\tch.Send(ctx2, errStr)\n\t\t})\n\t}\n\tsuccessfulArchivalsCount := 0\n\tfor i := 0; i < numHistoryArchivals; i++ {\n\t\tvar errStr string\n\t\tch.Receive(ctx, &errStr)\n\t\tif errStr != \"\" {\n\t\t\tworkflow.GetLogger(ctx).Error(\"at least one archival failed\", zap.Int(\"success-count\", successfulArchivalsCount), zap.String(\"err-string\", errStr))\n\t\t\treturn profile.end(errors.New(errStr))\n\t\t}\n\t\tsuccessfulArchivalsCount++\n\t}\n\n\treturn profile.end(nil)\n}\n\nfunc historyArchivalActivity(ctx context.Context, scheduledTimeNanos int64) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeHistoryArchival, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityArchivalContext(ctx).cadence\n\texecution, err := executeArchivalExeternalWorkflow(ctx, client, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\tgetHistoryReq := &shared.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:    stringPtr(archivalDomain),\n\t\tExecution: execution,\n\t}\n\n\tfailureReason := \"\"\n\tattempts := 0\n\texpireTime := time.Now().Add(activityTaskTimeout)\n\tfor {\n\t\t<-time.After(5 * time.Second)\n\t\tif time.Now().After(expireTime) {\n\t\t\tbreak\n\t\t}\n\t\tattempts++\n\t\tbCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\t\thistory, err := client.Service.GetWorkflowExecutionHistory(bCtx, getHistoryReq)\n\t\tcancel()\n\t\tif err != nil {\n\t\t\tfailureReason = fmt.Sprintf(\"error accessing history, %v\", err.Error())\n\t\t} else if !history.GetArchived() {\n\t\t\tfailureReason = \"history is not archived\"\n\t\t} else if len(history.History.Events) == 0 {\n\t\t\tfailureReason = \"got empty history\"\n\t\t} else {\n\t\t\treturn nil\n\t\t}\n\t}\n\tactivity.GetLogger(ctx).Error(\"failed to get archived history within time limit\",\n\t\tzap.String(\"failure_reason\", failureReason),\n\t\tzap.String(\"domain\", archivalDomain),\n\t\tzap.String(\"workflow_id\", execution.GetWorkflowId()),\n\t\tzap.String(\"run_id\", execution.GetRunId()),\n\t\tzap.Int(\"attempts\", attempts))\n\treturn fmt.Errorf(\"failed to get archived history within time limit, %v\", failureReason)\n}\n\nfunc executeArchivalExeternalWorkflow(\n\tctx context.Context,\n\tclient cadenceClient,\n\tscheduledTimeNanos int64,\n) (*shared.WorkflowExecution, error) {\n\tworkflowID := fmt.Sprintf(\"%v.%v\", wfTypeArchivalExternal, uuid.New())\n\tops := newWorkflowOptions(workflowID, childWorkflowTimeout)\n\tops.TaskList = archivalTaskListName\n\tworkflowRun, err := client.ExecuteWorkflow(ctx, ops, wfTypeArchivalExternal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := workflowRun.Get(ctx, nil); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &shared.WorkflowExecution{\n\t\tWorkflowId: stringPtr(workflowID),\n\t\tRunId:      stringPtr(workflowRun.GetRunID()),\n\t}, nil\n}\n\nfunc archivalExternalWorkflow(ctx workflow.Context, scheduledTimeNanos int64) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeArchivalExternal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t}\n\tvar result []byte\n\tnumActs := rand.Intn(10) + 1\n\tfor i := 0; i != numActs; i++ {\n\t\taCtx := workflow.WithActivityOptions(ctx, ao)\n\t\tif err = workflow.ExecuteActivity(aCtx, activityTypeLargeResult).Get(aCtx, &result); err != nil {\n\t\t\tbreak\n\t\t}\n\t}\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\treturn profile.end(nil)\n}\n\nfunc largeResultActivity() ([]byte, error) {\n\treturn make([]byte, resultSize), nil\n}\n"
  },
  {
    "path": "canary/localactivity.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\nfunc init() {\n\tregisterWorkflow(localActivityWorkfow, wfTypeLocalActivity)\n\n\tactivity.Register(activityForCondition0)\n\tactivity.Register(activityForCondition1)\n\tactivity.Register(activityForCondition2)\n\tactivity.Register(activityForCondition3)\n\tactivity.Register(activityForCondition4)\n}\n\ntype conditionAndAction struct {\n\t// condition is a function pointer to a local activity\n\tcondition interface{}\n\t// action is a function pointer to a regular activity\n\taction interface{}\n}\n\nvar checks = []conditionAndAction{\n\t{checkCondition0, activityForCondition0},\n\t{checkCondition1, activityForCondition1},\n\t{checkCondition2, activityForCondition2},\n\t{checkCondition3, activityForCondition3},\n\t{checkCondition4, activityForCondition4},\n}\n\nfunc localActivityWorkfow(ctx workflow.Context, inputScheduledTimeNanos int64) (string, error) {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeSignal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tlogger := workflow.GetLogger(ctx)\n\n\tlao := workflow.LocalActivityOptions{\n\t\t// use short timeout as local activity is execute as function locally.\n\t\tScheduleToCloseTimeout: time.Second,\n\t}\n\tctx = workflow.WithLocalActivityOptions(ctx, lao)\n\n\tvar data int32\n\terr = workflow.ExecuteLocalActivity(ctx, getConditionData).Get(ctx, &data)\n\tif err != nil {\n\t\treturn \"\", profile.end(err)\n\t}\n\tlogger.Sugar().Infof(\"Get condition data %v\", data)\n\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    time.Minute,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tvar actionFutures []workflow.Future\n\n\tfor i, check := range checks {\n\t\tvar conditionMeet bool\n\t\terr := workflow.ExecuteLocalActivity(ctx, check.condition, data).Get(ctx, &conditionMeet)\n\t\tif err != nil {\n\t\t\treturn \"\", profile.end(err)\n\t\t}\n\n\t\tlogger.Sugar().Infof(\"condition meet for %v: %v\", i, conditionMeet)\n\t\tif conditionMeet {\n\t\t\tf := workflow.ExecuteActivity(ctx, check.action)\n\t\t\tactionFutures = append(actionFutures, f)\n\t\t}\n\t}\n\n\tvar processResult string\n\tfor _, f := range actionFutures {\n\t\tvar actionResult string\n\t\tif err := f.Get(ctx, &actionResult); err != nil {\n\t\t\treturn \"\", profile.end(err)\n\t\t}\n\t\tif len(processResult) > 0 {\n\t\t\tprocessResult += \" and \"\n\t\t}\n\t\tprocessResult += actionResult\n\t}\n\n\tlogger.Sugar().Infof(\"Processed condition %v: %v\", data, processResult)\n\n\treturn processResult, profile.end(nil)\n}\n\nfunc getConditionData() (int32, error) {\n\treturn rand.Int31n(100), nil\n}\n\nfunc checkCondition0(ctx context.Context, data int32) (bool, error) {\n\treturn data < 10, nil\n}\n\nfunc checkCondition1(ctx context.Context, data int32) (bool, error) {\n\treturn data >= 90, nil\n}\n\nfunc checkCondition2(ctx context.Context, data int32) (bool, error) {\n\treturn data%2 == 0, nil\n}\n\nfunc checkCondition3(ctx context.Context, data int32) (bool, error) {\n\treturn data%3 == 0, nil\n}\n\nfunc checkCondition4(ctx context.Context, data int32) (bool, error) {\n\treturn data%5 == 0, nil\n}\n\nfunc activityForCondition0(ctx context.Context) (string, error) {\n\tactivity.GetLogger(ctx).Info(\"process for condition 0\")\n\treturn \"data < 10\", nil\n}\n\nfunc activityForCondition1(ctx context.Context) (string, error) {\n\tactivity.GetLogger(ctx).Info(\"process for condition 1\")\n\treturn \"data >= 90\", nil\n}\n\nfunc activityForCondition2(ctx context.Context) (string, error) {\n\tactivity.GetLogger(ctx).Info(\"process for condition 2\")\n\treturn \"data%2 == 0\", nil\n}\n\nfunc activityForCondition3(ctx context.Context) (string, error) {\n\tactivity.GetLogger(ctx).Info(\"process for condition 3\")\n\treturn \"data%3 == 0\", nil\n}\n\nfunc activityForCondition4(ctx context.Context) (string, error) {\n\tactivity.GetLogger(ctx).Info(\"process for condition 4\")\n\treturn \"data%5 == 0\", nil\n}\n"
  },
  {
    "path": "canary/metrics.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\n// counters go here\nconst (\n\tstartedCount                      = \"started\"\n\tfailedCount                       = \"failed\"\n\tsuccessCount                      = \"succeeded\"\n\tstartWorkflowCount                = \"startworkflow\"\n\tstartWorkflowSuccessCount         = \"startworkflow.success\"\n\tstartWorkflowFailureCount         = \"startworkflow.failures\"\n\tstartWorkflowAlreadyStartedCount  = \"startworkflow.failures.alreadystarted\"\n\tstartWorkflowDomainNotActiveCount = \"startworkflow.failures.domainnotactive\"\n\tlistOpenWorkflowsCount            = \"list-open-workflows\"\n\tlistOpenWorkflowsFailureCount     = \"list-open-workflows.failures\"\n\tlistWorkflowsCount                = \"list-workflows\"\n\tlistWorkflowsFailureCount         = \"list-workflows.failures\"\n\tlistArchivedWorkflowCount         = \"list-archived-workflows\"\n\tlistArchivedWorkflowFailureCount  = \"list-archived-workflows.failures\"\n\tgetWorkflowHistoryCount           = \"get-workflow-history\"\n\tgetWorkflowHistoryFailureCount    = \"get-workflow-history.failures\"\n\terrTimeoutCount                   = \"errors.timeout\"\n\terrIncompatibleVersion            = \"errors.incompatibleversion\"\n)\n\n// latency metrics go here\nconst (\n\tlatency                      = \"latency\"\n\tstartLatency                 = \"latency.schedule-to-start\"\n\tstartWorkflowLatency         = \"latency.startworkflow\"\n\tlistOpenWorkflowsLatency     = \"latency.list-open-workflows\"\n\tlistWorkflowsLatency         = \"latency.list-workflows\"\n\tlistArchivedWorkflowsLatency = \"latency.list-archived-workflows\"\n\tgetWorkflowHistoryLatency    = \"latency.get-workflow-history\"\n\ttimerDriftLatency            = \"latency.timer-drift\"\n)\n\n// workflowMetricsProfile is the state that's needed to\n// record success/failed and latency metrics at the end\n// of a workflow\ntype workflowMetricsProfile struct {\n\tctx            workflow.Context\n\tstartTimestamp int64\n\tscope          tally.Scope\n}\n\n// workflowMetricScope creates and returns a child metric scope with tags\n// that identify the current workflow type\nfunc workflowMetricScope(ctx workflow.Context, wfType string) tally.Scope {\n\tparent := workflow.GetMetricsScope(ctx)\n\treturn parent.Tagged(map[string]string{\"operation\": wfType})\n}\n\n// recordWorkflowStart emits metrics at the beginning of a workflow function\nfunc recordWorkflowStart(ctx workflow.Context, wfType string, scheduledTimeNanos int64) *workflowMetricsProfile {\n\tnow := workflow.Now(ctx).UnixNano()\n\tscope := workflowMetricScope(ctx, wfType)\n\telapsed := max(0, now-scheduledTimeNanos)\n\tscope.Timer(startLatency).Record(time.Duration(elapsed))\n\tscope.Counter(startedCount).Inc(1)\n\treturn &workflowMetricsProfile{\n\t\tctx:            ctx,\n\t\tstartTimestamp: now,\n\t\tscope:          scope,\n\t}\n}\n\n// recordWorkflowEnd emits metrics at the end of a workflow function\nfunc recordWorkflowEnd(scope tally.Scope, elapsed time.Duration, err error) error {\n\tscope.Timer(latency).Record(elapsed)\n\tif err == nil {\n\t\tscope.Counter(successCount).Inc(1)\n\t\treturn err\n\t}\n\tscope.Counter(failedCount).Inc(1)\n\tif _, ok := err.(*workflow.TimeoutError); ok {\n\t\tscope.Counter(errTimeoutCount).Inc(1)\n\t}\n\treturn err\n}\n\n// end records the elapsed time and reports the latency,\n// success, failed counts to m3\nfunc (profile *workflowMetricsProfile) end(err error) error {\n\tnow := workflow.Now(profile.ctx).UnixNano()\n\telapsed := time.Duration(now - profile.startTimestamp)\n\treturn recordWorkflowEnd(profile.scope, elapsed, err)\n}\n\n// recordActivityStart emits metrics at the beginning of an activity function\nfunc recordActivityStart(\n\tscope tally.Scope, activityType string, scheduledTimeNanos int64) (tally.Scope, tally.Stopwatch) {\n\tscope = scope.Tagged(map[string]string{\"operation\": activityType})\n\telapsed := max(0, time.Now().UnixNano()-scheduledTimeNanos)\n\tscope.Timer(startLatency).Record(time.Duration(elapsed))\n\tscope.Counter(startedCount).Inc(1)\n\tsw := scope.Timer(latency).Start()\n\treturn scope, sw\n}\n\n// recordActivityEnd emits metrics at the end of an activity function\nfunc recordActivityEnd(scope tally.Scope, sw tally.Stopwatch, err error) {\n\tsw.Stop()\n\tif err != nil {\n\t\tscope.Counter(failedCount).Inc(1)\n\t\treturn\n\t}\n\tscope.Counter(successCount).Inc(1)\n}\n"
  },
  {
    "path": "canary/query.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\tqueryType                     = \"QueryWorkflow.QueryType\"\n\tqueryStatusStarted            = \"QueryWorkflow.Status.Started\"\n\tqueryStatusCloseToFinished    = \"QueryWorkflow.Status.CloseToFinished\"\n\tqueryStatusQueryHandlerFailed = \"QueryWorkflow.Status.QueryHandlerFailed\"\n)\n\nfunc init() {\n\tregisterWorkflow(queryWorkflow, wfTypeQuery)\n\tregisterActivity(queryActivity, activityTypeQuery1)\n\tregisterActivity(queryActivity, activityTypeQuery2)\n}\n\n// queryWorkflow is the workflow implementation to test for querying workflow status\nfunc queryWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tstatus := queryStatusStarted\n\terr := workflow.SetQueryHandler(ctx, queryType, func() (string, error) {\n\t\treturn status, nil\n\t})\n\tif err != nil {\n\t\tstatus = queryStatusQueryHandlerFailed\n\t\treturn err\n\t}\n\n\tprofile, err := beginWorkflow(ctx, wfTypeQuery, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecution := workflow.GetInfo(ctx).WorkflowExecution\n\n\tvar queryStatus string\n\n\tnow := workflow.Now(ctx).UnixNano()\n\tactivityCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\terr = workflow.ExecuteActivity(activityCtx, activityTypeQuery1, now, execution).Get(activityCtx, &queryStatus)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Info(\"queryWorkflow failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\tif queryStatus != status {\n\t\terr := fmt.Errorf(\"queryWorkflow status expected %s, but is %s\", status, queryStatus)\n\t\tworkflow.GetLogger(ctx).Info(\"queryWorkflow failed, status is not expected\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tstatus = queryStatusCloseToFinished\n\n\tactivityCtx = workflow.WithActivityOptions(ctx, newActivityOptions())\n\terr = workflow.ExecuteActivity(activityCtx, activityTypeQuery2, now, execution).Get(activityCtx, &queryStatus)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Info(\"queryWorkflow failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\tif queryStatus != status {\n\t\terr := fmt.Errorf(\"queryWorkflow status expected %s, but is %s\", status, queryStatus)\n\t\tworkflow.GetLogger(ctx).Info(\"queryWorkflow failed, status is not expected\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\treturn profile.end(nil)\n}\n\n// queryActivity is the activity implementation for querying workflow status\nfunc queryActivity(ctx context.Context, scheduledTimeNanos int64, execution workflow.Execution) (string, error) {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeConcurrentExec, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\tencodedValue, err := client.QueryWorkflow(context.Background(), execution.ID, execution.RunID, queryType)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tvar result string\n\terr = encodedValue.Get(&result)\n\n\treturn result, err\n}\n"
  },
  {
    "path": "canary/reset.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\nconst (\n\tsmallWait         = 1\n\tbigWait           = 30\n\tsignalToTrigger   = \"signalToTrigger\"\n\tsignalBeforeReset = \"signalBeforeReset\"\n\tsignalAfterReset  = \"signalAfterReset\"\n)\n\nfunc init() {\n\tregisterWorkflow(resetWorkflow, wfTypeReset)\n\tregisterWorkflow(resetBaseWorkflow, wfTypeResetBase)\n\n\tregisterActivity(triggerResetActivity, activityTypeTriggerReset)\n\tregisterActivity(verifyResetActivity, activityTypeVerifyReset)\n\tregisterActivity(resetBaseActivity, activityTypeResetBase)\n}\n\nfunc resetWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tprofile, err := beginWorkflow(ctx, wfTypeReset, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\tinfo := workflow.GetInfo(ctx)\n\n\tcwo := newChildWorkflowOptions(domain, wfTypeResetBase+\"-\"+info.WorkflowExecution.RunID)\n\tbaseCtx := workflow.WithChildOptions(ctx, cwo)\n\tbaseFuture := workflow.ExecuteChildWorkflow(baseCtx, wfTypeResetBase, workflow.Now(ctx).UnixNano(), info.WorkflowExecution.ID, info.WorkflowExecution.RunID, domain)\n\tvar baseWE workflow.Execution\n\tif err := baseFuture.GetChildWorkflowExecution().Get(baseCtx, &baseWE); err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tsignalFuture1 := baseFuture.SignalChildWorkflow(baseCtx, signalBeforeReset, \"signalValue\")\n\terr = signalFuture1.Get(baseCtx, nil)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\t// use signal to wait for baseWF to get reach reset point\n\tvar value string\n\tsignalCh := workflow.GetSignalChannel(ctx, signalToTrigger)\n\tsignalCh.Receive(ctx, &value)\n\n\tsignalFuture2 := baseFuture.SignalChildWorkflow(baseCtx, signalAfterReset, \"signalValue\")\n\terr = signalFuture2.Get(baseCtx, nil)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\texpiration := time.Duration(info.ExecutionStartToCloseTimeoutSeconds) * time.Second\n\tactivityCtx := workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tScheduleToStartTimeout: expiration,\n\t\tStartToCloseTimeout:    expiration,\n\t})\n\tvar newWE workflow.Execution\n\terr = workflow.ExecuteActivity(activityCtx, activityTypeTriggerReset, domain, baseWE).Get(activityCtx, &newWE)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\tif err := workflow.Sleep(ctx, time.Duration(bigWait*2)*time.Second); err != nil {\n\t\treturn profile.end(err)\n\t}\n\terr = workflow.ExecuteActivity(activityCtx, activityTypeVerifyReset, domain, newWE).Get(activityCtx, nil)\n\n\treturn profile.end(err)\n}\n\nfunc resetBaseWorkflow(ctx workflow.Context, scheduledTimeNanos int64, parentID, parentRunID, domain string) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeResetBase, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tinfo := workflow.GetInfo(ctx)\n\texpiration := time.Duration(info.ExecutionStartToCloseTimeoutSeconds) * time.Second\n\tretryPolicy := &cadence.RetryPolicy{\n\t\tInitialInterval:    time.Second * 5,\n\t\tBackoffCoefficient: 1,\n\t\tMaximumInterval:    time.Second * 5,\n\t\tExpirationInterval: expiration,\n\t\tMaximumAttempts:    5,\n\t}\n\n\tactivityCtxWithRetry := workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tScheduleToStartTimeout: expiration,\n\t\tStartToCloseTimeout:    expiration,\n\t\tRetryPolicy:            retryPolicy,\n\t})\n\tactivityCtxNoRetry := workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tScheduleToStartTimeout: expiration,\n\t\tStartToCloseTimeout:    expiration,\n\t})\n\n\tvar f workflow.Future\n\tfutures1 := []workflow.Future{}\n\tfutures2 := []workflow.Future{}\n\t// completed before reset\n\tf = workflow.ExecuteActivity(activityCtxWithRetry, activityTypeResetBase, smallWait)\n\tfutures1 = append(futures1, f)\n\n\t// started before reset\n\tf = workflow.ExecuteActivity(activityCtxNoRetry, activityTypeResetBase, bigWait)\n\tfutures2 = append(futures2, f)\n\n\t// not started before reset(because it uses retry)\n\tf = workflow.ExecuteActivity(activityCtxWithRetry, activityTypeResetBase, bigWait)\n\tfutures2 = append(futures2, f)\n\n\t// fired before reset\n\tf = workflow.NewTimer(ctx, time.Duration(smallWait*2)*time.Second)\n\tfutures1 = append(futures1, f)\n\n\t// fired after reset\n\tf = workflow.NewTimer(ctx, time.Duration(bigWait)*time.Second)\n\tfutures2 = append(futures2, f)\n\n\t// wait until first set of futures1 are done: 1 act, 1 timer\n\tfor _, future := range futures1 {\n\t\tif err := future.Get(ctx, nil); err != nil {\n\t\t\treturn profile.end(err)\n\t\t}\n\t}\n\n\tsignalFuture := workflow.SignalExternalWorkflow(ctx, parentID, parentRunID, signalToTrigger, \"signalValue\")\n\terr = signalFuture.Get(ctx, nil)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tvar value string\n\tsignalCh := workflow.GetSignalChannel(ctx, signalBeforeReset)\n\tsignalCh.Receive(ctx, &value)\n\n\tsignalCh = workflow.GetSignalChannel(ctx, signalAfterReset)\n\tsignalCh.Receive(ctx, &value)\n\n\treturn profile.end(err)\n}\n\nfunc triggerResetActivity(ctx context.Context, domain string, baseWE workflow.Execution) (workflow.Execution, error) {\n\tsvClient := getActivityContext(ctx).cadence.Service\n\n\treason := \"reset canary\"\n\n\tclient := getActivityContext(ctx).cadence\n\tscope := activity.GetMetricsScope(ctx)\n\tresetEventID := int64(0)\n\tseenTrigger := false\n\n\t// reset to last decisionCompleted before signalToTrigger was sent\n\t// Since we are in the trigger activity, baseWF must have reached there\n\n\tevents, err := getMyHistory(client, baseWE, scope)\n\tif err != nil {\n\t\treturn workflow.Execution{}, err\n\t}\n\tfor _, event := range events {\n\t\tif event.GetEventType() == shared.EventTypeDecisionTaskCompleted {\n\t\t\tresetEventID = event.GetEventId()\n\t\t}\n\t\tif event.GetEventType() == shared.EventTypeSignalExternalWorkflowExecutionInitiated {\n\t\t\tseenTrigger = true\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif resetEventID == 0 || !seenTrigger {\n\t\treturn workflow.Execution{}, fmt.Errorf(\"something went wrong...base workflow has not reach reset point, %v, %v\", resetEventID, seenTrigger)\n\t}\n\n\treq := &shared.ResetWorkflowExecutionRequest{\n\t\tDomain: &domain,\n\t\tWorkflowExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: &baseWE.ID,\n\t\t\tRunId:      &baseWE.RunID,\n\t\t},\n\t\tReason:                &reason,\n\t\tDecisionFinishEventId: &resetEventID,\n\t\tRequestId:             &baseWE.RunID,\n\t}\n\tresp, err := svClient.ResetWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\treturn workflow.Execution{}, err\n\t}\n\tbaseWE.RunID = *resp.RunId\n\treturn baseWE, nil\n}\n\nfunc verifyResetActivity(ctx context.Context, domain string, newWE workflow.Execution) error {\n\tsvClient := getActivityContext(ctx).cadence.Service\n\n\tresp, err := svClient.DescribeWorkflowExecution(ctx, &shared.DescribeWorkflowExecutionRequest{\n\t\tDomain: &domain,\n\t\tExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: &newWE.ID,\n\t\t\tRunId:      &newWE.RunID,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.WorkflowExecutionInfo.CloseStatus == nil || resp.WorkflowExecutionInfo.GetCloseStatus() != shared.WorkflowExecutionCloseStatusCompleted {\n\t\treturn fmt.Errorf(\"new execution triggered by reset is not completed\")\n\t}\n\treturn nil\n}\n\nfunc resetBaseActivity(ctx context.Context, waitSecs int64) error {\n\ttime.Sleep(time.Second * time.Duration(waitSecs))\n\treturn nil\n}\n"
  },
  {
    "path": "canary/retry.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nvar (\n\terrRetryableActivityError = errors.New(\"retry me\")\n\terrUnexpectedProgress     = errors.New(\"unexpected progress\")\n\terrUnexpectedResult       = errors.New(\"unexpected result\")\n)\n\nfunc init() {\n\tregisterWorkflow(retryWorkflow, wfTypeRetry)\n\tregisterActivity(retryOnTimeoutActivity, activityTypeRetryOnTimeout)\n\tregisterActivity(retryOnFailureActivity, activityTypeRetryOnFailure)\n}\n\nfunc retryWorkflow(ctx workflow.Context, scheduledTimeNanos int64) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeRetry, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tinfo := workflow.GetInfo(ctx)\n\tnow := workflow.Now(ctx).UnixNano()\n\texpiration := time.Duration(info.ExecutionStartToCloseTimeoutSeconds) * time.Second\n\tretryPolicy := &cadence.RetryPolicy{\n\t\tInitialInterval:    time.Second * 5,\n\t\tBackoffCoefficient: 1,\n\t\tMaximumInterval:    time.Second * 5,\n\t\tExpirationInterval: expiration,\n\t\tMaximumAttempts:    5,\n\t}\n\n\tactivityCtx := workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\tTaskList:               taskListName,\n\t\tScheduleToStartTimeout: expiration,\n\t\tStartToCloseTimeout:    expiration,\n\t\tHeartbeatTimeout:       5 * time.Second,\n\t\tRetryPolicy:            retryPolicy,\n\t})\n\n\tf1 := workflow.ExecuteActivity(activityCtx, activityTypeRetryOnTimeout, now)\n\tf2 := workflow.ExecuteActivity(activityCtx, activityTypeRetryOnFailure, now)\n\n\tvar progress int\n\terr = f1.Get(ctx, &progress)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"retryWorkflow failed on RetryOnTimeout\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tif progress < 200 {\n\t\tworkflow.GetLogger(ctx).Error(\"Unexpected activity progress.\", zap.Int(\"Progress\", progress))\n\t\treturn profile.end(errUnexpectedProgress)\n\t}\n\n\tvar result int32\n\terr = f2.Get(ctx, &result)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"retryWorkflow failed on RetryOnFailure\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tif result < 3 {\n\t\tworkflow.GetLogger(ctx).Error(\"Unexpected activity result.\", zap.Int32(\"Result\", result))\n\t\treturn profile.end(errUnexpectedResult)\n\t}\n\n\treturn profile.end(nil)\n\n}\n\nfunc retryOnTimeoutActivity(ctx context.Context, scheduledTimeNanos int64) (int, error) {\n\tvar err error\n\n\tprogress := 0\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\terr = activity.GetHeartbeatDetails(ctx, &progress)\n\t\tif err != nil {\n\t\t\tactivity.GetLogger(ctx).Error(\"GetProgress failed.\", zap.Error(err))\n\t\t\treturn 0, err\n\t\t}\n\t}\n\n\tinfo := activity.GetInfo(ctx)\n\tif info.Attempt < 3 {\n\t\tactivity.RecordHeartbeat(ctx, info.Attempt*100)\n\t\ttime.Sleep(2 * info.HeartbeatTimeout)\n\t}\n\n\treturn progress, nil\n}\n\nfunc retryOnFailureActivity(ctx context.Context, scheduledTimeNanos int64) (int32, error) {\n\tinfo := activity.GetInfo(ctx)\n\tif info.Attempt < 3 {\n\t\treturn 0, errRetryableActivityError\n\t}\n\n\treturn info.Attempt, nil\n}\n"
  },
  {
    "path": "canary/runner.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"crypto/tls\"\n\t\"crypto/x509\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"sync\"\n\t\"time\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/compatibility\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer\"\n\t\"go.uber.org/yarpc/peer/hostport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\t\"go.uber.org/zap\"\n\t\"google.golang.org/grpc/credentials\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype canaryRunner struct {\n\t*RuntimeContext\n\tconfig *Canary\n}\n\n// NewCanaryRunner creates and returns a runnable which spins\n// up a set of canaries based on supplied config\nfunc NewCanaryRunner(cfg *Config) (Runnable, error) {\n\tlogger, err := cfg.Log.NewZapLogger()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create logger: %v\", err)\n\t}\n\n\tmetricsScope := cfg.Metrics.NewScope(log.NewLogger(logger), \"cadence-canary\")\n\n\tif cfg.Cadence.ServiceName == \"\" {\n\t\tcfg.Cadence.ServiceName = CadenceServiceName\n\t}\n\n\tvar dispatcher *yarpc.Dispatcher\n\tvar runtimeContext *RuntimeContext\n\tif cfg.Cadence.GRPCHostNameAndPort != \"\" {\n\t\tvar outbounds transport.Outbounds\n\t\tif cfg.Cadence.TLSCAFile != \"\" {\n\t\t\tcaCert, err := ioutil.ReadFile(cfg.Cadence.TLSCAFile)\n\t\t\tif err != nil {\n\t\t\t\tlogger.Fatal(\"Failed to load server CA certificate\", zap.Error(err))\n\t\t\t}\n\t\t\tcaCertPool := x509.NewCertPool()\n\t\t\tif !caCertPool.AppendCertsFromPEM(caCert) {\n\t\t\t\tlogger.Fatal(\"Failed to add server CA certificate\", zap.Error(err))\n\t\t\t}\n\t\t\ttlsConfig := tls.Config{\n\t\t\t\tRootCAs: caCertPool,\n\t\t\t}\n\t\t\ttlsCreds := credentials.NewTLS(&tlsConfig)\n\t\t\tgrpcTransport := grpc.NewTransport()\n\t\t\ttlsChooser := peer.NewSingle(hostport.Identify(cfg.Cadence.GRPCHostNameAndPort), grpcTransport.NewDialer(grpc.DialerCredentials(tlsCreds)))\n\t\t\toutbounds = transport.Outbounds{Unary: grpcTransport.NewOutbound(tlsChooser)}\n\t\t} else {\n\t\t\toutbounds = transport.Outbounds{Unary: grpc.NewTransport().NewSingleOutbound(cfg.Cadence.GRPCHostNameAndPort)}\n\t\t}\n\t\tdispatcher = yarpc.NewDispatcher(yarpc.Config{\n\t\t\tName: CanaryServiceName,\n\t\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\tcfg.Cadence.ServiceName: outbounds,\n\t\t\t},\n\t\t})\n\t\tclientConfig := dispatcher.ClientConfig(cfg.Cadence.ServiceName)\n\t\truntimeContext = NewRuntimeContext(\n\t\t\tlogger,\n\t\t\tmetricsScope,\n\t\t\tcompatibility.NewThrift2ProtoAdapter(\n\t\t\t\tapiv1.NewDomainAPIYARPCClient(clientConfig),\n\t\t\t\tapiv1.NewWorkflowAPIYARPCClient(clientConfig),\n\t\t\t\tapiv1.NewWorkerAPIYARPCClient(clientConfig),\n\t\t\t\tapiv1.NewVisibilityAPIYARPCClient(clientConfig),\n\t\t\t),\n\t\t)\n\t} else if cfg.Cadence.ThriftHostNameAndPort != \"\" {\n\t\ttch, err := tchannel.NewChannelTransport(\n\t\t\ttchannel.ServiceName(CanaryServiceName),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to create transport channel: %v\", err)\n\t\t}\n\t\tdispatcher = yarpc.NewDispatcher(yarpc.Config{\n\t\t\tName: CanaryServiceName,\n\t\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\tcfg.Cadence.ServiceName: {Unary: tch.NewSingleOutbound(cfg.Cadence.ThriftHostNameAndPort)},\n\t\t\t},\n\t\t})\n\t\truntimeContext = NewRuntimeContext(\n\t\t\tlogger,\n\t\t\tmetricsScope,\n\t\t\tworkflowserviceclient.New(dispatcher.ClientConfig(cfg.Cadence.ServiceName)),\n\t\t)\n\t} else {\n\t\treturn nil, fmt.Errorf(\"must specify either gRPC address(address) or Thrift address (host) in the config\")\n\t}\n\n\tif err := dispatcher.Start(); err != nil {\n\t\tdispatcher.Stop()\n\t\treturn nil, fmt.Errorf(\"failed to create outbound transport channel: %v\", err)\n\t}\n\n\treturn &canaryRunner{\n\t\tRuntimeContext: runtimeContext,\n\t\tconfig:         &cfg.Canary,\n\t}, nil\n}\n\n// Run runs the canaries\nfunc (r *canaryRunner) Run(mode string) error {\n\tr.metrics.Counter(\"restarts\").Inc(1)\n\tif len(r.config.Excludes) != 0 {\n\t\tupdateSanityChildWFList(r.config.Excludes)\n\t}\n\n\tif r.config.Cron.CronSchedule == \"\" {\n\t\tr.config.Cron.CronSchedule = \"@every 30s\"\n\t}\n\tif r.config.Cron.CronExecutionTimeout == 0 {\n\t\tr.config.Cron.CronExecutionTimeout = 18 * time.Minute\n\t}\n\tif r.config.Cron.StartJobTimeout == 0 {\n\t\tr.config.Cron.StartJobTimeout = 9 * time.Minute\n\t}\n\n\tvar wg sync.WaitGroup\n\tfor _, d := range r.config.Domains {\n\t\tcanary := newCanary(d, r.RuntimeContext, r.config)\n\t\tr.logger.Info(\"starting canary\", zap.String(\"domain\", d))\n\t\tr.execute(canary, mode, &wg)\n\t}\n\twg.Wait()\n\treturn nil\n}\n\nfunc (r *canaryRunner) execute(task Runnable, mode string, wg *sync.WaitGroup) {\n\twg.Add(1)\n\tgo func() {\n\t\ttask.Run(mode)\n\t\twg.Done()\n\t}()\n}\n\nfunc updateSanityChildWFList(excludes []string) {\n\tvar temp []string\n\tfor _, childName := range sanityChildWFList {\n\t\tif !isStringInList(childName, excludes) {\n\t\t\ttemp = append(temp, childName)\n\t\t}\n\t}\n\tsanityChildWFList = temp\n}\n\nfunc isStringInList(str string, list []string) bool {\n\tfor _, l := range list {\n\t\tif l == str {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "canary/sanity.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"fmt\"\n\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/multierr\"\n\t\"go.uber.org/zap\"\n)\n\n// sanityChildWFList is the list of tests / child workflows invoked by the sanity canary\nvar sanityChildWFList = []string{\n\twfTypeEcho,\n\twfTypeSignal,\n\twfTypeVisibility,\n\twfTypeSearchAttributes,\n\twfTypeConcurrentExec,\n\twfTypeQuery,\n\twfTypeTimeout,\n\twfTypeLocalActivity,\n\twfTypeCancellation,\n\twfTypeRetry,\n\twfTypeReset,\n\twfTypeHistoryArchival,\n\twfTypeVisibilityArchival,\n\twfTypeBatch,\n\twfTypeCrossClusterParent,\n}\n\nfunc init() {\n\tregisterWorkflow(sanityWorkflow, wfTypeSanity)\n}\n\n// sanityWorkflow represents a canary implementation that tests the\n// sanity of a cadence cluster - it performs probes / tests that\n// exercises the frontend APIs and the basic functionality of the\n// client library - each probe / test MUST be implemented as a\n// child workflow of this workflow\nfunc sanityWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tvar err error\n\tprofile, err := beginWorkflow(ctx, wfTypeSanity, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tchildNames, err := getChildWorkflowNames(ctx)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tselector, resultC := forkChildWorkflows(ctx, domain, childNames)\n\terr = joinChildWorkflows(ctx, childNames, selector, resultC)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"sanity workflow failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tworkflow.GetLogger(ctx).Info(\"sanity workflow finished successfully\")\n\treturn profile.end(err)\n}\n\n// forkChildWorkflows spawns child workflows with the given names\n// this method assumes that all child workflows have the same method signature\nfunc forkChildWorkflows(ctx workflow.Context, domain string, names []string) (workflow.Selector, workflow.Channel) {\n\tnow := workflow.Now(ctx).UnixNano()\n\tselector := workflow.NewSelector(ctx)\n\tresultC := workflow.NewBufferedChannel(ctx, len(names))\n\n\tmyID := workflow.GetInfo(ctx).WorkflowExecution.ID\n\tfor _, childName := range names {\n\t\tcwo := newChildWorkflowOptions(domain, concat(myID, childName))\n\t\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\t\tfuture := workflow.ExecuteChildWorkflow(childCtx, childName, now)\n\t\tselector.AddFuture(future, func(f workflow.Future) {\n\t\t\tif err := f.Get(ctx, nil); err != nil {\n\t\t\t\tworkflow.GetLogger(ctx).Error(\"child workflow failed\", zap.Error(err))\n\t\t\t\terr = fmt.Errorf(\"child workflow %s failed: %w\", childName, err)\n\t\t\t\tresultC.Send(ctx, err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tresultC.Send(ctx, nil)\n\t\t})\n\t}\n\n\treturn selector, resultC\n}\n\n// joinChildWorkflows waits for all the given child workflows to complete\n// returns error even if atleast one of the child workflow returns an error\nfunc joinChildWorkflows(ctx workflow.Context, names []string, selector workflow.Selector, resultC workflow.Channel) error {\n\tvar err error\n\tfor i := 0; i < len(names); i++ {\n\t\tselector.Select(ctx)\n\t\tvar err1 error\n\t\tresultC.Receive(ctx, &err1)\n\t\terr = multierr.Append(err, err1)\n\t}\n\treturn err\n}\n\n// getChildWorkflowNames exist mainly to make sure replays result in\n// deterministic behavior for the workflow - the set of child workflows\n// to be spawned are recorded in history as a side effect and result\n// from the side effect is returned as a result\nfunc getChildWorkflowNames(ctx workflow.Context) ([]string, error) {\n\tvar names []string\n\terr := workflow.SideEffect(ctx, func(workflow.Context) interface{} { return sanityChildWFList }).Get(&names)\n\treturn names, err\n}\n"
  },
  {
    "path": "canary/searchAttributes.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\ttimeToWaitVisibilityDataOnESInDuration = 2 * time.Second\n)\n\nfunc init() {\n\tregisterWorkflow(searchAttributesWorkflow, wfTypeSearchAttributes)\n\tregisterActivity(searchAttributesActivity, activityTypeSearchAttributes)\n}\n\n// searchAttributesWorkflow tests the search attributes apis\nfunc searchAttributesWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tvar err error\n\tprofile, err := beginWorkflow(ctx, wfTypeSearchAttributes, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tattr := map[string]interface{}{\n\t\t\"CustomKeywordField\": \"canaryTest\",\n\t}\n\terr = workflow.UpsertSearchAttributes(ctx, attr)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"searchAttributes test failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\texecInfo := workflow.GetInfo(ctx).WorkflowExecution\n\taCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\t// wait for visibility on ES\n\tif err := workflow.Sleep(ctx, timeToWaitVisibilityDataOnESInDuration); err != nil {\n\t\treturn profile.end(err)\n\t}\n\tnow := workflow.Now(ctx).UnixNano()\n\terr = workflow.ExecuteActivity(aCtx, activityTypeSearchAttributes, now, execInfo).Get(ctx, nil)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"searchAttributes test failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\treturn profile.end(nil)\n}\n\n// searchAttributesActivity exercises the visibility apis\nfunc searchAttributesActivity(ctx context.Context, scheduledTimeNanos int64, parentInfo workflow.Execution) error {\n\tvar err error\n\tscope := activity.GetMetricsScope(ctx)\n\tscope, sw := recordActivityStart(scope, activityTypeSearchAttributes, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\tif err := listWorkflow(client, parentInfo.ID, scope); err != nil {\n\t\treturn err\n\t}\n\tif _, err := getMyHistory(client, parentInfo, scope); err != nil {\n\t\treturn err\n\t}\n\treturn err\n}\n\nfunc listWorkflow(client cadenceClient, wfID string, scope tally.Scope) error {\n\tpageSz := int32(1)\n\tstartTime := time.Now().UnixNano() - int64(timeSkewToleranceDuration)\n\tendTime := time.Now().UnixNano() + int64(timeSkewToleranceDuration)\n\tqueryStr := fmt.Sprintf(\"WorkflowID = '%s' and CustomKeywordField = '%s' and StartTime between %d and %d\",\n\t\twfID, \"canaryTest\", startTime, endTime)\n\trequest := &shared.ListWorkflowExecutionsRequest{\n\t\tPageSize: &pageSz,\n\t\tQuery:    &queryStr,\n\t}\n\n\tscope.Counter(listWorkflowsCount).Inc(1)\n\tsw := scope.Timer(listWorkflowsLatency).Start()\n\tresp, err := client.ListWorkflow(context.Background(), request)\n\tsw.Stop()\n\tif err != nil {\n\t\tscope.Counter(listWorkflowsFailureCount).Inc(1)\n\t\treturn err\n\t}\n\n\tif len(resp.Executions) != 1 {\n\t\tscope.Counter(listWorkflowsFailureCount).Inc(1)\n\t\terr := fmt.Errorf(\"listWorkflow returned %d executions, expected=1\", len(resp.Executions))\n\t\treturn err\n\t}\n\n\tid := resp.Executions[0].Execution.GetWorkflowId()\n\tif id != wfID {\n\t\tscope.Counter(listWorkflowsFailureCount).Inc(1)\n\t\terr := fmt.Errorf(\"listWorkflow returned wrong workflow id %v\", id)\n\t\treturn err\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "canary/signal.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\tsignalName  = \"signal-name\"\n\tsignalValue = \"canary.signal\"\n)\n\nfunc init() {\n\tregisterWorkflow(signalWorkflow, wfTypeSignal)\n\tregisterWorkflow(signalExternalWorkflow, wfTypeSignalExternal)\n\tregisterActivity(signalActivity, activityTypeSignal)\n}\n\n// signalWorkflow is the workflow implementation to test SignalWorkflowExecution API\n// it simply spins up an activity which sends a signal to the parent\nfunc signalWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\tdomain := workflow.GetInfo(ctx).Domain\n\n\tvar err error\n\tprofile, err := beginWorkflow(ctx, wfTypeSignal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecInfo := workflow.GetInfo(ctx).WorkflowExecution\n\tsigName := fmt.Sprintf(\"sig.%v\", execInfo.RunID)\n\tsigCh := workflow.GetSignalChannel(ctx, sigName)\n\n\taCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\terr = workflow.ExecuteActivity(aCtx, activityTypeSignal, workflow.Now(ctx).UnixNano(), execInfo, sigName).Get(ctx, nil)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"signal activity failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\tvar sigValue string\n\tsigCh.Receive(ctx, &sigValue)\n\tif sigValue != signalValue {\n\t\tworkflow.GetLogger(ctx).Error(\"wrong signal value received\", zap.String(\"value\", sigValue))\n\t\treturn profile.end(errors.New(\"invalid signal value\"))\n\t}\n\tsigCh.Receive(ctx, &sigValue)\n\tif sigValue != signalValue {\n\t\tworkflow.GetLogger(ctx).Error(\"wrong signal value received\", zap.String(\"value\", sigValue))\n\t\treturn profile.end(errors.New(\"invalid signal value\"))\n\t}\n\n\tcwo := newChildWorkflowOptions(domain, wfTypeSignalExternal+\"-child\")\n\tchildCtx := workflow.WithChildOptions(ctx, cwo)\n\tchildFuture := workflow.ExecuteChildWorkflow(childCtx, wfTypeSignalExternal, workflow.Now(ctx).UnixNano(), sleepDuration)\n\tsignalFuture := childFuture.SignalChildWorkflow(childCtx, signalName, signalValue)\n\terr = signalFuture.Get(childCtx, nil)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tdecisionSignalTestFn := func(useRunID bool) error {\n\t\tworkflowIDSuffix := \"-with-run-ID\"\n\t\tif !useRunID {\n\t\t\tworkflowIDSuffix = \"-without-run-ID\"\n\t\t}\n\t\tcwo = newChildWorkflowOptions(domain, wfTypeSignalExternal+workflowIDSuffix)\n\t\tchildCtx = workflow.WithChildOptions(ctx, cwo)\n\t\tchildFuture = workflow.ExecuteChildWorkflow(childCtx, wfTypeSignalExternal, workflow.Now(ctx).UnixNano(), sleepDuration)\n\t\tchildExecution := &workflow.Execution{}\n\t\terr = childFuture.GetChildWorkflowExecution().Get(childCtx, childExecution)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !useRunID {\n\t\t\tchildExecution.RunID = \"\"\n\t\t}\n\t\tsignalFuture = workflow.SignalExternalWorkflow(ctx, childExecution.ID, childExecution.RunID, signalName, signalValue)\n\t\terr = signalFuture.Get(ctx, nil)\n\t\tif err != nil {\n\t\t\tworkflow.GetLogger(ctx).Info(\"signalWorkflow failed: fail to send siganl to child workflow\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\terr = childFuture.Get(childCtx, nil)\n\t\tif err != nil {\n\t\t\tworkflow.GetLogger(ctx).Info(\"signalWorkflow failed: child workflow return err\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\terr = decisionSignalTestFn(false)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\terr = decisionSignalTestFn(true)\n\treturn profile.end(err)\n}\n\n// signalActivity sends a signal to the parent workflow\n// that created this activity\nfunc signalActivity(ctx context.Context, scheduledTimeNanos int64, execInfo workflow.Execution, signalName string) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeSignal, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\terr = client.SignalWorkflow(context.Background(), execInfo.ID, execInfo.RunID, signalName, signalValue)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = client.SignalWorkflow(context.Background(), execInfo.ID, \"\", signalName, signalValue)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// signalExternalWorkflow receive a signal from the parent workflow\nfunc signalExternalWorkflow(ctx workflow.Context, scheduledTimeNanos int64) error {\n\tprofile, err := beginWorkflow(ctx, wfTypeSignalExternal, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn profile.end(err)\n\t}\n\n\tvar value string\n\tsignalCh := workflow.GetSignalChannel(ctx, signalName)\n\tsignalCh.Receive(ctx, &value)\n\tif value != signalValue {\n\t\tworkflow.GetLogger(ctx).Error(\"wrong signal value received\", zap.String(\"value\", value))\n\t\treturn profile.end(errors.New(\"invalid signal value\"))\n\t}\n\n\treturn profile.end(nil)\n}\n"
  },
  {
    "path": "canary/timeout.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\tactivityDelay                           = 4 * time.Second\n\tactivityStartToCloseTimeoutThreshold    = 1 * time.Second\n\tactivityScheduleToStartTimeoutThreshold = 1 * time.Second\n\tactivityScheduleToCloseTimeoutThreshold = activityStartToCloseTimeoutThreshold + activityScheduleToStartTimeoutThreshold\n)\n\nfunc init() {\n\tregisterWorkflow(timeoutWorkflow, wfTypeTimeout)\n\tregisterActivity(timeoutActivity, activityTypeTimeout)\n}\n\n// timeoutWorkflow is the workflow implementation to test for querying workflow status\nfunc timeoutWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeConcurrentExec, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tnow := workflow.Now(ctx).UnixNano()\n\tactivityOptions := newActivityOptions()\n\tactivityOptions.StartToCloseTimeout = activityStartToCloseTimeoutThreshold\n\tactivityOptions.ScheduleToStartTimeout = activityScheduleToStartTimeoutThreshold\n\tactivityOptions.ScheduleToCloseTimeout = activityScheduleToCloseTimeoutThreshold\n\tactivityCtx := workflow.WithActivityOptions(ctx, activityOptions)\n\tactivityFuture := workflow.ExecuteActivity(activityCtx, timeoutActivity, now)\n\n\tactivityErr := activityFuture.Get(ctx, nil)\n\tif activityErr != nil {\n\t\tif _, ok := activityErr.(*workflow.TimeoutError); !ok {\n\t\t\tworkflow.GetLogger(ctx).Info(\"activity timeout failed\", zap.Error(activityErr))\n\t\t} else {\n\t\t\tactivityErr = nil\n\t\t}\n\t} else {\n\t\tactivityErr = fmt.Errorf(\"activity timeout error expected, but no error received\")\n\t\tworkflow.GetLogger(ctx).Info(\"activity timeout failed\", zap.Error(activityErr))\n\t}\n\n\treturn profile.end(activityErr)\n}\n\n// timeoutActivity is the activity implementation for timeout test\nfunc timeoutActivity(ctx context.Context, scheduledTimeNanos int64) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeConcurrentExec, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\ttimer := time.NewTimer(activityDelay)\n\t<-timer.C\n\ttimer.Stop()\n\n\treturn nil\n}\n"
  },
  {
    "path": "canary/visibility.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\nconst (\n\ttimeSkewToleranceDuration = 5 * time.Minute\n)\n\nfunc init() {\n\tregisterWorkflow(visibilityWorkflow, wfTypeVisibility)\n\tregisterActivity(visibilityActivity, activityTypeVisibility)\n}\n\n// visibilityWorkflow tests the visibility apis\nfunc visibilityWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tvar err error\n\tprofile, err := beginWorkflow(ctx, wfTypeVisibility, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecInfo := workflow.GetInfo(ctx).WorkflowExecution\n\taCtx := workflow.WithActivityOptions(ctx, newActivityOptions())\n\t// wait for visibility on ES {}\n\tif err := workflow.Sleep(ctx, 2*time.Second); err != nil {\n\t\treturn profile.end(err)\n\t}\n\tnow := workflow.Now(ctx).UnixNano()\n\terr = workflow.ExecuteActivity(aCtx, activityTypeVisibility, now, execInfo).Get(ctx, nil)\n\tif err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"visibility test failed\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\treturn profile.end(nil)\n}\n\n// visibilityActivity exercises the visibility apis\nfunc visibilityActivity(ctx context.Context, scheduledTimeNanos int64, parentInfo workflow.Execution) error {\n\tvar err error\n\tscope := activity.GetMetricsScope(ctx)\n\tscope, sw := recordActivityStart(scope, activityTypeVisibility, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityContext(ctx).cadence\n\tif err := listMyWorkflow(client, parentInfo.ID, scope); err != nil {\n\t\treturn err\n\t}\n\tif _, err := getMyHistory(client, parentInfo, scope); err != nil {\n\t\treturn err\n\t}\n\treturn err\n}\n\nfunc listMyWorkflow(client cadenceClient, wfID string, scope tally.Scope) error {\n\tpageSz := int32(1)\n\tstartTime := time.Now().UnixNano() - int64(timeSkewToleranceDuration)\n\tendTime := time.Now().UnixNano() + int64(timeSkewToleranceDuration)\n\trequest := &shared.ListOpenWorkflowExecutionsRequest{\n\t\tMaximumPageSize: &pageSz,\n\t\tExecutionFilter: &shared.WorkflowExecutionFilter{WorkflowId: &wfID},\n\t\tStartTimeFilter: &shared.StartTimeFilter{\n\t\t\tEarliestTime: &startTime,\n\t\t\tLatestTime:   &endTime,\n\t\t},\n\t}\n\n\tscope.Counter(listOpenWorkflowsCount).Inc(1)\n\tsw := scope.Timer(listOpenWorkflowsLatency).Start()\n\tresp, err := client.ListOpenWorkflow(context.Background(), request)\n\tsw.Stop()\n\tif err != nil {\n\t\tscope.Counter(listOpenWorkflowsFailureCount).Inc(1)\n\t\treturn err\n\t}\n\n\tif len(resp.Executions) != 1 {\n\t\tscope.Counter(listOpenWorkflowsFailureCount).Inc(1)\n\t\terr := fmt.Errorf(\"listOpenWorkflow returned %d executions, expected=1\", len(resp.Executions))\n\t\treturn err\n\t}\n\n\tid := resp.Executions[0].Execution.GetWorkflowId()\n\tif id != wfID {\n\t\tscope.Counter(listOpenWorkflowsFailureCount).Inc(1)\n\t\terr := fmt.Errorf(\"listOpenWorkflow returned wrong workflow id %v\", id)\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc getMyHistory(client cadenceClient, execInfo workflow.Execution, scope tally.Scope) ([]*shared.HistoryEvent, error) {\n\tscope.Counter(getWorkflowHistoryCount).Inc(1)\n\tsw := scope.Timer(getWorkflowHistoryLatency).Start()\n\tdefer sw.Stop()\n\n\tevents := []*shared.HistoryEvent{}\n\titer := client.GetWorkflowHistory(context.Background(), execInfo.ID, execInfo.RunID, false, shared.HistoryEventFilterTypeAllEvent)\n\n\tfor iter.HasNext() {\n\t\tevent, err := iter.Next()\n\t\tif err != nil {\n\t\t\tscope.Counter(getWorkflowHistoryFailureCount).Inc(1)\n\t\t\treturn nil, err\n\t\t}\n\t\tevents = append(events, event)\n\t}\n\n\tif len(events) == 0 {\n\t\treturn nil, errors.New(\"getWorkflowHistory returned history with 0 events\")\n\t}\n\treturn events, nil\n}\n"
  },
  {
    "path": "canary/visibilityArchival.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n)\n\ntype (\n\tvisibilityArchivalValidator interface {\n\t\tshouldRun() bool\n\t\tgetQuery(workflowID, runID, workflowType string, startTime, closeTime time.Time) string\n\t\tvalidateExecutions([]*shared.WorkflowExecutionInfo) error\n\t}\n\n\tfilestoreVisibilityArchivalValidator struct {\n\t\texpectedRunID string\n\t}\n)\n\nconst (\n\tschemeFilestore = \"file\"\n\n\tqueryPageSize = 100\n)\n\nfunc init() {\n\tregisterWorkflow(visibilityArchivalWorkflow, wfTypeVisibilityArchival)\n\tregisterActivity(visibilityArchivalActivity, activityTypeVisibilityArchival)\n}\n\nfunc visibilityArchivalWorkflow(ctx workflow.Context, inputScheduledTimeNanos int64) error {\n\tscheduledTimeNanos := getScheduledTimeFromInputIfNonZero(ctx, inputScheduledTimeNanos)\n\n\tprofile, err := beginWorkflow(ctx, wfTypeVisibilityArchival, scheduledTimeNanos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, newActivityOptions()),\n\t\tactivityTypeVisibilityArchival,\n\t\tworkflow.Now(ctx).UnixNano(),\n\t).Get(ctx, nil); err != nil {\n\t\tworkflow.GetLogger(ctx).Error(\"failed to list archived workflows\", zap.Error(err))\n\t\treturn profile.end(err)\n\t}\n\n\treturn profile.end(nil)\n}\n\nfunc visibilityArchivalActivity(ctx context.Context, scheduledTimeNanos int64) error {\n\tscope := activity.GetMetricsScope(ctx)\n\tvar err error\n\tscope, sw := recordActivityStart(scope, activityTypeVisibilityArchival, scheduledTimeNanos)\n\tdefer recordActivityEnd(scope, sw, err)\n\n\tclient := getActivityArchivalContext(ctx).cadence\n\tresp, err := client.Describe(ctx, archivalDomain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif resp.Configuration != nil &&\n\t\tresp.Configuration.GetVisibilityArchivalStatus() == shared.ArchivalStatusDisabled {\n\t\treturn errors.New(\"domain not configured for visibility archival\")\n\t}\n\n\tvisArchivalURI := \"\"\n\tif resp.Configuration != nil {\n\t\tvisArchivalURI = resp.Configuration.GetVisibilityArchivalURI()\n\t}\n\n\tvar validator visibilityArchivalValidator\n\tscheme := getURIScheme(visArchivalURI)\n\tswitch scheme {\n\tcase schemeFilestore:\n\t\tvalidator = newFilestoreVisibilityArchivalValidator()\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown visibility archival scheme: %s\", scheme)\n\t}\n\n\tif !validator.shouldRun() {\n\t\treturn nil\n\t}\n\n\tstartTime := time.Now()\n\texecution, err := executeArchivalExeternalWorkflow(ctx, client, startTime.UnixNano())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tquery := validator.getQuery(\n\t\texecution.GetWorkflowId(),\n\t\texecution.GetRunId(),\n\t\twfTypeArchivalExternal,\n\t\tstartTime,\n\t\ttime.Now(),\n\t)\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn ctx.Err()\n\t\tdefault:\n\t\t}\n\n\t\texecutions, err := listarchivedWorkflow(\n\t\t\tctx,\n\t\t\tscope,\n\t\t\tclient,\n\t\t\t&shared.ListArchivedWorkflowExecutionsRequest{\n\t\t\t\tDomain:   stringPtr(archivalDomain),\n\t\t\t\tPageSize: int32Ptr(queryPageSize),\n\t\t\t\tQuery:    stringPtr(query),\n\t\t\t},\n\t\t)\n\t\tif err != nil && isBadRequestError(err) {\n\t\t\treturn err\n\t\t}\n\n\t\tif err == nil && validator.validateExecutions(executions) == nil {\n\t\t\treturn nil\n\t\t}\n\n\t\t<-time.After(5 * time.Second)\n\t}\n}\n\nfunc listarchivedWorkflow(\n\tctx context.Context,\n\tscope tally.Scope,\n\tclient cadenceClient,\n\trequest *shared.ListArchivedWorkflowExecutionsRequest,\n) ([]*shared.WorkflowExecutionInfo, error) {\n\tvar executions []*shared.WorkflowExecutionInfo\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn nil, ctx.Err()\n\t\tdefault:\n\t\t}\n\n\t\tscope.Counter(listArchivedWorkflowCount).Inc(1)\n\t\tsw := scope.Timer(listArchivedWorkflowsLatency).Start()\n\t\tresponse, err := client.ListArchivedWorkflow(ctx, request)\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tscope.Counter(listArchivedWorkflowFailureCount).Inc(1)\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif len(response.Executions) != 0 {\n\t\t\texecutions = append(executions, response.Executions...)\n\t\t}\n\n\t\tif response.NextPageToken == nil {\n\t\t\tbreak\n\t\t}\n\n\t\trequest.NextPageToken = response.NextPageToken\n\t}\n\n\treturn executions, nil\n}\n\nfunc newFilestoreVisibilityArchivalValidator() visibilityArchivalValidator {\n\treturn &filestoreVisibilityArchivalValidator{}\n}\n\nfunc (v *filestoreVisibilityArchivalValidator) shouldRun() bool {\n\treturn true\n}\n\nfunc (v *filestoreVisibilityArchivalValidator) getQuery(\n\tworkflowID, runID, workflowType string,\n\tstartTime, closeTime time.Time,\n) string {\n\tv.expectedRunID = runID\n\treturn fmt.Sprintf(\n\t\t\"WorkflowType = '%s' and WorkflowID = '%s' and CloseTime >= %v and CloseTime <= %v\",\n\t\tworkflowType,\n\t\tworkflowID,\n\t\tstartTime.UnixNano(),\n\t\tcloseTime.UnixNano(),\n\t)\n}\n\nfunc (v *filestoreVisibilityArchivalValidator) validateExecutions(\n\texecutions []*shared.WorkflowExecutionInfo,\n) error {\n\tif len(executions) != 1 {\n\t\treturn fmt.Errorf(\"listarchivedWorkflow returned %d executions, expecting 1\", len(executions))\n\t}\n\n\trunID := executions[0].Execution.GetRunId()\n\tif runID != v.expectedRunID {\n\t\treturn fmt.Errorf(\"listarchivedWorkflow returned wrong runID %v, expecting %s\", runID, v.expectedRunID)\n\t}\n\n\treturn nil\n}\n\nfunc getURIScheme(URI string) string {\n\tif idx := strings.Index(URI, \"://\"); idx != -1 {\n\t\treturn URI[:idx]\n\t}\n\treturn \"\"\n}\n\nfunc isBadRequestError(\n\terr error,\n) bool {\n\t_, ok := err.(*shared.BadRequestError)\n\treturn ok\n}\n"
  },
  {
    "path": "canary/workflow_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage canary\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/mocks\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\ntype (\n\tworkflowTestSuite struct {\n\t\tsuite.Suite\n\t\ttestsuite.WorkflowTestSuite\n\t\tenv *testsuite.TestWorkflowEnvironment\n\t}\n\n\tmockCadenceClient struct {\n\t\tclient       *mocks.Client\n\t\tdomainClient *mocks.DomainClient\n\t}\n)\n\nfunc TestWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(workflowTestSuite))\n}\n\nfunc (s *workflowTestSuite) SetupTest() {\n\ts.SetLogger(testlogger.NewZap(s.T()))\n\ts.env = s.NewTestWorkflowEnvironment()\n\ts.env.Test(s.T())\n}\n\nfunc (s *workflowTestSuite) TearDownTest() {\n\ts.env.AssertExpectations(s.T())\n}\n\nfunc (s *workflowTestSuite) TestConcurrentExecWorkflow() {\n\ts.env.SetWorkerOptions(newTestWorkerOptions(newMockActivityContext(newMockCadenceClient())))\n\ts.env.ExecuteWorkflow(wfTypeConcurrentExec, time.Now().UnixNano(), \"\")\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowTestSuite) TestQueryWorkflow() {\n\t// DOTO implement this test once client test framework\n\t// SetOnActivityStartedListener is guaranteed to\n\t// be executed and finished before activity started\n}\n\nfunc (s *workflowTestSuite) TestTimeout() {\n\t// DOTO implement this test once client test framework\n\t// can handle timeout error correctly\n}\n\nfunc (s *workflowTestSuite) TestSignalWorkflow() {\n\tmockClient := newMockCadenceClient()\n\tmockClient.client.On(\"SignalWorkflow\",\n\t\tmock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Run(func(args mock.Arguments) {\n\t\targ3 := args.Get(3).(string)\n\t\targ4 := args.Get(4).(string)\n\t\ts.env.SignalWorkflow(arg3, arg4)\n\t}).Return(nil)\n\ts.env.SetWorkerOptions(newTestWorkerOptions(newMockActivityContext(mockClient)))\n\ts.env.ExecuteWorkflow(wfTypeSignal, time.Now().UnixNano(), \"\")\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowTestSuite) TestEchoWorkflow() {\n\ts.env.SetWorkerOptions(newTestWorkerOptions(newMockActivityContext(newMockCadenceClient())))\n\ts.env.ExecuteWorkflow(wfTypeEcho, time.Now().UnixNano(), \"\")\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowTestSuite) TestVisibilityWorkflow() {\n\t// uncomment below when race condition on s.env.SetOnActivityStartedListener is resolved\n\n\t// mockClient := newMockCadenceClient()\n\t// // setup the mock for visibility apis after the activity is invoked, because\n\t// // we need the workflow id and run id to construct a response for the mock\n\t// s.env.SetOnActivityStartedListener(func(activityInfo *cadence.ActivityInfo, ctx context.Context, args cadence.EncodedValues) {\n\t// \tif activityInfo.ActivityType.Name != activityTypeVisibility {\n\t// \t\treturn\n\t// \t}\n\t// \tresp := newMockOpenWorkflowResponse(activityInfo.WorkflowExecution.ID, activityInfo.WorkflowExecution.RunID)\n\t// \tmockClient.client.On(\"ListOpenWorkflow\", mock.Anything, mock.Anything).Return(resp, nil)\n\t// \thistory := &shared.History{Events: []*shared.HistoryEvent{{}}}\n\t// \tmockClient.client.On(\"GetWorkflowHistory\", mock.Anything, mock.Anything, mock.Anything).Return(history, nil)\n\t// })\n\t// s.env.SetWorkerOptions(newTestWorkerOptions(newMockActivityContext(mockClient)))\n\t// s.env.ExecuteWorkflow(wfTypeVisibility, time.Now().UnixNano(), &ServiceConfig{})\n\t// s.True(s.env.IsWorkflowCompleted())\n\t// s.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowTestSuite) TestSanityWorkflow() {\n\toldWFList := sanityChildWFList\n\tsanityChildWFList = []string{wfTypeEcho}\n\ts.env.ExecuteWorkflow(wfTypeSanity, time.Now().UnixNano(), \"\")\n\tsanityChildWFList = oldWFList\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowTestSuite) TestLocalActivityWorkflow() {\n\ts.env.OnActivity(getConditionData).Return(int32(20), nil).Once()\n\ts.env.ExecuteWorkflow(localActivityWorkfow, time.Now().UnixNano())\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n\n\tvar result string\n\terr := s.env.GetWorkflowResult(&result)\n\ts.NoError(err)\n\ts.Equal(\"data%2 == 0 and data%5 == 0\", result)\n}\n\nfunc newTestWorkerOptions(ctx *activityContext) worker.Options {\n\treturn worker.Options{\n\t\tMetricsScope:              tally.NoopScope,\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), ctxKeyActivityRuntime, ctx),\n\t}\n}\n\nfunc newMockActivityContext(client mockCadenceClient) *activityContext {\n\treturn &activityContext{\n\t\tcadence: cadenceClient{\n\t\t\tClient:       client.client,\n\t\t\tDomainClient: client.domainClient,\n\t\t},\n\t}\n}\n\nfunc newMockCadenceClient() mockCadenceClient {\n\treturn mockCadenceClient{\n\t\tclient:       new(mocks.Client),\n\t\tdomainClient: new(mocks.DomainClient),\n\t}\n}\n"
  },
  {
    "path": "client/admin/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage admin\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -package admin github.com/uber/cadence/client/admin Client\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/admin_generated.go -v client=Admin\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/admin_generated.go -v client=Admin\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/admin_generated.go -v client=Admin\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/admin_generated.go -v client=Admin -v package=adminv1 -v path=github.com/uber/cadence-idl/go/proto/admin/v1 -v prefix=Admin\n//go:generate gowrap gen -g -p . -i Client -t ../templates/thrift.tmpl -o ../wrappers/thrift/admin_generated.go -v client=Admin -v prefix=Admin\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/admin_generated.go -v client=Admin\n\n// Client is the interface exposed by admin service client\ntype Client interface {\n\tAddSearchAttribute(context.Context, *types.AddSearchAttributeRequest, ...yarpc.CallOption) error\n\tCloseShard(context.Context, *types.CloseShardRequest, ...yarpc.CallOption) error\n\tDescribeCluster(context.Context, ...yarpc.CallOption) (*types.DescribeClusterResponse, error)\n\tDescribeShardDistribution(context.Context, *types.DescribeShardDistributionRequest, ...yarpc.CallOption) (*types.DescribeShardDistributionResponse, error)\n\tDescribeHistoryHost(context.Context, *types.DescribeHistoryHostRequest, ...yarpc.CallOption) (*types.DescribeHistoryHostResponse, error)\n\tDescribeQueue(context.Context, *types.DescribeQueueRequest, ...yarpc.CallOption) (*types.DescribeQueueResponse, error)\n\tDescribeWorkflowExecution(context.Context, *types.AdminDescribeWorkflowExecutionRequest, ...yarpc.CallOption) (*types.AdminDescribeWorkflowExecutionResponse, error)\n\tGetDLQReplicationMessages(context.Context, *types.GetDLQReplicationMessagesRequest, ...yarpc.CallOption) (*types.GetDLQReplicationMessagesResponse, error)\n\tGetDomainReplicationMessages(context.Context, *types.GetDomainReplicationMessagesRequest, ...yarpc.CallOption) (*types.GetDomainReplicationMessagesResponse, error)\n\tGetReplicationMessages(context.Context, *types.GetReplicationMessagesRequest, ...yarpc.CallOption) (*types.GetReplicationMessagesResponse, error)\n\tGetWorkflowExecutionRawHistoryV2(context.Context, *types.GetWorkflowExecutionRawHistoryV2Request, ...yarpc.CallOption) (*types.GetWorkflowExecutionRawHistoryV2Response, error)\n\tCountDLQMessages(context.Context, *types.CountDLQMessagesRequest, ...yarpc.CallOption) (*types.CountDLQMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *types.MergeDLQMessagesRequest, ...yarpc.CallOption) (*types.MergeDLQMessagesResponse, error)\n\tPurgeDLQMessages(context.Context, *types.PurgeDLQMessagesRequest, ...yarpc.CallOption) error\n\tReadDLQMessages(context.Context, *types.ReadDLQMessagesRequest, ...yarpc.CallOption) (*types.ReadDLQMessagesResponse, error)\n\tReapplyEvents(context.Context, *types.ReapplyEventsRequest, ...yarpc.CallOption) error\n\tRefreshWorkflowTasks(context.Context, *types.RefreshWorkflowTasksRequest, ...yarpc.CallOption) error\n\tRemoveTask(context.Context, *types.RemoveTaskRequest, ...yarpc.CallOption) error\n\tResendReplicationTasks(context.Context, *types.ResendReplicationTasksRequest, ...yarpc.CallOption) error\n\tResetQueue(context.Context, *types.ResetQueueRequest, ...yarpc.CallOption) error\n\tGetDynamicConfig(context.Context, *types.GetDynamicConfigRequest, ...yarpc.CallOption) (*types.GetDynamicConfigResponse, error)\n\tUpdateDynamicConfig(context.Context, *types.UpdateDynamicConfigRequest, ...yarpc.CallOption) error\n\tRestoreDynamicConfig(context.Context, *types.RestoreDynamicConfigRequest, ...yarpc.CallOption) error\n\tListDynamicConfig(context.Context, *types.ListDynamicConfigRequest, ...yarpc.CallOption) (*types.ListDynamicConfigResponse, error)\n\tDeleteWorkflow(context.Context, *types.AdminDeleteWorkflowRequest, ...yarpc.CallOption) (*types.AdminDeleteWorkflowResponse, error)\n\tMaintainCorruptWorkflow(context.Context, *types.AdminMaintainWorkflowRequest, ...yarpc.CallOption) (*types.AdminMaintainWorkflowResponse, error)\n\tGetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.GetGlobalIsolationGroupsResponse, error)\n\tUpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.UpdateGlobalIsolationGroupsResponse, error)\n\tGetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.GetDomainIsolationGroupsResponse, error)\n\tUpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.UpdateDomainIsolationGroupsResponse, error)\n\tGetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error)\n\tUpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n\tUpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (*types.UpdateTaskListPartitionConfigResponse, error)\n}\n"
  },
  {
    "path": "client/admin/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package admin -source interface.go -destination interface_mock.go -package admin github.com/uber/cadence/client/admin Client\n//\n\n// Package admin is a generated GoMock package.\npackage admin\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// AddSearchAttribute mocks base method.\nfunc (m *MockClient) AddSearchAttribute(arg0 context.Context, arg1 *types.AddSearchAttributeRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"AddSearchAttribute\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AddSearchAttribute indicates an expected call of AddSearchAttribute.\nfunc (mr *MockClientMockRecorder) AddSearchAttribute(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddSearchAttribute\", reflect.TypeOf((*MockClient)(nil).AddSearchAttribute), varargs...)\n}\n\n// CloseShard mocks base method.\nfunc (m *MockClient) CloseShard(arg0 context.Context, arg1 *types.CloseShardRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CloseShard\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CloseShard indicates an expected call of CloseShard.\nfunc (mr *MockClientMockRecorder) CloseShard(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseShard\", reflect.TypeOf((*MockClient)(nil).CloseShard), varargs...)\n}\n\n// CountDLQMessages mocks base method.\nfunc (m *MockClient) CountDLQMessages(arg0 context.Context, arg1 *types.CountDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.CountDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CountDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.CountDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountDLQMessages indicates an expected call of CountDLQMessages.\nfunc (mr *MockClientMockRecorder) CountDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountDLQMessages\", reflect.TypeOf((*MockClient)(nil).CountDLQMessages), varargs...)\n}\n\n// DeleteWorkflow mocks base method.\nfunc (m *MockClient) DeleteWorkflow(arg0 context.Context, arg1 *types.AdminDeleteWorkflowRequest, arg2 ...yarpc.CallOption) (*types.AdminDeleteWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DeleteWorkflow\", varargs...)\n\tret0, _ := ret[0].(*types.AdminDeleteWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteWorkflow indicates an expected call of DeleteWorkflow.\nfunc (mr *MockClientMockRecorder) DeleteWorkflow(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflow\", reflect.TypeOf((*MockClient)(nil).DeleteWorkflow), varargs...)\n}\n\n// DescribeCluster mocks base method.\nfunc (m *MockClient) DescribeCluster(arg0 context.Context, arg1 ...yarpc.CallOption) (*types.DescribeClusterResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0}\n\tfor _, a := range arg1 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeCluster\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeClusterResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeCluster indicates an expected call of DescribeCluster.\nfunc (mr *MockClientMockRecorder) DescribeCluster(arg0 any, arg1 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0}, arg1...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeCluster\", reflect.TypeOf((*MockClient)(nil).DescribeCluster), varargs...)\n}\n\n// DescribeHistoryHost mocks base method.\nfunc (m *MockClient) DescribeHistoryHost(arg0 context.Context, arg1 *types.DescribeHistoryHostRequest, arg2 ...yarpc.CallOption) (*types.DescribeHistoryHostResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeHistoryHostResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeHistoryHost indicates an expected call of DescribeHistoryHost.\nfunc (mr *MockClientMockRecorder) DescribeHistoryHost(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeHistoryHost\", reflect.TypeOf((*MockClient)(nil).DescribeHistoryHost), varargs...)\n}\n\n// DescribeQueue mocks base method.\nfunc (m *MockClient) DescribeQueue(arg0 context.Context, arg1 *types.DescribeQueueRequest, arg2 ...yarpc.CallOption) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeQueue\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeQueue indicates an expected call of DescribeQueue.\nfunc (mr *MockClientMockRecorder) DescribeQueue(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeQueue\", reflect.TypeOf((*MockClient)(nil).DescribeQueue), varargs...)\n}\n\n// DescribeShardDistribution mocks base method.\nfunc (m *MockClient) DescribeShardDistribution(arg0 context.Context, arg1 *types.DescribeShardDistributionRequest, arg2 ...yarpc.CallOption) (*types.DescribeShardDistributionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeShardDistribution\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeShardDistributionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeShardDistribution indicates an expected call of DescribeShardDistribution.\nfunc (mr *MockClientMockRecorder) DescribeShardDistribution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeShardDistribution\", reflect.TypeOf((*MockClient)(nil).DescribeShardDistribution), varargs...)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockClient) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.AdminDescribeWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.AdminDescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.AdminDescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockClientMockRecorder) DescribeWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).DescribeWorkflowExecution), varargs...)\n}\n\n// GetDLQReplicationMessages mocks base method.\nfunc (m *MockClient) GetDLQReplicationMessages(arg0 context.Context, arg1 *types.GetDLQReplicationMessagesRequest, arg2 ...yarpc.CallOption) (*types.GetDLQReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", varargs...)\n\tret0, _ := ret[0].(*types.GetDLQReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQReplicationMessages indicates an expected call of GetDLQReplicationMessages.\nfunc (mr *MockClientMockRecorder) GetDLQReplicationMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQReplicationMessages\", reflect.TypeOf((*MockClient)(nil).GetDLQReplicationMessages), varargs...)\n}\n\n// GetDomainAsyncWorkflowConfiguraton mocks base method.\nfunc (m *MockClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDomainAsyncWorkflowConfiguraton\", varargs...)\n\tret0, _ := ret[0].(*types.GetDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainAsyncWorkflowConfiguraton indicates an expected call of GetDomainAsyncWorkflowConfiguraton.\nfunc (mr *MockClientMockRecorder) GetDomainAsyncWorkflowConfiguraton(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAsyncWorkflowConfiguraton\", reflect.TypeOf((*MockClient)(nil).GetDomainAsyncWorkflowConfiguraton), varargs...)\n}\n\n// GetDomainIsolationGroups mocks base method.\nfunc (m *MockClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.GetDomainIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDomainIsolationGroups\", varargs...)\n\tret0, _ := ret[0].(*types.GetDomainIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainIsolationGroups indicates an expected call of GetDomainIsolationGroups.\nfunc (mr *MockClientMockRecorder) GetDomainIsolationGroups(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainIsolationGroups\", reflect.TypeOf((*MockClient)(nil).GetDomainIsolationGroups), varargs...)\n}\n\n// GetDomainReplicationMessages mocks base method.\nfunc (m *MockClient) GetDomainReplicationMessages(arg0 context.Context, arg1 *types.GetDomainReplicationMessagesRequest, arg2 ...yarpc.CallOption) (*types.GetDomainReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDomainReplicationMessages\", varargs...)\n\tret0, _ := ret[0].(*types.GetDomainReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainReplicationMessages indicates an expected call of GetDomainReplicationMessages.\nfunc (mr *MockClientMockRecorder) GetDomainReplicationMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainReplicationMessages\", reflect.TypeOf((*MockClient)(nil).GetDomainReplicationMessages), varargs...)\n}\n\n// GetDynamicConfig mocks base method.\nfunc (m *MockClient) GetDynamicConfig(arg0 context.Context, arg1 *types.GetDynamicConfigRequest, arg2 ...yarpc.CallOption) (*types.GetDynamicConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDynamicConfig\", varargs...)\n\tret0, _ := ret[0].(*types.GetDynamicConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDynamicConfig indicates an expected call of GetDynamicConfig.\nfunc (mr *MockClientMockRecorder) GetDynamicConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDynamicConfig\", reflect.TypeOf((*MockClient)(nil).GetDynamicConfig), varargs...)\n}\n\n// GetGlobalIsolationGroups mocks base method.\nfunc (m *MockClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.GetGlobalIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetGlobalIsolationGroups\", varargs...)\n\tret0, _ := ret[0].(*types.GetGlobalIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetGlobalIsolationGroups indicates an expected call of GetGlobalIsolationGroups.\nfunc (mr *MockClientMockRecorder) GetGlobalIsolationGroups(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetGlobalIsolationGroups\", reflect.TypeOf((*MockClient)(nil).GetGlobalIsolationGroups), varargs...)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockClient) GetReplicationMessages(arg0 context.Context, arg1 *types.GetReplicationMessagesRequest, arg2 ...yarpc.CallOption) (*types.GetReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", varargs...)\n\tret0, _ := ret[0].(*types.GetReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockClientMockRecorder) GetReplicationMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockClient)(nil).GetReplicationMessages), varargs...)\n}\n\n// GetWorkflowExecutionRawHistoryV2 mocks base method.\nfunc (m *MockClient) GetWorkflowExecutionRawHistoryV2(arg0 context.Context, arg1 *types.GetWorkflowExecutionRawHistoryV2Request, arg2 ...yarpc.CallOption) (*types.GetWorkflowExecutionRawHistoryV2Response, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionRawHistoryV2\", varargs...)\n\tret0, _ := ret[0].(*types.GetWorkflowExecutionRawHistoryV2Response)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecutionRawHistoryV2 indicates an expected call of GetWorkflowExecutionRawHistoryV2.\nfunc (mr *MockClientMockRecorder) GetWorkflowExecutionRawHistoryV2(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecutionRawHistoryV2\", reflect.TypeOf((*MockClient)(nil).GetWorkflowExecutionRawHistoryV2), varargs...)\n}\n\n// ListDynamicConfig mocks base method.\nfunc (m *MockClient) ListDynamicConfig(arg0 context.Context, arg1 *types.ListDynamicConfigRequest, arg2 ...yarpc.CallOption) (*types.ListDynamicConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListDynamicConfig\", varargs...)\n\tret0, _ := ret[0].(*types.ListDynamicConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDynamicConfig indicates an expected call of ListDynamicConfig.\nfunc (mr *MockClientMockRecorder) ListDynamicConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDynamicConfig\", reflect.TypeOf((*MockClient)(nil).ListDynamicConfig), varargs...)\n}\n\n// MaintainCorruptWorkflow mocks base method.\nfunc (m *MockClient) MaintainCorruptWorkflow(arg0 context.Context, arg1 *types.AdminMaintainWorkflowRequest, arg2 ...yarpc.CallOption) (*types.AdminMaintainWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"MaintainCorruptWorkflow\", varargs...)\n\tret0, _ := ret[0].(*types.AdminMaintainWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MaintainCorruptWorkflow indicates an expected call of MaintainCorruptWorkflow.\nfunc (mr *MockClientMockRecorder) MaintainCorruptWorkflow(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaintainCorruptWorkflow\", reflect.TypeOf((*MockClient)(nil).MaintainCorruptWorkflow), varargs...)\n}\n\n// MergeDLQMessages mocks base method.\nfunc (m *MockClient) MergeDLQMessages(arg0 context.Context, arg1 *types.MergeDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.MergeDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.MergeDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MergeDLQMessages indicates an expected call of MergeDLQMessages.\nfunc (mr *MockClientMockRecorder) MergeDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeDLQMessages\", reflect.TypeOf((*MockClient)(nil).MergeDLQMessages), varargs...)\n}\n\n// PurgeDLQMessages mocks base method.\nfunc (m *MockClient) PurgeDLQMessages(arg0 context.Context, arg1 *types.PurgeDLQMessagesRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PurgeDLQMessages indicates an expected call of PurgeDLQMessages.\nfunc (mr *MockClientMockRecorder) PurgeDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PurgeDLQMessages\", reflect.TypeOf((*MockClient)(nil).PurgeDLQMessages), varargs...)\n}\n\n// ReadDLQMessages mocks base method.\nfunc (m *MockClient) ReadDLQMessages(arg0 context.Context, arg1 *types.ReadDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.ReadDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.ReadDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadDLQMessages indicates an expected call of ReadDLQMessages.\nfunc (mr *MockClientMockRecorder) ReadDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadDLQMessages\", reflect.TypeOf((*MockClient)(nil).ReadDLQMessages), varargs...)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockClient) ReapplyEvents(arg0 context.Context, arg1 *types.ReapplyEventsRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockClientMockRecorder) ReapplyEvents(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockClient)(nil).ReapplyEvents), varargs...)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockClient) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.RefreshWorkflowTasksRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockClientMockRecorder) RefreshWorkflowTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockClient)(nil).RefreshWorkflowTasks), varargs...)\n}\n\n// RemoveTask mocks base method.\nfunc (m *MockClient) RemoveTask(arg0 context.Context, arg1 *types.RemoveTaskRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RemoveTask\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveTask indicates an expected call of RemoveTask.\nfunc (mr *MockClientMockRecorder) RemoveTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveTask\", reflect.TypeOf((*MockClient)(nil).RemoveTask), varargs...)\n}\n\n// ResendReplicationTasks mocks base method.\nfunc (m *MockClient) ResendReplicationTasks(arg0 context.Context, arg1 *types.ResendReplicationTasksRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResendReplicationTasks\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResendReplicationTasks indicates an expected call of ResendReplicationTasks.\nfunc (mr *MockClientMockRecorder) ResendReplicationTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResendReplicationTasks\", reflect.TypeOf((*MockClient)(nil).ResendReplicationTasks), varargs...)\n}\n\n// ResetQueue mocks base method.\nfunc (m *MockClient) ResetQueue(arg0 context.Context, arg1 *types.ResetQueueRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetQueue\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetQueue indicates an expected call of ResetQueue.\nfunc (mr *MockClientMockRecorder) ResetQueue(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetQueue\", reflect.TypeOf((*MockClient)(nil).ResetQueue), varargs...)\n}\n\n// RestoreDynamicConfig mocks base method.\nfunc (m *MockClient) RestoreDynamicConfig(arg0 context.Context, arg1 *types.RestoreDynamicConfigRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RestoreDynamicConfig\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RestoreDynamicConfig indicates an expected call of RestoreDynamicConfig.\nfunc (mr *MockClientMockRecorder) RestoreDynamicConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestoreDynamicConfig\", reflect.TypeOf((*MockClient)(nil).RestoreDynamicConfig), varargs...)\n}\n\n// UpdateDomainAsyncWorkflowConfiguraton mocks base method.\nfunc (m *MockClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateDomainAsyncWorkflowConfiguraton\", varargs...)\n\tret0, _ := ret[0].(*types.UpdateDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainAsyncWorkflowConfiguraton indicates an expected call of UpdateDomainAsyncWorkflowConfiguraton.\nfunc (mr *MockClientMockRecorder) UpdateDomainAsyncWorkflowConfiguraton(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainAsyncWorkflowConfiguraton\", reflect.TypeOf((*MockClient)(nil).UpdateDomainAsyncWorkflowConfiguraton), varargs...)\n}\n\n// UpdateDomainIsolationGroups mocks base method.\nfunc (m *MockClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.UpdateDomainIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateDomainIsolationGroups\", varargs...)\n\tret0, _ := ret[0].(*types.UpdateDomainIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainIsolationGroups indicates an expected call of UpdateDomainIsolationGroups.\nfunc (mr *MockClientMockRecorder) UpdateDomainIsolationGroups(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainIsolationGroups\", reflect.TypeOf((*MockClient)(nil).UpdateDomainIsolationGroups), varargs...)\n}\n\n// UpdateDynamicConfig mocks base method.\nfunc (m *MockClient) UpdateDynamicConfig(arg0 context.Context, arg1 *types.UpdateDynamicConfigRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateDynamicConfig\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDynamicConfig indicates an expected call of UpdateDynamicConfig.\nfunc (mr *MockClientMockRecorder) UpdateDynamicConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDynamicConfig\", reflect.TypeOf((*MockClient)(nil).UpdateDynamicConfig), varargs...)\n}\n\n// UpdateGlobalIsolationGroups mocks base method.\nfunc (m *MockClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (*types.UpdateGlobalIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateGlobalIsolationGroups\", varargs...)\n\tret0, _ := ret[0].(*types.UpdateGlobalIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateGlobalIsolationGroups indicates an expected call of UpdateGlobalIsolationGroups.\nfunc (mr *MockClientMockRecorder) UpdateGlobalIsolationGroups(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateGlobalIsolationGroups\", reflect.TypeOf((*MockClient)(nil).UpdateGlobalIsolationGroups), varargs...)\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (*types.UpdateTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", varargs...)\n\tret0, _ := ret[0].(*types.UpdateTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockClientMockRecorder) UpdateTaskListPartitionConfig(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockClient)(nil).UpdateTaskListPartitionConfig), varargs...)\n}\n"
  },
  {
    "path": "client/clientBean.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination clientBean_mock.go -self_package github.com/uber/cadence/client\n\npackage client\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype (\n\t// Bean in an collection of clients\n\tBean interface {\n\t\tGetHistoryClient() history.Client\n\t\tGetHistoryPeers() history.PeerResolver\n\t\tGetMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error)\n\t\tGetFrontendClient() frontend.Client\n\t\tGetShardDistributorClient() sharddistributor.Client\n\t\tGetShardDistributorExecutorClient() executorclient.Client\n\t\tGetRemoteAdminClient(cluster string) (admin.Client, error)\n\t\tSetRemoteAdminClient(cluster string, client admin.Client)\n\t\tGetRemoteFrontendClient(cluster string) (frontend.Client, error)\n\t}\n\n\tclientBeanImpl struct {\n\t\tsync.Mutex\n\t\thistoryClient                  history.Client\n\t\thistoryPeers                   history.PeerResolver\n\t\tmatchingClient                 atomic.Value\n\t\tfrontendClient                 frontend.Client\n\t\tshardDistributorClient         sharddistributor.Client\n\t\tshardDistributorExecutorClient executorclient.Client\n\t\tremoteAdminClients             map[string]admin.Client\n\t\tremoteFrontendClients          map[string]frontend.Client\n\t\tfactory                        Factory\n\t}\n)\n\n// NewClientBean provides a collection of clients\nfunc NewClientBean(factory Factory, dispatcher *yarpc.Dispatcher, clusterMetadata cluster.Metadata) (Bean, error) {\n\n\thistoryClient, historyPeers, err := factory.NewHistoryClient()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tremoteAdminClients := map[string]admin.Client{}\n\tremoteFrontendClients := map[string]frontend.Client{}\n\tfor clusterName := range clusterMetadata.GetEnabledClusterInfo() {\n\t\tclientConfig := dispatcher.ClientConfig(clusterName)\n\n\t\tadminClient, err := factory.NewAdminClientWithTimeoutAndConfig(\n\t\t\tclientConfig,\n\t\t\ttimeout.AdminDefaultTimeout,\n\t\t\ttimeout.AdminDefaultLargeTimeout,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tfrontendClient, err := factory.NewFrontendClientWithTimeoutAndConfig(\n\t\t\tclientConfig,\n\t\t\ttimeout.FrontendDefaultTimeout,\n\t\t\ttimeout.FrontendDefaultLongPollTimeout,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tremoteAdminClients[clusterName] = adminClient\n\t\tremoteFrontendClients[clusterName] = frontendClient\n\t}\n\n\tshardDistributorClient, err := factory.NewShardDistributorClient()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tshardDistributorExecutorClient, err := factory.NewShardDistributorExecutorClient()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &clientBeanImpl{\n\t\tfactory:                        factory,\n\t\thistoryClient:                  historyClient,\n\t\thistoryPeers:                   historyPeers,\n\t\tfrontendClient:                 remoteFrontendClients[clusterMetadata.GetCurrentClusterName()],\n\t\tshardDistributorClient:         shardDistributorClient,\n\t\tshardDistributorExecutorClient: shardDistributorExecutorClient,\n\t\tremoteAdminClients:             remoteAdminClients,\n\t\tremoteFrontendClients:          remoteFrontendClients,\n\t}, nil\n}\n\nfunc (h *clientBeanImpl) GetHistoryClient() history.Client {\n\treturn h.historyClient\n}\n\nfunc (h *clientBeanImpl) GetHistoryPeers() history.PeerResolver {\n\treturn h.historyPeers\n}\n\nfunc (h *clientBeanImpl) GetMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error) {\n\tif client := h.matchingClient.Load(); client != nil {\n\t\treturn client.(matching.Client), nil\n\t}\n\treturn h.lazyInitMatchingClient(domainIDToName)\n}\n\nfunc (h *clientBeanImpl) GetFrontendClient() frontend.Client {\n\treturn h.frontendClient\n}\n\nfunc (h *clientBeanImpl) GetShardDistributorClient() sharddistributor.Client {\n\treturn h.shardDistributorClient\n}\n\nfunc (h *clientBeanImpl) GetShardDistributorExecutorClient() executorclient.Client {\n\treturn h.shardDistributorExecutorClient\n}\n\nfunc (h *clientBeanImpl) GetRemoteAdminClient(cluster string) (admin.Client, error) {\n\tclient, ok := h.remoteAdminClients[cluster]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"unknown cluster name: %v with given cluster client map: %v\", cluster, h.remoteAdminClients)\n\t}\n\treturn client, nil\n}\n\nfunc (h *clientBeanImpl) SetRemoteAdminClient(\n\tcluster string,\n\tclient admin.Client,\n) {\n\n\th.remoteAdminClients[cluster] = client\n}\n\nfunc (h *clientBeanImpl) GetRemoteFrontendClient(cluster string) (frontend.Client, error) {\n\tclient, ok := h.remoteFrontendClients[cluster]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"unknown cluster name: %v with given cluster client map: %v\", cluster, h.remoteFrontendClients)\n\t}\n\treturn client, nil\n}\n\nfunc (h *clientBeanImpl) lazyInitMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error) {\n\th.Lock()\n\tdefer h.Unlock()\n\tif cached := h.matchingClient.Load(); cached != nil {\n\t\treturn cached.(matching.Client), nil\n\t}\n\tclient, err := h.factory.NewMatchingClient(domainIDToName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\th.matchingClient.Store(client)\n\treturn client, nil\n}\n"
  },
  {
    "path": "client/clientBean_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: clientBean.go\n//\n// Generated by this command:\n//\n//\tmockgen -package client -source clientBean.go -destination clientBean_mock.go -self_package github.com/uber/cadence/client\n//\n\n// Package client is a generated GoMock package.\npackage client\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tadmin \"github.com/uber/cadence/client/admin\"\n\tfrontend \"github.com/uber/cadence/client/frontend\"\n\thistory \"github.com/uber/cadence/client/history\"\n\tmatching \"github.com/uber/cadence/client/matching\"\n\tsharddistributor \"github.com/uber/cadence/client/sharddistributor\"\n\texecutorclient \"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// MockBean is a mock of Bean interface.\ntype MockBean struct {\n\tctrl     *gomock.Controller\n\trecorder *MockBeanMockRecorder\n\tisgomock struct{}\n}\n\n// MockBeanMockRecorder is the mock recorder for MockBean.\ntype MockBeanMockRecorder struct {\n\tmock *MockBean\n}\n\n// NewMockBean creates a new mock instance.\nfunc NewMockBean(ctrl *gomock.Controller) *MockBean {\n\tmock := &MockBean{ctrl: ctrl}\n\tmock.recorder = &MockBeanMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockBean) EXPECT() *MockBeanMockRecorder {\n\treturn m.recorder\n}\n\n// GetFrontendClient mocks base method.\nfunc (m *MockBean) GetFrontendClient() frontend.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFrontendClient\")\n\tret0, _ := ret[0].(frontend.Client)\n\treturn ret0\n}\n\n// GetFrontendClient indicates an expected call of GetFrontendClient.\nfunc (mr *MockBeanMockRecorder) GetFrontendClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFrontendClient\", reflect.TypeOf((*MockBean)(nil).GetFrontendClient))\n}\n\n// GetHistoryClient mocks base method.\nfunc (m *MockBean) GetHistoryClient() history.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryClient\")\n\tret0, _ := ret[0].(history.Client)\n\treturn ret0\n}\n\n// GetHistoryClient indicates an expected call of GetHistoryClient.\nfunc (mr *MockBeanMockRecorder) GetHistoryClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryClient\", reflect.TypeOf((*MockBean)(nil).GetHistoryClient))\n}\n\n// GetHistoryPeers mocks base method.\nfunc (m *MockBean) GetHistoryPeers() history.PeerResolver {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryPeers\")\n\tret0, _ := ret[0].(history.PeerResolver)\n\treturn ret0\n}\n\n// GetHistoryPeers indicates an expected call of GetHistoryPeers.\nfunc (mr *MockBeanMockRecorder) GetHistoryPeers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryPeers\", reflect.TypeOf((*MockBean)(nil).GetHistoryPeers))\n}\n\n// GetMatchingClient mocks base method.\nfunc (m *MockBean) GetMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMatchingClient\", domainIDToName)\n\tret0, _ := ret[0].(matching.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMatchingClient indicates an expected call of GetMatchingClient.\nfunc (mr *MockBeanMockRecorder) GetMatchingClient(domainIDToName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMatchingClient\", reflect.TypeOf((*MockBean)(nil).GetMatchingClient), domainIDToName)\n}\n\n// GetRemoteAdminClient mocks base method.\nfunc (m *MockBean) GetRemoteAdminClient(cluster string) (admin.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteAdminClient\", cluster)\n\tret0, _ := ret[0].(admin.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteAdminClient indicates an expected call of GetRemoteAdminClient.\nfunc (mr *MockBeanMockRecorder) GetRemoteAdminClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteAdminClient\", reflect.TypeOf((*MockBean)(nil).GetRemoteAdminClient), cluster)\n}\n\n// GetRemoteFrontendClient mocks base method.\nfunc (m *MockBean) GetRemoteFrontendClient(cluster string) (frontend.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteFrontendClient\", cluster)\n\tret0, _ := ret[0].(frontend.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteFrontendClient indicates an expected call of GetRemoteFrontendClient.\nfunc (mr *MockBeanMockRecorder) GetRemoteFrontendClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteFrontendClient\", reflect.TypeOf((*MockBean)(nil).GetRemoteFrontendClient), cluster)\n}\n\n// GetShardDistributorClient mocks base method.\nfunc (m *MockBean) GetShardDistributorClient() sharddistributor.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardDistributorClient\")\n\tret0, _ := ret[0].(sharddistributor.Client)\n\treturn ret0\n}\n\n// GetShardDistributorClient indicates an expected call of GetShardDistributorClient.\nfunc (mr *MockBeanMockRecorder) GetShardDistributorClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardDistributorClient\", reflect.TypeOf((*MockBean)(nil).GetShardDistributorClient))\n}\n\n// GetShardDistributorExecutorClient mocks base method.\nfunc (m *MockBean) GetShardDistributorExecutorClient() executorclient.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardDistributorExecutorClient\")\n\tret0, _ := ret[0].(executorclient.Client)\n\treturn ret0\n}\n\n// GetShardDistributorExecutorClient indicates an expected call of GetShardDistributorExecutorClient.\nfunc (mr *MockBeanMockRecorder) GetShardDistributorExecutorClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardDistributorExecutorClient\", reflect.TypeOf((*MockBean)(nil).GetShardDistributorExecutorClient))\n}\n\n// SetRemoteAdminClient mocks base method.\nfunc (m *MockBean) SetRemoteAdminClient(cluster string, client admin.Client) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetRemoteAdminClient\", cluster, client)\n}\n\n// SetRemoteAdminClient indicates an expected call of SetRemoteAdminClient.\nfunc (mr *MockBeanMockRecorder) SetRemoteAdminClient(cluster, client any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetRemoteAdminClient\", reflect.TypeOf((*MockBean)(nil).SetRemoteAdminClient), cluster, client)\n}\n"
  },
  {
    "path": "client/clientfactory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage client\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc/api/transport\"\n\n\t\"github.com/uber/cadence/.gen/go/admin/adminserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/history/historyserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/matching/matchingserviceclient\"\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/errorinjectors\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/client/wrappers/metered\"\n\t\"github.com/uber/cadence/client/wrappers/thrift\"\n\ttimeoutwrapper \"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype (\n\t// Factory can be used to create RPC clients for cadence services\n\tFactory interface {\n\t\tNewHistoryClient() (history.Client, history.PeerResolver, error)\n\t\tNewMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error)\n\n\t\tNewHistoryClientWithTimeout(timeout time.Duration) (history.Client, history.PeerResolver, error)\n\t\tNewMatchingClientWithTimeout(domainIDToName DomainIDToNameFunc, timeout time.Duration, longPollTimeout time.Duration) (matching.Client, error)\n\n\t\tNewAdminClientWithTimeoutAndConfig(config transport.ClientConfig, timeout time.Duration, largeTimeout time.Duration) (admin.Client, error)\n\t\tNewFrontendClientWithTimeoutAndConfig(config transport.ClientConfig, timeout time.Duration, longPollTimeout time.Duration) (frontend.Client, error)\n\n\t\tNewShardDistributorClient() (sharddistributor.Client, error)\n\t\tNewShardDistributorClientWithTimeout(timeout time.Duration) (sharddistributor.Client, error)\n\t\tNewShardDistributorExecutorClient() (executorclient.Client, error)\n\t}\n\n\t// DomainIDToNameFunc maps a domainID to domain name. Returns error when mapping is not possible.\n\tDomainIDToNameFunc func(string) (string, error)\n\n\trpcClientFactory struct {\n\t\trpcFactory            rpc.Factory\n\t\tresolver              membership.Resolver\n\t\tmetricsClient         metrics.Client\n\t\tdynConfig             *dynamicconfig.Collection\n\t\tnumberOfHistoryShards int\n\t\tallIsolationGroups    func() []string\n\t\tlogger                log.Logger\n\t}\n)\n\n// NewRPCClientFactory creates an instance of client factory that knows how to dispatch RPC calls.\nfunc NewRPCClientFactory(\n\trpcFactory rpc.Factory,\n\tresolver membership.Resolver,\n\tmetricsClient metrics.Client,\n\tdc *dynamicconfig.Collection,\n\tnumberOfHistoryShards int,\n\tallIsolationGroups func() []string,\n\tlogger log.Logger,\n) Factory {\n\treturn &rpcClientFactory{\n\t\trpcFactory:            rpcFactory,\n\t\tresolver:              resolver,\n\t\tmetricsClient:         metricsClient,\n\t\tdynConfig:             dc,\n\t\tnumberOfHistoryShards: numberOfHistoryShards,\n\t\tallIsolationGroups:    allIsolationGroups,\n\t\tlogger:                logger,\n\t}\n}\n\nfunc (cf *rpcClientFactory) NewHistoryClient() (history.Client, history.PeerResolver, error) {\n\treturn cf.NewHistoryClientWithTimeout(timeoutwrapper.HistoryDefaultTimeout)\n}\n\nfunc (cf *rpcClientFactory) NewMatchingClient(domainIDToName DomainIDToNameFunc) (matching.Client, error) {\n\treturn cf.NewMatchingClientWithTimeout(domainIDToName, timeoutwrapper.MatchingDefaultTimeout, timeoutwrapper.MatchingDefaultLongPollTimeout)\n}\n\nfunc (cf *rpcClientFactory) NewHistoryClientWithTimeout(timeout time.Duration) (history.Client, history.PeerResolver, error) {\n\tvar rawClient history.Client\n\tvar namedPort = membership.PortTchannel\n\n\toutboundConfig := cf.rpcFactory.GetDispatcher().ClientConfig(service.History)\n\tif rpc.IsGRPCOutbound(outboundConfig) {\n\t\trawClient = grpc.NewHistoryClient(historyv1.NewHistoryAPIYARPCClient(outboundConfig))\n\t\tnamedPort = membership.PortGRPC\n\t} else {\n\t\trawClient = thrift.NewHistoryClient(historyserviceclient.New(outboundConfig))\n\t}\n\n\tpeerResolver := history.NewPeerResolver(cf.numberOfHistoryShards, cf.resolver, namedPort)\n\n\tclient := history.NewClient(\n\t\tcf.numberOfHistoryShards,\n\t\tcf.rpcFactory.GetMaxMessageSize(),\n\t\trawClient,\n\t\tpeerResolver,\n\t\tcf.logger,\n\t)\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.HistoryErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewHistoryClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewHistoryClient(client, cf.metricsClient)\n\t}\n\tclient = timeoutwrapper.NewHistoryClient(client, timeout)\n\treturn client, peerResolver, nil\n}\n\nfunc (cf *rpcClientFactory) NewMatchingClientWithTimeout(\n\tdomainIDToName DomainIDToNameFunc,\n\ttimeout time.Duration,\n\tlongPollTimeout time.Duration,\n) (matching.Client, error) {\n\tvar rawClient matching.Client\n\tvar namedPort = membership.PortTchannel\n\toutboundConfig := cf.rpcFactory.GetDispatcher().ClientConfig(service.Matching)\n\tif rpc.IsGRPCOutbound(outboundConfig) {\n\t\trawClient = grpc.NewMatchingClient(matchingv1.NewMatchingAPIYARPCClient(outboundConfig))\n\t\tnamedPort = membership.PortGRPC\n\t} else {\n\t\trawClient = thrift.NewMatchingClient(matchingserviceclient.New(outboundConfig))\n\t}\n\n\tpeerResolver := matching.NewPeerResolver(cf.resolver, namedPort)\n\n\tpartitionConfigProvider := matching.NewPartitionConfigProvider(cf.logger, cf.metricsClient, domainIDToName, cf.dynConfig)\n\tdefaultLoadBalancer := matching.NewLoadBalancer(partitionConfigProvider)\n\troundRobinLoadBalancer := matching.NewRoundRobinLoadBalancer(partitionConfigProvider)\n\tweightedLoadBalancer := matching.NewWeightedLoadBalancer(roundRobinLoadBalancer, partitionConfigProvider, cf.logger)\n\tigLoadBalancer := matching.NewIsolationLoadBalancer(weightedLoadBalancer, partitionConfigProvider, domainIDToName, cf.dynConfig)\n\tloadBalancers := map[string]matching.LoadBalancer{\n\t\t\"random\":      defaultLoadBalancer,\n\t\t\"round-robin\": roundRobinLoadBalancer,\n\t\t\"weighted\":    weightedLoadBalancer,\n\t\t\"isolation\":   igLoadBalancer,\n\t}\n\tclient := matching.NewClient(\n\t\trawClient,\n\t\tpeerResolver,\n\t\tmatching.NewMultiLoadBalancer(defaultLoadBalancer, loadBalancers, domainIDToName, cf.dynConfig, cf.logger),\n\t\tpartitionConfigProvider,\n\t)\n\tclient = timeoutwrapper.NewMatchingClient(client, longPollTimeout, timeout)\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.MatchingErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewMatchingClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewMatchingClient(client, cf.metricsClient)\n\t}\n\treturn client, nil\n}\n\nfunc (cf *rpcClientFactory) NewAdminClientWithTimeoutAndConfig(\n\tconfig transport.ClientConfig,\n\ttimeout time.Duration,\n\tlargeTimeout time.Duration,\n) (admin.Client, error) {\n\tvar client admin.Client\n\tif rpc.IsGRPCOutbound(config) {\n\t\tclient = grpc.NewAdminClient(adminv1.NewAdminAPIYARPCClient(config))\n\t} else {\n\t\tclient = thrift.NewAdminClient(adminserviceclient.New(config))\n\t}\n\n\tclient = timeoutwrapper.NewAdminClient(client, largeTimeout, timeout)\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.AdminErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewAdminClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewAdminClient(client, cf.metricsClient)\n\t}\n\treturn client, nil\n}\n\nfunc (cf *rpcClientFactory) NewFrontendClientWithTimeoutAndConfig(\n\tconfig transport.ClientConfig,\n\ttimeout time.Duration,\n\tlongPollTimeout time.Duration,\n) (frontend.Client, error) {\n\tvar client frontend.Client\n\tif rpc.IsGRPCOutbound(config) {\n\t\tclient = grpc.NewFrontendClient(\n\t\t\tapiv1.NewDomainAPIYARPCClient(config),\n\t\t\tapiv1.NewWorkflowAPIYARPCClient(config),\n\t\t\tapiv1.NewWorkerAPIYARPCClient(config),\n\t\t\tapiv1.NewVisibilityAPIYARPCClient(config),\n\t\t)\n\t} else {\n\t\tclient = thrift.NewFrontendClient(workflowserviceclient.New(config))\n\t}\n\n\tclient = timeoutwrapper.NewFrontendClient(client, longPollTimeout, timeout)\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.FrontendErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewFrontendClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewFrontendClient(client, cf.metricsClient)\n\t}\n\treturn client, nil\n}\n\nfunc (cf *rpcClientFactory) NewShardDistributorClient() (sharddistributor.Client, error) {\n\treturn cf.NewShardDistributorClientWithTimeout(timeoutwrapper.ShardDistributorDefaultTimeout)\n}\n\nfunc (cf *rpcClientFactory) NewShardDistributorClientWithTimeout(\n\ttimeout time.Duration,\n) (sharddistributor.Client, error) {\n\toutboundConfig, ok := cf.rpcFactory.GetDispatcher().OutboundConfig(service.ShardDistributor)\n\t// If no outbound config is found, it means the service is not enabled, we just return nil as we don't want to\n\t// break existing configs.\n\tif !ok {\n\t\treturn nil, nil\n\t}\n\n\tif !rpc.IsGRPCOutbound(outboundConfig) {\n\t\treturn nil, fmt.Errorf(\"shard distributor client does not support non-GRPC outbound\")\n\t}\n\n\tclient := grpc.NewShardDistributorClient(\n\t\tsharddistributorv1.NewShardDistributorAPIYARPCClient(outboundConfig),\n\t)\n\n\tclient = timeoutwrapper.NewShardDistributorClient(client, timeout)\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.ShardDistributorErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewShardDistributorClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewShardDistributorClient(client, cf.metricsClient)\n\t}\n\n\treturn client, nil\n}\n\nfunc (cf *rpcClientFactory) NewShardDistributorExecutorClient() (executorclient.Client, error) {\n\toutboundConfig, ok := cf.rpcFactory.GetDispatcher().OutboundConfig(service.ShardDistributor)\n\t// If no outbound config is found, it means the service is not enabled, we just return nil as we don't want to\n\t// break existing configs.\n\tif !ok {\n\t\treturn nil, nil\n\t}\n\n\tif !rpc.IsGRPCOutbound(outboundConfig) {\n\t\treturn nil, fmt.Errorf(\"shard distributor client does not support non-GRPC outbound\")\n\t}\n\n\tclient := grpc.NewShardDistributorExecutorClient(sharddistributorv1.NewShardDistributorExecutorAPIYARPCClient(outboundConfig))\n\tif errorRate := cf.dynConfig.GetFloat64Property(dynamicproperties.ShardDistributorErrorInjectionRate)(); errorRate != 0 {\n\t\tclient = errorinjectors.NewShardDistributorExecutorClient(client, errorRate, cf.logger)\n\t}\n\tif cf.metricsClient != nil {\n\t\tclient = metered.NewShardDistributorExecutorClient(client, cf.metricsClient)\n\t}\n\n\treturn client, nil\n}\n"
  },
  {
    "path": "client/frontend/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage frontend\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/client/frontend\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/frontend_generated.go -v client=Frontend\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/frontend_generated.go -v client=Frontend\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/frontend_generated.go -v client=Frontend\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/frontend_generated.go -v client=Frontend -v package=apiv1 -v path=github.com/uber/cadence-idl/go/proto/api/v1 -v prefix=\n//go:generate gowrap gen -g -p . -i Client -t ../templates/thrift.tmpl -o ../wrappers/thrift/frontend_generated.go -v client=Frontend -v prefix=\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/frontend_generated.go -v client=Frontend\n\n// Client is the interface exposed by frontend service client\ntype Client interface {\n\tCountWorkflowExecutions(context.Context, *types.CountWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.CountWorkflowExecutionsResponse, error)\n\tDeleteDomain(context.Context, *types.DeleteDomainRequest, ...yarpc.CallOption) error\n\tDeprecateDomain(context.Context, *types.DeprecateDomainRequest, ...yarpc.CallOption) error\n\tDescribeDomain(context.Context, *types.DescribeDomainRequest, ...yarpc.CallOption) (*types.DescribeDomainResponse, error)\n\tDescribeTaskList(context.Context, *types.DescribeTaskListRequest, ...yarpc.CallOption) (*types.DescribeTaskListResponse, error)\n\tDescribeWorkflowExecution(context.Context, *types.DescribeWorkflowExecutionRequest, ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error)\n\tDiagnoseWorkflowExecution(context.Context, *types.DiagnoseWorkflowExecutionRequest, ...yarpc.CallOption) (*types.DiagnoseWorkflowExecutionResponse, error)\n\tGetClusterInfo(context.Context, ...yarpc.CallOption) (*types.ClusterInfo, error)\n\tGetSearchAttributes(context.Context, ...yarpc.CallOption) (*types.GetSearchAttributesResponse, error)\n\tGetWorkflowExecutionHistory(context.Context, *types.GetWorkflowExecutionHistoryRequest, ...yarpc.CallOption) (*types.GetWorkflowExecutionHistoryResponse, error)\n\tListArchivedWorkflowExecutions(context.Context, *types.ListArchivedWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.ListArchivedWorkflowExecutionsResponse, error)\n\tListClosedWorkflowExecutions(context.Context, *types.ListClosedWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.ListClosedWorkflowExecutionsResponse, error)\n\tListDomains(context.Context, *types.ListDomainsRequest, ...yarpc.CallOption) (*types.ListDomainsResponse, error)\n\tListOpenWorkflowExecutions(context.Context, *types.ListOpenWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.ListOpenWorkflowExecutionsResponse, error)\n\tListTaskListPartitions(context.Context, *types.ListTaskListPartitionsRequest, ...yarpc.CallOption) (*types.ListTaskListPartitionsResponse, error)\n\tGetTaskListsByDomain(context.Context, *types.GetTaskListsByDomainRequest, ...yarpc.CallOption) (*types.GetTaskListsByDomainResponse, error)\n\tRefreshWorkflowTasks(context.Context, *types.RefreshWorkflowTasksRequest, ...yarpc.CallOption) error\n\tListWorkflowExecutions(context.Context, *types.ListWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.ListWorkflowExecutionsResponse, error)\n\tPollForActivityTask(context.Context, *types.PollForActivityTaskRequest, ...yarpc.CallOption) (*types.PollForActivityTaskResponse, error)\n\tPollForDecisionTask(context.Context, *types.PollForDecisionTaskRequest, ...yarpc.CallOption) (*types.PollForDecisionTaskResponse, error)\n\tQueryWorkflow(context.Context, *types.QueryWorkflowRequest, ...yarpc.CallOption) (*types.QueryWorkflowResponse, error)\n\tRecordActivityTaskHeartbeat(context.Context, *types.RecordActivityTaskHeartbeatRequest, ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error)\n\tRecordActivityTaskHeartbeatByID(context.Context, *types.RecordActivityTaskHeartbeatByIDRequest, ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error)\n\tRegisterDomain(context.Context, *types.RegisterDomainRequest, ...yarpc.CallOption) error\n\tRequestCancelWorkflowExecution(context.Context, *types.RequestCancelWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tResetStickyTaskList(context.Context, *types.ResetStickyTaskListRequest, ...yarpc.CallOption) (*types.ResetStickyTaskListResponse, error)\n\tResetWorkflowExecution(context.Context, *types.ResetWorkflowExecutionRequest, ...yarpc.CallOption) (*types.ResetWorkflowExecutionResponse, error)\n\tRespondActivityTaskCanceled(context.Context, *types.RespondActivityTaskCanceledRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskCanceledByID(context.Context, *types.RespondActivityTaskCanceledByIDRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskCompleted(context.Context, *types.RespondActivityTaskCompletedRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskCompletedByID(context.Context, *types.RespondActivityTaskCompletedByIDRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskFailed(context.Context, *types.RespondActivityTaskFailedRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskFailedByID(context.Context, *types.RespondActivityTaskFailedByIDRequest, ...yarpc.CallOption) error\n\tRespondDecisionTaskCompleted(context.Context, *types.RespondDecisionTaskCompletedRequest, ...yarpc.CallOption) (*types.RespondDecisionTaskCompletedResponse, error)\n\tRespondDecisionTaskFailed(context.Context, *types.RespondDecisionTaskFailedRequest, ...yarpc.CallOption) error\n\tRespondQueryTaskCompleted(context.Context, *types.RespondQueryTaskCompletedRequest, ...yarpc.CallOption) error\n\tRestartWorkflowExecution(context.Context, *types.RestartWorkflowExecutionRequest, ...yarpc.CallOption) (*types.RestartWorkflowExecutionResponse, error)\n\tScanWorkflowExecutions(context.Context, *types.ListWorkflowExecutionsRequest, ...yarpc.CallOption) (*types.ListWorkflowExecutionsResponse, error)\n\tSignalWithStartWorkflowExecution(context.Context, *types.SignalWithStartWorkflowExecutionRequest, ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error)\n\tSignalWithStartWorkflowExecutionAsync(context.Context, *types.SignalWithStartWorkflowExecutionAsyncRequest, ...yarpc.CallOption) (*types.SignalWithStartWorkflowExecutionAsyncResponse, error)\n\tSignalWorkflowExecution(context.Context, *types.SignalWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tStartWorkflowExecution(context.Context, *types.StartWorkflowExecutionRequest, ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error)\n\tStartWorkflowExecutionAsync(context.Context, *types.StartWorkflowExecutionAsyncRequest, ...yarpc.CallOption) (*types.StartWorkflowExecutionAsyncResponse, error)\n\tTerminateWorkflowExecution(context.Context, *types.TerminateWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tUpdateDomain(context.Context, *types.UpdateDomainRequest, ...yarpc.CallOption) (*types.UpdateDomainResponse, error)\n\tFailoverDomain(context.Context, *types.FailoverDomainRequest, ...yarpc.CallOption) (*types.FailoverDomainResponse, error)\n\tListFailoverHistory(context.Context, *types.ListFailoverHistoryRequest, ...yarpc.CallOption) (*types.ListFailoverHistoryResponse, error)\n}\n"
  },
  {
    "path": "client/frontend/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package frontend -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/client/frontend\n//\n\n// Package frontend is a generated GoMock package.\npackage frontend\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// CountWorkflowExecutions mocks base method.\nfunc (m *MockClient) CountWorkflowExecutions(arg0 context.Context, arg1 *types.CountWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.CountWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CountWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.CountWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountWorkflowExecutions indicates an expected call of CountWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) CountWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).CountWorkflowExecutions), varargs...)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockClient) DeleteDomain(arg0 context.Context, arg1 *types.DeleteDomainRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DeleteDomain\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockClientMockRecorder) DeleteDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockClient)(nil).DeleteDomain), varargs...)\n}\n\n// DeprecateDomain mocks base method.\nfunc (m *MockClient) DeprecateDomain(arg0 context.Context, arg1 *types.DeprecateDomainRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DeprecateDomain\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeprecateDomain indicates an expected call of DeprecateDomain.\nfunc (mr *MockClientMockRecorder) DeprecateDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeprecateDomain\", reflect.TypeOf((*MockClient)(nil).DeprecateDomain), varargs...)\n}\n\n// DescribeDomain mocks base method.\nfunc (m *MockClient) DescribeDomain(arg0 context.Context, arg1 *types.DescribeDomainRequest, arg2 ...yarpc.CallOption) (*types.DescribeDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeDomain\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeDomain indicates an expected call of DescribeDomain.\nfunc (mr *MockClientMockRecorder) DescribeDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeDomain\", reflect.TypeOf((*MockClient)(nil).DescribeDomain), varargs...)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockClient) DescribeTaskList(arg0 context.Context, arg1 *types.DescribeTaskListRequest, arg2 ...yarpc.CallOption) (*types.DescribeTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockClientMockRecorder) DescribeTaskList(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockClient)(nil).DescribeTaskList), varargs...)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockClient) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.DescribeWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockClientMockRecorder) DescribeWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).DescribeWorkflowExecution), varargs...)\n}\n\n// DiagnoseWorkflowExecution mocks base method.\nfunc (m *MockClient) DiagnoseWorkflowExecution(arg0 context.Context, arg1 *types.DiagnoseWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.DiagnoseWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DiagnoseWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.DiagnoseWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DiagnoseWorkflowExecution indicates an expected call of DiagnoseWorkflowExecution.\nfunc (mr *MockClientMockRecorder) DiagnoseWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DiagnoseWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).DiagnoseWorkflowExecution), varargs...)\n}\n\n// FailoverDomain mocks base method.\nfunc (m *MockClient) FailoverDomain(arg0 context.Context, arg1 *types.FailoverDomainRequest, arg2 ...yarpc.CallOption) (*types.FailoverDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"FailoverDomain\", varargs...)\n\tret0, _ := ret[0].(*types.FailoverDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FailoverDomain indicates an expected call of FailoverDomain.\nfunc (mr *MockClientMockRecorder) FailoverDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailoverDomain\", reflect.TypeOf((*MockClient)(nil).FailoverDomain), varargs...)\n}\n\n// GetClusterInfo mocks base method.\nfunc (m *MockClient) GetClusterInfo(arg0 context.Context, arg1 ...yarpc.CallOption) (*types.ClusterInfo, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0}\n\tfor _, a := range arg1 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetClusterInfo\", varargs...)\n\tret0, _ := ret[0].(*types.ClusterInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetClusterInfo indicates an expected call of GetClusterInfo.\nfunc (mr *MockClientMockRecorder) GetClusterInfo(arg0 any, arg1 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0}, arg1...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClusterInfo\", reflect.TypeOf((*MockClient)(nil).GetClusterInfo), varargs...)\n}\n\n// GetSearchAttributes mocks base method.\nfunc (m *MockClient) GetSearchAttributes(arg0 context.Context, arg1 ...yarpc.CallOption) (*types.GetSearchAttributesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0}\n\tfor _, a := range arg1 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetSearchAttributes\", varargs...)\n\tret0, _ := ret[0].(*types.GetSearchAttributesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetSearchAttributes indicates an expected call of GetSearchAttributes.\nfunc (mr *MockClientMockRecorder) GetSearchAttributes(arg0 any, arg1 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0}, arg1...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSearchAttributes\", reflect.TypeOf((*MockClient)(nil).GetSearchAttributes), varargs...)\n}\n\n// GetTaskListsByDomain mocks base method.\nfunc (m *MockClient) GetTaskListsByDomain(arg0 context.Context, arg1 *types.GetTaskListsByDomainRequest, arg2 ...yarpc.CallOption) (*types.GetTaskListsByDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", varargs...)\n\tret0, _ := ret[0].(*types.GetTaskListsByDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListsByDomain indicates an expected call of GetTaskListsByDomain.\nfunc (mr *MockClientMockRecorder) GetTaskListsByDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListsByDomain\", reflect.TypeOf((*MockClient)(nil).GetTaskListsByDomain), varargs...)\n}\n\n// GetWorkflowExecutionHistory mocks base method.\nfunc (m *MockClient) GetWorkflowExecutionHistory(arg0 context.Context, arg1 *types.GetWorkflowExecutionHistoryRequest, arg2 ...yarpc.CallOption) (*types.GetWorkflowExecutionHistoryResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionHistory\", varargs...)\n\tret0, _ := ret[0].(*types.GetWorkflowExecutionHistoryResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecutionHistory indicates an expected call of GetWorkflowExecutionHistory.\nfunc (mr *MockClientMockRecorder) GetWorkflowExecutionHistory(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecutionHistory\", reflect.TypeOf((*MockClient)(nil).GetWorkflowExecutionHistory), varargs...)\n}\n\n// ListArchivedWorkflowExecutions mocks base method.\nfunc (m *MockClient) ListArchivedWorkflowExecutions(arg0 context.Context, arg1 *types.ListArchivedWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.ListArchivedWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListArchivedWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.ListArchivedWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListArchivedWorkflowExecutions indicates an expected call of ListArchivedWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) ListArchivedWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListArchivedWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).ListArchivedWorkflowExecutions), varargs...)\n}\n\n// ListClosedWorkflowExecutions mocks base method.\nfunc (m *MockClient) ListClosedWorkflowExecutions(arg0 context.Context, arg1 *types.ListClosedWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.ListClosedWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.ListClosedWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutions indicates an expected call of ListClosedWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) ListClosedWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).ListClosedWorkflowExecutions), varargs...)\n}\n\n// ListDomains mocks base method.\nfunc (m *MockClient) ListDomains(arg0 context.Context, arg1 *types.ListDomainsRequest, arg2 ...yarpc.CallOption) (*types.ListDomainsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListDomains\", varargs...)\n\tret0, _ := ret[0].(*types.ListDomainsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDomains indicates an expected call of ListDomains.\nfunc (mr *MockClientMockRecorder) ListDomains(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDomains\", reflect.TypeOf((*MockClient)(nil).ListDomains), varargs...)\n}\n\n// ListFailoverHistory mocks base method.\nfunc (m *MockClient) ListFailoverHistory(arg0 context.Context, arg1 *types.ListFailoverHistoryRequest, arg2 ...yarpc.CallOption) (*types.ListFailoverHistoryResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListFailoverHistory\", varargs...)\n\tret0, _ := ret[0].(*types.ListFailoverHistoryResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListFailoverHistory indicates an expected call of ListFailoverHistory.\nfunc (mr *MockClientMockRecorder) ListFailoverHistory(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListFailoverHistory\", reflect.TypeOf((*MockClient)(nil).ListFailoverHistory), varargs...)\n}\n\n// ListOpenWorkflowExecutions mocks base method.\nfunc (m *MockClient) ListOpenWorkflowExecutions(arg0 context.Context, arg1 *types.ListOpenWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.ListOpenWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.ListOpenWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutions indicates an expected call of ListOpenWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) ListOpenWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).ListOpenWorkflowExecutions), varargs...)\n}\n\n// ListTaskListPartitions mocks base method.\nfunc (m *MockClient) ListTaskListPartitions(arg0 context.Context, arg1 *types.ListTaskListPartitionsRequest, arg2 ...yarpc.CallOption) (*types.ListTaskListPartitionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", varargs...)\n\tret0, _ := ret[0].(*types.ListTaskListPartitionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskListPartitions indicates an expected call of ListTaskListPartitions.\nfunc (mr *MockClientMockRecorder) ListTaskListPartitions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskListPartitions\", reflect.TypeOf((*MockClient)(nil).ListTaskListPartitions), varargs...)\n}\n\n// ListWorkflowExecutions mocks base method.\nfunc (m *MockClient) ListWorkflowExecutions(arg0 context.Context, arg1 *types.ListWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListWorkflowExecutions indicates an expected call of ListWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) ListWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).ListWorkflowExecutions), varargs...)\n}\n\n// PollForActivityTask mocks base method.\nfunc (m *MockClient) PollForActivityTask(arg0 context.Context, arg1 *types.PollForActivityTaskRequest, arg2 ...yarpc.CallOption) (*types.PollForActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", varargs...)\n\tret0, _ := ret[0].(*types.PollForActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForActivityTask indicates an expected call of PollForActivityTask.\nfunc (mr *MockClientMockRecorder) PollForActivityTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForActivityTask\", reflect.TypeOf((*MockClient)(nil).PollForActivityTask), varargs...)\n}\n\n// PollForDecisionTask mocks base method.\nfunc (m *MockClient) PollForDecisionTask(arg0 context.Context, arg1 *types.PollForDecisionTaskRequest, arg2 ...yarpc.CallOption) (*types.PollForDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", varargs...)\n\tret0, _ := ret[0].(*types.PollForDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForDecisionTask indicates an expected call of PollForDecisionTask.\nfunc (mr *MockClientMockRecorder) PollForDecisionTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForDecisionTask\", reflect.TypeOf((*MockClient)(nil).PollForDecisionTask), varargs...)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockClient) QueryWorkflow(arg0 context.Context, arg1 *types.QueryWorkflowRequest, arg2 ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", varargs...)\n\tret0, _ := ret[0].(*types.QueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockClientMockRecorder) QueryWorkflow(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockClient)(nil).QueryWorkflow), varargs...)\n}\n\n// RecordActivityTaskHeartbeat mocks base method.\nfunc (m *MockClient) RecordActivityTaskHeartbeat(arg0 context.Context, arg1 *types.RecordActivityTaskHeartbeatRequest, arg2 ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", varargs...)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeat indicates an expected call of RecordActivityTaskHeartbeat.\nfunc (mr *MockClientMockRecorder) RecordActivityTaskHeartbeat(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeat\", reflect.TypeOf((*MockClient)(nil).RecordActivityTaskHeartbeat), varargs...)\n}\n\n// RecordActivityTaskHeartbeatByID mocks base method.\nfunc (m *MockClient) RecordActivityTaskHeartbeatByID(arg0 context.Context, arg1 *types.RecordActivityTaskHeartbeatByIDRequest, arg2 ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeatByID\", varargs...)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeatByID indicates an expected call of RecordActivityTaskHeartbeatByID.\nfunc (mr *MockClientMockRecorder) RecordActivityTaskHeartbeatByID(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeatByID\", reflect.TypeOf((*MockClient)(nil).RecordActivityTaskHeartbeatByID), varargs...)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockClient) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.RefreshWorkflowTasksRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockClientMockRecorder) RefreshWorkflowTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockClient)(nil).RefreshWorkflowTasks), varargs...)\n}\n\n// RegisterDomain mocks base method.\nfunc (m *MockClient) RegisterDomain(arg0 context.Context, arg1 *types.RegisterDomainRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RegisterDomain\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RegisterDomain indicates an expected call of RegisterDomain.\nfunc (mr *MockClientMockRecorder) RegisterDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterDomain\", reflect.TypeOf((*MockClient)(nil).RegisterDomain), varargs...)\n}\n\n// RequestCancelWorkflowExecution mocks base method.\nfunc (m *MockClient) RequestCancelWorkflowExecution(arg0 context.Context, arg1 *types.RequestCancelWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestCancelWorkflowExecution indicates an expected call of RequestCancelWorkflowExecution.\nfunc (mr *MockClientMockRecorder) RequestCancelWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).RequestCancelWorkflowExecution), varargs...)\n}\n\n// ResetStickyTaskList mocks base method.\nfunc (m *MockClient) ResetStickyTaskList(arg0 context.Context, arg1 *types.ResetStickyTaskListRequest, arg2 ...yarpc.CallOption) (*types.ResetStickyTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", varargs...)\n\tret0, _ := ret[0].(*types.ResetStickyTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetStickyTaskList indicates an expected call of ResetStickyTaskList.\nfunc (mr *MockClientMockRecorder) ResetStickyTaskList(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetStickyTaskList\", reflect.TypeOf((*MockClient)(nil).ResetStickyTaskList), varargs...)\n}\n\n// ResetWorkflowExecution mocks base method.\nfunc (m *MockClient) ResetWorkflowExecution(arg0 context.Context, arg1 *types.ResetWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.ResetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.ResetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution.\nfunc (mr *MockClientMockRecorder) ResetWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).ResetWorkflowExecution), varargs...)\n}\n\n// RespondActivityTaskCanceled mocks base method.\nfunc (m *MockClient) RespondActivityTaskCanceled(arg0 context.Context, arg1 *types.RespondActivityTaskCanceledRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceled indicates an expected call of RespondActivityTaskCanceled.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCanceled(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceled\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCanceled), varargs...)\n}\n\n// RespondActivityTaskCanceledByID mocks base method.\nfunc (m *MockClient) RespondActivityTaskCanceledByID(arg0 context.Context, arg1 *types.RespondActivityTaskCanceledByIDRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceledByID\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceledByID indicates an expected call of RespondActivityTaskCanceledByID.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCanceledByID(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceledByID\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCanceledByID), varargs...)\n}\n\n// RespondActivityTaskCompleted mocks base method.\nfunc (m *MockClient) RespondActivityTaskCompleted(arg0 context.Context, arg1 *types.RespondActivityTaskCompletedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompleted indicates an expected call of RespondActivityTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCompleted), varargs...)\n}\n\n// RespondActivityTaskCompletedByID mocks base method.\nfunc (m *MockClient) RespondActivityTaskCompletedByID(arg0 context.Context, arg1 *types.RespondActivityTaskCompletedByIDRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompletedByID\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompletedByID indicates an expected call of RespondActivityTaskCompletedByID.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCompletedByID(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompletedByID\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCompletedByID), varargs...)\n}\n\n// RespondActivityTaskFailed mocks base method.\nfunc (m *MockClient) RespondActivityTaskFailed(arg0 context.Context, arg1 *types.RespondActivityTaskFailedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailed indicates an expected call of RespondActivityTaskFailed.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskFailed(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailed\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskFailed), varargs...)\n}\n\n// RespondActivityTaskFailedByID mocks base method.\nfunc (m *MockClient) RespondActivityTaskFailedByID(arg0 context.Context, arg1 *types.RespondActivityTaskFailedByIDRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailedByID\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailedByID indicates an expected call of RespondActivityTaskFailedByID.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskFailedByID(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailedByID\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskFailedByID), varargs...)\n}\n\n// RespondDecisionTaskCompleted mocks base method.\nfunc (m *MockClient) RespondDecisionTaskCompleted(arg0 context.Context, arg1 *types.RespondDecisionTaskCompletedRequest, arg2 ...yarpc.CallOption) (*types.RespondDecisionTaskCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(*types.RespondDecisionTaskCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondDecisionTaskCompleted indicates an expected call of RespondDecisionTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondDecisionTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondDecisionTaskCompleted), varargs...)\n}\n\n// RespondDecisionTaskFailed mocks base method.\nfunc (m *MockClient) RespondDecisionTaskFailed(arg0 context.Context, arg1 *types.RespondDecisionTaskFailedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondDecisionTaskFailed indicates an expected call of RespondDecisionTaskFailed.\nfunc (mr *MockClientMockRecorder) RespondDecisionTaskFailed(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskFailed\", reflect.TypeOf((*MockClient)(nil).RespondDecisionTaskFailed), varargs...)\n}\n\n// RespondQueryTaskCompleted mocks base method.\nfunc (m *MockClient) RespondQueryTaskCompleted(arg0 context.Context, arg1 *types.RespondQueryTaskCompletedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondQueryTaskCompleted indicates an expected call of RespondQueryTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondQueryTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondQueryTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondQueryTaskCompleted), varargs...)\n}\n\n// RestartWorkflowExecution mocks base method.\nfunc (m *MockClient) RestartWorkflowExecution(arg0 context.Context, arg1 *types.RestartWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.RestartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RestartWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.RestartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RestartWorkflowExecution indicates an expected call of RestartWorkflowExecution.\nfunc (mr *MockClientMockRecorder) RestartWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestartWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).RestartWorkflowExecution), varargs...)\n}\n\n// ScanWorkflowExecutions mocks base method.\nfunc (m *MockClient) ScanWorkflowExecutions(arg0 context.Context, arg1 *types.ListWorkflowExecutionsRequest, arg2 ...yarpc.CallOption) (*types.ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ScanWorkflowExecutions\", varargs...)\n\tret0, _ := ret[0].(*types.ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ScanWorkflowExecutions indicates an expected call of ScanWorkflowExecutions.\nfunc (mr *MockClientMockRecorder) ScanWorkflowExecutions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScanWorkflowExecutions\", reflect.TypeOf((*MockClient)(nil).ScanWorkflowExecutions), varargs...)\n}\n\n// SignalWithStartWorkflowExecution mocks base method.\nfunc (m *MockClient) SignalWithStartWorkflowExecution(arg0 context.Context, arg1 *types.SignalWithStartWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecution indicates an expected call of SignalWithStartWorkflowExecution.\nfunc (mr *MockClientMockRecorder) SignalWithStartWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).SignalWithStartWorkflowExecution), varargs...)\n}\n\n// SignalWithStartWorkflowExecutionAsync mocks base method.\nfunc (m *MockClient) SignalWithStartWorkflowExecutionAsync(arg0 context.Context, arg1 *types.SignalWithStartWorkflowExecutionAsyncRequest, arg2 ...yarpc.CallOption) (*types.SignalWithStartWorkflowExecutionAsyncResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecutionAsync\", varargs...)\n\tret0, _ := ret[0].(*types.SignalWithStartWorkflowExecutionAsyncResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecutionAsync indicates an expected call of SignalWithStartWorkflowExecutionAsync.\nfunc (mr *MockClientMockRecorder) SignalWithStartWorkflowExecutionAsync(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecutionAsync\", reflect.TypeOf((*MockClient)(nil).SignalWithStartWorkflowExecutionAsync), varargs...)\n}\n\n// SignalWorkflowExecution mocks base method.\nfunc (m *MockClient) SignalWorkflowExecution(arg0 context.Context, arg1 *types.SignalWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SignalWorkflowExecution indicates an expected call of SignalWorkflowExecution.\nfunc (mr *MockClientMockRecorder) SignalWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).SignalWorkflowExecution), varargs...)\n}\n\n// StartWorkflowExecution mocks base method.\nfunc (m *MockClient) StartWorkflowExecution(arg0 context.Context, arg1 *types.StartWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecution indicates an expected call of StartWorkflowExecution.\nfunc (mr *MockClientMockRecorder) StartWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).StartWorkflowExecution), varargs...)\n}\n\n// StartWorkflowExecutionAsync mocks base method.\nfunc (m *MockClient) StartWorkflowExecutionAsync(arg0 context.Context, arg1 *types.StartWorkflowExecutionAsyncRequest, arg2 ...yarpc.CallOption) (*types.StartWorkflowExecutionAsyncResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"StartWorkflowExecutionAsync\", varargs...)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionAsyncResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecutionAsync indicates an expected call of StartWorkflowExecutionAsync.\nfunc (mr *MockClientMockRecorder) StartWorkflowExecutionAsync(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecutionAsync\", reflect.TypeOf((*MockClient)(nil).StartWorkflowExecutionAsync), varargs...)\n}\n\n// TerminateWorkflowExecution mocks base method.\nfunc (m *MockClient) TerminateWorkflowExecution(arg0 context.Context, arg1 *types.TerminateWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TerminateWorkflowExecution indicates an expected call of TerminateWorkflowExecution.\nfunc (mr *MockClientMockRecorder) TerminateWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TerminateWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).TerminateWorkflowExecution), varargs...)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockClient) UpdateDomain(arg0 context.Context, arg1 *types.UpdateDomainRequest, arg2 ...yarpc.CallOption) (*types.UpdateDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateDomain\", varargs...)\n\tret0, _ := ret[0].(*types.UpdateDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockClientMockRecorder) UpdateDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockClient)(nil).UpdateDomain), varargs...)\n}\n"
  },
  {
    "path": "client/history/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"slices\"\n\t\"sync\"\n\n\t\"go.uber.org/yarpc\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/future\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ Client = (*clientImpl)(nil)\n\ntype (\n\tclientImpl struct {\n\t\tnumberOfShards    int\n\t\trpcMaxSizeInBytes int // This value currently only used in GetReplicationMessage API\n\t\ttokenSerializer   common.TaskTokenSerializer\n\t\tclient            Client\n\t\tpeerResolver      PeerResolver\n\t\tlogger            log.Logger\n\t}\n\n\tgetReplicationMessagesWithSize struct {\n\t\tresponse *types.GetReplicationMessagesResponse\n\t\tsize     int\n\t\tpeer     string\n\n\t\t// earliestCreationTime of replication tasks of response\n\t\tearliestCreationTime *int64\n\t}\n)\n\n// NewClient creates a new history service TChannel client\nfunc NewClient(\n\tnumberOfShards int,\n\trpcMaxSizeInBytes int,\n\tclient Client,\n\tpeerResolver PeerResolver,\n\tlogger log.Logger,\n) Client {\n\treturn &clientImpl{\n\t\tnumberOfShards:    numberOfShards,\n\t\trpcMaxSizeInBytes: rpcMaxSizeInBytes,\n\t\ttokenSerializer:   common.NewJSONTaskTokenSerializer(),\n\t\tclient:            client,\n\t\tpeerResolver:      peerResolver,\n\t\tlogger:            logger,\n\t}\n}\n\nfunc (c *clientImpl) StartWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (*types.StartWorkflowExecutionResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.StartRequest.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.StartWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) GetMutableState(\n\tctx context.Context,\n\trequest *types.GetMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetMutableStateResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.Execution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.GetMutableStateResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.GetMutableState(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) PollMutableState(\n\tctx context.Context,\n\trequest *types.PollMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (*types.PollMutableStateResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.Execution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.PollMutableStateResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.PollMutableState(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) DescribeHistoryHost(\n\tctx context.Context,\n\trequest *types.DescribeHistoryHostRequest,\n\topts ...yarpc.CallOption,\n) (*types.DescribeHistoryHostResponse, error) {\n\n\tvar err error\n\tvar peer string\n\n\tif request.ShardIDForHost != nil {\n\t\tpeer, err = c.peerResolver.FromShardID(int(request.GetShardIDForHost()))\n\t\tif err != nil {\n\t\t\tc.logger.Error(\"peer could not be resolved for host.\", tag.Error(err), tag.ShardID(int(request.GetShardIDForHost())))\n\t\t\treturn nil, err\n\t\t}\n\n\t} else if request.ExecutionForHost != nil {\n\t\tpeer, err = c.peerResolver.FromWorkflowID(request.ExecutionForHost.GetWorkflowID())\n\t\tif err != nil {\n\t\t\tc.logger.Error(\"peer could not be resolved for workflow.\", tag.Error(err), tag.WorkflowID(request.ExecutionForHost.GetWorkflowID()))\n\t\t\treturn nil, err\n\t\t}\n\n\t} else {\n\t\tpeer, err = c.peerResolver.FromHostAddress(request.GetHostAddress())\n\t\tif err != nil {\n\t\t\tc.logger.Error(\"peer could not be resolved for address.\", tag.Error(err), tag.Address(request.GetHostAddress()))\n\t\t\treturn nil, err\n\t\t}\n\n\t}\n\tvar response *types.DescribeHistoryHostResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.DescribeHistoryHost(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RemoveTask(\n\tctx context.Context,\n\trequest *types.RemoveTaskRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\terr = c.client.RemoveTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) CloseShard(\n\tctx context.Context,\n\trequest *types.CloseShardRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\terr = c.client.CloseShard(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c *clientImpl) ResetQueue(\n\tctx context.Context,\n\trequest *types.ResetQueueRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\terr = c.client.ResetQueue(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c *clientImpl) DescribeQueue(\n\tctx context.Context,\n\trequest *types.DescribeQueueRequest,\n\topts ...yarpc.CallOption,\n) (*types.DescribeQueueResponse, error) {\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.DescribeQueueResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.DescribeQueue(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) DescribeMutableState(\n\tctx context.Context,\n\trequest *types.DescribeMutableStateRequest,\n\topts ...yarpc.CallOption,\n) (*types.DescribeMutableStateResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.Execution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.DescribeMutableStateResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.DescribeMutableState(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) ResetStickyTaskList(\n\tctx context.Context,\n\trequest *types.HistoryResetStickyTaskListRequest,\n\topts ...yarpc.CallOption,\n) (*types.HistoryResetStickyTaskListResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.Execution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.HistoryResetStickyTaskListResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.ResetStickyTaskList(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) DescribeWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryDescribeWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (*types.DescribeWorkflowExecutionResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.Request.Execution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.DescribeWorkflowExecutionResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.DescribeWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RecordDecisionTaskStarted(\n\tctx context.Context,\n\trequest *types.RecordDecisionTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (*types.RecordDecisionTaskStartedResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.RecordDecisionTaskStartedResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.RecordDecisionTaskStarted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RecordActivityTaskStarted(\n\tctx context.Context,\n\trequest *types.RecordActivityTaskStartedRequest,\n\topts ...yarpc.CallOption,\n) (*types.RecordActivityTaskStartedResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.RecordActivityTaskStartedResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.RecordActivityTaskStarted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\trequest *types.HistoryRespondDecisionTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) (*types.HistoryRespondDecisionTaskCompletedResponse, error) {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.CompleteRequest.TaskToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.HistoryRespondDecisionTaskCompletedResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tresponse, err = c.client.RespondDecisionTaskCompleted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn response, err\n}\n\nfunc (c *clientImpl) RespondDecisionTaskFailed(\n\tctx context.Context,\n\trequest *types.HistoryRespondDecisionTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) error {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.FailedRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RespondDecisionTaskFailed(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) RespondActivityTaskCompleted(\n\tctx context.Context,\n\trequest *types.HistoryRespondActivityTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) error {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.CompleteRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RespondActivityTaskCompleted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) RespondActivityTaskFailed(\n\tctx context.Context,\n\trequest *types.HistoryRespondActivityTaskFailedRequest,\n\topts ...yarpc.CallOption,\n) error {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.FailedRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RespondActivityTaskFailed(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) RespondActivityTaskCanceled(\n\tctx context.Context,\n\trequest *types.HistoryRespondActivityTaskCanceledRequest,\n\topts ...yarpc.CallOption,\n) error {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.CancelRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RespondActivityTaskCanceled(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\trequest *types.HistoryRecordActivityTaskHeartbeatRequest,\n\topts ...yarpc.CallOption,\n) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\ttaskToken, err := c.tokenSerializer.Deserialize(request.HeartbeatRequest.TaskToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpeer, err := c.peerResolver.FromWorkflowID(taskToken.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.RecordActivityTaskHeartbeatResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.RecordActivityTaskHeartbeat(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.CancelRequest.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RequestCancelWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\treturn c.executeWithRedirect(ctx, peer, op)\n}\n\nfunc (c *clientImpl) SignalWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistorySignalWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.SignalRequest.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.SignalWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\n\treturn err\n}\n\nfunc (c *clientImpl) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistorySignalWithStartWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (*types.StartWorkflowExecutionResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.SignalWithStartRequest.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.SignalWithStartWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, err\n}\n\nfunc (c *clientImpl) RemoveSignalMutableState(\n\tctx context.Context,\n\trequest *types.RemoveSignalMutableStateRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RemoveSignalMutableState(ctx, request, yarpc.WithShardKey(peer))\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\n\treturn err\n}\n\nfunc (c *clientImpl) TerminateWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryTerminateWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.TerminateRequest.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.TerminateWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) ResetWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryResetWorkflowExecutionRequest,\n\topts ...yarpc.CallOption,\n) (*types.ResetWorkflowExecutionResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.ResetRequest.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.ResetWorkflowExecutionResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tresponse, err = c.client.ResetWorkflowExecution(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, err\n}\n\nfunc (c *clientImpl) ScheduleDecisionTask(\n\tctx context.Context,\n\trequest *types.ScheduleDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.ScheduleDecisionTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) RecordChildExecutionCompleted(\n\tctx context.Context,\n\trequest *types.RecordChildExecutionCompletedRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.WorkflowID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RecordChildExecutionCompleted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) ReplicateEventsV2(\n\tctx context.Context,\n\trequest *types.ReplicateEventsV2Request,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.WorkflowExecution.GetWorkflowID())\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.ReplicateEventsV2(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) SyncShardStatus(\n\tctx context.Context,\n\trequest *types.SyncShardStatusRequest,\n\topts ...yarpc.CallOption,\n) error {\n\n\t// we do not have a workflow ID here, instead, we have something even better\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.SyncShardStatus(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) SyncActivity(\n\tctx context.Context,\n\trequest *types.SyncActivityRequest,\n\topts ...yarpc.CallOption,\n) error {\n\n\tpeer, err := c.peerResolver.FromWorkflowID(request.GetWorkflowID())\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.SyncActivity(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) QueryWorkflow(\n\tctx context.Context,\n\trequest *types.HistoryQueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (*types.HistoryQueryWorkflowResponse, error) {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.GetRequest().GetExecution().GetWorkflowID())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar response *types.HistoryQueryWorkflowResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.QueryWorkflow(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) GetReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetReplicationMessagesResponse, error) {\n\trequestsByPeer := make(map[string]*types.GetReplicationMessagesRequest)\n\n\tfor _, token := range request.Tokens {\n\t\tpeer, err := c.peerResolver.FromShardID(int(token.GetShardID()))\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif _, ok := requestsByPeer[peer]; !ok {\n\t\t\trequestsByPeer[peer] = &types.GetReplicationMessagesRequest{\n\t\t\t\tClusterName: request.ClusterName,\n\t\t\t}\n\t\t}\n\n\t\treq := requestsByPeer[peer]\n\t\treq.Tokens = append(req.Tokens, token)\n\t}\n\n\tg := &errgroup.Group{}\n\tvar responseMutex sync.Mutex\n\tpeerResponses := make([]*getReplicationMessagesWithSize, 0, len(requestsByPeer))\n\n\tfor peer, req := range requestsByPeer {\n\t\tpeer, req := peer, req\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), c.logger, &e) }()\n\n\t\t\trequestContext, cancel := common.CreateChildContext(ctx, 0.05)\n\t\t\tdefer cancel()\n\n\t\t\trequestContext, responseInfo := rpc.ContextWithResponseInfo(requestContext)\n\t\t\tresp, err := c.client.GetReplicationMessages(requestContext, req, append(opts, yarpc.WithShardKey(peer))...)\n\t\t\tif err != nil {\n\t\t\t\tc.logger.Warn(\"Failed to get replication tasks from client\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t\ttag.ShardReplicationToken(req),\n\t\t\t\t)\n\t\t\t\t// Returns service busy error to notify replication\n\t\t\t\tif _, ok := err.(*types.ServiceBusyError); ok {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tresponseMutex.Lock()\n\t\t\tpeerResponses = append(peerResponses, &getReplicationMessagesWithSize{\n\t\t\t\tresponse:             resp,\n\t\t\t\tsize:                 responseInfo.Size,\n\t\t\t\tpeer:                 peer,\n\t\t\t\tearliestCreationTime: resp.GetEarliestCreationTime(),\n\t\t\t})\n\t\t\tresponseMutex.Unlock()\n\t\t\treturn nil\n\t\t})\n\t}\n\n\tif err := g.Wait(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn c.buildGetReplicationMessagesResponse(peerResponses), nil\n}\n\n// buildGetReplicationMessagesResponse builds a new GetReplicationMessagesResponse from peer responses\n// The response can be partial if the total size of the response exceeds the max size.\n// In this case, responses with oldest replication tasks will be returned\nfunc (c *clientImpl) buildGetReplicationMessagesResponse(peerResponses []*getReplicationMessagesWithSize) *types.GetReplicationMessagesResponse {\n\t// Peers with large response can cause the response to exceed the max size.\n\t// In this case, we need to skip some peer responses to make sure the result response size is within the limit.\n\t// To prevent a replication lag in the future, we should return the response with the oldest replication task.\n\t// So we sort the peer responses by the earliest creation time of the replication task.\n\t// If the earliest creation time is the same, we compare the size of the response.\n\t// This will sure that shards with the oldest replication tasks will be processed first.\n\tsortGetReplicationMessageWithSize(peerResponses)\n\n\tresponse := &types.GetReplicationMessagesResponse{MessagesByShard: make(map[int32]*types.ReplicationMessages)}\n\tresponseTotalSize := 0\n\trpcMaxResponseSize := c.rpcMaxSizeInBytes\n\tfor _, resp := range peerResponses {\n\t\tif (responseTotalSize + resp.size) >= rpcMaxResponseSize {\n\t\t\t// Log shards that did not fit for debugging purposes\n\t\t\tfor shardID := range resp.response.GetMessagesByShard() {\n\t\t\t\tc.logger.Warn(\"Replication messages did not fit in the response\",\n\t\t\t\t\ttag.ShardID(int(shardID)),\n\t\t\t\t\ttag.Address(resp.peer),\n\t\t\t\t\ttag.ResponseSize(resp.size),\n\t\t\t\t\ttag.ResponseTotalSize(responseTotalSize),\n\t\t\t\t\ttag.ResponseMaxSize(rpcMaxResponseSize),\n\t\t\t\t)\n\t\t\t}\n\n\t\t\t// return partial response if the response size exceeded supported max size\n\t\t\t// but continue with next peer response, as it may still fit\n\t\t\tcontinue\n\t\t}\n\n\t\tresponseTotalSize += resp.size\n\n\t\tfor shardID, tasks := range resp.response.GetMessagesByShard() {\n\t\t\tresponse.MessagesByShard[shardID] = tasks\n\t\t}\n\t}\n\n\treturn response\n}\n\n// sortGetReplicationMessageWithSize sorts the peer responses by the earliest creation time of the replication tasks\nfunc sortGetReplicationMessageWithSize(peerResponses []*getReplicationMessagesWithSize) {\n\tslices.SortStableFunc(peerResponses, cmpGetReplicationMessagesWithSize)\n}\n\n// cmpGetReplicationMessagesWithSize compares\n// two getReplicationMessagesWithSize objects by earliest creation time\n// it can be used as a comparison func for slices.SortStableFunc\n// if a, b, or their earliestCreationTime is nil, slices.SortStableFunc will put them to the end of a slice\n// otherwise it will compare the earliestCreationTime of the replication tasks\n// if earliestCreationTime is equal, it will compare the size of the response\nfunc cmpGetReplicationMessagesWithSize(a, b *getReplicationMessagesWithSize) int {\n\t// a > b\n\tif a == nil || a.earliestCreationTime == nil {\n\t\treturn 1\n\t}\n\t// a < b\n\tif b == nil || b.earliestCreationTime == nil {\n\t\treturn -1\n\t}\n\n\t// if both are not nil, compare the creation time\n\tif *a.earliestCreationTime < *b.earliestCreationTime {\n\t\treturn -1\n\t}\n\n\tif *a.earliestCreationTime > *b.earliestCreationTime {\n\t\treturn 1\n\t}\n\n\t// if both equal, compare the size\n\tif a.size < b.size {\n\t\treturn -1\n\t}\n\n\tif a.size > b.size {\n\t\treturn 1\n\t}\n\n\treturn 0\n}\n\nfunc (c *clientImpl) GetDLQReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetDLQReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetDLQReplicationMessagesResponse, error) {\n\t// All workflow IDs are in the same shard per request\n\tworkflowID := request.GetTaskInfos()[0].GetWorkflowID()\n\tpeer, err := c.peerResolver.FromWorkflowID(workflowID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn c.client.GetDLQReplicationMessages(\n\t\tctx,\n\t\trequest,\n\t\tappend(opts, yarpc.WithShardKey(peer))...,\n\t)\n}\n\nfunc (c *clientImpl) ReapplyEvents(\n\tctx context.Context,\n\trequest *types.HistoryReapplyEventsRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.GetRequest().GetWorkflowExecution().GetWorkflowID())\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.ReapplyEvents(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) CountDLQMessages(\n\tctx context.Context,\n\trequest *types.CountDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.HistoryCountDLQMessagesResponse, error) {\n\n\tpeers, err := c.peerResolver.GetAllPeers()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar mu sync.Mutex\n\tresponses := make([]*types.HistoryCountDLQMessagesResponse, 0, len(peers))\n\n\tg := &errgroup.Group{}\n\tfor _, peer := range peers {\n\t\tpeer := peer\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), c.logger, &e) }()\n\n\t\t\tresponse, err := c.client.CountDLQMessages(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\t\tif err == nil {\n\t\t\t\tmu.Lock()\n\t\t\t\tresponses = append(responses, response)\n\t\t\t\tmu.Unlock()\n\t\t\t}\n\n\t\t\treturn err\n\t\t})\n\t}\n\n\terr = g.Wait()\n\n\tentries := map[types.HistoryDLQCountKey]int64{}\n\tfor _, response := range responses {\n\t\tfor key, count := range response.Entries {\n\t\t\tentries[key] = count\n\t\t}\n\t}\n\treturn &types.HistoryCountDLQMessagesResponse{Entries: entries}, err\n}\n\nfunc (c *clientImpl) ReadDLQMessages(\n\tctx context.Context,\n\trequest *types.ReadDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.ReadDLQMessagesResponse, error) {\n\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn c.client.ReadDLQMessages(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n}\n\nfunc (c *clientImpl) PurgeDLQMessages(\n\tctx context.Context,\n\trequest *types.PurgeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) error {\n\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn c.client.PurgeDLQMessages(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n}\n\nfunc (c *clientImpl) MergeDLQMessages(\n\tctx context.Context,\n\trequest *types.MergeDLQMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.MergeDLQMessagesResponse, error) {\n\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn c.client.MergeDLQMessages(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n}\n\nfunc (c *clientImpl) RefreshWorkflowTasks(\n\tctx context.Context,\n\trequest *types.HistoryRefreshWorkflowTasksRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromWorkflowID(request.GetRequest().GetExecution().GetWorkflowID())\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context, peer string) error {\n\t\treturn c.client.RefreshWorkflowTasks(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t}\n\terr = c.executeWithRedirect(ctx, peer, op)\n\treturn err\n}\n\nfunc (c *clientImpl) NotifyFailoverMarkers(\n\tctx context.Context,\n\trequest *types.NotifyFailoverMarkersRequest,\n\topts ...yarpc.CallOption,\n) error {\n\trequestsByPeer := make(map[string]*types.NotifyFailoverMarkersRequest)\n\n\tfor _, token := range request.GetFailoverMarkerTokens() {\n\t\tmarker := token.GetFailoverMarker()\n\t\tpeer, err := c.peerResolver.FromDomainID(marker.GetDomainID())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif _, ok := requestsByPeer[peer]; !ok {\n\t\t\trequestsByPeer[peer] = &types.NotifyFailoverMarkersRequest{\n\t\t\t\tFailoverMarkerTokens: []*types.FailoverMarkerToken{},\n\t\t\t}\n\t\t}\n\n\t\treq := requestsByPeer[peer]\n\t\treq.FailoverMarkerTokens = append(req.FailoverMarkerTokens, token)\n\t}\n\n\tg := &errgroup.Group{}\n\tfor peer, req := range requestsByPeer {\n\t\tpeer, req := peer, req\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), c.logger, &e) }()\n\t\t\treturn c.client.NotifyFailoverMarkers(ctx, req, append(opts, yarpc.WithShardKey(peer))...)\n\t\t})\n\t}\n\n\treturn g.Wait()\n}\n\nfunc (c *clientImpl) GetCrossClusterTasks(\n\tctx context.Context,\n\trequest *types.GetCrossClusterTasksRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetCrossClusterTasksResponse, error) {\n\trequestByPeer := make(map[string]*types.GetCrossClusterTasksRequest)\n\tfor _, shardID := range request.GetShardIDs() {\n\t\tpeer, err := c.peerResolver.FromShardID(int(shardID))\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif _, ok := requestByPeer[peer]; !ok {\n\t\t\trequestByPeer[peer] = &types.GetCrossClusterTasksRequest{\n\t\t\t\tTargetCluster: request.TargetCluster,\n\t\t\t}\n\t\t}\n\t\trequestByPeer[peer].ShardIDs = append(requestByPeer[peer].ShardIDs, shardID)\n\t}\n\n\t// preserve 5% timeout to return partial of the result if context is timing out\n\tctx, cancel := common.CreateChildContext(ctx, 0.05)\n\tdefer cancel()\n\n\tfutureByPeer := make(map[string]future.Future, len(requestByPeer))\n\tfor peer, req := range requestByPeer {\n\t\tfuture, settable := future.NewFuture()\n\t\tgo func(ctx context.Context, peer string, req *types.GetCrossClusterTasksRequest) {\n\t\t\tsettable.Set(c.client.GetCrossClusterTasks(ctx, req, yarpc.WithShardKey(peer)))\n\t\t}(ctx, peer, req)\n\n\t\tfutureByPeer[peer] = future\n\t}\n\n\tresponse := &types.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       make(map[int32][]*types.CrossClusterTaskRequest),\n\t\tFailedCauseByShard: make(map[int32]types.GetTaskFailedCause),\n\t}\n\tfor peer, future := range futureByPeer {\n\t\tvar resp *types.GetCrossClusterTasksResponse\n\t\tif futureErr := future.Get(ctx, &resp); futureErr != nil {\n\t\t\tc.logger.Error(\"Failed to get cross cluster tasks\", tag.Error(futureErr))\n\t\t\tfor _, failedShardID := range requestByPeer[peer].ShardIDs {\n\t\t\t\tresponse.FailedCauseByShard[failedShardID] = common.ConvertErrToGetTaskFailedCause(futureErr)\n\t\t\t}\n\t\t} else {\n\t\t\tfor shardID, tasks := range resp.TasksByShard {\n\t\t\t\tresponse.TasksByShard[shardID] = tasks\n\t\t\t}\n\t\t\tfor shardID, failedCause := range resp.FailedCauseByShard {\n\t\t\t\tresponse.FailedCauseByShard[shardID] = failedCause\n\t\t\t}\n\t\t}\n\t}\n\t// not using a waitGroup for created goroutines as once all futures are unblocked,\n\t// those goroutines will eventually be completed\n\n\treturn response, nil\n}\n\nfunc (c *clientImpl) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\trequest *types.RespondCrossClusterTasksCompletedRequest,\n\topts ...yarpc.CallOption,\n) (*types.RespondCrossClusterTasksCompletedResponse, error) {\n\tpeer, err := c.peerResolver.FromShardID(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar response *types.RespondCrossClusterTasksCompletedResponse\n\top := func(ctx context.Context, peer string) error {\n\t\tvar err error\n\t\tresponse, err = c.client.RespondCrossClusterTasksCompleted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\t\treturn err\n\t}\n\n\terr = c.executeWithRedirect(ctx, peer, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\nfunc (c *clientImpl) GetFailoverInfo(\n\tctx context.Context,\n\trequest *types.GetFailoverInfoRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetFailoverInfoResponse, error) {\n\tpeer, err := c.peerResolver.FromDomainID(request.GetDomainID())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn c.client.GetFailoverInfo(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n}\n\nfunc (c *clientImpl) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (*types.RatelimitUpdateResponse, error) {\n\tif len(opts) == 0 {\n\t\t// unfortunately there is not really any way to ensure \"must have a shard key option\"\n\t\t// due to the closed nature of yarpc.CallOption's implementation, outside private-field-reading reflection.\n\t\t//\n\t\t// there are a few options to work around this, but they are currently rather high effort or\n\t\t// run into import cycles or similar.  risk is low for now as there is only one caller, and likely\n\t\t// never will be others.\n\t\treturn nil, fmt.Errorf(\"invalid arguments, missing yarpc.WithShardKey(peer) at a minimum\")\n\t}\n\n\t// intentionally does not use peer-redirecting retries, as keys in this request\n\t// could end up on multiple different hosts after a peer change.\n\treturn c.client.RatelimitUpdate(ctx, request, opts...)\n}\n\nfunc (c *clientImpl) executeWithRedirect(\n\tctx context.Context,\n\tpeer string,\n\top func(ctx context.Context, peer string) error,\n) error {\n\tvar err error\n\tif ctx == nil {\n\t\tctx = context.Background()\n\t}\nredirectLoop:\n\tfor {\n\t\terr = common.IsValidContext(ctx)\n\t\tif err != nil {\n\t\t\tbreak redirectLoop\n\t\t}\n\t\terr = op(ctx, peer)\n\t\tif err != nil {\n\t\t\tif s, ok := err.(*types.ShardOwnershipLostError); ok {\n\t\t\t\t// TODO: consider emitting a metric for number of redirects\n\t\t\t\tpeer, err = c.peerResolver.FromHostAddress(s.GetOwner())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcontinue redirectLoop\n\t\t\t}\n\t\t}\n\t\tbreak redirectLoop\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "client/history/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage history\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewClient(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tnumberOfShards := 10\n\trpcMaxSizeInBytes := 1024\n\tclient := NewMockClient(ctrl)\n\tpeerResolver := NewMockPeerResolver(ctrl)\n\tlogger := log.NewNoop()\n\n\tc := NewClient(numberOfShards, rpcMaxSizeInBytes, client, peerResolver, logger)\n\tassert.NotNil(t, c)\n}\n\nfunc TestClient_withResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\top        func(Client) (any, error)\n\t\tmock      func(*MockPeerResolver, *MockClient)\n\t\twant      any\n\t\twantError bool\n\t}{\n\t\t{\n\t\t\tname: \"StartWorkflowExecution\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.StartWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"GetMutableState\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetMutableState(context.Background(), &types.GetMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.GetMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.GetMutableStateResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"PollMutableState\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollMutableState(context.Background(), &types.PollMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().PollMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.PollMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.PollMutableStateResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"ResetWorkflowExecution\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ResetWorkflowExecution(context.Background(), &types.HistoryResetWorkflowExecutionRequest{\n\t\t\t\t\tResetRequest: &types.ResetWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.ResetWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.ResetWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeWorkflowExecution\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeWorkflowExecution(context.Background(), &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RecordActivityTaskHeartbeat\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.RecordActivityTaskHeartbeatResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.RecordActivityTaskHeartbeatResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RecordActivityTaskStarted\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordActivityTaskStarted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.RecordActivityTaskStartedResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.RecordActivityTaskStartedResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RecordDecisionTaskStarted\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordDecisionTaskStarted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.RecordDecisionTaskStartedResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.RecordDecisionTaskStartedResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationMessages\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetReplicationMessages(context.Background(), &types.GetReplicationMessagesRequest{\n\t\t\t\t\tTokens: []*types.ReplicationToken{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 102,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(100).Return(\"test-peer-0\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromShardID(101).Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromShardID(102).Return(\"test-peer-2\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-0\")}).\n\t\t\t\t\tReturn(&types.GetReplicationMessagesResponse{\n\t\t\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{100: {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(&types.GetReplicationMessagesResponse{\n\t\t\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{101: {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-2\")}).\n\t\t\t\t\tReturn(&types.GetReplicationMessagesResponse{\n\t\t\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{102: {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{\n\t\t\t\t\t100: {},\n\t\t\t\t\t101: {},\n\t\t\t\t\t102: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetDLQReplicationMessages\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetDLQReplicationMessages(context.Background(), &types.GetDLQReplicationMessagesRequest{\n\t\t\t\t\tTaskInfos: []*types.ReplicationTaskInfo{\n\t\t\t\t\t\t{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.GetDLQReplicationMessagesResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.GetDLQReplicationMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"ReadDLQMessages\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ReadDLQMessages(context.Background(), &types.ReadDLQMessagesRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ReadDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.ReadDLQMessagesResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.ReadDLQMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"MergeDLQMessages\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.MergeDLQMessages(context.Background(), &types.MergeDLQMessagesRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().MergeDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.MergeDLQMessagesResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.MergeDLQMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"GetFailoverInfo\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetFailoverInfo(context.Background(), &types.GetFailoverInfoRequest{\n\t\t\t\t\tDomainID: \"test-domain\",\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromDomainID(\"test-domain\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetFailoverInfo(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.GetFailoverInfoResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.GetFailoverInfoResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeHistoryHost by host address\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeHistoryHost(context.Background(), &types.DescribeHistoryHostRequest{\n\t\t\t\t\tHostAddress: common.StringPtr(\"test-host\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromHostAddress(\"test-host\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeHistoryHost(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeHistoryHostResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeHistoryHostResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeHistoryHost by workflow id\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeHistoryHost(context.Background(), &types.DescribeHistoryHostRequest{\n\t\t\t\t\tExecutionForHost: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\tHostAddress:      common.StringPtr(\"test-host\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeHistoryHost(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeHistoryHostResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeHistoryHostResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeHistoryHost by shard id\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeHistoryHost(context.Background(), &types.DescribeHistoryHostRequest{\n\t\t\t\t\tShardIDForHost:   common.Int32Ptr(123),\n\t\t\t\t\tExecutionForHost: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\tHostAddress:      common.StringPtr(\"test-host\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeHistoryHost(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeHistoryHostResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeHistoryHostResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeMutableState\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeMutableState(context.Background(), &types.DescribeMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeMutableStateResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeQueue\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeQueue(context.Background(), &types.DescribeQueueRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeQueue(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.DescribeQueueResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.DescribeQueueResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"CountDLQMessages\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.CountDLQMessages(context.Background(), &types.CountDLQMessagesRequest{})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().GetAllPeers().Return([]string{\"test-peer-0\", \"test-peer-1\"}, nil).Times(1)\n\t\t\t\tc.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-0\")}).\n\t\t\t\t\tReturn(&types.HistoryCountDLQMessagesResponse{\n\t\t\t\t\t\tEntries: map[types.HistoryDLQCountKey]int64{\n\t\t\t\t\t\t\t{ShardID: 1}: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tc.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(&types.HistoryCountDLQMessagesResponse{\n\t\t\t\t\t\tEntries: map[types.HistoryDLQCountKey]int64{\n\t\t\t\t\t\t\t{ShardID: 2}: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.HistoryCountDLQMessagesResponse{\n\t\t\t\tEntries: map[types.HistoryDLQCountKey]int64{\n\t\t\t\t\t{ShardID: 1}: 1,\n\t\t\t\t\t{ShardID: 2}: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"QueryWorkflow\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.QueryWorkflow(context.Background(), &types.HistoryQueryWorkflowRequest{\n\t\t\t\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.HistoryQueryWorkflowResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.HistoryQueryWorkflowResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"ResetStickyTaskList\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ResetStickyTaskList(context.Background(), &types.HistoryResetStickyTaskListRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ResetStickyTaskList(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.HistoryResetStickyTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.HistoryResetStickyTaskListResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondDecisionTaskCompleted\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.HistoryRespondDecisionTaskCompletedResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.HistoryRespondDecisionTaskCompletedResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondDecisionTaskCompleted\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.HistoryRespondDecisionTaskCompletedResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.HistoryRespondDecisionTaskCompletedResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalWithStartWorkflowExecution\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.SignalWithStartWorkflowExecution(context.Background(), &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.StartWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RatelimitUpdate\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RatelimitUpdate(context.Background(), &types.RatelimitUpdateRequest{\n\t\t\t\t\tAny: &types.Any{\n\t\t\t\t\t\tValueType: \"something\",\n\t\t\t\t\t\tValue:     []byte(\"data\"),\n\t\t\t\t\t},\n\t\t\t\t}, yarpc.WithShardKey(\"test-peer\"))\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tc.EXPECT().RatelimitUpdate(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(&types.RatelimitUpdateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.RatelimitUpdateResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecution peer resolve failure\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"\", fmt.Errorf(\"some error\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecution failure\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"StartWorkflowExecution failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecution redirected success with host lost error\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromHostAddress(\"host-test-peer-2\").Return(\"test-peer-2\", nil).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil, &types.ShardOwnershipLostError{\n\t\t\t\t\t\tMessage: \"test-peer-1 lost the shard\",\n\t\t\t\t\t\tOwner:   \"host-test-peer-2\",\n\t\t\t\t\t}).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-2\")}).\n\t\t\t\t\tReturn(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.StartWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecution redirected failed again with peer resolve\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromHostAddress(\"host-test-peer-2\").Return(\"\", fmt.Errorf(\"not found\")).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil, &types.ShardOwnershipLostError{\n\t\t\t\t\t\tMessage: \"test-peer-1 lost the shard\",\n\t\t\t\t\t\tOwner:   \"host-test-peer-2\",\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecution redirected failed again with error\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromHostAddress(\"host-test-peer-2\").Return(\"test-peer-2\", nil).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil, &types.ShardOwnershipLostError{\n\t\t\t\t\t\tMessage: \"test-peer-1 lost the shard\",\n\t\t\t\t\t\tOwner:   \"host-test-peer-2\",\n\t\t\t\t\t}).Times(1)\n\t\t\t\tc.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-2\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"some error\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"GetMutableState fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetMutableState(context.Background(), &types.GetMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"GetMutableState failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"PollMutableState fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollMutableState(context.Background(), &types.PollMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().PollMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"PollMutableState failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ResetWorkflowExecution fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ResetWorkflowExecution(context.Background(), &types.HistoryResetWorkflowExecutionRequest{\n\t\t\t\t\tResetRequest: &types.ResetWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"ResetWorkflowExecution failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeWorkflowExecution fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeWorkflowExecution(context.Background(), &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"DescribeWorkflowExecution failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RecordActivityTaskHeartbeat fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"RecordActivityTaskHeartbeat failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RecordActivityTaskStarted fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordActivityTaskStarted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"RecordActivityTaskStarted failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RecordDecisionTaskStarted fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordDecisionTaskStarted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"RecordDecisionTaskStarted failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationMessages fail open on unknow error\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetReplicationMessages(context.Background(), &types.GetReplicationMessagesRequest{\n\t\t\t\t\tTokens: []*types.ReplicationToken{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 101,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromShardID(100).Return(\"test-peer-0\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromShardID(101).Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-0\")}).\n\t\t\t\t\tReturn(&types.GetReplicationMessagesResponse{\n\t\t\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{100: {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"GetReplicationMessages failed\")).Times(1)\n\t\t\t},\n\t\t\twant: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{\n\t\t\t\t\t100: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationMessages fail open on unknow error\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetReplicationMessages(context.Background(), &types.GetReplicationMessagesRequest{\n\t\t\t\t\tTokens: []*types.ReplicationToken{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID: 101,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromShardID(100).Return(\"test-peer-0\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromShardID(101).Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-0\")}).\n\t\t\t\t\tReturn(&types.GetReplicationMessagesResponse{\n\t\t\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{100: {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tc.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil, &types.ServiceBusyError{}).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"GetDLQReplicationMessages fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetDLQReplicationMessages(context.Background(), &types.GetDLQReplicationMessagesRequest{\n\t\t\t\t\tTaskInfos: []*types.ReplicationTaskInfo{\n\t\t\t\t\t\t{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"GetDLQReplicationMessages failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ReadDLQMessages fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ReadDLQMessages(context.Background(), &types.ReadDLQMessagesRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ReadDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"ReadDLQMessages failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"MergeDLQMessages fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.MergeDLQMessages(context.Background(), &types.MergeDLQMessagesRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().MergeDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"MergeDLQMessages failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"GetFailoverInfo fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetFailoverInfo(context.Background(), &types.GetFailoverInfoRequest{\n\t\t\t\t\tDomainID: \"test-domain\",\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromDomainID(\"test-domain\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().GetFailoverInfo(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"GetFailoverInfo failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeMutableState fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeMutableState(context.Background(), &types.DescribeMutableStateRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(gomock.Any()).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"DescribeMutableState failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeQueue fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeQueue(context.Background(), &types.DescribeQueueRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().DescribeQueue(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"DescribeQueue failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"CountDLQMessages fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.CountDLQMessages(context.Background(), &types.CountDLQMessagesRequest{})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().GetAllPeers().Return([]string{\"test-peer\"}, nil).Times(1)\n\t\t\t\tc.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"CountDLQMessages failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"QueryWorkflow fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.QueryWorkflow(context.Background(), &types.HistoryQueryWorkflowRequest{\n\t\t\t\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(gomock.Any()).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"QueryWorkflow failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ResetStickyTaskList fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ResetStickyTaskList(context.Background(), &types.HistoryResetStickyTaskListRequest{\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ResetStickyTaskList(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"ResetStickyTaskList failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RespondDecisionTaskCompleted fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"RespondDecisionTaskCompleted failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RespondDecisionTaskCompleted fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"RespondDecisionTaskCompleted failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"SignalWithStartWorkflowExecution fail\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.SignalWithStartWorkflowExecution(context.Background(), &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// Add your mock expectations here\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"SignalWithStartWorkflowExecution failed\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RatelimitUpdate requires explicit shard key arg\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\t// same as successful call...\n\t\t\t\treturn c.RatelimitUpdate(context.Background(), &types.RatelimitUpdateRequest{\n\t\t\t\t\t// Peer: \"\", // intentionally the zero value\n\t\t\t\t\tAny: &types.Any{\n\t\t\t\t\t\tValueType: \"something\",\n\t\t\t\t\t\tValue:     []byte(\"data\"),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\t// no calls expected\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\ttt := tt\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockClient := NewMockClient(ctrl)\n\t\t\tmockPeerResolver := NewMockPeerResolver(ctrl)\n\t\t\tc := NewClient(10, 1024, mockClient, mockPeerResolver, log.NewNoop())\n\t\t\ttt.mock(mockPeerResolver, mockClient)\n\t\t\tres, err := tt.op(c)\n\t\t\tif tt.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, res)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestClient_withNoResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\top        func(Client) error\n\t\tmock      func(*MockPeerResolver, *MockClient)\n\t\twantError bool\n\t}{\n\t\t{\n\t\t\tname: \"RefreshWorkflowTasks\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RefreshWorkflowTasks(context.Background(), &types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PurgeDLQMessages\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.PurgeDLQMessages(context.Background(), &types.PurgeDLQMessagesRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().PurgeDLQMessages(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ReapplyEvents\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.ReapplyEvents(context.Background(), &types.HistoryReapplyEventsRequest{\n\t\t\t\t\tRequest: &types.ReapplyEventsRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TerminateWorkflowExecution\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.TerminateWorkflowExecution(context.Background(), &types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"NotifyFailoverMarkers\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.NotifyFailoverMarkers(context.Background(), &types.NotifyFailoverMarkersRequest{\n\t\t\t\t\tFailoverMarkerTokens: []*types.FailoverMarkerToken{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tFailoverMarker: &types.FailoverMarkerAttributes{\n\t\t\t\t\t\t\t\tDomainID: \"test-domain-0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tFailoverMarker: &types.FailoverMarkerAttributes{\n\t\t\t\t\t\t\t\tDomainID: \"test-domain-1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromDomainID(\"test-domain-0\").Return(\"test-peer-0\", nil).Times(1)\n\t\t\t\tp.EXPECT().FromDomainID(\"test-domain-1\").Return(\"test-peer-1\", nil).Times(1)\n\t\t\t\tc.EXPECT().NotifyFailoverMarkers(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-0\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t\tc.EXPECT().NotifyFailoverMarkers(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer-1\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CloseShard\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.CloseShard(context.Background(), &types.CloseShardRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().CloseShard(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RemoveTask\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RemoveTask(context.Background(), &types.RemoveTaskRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RemoveTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RecordChildExecutionCompleted\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RecordChildExecutionCompleted(context.Background(), &types.RecordChildExecutionCompletedRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RecordChildExecutionCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RefreshWorkflowTasks\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RefreshWorkflowTasks(context.Background(), &types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RemoveSignalMutableState\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RemoveSignalMutableState(context.Background(), &types.RemoveSignalMutableStateRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RemoveSignalMutableState(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ReplicateEventsV2\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.ReplicateEventsV2(context.Background(), &types.ReplicateEventsV2Request{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ReplicateEventsV2(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RequestCancelWorkflowExecution\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\t\t\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ResetQueue\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.ResetQueue(context.Background(), &types.ResetQueueRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ResetQueue(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondActivityTaskCanceled\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondActivityTaskCanceled(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondActivityTaskCompleted\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondActivityTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondActivityTaskFailed\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondActivityTaskFailed(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondDecisionTaskFailed\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondDecisionTaskFailed(context.Background(), &types.HistoryRespondDecisionTaskFailedRequest{\n\t\t\t\t\tFailedRequest: &types.RespondDecisionTaskFailedRequest{\n\t\t\t\t\t\tTaskToken: []byte(`{\"workflowId\": \"test-workflow\"}`),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().RespondDecisionTaskFailed(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ScheduleDecisionTask\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.ScheduleDecisionTask(context.Background(), &types.ScheduleDecisionTaskRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().ScheduleDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalWorkflowExecution\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.SignalWorkflowExecution(context.Background(), &types.HistorySignalWorkflowExecutionRequest{\n\t\t\t\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SyncActivity\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.SyncActivity(context.Background(), &types.SyncActivityRequest{\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromWorkflowID(\"test-workflow\").Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().SyncActivity(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SyncShardStatus\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.SyncShardStatus(context.Background(), &types.SyncShardStatusRequest{\n\t\t\t\t\tShardID: 123,\n\t\t\t\t})\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, c *MockClient) {\n\t\t\t\tp.EXPECT().FromShardID(123).Return(\"test-peer\", nil).Times(1)\n\t\t\t\tc.EXPECT().SyncShardStatus(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"test-peer\")}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\ttt := tt\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockClient := NewMockClient(ctrl)\n\t\t\tmockPeerResolver := NewMockPeerResolver(ctrl)\n\t\t\tc := NewClient(10, 1024, mockClient, mockPeerResolver, log.NewNoop())\n\t\t\ttt.mock(mockPeerResolver, mockClient)\n\t\t\terr := tt.op(c)\n\t\t\tif tt.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_cmpGetReplicationMessagesWithSize(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\ta, b *getReplicationMessagesWithSize\n\t\twant int\n\t}{\n\t\t\"both nil\": {\n\t\t\ta: nil, b: nil, want: 1,\n\t\t},\n\t\t\"a time is nil, b is nil\": {\n\t\t\ta: &getReplicationMessagesWithSize{earliestCreationTime: nil}, b: nil, want: 1,\n\t\t},\n\t\t\"a time is not nil, b is nil\": {\n\t\t\ta: &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)}, b: nil, want: -1,\n\t\t},\n\t\t\"a time is not nil, b time is nil\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: nil},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a time less b time\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a time greater b time\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\twant: 1,\n\t\t},\n\t\t\"a size less b size\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10), size: 10},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10), size: 20},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a size greater b size\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10), size: 20},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10), size: 10},\n\t\t\twant: 1,\n\t\t},\n\t\t\"a equal b\": {\n\t\t\ta:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    &getReplicationMessagesWithSize{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\twant: 0,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, c.want, cmpGetReplicationMessagesWithSize(c.a, c.b))\n\t\t})\n\t}\n}\n\nfunc Test_sortGetReplicationMessageWithSize(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tresponses []*getReplicationMessagesWithSize\n\t\twant      []*getReplicationMessagesWithSize\n\t}{\n\t\t\"empty\": {},\n\t\t\"multiple nil, non nil earliestCreationTime\": {\n\t\t\tresponses: []*getReplicationMessagesWithSize{\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\t},\n\t\t\twant: []*getReplicationMessagesWithSize{\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t},\n\t\t},\n\t\t\"multiple nil, non nil same earliestCreationTime, different size\": {\n\t\t\tresponses: []*getReplicationMessagesWithSize{\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 50},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 30},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t},\n\t\t\twant: []*getReplicationMessagesWithSize{\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 30},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 50},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tsortGetReplicationMessageWithSize(c.responses)\n\t\t\tassert.Equal(t, c.want, c.responses)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "client/history/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -package history github.com/uber/cadence/client/history Client\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/history_generated.go -v client=History\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/history_generated.go -v client=History\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/history_generated.go -v client=History\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/history_generated.go -v client=History -v package=historyv1 -v path=github.com/uber/cadence/.gen/proto/history/v1 -v prefix=History\n//go:generate gowrap gen -g -p . -i Client -t ../templates/thrift.tmpl -o ../wrappers/thrift/history_generated.go -v client=History -v prefix=History\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/history_generated.go -v client=History -v exclude=GetReplicationMessages|GetDLQReplicationMessages|CountDLQMessages|ReadDLQMessages|PurgeDLQMessages|MergeDLQMessages|GetCrossClusterTasks|GetFailoverInfo\n\n// Client is the interface exposed by history service client\ntype Client interface {\n\tCloseShard(context.Context, *types.CloseShardRequest, ...yarpc.CallOption) error\n\tDescribeHistoryHost(context.Context, *types.DescribeHistoryHostRequest, ...yarpc.CallOption) (*types.DescribeHistoryHostResponse, error)\n\tDescribeMutableState(context.Context, *types.DescribeMutableStateRequest, ...yarpc.CallOption) (*types.DescribeMutableStateResponse, error)\n\tDescribeQueue(context.Context, *types.DescribeQueueRequest, ...yarpc.CallOption) (*types.DescribeQueueResponse, error)\n\tDescribeWorkflowExecution(context.Context, *types.HistoryDescribeWorkflowExecutionRequest, ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error)\n\tGetCrossClusterTasks(context.Context, *types.GetCrossClusterTasksRequest, ...yarpc.CallOption) (*types.GetCrossClusterTasksResponse, error)\n\tGetDLQReplicationMessages(context.Context, *types.GetDLQReplicationMessagesRequest, ...yarpc.CallOption) (*types.GetDLQReplicationMessagesResponse, error)\n\tCountDLQMessages(context.Context, *types.CountDLQMessagesRequest, ...yarpc.CallOption) (*types.HistoryCountDLQMessagesResponse, error)\n\tGetMutableState(context.Context, *types.GetMutableStateRequest, ...yarpc.CallOption) (*types.GetMutableStateResponse, error)\n\tGetReplicationMessages(context.Context, *types.GetReplicationMessagesRequest, ...yarpc.CallOption) (*types.GetReplicationMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *types.MergeDLQMessagesRequest, ...yarpc.CallOption) (*types.MergeDLQMessagesResponse, error)\n\tNotifyFailoverMarkers(context.Context, *types.NotifyFailoverMarkersRequest, ...yarpc.CallOption) error\n\tPollMutableState(context.Context, *types.PollMutableStateRequest, ...yarpc.CallOption) (*types.PollMutableStateResponse, error)\n\tPurgeDLQMessages(context.Context, *types.PurgeDLQMessagesRequest, ...yarpc.CallOption) error\n\tQueryWorkflow(context.Context, *types.HistoryQueryWorkflowRequest, ...yarpc.CallOption) (*types.HistoryQueryWorkflowResponse, error)\n\tReadDLQMessages(context.Context, *types.ReadDLQMessagesRequest, ...yarpc.CallOption) (*types.ReadDLQMessagesResponse, error)\n\tReapplyEvents(context.Context, *types.HistoryReapplyEventsRequest, ...yarpc.CallOption) error\n\tRecordActivityTaskHeartbeat(context.Context, *types.HistoryRecordActivityTaskHeartbeatRequest, ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error)\n\tRecordActivityTaskStarted(context.Context, *types.RecordActivityTaskStartedRequest, ...yarpc.CallOption) (*types.RecordActivityTaskStartedResponse, error)\n\tRecordChildExecutionCompleted(context.Context, *types.RecordChildExecutionCompletedRequest, ...yarpc.CallOption) error\n\tRecordDecisionTaskStarted(context.Context, *types.RecordDecisionTaskStartedRequest, ...yarpc.CallOption) (*types.RecordDecisionTaskStartedResponse, error)\n\tRefreshWorkflowTasks(context.Context, *types.HistoryRefreshWorkflowTasksRequest, ...yarpc.CallOption) error\n\tRemoveSignalMutableState(context.Context, *types.RemoveSignalMutableStateRequest, ...yarpc.CallOption) error\n\tRemoveTask(context.Context, *types.RemoveTaskRequest, ...yarpc.CallOption) error\n\tReplicateEventsV2(context.Context, *types.ReplicateEventsV2Request, ...yarpc.CallOption) error\n\tRequestCancelWorkflowExecution(context.Context, *types.HistoryRequestCancelWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tResetQueue(context.Context, *types.ResetQueueRequest, ...yarpc.CallOption) error\n\tResetStickyTaskList(context.Context, *types.HistoryResetStickyTaskListRequest, ...yarpc.CallOption) (*types.HistoryResetStickyTaskListResponse, error)\n\tResetWorkflowExecution(context.Context, *types.HistoryResetWorkflowExecutionRequest, ...yarpc.CallOption) (*types.ResetWorkflowExecutionResponse, error)\n\tRespondActivityTaskCanceled(context.Context, *types.HistoryRespondActivityTaskCanceledRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskCompleted(context.Context, *types.HistoryRespondActivityTaskCompletedRequest, ...yarpc.CallOption) error\n\tRespondActivityTaskFailed(context.Context, *types.HistoryRespondActivityTaskFailedRequest, ...yarpc.CallOption) error\n\tRespondCrossClusterTasksCompleted(context.Context, *types.RespondCrossClusterTasksCompletedRequest, ...yarpc.CallOption) (*types.RespondCrossClusterTasksCompletedResponse, error)\n\tRespondDecisionTaskCompleted(context.Context, *types.HistoryRespondDecisionTaskCompletedRequest, ...yarpc.CallOption) (*types.HistoryRespondDecisionTaskCompletedResponse, error)\n\tRespondDecisionTaskFailed(context.Context, *types.HistoryRespondDecisionTaskFailedRequest, ...yarpc.CallOption) error\n\tScheduleDecisionTask(context.Context, *types.ScheduleDecisionTaskRequest, ...yarpc.CallOption) error\n\tSignalWithStartWorkflowExecution(context.Context, *types.HistorySignalWithStartWorkflowExecutionRequest, ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error)\n\tSignalWorkflowExecution(context.Context, *types.HistorySignalWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tStartWorkflowExecution(context.Context, *types.HistoryStartWorkflowExecutionRequest, ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error)\n\tSyncActivity(context.Context, *types.SyncActivityRequest, ...yarpc.CallOption) error\n\tSyncShardStatus(context.Context, *types.SyncShardStatusRequest, ...yarpc.CallOption) error\n\tTerminateWorkflowExecution(context.Context, *types.HistoryTerminateWorkflowExecutionRequest, ...yarpc.CallOption) error\n\tGetFailoverInfo(context.Context, *types.GetFailoverInfoRequest, ...yarpc.CallOption) (*types.GetFailoverInfoResponse, error)\n\n\t// RatelimitUpdate pushes usage info for the passed ratelimit keys, and requests updated weight info from aggregating hosts.\n\t// Exact semantics beyond this depend on the load-balanced ratelimit implementation.\n\t//\n\t// A peer (via yarpc.WithShardkey) MUST be determined before calling and passed in yarpc opts,\n\t// and unlike most endpoints this will NOT be forwarded to a new peer if the ring membership changes.\n\t// To correctly forward keys to the new hosts, they must be re-sharded to find their new hosts.\n\tRatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (*types.RatelimitUpdateResponse, error)\n}\n"
  },
  {
    "path": "client/history/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package history -source interface.go -destination interface_mock.go -package history github.com/uber/cadence/client/history Client\n//\n\n// Package history is a generated GoMock package.\npackage history\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// CloseShard mocks base method.\nfunc (m *MockClient) CloseShard(arg0 context.Context, arg1 *types.CloseShardRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CloseShard\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CloseShard indicates an expected call of CloseShard.\nfunc (mr *MockClientMockRecorder) CloseShard(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseShard\", reflect.TypeOf((*MockClient)(nil).CloseShard), varargs...)\n}\n\n// CountDLQMessages mocks base method.\nfunc (m *MockClient) CountDLQMessages(arg0 context.Context, arg1 *types.CountDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.HistoryCountDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CountDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.HistoryCountDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountDLQMessages indicates an expected call of CountDLQMessages.\nfunc (mr *MockClientMockRecorder) CountDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountDLQMessages\", reflect.TypeOf((*MockClient)(nil).CountDLQMessages), varargs...)\n}\n\n// DescribeHistoryHost mocks base method.\nfunc (m *MockClient) DescribeHistoryHost(arg0 context.Context, arg1 *types.DescribeHistoryHostRequest, arg2 ...yarpc.CallOption) (*types.DescribeHistoryHostResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeHistoryHostResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeHistoryHost indicates an expected call of DescribeHistoryHost.\nfunc (mr *MockClientMockRecorder) DescribeHistoryHost(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeHistoryHost\", reflect.TypeOf((*MockClient)(nil).DescribeHistoryHost), varargs...)\n}\n\n// DescribeMutableState mocks base method.\nfunc (m *MockClient) DescribeMutableState(arg0 context.Context, arg1 *types.DescribeMutableStateRequest, arg2 ...yarpc.CallOption) (*types.DescribeMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeMutableState\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeMutableState indicates an expected call of DescribeMutableState.\nfunc (mr *MockClientMockRecorder) DescribeMutableState(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeMutableState\", reflect.TypeOf((*MockClient)(nil).DescribeMutableState), varargs...)\n}\n\n// DescribeQueue mocks base method.\nfunc (m *MockClient) DescribeQueue(arg0 context.Context, arg1 *types.DescribeQueueRequest, arg2 ...yarpc.CallOption) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeQueue\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeQueue indicates an expected call of DescribeQueue.\nfunc (mr *MockClientMockRecorder) DescribeQueue(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeQueue\", reflect.TypeOf((*MockClient)(nil).DescribeQueue), varargs...)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockClient) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.HistoryDescribeWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockClientMockRecorder) DescribeWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).DescribeWorkflowExecution), varargs...)\n}\n\n// GetCrossClusterTasks mocks base method.\nfunc (m *MockClient) GetCrossClusterTasks(arg0 context.Context, arg1 *types.GetCrossClusterTasksRequest, arg2 ...yarpc.CallOption) (*types.GetCrossClusterTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetCrossClusterTasks\", varargs...)\n\tret0, _ := ret[0].(*types.GetCrossClusterTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCrossClusterTasks indicates an expected call of GetCrossClusterTasks.\nfunc (mr *MockClientMockRecorder) GetCrossClusterTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCrossClusterTasks\", reflect.TypeOf((*MockClient)(nil).GetCrossClusterTasks), varargs...)\n}\n\n// GetDLQReplicationMessages mocks base method.\nfunc (m *MockClient) GetDLQReplicationMessages(arg0 context.Context, arg1 *types.GetDLQReplicationMessagesRequest, arg2 ...yarpc.CallOption) (*types.GetDLQReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", varargs...)\n\tret0, _ := ret[0].(*types.GetDLQReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQReplicationMessages indicates an expected call of GetDLQReplicationMessages.\nfunc (mr *MockClientMockRecorder) GetDLQReplicationMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQReplicationMessages\", reflect.TypeOf((*MockClient)(nil).GetDLQReplicationMessages), varargs...)\n}\n\n// GetFailoverInfo mocks base method.\nfunc (m *MockClient) GetFailoverInfo(arg0 context.Context, arg1 *types.GetFailoverInfoRequest, arg2 ...yarpc.CallOption) (*types.GetFailoverInfoResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetFailoverInfo\", varargs...)\n\tret0, _ := ret[0].(*types.GetFailoverInfoResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetFailoverInfo indicates an expected call of GetFailoverInfo.\nfunc (mr *MockClientMockRecorder) GetFailoverInfo(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFailoverInfo\", reflect.TypeOf((*MockClient)(nil).GetFailoverInfo), varargs...)\n}\n\n// GetMutableState mocks base method.\nfunc (m *MockClient) GetMutableState(arg0 context.Context, arg1 *types.GetMutableStateRequest, arg2 ...yarpc.CallOption) (*types.GetMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetMutableState\", varargs...)\n\tret0, _ := ret[0].(*types.GetMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMutableState indicates an expected call of GetMutableState.\nfunc (mr *MockClientMockRecorder) GetMutableState(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMutableState\", reflect.TypeOf((*MockClient)(nil).GetMutableState), varargs...)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockClient) GetReplicationMessages(arg0 context.Context, arg1 *types.GetReplicationMessagesRequest, arg2 ...yarpc.CallOption) (*types.GetReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", varargs...)\n\tret0, _ := ret[0].(*types.GetReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockClientMockRecorder) GetReplicationMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockClient)(nil).GetReplicationMessages), varargs...)\n}\n\n// MergeDLQMessages mocks base method.\nfunc (m *MockClient) MergeDLQMessages(arg0 context.Context, arg1 *types.MergeDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.MergeDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.MergeDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MergeDLQMessages indicates an expected call of MergeDLQMessages.\nfunc (mr *MockClientMockRecorder) MergeDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeDLQMessages\", reflect.TypeOf((*MockClient)(nil).MergeDLQMessages), varargs...)\n}\n\n// NotifyFailoverMarkers mocks base method.\nfunc (m *MockClient) NotifyFailoverMarkers(arg0 context.Context, arg1 *types.NotifyFailoverMarkersRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"NotifyFailoverMarkers\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// NotifyFailoverMarkers indicates an expected call of NotifyFailoverMarkers.\nfunc (mr *MockClientMockRecorder) NotifyFailoverMarkers(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyFailoverMarkers\", reflect.TypeOf((*MockClient)(nil).NotifyFailoverMarkers), varargs...)\n}\n\n// PollMutableState mocks base method.\nfunc (m *MockClient) PollMutableState(arg0 context.Context, arg1 *types.PollMutableStateRequest, arg2 ...yarpc.CallOption) (*types.PollMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PollMutableState\", varargs...)\n\tret0, _ := ret[0].(*types.PollMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollMutableState indicates an expected call of PollMutableState.\nfunc (mr *MockClientMockRecorder) PollMutableState(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollMutableState\", reflect.TypeOf((*MockClient)(nil).PollMutableState), varargs...)\n}\n\n// PurgeDLQMessages mocks base method.\nfunc (m *MockClient) PurgeDLQMessages(arg0 context.Context, arg1 *types.PurgeDLQMessagesRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PurgeDLQMessages indicates an expected call of PurgeDLQMessages.\nfunc (mr *MockClientMockRecorder) PurgeDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PurgeDLQMessages\", reflect.TypeOf((*MockClient)(nil).PurgeDLQMessages), varargs...)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockClient) QueryWorkflow(arg0 context.Context, arg1 *types.HistoryQueryWorkflowRequest, arg2 ...yarpc.CallOption) (*types.HistoryQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", varargs...)\n\tret0, _ := ret[0].(*types.HistoryQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockClientMockRecorder) QueryWorkflow(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockClient)(nil).QueryWorkflow), varargs...)\n}\n\n// RatelimitUpdate mocks base method.\nfunc (m *MockClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (*types.RatelimitUpdateResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, request}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RatelimitUpdate\", varargs...)\n\tret0, _ := ret[0].(*types.RatelimitUpdateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RatelimitUpdate indicates an expected call of RatelimitUpdate.\nfunc (mr *MockClientMockRecorder) RatelimitUpdate(ctx, request any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, request}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RatelimitUpdate\", reflect.TypeOf((*MockClient)(nil).RatelimitUpdate), varargs...)\n}\n\n// ReadDLQMessages mocks base method.\nfunc (m *MockClient) ReadDLQMessages(arg0 context.Context, arg1 *types.ReadDLQMessagesRequest, arg2 ...yarpc.CallOption) (*types.ReadDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", varargs...)\n\tret0, _ := ret[0].(*types.ReadDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadDLQMessages indicates an expected call of ReadDLQMessages.\nfunc (mr *MockClientMockRecorder) ReadDLQMessages(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadDLQMessages\", reflect.TypeOf((*MockClient)(nil).ReadDLQMessages), varargs...)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockClient) ReapplyEvents(arg0 context.Context, arg1 *types.HistoryReapplyEventsRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockClientMockRecorder) ReapplyEvents(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockClient)(nil).ReapplyEvents), varargs...)\n}\n\n// RecordActivityTaskHeartbeat mocks base method.\nfunc (m *MockClient) RecordActivityTaskHeartbeat(arg0 context.Context, arg1 *types.HistoryRecordActivityTaskHeartbeatRequest, arg2 ...yarpc.CallOption) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", varargs...)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeat indicates an expected call of RecordActivityTaskHeartbeat.\nfunc (mr *MockClientMockRecorder) RecordActivityTaskHeartbeat(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeat\", reflect.TypeOf((*MockClient)(nil).RecordActivityTaskHeartbeat), varargs...)\n}\n\n// RecordActivityTaskStarted mocks base method.\nfunc (m *MockClient) RecordActivityTaskStarted(arg0 context.Context, arg1 *types.RecordActivityTaskStartedRequest, arg2 ...yarpc.CallOption) (*types.RecordActivityTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordActivityTaskStarted\", varargs...)\n\tret0, _ := ret[0].(*types.RecordActivityTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskStarted indicates an expected call of RecordActivityTaskStarted.\nfunc (mr *MockClientMockRecorder) RecordActivityTaskStarted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskStarted\", reflect.TypeOf((*MockClient)(nil).RecordActivityTaskStarted), varargs...)\n}\n\n// RecordChildExecutionCompleted mocks base method.\nfunc (m *MockClient) RecordChildExecutionCompleted(arg0 context.Context, arg1 *types.RecordChildExecutionCompletedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordChildExecutionCompleted\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordChildExecutionCompleted indicates an expected call of RecordChildExecutionCompleted.\nfunc (mr *MockClientMockRecorder) RecordChildExecutionCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordChildExecutionCompleted\", reflect.TypeOf((*MockClient)(nil).RecordChildExecutionCompleted), varargs...)\n}\n\n// RecordDecisionTaskStarted mocks base method.\nfunc (m *MockClient) RecordDecisionTaskStarted(arg0 context.Context, arg1 *types.RecordDecisionTaskStartedRequest, arg2 ...yarpc.CallOption) (*types.RecordDecisionTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RecordDecisionTaskStarted\", varargs...)\n\tret0, _ := ret[0].(*types.RecordDecisionTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordDecisionTaskStarted indicates an expected call of RecordDecisionTaskStarted.\nfunc (mr *MockClientMockRecorder) RecordDecisionTaskStarted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordDecisionTaskStarted\", reflect.TypeOf((*MockClient)(nil).RecordDecisionTaskStarted), varargs...)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockClient) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.HistoryRefreshWorkflowTasksRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockClientMockRecorder) RefreshWorkflowTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockClient)(nil).RefreshWorkflowTasks), varargs...)\n}\n\n// RemoveSignalMutableState mocks base method.\nfunc (m *MockClient) RemoveSignalMutableState(arg0 context.Context, arg1 *types.RemoveSignalMutableStateRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RemoveSignalMutableState\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveSignalMutableState indicates an expected call of RemoveSignalMutableState.\nfunc (mr *MockClientMockRecorder) RemoveSignalMutableState(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveSignalMutableState\", reflect.TypeOf((*MockClient)(nil).RemoveSignalMutableState), varargs...)\n}\n\n// RemoveTask mocks base method.\nfunc (m *MockClient) RemoveTask(arg0 context.Context, arg1 *types.RemoveTaskRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RemoveTask\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveTask indicates an expected call of RemoveTask.\nfunc (mr *MockClientMockRecorder) RemoveTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveTask\", reflect.TypeOf((*MockClient)(nil).RemoveTask), varargs...)\n}\n\n// ReplicateEventsV2 mocks base method.\nfunc (m *MockClient) ReplicateEventsV2(arg0 context.Context, arg1 *types.ReplicateEventsV2Request, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ReplicateEventsV2\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateEventsV2 indicates an expected call of ReplicateEventsV2.\nfunc (mr *MockClientMockRecorder) ReplicateEventsV2(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateEventsV2\", reflect.TypeOf((*MockClient)(nil).ReplicateEventsV2), varargs...)\n}\n\n// RequestCancelWorkflowExecution mocks base method.\nfunc (m *MockClient) RequestCancelWorkflowExecution(arg0 context.Context, arg1 *types.HistoryRequestCancelWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestCancelWorkflowExecution indicates an expected call of RequestCancelWorkflowExecution.\nfunc (mr *MockClientMockRecorder) RequestCancelWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).RequestCancelWorkflowExecution), varargs...)\n}\n\n// ResetQueue mocks base method.\nfunc (m *MockClient) ResetQueue(arg0 context.Context, arg1 *types.ResetQueueRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetQueue\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetQueue indicates an expected call of ResetQueue.\nfunc (mr *MockClientMockRecorder) ResetQueue(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetQueue\", reflect.TypeOf((*MockClient)(nil).ResetQueue), varargs...)\n}\n\n// ResetStickyTaskList mocks base method.\nfunc (m *MockClient) ResetStickyTaskList(arg0 context.Context, arg1 *types.HistoryResetStickyTaskListRequest, arg2 ...yarpc.CallOption) (*types.HistoryResetStickyTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", varargs...)\n\tret0, _ := ret[0].(*types.HistoryResetStickyTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetStickyTaskList indicates an expected call of ResetStickyTaskList.\nfunc (mr *MockClientMockRecorder) ResetStickyTaskList(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetStickyTaskList\", reflect.TypeOf((*MockClient)(nil).ResetStickyTaskList), varargs...)\n}\n\n// ResetWorkflowExecution mocks base method.\nfunc (m *MockClient) ResetWorkflowExecution(arg0 context.Context, arg1 *types.HistoryResetWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.ResetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.ResetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution.\nfunc (mr *MockClientMockRecorder) ResetWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).ResetWorkflowExecution), varargs...)\n}\n\n// RespondActivityTaskCanceled mocks base method.\nfunc (m *MockClient) RespondActivityTaskCanceled(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskCanceledRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceled indicates an expected call of RespondActivityTaskCanceled.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCanceled(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceled\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCanceled), varargs...)\n}\n\n// RespondActivityTaskCompleted mocks base method.\nfunc (m *MockClient) RespondActivityTaskCompleted(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskCompletedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompleted indicates an expected call of RespondActivityTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskCompleted), varargs...)\n}\n\n// RespondActivityTaskFailed mocks base method.\nfunc (m *MockClient) RespondActivityTaskFailed(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskFailedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailed indicates an expected call of RespondActivityTaskFailed.\nfunc (mr *MockClientMockRecorder) RespondActivityTaskFailed(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailed\", reflect.TypeOf((*MockClient)(nil).RespondActivityTaskFailed), varargs...)\n}\n\n// RespondCrossClusterTasksCompleted mocks base method.\nfunc (m *MockClient) RespondCrossClusterTasksCompleted(arg0 context.Context, arg1 *types.RespondCrossClusterTasksCompletedRequest, arg2 ...yarpc.CallOption) (*types.RespondCrossClusterTasksCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondCrossClusterTasksCompleted\", varargs...)\n\tret0, _ := ret[0].(*types.RespondCrossClusterTasksCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondCrossClusterTasksCompleted indicates an expected call of RespondCrossClusterTasksCompleted.\nfunc (mr *MockClientMockRecorder) RespondCrossClusterTasksCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondCrossClusterTasksCompleted\", reflect.TypeOf((*MockClient)(nil).RespondCrossClusterTasksCompleted), varargs...)\n}\n\n// RespondDecisionTaskCompleted mocks base method.\nfunc (m *MockClient) RespondDecisionTaskCompleted(arg0 context.Context, arg1 *types.HistoryRespondDecisionTaskCompletedRequest, arg2 ...yarpc.CallOption) (*types.HistoryRespondDecisionTaskCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(*types.HistoryRespondDecisionTaskCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondDecisionTaskCompleted indicates an expected call of RespondDecisionTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondDecisionTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondDecisionTaskCompleted), varargs...)\n}\n\n// RespondDecisionTaskFailed mocks base method.\nfunc (m *MockClient) RespondDecisionTaskFailed(arg0 context.Context, arg1 *types.HistoryRespondDecisionTaskFailedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondDecisionTaskFailed indicates an expected call of RespondDecisionTaskFailed.\nfunc (mr *MockClientMockRecorder) RespondDecisionTaskFailed(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskFailed\", reflect.TypeOf((*MockClient)(nil).RespondDecisionTaskFailed), varargs...)\n}\n\n// ScheduleDecisionTask mocks base method.\nfunc (m *MockClient) ScheduleDecisionTask(arg0 context.Context, arg1 *types.ScheduleDecisionTaskRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ScheduleDecisionTask\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ScheduleDecisionTask indicates an expected call of ScheduleDecisionTask.\nfunc (mr *MockClientMockRecorder) ScheduleDecisionTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScheduleDecisionTask\", reflect.TypeOf((*MockClient)(nil).ScheduleDecisionTask), varargs...)\n}\n\n// SignalWithStartWorkflowExecution mocks base method.\nfunc (m *MockClient) SignalWithStartWorkflowExecution(arg0 context.Context, arg1 *types.HistorySignalWithStartWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecution indicates an expected call of SignalWithStartWorkflowExecution.\nfunc (mr *MockClientMockRecorder) SignalWithStartWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).SignalWithStartWorkflowExecution), varargs...)\n}\n\n// SignalWorkflowExecution mocks base method.\nfunc (m *MockClient) SignalWorkflowExecution(arg0 context.Context, arg1 *types.HistorySignalWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SignalWorkflowExecution indicates an expected call of SignalWorkflowExecution.\nfunc (mr *MockClientMockRecorder) SignalWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).SignalWorkflowExecution), varargs...)\n}\n\n// StartWorkflowExecution mocks base method.\nfunc (m *MockClient) StartWorkflowExecution(arg0 context.Context, arg1 *types.HistoryStartWorkflowExecutionRequest, arg2 ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecution indicates an expected call of StartWorkflowExecution.\nfunc (mr *MockClientMockRecorder) StartWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).StartWorkflowExecution), varargs...)\n}\n\n// SyncActivity mocks base method.\nfunc (m *MockClient) SyncActivity(arg0 context.Context, arg1 *types.SyncActivityRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SyncActivity\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncActivity indicates an expected call of SyncActivity.\nfunc (mr *MockClientMockRecorder) SyncActivity(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncActivity\", reflect.TypeOf((*MockClient)(nil).SyncActivity), varargs...)\n}\n\n// SyncShardStatus mocks base method.\nfunc (m *MockClient) SyncShardStatus(arg0 context.Context, arg1 *types.SyncShardStatusRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SyncShardStatus\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncShardStatus indicates an expected call of SyncShardStatus.\nfunc (mr *MockClientMockRecorder) SyncShardStatus(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncShardStatus\", reflect.TypeOf((*MockClient)(nil).SyncShardStatus), varargs...)\n}\n\n// TerminateWorkflowExecution mocks base method.\nfunc (m *MockClient) TerminateWorkflowExecution(arg0 context.Context, arg1 *types.HistoryTerminateWorkflowExecutionRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TerminateWorkflowExecution indicates an expected call of TerminateWorkflowExecution.\nfunc (mr *MockClientMockRecorder) TerminateWorkflowExecution(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TerminateWorkflowExecution\", reflect.TypeOf((*MockClient)(nil).TerminateWorkflowExecution), varargs...)\n}\n"
  },
  {
    "path": "client/history/peer_resolver.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"fmt\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/history/lookup\"\n)\n\n// PeerResolver is used to resolve history peers.\n// Those are deployed instances of Cadence history services that participate in the cluster ring.\n// The resulting peer is simply an address of form ip:port where RPC calls can be routed to.\n//\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination peer_resolver_mock.go -package history github.com/uber/cadence/client/history PeerResolver\ntype PeerResolver interface {\n\tFromWorkflowID(workflowID string) (string, error)\n\tFromDomainID(domainID string) (string, error)\n\tFromShardID(shardID int) (string, error)\n\tFromHostAddress(hostAddress string) (string, error)\n\tGetAllPeers() ([]string, error)\n\n\t// GlobalRatelimitPeers partitions the ratelimit keys into map[yarpc peer][]limits_for_peer\n\tGlobalRatelimitPeers(ratelimits []string) (ratelimitsByPeer map[Peer][]string, err error)\n}\n\n// Peer is used to mark a string as the routing information to a peer process.\n//\n// This is essentially the host:port address of the peer to be contacted,\n// but it is meant to be treated as an opaque blob until given to yarpc\n// via ToYarpcShardKey.\ntype Peer string\n\nfunc (s Peer) ToYarpcShardKey() yarpc.CallOption {\n\treturn yarpc.WithShardKey(string(s))\n}\n\ntype peerResolver struct {\n\tnumberOfShards int\n\tresolver       membership.Resolver\n\tnamedPort      string // grpc or tchannel, depends on yarpc configuration\n}\n\n// NewPeerResolver creates a new history peer resolver.\nfunc NewPeerResolver(numberOfShards int, resolver membership.Resolver, namedPort string) PeerResolver {\n\treturn peerResolver{\n\t\tnumberOfShards: numberOfShards,\n\t\tresolver:       resolver,\n\t\tnamedPort:      namedPort,\n\t}\n}\n\n// FromWorkflowID resolves the history peer responsible for a given workflowID.\n// WorkflowID is converted to logical shardID using a consistent hash function.\n// FromShardID is used for further resolving.\nfunc (pr peerResolver) FromWorkflowID(workflowID string) (string, error) {\n\tshardID := common.WorkflowIDToHistoryShard(workflowID, pr.numberOfShards)\n\treturn pr.FromShardID(shardID)\n}\n\n// FromDomainID resolves the history peer responsible for a given domainID.\n// DomainID is converted to logical shardID using a consistent hash function.\n// FromShardID is used for further resolving.\nfunc (pr peerResolver) FromDomainID(domainID string) (string, error) {\n\tshardID := common.DomainIDToHistoryShard(domainID, pr.numberOfShards)\n\treturn pr.FromShardID(shardID)\n}\n\n// FromShardID resolves the history peer responsible for a given logical shardID.\n// It uses our membership provider to lookup which instance currently owns the given shard.\n// FromHostAddress is used for further resolving.\nfunc (pr peerResolver) FromShardID(shardID int) (string, error) {\n\thost, err := lookup.HistoryServerByShardID(pr.resolver, shardID)\n\tif err != nil {\n\t\treturn \"\", common.ToServiceTransientError(err)\n\t}\n\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\treturn peer, common.ToServiceTransientError(err)\n}\n\n// FromHostAddress resolves the final history peer responsible for the given host address.\n// The address is formed by adding port for specified transport\nfunc (pr peerResolver) FromHostAddress(hostAddress string) (string, error) {\n\thost, err := pr.resolver.LookupByAddress(service.History, hostAddress)\n\tif err != nil {\n\t\treturn \"\", common.ToServiceTransientError(err)\n\t}\n\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\treturn peer, common.ToServiceTransientError(err)\n}\n\n// GetAllPeers returns all history service peers in the cluster ring.\nfunc (pr peerResolver) GetAllPeers() ([]string, error) {\n\thosts, err := pr.resolver.Members(service.History)\n\tif err != nil {\n\t\treturn nil, common.ToServiceTransientError(err)\n\t}\n\tpeers := make([]string, 0, len(hosts))\n\tfor _, host := range hosts {\n\t\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\t\tif err != nil {\n\t\t\treturn nil, common.ToServiceTransientError(err)\n\t\t}\n\t\tpeers = append(peers, peer)\n\t}\n\treturn peers, nil\n}\n\nfunc (pr peerResolver) GlobalRatelimitPeers(ratelimits []string) (map[Peer][]string, error) {\n\t// History was selected simply because it already has a ring and an internal-only API.\n\t// Any service should be fine, it just needs to be shared by both ends of the system.\n\thosts, err := pr.resolver.Members(service.History)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"unable to get history peers: %w\", err)\n\t}\n\tif len(hosts) == 0 {\n\t\t// can occur when shutting down the only instance because it calls EvictSelf ASAP.\n\t\t// this is common in dev, but otherwise *probably* should not be possible in a healthy system.\n\t\treturn nil, fmt.Errorf(\"unable to get history peers: no peers available\")\n\t}\n\n\tresults := make(map[Peer][]string, len(hosts))\n\tinitialCapacity := len(ratelimits) / len(hosts)\n\t// add a small buffer to reduce copies, as this is only an estimate\n\tinitialCapacity += initialCapacity / 10\n\t// but don't use zero, that'd be pointless\n\tif initialCapacity < 1 {\n\t\tinitialCapacity = 1\n\t}\n\n\tfor _, r := range ratelimits {\n\t\t// figure out the destination for this ratelimit\n\t\thost, err := pr.resolver.Lookup(service.History, r)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\n\t\t\t\t\"unable to find host for ratelimit key %q: %w\", r, err)\n\t\t}\n\t\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\t\tif err != nil {\n\t\t\t// AFAICT should not happen unless the peer data is incomplete / the ring has no port info.\n\t\t\t// retry may work?  eventually?\n\t\t\treturn nil, fmt.Errorf(\n\t\t\t\t\"unable to get address from host: %s, %w\",\n\t\t\t\thost.String(), // HostInfo has a readable String(), it's good enough\n\t\t\t\terr,\n\t\t\t)\n\t\t}\n\n\t\t// add to the peer's partition\n\t\tcurrent := results[Peer(peer)]\n\t\tif len(current) == 0 {\n\t\t\tcurrent = make([]string, 0, initialCapacity)\n\t\t}\n\t\tcurrent = append(current, r)\n\t\tresults[Peer(peer)] = current\n\t}\n\treturn results, nil\n}\n"
  },
  {
    "path": "client/history/peer_resolver_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: peer_resolver.go\n//\n// Generated by this command:\n//\n//\tmockgen -package history -source peer_resolver.go -destination peer_resolver_mock.go -package history github.com/uber/cadence/client/history PeerResolver\n//\n\n// Package history is a generated GoMock package.\npackage history\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPeerResolver is a mock of PeerResolver interface.\ntype MockPeerResolver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerResolverMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerResolverMockRecorder is the mock recorder for MockPeerResolver.\ntype MockPeerResolverMockRecorder struct {\n\tmock *MockPeerResolver\n}\n\n// NewMockPeerResolver creates a new mock instance.\nfunc NewMockPeerResolver(ctrl *gomock.Controller) *MockPeerResolver {\n\tmock := &MockPeerResolver{ctrl: ctrl}\n\tmock.recorder = &MockPeerResolverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerResolver) EXPECT() *MockPeerResolverMockRecorder {\n\treturn m.recorder\n}\n\n// FromDomainID mocks base method.\nfunc (m *MockPeerResolver) FromDomainID(domainID string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromDomainID\", domainID)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromDomainID indicates an expected call of FromDomainID.\nfunc (mr *MockPeerResolverMockRecorder) FromDomainID(domainID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromDomainID\", reflect.TypeOf((*MockPeerResolver)(nil).FromDomainID), domainID)\n}\n\n// FromHostAddress mocks base method.\nfunc (m *MockPeerResolver) FromHostAddress(hostAddress string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromHostAddress\", hostAddress)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromHostAddress indicates an expected call of FromHostAddress.\nfunc (mr *MockPeerResolverMockRecorder) FromHostAddress(hostAddress any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromHostAddress\", reflect.TypeOf((*MockPeerResolver)(nil).FromHostAddress), hostAddress)\n}\n\n// FromShardID mocks base method.\nfunc (m *MockPeerResolver) FromShardID(shardID int) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromShardID\", shardID)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromShardID indicates an expected call of FromShardID.\nfunc (mr *MockPeerResolverMockRecorder) FromShardID(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromShardID\", reflect.TypeOf((*MockPeerResolver)(nil).FromShardID), shardID)\n}\n\n// FromWorkflowID mocks base method.\nfunc (m *MockPeerResolver) FromWorkflowID(workflowID string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromWorkflowID\", workflowID)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromWorkflowID indicates an expected call of FromWorkflowID.\nfunc (mr *MockPeerResolverMockRecorder) FromWorkflowID(workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromWorkflowID\", reflect.TypeOf((*MockPeerResolver)(nil).FromWorkflowID), workflowID)\n}\n\n// GetAllPeers mocks base method.\nfunc (m *MockPeerResolver) GetAllPeers() ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllPeers\")\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllPeers indicates an expected call of GetAllPeers.\nfunc (mr *MockPeerResolverMockRecorder) GetAllPeers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllPeers\", reflect.TypeOf((*MockPeerResolver)(nil).GetAllPeers))\n}\n\n// GlobalRatelimitPeers mocks base method.\nfunc (m *MockPeerResolver) GlobalRatelimitPeers(ratelimits []string) (map[Peer][]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GlobalRatelimitPeers\", ratelimits)\n\tret0, _ := ret[0].(map[Peer][]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GlobalRatelimitPeers indicates an expected call of GlobalRatelimitPeers.\nfunc (mr *MockPeerResolverMockRecorder) GlobalRatelimitPeers(ratelimits any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GlobalRatelimitPeers\", reflect.TypeOf((*MockPeerResolver)(nil).GlobalRatelimitPeers), ratelimits)\n}\n"
  },
  {
    "path": "client/history/peer_resolver_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"errors\"\n\t\"slices\"\n\t\"testing\"\n\n\t\"github.com/dgryski/go-farm\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber/ringpop-go/hashring\"\n\tgomock \"go.uber.org/mock/gomock\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestPeerResolver(t *testing.T) {\n\tnumShards := 123\n\tt.Run(\"FromIDs\", func(t *testing.T) {\n\t\tcontroller := gomock.NewController(t)\n\t\tserviceResolver := membership.NewMockResolver(controller)\n\t\tserviceResolver.EXPECT().Lookup(\n\t\t\tservice.History, string(rune(common.DomainIDToHistoryShard(\"domainID\", numShards)))).Return(\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"domainHost:123\",\n\t\t\t\t\"domainHost_123\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1234}),\n\t\t\tnil)\n\t\tserviceResolver.EXPECT().Lookup(service.History, string(rune(common.WorkflowIDToHistoryShard(\"workflowID\", numShards)))).Return(\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"workflowHost:123\",\n\t\t\t\t\"workflow\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235, membership.PortGRPC: 1666}), nil)\n\n\t\tserviceResolver.EXPECT().Lookup(service.History, string(rune(99))).Return(\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"shardHost:123\",\n\t\t\t\t\"shard_123\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235}),\n\t\t\tnil)\n\n\t\tserviceResolver.EXPECT().Lookup(service.History, string(rune(11))).Return(membership.HostInfo{}, assert.AnError)\n\n\t\tr := NewPeerResolver(numShards, serviceResolver, membership.PortTchannel)\n\n\t\tpeer, err := r.FromDomainID(\"domainID\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, \"domainHost:1234\", peer)\n\n\t\tpeer, err = r.FromWorkflowID(\"workflowID\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, \"workflowHost:1235\", peer)\n\n\t\tpeer, err = r.FromShardID(99)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, \"shardHost:1235\", peer)\n\n\t\t_, err = r.FromShardID(11)\n\t\tassert.Error(t, err)\n\t})\n\n\tt.Run(\"FromHostAddress\", func(t *testing.T) {\n\t\ttests := []struct {\n\t\t\tname      string\n\t\t\taddress   string\n\t\t\tmock      func(*membership.MockResolver)\n\t\t\twant      string\n\t\t\twantError bool\n\t\t}{\n\t\t\t{\n\t\t\t\tname:    \"success\",\n\t\t\t\taddress: \"addressHost:123\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().LookupByAddress(service.History, \"addressHost:123\").Return(\n\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\"addressHost:123\",\n\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235, membership.PortGRPC: 1666}),\n\t\t\t\t\t\tnil,\n\t\t\t\t\t)\n\t\t\t\t},\n\t\t\t\twant: \"addressHost:1235\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tname:    \"invalid address\",\n\t\t\t\taddress: \"invalid address\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().LookupByAddress(service.History, \"invalid address\").Return(\n\t\t\t\t\t\tmembership.HostInfo{},\n\t\t\t\t\t\terrors.New(\"host not found\"),\n\t\t\t\t\t)\n\t\t\t\t},\n\t\t\t\twantError: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tname:    \"fail on no port\",\n\t\t\t\taddress: \"addressHost:123\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().LookupByAddress(service.History, \"addressHost:123\").Return(\n\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\"addressHost:123\",\n\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\tmembership.PortMap{}),\n\t\t\t\t\t\tnil,\n\t\t\t\t\t)\n\t\t\t\t},\n\t\t\t\twantError: true,\n\t\t\t},\n\t\t}\n\n\t\tfor _, tt := range tests {\n\t\t\ttt := tt\n\t\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t\tcontroller := gomock.NewController(t)\n\t\t\t\tserviceResolver := membership.NewMockResolver(controller)\n\t\t\t\ttt.mock(serviceResolver)\n\t\t\t\tr := NewPeerResolver(numShards, serviceResolver, membership.PortTchannel)\n\t\t\t\tres, err := r.FromHostAddress(tt.address)\n\t\t\t\tif tt.wantError {\n\t\t\t\t\tassert.True(t, common.IsServiceTransientError(err))\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, tt.want, res)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t}\n\t\t\t})\n\t\t}\n\t})\n\n\tt.Run(\"GetAllPeers\", func(t *testing.T) {\n\t\ttests := []struct {\n\t\t\tname      string\n\t\t\tmock      func(*membership.MockResolver)\n\t\t\twant      []string\n\t\t\twantError bool\n\t\t}{\n\t\t\t{\n\t\t\t\tname: \"success\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().Members(service.History).Return(\n\t\t\t\t\t\t[]membership.HostInfo{\n\t\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\t\"host1:123\",\n\t\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235, membership.PortGRPC: 1666}),\n\t\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\t\"host2:123\",\n\t\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235, membership.PortGRPC: 1666}),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t)\n\t\t\t\t},\n\t\t\t\twant: []string{\"host1:1235\", \"host2:1235\"},\n\t\t\t},\n\t\t\t{\n\t\t\t\tname: \"failed on peer resolve\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().Members(service.History).Return(nil, assert.AnError)\n\n\t\t\t\t},\n\t\t\t\twantError: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tname: \"failed on no port\",\n\t\t\t\tmock: func(mr *membership.MockResolver) {\n\t\t\t\t\tmr.EXPECT().Members(service.History).Return(\n\t\t\t\t\t\t[]membership.HostInfo{\n\t\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\t\"host1:123\",\n\t\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235, membership.PortGRPC: 1666}),\n\t\t\t\t\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\t\t\t\t\"host2:123\",\n\t\t\t\t\t\t\t\t\"address\",\n\t\t\t\t\t\t\t\tmembership.PortMap{membership.PortGRPC: 1666}),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t)\n\t\t\t\t},\n\t\t\t\twantError: true,\n\t\t\t},\n\t\t}\n\n\t\tfor _, tt := range tests {\n\t\t\ttt := tt\n\t\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t\tcontroller := gomock.NewController(t)\n\t\t\t\tserviceResolver := membership.NewMockResolver(controller)\n\t\t\t\ttt.mock(serviceResolver)\n\t\t\t\tr := NewPeerResolver(numShards, serviceResolver, membership.PortTchannel)\n\t\t\t\tres, err := r.GetAllPeers()\n\t\t\t\tif tt.wantError {\n\t\t\t\t\tassert.True(t, common.IsServiceTransientError(err))\n\t\t\t\t} else {\n\t\t\t\t\tassert.ElementsMatch(t, tt.want, res)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t}\n\t\t\t})\n\t\t}\n\t})\n\tt.Run(\"GlobalRatelimitPeers\", func(t *testing.T) {\n\t\tpeers := []membership.HostInfo{\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"abc:123\",\n\t\t\t\t\"abc-id\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235}),\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"def:456\",\n\t\t\t\t\"def-id\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235}),\n\t\t\tmembership.NewDetailedHostInfo(\n\t\t\t\t\"xyz:789\",\n\t\t\t\t\"xyz-id\",\n\t\t\t\tmembership.PortMap{membership.PortTchannel: 1235}),\n\t\t}\n\t\tlimitKeys := []string{\n\t\t\t\"domain-w-async\",\n\t\t\t\"domain-x-user\",\n\t\t\t\"domain-x-worker\",\n\t\t\t\"domain-y-worker\",\n\t\t\t\"domain-z-visibility\",\n\t\t}\n\n\t\tt.Run(\"errs\", func(t *testing.T) {\n\t\t\tt.Run(\"no members\", func(t *testing.T) {\n\t\t\t\tresolver := membership.NewMockResolver(gomock.NewController(t))\n\t\t\t\tresolver.EXPECT().Members(service.History).Return(nil, errors.New(\"no members\"))\n\n\t\t\t\tr := NewPeerResolver(numShards, resolver, membership.PortTchannel)\n\t\t\t\t_, err := r.GlobalRatelimitPeers([]string{\"test\"})\n\t\t\t\tassert.ErrorContains(t, err, \"no members\")\n\t\t\t})\n\t\t\tt.Run(\"no host for key\", func(t *testing.T) {\n\t\t\t\tresolver := membership.NewMockResolver(gomock.NewController(t))\n\t\t\t\tresolver.EXPECT().Members(service.History).Return(peers, nil)\n\t\t\t\t// seems likely impossible, unless the service is unknown\n\t\t\t\tresolver.EXPECT().Lookup(service.History, gomock.Any()).Return(membership.HostInfo{}, errors.New(\"no host\"))\n\n\t\t\t\tr := NewPeerResolver(numShards, resolver, membership.PortTchannel)\n\t\t\t\t_, err := r.GlobalRatelimitPeers(limitKeys)\n\t\t\t\tassert.ErrorContains(t, err, \"no host\")\n\t\t\t})\n\t\t\tt.Run(\"no protocol\", func(t *testing.T) {\n\t\t\t\tresolver := membership.NewMockResolver(gomock.NewController(t))\n\t\t\t\tresolver.EXPECT().Members(service.History).Return(peers, nil)\n\t\t\t\tresolver.EXPECT().Lookup(service.History, gomock.Any()).Return(peers[0], nil)\n\n\t\t\t\t// request GRPC, but only configured for TChannel\n\t\t\t\tr := NewPeerResolver(numShards, resolver, membership.PortGRPC)\n\t\t\t\t_, err := r.GlobalRatelimitPeers(limitKeys)\n\t\t\t\tassert.ErrorContains(t, err, \"unable to get address\")\n\t\t\t})\n\t\t})\n\t\tt.Run(\"success\", func(t *testing.T) {\n\t\t\tresolver := membership.NewMockResolver(gomock.NewController(t))\n\t\t\tresolver.EXPECT().Members(service.History).Return(peers, nil).Times(2)\n\n\t\t\t// use a real hashring to resolve the keys.\n\t\t\t// partly because it's easy and removes the need to mock,\n\t\t\t// and partly because this must remain stable across hashring upgrades.\n\t\t\tring := hashring.New(farm.Fingerprint32, 100 /* arbitrary, but affects sharding */)\n\t\t\tfor _, p := range peers {\n\t\t\t\tring.AddMembers(p) // casts type\n\t\t\t}\n\t\t\tresolver.EXPECT().Lookup(service.History, gomock.Any()).DoAndReturn(func(service, key string) (membership.HostInfo, error) {\n\t\t\t\tpeer, ok := ring.Lookup(key)\n\t\t\t\trequire.True(t, ok, \"keys should always be findable\")\n\t\t\t\treturn find(t, peers, func(info membership.HostInfo) bool {\n\t\t\t\t\treturn info.GetAddress() == peer\n\t\t\t\t}), nil\n\t\t\t}).AnyTimes()\n\n\t\t\t// small sanity check: sharded response should return all inputs\n\t\t\tassertAllKeysPresent := func(t *testing.T, sharded map[Peer][]string, limits []string) {\n\t\t\t\tresponded := maps.Values(sharded)\n\t\t\t\tassert.ElementsMatchf(t,\n\t\t\t\t\tlimits,\n\t\t\t\t\tslices.Concat(responded...), // flatten the [][]string\n\t\t\t\t\t\"sharded response does not contain exactly the limits requested:\\n\\trequested: %v\\n\\tresponse: %v\",\n\t\t\t\t\tlimits, sharded)\n\t\t\t}\n\n\t\t\t// shard the keys to hosts\n\t\t\tr := NewPeerResolver(numShards, resolver, membership.PortTchannel)\n\t\t\tres, err := r.GlobalRatelimitPeers(limitKeys)\n\t\t\trequire.NoError(t, err)\n\t\t\tassertAllKeysPresent(t, res, limitKeys)\n\t\t\tassert.Equal(t, map[Peer][]string{ // determined experimentally, but needs to be stable\n\t\t\t\t\"abc:1235\": {\"domain-w-async\", \"domain-x-worker\", \"domain-y-worker\"},\n\t\t\t\t\"def:1235\": {\"domain-x-user\"},\n\t\t\t\t\"xyz:1235\": {\"domain-z-visibility\"},\n\t\t\t}, res, \"\")\n\n\t\t\t// request with another key, response should be the same but with a new key somewhere\n\t\t\toneMoreKey := append(limitKeys, \"domain-other-user\")\n\t\t\tres, err = r.GlobalRatelimitPeers(oneMoreKey)\n\t\t\trequire.NoError(t, err)\n\t\t\tassertAllKeysPresent(t, res, oneMoreKey)\n\t\t\tassert.Equal(t, map[Peer][]string{\n\t\t\t\t\"abc:1235\": {\"domain-w-async\", \"domain-x-worker\", \"domain-y-worker\"},\n\t\t\t\t\"def:1235\": {\"domain-x-user\"},\n\t\t\t\t\"xyz:1235\": {\"domain-z-visibility\", \"domain-other-user\"},\n\t\t\t}, res, \"\")\n\t\t})\n\t})\n}\n\n// finds a single element that matches, or fails.\n// surprisingly \"slices\" doesn't have any filter or find equivalent, only Index,\n// and either way I want to ensure no multi-matches.\nfunc find[T any](t *testing.T, elems []T, equals func(T) bool) T {\n\tvar zero T\n\tfiltered := filter(elems, equals)\n\tif len(filtered) == 0 {\n\t\tt.Fatalf(\"no matching %T elements found\", zero)\n\t\treturn zero // unreachable\n\t}\n\tif len(filtered) > 1 {\n\t\tt.Fatalf(\"found multiple matching %T elements: %v\", zero, filtered)\n\t\treturn zero // unreachable\n\t}\n\treturn filtered[0]\n}\n\nfunc filter[T any](elems []T, match func(T) bool) (filtered []T) {\n\tfor _, e := range elems {\n\t\tif match(e) {\n\t\t\tfiltered = append(filtered, e)\n\t\t}\n\t}\n\treturn filtered\n}\n"
  },
  {
    "path": "client/matching/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage matching\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"go.uber.org/yarpc\"\n\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/future\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ Client = (*clientImpl)(nil)\n\ntype clientImpl struct {\n\tclient       Client\n\tpeerResolver PeerResolver\n\tloadBalancer LoadBalancer\n\tprovider     PartitionConfigProvider\n}\n\n// NewClient creates a new history service TChannel client\nfunc NewClient(\n\tclient Client,\n\tpeerResolver PeerResolver,\n\tlb LoadBalancer,\n\tprovider PartitionConfigProvider,\n) Client {\n\treturn &clientImpl{\n\t\tclient:       client,\n\t\tpeerResolver: peerResolver,\n\t\tloadBalancer: lb,\n\t\tprovider:     provider,\n\t}\n}\n\nfunc (c *clientImpl) AddActivityTask(\n\tctx context.Context,\n\trequest *types.AddActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (*types.AddActivityTaskResponse, error) {\n\tpartition := c.loadBalancer.PickWritePartition(\n\t\tpersistence.TaskListTypeActivity,\n\t\trequest,\n\t)\n\toriginalTaskList := request.TaskList\n\trequest.TaskList = &types.TaskList{\n\t\tName: partition,\n\t\tKind: originalTaskList.Kind,\n\t}\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.AddActivityTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\t// ReadOnlyPartitionError indicates the partition is being drained - invalidate cache to force next request to route to root partition\n\t\tvar readOnlyErr *types.ReadOnlyPartitionError\n\t\tif errors.As(err, &readOnlyErr) {\n\t\t\tc.provider.InvalidatePartitionCache(request.GetDomainUUID(), *originalTaskList, persistence.TaskListTypeActivity)\n\t\t}\n\t\treturn nil, err\n\t}\n\trequest.TaskList = originalTaskList\n\tc.provider.UpdatePartitionConfig(\n\t\trequest.GetDomainUUID(),\n\t\t*request.TaskList,\n\t\tpersistence.TaskListTypeActivity,\n\t\tresp.PartitionConfig,\n\t)\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) AddDecisionTask(\n\tctx context.Context,\n\trequest *types.AddDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (*types.AddDecisionTaskResponse, error) {\n\tpartition := c.loadBalancer.PickWritePartition(\n\t\tpersistence.TaskListTypeDecision,\n\t\trequest,\n\t)\n\toriginalTaskList := request.TaskList\n\trequest.TaskList = &types.TaskList{\n\t\tName: partition,\n\t\tKind: originalTaskList.Kind,\n\t}\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.AddDecisionTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\t// ReadOnlyPartitionError indicates the partition is being drained - invalidate cache to force next request to route to root partition\n\t\tvar readOnlyErr *types.ReadOnlyPartitionError\n\t\tif errors.As(err, &readOnlyErr) {\n\t\t\tc.provider.InvalidatePartitionCache(request.GetDomainUUID(), *originalTaskList, persistence.TaskListTypeDecision)\n\t\t}\n\t\treturn nil, err\n\t}\n\trequest.TaskList = originalTaskList\n\tc.provider.UpdatePartitionConfig(\n\t\trequest.GetDomainUUID(),\n\t\t*request.TaskList,\n\t\tpersistence.TaskListTypeDecision,\n\t\tresp.PartitionConfig,\n\t)\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) PollForActivityTask(\n\tctx context.Context,\n\trequest *types.MatchingPollForActivityTaskRequest,\n\topts ...yarpc.CallOption,\n) (*types.MatchingPollForActivityTaskResponse, error) {\n\tpartition := c.loadBalancer.PickReadPartition(\n\t\tpersistence.TaskListTypeActivity,\n\t\trequest,\n\t\trequest.GetIsolationGroup(),\n\t)\n\toriginalTaskList := request.PollRequest.TaskList\n\trequest.PollRequest.TaskList = &types.TaskList{\n\t\tName: partition,\n\t\tKind: originalTaskList.Kind,\n\t}\n\tpeer, err := c.peerResolver.FromTaskList(request.PollRequest.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.PollForActivityTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, cadence_errors.NewPeerHostnameError(err, peer)\n\t}\n\n\trequest.PollRequest.TaskList = originalTaskList\n\tc.provider.UpdatePartitionConfig(\n\t\trequest.GetDomainUUID(),\n\t\t*request.PollRequest.GetTaskList(),\n\t\tpersistence.TaskListTypeActivity,\n\t\tresp.PartitionConfig,\n\t)\n\tc.loadBalancer.UpdateWeight(\n\t\tpersistence.TaskListTypeActivity,\n\t\trequest,\n\t\tpartition,\n\t\tresp.LoadBalancerHints,\n\t)\n\t// caller needs to know the actual partition for cancelling long poll, so modify the request to pass this information\n\trequest.PollRequest.TaskList.Name = partition\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) PollForDecisionTask(\n\tctx context.Context,\n\trequest *types.MatchingPollForDecisionTaskRequest,\n\topts ...yarpc.CallOption,\n) (*types.MatchingPollForDecisionTaskResponse, error) {\n\tpartition := c.loadBalancer.PickReadPartition(\n\t\tpersistence.TaskListTypeDecision,\n\t\trequest,\n\t\trequest.GetIsolationGroup(),\n\t)\n\toriginalTaskList := request.PollRequest.TaskList\n\trequest.PollRequest.TaskList = &types.TaskList{\n\t\tName: partition,\n\t\tKind: originalTaskList.Kind,\n\t}\n\tpeer, err := c.peerResolver.FromTaskList(request.PollRequest.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.PollForDecisionTask(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, cadence_errors.NewPeerHostnameError(err, peer)\n\t}\n\trequest.PollRequest.TaskList = originalTaskList\n\tc.provider.UpdatePartitionConfig(\n\t\trequest.GetDomainUUID(),\n\t\t*request.PollRequest.GetTaskList(),\n\t\tpersistence.TaskListTypeDecision,\n\t\tresp.PartitionConfig,\n\t)\n\tc.loadBalancer.UpdateWeight(\n\t\tpersistence.TaskListTypeDecision,\n\t\trequest,\n\t\tpartition,\n\t\tresp.LoadBalancerHints,\n\t)\n\t// caller needs to know the actual partition for cancelling long poll, so modify the request to pass this information\n\trequest.PollRequest.TaskList.Name = partition\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) QueryWorkflow(\n\tctx context.Context,\n\trequest *types.MatchingQueryWorkflowRequest,\n\topts ...yarpc.CallOption,\n) (*types.MatchingQueryWorkflowResponse, error) {\n\tpartition := c.loadBalancer.PickReadPartition(\n\t\tpersistence.TaskListTypeDecision,\n\t\trequest,\n\t\t\"\",\n\t)\n\toriginalTaskList := request.TaskList\n\trequest.TaskList = &types.TaskList{\n\t\tName: partition,\n\t\tKind: originalTaskList.Kind,\n\t}\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.QueryWorkflow(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trequest.TaskList = originalTaskList\n\tc.provider.UpdatePartitionConfig(\n\t\trequest.GetDomainUUID(),\n\t\t*request.TaskList,\n\t\tpersistence.TaskListTypeDecision,\n\t\tresp.PartitionConfig,\n\t)\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) RespondQueryTaskCompleted(\n\tctx context.Context,\n\trequest *types.MatchingRespondQueryTaskCompletedRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = c.client.RespondQueryTaskCompleted(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c *clientImpl) CancelOutstandingPoll(\n\tctx context.Context,\n\trequest *types.CancelOutstandingPollRequest,\n\topts ...yarpc.CallOption,\n) error {\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = c.client.CancelOutstandingPoll(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c *clientImpl) DescribeTaskList(\n\tctx context.Context,\n\trequest *types.MatchingDescribeTaskListRequest,\n\topts ...yarpc.CallOption,\n) (*types.DescribeTaskListResponse, error) {\n\tpeer, err := c.peerResolver.FromTaskList(request.DescRequest.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.DescribeTaskList(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) ListTaskListPartitions(\n\tctx context.Context,\n\trequest *types.MatchingListTaskListPartitionsRequest,\n\topts ...yarpc.CallOption,\n) (*types.ListTaskListPartitionsResponse, error) {\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.ListTaskListPartitions(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) GetTaskListsByDomain(\n\tctx context.Context,\n\trequest *types.GetTaskListsByDomainRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetTaskListsByDomainResponse, error) {\n\tpeers, err := c.peerResolver.GetAllPeers()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar futures []future.Future\n\tfor _, peer := range peers {\n\t\tfuture, settable := future.NewFuture()\n\t\tsettable.Set(c.client.GetTaskListsByDomain(ctx, request, append(opts, yarpc.WithShardKey(peer))...))\n\t\tfutures = append(futures, future)\n\t}\n\n\tdecisionTaskListMap := make(map[string]*types.DescribeTaskListResponse)\n\tactivityTaskListMap := make(map[string]*types.DescribeTaskListResponse)\n\tfor i, future := range futures {\n\t\tvar resp *types.GetTaskListsByDomainResponse\n\t\tif err = future.Get(ctx, &resp); err != nil {\n\t\t\treturn nil, cadence_errors.NewPeerHostnameError(err, peers[i])\n\t\t}\n\t\tfor name, tl := range resp.GetDecisionTaskListMap() {\n\t\t\tif _, ok := decisionTaskListMap[name]; !ok {\n\t\t\t\tdecisionTaskListMap[name] = tl\n\t\t\t} else {\n\t\t\t\tdecisionTaskListMap[name].Pollers = append(decisionTaskListMap[name].Pollers, tl.GetPollers()...)\n\t\t\t}\n\t\t}\n\t\tfor name, tl := range resp.GetActivityTaskListMap() {\n\t\t\tif _, ok := activityTaskListMap[name]; !ok {\n\t\t\t\tactivityTaskListMap[name] = tl\n\t\t\t} else {\n\t\t\t\tactivityTaskListMap[name].Pollers = append(activityTaskListMap[name].Pollers, tl.GetPollers()...)\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: decisionTaskListMap,\n\t\tActivityTaskListMap: activityTaskListMap,\n\t}, nil\n}\n\nfunc (c *clientImpl) UpdateTaskListPartitionConfig(\n\tctx context.Context,\n\trequest *types.MatchingUpdateTaskListPartitionConfigRequest,\n\topts ...yarpc.CallOption,\n) (*types.MatchingUpdateTaskListPartitionConfigResponse, error) {\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.UpdateTaskListPartitionConfig(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *clientImpl) RefreshTaskListPartitionConfig(\n\tctx context.Context,\n\trequest *types.MatchingRefreshTaskListPartitionConfigRequest,\n\topts ...yarpc.CallOption,\n) (*types.MatchingRefreshTaskListPartitionConfigResponse, error) {\n\tpeer, err := c.peerResolver.FromTaskList(request.TaskList.GetName())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := c.client.RefreshTaskListPartitionConfig(ctx, request, append(opts, yarpc.WithShardKey(peer))...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n"
  },
  {
    "path": "client/matching/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"context\"\n\tstdErrors \"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t_testDomainUUID = \"123\"\n\t_testDomain     = \"test-domain\"\n\t_testTaskList   = \"test-tasklist\"\n\t_testPartition  = \"test-partition\"\n)\n\ntype testCase struct {\n\tname          string\n\top            func(Client) (any, error)\n\tmock          func(*MockPeerResolver, *MockLoadBalancer, *MockClient, *MockPartitionConfigProvider)\n\twant          any\n\twantError     bool\n\tvalidateError func(*testing.T, error)\n}\n\nfunc TestNewClient(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclient := NewMockClient(ctrl)\n\tpeerResolver := NewMockPeerResolver(ctrl)\n\tloadbalancer := NewMockLoadBalancer(ctrl)\n\tprovider := NewMockPartitionConfigProvider(ctrl)\n\n\tc := NewClient(client, peerResolver, loadbalancer, provider)\n\tassert.NotNil(t, c)\n}\n\nfunc TestClient_withoutResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\top        func(Client) error\n\t\tmock      func(*MockPeerResolver, *MockLoadBalancer, *MockClient)\n\t\twantError bool\n\t}{\n\t\t{\n\t\t\tname: \"RespondQueryTaskCompleted\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondQueryTaskCompleted(context.Background(), testRespondQueryTaskCompletedRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().RespondQueryTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RespondQueryTaskCompleted - Error in resolving peer\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondQueryTaskCompleted(context.Background(), testRespondQueryTaskCompletedRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RespondQueryTaskCompleted - Error while responding to completion of QueryTask\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.RespondQueryTaskCompleted(context.Background(), testRespondQueryTaskCompletedRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().RespondQueryTaskCompleted(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"CancelOutstandingPoll\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.CancelOutstandingPoll(context.Background(), testCancelOutstandingPollRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().CancelOutstandingPoll(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CancelOutstandingPoll - Error in resolving peer\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.CancelOutstandingPoll(context.Background(), testCancelOutstandingPollRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"CancelOutstandingPoll - Error while cancelling outstanding poll\",\n\t\t\top: func(c Client) error {\n\t\t\t\treturn c.CancelOutstandingPoll(context.Background(), testCancelOutstandingPollRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().CancelOutstandingPoll(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\ttt := tt\n\t\tt.Run(tt.name, func(t *testing.T) {\n\n\t\t\t// setting up client\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tclient := NewMockClient(ctrl)\n\t\t\tpeerResolverMock := NewMockPeerResolver(ctrl)\n\t\t\tloadbalancerMock := NewMockLoadBalancer(ctrl)\n\t\t\tproviderMock := NewMockPartitionConfigProvider(ctrl)\n\t\t\ttt.mock(peerResolverMock, loadbalancerMock, client)\n\t\t\tc := NewClient(client, peerResolverMock, loadbalancerMock, providerMock)\n\n\t\t\terr := tt.op(c)\n\t\t\tif tt.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t})\n\t}\n}\n\nfunc TestClient_withResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\top            func(Client) (any, error)\n\t\tmock          func(*MockPeerResolver, *MockLoadBalancer, *MockClient, *MockPartitionConfigProvider)\n\t\twant          any\n\t\twantError     bool\n\t\tvalidateError func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"AddActivityTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddActivityTask(context.Background(), testAddActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeActivity, testAddActivityTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.AddActivityTaskResponse{}, nil)\n\t\t\t\tmp.EXPECT().UpdatePartitionConfig(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeActivity, nil)\n\t\t\t},\n\t\t\twant: &types.AddActivityTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"AddActivityTask - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddActivityTask(context.Background(), testAddActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeActivity, testAddActivityTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"AddActivityTask - Error while adding activity task\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddActivityTask(context.Background(), testAddActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeActivity, testAddActivityTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"AddActivityTask - ReadOnlyPartitionError triggers cache invalidation\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddActivityTask(context.Background(), testAddActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeActivity, testAddActivityTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, &types.ReadOnlyPartitionError{Message: \"partition drained\"})\n\t\t\t\tmp.EXPECT().InvalidatePartitionCache(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeActivity)\n\t\t\t},\n\t\t\twantError: true,\n\t\t\tvalidateError: func(t *testing.T, err error) {\n\t\t\t\tvar readOnlyErr *types.ReadOnlyPartitionError\n\t\t\t\tassert.True(t, stdErrors.As(err, &readOnlyErr))\n\t\t\t\tassert.Equal(t, \"partition drained\", readOnlyErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"AddDecisionTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddDecisionTask(context.Background(), testAddDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeDecision, testAddDecisionTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.AddDecisionTaskResponse{}, nil)\n\t\t\t\tmp.EXPECT().UpdatePartitionConfig(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeDecision, nil)\n\t\t\t},\n\t\t\twant: &types.AddDecisionTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"AddDecisionTask - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddDecisionTask(context.Background(), testAddDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeDecision, testAddDecisionTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"AddDecisionTask - Error while adding decision task\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddDecisionTask(context.Background(), testAddDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeDecision, testAddDecisionTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"AddDecisionTask - ReadOnlyPartitionError triggers cache invalidation\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.AddDecisionTask(context.Background(), testAddDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickWritePartition(persistence.TaskListTypeDecision, testAddDecisionTaskRequest()).Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, &types.ReadOnlyPartitionError{Message: \"partition drained\"})\n\t\t\t\tmp.EXPECT().InvalidatePartitionCache(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeDecision)\n\t\t\t},\n\t\t\twantError: true,\n\t\t\tvalidateError: func(t *testing.T, err error) {\n\t\t\t\tvar readOnlyErr *types.ReadOnlyPartitionError\n\t\t\t\tassert.True(t, stdErrors.As(err, &readOnlyErr))\n\t\t\t\tassert.Equal(t, \"partition drained\", readOnlyErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PollForActivityTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForActivityTask(context.Background(), testMatchingPollForActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeActivity, testMatchingPollForActivityTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.MatchingPollForActivityTaskResponse{}, nil)\n\t\t\t\tmp.EXPECT().UpdatePartitionConfig(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeActivity, nil)\n\t\t\t\tbalancer.EXPECT().UpdateWeight(persistence.TaskListTypeActivity, testMatchingPollForActivityTaskRequest(), _testPartition, nil)\n\t\t\t},\n\t\t\twant: &types.MatchingPollForActivityTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"PollForActivityTask - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForActivityTask(context.Background(), testMatchingPollForActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeActivity, testMatchingPollForActivityTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"PollForActivityTask - Error while polling for ActivityTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForActivityTask(context.Background(), testMatchingPollForActivityTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeActivity, testMatchingPollForActivityTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, &types.InternalServiceError{Message: \"test error\"})\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t\tvalidateError: func(t *testing.T, err error) {\n\t\t\t\tvar peerErr *errors.PeerHostnameError\n\t\t\t\tassert.True(t, stdErrors.As(err, &peerErr))\n\t\t\t\tassert.Equal(t, \"peer0\", peerErr.PeerHostname)\n\t\t\t\tvar internalErr *types.InternalServiceError\n\t\t\t\tassert.True(t, stdErrors.As(peerErr.WrappedError, &internalErr))\n\t\t\t\tassert.Equal(t, \"test error\", internalErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PollForDecisionTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForDecisionTask(context.Background(), testMatchingPollForDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingPollForDecisionTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.MatchingPollForDecisionTaskResponse{}, nil)\n\t\t\t\tmp.EXPECT().UpdatePartitionConfig(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeDecision, nil)\n\t\t\t\tbalancer.EXPECT().UpdateWeight(persistence.TaskListTypeDecision, testMatchingPollForDecisionTaskRequest(), _testPartition, nil)\n\t\t\t},\n\t\t\twant: &types.MatchingPollForDecisionTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"PollForDecisionTask - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForDecisionTask(context.Background(), testMatchingPollForDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingPollForDecisionTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"PollForDecisionTask - Error while polling for DecisionTask\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.PollForDecisionTask(context.Background(), testMatchingPollForDecisionTaskRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingPollForDecisionTaskRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, &types.InternalServiceError{Message: \"test error\"})\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t\tvalidateError: func(t *testing.T, err error) {\n\t\t\t\tvar peerErr *errors.PeerHostnameError\n\t\t\t\tassert.True(t, stdErrors.As(err, &peerErr))\n\t\t\t\tassert.Equal(t, \"peer0\", peerErr.PeerHostname)\n\t\t\t\tvar internalErr *types.InternalServiceError\n\t\t\t\tassert.True(t, stdErrors.As(peerErr.WrappedError, &internalErr))\n\t\t\t\tassert.Equal(t, \"test error\", internalErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"QueryWorkflow\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.QueryWorkflow(context.Background(), testMatchingQueryWorkflowRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingQueryWorkflowRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.MatchingQueryWorkflowResponse{}, nil)\n\t\t\t\tmp.EXPECT().UpdatePartitionConfig(_testDomainUUID, types.TaskList{Name: _testTaskList}, persistence.TaskListTypeDecision, nil)\n\t\t\t},\n\t\t\twant: &types.MatchingQueryWorkflowResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"QueryWorkflow - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.QueryWorkflow(context.Background(), testMatchingQueryWorkflowRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingQueryWorkflowRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"QueryWorkflow - Error while querying workflow\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.QueryWorkflow(context.Background(), testMatchingQueryWorkflowRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(persistence.TaskListTypeDecision, testMatchingQueryWorkflowRequest(), \"\").Return(_testPartition)\n\t\t\t\tp.EXPECT().FromTaskList(_testPartition).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeTaskList\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeTaskList(context.Background(), testMatchingDescribeTaskListRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.DescribeTaskListResponse{}, nil)\n\t\t\t},\n\t\t\twant: &types.DescribeTaskListResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeTaskList - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeTaskList(context.Background(), testMatchingDescribeTaskListRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeTaskList - Error while describing tasklist\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.DescribeTaskList(context.Background(), testMatchingDescribeTaskListRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ListTaskListPartitions\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ListTaskListPartitions(context.Background(), testMatchingListTaskListPartitionsRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().ListTaskListPartitions(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.ListTaskListPartitionsResponse{}, nil)\n\t\t\t},\n\t\t\twant: &types.ListTaskListPartitionsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"ListTaskListPartitions - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ListTaskListPartitions(context.Background(), testMatchingListTaskListPartitionsRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ListTaskListPartitions - Error while listing tasklist partitions\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.ListTaskListPartitions(context.Background(), testMatchingListTaskListPartitionsRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().ListTaskListPartitions(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"GetTaskListsByDomain\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetTaskListsByDomain(context.Background(), testGetTaskListsByDomainRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().GetAllPeers().Return([]string{\"peer0\", \"peer1\"}, nil)\n\t\t\t\tc.EXPECT().GetTaskListsByDomain(gomock.Any(), testGetTaskListsByDomainRequest(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.GetTaskListsByDomainResponse{}, nil)\n\t\t\t\tc.EXPECT().GetTaskListsByDomain(gomock.Any(), testGetTaskListsByDomainRequest(), []yarpc.CallOption{yarpc.WithShardKey(\"peer1\")}).Return(&types.GetTaskListsByDomainResponse{}, nil)\n\t\t\t},\n\t\t\twant: &types.GetTaskListsByDomainResponse{\n\t\t\t\tDecisionTaskListMap: make(map[string]*types.DescribeTaskListResponse),\n\t\t\t\tActivityTaskListMap: make(map[string]*types.DescribeTaskListResponse),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetTaskListsByDomain - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.GetTaskListsByDomain(context.Background(), testGetTaskListsByDomainRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().GetAllPeers().Return([]string{\"peer0\", \"peer1\"}, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateTaskListPartitionConfig\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.UpdateTaskListPartitionConfig(context.Background(), testMatchingUpdateTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.MatchingUpdateTaskListPartitionConfigResponse{}, nil)\n\t\t\t},\n\t\t\twant: &types.MatchingUpdateTaskListPartitionConfigResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateTaskListPartitionConfig - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.UpdateTaskListPartitionConfig(context.Background(), testMatchingUpdateTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateTaskListPartitionConfig - Error while listing tasklist partitions\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.UpdateTaskListPartitionConfig(context.Background(), testMatchingUpdateTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RefreshTaskListPartitionConfig\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RefreshTaskListPartitionConfig(context.Background(), testMatchingRefreshTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil)\n\t\t\t},\n\t\t\twant: &types.MatchingRefreshTaskListPartitionConfigResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"RefreshTaskListPartitionConfig - Error in resolving peer\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RefreshTaskListPartitionConfig(context.Background(), testMatchingRefreshTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"RefreshTaskListPartitionConfig - Error while listing tasklist partitions\",\n\t\t\top: func(c Client) (any, error) {\n\t\t\t\treturn c.RefreshTaskListPartitionConfig(context.Background(), testMatchingRefreshTaskListPartitionConfigRequest())\n\t\t\t},\n\t\t\tmock: func(p *MockPeerResolver, balancer *MockLoadBalancer, c *MockClient, mp *MockPartitionConfigProvider) {\n\t\t\t\tp.EXPECT().FromTaskList(_testTaskList).Return(\"peer0\", nil)\n\t\t\t\tc.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), gomock.Any(), []yarpc.CallOption{yarpc.WithShardKey(\"peer0\")}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantError: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\ttt := tt\n\t\tt.Run(tt.name, func(t *testing.T) {\n\n\t\t\t// setting up client\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tclient := NewMockClient(ctrl)\n\t\t\tpeerResolverMock := NewMockPeerResolver(ctrl)\n\t\t\tloadbalancerMock := NewMockLoadBalancer(ctrl)\n\t\t\tproviderMock := NewMockPartitionConfigProvider(ctrl)\n\t\t\ttt.mock(peerResolverMock, loadbalancerMock, client, providerMock)\n\t\t\tc := NewClient(client, peerResolverMock, loadbalancerMock, providerMock)\n\n\t\t\tres, err := tt.op(c)\n\t\t\tif tt.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tif tt.validateError != nil {\n\t\t\t\t\ttt.validateError(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, res)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc testAddActivityTaskRequest() *types.AddActivityTaskRequest {\n\treturn &types.AddActivityTaskRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t}\n}\n\nfunc testAddDecisionTaskRequest() *types.AddDecisionTaskRequest {\n\treturn &types.AddDecisionTaskRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t}\n}\n\nfunc testRespondQueryTaskCompletedRequest() *types.MatchingRespondQueryTaskCompletedRequest {\n\treturn &types.MatchingRespondQueryTaskCompletedRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t}\n}\n\nfunc testCancelOutstandingPollRequest() *types.CancelOutstandingPollRequest {\n\treturn &types.CancelOutstandingPollRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t}\n}\n\nfunc testMatchingPollForActivityTaskRequest() *types.MatchingPollForActivityTaskRequest {\n\treturn &types.MatchingPollForActivityTaskRequest{\n\t\tDomainUUID:  _testDomainUUID,\n\t\tPollRequest: &types.PollForActivityTaskRequest{TaskList: &types.TaskList{Name: _testTaskList}},\n\t}\n}\n\nfunc testMatchingPollForDecisionTaskRequest() *types.MatchingPollForDecisionTaskRequest {\n\treturn &types.MatchingPollForDecisionTaskRequest{\n\t\tDomainUUID:  _testDomainUUID,\n\t\tPollRequest: &types.PollForDecisionTaskRequest{TaskList: &types.TaskList{Name: _testTaskList}},\n\t}\n}\n\nfunc testMatchingQueryWorkflowRequest() *types.MatchingQueryWorkflowRequest {\n\treturn &types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID:   _testDomainUUID,\n\t\tTaskList:     &types.TaskList{Name: _testTaskList},\n\t\tQueryRequest: &types.QueryWorkflowRequest{Domain: _testDomain},\n\t}\n}\n\nfunc testMatchingDescribeTaskListRequest() *types.MatchingDescribeTaskListRequest {\n\treturn &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID:  _testDomainUUID,\n\t\tDescRequest: &types.DescribeTaskListRequest{TaskList: &types.TaskList{Name: _testTaskList}},\n\t}\n}\n\nfunc testMatchingListTaskListPartitionsRequest() *types.MatchingListTaskListPartitionsRequest {\n\treturn &types.MatchingListTaskListPartitionsRequest{\n\t\tDomain:   _testDomainUUID,\n\t\tTaskList: &types.TaskList{Name: _testTaskList},\n\t}\n}\n\nfunc testGetTaskListsByDomainRequest() *types.GetTaskListsByDomainRequest {\n\treturn &types.GetTaskListsByDomainRequest{\n\t\tDomain: _testDomain,\n\t}\n}\n\nfunc testMatchingUpdateTaskListPartitionConfigRequest() *types.MatchingUpdateTaskListPartitionConfigRequest {\n\treturn &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\tVersion: 1,\n\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {},\n\t\t\t\t2: {\n\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {\n\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testMatchingRefreshTaskListPartitionConfigRequest() *types.MatchingRefreshTaskListPartitionConfigRequest {\n\treturn &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\tDomainUUID: _testDomainUUID,\n\t\tTaskList:   &types.TaskList{Name: _testTaskList},\n\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\tVersion: 1,\n\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {},\n\t\t\t\t2: {\n\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {\n\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "client/matching/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage matching\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -package matching github.com/uber/cadence/client/matching Client\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/matching_generated.go -v client=Matching\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/matching_generated.go -v client=Matching\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/matching_generated.go -v client=Matching\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/matching_generated.go -v client=Matching -v package=matchingv1 -v path=github.com/uber/cadence/.gen/proto/matching/v1 -v prefix=Matching\n//go:generate gowrap gen -g -p . -i Client -t ../templates/thrift.tmpl -o ../wrappers/thrift/matching_generated.go -v client=Matching -v prefix=Matching\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/matching_generated.go -v client=Matching\n\n// Client is the interface exposed by types service client\ntype Client interface {\n\tAddActivityTask(context.Context, *types.AddActivityTaskRequest, ...yarpc.CallOption) (*types.AddActivityTaskResponse, error)\n\tAddDecisionTask(context.Context, *types.AddDecisionTaskRequest, ...yarpc.CallOption) (*types.AddDecisionTaskResponse, error)\n\tCancelOutstandingPoll(context.Context, *types.CancelOutstandingPollRequest, ...yarpc.CallOption) error\n\tDescribeTaskList(context.Context, *types.MatchingDescribeTaskListRequest, ...yarpc.CallOption) (*types.DescribeTaskListResponse, error)\n\tListTaskListPartitions(context.Context, *types.MatchingListTaskListPartitionsRequest, ...yarpc.CallOption) (*types.ListTaskListPartitionsResponse, error)\n\tGetTaskListsByDomain(context.Context, *types.GetTaskListsByDomainRequest, ...yarpc.CallOption) (*types.GetTaskListsByDomainResponse, error)\n\tPollForActivityTask(context.Context, *types.MatchingPollForActivityTaskRequest, ...yarpc.CallOption) (*types.MatchingPollForActivityTaskResponse, error)\n\tPollForDecisionTask(context.Context, *types.MatchingPollForDecisionTaskRequest, ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error)\n\tQueryWorkflow(context.Context, *types.MatchingQueryWorkflowRequest, ...yarpc.CallOption) (*types.MatchingQueryWorkflowResponse, error)\n\tRespondQueryTaskCompleted(context.Context, *types.MatchingRespondQueryTaskCompletedRequest, ...yarpc.CallOption) error\n\tUpdateTaskListPartitionConfig(context.Context, *types.MatchingUpdateTaskListPartitionConfigRequest, ...yarpc.CallOption) (*types.MatchingUpdateTaskListPartitionConfigResponse, error)\n\tRefreshTaskListPartitionConfig(context.Context, *types.MatchingRefreshTaskListPartitionConfigRequest, ...yarpc.CallOption) (*types.MatchingRefreshTaskListPartitionConfigResponse, error)\n}\n"
  },
  {
    "path": "client/matching/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package matching -source interface.go -destination interface_mock.go -package matching github.com/uber/cadence/client/matching Client\n//\n\n// Package matching is a generated GoMock package.\npackage matching\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// AddActivityTask mocks base method.\nfunc (m *MockClient) AddActivityTask(arg0 context.Context, arg1 *types.AddActivityTaskRequest, arg2 ...yarpc.CallOption) (*types.AddActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"AddActivityTask\", varargs...)\n\tret0, _ := ret[0].(*types.AddActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTask indicates an expected call of AddActivityTask.\nfunc (mr *MockClientMockRecorder) AddActivityTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTask\", reflect.TypeOf((*MockClient)(nil).AddActivityTask), varargs...)\n}\n\n// AddDecisionTask mocks base method.\nfunc (m *MockClient) AddDecisionTask(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, arg2 ...yarpc.CallOption) (*types.AddDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"AddDecisionTask\", varargs...)\n\tret0, _ := ret[0].(*types.AddDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTask indicates an expected call of AddDecisionTask.\nfunc (mr *MockClientMockRecorder) AddDecisionTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTask\", reflect.TypeOf((*MockClient)(nil).AddDecisionTask), varargs...)\n}\n\n// CancelOutstandingPoll mocks base method.\nfunc (m *MockClient) CancelOutstandingPoll(arg0 context.Context, arg1 *types.CancelOutstandingPollRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CancelOutstandingPoll\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CancelOutstandingPoll indicates an expected call of CancelOutstandingPoll.\nfunc (mr *MockClientMockRecorder) CancelOutstandingPoll(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CancelOutstandingPoll\", reflect.TypeOf((*MockClient)(nil).CancelOutstandingPoll), varargs...)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockClient) DescribeTaskList(arg0 context.Context, arg1 *types.MatchingDescribeTaskListRequest, arg2 ...yarpc.CallOption) (*types.DescribeTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", varargs...)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockClientMockRecorder) DescribeTaskList(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockClient)(nil).DescribeTaskList), varargs...)\n}\n\n// GetTaskListsByDomain mocks base method.\nfunc (m *MockClient) GetTaskListsByDomain(arg0 context.Context, arg1 *types.GetTaskListsByDomainRequest, arg2 ...yarpc.CallOption) (*types.GetTaskListsByDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", varargs...)\n\tret0, _ := ret[0].(*types.GetTaskListsByDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListsByDomain indicates an expected call of GetTaskListsByDomain.\nfunc (mr *MockClientMockRecorder) GetTaskListsByDomain(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListsByDomain\", reflect.TypeOf((*MockClient)(nil).GetTaskListsByDomain), varargs...)\n}\n\n// ListTaskListPartitions mocks base method.\nfunc (m *MockClient) ListTaskListPartitions(arg0 context.Context, arg1 *types.MatchingListTaskListPartitionsRequest, arg2 ...yarpc.CallOption) (*types.ListTaskListPartitionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", varargs...)\n\tret0, _ := ret[0].(*types.ListTaskListPartitionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskListPartitions indicates an expected call of ListTaskListPartitions.\nfunc (mr *MockClientMockRecorder) ListTaskListPartitions(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskListPartitions\", reflect.TypeOf((*MockClient)(nil).ListTaskListPartitions), varargs...)\n}\n\n// PollForActivityTask mocks base method.\nfunc (m *MockClient) PollForActivityTask(arg0 context.Context, arg1 *types.MatchingPollForActivityTaskRequest, arg2 ...yarpc.CallOption) (*types.MatchingPollForActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", varargs...)\n\tret0, _ := ret[0].(*types.MatchingPollForActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForActivityTask indicates an expected call of PollForActivityTask.\nfunc (mr *MockClientMockRecorder) PollForActivityTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForActivityTask\", reflect.TypeOf((*MockClient)(nil).PollForActivityTask), varargs...)\n}\n\n// PollForDecisionTask mocks base method.\nfunc (m *MockClient) PollForDecisionTask(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, arg2 ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", varargs...)\n\tret0, _ := ret[0].(*types.MatchingPollForDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForDecisionTask indicates an expected call of PollForDecisionTask.\nfunc (mr *MockClientMockRecorder) PollForDecisionTask(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForDecisionTask\", reflect.TypeOf((*MockClient)(nil).PollForDecisionTask), varargs...)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockClient) QueryWorkflow(arg0 context.Context, arg1 *types.MatchingQueryWorkflowRequest, arg2 ...yarpc.CallOption) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", varargs...)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockClientMockRecorder) QueryWorkflow(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockClient)(nil).QueryWorkflow), varargs...)\n}\n\n// RefreshTaskListPartitionConfig mocks base method.\nfunc (m *MockClient) RefreshTaskListPartitionConfig(arg0 context.Context, arg1 *types.MatchingRefreshTaskListPartitionConfigRequest, arg2 ...yarpc.CallOption) (*types.MatchingRefreshTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RefreshTaskListPartitionConfig\", varargs...)\n\tret0, _ := ret[0].(*types.MatchingRefreshTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RefreshTaskListPartitionConfig indicates an expected call of RefreshTaskListPartitionConfig.\nfunc (mr *MockClientMockRecorder) RefreshTaskListPartitionConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshTaskListPartitionConfig\", reflect.TypeOf((*MockClient)(nil).RefreshTaskListPartitionConfig), varargs...)\n}\n\n// RespondQueryTaskCompleted mocks base method.\nfunc (m *MockClient) RespondQueryTaskCompleted(arg0 context.Context, arg1 *types.MatchingRespondQueryTaskCompletedRequest, arg2 ...yarpc.CallOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondQueryTaskCompleted indicates an expected call of RespondQueryTaskCompleted.\nfunc (mr *MockClientMockRecorder) RespondQueryTaskCompleted(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondQueryTaskCompleted\", reflect.TypeOf((*MockClient)(nil).RespondQueryTaskCompleted), varargs...)\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockClient) UpdateTaskListPartitionConfig(arg0 context.Context, arg1 *types.MatchingUpdateTaskListPartitionConfigRequest, arg2 ...yarpc.CallOption) (*types.MatchingUpdateTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", varargs...)\n\tret0, _ := ret[0].(*types.MatchingUpdateTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockClientMockRecorder) UpdateTaskListPartitionConfig(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockClient)(nil).UpdateTaskListPartitionConfig), varargs...)\n}\n"
  },
  {
    "path": "client/matching/isolation_loadbalancer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"math/rand\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype isolationLoadBalancer struct {\n\tprovider                   PartitionConfigProvider\n\tfallback                   WeightedLoadBalancer\n\tdomainIDToName             func(string) (string, error)\n\tisolationAssignmentEnabled func(string) bool\n}\n\nfunc NewIsolationLoadBalancer(fallback WeightedLoadBalancer, provider PartitionConfigProvider, domainIDToName func(string) (string, error), config *dynamicconfig.Collection) LoadBalancer {\n\tisolationAssignmentEnabled := config.GetBoolPropertyFilteredByDomain(dynamicproperties.EnablePartitionIsolationGroupAssignment)\n\treturn &isolationLoadBalancer{\n\t\tprovider:                   provider,\n\t\tfallback:                   fallback,\n\t\tdomainIDToName:             domainIDToName,\n\t\tisolationAssignmentEnabled: isolationAssignmentEnabled,\n\t}\n}\n\nfunc (i *isolationLoadBalancer) PickWritePartition(taskListType int, req WriteRequest) string {\n\ttaskList := *req.GetTaskList()\n\n\tdomainName, err := i.domainIDToName(req.GetDomainUUID())\n\tif err != nil || !i.isolationAssignmentEnabled(domainName) {\n\t\treturn i.fallback.PickWritePartition(taskListType, req)\n\t}\n\n\ttaskGroup, ok := req.GetPartitionConfig()[isolationgroup.GroupKey]\n\tif !ok || taskGroup == \"\" {\n\t\treturn i.fallback.PickWritePartition(taskListType, req)\n\t}\n\n\tconfig := i.provider.GetPartitionConfig(req.GetDomainUUID(), taskList, taskListType)\n\tif len(config.WritePartitions) == 1 {\n\t\treturn taskList.GetName()\n\t}\n\n\tpartitions := getPartitionsForGroup(taskGroup, config.WritePartitions)\n\tif len(partitions) == 0 {\n\t\treturn i.fallback.PickWritePartition(taskListType, req)\n\t}\n\n\tp := pickRandom(partitions)\n\n\treturn getPartitionTaskListName(taskList.GetName(), p)\n}\n\nfunc (i *isolationLoadBalancer) PickReadPartition(taskListType int, req ReadRequest, isolationGroup string) string {\n\ttaskList := *req.GetTaskList()\n\n\tdomainName, err := i.domainIDToName(req.GetDomainUUID())\n\tif err != nil || !i.isolationAssignmentEnabled(domainName) || isolationGroup == \"\" {\n\t\treturn i.fallback.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\n\tconfig := i.provider.GetPartitionConfig(req.GetDomainUUID(), taskList, taskListType)\n\tif len(config.ReadPartitions) == 1 {\n\t\treturn taskList.GetName()\n\t}\n\n\tpartitions := getPartitionsForGroup(isolationGroup, config.ReadPartitions)\n\tif len(partitions) == 0 {\n\t\treturn i.fallback.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\n\tp := partitions[0]\n\tif len(partitions) > 1 {\n\t\tp = i.fallback.PickBetween(req.GetDomainUUID(), taskList.GetName(), taskListType, partitions)\n\n\t\tif p == -1 {\n\t\t\tp = pickRandom(partitions)\n\t\t}\n\t}\n\n\treturn getPartitionTaskListName(taskList.GetName(), p)\n}\n\nfunc (i *isolationLoadBalancer) UpdateWeight(taskListType int, req ReadRequest, partition string, info *types.LoadBalancerHints) {\n\ti.fallback.UpdateWeight(taskListType, req, partition, info)\n}\n\nfunc getPartitionsForGroup(taskGroup string, partitions map[int]*types.TaskListPartition) []int {\n\tif taskGroup == \"\" {\n\t\treturn nil\n\t}\n\n\tvar res []int\n\tfor id := range len(partitions) {\n\t\tp := partitions[id]\n\t\tif partitionAcceptsGroup(p, taskGroup) {\n\t\t\tres = append(res, id)\n\t\t}\n\t}\n\treturn res\n}\n\nfunc pickRandom(partitions []int) int {\n\tpicked := rand.Intn(len(partitions))\n\treturn partitions[picked]\n}\n\nfunc partitionAcceptsGroup(partition *types.TaskListPartition, taskGroup string) bool {\n\t// Accepts all groups\n\tif len(partition.IsolationGroups) == 0 {\n\t\treturn true\n\t}\n\tfor _, ig := range partition.IsolationGroups {\n\t\tif ig == taskGroup {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "client/matching/isolation_loadbalancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar testDomain = \"domainID\"\n\nfunc TestIsolationPickWritePartition(t *testing.T) {\n\ttl := \"tl\"\n\tcases := []struct {\n\t\tname             string\n\t\tgroup            string\n\t\tconfig           *types.TaskListPartitionConfig\n\t\tdisableIsolation bool\n\t\tshouldFallback   bool\n\t\tallowed          []string\n\t}{\n\t\t{\n\t\t\tname:  \"single partition\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowed: []string{tl},\n\t\t},\n\t\t{\n\t\t\tname:  \"single partition - isolation disabled\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tdisableIsolation: true,\n\t\t\tshouldFallback:   true,\n\t\t\tallowed:          []string{\"fallback\"},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - single option\",\n\t\t\tgroup: \"b\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowed: []string{getPartitionTaskListName(tl, 1)},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - multiple options\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"a\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowed: []string{tl, getPartitionTaskListName(tl, 1)},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - no match\",\n\t\t\tgroup: \"c\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tshouldFallback: true,\n\t\t\tallowed:        []string{\"fallback\"},\n\t\t},\n\t\t{\n\t\t\tname: \"fallback - no group\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tshouldFallback: true,\n\t\t\tallowed:        []string{\"fallback\"},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlb, fallback := createWithMocks(t, !tc.disableIsolation, tc.config)\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID: testDomain,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: tl,\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t}\n\t\t\tif tc.group != \"\" {\n\t\t\t\treq.PartitionConfig = map[string]string{\n\t\t\t\t\tisolationgroup.GroupKey: tc.group,\n\t\t\t\t}\n\t\t\t}\n\t\t\tif tc.shouldFallback {\n\t\t\t\tfallback.EXPECT().PickWritePartition(int(types.TaskListTypeDecision), req).Return(\"fallback\").Times(1)\n\t\t\t}\n\t\t\tp := lb.PickWritePartition(0, req)\n\t\t\tassert.Contains(t, tc.allowed, p)\n\t\t})\n\t}\n}\n\nfunc TestIsolationPickReadPartition(t *testing.T) {\n\ttl := \"tl\"\n\tcases := []struct {\n\t\tname             string\n\t\tgroup            string\n\t\tconfig           *types.TaskListPartitionConfig\n\t\tdisableIsolation bool\n\t\tallowance        func(balancer *MockWeightedLoadBalancer)\n\t\texpected         string\n\t}{\n\t\t{\n\t\t\tname:  \"single partition\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: tl,\n\t\t},\n\t\t{\n\t\t\tname:  \"single partition - isolation disabled\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tdisableIsolation: true,\n\t\t\tallowance: func(balancer *MockWeightedLoadBalancer) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(gomock.Any(), gomock.Any(), gomock.Any()).Return(\"fallback\")\n\t\t\t},\n\t\t\texpected: \"fallback\",\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - single option\",\n\t\t\tgroup: \"b\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: getPartitionTaskListName(tl, 1),\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - multiple options\",\n\t\t\tgroup: \"a\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"a\"}},\n\t\t\t\t\t2: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowance: func(balancer *MockWeightedLoadBalancer) {\n\t\t\t\tbalancer.EXPECT().PickBetween(testDomain, tl, 0, []int{0, 1}).Return(1)\n\t\t\t},\n\t\t\texpected: getPartitionTaskListName(tl, 1),\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - no matching\",\n\t\t\tgroup: \"c\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowance: func(balancer *MockWeightedLoadBalancer) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(gomock.Any(), gomock.Any(), gomock.Any()).Return(\"fallback\")\n\t\t\t},\n\t\t\texpected: \"fallback\",\n\t\t},\n\t\t{\n\t\t\tname: \"fallback - no group\",\n\t\t\tconfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {[]string{\"a\"}},\n\t\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowance: func(balancer *MockWeightedLoadBalancer) {\n\t\t\t\tbalancer.EXPECT().PickReadPartition(gomock.Any(), gomock.Any(), gomock.Any()).Return(\"fallback\")\n\t\t\t},\n\t\t\texpected: \"fallback\",\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlb, fallback := createWithMocks(t, !tc.disableIsolation, tc.config)\n\t\t\treq := &types.MatchingQueryWorkflowRequest{\n\t\t\t\tDomainUUID: testDomain,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: tl,\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t}\n\t\t\tif tc.allowance != nil {\n\t\t\t\ttc.allowance(fallback)\n\t\t\t}\n\t\t\tactual := lb.PickReadPartition(0, req, tc.group)\n\t\t\tassert.Equal(t, tc.expected, actual)\n\t\t})\n\t}\n}\n\nfunc TestIsolationGetPartitionsForGroup(t *testing.T) {\n\tcases := []struct {\n\t\tname       string\n\t\tgroup      string\n\t\tpartitions map[int]*types.TaskListPartition\n\t\texpected   []int\n\t}{\n\t\t{\n\t\t\tname:  \"single partition\",\n\t\t\tgroup: \"a\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"b\", \"c\"}},\n\t\t\t},\n\t\t\texpected: []int{0},\n\t\t},\n\t\t{\n\t\t\tname:  \"single partition - wildcard\",\n\t\t\tgroup: \"a\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t},\n\t\t\texpected: []int{0},\n\t\t},\n\t\t{\n\t\t\tname:  \"single partition - no options\",\n\t\t\tgroup: \"a\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"b\"}},\n\t\t\t},\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - single option\",\n\t\t\tgroup: \"b\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"c\"}},\n\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t},\n\t\t\texpected: []int{1},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - multiple options\",\n\t\t\tgroup: \"b\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"b\", \"c\"}},\n\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t2: {[]string{\"d\"}},\n\t\t\t},\n\t\t\texpected: []int{0, 1},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - multiple options with wildcard\",\n\t\t\tgroup: \"b\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"c\"}},\n\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t2: {},\n\t\t\t},\n\t\t\texpected: []int{1, 2},\n\t\t},\n\t\t{\n\t\t\tname:  \"multiple partitions - no options\",\n\t\t\tgroup: \"d\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"c\"}},\n\t\t\t\t1: {[]string{\"b\"}},\n\t\t\t\t2: {[]string{\"c\"}},\n\t\t\t},\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"no group\",\n\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {[]string{\"a\", \"b\", \"c\"}},\n\t\t\t},\n\t\t\texpected: nil,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tactual := getPartitionsForGroup(tc.group, tc.partitions)\n\t\t\tif tc.expected == nil {\n\t\t\t\tassert.Nil(t, actual)\n\t\t\t} else {\n\t\t\t\tassert.ElementsMatch(t, tc.expected, actual)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc createWithMocks(t *testing.T, isolationEnabled bool, config *types.TaskListPartitionConfig) (*isolationLoadBalancer, *MockWeightedLoadBalancer) {\n\tctrl := gomock.NewController(t)\n\tfallback := NewMockWeightedLoadBalancer(ctrl)\n\tcfg := NewMockPartitionConfigProvider(ctrl)\n\tcfg.EXPECT().GetPartitionConfig(gomock.Any(), gomock.Any(), gomock.Any()).Return(config).AnyTimes()\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.EnablePartitionIsolationGroupAssignment, isolationEnabled))\n\tdc := dynamicconfig.NewCollection(dynamicClient, testlogger.New(t))\n\tlb := NewIsolationLoadBalancer(fallback, cfg, func(s string) (string, error) {\n\t\treturn s, nil\n\t}, dc).(*isolationLoadBalancer)\n\n\treturn lb, fallback\n}\n"
  },
  {
    "path": "client/matching/loadbalancer.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination loadbalancer_mock.go -package matching github.com/uber/cadence/client/matching LoadBalancer,WeightedLoadBalancer\n\npackage matching\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// WriteRequest is the interface for all types of AddTask* requests\n\tWriteRequest interface {\n\t\tReadRequest\n\t\tGetPartitionConfig() map[string]string\n\t}\n\t// ReadRequest is the interface for all types of Poll* requests\n\tReadRequest interface {\n\t\tGetDomainUUID() string\n\t\tGetTaskList() *types.TaskList\n\t\tGetForwardedFrom() string\n\t}\n\t// LoadBalancer is the interface for implementers of\n\t// component that distributes add/poll api calls across\n\t// available task list partitions when possible\n\tLoadBalancer interface {\n\t\t// PickWritePartition returns the task list partition for adding\n\t\t// an activity or decision task. The input is the name of the\n\t\t// original task list (with no partition info). When forwardedFrom\n\t\t// is non-empty, this call is forwardedFrom from a child partition\n\t\t// to a parent partition in which case, no load balancing should be\n\t\t// performed\n\t\tPickWritePartition(\n\t\t\ttaskListType int,\n\t\t\trequest WriteRequest,\n\t\t) string\n\n\t\t// PickReadPartition returns the task list partition to send a poller to.\n\t\t// Input is name of the original task list as specified by caller. When\n\t\t// forwardedFrom is non-empty, no load balancing should be done.\n\t\tPickReadPartition(\n\t\t\ttaskListType int,\n\t\t\trequest ReadRequest,\n\t\t\tisolationGroup string,\n\t\t) string\n\n\t\t// UpdateWeight updates the weight of a task list partition.\n\t\t// Input is name of the original task list as specified by caller. When\n\t\t// the original task list is a partition, no update should be done.\n\t\tUpdateWeight(\n\t\t\ttaskListType int,\n\t\t\trequest ReadRequest,\n\t\t\tpartition string,\n\t\t\tinfo *types.LoadBalancerHints,\n\t\t)\n\t}\n\tWeightedLoadBalancer interface {\n\t\tLoadBalancer\n\t\tPickBetween(domainID, taskListName string, taskListType int, partitions []int) int\n\t}\n\n\tdefaultLoadBalancer struct {\n\t\tprovider PartitionConfigProvider\n\t}\n)\n\n// NewLoadBalancer returns an instance of matching load balancer that\n// can help distribute api calls across task list partitions\nfunc NewLoadBalancer(\n\tprovider PartitionConfigProvider,\n) LoadBalancer {\n\treturn &defaultLoadBalancer{\n\t\tprovider: provider,\n\t}\n}\n\nfunc (lb *defaultLoadBalancer) PickWritePartition(\n\ttaskListType int, req WriteRequest,\n) string {\n\tnPartitions := lb.provider.GetNumberOfWritePartitions(req.GetDomainUUID(), *req.GetTaskList(), taskListType)\n\treturn lb.pickPartition(*req.GetTaskList(), req.GetForwardedFrom(), nPartitions)\n\n}\n\nfunc (lb *defaultLoadBalancer) PickReadPartition(\n\ttaskListType int, req ReadRequest, _ string,\n) string {\n\tn := lb.provider.GetNumberOfReadPartitions(req.GetDomainUUID(), *req.GetTaskList(), taskListType)\n\treturn lb.pickPartition(*req.GetTaskList(), req.GetForwardedFrom(), n)\n\n}\n\nfunc (lb *defaultLoadBalancer) pickPartition(\n\ttaskList types.TaskList,\n\tforwardedFrom string,\n\tnPartitions int,\n) string {\n\n\tif nPartitions <= 1 {\n\t\treturn taskList.GetName()\n\t}\n\n\tp := rand.Intn(nPartitions)\n\treturn getPartitionTaskListName(taskList.GetName(), p)\n}\n\nfunc (lb *defaultLoadBalancer) UpdateWeight(\n\ttaskListType int,\n\treq ReadRequest,\n\tpartition string,\n\tinfo *types.LoadBalancerHints,\n) {\n}\n\nfunc getPartitionTaskListName(root string, partition int) string {\n\tif partition <= 0 {\n\t\treturn root\n\t}\n\treturn fmt.Sprintf(\"%v%v/%v\", constants.ReservedTaskListPrefix, root, partition)\n}\n"
  },
  {
    "path": "client/matching/loadbalancer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: loadbalancer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package matching -source loadbalancer.go -destination loadbalancer_mock.go -package matching github.com/uber/cadence/client/matching LoadBalancer,WeightedLoadBalancer\n//\n\n// Package matching is a generated GoMock package.\npackage matching\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockWriteRequest is a mock of WriteRequest interface.\ntype MockWriteRequest struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWriteRequestMockRecorder\n\tisgomock struct{}\n}\n\n// MockWriteRequestMockRecorder is the mock recorder for MockWriteRequest.\ntype MockWriteRequestMockRecorder struct {\n\tmock *MockWriteRequest\n}\n\n// NewMockWriteRequest creates a new mock instance.\nfunc NewMockWriteRequest(ctrl *gomock.Controller) *MockWriteRequest {\n\tmock := &MockWriteRequest{ctrl: ctrl}\n\tmock.recorder = &MockWriteRequestMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWriteRequest) EXPECT() *MockWriteRequestMockRecorder {\n\treturn m.recorder\n}\n\n// GetDomainUUID mocks base method.\nfunc (m *MockWriteRequest) GetDomainUUID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainUUID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainUUID indicates an expected call of GetDomainUUID.\nfunc (mr *MockWriteRequestMockRecorder) GetDomainUUID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainUUID\", reflect.TypeOf((*MockWriteRequest)(nil).GetDomainUUID))\n}\n\n// GetForwardedFrom mocks base method.\nfunc (m *MockWriteRequest) GetForwardedFrom() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetForwardedFrom\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetForwardedFrom indicates an expected call of GetForwardedFrom.\nfunc (mr *MockWriteRequestMockRecorder) GetForwardedFrom() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetForwardedFrom\", reflect.TypeOf((*MockWriteRequest)(nil).GetForwardedFrom))\n}\n\n// GetPartitionConfig mocks base method.\nfunc (m *MockWriteRequest) GetPartitionConfig() map[string]string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionConfig\")\n\tret0, _ := ret[0].(map[string]string)\n\treturn ret0\n}\n\n// GetPartitionConfig indicates an expected call of GetPartitionConfig.\nfunc (mr *MockWriteRequestMockRecorder) GetPartitionConfig() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionConfig\", reflect.TypeOf((*MockWriteRequest)(nil).GetPartitionConfig))\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockWriteRequest) GetTaskList() *types.TaskList {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\")\n\tret0, _ := ret[0].(*types.TaskList)\n\treturn ret0\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockWriteRequestMockRecorder) GetTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockWriteRequest)(nil).GetTaskList))\n}\n\n// MockReadRequest is a mock of ReadRequest interface.\ntype MockReadRequest struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReadRequestMockRecorder\n\tisgomock struct{}\n}\n\n// MockReadRequestMockRecorder is the mock recorder for MockReadRequest.\ntype MockReadRequestMockRecorder struct {\n\tmock *MockReadRequest\n}\n\n// NewMockReadRequest creates a new mock instance.\nfunc NewMockReadRequest(ctrl *gomock.Controller) *MockReadRequest {\n\tmock := &MockReadRequest{ctrl: ctrl}\n\tmock.recorder = &MockReadRequestMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockReadRequest) EXPECT() *MockReadRequestMockRecorder {\n\treturn m.recorder\n}\n\n// GetDomainUUID mocks base method.\nfunc (m *MockReadRequest) GetDomainUUID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainUUID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainUUID indicates an expected call of GetDomainUUID.\nfunc (mr *MockReadRequestMockRecorder) GetDomainUUID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainUUID\", reflect.TypeOf((*MockReadRequest)(nil).GetDomainUUID))\n}\n\n// GetForwardedFrom mocks base method.\nfunc (m *MockReadRequest) GetForwardedFrom() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetForwardedFrom\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetForwardedFrom indicates an expected call of GetForwardedFrom.\nfunc (mr *MockReadRequestMockRecorder) GetForwardedFrom() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetForwardedFrom\", reflect.TypeOf((*MockReadRequest)(nil).GetForwardedFrom))\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockReadRequest) GetTaskList() *types.TaskList {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\")\n\tret0, _ := ret[0].(*types.TaskList)\n\treturn ret0\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockReadRequestMockRecorder) GetTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockReadRequest)(nil).GetTaskList))\n}\n\n// MockLoadBalancer is a mock of LoadBalancer interface.\ntype MockLoadBalancer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockLoadBalancerMockRecorder\n\tisgomock struct{}\n}\n\n// MockLoadBalancerMockRecorder is the mock recorder for MockLoadBalancer.\ntype MockLoadBalancerMockRecorder struct {\n\tmock *MockLoadBalancer\n}\n\n// NewMockLoadBalancer creates a new mock instance.\nfunc NewMockLoadBalancer(ctrl *gomock.Controller) *MockLoadBalancer {\n\tmock := &MockLoadBalancer{ctrl: ctrl}\n\tmock.recorder = &MockLoadBalancerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockLoadBalancer) EXPECT() *MockLoadBalancerMockRecorder {\n\treturn m.recorder\n}\n\n// PickReadPartition mocks base method.\nfunc (m *MockLoadBalancer) PickReadPartition(taskListType int, request ReadRequest, isolationGroup string) string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PickReadPartition\", taskListType, request, isolationGroup)\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PickReadPartition indicates an expected call of PickReadPartition.\nfunc (mr *MockLoadBalancerMockRecorder) PickReadPartition(taskListType, request, isolationGroup any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PickReadPartition\", reflect.TypeOf((*MockLoadBalancer)(nil).PickReadPartition), taskListType, request, isolationGroup)\n}\n\n// PickWritePartition mocks base method.\nfunc (m *MockLoadBalancer) PickWritePartition(taskListType int, request WriteRequest) string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PickWritePartition\", taskListType, request)\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PickWritePartition indicates an expected call of PickWritePartition.\nfunc (mr *MockLoadBalancerMockRecorder) PickWritePartition(taskListType, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PickWritePartition\", reflect.TypeOf((*MockLoadBalancer)(nil).PickWritePartition), taskListType, request)\n}\n\n// UpdateWeight mocks base method.\nfunc (m *MockLoadBalancer) UpdateWeight(taskListType int, request ReadRequest, partition string, info *types.LoadBalancerHints) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateWeight\", taskListType, request, partition, info)\n}\n\n// UpdateWeight indicates an expected call of UpdateWeight.\nfunc (mr *MockLoadBalancerMockRecorder) UpdateWeight(taskListType, request, partition, info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWeight\", reflect.TypeOf((*MockLoadBalancer)(nil).UpdateWeight), taskListType, request, partition, info)\n}\n\n// MockWeightedLoadBalancer is a mock of WeightedLoadBalancer interface.\ntype MockWeightedLoadBalancer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWeightedLoadBalancerMockRecorder\n\tisgomock struct{}\n}\n\n// MockWeightedLoadBalancerMockRecorder is the mock recorder for MockWeightedLoadBalancer.\ntype MockWeightedLoadBalancerMockRecorder struct {\n\tmock *MockWeightedLoadBalancer\n}\n\n// NewMockWeightedLoadBalancer creates a new mock instance.\nfunc NewMockWeightedLoadBalancer(ctrl *gomock.Controller) *MockWeightedLoadBalancer {\n\tmock := &MockWeightedLoadBalancer{ctrl: ctrl}\n\tmock.recorder = &MockWeightedLoadBalancerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWeightedLoadBalancer) EXPECT() *MockWeightedLoadBalancerMockRecorder {\n\treturn m.recorder\n}\n\n// PickBetween mocks base method.\nfunc (m *MockWeightedLoadBalancer) PickBetween(domainID, taskListName string, taskListType int, partitions []int) int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PickBetween\", domainID, taskListName, taskListType, partitions)\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// PickBetween indicates an expected call of PickBetween.\nfunc (mr *MockWeightedLoadBalancerMockRecorder) PickBetween(domainID, taskListName, taskListType, partitions any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PickBetween\", reflect.TypeOf((*MockWeightedLoadBalancer)(nil).PickBetween), domainID, taskListName, taskListType, partitions)\n}\n\n// PickReadPartition mocks base method.\nfunc (m *MockWeightedLoadBalancer) PickReadPartition(taskListType int, request ReadRequest, isolationGroup string) string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PickReadPartition\", taskListType, request, isolationGroup)\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PickReadPartition indicates an expected call of PickReadPartition.\nfunc (mr *MockWeightedLoadBalancerMockRecorder) PickReadPartition(taskListType, request, isolationGroup any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PickReadPartition\", reflect.TypeOf((*MockWeightedLoadBalancer)(nil).PickReadPartition), taskListType, request, isolationGroup)\n}\n\n// PickWritePartition mocks base method.\nfunc (m *MockWeightedLoadBalancer) PickWritePartition(taskListType int, request WriteRequest) string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PickWritePartition\", taskListType, request)\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PickWritePartition indicates an expected call of PickWritePartition.\nfunc (mr *MockWeightedLoadBalancerMockRecorder) PickWritePartition(taskListType, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PickWritePartition\", reflect.TypeOf((*MockWeightedLoadBalancer)(nil).PickWritePartition), taskListType, request)\n}\n\n// UpdateWeight mocks base method.\nfunc (m *MockWeightedLoadBalancer) UpdateWeight(taskListType int, request ReadRequest, partition string, info *types.LoadBalancerHints) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateWeight\", taskListType, request, partition, info)\n}\n\n// UpdateWeight indicates an expected call of UpdateWeight.\nfunc (mr *MockWeightedLoadBalancerMockRecorder) UpdateWeight(taskListType, request, partition, info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWeight\", reflect.TypeOf((*MockWeightedLoadBalancer)(nil).UpdateWeight), taskListType, request, partition, info)\n}\n"
  },
  {
    "path": "client/matching/loadbalancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc setUpMocksForLoadBalancer(t *testing.T) (*defaultLoadBalancer, *MockPartitionConfigProvider) {\n\tctrl := gomock.NewController(t)\n\tmockProvider := NewMockPartitionConfigProvider(ctrl)\n\n\treturn &defaultLoadBalancer{\n\t\tprovider: mockProvider,\n\t}, mockProvider\n}\n\nfunc Test_defaultLoadBalancer_PickWritePartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tforwardedFrom      string\n\t\ttaskListType       int\n\t\tnPartitions        int\n\t\ttaskListKind       types.TaskListKind\n\t\texpectedPartitions []string\n\t}{\n\t\t{\n\t\t\tname:               \"single write partition, forwarded\",\n\t\t\tforwardedFrom:      \"parent-task-list\",\n\t\t\ttaskListType:       0,\n\t\t\tnPartitions:        1,\n\t\t\ttaskListKind:       types.TaskListKindNormal,\n\t\t\texpectedPartitions: []string{\"test-task-list\"},\n\t\t},\n\t\t{\n\t\t\tname:               \"multiple write partitions, no forward\",\n\t\t\tforwardedFrom:      \"\",\n\t\t\ttaskListType:       0,\n\t\t\tnPartitions:        3,\n\t\t\ttaskListKind:       types.TaskListKindNormal,\n\t\t\texpectedPartitions: []string{\"test-task-list\", \"/__cadence_sys/test-task-list/1\", \"/__cadence_sys/test-task-list/2\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\tloadBalancer, mockProvider := setUpMocksForLoadBalancer(t)\n\n\t\t\tmockProvider.EXPECT().\n\t\t\t\tGetNumberOfWritePartitions(\"test-domain-id\", types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind}, tc.taskListType).\n\t\t\t\tReturn(tc.nPartitions).\n\t\t\t\tTimes(1)\n\n\t\t\t// Pick write partition\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    \"test-domain-id\",\n\t\t\t\tTaskList:      &types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind},\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tpartition := loadBalancer.PickWritePartition(tc.taskListType, req)\n\n\t\t\t// Validate result\n\t\t\tassert.Contains(t, tc.expectedPartitions, partition)\n\t\t})\n\t}\n}\n\nfunc Test_defaultLoadBalancer_PickReadPartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tforwardedFrom      string\n\t\ttaskListType       int\n\t\tnPartitions        int\n\t\ttaskListKind       types.TaskListKind\n\t\texpectedPartitions []string\n\t}{\n\t\t{\n\t\t\tname:               \"single read partition, forwarded\",\n\t\t\tforwardedFrom:      \"parent-task-list\",\n\t\t\ttaskListType:       0,\n\t\t\tnPartitions:        1,\n\t\t\ttaskListKind:       types.TaskListKindNormal,\n\t\t\texpectedPartitions: []string{\"test-task-list\"},\n\t\t},\n\t\t{\n\t\t\tname:               \"multiple read partitions, no forward\",\n\t\t\tforwardedFrom:      \"\",\n\t\t\ttaskListType:       0,\n\t\t\tnPartitions:        3,\n\t\t\ttaskListKind:       types.TaskListKindNormal,\n\t\t\texpectedPartitions: []string{\"test-task-list\", \"/__cadence_sys/test-task-list/1\", \"/__cadence_sys/test-task-list/2\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\tloadBalancer, mockProvider := setUpMocksForLoadBalancer(t)\n\n\t\t\tmockProvider.EXPECT().\n\t\t\t\tGetNumberOfReadPartitions(\"test-domain-id\", types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind}, tc.taskListType).\n\t\t\t\tReturn(tc.nPartitions).\n\t\t\t\tTimes(1)\n\n\t\t\t// Pick read partition\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    \"test-domain-id\",\n\t\t\t\tTaskList:      &types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind},\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tpartition := loadBalancer.PickReadPartition(tc.taskListType, req, \"\")\n\n\t\t\t// Validate result\n\t\t\tassert.Contains(t, tc.expectedPartitions, partition)\n\t\t})\n\t}\n}\n\nfunc Test_defaultLoadBalancer_UpdateWeight(t *testing.T) {\n\tt.Run(\"no-op for task list partitions\", func(t *testing.T) {\n\t\t// Set up mocks\n\t\tloadBalancer, _ := setUpMocksForLoadBalancer(t)\n\n\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: types.TaskListKindNormal.Ptr()}\n\n\t\t// Call UpdateWeight, should do nothing\n\t\treq := &types.AddDecisionTaskRequest{\n\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\tTaskList:   &taskList,\n\t\t}\n\t\tloadBalancer.UpdateWeight(0, req, \"partition\", nil)\n\n\t\t// No expectations, just ensure no-op\n\t})\n}\n\nfunc Test_defaultLoadBalancer_pickPartition(t *testing.T) {\n\ttype args struct {\n\t\ttaskList      types.TaskList\n\t\tforwardedFrom string\n\t\tnPartitions   int\n\t}\n\ttests := []struct {\n\t\tname string\n\t\targs args\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"Test: nPartitions <= 0\",\n\t\t\targs: args{\n\t\t\t\ttaskList: types.TaskList{\n\t\t\t\t\tName: \"taskList4\",\n\t\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t\t},\n\t\t\t\tforwardedFrom: \"\",\n\t\t\t\tnPartitions:   0,\n\t\t\t},\n\t\t\twant: \"taskList4\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tlb := &defaultLoadBalancer{}\n\t\t\tgot := lb.pickPartition(tt.args.taskList, tt.args.forwardedFrom, tt.args.nPartitions)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "client/matching/multi_loadbalancer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tmultiLoadBalancer struct {\n\t\tdefaultLoadBalancer  LoadBalancer\n\t\tloadBalancers        map[string]LoadBalancer\n\t\tdomainIDToName       func(string) (string, error)\n\t\tloadbalancerStrategy dynamicproperties.StringPropertyFnWithTaskListInfoFilters\n\t\tlogger               log.Logger\n\t}\n)\n\nfunc NewMultiLoadBalancer(\n\tdefaultLoadBalancer LoadBalancer,\n\tloadBalancers map[string]LoadBalancer,\n\tdomainIDToName func(string) (string, error),\n\tdc *dynamicconfig.Collection,\n\tlogger log.Logger,\n) LoadBalancer {\n\treturn &multiLoadBalancer{\n\t\tdefaultLoadBalancer:  defaultLoadBalancer,\n\t\tloadBalancers:        loadBalancers,\n\t\tdomainIDToName:       domainIDToName,\n\t\tloadbalancerStrategy: dc.GetStringPropertyFilteredByTaskListInfo(dynamicproperties.TasklistLoadBalancerStrategy),\n\t\tlogger:               logger,\n\t}\n}\n\nfunc (lb *multiLoadBalancer) PickWritePartition(\n\ttaskListType int,\n\treq WriteRequest,\n) string {\n\tif !lb.canRedirectToPartition(req) {\n\t\treturn req.GetTaskList().GetName()\n\t}\n\tdomainName, err := lb.domainIDToName(req.GetDomainUUID())\n\tif err != nil {\n\t\treturn lb.defaultLoadBalancer.PickWritePartition(taskListType, req)\n\t}\n\tstrategy := lb.loadbalancerStrategy(domainName, req.GetTaskList().GetName(), taskListType)\n\tloadBalancer, ok := lb.loadBalancers[strategy]\n\tif !ok {\n\t\tlb.logger.Warn(\"unsupported load balancer strategy\", tag.Value(strategy))\n\t\treturn lb.defaultLoadBalancer.PickWritePartition(taskListType, req)\n\t}\n\treturn loadBalancer.PickWritePartition(taskListType, req)\n}\n\nfunc (lb *multiLoadBalancer) PickReadPartition(\n\ttaskListType int,\n\treq ReadRequest,\n\tisolationGroup string,\n) string {\n\tif !lb.canRedirectToPartition(req) {\n\t\treturn req.GetTaskList().GetName()\n\t}\n\tdomainName, err := lb.domainIDToName(req.GetDomainUUID())\n\tif err != nil {\n\t\treturn lb.defaultLoadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\tstrategy := lb.loadbalancerStrategy(domainName, req.GetTaskList().GetName(), taskListType)\n\tloadBalancer, ok := lb.loadBalancers[strategy]\n\tif !ok {\n\t\tlb.logger.Warn(\"unsupported load balancer strategy\", tag.Value(strategy))\n\t\treturn lb.defaultLoadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\treturn loadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n}\n\nfunc (lb *multiLoadBalancer) UpdateWeight(\n\ttaskListType int,\n\treq ReadRequest,\n\tpartition string,\n\tinfo *types.LoadBalancerHints,\n) {\n\tif !lb.canRedirectToPartition(req) {\n\t\treturn\n\t}\n\tdomainName, err := lb.domainIDToName(req.GetDomainUUID())\n\tif err != nil {\n\t\treturn\n\t}\n\tstrategy := lb.loadbalancerStrategy(domainName, req.GetTaskList().GetName(), taskListType)\n\tloadBalancer, ok := lb.loadBalancers[strategy]\n\tif !ok {\n\t\tlb.logger.Warn(\"unsupported load balancer strategy\", tag.Value(strategy))\n\t\treturn\n\t}\n\tloadBalancer.UpdateWeight(taskListType, req, partition, info)\n}\n\nfunc (lb *multiLoadBalancer) canRedirectToPartition(req ReadRequest) bool {\n\treturn req.GetForwardedFrom() == \"\" && req.GetTaskList().GetKind() == types.TaskListKindNormal && !strings.HasPrefix(req.GetTaskList().GetName(), constants.ReservedTaskListPrefix)\n}\n"
  },
  {
    "path": "client/matching/multi_loadbalancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewMultiLoadBalancer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\trandomMock := NewMockLoadBalancer(ctrl)\n\troundRobinMock := NewMockLoadBalancer(ctrl)\n\tlbs := map[string]LoadBalancer{\n\t\t\"random\":      randomMock,\n\t\t\"round-robin\": roundRobinMock,\n\t}\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\treturn \"testDomainName\", nil\n\t}\n\tdc := dynamicconfig.NewCollection(dynamicconfig.NewNopClient(), testlogger.New(t))\n\tlb := NewMultiLoadBalancer(randomMock, lbs, domainIDToName, dc, testlogger.New(t))\n\tassert.NotNil(t, lb)\n\tmultiLB, ok := lb.(*multiLoadBalancer)\n\tassert.NotNil(t, multiLB)\n\tassert.True(t, ok)\n\tassert.NotNil(t, multiLB.defaultLoadBalancer)\n\tassert.NotNil(t, multiLB.loadBalancers)\n\tassert.NotNil(t, multiLB.domainIDToName)\n\tassert.NotNil(t, multiLB.loadbalancerStrategy)\n\tassert.NotNil(t, multiLB.logger)\n}\n\nfunc TestMultiLoadBalancer_PickWritePartition(t *testing.T) {\n\n\t// Mock the domainIDToName function\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\tif domainID == \"valid-domain\" {\n\t\t\treturn \"valid-domain-name\", nil\n\t\t}\n\t\treturn \"\", errors.New(\"domain not found\")\n\t}\n\n\t// Test cases\n\ttests := []struct {\n\t\tname                 string\n\t\tdomainID             string\n\t\ttaskList             types.TaskList\n\t\ttaskListType         int\n\t\tforwardedFrom        string\n\t\tloadbalancerStrategy string\n\t\texpectedPartition    string\n\t}{\n\t\t{\n\t\t\tname:                 \"random partition when domainIDToName fails\",\n\t\t\tdomainID:             \"invalid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"random-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"round-robin partition enabled\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"round-robin\",\n\t\t\texpectedPartition:    \"roundrobin-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"random partition when round-robin disabled\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"invalid-enum\",\n\t\t\texpectedPartition:    \"random-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - forwarded\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"somewhere\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - sticky\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindSticky.Ptr()},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - partition\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"/__cadence_sys/test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"/__cadence_sys/test-tasklist\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tt.domainID,\n\t\t\t\tTaskList:      &tt.taskList,\n\t\t\t\tForwardedFrom: tt.forwardedFrom,\n\t\t\t}\n\t\t\t// Mock behavior for random and round robin load balancers\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Mock the LoadBalancer interface\n\t\t\trandomMock := NewMockLoadBalancer(ctrl)\n\t\t\troundRobinMock := NewMockLoadBalancer(ctrl)\n\t\t\trandomMock.EXPECT().PickWritePartition(tt.taskListType, req).Return(\"random-partition\").AnyTimes()\n\t\t\troundRobinMock.EXPECT().PickWritePartition(tt.taskListType, req).Return(\"roundrobin-partition\").AnyTimes()\n\n\t\t\tloadbalancerStrategyFn := func(domainName, taskListName string, taskListType int) string {\n\t\t\t\treturn tt.loadbalancerStrategy\n\t\t\t}\n\n\t\t\t// Create multiLoadBalancer\n\t\t\tlb := &multiLoadBalancer{\n\t\t\t\tdefaultLoadBalancer: randomMock,\n\t\t\t\tloadBalancers: map[string]LoadBalancer{\n\t\t\t\t\t\"round-robin\": roundRobinMock,\n\t\t\t\t},\n\t\t\t\tdomainIDToName:       domainIDToName,\n\t\t\t\tloadbalancerStrategy: loadbalancerStrategyFn,\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t}\n\n\t\t\t// Call PickWritePartition and assert result\n\t\t\tpartition := lb.PickWritePartition(tt.taskListType, req)\n\t\t\tassert.Equal(t, tt.expectedPartition, partition)\n\t\t})\n\t}\n}\n\nfunc TestMultiLoadBalancer_PickReadPartition(t *testing.T) {\n\n\t// Mock the domainIDToName function\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\tif domainID == \"valid-domain\" {\n\t\t\treturn \"valid-domain-name\", nil\n\t\t}\n\t\treturn \"\", errors.New(\"domain not found\")\n\t}\n\n\t// Test cases\n\ttests := []struct {\n\t\tname                 string\n\t\tdomainID             string\n\t\ttaskList             types.TaskList\n\t\ttaskListType         int\n\t\tforwardedFrom        string\n\t\tloadbalancerStrategy string\n\t\texpectedPartition    string\n\t}{\n\t\t{\n\t\t\tname:                 \"random partition when domainIDToName fails\",\n\t\t\tdomainID:             \"invalid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"random-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"round-robin partition enabled\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"round-robin\",\n\t\t\texpectedPartition:    \"roundrobin-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"random partition when round-robin disabled\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"invalid-enum\",\n\t\t\texpectedPartition:    \"random-partition\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - forwarded\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"somewhere\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - sticky\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindSticky.Ptr()},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname:                 \"cannot repartition - partition\",\n\t\t\tdomainID:             \"valid-domain\",\n\t\t\ttaskList:             types.TaskList{Name: \"/__cadence_sys/test-tasklist\"},\n\t\t\ttaskListType:         1,\n\t\t\tforwardedFrom:        \"\",\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\texpectedPartition:    \"/__cadence_sys/test-tasklist\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tt.domainID,\n\t\t\t\tTaskList:      &tt.taskList,\n\t\t\t\tForwardedFrom: tt.forwardedFrom,\n\t\t\t}\n\t\t\t// Mock behavior for random and round robin load balancers\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Mock the LoadBalancer interface\n\t\t\trandomMock := NewMockLoadBalancer(ctrl)\n\t\t\troundRobinMock := NewMockLoadBalancer(ctrl)\n\t\t\trandomMock.EXPECT().PickReadPartition(tt.taskListType, req, \"\").Return(\"random-partition\").AnyTimes()\n\t\t\troundRobinMock.EXPECT().PickReadPartition(tt.taskListType, req, \"\").Return(\"roundrobin-partition\").AnyTimes()\n\n\t\t\t// Mock dynamic config for loadbalancer strategy\n\t\t\tloadbalancerStrategyFn := func(domainName, taskListName string, taskListType int) string {\n\t\t\t\treturn tt.loadbalancerStrategy\n\t\t\t}\n\n\t\t\t// Create multiLoadBalancer\n\t\t\tlb := &multiLoadBalancer{\n\t\t\t\tdefaultLoadBalancer: randomMock,\n\t\t\t\tloadBalancers: map[string]LoadBalancer{\n\t\t\t\t\t\"round-robin\": roundRobinMock,\n\t\t\t\t},\n\t\t\t\tdomainIDToName:       domainIDToName,\n\t\t\t\tloadbalancerStrategy: loadbalancerStrategyFn,\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t}\n\n\t\t\t// Call PickReadPartition and assert result\n\t\t\tpartition := lb.PickReadPartition(tt.taskListType, req, \"\")\n\t\t\tassert.Equal(t, tt.expectedPartition, partition)\n\t\t})\n\t}\n}\n\nfunc TestMultiLoadBalancer_UpdateWeight(t *testing.T) {\n\t// Mock the domainIDToName function\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\tif domainID == \"valid-domain\" {\n\t\t\treturn \"valid-domain-name\", nil\n\t\t}\n\t\treturn \"\", errors.New(\"domain not found\")\n\t}\n\n\t// Test cases\n\ttests := []struct {\n\t\tname                 string\n\t\tdomainID             string\n\t\ttaskList             types.TaskList\n\t\ttaskListType         int\n\t\tforwardedFrom        string\n\t\tpartition            string\n\t\tloadBalancerHints    *types.LoadBalancerHints\n\t\tloadbalancerStrategy string\n\t\tshouldUpdate         bool\n\t}{\n\t\t{\n\t\t\tname:          \"do nothing when domainIDToName fails\",\n\t\t\tdomainID:      \"invalid-domain\",\n\t\t\ttaskList:      types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tpartition:     \"partition-1\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount:  10,\n\t\t\t\tRatePerSecond: 1,\n\t\t\t},\n\t\t\tloadbalancerStrategy: \"random\",\n\t\t\tshouldUpdate:         false,\n\t\t},\n\t\t{\n\t\t\tname:          \"update weight with round-robin load balancer\",\n\t\t\tdomainID:      \"valid-domain\",\n\t\t\ttaskList:      types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tpartition:     \"partition-2\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount:  20,\n\t\t\t\tRatePerSecond: 2,\n\t\t\t},\n\t\t\tloadbalancerStrategy: \"round-robin\",\n\t\t\tshouldUpdate:         true,\n\t\t},\n\t\t{\n\t\t\tname:          \"do nothing when strategy is unsupported\",\n\t\t\tdomainID:      \"valid-domain\",\n\t\t\ttaskList:      types.TaskList{Name: \"test-tasklist\"},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tpartition:     \"partition-3\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount:  30,\n\t\t\t\tRatePerSecond: 3,\n\t\t\t},\n\t\t\tloadbalancerStrategy: \"invalid-strategy\",\n\t\t\tshouldUpdate:         false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tt.domainID,\n\t\t\t\tTaskList:      &tt.taskList,\n\t\t\t\tForwardedFrom: tt.forwardedFrom,\n\t\t\t}\n\t\t\t// Mock behavior for random and round-robin load balancers\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Mock the LoadBalancer interface\n\t\t\trandomMock := NewMockLoadBalancer(ctrl)\n\t\t\troundRobinMock := NewMockLoadBalancer(ctrl)\n\n\t\t\tif tt.shouldUpdate {\n\t\t\t\troundRobinMock.EXPECT().UpdateWeight(tt.taskListType, req, tt.partition, tt.loadBalancerHints).Times(1)\n\t\t\t} else {\n\t\t\t\troundRobinMock.EXPECT().UpdateWeight(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(0)\n\t\t\t}\n\n\t\t\tloadbalancerStrategyFn := func(domainName, taskListName string, taskListType int) string {\n\t\t\t\treturn tt.loadbalancerStrategy\n\t\t\t}\n\n\t\t\t// Create multiLoadBalancer\n\t\t\tlb := &multiLoadBalancer{\n\t\t\t\tdefaultLoadBalancer: randomMock,\n\t\t\t\tloadBalancers: map[string]LoadBalancer{\n\t\t\t\t\t\"round-robin\": roundRobinMock,\n\t\t\t\t},\n\t\t\t\tdomainIDToName:       domainIDToName,\n\t\t\t\tloadbalancerStrategy: loadbalancerStrategyFn,\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t}\n\n\t\t\t// Call UpdateWeight\n\t\t\tlb.UpdateWeight(tt.taskListType, req, tt.partition, tt.loadBalancerHints)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "client/matching/partition_config_provider.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination partition_config_provider_mock.go -package matching github.com/uber/cadence/client/matching PartitionConfigProvider\n\npackage matching\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// PartitionConfigProvider is the interface for implementers of\n\t// component that provides partition configuration for task list\n\t// partitions\n\tPartitionConfigProvider interface {\n\t\t// GetNumberOfReadPartitions returns the number of read partitions\n\t\tGetNumberOfReadPartitions(domainID string, taskList types.TaskList, taskListType int) int\n\t\t// GetNumberOfWritePartitions returns the number of write partitions\n\t\tGetNumberOfWritePartitions(domainID string, taskList types.TaskList, taskListType int) int\n\t\t// GetPartitionConfig returns the cached partition configuration\n\t\tGetPartitionConfig(domainID string, taskList types.TaskList, taskListType int) *types.TaskListPartitionConfig\n\t\t// UpdatePartitionConfig updates the partition configuration for a task list\n\t\tUpdatePartitionConfig(domainID string, taskList types.TaskList, taskListType int, config *types.TaskListPartitionConfig)\n\t\t// InvalidatePartitionCache invalidates the cached partition configuration for a task list\n\t\tInvalidatePartitionCache(domainID string, taskList types.TaskList, taskListType int)\n\t\t// GetMetricsClient returns the metrics client\n\t\tGetMetricsClient() metrics.Client\n\t\t// GetLogger returns the logger\n\t\tGetLogger() log.Logger\n\t}\n\n\tsyncedTaskListPartitionConfig struct {\n\t\tsync.RWMutex\n\t\ttypes.TaskListPartitionConfig\n\t}\n\n\tpartitionConfigProviderImpl struct {\n\t\tconfigCache         cache.Cache\n\t\tlogger              log.Logger\n\t\tmetricsClient       metrics.Client\n\t\tdomainIDToName      func(string) (string, error)\n\t\tenableReadFromCache dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tnReadPartitions     dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tnWritePartitions    dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t}\n)\n\nvar singlePartitionConfig = createDefaultConfig(1, 1)\n\nfunc (c *syncedTaskListPartitionConfig) updateConfig(newConfig types.TaskListPartitionConfig) bool {\n\tc.Lock()\n\tdefer c.Unlock()\n\tif c.Version < newConfig.Version {\n\t\tc.TaskListPartitionConfig = newConfig\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc NewPartitionConfigProvider(\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdomainIDToName func(string) (string, error),\n\tdc *dynamicconfig.Collection,\n) PartitionConfigProvider {\n\treturn &partitionConfigProviderImpl{\n\t\tlogger:              logger,\n\t\tmetricsClient:       metricsClient,\n\t\tdomainIDToName:      domainIDToName,\n\t\tenableReadFromCache: dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache),\n\t\tnReadPartitions:     dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingNumTasklistReadPartitions),\n\t\tnWritePartitions:    dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingNumTasklistWritePartitions),\n\t\tconfigCache: cache.New(&cache.Options{\n\t\t\tTTL:             0,\n\t\t\tInitialCapacity: 100,\n\t\t\tPin:             false,\n\t\t\tMaxCount:        3000,\n\t\t\tActivelyEvict:   false,\n\t\t\tMetricsScope:    metricsClient.Scope(metrics.PartitionConfigProviderScope),\n\t\t\tLogger:          logger,\n\t\t}),\n\t}\n}\n\nfunc (p *partitionConfigProviderImpl) GetNumberOfReadPartitions(domainID string, taskList types.TaskList, taskListType int) int {\n\tconfig := p.GetPartitionConfig(domainID, taskList, taskListType)\n\treturn len(config.ReadPartitions)\n}\n\nfunc (p *partitionConfigProviderImpl) GetNumberOfWritePartitions(domainID string, taskList types.TaskList, taskListType int) int {\n\tconfig := p.GetPartitionConfig(domainID, taskList, taskListType)\n\tv := config.Version\n\tw := len(config.WritePartitions)\n\tr := len(config.ReadPartitions)\n\tif w > r {\n\t\tp.logger.Warn(\"Number of write partitions exceeds number of read partitions, using number of read partitions\", tag.WorkflowDomainID(domainID), tag.WorkflowTaskListName(taskList.GetName()), tag.WorkflowTaskListType(taskListType), tag.Dynamic(\"read-partition\", r), tag.Dynamic(\"write-partition\", w), tag.Dynamic(\"config-version\", v))\n\t\treturn r\n\t}\n\treturn w\n}\n\nfunc (p *partitionConfigProviderImpl) GetPartitionConfig(domainID string, taskList types.TaskList, taskListType int) *types.TaskListPartitionConfig {\n\tdomainName, err := p.domainIDToName(domainID)\n\tif err != nil {\n\t\treturn createDefaultConfig(1, 1)\n\t}\n\tif !p.enableReadFromCache(domainName, taskList.GetName(), taskListType) {\n\t\tnWrite := p.nWritePartitions(domainName, taskList.GetName(), taskListType)\n\t\tnRead := p.nReadPartitions(domainName, taskList.GetName(), taskListType)\n\t\t// checks to make sure number of writes never exceeds number of reads\n\t\tif nWrite > nRead {\n\t\t\tp.logger.Warn(\"Number of write partitions exceeds number of read partitions, using number of read partitions\", tag.WorkflowDomainID(domainID), tag.WorkflowTaskListName(taskList.GetName()), tag.WorkflowTaskListType(taskListType), tag.Dynamic(\"read-partition\", nRead), tag.Dynamic(\"write-partition\", nWrite))\n\t\t\tnWrite = nRead\n\t\t}\n\t\treturn createDefaultConfig(nRead, nWrite)\n\t}\n\tc := p.getCachedPartitionConfig(domainID, taskList, taskListType)\n\tif c == nil {\n\t\treturn singlePartitionConfig\n\t}\n\tc.RLock()\n\tconfig := c.TaskListPartitionConfig\n\tc.RUnlock()\n\tv := config.Version\n\tw := len(config.WritePartitions)\n\tr := len(config.ReadPartitions)\n\tscope := p.metricsClient.Scope(metrics.PartitionConfigProviderScope, metrics.DomainTag(domainName), metrics.TaskListRootPartitionTag(taskList.GetName()), getTaskListTypeTag(taskListType))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumReadGauge, float64(r))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumWriteGauge, float64(w))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigVersionGauge, float64(v))\n\n\treturn &config\n}\n\nfunc (p *partitionConfigProviderImpl) UpdatePartitionConfig(domainID string, taskList types.TaskList, taskListType int, config *types.TaskListPartitionConfig) {\n\tif config == nil || taskList.GetKind() != types.TaskListKindNormal {\n\t\treturn\n\t}\n\ttaskListKey := key{\n\t\tdomainID:     domainID,\n\t\ttaskListName: taskList.Name,\n\t\ttaskListType: taskListType,\n\t}\n\tvar err error\n\tcI := p.configCache.Get(taskListKey)\n\tif cI == nil {\n\t\tcI, err = p.configCache.PutIfNotExist(taskListKey, &syncedTaskListPartitionConfig{TaskListPartitionConfig: *config})\n\t\tif err != nil {\n\t\t\tp.logger.Error(\"Failed put partition config into cache\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t}\n\tc, ok := cI.(*syncedTaskListPartitionConfig)\n\tif !ok {\n\t\treturn\n\t}\n\tupdated := c.updateConfig(*config)\n\tif updated {\n\t\tw := len(c.WritePartitions)\n\t\tr := len(c.ReadPartitions)\n\t\tp.logger.Info(\"tasklist partition config updated\", tag.WorkflowDomainID(domainID), tag.WorkflowTaskListName(taskList.Name), tag.WorkflowTaskListType(taskListType), tag.Dynamic(\"read-partition\", r), tag.Dynamic(\"write-partition\", w), tag.Dynamic(\"config-version\", config.Version))\n\t}\n}\n\nfunc (p *partitionConfigProviderImpl) GetMetricsClient() metrics.Client {\n\treturn p.metricsClient\n}\n\nfunc (p *partitionConfigProviderImpl) GetLogger() log.Logger {\n\treturn p.logger\n}\n\nfunc (p *partitionConfigProviderImpl) InvalidatePartitionCache(domainID string, taskList types.TaskList, taskListType int) {\n\tif taskList.GetKind() != types.TaskListKindNormal {\n\t\treturn\n\t}\n\ttaskListKey := key{\n\t\tdomainID:     domainID,\n\t\ttaskListName: taskList.Name,\n\t\ttaskListType: taskListType,\n\t}\n\tp.configCache.Delete(taskListKey)\n\tp.logger.Info(\"tasklist partition config cache invalidated\", tag.WorkflowDomainID(domainID), tag.WorkflowTaskListName(taskList.Name), tag.WorkflowTaskListType(taskListType))\n}\n\nfunc (p *partitionConfigProviderImpl) getCachedPartitionConfig(domainID string, taskList types.TaskList, taskListType int) *syncedTaskListPartitionConfig {\n\tif taskList.GetKind() != types.TaskListKindNormal {\n\t\treturn nil\n\t}\n\ttaskListKey := key{\n\t\tdomainID:     domainID,\n\t\ttaskListName: taskList.Name,\n\t\ttaskListType: taskListType,\n\t}\n\tcI := p.configCache.Get(taskListKey)\n\tif cI == nil {\n\t\tp.logger.Debug(\"Partition config not found in cache\", tag.WorkflowDomainID(domainID), tag.WorkflowTaskListName(taskList.Name), tag.WorkflowTaskListType(taskListType))\n\t\treturn nil\n\t}\n\tc, ok := cI.(*syncedTaskListPartitionConfig)\n\tif !ok {\n\t\treturn nil\n\t}\n\treturn c\n}\n\nfunc getTaskListTypeTag(taskListType int) metrics.Tag {\n\tswitch taskListType {\n\tcase persistence.TaskListTypeActivity:\n\t\treturn metrics.TaskListTypeTag(\"activity\")\n\tcase persistence.TaskListTypeDecision:\n\t\treturn metrics.TaskListTypeTag(\"decision\")\n\tdefault:\n\t\treturn metrics.TaskListTypeTag(\"\")\n\t}\n}\n\nfunc createDefaultConfig(nRead, nWrite int) *types.TaskListPartitionConfig {\n\tread := make(map[int]*types.TaskListPartition, nRead)\n\tfor i := 0; i < nRead; i++ {\n\t\tread[i] = &types.TaskListPartition{}\n\t}\n\twrite := make(map[int]*types.TaskListPartition, nWrite)\n\tfor i := 0; i < nWrite; i++ {\n\t\twrite[i] = &types.TaskListPartition{}\n\t}\n\treturn &types.TaskListPartitionConfig{\n\t\tVersion:         0,\n\t\tReadPartitions:  read,\n\t\tWritePartitions: write,\n\t}\n}\n"
  },
  {
    "path": "client/matching/partition_config_provider_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: partition_config_provider.go\n//\n// Generated by this command:\n//\n//\tmockgen -package matching -source partition_config_provider.go -destination partition_config_provider_mock.go -package matching github.com/uber/cadence/client/matching PartitionConfigProvider\n//\n\n// Package matching is a generated GoMock package.\npackage matching\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tlog \"github.com/uber/cadence/common/log\"\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockPartitionConfigProvider is a mock of PartitionConfigProvider interface.\ntype MockPartitionConfigProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPartitionConfigProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockPartitionConfigProviderMockRecorder is the mock recorder for MockPartitionConfigProvider.\ntype MockPartitionConfigProviderMockRecorder struct {\n\tmock *MockPartitionConfigProvider\n}\n\n// NewMockPartitionConfigProvider creates a new mock instance.\nfunc NewMockPartitionConfigProvider(ctrl *gomock.Controller) *MockPartitionConfigProvider {\n\tmock := &MockPartitionConfigProvider{ctrl: ctrl}\n\tmock.recorder = &MockPartitionConfigProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPartitionConfigProvider) EXPECT() *MockPartitionConfigProviderMockRecorder {\n\treturn m.recorder\n}\n\n// GetLogger mocks base method.\nfunc (m *MockPartitionConfigProvider) GetLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetLogger indicates an expected call of GetLogger.\nfunc (mr *MockPartitionConfigProviderMockRecorder) GetLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLogger\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).GetLogger))\n}\n\n// GetMetricsClient mocks base method.\nfunc (m *MockPartitionConfigProvider) GetMetricsClient() metrics.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsClient\")\n\tret0, _ := ret[0].(metrics.Client)\n\treturn ret0\n}\n\n// GetMetricsClient indicates an expected call of GetMetricsClient.\nfunc (mr *MockPartitionConfigProviderMockRecorder) GetMetricsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsClient\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).GetMetricsClient))\n}\n\n// GetNumberOfReadPartitions mocks base method.\nfunc (m *MockPartitionConfigProvider) GetNumberOfReadPartitions(domainID string, taskList types.TaskList, taskListType int) int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetNumberOfReadPartitions\", domainID, taskList, taskListType)\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetNumberOfReadPartitions indicates an expected call of GetNumberOfReadPartitions.\nfunc (mr *MockPartitionConfigProviderMockRecorder) GetNumberOfReadPartitions(domainID, taskList, taskListType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetNumberOfReadPartitions\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).GetNumberOfReadPartitions), domainID, taskList, taskListType)\n}\n\n// GetNumberOfWritePartitions mocks base method.\nfunc (m *MockPartitionConfigProvider) GetNumberOfWritePartitions(domainID string, taskList types.TaskList, taskListType int) int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetNumberOfWritePartitions\", domainID, taskList, taskListType)\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetNumberOfWritePartitions indicates an expected call of GetNumberOfWritePartitions.\nfunc (mr *MockPartitionConfigProviderMockRecorder) GetNumberOfWritePartitions(domainID, taskList, taskListType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetNumberOfWritePartitions\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).GetNumberOfWritePartitions), domainID, taskList, taskListType)\n}\n\n// GetPartitionConfig mocks base method.\nfunc (m *MockPartitionConfigProvider) GetPartitionConfig(domainID string, taskList types.TaskList, taskListType int) *types.TaskListPartitionConfig {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionConfig\", domainID, taskList, taskListType)\n\tret0, _ := ret[0].(*types.TaskListPartitionConfig)\n\treturn ret0\n}\n\n// GetPartitionConfig indicates an expected call of GetPartitionConfig.\nfunc (mr *MockPartitionConfigProviderMockRecorder) GetPartitionConfig(domainID, taskList, taskListType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionConfig\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).GetPartitionConfig), domainID, taskList, taskListType)\n}\n\n// InvalidatePartitionCache mocks base method.\nfunc (m *MockPartitionConfigProvider) InvalidatePartitionCache(domainID string, taskList types.TaskList, taskListType int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"InvalidatePartitionCache\", domainID, taskList, taskListType)\n}\n\n// InvalidatePartitionCache indicates an expected call of InvalidatePartitionCache.\nfunc (mr *MockPartitionConfigProviderMockRecorder) InvalidatePartitionCache(domainID, taskList, taskListType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InvalidatePartitionCache\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).InvalidatePartitionCache), domainID, taskList, taskListType)\n}\n\n// UpdatePartitionConfig mocks base method.\nfunc (m *MockPartitionConfigProvider) UpdatePartitionConfig(domainID string, taskList types.TaskList, taskListType int, config *types.TaskListPartitionConfig) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdatePartitionConfig\", domainID, taskList, taskListType, config)\n}\n\n// UpdatePartitionConfig indicates an expected call of UpdatePartitionConfig.\nfunc (mr *MockPartitionConfigProviderMockRecorder) UpdatePartitionConfig(domainID, taskList, taskListType, config any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdatePartitionConfig\", reflect.TypeOf((*MockPartitionConfigProvider)(nil).UpdatePartitionConfig), domainID, taskList, taskListType, config)\n}\n"
  },
  {
    "path": "client/matching/partition_config_provider_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestSyncedTaskListPartitionConfig(t *testing.T) {\n\tc := &syncedTaskListPartitionConfig{TaskListPartitionConfig: types.TaskListPartitionConfig{Version: 50}}\n\n\tvar g errgroup.Group\n\tfor i := int64(0); i < 100; i++ {\n\t\tv := i\n\t\tg.Go(func() error {\n\t\t\tc.updateConfig(types.TaskListPartitionConfig{Version: v})\n\t\t\treturn nil\n\t\t})\n\t}\n\trequire.NoError(t, g.Wait())\n\n\tassert.Equal(t, int64(99), c.Version)\n}\n\nfunc setUpMocksForPartitionConfigProvider(t *testing.T, enableReadFromCache bool) (*partitionConfigProviderImpl, *cache.MockCache) {\n\tctrl := gomock.NewController(t)\n\tmockCache := cache.NewMockCache(ctrl)\n\tlogger := log.NewNoop()\n\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\treturn \"test-domain\", nil\n\t}\n\n\treturn &partitionConfigProviderImpl{\n\t\tconfigCache:         mockCache,\n\t\tlogger:              logger,\n\t\tmetricsClient:       metrics.NewNoopMetricsClient(),\n\t\tdomainIDToName:      domainIDToName,\n\t\tenableReadFromCache: dynamicproperties.GetBoolPropertyFilteredByTaskListInfo(enableReadFromCache),\n\t\tnReadPartitions:     dynamicproperties.GetIntPropertyFilteredByTaskListInfo(3),\n\t\tnWritePartitions:    dynamicproperties.GetIntPropertyFilteredByTaskListInfo(5),\n\t}, mockCache\n}\n\nfunc TestNewPartitionConfigProvider(t *testing.T) {\n\tdc := dynamicconfig.NewCollection(dynamicconfig.NewNopClient(), testlogger.New(t))\n\tlogger := testlogger.New(t)\n\tdomainIDToName := func(domainID string) (string, error) {\n\t\treturn \"test-domain\", nil\n\t}\n\tp := NewPartitionConfigProvider(logger, metrics.NewNoopMetricsClient(), domainIDToName, dc)\n\tassert.NotNil(t, p)\n\tpImpl, ok := p.(*partitionConfigProviderImpl)\n\tassert.True(t, ok)\n\tassert.NotNil(t, pImpl)\n\tassert.Equal(t, logger, pImpl.logger)\n\tassert.NotNil(t, pImpl.configCache)\n\tassert.NotNil(t, pImpl.domainIDToName)\n\tassert.NotNil(t, pImpl.enableReadFromCache)\n\tassert.NotNil(t, pImpl.nReadPartitions)\n\tassert.NotNil(t, pImpl.nWritePartitions)\n}\n\nfunc TestGetNumberOfReadPartitions(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                string\n\t\ttaskListKind        types.TaskListKind\n\t\tenableReadFromCache bool\n\t\tcachedConfigExists  bool\n\t\texpectedPartitions  int\n\t}{\n\t\t{\n\t\t\tname:                \"get read partitions from dynamic config\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: false,\n\t\t\texpectedPartitions:  3,\n\t\t},\n\t\t{\n\t\t\tname:                \"get read partitions from cache\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  true,\n\t\t\texpectedPartitions:  4,\n\t\t},\n\t\t{\n\t\t\tname:                \"get read partitions for sticky tasklist\",\n\t\t\ttaskListKind:        types.TaskListKindSticky,\n\t\t\tenableReadFromCache: true,\n\t\t\texpectedPartitions:  1,\n\t\t},\n\t\t{\n\t\t\tname:                \"cache config missing, fallback to default\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  false,\n\t\t\texpectedPartitions:  1,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpartitionProvider, mockCache := setUpMocksForPartitionConfigProvider(t, tc.enableReadFromCache)\n\n\t\t\tif tc.enableReadFromCache && tc.taskListKind == types.TaskListKindNormal {\n\t\t\t\tif tc.cachedConfigExists {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(&syncedTaskListPartitionConfig{\n\t\t\t\t\t\tTaskListPartitionConfig: types.TaskListPartitionConfig{ReadPartitions: partitions(4)},\n\t\t\t\t\t}).Times(1)\n\t\t\t\t} else {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(nil).Times(1)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\t\t\tp := partitionProvider.GetNumberOfReadPartitions(\"test-domain-id\", taskList, 0)\n\n\t\t\t// Validate result\n\t\t\tassert.Equal(t, tc.expectedPartitions, p)\n\t\t})\n\t}\n}\n\nfunc TestGetNumberOfWritePartitions(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                string\n\t\ttaskListKind        types.TaskListKind\n\t\tenableReadFromCache bool\n\t\tcachedConfigExists  bool\n\t\texpectedPartitions  int\n\t}{\n\t\t{\n\t\t\tname:                \"get write partitions from dynamic config\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: false,\n\t\t\texpectedPartitions:  3, // nWritePartitions is 5 but capped by nReadPartitions which is 3\n\t\t},\n\t\t{\n\t\t\tname:                \"get write partitions from cache\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  true,\n\t\t\texpectedPartitions:  2,\n\t\t},\n\t\t{\n\t\t\tname:                \"get write partitions from cache for sticky tasklist\",\n\t\t\ttaskListKind:        types.TaskListKindSticky,\n\t\t\tenableReadFromCache: true,\n\t\t\texpectedPartitions:  1,\n\t\t},\n\t\t{\n\t\t\tname:                \"cache config missing, fallback to default\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  false,\n\t\t\texpectedPartitions:  1,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpartitionProvider, mockCache := setUpMocksForPartitionConfigProvider(t, tc.enableReadFromCache)\n\n\t\t\tif tc.enableReadFromCache && tc.taskListKind == types.TaskListKindNormal {\n\t\t\t\tif tc.cachedConfigExists {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(&syncedTaskListPartitionConfig{\n\t\t\t\t\t\tTaskListPartitionConfig: types.TaskListPartitionConfig{ReadPartitions: partitions(2), WritePartitions: partitions(5)},\n\t\t\t\t\t}).Times(1)\n\t\t\t\t} else {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(nil).Times(1)\n\t\t\t\t}\n\t\t\t}\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\t\t\tp := partitionProvider.GetNumberOfWritePartitions(\"test-domain-id\", taskList, 0)\n\n\t\t\t// Validate result\n\t\t\tassert.Equal(t, tc.expectedPartitions, p)\n\t\t})\n\t}\n}\n\nfunc TestGetPartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                string\n\t\ttaskListKind        types.TaskListKind\n\t\tenableReadFromCache bool\n\t\tcachedConfigExists  bool\n\t\texpected            *types.TaskListPartitionConfig\n\t}{\n\t\t{\n\t\t\tname:                \"get from dynamic config\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: false,\n\t\t\texpected:            createDefaultConfig(3, 3),\n\t\t},\n\t\t{\n\t\t\tname:                \"get from cache\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  true,\n\t\t\t// It's wrong but it's the value we've got\n\t\t\texpected: createDefaultConfig(2, 5),\n\t\t},\n\t\t{\n\t\t\tname:                \"get from cache for sticky tasklist\",\n\t\t\ttaskListKind:        types.TaskListKindSticky,\n\t\t\tenableReadFromCache: true,\n\t\t\texpected:            createDefaultConfig(1, 1),\n\t\t},\n\t\t{\n\t\t\tname:                \"cache config missing, fallback to default\",\n\t\t\ttaskListKind:        types.TaskListKindNormal,\n\t\t\tenableReadFromCache: true,\n\t\t\tcachedConfigExists:  false,\n\t\t\texpected:            createDefaultConfig(1, 1),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpartitionProvider, mockCache := setUpMocksForPartitionConfigProvider(t, tc.enableReadFromCache)\n\n\t\t\tif tc.enableReadFromCache && tc.taskListKind == types.TaskListKindNormal {\n\t\t\t\tif tc.cachedConfigExists {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(&syncedTaskListPartitionConfig{\n\t\t\t\t\t\tTaskListPartitionConfig: types.TaskListPartitionConfig{ReadPartitions: partitions(2), WritePartitions: partitions(5)},\n\t\t\t\t\t}).Times(1)\n\t\t\t\t} else {\n\t\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(nil).Times(1)\n\t\t\t\t}\n\t\t\t}\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\t\t\tconfig := partitionProvider.GetPartitionConfig(\"test-domain-id\", taskList, 0)\n\n\t\t\t// Validate result\n\t\t\tassert.Equal(t, tc.expected, config)\n\t\t})\n\t}\n}\n\nfunc TestUpdatePartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tinput             *types.TaskListPartitionConfig\n\t\tconfigExists      bool\n\t\texpectedNewConfig bool\n\t}{\n\t\t{\n\t\t\tname:              \"config exists, update config\",\n\t\t\tinput:             &types.TaskListPartitionConfig{Version: 2},\n\t\t\tconfigExists:      true,\n\t\t\texpectedNewConfig: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"config does not exist, create new config\",\n\t\t\tinput:             &types.TaskListPartitionConfig{Version: 2},\n\t\t\tconfigExists:      false,\n\t\t\texpectedNewConfig: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpartitionProvider, mockCache := setUpMocksForPartitionConfigProvider(t, true)\n\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\"}\n\n\t\t\tif tc.configExists {\n\t\t\t\texistingConfig := &syncedTaskListPartitionConfig{\n\t\t\t\t\tTaskListPartitionConfig: types.TaskListPartitionConfig{Version: 1},\n\t\t\t\t}\n\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(existingConfig).Times(1)\n\t\t\t} else {\n\t\t\t\tmockCache.EXPECT().Get(gomock.Any()).Return(nil).Times(1)\n\t\t\t\tmockCache.EXPECT().PutIfNotExist(gomock.Any(), gomock.Any()).Return(&syncedTaskListPartitionConfig{}, nil).Times(1)\n\t\t\t}\n\n\t\t\tpartitionProvider.UpdatePartitionConfig(\"test-domain-id\", taskList, 0, tc.input)\n\t\t})\n\t}\n}\n\nfunc TestInvalidatePartitionCache(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\ttaskListKind     types.TaskListKind\n\t\texpectDeleteCall bool\n\t}{\n\t\t{\n\t\t\tname:             \"invalidate cache for normal task list\",\n\t\t\ttaskListKind:     types.TaskListKindNormal,\n\t\t\texpectDeleteCall: true,\n\t\t},\n\t\t{\n\t\t\tname:             \"skip invalidation for sticky task list\",\n\t\t\ttaskListKind:     types.TaskListKindSticky,\n\t\t\texpectDeleteCall: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpartitionProvider, mockCache := setUpMocksForPartitionConfigProvider(t, true)\n\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\n\t\t\tif tc.expectDeleteCall {\n\t\t\t\texpectedKey := key{\n\t\t\t\t\tdomainID:     \"test-domain-id\",\n\t\t\t\t\ttaskListName: \"test-task-list\",\n\t\t\t\t\ttaskListType: 0,\n\t\t\t\t}\n\t\t\t\tmockCache.EXPECT().Delete(expectedKey).Times(1)\n\t\t\t}\n\n\t\t\tpartitionProvider.InvalidatePartitionCache(\"test-domain-id\", taskList, 0)\n\t\t})\n\t}\n}\n\nfunc partitions(num int) map[int]*types.TaskListPartition {\n\tresult := make(map[int]*types.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tresult[i] = &types.TaskListPartition{}\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "client/matching/peer_resolver.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage matching\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// PeerResolver is used to resolve matching peers.\n// Those are deployed instances of Cadence matching services that participate in the cluster ring.\n// The resulting peer is simply an address of form ip:port where RPC calls can be routed to.\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination peer_resolver_mock.go -package matching github.com/uber/cadence/client/matching PeerResolver\n\ntype PeerResolver interface {\n\tFromTaskList(taskListName string) (string, error)\n\tGetAllPeers() ([]string, error)\n\tFromHostAddress(hostAddress string) (string, error)\n}\n\ntype peerResolver struct {\n\tresolver  membership.Resolver\n\tnamedPort string // grpc or tchannel, depends on yarpc configuration\n}\n\n// NewPeerResolver creates a new matching peer resolver.\nfunc NewPeerResolver(membership membership.Resolver, namedPort string) PeerResolver {\n\treturn peerResolver{\n\t\tresolver:  membership,\n\t\tnamedPort: namedPort,\n\t}\n}\n\n// FromTaskList resolves the matching peer responsible for the given task list name.\n// It uses our membership provider to lookup which instance currently owns the given task list.\n// FromHostAddress is used for further resolving.\nfunc (pr peerResolver) FromTaskList(taskListName string) (string, error) {\n\thost, err := pr.resolver.Lookup(service.Matching, taskListName)\n\tif err != nil {\n\t\treturn \"\", common.ToServiceTransientError(err)\n\t}\n\n\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\treturn peer, common.ToServiceTransientError(err)\n}\n\n// GetAllPeers returns all matching service peers in the cluster ring.\nfunc (pr peerResolver) GetAllPeers() ([]string, error) {\n\thosts, err := pr.resolver.Members(service.Matching)\n\tif err != nil {\n\t\treturn nil, common.ToServiceTransientError(err)\n\t}\n\tpeers := make([]string, 0, len(hosts))\n\tfor _, host := range hosts {\n\t\tpeer, err := pr.FromHostAddress(host.GetAddress())\n\t\tif err != nil {\n\t\t\treturn nil, common.ToServiceTransientError(err)\n\t\t}\n\t\tpeers = append(peers, peer)\n\t}\n\treturn peers, nil\n}\n\n// FromHostAddress resolves the final matching peer responsible for the given host address.\n// The address may be used as is, or processed with additional address mapper.\n// In case of gRPC transport, the port within the address is replaced with gRPC port.\nfunc (pr peerResolver) FromHostAddress(hostAddress string) (string, error) {\n\thost, err := pr.resolver.LookupByAddress(service.Matching, hostAddress)\n\tif err != nil {\n\t\treturn \"\", common.ToServiceTransientError(err)\n\t}\n\n\tpeer, err := host.GetNamedAddress(pr.namedPort)\n\treturn peer, common.ToServiceTransientError(err)\n}\n"
  },
  {
    "path": "client/matching/peer_resolver_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: peer_resolver.go\n//\n// Generated by this command:\n//\n//\tmockgen -package matching -source peer_resolver.go -destination peer_resolver_mock.go -package matching github.com/uber/cadence/client/matching PeerResolver\n//\n\n// Package matching is a generated GoMock package.\npackage matching\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPeerResolver is a mock of PeerResolver interface.\ntype MockPeerResolver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerResolverMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerResolverMockRecorder is the mock recorder for MockPeerResolver.\ntype MockPeerResolverMockRecorder struct {\n\tmock *MockPeerResolver\n}\n\n// NewMockPeerResolver creates a new mock instance.\nfunc NewMockPeerResolver(ctrl *gomock.Controller) *MockPeerResolver {\n\tmock := &MockPeerResolver{ctrl: ctrl}\n\tmock.recorder = &MockPeerResolverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerResolver) EXPECT() *MockPeerResolverMockRecorder {\n\treturn m.recorder\n}\n\n// FromHostAddress mocks base method.\nfunc (m *MockPeerResolver) FromHostAddress(hostAddress string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromHostAddress\", hostAddress)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromHostAddress indicates an expected call of FromHostAddress.\nfunc (mr *MockPeerResolverMockRecorder) FromHostAddress(hostAddress any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromHostAddress\", reflect.TypeOf((*MockPeerResolver)(nil).FromHostAddress), hostAddress)\n}\n\n// FromTaskList mocks base method.\nfunc (m *MockPeerResolver) FromTaskList(taskListName string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromTaskList\", taskListName)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FromTaskList indicates an expected call of FromTaskList.\nfunc (mr *MockPeerResolverMockRecorder) FromTaskList(taskListName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromTaskList\", reflect.TypeOf((*MockPeerResolver)(nil).FromTaskList), taskListName)\n}\n\n// GetAllPeers mocks base method.\nfunc (m *MockPeerResolver) GetAllPeers() ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllPeers\")\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllPeers indicates an expected call of GetAllPeers.\nfunc (mr *MockPeerResolverMockRecorder) GetAllPeers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllPeers\", reflect.TypeOf((*MockPeerResolver)(nil).GetAllPeers))\n}\n"
  },
  {
    "path": "client/matching/peer_resolver_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage matching\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\tgomock \"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestPeerResolver(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tserviceResolver := membership.NewMockResolver(controller)\n\thost1 := membership.NewDetailedHostInfo(\n\t\t\"tasklistHost:1234\",\n\t\t\"tasklistHost_1234\",\n\t\tmembership.PortMap{\n\t\t\tmembership.PortTchannel: 1234,\n\t\t\tmembership.PortGRPC:     1244,\n\t\t},\n\t)\n\thost2 := membership.NewDetailedHostInfo(\n\t\t\"tasklistHost2:1235\",\n\t\t\"tasklistHost2_1235\",\n\t\tmembership.PortMap{\n\t\t\tmembership.PortTchannel: 1235,\n\t\t\tmembership.PortGRPC:     1245,\n\t\t},\n\t)\n\tserviceResolver.EXPECT().Lookup(service.Matching, \"taskListA\").Return(\n\t\thost1,\n\t\tnil)\n\tserviceResolver.EXPECT().Lookup(service.Matching, \"invalid\").Return(membership.HostInfo{}, assert.AnError)\n\tserviceResolver.EXPECT().LookupByAddress(service.Matching, \"invalid address\").Return(membership.HostInfo{}, assert.AnError)\n\n\tserviceResolver.EXPECT().Members(service.Matching).Return([]membership.HostInfo{\n\t\thost1,\n\t\thost2,\n\t}, nil)\n\n\tserviceResolver.EXPECT().LookupByAddress(service.Matching, \"tasklistHost2:1235\").Return(\n\t\thost2,\n\t\tnil,\n\t).AnyTimes()\n\tserviceResolver.EXPECT().LookupByAddress(service.Matching, \"tasklistHost:1234\").Return(\n\t\thost1,\n\t\tnil,\n\t).AnyTimes()\n\tr := NewPeerResolver(serviceResolver, membership.PortGRPC)\n\n\tpeer, err := r.FromTaskList(\"taskListA\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"tasklistHost:1244\", peer)\n\n\t_, err = r.FromTaskList(\"invalid\")\n\tassert.Error(t, err)\n\n\t_, err = r.FromHostAddress(\"invalid address\")\n\tassert.Error(t, err)\n\n\tpeers, err := r.GetAllPeers()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []string{\"tasklistHost:1244\", \"tasklistHost2:1245\"}, peers)\n\n}\n"
  },
  {
    "path": "client/matching/rr_loadbalancer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tkey struct {\n\t\tdomainID     string\n\t\ttaskListName string\n\t\ttaskListType int\n\t}\n\n\troundRobinLoadBalancer struct {\n\t\tprovider   PartitionConfigProvider\n\t\treadCache  cache.Cache\n\t\twriteCache cache.Cache\n\n\t\tpickPartitionFn func(domainName string, taskList types.TaskList, taskListType int, nPartitions int, partitionCache cache.Cache) string\n\t}\n)\n\nfunc NewRoundRobinLoadBalancer(\n\tprovider PartitionConfigProvider,\n) LoadBalancer {\n\treturn &roundRobinLoadBalancer{\n\t\tprovider: provider,\n\t\treadCache: cache.New(&cache.Options{\n\t\t\tTTL:             0,\n\t\t\tInitialCapacity: 100,\n\t\t\tPin:             false,\n\t\t\tMaxCount:        3000,\n\t\t\tActivelyEvict:   false,\n\t\t\tMetricsScope:    provider.GetMetricsClient().Scope(metrics.LoadBalancerScope),\n\t\t\tLogger:          provider.GetLogger(),\n\t\t}),\n\t\twriteCache: cache.New(&cache.Options{\n\t\t\tTTL:             0,\n\t\t\tInitialCapacity: 100,\n\t\t\tPin:             false,\n\t\t\tMaxCount:        3000,\n\t\t\tActivelyEvict:   false,\n\t\t\tMetricsScope:    provider.GetMetricsClient().Scope(metrics.LoadBalancerScope),\n\t\t\tLogger:          provider.GetLogger(),\n\t\t}),\n\t\tpickPartitionFn: pickPartition,\n\t}\n}\n\nfunc (lb *roundRobinLoadBalancer) PickWritePartition(\n\ttaskListType int,\n\treq WriteRequest,\n) string {\n\tnPartitions := lb.provider.GetNumberOfWritePartitions(req.GetDomainUUID(), *req.GetTaskList(), taskListType)\n\treturn lb.pickPartitionFn(req.GetDomainUUID(), *req.GetTaskList(), taskListType, nPartitions, lb.writeCache)\n}\n\nfunc (lb *roundRobinLoadBalancer) PickReadPartition(\n\ttaskListType int,\n\treq ReadRequest,\n\t_ string,\n) string {\n\tn := lb.provider.GetNumberOfReadPartitions(req.GetDomainUUID(), *req.GetTaskList(), taskListType)\n\treturn lb.pickPartitionFn(req.GetDomainUUID(), *req.GetTaskList(), taskListType, n, lb.readCache)\n}\n\nfunc pickPartition(\n\tdomainID string,\n\ttaskList types.TaskList,\n\ttaskListType int,\n\tnPartitions int,\n\tpartitionCache cache.Cache,\n) string {\n\ttaskListName := taskList.GetName()\n\tif nPartitions <= 1 {\n\t\treturn taskListName\n\t}\n\n\ttaskListKey := key{\n\t\tdomainID:     domainID,\n\t\ttaskListName: taskListName,\n\t\ttaskListType: taskListType,\n\t}\n\n\tvalI := partitionCache.Get(taskListKey)\n\tif valI == nil {\n\t\tval := int64(-1)\n\t\tvar err error\n\t\tvalI, err = partitionCache.PutIfNotExist(taskListKey, &val)\n\t\tif err != nil {\n\t\t\treturn taskListName\n\t\t}\n\t}\n\tvalAddr, ok := valI.(*int64)\n\tif !ok {\n\t\treturn taskListName\n\t}\n\n\tp := int(atomic.AddInt64(valAddr, 1) % int64(nPartitions))\n\treturn getPartitionTaskListName(taskList.GetName(), p)\n}\n\nfunc (lb *roundRobinLoadBalancer) UpdateWeight(\n\ttaskListType int,\n\treq ReadRequest,\n\tpartition string,\n\tinfo *types.LoadBalancerHints,\n) {\n}\n"
  },
  {
    "path": "client/matching/rr_loadbalancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewRoundRobinLoadBalancer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tp := NewMockPartitionConfigProvider(ctrl)\n\tp.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tp.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tlb := NewRoundRobinLoadBalancer(p)\n\tassert.NotNil(t, lb)\n\trb, ok := lb.(*roundRobinLoadBalancer)\n\tassert.NotNil(t, rb)\n\tassert.True(t, ok)\n\tassert.Equal(t, p, rb.provider)\n}\n\nfunc TestPickPartition(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tdomainID       string\n\t\ttaskList       types.TaskList\n\t\ttaskListType   int\n\t\tforwardedFrom  string\n\t\tnPartitions    int\n\t\tsetupCache     func(mockCache *cache.MockCache)\n\t\texpectedResult string\n\t}{\n\t\t{\n\t\t\tname:           \"nPartitions <= 1\",\n\t\t\tdomainID:       \"testDomain\",\n\t\t\ttaskList:       types.TaskList{Name: \"testTaskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\ttaskListType:   1,\n\t\t\tforwardedFrom:  \"\",\n\t\t\tnPartitions:    1,\n\t\t\tsetupCache:     nil,\n\t\t\texpectedResult: \"testTaskList\",\n\t\t},\n\t\t{\n\t\t\tname:          \"Cache miss and partitioned task list\",\n\t\t\tdomainID:      \"testDomain\",\n\t\t\ttaskList:      types.TaskList{Name: \"testTaskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tnPartitions:   3,\n\t\t\tsetupCache: func(mockCache *cache.MockCache) {\n\t\t\t\tmockCache.EXPECT().Get(key{\n\t\t\t\t\tdomainID:     \"testDomain\",\n\t\t\t\t\ttaskListName: \"testTaskList\",\n\t\t\t\t\ttaskListType: 1,\n\t\t\t\t}).Return(nil)\n\t\t\t\tmockCache.EXPECT().PutIfNotExist(key{\n\t\t\t\t\tdomainID:     \"testDomain\",\n\t\t\t\t\ttaskListName: \"testTaskList\",\n\t\t\t\t\ttaskListType: 1,\n\t\t\t\t}, gomock.Any()).DoAndReturn(func(key key, val interface{}) (interface{}, error) {\n\t\t\t\t\tif *val.(*int64) != -1 {\n\t\t\t\t\t\tpanic(\"Expected value to be -1\")\n\t\t\t\t\t}\n\t\t\t\t\treturn val, nil\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedResult: \"testTaskList\",\n\t\t},\n\t\t{\n\t\t\tname:          \"Cache error and partitioned task list\",\n\t\t\tdomainID:      \"testDomain\",\n\t\t\ttaskList:      types.TaskList{Name: \"testTaskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tnPartitions:   3,\n\t\t\tsetupCache: func(mockCache *cache.MockCache) {\n\t\t\t\tmockCache.EXPECT().Get(key{\n\t\t\t\t\tdomainID:     \"testDomain\",\n\t\t\t\t\ttaskListName: \"testTaskList\",\n\t\t\t\t\ttaskListType: 1,\n\t\t\t\t}).Return(nil)\n\t\t\t\tmockCache.EXPECT().PutIfNotExist(key{\n\t\t\t\t\tdomainID:     \"testDomain\",\n\t\t\t\t\ttaskListName: \"testTaskList\",\n\t\t\t\t\ttaskListType: 1,\n\t\t\t\t}, gomock.Any()).Return(nil, fmt.Errorf(\"cache error\"))\n\t\t\t},\n\t\t\texpectedResult: \"testTaskList\",\n\t\t},\n\t\t{\n\t\t\tname:          \"Cache hit and partitioned task list\",\n\t\t\tdomainID:      \"testDomain\",\n\t\t\ttaskList:      types.TaskList{Name: \"testTaskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\ttaskListType:  1,\n\t\t\tforwardedFrom: \"\",\n\t\t\tnPartitions:   3,\n\t\t\tsetupCache: func(mockCache *cache.MockCache) {\n\t\t\t\tmockCache.EXPECT().Get(key{\n\t\t\t\t\tdomainID:     \"testDomain\",\n\t\t\t\t\ttaskListName: \"testTaskList\",\n\t\t\t\t\ttaskListType: 1,\n\t\t\t\t}).Return(new(int64))\n\t\t\t},\n\t\t\texpectedResult: \"/__cadence_sys/testTaskList/1\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockCache := cache.NewMockCache(ctrl)\n\n\t\t\t// If the test requires setting up cache behavior, call setupCache\n\t\t\tif tt.setupCache != nil {\n\t\t\t\ttt.setupCache(mockCache)\n\t\t\t}\n\n\t\t\t// Call the pickPartition function\n\t\t\tresult := pickPartition(\n\t\t\t\ttt.domainID,\n\t\t\t\ttt.taskList,\n\t\t\t\ttt.taskListType,\n\t\t\t\ttt.nPartitions,\n\t\t\t\tmockCache,\n\t\t\t)\n\n\t\t\t// Assert that the result matches the expected result\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n}\n\nfunc setUpMocksForRoundRobinLoadBalancer(t *testing.T, pickPartitionFn func(domainID string, taskList types.TaskList, taskListType int, nPartitions int, partitionCache cache.Cache) string) (*roundRobinLoadBalancer, *MockPartitionConfigProvider, *cache.MockCache) {\n\tctrl := gomock.NewController(t)\n\tmockProvider := NewMockPartitionConfigProvider(ctrl)\n\tmockCache := cache.NewMockCache(ctrl)\n\n\treturn &roundRobinLoadBalancer{\n\t\tprovider:        mockProvider,\n\t\treadCache:       mockCache,\n\t\twriteCache:      mockCache,\n\t\tpickPartitionFn: pickPartitionFn,\n\t}, mockProvider, mockCache\n}\n\nfunc TestRoundRobinPickWritePartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tforwardedFrom     string\n\t\ttaskListType      int\n\t\tnPartitions       int\n\t\ttaskListKind      types.TaskListKind\n\t\texpectedPartition string\n\t}{\n\t\t{\n\t\t\tname:              \"multiple write partitions, no forward\",\n\t\t\tforwardedFrom:     \"\",\n\t\t\ttaskListType:      0,\n\t\t\tnPartitions:       3,\n\t\t\ttaskListKind:      types.TaskListKindNormal,\n\t\t\texpectedPartition: \"custom-partition\", // Simulated by fake pickPartitionFn\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Fake pickPartitionFn behavior\n\t\t\tfakePickPartitionFn := func(domainID string, taskList types.TaskList, taskListType int, nPartitions int, partitionCache cache.Cache) string {\n\t\t\t\tassert.Equal(t, \"test-domain-id\", domainID)\n\t\t\t\tassert.Equal(t, \"test-task-list\", taskList.Name)\n\t\t\t\tassert.Equal(t, tc.taskListKind, taskList.GetKind())\n\t\t\t\tassert.Equal(t, tc.taskListType, taskListType)\n\t\t\t\tassert.Equal(t, tc.nPartitions, nPartitions)\n\t\t\t\tif taskList.GetKind() == types.TaskListKindSticky {\n\t\t\t\t\treturn taskList.GetName()\n\t\t\t\t}\n\t\t\t\treturn \"custom-partition\"\n\t\t\t}\n\n\t\t\t// Set up mocks with the fake pickPartitionFn\n\t\t\tloadBalancer, mockProvider, _ := setUpMocksForRoundRobinLoadBalancer(t, fakePickPartitionFn)\n\n\t\t\tmockProvider.EXPECT().\n\t\t\t\tGetNumberOfWritePartitions(\"test-domain-id\", types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind}, tc.taskListType).\n\t\t\t\tReturn(tc.nPartitions).\n\t\t\t\tTimes(1)\n\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    \"test-domain-id\",\n\t\t\t\tTaskList:      &taskList,\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tpartition := loadBalancer.PickWritePartition(tc.taskListType, req)\n\n\t\t\t// Validate result\n\t\t\tassert.Equal(t, tc.expectedPartition, partition)\n\t\t})\n\t}\n}\n\nfunc TestRoundRobinPickReadPartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tforwardedFrom     string\n\t\ttaskListType      int\n\t\tnPartitions       int\n\t\ttaskListKind      types.TaskListKind\n\t\texpectedPartition string\n\t}{\n\t\t{\n\t\t\tname:              \"multiple read partitions, no forward\",\n\t\t\tforwardedFrom:     \"\",\n\t\t\ttaskListType:      0,\n\t\t\tnPartitions:       3,\n\t\t\ttaskListKind:      types.TaskListKindNormal,\n\t\t\texpectedPartition: \"custom-partition\", // Simulated by fake pickPartitionFn\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Fake pickPartitionFn behavior\n\t\t\tfakePickPartitionFn := func(domainID string, taskList types.TaskList, taskListType int, nPartitions int, partitionCache cache.Cache) string {\n\t\t\t\tassert.Equal(t, \"test-domain-id\", domainID)\n\t\t\t\tassert.Equal(t, \"test-task-list\", taskList.Name)\n\t\t\t\tassert.Equal(t, tc.taskListKind, taskList.GetKind())\n\t\t\t\tassert.Equal(t, tc.taskListType, taskListType)\n\t\t\t\tassert.Equal(t, tc.nPartitions, nPartitions)\n\t\t\t\tif taskList.GetKind() == types.TaskListKindSticky {\n\t\t\t\t\treturn taskList.GetName()\n\t\t\t\t}\n\t\t\t\treturn \"custom-partition\"\n\t\t\t}\n\n\t\t\t// Set up mocks with the fake pickPartitionFn\n\t\t\tloadBalancer, mockProvider, _ := setUpMocksForRoundRobinLoadBalancer(t, fakePickPartitionFn)\n\n\t\t\tmockProvider.EXPECT().\n\t\t\t\tGetNumberOfReadPartitions(\"test-domain-id\", types.TaskList{Name: \"test-task-list\", Kind: &tc.taskListKind}, tc.taskListType).\n\t\t\t\tReturn(tc.nPartitions).\n\t\t\t\tTimes(1)\n\n\t\t\tkind := tc.taskListKind\n\t\t\ttaskList := types.TaskList{Name: \"test-task-list\", Kind: &kind}\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    \"test-domain-id\",\n\t\t\t\tTaskList:      &taskList,\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tpartition := loadBalancer.PickReadPartition(tc.taskListType, req, \"\")\n\n\t\t\t// Validate result\n\t\t\tassert.Equal(t, tc.expectedPartition, partition)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "client/matching/weighted_loadbalancer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"math\"\n\t\"math/rand\"\n\t\"path\"\n\t\"sort\"\n\t\"strconv\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst _backlogThreshold = 100\n\ntype (\n\tweightSelector struct {\n\t\tsync.RWMutex\n\t\tweights     []int64\n\t\tinitialized bool\n\t\tthreshold   int64\n\t}\n\n\tweightedLoadBalancer struct {\n\t\tfallbackLoadBalancer LoadBalancer\n\t\tprovider             PartitionConfigProvider\n\t\tweightCache          cache.Cache\n\t\tlogger               log.Logger\n\t}\n)\n\nfunc newWeightSelector(n int, threshold int64) *weightSelector {\n\tpw := &weightSelector{weights: make([]int64, n), threshold: threshold}\n\tfor i := range pw.weights {\n\t\tpw.weights[i] = -1\n\t}\n\treturn pw\n}\n\nfunc (pw *weightSelector) pickBetween(partitions []int) int {\n\ttotalWeight := int64(0)\n\tcumulativeWeights := make([]int64, len(partitions))\n\tpw.RLock()\n\tdefer pw.RUnlock()\n\n\tif !pw.initialized {\n\t\treturn -1\n\t}\n\tshouldEnable := false\n\tfor i, partitionID := range partitions {\n\t\t// Invalid partition specified, bail out\n\t\tif partitionID >= len(pw.weights) {\n\t\t\treturn -1\n\t\t}\n\t\tweight := pw.weights[partitionID]\n\t\ttotalWeight += weight\n\t\tcumulativeWeights[i] = totalWeight\n\t\tif weight > pw.threshold {\n\t\t\tshouldEnable = true // only enable weight selection if backlog size is larger than the threshold\n\t\t}\n\t}\n\tif totalWeight <= 0 || !shouldEnable {\n\t\treturn -1\n\t}\n\tr := rand.Int63n(totalWeight)\n\tindex := sort.Search(len(cumulativeWeights), func(i int) bool {\n\t\treturn cumulativeWeights[i] > r\n\t})\n\treturn partitions[index]\n}\n\nfunc (pw *weightSelector) pick() (int, []int64) {\n\ttotalWeight := int64(0)\n\tpw.RLock()\n\tdefer pw.RUnlock()\n\tcumulativeWeights := make([]int64, len(pw.weights))\n\tif !pw.initialized {\n\t\treturn -1, cumulativeWeights\n\t}\n\tshouldDrain := false\n\tfor i, w := range pw.weights {\n\t\ttotalWeight += w\n\t\tcumulativeWeights[i] = totalWeight\n\t\tif w > pw.threshold {\n\t\t\tshouldDrain = true // only enable weight selection if backlog size is larger than the threshold\n\t\t}\n\t}\n\tif totalWeight <= 0 || !shouldDrain {\n\t\treturn -1, cumulativeWeights\n\t}\n\tr := rand.Int63n(totalWeight)\n\tindex := sort.Search(len(cumulativeWeights), func(i int) bool {\n\t\treturn cumulativeWeights[i] > r\n\t})\n\treturn index, cumulativeWeights\n}\n\nfunc (pw *weightSelector) update(n, p int, weight int64) {\n\tpw.Lock()\n\tdefer pw.Unlock()\n\tif n > len(pw.weights) {\n\t\tnewWeights := make([]int64, n)\n\t\tcopy(newWeights, pw.weights)\n\t\tfor i := len(pw.weights); i < n; i++ {\n\t\t\tnewWeights[i] = -1\n\t\t}\n\t\tpw.weights = newWeights\n\t\tpw.initialized = false\n\t} else if n < len(pw.weights) {\n\t\tpw.weights = pw.weights[:n]\n\t}\n\tif p < n { // the opposite condition can happen when the task list is scaled down and we get an update from a drained partition\n\t\tpw.weights[p] = weight\n\t}\n\tfor _, w := range pw.weights {\n\t\tif w == -1 {\n\t\t\treturn\n\t\t}\n\t}\n\tpw.initialized = true\n}\n\nfunc NewWeightedLoadBalancer(\n\tlb LoadBalancer,\n\tprovider PartitionConfigProvider,\n\tlogger log.Logger,\n) WeightedLoadBalancer {\n\treturn &weightedLoadBalancer{\n\t\tfallbackLoadBalancer: lb,\n\t\tprovider:             provider,\n\t\tweightCache: cache.New(&cache.Options{\n\t\t\tTTL:             0,\n\t\t\tInitialCapacity: 100,\n\t\t\tPin:             false,\n\t\t\tMaxCount:        3000,\n\t\t\tActivelyEvict:   false,\n\t\t\tMetricsScope:    provider.GetMetricsClient().Scope(metrics.LoadBalancerScope),\n\t\t\tLogger:          logger,\n\t\t}),\n\t\tlogger: logger,\n\t}\n}\n\nfunc (lb *weightedLoadBalancer) PickWritePartition(\n\ttaskListType int,\n\treq WriteRequest,\n) string {\n\treturn lb.fallbackLoadBalancer.PickWritePartition(taskListType, req)\n}\n\nfunc (lb *weightedLoadBalancer) PickReadPartition(\n\ttaskListType int,\n\treq ReadRequest,\n\tisolationGroup string,\n) string {\n\ttaskListKey := key{\n\t\tdomainID:     req.GetDomainUUID(),\n\t\ttaskListName: req.GetTaskList().GetName(),\n\t\ttaskListType: taskListType,\n\t}\n\twI := lb.weightCache.Get(taskListKey)\n\tif wI == nil {\n\t\treturn lb.fallbackLoadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\tw, ok := wI.(*weightSelector)\n\tif !ok {\n\t\treturn lb.fallbackLoadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\tp, cumulativeWeights := w.pick()\n\tlb.logger.Debug(\"pick read partition\", tag.WorkflowDomainID(req.GetDomainUUID()), tag.WorkflowTaskListName(req.GetTaskList().Name), tag.WorkflowTaskListType(taskListType), tag.Dynamic(\"cumulative-weights\", cumulativeWeights), tag.Dynamic(\"task-list-partition\", p))\n\tif p < 0 {\n\t\treturn lb.fallbackLoadBalancer.PickReadPartition(taskListType, req, isolationGroup)\n\t}\n\treturn getPartitionTaskListName(req.GetTaskList().GetName(), p)\n}\n\nfunc (lb *weightedLoadBalancer) UpdateWeight(\n\ttaskListType int,\n\treq ReadRequest,\n\tpartition string,\n\tinfo *types.LoadBalancerHints,\n) {\n\ttaskList := *req.GetTaskList()\n\tif info == nil {\n\t\treturn\n\t}\n\tp := 0\n\tif partition != taskList.GetName() {\n\t\tvar err error\n\t\tp, err = strconv.Atoi(path.Base(partition))\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\ttaskListKey := key{\n\t\tdomainID:     req.GetDomainUUID(),\n\t\ttaskListName: taskList.GetName(),\n\t\ttaskListType: taskListType,\n\t}\n\tn := lb.provider.GetNumberOfReadPartitions(req.GetDomainUUID(), taskList, taskListType)\n\tif n <= 1 {\n\t\tlb.weightCache.Delete(taskListKey)\n\t\treturn\n\t}\n\twI := lb.weightCache.Get(taskListKey)\n\tif wI == nil {\n\t\tvar err error\n\t\tw := newWeightSelector(n, _backlogThreshold)\n\t\twI, err = lb.weightCache.PutIfNotExist(taskListKey, w)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\tw, ok := wI.(*weightSelector)\n\tif !ok {\n\t\treturn\n\t}\n\tweight := calcWeightFromLoadBalancerHints(info)\n\tlb.logger.Debug(\"update task list partition weight\", tag.WorkflowDomainID(req.GetDomainUUID()), tag.WorkflowTaskListName(taskList.GetName()), tag.WorkflowTaskListType(taskListType), tag.Dynamic(\"task-list-partition\", p), tag.Dynamic(\"weight\", weight), tag.Dynamic(\"load-balancer-hints\", info))\n\tw.update(n, p, weight)\n}\n\nfunc (lb *weightedLoadBalancer) PickBetween(domainID, taskListName string, taskListType int, partitions []int) int {\n\tif len(partitions) == 0 {\n\t\treturn -1\n\t}\n\tif len(partitions) == 1 {\n\t\treturn partitions[0]\n\t}\n\ttaskListKey := key{\n\t\tdomainID:     domainID,\n\t\ttaskListName: taskListName,\n\t\ttaskListType: taskListType,\n\t}\n\twI := lb.weightCache.Get(taskListKey)\n\tif wI == nil {\n\t\treturn -1\n\t}\n\tw, ok := wI.(*weightSelector)\n\tif !ok {\n\t\treturn -1\n\t}\n\treturn w.pickBetween(partitions)\n}\n\nfunc calcWeightFromLoadBalancerHints(info *types.LoadBalancerHints) int64 {\n\t// according to Little's Law, the average number of tasks in the queue L = λW\n\t// where λ is the average arrival rate and W is the average wait time a task spends in the queue\n\t// here λ is the QPS and W is the average match latency which is 10ms\n\t// so the backlog hint should be backlog count + L.\n\tsmoothingNumber := int64(0)\n\tqps := info.RatePerSecond\n\tif qps > 0.01 {\n\t\tsmoothingNumber = int64(math.Ceil(qps * 0.01))\n\t}\n\treturn info.BacklogCount + smoothingNumber\n}\n"
  },
  {
    "path": "client/matching/weighted_loadbalancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage matching\n\nimport (\n\t\"math\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestPollerWeight(t *testing.T) {\n\tn := 4\n\tpw := newWeightSelector(n, 100)\n\t// uninitialized weights should return -1\n\tp, cumulativeWeights := pw.pick()\n\tassert.Equal(t, -1, p)\n\tassert.Equal(t, []int64{0, 0, 0, 0}, cumulativeWeights)\n\t// all 0 weights should return -1\n\tfor i := 0; i < n; i++ {\n\t\tpw.update(n, i, 0)\n\t\tp, cumulativeWeights := pw.pick()\n\t\tassert.Equal(t, -1, p)\n\t\tassert.Equal(t, []int64{0, 0, 0, 0}, cumulativeWeights)\n\t}\n\t// if only one item has non-zero weight, always pick that item\n\tpw.update(n, 3, 400)\n\tfor i := 0; i < 100; i++ {\n\t\tp, cumulativeWeights := pw.pick()\n\t\tassert.Equal(t, 3, p)\n\t\tassert.Equal(t, []int64{0, 0, 0, 400}, cumulativeWeights)\n\t}\n\tpw.update(n, 2, 300)\n\tpw.update(n, 1, 200)\n\tpw.update(n, 0, 100)\n\t// test pick probabilities\n\ttestPickProbHelper(t, pw, time.Now().UnixNano())\n\n\t// shrink size and test pick probabilities\n\tpw.update(n-1, 2, 200)\n\ttestPickProbHelper(t, pw, time.Now().UnixNano())\n\n\t// expand size and test pick probabilities\n\tpw.update(n, 3, 300)\n\tpw.update(n+1, 4, 400)\n\ttestPickProbHelper(t, pw, time.Now().UnixNano())\n\n\t// no panics\n\tpw.update(n, 4, 0)\n}\n\nfunc testPickProbHelper(t *testing.T, pw *weightSelector, seed int64) {\n\tt.Helper()\n\trand.Seed(seed)\n\t// Collect pick results\n\tresults := make(map[int]int)\n\tnumPicks := 1000000\n\tfor i := 0; i < numPicks; i++ {\n\t\tindex, _ := pw.pick()\n\t\tresults[index]++\n\t}\n\t// Calculate expected probabilities\n\ttotalWeight := int64(0)\n\tfor _, w := range pw.weights {\n\t\ttotalWeight += w\n\t}\n\texpectedProbs := make([]float64, len(pw.weights))\n\tfor i, w := range pw.weights {\n\t\texpectedProbs[i] = float64(w) / float64(totalWeight)\n\t}\n\t// Check that pick results are approximately proportional to weights\n\tfor i := 0; i < len(pw.weights); i++ {\n\t\texpectedCount := expectedProbs[i] * float64(numPicks)\n\t\tactualCount := float64(results[i])\n\t\tdelta := expectedCount * 0.02 // Allow 2% error margin\n\t\tif actualCount < expectedCount-delta || actualCount > expectedCount+delta {\n\t\t\tt.Errorf(\"Index %d: expected count approximately %.0f, got %d\", i, expectedCount, results[i])\n\t\t}\n\t}\n}\n\nfunc TestNewWeightedLoadBalancer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\troundRobinMock := NewMockLoadBalancer(ctrl)\n\tp := NewMockPartitionConfigProvider(ctrl)\n\tp.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tlogger := testlogger.New(t)\n\tp.EXPECT().GetLogger().Return(logger).AnyTimes()\n\tlb := NewWeightedLoadBalancer(roundRobinMock, p, logger)\n\tassert.NotNil(t, lb)\n\tweightedLB, ok := lb.(*weightedLoadBalancer)\n\tassert.NotNil(t, weightedLB)\n\tassert.True(t, ok)\n\tassert.Equal(t, roundRobinMock, weightedLB.fallbackLoadBalancer)\n\tassert.Equal(t, p, weightedLB.provider)\n\tassert.NotNil(t, weightedLB.weightCache)\n\tassert.NotNil(t, weightedLB.logger)\n}\n\nfunc TestWeightedLoadBalancer_PickWritePartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tdomainID       string\n\t\ttaskList       types.TaskList\n\t\ttaskListType   int\n\t\tforwardedFrom  string\n\t\texpectedResult string\n\t\tsetupMock      func(m *MockLoadBalancer)\n\t}{\n\t\t{\n\t\t\tname:     \"Basic case\",\n\t\t\tdomainID: \"domainA\",\n\t\t\ttaskList: types.TaskList{Name: \"taskListA\"},\n\t\t\tsetupMock: func(m *MockLoadBalancer) {\n\t\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\t\tDomainUUID:    \"domainA\",\n\t\t\t\t\tTaskList:      &types.TaskList{Name: \"taskListA\"},\n\t\t\t\t\tForwardedFrom: \"\",\n\t\t\t\t}\n\t\t\t\tm.EXPECT().\n\t\t\t\t\tPickWritePartition(0, req).\n\t\t\t\t\tReturn(\"partitionA\")\n\t\t\t},\n\t\t\texpectedResult: \"partitionA\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockFallbackLB := NewMockLoadBalancer(ctrl)\n\t\t\tif tc.setupMock != nil {\n\t\t\t\ttc.setupMock(mockFallbackLB)\n\t\t\t}\n\n\t\t\tlb := &weightedLoadBalancer{\n\t\t\t\tfallbackLoadBalancer: mockFallbackLB,\n\t\t\t}\n\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tc.domainID,\n\t\t\t\tTaskList:      &tc.taskList,\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\n\t\t\tresult := lb.PickWritePartition(tc.taskListType, req)\n\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t})\n\t}\n}\n\nfunc TestWeightedLoadBalancer_PickReadPartition(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tdomainID           string\n\t\ttaskList           types.TaskList\n\t\ttaskListType       int\n\t\tforwardedFrom      string\n\t\tweightCacheReturn  interface{}\n\t\tfallbackReturn     string\n\t\texpectedResult     string\n\t\texpectFallbackCall bool\n\t}{\n\t\t{\n\t\t\tname:               \"WeightCache returns nil\",\n\t\t\tdomainID:           \"domainA\",\n\t\t\ttaskList:           types.TaskList{Name: \"taskListA\"},\n\t\t\tweightCacheReturn:  nil,\n\t\t\tfallbackReturn:     \"fallbackPartition\",\n\t\t\texpectedResult:     \"fallbackPartition\",\n\t\t\texpectFallbackCall: true,\n\t\t},\n\t\t{\n\t\t\tname:               \"WeightCache returns invalid type\",\n\t\t\tdomainID:           \"domainB\",\n\t\t\ttaskList:           types.TaskList{Name: \"taskListB\"},\n\t\t\tweightCacheReturn:  \"invalidType\",\n\t\t\tfallbackReturn:     \"fallbackPartition\",\n\t\t\texpectedResult:     \"fallbackPartition\",\n\t\t\texpectFallbackCall: true,\n\t\t},\n\t\t{\n\t\t\tname:               \"WeightSelector pick returns negative\",\n\t\t\tdomainID:           \"domainC\",\n\t\t\ttaskList:           types.TaskList{Name: \"taskListC\"},\n\t\t\tweightCacheReturn:  newWeightSelector(2, 100),\n\t\t\tfallbackReturn:     \"fallbackPartition\",\n\t\t\texpectedResult:     \"fallbackPartition\",\n\t\t\texpectFallbackCall: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"WeightSelector pick returns non-negative\",\n\t\t\tdomainID: \"domainD\",\n\t\t\ttaskList: types.TaskList{Name: \"taskListD\"},\n\t\t\tweightCacheReturn: func() *weightSelector {\n\t\t\t\tpw := newWeightSelector(2, 10)\n\t\t\t\tpw.update(2, 0, 0)\n\t\t\t\tpw.update(2, 1, 11)\n\t\t\t\treturn pw\n\t\t\t}(),\n\t\t\texpectedResult:     getPartitionTaskListName(\"taskListD\", 1),\n\t\t\texpectFallbackCall: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tc.domainID,\n\t\t\t\tTaskList:      &tc.taskList,\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tctrl := gomock.NewController(t)\n\t\t\t// Create mocks.\n\t\t\tmockWeightCache := cache.NewMockCache(ctrl)\n\t\t\tmockFallbackLoadBalancer := NewMockLoadBalancer(ctrl)\n\n\t\t\t// Set up the mocks.\n\t\t\ttaskListKey := key{\n\t\t\t\tdomainID:     tc.domainID,\n\t\t\t\ttaskListName: tc.taskList.GetName(),\n\t\t\t\ttaskListType: tc.taskListType,\n\t\t\t}\n\t\t\tmockWeightCache.EXPECT().\n\t\t\t\tGet(taskListKey).\n\t\t\t\tReturn(tc.weightCacheReturn)\n\n\t\t\tif tc.expectFallbackCall {\n\t\t\t\tmockFallbackLoadBalancer.EXPECT().\n\t\t\t\t\tPickReadPartition(tc.taskListType, req, \"\").\n\t\t\t\t\tReturn(tc.fallbackReturn)\n\t\t\t}\n\n\t\t\tlogger := testlogger.New(t)\n\n\t\t\t// Create the weightedLoadBalancer instance.\n\t\t\tlb := &weightedLoadBalancer{\n\t\t\t\tweightCache:          mockWeightCache,\n\t\t\t\tfallbackLoadBalancer: mockFallbackLoadBalancer,\n\t\t\t\tlogger:               logger,\n\t\t\t}\n\n\t\t\t// Call the method under test.\n\t\t\tresult := lb.PickReadPartition(tc.taskListType, req, \"\")\n\n\t\t\t// Assert the result.\n\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t})\n\t}\n}\n\nfunc TestWeightedLoadBalancer_UpdateWeight(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tdomainID          string\n\t\ttaskList          types.TaskList\n\t\ttaskListType      int\n\t\tforwardedFrom     string\n\t\tpartition         string\n\t\tloadBalancerHints *types.LoadBalancerHints\n\t\tsetupMock         func(*cache.MockCache, *MockPartitionConfigProvider)\n\t}{\n\t\t{\n\t\t\tname:     \"Sticky task list\",\n\t\t\tdomainID: \"domainA\",\n\t\t\ttaskList: types.TaskList{Name: \"a\", Kind: types.TaskListKindSticky.Ptr()},\n\t\t},\n\t\t{\n\t\t\tname:          \"forwarded request\",\n\t\t\tdomainID:      \"domainA\",\n\t\t\ttaskList:      types.TaskList{Name: \"a\"},\n\t\t\tforwardedFrom: \"tasklist\",\n\t\t},\n\t\t{\n\t\t\tname:     \"partitioned task list\",\n\t\t\tdomainID: \"domainA\",\n\t\t\ttaskList: types.TaskList{Name: \"/__cadence_sys/aaa/1\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"nil loadBalancerHints\",\n\t\t\tdomainID: \"domainA\",\n\t\t\ttaskList: types.TaskList{Name: \"a\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"1 partition\",\n\t\t\tdomainID:  \"domainA\",\n\t\t\ttaskList:  types.TaskList{Name: \"a\"},\n\t\t\tpartition: \"a\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount: 1,\n\t\t\t},\n\t\t\tsetupMock: func(mockCache *cache.MockCache, mockPartitionConfigProvider *MockPartitionConfigProvider) {\n\t\t\t\tmockPartitionConfigProvider.EXPECT().GetNumberOfReadPartitions(\"domainA\", types.TaskList{Name: \"a\"}, 0).Return(1)\n\t\t\t\tmockCache.EXPECT().Delete(key{\n\t\t\t\t\tdomainID:     \"domainA\",\n\t\t\t\t\ttaskListName: \"a\",\n\t\t\t\t\ttaskListType: 0,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"partition 0\",\n\t\t\tdomainID:  \"domainA\",\n\t\t\ttaskList:  types.TaskList{Name: \"a\"},\n\t\t\tpartition: \"a\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount: 1,\n\t\t\t},\n\t\t\tsetupMock: func(mockCache *cache.MockCache, mockPartitionConfigProvider *MockPartitionConfigProvider) {\n\t\t\t\tmockPartitionConfigProvider.EXPECT().GetNumberOfReadPartitions(\"domainA\", types.TaskList{Name: \"a\"}, 0).Return(2)\n\t\t\t\tmockCache.EXPECT().Get(key{\n\t\t\t\t\tdomainID:     \"domainA\",\n\t\t\t\t\ttaskListName: \"a\",\n\t\t\t\t\ttaskListType: 0,\n\t\t\t\t}).Return(nil)\n\t\t\t\tmockCache.EXPECT().PutIfNotExist(key{\n\t\t\t\t\tdomainID:     \"domainA\",\n\t\t\t\t\ttaskListName: \"a\",\n\t\t\t\t\ttaskListType: 0,\n\t\t\t\t}, newWeightSelector(2, 100)).Return(newWeightSelector(2, 100), nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"partition 1\",\n\t\t\tdomainID:  \"domainA\",\n\t\t\ttaskList:  types.TaskList{Name: \"a\"},\n\t\t\tpartition: \"/__cadence_sys/a/1\",\n\t\t\tloadBalancerHints: &types.LoadBalancerHints{\n\t\t\t\tBacklogCount: 1,\n\t\t\t},\n\t\t\tsetupMock: func(mockCache *cache.MockCache, mockPartitionConfigProvider *MockPartitionConfigProvider) {\n\t\t\t\tmockPartitionConfigProvider.EXPECT().GetNumberOfReadPartitions(\"domainA\", types.TaskList{Name: \"a\"}, 0).Return(2)\n\t\t\t\tmockCache.EXPECT().Get(key{\n\t\t\t\t\tdomainID:     \"domainA\",\n\t\t\t\t\ttaskListName: \"a\",\n\t\t\t\t\ttaskListType: 0,\n\t\t\t\t}).Return(newWeightSelector(2, 100))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    tc.domainID,\n\t\t\t\tTaskList:      &tc.taskList,\n\t\t\t\tForwardedFrom: tc.forwardedFrom,\n\t\t\t}\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockWeightCache := cache.NewMockCache(ctrl)\n\t\t\tmockPartitionConfigProvider := NewMockPartitionConfigProvider(ctrl)\n\t\t\tlb := &weightedLoadBalancer{\n\t\t\t\tweightCache: mockWeightCache,\n\t\t\t\tprovider:    mockPartitionConfigProvider,\n\t\t\t\tlogger:      testlogger.New(t),\n\t\t\t}\n\t\t\tif tc.setupMock != nil {\n\t\t\t\ttc.setupMock(mockWeightCache, mockPartitionConfigProvider)\n\t\t\t}\n\n\t\t\tlb.UpdateWeight(tc.taskListType, req, tc.partition, tc.loadBalancerHints)\n\t\t})\n\t}\n}\n\nfunc TestCalcWeightFromLoadBalancerHints(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tinfo     types.LoadBalancerHints\n\t\texpected int64\n\t}{\n\t\t{\n\t\t\tname:     \"Zero QPS and backlog count\",\n\t\t\tinfo:     types.LoadBalancerHints{BacklogCount: 0, RatePerSecond: 0},\n\t\t\texpected: 0,\n\t\t},\n\t\t{\n\t\t\tname:     \"Small QPS below threshold\",\n\t\t\tinfo:     types.LoadBalancerHints{BacklogCount: 10, RatePerSecond: 0.005},\n\t\t\texpected: 10,\n\t\t},\n\t\t{\n\t\t\tname:     \"QPS above threshold with no backlog\",\n\t\t\tinfo:     types.LoadBalancerHints{BacklogCount: 0, RatePerSecond: 2},\n\t\t\texpected: int64(math.Ceil(2 * 0.01)), // smoothingNumber calculation\n\t\t},\n\t\t{\n\t\t\tname:     \"QPS above threshold with backlog\",\n\t\t\tinfo:     types.LoadBalancerHints{BacklogCount: 100, RatePerSecond: 5},\n\t\t\texpected: 100 + int64(math.Ceil(5*0.01)), // backlog + smoothingNumber\n\t\t},\n\t\t{\n\t\t\tname:     \"Large QPS\",\n\t\t\tinfo:     types.LoadBalancerHints{BacklogCount: 50, RatePerSecond: 100},\n\t\t\texpected: 50 + int64(math.Ceil(100*0.01)), // backlog + smoothingNumber\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := calcWeightFromLoadBalancerHints(&tt.info)\n\t\t\tassert.Equal(t, tt.expected, result, \"unexpected result for %s\", tt.name)\n\t\t})\n\t}\n}\n\nfunc TestPickBetween(t *testing.T) {\n\ttl := types.TaskList{Name: \"a\"}\n\tcases := []struct {\n\t\tname    string\n\t\thints   map[int]*types.LoadBalancerHints\n\t\tconfig  *types.TaskListPartitionConfig\n\t\tfrom    []int\n\t\tallowed []int\n\t}{\n\t\t{\n\t\t\tname:    \"unknown TL\",\n\t\t\tconfig:  createDefaultConfig(1, 1),\n\t\t\tfrom:    []int{0, 1, 2, 3},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:    \"no partitions specified\",\n\t\t\tconfig:  createDefaultConfig(1, 1),\n\t\t\tfrom:    []int{},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:    \"single partition specified\",\n\t\t\tconfig:  createDefaultConfig(1, 1),\n\t\t\tfrom:    []int{0},\n\t\t\tallowed: []int{0},\n\t\t},\n\t\t{\n\t\t\tname:    \"single partition\",\n\t\t\tconfig:  createDefaultConfig(1, 1),\n\t\t\tfrom:    []int{0, 1, 2, 3},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:    \"not initialized\",\n\t\t\tconfig:  createDefaultConfig(4, 4),\n\t\t\tfrom:    []int{0, 1, 2, 3},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:   \"partially initialized\",\n\t\t\tconfig: createDefaultConfig(4, 4),\n\t\t\thints: map[int]*types.LoadBalancerHints{\n\t\t\t\t0: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t1: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t},\n\t\t\tfrom:    []int{0, 1, 2, 3},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:   \"fully initialized\",\n\t\t\tconfig: createDefaultConfig(4, 4),\n\t\t\thints: map[int]*types.LoadBalancerHints{\n\t\t\t\t0: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t1: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t3: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t},\n\t\t\tfrom:    []int{0, 1, 2, 3},\n\t\t\tallowed: []int{0, 1, 2, 3},\n\t\t},\n\t\t{\n\t\t\tname:   \"invalid partition\",\n\t\t\tconfig: createDefaultConfig(4, 4),\n\t\t\thints: map[int]*types.LoadBalancerHints{\n\t\t\t\t0: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t1: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t3: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t},\n\t\t\tfrom:    []int{0, 1, 2, 4},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:   \"only consider specified - below threshold\",\n\t\t\tconfig: createDefaultConfig(4, 4),\n\t\t\thints: map[int]*types.LoadBalancerHints{\n\t\t\t\t0: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t1: {\n\t\t\t\t\tBacklogCount:  1,\n\t\t\t\t\tRatePerSecond: 1,\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t3: {\n\t\t\t\t\tBacklogCount:  1,\n\t\t\t\t\tRatePerSecond: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tfrom:    []int{1, 3},\n\t\t\tallowed: []int{-1},\n\t\t},\n\t\t{\n\t\t\tname:   \"only consider specified - above threshold\",\n\t\t\tconfig: createDefaultConfig(4, 4),\n\t\t\thints: map[int]*types.LoadBalancerHints{\n\t\t\t\t0: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t1: {\n\t\t\t\t\tBacklogCount:  _backlogThreshold,\n\t\t\t\t\tRatePerSecond: 1,\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tBacklogCount:  1000000,\n\t\t\t\t\tRatePerSecond: 100,\n\t\t\t\t},\n\t\t\t\t3: {\n\t\t\t\t\tBacklogCount:  1,\n\t\t\t\t\tRatePerSecond: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tfrom:    []int{1, 3},\n\t\t\tallowed: []int{1, 3},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\treq := &types.AddDecisionTaskRequest{\n\t\t\t\tDomainUUID:    testDomain,\n\t\t\t\tTaskList:      &tl,\n\t\t\t\tForwardedFrom: \"\",\n\t\t\t}\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tweightCache := cache.New(&cache.Options{\n\t\t\t\tTTL:             0,\n\t\t\t\tInitialCapacity: 100,\n\t\t\t\tPin:             false,\n\t\t\t\tMaxCount:        3000,\n\t\t\t\tActivelyEvict:   false,\n\t\t\t\tMetricsScope:    metrics.NoopScope,\n\t\t\t\tLogger:          testlogger.New(t),\n\t\t\t})\n\t\t\tmockPartitionConfigProvider := NewMockPartitionConfigProvider(ctrl)\n\t\t\tmockPartitionConfigProvider.EXPECT().GetNumberOfReadPartitions(testDomain, tl, 0).Return(len(tc.config.ReadPartitions)).Times(len(tc.hints))\n\t\t\tlb := &weightedLoadBalancer{\n\t\t\t\tweightCache: weightCache,\n\t\t\t\tprovider:    mockPartitionConfigProvider,\n\t\t\t\tlogger:      testlogger.New(t),\n\t\t\t}\n\t\t\tfor partitionID, hint := range tc.hints {\n\t\t\t\tlb.UpdateWeight(0, req, getPartitionTaskListName(tl.Name, partitionID), hint)\n\t\t\t}\n\t\t\tactual := lb.PickBetween(testDomain, tl.Name, 0, tc.from)\n\t\t\tassert.Contains(t, tc.allowed, actual)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "client/sharddistributor/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sharddistributor\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/client/sharddistributor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/sharddistributor_generated.go -v client=ShardDistributor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/sharddistributor_generated.go -v client=ShardDistributor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/sharddistributor_generated.go -v client=ShardDistributor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/sharddistributor_generated.go -v client=ShardDistributor -v package=sharddistributorv1 -v path=github.com/uber/cadence/.gen/proto/sharddistributor/v1 -v prefix=ShardDistributor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/sharddistributor_generated.go -v client=ShardDistributor\n\ntype Client interface {\n\tGetShardOwner(context.Context, *types.GetShardOwnerRequest, ...yarpc.CallOption) (*types.GetShardOwnerResponse, error)\n\tWatchNamespaceState(context.Context, *types.WatchNamespaceStateRequest, ...yarpc.CallOption) (WatchNamespaceStateClient, error)\n}\n\ntype WatchNamespaceStateClient interface {\n\tContext() context.Context\n\tRecv(...yarpc.StreamOption) (*types.WatchNamespaceStateResponse, error)\n\tCloseSend(...yarpc.StreamOption) error\n}\n"
  },
  {
    "path": "client/sharddistributor/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package sharddistributor -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/client/sharddistributor\n//\n\n// Package sharddistributor is a generated GoMock package.\npackage sharddistributor\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// GetShardOwner mocks base method.\nfunc (m *MockClient) GetShardOwner(arg0 context.Context, arg1 *types.GetShardOwnerRequest, arg2 ...yarpc.CallOption) (*types.GetShardOwnerResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetShardOwner\", varargs...)\n\tret0, _ := ret[0].(*types.GetShardOwnerResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShardOwner indicates an expected call of GetShardOwner.\nfunc (mr *MockClientMockRecorder) GetShardOwner(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardOwner\", reflect.TypeOf((*MockClient)(nil).GetShardOwner), varargs...)\n}\n\n// WatchNamespaceState mocks base method.\nfunc (m *MockClient) WatchNamespaceState(arg0 context.Context, arg1 *types.WatchNamespaceStateRequest, arg2 ...yarpc.CallOption) (WatchNamespaceStateClient, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"WatchNamespaceState\", varargs...)\n\tret0, _ := ret[0].(WatchNamespaceStateClient)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WatchNamespaceState indicates an expected call of WatchNamespaceState.\nfunc (mr *MockClientMockRecorder) WatchNamespaceState(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WatchNamespaceState\", reflect.TypeOf((*MockClient)(nil).WatchNamespaceState), varargs...)\n}\n\n// MockWatchNamespaceStateClient is a mock of WatchNamespaceStateClient interface.\ntype MockWatchNamespaceStateClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWatchNamespaceStateClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockWatchNamespaceStateClientMockRecorder is the mock recorder for MockWatchNamespaceStateClient.\ntype MockWatchNamespaceStateClientMockRecorder struct {\n\tmock *MockWatchNamespaceStateClient\n}\n\n// NewMockWatchNamespaceStateClient creates a new mock instance.\nfunc NewMockWatchNamespaceStateClient(ctrl *gomock.Controller) *MockWatchNamespaceStateClient {\n\tmock := &MockWatchNamespaceStateClient{ctrl: ctrl}\n\tmock.recorder = &MockWatchNamespaceStateClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWatchNamespaceStateClient) EXPECT() *MockWatchNamespaceStateClientMockRecorder {\n\treturn m.recorder\n}\n\n// CloseSend mocks base method.\nfunc (m *MockWatchNamespaceStateClient) CloseSend(arg0 ...yarpc.StreamOption) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"CloseSend\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CloseSend indicates an expected call of CloseSend.\nfunc (mr *MockWatchNamespaceStateClientMockRecorder) CloseSend(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseSend\", reflect.TypeOf((*MockWatchNamespaceStateClient)(nil).CloseSend), arg0...)\n}\n\n// Context mocks base method.\nfunc (m *MockWatchNamespaceStateClient) Context() context.Context {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Context\")\n\tret0, _ := ret[0].(context.Context)\n\treturn ret0\n}\n\n// Context indicates an expected call of Context.\nfunc (mr *MockWatchNamespaceStateClientMockRecorder) Context() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Context\", reflect.TypeOf((*MockWatchNamespaceStateClient)(nil).Context))\n}\n\n// Recv mocks base method.\nfunc (m *MockWatchNamespaceStateClient) Recv(arg0 ...yarpc.StreamOption) (*types.WatchNamespaceStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Recv\", varargs...)\n\tret0, _ := ret[0].(*types.WatchNamespaceStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Recv indicates an expected call of Recv.\nfunc (mr *MockWatchNamespaceStateClientMockRecorder) Recv(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Recv\", reflect.TypeOf((*MockWatchNamespaceStateClient)(nil).Recv), arg0...)\n}\n"
  },
  {
    "path": "client/sharddistributorexecutor/executorinterface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sharddistributorexecutor\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination executorinterface_mock.go -self_package github.com/uber/cadence/client/sharddistributorexecutor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/retry.tmpl -o ../wrappers/retryable/sharddistributorexecutor_generated.go -v client=ShardDistributorExecutor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/metered.tmpl -o ../wrappers/metered/sharddistributorexecutor_generated.go -v client=ShardDistributorExecutor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/errorinjectors.tmpl -o ../wrappers/errorinjectors/sharddistributorexecutor_generated.go -v client=ShardDistributorExecutor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/grpc.tmpl -o ../wrappers/grpc/sharddistributorexecutor_generated.go -v client=ShardDistributorExecutor -v package=apiv1 -v path=github.com/uber/cadence/proto/internal/uber/cadence/sharddistributor/v1 -v prefix=ShardDistributorExecutor\n//go:generate gowrap gen -g -p . -i Client -t ../templates/timeout.tmpl -o ../wrappers/timeout/sharddistributorexecutor_generated.go -v client=ShardDistributorExecutor\n\ntype Client interface {\n\tHeartbeat(context.Context, *types.ExecutorHeartbeatRequest, ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error)\n}\n"
  },
  {
    "path": "client/sharddistributorexecutor/executorinterface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: executorinterface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package sharddistributorexecutor -source executorinterface.go -destination executorinterface_mock.go -self_package github.com/uber/cadence/client/sharddistributorexecutor\n//\n\n// Package sharddistributorexecutor is a generated GoMock package.\npackage sharddistributorexecutor\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// Heartbeat mocks base method.\nfunc (m *MockClient) Heartbeat(arg0 context.Context, arg1 *types.ExecutorHeartbeatRequest, arg2 ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Heartbeat\", varargs...)\n\tret0, _ := ret[0].(*types.ExecutorHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Heartbeat indicates an expected call of Heartbeat.\nfunc (mr *MockClientMockRecorder) Heartbeat(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Heartbeat\", reflect.TypeOf((*MockClient)(nil).Heartbeat), varargs...)\n}\n"
  },
  {
    "path": "client/templates/errorinjectors.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n{{$clientName := (index .Vars \"client\")}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n{{ $Decorator := (printf \"%s%s\" $clientName .Interface.Name) }}\n\n\n{{$fakeErr := printf \"msg%sInjectedFakeErr\" $clientName}}\nconst (\n\t{{$fakeErr}} = \"Injected fake {{down $clientName}} client error\"\n)\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with retries\ntype {{$decorator}} struct {\n    client        {{.Interface.Type}}\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// New{{$Decorator}} creates a new instance of {{$decorator}} that injects error into every call with a given rate.\nfunc New{{$Decorator}}(client {{.Interface.Type}}, errorRate float64, logger log.Logger) {{.Interface.Type}} {\n    return &{{$decorator}}{\n        client:        client,\n        errorRate:     errorRate,\n        logger:        logger,\n        fakeErrFn:     errors.GenerateFakeError,\n        forwardCallFn: errors.ShouldForwardCall,\n    }\n}\n\n{{range $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n\t        fakeErr := c.fakeErrFn(c.errorRate)\n\t        var forwardCall bool\n\t        if forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t            {{$method.ResultsNames}} = c.client.{{$method.Call}}\n\t        }\n\n\t        if fakeErr != nil {\n                c.logger.Error({{$fakeErr}},\n\t\t\t        {{printf \"tag.%sClientOperation%s\" $clientName .Name}},\n\t\t\t        tag.Error(fakeErr),\n\t\t\t        tag.Bool(forwardCall),\n\t\t\t        tag.ClientError(err),\n\t\t        )\n\t            err = fakeErr\n\t            return\n            }\n            return\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "client/templates/grpc.tmpl",
    "content": "{{$packagePath := (index .Vars \"path\")}}\n{{$package := (index .Vars \"package\")}}\n{{$prefix := (index .Vars \"prefix\")}}\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t{{$package}} \"{{$packagePath}}\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\n{{$interfaceName := .Interface.Name}}\n{{$clientName := (index .Vars \"client\")}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n\n{{range $method := .Interface.Methods}}\n{{$Request := printf \"%sRequest\" $method.Name}}\n{{$Response := printf \"%sResponse\" $method.Name}}\n{{- $isStreaming := false}}\n{{- range $method.Results}}\n\t{{- if contains \"Client\" .Type}}\n\t\t{{- $isStreaming = true}}\n\t{{- end}}\n{{- end}}\n{{- if $isStreaming}}\nfunc (g {{$decorator}}) {{$method.Declaration}} {\n\tstream, {{(index $method.Results 1).Name}} := g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, proto.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\tif {{(index $method.Results 1).Name}} != nil {\n\t\treturn nil, proto.ToError({{(index $method.Results 1).Name}})\n\t}\n\treturn &{{lower $method.Name}}Client{stream: stream}, nil\n}\n\ntype {{lower $method.Name}}Client struct {\n\tstream {{$package}}.{{$prefix}}APIService{{$method.Name}}YARPCClient\n}\n\nfunc (c *{{lower $method.Name}}Client) Context() context.Context {\n\treturn c.stream.Context()\n}\n\nfunc (c *{{lower $method.Name}}Client) Recv(options ...yarpc.StreamOption) (*types.{{$Response}}, error) {\n\tresponse, err := c.stream.Recv(options...)\n\tif err != nil {\n\t\treturn nil, proto.ToError(err)\n\t}\n\treturn proto.To{{$prefix}}{{$Response}}(response), nil\n}\n\nfunc (c *{{lower $method.Name}}Client) CloseSend(options ...yarpc.StreamOption) error {\n\treturn proto.ToError(c.stream.CloseSend(options...))\n}\n{{- else}}\nfunc (g {{$decorator}}) {{$method.Declaration}} {\n\t{{- if eq (len $method.Params) 2}}\n\t{{- if eq (len $method.Results) 1}}\n\t_, {{(index $method.Results 0).Name}} = g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, &{{$package}}.{{$method.Name}}Request{}, {{(index $method.Params 1).Pass}})\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, &{{$package}}.{{$method.Name}}Request{}, {{(index $method.Params 1).Pass}})\n\t{{- end}}\n\t{{- else}}\n\t{{- if eq (len $method.Results) 1}}\n\t_, {{(index $method.Results 0).Name}} = g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, proto.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, proto.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\t{{- end}}\n\t{{- end}}\n\n\t{{- if eq (len $method.Results) 1}}\n\treturn proto.ToError({{(index $method.Results 0).Name}})\n\t{{- else}}\n\treturn proto.To{{$prefix}}{{$Response}}(response), proto.ToError({{(index $method.Results 1).Name}})\n\t{{- end}}\n}\n{{- end}}\n{{end}}\n"
  },
  {
    "path": "client/templates/metered.tmpl",
    "content": "import (\n\t\"context\"\n\t\"strings\"\n\n\t\"go.uber.org/yarpc\"\n    \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n{{$clientName := (index .Vars \"client\")}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n{{ $Decorator := (printf \"%s%s\" $clientName .Interface.Name) }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with retries\ntype {{$decorator}} struct {\n    client        {{.Interface.Type}}\n    metricsClient metrics.Client\n}\n\n// New{{$Decorator}} creates a new instance of {{$decorator}} with retry policy\nfunc New{{$Decorator}}(client {{.Interface.Type}}, metricsClient metrics.Client) {{.Interface.Type}} {\n    return &{{$decorator}}{\n        client:        client,\n        metricsClient: metricsClient,\n    }\n}\n\n{{range $method := .Interface.Methods}}\nfunc (c *{{$decorator}}) {{$method.Declaration}} {\n    {{- $scopeName:=printf \"metrics.%sClient%sScope\" $clientName $method.Name }}\n\tretryCount := getRetryCountFromContext({{(index $method.Params 0).Name}})\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope({{$scopeName}})\n\t} else {\n\t\tscope = c.metricsClient.Scope({{$scopeName}}, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n    {{- if eq $clientName \"Matching\" }}\n    c.emitForwardedFromStats(scope, {{(index $method.Params 1).Name}})\n    {{ end }}\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\t{{$method.ResultsNames}} = c.client.{{$method.Call}}\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn {{$method.ResultsNames}}\n}\n{{end}}\n\n{{- if eq $clientName \"Matching\" }}\ntype forwardedRequest interface {\n    GetForwardedFrom() string\n    GetTaskList() *types.TaskList\n}\n\nfunc (c *{{$decorator}}) emitForwardedFromStats(scope metrics.Scope, req any) {\n    p, ok := req.(forwardedRequest)\n    if !ok || p.GetTaskList() == nil {\n        return\n    }\n\n    taskList := p.GetTaskList()\n    forwardedFrom := p.GetForwardedFrom()\n\n    isChildPartition := strings.HasPrefix(taskList.GetName(), constants.ReservedTaskListPrefix)\n    if forwardedFrom != \"\"{\n        scope.IncCounter(metrics.MatchingClientForwardedCounter)\n        return\n    }\n\n    if isChildPartition {\n        scope.IncCounter(metrics.MatchingClientInvalidTaskListName)\n    }\n    return\n}\n{{ end -}}\n"
  },
  {
    "path": "client/templates/retry.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n{{$clientName := (index .Vars \"client\")}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n{{ $Decorator := (printf \"%s%s\" $clientName .Interface.Name) }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with retries\ntype {{$decorator}} struct {\n    client        {{.Interface.Type}}\n    throttleRetry *backoff.ThrottleRetry\n}\n\n// New{{$Decorator}} creates a new instance of {{$decorator}} with retry policy\nfunc New{{$Decorator}}(client {{.Interface.Type}}, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) {{.Interface.Type}} {\n    return &{{$decorator}}{\n        client: client,\n        throttleRetry: backoff.NewThrottleRetry(\n            backoff.WithRetryPolicy(policy),\n            backoff.WithRetryableError(isRetryable),\n        ),\n    }\n}\n\n{{range $method := .Interface.Methods}}\n    {{$resultsLength := len ($method.Results)}}\n    {{if eq $resultsLength 1}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n            op := func(ctx context.Context) error {\n                return c.client.{{$method.Call}}\n            }\n            return c.throttleRetry.Do(ctx, op)\n        }\n    {{else}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n            var resp {{ (index $method.Results 0).Type }}\n            op := func(ctx context.Context) error {\n                var err error\n                resp, err = c.client.{{$method.Call}}\n                return err\n            }\n            err = c.throttleRetry.Do(ctx, op)\n            return resp, err\n        }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "client/templates/thrift.tmpl",
    "content": "{{$prefix := (index .Vars \"prefix\")}}\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\n{{$unsupportedMethods := list \"CountDLQMessages\" \"UpdateTaskListPartitionConfig\" \"RefreshTaskListPartitionConfig\"}}\n\n{{$interfaceName := .Interface.Name}}\n{{$clientName := (index .Vars \"client\")}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n\n{{range $method := .Interface.Methods}}\n{{$Request := printf \"%sRequest\" $method.Name}}\n{{$Response := printf \"%sResponse\" $method.Name}}\nfunc (g {{$decorator}}) {{$method.Declaration}} {\n\t{{- if has $method.Name $unsupportedMethods}}\n\t\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n\t{{- else if or (eq $method.Name \"AddDecisionTask\") (eq $method.Name \"AddActivityTask\")}}\n\t\t{{(index $method.Results 1).Name}} = g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\t\tif {{(index $method.Results 1).Name}} != nil {\n\t\t\treturn nil, {{(index $method.Results 1).Name}}\n\t\t}\n\t\treturn &types.{{$method.Name}}Response{}, nil\n\t{{- else}}\n\t{{- if eq (len $method.Params) 2}}\n\t{{- if eq (len $method.Results) 1}}\n\t{{(index $method.Results 0).Name}} = g.c.{{$method.Call}}\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.c.{{$method.Call}}\n\t{{- end}}\n\t{{- else}}\n\t{{- if eq (len $method.Results) 1}}\n\t{{(index $method.Results 0).Name}} = g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.c.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.From{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}), {{(index $method.Params 2).Pass}})\n\t{{- end}}\n\t{{- end}}\n\n\t{{- if eq (len $method.Results) 1}}\n\treturn thrift.ToError({{(index $method.Results 0).Name}})\n\t{{- else}}\n\treturn thrift.To{{$prefix}}{{$Response}}(response), thrift.ToError({{(index $method.Results 1).Name}})\n\t{{- end}}\n\t{{- end}}\n}\n{{end}}\n"
  },
  {
    "path": "client/templates/timeout.tmpl",
    "content": "{{$ClientName := (index .Vars \"client\")}}\n{{$clientName := (down $ClientName)}}\n{{$interfaceName := .Interface.Name}}\n{{ $decorator := (printf \"%s%s\" (down $clientName) .Interface.Name) }}\n{{ $Decorator := (printf \"%s%s\" $ClientName .Interface.Name) }}\n{{$largeTimeoutAPIs := list \"adminClient.GetCrossClusterTasks\" \"adminClient.GetReplicationMessages\"}}\n{{$longPollTimeoutAPIs := list \"frontendClient.ListArchivedWorkflowExecutions\" \"frontendClient.PollForActivityTask\" \"frontendClient.PollForDecisionTask\" \"matchingClient.PollForActivityTask\" \"matchingClient.PollForDecisionTask\"}}\n{{$noTimeoutAPIs := list \"historyClient.GetReplicationMessages\" \"historyClient.GetDLQReplicationMessages\" \"historyClient.CountDLQMessages\" \"historyClient.ReadDLQMessages\" \"historyClient.PurgeDLQMessages\" \"historyClient.MergeDLQMessages\" \"historyClient.GetCrossClusterTasks\" \"historyClient.GetFailoverInfo\" \"matchingClient.GetTaskListsByDomain\" \"sharddistributorClient.WatchNamespaceState\"}}\n{{/*\n $fieldMap defines a map of the decorator struct fields\n with field name as the key and field type as the value\n */}}\n{{$fieldMap := dict }}\n{{ if has $ClientName (list \"History\" \"ShardDistributor\" \"ShardDistributorExecutor\") }}\n    {{$fieldMap = merge $fieldMap (dict  \"timeout\" \"time.Duration\" \"client\" .Interface.Type) }}\n{{ else if eq $ClientName \"Admin\" }}\n    {{$fieldMap = merge $fieldMap (dict  \"timeout\" \"time.Duration\" \"client\" .Interface.Type \"largeTimeout\" \"time.Duration\") }}\n{{ else }}\n    {{$fieldMap = merge $fieldMap (dict \"timeout\" \"time.Duration\" \"client\" .Interface.Type \"longPollTimeout\" \"time.Duration\") }}\n{{ end }}\n\nimport (\n    \"context\"\n    \"time\"\n\n    \"go.uber.org/yarpc\"\n\n    \"github.com/uber/cadence/client/{{$clientName}}\"\n    \"github.com/uber/cadence/common/future\"\n    \"github.com/uber/cadence/common/persistence\"\n    \"github.com/uber/cadence/common/types\"\n)\n\nvar _ {{.Interface.Type}} = (*{{$decorator}})(nil)\n\n// {{$decorator}} implements the {{.Interface.Type}} interface instrumented with timeouts\ntype {{$decorator}} struct {\n    {{ range $fieldName, $fieldType := $fieldMap -}}\n           {{$fieldName}} {{$fieldType}}\n    {{ end }}\n}\n\n// New{{$Decorator}} creates a new {{$decorator}} instance\nfunc New{{$Decorator}}(\n   {{- range $fieldName, $fieldType := $fieldMap}}\n               {{$fieldName}} {{$fieldType}},\n    {{- end }}\n) {{.Interface.Type}} {\n   return &{{$decorator}}{\n      {{- range $fieldName, $fieldType := $fieldMap}}\n            {{$fieldName}}: {{$fieldName}},\n        {{- end }}\n   }\n}\n\n{{range $method := .Interface.Methods}}\nfunc (c * {{$decorator}}) {{$method.Declaration}} {\n    {{- if has (printf \"%s.%s\" $decorator $method.Name) $largeTimeoutAPIs -}}\n            ctx, cancel := createContext(ctx, c.largeTimeout)\n            defer cancel()\n    {{- else if has (printf \"%s.%s\" $decorator $method.Name) $longPollTimeoutAPIs -}}\n            ctx, cancel := createContext(ctx, c.longPollTimeout)\n            defer cancel()\n    {{- else if not (has (printf \"%s.%s\" $decorator $method.Name) $noTimeoutAPIs) -}}\n            ctx, cancel := createContext(ctx, c.timeout)\n            defer cancel()\n    {{- end }}\n    {{$method.Pass (\"c.client.\") }}\n}\n{{end}}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/admin_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgAdminInjectedFakeErr = \"Injected fake admin client error\"\n)\n\n// adminClient implements admin.Client interface instrumented with retries\ntype adminClient struct {\n\tclient        admin.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewAdminClient creates a new instance of adminClient that injects error into every call with a given rate.\nfunc NewAdminClient(client admin.Client, errorRate float64, logger log.Logger) admin.Client {\n\treturn &adminClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.AddSearchAttribute(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationAddSearchAttribute,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.CloseShard(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationCloseShard,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tcp2, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationCountDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tap2, err = c.client.DeleteWorkflow(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDeleteWorkflow,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp1, err = c.client.DescribeCluster(ctx, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDescribeCluster,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDescribeHistoryHost,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDescribeQueue,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeShardDistribution(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDescribeShardDistribution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tap2, err = c.client.DescribeWorkflowExecution(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationDescribeWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetDLQReplicationMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp1, err = c.client.GetDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetDomainAsyncWorkflowConfiguraton,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp1, err = c.client.GetDomainIsolationGroups(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetDomainIsolationGroups,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetDomainReplicationMessages(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetDomainReplicationMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetDynamicConfig(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetDynamicConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp1, err = c.client.GetGlobalIsolationGroups(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetGlobalIsolationGroups,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetReplicationMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetWorkflowExecutionRawHistoryV2(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationGetWorkflowExecutionRawHistoryV2,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListDynamicConfig(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationListDynamicConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tap2, err = c.client.MaintainCorruptWorkflow(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationMaintainCorruptWorkflow,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationMergeDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationPurgeDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationReadDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ReapplyEvents(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationReapplyEvents,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationRefreshWorkflowTasks,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RemoveTask(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationRemoveTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ResendReplicationTasks(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationResendReplicationTasks,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ResetQueue(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationResetQueue,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RestoreDynamicConfig(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationRestoreDynamicConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tup1, err = c.client.UpdateDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationUpdateDomainAsyncWorkflowConfiguraton,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tup1, err = c.client.UpdateDomainIsolationGroups(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationUpdateDomainIsolationGroups,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.UpdateDynamicConfig(ctx, up1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationUpdateDynamicConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tup1, err = c.client.UpdateGlobalIsolationGroups(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationUpdateGlobalIsolationGroups,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tup1, err = c.client.UpdateTaskListPartitionConfig(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgAdminInjectedFakeErr,\n\t\t\ttag.AdminClientOperationUpdateTaskListPartitionConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/errorinjectors_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errorinjectors\n\nimport (\n\t\"context\"\n\tstdErrors \"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestInjector(t *testing.T) {\n\t// All wrappers follow the same logic, so in this test we only test one of them.\n\tt.Run(\"no error is forwarded\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := admin.NewMockClient(ctrl)\n\n\t\tclientMock.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\n\t\tinjector := NewAdminClient(clientMock, 1, testlogger.New(t)).(*adminClient)\n\n\t\tinjector.fakeErrFn = func(float64) error {\n\t\t\treturn nil\n\t\t}\n\t\tinjector.forwardCallFn = func(err error) bool {\n\t\t\treturn err == nil\n\t\t}\n\t\t_, err := injector.DescribeWorkflowExecution(context.Background(), &types.AdminDescribeWorkflowExecutionRequest{})\n\t\tassert.NoError(t, err)\n\t})\n\tt.Run(\"no fake error, but client returns an error\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := admin.NewMockClient(ctrl)\n\n\t\tclientErr := fmt.Errorf(\"fail\")\n\n\t\tclientMock.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, clientErr).Times(1)\n\n\t\tinjector := NewAdminClient(clientMock, 1, testlogger.New(t)).(*adminClient)\n\n\t\tinjector.fakeErrFn = func(float64) error {\n\t\t\treturn nil\n\t\t}\n\t\tinjector.forwardCallFn = func(err error) bool {\n\t\t\treturn err == nil\n\t\t}\n\t\t_, err := injector.DescribeWorkflowExecution(context.Background(), &types.AdminDescribeWorkflowExecutionRequest{})\n\t\tassert.Error(t, err)\n\t\tassert.True(t, stdErrors.Is(clientErr, err))\n\t})\n\tt.Run(\"fake error overrides client error\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := admin.NewMockClient(ctrl)\n\n\t\tclientErr := fmt.Errorf(\"fail\")\n\n\t\tclientMock.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, clientErr).Times(1)\n\n\t\tinjector := NewAdminClient(clientMock, 1, testlogger.New(t)).(*adminClient)\n\n\t\tinjector.fakeErrFn = func(float64) error {\n\t\t\treturn errors.ErrFakeServiceBusy\n\t\t}\n\t\tinjector.forwardCallFn = func(err error) bool {\n\t\t\treturn true\n\t\t}\n\t\t_, err := injector.DescribeWorkflowExecution(context.Background(), &types.AdminDescribeWorkflowExecutionRequest{})\n\t\tassert.Error(t, err)\n\t\tassert.True(t, stdErrors.Is(errors.ErrFakeServiceBusy, err))\n\t})\n\tt.Run(\"failed but not forwarded\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := admin.NewMockClient(ctrl)\n\n\t\tclientMock.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\n\t\tinjector := NewAdminClient(clientMock, 1, testlogger.New(t)).(*adminClient)\n\n\t\tinjector.fakeErrFn = func(float64) error {\n\t\t\treturn errors.ErrFakeServiceBusy\n\t\t}\n\t\tinjector.forwardCallFn = func(err error) bool {\n\t\t\treturn true\n\t\t}\n\t\t_, err := injector.DescribeWorkflowExecution(context.Background(), &types.AdminDescribeWorkflowExecutionRequest{})\n\t\tassert.Error(t, err)\n\t})\n\tt.Run(\"failed and not forwarded\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := admin.NewMockClient(ctrl)\n\n\t\tinjector := NewAdminClient(clientMock, 1, testlogger.New(t)).(*adminClient)\n\n\t\tinjector.fakeErrFn = func(float64) error {\n\t\t\treturn errors.ErrFakeServiceBusy\n\t\t}\n\t\tinjector.forwardCallFn = func(err error) bool {\n\t\t\treturn false\n\t\t}\n\t\t_, err := injector.DescribeWorkflowExecution(context.Background(), &types.AdminDescribeWorkflowExecutionRequest{})\n\t\tassert.Error(t, err)\n\t})\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/frontend_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgFrontendInjectedFakeErr = \"Injected fake frontend client error\"\n)\n\n// frontendClient implements frontend.Client interface instrumented with retries\ntype frontendClient struct {\n\tclient        frontend.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewFrontendClient creates a new instance of frontendClient that injects error into every call with a given rate.\nfunc NewFrontendClient(client frontend.Client, errorRate float64, logger log.Logger) frontend.Client {\n\treturn &frontendClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tcp2, err = c.client.CountWorkflowExecutions(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationCountWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.DeleteDomain(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDeleteDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.DeprecateDomain(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDeprecateDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeDomain(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDescribeDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeTaskList(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDescribeTaskList,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeWorkflowExecution(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDescribeWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DiagnoseWorkflowExecution(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationDiagnoseWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tfp2, err = c.client.FailoverDomain(ctx, fp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationFailoverDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tcp1, err = c.client.GetClusterInfo(ctx, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationGetClusterInfo,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp1, err = c.client.GetSearchAttributes(ctx, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationGetSearchAttributes,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationGetTaskListsByDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetWorkflowExecutionHistory(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationGetWorkflowExecutionHistory,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListArchivedWorkflowExecutions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListArchivedWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListClosedWorkflowExecutions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListClosedWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListDomains(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListDomains,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListFailoverHistory(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListFailoverHistory,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListOpenWorkflowExecutions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListOpenWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListTaskListPartitions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListTaskListPartitions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ListWorkflowExecutions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationListWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tpp2, err = c.client.PollForActivityTask(ctx, pp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationPollForActivityTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tpp2, err = c.client.PollForDecisionTask(ctx, pp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationPollForDecisionTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tqp2, err = c.client.QueryWorkflow(ctx, qp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationQueryWorkflow,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RecordActivityTaskHeartbeat(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRecordActivityTaskHeartbeat,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RecordActivityTaskHeartbeatByID(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRecordActivityTaskHeartbeatByID,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRefreshWorkflowTasks,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RegisterDomain(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRegisterDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RequestCancelWorkflowExecution(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRequestCancelWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.ResetStickyTaskList(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationResetStickyTaskList,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.ResetWorkflowExecution(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationResetWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCanceled(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskCanceled,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCanceledByID(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskCanceledByID,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCompleted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCompletedByID(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskCompletedByID,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskFailed(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskFailed,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskFailedByID(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondActivityTaskFailedByID,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RespondDecisionTaskCompleted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondDecisionTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondDecisionTaskFailed(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondDecisionTaskFailed,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondQueryTaskCompleted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRespondQueryTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RestartWorkflowExecution(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationRestartWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp2, err = c.client.ScanWorkflowExecutions(ctx, lp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationScanWorkflowExecutions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp2, err = c.client.SignalWithStartWorkflowExecution(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationSignalWithStartWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp2, err = c.client.SignalWithStartWorkflowExecutionAsync(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationSignalWithStartWorkflowExecutionAsync,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.SignalWorkflowExecution(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationSignalWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp2, err = c.client.StartWorkflowExecution(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationStartWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp2, err = c.client.StartWorkflowExecutionAsync(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationStartWorkflowExecutionAsync,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.TerminateWorkflowExecution(ctx, tp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationTerminateWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tup2, err = c.client.UpdateDomain(ctx, up1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgFrontendInjectedFakeErr,\n\t\t\ttag.FrontendClientOperationUpdateDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/history_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgHistoryInjectedFakeErr = \"Injected fake history client error\"\n)\n\n// historyClient implements history.Client interface instrumented with retries\ntype historyClient struct {\n\tclient        history.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewHistoryClient creates a new instance of historyClient that injects error into every call with a given rate.\nfunc NewHistoryClient(client history.Client, errorRate float64, logger log.Logger) history.Client {\n\treturn &historyClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.CloseShard(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationCloseShard,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\thp1, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationCountDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationDescribeHistoryHost,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeMutableState(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationDescribeMutableState,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp2, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationDescribeQueue,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp1, err = c.client.DescribeWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationDescribeWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetCrossClusterTasks(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationGetCrossClusterTasks,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationGetDLQReplicationMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetFailoverInfo(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationGetFailoverInfo,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetMutableState(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationGetMutableState,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationGetReplicationMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationMergeDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.NotifyFailoverMarkers(ctx, np1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationNotifyFailoverMarkers,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tpp2, err = c.client.PollMutableState(ctx, pp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationPollMutableState,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationPurgeDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\thp2, err = c.client.QueryWorkflow(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationQueryWorkflow,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp1, err = c.client.RatelimitUpdate(ctx, request, opts...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRatelimitUpdate,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationReadDLQMessages,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ReapplyEvents(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationReapplyEvents,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp1, err = c.client.RecordActivityTaskHeartbeat(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRecordActivityTaskHeartbeat,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RecordActivityTaskStarted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRecordActivityTaskStarted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RecordChildExecutionCompleted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRecordChildExecutionCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RecordDecisionTaskStarted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRecordDecisionTaskStarted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RefreshWorkflowTasks(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRefreshWorkflowTasks,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RemoveSignalMutableState(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRemoveSignalMutableState,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RemoveTask(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRemoveTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ReplicateEventsV2(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationReplicateEventsV2,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RequestCancelWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRequestCancelWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ResetQueue(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationResetQueue,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\thp2, err = c.client.ResetStickyTaskList(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationResetStickyTaskList,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp1, err = c.client.ResetWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationResetWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCanceled(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondActivityTaskCanceled,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskCompleted(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondActivityTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondActivityTaskFailed(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondActivityTaskFailed,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\trp2, err = c.client.RespondCrossClusterTasksCompleted(ctx, rp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondCrossClusterTasksCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\thp2, err = c.client.RespondDecisionTaskCompleted(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondDecisionTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondDecisionTaskFailed(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationRespondDecisionTaskFailed,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.ScheduleDecisionTask(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationScheduleDecisionTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp1, err = c.client.SignalWithStartWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationSignalWithStartWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.SignalWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationSignalWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tsp1, err = c.client.StartWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationStartWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.SyncActivity(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationSyncActivity,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.SyncShardStatus(ctx, sp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationSyncShardStatus,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.TerminateWorkflowExecution(ctx, hp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgHistoryInjectedFakeErr,\n\t\t\ttag.HistoryClientOperationTerminateWorkflowExecution,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/matching_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgMatchingInjectedFakeErr = \"Injected fake matching client error\"\n)\n\n// matchingClient implements matching.Client interface instrumented with retries\ntype matchingClient struct {\n\tclient        matching.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewMatchingClient creates a new instance of matchingClient that injects error into every call with a given rate.\nfunc NewMatchingClient(client matching.Client, errorRate float64, logger log.Logger) matching.Client {\n\treturn &matchingClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tap2, err = c.client.AddActivityTask(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationAddActivityTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tap2, err = c.client.AddDecisionTask(ctx, ap1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationAddDecisionTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.CancelOutstandingPoll(ctx, cp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationCancelOutstandingPoll,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tdp1, err = c.client.DescribeTaskList(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationDescribeTaskList,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationGetTaskListsByDomain,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tlp1, err = c.client.ListTaskListPartitions(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationListTaskListPartitions,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.PollForActivityTask(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationPollForActivityTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.PollForDecisionTask(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationPollForDecisionTask,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.QueryWorkflow(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationQueryWorkflow,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.RefreshTaskListPartitionConfig(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationRefreshTaskListPartitionConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\terr = c.client.RespondQueryTaskCompleted(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationRespondQueryTaskCompleted,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tmp2, err = c.client.UpdateTaskListPartitionConfig(ctx, mp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgMatchingInjectedFakeErr,\n\t\t\ttag.MatchingClientOperationUpdateTaskListPartitionConfig,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/sharddistributor_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgShardDistributorInjectedFakeErr = \"Injected fake sharddistributor client error\"\n)\n\n// sharddistributorClient implements sharddistributor.Client interface instrumented with retries\ntype sharddistributorClient struct {\n\tclient        sharddistributor.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewShardDistributorClient creates a new instance of sharddistributorClient that injects error into every call with a given rate.\nfunc NewShardDistributorClient(client sharddistributor.Client, errorRate float64, logger log.Logger) sharddistributor.Client {\n\treturn &sharddistributorClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *sharddistributorClient) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest, p1 ...yarpc.CallOption) (gp2 *types.GetShardOwnerResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tgp2, err = c.client.GetShardOwner(ctx, gp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgShardDistributorInjectedFakeErr,\n\t\t\ttag.ShardDistributorClientOperationGetShardOwner,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *sharddistributorClient) WatchNamespaceState(ctx context.Context, wp1 *types.WatchNamespaceStateRequest, p1 ...yarpc.CallOption) (w1 sharddistributor.WatchNamespaceStateClient, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tw1, err = c.client.WatchNamespaceState(ctx, wp1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgShardDistributorInjectedFakeErr,\n\t\t\ttag.ShardDistributorClientOperationWatchNamespaceState,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/errorinjectors/sharddistributorexecutor_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/errorinjectors.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tmsgShardDistributorExecutorInjectedFakeErr = \"Injected fake sharddistributorexecutor client error\"\n)\n\n// sharddistributorexecutorClient implements sharddistributorexecutor.Client interface instrumented with retries\ntype sharddistributorexecutorClient struct {\n\tclient        sharddistributorexecutor.Client\n\terrorRate     float64\n\tlogger        log.Logger\n\tfakeErrFn     func(float64) error\n\tforwardCallFn func(error) bool\n}\n\n// NewShardDistributorExecutorClient creates a new instance of sharddistributorexecutorClient that injects error into every call with a given rate.\nfunc NewShardDistributorExecutorClient(client sharddistributorexecutor.Client, errorRate float64, logger log.Logger) sharddistributorexecutor.Client {\n\treturn &sharddistributorexecutorClient{\n\t\tclient:        client,\n\t\terrorRate:     errorRate,\n\t\tlogger:        logger,\n\t\tfakeErrFn:     errors.GenerateFakeError,\n\t\tforwardCallFn: errors.ShouldForwardCall,\n\t}\n}\n\nfunc (c *sharddistributorexecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tfakeErr := c.fakeErrFn(c.errorRate)\n\tvar forwardCall bool\n\tif forwardCall = c.forwardCallFn(fakeErr); forwardCall {\n\t\tep2, err = c.client.Heartbeat(ctx, ep1, p1...)\n\t}\n\n\tif fakeErr != nil {\n\t\tc.logger.Error(msgShardDistributorExecutorInjectedFakeErr,\n\t\t\ttag.ShardDistributorExecutorClientOperationHeartbeat,\n\t\t\ttag.Error(fakeErr),\n\t\t\ttag.Bool(forwardCall),\n\t\t\ttag.ClientError(err),\n\t\t)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/grpc/admin_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.AddSearchAttribute(ctx, proto.FromAdminAddSearchAttributeRequest(ap1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.CloseShard(ctx, proto.FromAdminCloseShardRequest(cp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tresponse, err := g.c.CountDLQMessages(ctx, proto.FromAdminCountDLQMessagesRequest(cp1), p1...)\n\treturn proto.ToAdminCountDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tresponse, err := g.c.DeleteWorkflow(ctx, proto.FromAdminDeleteWorkflowRequest(ap1), p1...)\n\treturn proto.ToAdminDeleteWorkflowResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tresponse, err := g.c.DescribeCluster(ctx, &adminv1.DescribeClusterRequest{}, p1...)\n\treturn proto.ToAdminDescribeClusterResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.c.DescribeHistoryHost(ctx, proto.FromAdminDescribeHistoryHostRequest(dp1), p1...)\n\treturn proto.ToAdminDescribeHistoryHostResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tresponse, err := g.c.DescribeQueue(ctx, proto.FromAdminDescribeQueueRequest(dp1), p1...)\n\treturn proto.ToAdminDescribeQueueResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tresponse, err := g.c.DescribeShardDistribution(ctx, proto.FromAdminDescribeShardDistributionRequest(dp1), p1...)\n\treturn proto.ToAdminDescribeShardDistributionResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, proto.FromAdminDescribeWorkflowExecutionRequest(ap1), p1...)\n\treturn proto.ToAdminDescribeWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDLQReplicationMessages(ctx, proto.FromAdminGetDLQReplicationMessagesRequest(gp1), p1...)\n\treturn proto.ToAdminGetDLQReplicationMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.c.GetDomainAsyncWorkflowConfiguraton(ctx, proto.FromAdminGetDomainAsyncWorkflowConfiguratonRequest(request), opts...)\n\treturn proto.ToAdminGetDomainAsyncWorkflowConfiguratonResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.GetDomainIsolationGroups(ctx, proto.FromAdminGetDomainIsolationGroupsRequest(request), opts...)\n\treturn proto.ToAdminGetDomainIsolationGroupsResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDomainReplicationMessages(ctx, proto.FromAdminGetDomainReplicationMessagesRequest(gp1), p1...)\n\treturn proto.ToAdminGetDomainReplicationMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tresponse, err := g.c.GetDynamicConfig(ctx, proto.FromAdminGetDynamicConfigRequest(gp1), p1...)\n\treturn proto.ToAdminGetDynamicConfigResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.GetGlobalIsolationGroups(ctx, proto.FromAdminGetGlobalIsolationGroupsRequest(request), opts...)\n\treturn proto.ToAdminGetGlobalIsolationGroupsResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetReplicationMessages(ctx, proto.FromAdminGetReplicationMessagesRequest(gp1), p1...)\n\treturn proto.ToAdminGetReplicationMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tresponse, err := g.c.GetWorkflowExecutionRawHistoryV2(ctx, proto.FromAdminGetWorkflowExecutionRawHistoryV2Request(gp1), p1...)\n\treturn proto.ToAdminGetWorkflowExecutionRawHistoryV2Response(response), proto.ToError(err)\n}\n\nfunc (g adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tresponse, err := g.c.ListDynamicConfig(ctx, proto.FromAdminListDynamicConfigRequest(lp1), p1...)\n\treturn proto.ToAdminListDynamicConfigResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tresponse, err := g.c.MaintainCorruptWorkflow(ctx, proto.FromAdminMaintainCorruptWorkflowRequest(ap1), p1...)\n\treturn proto.ToAdminMaintainCorruptWorkflowResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.c.MergeDLQMessages(ctx, proto.FromAdminMergeDLQMessagesRequest(mp1), p1...)\n\treturn proto.ToAdminMergeDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.PurgeDLQMessages(ctx, proto.FromAdminPurgeDLQMessagesRequest(pp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.c.ReadDLQMessages(ctx, proto.FromAdminReadDLQMessagesRequest(rp1), p1...)\n\treturn proto.ToAdminReadDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ReapplyEvents(ctx, proto.FromAdminReapplyEventsRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RefreshWorkflowTasks(ctx, proto.FromAdminRefreshWorkflowTasksRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RemoveTask(ctx, proto.FromAdminRemoveTaskRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ResendReplicationTasks(ctx, proto.FromAdminResendReplicationTasksRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ResetQueue(ctx, proto.FromAdminResetQueueRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RestoreDynamicConfig(ctx, proto.FromAdminRestoreDynamicConfigRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.c.UpdateDomainAsyncWorkflowConfiguraton(ctx, proto.FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(request), opts...)\n\treturn proto.ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.UpdateDomainIsolationGroups(ctx, proto.FromAdminUpdateDomainIsolationGroupsRequest(request), opts...)\n\treturn proto.ToAdminUpdateDomainIsolationGroupsResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.UpdateDynamicConfig(ctx, proto.FromAdminUpdateDynamicConfigRequest(up1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.UpdateGlobalIsolationGroups(ctx, proto.FromAdminUpdateGlobalIsolationGroupsRequest(request), opts...)\n\treturn proto.ToAdminUpdateGlobalIsolationGroupsResponse(response), proto.ToError(err)\n}\n\nfunc (g adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tresponse, err := g.c.UpdateTaskListPartitionConfig(ctx, proto.FromAdminUpdateTaskListPartitionConfigRequest(request), opts...)\n\treturn proto.ToAdminUpdateTaskListPartitionConfigResponse(response), proto.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/grpc/constructor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage grpc\n\nimport (\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n)\n\ntype (\n\tadminClient struct {\n\t\tc adminv1.AdminAPIYARPCClient\n\t}\n\n\tfrontendGRPCClientWrapper struct {\n\t\tapiv1.DomainAPIYARPCClient\n\t\tapiv1.WorkflowAPIYARPCClient\n\t\tapiv1.WorkerAPIYARPCClient\n\t\tapiv1.VisibilityAPIYARPCClient\n\t}\n\tfrontendClient struct {\n\t\tc *frontendGRPCClientWrapper\n\t}\n\thistoryClient struct {\n\t\tc historyv1.HistoryAPIYARPCClient\n\t}\n\tmatchingClient struct {\n\t\tc matchingv1.MatchingAPIYARPCClient\n\t}\n\tsharddistributorClient struct {\n\t\tc sharddistributorv1.ShardDistributorAPIYARPCClient\n\t}\n\tsharddistributorexecutorClient struct {\n\t\tc sharddistributorv1.ShardDistributorExecutorAPIYARPCClient\n\t}\n)\n\nfunc NewAdminClient(c adminv1.AdminAPIYARPCClient) admin.Client {\n\treturn adminClient{c}\n}\n\nfunc NewFrontendClient(\n\tdomain apiv1.DomainAPIYARPCClient,\n\tworkflow apiv1.WorkflowAPIYARPCClient,\n\tworker apiv1.WorkerAPIYARPCClient,\n\tvisibility apiv1.VisibilityAPIYARPCClient,\n) frontend.Client {\n\treturn frontendClient{&frontendGRPCClientWrapper{domain, workflow, worker, visibility}}\n}\n\nfunc NewHistoryClient(c historyv1.HistoryAPIYARPCClient) history.Client {\n\treturn historyClient{c}\n}\n\nfunc NewMatchingClient(c matchingv1.MatchingAPIYARPCClient) matching.Client {\n\treturn matchingClient{c}\n}\n\nfunc NewShardDistributorClient(c sharddistributorv1.ShardDistributorAPIYARPCClient) sharddistributor.Client {\n\treturn sharddistributorClient{c}\n}\n\nfunc NewShardDistributorExecutorClient(c sharddistributorv1.ShardDistributorExecutorAPIYARPCClient) sharddistributorexecutor.Client {\n\treturn sharddistributorexecutorClient{c}\n}\n"
  },
  {
    "path": "client/wrappers/grpc/frontend_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.CountWorkflowExecutions(ctx, proto.FromCountWorkflowExecutionsRequest(cp1), p1...)\n\treturn proto.ToCountWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.DeleteDomain(ctx, proto.FromDeleteDomainRequest(dp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.DeprecateDomain(ctx, proto.FromDeprecateDomainRequest(dp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tresponse, err := g.c.DescribeDomain(ctx, proto.FromDescribeDomainRequest(dp1), p1...)\n\treturn proto.ToDescribeDomainResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tresponse, err := g.c.DescribeTaskList(ctx, proto.FromDescribeTaskListRequest(dp1), p1...)\n\treturn proto.ToDescribeTaskListResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, proto.FromDescribeWorkflowExecutionRequest(dp1), p1...)\n\treturn proto.ToDescribeWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DiagnoseWorkflowExecution(ctx, proto.FromDiagnoseWorkflowExecutionRequest(dp1), p1...)\n\treturn proto.ToDiagnoseWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tresponse, err := g.c.FailoverDomain(ctx, proto.FromFailoverDomainRequest(fp1), p1...)\n\treturn proto.ToFailoverDomainResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tresponse, err := g.c.GetClusterInfo(ctx, &apiv1.GetClusterInfoRequest{}, p1...)\n\treturn proto.ToGetClusterInfoResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tresponse, err := g.c.GetSearchAttributes(ctx, &apiv1.GetSearchAttributesRequest{}, p1...)\n\treturn proto.ToGetSearchAttributesResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.c.GetTaskListsByDomain(ctx, proto.FromGetTaskListsByDomainRequest(gp1), p1...)\n\treturn proto.ToGetTaskListsByDomainResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tresponse, err := g.c.GetWorkflowExecutionHistory(ctx, proto.FromGetWorkflowExecutionHistoryRequest(gp1), p1...)\n\treturn proto.ToGetWorkflowExecutionHistoryResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListArchivedWorkflowExecutions(ctx, proto.FromListArchivedWorkflowExecutionsRequest(lp1), p1...)\n\treturn proto.ToListArchivedWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListClosedWorkflowExecutions(ctx, proto.FromListClosedWorkflowExecutionsRequest(lp1), p1...)\n\treturn proto.ToListClosedWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tresponse, err := g.c.ListDomains(ctx, proto.FromListDomainsRequest(lp1), p1...)\n\treturn proto.ToListDomainsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tresponse, err := g.c.ListFailoverHistory(ctx, proto.FromListFailoverHistoryRequest(lp1), p1...)\n\treturn proto.ToListFailoverHistoryResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListOpenWorkflowExecutions(ctx, proto.FromListOpenWorkflowExecutionsRequest(lp1), p1...)\n\treturn proto.ToListOpenWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.c.ListTaskListPartitions(ctx, proto.FromListTaskListPartitionsRequest(lp1), p1...)\n\treturn proto.ToListTaskListPartitionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListWorkflowExecutions(ctx, proto.FromListWorkflowExecutionsRequest(lp1), p1...)\n\treturn proto.ToListWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tresponse, err := g.c.PollForActivityTask(ctx, proto.FromPollForActivityTaskRequest(pp1), p1...)\n\treturn proto.ToPollForActivityTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tresponse, err := g.c.PollForDecisionTask(ctx, proto.FromPollForDecisionTaskRequest(pp1), p1...)\n\treturn proto.ToPollForDecisionTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, proto.FromQueryWorkflowRequest(qp1), p1...)\n\treturn proto.ToQueryWorkflowResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeat(ctx, proto.FromRecordActivityTaskHeartbeatRequest(rp1), p1...)\n\treturn proto.ToRecordActivityTaskHeartbeatResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeatByID(ctx, proto.FromRecordActivityTaskHeartbeatByIDRequest(rp1), p1...)\n\treturn proto.ToRecordActivityTaskHeartbeatByIDResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RefreshWorkflowTasks(ctx, proto.FromRefreshWorkflowTasksRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RegisterDomain(ctx, proto.FromRegisterDomainRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RequestCancelWorkflowExecution(ctx, proto.FromRequestCancelWorkflowExecutionRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tresponse, err := g.c.ResetStickyTaskList(ctx, proto.FromResetStickyTaskListRequest(rp1), p1...)\n\treturn proto.ToResetStickyTaskListResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.ResetWorkflowExecution(ctx, proto.FromResetWorkflowExecutionRequest(rp1), p1...)\n\treturn proto.ToResetWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCanceled(ctx, proto.FromRespondActivityTaskCanceledRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCanceledByID(ctx, proto.FromRespondActivityTaskCanceledByIDRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCompleted(ctx, proto.FromRespondActivityTaskCompletedRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCompletedByID(ctx, proto.FromRespondActivityTaskCompletedByIDRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskFailed(ctx, proto.FromRespondActivityTaskFailedRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskFailedByID(ctx, proto.FromRespondActivityTaskFailedByIDRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.c.RespondDecisionTaskCompleted(ctx, proto.FromRespondDecisionTaskCompletedRequest(rp1), p1...)\n\treturn proto.ToRespondDecisionTaskCompletedResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondDecisionTaskFailed(ctx, proto.FromRespondDecisionTaskFailedRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondQueryTaskCompleted(ctx, proto.FromRespondQueryTaskCompletedRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.RestartWorkflowExecution(ctx, proto.FromRestartWorkflowExecutionRequest(rp1), p1...)\n\treturn proto.ToRestartWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ScanWorkflowExecutions(ctx, proto.FromScanWorkflowExecutionsRequest(lp1), p1...)\n\treturn proto.ToScanWorkflowExecutionsResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecution(ctx, proto.FromSignalWithStartWorkflowExecutionRequest(sp1), p1...)\n\treturn proto.ToSignalWithStartWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecutionAsync(ctx, proto.FromSignalWithStartWorkflowExecutionAsyncRequest(sp1), p1...)\n\treturn proto.ToSignalWithStartWorkflowExecutionAsyncResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.SignalWorkflowExecution(ctx, proto.FromSignalWorkflowExecutionRequest(sp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecution(ctx, proto.FromStartWorkflowExecutionRequest(sp1), p1...)\n\treturn proto.ToStartWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecutionAsync(ctx, proto.FromStartWorkflowExecutionAsyncRequest(sp1), p1...)\n\treturn proto.ToStartWorkflowExecutionAsyncResponse(response), proto.ToError(err)\n}\n\nfunc (g frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.TerminateWorkflowExecution(ctx, proto.FromTerminateWorkflowExecutionRequest(tp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tresponse, err := g.c.UpdateDomain(ctx, proto.FromUpdateDomainRequest(up1), p1...)\n\treturn proto.ToUpdateDomainResponse(response), proto.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/grpc/history_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.CloseShard(ctx, proto.FromHistoryCloseShardRequest(cp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\tresponse, err := g.c.CountDLQMessages(ctx, proto.FromHistoryCountDLQMessagesRequest(cp1), p1...)\n\treturn proto.ToHistoryCountDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.c.DescribeHistoryHost(ctx, proto.FromHistoryDescribeHistoryHostRequest(dp1), p1...)\n\treturn proto.ToHistoryDescribeHistoryHostResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tresponse, err := g.c.DescribeMutableState(ctx, proto.FromHistoryDescribeMutableStateRequest(dp1), p1...)\n\treturn proto.ToHistoryDescribeMutableStateResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tresponse, err := g.c.DescribeQueue(ctx, proto.FromHistoryDescribeQueueRequest(dp1), p1...)\n\treturn proto.ToHistoryDescribeQueueResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, proto.FromHistoryDescribeWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToHistoryDescribeWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tresponse, err := g.c.GetCrossClusterTasks(ctx, proto.FromHistoryGetCrossClusterTasksRequest(gp1), p1...)\n\treturn proto.ToHistoryGetCrossClusterTasksResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDLQReplicationMessages(ctx, proto.FromHistoryGetDLQReplicationMessagesRequest(gp1), p1...)\n\treturn proto.ToHistoryGetDLQReplicationMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\tresponse, err := g.c.GetFailoverInfo(ctx, proto.FromHistoryGetFailoverInfoRequest(gp1), p1...)\n\treturn proto.ToHistoryGetFailoverInfoResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tresponse, err := g.c.GetMutableState(ctx, proto.FromHistoryGetMutableStateRequest(gp1), p1...)\n\treturn proto.ToHistoryGetMutableStateResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetReplicationMessages(ctx, proto.FromHistoryGetReplicationMessagesRequest(gp1), p1...)\n\treturn proto.ToHistoryGetReplicationMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.c.MergeDLQMessages(ctx, proto.FromHistoryMergeDLQMessagesRequest(mp1), p1...)\n\treturn proto.ToHistoryMergeDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.NotifyFailoverMarkers(ctx, proto.FromHistoryNotifyFailoverMarkersRequest(np1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tresponse, err := g.c.PollMutableState(ctx, proto.FromHistoryPollMutableStateRequest(pp1), p1...)\n\treturn proto.ToHistoryPollMutableStateResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.PurgeDLQMessages(ctx, proto.FromHistoryPurgeDLQMessagesRequest(pp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, proto.FromHistoryQueryWorkflowRequest(hp1), p1...)\n\treturn proto.ToHistoryQueryWorkflowResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tresponse, err := g.c.RatelimitUpdate(ctx, proto.FromHistoryRatelimitUpdateRequest(request), opts...)\n\treturn proto.ToHistoryRatelimitUpdateResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.c.ReadDLQMessages(ctx, proto.FromHistoryReadDLQMessagesRequest(rp1), p1...)\n\treturn proto.ToHistoryReadDLQMessagesResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ReapplyEvents(ctx, proto.FromHistoryReapplyEventsRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeat(ctx, proto.FromHistoryRecordActivityTaskHeartbeatRequest(hp1), p1...)\n\treturn proto.ToHistoryRecordActivityTaskHeartbeatResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskStarted(ctx, proto.FromHistoryRecordActivityTaskStartedRequest(rp1), p1...)\n\treturn proto.ToHistoryRecordActivityTaskStartedResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RecordChildExecutionCompleted(ctx, proto.FromHistoryRecordChildExecutionCompletedRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tresponse, err := g.c.RecordDecisionTaskStarted(ctx, proto.FromHistoryRecordDecisionTaskStartedRequest(rp1), p1...)\n\treturn proto.ToHistoryRecordDecisionTaskStartedResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RefreshWorkflowTasks(ctx, proto.FromHistoryRefreshWorkflowTasksRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RemoveSignalMutableState(ctx, proto.FromHistoryRemoveSignalMutableStateRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RemoveTask(ctx, proto.FromHistoryRemoveTaskRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ReplicateEventsV2(ctx, proto.FromHistoryReplicateEventsV2Request(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RequestCancelWorkflowExecution(ctx, proto.FromHistoryRequestCancelWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ResetQueue(ctx, proto.FromHistoryResetQueueRequest(rp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tresponse, err := g.c.ResetStickyTaskList(ctx, proto.FromHistoryResetStickyTaskListRequest(hp1), p1...)\n\treturn proto.ToHistoryResetStickyTaskListResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.ResetWorkflowExecution(ctx, proto.FromHistoryResetWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToHistoryResetWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCanceled(ctx, proto.FromHistoryRespondActivityTaskCanceledRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskCompleted(ctx, proto.FromHistoryRespondActivityTaskCompletedRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondActivityTaskFailed(ctx, proto.FromHistoryRespondActivityTaskFailedRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tresponse, err := g.c.RespondCrossClusterTasksCompleted(ctx, proto.FromHistoryRespondCrossClusterTasksCompletedRequest(rp1), p1...)\n\treturn proto.ToHistoryRespondCrossClusterTasksCompletedResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.c.RespondDecisionTaskCompleted(ctx, proto.FromHistoryRespondDecisionTaskCompletedRequest(hp1), p1...)\n\treturn proto.ToHistoryRespondDecisionTaskCompletedResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondDecisionTaskFailed(ctx, proto.FromHistoryRespondDecisionTaskFailedRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.ScheduleDecisionTask(ctx, proto.FromHistoryScheduleDecisionTaskRequest(sp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecution(ctx, proto.FromHistorySignalWithStartWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToHistorySignalWithStartWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.SignalWorkflowExecution(ctx, proto.FromHistorySignalWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecution(ctx, proto.FromHistoryStartWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToHistoryStartWorkflowExecutionResponse(response), proto.ToError(err)\n}\n\nfunc (g historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.SyncActivity(ctx, proto.FromHistorySyncActivityRequest(sp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.SyncShardStatus(ctx, proto.FromHistorySyncShardStatusRequest(sp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.TerminateWorkflowExecution(ctx, proto.FromHistoryTerminateWorkflowExecutionRequest(hp1), p1...)\n\treturn proto.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/grpc/matching_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\tresponse, err := g.c.AddActivityTask(ctx, proto.FromMatchingAddActivityTaskRequest(ap1), p1...)\n\treturn proto.ToMatchingAddActivityTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\tresponse, err := g.c.AddDecisionTask(ctx, proto.FromMatchingAddDecisionTaskRequest(ap1), p1...)\n\treturn proto.ToMatchingAddDecisionTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.CancelOutstandingPoll(ctx, proto.FromMatchingCancelOutstandingPollRequest(cp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tresponse, err := g.c.DescribeTaskList(ctx, proto.FromMatchingDescribeTaskListRequest(mp1), p1...)\n\treturn proto.ToMatchingDescribeTaskListResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.c.GetTaskListsByDomain(ctx, proto.FromMatchingGetTaskListsByDomainRequest(gp1), p1...)\n\treturn proto.ToMatchingGetTaskListsByDomainResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.c.ListTaskListPartitions(ctx, proto.FromMatchingListTaskListPartitionsRequest(mp1), p1...)\n\treturn proto.ToMatchingListTaskListPartitionsResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tresponse, err := g.c.PollForActivityTask(ctx, proto.FromMatchingPollForActivityTaskRequest(mp1), p1...)\n\treturn proto.ToMatchingPollForActivityTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tresponse, err := g.c.PollForDecisionTask(ctx, proto.FromMatchingPollForDecisionTaskRequest(mp1), p1...)\n\treturn proto.ToMatchingPollForDecisionTaskResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, proto.FromMatchingQueryWorkflowRequest(mp1), p1...)\n\treturn proto.ToMatchingQueryWorkflowResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\tresponse, err := g.c.RefreshTaskListPartitionConfig(ctx, proto.FromMatchingRefreshTaskListPartitionConfigRequest(mp1), p1...)\n\treturn proto.ToMatchingRefreshTaskListPartitionConfigResponse(response), proto.ToError(err)\n}\n\nfunc (g matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\t_, err = g.c.RespondQueryTaskCompleted(ctx, proto.FromMatchingRespondQueryTaskCompletedRequest(mp1), p1...)\n\treturn proto.ToError(err)\n}\n\nfunc (g matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\tresponse, err := g.c.UpdateTaskListPartitionConfig(ctx, proto.FromMatchingUpdateTaskListPartitionConfigRequest(mp1), p1...)\n\treturn proto.ToMatchingUpdateTaskListPartitionConfigResponse(response), proto.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/grpc/sharddistributor_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g sharddistributorClient) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest, p1 ...yarpc.CallOption) (gp2 *types.GetShardOwnerResponse, err error) {\n\tresponse, err := g.c.GetShardOwner(ctx, proto.FromShardDistributorGetShardOwnerRequest(gp1), p1...)\n\treturn proto.ToShardDistributorGetShardOwnerResponse(response), proto.ToError(err)\n}\n\nfunc (g sharddistributorClient) WatchNamespaceState(ctx context.Context, wp1 *types.WatchNamespaceStateRequest, p1 ...yarpc.CallOption) (w1 sharddistributor.WatchNamespaceStateClient, err error) {\n\tstream, err := g.c.WatchNamespaceState(ctx, proto.FromShardDistributorWatchNamespaceStateRequest(wp1), p1...)\n\tif err != nil {\n\t\treturn nil, proto.ToError(err)\n\t}\n\treturn &watchnamespacestateClient{stream: stream}, nil\n}\n\ntype watchnamespacestateClient struct {\n\tstream sharddistributorv1.ShardDistributorAPIServiceWatchNamespaceStateYARPCClient\n}\n\nfunc (c *watchnamespacestateClient) Context() context.Context {\n\treturn c.stream.Context()\n}\n\nfunc (c *watchnamespacestateClient) Recv(options ...yarpc.StreamOption) (*types.WatchNamespaceStateResponse, error) {\n\tresponse, err := c.stream.Recv(options...)\n\tif err != nil {\n\t\treturn nil, proto.ToError(err)\n\t}\n\treturn proto.ToShardDistributorWatchNamespaceStateResponse(response), nil\n}\n\nfunc (c *watchnamespacestateClient) CloseSend(options ...yarpc.StreamOption) error {\n\treturn proto.ToError(c.stream.CloseSend(options...))\n}\n"
  },
  {
    "path": "client/wrappers/grpc/sharddistributorexecutor_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g sharddistributorexecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tresponse, err := g.c.Heartbeat(ctx, proto.FromShardDistributorExecutorHeartbeatRequest(ep1), p1...)\n\treturn proto.ToShardDistributorExecutorHeartbeatResponse(response), proto.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/metered/admin_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// adminClient implements admin.Client interface instrumented with retries\ntype adminClient struct {\n\tclient        admin.Client\n\tmetricsClient metrics.Client\n}\n\n// NewAdminClient creates a new instance of adminClient with retry policy\nfunc NewAdminClient(client admin.Client, metricsClient metrics.Client) admin.Client {\n\treturn &adminClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientAddSearchAttributeScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientAddSearchAttributeScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.AddSearchAttribute(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientCloseShardScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientCloseShardScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.CloseShard(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientCountDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientCountDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tcp2, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn cp2, err\n}\n\nfunc (c *adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDeleteWorkflowScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDeleteWorkflowScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tap2, err = c.client.DeleteWorkflow(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ap2, err\n}\n\nfunc (c *adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeClusterScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeClusterScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp1, err = c.client.DescribeCluster(ctx, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp1, err\n}\n\nfunc (c *adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeHistoryHostScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeHistoryHostScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeQueueScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeQueueScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeShardDistributionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeShardDistributionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeShardDistribution(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientDescribeWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tap2, err = c.client.DescribeWorkflowExecution(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ap2, err\n}\n\nfunc (c *adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDLQReplicationMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDLQReplicationMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainAsyncWorkflowConfiguratonScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainAsyncWorkflowConfiguratonScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp1, err = c.client.GetDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp1, err\n}\n\nfunc (c *adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainIsolationGroupsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainIsolationGroupsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp1, err = c.client.GetDomainIsolationGroups(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp1, err\n}\n\nfunc (c *adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainReplicationMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDomainReplicationMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetDomainReplicationMessages(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDynamicConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetDynamicConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetDynamicConfig(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetGlobalIsolationGroupsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetGlobalIsolationGroupsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp1, err = c.client.GetGlobalIsolationGroups(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp1, err\n}\n\nfunc (c *adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetReplicationMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetReplicationMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetWorkflowExecutionRawHistoryV2Scope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientGetWorkflowExecutionRawHistoryV2Scope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetWorkflowExecutionRawHistoryV2(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientListDynamicConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientListDynamicConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListDynamicConfig(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientMaintainCorruptWorkflowScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientMaintainCorruptWorkflowScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tap2, err = c.client.MaintainCorruptWorkflow(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ap2, err\n}\n\nfunc (c *adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientMergeDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientMergeDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientPurgeDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientPurgeDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientReadDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientReadDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientReapplyEventsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientReapplyEventsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ReapplyEvents(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRefreshWorkflowTasksScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRefreshWorkflowTasksScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRemoveTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRemoveTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RemoveTask(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientResendReplicationTasksScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientResendReplicationTasksScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ResendReplicationTasks(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientResetQueueScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientResetQueueScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ResetQueue(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRestoreDynamicConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientRestoreDynamicConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RestoreDynamicConfig(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDomainAsyncWorkflowConfiguratonScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDomainAsyncWorkflowConfiguratonScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tup1, err = c.client.UpdateDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn up1, err\n}\n\nfunc (c *adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDomainIsolationGroupsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDomainIsolationGroupsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tup1, err = c.client.UpdateDomainIsolationGroups(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn up1, err\n}\n\nfunc (c *adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDynamicConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateDynamicConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.UpdateDynamicConfig(ctx, up1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateGlobalIsolationGroupsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateGlobalIsolationGroupsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tup1, err = c.client.UpdateGlobalIsolationGroups(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn up1, err\n}\n\nfunc (c *adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateTaskListPartitionConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.AdminClientUpdateTaskListPartitionConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tup1, err = c.client.UpdateTaskListPartitionConfig(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn up1, err\n}\n"
  },
  {
    "path": "client/wrappers/metered/base.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport \"context\"\n\ntype retryCountKeyType string\n\nconst retryCountKey = retryCountKeyType(\"retryCount\")\n\nfunc getRetryCountFromContext(ctx context.Context) int {\n\tif retryCount, ok := ctx.Value(retryCountKey).(int); ok {\n\t\treturn retryCount\n\t}\n\treturn -1\n}\n"
  },
  {
    "path": "client/wrappers/metered/frontend_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// frontendClient implements frontend.Client interface instrumented with retries\ntype frontendClient struct {\n\tclient        frontend.Client\n\tmetricsClient metrics.Client\n}\n\n// NewFrontendClient creates a new instance of frontendClient with retry policy\nfunc NewFrontendClient(client frontend.Client, metricsClient metrics.Client) frontend.Client {\n\treturn &frontendClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientCountWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientCountWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tcp2, err = c.client.CountWorkflowExecutions(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn cp2, err\n}\n\nfunc (c *frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDeleteDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDeleteDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.DeleteDomain(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDeprecateDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDeprecateDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.DeprecateDomain(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeDomain(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeTaskListScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeTaskListScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeTaskList(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDescribeWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeWorkflowExecution(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDiagnoseWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientDiagnoseWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DiagnoseWorkflowExecution(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientFailoverDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientFailoverDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tfp2, err = c.client.FailoverDomain(ctx, fp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn fp2, err\n}\n\nfunc (c *frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetClusterInfoScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetClusterInfoScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tcp1, err = c.client.GetClusterInfo(ctx, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn cp1, err\n}\n\nfunc (c *frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetSearchAttributesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetSearchAttributesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp1, err = c.client.GetSearchAttributes(ctx, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp1, err\n}\n\nfunc (c *frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetTaskListsByDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetTaskListsByDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetWorkflowExecutionHistoryScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientGetWorkflowExecutionHistoryScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetWorkflowExecutionHistory(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListArchivedWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListArchivedWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListArchivedWorkflowExecutions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListClosedWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListClosedWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListClosedWorkflowExecutions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListDomainsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListDomainsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListDomains(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListFailoverHistoryScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListFailoverHistoryScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListFailoverHistory(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListOpenWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListOpenWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListOpenWorkflowExecutions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListTaskListPartitionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListTaskListPartitionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListTaskListPartitions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientListWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ListWorkflowExecutions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientPollForActivityTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientPollForActivityTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tpp2, err = c.client.PollForActivityTask(ctx, pp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn pp2, err\n}\n\nfunc (c *frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientPollForDecisionTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientPollForDecisionTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tpp2, err = c.client.PollForDecisionTask(ctx, pp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn pp2, err\n}\n\nfunc (c *frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientQueryWorkflowScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientQueryWorkflowScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tqp2, err = c.client.QueryWorkflow(ctx, qp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn qp2, err\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRecordActivityTaskHeartbeatScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRecordActivityTaskHeartbeatScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RecordActivityTaskHeartbeat(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRecordActivityTaskHeartbeatByIDScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRecordActivityTaskHeartbeatByIDScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RecordActivityTaskHeartbeatByID(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRefreshWorkflowTasksScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRefreshWorkflowTasksScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRegisterDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRegisterDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RegisterDomain(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRequestCancelWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRequestCancelWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RequestCancelWorkflowExecution(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientResetStickyTaskListScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientResetStickyTaskListScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.ResetStickyTaskList(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientResetWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientResetWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.ResetWorkflowExecution(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCanceledScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCanceledScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCanceled(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCanceledByIDScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCanceledByIDScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCanceledByID(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCompleted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCompletedByIDScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskCompletedByIDScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCompletedByID(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskFailedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskFailedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskFailed(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskFailedByIDScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondActivityTaskFailedByIDScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskFailedByID(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondDecisionTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondDecisionTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RespondDecisionTaskCompleted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondDecisionTaskFailedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondDecisionTaskFailedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondDecisionTaskFailed(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondQueryTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRespondQueryTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondQueryTaskCompleted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRestartWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientRestartWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RestartWorkflowExecution(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientScanWorkflowExecutionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientScanWorkflowExecutionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp2, err = c.client.ScanWorkflowExecutions(ctx, lp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp2, err\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWithStartWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWithStartWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp2, err = c.client.SignalWithStartWorkflowExecution(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp2, err\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWithStartWorkflowExecutionAsyncScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWithStartWorkflowExecutionAsyncScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp2, err = c.client.SignalWithStartWorkflowExecutionAsync(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp2, err\n}\n\nfunc (c *frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientSignalWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.SignalWorkflowExecution(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientStartWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientStartWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp2, err = c.client.StartWorkflowExecution(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp2, err\n}\n\nfunc (c *frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientStartWorkflowExecutionAsyncScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientStartWorkflowExecutionAsyncScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp2, err = c.client.StartWorkflowExecutionAsync(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp2, err\n}\n\nfunc (c *frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientTerminateWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientTerminateWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.TerminateWorkflowExecution(ctx, tp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientUpdateDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.FrontendClientUpdateDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tup2, err = c.client.UpdateDomain(ctx, up1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn up2, err\n}\n"
  },
  {
    "path": "client/wrappers/metered/history_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// historyClient implements history.Client interface instrumented with retries\ntype historyClient struct {\n\tclient        history.Client\n\tmetricsClient metrics.Client\n}\n\n// NewHistoryClient creates a new instance of historyClient with retry policy\nfunc NewHistoryClient(client history.Client, metricsClient metrics.Client) history.Client {\n\treturn &historyClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientCloseShardScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientCloseShardScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.CloseShard(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientCountDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientCountDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\thp1, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn hp1, err\n}\n\nfunc (c *historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeHistoryHostScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeHistoryHostScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeMutableStateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeMutableStateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeMutableState(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeQueueScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeQueueScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp2, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp2, err\n}\n\nfunc (c *historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientDescribeWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp1, err = c.client.DescribeWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp1, err\n}\n\nfunc (c *historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetCrossClusterTasksScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetCrossClusterTasksScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetCrossClusterTasks(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetDLQReplicationMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetDLQReplicationMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetFailoverInfoScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetFailoverInfoScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetFailoverInfo(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetMutableStateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetMutableStateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetMutableState(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetReplicationMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientGetReplicationMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientMergeDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientMergeDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientNotifyFailoverMarkersScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientNotifyFailoverMarkersScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.NotifyFailoverMarkers(ctx, np1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientPollMutableStateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientPollMutableStateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tpp2, err = c.client.PollMutableState(ctx, pp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn pp2, err\n}\n\nfunc (c *historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientPurgeDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientPurgeDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientQueryWorkflowScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientQueryWorkflowScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\thp2, err = c.client.QueryWorkflow(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn hp2, err\n}\n\nfunc (c *historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRatelimitUpdateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRatelimitUpdateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp1, err = c.client.RatelimitUpdate(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp1, err\n}\n\nfunc (c *historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReadDLQMessagesScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReadDLQMessagesScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReapplyEventsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReapplyEventsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ReapplyEvents(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordActivityTaskHeartbeatScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordActivityTaskHeartbeatScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp1, err = c.client.RecordActivityTaskHeartbeat(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp1, err\n}\n\nfunc (c *historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordActivityTaskStartedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordActivityTaskStartedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RecordActivityTaskStarted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordChildExecutionCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordChildExecutionCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RecordChildExecutionCompleted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordDecisionTaskStartedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRecordDecisionTaskStartedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RecordDecisionTaskStarted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRefreshWorkflowTasksScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRefreshWorkflowTasksScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RefreshWorkflowTasks(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRemoveSignalMutableStateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRemoveSignalMutableStateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RemoveSignalMutableState(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRemoveTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRemoveTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RemoveTask(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReplicateEventsV2Scope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientReplicateEventsV2Scope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ReplicateEventsV2(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRequestCancelWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRequestCancelWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RequestCancelWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetQueueScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetQueueScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ResetQueue(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetStickyTaskListScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetStickyTaskListScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\thp2, err = c.client.ResetStickyTaskList(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn hp2, err\n}\n\nfunc (c *historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientResetWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp1, err = c.client.ResetWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp1, err\n}\n\nfunc (c *historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskCanceledScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskCanceledScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCanceled(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskCompleted(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskFailedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskFailedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondActivityTaskFailed(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondCrossClusterTasksCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondCrossClusterTasksCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\trp2, err = c.client.RespondCrossClusterTasksCompleted(ctx, rp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn rp2, err\n}\n\nfunc (c *historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondDecisionTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondDecisionTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\thp2, err = c.client.RespondDecisionTaskCompleted(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn hp2, err\n}\n\nfunc (c *historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondDecisionTaskFailedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientRespondDecisionTaskFailedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondDecisionTaskFailed(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientScheduleDecisionTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientScheduleDecisionTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.ScheduleDecisionTask(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSignalWithStartWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSignalWithStartWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp1, err = c.client.SignalWithStartWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp1, err\n}\n\nfunc (c *historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSignalWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSignalWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.SignalWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientStartWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientStartWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tsp1, err = c.client.StartWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn sp1, err\n}\n\nfunc (c *historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSyncActivityScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSyncActivityScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.SyncActivity(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSyncShardStatusScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientSyncShardStatusScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.SyncShardStatus(ctx, sp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientTerminateWorkflowExecutionScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.HistoryClientTerminateWorkflowExecutionScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.TerminateWorkflowExecution(ctx, hp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "client/wrappers/metered/matching_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"strings\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// matchingClient implements matching.Client interface instrumented with retries\ntype matchingClient struct {\n\tclient        matching.Client\n\tmetricsClient metrics.Client\n}\n\n// NewMatchingClient creates a new instance of matchingClient with retry policy\nfunc NewMatchingClient(client matching.Client, metricsClient metrics.Client) matching.Client {\n\treturn &matchingClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientAddActivityTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientAddActivityTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, ap1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tap2, err = c.client.AddActivityTask(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ap2, err\n}\n\nfunc (c *matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientAddDecisionTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientAddDecisionTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, ap1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tap2, err = c.client.AddDecisionTask(ctx, ap1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ap2, err\n}\n\nfunc (c *matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientCancelOutstandingPollScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientCancelOutstandingPollScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, cp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.CancelOutstandingPoll(ctx, cp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientDescribeTaskListScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientDescribeTaskListScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tdp1, err = c.client.DescribeTaskList(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn dp1, err\n}\n\nfunc (c *matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientGetTaskListsByDomainScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientGetTaskListsByDomainScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, gp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientListTaskListPartitionsScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientListTaskListPartitionsScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tlp1, err = c.client.ListTaskListPartitions(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn lp1, err\n}\n\nfunc (c *matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientPollForActivityTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientPollForActivityTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.PollForActivityTask(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientPollForDecisionTaskScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientPollForDecisionTaskScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.PollForDecisionTask(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientQueryWorkflowScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientQueryWorkflowScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.QueryWorkflow(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientRefreshTaskListPartitionConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientRefreshTaskListPartitionConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.RefreshTaskListPartitionConfig(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\nfunc (c *matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientRespondQueryTaskCompletedScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientRespondQueryTaskCompletedScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\terr = c.client.RespondQueryTaskCompleted(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (c *matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientUpdateTaskListPartitionConfigScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.MatchingClientUpdateTaskListPartitionConfigScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\tc.emitForwardedFromStats(scope, mp1)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tmp2, err = c.client.UpdateTaskListPartitionConfig(ctx, mp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn mp2, err\n}\n\ntype forwardedRequest interface {\n\tGetForwardedFrom() string\n\tGetTaskList() *types.TaskList\n}\n\nfunc (c *matchingClient) emitForwardedFromStats(scope metrics.Scope, req any) {\n\tp, ok := req.(forwardedRequest)\n\tif !ok || p.GetTaskList() == nil {\n\t\treturn\n\t}\n\n\ttaskList := p.GetTaskList()\n\tforwardedFrom := p.GetForwardedFrom()\n\n\tisChildPartition := strings.HasPrefix(taskList.GetName(), constants.ReservedTaskListPrefix)\n\tif forwardedFrom != \"\" {\n\t\tscope.IncCounter(metrics.MatchingClientForwardedCounter)\n\t\treturn\n\t}\n\n\tif isChildPartition {\n\t\tscope.IncCounter(metrics.MatchingClientInvalidTaskListName)\n\t}\n\treturn\n}\n"
  },
  {
    "path": "client/wrappers/metered/metered_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestWrappers(t *testing.T) {\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := frontend.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(nil, nil).Times(1)\n\n\t\tretryableClient := NewFrontendClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.CountWorkflowExecutions(context.Background(), &types.CountWorkflowExecutionsRequest{})\n\t\tassert.NoError(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 1, \"there should be a single counter registered\")\n\t\tassert.Len(t, testScope.Snapshot().Timers(), 1, \"there should be a single timer registered\")\n\t})\n\tt.Run(\"error\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := frontend.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(nil, errors.New(\"error\"))\n\n\t\tretryableClient := NewFrontendClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.CountWorkflowExecutions(context.Background(), &types.CountWorkflowExecutionsRequest{})\n\t\tassert.Error(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 2, \"there should be two counters registered, one for call and one for failure\")\n\t})\n}\n\n// Matching has a special logic that should emit a metric if a request is forwarded.\nfunc TestMatching(t *testing.T) {\n\tt.Run(\"forwarded\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := matching.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\t\tretryableClient := NewMatchingClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.AddActivityTask(context.Background(), &types.AddActivityTaskRequest{\n\t\t\tForwardedFrom: \"test\",\n\t\t\tTaskList:      &types.TaskList{Name: \"test\"},\n\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 2, \"there should be two counters registered, one for call and one for forwarded\")\n\t})\n\tt.Run(\"forwarded poller\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := matching.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(&types.MatchingPollForDecisionTaskResponse{}, nil).Times(1)\n\n\t\tretryableClient := NewMatchingClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.PollForDecisionTask(context.Background(), &types.MatchingPollForDecisionTaskRequest{\n\t\t\tForwardedFrom: \"test\",\n\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\tTaskList: &types.TaskList{Name: \"test\"},\n\t\t\t},\n\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 2, \"there should be two counters registered, one for call and one for forwarded\")\n\t})\n\tt.Run(\"not forwarded\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := matching.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\t\tretryableClient := NewMatchingClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.AddActivityTask(context.Background(), &types.AddActivityTaskRequest{\n\t\t\tForwardedFrom: \"\",\n\t\t\tTaskList:      &types.TaskList{Name: \"test\"},\n\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 1, \"there should be one counters registered, one for call\")\n\t})\n\tt.Run(\"invalid task list\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tclientMock := matching.NewMockClient(ctrl)\n\n\t\ttestScope := tally.NewTestScope(\"\", nil)\n\t\tmetricsClient := metrics.NewClient(testScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\n\t\tclientMock.EXPECT().AddActivityTask(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\t\tretryableClient := NewMatchingClient(\n\t\t\tclientMock,\n\t\t\tmetricsClient)\n\n\t\t_, err := retryableClient.AddActivityTask(context.Background(), &types.AddActivityTaskRequest{\n\t\t\tForwardedFrom: \"\",\n\t\t\tTaskList:      &types.TaskList{Name: constants.ReservedTaskListPrefix + \"test\"},\n\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Len(t, testScope.Snapshot().Counters(), 2, \"there should be two counters registered, one for call and one for invalid task list\")\n\t})\n}\n"
  },
  {
    "path": "client/wrappers/metered/sharddistributor_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// sharddistributorClient implements sharddistributor.Client interface instrumented with retries\ntype sharddistributorClient struct {\n\tclient        sharddistributor.Client\n\tmetricsClient metrics.Client\n}\n\n// NewShardDistributorClient creates a new instance of sharddistributorClient with retry policy\nfunc NewShardDistributorClient(client sharddistributor.Client, metricsClient metrics.Client) sharddistributor.Client {\n\treturn &sharddistributorClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *sharddistributorClient) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest, p1 ...yarpc.CallOption) (gp2 *types.GetShardOwnerResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorClientGetShardOwnerScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorClientGetShardOwnerScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tgp2, err = c.client.GetShardOwner(ctx, gp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn gp2, err\n}\n\nfunc (c *sharddistributorClient) WatchNamespaceState(ctx context.Context, wp1 *types.WatchNamespaceStateRequest, p1 ...yarpc.CallOption) (w1 sharddistributor.WatchNamespaceStateClient, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorClientWatchNamespaceStateScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorClientWatchNamespaceStateScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tw1, err = c.client.WatchNamespaceState(ctx, wp1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn w1, err\n}\n"
  },
  {
    "path": "client/wrappers/metered/sharddistributorexecutor_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// sharddistributorexecutorClient implements sharddistributorexecutor.Client interface instrumented with retries\ntype sharddistributorexecutorClient struct {\n\tclient        sharddistributorexecutor.Client\n\tmetricsClient metrics.Client\n}\n\n// NewShardDistributorExecutorClient creates a new instance of sharddistributorexecutorClient with retry policy\nfunc NewShardDistributorExecutorClient(client sharddistributorexecutor.Client, metricsClient metrics.Client) sharddistributorexecutor.Client {\n\treturn &sharddistributorexecutorClient{\n\t\tclient:        client,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (c *sharddistributorexecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tretryCount := getRetryCountFromContext(ctx)\n\n\tvar scope metrics.Scope\n\tif retryCount == -1 {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorExecutorClientHeartbeatScope)\n\t} else {\n\t\tscope = c.metricsClient.Scope(metrics.ShardDistributorExecutorClientHeartbeatScope, metrics.IsRetryTag(retryCount > 0))\n\t}\n\n\tscope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := scope.StartTimer(metrics.CadenceClientLatency)\n\tep2, err = c.client.Heartbeat(ctx, ep1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn ep2, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/admin_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// adminClient implements admin.Client interface instrumented with retries\ntype adminClient struct {\n\tclient        admin.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewAdminClient creates a new instance of adminClient with retry policy\nfunc NewAdminClient(client admin.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) admin.Client {\n\treturn &adminClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.AddSearchAttribute(ctx, ap1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.CloseShard(ctx, cp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tvar resp *types.CountDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tvar resp *types.AdminDeleteWorkflowResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DeleteWorkflow(ctx, ap1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tvar resp *types.DescribeClusterResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeCluster(ctx, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tvar resp *types.DescribeHistoryHostResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tvar resp *types.DescribeQueueResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tvar resp *types.DescribeShardDistributionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeShardDistribution(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tvar resp *types.AdminDescribeWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeWorkflowExecution(ctx, ap1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tvar resp *types.GetDLQReplicationMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tvar resp *types.GetDomainAsyncWorkflowConfiguratonResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tvar resp *types.GetDomainIsolationGroupsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDomainIsolationGroups(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tvar resp *types.GetDomainReplicationMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDomainReplicationMessages(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tvar resp *types.GetDynamicConfigResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDynamicConfig(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tvar resp *types.GetGlobalIsolationGroupsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetGlobalIsolationGroups(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tvar resp *types.GetReplicationMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tvar resp *types.GetWorkflowExecutionRawHistoryV2Response\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetWorkflowExecutionRawHistoryV2(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tvar resp *types.ListDynamicConfigResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListDynamicConfig(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tvar resp *types.AdminMaintainWorkflowResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.MaintainCorruptWorkflow(ctx, ap1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tvar resp *types.MergeDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tvar resp *types.ReadDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ReapplyEvents(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RemoveTask(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ResendReplicationTasks(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ResetQueue(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RestoreDynamicConfig(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tvar resp *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tvar resp *types.UpdateDomainIsolationGroupsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateDomainIsolationGroups(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.UpdateDynamicConfig(ctx, up1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tvar resp *types.UpdateGlobalIsolationGroupsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateGlobalIsolationGroups(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tvar resp *types.UpdateTaskListPartitionConfigResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateTaskListPartitionConfig(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/frontend_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// frontendClient implements frontend.Client interface instrumented with retries\ntype frontendClient struct {\n\tclient        frontend.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewFrontendClient creates a new instance of frontendClient with retry policy\nfunc NewFrontendClient(client frontend.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) frontend.Client {\n\treturn &frontendClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tvar resp *types.CountWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.CountWorkflowExecutions(ctx, cp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.DeleteDomain(ctx, dp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.DeprecateDomain(ctx, dp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tvar resp *types.DescribeDomainResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeDomain(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tvar resp *types.DescribeTaskListResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeTaskList(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tvar resp *types.DescribeWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeWorkflowExecution(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tvar resp *types.DiagnoseWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DiagnoseWorkflowExecution(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tvar resp *types.FailoverDomainResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.FailoverDomain(ctx, fp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tvar resp *types.ClusterInfo\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetClusterInfo(ctx, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tvar resp *types.GetSearchAttributesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetSearchAttributes(ctx, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tvar resp *types.GetTaskListsByDomainResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tvar resp *types.GetWorkflowExecutionHistoryResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetWorkflowExecutionHistory(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tvar resp *types.ListArchivedWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListArchivedWorkflowExecutions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tvar resp *types.ListClosedWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListClosedWorkflowExecutions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tvar resp *types.ListDomainsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListDomains(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tvar resp *types.ListFailoverHistoryResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListFailoverHistory(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tvar resp *types.ListOpenWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListOpenWorkflowExecutions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tvar resp *types.ListTaskListPartitionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListTaskListPartitions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tvar resp *types.ListWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListWorkflowExecutions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tvar resp *types.PollForActivityTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.PollForActivityTask(ctx, pp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tvar resp *types.PollForDecisionTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.PollForDecisionTask(ctx, pp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tvar resp *types.QueryWorkflowResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.QueryWorkflow(ctx, qp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tvar resp *types.RecordActivityTaskHeartbeatResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RecordActivityTaskHeartbeat(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tvar resp *types.RecordActivityTaskHeartbeatResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RecordActivityTaskHeartbeatByID(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RegisterDomain(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RequestCancelWorkflowExecution(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tvar resp *types.ResetStickyTaskListResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ResetStickyTaskList(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tvar resp *types.ResetWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ResetWorkflowExecution(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCanceled(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCanceledByID(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCompleted(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCompletedByID(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskFailed(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskFailedByID(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tvar resp *types.RespondDecisionTaskCompletedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RespondDecisionTaskCompleted(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondDecisionTaskFailed(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondQueryTaskCompleted(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tvar resp *types.RestartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RestartWorkflowExecution(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tvar resp *types.ListWorkflowExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ScanWorkflowExecutions(ctx, lp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tvar resp *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.SignalWithStartWorkflowExecution(ctx, sp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tvar resp *types.SignalWithStartWorkflowExecutionAsyncResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.SignalWithStartWorkflowExecutionAsync(ctx, sp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.SignalWorkflowExecution(ctx, sp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tvar resp *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.StartWorkflowExecution(ctx, sp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tvar resp *types.StartWorkflowExecutionAsyncResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.StartWorkflowExecutionAsync(ctx, sp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.TerminateWorkflowExecution(ctx, tp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tvar resp *types.UpdateDomainResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateDomain(ctx, up1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/history_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// historyClient implements history.Client interface instrumented with retries\ntype historyClient struct {\n\tclient        history.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewHistoryClient creates a new instance of historyClient with retry policy\nfunc NewHistoryClient(client history.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) history.Client {\n\treturn &historyClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.CloseShard(ctx, cp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\tvar resp *types.HistoryCountDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.CountDLQMessages(ctx, cp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tvar resp *types.DescribeHistoryHostResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeHistoryHost(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tvar resp *types.DescribeMutableStateResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeMutableState(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tvar resp *types.DescribeQueueResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeQueue(ctx, dp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tvar resp *types.DescribeWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeWorkflowExecution(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tvar resp *types.GetCrossClusterTasksResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetCrossClusterTasks(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tvar resp *types.GetDLQReplicationMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\tvar resp *types.GetFailoverInfoResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetFailoverInfo(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tvar resp *types.GetMutableStateResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetMutableState(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tvar resp *types.GetReplicationMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetReplicationMessages(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tvar resp *types.MergeDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.MergeDLQMessages(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.NotifyFailoverMarkers(ctx, np1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tvar resp *types.PollMutableStateResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.PollMutableState(ctx, pp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.PurgeDLQMessages(ctx, pp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tvar resp *types.HistoryQueryWorkflowResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.QueryWorkflow(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tvar resp *types.RatelimitUpdateResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RatelimitUpdate(ctx, request, opts...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tvar resp *types.ReadDLQMessagesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ReadDLQMessages(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ReapplyEvents(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tvar resp *types.RecordActivityTaskHeartbeatResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RecordActivityTaskHeartbeat(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tvar resp *types.RecordActivityTaskStartedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RecordActivityTaskStarted(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RecordChildExecutionCompleted(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tvar resp *types.RecordDecisionTaskStartedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RecordDecisionTaskStarted(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RefreshWorkflowTasks(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RemoveSignalMutableState(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RemoveTask(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ReplicateEventsV2(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RequestCancelWorkflowExecution(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ResetQueue(ctx, rp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tvar resp *types.HistoryResetStickyTaskListResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ResetStickyTaskList(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tvar resp *types.ResetWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ResetWorkflowExecution(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCanceled(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskCompleted(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondActivityTaskFailed(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tvar resp *types.RespondCrossClusterTasksCompletedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RespondCrossClusterTasksCompleted(ctx, rp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tvar resp *types.HistoryRespondDecisionTaskCompletedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RespondDecisionTaskCompleted(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondDecisionTaskFailed(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.ScheduleDecisionTask(ctx, sp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tvar resp *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.SignalWithStartWorkflowExecution(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.SignalWorkflowExecution(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tvar resp *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.StartWorkflowExecution(ctx, hp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.SyncActivity(ctx, sp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.SyncShardStatus(ctx, sp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.TerminateWorkflowExecution(ctx, hp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n"
  },
  {
    "path": "client/wrappers/retryable/matching_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// matchingClient implements matching.Client interface instrumented with retries\ntype matchingClient struct {\n\tclient        matching.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewMatchingClient creates a new instance of matchingClient with retry policy\nfunc NewMatchingClient(client matching.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) matching.Client {\n\treturn &matchingClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\tvar resp *types.AddActivityTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.AddActivityTask(ctx, ap1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\tvar resp *types.AddDecisionTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.AddDecisionTask(ctx, ap1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.CancelOutstandingPoll(ctx, cp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tvar resp *types.DescribeTaskListResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.DescribeTaskList(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tvar resp *types.GetTaskListsByDomainResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tvar resp *types.ListTaskListPartitionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.ListTaskListPartitions(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tvar resp *types.MatchingPollForActivityTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.PollForActivityTask(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tvar resp *types.MatchingPollForDecisionTaskResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.PollForDecisionTask(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tvar resp *types.MatchingQueryWorkflowResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.QueryWorkflow(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\tvar resp *types.MatchingRefreshTaskListPartitionConfigResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.RefreshTaskListPartitionConfig(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\top := func(ctx context.Context) error {\n\t\treturn c.client.RespondQueryTaskCompleted(ctx, mp1, p1...)\n\t}\n\treturn c.throttleRetry.Do(ctx, op)\n}\n\nfunc (c *matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\tvar resp *types.MatchingUpdateTaskListPartitionConfigResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.UpdateTaskListPartitionConfig(ctx, mp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/sharddistributor_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// sharddistributorClient implements sharddistributor.Client interface instrumented with retries\ntype sharddistributorClient struct {\n\tclient        sharddistributor.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewShardDistributorClient creates a new instance of sharddistributorClient with retry policy\nfunc NewShardDistributorClient(client sharddistributor.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) sharddistributor.Client {\n\treturn &sharddistributorClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *sharddistributorClient) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest, p1 ...yarpc.CallOption) (gp2 *types.GetShardOwnerResponse, err error) {\n\tvar resp *types.GetShardOwnerResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.GetShardOwner(ctx, gp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (c *sharddistributorClient) WatchNamespaceState(ctx context.Context, wp1 *types.WatchNamespaceStateRequest, p1 ...yarpc.CallOption) (w1 sharddistributor.WatchNamespaceStateClient, err error) {\n\tvar resp sharddistributor.WatchNamespaceStateClient\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.WatchNamespaceState(ctx, wp1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/sharddistributorexecutor_generated.go",
    "content": "package retryable\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/retry.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// sharddistributorexecutorClient implements sharddistributorexecutor.Client interface instrumented with retries\ntype sharddistributorexecutorClient struct {\n\tclient        sharddistributorexecutor.Client\n\tthrottleRetry *backoff.ThrottleRetry\n}\n\n// NewShardDistributorExecutorClient creates a new instance of sharddistributorexecutorClient with retry policy\nfunc NewShardDistributorExecutorClient(client sharddistributorexecutor.Client, policy backoff.RetryPolicy, isRetryable backoff.IsRetryable) sharddistributorexecutor.Client {\n\treturn &sharddistributorexecutorClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isRetryable),\n\t\t),\n\t}\n}\n\nfunc (c *sharddistributorexecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tvar resp *types.ExecutorHeartbeatResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = c.client.Heartbeat(ctx, ep1, p1...)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n"
  },
  {
    "path": "client/wrappers/retryable/wrappers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage retryable\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestFrontendClientRetryableError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclientMock := frontend.NewMockClient(ctrl)\n\t// One failure, one success\n\tclientMock.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(nil, &types.ServiceBusyError{\n\t\t\tMessage: \"error\",\n\t\t}).Times(1)\n\tclientMock.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(nil, nil).Times(1)\n\n\tretryableClient := NewFrontendClient(\n\t\tclientMock,\n\t\tcommon.CreateFrontendServiceRetryPolicy(),\n\t\tcommon.IsServiceBusyError)\n\n\t_, err := retryableClient.CountWorkflowExecutions(context.Background(), &types.CountWorkflowExecutionsRequest{})\n\tassert.NoError(t, err)\n}\n\nfunc TestFrontendClientNonRetryableError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclientMock := frontend.NewMockClient(ctrl)\n\t// One failure, one success\n\tclientMock.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(nil, &types.BadRequestError{\n\t\t\tMessage: \"error\",\n\t\t}).Times(1)\n\n\tretryableClient := NewFrontendClient(\n\t\tclientMock,\n\t\tcommon.CreateFrontendServiceRetryPolicy(),\n\t\tcommon.IsServiceBusyError)\n\n\t_, err := retryableClient.CountWorkflowExecutions(context.Background(), &types.CountWorkflowExecutionsRequest{})\n\tassert.Error(t, err)\n}\n"
  },
  {
    "path": "client/wrappers/thrift/admin_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.AddSearchAttribute(ctx, thrift.FromAdminAddSearchAttributeRequest(ap1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.CloseShard(ctx, thrift.FromAdminCloseShardRequest(cp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n}\n\nfunc (g adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tresponse, err := g.c.DeleteWorkflow(ctx, thrift.FromAdminDeleteWorkflowRequest(ap1), p1...)\n\treturn thrift.ToAdminDeleteWorkflowResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tresponse, err := g.c.DescribeCluster(ctx, p1...)\n\treturn thrift.ToAdminDescribeClusterResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.c.DescribeHistoryHost(ctx, thrift.FromAdminDescribeHistoryHostRequest(dp1), p1...)\n\treturn thrift.ToAdminDescribeHistoryHostResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tresponse, err := g.c.DescribeQueue(ctx, thrift.FromAdminDescribeQueueRequest(dp1), p1...)\n\treturn thrift.ToAdminDescribeQueueResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tresponse, err := g.c.DescribeShardDistribution(ctx, thrift.FromAdminDescribeShardDistributionRequest(dp1), p1...)\n\treturn thrift.ToAdminDescribeShardDistributionResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, thrift.FromAdminDescribeWorkflowExecutionRequest(ap1), p1...)\n\treturn thrift.ToAdminDescribeWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDLQReplicationMessages(ctx, thrift.FromAdminGetDLQReplicationMessagesRequest(gp1), p1...)\n\treturn thrift.ToAdminGetDLQReplicationMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.c.GetDomainAsyncWorkflowConfiguraton(ctx, thrift.FromAdminGetDomainAsyncWorkflowConfiguratonRequest(request), opts...)\n\treturn thrift.ToAdminGetDomainAsyncWorkflowConfiguratonResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.GetDomainIsolationGroups(ctx, thrift.FromAdminGetDomainIsolationGroupsRequest(request), opts...)\n\treturn thrift.ToAdminGetDomainIsolationGroupsResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDomainReplicationMessages(ctx, thrift.FromAdminGetDomainReplicationMessagesRequest(gp1), p1...)\n\treturn thrift.ToAdminGetDomainReplicationMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tresponse, err := g.c.GetDynamicConfig(ctx, thrift.FromAdminGetDynamicConfigRequest(gp1), p1...)\n\treturn thrift.ToAdminGetDynamicConfigResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.GetGlobalIsolationGroups(ctx, thrift.FromAdminGetGlobalIsolationGroupsRequest(request), opts...)\n\treturn thrift.ToAdminGetGlobalIsolationGroupsResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetReplicationMessages(ctx, thrift.FromAdminGetReplicationMessagesRequest(gp1), p1...)\n\treturn thrift.ToAdminGetReplicationMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tresponse, err := g.c.GetWorkflowExecutionRawHistoryV2(ctx, thrift.FromAdminGetWorkflowExecutionRawHistoryV2Request(gp1), p1...)\n\treturn thrift.ToAdminGetWorkflowExecutionRawHistoryV2Response(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tresponse, err := g.c.ListDynamicConfig(ctx, thrift.FromAdminListDynamicConfigRequest(lp1), p1...)\n\treturn thrift.ToAdminListDynamicConfigResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tresponse, err := g.c.MaintainCorruptWorkflow(ctx, thrift.FromAdminMaintainCorruptWorkflowRequest(ap1), p1...)\n\treturn thrift.ToAdminMaintainCorruptWorkflowResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.c.MergeDLQMessages(ctx, thrift.FromAdminMergeDLQMessagesRequest(mp1), p1...)\n\treturn thrift.ToAdminMergeDLQMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.PurgeDLQMessages(ctx, thrift.FromAdminPurgeDLQMessagesRequest(pp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.c.ReadDLQMessages(ctx, thrift.FromAdminReadDLQMessagesRequest(rp1), p1...)\n\treturn thrift.ToAdminReadDLQMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ReapplyEvents(ctx, thrift.FromAdminReapplyEventsRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RefreshWorkflowTasks(ctx, thrift.FromAdminRefreshWorkflowTasksRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RemoveTask(ctx, thrift.FromAdminRemoveTaskRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ResendReplicationTasks(ctx, thrift.FromAdminResendReplicationTasksRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ResetQueue(ctx, thrift.FromAdminResetQueueRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RestoreDynamicConfig(ctx, thrift.FromAdminRestoreDynamicConfigRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.c.UpdateDomainAsyncWorkflowConfiguraton(ctx, thrift.FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(request), opts...)\n\treturn thrift.ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.UpdateDomainIsolationGroups(ctx, thrift.FromAdminUpdateDomainIsolationGroupsRequest(request), opts...)\n\treturn thrift.ToAdminUpdateDomainIsolationGroupsResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.UpdateDynamicConfig(ctx, thrift.FromAdminUpdateDynamicConfigRequest(up1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.c.UpdateGlobalIsolationGroups(ctx, thrift.FromAdminUpdateGlobalIsolationGroupsRequest(request), opts...)\n\treturn thrift.ToAdminUpdateGlobalIsolationGroupsResponse(response), thrift.ToError(err)\n}\n\nfunc (g adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n}\n"
  },
  {
    "path": "client/wrappers/thrift/constructor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/admin/adminserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/history/historyserviceclient\"\n\t\"github.com/uber/cadence/.gen/go/matching/matchingserviceclient\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n)\n\ntype (\n\tadminClient struct {\n\t\tc adminserviceclient.Interface\n\t}\n\tfrontendClient struct {\n\t\tc workflowserviceclient.Interface\n\t}\n\thistoryClient struct {\n\t\tc historyserviceclient.Interface\n\t}\n\tmatchingClient struct {\n\t\tc matchingserviceclient.Interface\n\t}\n)\n\nfunc NewAdminClient(c adminserviceclient.Interface) admin.Client {\n\treturn adminClient{c}\n}\n\nfunc NewFrontendClient(c workflowserviceclient.Interface) frontend.Client {\n\treturn frontendClient{c}\n}\n\nfunc NewHistoryClient(c historyserviceclient.Interface) history.Client {\n\treturn historyClient{c}\n}\n\nfunc NewMatchingClient(c matchingserviceclient.Interface) matching.Client {\n\treturn matchingClient{c}\n}\n"
  },
  {
    "path": "client/wrappers/thrift/frontend_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.CountWorkflowExecutions(ctx, thrift.FromCountWorkflowExecutionsRequest(cp1), p1...)\n\treturn thrift.ToCountWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.DeleteDomain(ctx, thrift.FromDeleteDomainRequest(dp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.DeprecateDomain(ctx, thrift.FromDeprecateDomainRequest(dp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tresponse, err := g.c.DescribeDomain(ctx, thrift.FromDescribeDomainRequest(dp1), p1...)\n\treturn thrift.ToDescribeDomainResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tresponse, err := g.c.DescribeTaskList(ctx, thrift.FromDescribeTaskListRequest(dp1), p1...)\n\treturn thrift.ToDescribeTaskListResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, thrift.FromDescribeWorkflowExecutionRequest(dp1), p1...)\n\treturn thrift.ToDescribeWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DiagnoseWorkflowExecution(ctx, thrift.FromDiagnoseWorkflowExecutionRequest(dp1), p1...)\n\treturn thrift.ToDiagnoseWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tresponse, err := g.c.FailoverDomain(ctx, thrift.FromFailoverDomainRequest(fp1), p1...)\n\treturn thrift.ToFailoverDomainResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tresponse, err := g.c.GetClusterInfo(ctx, p1...)\n\treturn thrift.ToGetClusterInfoResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tresponse, err := g.c.GetSearchAttributes(ctx, p1...)\n\treturn thrift.ToGetSearchAttributesResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.c.GetTaskListsByDomain(ctx, thrift.FromGetTaskListsByDomainRequest(gp1), p1...)\n\treturn thrift.ToGetTaskListsByDomainResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tresponse, err := g.c.GetWorkflowExecutionHistory(ctx, thrift.FromGetWorkflowExecutionHistoryRequest(gp1), p1...)\n\treturn thrift.ToGetWorkflowExecutionHistoryResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListArchivedWorkflowExecutions(ctx, thrift.FromListArchivedWorkflowExecutionsRequest(lp1), p1...)\n\treturn thrift.ToListArchivedWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListClosedWorkflowExecutions(ctx, thrift.FromListClosedWorkflowExecutionsRequest(lp1), p1...)\n\treturn thrift.ToListClosedWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tresponse, err := g.c.ListDomains(ctx, thrift.FromListDomainsRequest(lp1), p1...)\n\treturn thrift.ToListDomainsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tresponse, err := g.c.ListFailoverHistory(ctx, thrift.FromListFailoverHistoryRequest(lp1), p1...)\n\treturn thrift.ToListFailoverHistoryResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListOpenWorkflowExecutions(ctx, thrift.FromListOpenWorkflowExecutionsRequest(lp1), p1...)\n\treturn thrift.ToListOpenWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.c.ListTaskListPartitions(ctx, thrift.FromListTaskListPartitionsRequest(lp1), p1...)\n\treturn thrift.ToListTaskListPartitionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ListWorkflowExecutions(ctx, thrift.FromListWorkflowExecutionsRequest(lp1), p1...)\n\treturn thrift.ToListWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tresponse, err := g.c.PollForActivityTask(ctx, thrift.FromPollForActivityTaskRequest(pp1), p1...)\n\treturn thrift.ToPollForActivityTaskResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tresponse, err := g.c.PollForDecisionTask(ctx, thrift.FromPollForDecisionTaskRequest(pp1), p1...)\n\treturn thrift.ToPollForDecisionTaskResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, thrift.FromQueryWorkflowRequest(qp1), p1...)\n\treturn thrift.ToQueryWorkflowResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeat(ctx, thrift.FromRecordActivityTaskHeartbeatRequest(rp1), p1...)\n\treturn thrift.ToRecordActivityTaskHeartbeatResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeatByID(ctx, thrift.FromRecordActivityTaskHeartbeatByIDRequest(rp1), p1...)\n\treturn thrift.ToRecordActivityTaskHeartbeatByIDResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RefreshWorkflowTasks(ctx, thrift.FromRefreshWorkflowTasksRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RegisterDomain(ctx, thrift.FromRegisterDomainRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RequestCancelWorkflowExecution(ctx, thrift.FromRequestCancelWorkflowExecutionRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tresponse, err := g.c.ResetStickyTaskList(ctx, thrift.FromResetStickyTaskListRequest(rp1), p1...)\n\treturn thrift.ToResetStickyTaskListResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.ResetWorkflowExecution(ctx, thrift.FromResetWorkflowExecutionRequest(rp1), p1...)\n\treturn thrift.ToResetWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCanceled(ctx, thrift.FromRespondActivityTaskCanceledRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCanceledByID(ctx, thrift.FromRespondActivityTaskCanceledByIDRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCompleted(ctx, thrift.FromRespondActivityTaskCompletedRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCompletedByID(ctx, thrift.FromRespondActivityTaskCompletedByIDRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskFailed(ctx, thrift.FromRespondActivityTaskFailedRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskFailedByID(ctx, thrift.FromRespondActivityTaskFailedByIDRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.c.RespondDecisionTaskCompleted(ctx, thrift.FromRespondDecisionTaskCompletedRequest(rp1), p1...)\n\treturn thrift.ToRespondDecisionTaskCompletedResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondDecisionTaskFailed(ctx, thrift.FromRespondDecisionTaskFailedRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondQueryTaskCompleted(ctx, thrift.FromRespondQueryTaskCompletedRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.RestartWorkflowExecution(ctx, thrift.FromRestartWorkflowExecutionRequest(rp1), p1...)\n\treturn thrift.ToRestartWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.c.ScanWorkflowExecutions(ctx, thrift.FromScanWorkflowExecutionsRequest(lp1), p1...)\n\treturn thrift.ToScanWorkflowExecutionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecution(ctx, thrift.FromSignalWithStartWorkflowExecutionRequest(sp1), p1...)\n\treturn thrift.ToSignalWithStartWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecutionAsync(ctx, thrift.FromSignalWithStartWorkflowExecutionAsyncRequest(sp1), p1...)\n\treturn thrift.ToSignalWithStartWorkflowExecutionAsyncResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.SignalWorkflowExecution(ctx, thrift.FromSignalWorkflowExecutionRequest(sp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecution(ctx, thrift.FromStartWorkflowExecutionRequest(sp1), p1...)\n\treturn thrift.ToStartWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecutionAsync(ctx, thrift.FromStartWorkflowExecutionAsyncRequest(sp1), p1...)\n\treturn thrift.ToStartWorkflowExecutionAsyncResponse(response), thrift.ToError(err)\n}\n\nfunc (g frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.TerminateWorkflowExecution(ctx, thrift.FromTerminateWorkflowExecutionRequest(tp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tresponse, err := g.c.UpdateDomain(ctx, thrift.FromUpdateDomainRequest(up1), p1...)\n\treturn thrift.ToUpdateDomainResponse(response), thrift.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/thrift/history_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.CloseShard(ctx, thrift.FromHistoryCloseShardRequest(cp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n}\n\nfunc (g historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.c.DescribeHistoryHost(ctx, thrift.FromHistoryDescribeHistoryHostRequest(dp1), p1...)\n\treturn thrift.ToHistoryDescribeHistoryHostResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tresponse, err := g.c.DescribeMutableState(ctx, thrift.FromHistoryDescribeMutableStateRequest(dp1), p1...)\n\treturn thrift.ToHistoryDescribeMutableStateResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tresponse, err := g.c.DescribeQueue(ctx, thrift.FromHistoryDescribeQueueRequest(dp1), p1...)\n\treturn thrift.ToHistoryDescribeQueueResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.DescribeWorkflowExecution(ctx, thrift.FromHistoryDescribeWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToHistoryDescribeWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tresponse, err := g.c.GetCrossClusterTasks(ctx, thrift.FromHistoryGetCrossClusterTasksRequest(gp1), p1...)\n\treturn thrift.ToHistoryGetCrossClusterTasksResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetDLQReplicationMessages(ctx, thrift.FromHistoryGetDLQReplicationMessagesRequest(gp1), p1...)\n\treturn thrift.ToHistoryGetDLQReplicationMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\tresponse, err := g.c.GetFailoverInfo(ctx, thrift.FromHistoryGetFailoverInfoRequest(gp1), p1...)\n\treturn thrift.ToHistoryGetFailoverInfoResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tresponse, err := g.c.GetMutableState(ctx, thrift.FromHistoryGetMutableStateRequest(gp1), p1...)\n\treturn thrift.ToHistoryGetMutableStateResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.c.GetReplicationMessages(ctx, thrift.FromHistoryGetReplicationMessagesRequest(gp1), p1...)\n\treturn thrift.ToHistoryGetReplicationMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.c.MergeDLQMessages(ctx, thrift.FromHistoryMergeDLQMessagesRequest(mp1), p1...)\n\treturn thrift.ToHistoryMergeDLQMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.NotifyFailoverMarkers(ctx, thrift.FromHistoryNotifyFailoverMarkersRequest(np1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tresponse, err := g.c.PollMutableState(ctx, thrift.FromHistoryPollMutableStateRequest(pp1), p1...)\n\treturn thrift.ToHistoryPollMutableStateResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.PurgeDLQMessages(ctx, thrift.FromHistoryPurgeDLQMessagesRequest(pp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, thrift.FromHistoryQueryWorkflowRequest(hp1), p1...)\n\treturn thrift.ToHistoryQueryWorkflowResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tresponse, err := g.c.RatelimitUpdate(ctx, thrift.FromHistoryRatelimitUpdateRequest(request), opts...)\n\treturn thrift.ToHistoryRatelimitUpdateResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.c.ReadDLQMessages(ctx, thrift.FromHistoryReadDLQMessagesRequest(rp1), p1...)\n\treturn thrift.ToHistoryReadDLQMessagesResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ReapplyEvents(ctx, thrift.FromHistoryReapplyEventsRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskHeartbeat(ctx, thrift.FromHistoryRecordActivityTaskHeartbeatRequest(hp1), p1...)\n\treturn thrift.ToHistoryRecordActivityTaskHeartbeatResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tresponse, err := g.c.RecordActivityTaskStarted(ctx, thrift.FromHistoryRecordActivityTaskStartedRequest(rp1), p1...)\n\treturn thrift.ToHistoryRecordActivityTaskStartedResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RecordChildExecutionCompleted(ctx, thrift.FromHistoryRecordChildExecutionCompletedRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tresponse, err := g.c.RecordDecisionTaskStarted(ctx, thrift.FromHistoryRecordDecisionTaskStartedRequest(rp1), p1...)\n\treturn thrift.ToHistoryRecordDecisionTaskStartedResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RefreshWorkflowTasks(ctx, thrift.FromHistoryRefreshWorkflowTasksRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RemoveSignalMutableState(ctx, thrift.FromHistoryRemoveSignalMutableStateRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RemoveTask(ctx, thrift.FromHistoryRemoveTaskRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ReplicateEventsV2(ctx, thrift.FromHistoryReplicateEventsV2Request(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RequestCancelWorkflowExecution(ctx, thrift.FromHistoryRequestCancelWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ResetQueue(ctx, thrift.FromHistoryResetQueueRequest(rp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tresponse, err := g.c.ResetStickyTaskList(ctx, thrift.FromHistoryResetStickyTaskListRequest(hp1), p1...)\n\treturn thrift.ToHistoryResetStickyTaskListResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.ResetWorkflowExecution(ctx, thrift.FromHistoryResetWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToHistoryResetWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCanceled(ctx, thrift.FromHistoryRespondActivityTaskCanceledRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskCompleted(ctx, thrift.FromHistoryRespondActivityTaskCompletedRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondActivityTaskFailed(ctx, thrift.FromHistoryRespondActivityTaskFailedRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tresponse, err := g.c.RespondCrossClusterTasksCompleted(ctx, thrift.FromHistoryRespondCrossClusterTasksCompletedRequest(rp1), p1...)\n\treturn thrift.ToHistoryRespondCrossClusterTasksCompletedResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.c.RespondDecisionTaskCompleted(ctx, thrift.FromHistoryRespondDecisionTaskCompletedRequest(hp1), p1...)\n\treturn thrift.ToHistoryRespondDecisionTaskCompletedResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondDecisionTaskFailed(ctx, thrift.FromHistoryRespondDecisionTaskFailedRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.ScheduleDecisionTask(ctx, thrift.FromHistoryScheduleDecisionTaskRequest(sp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.SignalWithStartWorkflowExecution(ctx, thrift.FromHistorySignalWithStartWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToHistorySignalWithStartWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.SignalWorkflowExecution(ctx, thrift.FromHistorySignalWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.c.StartWorkflowExecution(ctx, thrift.FromHistoryStartWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToHistoryStartWorkflowExecutionResponse(response), thrift.ToError(err)\n}\n\nfunc (g historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.SyncActivity(ctx, thrift.FromHistorySyncActivityRequest(sp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.SyncShardStatus(ctx, thrift.FromHistorySyncShardStatusRequest(sp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.TerminateWorkflowExecution(ctx, thrift.FromHistoryTerminateWorkflowExecutionRequest(hp1), p1...)\n\treturn thrift.ToError(err)\n}\n"
  },
  {
    "path": "client/wrappers/thrift/matching_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\terr = g.c.AddActivityTask(ctx, thrift.FromMatchingAddActivityTaskRequest(ap1), p1...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.AddActivityTaskResponse{}, nil\n}\n\nfunc (g matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\terr = g.c.AddDecisionTask(ctx, thrift.FromMatchingAddDecisionTaskRequest(ap1), p1...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.AddDecisionTaskResponse{}, nil\n}\n\nfunc (g matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.CancelOutstandingPoll(ctx, thrift.FromMatchingCancelOutstandingPollRequest(cp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tresponse, err := g.c.DescribeTaskList(ctx, thrift.FromMatchingDescribeTaskListRequest(mp1), p1...)\n\treturn thrift.ToMatchingDescribeTaskListResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.c.GetTaskListsByDomain(ctx, thrift.FromMatchingGetTaskListsByDomainRequest(gp1), p1...)\n\treturn thrift.ToMatchingGetTaskListsByDomainResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.c.ListTaskListPartitions(ctx, thrift.FromMatchingListTaskListPartitionsRequest(mp1), p1...)\n\treturn thrift.ToMatchingListTaskListPartitionsResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tresponse, err := g.c.PollForActivityTask(ctx, thrift.FromMatchingPollForActivityTaskRequest(mp1), p1...)\n\treturn thrift.ToMatchingPollForActivityTaskResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tresponse, err := g.c.PollForDecisionTask(ctx, thrift.FromMatchingPollForDecisionTaskRequest(mp1), p1...)\n\treturn thrift.ToMatchingPollForDecisionTaskResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tresponse, err := g.c.QueryWorkflow(ctx, thrift.FromMatchingQueryWorkflowRequest(mp1), p1...)\n\treturn thrift.ToMatchingQueryWorkflowResponse(response), thrift.ToError(err)\n}\n\nfunc (g matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n}\n\nfunc (g matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\terr = g.c.RespondQueryTaskCompleted(ctx, thrift.FromMatchingRespondQueryTaskCompletedRequest(mp1), p1...)\n\treturn thrift.ToError(err)\n}\n\nfunc (g matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\treturn nil, thrift.ToError(&types.BadRequestError{Message: \"Feature not supported on TChannel\"})\n}\n"
  },
  {
    "path": "client/wrappers/timeout/admin_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ admin.Client = (*adminClient)(nil)\n\n// adminClient implements the admin.Client interface instrumented with timeouts\ntype adminClient struct {\n\tclient       admin.Client\n\tlargeTimeout time.Duration\n\ttimeout      time.Duration\n}\n\n// NewAdminClient creates a new adminClient instance\nfunc NewAdminClient(\n\tclient admin.Client,\n\tlargeTimeout time.Duration,\n\ttimeout time.Duration,\n) admin.Client {\n\treturn &adminClient{\n\t\tclient:       client,\n\t\tlargeTimeout: largeTimeout,\n\t\ttimeout:      timeout,\n\t}\n}\n\nfunc (c *adminClient) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.AddSearchAttribute(ctx, ap1, p1...)\n}\n\nfunc (c *adminClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.CloseShard(ctx, cp1, p1...)\n}\n\nfunc (c *adminClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.CountDLQMessages(ctx, cp1, p1...)\n}\n\nfunc (c *adminClient) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DeleteWorkflow(ctx, ap1, p1...)\n}\n\nfunc (c *adminClient) DescribeCluster(ctx context.Context, p1 ...yarpc.CallOption) (dp1 *types.DescribeClusterResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeCluster(ctx, p1...)\n}\n\nfunc (c *adminClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeHistoryHost(ctx, dp1, p1...)\n}\n\nfunc (c *adminClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeQueue(ctx, dp1, p1...)\n}\n\nfunc (c *adminClient) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeShardDistribution(ctx, dp1, p1...)\n}\n\nfunc (c *adminClient) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeWorkflowExecution(ctx, ap1, p1...)\n}\n\nfunc (c *adminClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n}\n\nfunc (c *adminClient) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n}\n\nfunc (c *adminClient) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetDomainIsolationGroups(ctx, request, opts...)\n}\n\nfunc (c *adminClient) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetDomainReplicationMessages(ctx, gp1, p1...)\n}\n\nfunc (c *adminClient) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetDynamicConfig(ctx, gp1, p1...)\n}\n\nfunc (c *adminClient) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetGlobalIsolationGroups(ctx, request, opts...)\n}\n\nfunc (c *adminClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.largeTimeout)\n\tdefer cancel()\n\treturn c.client.GetReplicationMessages(ctx, gp1, p1...)\n}\n\nfunc (c *adminClient) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetWorkflowExecutionRawHistoryV2(ctx, gp1, p1...)\n}\n\nfunc (c *adminClient) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListDynamicConfig(ctx, lp1, p1...)\n}\n\nfunc (c *adminClient) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest, p1 ...yarpc.CallOption) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.MaintainCorruptWorkflow(ctx, ap1, p1...)\n}\n\nfunc (c *adminClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.MergeDLQMessages(ctx, mp1, p1...)\n}\n\nfunc (c *adminClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.PurgeDLQMessages(ctx, pp1, p1...)\n}\n\nfunc (c *adminClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ReadDLQMessages(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ReapplyEvents(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RemoveTask(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResendReplicationTasks(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetQueue(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RestoreDynamicConfig(ctx, rp1, p1...)\n}\n\nfunc (c *adminClient) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateDomainAsyncWorkflowConfiguraton(ctx, request, opts...)\n}\n\nfunc (c *adminClient) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateDomainIsolationGroups(ctx, request, opts...)\n}\n\nfunc (c *adminClient) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateDynamicConfig(ctx, up1, p1...)\n}\n\nfunc (c *adminClient) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest, opts ...yarpc.CallOption) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateGlobalIsolationGroups(ctx, request, opts...)\n}\n\nfunc (c *adminClient) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest, opts ...yarpc.CallOption) (up1 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateTaskListPartitionConfig(ctx, request, opts...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/frontend_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ frontend.Client = (*frontendClient)(nil)\n\n// frontendClient implements the frontend.Client interface instrumented with timeouts\ntype frontendClient struct {\n\tclient          frontend.Client\n\tlongPollTimeout time.Duration\n\ttimeout         time.Duration\n}\n\n// NewFrontendClient creates a new frontendClient instance\nfunc NewFrontendClient(\n\tclient frontend.Client,\n\tlongPollTimeout time.Duration,\n\ttimeout time.Duration,\n) frontend.Client {\n\treturn &frontendClient{\n\t\tclient:          client,\n\t\tlongPollTimeout: longPollTimeout,\n\t\ttimeout:         timeout,\n\t}\n}\n\nfunc (c *frontendClient) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.CountWorkflowExecutions(ctx, cp1, p1...)\n}\n\nfunc (c *frontendClient) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DeleteDomain(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DeprecateDomain(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeDomainResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeDomain(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeTaskListResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeTaskList(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeWorkflowExecution(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DiagnoseWorkflowExecution(ctx, dp1, p1...)\n}\n\nfunc (c *frontendClient) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest, p1 ...yarpc.CallOption) (fp2 *types.FailoverDomainResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.FailoverDomain(ctx, fp1, p1...)\n}\n\nfunc (c *frontendClient) GetClusterInfo(ctx context.Context, p1 ...yarpc.CallOption) (cp1 *types.ClusterInfo, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetClusterInfo(ctx, p1...)\n}\n\nfunc (c *frontendClient) GetSearchAttributes(ctx context.Context, p1 ...yarpc.CallOption) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetSearchAttributes(ctx, p1...)\n}\n\nfunc (c *frontendClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n}\n\nfunc (c *frontendClient) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest, p1 ...yarpc.CallOption) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetWorkflowExecutionHistory(ctx, gp1, p1...)\n}\n\nfunc (c *frontendClient) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.longPollTimeout)\n\tdefer cancel()\n\treturn c.client.ListArchivedWorkflowExecutions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListClosedWorkflowExecutions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListDomainsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListDomains(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest, p1 ...yarpc.CallOption) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListFailoverHistory(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListOpenWorkflowExecutions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListTaskListPartitions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListWorkflowExecutions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.longPollTimeout)\n\tdefer cancel()\n\treturn c.client.PollForActivityTask(ctx, pp1, p1...)\n}\n\nfunc (c *frontendClient) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest, p1 ...yarpc.CallOption) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.longPollTimeout)\n\tdefer cancel()\n\treturn c.client.PollForDecisionTask(ctx, pp1, p1...)\n}\n\nfunc (c *frontendClient) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest, p1 ...yarpc.CallOption) (qp2 *types.QueryWorkflowResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.QueryWorkflow(ctx, qp1, p1...)\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordActivityTaskHeartbeat(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordActivityTaskHeartbeatByID(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RefreshWorkflowTasks(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RegisterDomain(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RequestCancelWorkflowExecution(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetStickyTaskList(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetWorkflowExecution(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCanceled(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCanceledByID(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCompleted(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCompletedByID(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskFailed(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskFailedByID(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondDecisionTaskCompleted(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondDecisionTaskFailed(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondQueryTaskCompleted(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RestartWorkflowExecution(ctx, rp1, p1...)\n}\n\nfunc (c *frontendClient) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest, p1 ...yarpc.CallOption) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ScanWorkflowExecutions(ctx, lp1, p1...)\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SignalWithStartWorkflowExecution(ctx, sp1, p1...)\n}\n\nfunc (c *frontendClient) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SignalWithStartWorkflowExecutionAsync(ctx, sp1, p1...)\n}\n\nfunc (c *frontendClient) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SignalWorkflowExecution(ctx, sp1, p1...)\n}\n\nfunc (c *frontendClient) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.StartWorkflowExecution(ctx, sp1, p1...)\n}\n\nfunc (c *frontendClient) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest, p1 ...yarpc.CallOption) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.StartWorkflowExecutionAsync(ctx, sp1, p1...)\n}\n\nfunc (c *frontendClient) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.TerminateWorkflowExecution(ctx, tp1, p1...)\n}\n\nfunc (c *frontendClient) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest, p1 ...yarpc.CallOption) (up2 *types.UpdateDomainResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateDomain(ctx, up1, p1...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/history_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ history.Client = (*historyClient)(nil)\n\n// historyClient implements the history.Client interface instrumented with timeouts\ntype historyClient struct {\n\tclient  history.Client\n\ttimeout time.Duration\n}\n\n// NewHistoryClient creates a new historyClient instance\nfunc NewHistoryClient(\n\tclient history.Client,\n\ttimeout time.Duration,\n) history.Client {\n\treturn &historyClient{\n\t\tclient:  client,\n\t\ttimeout: timeout,\n\t}\n}\n\nfunc (c *historyClient) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.CloseShard(ctx, cp1, p1...)\n}\n\nfunc (c *historyClient) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest, p1 ...yarpc.CallOption) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\treturn c.client.CountDLQMessages(ctx, cp1, p1...)\n}\n\nfunc (c *historyClient) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeHistoryHost(ctx, dp1, p1...)\n}\n\nfunc (c *historyClient) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeMutableStateResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeMutableState(ctx, dp1, p1...)\n}\n\nfunc (c *historyClient) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest, p1 ...yarpc.CallOption) (dp2 *types.DescribeQueueResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeQueue(ctx, dp1, p1...)\n}\n\nfunc (c *historyClient) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest, p1 ...yarpc.CallOption) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\treturn c.client.GetCrossClusterTasks(ctx, gp1, p1...)\n}\n\nfunc (c *historyClient) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\treturn c.client.GetDLQReplicationMessages(ctx, gp1, p1...)\n}\n\nfunc (c *historyClient) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest, p1 ...yarpc.CallOption) (gp2 *types.GetFailoverInfoResponse, err error) {\n\treturn c.client.GetFailoverInfo(ctx, gp1, p1...)\n}\n\nfunc (c *historyClient) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest, p1 ...yarpc.CallOption) (gp2 *types.GetMutableStateResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetMutableState(ctx, gp1, p1...)\n}\n\nfunc (c *historyClient) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest, p1 ...yarpc.CallOption) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\treturn c.client.GetReplicationMessages(ctx, gp1, p1...)\n}\n\nfunc (c *historyClient) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest, p1 ...yarpc.CallOption) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\treturn c.client.MergeDLQMessages(ctx, mp1, p1...)\n}\n\nfunc (c *historyClient) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.NotifyFailoverMarkers(ctx, np1, p1...)\n}\n\nfunc (c *historyClient) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest, p1 ...yarpc.CallOption) (pp2 *types.PollMutableStateResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.PollMutableState(ctx, pp1, p1...)\n}\n\nfunc (c *historyClient) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest, p1 ...yarpc.CallOption) (err error) {\n\treturn c.client.PurgeDLQMessages(ctx, pp1, p1...)\n}\n\nfunc (c *historyClient) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.QueryWorkflow(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RatelimitUpdate(ctx context.Context, request *types.RatelimitUpdateRequest, opts ...yarpc.CallOption) (rp1 *types.RatelimitUpdateResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RatelimitUpdate(ctx, request, opts...)\n}\n\nfunc (c *historyClient) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest, p1 ...yarpc.CallOption) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\treturn c.client.ReadDLQMessages(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ReapplyEvents(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest, p1 ...yarpc.CallOption) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordActivityTaskHeartbeat(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordActivityTaskStarted(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordChildExecutionCompleted(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest, p1 ...yarpc.CallOption) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RecordDecisionTaskStarted(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RefreshWorkflowTasks(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RemoveSignalMutableState(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RemoveTask(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ReplicateEventsV2(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RequestCancelWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetQueue(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetStickyTaskList(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest, p1 ...yarpc.CallOption) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ResetWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCanceled(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskCompleted(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondActivityTaskFailed(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest, p1 ...yarpc.CallOption) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondCrossClusterTasksCompleted(ctx, rp1, p1...)\n}\n\nfunc (c *historyClient) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest, p1 ...yarpc.CallOption) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondDecisionTaskCompleted(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondDecisionTaskFailed(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ScheduleDecisionTask(ctx, sp1, p1...)\n}\n\nfunc (c *historyClient) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SignalWithStartWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SignalWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest, p1 ...yarpc.CallOption) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.StartWorkflowExecution(ctx, hp1, p1...)\n}\n\nfunc (c *historyClient) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SyncActivity(ctx, sp1, p1...)\n}\n\nfunc (c *historyClient) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.SyncShardStatus(ctx, sp1, p1...)\n}\n\nfunc (c *historyClient) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.TerminateWorkflowExecution(ctx, hp1, p1...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/history_generated_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc Test_historyClient_CloseShard(t *testing.T) {\n\ttype fields struct {\n\t\tclient  func(t *testing.T) history.Client\n\t\ttimeout time.Duration\n\t}\n\ttype args struct {\n\t\tctx context.Context\n\t\tcp1 *types.CloseShardRequest\n\t\tp1  []yarpc.CallOption\n\t}\n\ttests := []struct {\n\t\tname    string\n\t\tfields  fields\n\t\targs    args\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"nil context success\",\n\t\t\tfields: fields{\n\t\t\t\ttimeout: time.Millisecond * 150,\n\t\t\t},\n\t\t\targs: args{\n\t\t\t\tctx: nil,\n\t\t\t\tcp1: &types.CloseShardRequest{},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"context failed with timeout\",\n\t\t\tfields: fields{\n\t\t\t\ttimeout: time.Millisecond * 50,\n\t\t\t},\n\t\t\targs: args{\n\t\t\t\tctx: context.Background(),\n\t\t\t\tcp1: &types.CloseShardRequest{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid timeout value success\",\n\t\t\tfields: fields{\n\t\t\t\ttimeout: -10,\n\t\t\t},\n\t\t\targs: args{\n\t\t\t\tctx: context.Background(),\n\t\t\t\tcp1: &types.CloseShardRequest{},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tm := history.NewMockClient(gomock.NewController(t))\n\t\t\tm.EXPECT().CloseShard(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, r *types.CloseShardRequest, opt ...yarpc.CallOption) error {\n\t\t\t\tfor {\n\t\t\t\t\tselect {\n\t\t\t\t\tcase <-ctx.Done():\n\t\t\t\t\t\treturn ctx.Err()\n\t\t\t\t\tcase <-time.After(time.Millisecond * 100):\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t})\n\t\t\tc := historyClient{\n\t\t\t\tclient:  m,\n\t\t\t\ttimeout: tt.fields.timeout,\n\t\t\t}\n\t\t\tif err := c.CloseShard(tt.args.ctx, tt.args.cp1, tt.args.p1...); (err != nil) != tt.wantErr {\n\t\t\t\tt.Errorf(\"CloseShard() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_historyClient_GetReplicationMessages(t *testing.T) {\n\tt.Run(\"no timeout\", func(t *testing.T) {\n\t\tm := history.NewMockClient(gomock.NewController(t))\n\t\tm.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, r *types.GetReplicationMessagesRequest, opt ...yarpc.CallOption) (*types.GetReplicationMessagesResponse, error) {\n\t\t\tfor {\n\t\t\t\tselect {\n\t\t\t\tcase <-ctx.Done():\n\t\t\t\t\treturn nil, ctx.Err()\n\t\t\t\tcase <-time.After(time.Millisecond * 100):\n\t\t\t\t\treturn &types.GetReplicationMessagesResponse{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t\tc := historyClient{\n\t\t\tclient:  m,\n\t\t\ttimeout: time.Millisecond * 10,\n\t\t}\n\t\t_, err := c.GetReplicationMessages(context.Background(), &types.GetReplicationMessagesRequest{})\n\t\tassert.NoError(t, err)\n\t})\n}\n"
  },
  {
    "path": "client/wrappers/timeout/matching_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ matching.Client = (*matchingClient)(nil)\n\n// matchingClient implements the matching.Client interface instrumented with timeouts\ntype matchingClient struct {\n\tclient          matching.Client\n\tlongPollTimeout time.Duration\n\ttimeout         time.Duration\n}\n\n// NewMatchingClient creates a new matchingClient instance\nfunc NewMatchingClient(\n\tclient matching.Client,\n\tlongPollTimeout time.Duration,\n\ttimeout time.Duration,\n) matching.Client {\n\treturn &matchingClient{\n\t\tclient:          client,\n\t\tlongPollTimeout: longPollTimeout,\n\t\ttimeout:         timeout,\n\t}\n}\n\nfunc (c *matchingClient) AddActivityTask(ctx context.Context, ap1 *types.AddActivityTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddActivityTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.AddActivityTask(ctx, ap1, p1...)\n}\n\nfunc (c *matchingClient) AddDecisionTask(ctx context.Context, ap1 *types.AddDecisionTaskRequest, p1 ...yarpc.CallOption) (ap2 *types.AddDecisionTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.AddDecisionTask(ctx, ap1, p1...)\n}\n\nfunc (c *matchingClient) CancelOutstandingPoll(ctx context.Context, cp1 *types.CancelOutstandingPollRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.CancelOutstandingPoll(ctx, cp1, p1...)\n}\n\nfunc (c *matchingClient) DescribeTaskList(ctx context.Context, mp1 *types.MatchingDescribeTaskListRequest, p1 ...yarpc.CallOption) (dp1 *types.DescribeTaskListResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.DescribeTaskList(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest, p1 ...yarpc.CallOption) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\treturn c.client.GetTaskListsByDomain(ctx, gp1, p1...)\n}\n\nfunc (c *matchingClient) ListTaskListPartitions(ctx context.Context, mp1 *types.MatchingListTaskListPartitionsRequest, p1 ...yarpc.CallOption) (lp1 *types.ListTaskListPartitionsResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.ListTaskListPartitions(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) PollForActivityTask(ctx context.Context, mp1 *types.MatchingPollForActivityTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForActivityTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.longPollTimeout)\n\tdefer cancel()\n\treturn c.client.PollForActivityTask(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) PollForDecisionTask(ctx context.Context, mp1 *types.MatchingPollForDecisionTaskRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingPollForDecisionTaskResponse, err error) {\n\tctx, cancel := createContext(ctx, c.longPollTimeout)\n\tdefer cancel()\n\treturn c.client.PollForDecisionTask(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) QueryWorkflow(ctx context.Context, mp1 *types.MatchingQueryWorkflowRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingQueryWorkflowResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.QueryWorkflow(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) RefreshTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingRefreshTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingRefreshTaskListPartitionConfigResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RefreshTaskListPartitionConfig(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) RespondQueryTaskCompleted(ctx context.Context, mp1 *types.MatchingRespondQueryTaskCompletedRequest, p1 ...yarpc.CallOption) (err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.RespondQueryTaskCompleted(ctx, mp1, p1...)\n}\n\nfunc (c *matchingClient) UpdateTaskListPartitionConfig(ctx context.Context, mp1 *types.MatchingUpdateTaskListPartitionConfigRequest, p1 ...yarpc.CallOption) (mp2 *types.MatchingUpdateTaskListPartitionConfigResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.UpdateTaskListPartitionConfig(ctx, mp1, p1...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/sharddistributor_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ sharddistributor.Client = (*sharddistributorClient)(nil)\n\n// sharddistributorClient implements the sharddistributor.Client interface instrumented with timeouts\ntype sharddistributorClient struct {\n\tclient  sharddistributor.Client\n\ttimeout time.Duration\n}\n\n// NewShardDistributorClient creates a new sharddistributorClient instance\nfunc NewShardDistributorClient(\n\tclient sharddistributor.Client,\n\ttimeout time.Duration,\n) sharddistributor.Client {\n\treturn &sharddistributorClient{\n\t\tclient:  client,\n\t\ttimeout: timeout,\n\t}\n}\n\nfunc (c *sharddistributorClient) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest, p1 ...yarpc.CallOption) (gp2 *types.GetShardOwnerResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.GetShardOwner(ctx, gp1, p1...)\n}\n\nfunc (c *sharddistributorClient) WatchNamespaceState(ctx context.Context, wp1 *types.WatchNamespaceStateRequest, p1 ...yarpc.CallOption) (w1 sharddistributor.WatchNamespaceStateClient, err error) {\n\treturn c.client.WatchNamespaceState(ctx, wp1, p1...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/sharddistributorexecutor_generated.go",
    "content": "package timeout\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/timeout.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ sharddistributorexecutor.Client = (*sharddistributorexecutorClient)(nil)\n\n// sharddistributorexecutorClient implements the sharddistributorexecutor.Client interface instrumented with timeouts\ntype sharddistributorexecutorClient struct {\n\tclient  sharddistributorexecutor.Client\n\ttimeout time.Duration\n}\n\n// NewShardDistributorExecutorClient creates a new sharddistributorexecutorClient instance\nfunc NewShardDistributorExecutorClient(\n\tclient sharddistributorexecutor.Client,\n\ttimeout time.Duration,\n) sharddistributorexecutor.Client {\n\treturn &sharddistributorexecutorClient{\n\t\tclient:  client,\n\t\ttimeout: timeout,\n\t}\n}\n\nfunc (c *sharddistributorexecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tctx, cancel := createContext(ctx, c.timeout)\n\tdefer cancel()\n\treturn c.client.Heartbeat(ctx, ep1, p1...)\n}\n"
  },
  {
    "path": "client/wrappers/timeout/timeout.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst (\n\t// AdminDefaultTimeout is the admin service default timeout used to make calls\n\tAdminDefaultTimeout = 10 * time.Second\n\t// AdminDefaultLargeTimeout is the admin service large default timeout used to make calls\n\tAdminDefaultLargeTimeout = time.Minute\n\t// FrontendDefaultTimeout is the frontend service default timeout used to make calls\n\tFrontendDefaultTimeout = 10 * time.Second\n\t// FrontendDefaultLongPollTimeout is the frontend service long poll default timeout used to make calls\n\tFrontendDefaultLongPollTimeout = time.Minute * 3\n\t// MatchingDefaultTimeout is the default timeout used to make calls\n\tMatchingDefaultTimeout = time.Minute\n\t// MatchingDefaultLongPollTimeout is the long poll default timeout used to make calls\n\tMatchingDefaultLongPollTimeout = time.Minute * 2\n\t// HistoryDefaultTimeout is the default timeout used to make calls\n\tHistoryDefaultTimeout = time.Second * 30\n\t// ShardDistributorDefaultTimeout is the default timeout used to make calls\n\tShardDistributorDefaultTimeout = time.Second * 10\n\t// ShardDistributorExecutorDefaultTimeout is the default timeout used to make calls\n\tShardDistributorExecutorDefaultTimeout = time.Second * 10\n)\n\nfunc createContext(parent context.Context, timeout time.Duration) (context.Context, context.CancelFunc) {\n\tif parent == nil {\n\t\tparent = context.Background()\n\t}\n\tif timeout > 0 {\n\t\treturn context.WithTimeout(parent, timeout)\n\t}\n\treturn context.WithCancel(parent)\n}\n"
  },
  {
    "path": "cmd/bench/main.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage main\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"path\"\n\t\"strings\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/bench\"\n\t\"github.com/uber/cadence/bench/lib\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tflagRoot   = \"root\"\n\tflagConfig = \"config\"\n\tflagEnv    = \"env\"\n\tflagZone   = \"zone\"\n)\n\nconst (\n\tdefaultRoot   = \".\"\n\tdefaultConfig = \"config/bench\"\n\tdefaultEnv    = \"development\"\n\tdefaultZone   = \"\"\n)\n\nfunc startHandler(c *cli.Context) error {\n\tenv := getEnvironment(c)\n\tzone := getZone(c)\n\tconfigDir := getConfigDir(c)\n\n\tlog.Printf(\"Loading config; env=%v,zone=%v,configDir=%v\\n\", env, zone, configDir)\n\n\tvar cfg lib.Config\n\tif err := config.Load(env, configDir, zone, &cfg); err != nil {\n\t\treturn fmt.Errorf(\"failed to load config file: %w\", err)\n\t}\n\n\tif err := cfg.Validate(); err != nil {\n\t\treturn fmt.Errorf(\"invalid config: %w\", err)\n\t}\n\n\tbenchWorker, err := bench.NewWorker(&cfg)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to initialize bench worker: %w\", err)\n\t}\n\n\tif err := benchWorker.Run(); err != nil {\n\t\treturn fmt.Errorf(\"failed to run bench worker: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc getRootDir(c *cli.Context) string {\n\trootDir := c.String(flagRoot)\n\tif len(rootDir) == 0 {\n\t\tvar err error\n\t\tif rootDir, err = os.Getwd(); err != nil {\n\t\t\trootDir = \".\"\n\t\t}\n\t}\n\treturn rootDir\n}\n\nfunc getConfigDir(c *cli.Context) string {\n\trootDir := getRootDir(c)\n\tconfigDir := c.String(flagConfig)\n\treturn path.Join(rootDir, configDir)\n}\n\nfunc getEnvironment(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(flagEnv))\n}\n\nfunc getZone(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(flagZone))\n}\n\nfunc buildCLI() *cli.App {\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence-bench\"\n\tapp.Usage = \"Cadence bench\"\n\tapp.Version = \"0.0.1\"\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    flagRoot,\n\t\t\tAliases: []string{\"r\"},\n\t\t\tValue:   defaultRoot,\n\t\t\tUsage:   \"root directory of execution environment\",\n\t\t\tEnvVars: []string{lib.EnvKeyRoot},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    flagConfig,\n\t\t\tAliases: []string{\"c\"},\n\t\t\tValue:   defaultConfig,\n\t\t\tUsage:   \"config dir path relative to root\",\n\t\t\tEnvVars: []string{lib.EnvKeyConfigDir},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    flagEnv,\n\t\t\tAliases: []string{\"e\"},\n\t\t\tValue:   defaultEnv,\n\t\t\tUsage:   \"runtime environment\",\n\t\t\tEnvVars: []string{lib.EnvKeyEnvironment},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    flagZone,\n\t\t\tAliases: []string{\"z\"},\n\t\t\tValue:   defaultZone,\n\t\t\tUsage:   \"availability zone\",\n\t\t\tEnvVars: []string{lib.EnvKeyAvailabilityZone},\n\t\t},\n\t}\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:  \"start\",\n\t\t\tUsage: \"start cadence bench worker\",\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn startHandler(c)\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n\nfunc main() {\n\tapp := buildCLI()\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n"
  },
  {
    "path": "cmd/canary/main.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"path\"\n\t\"strings\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/canary\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nfunc startHandler(c *cli.Context) error {\n\tenv := getEnvironment(c)\n\tzone := getZone(c)\n\tconfigDir := getConfigDir(c)\n\n\tlog.Printf(\"Loading config; env=%v,zone=%v,configDir=%v\\n\", env, zone, configDir)\n\n\tvar cfg canary.Config\n\tif err := config.Load(env, configDir, zone, &cfg); err != nil {\n\t\treturn fmt.Errorf(\"failed to load config file: %w\", err)\n\t}\n\n\tif err := cfg.Validate(); err != nil {\n\t\treturn fmt.Errorf(\"invalid config: %w\", err)\n\t}\n\n\tmode := c.String(\"mode\")\n\tcanary, err := canary.NewCanaryRunner(&cfg)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to initialize canary: %w\", err)\n\t}\n\n\tif err := canary.Run(mode); err != nil {\n\t\treturn fmt.Errorf(\"failed to run canary: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc getRootDir(c *cli.Context) string {\n\trootDir := c.String(\"root\")\n\tif len(rootDir) == 0 {\n\t\tvar err error\n\t\tif rootDir, err = os.Getwd(); err != nil {\n\t\t\trootDir = \".\"\n\t\t}\n\t}\n\treturn rootDir\n}\n\nfunc getConfigDir(c *cli.Context) string {\n\trootDir := getRootDir(c)\n\tconfigDir := c.String(\"config\")\n\treturn path.Join(rootDir, configDir)\n}\n\nfunc getEnvironment(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(\"env\"))\n}\n\nfunc getZone(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(\"zone\"))\n}\n\nfunc buildCLI() *cli.App {\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence-canary\"\n\tapp.Usage = \"Cadence canary\"\n\tapp.Version = \"0.0.1\"\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    \"root\",\n\t\t\tAliases: []string{\"r\"},\n\t\t\tValue:   \".\",\n\t\t\tUsage:   \"root directory of execution environment\",\n\t\t\tEnvVars: []string{canary.EnvKeyRoot},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"config\",\n\t\t\tAliases: []string{\"c\"},\n\t\t\tValue:   \"config/canary\",\n\t\t\tUsage:   \"config dir path relative to root\",\n\t\t\tEnvVars: []string{canary.EnvKeyConfigDir},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"env\",\n\t\t\tAliases: []string{\"e\"},\n\t\t\tValue:   \"development\",\n\t\t\tUsage:   \"runtime environment\",\n\t\t\tEnvVars: []string{canary.EnvKeyEnvironment},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"zone\",\n\t\t\tAliases: []string{\"az\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"availability zone\",\n\t\t\tEnvVars: []string{canary.EnvKeyAvailabilityZone},\n\t\t},\n\t}\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:  \"start\",\n\t\t\tUsage: \"start cadence canary worker or cron, or both\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    \"mode\",\n\t\t\t\t\tAliases: []string{\"m\"},\n\t\t\t\t\tValue:   canary.ModeAll,\n\t\t\t\t\tUsage:   fmt.Sprintf(\"%v, %v or %v\", canary.ModeWorker, canary.ModeCronCanary, canary.ModeAll),\n\t\t\t\t\tEnvVars: []string{canary.EnvKeyMode},\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn startHandler(c)\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n\nfunc main() {\n\tapp := buildCLI()\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n"
  },
  {
    "path": "cmd/server/README.md",
    "content": "This is the primary \"kitchen sink\" server build, which includes all first-party optional plugins, and is used as our Docker-provided binary.\n\nFor the most part, this means that day-to-day library upgrades should:\n1. Update the main `go.mod`'s dependencies\n2. `make tidy`\n\nNew submodules we want to include by default should:\n1. Add a replace in this `go.mod` like others\n2. Import and register it somehow\n3. `make tidy`\n\nAnd if you have problems tidying:\n1. Copy/paste all included modules into this `go.mod` (`/go.mod` + `/common/archiver/gcloud/go.mod` currently)\n2. `go mod tidy` this submodule and it will probably be correct, `make build` to make sure\n3. Commit to save your results!\n4. `make tidy` to make sure it's stable\n"
  },
  {
    "path": "cmd/server/cadence/cadence.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cadence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\tstdLog \"log\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"strings\"\n\t\"sync\"\n\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/service\"\n\n\t_ \"go.uber.org/automaxprocs\" // defines automaxpocs for dockerized usage.\n)\n\n// validServices is the list of all valid cadence services\nvar validServices = service.ShortNames(service.List)\nvar defaultServices = service.ShortNames(service.ListWithRing)\n\nfunc isValidService(in string) bool {\n\tfor _, s := range validServices {\n\t\tif s == in {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// BuildCLI is the main entry point for the cadence server\nfunc BuildCLI(releaseVersion string, gitRevision string) *cli.App {\n\tversion := fmt.Sprintf(\" Release version: %v \\n\"+\n\t\t\"   Build commit: %v\\n\"+\n\t\t\"   Max Support CLI feature version: %v \\n\"+\n\t\t\"   Max Support GoSDK feature version: %v \\n\"+\n\t\t\"   Max Support JavaSDK feature version: %v \\n\"+\n\t\t\"   Note:  Feature version is for compatibility checking between server and clients if enabled feature checking. Server is always backward compatible to older CLI versions, but not accepting newer than it can support.\",\n\t\treleaseVersion, gitRevision, client.SupportedCLIVersion, client.SupportedGoSDKVersion, client.SupportedJavaSDKVersion)\n\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence\"\n\tapp.Usage = \"Cadence server\"\n\tapp.Version = version\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    \"root\",\n\t\t\tAliases: []string{\"r\"},\n\t\t\tValue:   \".\",\n\t\t\tUsage:   \"root directory of execution environment\",\n\t\t\tEnvVars: []string{config.EnvKeyRoot},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"config\",\n\t\t\tAliases: []string{\"c\"},\n\t\t\tValue:   \"config\",\n\t\t\tUsage:   \"config dir is a path relative to root, or an absolute path\",\n\t\t\tEnvVars: []string{config.EnvKeyConfigDir},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"env\",\n\t\t\tAliases: []string{\"e\"},\n\t\t\tValue:   \"development\",\n\t\t\tUsage:   \"runtime environment\",\n\t\t\tEnvVars: []string{config.EnvKeyEnvironment},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    \"zone\",\n\t\t\tAliases: []string{\"az\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"availability zone\",\n\t\t\tEnvVars: []string{config.EnvKeyAvailabilityZone},\n\t\t},\n\t}\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:    \"start\",\n\t\t\tAliases: []string{\"\"},\n\t\t\tUsage:   \"start cadence server\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    \"services\",\n\t\t\t\t\tAliases: []string{\"s\"},\n\t\t\t\t\tValue:   strings.Join(defaultServices, \",\"),\n\t\t\t\t\tUsage:   \"list of services to start\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\thost, err := os.Hostname()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn fmt.Errorf(\"get hostname: %w\", err)\n\t\t\t\t}\n\n\t\t\t\tappCtx := appContext{\n\t\t\t\t\tCfgContext: config.Context{\n\t\t\t\t\t\tEnvironment: getEnvironment(c),\n\t\t\t\t\t\tZone:        getZone(c),\n\t\t\t\t\t},\n\t\t\t\t\tConfigDir: getConfigDir(c),\n\t\t\t\t\tRootDir:   getRootDir(c),\n\t\t\t\t\tHostName:  host,\n\t\t\t\t}\n\n\t\t\t\tservices := getServices(c)\n\n\t\t\t\treturn runServices(\n\t\t\t\t\tservices,\n\t\t\t\t\tfunc(serviceName string) fxAppInterface {\n\t\t\t\t\t\treturn fx.New(\n\t\t\t\t\t\t\tfx.Module(serviceName,\n\t\t\t\t\t\t\t\t_commonModule,\n\t\t\t\t\t\t\t\tfx.Provide(\n\t\t\t\t\t\t\t\t\tfunc() appContext {\n\t\t\t\t\t\t\t\t\t\treturn appCtx\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t\tModule(serviceName),\n\t\t\t\t\t\t\t),\n\t\t\t\t\t\t)\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n\nfunc runServices(services []string, appBuilder func(serviceName string) fxAppInterface) error {\n\tstoppedWg := &sync.WaitGroup{}\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\terrChan := make(chan error, len(services))\n\n\tfor _, serv := range services {\n\t\tstoppedWg.Add(1)\n\t\tgo func(s string) {\n\t\t\tdefer stoppedWg.Done()\n\t\t\tfxApp := appBuilder(s)\n\n\t\t\t//  If any of the start hooks return an error, Start short-circuits, calls Stop, and returns the inciting error.\n\t\t\tif err := fxApp.Start(ctx); err != nil {\n\t\t\t\t// If any of the apps fails to start, immediately cancel the context so others will also stop.\n\t\t\t\tcancel()\n\t\t\t\terrChan <- fmt.Errorf(\"service %s start: %w\", s, err)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tselect {\n\t\t\t// Block until FX receives a shutdown signal\n\t\t\tcase <-fxApp.Done():\n\t\t\t}\n\n\t\t\t// Stop the application\n\t\t\terr := fxApp.Stop(ctx)\n\t\t\tif err != nil {\n\t\t\t\terrChan <- fmt.Errorf(\"service %s stop: %w\", s, err)\n\t\t\t}\n\t\t}(serv)\n\t}\n\tgo func() {\n\t\tstoppedWg.Wait()\n\t\t// After stoppedWg unblocked all services are stopped to we no longer wait for errors.\n\t\tclose(errChan)\n\t}()\n\n\tvar resErrors error\n\tfor err := range errChan {\n\t\t// skip canceled errors, since they are caused by context cancelation and only focus on actual errors.\n\t\tif err != nil && !errors.Is(err, context.Canceled) {\n\t\t\tresErrors = multierr.Append(resErrors, err)\n\t\t}\n\t}\n\tif resErrors != nil {\n\t\treturn resErrors\n\t}\n\treturn nil\n}\n\ntype appContext struct {\n\tfx.Out\n\n\tCfgContext config.Context\n\tConfigDir  string `name:\"config-dir\"`\n\tRootDir    string `name:\"root-dir\"`\n\tHostName   string `name:\"hostname\"`\n}\n\nfunc getEnvironment(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(\"env\"))\n}\n\nfunc getZone(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(\"zone\"))\n}\n\n// getServices parses the services arg from cli\n// and returns a list of services to start\nfunc getServices(c *cli.Context) []string {\n\tval := strings.TrimSpace(c.String(\"services\"))\n\ttokens := strings.Split(val, \",\")\n\n\tif len(tokens) == 0 {\n\t\tstdLog.Fatal(\"list of services is empty\")\n\t}\n\n\tfor _, t := range tokens {\n\t\tif !isValidService(t) {\n\t\t\tstdLog.Fatalf(\"invalid service `%v` in service list [%v]\", t, val)\n\t\t}\n\t}\n\n\treturn tokens\n}\n\nfunc getConfigDir(c *cli.Context) string {\n\treturn constructPathIfNeed(getRootDir(c), c.String(\"config\"))\n}\n\nfunc getRootDir(c *cli.Context) string {\n\tdirpath := c.String(\"root\")\n\tif len(dirpath) == 0 {\n\t\tcwd, err := os.Getwd()\n\t\tif err != nil {\n\t\t\tstdLog.Fatalf(\"os.Getwd() failed, err=%v\", err)\n\t\t}\n\t\treturn cwd\n\t}\n\treturn dirpath\n}\n\n// constructPathIfNeed would append the dir as the root dir\n// when the file wasn't absolute path.\nfunc constructPathIfNeed(dir string, file string) string {\n\tif !filepath.IsAbs(file) {\n\t\treturn dir + \"/\" + file\n\t}\n\treturn file\n}\n\ntype fxAppInterface interface {\n\tStart(context.Context) error\n\tStop(context.Context) error\n\tDone() <-chan os.Signal\n}\n"
  },
  {
    "path": "cmd/server/cadence/cadence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cadence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"os\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n)\n\ntype CadenceSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestCadenceSuite(t *testing.T) {\n\tsuite.Run(t, new(CadenceSuite))\n}\n\nfunc (s *CadenceSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *CadenceSuite) TestIsValidService() {\n\ts.True(isValidService(\"history\"))\n\ts.True(isValidService(\"matching\"))\n\ts.True(isValidService(\"frontend\"))\n\ts.False(isValidService(\"cadence-history\"))\n\ts.False(isValidService(\"cadence-matching\"))\n\ts.False(isValidService(\"cadence-frontend\"))\n\ts.False(isValidService(\"foobar\"))\n}\n\nfunc (s *CadenceSuite) TestPath() {\n\ts.Equal(\"foo/bar\", constructPathIfNeed(\"foo\", \"bar\"))\n\ts.Equal(\"/bar\", constructPathIfNeed(\"foo\", \"/bar\"))\n}\n\n// MockFxApp implements fxAppInterface for testing\ntype MockFxApp struct {\n\tStartFunc func(context.Context) error\n\tStopFunc  func(context.Context) error\n\tDoneFunc  func() <-chan os.Signal\n}\n\nfunc (m *MockFxApp) Start(ctx context.Context) error {\n\treturn m.StartFunc(ctx)\n}\n\nfunc (m *MockFxApp) Stop(ctx context.Context) error {\n\treturn m.StopFunc(ctx)\n}\n\nfunc (m *MockFxApp) Done() <-chan os.Signal {\n\treturn m.DoneFunc()\n}\n\n// TestRunServicesSuccess tests successful service execution\nfunc TestRunServicesSuccess(t *testing.T) {\n\t// Create a test application using fxtest\n\tapp := fxtest.New(t,\n\t\tfx.Provide(func() string { return \"test-service\" }),\n\t\tfx.Invoke(func(s string) {\n\t\t\t// Just a simple component that does nothing\n\t\t}),\n\t)\n\n\t// Create a done channel\n\tdone := make(chan os.Signal, 1)\n\n\t// Wrap fxtest.App in our interface\n\tappInterface := &MockFxApp{\n\t\tStartFunc: app.Start,\n\t\tStopFunc:  app.Stop,\n\t\tDoneFunc:  func() <-chan os.Signal { return done },\n\t}\n\n\t// Run in a goroutine\n\terrCh := make(chan error, 1)\n\tgo func() {\n\t\terrCh <- runServices([]string{\"service1\"}, func(name string) fxAppInterface {\n\t\t\treturn appInterface\n\t\t})\n\t}()\n\n\t// Give it a moment to start\n\ttime.Sleep(50 * time.Millisecond)\n\n\t// Send signal to stop\n\tclose(done)\n\n\t// Check result\n\terr := <-errCh\n\tassert.NoError(t, err)\n}\n\n// TestRunServicesStartError tests failure during service start\nfunc TestRunServicesStartError(t *testing.T) {\n\t// Create a mock app that fails on start\n\tstartError := errors.New(\"failed to start\")\n\tapp := &MockFxApp{\n\t\tStartFunc: func(ctx context.Context) error {\n\t\t\treturn startError\n\t\t},\n\t\tStopFunc: func(ctx context.Context) error {\n\t\t\treturn nil\n\t\t},\n\t\tDoneFunc: func() <-chan os.Signal {\n\t\t\tch := make(chan os.Signal)\n\t\t\treturn ch\n\t\t},\n\t}\n\n\t// Run the services\n\terr := runServices([]string{\"service1\"}, func(name string) fxAppInterface {\n\t\treturn app\n\t})\n\n\t// Verify error was returned\n\tassert.Error(t, err)\n\tassert.True(t, errors.Is(err, startError), \"Error chain should contain the original error\")\n}\n\n// TestRunServicesStopError tests failure during service stop\nfunc TestRunServicesStopError(t *testing.T) {\n\t// Create a mock app that fails on stop\n\tstopError := errors.New(\"failed to stop\")\n\tdone := make(chan os.Signal, 1)\n\tapp := &MockFxApp{\n\t\tStartFunc: func(ctx context.Context) error {\n\t\t\treturn nil\n\t\t},\n\t\tStopFunc: func(ctx context.Context) error {\n\t\t\treturn stopError\n\t\t},\n\t\tDoneFunc: func() <-chan os.Signal {\n\t\t\treturn done\n\t\t},\n\t}\n\n\t// Run in a goroutine\n\terrCh := make(chan error, 1)\n\tgo func() {\n\t\terrCh <- runServices([]string{\"service1\"}, func(name string) fxAppInterface {\n\t\t\treturn app\n\t\t})\n\t}()\n\n\t// Give it a moment to start\n\ttime.Sleep(50 * time.Millisecond)\n\n\t// Signal that the service is done\n\tclose(done)\n\n\t// Check the error\n\terr := <-errCh\n\tassert.Error(t, err)\n\tassert.True(t, errors.Is(err, stopError), \"Error chain should contain the stop error\")\n}\n\n// TestRunServicesCascadeFailure tests that when one service fails, others get stopped\nfunc TestRunServicesCascadeFailure(t *testing.T) {\n\t// We'll use this to track context cancellation\n\tvar contextCancelled bool\n\tvar contextMu sync.Mutex\n\n\t// Create two apps - one will fail, one will succeed but should be stopped\n\tstartErr := errors.New(\"service 2 failed to start\")\n\n\t// Track app lifecycle events\n\tapp1Started := false\n\tapp1Stopped := false\n\tapp2Started := false\n\n\t// First app - will start successfully\n\tdone1 := make(chan os.Signal, 1)\n\tapp1 := &MockFxApp{\n\t\tStartFunc: func(ctx context.Context) error {\n\t\t\tapp1Started = true\n\n\t\t\t// Monitor for context cancellation in a goroutine\n\t\t\tgo func() {\n\t\t\t\t<-ctx.Done()\n\t\t\t\tcontextMu.Lock()\n\t\t\t\tcontextCancelled = true\n\t\t\t\tcontextMu.Unlock()\n\t\t\t\t// Close done channel to simulate app stopping due to context\n\t\t\t\tclose(done1)\n\t\t\t}()\n\n\t\t\treturn nil\n\t\t},\n\t\tStopFunc: func(ctx context.Context) error {\n\t\t\tapp1Stopped = true\n\t\t\treturn nil\n\t\t},\n\t\tDoneFunc: func() <-chan os.Signal {\n\t\t\treturn done1\n\t\t},\n\t}\n\n\t// Second app - will fail to start\n\tapp2 := &MockFxApp{\n\t\tStartFunc: func(ctx context.Context) error {\n\t\t\tapp2Started = true\n\t\t\treturn startErr\n\t\t},\n\t\tStopFunc: func(ctx context.Context) error {\n\t\t\tt.Fatal(\"App2 Stop should never be called since Start failed\")\n\t\t\treturn nil\n\t\t},\n\t\tDoneFunc: func() <-chan os.Signal {\n\t\t\tch := make(chan os.Signal)\n\t\t\treturn ch // Never signals\n\t\t},\n\t}\n\n\t// Build app provider that returns different apps for different services\n\tappProvider := func(name string) fxAppInterface {\n\t\tswitch name {\n\t\tcase \"service1\":\n\t\t\treturn app1\n\t\tcase \"service2\":\n\t\t\treturn app2\n\t\tdefault:\n\t\t\tt.Fatalf(\"Unexpected service name: %s\", name)\n\t\t\treturn nil\n\t\t}\n\t}\n\n\t// Run services\n\terr := runServices([]string{\"service1\", \"service2\"}, appProvider)\n\n\t// Verify results\n\trequire.Error(t, err, \"Should return an error\")\n\tassert.True(t, app1Started, \"App1 should have started\")\n\tassert.True(t, app2Started, \"App2 should have started\")\n\tassert.True(t, app1Stopped, \"App1 should have been stopped due to context cancellation\")\n\n\t// Check if context was cancelled\n\tcontextMu.Lock()\n\tassert.True(t, contextCancelled, \"Context should have been cancelled\")\n\tcontextMu.Unlock()\n\n\t// Check error content\n\tassert.Contains(t, err.Error(), \"service 2\")\n}\n"
  },
  {
    "path": "cmd/server/cadence/fx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cadence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock/clockfx\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicconfigfx\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/logfx\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/metricsfx\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/rpc/rpcfx\"\n\t\"github.com/uber/cadence/common/service\"\n\tshardDistributorCfg \"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/sharddistributorfx\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\nvar _commonModule = fx.Options(\n\tconfig.Module,\n\tdynamicconfigfx.Module,\n\tlogfx.Module,\n\tmetricsfx.Module,\n\tclockfx.Module)\n\n// Module provides a cadence server initialization with root components.\n// AppParams allows to provide optional/overrides for implementation specific dependencies.\nfunc Module(serviceName string) fx.Option {\n\tif serviceName == service.ShortName(service.ShardDistributor) {\n\t\treturn fx.Options(\n\t\t\tfx.Supply(serviceContext{\n\t\t\t\tName:     serviceName,\n\t\t\t\tFullName: service.FullName(serviceName),\n\t\t\t}),\n\t\t\tfx.Provide(func(cfg config.Config) shardDistributorCfg.ShardDistribution {\n\t\t\t\treturn cfg.ShardDistribution\n\t\t\t}),\n\t\t\t// Decorate both logger so all components use proper service name.\n\t\t\tfx.Decorate(func(z *zap.Logger, l log.Logger) (*zap.Logger, log.Logger) {\n\t\t\t\treturn z.With(zap.String(\"service\", service.ShardDistributor)), l.WithTags(tag.Service(service.ShardDistributor))\n\t\t\t}),\n\n\t\t\tetcd.Module,\n\n\t\t\trpcfx.Module,\n\t\t\tsharddistributorfx.Module)\n\t}\n\treturn fx.Options(\n\t\tfx.Supply(serviceContext{\n\t\t\tName:     serviceName,\n\t\t\tFullName: service.FullName(serviceName),\n\t\t}),\n\t\tfx.Provide(NewApp),\n\t\t// empty invoke so fx won't drop the application from the dependencies.\n\t\tfx.Invoke(func(a *App) {}),\n\t)\n}\n\ntype AppParams struct {\n\tfx.In\n\n\tService       string `name:\"service\"`\n\tAppContext    config.Context\n\tConfig        config.Config\n\tLogger        log.Logger\n\tLifeCycle     fx.Lifecycle\n\tDynamicConfig dynamicconfig.Client\n\tScope         tally.Scope\n\tMetricsClient metrics.Client\n}\n\n// NewApp created a new Application from pre initalized config and logger.\nfunc NewApp(params AppParams) *App {\n\tapp := &App{\n\t\tcfg:           params.Config,\n\t\tlogger:        params.Logger,\n\t\tservice:       params.Service,\n\t\tdynamicConfig: params.DynamicConfig,\n\t\tscope:         params.Scope,\n\t\tmetricsClient: params.MetricsClient,\n\t}\n\n\tparams.LifeCycle.Append(fx.StartHook(app.verifySchema))\n\tparams.LifeCycle.Append(fx.StartStopHook(app.Start, app.Stop))\n\treturn app\n}\n\n// App is a fx application that registers itself into fx.Lifecycle and runs.\n// It is done implicitly, since it provides methods Start and Stop which are picked up by fx.\ntype App struct {\n\tcfg           config.Config\n\trootDir       string\n\tlogger        log.Logger\n\tdynamicConfig dynamicconfig.Client\n\tscope         tally.Scope\n\tmetricsClient metrics.Client\n\n\tdaemon  common.Daemon\n\tservice string\n}\n\nfunc (a *App) Start(_ context.Context) error {\n\ta.daemon = newServer(a.service, a.cfg, a.logger, a.dynamicConfig, a.scope, a.metricsClient)\n\ta.daemon.Start()\n\treturn nil\n}\n\nfunc (a *App) Stop(ctx context.Context) error {\n\ta.daemon.Stop()\n\treturn nil\n}\n\nfunc (a *App) verifySchema(ctx context.Context) error {\n\t// cassandra schema version validation\n\tif err := cassandra.VerifyCompatibleVersion(a.cfg.Persistence, gocql.Quorum); err != nil {\n\t\treturn fmt.Errorf(\"cassandra schema version compatibility check failed: %w\", err)\n\t}\n\t// sql schema version validation\n\tif err := sql.VerifyCompatibleVersion(a.cfg.Persistence); err != nil {\n\t\treturn fmt.Errorf(\"sql schema version compatibility check failed: %w\", err)\n\t}\n\treturn nil\n}\n\ntype serviceContext struct {\n\tfx.Out\n\n\tName     string `name:\"service\"`\n\tFullName string `name:\"service-full-name\"`\n}\n"
  },
  {
    "path": "cmd/server/cadence/fx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cadence\n\nimport (\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/testflags\"\n\n\t_ \"github.com/uber/cadence/service/sharddistributor/store/etcd\" // needed for shard distributor storage\n)\n\nfunc TestFxDependencies(t *testing.T) {\n\terr := fx.ValidateApp(_commonModule,\n\t\tfx.Supply(appContext{\n\t\t\tCfgContext: config.Context{\n\t\t\t\tEnvironment: \"\",\n\t\t\t\tZone:        \"\",\n\t\t\t},\n\t\t\tConfigDir: \"\",\n\t\t\tRootDir:   \"\",\n\t\t}),\n\t\tModule(\"\"))\n\trequire.NoError(t, err)\n}\n\nfunc TestFxDependenciesForShardDistributor(t *testing.T) {\n\terr := fx.ValidateApp(_commonModule,\n\t\tfx.Supply(appContext{\n\t\t\tCfgContext: config.Context{\n\t\t\t\tEnvironment: \"\",\n\t\t\t\tZone:        \"\",\n\t\t\t},\n\t\t\tConfigDir: \"\",\n\t\t\tRootDir:   \"\",\n\t\t}),\n\t\tModule(service.ShortName(service.ShardDistributor)))\n\trequire.NoError(t, err)\n}\n\nfunc TestShardDistributorStartStop(t *testing.T) {\n\tflag.Parse()\n\ttestflags.RequireEtcd(t)\n\n\twd, err := os.Getwd()\n\trequire.NoError(t, err)\n\tapp := fxtest.New(t, _commonModule,\n\t\tfx.Supply(appContext{\n\t\t\tCfgContext: config.Context{\n\t\t\t\tEnvironment: \"development\",\n\t\t\t\tZone:        \"\",\n\t\t\t},\n\t\t\tConfigDir: fmt.Sprintf(\"%s/testdata/config\", wd),\n\t\t\tRootDir:   \"\",\n\t\t}),\n\t\tModule(service.ShortName(service.ShardDistributor)))\n\tapp.RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "cmd/server/cadence/server.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cadence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/startreedata/pinot-client-go/pinot\"\n\t\"github.com/uber-go/tally\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/compatibility\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\tsharddistributorClient \"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/errorinjectors\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/client/wrappers/metered\"\n\t\"github.com/uber/cadence/client/wrappers/retryable\"\n\ttimeoutwrapper \"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/blobstore/filestore\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging/kafka\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/peerprovider/ringpopprovider\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/frontend\"\n\t\"github.com/uber/cadence/service/history\"\n\t\"github.com/uber/cadence/service/matching\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n\t\"github.com/uber/cadence/service/worker\"\n\tdiagnosticsInvariant \"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/failure\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/retry\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/timeout\"\n)\n\ntype (\n\tserver struct {\n\t\tname             string\n\t\tcfg              config.Config\n\t\tlogger           log.Logger\n\t\tdoneC            chan struct{}\n\t\tdaemon           common.Daemon\n\t\tdynamicCfgClient dynamicconfig.Client\n\t\tscope            tally.Scope\n\t\tmetricsClient    metrics.Client\n\t}\n)\n\n// newServer returns a new instance of a daemon\n// that represents a cadence service\nfunc newServer(service string, cfg config.Config, logger log.Logger, dynamicCfgClient dynamicconfig.Client, scope tally.Scope, metricsClient metrics.Client) common.Daemon {\n\treturn &server{\n\t\tcfg:              cfg,\n\t\tname:             service,\n\t\tdoneC:            make(chan struct{}),\n\t\tlogger:           logger,\n\t\tdynamicCfgClient: dynamicCfgClient,\n\t\tscope:            scope,\n\t\tmetricsClient:    metricsClient,\n\t}\n}\n\n// Start starts the server\nfunc (s *server) Start() {\n\ts.daemon = s.startService()\n}\n\n// Stop stops the server\nfunc (s *server) Stop() {\n\tif s.daemon == nil {\n\t\treturn\n\t}\n\n\tselect {\n\tcase <-s.doneC:\n\tdefault:\n\t\ts.daemon.Stop()\n\t\tselect {\n\t\tcase <-s.doneC:\n\t\tcase <-time.After(time.Minute):\n\t\t\ts.logger.Warn(\"timed out waiting for server to exit\")\n\t\t}\n\t}\n}\n\n// startService starts a service with the given name and config\nfunc (s *server) startService() common.Daemon {\n\tsvcCfg, err := s.cfg.GetServiceConfig(s.name)\n\tif err != nil {\n\t\ts.logger.Fatal(err.Error())\n\t}\n\n\tparams := resource.Params{\n\t\tName:              service.FullName(s.name),\n\t\tLogger:            s.logger.WithTags(tag.Service(service.FullName(s.name))),\n\t\tPersistenceConfig: s.cfg.Persistence,\n\t\tDynamicConfig:     s.dynamicCfgClient,\n\t\tRPCConfig:         svcCfg.RPC,\n\t}\n\n\tclusterGroupMetadata := s.cfg.ClusterGroupMetadata\n\tdc := dynamicconfig.NewCollection(\n\t\tparams.DynamicConfig,\n\t\tparams.Logger,\n\t\tdynamicproperties.ClusterNameFilter(clusterGroupMetadata.CurrentClusterName),\n\t)\n\n\tparams.MetricScope = s.scope\n\tparams.MetricsClient = s.metricsClient\n\n\trpcParams, err := rpc.NewParams(params.Name, &s.cfg, dc, params.Logger, params.MetricsClient)\n\tif err != nil {\n\t\ts.logger.Fatal(\"error creating rpc factory params\", tag.Error(err))\n\t}\n\trpcParams.OutboundsBuilder = rpc.CombineOutbounds(\n\t\trpcParams.OutboundsBuilder,\n\t\trpc.NewCrossDCOutbounds(clusterGroupMetadata.ClusterGroup, rpc.NewDNSPeerChooserFactory(s.cfg.PublicClient.RefreshInterval, params.Logger)),\n\t)\n\trpcFactory := rpc.NewFactory(params.Logger, rpcParams)\n\tparams.RPCFactory = rpcFactory\n\n\tpeerProvider, err := ringpopprovider.New(\n\t\tparams.Name,\n\t\t&s.cfg.Ringpop,\n\t\trpcFactory.GetTChannel(),\n\t\tmembership.PortMap{\n\t\t\tmembership.PortGRPC:     svcCfg.RPC.GRPCPort,\n\t\t\tmembership.PortTchannel: svcCfg.RPC.Port,\n\t\t},\n\t\tparams.Logger,\n\t)\n\n\tif err != nil {\n\t\ts.logger.Fatal(\"ringpop provider failed\", tag.Error(err))\n\t}\n\n\tshardDistributorClient := s.createShardDistributorClient(params, dc)\n\tvar spectator spectatorclient.Spectator\n\tif shardDistributorClient != nil && len(s.cfg.ShardDistributorMatchingConfig.Namespaces) > 0 {\n\t\tif len(s.cfg.ShardDistributorMatchingConfig.Namespaces) > 1 {\n\t\t\ts.logger.Fatal(\"spectator does not support multiple namespaces\", tag.Value(s.cfg.ShardDistributorMatchingConfig.Namespaces))\n\t\t}\n\t\tmatchingShardDistributionMode := dc.GetStringProperty(dynamicproperties.MatchingShardDistributionMode)\n\n\t\tspectatorParams := spectatorclient.Params{\n\t\t\tClient:       shardDistributorClient,\n\t\t\tMetricsScope: params.MetricScope,\n\t\t\tLogger:       params.Logger,\n\t\t\tConfig:       s.cfg.ShardDistributorMatchingConfig,\n\t\t\tTimeSource:   clock.NewRealTimeSource(),\n\t\t\tEnabled: func() bool {\n\t\t\t\treturn membership.ModeKey(matchingShardDistributionMode()) != membership.ModeKeyHashRing\n\t\t\t},\n\t\t}\n\t\tnamespace := s.cfg.ShardDistributorMatchingConfig.Namespaces[0].Namespace\n\t\tspectator, err = spectatorclient.NewSpectatorWithNamespace(\n\t\t\tspectatorParams,\n\t\t\tnamespace,\n\t\t)\n\t\tif err != nil {\n\t\t\ts.logger.Fatal(\"error creating spectator\", tag.Error(err))\n\t\t}\n\n\t\t// Start the spectator to begin watching namespace state\n\t\tif err := spectator.Start(context.Background()); err != nil {\n\t\t\ts.logger.Fatal(\"error starting spectator\", tag.Error(err))\n\t\t}\n\t} else {\n\t\ts.logger.Warn(\"Shard distributor client not configured, spectator will not be started\")\n\t}\n\n\tparams.HashRings = make(map[string]membership.SingleProvider)\n\tfor _, s := range service.ListWithRing {\n\t\tparams.HashRings[s] = membership.NewHashring(s, peerProvider, clock.NewRealTimeSource(), params.Logger, params.MetricsClient.Scope(metrics.HashringScope))\n\t}\n\n\twrappedRings := s.wrapHashRingsWithShardDistributor(params.HashRings, spectator, dc, params.Logger)\n\n\tparams.MembershipResolver, err = membership.NewResolver(\n\t\tpeerProvider,\n\t\tparams.MetricsClient,\n\t\tparams.Logger,\n\t\twrappedRings,\n\t)\n\n\tif err != nil {\n\t\ts.logger.Fatal(\"error creating membership monitor\", tag.Error(err))\n\t}\n\tparams.PProfInitializer = svcCfg.PProf.NewInitializer(params.Logger)\n\n\tparams.ClusterRedirectionPolicy = s.cfg.ClusterGroupMetadata.ClusterRedirectionPolicy\n\n\tparams.GetIsolationGroups = getFromDynamicConfig(params, dc)\n\n\tparams.ClusterMetadata = cluster.NewMetadata(\n\t\t*clusterGroupMetadata,\n\t\tdc.GetBoolPropertyFilteredByDomain(dynamicproperties.UseNewInitialFailoverVersion),\n\t\tparams.MetricsClient,\n\t\tparams.Logger,\n\t)\n\n\tadvancedVisMode := dc.GetStringProperty(\n\t\tdynamicproperties.WriteVisibilityStoreName,\n\t)()\n\tisAdvancedVisEnabled := common.IsAdvancedVisibilityWritingEnabled(advancedVisMode, params.PersistenceConfig.IsAdvancedVisibilityConfigExist())\n\tif isAdvancedVisEnabled {\n\t\tparams.MessagingClient = kafka.NewKafkaClient(&s.cfg.Kafka, params.MetricsClient, params.Logger, params.MetricScope, isAdvancedVisEnabled)\n\t} else {\n\t\tparams.MessagingClient = nil\n\t}\n\n\tif isAdvancedVisEnabled {\n\t\ts.setupVisibilityClients(&params)\n\t}\n\n\tpublicClientConfig := params.RPCFactory.GetDispatcher().ClientConfig(rpc.OutboundPublicClient)\n\tif rpc.IsGRPCOutbound(publicClientConfig) {\n\t\tparams.PublicClient = compatibility.NewThrift2ProtoAdapter(\n\t\t\tapiv1.NewDomainAPIYARPCClient(publicClientConfig),\n\t\t\tapiv1.NewWorkflowAPIYARPCClient(publicClientConfig),\n\t\t\tapiv1.NewWorkerAPIYARPCClient(publicClientConfig),\n\t\t\tapiv1.NewVisibilityAPIYARPCClient(publicClientConfig),\n\t\t)\n\t} else {\n\t\tparams.PublicClient = workflowserviceclient.New(publicClientConfig)\n\t}\n\n\tparams.ArchivalMetadata = archiver.NewArchivalMetadata(\n\t\tdc,\n\t\ts.cfg.Archival.History.Status,\n\t\ts.cfg.Archival.History.EnableRead,\n\t\ts.cfg.Archival.Visibility.Status,\n\t\ts.cfg.Archival.Visibility.EnableRead,\n\t\t&s.cfg.DomainDefaults.Archival,\n\t)\n\n\tparams.ArchiverProvider = provider.NewArchiverProvider(s.cfg.Archival.History.Provider, s.cfg.Archival.Visibility.Provider)\n\tparams.PersistenceConfig.TransactionSizeLimit = dc.GetIntProperty(dynamicproperties.TransactionSizeLimit)\n\tparams.PersistenceConfig.ErrorInjectionRate = dc.GetFloat64Property(dynamicproperties.PersistenceErrorInjectionRate)\n\tparams.AuthorizationConfig = s.cfg.Authorization\n\tparams.BlobstoreClient, err = filestore.NewFilestoreClient(s.cfg.Blobstore.Filestore)\n\tif err != nil {\n\t\ts.logger.Warn(\"failed to create file blobstore client, will continue startup without it: %v\", tag.Error(err))\n\t\tparams.BlobstoreClient = nil\n\t}\n\n\tparams.AsyncWorkflowQueueProvider, err = queue.NewAsyncQueueProvider(s.cfg.AsyncWorkflowQueues)\n\tif err != nil {\n\t\ts.logger.Fatal(\"error creating async queue provider\", tag.Error(err))\n\t}\n\n\tparams.KafkaConfig = s.cfg.Kafka\n\tparams.DiagnosticsInvariants = []diagnosticsInvariant.Invariant{timeout.NewInvariant(timeout.Params{Client: params.PublicClient}), failure.NewInvariant(), retry.NewInvariant()}\n\tparams.ShardDistributorMatchingConfig = s.cfg.ShardDistributorMatchingConfig\n\n\tparams.Logger.Info(\"Starting service \" + s.name)\n\n\tvar daemon common.Daemon\n\n\tswitch params.Name {\n\tcase service.Frontend:\n\t\tdaemon, err = frontend.NewService(&params)\n\tcase service.History:\n\t\tdaemon, err = history.NewService(&params)\n\tcase service.Matching:\n\t\tdaemon, err = matching.NewService(&params)\n\tcase service.Worker:\n\t\tdaemon, err = worker.NewService(&params)\n\tdefault:\n\t\tparams.Logger.Fatal(\"unknown service\", tag.Service(params.Name))\n\t}\n\tif err != nil {\n\t\tparams.Logger.Fatal(\"Fail to start \"+s.name+\" service \", tag.Error(err))\n\t}\n\n\tgo execute(daemon, s.doneC)\n\n\treturn daemon\n}\n\nfunc (*server) wrapHashRingsWithShardDistributor(\n\thashRings map[string]membership.SingleProvider,\n\tspectator spectatorclient.Spectator,\n\tdc *dynamicconfig.Collection,\n\tlogger log.Logger,\n) map[string]membership.SingleProvider {\n\tif _, ok := hashRings[service.Matching]; ok {\n\t\thashRings[service.Matching] = membership.NewShardDistributorResolver(\n\t\t\tspectator,\n\t\t\tdc.GetStringProperty(dynamicproperties.MatchingShardDistributionMode),\n\t\t\tdc.GetBoolProperty(dynamicproperties.MatchingExcludeShortLivedTaskListsFromShardManager),\n\t\t\tdc.GetIntProperty(dynamicproperties.MatchingPercentageOnboardedToShardManager),\n\t\t\thashRings[service.Matching],\n\t\t\tlogger,\n\t\t)\n\t}\n\treturn hashRings\n}\n\nfunc (*server) createShardDistributorClient(\n\tparams resource.Params,\n\tdc *dynamicconfig.Collection,\n) sharddistributorClient.Client {\n\tshardDistributorClientConfig, ok := params.RPCFactory.GetDispatcher().OutboundConfig(service.ShardDistributor)\n\tvar shardDistributorClient sharddistributorClient.Client\n\tif ok {\n\t\tif !rpc.IsGRPCOutbound(shardDistributorClientConfig) {\n\t\t\tparams.Logger.Error(\"shard distributor client does not support non-GRPC outbound will fail back to hashring\")\n\t\t\treturn nil\n\t\t}\n\t\tif shardDistributorClientConfig.Outbounds.Stream == nil {\n\t\t\tparams.Logger.Error(\"shard distributor client does not support stream outbound will fail back to hashring\")\n\t\t\treturn nil\n\t\t}\n\n\t\tshardDistributorClient = grpc.NewShardDistributorClient(\n\t\t\tsharddistributorv1.NewShardDistributorAPIYARPCClient(shardDistributorClientConfig),\n\t\t)\n\n\t\tshardDistributorClient = timeoutwrapper.NewShardDistributorClient(shardDistributorClient, timeoutwrapper.ShardDistributorDefaultTimeout)\n\t\tshardDistributorClient = retryable.NewShardDistributorClient(\n\t\t\tshardDistributorClient,\n\t\t\tcommon.CreateShardDistributorServiceRetryPolicy(),\n\t\t\tcommon.IsServiceTransientError,\n\t\t)\n\t\tif errorRate := dc.GetFloat64Property(dynamicproperties.ShardDistributorErrorInjectionRate)(); errorRate != 0 {\n\t\t\tshardDistributorClient = errorinjectors.NewShardDistributorClient(shardDistributorClient, errorRate, params.Logger)\n\t\t}\n\t\tif params.MetricsClient != nil {\n\t\t\tshardDistributorClient = metered.NewShardDistributorClient(shardDistributorClient, params.MetricsClient)\n\t\t}\n\t}\n\treturn shardDistributorClient\n}\n\n// execute runs the daemon in a separate go routine\nfunc execute(d common.Daemon, doneC chan struct{}) {\n\td.Start()\n\tclose(doneC)\n}\n\n// there are multiple circumstances:\n// 1. advanced visibility store == elasticsearch, use ESClient and visibilityDualManager\n// 2. advanced visibility store == pinot and in process of migration, use ESClient, PinotClient and and visibilityTripleManager\n// 3. advanced visibility store == pinot and not migrating, use PinotClient and visibilityDualManager\n// 4. advanced visibility store == opensearch and not migrating, this performs the same as 1, just use different version ES client and visibilityDualManager\n// 5. advanced visibility store == opensearch and in process of migration, use ESClient and visibilityTripleManager\nfunc (s *server) setupVisibilityClients(params *resource.Params) {\n\tadvancedVisStoreKey := s.cfg.Persistence.AdvancedVisibilityStore\n\tadvancedVisStore, ok := s.cfg.Persistence.DataStores[advancedVisStoreKey]\n\tif !ok {\n\t\ts.logger.Fatal(\"Cannot find advanced visibility store in config\", tag.Value(advancedVisStoreKey))\n\t}\n\n\t// Handle advanced visibility store based on type and migration state\n\tswitch advancedVisStoreKey {\n\tcase constants.PinotVisibilityStoreName:\n\t\ts.setupPinotClient(params, advancedVisStore)\n\tcase constants.OSVisibilityStoreName:\n\t\ts.setupOSClient(params, advancedVisStore)\n\tdefault: // Assume Elasticsearch by default\n\t\ts.setupESClient(params)\n\t}\n}\n\nfunc (s *server) setupPinotClient(params *resource.Params, advancedVisStore config.DataStore) {\n\tparams.PinotConfig = advancedVisStore.Pinot\n\tpinotBroker := params.PinotConfig.Broker\n\tpinotRawClient, err := pinot.NewFromBrokerList([]string{pinotBroker})\n\tif err != nil || pinotRawClient == nil {\n\t\ts.logger.Fatal(\"Creating Pinot visibility client failed\", tag.Error(err))\n\t}\n\tparams.PinotClient = pnt.NewPinotClient(pinotRawClient, params.Logger, params.PinotConfig)\n\tif advancedVisStore.Pinot.Migration.Enabled {\n\t\ts.setupESClient(params)\n\t}\n}\n\nfunc (s *server) setupESClient(params *resource.Params) {\n\tesVisibilityStore, ok := s.cfg.Persistence.DataStores[constants.ESVisibilityStoreName]\n\tif !ok {\n\t\ts.logger.Fatal(\"Cannot find Elasticsearch visibility store in config\")\n\t}\n\n\tparams.ESConfig = esVisibilityStore.ElasticSearch\n\tparams.ESConfig.SetUsernamePassword()\n\n\tesClient, err := elasticsearch.NewGenericClient(params.ESConfig, params.Logger)\n\tif err != nil {\n\t\ts.logger.Fatal(\"Error creating Elasticsearch client\", tag.Error(err))\n\t}\n\tparams.ESClient = esClient\n\n\terr = validateIndex(params.ESConfig)\n\tif err != nil {\n\t\ts.logger.Fatal(\"Error creating OpenSearch client\", tag.Error(err))\n\t}\n}\n\nfunc (s *server) setupOSClient(params *resource.Params, advancedVisStore config.DataStore) {\n\t// OpenSearch client setup (same structure as Elasticsearch, just version difference)\n\t// This is only for migration purposes\n\tparams.OSConfig = advancedVisStore.ElasticSearch\n\tparams.OSConfig.SetUsernamePassword()\n\n\tosClient, err := elasticsearch.NewGenericClient(params.OSConfig, params.Logger)\n\tif err != nil {\n\t\ts.logger.Fatal(\"Error creating OpenSearch client\", tag.Error(err))\n\t}\n\tparams.OSClient = osClient\n\n\terr = validateIndex(params.OSConfig)\n\tif err != nil {\n\t\ts.logger.Fatal(\"Error creating OpenSearch client\", tag.Error(err))\n\t}\n\n\tif advancedVisStore.ElasticSearch.Migration.Enabled {\n\t\ts.setupESClient(params)\n\t} else {\n\t\t// to avoid code duplication, we will use es-visibility and set the version to os2 instead of using os-visibility directly\n\t\tparams.ESConfig = advancedVisStore.ElasticSearch\n\t\tparams.ESConfig.SetUsernamePassword()\n\t\tparams.ESClient = osClient\n\t}\n}\n\nfunc validateIndex(config *config.ElasticSearchConfig) error {\n\tindexName, ok := config.Indices[constants.VisibilityAppName]\n\tif !ok || len(indexName) == 0 {\n\t\treturn fmt.Errorf(\"visibility index is missing in config\")\n\t}\n\treturn nil\n}\n\nfunc getFromDynamicConfig(params resource.Params, dc *dynamicconfig.Collection) func() []string {\n\treturn func() []string {\n\t\tres, err := isolationgroupapi.MapAllIsolationGroupsResponse(dc.GetListProperty(dynamicproperties.AllIsolationGroups)())\n\t\tif err != nil {\n\t\t\tparams.Logger.Error(\"failed to get isolation groups from config\", tag.Error(err))\n\t\t\treturn nil\n\t\t}\n\t\treturn res\n\t}\n}\n"
  },
  {
    "path": "cmd/server/cadence/server_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build !race\n// +build !race\n\npackage cadence\n\nimport (\n\t\"context\"\n\t\"slices\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\ntype ServerSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tlogger log.Logger\n}\n\nfunc TestServerSuite(t *testing.T) {\n\tsuite.Run(t, new(ServerSuite))\n}\n\nfunc (s *ServerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.logger = testlogger.New(s.T())\n}\n\n/*\nTestServerStartup tests the startup logic for the binary. When this fails, you should be able to reproduce by running \"cadence-server start\"\nIf you need to run locally, make sure Cassandra is up and schema is installed(run `make install-schema`)\n*/\nfunc (s *ServerSuite) TestServerStartup() {\n\tenv := \"development\"\n\tzone := \"\"\n\trootDir := \"../../../\"\n\tconfigDir := constructPathIfNeed(rootDir, \"config\")\n\n\ts.T().Logf(\"Loading config; env=%v,zone=%v,configDir=%v\\n\", env, zone, configDir)\n\n\tvar cfg config.Config\n\terr := config.Load(env, configDir, zone, &cfg)\n\tif err != nil {\n\t\ts.logger.Fatal(\"Config file corrupted.\", tag.Error(err))\n\t}\n\n\t// set up sqlite persistence layer and apply schema to sqlite db\n\ttestBase := pt.NewTestBaseWithSQL(s.T(), sqlite.GetTestClusterOption())\n\tcfg.Persistence = testBase.Config()\n\ttestBase.Setup()\n\n\ts.T().Logf(\"config=\\n%v\\n\", cfg.String())\n\n\tcfg.DynamicConfig.FileBased.Filepath = constructPathIfNeed(rootDir, cfg.DynamicConfig.FileBased.Filepath)\n\n\tif err := cfg.ValidateAndFillDefaults(); err != nil {\n\t\ts.logger.Fatal(\"config validation failed\", tag.Error(err))\n\t}\n\n\tlogger := testlogger.New(s.T())\n\n\tlifecycle := fxtest.NewLifecycle(s.T())\n\n\tvar daemons []common.Daemon\n\t// Shard distributor should be tested separately\n\tdistributorShortName := service.ShortName(service.ShardDistributor)\n\tservices := slices.DeleteFunc(service.ShortNames(service.List),\n\t\tfunc(s string) bool {\n\t\t\treturn s == distributorShortName\n\t\t})\n\n\tfor _, svc := range services {\n\t\tserver := newServer(svc, cfg, logger, dynamicconfig.NewNopClient(), tally.NoopScope, metrics.NewNoopMetricsClient())\n\t\tdaemons = append(daemons, server)\n\t\tserver.Start()\n\t}\n\n\ttimer := time.NewTimer(time.Second * 10)\n\n\t<-timer.C\n\ts.NoError(lifecycle.Stop(context.Background()))\n\tfor _, daemon := range daemons {\n\t\tdaemon.Stop()\n\t}\n}\n\nfunc TestSettingGettingZonalIsolationGroupsFromIG(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclient := dynamicconfig.NewMockClient(ctrl)\n\tclient.EXPECT().GetListValue(dynamicproperties.AllIsolationGroups, gomock.Any()).Return([]interface{}{\n\t\t\"zone-1\", \"zone-2\",\n\t}, nil)\n\n\tdc := dynamicconfig.NewCollection(client, log.NewNoop())\n\n\tassert.NotPanics(t, func() {\n\t\tfn := getFromDynamicConfig(resource.Params{\n\t\t\tLogger: log.NewNoop(),\n\t\t}, dc)\n\t\tout := fn()\n\t\tassert.Equal(t, []string{\"zone-1\", \"zone-2\"}, out)\n\t})\n}\n\nfunc TestSettingGettingZonalIsolationGroupsFromIGError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclient := dynamicconfig.NewMockClient(ctrl)\n\tclient.EXPECT().GetListValue(dynamicproperties.AllIsolationGroups, gomock.Any()).Return(nil, assert.AnError)\n\tdc := dynamicconfig.NewCollection(client, log.NewNoop())\n\n\tassert.NotPanics(t, func() {\n\t\tgetFromDynamicConfig(resource.Params{\n\t\t\tLogger: log.NewNoop(),\n\t\t}, dc)()\n\t})\n}\n"
  },
  {
    "path": "cmd/server/cadence/testdata/config/development.yaml",
    "content": "shardDistribution:\n  leaderStore:\n    storageParams:\n      endpoints:\n      - ${ETCD_ENDPOINTS:\"localhost:2379\"}\n  store:\n    storageParams:\n      endpoints:\n        - ${ETCD_ENDPOINTS:\"localhost:2379\"}\n  election:\n    leaderPeriod: 30s\n    maxRandomDelay: 5s\n    failedElectionCooldown: 30s\n  namespaces:\n    - name: cadence-matching-dev\n      type: ephemeral\n      mode: shadow\n  process:\n    period: 2s\n\npersistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence\"\n        connectTimeout: 2s # defaults to 2s if not defined\n        timeout: 5s # defaults to 10s if not defined\n        consistency: LOCAL_QUORUM # default value\n        serialConsistency: LOCAL_SERIAL # default value\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\nservices:\n  shard-distributor:\n    rpc:\n      port: 7941\n      grpcPort: 7943\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7942\n\nringpop:\n  name: cadence\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:7941\"]\n  maxJoinDuration: 30s\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      newInitialFailoverVersion: 1 # migrating to this new failover version\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n"
  },
  {
    "path": "cmd/server/go.mod",
    "content": "module github.com/uber/cadence/cmd/server\n\ngo 1.23.0\n\ntoolchain go1.23.4\n\n// build against the current code in the \"main\" (and gcloud) module, not a specific SHA.\n//\n// anyone outside this repo using this needs to ensure that both the \"main\" module and this module\n// are at the same SHA for consistency, but internally we can cheat by telling Go that it's at a\n// relative file path.\nreplace github.com/uber/cadence => ../..\n\nreplace github.com/uber/cadence/common/archiver/gcloud => ../../common/archiver/gcloud\n\nrequire (\n\tgithub.com/VividCortex/mysqlerr v1.0.0 // indirect\n\tgithub.com/aws/aws-sdk-go v1.54.12 // indirect\n\tgithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 // indirect\n\tgithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9 // indirect\n\tgithub.com/cristalhq/jwt/v3 v3.1.0 // indirect\n\tgithub.com/davecgh/go-spew v1.1.1 // indirect\n\tgithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da // indirect\n\tgithub.com/go-sql-driver/mysql v1.7.1 // indirect\n\tgithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1 // indirect\n\tgithub.com/gogo/protobuf v1.3.2 // indirect\n\tgithub.com/golang/mock v1.6.0 // indirect\n\tgithub.com/google/uuid v1.6.0 // indirect\n\tgithub.com/hashicorp/go-version v1.2.0 // indirect\n\tgithub.com/iancoleman/strcase v0.2.0 // indirect\n\tgithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee // indirect\n\tgithub.com/lib/pq v1.2.0 // indirect\n\tgithub.com/m3db/prometheus_client_golang v0.8.1 // indirect\n\tgithub.com/olivere/elastic v6.2.37+incompatible // indirect\n\tgithub.com/olivere/elastic/v7 v7.0.21 // indirect\n\tgithub.com/opentracing/opentracing-go v1.2.0 // indirect\n\tgithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709 // indirect\n\tgithub.com/robfig/cron v1.2.0 // indirect\n\tgithub.com/sirupsen/logrus v1.9.0 // indirect\n\tgithub.com/startreedata/pinot-client-go v0.2.0 // latest release supports pinot v0.12.0 which is also internal version\n\tgithub.com/stretchr/testify v1.10.0\n\tgithub.com/uber-go/tally v3.3.15+incompatible\n\tgithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4\n\tgithub.com/uber/ringpop-go v0.8.5 // indirect\n\tgithub.com/uber/tchannel-go v1.22.2 // indirect\n\tgithub.com/valyala/fastjson v1.4.1 // indirect\n\tgithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c // indirect\n\tgithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2 // indirect\n\tgo.uber.org/atomic v1.10.0 // indirect\n\tgo.uber.org/cadence v0.19.0\n\tgo.uber.org/config v1.4.0 // indirect\n\tgo.uber.org/fx v1.23.0\n\tgo.uber.org/multierr v1.10.0\n\tgo.uber.org/thriftrw v1.29.2 // indirect\n\tgo.uber.org/yarpc v1.70.3 // indirect\n\tgo.uber.org/zap v1.26.0\n\tgolang.org/x/net v0.40.0 // indirect\n\tgolang.org/x/sync v0.14.0 // indirect\n\tgolang.org/x/time v0.5.0 // indirect\n\tgolang.org/x/tools v0.22.0 // indirect\n\tgonum.org/v1/gonum v0.7.0 // indirect\n\tgoogle.golang.org/grpc v1.59.0 // indirect\n\tgopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19 // indirect\n\tgopkg.in/yaml.v2 v2.4.0 // indirect\n)\n\nrequire (\n\tgithub.com/uber/cadence v0.0.0-00010101000000-000000000000\n\tgithub.com/uber/cadence/common/archiver/gcloud v0.0.0-00010101000000-000000000000\n\tgo.uber.org/automaxprocs v1.6.0\n\tgo.uber.org/mock v0.5.0\n)\n\nrequire (\n\tgithub.com/IBM/sarama v1.45.2 // indirect\n\tgithub.com/coreos/go-semver v0.3.0 // indirect\n\tgithub.com/coreos/go-systemd/v22 v22.3.2 // indirect\n\tgithub.com/fatih/color v1.13.0 // indirect\n\tgithub.com/google/gofuzz v1.0.0 // indirect\n\tgithub.com/mattn/go-colorable v0.1.9 // indirect\n\tgithub.com/mattn/go-isatty v0.0.14 // indirect\n\tgithub.com/ncruces/go-sqlite3 v0.22.0 // indirect\n\tgithub.com/ncruces/julianday v1.0.0 // indirect\n\tgithub.com/opensearch-project/opensearch-go/v4 v4.1.0 // indirect\n\tgithub.com/pierrec/lz4/v4 v4.1.22 // indirect\n\tgithub.com/robfig/cron/v3 v3.0.1 // indirect\n\tgithub.com/tetratelabs/wazero v1.8.2 // indirect\n\tgithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 // indirect\n\tgo.etcd.io/etcd/api/v3 v3.5.5 // indirect\n\tgo.etcd.io/etcd/client/pkg/v3 v3.5.5 // indirect\n\tgo.etcd.io/etcd/client/v3 v3.5.5 // indirect\n)\n\nrequire (\n\tcloud.google.com/go v0.110.8 // indirect\n\tcloud.google.com/go/compute v1.23.1 // indirect\n\tcloud.google.com/go/compute/metadata v0.2.3 // indirect\n\tcloud.google.com/go/iam v1.1.3 // indirect\n\tcloud.google.com/go/storage v1.30.1 // indirect\n\tgithub.com/BurntSushi/toml v1.3.2 // indirect\n\tgithub.com/MicahParks/keyfunc/v2 v2.1.0 // indirect\n\tgithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 // indirect\n\tgithub.com/apache/thrift v0.17.0 // indirect\n\tgithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3 // indirect\n\tgithub.com/beorn7/perks v1.0.1 // indirect\n\tgithub.com/cespare/xxhash/v2 v2.2.0 // indirect\n\tgithub.com/cpuguy83/go-md2man/v2 v2.0.4 // indirect\n\tgithub.com/eapache/go-resiliency v1.7.0 // indirect\n\tgithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3 // indirect\n\tgithub.com/eapache/queue v1.1.0 // indirect\n\tgithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab // indirect\n\tgithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a // indirect\n\tgithub.com/fatih/structtag v1.2.0 // indirect\n\tgithub.com/go-zookeeper/zk v1.0.3 // indirect\n\tgithub.com/gogo/googleapis v1.3.2 // indirect\n\tgithub.com/gogo/status v1.1.0 // indirect\n\tgithub.com/golang-jwt/jwt/v5 v5.2.0 // indirect\n\tgithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect\n\tgithub.com/golang/protobuf v1.5.4 // indirect\n\tgithub.com/golang/snappy v0.0.4 // indirect\n\tgithub.com/google/s2a-go v0.1.4 // indirect\n\tgithub.com/googleapis/enterprise-certificate-proxy v0.2.4 // indirect\n\tgithub.com/googleapis/gax-go/v2 v2.12.0 // indirect\n\tgithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed // indirect\n\tgithub.com/hashicorp/errwrap v1.0.0 // indirect\n\tgithub.com/hashicorp/go-multierror v1.1.1 // indirect\n\tgithub.com/hashicorp/go-uuid v1.0.3 // indirect\n\tgithub.com/jcmturner/aescts/v2 v2.0.0 // indirect\n\tgithub.com/jcmturner/dnsutils/v2 v2.0.0 // indirect\n\tgithub.com/jcmturner/gofork v1.7.6 // indirect\n\tgithub.com/jcmturner/gokrb5/v8 v8.4.4 // indirect\n\tgithub.com/jcmturner/rpc/v2 v2.0.3 // indirect\n\tgithub.com/jessevdk/go-flags v1.4.0 // indirect\n\tgithub.com/jmespath/go-jmespath v0.4.0 // indirect\n\tgithub.com/jonboulle/clockwork v0.5.0 // indirect\n\tgithub.com/josharian/intern v1.0.0 // indirect\n\tgithub.com/kisielk/errcheck v1.5.0 // indirect\n\tgithub.com/klauspost/compress v1.18.0 // indirect\n\tgithub.com/m3db/prometheus_client_model v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_common v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_procfs v0.8.1 // indirect\n\tgithub.com/mailru/easyjson v0.7.7 // indirect\n\tgithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect\n\tgithub.com/pkg/errors v0.9.1 // indirect\n\tgithub.com/pmezard/go-difflib v1.0.0 // indirect\n\tgithub.com/prometheus/client_golang v1.11.1 // indirect\n\tgithub.com/prometheus/client_model v0.4.0 // indirect\n\tgithub.com/prometheus/common v0.26.0 // indirect\n\tgithub.com/prometheus/procfs v0.6.0 // indirect\n\tgithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475 // indirect\n\tgithub.com/russross/blackfriday/v2 v2.1.0 // indirect\n\tgithub.com/stretchr/objx v0.5.2 // indirect\n\tgithub.com/uber-common/bark v1.2.1 // indirect\n\tgithub.com/uber-go/mapdecode v1.0.0 // indirect\n\tgithub.com/urfave/cli/v2 v2.27.4\n\tgithub.com/xdg/stringprep v1.0.0 // indirect\n\tgo.opencensus.io v0.24.0 // indirect\n\tgo.uber.org/dig v1.18.0 // indirect\n\tgo.uber.org/net/metrics v1.3.0 // indirect\n\tgolang.org/x/crypto v0.38.0 // indirect\n\tgolang.org/x/exp v0.0.0-20231226003508-02704c960a9b // indirect\n\tgolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e // indirect\n\tgolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 // indirect\n\tgolang.org/x/mod v0.18.0 // indirect\n\tgolang.org/x/oauth2 v0.11.0 // indirect\n\tgolang.org/x/sys v0.33.0 // indirect\n\tgolang.org/x/text v0.25.0 // indirect\n\tgolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 // indirect\n\tgoogle.golang.org/api v0.128.0 // indirect\n\tgoogle.golang.org/appengine v1.6.7 // indirect\n\tgoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b // indirect\n\tgoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a // indirect\n\tgoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 // indirect\n\tgoogle.golang.org/protobuf v1.33.0 // indirect\n\tgopkg.in/inf.v0 v0.9.1 // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n\thonnef.co/go/tools v0.3.2 // indirect\n)\n"
  },
  {
    "path": "cmd/server/go.sum",
    "content": "cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.110.8 h1:tyNdfIxjzaWctIiLYOTalaLKZ17SI44SKFW26QbOhME=\ncloud.google.com/go v0.110.8/go.mod h1:Iz8AkXJf1qmxC3Oxoep8R1T36w8B92yU29PcBhHO5fk=\ncloud.google.com/go/compute v1.23.1 h1:V97tBoDaZHb6leicZ1G6DLK2BAaZLJ/7+9BB/En3hR0=\ncloud.google.com/go/compute v1.23.1/go.mod h1:CqB3xpmPKKt3OJpW2ndFIXnA9A4xAy/F3Xp1ixncW78=\ncloud.google.com/go/compute/metadata v0.2.3 h1:mg4jlk7mCAj6xXp9UJ4fjI9VUI5rubuGBW5aJ7UnBMY=\ncloud.google.com/go/compute/metadata v0.2.3/go.mod h1:VAV5nSsACxMJvgaAuX6Pk2AawlZn8kiOGuCv6gTkwuA=\ncloud.google.com/go/iam v1.1.3 h1:18tKG7DzydKWUnLjonWcJO6wjSCAtzh4GcRKlH/Hrzc=\ncloud.google.com/go/iam v1.1.3/go.mod h1:3khUlaBXfPKKe7huYgEpDn6FtgRyMEqbkvBxrQyY5SE=\ncloud.google.com/go/storage v1.30.1 h1:uOdMxAs8HExqBlnLtnQyP0YkvbiDpdGShGKtx6U/oNM=\ncloud.google.com/go/storage v1.30.1/go.mod h1:NfxhC0UJE1aXSx7CIIbCf7y9HKT7BiccwkR7+P7gN8E=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/BurntSushi/toml v0.4.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8=\ngithub.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/IBM/sarama v1.45.2 h1:8m8LcMCu3REcwpa7fCP6v2fuPuzVwXDAM2DOv3CBrKw=\ngithub.com/IBM/sarama v1.45.2/go.mod h1:ppaoTcVdGv186/z6MEKsMm70A5fwJfRTpstI37kVn3Y=\ngithub.com/MicahParks/keyfunc/v2 v2.1.0 h1:6ZXKb9Rp6qp1bDbJefnG7cTH8yMN1IC/4nf+GVjO99k=\ngithub.com/MicahParks/keyfunc/v2 v2.1.0/go.mod h1:rW42fi+xgLJ2FRRXAfNx9ZA8WpD4OeE/yHVMteCkw9k=\ngithub.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=\ngithub.com/VividCortex/mysqlerr v1.0.0 h1:5pZ2TZA+YnzPgzBfiUWGqWmKDVNBdrkf9g+DNe1Tiq8=\ngithub.com/VividCortex/mysqlerr v1.0.0/go.mod h1:xERx8E4tBhLvpjzdUyQiSfUxeMcATEQrflDAfXsqcAE=\ngithub.com/ajstarks/svgo v0.0.0-20180226025133-644b8db467af/go.mod h1:K08gAheRH3/J6wwsYMMT4xOr94bZjxIelGM0+d/wbFw=\ngithub.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 h1:kFOfPq6dUM1hTo4JG6LR5AXSUEsOjtdm0kw0FtQtMJA=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c=\ngithub.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=\ngithub.com/apache/thrift v0.17.0 h1:cMd2aj52n+8VoAtvSvLn4kDC3aZ6IAkBuqWQ2IDu7wo=\ngithub.com/apache/thrift v0.17.0/go.mod h1:OLxhMRJxomX+1I/KUw03qoV3mMz16BwaKI+d4fPBx7Q=\ngithub.com/aws/aws-sdk-go v1.34.13/go.mod h1:5zCpMtNQVjRREroY7sYe8lOMRSxkhG6MZveU8YkpAk0=\ngithub.com/aws/aws-sdk-go v1.54.12 h1:xPDB+GSBZq0rJbmDZF+EyfMbnWRyfEPcn7PZ7bJjXSw=\ngithub.com/aws/aws-sdk-go v1.54.12/go.mod h1:eRwEWoyTWFMVYVQzKMNHWP5/RV4xIUGMQfXQHfHkpNU=\ngithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3 h1:wOysYcIdqv3WnvwqFFzrYCFALPED7qkUGaLXu359GSc=\ngithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3/go.mod h1:UMqtWQTnOe4byzwe7Zhwh8f8s+36uszN51sJrSIZlTE=\ngithub.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=\ngithub.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=\ngithub.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=\ngithub.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=\ngithub.com/bitly/go-hostpool v0.0.0-20171023180738-a3a6125de932 h1:mXoPYz/Ul5HYEDvkta6I8/rnYM5gSdSV2tJ6XbZuEtY=\ngithub.com/bitly/go-hostpool v0.0.0-20171023180738-a3a6125de932/go.mod h1:NOuUCSz6Q9T7+igc/hlvDOUdtWKryOrtFyIVABv/p7k=\ngithub.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869 h1:DDGfHa7BWjL4YnC6+E63dPcxHo2sUxDIu8g3QgEJdRY=\ngithub.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869/go.mod h1:Ekp36dRnpXw/yCqJaO+ZrUyxD+3VXMFFr56k5XYrpB4=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b h1:AP/Y7sqYicnjGDfD5VcY4CIfh1hRXBUavxrvELjTiOE=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b/go.mod h1:ac9efd0D1fsDb3EJvhqgXRbFx7bs2wqZ10HQPeU8U/Q=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 h1:bXxS5/Z3/dfc8iFniQfgogNBomo0u+1//9eP+jl8GVo=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748/go.mod h1:l/bIBLeOl9eX+wxJAzxS4TveKRtAqlyDpHjhkfO0MEI=\ngithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9 h1:2rukpuvOpZryti4j58JHH5f0qJXxYdTYpkgNYx8iLdg=\ngithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9/go.mod h1:h4Tt1A91nOVAYsWdoxlXwKYPfxkxeTuRFkEMUQaRVBo=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=\ngithub.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=\ngithub.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=\ngithub.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk=\ngithub.com/cncf/udpa/go v0.0.0-20210930031921-04548b0d99d4/go.mod h1:6pvJx4me5XPnfI9Z40ddWsdw2W/uZgQLFXToKeRcDiI=\ngithub.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210805033703-aa0b78936158/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210922020428-25de7278fc84/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20211011173535-cb28da3451f1/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI=\ngithub.com/coreos/go-semver v0.3.0 h1:wkHLiw0WNATZnSG7epLsujiMCgPAc9xhjJ4tgnAxmfM=\ngithub.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=\ngithub.com/coreos/go-systemd/v22 v22.3.2 h1:D9/bQk5vlXQFZ6Kwuu6zaiXJ9oTPe68++AzAJc1DzSI=\ngithub.com/coreos/go-systemd/v22 v22.3.2/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4 h1:wfIWP927BUkWJb2NmU/kNDYIBTh/ziUX91+lVfRxZq4=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=\ngithub.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=\ngithub.com/cristalhq/jwt/v3 v3.1.0 h1:iLeL9VzB0SCtjCy9Kg53rMwTcrNm+GHyVcz2eUujz6s=\ngithub.com/cristalhq/jwt/v3 v3.1.0/go.mod h1:XOnIXst8ozq/esy5N1XOlSyQqBd+84fxJ99FK+1jgL8=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/dgryski/go-farm v0.0.0-20200201041132-a6ae2369ad13/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da h1:aIftn67I1fkbMa512G+w+Pxci9hJPB8oMnkcP3iZF38=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=\ngithub.com/eapache/go-resiliency v1.7.0 h1:n3NRTnBn5N0Cbi/IeOHuQn9s2UwVUH7Ga0ZWcP+9JTA=\ngithub.com/eapache/go-resiliency v1.7.0/go.mod h1:5yPzW0MIvSe0JDsv0v+DvcjEv2FyD6iZYSs1ZI+iQho=\ngithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3 h1:Oy0F4ALJ04o5Qqpdz8XLIpNA3WM/iSIXqxtqo7UGVws=\ngithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3/go.mod h1:YvSRo5mw33fLEx1+DlK6L2VV43tJt5Eyel9n9XBcR+0=\ngithub.com/eapache/queue v1.1.0 h1:YOEu7KNc61ntiQlcEeUIoDTJ2o8mQznoNvUhiigpIqc=\ngithub.com/eapache/queue v1.1.0/go.mod h1:6eCeP0CKFpHLu8blIFXhExK/dRa7WDZfr6jVFPTqq+I=\ngithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab h1:eTc1vwMHNg4WtS95PtYi3FFCKwlPjtN/Lw9IALTRtd8=\ngithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab/go.mod h1:YfzfFFoVP/catgzJb4IKIqXjX78Ha8FMSDh3ymbK86o=\ngithub.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210217033140-668b12f5399d/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.mod h1:hliV/p42l8fGbc6Y9bQ70uLwIvmJyVE5k4iMKlh8wCQ=\ngithub.com/envoyproxy/go-control-plane v0.9.10-0.20210907150352-cf90f659a021/go.mod h1:AFq3mo9L8Lqqiid3OhADV3RfLJnjiw63cSpi+fDTRC0=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a h1:yDWHCSQ40h88yih2JAcL6Ls/kVkSE8GFACTGVnMPruw=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a/go.mod h1:7Ga40egUymuWXxAe151lTNnCv97MddSOVsjpPPkityA=\ngithub.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w=\ngithub.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk=\ngithub.com/fatih/structtag v1.0.0/go.mod h1:IKitwq45uXL/yqi5mYghiD3w9H6eTOvI9vnk8tXMphA=\ngithub.com/fatih/structtag v1.2.0 h1:/OdNE99OxoI/PqaW/SuSK9uxxT3f/tcSZgon/ssNSx4=\ngithub.com/fatih/structtag v1.2.0/go.mod h1:mBJUNpUnHmRKrKlQQlmCrh5PuhftFbNv8Ys4/aAZl94=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568 h1:BHsljHzVlRcyQhjrss6TZTdY2VfCqZPbv5k3iBFa2ZQ=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568/go.mod h1:xEzjJPgXI435gkrCt3MPfRiAkVrwSbHsst4LCFVfpJc=\ngithub.com/fogleman/gg v1.2.1-0.20190220221249-0403632d5b90/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k=\ngithub.com/fortytw2/leaktest v1.3.0 h1:u8491cBMTQ8ft8aeV+adlcytMZylmA5nnwwkRZjI8vw=\ngithub.com/fortytw2/leaktest v1.3.0/go.mod h1:jDsjWgpAGjm2CA7WthBh/CdZYEPF31XHquHwclZch5g=\ngithub.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=\ngithub.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=\ngithub.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=\ngithub.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=\ngithub.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=\ngithub.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=\ngithub.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI=\ngithub.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=\ngithub.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=\ngithub.com/go-zookeeper/zk v1.0.3 h1:7M2kwOsc//9VeeFiPtf+uSJlVpU66x9Ba5+8XK7/TDg=\ngithub.com/go-zookeeper/zk v1.0.3/go.mod h1:nOB03cncLtlp4t+UAkGSV+9beXP/akpekBwL+UX1Qcw=\ngithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1 h1:px9qUCy/RNJNsfCam4m2IxWGxNuimkrioEF0vrrbPsg=\ngithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1/go.mod h1:3gM2c4D3AnkISwBxGnMMsS8Oy4y2lhbPRsH4xnJrHG8=\ngithub.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=\ngithub.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a/go.mod h1:gf4bu3Q80BeJ6H1S1vYPm8/ELATdvryBaNFGgqEef3s=\ngithub.com/gogo/googleapis v1.3.2 h1:kX1es4djPJrsDhY7aZKJy7aZasdcB5oSOEphMjSB53c=\ngithub.com/gogo/googleapis v1.3.2/go.mod h1:5YRNX2z1oM5gXdAkurHa942MDgEJyk02w4OecKY87+c=\ngithub.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.2.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.3.1/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=\ngithub.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=\ngithub.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=\ngithub.com/gogo/status v1.1.0 h1:+eIkrewn5q6b30y+g/BJINVVdi2xH7je5MPJ3ZPK3JA=\ngithub.com/gogo/status v1.1.0/go.mod h1:BFv9nrluPLmrS0EmGVvLaPNmRosr9KapBYd5/hpY1WM=\ngithub.com/golang-jwt/jwt/v5 v5.2.0 h1:d/ix8ftRUorsN+5eMIlF4T6J8CAt9rch3My2winC1Jw=\ngithub.com/golang-jwt/jwt/v5 v5.2.0/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=\ngithub.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.4.0/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=\ngithub.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=\ngithub.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.3-0.20190920234318-1680a479a2cf/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=\ngithub.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=\ngithub.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=\ngithub.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=\ngithub.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=\ngithub.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=\ngithub.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=\ngithub.com/google/gofuzz v1.0.0 h1:A8PeW59pxE9IoFRqBp37U+mSNaQoZ46F1f0f863XSXw=\ngithub.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=\ngithub.com/google/martian/v3 v3.3.2 h1:IqNFLAmvJOgVlpdEBiQbDc2EwKW77amAycfTuWKdfvw=\ngithub.com/google/martian/v3 v3.3.2/go.mod h1:oBOf6HBosgwRXnUGWUB05QECsc6uvmMiJ3+6W4l/CUk=\ngithub.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=\ngithub.com/google/s2a-go v0.1.4 h1:1kZ/sQM3srePvKs3tXAvQzo66XfcReoqFpIpIccE7Oc=\ngithub.com/google/s2a-go v0.1.4/go.mod h1:Ej+mSEMGRnqRzjc7VtF+jdBwYG5fuJfiZ8ELkjEwM0A=\ngithub.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.4 h1:uGy6JWR/uMIILU8wbf+OkstIrNiMjGpEIyhx8f6W7s4=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.4/go.mod h1:AwSRAtLfXpU5Nm3pW+v7rGDHp09LsPtGY9MduiEsR9k=\ngithub.com/googleapis/gax-go/v2 v2.12.0 h1:A+gCJKdRfqXkr+BIRGtZLibNXf0m1f9E4HG56etFpas=\ngithub.com/googleapis/gax-go/v2 v2.12.0/go.mod h1:y+aIqrI5eb1YGMVJfuV3185Ts/D7qKpsEkdD5+I6QGU=\ngithub.com/gorilla/securecookie v1.1.1/go.mod h1:ra0sb63/xPlUeL+yeDciTfxMRAA+MP+HVt/4epWDjd4=\ngithub.com/gorilla/sessions v1.2.1/go.mod h1:dk2InVEVJ0sfLlnXv9EAgkf6ecYs/i80K/zI+bUmuGM=\ngithub.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw=\ngithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed h1:5upAirOpQc1Q53c0bnx2ufif5kANL7bfZWcc6VJWJd8=\ngithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed/go.mod h1:tMWxXQ9wFIaZeTI9F+hmhFiGpFmhOHzyShyFUhRm0H4=\ngithub.com/hashicorp/errwrap v1.0.0 h1:hLrqtEDnRye3+sgx6z4qVLNuviH3MR5aQ0ykNJa/UYA=\ngithub.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=\ngithub.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=\ngithub.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=\ngithub.com/hashicorp/go-uuid v1.0.2/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-uuid v1.0.3 h1:2gKiV6YVmrJ1i2CKKa9obLvRieoRGviZFL26PcT/Co8=\ngithub.com/hashicorp/go-uuid v1.0.3/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-version v1.2.0 h1:3vNe/fWF5CBgRIguda1meWhsZHy3m8gCJ5wx+dIzX/E=\ngithub.com/hashicorp/go-version v1.2.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=\ngithub.com/iancoleman/strcase v0.2.0 h1:05I4QRnGpI0m37iZQRuskXh+w77mr6Z41lwQzuHLwW0=\ngithub.com/iancoleman/strcase v0.2.0/go.mod h1:iwCmte+B7n89clKwxIoIXy/HfoL7AsD47ZCWhYzw7ho=\ngithub.com/jcmturner/aescts/v2 v2.0.0 h1:9YKLH6ey7H4eDBXW8khjYslgyqG2xZikXP0EQFKrle8=\ngithub.com/jcmturner/aescts/v2 v2.0.0/go.mod h1:AiaICIRyfYg35RUkr8yESTqvSy7csK90qZ5xfvvsoNs=\ngithub.com/jcmturner/dnsutils/v2 v2.0.0 h1:lltnkeZGL0wILNvrNiVCR6Ro5PGU/SeBvVO/8c/iPbo=\ngithub.com/jcmturner/dnsutils/v2 v2.0.0/go.mod h1:b0TnjGOvI/n42bZa+hmXL+kFJZsFT7G4t3HTlQ184QM=\ngithub.com/jcmturner/gofork v1.7.6 h1:QH0l3hzAU1tfT3rZCnW5zXl+orbkNMMRGJfdJjHVETg=\ngithub.com/jcmturner/gofork v1.7.6/go.mod h1:1622LH6i/EZqLloHfE7IeZ0uEJwMSUyQ/nDd82IeqRo=\ngithub.com/jcmturner/goidentity/v6 v6.0.1 h1:VKnZd2oEIMorCTsFBnJWbExfNN7yZr3EhJAxwOkZg6o=\ngithub.com/jcmturner/goidentity/v6 v6.0.1/go.mod h1:X1YW3bgtvwAXju7V3LCIMpY0Gbxyjn/mY9zx4tFonSg=\ngithub.com/jcmturner/gokrb5/v8 v8.4.4 h1:x1Sv4HaTpepFkXbt2IkL29DXRf8sOfZXo8eRKh687T8=\ngithub.com/jcmturner/gokrb5/v8 v8.4.4/go.mod h1:1btQEpgT6k+unzCwX1KdWMEwPPkkgBtP+F6aCACiMrs=\ngithub.com/jcmturner/rpc/v2 v2.0.3 h1:7FXXj8Ti1IaVFpSAziCZWNzbNuZmnvw/i6CqLNdWfZY=\ngithub.com/jcmturner/rpc/v2 v2.0.3/go.mod h1:VUJYCIDm3PVOEHw8sgt091/20OJjskO/YJki3ELg/Hc=\ngithub.com/jessevdk/go-flags v1.4.0 h1:4IU2WS7AumrZ/40jfhf4QVDMsQwqA7VEHozFRrGARJA=\ngithub.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=\ngithub.com/jmespath/go-jmespath v0.3.0/go.mod h1:9QtRXoHjLGCJ5IBSaohpXITPlowMeeYCZ7fLUTSywik=\ngithub.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=\ngithub.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=\ngithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee h1:59lyMGvZusByi7Rvctn8cxdVAjhiOnqCv3G5DrYApYQ=\ngithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee/go.mod h1:ClpsPFzLpSBl7MvJ+BhV0JHz4vmKRBarpvZ9644v9Oo=\ngithub.com/jonboulle/clockwork v0.5.0 h1:Hyh9A8u51kptdkR+cqRpT1EebBwTn1oK9YfGYbdFz6I=\ngithub.com/jonboulle/clockwork v0.5.0/go.mod h1:3mZlmanh0g2NDKO5TWZVJAfofYk64M7XN3SzBPjZF60=\ngithub.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY=\ngithub.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=\ngithub.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=\ngithub.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=\ngithub.com/json-iterator/go v1.1.8/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.11/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=\ngithub.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=\ngithub.com/jung-kurt/gofpdf v1.0.3-0.20190309125859-24315acbbda5/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes=\ngithub.com/kisielk/errcheck v1.2.0/go.mod h1:/BMXB+zMLi60iA8Vv6Ksmxu/1UDYcXs4uQLJ+jE2L00=\ngithub.com/kisielk/errcheck v1.5.0 h1:e8esj/e4R+SAOwFwN+n3zr0nYeCyeweozKfO23MvHzY=\ngithub.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=\ngithub.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=\ngithub.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=\ngithub.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=\ngithub.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=\ngithub.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=\ngithub.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=\ngithub.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=\ngithub.com/lib/pq v1.2.0 h1:LXpIM/LZ5xGFhOpXAQUIMM1HdyqzVYM13zNdjCEEcA0=\ngithub.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=\ngithub.com/m3db/prometheus_client_golang v0.8.1 h1:t7w/tcFws81JL1j5sqmpqcOyQOpH4RDOmIe3A3fdN3w=\ngithub.com/m3db/prometheus_client_golang v0.8.1/go.mod h1:8R/f1xYhXWq59KD/mbRqoBulXejss7vYtYzWmruNUwI=\ngithub.com/m3db/prometheus_client_model v0.1.0 h1:cg1+DiuyT6x8h9voibtarkH1KT6CmsewBSaBhe8wzLo=\ngithub.com/m3db/prometheus_client_model v0.1.0/go.mod h1:Qfsxn+LypxzF+lNhak7cF7k0zxK7uB/ynGYoj80zcD4=\ngithub.com/m3db/prometheus_common v0.1.0 h1:YJu6eCIV6MQlcwND24cRG/aRkZDX1jvYbsNNs1ZYr0w=\ngithub.com/m3db/prometheus_common v0.1.0/go.mod h1:EBmDQaMAy4B8i+qsg1wMXAelLNVbp49i/JOeVszQ/rs=\ngithub.com/m3db/prometheus_procfs v0.8.1 h1:LsxWzVELhDU9sLsZTaFLCeAwCn7bC7qecZcK4zobs/g=\ngithub.com/m3db/prometheus_procfs v0.8.1/go.mod h1:N8lv8fLh3U3koZx1Bnisj60GYUMDpWb09x1R+dmMOJo=\ngithub.com/mailru/easyjson v0.7.6/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=\ngithub.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0=\ngithub.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=\ngithub.com/mattn/go-colorable v0.1.9 h1:sqDoxXbdeALODt0DAeJCVp38ps9ZogZEAXjus69YV3U=\ngithub.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=\ngithub.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=\ngithub.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=\ngithub.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=\ngithub.com/mattn/go-shellwords v1.0.10/go.mod h1:EZzvwXDESEeg03EKmM+RmDnNOPKG4lLtQsUlTZDWQ8Y=\ngithub.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=\ngithub.com/mattn/go-sqlite3 v1.11.0 h1:LDdKkqtYlom37fkvqs8rMPFKAMe8+SgjbwZ6ex1/A/Q=\ngithub.com/mattn/go-sqlite3 v1.11.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 h1:I0XW9+e1XWDxdcEniV4rQAIOPUGDq67JSCiRCgGCZLI=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=\ngithub.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/ncruces/go-sqlite3 v0.22.0 h1:FkGSBhd0TY6e66k1LVhyEpA+RnG/8QkQNed5pjIk4cs=\ngithub.com/ncruces/go-sqlite3 v0.22.0/go.mod h1:ueXOZXYZS2OFQirCU3mHneDwJm5fGKHrtccYBeGEV7M=\ngithub.com/ncruces/julianday v1.0.0 h1:fH0OKwa7NWvniGQtxdJRxAgkBMolni2BjDHaWTxqt7M=\ngithub.com/ncruces/julianday v1.0.0/go.mod h1:Dusn2KvZrrovOMJuOt0TNXL6tB7U2E8kvza5fFc9G7g=\ngithub.com/olivere/elastic v6.2.37+incompatible h1:UfSGJem5czY+x/LqxgeCBgjDn6St+z8OnsCuxwD3L0U=\ngithub.com/olivere/elastic v6.2.37+incompatible/go.mod h1:J+q1zQJTgAz9woqsbVRqGeB5G1iqDKVBWLNSYW8yfJ8=\ngithub.com/olivere/elastic/v7 v7.0.21 h1:58a2pMlLketCsLyKg8kJNJG+OZIFKrSQXX6gJBpqqlg=\ngithub.com/olivere/elastic/v7 v7.0.21/go.mod h1:Kh7iIsXIBl5qRQOBFoylCsXVTtye3keQU2Y/YbR7HD8=\ngithub.com/opensearch-project/opensearch-go/v4 v4.1.0 h1:YXNaMpMU0PC7suGyP13EuczkDT3K54QajgDnLKCZAz8=\ngithub.com/opensearch-project/opensearch-go/v4 v4.1.0/go.mod h1:aSTMFGSLEoiG19US6Oo5udvWCjHap3mRcWBNV8rAFak=\ngithub.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=\ngithub.com/opentracing/opentracing-go v1.2.0 h1:uEJPy/1a5RIPAJ0Ov+OIO8OxWu77jEv+1B0VhjKrZUs=\ngithub.com/opentracing/opentracing-go v1.2.0/go.mod h1:GxEUsuufX4nBwe+T+Wl9TAgYrxe9dPLANfrWvHYVTgc=\ngithub.com/pborman/uuid v0.0.0-20160209185913-a97ce2ca70fa/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709 h1:zNBQb37RGLmJybyMcs983HfUfpkw9OTFD9tbBfAViHE=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pierrec/lz4/v4 v4.1.22 h1:cKFw6uJDK+/gfw5BcDL0JL5aBsAFdsIT18eRtLj7VIU=\ngithub.com/pierrec/lz4/v4 v4.1.22/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=\ngithub.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g=\ngithub.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a h1:AA9vgIBDjMHPC2McaGPojgV2dcI78ZC0TLNhYCXEKH8=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a/go.mod h1:lzZQ3Noex5pfAy7mkAeCjcBDteYU85uWWnJ/y6gKU8k=\ngithub.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=\ngithub.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo=\ngithub.com/prometheus/client_golang v1.3.0/go.mod h1:hJaj2vgQTGQmVCsAACORcieXFeDPbaTKGT+JTgUa3og=\ngithub.com/prometheus/client_golang v1.4.1/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=\ngithub.com/prometheus/client_golang v1.7.1/go.mod h1:PY5Wy2awLA44sXw4AOSfFBetzPP4j5+D6mVACh+pe2M=\ngithub.com/prometheus/client_golang v1.11.1 h1:+4eQaD7vAZ6DsfsxB15hbE0odUjGI5ARs9yskGu1v4s=\ngithub.com/prometheus/client_golang v1.11.1/go.mod h1:Z6t4BnS23TR94PD6BsDNk8yVqroYurpAkEiz0P2BEV0=\ngithub.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=\ngithub.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.1.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.4.0 h1:5lQXD3cAg1OXBf4Wq03gTrXHeaV0TQvGfUooCfx1yqY=\ngithub.com/prometheus/client_model v0.4.0/go.mod h1:oMQmHW1/JoDwqLtg57MGgP/Fb1CJEYF2imWWhWtMkYU=\ngithub.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=\ngithub.com/prometheus/common v0.7.0/go.mod h1:DjGbpBbp5NYNiECxcL/VnbXCCaQpKd3tt26CguLLsqA=\ngithub.com/prometheus/common v0.8.0/go.mod h1:PC/OgXc+UN7B4ALwvn1yzVZmVwvhXp5JsbBv6wSv6i0=\ngithub.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4=\ngithub.com/prometheus/common v0.10.0/go.mod h1:Tlit/dnDKsSWFlCLTWaA1cyBgKHSMdTB80sz/V91rCo=\ngithub.com/prometheus/common v0.26.0 h1:iMAkS2TDoNWnKM+Kopnx/8tnEStIfpYA0ur0xQzzhMQ=\ngithub.com/prometheus/common v0.26.0/go.mod h1:M7rCNAaPfAosfx8veZJCuw84e35h3Cfd9VFqTh1DIvc=\ngithub.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=\ngithub.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=\ngithub.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.0.9/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.1.3/go.mod h1:lV6e/gmhEcM9IjHGsFOCxxuZ+z1YqCvr4OA4YeYWdaU=\ngithub.com/prometheus/procfs v0.6.0 h1:mxy4L2jP6qMonqmq+aTtOx1ifVWUgG/TAmntgbh3xv4=\ngithub.com/prometheus/procfs v0.6.0/go.mod h1:cz+aTbrPOrUb4q7XlbU9ygM+/jj0fzG6c1xBZuNvfVA=\ngithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475 h1:N/ElC8H3+5XpJzTSTfLsJV/mx9Q9g7kxmchpfZyxgzM=\ngithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475/go.mod h1:bCqnVzQkZxMG4s8nGwiZ5l3QUCyqpo9Y+/ZMZ9VjZe4=\ngithub.com/robfig/cron v1.2.0 h1:ZjScXvvxeQ63Dbyxy76Fj3AT3Ut0aKsyd2/tl3DTMuQ=\ngithub.com/robfig/cron v1.2.0/go.mod h1:JGuDeoQd7Z6yL4zQhZ3OPEVHB7fL6Ka6skscFHfmt2k=\ngithub.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=\ngithub.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=\ngithub.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=\ngithub.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.6.1 h1:/FiVV8dS/e+YqF2JvO3yXRFbBLTIuSDkuC7aBOAvL+k=\ngithub.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=\ngithub.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=\ngithub.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af h1:EiWVfh8mr40yFZEui2oF0d45KgH48PkB2H0Z0GANvSI=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af/go.mod h1:Vrkh1pnjV9Bl8c3P9zH0/D4NlOHWP5d4/hF4YTULaec=\ngithub.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=\ngithub.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=\ngithub.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=\ngithub.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=\ngithub.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=\ngithub.com/smartystreets/assertions v1.1.1/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=\ngithub.com/smartystreets/go-aws-auth v0.0.0-20180515143844-0c1422d1fdb9/go.mod h1:SnhjPscd9TpLiy1LpzGSKh3bXCfxxXuqd9xmQJy3slM=\ngithub.com/smartystreets/gunit v1.4.2/go.mod h1:ZjM1ozSIMJlAz/ay4SG8PeKF00ckUp+zMHZXV9/bvak=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=\ngithub.com/startreedata/pinot-client-go v0.2.0 h1:Pv4W3HgxxGbB9GogRwNqfNyqPrOpScZuhQRc9kLM90A=\ngithub.com/startreedata/pinot-client-go v0.2.0/go.mod h1:vTz6Bu4dWIQIsfUoqFtgMV2QqBjeuSaDA8vxkOoYnLg=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25 h1:7z3LSn867ex6VSaahyKadf4WtSsJIgne6A1WLOAGM8A=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25/go.mod h1:lbP8tGiBjZ5YWIc2fzuRpTaz0b/53vT6PEs3QuAWzuU=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=\ngithub.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=\ngithub.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=\ngithub.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=\ngithub.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=\ngithub.com/tetratelabs/wazero v1.8.2 h1:yIgLR/b2bN31bjxwXHD8a3d+BogigR952csSDdLYEv4=\ngithub.com/tetratelabs/wazero v1.8.2/go.mod h1:yAI0XTsMBhREkM/YDAK/zNou3GoiAce1P6+rp/wQhjs=\ngithub.com/tidwall/gjson v1.17.1 h1:wlYEnwqAHgzmhNUFfw7Xalt2JzQvsMx2Se4PcoFCT/U=\ngithub.com/tidwall/gjson v1.17.1/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=\ngithub.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=\ngithub.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=\ngithub.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4=\ngithub.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=\ngithub.com/tidwall/sjson v1.2.5 h1:kLy8mja+1c9jlljvWTlSazM7cKDRfJuR/bOJhcY5NcY=\ngithub.com/tidwall/sjson v1.2.5/go.mod h1:Fvgq9kS/6ociJEDnK0Fk1cpYF4FIW6ZF7LAe+6jwd28=\ngithub.com/uber-common/bark v1.2.1 h1:cREJ9b7CpTjwZr0/5wV82fXlitoCIEHHnt9WkQ4lIk0=\ngithub.com/uber-common/bark v1.2.1/go.mod h1:g0ZuPcD7XiExKHynr93Q742G/sbrdVQkghrqLGOoFuY=\ngithub.com/uber-go/mapdecode v1.0.0 h1:euUEFM9KnuCa1OBixz1xM+FIXmpixyay5DLymceOVrU=\ngithub.com/uber-go/mapdecode v1.0.0/go.mod h1:b5nP15FwXTgpjTjeA9A2uTHXV5UJCl4arwKpP0FP1Hw=\ngithub.com/uber-go/tally v3.3.12+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber-go/tally v3.3.15+incompatible h1:9hLSgNBP28CjIaDmAuRTq9qV+UZY+9PcvAkXO4nNMwg=\ngithub.com/uber-go/tally v3.3.15+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber/cadence-idl v0.0.0-20211111101836-d6b70b60eb8c/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4 h1:1kAlcf7STXalFhDzv6vUz3XZtGOXQfwDF7IBKHu8GwI=\ngithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible h1:NHcubEkVbahf9t3p75TOCR83gdUHXjRJvjoBh1yACsM=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible/go.mod h1:WVhlPFC8FDjOFMMWRy2pZqQJSXxYSwNYOkTr/Z6d3Kk=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible h1:MxZXOiR2JuoANZ3J6DE/U0kSFv/eJ/GfSYVCjK7dyaw=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible/go.mod h1:ComeNDZlWwrWnDv8aPp0Ba6+uUTzImX/AauajbLI56U=\ngithub.com/uber/ringpop-go v0.8.5 h1:aBa/SHmmFRcAXA63k7uBheoTL8tCmH7L+OgktB1AF/o=\ngithub.com/uber/ringpop-go v0.8.5/go.mod h1:zVI6eGO6L7pG14GkntHsSOfmUAWQ7B4lvmzly4IT4ls=\ngithub.com/uber/tchannel-go v1.16.0/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/uber/tchannel-go v1.22.2 h1:NKA5FVESYh6Ij6V+tujK+IFZnBKDyUHdsBY264UYhgk=\ngithub.com/uber/tchannel-go v1.22.2/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/urfave/cli/v2 v2.27.4 h1:o1owoI+02Eb+K107p27wEX9Bb8eqIoZCfLXloLUSWJ8=\ngithub.com/urfave/cli/v2 v2.27.4/go.mod h1:m4QzxcD2qpra4z7WhzEGn74WZLViBnMpb1ToCAKdGRQ=\ngithub.com/valyala/fastjson v1.4.1 h1:hrltpHpIpkaxll8QltMU8c3QZ5+qIiCL8yKqPFJI/yE=\ngithub.com/valyala/fastjson v1.4.1/go.mod h1:nV6MsjxL2IMJQUoHDIrjEI7oLyeqK6aBD7EFWPsvP8o=\ngithub.com/wI2L/jsondiff v0.6.0 h1:zrsH3FbfVa3JO9llxrcDy/XLkYPLgoMX6Mz3T2PP2AI=\ngithub.com/wI2L/jsondiff v0.6.0/go.mod h1:D6aQ5gKgPF9g17j+E9N7aasmU1O+XvfmWm1y8UMmNpw=\ngithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c h1:u40Z8hqBAAQyv+vATcGgV0YCnDjqSL7/q/JyPhhJSPk=\ngithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c/go.mod h1:lB8K/P019DLNhemzwFU4jHLhdvlE6uDZjXFejJXr49I=\ngithub.com/xdg/stringprep v1.0.0 h1:d9X0esnoa3dFsV0FG35rAT0RIhYFlPq7MiP+DW89La0=\ngithub.com/xdg/stringprep v1.0.0/go.mod h1:Jhud4/sHMO4oL310DaZAKk9ZaJ08SJfe+sJh0HrGL1Y=\ngithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 h1:gEOO8jv9F4OT7lGCjxCBTO/36wtF6j2nSip77qHd4x4=\ngithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1/go.mod h1:Ohn+xnUBiLI6FVj/9LpzZWtj1/D6lUovWYBkxHVV3aM=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2 h1:zzrxE1FKn5ryBNl9eKOeqQ58Y/Qpo3Q9QNxKHX5uzzQ=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2/go.mod h1:hzfGeIUDq/j97IG+FhNqkowIyEcD88LrW6fyU3K3WqY=\ngithub.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=\ngo.etcd.io/etcd/api/v3 v3.5.5 h1:BX4JIbQ7hl7+jL+g+2j5UAr0o1bctCm6/Ct+ArBGkf0=\ngo.etcd.io/etcd/api/v3 v3.5.5/go.mod h1:KFtNaxGDw4Yx/BA4iPPwevUTAuqcsPxzyX8PHydchN8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.5 h1:9S0JUVvmrVl7wCF39iTQthdaaNIiAaQbmK75ogO6GU8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.5/go.mod h1:ggrwbk069qxpKPq8/FKkQ3Xq9y39kbFR4LnKszpRXeQ=\ngo.etcd.io/etcd/client/v3 v3.5.5 h1:q++2WTJbUgpQu4B6hCuT7VkdwaTP7Qz6Daak3WzbrlI=\ngo.etcd.io/etcd/client/v3 v3.5.5/go.mod h1:aApjR4WGlSumpnJ2kloS75h6aHUmAyaPLjHMxpc7E7c=\ngo.opencensus.io v0.22.4/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=\ngo.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=\ngo.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=\ngo.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=\ngo.uber.org/atomic v1.3.2/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=\ngo.uber.org/atomic v1.5.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.5.1/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/atomic v1.10.0 h1:9qC72Qh0+3MqyJbAn8YU5xVq1frD8bn3JtD2oXtafVQ=\ngo.uber.org/atomic v1.10.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=\ngo.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs=\ngo.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8=\ngo.uber.org/cadence v0.19.0 h1:EvDXwIJ0lAAxL2i8ne/vG/TeoJM6xkAyqgTRFmIWG+c=\ngo.uber.org/cadence v0.19.0/go.mod h1:s91dOf0kcJbumPscRIVFV/4Xq/exhefzpXmnDiRRTxs=\ngo.uber.org/config v1.4.0 h1:upnMPpMm6WlbZtXoasNkK4f0FhxwS+W4Iqz5oNznehQ=\ngo.uber.org/config v1.4.0/go.mod h1:aCyrMHmUAc/s2h9sv1koP84M9ZF/4K+g2oleyESO/Ig=\ngo.uber.org/dig v1.8.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.10.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.18.0 h1:imUL1UiY0Mg4bqbFfsRQO5G4CGRBec/ZujWTvSVp3pw=\ngo.uber.org/dig v1.18.0/go.mod h1:Us0rSJiThwCv2GteUN0Q7OKvU7n5J4dxZ9JKUXozFdE=\ngo.uber.org/fx v1.10.0/go.mod h1:vLRicqpG/qQEzno4SYU86iCwfT95EZza+Eba0ItuxqY=\ngo.uber.org/fx v1.13.1/go.mod h1:bREWhavnedxpJeTq9pQT53BbvwhUv7TcpsOqcH4a+3w=\ngo.uber.org/fx v1.23.0 h1:lIr/gYWQGfTwGcSXWXu4vP5Ws6iqnNEIY+F/aFzCKTg=\ngo.uber.org/fx v1.23.0/go.mod h1:o/D9n+2mLP6v1EG+qsdT1O8wKopYAsqZasju97SDFCU=\ngo.uber.org/goleak v0.10.0/go.mod h1:VCZuO8V8mFPlL0F5J5GK1rtHV3DrFcQ1R8ryq7FK0aI=\ngo.uber.org/goleak v1.0.0/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=\ngo.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=\ngo.uber.org/goleak v1.2.0/go.mod h1:XJYK+MuIchqpmGmUSAzotztawfKvYLUIgg7guXrwVUo=\ngo.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=\ngo.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=\ngo.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=\ngo.uber.org/multierr v1.3.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.4.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=\ngo.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=\ngo.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=\ngo.uber.org/net/metrics v1.3.0 h1:iRLPuVecNYf/wIV+mQaA4IgN8ghifu3q1B4IT6HfwyY=\ngo.uber.org/net/metrics v1.3.0/go.mod h1:pEQrSDGNWT5IVpekWzee5//uHjI4gmgZFkobfw3bv8I=\ngo.uber.org/thriftrw v1.25.0/go.mod h1:IcIfSeZgc59AlYb0xr0DlDKIdD7SgjnFpG9BXCPyy9g=\ngo.uber.org/thriftrw v1.29.2 h1:pRuFLzbGvTcnYwGSjizWRHlbJUzGhu84sRiL1h1kUd8=\ngo.uber.org/thriftrw v1.29.2/go.mod h1:YcjXveberDd28/Bs34SwHy3yu85x/jB4UA2gIcz/Eo0=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee h1:0mgffUl7nfd+FpvXMVz4IDEaUSmT1ysygQC7qYo7sG4=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA=\ngo.uber.org/yarpc v1.55.0/go.mod h1:V2JUPDWHYGNpvyuroYjf0KFjwvBCtcFJLuvZqv7TWA0=\ngo.uber.org/yarpc v1.70.3 h1:yykHwzRD9/bgDtlOWoVuXbSZoU91Id2dWJO1CDSRHnI=\ngo.uber.org/yarpc v1.70.3/go.mod h1:EH6I6K1HxBbOxZIJfhdDf+H+cvXPHmJyRvpfPqES20U=\ngo.uber.org/zap v1.9.1/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=\ngo.uber.org/zap v1.13.0/go.mod h1:zwrFLgMcdUuIBviXEYEH1YKNaOBnKXsx2IPda5bBwHM=\ngo.uber.org/zap v1.17.0/go.mod h1:MXVU+bhUf/A7Xi2HNOnopQOrmycQ5Ih87HtOu4q5SSo=\ngo.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo=\ngo.uber.org/zap v1.26.0/go.mod h1:dtElttAiwGvoJ/vj4IwHBS/gXsEu/pZ50mUIRWuG0so=\ngolang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.0.0-20220314234659-1baeb1ce4c0b/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=\ngolang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58=\ngolang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8=\ngolang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw=\ngolang.org/x/exp v0.0.0-20180321215751-8460e604b9de/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20180807140117-3d87b88a115f/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190125153040-c74c464bbbf2/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b h1:kLiC65FbiHWFAOu+lxwNPujcsl8VYyTYYEZnsOO1WK4=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b/go.mod h1:iRJReGqOEeBhDZGkGbynYwcHlctCvnjTYIamk7uXpHI=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e h1:qyrTQ++p1afMkO4DPEeLGq/3oTsdlvdH4vqZUBWzUKM=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e/go.mod h1:AbB0pIl9nAr9wVwH+Z2ZpaocVmF5I4GyWCDIsVjR0bk=\ngolang.org/x/image v0.0.0-20180708004352-c73c2afc3b81/go.mod h1:ux5Hcp/YLpHSI86hEcLt0YII63i6oz57MZXIpbrjZUs=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20191125180803-fdd1cda4f05f/go.mod h1:5qLYkcX4OjUUV8bRuDixDT3tpyyb+LUpUlRWLxfhWrs=\ngolang.org/x/lint v0.0.0-20200130185559-910be7a94367/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 h1:VLliZ0d+/avPrXXH+OakdXhpJuEoBZuwh1m2j7U6Iug=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc=\ngolang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=\ngolang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=\ngolang.org/x/mod v0.18.0 h1:5+9lSbEzPSdWkH32vYPBwEpX8KwDbM52Ud9xBUvNlb0=\ngolang.org/x/mod v0.18.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=\ngolang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200114155413-6afb5195e5aa/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=\ngolang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=\ngolang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=\ngolang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=\ngolang.org/x/net v0.40.0 h1:79Xs7wF06Gbdcg4kdCCIQArK11Z1hr5POQ6+fIYHNuY=\ngolang.org/x/net v0.40.0/go.mod h1:y0hY0exeL2Pku80/zKK7tpntoX23cqL3Oa6njdgRtds=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.11.0 h1:vPL4xzxBM4niKCW6g9whtaWVXTJf1U5e4aZxxFx/gbU=\ngolang.org/x/oauth2 v0.11.0/go.mod h1:LdF7O/8bLR/qWK9DrpXmbHLTouvRHK0SgJl0GmDBchk=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.14.0 h1:woo0S4Yywslg6hp4eUFjTVOyKt0RookbpAHG4c1HmhQ=\ngolang.org/x/sync v0.14.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191220142924-d4481acd189f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200117145432-59e60aa80a0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200212091648-12a6c2dcc1e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200615200032-f1bc736245b1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200625212154-ddb9806d33ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210403161142-5e06dd20ab57/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=\ngolang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=\ngolang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=\ngolang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=\ngolang.org/x/text v0.0.0-20170915032832-14c0d48ead0c/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=\ngolang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=\ngolang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=\ngolang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=\ngolang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=\ngolang.org/x/time v0.0.0-20170927054726-6dc17368e09b/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=\ngolang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180525024113-a5b4c53f6e8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190206041539-40960b6deb8e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191030062658-86caa796c7ab/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191104232314-dc038396d1f0/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191108193012-7d206e10da11/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191114200427-caa0b0f7d508/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191125144606-a911d9008d1f/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191226212025-6b505debf4bc/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117215004-fe56e6335763/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200216192241-b320d3a0f5a2/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.2/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.11-0.20220513221640-090b14e8501f/go.mod h1:SgwaegtQh8clINPpECJMqnxLv9I09HLqnW3RMqW0CA4=\ngolang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=\ngolang.org/x/tools v0.22.0 h1:gqSGLZqv+AI9lIQzniJ0nZDRG5GBPsSi+DRNHWNz6yA=\ngolang.org/x/tools v0.22.0/go.mod h1:aCwcsjqvq7Yqt6TNyX7QMU2enbQ/Gt0bo6krSeEri+c=\ngolang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 h1:H2TDz8ibqkAF6YGhCdN3jS9O0/s90v0rJh3X/OLHEUk=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2/go.mod h1:K8+ghG5WaK9qNqU5K3HdILfMLy1f3aNYFI/wnl100a8=\ngonum.org/v1/gonum v0.0.0-20180816165407-929014505bf4/go.mod h1:Y+Yx5eoAFn32cQvJDxZx5Dpnq+c3wtXuadVZAcxbbBo=\ngonum.org/v1/gonum v0.7.0 h1:Hdks0L0hgznZLG9nzXb8vZ0rRvqNvAcgAp84y7Mwkgw=\ngonum.org/v1/gonum v0.7.0/go.mod h1:L02bwd0sqlsvRv41G7wGWFCsVNZFv/k1xzGIxeANHGM=\ngonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0 h1:OE9mWmgKkjJyEmDAAtGMPjXu+YNeGvK9VTSHY6+Qihc=\ngonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0/go.mod h1:wa6Ws7BG/ESfp6dHfk7C6KdzKA7wR7u/rKwOGE66zvw=\ngonum.org/v1/plot v0.0.0-20190515093506-e2840ee46a6b/go.mod h1:Wt8AAjI+ypCyYX3nZBvf6cAIx93T+c/OS2HFAYskSZc=\ngoogle.golang.org/api v0.128.0 h1:RjPESny5CnQRn9V6siglged+DZCgfu9l6mO9dkX9VOg=\ngoogle.golang.org/api v0.128.0/go.mod h1:Y611qgqaE92On/7g65MQgxYul3c0rEB894kniWLY750=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/appengine v1.6.7 h1:FZR1q0exgwxzPzp/aF+VccGrSfxfPpkBqjIIEq3ru6c=\ngoogle.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=\ngoogle.golang.org/genproto v0.0.0-20180518175338-11a468237815/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20200212174721-66ed5ce911ce/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/genproto v0.0.0-20210602131652-f16073e35f0c/go.mod h1:UODoCrxHCcBojKKwX1terBiRUaqAsFqJiF615XL43r0=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b h1:+YaDE2r2OG8t/z5qmsh7Y+XXwCbvadxxZ0YY6mTdrVA=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b/go.mod h1:CgAqfJo+Xmu0GwA0411Ht3OU3OntXwsGmrmjI8ioGXI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a h1:myvhA4is3vrit1a6NZCWBIwN0kNEnX21DJOJX/NvIfI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a/go.mod h1:SUBoKXbI1Efip18FClrQVGjWcyd0QZd8KkvdP34t7ww=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 h1:AB/lmRny7e2pLhFEYIbl5qkDAUt2h0ZRO4wGPhZf+ik=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405/go.mod h1:67X1fPuzjcrkymZzZV1vvkFeTn2Rvc6lYF9MYFGCcwE=\ngoogle.golang.org/grpc v1.12.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.28.0/go.mod h1:rpkK4SK4GF4Ach/+MFLZUBavHOvF2JJB5uozKKal+60=\ngoogle.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0=\ngoogle.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc=\ngoogle.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.38.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.40.1/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34=\ngoogle.golang.org/grpc v1.41.0/go.mod h1:U3l9uK9J0sini8mHphKoXyaqDA/8VyGnDee1zzIUK6k=\ngoogle.golang.org/grpc v1.45.0/go.mod h1:lN7owxKUQEqMfSyQikvvk5tf/6zMPsrK+ONuO11+0rQ=\ngoogle.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk=\ngoogle.golang.org/grpc v1.59.0/go.mod h1:aUPDwccQo6OTjy7Hct4AfBPD1GptF4fyUjIkQ9YtF98=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=\ngoogle.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=\ngoogle.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=\ngoogle.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=\ngopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=\ngopkg.in/inf.v0 v0.9.1 h1:73M5CoZyi3ZLMOyDlQh031Cx6N9NDJ2Vvfl76EDAgDc=\ngopkg.in/inf.v0 v0.9.1/go.mod h1:cWUDdTG/fYaXco+Dcufb5Vnc6Gp2YChqWtbxRZE0mXw=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19 h1:WB265cn5OpO+hK3pikC9hpP1zI/KTwmyMFKloW9eOVc=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19/go.mod h1:o4V0GXN9/CAmCsvJ0oXYZvrZOe7syiDZSN1GWGZTGzc=\ngopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=\nhonnef.co/go/tools v0.3.2 h1:ytYb4rOqyp1TSa2EPvNVwtPQJctSELKaMyLfqNP4+34=\nhonnef.co/go/tools v0.3.2/go.mod h1:jzwdWgg7Jdq75wlfblQxO4neNaFFSvgc1tD5Wv8U0Yw=\nrsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=\nrsc.io/quote/v3 v3.1.0/go.mod h1:yEA65RcK8LyAZtP9Kv3t0HmxON59tX3rD+tICJqUlj0=\nrsc.io/sampler v1.3.0/go.mod h1:T1hPZKmBbMNahiBKFy5HrXp6adAjACjK9JXDnKaTXpA=\nsigs.k8s.io/yaml v1.2.0/go.mod h1:yfXDCHCao9+ENCvLSE62v9VSji2MKu5jeNfTrofGhJc=\n"
  },
  {
    "path": "cmd/server/main.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"os\"\n\n\t\"github.com/uber/cadence/cmd/server/cadence\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n\n\t_ \"github.com/uber/cadence/common/archiver/gcloud\"                                      // needed to load the optional gcloud archiver plugin\n\t_ \"github.com/uber/cadence/common/asyncworkflow/queue/kafka\"                            // needed to load kafka asyncworkflow queue\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\"              // needed to load cassandra plugin\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\" // needed to load the default gocql client\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"                      // needed to load mysql plugin\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\"                   // needed to load postgres plugin\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"                     // needed to load sqlite plugin\n\t_ \"github.com/uber/cadence/service/sharddistributor/store/etcd\"                         // needed for shard distributor shard/heartbeat and leader election\n)\n\n// main entry point for the cadence server\nfunc main() {\n\tapp := cadence.BuildCLI(metrics.ReleaseVersion, metrics.Revision)\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n"
  },
  {
    "path": "cmd/sharddistributor-canary/main.go",
    "content": "package main\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary\"\n\tcanaryConfig \"github.com/uber/cadence/service/sharddistributor/canary/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/executors\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\t// Default configuration\n\tdefaultShardDistributorEndpoint = \"127.0.0.1:7943\"\n\tdefaultFixedNamespace           = \"shard-distributor-canary\"\n\tdefaultEphemeralNamespace       = \"shard-distributor-canary-ephemeral\"\n\tdefaultCanaryGRPCPort           = 7953 // Port for canary to receive ping requests\n\tdefaultNumExecutors             = 1\n\n\tshardDistributorServiceName = \"cadence-shard-distributor\"\n)\n\nfunc runApp(c *cli.Context) {\n\tendpoint := c.String(\"endpoint\")\n\tfixedNamespace := c.String(\"fixed-namespace\")\n\tephemeralNamespace := c.String(\"ephemeral-namespace\")\n\tcanaryGRPCPort := c.Int(\"canary-grpc-port\")\n\n\tnumFixedExecutors := c.Int(\"num-fixed-executors\")\n\tnumEphemeralExecutors := c.Int(\"num-ephemeral-executors\")\n\n\tif c.IsSet(\"num-executors\") {\n\t\tnumExecutors := c.Int(\"num-executors\")\n\t\tnumFixedExecutors = numExecutors\n\t\tnumEphemeralExecutors = numExecutors\n\n\t}\n\n\tfx.New(opts(fixedNamespace, ephemeralNamespace, endpoint, canaryGRPCPort, numFixedExecutors, numEphemeralExecutors)).Run()\n}\n\nfunc opts(fixedNamespace, ephemeralNamespace, endpoint string, canaryGRPCPort int, numFixedExecutors, numEphemeral int) fx.Option {\n\tconfiguration := clientcommon.Config{\n\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t{Namespace: fixedNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeONBOARDED},\n\t\t\t{Namespace: ephemeralNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeONBOARDED},\n\t\t\t{Namespace: executors.LocalPassthroughNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeLOCALPASSTHROUGH},\n\t\t\t{Namespace: executors.LocalPassthroughShadowNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeLOCALPASSTHROUGHSHADOW},\n\t\t\t{Namespace: executors.DistributedPassthroughNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeDISTRIBUTEDPASSTHROUGH},\n\t\t\t{Namespace: executors.ExternalAssignmentNamespace, HeartBeatInterval: 1 * time.Second, MigrationMode: config.MigrationModeDISTRIBUTEDPASSTHROUGH},\n\t\t},\n\t}\n\n\tcanaryGRPCAddress := fmt.Sprintf(\"127.0.0.1:%d\", canaryGRPCPort)\n\n\t// Create listener for GRPC inbound\n\tlistener, err := net.Listen(\"tcp\", canaryGRPCAddress)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\ttransport := grpc.NewTransport()\n\n\texecutorMetadata := executorclient.ExecutorMetadata{\n\t\tclientcommon.GrpcAddressMetadataKey: canaryGRPCAddress,\n\t}\n\n\treturn fx.Options(\n\t\tfx.Supply(\n\t\t\tfx.Annotate(tally.NoopScope, fx.As(new(tally.Scope))),\n\t\t\tfx.Annotate(clock.NewRealTimeSource(), fx.As(new(clock.TimeSource))),\n\t\t\tconfiguration,\n\t\t\ttransport,\n\t\t\texecutorMetadata,\n\t\t),\n\n\t\tfx.Provide(func(peerChooser spectatorclient.SpectatorPeerChooserInterface) yarpc.Config {\n\t\t\treturn yarpc.Config{\n\t\t\t\tName: \"shard-distributor-canary\",\n\t\t\t\tInbounds: yarpc.Inbounds{\n\t\t\t\t\ttransport.NewInbound(listener), // Listen for incoming ping requests\n\t\t\t\t},\n\t\t\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\t\tshardDistributorServiceName: {\n\t\t\t\t\t\tUnary:  transport.NewSingleOutbound(endpoint),\n\t\t\t\t\t\tStream: transport.NewSingleOutbound(endpoint),\n\t\t\t\t\t},\n\t\t\t\t\t// canary-to-canary outbound uses peer chooser to route to other canary instances\n\t\t\t\t\t\"shard-distributor-canary\": {\n\t\t\t\t\t\tUnary:  transport.NewOutbound(peerChooser),\n\t\t\t\t\t\tStream: transport.NewOutbound(peerChooser),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t}),\n\n\t\tfx.Provide(\n\t\t\tfunc(t *grpc.Transport) peer.Transport { return t },\n\t\t),\n\t\tfx.Provide(\n\t\t\tyarpc.NewDispatcher,\n\t\t\tfunc(d *yarpc.Dispatcher) yarpc.ClientConfig { return d }, // Reprovide the dispatcher as a client config\n\t\t),\n\t\tfx.Provide(zap.NewDevelopment),\n\t\tfx.Provide(log.NewLogger),\n\n\t\t// We do decorate instead of Invoke because we want to start and stop the dispatcher at the\n\t\t// correct time.\n\t\t// It will start before all dependencies are started and stop after all dependencies are stopped.\n\t\t// The Decorate gives fx enough information, so it can start and stop the dispatcher at the correct time.\n\t\t//\n\t\t// It is critical to start and stop the dispatcher at the correct time.\n\t\t// Since the executors need to\n\t\t// be able to send a final \"drain\" request to the shard distributor before the application is stopped.\n\t\tfx.Decorate(func(\n\t\t\tlc fx.Lifecycle,\n\t\t\tdispatcher *yarpc.Dispatcher,\n\t\t\tserver sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCServer,\n\t\t) *yarpc.Dispatcher {\n\t\t\t// Register canary procedures and ensure dispatcher lifecycle is managed by fx.\n\t\t\tdispatcher.Register(sharddistributorv1.BuildShardDistributorExecutorCanaryAPIYARPCProcedures(server))\n\t\t\tlc.Append(fx.StartStopHook(dispatcher.Start, dispatcher.Stop))\n\t\t\treturn dispatcher\n\t\t}),\n\n\t\t// Include the canary module - it will set up spectator peer choosers and canary client\n\t\tcanary.Module(canary.NamespacesNames{\n\t\t\tFixedNamespace:              fixedNamespace,\n\t\t\tEphemeralNamespace:          ephemeralNamespace,\n\t\t\tExternalAssignmentNamespace: executors.ExternalAssignmentNamespace,\n\t\t\tSharddistributorServiceName: shardDistributorServiceName,\n\t\t\tConfig: canaryConfig.Config{\n\t\t\t\tCanary: canaryConfig.CanaryConfig{\n\t\t\t\t\tNumFixedExecutors:     numFixedExecutors,\n\t\t\t\t\tNumEphemeralExecutors: numEphemeral,\n\t\t\t\t},\n\t\t\t},\n\t\t}),\n\t)\n}\n\nfunc buildCLI() *cli.App {\n\tapp := cli.NewApp()\n\tapp.Name = \"sharddistributor-canary\"\n\tapp.Usage = \"Cadence shard distributor canary\"\n\tapp.Version = \"0.0.1\"\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:  \"start\",\n\t\t\tUsage: \"start shard distributor canary\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    \"endpoint\",\n\t\t\t\t\tAliases: []string{\"e\"},\n\t\t\t\t\tValue:   defaultShardDistributorEndpoint,\n\t\t\t\t\tUsage:   \"shard distributor endpoint address\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  \"fixed-namespace\",\n\t\t\t\t\tValue: defaultFixedNamespace,\n\t\t\t\t\tUsage: \"namespace for fixed shard processing\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  \"ephemeral-namespace\",\n\t\t\t\t\tValue: defaultEphemeralNamespace,\n\t\t\t\t\tUsage: \"namespace for ephemeral shard creation testing\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  \"canary-grpc-port\",\n\t\t\t\t\tValue: defaultCanaryGRPCPort,\n\t\t\t\t\tUsage: \"port for canary to receive ping requests\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  \"num-executors\",\n\t\t\t\t\tValue: defaultNumExecutors,\n\t\t\t\t\tUsage: \"number of executors for fixed and ephemeral to start. Overrides num-fixed-executors and num-ephemeral-executors flags\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  \"num-fixed-executors\",\n\t\t\t\t\tValue: defaultNumExecutors,\n\t\t\t\t\tUsage: \"number of executors of fixed namespace to start. Don't use with num-executors\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  \"num-ephemeral-executors\",\n\t\t\t\t\tValue: defaultNumExecutors,\n\t\t\t\t\tUsage: \"number of executors of ephemeral namespace to start. Don't use with num-executors\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\trunApp(c)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n\nfunc main() {\n\tapp := buildCLI()\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n"
  },
  {
    "path": "cmd/sharddistributor-canary/main_test.go",
    "content": "package main\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/fx\"\n)\n\nfunc TestDependenciesAreSatisfied(t *testing.T) {\n\tassert.NoError(t, fx.ValidateApp(opts(\n\t\tdefaultFixedNamespace,\n\t\tdefaultEphemeralNamespace,\n\t\tdefaultShardDistributorEndpoint,\n\t\tdefaultCanaryGRPCPort,\n\t\tdefaultNumExecutors,\n\t\tdefaultNumExecutors,\n\t)))\n}\n"
  },
  {
    "path": "cmd/tools/cassandra/main.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"os\"\n\n\t\"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\" // needed to load the default gocql client\n)\n\nfunc main() {\n\tapp := cassandra.BuildCLIOptions()\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n"
  },
  {
    "path": "cmd/tools/cli/main.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"os\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/tools/cli\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\"              // needed to load cassandra plugin\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\" // needed to load the default gocql client\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"                      // needed to load mysql plugin\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\"                   // needed to load postgres plugin\n)\n\n// Start using this CLI tool with command\n// See cadence/tools/cli/README.md for usage\nfunc main() {\n\tapp := cli.NewCliApp(cli.NewClientFactory(must(zap.NewDevelopment())))\n\tcommoncli.ExitHandler(app.Run(os.Args))\n}\n\nfunc must[T any](v T, err error) T {\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn v\n}\n"
  },
  {
    "path": "cmd/tools/copyright/licensegen.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"bufio\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"strings\"\n)\n\ntype (\n\t// task that adds license header to source\n\t// files, if they don't already exist\n\taddLicenseHeaderTask struct {\n\t\tlicense string  // license header string to add\n\t\tconfig  *config // root directory of the project source\n\t}\n\n\t// command line config params\n\tconfig struct {\n\t\trootDir            string\n\t\tverifyOnly         bool\n\t\ttemporalAddMode    bool\n\t\ttemporalModifyMode bool\n\t\tfilePaths          string\n\t}\n)\n\n// licenseFileName is the name of the license file\nconst licenseFileName = \"LICENSE\"\n\n// unique prefix that identifies a license header\nconst licenseHeaderPrefixOld = \"Copyright (c)\"\nconst licenseHeaderPrefix = \"// The MIT License (MIT)\"\nconst cadenceCopyright = \"// Copyright (c) 2017-2020 Uber Technologies Inc.\"\nconst cadenceModificationHeader = \"// Modifications Copyright (c) 2020 Uber Technologies Inc.\"\nconst temporalCopyright = \"// Copyright (c) 2020 Temporal Technologies, Inc.\"\nconst temporalPartialCopyright = \"// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\"\n\nconst firstLinesToCheck = 10\n\nvar (\n\t// directories to be excluded\n\tdirDenylist = []string{\"vendor/\"}\n\t// default perms for the newly created files\n\tdefaultFilePerms = os.FileMode(0644)\n)\n\n// command line utility that adds license header\n// to the source files. Usage as follows:\n//\n//\t./cmd/tools/copyright/licensegen.go\nfunc main() {\n\n\tvar cfg config\n\tflag.StringVar(&cfg.rootDir, \"rootDir\", \".\", \"project root directory\")\n\tflag.BoolVar(&cfg.verifyOnly, \"verifyOnly\", false,\n\t\t\"don't automatically add headers, just verify all files\")\n\tflag.BoolVar(&cfg.temporalAddMode, \"temporalAddMode\", false, \"add copyright for new file copied from temporal\")\n\tflag.BoolVar(&cfg.temporalModifyMode, \"temporalModifyMode\", false, \"add copyright for existing file which has parts copied from temporal\")\n\tflag.StringVar(&cfg.filePaths, \"filePaths\", \"\", \"comma separated list of files to run temporal license on\")\n\tflag.Parse()\n\n\tif err := verifyCfg(cfg); err != nil {\n\t\tfmt.Println(err)\n\t\tos.Exit(-1)\n\t}\n\n\ttask := newAddLicenseHeaderTask(&cfg)\n\tif err := task.run(); err != nil {\n\t\tfmt.Println(err)\n\t\tos.Exit(-1)\n\t}\n}\n\nfunc verifyCfg(cfg config) error {\n\tif cfg.verifyOnly {\n\t\tif cfg.temporalModifyMode || cfg.temporalAddMode {\n\t\t\treturn errors.New(\"invalid config, can only specify one of temporalAddMode, temporalModifyMode or verifyOnly\")\n\t\t}\n\t}\n\tif cfg.temporalAddMode && cfg.temporalModifyMode {\n\t\treturn errors.New(\"invalid config, can only specify temporalAddMode or temporalModifyMode\")\n\t}\n\tif (cfg.temporalModifyMode || cfg.temporalAddMode) && len(cfg.filePaths) == 0 {\n\t\treturn errors.New(\"invalid config, when running in temporalAddMode or temporalModifyMode must provide filePaths\")\n\t}\n\treturn nil\n}\n\nfunc newAddLicenseHeaderTask(cfg *config) *addLicenseHeaderTask {\n\treturn &addLicenseHeaderTask{\n\t\tconfig: cfg,\n\t}\n}\n\nfunc (task *addLicenseHeaderTask) run() error {\n\tdata, err := ioutil.ReadFile(task.config.rootDir + \"/\" + licenseFileName)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"error reading license file, errr=%v\", err.Error())\n\t}\n\ttask.license, err = commentOutLines(string(data))\n\tif err != nil {\n\t\treturn fmt.Errorf(\"copyright header failed to comment out lines, err=%v\", err.Error())\n\t}\n\tif task.config.temporalAddMode {\n\t\ttask.license = fmt.Sprintf(\"%v\\n\\n%v\\n\\n%v\", cadenceModificationHeader, temporalCopyright, task.license)\n\t} else if task.config.temporalModifyMode {\n\t\ttask.license = fmt.Sprintf(\"%v\\n\\n%v\\n\\n%v\", cadenceCopyright, temporalPartialCopyright, task.license)\n\t}\n\tif task.config.temporalModifyMode || task.config.temporalAddMode {\n\t\tfilePaths, fileInfos, err := getFilePaths(task.config.filePaths)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfor i := 0; i < len(filePaths); i++ {\n\t\t\tif err := task.handleFile(filePaths[i], fileInfos[i], nil); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t}\n\ttask.license = fmt.Sprintf(\"%v\\n\\n%v\\n\\n%v\", licenseHeaderPrefix, cadenceCopyright, task.license)\n\terr = filepath.Walk(task.config.rootDir, task.handleFile)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"copyright header check failed, err=%v\", err.Error())\n\t}\n\treturn nil\n}\n\nfunc (task *addLicenseHeaderTask) handleFile(path string, fileInfo os.FileInfo, err error) error {\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif fileInfo.IsDir() {\n\t\tif strings.HasPrefix(fileInfo.Name(), \"_vendor-\") || fileInfo.Name() == \".build\" || fileInfo.Name() == \".bin\" {\n\t\t\treturn filepath.SkipDir\n\t\t}\n\t\treturn nil\n\t}\n\n\tif !mustProcessPath(path) {\n\t\treturn nil\n\t}\n\n\tif !strings.HasSuffix(fileInfo.Name(), \".go\") && !strings.HasSuffix(fileInfo.Name(), \".proto\") {\n\t\treturn nil\n\t}\n\n\t// Used as part of the cli to write licence headers on files, does not use user supplied input so marked as nosec\n\t// #nosec\n\tf, err := os.Open(path)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tok, err := hasCopyright(f)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := f.Close(); err != nil {\n\t\treturn err\n\t}\n\n\tif ok {\n\t\tif task.config.temporalModifyMode || task.config.temporalAddMode {\n\t\t\treturn fmt.Errorf(\"when running in temporalModifyMode or temporalAddMode please first remove existing license header: %v\", path)\n\t\t}\n\t\treturn nil\n\t}\n\n\t// at this point, src file is missing the header\n\tif task.config.verifyOnly {\n\t\tif !isFileAutogenerated(path) {\n\t\t\treturn fmt.Errorf(\"%v missing license header\", path)\n\t\t}\n\t}\n\n\t// Used as part of the cli to write licence headers on files, does not use user supplied input so marked as nosec\n\t// #nosec\n\tdata, err := ioutil.ReadFile(path)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn ioutil.WriteFile(path, []byte(task.license+string(data)), defaultFilePerms)\n}\n\nfunc hasCopyright(f *os.File) (bool, error) {\n\tscanner := bufio.NewScanner(f)\n\tlineSuccess := scanner.Scan()\n\tif !lineSuccess {\n\t\treturn false, fmt.Errorf(\"fail to read first line of file %v\", f.Name())\n\t}\n\ti := 0\n\tfor i < firstLinesToCheck && lineSuccess {\n\t\ti++\n\t\tline := strings.TrimSpace(scanner.Text())\n\t\tif err := scanner.Err(); err != nil {\n\t\t\treturn false, err\n\t\t}\n\t\tif lineHasCopyright(line) {\n\t\t\treturn true, nil\n\t\t}\n\t\tlineSuccess = scanner.Scan()\n\t}\n\treturn false, nil\n}\n\nfunc isFileAutogenerated(path string) bool {\n\treturn strings.HasPrefix(path, \".gen\")\n}\n\nfunc mustProcessPath(path string) bool {\n\tfor _, d := range dirDenylist {\n\t\tif strings.HasPrefix(path, d) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc commentOutLines(str string) (string, error) {\n\tvar lines []string\n\tscanner := bufio.NewScanner(strings.NewReader(str))\n\tfor scanner.Scan() {\n\t\ttext := scanner.Text()\n\t\tif text == \"\" {\n\t\t\t// do not add trailing whitespace, gofmt / goimports removes it\n\t\t\tlines = append(lines, \"//\\n\")\n\t\t} else {\n\t\t\tlines = append(lines, \"// \"+scanner.Text()+\"\\n\")\n\t\t}\n\t}\n\tlines = append(lines, \"\\n\")\n\n\tif err := scanner.Err(); err != nil {\n\t\treturn \"\", err\n\t}\n\treturn strings.Join(lines, \"\"), nil\n}\n\nfunc lineHasCopyright(line string) bool {\n\treturn strings.Contains(line, licenseHeaderPrefixOld) ||\n\t\tstrings.Contains(line, licenseHeaderPrefix)\n}\n\nfunc getFilePaths(filePaths string) ([]string, []os.FileInfo, error) {\n\tpaths := strings.Split(filePaths, \",\")\n\tvar fileInfos []os.FileInfo\n\tfor _, p := range paths {\n\t\tfileInfo, err := os.Stat(p)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\tfileInfos = append(fileInfos, fileInfo)\n\t}\n\treturn paths, fileInfos, nil\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/console/console.go",
    "content": "package console\n\nimport (\n\t\"bufio\"\n\t\"context\"\n\t\"fmt\"\n\t\"io\"\n\t\"strings\"\n)\n\n// Manager handles console interactions\ntype Manager struct {\n\treader io.Reader\n\twriter io.Writer\n}\n\n// NewManager creates a new console manager\nfunc NewManager(reader io.Reader, writer io.Writer) *Manager {\n\treturn &Manager{\n\t\treader: reader,\n\t\twriter: writer,\n\t}\n}\n\n// Confirm asks for user confirmation and returns true for 'y', false for 'n'\nfunc (m *Manager) Confirm(ctx context.Context, message string) (bool, error) {\n\treturn m.ConfirmWithDefault(ctx, message, false)\n}\n\n// ConfirmWithDefault asks for user confirmation with a default value\n// Returns defaultValue if user just presses enter\nfunc (m *Manager) ConfirmWithDefault(ctx context.Context, message string, defaultValue bool) (bool, error) {\n\tprompt := fmt.Sprintf(\"%s [y/N]: \", message)\n\tif defaultValue {\n\t\tprompt = fmt.Sprintf(\"%s [Y/n]: \", message)\n\t}\n\n\t_, _ = fmt.Fprint(m.writer, prompt)\n\n\tinputChan := make(chan inputResult, 1)\n\n\t// Start goroutine to read input\n\tgo func() {\n\t\tscanner := bufio.NewScanner(m.reader)\n\t\tif !scanner.Scan() {\n\t\t\tif err := scanner.Err(); err != nil {\n\t\t\t\tinputChan <- inputResult{err: fmt.Errorf(\"failed to read input: %w\", err)}\n\t\t\t\treturn\n\t\t\t}\n\t\t\t// EOF or no input - use default\n\t\t\tinputChan <- inputResult{text: \"\", err: nil}\n\t\t\treturn\n\t\t}\n\t\tinputChan <- inputResult{text: scanner.Text(), err: nil}\n\t}()\n\n\tvar result inputResult\n\t// Wait for either input or context cancellation\n\tselect {\n\tcase <-ctx.Done():\n\t\t// Context was cancelled\n\t\treturn false, ctx.Err()\n\tcase result = <-inputChan:\n\t}\n\t// Got input from user\n\tif result.err != nil {\n\t\treturn false, result.err\n\t}\n\n\tinput := strings.TrimSpace(strings.ToLower(result.text))\n\n\t// Empty input uses default\n\tif input == \"\" {\n\t\treturn defaultValue, nil\n\t}\n\n\tswitch input {\n\tcase \"y\", \"yes\":\n\t\treturn true, nil\n\tcase \"n\", \"no\":\n\t\treturn false, nil\n\tdefault:\n\t\treturn false, fmt.Errorf(\"invalid input: %s\", input)\n\t}\n}\n\ntype inputResult struct {\n\ttext string\n\terr  error\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/console/console_test.go",
    "content": "package console\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"io\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestConfirm(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tinput    string\n\t\texpected bool\n\t\thasError bool\n\t}{\n\t\t{\"yes response\", \"y\\n\", true, false},\n\t\t{\"Yes response\", \"Y\\n\", true, false},\n\t\t{\"yes full\", \"yes\\n\", true, false},\n\t\t{\"no response\", \"n\\n\", false, false},\n\t\t{\"No response\", \"N\\n\", false, false},\n\t\t{\"no full\", \"no\\n\", false, false},\n\t\t{\"empty input uses default false\", \"\\n\", false, false},\n\t\t{\"invalid input\", \"maybe\\n\", false, true},\n\t\t{\"whitespace input\", \"  \\n\", false, false},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\treader := strings.NewReader(tt.input)\n\t\t\twriter := &bytes.Buffer{}\n\t\t\tmanager := NewManager(reader, writer)\n\n\t\t\tctx := context.Background()\n\t\t\tresult, err := manager.Confirm(ctx, \"Test message\")\n\n\t\t\tif tt.hasError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\n\t\t\t// Check that prompt was written\n\t\t\toutput := writer.String()\n\t\t\tassert.Contains(t, output, \"Test message [y/N]:\")\n\t\t})\n\t}\n}\n\nfunc TestConfirmWithDefault(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tinput        string\n\t\tdefaultValue bool\n\t\texpected     bool\n\t\thasError     bool\n\t}{\n\t\t{\"yes with default false\", \"y\\n\", false, true, false},\n\t\t{\"no with default true\", \"n\\n\", true, false, false},\n\t\t{\"empty uses default true\", \"\\n\", true, true, false},\n\t\t{\"empty uses default false\", \"\\n\", false, false, false},\n\t\t{\"invalid input\", \"invalid\\n\", true, false, true},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\treader := strings.NewReader(tt.input)\n\t\t\twriter := &bytes.Buffer{}\n\t\t\tmanager := NewManager(reader, writer)\n\n\t\t\tctx := context.Background()\n\t\t\tresult, err := manager.ConfirmWithDefault(ctx, \"Test message\", tt.defaultValue)\n\n\t\t\tif tt.hasError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\n\t\t\t// Check prompt format based on default\n\t\t\toutput := writer.String()\n\t\t\tif tt.defaultValue {\n\t\t\t\tassert.Contains(t, output, \"[Y/n]:\")\n\t\t\t} else {\n\t\t\t\tassert.Contains(t, output, \"[y/N]:\")\n\t\t\t}\n\t\t})\n\t}\n}\n\n// blockingReader blocks on Read until unblocked\ntype blockingReader struct {\n\tunblock chan struct{}\n\tdata    []byte\n\tread    bool\n}\n\nfunc newBlockingReader() *blockingReader {\n\treturn &blockingReader{\n\t\tunblock: make(chan struct{}),\n\t\tdata:    []byte(\"y\\n\"),\n\t}\n}\n\nfunc (br *blockingReader) Read(p []byte) (n int, err error) {\n\tif br.read {\n\t\treturn 0, io.EOF\n\t}\n\n\t// Block until unblocked\n\t<-br.unblock\n\n\tbr.read = true\n\tn = copy(p, br.data)\n\treturn n, nil\n}\n\nfunc (br *blockingReader) Unblock() {\n\tclose(br.unblock)\n}\n\nfunc TestContextCancellation(t *testing.T) {\n\tt.Run(\"context cancelled before input\", func(t *testing.T) {\n\t\t// Create a context that's already cancelled\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tcancel()\n\n\t\treader := strings.NewReader(\"y\\n\") // This won't be read due to cancellation\n\t\twriter := &bytes.Buffer{}\n\t\tmanager := NewManager(reader, writer)\n\n\t\tresult, err := manager.ConfirmWithDefault(ctx, \"Test\", false)\n\n\t\trequire.Error(t, err)\n\t\tassert.True(t, errors.Is(err, context.Canceled))\n\t\tassert.False(t, result)\n\t})\n\n\tt.Run(\"context cancelled during input wait\", func(t *testing.T) {\n\t\tctx, cancel := context.WithCancel(context.Background())\n\n\t\t// Use a blocking reader that actually blocks\n\t\treader := newBlockingReader()\n\t\twriter := &bytes.Buffer{}\n\t\tmanager := NewManager(reader, writer)\n\n\t\t// Cancel context after a short delay\n\t\tgo func() {\n\t\t\ttime.Sleep(10 * time.Millisecond)\n\t\t\tcancel()\n\t\t}()\n\n\t\tresult, err := manager.ConfirmWithDefault(ctx, \"Test\", false)\n\n\t\trequire.Error(t, err)\n\t\tassert.True(t, errors.Is(err, context.Canceled))\n\t\tassert.False(t, result)\n\t})\n}\n\n// errorReader always returns an error when scanning\ntype errorReader struct{}\n\nfunc (e errorReader) Read(p []byte) (n int, err error) {\n\treturn 0, errors.New(\"read error\")\n}\n\nfunc TestScannerError(t *testing.T) {\n\treader := errorReader{}\n\twriter := &bytes.Buffer{}\n\tmanager := NewManager(reader, writer)\n\n\tctx := context.Background()\n\tresult, err := manager.ConfirmWithDefault(ctx, \"Test\", false)\n\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"failed to read input\")\n\tassert.False(t, result)\n}\n\nfunc TestEOFHandling(t *testing.T) {\n\t// Reader with no content (immediate EOF)\n\treader := strings.NewReader(\"\")\n\twriter := &bytes.Buffer{}\n\tmanager := NewManager(reader, writer)\n\n\tctx := context.Background()\n\tresult, err := manager.ConfirmWithDefault(ctx, \"Test\", true)\n\n\tassert.NoError(t, err)\n\tassert.True(t, result, \"Expected default value (true) on EOF\")\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/fs/fs.go",
    "content": "package fs\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"os\"\n\t\"path/filepath\"\n\n\t\"golang.org/x/mod/modfile\"\n)\n\n// Client implements Interface\ntype Client struct {\n\tverbose bool\n}\n\nfunc NewFileSystemClient(verbose bool) *Client {\n\treturn &Client{verbose: verbose}\n}\n\n// FindGoModFiles reads go.work file and returns module directories\nfunc (f *Client) FindGoModFiles(ctx context.Context, root string) ([]string, error) {\n\tf.logDebug(\"Finding modules from go.work file\")\n\n\tworkFilePath := filepath.Join(root, \"go.work\")\n\tmodules, err := f.parseGoWorkFile(workFilePath, root)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to parse go.work file: %w\", err)\n\t}\n\n\tf.logDebug(\"Found modules from go.work: %v\", modules)\n\treturn modules, nil\n}\n\n// parseGoWorkFile parses the go.work file using the official modfile package\nfunc (f *Client) parseGoWorkFile(workFilePath, root string) ([]string, error) {\n\tworkFileData, err := os.ReadFile(workFilePath)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read go.work file: %w\", err)\n\t}\n\n\tworkFile, err := modfile.ParseWork(workFilePath, workFileData, nil)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to parse go.work file: %w\", err)\n\t}\n\n\tmodules := make([]string, 0, len(workFile.Use))\n\tfor _, use := range workFile.Use {\n\t\tmodules = append(modules, use.Path)\n\t}\n\n\treturn modules, nil\n}\n\n// resolveModulePath converts relative path to absolute path\nfunc (f *Client) resolveModulePath(modulePath, root string) string {\n\tif filepath.IsAbs(modulePath) {\n\t\treturn modulePath\n\t}\n\n\treturn filepath.Join(root, modulePath)\n}\n\nfunc (f *Client) logDebug(msg string, args ...interface{}) {\n\tif f.verbose {\n\t\tfmt.Printf(\"%s\\n\", fmt.Sprintf(msg, args...))\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/git/git.go",
    "content": "package git\n\nimport (\n\t\"bufio\"\n\t\"context\"\n\t\"fmt\"\n\t\"os/exec\"\n\t\"strings\"\n)\n\n// Client implements Interface\ntype Client struct {\n\tverbose bool\n}\n\nfunc NewGitClient(verbose bool) *Client {\n\treturn &Client{verbose: verbose}\n}\n\nfunc (g *Client) GetCurrentBranch(ctx context.Context) (string, error) {\n\tg.logDebug(\"Getting current git branch\")\n\tcmd := exec.CommandContext(ctx, \"git\", \"rev-parse\", \"--abbrev-ref\", \"HEAD\")\n\toutput, err := cmd.Output()\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"get current branch: %w\", err)\n\t}\n\tbranch := strings.TrimSpace(string(output))\n\tg.logDebug(\"Current branch: %s\", branch)\n\treturn branch, nil\n}\n\nfunc (g *Client) GetTags(ctx context.Context) ([]string, error) {\n\tg.logDebug(\"Fetching git tags\")\n\tcmd := exec.CommandContext(ctx, \"git\", \"tag\", \"-l\")\n\toutput, err := cmd.Output()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get git tags: %w\", err)\n\t}\n\n\tvar tags []string\n\tscanner := bufio.NewScanner(strings.NewReader(string(output)))\n\tfor scanner.Scan() {\n\t\tif tag := strings.TrimSpace(scanner.Text()); tag != \"\" {\n\t\t\ttags = append(tags, tag)\n\t\t}\n\t}\n\tg.logDebug(\"Found git tags %v\", tags)\n\treturn tags, scanner.Err()\n}\n\nfunc (g *Client) CreateTag(ctx context.Context, tag string) error {\n\tg.logDebug(\"Creating git tag %s\", tag)\n\tcmd := exec.CommandContext(ctx, \"git\", \"tag\", tag)\n\terr := cmd.Run()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"create tag %s: %w\", tag, err)\n\t}\n\treturn err\n}\n\nfunc (g *Client) PushTag(ctx context.Context, tag string) error {\n\tg.logDebug(\"Pushing git tag %s\", tag)\n\tcmd := exec.CommandContext(ctx, \"git\", \"push\", \"origin\", tag)\n\terr := cmd.Run()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"push tag %s: %w\", tag, err)\n\t}\n\treturn err\n}\n\nfunc (g *Client) GetRepoRoot(ctx context.Context) (string, error) {\n\tg.logDebug(\"Getting repository root\")\n\tcmd := exec.CommandContext(ctx, \"git\", \"rev-parse\", \"--show-toplevel\")\n\toutput, err := cmd.Output()\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"get repository root: %w\", err)\n\t}\n\troot := strings.TrimSpace(string(output))\n\tg.logDebug(\"Repository root %s\", root)\n\treturn root, nil\n}\n\nfunc (g *Client) logDebug(msg string, args ...interface{}) {\n\tif g.verbose {\n\t\tfmt.Printf(\"%s\\n\", fmt.Sprintf(msg, args...))\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/release.go",
    "content": "package release\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"regexp\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/Masterminds/semver/v3\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination release_mocks_test.go -self_package github.com/uber/cadence/cmd/tools/releaser/release\n\n// Git defines git operations for testing\ntype Git interface {\n\tGetCurrentBranch(ctx context.Context) (string, error)\n\tGetTags(ctx context.Context) ([]string, error)\n\tCreateTag(ctx context.Context, tag string) error\n\tPushTag(ctx context.Context, tag string) error\n\tGetRepoRoot(ctx context.Context) (string, error)\n}\n\n// FS defines filesystem operations for testing\ntype FS interface {\n\tFindGoModFiles(ctx context.Context, root string) ([]string, error)\n}\n\n// UserInteraction defines the interface for user interactions\ntype UserInteraction interface {\n\tConfirm(ctx context.Context, message string) (bool, error)\n\tConfirmWithDefault(ctx context.Context, message string, defaultValue bool) (bool, error)\n}\n\nvar (\n\tversionRegex    = regexp.MustCompile(`v[0-9]+\\.[0-9]+\\.[0-9]+(?:-prerelease[0-9]+)?`)\n\tprereleaseRegex = regexp.MustCompile(`v([0-9]+\\.[0-9]+\\.[0-9]+)-prerelease(\\d+)`)\n)\n\n// Config holds the release configuration\ntype Config struct {\n\tRepoRoot          string\n\tExcludedDirs      []string\n\tRequiredBranch    string\n\tVerbose           bool\n\tCommand           string // The subcommand being executed\n\tSkipConfirmations bool\n\tManualVersion     string // Manual version override\n}\n\n// Manager handles the release process\ntype Manager struct {\n\tconfig      Config\n\tgit         Git\n\tfs          FS\n\tinteraction UserInteraction\n\ttagCache    *TagCache\n}\n\nfunc NewReleaseManager(config Config, git Git, fs FS, interaction UserInteraction) *Manager {\n\treturn &Manager{\n\t\tconfig:      config,\n\t\tgit:         git,\n\t\tfs:          fs,\n\t\tinteraction: interaction,\n\t}\n}\n\nfunc (rm *Manager) RunRelease(ctx context.Context) error {\n\treturn rm.executeCommand(ctx, rm.calculateReleaseVersion, rm.validateReleaseCommand)\n}\n\nfunc (rm *Manager) RunPatch(ctx context.Context) error {\n\treturn rm.executeCommand(ctx, rm.calculatePatchVersion, rm.validateMinorMajorCommand)\n}\n\nfunc (rm *Manager) RunMinor(ctx context.Context) error {\n\treturn rm.executeCommand(ctx, rm.calculateMinorVersion, rm.validateMinorMajorCommand)\n}\n\nfunc (rm *Manager) RunMajor(ctx context.Context) error {\n\treturn rm.executeCommand(ctx, rm.calculateMajorVersion, rm.validateMinorMajorCommand)\n}\n\nfunc (rm *Manager) RunPrerelease(ctx context.Context) error {\n\treturn rm.executeCommand(ctx, rm.calculatePrereleaseVersion, nil)\n}\n\n// Generic command execution flow\nfunc (rm *Manager) executeCommand(\n\tctx context.Context,\n\tcalculateVersion func(string) (string, error),\n\tvalidate func(string) error,\n) error {\n\t// Initialize tag cache\n\tif err := rm.GetKnownReleases(ctx); err != nil {\n\t\treturn err\n\t}\n\n\t// Get current version\n\tcurrentVersion := rm.GetCurrentGlobalVersion()\n\trm.logDebug(\"Current version: %s\", currentVersion)\n\n\tvar targetVersion string\n\tvar err error\n\n\t// Check for manual version override\n\tif rm.config.ManualVersion != \"\" {\n\t\trm.logDebug(\"Using manual version override: %s\", rm.config.ManualVersion)\n\t\ttargetVersion, err = rm.processManualVersion(rm.config.ManualVersion)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\t// Run validation if provided (only for automatic version calculation)\n\t\tif validate != nil {\n\t\t\tif err := validate(currentVersion); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\t// Calculate target version automatically\n\t\ttargetVersion, err = calculateVersion(currentVersion)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tfmt.Printf(\"Version transition: %s → %s\\n\", currentVersion, targetVersion)\n\n\treturn rm.executeRelease(ctx, targetVersion)\n}\n\n// Version calculation methods for each subcommand\nfunc (rm *Manager) calculateReleaseVersion(currentVersionStr string) (string, error) {\n\trm.logDebug(\"Calculating release version from: %s\", currentVersionStr)\n\n\tcurrentVersion, err := semver.NewVersion(currentVersionStr)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"failed to parse current version %s: %w\", currentVersionStr, err)\n\t}\n\n\t// Remove prerelease suffix\n\treleaseVersion := fmt.Sprintf(\"v%d.%d.%d\",\n\t\tcurrentVersion.Major(),\n\t\tcurrentVersion.Minor(),\n\t\tcurrentVersion.Patch())\n\n\treturn releaseVersion, nil\n}\n\n// Generic version calculation that creates a closure for different version types\nfunc (rm *Manager) calculateVersionIncrement(versionType string) func(string) (string, error) {\n\treturn func(currentVersionStr string) (string, error) {\n\t\trm.logDebug(\"Calculating %s version from: %s\", versionType, currentVersionStr)\n\n\t\t// First, get the base version (remove prerelease if present)\n\t\tbaseVersion := rm.getBaseVersion(currentVersionStr)\n\n\t\t// Increment version\n\t\tnewVersion, err := IncrementVersion(baseVersion, versionType)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\n\t\t// Create first prerelease\n\t\treturn rm.GetNextPrereleaseVersion(newVersion)\n\t}\n}\n\n// Update the existing methods to use the generic function\nfunc (rm *Manager) calculateMinorVersion(currentVersionStr string) (string, error) {\n\treturn rm.calculateVersionIncrement(\"minor\")(currentVersionStr)\n}\n\nfunc (rm *Manager) calculateMajorVersion(currentVersionStr string) (string, error) {\n\treturn rm.calculateVersionIncrement(\"major\")(currentVersionStr)\n}\n\nfunc (rm *Manager) calculatePatchVersion(currentVersionStr string) (string, error) {\n\treturn rm.calculateVersionIncrement(\"patch\")(currentVersionStr)\n}\n\nfunc (rm *Manager) calculatePrereleaseVersion(currentVersionStr string) (string, error) {\n\trm.logDebug(\"Calculating prerelease version from: %s\", currentVersionStr)\n\n\tcurrentVersion, err := semver.NewVersion(currentVersionStr)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"failed to parse current version %s: %w\", currentVersionStr, err)\n\t}\n\n\tbaseVersionStr := fmt.Sprintf(\"v%d.%d.%d\",\n\t\tcurrentVersion.Major(),\n\t\tcurrentVersion.Minor(),\n\t\tcurrentVersion.Patch())\n\n\treturn rm.GetNextPrereleaseVersion(baseVersionStr)\n}\n\n// Validation methods\nfunc (rm *Manager) validateReleaseCommand(currentVersion string) error {\n\tif !strings.Contains(currentVersion, \"-prerelease\") {\n\t\treturn fmt.Errorf(\"release command requires existing prerelease version, current: %s\", currentVersion)\n\t}\n\treturn nil\n}\n\nfunc (rm *Manager) validateMinorMajorCommand(currentVersion string) error {\n\tif strings.Contains(currentVersion, \"-prerelease\") {\n\t\treturn fmt.Errorf(\"minor/major commands should be run from stable versions, current: %s (consider using 'release' first)\", currentVersion)\n\t}\n\treturn nil\n}\n\n// Helper method to get base version (remove prerelease suffix)\nfunc (rm *Manager) getBaseVersion(versionStr string) string {\n\tversion, err := semver.NewVersion(versionStr)\n\tif err != nil {\n\t\treturn versionStr\n\t}\n\n\treturn fmt.Sprintf(\"v%d.%d.%d\", version.Major(), version.Minor(), version.Patch())\n}\n\n// processManualVersion validates and normalizes manual version input\nfunc (rm *Manager) processManualVersion(manualVersion string) (string, error) {\n\t// Normalize the version (ensure v prefix, validate semver)\n\tnormalizedVersion, err := NormalizeVersion(manualVersion)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"invalid manual version format: %w\", err)\n\t}\n\n\trm.logDebug(\"Manual version normalized: %s → %s\", manualVersion, normalizedVersion)\n\treturn normalizedVersion, nil\n}\n\n// Execute the actual release\nfunc (rm *Manager) executeRelease(ctx context.Context, targetVersion string) error {\n\t// Assess state\n\tstate, err := rm.AssessCurrentState(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// Plan actions\n\tactions, warnings := rm.planReleaseActions(state, targetVersion)\n\n\t// Handle warnings and get confirmation to continue\n\tif err = rm.handleWarningsAndConfirmations(ctx, warnings); err != nil {\n\t\treturn err\n\t}\n\n\t// Check for version conflicts and handle them\n\tconflictInfo, conflictErr := rm.CheckVersionExists(targetVersion, state.Modules)\n\tfinalActions := actions\n\n\tif conflictErr != nil {\n\t\tfmt.Printf(\"❌ Version conflict detected:\\n\")\n\t\tfmt.Printf(\"   Existing tags: %v\\n\", conflictInfo.ExistingTags)\n\t\tfmt.Printf(\"   Tags to create: %v\\n\", conflictInfo.MissingTags)\n\n\t\tif len(conflictInfo.MissingTags) == 0 {\n\t\t\treturn fmt.Errorf(\"all tags already exist for version %s\", targetVersion)\n\t\t}\n\n\t\tif !rm.config.SkipConfirmations {\n\t\t\tconfirmed, err := rm.interaction.ConfirmWithDefault(ctx,\n\t\t\t\tfmt.Sprintf(\"Continue and create only missing tags (%d remaining)?\", len(conflictInfo.MissingTags)),\n\t\t\t\tfalse)\n\t\t\tif err != nil || !confirmed {\n\t\t\t\treturn fmt.Errorf(\"operation cancelled due to version conflict\")\n\t\t\t}\n\t\t}\n\n\t\t// Filter actions to only include missing tags\n\t\tfinalActions = rm.filterActionsForMissingTags(actions, conflictInfo.MissingTags)\n\t\tfmt.Printf(\"✓ Will skip existing tags and create only: %v\\n\", conflictInfo.MissingTags)\n\t}\n\n\t// Show planned actions (filtered if there were conflicts)\n\trm.ShowPlannedActions(finalActions)\n\n\t// Confirm tag creation\n\tif !rm.config.SkipConfirmations {\n\t\ttagCount := len(finalActions) / 2 // Each tag has create + push action\n\t\tmessage := fmt.Sprintf(\"Create %d tags?\", tagCount)\n\t\tif conflictErr != nil {\n\t\t\tmessage = fmt.Sprintf(\"Create %d missing tags (skipping %d existing)?\", len(conflictInfo.MissingTags), len(conflictInfo.ExistingTags))\n\t\t}\n\n\t\tconfirmed, err := rm.interaction.Confirm(ctx, message)\n\t\tif err != nil || !confirmed {\n\t\t\tif ctx.Err() != nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn fmt.Errorf(\"tag creation cancelled\")\n\t\t}\n\t}\n\n\t// Create tags (only missing ones if there were conflicts)\n\tif err = rm.executeTagCreation(ctx, finalActions); err != nil {\n\t\treturn err\n\t}\n\n\t// Confirm tag pushing\n\tif !rm.config.SkipConfirmations {\n\t\tpushCount := 0\n\t\tfor _, action := range finalActions {\n\t\t\tif action.Type == ActionPushTags {\n\t\t\t\tpushCount++\n\t\t\t}\n\t\t}\n\n\t\tmessage := fmt.Sprintf(\"Push %d tags?\", pushCount)\n\t\tconfirmed, err := rm.interaction.Confirm(ctx, message)\n\t\tif err != nil || !confirmed {\n\t\t\tif ctx.Err() != nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tfmt.Printf(\"Tags created locally but not pushed\\n\")\n\t\t\tif conflictErr != nil {\n\t\t\t\tfmt.Printf(\"Created: %v\\n\", conflictInfo.MissingTags)\n\t\t\t\tfmt.Printf(\"Skipped: %v\\n\", conflictInfo.ExistingTags)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\t}\n\n\t// Push tags\n\tif err = rm.executeTagPushing(ctx, finalActions); err != nil {\n\t\treturn fmt.Errorf(\"push tags: %w\", err)\n\t}\n\n\t// Success message\n\tif conflictErr != nil {\n\t\tfmt.Printf(\"✓ Release %s completed successfully\\n\", targetVersion)\n\t\tfmt.Printf(\"  Created: %v\\n\", conflictInfo.MissingTags)\n\t\tfmt.Printf(\"  Skipped: %v (already existed)\\n\", conflictInfo.ExistingTags)\n\t} else {\n\t\tfmt.Printf(\"✓ Release %s completed successfully\\n\", targetVersion)\n\t}\n\n\treturn nil\n}\n\n// GetKnownReleases fetches and parses all tags once\nfunc (rm *Manager) GetKnownReleases(ctx context.Context) error {\n\tfmt.Println(\"Getting known releases\")\n\n\t// Fetch raw tags once\n\trawTags, err := rm.git.GetTags(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to fetch tags: %w\", err)\n\t}\n\n\trm.tagCache = &TagCache{\n\t\tAllTags:         make([]ParsedTag, 0, len(rawTags)),\n\t\tVersionTags:     make([]ParsedTag, 0, len(rawTags)),\n\t\tModuleTags:      make(map[string][]ParsedTag),\n\t\tPrereleaseCache: make(map[string][]int),\n\t}\n\n\tfor _, rawTag := range rawTags {\n\t\tparsedTag := rm.parseTag(rawTag)\n\t\trm.tagCache.AllTags = append(rm.tagCache.AllTags, parsedTag)\n\n\t\t// Skip not version tags\n\t\tif parsedTag.Version == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\trm.tagCache.VersionTags = append(rm.tagCache.VersionTags, parsedTag)\n\n\t\t// Group by module\n\t\tif rm.tagCache.ModuleTags[parsedTag.ModulePath] == nil {\n\t\t\trm.tagCache.ModuleTags[parsedTag.ModulePath] = make([]ParsedTag, 0)\n\t\t}\n\t\trm.tagCache.ModuleTags[parsedTag.ModulePath] = append(\n\t\t\trm.tagCache.ModuleTags[parsedTag.ModulePath], parsedTag)\n\n\t\t// Cache prerelease numbers\n\t\tif parsedTag.IsPrerelease {\n\t\t\tbaseVersion := fmt.Sprintf(\"v%d.%d.%d\",\n\t\t\t\tparsedTag.Version.Major(),\n\t\t\t\tparsedTag.Version.Minor(),\n\t\t\t\tparsedTag.Version.Patch())\n\n\t\t\tif rm.tagCache.PrereleaseCache[baseVersion] == nil {\n\t\t\t\trm.tagCache.PrereleaseCache[baseVersion] = make([]int, 0)\n\t\t\t}\n\t\t\trm.tagCache.PrereleaseCache[baseVersion] = append(\n\t\t\t\trm.tagCache.PrereleaseCache[baseVersion], parsedTag.PrereleaseNum)\n\t\t}\n\t}\n\n\t// Sort and cache highest version\n\tif len(rm.tagCache.VersionTags) > 0 {\n\t\trm.sortVersionTags()\n\t\trm.tagCache.HighestVersion = rm.tagCache.VersionTags[len(rm.tagCache.VersionTags)-1].Version\n\t}\n\n\trm.logDebug(\"Known releases total tags (%d), version_tags(%d)\", len(rm.tagCache.AllTags), len(rm.tagCache.VersionTags))\n\n\treturn nil\n}\n\nfunc (rm *Manager) parseTag(rawTag string) ParsedTag {\n\tparsed := ParsedTag{Raw: rawTag}\n\n\t// Extract module path and version part\n\tif idx := strings.LastIndex(rawTag, \"/v\"); idx != -1 {\n\t\tparsed.ModulePath = rawTag[:idx]\n\t\tversionPart := rawTag[idx+1:]\n\n\t\tif versionRegex.MatchString(versionPart) {\n\t\t\tif version, err := semver.NewVersion(versionPart); err == nil {\n\t\t\t\tparsed.Version = version\n\n\t\t\t\t// Check for prerelease\n\t\t\t\tif matches := prereleaseRegex.FindStringSubmatch(versionPart); len(matches) > 2 {\n\t\t\t\t\tparsed.IsPrerelease = true\n\t\t\t\t\tif num, err := strconv.Atoi(matches[2]); err == nil {\n\t\t\t\t\t\tparsed.PrereleaseNum = num\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\treturn parsed\n\t}\n\n\t// Root module tag\n\tif versionRegex.MatchString(rawTag) {\n\t\tif version, err := semver.NewVersion(rawTag); err == nil {\n\t\t\tparsed.Version = version\n\n\t\t\tif matches := prereleaseRegex.FindStringSubmatch(rawTag); len(matches) > 2 {\n\t\t\t\tparsed.IsPrerelease = true\n\t\t\t\tif num, err := strconv.Atoi(matches[2]); err == nil {\n\t\t\t\t\tparsed.PrereleaseNum = num\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn parsed\n}\n\nfunc (rm *Manager) sortVersionTags() {\n\tsort.Slice(rm.tagCache.VersionTags, func(i, j int) bool {\n\t\treturn rm.tagCache.VersionTags[i].Version.LessThan(rm.tagCache.VersionTags[j].Version)\n\t})\n}\n\nfunc (rm *Manager) GetCurrentGlobalVersion() string {\n\tif rm.tagCache.HighestVersion == nil {\n\t\treturn \"v0.0.0\"\n\t}\n\n\treturn \"v\" + rm.tagCache.HighestVersion.String()\n}\n\nfunc (rm *Manager) GetNextPrereleaseVersion(baseVersionStr string) (string, error) {\n\t// Parse base version\n\tbaseVersion, err := semver.NewVersion(baseVersionStr)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"failed to parse base version %s: %w\", baseVersionStr, err)\n\t}\n\n\tcleanBaseStr := fmt.Sprintf(\"v%d.%d.%d\", baseVersion.Major(), baseVersion.Minor(), baseVersion.Patch())\n\n\tprereleaseNumbers := rm.tagCache.PrereleaseCache[cleanBaseStr]\n\tif len(prereleaseNumbers) == 0 {\n\t\treturn fmt.Sprintf(\"%s-prerelease01\", cleanBaseStr), nil\n\t}\n\n\t// Sort and get next number\n\tsort.Ints(prereleaseNumbers)\n\tnextNum := prereleaseNumbers[len(prereleaseNumbers)-1] + 1\n\n\tif nextNum > 99 {\n\t\treturn \"\", fmt.Errorf(\"maximum prerelease number (99) exceeded, base (%s)\", cleanBaseStr)\n\t}\n\n\treturn fmt.Sprintf(\"%s-prerelease%02d\", cleanBaseStr, nextNum), nil\n}\n\nfunc (rm *Manager) CheckVersionExists(version string, modules []Module) (VersionConflictInfo, error) {\n\tconflictInfo := VersionConflictInfo{\n\t\tExistingTags: make([]string, 0),\n\t\tMissingTags:  make([]string, 0),\n\t}\n\n\tfor _, module := range modules {\n\t\texpectedTag := version\n\t\tif module.Path != \"\" {\n\t\t\texpectedTag = module.Path + \"/\" + version\n\t\t}\n\n\t\texists := false\n\t\tfor _, tag := range rm.tagCache.AllTags {\n\t\t\tif tag.Raw == expectedTag {\n\t\t\t\tconflictInfo.ExistingTags = append(conflictInfo.ExistingTags, expectedTag)\n\t\t\t\texists = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tif !exists {\n\t\t\tconflictInfo.MissingTags = append(conflictInfo.MissingTags, expectedTag)\n\t\t}\n\t}\n\n\tif len(conflictInfo.ExistingTags) > 0 {\n\t\treturn conflictInfo, fmt.Errorf(\"some tags already exist: %v\", conflictInfo.ExistingTags)\n\t}\n\n\treturn conflictInfo, nil\n}\n\n// AssessCurrentState gathers repository state (assumes cache is already populated)\nfunc (rm *Manager) AssessCurrentState(ctx context.Context) (*State, error) {\n\tstate := &State{}\n\n\tvar err error\n\t// Gather information (cache should already be populated)\n\tstate.CurrentBranch, err = rm.git.GetCurrentBranch(ctx)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get current branch: %w\", err)\n\t}\n\tstate.Modules, err = rm.FindModules(ctx)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"find modules: %w\", err)\n\t}\n\tstate.CurrentVersion = rm.GetCurrentGlobalVersion()\n\tstate.TagCache = rm.tagCache\n\n\treturn state, nil\n}\n\n// ShowCurrentState displays current release state\nfunc (rm *Manager) ShowCurrentState(ctx context.Context) error {\n\t// Initialize tag cache\n\tif err := rm.GetKnownReleases(ctx); err != nil {\n\t\treturn err\n\t}\n\n\tstate, err := rm.AssessCurrentState(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfmt.Printf(\"Repository Release Status\\n\")\n\tfmt.Printf(\"========================\\n\")\n\tfmt.Printf(\"Branch: %s\\n\", state.CurrentBranch)\n\tfmt.Printf(\"Global Version: %s\\n\", state.CurrentVersion)\n\tfmt.Printf(\"\\n\")\n\n\tfmt.Printf(\"Modules and Versions:\\n\")\n\tfor _, module := range state.Modules {\n\t\tmoduleName := module.Path\n\t\tif moduleName == \"\" {\n\t\t\tmoduleName = \"root\"\n\t\t}\n\t\tfmt.Printf(\"  %-20s %s\\n\", moduleName, module.Version)\n\t}\n\n\t// Show what commands are available\n\tfmt.Printf(\"\\nAvailable Commands:\\n\")\n\tif strings.Contains(state.CurrentVersion, \"-prerelease\") {\n\t\tfmt.Printf(\"  releaser prerelease              # Increment prerelease number\\n\")\n\t\tfmt.Printf(\"  releaser release                 # Promote to final release\\n\")\n\t\tfmt.Printf(\"  releaser release -s v1.4.0       # Override with specific version\\n\")\n\t} else {\n\t\tfmt.Printf(\"  releaser minor                   # Start new minor version cycle\\n\")\n\t\tfmt.Printf(\"  releaser major                   # Start new major version cycle\\n\")\n\t\tfmt.Printf(\"  releaser patch                   # Start new patch version cycle\\n\")\n\t\tfmt.Printf(\"  releaser minor -s v1.4.0-prerelease01  # Override with specific version\\n\")\n\t}\n\n\treturn nil\n}\n\n// ShowPlannedActions displays what actions will be performed\nfunc (rm *Manager) ShowPlannedActions(actions []Action) {\n\tif len(actions) == 0 {\n\t\tfmt.Println(\"No actions planned\")\n\t\treturn\n\t}\n\n\tfmt.Println(\"\\nPlanned Release Actions:\")\n\n\t// Group actions by type for better readability\n\tcreateActions := make([]Action, 0)\n\tpushActions := make([]Action, 0)\n\n\tfor _, action := range actions {\n\t\tswitch action.Type {\n\t\tcase ActionCreateTag:\n\t\t\tcreateActions = append(createActions, action)\n\t\tcase ActionPushTags:\n\t\t\tpushActions = append(pushActions, action)\n\t\t}\n\t}\n\n\tif len(createActions) > 0 {\n\t\tfmt.Println(\"\\nCreate Tags:\")\n\t\tfor _, action := range createActions {\n\t\t\tfmt.Printf(\"  git tag %s\\n\", action.Target)\n\t\t}\n\t}\n\n\tif len(pushActions) > 0 {\n\t\tfmt.Println(\"\\nPush Tags:\")\n\t\tfor _, action := range pushActions {\n\t\t\tfmt.Printf(\"  git push origin %s\\n\", action.Target)\n\t\t}\n\t}\n}\n\nfunc (rm *Manager) planReleaseActions(state *State, targetVersion string) ([]Action, []Warning) {\n\tvar actions []Action\n\tvar warnings []Warning\n\n\t// Add actions for each module\n\tfor _, module := range state.Modules {\n\t\ttagName := rm.getTagName(module, targetVersion)\n\n\t\tactions = append(actions, Action{\n\t\t\tType:        ActionCreateTag,\n\t\t\tTarget:      tagName,\n\t\t\tDescription: fmt.Sprintf(\"Create tag %s\", tagName),\n\t\t})\n\n\t\tactions = append(actions, Action{\n\t\t\tType:        ActionPushTags,\n\t\t\tTarget:      tagName,\n\t\t\tDescription: fmt.Sprintf(\"Push tag %s\", tagName),\n\t\t})\n\t}\n\n\t// Add warnings\n\twarnings = append(warnings, rm.validateWithWarnings(state, targetVersion)...)\n\n\treturn actions, warnings\n}\n\nfunc (rm *Manager) validateWithWarnings(state *State, targetVersion string) []Warning {\n\tvar warnings []Warning\n\n\t// Branch check -> warning\n\tif rm.config.RequiredBranch != \"\" && state.CurrentBranch != rm.config.RequiredBranch {\n\t\twarnings = append(warnings, Warning{\n\t\t\tType:    WrongBranch,\n\t\t\tMessage: fmt.Sprintf(\"you are not on %s\", rm.config.RequiredBranch),\n\t\t})\n\t}\n\n\treturn warnings\n}\n\n// filterActionsForMissingTags filters actions to only include missing tags\nfunc (rm *Manager) filterActionsForMissingTags(actions []Action, missingTags []string) []Action {\n\tvar filteredActions []Action\n\n\t// Create a set of missing tags for quick lookup\n\tmissingTagSet := make(map[string]bool)\n\tfor _, tag := range missingTags {\n\t\tmissingTagSet[tag] = true\n\t}\n\n\t// Filter actions to only include missing tags\n\tfor _, action := range actions {\n\t\tif missingTagSet[action.Target] {\n\t\t\tfilteredActions = append(filteredActions, action)\n\t\t}\n\t}\n\n\treturn filteredActions\n}\n\n// getLatestVersionForModule returns the latest version for a given module path\nfunc (rm *Manager) getLatestVersionForModule(modulePath string) string {\n\t// Handle case where tag cache isn't initialized yet\n\tif rm.tagCache == nil || rm.tagCache.ModuleTags == nil {\n\t\treturn \"v0.0.0\" // Default for modules with no releases yet\n\t}\n\n\tmoduleTags, exists := rm.tagCache.ModuleTags[modulePath]\n\tif !exists || len(moduleTags) == 0 {\n\t\treturn \"v0.0.0\" // No releases for this module yet\n\t}\n\n\t// Find the latest version among all tags for this module\n\tvar latestVersion *semver.Version\n\tfor _, tag := range moduleTags {\n\t\tif tag.Version != nil {\n\t\t\tif latestVersion == nil || tag.Version.GreaterThan(latestVersion) {\n\t\t\t\tlatestVersion = tag.Version\n\t\t\t}\n\t\t}\n\t}\n\n\tif latestVersion == nil {\n\t\treturn \"v0.0.0\"\n\t}\n\n\treturn \"v\" + latestVersion.String()\n}\n\nfunc (rm *Manager) handleWarningsAndConfirmations(ctx context.Context, warnings []Warning) error {\n\tfor _, warning := range warnings {\n\t\tfmt.Printf(\"⚠️  %s\\n\", warning.Message)\n\n\t\tif rm.config.SkipConfirmations {\n\t\t\tcontinue\n\t\t}\n\n\t\tconfirmed, err := rm.interaction.ConfirmWithDefault(ctx, \"Continue?\", false)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !confirmed {\n\t\t\treturn fmt.Errorf(\"operation cancelled due to: %s\", warning.Message)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (rm *Manager) executeTagCreation(ctx context.Context, actions []Action) error {\n\tfmt.Println(\"Creating tags...\")\n\tfor _, action := range actions {\n\t\tif action.Type == ActionCreateTag {\n\t\t\tif err := rm.git.CreateTag(ctx, action.Target); err != nil {\n\t\t\t\treturn fmt.Errorf(\"failed to create tag %s: %w\", action.Target, err)\n\t\t\t}\n\t\t\tfmt.Printf(\"✓ Created tag %s\\n\", action.Target)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (rm *Manager) executeTagPushing(ctx context.Context, actions []Action) error {\n\tfmt.Println(\"Pushing tags...\")\n\tfor _, action := range actions {\n\t\tif action.Type == ActionPushTags {\n\t\t\tif err := rm.git.PushTag(ctx, action.Target); err != nil {\n\t\t\t\treturn fmt.Errorf(\"failed to push tag %s: %w\", action.Target, err)\n\t\t\t}\n\t\t\tfmt.Printf(\"✓ Pushed tag %s\\n\", action.Target)\n\t\t}\n\t}\n\treturn nil\n}\n\n// NormalizeVersion ensures version has 'v' prefix and is valid semver\nfunc NormalizeVersion(v string) (string, error) {\n\tif !strings.HasPrefix(v, \"v\") {\n\t\tv = \"v\" + v\n\t}\n\n\t// Parse with Masterminds/semver to validate\n\tsemVer, err := semver.NewVersion(v)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"invalid semantic version: %s\", v)\n\t}\n\n\treturn \"v\" + semVer.String(), nil\n}\n\n// IncrementVersion increments a version based on type\nfunc IncrementVersion(currentVersionStr, versionType string) (string, error) {\n\t// Parse the current version\n\tcurrentVersion, err := semver.NewVersion(currentVersionStr)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"failed to parse current version %s: %w\", currentVersionStr, err)\n\t}\n\n\tvar newVersion semver.Version\n\tswitch versionType {\n\tcase \"major\":\n\t\tnewVersion = currentVersion.IncMajor()\n\tcase \"minor\":\n\t\tnewVersion = currentVersion.IncMinor()\n\tcase \"patch\":\n\t\tnewVersion = currentVersion.IncPatch()\n\tdefault:\n\t\treturn \"\", fmt.Errorf(\"invalid version type: %s\", versionType)\n\t}\n\n\treturn \"v\" + newVersion.String(), nil\n}\n\n// FindModules discovers all Go modules in the repository\nfunc (rm *Manager) FindModules(ctx context.Context) ([]Module, error) {\n\trm.logDebug(\"Discovering Go modules in path %s\", rm.config.RepoRoot)\n\tgoModPaths, err := rm.fs.FindGoModFiles(ctx, rm.config.RepoRoot)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to find go.mod files: %w\", err)\n\t}\n\n\tvar modules []Module\n\tseen := make(map[string]bool)\n\n\tfor _, path := range goModPaths {\n\n\t\t// Normalize relative path\n\t\tif path == \".\" {\n\t\t\tpath = \"\"\n\t\t}\n\t\tpath = strings.TrimPrefix(path, \"./\")\n\n\t\t// Check if should be excluded\n\t\tif rm.shouldExcludeModule(path) {\n\t\t\trm.logDebug(\"Excluding module %s\", path)\n\t\t\tcontinue\n\t\t}\n\n\t\t// Deduplicate\n\t\tif seen[path] {\n\t\t\tcontinue\n\t\t}\n\t\tseen[path] = true\n\n\t\t// Get latest version for this module from cache\n\t\tlatestVersion := rm.getLatestVersionForModule(path)\n\n\t\tmodules = append(modules, Module{\n\t\t\tPath:    path,\n\t\t\tVersion: latestVersion,\n\t\t})\n\t\trm.logDebug(\"Found module\\n%s\\nversion: %s\\n\", path, latestVersion)\n\t}\n\n\treturn modules, nil\n}\n\n// shouldExcludeModule checks if a module should be excluded\nfunc (rm *Manager) shouldExcludeModule(relPath string) bool {\n\tfor _, excluded := range rm.config.ExcludedDirs {\n\t\tif relPath == excluded || strings.HasPrefix(relPath, excluded+\"/\") {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// getTagName generates the tag name for a module and version\nfunc (rm *Manager) getTagName(module Module, version string) string {\n\tif module.Path == \"\" {\n\t\treturn version\n\t}\n\treturn module.Path + \"/\" + version\n}\n\nfunc (rm *Manager) logDebug(msg string, args ...interface{}) {\n\tif rm.config.Verbose {\n\t\tfmt.Printf(\"DEBUG: %s\\n\", fmt.Sprintf(msg, args...))\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/release_mocks_test.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: release.go\n//\n// Generated by this command:\n//\n//\tmockgen -package release -source release.go -destination release_mocks_test.go -self_package github.com/uber/cadence/cmd/tools/releaser/release\n//\n\n// Package release is a generated GoMock package.\npackage release\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockGit is a mock of Git interface.\ntype MockGit struct {\n\tctrl     *gomock.Controller\n\trecorder *MockGitMockRecorder\n\tisgomock struct{}\n}\n\n// MockGitMockRecorder is the mock recorder for MockGit.\ntype MockGitMockRecorder struct {\n\tmock *MockGit\n}\n\n// NewMockGit creates a new mock instance.\nfunc NewMockGit(ctrl *gomock.Controller) *MockGit {\n\tmock := &MockGit{ctrl: ctrl}\n\tmock.recorder = &MockGitMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockGit) EXPECT() *MockGitMockRecorder {\n\treturn m.recorder\n}\n\n// CreateTag mocks base method.\nfunc (m *MockGit) CreateTag(ctx context.Context, tag string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTag\", ctx, tag)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateTag indicates an expected call of CreateTag.\nfunc (mr *MockGitMockRecorder) CreateTag(ctx, tag any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTag\", reflect.TypeOf((*MockGit)(nil).CreateTag), ctx, tag)\n}\n\n// GetCurrentBranch mocks base method.\nfunc (m *MockGit) GetCurrentBranch(ctx context.Context) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentBranch\", ctx)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCurrentBranch indicates an expected call of GetCurrentBranch.\nfunc (mr *MockGitMockRecorder) GetCurrentBranch(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentBranch\", reflect.TypeOf((*MockGit)(nil).GetCurrentBranch), ctx)\n}\n\n// GetRepoRoot mocks base method.\nfunc (m *MockGit) GetRepoRoot(ctx context.Context) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRepoRoot\", ctx)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRepoRoot indicates an expected call of GetRepoRoot.\nfunc (mr *MockGitMockRecorder) GetRepoRoot(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRepoRoot\", reflect.TypeOf((*MockGit)(nil).GetRepoRoot), ctx)\n}\n\n// GetTags mocks base method.\nfunc (m *MockGit) GetTags(ctx context.Context) ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTags\", ctx)\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTags indicates an expected call of GetTags.\nfunc (mr *MockGitMockRecorder) GetTags(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTags\", reflect.TypeOf((*MockGit)(nil).GetTags), ctx)\n}\n\n// PushTag mocks base method.\nfunc (m *MockGit) PushTag(ctx context.Context, tag string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PushTag\", ctx, tag)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PushTag indicates an expected call of PushTag.\nfunc (mr *MockGitMockRecorder) PushTag(ctx, tag any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PushTag\", reflect.TypeOf((*MockGit)(nil).PushTag), ctx, tag)\n}\n\n// MockFS is a mock of FS interface.\ntype MockFS struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFSMockRecorder\n\tisgomock struct{}\n}\n\n// MockFSMockRecorder is the mock recorder for MockFS.\ntype MockFSMockRecorder struct {\n\tmock *MockFS\n}\n\n// NewMockFS creates a new mock instance.\nfunc NewMockFS(ctrl *gomock.Controller) *MockFS {\n\tmock := &MockFS{ctrl: ctrl}\n\tmock.recorder = &MockFSMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFS) EXPECT() *MockFSMockRecorder {\n\treturn m.recorder\n}\n\n// FindGoModFiles mocks base method.\nfunc (m *MockFS) FindGoModFiles(ctx context.Context, root string) ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FindGoModFiles\", ctx, root)\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FindGoModFiles indicates an expected call of FindGoModFiles.\nfunc (mr *MockFSMockRecorder) FindGoModFiles(ctx, root any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FindGoModFiles\", reflect.TypeOf((*MockFS)(nil).FindGoModFiles), ctx, root)\n}\n\n// MockUserInteraction is a mock of UserInteraction interface.\ntype MockUserInteraction struct {\n\tctrl     *gomock.Controller\n\trecorder *MockUserInteractionMockRecorder\n\tisgomock struct{}\n}\n\n// MockUserInteractionMockRecorder is the mock recorder for MockUserInteraction.\ntype MockUserInteractionMockRecorder struct {\n\tmock *MockUserInteraction\n}\n\n// NewMockUserInteraction creates a new mock instance.\nfunc NewMockUserInteraction(ctrl *gomock.Controller) *MockUserInteraction {\n\tmock := &MockUserInteraction{ctrl: ctrl}\n\tmock.recorder = &MockUserInteractionMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockUserInteraction) EXPECT() *MockUserInteractionMockRecorder {\n\treturn m.recorder\n}\n\n// Confirm mocks base method.\nfunc (m *MockUserInteraction) Confirm(ctx context.Context, message string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Confirm\", ctx, message)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Confirm indicates an expected call of Confirm.\nfunc (mr *MockUserInteractionMockRecorder) Confirm(ctx, message any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Confirm\", reflect.TypeOf((*MockUserInteraction)(nil).Confirm), ctx, message)\n}\n\n// ConfirmWithDefault mocks base method.\nfunc (m *MockUserInteraction) ConfirmWithDefault(ctx context.Context, message string, defaultValue bool) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ConfirmWithDefault\", ctx, message, defaultValue)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ConfirmWithDefault indicates an expected call of ConfirmWithDefault.\nfunc (mr *MockUserInteractionMockRecorder) ConfirmWithDefault(ctx, message, defaultValue any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ConfirmWithDefault\", reflect.TypeOf((*MockUserInteraction)(nil).ConfirmWithDefault), ctx, message, defaultValue)\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/release_test.go",
    "content": "package release\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n)\n\nfunc TestManager_RunRelease(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t{\n\t\t\tname:           \"successful release from prerelease\",\n\t\t\tcurrentTags:    []string{\"v1.2.3-prerelease01\"},\n\t\t\texpectedTarget: \"v1.2.3\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:        \"error when no prerelease exists\",\n\t\t\tcurrentTags: []string{\"v1.2.3\"},\n\t\t\tshouldError: true,\n\t\t\terrorMsg:    \"release command requires existing prerelease version\",\n\t\t},\n\t\t{\n\t\t\tname:           \"release from higher prerelease number\",\n\t\t\tcurrentTags:    []string{\"v1.2.3-prerelease05\"},\n\t\t\texpectedTarget: \"v1.2.3\",\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"release\",\n\t\t\t\tSkipConfirmations: true, // Skip confirmations for testing\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\n\t\t\tif !tt.shouldError {\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t}\n\n\t\t\terr := manager.RunRelease(context.Background())\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestManager_RunMinor(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t{\n\t\t\tname:           \"successful minor from stable\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v1.3.0-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:        \"error when current is prerelease\",\n\t\t\tcurrentTags: []string{\"v1.2.3-prerelease01\"},\n\t\t\tshouldError: true,\n\t\t\terrorMsg:    \"minor/major commands should be run from stable versions\",\n\t\t},\n\t\t{\n\t\t\tname:           \"minor from initial version\",\n\t\t\tcurrentTags:    []string{},\n\t\t\texpectedTarget: \"v0.1.0-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"minor\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\n\t\t\tif !tt.shouldError {\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t}\n\n\t\t\terr := manager.RunMinor(context.Background())\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestManager_RunMajor(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t\tshouldError    bool\n\t}{\n\t\t{\n\t\t\tname:           \"successful major from stable\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v2.0.0-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"major from initial version\",\n\t\t\tcurrentTags:    []string{},\n\t\t\texpectedTarget: \"v1.0.0-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"major\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\n\t\t\terr := manager.RunMajor(context.Background())\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n}\n\nfunc TestManager_RunPrerelease(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t}{\n\t\t{\n\t\t\tname:           \"increment prerelease number\",\n\t\t\tcurrentTags:    []string{\"v1.2.3-prerelease01\"},\n\t\t\texpectedTarget: \"v1.2.3-prerelease02\",\n\t\t},\n\t\t{\n\t\t\tname:           \"create first prerelease from stable\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v1.2.3-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:           \"increment from higher prerelease\",\n\t\t\tcurrentTags:    []string{\"v1.2.3-prerelease05\"},\n\t\t\texpectedTarget: \"v1.2.3-prerelease06\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"prerelease\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\n\t\t\terr := manager.RunPrerelease(context.Background())\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n}\n\nfunc TestManager_ManualVersionOverride(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmanualVersion  string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t{\n\t\t\tname:           \"valid manual version\",\n\t\t\tmanualVersion:  \"v2.5.0\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v2.5.0\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"manual version without v prefix\",\n\t\t\tmanualVersion:  \"2.5.0\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v2.5.0\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:          \"invalid manual version\",\n\t\t\tmanualVersion: \"invalid\",\n\t\t\tcurrentTags:   []string{\"v1.2.3\"},\n\t\t\tshouldError:   true,\n\t\t\terrorMsg:      \"invalid manual version format\",\n\t\t},\n\t\t{\n\t\t\tname:           \"manual prerelease version\",\n\t\t\tmanualVersion:  \"v2.5.0-prerelease03\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v2.5.0-prerelease03\",\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"release\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t\tManualVersion:     tt.manualVersion,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\n\t\t\tif !tt.shouldError {\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\t}\n\n\t\t\terr := manager.RunRelease(context.Background())\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestManager_ConflictResolution(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttargetVersion  string\n\t\texistingTags   []string\n\t\tmodules        []Module\n\t\texpectedCreate []string\n\t\texpectedSkip   []string\n\t\tshouldError    bool\n\t}{\n\t\t{\n\t\t\tname:          \"partial conflict - some modules have version\",\n\t\t\ttargetVersion: \"v1.3.0\",\n\t\t\texistingTags:  []string{\"moduleA/v1.3.0\"},\n\t\t\tmodules: []Module{\n\t\t\t\t{Path: \"moduleA\", Version: \"v1.2.0\"},\n\t\t\t\t{Path: \"moduleB\", Version: \"v1.2.0\"},\n\t\t\t\t{Path: \"\", Version: \"v1.2.0\"}, // root module\n\t\t\t},\n\t\t\texpectedCreate: []string{\"moduleB/v1.3.0\", \"v1.3.0\"},\n\t\t\texpectedSkip:   []string{\"moduleA/v1.3.0\"},\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:          \"complete conflict - all modules have version\",\n\t\t\ttargetVersion: \"v1.3.0\",\n\t\t\texistingTags:  []string{\"moduleA/v1.3.0\", \"moduleB/v1.3.0\", \"v1.3.0\"},\n\t\t\tmodules: []Module{\n\t\t\t\t{Path: \"moduleA\", Version: \"v1.2.0\"},\n\t\t\t\t{Path: \"moduleB\", Version: \"v1.2.0\"},\n\t\t\t\t{Path: \"\", Version: \"v1.2.0\"},\n\t\t\t},\n\t\t\texpectedCreate: []string{},\n\t\t\texpectedSkip:   []string{\"moduleA/v1.3.0\", \"moduleB/v1.3.0\", \"v1.3.0\"},\n\t\t\tshouldError:    true,\n\t\t},\n\t\t{\n\t\t\tname:          \"no conflict\",\n\t\t\ttargetVersion: \"v1.3.0\",\n\t\t\texistingTags:  []string{},\n\t\t\tmodules: []Module{\n\t\t\t\t{Path: \"moduleA\", Version: \"v1.2.0\"},\n\t\t\t\t{Path: \"moduleB\", Version: \"v1.2.0\"},\n\t\t\t},\n\t\t\texpectedCreate: []string{\"moduleA/v1.3.0\", \"moduleB/v1.3.0\"},\n\t\t\texpectedSkip:   []string{},\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"release\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t\tManualVersion:     tt.targetVersion,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup tag cache\n\t\t\tmanager.tagCache = &TagCache{\n\t\t\t\tAllTags: make([]ParsedTag, 0),\n\t\t\t}\n\n\t\t\t// Parse existing tags\n\t\t\tfor _, tag := range tt.existingTags {\n\t\t\t\tmanager.tagCache.AllTags = append(manager.tagCache.AllTags, ParsedTag{Raw: tag})\n\t\t\t}\n\n\t\t\tconflictInfo, err := manager.CheckVersionExists(tt.targetVersion, tt.modules)\n\n\t\t\tif tt.shouldError && len(tt.expectedCreate) == 0 {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.ElementsMatch(t, tt.expectedSkip, conflictInfo.ExistingTags)\n\t\t\t\tassert.Empty(t, conflictInfo.MissingTags)\n\t\t\t} else {\n\t\t\t\tif len(tt.expectedSkip) > 0 {\n\t\t\t\t\trequire.Error(t, err) // Should have conflict error\n\t\t\t\t\tassert.ElementsMatch(t, tt.expectedSkip, conflictInfo.ExistingTags)\n\t\t\t\t} else {\n\t\t\t\t\trequire.NoError(t, err) // No conflict\n\t\t\t\t}\n\t\t\t\tassert.ElementsMatch(t, tt.expectedCreate, conflictInfo.MissingTags)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestManager_MultiModuleOperations(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockGit := NewMockGit(ctrl)\n\tmockFS := NewMockFS(ctrl)\n\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\tconfig := Config{\n\t\tRepoRoot:          \"/test\",\n\t\tExcludedDirs:      []string{\"cmd\", \"internal/tools\"},\n\t\tRequiredBranch:    \"master\",\n\t\tVerbose:           false,\n\t\tCommand:           \"minor\",\n\t\tSkipConfirmations: true,\n\t}\n\n\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t// Setup mocks for multi-module repo\n\tcurrentTags := []string{\n\t\t\"v1.2.3\",\n\t\t\"service1/v1.2.3\",\n\t\t\"service2/v1.2.3\",\n\t}\n\tgoModPaths := []string{\".\", \"service1\", \"service2\", \"cmd/tool\", \"internal/tools/helper\"}\n\n\texpectedTags := []string{\n\t\t\"v1.3.0-prerelease01\",\n\t\t\"service1/v1.3.0-prerelease01\",\n\t\t\"service2/v1.3.0-prerelease01\",\n\t}\n\n\tmockGit.EXPECT().GetTags(gomock.Any()).Return(currentTags, nil)\n\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return(goModPaths, nil)\n\n\t// Expect tag creation for non-excluded modules only\n\tfor _, tag := range expectedTags {\n\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tag).Return(nil)\n\t\tmockGit.EXPECT().PushTag(gomock.Any(), tag).Return(nil)\n\t}\n\n\terr := manager.RunMinor(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestManager_ErrorHandling(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tsetupMock   func(*MockGit, *MockFS, *MockUserInteraction)\n\t\texpectedErr string\n\t}{\n\t\t{\n\t\t\tname: \"git tags fetch error\",\n\t\t\tsetupMock: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(nil, errors.New(\"git error\"))\n\t\t\t},\n\t\t\texpectedErr: \"failed to fetch tags\",\n\t\t},\n\t\t{\n\t\t\tname: \"git branch fetch error\",\n\t\t\tsetupMock: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"\", errors.New(\"branch error\"))\n\t\t\t},\n\t\t\texpectedErr: \"get current branch\",\n\t\t},\n\t\t{\n\t\t\tname: \"fs find modules error\",\n\t\t\tsetupMock: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return(nil, errors.New(\"fs error\"))\n\t\t\t},\n\t\t\texpectedErr: \"find modules\",\n\t\t},\n\t\t{\n\t\t\tname: \"tag creation error\",\n\t\t\tsetupMock: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), \"v1.3.0-prerelease01\").Return(errors.New(\"tag error\"))\n\t\t\t},\n\t\t\texpectedErr: \"failed to create tag\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"minor\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\ttt.setupMock(mockGit, mockFS, mockInteraction)\n\n\t\t\terr := manager.RunMinor(context.Background())\n\t\t\trequire.Error(t, err)\n\t\t\tassert.Contains(t, err.Error(), tt.expectedErr)\n\t\t})\n\t}\n}\n\nfunc TestManager_RunPatch(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentTags    []string\n\t\texpectedTarget string\n\t\tshouldError    bool\n\t}{\n\t\t{\n\t\t\tname:           \"successful patch from stable\",\n\t\t\tcurrentTags:    []string{\"v1.2.3\"},\n\t\t\texpectedTarget: \"v1.2.4-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"patch from initial version\",\n\t\t\tcurrentTags:    []string{},\n\t\t\texpectedTarget: \"v0.0.1-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"patch\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedTarget).Return(nil)\n\n\t\t\terr := manager.RunPatch(context.Background())\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/scenario_test.go",
    "content": "package release\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n)\n\nfunc TestCompleteReleaseWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tworkflow    []workflowStep\n\t\tdescription string\n\t}{\n\t\t{\n\t\t\tname:        \"feature development cycle\",\n\t\t\tdescription: \"Complete feature development from minor through prerelease iterations to final release\",\n\t\t\tworkflow: []workflowStep{\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"minor\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.2.3\"},\n\t\t\t\t\texpectedTag:   \"v1.3.0-prerelease01\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"prerelease\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.2.3\", \"v1.3.0-prerelease01\"},\n\t\t\t\t\texpectedTag:   \"v1.3.0-prerelease02\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"prerelease\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.2.3\", \"v1.3.0-prerelease01\", \"v1.3.0-prerelease02\"},\n\t\t\t\t\texpectedTag:   \"v1.3.0-prerelease03\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"release\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.2.3\", \"v1.3.0-prerelease01\", \"v1.3.0-prerelease02\", \"v1.3.0-prerelease03\"},\n\t\t\t\t\texpectedTag:   \"v1.3.0\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"major version cycle\",\n\t\t\tdescription: \"Major version development cycle with breaking changes\",\n\t\t\tworkflow: []workflowStep{\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"major\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.3.0\"},\n\t\t\t\t\texpectedTag:   \"v2.0.0-prerelease01\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"prerelease\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.3.0\", \"v2.0.0-prerelease01\"},\n\t\t\t\t\texpectedTag:   \"v2.0.0-prerelease02\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"release\",\n\t\t\t\t\tcurrentTags:   []string{\"v1.3.0\", \"v2.0.0-prerelease01\", \"v2.0.0-prerelease02\"},\n\t\t\t\t\texpectedTag:   \"v2.0.0\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"initial repository setup\",\n\t\t\tdescription: \"Starting from empty repository to first release\",\n\t\t\tworkflow: []workflowStep{\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"minor\",\n\t\t\t\t\tcurrentTags:   []string{},\n\t\t\t\t\texpectedTag:   \"v0.1.0-prerelease01\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcommand:       \"release\",\n\t\t\t\t\tcurrentTags:   []string{\"v0.1.0-prerelease01\"},\n\t\t\t\t\texpectedTag:   \"v0.1.0\",\n\t\t\t\t\texpectedError: false,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tt.Logf(\"Testing workflow: %s\", tt.description)\n\n\t\t\tfor i, step := range tt.workflow {\n\t\t\t\tt.Logf(\"Step %d: %s command\", i+1, step.command)\n\n\t\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\t\tconfig := Config{\n\t\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\t\tVerbose:           false,\n\t\t\t\t\tCommand:           step.command,\n\t\t\t\t\tSkipConfirmations: true,\n\t\t\t\t}\n\n\t\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t\t// Setup mocks\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(step.currentTags, nil)\n\n\t\t\t\tif !step.expectedError {\n\t\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), step.expectedTag).Return(nil)\n\t\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), step.expectedTag).Return(nil)\n\t\t\t\t}\n\n\t\t\t\t// Execute command\n\t\t\t\tvar err error\n\t\t\t\tswitch step.command {\n\t\t\t\tcase \"minor\":\n\t\t\t\t\terr = manager.RunMinor(context.Background())\n\t\t\t\tcase \"major\":\n\t\t\t\t\terr = manager.RunMajor(context.Background())\n\t\t\t\tcase \"prerelease\":\n\t\t\t\t\terr = manager.RunPrerelease(context.Background())\n\t\t\t\tcase \"release\":\n\t\t\t\t\terr = manager.RunRelease(context.Background())\n\t\t\t\t}\n\n\t\t\t\tif step.expectedError {\n\t\t\t\t\trequire.Error(t, err)\n\t\t\t\t} else {\n\t\t\t\t\trequire.NoError(t, err)\n\t\t\t\t}\n\n\t\t\t\tctrl.Finish()\n\t\t\t}\n\t\t})\n\t}\n}\n\ntype workflowStep struct {\n\tcommand       string\n\tcurrentTags   []string\n\texpectedTag   string\n\texpectedError bool\n}\n\nfunc TestMultiModuleComplexScenarios(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tcurrentTags     []string\n\t\tmodules         []string\n\t\tcommand         string\n\t\texpectedTags    []string\n\t\texcludedModules []string\n\t}{\n\t\t{\n\t\t\tname: \"multi-module minor release\",\n\t\t\tcurrentTags: []string{\n\t\t\t\t\"v1.2.3\",\n\t\t\t\t\"service1/v1.2.3\",\n\t\t\t\t\"service2/v1.2.3\",\n\t\t\t\t\"cmd/tool/v1.0.0\", // This should be excluded\n\t\t\t},\n\t\t\tmodules: []string{\".\", \"service1\", \"service2\", \"cmd/tool\"},\n\t\t\tcommand: \"minor\",\n\t\t\texpectedTags: []string{\n\t\t\t\t\"v1.3.0-prerelease01\",\n\t\t\t\t\"service1/v1.3.0-prerelease01\",\n\t\t\t\t\"service2/v1.3.0-prerelease01\",\n\t\t\t},\n\t\t\texcludedModules: []string{\"cmd/tool\"},\n\t\t},\n\t\t{\n\t\t\tname: \"multi-module with version conflicts\",\n\t\t\tcurrentTags: []string{\n\t\t\t\t\"v1.2.3\",\n\t\t\t\t\"service1/v1.3.0\", // Already has target version\n\t\t\t\t\"service2/v1.2.3\",\n\t\t\t},\n\t\t\tmodules: []string{\".\", \"service1\", \"service2\"},\n\t\t\tcommand: \"manual\",\n\t\t\texpectedTags: []string{\n\t\t\t\t\"v1.3.0\",          // Root module\n\t\t\t\t\"service2/v1.3.0\", // Service2 missing\n\t\t\t\t// service1/v1.3.0 should be skipped\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      tt.excludedModules,\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           tt.command,\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tif tt.command == \"manual\" {\n\t\t\t\tconfig.ManualVersion = \"v1.3.0\"\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.currentTags, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return(tt.modules, nil)\n\n\t\t\t// Expect tag creation only for expected tags\n\t\t\tfor _, tag := range tt.expectedTags {\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tag).Return(nil)\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tag).Return(nil)\n\t\t\t}\n\n\t\t\t// Execute command\n\t\t\tvar err error\n\t\t\tif tt.command == \"manual\" {\n\t\t\t\terr = manager.RunRelease(context.Background())\n\t\t\t} else {\n\t\t\t\terr = manager.RunMinor(context.Background())\n\t\t\t}\n\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n}\n\nfunc TestErrorRecoveryScenarios(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tdescription   string\n\t\tsetupMocks    func(*MockGit, *MockFS, *MockUserInteraction)\n\t\tcommand       string\n\t\texpectedErr   string\n\t\tshouldRecover bool\n\t}{\n\t\t{\n\t\t\tname:        \"partial tag creation failure\",\n\t\t\tdescription: \"Some tags created successfully, others failed\",\n\t\t\tsetupMocks: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\", \"service1\"}, nil)\n\n\t\t\t\t// First tag succeeds, second fails\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), \"v1.3.0-prerelease01\").Return(nil)\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), \"service1/v1.3.0-prerelease01\").\n\t\t\t\t\tReturn(assert.AnError)\n\t\t\t},\n\t\t\tcommand:     \"minor\",\n\t\t\texpectedErr: \"failed to create tag\",\n\t\t},\n\t\t{\n\t\t\tname:        \"network failure during push\",\n\t\t\tdescription: \"Tags created locally but push failed\",\n\t\t\tsetupMocks: func(mockGit *MockGit, mockFS *MockFS, mockInteraction *MockUserInteraction) {\n\t\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\n\t\t\t\t// Tag creation succeeds\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), \"v1.3.0-prerelease01\").Return(nil)\n\t\t\t\t// Push fails\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), \"v1.3.0-prerelease01\").\n\t\t\t\t\tReturn(assert.AnError)\n\t\t\t},\n\t\t\tcommand:     \"minor\",\n\t\t\texpectedErr: \"push tags\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tt.Logf(\"Testing scenario: %s\", tt.description)\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           tt.command,\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\ttt.setupMocks(mockGit, mockFS, mockInteraction)\n\n\t\t\tvar err error\n\t\t\tswitch tt.command {\n\t\t\tcase \"minor\":\n\t\t\t\terr = manager.RunMinor(context.Background())\n\t\t\tcase \"release\":\n\t\t\t\terr = manager.RunRelease(context.Background())\n\t\t\t}\n\n\t\t\trequire.Error(t, err)\n\t\t\tassert.Contains(t, err.Error(), tt.expectedErr)\n\t\t})\n\t}\n}\n\nfunc TestConcurrentPrereleaseScenarios(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texistingTags    []string\n\t\texpectedVersion string\n\t\tdescription     string\n\t}{\n\t\t{\n\t\t\tname: \"gap in prerelease sequence\",\n\t\t\texistingTags: []string{\n\t\t\t\t\"v1.3.0-prerelease01\",\n\t\t\t\t\"v1.3.0-prerelease03\", // Missing 02\n\t\t\t\t\"v1.3.0-prerelease05\",\n\t\t\t},\n\t\t\texpectedVersion: \"v1.3.0-prerelease06\", // Should continue from highest\n\t\t\tdescription:     \"Should continue from highest prerelease number even with gaps\",\n\t\t},\n\t\t{\n\t\t\tname: \"mixed prerelease versions\",\n\t\t\texistingTags: []string{\n\t\t\t\t\"v1.2.0-prerelease05\",\n\t\t\t\t\"v1.3.0-prerelease01\",\n\t\t\t\t\"v1.3.0-prerelease02\",\n\t\t\t\t\"v1.4.0-prerelease01\",\n\t\t\t},\n\t\t\texpectedVersion: \"v1.4.0-prerelease02\", // Should work with current global version\n\t\t\tdescription:     \"Should handle mixed prerelease versions for different base versions\",\n\t\t},\n\t\t{\n\t\t\tname: \"approaching prerelease limit\",\n\t\t\texistingTags: []string{\n\t\t\t\t\"v1.3.0-prerelease98\",\n\t\t\t},\n\t\t\texpectedVersion: \"v1.3.0-prerelease99\",\n\t\t\tdescription:     \"Should handle high prerelease numbers approaching limit\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tt.Logf(\"Testing scenario: %s\", tt.description)\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"prerelease\",\n\t\t\t\tSkipConfirmations: true,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return(tt.existingTags, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(\"master\", nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), tt.expectedVersion).Return(nil)\n\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), tt.expectedVersion).Return(nil)\n\n\t\t\terr := manager.RunPrerelease(context.Background())\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n}\n\nfunc TestUserInteractionScenarios(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tcurrentBranch     string\n\t\tuserResponses     []bool // Responses to confirmation prompts\n\t\tskipConfirmations bool\n\t\tshouldComplete    bool\n\t\texpectedActions   int // Number of git operations expected\n\t}{\n\t\t{\n\t\t\tname:              \"user confirms all warnings and actions\",\n\t\t\tcurrentBranch:     \"feature-branch\",         // Wrong branch\n\t\t\tuserResponses:     []bool{true, true, true}, // Continue despite warning, create tags, push tags\n\t\t\tskipConfirmations: false,\n\t\t\tshouldComplete:    true,\n\t\t\texpectedActions:   2, // create + push\n\t\t},\n\t\t{\n\t\t\tname:              \"user rejects branch warning\",\n\t\t\tcurrentBranch:     \"feature-branch\",\n\t\t\tuserResponses:     []bool{false}, // Reject continuing despite warning\n\t\t\tskipConfirmations: false,\n\t\t\tshouldComplete:    false,\n\t\t\texpectedActions:   0,\n\t\t},\n\t\t{\n\t\t\tname:              \"user confirms warning but rejects tag creation\",\n\t\t\tcurrentBranch:     \"feature-branch\",\n\t\t\tuserResponses:     []bool{true, false}, // Continue despite warning, but reject tag creation\n\t\t\tskipConfirmations: false,\n\t\t\tshouldComplete:    false,\n\t\t\texpectedActions:   0,\n\t\t},\n\t\t{\n\t\t\tname:              \"user creates tags but skips push\",\n\t\t\tcurrentBranch:     \"feature-branch\",\n\t\t\tuserResponses:     []bool{true, true, false}, // Continue, create, but don't push\n\t\t\tskipConfirmations: false,\n\t\t\tshouldComplete:    false, // Partial completion\n\t\t\texpectedActions:   1,     // Only create\n\t\t},\n\t\t{\n\t\t\tname:              \"skip all confirmations\",\n\t\t\tcurrentBranch:     \"feature-branch\",\n\t\t\tuserResponses:     []bool{}, // No interactions expected\n\t\t\tskipConfirmations: true,\n\t\t\tshouldComplete:    true,\n\t\t\texpectedActions:   2, // create + push\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockGit := NewMockGit(ctrl)\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\n\t\t\tconfig := Config{\n\t\t\t\tRepoRoot:          \"/test\",\n\t\t\t\tExcludedDirs:      []string{},\n\t\t\t\tRequiredBranch:    \"master\",\n\t\t\t\tVerbose:           false,\n\t\t\t\tCommand:           \"minor\",\n\t\t\t\tSkipConfirmations: tt.skipConfirmations,\n\t\t\t}\n\n\t\t\tmanager := NewReleaseManager(config, mockGit, mockFS, mockInteraction)\n\n\t\t\t// Setup basic mocks\n\t\t\tmockGit.EXPECT().GetTags(gomock.Any()).Return([]string{\"v1.2.3\"}, nil)\n\t\t\tmockGit.EXPECT().GetCurrentBranch(gomock.Any()).Return(tt.currentBranch, nil)\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return([]string{\".\"}, nil)\n\n\t\t\t// Setup user interaction mocks\n\t\t\tresponseIndex := 0\n\t\t\tif !tt.skipConfirmations {\n\t\t\t\tif tt.currentBranch != \"master\" {\n\t\t\t\t\t// Branch warning confirmation\n\t\t\t\t\tif responseIndex < len(tt.userResponses) {\n\t\t\t\t\t\tmockInteraction.EXPECT().\n\t\t\t\t\t\t\tConfirmWithDefault(gomock.Any(), \"Continue?\", false).\n\t\t\t\t\t\t\tReturn(tt.userResponses[responseIndex], nil)\n\t\t\t\t\t\tresponseIndex++\n\n\t\t\t\t\t\tif !tt.userResponses[responseIndex-1] {\n\t\t\t\t\t\t\t// User rejected, no more interactions\n\t\t\t\t\t\t\tgoto executeTest\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Tag creation confirmation\n\t\t\t\tif responseIndex < len(tt.userResponses) {\n\t\t\t\t\tmockInteraction.EXPECT().\n\t\t\t\t\t\tConfirm(gomock.Any(), gomock.Any()).\n\t\t\t\t\t\tReturn(tt.userResponses[responseIndex], nil)\n\t\t\t\t\tresponseIndex++\n\n\t\t\t\t\tif !tt.userResponses[responseIndex-1] {\n\t\t\t\t\t\t// User rejected tag creation\n\t\t\t\t\t\tgoto executeTest\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Tag push confirmation\n\t\t\t\tif responseIndex < len(tt.userResponses) {\n\t\t\t\t\tmockInteraction.EXPECT().\n\t\t\t\t\t\tConfirm(gomock.Any(), gomock.Any()).\n\t\t\t\t\t\tReturn(tt.userResponses[responseIndex], nil)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Setup git operation mocks based on expected actions\n\t\t\tif tt.expectedActions >= 1 {\n\t\t\t\tmockGit.EXPECT().CreateTag(gomock.Any(), \"v1.3.0-prerelease01\").Return(nil)\n\t\t\t}\n\t\t\tif tt.expectedActions >= 2 {\n\t\t\t\tmockGit.EXPECT().PushTag(gomock.Any(), \"v1.3.0-prerelease01\").Return(nil)\n\t\t\t}\n\n\t\texecuteTest:\n\t\t\terr := manager.RunMinor(context.Background())\n\n\t\t\tif tt.shouldComplete {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t} else {\n\t\t\t\t// May have error or be cancelled\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Logf(\"Expected cancellation/error: %v\", err)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/types.go",
    "content": "package release\n\nimport \"github.com/Masterminds/semver/v3\"\n\n// State holds the current state of the repository\ntype State struct {\n\tCurrentBranch  string\n\tModules        []Module\n\tCurrentVersion string\n\tTagCache       *TagCache\n}\n\n// Module represents a Go module\ntype Module struct {\n\tPath    string\n\tVersion string\n}\n\n// ParsedTag represents a single parsed git tag\ntype ParsedTag struct {\n\tRaw           string          // Original tag name (e.g., \"service1/v1.2.3-prerelease01\")\n\tModulePath    string          // Module path (e.g., \"service1\", \"\" for root)\n\tVersion       *semver.Version // Parsed semantic version\n\tGitSHA        string          // Git commit SHA\n\tIsPrerelease  bool\n\tPrereleaseNum int // Extracted prerelease number (01, 02, etc.)\n}\n\n// TagCache holds all parsed tag information\ntype TagCache struct {\n\tAllTags         []ParsedTag\n\tVersionTags     []ParsedTag            // Only valid semver tags\n\tModuleTags      map[string][]ParsedTag // Tags grouped by module path\n\tPrereleaseCache map[string][]int       // Base version -> prerelease numbers\n\tHighestVersion  *semver.Version        // Cached highest version\n}\n\n// VersionConflictInfo holds information about version conflicts\ntype VersionConflictInfo struct {\n\tExistingTags []string // Tags that already exist\n\tMissingTags  []string // Tags that need to be created\n}\n\n// WarningType represents different types of warnings\ntype WarningType int\n\nconst (\n\tWrongBranch WarningType = iota\n\tExistingTags\n)\n\n// Warning represents a validation warning that can be overridden\ntype Warning struct {\n\tType    WarningType\n\tMessage string\n}\n\n// Action represents a planned release action\ntype Action struct {\n\tType        ActionType\n\tTarget      string // tag name, module path\n\tDescription string\n\tGitSHA      string // for existing tags\n}\n\ntype ActionType int\n\nconst (\n\tActionCreateTag ActionType = iota\n\tActionPushTags\n)\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/validate_test.go",
    "content": "package release\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n)\n\nfunc TestValidateReleaseCommand(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentVersion string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t{\n\t\t\tname:           \"valid prerelease version\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease01\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"valid higher prerelease\",\n\t\t\tcurrentVersion: \"v2.5.0-prerelease15\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"invalid stable version\",\n\t\t\tcurrentVersion: \"v1.2.3\",\n\t\t\tshouldError:    true,\n\t\t\terrorMsg:       \"release command requires existing prerelease version\",\n\t\t},\n\t\t{\n\t\t\tname:           \"invalid initial version\",\n\t\t\tcurrentVersion: \"v0.0.0\",\n\t\t\tshouldError:    true,\n\t\t\terrorMsg:       \"release command requires existing prerelease version\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\terr := manager.validateReleaseCommand(tt.currentVersion)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateMinorMajorCommand(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentVersion string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t{\n\t\t\tname:           \"valid stable version\",\n\t\t\tcurrentVersion: \"v1.2.3\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"valid initial version\",\n\t\t\tcurrentVersion: \"v0.0.0\",\n\t\t\tshouldError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"invalid prerelease version\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease01\",\n\t\t\tshouldError:    true,\n\t\t\terrorMsg:       \"minor/major commands should be run from stable versions\",\n\t\t},\n\t\t{\n\t\t\tname:           \"invalid higher prerelease\",\n\t\t\tcurrentVersion: \"v2.5.0-prerelease15\",\n\t\t\tshouldError:    true,\n\t\t\terrorMsg:       \"minor/major commands should be run from stable versions\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\terr := manager.validateMinorMajorCommand(tt.currentVersion)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestProcessManualVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tmanualVersion string\n\t\texpected      string\n\t\tshouldError   bool\n\t\terrorMsg      string\n\t}{\n\t\t{\n\t\t\tname:          \"valid version with v\",\n\t\t\tmanualVersion: \"v1.2.3\",\n\t\t\texpected:      \"v1.2.3\",\n\t\t\tshouldError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"valid version without v\",\n\t\t\tmanualVersion: \"1.2.3\",\n\t\t\texpected:      \"v1.2.3\",\n\t\t\tshouldError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"valid prerelease with v\",\n\t\t\tmanualVersion: \"v2.0.0-prerelease01\",\n\t\t\texpected:      \"v2.0.0-prerelease01\",\n\t\t\tshouldError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"valid prerelease without v\",\n\t\t\tmanualVersion: \"2.0.0-prerelease01\",\n\t\t\texpected:      \"v2.0.0-prerelease01\",\n\t\t\tshouldError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"auto-complete version format\",\n\t\t\tmanualVersion: \"1.2\",\n\t\t\texpected:      \"v1.2.0\",\n\t\t\tshouldError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"completely invalid\",\n\t\t\tmanualVersion: \"not-a-version\",\n\t\t\tshouldError:   true,\n\t\t\terrorMsg:      \"invalid manual version format\",\n\t\t},\n\t\t{\n\t\t\tname:          \"empty version\",\n\t\t\tmanualVersion: \"\",\n\t\t\tshouldError:   true,\n\t\t\terrorMsg:      \"invalid manual version format\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\tresult, err := manager.processManualVersion(tt.manualVersion)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShouldExcludeModule(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texcludedDirs  []string\n\t\tmodulePath    string\n\t\tshouldExclude bool\n\t}{\n\t\t{\n\t\t\tname:          \"not excluded\",\n\t\t\texcludedDirs:  []string{\"cmd\", \"internal/tools\"},\n\t\t\tmodulePath:    \"service1\",\n\t\t\tshouldExclude: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"exact match\",\n\t\t\texcludedDirs:  []string{\"cmd\", \"internal/tools\"},\n\t\t\tmodulePath:    \"cmd\",\n\t\t\tshouldExclude: true,\n\t\t},\n\t\t{\n\t\t\tname:          \"prefix match\",\n\t\t\texcludedDirs:  []string{\"cmd\", \"internal/tools\"},\n\t\t\tmodulePath:    \"cmd/tool1\",\n\t\t\tshouldExclude: true,\n\t\t},\n\t\t{\n\t\t\tname:          \"nested prefix match\",\n\t\t\texcludedDirs:  []string{\"cmd\", \"internal/tools\"},\n\t\t\tmodulePath:    \"internal/tools/helper\",\n\t\t\tshouldExclude: true,\n\t\t},\n\t\t{\n\t\t\tname:          \"partial match should not exclude\",\n\t\t\texcludedDirs:  []string{\"cmd\", \"internal/tools\"},\n\t\t\tmodulePath:    \"cmdline\",\n\t\t\tshouldExclude: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"empty excluded dirs\",\n\t\t\texcludedDirs:  []string{},\n\t\t\tmodulePath:    \"any/path\",\n\t\t\tshouldExclude: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"root module\",\n\t\t\texcludedDirs:  []string{\"cmd\"},\n\t\t\tmodulePath:    \"\",\n\t\t\tshouldExclude: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{\n\t\t\t\tconfig: Config{\n\t\t\t\t\tExcludedDirs: tt.excludedDirs,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tresult := manager.shouldExcludeModule(tt.modulePath)\n\t\t\tassert.Equal(t, tt.shouldExclude, result)\n\t\t})\n\t}\n}\n\nfunc TestGetTagName(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tmodule   Module\n\t\tversion  string\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"root module\",\n\t\t\tmodule:   Module{Path: \"\", Version: \"v1.2.3\"},\n\t\t\tversion:  \"v1.3.0\",\n\t\t\texpected: \"v1.3.0\",\n\t\t},\n\t\t{\n\t\t\tname:     \"submodule\",\n\t\t\tmodule:   Module{Path: \"service1\", Version: \"v1.2.3\"},\n\t\t\tversion:  \"v1.3.0\",\n\t\t\texpected: \"service1/v1.3.0\",\n\t\t},\n\t\t{\n\t\t\tname:     \"nested module\",\n\t\t\tmodule:   Module{Path: \"services/auth\", Version: \"v1.2.3\"},\n\t\t\tversion:  \"v2.0.0\",\n\t\t\texpected: \"services/auth/v2.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:     \"prerelease version\",\n\t\t\tmodule:   Module{Path: \"service1\", Version: \"v1.2.3\"},\n\t\t\tversion:  \"v1.3.0-prerelease01\",\n\t\t\texpected: \"service1/v1.3.0-prerelease01\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\tresult := manager.getTagName(tt.module, tt.version)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestGetLatestVersionForModule(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tmodulePath string\n\t\ttags       []string\n\t\texpected   string\n\t}{\n\t\t{\n\t\t\tname:       \"no tags for module\",\n\t\t\tmodulePath: \"service1\",\n\t\t\ttags:       []string{},\n\t\t\texpected:   \"v0.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:       \"single tag\",\n\t\t\tmodulePath: \"service1\",\n\t\t\ttags:       []string{\"service1/v1.2.3\"},\n\t\t\texpected:   \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:       \"multiple tags - choose latest\",\n\t\t\tmodulePath: \"service1\",\n\t\t\ttags:       []string{\"service1/v1.2.3\", \"service1/v1.1.0\", \"service1/v2.0.0\"},\n\t\t\texpected:   \"v2.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:       \"mix of stable and prerelease\",\n\t\t\tmodulePath: \"service1\",\n\t\t\ttags:       []string{\"service1/v1.2.3\", \"service1/v1.3.0-prerelease01\"},\n\t\t\texpected:   \"v1.3.0-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:       \"root module\",\n\t\t\tmodulePath: \"\",\n\t\t\ttags:       []string{\"v1.2.3\", \"v1.1.0\", \"v2.0.0\"},\n\t\t\texpected:   \"v2.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:       \"no cache initialized\",\n\t\t\tmodulePath: \"service1\",\n\t\t\ttags:       nil, // Will simulate uninitialized cache\n\t\t\texpected:   \"v0.0.0\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\n\t\t\tif tt.tags != nil {\n\t\t\t\t// Initialize cache\n\t\t\t\tmanager.tagCache = &TagCache{\n\t\t\t\t\tModuleTags: make(map[string][]ParsedTag),\n\t\t\t\t}\n\n\t\t\t\t// Parse and add tags\n\t\t\t\tfor _, tag := range tt.tags {\n\t\t\t\t\tparsed := manager.parseTag(tag)\n\t\t\t\t\tif parsed.Version != nil {\n\t\t\t\t\t\tif manager.tagCache.ModuleTags[parsed.ModulePath] == nil {\n\t\t\t\t\t\t\tmanager.tagCache.ModuleTags[parsed.ModulePath] = make([]ParsedTag, 0)\n\t\t\t\t\t\t}\n\t\t\t\t\t\tmanager.tagCache.ModuleTags[parsed.ModulePath] = append(\n\t\t\t\t\t\t\tmanager.tagCache.ModuleTags[parsed.ModulePath], parsed)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tresult := manager.getLatestVersionForModule(tt.modulePath)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestFilterActionsForMissingTags(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tactions     []Action\n\t\tmissingTags []string\n\t\texpected    []Action\n\t}{\n\t\t{\n\t\t\tname: \"filter some actions\",\n\t\t\tactions: []Action{\n\t\t\t\t{Type: ActionCreateTag, Target: \"v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"v1.3.0\"},\n\t\t\t\t{Type: ActionCreateTag, Target: \"service1/v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"service1/v1.3.0\"},\n\t\t\t\t{Type: ActionCreateTag, Target: \"service2/v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"service2/v1.3.0\"},\n\t\t\t},\n\t\t\tmissingTags: []string{\"v1.3.0\", \"service2/v1.3.0\"},\n\t\t\texpected: []Action{\n\t\t\t\t{Type: ActionCreateTag, Target: \"v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"v1.3.0\"},\n\t\t\t\t{Type: ActionCreateTag, Target: \"service2/v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"service2/v1.3.0\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"no missing tags\",\n\t\t\tactions: []Action{\n\t\t\t\t{Type: ActionCreateTag, Target: \"v1.3.0\"},\n\t\t\t\t{Type: ActionPushTags, Target: \"v1.3.0\"},\n\t\t\t},\n\t\t\tmissingTags: []string{},\n\t\t\texpected:    []Action(nil),\n\t\t},\n\t\t{\n\t\t\tname:        \"no actions\",\n\t\t\tactions:     []Action{},\n\t\t\tmissingTags: []string{\"v1.3.0\"},\n\t\t\texpected:    []Action(nil),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\tresult := manager.filterActionsForMissingTags(tt.actions, tt.missingTags)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestValidateWithWarnings(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\tcurrentBranch    string\n\t\trequiredBranch   string\n\t\texpectedWarnings int\n\t\texpectedMessage  string\n\t}{\n\t\t{\n\t\t\tname:             \"correct branch\",\n\t\t\tcurrentBranch:    \"master\",\n\t\t\trequiredBranch:   \"master\",\n\t\t\texpectedWarnings: 0,\n\t\t},\n\t\t{\n\t\t\tname:             \"wrong branch\",\n\t\t\tcurrentBranch:    \"feature-branch\",\n\t\t\trequiredBranch:   \"master\",\n\t\t\texpectedWarnings: 1,\n\t\t\texpectedMessage:  \"you are not on master\",\n\t\t},\n\t\t{\n\t\t\tname:             \"no required branch\",\n\t\t\tcurrentBranch:    \"any-branch\",\n\t\t\trequiredBranch:   \"\",\n\t\t\texpectedWarnings: 0,\n\t\t},\n\t\t{\n\t\t\tname:             \"different wrong branch\",\n\t\t\tcurrentBranch:    \"develop\",\n\t\t\trequiredBranch:   \"main\",\n\t\t\texpectedWarnings: 1,\n\t\t\texpectedMessage:  \"you are not on main\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{\n\t\t\t\tconfig: Config{\n\t\t\t\t\tRequiredBranch: tt.requiredBranch,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tstate := &State{\n\t\t\t\tCurrentBranch: tt.currentBranch,\n\t\t\t}\n\n\t\t\twarnings := manager.validateWithWarnings(state, \"v1.3.0\")\n\n\t\t\tassert.Len(t, warnings, tt.expectedWarnings)\n\t\t\tif tt.expectedWarnings > 0 {\n\t\t\t\tassert.Contains(t, warnings[0].Message, tt.expectedMessage)\n\t\t\t\tassert.Equal(t, WrongBranch, warnings[0].Type)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFindModules(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tgoModPaths   []string\n\t\texcludedDirs []string\n\t\texpected     []Module\n\t}{\n\t\t{\n\t\t\tname:         \"single root module\",\n\t\t\tgoModPaths:   []string{\".\"},\n\t\t\texcludedDirs: []string{},\n\t\t\texpected: []Module{\n\t\t\t\t{Path: \"\", Version: \"v0.0.0\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"multiple modules\",\n\t\t\tgoModPaths:   []string{\".\", \"service1\", \"service2\"},\n\t\t\texcludedDirs: []string{},\n\t\t\texpected: []Module{\n\t\t\t\t{Path: \"\", Version: \"v0.0.0\"},\n\t\t\t\t{Path: \"service1\", Version: \"v0.0.0\"},\n\t\t\t\t{Path: \"service2\", Version: \"v0.0.0\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"modules with exclusions\",\n\t\t\tgoModPaths:   []string{\".\", \"service1\", \"cmd/tool\", \"internal/tools/helper\"},\n\t\t\texcludedDirs: []string{\"cmd\", \"internal/tools\"},\n\t\t\texpected: []Module{\n\t\t\t\t{Path: \"\", Version: \"v0.0.0\"},\n\t\t\t\t{Path: \"service1\", Version: \"v0.0.0\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"deduplicate paths\",\n\t\t\tgoModPaths:   []string{\".\", \"./\", \"service1\", \"./service1\"},\n\t\t\texcludedDirs: []string{},\n\t\t\texpected: []Module{\n\t\t\t\t{Path: \"\", Version: \"v0.0.0\"},\n\t\t\t\t{Path: \"service1\", Version: \"v0.0.0\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockFS := NewMockFS(ctrl)\n\t\t\tmanager := &Manager{\n\t\t\t\tconfig: Config{\n\t\t\t\t\tRepoRoot:     \"/test\",\n\t\t\t\t\tExcludedDirs: tt.excludedDirs,\n\t\t\t\t},\n\t\t\t\tfs: mockFS,\n\t\t\t\ttagCache: &TagCache{\n\t\t\t\t\tModuleTags: make(map[string][]ParsedTag),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tmockFS.EXPECT().FindGoModFiles(gomock.Any(), \"/test\").Return(tt.goModPaths, nil)\n\n\t\t\tresult, err := manager.FindModules(context.Background())\n\t\t\trequire.NoError(t, err)\n\n\t\t\tassert.ElementsMatch(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestHandleWarningsAndConfirmations(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\twarnings          []Warning\n\t\tskipConfirmations bool\n\t\tuserResponses     []bool\n\t\tshouldError       bool\n\t\terrorMsg          string\n\t}{\n\t\t{\n\t\t\tname:              \"no warnings\",\n\t\t\twarnings:          []Warning{},\n\t\t\tskipConfirmations: false,\n\t\t\tshouldError:       false,\n\t\t},\n\t\t{\n\t\t\tname: \"skip confirmations with warnings\",\n\t\t\twarnings: []Warning{\n\t\t\t\t{Type: WrongBranch, Message: \"you are not on master\"},\n\t\t\t},\n\t\t\tskipConfirmations: true,\n\t\t\tshouldError:       false,\n\t\t},\n\t\t{\n\t\t\tname: \"user confirms warning\",\n\t\t\twarnings: []Warning{\n\t\t\t\t{Type: WrongBranch, Message: \"you are not on master\"},\n\t\t\t},\n\t\t\tskipConfirmations: false,\n\t\t\tuserResponses:     []bool{true},\n\t\t\tshouldError:       false,\n\t\t},\n\t\t{\n\t\t\tname: \"user rejects warning\",\n\t\t\twarnings: []Warning{\n\t\t\t\t{Type: WrongBranch, Message: \"you are not on master\"},\n\t\t\t},\n\t\t\tskipConfirmations: false,\n\t\t\tuserResponses:     []bool{false},\n\t\t\tshouldError:       true,\n\t\t\terrorMsg:          \"operation cancelled due to: you are not on master\",\n\t\t},\n\t\t{\n\t\t\tname: \"multiple warnings - all confirmed\",\n\t\t\twarnings: []Warning{\n\t\t\t\t{Type: WrongBranch, Message: \"you are not on master\"},\n\t\t\t\t{Type: ExistingTags, Message: \"some tags exist\"},\n\t\t\t},\n\t\t\tskipConfirmations: false,\n\t\t\tuserResponses:     []bool{true, true},\n\t\t\tshouldError:       false,\n\t\t},\n\t\t{\n\t\t\tname: \"multiple warnings - second rejected\",\n\t\t\twarnings: []Warning{\n\t\t\t\t{Type: WrongBranch, Message: \"you are not on master\"},\n\t\t\t\t{Type: ExistingTags, Message: \"some tags exist\"},\n\t\t\t},\n\t\t\tskipConfirmations: false,\n\t\t\tuserResponses:     []bool{true, false},\n\t\t\tshouldError:       true,\n\t\t\terrorMsg:          \"operation cancelled due to: some tags exist\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockInteraction := NewMockUserInteraction(ctrl)\n\t\t\tmanager := &Manager{\n\t\t\t\tconfig: Config{\n\t\t\t\t\tSkipConfirmations: tt.skipConfirmations,\n\t\t\t\t},\n\t\t\t\tinteraction: mockInteraction,\n\t\t\t}\n\n\t\t\t// Setup mock expectations\n\t\t\tif !tt.skipConfirmations && len(tt.warnings) > 0 {\n\t\t\t\tfor i, response := range tt.userResponses {\n\t\t\t\t\tif i < len(tt.warnings) {\n\t\t\t\t\t\tmockInteraction.EXPECT().\n\t\t\t\t\t\t\tConfirmWithDefault(gomock.Any(), \"Continue?\", false).\n\t\t\t\t\t\t\tReturn(response, nil)\n\n\t\t\t\t\t\tif !response {\n\t\t\t\t\t\t\tbreak // Stop after first rejection\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\terr := manager.handleWarningsAndConfirmations(context.Background(), tt.warnings)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tif tt.errorMsg != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "cmd/tools/releaser/internal/release/version_test.go",
    "content": "package release\n\nimport (\n\t\"testing\"\n\n\t\"github.com/Masterminds/semver/v3\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestParseTag(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\trawTag         string\n\t\texpectedModule string\n\t\texpectedVer    string\n\t\texpectedPre    bool\n\t\texpectedPreNum int\n\t\tshouldHaveVer  bool\n\t}{\n\t\t{\n\t\t\tname:           \"root module stable version\",\n\t\t\trawTag:         \"v1.2.3\",\n\t\t\texpectedModule: \"\",\n\t\t\texpectedVer:    \"1.2.3\",\n\t\t\texpectedPre:    false,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:           \"root module prerelease\",\n\t\t\trawTag:         \"v1.2.3-prerelease01\",\n\t\t\texpectedModule: \"\",\n\t\t\texpectedVer:    \"1.2.3-prerelease01\",\n\t\t\texpectedPre:    true,\n\t\t\texpectedPreNum: 1,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:           \"submodule stable version\",\n\t\t\trawTag:         \"service1/v1.2.3\",\n\t\t\texpectedModule: \"service1\",\n\t\t\texpectedVer:    \"1.2.3\",\n\t\t\texpectedPre:    false,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:           \"submodule prerelease\",\n\t\t\trawTag:         \"service1/v1.2.3-prerelease05\",\n\t\t\texpectedModule: \"service1\",\n\t\t\texpectedVer:    \"1.2.3-prerelease05\",\n\t\t\texpectedPre:    true,\n\t\t\texpectedPreNum: 5,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:           \"nested module\",\n\t\t\trawTag:         \"services/auth/v2.1.0\",\n\t\t\texpectedModule: \"services/auth\",\n\t\t\texpectedVer:    \"2.1.0\",\n\t\t\texpectedPre:    false,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:           \"high prerelease number\",\n\t\t\trawTag:         \"v1.2.3-prerelease99\",\n\t\t\texpectedModule: \"\",\n\t\t\texpectedVer:    \"1.2.3-prerelease99\",\n\t\t\texpectedPre:    true,\n\t\t\texpectedPreNum: 99,\n\t\t\tshouldHaveVer:  true,\n\t\t},\n\t\t{\n\t\t\tname:          \"invalid version tag\",\n\t\t\trawTag:        \"invalid-tag\",\n\t\t\tshouldHaveVer: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not a version tag\",\n\t\t\trawTag:        \"refs/heads/master\",\n\t\t\tshouldHaveVer: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\tparsed := manager.parseTag(tt.rawTag)\n\n\t\t\tassert.Equal(t, tt.rawTag, parsed.Raw)\n\t\t\tassert.Equal(t, tt.expectedModule, parsed.ModulePath)\n\t\t\tassert.Equal(t, tt.expectedPre, parsed.IsPrerelease)\n\n\t\t\tif tt.shouldHaveVer {\n\t\t\t\trequire.NotNil(t, parsed.Version)\n\t\t\t\tassert.Equal(t, tt.expectedVer, parsed.Version.String())\n\t\t\t\tif tt.expectedPre {\n\t\t\t\t\tassert.Equal(t, tt.expectedPreNum, parsed.PrereleaseNum)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, parsed.Version)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIncrementVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tcurrent     string\n\t\tversionType string\n\t\texpected    string\n\t\tshouldError bool\n\t}{\n\t\t{\n\t\t\tname:        \"increment major\",\n\t\t\tcurrent:     \"v1.2.3\",\n\t\t\tversionType: \"major\",\n\t\t\texpected:    \"v2.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"increment minor\",\n\t\t\tcurrent:     \"v1.2.3\",\n\t\t\tversionType: \"minor\",\n\t\t\texpected:    \"v1.3.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"increment patch\",\n\t\t\tcurrent:     \"v1.2.3\",\n\t\t\tversionType: \"patch\",\n\t\t\texpected:    \"v1.2.4\",\n\t\t},\n\t\t{\n\t\t\tname:        \"increment from zero\",\n\t\t\tcurrent:     \"v0.0.0\",\n\t\t\tversionType: \"minor\",\n\t\t\texpected:    \"v0.1.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"increment major from high version\",\n\t\t\tcurrent:     \"v15.27.99\",\n\t\t\tversionType: \"major\",\n\t\t\texpected:    \"v16.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid version type\",\n\t\t\tcurrent:     \"v1.2.3\",\n\t\t\tversionType: \"invalid\",\n\t\t\tshouldError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid current version\",\n\t\t\tcurrent:     \"invalid\",\n\t\t\tversionType: \"major\",\n\t\t\tshouldError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := IncrementVersion(tt.current, tt.versionType)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNormalizeVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tinput       string\n\t\texpected    string\n\t\tshouldError bool\n\t}{\n\t\t{\n\t\t\tname:     \"already normalized\",\n\t\t\tinput:    \"v1.2.3\",\n\t\t\texpected: \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:     \"add v prefix\",\n\t\t\tinput:    \"1.2.3\",\n\t\t\texpected: \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:     \"prerelease with v\",\n\t\t\tinput:    \"v1.2.3-prerelease01\",\n\t\t\texpected: \"v1.2.3-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:     \"prerelease without v\",\n\t\t\tinput:    \"1.2.3-prerelease01\",\n\t\t\texpected: \"v1.2.3-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:     \"invalid semver\",\n\t\t\tinput:    \"1.2\",\n\t\t\texpected: \"v1.2.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"completely invalid\",\n\t\t\tinput:       \"not-a-version\",\n\t\t\tshouldError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := NormalizeVersion(tt.input)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetNextPrereleaseVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tbaseVersion  string\n\t\texistingPres []int\n\t\texpected     string\n\t\tshouldError  bool\n\t\terrorMsg     string\n\t}{\n\t\t{\n\t\t\tname:         \"first prerelease\",\n\t\t\tbaseVersion:  \"v1.2.3\",\n\t\t\texistingPres: []int{},\n\t\t\texpected:     \"v1.2.3-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:         \"increment from existing\",\n\t\t\tbaseVersion:  \"v1.2.3\",\n\t\t\texistingPres: []int{1, 2, 3},\n\t\t\texpected:     \"v1.2.3-prerelease04\",\n\t\t},\n\t\t{\n\t\t\tname:         \"increment from high number\",\n\t\t\tbaseVersion:  \"v1.2.3\",\n\t\t\texistingPres: []int{25},\n\t\t\texpected:     \"v1.2.3-prerelease26\",\n\t\t},\n\t\t{\n\t\t\tname:         \"unordered existing prereleases\",\n\t\t\tbaseVersion:  \"v1.2.3\",\n\t\t\texistingPres: []int{3, 1, 5, 2},\n\t\t\texpected:     \"v1.2.3-prerelease06\",\n\t\t},\n\t\t{\n\t\t\tname:         \"max prerelease exceeded\",\n\t\t\tbaseVersion:  \"v1.2.3\",\n\t\t\texistingPres: []int{99}, // Next would be 100\n\t\t\tshouldError:  true,\n\t\t\terrorMsg:     \"maximum prerelease number (99) exceeded\",\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid base version\",\n\t\t\tbaseVersion: \"invalid\",\n\t\t\tshouldError: true,\n\t\t\terrorMsg:    \"failed to parse base version\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{\n\t\t\t\ttagCache: &TagCache{\n\t\t\t\t\tPrereleaseCache: make(map[string][]int),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Setup prerelease cache\n\t\t\tif len(tt.existingPres) > 0 {\n\t\t\t\t// Parse base version to get the clean string\n\t\t\t\tif baseVer, err := semver.NewVersion(tt.baseVersion); err == nil {\n\t\t\t\t\tcleanBase := \"v\" + baseVer.String()\n\t\t\t\t\tmanager.tagCache.PrereleaseCache[cleanBase] = tt.existingPres\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tresult, err := manager.GetNextPrereleaseVersion(tt.baseVersion)\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVersionCalculationMethods(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmethod         string\n\t\tcurrentVersion string\n\t\texpected       string\n\t\tshouldError    bool\n\t\terrorMsg       string\n\t}{\n\t\t// Release version calculations\n\t\t{\n\t\t\tname:           \"release from prerelease\",\n\t\t\tmethod:         \"release\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease01\",\n\t\t\texpected:       \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:           \"release from high prerelease\",\n\t\t\tcurrentVersion: \"v2.5.10-prerelease15\",\n\t\t\tmethod:         \"release\",\n\t\t\texpected:       \"v2.5.10\",\n\t\t},\n\n\t\t// Minor version calculations\n\t\t{\n\t\t\tname:           \"minor from stable\",\n\t\t\tmethod:         \"minor\",\n\t\t\tcurrentVersion: \"v1.2.3\",\n\t\t\texpected:       \"v1.3.0-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:           \"minor from prerelease base\",\n\t\t\tmethod:         \"minor\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease05\",\n\t\t\texpected:       \"v1.3.0-prerelease01\",\n\t\t},\n\n\t\t// Major version calculations\n\t\t{\n\t\t\tname:           \"major from stable\",\n\t\t\tmethod:         \"major\",\n\t\t\tcurrentVersion: \"v1.2.3\",\n\t\t\texpected:       \"v2.0.0-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:           \"major from prerelease base\",\n\t\t\tmethod:         \"major\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease05\",\n\t\t\texpected:       \"v2.0.0-prerelease01\",\n\t\t},\n\n\t\t// Prerelease calculations\n\t\t{\n\t\t\tname:           \"prerelease from stable\",\n\t\t\tmethod:         \"prerelease\",\n\t\t\tcurrentVersion: \"v1.2.3\",\n\t\t\texpected:       \"v1.2.3-prerelease01\",\n\t\t},\n\t\t{\n\t\t\tname:           \"prerelease increment\",\n\t\t\tmethod:         \"prerelease\",\n\t\t\tcurrentVersion: \"v1.2.3-prerelease01\",\n\t\t\texpected:       \"v1.2.3-prerelease02\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{\n\t\t\t\ttagCache: &TagCache{\n\t\t\t\t\tPrereleaseCache: make(map[string][]int),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Setup prerelease cache for increment scenarios\n\t\t\tif tt.method == \"prerelease\" && tt.currentVersion == \"v1.2.3-prerelease01\" {\n\t\t\t\tmanager.tagCache.PrereleaseCache[\"v1.2.3\"] = []int{1}\n\t\t\t}\n\n\t\t\tvar result string\n\t\t\tvar err error\n\n\t\t\tswitch tt.method {\n\t\t\tcase \"release\":\n\t\t\t\tresult, err = manager.calculateReleaseVersion(tt.currentVersion)\n\t\t\tcase \"minor\":\n\t\t\t\tresult, err = manager.calculateMinorVersion(tt.currentVersion)\n\t\t\tcase \"major\":\n\t\t\t\tresult, err = manager.calculateMajorVersion(tt.currentVersion)\n\t\t\tcase \"prerelease\":\n\t\t\t\tresult, err = manager.calculatePrereleaseVersion(tt.currentVersion)\n\t\t\t}\n\n\t\t\tif tt.shouldError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errorMsg)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetBaseVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tinput    string\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"stable version\",\n\t\t\tinput:    \"v1.2.3\",\n\t\t\texpected: \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:     \"prerelease version\",\n\t\t\tinput:    \"v1.2.3-prerelease01\",\n\t\t\texpected: \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:     \"high prerelease version\",\n\t\t\tinput:    \"v2.5.10-prerelease25\",\n\t\t\texpected: \"v2.5.10\",\n\t\t},\n\t\t{\n\t\t\tname:     \"invalid version\",\n\t\t\tinput:    \"invalid\",\n\t\t\texpected: \"invalid\", // Should return input unchanged\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{}\n\t\t\tresult := manager.getBaseVersion(tt.input)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestGetCurrentGlobalVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tversionTags []string\n\t\texpected    string\n\t}{\n\t\t{\n\t\t\tname:        \"no versions\",\n\t\t\tversionTags: []string{},\n\t\t\texpected:    \"v0.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"single version\",\n\t\t\tversionTags: []string{\"v1.2.3\"},\n\t\t\texpected:    \"v1.2.3\",\n\t\t},\n\t\t{\n\t\t\tname:        \"multiple versions\",\n\t\t\tversionTags: []string{\"v1.2.3\", \"v1.1.0\", \"v2.0.0\", \"v1.3.0\"},\n\t\t\texpected:    \"v2.0.0\",\n\t\t},\n\t\t{\n\t\t\tname:        \"prerelease versions\",\n\t\t\tversionTags: []string{\"v1.2.3\", \"v1.3.0-prerelease01\", \"v1.2.4\"},\n\t\t\texpected:    \"v1.3.0-prerelease01\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmanager := &Manager{\n\t\t\t\ttagCache: &TagCache{\n\t\t\t\t\tVersionTags: make([]ParsedTag, 0),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Parse and add version tags\n\t\t\tfor _, tag := range tt.versionTags {\n\t\t\t\tparsed := manager.parseTag(tag)\n\t\t\t\tif parsed.Version != nil {\n\t\t\t\t\tmanager.tagCache.VersionTags = append(manager.tagCache.VersionTags, parsed)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Sort and set highest version\n\t\t\tif len(manager.tagCache.VersionTags) > 0 {\n\t\t\t\tmanager.sortVersionTags()\n\t\t\t\tmanager.tagCache.HighestVersion = manager.tagCache.VersionTags[len(manager.tagCache.VersionTags)-1].Version\n\t\t\t}\n\n\t\t\tresult := manager.GetCurrentGlobalVersion()\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestSortVersionTags(t *testing.T) {\n\tmanager := &Manager{\n\t\ttagCache: &TagCache{\n\t\t\tVersionTags: []ParsedTag{\n\t\t\t\t{Raw: \"v2.0.0\", Version: semver.MustParse(\"v2.0.0\")},\n\t\t\t\t{Raw: \"v1.2.3\", Version: semver.MustParse(\"v1.2.3\")},\n\t\t\t\t{Raw: \"v1.3.0-prerelease01\", Version: semver.MustParse(\"v1.3.0-prerelease01\")},\n\t\t\t\t{Raw: \"v1.1.0\", Version: semver.MustParse(\"v1.1.0\")},\n\t\t\t},\n\t\t},\n\t}\n\n\tmanager.sortVersionTags()\n\n\texpected := []string{\n\t\t\"v1.1.0\",\n\t\t\"v1.2.3\",\n\t\t\"v1.3.0-prerelease01\",\n\t\t\"v2.0.0\",\n\t}\n\n\tactual := make([]string, len(manager.tagCache.VersionTags))\n\tfor i, tag := range manager.tagCache.VersionTags {\n\t\tactual[i] = tag.Raw\n\t}\n\n\tassert.Equal(t, expected, actual)\n}\n"
  },
  {
    "path": "cmd/tools/releaser/releaser.go",
    "content": "package main\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"os\"\n\t\"os/signal\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/cmd/tools/releaser/internal/console\"\n\t\"github.com/uber/cadence/cmd/tools/releaser/internal/fs\"\n\t\"github.com/uber/cadence/cmd/tools/releaser/internal/git\"\n\t\"github.com/uber/cadence/cmd/tools/releaser/internal/release\"\n)\n\nfunc main() {\n\tcliApp := &cli.App{\n\t\tName:    \"releaser\",\n\t\tUsage:   \"Cadence workflow release management tool\",\n\t\tVersion: \"0.1.0\",\n\t\tFlags: []cli.Flag{\n\t\t\t&cli.BoolFlag{\n\t\t\t\tName:    \"verbose\",\n\t\t\t\tAliases: []string{\"i\"},\n\t\t\t\tUsage:   \"Enable verbose output\",\n\t\t\t},\n\t\t\t&cli.BoolFlag{\n\t\t\t\tName:  \"yes\",\n\t\t\t\tUsage: \"Skip confirmation prompts\",\n\t\t\t},\n\t\t\t&cli.StringFlag{\n\t\t\t\tName:    \"set-version\",\n\t\t\t\tAliases: []string{\"s\"},\n\t\t\t\tUsage:   \"Override automatic version calculation with specific version\",\n\t\t\t},\n\t\t},\n\t\tCommands: []*cli.Command{\n\t\t\t{\n\t\t\t\tName:   \"release\",\n\t\t\t\tUsage:  \"Promote latest prerelease to final release\",\n\t\t\t\tAction: releaseCommand,\n\t\t\t\tDescription: `Converts the current prerelease version to a final release.\nExample: v1.2.3-prerelease01 → v1.2.3\n\nThis command requires that the current version is a prerelease.`,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName:   \"minor\",\n\t\t\t\tUsage:  \"Start new minor version development cycle\",\n\t\t\t\tAction: minorCommand,\n\t\t\t\tDescription: `Increments the minor version and creates the first prerelease.\nExample: v1.2.3 → v1.3.0-prerelease01\n\nUse this when starting development of new features.`,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName:   \"major\",\n\t\t\t\tUsage:  \"Start new major version development cycle\",\n\t\t\t\tAction: majorCommand,\n\t\t\t\tDescription: `Increments the major version and creates the first prerelease.\nExample: v1.2.3 → v2.0.0-prerelease01\n\nUse this when introducing breaking changes.`,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName:   \"patch\",\n\t\t\t\tUsage:  \"Start new patch version development cycle\",\n\t\t\t\tAction: patchCommand,\n\t\t\t\tDescription: `Increments the patch version and creates the first prerelease.\nExample: v1.2.3 → v1.2.4-prerelease01\n\nUse this when starting hotfix or patch development.`,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName:   \"prerelease\",\n\t\t\t\tUsage:  \"Increment prerelease number\",\n\t\t\t\tAction: prereleaseCommand,\n\t\t\t\tDescription: `Increments the prerelease number for iterative development.\nExample: v1.2.3-prerelease01 → v1.2.3-prerelease02\n\nUse this during active development within a version cycle.`,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName:   \"status\",\n\t\t\t\tUsage:  \"Show current repository release status\",\n\t\t\t\tAction: statusCommand,\n\t\t\t\tDescription: `Displays current branch, version, and module information.\nUse this to understand the current state before making releases.`,\n\t\t\t},\n\t\t},\n\t\tCustomAppHelpTemplate: `NAME:\n   {{.Name}} - {{.Usage}}\n\nUSAGE:\n   {{.HelpName}} [global options] command [command options]\n\nVERSION:\n   {{.Version}}\n\nGLOBAL OPTIONS:\n   {{range .VisibleFlags}}{{.}}\n   {{end}}\n\nCOMMANDS:\n{{range .Commands}}{{if not .HideHelp}}   {{join .Names \", \"}}{{ \"\\t\\t\"}}{{.Usage}}{{ \"\\n\" }}{{end}}{{end}}\n\nEXAMPLES:\n   # Check current status\n   {{.HelpName}} status\n\n   # Development workflow\n   {{.HelpName}} minor           # Start new minor version: v1.2.3 → v1.3.0-prerelease01\n   {{.HelpName}} major           # Start new major version: v1.2.3 → v2.0.0-prerelease01\n   {{.HelpName}} patch           # Start new patch version: v1.2.3 → v1.2.4-prerelease01\n   {{.HelpName}} prerelease      # Iterate: v1.3.0-prerelease01 → v1.3.0-prerelease02\n   {{.HelpName}} release         # Finalize: v1.3.0-prerelease03 → v1.3.0\n\n   # Major version workflow  \n   {{.HelpName}} major           # Start major version: v1.3.0 → v2.0.0-prerelease01\n   {{.HelpName}} prerelease      # Iterate: v2.0.0-prerelease01 → v2.0.0-prerelease02\n   {{.HelpName}} release         # Finalize: v2.0.0-prerelease02 → v2.0.0\n\n   # Manual version override\n   {{.HelpName}} release --set-version v1.4.0    # Override automatic calculation\n\nSAFETY FEATURES:\n   - Validates current version state before operations\n   - Requires clean git working directory\n   - Enforces releases only from master branch\n   - Prevents creating duplicate versions\n   - Builds and tests before creating tags\n   - Interactive confirmations for all operations\n\nUse --yes to skip confirmation prompts for automation.\n`,\n\t}\n\n\tctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)\n\tdefer stop()\n\n\tif err := cliApp.RunContext(ctx, os.Args); err != nil {\n\t\t_, _ = fmt.Fprintf(os.Stderr, \"Error: %v\\n\", err)\n\t\tos.Exit(1)\n\t}\n}\n\nfunc createManager(c *cli.Context, command string) (*release.Manager, error) {\n\tcfg := release.Config{\n\t\tExcludedDirs:      []string{\"cmd\", \"internal/tools\", \"idls\"},\n\t\tRequiredBranch:    \"master\",\n\t\tVerbose:           c.Bool(\"verbose\"),\n\t\tCommand:           command,\n\t\tSkipConfirmations: c.Bool(\"yes\"),\n\t\tManualVersion:     c.String(\"set-version\"),\n\t}\n\n\tgitClient := git.NewGitClient(cfg.Verbose)\n\trepo := fs.NewFileSystemClient(cfg.Verbose)\n\tinteraction := console.NewManager(os.Stdout, os.Stdin)\n\n\tmanager := release.NewReleaseManager(cfg, gitClient, repo, interaction)\n\n\t// Get repo root and update cfg\n\trepoRoot, err := gitClient.GetRepoRoot(c.Context)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to get repository root: %w\", err)\n\t}\n\tcfg.RepoRoot = repoRoot\n\n\treturn manager, nil\n}\n\nfunc releaseCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"release\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.RunRelease(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n\nfunc minorCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"minor\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.RunMinor(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n\nfunc majorCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"major\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.RunMajor(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n\nfunc patchCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"patch\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.RunPatch(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n\nfunc prereleaseCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"prerelease\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.RunPrerelease(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n\nfunc statusCommand(c *cli.Context) error {\n\tmanager, err := createManager(c, \"status\")\n\tif err != nil {\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\tif err := manager.ShowCurrentState(c.Context); err != nil {\n\t\tif c.Context.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn cli.Exit(err.Error(), 1)\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "cmd/tools/sql/main.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"os\"\n\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n\t\"github.com/uber/cadence/tools/sql\"\n\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"    // needed to load mysql plugin\n\t_ \"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\" // needed to load postgres plugin\n)\n\nfunc main() {\n\tcommoncli.ExitHandler(sql.RunTool(os.Args))\n}\n"
  },
  {
    "path": "codecov.yml",
    "content": "# Refs:\n# - https://docs.codecov.com/docs/common-recipe-list\n# - https://docs.codecov.com/docs/codecovyml-reference\n#\n# After making changes, run below command to validate\n# curl --data-binary @codecov.yml https://codecov.io/validate\ncoverage:\n  range: 80...100\n  round: down\n  precision: 2\n  status:\n    project:                   # measuring the overall project coverage\n      default:                 # context, you can create multiple ones with custom titles\n        informational: true\n        target: 85%            # specify the target coverage for each commit status\n                               #   option: \"auto\" (compare against parent commit or pull request base)\n                               #   option: \"X%\" a static target percentage to hit\n        threshold: 0%          # allow the coverage drop by x% before marking as failure\n        if_ci_failed: ignore   # require the CI to pass before setting the status\n    patch:\n      default:\n        target: 75%            # specify the target coverage for each commit status\n                               #   option: \"auto\" (compare against parent commit or pull request base)\n                               #   option: \"X%\" a static target percentage to hit\n        threshold: 0%          # allow the coverage drop by x% before marking as failure\ncomment:\n  layout: \"header, files, footer\"\n  hide_project_coverage: false\ncodecov:\n  require_ci_to_pass: false\nignore:\n  - \"**/*_cql.go\"\n  - \"**/*_generated.go\"\n  - \"**/*_mock.go\"\n  - \"**/*_test.go\"\n  - \"**/*Test.go\"\n  - \"**/*_test_utils.go\"\n  - \"**/constants.go\"\n  - \"**/interface.go\"\n  - \"**/interfaces.go\"\n  - \"**/main.go\"\n  - \"**/*mocks.go\"\n  - \"**/mocks/**\"\n  - \"**/testdata/**\"\n  - \"**/testing/**\"\n  - \"**/types.go\"\n  - \"**/version.go\"\n  - \"bench/**\"\n  - \"canary/**\"\n  - \"cmd/**\"\n  - \"common/persistence/persistence-tests/**\"\n  - \"common/domain/errors.go\"\n  - \"common/log/**\"\n  - \"common/metrics/**\"\n  - \"common/persistence/nosql/nosqlplugin/cassandra/admin.go\"\n  - \"common/persistence/nosql/nosqlplugin/dynamodb/**\"\n  - \"common/persistence/nosql/nosqlplugin/mongodb/**\"\n  - \"common/types/shared.go\" # 8k lines of getters. Not worth testing manually but consider switching to generated code.\n  - \"host/**\"\n  - \"idls/**\"\n  - \"service/frontend/service.go\"\n  - \"service/history/constants/test_constants.go\"\n  - \"service/history/execution/mutable_state.go\"\n  - \"service/history/shard/contextTest.go\"\n  - \"service/history/workflow/errors.go\"\n  - \"service/history/service.go\"\n  - \"service/matching/service.go\"\n  - \"service/matching/tasklist/testing.go\"\n  - \"service/worker/service.go\"\n  - \"simulation/**\"\n  - \"testflags/**\"\n  - \"tools/common/schema/test/**\"\n  - \"tools/linter/**\"\n  - \"tools/matchingsimulationcomparison/**\"\n  - \"tools/cli/factory.go\"\n"
  },
  {
    "path": "common/activecluster/execution_manager_provider_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/activecluster (interfaces: ExecutionManagerProvider)\n//\n// Generated by this command:\n//\n//\tmockgen -package activecluster -destination execution_manager_provider_mock.go -self_package github.com/uber/cadence/common/activecluster github.com/uber/cadence/common/activecluster ExecutionManagerProvider\n//\n\n// Package activecluster is a generated GoMock package.\npackage activecluster\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockExecutionManagerProvider is a mock of ExecutionManagerProvider interface.\ntype MockExecutionManagerProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutionManagerProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutionManagerProviderMockRecorder is the mock recorder for MockExecutionManagerProvider.\ntype MockExecutionManagerProviderMockRecorder struct {\n\tmock *MockExecutionManagerProvider\n}\n\n// NewMockExecutionManagerProvider creates a new mock instance.\nfunc NewMockExecutionManagerProvider(ctrl *gomock.Controller) *MockExecutionManagerProvider {\n\tmock := &MockExecutionManagerProvider{ctrl: ctrl}\n\tmock.recorder = &MockExecutionManagerProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutionManagerProvider) EXPECT() *MockExecutionManagerProviderMockRecorder {\n\treturn m.recorder\n}\n\n// GetExecutionManager mocks base method.\nfunc (m *MockExecutionManagerProvider) GetExecutionManager(shardID int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionManager\", shardID)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetExecutionManager indicates an expected call of GetExecutionManager.\nfunc (mr *MockExecutionManagerProviderMockRecorder) GetExecutionManager(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionManager\", reflect.TypeOf((*MockExecutionManagerProvider)(nil).GetExecutionManager), shardID)\n}\n"
  },
  {
    "path": "common/activecluster/manager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage activecluster\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype DomainIDToDomainFn func(id string) (*cache.DomainCacheEntry, error)\n\nconst (\n\tLookupNewWorkflowOpName                                 = \"LookupNewWorkflow\"\n\tLookupWorkflowOpName                                    = \"LookupWorkflow\"\n\tGetActiveClusterInfoByClusterAttributeOpName            = \"GetActiveClusterInfoByClusterAttribute\"\n\tGetActiveClusterInfoByWorkflowOpName                    = \"GetActiveClusterInfoByWorkflow\"\n\tGetActiveClusterSelectionPolicyForWorkflowOpName        = \"GetActiveClusterSelectionPolicyForWorkflow\"\n\tGetActiveClusterSelectionPolicyForCurrentWorkflowOpName = \"GetActiveClusterSelectionPolicyForCurrentWorkflow\"\n\tDomainIDToDomainFnErrorReason                           = \"domain_id_to_name_fn_error\"\n\n\tworkflowPolicyCacheTTL      = 10 * time.Second\n\tworkflowPolicyCacheMaxCount = 1000\n)\n\ntype managerImpl struct {\n\tdomainIDToDomainFn       DomainIDToDomainFn\n\tmetricsCl                metrics.Client\n\tlogger                   log.Logger\n\texecutionManagerProvider ExecutionManagerProvider\n\tnumShards                int\n\tworkflowPolicyCache      cache.Cache\n}\n\ntype ManagerOption func(*managerImpl)\n\nfunc NewManager(\n\tdomainIDToDomainFn DomainIDToDomainFn,\n\tmetricsCl metrics.Client,\n\tlogger log.Logger,\n\texecutionManagerProvider ExecutionManagerProvider,\n\tnumShards int,\n\topts ...ManagerOption,\n) (Manager, error) {\n\tm := &managerImpl{\n\t\tdomainIDToDomainFn:       domainIDToDomainFn,\n\t\tmetricsCl:                metricsCl,\n\t\tlogger:                   logger.WithTags(tag.ComponentActiveClusterManager),\n\t\texecutionManagerProvider: executionManagerProvider,\n\t\tnumShards:                numShards,\n\t\tworkflowPolicyCache: cache.New(&cache.Options{\n\t\t\tTTL:           workflowPolicyCacheTTL,\n\t\t\tMaxCount:      workflowPolicyCacheMaxCount,\n\t\t\tActivelyEvict: true,\n\t\t\tPin:           false,\n\t\t\tMetricsScope:  metricsCl.Scope(metrics.ActiveClusterManagerWorkflowCacheScope),\n\t\t\tLogger:        logger,\n\t\t}),\n\t}\n\tfor _, opt := range opts {\n\t\topt(m)\n\t}\n\treturn m, nil\n}\n\nfunc (m *managerImpl) getClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error) {\n\tshardID := common.WorkflowIDToHistoryShard(wfID, m.numShards)\n\texecutionManager, err := m.executionManagerProvider.GetExecutionManager(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif rID == \"\" {\n\t\texecution, err := executionManager.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trID = execution.RunID\n\t}\n\t// Check if the policy is already in the cache. create a key from domainID, wfID, rID\n\tkey := fmt.Sprintf(\"%s:%s:%s\", domainID, wfID, rID)\n\tcacheData := m.workflowPolicyCache.Get(key)\n\tif cacheData != nil {\n\t\tplcy, ok := cacheData.(*types.ActiveClusterSelectionPolicy)\n\t\tif ok {\n\t\t\treturn plcy, nil\n\t\t}\n\n\t\t// This should never happen. If it does, we ignore cache value and get it from DB.\n\t\tm.logger.Warn(fmt.Sprintf(\"Cache data for key %s is of type %T, not a *types.ActiveClusterSelectionPolicy\", key, cacheData))\n\t}\n\n\tplcy, err := executionManager.GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif plcy == nil {\n\t\treturn nil, &types.EntityNotExistsError{\n\t\t\tMessage: \"active cluster selection policy not found\",\n\t\t}\n\t}\n\n\tm.workflowPolicyCache.Put(key, plcy)\n\treturn plcy, nil\n}\n\nfunc (m *managerImpl) getDomainAndScope(domainID, fn string) (*cache.DomainCacheEntry, metrics.Scope, error) {\n\tscope := m.metricsCl.Scope(metrics.ActiveClusterManager).Tagged(metrics.ActiveClusterLookupFnTag(fn))\n\n\td, err := m.domainIDToDomainFn(domainID)\n\tif err != nil {\n\t\tscope = scope.Tagged(metrics.ReasonTag(DomainIDToDomainFnErrorReason))\n\t\tscope.IncCounter(metrics.ActiveClusterManagerLookupFailureCount)\n\t\treturn nil, nil, err\n\t}\n\n\tscope = scope.Tagged(metrics.DomainTag(d.GetInfo().Name))\n\tscope = scope.Tagged(metrics.IsActiveActiveDomainTag(d.GetReplicationConfig().IsActiveActive()))\n\tscope.IncCounter(metrics.ActiveClusterManagerLookupRequestCount)\n\n\treturn d, scope, nil\n}\n\nfunc (m *managerImpl) handleError(scope metrics.Scope, err *error, start time.Time) {\n\tif err != nil && *err != nil {\n\t\tscope.IncCounter(metrics.ActiveClusterManagerLookupFailureCount)\n\t} else {\n\t\tscope.IncCounter(metrics.ActiveClusterManagerLookupSuccessCount)\n\t}\n\tscope.RecordHistogramDuration(metrics.ActiveClusterManagerLookupLatency, time.Since(start))\n}\n\nfunc (m *managerImpl) GetActiveClusterInfoByClusterAttribute(ctx context.Context, domainID string, clusterAttribute *types.ClusterAttribute) (res *types.ActiveClusterInfo, e error) {\n\tdefer func() {\n\t\tlogFn := m.logger.Debug\n\t\tif e != nil {\n\t\t\tlogFn = m.logger.Warn\n\t\t}\n\t\tlogFn(\"GetActiveClusterInfoByClusterAttribute\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.Dynamic(\"clusterAttribute\", clusterAttribute),\n\t\t\ttag.Dynamic(\"result\", res),\n\t\t\ttag.Error(e),\n\t\t)\n\t}()\n\n\td, scope, err := m.getDomainAndScope(domainID, GetActiveClusterInfoByClusterAttributeOpName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer m.handleError(scope, &e, time.Now())\n\n\tres, ok := d.GetActiveClusterInfoByClusterAttribute(clusterAttribute)\n\tif !ok {\n\t\treturn nil, &ClusterAttributeNotFoundError{\n\t\t\tDomainID:         domainID,\n\t\t\tClusterAttribute: clusterAttribute,\n\t\t}\n\t}\n\treturn res, nil\n}\n\nfunc (m *managerImpl) GetActiveClusterInfoByWorkflow(ctx context.Context, domainID, wfID, rID string) (res *types.ActiveClusterInfo, e error) {\n\td, scope, err := m.getDomainAndScope(domainID, GetActiveClusterInfoByWorkflowOpName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer m.handleError(scope, &e, time.Now())\n\n\tif !d.GetReplicationConfig().IsActiveActive() {\n\t\t// Not an active-active domain. return ActiveClusterName from domain entry\n\t\tm.logger.Debug(\"GetActiveClusterInfoByWorkflow: not an active-active domain. returning ActiveClusterName from domain entry\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.ActiveClusterName(d.GetReplicationConfig().ActiveClusterName),\n\t\t\ttag.FailoverVersion(d.GetFailoverVersion()),\n\t\t)\n\t\treturn &types.ActiveClusterInfo{\n\t\t\tActiveClusterName: d.GetReplicationConfig().ActiveClusterName,\n\t\t\tFailoverVersion:   d.GetFailoverVersion(),\n\t\t}, nil\n\t}\n\n\tpolicy, err := m.getClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\tif err != nil {\n\t\tvar notExistsErr *types.EntityNotExistsError\n\t\tif !errors.As(err, &notExistsErr) {\n\t\t\treturn nil, err\n\t\t}\n\t\tpolicy = &types.ActiveClusterSelectionPolicy{}\n\t}\n\tres, ok := d.GetActiveClusterInfoByClusterAttribute(policy.ClusterAttribute)\n\tif !ok {\n\t\tm.logger.Debug(\"GetActiveClusterInfoByWorkflow: cluster attribute not found\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.ActiveClusterName(d.GetReplicationConfig().ActiveClusterName),\n\t\t\ttag.FailoverVersion(d.GetFailoverVersion()),\n\t\t)\n\t\treturn nil, &ClusterAttributeNotFoundError{\n\t\t\tDomainID:         domainID,\n\t\t\tClusterAttribute: policy.ClusterAttribute,\n\t\t}\n\t}\n\tm.logger.Debug(\"GetActiveClusterInfoByWorkflow: returning active cluster info\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(wfID),\n\t\ttag.WorkflowRunID(rID),\n\t\ttag.ActiveClusterName(res.ActiveClusterName),\n\t\ttag.FailoverVersion(res.FailoverVersion),\n\t)\n\n\treturn res, nil\n}\n\nfunc (m *managerImpl) GetActiveClusterSelectionPolicyForWorkflow(ctx context.Context, domainID, wfID, rID string) (res *types.ActiveClusterSelectionPolicy, e error) {\n\td, scope, err := m.getDomainAndScope(domainID, GetActiveClusterSelectionPolicyForWorkflowOpName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer m.handleError(scope, &e, time.Now())\n\n\tif !d.GetReplicationConfig().IsActiveActive() {\n\t\t// Not an active-active domain. return ActiveClusterName from domain entry\n\t\tm.logger.Debug(\"GetActiveClusterSelectionPolicyForWorkflow: not an active-active domain. returning ActiveClusterName from domain entry\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.ActiveClusterName(d.GetReplicationConfig().ActiveClusterName),\n\t\t\ttag.FailoverVersion(d.GetFailoverVersion()),\n\t\t)\n\t\treturn nil, nil\n\t}\n\n\tplcy, err := m.getClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\tif err != nil {\n\t\tvar notExistsErr *types.EntityNotExistsError\n\t\tif !errors.As(err, &notExistsErr) {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, nil\n\t}\n\treturn plcy, nil\n}\n\nfunc (m *managerImpl) GetActiveClusterSelectionPolicyForCurrentWorkflow(ctx context.Context, domainID, wfID string) (res *types.ActiveClusterSelectionPolicy, running bool, e error) {\n\td, scope, err := m.getDomainAndScope(domainID, GetActiveClusterSelectionPolicyForCurrentWorkflowOpName)\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\tdefer m.handleError(scope, &e, time.Now())\n\tif !d.GetReplicationConfig().IsActiveActive() {\n\t\t// Not an active-active domain. return nil\n\t\tm.logger.Debug(\"GetActiveClusterSelectionPolicyForCurrentWorkflow: not an active-active domain. returning nil\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t)\n\t\treturn nil, false, nil\n\t}\n\n\tshardID := common.WorkflowIDToHistoryShard(wfID, m.numShards)\n\texecutionManager, err := m.executionManagerProvider.GetExecutionManager(shardID)\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\texecution, err := executionManager.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: wfID,\n\t})\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\tif persistence.IsWorkflowRunning(execution.State) {\n\t\tpolicy, err := m.getClusterSelectionPolicy(ctx, domainID, wfID, execution.RunID)\n\t\tif err != nil {\n\t\t\treturn nil, false, err\n\t\t}\n\t\treturn policy, true, nil\n\t}\n\treturn nil, false, nil\n}\n"
  },
  {
    "path": "common/activecluster/manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/activecluster (interfaces: Manager)\n//\n// Generated by this command:\n//\n//\tmockgen -package activecluster -destination manager_mock.go -self_package github.com/uber/cadence/common/activecluster github.com/uber/cadence/common/activecluster Manager\n//\n\n// Package activecluster is a generated GoMock package.\npackage activecluster\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockManager is a mock of Manager interface.\ntype MockManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockManagerMockRecorder is the mock recorder for MockManager.\ntype MockManagerMockRecorder struct {\n\tmock *MockManager\n}\n\n// NewMockManager creates a new mock instance.\nfunc NewMockManager(ctrl *gomock.Controller) *MockManager {\n\tmock := &MockManager{ctrl: ctrl}\n\tmock.recorder = &MockManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockManager) EXPECT() *MockManagerMockRecorder {\n\treturn m.recorder\n}\n\n// GetActiveClusterInfoByClusterAttribute mocks base method.\nfunc (m *MockManager) GetActiveClusterInfoByClusterAttribute(ctx context.Context, domainID string, clusterAttribute *types.ClusterAttribute) (*types.ActiveClusterInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterInfoByClusterAttribute\", ctx, domainID, clusterAttribute)\n\tret0, _ := ret[0].(*types.ActiveClusterInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActiveClusterInfoByClusterAttribute indicates an expected call of GetActiveClusterInfoByClusterAttribute.\nfunc (mr *MockManagerMockRecorder) GetActiveClusterInfoByClusterAttribute(ctx, domainID, clusterAttribute any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterInfoByClusterAttribute\", reflect.TypeOf((*MockManager)(nil).GetActiveClusterInfoByClusterAttribute), ctx, domainID, clusterAttribute)\n}\n\n// GetActiveClusterInfoByWorkflow mocks base method.\nfunc (m *MockManager) GetActiveClusterInfoByWorkflow(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterInfoByWorkflow\", ctx, domainID, wfID, rID)\n\tret0, _ := ret[0].(*types.ActiveClusterInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActiveClusterInfoByWorkflow indicates an expected call of GetActiveClusterInfoByWorkflow.\nfunc (mr *MockManagerMockRecorder) GetActiveClusterInfoByWorkflow(ctx, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterInfoByWorkflow\", reflect.TypeOf((*MockManager)(nil).GetActiveClusterInfoByWorkflow), ctx, domainID, wfID, rID)\n}\n\n// GetActiveClusterSelectionPolicyForCurrentWorkflow mocks base method.\nfunc (m *MockManager) GetActiveClusterSelectionPolicyForCurrentWorkflow(ctx context.Context, domainID, wfID string) (*types.ActiveClusterSelectionPolicy, bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterSelectionPolicyForCurrentWorkflow\", ctx, domainID, wfID)\n\tret0, _ := ret[0].(*types.ActiveClusterSelectionPolicy)\n\tret1, _ := ret[1].(bool)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetActiveClusterSelectionPolicyForCurrentWorkflow indicates an expected call of GetActiveClusterSelectionPolicyForCurrentWorkflow.\nfunc (mr *MockManagerMockRecorder) GetActiveClusterSelectionPolicyForCurrentWorkflow(ctx, domainID, wfID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterSelectionPolicyForCurrentWorkflow\", reflect.TypeOf((*MockManager)(nil).GetActiveClusterSelectionPolicyForCurrentWorkflow), ctx, domainID, wfID)\n}\n\n// GetActiveClusterSelectionPolicyForWorkflow mocks base method.\nfunc (m *MockManager) GetActiveClusterSelectionPolicyForWorkflow(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterSelectionPolicyForWorkflow\", ctx, domainID, wfID, rID)\n\tret0, _ := ret[0].(*types.ActiveClusterSelectionPolicy)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActiveClusterSelectionPolicyForWorkflow indicates an expected call of GetActiveClusterSelectionPolicyForWorkflow.\nfunc (mr *MockManagerMockRecorder) GetActiveClusterSelectionPolicyForWorkflow(ctx, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterSelectionPolicyForWorkflow\", reflect.TypeOf((*MockManager)(nil).GetActiveClusterSelectionPolicyForWorkflow), ctx, domainID, wfID, rID)\n}\n"
  },
  {
    "path": "common/activecluster/manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage activecluster\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tnumShards = 10\n)\n\nfunc TestGetActiveClusterInfoByClusterAttribute(t *testing.T) {\n\tmetricsCl := metrics.NewNoopMetricsClient()\n\tlogger := log.NewNoop()\n\n\ttests := []struct {\n\t\tname              string\n\t\tclusterAttribute  *types.ClusterAttribute\n\t\tactiveClusterCfg  *types.ActiveClusters\n\t\tdomainIDToNameErr error\n\t\texpectedResult    *types.ActiveClusterInfo\n\t\texpectedError     string\n\t}{\n\t\t{\n\t\t\tname:             \"nil cluster attribute - returns domain-level active cluster info\",\n\t\t\tclusterAttribute: nil,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   20,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\tFailoverVersion:   201, // domain failover version\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"domain ID to name function returns error\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tdomainIDToNameErr: errors.New(\"failed to find domain by id\"),\n\t\t\texpectedError:     \"failed to find domain by id\",\n\t\t},\n\t\t{\n\t\t\tname: \"nil active clusters - returns ClusterAttributeNotFoundError\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\tactiveClusterCfg: nil,\n\t\t\texpectedError:    \"could not find cluster attribute &{region us-west} in the domain test-domain-id's active cluster config\",\n\t\t},\n\t\t{\n\t\t\tname: \"nil attribute scopes - returns ClusterAttributeNotFoundError\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: nil,\n\t\t\t},\n\t\t\texpectedError: \"could not find cluster attribute &{region us-west} in the domain test-domain-id's active cluster config\",\n\t\t},\n\t\t{\n\t\t\tname: \"scope not found - returns ClusterAttributeNotFoundError\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"datacenter\",\n\t\t\t\tName:  \"dc1\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: \"could not find cluster attribute &{datacenter dc1} in the domain test-domain-id's active cluster config\",\n\t\t},\n\t\t{\n\t\t\tname: \"attribute name not found in scope - returns ClusterAttributeNotFoundError\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-central\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: \"could not find cluster attribute &{region us-central} in the domain test-domain-id's active cluster config\",\n\t\t},\n\t\t{\n\t\t\tname: \"successful lookup - returns active cluster info for region attribute\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\tFailoverVersion:   100,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"successful lookup - returns active cluster info for custom scope\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"datacenter\",\n\t\t\t\tName:  \"dc1\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   150,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   250,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\tFailoverVersion:   150,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"successful lookup - multiple scopes with different attributes\",\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"city\",\n\t\t\t\tName:  \"seattle\",\n\t\t\t},\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"city\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"san_francisco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   350,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\tFailoverVersion:   300,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainIDToDomainFn := func(id string) (*cache.DomainCacheEntry, error) {\n\t\t\t\treturn getDomainCacheEntryWithAttributeScopes(tc.activeClusterCfg), tc.domainIDToNameErr\n\t\t\t}\n\n\t\t\tmgr, err := NewManager(\n\t\t\t\tdomainIDToDomainFn,\n\t\t\t\tmetricsCl,\n\t\t\t\tlogger,\n\t\t\t\tnil,\n\t\t\t\tnumShards,\n\t\t\t)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tresult, err := mgr.GetActiveClusterInfoByClusterAttribute(context.Background(), \"test-domain-id\", tc.clusterAttribute)\n\t\t\tif tc.expectedError != \"\" {\n\t\t\t\tassert.EqualError(t, err, tc.expectedError)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getDomainCacheEntry(cfg *types.ActiveClusters, migratedFromActivePassive bool) *cache.DomainCacheEntry {\n\t// only thing we care in domain cache entry is the active clusters config\n\t// for domains migrated from active-passive to active-active, we set the failover version to 201\n\tactiveClusterName := \"\"\n\tif migratedFromActivePassive || cfg == nil {\n\t\tactiveClusterName = \"cluster0\"\n\t}\n\treturn cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tName: \"test-domain-id\",\n\t\t},\n\t\tnil,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters:    cfg,\n\t\t\tActiveClusterName: activeClusterName,\n\t\t},\n\t\t201,\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t1,\n\t)\n}\n\nfunc TestGetActiveClusterInfoByWorkflow(t *testing.T) {\n\tmetricsCl := metrics.NewNoopMetricsClient()\n\tlogger := log.NewNoop()\n\n\ttests := []struct {\n\t\tname                           string\n\t\trunID                          string\n\t\tactiveClusterCfg               *types.ActiveClusters\n\t\tdomainIDToNameErr              error\n\t\tmockExecutionManagerFn         func(em *persistence.MockExecutionManager)\n\t\tmockExecutionManagerProviderFn func(emp *MockExecutionManagerProvider)\n\t\tcachedPolicy                   *types.ActiveClusterSelectionPolicy\n\t\texpectedResult                 *types.ActiveClusterInfo\n\t\texpectedError                  string\n\t}{\n\t\t{\n\t\t\tname:  \"domain ID to name function returns error\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tdomainIDToNameErr: errors.New(\"failed to find domain by id\"),\n\t\t\texpectedError:     \"failed to find domain by id\",\n\t\t},\n\t\t{\n\t\t\tname:  \"execution manager provider returns error\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerProviderFn: func(emp *MockExecutionManagerProvider) {\n\t\t\t\temp.EXPECT().GetExecutionManager(6).Return(nil, errors.New(\"failed to get execution manager\")).AnyTimes()\n\t\t\t},\n\t\t\texpectedError: \"failed to get execution manager\",\n\t\t},\n\t\t{\n\t\t\tname:  \"execution manager GetActiveClusterSelectionPolicy returns error\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(nil, errors.New(\"database error\"))\n\t\t\t},\n\t\t\texpectedError: \"database error\",\n\t\t},\n\t\t{\n\t\t\tname:  \"policy not found (EntityNotExistsError) - uses empty policy with nil cluster attribute\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{Message: \"policy not found\"})\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\", // domain-level active cluster\n\t\t\t\tFailoverVersion:   201,        // domain failover version\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"policy not found (EntityNotExistsError) - empty policy with cluster attribute not found\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{Message: \"policy not found\"})\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\", // domain-level active cluster\n\t\t\t\tFailoverVersion:   201,        // domain failover version\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"policy found but cluster attribute not found in domain config\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"datacenter\",\n\t\t\t\t\t\t\tName:  \"dc1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: \"could not find cluster attribute &{datacenter dc1} in the domain test-domain-id's active cluster config\",\n\t\t},\n\t\t{\n\t\t\tname:  \"successful lookup - policy found and cluster attribute exists\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\tFailoverVersion:   200,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"successful lookup - policy from cache\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   150,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   250,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tcachedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"datacenter\",\n\t\t\t\t\tName:  \"dc2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\t// No expectations needed since policy comes from cache, but we need the provider to be non-nil\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\tFailoverVersion:   250,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"successful lookup - nil cluster attribute in policy uses domain-level info\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: nil, // nil cluster attribute\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster0\", // domain-level active cluster\n\t\t\t\tFailoverVersion:   201,        // domain failover version\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"successful lookup - multiple scopes with different attributes\",\n\t\t\trunID: \"test-run-id\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"city\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"san_francisco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   350,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"city\",\n\t\t\t\t\t\t\tName:  \"seattle\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\tFailoverVersion:   300,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"successful lookup - empty runID triggers GetCurrentExecution\",\n\t\t\trunID: \"\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\t// Expect GetCurrentExecution to be called first to get the runID\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"current-run-id\",\n\t\t\t\t}, nil)\n\t\t\t\t// Then expect GetActiveClusterSelectionPolicy with the returned runID\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"current-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\tFailoverVersion:   200,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"empty runID - GetCurrentExecution returns error\",\n\t\t\trunID: \"\",\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\t// GetCurrentExecution returns an error\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(nil, errors.New(\"workflow not found\"))\n\t\t\t},\n\t\t\texpectedError: \"workflow not found\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainIDToDomainFn := func(id string) (*cache.DomainCacheEntry, error) {\n\t\t\t\treturn getDomainCacheEntryWithAttributeScopes(tc.activeClusterCfg), tc.domainIDToNameErr\n\t\t\t}\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\twfID := \"test-workflow-id\"\n\t\t\tshardID := 6 // corresponds to wfID given numShards\n\n\t\t\tvar emProvider *MockExecutionManagerProvider\n\t\t\tif tc.mockExecutionManagerProviderFn != nil || tc.mockExecutionManagerFn != nil {\n\t\t\t\temProvider = NewMockExecutionManagerProvider(ctrl)\n\n\t\t\t\tif tc.mockExecutionManagerProviderFn != nil {\n\t\t\t\t\ttc.mockExecutionManagerProviderFn(emProvider)\n\t\t\t\t} else {\n\t\t\t\t\t// Default case: create execution manager and set up provider\n\t\t\t\t\tem := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\t\tif tc.mockExecutionManagerFn != nil {\n\t\t\t\t\t\ttc.mockExecutionManagerFn(em)\n\t\t\t\t\t}\n\t\t\t\t\temProvider.EXPECT().GetExecutionManager(shardID).Return(em, nil).AnyTimes()\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tmgr, err := NewManager(\n\t\t\t\tdomainIDToDomainFn,\n\t\t\t\tmetricsCl,\n\t\t\t\tlogger,\n\t\t\t\temProvider,\n\t\t\t\tnumShards,\n\t\t\t)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif tc.cachedPolicy != nil {\n\t\t\t\tkey := fmt.Sprintf(\"%s:%s:%s\", \"test-domain-id\", wfID, tc.runID)\n\t\t\t\tmgr.(*managerImpl).workflowPolicyCache.Put(key, tc.cachedPolicy)\n\t\t\t}\n\n\t\t\tresult, err := mgr.GetActiveClusterInfoByWorkflow(context.Background(), \"test-domain-id\", wfID, tc.runID)\n\t\t\tif tc.expectedError != \"\" {\n\t\t\t\tassert.EqualError(t, err, tc.expectedError)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getDomainCacheEntryWithAttributeScopes(cfg *types.ActiveClusters) *cache.DomainCacheEntry {\n\t// Helper function specifically for GetActiveClusterInfoByClusterAttribute tests\n\t// Always sets activeClusterName to \"cluster0\" for consistency\n\treturn cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tName: \"test-domain-id\",\n\t\t},\n\t\tnil,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters:    cfg,\n\t\t\tActiveClusterName: \"cluster0\",\n\t\t},\n\t\t201, // domain failover version\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t1,\n\t)\n}\n\nfunc TestGetActiveClusterSelectionPolicyForCurrentWorkflow(t *testing.T) {\n\tmetricsCl := metrics.NewNoopMetricsClient()\n\tlogger := log.NewNoop()\n\n\ttests := []struct {\n\t\tname                           string\n\t\tactiveClusterCfg               *types.ActiveClusters\n\t\tisActiveActive                 bool\n\t\tdomainIDToNameErr              error\n\t\tmockExecutionManagerFn         func(em *persistence.MockExecutionManager)\n\t\tmockExecutionManagerProviderFn func(emp *MockExecutionManagerProvider)\n\t\tcachedPolicy                   *types.ActiveClusterSelectionPolicy\n\t\texpectedPolicy                 *types.ActiveClusterSelectionPolicy\n\t\texpectedRunning                bool\n\t\texpectedError                  string\n\t}{\n\t\t{\n\t\t\tname:              \"domain ID to name function returns error\",\n\t\t\tisActiveActive:    true,\n\t\t\tdomainIDToNameErr: errors.New(\"failed to find domain by id\"),\n\t\t\texpectedError:     \"failed to find domain by id\",\n\t\t\texpectedRunning:   false,\n\t\t},\n\t\t{\n\t\t\tname:            \"non-active-active domain returns nil\",\n\t\t\tisActiveActive:  false,\n\t\t\texpectedPolicy:  nil,\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"execution manager provider returns error\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerProviderFn: func(emp *MockExecutionManagerProvider) {\n\t\t\t\temp.EXPECT().GetExecutionManager(6).Return(nil, errors.New(\"failed to get execution manager\")).AnyTimes()\n\t\t\t},\n\t\t\texpectedError:   \"failed to get execution manager\",\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"GetCurrentExecution returns error\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(nil, errors.New(\"workflow not found\"))\n\t\t\t},\n\t\t\texpectedError:   \"workflow not found\",\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow completed - returns nil, false\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedPolicy:  nil,\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow running - successfully returns policy\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil)\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"region\",\n\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRunning: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow created state - successfully returns policy\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   150,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   250,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t}, nil)\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"datacenter\",\n\t\t\t\t\t\t\tName:  \"dc1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"datacenter\",\n\t\t\t\t\tName:  \"dc1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRunning: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow running but GetActiveClusterSelectionPolicy fails\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil)\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(nil, errors.New(\"database error\"))\n\t\t\t},\n\t\t\texpectedError:   \"database error\",\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow running - policy from cache\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"city\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"san_francisco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   350,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil)\n\t\t\t\t// No expectation for GetActiveClusterSelectionPolicy since it comes from cache\n\t\t\t},\n\t\t\tcachedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"city\",\n\t\t\t\t\tName:  \"seattle\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"city\",\n\t\t\t\t\tName:  \"seattle\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRunning: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow running with EntityNotExistsError returns error\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil)\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{Message: \"policy not found\"})\n\t\t\t},\n\t\t\texpectedError:   \"policy not found\",\n\t\t\texpectedRunning: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"workflow running with nil cluster attribute in policy\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveClusterCfg: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockExecutionManagerFn: func(em *persistence.MockExecutionManager) {\n\t\t\t\tem.EXPECT().GetCurrentExecution(gomock.Any(), &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil)\n\t\t\t\tem.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").\n\t\t\t\t\tReturn(&types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\tClusterAttribute: nil,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: nil,\n\t\t\t},\n\t\t\texpectedRunning: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainIDToDomainFn := func(id string) (*cache.DomainCacheEntry, error) {\n\t\t\t\treturn getDomainCacheEntry(tc.activeClusterCfg, tc.isActiveActive), tc.domainIDToNameErr\n\t\t\t}\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\twfID := \"test-workflow-id\"\n\t\t\tshardID := 6 // corresponds to wfID given numShards\n\n\t\t\tvar emProvider *MockExecutionManagerProvider\n\t\t\tif tc.mockExecutionManagerProviderFn != nil || tc.mockExecutionManagerFn != nil {\n\t\t\t\temProvider = NewMockExecutionManagerProvider(ctrl)\n\n\t\t\t\tif tc.mockExecutionManagerProviderFn != nil {\n\t\t\t\t\ttc.mockExecutionManagerProviderFn(emProvider)\n\t\t\t\t} else {\n\t\t\t\t\t// Default case: create execution manager and set up provider\n\t\t\t\t\tem := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\t\tif tc.mockExecutionManagerFn != nil {\n\t\t\t\t\t\ttc.mockExecutionManagerFn(em)\n\t\t\t\t\t}\n\t\t\t\t\temProvider.EXPECT().GetExecutionManager(shardID).Return(em, nil).AnyTimes()\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tmgr, err := NewManager(\n\t\t\t\tdomainIDToDomainFn,\n\t\t\t\tmetricsCl,\n\t\t\t\tlogger,\n\t\t\t\temProvider,\n\t\t\t\tnumShards,\n\t\t\t)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif tc.cachedPolicy != nil {\n\t\t\t\t// Need to get the runID from the test case to build the cache key\n\t\t\t\t// For cached policy tests, we assume the runID is \"test-run-id\"\n\t\t\t\tkey := fmt.Sprintf(\"%s:%s:%s\", \"test-domain-id\", wfID, \"test-run-id\")\n\t\t\t\tmgr.(*managerImpl).workflowPolicyCache.Put(key, tc.cachedPolicy)\n\t\t\t}\n\n\t\t\tpolicy, running, err := mgr.GetActiveClusterSelectionPolicyForCurrentWorkflow(context.Background(), \"test-domain-id\", wfID)\n\t\t\tif tc.expectedError != \"\" {\n\t\t\t\tassert.EqualError(t, err, tc.expectedError)\n\t\t\t\tassert.Nil(t, policy)\n\t\t\t\tassert.Equal(t, tc.expectedRunning, running)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedPolicy, policy)\n\t\t\t\tassert.Equal(t, tc.expectedRunning, running)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/activecluster/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage activecluster\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination manager_mock.go -self_package github.com/uber/cadence/common/activecluster github.com/uber/cadence/common/activecluster Manager\n//go:generate mockgen -package $GOPACKAGE -destination execution_manager_provider_mock.go -self_package github.com/uber/cadence/common/activecluster github.com/uber/cadence/common/activecluster ExecutionManagerProvider\n\n// Manager is the interface for active cluster manager.\n// It is used to get active cluster info by cluster attribute or workflow.\ntype Manager interface {\n\t// GetActiveClusterInfoByClusterAttribute returns the active cluster info by cluster attribute\n\t// If clusterAttribute is nil, returns the domain-level active cluster info\n\t// If clusterAttribute is not nil and exists in the domain metadata, returns the active cluster info of the cluster attribute\n\t// If clusterAttribute is not nil and does not exist in the domain metadata, returns an error\n\tGetActiveClusterInfoByClusterAttribute(ctx context.Context, domainID string, clusterAttribute *types.ClusterAttribute) (*types.ActiveClusterInfo, error)\n\n\t// GetActiveClusterInfoByWorkflow returns the active cluster info by workflow\n\t// It will first look up the cluster selection policy for the workflow and then get the active cluster info by cluster attribute from the policy\n\tGetActiveClusterInfoByWorkflow(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterInfo, error)\n\n\t// GetActiveClusterSelectionPolicyForWorkflow returns the active cluster selection policy for a workflow\n\tGetActiveClusterSelectionPolicyForWorkflow(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error)\n\n\t// GetActiveClusterSelectionPolicyForCurrentWorkflow returns the active cluster selection policy for the current workflow\n\t// if the workflow is NOT closed, returns policy and true, otherwise returns nil and false\n\tGetActiveClusterSelectionPolicyForCurrentWorkflow(ctx context.Context, domainID, wfID string) (*types.ActiveClusterSelectionPolicy, bool, error)\n}\n\ntype ExecutionManagerProvider interface {\n\tGetExecutionManager(shardID int) (persistence.ExecutionManager, error)\n}\n\ntype ClusterAttributeNotFoundError struct {\n\tDomainID         string\n\tClusterAttribute *types.ClusterAttribute\n\t// ActiveClusterInfo *types.ActiveClusterInfo\n}\n\nfunc (e *ClusterAttributeNotFoundError) Error() string {\n\treturn fmt.Sprintf(\"could not find cluster attribute %s in the domain %s's active cluster config\", e.ClusterAttribute, e.DomainID)\n}\n"
  },
  {
    "path": "common/archiver/README.md",
    "content": "## What\n\nThis README explains how to add new Archiver implementations.\n\n## Steps\n\n**Step 1: Create a new package for your implementation**\n\nCreate a new directory in the `archiver` folder. The structure should look like the following:\n```\n./common/archiver\n  - filestore/                      -- Filestore implementation \n  - provider/\n      - provider.go                 -- Provider of archiver instances\n  - yourImplementation/\n      - historyArchiver.go          -- HistoryArchiver implementation\n      - historyArchiver_test.go     -- Unit tests for HistoryArchiver\n      - visibilityArchiver.go       -- VisibilityArchiver implementations\n      - visibilityArchiver_test.go  -- Unit tests for VisibilityArchiver\n```\n\n**Step 2: Implement the HistoryArchiver interface**\n\n```go\ntype HistoryArchiver interface {\n\t// Archive is used to archive a workflow history. When the context expires the method should stop trying to archive.\n\t// Implementors are free to archive however they want, including implementing retries of sub-operations. The URI defines\n\t// the resource that histories should be archived into. The implementor gets to determine how to interpret the URI.\n\t// The Archive method may or may not be automatically retried by the caller. The ArchiveOptions are used\n\t// to interact with these retries including giving the implementor the ability to cancel retries and record progress\n  // between retry attempts. \n  // This method will be invoked after a workflow passes its retention period.\n  // It's possible that this method will be invoked for one workflow multiple times and potentially concurrently,\n  // implementation should correctly handle the workflow not exist case and return nil error.\n    Archive(context.Context, URI, *ArchiveHistoryRequest, ...ArchiveOption) error\n    \n    // Get is used to access an archived history. When context expires method should stop trying to fetch history.\n    // The URI identifies the resource from which history should be accessed and it is up to the implementor to interpret this URI.\n    // This method should thrift errors - see filestore as an example.\n    Get(context.Context, URI, *GetHistoryRequest) (*GetHistoryResponse, error)\n    \n    // ValidateURI is used to define what a valid URI for an implementation is.\n    ValidateURI(URI) error\n}\n```\n\n**Step 3: Implement the VisibilityArchiver interface**\n\n```go\ntype VisibilityArchiver interface {\n    // Archive is used to archive one workflow visibility record. \n    // Check the Archive() method of the HistoryArchiver interface in Step 2 for parameters' meaning and requirements. \n    // The only difference is that the ArchiveOption parameter won't include an option for recording process. \n    // Please make sure your implementation is lossless. If any in-memory batching mechanism is used, then those batched records will be lost during server restarts. \n    // This method will be invoked when workflow closes. Note that because of conflict resolution, it is possible for a workflow to through the closing process multiple times, which means that this method can be invoked more than once after a workflow closes.\n    Archive(context.Context, URI, *ArchiveVisibilityRequest, ...ArchiveOption) error\n    \n    // Query is used to retrieve archived visibility records. \n    // Check the Get() method of the HistoryArchiver interface in Step 2 for parameters' meaning and requirements.\n    // The request includes a string field called query, which describes what kind of visibility records should be returned. For example, it can be some SQL-like syntax query string. \n    // Your implementation is responsible for parsing and validating the query, and also returning all visibility records that match the query. \n    // Currently the maximum context timeout passed into the method is 3 minutes, so it's ok if this method takes a long time to run.\n    Query(context.Context, URI, *QueryVisibilityRequest) (*QueryVisibilityResponse, error)\n\n    // ValidateURI is used to define what a valid URI for an implementation is.\n    ValidateURI(URI) error\n}\n```\n\n**Step 4: Update provider to provide access to your implementation**\n\nModify the `./provider/provider.go` file so that the `ArchiverProvider` knows how to create an instance of your archiver. \nAlso, add configs for you archiver to static yaml config files and modify the `HistoryArchiverProvider` \nand `VisibilityArchiverProvider` struct in the `../common/config.go` accordingly.\n\n\n## FAQ\n**If my Archive method can automatically be retried by caller how can I record and access progress between retries?**\n\nArchiverOptions is used to handle this. The following shows and example: \n```go\nfunc (a *Archiver) Archive(\n\tctx context.Context,\n\tURI string,\n\trequest *ArchiveRequest,\n\topts ...ArchiveOption,\n) error {\n  featureCatalog := GetFeatureCatalog(opts...) // this function is defined in options.go\n\n  var progress progress\n\n  // Check if the feature for recording progress is enabled.\n  if featureCatalog.ProgressManager != nil {\n    if err := featureCatalog.ProgressManager.LoadProgress(ctx, &prevProgress); err != nil {\n      // log some error message and return error if needed.\n    }\n  }\n\n  // Your archiver implementation...\n\n  // Record current progress\n  if featureCatalog.ProgressManager != nil {\n    if err := featureCatalog.ProgressManager.RecordProgress(ctx, progress); err != nil {\n      // log some error message and return error if needed. \n    }\n  }\n}\n```\n\n**If my Archive method encounters an error which is non-retryable how do I indicate that the caller should not retry?**\n\n```go\nfunc (a *Archiver) Archive(\n\tctx context.Context,\n\tURI string,\n\trequest *ArchiveRequest,\n\topts ...ArchiveOption,\n) error {\n  featureCatalog := GetFeatureCatalog(opts...) // this function is defined in options.go\n\n  err := youArchiverImpl()\n  if nonRetryableErr(err) {\n    if featureCatalog.NonRetriableError != nil {\n\t  return featureCatalog.NonRetriableError() // when the caller gets this error type back it will not retry anymore.\n    }\n  }\n}\n```\n\n**How does my history archiver implementation read history?**\n\nThe `archiver` package provides a utility class called `HistoryIterator` which is a wrapper of `HistoryManager`. \nIts usage is simpler than the `HistoryManager` given in the `BootstrapContainer`, \nso archiver implementations can choose to use it when reading workflow histories. \nSee the `historyIterator.go` file for more details. \nSample usage can be found in the filestore historyArchiver implementation.\n\n**Should my archiver define all its own error types?**\n\nEach archiver is free to define and return any errors it wants. However many common errors which\nexist between archivers are already defined in `constants.go`.\n\n**Is there a generic query syntax for visibility archiver?**\n\nCurrently no. But this is something we plan to do in the future. As for now, try to make your syntax similar to the one used by our advanced list workflow API.\n"
  },
  {
    "path": "common/archiver/URI.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport \"net/url\"\n\ntype (\n\t// URI identifies the archival resource to which records are written to and read from.\n\tURI interface {\n\t\tScheme() string\n\t\tPath() string\n\t\tHostname() string\n\t\tPort() string\n\t\tUsername() string\n\t\tPassword() string\n\t\tString() string\n\t\tOpaque() string\n\t\tQuery() map[string][]string\n\t}\n\n\turi struct {\n\t\turl *url.URL\n\t}\n)\n\n// NewURI constructs a new archiver URI from string.\nfunc NewURI(s string) (URI, error) {\n\turl, err := url.ParseRequestURI(s)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &uri{url: url}, nil\n}\n\nfunc (u *uri) Scheme() string {\n\treturn u.url.Scheme\n}\n\nfunc (u *uri) Path() string {\n\treturn u.url.Path\n}\n\nfunc (u *uri) Hostname() string {\n\treturn u.url.Hostname()\n}\n\nfunc (u *uri) Port() string {\n\treturn u.url.Port()\n}\n\nfunc (u *uri) Username() string {\n\tif u.url.User == nil {\n\t\treturn \"\"\n\t}\n\treturn u.url.User.Username()\n}\n\nfunc (u *uri) Password() string {\n\tif u.url.User == nil {\n\t\treturn \"\"\n\t}\n\tpassword, exist := u.url.User.Password()\n\tif !exist {\n\t\treturn \"\"\n\t}\n\treturn password\n}\n\nfunc (u *uri) Opaque() string {\n\treturn u.url.Opaque\n}\n\nfunc (u *uri) Query() map[string][]string {\n\treturn u.url.Query()\n}\n\nfunc (u *uri) String() string {\n\treturn u.url.String()\n}\n"
  },
  {
    "path": "common/archiver/URI_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tURISuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestURISuite(t *testing.T) {\n\tsuite.Run(t, new(URISuite))\n}\n\nfunc (s *URISuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *URISuite) TestURI() {\n\ttestCases := []struct {\n\t\tURIString string\n\t\tvalid     bool\n\t\tscheme    string\n\t\tpath      string\n\t\thostname  string\n\t\tport      string\n\t\tusername  string\n\t\tpassword  string\n\t\topaque    string\n\t\tquery     map[string][]string\n\t}{\n\t\t{\n\t\t\tURIString: \"\",\n\t\t\tvalid:     false,\n\t\t},\n\t\t{\n\t\t\tURIString: \"some random string\",\n\t\t\tvalid:     false,\n\t\t},\n\t\t{\n\t\t\tURIString: \"mailto:a@b.com\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"mailto\",\n\t\t\topaque:    \"a@b.com\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"test://\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"test\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"http://example.com/path\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"http\",\n\t\t\thostname:  \"example.com\",\n\t\t\tpath:      \"/path\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"http://example.com/path with space\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"http\",\n\t\t\thostname:  \"example.com\",\n\t\t\tpath:      \"/path with space\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"https://localhost:8080?key1=value1&key1=value2&key2=value3\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"https\",\n\t\t\thostname:  \"localhost\",\n\t\t\tport:      \"8080\",\n\t\t\tquery: map[string][]string{\n\t\t\t\t\"key1\": []string{\"value1\", \"value2\"},\n\t\t\t\t\"key2\": []string{\"value3\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tURIString: \"file:///absolute/path/to/dir\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"file\",\n\t\t\tpath:      \"/absolute/path/to/dir\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"test://person:password@host/path\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"test\",\n\t\t\thostname:  \"host\",\n\t\t\tpath:      \"/path\",\n\t\t\tusername:  \"person\",\n\t\t\tpassword:  \"password\",\n\t\t},\n\t\t{\n\t\t\tURIString: \"test:opaque?key1=value1&key1=value2&key2=value3\",\n\t\t\tvalid:     true,\n\t\t\tscheme:    \"test\",\n\t\t\topaque:    \"opaque\",\n\t\t\tquery: map[string][]string{\n\t\t\t\t\"key1\": []string{\"value1\", \"value2\"},\n\t\t\t\t\"key2\": []string{\"value3\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tURI, err := NewURI(tc.URIString)\n\t\tif !tc.valid {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\n\t\ts.NoError(err)\n\t\ts.Equal(tc.scheme, URI.Scheme())\n\t\ts.Equal(tc.path, URI.Path())\n\t\ts.Equal(tc.hostname, URI.Hostname())\n\t\ts.Equal(tc.port, URI.Port())\n\t\ts.Equal(tc.username, URI.Username())\n\t\ts.Equal(tc.password, URI.Password())\n\t\ts.Equal(tc.opaque, URI.Opaque())\n\t\tif tc.query != nil {\n\t\t\ts.Equal(tc.query, URI.Query())\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/archiver/archivalMetadata.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ArchivalMetadata provides cluster level archival information\n\tArchivalMetadata interface {\n\t\tGetHistoryConfig() ArchivalConfig\n\t\tGetVisibilityConfig() ArchivalConfig\n\t}\n\n\t// ArchivalConfig is an immutable representation of the archival configuration of the cluster\n\t// This config is determined at cluster startup time\n\tArchivalConfig interface {\n\t\tClusterConfiguredForArchival() bool\n\t\tGetClusterStatus() ArchivalStatus\n\t\tReadEnabled() bool\n\t\tGetDomainDefaultStatus() types.ArchivalStatus\n\t\tGetDomainDefaultURI() string\n\t}\n\n\tarchivalMetadata struct {\n\t\thistoryConfig    ArchivalConfig\n\t\tvisibilityConfig ArchivalConfig\n\t}\n\n\tarchivalConfig struct {\n\t\tstaticClusterStatus  ArchivalStatus\n\t\tdynamicClusterStatus dynamicproperties.StringPropertyFn\n\t\tstaticEnableRead     bool\n\t\tdynamicEnableRead    dynamicproperties.BoolPropertyFn\n\t\tdomainDefaultStatus  types.ArchivalStatus\n\t\tdomainDefaultURI     string\n\t}\n\n\t// ArchivalStatus represents the archival status of the cluster\n\tArchivalStatus int\n)\n\nconst (\n\t// ArchivalDisabled means this cluster is not configured to handle archival\n\tArchivalDisabled ArchivalStatus = iota\n\t// ArchivalPaused means this cluster is configured to handle archival but is currently not archiving\n\t// This state is not yet implemented, as of now ArchivalPaused is treated the same way as ArchivalDisabled\n\tArchivalPaused\n\t// ArchivalEnabled means this cluster is currently archiving\n\tArchivalEnabled\n)\n\n// NewArchivalMetadata constructs a new ArchivalMetadata\nfunc NewArchivalMetadata(\n\tdc *dynamicconfig.Collection,\n\thistoryStatus string,\n\thistoryReadEnabled bool,\n\tvisibilityStatus string,\n\tvisibilityReadEnabled bool,\n\tdomainDefaults *config.ArchivalDomainDefaults,\n) ArchivalMetadata {\n\thistoryConfig := NewArchivalConfig(\n\t\thistoryStatus,\n\t\tdc.GetStringProperty(dynamicproperties.HistoryArchivalStatus),\n\t\thistoryReadEnabled,\n\t\tdc.GetBoolProperty(dynamicproperties.EnableReadFromHistoryArchival),\n\t\tdomainDefaults.History.Status,\n\t\tdomainDefaults.History.URI,\n\t)\n\n\tvisibilityConfig := NewArchivalConfig(\n\t\tvisibilityStatus,\n\t\tdc.GetStringProperty(dynamicproperties.VisibilityArchivalStatus),\n\t\tvisibilityReadEnabled,\n\t\tdc.GetBoolProperty(dynamicproperties.EnableReadFromVisibilityArchival),\n\t\tdomainDefaults.Visibility.Status,\n\t\tdomainDefaults.Visibility.URI,\n\t)\n\n\treturn &archivalMetadata{\n\t\thistoryConfig:    historyConfig,\n\t\tvisibilityConfig: visibilityConfig,\n\t}\n}\n\nfunc (metadata *archivalMetadata) GetHistoryConfig() ArchivalConfig {\n\treturn metadata.historyConfig\n}\n\nfunc (metadata *archivalMetadata) GetVisibilityConfig() ArchivalConfig {\n\treturn metadata.visibilityConfig\n}\n\n// NewArchivalConfig constructs a new valid ArchivalConfig\nfunc NewArchivalConfig(\n\tstaticClusterStatusStr string,\n\tdynamicClusterStatus dynamicproperties.StringPropertyFn,\n\tstaticEnableRead bool,\n\tdynamicEnableRead dynamicproperties.BoolPropertyFn,\n\tdomainDefaultStatusStr string,\n\tdomainDefaultURI string,\n) ArchivalConfig {\n\tstaticClusterStatus, err := getClusterArchivalStatus(staticClusterStatusStr)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tdomainDefaultStatus, err := getDomainArchivalStatus(domainDefaultStatusStr)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\treturn &archivalConfig{\n\t\tstaticClusterStatus:  staticClusterStatus,\n\t\tdynamicClusterStatus: dynamicClusterStatus,\n\t\tstaticEnableRead:     staticEnableRead,\n\t\tdynamicEnableRead:    dynamicEnableRead,\n\t\tdomainDefaultStatus:  domainDefaultStatus,\n\t\tdomainDefaultURI:     domainDefaultURI,\n\t}\n}\n\n// NewDisabledArchvialConfig returns a disabled ArchivalConfig\nfunc NewDisabledArchvialConfig() ArchivalConfig {\n\treturn &archivalConfig{\n\t\tstaticClusterStatus:  ArchivalDisabled,\n\t\tdynamicClusterStatus: nil,\n\t\tstaticEnableRead:     false,\n\t\tdynamicEnableRead:    nil,\n\t\tdomainDefaultStatus:  types.ArchivalStatusDisabled,\n\t\tdomainDefaultURI:     \"\",\n\t}\n}\n\n// ClusterConfiguredForArchival returns true if cluster is configured to handle archival, false otherwise\nfunc (a *archivalConfig) ClusterConfiguredForArchival() bool {\n\treturn a.GetClusterStatus() == ArchivalEnabled\n}\n\nfunc (a *archivalConfig) GetClusterStatus() ArchivalStatus {\n\t// Only check dynamic config when archival is enabled in static config.\n\t// If archival is disabled in static config, there will be no provider section in the static config\n\t// and the archiver provider can not create any archiver. Therefore, in that case,\n\t// even dynamic config says archival is enabled, we should ignore that.\n\t// Only when archival is enabled in static config, should we check if there's any difference between static config and dynamic config.\n\tif a.staticClusterStatus != ArchivalEnabled {\n\t\treturn a.staticClusterStatus\n\t}\n\n\tdynamicStatusStr := a.dynamicClusterStatus()\n\tdynamicStatus, err := getClusterArchivalStatus(dynamicStatusStr)\n\tif err != nil {\n\t\treturn ArchivalDisabled\n\t}\n\treturn dynamicStatus\n}\n\nfunc (a *archivalConfig) ReadEnabled() bool {\n\tif !a.ClusterConfiguredForArchival() {\n\t\treturn false\n\t}\n\treturn a.staticEnableRead && a.dynamicEnableRead()\n}\n\nfunc (a *archivalConfig) GetDomainDefaultStatus() types.ArchivalStatus {\n\treturn a.domainDefaultStatus\n}\n\nfunc (a *archivalConfig) GetDomainDefaultURI() string {\n\treturn a.domainDefaultURI\n}\n\nfunc getClusterArchivalStatus(str string) (ArchivalStatus, error) {\n\tstr = strings.TrimSpace(strings.ToLower(str))\n\tswitch str {\n\tcase \"\", constants.ArchivalDisabled:\n\t\treturn ArchivalDisabled, nil\n\tcase constants.ArchivalPaused:\n\t\treturn ArchivalPaused, nil\n\tcase constants.ArchivalEnabled:\n\t\treturn ArchivalEnabled, nil\n\t}\n\treturn ArchivalDisabled, fmt.Errorf(\"invalid archival status of %v for cluster, valid status are: {\\\"\\\", \\\"disabled\\\", \\\"paused\\\", \\\"enabled\\\"}\", str)\n}\n\nfunc getDomainArchivalStatus(str string) (types.ArchivalStatus, error) {\n\tstr = strings.TrimSpace(strings.ToLower(str))\n\tswitch str {\n\tcase \"\", constants.ArchivalDisabled:\n\t\treturn types.ArchivalStatusDisabled, nil\n\tcase constants.ArchivalEnabled:\n\t\treturn types.ArchivalStatusEnabled, nil\n\t}\n\treturn types.ArchivalStatusDisabled, fmt.Errorf(\"invalid archival status of %v for domain, valid status are: {\\\"\\\", \\\"disabled\\\", \\\"enabled\\\"}\", str)\n}\n"
  },
  {
    "path": "common/archiver/archivalMetadata_mock.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport \"github.com/stretchr/testify/mock\"\n\n// MockArchivalMetadata is an autogenerated mock type for the ArchivalMetadata type\ntype MockArchivalMetadata struct {\n\tmock.Mock\n}\n\n// GetHistoryConfig provides a mock function with given fields:\nfunc (_m *MockArchivalMetadata) GetHistoryConfig() ArchivalConfig {\n\tret := _m.Called()\n\n\tvar r0 ArchivalConfig\n\tif rf, ok := ret.Get(0).(func() ArchivalConfig); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(ArchivalConfig)\n\t\t}\n\t}\n\n\treturn r0\n}\n\n// GetVisibilityConfig provides a mock function with given fields:\nfunc (_m *MockArchivalMetadata) GetVisibilityConfig() ArchivalConfig {\n\tret := _m.Called()\n\n\tvar r0 ArchivalConfig\n\tif rf, ok := ret.Get(0).(func() ArchivalConfig); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(ArchivalConfig)\n\t\t}\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/constants.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport \"errors\"\n\nconst (\n\t// ArchiveNonRetriableErrorMsg is the log message when the Archive() method encounters a non-retriable error\n\tArchiveNonRetriableErrorMsg = \"Archive method encountered an non-retriable error.\"\n\t// ArchiveTransientErrorMsg is the log message when the Archive() method encounters a transient error\n\tArchiveTransientErrorMsg = \"Archive method encountered a transient error.\"\n\t// ArchiveSkippedInfoMsg is the log messsage when the Archive() method encounter an entity not exists error\n\tArchiveSkippedInfoMsg = \"Archive method encountered entity not exists error and skipped the archival\"\n\n\t// ErrReasonInvalidURI is the error reason for invalid URI\n\tErrReasonInvalidURI = \"URI is invalid\"\n\t// ErrReasonInvalidArchiveRequest is the error reason for invalid archive request\n\tErrReasonInvalidArchiveRequest = \"archive request is invalid\"\n\t// ErrReasonConstructHistoryIterator is the error reason for failing to construct history iterator\n\tErrReasonConstructHistoryIterator = \"failed to construct history iterator\"\n\t// ErrReasonReadHistory is the error reason for failing to read history\n\tErrReasonReadHistory = \"failed to read history batches\"\n\t// ErrReasonHistoryMutated is the error reason for mutated history\n\tErrReasonHistoryMutated = \"history was mutated\"\n)\n\nvar (\n\t// ErrInvalidURI is the error for invalid URI\n\tErrInvalidURI = errors.New(\"URI is invalid\")\n\t// ErrURISchemeMismatch is the error for mismatch between URI scheme and archiver\n\tErrURISchemeMismatch = errors.New(\"URI scheme does not match the archiver\")\n\t// ErrHistoryMutated is the error for mutated history\n\tErrHistoryMutated = errors.New(\"history was mutated\")\n\t// ErrContextTimeout is the error for context timeout\n\tErrContextTimeout = errors.New(\"archive aborted because context timed out\")\n\t// ErrInvalidGetHistoryRequest is the error for invalid GetHistory request\n\tErrInvalidGetHistoryRequest = errors.New(\"get archived history request is invalid\")\n\t// ErrInvalidQueryVisibilityRequest is the error for invalid Query Visibility request\n\tErrInvalidQueryVisibilityRequest = errors.New(\"query visiblity request is invalid\")\n\t// ErrNextPageTokenCorrupted is the error for corrupted GetHistory token\n\tErrNextPageTokenCorrupted = errors.New(\"next page token is corrupted\")\n\t// ErrHistoryNotExist is the error for non-exist history\n\tErrHistoryNotExist = errors.New(\"requested workflow history does not exist\")\n)\n"
  },
  {
    "path": "common/archiver/filestore/historyArchiver.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Filestore History Archiver will archive workflow histories to local disk.\n\n// Each Archive() request results in a file named in the format of\n// hash(domainID, workflowID, runID)_version.history being created in the specified\n// directory. Workflow histories stored in that file are encoded in JSON format.\n\n// The Get() method retrieves the archived histories from the directory specified in the\n// URI. It optionally takes in a NextPageToken which specifies the workflow close failover\n// version and the index of the first history batch that should be returned. Instead of\n// NextPageToken, caller can also provide a close failover version, in which case, Get() method\n// will return history batches starting from the beginning of that history version. If neither\n// of NextPageToken or close failover version is specified, the highest close failover version\n// will be picked.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"os\"\n\t\"path\"\n\t\"strconv\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nconst (\n\t// URIScheme is the scheme for the filestore implementation\n\tURIScheme = \"file\"\n\n\terrEncodeHistory = \"failed to encode history batches\"\n\terrMakeDirectory = \"failed to make directory\"\n\terrWriteFile     = \"failed to write history to file\"\n\n\ttargetHistoryBlobSize = 2 * 1024 * 1024 // 2MB\n)\n\nvar (\n\terrInvalidFileMode = errors.New(\"invalid file mode\")\n\terrInvalidDirMode  = errors.New(\"invalid directory mode\")\n)\n\ntype (\n\thistoryArchiver struct {\n\t\tcontainer *archiver.HistoryBootstrapContainer\n\t\tfileMode  os.FileMode\n\t\tdirMode   os.FileMode\n\n\t\t// only set in test code\n\t\thistoryIterator archiver.HistoryIterator\n\t}\n\n\tgetHistoryToken struct {\n\t\tCloseFailoverVersion int64\n\t\tNextBatchIdx         int\n\t}\n)\n\n// NewHistoryArchiver creates a new archiver.HistoryArchiver based on filestore\nfunc NewHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\tconfig *config.FilestoreArchiver,\n) (archiver.HistoryArchiver, error) {\n\treturn newHistoryArchiver(container, config, nil)\n}\n\nfunc newHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\tconfig *config.FilestoreArchiver,\n\thistoryIterator archiver.HistoryIterator,\n) (*historyArchiver, error) {\n\tfileMode, err := strconv.ParseUint(config.FileMode, 0, 32)\n\tif err != nil {\n\t\treturn nil, errInvalidFileMode\n\t}\n\tdirMode, err := strconv.ParseUint(config.DirMode, 0, 32)\n\tif err != nil {\n\t\treturn nil, errInvalidDirMode\n\t}\n\treturn &historyArchiver{\n\t\tcontainer:       container,\n\t\tfileMode:        os.FileMode(fileMode),\n\t\tdirMode:         os.FileMode(dirMode),\n\t\thistoryIterator: historyIterator,\n\t}, nil\n}\n\nfunc (h *historyArchiver) Archive(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.ArchiveHistoryRequest,\n\topts ...archiver.ArchiveOption,\n) (err error) {\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tdefer func() {\n\t\tif err != nil && !persistence.IsTransientError(err) && featureCatalog.NonRetriableError != nil {\n\t\t\terr = featureCatalog.NonRetriableError()\n\t\t}\n\t}()\n\n\tlogger := archiver.TagLoggerWithArchiveHistoryRequestAndURI(h.container.Logger, request, URI.String())\n\n\tif err := h.ValidateURI(URI); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\treturn err\n\t}\n\n\tif err := archiver.ValidateHistoryArchiveRequest(request); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidArchiveRequest), tag.Error(err))\n\t\treturn err\n\t}\n\n\thistoryIterator := h.historyIterator\n\tif historyIterator == nil { // will only be set by testing code\n\t\thistoryIterator = archiver.NewHistoryIterator(ctx, request, h.container.HistoryV2Manager, targetHistoryBlobSize)\n\t}\n\n\thistoryBatches := []*types.History{}\n\tfor historyIterator.HasNext() {\n\t\thistoryBlob, err := getNextHistoryBlob(ctx, historyIterator)\n\t\tif err != nil {\n\t\t\tif common.IsEntityNotExistsError(err) {\n\t\t\t\t// workflow history no longer exists, may due to duplicated archival signal\n\t\t\t\t// this may happen even in the middle of iterating history as two archival signals\n\t\t\t\t// can be processed concurrently.\n\t\t\t\tlogger.Info(archiver.ArchiveSkippedInfoMsg)\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tlogger := logger.WithTags(tag.ArchivalArchiveFailReason(archiver.ErrReasonReadHistory), tag.Error(err))\n\t\t\tif !persistence.IsTransientError(err) {\n\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg)\n\t\t\t} else {\n\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg)\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\tif archiver.IsHistoryMutated(request, historyBlob.Body, *historyBlob.Header.IsLast, logger) {\n\t\t\tif !featureCatalog.ArchiveIncompleteHistory() {\n\t\t\t\treturn archiver.ErrHistoryMutated\n\t\t\t}\n\t\t}\n\n\t\thistoryBatches = append(historyBatches, historyBlob.Body...)\n\t}\n\n\tencodedHistoryBatches, err := encode(historyBatches)\n\tif err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errEncodeHistory), tag.Error(err))\n\t\treturn err\n\t}\n\n\tdirPath := URI.Path()\n\tif err = util.MkdirAll(dirPath, h.dirMode); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errMakeDirectory), tag.Error(err))\n\t\treturn err\n\t}\n\n\tfilename := constructHistoryFilename(request.DomainID, request.WorkflowID, request.RunID, request.CloseFailoverVersion)\n\tif err := util.WriteFile(path.Join(dirPath, filename), encodedHistoryBatches, h.fileMode); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errWriteFile), tag.Error(err))\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc (h *historyArchiver) Get(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.GetHistoryRequest,\n) (*archiver.GetHistoryResponse, error) {\n\tif err := h.ValidateURI(URI); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateGetRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidGetHistoryRequest.Error()}\n\t}\n\n\tdirPath := URI.Path()\n\texists, err := util.DirectoryExists(dirPath)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\tif !exists {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrHistoryNotExist.Error()}\n\t}\n\n\tvar token *getHistoryToken\n\tif request.NextPageToken != nil {\n\t\ttoken, err = deserializeGetHistoryToken(request.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: archiver.ErrNextPageTokenCorrupted.Error()}\n\t\t}\n\t} else if request.CloseFailoverVersion != nil {\n\t\ttoken = &getHistoryToken{\n\t\t\tCloseFailoverVersion: *request.CloseFailoverVersion,\n\t\t\tNextBatchIdx:         0,\n\t\t}\n\t} else {\n\t\thighestVersion, err := getHighestVersion(dirPath, request)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\ttoken = &getHistoryToken{\n\t\t\tCloseFailoverVersion: *highestVersion,\n\t\t\tNextBatchIdx:         0,\n\t\t}\n\t}\n\n\tfilename := constructHistoryFilename(request.DomainID, request.WorkflowID, request.RunID, token.CloseFailoverVersion)\n\tfilepath := path.Join(dirPath, filename)\n\texists, err = util.FileExists(filepath)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\tif !exists {\n\t\treturn nil, &types.EntityNotExistsError{Message: archiver.ErrHistoryNotExist.Error()}\n\t}\n\n\tencodedHistoryBatches, err := util.ReadFile(filepath)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\n\thistoryBatches, err := decodeHistoryBatches(encodedHistoryBatches)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\thistoryBatches = historyBatches[token.NextBatchIdx:]\n\n\tresponse := &archiver.GetHistoryResponse{}\n\tnumOfEvents := 0\n\tnumOfBatches := 0\n\tfor _, batch := range historyBatches {\n\t\tresponse.HistoryBatches = append(response.HistoryBatches, batch)\n\t\tnumOfBatches++\n\t\tnumOfEvents += len(batch.Events)\n\t\tif numOfEvents >= request.PageSize {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif numOfBatches < len(historyBatches) {\n\t\ttoken.NextBatchIdx += numOfBatches\n\t\tnextToken, err := serializeToken(token)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\tresponse.NextPageToken = nextToken\n\t}\n\n\treturn response, nil\n}\n\nfunc (h *historyArchiver) ValidateURI(URI archiver.URI) error {\n\tif URI.Scheme() != URIScheme {\n\t\treturn archiver.ErrURISchemeMismatch\n\t}\n\n\treturn validateDirPath(URI.Path())\n}\n\nfunc getNextHistoryBlob(ctx context.Context, historyIterator archiver.HistoryIterator) (*archiver.HistoryBlob, error) {\n\thistoryBlob, err := historyIterator.Next()\n\top := func(ctx context.Context) error {\n\t\thistoryBlob, err = historyIterator.Next()\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreatePersistenceRetryPolicy()),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\tfor err != nil {\n\t\tif contextExpired(ctx) {\n\t\t\treturn nil, archiver.ErrContextTimeout\n\t\t}\n\t\tif !persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\terr = throttleRetry.Do(ctx, op)\n\t}\n\treturn historyBlob, nil\n}\n\nfunc getHighestVersion(dirPath string, request *archiver.GetHistoryRequest) (*int64, error) {\n\tfilenames, err := util.ListFilesByPrefix(dirPath, constructHistoryFilenamePrefix(request.DomainID, request.WorkflowID, request.RunID))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar highestVersion *int64\n\tfor _, filename := range filenames {\n\t\tversion, err := extractCloseFailoverVersion(filename)\n\t\tif err != nil {\n\t\t\tcontinue\n\t\t}\n\t\tif highestVersion == nil || version > *highestVersion {\n\t\t\thighestVersion = &version\n\t\t}\n\t}\n\tif highestVersion == nil {\n\t\treturn nil, archiver.ErrHistoryNotExist\n\t}\n\treturn highestVersion, nil\n}\n"
  },
  {
    "path": "common/archiver/filestore/historyArchiver_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"path\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nconst (\n\ttestDomainID             = \"test-domain-id\"\n\ttestDomainName           = \"test-domain-name\"\n\ttestWorkflowID           = \"test-workflow-id\"\n\ttestRunID                = \"test-run-id\"\n\ttestNextEventID          = 1800\n\ttestCloseFailoverVersion = 100\n\ttestPageSize             = 100\n\n\ttestFileModeStr = \"0666\"\n\ttestDirModeStr  = \"0766\"\n)\n\nvar (\n\ttestBranchToken = []byte{1, 2, 3}\n)\n\ntype historyArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tcontainer          *archiver.HistoryBootstrapContainer\n\ttestArchivalURI    archiver.URI\n\ttestGetDirectory   string\n\thistoryBatchesV1   []*types.History\n\thistoryBatchesV100 []*types.History\n}\n\nfunc TestHistoryArchiverSuite(t *testing.T) {\n\tsuite.Run(t, new(historyArchiverSuite))\n}\n\nfunc (s *historyArchiverSuite) SetupSuite() {\n\tvar err error\n\ts.testGetDirectory, err = ioutil.TempDir(\"\", \"TestGet\")\n\ts.Require().NoError(err)\n\ts.setupHistoryDirectory()\n\ts.testArchivalURI, err = archiver.NewURI(\"file:///a/b/c\")\n\ts.Require().NoError(err)\n}\n\nfunc (s *historyArchiverSuite) TearDownSuite() {\n\tos.RemoveAll(s.testGetDirectory)\n}\n\nfunc (s *historyArchiverSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.container = &archiver.HistoryBootstrapContainer{\n\t\tLogger: testlogger.New(s.T()),\n\t}\n}\n\nfunc (s *historyArchiverSuite) TestValidateURI() {\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"file://\",\n\t\t\texpectedErr: errEmptyDirectoryPath,\n\t\t},\n\t\t{\n\t\t\tURI:         \"file:///a/b/c\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedErr, historyArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_InvalidURI() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_InvalidRequest() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           \"\", // an invalid request\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_ErrorOnReadHistory() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"some random error\")),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_TimeoutWhenReadingHistory() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.ServiceBusyError{}),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(getCanceledContext(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_HistoryMutated() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion + 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_NonRetriableErrorOption() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"some random error\")),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tnonRetryableErr := errors.New(\"some non-retryable error\")\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request, archiver.GetNonRetriableErrorOption(nonRetryableErr))\n\ts.Equal(nonRetryableErr, err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Skip() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(false),\n\t\t},\n\t\tBody: []*types.History{\n\t\t\t{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:        constants.FirstEventID,\n\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.EntityNotExistsError{Message: \"workflow not found\"}),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.NoError(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Success() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 2,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(false),\n\t)\n\n\tdir, err := ioutil.TempDir(\"\", \"TestArchiveSingleRead\")\n\ts.NoError(err)\n\tdefer os.RemoveAll(dir)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + dir)\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\ts.NoError(err)\n\n\texpectedFilename := constructHistoryFilename(testDomainID, testWorkflowID, testRunID, testCloseFailoverVersion)\n\ts.assertFileExists(path.Join(dir, expectedFilename))\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidURI() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   100,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidRequest() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   0, // pageSize should be greater than 0\n\t}\n\tresponse, err := historyArchiver.Get(context.Background(), s.testArchivalURI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_DirectoryNotExist() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\tresponse, err := historyArchiver.Get(context.Background(), s.testArchivalURI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidToken() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:      testDomainID,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tPageSize:      testPageSize,\n\t\tNextPageToken: []byte{'r', 'a', 'n', 'd', 'o', 'm'},\n\t}\n\tURI, err := archiver.NewURI(\"file:///\")\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_FileNotExist() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             testPageSize,\n\t\tCloseFailoverVersion: common.Int64Ptr(testCloseFailoverVersion),\n\t}\n\tURI, err := archiver.NewURI(\"file:///\")\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_PickHighestVersion() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + s.testGetDirectory)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(s.historyBatchesV100, response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_UseProvidedVersion() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             testPageSize,\n\t\tCloseFailoverVersion: common.Int64Ptr(1),\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + s.testGetDirectory)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(s.historyBatchesV1, response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_SmallPageSize() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             1,\n\t\tCloseFailoverVersion: common.Int64Ptr(100),\n\t}\n\tcombinedHistory := []*types.History{}\n\n\tURI, err := archiver.NewURI(\"file://\" + s.testGetDirectory)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.NotNil(response.NextPageToken)\n\ts.NotNil(response.HistoryBatches)\n\ts.Len(response.HistoryBatches, 1)\n\tcombinedHistory = append(combinedHistory, response.HistoryBatches...)\n\n\trequest.NextPageToken = response.NextPageToken\n\tresponse, err = historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.NotNil(response.HistoryBatches)\n\ts.Len(response.HistoryBatches, 1)\n\tcombinedHistory = append(combinedHistory, response.HistoryBatches...)\n\n\ts.Equal(s.historyBatchesV100, combinedHistory)\n}\n\nfunc (s *historyArchiverSuite) TestArchiveAndGet() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: s.historyBatchesV100,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(false),\n\t)\n\n\tdir, err := ioutil.TempDir(\"\", \"TestArchiveAndGet\")\n\ts.NoError(err)\n\tdefer os.RemoveAll(dir)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\tarchiveRequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + dir)\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, archiveRequest)\n\ts.NoError(err)\n\n\texpectedFilename := constructHistoryFilename(testDomainID, testWorkflowID, testRunID, testCloseFailoverVersion)\n\ts.assertFileExists(path.Join(dir, expectedFilename))\n\n\tgetRequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\tresponse, err := historyArchiver.Get(context.Background(), URI, getRequest)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(s.historyBatchesV100, response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) newTestHistoryArchiver(historyIterator archiver.HistoryIterator) *historyArchiver {\n\tconfig := &config.FilestoreArchiver{\n\t\tFileMode: testFileModeStr,\n\t\tDirMode:  testDirModeStr,\n\t}\n\tarchiver, err := newHistoryArchiver(s.container, config, historyIterator)\n\ts.NoError(err)\n\treturn archiver\n}\n\nfunc (s *historyArchiverSuite) setupHistoryDirectory() {\n\ts.historyBatchesV1 = []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.historyBatchesV100 = []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.writeHistoryBatchesForGetTest(s.historyBatchesV1, int64(1))\n\ts.writeHistoryBatchesForGetTest(s.historyBatchesV100, testCloseFailoverVersion)\n}\n\nfunc (s *historyArchiverSuite) writeHistoryBatchesForGetTest(historyBatches []*types.History, version int64) {\n\tdata, err := encode(historyBatches)\n\ts.Require().NoError(err)\n\tfilename := constructHistoryFilename(testDomainID, testWorkflowID, testRunID, version)\n\terr = util.WriteFile(path.Join(s.testGetDirectory, filename), data, testFileMode)\n\ts.Require().NoError(err)\n}\n\nfunc (s *historyArchiverSuite) assertFileExists(filepath string) {\n\texists, err := util.FileExists(filepath)\n\ts.NoError(err)\n\ts.True(exists)\n}\n\nfunc getCanceledContext() context.Context {\n\tctx, cancel := context.WithCancel(context.Background())\n\tcancel()\n\treturn ctx\n}\n"
  },
  {
    "path": "common/archiver/filestore/queryParser.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source queryParser.go -destination queryParser_mock.go -mock_names Interface=MockQueryParser\n\npackage filestore\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// QueryParser parses a limited SQL where clause into a struct\n\tQueryParser interface {\n\t\tParse(query string) (*parsedQuery, error)\n\t}\n\n\tqueryParser struct{}\n\n\tparsedQuery struct {\n\t\tearliestCloseTime int64\n\t\tlatestCloseTime   int64\n\t\tworkflowID        *string\n\t\trunID             *string\n\t\tworkflowTypeName  *string\n\t\tcloseStatus       *types.WorkflowExecutionCloseStatus\n\t\temptyResult       bool\n\t}\n)\n\n// All allowed fields for filtering\nconst (\n\tWorkflowID   = \"WorkflowID\"\n\tRunID        = \"RunID\"\n\tWorkflowType = \"WorkflowType\"\n\tCloseTime    = \"CloseTime\"\n\tCloseStatus  = \"CloseStatus\"\n)\n\nconst (\n\tqueryTemplate = \"select * from dummy where %s\"\n\n\tdefaultDateTimeFormat = time.RFC3339\n)\n\n// NewQueryParser creates a new query parser for filestore\nfunc NewQueryParser() QueryParser {\n\treturn &queryParser{}\n}\n\nfunc (p *queryParser) Parse(query string) (*parsedQuery, error) {\n\tstmt, err := sqlparser.Parse(fmt.Sprintf(queryTemplate, query))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\twhereExpr := stmt.(*sqlparser.Select).Where.Expr\n\tparsedQuery := &parsedQuery{\n\t\tearliestCloseTime: 0,\n\t\tlatestCloseTime:   time.Now().UnixNano(),\n\t}\n\tif err := p.convertWhereExpr(whereExpr, parsedQuery); err != nil {\n\t\treturn nil, err\n\t}\n\treturn parsedQuery, nil\n}\n\nfunc (p *queryParser) convertWhereExpr(expr sqlparser.Expr, parsedQuery *parsedQuery) error {\n\tif expr == nil {\n\t\treturn errors.New(\"where expression is nil\")\n\t}\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn p.convertComparisonExpr(expr, parsedQuery)\n\tcase *sqlparser.AndExpr:\n\t\treturn p.convertAndExpr(expr, parsedQuery)\n\tcase *sqlparser.ParenExpr:\n\t\treturn p.convertParenExpr(expr, parsedQuery)\n\tdefault:\n\t\treturn errors.New(\"only comparison and \\\"and\\\" expression is supported\")\n\t}\n}\n\nfunc (p *queryParser) convertParenExpr(parenExpr *sqlparser.ParenExpr, parsedQuery *parsedQuery) error {\n\treturn p.convertWhereExpr(parenExpr.Expr, parsedQuery)\n}\n\nfunc (p *queryParser) convertAndExpr(andExpr *sqlparser.AndExpr, parsedQuery *parsedQuery) error {\n\tif err := p.convertWhereExpr(andExpr.Left, parsedQuery); err != nil {\n\t\treturn err\n\t}\n\treturn p.convertWhereExpr(andExpr.Right, parsedQuery)\n}\n\nfunc (p *queryParser) convertComparisonExpr(compExpr *sqlparser.ComparisonExpr, parsedQuery *parsedQuery) error {\n\tcolName, ok := compExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid filter name: %s\", sqlparser.String(compExpr.Left))\n\t}\n\tcolNameStr := sqlparser.String(colName)\n\top := compExpr.Operator\n\tvalExpr, ok := compExpr.Right.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid value: %s\", sqlparser.String(compExpr.Right))\n\t}\n\tvalStr := sqlparser.String(valExpr)\n\n\tswitch colNameStr {\n\tcase WorkflowID:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with file system\", WorkflowID)\n\t\t}\n\t\tif parsedQuery.workflowID != nil && *parsedQuery.workflowID != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.workflowID = common.StringPtr(val)\n\tcase RunID:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with file system\", RunID)\n\t\t}\n\t\tif parsedQuery.runID != nil && *parsedQuery.runID != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.runID = common.StringPtr(val)\n\tcase WorkflowType:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with file system\", WorkflowType)\n\t\t}\n\t\tif parsedQuery.workflowTypeName != nil && *parsedQuery.workflowTypeName != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.workflowTypeName = common.StringPtr(val)\n\tcase CloseStatus:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\t// if failed to extract string value, it means user input close status as a number\n\t\t\tval = valStr\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with file system\", CloseStatus)\n\t\t}\n\t\tstatus, err := convertStatusStr(val)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif parsedQuery.closeStatus != nil && *parsedQuery.closeStatus != status {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.closeStatus = status.Ptr()\n\tcase CloseTime:\n\t\ttimestamp, err := convertToTimestamp(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn p.convertCloseTime(timestamp, op, parsedQuery)\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown filter name: %s\", colNameStr)\n\t}\n\n\treturn nil\n}\n\nfunc (p *queryParser) convertCloseTime(timestamp int64, op string, parsedQuery *parsedQuery) error {\n\tswitch op {\n\tcase \"=\":\n\t\tif err := p.convertCloseTime(timestamp, \">=\", parsedQuery); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := p.convertCloseTime(timestamp, \"<=\", parsedQuery); err != nil {\n\t\t\treturn err\n\t\t}\n\tcase \"<\":\n\t\tparsedQuery.latestCloseTime = min(parsedQuery.latestCloseTime, timestamp-1)\n\tcase \"<=\":\n\t\tparsedQuery.latestCloseTime = min(parsedQuery.latestCloseTime, timestamp)\n\tcase \">\":\n\t\tparsedQuery.earliestCloseTime = max(parsedQuery.earliestCloseTime, timestamp+1)\n\tcase \">=\":\n\t\tparsedQuery.earliestCloseTime = max(parsedQuery.earliestCloseTime, timestamp)\n\tdefault:\n\t\treturn fmt.Errorf(\"operator %s is not supported for close time\", op)\n\t}\n\treturn nil\n}\n\nfunc convertToTimestamp(timeStr string) (int64, error) {\n\ttimestamp, err := strconv.ParseInt(timeStr, 10, 64)\n\tif err == nil {\n\t\treturn timestamp, nil\n\t}\n\ttimestampStr, err := extractStringValue(timeStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tparsedTime, err := time.Parse(defaultDateTimeFormat, timestampStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn parsedTime.UnixNano(), nil\n}\n\nfunc convertStatusStr(statusStr string) (types.WorkflowExecutionCloseStatus, error) {\n\tstatusStr = strings.ToLower(strings.TrimSpace(statusStr))\n\tswitch statusStr {\n\tcase \"completed\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusCompleted)):\n\t\treturn types.WorkflowExecutionCloseStatusCompleted, nil\n\tcase \"failed\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusFailed)):\n\t\treturn types.WorkflowExecutionCloseStatusFailed, nil\n\tcase \"canceled\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusCanceled)):\n\t\treturn types.WorkflowExecutionCloseStatusCanceled, nil\n\tcase \"terminated\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusTerminated)):\n\t\treturn types.WorkflowExecutionCloseStatusTerminated, nil\n\tcase \"continuedasnew\", \"continued_as_new\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusContinuedAsNew)):\n\t\treturn types.WorkflowExecutionCloseStatusContinuedAsNew, nil\n\tcase \"timedout\", \"timed_out\", strconv.Itoa(int(types.WorkflowExecutionCloseStatusTimedOut)):\n\t\treturn types.WorkflowExecutionCloseStatusTimedOut, nil\n\tdefault:\n\t\treturn 0, fmt.Errorf(\"unknown workflow close status: %s\", statusStr)\n\t}\n}\n\nfunc extractStringValue(s string) (string, error) {\n\tif len(s) >= 2 && s[0] == '\\'' && s[len(s)-1] == '\\'' {\n\t\treturn s[1 : len(s)-1], nil\n\t}\n\treturn \"\", fmt.Errorf(\"value %s is not a string value\", s)\n}\n"
  },
  {
    "path": "common/archiver/filestore/queryParser_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: queryParser.go\n//\n// Generated by this command:\n//\n//\tmockgen -package filestore -source queryParser.go -destination queryParser_mock.go -mock_names Interface=MockQueryParser\n//\n\n// Package filestore is a generated GoMock package.\npackage filestore\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockQueryParser is a mock of QueryParser interface.\ntype MockQueryParser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueryParserMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueryParserMockRecorder is the mock recorder for MockQueryParser.\ntype MockQueryParserMockRecorder struct {\n\tmock *MockQueryParser\n}\n\n// NewMockQueryParser creates a new mock instance.\nfunc NewMockQueryParser(ctrl *gomock.Controller) *MockQueryParser {\n\tmock := &MockQueryParser{ctrl: ctrl}\n\tmock.recorder = &MockQueryParserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueryParser) EXPECT() *MockQueryParserMockRecorder {\n\treturn m.recorder\n}\n\n// Parse mocks base method.\nfunc (m *MockQueryParser) Parse(query string) (*parsedQuery, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Parse\", query)\n\tret0, _ := ret[0].(*parsedQuery)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Parse indicates an expected call of Parse.\nfunc (mr *MockQueryParserMockRecorder) Parse(query any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Parse\", reflect.TypeOf((*MockQueryParser)(nil).Parse), query)\n}\n"
  },
  {
    "path": "common/archiver/filestore/queryParser_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype queryParserSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tparser QueryParser\n}\n\nfunc TestQueryParserSuite(t *testing.T) {\n\tsuite.Run(t, new(queryParserSuite))\n}\n\nfunc (s *queryParserSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.parser = NewQueryParser()\n}\n\nfunc (s *queryParserSuite) TestParseWorkflowID_RunID_WorkflowType() {\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" and WorkflowID = \\\"random workflowID\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"RunID = \\\"random runID\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\trunID: common.StringPtr(\"random runID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowType = \\\"random typeName\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowTypeName: common.StringPtr(\"random typeName\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = 'random workflowID'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowType = 'random typeName' and WorkflowType = \\\"another typeName\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\temptyResult: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowType = 'random typeName' and (WorkflowID = \\\"random workflowID\\\" and RunID='random runID')\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID:       common.StringPtr(\"random workflowID\"),\n\t\t\t\trunID:            common.StringPtr(\"random runID\"),\n\t\t\t\tworkflowTypeName: common.StringPtr(\"random typeName\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"runID = random workflowID\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" or WorkflowID = \\\"another workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" or runID = \\\"random runID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"workflowid = \\\"random workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"runID > \\\"random workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.emptyResult, parsedQuery.emptyResult)\n\t\tif !tc.parsedQuery.emptyResult {\n\t\t\ts.Equal(tc.parsedQuery.workflowID, parsedQuery.workflowID)\n\t\t\ts.Equal(tc.parsedQuery.runID, parsedQuery.runID)\n\t\t\ts.Equal(tc.parsedQuery.workflowTypeName, parsedQuery.workflowTypeName)\n\t\t}\n\t}\n}\n\nfunc (s *queryParserSuite) TestParseCloseStatus() {\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     \"CloseStatus = \\\"Completed\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseStatus: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = 'continuedasnew'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseStatus: types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = 'TIMED_OUT'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseStatus: types.WorkflowExecutionCloseStatusTimedOut.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = 'Failed' and CloseStatus = \\\"Failed\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseStatus: types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"(CloseStatus = 'Timedout' and CloseStatus = \\\"canceled\\\")\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\temptyResult: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"closeStatus = \\\"Failed\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = \\\"Failed\\\" or CloseStatus = \\\"Failed\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = \\\"unknown\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus > \\\"Failed\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = 1\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseStatus: types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus = 10\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.emptyResult, parsedQuery.emptyResult)\n\t\tif !tc.parsedQuery.emptyResult {\n\t\t\ts.Equal(tc.parsedQuery.closeStatus, parsedQuery.closeStatus)\n\t\t}\n\t}\n}\n\nfunc (s *queryParserSuite) TestParseCloseTime() {\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     \"CloseTime <= 1000\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 0,\n\t\t\t\tlatestCloseTime:   1000,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime < 2000 and CloseTime <= 1000 and CloseTime > 300\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 301,\n\t\t\t\tlatestCloseTime:   1000,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime = 2000 and (CloseTime > 1000 and CloseTime <= 9999)\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 2000,\n\t\t\t\tlatestCloseTime:   2000,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime <= \\\"2019-01-01T11:11:11Z\\\" and CloseTime >= 1000000\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 1000000,\n\t\t\t\tlatestCloseTime:   1546341071000000000,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"closeTime = 2000\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime > \\\"2019-01-01 00:00:00\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseStatus > 2000 or CloseStatus < 1000\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.emptyResult, parsedQuery.emptyResult)\n\t\tif !tc.parsedQuery.emptyResult {\n\t\t\ts.Equal(tc.parsedQuery.earliestCloseTime, parsedQuery.earliestCloseTime)\n\t\t\ts.Equal(tc.parsedQuery.latestCloseTime, parsedQuery.latestCloseTime)\n\t\t}\n\t}\n}\n\nfunc (s *queryParserSuite) TestParse() {\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     \"CloseTime <= \\\"2019-01-01T11:11:11Z\\\" and WorkflowID = 'random workflowID'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 0,\n\t\t\t\tlatestCloseTime:   1546341071000000000,\n\t\t\t\tworkflowID:        common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime > 1999 and CloseTime < 10000 and RunID = 'random runID' and CloseStatus = 'Failed'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tearliestCloseTime: 2000,\n\t\t\t\tlatestCloseTime:   9999,\n\t\t\t\trunID:             common.StringPtr(\"random runID\"),\n\t\t\t\tcloseStatus:       types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"CloseTime > 2001 and CloseTime < 10000 and (RunID = 'random runID') and CloseStatus = 'Failed' and (RunID = 'another ID')\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\temptyResult: true,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.emptyResult, parsedQuery.emptyResult)\n\t\tif !tc.parsedQuery.emptyResult {\n\t\t\ts.Equal(tc.parsedQuery, parsedQuery)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/archiver/filestore/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"os\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/dgryski/go-farm\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nvar (\n\terrEmptyDirectoryPath = errors.New(\"directory path is empty\")\n)\n\n// encoding & decoding util\n\nfunc encode(v interface{}) ([]byte, error) {\n\treturn json.Marshal(v)\n}\n\nfunc decodeHistoryBatches(data []byte) ([]*types.History, error) {\n\thistoryBatches := []*types.History{}\n\terr := json.Unmarshal(data, &historyBatches)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn historyBatches, nil\n}\n\nfunc decodeVisibilityRecord(data []byte) (*visibilityRecord, error) {\n\trecord := &visibilityRecord{}\n\terr := json.Unmarshal(data, record)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn record, nil\n}\n\nfunc serializeToken(token interface{}) ([]byte, error) {\n\tif token == nil {\n\t\treturn nil, nil\n\t}\n\treturn json.Marshal(token)\n}\n\nfunc deserializeGetHistoryToken(bytes []byte) (*getHistoryToken, error) {\n\ttoken := &getHistoryToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc deserializeQueryVisibilityToken(bytes []byte) (*queryVisibilityToken, error) {\n\ttoken := &queryVisibilityToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\n// File name construction\n\nfunc constructHistoryFilename(domainID, workflowID, runID string, version int64) string {\n\tcombinedHash := constructHistoryFilenamePrefix(domainID, workflowID, runID)\n\treturn fmt.Sprintf(\"%s_%v.history\", combinedHash, version)\n}\n\nfunc constructHistoryFilenamePrefix(domainID, workflowID, runID string) string {\n\treturn strings.Join([]string{hash(domainID), hash(workflowID), hash(runID)}, \"\")\n}\n\nfunc constructVisibilityFilename(closeTimestamp int64, runID string) string {\n\treturn fmt.Sprintf(\"%v_%s.visibility\", closeTimestamp, hash(runID))\n}\n\nfunc hash(s string) string {\n\treturn fmt.Sprintf(\"%v\", farm.Fingerprint64([]byte(s)))\n}\n\n// Validation\n\nfunc validateDirPath(dirPath string) error {\n\tif len(dirPath) == 0 {\n\t\treturn errEmptyDirectoryPath\n\t}\n\tinfo, err := os.Stat(dirPath)\n\tif os.IsNotExist(err) {\n\t\treturn nil\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !info.IsDir() {\n\t\treturn util.ErrDirectoryExpected\n\t}\n\treturn nil\n}\n\n// Misc.\n\nfunc extractCloseFailoverVersion(filename string) (int64, error) {\n\tfilenameParts := strings.FieldsFunc(filename, func(r rune) bool {\n\t\treturn r == '_' || r == '.'\n\t})\n\tif len(filenameParts) != 3 {\n\t\treturn -1, errors.New(\"unknown filename structure\")\n\t}\n\treturn strconv.ParseInt(filenameParts[1], 10, 64)\n}\n\nfunc contextExpired(ctx context.Context) bool {\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "common/archiver/filestore/util_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"io/ioutil\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nconst (\n\ttestFileMode = os.FileMode(0600)\n\ttestDirMode  = os.FileMode(0700)\n)\n\ntype UtilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(UtilSuite))\n}\n\nfunc (s *UtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *UtilSuite) TestEncodeDecodeHistoryBatches() {\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:      constants.FirstEventID,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:      constants.FirstEventID + 2,\n\t\t\t\t\tVersion: 2,\n\t\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\t\tIdentity: \"some random identity\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tencodedHistoryBatches, err := encode(historyBatches)\n\ts.NoError(err)\n\n\tdecodedHistoryBatches, err := decodeHistoryBatches(encodedHistoryBatches)\n\ts.NoError(err)\n\ts.Equal(historyBatches, decodedHistoryBatches)\n}\n\nfunc (s *UtilSuite) TestValidateDirPath() {\n\tdir := s.T().TempDir()\n\ts.assertDirectoryExists(dir)\n\tfilename := \"test-file-name\"\n\ts.createFile(dir, filename)\n\tfpath := filepath.Join(dir, filename)\n\n\ttestCases := []struct {\n\t\tdirPath     string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tdirPath:     \"\",\n\t\t\texpectedErr: errEmptyDirectoryPath,\n\t\t},\n\t\t{\n\t\t\tdirPath:     \"/absolute/path\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tdirPath:     \"relative/path\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tdirPath:     dir,\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tdirPath:     fpath,\n\t\t\texpectedErr: util.ErrDirectoryExpected,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.expectedErr, validateDirPath(tc.dirPath))\n\t}\n}\n\nfunc (s *UtilSuite) TestconstructHistoryFilename() {\n\ttestCases := []struct {\n\t\tdomainID             string\n\t\tworkflowID           string\n\t\trunID                string\n\t\tcloseFailoverVersion int64\n\t\texpectBuiltName      string\n\t}{\n\t\t{\n\t\t\tdomainID:             \"testDomainID\",\n\t\t\tworkflowID:           \"testWorkflowID\",\n\t\t\trunID:                \"testRunID\",\n\t\t\tcloseFailoverVersion: 5,\n\t\t\texpectBuiltName:      \"17971674567288329890367046253745284795510285995943906173973_5.history\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tfilename := constructHistoryFilename(tc.domainID, tc.workflowID, tc.runID, tc.closeFailoverVersion)\n\t\ts.Equal(tc.expectBuiltName, filename)\n\t}\n}\n\nfunc (s *UtilSuite) TestExtractCloseFailoverVersion() {\n\ttestCases := []struct {\n\t\tfilename        string\n\t\texpectedVersion int64\n\t\texpectedErr     bool\n\t}{\n\t\t{\n\t\t\tfilename:        \"17971674567288329890367046253745284795510285995943906173973_5.history\",\n\t\t\texpectedVersion: 5,\n\t\t\texpectedErr:     false,\n\t\t},\n\t\t{\n\t\t\tfilename:    \"history\",\n\t\t\texpectedErr: true,\n\t\t},\n\t\t{\n\t\t\tfilename:    \"some.random.filename\",\n\t\t\texpectedErr: true,\n\t\t},\n\t\t{\n\t\t\tfilename:        \"some-random_101.filename\",\n\t\t\texpectedVersion: 101,\n\t\t\texpectedErr:     false,\n\t\t},\n\t\t{\n\t\t\tfilename:        \"random_-100.filename\",\n\t\t\texpectedVersion: -100,\n\t\t\texpectedErr:     false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tversion, err := extractCloseFailoverVersion(tc.filename)\n\t\tif tc.expectedErr {\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(tc.expectedVersion, version)\n\t\t}\n\t}\n}\n\nfunc (s *UtilSuite) TestSerializeDeserializeGetHistoryToken() {\n\ttoken := &getHistoryToken{\n\t\tCloseFailoverVersion: 101,\n\t\tNextBatchIdx:         20,\n\t}\n\n\tserializedToken, err := serializeToken(token)\n\ts.Nil(err)\n\n\tdeserializedToken, err := deserializeGetHistoryToken(serializedToken)\n\ts.Nil(err)\n\ts.Equal(token, deserializedToken)\n}\n\nfunc (s *UtilSuite) createFile(dir string, filename string) {\n\terr := ioutil.WriteFile(filepath.Join(dir, filename), []byte(\"file contents\"), testFileMode)\n\ts.Nil(err)\n}\n\nfunc (s *UtilSuite) assertDirectoryExists(path string) {\n\texists, err := util.DirectoryExists(path)\n\ts.NoError(err)\n\ts.True(exists)\n}\n"
  },
  {
    "path": "common/archiver/filestore/visibilityArchiver.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"os\"\n\t\"path\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nconst (\n\terrEncodeVisibilityRecord = \"failed to encode visibility record\"\n)\n\ntype (\n\tvisibilityArchiver struct {\n\t\tcontainer   *archiver.VisibilityBootstrapContainer\n\t\tfileMode    os.FileMode\n\t\tdirMode     os.FileMode\n\t\tqueryParser QueryParser\n\t}\n\n\tqueryVisibilityToken struct {\n\t\tLastCloseTime int64\n\t\tLastRunID     string\n\t}\n\n\tvisibilityRecord archiver.ArchiveVisibilityRequest\n\n\tqueryVisibilityRequest struct {\n\t\tdomainID      string\n\t\tpageSize      int\n\t\tnextPageToken []byte\n\t\tparsedQuery   *parsedQuery\n\t}\n)\n\n// NewVisibilityArchiver creates a new archiver.VisibilityArchiver based on filestore\nfunc NewVisibilityArchiver(\n\tcontainer *archiver.VisibilityBootstrapContainer,\n\tconfig *config.FilestoreArchiver,\n) (archiver.VisibilityArchiver, error) {\n\tfileMode, err := strconv.ParseUint(config.FileMode, 0, 32)\n\tif err != nil {\n\t\treturn nil, errInvalidFileMode\n\t}\n\tdirMode, err := strconv.ParseUint(config.DirMode, 0, 32)\n\tif err != nil {\n\t\treturn nil, errInvalidDirMode\n\t}\n\treturn &visibilityArchiver{\n\t\tcontainer:   container,\n\t\tfileMode:    os.FileMode(fileMode),\n\t\tdirMode:     os.FileMode(dirMode),\n\t\tqueryParser: NewQueryParser(),\n\t}, nil\n}\n\nfunc (v *visibilityArchiver) Archive(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.ArchiveVisibilityRequest,\n\topts ...archiver.ArchiveOption,\n) (err error) {\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tdefer func() {\n\t\tif err != nil && featureCatalog.NonRetriableError != nil {\n\t\t\terr = featureCatalog.NonRetriableError()\n\t\t}\n\t}()\n\n\tlogger := archiver.TagLoggerWithArchiveVisibilityRequestAndURI(v.container.Logger, request, URI.String())\n\n\tif err := v.ValidateURI(URI); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\treturn err\n\t}\n\n\tif err := archiver.ValidateVisibilityArchivalRequest(request); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidArchiveRequest), tag.Error(err))\n\t\treturn err\n\t}\n\n\tdirPath := path.Join(URI.Path(), request.DomainID)\n\tif err = util.MkdirAll(dirPath, v.dirMode); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errMakeDirectory), tag.Error(err))\n\t\treturn err\n\t}\n\n\tencodedVisibilityRecord, err := encode(request)\n\tif err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errEncodeVisibilityRecord), tag.Error(err))\n\t\treturn err\n\t}\n\n\t// The filename has the format: closeTimestamp_hash(runID).visibility\n\t// This format allows the archiver to sort all records without reading the file contents\n\tfilename := constructVisibilityFilename(request.CloseTimestamp, request.RunID)\n\tif err := util.WriteFile(path.Join(dirPath, filename), encodedVisibilityRecord, v.fileMode); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errWriteFile), tag.Error(err))\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc (v *visibilityArchiver) Query(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.QueryVisibilityRequest,\n) (*archiver.QueryVisibilityResponse, error) {\n\tif err := v.ValidateURI(URI); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateQueryRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidQueryVisibilityRequest.Error()}\n\t}\n\n\tparsedQuery, err := v.queryParser.Parse(request.Query)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: err.Error()}\n\t}\n\n\tif parsedQuery.emptyResult {\n\t\treturn &archiver.QueryVisibilityResponse{}, nil\n\t}\n\n\treturn v.query(ctx, URI, &queryVisibilityRequest{\n\t\tdomainID:      request.DomainID,\n\t\tpageSize:      request.PageSize,\n\t\tnextPageToken: request.NextPageToken,\n\t\tparsedQuery:   parsedQuery,\n\t})\n}\n\nfunc (v *visibilityArchiver) query(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *queryVisibilityRequest,\n) (*archiver.QueryVisibilityResponse, error) {\n\tvar token *queryVisibilityToken\n\tif request.nextPageToken != nil {\n\t\tvar err error\n\t\ttoken, err = deserializeQueryVisibilityToken(request.nextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: archiver.ErrNextPageTokenCorrupted.Error()}\n\t\t}\n\t}\n\n\tdirPath := path.Join(URI.Path(), request.domainID)\n\texists, err := util.DirectoryExists(dirPath)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\tif !exists {\n\t\treturn &archiver.QueryVisibilityResponse{}, nil\n\t}\n\n\tfiles, err := util.ListFiles(dirPath)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\n\tfiles, err = sortAndFilterFiles(files, token)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\tif len(files) == 0 {\n\t\treturn &archiver.QueryVisibilityResponse{}, nil\n\t}\n\n\tresponse := &archiver.QueryVisibilityResponse{}\n\tfor idx, file := range files {\n\t\tencodedRecord, err := util.ReadFile(path.Join(dirPath, file))\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\trecord, err := decodeVisibilityRecord(encodedRecord)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\tif record.CloseTimestamp < request.parsedQuery.earliestCloseTime {\n\t\t\tbreak\n\t\t}\n\n\t\tif matchQuery(record, request.parsedQuery) {\n\t\t\tresponse.Executions = append(response.Executions, convertToExecutionInfo(record))\n\t\t\tif len(response.Executions) == request.pageSize {\n\t\t\t\tif idx != len(files) {\n\t\t\t\t\tnewToken := &queryVisibilityToken{\n\t\t\t\t\t\tLastCloseTime: record.CloseTimestamp,\n\t\t\t\t\t\tLastRunID:     record.RunID,\n\t\t\t\t\t}\n\t\t\t\t\tencodedToken, err := serializeToken(newToken)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t\t\t\t}\n\t\t\t\t\tresponse.NextPageToken = encodedToken\n\t\t\t\t}\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n\n\treturn response, nil\n}\n\nfunc (v *visibilityArchiver) ValidateURI(URI archiver.URI) error {\n\tif URI.Scheme() != URIScheme {\n\t\treturn archiver.ErrURISchemeMismatch\n\t}\n\n\treturn validateDirPath((URI.Path()))\n}\n\ntype parsedVisFilename struct {\n\tname        string\n\tcloseTime   int64\n\thashedRunID string\n}\n\n// sortAndFilterFiles sort visibility record file names based on close timestamp (desc) and use hashed runID to break ties.\n// if a nextPageToken is give, it only returns filenames that have a smaller close timestamp\nfunc sortAndFilterFiles(filenames []string, token *queryVisibilityToken) ([]string, error) {\n\tvar parsedFilenames []*parsedVisFilename\n\tfor _, name := range filenames {\n\t\tpieces := strings.FieldsFunc(name, func(r rune) bool {\n\t\t\treturn r == '_' || r == '.'\n\t\t})\n\t\tif len(pieces) != 3 {\n\t\t\treturn nil, fmt.Errorf(\"failed to parse visibility filename %s\", name)\n\t\t}\n\n\t\tcloseTime, err := strconv.ParseInt(pieces[0], 10, 64)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to parse visibility filename %s\", name)\n\t\t}\n\t\tparsedFilenames = append(parsedFilenames, &parsedVisFilename{\n\t\t\tname:        name,\n\t\t\tcloseTime:   closeTime,\n\t\t\thashedRunID: pieces[1],\n\t\t})\n\t}\n\n\tsort.Slice(parsedFilenames, func(i, j int) bool {\n\t\tif parsedFilenames[i].closeTime == parsedFilenames[j].closeTime {\n\t\t\treturn parsedFilenames[i].hashedRunID > parsedFilenames[j].hashedRunID\n\t\t}\n\t\treturn parsedFilenames[i].closeTime > parsedFilenames[j].closeTime\n\t})\n\n\tstartIdx := 0\n\tif token != nil {\n\t\tLastHashedRunID := hash(token.LastRunID)\n\t\tstartIdx = sort.Search(len(parsedFilenames), func(i int) bool {\n\t\t\tif parsedFilenames[i].closeTime == token.LastCloseTime {\n\t\t\t\treturn parsedFilenames[i].hashedRunID < LastHashedRunID\n\t\t\t}\n\t\t\treturn parsedFilenames[i].closeTime < token.LastCloseTime\n\t\t})\n\t}\n\n\tif startIdx == len(parsedFilenames) {\n\t\treturn []string{}, nil\n\t}\n\n\tvar filteredFilenames []string\n\tfor _, parsedFilename := range parsedFilenames[startIdx:] {\n\t\tfilteredFilenames = append(filteredFilenames, parsedFilename.name)\n\t}\n\treturn filteredFilenames, nil\n}\n\nfunc matchQuery(record *visibilityRecord, query *parsedQuery) bool {\n\tif record.CloseTimestamp < query.earliestCloseTime || record.CloseTimestamp > query.latestCloseTime {\n\t\treturn false\n\t}\n\tif query.workflowID != nil && record.WorkflowID != *query.workflowID {\n\t\treturn false\n\t}\n\tif query.runID != nil && record.RunID != *query.runID {\n\t\treturn false\n\t}\n\tif query.workflowTypeName != nil && record.WorkflowTypeName != *query.workflowTypeName {\n\t\treturn false\n\t}\n\tif query.closeStatus != nil && record.CloseStatus != *query.closeStatus {\n\t\treturn false\n\t}\n\treturn true\n}\n\nfunc convertToExecutionInfo(record *visibilityRecord) *types.WorkflowExecutionInfo {\n\treturn &types.WorkflowExecutionInfo{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: record.WorkflowID,\n\t\t\tRunID:      record.RunID,\n\t\t},\n\t\tType: &types.WorkflowType{\n\t\t\tName: record.WorkflowTypeName,\n\t\t},\n\t\tStartTime:     common.Int64Ptr(record.StartTimestamp),\n\t\tExecutionTime: common.Int64Ptr(record.ExecutionTimestamp),\n\t\tCloseTime:     common.Int64Ptr(record.CloseTimestamp),\n\t\tCloseStatus:   record.CloseStatus.Ptr(),\n\t\tHistoryLength: record.HistoryLength,\n\t\tMemo:          record.Memo,\n\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\tIndexedFields: archiver.ConvertSearchAttrToBytes(record.SearchAttributes),\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/archiver/filestore/visibilityArchiver_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"os\"\n\t\"path\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\nconst (\n\ttestWorkflowTypeName = \"test-workflow-type\"\n)\n\ntype visibilityArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tcontainer          *archiver.VisibilityBootstrapContainer\n\ttestArchivalURI    archiver.URI\n\ttestQueryDirectory string\n\tvisibilityRecords  []*visibilityRecord\n\n\tcontroller *gomock.Controller\n}\n\nfunc TestVisibilityArchiverSuite(t *testing.T) {\n\tsuite.Run(t, new(visibilityArchiverSuite))\n}\n\nfunc (s *visibilityArchiverSuite) SetupSuite() {\n\tvar err error\n\ts.testQueryDirectory = s.T().TempDir()\n\ts.setupVisibilityDirectory()\n\ts.testArchivalURI, err = archiver.NewURI(\"file:///a/b/c\")\n\ts.Require().NoError(err)\n}\n\nfunc (s *visibilityArchiverSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.container = &archiver.VisibilityBootstrapContainer{\n\t\tLogger: testlogger.New(s.T()),\n\t}\n\ts.controller = gomock.NewController(s.T())\n}\n\nfunc (s *visibilityArchiverSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *visibilityArchiverSuite) TestValidateURI() {\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"file://\",\n\t\t\texpectedErr: errEmptyDirectoryPath,\n\t\t},\n\t\t{\n\t\t\tURI:         \"file:///a/b/c\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedErr, visibilityArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_InvalidURI() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainName:         testDomainName,\n\t\tDomainID:           testDomainID,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     time.Now().UnixNano(),\n\t\tExecutionTimestamp: 0, // workflow without backoff\n\t\tCloseTimestamp:     time.Now().UnixNano(),\n\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\tHistoryLength:      int64(101),\n\t}\n\terr = visibilityArchiver.Archive(context.Background(), URI, request)\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_InvalidRequest() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\terr := visibilityArchiver.Archive(context.Background(), s.testArchivalURI, &archiver.ArchiveVisibilityRequest{})\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_NonRetriableErrorOption() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tnonRetriableErr := errors.New(\"some non-retryable error\")\n\terr := visibilityArchiver.Archive(\n\t\tcontext.Background(),\n\t\ts.testArchivalURI,\n\t\t&archiver.ArchiveVisibilityRequest{},\n\t\tarchiver.GetNonRetriableErrorOption(nonRetriableErr),\n\t)\n\ts.Equal(nonRetriableErr, err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Success() {\n\tdir := s.T().TempDir()\n\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tcloseTimestamp := time.Now()\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainID:           testDomainID,\n\t\tDomainName:         testDomainName,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     closeTimestamp.Add(-time.Hour).UnixNano(),\n\t\tExecutionTimestamp: 0, // workflow without backoff\n\t\tCloseTimestamp:     closeTimestamp.UnixNano(),\n\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\tHistoryLength:      int64(101),\n\t\tMemo: &types.Memo{\n\t\t\tFields: map[string][]byte{\n\t\t\t\t\"testFields\": []byte{1, 2, 3},\n\t\t\t},\n\t\t},\n\t\tSearchAttributes: map[string]string{\n\t\t\t\"testAttribute\": \"456\",\n\t\t},\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + dir)\n\ts.NoError(err)\n\terr = visibilityArchiver.Archive(context.Background(), URI, request)\n\ts.NoError(err)\n\n\texpectedFilename := constructVisibilityFilename(closeTimestamp.UnixNano(), testRunID)\n\tfilepath := path.Join(dir, testDomainID, expectedFilename)\n\ts.assertFileExists(filepath)\n\n\tdata, err := util.ReadFile(filepath)\n\ts.NoError(err)\n\n\tarchivedRecord := &archiver.ArchiveVisibilityRequest{}\n\terr = json.Unmarshal(data, archivedRecord)\n\ts.NoError(err)\n\ts.Equal(request, archivedRecord)\n}\n\nfunc (s *visibilityArchiverSuite) TestMatchQuery() {\n\ttestCases := []struct {\n\t\tquery       *parsedQuery\n\t\trecord      *visibilityRecord\n\t\tshouldMatch bool\n\t}{\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp: int64(1999),\n\t\t\t},\n\t\t\tshouldMatch: true,\n\t\t},\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp: int64(999),\n\t\t\t},\n\t\t\tshouldMatch: false,\n\t\t},\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t\tworkflowID:        common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp: int64(2000),\n\t\t\t},\n\t\t\tshouldMatch: false,\n\t\t},\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t\tworkflowID:        common.StringPtr(\"random workflowID\"),\n\t\t\t\trunID:             common.StringPtr(\"random runID\"),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp:   int64(12345),\n\t\t\t\tWorkflowID:       \"random workflowID\",\n\t\t\t\tRunID:            \"random runID\",\n\t\t\t\tWorkflowTypeName: \"random type name\",\n\t\t\t},\n\t\t\tshouldMatch: true,\n\t\t},\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t\tworkflowTypeName:  common.StringPtr(\"some random type name\"),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp: int64(12345),\n\t\t\t},\n\t\t\tshouldMatch: false,\n\t\t},\n\t\t{\n\t\t\tquery: &parsedQuery{\n\t\t\t\tearliestCloseTime: int64(1000),\n\t\t\t\tlatestCloseTime:   int64(12345),\n\t\t\t\tworkflowTypeName:  common.StringPtr(\"some random type name\"),\n\t\t\t\tcloseStatus:       types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(),\n\t\t\t},\n\t\t\trecord: &visibilityRecord{\n\t\t\t\tCloseTimestamp:   int64(12345),\n\t\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\t\tWorkflowTypeName: \"some random type name\",\n\t\t\t},\n\t\t\tshouldMatch: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.shouldMatch, matchQuery(tc.record, tc.query))\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) TestSortAndFilterFiles() {\n\ttestCases := []struct {\n\t\tfilenames      []string\n\t\ttoken          *queryVisibilityToken\n\t\texpectedResult []string\n\t}{\n\t\t{\n\t\t\tfilenames:      []string{\"9_12345.vis\", \"5_0.vis\", \"9_54321.vis\", \"1000_654.vis\", \"1000_78.vis\"},\n\t\t\texpectedResult: []string{\"1000_78.vis\", \"1000_654.vis\", \"9_54321.vis\", \"9_12345.vis\", \"5_0.vis\"},\n\t\t},\n\t\t{\n\t\t\tfilenames: []string{\"9_12345.vis\", \"5_0.vis\", \"9_54321.vis\", \"1000_654.vis\", \"1000_78.vis\"},\n\t\t\ttoken: &queryVisibilityToken{\n\t\t\t\tLastCloseTime: 3,\n\t\t\t},\n\t\t\texpectedResult: []string{},\n\t\t},\n\t\t{\n\t\t\tfilenames: []string{\"9_12345.vis\", \"5_0.vis\", \"9_54321.vis\", \"1000_654.vis\", \"1000_78.vis\"},\n\t\t\ttoken: &queryVisibilityToken{\n\t\t\t\tLastCloseTime: 999,\n\t\t\t},\n\t\t\texpectedResult: []string{\"9_54321.vis\", \"9_12345.vis\", \"5_0.vis\"},\n\t\t},\n\t\t{\n\t\t\tfilenames: []string{\"9_12345.vis\", \"5_0.vis\", \"9_54321.vis\", \"1000_654.vis\", \"1000_78.vis\"},\n\t\t\ttoken: &queryVisibilityToken{\n\t\t\t\tLastCloseTime: 5,\n\t\t\t},\n\t\t\texpectedResult: []string{\"5_0.vis\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tresult, err := sortAndFilterFiles(tc.filenames, tc.token)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedResult, result)\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidURI() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 1,\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidRequest() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, &archiver.QueryVisibilityRequest{})\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidQuery() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(nil, errors.New(\"invalid query\"))\n\tvisibilityArchiver.queryParser = mockParser\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, &archiver.QueryVisibilityRequest{\n\t\tDomainID: \"some random domainID\",\n\t\tPageSize: 10,\n\t\tQuery:    \"some invalid query\",\n\t})\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_DirectoryNotExist() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tearliestCloseTime: int64(1),\n\t\tlatestCloseTime:   int64(101),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tQuery:    \"parsed by mockParser\",\n\t\tPageSize: 1,\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Empty(response.Executions)\n\ts.Empty(response.NextPageToken)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidToken() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tearliestCloseTime: int64(1),\n\t\tlatestCloseTime:   int64(101),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID:      testDomainID,\n\t\tQuery:         \"parsed by mockParser\",\n\t\tPageSize:      1,\n\t\tNextPageToken: []byte{1, 2, 3},\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_NoNextPageToken() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tearliestCloseTime: int64(1),\n\t\tlatestCloseTime:   int64(10001),\n\t\tworkflowID:        common.StringPtr(testWorkflowID),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 10,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + s.testQueryDirectory)\n\ts.NoError(err)\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 1)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), response.Executions[0])\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_SmallPageSize() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tearliestCloseTime: int64(1),\n\t\tlatestCloseTime:   int64(10001),\n\t\tcloseStatus:       types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 2,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\tURI, err := archiver.NewURI(\"file://\" + s.testQueryDirectory)\n\ts.NoError(err)\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.NotNil(response.NextPageToken)\n\ts.Len(response.Executions, 2)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), response.Executions[0])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[1]), response.Executions[1])\n\n\trequest.NextPageToken = response.NextPageToken\n\tresponse, err = visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 1)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[3]), response.Executions[0])\n}\n\nfunc (s *visibilityArchiverSuite) TestArchiveAndQuery() {\n\tdir := s.T().TempDir()\n\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tearliestCloseTime: int64(10),\n\t\tlatestCloseTime:   int64(10001),\n\t\tcloseStatus:       types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\tURI, err := archiver.NewURI(\"file://\" + dir)\n\ts.NoError(err)\n\tfor _, record := range s.visibilityRecords {\n\t\terr := visibilityArchiver.Archive(context.Background(), URI, (*archiver.ArchiveVisibilityRequest)(record))\n\t\ts.NoError(err)\n\t}\n\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 1,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\texecutions := []*types.WorkflowExecutionInfo{}\n\tfor len(executions) == 0 || request.NextPageToken != nil {\n\t\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\texecutions = append(executions, response.Executions...)\n\t\trequest.NextPageToken = response.NextPageToken\n\t}\n\ts.Len(executions, 2)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), executions[0])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[1]), executions[1])\n}\n\nfunc (s *visibilityArchiverSuite) newTestVisibilityArchiver() *visibilityArchiver {\n\tconfig := &config.FilestoreArchiver{\n\t\tFileMode: testFileModeStr,\n\t\tDirMode:  testDirModeStr,\n\t}\n\tarchiver, err := NewVisibilityArchiver(s.container, config)\n\ts.NoError(err)\n\treturn archiver.(*visibilityArchiver)\n}\n\nfunc (s *visibilityArchiverSuite) setupVisibilityDirectory() {\n\ts.visibilityRecords = []*visibilityRecord{\n\t\t{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            testRunID,\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   1,\n\t\t\tCloseTimestamp:   10000,\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:    101,\n\t\t},\n\t\t{\n\t\t\tDomainID:           testDomainID,\n\t\t\tDomainName:         testDomainName,\n\t\t\tWorkflowID:         \"some random workflow ID\",\n\t\t\tRunID:              \"some random run ID\",\n\t\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\t\tStartTimestamp:     2,\n\t\t\tExecutionTimestamp: 0,\n\t\t\tCloseTimestamp:     1000,\n\t\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:      123,\n\t\t},\n\t\t{\n\t\t\tDomainID:           testDomainID,\n\t\t\tDomainName:         testDomainName,\n\t\t\tWorkflowID:         \"another workflow ID\",\n\t\t\tRunID:              \"another run ID\",\n\t\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\t\tStartTimestamp:     3,\n\t\t\tExecutionTimestamp: 0,\n\t\t\tCloseTimestamp:     10,\n\t\t\tCloseStatus:        types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\tHistoryLength:      456,\n\t\t},\n\t\t{\n\t\t\tDomainID:           testDomainID,\n\t\t\tDomainName:         testDomainName,\n\t\t\tWorkflowID:         \"and another workflow ID\",\n\t\t\tRunID:              \"and another run ID\",\n\t\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\t\tStartTimestamp:     3,\n\t\t\tExecutionTimestamp: 0,\n\t\t\tCloseTimestamp:     5,\n\t\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:      456,\n\t\t},\n\t\t{\n\t\t\tDomainID:           \"some random domain ID\",\n\t\t\tDomainName:         \"some random domain name\",\n\t\t\tWorkflowID:         \"another workflow ID\",\n\t\t\tRunID:              \"another run ID\",\n\t\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\t\tStartTimestamp:     3,\n\t\t\tExecutionTimestamp: 0,\n\t\t\tCloseTimestamp:     10000,\n\t\t\tCloseStatus:        types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\tHistoryLength:      456,\n\t\t},\n\t}\n\n\tfor _, record := range s.visibilityRecords {\n\t\ts.writeVisibilityRecordForQueryTest(record)\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) writeVisibilityRecordForQueryTest(record *visibilityRecord) {\n\tdata, err := encode(record)\n\ts.Require().NoError(err)\n\tfilename := constructVisibilityFilename(record.CloseTimestamp, record.RunID)\n\ts.Require().NoError(os.MkdirAll(path.Join(s.testQueryDirectory, record.DomainID), testDirMode))\n\terr = util.WriteFile(path.Join(s.testQueryDirectory, record.DomainID, filename), data, testFileMode)\n\ts.Require().NoError(err)\n}\n\nfunc (s *visibilityArchiverSuite) assertFileExists(filepath string) {\n\texists, err := util.FileExists(filepath)\n\ts.NoError(err)\n\ts.True(exists)\n}\n"
  },
  {
    "path": "common/archiver/gcloud/README.md",
    "content": "# Google Storage blobstore\n## Configuration\nSee https://cloud.google.com/docs/authentication/production to understand how is made the authentication against google cloud storage\n\nNowdays we support three different ways in order to let Cadence know where your google keyfile credentials are located\n\n* Cadence archival deployment.yaml configuration file\n* `GOOGLE_APPLICATION_CREDENTIALS` environment variable\n*  Google default credentials location\n\nIf more than one credentials location is given, then Cadence will resolve the conflicts by the following priority:\n\n`GOOGLE_APPLICATION_CREDENTIALS > Cadencen archival deployment.yaml > Google default credentials`\n\nBe sure that you have created your bucket first, and have enought rights in order to read/write over your bucket.\n\n### Gcloud Archival example\n\nEnabling archival is done by using the configuration below. `credentialsPath` is required but could be empty \"\" in order to allow a Google default credentials.\n\n```\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      gstorage:\n        credentialsPath: \"/tmp/keyfile.json\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      gstorage:\n        credentialsPath: \"/tmp/keyfile.json\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"gs://my-bucket-cad/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"gs://my-bucket-cad/cadence_archival/visibility\"\n```\n\n## Visibility query syntax\nYou can query the visibility store by using the `cadence workflow listarchived` command\n\nThe syntax for the query is based on SQL\n\nSupported column names are\n- WorkflowType *String*\n- WorkflowID *String*\n- StartTime *Date*\n- CloseTime *Date*\n- SearchPrecision *String - Day, Hour, Minute, Second*\n\nOne of these fields are required, StartTime or CloseTime and they are mutually exclusive and also SearchPrecision.\n\nSearching for a record will be done in times in the UTC timezone\n\nSearchPrecision specifies what range you want to search for records. If you use `SearchPrecision = 'Day'`\nit will search all records starting from `2020-01-21T00:00:00Z` to `2020-01-21T59:59:59Z` \n\n### Limitations\n\n- The only operator supported is `=`\n- Currently It's not possible to guarantee the resulSet order, specially if the pageSize it's fullfilled.  \n\n### Example\n\n*Searches the first 20 records for a given day 2020-01-21*\n\n`./cadence --do samples-domain workflow listarchived -ps=\"20\" -q \"StartTime = '2020-01-21T00:00:00Z' AND SearchPrecision='Day'\"`\n\n## Archival query syntax\n\nOnce you have a workflowId and a runId you can retrieve your workflow history.\n\nexample:\n\n`./cadence --do samples-domain  workflow  show  -w workflow-id -r runId`"
  },
  {
    "path": "common/archiver/gcloud/connector/client.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage connector\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"io\"\n\t\"io/ioutil\"\n\t\"os\"\n\n\t\"cloud.google.com/go/storage\"\n\t\"google.golang.org/api/iterator\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n)\n\nvar (\n\t// ErrBucketNotFound is non retriable error that is thrown when the bucket doesn't exist\n\tErrBucketNotFound = errors.New(\"bucket not found\")\n\terrObjectNotFound = errors.New(\"object not found\")\n)\n\ntype (\n\t// Precondition is a function that allow you to filter a query result.\n\t// If subject match params conditions then return true, else return false.\n\tPrecondition func(subject interface{}) bool\n\n\t// Client is a wrapper around Google cloud storages client library.\n\tClient interface {\n\t\tUpload(ctx context.Context, URI archiver.URI, fileName string, file []byte) error\n\t\tGet(ctx context.Context, URI archiver.URI, file string) ([]byte, error)\n\t\tQuery(ctx context.Context, URI archiver.URI, fileNamePrefix string) ([]string, error)\n\t\tQueryWithFilters(ctx context.Context, URI archiver.URI, fileNamePrefix string, pageSize, offset int, filters []Precondition) ([]string, bool, int, error)\n\t\tExist(ctx context.Context, URI archiver.URI, fileName string) (bool, error)\n\t}\n\n\t// Config structure used to parse from the storage-provider yaml nodes in [github.com/uber/cadence/common/config.HistoryArchiverProvider]\n\t// and [github.com/uber/cadence/common/config.VisibilityArchiverProvider] and\n\tConfig struct {\n\t\tCredentialsPath string `yaml:\"credentialsPath\"`\n\t}\n\n\tstorageWrapper struct {\n\t\tclient GcloudStorageClient\n\t}\n)\n\n// NewClient return a Cadence gcloudstorage.Client based on default google service account credentials (ScopeFullControl required).\n// Bucket must be created by Iaas scripts, in other words, this library doesn't create the required Bucket.\n// Optionally you can set your credential path through the \"GOOGLE_APPLICATION_CREDENTIALS\" environment variable or through cadence config file.\n// You can find more info about \"Google Setting Up Authentication for Server to Server Production Applications\" under the following link\n// https://cloud.google.com/docs/authentication/production\nfunc NewClient(ctx context.Context, config Config) (Client, error) {\n\tif credentialsPath := os.Getenv(\"GOOGLE_APPLICATION_CREDENTIALS\"); credentialsPath != \"\" {\n\t\tclientDelegate, err := newClientDelegateWithCredentials(ctx, credentialsPath)\n\t\treturn &storageWrapper{client: clientDelegate}, err\n\t}\n\n\tif config.CredentialsPath != \"\" {\n\t\tclientDelegate, err := newClientDelegateWithCredentials(ctx, config.CredentialsPath)\n\t\treturn &storageWrapper{client: clientDelegate}, err\n\t}\n\n\tclientDelegate, err := newDefaultClientDelegate(ctx)\n\treturn &storageWrapper{client: clientDelegate}, err\n\n}\n\n// NewClientWithParams return a gcloudstorage.Client based on input parameters\nfunc NewClientWithParams(clientD GcloudStorageClient) (Client, error) {\n\treturn &storageWrapper{client: clientD}, nil\n}\n\n// Upload push a file to gcloud storage bucket (sinkPath)\n// example:\n// Upload(ctx, mockBucketHandleClient, \"gs://my-bucket-cad/cadence_archival/development\", \"45273645-fileName.history\", fileReader)\nfunc (s *storageWrapper) Upload(ctx context.Context, URI archiver.URI, fileName string, file []byte) (err error) {\n\tbucket := s.client.Bucket(URI.Hostname())\n\twriter := bucket.Object(formatSinkPath(URI.Path()) + \"/\" + fileName).NewWriter(ctx)\n\t_, err = io.Copy(writer, bytes.NewReader(file))\n\tif err == nil {\n\t\terr = writer.Close()\n\t}\n\n\treturn err\n}\n\n// Exist check if a bucket or an object exist\n// If fileName is empty, then 'Exist' function will only check if the given bucket exist.\nfunc (s *storageWrapper) Exist(ctx context.Context, URI archiver.URI, fileName string) (exists bool, err error) {\n\tbucket := s.client.Bucket(URI.Hostname())\n\tif _, err := bucket.Attrs(ctx); err != nil {\n\t\treturn false, err\n\t}\n\n\tif fileName == \"\" {\n\t\treturn true, nil\n\t}\n\n\tif _, err = bucket.Object(fileName).Attrs(ctx); err != nil {\n\t\treturn false, errObjectNotFound\n\t}\n\n\treturn true, nil\n}\n\n// Get retrieve a file\nfunc (s *storageWrapper) Get(ctx context.Context, URI archiver.URI, fileName string) ([]byte, error) {\n\tbucket := s.client.Bucket(URI.Hostname())\n\treader, err := bucket.Object(formatSinkPath(URI.Path()) + \"/\" + fileName).NewReader(ctx)\n\tif err == nil {\n\t\tdefer reader.Close()\n\t\treturn ioutil.ReadAll(reader)\n\t}\n\n\treturn nil, err\n}\n\n// Query, retieves file names by provided storage query\nfunc (s *storageWrapper) Query(ctx context.Context, URI archiver.URI, fileNamePrefix string) (fileNames []string, err error) {\n\tfileNames = make([]string, 0)\n\tbucket := s.client.Bucket(URI.Hostname())\n\tvar attrs = new(storage.ObjectAttrs)\n\tit := bucket.Objects(ctx, &storage.Query{\n\t\tPrefix: formatSinkPath(URI.Path()) + \"/\" + fileNamePrefix,\n\t})\n\n\tfor {\n\t\tattrs, err = it.Next()\n\t\tif err == iterator.Done {\n\t\t\treturn fileNames, nil\n\t\t}\n\t\tfileNames = append(fileNames, attrs.Name)\n\t}\n\n}\n\n// QueryWithFilter, retieves filenames that match filter parameters. PageSize is optional, 0 means all records.\nfunc (s *storageWrapper) QueryWithFilters(ctx context.Context, URI archiver.URI, fileNamePrefix string, pageSize, offset int, filters []Precondition) ([]string, bool, int, error) {\n\n\tvar err error\n\tcurrentPos := offset\n\tresultSet := make([]string, 0)\n\tbucket := s.client.Bucket(URI.Hostname())\n\tvar attrs = new(storage.ObjectAttrs)\n\tit := bucket.Objects(ctx, &storage.Query{\n\t\tPrefix: formatSinkPath(URI.Path()) + \"/\" + fileNamePrefix,\n\t})\n\n\tfor {\n\t\tattrs, err = it.Next()\n\t\tif err == iterator.Done {\n\t\t\treturn resultSet, true, currentPos, nil\n\t\t}\n\n\t\tif completed := isPageCompleted(pageSize, len(resultSet)); completed {\n\t\t\treturn resultSet, completed, currentPos, err\n\t\t}\n\n\t\tvalid := true\n\t\tfor _, f := range filters {\n\t\t\tif valid = f(attrs.Name); !valid {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tif valid {\n\t\t\tif offset > 0 {\n\t\t\t\toffset--\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\t// if match parsedQuery criteria and current cursor position is the last known position (offset is zero), append fileName to resultSet\n\t\t\tresultSet = append(resultSet, attrs.Name)\n\t\t\tcurrentPos++\n\t\t}\n\t}\n\n}\n\nfunc isPageCompleted(pageSize, currentPosition int) bool {\n\treturn pageSize != 0 && currentPosition > 0 && pageSize <= currentPosition\n}\n\nfunc formatSinkPath(sinkPath string) string {\n\treturn sinkPath[1:]\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/clientDelegate.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage connector\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\n\t\"cloud.google.com/go/storage\"\n\t\"golang.org/x/oauth2/google\"\n\t\"google.golang.org/api/option\"\n)\n\ntype (\n\t// GcloudStorageClient is an interface that expose some methods from gcloud storage client\n\tGcloudStorageClient interface {\n\t\tBucket(URI string) BucketHandleWrapper\n\t}\n\n\tclientDelegate struct {\n\t\tnativeClient *storage.Client\n\t}\n)\n\ntype (\n\t// BucketHandleWrapper is an interface that expose some methods from gcloud storage bucket\n\tBucketHandleWrapper interface {\n\t\tObject(name string) ObjectHandleWrapper\n\t\tObjects(ctx context.Context, q *storage.Query) ObjectIteratorWrapper\n\t\tAttrs(ctx context.Context) (*storage.BucketAttrs, error)\n\t}\n\n\tbucketDelegate struct {\n\t\tbucket *storage.BucketHandle\n\t}\n)\n\ntype (\n\t// ObjectHandleWrapper is an interface that expose some methods from gcloud storage object\n\tObjectHandleWrapper interface {\n\t\tNewWriter(ctx context.Context) WriterWrapper\n\t\tNewReader(ctx context.Context) (ReaderWrapper, error)\n\t\tAttrs(ctx context.Context) (*storage.ObjectAttrs, error)\n\t}\n\n\tobjectDelegate struct {\n\t\tobject *storage.ObjectHandle\n\t}\n)\n\ntype (\n\t// WriterWrapper is an interface that expose some methods from gcloud storage writer\n\tWriterWrapper interface {\n\t\tClose() error\n\t\tWrite(p []byte) (n int, err error)\n\t\tCloseWithError(err error) error\n\t}\n\n\twriterDelegate struct {\n\t\twriter *storage.Writer\n\t}\n)\n\ntype (\n\t// ReaderWrapper is an interface that expose some methods from gcloud storage reader\n\tReaderWrapper interface {\n\t\tClose() error\n\t\tRead(p []byte) (int, error)\n\t}\n\n\treaderDelegate struct {\n\t\treader *storage.Reader\n\t}\n)\n\ntype (\n\t// ObjectIteratorWrapper is an interface that expose some methods from gcloud storage objectIterator\n\tObjectIteratorWrapper interface {\n\t\tNext() (*storage.ObjectAttrs, error)\n\t}\n)\n\nfunc newDefaultClientDelegate(ctx context.Context) (*clientDelegate, error) {\n\tnativeClient, err := storage.NewClient(ctx)\n\treturn &clientDelegate{nativeClient: nativeClient}, err\n}\n\nfunc newClientDelegateWithCredentials(ctx context.Context, credentialsPath string) (*clientDelegate, error) {\n\n\tjsonKey, err := ioutil.ReadFile(credentialsPath)\n\tif err != nil {\n\t\treturn newDefaultClientDelegate(ctx)\n\t}\n\n\tconf, err := google.JWTConfigFromJSON(jsonKey, storage.ScopeFullControl)\n\tif err != nil {\n\t\treturn newDefaultClientDelegate(ctx)\n\t}\n\n\tnativeClient, err := storage.NewClient(ctx, option.WithTokenSource(conf.TokenSource(ctx)))\n\treturn &clientDelegate{nativeClient: nativeClient}, err\n}\n\n// Bucket returns a BucketHandle, which provides operations on the named bucket.\n// This call does not perform any network operations.\n//\n// The supplied name must contain only lowercase letters, numbers, dashes,\n// underscores, and dots. The full specification for valid bucket names can be\n// found at:\n//\n//\thttps://cloud.google.com/storage/docs/bucket-naming\nfunc (c *clientDelegate) Bucket(bucketName string) BucketHandleWrapper {\n\treturn &bucketDelegate{bucket: c.nativeClient.Bucket(bucketName)}\n}\n\n// Object returns an ObjectHandle, which provides operations on the named object.\n// This call does not perform any network operations.\n//\n// name must consist entirely of valid UTF-8-encoded runes. The full specification\n// for valid object names can be found at:\n//\n//\thttps://cloud.google.com/storage/docs/bucket-naming\nfunc (b *bucketDelegate) Object(name string) ObjectHandleWrapper {\n\treturn &objectDelegate{object: b.bucket.Object(name)}\n}\n\n// Objects returns an iterator over the objects in the bucket that match the Query q.\n// If q is nil, no filtering is done.\nfunc (b *bucketDelegate) Objects(ctx context.Context, q *storage.Query) ObjectIteratorWrapper {\n\treturn b.bucket.Objects(ctx, q)\n}\n\n// Attrs returns the metadata for the bucket.\nfunc (b *bucketDelegate) Attrs(ctx context.Context) (*storage.BucketAttrs, error) {\n\treturn b.bucket.Attrs(ctx)\n}\n\n// NewWriter returns a storage Writer that writes to the GCS object\n// associated with this ObjectHandle.\n//\n// A new object will be created unless an object with this name already exists.\n// Otherwise any previous object with the same name will be replaced.\n// The object will not be available (and any previous object will remain)\n// until Close has been called.\n//\n// Attributes can be set on the object by modifying the returned Writer's\n// ObjectAttrs field before the first call to Write. If no ContentType\n// attribute is specified, the content type will be automatically sniffed\n// using net/http.DetectContentType.\n//\n// It is the caller's responsibility to call Close when writing is done. To\n// stop writing without saving the data, cancel the context.\nfunc (o *objectDelegate) NewWriter(ctx context.Context) WriterWrapper {\n\treturn &writerDelegate{writer: o.object.NewWriter(ctx)}\n}\n\n// NewReader creates a new Reader to read the contents of the\n// object.\n// ErrObjectNotExist will be returned if the object is not found.\n//\n// The caller must call Close on the returned Reader when done reading.\nfunc (o *objectDelegate) NewReader(ctx context.Context) (ReaderWrapper, error) {\n\tr, err := o.object.NewReader(ctx)\n\treturn &readerDelegate{reader: r}, err\n}\n\nfunc (o *objectDelegate) Attrs(ctx context.Context) (attrs *storage.ObjectAttrs, err error) {\n\treturn o.object.Attrs(ctx)\n}\n\n// Close completes the write operation and flushes any buffered data.\n// If Close doesn't return an error, metadata about the written object\n// can be retrieved by calling Attrs.\nfunc (w *writerDelegate) Close() error {\n\treturn w.writer.Close()\n}\n\n// Write appends to w. It implements the io.Writer interface.\n//\n// Since writes happen asynchronously, Write may return a nil\n// error even though the write failed (or will fail). Always\n// use the error returned from Writer.Close to determine if\n// the upload was successful.\nfunc (w *writerDelegate) Write(p []byte) (int, error) {\n\treturn w.writer.Write(p)\n}\n\n// CloseWithError aborts the write operation with the provided error.\n// CloseWithError always returns nil.\n//\n// Deprecated: cancel the context passed to NewWriter instead.\nfunc (w *writerDelegate) CloseWithError(err error) error {\n\treturn w.writer.CloseWithError(err)\n}\n\n// Close closes the Reader. It must be called when done reading.\nfunc (r *readerDelegate) Close() error {\n\treturn r.reader.Close()\n}\n\nfunc (r *readerDelegate) Read(p []byte) (int, error) {\n\treturn r.reader.Read(p)\n\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/client_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage connector_test\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"io\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"cloud.google.com/go/storage\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"google.golang.org/api/iterator\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector/mocks\"\n)\n\nfunc (s *clientSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\tfile, _ := json.MarshalIndent(&fakeData{data: \"example\"}, \"\", \" \")\n\n\tos.MkdirAll(\"/tmp/cadence_archival/development\", os.ModePerm)\n\ts.Require().NoError(ioutil.WriteFile(\"/tmp/cadence_archival/development/myfile.history\", file, 0644))\n}\n\nfunc (s *clientSuite) TearDownTest() {\n\tos.Remove(\"/tmp/cadence_archival/development/myfile.history\")\n}\n\nfunc TestClientSuite(t *testing.T) {\n\tsuite.Run(t, new(clientSuite))\n}\n\ntype clientSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\ntype fakeData struct {\n\tdata string\n}\n\nfunc (s *clientSuite) TestUpload() {\n\tctx := context.Background()\n\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\tmockObjectHandler := &mocks.ObjectHandleWrapper{}\n\tmockWriter := &mocks.WriterWrapper{}\n\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tmockStorageClient.On(\"Bucket\", \"my-bucket-cad\").Return(mockBucketHandleClient).Times(1)\n\tmockBucketHandleClient.On(\"Object\", \"cadence_archival/development/myfile.history\").Return(mockObjectHandler).Times(1)\n\tmockObjectHandler.On(\"NewWriter\", ctx).Return(mockWriter).Times(1)\n\tmockWriter.On(\"Write\", mock.Anything).Return(2, nil).Times(2)\n\tmockWriter.On(\"Close\").Return(nil).Times(1)\n\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\ts.Require().NoError(err)\n\terr = storageWrapper.Upload(ctx, URI, \"myfile.history\", []byte(\"{}\"))\n\ts.Require().NoError(err)\n}\n\nfunc (s *clientSuite) TestUploadWriterCloseError() {\n\tctx := context.Background()\n\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\tmockObjectHandler := &mocks.ObjectHandleWrapper{}\n\tmockWriter := &mocks.WriterWrapper{}\n\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tmockStorageClient.On(\"Bucket\", \"my-bucket-cad\").Return(mockBucketHandleClient).Times(1)\n\tmockBucketHandleClient.On(\"Object\", \"cadence_archival/development/myfile.history\").Return(mockObjectHandler).Times(1)\n\tmockObjectHandler.On(\"NewWriter\", ctx).Return(mockWriter).Times(1)\n\tmockWriter.On(\"Write\", mock.Anything).Return(2, nil).Times(2)\n\tmockWriter.On(\"Close\").Return(errors.New(\"Not Found\")).Times(1)\n\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\ts.Require().NoError(err)\n\terr = storageWrapper.Upload(ctx, URI, \"myfile.history\", []byte(\"{}\"))\n\ts.Require().EqualError(err, \"Not Found\")\n}\n\nfunc (s *clientSuite) TestExist() {\n\tctx := context.Background()\n\ttestCases := []struct {\n\t\tcallContext         context.Context\n\t\tURI                 string\n\t\tfileName            string\n\t\tbucketName          string\n\t\tbucketExists        bool\n\t\tobjectExists        bool\n\t\tbucketExpectedError error\n\t\tobjectExpectedError error\n\t}{\n\t\t{\n\t\t\tcallContext:         ctx,\n\t\t\tURI:                 \"gs://my-bucket-cad/cadence_archival/development\",\n\t\t\tfileName:            \"\",\n\t\t\tbucketName:          \"my-bucket-cad\",\n\t\t\tbucketExists:        true,\n\t\t\tobjectExists:        false,\n\t\t\tbucketExpectedError: nil,\n\t\t\tobjectExpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tcallContext:         ctx,\n\t\t\tURI:                 \"gs://my-bucket-cad/cadence_archival/development\",\n\t\t\tfileName:            \"\",\n\t\t\tbucketName:          \"my-bucket-cad\",\n\t\t\tbucketExists:        false,\n\t\t\tobjectExists:        false,\n\t\t\tbucketExpectedError: errors.New(\"bucket not found\"),\n\t\t\tobjectExpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tcallContext:         ctx,\n\t\t\tURI:                 \"gs://my-bucket-cad/cadence_archival/development\",\n\t\t\tfileName:            \"myfile.history\",\n\t\t\tbucketName:          \"my-bucket-cad\",\n\t\t\tbucketExists:        true,\n\t\t\tobjectExists:        true,\n\t\t\tbucketExpectedError: nil,\n\t\t\tobjectExpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tcallContext:         ctx,\n\t\t\tURI:                 \"gs://my-bucket-cad/cadence_archival/development\",\n\t\t\tfileName:            \"myfile.history\",\n\t\t\tbucketName:          \"my-bucket-cad\",\n\t\t\tbucketExists:        true,\n\t\t\tobjectExists:        false,\n\t\t\tbucketExpectedError: nil,\n\t\t\tobjectExpectedError: errors.New(\"object not found\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\n\t\tmockStorageClient := &mocks.GcloudStorageClient{}\n\t\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\t\tmockObjectHandler := &mocks.ObjectHandleWrapper{}\n\n\t\tmockStorageClient.On(\"Bucket\", tc.bucketName).Return(mockBucketHandleClient).Times(1)\n\t\tmockBucketHandleClient.On(\"Attrs\", tc.callContext).Return(nil, tc.bucketExpectedError).Times(1)\n\t\tmockBucketHandleClient.On(\"Object\", tc.fileName).Return(mockObjectHandler).Times(1)\n\t\tmockObjectHandler.On(\"Attrs\", tc.callContext).Return(nil, tc.objectExpectedError).Times(1)\n\t\tURI, _ := archiver.NewURI(tc.URI)\n\t\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\t\texists, err := storageWrapper.Exist(tc.callContext, URI, tc.fileName)\n\t\tif tc.bucketExists && tc.fileName == \"\" {\n\t\t\ts.Require().NoError(err)\n\t\t\ts.True(exists)\n\t\t} else if !tc.bucketExists {\n\t\t\ts.Require().Error(err)\n\t\t\ts.Require().EqualError(err, \"bucket not found\")\n\t\t\ts.False(exists)\n\t\t} else if tc.bucketExists && tc.fileName != \"\" && tc.objectExists {\n\t\t\ts.Require().NoError(err)\n\t\t\ts.True(exists)\n\t\t} else if tc.bucketExists && tc.fileName != \"\" && !tc.objectExists {\n\t\t\ts.Require().Error(err)\n\t\t\ts.Require().EqualError(err, \"object not found\")\n\t\t\ts.False(exists)\n\t\t}\n\t}\n\n}\n\nfunc (s *clientSuite) TestGet() {\n\tctx := context.Background()\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\tmockObjectHandler := &mocks.ObjectHandleWrapper{}\n\tmockReader := &mocks.ReaderWrapper{}\n\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tmockStorageClient.On(\"Bucket\", \"my-bucket-cad\").Return(mockBucketHandleClient).Times(1)\n\tmockBucketHandleClient.On(\"Object\", \"cadence_archival/development/myfile.history\").Return(mockObjectHandler).Times(1)\n\tmockObjectHandler.On(\"NewReader\", ctx).Return(mockReader, nil).Times(1)\n\tmockReader.On(\"Read\", mock.Anything).Return(2, io.EOF).Times(2)\n\tmockReader.On(\"Close\").Return(nil).Times(1)\n\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\ts.Require().NoError(err)\n\t_, err = storageWrapper.Get(ctx, URI, \"myfile.history\")\n\ts.Require().NoError(err)\n}\n\nfunc (s *clientSuite) TestWrongGoogleCredentialsPath() {\n\tctx := context.Background()\n\ts.T().Setenv(\"GOOGLE_APPLICATION_CREDENTIALS\", \"/Wrong/path\")\n\t_, err := connector.NewClient(ctx, connector.Config{}) // config is ignored, prefers env var\n\ts.Require().Error(err)\n}\n\nfunc (s *clientSuite) TestQuery() {\n\n\tctx := context.Background()\n\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tmockObjectIterator := &mocks.ObjectIteratorWrapper{}\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tattr := new(storage.ObjectAttrs)\n\tattr.Name = \"fileName_01\"\n\n\tmockStorageClient.On(\"Bucket\", \"my-bucket-cad\").Return(mockBucketHandleClient).Times(1)\n\tmockBucketHandleClient.On(\"Objects\", ctx, mock.Anything).Return(mockObjectIterator).Times(1)\n\tmockIterator := 0\n\tmockObjectIterator.On(\"Next\").Return(func() *storage.ObjectAttrs {\n\t\tmockIterator++\n\t\tif mockIterator == 1 {\n\t\t\treturn attr\n\t\t}\n\t\treturn nil\n\n\t}, func() error {\n\t\tif mockIterator == 1 {\n\t\t\treturn nil\n\t\t}\n\t\treturn iterator.Done\n\n\t}).Times(2)\n\n\tvar fileNames []string\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\ts.Require().NoError(err)\n\tfileNames, err = storageWrapper.Query(ctx, URI, \"7478875943689868082123907395549832634615673687049942026838\")\n\ts.Require().NoError(err)\n\ts.Equal(strings.Join(fileNames, \", \"), \"fileName_01\")\n}\n\nfunc (s *clientSuite) TestQueryWithFilter() {\n\n\tctx := context.Background()\n\tmockBucketHandleClient := &mocks.BucketHandleWrapper{}\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tmockObjectIterator := &mocks.ObjectIteratorWrapper{}\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tattr := new(storage.ObjectAttrs)\n\tattr.Name = \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\"\n\tattrInvalid := new(storage.ObjectAttrs)\n\tattrInvalid.Name = \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464321_15619178330501475177.visibility\"\n\n\tmockStorageClient.On(\"Bucket\", \"my-bucket-cad\").Return(mockBucketHandleClient).Times(1)\n\tmockBucketHandleClient.On(\"Objects\", ctx, mock.Anything).Return(mockObjectIterator).Times(1)\n\tmockIterator := 0\n\tmockObjectIterator.On(\"Next\").Return(func() *storage.ObjectAttrs {\n\t\tmockIterator++\n\t\tswitch mockIterator {\n\t\tcase 1:\n\t\t\treturn attr\n\t\tcase 2:\n\t\t\treturn attrInvalid\n\t\tdefault:\n\t\t\treturn nil\n\t\t}\n\n\t}, func() error {\n\t\tswitch mockIterator {\n\t\tcase 1:\n\t\t\treturn nil\n\t\tcase 2:\n\t\t\treturn nil\n\t\tdefault:\n\t\t\treturn iterator.Done\n\t\t}\n\n\t}).Times(3)\n\n\tvar fileNames []string\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\ts.Require().NoError(err)\n\tfileNames, _, _, err = storageWrapper.QueryWithFilters(ctx, URI, \"closeTimeout_2020-02-27T09:42:28Z\", 0, 0, []connector.Precondition{newWorkflowIDPrecondition(\"4418294404690464320\")})\n\n\ts.Require().NoError(err)\n\ts.Equal(strings.Join(fileNames, \", \"), \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\")\n}\n\nfunc newWorkflowIDPrecondition(workflowID string) connector.Precondition {\n\treturn func(subject interface{}) bool {\n\n\t\tif workflowID == \"\" {\n\t\t\treturn true\n\t\t}\n\n\t\tfileName, ok := subject.(string)\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\n\t\tif strings.Contains(fileName, workflowID) {\n\t\t\tfileNameParts := strings.Split(fileName, \"_\")\n\t\t\tif len(fileNameParts) != 5 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\treturn strings.Contains(fileName, fileNameParts[3])\n\t\t}\n\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/BucketHandleWrapper.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tstorage \"cloud.google.com/go/storage\"\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tconnector \"github.com/uber/cadence/common/archiver/gcloud/connector\"\n)\n\n// BucketHandleWrapper is an autogenerated mock type for the BucketHandleWrapper type\ntype BucketHandleWrapper struct {\n\tmock.Mock\n}\n\n// Attrs provides a mock function with given fields: ctx\nfunc (_m *BucketHandleWrapper) Attrs(ctx context.Context) (*storage.BucketAttrs, error) {\n\tret := _m.Called(ctx)\n\n\tvar r0 *storage.BucketAttrs\n\tif rf, ok := ret.Get(0).(func(context.Context) *storage.BucketAttrs); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*storage.BucketAttrs)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context) error); ok {\n\t\tr1 = rf(ctx)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Object provides a mock function with given fields: name\nfunc (_m *BucketHandleWrapper) Object(name string) connector.ObjectHandleWrapper {\n\tret := _m.Called(name)\n\n\tvar r0 connector.ObjectHandleWrapper\n\tif rf, ok := ret.Get(0).(func(string) connector.ObjectHandleWrapper); ok {\n\t\tr0 = rf(name)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(connector.ObjectHandleWrapper)\n\t\t}\n\t}\n\n\treturn r0\n}\n\n// Objects provides a mock function with given fields: ctx, q\nfunc (_m *BucketHandleWrapper) Objects(ctx context.Context, q *storage.Query) connector.ObjectIteratorWrapper {\n\tret := _m.Called(ctx, q)\n\n\tvar r0 connector.ObjectIteratorWrapper\n\tif rf, ok := ret.Get(0).(func(context.Context, *storage.Query) connector.ObjectIteratorWrapper); ok {\n\t\tr0 = rf(ctx, q)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(connector.ObjectIteratorWrapper)\n\t\t}\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/Client.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tarchiver \"github.com/uber/cadence/common/archiver\"\n\tconnector \"github.com/uber/cadence/common/archiver/gcloud/connector\"\n)\n\n// Client is an autogenerated mock type for the Client type\ntype Client struct {\n\tmock.Mock\n}\n\n// Exist provides a mock function with given fields: ctx, URI, fileName\nfunc (_m *Client) Exist(ctx context.Context, URI archiver.URI, fileName string) (bool, error) {\n\tret := _m.Called(ctx, URI, fileName)\n\n\tvar r0 bool\n\tif rf, ok := ret.Get(0).(func(context.Context, archiver.URI, string) bool); ok {\n\t\tr0 = rf(ctx, URI, fileName)\n\t} else {\n\t\tr0 = ret.Get(0).(bool)\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, archiver.URI, string) error); ok {\n\t\tr1 = rf(ctx, URI, fileName)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Get provides a mock function with given fields: ctx, URI, file\nfunc (_m *Client) Get(ctx context.Context, URI archiver.URI, file string) ([]byte, error) {\n\tret := _m.Called(ctx, URI, file)\n\n\tvar r0 []byte\n\tif rf, ok := ret.Get(0).(func(context.Context, archiver.URI, string) []byte); ok {\n\t\tr0 = rf(ctx, URI, file)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).([]byte)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, archiver.URI, string) error); ok {\n\t\tr1 = rf(ctx, URI, file)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Query provides a mock function with given fields: ctx, URI, fileNamePrefix\nfunc (_m *Client) Query(ctx context.Context, URI archiver.URI, fileNamePrefix string) ([]string, error) {\n\tret := _m.Called(ctx, URI, fileNamePrefix)\n\n\tvar r0 []string\n\tif rf, ok := ret.Get(0).(func(context.Context, archiver.URI, string) []string); ok {\n\t\tr0 = rf(ctx, URI, fileNamePrefix)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).([]string)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, archiver.URI, string) error); ok {\n\t\tr1 = rf(ctx, URI, fileNamePrefix)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// QueryWithFilters provides a mock function with given fields: ctx, URI, fileNamePrefix, pageSize, offset, filters\nfunc (_m *Client) QueryWithFilters(ctx context.Context, URI archiver.URI, fileNamePrefix string, pageSize int, offset int, filters []connector.Precondition) ([]string, bool, int, error) {\n\tret := _m.Called(ctx, URI, fileNamePrefix, pageSize, offset, filters)\n\n\tvar r0 []string\n\tif rf, ok := ret.Get(0).(func(context.Context, archiver.URI, string, int, int, []connector.Precondition) []string); ok {\n\t\tr0 = rf(ctx, URI, fileNamePrefix, pageSize, offset, filters)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).([]string)\n\t\t}\n\t}\n\n\tvar r1 bool\n\tif rf, ok := ret.Get(1).(func(context.Context, archiver.URI, string, int, int, []connector.Precondition) bool); ok {\n\t\tr1 = rf(ctx, URI, fileNamePrefix, pageSize, offset, filters)\n\t} else {\n\t\tr1 = ret.Get(1).(bool)\n\t}\n\n\tvar r2 int\n\tif rf, ok := ret.Get(2).(func(context.Context, archiver.URI, string, int, int, []connector.Precondition) int); ok {\n\t\tr2 = rf(ctx, URI, fileNamePrefix, pageSize, offset, filters)\n\t} else {\n\t\tr2 = ret.Get(2).(int)\n\t}\n\n\tvar r3 error\n\tif rf, ok := ret.Get(3).(func(context.Context, archiver.URI, string, int, int, []connector.Precondition) error); ok {\n\t\tr3 = rf(ctx, URI, fileNamePrefix, pageSize, offset, filters)\n\t} else {\n\t\tr3 = ret.Error(3)\n\t}\n\n\treturn r0, r1, r2, r3\n}\n\n// Upload provides a mock function with given fields: ctx, URI, fileName, file\nfunc (_m *Client) Upload(ctx context.Context, URI archiver.URI, fileName string, file []byte) error {\n\tret := _m.Called(ctx, URI, fileName, file)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, archiver.URI, string, []byte) error); ok {\n\t\tr0 = rf(ctx, URI, fileName, file)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/GcloudStorageClient.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tconnector \"github.com/uber/cadence/common/archiver/gcloud/connector\"\n)\n\n// GcloudStorageClient is an autogenerated mock type for the GcloudStorageClient type\ntype GcloudStorageClient struct {\n\tmock.Mock\n}\n\n// Bucket provides a mock function with given fields: URI\nfunc (_m *GcloudStorageClient) Bucket(URI string) connector.BucketHandleWrapper {\n\tret := _m.Called(URI)\n\n\tvar r0 connector.BucketHandleWrapper\n\tif rf, ok := ret.Get(0).(func(string) connector.BucketHandleWrapper); ok {\n\t\tr0 = rf(URI)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(connector.BucketHandleWrapper)\n\t\t}\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/ObjectHandleWrapper.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tstorage \"cloud.google.com/go/storage\"\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tconnector \"github.com/uber/cadence/common/archiver/gcloud/connector\"\n)\n\n// ObjectHandleWrapper is an autogenerated mock type for the ObjectHandleWrapper type\ntype ObjectHandleWrapper struct {\n\tmock.Mock\n}\n\n// Attrs provides a mock function with given fields: ctx\nfunc (_m *ObjectHandleWrapper) Attrs(ctx context.Context) (*storage.ObjectAttrs, error) {\n\tret := _m.Called(ctx)\n\n\tvar r0 *storage.ObjectAttrs\n\tif rf, ok := ret.Get(0).(func(context.Context) *storage.ObjectAttrs); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*storage.ObjectAttrs)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context) error); ok {\n\t\tr1 = rf(ctx)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// NewReader provides a mock function with given fields: ctx\nfunc (_m *ObjectHandleWrapper) NewReader(ctx context.Context) (connector.ReaderWrapper, error) {\n\tret := _m.Called(ctx)\n\n\tvar r0 connector.ReaderWrapper\n\tif rf, ok := ret.Get(0).(func(context.Context) connector.ReaderWrapper); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(connector.ReaderWrapper)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context) error); ok {\n\t\tr1 = rf(ctx)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// NewWriter provides a mock function with given fields: ctx\nfunc (_m *ObjectHandleWrapper) NewWriter(ctx context.Context) connector.WriterWrapper {\n\tret := _m.Called(ctx)\n\n\tvar r0 connector.WriterWrapper\n\tif rf, ok := ret.Get(0).(func(context.Context) connector.WriterWrapper); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(connector.WriterWrapper)\n\t\t}\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/ObjectIteratorWrapper.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tstorage \"cloud.google.com/go/storage\"\n\tmock \"github.com/stretchr/testify/mock\"\n)\n\n// ObjectIteratorWrapper is an autogenerated mock type for the ObjectIteratorWrapper type\ntype ObjectIteratorWrapper struct {\n\tmock.Mock\n}\n\n// Next provides a mock function with given fields:\nfunc (_m *ObjectIteratorWrapper) Next() (*storage.ObjectAttrs, error) {\n\tret := _m.Called()\n\n\tvar r0 *storage.ObjectAttrs\n\tif rf, ok := ret.Get(0).(func() *storage.ObjectAttrs); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*storage.ObjectAttrs)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func() error); ok {\n\t\tr1 = rf()\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/ReaderWrapper.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport mock \"github.com/stretchr/testify/mock\"\n\n// ReaderWrapper is an autogenerated mock type for the ReaderWrapper type\ntype ReaderWrapper struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *ReaderWrapper) Close() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Read provides a mock function with given fields: p\nfunc (_m *ReaderWrapper) Read(p []byte) (int, error) {\n\tret := _m.Called(p)\n\n\tvar r0 int\n\tif rf, ok := ret.Get(0).(func([]byte) int); ok {\n\t\tr0 = rf(p)\n\t} else {\n\t\tr0 = ret.Get(0).(int)\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func([]byte) error); ok {\n\t\tr1 = rf(p)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n"
  },
  {
    "path": "common/archiver/gcloud/connector/mocks/WriterWrapper.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport mock \"github.com/stretchr/testify/mock\"\n\n// WriterWrapper is an autogenerated mock type for the WriterWrapper type\ntype WriterWrapper struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *WriterWrapper) Close() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// CloseWithError provides a mock function with given fields: err\nfunc (_m *WriterWrapper) CloseWithError(err error) error {\n\tret := _m.Called(err)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(error) error); ok {\n\t\tr0 = rf(err)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Write provides a mock function with given fields: p\nfunc (_m *WriterWrapper) Write(p []byte) (int, error) {\n\tret := _m.Called(p)\n\n\tvar r0 int\n\tif rf, ok := ret.Get(0).(func([]byte) int); ok {\n\t\tr0 = rf(p)\n\t} else {\n\t\tr0 = ret.Get(0).(int)\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func([]byte) error); ok {\n\t\tr1 = rf(p)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n"
  },
  {
    "path": "common/archiver/gcloud/go.mod",
    "content": "module github.com/uber/cadence/common/archiver/gcloud\n\ngo 1.23.0\n\ntoolchain go1.23.4\n\n// build against the current code in the \"main\" module, not a specific SHA.\n//\n// anyone outside this repo using this needs to ensure that both the \"main\" module and this module\n// are at the same SHA for consistency, but internally we can cheat by telling Go that it's at a\n// relative file path.\nreplace github.com/uber/cadence => ../../..\n\n// ringpop-go and tchannel-go depends on older version of thrift, yarpc brings up newer version\nreplace github.com/apache/thrift => github.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7\n\nrequire (\n\tgithub.com/aws/aws-sdk-go v1.54.12 // indirect\n\tgithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 // indirect\n\tgithub.com/cristalhq/jwt/v3 v3.1.0 // indirect\n\tgithub.com/davecgh/go-spew v1.1.1 // indirect\n\tgithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da\n\tgithub.com/gogo/protobuf v1.3.2 // indirect\n\tgithub.com/golang/mock v1.6.0 // indirect\n\tgithub.com/google/uuid v1.6.0 // indirect\n\tgithub.com/m3db/prometheus_client_golang v0.8.1 // indirect\n\tgithub.com/opentracing/opentracing-go v1.2.0 // indirect\n\tgithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709 // indirect\n\tgithub.com/robfig/cron v1.2.0 // indirect\n\tgithub.com/stretchr/testify v1.10.0\n\tgithub.com/uber-go/tally v3.3.15+incompatible\n\tgithub.com/uber/ringpop-go v0.8.5 // indirect\n\tgithub.com/uber/tchannel-go v1.22.2 // indirect\n\tgithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2\n\tgo.uber.org/atomic v1.10.0 // indirect\n\tgo.uber.org/cadence v0.19.0 // indirect\n\tgo.uber.org/config v1.4.0 // indirect\n\tgo.uber.org/multierr v1.10.0 // indirect\n\tgo.uber.org/thriftrw v1.29.2 // indirect\n\tgo.uber.org/yarpc v1.70.3 // indirect\n\tgo.uber.org/zap v1.26.0 // indirect\n\tgolang.org/x/net v0.40.0 // indirect\n\tgolang.org/x/sync v0.14.0 // indirect\n\tgolang.org/x/time v0.5.0 // indirect\n\tgolang.org/x/tools v0.22.0 // indirect\n\tgoogle.golang.org/grpc v1.59.0 // indirect\n\tgopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19 // indirect\n\tgopkg.in/yaml.v2 v2.4.0 // indirect\n)\n\nrequire (\n\tcloud.google.com/go/storage v1.30.1\n\tgithub.com/uber/cadence v0.0.0-00010101000000-000000000000\n\tgo.uber.org/mock v0.5.0\n\tgolang.org/x/oauth2 v0.11.0\n\tgoogle.golang.org/api v0.128.0\n)\n\nrequire (\n\tcloud.google.com/go v0.110.8 // indirect\n\tcloud.google.com/go/compute v1.23.1 // indirect\n\tcloud.google.com/go/compute/metadata v0.2.3 // indirect\n\tcloud.google.com/go/iam v1.1.3 // indirect\n\tgithub.com/BurntSushi/toml v1.3.2 // indirect\n\tgithub.com/apache/thrift v0.17.0 // indirect\n\tgithub.com/beorn7/perks v1.0.1 // indirect\n\tgithub.com/cespare/xxhash/v2 v2.2.0 // indirect\n\tgithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a // indirect\n\tgithub.com/gogo/googleapis v1.3.2 // indirect\n\tgithub.com/gogo/status v1.1.0 // indirect\n\tgithub.com/golang-jwt/jwt/v5 v5.2.0 // indirect\n\tgithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect\n\tgithub.com/golang/protobuf v1.5.4 // indirect\n\tgithub.com/golang/snappy v0.0.4 // indirect\n\tgithub.com/google/gofuzz v1.0.0 // indirect\n\tgithub.com/google/s2a-go v0.1.4 // indirect\n\tgithub.com/googleapis/enterprise-certificate-proxy v0.2.4 // indirect\n\tgithub.com/googleapis/gax-go/v2 v2.12.0 // indirect\n\tgithub.com/jmespath/go-jmespath v0.4.0 // indirect\n\tgithub.com/jonboulle/clockwork v0.5.0 // indirect\n\tgithub.com/kr/pretty v0.3.0 // indirect\n\tgithub.com/m3db/prometheus_client_model v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_common v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_procfs v0.8.1 // indirect\n\tgithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect\n\tgithub.com/pkg/errors v0.9.1 // indirect\n\tgithub.com/pmezard/go-difflib v1.0.0 // indirect\n\tgithub.com/prometheus/client_golang v1.11.1 // indirect\n\tgithub.com/prometheus/client_model v0.4.0 // indirect\n\tgithub.com/prometheus/common v0.26.0 // indirect\n\tgithub.com/prometheus/procfs v0.6.0 // indirect\n\tgithub.com/robfig/cron/v3 v3.0.1 // indirect\n\tgithub.com/stretchr/objx v0.5.2 // indirect\n\tgithub.com/uber-go/mapdecode v1.0.0 // indirect\n\tgo.opencensus.io v0.24.0 // indirect\n\tgo.uber.org/dig v1.18.0 // indirect\n\tgo.uber.org/fx v1.23.0 // indirect\n\tgo.uber.org/net/metrics v1.3.0 // indirect\n\tgolang.org/x/crypto v0.38.0 // indirect\n\tgolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e // indirect\n\tgolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 // indirect\n\tgolang.org/x/mod v0.18.0 // indirect\n\tgolang.org/x/sys v0.33.0 // indirect\n\tgolang.org/x/text v0.25.0 // indirect\n\tgolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 // indirect\n\tgoogle.golang.org/appengine v1.6.7 // indirect\n\tgoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b // indirect\n\tgoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a // indirect\n\tgoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 // indirect\n\tgoogle.golang.org/protobuf v1.33.0 // indirect\n\tgopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n\thonnef.co/go/tools v0.3.2 // indirect\n)\n"
  },
  {
    "path": "common/archiver/gcloud/go.sum",
    "content": "cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.110.8 h1:tyNdfIxjzaWctIiLYOTalaLKZ17SI44SKFW26QbOhME=\ncloud.google.com/go v0.110.8/go.mod h1:Iz8AkXJf1qmxC3Oxoep8R1T36w8B92yU29PcBhHO5fk=\ncloud.google.com/go/compute v1.23.1 h1:V97tBoDaZHb6leicZ1G6DLK2BAaZLJ/7+9BB/En3hR0=\ncloud.google.com/go/compute v1.23.1/go.mod h1:CqB3xpmPKKt3OJpW2ndFIXnA9A4xAy/F3Xp1ixncW78=\ncloud.google.com/go/compute/metadata v0.2.3 h1:mg4jlk7mCAj6xXp9UJ4fjI9VUI5rubuGBW5aJ7UnBMY=\ncloud.google.com/go/compute/metadata v0.2.3/go.mod h1:VAV5nSsACxMJvgaAuX6Pk2AawlZn8kiOGuCv6gTkwuA=\ncloud.google.com/go/iam v1.1.3 h1:18tKG7DzydKWUnLjonWcJO6wjSCAtzh4GcRKlH/Hrzc=\ncloud.google.com/go/iam v1.1.3/go.mod h1:3khUlaBXfPKKe7huYgEpDn6FtgRyMEqbkvBxrQyY5SE=\ncloud.google.com/go/storage v1.30.1 h1:uOdMxAs8HExqBlnLtnQyP0YkvbiDpdGShGKtx6U/oNM=\ncloud.google.com/go/storage v1.30.1/go.mod h1:NfxhC0UJE1aXSx7CIIbCf7y9HKT7BiccwkR7+P7gN8E=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/BurntSushi/toml v0.4.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8=\ngithub.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=\ngithub.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 h1:kFOfPq6dUM1hTo4JG6LR5AXSUEsOjtdm0kw0FtQtMJA=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c=\ngithub.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7 h1:Fv9bK1Q+ly/ROk4aJsVMeuIwPel4bEnD8EPiI91nZMg=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=\ngithub.com/aws/aws-sdk-go v1.54.12 h1:xPDB+GSBZq0rJbmDZF+EyfMbnWRyfEPcn7PZ7bJjXSw=\ngithub.com/aws/aws-sdk-go v1.54.12/go.mod h1:eRwEWoyTWFMVYVQzKMNHWP5/RV4xIUGMQfXQHfHkpNU=\ngithub.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=\ngithub.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=\ngithub.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=\ngithub.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b h1:AP/Y7sqYicnjGDfD5VcY4CIfh1hRXBUavxrvELjTiOE=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b/go.mod h1:ac9efd0D1fsDb3EJvhqgXRbFx7bs2wqZ10HQPeU8U/Q=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 h1:bXxS5/Z3/dfc8iFniQfgogNBomo0u+1//9eP+jl8GVo=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748/go.mod h1:l/bIBLeOl9eX+wxJAzxS4TveKRtAqlyDpHjhkfO0MEI=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=\ngithub.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=\ngithub.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=\ngithub.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk=\ngithub.com/cncf/udpa/go v0.0.0-20210930031921-04548b0d99d4/go.mod h1:6pvJx4me5XPnfI9Z40ddWsdw2W/uZgQLFXToKeRcDiI=\ngithub.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210805033703-aa0b78936158/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210922020428-25de7278fc84/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20211011173535-cb28da3451f1/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI=\ngithub.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=\ngithub.com/cristalhq/jwt/v3 v3.1.0 h1:iLeL9VzB0SCtjCy9Kg53rMwTcrNm+GHyVcz2eUujz6s=\ngithub.com/cristalhq/jwt/v3 v3.1.0/go.mod h1:XOnIXst8ozq/esy5N1XOlSyQqBd+84fxJ99FK+1jgL8=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/dgryski/go-farm v0.0.0-20200201041132-a6ae2369ad13/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da h1:aIftn67I1fkbMa512G+w+Pxci9hJPB8oMnkcP3iZF38=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.mod h1:hliV/p42l8fGbc6Y9bQ70uLwIvmJyVE5k4iMKlh8wCQ=\ngithub.com/envoyproxy/go-control-plane v0.9.10-0.20210907150352-cf90f659a021/go.mod h1:AFq3mo9L8Lqqiid3OhADV3RfLJnjiw63cSpi+fDTRC0=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a h1:yDWHCSQ40h88yih2JAcL6Ls/kVkSE8GFACTGVnMPruw=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a/go.mod h1:7Ga40egUymuWXxAe151lTNnCv97MddSOVsjpPPkityA=\ngithub.com/fatih/structtag v1.0.0/go.mod h1:IKitwq45uXL/yqi5mYghiD3w9H6eTOvI9vnk8tXMphA=\ngithub.com/fatih/structtag v1.2.0 h1:/OdNE99OxoI/PqaW/SuSK9uxxT3f/tcSZgon/ssNSx4=\ngithub.com/fatih/structtag v1.2.0/go.mod h1:mBJUNpUnHmRKrKlQQlmCrh5PuhftFbNv8Ys4/aAZl94=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568/go.mod h1:xEzjJPgXI435gkrCt3MPfRiAkVrwSbHsst4LCFVfpJc=\ngithub.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=\ngithub.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=\ngithub.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=\ngithub.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=\ngithub.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=\ngithub.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=\ngithub.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a/go.mod h1:gf4bu3Q80BeJ6H1S1vYPm8/ELATdvryBaNFGgqEef3s=\ngithub.com/gogo/googleapis v1.3.2 h1:kX1es4djPJrsDhY7aZKJy7aZasdcB5oSOEphMjSB53c=\ngithub.com/gogo/googleapis v1.3.2/go.mod h1:5YRNX2z1oM5gXdAkurHa942MDgEJyk02w4OecKY87+c=\ngithub.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.2.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.3.1/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=\ngithub.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=\ngithub.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=\ngithub.com/gogo/status v1.1.0 h1:+eIkrewn5q6b30y+g/BJINVVdi2xH7je5MPJ3ZPK3JA=\ngithub.com/gogo/status v1.1.0/go.mod h1:BFv9nrluPLmrS0EmGVvLaPNmRosr9KapBYd5/hpY1WM=\ngithub.com/golang-jwt/jwt/v5 v5.2.0 h1:d/ix8ftRUorsN+5eMIlF4T6J8CAt9rch3My2winC1Jw=\ngithub.com/golang-jwt/jwt/v5 v5.2.0/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.4.0/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=\ngithub.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=\ngithub.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.3-0.20190920234318-1680a479a2cf/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=\ngithub.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=\ngithub.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=\ngithub.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=\ngithub.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=\ngithub.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=\ngithub.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=\ngithub.com/google/gofuzz v1.0.0 h1:A8PeW59pxE9IoFRqBp37U+mSNaQoZ46F1f0f863XSXw=\ngithub.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=\ngithub.com/google/martian/v3 v3.3.2 h1:IqNFLAmvJOgVlpdEBiQbDc2EwKW77amAycfTuWKdfvw=\ngithub.com/google/martian/v3 v3.3.2/go.mod h1:oBOf6HBosgwRXnUGWUB05QECsc6uvmMiJ3+6W4l/CUk=\ngithub.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=\ngithub.com/google/s2a-go v0.1.4 h1:1kZ/sQM3srePvKs3tXAvQzo66XfcReoqFpIpIccE7Oc=\ngithub.com/google/s2a-go v0.1.4/go.mod h1:Ej+mSEMGRnqRzjc7VtF+jdBwYG5fuJfiZ8ELkjEwM0A=\ngithub.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.4 h1:uGy6JWR/uMIILU8wbf+OkstIrNiMjGpEIyhx8f6W7s4=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.4/go.mod h1:AwSRAtLfXpU5Nm3pW+v7rGDHp09LsPtGY9MduiEsR9k=\ngithub.com/googleapis/gax-go/v2 v2.12.0 h1:A+gCJKdRfqXkr+BIRGtZLibNXf0m1f9E4HG56etFpas=\ngithub.com/googleapis/gax-go/v2 v2.12.0/go.mod h1:y+aIqrI5eb1YGMVJfuV3185Ts/D7qKpsEkdD5+I6QGU=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw=\ngithub.com/jessevdk/go-flags v1.4.0 h1:4IU2WS7AumrZ/40jfhf4QVDMsQwqA7VEHozFRrGARJA=\ngithub.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=\ngithub.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=\ngithub.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=\ngithub.com/jonboulle/clockwork v0.5.0 h1:Hyh9A8u51kptdkR+cqRpT1EebBwTn1oK9YfGYbdFz6I=\ngithub.com/jonboulle/clockwork v0.5.0/go.mod h1:3mZlmanh0g2NDKO5TWZVJAfofYk64M7XN3SzBPjZF60=\ngithub.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=\ngithub.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=\ngithub.com/json-iterator/go v1.1.8/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.11/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=\ngithub.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=\ngithub.com/kisielk/errcheck v1.2.0/go.mod h1:/BMXB+zMLi60iA8Vv6Ksmxu/1UDYcXs4uQLJ+jE2L00=\ngithub.com/kisielk/errcheck v1.5.0 h1:e8esj/e4R+SAOwFwN+n3zr0nYeCyeweozKfO23MvHzY=\ngithub.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=\ngithub.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=\ngithub.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=\ngithub.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=\ngithub.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=\ngithub.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=\ngithub.com/m3db/prometheus_client_golang v0.8.1 h1:t7w/tcFws81JL1j5sqmpqcOyQOpH4RDOmIe3A3fdN3w=\ngithub.com/m3db/prometheus_client_golang v0.8.1/go.mod h1:8R/f1xYhXWq59KD/mbRqoBulXejss7vYtYzWmruNUwI=\ngithub.com/m3db/prometheus_client_model v0.1.0 h1:cg1+DiuyT6x8h9voibtarkH1KT6CmsewBSaBhe8wzLo=\ngithub.com/m3db/prometheus_client_model v0.1.0/go.mod h1:Qfsxn+LypxzF+lNhak7cF7k0zxK7uB/ynGYoj80zcD4=\ngithub.com/m3db/prometheus_common v0.1.0 h1:YJu6eCIV6MQlcwND24cRG/aRkZDX1jvYbsNNs1ZYr0w=\ngithub.com/m3db/prometheus_common v0.1.0/go.mod h1:EBmDQaMAy4B8i+qsg1wMXAelLNVbp49i/JOeVszQ/rs=\ngithub.com/m3db/prometheus_procfs v0.8.1 h1:LsxWzVELhDU9sLsZTaFLCeAwCn7bC7qecZcK4zobs/g=\ngithub.com/m3db/prometheus_procfs v0.8.1/go.mod h1:N8lv8fLh3U3koZx1Bnisj60GYUMDpWb09x1R+dmMOJo=\ngithub.com/mattn/go-shellwords v1.0.10/go.mod h1:EZzvwXDESEeg03EKmM+RmDnNOPKG4lLtQsUlTZDWQ8Y=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 h1:I0XW9+e1XWDxdcEniV4rQAIOPUGDq67JSCiRCgGCZLI=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=\ngithub.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=\ngithub.com/opentracing/opentracing-go v1.2.0 h1:uEJPy/1a5RIPAJ0Ov+OIO8OxWu77jEv+1B0VhjKrZUs=\ngithub.com/opentracing/opentracing-go v1.2.0/go.mod h1:GxEUsuufX4nBwe+T+Wl9TAgYrxe9dPLANfrWvHYVTgc=\ngithub.com/pborman/uuid v0.0.0-20160209185913-a97ce2ca70fa/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709 h1:zNBQb37RGLmJybyMcs983HfUfpkw9OTFD9tbBfAViHE=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a h1:AA9vgIBDjMHPC2McaGPojgV2dcI78ZC0TLNhYCXEKH8=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a/go.mod h1:lzZQ3Noex5pfAy7mkAeCjcBDteYU85uWWnJ/y6gKU8k=\ngithub.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=\ngithub.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo=\ngithub.com/prometheus/client_golang v1.3.0/go.mod h1:hJaj2vgQTGQmVCsAACORcieXFeDPbaTKGT+JTgUa3og=\ngithub.com/prometheus/client_golang v1.4.1/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=\ngithub.com/prometheus/client_golang v1.7.1/go.mod h1:PY5Wy2awLA44sXw4AOSfFBetzPP4j5+D6mVACh+pe2M=\ngithub.com/prometheus/client_golang v1.11.1 h1:+4eQaD7vAZ6DsfsxB15hbE0odUjGI5ARs9yskGu1v4s=\ngithub.com/prometheus/client_golang v1.11.1/go.mod h1:Z6t4BnS23TR94PD6BsDNk8yVqroYurpAkEiz0P2BEV0=\ngithub.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=\ngithub.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.1.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.4.0 h1:5lQXD3cAg1OXBf4Wq03gTrXHeaV0TQvGfUooCfx1yqY=\ngithub.com/prometheus/client_model v0.4.0/go.mod h1:oMQmHW1/JoDwqLtg57MGgP/Fb1CJEYF2imWWhWtMkYU=\ngithub.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=\ngithub.com/prometheus/common v0.7.0/go.mod h1:DjGbpBbp5NYNiECxcL/VnbXCCaQpKd3tt26CguLLsqA=\ngithub.com/prometheus/common v0.8.0/go.mod h1:PC/OgXc+UN7B4ALwvn1yzVZmVwvhXp5JsbBv6wSv6i0=\ngithub.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4=\ngithub.com/prometheus/common v0.10.0/go.mod h1:Tlit/dnDKsSWFlCLTWaA1cyBgKHSMdTB80sz/V91rCo=\ngithub.com/prometheus/common v0.26.0 h1:iMAkS2TDoNWnKM+Kopnx/8tnEStIfpYA0ur0xQzzhMQ=\ngithub.com/prometheus/common v0.26.0/go.mod h1:M7rCNAaPfAosfx8veZJCuw84e35h3Cfd9VFqTh1DIvc=\ngithub.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=\ngithub.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=\ngithub.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.0.9/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.1.3/go.mod h1:lV6e/gmhEcM9IjHGsFOCxxuZ+z1YqCvr4OA4YeYWdaU=\ngithub.com/prometheus/procfs v0.6.0 h1:mxy4L2jP6qMonqmq+aTtOx1ifVWUgG/TAmntgbh3xv4=\ngithub.com/prometheus/procfs v0.6.0/go.mod h1:cz+aTbrPOrUb4q7XlbU9ygM+/jj0fzG6c1xBZuNvfVA=\ngithub.com/robfig/cron v1.2.0 h1:ZjScXvvxeQ63Dbyxy76Fj3AT3Ut0aKsyd2/tl3DTMuQ=\ngithub.com/robfig/cron v1.2.0/go.mod h1:JGuDeoQd7Z6yL4zQhZ3OPEVHB7fL6Ka6skscFHfmt2k=\ngithub.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=\ngithub.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=\ngithub.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=\ngithub.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.6.1 h1:/FiVV8dS/e+YqF2JvO3yXRFbBLTIuSDkuC7aBOAvL+k=\ngithub.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af h1:EiWVfh8mr40yFZEui2oF0d45KgH48PkB2H0Z0GANvSI=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af/go.mod h1:Vrkh1pnjV9Bl8c3P9zH0/D4NlOHWP5d4/hF4YTULaec=\ngithub.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=\ngithub.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=\ngithub.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25 h1:7z3LSn867ex6VSaahyKadf4WtSsJIgne6A1WLOAGM8A=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25/go.mod h1:lbP8tGiBjZ5YWIc2fzuRpTaz0b/53vT6PEs3QuAWzuU=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=\ngithub.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=\ngithub.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=\ngithub.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=\ngithub.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=\ngithub.com/uber-common/bark v1.2.1/go.mod h1:g0ZuPcD7XiExKHynr93Q742G/sbrdVQkghrqLGOoFuY=\ngithub.com/uber-go/mapdecode v1.0.0 h1:euUEFM9KnuCa1OBixz1xM+FIXmpixyay5DLymceOVrU=\ngithub.com/uber-go/mapdecode v1.0.0/go.mod h1:b5nP15FwXTgpjTjeA9A2uTHXV5UJCl4arwKpP0FP1Hw=\ngithub.com/uber-go/tally v3.3.12+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber-go/tally v3.3.15+incompatible h1:9hLSgNBP28CjIaDmAuRTq9qV+UZY+9PcvAkXO4nNMwg=\ngithub.com/uber-go/tally v3.3.15+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber/cadence-idl v0.0.0-20211111101836-d6b70b60eb8c/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible h1:NHcubEkVbahf9t3p75TOCR83gdUHXjRJvjoBh1yACsM=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible/go.mod h1:WVhlPFC8FDjOFMMWRy2pZqQJSXxYSwNYOkTr/Z6d3Kk=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible h1:MxZXOiR2JuoANZ3J6DE/U0kSFv/eJ/GfSYVCjK7dyaw=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible/go.mod h1:ComeNDZlWwrWnDv8aPp0Ba6+uUTzImX/AauajbLI56U=\ngithub.com/uber/ringpop-go v0.8.5 h1:aBa/SHmmFRcAXA63k7uBheoTL8tCmH7L+OgktB1AF/o=\ngithub.com/uber/ringpop-go v0.8.5/go.mod h1:zVI6eGO6L7pG14GkntHsSOfmUAWQ7B4lvmzly4IT4ls=\ngithub.com/uber/tchannel-go v1.16.0/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/uber/tchannel-go v1.22.2 h1:NKA5FVESYh6Ij6V+tujK+IFZnBKDyUHdsBY264UYhgk=\ngithub.com/uber/tchannel-go v1.22.2/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2 h1:zzrxE1FKn5ryBNl9eKOeqQ58Y/Qpo3Q9QNxKHX5uzzQ=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2/go.mod h1:hzfGeIUDq/j97IG+FhNqkowIyEcD88LrW6fyU3K3WqY=\ngithub.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=\ngo.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=\ngo.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=\ngo.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=\ngo.uber.org/atomic v1.3.2/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=\ngo.uber.org/atomic v1.5.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.5.1/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/atomic v1.10.0 h1:9qC72Qh0+3MqyJbAn8YU5xVq1frD8bn3JtD2oXtafVQ=\ngo.uber.org/atomic v1.10.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=\ngo.uber.org/cadence v0.19.0 h1:EvDXwIJ0lAAxL2i8ne/vG/TeoJM6xkAyqgTRFmIWG+c=\ngo.uber.org/cadence v0.19.0/go.mod h1:s91dOf0kcJbumPscRIVFV/4Xq/exhefzpXmnDiRRTxs=\ngo.uber.org/config v1.4.0 h1:upnMPpMm6WlbZtXoasNkK4f0FhxwS+W4Iqz5oNznehQ=\ngo.uber.org/config v1.4.0/go.mod h1:aCyrMHmUAc/s2h9sv1koP84M9ZF/4K+g2oleyESO/Ig=\ngo.uber.org/dig v1.8.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.10.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.18.0 h1:imUL1UiY0Mg4bqbFfsRQO5G4CGRBec/ZujWTvSVp3pw=\ngo.uber.org/dig v1.18.0/go.mod h1:Us0rSJiThwCv2GteUN0Q7OKvU7n5J4dxZ9JKUXozFdE=\ngo.uber.org/fx v1.10.0/go.mod h1:vLRicqpG/qQEzno4SYU86iCwfT95EZza+Eba0ItuxqY=\ngo.uber.org/fx v1.13.1/go.mod h1:bREWhavnedxpJeTq9pQT53BbvwhUv7TcpsOqcH4a+3w=\ngo.uber.org/fx v1.23.0 h1:lIr/gYWQGfTwGcSXWXu4vP5Ws6iqnNEIY+F/aFzCKTg=\ngo.uber.org/fx v1.23.0/go.mod h1:o/D9n+2mLP6v1EG+qsdT1O8wKopYAsqZasju97SDFCU=\ngo.uber.org/goleak v0.10.0/go.mod h1:VCZuO8V8mFPlL0F5J5GK1rtHV3DrFcQ1R8ryq7FK0aI=\ngo.uber.org/goleak v1.0.0/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=\ngo.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=\ngo.uber.org/goleak v1.2.0/go.mod h1:XJYK+MuIchqpmGmUSAzotztawfKvYLUIgg7guXrwVUo=\ngo.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=\ngo.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=\ngo.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=\ngo.uber.org/multierr v1.3.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.4.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=\ngo.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=\ngo.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=\ngo.uber.org/net/metrics v1.3.0 h1:iRLPuVecNYf/wIV+mQaA4IgN8ghifu3q1B4IT6HfwyY=\ngo.uber.org/net/metrics v1.3.0/go.mod h1:pEQrSDGNWT5IVpekWzee5//uHjI4gmgZFkobfw3bv8I=\ngo.uber.org/thriftrw v1.25.0/go.mod h1:IcIfSeZgc59AlYb0xr0DlDKIdD7SgjnFpG9BXCPyy9g=\ngo.uber.org/thriftrw v1.29.2 h1:pRuFLzbGvTcnYwGSjizWRHlbJUzGhu84sRiL1h1kUd8=\ngo.uber.org/thriftrw v1.29.2/go.mod h1:YcjXveberDd28/Bs34SwHy3yu85x/jB4UA2gIcz/Eo0=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee h1:0mgffUl7nfd+FpvXMVz4IDEaUSmT1ysygQC7qYo7sG4=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA=\ngo.uber.org/yarpc v1.55.0/go.mod h1:V2JUPDWHYGNpvyuroYjf0KFjwvBCtcFJLuvZqv7TWA0=\ngo.uber.org/yarpc v1.70.3 h1:yykHwzRD9/bgDtlOWoVuXbSZoU91Id2dWJO1CDSRHnI=\ngo.uber.org/yarpc v1.70.3/go.mod h1:EH6I6K1HxBbOxZIJfhdDf+H+cvXPHmJyRvpfPqES20U=\ngo.uber.org/zap v1.9.1/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=\ngo.uber.org/zap v1.13.0/go.mod h1:zwrFLgMcdUuIBviXEYEH1YKNaOBnKXsx2IPda5bBwHM=\ngo.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo=\ngo.uber.org/zap v1.26.0/go.mod h1:dtElttAiwGvoJ/vj4IwHBS/gXsEu/pZ50mUIRWuG0so=\ngolang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.0.0-20220314234659-1baeb1ce4c0b/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=\ngolang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8=\ngolang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b h1:kLiC65FbiHWFAOu+lxwNPujcsl8VYyTYYEZnsOO1WK4=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b/go.mod h1:iRJReGqOEeBhDZGkGbynYwcHlctCvnjTYIamk7uXpHI=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e h1:qyrTQ++p1afMkO4DPEeLGq/3oTsdlvdH4vqZUBWzUKM=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e/go.mod h1:AbB0pIl9nAr9wVwH+Z2ZpaocVmF5I4GyWCDIsVjR0bk=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20191125180803-fdd1cda4f05f/go.mod h1:5qLYkcX4OjUUV8bRuDixDT3tpyyb+LUpUlRWLxfhWrs=\ngolang.org/x/lint v0.0.0-20200130185559-910be7a94367/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 h1:VLliZ0d+/avPrXXH+OakdXhpJuEoBZuwh1m2j7U6Iug=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc=\ngolang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=\ngolang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=\ngolang.org/x/mod v0.18.0 h1:5+9lSbEzPSdWkH32vYPBwEpX8KwDbM52Ud9xBUvNlb0=\ngolang.org/x/mod v0.18.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=\ngolang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=\ngolang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=\ngolang.org/x/net v0.40.0 h1:79Xs7wF06Gbdcg4kdCCIQArK11Z1hr5POQ6+fIYHNuY=\ngolang.org/x/net v0.40.0/go.mod h1:y0hY0exeL2Pku80/zKK7tpntoX23cqL3Oa6njdgRtds=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.11.0 h1:vPL4xzxBM4niKCW6g9whtaWVXTJf1U5e4aZxxFx/gbU=\ngolang.org/x/oauth2 v0.11.0/go.mod h1:LdF7O/8bLR/qWK9DrpXmbHLTouvRHK0SgJl0GmDBchk=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.14.0 h1:woo0S4Yywslg6hp4eUFjTVOyKt0RookbpAHG4c1HmhQ=\ngolang.org/x/sync v0.14.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191220142924-d4481acd189f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200117145432-59e60aa80a0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200212091648-12a6c2dcc1e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200615200032-f1bc736245b1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200625212154-ddb9806d33ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=\ngolang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=\ngolang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=\ngolang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/text v0.0.0-20170915032832-14c0d48ead0c/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=\ngolang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=\ngolang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=\ngolang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=\ngolang.org/x/time v0.0.0-20170927054726-6dc17368e09b/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=\ngolang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191030062658-86caa796c7ab/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191104232314-dc038396d1f0/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191108193012-7d206e10da11/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191114200427-caa0b0f7d508/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191125144606-a911d9008d1f/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191226212025-6b505debf4bc/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117215004-fe56e6335763/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200216192241-b320d3a0f5a2/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.11-0.20220513221640-090b14e8501f/go.mod h1:SgwaegtQh8clINPpECJMqnxLv9I09HLqnW3RMqW0CA4=\ngolang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=\ngolang.org/x/tools v0.22.0 h1:gqSGLZqv+AI9lIQzniJ0nZDRG5GBPsSi+DRNHWNz6yA=\ngolang.org/x/tools v0.22.0/go.mod h1:aCwcsjqvq7Yqt6TNyX7QMU2enbQ/Gt0bo6krSeEri+c=\ngolang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 h1:H2TDz8ibqkAF6YGhCdN3jS9O0/s90v0rJh3X/OLHEUk=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2/go.mod h1:K8+ghG5WaK9qNqU5K3HdILfMLy1f3aNYFI/wnl100a8=\ngoogle.golang.org/api v0.128.0 h1:RjPESny5CnQRn9V6siglged+DZCgfu9l6mO9dkX9VOg=\ngoogle.golang.org/api v0.128.0/go.mod h1:Y611qgqaE92On/7g65MQgxYul3c0rEB894kniWLY750=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/appengine v1.6.7 h1:FZR1q0exgwxzPzp/aF+VccGrSfxfPpkBqjIIEq3ru6c=\ngoogle.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=\ngoogle.golang.org/genproto v0.0.0-20180518175338-11a468237815/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20200212174721-66ed5ce911ce/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b h1:+YaDE2r2OG8t/z5qmsh7Y+XXwCbvadxxZ0YY6mTdrVA=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b/go.mod h1:CgAqfJo+Xmu0GwA0411Ht3OU3OntXwsGmrmjI8ioGXI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a h1:myvhA4is3vrit1a6NZCWBIwN0kNEnX21DJOJX/NvIfI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a/go.mod h1:SUBoKXbI1Efip18FClrQVGjWcyd0QZd8KkvdP34t7ww=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 h1:AB/lmRny7e2pLhFEYIbl5qkDAUt2h0ZRO4wGPhZf+ik=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405/go.mod h1:67X1fPuzjcrkymZzZV1vvkFeTn2Rvc6lYF9MYFGCcwE=\ngoogle.golang.org/grpc v1.12.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.28.0/go.mod h1:rpkK4SK4GF4Ach/+MFLZUBavHOvF2JJB5uozKKal+60=\ngoogle.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0=\ngoogle.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc=\ngoogle.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.40.1/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34=\ngoogle.golang.org/grpc v1.45.0/go.mod h1:lN7owxKUQEqMfSyQikvvk5tf/6zMPsrK+ONuO11+0rQ=\ngoogle.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk=\ngoogle.golang.org/grpc v1.59.0/go.mod h1:aUPDwccQo6OTjy7Hct4AfBPD1GptF4fyUjIkQ9YtF98=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=\ngoogle.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=\ngoogle.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=\ngoogle.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=\ngopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19 h1:WB265cn5OpO+hK3pikC9hpP1zI/KTwmyMFKloW9eOVc=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19/go.mod h1:o4V0GXN9/CAmCsvJ0oXYZvrZOe7syiDZSN1GWGZTGzc=\ngopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=\nhonnef.co/go/tools v0.3.2 h1:ytYb4rOqyp1TSa2EPvNVwtPQJctSELKaMyLfqNP4+34=\nhonnef.co/go/tools v0.3.2/go.mod h1:jzwdWgg7Jdq75wlfblQxO4neNaFFSvgc1tD5Wv8U0Yw=\nrsc.io/quote/v3 v3.1.0/go.mod h1:yEA65RcK8LyAZtP9Kv3t0HmxON59tX3rD+tICJqUlj0=\nrsc.io/sampler v1.3.0/go.mod h1:T1hPZKmBbMNahiBKFy5HrXp6adAjACjK9JXDnKaTXpA=\n"
  },
  {
    "path": "common/archiver/gcloud/historyArchiver.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"context\"\n\t\"encoding/binary\"\n\t\"errors\"\n\t\"path/filepath\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\terrUploadNonRetriable = errors.New(\"upload non-retriable error\")\n)\n\nconst (\n\t// URIScheme is the scheme for the gcloud storage implementation\n\tURIScheme = \"gs\"\n\tConfigKey = \"gstorage\"\n\n\ttargetHistoryBlobSize = 2 * 1024 * 1024 // 2MB\n\terrEncodeHistory      = \"failed to encode history batches\"\n\terrBucketHistory      = \"failed to get google storage bucket handle\"\n\terrWriteFile          = \"failed to write history to google storage\"\n)\n\ntype historyArchiver struct {\n\tcontainer     *archiver.HistoryBootstrapContainer\n\tgcloudStorage connector.Client\n\n\t// only set in test code\n\thistoryIterator archiver.HistoryIterator\n}\n\ntype progress struct {\n\tCurrentPageNumber int\n\tIteratorState     []byte\n}\n\ntype getHistoryToken struct {\n\tCloseFailoverVersion int64\n\tHighestPart          int\n\tCurrentPart          int\n\tBatchIdxOffset       int\n}\n\n// NewHistoryArchiver creates a new gcloud storage HistoryArchiver\nfunc NewHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\tconfig connector.Config,\n) (archiver.HistoryArchiver, error) {\n\tstorage, err := connector.NewClient(context.Background(), config)\n\tif err == nil {\n\t\treturn newHistoryArchiver(container, nil, storage), nil\n\t}\n\treturn nil, err\n}\n\nfunc newHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\thistoryIterator archiver.HistoryIterator,\n\tstorage connector.Client,\n) archiver.HistoryArchiver {\n\treturn &historyArchiver{\n\t\tcontainer:       container,\n\t\tgcloudStorage:   storage,\n\t\thistoryIterator: historyIterator,\n\t}\n}\n\n// Archive is used to archive a workflow history. When the context expires the method should stop trying to archive.\n// Implementors are free to archive however they want, including implementing retries of sub-operations. The URI defines\n// the resource that histories should be archived into. The implementor gets to determine how to interpret the URI.\n// The Archive method may or may not be automatically retried by the caller. The ArchiveOptions are used\n// to interact with these retries including giving the implementor the ability to cancel retries and record progress\n// between retry attempts.\n// This method will be invoked after a workflow passes its retention period.\nfunc (h *historyArchiver) Archive(ctx context.Context, URI archiver.URI, request *archiver.ArchiveHistoryRequest, opts ...archiver.ArchiveOption) (err error) {\n\tscope := h.container.MetricsClient.Scope(metrics.HistoryArchiverScope, metrics.DomainTag(request.DomainName))\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\n\t\t\tif err.Error() != errUploadNonRetriable.Error() {\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverArchiveTransientErrorCount)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tscope.IncCounter(metrics.HistoryArchiverArchiveNonRetryableErrorCount)\n\t\t\tif featureCatalog.NonRetriableError != nil {\n\t\t\t\terr = featureCatalog.NonRetriableError()\n\t\t\t}\n\n\t\t}\n\t}()\n\n\tlogger := archiver.TagLoggerWithArchiveHistoryRequestAndURI(h.container.Logger, request, URI.String())\n\n\tif err := h.ValidateURI(URI); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\treturn errUploadNonRetriable\n\t}\n\n\tif err := archiver.ValidateHistoryArchiveRequest(request); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidArchiveRequest), tag.Error(err))\n\t\treturn errUploadNonRetriable\n\t}\n\n\tvar totalUploadSize int64\n\thistoryIterator := h.historyIterator\n\tvar progress progress\n\tif historyIterator == nil { // will only be set by testing code\n\t\thistoryIterator, _ = loadHistoryIterator(ctx, request, h.container.HistoryV2Manager, featureCatalog, &progress)\n\t}\n\n\tfor historyIterator.HasNext() {\n\t\tpart := progress.CurrentPageNumber\n\t\thistoryBlob, err := getNextHistoryBlob(ctx, historyIterator)\n\n\t\tif err != nil {\n\t\t\tif common.IsEntityNotExistsError(err) {\n\t\t\t\t// workflow history no longer exists, may due to duplicated archival signal\n\t\t\t\t// this may happen even in the middle of iterating history as two archival signals\n\t\t\t\t// can be processed concurrently.\n\t\t\t\tlogger.Info(archiver.ArchiveSkippedInfoMsg)\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverDuplicateArchivalsCount)\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tlogger := logger.WithTags(tag.ArchivalArchiveFailReason(archiver.ErrReasonReadHistory), tag.Error(err))\n\t\t\tif !persistence.IsTransientError(err) {\n\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg)\n\t\t\t\treturn errUploadNonRetriable\n\t\t\t}\n\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg)\n\t\t\treturn err\n\t\t}\n\n\t\tif archiver.IsHistoryMutated(request, historyBlob.Body, *historyBlob.Header.IsLast, logger) {\n\t\t\tif !featureCatalog.ArchiveIncompleteHistory() {\n\t\t\t\treturn archiver.ErrHistoryMutated\n\t\t\t}\n\n\t\t}\n\n\t\tencodedHistoryPart, err := encode(historyBlob.Body)\n\t\tif err != nil {\n\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errEncodeHistory), tag.Error(err))\n\t\t\treturn errUploadNonRetriable\n\t\t}\n\n\t\tfilename := constructHistoryFilenameMultipart(request.DomainID, request.WorkflowID, request.RunID, request.CloseFailoverVersion, part)\n\t\tif exist, _ := h.gcloudStorage.Exist(ctx, URI, filename); !exist {\n\t\t\tif err := h.gcloudStorage.Upload(ctx, URI, filename, encodedHistoryPart); err != nil {\n\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(errWriteFile), tag.Error(err))\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverArchiveTransientErrorCount)\n\t\t\t\treturn err\n\t\t\t}\n\n\t\t\ttotalUploadSize = totalUploadSize + int64(binary.Size(encodedHistoryPart))\n\t\t}\n\n\t\tsaveHistoryIteratorState(ctx, featureCatalog, historyIterator, part, &progress)\n\t}\n\n\tscope.AddCounter(metrics.HistoryArchiverTotalUploadSize, totalUploadSize)\n\tscope.AddCounter(metrics.HistoryArchiverHistorySize, totalUploadSize)\n\tscope.IncCounter(metrics.HistoryArchiverArchiveSuccessCount)\n\treturn\n}\n\n// Get is used to access an archived history. When context expires method should stop trying to fetch history.\n// The URI identifies the resource from which history should be accessed and it is up to the implementor to interpret this URI.\n// This method should thrift errors - see filestore as an example.\nfunc (h *historyArchiver) Get(ctx context.Context, URI archiver.URI, request *archiver.GetHistoryRequest) (*archiver.GetHistoryResponse, error) {\n\n\terr := h.ValidateURI(URI)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateGetRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidGetHistoryRequest.Error()}\n\t}\n\n\tvar token *getHistoryToken\n\tif request.NextPageToken != nil {\n\t\ttoken, err = deserializeGetHistoryToken(request.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: archiver.ErrNextPageTokenCorrupted.Error()}\n\t\t}\n\t} else {\n\t\thighestVersion, historyhighestPart, historyCurrentPart, err := h.getHighestVersion(ctx, URI, request)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\ttoken = &getHistoryToken{\n\t\t\tCloseFailoverVersion: *highestVersion,\n\t\t\tHighestPart:          *historyhighestPart,\n\t\t\tCurrentPart:          *historyCurrentPart,\n\t\t\tBatchIdxOffset:       0,\n\t\t}\n\t}\n\n\tresponse := &archiver.GetHistoryResponse{}\n\tresponse.HistoryBatches = []*types.History{}\n\tnumOfEvents := 0\n\nouter:\n\tfor token.CurrentPart <= token.HighestPart {\n\n\t\tfilename := constructHistoryFilenameMultipart(request.DomainID, request.WorkflowID, request.RunID, token.CloseFailoverVersion, token.CurrentPart)\n\t\tencodedHistoryBatches, err := h.gcloudStorage.Get(ctx, URI, filename)\n\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\tif encodedHistoryBatches == nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: \"Fail retrieving history file: \" + URI.String() + \"/\" + filename}\n\t\t}\n\n\t\tbatches, err := decodeHistoryBatches(encodedHistoryBatches)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\t// trim the batches in the beginning based on token.BatchIdxOffset\n\t\tbatches = batches[token.BatchIdxOffset:]\n\n\t\tfor idx, batch := range batches {\n\t\t\tresponse.HistoryBatches = append(response.HistoryBatches, batch)\n\t\t\ttoken.BatchIdxOffset++\n\t\t\tnumOfEvents += len(batch.Events)\n\n\t\t\tif numOfEvents >= request.PageSize {\n\t\t\t\tif idx == len(batches)-1 {\n\t\t\t\t\t// handle the edge case where page size is meeted after adding the last batch\n\t\t\t\t\ttoken.BatchIdxOffset = 0\n\t\t\t\t\ttoken.CurrentPart++\n\t\t\t\t}\n\t\t\t\tbreak outer\n\t\t\t}\n\t\t}\n\n\t\t// reset the offset to 0 as we will read a new page\n\t\ttoken.BatchIdxOffset = 0\n\t\ttoken.CurrentPart++\n\n\t}\n\n\tif token.CurrentPart <= token.HighestPart {\n\t\tnextToken, err := serializeToken(token)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\tresponse.NextPageToken = nextToken\n\t}\n\n\treturn response, nil\n}\n\n// ValidateURI is used to define what a valid URI for an implementation is.\nfunc (h *historyArchiver) ValidateURI(URI archiver.URI) (err error) {\n\n\tif err = h.validateURI(URI); err == nil {\n\t\t_, err = h.gcloudStorage.Exist(context.Background(), URI, \"\")\n\t}\n\n\treturn\n}\n\nfunc (h *historyArchiver) validateURI(URI archiver.URI) (err error) {\n\tif URI.Scheme() != URIScheme {\n\t\treturn archiver.ErrURISchemeMismatch\n\t}\n\n\tif URI.Path() == \"\" || URI.Hostname() == \"\" {\n\t\treturn archiver.ErrInvalidURI\n\t}\n\n\treturn\n}\n\nfunc getNextHistoryBlob(ctx context.Context, historyIterator archiver.HistoryIterator) (*archiver.HistoryBlob, error) {\n\thistoryBlob, err := historyIterator.Next()\n\top := func(ctx context.Context) error {\n\t\thistoryBlob, err = historyIterator.Next()\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreatePersistenceRetryPolicy()),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\tfor err != nil {\n\t\tif contextExpired(ctx) {\n\t\t\treturn nil, archiver.ErrContextTimeout\n\t\t}\n\t\tif !persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\terr = throttleRetry.Do(ctx, op)\n\t}\n\treturn historyBlob, nil\n}\n\n// with XDC(global domain) concept, archival may write different history with the same RunID, with different failoverVersion.\n// In that case, the history/runID with the highest failoverVersion wins.\n// getHighestVersion look up all archived files to find the highest failoverVersion.\n// Since a history is written into different parts in this archival implementation, it also returns the highest and lowest partVersionID.\nfunc (h *historyArchiver) getHighestVersion(ctx context.Context, URI archiver.URI, request *archiver.GetHistoryRequest) (*int64, *int, *int, error) {\n\n\tfilenames, err := h.gcloudStorage.Query(ctx, URI, constructHistoryFilenamePrefix(request.DomainID, request.WorkflowID, request.RunID))\n\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\n\tvar highestVersion *int64\n\tvar highestVersionPart *int\n\tvar lowestVersionPart *int\n\n\tfor _, filename := range filenames {\n\t\tversion, partVersionID, err := extractCloseFailoverVersion(filepath.Base(filename))\n\t\tif err != nil || (request.CloseFailoverVersion != nil && version != *request.CloseFailoverVersion) {\n\t\t\tcontinue\n\t\t}\n\n\t\tif highestVersion == nil || version > *highestVersion {\n\t\t\thighestVersion = &version\n\t\t\thighestVersionPart = new(int)\n\t\t\tlowestVersionPart = new(int)\n\t\t}\n\n\t\tif *highestVersion == version {\n\t\t\tif highestVersionPart == nil || partVersionID > *highestVersionPart {\n\t\t\t\thighestVersionPart = &partVersionID\n\t\t\t}\n\n\t\t\tif lowestVersionPart == nil || partVersionID < *lowestVersionPart {\n\t\t\t\tlowestVersionPart = &partVersionID\n\t\t\t}\n\t\t}\n\n\t}\n\n\tif highestVersion == nil {\n\t\treturn nil, nil, nil, archiver.ErrHistoryNotExist\n\t}\n\treturn highestVersion, highestVersionPart, lowestVersionPart, nil\n}\n\nfunc loadHistoryIterator(ctx context.Context, request *archiver.ArchiveHistoryRequest, historyManager persistence.HistoryManager, featureCatalog *archiver.ArchiveFeatureCatalog, progress *progress) (historyIterator archiver.HistoryIterator, err error) {\n\n\tdefer func() {\n\t\tif err != nil || historyIterator == nil {\n\t\t\thistoryIterator, err = archiver.NewHistoryIteratorFromState(ctx, request, historyManager, targetHistoryBlobSize, nil)\n\t\t}\n\t}()\n\n\tif featureCatalog.ProgressManager != nil {\n\t\tif featureCatalog.ProgressManager.HasProgress(ctx) {\n\t\t\terr = featureCatalog.ProgressManager.LoadProgress(ctx, &progress)\n\t\t\tif err == nil {\n\t\t\t\thistoryIterator, err = archiver.NewHistoryIteratorFromState(ctx, request, historyManager, targetHistoryBlobSize, progress.IteratorState)\n\t\t\t}\n\t\t}\n\n\t}\n\treturn\n}\n\nfunc saveHistoryIteratorState(ctx context.Context, featureCatalog *archiver.ArchiveFeatureCatalog, historyIterator archiver.HistoryIterator, currentPartNum int, progress *progress) (err error) {\n\tvar state []byte\n\tif featureCatalog.ProgressManager != nil {\n\t\tstate, err = historyIterator.GetState()\n\t\tif err == nil {\n\t\t\tprogress.CurrentPageNumber = currentPartNum + 1\n\t\t\tprogress.IteratorState = state\n\n\t\t\terr = featureCatalog.ProgressManager.RecordProgress(ctx, progress)\n\t\t}\n\t}\n\n\treturn err\n}\n"
  },
  {
    "path": "common/archiver/gcloud/historyArchiver_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector/mocks\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestDomainID                  = \"test-domain-id\"\n\ttestDomainName                = \"test-domain-name\"\n\ttestWorkflowID                = \"test-workflow-id\"\n\ttestRunID                     = \"test-run-id\"\n\ttestNextEventID               = 1800\n\ttestCloseFailoverVersion      = 100\n\ttestPageSize                  = 100\n\texampleHistoryRecord          = `[{\"events\":[{\"eventId\":1,\"timestamp\":1576800090315103000,\"eventType\":\"WorkflowExecutionStarted\",\"version\":-24,\"taskId\":5242897,\"workflowExecutionStartedEventAttributes\":{\"workflowType\":{\"name\":\"MobileOnlyWorkflow::processMobileOnly\"},\"taskList\":{\"name\":\"MobileOnly\"},\"input\":\"eyJkbmkiOiI0ODY5NGJmZi04MTU2LTRjZDEtYTYzZi0wZTM0ZDBlYzljMWEiLCJjYXRlZ29yeSI6InVwZGF0ZV9jbGllbnQiLCJhZ2UiOjE4LCJzY29yZSI6MCwic3RhdHVzIjoicGVuZGluZyIsImRlc2NyaXB0aW9uIjoiTG9yZW0gSXBzdW0gaXMgc2ltcGx5IGR1bW15IHRleHQgb2YgdGhlIHByaW50aW5nIGFuZCB0eXBlc2V0dGluZyBpbmR1c3RyeS4gTG9yZW0gSXBzdW0gaGFzIGJlZW4gdGhlIGluZHVzdHJ5XHUwMDI3cyBzdGFuZGFyZCBkdW1teSB0ZXh0IGV2ZXIgc2luY2UgdGhlIDE1MDBzLCB3aGVuIGFuIHVua25vd24gcHJpbnRlciB0b29rIGEgZ2FsbGV5IG9mIHR5cGUgYW5kIHNjcmFtYmxlZCBpdCB0byBtYWtlIGEgdHlwZSBzcGVjaW1lbiBib29rLkl0IGhhcyBzdXJ2aXZlZCBub3Qgb25seSBmaXZlIGNlbnR1cmllcywgYnV0IGFsc28gdGhlIGxlYXAgaW50byBlbGVjdHJvbmljIHR5cGVzZXR0aW5nLCByZW1haW5pbmcgZXNzZW50aWFsbHkgdW5jaGFuZ2VkLkl0IHdhcyBwb3B1bGFyaXNlZCBpbiB0aGUgMTk2MHMgd2l0aCB0aGUgcmVsZWFzZSBvZiBMZXRyYXNldCBzaGVldHMgY29udGFpbmluZyBMb3JlbSBJcHN1bSBwYXNzYWdlcywgYW5kIG1vcmUgcmVjZW50bHkgd2l0aCBkZXNrdG9wIHB1Ymxpc2hpbmcgc29mdHdhcmUgbGlrZSBBbGR1cyBQYWdlTWFrZXIgaW5jbHVkaW5nIHZlcnNpb25zIG9mIExvcmVtIElwc3VtLkl0IGlzIGEgbG9uZyBlc3RhYmxpc2hlZCBmYWN0IHRoYXQgYSByZWFkZXIgd2lsbCBiZSBkaXN0cmFjdGVkIGJ5IHRoZSByZWFkYWJsZSBjb250ZW50IG9mIGEgcGFnZSB3aGVuIGxvb2tpbmcgYXQgaXRzIGxheW91dC4gVGhlIHBvaW50IG9mIHVzaW5nIExvcmVtIElwc3VtIGlzIHRoYXQgaXQgaGFzIGEgbW9yZS1vci1sZXNzIG5vcm1hbCBkaXN0cmlidXRpb24gb2YgbGV0dGVycywgYXMgb3Bwb3NlZCB0byB1c2luZyBcdTAwMjdDb250ZW50IGhlcmUsIGNvbnRlbnQgaGVyZVx1MDAyNywgbWFraW5nIGl0IGxvb2sgbGlrZSByZWFkYWJsZSBFbmdsaXNoLiJ9\",\"executionStartToCloseTimeoutSeconds\":300,\"taskStartToCloseTimeoutSeconds\":60,\"originalExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"identity\":\"\",\"firstExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"attempt\":0,\"firstDecisionTaskBackoffSeconds\":0}}]}]`\n\ttwoEventsExampleHistoryRecord = `[{\"events\":[{\"eventId\":1,\"timestamp\":1576800090315103000,\"eventType\":\"WorkflowExecutionStarted\",\"version\":-24,\"taskId\":5242897,\"workflowExecutionStartedEventAttributes\":{\"workflowType\":{\"name\":\"MobileOnlyWorkflow::processMobileOnly\"},\"taskList\":{\"name\":\"MobileOnly\"},\"input\":\"eyJkbmkiOiI0ODY5NGJmZi04MTU2LTRjZDEtYTYzZi0wZTM0ZDBlYzljMWEiLCJjYXRlZ29yeSI6InVwZGF0ZV9jbGllbnQiLCJhZ2UiOjE4LCJzY29yZSI6MCwic3RhdHVzIjoicGVuZGluZyIsImRlc2NyaXB0aW9uIjoiTG9yZW0gSXBzdW0gaXMgc2ltcGx5IGR1bW15IHRleHQgb2YgdGhlIHByaW50aW5nIGFuZCB0eXBlc2V0dGluZyBpbmR1c3RyeS4gTG9yZW0gSXBzdW0gaGFzIGJlZW4gdGhlIGluZHVzdHJ5XHUwMDI3cyBzdGFuZGFyZCBkdW1teSB0ZXh0IGV2ZXIgc2luY2UgdGhlIDE1MDBzLCB3aGVuIGFuIHVua25vd24gcHJpbnRlciB0b29rIGEgZ2FsbGV5IG9mIHR5cGUgYW5kIHNjcmFtYmxlZCBpdCB0byBtYWtlIGEgdHlwZSBzcGVjaW1lbiBib29rLkl0IGhhcyBzdXJ2aXZlZCBub3Qgb25seSBmaXZlIGNlbnR1cmllcywgYnV0IGFsc28gdGhlIGxlYXAgaW50byBlbGVjdHJvbmljIHR5cGVzZXR0aW5nLCByZW1haW5pbmcgZXNzZW50aWFsbHkgdW5jaGFuZ2VkLkl0IHdhcyBwb3B1bGFyaXNlZCBpbiB0aGUgMTk2MHMgd2l0aCB0aGUgcmVsZWFzZSBvZiBMZXRyYXNldCBzaGVldHMgY29udGFpbmluZyBMb3JlbSBJcHN1bSBwYXNzYWdlcywgYW5kIG1vcmUgcmVjZW50bHkgd2l0aCBkZXNrdG9wIHB1Ymxpc2hpbmcgc29mdHdhcmUgbGlrZSBBbGR1cyBQYWdlTWFrZXIgaW5jbHVkaW5nIHZlcnNpb25zIG9mIExvcmVtIElwc3VtLkl0IGlzIGEgbG9uZyBlc3RhYmxpc2hlZCBmYWN0IHRoYXQgYSByZWFkZXIgd2lsbCBiZSBkaXN0cmFjdGVkIGJ5IHRoZSByZWFkYWJsZSBjb250ZW50IG9mIGEgcGFnZSB3aGVuIGxvb2tpbmcgYXQgaXRzIGxheW91dC4gVGhlIHBvaW50IG9mIHVzaW5nIExvcmVtIElwc3VtIGlzIHRoYXQgaXQgaGFzIGEgbW9yZS1vci1sZXNzIG5vcm1hbCBkaXN0cmlidXRpb24gb2YgbGV0dGVycywgYXMgb3Bwb3NlZCB0byB1c2luZyBcdTAwMjdDb250ZW50IGhlcmUsIGNvbnRlbnQgaGVyZVx1MDAyNywgbWFraW5nIGl0IGxvb2sgbGlrZSByZWFkYWJsZSBFbmdsaXNoLiJ9\",\"executionStartToCloseTimeoutSeconds\":300,\"taskStartToCloseTimeoutSeconds\":60,\"originalExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"identity\":\"\",\"firstExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"attempt\":0,\"firstDecisionTaskBackoffSeconds\":0}},\n\t{\"eventId\":2,\"timestamp\":1576800090315103000,\"eventType\":\"WorkflowExecutionStarted\",\"version\":-24,\"taskId\":5242897,\"workflowExecutionStartedEventAttributes\":{\"workflowType\":{\"name\":\"MobileOnlyWorkflow::processMobileOnly\"},\"taskList\":{\"name\":\"MobileOnly\"},\"input\":\"eyJkbmkiOiI0ODY5NGJmZi04MTU2LTRjZDEtYTYzZi0wZTM0ZDBlYzljMWEiLCJjYXRlZ29yeSI6InVwZGF0ZV9jbGllbnQiLCJhZ2UiOjE4LCJzY29yZSI6MCwic3RhdHVzIjoicGVuZGluZyIsImRlc2NyaXB0aW9uIjoiTG9yZW0gSXBzdW0gaXMgc2ltcGx5IGR1bW15IHRleHQgb2YgdGhlIHByaW50aW5nIGFuZCB0eXBlc2V0dGluZyBpbmR1c3RyeS4gTG9yZW0gSXBzdW0gaGFzIGJlZW4gdGhlIGluZHVzdHJ5XHUwMDI3cyBzdGFuZGFyZCBkdW1teSB0ZXh0IGV2ZXIgc2luY2UgdGhlIDE1MDBzLCB3aGVuIGFuIHVua25vd24gcHJpbnRlciB0b29rIGEgZ2FsbGV5IG9mIHR5cGUgYW5kIHNjcmFtYmxlZCBpdCB0byBtYWtlIGEgdHlwZSBzcGVjaW1lbiBib29rLkl0IGhhcyBzdXJ2aXZlZCBub3Qgb25seSBmaXZlIGNlbnR1cmllcywgYnV0IGFsc28gdGhlIGxlYXAgaW50byBlbGVjdHJvbmljIHR5cGVzZXR0aW5nLCByZW1haW5pbmcgZXNzZW50aWFsbHkgdW5jaGFuZ2VkLkl0IHdhcyBwb3B1bGFyaXNlZCBpbiB0aGUgMTk2MHMgd2l0aCB0aGUgcmVsZWFzZSBvZiBMZXRyYXNldCBzaGVldHMgY29udGFpbmluZyBMb3JlbSBJcHN1bSBwYXNzYWdlcywgYW5kIG1vcmUgcmVjZW50bHkgd2l0aCBkZXNrdG9wIHB1Ymxpc2hpbmcgc29mdHdhcmUgbGlrZSBBbGR1cyBQYWdlTWFrZXIgaW5jbHVkaW5nIHZlcnNpb25zIG9mIExvcmVtIElwc3VtLkl0IGlzIGEgbG9uZyBlc3RhYmxpc2hlZCBmYWN0IHRoYXQgYSByZWFkZXIgd2lsbCBiZSBkaXN0cmFjdGVkIGJ5IHRoZSByZWFkYWJsZSBjb250ZW50IG9mIGEgcGFnZSB3aGVuIGxvb2tpbmcgYXQgaXRzIGxheW91dC4gVGhlIHBvaW50IG9mIHVzaW5nIExvcmVtIElwc3VtIGlzIHRoYXQgaXQgaGFzIGEgbW9yZS1vci1sZXNzIG5vcm1hbCBkaXN0cmlidXRpb24gb2YgbGV0dGVycywgYXMgb3Bwb3NlZCB0byB1c2luZyBcdTAwMjdDb250ZW50IGhlcmUsIGNvbnRlbnQgaGVyZVx1MDAyNywgbWFraW5nIGl0IGxvb2sgbGlrZSByZWFkYWJsZSBFbmdsaXNoLiJ9\",\"executionStartToCloseTimeoutSeconds\":300,\"taskStartToCloseTimeoutSeconds\":60,\"originalExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"identity\":\"\",\"firstExecutionRunId\":\"1fd5d4c8-1590-4a0a-8027-535e8729de8e\",\"attempt\":0,\"firstDecisionTaskBackoffSeconds\":0}}]}]`\n)\n\nvar (\n\ttestBranchToken = []byte{1, 2, 3}\n)\n\nfunc (h *historyArchiverSuite) SetupTest() {\n\th.Assertions = require.New(h.T())\n\th.container = &archiver.HistoryBootstrapContainer{\n\t\tLogger:        testlogger.New(h.T()),\n\t\tMetricsClient: metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t}\n\th.testArchivalURI, _ = archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n}\n\nfunc TestHistoryArchiverSuite(t *testing.T) {\n\tsuite.Run(t, new(historyArchiverSuite))\n}\n\ntype historyArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\tcontainer       *archiver.HistoryBootstrapContainer\n\ttestArchivalURI archiver.URI\n}\n\nfunc getCanceledContext() context.Context {\n\tctx, cancel := context.WithCancel(context.Background())\n\tcancel()\n\treturn ctx\n}\n\nfunc (h *historyArchiverSuite) TestValidateURI() {\n\tctx := context.Background()\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs:my-bucket-cad/cadence_archival/development\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://my-bucket-cad\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs:/my-bucket-cad/cadence_archival/development\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://my-bucket-cad/cadence_archival/development\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, mock.Anything, \"\").Return(false, nil)\n\thistoryArchiver := new(historyArchiver)\n\thistoryArchiver.gcloudStorage = storageWrapper\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\th.NoError(err)\n\t\th.Equal(tc.expectedErr, historyArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_InvalidURI() {\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\n\tmockCtrl := gomock.NewController(h.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\th.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_InvalidRequest() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\tmockCtrl := gomock.NewController(h.T())\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           \"\",\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\n\terr = historyArchiver.Archive(ctx, h.testArchivalURI, request)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_ErrorOnReadHistory() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\n\tmockCtrl := gomock.NewController(h.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"some random error\")),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr = historyArchiver.Archive(ctx, h.testArchivalURI, request)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_TimeoutWhenReadingHistory() {\n\n\tctx := getCanceledContext()\n\tmockCtrl := gomock.NewController(h.T())\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, mock.Anything, \"\").Return(true, nil).Times(1)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.ServiceBusyError{}),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(ctx, h.testArchivalURI, request)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_HistoryMutated() {\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion + 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr = historyArchiver.Archive(ctx, h.testArchivalURI, request)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Fail_NonRetriableErrorOption() {\n\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"upload non-retriable error\")),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr = historyArchiver.Archive(ctx, h.testArchivalURI, request, archiver.GetNonRetriableErrorOption(errUploadNonRetriable))\n\th.Equal(errUploadNonRetriable, err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Skip() {\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, mock.Anything).Return(true, nil)\n\tstorageWrapper.On(\"Upload\", ctx, URI, mock.Anything, mock.Anything).Return(nil)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(false),\n\t\t},\n\t\tBody: []*types.History{\n\t\t\t{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:        constants.FirstEventID,\n\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.EntityNotExistsError{Message: \"workflow not found\"}),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr = historyArchiver.Archive(ctx, h.testArchivalURI, request)\n\th.NoError(err)\n}\n\nfunc (h *historyArchiverSuite) TestArchive_Success() {\n\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, mock.Anything).Return(false, nil)\n\tstorageWrapper.On(\"Upload\", ctx, URI, mock.Anything, mock.Anything).Return(nil)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 2,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(false),\n\t)\n\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\n\th.NoError(err)\n\terr = historyArchiver.Archive(ctx, URI, request)\n\n\th.NoError(err)\n\n}\n\nfunc (h *historyArchiverSuite) TestGet_Fail_InvalidURI() {\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   100,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.Nil(response)\n\th.Error(err)\n}\n\nfunc (h *historyArchiverSuite) TestGet_Fail_InvalidToken() {\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tmockStorageClient := &mocks.GcloudStorageClient{}\n\tstorageWrapper, _ := connector.NewClientWithParams(mockStorageClient)\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:      testDomainID,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tPageSize:      testPageSize,\n\t\tNextPageToken: []byte{'r', 'a', 'n', 'd', 'o', 'm'},\n\t}\n\tURI, err := archiver.NewURI(\"gs:///\")\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.Nil(response)\n\th.Error(err)\n\th.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (h *historyArchiverSuite) TestGet_Success_PickHighestVersion() {\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\tstorageWrapper.On(\"Query\", ctx, URI, mock.Anything).Return([]string{\"905702227796330300141628222723188294514017512010591354159_-24_0.history\", \"905702227796330300141628222723188294514017512010591354159_-25_0.history\"}, nil).Times(1)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_0.history\").Return([]byte(exampleHistoryRecord), nil)\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.NoError(err)\n\th.Nil(response.NextPageToken)\n}\n\nfunc (h *historyArchiverSuite) TestGet_Success_UseProvidedVersion() {\n\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\tstorageWrapper.On(\"Query\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470\").Return([]string{\"905702227796330300141628222723188294514017512010591354159_-24_0.history\", \"905702227796330300141628222723188294514017512010591354159_-25_0.history\"}, nil).Times(1)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-25_0.history\").Return([]byte(exampleHistoryRecord), nil)\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             testPageSize,\n\t\tCloseFailoverVersion: common.Int64Ptr(-25),\n\t}\n\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.NoError(err)\n\th.Nil(response.NextPageToken)\n}\n\nfunc (h *historyArchiverSuite) TestGet_Success_PageSize() {\n\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\tstorageWrapper.On(\"Query\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470\").Return([]string{\"905702227796330300141628222723188294514017512010591354159_-24_0.history\", \"905702227796330300141628222723188294514017512010591354159_-24_1.history\", \"905702227796330300141628222723188294514017512010591354159_-24_2.history\", \"905702227796330300141628222723188294514017512010591354159_-24_3.history\", \"905702227796330300141628222723188294514017512010591354159_-25_0.history\"}, nil).Times(1)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_0.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_1.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_2.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_3.history\").Return([]byte(exampleHistoryRecord), nil)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   2,\n\t}\n\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.NoError(err)\n\th.NotNil(response.NextPageToken)\n\th.EqualValues(len(response.HistoryBatches), 2)\n}\n\nfunc (h *historyArchiverSuite) TestGet_Success_FromToken() {\n\n\tctx := context.Background()\n\tmockCtrl := gomock.NewController(h.T())\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/development\")\n\th.Require().NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", ctx, URI, \"\").Return(true, nil).Times(1)\n\tstorageWrapper.On(\"Query\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470\").Return([]string{\"905702227796330300141628222723188294514017512010591354159_-24_0.history\", \"905702227796330300141628222723188294514017512010591354159_-24_1.history\", \"905702227796330300141628222723188294514017512010591354159_-24_2.history\", \"905702227796330300141628222723188294514017512010591354159_-24_3.history\", \"905702227796330300141628222723188294514017512010591354159_-25_0.history\"}, nil).Times(1)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_0.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_1.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_2.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_3.history\").Return([]byte(twoEventsExampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_4.history\").Return([]byte(exampleHistoryRecord), nil)\n\tstorageWrapper.On(\"Get\", ctx, URI, \"71817125141568232911739672280485489488911532452831150339470_-24_5.history\").Return([]byte(exampleHistoryRecord), nil)\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryArchiver := newHistoryArchiver(h.container, historyIterator, storageWrapper)\n\n\ttoken := &getHistoryToken{\n\t\tCloseFailoverVersion: -24,\n\t\tHighestPart:          5,\n\t\tCurrentPart:          2,\n\t\tBatchIdxOffset:       0,\n\t}\n\n\tnextPageToken, err := serializeToken(token)\n\th.NoError(err)\n\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:      testDomainID,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tPageSize:      4,\n\t\tNextPageToken: nextPageToken,\n\t}\n\n\th.NoError(err)\n\tresponse, err := historyArchiver.Get(ctx, URI, request)\n\th.NoError(err)\n\th.NotNil(response.NextPageToken)\n\n\ttoken, err = deserializeGetHistoryToken(response.NextPageToken)\n\th.NoError(err)\n\n\th.EqualValues(5, token.HighestPart)\n\th.EqualValues(5, token.CurrentPart)\n\th.EqualValues(3, len(response.HistoryBatches))\n\tnumOfEvents := 0\n\tfor _, batch := range response.HistoryBatches {\n\t\tnumOfEvents += len(batch.Events)\n\t}\n\n\th.EqualValues(4, numOfEvents)\n}\n"
  },
  {
    "path": "common/archiver/gcloud/init.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc init() {\n\t// register default providers, ideally remove this and trigger manually during startup\n\n\tmust := func(err error) {\n\t\tif err != nil {\n\t\t\tpanic(fmt.Errorf(\"failed to register gcloud archivers: %w\", err))\n\t\t}\n\t}\n\n\tmust(provider.RegisterHistoryArchiver(URIScheme, ConfigKey, func(cfg *config.YamlNode, container *archiver.HistoryBootstrapContainer) (archiver.HistoryArchiver, error) {\n\t\tvar out connector.Config\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn NewHistoryArchiver(container, out)\n\t}))\n\tmust(provider.RegisterVisibilityArchiver(URIScheme, ConfigKey, func(cfg *config.YamlNode, container *archiver.VisibilityBootstrapContainer) (archiver.VisibilityArchiver, error) {\n\t\tvar out connector.Config\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn NewVisibilityArchiver(container, out)\n\t}))\n}\n"
  },
  {
    "path": "common/archiver/gcloud/queryParser.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source queryParser.go -destination queryParser_mock.go -mock_names Interface=MockQueryParser\n\npackage gcloud\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\ntype (\n\t// QueryParser parses a limited SQL where clause into a struct\n\tQueryParser interface {\n\t\tParse(query string) (*parsedQuery, error)\n\t}\n\n\tqueryParser struct{}\n\n\tparsedQuery struct {\n\t\tworkflowID      *string\n\t\tworkflowType    *string\n\t\tstartTime       int64\n\t\tcloseTime       int64\n\t\tsearchPrecision *string\n\t\trunID           *string\n\t\temptyResult     bool\n\t}\n)\n\n// All allowed fields for filtering\nconst (\n\tWorkflowID      = \"WorkflowID\"\n\tRunID           = \"RunID\"\n\tWorkflowType    = \"WorkflowType\"\n\tCloseTime       = \"CloseTime\"\n\tStartTime       = \"StartTime\"\n\tCloseStatus     = \"CloseStatus\"\n\tSearchPrecision = \"SearchPrecision\"\n)\n\n// Precision specific values\nconst (\n\tPrecisionDay    = \"Day\"\n\tPrecisionHour   = \"Hour\"\n\tPrecisionMinute = \"Minute\"\n\tPrecisionSecond = \"Second\"\n)\n\nconst (\n\tqueryTemplate = \"select * from dummy where %s\"\n\n\tdefaultDateTimeFormat = time.RFC3339\n)\n\n// NewQueryParser creates a new query parser for filestore\nfunc NewQueryParser() QueryParser {\n\treturn &queryParser{}\n}\n\nfunc (p *queryParser) Parse(query string) (*parsedQuery, error) {\n\tstmt, err := sqlparser.Parse(fmt.Sprintf(queryTemplate, query))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\twhereExpr := stmt.(*sqlparser.Select).Where.Expr\n\tparsedQuery := &parsedQuery{}\n\tif err := p.convertWhereExpr(whereExpr, parsedQuery); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif (parsedQuery.closeTime == 0 && parsedQuery.startTime == 0) || (parsedQuery.closeTime != 0 && parsedQuery.startTime != 0) {\n\t\treturn nil, errors.New(\"requires a StartTime or CloseTime\")\n\t}\n\n\tif parsedQuery.searchPrecision == nil {\n\t\treturn nil, errors.New(\"SearchPrecision is required when searching for a StartTime or CloseTime\")\n\t}\n\n\treturn parsedQuery, nil\n}\n\nfunc (p *queryParser) convertWhereExpr(expr sqlparser.Expr, parsedQuery *parsedQuery) error {\n\tif expr == nil {\n\t\treturn errors.New(\"where expression is nil\")\n\t}\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn p.convertComparisonExpr(expr, parsedQuery)\n\tcase *sqlparser.AndExpr:\n\t\treturn p.convertAndExpr(expr, parsedQuery)\n\tcase *sqlparser.ParenExpr:\n\t\treturn p.convertParenExpr(expr, parsedQuery)\n\tdefault:\n\t\treturn errors.New(\"only comparison and \\\"and\\\" expression is supported\")\n\t}\n}\n\nfunc (p *queryParser) convertParenExpr(parenExpr *sqlparser.ParenExpr, parsedQuery *parsedQuery) error {\n\treturn p.convertWhereExpr(parenExpr.Expr, parsedQuery)\n}\n\nfunc (p *queryParser) convertAndExpr(andExpr *sqlparser.AndExpr, parsedQuery *parsedQuery) error {\n\tif err := p.convertWhereExpr(andExpr.Left, parsedQuery); err != nil {\n\t\treturn err\n\t}\n\treturn p.convertWhereExpr(andExpr.Right, parsedQuery)\n}\n\nfunc (p *queryParser) convertComparisonExpr(compExpr *sqlparser.ComparisonExpr, parsedQuery *parsedQuery) error {\n\tcolName, ok := compExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid filter name: %s\", sqlparser.String(compExpr.Left))\n\t}\n\tcolNameStr := sqlparser.String(colName)\n\top := compExpr.Operator\n\tvalExpr, ok := compExpr.Right.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid value: %s\", sqlparser.String(compExpr.Right))\n\t}\n\tvalStr := sqlparser.String(valExpr)\n\n\tswitch colNameStr {\n\tcase WorkflowID:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", WorkflowID)\n\t\t}\n\t\tif parsedQuery.workflowID != nil && *parsedQuery.workflowID != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.workflowID = common.StringPtr(val)\n\tcase RunID:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", RunID)\n\t\t}\n\t\tif parsedQuery.runID != nil && *parsedQuery.runID != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.runID = common.StringPtr(val)\n\tcase CloseTime:\n\t\ttimestamp, err := convertToTimestamp(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", CloseTime)\n\t\t}\n\t\tparsedQuery.closeTime = timestamp\n\n\tcase StartTime:\n\t\ttimestamp, err := convertToTimestamp(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", StartTime)\n\t\t}\n\t\tparsedQuery.startTime = timestamp\n\tcase WorkflowType:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", WorkflowType)\n\t\t}\n\t\tif parsedQuery.workflowType != nil && *parsedQuery.workflowType != val {\n\t\t\tparsedQuery.emptyResult = true\n\t\t\treturn nil\n\t\t}\n\t\tparsedQuery.workflowType = common.StringPtr(val)\n\tcase SearchPrecision:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Google Cloud Storage\", SearchPrecision)\n\t\t}\n\t\tif parsedQuery.searchPrecision != nil && *parsedQuery.searchPrecision != val {\n\t\t\treturn fmt.Errorf(\"only one expression is allowed for %s\", SearchPrecision)\n\t\t}\n\t\tswitch val {\n\t\tcase PrecisionDay:\n\t\tcase PrecisionHour:\n\t\tcase PrecisionMinute:\n\t\tcase PrecisionSecond:\n\t\tdefault:\n\t\t\treturn fmt.Errorf(\"invalid value for %s: %s\", SearchPrecision, val)\n\t\t}\n\t\tparsedQuery.searchPrecision = common.StringPtr(val)\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown filter name: %s\", colNameStr)\n\t}\n\n\treturn nil\n}\n\nfunc convertToTimestamp(timeStr string) (int64, error) {\n\ttimestamp, err := strconv.ParseInt(timeStr, 10, 64)\n\tif err == nil {\n\t\treturn timestamp, nil\n\t}\n\ttimestampStr, err := extractStringValue(timeStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tparsedTime, err := time.Parse(defaultDateTimeFormat, timestampStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn parsedTime.UnixNano(), nil\n}\n\nfunc extractStringValue(s string) (string, error) {\n\tif len(s) >= 2 && s[0] == '\\'' && s[len(s)-1] == '\\'' {\n\t\treturn s[1 : len(s)-1], nil\n\t}\n\treturn \"\", fmt.Errorf(\"value %s is not a string value\", s)\n}\n"
  },
  {
    "path": "common/archiver/gcloud/queryParser_mock.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by MockGen. DO NOT EDIT.\n// Source: queryParser.go\n\n// Package gcloud is a generated GoMock package.\npackage gcloud\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockQueryParser is a mock of QueryParser interface.\ntype MockQueryParser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueryParserMockRecorder\n}\n\n// MockQueryParserMockRecorder is the mock recorder for MockQueryParser.\ntype MockQueryParserMockRecorder struct {\n\tmock *MockQueryParser\n}\n\n// NewMockQueryParser creates a new mock instance.\nfunc NewMockQueryParser(ctrl *gomock.Controller) *MockQueryParser {\n\tmock := &MockQueryParser{ctrl: ctrl}\n\tmock.recorder = &MockQueryParserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueryParser) EXPECT() *MockQueryParserMockRecorder {\n\treturn m.recorder\n}\n\n// Parse mocks base method.\nfunc (m *MockQueryParser) Parse(query string) (*parsedQuery, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Parse\", query)\n\tret0, _ := ret[0].(*parsedQuery)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Parse indicates an expected call of Parse.\nfunc (mr *MockQueryParserMockRecorder) Parse(query interface{}) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Parse\", reflect.TypeOf((*MockQueryParser)(nil).Parse), query)\n}\n"
  },
  {
    "path": "common/archiver/gcloud/util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/dgryski/go-farm\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc encode(v interface{}) ([]byte, error) {\n\treturn json.Marshal(v)\n}\n\nfunc decodeHistoryBatches(data []byte) ([]*types.History, error) {\n\thistoryBatches := []*types.History{}\n\terr := json.Unmarshal(data, &historyBatches)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn historyBatches, nil\n}\n\nfunc constructHistoryFilenameMultipart(domainID, workflowID, runID string, version int64, partNumber int) string {\n\tcombinedHash := constructHistoryFilenamePrefix(domainID, workflowID, runID)\n\treturn fmt.Sprintf(\"%s_%v_%v.history\", combinedHash, version, partNumber)\n}\n\nfunc constructHistoryFilenamePrefix(domainID, workflowID, runID string) string {\n\treturn strings.Join([]string{hash(domainID), hash(workflowID), hash(runID)}, \"\")\n}\n\nfunc constructVisibilityFilenamePrefix(domainID, tag string) string {\n\treturn fmt.Sprintf(\"%s/%s\", domainID, tag)\n}\n\nfunc constructTimeBasedSearchKey(domainID, tag string, timestamp int64, precision string) string {\n\tt := time.Unix(0, timestamp).In(time.UTC)\n\tvar timeFormat = \"\"\n\tswitch precision {\n\tcase PrecisionSecond:\n\t\ttimeFormat = \":05\"\n\t\tfallthrough\n\tcase PrecisionMinute:\n\t\ttimeFormat = \":04\" + timeFormat\n\t\tfallthrough\n\tcase PrecisionHour:\n\t\ttimeFormat = \"15\" + timeFormat\n\t\tfallthrough\n\tcase PrecisionDay:\n\t\ttimeFormat = \"2006-01-02T\" + timeFormat\n\t}\n\n\treturn fmt.Sprintf(\"%s_%s\", constructVisibilityFilenamePrefix(domainID, tag), t.Format(timeFormat))\n}\n\nfunc hash(s string) (result string) {\n\tif s != \"\" {\n\t\treturn fmt.Sprintf(\"%v\", farm.Fingerprint64([]byte(s)))\n\t}\n\treturn\n}\n\nfunc contextExpired(ctx context.Context) bool {\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc deserializeGetHistoryToken(bytes []byte) (*getHistoryToken, error) {\n\ttoken := &getHistoryToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc extractCloseFailoverVersion(filename string) (int64, int, error) {\n\tfilenameParts := strings.FieldsFunc(filename, func(r rune) bool {\n\t\treturn r == '_' || r == '.'\n\t})\n\tif len(filenameParts) != 4 {\n\t\treturn -1, 0, errors.New(\"unknown filename structure\")\n\t}\n\n\tfailoverVersion, err := strconv.ParseInt(filenameParts[1], 10, 64)\n\tif err != nil {\n\t\treturn -1, 0, err\n\t}\n\n\thighestPart, err := strconv.Atoi(filenameParts[2])\n\treturn failoverVersion, highestPart, err\n}\n\nfunc serializeToken(token interface{}) ([]byte, error) {\n\tif token == nil {\n\t\treturn nil, nil\n\t}\n\treturn json.Marshal(token)\n}\n\nfunc decodeVisibilityRecord(data []byte) (*visibilityRecord, error) {\n\trecord := &visibilityRecord{}\n\terr := json.Unmarshal(data, record)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn record, nil\n}\n\nfunc constructVisibilityFilename(domain, workflowTypeName, workflowID, runID, tag string, timestamp int64) string {\n\tt := time.Unix(0, timestamp).In(time.UTC)\n\tprefix := constructVisibilityFilenamePrefix(domain, tag)\n\treturn fmt.Sprintf(\"%s_%s_%s_%s_%s.visibility\", prefix, t.Format(time.RFC3339), hash(workflowTypeName), hash(workflowID), hash(runID))\n}\n\nfunc deserializeQueryVisibilityToken(bytes []byte) (*queryVisibilityToken, error) {\n\ttoken := &queryVisibilityToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc convertToExecutionInfo(record *visibilityRecord) *types.WorkflowExecutionInfo {\n\treturn &types.WorkflowExecutionInfo{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: record.WorkflowID,\n\t\t\tRunID:      record.RunID,\n\t\t},\n\t\tType: &types.WorkflowType{\n\t\t\tName: record.WorkflowTypeName,\n\t\t},\n\t\tStartTime:     common.Int64Ptr(record.StartTimestamp),\n\t\tExecutionTime: common.Int64Ptr(record.ExecutionTimestamp),\n\t\tCloseTime:     common.Int64Ptr(record.CloseTimestamp),\n\t\tCloseStatus:   record.CloseStatus.Ptr(),\n\t\tHistoryLength: record.HistoryLength,\n\t\tMemo:          record.Memo,\n\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\tIndexedFields: archiver.ConvertSearchAttrToBytes(record.SearchAttributes),\n\t\t},\n\t}\n}\n\nfunc newRunIDPrecondition(runID string) connector.Precondition {\n\treturn func(subject interface{}) bool {\n\n\t\tif runID == \"\" {\n\t\t\treturn true\n\t\t}\n\n\t\tfileName, ok := subject.(string)\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\n\t\tif strings.Contains(fileName, runID) {\n\t\t\tfileNameParts := strings.Split(fileName, \"_\")\n\t\t\tif len(fileNameParts) != 5 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\treturn strings.Contains(fileName, fileNameParts[4])\n\t\t}\n\n\t\treturn false\n\t}\n}\n\nfunc newWorkflowIDPrecondition(workflowID string) connector.Precondition {\n\treturn func(subject interface{}) bool {\n\n\t\tif workflowID == \"\" {\n\t\t\treturn true\n\t\t}\n\n\t\tfileName, ok := subject.(string)\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\n\t\tif strings.Contains(fileName, workflowID) {\n\t\t\tfileNameParts := strings.Split(fileName, \"_\")\n\t\t\tif len(fileNameParts) != 5 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\treturn strings.Contains(fileName, fileNameParts[3])\n\t\t}\n\n\t\treturn false\n\t}\n}\n\nfunc newWorkflowTypeNamePrecondition(workflowTypeName string) connector.Precondition {\n\treturn func(subject interface{}) bool {\n\n\t\tif workflowTypeName == \"\" {\n\t\t\treturn true\n\t\t}\n\n\t\tfileName, ok := subject.(string)\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\n\t\tif strings.Contains(fileName, workflowTypeName) {\n\t\t\tfileNameParts := strings.Split(fileName, \"_\")\n\t\t\tif len(fileNameParts) != 5 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\treturn strings.Contains(fileName, fileNameParts[2])\n\t\t}\n\n\t\treturn false\n\t}\n}\n\nfunc isRetryableError(err error) (retryable bool) {\n\tswitch err.Error() {\n\tcase connector.ErrBucketNotFound.Error(),\n\t\tarchiver.ErrURISchemeMismatch.Error(),\n\t\tarchiver.ErrInvalidURI.Error():\n\t\tretryable = false\n\tdefault:\n\t\tretryable = true\n\t}\n\n\treturn\n}\n"
  },
  {
    "path": "common/archiver/gcloud/util_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *utilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(utilSuite))\n}\n\ntype utilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc (s *utilSuite) TestEncodeDecodeHistoryBatches() {\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:      constants.FirstEventID,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:      constants.FirstEventID + 2,\n\t\t\t\t\tVersion: 2,\n\t\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\t\tIdentity: \"some random identity\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tencodedHistoryBatches, err := encode(historyBatches)\n\ts.NoError(err)\n\n\tdecodedHistoryBatches, err := decodeHistoryBatches(encodedHistoryBatches)\n\ts.NoError(err)\n\ts.Equal(historyBatches, decodedHistoryBatches)\n}\n\nfunc (s *utilSuite) TestconstructHistoryFilename() {\n\ttestCases := []struct {\n\t\tdomainID             string\n\t\tworkflowID           string\n\t\trunID                string\n\t\tcloseFailoverVersion int64\n\t\texpectBuiltName      string\n\t}{\n\t\t{\n\t\t\tdomainID:             \"testDomainID\",\n\t\t\tworkflowID:           \"testWorkflowID\",\n\t\t\trunID:                \"testRunID\",\n\t\t\tcloseFailoverVersion: 5,\n\t\t\texpectBuiltName:      \"17971674567288329890367046253745284795510285995943906173973_5_0.history\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tfilename := constructHistoryFilenameMultipart(tc.domainID, tc.workflowID, tc.runID, tc.closeFailoverVersion, 0)\n\t\ts.Equal(tc.expectBuiltName, filename)\n\t}\n}\n\nfunc (s *utilSuite) TestSerializeDeserializeGetHistoryToken() {\n\ttoken := &getHistoryToken{\n\t\tCloseFailoverVersion: 101,\n\t\tBatchIdxOffset:       20,\n\t}\n\n\tserializedToken, err := serializeToken(token)\n\ts.Nil(err)\n\n\tdeserializedToken, err := deserializeGetHistoryToken(serializedToken)\n\ts.Nil(err)\n\ts.Equal(token, deserializedToken)\n}\n\nfunc (s *utilSuite) TestConstructHistoryFilenamePrefix() {\n\ts.Equal(\"28646288347718592068344541402884576509131521284625246243\", constructHistoryFilenamePrefix(\"domainID\", \"workflowID\", \"runID\"))\n}\n\nfunc (s *utilSuite) TestConstructHistoryFilenameMultipart() {\n\ts.Equal(\"28646288347718592068344541402884576509131521284625246243_-24_0.history\", constructHistoryFilenameMultipart(\"domainID\", \"workflowID\", \"runID\", -24, 0))\n}\n\nfunc (s *utilSuite) TestConstructVisibilityFilenamePrefix() {\n\ts.Equal(\"domainID/startTimeout\", constructVisibilityFilenamePrefix(\"domainID\", indexKeyStartTimeout))\n}\n\nfunc (s *utilSuite) TestConstructTimeBasedSearchKey() {\n\ts.Equal(\"domainID/startTimeout_1970-01-01T\", constructTimeBasedSearchKey(\"domainID\", indexKeyStartTimeout, 1580819141, \"Day\"))\n}\n\nfunc (s *utilSuite) TestConstructVisibilityFilename() {\n\ts.Equal(\"domainID/startTimeout_1970-01-01T00:24:32Z_4346151385925082125_8344541402884576509_131521284625246243.visibility\", constructVisibilityFilename(\"domainID\", \"workflowTypeName\", \"workflowID\", \"runID\", indexKeyStartTimeout, 1472313624305))\n}\n\nfunc (s *utilSuite) TestWorkflowIdPrecondition() {\n\ttestCases := []struct {\n\t\tworkflowID     string\n\t\tfileName       string\n\t\texpectedResult bool\n\t}{\n\t\t{\n\t\t\tworkflowID:     \"4418294404690464320\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tworkflowID:     \"testWorkflowID\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tworkflowID:     \"\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult: true,\n\t\t},\n\t}\n\n\tfor _, testCase := range testCases {\n\t\ts.Equal(newWorkflowIDPrecondition(testCase.workflowID)(testCase.fileName), testCase.expectedResult)\n\t}\n\n}\n\nfunc (s *utilSuite) TestRunIdPrecondition() {\n\ttestCases := []struct {\n\t\tworkflowID     string\n\t\trunID          string\n\t\tfileName       string\n\t\texpectedResult bool\n\t}{\n\t\t{\n\t\t\tworkflowID:     \"4418294404690464320\",\n\t\t\trunID:          \"15619178330501475177\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tworkflowID:     \"4418294404690464320\",\n\t\t\trunID:          \"15619178330501475177\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_unkonwnRunID.visibility\",\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tworkflowID:     \"4418294404690464320\",\n\t\t\trunID:          \"\",\n\t\t\tfileName:       \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_unkonwnRunID.visibility\",\n\t\t\texpectedResult: true,\n\t\t},\n\t}\n\n\tfor _, testCase := range testCases {\n\t\ts.Equal(newRunIDPrecondition(testCase.runID)(testCase.fileName), testCase.expectedResult)\n\t}\n\n}\n\nfunc (s *utilSuite) TestWorkflowTypeNamePrecondition() {\n\ttestCases := []struct {\n\t\tworkflowID       string\n\t\trunID            string\n\t\tworkflowTypeName string\n\t\tfileName         string\n\t\texpectedResult   bool\n\t}{\n\t\t{\n\t\t\tworkflowID:       \"4418294404690464320\",\n\t\t\trunID:            \"15619178330501475177\",\n\t\t\tworkflowTypeName: \"12851121011173788097\",\n\t\t\tfileName:         \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788097_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult:   true,\n\t\t},\n\t\t{\n\t\t\tworkflowID:       \"4418294404690464320\",\n\t\t\trunID:            \"15619178330501475177\",\n\t\t\tworkflowTypeName: \"12851121011173788097\",\n\t\t\tfileName:         \"closeTimeout_2020-02-27T09:42:28Z_12851121011173788098_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult:   false,\n\t\t},\n\t\t{\n\t\t\tworkflowID:       \"4418294404690464320\",\n\t\t\trunID:            \"15619178330501475177\",\n\t\t\tworkflowTypeName: \"\",\n\t\t\tfileName:         \"closeTimeout_2020-02-27T09:42:28Z_unkownWorkflowTypeName_4418294404690464320_15619178330501475177.visibility\",\n\t\t\texpectedResult:   true,\n\t\t},\n\t}\n\n\tfor _, testCase := range testCases {\n\t\ts.Equal(newWorkflowTypeNamePrecondition(testCase.workflowTypeName)(testCase.fileName), testCase.expectedResult)\n\t}\n\n}\n"
  },
  {
    "path": "common/archiver/gcloud/visibilityArchiver.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"path/filepath\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\terrEncodeVisibilityRecord = \"failed to encode visibility record\"\n\tindexKeyStartTimeout      = \"startTimeout\"\n\tindexKeyCloseTimeout      = \"closeTimeout\"\n\ttimeoutInSeconds          = 5\n)\n\nvar (\n\terrRetriable = errors.New(\"retriable error\")\n)\n\ntype (\n\tvisibilityArchiver struct {\n\t\tcontainer     *archiver.VisibilityBootstrapContainer\n\t\tgcloudStorage connector.Client\n\t\tqueryParser   QueryParser\n\t}\n\n\tqueryVisibilityToken struct {\n\t\tOffset int\n\t}\n\n\tvisibilityRecord archiver.ArchiveVisibilityRequest\n\n\tqueryVisibilityRequest struct {\n\t\tdomainID      string\n\t\tpageSize      int\n\t\tnextPageToken []byte\n\t\tparsedQuery   *parsedQuery\n\t}\n)\n\nfunc newVisibilityArchiver(container *archiver.VisibilityBootstrapContainer, storage connector.Client) *visibilityArchiver {\n\treturn &visibilityArchiver{\n\t\tcontainer:     container,\n\t\tgcloudStorage: storage,\n\t\tqueryParser:   NewQueryParser(),\n\t}\n}\n\n// NewVisibilityArchiver creates a new archiver.VisibilityArchiver based on filestore\nfunc NewVisibilityArchiver(container *archiver.VisibilityBootstrapContainer, config connector.Config) (archiver.VisibilityArchiver, error) {\n\tstorage, err := connector.NewClient(context.Background(), config)\n\treturn newVisibilityArchiver(container, storage), err\n}\n\n// Archive is used to archive one workflow visibility record.\n// Check the Archive() method of the HistoryArchiver interface in Step 2 for parameters' meaning and requirements.\n// The only difference is that the ArchiveOption parameter won't include an option for recording process.\n// Please make sure your implementation is lossless. If any in-memory batching mechanism is used, then those batched records will be lost during server restarts.\n// This method will be invoked when workflow closes. Note that because of conflict resolution, it is possible for a workflow to through the closing process multiple times, which means that this method can be invoked more than once after a workflow closes.\nfunc (v *visibilityArchiver) Archive(ctx context.Context, URI archiver.URI, request *archiver.ArchiveVisibilityRequest, opts ...archiver.ArchiveOption) (err error) {\n\tscope := v.container.MetricsClient.Scope(metrics.HistoryArchiverScope, metrics.DomainTag(request.DomainName))\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif isRetryableError(err) {\n\t\t\t\tscope.IncCounter(metrics.VisibilityArchiverArchiveTransientErrorCount)\n\t\t\t} else {\n\t\t\t\tscope.IncCounter(metrics.VisibilityArchiverArchiveNonRetryableErrorCount)\n\t\t\t\tif featureCatalog.NonRetriableError != nil {\n\t\t\t\t\terr = featureCatalog.NonRetriableError()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}()\n\n\tlogger := archiver.TagLoggerWithArchiveVisibilityRequestAndURI(v.container.Logger, request, URI.String())\n\n\tif err := v.ValidateURI(URI); err != nil {\n\t\tif isRetryableError(err) {\n\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\treturn err\n\t}\n\n\tif err := archiver.ValidateVisibilityArchivalRequest(request); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidArchiveRequest), tag.Error(err))\n\t\treturn err\n\t}\n\n\tencodedVisibilityRecord, err := encode(request)\n\tif err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errEncodeVisibilityRecord), tag.Error(err))\n\t\treturn err\n\t}\n\n\t// The filename has the format: closeTimestamp_hash(runID).visibility\n\t// This format allows the archiver to sort all records without reading the file contents\n\tfilename := constructVisibilityFilename(request.DomainID, request.WorkflowTypeName, request.WorkflowID, request.RunID, indexKeyCloseTimeout, request.CloseTimestamp)\n\tif err := v.gcloudStorage.Upload(ctx, URI, filename, encodedVisibilityRecord); err != nil {\n\t\tlogger.Error(archiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(errWriteFile), tag.Error(err))\n\t\treturn errRetriable\n\t}\n\n\tfilename = constructVisibilityFilename(request.DomainID, request.WorkflowTypeName, request.WorkflowID, request.RunID, indexKeyStartTimeout, request.StartTimestamp)\n\tif err := v.gcloudStorage.Upload(ctx, URI, filename, encodedVisibilityRecord); err != nil {\n\t\tlogger.Error(archiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(errWriteFile), tag.Error(err))\n\t\treturn errRetriable\n\t}\n\n\tscope.IncCounter(metrics.VisibilityArchiveSuccessCount)\n\treturn nil\n}\n\n// Query is used to retrieve archived visibility records.\n// Check the Get() method of the HistoryArchiver interface in Step 2 for parameters' meaning and requirements.\n// The request includes a string field called query, which describes what kind of visibility records should be returned. For example, it can be some SQL-like syntax query string.\n// Your implementation is responsible for parsing and validating the query, and also returning all visibility records that match the query.\n// Currently the maximum context timeout passed into the method is 3 minutes, so it's ok if this method takes a long time to run.\nfunc (v *visibilityArchiver) Query(ctx context.Context, URI archiver.URI, request *archiver.QueryVisibilityRequest) (*archiver.QueryVisibilityResponse, error) {\n\tif err := v.ValidateURI(URI); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateQueryRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidQueryVisibilityRequest.Error()}\n\t}\n\n\tparsedQuery, err := v.queryParser.Parse(request.Query)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: err.Error()}\n\t}\n\n\tif parsedQuery.emptyResult {\n\t\treturn &archiver.QueryVisibilityResponse{}, nil\n\t}\n\n\treturn v.query(ctx, URI, &queryVisibilityRequest{\n\t\tdomainID:      request.DomainID,\n\t\tpageSize:      request.PageSize,\n\t\tnextPageToken: request.NextPageToken,\n\t\tparsedQuery:   parsedQuery,\n\t})\n}\n\nfunc (v *visibilityArchiver) query(ctx context.Context, URI archiver.URI, request *queryVisibilityRequest) (*archiver.QueryVisibilityResponse, error) {\n\ttoken := new(queryVisibilityToken)\n\tif request.nextPageToken != nil {\n\t\tvar err error\n\t\ttoken, err = deserializeQueryVisibilityToken(request.nextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: archiver.ErrNextPageTokenCorrupted.Error()}\n\t\t}\n\t}\n\n\tvar prefix = constructVisibilityFilenamePrefix(request.domainID, indexKeyCloseTimeout)\n\tif request.parsedQuery.closeTime != 0 {\n\t\tprefix = constructTimeBasedSearchKey(request.domainID, indexKeyCloseTimeout, request.parsedQuery.closeTime, *request.parsedQuery.searchPrecision)\n\t}\n\tif request.parsedQuery.startTime != 0 {\n\t\tprefix = constructTimeBasedSearchKey(request.domainID, indexKeyStartTimeout, request.parsedQuery.startTime, *request.parsedQuery.searchPrecision)\n\t}\n\n\tfilters := make([]connector.Precondition, 0)\n\tif request.parsedQuery.workflowID != nil {\n\t\tfilters = append(filters, newWorkflowIDPrecondition(hash(*request.parsedQuery.workflowID)))\n\t}\n\n\tif request.parsedQuery.runID != nil {\n\t\tfilters = append(filters, newWorkflowIDPrecondition(hash(*request.parsedQuery.runID)))\n\t}\n\n\tif request.parsedQuery.workflowType != nil {\n\t\tfilters = append(filters, newWorkflowIDPrecondition(hash(*request.parsedQuery.workflowType)))\n\t}\n\n\tfilenames, completed, currentCursorPos, err := v.gcloudStorage.QueryWithFilters(ctx, URI, prefix, request.pageSize, token.Offset, filters)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\n\tresponse := &archiver.QueryVisibilityResponse{}\n\tfor _, file := range filenames {\n\t\tencodedRecord, err := v.gcloudStorage.Get(ctx, URI, fmt.Sprintf(\"%s/%s\", request.domainID, filepath.Base(file)))\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\trecord, err := decodeVisibilityRecord(encodedRecord)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\tresponse.Executions = append(response.Executions, convertToExecutionInfo(record))\n\t}\n\n\tif !completed {\n\t\tnewToken := &queryVisibilityToken{\n\t\t\tOffset: currentCursorPos,\n\t\t}\n\t\tencodedToken, err := serializeToken(newToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\tresponse.NextPageToken = encodedToken\n\t}\n\n\treturn response, nil\n}\n\n// ValidateURI is used to define what a valid URI for an implementation is.\nfunc (v *visibilityArchiver) ValidateURI(URI archiver.URI) (err error) {\n\tctx, cancel := context.WithTimeout(context.Background(), timeoutInSeconds*time.Second)\n\tdefer cancel()\n\n\tif err = v.validateURI(URI); err == nil {\n\t\t_, err = v.gcloudStorage.Exist(ctx, URI, \"\")\n\t}\n\n\treturn\n}\n\nfunc (v *visibilityArchiver) validateURI(URI archiver.URI) (err error) {\n\tif URI.Scheme() != URIScheme {\n\t\treturn archiver.ErrURISchemeMismatch\n\t}\n\n\tif URI.Path() == \"\" || URI.Hostname() == \"\" {\n\t\treturn archiver.ErrInvalidURI\n\t}\n\n\treturn\n}\n"
  },
  {
    "path": "common/archiver/gcloud/visibilityArchiver_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gcloud\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/gcloud/connector/mocks\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestWorkflowTypeName    = \"test-workflow-type\"\n\texampleVisibilityRecord = `{\"DomainID\":\"test-domain-id\",\"DomainName\":\"test-domain-name\",\"WorkflowID\":\"test-workflow-id\",\"RunID\":\"test-run-id\",\"WorkflowTypeName\":\"test-workflow-type\",\"StartTimestamp\":1580896574804475000,\"ExecutionTimestamp\":0,\"CloseTimestamp\":1580896575946478000,\"CloseStatus\":\"COMPLETED\",\"HistoryLength\":36,\"Memo\":null,\"SearchAttributes\":{},\"HistoryArchivalURI\":\"gs://my-bucket-cad/cadence_archival/development\"}`\n)\n\nfunc (s *visibilityArchiverSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.container = &archiver.VisibilityBootstrapContainer{\n\t\tLogger:        testlogger.New(s.T()),\n\t\tMetricsClient: metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t}\n\ts.expectedVisibilityRecords = []*visibilityRecord{\n\t\t{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            testRunID,\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   1580896574804475000,\n\t\t\tCloseTimestamp:   1580896575946478000,\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusCompleted,\n\t\t\tHistoryLength:    36,\n\t\t},\n\t}\n}\nfunc TestVisibilityArchiverSuiteSuite(t *testing.T) {\n\tsuite.Run(t, new(visibilityArchiverSuite))\n}\n\ntype visibilityArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\tcontainer                 *archiver.VisibilityBootstrapContainer\n\texpectedVisibilityRecords []*visibilityRecord\n}\n\nfunc (s *visibilityArchiverSuite) TestValidateVisibilityURI() {\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs:my-bucket-cad/cadence_archival/visibility\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://my-bucket-cad\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs:/my-bucket-cad/cadence_archival/visibility\",\n\t\t\texpectedErr: archiver.ErrInvalidURI,\n\t\t},\n\t\t{\n\t\t\tURI:         \"gs://my-bucket-cad/cadence_archival/visibility\",\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, mock.Anything, \"\").Return(false, nil)\n\tvisibilityArchiver := new(visibilityArchiver)\n\tvisibilityArchiver.gcloudStorage = storageWrapper\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedErr, visibilityArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_InvalidVisibilityURI() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, \"\").Return(true, nil).Times(1)\n\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainID:   testDomainID,\n\t\tDomainName: testDomainName,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t}\n\n\terr = visibilityArchiver.Archive(ctx, URI, request)\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidVisibilityURI() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, \"\").Return(true, nil).Times(1)\n\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 10,\n\t\tQuery:    \"WorkflowType='type::example' AND CloseTime='2020-02-05T11:00:00Z' AND SearchPrecision='Day'\",\n\t}\n\n\t_, err = visibilityArchiver.Query(ctx, URI, request)\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestVisibilityArchive() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/visibility\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, mock.Anything).Return(false, nil)\n\tstorageWrapper.On(\"Upload\", mock.Anything, URI, mock.Anything, mock.Anything).Return(nil)\n\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainName:         testDomainName,\n\t\tDomainID:           testDomainID,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     time.Now().UnixNano(),\n\t\tExecutionTimestamp: 0, // workflow without backoff\n\t\tCloseTimestamp:     time.Now().UnixNano(),\n\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\tHistoryLength:      int64(101),\n\t}\n\n\terr = visibilityArchiver.Archive(ctx, URI, request)\n\ts.NoError(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidQuery() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/visibility\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, mock.Anything).Return(false, nil)\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\tmockCtrl := gomock.NewController(s.T())\n\n\tmockParser := NewMockQueryParser(mockCtrl)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(nil, errors.New(\"invalid query\"))\n\tvisibilityArchiver.queryParser = mockParser\n\tresponse, err := visibilityArchiver.Query(ctx, URI, &archiver.QueryVisibilityRequest{\n\t\tDomainID: \"some random domainID\",\n\t\tPageSize: 10,\n\t\tQuery:    \"some invalid query\",\n\t})\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidToken() {\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/visibility\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, mock.Anything).Return(false, nil)\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\tmockCtrl := gomock.NewController(s.T())\n\n\tmockParser := NewMockQueryParser(mockCtrl)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tcloseTime: int64(101),\n\t\tstartTime: int64(1),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID:      testDomainID,\n\t\tQuery:         \"parsed by mockParser\",\n\t\tPageSize:      1,\n\t\tNextPageToken: []byte{1, 2, 3},\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_NoNextPageToken() {\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/visibility\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, mock.Anything).Return(false, nil)\n\tstorageWrapper.On(\"QueryWithFilters\", mock.Anything, URI, mock.Anything, 10, 0, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return([]string{\"closeTimeout_2020-02-05T09:56:14Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\"}, true, 1, nil).Times(1)\n\tstorageWrapper.On(\"Get\", mock.Anything, URI, \"test-domain-id/closeTimeout_2020-02-05T09:56:14Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\").Return([]byte(exampleVisibilityRecord), nil)\n\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\tmockCtrl := gomock.NewController(s.T())\n\n\tmockParser := NewMockQueryParser(mockCtrl)\n\tdayPrecision := string(\"Day\")\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tcloseTime:       int64(101),\n\t\tsearchPrecision: &dayPrecision,\n\t\tworkflowType:    common.StringPtr(\"MobileOnlyWorkflow::processMobileOnly\"),\n\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t\trunID:           common.StringPtr(testRunID),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 10,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\n\tresponse, err := visibilityArchiver.Query(ctx, URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 1)\n\ts.Equal(convertToExecutionInfo(s.expectedVisibilityRecords[0]), response.Executions[0])\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_SmallPageSize() {\n\n\tpageSize := 2\n\tctx := context.Background()\n\tURI, err := archiver.NewURI(\"gs://my-bucket-cad/cadence_archival/visibility\")\n\ts.NoError(err)\n\tstorageWrapper := &mocks.Client{}\n\tstorageWrapper.On(\"Exist\", mock.Anything, URI, mock.Anything).Return(false, nil)\n\tstorageWrapper.On(\"QueryWithFilters\", mock.Anything, URI, mock.Anything, pageSize, 0, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return([]string{\"closeTimeout_2020-02-05T09:56:14Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\", \"closeTimeout_2020-02-05T09:56:15Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\"}, false, 1, nil).Times(2)\n\tstorageWrapper.On(\"QueryWithFilters\", mock.Anything, URI, mock.Anything, pageSize, 1, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return([]string{\"closeTimeout_2020-02-05T09:56:16Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\"}, true, 2, nil).Times(2)\n\tstorageWrapper.On(\"Get\", mock.Anything, URI, \"test-domain-id/closeTimeout_2020-02-05T09:56:14Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\").Return([]byte(exampleVisibilityRecord), nil)\n\tstorageWrapper.On(\"Get\", mock.Anything, URI, \"test-domain-id/closeTimeout_2020-02-05T09:56:15Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\").Return([]byte(exampleVisibilityRecord), nil)\n\tstorageWrapper.On(\"Get\", mock.Anything, URI, \"test-domain-id/closeTimeout_2020-02-05T09:56:16Z_test-workflow-id_MobileOnlyWorkflow::processMobileOnly_test-run-id.visibility\").Return([]byte(exampleVisibilityRecord), nil)\n\n\tvisibilityArchiver := newVisibilityArchiver(s.container, storageWrapper)\n\ts.NoError(err)\n\tmockCtrl := gomock.NewController(s.T())\n\n\tmockParser := NewMockQueryParser(mockCtrl)\n\tdayPrecision := string(\"Day\")\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tcloseTime:       int64(101),\n\t\tsearchPrecision: &dayPrecision,\n\t\tworkflowType:    common.StringPtr(\"MobileOnlyWorkflow::processMobileOnly\"),\n\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t\trunID:           common.StringPtr(testRunID),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: pageSize,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\n\tresponse, err := visibilityArchiver.Query(ctx, URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.NotNil(response.NextPageToken)\n\ts.Len(response.Executions, 2)\n\ts.Equal(convertToExecutionInfo(s.expectedVisibilityRecords[0]), response.Executions[0])\n\ts.Equal(convertToExecutionInfo(s.expectedVisibilityRecords[0]), response.Executions[1])\n\n\trequest.NextPageToken = response.NextPageToken\n\tresponse, err = visibilityArchiver.Query(ctx, URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 1)\n\ts.Equal(convertToExecutionInfo(s.expectedVisibilityRecords[0]), response.Executions[0])\n}\n"
  },
  {
    "path": "common/archiver/historyIterator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination historyIterator_mock.go -self_package github.com/uber/cadence/common/archiver\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\thistoryPageSize = 250\n)\n\ntype (\n\t// HistoryIterator is used to get history batches\n\tHistoryIterator interface {\n\t\tNext() (*HistoryBlob, error)\n\t\tHasNext() bool\n\t\tGetState() ([]byte, error)\n\t}\n\n\t// HistoryBlobHeader is the header attached to all history blobs\n\tHistoryBlobHeader struct {\n\t\tDomainName           *string `json:\"domain_name,omitempty\"`\n\t\tDomainID             *string `json:\"domain_id,omitempty\"`\n\t\tWorkflowID           *string `json:\"workflow_id,omitempty\"`\n\t\tRunID                *string `json:\"run_id,omitempty\"`\n\t\tIsLast               *bool   `json:\"is_last,omitempty\"`\n\t\tFirstFailoverVersion *int64  `json:\"first_failover_version,omitempty\"`\n\t\tLastFailoverVersion  *int64  `json:\"last_failover_version,omitempty\"`\n\t\tFirstEventID         *int64  `json:\"first_event_id,omitempty\"`\n\t\tLastEventID          *int64  `json:\"last_event_id,omitempty\"`\n\t\tEventCount           *int64  `json:\"event_count,omitempty\"`\n\t}\n\n\t// HistoryBlob is the serializable data that forms the body of a blob\n\tHistoryBlob struct {\n\t\tHeader *HistoryBlobHeader `json:\"header\"`\n\t\tBody   []*types.History   `json:\"body\"`\n\t}\n\n\thistoryIteratorState struct {\n\t\tNextEventID       int64\n\t\tFinishedIteration bool\n\t}\n\n\thistoryIterator struct {\n\t\thistoryIteratorState\n\n\t\tctx                   context.Context\n\t\trequest               *ArchiveHistoryRequest\n\t\thistoryV2Manager      persistence.HistoryManager\n\t\tsizeEstimator         SizeEstimator\n\t\thistoryPageSize       int\n\t\ttargetHistoryBlobSize int\n\t}\n)\n\nvar (\n\terrIteratorDepleted = errors.New(\"iterator is depleted\")\n)\n\n// NewHistoryIterator returns a new HistoryIterator\nfunc NewHistoryIterator(\n\tctx context.Context,\n\trequest *ArchiveHistoryRequest,\n\thistoryV2Manager persistence.HistoryManager,\n\ttargetHistoryBlobSize int,\n) HistoryIterator {\n\treturn newHistoryIterator(ctx, request, historyV2Manager, targetHistoryBlobSize)\n}\n\n// NewHistoryIteratorFromState returns a new HistoryIterator with specified state\nfunc NewHistoryIteratorFromState(\n\tctx context.Context,\n\trequest *ArchiveHistoryRequest,\n\thistoryV2Manager persistence.HistoryManager,\n\ttargetHistoryBlobSize int,\n\tinitialState []byte,\n) (HistoryIterator, error) {\n\tit := newHistoryIterator(ctx, request, historyV2Manager, targetHistoryBlobSize)\n\tif initialState == nil {\n\t\treturn it, nil\n\t}\n\tif err := it.reset(initialState); err != nil {\n\t\treturn nil, err\n\t}\n\treturn it, nil\n}\n\nfunc newHistoryIterator(\n\tctx context.Context,\n\trequest *ArchiveHistoryRequest,\n\thistoryV2Manager persistence.HistoryManager,\n\ttargetHistoryBlobSize int,\n) *historyIterator {\n\treturn &historyIterator{\n\t\thistoryIteratorState: historyIteratorState{\n\t\t\tNextEventID:       constants.FirstEventID,\n\t\t\tFinishedIteration: false,\n\t\t},\n\t\tctx:                   ctx,\n\t\trequest:               request,\n\t\thistoryV2Manager:      historyV2Manager,\n\t\thistoryPageSize:       historyPageSize,\n\t\ttargetHistoryBlobSize: targetHistoryBlobSize,\n\t\tsizeEstimator:         NewJSONSizeEstimator(),\n\t}\n}\n\nfunc (i *historyIterator) Next() (*HistoryBlob, error) {\n\tif !i.HasNext() {\n\t\treturn nil, errIteratorDepleted\n\t}\n\n\thistoryBatches, newIterState, err := i.readHistoryBatches(i.ctx, i.NextEventID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ti.historyIteratorState = newIterState\n\tfirstEvent := historyBatches[0].Events[0]\n\tlastBatch := historyBatches[len(historyBatches)-1]\n\tlastEvent := lastBatch.Events[len(lastBatch.Events)-1]\n\teventCount := int64(0)\n\tfor _, batch := range historyBatches {\n\t\teventCount += int64(len(batch.Events))\n\t}\n\theader := &HistoryBlobHeader{\n\t\tDomainName:           common.StringPtr(i.request.DomainName),\n\t\tDomainID:             common.StringPtr(i.request.DomainID),\n\t\tWorkflowID:           common.StringPtr(i.request.WorkflowID),\n\t\tRunID:                common.StringPtr(i.request.RunID),\n\t\tIsLast:               common.BoolPtr(i.FinishedIteration),\n\t\tFirstFailoverVersion: common.Int64Ptr(firstEvent.Version),\n\t\tLastFailoverVersion:  common.Int64Ptr(lastEvent.Version),\n\t\tFirstEventID:         common.Int64Ptr(firstEvent.ID),\n\t\tLastEventID:          common.Int64Ptr(lastEvent.ID),\n\t\tEventCount:           common.Int64Ptr(eventCount),\n\t}\n\n\treturn &HistoryBlob{\n\t\tHeader: header,\n\t\tBody:   historyBatches,\n\t}, nil\n}\n\n// HasNext returns true if there are more items to iterate over.\nfunc (i *historyIterator) HasNext() bool {\n\treturn !i.FinishedIteration\n}\n\n// GetState returns the encoded iterator state\nfunc (i *historyIterator) GetState() ([]byte, error) {\n\treturn json.Marshal(i.historyIteratorState)\n}\n\nfunc (i *historyIterator) readHistoryBatches(ctx context.Context, firstEventID int64) ([]*types.History, historyIteratorState, error) {\n\tsize := 0\n\ttargetSize := i.targetHistoryBlobSize\n\tvar historyBatches []*types.History\n\tnewIterState := historyIteratorState{}\n\tfor size < targetSize {\n\t\tcurrHistoryBatches, err := i.readHistory(ctx, firstEventID)\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok && firstEventID != constants.FirstEventID {\n\t\t\tnewIterState.FinishedIteration = true\n\t\t\treturn historyBatches, newIterState, nil\n\t\t}\n\t\tif err != nil {\n\t\t\treturn nil, newIterState, err\n\t\t}\n\t\tfor idx, batch := range currHistoryBatches {\n\t\t\thistoryBatchSize, err := i.sizeEstimator.EstimateSize(batch)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, newIterState, err\n\t\t\t}\n\t\t\tsize += historyBatchSize\n\t\t\thistoryBatches = append(historyBatches, batch)\n\t\t\tfirstEventID = batch.Events[len(batch.Events)-1].ID + 1\n\n\t\t\t// In case targetSize is satisfied before reaching the end of current set of batches, return immediately.\n\t\t\t// Otherwise, we need to look ahead to see if there's more history batches.\n\t\t\tif size >= targetSize && idx != len(currHistoryBatches)-1 {\n\t\t\t\tnewIterState.FinishedIteration = false\n\t\t\t\tnewIterState.NextEventID = firstEventID\n\t\t\t\treturn historyBatches, newIterState, nil\n\t\t\t}\n\t\t}\n\t}\n\n\t// If you are here, it means the target size is met after adding the last batch of read history.\n\t// We need to check if there's more history batches.\n\t_, err := i.readHistory(ctx, firstEventID)\n\tif _, ok := err.(*types.EntityNotExistsError); ok && firstEventID != constants.FirstEventID {\n\t\tnewIterState.FinishedIteration = true\n\t\treturn historyBatches, newIterState, nil\n\t}\n\tif err != nil {\n\t\treturn nil, newIterState, err\n\t}\n\tnewIterState.FinishedIteration = false\n\tnewIterState.NextEventID = firstEventID\n\treturn historyBatches, newIterState, nil\n}\n\nfunc (i *historyIterator) readHistory(ctx context.Context, firstEventID int64) ([]*types.History, error) {\n\treq := &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken: i.request.BranchToken,\n\t\tMinEventID:  firstEventID,\n\t\tMaxEventID:  constants.EndEventID,\n\t\tPageSize:    i.historyPageSize,\n\t\tShardID:     common.IntPtr(i.request.ShardID),\n\t\tDomainName:  i.request.DomainName,\n\t}\n\thistoryBatches, _, _, err := persistenceutils.ReadFullPageV2EventsByBatch(ctx, i.historyV2Manager, req)\n\treturn historyBatches, err\n\n}\n\n// reset resets iterator to a certain state given its encoded representation\n// if it returns an error, the operation will have no effect on the iterator\nfunc (i *historyIterator) reset(stateToken []byte) error {\n\tvar iteratorState historyIteratorState\n\tif err := json.Unmarshal(stateToken, &iteratorState); err != nil {\n\t\treturn err\n\t}\n\ti.historyIteratorState = iteratorState\n\treturn nil\n}\n\ntype (\n\t// SizeEstimator is used to estimate the size of any object\n\tSizeEstimator interface {\n\t\tEstimateSize(v interface{}) (int, error)\n\t}\n\n\tjsonSizeEstimator struct{}\n)\n\nfunc (e *jsonSizeEstimator) EstimateSize(v interface{}) (int, error) {\n\tdata, err := json.Marshal(v)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn len(data), nil\n}\n\n// NewJSONSizeEstimator returns a new SizeEstimator which uses json encoding to\n// estimate size\nfunc NewJSONSizeEstimator() SizeEstimator {\n\treturn &jsonSizeEstimator{}\n}\n"
  },
  {
    "path": "common/archiver/historyIterator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: historyIterator.go\n//\n// Generated by this command:\n//\n//\tmockgen -package archiver -source historyIterator.go -destination historyIterator_mock.go -self_package github.com/uber/cadence/common/archiver\n//\n\n// Package archiver is a generated GoMock package.\npackage archiver\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockHistoryIterator is a mock of HistoryIterator interface.\ntype MockHistoryIterator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHistoryIteratorMockRecorder\n\tisgomock struct{}\n}\n\n// MockHistoryIteratorMockRecorder is the mock recorder for MockHistoryIterator.\ntype MockHistoryIteratorMockRecorder struct {\n\tmock *MockHistoryIterator\n}\n\n// NewMockHistoryIterator creates a new mock instance.\nfunc NewMockHistoryIterator(ctrl *gomock.Controller) *MockHistoryIterator {\n\tmock := &MockHistoryIterator{ctrl: ctrl}\n\tmock.recorder = &MockHistoryIteratorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHistoryIterator) EXPECT() *MockHistoryIteratorMockRecorder {\n\treturn m.recorder\n}\n\n// GetState mocks base method.\nfunc (m *MockHistoryIterator) GetState() ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetState\")\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetState indicates an expected call of GetState.\nfunc (mr *MockHistoryIteratorMockRecorder) GetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetState\", reflect.TypeOf((*MockHistoryIterator)(nil).GetState))\n}\n\n// HasNext mocks base method.\nfunc (m *MockHistoryIterator) HasNext() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasNext\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasNext indicates an expected call of HasNext.\nfunc (mr *MockHistoryIteratorMockRecorder) HasNext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasNext\", reflect.TypeOf((*MockHistoryIterator)(nil).HasNext))\n}\n\n// Next mocks base method.\nfunc (m *MockHistoryIterator) Next() (*HistoryBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Next\")\n\tret0, _ := ret[0].(*HistoryBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Next indicates an expected call of Next.\nfunc (mr *MockHistoryIteratorMockRecorder) Next() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Next\", reflect.TypeOf((*MockHistoryIterator)(nil).Next))\n}\n\n// MockSizeEstimator is a mock of SizeEstimator interface.\ntype MockSizeEstimator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSizeEstimatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockSizeEstimatorMockRecorder is the mock recorder for MockSizeEstimator.\ntype MockSizeEstimatorMockRecorder struct {\n\tmock *MockSizeEstimator\n}\n\n// NewMockSizeEstimator creates a new mock instance.\nfunc NewMockSizeEstimator(ctrl *gomock.Controller) *MockSizeEstimator {\n\tmock := &MockSizeEstimator{ctrl: ctrl}\n\tmock.recorder = &MockSizeEstimatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSizeEstimator) EXPECT() *MockSizeEstimatorMockRecorder {\n\treturn m.recorder\n}\n\n// EstimateSize mocks base method.\nfunc (m *MockSizeEstimator) EstimateSize(v any) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"EstimateSize\", v)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// EstimateSize indicates an expected call of EstimateSize.\nfunc (mr *MockSizeEstimatorMockRecorder) EstimateSize(v any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"EstimateSize\", reflect.TypeOf((*MockSizeEstimator)(nil).EstimateSize), v)\n}\n"
  },
  {
    "path": "common/archiver/historyIterator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestDomainID                     = \"test-domain-id\"\n\ttestDomainName                   = \"test-domain-name\"\n\ttestWorkflowID                   = \"test-workflow-id\"\n\ttestRunID                        = \"test-run-id\"\n\ttestShardID                      = 1\n\ttestNextEventID                  = 1800\n\ttestCloseFailoverVersion         = 100\n\ttestDefaultPersistencePageSize   = 250\n\ttestDefaultTargetHistoryBlobSize = 2 * 1024 * 124\n\ttestDefaultHistoryEventSize      = 50\n)\n\nvar (\n\ttestBranchToken = []byte{1, 2, 3}\n)\n\ntype (\n\tHistoryIteratorSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\t}\n\n\tpage struct {\n\t\tfirstbatchIdx             int\n\t\tnumBatches                int\n\t\tfirstEventFailoverVersion int64\n\t\tlastEventFailoverVersion  int64\n\t}\n\n\ttestSizeEstimator struct{}\n)\n\nfunc (e *testSizeEstimator) EstimateSize(v interface{}) (int, error) {\n\thistoryBatch, ok := v.(*types.History)\n\tif !ok {\n\t\treturn -1, errors.New(\"test size estimator only estimate the size of history batches\")\n\t}\n\treturn testDefaultHistoryEventSize * len(historyBatch.Events), nil\n}\n\nfunc newTestSizeEstimator() SizeEstimator {\n\treturn &testSizeEstimator{}\n}\n\nfunc TestHistoryIteratorSuite(t *testing.T) {\n\tsuite.Run(t, new(HistoryIteratorSuite))\n}\n\nfunc (s *HistoryIteratorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistory_Failed_EventsV2() {\n\tmockHistoryV2Manager := &mocks.HistoryV2Manager{}\n\tmockHistoryV2Manager.On(\"ReadHistoryBranchByBatch\", mock.Anything, mock.Anything).Return(nil, errors.New(\"got error reading history branch\"))\n\titr := s.constructTestHistoryIterator(mockHistoryV2Manager, testDefaultTargetHistoryBlobSize, nil)\n\thistory, err := itr.readHistory(context.Background(), constants.FirstEventID)\n\ts.Error(err)\n\ts.Nil(history)\n\tmockHistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistory_Success_EventsV2() {\n\tmockHistoryV2Manager := &mocks.HistoryV2Manager{}\n\tresp := persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       []*types.History{},\n\t\tNextPageToken: []byte{},\n\t}\n\tmockHistoryV2Manager.On(\"ReadHistoryBranchByBatch\", mock.Anything, mock.Anything).Return(&resp, nil)\n\titr := s.constructTestHistoryIterator(mockHistoryV2Manager, testDefaultTargetHistoryBlobSize, nil)\n\thistory, err := itr.readHistory(context.Background(), constants.FirstEventID)\n\ts.NoError(err)\n\ts.NotNil(history)\n\tmockHistoryV2Manager.AssertExpectations(s.T())\n}\n\n// In the following test:\n//   batchInfo represents # of events for each history batch.\n//   page represents the metadata of the set of history batches that should be requested by the iterator\n//   and returned by the history manager. Each page specifies the index of the first history batch it should\n//   return, # of batches to return and first/last event failover version for the set of batches returned.\n//   Note that is possible that a history batch is contained in multiple pages.\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Fail_FirstCallToReadHistoryGivesError() {\n\tbatchInfo := []int{1}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, 0, false, pages...)\n\titr := s.constructTestHistoryIterator(historyV2Manager, testDefaultTargetHistoryBlobSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\tevents, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.Nil(events)\n\ts.False(nextIterState.FinishedIteration)\n\ts.Zero(nextIterState.NextEventID)\n\ts.Error(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Fail_NonFirstCallToReadHistoryGivesError() {\n\tbatchInfo := []int{1, 1}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             1,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, 1, false, pages...)\n\titr := s.constructTestHistoryIterator(historyV2Manager, testDefaultTargetHistoryBlobSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\tevents, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.Nil(events)\n\ts.False(nextIterState.FinishedIteration)\n\ts.Zero(nextIterState.NextEventID)\n\ts.Error(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Success_ReadToHistoryEnd() {\n\tbatchInfo := []int{1, 2, 1, 1, 1, 3, 3, 1, 3}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             5,\n\t\t\tnumBatches:                4,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\t// ensure target history batches size is greater than total history length to ensure all of history is read\n\titr := s.constructTestHistoryIterator(historyV2Manager, 20*testDefaultHistoryEventSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\thistory, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.NotNil(history)\n\ts.Len(history, 9)\n\ts.True(nextIterState.FinishedIteration)\n\ts.Zero(nextIterState.NextEventID)\n\ts.NoError(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Success_TargetSizeSatisfiedWithoutReadingToEnd() {\n\tbatchInfo := []int{1, 2, 1, 1, 1, 3, 3, 1, 3}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             5,\n\t\t\tnumBatches:                4,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, false, pages...)\n\t// ensure target history batches is smaller than full length of history so that not all of history is read\n\titr := s.constructTestHistoryIterator(historyV2Manager, 11*testDefaultHistoryEventSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\thistory, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.NotNil(history)\n\ts.Len(history, 7)\n\ts.False(nextIterState.FinishedIteration)\n\ts.Equal(int64(13), nextIterState.NextEventID)\n\ts.NoError(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Success_ReadExactlyToHistoryEnd() {\n\tbatchInfo := []int{1, 2, 1, 1, 1, 3, 3, 1, 3}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             5,\n\t\t\tnumBatches:                4,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\t// ensure target history batches size is equal to the full length of history so that all of history is read\n\titr := s.constructTestHistoryIterator(historyV2Manager, 16*testDefaultHistoryEventSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\thistory, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.NotNil(history)\n\ts.Len(history, 9)\n\ts.True(nextIterState.FinishedIteration)\n\ts.Zero(nextIterState.NextEventID)\n\ts.NoError(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestReadHistoryBatches_Success_ReadPageMultipleTimes() {\n\tbatchInfo := []int{1, 3, 2}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             2,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\t// ensure target history batches is very small so that one page needs multiple read\n\titr := s.constructTestHistoryIterator(historyV2Manager, 2*testDefaultHistoryEventSize, nil)\n\tstartingIteratorState := s.copyIteratorState(itr)\n\thistory, nextIterState, err := itr.readHistoryBatches(context.Background(), constants.FirstEventID)\n\ts.NotNil(history)\n\ts.Len(history, 2)\n\ts.False(nextIterState.FinishedIteration)\n\ts.Equal(int64(5), nextIterState.NextEventID)\n\ts.NoError(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\n\thistory, nextIterState, err = itr.readHistoryBatches(context.Background(), nextIterState.NextEventID)\n\ts.NotNil(history)\n\ts.Len(history, 1)\n\ts.True(nextIterState.FinishedIteration)\n\ts.Zero(nextIterState.NextEventID)\n\ts.NoError(err)\n\ts.assertStateMatches(startingIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestNext_Fail_IteratorDepleted() {\n\tbatchInfo := []int{1, 3, 2, 1, 2, 3, 4}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             2,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  2,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                4,\n\t\t\tfirstEventFailoverVersion: 2,\n\t\t\tlastEventFailoverVersion:  5,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\t// set target history batches such that a single call to next will read all of history\n\titr := s.constructTestHistoryIterator(historyV2Manager, 16*testDefaultHistoryEventSize, nil)\n\tblob, err := itr.Next()\n\ts.Nil(err)\n\n\texpectedIteratorState := historyIteratorState{\n\t\t// when iteration is finished page token is not advanced\n\t\tFinishedIteration: true,\n\t\tNextEventID:       0,\n\t}\n\ts.assertStateMatches(expectedIteratorState, itr)\n\ts.NotNil(blob)\n\texpectedHeader := &HistoryBlobHeader{\n\t\tDomainName:           common.StringPtr(testDomainName),\n\t\tDomainID:             common.StringPtr(testDomainID),\n\t\tWorkflowID:           common.StringPtr(testWorkflowID),\n\t\tRunID:                common.StringPtr(testRunID),\n\t\tIsLast:               common.BoolPtr(true),\n\t\tFirstFailoverVersion: common.Int64Ptr(1),\n\t\tLastFailoverVersion:  common.Int64Ptr(5),\n\t\tFirstEventID:         common.Int64Ptr(constants.FirstEventID),\n\t\tLastEventID:          common.Int64Ptr(16),\n\t\tEventCount:           common.Int64Ptr(16),\n\t}\n\ts.Equal(expectedHeader, blob.Header)\n\ts.Len(blob.Body, 7)\n\ts.NoError(err)\n\ts.False(itr.HasNext())\n\n\tblob, err = itr.Next()\n\ts.Equal(err, errIteratorDepleted)\n\ts.Nil(blob)\n\ts.assertStateMatches(expectedIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestNext_Fail_ReturnErrOnSecondCallToNext() {\n\tbatchInfo := []int{1, 3, 2, 1, 3, 2}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             2,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             5,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, 3, false, pages...)\n\t// set target blob size such that the first two pages are read for blob one without error, third page will return error\n\titr := s.constructTestHistoryIterator(historyV2Manager, 6*testDefaultHistoryEventSize, nil)\n\tblob, err := itr.Next()\n\texpectedIteratorState := historyIteratorState{\n\t\tFinishedIteration: false,\n\t\tNextEventID:       7,\n\t}\n\ts.assertStateMatches(expectedIteratorState, itr)\n\ts.NotNil(blob)\n\texpectedHeader := &HistoryBlobHeader{\n\t\tDomainName:           common.StringPtr(testDomainName),\n\t\tDomainID:             common.StringPtr(testDomainID),\n\t\tWorkflowID:           common.StringPtr(testWorkflowID),\n\t\tRunID:                common.StringPtr(testRunID),\n\t\tIsLast:               common.BoolPtr(false),\n\t\tFirstFailoverVersion: common.Int64Ptr(1),\n\t\tLastFailoverVersion:  common.Int64Ptr(1),\n\t\tFirstEventID:         common.Int64Ptr(constants.FirstEventID),\n\t\tLastEventID:          common.Int64Ptr(6),\n\t\tEventCount:           common.Int64Ptr(6),\n\t}\n\ts.Equal(expectedHeader, blob.Header)\n\ts.NoError(err)\n\ts.True(itr.HasNext())\n\n\tblob, err = itr.Next()\n\ts.Error(err)\n\ts.Nil(blob)\n\ts.assertStateMatches(expectedIteratorState, itr)\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestNext_Success_TenCallsToNext() {\n\tbatchInfo := []int{}\n\tfor i := 0; i < 100; i++ {\n\t\tbatchInfo = append(batchInfo, []int{1, 2, 3, 4, 4, 3, 2, 1}...)\n\t}\n\tvar pages []page\n\tfor i := 0; i < 100; i++ {\n\t\tp := page{\n\t\t\tfirstbatchIdx:             i * 8,\n\t\t\tnumBatches:                8,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t}\n\t\tpages = append(pages, p)\n\t}\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\t// set target blob size size such that every 10 persistence pages is one group of history batches\n\titr := s.constructTestHistoryIterator(historyV2Manager, 20*10*testDefaultHistoryEventSize, nil)\n\texpectedIteratorState := historyIteratorState{\n\t\tFinishedIteration: false,\n\t\tNextEventID:       constants.FirstEventID,\n\t}\n\tfor i := 0; i < 10; i++ {\n\t\ts.assertStateMatches(expectedIteratorState, itr)\n\t\ts.True(itr.HasNext())\n\t\tblob, err := itr.Next()\n\t\ts.NoError(err)\n\t\ts.NotNil(blob)\n\t\texpectedHeader := &HistoryBlobHeader{\n\t\t\tDomainName:           common.StringPtr(testDomainName),\n\t\t\tDomainID:             common.StringPtr(testDomainID),\n\t\t\tWorkflowID:           common.StringPtr(testWorkflowID),\n\t\t\tRunID:                common.StringPtr(testRunID),\n\t\t\tIsLast:               common.BoolPtr(false),\n\t\t\tFirstFailoverVersion: common.Int64Ptr(1),\n\t\t\tLastFailoverVersion:  common.Int64Ptr(1),\n\t\t\tFirstEventID:         common.Int64Ptr(constants.FirstEventID + int64(i*200)),\n\t\t\tLastEventID:          common.Int64Ptr(int64(200 + (i * 200))),\n\t\t\tEventCount:           common.Int64Ptr(200),\n\t\t}\n\t\tif i == 9 {\n\t\t\texpectedHeader.IsLast = common.BoolPtr(true)\n\t\t}\n\t\ts.Equal(expectedHeader, blob.Header)\n\n\t\tif i < 9 {\n\t\t\texpectedIteratorState.FinishedIteration = false\n\t\t\texpectedIteratorState.NextEventID = int64(200*(i+1) + 1)\n\t\t} else {\n\t\t\texpectedIteratorState.NextEventID = 0\n\t\t\texpectedIteratorState.FinishedIteration = true\n\t\t}\n\t}\n\ts.assertStateMatches(expectedIteratorState, itr)\n\ts.False(itr.HasNext())\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestNext_Success_SameHistoryDifferentPage() {\n\tbatchInfo := []int{2, 4, 4, 3, 2, 1, 1, 2}\n\tpages := []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             2,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                2,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             4,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             5,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\teventsPerRead := 6\n\ttargetBlobSize := eventsPerRead * testDefaultHistoryEventSize\n\thistoryV2Manager := s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\titr1 := s.constructTestHistoryIterator(historyV2Manager, targetBlobSize, nil)\n\n\tpages = []page{\n\t\t{\n\t\t\tfirstbatchIdx:             0,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             1,\n\t\t\tnumBatches:                3,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             2,\n\t\t\tnumBatches:                1,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             3,\n\t\t\tnumBatches:                5,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t\t{\n\t\t\tfirstbatchIdx:             4,\n\t\t\tnumBatches:                4,\n\t\t\tfirstEventFailoverVersion: 1,\n\t\t\tlastEventFailoverVersion:  1,\n\t\t},\n\t}\n\thistoryV2Manager = s.constructMockHistoryV2Manager(batchInfo, -1, true, pages...)\n\titr2 := s.constructTestHistoryIterator(historyV2Manager, targetBlobSize, nil)\n\n\ttotalPages := 3\n\texpectedFirstEventID := []int64{1, 7, 14}\n\tfor i := 0; i != totalPages; i++ {\n\t\ts.True(itr1.HasNext())\n\t\thistory1, err := itr1.Next()\n\t\ts.NoError(err)\n\n\t\ts.True(itr2.HasNext())\n\t\thistory2, err := itr2.Next()\n\t\ts.NoError(err)\n\n\t\ts.Equal(history1.Header, history2.Header)\n\t\ts.Equal(len(history1.Body), len(history2.Body))\n\t\ts.Equal(expectedFirstEventID[i], history1.Body[0].Events[0].ID)\n\t\ts.Equal(expectedFirstEventID[i], history2.Body[0].Events[0].ID)\n\t}\n\texpectedIteratorState := historyIteratorState{\n\t\tNextEventID:       0,\n\t\tFinishedIteration: true,\n\t}\n\ts.assertStateMatches(expectedIteratorState, itr1)\n\ts.assertStateMatches(expectedIteratorState, itr2)\n\ts.False(itr1.HasNext())\n\ts.False(itr2.HasNext())\n\thistoryV2Manager.AssertExpectations(s.T())\n}\n\nfunc (s *HistoryIteratorSuite) TestNewIteratorWithState() {\n\titr := s.constructTestHistoryIterator(nil, testDefaultTargetHistoryBlobSize, nil)\n\ttestIteratorState := historyIteratorState{\n\t\tFinishedIteration: true,\n\t\tNextEventID:       4,\n\t}\n\titr.historyIteratorState = testIteratorState\n\tstateToken, err := itr.GetState()\n\ts.NoError(err)\n\n\tnewItr := s.constructTestHistoryIterator(nil, testDefaultTargetHistoryBlobSize, stateToken)\n\ts.assertStateMatches(testIteratorState, newItr)\n}\n\nfunc (s *HistoryIteratorSuite) constructMockHistoryV2Manager(batchInfo []int, returnErrorOnPage int, addNotExistCall bool, pages ...page) *mocks.HistoryV2Manager {\n\tmockHistoryV2Manager := &mocks.HistoryV2Manager{}\n\n\tfirstEventIDs := []int64{constants.FirstEventID}\n\tfor i, batchSize := range batchInfo {\n\t\tfirstEventIDs = append(firstEventIDs, firstEventIDs[i]+int64(batchSize))\n\t}\n\n\tfor i, p := range pages {\n\t\treq := &persistence.ReadHistoryBranchRequest{\n\t\t\tBranchToken: testBranchToken,\n\t\t\tMinEventID:  firstEventIDs[p.firstbatchIdx],\n\t\t\tMaxEventID:  constants.EndEventID,\n\t\t\tPageSize:    testDefaultPersistencePageSize,\n\t\t\tShardID:     common.IntPtr(testShardID),\n\t\t\tDomainName:  testDomainName,\n\t\t}\n\t\tif returnErrorOnPage == i {\n\t\t\tmockHistoryV2Manager.On(\"ReadHistoryBranchByBatch\", mock.Anything, req).Return(nil, errors.New(\"got error getting workflow execution history\"))\n\t\t\treturn mockHistoryV2Manager\n\t\t}\n\n\t\tresp := &persistence.ReadHistoryBranchByBatchResponse{\n\t\t\tHistory: s.constructHistoryBatches(batchInfo, p, firstEventIDs[p.firstbatchIdx]),\n\t\t}\n\t\tmockHistoryV2Manager.On(\"ReadHistoryBranchByBatch\", mock.Anything, req).Return(resp, nil)\n\t}\n\n\tif addNotExistCall {\n\t\treq := &persistence.ReadHistoryBranchRequest{\n\t\t\tBranchToken: testBranchToken,\n\t\t\tMinEventID:  firstEventIDs[len(firstEventIDs)-1],\n\t\t\tMaxEventID:  constants.EndEventID,\n\t\t\tPageSize:    testDefaultPersistencePageSize,\n\t\t\tShardID:     common.IntPtr(testShardID),\n\t\t\tDomainName:  testDomainName,\n\t\t}\n\t\tmockHistoryV2Manager.On(\"ReadHistoryBranchByBatch\", mock.Anything, req).Return(nil, &types.EntityNotExistsError{Message: \"Reach the end\"})\n\t}\n\n\treturn mockHistoryV2Manager\n}\n\nfunc (s *HistoryIteratorSuite) copyIteratorState(itr *historyIterator) historyIteratorState {\n\treturn itr.historyIteratorState\n}\n\nfunc (s *HistoryIteratorSuite) assertStateMatches(expected historyIteratorState, itr *historyIterator) {\n\ts.Equal(expected.NextEventID, itr.NextEventID)\n\ts.Equal(expected.FinishedIteration, itr.FinishedIteration)\n}\n\nfunc (s *HistoryIteratorSuite) constructHistoryBatches(batchInfo []int, page page, firstEventID int64) []*types.History {\n\tbatches := []*types.History{}\n\teventsID := firstEventID\n\tfor batchIdx, numEvents := range batchInfo[page.firstbatchIdx : page.firstbatchIdx+page.numBatches] {\n\t\tevents := []*types.HistoryEvent{}\n\t\tfor i := 0; i < numEvents; i++ {\n\t\t\tevent := &types.HistoryEvent{\n\t\t\t\tID:      eventsID,\n\t\t\t\tVersion: page.firstEventFailoverVersion,\n\t\t\t}\n\t\t\teventsID++\n\t\t\tif batchIdx == page.numBatches-1 {\n\t\t\t\tevent.Version = page.lastEventFailoverVersion\n\t\t\t}\n\t\t\tevents = append(events, event)\n\t\t}\n\t\tbatches = append(batches, &types.History{\n\t\t\tEvents: events,\n\t\t})\n\t}\n\treturn batches\n}\n\nfunc (s *HistoryIteratorSuite) constructTestHistoryIterator(\n\tmockHistoryV2Manager *mocks.HistoryV2Manager,\n\ttargetHistoryBlobSize int,\n\tinitialState []byte,\n) *historyIterator {\n\trequest := &ArchiveHistoryRequest{\n\t\tShardID:              testShardID,\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\titr := newHistoryIterator(context.Background(), request, mockHistoryV2Manager, targetHistoryBlobSize)\n\tif initialState != nil {\n\t\terr := itr.reset(initialState)\n\t\ts.NoError(err)\n\t}\n\titr.sizeEstimator = newTestSizeEstimator()\n\treturn itr\n}\n"
  },
  {
    "path": "common/archiver/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ArchiveHistoryRequest is request to Archive workflow history\n\tArchiveHistoryRequest struct {\n\t\tShardID              int\n\t\tDomainID             string\n\t\tDomainName           string\n\t\tWorkflowID           string\n\t\tRunID                string\n\t\tBranchToken          []byte\n\t\tNextEventID          int64\n\t\tCloseFailoverVersion int64\n\t}\n\n\t// GetHistoryRequest is the request to Get archived history\n\tGetHistoryRequest struct {\n\t\tDomainID             string\n\t\tWorkflowID           string\n\t\tRunID                string\n\t\tCloseFailoverVersion *int64\n\t\tNextPageToken        []byte\n\t\tPageSize             int\n\t}\n\n\t// GetHistoryResponse is the response of Get archived history\n\tGetHistoryResponse struct {\n\t\tHistoryBatches []*types.History\n\t\tNextPageToken  []byte\n\t}\n\n\t// HistoryBootstrapContainer contains components needed by all history Archiver implementations\n\tHistoryBootstrapContainer struct {\n\t\tHistoryV2Manager persistence.HistoryManager\n\t\tLogger           log.Logger\n\t\tMetricsClient    metrics.Client\n\t\tClusterMetadata  cluster.Metadata\n\t\tDomainCache      cache.DomainCache\n\t}\n\n\t// HistoryArchiver is used to archive history and read archived history\n\tHistoryArchiver interface {\n\t\tArchive(context.Context, URI, *ArchiveHistoryRequest, ...ArchiveOption) error\n\t\tGet(context.Context, URI, *GetHistoryRequest) (*GetHistoryResponse, error)\n\t\tValidateURI(URI) error\n\t}\n\n\t// VisibilityBootstrapContainer contains components needed by all visibility Archiver implementations\n\tVisibilityBootstrapContainer struct {\n\t\tLogger          log.Logger\n\t\tMetricsClient   metrics.Client\n\t\tClusterMetadata cluster.Metadata\n\t\tDomainCache     cache.DomainCache\n\t}\n\n\t// ArchiveVisibilityRequest is request to Archive single workflow visibility record\n\tArchiveVisibilityRequest struct {\n\t\tDomainID           string\n\t\tDomainName         string // doesn't need to be archived\n\t\tWorkflowID         string\n\t\tRunID              string\n\t\tWorkflowTypeName   string\n\t\tStartTimestamp     int64\n\t\tExecutionTimestamp int64\n\t\tCloseTimestamp     int64\n\t\tCloseStatus        types.WorkflowExecutionCloseStatus\n\t\tHistoryLength      int64\n\t\tMemo               *types.Memo\n\t\tSearchAttributes   map[string]string\n\t\tHistoryArchivalURI string\n\t}\n\n\t// QueryVisibilityRequest is the request to query archived visibility records\n\tQueryVisibilityRequest struct {\n\t\tDomainID      string\n\t\tPageSize      int\n\t\tNextPageToken []byte\n\t\tQuery         string\n\t}\n\n\t// QueryVisibilityResponse is the response of querying archived visibility records\n\tQueryVisibilityResponse struct {\n\t\tExecutions    []*types.WorkflowExecutionInfo\n\t\tNextPageToken []byte\n\t}\n\n\t// VisibilityArchiver is used to archive visibility and read archived visibility\n\tVisibilityArchiver interface {\n\t\tArchive(context.Context, URI, *ArchiveVisibilityRequest, ...ArchiveOption) error\n\t\tQuery(context.Context, URI, *QueryVisibilityRequest) (*QueryVisibilityResponse, error)\n\t\tValidateURI(URI) error\n\t}\n)\n"
  },
  {
    "path": "common/archiver/interface_mock.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage archiver\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n)\n\n// HistoryArchiverMock is an autogenerated mock type for the HistoryArchiver type\ntype HistoryArchiverMock struct {\n\tmock.Mock\n}\n\n// Archive provides a mock function with given fields: ctx, uri, request, opts\nfunc (_m *HistoryArchiverMock) Archive(ctx context.Context, uri URI, request *ArchiveHistoryRequest, opts ...ArchiveOption) error {\n\t_va := make([]interface{}, len(opts))\n\tfor _i := range opts {\n\t\t_va[_i] = opts[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, ctx, uri, request)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, URI, *ArchiveHistoryRequest, ...ArchiveOption) error); ok {\n\t\tr0 = rf(ctx, uri, request, opts...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Get provides a mock function with given fields: ctx, uri, request\nfunc (_m *HistoryArchiverMock) Get(ctx context.Context, uri URI, request *GetHistoryRequest) (*GetHistoryResponse, error) {\n\tret := _m.Called(ctx, uri, request)\n\n\tvar r0 *GetHistoryResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, URI, *GetHistoryRequest) *GetHistoryResponse); ok {\n\t\tr0 = rf(ctx, uri, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*GetHistoryResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, URI, *GetHistoryRequest) error); ok {\n\t\tr1 = rf(ctx, uri, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ValidateURI provides a mock function with given fields: uri\nfunc (_m *HistoryArchiverMock) ValidateURI(uri URI) error {\n\tret := _m.Called(uri)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(URI) error); ok {\n\t\tr0 = rf(uri)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// VisibilityArchiverMock is an autogenerated mock type for the VisibilityArchiver type\ntype VisibilityArchiverMock struct {\n\tmock.Mock\n}\n\n// Archive provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *VisibilityArchiverMock) Archive(_a0 context.Context, _a1 URI, _a2 *ArchiveVisibilityRequest, _a3 ...ArchiveOption) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, URI, *ArchiveVisibilityRequest, ...ArchiveOption) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Query provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *VisibilityArchiverMock) Query(_a0 context.Context, _a1 URI, _a2 *QueryVisibilityRequest) (*QueryVisibilityResponse, error) {\n\tret := _m.Called(_a0, _a1, _a2)\n\n\tvar r0 *QueryVisibilityResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, URI, *QueryVisibilityRequest) *QueryVisibilityResponse); ok {\n\t\tr0 = rf(_a0, _a1, _a2)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*QueryVisibilityResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, URI, *QueryVisibilityRequest) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ValidateURI provides a mock function with given fields: uri\nfunc (_m *VisibilityArchiverMock) ValidateURI(uri URI) error {\n\tret := _m.Called(uri)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(URI) error); ok {\n\t\tr0 = rf(uri)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/archiver/options.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"go.uber.org/cadence/activity\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype (\n\t// ArchiveOption is used to provide options for adding features to\n\t// the Archive method of History/Visibility Archiver\n\tArchiveOption func(featureCatalog *ArchiveFeatureCatalog)\n\n\t// ArchiveFeatureCatalog is a collection features for the Archive method of\n\t// History/Visibility Archiver\n\tArchiveFeatureCatalog struct {\n\t\tProgressManager          ProgressManager\n\t\tNonRetriableError        NonRetriableError\n\t\tArchiveIncompleteHistory dynamicproperties.BoolPropertyFn\n\t}\n\n\t// NonRetriableError returns an error indicating archiver has encountered an non-retriable error\n\tNonRetriableError func() error\n\n\t// ProgressManager is used to record and load archive progress\n\tProgressManager interface {\n\t\tRecordProgress(ctx context.Context, progress interface{}) error\n\t\tLoadProgress(ctx context.Context, valuePtr interface{}) error\n\t\tHasProgress(ctx context.Context) bool\n\t}\n)\n\n// GetFeatureCatalog applies all the ArchiveOptions to the catalog and returns the catalog.\n// It should be called inside the Archive method.\nfunc GetFeatureCatalog(opts ...ArchiveOption) *ArchiveFeatureCatalog {\n\tcatalog := &ArchiveFeatureCatalog{\n\t\tArchiveIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\tfor _, opt := range opts {\n\t\topt(catalog)\n\t}\n\treturn catalog\n}\n\n// GetHeartbeatArchiveOption returns an ArchiveOption for enabling heartbeating.\n// It should be used when the Archive method is invoked inside an activity.\nfunc GetHeartbeatArchiveOption() ArchiveOption {\n\treturn func(catalog *ArchiveFeatureCatalog) {\n\t\tcatalog.ProgressManager = &heartbeatProgressManager{}\n\t}\n}\n\ntype heartbeatProgressManager struct{}\n\nfunc (h *heartbeatProgressManager) RecordProgress(ctx context.Context, progress interface{}) error {\n\tactivity.RecordHeartbeat(ctx, progress)\n\treturn nil\n}\n\nfunc (h *heartbeatProgressManager) LoadProgress(ctx context.Context, valuePtr interface{}) error {\n\tif !h.HasProgress(ctx) {\n\t\treturn errors.New(\"no progress information in the context\")\n\t}\n\treturn activity.GetHeartbeatDetails(ctx, valuePtr)\n}\n\nfunc (h *heartbeatProgressManager) HasProgress(ctx context.Context) bool {\n\treturn activity.HasHeartbeatDetails(ctx)\n}\n\n// GetNonRetriableErrorOption returns an ArchiveOption so that archiver knows what should\n// be returned when an non-retryable error is encountered.\nfunc GetNonRetriableErrorOption(nonRetryableErr error) ArchiveOption {\n\treturn func(catalog *ArchiveFeatureCatalog) {\n\t\tcatalog.NonRetriableError = func() error {\n\t\t\treturn nonRetryableErr\n\t\t}\n\t}\n}\n\n// GetArchivingIncompleteHistoryOption returns an ArchiveOption so that archiver would archive incomplete history\nfunc GetArchivingIncompleteHistoryOption(allow dynamicproperties.BoolPropertyFn) ArchiveOption {\n\treturn func(catalog *ArchiveFeatureCatalog) {\n\t\tcatalog.ArchiveIncompleteHistory = allow\n\t}\n}\n"
  },
  {
    "path": "common/archiver/provider/init.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage provider\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/filestore\"\n\t\"github.com/uber/cadence/common/archiver/s3store\"\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc init() {\n\t// TODO: ideally remove this and handle per-instance registration during startup somehow,\n\t// as globals and inits have consistently caused issues.\n\t//\n\t// For now though, it's replacing a hard-coded switch statement, so an init func\n\t// is the most straightforward and should-be-identical conversion.\n\n\tmust := func(err error) {\n\t\tif err != nil {\n\t\t\tpanic(fmt.Errorf(\"failed to register default provider: %w\", err))\n\t\t}\n\t}\n\n\tmust(RegisterHistoryArchiver(filestore.URIScheme, config.FilestoreConfig, func(cfg *config.YamlNode, container *archiver.HistoryBootstrapContainer) (archiver.HistoryArchiver, error) {\n\t\tvar out *config.FilestoreArchiver\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn filestore.NewHistoryArchiver(container, out)\n\t}))\n\tmust(RegisterHistoryArchiver(s3store.URIScheme, config.S3storeConfig, func(cfg *config.YamlNode, container *archiver.HistoryBootstrapContainer) (archiver.HistoryArchiver, error) {\n\t\tvar out *config.S3Archiver\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn s3store.NewHistoryArchiver(container, out)\n\t}))\n\n\tmust(RegisterVisibilityArchiver(filestore.URIScheme, config.FilestoreConfig, func(cfg *config.YamlNode, container *archiver.VisibilityBootstrapContainer) (archiver.VisibilityArchiver, error) {\n\t\tvar out *config.FilestoreArchiver\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn filestore.NewVisibilityArchiver(container, out)\n\t}))\n\tmust(RegisterVisibilityArchiver(s3store.URIScheme, config.S3storeConfig, func(cfg *config.YamlNode, container *archiver.VisibilityBootstrapContainer) (archiver.VisibilityArchiver, error) {\n\t\tvar out *config.S3Archiver\n\t\tif err := cfg.Decode(&out); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t\t}\n\t\treturn s3store.NewVisibilityArchiver(container, out)\n\t}))\n}\n"
  },
  {
    "path": "common/archiver/provider/noop_provider.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage provider\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n)\n\ntype noOpArchiverProvider struct{}\n\nfunc NewNoOpArchiverProvider() ArchiverProvider {\n\treturn &noOpArchiverProvider{}\n}\n\nfunc (*noOpArchiverProvider) RegisterBootstrapContainer(\n\tserviceName string,\n\thistoryContainer *archiver.HistoryBootstrapContainer,\n\tvisibilityContainter *archiver.VisibilityBootstrapContainer,\n) error {\n\treturn nil\n}\n\nfunc (*noOpArchiverProvider) GetHistoryArchiver(scheme, serviceName string) (archiver.HistoryArchiver, error) {\n\treturn &noOpHistoryArchiver{}, nil\n}\n\nfunc (*noOpArchiverProvider) GetVisibilityArchiver(scheme, serviceName string) (archiver.VisibilityArchiver, error) {\n\treturn &noOpVisibilityArchiver{}, nil\n}\n\ntype noOpHistoryArchiver struct{}\n\nfunc (*noOpHistoryArchiver) Archive(context.Context, archiver.URI, *archiver.ArchiveHistoryRequest, ...archiver.ArchiveOption) error {\n\treturn nil\n}\n\nfunc (*noOpHistoryArchiver) Get(context.Context, archiver.URI, *archiver.GetHistoryRequest) (*archiver.GetHistoryResponse, error) {\n\treturn &archiver.GetHistoryResponse{}, nil\n}\n\nfunc (*noOpHistoryArchiver) ValidateURI(archiver.URI) error {\n\treturn nil\n}\n\ntype noOpVisibilityArchiver struct{}\n\nfunc (*noOpVisibilityArchiver) Archive(context.Context, archiver.URI, *archiver.ArchiveVisibilityRequest, ...archiver.ArchiveOption) error {\n\treturn nil\n}\n\nfunc (*noOpVisibilityArchiver) Query(context.Context, archiver.URI, *archiver.QueryVisibilityRequest) (*archiver.QueryVisibilityResponse, error) {\n\treturn &archiver.QueryVisibilityResponse{}, nil\n}\n\nfunc (*noOpVisibilityArchiver) ValidateURI(archiver.URI) error {\n\treturn nil\n}\n"
  },
  {
    "path": "common/archiver/provider/provider.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage provider\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/syncmap\"\n)\n\n//go:generate mockgen -package=$GOPACKAGE -destination=provider_mock.go -self_package=github.com/uber/cadence/common/archiver/provider github.com/uber/cadence/common/archiver/provider ArchiverProvider\n\nvar (\n\t// ErrUnknownScheme is the error for unknown archiver scheme\n\tErrUnknownScheme = errors.New(\"unknown archiver scheme\")\n\t// ErrNotSupported is the error for not supported archiver implementation\n\tErrNotSupported = errors.New(\"archiver provider not supported\")\n\t// ErrBootstrapContainerNotFound is the error for unable to find the bootstrap container given serviceName\n\tErrBootstrapContainerNotFound = errors.New(\"unable to find bootstrap container for the given service name\")\n\t// ErrArchiverConfigNotFound is the error for unable to find the config for an archiver given scheme\n\tErrArchiverConfigNotFound = errors.New(\"unable to find archiver config for the given scheme\")\n\t// ErrBootstrapContainerAlreadyRegistered is the error for registering multiple containers for the same serviceName\n\tErrBootstrapContainerAlreadyRegistered = errors.New(\"bootstrap container has already been registered\")\n)\n\ntype (\n\t// ArchiverProvider returns history or visibility archiver based on the scheme and serviceName.\n\t// The archiver for each combination of scheme and serviceName will be created only once and cached.\n\tArchiverProvider interface {\n\t\tRegisterBootstrapContainer(\n\t\t\tserviceName string,\n\t\t\thistoryContainer *archiver.HistoryBootstrapContainer,\n\t\t\tvisibilityContainter *archiver.VisibilityBootstrapContainer,\n\t\t) error\n\t\tGetHistoryArchiver(scheme, serviceName string) (archiver.HistoryArchiver, error)\n\t\tGetVisibilityArchiver(scheme, serviceName string) (archiver.VisibilityArchiver, error)\n\t}\n\n\tarchiverProvider struct {\n\t\tsync.RWMutex\n\n\t\thistoryArchiverConfigs    config.HistoryArchiverProvider\n\t\tvisibilityArchiverConfigs config.VisibilityArchiverProvider\n\n\t\t// Key for the container is just serviceName\n\t\thistoryContainers    map[string]*archiver.HistoryBootstrapContainer\n\t\tvisibilityContainers map[string]*archiver.VisibilityBootstrapContainer\n\n\t\t// Key for the archiver is scheme + serviceName\n\t\thistoryArchivers    map[string]archiver.HistoryArchiver\n\t\tvisibilityArchivers map[string]archiver.VisibilityArchiver\n\t}\n\n\thistoryConstructor struct {\n\t\tfn func(cfg *config.YamlNode, container *archiver.HistoryBootstrapContainer) (archiver.HistoryArchiver, error)\n\t\t// yaml key where this config exists, under archival.history.provider.\n\t\t// This almost certainly should be the same as the scheme, but that'll need more work.\n\t\tconfigKey string\n\t}\n\tvisibilityConstructor struct {\n\t\tfn func(cfg *config.YamlNode, container *archiver.VisibilityBootstrapContainer) (archiver.VisibilityArchiver, error)\n\t\t// yaml key where this config exists, under archival.visibility.provider.\n\t\t// This almost certainly should be the same as the scheme, but that'll need more work.\n\t\tconfigKey string\n\t}\n)\n\nvar (\n\thistoryConstructors    = syncmap.New[string, historyConstructor]()\n\tvisibilityConstructors = syncmap.New[string, visibilityConstructor]()\n)\n\nfunc RegisterHistoryArchiver(scheme, configKey string, constructor func(cfg *config.YamlNode, container *archiver.HistoryBootstrapContainer) (archiver.HistoryArchiver, error)) error {\n\tinserted := historyConstructors.Put(scheme, historyConstructor{\n\t\tfn:        constructor,\n\t\tconfigKey: configKey,\n\t})\n\tif !inserted {\n\t\treturn fmt.Errorf(\"history archiver already registered for scheme %q\", scheme)\n\t}\n\treturn nil\n}\n\nfunc RegisterVisibilityArchiver(scheme, configKey string, constructor func(cfg *config.YamlNode, container *archiver.VisibilityBootstrapContainer) (archiver.VisibilityArchiver, error)) error {\n\tinserted := visibilityConstructors.Put(scheme, visibilityConstructor{\n\t\tfn:        constructor,\n\t\tconfigKey: configKey,\n\t})\n\tif !inserted {\n\t\treturn fmt.Errorf(\"visibility archiver already registered for scheme %q\", scheme)\n\t}\n\treturn nil\n}\n\n// NewArchiverProvider returns a new Archiver provider\nfunc NewArchiverProvider(\n\thistoryArchiverConfigs config.HistoryArchiverProvider,\n\tvisibilityArchiverConfigs config.VisibilityArchiverProvider,\n) ArchiverProvider {\n\treturn &archiverProvider{\n\t\thistoryArchiverConfigs:    historyArchiverConfigs,\n\t\tvisibilityArchiverConfigs: visibilityArchiverConfigs,\n\t\thistoryContainers:         make(map[string]*archiver.HistoryBootstrapContainer),\n\t\tvisibilityContainers:      make(map[string]*archiver.VisibilityBootstrapContainer),\n\t\thistoryArchivers:          make(map[string]archiver.HistoryArchiver),\n\t\tvisibilityArchivers:       make(map[string]archiver.VisibilityArchiver),\n\t}\n}\n\n// RegisterBootstrapContainer stores the given bootstrap container given the serviceName\n// The container should be registered when a service starts up and before GetArchiver() is ever called.\n// Later calls to GetArchiver() will used the registered container to initialize new archivers.\n// If the container for a service has already registered, and this method is invoked for that service again\n// with an non-nil container, an error will be returned.\nfunc (p *archiverProvider) RegisterBootstrapContainer(\n\tserviceName string,\n\thistoryContainer *archiver.HistoryBootstrapContainer,\n\tvisibilityContainter *archiver.VisibilityBootstrapContainer,\n) error {\n\tp.Lock()\n\tdefer p.Unlock()\n\n\tif _, ok := p.historyContainers[serviceName]; ok && historyContainer != nil {\n\t\treturn ErrBootstrapContainerAlreadyRegistered\n\t}\n\tif _, ok := p.visibilityContainers[serviceName]; ok && visibilityContainter != nil {\n\t\treturn ErrBootstrapContainerAlreadyRegistered\n\t}\n\n\tif historyContainer != nil {\n\t\tp.historyContainers[serviceName] = historyContainer\n\t}\n\tif visibilityContainter != nil {\n\t\tp.visibilityContainers[serviceName] = visibilityContainter\n\t}\n\treturn nil\n}\n\nfunc (p *archiverProvider) GetHistoryArchiver(scheme, serviceName string) (historyArchiver archiver.HistoryArchiver, err error) {\n\tarchiverKey := p.getArchiverKey(scheme, serviceName)\n\tp.RLock()\n\tif historyArchiver, ok := p.historyArchivers[archiverKey]; ok {\n\t\tp.RUnlock()\n\t\treturn historyArchiver, nil\n\t}\n\tp.RUnlock()\n\n\tcontainer, ok := p.historyContainers[serviceName]\n\tif !ok {\n\t\treturn nil, ErrBootstrapContainerNotFound\n\t}\n\n\tconstructor, ok := historyConstructors.Get(scheme)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"no history archiver constructor for scheme %q\", scheme)\n\t}\n\n\tcfg, ok := p.historyArchiverConfigs[constructor.configKey]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"no history archiver config for scheme %q, config key %q\", scheme, constructor.configKey)\n\t}\n\n\thistoryArchiver, err = constructor.fn(cfg, container)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"history archiver constructor failed for scheme %q, config key %q: err: %w\", scheme, constructor.configKey, err)\n\t}\n\n\tp.Lock()\n\tdefer p.Unlock()\n\tif existingHistoryArchiver, ok := p.historyArchivers[archiverKey]; ok {\n\t\treturn existingHistoryArchiver, nil\n\t}\n\tp.historyArchivers[archiverKey] = historyArchiver\n\treturn historyArchiver, nil\n}\n\nfunc (p *archiverProvider) GetVisibilityArchiver(scheme, serviceName string) (archiver.VisibilityArchiver, error) {\n\tarchiverKey := p.getArchiverKey(scheme, serviceName)\n\tp.RLock()\n\tif visibilityArchiver, ok := p.visibilityArchivers[archiverKey]; ok {\n\t\tp.RUnlock()\n\t\treturn visibilityArchiver, nil\n\t}\n\tp.RUnlock()\n\n\tcontainer, ok := p.visibilityContainers[serviceName]\n\tif !ok {\n\t\treturn nil, ErrBootstrapContainerNotFound\n\t}\n\n\tconstructor, ok := visibilityConstructors.Get(scheme)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"no visibility archiver constructor for scheme %q\", scheme)\n\t}\n\n\tcfg, ok := p.visibilityArchiverConfigs[constructor.configKey]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"no visibility archiver config for scheme %q, config key %q\", scheme, constructor.configKey)\n\t}\n\n\tvisibilityArchiver, err := constructor.fn(cfg, container)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"visibility archiver constructor failed for scheme %q, config key %q: err: %w\", scheme, constructor.configKey, err)\n\t}\n\n\tp.Lock()\n\tdefer p.Unlock()\n\tif existingVisibilityArchiver, ok := p.visibilityArchivers[archiverKey]; ok {\n\t\treturn existingVisibilityArchiver, nil\n\t}\n\tp.visibilityArchivers[archiverKey] = visibilityArchiver\n\treturn visibilityArchiver, nil\n\n}\n\nfunc (p *archiverProvider) getArchiverKey(scheme, serviceName string) string {\n\treturn scheme + \":\" + serviceName\n}\n"
  },
  {
    "path": "common/archiver/provider/provider_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/archiver/provider (interfaces: ArchiverProvider)\n//\n// Generated by this command:\n//\n//\tmockgen -package=provider -destination=provider_mock.go -self_package=github.com/uber/cadence/common/archiver/provider github.com/uber/cadence/common/archiver/provider ArchiverProvider\n//\n\n// Package provider is a generated GoMock package.\npackage provider\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tarchiver \"github.com/uber/cadence/common/archiver\"\n)\n\n// MockArchiverProvider is a mock of ArchiverProvider interface.\ntype MockArchiverProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockArchiverProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockArchiverProviderMockRecorder is the mock recorder for MockArchiverProvider.\ntype MockArchiverProviderMockRecorder struct {\n\tmock *MockArchiverProvider\n}\n\n// NewMockArchiverProvider creates a new mock instance.\nfunc NewMockArchiverProvider(ctrl *gomock.Controller) *MockArchiverProvider {\n\tmock := &MockArchiverProvider{ctrl: ctrl}\n\tmock.recorder = &MockArchiverProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockArchiverProvider) EXPECT() *MockArchiverProviderMockRecorder {\n\treturn m.recorder\n}\n\n// GetHistoryArchiver mocks base method.\nfunc (m *MockArchiverProvider) GetHistoryArchiver(scheme, serviceName string) (archiver.HistoryArchiver, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryArchiver\", scheme, serviceName)\n\tret0, _ := ret[0].(archiver.HistoryArchiver)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryArchiver indicates an expected call of GetHistoryArchiver.\nfunc (mr *MockArchiverProviderMockRecorder) GetHistoryArchiver(scheme, serviceName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryArchiver\", reflect.TypeOf((*MockArchiverProvider)(nil).GetHistoryArchiver), scheme, serviceName)\n}\n\n// GetVisibilityArchiver mocks base method.\nfunc (m *MockArchiverProvider) GetVisibilityArchiver(scheme, serviceName string) (archiver.VisibilityArchiver, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityArchiver\", scheme, serviceName)\n\tret0, _ := ret[0].(archiver.VisibilityArchiver)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetVisibilityArchiver indicates an expected call of GetVisibilityArchiver.\nfunc (mr *MockArchiverProviderMockRecorder) GetVisibilityArchiver(scheme, serviceName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityArchiver\", reflect.TypeOf((*MockArchiverProvider)(nil).GetVisibilityArchiver), scheme, serviceName)\n}\n\n// RegisterBootstrapContainer mocks base method.\nfunc (m *MockArchiverProvider) RegisterBootstrapContainer(serviceName string, historyContainer *archiver.HistoryBootstrapContainer, visibilityContainter *archiver.VisibilityBootstrapContainer) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RegisterBootstrapContainer\", serviceName, historyContainer, visibilityContainter)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RegisterBootstrapContainer indicates an expected call of RegisterBootstrapContainer.\nfunc (mr *MockArchiverProviderMockRecorder) RegisterBootstrapContainer(serviceName, historyContainer, visibilityContainter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterBootstrapContainer\", reflect.TypeOf((*MockArchiverProvider)(nil).RegisterBootstrapContainer), serviceName, historyContainer, visibilityContainter)\n}\n"
  },
  {
    "path": "common/archiver/s3store/README.md",
    "content": "# Amazon S3 blobstore\n## Configuration\nSee https://docs.aws.amazon.com/sdk-for-go/v1/developer-guide/configuring-sdk.html#specifying-credentials on how to set up authentication against s3\n\nEnabling archival is done by using the configuration below. `Region` and `bucket URI` are required\n```\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      s3store:\n        region: \"us-east-1\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      s3store:\n        region: \"us-east-1\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"s3://<bucket-name>\"\n    visibility:\n      status: \"enabled\"\n      URI: \"s3://<bucket-name>\"\n```\n\n## Visibility query syntax\nYou can query the visibility store by using the `cadence workflow listarchived` command\n\nThe syntax for the query is based on SQL\n\nSupported column names are\n- WorkflowID *String*\n- WorkflowTypeName *String*\n- StartTime *Date*\n- CloseTime *Date*\n- SearchPrecision *String - Day, Hour, Minute, Second*\n\nWorkflowID or WorkflowTypeName is required. If filtering on date use StartTime or CloseTime in combination with SearchPrecision.\n\nSearching for a record will be done in times in the UTC timezone\n\nSearchPrecision specifies what range you want to search for records. If you use `SearchPrecision = 'Day'`\nit will search all records starting from `2020-01-21T00:00:00Z` to `2020-01-21T59:59:59Z` \n\n### Limitations\n\n- The only operator supported is `=` due to how records are stored in s3.\n\n### Example\n\n*Searches for all records done in day 2020-01-21 with the specified workflow id*\n\n`./cadence --do samples-domain workflow listarchived -q \"StartTime = '2020-01-21T00:00:00Z' AND WorkflowID='workflow-id' AND SearchPrecision='Day'\"`\n## Storage in S3\nWorkflow runs are stored in s3 using the following structure\n```\ns3://<bucket-name>/<domain-id>/\n\thistory/<workflow-id>/<run-id>\n\tvisibility/\n            workflowTypeName/<workflow-type-name>/\n                startTimeout/2020-01-21T16:16:11Z/<run-id>\n                closeTimeout/2020-01-21T16:16:11Z/<run-id>\n            workflowID/<workflow-id>/\n                startTimeout/2020-01-21T16:16:11Z/<run-id>\n                closeTimeout/2020-01-21T16:16:11Z/<run-id>\n```\n\n## Using localstack for local development\n1. Install awscli from [here](https://docs.aws.amazon.com/cli/latest/userguide/cli-chap-install.html)\n2. Install localstack from [here](https://github.com/localstack/localstack#installing)\n3. Launch localstack with `SERVICES=s3 localstack start`\n4. Create a bucket using `aws --endpoint-url=http://localhost:4572 s3 mb s3://cadence-development` \n5. Configure archival and domainDefaults with the following configuration\n```\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      s3store:\n        region: \"us-east-1\"\n        endpoint: \"http://127.0.0.1:4572\"\n        s3ForcePathStyle: true\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      s3store:\n        region: \"us-east-1\"\n        endpoint: \"http://127.0.0.1:4572\"\n        s3ForcePathStyle: true\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"s3://cadence-development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"s3://cadence-development\"\n```\n"
  },
  {
    "path": "common/archiver/s3store/historyArchiver.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// S3 History Archiver will archive workflow histories to amazon s3\n\npackage s3store\n\nimport (\n\t\"context\"\n\t\"encoding/binary\"\n\t\"errors\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/awserr\"\n\t\"github.com/aws/aws-sdk-go/aws/request\"\n\t\"github.com/aws/aws-sdk-go/aws/session\"\n\t\"github.com/aws/aws-sdk-go/service/s3\"\n\t\"github.com/aws/aws-sdk-go/service/s3/s3iface\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// URIScheme is the scheme for the s3 implementation\n\tURIScheme               = \"s3\"\n\terrEncodeHistory        = \"failed to encode history batches\"\n\terrWriteKey             = \"failed to write history to s3\"\n\tdefaultBlobstoreTimeout = 60 * time.Second\n\ttargetHistoryBlobSize   = 2 * 1024 * 1024 // 2MB\n)\n\nvar (\n\terrNoBucketSpecified = errors.New(\"no bucket specified\")\n\terrBucketNotExists   = errors.New(\"requested bucket does not exist\")\n\terrEmptyAwsRegion    = errors.New(\"empty aws region\")\n)\n\ntype (\n\thistoryArchiver struct {\n\t\tcontainer *archiver.HistoryBootstrapContainer\n\t\ts3cli     s3iface.S3API\n\t\t// only set in test code\n\t\thistoryIterator archiver.HistoryIterator\n\t}\n\n\tgetHistoryToken struct {\n\t\tCloseFailoverVersion int64\n\t\tBatchIdx             int\n\t}\n\n\tuploadProgress struct {\n\t\tBatchIdx      int\n\t\tIteratorState []byte\n\t\tuploadedSize  int64\n\t\thistorySize   int64\n\t}\n)\n\n// NewHistoryArchiver creates a new archiver.HistoryArchiver based on s3\nfunc NewHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\tconfig *config.S3Archiver,\n) (archiver.HistoryArchiver, error) {\n\treturn newHistoryArchiver(container, config, nil)\n}\n\nfunc newHistoryArchiver(\n\tcontainer *archiver.HistoryBootstrapContainer,\n\tconfig *config.S3Archiver,\n\thistoryIterator archiver.HistoryIterator,\n) (*historyArchiver, error) {\n\tif len(config.Region) == 0 {\n\t\treturn nil, errEmptyAwsRegion\n\t}\n\ts3Config := &aws.Config{\n\t\tEndpoint:         config.Endpoint,\n\t\tRegion:           aws.String(config.Region),\n\t\tS3ForcePathStyle: aws.Bool(config.S3ForcePathStyle),\n\t}\n\tsess, err := session.NewSession(s3Config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &historyArchiver{\n\t\tcontainer:       container,\n\t\ts3cli:           s3.New(sess),\n\t\thistoryIterator: historyIterator,\n\t}, nil\n}\nfunc (h *historyArchiver) Archive(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.ArchiveHistoryRequest,\n\topts ...archiver.ArchiveOption,\n) (err error) {\n\tscope := h.container.MetricsClient.Scope(metrics.HistoryArchiverScope, metrics.DomainTag(request.DomainName))\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif persistence.IsTransientError(err) || isRetryableError(err) {\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverArchiveTransientErrorCount)\n\t\t\t} else {\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverArchiveNonRetryableErrorCount)\n\t\t\t\tif featureCatalog.NonRetriableError != nil {\n\t\t\t\t\terr = featureCatalog.NonRetriableError()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}()\n\n\tlogger := archiver.TagLoggerWithArchiveHistoryRequestAndURI(h.container.Logger, request, URI.String())\n\n\tif err := softValidateURI(URI); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidURI), tag.Error(err))\n\t\treturn err\n\t}\n\n\tif err := archiver.ValidateHistoryArchiveRequest(request); err != nil {\n\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiver.ErrReasonInvalidArchiveRequest), tag.Error(err))\n\t\treturn err\n\t}\n\n\tvar progress uploadProgress\n\thistoryIterator := h.historyIterator\n\tif historyIterator == nil { // will only be set by testing code\n\t\thistoryIterator = loadHistoryIterator(ctx, request, h.container.HistoryV2Manager, featureCatalog, &progress)\n\t}\n\tfor historyIterator.HasNext() {\n\t\thistoryBlob, err := getNextHistoryBlob(ctx, historyIterator)\n\t\tif err != nil {\n\t\t\tif common.IsEntityNotExistsError(err) {\n\t\t\t\t// workflow history no longer exists, may due to duplicated archival signal\n\t\t\t\t// this may happen even in the middle of iterating history as two archival signals\n\t\t\t\t// can be processed concurrently.\n\t\t\t\tlogger.Info(archiver.ArchiveSkippedInfoMsg)\n\t\t\t\tscope.IncCounter(metrics.HistoryArchiverDuplicateArchivalsCount)\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tlogger := logger.WithTags(tag.ArchivalArchiveFailReason(archiver.ErrReasonReadHistory), tag.Error(err))\n\t\t\tif persistence.IsTransientError(err) {\n\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg)\n\t\t\t} else {\n\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg)\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\tif archiver.IsHistoryMutated(request, historyBlob.Body, *historyBlob.Header.IsLast, logger) {\n\t\t\tif !featureCatalog.ArchiveIncompleteHistory() {\n\t\t\t\treturn archiver.ErrHistoryMutated\n\t\t\t}\n\t\t}\n\n\t\tencodedHistoryBlob, err := encode(historyBlob)\n\t\tif err != nil {\n\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(errEncodeHistory), tag.Error(err))\n\t\t\treturn err\n\t\t}\n\n\t\tkey := constructHistoryKey(URI.Path(), request.DomainID, request.WorkflowID, request.RunID, request.CloseFailoverVersion, progress.BatchIdx)\n\n\t\texists, err := keyExists(ctx, h.s3cli, URI, key)\n\t\tif err != nil {\n\t\t\tlogger := logger.WithTags(tag.ArchivalArchiveFailReason(errWriteKey), tag.Error(err))\n\t\t\tif isRetryableError(err) {\n\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg)\n\t\t\t} else {\n\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg)\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\tblobSize := int64(binary.Size(encodedHistoryBlob))\n\t\tif exists {\n\t\t\tscope.IncCounter(metrics.HistoryArchiverBlobExistsCount)\n\t\t} else {\n\t\t\tif err := upload(ctx, h.s3cli, URI, key, encodedHistoryBlob); err != nil {\n\t\t\t\tlogger := logger.WithTags(tag.ArchivalArchiveFailReason(errWriteKey), tag.Error(err))\n\t\t\t\tif isRetryableError(err) {\n\t\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg)\n\t\t\t\t} else {\n\t\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg)\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tprogress.uploadedSize += blobSize\n\t\t\tscope.RecordTimer(metrics.HistoryArchiverBlobSize, time.Duration(blobSize))\n\t\t}\n\n\t\tprogress.historySize += blobSize\n\t\tprogress.BatchIdx = progress.BatchIdx + 1\n\t\tsaveHistoryIteratorState(ctx, featureCatalog, historyIterator, &progress)\n\t}\n\n\tscope.RecordTimer(metrics.HistoryArchiverTotalUploadSize, time.Duration(progress.uploadedSize))\n\tscope.RecordTimer(metrics.HistoryArchiverHistorySize, time.Duration(progress.historySize))\n\tscope.IncCounter(metrics.HistoryArchiverArchiveSuccessCount)\n\treturn nil\n}\n\nfunc loadHistoryIterator(ctx context.Context, request *archiver.ArchiveHistoryRequest, historyManager persistence.HistoryManager, featureCatalog *archiver.ArchiveFeatureCatalog, progress *uploadProgress) (historyIterator archiver.HistoryIterator) {\n\tif featureCatalog.ProgressManager != nil {\n\t\tif featureCatalog.ProgressManager.HasProgress(ctx) {\n\t\t\terr := featureCatalog.ProgressManager.LoadProgress(ctx, progress)\n\t\t\tif err == nil {\n\t\t\t\thistoryIterator, err := archiver.NewHistoryIteratorFromState(ctx, request, historyManager, targetHistoryBlobSize, progress.IteratorState)\n\t\t\t\tif err == nil {\n\t\t\t\t\treturn historyIterator\n\t\t\t\t}\n\t\t\t}\n\t\t\tprogress.IteratorState = nil\n\t\t\tprogress.BatchIdx = 0\n\t\t\tprogress.historySize = 0\n\t\t\tprogress.uploadedSize = 0\n\t\t}\n\t}\n\treturn archiver.NewHistoryIterator(ctx, request, historyManager, targetHistoryBlobSize)\n}\n\nfunc saveHistoryIteratorState(ctx context.Context, featureCatalog *archiver.ArchiveFeatureCatalog, historyIterator archiver.HistoryIterator, progress *uploadProgress) {\n\t// Saving history state is a best effort operation. Ignore errors and continue\n\tif featureCatalog.ProgressManager != nil {\n\t\tstate, err := historyIterator.GetState()\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t\tprogress.IteratorState = state\n\t\terr = featureCatalog.ProgressManager.RecordProgress(ctx, progress)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (h *historyArchiver) Get(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.GetHistoryRequest,\n) (*archiver.GetHistoryResponse, error) {\n\tif err := softValidateURI(URI); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateGetRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidGetHistoryRequest.Error()}\n\t}\n\n\tvar err error\n\tvar token *getHistoryToken\n\tif request.NextPageToken != nil {\n\t\ttoken, err = deserializeGetHistoryToken(request.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: archiver.ErrNextPageTokenCorrupted.Error()}\n\t\t}\n\t} else if request.CloseFailoverVersion != nil {\n\t\ttoken = &getHistoryToken{\n\t\t\tCloseFailoverVersion: *request.CloseFailoverVersion,\n\t\t}\n\t} else {\n\t\thighestVersion, err := h.getHighestVersion(ctx, URI, request)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: err.Error()}\n\t\t}\n\t\ttoken = &getHistoryToken{\n\t\t\tCloseFailoverVersion: *highestVersion,\n\t\t}\n\t}\n\n\tresponse := &archiver.GetHistoryResponse{}\n\tnumOfEvents := 0\n\tisTruncated := false\n\tfor {\n\t\tif numOfEvents >= request.PageSize {\n\t\t\tisTruncated = true\n\t\t\tbreak\n\t\t}\n\t\tkey := constructHistoryKey(URI.Path(), request.DomainID, request.WorkflowID, request.RunID, token.CloseFailoverVersion, token.BatchIdx)\n\n\t\tencodedRecord, err := download(ctx, h.s3cli, URI, key)\n\t\tif err != nil {\n\t\t\tif isRetryableError(err) {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t\t}\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.BadRequestError, *types.InternalServiceError, *types.EntityNotExistsError:\n\t\t\t\treturn nil, err\n\t\t\tdefault:\n\t\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t\t}\n\t\t}\n\n\t\thistoryBlob, err := decodeHistoryBlob(encodedRecord)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\tfor _, batch := range historyBlob.Body {\n\t\t\tresponse.HistoryBatches = append(response.HistoryBatches, batch)\n\t\t\tnumOfEvents += len(batch.Events)\n\t\t}\n\n\t\tif *historyBlob.Header.IsLast {\n\t\t\tbreak\n\t\t}\n\t\ttoken.BatchIdx++\n\t}\n\n\tif isTruncated {\n\t\tnextToken, err := serializeToken(token)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\tresponse.NextPageToken = nextToken\n\t}\n\n\treturn response, nil\n}\n\nfunc (h *historyArchiver) ValidateURI(URI archiver.URI) error {\n\terr := softValidateURI(URI)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn bucketExists(context.TODO(), h.s3cli, URI)\n}\n\nfunc getNextHistoryBlob(ctx context.Context, historyIterator archiver.HistoryIterator) (*archiver.HistoryBlob, error) {\n\thistoryBlob, err := historyIterator.Next()\n\top := func(ctx context.Context) error {\n\t\thistoryBlob, err = historyIterator.Next()\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreatePersistenceRetryPolicy()),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\tfor err != nil {\n\t\tif contextExpired(ctx) {\n\t\t\treturn nil, archiver.ErrContextTimeout\n\t\t}\n\t\tif !persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\terr = throttleRetry.Do(ctx, op)\n\t}\n\treturn historyBlob, nil\n}\n\n// with XDC(global domain) concept, archival may write different history with the same RunID, with different failoverVersion.\n// In that case, the history/runID with the highest failoverVersion wins.\n// getHighestVersion look up all archived files to find the highest failoverVersion.\nfunc (h *historyArchiver) getHighestVersion(ctx context.Context, URI archiver.URI, request *archiver.GetHistoryRequest) (*int64, error) {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\tvar prefix = constructHistoryKeyPrefix(URI.Path(), request.DomainID, request.WorkflowID, request.RunID) + \"/\"\n\tresults, err := h.s3cli.ListObjectsV2WithContext(ctx, &s3.ListObjectsV2Input{\n\t\tBucket:    aws.String(URI.Hostname()),\n\t\tPrefix:    aws.String(prefix),\n\t\tDelimiter: aws.String(\"/\"),\n\t})\n\tif err != nil {\n\t\tif aerr, ok := err.(awserr.Error); ok && aerr.Code() == s3.ErrCodeNoSuchBucket {\n\t\t\treturn nil, &types.BadRequestError{Message: errBucketNotExists.Error()}\n\t\t}\n\t\treturn nil, err\n\t}\n\tvar highestVersion *int64\n\n\tfor _, v := range results.CommonPrefixes {\n\t\tvar version int64\n\t\tversion, err = strconv.ParseInt(strings.Replace(strings.Replace(*v.Prefix, prefix, \"\", 1), \"/\", \"\", 1), 10, 64)\n\t\tif err != nil {\n\t\t\tcontinue\n\t\t}\n\t\tif highestVersion == nil || version > *highestVersion {\n\t\t\thighestVersion = &version\n\t\t}\n\t}\n\tif highestVersion == nil {\n\t\treturn nil, archiver.ErrHistoryNotExist\n\t}\n\treturn highestVersion, nil\n}\n\nfunc isRetryableError(err error) bool {\n\tif err == nil {\n\t\treturn false\n\t}\n\tif aerr, ok := err.(awserr.Error); ok {\n\t\treturn isStatusCodeRetryable(aerr) || request.IsErrorRetryable(aerr) || request.IsErrorThrottle(aerr)\n\t}\n\treturn false\n}\n\nfunc isStatusCodeRetryable(err error) bool {\n\tif aerr, ok := err.(awserr.Error); ok {\n\t\tif rerr, ok := err.(awserr.RequestFailure); ok {\n\t\t\tif rerr.StatusCode() == 429 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tif rerr.StatusCode() >= 500 && rerr.StatusCode() != 501 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t\treturn isStatusCodeRetryable(aerr.OrigErr())\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/archiver/s3store/historyArchiver_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage s3store\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/awserr\"\n\t\"github.com/aws/aws-sdk-go/aws/request\"\n\t\"github.com/aws/aws-sdk-go/service/s3\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/s3store/mocks\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestDomainID             = \"test-domain-id\"\n\ttestDomainName           = \"test-domain-name\"\n\ttestWorkflowID           = \"test-workflow-id\"\n\ttestRunID                = \"test-run-id\"\n\ttestNextEventID          = 1800\n\ttestCloseFailoverVersion = 100\n\ttestPageSize             = 100\n\ttestBucket               = \"test-bucket\"\n\ttestBucketURI            = \"s3://test-bucket\"\n)\n\nvar (\n\ttestBranchToken = []byte{1, 2, 3}\n)\n\ntype historyArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\ts3cli              *mocks.S3API\n\tcontainer          *archiver.HistoryBootstrapContainer\n\ttestArchivalURI    archiver.URI\n\thistoryBatchesV1   []*archiver.HistoryBlob\n\thistoryBatchesV100 []*archiver.HistoryBlob\n}\n\nfunc TestHistoryArchiverSuite(t *testing.T) {\n\tsuite.Run(t, new(historyArchiverSuite))\n}\n\nfunc (s *historyArchiverSuite) SetupSuite() {\n\tvar err error\n\ts.s3cli = &mocks.S3API{}\n\tsetupFsEmulation(s.s3cli)\n\ts.setupHistoryDirectory()\n\ts.testArchivalURI, err = archiver.NewURI(testBucketURI)\n\n\ts.Require().NoError(err)\n}\n\nfunc (s *historyArchiverSuite) TearDownSuite() {\n}\n\nfunc (s *historyArchiverSuite) SetupTest() {\n\tscope := tally.NewTestScope(\"test\", nil)\n\ts.Assertions = require.New(s.T())\n\ts.container = &archiver.HistoryBootstrapContainer{\n\t\tLogger:        testlogger.New(s.T()),\n\t\tMetricsClient: metrics.NewClient(scope, metrics.History, metrics.HistogramMigration{}),\n\t}\n}\n\nfunc setupFsEmulation(s3cli *mocks.S3API) {\n\tfs := make(map[string][]byte)\n\n\tputObjectFn := func(_ aws.Context, input *s3.PutObjectInput, _ ...request.Option) *s3.PutObjectOutput {\n\t\tbuf := new(bytes.Buffer)\n\t\tbuf.ReadFrom(input.Body)\n\t\tfs[*input.Bucket+*input.Key] = buf.Bytes()\n\t\treturn &s3.PutObjectOutput{}\n\t}\n\tgetObjectFn := func(_ aws.Context, input *s3.GetObjectInput, _ ...request.Option) *s3.GetObjectOutput {\n\t\treturn &s3.GetObjectOutput{\n\t\t\tBody: ioutil.NopCloser(bytes.NewReader(fs[*input.Bucket+*input.Key])),\n\t\t}\n\t}\n\ts3cli.On(\"ListObjectsV2WithContext\", mock.Anything, mock.Anything).\n\t\tReturn(func(_ context.Context, input *s3.ListObjectsV2Input, opts ...request.Option) *s3.ListObjectsV2Output {\n\t\t\tobjects := make([]*s3.Object, 0)\n\t\t\tcommonPrefixMap := map[string]bool{}\n\t\t\tfor k := range fs {\n\t\t\t\tif strings.HasPrefix(k, *input.Bucket+*input.Prefix) {\n\t\t\t\t\tkey := k[len(*input.Bucket):]\n\t\t\t\t\tkeyWithoutPrefix := key[len(*input.Prefix):]\n\t\t\t\t\tindex := strings.Index(keyWithoutPrefix, \"/\")\n\t\t\t\t\tif index == -1 || input.Delimiter == nil {\n\t\t\t\t\t\tobjects = append(objects, &s3.Object{\n\t\t\t\t\t\t\tKey: aws.String(key),\n\t\t\t\t\t\t})\n\t\t\t\t\t} else {\n\t\t\t\t\t\tcommonPrefixMap[key[:len(*input.Prefix)+index]] = true\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tcommonPrefixes := make([]*s3.CommonPrefix, 0)\n\t\t\tfor k := range commonPrefixMap {\n\t\t\t\tcommonPrefixes = append(commonPrefixes, &s3.CommonPrefix{\n\t\t\t\t\tPrefix: aws.String(k),\n\t\t\t\t})\n\t\t\t}\n\n\t\t\tsort.SliceStable(objects, func(i, j int) bool {\n\t\t\t\treturn *objects[i].Key < *objects[j].Key\n\t\t\t})\n\t\t\tmaxKeys := 1000\n\t\t\tif input.MaxKeys != nil {\n\t\t\t\tmaxKeys = int(*input.MaxKeys)\n\t\t\t}\n\t\t\tstart := 0\n\t\t\tif input.ContinuationToken != nil {\n\t\t\t\tstart, _ = strconv.Atoi(*input.ContinuationToken)\n\t\t\t}\n\n\t\t\tif input.StartAfter != nil {\n\t\t\t\tfor k, v := range objects {\n\t\t\t\t\tif *input.StartAfter == *v.Key {\n\t\t\t\t\t\tstart = k + 1\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tisTruncated := false\n\t\t\tvar nextContinuationToken *string\n\t\t\tif len(objects) > start+maxKeys {\n\t\t\t\tisTruncated = true\n\t\t\t\tnextContinuationToken = common.StringPtr(fmt.Sprintf(\"%d\", start+maxKeys))\n\t\t\t\tobjects = objects[start : start+maxKeys]\n\t\t\t} else {\n\t\t\t\tobjects = objects[start:]\n\t\t\t}\n\n\t\t\tif input.StartAfter != nil {\n\t\t\t\tfor k, v := range commonPrefixes {\n\t\t\t\t\tif *input.StartAfter == *v.Prefix {\n\t\t\t\t\t\tstart = k + 1\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif len(commonPrefixes) > start+maxKeys {\n\t\t\t\tisTruncated = true\n\t\t\t\tnextContinuationToken = common.StringPtr(fmt.Sprintf(\"%d\", start+maxKeys))\n\t\t\t\tcommonPrefixes = commonPrefixes[start : start+maxKeys]\n\t\t\t} else if len(commonPrefixes) > 0 {\n\t\t\t\tcommonPrefixes = commonPrefixes[start:]\n\t\t\t}\n\n\t\t\treturn &s3.ListObjectsV2Output{\n\t\t\t\tCommonPrefixes:        commonPrefixes,\n\t\t\t\tContents:              objects,\n\t\t\t\tIsTruncated:           &isTruncated,\n\t\t\t\tNextContinuationToken: nextContinuationToken,\n\t\t\t}\n\t\t}, nil)\n\ts3cli.On(\"PutObjectWithContext\", mock.Anything, mock.Anything).Return(putObjectFn, nil)\n\n\ts3cli.On(\"HeadObjectWithContext\", mock.Anything, mock.MatchedBy(func(input *s3.HeadObjectInput) bool {\n\t\t_, ok := fs[*input.Bucket+*input.Key]\n\t\treturn !ok\n\t})).Return(nil, awserr.New(\"NotFound\", \"\", nil))\n\ts3cli.On(\"HeadObjectWithContext\", mock.Anything, mock.Anything).Return(&s3.HeadObjectOutput{}, nil)\n\n\ts3cli.On(\"GetObjectWithContext\", mock.Anything, mock.MatchedBy(func(input *s3.GetObjectInput) bool {\n\t\t_, ok := fs[*input.Bucket+*input.Key]\n\t\treturn !ok\n\t})).Return(nil, awserr.New(s3.ErrCodeNoSuchKey, \"\", nil))\n\ts3cli.On(\"GetObjectWithContext\", mock.Anything, mock.Anything).Return(getObjectFn, nil)\n}\n\nfunc (s *historyArchiverSuite) TestValidateURI() {\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"s3://\",\n\t\t\texpectedErr: errNoBucketSpecified,\n\t\t},\n\t\t{\n\t\t\tURI:         \"s3://bucket/a/b/c\",\n\t\t\texpectedErr: errBucketNotExists,\n\t\t},\n\t\t{\n\t\t\tURI:         testBucketURI,\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\ts.s3cli.On(\"HeadBucketWithContext\", mock.Anything, mock.MatchedBy(func(input *s3.HeadBucketInput) bool {\n\t\treturn *input.Bucket != s.testArchivalURI.Hostname()\n\t})).Return(nil, awserr.New(\"NotFound\", \"\", nil))\n\ts.s3cli.On(\"HeadBucketWithContext\", mock.Anything, mock.Anything).Return(&s3.HeadBucketOutput{}, nil)\n\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedErr, historyArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_InvalidURI() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_InvalidRequest() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           \"\", // an invalid request\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_ErrorOnReadHistory() {\n\tmockCtrl := gomock.NewController(s.T())\n\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"some random error\")),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_TimeoutWhenReadingHistory() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.ServiceBusyError{}),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(getCanceledContext(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_HistoryMutated() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion + 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Fail_NonRetriableErrorOption() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, errors.New(\"some random error\")),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tnonRetryableErr := errors.New(\"some non-retryable error\")\n\terr := historyArchiver.Archive(context.Background(), s.testArchivalURI, request, archiver.GetNonRetriableErrorOption(nonRetryableErr))\n\ts.Equal(nonRetryableErr, err)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Skip() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(false),\n\t\t},\n\t\tBody: []*types.History{\n\t\t\t{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:        constants.FirstEventID,\n\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(nil, &types.EntityNotExistsError{Message: \"workflow not found\"}),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(testBucketURI + \"/TestArchive_Skip\")\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\ts.NoError(err)\n\n\texpectedkey := constructHistoryKey(\"\", testDomainID, testWorkflowID, testRunID, testCloseFailoverVersion, 0)\n\ts.assertKeyExists(expectedkey)\n}\n\nfunc (s *historyArchiverSuite) TestArchive_Success() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\thistoryBatches := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        constants.FirstEventID + 2,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\thistoryBlob := &archiver.HistoryBlob{\n\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\tIsLast: common.BoolPtr(true),\n\t\t},\n\t\tBody: historyBatches,\n\t}\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(historyBlob, nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(false),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\trequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(testBucketURI + \"/TestArchive_Success\")\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, request)\n\ts.NoError(err)\n\n\texpectedkey := constructHistoryKey(\"\", testDomainID, testWorkflowID, testRunID, testCloseFailoverVersion, 0)\n\ts.assertKeyExists(expectedkey)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidURI() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   100,\n\t}\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidRequest() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   0, // pageSize should be greater than 0\n\t}\n\tresponse, err := historyArchiver.Get(context.Background(), s.testArchivalURI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_InvalidToken() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:      testDomainID,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tPageSize:      testPageSize,\n\t\tNextPageToken: []byte{'r', 'a', 'n', 'd', 'o', 'm'},\n\t}\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Fail_KeyNotExist() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             testPageSize,\n\t\tCloseFailoverVersion: common.Int64Ptr(testCloseFailoverVersion),\n\t}\n\tURI, err := archiver.NewURI(\"s3://test-bucket/non-existent\")\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.Nil(response)\n\ts.Error(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_PickHighestVersion() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(append(s.historyBatchesV100[0].Body, s.historyBatchesV100[1].Body...), response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_UseProvidedVersion() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             testPageSize,\n\t\tCloseFailoverVersion: common.Int64Ptr(1),\n\t}\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(s.historyBatchesV1[0].Body, response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) TestGet_Success_SmallPageSize() {\n\thistoryArchiver := s.newTestHistoryArchiver(nil)\n\trequest := &archiver.GetHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tPageSize:             1,\n\t\tCloseFailoverVersion: common.Int64Ptr(100),\n\t}\n\tcombinedHistory := []*types.History{}\n\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.NotNil(response.NextPageToken)\n\ts.NotNil(response.HistoryBatches)\n\ts.Len(response.HistoryBatches, 1)\n\tcombinedHistory = append(combinedHistory, response.HistoryBatches...)\n\n\trequest.NextPageToken = response.NextPageToken\n\tresponse, err = historyArchiver.Get(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.NotNil(response.HistoryBatches)\n\ts.Len(response.HistoryBatches, 1)\n\tcombinedHistory = append(combinedHistory, response.HistoryBatches...)\n\n\ts.Equal(append(s.historyBatchesV100[0].Body, s.historyBatchesV100[1].Body...), combinedHistory)\n}\n\nfunc (s *historyArchiverSuite) TestArchiveAndGet() {\n\tmockCtrl := gomock.NewController(s.T())\n\thistoryIterator := archiver.NewMockHistoryIterator(mockCtrl)\n\tgomock.InOrder(\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(s.historyBatchesV100[0], nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(true),\n\t\thistoryIterator.EXPECT().Next().Return(s.historyBatchesV100[1], nil),\n\t\thistoryIterator.EXPECT().HasNext().Return(false),\n\t)\n\n\thistoryArchiver := s.newTestHistoryArchiver(historyIterator)\n\tarchiveRequest := &archiver.ArchiveHistoryRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t}\n\tURI, err := archiver.NewURI(testBucketURI + \"/TestArchiveAndGet\")\n\ts.NoError(err)\n\terr = historyArchiver.Archive(context.Background(), URI, archiveRequest)\n\ts.NoError(err)\n\n\tgetRequest := &archiver.GetHistoryRequest{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tPageSize:   testPageSize,\n\t}\n\tresponse, err := historyArchiver.Get(context.Background(), URI, getRequest)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Equal(append(s.historyBatchesV100[0].Body, s.historyBatchesV100[1].Body...), response.HistoryBatches)\n}\n\nfunc (s *historyArchiverSuite) newTestHistoryArchiver(historyIterator archiver.HistoryIterator) *historyArchiver {\n\t// config := &config.S3Archiver{}\n\t// archiver, err := newHistoryArchiver(s.container, config, historyIterator)\n\tarchiver := &historyArchiver{\n\t\tcontainer:       s.container,\n\t\ts3cli:           s.s3cli,\n\t\thistoryIterator: historyIterator,\n\t}\n\treturn archiver\n}\n\nfunc (s *historyArchiverSuite) setupHistoryDirectory() {\n\ts.historyBatchesV1 = []*archiver.HistoryBlob{\n\t\t{\n\t\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\t\tIsLast: common.BoolPtr(true),\n\t\t\t},\n\t\t\tBody: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\t\tVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.historyBatchesV100 = []*archiver.HistoryBlob{\n\t\t{\n\t\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\t\tIsLast: common.BoolPtr(false),\n\t\t\t},\n\t\t\tBody: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:        constants.FirstEventID + 1,\n\t\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tHeader: &archiver.HistoryBlobHeader{\n\t\t\t\tIsLast: common.BoolPtr(true),\n\t\t\t},\n\t\t\tBody: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:        testNextEventID - 1,\n\t\t\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\t\tVersion:   testCloseFailoverVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.writeHistoryBatchesForGetTest(s.historyBatchesV1, int64(1))\n\ts.writeHistoryBatchesForGetTest(s.historyBatchesV100, testCloseFailoverVersion)\n}\n\nfunc (s *historyArchiverSuite) writeHistoryBatchesForGetTest(historyBatches []*archiver.HistoryBlob, version int64) {\n\tfor i, batch := range historyBatches {\n\t\tdata, err := encode(batch)\n\t\ts.Require().NoError(err)\n\t\tkey := constructHistoryKey(\"\", testDomainID, testWorkflowID, testRunID, version, i)\n\t\t_, err = s.s3cli.PutObjectWithContext(context.Background(), &s3.PutObjectInput{\n\t\t\tBucket: aws.String(testBucket),\n\t\t\tKey:    aws.String(key),\n\t\t\tBody:   bytes.NewReader(data),\n\t\t})\n\t\ts.Require().NoError(err)\n\t}\n}\n\nfunc (s *historyArchiverSuite) assertKeyExists(key string) {\n\t_, err := s.s3cli.GetObjectWithContext(context.Background(), &s3.GetObjectInput{\n\t\tBucket: aws.String(testBucket),\n\t\tKey:    aws.String(key),\n\t})\n\ts.NoError(err)\n}\n\nfunc getCanceledContext() context.Context {\n\tctx, cancel := context.WithCancel(context.Background())\n\tcancel()\n\treturn ctx\n}\n"
  },
  {
    "path": "common/archiver/s3store/mocks/S3API_generate.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mocks\n\n//go:generate mockery --srcpkg github.com/aws/aws-sdk-go/service/s3/s3iface --quiet --name S3API --output . --filename s3_api_mock.go\n"
  },
  {
    "path": "common/archiver/s3store/mocks/s3_api_mock.go",
    "content": "// Code generated by mockery v2.53.5. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\trequest \"github.com/aws/aws-sdk-go/aws/request\"\n\ts3 \"github.com/aws/aws-sdk-go/service/s3\"\n\tmock \"github.com/stretchr/testify/mock\"\n)\n\n// S3API is an autogenerated mock type for the S3API type\ntype S3API struct {\n\tmock.Mock\n}\n\n// AbortMultipartUpload provides a mock function with given fields: _a0\nfunc (_m *S3API) AbortMultipartUpload(_a0 *s3.AbortMultipartUploadInput) (*s3.AbortMultipartUploadOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for AbortMultipartUpload\")\n\t}\n\n\tvar r0 *s3.AbortMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.AbortMultipartUploadInput) (*s3.AbortMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.AbortMultipartUploadInput) *s3.AbortMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.AbortMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.AbortMultipartUploadInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// AbortMultipartUploadRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) AbortMultipartUploadRequest(_a0 *s3.AbortMultipartUploadInput) (*request.Request, *s3.AbortMultipartUploadOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for AbortMultipartUploadRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.AbortMultipartUploadOutput\n\tif rf, ok := ret.Get(0).(func(*s3.AbortMultipartUploadInput) (*request.Request, *s3.AbortMultipartUploadOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.AbortMultipartUploadInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.AbortMultipartUploadInput) *s3.AbortMultipartUploadOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.AbortMultipartUploadOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// AbortMultipartUploadWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) AbortMultipartUploadWithContext(_a0 context.Context, _a1 *s3.AbortMultipartUploadInput, _a2 ...request.Option) (*s3.AbortMultipartUploadOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for AbortMultipartUploadWithContext\")\n\t}\n\n\tvar r0 *s3.AbortMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.AbortMultipartUploadInput, ...request.Option) (*s3.AbortMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.AbortMultipartUploadInput, ...request.Option) *s3.AbortMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.AbortMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.AbortMultipartUploadInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CompleteMultipartUpload provides a mock function with given fields: _a0\nfunc (_m *S3API) CompleteMultipartUpload(_a0 *s3.CompleteMultipartUploadInput) (*s3.CompleteMultipartUploadOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CompleteMultipartUpload\")\n\t}\n\n\tvar r0 *s3.CompleteMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.CompleteMultipartUploadInput) (*s3.CompleteMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CompleteMultipartUploadInput) *s3.CompleteMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CompleteMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CompleteMultipartUploadInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CompleteMultipartUploadRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) CompleteMultipartUploadRequest(_a0 *s3.CompleteMultipartUploadInput) (*request.Request, *s3.CompleteMultipartUploadOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CompleteMultipartUploadRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.CompleteMultipartUploadOutput\n\tif rf, ok := ret.Get(0).(func(*s3.CompleteMultipartUploadInput) (*request.Request, *s3.CompleteMultipartUploadOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CompleteMultipartUploadInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CompleteMultipartUploadInput) *s3.CompleteMultipartUploadOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.CompleteMultipartUploadOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// CompleteMultipartUploadWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) CompleteMultipartUploadWithContext(_a0 context.Context, _a1 *s3.CompleteMultipartUploadInput, _a2 ...request.Option) (*s3.CompleteMultipartUploadOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CompleteMultipartUploadWithContext\")\n\t}\n\n\tvar r0 *s3.CompleteMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CompleteMultipartUploadInput, ...request.Option) (*s3.CompleteMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CompleteMultipartUploadInput, ...request.Option) *s3.CompleteMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CompleteMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.CompleteMultipartUploadInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CopyObject provides a mock function with given fields: _a0\nfunc (_m *S3API) CopyObject(_a0 *s3.CopyObjectInput) (*s3.CopyObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CopyObject\")\n\t}\n\n\tvar r0 *s3.CopyObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.CopyObjectInput) (*s3.CopyObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CopyObjectInput) *s3.CopyObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CopyObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CopyObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CopyObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) CopyObjectRequest(_a0 *s3.CopyObjectInput) (*request.Request, *s3.CopyObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CopyObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.CopyObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.CopyObjectInput) (*request.Request, *s3.CopyObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CopyObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CopyObjectInput) *s3.CopyObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.CopyObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// CopyObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) CopyObjectWithContext(_a0 context.Context, _a1 *s3.CopyObjectInput, _a2 ...request.Option) (*s3.CopyObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CopyObjectWithContext\")\n\t}\n\n\tvar r0 *s3.CopyObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CopyObjectInput, ...request.Option) (*s3.CopyObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CopyObjectInput, ...request.Option) *s3.CopyObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CopyObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.CopyObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateBucket provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateBucket(_a0 *s3.CreateBucketInput) (*s3.CreateBucketOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateBucket\")\n\t}\n\n\tvar r0 *s3.CreateBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.CreateBucketInput) (*s3.CreateBucketOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateBucketInput) *s3.CreateBucketOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateBucketInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateBucketRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateBucketRequest(_a0 *s3.CreateBucketInput) (*request.Request, *s3.CreateBucketOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateBucketRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.CreateBucketOutput\n\tif rf, ok := ret.Get(0).(func(*s3.CreateBucketInput) (*request.Request, *s3.CreateBucketOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateBucketInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateBucketInput) *s3.CreateBucketOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.CreateBucketOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// CreateBucketWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) CreateBucketWithContext(_a0 context.Context, _a1 *s3.CreateBucketInput, _a2 ...request.Option) (*s3.CreateBucketOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateBucketWithContext\")\n\t}\n\n\tvar r0 *s3.CreateBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateBucketInput, ...request.Option) (*s3.CreateBucketOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateBucketInput, ...request.Option) *s3.CreateBucketOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.CreateBucketInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateMultipartUpload provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateMultipartUpload(_a0 *s3.CreateMultipartUploadInput) (*s3.CreateMultipartUploadOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateMultipartUpload\")\n\t}\n\n\tvar r0 *s3.CreateMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.CreateMultipartUploadInput) (*s3.CreateMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateMultipartUploadInput) *s3.CreateMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateMultipartUploadInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateMultipartUploadRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateMultipartUploadRequest(_a0 *s3.CreateMultipartUploadInput) (*request.Request, *s3.CreateMultipartUploadOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateMultipartUploadRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.CreateMultipartUploadOutput\n\tif rf, ok := ret.Get(0).(func(*s3.CreateMultipartUploadInput) (*request.Request, *s3.CreateMultipartUploadOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateMultipartUploadInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateMultipartUploadInput) *s3.CreateMultipartUploadOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.CreateMultipartUploadOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// CreateMultipartUploadWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) CreateMultipartUploadWithContext(_a0 context.Context, _a1 *s3.CreateMultipartUploadInput, _a2 ...request.Option) (*s3.CreateMultipartUploadOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateMultipartUploadWithContext\")\n\t}\n\n\tvar r0 *s3.CreateMultipartUploadOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateMultipartUploadInput, ...request.Option) (*s3.CreateMultipartUploadOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateMultipartUploadInput, ...request.Option) *s3.CreateMultipartUploadOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateMultipartUploadOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.CreateMultipartUploadInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateSession provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateSession(_a0 *s3.CreateSessionInput) (*s3.CreateSessionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateSession\")\n\t}\n\n\tvar r0 *s3.CreateSessionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.CreateSessionInput) (*s3.CreateSessionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateSessionInput) *s3.CreateSessionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateSessionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateSessionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateSessionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) CreateSessionRequest(_a0 *s3.CreateSessionInput) (*request.Request, *s3.CreateSessionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateSessionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.CreateSessionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.CreateSessionInput) (*request.Request, *s3.CreateSessionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.CreateSessionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.CreateSessionInput) *s3.CreateSessionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.CreateSessionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// CreateSessionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) CreateSessionWithContext(_a0 context.Context, _a1 *s3.CreateSessionInput, _a2 ...request.Option) (*s3.CreateSessionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CreateSessionWithContext\")\n\t}\n\n\tvar r0 *s3.CreateSessionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateSessionInput, ...request.Option) (*s3.CreateSessionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.CreateSessionInput, ...request.Option) *s3.CreateSessionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.CreateSessionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.CreateSessionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucket provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucket(_a0 *s3.DeleteBucketInput) (*s3.DeleteBucketOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucket\")\n\t}\n\n\tvar r0 *s3.DeleteBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInput) (*s3.DeleteBucketOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInput) *s3.DeleteBucketOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketAnalyticsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketAnalyticsConfiguration(_a0 *s3.DeleteBucketAnalyticsConfigurationInput) (*s3.DeleteBucketAnalyticsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketAnalyticsConfiguration\")\n\t}\n\n\tvar r0 *s3.DeleteBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketAnalyticsConfigurationInput) (*s3.DeleteBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketAnalyticsConfigurationInput) *s3.DeleteBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketAnalyticsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketAnalyticsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketAnalyticsConfigurationRequest(_a0 *s3.DeleteBucketAnalyticsConfigurationInput) (*request.Request, *s3.DeleteBucketAnalyticsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketAnalyticsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketAnalyticsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketAnalyticsConfigurationInput) (*request.Request, *s3.DeleteBucketAnalyticsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketAnalyticsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketAnalyticsConfigurationInput) *s3.DeleteBucketAnalyticsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketAnalyticsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketAnalyticsConfigurationWithContext(_a0 context.Context, _a1 *s3.DeleteBucketAnalyticsConfigurationInput, _a2 ...request.Option) (*s3.DeleteBucketAnalyticsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketAnalyticsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketAnalyticsConfigurationInput, ...request.Option) (*s3.DeleteBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketAnalyticsConfigurationInput, ...request.Option) *s3.DeleteBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketAnalyticsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketCors provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketCors(_a0 *s3.DeleteBucketCorsInput) (*s3.DeleteBucketCorsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketCors\")\n\t}\n\n\tvar r0 *s3.DeleteBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketCorsInput) (*s3.DeleteBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketCorsInput) *s3.DeleteBucketCorsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketCorsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketCorsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketCorsRequest(_a0 *s3.DeleteBucketCorsInput) (*request.Request, *s3.DeleteBucketCorsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketCorsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketCorsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketCorsInput) (*request.Request, *s3.DeleteBucketCorsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketCorsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketCorsInput) *s3.DeleteBucketCorsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketCorsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketCorsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketCorsWithContext(_a0 context.Context, _a1 *s3.DeleteBucketCorsInput, _a2 ...request.Option) (*s3.DeleteBucketCorsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketCorsWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketCorsInput, ...request.Option) (*s3.DeleteBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketCorsInput, ...request.Option) *s3.DeleteBucketCorsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketCorsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketEncryption provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketEncryption(_a0 *s3.DeleteBucketEncryptionInput) (*s3.DeleteBucketEncryptionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketEncryption\")\n\t}\n\n\tvar r0 *s3.DeleteBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketEncryptionInput) (*s3.DeleteBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketEncryptionInput) *s3.DeleteBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketEncryptionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketEncryptionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketEncryptionRequest(_a0 *s3.DeleteBucketEncryptionInput) (*request.Request, *s3.DeleteBucketEncryptionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketEncryptionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketEncryptionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketEncryptionInput) (*request.Request, *s3.DeleteBucketEncryptionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketEncryptionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketEncryptionInput) *s3.DeleteBucketEncryptionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketEncryptionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketEncryptionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketEncryptionWithContext(_a0 context.Context, _a1 *s3.DeleteBucketEncryptionInput, _a2 ...request.Option) (*s3.DeleteBucketEncryptionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketEncryptionWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketEncryptionInput, ...request.Option) (*s3.DeleteBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketEncryptionInput, ...request.Option) *s3.DeleteBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketEncryptionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketIntelligentTieringConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketIntelligentTieringConfiguration(_a0 *s3.DeleteBucketIntelligentTieringConfigurationInput) (*s3.DeleteBucketIntelligentTieringConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketIntelligentTieringConfiguration\")\n\t}\n\n\tvar r0 *s3.DeleteBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) (*s3.DeleteBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) *s3.DeleteBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketIntelligentTieringConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketIntelligentTieringConfigurationRequest(_a0 *s3.DeleteBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.DeleteBucketIntelligentTieringConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketIntelligentTieringConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketIntelligentTieringConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.DeleteBucketIntelligentTieringConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketIntelligentTieringConfigurationInput) *s3.DeleteBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketIntelligentTieringConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketIntelligentTieringConfigurationWithContext(_a0 context.Context, _a1 *s3.DeleteBucketIntelligentTieringConfigurationInput, _a2 ...request.Option) (*s3.DeleteBucketIntelligentTieringConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketIntelligentTieringConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketIntelligentTieringConfigurationInput, ...request.Option) (*s3.DeleteBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketIntelligentTieringConfigurationInput, ...request.Option) *s3.DeleteBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketIntelligentTieringConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketInventoryConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketInventoryConfiguration(_a0 *s3.DeleteBucketInventoryConfigurationInput) (*s3.DeleteBucketInventoryConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketInventoryConfiguration\")\n\t}\n\n\tvar r0 *s3.DeleteBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInventoryConfigurationInput) (*s3.DeleteBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInventoryConfigurationInput) *s3.DeleteBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketInventoryConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketInventoryConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketInventoryConfigurationRequest(_a0 *s3.DeleteBucketInventoryConfigurationInput) (*request.Request, *s3.DeleteBucketInventoryConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketInventoryConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketInventoryConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInventoryConfigurationInput) (*request.Request, *s3.DeleteBucketInventoryConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInventoryConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketInventoryConfigurationInput) *s3.DeleteBucketInventoryConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketInventoryConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketInventoryConfigurationWithContext(_a0 context.Context, _a1 *s3.DeleteBucketInventoryConfigurationInput, _a2 ...request.Option) (*s3.DeleteBucketInventoryConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketInventoryConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketInventoryConfigurationInput, ...request.Option) (*s3.DeleteBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketInventoryConfigurationInput, ...request.Option) *s3.DeleteBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketInventoryConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketLifecycle provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketLifecycle(_a0 *s3.DeleteBucketLifecycleInput) (*s3.DeleteBucketLifecycleOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketLifecycle\")\n\t}\n\n\tvar r0 *s3.DeleteBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketLifecycleInput) (*s3.DeleteBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketLifecycleInput) *s3.DeleteBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketLifecycleInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketLifecycleRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketLifecycleRequest(_a0 *s3.DeleteBucketLifecycleInput) (*request.Request, *s3.DeleteBucketLifecycleOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketLifecycleRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketLifecycleOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketLifecycleInput) (*request.Request, *s3.DeleteBucketLifecycleOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketLifecycleInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketLifecycleInput) *s3.DeleteBucketLifecycleOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketLifecycleOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketLifecycleWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketLifecycleWithContext(_a0 context.Context, _a1 *s3.DeleteBucketLifecycleInput, _a2 ...request.Option) (*s3.DeleteBucketLifecycleOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketLifecycleWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketLifecycleInput, ...request.Option) (*s3.DeleteBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketLifecycleInput, ...request.Option) *s3.DeleteBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketLifecycleInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketMetricsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketMetricsConfiguration(_a0 *s3.DeleteBucketMetricsConfigurationInput) (*s3.DeleteBucketMetricsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketMetricsConfiguration\")\n\t}\n\n\tvar r0 *s3.DeleteBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketMetricsConfigurationInput) (*s3.DeleteBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketMetricsConfigurationInput) *s3.DeleteBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketMetricsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketMetricsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketMetricsConfigurationRequest(_a0 *s3.DeleteBucketMetricsConfigurationInput) (*request.Request, *s3.DeleteBucketMetricsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketMetricsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketMetricsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketMetricsConfigurationInput) (*request.Request, *s3.DeleteBucketMetricsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketMetricsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketMetricsConfigurationInput) *s3.DeleteBucketMetricsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketMetricsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketMetricsConfigurationWithContext(_a0 context.Context, _a1 *s3.DeleteBucketMetricsConfigurationInput, _a2 ...request.Option) (*s3.DeleteBucketMetricsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketMetricsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketMetricsConfigurationInput, ...request.Option) (*s3.DeleteBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketMetricsConfigurationInput, ...request.Option) *s3.DeleteBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketMetricsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketOwnershipControls provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketOwnershipControls(_a0 *s3.DeleteBucketOwnershipControlsInput) (*s3.DeleteBucketOwnershipControlsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketOwnershipControls\")\n\t}\n\n\tvar r0 *s3.DeleteBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketOwnershipControlsInput) (*s3.DeleteBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketOwnershipControlsInput) *s3.DeleteBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketOwnershipControlsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketOwnershipControlsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketOwnershipControlsRequest(_a0 *s3.DeleteBucketOwnershipControlsInput) (*request.Request, *s3.DeleteBucketOwnershipControlsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketOwnershipControlsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketOwnershipControlsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketOwnershipControlsInput) (*request.Request, *s3.DeleteBucketOwnershipControlsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketOwnershipControlsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketOwnershipControlsInput) *s3.DeleteBucketOwnershipControlsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketOwnershipControlsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketOwnershipControlsWithContext(_a0 context.Context, _a1 *s3.DeleteBucketOwnershipControlsInput, _a2 ...request.Option) (*s3.DeleteBucketOwnershipControlsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketOwnershipControlsWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketOwnershipControlsInput, ...request.Option) (*s3.DeleteBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketOwnershipControlsInput, ...request.Option) *s3.DeleteBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketOwnershipControlsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketPolicy provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketPolicy(_a0 *s3.DeleteBucketPolicyInput) (*s3.DeleteBucketPolicyOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketPolicy\")\n\t}\n\n\tvar r0 *s3.DeleteBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketPolicyInput) (*s3.DeleteBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketPolicyInput) *s3.DeleteBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketPolicyInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketPolicyRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketPolicyRequest(_a0 *s3.DeleteBucketPolicyInput) (*request.Request, *s3.DeleteBucketPolicyOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketPolicyRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketPolicyOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketPolicyInput) (*request.Request, *s3.DeleteBucketPolicyOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketPolicyInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketPolicyInput) *s3.DeleteBucketPolicyOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketPolicyOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketPolicyWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketPolicyWithContext(_a0 context.Context, _a1 *s3.DeleteBucketPolicyInput, _a2 ...request.Option) (*s3.DeleteBucketPolicyOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketPolicyWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketPolicyInput, ...request.Option) (*s3.DeleteBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketPolicyInput, ...request.Option) *s3.DeleteBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketPolicyInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketReplication provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketReplication(_a0 *s3.DeleteBucketReplicationInput) (*s3.DeleteBucketReplicationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketReplication\")\n\t}\n\n\tvar r0 *s3.DeleteBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketReplicationInput) (*s3.DeleteBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketReplicationInput) *s3.DeleteBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketReplicationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketReplicationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketReplicationRequest(_a0 *s3.DeleteBucketReplicationInput) (*request.Request, *s3.DeleteBucketReplicationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketReplicationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketReplicationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketReplicationInput) (*request.Request, *s3.DeleteBucketReplicationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketReplicationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketReplicationInput) *s3.DeleteBucketReplicationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketReplicationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketReplicationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketReplicationWithContext(_a0 context.Context, _a1 *s3.DeleteBucketReplicationInput, _a2 ...request.Option) (*s3.DeleteBucketReplicationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketReplicationWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketReplicationInput, ...request.Option) (*s3.DeleteBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketReplicationInput, ...request.Option) *s3.DeleteBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketReplicationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketRequest(_a0 *s3.DeleteBucketInput) (*request.Request, *s3.DeleteBucketOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInput) (*request.Request, *s3.DeleteBucketOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketInput) *s3.DeleteBucketOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketTagging(_a0 *s3.DeleteBucketTaggingInput) (*s3.DeleteBucketTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketTagging\")\n\t}\n\n\tvar r0 *s3.DeleteBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketTaggingInput) (*s3.DeleteBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketTaggingInput) *s3.DeleteBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketTaggingRequest(_a0 *s3.DeleteBucketTaggingInput) (*request.Request, *s3.DeleteBucketTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketTaggingInput) (*request.Request, *s3.DeleteBucketTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketTaggingInput) *s3.DeleteBucketTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketTaggingWithContext(_a0 context.Context, _a1 *s3.DeleteBucketTaggingInput, _a2 ...request.Option) (*s3.DeleteBucketTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketTaggingInput, ...request.Option) (*s3.DeleteBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketTaggingInput, ...request.Option) *s3.DeleteBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketWebsite provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketWebsite(_a0 *s3.DeleteBucketWebsiteInput) (*s3.DeleteBucketWebsiteOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketWebsite\")\n\t}\n\n\tvar r0 *s3.DeleteBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketWebsiteInput) (*s3.DeleteBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketWebsiteInput) *s3.DeleteBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketWebsiteInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketWebsiteRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteBucketWebsiteRequest(_a0 *s3.DeleteBucketWebsiteInput) (*request.Request, *s3.DeleteBucketWebsiteOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketWebsiteRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteBucketWebsiteOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketWebsiteInput) (*request.Request, *s3.DeleteBucketWebsiteOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteBucketWebsiteInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteBucketWebsiteInput) *s3.DeleteBucketWebsiteOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteBucketWebsiteOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketWebsiteWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketWebsiteWithContext(_a0 context.Context, _a1 *s3.DeleteBucketWebsiteInput, _a2 ...request.Option) (*s3.DeleteBucketWebsiteOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketWebsiteWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketWebsiteInput, ...request.Option) (*s3.DeleteBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketWebsiteInput, ...request.Option) *s3.DeleteBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketWebsiteInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteBucketWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteBucketWithContext(_a0 context.Context, _a1 *s3.DeleteBucketInput, _a2 ...request.Option) (*s3.DeleteBucketOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteBucketWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketInput, ...request.Option) (*s3.DeleteBucketOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteBucketInput, ...request.Option) *s3.DeleteBucketOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteBucketInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObject provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObject(_a0 *s3.DeleteObjectInput) (*s3.DeleteObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObject\")\n\t}\n\n\tvar r0 *s3.DeleteObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectInput) (*s3.DeleteObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectInput) *s3.DeleteObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObjectRequest(_a0 *s3.DeleteObjectInput) (*request.Request, *s3.DeleteObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectInput) (*request.Request, *s3.DeleteObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectInput) *s3.DeleteObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObjectTagging(_a0 *s3.DeleteObjectTaggingInput) (*s3.DeleteObjectTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectTagging\")\n\t}\n\n\tvar r0 *s3.DeleteObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectTaggingInput) (*s3.DeleteObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectTaggingInput) *s3.DeleteObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObjectTaggingRequest(_a0 *s3.DeleteObjectTaggingInput) (*request.Request, *s3.DeleteObjectTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteObjectTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectTaggingInput) (*request.Request, *s3.DeleteObjectTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectTaggingInput) *s3.DeleteObjectTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteObjectTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteObjectTaggingWithContext(_a0 context.Context, _a1 *s3.DeleteObjectTaggingInput, _a2 ...request.Option) (*s3.DeleteObjectTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectTaggingInput, ...request.Option) (*s3.DeleteObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectTaggingInput, ...request.Option) *s3.DeleteObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteObjectTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteObjectWithContext(_a0 context.Context, _a1 *s3.DeleteObjectInput, _a2 ...request.Option) (*s3.DeleteObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectInput, ...request.Option) (*s3.DeleteObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectInput, ...request.Option) *s3.DeleteObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjects provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObjects(_a0 *s3.DeleteObjectsInput) (*s3.DeleteObjectsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjects\")\n\t}\n\n\tvar r0 *s3.DeleteObjectsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectsInput) (*s3.DeleteObjectsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectsInput) *s3.DeleteObjectsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeleteObjectsRequest(_a0 *s3.DeleteObjectsInput) (*request.Request, *s3.DeleteObjectsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeleteObjectsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectsInput) (*request.Request, *s3.DeleteObjectsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeleteObjectsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeleteObjectsInput) *s3.DeleteObjectsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeleteObjectsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteObjectsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeleteObjectsWithContext(_a0 context.Context, _a1 *s3.DeleteObjectsInput, _a2 ...request.Option) (*s3.DeleteObjectsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteObjectsWithContext\")\n\t}\n\n\tvar r0 *s3.DeleteObjectsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectsInput, ...request.Option) (*s3.DeleteObjectsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeleteObjectsInput, ...request.Option) *s3.DeleteObjectsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeleteObjectsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeleteObjectsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeletePublicAccessBlock provides a mock function with given fields: _a0\nfunc (_m *S3API) DeletePublicAccessBlock(_a0 *s3.DeletePublicAccessBlockInput) (*s3.DeletePublicAccessBlockOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeletePublicAccessBlock\")\n\t}\n\n\tvar r0 *s3.DeletePublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.DeletePublicAccessBlockInput) (*s3.DeletePublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeletePublicAccessBlockInput) *s3.DeletePublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeletePublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeletePublicAccessBlockInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeletePublicAccessBlockRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) DeletePublicAccessBlockRequest(_a0 *s3.DeletePublicAccessBlockInput) (*request.Request, *s3.DeletePublicAccessBlockOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeletePublicAccessBlockRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.DeletePublicAccessBlockOutput\n\tif rf, ok := ret.Get(0).(func(*s3.DeletePublicAccessBlockInput) (*request.Request, *s3.DeletePublicAccessBlockOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.DeletePublicAccessBlockInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.DeletePublicAccessBlockInput) *s3.DeletePublicAccessBlockOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.DeletePublicAccessBlockOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// DeletePublicAccessBlockWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) DeletePublicAccessBlockWithContext(_a0 context.Context, _a1 *s3.DeletePublicAccessBlockInput, _a2 ...request.Option) (*s3.DeletePublicAccessBlockOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeletePublicAccessBlockWithContext\")\n\t}\n\n\tvar r0 *s3.DeletePublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeletePublicAccessBlockInput, ...request.Option) (*s3.DeletePublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.DeletePublicAccessBlockInput, ...request.Option) *s3.DeletePublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.DeletePublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.DeletePublicAccessBlockInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAccelerateConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAccelerateConfiguration(_a0 *s3.GetBucketAccelerateConfigurationInput) (*s3.GetBucketAccelerateConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAccelerateConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketAccelerateConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAccelerateConfigurationInput) (*s3.GetBucketAccelerateConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAccelerateConfigurationInput) *s3.GetBucketAccelerateConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAccelerateConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAccelerateConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAccelerateConfigurationRequest(_a0 *s3.GetBucketAccelerateConfigurationInput) (*request.Request, *s3.GetBucketAccelerateConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAccelerateConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketAccelerateConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAccelerateConfigurationInput) (*request.Request, *s3.GetBucketAccelerateConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAccelerateConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAccelerateConfigurationInput) *s3.GetBucketAccelerateConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAccelerateConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketAccelerateConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketAccelerateConfigurationInput, _a2 ...request.Option) (*s3.GetBucketAccelerateConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAccelerateConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketAccelerateConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAccelerateConfigurationInput, ...request.Option) (*s3.GetBucketAccelerateConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAccelerateConfigurationInput, ...request.Option) *s3.GetBucketAccelerateConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketAccelerateConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAcl provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAcl(_a0 *s3.GetBucketAclInput) (*s3.GetBucketAclOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAcl\")\n\t}\n\n\tvar r0 *s3.GetBucketAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAclInput) (*s3.GetBucketAclOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAclInput) *s3.GetBucketAclOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAclInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAclRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAclRequest(_a0 *s3.GetBucketAclInput) (*request.Request, *s3.GetBucketAclOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAclRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketAclOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAclInput) (*request.Request, *s3.GetBucketAclOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAclInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAclInput) *s3.GetBucketAclOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketAclOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAclWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketAclWithContext(_a0 context.Context, _a1 *s3.GetBucketAclInput, _a2 ...request.Option) (*s3.GetBucketAclOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAclWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAclInput, ...request.Option) (*s3.GetBucketAclOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAclInput, ...request.Option) *s3.GetBucketAclOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketAclInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAnalyticsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAnalyticsConfiguration(_a0 *s3.GetBucketAnalyticsConfigurationInput) (*s3.GetBucketAnalyticsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAnalyticsConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAnalyticsConfigurationInput) (*s3.GetBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAnalyticsConfigurationInput) *s3.GetBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAnalyticsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAnalyticsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketAnalyticsConfigurationRequest(_a0 *s3.GetBucketAnalyticsConfigurationInput) (*request.Request, *s3.GetBucketAnalyticsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAnalyticsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketAnalyticsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAnalyticsConfigurationInput) (*request.Request, *s3.GetBucketAnalyticsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketAnalyticsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketAnalyticsConfigurationInput) *s3.GetBucketAnalyticsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketAnalyticsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketAnalyticsConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketAnalyticsConfigurationInput, _a2 ...request.Option) (*s3.GetBucketAnalyticsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketAnalyticsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAnalyticsConfigurationInput, ...request.Option) (*s3.GetBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketAnalyticsConfigurationInput, ...request.Option) *s3.GetBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketAnalyticsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketCors provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketCors(_a0 *s3.GetBucketCorsInput) (*s3.GetBucketCorsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketCors\")\n\t}\n\n\tvar r0 *s3.GetBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketCorsInput) (*s3.GetBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketCorsInput) *s3.GetBucketCorsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketCorsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketCorsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketCorsRequest(_a0 *s3.GetBucketCorsInput) (*request.Request, *s3.GetBucketCorsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketCorsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketCorsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketCorsInput) (*request.Request, *s3.GetBucketCorsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketCorsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketCorsInput) *s3.GetBucketCorsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketCorsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketCorsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketCorsWithContext(_a0 context.Context, _a1 *s3.GetBucketCorsInput, _a2 ...request.Option) (*s3.GetBucketCorsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketCorsWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketCorsInput, ...request.Option) (*s3.GetBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketCorsInput, ...request.Option) *s3.GetBucketCorsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketCorsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketEncryption provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketEncryption(_a0 *s3.GetBucketEncryptionInput) (*s3.GetBucketEncryptionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketEncryption\")\n\t}\n\n\tvar r0 *s3.GetBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketEncryptionInput) (*s3.GetBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketEncryptionInput) *s3.GetBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketEncryptionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketEncryptionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketEncryptionRequest(_a0 *s3.GetBucketEncryptionInput) (*request.Request, *s3.GetBucketEncryptionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketEncryptionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketEncryptionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketEncryptionInput) (*request.Request, *s3.GetBucketEncryptionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketEncryptionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketEncryptionInput) *s3.GetBucketEncryptionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketEncryptionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketEncryptionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketEncryptionWithContext(_a0 context.Context, _a1 *s3.GetBucketEncryptionInput, _a2 ...request.Option) (*s3.GetBucketEncryptionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketEncryptionWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketEncryptionInput, ...request.Option) (*s3.GetBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketEncryptionInput, ...request.Option) *s3.GetBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketEncryptionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketIntelligentTieringConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketIntelligentTieringConfiguration(_a0 *s3.GetBucketIntelligentTieringConfigurationInput) (*s3.GetBucketIntelligentTieringConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketIntelligentTieringConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketIntelligentTieringConfigurationInput) (*s3.GetBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketIntelligentTieringConfigurationInput) *s3.GetBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketIntelligentTieringConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketIntelligentTieringConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketIntelligentTieringConfigurationRequest(_a0 *s3.GetBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.GetBucketIntelligentTieringConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketIntelligentTieringConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketIntelligentTieringConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.GetBucketIntelligentTieringConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketIntelligentTieringConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketIntelligentTieringConfigurationInput) *s3.GetBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketIntelligentTieringConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketIntelligentTieringConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketIntelligentTieringConfigurationInput, _a2 ...request.Option) (*s3.GetBucketIntelligentTieringConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketIntelligentTieringConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketIntelligentTieringConfigurationInput, ...request.Option) (*s3.GetBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketIntelligentTieringConfigurationInput, ...request.Option) *s3.GetBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketIntelligentTieringConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketInventoryConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketInventoryConfiguration(_a0 *s3.GetBucketInventoryConfigurationInput) (*s3.GetBucketInventoryConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketInventoryConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketInventoryConfigurationInput) (*s3.GetBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketInventoryConfigurationInput) *s3.GetBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketInventoryConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketInventoryConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketInventoryConfigurationRequest(_a0 *s3.GetBucketInventoryConfigurationInput) (*request.Request, *s3.GetBucketInventoryConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketInventoryConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketInventoryConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketInventoryConfigurationInput) (*request.Request, *s3.GetBucketInventoryConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketInventoryConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketInventoryConfigurationInput) *s3.GetBucketInventoryConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketInventoryConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketInventoryConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketInventoryConfigurationInput, _a2 ...request.Option) (*s3.GetBucketInventoryConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketInventoryConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketInventoryConfigurationInput, ...request.Option) (*s3.GetBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketInventoryConfigurationInput, ...request.Option) *s3.GetBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketInventoryConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycle provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLifecycle(_a0 *s3.GetBucketLifecycleInput) (*s3.GetBucketLifecycleOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycle\")\n\t}\n\n\tvar r0 *s3.GetBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleInput) (*s3.GetBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleInput) *s3.GetBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLifecycleInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycleConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLifecycleConfiguration(_a0 *s3.GetBucketLifecycleConfigurationInput) (*s3.GetBucketLifecycleConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycleConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketLifecycleConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleConfigurationInput) (*s3.GetBucketLifecycleConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleConfigurationInput) *s3.GetBucketLifecycleConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLifecycleConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycleConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLifecycleConfigurationRequest(_a0 *s3.GetBucketLifecycleConfigurationInput) (*request.Request, *s3.GetBucketLifecycleConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycleConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketLifecycleConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleConfigurationInput) (*request.Request, *s3.GetBucketLifecycleConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLifecycleConfigurationInput) *s3.GetBucketLifecycleConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycleConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketLifecycleConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketLifecycleConfigurationInput, _a2 ...request.Option) (*s3.GetBucketLifecycleConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycleConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketLifecycleConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLifecycleConfigurationInput, ...request.Option) (*s3.GetBucketLifecycleConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLifecycleConfigurationInput, ...request.Option) *s3.GetBucketLifecycleConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketLifecycleConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycleRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLifecycleRequest(_a0 *s3.GetBucketLifecycleInput) (*request.Request, *s3.GetBucketLifecycleOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycleRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketLifecycleOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleInput) (*request.Request, *s3.GetBucketLifecycleOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLifecycleInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLifecycleInput) *s3.GetBucketLifecycleOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketLifecycleOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLifecycleWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketLifecycleWithContext(_a0 context.Context, _a1 *s3.GetBucketLifecycleInput, _a2 ...request.Option) (*s3.GetBucketLifecycleOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLifecycleWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLifecycleInput, ...request.Option) (*s3.GetBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLifecycleInput, ...request.Option) *s3.GetBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketLifecycleInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLocation provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLocation(_a0 *s3.GetBucketLocationInput) (*s3.GetBucketLocationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLocation\")\n\t}\n\n\tvar r0 *s3.GetBucketLocationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLocationInput) (*s3.GetBucketLocationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLocationInput) *s3.GetBucketLocationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLocationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLocationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLocationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLocationRequest(_a0 *s3.GetBucketLocationInput) (*request.Request, *s3.GetBucketLocationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLocationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketLocationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLocationInput) (*request.Request, *s3.GetBucketLocationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLocationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLocationInput) *s3.GetBucketLocationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketLocationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLocationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketLocationWithContext(_a0 context.Context, _a1 *s3.GetBucketLocationInput, _a2 ...request.Option) (*s3.GetBucketLocationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLocationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketLocationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLocationInput, ...request.Option) (*s3.GetBucketLocationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLocationInput, ...request.Option) *s3.GetBucketLocationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLocationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketLocationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLogging provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLogging(_a0 *s3.GetBucketLoggingInput) (*s3.GetBucketLoggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLogging\")\n\t}\n\n\tvar r0 *s3.GetBucketLoggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLoggingInput) (*s3.GetBucketLoggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLoggingInput) *s3.GetBucketLoggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLoggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLoggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLoggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketLoggingRequest(_a0 *s3.GetBucketLoggingInput) (*request.Request, *s3.GetBucketLoggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLoggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketLoggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLoggingInput) (*request.Request, *s3.GetBucketLoggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketLoggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketLoggingInput) *s3.GetBucketLoggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketLoggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketLoggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketLoggingWithContext(_a0 context.Context, _a1 *s3.GetBucketLoggingInput, _a2 ...request.Option) (*s3.GetBucketLoggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketLoggingWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketLoggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLoggingInput, ...request.Option) (*s3.GetBucketLoggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketLoggingInput, ...request.Option) *s3.GetBucketLoggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketLoggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketLoggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketMetricsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketMetricsConfiguration(_a0 *s3.GetBucketMetricsConfigurationInput) (*s3.GetBucketMetricsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketMetricsConfiguration\")\n\t}\n\n\tvar r0 *s3.GetBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketMetricsConfigurationInput) (*s3.GetBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketMetricsConfigurationInput) *s3.GetBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketMetricsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketMetricsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketMetricsConfigurationRequest(_a0 *s3.GetBucketMetricsConfigurationInput) (*request.Request, *s3.GetBucketMetricsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketMetricsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketMetricsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketMetricsConfigurationInput) (*request.Request, *s3.GetBucketMetricsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketMetricsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketMetricsConfigurationInput) *s3.GetBucketMetricsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketMetricsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketMetricsConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketMetricsConfigurationInput, _a2 ...request.Option) (*s3.GetBucketMetricsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketMetricsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketMetricsConfigurationInput, ...request.Option) (*s3.GetBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketMetricsConfigurationInput, ...request.Option) *s3.GetBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketMetricsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotification provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketNotification(_a0 *s3.GetBucketNotificationConfigurationRequest) (*s3.NotificationConfigurationDeprecated, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotification\")\n\t}\n\n\tvar r0 *s3.NotificationConfigurationDeprecated\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) (*s3.NotificationConfigurationDeprecated, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) *s3.NotificationConfigurationDeprecated); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.NotificationConfigurationDeprecated)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketNotificationConfigurationRequest) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotificationConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketNotificationConfiguration(_a0 *s3.GetBucketNotificationConfigurationRequest) (*s3.NotificationConfiguration, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotificationConfiguration\")\n\t}\n\n\tvar r0 *s3.NotificationConfiguration\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) (*s3.NotificationConfiguration, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) *s3.NotificationConfiguration); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.NotificationConfiguration)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketNotificationConfigurationRequest) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotificationConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketNotificationConfigurationRequest(_a0 *s3.GetBucketNotificationConfigurationRequest) (*request.Request, *s3.NotificationConfiguration) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotificationConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.NotificationConfiguration\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) (*request.Request, *s3.NotificationConfiguration)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketNotificationConfigurationRequest) *s3.NotificationConfiguration); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.NotificationConfiguration)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotificationConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketNotificationConfigurationWithContext(_a0 context.Context, _a1 *s3.GetBucketNotificationConfigurationRequest, _a2 ...request.Option) (*s3.NotificationConfiguration, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotificationConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.NotificationConfiguration\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) (*s3.NotificationConfiguration, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) *s3.NotificationConfiguration); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.NotificationConfiguration)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotificationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketNotificationRequest(_a0 *s3.GetBucketNotificationConfigurationRequest) (*request.Request, *s3.NotificationConfigurationDeprecated) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotificationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.NotificationConfigurationDeprecated\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) (*request.Request, *s3.NotificationConfigurationDeprecated)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketNotificationConfigurationRequest) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketNotificationConfigurationRequest) *s3.NotificationConfigurationDeprecated); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.NotificationConfigurationDeprecated)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketNotificationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketNotificationWithContext(_a0 context.Context, _a1 *s3.GetBucketNotificationConfigurationRequest, _a2 ...request.Option) (*s3.NotificationConfigurationDeprecated, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketNotificationWithContext\")\n\t}\n\n\tvar r0 *s3.NotificationConfigurationDeprecated\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) (*s3.NotificationConfigurationDeprecated, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) *s3.NotificationConfigurationDeprecated); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.NotificationConfigurationDeprecated)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketNotificationConfigurationRequest, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketOwnershipControls provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketOwnershipControls(_a0 *s3.GetBucketOwnershipControlsInput) (*s3.GetBucketOwnershipControlsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketOwnershipControls\")\n\t}\n\n\tvar r0 *s3.GetBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketOwnershipControlsInput) (*s3.GetBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketOwnershipControlsInput) *s3.GetBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketOwnershipControlsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketOwnershipControlsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketOwnershipControlsRequest(_a0 *s3.GetBucketOwnershipControlsInput) (*request.Request, *s3.GetBucketOwnershipControlsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketOwnershipControlsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketOwnershipControlsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketOwnershipControlsInput) (*request.Request, *s3.GetBucketOwnershipControlsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketOwnershipControlsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketOwnershipControlsInput) *s3.GetBucketOwnershipControlsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketOwnershipControlsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketOwnershipControlsWithContext(_a0 context.Context, _a1 *s3.GetBucketOwnershipControlsInput, _a2 ...request.Option) (*s3.GetBucketOwnershipControlsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketOwnershipControlsWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketOwnershipControlsInput, ...request.Option) (*s3.GetBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketOwnershipControlsInput, ...request.Option) *s3.GetBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketOwnershipControlsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicy provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketPolicy(_a0 *s3.GetBucketPolicyInput) (*s3.GetBucketPolicyOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicy\")\n\t}\n\n\tvar r0 *s3.GetBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyInput) (*s3.GetBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyInput) *s3.GetBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketPolicyInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicyRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketPolicyRequest(_a0 *s3.GetBucketPolicyInput) (*request.Request, *s3.GetBucketPolicyOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicyRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketPolicyOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyInput) (*request.Request, *s3.GetBucketPolicyOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketPolicyInput) *s3.GetBucketPolicyOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketPolicyOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicyStatus provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketPolicyStatus(_a0 *s3.GetBucketPolicyStatusInput) (*s3.GetBucketPolicyStatusOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicyStatus\")\n\t}\n\n\tvar r0 *s3.GetBucketPolicyStatusOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyStatusInput) (*s3.GetBucketPolicyStatusOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyStatusInput) *s3.GetBucketPolicyStatusOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketPolicyStatusOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketPolicyStatusInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicyStatusRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketPolicyStatusRequest(_a0 *s3.GetBucketPolicyStatusInput) (*request.Request, *s3.GetBucketPolicyStatusOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicyStatusRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketPolicyStatusOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyStatusInput) (*request.Request, *s3.GetBucketPolicyStatusOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketPolicyStatusInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketPolicyStatusInput) *s3.GetBucketPolicyStatusOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketPolicyStatusOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicyStatusWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketPolicyStatusWithContext(_a0 context.Context, _a1 *s3.GetBucketPolicyStatusInput, _a2 ...request.Option) (*s3.GetBucketPolicyStatusOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicyStatusWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketPolicyStatusOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketPolicyStatusInput, ...request.Option) (*s3.GetBucketPolicyStatusOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketPolicyStatusInput, ...request.Option) *s3.GetBucketPolicyStatusOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketPolicyStatusOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketPolicyStatusInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketPolicyWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketPolicyWithContext(_a0 context.Context, _a1 *s3.GetBucketPolicyInput, _a2 ...request.Option) (*s3.GetBucketPolicyOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketPolicyWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketPolicyInput, ...request.Option) (*s3.GetBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketPolicyInput, ...request.Option) *s3.GetBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketPolicyInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketReplication provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketReplication(_a0 *s3.GetBucketReplicationInput) (*s3.GetBucketReplicationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketReplication\")\n\t}\n\n\tvar r0 *s3.GetBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketReplicationInput) (*s3.GetBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketReplicationInput) *s3.GetBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketReplicationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketReplicationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketReplicationRequest(_a0 *s3.GetBucketReplicationInput) (*request.Request, *s3.GetBucketReplicationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketReplicationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketReplicationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketReplicationInput) (*request.Request, *s3.GetBucketReplicationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketReplicationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketReplicationInput) *s3.GetBucketReplicationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketReplicationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketReplicationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketReplicationWithContext(_a0 context.Context, _a1 *s3.GetBucketReplicationInput, _a2 ...request.Option) (*s3.GetBucketReplicationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketReplicationWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketReplicationInput, ...request.Option) (*s3.GetBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketReplicationInput, ...request.Option) *s3.GetBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketReplicationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketRequestPayment provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketRequestPayment(_a0 *s3.GetBucketRequestPaymentInput) (*s3.GetBucketRequestPaymentOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketRequestPayment\")\n\t}\n\n\tvar r0 *s3.GetBucketRequestPaymentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketRequestPaymentInput) (*s3.GetBucketRequestPaymentOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketRequestPaymentInput) *s3.GetBucketRequestPaymentOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketRequestPaymentInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketRequestPaymentRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketRequestPaymentRequest(_a0 *s3.GetBucketRequestPaymentInput) (*request.Request, *s3.GetBucketRequestPaymentOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketRequestPaymentRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketRequestPaymentOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketRequestPaymentInput) (*request.Request, *s3.GetBucketRequestPaymentOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketRequestPaymentInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketRequestPaymentInput) *s3.GetBucketRequestPaymentOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketRequestPaymentWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketRequestPaymentWithContext(_a0 context.Context, _a1 *s3.GetBucketRequestPaymentInput, _a2 ...request.Option) (*s3.GetBucketRequestPaymentOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketRequestPaymentWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketRequestPaymentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketRequestPaymentInput, ...request.Option) (*s3.GetBucketRequestPaymentOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketRequestPaymentInput, ...request.Option) *s3.GetBucketRequestPaymentOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketRequestPaymentInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketTagging(_a0 *s3.GetBucketTaggingInput) (*s3.GetBucketTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketTagging\")\n\t}\n\n\tvar r0 *s3.GetBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketTaggingInput) (*s3.GetBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketTaggingInput) *s3.GetBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketTaggingRequest(_a0 *s3.GetBucketTaggingInput) (*request.Request, *s3.GetBucketTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketTaggingInput) (*request.Request, *s3.GetBucketTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketTaggingInput) *s3.GetBucketTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketTaggingWithContext(_a0 context.Context, _a1 *s3.GetBucketTaggingInput, _a2 ...request.Option) (*s3.GetBucketTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketTaggingInput, ...request.Option) (*s3.GetBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketTaggingInput, ...request.Option) *s3.GetBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketVersioning provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketVersioning(_a0 *s3.GetBucketVersioningInput) (*s3.GetBucketVersioningOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketVersioning\")\n\t}\n\n\tvar r0 *s3.GetBucketVersioningOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketVersioningInput) (*s3.GetBucketVersioningOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketVersioningInput) *s3.GetBucketVersioningOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketVersioningOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketVersioningInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketVersioningRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketVersioningRequest(_a0 *s3.GetBucketVersioningInput) (*request.Request, *s3.GetBucketVersioningOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketVersioningRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketVersioningOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketVersioningInput) (*request.Request, *s3.GetBucketVersioningOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketVersioningInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketVersioningInput) *s3.GetBucketVersioningOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketVersioningOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketVersioningWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketVersioningWithContext(_a0 context.Context, _a1 *s3.GetBucketVersioningInput, _a2 ...request.Option) (*s3.GetBucketVersioningOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketVersioningWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketVersioningOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketVersioningInput, ...request.Option) (*s3.GetBucketVersioningOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketVersioningInput, ...request.Option) *s3.GetBucketVersioningOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketVersioningOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketVersioningInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketWebsite provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketWebsite(_a0 *s3.GetBucketWebsiteInput) (*s3.GetBucketWebsiteOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketWebsite\")\n\t}\n\n\tvar r0 *s3.GetBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketWebsiteInput) (*s3.GetBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketWebsiteInput) *s3.GetBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketWebsiteInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketWebsiteRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetBucketWebsiteRequest(_a0 *s3.GetBucketWebsiteInput) (*request.Request, *s3.GetBucketWebsiteOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketWebsiteRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetBucketWebsiteOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketWebsiteInput) (*request.Request, *s3.GetBucketWebsiteOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetBucketWebsiteInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetBucketWebsiteInput) *s3.GetBucketWebsiteOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetBucketWebsiteOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetBucketWebsiteWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetBucketWebsiteWithContext(_a0 context.Context, _a1 *s3.GetBucketWebsiteInput, _a2 ...request.Option) (*s3.GetBucketWebsiteOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetBucketWebsiteWithContext\")\n\t}\n\n\tvar r0 *s3.GetBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketWebsiteInput, ...request.Option) (*s3.GetBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetBucketWebsiteInput, ...request.Option) *s3.GetBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetBucketWebsiteInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObject provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObject(_a0 *s3.GetObjectInput) (*s3.GetObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObject\")\n\t}\n\n\tvar r0 *s3.GetObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectInput) (*s3.GetObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectInput) *s3.GetObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAcl provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectAcl(_a0 *s3.GetObjectAclInput) (*s3.GetObjectAclOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAcl\")\n\t}\n\n\tvar r0 *s3.GetObjectAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAclInput) (*s3.GetObjectAclOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAclInput) *s3.GetObjectAclOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectAclInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAclRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectAclRequest(_a0 *s3.GetObjectAclInput) (*request.Request, *s3.GetObjectAclOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAclRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectAclOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAclInput) (*request.Request, *s3.GetObjectAclOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAclInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectAclInput) *s3.GetObjectAclOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectAclOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAclWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectAclWithContext(_a0 context.Context, _a1 *s3.GetObjectAclInput, _a2 ...request.Option) (*s3.GetObjectAclOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAclWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectAclInput, ...request.Option) (*s3.GetObjectAclOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectAclInput, ...request.Option) *s3.GetObjectAclOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectAclInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAttributes provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectAttributes(_a0 *s3.GetObjectAttributesInput) (*s3.GetObjectAttributesOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAttributes\")\n\t}\n\n\tvar r0 *s3.GetObjectAttributesOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAttributesInput) (*s3.GetObjectAttributesOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAttributesInput) *s3.GetObjectAttributesOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectAttributesOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectAttributesInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAttributesRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectAttributesRequest(_a0 *s3.GetObjectAttributesInput) (*request.Request, *s3.GetObjectAttributesOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAttributesRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectAttributesOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAttributesInput) (*request.Request, *s3.GetObjectAttributesOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectAttributesInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectAttributesInput) *s3.GetObjectAttributesOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectAttributesOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectAttributesWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectAttributesWithContext(_a0 context.Context, _a1 *s3.GetObjectAttributesInput, _a2 ...request.Option) (*s3.GetObjectAttributesOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectAttributesWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectAttributesOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectAttributesInput, ...request.Option) (*s3.GetObjectAttributesOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectAttributesInput, ...request.Option) *s3.GetObjectAttributesOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectAttributesOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectAttributesInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLegalHold provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectLegalHold(_a0 *s3.GetObjectLegalHoldInput) (*s3.GetObjectLegalHoldOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLegalHold\")\n\t}\n\n\tvar r0 *s3.GetObjectLegalHoldOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLegalHoldInput) (*s3.GetObjectLegalHoldOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLegalHoldInput) *s3.GetObjectLegalHoldOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectLegalHoldInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLegalHoldRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectLegalHoldRequest(_a0 *s3.GetObjectLegalHoldInput) (*request.Request, *s3.GetObjectLegalHoldOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLegalHoldRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectLegalHoldOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLegalHoldInput) (*request.Request, *s3.GetObjectLegalHoldOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLegalHoldInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectLegalHoldInput) *s3.GetObjectLegalHoldOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLegalHoldWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectLegalHoldWithContext(_a0 context.Context, _a1 *s3.GetObjectLegalHoldInput, _a2 ...request.Option) (*s3.GetObjectLegalHoldOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLegalHoldWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectLegalHoldOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectLegalHoldInput, ...request.Option) (*s3.GetObjectLegalHoldOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectLegalHoldInput, ...request.Option) *s3.GetObjectLegalHoldOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectLegalHoldInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLockConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectLockConfiguration(_a0 *s3.GetObjectLockConfigurationInput) (*s3.GetObjectLockConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLockConfiguration\")\n\t}\n\n\tvar r0 *s3.GetObjectLockConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLockConfigurationInput) (*s3.GetObjectLockConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLockConfigurationInput) *s3.GetObjectLockConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectLockConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLockConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectLockConfigurationRequest(_a0 *s3.GetObjectLockConfigurationInput) (*request.Request, *s3.GetObjectLockConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLockConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectLockConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLockConfigurationInput) (*request.Request, *s3.GetObjectLockConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectLockConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectLockConfigurationInput) *s3.GetObjectLockConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectLockConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectLockConfigurationWithContext(_a0 context.Context, _a1 *s3.GetObjectLockConfigurationInput, _a2 ...request.Option) (*s3.GetObjectLockConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectLockConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectLockConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectLockConfigurationInput, ...request.Option) (*s3.GetObjectLockConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectLockConfigurationInput, ...request.Option) *s3.GetObjectLockConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectLockConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectRequest(_a0 *s3.GetObjectInput) (*request.Request, *s3.GetObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectInput) (*request.Request, *s3.GetObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectInput) *s3.GetObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectRetention provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectRetention(_a0 *s3.GetObjectRetentionInput) (*s3.GetObjectRetentionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectRetention\")\n\t}\n\n\tvar r0 *s3.GetObjectRetentionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectRetentionInput) (*s3.GetObjectRetentionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectRetentionInput) *s3.GetObjectRetentionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectRetentionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectRetentionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectRetentionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectRetentionRequest(_a0 *s3.GetObjectRetentionInput) (*request.Request, *s3.GetObjectRetentionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectRetentionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectRetentionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectRetentionInput) (*request.Request, *s3.GetObjectRetentionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectRetentionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectRetentionInput) *s3.GetObjectRetentionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectRetentionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectRetentionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectRetentionWithContext(_a0 context.Context, _a1 *s3.GetObjectRetentionInput, _a2 ...request.Option) (*s3.GetObjectRetentionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectRetentionWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectRetentionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectRetentionInput, ...request.Option) (*s3.GetObjectRetentionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectRetentionInput, ...request.Option) *s3.GetObjectRetentionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectRetentionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectRetentionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectTagging(_a0 *s3.GetObjectTaggingInput) (*s3.GetObjectTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTagging\")\n\t}\n\n\tvar r0 *s3.GetObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTaggingInput) (*s3.GetObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTaggingInput) *s3.GetObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectTaggingRequest(_a0 *s3.GetObjectTaggingInput) (*request.Request, *s3.GetObjectTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTaggingInput) (*request.Request, *s3.GetObjectTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectTaggingInput) *s3.GetObjectTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectTaggingWithContext(_a0 context.Context, _a1 *s3.GetObjectTaggingInput, _a2 ...request.Option) (*s3.GetObjectTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectTaggingInput, ...request.Option) (*s3.GetObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectTaggingInput, ...request.Option) *s3.GetObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTorrent provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectTorrent(_a0 *s3.GetObjectTorrentInput) (*s3.GetObjectTorrentOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTorrent\")\n\t}\n\n\tvar r0 *s3.GetObjectTorrentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTorrentInput) (*s3.GetObjectTorrentOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTorrentInput) *s3.GetObjectTorrentOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectTorrentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectTorrentInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTorrentRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetObjectTorrentRequest(_a0 *s3.GetObjectTorrentInput) (*request.Request, *s3.GetObjectTorrentOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTorrentRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetObjectTorrentOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTorrentInput) (*request.Request, *s3.GetObjectTorrentOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetObjectTorrentInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetObjectTorrentInput) *s3.GetObjectTorrentOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetObjectTorrentOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectTorrentWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectTorrentWithContext(_a0 context.Context, _a1 *s3.GetObjectTorrentInput, _a2 ...request.Option) (*s3.GetObjectTorrentOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectTorrentWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectTorrentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectTorrentInput, ...request.Option) (*s3.GetObjectTorrentOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectTorrentInput, ...request.Option) *s3.GetObjectTorrentOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectTorrentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectTorrentInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetObjectWithContext(_a0 context.Context, _a1 *s3.GetObjectInput, _a2 ...request.Option) (*s3.GetObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetObjectWithContext\")\n\t}\n\n\tvar r0 *s3.GetObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectInput, ...request.Option) (*s3.GetObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetObjectInput, ...request.Option) *s3.GetObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetPublicAccessBlock provides a mock function with given fields: _a0\nfunc (_m *S3API) GetPublicAccessBlock(_a0 *s3.GetPublicAccessBlockInput) (*s3.GetPublicAccessBlockOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetPublicAccessBlock\")\n\t}\n\n\tvar r0 *s3.GetPublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.GetPublicAccessBlockInput) (*s3.GetPublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetPublicAccessBlockInput) *s3.GetPublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetPublicAccessBlockInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetPublicAccessBlockRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) GetPublicAccessBlockRequest(_a0 *s3.GetPublicAccessBlockInput) (*request.Request, *s3.GetPublicAccessBlockOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetPublicAccessBlockRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.GetPublicAccessBlockOutput\n\tif rf, ok := ret.Get(0).(func(*s3.GetPublicAccessBlockInput) (*request.Request, *s3.GetPublicAccessBlockOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.GetPublicAccessBlockInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.GetPublicAccessBlockInput) *s3.GetPublicAccessBlockOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.GetPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// GetPublicAccessBlockWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) GetPublicAccessBlockWithContext(_a0 context.Context, _a1 *s3.GetPublicAccessBlockInput, _a2 ...request.Option) (*s3.GetPublicAccessBlockOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetPublicAccessBlockWithContext\")\n\t}\n\n\tvar r0 *s3.GetPublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetPublicAccessBlockInput, ...request.Option) (*s3.GetPublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.GetPublicAccessBlockInput, ...request.Option) *s3.GetPublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.GetPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.GetPublicAccessBlockInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// HeadBucket provides a mock function with given fields: _a0\nfunc (_m *S3API) HeadBucket(_a0 *s3.HeadBucketInput) (*s3.HeadBucketOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadBucket\")\n\t}\n\n\tvar r0 *s3.HeadBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) (*s3.HeadBucketOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) *s3.HeadBucketOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.HeadBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.HeadBucketInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// HeadBucketRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) HeadBucketRequest(_a0 *s3.HeadBucketInput) (*request.Request, *s3.HeadBucketOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadBucketRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.HeadBucketOutput\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) (*request.Request, *s3.HeadBucketOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.HeadBucketInput) *s3.HeadBucketOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.HeadBucketOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// HeadBucketWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) HeadBucketWithContext(_a0 context.Context, _a1 *s3.HeadBucketInput, _a2 ...request.Option) (*s3.HeadBucketOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadBucketWithContext\")\n\t}\n\n\tvar r0 *s3.HeadBucketOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadBucketInput, ...request.Option) (*s3.HeadBucketOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadBucketInput, ...request.Option) *s3.HeadBucketOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.HeadBucketOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.HeadBucketInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// HeadObject provides a mock function with given fields: _a0\nfunc (_m *S3API) HeadObject(_a0 *s3.HeadObjectInput) (*s3.HeadObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadObject\")\n\t}\n\n\tvar r0 *s3.HeadObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) (*s3.HeadObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) *s3.HeadObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.HeadObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.HeadObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// HeadObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) HeadObjectRequest(_a0 *s3.HeadObjectInput) (*request.Request, *s3.HeadObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.HeadObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) (*request.Request, *s3.HeadObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.HeadObjectInput) *s3.HeadObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.HeadObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// HeadObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) HeadObjectWithContext(_a0 context.Context, _a1 *s3.HeadObjectInput, _a2 ...request.Option) (*s3.HeadObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for HeadObjectWithContext\")\n\t}\n\n\tvar r0 *s3.HeadObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadObjectInput, ...request.Option) (*s3.HeadObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadObjectInput, ...request.Option) *s3.HeadObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.HeadObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.HeadObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketAnalyticsConfigurations provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketAnalyticsConfigurations(_a0 *s3.ListBucketAnalyticsConfigurationsInput) (*s3.ListBucketAnalyticsConfigurationsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketAnalyticsConfigurations\")\n\t}\n\n\tvar r0 *s3.ListBucketAnalyticsConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketAnalyticsConfigurationsInput) (*s3.ListBucketAnalyticsConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketAnalyticsConfigurationsInput) *s3.ListBucketAnalyticsConfigurationsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketAnalyticsConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketAnalyticsConfigurationsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketAnalyticsConfigurationsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketAnalyticsConfigurationsRequest(_a0 *s3.ListBucketAnalyticsConfigurationsInput) (*request.Request, *s3.ListBucketAnalyticsConfigurationsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketAnalyticsConfigurationsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListBucketAnalyticsConfigurationsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketAnalyticsConfigurationsInput) (*request.Request, *s3.ListBucketAnalyticsConfigurationsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketAnalyticsConfigurationsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketAnalyticsConfigurationsInput) *s3.ListBucketAnalyticsConfigurationsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListBucketAnalyticsConfigurationsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketAnalyticsConfigurationsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListBucketAnalyticsConfigurationsWithContext(_a0 context.Context, _a1 *s3.ListBucketAnalyticsConfigurationsInput, _a2 ...request.Option) (*s3.ListBucketAnalyticsConfigurationsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketAnalyticsConfigurationsWithContext\")\n\t}\n\n\tvar r0 *s3.ListBucketAnalyticsConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketAnalyticsConfigurationsInput, ...request.Option) (*s3.ListBucketAnalyticsConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketAnalyticsConfigurationsInput, ...request.Option) *s3.ListBucketAnalyticsConfigurationsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketAnalyticsConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListBucketAnalyticsConfigurationsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketIntelligentTieringConfigurations provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketIntelligentTieringConfigurations(_a0 *s3.ListBucketIntelligentTieringConfigurationsInput) (*s3.ListBucketIntelligentTieringConfigurationsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketIntelligentTieringConfigurations\")\n\t}\n\n\tvar r0 *s3.ListBucketIntelligentTieringConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) (*s3.ListBucketIntelligentTieringConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) *s3.ListBucketIntelligentTieringConfigurationsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketIntelligentTieringConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketIntelligentTieringConfigurationsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketIntelligentTieringConfigurationsRequest(_a0 *s3.ListBucketIntelligentTieringConfigurationsInput) (*request.Request, *s3.ListBucketIntelligentTieringConfigurationsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketIntelligentTieringConfigurationsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListBucketIntelligentTieringConfigurationsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) (*request.Request, *s3.ListBucketIntelligentTieringConfigurationsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketIntelligentTieringConfigurationsInput) *s3.ListBucketIntelligentTieringConfigurationsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListBucketIntelligentTieringConfigurationsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketIntelligentTieringConfigurationsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListBucketIntelligentTieringConfigurationsWithContext(_a0 context.Context, _a1 *s3.ListBucketIntelligentTieringConfigurationsInput, _a2 ...request.Option) (*s3.ListBucketIntelligentTieringConfigurationsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketIntelligentTieringConfigurationsWithContext\")\n\t}\n\n\tvar r0 *s3.ListBucketIntelligentTieringConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketIntelligentTieringConfigurationsInput, ...request.Option) (*s3.ListBucketIntelligentTieringConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketIntelligentTieringConfigurationsInput, ...request.Option) *s3.ListBucketIntelligentTieringConfigurationsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketIntelligentTieringConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListBucketIntelligentTieringConfigurationsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketInventoryConfigurations provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketInventoryConfigurations(_a0 *s3.ListBucketInventoryConfigurationsInput) (*s3.ListBucketInventoryConfigurationsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketInventoryConfigurations\")\n\t}\n\n\tvar r0 *s3.ListBucketInventoryConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketInventoryConfigurationsInput) (*s3.ListBucketInventoryConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketInventoryConfigurationsInput) *s3.ListBucketInventoryConfigurationsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketInventoryConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketInventoryConfigurationsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketInventoryConfigurationsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketInventoryConfigurationsRequest(_a0 *s3.ListBucketInventoryConfigurationsInput) (*request.Request, *s3.ListBucketInventoryConfigurationsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketInventoryConfigurationsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListBucketInventoryConfigurationsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketInventoryConfigurationsInput) (*request.Request, *s3.ListBucketInventoryConfigurationsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketInventoryConfigurationsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketInventoryConfigurationsInput) *s3.ListBucketInventoryConfigurationsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListBucketInventoryConfigurationsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketInventoryConfigurationsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListBucketInventoryConfigurationsWithContext(_a0 context.Context, _a1 *s3.ListBucketInventoryConfigurationsInput, _a2 ...request.Option) (*s3.ListBucketInventoryConfigurationsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketInventoryConfigurationsWithContext\")\n\t}\n\n\tvar r0 *s3.ListBucketInventoryConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketInventoryConfigurationsInput, ...request.Option) (*s3.ListBucketInventoryConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketInventoryConfigurationsInput, ...request.Option) *s3.ListBucketInventoryConfigurationsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketInventoryConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListBucketInventoryConfigurationsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketMetricsConfigurations provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketMetricsConfigurations(_a0 *s3.ListBucketMetricsConfigurationsInput) (*s3.ListBucketMetricsConfigurationsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketMetricsConfigurations\")\n\t}\n\n\tvar r0 *s3.ListBucketMetricsConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketMetricsConfigurationsInput) (*s3.ListBucketMetricsConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketMetricsConfigurationsInput) *s3.ListBucketMetricsConfigurationsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketMetricsConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketMetricsConfigurationsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketMetricsConfigurationsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketMetricsConfigurationsRequest(_a0 *s3.ListBucketMetricsConfigurationsInput) (*request.Request, *s3.ListBucketMetricsConfigurationsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketMetricsConfigurationsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListBucketMetricsConfigurationsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketMetricsConfigurationsInput) (*request.Request, *s3.ListBucketMetricsConfigurationsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketMetricsConfigurationsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketMetricsConfigurationsInput) *s3.ListBucketMetricsConfigurationsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListBucketMetricsConfigurationsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketMetricsConfigurationsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListBucketMetricsConfigurationsWithContext(_a0 context.Context, _a1 *s3.ListBucketMetricsConfigurationsInput, _a2 ...request.Option) (*s3.ListBucketMetricsConfigurationsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketMetricsConfigurationsWithContext\")\n\t}\n\n\tvar r0 *s3.ListBucketMetricsConfigurationsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketMetricsConfigurationsInput, ...request.Option) (*s3.ListBucketMetricsConfigurationsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketMetricsConfigurationsInput, ...request.Option) *s3.ListBucketMetricsConfigurationsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketMetricsConfigurationsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListBucketMetricsConfigurationsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBuckets provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBuckets(_a0 *s3.ListBucketsInput) (*s3.ListBucketsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBuckets\")\n\t}\n\n\tvar r0 *s3.ListBucketsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketsInput) (*s3.ListBucketsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketsInput) *s3.ListBucketsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListBucketsRequest(_a0 *s3.ListBucketsInput) (*request.Request, *s3.ListBucketsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListBucketsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketsInput) (*request.Request, *s3.ListBucketsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListBucketsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListBucketsInput) *s3.ListBucketsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListBucketsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListBucketsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListBucketsWithContext(_a0 context.Context, _a1 *s3.ListBucketsInput, _a2 ...request.Option) (*s3.ListBucketsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListBucketsWithContext\")\n\t}\n\n\tvar r0 *s3.ListBucketsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketsInput, ...request.Option) (*s3.ListBucketsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListBucketsInput, ...request.Option) *s3.ListBucketsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListBucketsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListBucketsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListDirectoryBuckets provides a mock function with given fields: _a0\nfunc (_m *S3API) ListDirectoryBuckets(_a0 *s3.ListDirectoryBucketsInput) (*s3.ListDirectoryBucketsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListDirectoryBuckets\")\n\t}\n\n\tvar r0 *s3.ListDirectoryBucketsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListDirectoryBucketsInput) (*s3.ListDirectoryBucketsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListDirectoryBucketsInput) *s3.ListDirectoryBucketsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListDirectoryBucketsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListDirectoryBucketsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListDirectoryBucketsPages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListDirectoryBucketsPages(_a0 *s3.ListDirectoryBucketsInput, _a1 func(*s3.ListDirectoryBucketsOutput, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListDirectoryBucketsPages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListDirectoryBucketsInput, func(*s3.ListDirectoryBucketsOutput, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListDirectoryBucketsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListDirectoryBucketsPagesWithContext(_a0 context.Context, _a1 *s3.ListDirectoryBucketsInput, _a2 func(*s3.ListDirectoryBucketsOutput, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListDirectoryBucketsPagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListDirectoryBucketsInput, func(*s3.ListDirectoryBucketsOutput, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListDirectoryBucketsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListDirectoryBucketsRequest(_a0 *s3.ListDirectoryBucketsInput) (*request.Request, *s3.ListDirectoryBucketsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListDirectoryBucketsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListDirectoryBucketsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListDirectoryBucketsInput) (*request.Request, *s3.ListDirectoryBucketsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListDirectoryBucketsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListDirectoryBucketsInput) *s3.ListDirectoryBucketsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListDirectoryBucketsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListDirectoryBucketsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListDirectoryBucketsWithContext(_a0 context.Context, _a1 *s3.ListDirectoryBucketsInput, _a2 ...request.Option) (*s3.ListDirectoryBucketsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListDirectoryBucketsWithContext\")\n\t}\n\n\tvar r0 *s3.ListDirectoryBucketsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListDirectoryBucketsInput, ...request.Option) (*s3.ListDirectoryBucketsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListDirectoryBucketsInput, ...request.Option) *s3.ListDirectoryBucketsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListDirectoryBucketsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListDirectoryBucketsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListMultipartUploads provides a mock function with given fields: _a0\nfunc (_m *S3API) ListMultipartUploads(_a0 *s3.ListMultipartUploadsInput) (*s3.ListMultipartUploadsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListMultipartUploads\")\n\t}\n\n\tvar r0 *s3.ListMultipartUploadsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListMultipartUploadsInput) (*s3.ListMultipartUploadsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListMultipartUploadsInput) *s3.ListMultipartUploadsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListMultipartUploadsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListMultipartUploadsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListMultipartUploadsPages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListMultipartUploadsPages(_a0 *s3.ListMultipartUploadsInput, _a1 func(*s3.ListMultipartUploadsOutput, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListMultipartUploadsPages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListMultipartUploadsInput, func(*s3.ListMultipartUploadsOutput, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListMultipartUploadsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListMultipartUploadsPagesWithContext(_a0 context.Context, _a1 *s3.ListMultipartUploadsInput, _a2 func(*s3.ListMultipartUploadsOutput, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListMultipartUploadsPagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListMultipartUploadsInput, func(*s3.ListMultipartUploadsOutput, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListMultipartUploadsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListMultipartUploadsRequest(_a0 *s3.ListMultipartUploadsInput) (*request.Request, *s3.ListMultipartUploadsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListMultipartUploadsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListMultipartUploadsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListMultipartUploadsInput) (*request.Request, *s3.ListMultipartUploadsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListMultipartUploadsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListMultipartUploadsInput) *s3.ListMultipartUploadsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListMultipartUploadsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListMultipartUploadsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListMultipartUploadsWithContext(_a0 context.Context, _a1 *s3.ListMultipartUploadsInput, _a2 ...request.Option) (*s3.ListMultipartUploadsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListMultipartUploadsWithContext\")\n\t}\n\n\tvar r0 *s3.ListMultipartUploadsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListMultipartUploadsInput, ...request.Option) (*s3.ListMultipartUploadsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListMultipartUploadsInput, ...request.Option) *s3.ListMultipartUploadsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListMultipartUploadsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListMultipartUploadsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectVersions provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjectVersions(_a0 *s3.ListObjectVersionsInput) (*s3.ListObjectVersionsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectVersions\")\n\t}\n\n\tvar r0 *s3.ListObjectVersionsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectVersionsInput) (*s3.ListObjectVersionsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectVersionsInput) *s3.ListObjectVersionsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectVersionsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectVersionsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectVersionsPages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListObjectVersionsPages(_a0 *s3.ListObjectVersionsInput, _a1 func(*s3.ListObjectVersionsOutput, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectVersionsPages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectVersionsInput, func(*s3.ListObjectVersionsOutput, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectVersionsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListObjectVersionsPagesWithContext(_a0 context.Context, _a1 *s3.ListObjectVersionsInput, _a2 func(*s3.ListObjectVersionsOutput, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectVersionsPagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectVersionsInput, func(*s3.ListObjectVersionsOutput, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectVersionsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjectVersionsRequest(_a0 *s3.ListObjectVersionsInput) (*request.Request, *s3.ListObjectVersionsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectVersionsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListObjectVersionsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectVersionsInput) (*request.Request, *s3.ListObjectVersionsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectVersionsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectVersionsInput) *s3.ListObjectVersionsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListObjectVersionsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectVersionsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListObjectVersionsWithContext(_a0 context.Context, _a1 *s3.ListObjectVersionsInput, _a2 ...request.Option) (*s3.ListObjectVersionsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectVersionsWithContext\")\n\t}\n\n\tvar r0 *s3.ListObjectVersionsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectVersionsInput, ...request.Option) (*s3.ListObjectVersionsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectVersionsInput, ...request.Option) *s3.ListObjectVersionsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectVersionsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListObjectVersionsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjects provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjects(_a0 *s3.ListObjectsInput) (*s3.ListObjectsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjects\")\n\t}\n\n\tvar r0 *s3.ListObjectsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsInput) (*s3.ListObjectsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsInput) *s3.ListObjectsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectsPages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListObjectsPages(_a0 *s3.ListObjectsInput, _a1 func(*s3.ListObjectsOutput, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsPages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsInput, func(*s3.ListObjectsOutput, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListObjectsPagesWithContext(_a0 context.Context, _a1 *s3.ListObjectsInput, _a2 func(*s3.ListObjectsOutput, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsPagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsInput, func(*s3.ListObjectsOutput, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjectsRequest(_a0 *s3.ListObjectsInput) (*request.Request, *s3.ListObjectsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListObjectsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsInput) (*request.Request, *s3.ListObjectsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectsInput) *s3.ListObjectsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListObjectsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectsV2 provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjectsV2(_a0 *s3.ListObjectsV2Input) (*s3.ListObjectsV2Output, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsV2\")\n\t}\n\n\tvar r0 *s3.ListObjectsV2Output\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsV2Input) (*s3.ListObjectsV2Output, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsV2Input) *s3.ListObjectsV2Output); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectsV2Output)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectsV2Input) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectsV2Pages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListObjectsV2Pages(_a0 *s3.ListObjectsV2Input, _a1 func(*s3.ListObjectsV2Output, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsV2Pages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsV2Input, func(*s3.ListObjectsV2Output, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectsV2PagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListObjectsV2PagesWithContext(_a0 context.Context, _a1 *s3.ListObjectsV2Input, _a2 func(*s3.ListObjectsV2Output, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsV2PagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsV2Input, func(*s3.ListObjectsV2Output, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListObjectsV2Request provides a mock function with given fields: _a0\nfunc (_m *S3API) ListObjectsV2Request(_a0 *s3.ListObjectsV2Input) (*request.Request, *s3.ListObjectsV2Output) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsV2Request\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListObjectsV2Output\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsV2Input) (*request.Request, *s3.ListObjectsV2Output)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListObjectsV2Input) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListObjectsV2Input) *s3.ListObjectsV2Output); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListObjectsV2Output)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectsV2WithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListObjectsV2WithContext(_a0 context.Context, _a1 *s3.ListObjectsV2Input, _a2 ...request.Option) (*s3.ListObjectsV2Output, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsV2WithContext\")\n\t}\n\n\tvar r0 *s3.ListObjectsV2Output\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsV2Input, ...request.Option) (*s3.ListObjectsV2Output, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsV2Input, ...request.Option) *s3.ListObjectsV2Output); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectsV2Output)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListObjectsV2Input, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListObjectsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListObjectsWithContext(_a0 context.Context, _a1 *s3.ListObjectsInput, _a2 ...request.Option) (*s3.ListObjectsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListObjectsWithContext\")\n\t}\n\n\tvar r0 *s3.ListObjectsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsInput, ...request.Option) (*s3.ListObjectsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListObjectsInput, ...request.Option) *s3.ListObjectsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListObjectsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListObjectsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListParts provides a mock function with given fields: _a0\nfunc (_m *S3API) ListParts(_a0 *s3.ListPartsInput) (*s3.ListPartsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListParts\")\n\t}\n\n\tvar r0 *s3.ListPartsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListPartsInput) (*s3.ListPartsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListPartsInput) *s3.ListPartsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListPartsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListPartsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListPartsPages provides a mock function with given fields: _a0, _a1\nfunc (_m *S3API) ListPartsPages(_a0 *s3.ListPartsInput, _a1 func(*s3.ListPartsOutput, bool) bool) error {\n\tret := _m.Called(_a0, _a1)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListPartsPages\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.ListPartsInput, func(*s3.ListPartsOutput, bool) bool) error); ok {\n\t\tr0 = rf(_a0, _a1)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListPartsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3\nfunc (_m *S3API) ListPartsPagesWithContext(_a0 context.Context, _a1 *s3.ListPartsInput, _a2 func(*s3.ListPartsOutput, bool) bool, _a3 ...request.Option) error {\n\t_va := make([]interface{}, len(_a3))\n\tfor _i := range _a3 {\n\t\t_va[_i] = _a3[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1, _a2)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListPartsPagesWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListPartsInput, func(*s3.ListPartsOutput, bool) bool, ...request.Option) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2, _a3...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ListPartsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) ListPartsRequest(_a0 *s3.ListPartsInput) (*request.Request, *s3.ListPartsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListPartsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.ListPartsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.ListPartsInput) (*request.Request, *s3.ListPartsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.ListPartsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.ListPartsInput) *s3.ListPartsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.ListPartsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// ListPartsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) ListPartsWithContext(_a0 context.Context, _a1 *s3.ListPartsInput, _a2 ...request.Option) (*s3.ListPartsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListPartsWithContext\")\n\t}\n\n\tvar r0 *s3.ListPartsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListPartsInput, ...request.Option) (*s3.ListPartsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.ListPartsInput, ...request.Option) *s3.ListPartsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.ListPartsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.ListPartsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAccelerateConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAccelerateConfiguration(_a0 *s3.PutBucketAccelerateConfigurationInput) (*s3.PutBucketAccelerateConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAccelerateConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketAccelerateConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAccelerateConfigurationInput) (*s3.PutBucketAccelerateConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAccelerateConfigurationInput) *s3.PutBucketAccelerateConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAccelerateConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAccelerateConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAccelerateConfigurationRequest(_a0 *s3.PutBucketAccelerateConfigurationInput) (*request.Request, *s3.PutBucketAccelerateConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAccelerateConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketAccelerateConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAccelerateConfigurationInput) (*request.Request, *s3.PutBucketAccelerateConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAccelerateConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAccelerateConfigurationInput) *s3.PutBucketAccelerateConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAccelerateConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketAccelerateConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketAccelerateConfigurationInput, _a2 ...request.Option) (*s3.PutBucketAccelerateConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAccelerateConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketAccelerateConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAccelerateConfigurationInput, ...request.Option) (*s3.PutBucketAccelerateConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAccelerateConfigurationInput, ...request.Option) *s3.PutBucketAccelerateConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAccelerateConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketAccelerateConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAcl provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAcl(_a0 *s3.PutBucketAclInput) (*s3.PutBucketAclOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAcl\")\n\t}\n\n\tvar r0 *s3.PutBucketAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAclInput) (*s3.PutBucketAclOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAclInput) *s3.PutBucketAclOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAclInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAclRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAclRequest(_a0 *s3.PutBucketAclInput) (*request.Request, *s3.PutBucketAclOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAclRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketAclOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAclInput) (*request.Request, *s3.PutBucketAclOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAclInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAclInput) *s3.PutBucketAclOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketAclOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAclWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketAclWithContext(_a0 context.Context, _a1 *s3.PutBucketAclInput, _a2 ...request.Option) (*s3.PutBucketAclOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAclWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAclInput, ...request.Option) (*s3.PutBucketAclOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAclInput, ...request.Option) *s3.PutBucketAclOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketAclInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAnalyticsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAnalyticsConfiguration(_a0 *s3.PutBucketAnalyticsConfigurationInput) (*s3.PutBucketAnalyticsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAnalyticsConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAnalyticsConfigurationInput) (*s3.PutBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAnalyticsConfigurationInput) *s3.PutBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAnalyticsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAnalyticsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketAnalyticsConfigurationRequest(_a0 *s3.PutBucketAnalyticsConfigurationInput) (*request.Request, *s3.PutBucketAnalyticsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAnalyticsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketAnalyticsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAnalyticsConfigurationInput) (*request.Request, *s3.PutBucketAnalyticsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketAnalyticsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketAnalyticsConfigurationInput) *s3.PutBucketAnalyticsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketAnalyticsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketAnalyticsConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketAnalyticsConfigurationInput, _a2 ...request.Option) (*s3.PutBucketAnalyticsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketAnalyticsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketAnalyticsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAnalyticsConfigurationInput, ...request.Option) (*s3.PutBucketAnalyticsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketAnalyticsConfigurationInput, ...request.Option) *s3.PutBucketAnalyticsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketAnalyticsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketAnalyticsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketCors provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketCors(_a0 *s3.PutBucketCorsInput) (*s3.PutBucketCorsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketCors\")\n\t}\n\n\tvar r0 *s3.PutBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketCorsInput) (*s3.PutBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketCorsInput) *s3.PutBucketCorsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketCorsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketCorsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketCorsRequest(_a0 *s3.PutBucketCorsInput) (*request.Request, *s3.PutBucketCorsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketCorsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketCorsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketCorsInput) (*request.Request, *s3.PutBucketCorsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketCorsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketCorsInput) *s3.PutBucketCorsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketCorsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketCorsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketCorsWithContext(_a0 context.Context, _a1 *s3.PutBucketCorsInput, _a2 ...request.Option) (*s3.PutBucketCorsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketCorsWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketCorsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketCorsInput, ...request.Option) (*s3.PutBucketCorsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketCorsInput, ...request.Option) *s3.PutBucketCorsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketCorsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketCorsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketEncryption provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketEncryption(_a0 *s3.PutBucketEncryptionInput) (*s3.PutBucketEncryptionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketEncryption\")\n\t}\n\n\tvar r0 *s3.PutBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketEncryptionInput) (*s3.PutBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketEncryptionInput) *s3.PutBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketEncryptionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketEncryptionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketEncryptionRequest(_a0 *s3.PutBucketEncryptionInput) (*request.Request, *s3.PutBucketEncryptionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketEncryptionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketEncryptionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketEncryptionInput) (*request.Request, *s3.PutBucketEncryptionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketEncryptionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketEncryptionInput) *s3.PutBucketEncryptionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketEncryptionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketEncryptionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketEncryptionWithContext(_a0 context.Context, _a1 *s3.PutBucketEncryptionInput, _a2 ...request.Option) (*s3.PutBucketEncryptionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketEncryptionWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketEncryptionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketEncryptionInput, ...request.Option) (*s3.PutBucketEncryptionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketEncryptionInput, ...request.Option) *s3.PutBucketEncryptionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketEncryptionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketEncryptionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketIntelligentTieringConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketIntelligentTieringConfiguration(_a0 *s3.PutBucketIntelligentTieringConfigurationInput) (*s3.PutBucketIntelligentTieringConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketIntelligentTieringConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketIntelligentTieringConfigurationInput) (*s3.PutBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketIntelligentTieringConfigurationInput) *s3.PutBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketIntelligentTieringConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketIntelligentTieringConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketIntelligentTieringConfigurationRequest(_a0 *s3.PutBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.PutBucketIntelligentTieringConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketIntelligentTieringConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketIntelligentTieringConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketIntelligentTieringConfigurationInput) (*request.Request, *s3.PutBucketIntelligentTieringConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketIntelligentTieringConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketIntelligentTieringConfigurationInput) *s3.PutBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketIntelligentTieringConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketIntelligentTieringConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketIntelligentTieringConfigurationInput, _a2 ...request.Option) (*s3.PutBucketIntelligentTieringConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketIntelligentTieringConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketIntelligentTieringConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketIntelligentTieringConfigurationInput, ...request.Option) (*s3.PutBucketIntelligentTieringConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketIntelligentTieringConfigurationInput, ...request.Option) *s3.PutBucketIntelligentTieringConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketIntelligentTieringConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketIntelligentTieringConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketInventoryConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketInventoryConfiguration(_a0 *s3.PutBucketInventoryConfigurationInput) (*s3.PutBucketInventoryConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketInventoryConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketInventoryConfigurationInput) (*s3.PutBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketInventoryConfigurationInput) *s3.PutBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketInventoryConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketInventoryConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketInventoryConfigurationRequest(_a0 *s3.PutBucketInventoryConfigurationInput) (*request.Request, *s3.PutBucketInventoryConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketInventoryConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketInventoryConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketInventoryConfigurationInput) (*request.Request, *s3.PutBucketInventoryConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketInventoryConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketInventoryConfigurationInput) *s3.PutBucketInventoryConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketInventoryConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketInventoryConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketInventoryConfigurationInput, _a2 ...request.Option) (*s3.PutBucketInventoryConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketInventoryConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketInventoryConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketInventoryConfigurationInput, ...request.Option) (*s3.PutBucketInventoryConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketInventoryConfigurationInput, ...request.Option) *s3.PutBucketInventoryConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketInventoryConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketInventoryConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycle provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLifecycle(_a0 *s3.PutBucketLifecycleInput) (*s3.PutBucketLifecycleOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycle\")\n\t}\n\n\tvar r0 *s3.PutBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleInput) (*s3.PutBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleInput) *s3.PutBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLifecycleInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycleConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLifecycleConfiguration(_a0 *s3.PutBucketLifecycleConfigurationInput) (*s3.PutBucketLifecycleConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycleConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketLifecycleConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleConfigurationInput) (*s3.PutBucketLifecycleConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleConfigurationInput) *s3.PutBucketLifecycleConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLifecycleConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycleConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLifecycleConfigurationRequest(_a0 *s3.PutBucketLifecycleConfigurationInput) (*request.Request, *s3.PutBucketLifecycleConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycleConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketLifecycleConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleConfigurationInput) (*request.Request, *s3.PutBucketLifecycleConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLifecycleConfigurationInput) *s3.PutBucketLifecycleConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycleConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketLifecycleConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketLifecycleConfigurationInput, _a2 ...request.Option) (*s3.PutBucketLifecycleConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycleConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketLifecycleConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLifecycleConfigurationInput, ...request.Option) (*s3.PutBucketLifecycleConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLifecycleConfigurationInput, ...request.Option) *s3.PutBucketLifecycleConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLifecycleConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketLifecycleConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycleRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLifecycleRequest(_a0 *s3.PutBucketLifecycleInput) (*request.Request, *s3.PutBucketLifecycleOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycleRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketLifecycleOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleInput) (*request.Request, *s3.PutBucketLifecycleOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLifecycleInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLifecycleInput) *s3.PutBucketLifecycleOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketLifecycleOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLifecycleWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketLifecycleWithContext(_a0 context.Context, _a1 *s3.PutBucketLifecycleInput, _a2 ...request.Option) (*s3.PutBucketLifecycleOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLifecycleWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketLifecycleOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLifecycleInput, ...request.Option) (*s3.PutBucketLifecycleOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLifecycleInput, ...request.Option) *s3.PutBucketLifecycleOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLifecycleOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketLifecycleInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLogging provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLogging(_a0 *s3.PutBucketLoggingInput) (*s3.PutBucketLoggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLogging\")\n\t}\n\n\tvar r0 *s3.PutBucketLoggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLoggingInput) (*s3.PutBucketLoggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLoggingInput) *s3.PutBucketLoggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLoggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLoggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLoggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketLoggingRequest(_a0 *s3.PutBucketLoggingInput) (*request.Request, *s3.PutBucketLoggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLoggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketLoggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLoggingInput) (*request.Request, *s3.PutBucketLoggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketLoggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketLoggingInput) *s3.PutBucketLoggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketLoggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketLoggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketLoggingWithContext(_a0 context.Context, _a1 *s3.PutBucketLoggingInput, _a2 ...request.Option) (*s3.PutBucketLoggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketLoggingWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketLoggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLoggingInput, ...request.Option) (*s3.PutBucketLoggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketLoggingInput, ...request.Option) *s3.PutBucketLoggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketLoggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketLoggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketMetricsConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketMetricsConfiguration(_a0 *s3.PutBucketMetricsConfigurationInput) (*s3.PutBucketMetricsConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketMetricsConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketMetricsConfigurationInput) (*s3.PutBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketMetricsConfigurationInput) *s3.PutBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketMetricsConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketMetricsConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketMetricsConfigurationRequest(_a0 *s3.PutBucketMetricsConfigurationInput) (*request.Request, *s3.PutBucketMetricsConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketMetricsConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketMetricsConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketMetricsConfigurationInput) (*request.Request, *s3.PutBucketMetricsConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketMetricsConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketMetricsConfigurationInput) *s3.PutBucketMetricsConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketMetricsConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketMetricsConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketMetricsConfigurationInput, _a2 ...request.Option) (*s3.PutBucketMetricsConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketMetricsConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketMetricsConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketMetricsConfigurationInput, ...request.Option) (*s3.PutBucketMetricsConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketMetricsConfigurationInput, ...request.Option) *s3.PutBucketMetricsConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketMetricsConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketMetricsConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotification provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketNotification(_a0 *s3.PutBucketNotificationInput) (*s3.PutBucketNotificationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotification\")\n\t}\n\n\tvar r0 *s3.PutBucketNotificationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationInput) (*s3.PutBucketNotificationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationInput) *s3.PutBucketNotificationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketNotificationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketNotificationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotificationConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketNotificationConfiguration(_a0 *s3.PutBucketNotificationConfigurationInput) (*s3.PutBucketNotificationConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotificationConfiguration\")\n\t}\n\n\tvar r0 *s3.PutBucketNotificationConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationConfigurationInput) (*s3.PutBucketNotificationConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationConfigurationInput) *s3.PutBucketNotificationConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketNotificationConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketNotificationConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotificationConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketNotificationConfigurationRequest(_a0 *s3.PutBucketNotificationConfigurationInput) (*request.Request, *s3.PutBucketNotificationConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotificationConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketNotificationConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationConfigurationInput) (*request.Request, *s3.PutBucketNotificationConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketNotificationConfigurationInput) *s3.PutBucketNotificationConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketNotificationConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotificationConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketNotificationConfigurationWithContext(_a0 context.Context, _a1 *s3.PutBucketNotificationConfigurationInput, _a2 ...request.Option) (*s3.PutBucketNotificationConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotificationConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketNotificationConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketNotificationConfigurationInput, ...request.Option) (*s3.PutBucketNotificationConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketNotificationConfigurationInput, ...request.Option) *s3.PutBucketNotificationConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketNotificationConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketNotificationConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotificationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketNotificationRequest(_a0 *s3.PutBucketNotificationInput) (*request.Request, *s3.PutBucketNotificationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotificationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketNotificationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationInput) (*request.Request, *s3.PutBucketNotificationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketNotificationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketNotificationInput) *s3.PutBucketNotificationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketNotificationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketNotificationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketNotificationWithContext(_a0 context.Context, _a1 *s3.PutBucketNotificationInput, _a2 ...request.Option) (*s3.PutBucketNotificationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketNotificationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketNotificationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketNotificationInput, ...request.Option) (*s3.PutBucketNotificationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketNotificationInput, ...request.Option) *s3.PutBucketNotificationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketNotificationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketNotificationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketOwnershipControls provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketOwnershipControls(_a0 *s3.PutBucketOwnershipControlsInput) (*s3.PutBucketOwnershipControlsOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketOwnershipControls\")\n\t}\n\n\tvar r0 *s3.PutBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketOwnershipControlsInput) (*s3.PutBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketOwnershipControlsInput) *s3.PutBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketOwnershipControlsInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketOwnershipControlsRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketOwnershipControlsRequest(_a0 *s3.PutBucketOwnershipControlsInput) (*request.Request, *s3.PutBucketOwnershipControlsOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketOwnershipControlsRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketOwnershipControlsOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketOwnershipControlsInput) (*request.Request, *s3.PutBucketOwnershipControlsOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketOwnershipControlsInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketOwnershipControlsInput) *s3.PutBucketOwnershipControlsOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketOwnershipControlsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketOwnershipControlsWithContext(_a0 context.Context, _a1 *s3.PutBucketOwnershipControlsInput, _a2 ...request.Option) (*s3.PutBucketOwnershipControlsOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketOwnershipControlsWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketOwnershipControlsOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketOwnershipControlsInput, ...request.Option) (*s3.PutBucketOwnershipControlsOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketOwnershipControlsInput, ...request.Option) *s3.PutBucketOwnershipControlsOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketOwnershipControlsOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketOwnershipControlsInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketPolicy provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketPolicy(_a0 *s3.PutBucketPolicyInput) (*s3.PutBucketPolicyOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketPolicy\")\n\t}\n\n\tvar r0 *s3.PutBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketPolicyInput) (*s3.PutBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketPolicyInput) *s3.PutBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketPolicyInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketPolicyRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketPolicyRequest(_a0 *s3.PutBucketPolicyInput) (*request.Request, *s3.PutBucketPolicyOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketPolicyRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketPolicyOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketPolicyInput) (*request.Request, *s3.PutBucketPolicyOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketPolicyInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketPolicyInput) *s3.PutBucketPolicyOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketPolicyOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketPolicyWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketPolicyWithContext(_a0 context.Context, _a1 *s3.PutBucketPolicyInput, _a2 ...request.Option) (*s3.PutBucketPolicyOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketPolicyWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketPolicyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketPolicyInput, ...request.Option) (*s3.PutBucketPolicyOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketPolicyInput, ...request.Option) *s3.PutBucketPolicyOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketPolicyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketPolicyInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketReplication provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketReplication(_a0 *s3.PutBucketReplicationInput) (*s3.PutBucketReplicationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketReplication\")\n\t}\n\n\tvar r0 *s3.PutBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketReplicationInput) (*s3.PutBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketReplicationInput) *s3.PutBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketReplicationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketReplicationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketReplicationRequest(_a0 *s3.PutBucketReplicationInput) (*request.Request, *s3.PutBucketReplicationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketReplicationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketReplicationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketReplicationInput) (*request.Request, *s3.PutBucketReplicationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketReplicationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketReplicationInput) *s3.PutBucketReplicationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketReplicationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketReplicationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketReplicationWithContext(_a0 context.Context, _a1 *s3.PutBucketReplicationInput, _a2 ...request.Option) (*s3.PutBucketReplicationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketReplicationWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketReplicationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketReplicationInput, ...request.Option) (*s3.PutBucketReplicationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketReplicationInput, ...request.Option) *s3.PutBucketReplicationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketReplicationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketReplicationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketRequestPayment provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketRequestPayment(_a0 *s3.PutBucketRequestPaymentInput) (*s3.PutBucketRequestPaymentOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketRequestPayment\")\n\t}\n\n\tvar r0 *s3.PutBucketRequestPaymentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketRequestPaymentInput) (*s3.PutBucketRequestPaymentOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketRequestPaymentInput) *s3.PutBucketRequestPaymentOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketRequestPaymentInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketRequestPaymentRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketRequestPaymentRequest(_a0 *s3.PutBucketRequestPaymentInput) (*request.Request, *s3.PutBucketRequestPaymentOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketRequestPaymentRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketRequestPaymentOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketRequestPaymentInput) (*request.Request, *s3.PutBucketRequestPaymentOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketRequestPaymentInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketRequestPaymentInput) *s3.PutBucketRequestPaymentOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketRequestPaymentWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketRequestPaymentWithContext(_a0 context.Context, _a1 *s3.PutBucketRequestPaymentInput, _a2 ...request.Option) (*s3.PutBucketRequestPaymentOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketRequestPaymentWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketRequestPaymentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketRequestPaymentInput, ...request.Option) (*s3.PutBucketRequestPaymentOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketRequestPaymentInput, ...request.Option) *s3.PutBucketRequestPaymentOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketRequestPaymentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketRequestPaymentInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketTagging(_a0 *s3.PutBucketTaggingInput) (*s3.PutBucketTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketTagging\")\n\t}\n\n\tvar r0 *s3.PutBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketTaggingInput) (*s3.PutBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketTaggingInput) *s3.PutBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketTaggingRequest(_a0 *s3.PutBucketTaggingInput) (*request.Request, *s3.PutBucketTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketTaggingInput) (*request.Request, *s3.PutBucketTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketTaggingInput) *s3.PutBucketTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketTaggingWithContext(_a0 context.Context, _a1 *s3.PutBucketTaggingInput, _a2 ...request.Option) (*s3.PutBucketTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketTaggingInput, ...request.Option) (*s3.PutBucketTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketTaggingInput, ...request.Option) *s3.PutBucketTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketVersioning provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketVersioning(_a0 *s3.PutBucketVersioningInput) (*s3.PutBucketVersioningOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketVersioning\")\n\t}\n\n\tvar r0 *s3.PutBucketVersioningOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketVersioningInput) (*s3.PutBucketVersioningOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketVersioningInput) *s3.PutBucketVersioningOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketVersioningOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketVersioningInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketVersioningRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketVersioningRequest(_a0 *s3.PutBucketVersioningInput) (*request.Request, *s3.PutBucketVersioningOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketVersioningRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketVersioningOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketVersioningInput) (*request.Request, *s3.PutBucketVersioningOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketVersioningInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketVersioningInput) *s3.PutBucketVersioningOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketVersioningOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketVersioningWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketVersioningWithContext(_a0 context.Context, _a1 *s3.PutBucketVersioningInput, _a2 ...request.Option) (*s3.PutBucketVersioningOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketVersioningWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketVersioningOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketVersioningInput, ...request.Option) (*s3.PutBucketVersioningOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketVersioningInput, ...request.Option) *s3.PutBucketVersioningOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketVersioningOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketVersioningInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketWebsite provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketWebsite(_a0 *s3.PutBucketWebsiteInput) (*s3.PutBucketWebsiteOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketWebsite\")\n\t}\n\n\tvar r0 *s3.PutBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketWebsiteInput) (*s3.PutBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketWebsiteInput) *s3.PutBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketWebsiteInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketWebsiteRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutBucketWebsiteRequest(_a0 *s3.PutBucketWebsiteInput) (*request.Request, *s3.PutBucketWebsiteOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketWebsiteRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutBucketWebsiteOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketWebsiteInput) (*request.Request, *s3.PutBucketWebsiteOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutBucketWebsiteInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutBucketWebsiteInput) *s3.PutBucketWebsiteOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutBucketWebsiteOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutBucketWebsiteWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutBucketWebsiteWithContext(_a0 context.Context, _a1 *s3.PutBucketWebsiteInput, _a2 ...request.Option) (*s3.PutBucketWebsiteOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutBucketWebsiteWithContext\")\n\t}\n\n\tvar r0 *s3.PutBucketWebsiteOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketWebsiteInput, ...request.Option) (*s3.PutBucketWebsiteOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutBucketWebsiteInput, ...request.Option) *s3.PutBucketWebsiteOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutBucketWebsiteOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutBucketWebsiteInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObject provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObject(_a0 *s3.PutObjectInput) (*s3.PutObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObject\")\n\t}\n\n\tvar r0 *s3.PutObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectInput) (*s3.PutObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectInput) *s3.PutObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectAcl provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectAcl(_a0 *s3.PutObjectAclInput) (*s3.PutObjectAclOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectAcl\")\n\t}\n\n\tvar r0 *s3.PutObjectAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectAclInput) (*s3.PutObjectAclOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectAclInput) *s3.PutObjectAclOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectAclInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectAclRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectAclRequest(_a0 *s3.PutObjectAclInput) (*request.Request, *s3.PutObjectAclOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectAclRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectAclOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectAclInput) (*request.Request, *s3.PutObjectAclOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectAclInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectAclInput) *s3.PutObjectAclOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectAclOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectAclWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectAclWithContext(_a0 context.Context, _a1 *s3.PutObjectAclInput, _a2 ...request.Option) (*s3.PutObjectAclOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectAclWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectAclOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectAclInput, ...request.Option) (*s3.PutObjectAclOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectAclInput, ...request.Option) *s3.PutObjectAclOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectAclOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectAclInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLegalHold provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectLegalHold(_a0 *s3.PutObjectLegalHoldInput) (*s3.PutObjectLegalHoldOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLegalHold\")\n\t}\n\n\tvar r0 *s3.PutObjectLegalHoldOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLegalHoldInput) (*s3.PutObjectLegalHoldOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLegalHoldInput) *s3.PutObjectLegalHoldOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectLegalHoldInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLegalHoldRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectLegalHoldRequest(_a0 *s3.PutObjectLegalHoldInput) (*request.Request, *s3.PutObjectLegalHoldOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLegalHoldRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectLegalHoldOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLegalHoldInput) (*request.Request, *s3.PutObjectLegalHoldOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLegalHoldInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectLegalHoldInput) *s3.PutObjectLegalHoldOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLegalHoldWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectLegalHoldWithContext(_a0 context.Context, _a1 *s3.PutObjectLegalHoldInput, _a2 ...request.Option) (*s3.PutObjectLegalHoldOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLegalHoldWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectLegalHoldOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectLegalHoldInput, ...request.Option) (*s3.PutObjectLegalHoldOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectLegalHoldInput, ...request.Option) *s3.PutObjectLegalHoldOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectLegalHoldOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectLegalHoldInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLockConfiguration provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectLockConfiguration(_a0 *s3.PutObjectLockConfigurationInput) (*s3.PutObjectLockConfigurationOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLockConfiguration\")\n\t}\n\n\tvar r0 *s3.PutObjectLockConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLockConfigurationInput) (*s3.PutObjectLockConfigurationOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLockConfigurationInput) *s3.PutObjectLockConfigurationOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectLockConfigurationInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLockConfigurationRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectLockConfigurationRequest(_a0 *s3.PutObjectLockConfigurationInput) (*request.Request, *s3.PutObjectLockConfigurationOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLockConfigurationRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectLockConfigurationOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLockConfigurationInput) (*request.Request, *s3.PutObjectLockConfigurationOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectLockConfigurationInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectLockConfigurationInput) *s3.PutObjectLockConfigurationOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectLockConfigurationWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectLockConfigurationWithContext(_a0 context.Context, _a1 *s3.PutObjectLockConfigurationInput, _a2 ...request.Option) (*s3.PutObjectLockConfigurationOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectLockConfigurationWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectLockConfigurationOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectLockConfigurationInput, ...request.Option) (*s3.PutObjectLockConfigurationOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectLockConfigurationInput, ...request.Option) *s3.PutObjectLockConfigurationOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectLockConfigurationOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectLockConfigurationInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectRequest(_a0 *s3.PutObjectInput) (*request.Request, *s3.PutObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectInput) (*request.Request, *s3.PutObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectInput) *s3.PutObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectRetention provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectRetention(_a0 *s3.PutObjectRetentionInput) (*s3.PutObjectRetentionOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectRetention\")\n\t}\n\n\tvar r0 *s3.PutObjectRetentionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectRetentionInput) (*s3.PutObjectRetentionOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectRetentionInput) *s3.PutObjectRetentionOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectRetentionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectRetentionInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectRetentionRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectRetentionRequest(_a0 *s3.PutObjectRetentionInput) (*request.Request, *s3.PutObjectRetentionOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectRetentionRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectRetentionOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectRetentionInput) (*request.Request, *s3.PutObjectRetentionOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectRetentionInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectRetentionInput) *s3.PutObjectRetentionOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectRetentionOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectRetentionWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectRetentionWithContext(_a0 context.Context, _a1 *s3.PutObjectRetentionInput, _a2 ...request.Option) (*s3.PutObjectRetentionOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectRetentionWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectRetentionOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectRetentionInput, ...request.Option) (*s3.PutObjectRetentionOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectRetentionInput, ...request.Option) *s3.PutObjectRetentionOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectRetentionOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectRetentionInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectTagging provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectTagging(_a0 *s3.PutObjectTaggingInput) (*s3.PutObjectTaggingOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectTagging\")\n\t}\n\n\tvar r0 *s3.PutObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectTaggingInput) (*s3.PutObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectTaggingInput) *s3.PutObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectTaggingInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectTaggingRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutObjectTaggingRequest(_a0 *s3.PutObjectTaggingInput) (*request.Request, *s3.PutObjectTaggingOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectTaggingRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutObjectTaggingOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectTaggingInput) (*request.Request, *s3.PutObjectTaggingOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutObjectTaggingInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutObjectTaggingInput) *s3.PutObjectTaggingOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutObjectTaggingOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectTaggingWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectTaggingWithContext(_a0 context.Context, _a1 *s3.PutObjectTaggingInput, _a2 ...request.Option) (*s3.PutObjectTaggingOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectTaggingWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectTaggingOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectTaggingInput, ...request.Option) (*s3.PutObjectTaggingOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectTaggingInput, ...request.Option) *s3.PutObjectTaggingOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectTaggingOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectTaggingInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutObjectWithContext(_a0 context.Context, _a1 *s3.PutObjectInput, _a2 ...request.Option) (*s3.PutObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutObjectWithContext\")\n\t}\n\n\tvar r0 *s3.PutObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectInput, ...request.Option) (*s3.PutObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutObjectInput, ...request.Option) *s3.PutObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutPublicAccessBlock provides a mock function with given fields: _a0\nfunc (_m *S3API) PutPublicAccessBlock(_a0 *s3.PutPublicAccessBlockInput) (*s3.PutPublicAccessBlockOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutPublicAccessBlock\")\n\t}\n\n\tvar r0 *s3.PutPublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.PutPublicAccessBlockInput) (*s3.PutPublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutPublicAccessBlockInput) *s3.PutPublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutPublicAccessBlockInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutPublicAccessBlockRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) PutPublicAccessBlockRequest(_a0 *s3.PutPublicAccessBlockInput) (*request.Request, *s3.PutPublicAccessBlockOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutPublicAccessBlockRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.PutPublicAccessBlockOutput\n\tif rf, ok := ret.Get(0).(func(*s3.PutPublicAccessBlockInput) (*request.Request, *s3.PutPublicAccessBlockOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.PutPublicAccessBlockInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.PutPublicAccessBlockInput) *s3.PutPublicAccessBlockOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.PutPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// PutPublicAccessBlockWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) PutPublicAccessBlockWithContext(_a0 context.Context, _a1 *s3.PutPublicAccessBlockInput, _a2 ...request.Option) (*s3.PutPublicAccessBlockOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for PutPublicAccessBlockWithContext\")\n\t}\n\n\tvar r0 *s3.PutPublicAccessBlockOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutPublicAccessBlockInput, ...request.Option) (*s3.PutPublicAccessBlockOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.PutPublicAccessBlockInput, ...request.Option) *s3.PutPublicAccessBlockOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.PutPublicAccessBlockOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.PutPublicAccessBlockInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// RestoreObject provides a mock function with given fields: _a0\nfunc (_m *S3API) RestoreObject(_a0 *s3.RestoreObjectInput) (*s3.RestoreObjectOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RestoreObject\")\n\t}\n\n\tvar r0 *s3.RestoreObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.RestoreObjectInput) (*s3.RestoreObjectOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.RestoreObjectInput) *s3.RestoreObjectOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.RestoreObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.RestoreObjectInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// RestoreObjectRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) RestoreObjectRequest(_a0 *s3.RestoreObjectInput) (*request.Request, *s3.RestoreObjectOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RestoreObjectRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.RestoreObjectOutput\n\tif rf, ok := ret.Get(0).(func(*s3.RestoreObjectInput) (*request.Request, *s3.RestoreObjectOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.RestoreObjectInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.RestoreObjectInput) *s3.RestoreObjectOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.RestoreObjectOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// RestoreObjectWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) RestoreObjectWithContext(_a0 context.Context, _a1 *s3.RestoreObjectInput, _a2 ...request.Option) (*s3.RestoreObjectOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RestoreObjectWithContext\")\n\t}\n\n\tvar r0 *s3.RestoreObjectOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.RestoreObjectInput, ...request.Option) (*s3.RestoreObjectOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.RestoreObjectInput, ...request.Option) *s3.RestoreObjectOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.RestoreObjectOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.RestoreObjectInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// SelectObjectContent provides a mock function with given fields: _a0\nfunc (_m *S3API) SelectObjectContent(_a0 *s3.SelectObjectContentInput) (*s3.SelectObjectContentOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for SelectObjectContent\")\n\t}\n\n\tvar r0 *s3.SelectObjectContentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.SelectObjectContentInput) (*s3.SelectObjectContentOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.SelectObjectContentInput) *s3.SelectObjectContentOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.SelectObjectContentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.SelectObjectContentInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// SelectObjectContentRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) SelectObjectContentRequest(_a0 *s3.SelectObjectContentInput) (*request.Request, *s3.SelectObjectContentOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for SelectObjectContentRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.SelectObjectContentOutput\n\tif rf, ok := ret.Get(0).(func(*s3.SelectObjectContentInput) (*request.Request, *s3.SelectObjectContentOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.SelectObjectContentInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.SelectObjectContentInput) *s3.SelectObjectContentOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.SelectObjectContentOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// SelectObjectContentWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) SelectObjectContentWithContext(_a0 context.Context, _a1 *s3.SelectObjectContentInput, _a2 ...request.Option) (*s3.SelectObjectContentOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for SelectObjectContentWithContext\")\n\t}\n\n\tvar r0 *s3.SelectObjectContentOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.SelectObjectContentInput, ...request.Option) (*s3.SelectObjectContentOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.SelectObjectContentInput, ...request.Option) *s3.SelectObjectContentOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.SelectObjectContentOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.SelectObjectContentInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPart provides a mock function with given fields: _a0\nfunc (_m *S3API) UploadPart(_a0 *s3.UploadPartInput) (*s3.UploadPartOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPart\")\n\t}\n\n\tvar r0 *s3.UploadPartOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartInput) (*s3.UploadPartOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartInput) *s3.UploadPartOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.UploadPartOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.UploadPartInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPartCopy provides a mock function with given fields: _a0\nfunc (_m *S3API) UploadPartCopy(_a0 *s3.UploadPartCopyInput) (*s3.UploadPartCopyOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPartCopy\")\n\t}\n\n\tvar r0 *s3.UploadPartCopyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartCopyInput) (*s3.UploadPartCopyOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartCopyInput) *s3.UploadPartCopyOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.UploadPartCopyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.UploadPartCopyInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPartCopyRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) UploadPartCopyRequest(_a0 *s3.UploadPartCopyInput) (*request.Request, *s3.UploadPartCopyOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPartCopyRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.UploadPartCopyOutput\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartCopyInput) (*request.Request, *s3.UploadPartCopyOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartCopyInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.UploadPartCopyInput) *s3.UploadPartCopyOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.UploadPartCopyOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPartCopyWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) UploadPartCopyWithContext(_a0 context.Context, _a1 *s3.UploadPartCopyInput, _a2 ...request.Option) (*s3.UploadPartCopyOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPartCopyWithContext\")\n\t}\n\n\tvar r0 *s3.UploadPartCopyOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.UploadPartCopyInput, ...request.Option) (*s3.UploadPartCopyOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.UploadPartCopyInput, ...request.Option) *s3.UploadPartCopyOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.UploadPartCopyOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.UploadPartCopyInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPartRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) UploadPartRequest(_a0 *s3.UploadPartInput) (*request.Request, *s3.UploadPartOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPartRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.UploadPartOutput\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartInput) (*request.Request, *s3.UploadPartOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.UploadPartInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.UploadPartInput) *s3.UploadPartOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.UploadPartOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// UploadPartWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) UploadPartWithContext(_a0 context.Context, _a1 *s3.UploadPartInput, _a2 ...request.Option) (*s3.UploadPartOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UploadPartWithContext\")\n\t}\n\n\tvar r0 *s3.UploadPartOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.UploadPartInput, ...request.Option) (*s3.UploadPartOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.UploadPartInput, ...request.Option) *s3.UploadPartOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.UploadPartOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.UploadPartInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// WaitUntilBucketExists provides a mock function with given fields: _a0\nfunc (_m *S3API) WaitUntilBucketExists(_a0 *s3.HeadBucketInput) error {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilBucketExists\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) error); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilBucketExistsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) WaitUntilBucketExistsWithContext(_a0 context.Context, _a1 *s3.HeadBucketInput, _a2 ...request.WaiterOption) error {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilBucketExistsWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadBucketInput, ...request.WaiterOption) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilBucketNotExists provides a mock function with given fields: _a0\nfunc (_m *S3API) WaitUntilBucketNotExists(_a0 *s3.HeadBucketInput) error {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilBucketNotExists\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadBucketInput) error); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilBucketNotExistsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) WaitUntilBucketNotExistsWithContext(_a0 context.Context, _a1 *s3.HeadBucketInput, _a2 ...request.WaiterOption) error {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilBucketNotExistsWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadBucketInput, ...request.WaiterOption) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilObjectExists provides a mock function with given fields: _a0\nfunc (_m *S3API) WaitUntilObjectExists(_a0 *s3.HeadObjectInput) error {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilObjectExists\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) error); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilObjectExistsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) WaitUntilObjectExistsWithContext(_a0 context.Context, _a1 *s3.HeadObjectInput, _a2 ...request.WaiterOption) error {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilObjectExistsWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadObjectInput, ...request.WaiterOption) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilObjectNotExists provides a mock function with given fields: _a0\nfunc (_m *S3API) WaitUntilObjectNotExists(_a0 *s3.HeadObjectInput) error {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilObjectNotExists\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(*s3.HeadObjectInput) error); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WaitUntilObjectNotExistsWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) WaitUntilObjectNotExistsWithContext(_a0 context.Context, _a1 *s3.HeadObjectInput, _a2 ...request.WaiterOption) error {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WaitUntilObjectNotExistsWithContext\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.HeadObjectInput, ...request.WaiterOption) error); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// WriteGetObjectResponse provides a mock function with given fields: _a0\nfunc (_m *S3API) WriteGetObjectResponse(_a0 *s3.WriteGetObjectResponseInput) (*s3.WriteGetObjectResponseOutput, error) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WriteGetObjectResponse\")\n\t}\n\n\tvar r0 *s3.WriteGetObjectResponseOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(*s3.WriteGetObjectResponseInput) (*s3.WriteGetObjectResponseOutput, error)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.WriteGetObjectResponseInput) *s3.WriteGetObjectResponseOutput); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.WriteGetObjectResponseOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.WriteGetObjectResponseInput) error); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// WriteGetObjectResponseRequest provides a mock function with given fields: _a0\nfunc (_m *S3API) WriteGetObjectResponseRequest(_a0 *s3.WriteGetObjectResponseInput) (*request.Request, *s3.WriteGetObjectResponseOutput) {\n\tret := _m.Called(_a0)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WriteGetObjectResponseRequest\")\n\t}\n\n\tvar r0 *request.Request\n\tvar r1 *s3.WriteGetObjectResponseOutput\n\tif rf, ok := ret.Get(0).(func(*s3.WriteGetObjectResponseInput) (*request.Request, *s3.WriteGetObjectResponseOutput)); ok {\n\t\treturn rf(_a0)\n\t}\n\tif rf, ok := ret.Get(0).(func(*s3.WriteGetObjectResponseInput) *request.Request); ok {\n\t\tr0 = rf(_a0)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*request.Request)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(*s3.WriteGetObjectResponseInput) *s3.WriteGetObjectResponseOutput); ok {\n\t\tr1 = rf(_a0)\n\t} else {\n\t\tif ret.Get(1) != nil {\n\t\t\tr1 = ret.Get(1).(*s3.WriteGetObjectResponseOutput)\n\t\t}\n\t}\n\n\treturn r0, r1\n}\n\n// WriteGetObjectResponseWithContext provides a mock function with given fields: _a0, _a1, _a2\nfunc (_m *S3API) WriteGetObjectResponseWithContext(_a0 context.Context, _a1 *s3.WriteGetObjectResponseInput, _a2 ...request.Option) (*s3.WriteGetObjectResponseOutput, error) {\n\t_va := make([]interface{}, len(_a2))\n\tfor _i := range _a2 {\n\t\t_va[_i] = _a2[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _a0, _a1)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for WriteGetObjectResponseWithContext\")\n\t}\n\n\tvar r0 *s3.WriteGetObjectResponseOutput\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.WriteGetObjectResponseInput, ...request.Option) (*s3.WriteGetObjectResponseOutput, error)); ok {\n\t\treturn rf(_a0, _a1, _a2...)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *s3.WriteGetObjectResponseInput, ...request.Option) *s3.WriteGetObjectResponseOutput); ok {\n\t\tr0 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*s3.WriteGetObjectResponseOutput)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *s3.WriteGetObjectResponseInput, ...request.Option) error); ok {\n\t\tr1 = rf(_a0, _a1, _a2...)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// NewS3API creates a new instance of S3API. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.\n// The first argument is typically a *testing.T value.\nfunc NewS3API(t interface {\n\tmock.TestingT\n\tCleanup(func())\n}) *S3API {\n\tmock := &S3API{}\n\tmock.Mock.Test(t)\n\n\tt.Cleanup(func() { mock.AssertExpectations(t) })\n\n\treturn mock\n}\n"
  },
  {
    "path": "common/archiver/s3store/queryParser.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source queryParser.go -destination queryParser_mock.go -mock_names Interface=MockQueryParser\n\npackage s3store\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\ntype (\n\t// QueryParser parses a limited SQL where clause into a struct\n\tQueryParser interface {\n\t\tParse(query string) (*parsedQuery, error)\n\t}\n\n\tqueryParser struct{}\n\n\tparsedQuery struct {\n\t\tworkflowTypeName *string\n\t\tworkflowID       *string\n\t\tstartTime        *int64\n\t\tcloseTime        *int64\n\t\tsearchPrecision  *string\n\t}\n)\n\n// All allowed fields for filtering\nconst (\n\tWorkflowTypeName = \"WorkflowTypeName\"\n\tWorkflowID       = \"WorkflowID\"\n\tStartTime        = \"StartTime\"\n\tCloseTime        = \"CloseTime\"\n\tSearchPrecision  = \"SearchPrecision\"\n)\n\n// Precision specific values\nconst (\n\tPrecisionDay    = \"Day\"\n\tPrecisionHour   = \"Hour\"\n\tPrecisionMinute = \"Minute\"\n\tPrecisionSecond = \"Second\"\n)\nconst (\n\tqueryTemplate         = \"select * from dummy where %s\"\n\tdefaultDateTimeFormat = time.RFC3339\n)\n\n// NewQueryParser creates a new query parser for filestore\nfunc NewQueryParser() QueryParser {\n\treturn &queryParser{}\n}\n\nfunc (p *queryParser) Parse(query string) (*parsedQuery, error) {\n\tstmt, err := sqlparser.Parse(fmt.Sprintf(queryTemplate, query))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\twhereExpr := stmt.(*sqlparser.Select).Where.Expr\n\tparsedQuery := &parsedQuery{}\n\tif err := p.convertWhereExpr(whereExpr, parsedQuery); err != nil {\n\t\treturn nil, err\n\t}\n\tif parsedQuery.workflowID == nil && parsedQuery.workflowTypeName == nil {\n\t\treturn nil, errors.New(\"WorkflowID or WorkflowTypeName is required in query\")\n\t}\n\tif parsedQuery.workflowID != nil && parsedQuery.workflowTypeName != nil {\n\t\treturn nil, errors.New(\"only one of WorkflowID or WorkflowTypeName can be specified in a query\")\n\t}\n\tif parsedQuery.closeTime != nil && parsedQuery.startTime != nil {\n\t\treturn nil, errors.New(\"only one of StartTime or CloseTime can be specified in a query\")\n\t}\n\tif (parsedQuery.closeTime != nil || parsedQuery.startTime != nil) && parsedQuery.searchPrecision == nil {\n\t\treturn nil, errors.New(\"SearchPrecision is required when searching for a StartTime or CloseTime\")\n\t}\n\n\tif parsedQuery.closeTime == nil && parsedQuery.startTime == nil && parsedQuery.searchPrecision != nil {\n\t\treturn nil, errors.New(\"SearchPrecision requires a StartTime or CloseTime\")\n\t}\n\treturn parsedQuery, nil\n}\n\nfunc (p *queryParser) convertWhereExpr(expr sqlparser.Expr, parsedQuery *parsedQuery) error {\n\tif expr == nil {\n\t\treturn errors.New(\"where expression is nil\")\n\t}\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn p.convertComparisonExpr(expr, parsedQuery)\n\tcase *sqlparser.AndExpr:\n\t\treturn p.convertAndExpr(expr, parsedQuery)\n\tcase *sqlparser.ParenExpr:\n\t\treturn p.convertParenExpr(expr, parsedQuery)\n\tdefault:\n\t\treturn errors.New(\"only comparison and \\\"and\\\" expression is supported\")\n\t}\n}\n\nfunc (p *queryParser) convertParenExpr(parenExpr *sqlparser.ParenExpr, parsedQuery *parsedQuery) error {\n\treturn p.convertWhereExpr(parenExpr.Expr, parsedQuery)\n}\n\nfunc (p *queryParser) convertAndExpr(andExpr *sqlparser.AndExpr, parsedQuery *parsedQuery) error {\n\tif err := p.convertWhereExpr(andExpr.Left, parsedQuery); err != nil {\n\t\treturn err\n\t}\n\treturn p.convertWhereExpr(andExpr.Right, parsedQuery)\n}\n\nfunc (p *queryParser) convertComparisonExpr(compExpr *sqlparser.ComparisonExpr, parsedQuery *parsedQuery) error {\n\tcolName, ok := compExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid filter name: %s\", sqlparser.String(compExpr.Left))\n\t}\n\tcolNameStr := sqlparser.String(colName)\n\top := compExpr.Operator\n\tvalExpr, ok := compExpr.Right.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn fmt.Errorf(\"invalid value: %s\", sqlparser.String(compExpr.Right))\n\t}\n\tvalStr := sqlparser.String(valExpr)\n\n\tswitch colNameStr {\n\tcase WorkflowTypeName:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Amazon S3\", WorkflowTypeName)\n\t\t}\n\t\tif parsedQuery.workflowTypeName != nil {\n\t\t\treturn fmt.Errorf(\"can not query %s multiple times\", WorkflowTypeName)\n\t\t}\n\t\tparsedQuery.workflowTypeName = common.StringPtr(val)\n\tcase WorkflowID:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Amazon S3\", WorkflowID)\n\t\t}\n\t\tif parsedQuery.workflowID != nil {\n\t\t\treturn fmt.Errorf(\"can not query %s multiple times\", WorkflowID)\n\t\t}\n\t\tparsedQuery.workflowID = common.StringPtr(val)\n\tcase CloseTime:\n\t\ttimestamp, err := convertToTimestamp(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Amazon S3\", CloseTime)\n\t\t}\n\t\tparsedQuery.closeTime = &timestamp\n\tcase StartTime:\n\t\ttimestamp, err := convertToTimestamp(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Amazon S3\", StartTime)\n\t\t}\n\t\tparsedQuery.startTime = &timestamp\n\tcase SearchPrecision:\n\t\tval, err := extractStringValue(valStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif op != \"=\" {\n\t\t\treturn fmt.Errorf(\"only operator = is supported for %s with Amazon S3\", SearchPrecision)\n\t\t}\n\t\tif parsedQuery.searchPrecision != nil && *parsedQuery.searchPrecision != val {\n\t\t\treturn fmt.Errorf(\"only one expression is allowed for %s\", SearchPrecision)\n\t\t}\n\t\tswitch val {\n\t\tcase PrecisionDay:\n\t\tcase PrecisionHour:\n\t\tcase PrecisionMinute:\n\t\tcase PrecisionSecond:\n\t\tdefault:\n\t\t\treturn fmt.Errorf(\"invalid value for %s: %s\", SearchPrecision, val)\n\t\t}\n\t\tparsedQuery.searchPrecision = common.StringPtr(val)\n\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown filter name: %s\", colNameStr)\n\t}\n\n\treturn nil\n}\n\nfunc convertToTimestamp(timeStr string) (int64, error) {\n\ttimestamp, err := strconv.ParseInt(timeStr, 10, 64)\n\tif err == nil {\n\t\treturn timestamp, nil\n\t}\n\ttimestampStr, err := extractStringValue(timeStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tparsedTime, err := time.Parse(defaultDateTimeFormat, timestampStr)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn parsedTime.UnixNano(), nil\n}\n\nfunc extractStringValue(s string) (string, error) {\n\tif len(s) >= 2 && s[0] == '\\'' && s[len(s)-1] == '\\'' {\n\t\treturn s[1 : len(s)-1], nil\n\t}\n\treturn \"\", fmt.Errorf(\"value %s is not a string value\", s)\n}\n"
  },
  {
    "path": "common/archiver/s3store/queryParser_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: queryParser.go\n//\n// Generated by this command:\n//\n//\tmockgen -package s3store -source queryParser.go -destination queryParser_mock.go -mock_names Interface=MockQueryParser\n//\n\n// Package s3store is a generated GoMock package.\npackage s3store\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockQueryParser is a mock of QueryParser interface.\ntype MockQueryParser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueryParserMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueryParserMockRecorder is the mock recorder for MockQueryParser.\ntype MockQueryParserMockRecorder struct {\n\tmock *MockQueryParser\n}\n\n// NewMockQueryParser creates a new mock instance.\nfunc NewMockQueryParser(ctrl *gomock.Controller) *MockQueryParser {\n\tmock := &MockQueryParser{ctrl: ctrl}\n\tmock.recorder = &MockQueryParserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueryParser) EXPECT() *MockQueryParserMockRecorder {\n\treturn m.recorder\n}\n\n// Parse mocks base method.\nfunc (m *MockQueryParser) Parse(query string) (*parsedQuery, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Parse\", query)\n\tret0, _ := ret[0].(*parsedQuery)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Parse indicates an expected call of Parse.\nfunc (mr *MockQueryParserMockRecorder) Parse(query any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Parse\", reflect.TypeOf((*MockQueryParser)(nil).Parse), query)\n}\n"
  },
  {
    "path": "common/archiver/s3store/queryParser_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage s3store\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\ntype queryParserSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tparser QueryParser\n}\n\nfunc TestQueryParserSuite(t *testing.T) {\n\tsuite.Run(t, new(queryParserSuite))\n}\n\nfunc (s *queryParserSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.parser = NewQueryParser()\n}\n\nfunc (s *queryParserSuite) TestParseWorkflowIDAndWorkflowTypeName() {\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowTypeName = \\\"random workflowTypeName\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowTypeName: common.StringPtr(\"random workflowTypeName\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" and WorkflowTypeName = \\\"random workflowTypeName\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" and WorkflowID = \\\"random workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"RunID = \\\"random runID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = 'random workflowID'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"(WorkflowID = \\\"random workflowID\\\")\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tworkflowID: common.StringPtr(\"random workflowID\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     \"runID = random workflowID\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" or WorkflowID = \\\"another workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"WorkflowID = \\\"random workflowID\\\" or runID = \\\"random runID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"workflowid = \\\"random workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     \"runID > \\\"random workflowID\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.workflowID, parsedQuery.workflowID)\n\t\ts.Equal(tc.parsedQuery.workflowTypeName, parsedQuery.workflowTypeName)\n\n\t}\n}\n\nfunc (s *queryParserSuite) TestParsePrecision() {\n\tcommonQueryPart := \"WorkflowID = \\\"random workflowID\\\" AND \"\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime = 1000 and SearchPrecision = 'Day'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tsearchPrecision: common.StringPtr(PrecisionDay),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime = 1000 and SearchPrecision = 'Hour'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tsearchPrecision: common.StringPtr(PrecisionHour),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime = 1000 and SearchPrecision = 'Minute'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tsearchPrecision: common.StringPtr(PrecisionMinute),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"StartTime = 1000 and SearchPrecision = 'Second'\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tsearchPrecision: common.StringPtr(PrecisionSecond),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"SearchPrecision = 'Second'\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"SearchPrecision = 'Invalid string'\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.searchPrecision, parsedQuery.searchPrecision)\n\t}\n}\n\nfunc (s *queryParserSuite) TestParseCloseTime() {\n\tcommonQueryPart := \"WorkflowID = \\\"random workflowID\\\" AND SearchPrecision = 'Day' AND \"\n\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime = 1000\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseTime: common.Int64Ptr(1000),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime = \\\"2019-01-01T11:11:11Z\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tcloseTime: common.Int64Ptr(1546341071000000000),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"closeTime = 2000\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"CloseTime > \\\"2019-01-01 00:00:00\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.closeTime, parsedQuery.closeTime)\n\n\t}\n}\n\nfunc (s *queryParserSuite) TestParseStartTime() {\n\tcommonQueryPart := \"WorkflowID = \\\"random workflowID\\\" AND SearchPrecision = 'Day' AND \"\n\n\ttestCases := []struct {\n\t\tquery       string\n\t\texpectErr   bool\n\t\tparsedQuery *parsedQuery\n\t}{\n\t\t{\n\t\t\tquery:     commonQueryPart + \"StartTime = 1000\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tstartTime: common.Int64Ptr(1000),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"StartTime = \\\"2019-01-01T11:11:11Z\\\"\",\n\t\t\texpectErr: false,\n\t\t\tparsedQuery: &parsedQuery{\n\t\t\t\tstartTime: common.Int64Ptr(1546341071000000000),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"startTime = 2000\",\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tquery:     commonQueryPart + \"StartTime > \\\"2019-01-01 00:00:00\\\"\",\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tparsedQuery, err := s.parser.Parse(tc.query)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.NoError(err)\n\t\ts.Equal(tc.parsedQuery.closeTime, parsedQuery.closeTime)\n\t}\n}\n"
  },
  {
    "path": "common/archiver/s3store/util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage s3store\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/awserr\"\n\t\"github.com/aws/aws-sdk-go/service/s3\"\n\t\"github.com/aws/aws-sdk-go/service/s3/s3iface\"\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// encoding & decoding util\n\nfunc encode(v interface{}) ([]byte, error) {\n\treturn json.Marshal(v)\n}\n\nfunc decodeHistoryBlob(data []byte) (*archiver.HistoryBlob, error) {\n\thistoryBlob := &archiver.HistoryBlob{}\n\terr := json.Unmarshal(data, historyBlob)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn historyBlob, nil\n}\nfunc decodeVisibilityRecord(data []byte) (*visibilityRecord, error) {\n\trecord := &visibilityRecord{}\n\terr := json.Unmarshal(data, record)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn record, nil\n}\n\nfunc serializeToken(token interface{}) ([]byte, error) {\n\tif token == nil {\n\t\treturn nil, nil\n\t}\n\treturn json.Marshal(token)\n}\n\nfunc deserializeGetHistoryToken(bytes []byte) (*getHistoryToken, error) {\n\ttoken := &getHistoryToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc deserializeQueryVisibilityToken(bytes []byte) *string {\n\tvar ret = string(bytes)\n\treturn &ret\n}\nfunc serializeQueryVisibilityToken(token string) []byte {\n\treturn []byte(token)\n}\n\n// Only validates the scheme and buckets are passed\nfunc softValidateURI(URI archiver.URI) error {\n\tif URI.Scheme() != URIScheme {\n\t\treturn archiver.ErrURISchemeMismatch\n\t}\n\tif len(URI.Hostname()) == 0 {\n\t\treturn errNoBucketSpecified\n\t}\n\treturn nil\n}\n\nfunc bucketExists(ctx context.Context, s3cli s3iface.S3API, URI archiver.URI) error {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\t_, err := s3cli.HeadBucketWithContext(ctx, &s3.HeadBucketInput{\n\t\tBucket: aws.String(URI.Hostname()),\n\t})\n\tif err == nil {\n\t\treturn nil\n\t}\n\tif isNotFoundError(err) {\n\t\treturn errBucketNotExists\n\t}\n\treturn err\n}\n\nfunc keyExists(ctx context.Context, s3cli s3iface.S3API, URI archiver.URI, key string) (bool, error) {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\t_, err := s3cli.HeadObjectWithContext(ctx, &s3.HeadObjectInput{\n\t\tBucket: aws.String(URI.Hostname()),\n\t\tKey:    aws.String(key),\n\t})\n\tif err != nil {\n\t\tif isNotFoundError(err) {\n\t\t\treturn false, nil\n\t\t}\n\t\treturn false, err\n\t}\n\treturn true, nil\n}\n\nfunc isNotFoundError(err error) bool {\n\taerr, ok := err.(awserr.Error)\n\treturn ok && (aerr.Code() == \"NotFound\")\n}\n\n// Key construction\nfunc constructHistoryKey(path, domainID, workflowID, runID string, version int64, batchIdx int) string {\n\tprefix := constructHistoryKeyPrefixWithVersion(path, domainID, workflowID, runID, version)\n\treturn fmt.Sprintf(\"%s%d\", prefix, batchIdx)\n}\n\nfunc constructHistoryKeyPrefixWithVersion(path, domainID, workflowID, runID string, version int64) string {\n\tprefix := constructHistoryKeyPrefix(path, domainID, workflowID, runID)\n\treturn fmt.Sprintf(\"%s/%v/\", prefix, version)\n}\n\nfunc constructHistoryKeyPrefix(path, domainID, workflowID, runID string) string {\n\treturn strings.TrimLeft(strings.Join([]string{path, domainID, \"history\", workflowID, runID}, \"/\"), \"/\")\n}\n\nfunc constructTimeBasedSearchKey(path, domainID, primaryIndexKey, primaryIndexValue, secondaryIndexKey string, timestamp int64, precision string) string {\n\tt := time.Unix(0, timestamp).In(time.UTC)\n\tvar timeFormat = \"\"\n\tswitch precision {\n\tcase PrecisionSecond:\n\t\ttimeFormat = \":05\"\n\t\tfallthrough\n\tcase PrecisionMinute:\n\t\ttimeFormat = \":04\" + timeFormat\n\t\tfallthrough\n\tcase PrecisionHour:\n\t\ttimeFormat = \"15\" + timeFormat\n\t\tfallthrough\n\tcase PrecisionDay:\n\t\ttimeFormat = \"2006-01-02T\" + timeFormat\n\t}\n\n\treturn fmt.Sprintf(\"%s/%s\", constructVisibilitySearchPrefix(path, domainID, primaryIndexKey, primaryIndexValue, secondaryIndexKey), t.Format(timeFormat))\n}\n\nfunc constructTimestampIndex(path, domainID, primaryIndexKey, primaryIndexValue, secondaryIndexKey string, timestamp int64, runID string) string {\n\tt := time.Unix(0, timestamp).In(time.UTC)\n\treturn fmt.Sprintf(\"%s/%s/%s\", constructVisibilitySearchPrefix(path, domainID, primaryIndexKey, primaryIndexValue, secondaryIndexKey), t.Format(time.RFC3339), runID)\n}\n\nfunc constructVisibilitySearchPrefix(path, domainID, primaryIndexKey, primaryIndexValue, secondaryIndexType string) string {\n\treturn strings.TrimLeft(strings.Join([]string{path, domainID, \"visibility\", primaryIndexKey, primaryIndexValue, secondaryIndexType}, \"/\"), \"/\")\n}\n\nfunc ensureContextTimeout(ctx context.Context) (context.Context, context.CancelFunc) {\n\tif _, ok := ctx.Deadline(); ok {\n\t\treturn ctx, func() {}\n\t}\n\treturn context.WithTimeout(ctx, defaultBlobstoreTimeout)\n}\nfunc upload(ctx context.Context, s3cli s3iface.S3API, URI archiver.URI, key string, data []byte) error {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\n\t_, err := s3cli.PutObjectWithContext(ctx, &s3.PutObjectInput{\n\t\tBucket: aws.String(URI.Hostname()),\n\t\tKey:    aws.String(key),\n\t\tBody:   bytes.NewReader(data),\n\t})\n\tif err != nil {\n\t\tif aerr, ok := err.(awserr.Error); ok {\n\t\t\tif aerr.Code() == s3.ErrCodeNoSuchBucket {\n\t\t\t\treturn &types.BadRequestError{Message: errBucketNotExists.Error()}\n\t\t\t}\n\t\t}\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc download(ctx context.Context, s3cli s3iface.S3API, URI archiver.URI, key string) ([]byte, error) {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\tresult, err := s3cli.GetObjectWithContext(ctx, &s3.GetObjectInput{\n\t\tBucket: aws.String(URI.Hostname()),\n\t\tKey:    aws.String(key),\n\t})\n\n\tif err != nil {\n\t\tif aerr, ok := err.(awserr.Error); ok {\n\t\t\tif aerr.Code() == s3.ErrCodeNoSuchBucket {\n\t\t\t\treturn nil, &types.BadRequestError{Message: errBucketNotExists.Error()}\n\t\t\t}\n\n\t\t\tif aerr.Code() == s3.ErrCodeNoSuchKey {\n\t\t\t\treturn nil, &types.EntityNotExistsError{Message: archiver.ErrHistoryNotExist.Error()}\n\t\t\t}\n\t\t}\n\t\treturn nil, err\n\t}\n\n\tdefer func() {\n\t\tif ierr := result.Body.Close(); ierr != nil {\n\t\t\terr = multierr.Append(err, ierr)\n\t\t}\n\t}()\n\n\tbody, err := ioutil.ReadAll(result.Body)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn body, nil\n}\n\nfunc contextExpired(ctx context.Context) bool {\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc convertToExecutionInfo(record *visibilityRecord) *types.WorkflowExecutionInfo {\n\treturn &types.WorkflowExecutionInfo{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: record.WorkflowID,\n\t\t\tRunID:      record.RunID,\n\t\t},\n\t\tType: &types.WorkflowType{\n\t\t\tName: record.WorkflowTypeName,\n\t\t},\n\t\tStartTime:     common.Int64Ptr(record.StartTimestamp),\n\t\tExecutionTime: common.Int64Ptr(record.ExecutionTimestamp),\n\t\tCloseTime:     common.Int64Ptr(record.CloseTimestamp),\n\t\tCloseStatus:   record.CloseStatus.Ptr(),\n\t\tHistoryLength: record.HistoryLength,\n\t\tMemo:          record.Memo,\n\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\tIndexedFields: archiver.ConvertSearchAttrToBytes(record.SearchAttributes),\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/archiver/s3store/visibilityArchiver.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage s3store\n\nimport (\n\t\"context\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/session\"\n\t\"github.com/aws/aws-sdk-go/service/s3\"\n\t\"github.com/aws/aws-sdk-go/service/s3/s3iface\"\n\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tvisibilityArchiver struct {\n\t\tcontainer   *archiver.VisibilityBootstrapContainer\n\t\ts3cli       s3iface.S3API\n\t\tqueryParser QueryParser\n\t}\n\n\tvisibilityRecord archiver.ArchiveVisibilityRequest\n\n\tqueryVisibilityRequest struct {\n\t\tdomainID      string\n\t\tpageSize      int\n\t\tnextPageToken []byte\n\t\tparsedQuery   *parsedQuery\n\t}\n\n\tindexToArchive struct {\n\t\tprimaryIndex            string\n\t\tprimaryIndexValue       string\n\t\tsecondaryIndex          string\n\t\tsecondaryIndexTimestamp int64\n\t}\n)\n\nconst (\n\terrEncodeVisibilityRecord       = \"failed to encode visibility record\"\n\tsecondaryIndexKeyStartTimeout   = \"startTimeout\"\n\tsecondaryIndexKeyCloseTimeout   = \"closeTimeout\"\n\tprimaryIndexKeyWorkflowTypeName = \"workflowTypeName\"\n\tprimaryIndexKeyWorkflowID       = \"workflowID\"\n)\n\n// NewVisibilityArchiver creates a new archiver.VisibilityArchiver based on s3\nfunc NewVisibilityArchiver(\n\tcontainer *archiver.VisibilityBootstrapContainer,\n\tconfig *config.S3Archiver,\n) (archiver.VisibilityArchiver, error) {\n\treturn newVisibilityArchiver(container, config)\n}\n\nfunc newVisibilityArchiver(\n\tcontainer *archiver.VisibilityBootstrapContainer,\n\tconfig *config.S3Archiver) (*visibilityArchiver, error) {\n\ts3Config := &aws.Config{\n\t\tEndpoint:         config.Endpoint,\n\t\tRegion:           aws.String(config.Region),\n\t\tS3ForcePathStyle: aws.Bool(config.S3ForcePathStyle),\n\t}\n\tsess, err := session.NewSession(s3Config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &visibilityArchiver{\n\t\tcontainer:   container,\n\t\ts3cli:       s3.New(sess),\n\t\tqueryParser: NewQueryParser(),\n\t}, nil\n}\n\nfunc (v *visibilityArchiver) Archive(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.ArchiveVisibilityRequest,\n\topts ...archiver.ArchiveOption,\n) (err error) {\n\tscope := v.container.MetricsClient.Scope(metrics.VisibilityArchiverScope, metrics.DomainTag(request.DomainName))\n\tfeatureCatalog := archiver.GetFeatureCatalog(opts...)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tlogger := archiver.TagLoggerWithArchiveVisibilityRequestAndURI(v.container.Logger, request, URI.String())\n\tarchiveFailReason := \"\"\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif isRetryableError(err) {\n\t\t\t\tscope.IncCounter(metrics.VisibilityArchiverArchiveTransientErrorCount)\n\t\t\t\tlogger.Error(archiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(archiveFailReason), tag.Error(err))\n\t\t\t} else {\n\t\t\t\tscope.IncCounter(metrics.VisibilityArchiverArchiveNonRetryableErrorCount)\n\t\t\t\tlogger.Error(archiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(archiveFailReason), tag.Error(err))\n\t\t\t\tif featureCatalog.NonRetriableError != nil {\n\t\t\t\t\terr = featureCatalog.NonRetriableError()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}()\n\n\tif err := softValidateURI(URI); err != nil {\n\t\tarchiveFailReason = archiver.ErrReasonInvalidURI\n\t\treturn err\n\t}\n\n\tif err := archiver.ValidateVisibilityArchivalRequest(request); err != nil {\n\t\tarchiveFailReason = archiver.ErrReasonInvalidArchiveRequest\n\t\treturn err\n\t}\n\n\tencodedVisibilityRecord, err := encode(request)\n\tif err != nil {\n\t\tarchiveFailReason = errEncodeVisibilityRecord\n\t\treturn err\n\t}\n\tindexes := createIndexesToArchive(request)\n\t// Upload archive to all indexes\n\tfor _, element := range indexes {\n\t\tkey := constructTimestampIndex(URI.Path(), request.DomainID, element.primaryIndex, element.primaryIndexValue, element.secondaryIndex, element.secondaryIndexTimestamp, request.RunID)\n\t\tif err := upload(ctx, v.s3cli, URI, key, encodedVisibilityRecord); err != nil {\n\t\t\tarchiveFailReason = errWriteKey\n\t\t\treturn err\n\t\t}\n\t}\n\tscope.IncCounter(metrics.VisibilityArchiveSuccessCount)\n\treturn nil\n}\nfunc createIndexesToArchive(request *archiver.ArchiveVisibilityRequest) []indexToArchive {\n\treturn []indexToArchive{\n\t\t{primaryIndexKeyWorkflowTypeName, request.WorkflowTypeName, secondaryIndexKeyCloseTimeout, request.CloseTimestamp},\n\t\t{primaryIndexKeyWorkflowTypeName, request.WorkflowTypeName, secondaryIndexKeyStartTimeout, request.StartTimestamp},\n\t\t{primaryIndexKeyWorkflowID, request.WorkflowID, secondaryIndexKeyCloseTimeout, request.CloseTimestamp},\n\t\t{primaryIndexKeyWorkflowID, request.WorkflowID, secondaryIndexKeyStartTimeout, request.StartTimestamp},\n\t}\n}\n\nfunc (v *visibilityArchiver) Query(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *archiver.QueryVisibilityRequest,\n) (*archiver.QueryVisibilityResponse, error) {\n\tif err := softValidateURI(URI); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidURI.Error()}\n\t}\n\n\tif err := archiver.ValidateQueryRequest(request); err != nil {\n\t\treturn nil, &types.BadRequestError{Message: archiver.ErrInvalidQueryVisibilityRequest.Error()}\n\t}\n\n\tparsedQuery, err := v.queryParser.Parse(request.Query)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: err.Error()}\n\t}\n\n\treturn v.query(ctx, URI, &queryVisibilityRequest{\n\t\tdomainID:      request.DomainID,\n\t\tpageSize:      request.PageSize,\n\t\tnextPageToken: request.NextPageToken,\n\t\tparsedQuery:   parsedQuery,\n\t})\n}\n\nfunc (v *visibilityArchiver) query(\n\tctx context.Context,\n\tURI archiver.URI,\n\trequest *queryVisibilityRequest,\n) (*archiver.QueryVisibilityResponse, error) {\n\tctx, cancel := ensureContextTimeout(ctx)\n\tdefer cancel()\n\tvar token *string\n\tif request.nextPageToken != nil {\n\t\ttoken = deserializeQueryVisibilityToken(request.nextPageToken)\n\t}\n\tprimaryIndex := primaryIndexKeyWorkflowTypeName\n\tprimaryIndexValue := request.parsedQuery.workflowTypeName\n\tif request.parsedQuery.workflowID != nil {\n\t\tprimaryIndex = primaryIndexKeyWorkflowID\n\t\tprimaryIndexValue = request.parsedQuery.workflowID\n\t}\n\tvar prefix = constructVisibilitySearchPrefix(URI.Path(), request.domainID, primaryIndex, *primaryIndexValue, secondaryIndexKeyCloseTimeout) + \"/\"\n\tif request.parsedQuery.closeTime != nil {\n\t\tprefix = constructTimeBasedSearchKey(URI.Path(), request.domainID, primaryIndex, *primaryIndexValue, secondaryIndexKeyCloseTimeout, *request.parsedQuery.closeTime, *request.parsedQuery.searchPrecision)\n\t}\n\tif request.parsedQuery.startTime != nil {\n\t\tprefix = constructTimeBasedSearchKey(URI.Path(), request.domainID, primaryIndex, *primaryIndexValue, secondaryIndexKeyStartTimeout, *request.parsedQuery.startTime, *request.parsedQuery.searchPrecision)\n\t}\n\n\tresults, err := v.s3cli.ListObjectsV2WithContext(ctx, &s3.ListObjectsV2Input{\n\t\tBucket:            aws.String(URI.Hostname()),\n\t\tPrefix:            aws.String(prefix),\n\t\tMaxKeys:           aws.Int64(int64(request.pageSize)),\n\t\tContinuationToken: token,\n\t})\n\tif err != nil {\n\t\tif isRetryableError(err) {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\treturn nil, &types.BadRequestError{Message: err.Error()}\n\t}\n\tif len(results.Contents) == 0 {\n\t\treturn &archiver.QueryVisibilityResponse{}, nil\n\t}\n\n\tresponse := &archiver.QueryVisibilityResponse{}\n\tif *results.IsTruncated {\n\t\tresponse.NextPageToken = serializeQueryVisibilityToken(*results.NextContinuationToken)\n\t}\n\tfor _, item := range results.Contents {\n\t\tencodedRecord, err := download(ctx, v.s3cli, URI, *item.Key)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\n\t\trecord, err := decodeVisibilityRecord(encodedRecord)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: err.Error()}\n\t\t}\n\t\tresponse.Executions = append(response.Executions, convertToExecutionInfo(record))\n\t}\n\treturn response, nil\n}\n\nfunc (v *visibilityArchiver) ValidateURI(URI archiver.URI) error {\n\terr := softValidateURI(URI)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn bucketExists(context.TODO(), v.s3cli, URI)\n}\n"
  },
  {
    "path": "common/archiver/s3store/visibilityArchiver_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage s3store\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/aws/aws-sdk-go/aws/awserr\"\n\t\"github.com/aws/aws-sdk-go/service/s3\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/s3store/mocks\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype visibilityArchiverSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\ts3cli *mocks.S3API\n\n\tcontainer         *archiver.VisibilityBootstrapContainer\n\tvisibilityRecords []*visibilityRecord\n\n\tcontroller      *gomock.Controller\n\ttestArchivalURI archiver.URI\n}\n\nfunc TestVisibilityArchiverSuite(t *testing.T) {\n\tsuite.Run(t, new(visibilityArchiverSuite))\n}\n\nfunc (s *visibilityArchiverSuite) TestValidateURI() {\n\ttestCases := []struct {\n\t\tURI         string\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tURI:         \"wrongscheme:///a/b/c\",\n\t\t\texpectedErr: archiver.ErrURISchemeMismatch,\n\t\t},\n\t\t{\n\t\t\tURI:         \"s3://\",\n\t\t\texpectedErr: errNoBucketSpecified,\n\t\t},\n\t\t{\n\t\t\tURI:         \"s3:///test\",\n\t\t\texpectedErr: errNoBucketSpecified,\n\t\t},\n\t\t{\n\t\t\tURI:         \"s3://bucket/a/b/c\",\n\t\t\texpectedErr: errBucketNotExists,\n\t\t},\n\t\t{\n\t\t\tURI:         testBucketURI,\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\ts.s3cli.On(\"HeadBucketWithContext\", mock.Anything, mock.MatchedBy(func(input *s3.HeadBucketInput) bool {\n\t\treturn *input.Bucket != s.testArchivalURI.Hostname()\n\t})).Return(nil, awserr.New(\"NotFound\", \"\", nil))\n\ts.s3cli.On(\"HeadBucketWithContext\", mock.Anything, mock.Anything).Return(&s3.HeadBucketOutput{}, nil)\n\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tfor _, tc := range testCases {\n\t\tURI, err := archiver.NewURI(tc.URI)\n\t\ts.NoError(err)\n\t\ts.Equal(tc.expectedErr, visibilityArchiver.ValidateURI(URI))\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) newTestVisibilityArchiver() *visibilityArchiver {\n\tarchiver := &visibilityArchiver{\n\t\tcontainer:   s.container,\n\t\ts3cli:       s.s3cli,\n\t\tqueryParser: NewQueryParser(),\n\t}\n\treturn archiver\n}\n\nconst (\n\ttestWorkflowTypeName = \"test-workflow-type\"\n)\n\nfunc (s *visibilityArchiverSuite) SetupSuite() {\n\tvar err error\n\tscope := tally.NewTestScope(\"test\", nil)\n\ts.s3cli = &mocks.S3API{}\n\tsetupFsEmulation(s.s3cli)\n\n\ts.testArchivalURI, err = archiver.NewURI(testBucketURI)\n\ts.Require().NoError(err)\n\n\ts.container = &archiver.VisibilityBootstrapContainer{\n\t\tLogger:        testlogger.New(s.T()),\n\t\tMetricsClient: metrics.NewClient(scope, metrics.History, metrics.HistogramMigration{}),\n\t}\n\ts.setupVisibilityDirectory()\n}\n\nfunc (s *visibilityArchiverSuite) TearDownSuite() {\n\n}\n\nfunc (s *visibilityArchiverSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n}\n\nfunc (s *visibilityArchiverSuite) TearDownTest() {\n\ts.controller.Finish()\n}\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_InvalidURI() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainName:         testDomainName,\n\t\tDomainID:           testDomainID,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     time.Now().UnixNano(),\n\t\tExecutionTimestamp: 0, // workflow without backoff\n\t\tCloseTimestamp:     time.Now().UnixNano(),\n\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\tHistoryLength:      int64(101),\n\t}\n\terr = visibilityArchiver.Archive(context.Background(), URI, request)\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_InvalidRequest() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\terr := visibilityArchiver.Archive(context.Background(), s.testArchivalURI, &archiver.ArchiveVisibilityRequest{})\n\ts.Error(err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Fail_NonRetriableErrorOption() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tnonRetriableErr := errors.New(\"some non-retryable error\")\n\terr := visibilityArchiver.Archive(\n\t\tcontext.Background(),\n\t\ts.testArchivalURI,\n\t\t&archiver.ArchiveVisibilityRequest{\n\t\t\tDomainID: testDomainID,\n\t\t},\n\t\tarchiver.GetNonRetriableErrorOption(nonRetriableErr),\n\t)\n\ts.Equal(nonRetriableErr, err)\n}\n\nfunc (s *visibilityArchiverSuite) TestArchive_Success() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tcloseTimestamp := time.Now()\n\trequest := &archiver.ArchiveVisibilityRequest{\n\t\tDomainID:           testDomainID,\n\t\tDomainName:         testDomainName,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     closeTimestamp.Add(-time.Hour).UnixNano(),\n\t\tExecutionTimestamp: 0, // workflow without backoff\n\t\tCloseTimestamp:     closeTimestamp.UnixNano(),\n\t\tCloseStatus:        types.WorkflowExecutionCloseStatusFailed,\n\t\tHistoryLength:      int64(101),\n\t\tMemo: &types.Memo{\n\t\t\tFields: map[string][]byte{\n\t\t\t\t\"testFields\": {1, 2, 3},\n\t\t\t},\n\t\t},\n\t\tSearchAttributes: map[string]string{\n\t\t\t\"testAttribute\": \"456\",\n\t\t},\n\t}\n\tURI, err := archiver.NewURI(testBucketURI + \"/test-archive-success\")\n\ts.NoError(err)\n\terr = visibilityArchiver.Archive(context.Background(), URI, request)\n\ts.NoError(err)\n\n\texpectedKey := constructTimestampIndex(URI.Path(), testDomainID, primaryIndexKeyWorkflowID, testWorkflowID, secondaryIndexKeyCloseTimeout, closeTimestamp.UnixNano(), testRunID)\n\tdata, err := download(context.Background(), visibilityArchiver.s3cli, URI, expectedKey)\n\ts.NoError(err, expectedKey)\n\n\tarchivedRecord := &archiver.ArchiveVisibilityRequest{}\n\terr = json.Unmarshal(data, archivedRecord)\n\ts.NoError(err)\n\ts.Equal(request, archivedRecord)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidURI() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(\"wrongscheme://\")\n\ts.NoError(err)\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 1,\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidRequest() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, &archiver.QueryVisibilityRequest{})\n\ts.Error(err)\n\ts.Nil(response)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Fail_InvalidQuery() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(nil, errors.New(\"invalid query\"))\n\tvisibilityArchiver.queryParser = mockParser\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, &archiver.QueryVisibilityRequest{\n\t\tDomainID: \"some random domainID\",\n\t\tPageSize: 10,\n\t\tQuery:    \"some invalid query\",\n\t})\n\ts.Error(err)\n\ts.Nil(response)\n}\nfunc (s *visibilityArchiverSuite) TestQuery_Success_DirectoryNotExist() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t\tcloseTime:       common.Int64Ptr(0),\n\t\tsearchPrecision: common.StringPtr(PrecisionSecond),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tQuery:    \"parsed by mockParser\",\n\t\tPageSize: 1,\n\t}\n\tresponse, err := visibilityArchiver.Query(context.Background(), s.testArchivalURI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Empty(response.Executions)\n\ts.Empty(response.NextPageToken)\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_NoNextPageToken() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tcloseTime:       common.Int64Ptr(int64(1 * time.Hour)),\n\t\tsearchPrecision: common.StringPtr(PrecisionHour),\n\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t}, nil)\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 10,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 2)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), response.Executions[0])\n}\n\nfunc (s *visibilityArchiverSuite) TestQuery_Success_SmallPageSize() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tcloseTime:       common.Int64Ptr(0),\n\t\tsearchPrecision: common.StringPtr(PrecisionDay),\n\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 2,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\tURI, err := archiver.NewURI(testBucketURI)\n\ts.NoError(err)\n\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.NotNil(response.NextPageToken)\n\ts.Len(response.Executions, 2)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), response.Executions[0])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[1]), response.Executions[1])\n\n\trequest.NextPageToken = response.NextPageToken\n\tresponse, err = visibilityArchiver.Query(context.Background(), URI, request)\n\ts.NoError(err)\n\ts.NotNil(response)\n\ts.Nil(response.NextPageToken)\n\ts.Len(response.Executions, 1)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[2]), response.Executions[0])\n}\n\ntype precisionTest struct {\n\tday       int64\n\thour      int64\n\tminute    int64\n\tsecond    int64\n\tprecision string\n}\n\nfunc (s *visibilityArchiverSuite) TestArchiveAndQueryPrecisions() {\n\tprecisionTests := []*precisionTest{\n\t\t{\n\t\t\tday:       1,\n\t\t\thour:      0,\n\t\t\tminute:    0,\n\t\t\tsecond:    0,\n\t\t\tprecision: PrecisionDay,\n\t\t},\n\t\t{\n\t\t\tday:       1,\n\t\t\thour:      1,\n\t\t\tminute:    0,\n\t\t\tsecond:    0,\n\t\t\tprecision: PrecisionDay,\n\t\t},\n\t\t{\n\t\t\tday:       2,\n\t\t\thour:      1,\n\t\t\tminute:    0,\n\t\t\tsecond:    0,\n\t\t\tprecision: PrecisionHour,\n\t\t},\n\t\t{\n\t\t\tday:       2,\n\t\t\thour:      1,\n\t\t\tminute:    30,\n\t\t\tsecond:    0,\n\t\t\tprecision: PrecisionHour,\n\t\t},\n\t\t{\n\t\t\tday:       3,\n\t\t\thour:      2,\n\t\t\tminute:    1,\n\t\t\tsecond:    0,\n\t\t\tprecision: PrecisionMinute,\n\t\t},\n\t\t{\n\t\t\tday:       3,\n\t\t\thour:      2,\n\t\t\tminute:    1,\n\t\t\tsecond:    30,\n\t\t\tprecision: PrecisionMinute,\n\t\t},\n\t\t{\n\t\t\tday:       4,\n\t\t\thour:      3,\n\t\t\tminute:    2,\n\t\t\tsecond:    1,\n\t\t\tprecision: PrecisionSecond,\n\t\t},\n\t\t{\n\t\t\tday:       4,\n\t\t\thour:      3,\n\t\t\tminute:    2,\n\t\t\tsecond:    1,\n\t\t\tprecision: PrecisionSecond,\n\t\t},\n\t\t{\n\t\t\tday:       4,\n\t\t\thour:      3,\n\t\t\tminute:    2,\n\t\t\tsecond:    2,\n\t\t\tprecision: PrecisionSecond,\n\t\t},\n\t\t{\n\t\t\tday:       4,\n\t\t\thour:      3,\n\t\t\tminute:    2,\n\t\t\tsecond:    2,\n\t\t\tprecision: PrecisionSecond,\n\t\t},\n\t}\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(testBucketURI + \"/archive-and-query-precision\")\n\ts.NoError(err)\n\n\tfor i, testData := range precisionTests {\n\t\trecord := archiver.ArchiveVisibilityRequest{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            fmt.Sprintf(\"%s-%d\", testRunID, i),\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   testData.day*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second),\n\t\t\tCloseTimestamp:   (testData.day+30)*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second),\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:    101,\n\t\t}\n\t\terr := visibilityArchiver.Archive(context.Background(), URI, &record)\n\t\ts.NoError(err)\n\t}\n\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 100,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\n\tfor i, testData := range precisionTests {\n\t\tmockParser := NewMockQueryParser(s.controller)\n\t\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\t\tcloseTime:       common.Int64Ptr((testData.day+30)*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second)),\n\t\t\tsearchPrecision: common.StringPtr(testData.precision),\n\t\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t\t}, nil).AnyTimes()\n\t\tvisibilityArchiver.queryParser = mockParser\n\n\t\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\ts.Len(response.Executions, 2, \"Iteration \", i)\n\n\t\tmockParser = NewMockQueryParser(s.controller)\n\t\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\t\tstartTime:       common.Int64Ptr((testData.day)*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second)),\n\t\t\tsearchPrecision: common.StringPtr(testData.precision),\n\t\t\tworkflowID:      common.StringPtr(testWorkflowID),\n\t\t}, nil).AnyTimes()\n\t\tvisibilityArchiver.queryParser = mockParser\n\n\t\tresponse, err = visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\ts.Len(response.Executions, 2, \"Iteration \", i)\n\n\t\tmockParser = NewMockQueryParser(s.controller)\n\t\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\t\tcloseTime:        common.Int64Ptr((testData.day+30)*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second)),\n\t\t\tsearchPrecision:  common.StringPtr(testData.precision),\n\t\t\tworkflowTypeName: common.StringPtr(testWorkflowTypeName),\n\t\t}, nil).AnyTimes()\n\t\tvisibilityArchiver.queryParser = mockParser\n\n\t\tresponse, err = visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\ts.Len(response.Executions, 2, \"Iteration \", i)\n\n\t\tmockParser = NewMockQueryParser(s.controller)\n\t\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\t\tstartTime:        common.Int64Ptr((testData.day)*int64(time.Hour)*24 + testData.hour*int64(time.Hour) + testData.minute*int64(time.Minute) + testData.second*int64(time.Second)),\n\t\t\tsearchPrecision:  common.StringPtr(testData.precision),\n\t\t\tworkflowTypeName: common.StringPtr(testWorkflowTypeName),\n\t\t}, nil).AnyTimes()\n\t\tvisibilityArchiver.queryParser = mockParser\n\n\t\tresponse, err = visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\ts.Len(response.Executions, 2, \"Iteration \", i)\n\t}\n}\nfunc (s *visibilityArchiverSuite) TestArchiveAndQuery() {\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tURI, err := archiver.NewURI(testBucketURI + \"/archive-and-query\")\n\ts.NoError(err)\n\tfor _, record := range s.visibilityRecords {\n\t\terr := visibilityArchiver.Archive(context.Background(), URI, (*archiver.ArchiveVisibilityRequest)(record))\n\t\ts.NoError(err)\n\t}\n\n\tmockParser := NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tworkflowID: common.StringPtr(testWorkflowID),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\trequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 1,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\texecutions := []*types.WorkflowExecutionInfo{}\n\tvar first = true\n\tfor first || request.NextPageToken != nil {\n\t\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\texecutions = append(executions, response.Executions...)\n\t\trequest.NextPageToken = response.NextPageToken\n\t\tfirst = false\n\t}\n\ts.Len(executions, 3)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), executions[0])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[1]), executions[1])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[2]), executions[2])\n\n\tmockParser = NewMockQueryParser(s.controller)\n\tmockParser.EXPECT().Parse(gomock.Any()).Return(&parsedQuery{\n\t\tworkflowTypeName: common.StringPtr(testWorkflowTypeName),\n\t}, nil).AnyTimes()\n\tvisibilityArchiver.queryParser = mockParser\n\trequest = &archiver.QueryVisibilityRequest{\n\t\tDomainID: testDomainID,\n\t\tPageSize: 1,\n\t\tQuery:    \"parsed by mockParser\",\n\t}\n\texecutions = []*types.WorkflowExecutionInfo{}\n\tfirst = true\n\tfor first || request.NextPageToken != nil {\n\t\tresponse, err := visibilityArchiver.Query(context.Background(), URI, request)\n\t\ts.NoError(err)\n\t\ts.NotNil(response)\n\t\texecutions = append(executions, response.Executions...)\n\t\trequest.NextPageToken = response.NextPageToken\n\t\tfirst = false\n\t}\n\ts.Len(executions, 3)\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[0]), executions[0])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[1]), executions[1])\n\ts.Equal(convertToExecutionInfo(s.visibilityRecords[2]), executions[2])\n}\n\nfunc (s *visibilityArchiverSuite) setupVisibilityDirectory() {\n\ts.visibilityRecords = []*visibilityRecord{\n\t\t{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            testRunID,\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   1,\n\t\t\tCloseTimestamp:   int64(1 * time.Hour),\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:    101,\n\t\t},\n\t\t{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            testRunID + \"1\",\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   1,\n\t\t\tCloseTimestamp:   int64(1*time.Hour + 30*time.Minute),\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:    101,\n\t\t},\n\t\t{\n\t\t\tDomainID:         testDomainID,\n\t\t\tDomainName:       testDomainName,\n\t\t\tWorkflowID:       testWorkflowID,\n\t\t\tRunID:            testRunID + \"1\",\n\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\tStartTimestamp:   1,\n\t\t\tCloseTimestamp:   int64(3 * time.Hour),\n\t\t\tCloseStatus:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\tHistoryLength:    101,\n\t\t},\n\t}\n\tvisibilityArchiver := s.newTestVisibilityArchiver()\n\tfor _, record := range s.visibilityRecords {\n\t\ts.writeVisibilityRecordForQueryTest(visibilityArchiver, record)\n\t}\n}\n\nfunc (s *visibilityArchiverSuite) writeVisibilityRecordForQueryTest(visibilityArchiver *visibilityArchiver, record *visibilityRecord) {\n\terr := visibilityArchiver.Archive(context.Background(), s.testArchivalURI, (*archiver.ArchiveVisibilityRequest)(record))\n\ts.Require().NoError(err)\n}\n"
  },
  {
    "path": "common/archiver/util.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\terrEmptyDomainID         = errors.New(\"DomainID is empty\")\n\terrEmptyDomainName       = errors.New(\"DomainName is empty\")\n\terrEmptyWorkflowID       = errors.New(\"WorkflowID is empty\")\n\terrEmptyRunID            = errors.New(\"RunID is empty\")\n\terrInvalidPageSize       = errors.New(\"PageSize should be greater than 0\")\n\terrEmptyWorkflowTypeName = errors.New(\"WorkflowTypeName is empty\")\n\terrEmptyStartTime        = errors.New(\"StartTimestamp is empty\")\n\terrEmptyCloseTime        = errors.New(\"CloseTimestamp is empty\")\n\terrEmptyQuery            = errors.New(\"Query string is empty\")\n)\n\n// TagLoggerWithArchiveHistoryRequestAndURI tags logger with fields in the archive history request and the URI\nfunc TagLoggerWithArchiveHistoryRequestAndURI(logger log.Logger, request *ArchiveHistoryRequest, URI string) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.ShardID(request.ShardID),\n\t\ttag.ArchivalRequestDomainID(request.DomainID),\n\t\ttag.ArchivalRequestDomainName(request.DomainName),\n\t\ttag.ArchivalRequestWorkflowID(request.WorkflowID),\n\t\ttag.ArchivalRequestRunID(request.RunID),\n\t\ttag.ArchivalRequestBranchToken(request.BranchToken),\n\t\ttag.ArchivalRequestNextEventID(request.NextEventID),\n\t\ttag.ArchivalRequestCloseFailoverVersion(request.CloseFailoverVersion),\n\t\ttag.ArchivalURI(URI),\n\t)\n}\n\n// TagLoggerWithArchiveVisibilityRequestAndURI tags logger with fields in the archive visibility request and the URI\nfunc TagLoggerWithArchiveVisibilityRequestAndURI(logger log.Logger, request *ArchiveVisibilityRequest, URI string) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.ArchivalRequestDomainID(request.DomainID),\n\t\ttag.ArchivalRequestDomainName(request.DomainName),\n\t\ttag.ArchivalRequestWorkflowID(request.WorkflowID),\n\t\ttag.ArchivalRequestRunID(request.RunID),\n\t\ttag.ArchvialRequestWorkflowType(request.WorkflowTypeName),\n\t\ttag.ArchivalRequestCloseTimestamp(request.CloseTimestamp),\n\t\ttag.ArchivalRequestCloseStatus(request.CloseStatus.String()),\n\t\ttag.ArchivalURI(URI),\n\t)\n}\n\n// ValidateHistoryArchiveRequest validates the archive history request\nfunc ValidateHistoryArchiveRequest(request *ArchiveHistoryRequest) error {\n\tif request.DomainID == \"\" {\n\t\treturn errEmptyDomainID\n\t}\n\tif request.WorkflowID == \"\" {\n\t\treturn errEmptyWorkflowID\n\t}\n\tif request.RunID == \"\" {\n\t\treturn errEmptyRunID\n\t}\n\tif request.DomainName == \"\" {\n\t\treturn errEmptyDomainName\n\t}\n\treturn nil\n}\n\n// ValidateGetRequest validates the get archived history request\nfunc ValidateGetRequest(request *GetHistoryRequest) error {\n\tif request.DomainID == \"\" {\n\t\treturn errEmptyDomainID\n\t}\n\tif request.WorkflowID == \"\" {\n\t\treturn errEmptyWorkflowID\n\t}\n\tif request.RunID == \"\" {\n\t\treturn errEmptyRunID\n\t}\n\tif request.PageSize == 0 {\n\t\treturn errInvalidPageSize\n\t}\n\treturn nil\n}\n\n// ValidateVisibilityArchivalRequest validates the archive visibility request\nfunc ValidateVisibilityArchivalRequest(request *ArchiveVisibilityRequest) error {\n\tif request.DomainID == \"\" {\n\t\treturn errEmptyDomainID\n\t}\n\tif request.DomainName == \"\" {\n\t\treturn errEmptyDomainName\n\t}\n\tif request.WorkflowID == \"\" {\n\t\treturn errEmptyWorkflowID\n\t}\n\tif request.RunID == \"\" {\n\t\treturn errEmptyRunID\n\t}\n\tif request.WorkflowTypeName == \"\" {\n\t\treturn errEmptyWorkflowTypeName\n\t}\n\tif request.StartTimestamp == 0 {\n\t\treturn errEmptyStartTime\n\t}\n\tif request.CloseTimestamp == 0 {\n\t\treturn errEmptyCloseTime\n\t}\n\treturn nil\n}\n\n// ValidateQueryRequest validates the query visibility request\nfunc ValidateQueryRequest(request *QueryVisibilityRequest) error {\n\tif request.DomainID == \"\" {\n\t\treturn errEmptyDomainID\n\t}\n\tif request.PageSize == 0 {\n\t\treturn errInvalidPageSize\n\t}\n\tif request.Query == \"\" {\n\t\treturn errEmptyQuery\n\t}\n\treturn nil\n}\n\n// ConvertSearchAttrToBytes converts search attribute value from string back to byte array\nfunc ConvertSearchAttrToBytes(searchAttrStr map[string]string) map[string][]byte {\n\tsearchAttr := make(map[string][]byte)\n\tfor k, v := range searchAttrStr {\n\t\tsearchAttr[k] = []byte(v)\n\t}\n\treturn searchAttr\n}\n\nfunc IsHistoryMutated(request *ArchiveHistoryRequest, historyBatches []*types.History, isLast bool, logger log.Logger) (mutated bool) {\n\tlastBatch := historyBatches[len(historyBatches)-1].Events\n\tlastEvent := lastBatch[len(lastBatch)-1]\n\tlastFailoverVersion := lastEvent.Version\n\tdefer func() {\n\t\tif mutated {\n\t\t\tlogger.Warn(ArchiveNonRetriableErrorMsg+\":history is mutated when during archival\",\n\t\t\t\ttag.ArchivalArchiveFailReason(ErrReasonHistoryMutated),\n\t\t\t\ttag.FailoverVersion(lastFailoverVersion),\n\t\t\t\ttag.TokenLastEventID(lastEvent.ID))\n\t\t}\n\t}()\n\tif lastFailoverVersion > request.CloseFailoverVersion {\n\t\treturn true\n\t}\n\n\tif !isLast {\n\t\treturn false\n\t}\n\tlastEventID := lastEvent.ID\n\treturn lastFailoverVersion != request.CloseFailoverVersion || lastEventID+1 != request.NextEventID\n}\n"
  },
  {
    "path": "common/archiver/util_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype UtilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(UtilSuite))\n}\n\nfunc (s *UtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *UtilSuite) TestHistoryMutated() {\n\ttestCases := []struct {\n\t\thistoryBatches []*types.History\n\t\trequest        *ArchiveHistoryRequest\n\t\tisLast         bool\n\t\tisMutated      bool\n\t}{\n\t\t{\n\t\t\thistoryBatches: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tVersion: 15,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\trequest: &ArchiveHistoryRequest{\n\t\t\t\tCloseFailoverVersion: 3,\n\t\t\t},\n\t\t\tisMutated: true,\n\t\t},\n\t\t{\n\t\t\thistoryBatches: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:      33,\n\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:      49,\n\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:      50,\n\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\trequest: &ArchiveHistoryRequest{\n\t\t\t\tCloseFailoverVersion: 10,\n\t\t\t\tNextEventID:          34,\n\t\t\t},\n\t\t\tisLast:    true,\n\t\t\tisMutated: true,\n\t\t},\n\t\t{\n\t\t\thistoryBatches: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tVersion: 9,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\trequest: &ArchiveHistoryRequest{\n\t\t\t\tCloseFailoverVersion: 10,\n\t\t\t},\n\t\t\tisLast:    true,\n\t\t\tisMutated: true,\n\t\t},\n\t\t{\n\t\t\thistoryBatches: []*types.History{\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:      20,\n\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:      33,\n\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\trequest: &ArchiveHistoryRequest{\n\t\t\t\tCloseFailoverVersion: 10,\n\t\t\t\tNextEventID:          34,\n\t\t\t},\n\t\t\tisLast:    true,\n\t\t\tisMutated: false,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.isMutated, IsHistoryMutated(tc.request, tc.historyBatches, tc.isLast, testlogger.New(s.T())))\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/consumer/default_consumer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage consumer\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sort\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nconst (\n\tdefaultShutdownTimeout = 5 * time.Second\n\tdefaultStartWFTimeout  = 3 * time.Second\n\tdefaultConcurrency     = 100\n)\n\ntype DefaultConsumer struct {\n\tqueueID         string\n\tinnerConsumer   messaging.Consumer\n\tlogger          log.Logger\n\tscope           metrics.Scope\n\tfrontendClient  frontend.Client\n\tctx             context.Context\n\tcancelFn        context.CancelFunc\n\twg              sync.WaitGroup\n\tshutdownTimeout time.Duration\n\tstartWFTimeout  time.Duration\n\tmsgDecoder      codec.BinaryEncoder\n\tconcurrency     int\n}\n\ntype Option func(*DefaultConsumer)\n\nfunc WithConcurrency(concurrency int) Option {\n\treturn func(c *DefaultConsumer) {\n\t\tc.concurrency = concurrency\n\t}\n}\n\nfunc New(\n\tqueueID string,\n\tinnerConsumer messaging.Consumer,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tfrontendClient frontend.Client,\n\toptions ...Option,\n) *DefaultConsumer {\n\tctx, cancelFn := context.WithCancel(context.Background())\n\tc := &DefaultConsumer{\n\t\tqueueID:         queueID,\n\t\tinnerConsumer:   innerConsumer,\n\t\tlogger:          logger.WithTags(tag.AsyncWFQueueID(queueID)),\n\t\tscope:           metricsClient.Scope(metrics.AsyncWorkflowConsumerScope),\n\t\tfrontendClient:  frontendClient,\n\t\tctx:             ctx,\n\t\tcancelFn:        cancelFn,\n\t\tshutdownTimeout: defaultShutdownTimeout,\n\t\tstartWFTimeout:  defaultStartWFTimeout,\n\t\tmsgDecoder:      codec.NewThriftRWEncoder(),\n\t\tconcurrency:     defaultConcurrency,\n\t}\n\n\tfor _, opt := range options {\n\t\topt(c)\n\t}\n\n\treturn c\n}\n\nfunc (c *DefaultConsumer) Start() error {\n\tif err := c.innerConsumer.Start(); err != nil {\n\t\treturn err\n\t}\n\n\tfor i := 0; i < c.concurrency; i++ {\n\t\tc.wg.Add(1)\n\t\tgo c.runProcessLoop()\n\t\tc.logger.Info(\"Started process loop\", tag.Counter(i))\n\t}\n\tc.logger.Info(\"Started consumer\", tag.Dynamic(\"concurrency\", c.concurrency))\n\treturn nil\n}\n\nfunc (c *DefaultConsumer) Stop() {\n\tc.logger.Info(\"Stopping consumer\")\n\tc.cancelFn()\n\tc.wg.Wait()\n\tif !common.AwaitWaitGroup(&c.wg, c.shutdownTimeout) {\n\t\tc.logger.Warn(\"Consumer timed out on shutdown\", tag.Dynamic(\"timeout\", c.shutdownTimeout))\n\t\treturn\n\t}\n\n\tc.innerConsumer.Stop()\n\tc.logger.Info(\"Stopped consumer\")\n}\n\nfunc (c *DefaultConsumer) runProcessLoop() {\n\tdefer c.wg.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase msg, ok := <-c.innerConsumer.Messages():\n\t\t\tif !ok {\n\t\t\t\tc.logger.Info(\"Consumer channel closed\")\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tc.processMessage(msg)\n\t\tcase <-c.ctx.Done():\n\t\t\tc.logger.Info(\"Consumer context done so terminating loop\")\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (c *DefaultConsumer) processMessage(msg messaging.Message) {\n\tlogger := c.logger.WithTags(tag.Dynamic(\"partition\", msg.Partition()), tag.Dynamic(\"offset\", msg.Offset()))\n\tlogger.Debug(\"Received message\")\n\n\tsw := c.scope.StartTimer(metrics.AsyncWorkflowProcessMsgLatency)\n\tdefer sw.Stop()\n\n\tvar request sqlblobs.AsyncRequestMessage\n\tif err := c.msgDecoder.Decode(msg.Value(), &request); err != nil {\n\t\tlogger.Error(\"Failed to decode message\", tag.Error(err))\n\t\tc.scope.IncCounter(metrics.AsyncWorkflowFailureCorruptMsgCount)\n\t\tif err := msg.Nack(); err != nil {\n\t\t\tlogger.Error(\"Failed to nack message\", tag.Error(err))\n\t\t}\n\t\treturn\n\t}\n\n\tlogTags, err := c.processRequest(logger, &request)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to process message\", append(logTags, tag.Error(err))...)\n\t\tif nackErr := msg.Nack(); nackErr != nil {\n\t\t\tlogger.Error(\"Failed to nack message\", append(logTags, tag.Dynamic(\"original-error\", err.Error()), tag.Error(nackErr))...)\n\t\t}\n\t\treturn\n\t}\n\n\tlogger = logger.WithTags(logTags...)\n\tif err := msg.Ack(); err != nil {\n\t\tlogger.Error(\"Failed to ack message\", tag.Error(err))\n\t}\n\tlogger.Info(\"Processed message successfully\")\n}\n\nfunc (c *DefaultConsumer) processRequest(logger log.Logger, request *sqlblobs.AsyncRequestMessage) ([]tag.Tag, error) {\n\trequestType := request.GetType().String()\n\tscope := c.scope.Tagged(metrics.AsyncWFRequestTypeTag(requestType))\n\tlogTags := []tag.Tag{tag.AsyncWFRequestType(requestType)}\n\tswitch request.GetType() {\n\tcase sqlblobs.AsyncRequestTypeStartWorkflowExecutionAsyncRequest:\n\t\tstartWFReq, err := c.decodeStartWorkflowRequest(request.GetPayload(), request.GetEncoding())\n\t\tif err != nil {\n\t\t\tscope.IncCounter(metrics.AsyncWorkflowFailureCorruptMsgCount)\n\t\t\treturn logTags, err\n\t\t}\n\n\t\tyarpcCallOpts := getYARPCOptions(request.GetHeader())\n\t\tscope := scope.Tagged(metrics.DomainTag(startWFReq.GetDomain()))\n\t\tlogTags = append(logTags, tag.WorkflowDomainName(startWFReq.GetDomain()), tag.WorkflowID(startWFReq.GetWorkflowID()))\n\n\t\tvar resp *types.StartWorkflowExecutionResponse\n\t\top := func(ctx1 context.Context) error {\n\t\t\tctx, cancel := context.WithTimeout(ctx1, c.startWFTimeout)\n\t\t\tdefer cancel()\n\t\t\tresp, err = c.frontendClient.StartWorkflowExecution(ctx, startWFReq, yarpcCallOpts...)\n\n\t\t\tvar startedError *types.WorkflowExecutionAlreadyStartedError\n\t\t\tif errors.As(err, &startedError) {\n\t\t\t\tlogger.Info(\"Received WorkflowExecutionAlreadyStartedError, treating it as a success\", tag.WorkflowID(startWFReq.GetWorkflowID()), tag.WorkflowRunID(startedError.RunID))\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\tif err := callFrontendWithRetries(c.ctx, op); err != nil {\n\t\t\tscope.IncCounter(metrics.AsyncWorkflowFailureByFrontendCount)\n\t\t\treturn logTags, fmt.Errorf(\"start workflow execution failed after all attempts: %w\", err)\n\t\t}\n\n\t\tlogTags = append(logTags, tag.WorkflowRunID(resp.GetRunID()))\n\t\tscope.IncCounter(metrics.AsyncWorkflowSuccessCount)\n\tcase sqlblobs.AsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest:\n\t\tstartWFReq, err := c.decodeSignalWithStartWorkflowRequest(request.GetPayload(), request.GetEncoding())\n\t\tif err != nil {\n\t\t\tc.scope.IncCounter(metrics.AsyncWorkflowFailureCorruptMsgCount)\n\t\t\treturn logTags, err\n\t\t}\n\n\t\tyarpcCallOpts := getYARPCOptions(request.GetHeader())\n\t\tscope := c.scope.Tagged(metrics.DomainTag(startWFReq.GetDomain()))\n\t\tlogTags = append(logTags, tag.WorkflowDomainName(startWFReq.GetDomain()), tag.WorkflowID(startWFReq.GetWorkflowID()))\n\t\tvar resp *types.StartWorkflowExecutionResponse\n\t\top := func(ctx1 context.Context) error {\n\t\t\tctx, cancel := context.WithTimeout(ctx1, c.startWFTimeout)\n\t\t\tdefer cancel()\n\t\t\tresp, err = c.frontendClient.SignalWithStartWorkflowExecution(ctx, startWFReq, yarpcCallOpts...)\n\n\t\t\tvar startedError *types.WorkflowExecutionAlreadyStartedError\n\t\t\tif errors.As(err, &startedError) {\n\t\t\t\tlogger.Info(\"Received WorkflowExecutionAlreadyStartedError, treating it as a success\", tag.WorkflowID(startWFReq.GetWorkflowID()), tag.WorkflowRunID(startedError.RunID))\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\tif err := callFrontendWithRetries(c.ctx, op); err != nil {\n\t\t\tscope.IncCounter(metrics.AsyncWorkflowFailureByFrontendCount)\n\t\t\treturn logTags, fmt.Errorf(\"signal with start workflow execution failed after all attempts: %w\", err)\n\t\t}\n\n\t\tscope.IncCounter(metrics.AsyncWorkflowSuccessCount)\n\t\tlogTags = append(logTags, tag.WorkflowRunID(resp.GetRunID()))\n\tdefault:\n\t\tc.scope.IncCounter(metrics.AsyncWorkflowFailureCorruptMsgCount)\n\t\treturn logTags, &UnsupportedRequestType{Type: request.GetType()}\n\t}\n\n\treturn logTags, nil\n}\n\nfunc callFrontendWithRetries(ctx context.Context, op func(ctx context.Context) error) error {\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreateFrontendServiceRetryPolicy()),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\n\treturn throttleRetry.Do(ctx, op)\n}\n\nfunc getYARPCOptions(header *shared.Header) []yarpc.CallOption {\n\tif header == nil || header.GetFields() == nil {\n\t\treturn nil\n\t}\n\n\t// sort the header fields to make the tests deterministic\n\tfields := header.GetFields()\n\tsortedKeys := make([]string, 0, len(fields))\n\tfor k := range fields {\n\t\tsortedKeys = append(sortedKeys, k)\n\t}\n\tsort.Strings(sortedKeys)\n\n\tvar opts []yarpc.CallOption\n\tfor _, k := range sortedKeys {\n\t\topts = append(opts, yarpc.WithHeader(k, string(fields[k])))\n\t}\n\treturn opts\n}\n\nfunc (c *DefaultConsumer) decodeStartWorkflowRequest(payload []byte, encoding string) (*types.StartWorkflowExecutionRequest, error) {\n\tif encoding != string(constants.EncodingTypeThriftRW) {\n\t\treturn nil, &UnsupportedEncoding{EncodingType: encoding}\n\t}\n\n\tvar thriftObj shared.StartWorkflowExecutionAsyncRequest\n\tif err := c.msgDecoder.Decode(payload, &thriftObj); err != nil {\n\t\treturn nil, err\n\t}\n\n\tstartRequest := thrift.ToStartWorkflowExecutionAsyncRequest(&thriftObj)\n\treturn startRequest.StartWorkflowExecutionRequest, nil\n}\n\nfunc (c *DefaultConsumer) decodeSignalWithStartWorkflowRequest(payload []byte, encoding string) (*types.SignalWithStartWorkflowExecutionRequest, error) {\n\tif encoding != string(constants.EncodingTypeThriftRW) {\n\t\treturn nil, &UnsupportedEncoding{EncodingType: encoding}\n\t}\n\n\tvar thriftObj shared.SignalWithStartWorkflowExecutionAsyncRequest\n\tif err := c.msgDecoder.Decode(payload, &thriftObj); err != nil {\n\t\treturn nil, err\n\t}\n\n\tsignalWithStartRequest := thrift.ToSignalWithStartWorkflowExecutionAsyncRequest(&thriftObj)\n\treturn signalWithStartRequest.SignalWithStartWorkflowExecutionRequest, nil\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/consumer/default_consumer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage consumer\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nvar (\n\ttestSignalWithStartAsyncReq = &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tSignalWithStartWorkflowExecutionRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:       \"test-domain\",\n\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{Name: \"test-workflow-type\"},\n\t\t\tInput:        []byte(\"test-input\"),\n\t\t\tSignalName:   \"test-signal-name\",\n\t\t},\n\t}\n\n\ttestStartReq = &types.StartWorkflowExecutionAsyncRequest{\n\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:       \"test-domain\",\n\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{Name: \"test-workflow-type\"},\n\t\t\tInput:        []byte(\"test-input\"),\n\t\t},\n\t}\n)\n\ntype fakeMessageConsumer struct {\n\t// input\n\tfailToStart bool\n\tch          chan messaging.Message\n\n\t// output\n\tstopped bool\n}\n\nfunc (f *fakeMessageConsumer) Start() error {\n\tif f.failToStart {\n\t\treturn errors.New(\"failed to start\")\n\t}\n\n\treturn nil\n}\n\nfunc (f *fakeMessageConsumer) Stop() {\n\tf.stopped = true\n}\n\nfunc (f *fakeMessageConsumer) Messages() <-chan messaging.Message {\n\treturn f.ch\n}\n\ntype fakeMessage struct {\n\t// input\n\tval     []byte\n\twantAck bool\n\n\t// output\n\tacked  bool\n\tnacked bool\n}\n\nfunc (m *fakeMessage) Value() []byte {\n\treturn m.val\n}\n\nfunc (m *fakeMessage) Partition() int32 {\n\treturn 0\n}\n\nfunc (m *fakeMessage) Offset() int64 {\n\treturn 0\n}\n\nfunc (m *fakeMessage) Ack() error {\n\tm.acked = true\n\treturn nil\n}\n\nfunc (m *fakeMessage) Nack() error {\n\tm.nacked = true\n\treturn nil\n}\n\nfunc TestDefaultConsumer(t *testing.T) {\n\ttests := []struct {\n\t\tname                         string\n\t\tinnerConsumerFailToStart     bool\n\t\tcloseChanBeforeStop          bool\n\t\tfrontendErr                  error\n\t\texpectStartRequest           bool\n\t\texpectSignalWithStartRequest bool\n\t\tmsgs                         []*fakeMessage\n\t}{\n\t\t{\n\t\t\tname:                     \"failed to start\",\n\t\t\tinnerConsumerFailToStart: true,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid messages\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: []byte(\"invalid payload\"), wantAck: false},\n\t\t\t\t{val: []byte(\"invalid payload 2\"), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"unsupported request type\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateUnsupportedRequestMsg(t), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"startworkflow request with invalid payload content\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, false), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"startworkflowfrontend error\",\n\t\t\tfrontendErr: &types.InternalServiceError{Message: \"oh no\"},\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: false},\n\t\t\t},\n\t\t\texpectStartRequest: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"startworkflowfrontend WorkflowExecutionAlreadyStartedError\",\n\t\t\tfrontendErr: &types.WorkflowExecutionAlreadyStartedError{Message: \"all good, already started\"},\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: true},\n\t\t\t},\n\t\t\texpectStartRequest: true,\n\t\t},\n\t\t{\n\t\t\tname: \"startworkflow unsupported encoding type. json encoding of requests are lossy due to PII masking so it shouldn't be used for async requests\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeJSON, true), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"startworkflow ok\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: true},\n\t\t\t},\n\t\t\texpectStartRequest: true,\n\t\t},\n\t\t{\n\t\t\tname:                \"startworkflow ok with chan closed before stopping\",\n\t\t\tcloseChanBeforeStop: true,\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: true},\n\t\t\t},\n\t\t\texpectStartRequest: true,\n\t\t},\n\t\t// signal with start test cases\n\t\t{\n\t\t\tname: \"signalwithstartworkflow request with invalid payload content\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, false), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"signalwithstartworkflow frontend error\",\n\t\t\tfrontendErr: &types.InternalServiceError{Message: \"oh no\"},\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: false},\n\t\t\t},\n\t\t\texpectSignalWithStartRequest: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"signalwithstartworkflow WorkflowExecutionAlreadyStartedError error\",\n\t\t\tfrontendErr: &types.WorkflowExecutionAlreadyStartedError{Message: \"All good\"},\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: true},\n\t\t\t},\n\t\t\texpectSignalWithStartRequest: true,\n\t\t},\n\t\t{\n\t\t\tname: \"signalwithstartworkflow unsupported encoding type. json encoding of requests are lossy due to PII masking so it shouldn't be used for async requests\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeJSON, true), wantAck: false},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"signalwithstartworkflow ok\",\n\t\t\tmsgs: []*fakeMessage{\n\t\t\t\t{val: mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t, constants.EncodingTypeThriftRW, true), wantAck: true},\n\t\t\t},\n\t\t\texpectSignalWithStartRequest: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tfakeConsumer := &fakeMessageConsumer{\n\t\t\t\tch:          make(chan messaging.Message),\n\t\t\t\tfailToStart: tc.innerConsumerFailToStart,\n\t\t\t}\n\n\t\t\tmockFrontend := frontend.NewMockClient(gomock.NewController(t))\n\t\t\t// we fake 2 headers and pass them manually to the mock because \"...\" extension doesn't work with mocked interface\n\t\t\topts := getYARPCOptions(fakeHeaders())\n\t\t\tif tc.expectStartRequest {\n\t\t\t\tmockFrontend.EXPECT().\n\t\t\t\t\tStartWorkflowExecution(gomock.Any(), gomock.Any(), opts[0], opts[1]).\n\t\t\t\t\tDoAndReturn(func(ctx interface{}, req *types.StartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tif diff := cmp.Diff(testStartReq.StartWorkflowExecutionRequest, req); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif tc.frontendErr != nil {\n\t\t\t\t\t\t\treturn nil, tc.frontendErr\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: \"test-run-id\"}, nil\n\t\t\t\t\t}).MinTimes(1)\n\t\t\t}\n\t\t\tif tc.expectSignalWithStartRequest {\n\t\t\t\tmockFrontend.EXPECT().\n\t\t\t\t\tSignalWithStartWorkflowExecution(gomock.Any(), gomock.Any(), opts[0], opts[1]).\n\t\t\t\t\tDoAndReturn(func(ctx interface{}, req *types.SignalWithStartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tif diff := cmp.Diff(testSignalWithStartAsyncReq.SignalWithStartWorkflowExecutionRequest, req); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif tc.frontendErr != nil {\n\t\t\t\t\t\t\treturn nil, tc.frontendErr\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: \"test-run-id\"}, nil\n\t\t\t\t\t}).MinTimes(1)\n\t\t\t}\n\n\t\t\tc := New(\"queueid1\", fakeConsumer, testlogger.New(t), metrics.NewNoopMetricsClient(), mockFrontend, WithConcurrency(2))\n\t\t\terr := c.Start()\n\t\t\tif tc.innerConsumerFailToStart != (err != nil) {\n\t\t\t\tt.Errorf(\"Start() err: %v, wantErr: %v\", err, tc.innerConsumerFailToStart)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tfor _, msg := range tc.msgs {\n\t\t\t\tfakeConsumer.ch <- msg\n\t\t\t}\n\n\t\t\tif tc.closeChanBeforeStop {\n\t\t\t\tclose(fakeConsumer.ch)\n\t\t\t}\n\n\t\t\tc.Stop()\n\t\t\tif !fakeConsumer.stopped {\n\t\t\t\tt.Error(\"innerConsumer.Stop() not called\")\n\t\t\t}\n\n\t\t\tfor i, msg := range tc.msgs {\n\t\t\t\tif msg.wantAck && !msg.acked {\n\t\t\t\t\tt.Errorf(\"message %d not acked\", i)\n\t\t\t\t}\n\t\t\t\tif !msg.wantAck && !msg.nacked {\n\t\t\t\t\tt.Errorf(\"message %d not nacked\", i)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mustGenerateStartWorkflowExecutionRequestMsg(t *testing.T, encodingType constants.EncodingType, validPayload bool) []byte {\n\tencoder := codec.NewThriftRWEncoder()\n\tpayload, err := encoder.Encode(thrift.FromStartWorkflowExecutionAsyncRequest(testStartReq))\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\tif !validPayload {\n\t\tpayload = []byte(\"invalid payload\")\n\t}\n\n\tmsg := &sqlblobs.AsyncRequestMessage{\n\t\tType:     sqlblobs.AsyncRequestTypeStartWorkflowExecutionAsyncRequest.Ptr(),\n\t\tHeader:   fakeHeaders(),\n\t\tEncoding: common.StringPtr(string(encodingType)),\n\t\tPayload:  payload,\n\t}\n\n\tres, err := codec.NewThriftRWEncoder().Encode(msg)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\treturn res\n}\n\nfunc mustGenerateSignalWithStartWorkflowExecutionRequestMsg(t *testing.T, encodingType constants.EncodingType, validPayload bool) []byte {\n\tencoder := codec.NewThriftRWEncoder()\n\tpayload, err := encoder.Encode(thrift.FromSignalWithStartWorkflowExecutionAsyncRequest(testSignalWithStartAsyncReq))\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\tif !validPayload {\n\t\tpayload = []byte(\"invalid payload\")\n\t}\n\n\tmsg := &sqlblobs.AsyncRequestMessage{\n\t\tType:     sqlblobs.AsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest.Ptr(),\n\t\tHeader:   fakeHeaders(),\n\t\tEncoding: common.StringPtr(string(encodingType)),\n\t\tPayload:  payload,\n\t}\n\n\tres, err := codec.NewThriftRWEncoder().Encode(msg)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\treturn res\n}\n\nfunc mustGenerateUnsupportedRequestMsg(t *testing.T) []byte {\n\tencoder := codec.NewThriftRWEncoder()\n\tpayload, err := encoder.Encode(thrift.FromStartWorkflowExecutionAsyncRequest(testStartReq))\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttp := sqlblobs.AsyncRequestType(-1)\n\tmsg := &sqlblobs.AsyncRequestMessage{\n\t\tType:     &tp,\n\t\tHeader:   &shared.Header{},\n\t\tEncoding: common.StringPtr(string(constants.EncodingTypeJSON)),\n\t\tPayload:  payload,\n\t}\n\n\tres, err := codec.NewThriftRWEncoder().Encode(msg)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\treturn res\n}\n\nfunc fakeHeaders() *shared.Header {\n\treturn &shared.Header{\n\t\tFields: map[string][]byte{\n\t\t\t\"key1\": []byte(\"val1\"),\n\t\t\t\"key2\": []byte(\"val2\"),\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/consumer/errors.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage consumer\n\nimport \"github.com/uber/cadence/.gen/go/sqlblobs\"\n\ntype UnsupportedRequestType struct {\n\tType sqlblobs.AsyncRequestType\n}\n\nfunc (e *UnsupportedRequestType) Error() string {\n\treturn \"unsupported request type: \" + e.Type.String()\n}\n\ntype UnsupportedEncoding struct {\n\tEncodingType string\n}\n\nfunc (e *UnsupportedEncoding) Error() string {\n\treturn \"unsupported encoding: \" + e.EncodingType\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queue\n\npackage queue\n\nimport (\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// Provider is used to get a queue\n\tProvider interface {\n\t\tGetPredefinedQueue(string) (provider.Queue, error)\n\t\tGetQueue(string, *types.DataBlob) (provider.Queue, error)\n\t}\n)\n"
  },
  {
    "path": "common/asyncworkflow/queue/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package queue -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queue\n//\n\n// Package queue is a generated GoMock package.\npackage queue\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tprovider \"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockProvider is a mock of Provider interface.\ntype MockProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockProviderMockRecorder is the mock recorder for MockProvider.\ntype MockProviderMockRecorder struct {\n\tmock *MockProvider\n}\n\n// NewMockProvider creates a new mock instance.\nfunc NewMockProvider(ctrl *gomock.Controller) *MockProvider {\n\tmock := &MockProvider{ctrl: ctrl}\n\tmock.recorder = &MockProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProvider) EXPECT() *MockProviderMockRecorder {\n\treturn m.recorder\n}\n\n// GetPredefinedQueue mocks base method.\nfunc (m *MockProvider) GetPredefinedQueue(arg0 string) (provider.Queue, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPredefinedQueue\", arg0)\n\tret0, _ := ret[0].(provider.Queue)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetPredefinedQueue indicates an expected call of GetPredefinedQueue.\nfunc (mr *MockProviderMockRecorder) GetPredefinedQueue(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPredefinedQueue\", reflect.TypeOf((*MockProvider)(nil).GetPredefinedQueue), arg0)\n}\n\n// GetQueue mocks base method.\nfunc (m *MockProvider) GetQueue(arg0 string, arg1 *types.DataBlob) (provider.Queue, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueue\", arg0, arg1)\n\tret0, _ := ret[0].(provider.Queue)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueue indicates an expected call of GetQueue.\nfunc (mr *MockProviderMockRecorder) GetQueue(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueue\", reflect.TypeOf((*MockProvider)(nil).GetQueue), arg0, arg1)\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/config.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/IBM/sarama\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n)\n\ntype (\n\tqueueConfig struct {\n\t\tConnection connectionConfig `yaml:\"connection\"`\n\t\tTopic      string           `yaml:\"topic\"`\n\t}\n\n\tconnectionConfig struct {\n\t\tBrokers []string    `yaml:\"brokers\"`\n\t\tTLS     config.TLS  `yaml:\"tls\"`\n\t\tSASL    config.SASL `yaml:\"sasl\"`\n\t}\n)\n\nfunc (c *queueConfig) ID() string {\n\treturn fmt.Sprintf(\"kafka::%s/%s\", c.Topic, strings.Join(c.Connection.Brokers, \",\"))\n}\n\nfunc newSaramaConfigWithAuth(tls *config.TLS, sasl *config.SASL) (*sarama.Config, error) {\n\tsaramaConfig := sarama.NewConfig()\n\n\t// TLS support\n\ttlsConfig, err := tls.ToTLSConfig()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error creating Kafka TLS config %w\", err)\n\t}\n\tif tlsConfig != nil {\n\t\tsaramaConfig.Net.TLS.Enable = true\n\t\tsaramaConfig.Net.TLS.Config = tlsConfig\n\t}\n\n\t// SASL support\n\tif sasl.Enabled {\n\t\tsaramaConfig.Net.SASL.Enable = true\n\t\tsaramaConfig.Net.SASL.Handshake = true\n\t\tsaramaConfig.Net.SASL.User = sasl.User\n\t\tsaramaConfig.Net.SASL.Password = sasl.Password\n\t\tswitch sasl.Algorithm {\n\t\tcase \"sha512\":\n\t\t\tsaramaConfig.Net.SASL.SCRAMClientGeneratorFunc = func() sarama.SCRAMClient {\n\t\t\t\treturn &authorization.XDGSCRAMClient{HashGeneratorFcn: authorization.SHA512}\n\t\t\t}\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypeSCRAMSHA512\n\t\tcase \"sha256\":\n\t\t\tsaramaConfig.Net.SASL.SCRAMClientGeneratorFunc = func() sarama.SCRAMClient {\n\t\t\t\treturn &authorization.XDGSCRAMClient{HashGeneratorFcn: authorization.SHA256}\n\t\t\t}\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypeSCRAMSHA256\n\t\tcase \"plain\":\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypePlaintext\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"unknown SASL algorithm %v\", sasl.Algorithm)\n\t\t}\n\t}\n\treturn saramaConfig, nil\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/config_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport \"testing\"\n\nfunc TestQueueConfigID(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tconfig   queueConfig\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname: \"single broker\",\n\t\t\tconfig: queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"broker1:9092\"},\n\t\t\t\t},\n\t\t\t\tTopic: \"topic1\",\n\t\t\t},\n\t\t\texpected: \"kafka::topic1/broker1:9092\",\n\t\t},\n\t\t{\n\t\t\tname: \"multiple brokers\",\n\t\t\tconfig: queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"broker1:9092\", \"broker2:9092\"},\n\t\t\t\t},\n\t\t\t\tTopic: \"topic2\",\n\t\t\t},\n\t\t\texpected: \"kafka::topic2/broker1:9092,broker2:9092\",\n\t\t},\n\t\t{\n\t\t\tname: \"no brokers\",\n\t\t\tconfig: queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{},\n\t\t\t\t},\n\t\t\t\tTopic: \"topic3\",\n\t\t\t},\n\t\t\texpected: \"kafka::topic3/\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty topic\",\n\t\t\tconfig: queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"broker1:9092\"},\n\t\t\t\t},\n\t\t\t\tTopic: \"\",\n\t\t\t},\n\t\t\texpected: \"kafka::/broker1:9092\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.config.ID()\n\t\t\tif got != tt.expected {\n\t\t\t\tt.Errorf(\"queueConfig.ID() = %v, want %v\", got, tt.expected)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/decoder.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdecoderImpl struct {\n\t\tblob *types.DataBlob\n\t}\n)\n\nfunc newDecoder(blob *types.DataBlob) provider.Decoder {\n\treturn &decoderImpl{\n\t\tblob: blob,\n\t}\n}\n\nfunc (d *decoderImpl) Decode(out any) error {\n\tif d.blob.GetEncodingType() != types.EncodingTypeJSON {\n\t\treturn fmt.Errorf(\"unsupported encoding type %v\", d.blob.GetEncodingType())\n\t}\n\treturn json.Unmarshal(d.blob.Data, out)\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/decoder_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDecode(t *testing.T) {\n\ttype testStruct struct {\n\t\tName string `json:\"name\"`\n\t}\n\n\ttests := []struct {\n\t\tname           string\n\t\tblob           *types.DataBlob\n\t\twant           *testStruct\n\t\twantErr        bool\n\t\texpectedErrMsg string\n\t}{\n\t\t{\n\t\t\tname: \"valid JSON encoding\",\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tData:         []byte(`{\"name\":\"test\"}`),\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t},\n\t\t\twant:    &testStruct{Name: \"test\"},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"unsupported encoding type\",\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tData:         []byte(\"aa\"),\n\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t},\n\t\t\twant:           nil,\n\t\t\twantErr:        true,\n\t\t\texpectedErrMsg: \"unsupported encoding type\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tdecoder := newDecoder(tt.blob)\n\t\t\tvar got testStruct\n\t\t\terr := decoder.Decode(&got)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedErrMsg)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, &got)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/init.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n)\n\nfunc init() {\n\tmust := func(err error) {\n\t\tif err != nil {\n\t\t\tpanic(fmt.Errorf(\"failed to register default provider: %w\", err))\n\t\t}\n\t}\n\tmust(provider.RegisterQueueProvider(\"kafka\", newQueue))\n\tmust(provider.RegisterDecoder(\"kafka\", newDecoder))\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/queue.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"time\"\n\n\t\"github.com/IBM/sarama\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/consumer\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/messaging/kafka\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tqueueImpl struct {\n\t\tconfig *queueConfig\n\t}\n)\n\nfunc newQueue(decoder provider.Decoder) (provider.Queue, error) {\n\tvar out queueConfig\n\tif err := decoder.Decode(&out); err != nil {\n\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t}\n\tsort.Strings(out.Connection.Brokers)\n\treturn &queueImpl{\n\t\tconfig: &out,\n\t}, nil\n}\n\nfunc (q *queueImpl) ID() string {\n\treturn q.config.ID()\n}\n\nfunc (q *queueImpl) CreateConsumer(p *provider.Params) (provider.Consumer, error) {\n\tconsumerGroup := fmt.Sprintf(\"%s-asyncwf-consumer\", q.config.Topic)\n\tdlqTopic := fmt.Sprintf(\"%s-dlq\", q.config.Topic)\n\tdlqConfig, err := newSaramaConfigWithAuth(&q.config.Connection.TLS, &q.config.Connection.SASL)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create kafka sarama config: %w\", err)\n\t}\n\tdlqConfig.Producer.Return.Successes = true\n\tdlqProducer, err := newProducer(dlqTopic, q.config.Connection.Brokers, dlqConfig, p.MetricsClient, p.Logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create kafka producer for dlq: %w\", err)\n\t}\n\n\tconsumerConfig, err := newSaramaConfigWithAuth(&q.config.Connection.TLS, &q.config.Connection.SASL)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create kafka sarama config: %w\", err)\n\t}\n\n\tconsumerConfig.Consumer.Fetch.Default = 30 * 1024 * 1024 // 30MB.\n\tconsumerConfig.Consumer.Return.Errors = true\n\tconsumerConfig.Consumer.Offsets.AutoCommit.Enable = false // Use manual commit\n\tconsumerConfig.Consumer.Offsets.Initial = sarama.OffsetOldest\n\tconsumerConfig.Consumer.MaxProcessingTime = 250 * time.Millisecond\n\tkafkaConsumer, err := kafka.NewKafkaConsumer(dlqProducer, q.config.Connection.Brokers, q.config.Topic, consumerGroup, consumerConfig, p.MetricsClient, p.Logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create kafka consumer: %w\", err)\n\t}\n\tp.Logger.Info(\"Creating async wf consumer\", tag.KafkaTopicName(q.config.Topic))\n\treturn consumer.New(q.ID(), kafkaConsumer, p.Logger, p.MetricsClient, p.FrontendClient), nil\n}\n\nfunc (q *queueImpl) CreateProducer(p *provider.Params) (messaging.Producer, error) {\n\tconfig, err := newSaramaConfigWithAuth(&q.config.Connection.TLS, &q.config.Connection.SASL)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tconfig.Producer.Return.Successes = true\n\tp.Logger.Info(\"Creating async wf producer\", tag.KafkaTopicName(q.config.Topic))\n\treturn newProducer(q.config.Topic, q.config.Connection.Brokers, config, p.MetricsClient, p.Logger)\n}\n\nfunc newProducer(topic string, brokers []string, saramaConfig *sarama.Config, metricsClient metrics.Client, logger log.Logger) (messaging.Producer, error) {\n\tp, err := sarama.NewSyncProducer(brokers, saramaConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\twithMetricsOpt := messaging.WithMetricTags(metrics.TopicTag(topic))\n\treturn messaging.NewMetricProducer(kafka.NewKafkaProducer(topic, p, logger), metricsClient, withMetricsOpt), nil\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/kafka/queue_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/IBM/sarama\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype MockDecoder struct {\n\tDecodeFunc func(v any) error\n}\n\nfunc (m *MockDecoder) Decode(v any) error {\n\treturn m.DecodeFunc(v)\n}\n\nfunc TestNewQueue(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tdecoder   *MockDecoder\n\t\twant      *queueImpl\n\t\twantErr   bool\n\t\terrString string\n\t}{\n\t\t{\n\t\t\tname: \"successful decoding\",\n\t\t\tdecoder: &MockDecoder{\n\t\t\t\tDecodeFunc: func(v any) error {\n\t\t\t\t\tout := v.(*queueConfig)\n\t\t\t\t\tout.Connection.Brokers = []string{\"broker2\", \"broker1\"}\n\t\t\t\t\treturn nil\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &queueImpl{\n\t\t\t\tconfig: &queueConfig{\n\t\t\t\t\tConnection: connectionConfig{Brokers: []string{\"broker1\", \"broker2\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"decoding failure\",\n\t\t\tdecoder: &MockDecoder{\n\t\t\t\tDecodeFunc: func(v any) error {\n\t\t\t\t\treturn errors.New(\"decoding error\")\n\t\t\t\t},\n\t\t\t},\n\t\t\twant:      nil,\n\t\t\twantErr:   true,\n\t\t\terrString: \"bad config: decoding error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, err := newQueue(tt.decoder)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateConsumer(t *testing.T) {\n\ttestCases := []struct {\n\t\tname    string\n\t\tconfig  *queueConfig\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tconfig: &queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"localhost:9092\"},\n\t\t\t\t\tTLS: config.TLS{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t\tSASL: config.SASL{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopic: \"test-topic\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Invalid SASL config\",\n\t\t\tconfig: &queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"localhost:9092\"},\n\t\t\t\t\tTLS: config.TLS{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t\tSASL: config.SASL{\n\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\tAlgorithm: \"test\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopic: \"test-topic\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockBroker := sarama.NewMockBroker(t, 0)\n\t\t\tdefer mockBroker.Close()\n\t\t\tmockBroker.SetHandlerByMap(map[string]sarama.MockResponse{\n\t\t\t\t\"MetadataRequest\": sarama.NewMockMetadataResponse(t).\n\t\t\t\t\tSetBroker(mockBroker.Addr(), mockBroker.BrokerID()).\n\t\t\t\t\tSetLeader(\"test-topic\", 0, mockBroker.BrokerID()).\n\t\t\t\t\tSetController(mockBroker.BrokerID()),\n\t\t\t})\n\t\t\ttc.config.Connection.Brokers = []string{mockBroker.Addr()}\n\n\t\t\tq := &queueImpl{\n\t\t\t\tconfig: tc.config,\n\t\t\t}\n\t\t\tp := &provider.Params{\n\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t}\n\t\t\tconsumer, err := q.CreateConsumer(p)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, consumer)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateProducer(t *testing.T) {\n\ttestCases := []struct {\n\t\tname    string\n\t\tconfig  *queueConfig\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tconfig: &queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"localhost:9092\"},\n\t\t\t\t\tTLS: config.TLS{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t\tSASL: config.SASL{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopic: \"test-topic\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Invalid SASL config\",\n\t\t\tconfig: &queueConfig{\n\t\t\t\tConnection: connectionConfig{\n\t\t\t\t\tBrokers: []string{\"localhost:9092\"},\n\t\t\t\t\tTLS: config.TLS{\n\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t},\n\t\t\t\t\tSASL: config.SASL{\n\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\tAlgorithm: \"test\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopic: \"test-topic\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockBroker := sarama.NewMockBroker(t, 0)\n\t\t\tdefer mockBroker.Close()\n\t\t\tmockBroker.SetHandlerByMap(map[string]sarama.MockResponse{\n\t\t\t\t\"MetadataRequest\": sarama.NewMockMetadataResponse(t).\n\t\t\t\t\tSetBroker(mockBroker.Addr(), mockBroker.BrokerID()).\n\t\t\t\t\tSetLeader(\"test-topic\", 0, mockBroker.BrokerID()).\n\t\t\t\t\tSetController(mockBroker.BrokerID()),\n\t\t\t})\n\t\t\ttc.config.Connection.Brokers = []string{mockBroker.Addr()}\n\n\t\t\tq := &queueImpl{\n\t\t\t\tconfig: tc.config,\n\t\t\t}\n\t\t\tp := &provider.Params{\n\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t}\n\t\t\tconsumer, err := q.CreateProducer(p)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, consumer)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/provider/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: provider.go\n//\n// Generated by this command:\n//\n//\tmockgen -package provider -source provider.go -destination interface_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queue/provider\n//\n\n// Package provider is a generated GoMock package.\npackage provider\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tmessaging \"github.com/uber/cadence/common/messaging\"\n)\n\n// MockDecoder is a mock of Decoder interface.\ntype MockDecoder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDecoderMockRecorder\n\tisgomock struct{}\n}\n\n// MockDecoderMockRecorder is the mock recorder for MockDecoder.\ntype MockDecoderMockRecorder struct {\n\tmock *MockDecoder\n}\n\n// NewMockDecoder creates a new mock instance.\nfunc NewMockDecoder(ctrl *gomock.Controller) *MockDecoder {\n\tmock := &MockDecoder{ctrl: ctrl}\n\tmock.recorder = &MockDecoderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDecoder) EXPECT() *MockDecoderMockRecorder {\n\treturn m.recorder\n}\n\n// Decode mocks base method.\nfunc (m *MockDecoder) Decode(arg0 any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Decode\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Decode indicates an expected call of Decode.\nfunc (mr *MockDecoderMockRecorder) Decode(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Decode\", reflect.TypeOf((*MockDecoder)(nil).Decode), arg0)\n}\n\n// MockConsumer is a mock of Consumer interface.\ntype MockConsumer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConsumerMockRecorder\n\tisgomock struct{}\n}\n\n// MockConsumerMockRecorder is the mock recorder for MockConsumer.\ntype MockConsumerMockRecorder struct {\n\tmock *MockConsumer\n}\n\n// NewMockConsumer creates a new mock instance.\nfunc NewMockConsumer(ctrl *gomock.Controller) *MockConsumer {\n\tmock := &MockConsumer{ctrl: ctrl}\n\tmock.recorder = &MockConsumerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConsumer) EXPECT() *MockConsumerMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockConsumer) Start() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockConsumerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockConsumer)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockConsumer) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockConsumerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockConsumer)(nil).Stop))\n}\n\n// MockQueue is a mock of Queue interface.\ntype MockQueue struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueueMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueueMockRecorder is the mock recorder for MockQueue.\ntype MockQueueMockRecorder struct {\n\tmock *MockQueue\n}\n\n// NewMockQueue creates a new mock instance.\nfunc NewMockQueue(ctrl *gomock.Controller) *MockQueue {\n\tmock := &MockQueue{ctrl: ctrl}\n\tmock.recorder = &MockQueueMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueue) EXPECT() *MockQueueMockRecorder {\n\treturn m.recorder\n}\n\n// CreateConsumer mocks base method.\nfunc (m *MockQueue) CreateConsumer(arg0 *Params) (Consumer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateConsumer\", arg0)\n\tret0, _ := ret[0].(Consumer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateConsumer indicates an expected call of CreateConsumer.\nfunc (mr *MockQueueMockRecorder) CreateConsumer(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateConsumer\", reflect.TypeOf((*MockQueue)(nil).CreateConsumer), arg0)\n}\n\n// CreateProducer mocks base method.\nfunc (m *MockQueue) CreateProducer(arg0 *Params) (messaging.Producer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateProducer\", arg0)\n\tret0, _ := ret[0].(messaging.Producer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateProducer indicates an expected call of CreateProducer.\nfunc (mr *MockQueueMockRecorder) CreateProducer(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateProducer\", reflect.TypeOf((*MockQueue)(nil).CreateProducer), arg0)\n}\n\n// ID mocks base method.\nfunc (m *MockQueue) ID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// ID indicates an expected call of ID.\nfunc (mr *MockQueueMockRecorder) ID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ID\", reflect.TypeOf((*MockQueue)(nil).ID))\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/provider/provider.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queue/provider\n\npackage provider\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/syncmap\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tParams struct {\n\t\tLogger         log.Logger\n\t\tMetricsClient  metrics.Client\n\t\tFrontendClient frontend.Client\n\t}\n\n\tDecoder interface {\n\t\tDecode(any) error\n\t}\n\n\tConsumer interface {\n\t\tStart() error\n\t\tStop()\n\t}\n\n\t// Queue is an interface for async queue\n\tQueue interface {\n\t\tID() string\n\t\tCreateConsumer(*Params) (Consumer, error)\n\t\tCreateProducer(*Params) (messaging.Producer, error)\n\t}\n\n\tQueueConstructor func(Decoder) (Queue, error)\n\n\tDecoderConstructor func(*types.DataBlob) Decoder\n)\n\nvar (\n\tqueueConstructors   = syncmap.New[string, QueueConstructor]()\n\tdecoderConstructors = syncmap.New[string, DecoderConstructor]()\n)\n\n// RegisterQueueProvider registers a queue constructor for a given queue type\nfunc RegisterQueueProvider(queueType string, queueConstructor QueueConstructor) error {\n\tinserted := queueConstructors.Put(queueType, queueConstructor)\n\tif !inserted {\n\t\treturn fmt.Errorf(\"queue type %v already registered\", queueType)\n\t}\n\treturn nil\n}\n\n// GetQueueProvider returns a queue constructor for a given queue type\nfunc GetQueueProvider(queueType string) (QueueConstructor, bool) {\n\treturn queueConstructors.Get(queueType)\n}\n\n// RegisterDecoder registers a decoder constructor for a given queue type\nfunc RegisterDecoder(queueType string, decoderConstructor DecoderConstructor) error {\n\tinserted := decoderConstructors.Put(queueType, decoderConstructor)\n\tif !inserted {\n\t\treturn fmt.Errorf(\"decoder type %v already registered\", queueType)\n\t}\n\treturn nil\n}\n\n// GetDecoder returns a decoder constructor for a given queue type\nfunc GetDecoder(queueType string) (DecoderConstructor, bool) {\n\treturn decoderConstructors.Get(queueType)\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/provider/provider_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage provider\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestQueueProvider(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType string\n\t\tsetup     func()\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: \"q1\",\n\t\t\twantErr:   false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Duplicate type\",\n\t\t\tqueueType: \"q2\",\n\t\t\tsetup: func() {\n\t\t\t\tRegisterQueueProvider(\"q2\", func(Decoder) (Queue, error) {\n\t\t\t\t\treturn nil, nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, ok := GetQueueProvider(tt.queueType)\n\t\t\tassert.False(t, ok)\n\n\t\t\tif tt.setup != nil {\n\t\t\t\ttt.setup()\n\t\t\t}\n\n\t\t\terr := RegisterQueueProvider(tt.queueType, func(Decoder) (Queue, error) {\n\t\t\t\treturn nil, nil\n\t\t\t})\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\t_, ok = GetQueueProvider(tt.queueType)\n\t\t\tassert.True(t, ok)\n\t\t})\n\t}\n}\n\nfunc TestDecoder(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType string\n\t\tsetup     func()\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: \"q1\",\n\t\t\twantErr:   false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Duplicate type\",\n\t\t\tqueueType: \"q2\",\n\t\t\tsetup: func() {\n\t\t\t\tRegisterDecoder(\"q2\", func(*types.DataBlob) Decoder {\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, ok := GetDecoder(tt.queueType)\n\t\t\tassert.False(t, ok)\n\n\t\t\tif tt.setup != nil {\n\t\t\t\ttt.setup()\n\t\t\t}\n\n\t\t\terr := RegisterDecoder(tt.queueType, func(*types.DataBlob) Decoder {\n\t\t\t\treturn nil\n\t\t\t})\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\t_, ok = GetDecoder(tt.queueType)\n\t\t\tassert.True(t, ok)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/provider.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tproviderImpl struct {\n\t\tqueues map[string]provider.Queue\n\t}\n)\n\n// NewAsyncQueueProvider returns a new async queue provider\nfunc NewAsyncQueueProvider(cfg map[string]config.AsyncWorkflowQueueProvider) (Provider, error) {\n\tp := &providerImpl{\n\t\tqueues: make(map[string]provider.Queue),\n\t}\n\tfor queueName, queueCfg := range cfg {\n\t\tqueueConstructor, ok := provider.GetQueueProvider(queueCfg.Type)\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(\"queue type %v not registered\", queueCfg.Type)\n\t\t}\n\t\tqueue, err := queueConstructor(queueCfg.Config)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tp.queues[queueName] = queue\n\t}\n\treturn p, nil\n}\n\nfunc (p *providerImpl) GetPredefinedQueue(name string) (provider.Queue, error) {\n\tqueue, ok := p.queues[name]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"queue %v not found\", name)\n\t}\n\treturn queue, nil\n}\n\nfunc (p *providerImpl) GetQueue(queueType string, queueConfig *types.DataBlob) (provider.Queue, error) {\n\tqueueConfigDecoder, ok := provider.GetDecoder(queueType)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"queue type %v not registered\", queueType)\n\t}\n\tdecoder := queueConfigDecoder(queueConfig)\n\tqueueConstructor, ok := provider.GetQueueProvider(queueType)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"queue type %v not registered\", queueType)\n\t}\n\treturn queueConstructor(decoder)\n}\n"
  },
  {
    "path": "common/asyncworkflow/queue/provider_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc mockQueueConstructor(cfg provider.Decoder) (provider.Queue, error) {\n\t// Mock implementation\n\treturn nil, nil\n}\n\nfunc mockDecoderConstructor(cfg *types.DataBlob) provider.Decoder {\n\t// Mock implementation\n\treturn nil\n}\n\nfunc TestNewAsyncQueueProvider(t *testing.T) {\n\t// Mock the provider registration functions\n\tprovider.RegisterQueueProvider(\"validType\", mockQueueConstructor)\n\tprovider.RegisterDecoder(\"validType\", mockDecoderConstructor)\n\n\ttests := []struct {\n\t\tname          string\n\t\tcfg           map[string]config.AsyncWorkflowQueueProvider\n\t\texpectError   bool\n\t\terrorContains string\n\t}{\n\t\t{\n\t\t\tname: \"Successful Initialization\",\n\t\t\tcfg: map[string]config.AsyncWorkflowQueueProvider{\n\t\t\t\t\"testQueue\": {Type: \"validType\", Config: &config.YamlNode{}},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Unregistered Queue Type\",\n\t\t\tcfg: map[string]config.AsyncWorkflowQueueProvider{\n\t\t\t\t\"testQueue\": {Type: \"invalidType\", Config: &config.YamlNode{}},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"not registered\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, err := NewAsyncQueueProvider(tt.cfg)\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tt.errorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tt.errorContains)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetPredefinedQueue(t *testing.T) {\n\tp := &providerImpl{\n\t\tqueues: map[string]provider.Queue{\n\t\t\t\"testQueue\": nil,\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname          string\n\t\tqueueName     string\n\t\texpectError   bool\n\t\terrorContains string\n\t}{\n\t\t{\n\t\t\tname:        \"Successful Get\",\n\t\t\tqueueName:   \"testQueue\",\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Queue Not Found\",\n\t\t\tqueueName:     \"invalidQueue\",\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"not found\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, err := p.GetPredefinedQueue(tt.queueName)\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tt.errorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tt.errorContains)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetQueue(t *testing.T) {\n\t// Mock the provider registration functions\n\tprovider.RegisterQueueProvider(\"validType\", mockQueueConstructor)\n\tprovider.RegisterDecoder(\"validType\", mockDecoderConstructor)\n\n\tp := &providerImpl{\n\t\tqueues: map[string]provider.Queue{},\n\t}\n\n\ttests := []struct {\n\t\tname          string\n\t\tqueueType     string\n\t\tqueueConfig   *types.DataBlob\n\t\texpectError   bool\n\t\terrorContains string\n\t}{\n\t\t{\n\t\t\tname:        \"Successful Get\",\n\t\t\tqueueType:   \"validType\",\n\t\t\tqueueConfig: &types.DataBlob{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Unregistered Queue Type\",\n\t\t\tqueueType:     \"invalidType\",\n\t\t\tqueueConfig:   &types.DataBlob{},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"not registered\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, err := p.GetQueue(tt.queueType, tt.queueConfig)\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tt.errorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tt.errorContains)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queueconfigapi/handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queueconfigapi\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype handlerImpl struct {\n\tlogger        log.Logger\n\tdomainHandler domain.Handler\n}\n\nfunc New(logger log.Logger, dh domain.Handler) Handler {\n\treturn &handlerImpl{\n\t\tlogger:        logger,\n\t\tdomainHandler: dh,\n\t}\n}\n\nfunc (h *handlerImpl) GetConfiguraton(ctx context.Context, req *types.GetDomainAsyncWorkflowConfiguratonRequest) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tresp, err := h.domainHandler.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: &req.Domain,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif resp == nil || resp.Configuration == nil || resp.Configuration.AsyncWorkflowConfig == nil {\n\t\treturn &types.GetDomainAsyncWorkflowConfiguratonResponse{}, nil\n\t}\n\n\treturn &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\tConfiguration: resp.Configuration.AsyncWorkflowConfig,\n\t}, nil\n}\n\nfunc (h *handlerImpl) UpdateConfiguration(ctx context.Context, req *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tif req == nil {\n\t\treturn nil, &types.BadRequestError{Message: \"Request is nil.\"}\n\t}\n\n\terr := h.domainHandler.UpdateAsyncWorkflowConfiguraton(ctx, *req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &types.UpdateDomainAsyncWorkflowConfiguratonResponse{}, nil\n}\n"
  },
  {
    "path": "common/asyncworkflow/queueconfigapi/handler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package queueconfigapi -source interface.go -destination handler_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queueconfigapi\n//\n\n// Package queueconfigapi is a generated GoMock package.\npackage queueconfigapi\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// GetConfiguraton mocks base method.\nfunc (m *MockHandler) GetConfiguraton(arg0 context.Context, arg1 *types.GetDomainAsyncWorkflowConfiguratonRequest) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetConfiguraton\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetConfiguraton indicates an expected call of GetConfiguraton.\nfunc (mr *MockHandlerMockRecorder) GetConfiguraton(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetConfiguraton\", reflect.TypeOf((*MockHandler)(nil).GetConfiguraton), arg0, arg1)\n}\n\n// UpdateConfiguration mocks base method.\nfunc (m *MockHandler) UpdateConfiguration(arg0 context.Context, arg1 *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateConfiguration\", arg0, arg1)\n\tret0, _ := ret[0].(*types.UpdateDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateConfiguration indicates an expected call of UpdateConfiguration.\nfunc (mr *MockHandlerMockRecorder) UpdateConfiguration(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateConfiguration\", reflect.TypeOf((*MockHandler)(nil).UpdateConfiguration), arg0, arg1)\n}\n"
  },
  {
    "path": "common/asyncworkflow/queueconfigapi/handler_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queueconfigapi\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetConfiguraton(t *testing.T) {\n\ttests := map[string]struct {\n\t\treq                 *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t\tdomainHandlerMockFn func(*domain.MockHandler)\n\t\twantResp            *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t\twantErr             bool\n\t}{\n\t\t\"Domain handler fails to DescribeDomain\": {\n\t\t\treq: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tm.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"Domain config is nil\": {\n\t\t\treq: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tresp := &types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: nil,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"Domain async wf config is nil\": {\n\t\t\treq: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tresp := &types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tAsyncWorkflowConfig: nil,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tm.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"Success\": {\n\t\t\treq: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tresp := &types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tAsyncWorkflowConfig: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t\t\t\tQueueType:           \"kafka\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker\"]}`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tm.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t\tQueueType:           \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker\"]}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdomainHandlerMock := domain.NewMockHandler(ctrl)\n\t\t\tif tc.domainHandlerMockFn != nil {\n\t\t\t\ttc.domainHandlerMockFn(domainHandlerMock)\n\t\t\t}\n\n\t\t\thandler := New(testlogger.New(t), domainHandlerMock)\n\t\t\tresp, err := handler.GetConfiguraton(context.Background(), tc.req)\n\n\t\t\tif tc.wantErr != (err != nil) {\n\t\t\t\tt.Fatalf(\"Error mismatch. Got: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tc.wantResp, resp)\n\t\t})\n\t}\n}\n\nfunc TestUpdateConfiguration(t *testing.T) {\n\ttests := map[string]struct {\n\t\treq                 *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\tdomainHandlerMockFn func(*domain.MockHandler)\n\t\twantResp            *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\twantErr             bool\n\t}{\n\t\t\"Domain handler fails to UpdateAsyncWorkflowConfiguraton\": {\n\t\t\treq: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tm.EXPECT().UpdateAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).Return(errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"nil request\": {\n\t\t\treq:     nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"Success\": {\n\t\t\treq: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\tdomainHandlerMockFn: func(m *domain.MockHandler) {\n\t\t\t\tm.EXPECT().UpdateAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdomainHandlerMock := domain.NewMockHandler(ctrl)\n\t\t\tif tc.domainHandlerMockFn != nil {\n\t\t\t\ttc.domainHandlerMockFn(domainHandlerMock)\n\t\t\t}\n\n\t\t\thandler := New(testlogger.New(t), domainHandlerMock)\n\t\t\tresp, err := handler.UpdateConfiguration(context.Background(), tc.req)\n\n\t\t\tif tc.wantErr != (err != nil) {\n\t\t\t\tt.Fatalf(\"Error mismatch. Got: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tc.wantResp, resp)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/asyncworkflow/queueconfigapi/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queueconfigapi\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination handler_mock.go -self_package github.com/uber/cadence/common/asyncworkflow/queueconfigapi\n\ntype Handler interface {\n\tGetConfiguraton(context.Context, *types.GetDomainAsyncWorkflowConfiguratonRequest) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error)\n\tUpdateConfiguration(context.Context, *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n}\n"
  },
  {
    "path": "common/authorization/README.md",
    "content": "## Cadence has two authorizer options:\n\n1. OAuthAuthorizer: validates JWTs issued by your Identity Provider and enforces permissions.\n2. NoopAuthorizer: turns authorization off.\n\nIn order to configure, add an authorization section to Cadence server config [example](https://github.com/cadence-workflow/cadence/blob/master/config/development_oauth.yaml). These fields map 1:1 to the Go structs in [common/config](https://github.com/cadence-workflow/cadence/blob/master/common/config/authorization.go).\n\n### Option A for OAuth : Validate tokens via JWKS\n\n\n    authorization:\n        oauthAuthorizer:\n            enable: true \n            # Reject tokens with excessively long TTL (seconds). Optional but recommended.\n            maxJwtTTL: 3600 \n    \n            # JWT verification config (algorithm + how to fetch public keys)\n            jwtCredentials:\n                algorithm: RS256         # supported: RS256\n            # publicKey is optional if you supply a JWKS URL (below)\n            # publicKey: /etc/cadence/keys/idp-public.pem\n\n            provider:\n                jwksURL: \"https://YOUR_IDP/.well-known/jwks.json\"\n                # Optional JSONPath-like claims locations used by Cadence:\n                groupsAttributePath: \"groups\"      \n                adminAttributePath: \"admin\"\n\n### Option B for OAuth : Validate tokens via a static public key\n\n\n    authorization:\n        oauthAuthorizer:\n            enable: true\n            maxJwtTTL: 3600\n            jwtCredentials:\n                algorithm: RS256\n                publicKey: /etc/cadence/keys/idp-public.pem\n\n### NoopAuthorizer: Turning authz off\n\n\n    authorization:\n        noopAuthorizer:\n            enable: true\n\n## Background\n\nThe server constructs an authorization.Attributes object for each API call (actor, API name, domain, optional workflow/tasklist), evaluates the token, and returns an allow/deny Decision. JWTs are expected to contain Cadence-specific claims including groups and (optionally) an admin flag.\n\n### Key structs & functions:\n\n```\nauthorization.Authorizer interface \n\nauthorization.Attributes \n \nauthorization.Decision\n\nauthorization.JWTClaims\n```\n\nWhen OAuth authZ is enabled, clients must present a valid JWT to the frontend service on every call (Cadence uses the provided token to authorize the API/Domain access). The exact header/wire placement is handled by Cadence’s server middleware and the client transport; the important bit is that the token must validate against your jwksURL/publicKey, include expected claims (groups/admin), and not exceed maxJwtTTL. \n"
  },
  {
    "path": "common/authorization/authority_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: authorizer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package authorization -source authorizer.go -destination authority_mock.go -self_package github.com/uber/cadence/common/authorization\n//\n\n// Package authorization is a generated GoMock package.\npackage authorization\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockAuthorizer is a mock of Authorizer interface.\ntype MockAuthorizer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockAuthorizerMockRecorder\n\tisgomock struct{}\n}\n\n// MockAuthorizerMockRecorder is the mock recorder for MockAuthorizer.\ntype MockAuthorizerMockRecorder struct {\n\tmock *MockAuthorizer\n}\n\n// NewMockAuthorizer creates a new mock instance.\nfunc NewMockAuthorizer(ctrl *gomock.Controller) *MockAuthorizer {\n\tmock := &MockAuthorizer{ctrl: ctrl}\n\tmock.recorder = &MockAuthorizerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockAuthorizer) EXPECT() *MockAuthorizerMockRecorder {\n\treturn m.recorder\n}\n\n// Authorize mocks base method.\nfunc (m *MockAuthorizer) Authorize(ctx context.Context, attributes *Attributes) (Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Authorize\", ctx, attributes)\n\tret0, _ := ret[0].(Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Authorize indicates an expected call of Authorize.\nfunc (mr *MockAuthorizerMockRecorder) Authorize(ctx, attributes any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Authorize\", reflect.TypeOf((*MockAuthorizer)(nil).Authorize), ctx, attributes)\n}\n\n// MockFilteredRequestBody is a mock of FilteredRequestBody interface.\ntype MockFilteredRequestBody struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFilteredRequestBodyMockRecorder\n\tisgomock struct{}\n}\n\n// MockFilteredRequestBodyMockRecorder is the mock recorder for MockFilteredRequestBody.\ntype MockFilteredRequestBodyMockRecorder struct {\n\tmock *MockFilteredRequestBody\n}\n\n// NewMockFilteredRequestBody creates a new mock instance.\nfunc NewMockFilteredRequestBody(ctrl *gomock.Controller) *MockFilteredRequestBody {\n\tmock := &MockFilteredRequestBody{ctrl: ctrl}\n\tmock.recorder = &MockFilteredRequestBodyMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFilteredRequestBody) EXPECT() *MockFilteredRequestBodyMockRecorder {\n\treturn m.recorder\n}\n\n// SerializeForLogging mocks base method.\nfunc (m *MockFilteredRequestBody) SerializeForLogging() (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeForLogging\")\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeForLogging indicates an expected call of SerializeForLogging.\nfunc (mr *MockFilteredRequestBodyMockRecorder) SerializeForLogging() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeForLogging\", reflect.TypeOf((*MockFilteredRequestBody)(nil).SerializeForLogging))\n}\n"
  },
  {
    "path": "common/authorization/authorizer.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination authority_mock.go -self_package github.com/uber/cadence/common/authorization\n\npackage authorization\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os\"\n\t\"reflect\"\n\t\"strings\"\n\n\tclientworker \"go.uber.org/cadence/worker\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// DecisionDeny means auth decision is deny\n\tDecisionDeny Decision = iota + 1\n\t// DecisionAllow means auth decision is allow\n\tDecisionAllow\n)\n\nconst (\n\t// PermissionRead means the user can write on the domain level APIs\n\tPermissionRead Permission = iota + 1\n\t// PermissionWrite means the user can write on the domain level APIs\n\tPermissionWrite\n\t// PermissionAdmin means the user can read+write on the domain level APIs\n\tPermissionAdmin\n\t// PermissionProcess means the user can process via the task execution related APIs\n\tPermissionProcess\n)\n\ntype (\n\tdomainData map[string]string\n\t// Attributes is input for authority to make decision.\n\t// It can be extended in future if required auth on resources like WorkflowType and TaskList\n\tAttributes struct {\n\t\tActor        string\n\t\tAPIName      string\n\t\tDomainName   string\n\t\tWorkflowType *types.WorkflowType\n\t\tTaskList     *types.TaskList\n\t\tPermission   Permission\n\t\tRequestBody  FilteredRequestBody // request object except for data inputs (PII)\n\t}\n\n\t// Result is result from authority.\n\tResult struct {\n\t\tDecision Decision\n\t}\n\n\t// Decision is enum type for auth decision\n\tDecision int\n\n\t// Permission is enum type for auth permission\n\tPermission int\n)\n\nfunc NewPermission(permission string) Permission {\n\tswitch permission {\n\tcase \"read\":\n\t\treturn PermissionRead\n\tcase \"write\":\n\t\treturn PermissionWrite\n\tcase \"admin\":\n\t\treturn PermissionAdmin\n\tcase \"process\":\n\t\treturn PermissionProcess\n\tdefault:\n\t\treturn -1\n\t}\n}\n\nfunc (d domainData) Groups(groupType string) []string {\n\tres, ok := d[groupType]\n\tif !ok {\n\t\treturn nil\n\t}\n\treturn strings.Split(res, groupSeparator)\n}\n\n// Authorizer is an interface for authorization\ntype Authorizer interface {\n\tAuthorize(ctx context.Context, attributes *Attributes) (Result, error)\n}\n\nfunc GetAuthProviderClient(privateKey string) (clientworker.AuthorizationProvider, error) {\n\tpk, err := os.ReadFile(privateKey)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid private key path %s\", privateKey)\n\t}\n\treturn clientworker.NewAdminJwtAuthorizationProvider(pk), nil\n}\n\n// FilteredRequestBody request object except for data inputs (PII)\ntype FilteredRequestBody interface {\n\tSerializeForLogging() (string, error)\n}\n\ntype simpleRequestLogWrapper struct {\n\trequest interface{}\n}\n\nfunc (f *simpleRequestLogWrapper) SerializeForLogging() (string, error) {\n\t// We have to check if the request is a typed nil. In the interface we have to handle typed nils.\n\t// The reflection check  is slow but this function is doing json marshalling, so performance\n\t// shouldn't be an issue.\n\tif f.request == nil || reflect.ValueOf(f.request).IsNil() {\n\t\treturn \"\", nil\n\t}\n\n\tres, err := json.Marshal(f.request)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\treturn string(res), nil\n}\n\nfunc NewFilteredRequestBody(request interface{}) FilteredRequestBody {\n\treturn &simpleRequestLogWrapper{request}\n}\n\nfunc validatePermission(claims *JWTClaims, attributes *Attributes, data domainData) error {\n\tif (attributes.Permission < PermissionRead) || (attributes.Permission > PermissionProcess) {\n\t\treturn fmt.Errorf(\"permission %v is not supported\", attributes.Permission)\n\t}\n\n\tallowedGroups := map[string]bool{}\n\t// groups that allowed by domain configuration(in domainData)\n\t// write groups are always checked\n\tfor _, g := range data.Groups(constants.DomainDataKeyForWriteGroups) {\n\t\tallowedGroups[g] = true\n\t}\n\n\tif attributes.Permission == PermissionRead {\n\t\tfor _, g := range data.Groups(constants.DomainDataKeyForReadGroups) {\n\t\t\tallowedGroups[g] = true\n\t\t}\n\t}\n\n\tif attributes.Permission == PermissionProcess {\n\t\tfor _, g := range data.Groups(constants.DomainDataKeyForProcessGroups) {\n\t\t\tallowedGroups[g] = true\n\t\t}\n\t}\n\n\tfor _, jwtGroup := range claims.GetGroups() {\n\t\tif _, ok := allowedGroups[jwtGroup]; ok {\n\t\t\treturn nil\n\t\t}\n\t}\n\n\treturn fmt.Errorf(\"token doesn't have the right permission, jwt groups: %v, allowed groups: %v\", claims.GetGroups(), allowedGroups)\n}\n"
  },
  {
    "path": "common/authorization/authorizer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc Test_validatePermission(t *testing.T) {\n\n\treadRequestAttr := &Attributes{Permission: PermissionRead}\n\twriteRequestAttr := &Attributes{Permission: PermissionWrite}\n\tprocessRequestAttr := &Attributes{Permission: PermissionProcess}\n\n\treadWriteDomainData := domainData{\n\t\tconstants.DomainDataKeyForReadGroups:  \"read1\",\n\t\tconstants.DomainDataKeyForWriteGroups: \"write1\",\n\t}\n\n\treadDomainData := domainData{\n\t\tconstants.DomainDataKeyForReadGroups: \"read1\",\n\t}\n\n\tprocessDomainData := domainData{\n\t\tconstants.DomainDataKeyForProcessGroups: \"process1\",\n\t}\n\n\temptyDomainData := domainData{}\n\n\ttests := []struct {\n\t\tname       string\n\t\tclaims     *JWTClaims\n\t\tattributes *Attributes\n\t\tdata       map[string]string\n\t\twantErr    assert.ErrorAssertionFunc\n\t}{\n\t\t{\n\t\t\tname:       \"no args should always fail\",\n\t\t\tclaims:     &JWTClaims{},\n\t\t\tattributes: &Attributes{},\n\t\t\tdata:       emptyDomainData,\n\t\t\twantErr:    assert.Error,\n\t\t},\n\t\t{\n\t\t\tname:       \"Empty claims will be denied even when domain data has no groups\",\n\t\t\tclaims:     &JWTClaims{},\n\t\t\tattributes: writeRequestAttr,\n\t\t\tdata:       emptyDomainData,\n\t\t\twantErr:    assert.Error,\n\t\t},\n\t\t{\n\t\t\tname:       \"Empty claims will be denied when domain data has at least one group\",\n\t\t\tclaims:     &JWTClaims{},\n\t\t\tattributes: writeRequestAttr,\n\t\t\tdata:       readDomainData,\n\t\t\twantErr:    assert.Error,\n\t\t},\n\t\t{\n\t\t\tname:       \"Read-only groups should not get access to write groups\",\n\t\t\tclaims:     &JWTClaims{Groups: \"read1\"},\n\t\t\tattributes: writeRequestAttr,\n\t\t\tdata:       readWriteDomainData,\n\t\t\twantErr:    assert.Error,\n\t\t},\n\t\t{\n\t\t\tname:       \"Process-only groups should not get access to write groups\",\n\t\t\tclaims:     &JWTClaims{Groups: \"process1\"},\n\t\t\tattributes: writeRequestAttr,\n\t\t\tdata:       processDomainData,\n\t\t\twantErr:    assert.Error,\n\t\t},\n\t\t{\n\t\t\tname:       \"Process-only groups should get access to process groups\",\n\t\t\tclaims:     &JWTClaims{Groups: \"process1\"},\n\t\t\tattributes: processRequestAttr,\n\t\t\tdata:       processDomainData,\n\t\t\twantErr:    assert.NoError,\n\t\t},\n\t\t{\n\t\t\tname:       \"Write-only groups should get access to read groups\",\n\t\t\tclaims:     &JWTClaims{Groups: \"write1\"},\n\t\t\tattributes: readRequestAttr,\n\t\t\tdata:       readWriteDomainData,\n\t\t\twantErr:    assert.NoError,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttt.wantErr(t, validatePermission(tt.claims, tt.attributes, tt.data))\n\t\t})\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecutionRequestSerializeForLogging(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput               interface{}\n\t\texpectedOutput      string\n\t\texpectedErrorOutput error\n\t}{\n\t\t\"complete request without error\": {\n\t\t\tinput:          createNewSignalWithStartWorkflowExecutionRequest(),\n\t\t\texpectedOutput: \"{\\\"domain\\\":\\\"testDomain\\\",\\\"workflowId\\\":\\\"testWorkflowID\\\",\\\"workflowType\\\":{\\\"name\\\":\\\"testWorkflowType\\\"},\\\"taskList\\\":{\\\"name\\\":\\\"testTaskList\\\",\\\"kind\\\":\\\"STICKY\\\"},\\\"executionStartToCloseTimeoutSeconds\\\":1,\\\"taskStartToCloseTimeoutSeconds\\\":1,\\\"identity\\\":\\\"testIdentity\\\",\\\"requestId\\\":\\\"DF66E35D-A5B0-425D-8731-6AAC4A4B6368\\\",\\\"workflowIdReusePolicy\\\":\\\"AllowDuplicate\\\",\\\"signalName\\\":\\\"testRequest\\\",\\\"control\\\":\\\"dGVzdENvbnRyb2w=\\\",\\\"retryPolicy\\\":{\\\"initialIntervalInSeconds\\\":1,\\\"backoffCoefficient\\\":1,\\\"maximumIntervalInSeconds\\\":1,\\\"maximumAttempts\\\":1,\\\"nonRetriableErrorReasons\\\":[\\\"testArray\\\"],\\\"expirationIntervalInSeconds\\\":1},\\\"cronSchedule\\\":\\\"testSchedule\\\",\\\"header\\\":{},\\\"delayStartSeconds\\\":1,\\\"jitterStartSeconds\\\":1,\\\"firstRunAtTimestamp\\\":1}\",\n\t\t},\n\n\t\t\"non marchalable struct should error\": {\n\t\t\tinput:               make(chan struct{}),\n\t\t\texpectedErrorOutput: &json.UnsupportedTypeError{},\n\t\t},\n\n\t\t\"empty request without error\": {\n\t\t\tinput:          &types.SignalWithStartWorkflowExecutionRequest{},\n\t\t\texpectedOutput: \"{}\",\n\t\t},\n\n\t\t\"typed nil request without error\": {\n\t\t\tinput:          (*types.SignalWithStartWorkflowExecutionRequest)(nil),\n\t\t\texpectedOutput: \"\",\n\t\t},\n\n\t\t\"nil request without error\": {\n\t\t\tinput:          nil,\n\t\t\texpectedOutput: \"\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\twrappedInput := NewFilteredRequestBody(test.input)\n\t\t\t\toutput, err := wrappedInput.SerializeForLogging()\n\t\t\t\tassert.Equal(t, test.expectedOutput, output)\n\t\t\t\tif test.expectedErrorOutput != nil {\n\t\t\t\t\tassert.ErrorAs(t, err, &test.expectedErrorOutput)\n\t\t\t\t} else {\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t}\n\n\t\t\t\tassert.NotContains(t, output, \"PII\")\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc createNewSignalWithStartWorkflowExecutionRequest() *types.SignalWithStartWorkflowExecutionRequest {\n\ttestTasklistKind := types.TaskListKind(1)\n\ttestExecutionStartToCloseTimeoutSeconds := int32(1)\n\ttestTaskStartToCloseTimeoutSeconds := int32(1)\n\ttestWorkflowIDReusePolicy := types.WorkflowIDReusePolicy(1)\n\ttestDelayStartSeconds := int32(1)\n\ttestJitterStartSeconds := int32(1)\n\ttestFirstRunAtTimestamp := int64(1)\n\tpiiTestArray := []byte(\"testInputPII\")\n\tpiiTestMap := make(map[string][]byte)\n\tpiiTestMap[\"PII\"] = piiTestArray\n\n\ttestReq := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:       \"testDomain\",\n\t\tWorkflowID:   \"testWorkflowID\",\n\t\tWorkflowType: &types.WorkflowType{Name: \"testWorkflowType\"},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"testTaskList\",\n\t\t\tKind: &testTasklistKind,\n\t\t},\n\t\tInput:                               piiTestArray,\n\t\tExecutionStartToCloseTimeoutSeconds: &testExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      &testTaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            \"testIdentity\",\n\t\tRequestID:                           \"DF66E35D-A5B0-425D-8731-6AAC4A4B6368\",\n\t\tWorkflowIDReusePolicy:               &testWorkflowIDReusePolicy,\n\t\tSignalName:                          \"testRequest\",\n\t\tSignalInput:                         piiTestArray,\n\t\tControl:                             []byte(\"testControl\"),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1,\n\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\tMaximumAttempts:             1,\n\t\t\tNonRetriableErrorReasons:    []string{\"testArray\"},\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tCronSchedule:        \"testSchedule\",\n\t\tMemo:                &types.Memo{Fields: piiTestMap},\n\t\tSearchAttributes:    &types.SearchAttributes{IndexedFields: piiTestMap},\n\t\tHeader:              &types.Header{Fields: map[string][]byte{}},\n\t\tDelayStartSeconds:   &testDelayStartSeconds,\n\t\tJitterStartSeconds:  &testJitterStartSeconds,\n\t\tFirstRunAtTimestamp: &testFirstRunAtTimestamp,\n\t}\n\treturn testReq\n}\n"
  },
  {
    "path": "common/authorization/factory.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nfunc NewAuthorizer(authorization config.Authorization, logger log.Logger, domainCache cache.DomainCache) (Authorizer, error) {\n\tswitch true {\n\tcase authorization.OAuthAuthorizer.Enable:\n\t\treturn NewOAuthAuthorizer(authorization.OAuthAuthorizer, logger, domainCache)\n\tdefault:\n\t\treturn NewNopAuthorizer()\n\t}\n}\n"
  },
  {
    "path": "common/authorization/factory_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"testing\"\n\n\t\"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\ntype (\n\tfactorySuite struct {\n\t\tsuite.Suite\n\t\tlogger log.Logger\n\t}\n)\n\nfunc TestFactorySuite(t *testing.T) {\n\tsuite.Run(t, new(factorySuite))\n}\n\nfunc (s *factorySuite) SetupTest() {\n\ts.logger = testlogger.New(s.Suite.T())\n}\n\nfunc cfgNoop() config.Authorization {\n\treturn config.Authorization{\n\t\tOAuthAuthorizer: config.OAuthAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t\tNoopAuthorizer: config.NoopAuthorizer{\n\t\t\tEnable: true,\n\t\t},\n\t}\n}\n\nfunc cfgOAuth() config.Authorization {\n\treturn config.Authorization{\n\t\tOAuthAuthorizer: config.OAuthAuthorizer{\n\t\t\tEnable: true,\n\t\t\tJwtCredentials: &config.JwtCredentials{\n\t\t\t\tAlgorithm: jwt.SigningMethodRS256.Name,\n\t\t\t\tPublicKey: \"../../config/credentials/keytest.pub\",\n\t\t\t},\n\t\t\tMaxJwtTTL: 12345,\n\t\t},\n\t}\n}\n\nfunc (s *factorySuite) TestFactoryNoopAuthorizer() {\n\tcfgOAuthVar := cfgOAuth()\n\n\tpublicKey, _ := common.LoadRSAPublicKey(cfgOAuthVar.OAuthAuthorizer.JwtCredentials.PublicKey)\n\n\tvar tests = []struct {\n\t\tcfg      config.Authorization\n\t\texpected Authorizer\n\t\terr      error\n\t}{\n\t\t{cfgNoop(), &nopAuthority{}, nil},\n\t\t{cfgOAuthVar, &oauthAuthority{\n\t\t\tconfig:    cfgOAuthVar.OAuthAuthorizer,\n\t\t\tlog:       s.logger,\n\t\t\tpublicKey: publicKey,\n\t\t\tparser:    jwt.NewParser(jwt.WithValidMethods([]string{cfgOAuthVar.OAuthAuthorizer.JwtCredentials.Algorithm}), jwt.WithIssuedAt()),\n\t\t}, nil},\n\t}\n\n\tfor _, test := range tests {\n\t\tauthorizer, err := NewAuthorizer(test.cfg, s.logger, nil)\n\t\ts.Equal(authorizer, test.expected)\n\t\ts.Equal(err, test.err)\n\t}\n}\n"
  },
  {
    "path": "common/authorization/nopAuthorizer.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport \"context\"\n\ntype nopAuthority struct{}\n\n// NewNopAuthorizer creates a no-op authority\nfunc NewNopAuthorizer() (Authorizer, error) {\n\treturn &nopAuthority{}, nil\n}\n\nfunc (a *nopAuthority) Authorize(\n\tctx context.Context,\n\tattributes *Attributes,\n) (Result, error) {\n\treturn Result{Decision: DecisionAllow}, nil\n}\n"
  },
  {
    "path": "common/authorization/oauthAuthorizer.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/MicahParks/keyfunc/v2\"\n\t\"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/jmespath/go-jmespath\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\nvar _ jwt.Claims = (*JWTClaims)(nil)\n\nconst (\n\tgroupSeparator    = \" \"\n\tjwtInternalIssuer = \"internal-jwt\"\n)\n\ntype oauthAuthority struct {\n\tconfig      config.OAuthAuthorizer\n\tdomainCache cache.DomainCache\n\tlog         log.Logger\n\tparser      *jwt.Parser\n\tpublicKey   interface{}\n\tjwks        *keyfunc.JWKS\n}\n\n// JWTClaims is a Cadence specific claim with embeded Claims defined https://datatracker.ietf.org/doc/html/rfc7519#section-4.1\ntype JWTClaims struct {\n\tjwt.RegisteredClaims\n\n\tName   string\n\tGroups string // separated by space\n\tAdmin  bool\n\tTTL    int64 // TODO should be removed. ExpiresAt should be used\n}\n\nfunc (j JWTClaims) GetGroups() []string {\n\treturn strings.Split(j.Groups, groupSeparator)\n}\n\n// NewOAuthAuthorizer creates an oauth Authorizer\nfunc NewOAuthAuthorizer(\n\toauthConfig config.OAuthAuthorizer,\n\tlog log.Logger,\n\tdomainCache cache.DomainCache,\n) (Authorizer, error) {\n\tvar jwks *keyfunc.JWKS\n\tvar key interface{}\n\tvar err error\n\n\tif oauthConfig.JwtCredentials != nil {\n\t\tif oauthConfig.JwtCredentials.Algorithm != jwt.SigningMethodRS256.Name {\n\t\t\treturn nil, fmt.Errorf(\"algorithm %q is not supported\", oauthConfig.JwtCredentials.Algorithm)\n\t\t}\n\n\t\tif key, err = common.LoadRSAPublicKey(oauthConfig.JwtCredentials.PublicKey); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"loading RSA public key: %w\", err)\n\t\t}\n\t}\n\n\tif oauthConfig.Provider != nil {\n\t\tif oauthConfig.Provider.JWKSURL == \"\" {\n\t\t\treturn nil, fmt.Errorf(\"JWKSURL is not set\")\n\t\t}\n\t\t// Create the JWKS from the resource at the given URL.\n\t\tif jwks, err = keyfunc.Get(oauthConfig.Provider.JWKSURL, keyfunc.Options{}); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"creating JWKS from resource: %s error: %w\", oauthConfig.Provider.JWKSURL, err)\n\t\t}\n\t}\n\n\treturn &oauthAuthority{\n\t\tconfig:      oauthConfig,\n\t\tdomainCache: domainCache,\n\t\tlog:         log,\n\t\tparser: jwt.NewParser(\n\t\t\tjwt.WithValidMethods([]string{jwt.SigningMethodRS256.Name}),\n\t\t\tjwt.WithIssuedAt(),\n\t\t),\n\t\tpublicKey: key,\n\t\tjwks:      jwks,\n\t}, nil\n}\n\n// Authorize defines the logic to verify get claims from token\nfunc (a *oauthAuthority) Authorize(ctx context.Context, attributes *Attributes) (Result, error) {\n\tcall := yarpc.CallFromContext(ctx)\n\n\ttoken := call.Header(common.AuthorizationTokenHeaderName)\n\tif token == \"\" {\n\t\ta.log.Debug(\"request is not authorized\", tag.Error(errors.New(\"token is not set in header\")))\n\t\treturn Result{Decision: DecisionDeny}, nil\n\t}\n\n\tvar claims JWTClaims\n\tparsedToken, err := a.parser.ParseWithClaims(token, &claims, a.keyFunc)\n\tif err != nil {\n\t\ta.log.Debug(\"request is not authorized\", tag.Error(err))\n\t\treturn Result{Decision: DecisionDeny}, nil\n\t}\n\n\tif !isTokenInternal(parsedToken) {\n\t\tparsed, _, err := a.parser.ParseUnverified(token, jwt.MapClaims{})\n\t\tif err != nil {\n\t\t\ta.log.Debug(\"request is not authorized\", tag.Error(err))\n\t\t\treturn Result{Decision: DecisionDeny}, nil\n\t\t}\n\n\t\tif err := a.parseExternal(parsed.Claims.(jwt.MapClaims), &claims); err != nil {\n\t\t\ta.log.Debug(\"request is not authorized\", tag.Error(err))\n\t\t\treturn Result{Decision: DecisionDeny}, nil\n\t\t}\n\t}\n\n\tif err := a.validateTTL(&claims); err != nil {\n\t\ta.log.Debug(\"request is not authorized\", tag.Error(err))\n\t\treturn Result{Decision: DecisionDeny}, nil\n\t}\n\n\tif claims.Admin {\n\t\treturn Result{Decision: DecisionAllow}, nil\n\t}\n\n\tdomain, err := a.domainCache.GetDomain(attributes.DomainName)\n\tif err != nil {\n\t\treturn Result{Decision: DecisionDeny}, err\n\t}\n\n\tif err := validatePermission(&claims, attributes, domain.GetInfo().Data); err != nil {\n\t\ta.log.Debug(\"request is not authorized\", tag.Error(err))\n\t\treturn Result{Decision: DecisionDeny}, nil\n\t}\n\n\treturn Result{Decision: DecisionAllow}, nil\n}\n\n// keyFunc returns correct key to check signature\nfunc (a *oauthAuthority) keyFunc(token *jwt.Token) (interface{}, error) {\n\tif isTokenInternal(token) && a.publicKey != nil {\n\t\treturn a.publicKey, nil\n\t}\n\t// External provider with JWKS provided\n\t// https://datatracker.ietf.org/doc/html/rfc7517\n\tif a.jwks != nil {\n\t\treturn a.jwks.Keyfunc(token)\n\t}\n\n\treturn nil, errors.New(\"no public key for verification\")\n}\n\nfunc (a *oauthAuthority) validateTTL(claims *JWTClaims) error {\n\t// Fill ExpiresAt when TTL is passed\n\tif claims.TTL > 0 {\n\t\tclaims.ExpiresAt = jwt.NewNumericDate(claims.IssuedAt.Time.Add(time.Second * time.Duration(claims.TTL)))\n\t}\n\n\texp, err := claims.GetExpirationTime()\n\n\tif err != nil || exp == nil {\n\t\treturn errors.New(\"ExpiresAt is not set\")\n\t}\n\n\ttimeLeft := exp.Unix() - time.Now().Unix()\n\tif timeLeft < 0 {\n\t\treturn errors.New(\"token is expired\")\n\t}\n\n\tif timeLeft > a.config.MaxJwtTTL {\n\t\treturn fmt.Errorf(\"token TTL: %d is larger than MaxTTL allowed: %d\", timeLeft, a.config.MaxJwtTTL)\n\t}\n\n\treturn nil\n}\n\nfunc isTokenInternal(token *jwt.Token) bool {\n\t// external providers should set kid part always\n\tif _, ok := token.Header[\"kid\"]; !ok {\n\t\treturn true\n\t}\n\n\tissuer, err := token.Claims.GetIssuer()\n\tif err != nil {\n\t\treturn false\n\t}\n\n\treturn issuer == jwtInternalIssuer\n}\n\nfunc (a *oauthAuthority) parseExternal(rawClaims map[string]interface{}, claims *JWTClaims) error {\n\tif a.config.Provider.GroupsAttributePath != \"\" {\n\t\tuserGroups, err := jmespath.Search(a.config.Provider.GroupsAttributePath, rawClaims)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"extracting JWT Groups claim: %w\", err)\n\t\t}\n\n\t\tif _, ok := userGroups.(string); !ok {\n\t\t\treturn errors.New(\"cannot convert groups to string\")\n\t\t}\n\n\t\tclaims.Groups = userGroups.(string)\n\t}\n\n\tif a.config.Provider.AdminAttributePath != \"\" {\n\t\tisAdmin, err := jmespath.Search(a.config.Provider.AdminAttributePath, rawClaims)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"extracting JWT Admin claim: %w\", err)\n\t\t}\n\t\tif _, ok := isAdmin.(bool); !ok {\n\t\t\treturn errors.New(\"cannot convert isAdmin to bool\")\n\t\t}\n\t\tclaims.Admin = isAdmin.(bool)\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/authorization/oauthAuthorizer_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/api/encoding\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"golang.org/x/net/context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\toauthSuite struct {\n\t\tsuite.Suite\n\t\tlogger      *log.MockLogger\n\t\tcfg         config.OAuthAuthorizer\n\t\tproviderCfg config.OAuthAuthorizer\n\t\tatt         Attributes\n\t\ttoken       string\n\t\tcontroller  *gomock.Controller\n\t\tdomainCache *cache.MockDomainCache\n\t\tctx         context.Context\n\t\tdomainEntry *cache.DomainCacheEntry\n\t}\n)\n\nfunc TestOAuthSuite(t *testing.T) {\n\tsuite.Run(t, new(oauthSuite))\n}\n\nfunc (s *oauthSuite) SetupTest() {\n\ts.logger = log.NewMockLogger(gomock.NewController(s.T()))\n\ts.cfg = config.OAuthAuthorizer{\n\t\tEnable: true,\n\t\tJwtCredentials: &config.JwtCredentials{\n\t\t\tAlgorithm: jwt.SigningMethodRS256.Name,\n\t\t\tPublicKey: \"../../config/credentials/keytest.pub\",\n\t\t},\n\t\tMaxJwtTTL: 300000001,\n\t}\n\ts.providerCfg = config.OAuthAuthorizer{\n\t\tEnable: true,\n\t\tProvider: &config.OAuthProvider{\n\t\t\tGroupsAttributePath: \"tst_group\",\n\t\t\tAdminAttributePath:  \"tst_admin\",\n\t\t},\n\t}\n\t// https://jwt.io/#debugger-io?token=eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6ZmFsc2UsIklhdCI6MTYyNzUzODcxMiwiVFRMIjozMDAwMDAwMDB9.bh4s8-l1bjG7-QFzuouPy9WPvkq3_9U2e815WFrN-M247NQROBii8ju_N21i6ixK0t-VZTgcJs2B4aN4w1uiCTCg6NyhdeeG8Xd8NcYw0Oq7fjSoFmOXzDzljY6oi9M1XXniNrDIMBLfKXx8tgseSBwOnWoT3vja3ioU6ReqD3Xsp-Wg_clDhb6vtA6pDtnaCVXJNStLSbgWyi-1Mxo9ar92zRDV5YsMaBdUjFUT2bW9QcFzMFAqpHin0QEIa6GPZezY-yn88k5S5cT6Yh7WA4C0Q6C3H1n3EOS05Phwpxt840w7zjh5XR0-rd8-kRX84pHMh0GwHfjV1K7jBQ2QnQ&publicKey=-----BEGIN%20PUBLIC%20KEY-----%0AMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAscukltHilaq%2Bo5gIVE4P%0AGwWl%2BesvJ2EaEpWw6ogr98Un11YJ4oKkwIkLw4iIo0tveCINA3cZmxaW1RejRWKE%0AqYFtQ1rYd6BsnFAHXWh2R3A1FtpG6ANUEGkE7OAJe2%2FL42E%2FImJ%2BGQxRvartInDM%0AyfiRfB7%2BL2n3wG%2BNi%2BhBNMtAaX4Wwbj2hup21Jjuo96TuhcGImBFBATGWaYR2wqe%0A%2F6by9wJexPHlY%2F1uDp3SnzF1dCLjp76SGCfyYqOGC%2FPxhQi7mDxeH9%2FtIC%2Blt%2FSz%0Awc1n8gZLtlRlZHinvYa8lhWXqVYw6WD8h4LTgALq9iY%2BbeD1PFQSY1GkQtt0RhRw%0AeQIDAQAB%0A-----END%20PUBLIC%20KEY-----\n\ts.token = `eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6ZmFsc2UsIklhdCI6MTYyNzUzODcxMiwiVFRMIjozMDAwMDAwMDB9.bh4s8-l1bjG7-QFzuouPy9WPvkq3_9U2e815WFrN-M247NQROBii8ju_N21i6ixK0t-VZTgcJs2B4aN4w1uiCTCg6NyhdeeG8Xd8NcYw0Oq7fjSoFmOXzDzljY6oi9M1XXniNrDIMBLfKXx8tgseSBwOnWoT3vja3ioU6ReqD3Xsp-Wg_clDhb6vtA6pDtnaCVXJNStLSbgWyi-1Mxo9ar92zRDV5YsMaBdUjFUT2bW9QcFzMFAqpHin0QEIa6GPZezY-yn88k5S5cT6Yh7WA4C0Q6C3H1n3EOS05Phwpxt840w7zjh5XR0-rd8-kRX84pHMh0GwHfjV1K7jBQ2QnQ`\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.AuthorizationTokenHeaderName, s.token),\n\t})\n\ts.NoError(err)\n\ts.att = Attributes{\n\t\tActor:      \"John Doe\",\n\t\tAPIName:    \"\",\n\t\tDomainName: \"test-domain\",\n\t\tTaskList:   nil,\n\t\tPermission: PermissionRead,\n\t}\n\n\ts.domainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   \"test-domain-id\",\n\t\t\tName: \"test-domain\",\n\t\t\tData: map[string]string{\n\t\t\t\tconstants.DomainDataKeyForReadGroups: \"c\",\n\t\t\t},\n\t\t},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234, // not used\n\t)\n\n\ts.controller = gomock.NewController(s.T())\n\ts.domainCache = cache.NewMockDomainCache(s.controller)\n\ts.ctx = ctx\n}\n\nfunc (s *oauthSuite) TestCorrectPayload() {\n\ts.domainCache.EXPECT().GetDomain(s.att.DomainName).Return(s.domainEntry, nil).Times(1)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\tresult, err := authorizer.Authorize(s.ctx, &s.att)\n\ts.NoError(err)\n\ts.Equal(result.Decision, DecisionAllow)\n}\n\nfunc (s *oauthSuite) TestItIsAdmin() {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\t// https://jwt.io/#debugger-io?token=eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6dHJ1ZSwiSWF0IjoxNjI3NTM4NzEyLCJUVEwiOjMwMDAwMDAwMH0.W_989GT8UWm-W7Hv0L2A3fND0Ly_CCuAdVMMoCs-l_GYxgxHP4_P5S9ejqh28AhUYllWNTRR_zM_hNakqnlufz09HP7mwlEKsxQrfoaycX20n8b7V-CktlysyVE2ZbCMt0Ef_MJF6bOOJ4JsayP6TQFXTP7QSUqNTpRYLZcBLlKHDZYm8uol_1EEs3kV5j3lP-WNcR18xBG0UIptakatm7aQEfPWOWnbRUpg9XVv3c4Bt8no4TW1z0XmFF9dD8vb2U-idPkPFstZwOZ0Ikn9nCt4W44kbeCC-i8uCe5SRiqNFWtvjnTBTVqXm27owT7ZbJwqvmMmhZ86Lz7eGtxgPQ&publicKey=-----BEGIN%20PUBLIC%20KEY-----%0AMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAscukltHilaq%2Bo5gIVE4P%0AGwWl%2BesvJ2EaEpWw6ogr98Un11YJ4oKkwIkLw4iIo0tveCINA3cZmxaW1RejRWKE%0AqYFtQ1rYd6BsnFAHXWh2R3A1FtpG6ANUEGkE7OAJe2%2FL42E%2FImJ%2BGQxRvartInDM%0AyfiRfB7%2BL2n3wG%2BNi%2BhBNMtAaX4Wwbj2hup21Jjuo96TuhcGImBFBATGWaYR2wqe%0A%2F6by9wJexPHlY%2F1uDp3SnzF1dCLjp76SGCfyYqOGC%2FPxhQi7mDxeH9%2FtIC%2Blt%2FSz%0Awc1n8gZLtlRlZHinvYa8lhWXqVYw6WD8h4LTgALq9iY%2BbeD1PFQSY1GkQtt0RhRw%0AeQIDAQAB%0A-----END%20PUBLIC%20KEY-----\n\ttoken := `eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6dHJ1ZSwiSWF0IjoxNjI3NTM4NzEyLCJUVEwiOjMwMDAwMDAwMH0.W_989GT8UWm-W7Hv0L2A3fND0Ly_CCuAdVMMoCs-l_GYxgxHP4_P5S9ejqh28AhUYllWNTRR_zM_hNakqnlufz09HP7mwlEKsxQrfoaycX20n8b7V-CktlysyVE2ZbCMt0Ef_MJF6bOOJ4JsayP6TQFXTP7QSUqNTpRYLZcBLlKHDZYm8uol_1EEs3kV5j3lP-WNcR18xBG0UIptakatm7aQEfPWOWnbRUpg9XVv3c4Bt8no4TW1z0XmFF9dD8vb2U-idPkPFstZwOZ0Ikn9nCt4W44kbeCC-i8uCe5SRiqNFWtvjnTBTVqXm27owT7ZbJwqvmMmhZ86Lz7eGtxgPQ`\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.AuthorizationTokenHeaderName, token),\n\t})\n\ts.NoError(err)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\tresult, err := authorizer.Authorize(ctx, &s.att)\n\ts.NoError(err)\n\ts.Equal(result.Decision, DecisionAllow)\n}\n\nfunc (s *oauthSuite) TestEmptyToken() {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.AuthorizationTokenHeaderName, \"\"),\n\t})\n\ts.NoError(err)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn fmt.Sprintf(\"%v\", t[0].Field().Interface) == \"token is not set in header\"\n\t}))\n\tresult, _ := authorizer.Authorize(ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc (s *oauthSuite) TestGetDomainError() {\n\ts.domainCache.EXPECT().GetDomain(s.att.DomainName).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\tresult, err := authorizer.Authorize(s.ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n\ts.EqualError(err, \"error\")\n}\n\nfunc (s *oauthSuite) TestIncorrectPublicKey() {\n\ts.cfg.JwtCredentials.PublicKey = \"incorrectPublicKey\"\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.Equal(nil, authorizer)\n\ts.EqualError(err, \"loading RSA public key: invalid public key path incorrectPublicKey\")\n}\n\nfunc (s *oauthSuite) TestIncorrectAlgorithm() {\n\ts.cfg.JwtCredentials.Algorithm = \"SHA256\"\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.Equal(nil, authorizer)\n\ts.ErrorContains(err, \"algorithm \\\"SHA256\\\" is not supported\")\n}\n\nfunc (s *oauthSuite) TestMaxTTLLargerInToken() {\n\ts.cfg.MaxJwtTTL = 1\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn strings.HasPrefix(fmt.Sprintf(\"%v\", t[0].Field().Interface), \"token TTL:\")\n\t}))\n\tresult, _ := authorizer.Authorize(s.ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc (s *oauthSuite) TestIncorrectToken() {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.AuthorizationTokenHeaderName, \"test\"),\n\t})\n\ts.NoError(err)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn fmt.Sprintf(\"%v\", t[0].Field().Interface) == \"token is malformed: token contains an invalid number of segments\"\n\t}))\n\tresult, _ := authorizer.Authorize(ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc (s *oauthSuite) TestIatExpiredToken() {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\t// https://jwt.io/#debugger-io?token=eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6ZmFsc2UsIklhdCI6MTYyNzUzODcxMiwiVFRMIjoxfQ.KLOkzV6sIBFCctbcbK98qT5v7ifL_H_6DAzkKsIE4124m5-LtVClA71o5ZtHuoZoiN2xwvGGnkOYg-LbrMajSjsixhGhgz0sAzAomufKACNX1eW9vB5onfTw2q26rpBz0vkIzBYFqUFor3BS30p0V_lnVQGYWRoIcDYspgTyDqMcJ_T77NVBlsyl6ISGiRdv_COcpMEqE_jse7ZKwuoNnQRQp97J3fapPXd6w6qB_PAPlZSXHikvIXG-_9o60RFcB8GDn1lvjZC1NUzGvM2CpVzS4r1_ViKjnjXMuWEPKOyNjQ6LBV9JkRx86N-6jy5V74OyXi-YkiSMplxAKY2G5g&publicKey=-----BEGIN%20PUBLIC%20KEY-----%0AMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAscukltHilaq%2Bo5gIVE4P%0AGwWl%2BesvJ2EaEpWw6ogr98Un11YJ4oKkwIkLw4iIo0tveCINA3cZmxaW1RejRWKE%0AqYFtQ1rYd6BsnFAHXWh2R3A1FtpG6ANUEGkE7OAJe2%2FL42E%2FImJ%2BGQxRvartInDM%0AyfiRfB7%2BL2n3wG%2BNi%2BhBNMtAaX4Wwbj2hup21Jjuo96TuhcGImBFBATGWaYR2wqe%0A%2F6by9wJexPHlY%2F1uDp3SnzF1dCLjp76SGCfyYqOGC%2FPxhQi7mDxeH9%2FtIC%2Blt%2FSz%0Awc1n8gZLtlRlZHinvYa8lhWXqVYw6WD8h4LTgALq9iY%2BbeD1PFQSY1GkQtt0RhRw%0AeQIDAQAB%0A-----END%20PUBLIC%20KEY-----\n\ttoken := `eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJTdWIiOiIxMjM0NTY3ODkwIiwiTmFtZSI6IkpvaG4gRG9lIiwiR3JvdXBzIjoiYSBiIGMiLCJBZG1pbiI6ZmFsc2UsIklhdCI6MTYyNzUzODcxMiwiVFRMIjoxfQ.KLOkzV6sIBFCctbcbK98qT5v7ifL_H_6DAzkKsIE4124m5-LtVClA71o5ZtHuoZoiN2xwvGGnkOYg-LbrMajSjsixhGhgz0sAzAomufKACNX1eW9vB5onfTw2q26rpBz0vkIzBYFqUFor3BS30p0V_lnVQGYWRoIcDYspgTyDqMcJ_T77NVBlsyl6ISGiRdv_COcpMEqE_jse7ZKwuoNnQRQp97J3fapPXd6w6qB_PAPlZSXHikvIXG-_9o60RFcB8GDn1lvjZC1NUzGvM2CpVzS4r1_ViKjnjXMuWEPKOyNjQ6LBV9JkRx86N-6jy5V74OyXi-YkiSMplxAKY2G5g`\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.AuthorizationTokenHeaderName, token),\n\t})\n\ts.NoError(err)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn fmt.Sprintf(\"%v\", t[0].Field().Interface) == \"token is expired\"\n\t}))\n\tresult, _ := authorizer.Authorize(ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc (s *oauthSuite) TestDifferentGroup() {\n\ts.domainEntry.GetInfo().Data[constants.DomainDataKeyForReadGroups] = \"AdifferentGroup\"\n\ts.domainCache.EXPECT().GetDomain(s.att.DomainName).Return(s.domainEntry, nil).Times(1)\n\ts.att.Permission = PermissionWrite\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn fmt.Sprintf(\"%v\", t[0].Field().Interface) == \"token doesn't have the right permission, jwt groups: [a b c], allowed groups: map[]\"\n\t}))\n\tresult, _ := authorizer.Authorize(s.ctx, &s.att)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc (s *oauthSuite) TestExternalProviderWithoutJWKSWillFail() {\n\tauthorizer, err := NewOAuthAuthorizer(s.providerCfg, s.logger, s.domainCache)\n\ts.Error(err)\n\ts.Equal(nil, authorizer)\n\n}\n\nfunc (s *oauthSuite) TestIncorrectPermission() {\n\ts.domainCache.EXPECT().GetDomain(s.att.DomainName).Return(s.domainEntry, nil).Times(1)\n\ts.att.Permission = Permission(15)\n\tauthorizer, err := NewOAuthAuthorizer(s.cfg, s.logger, s.domainCache)\n\ts.NoError(err)\n\ts.logger.EXPECT().Debug(\"request is not authorized\", gomock.Cond(func(t []tag.Tag) bool {\n\t\treturn fmt.Sprintf(\"%v\", t[0].Field().Interface) == \"permission 15 is not supported\"\n\t}))\n\tresult, err := authorizer.Authorize(s.ctx, &s.att)\n\ts.NoError(err)\n\ts.Equal(result.Decision, DecisionDeny)\n}\n\nfunc Test_oauthAuthority_validateTTL(t *testing.T) {\n\n\ttests := []struct {\n\t\tname      string\n\t\tclaims    *JWTClaims\n\t\tttlConfig int64\n\t\twantErr   assert.ErrorAssertionFunc\n\t}{\n\t\t{\n\t\t\tname:    \"Empty claims will fail TTL validation\",\n\t\t\tclaims:  &JWTClaims{},\n\t\t\twantErr: assert.Error,\n\t\t},\n\t\t{\n\t\t\tname: \"Claims with IAT and Claim TTL will pass\",\n\t\t\tclaims: &JWTClaims{\n\t\t\t\tTTL: 300,\n\t\t\t\tRegisteredClaims: jwt.RegisteredClaims{\n\t\t\t\t\tIssuedAt: jwt.NewNumericDate(time.Now()),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr:   assert.NoError,\n\t\t\tttlConfig: 500,\n\t\t},\n\n\t\t{\n\t\t\tname: \"Claims with IAT but without TTL or ExpiresAT will fail TTL validation\",\n\t\t\tclaims: &JWTClaims{\n\t\t\t\tRegisteredClaims: jwt.RegisteredClaims{\n\t\t\t\t\tIssuedAt: jwt.NewNumericDate(time.Now().Add(-time.Minute)),\n\t\t\t\t},\n\t\t\t},\n\t\t\tttlConfig: 1,\n\t\t\twantErr:   assert.Error,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvalidator := &oauthAuthority{\n\t\t\t\tconfig: config.OAuthAuthorizer{MaxJwtTTL: tt.ttlConfig},\n\t\t\t}\n\t\t\ttt.wantErr(t, validator.validateTTL(tt.claims), fmt.Sprintf(\"validateTTL(%v)\", tt.claims))\n\t\t})\n\t}\n}\n\nfunc TestIsTokenInternal(t *testing.T) {\n\tinternalToken := &jwt.Token{\n\t\tHeader: map[string]interface{}{},\n\t}\n\tinternalTokenWithKid := &jwt.Token{\n\t\tHeader: map[string]interface{}{\n\t\t\t\"kid\": jwtInternalIssuer,\n\t\t},\n\t\tClaims: JWTClaims{\n\t\t\tRegisteredClaims: jwt.RegisteredClaims{\n\t\t\t\tIssuer: jwtInternalIssuer,\n\t\t\t},\n\t\t},\n\t}\n\texternalToken := &jwt.Token{\n\t\tHeader: map[string]interface{}{\n\t\t\t\"kid\": \"3lkj323jkj3\",\n\t\t},\n\t\tClaims: JWTClaims{\n\t\t\tRegisteredClaims: jwt.RegisteredClaims{\n\t\t\t\tIssuer: \"https://cognito-idp.us-east-1.amazonaws.com/us-east-1_hNqHxsxaM\",\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname  string\n\t\ttoken *jwt.Token\n\t\twant  bool\n\t}{\n\t\t{\n\t\t\tname:  \"internal token w/o kid\",\n\t\t\ttoken: internalToken,\n\t\t\twant:  true,\n\t\t},\n\t\t{\n\t\t\tname:  \"internal token with kid\",\n\t\t\ttoken: internalTokenWithKid,\n\t\t\twant:  true,\n\t\t},\n\t\t{\n\t\t\tname:  \"external token with kid\",\n\t\t\ttoken: externalToken,\n\t\t\twant:  false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tassert.Equalf(t, tt.want, isTokenInternal(tt.token), \"isTokenInternal(%v)\", tt.token)\n\t\t})\n\t}\n}\n\nfunc Test_oauthAuthority_parseExternal(t *testing.T) {\n\tclaim := map[string]interface{}{\"cognito:groups\": []interface{}{\"domain2\", \"domain1\", \"group1\"}}\n\n\ttests := []struct {\n\t\tname       string\n\t\tconfig     config.OAuthAuthorizer\n\t\tmapToken   map[string]interface{}\n\t\tclaims     *JWTClaims\n\t\twantGroups string\n\t\twantAdmin  bool\n\t\twantErr    assert.ErrorAssertionFunc\n\t}{\n\t\t{\n\t\t\tname: \"empty config will not alter token\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tGroupsAttributePath: \"\",\n\t\t\t\t\tAdminAttributePath:  \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: assert.NoError,\n\t\t},\n\t\t{\n\t\t\tname: \"groups incorrect path will result into an error\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tGroupsAttributePath: \"/bad/path\",\n\t\t\t\t\tAdminAttributePath:  \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: assert.Error,\n\t\t},\n\t\t{\n\t\t\tname: \"admin incorrect path will result into an error\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tGroupsAttributePath: \"\",\n\t\t\t\t\tAdminAttributePath:  \"/bad/path\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: assert.Error,\n\t\t},\n\t\t{\n\t\t\tname: \"correct groups path will fill claims\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tGroupsAttributePath: \"\\\"cognito:groups\\\" | join(' ', @)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapToken:   claim,\n\t\t\twantErr:    assert.NoError,\n\t\t\twantGroups: \"domain2 domain1 group1\",\n\t\t\twantAdmin:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"correct admin path will fill claims\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tAdminAttributePath: \"\\\"cognito:groups\\\" | contains(@, 'group1')\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapToken:   claim,\n\t\t\twantErr:    assert.NoError,\n\t\t\twantGroups: \"\",\n\t\t\twantAdmin:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"non bool result for admin will result in error\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tAdminAttributePath: \"\\\"cognito:groups\\\"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapToken:   claim,\n\t\t\twantErr:    assert.Error,\n\t\t\twantGroups: \"\",\n\t\t\twantAdmin:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"non string result for groups will result in error\",\n\t\t\tconfig: config.OAuthAuthorizer{\n\t\t\t\tProvider: &config.OAuthProvider{\n\t\t\t\t\tGroupsAttributePath: \"\\\"cognito:groups\\\" | contains(@, 'group1')\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapToken:   claim,\n\t\t\twantErr:    assert.Error,\n\t\t\twantGroups: \"\",\n\t\t\twantAdmin:  false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ta := &oauthAuthority{\n\t\t\t\tconfig: tt.config,\n\t\t\t}\n\t\t\tactualClaim := &JWTClaims{}\n\t\t\terr := a.parseExternal(tt.mapToken, actualClaim)\n\t\t\ttt.wantErr(t, err)\n\t\t\tassert.Equal(t, tt.wantGroups, actualClaim.Groups)\n\t\t\tassert.Equal(t, tt.wantAdmin, actualClaim.Admin)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/authorization/scram_client.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage authorization\n\nimport (\n\t\"crypto/sha256\"\n\t\"crypto/sha512\"\n\t\"hash\"\n\n\t\"github.com/xdg/scram\"\n)\n\n// NOTE: the code is copied from https://github.com/IBM/sarama/blob/master/examples/sasl_scram_client/scram_client.go\n\n// SHA256 algorithm\nvar SHA256 scram.HashGeneratorFcn = func() hash.Hash { return sha256.New() }\n\n// SHA512 algorithm\nvar SHA512 scram.HashGeneratorFcn = func() hash.Hash { return sha512.New() }\n\n// XDGSCRAMClient is the scram client\ntype XDGSCRAMClient struct {\n\t*scram.Client\n\t*scram.ClientConversation\n\tscram.HashGeneratorFcn\n}\n\n// Begin creates new client\nfunc (x *XDGSCRAMClient) Begin(userName, password, authzID string) (err error) {\n\tx.Client, err = x.HashGeneratorFcn.NewClient(userName, password, authzID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tx.ClientConversation = x.Client.NewConversation()\n\treturn nil\n}\n\n// Step takes a string provided from a server (or just an empty string for the\n// very first conversation step) and attempts to move the authentication\n// conversation forward.  It returns a string to be sent to the server or an\n// error if the server message is invalid.  Calling Step after a conversation\n// completes is also an error.\nfunc (x *XDGSCRAMClient) Step(challenge string) (response string, err error) {\n\tresponse, err = x.ClientConversation.Step(challenge)\n\treturn\n}\n\n// Done stops the client\nfunc (x *XDGSCRAMClient) Done() bool {\n\treturn x.ClientConversation.Done()\n}\n"
  },
  {
    "path": "common/authorization/test_result",
    "content": ""
  },
  {
    "path": "common/backoff/cron.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/robfig/cron/v3\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// NoBackoff is used to represent backoff when no cron backoff is needed\nconst NoBackoff = time.Duration(-1)\n\n// ValidateSchedule validates a cron schedule spec\nfunc ValidateSchedule(cronSchedule string) (cron.Schedule, error) {\n\tsched, err := cron.ParseStandard(cronSchedule)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"Invalid CronSchedule, failed to parse: %q, err: %v\", cronSchedule, err),\n\t\t}\n\t}\n\t// schedule must parse and there must be a next-firing date (catches impossible dates like Feb 30)\n\tnext := sched.Next(time.Now())\n\tif next.IsZero() {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"Invalid CronSchedule, no next firing time found, maybe impossible date: %q\", cronSchedule),\n\t\t}\n\t}\n\treturn sched, nil\n}\n\n// GetBackoffForNextSchedule calculates the backoff time for the next run given\n// a cronSchedule, workflow start time and workflow close time\nfunc GetBackoffForNextSchedule(\n\tsched cron.Schedule,\n\tstartTime time.Time,\n\tcloseTime time.Time,\n\tjitterStartSeconds int32,\n\tcronOverlapPolicy types.CronOverlapPolicy,\n) (time.Duration, error) {\n\tstartUTCTime := startTime.In(time.UTC)\n\tcloseUTCTime := closeTime.In(time.UTC)\n\tnextScheduleTime := sched.Next(startUTCTime)\n\troundedInterval := time.Duration(0)\n\tif nextScheduleTime.IsZero() {\n\t\t// this should only occur for bad specs, e.g. impossible dates like Feb 30,\n\t\t// which should be prevented from being saved by the valid check.\n\t\treturn NoBackoff, fmt.Errorf(\"invalid CronSchedule, no next firing time found\")\n\t}\n\n\tif nextScheduleTime.Before(closeUTCTime) {\n\t\t// Cron overlap policy only applies if there were runs skipped\n\t\tswitch cronOverlapPolicy {\n\t\tcase types.CronOverlapPolicySkipped:\n\t\t\tfor nextScheduleTime.Before(closeUTCTime) {\n\t\t\t\tnextScheduleTime = sched.Next(nextScheduleTime)\n\t\t\t\tif nextScheduleTime.IsZero() {\n\t\t\t\t\t// this should only occur for bad specs, e.g. impossible dates like Feb 30,\n\t\t\t\t\t// which should be prevented from being saved by the valid check.\n\t\t\t\t\treturn NoBackoff, fmt.Errorf(\"invalid CronSchedule, no next firing time found\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tbackoffInterval := nextScheduleTime.Sub(closeUTCTime)\n\t\t\troundedInterval = time.Second * time.Duration(math.Ceil(backoffInterval.Seconds()))\n\t\tcase types.CronOverlapPolicyBufferOne:\n\t\t\t// we want to start the next run as soon as possible, so we don't need to buffer\n\t\t\troundedInterval = time.Duration(0)\n\t\t}\n\t} else {\n\t\tbackoffInterval := nextScheduleTime.Sub(closeUTCTime)\n\t\troundedInterval = time.Second * time.Duration(math.Ceil(backoffInterval.Seconds()))\n\t}\n\n\tvar jitter time.Duration\n\tif jitterStartSeconds > 0 {\n\t\tjitter = time.Duration(rand.Int31n(jitterStartSeconds+1)) * time.Second\n\t}\n\n\treturn roundedInterval + jitter, nil\n}\n\n// GetBackoffForNextScheduleInSeconds calculates the backoff time in seconds for the\n// next run given a cronSchedule and current time\nfunc GetBackoffForNextScheduleInSeconds(\n\tcronSchedule string,\n\tstartTime time.Time,\n\tcloseTime time.Time,\n\tjitterStartSeconds int32,\n\toverlapPolicy types.CronOverlapPolicy,\n) (int32, error) {\n\tsched, err := ValidateSchedule(cronSchedule)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tbackoffDuration, err := GetBackoffForNextSchedule(sched, startTime, closeTime, jitterStartSeconds, overlapPolicy)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn int32(math.Ceil(backoffDuration.Seconds())), nil\n}\n"
  },
  {
    "path": "common/backoff/cron_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestCron(t *testing.T) {\n\tvar crontests = []struct {\n\t\tcron      string\n\t\tstartTime string\n\t\tendTime   string\n\t\tresult    time.Duration\n\t}{\n\t\t{\"0 10 * * *\", \"2018-12-17T08:00:00-08:00\", \"\", time.Hour * 18},\n\t\t{\"0 10 * * *\", \"2018-12-18T02:00:00-08:00\", \"\", time.Hour * 24},\n\t\t{\"0 * * * *\", \"2018-12-17T08:08:00+00:00\", \"\", time.Minute * 52},\n\t\t{\"0 * * * *\", \"2018-12-17T09:00:00+00:00\", \"\", time.Hour},\n\t\t{\"* * * * *\", \"2018-12-17T08:08:18+00:00\", \"\", time.Second * 42},\n\t\t{\"0 * * * *\", \"2018-12-17T09:00:00+00:00\", \"\", time.Minute * 60},\n\t\t{\"0 10 * * *\", \"2018-12-17T08:00:00+00:00\", \"2018-12-20T00:00:00+00:00\", time.Hour * 10},\n\t\t{\"0 10 * * *\", \"2018-12-17T08:00:00+00:00\", \"2018-12-17T09:00:00+00:00\", time.Hour},\n\t\t{\"*/10 * * * *\", \"2018-12-17T00:04:00+00:00\", \"2018-12-17T01:02:00+00:00\", time.Minute * 8},\n\t\t{\"invalid-cron-spec\", \"2018-12-17T00:04:00+00:00\", \"2018-12-17T01:02:00+00:00\", NoBackoff},\n\t\t{\"@every 5h\", \"2018-12-17T08:00:00+00:00\", \"2018-12-17T09:00:00+00:00\", time.Hour * 4},\n\t\t{\"@every 5h\", \"2018-12-17T08:00:00+00:00\", \"2018-12-18T00:00:00+00:00\", time.Hour * 4},\n\t\t{\"0 3 * * 0-6\", \"2018-12-17T08:00:00-08:00\", \"\", time.Hour * 11},\n\t\t// At 16:05 East Coast (Day light saving on)\n\t\t{\"CRON_TZ=America/New_York 5 16 * * *\", \"2021-03-14T00:00:00-04:00\", \"2021-03-14T15:05:00-04:00\", time.Hour * 1},\n\t\t// At 04:05 East Coast (Day light saving off)\n\t\t{\"CRON_TZ=America/New_York 5 4 * * *\", \"2021-11-25T00:00:00-05:00\", \"2021-11-25T03:05:00-05:00\", time.Hour * 1},\n\t}\n\tfor idx, tt := range crontests {\n\t\tt.Run(strconv.Itoa(idx), func(t *testing.T) {\n\t\t\tstart, _ := time.Parse(time.RFC3339, tt.startTime)\n\t\t\tend := start\n\t\t\tif tt.endTime != \"\" {\n\t\t\t\tend, _ = time.Parse(time.RFC3339, tt.endTime)\n\t\t\t}\n\t\t\tsched, err := ValidateSchedule(tt.cron)\n\t\t\tif tt.result == NoBackoff {\n\t\t\t\t// no backoff == error, simplifies the test-struct a bit\n\t\t\t\trequire.ErrorContains(t, err, \"Invalid CronSchedule\")\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tbackoff, err := GetBackoffForNextSchedule(sched, start, end, 0, types.CronOverlapPolicySkipped)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.result, backoff, \"The cron spec is %s and the expected result is %s\", tt.cron, tt.result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCronWithJitterStart(t *testing.T) {\n\tvar cronWithJitterStartTests = []struct {\n\t\tcron                   string\n\t\tstartTime              string\n\t\tjitterStartSeconds     int32\n\t\texpectedResultSeconds  time.Duration\n\t\texpectedResultSeconds2 time.Duration\n\t}{\n\t\t// Note that the cron scheduler we use (robfig) schedules differently depending on the syntax:\n\t\t// 1) * * * syntax : next run is scheduled on the first second of each minute, starting at next minute\n\t\t// 2) @every X syntax: next run is scheduled X seconds from the time passed in to Next() call.\n\t\t{\"* * * * *\", \"2018-12-17T08:00:00+00:00\", 10, time.Second * 60, time.Second * 60},\n\t\t{\"* * * * *\", \"2018-12-17T08:00:10+00:00\", 30, time.Second * 50, time.Second * 60},\n\t\t{\"* * * * *\", \"2018-12-17T08:00:25+00:00\", 15, time.Second * 35, time.Second * 60},\n\t\t{\"* * * * *\", \"2018-12-17T08:00:45+00:00\", 0, time.Second * 15, time.Second * 60},\n\t\t{\"@every 60s\", \"2018-12-17T08:00:45+00:00\", 0, time.Second * 60, time.Second * 60},\n\t\t{\"* * * * *\", \"2018-12-17T08:00:45+00:00\", 45, time.Second * 15, time.Second * 60},\n\t\t{\"@every 60s\", \"2018-12-17T08:00:45+00:00\", 45, time.Second * 60, time.Second * 60},\n\t\t{\"* * * * *\", \"2018-12-17T08:00:00+00:00\", 70, time.Second * 60, time.Second * 60},\n\t\t{\"@every 20s\", \"2018-12-17T08:00:00+00:00\", 15, time.Second * 20, time.Second * 20},\n\t\t{\"@every 10s\", \"2018-12-17T08:00:09+00:00\", 0, time.Second * 10, time.Second * 10},\n\t\t{\"@every 20s\", \"2018-12-17T08:00:09+00:00\", 15, time.Second * 20, time.Second * 20},\n\t\t{\"* * * * *\", \"0001-01-01T00:00:00+00:00\", 0, time.Second * 60, time.Second * 60},\n\t\t{\"@every 20s\", \"0001-01-01T00:00:00+00:00\", 0, time.Second * 20, time.Second * 20},\n\t}\n\n\trand.Seed(int64(time.Now().Nanosecond()))\n\tfor idx, tt := range cronWithJitterStartTests {\n\t\tt.Run(strconv.Itoa(idx), func(t *testing.T) {\n\t\t\texactCount := 0\n\n\t\t\tstart, _ := time.Parse(time.RFC3339, tt.startTime)\n\t\t\tend := start\n\t\t\tsched, err := ValidateSchedule(tt.cron)\n\t\t\tif tt.expectedResultSeconds != NoBackoff {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tbackoff, err := GetBackoffForNextSchedule(sched, start, end, tt.jitterStartSeconds, types.CronOverlapPolicySkipped)\n\t\t\trequire.NoError(t, err)\n\t\t\tfmt.Printf(\"Backoff time for test %d = %v\\n\", idx, backoff)\n\t\t\tdelta := time.Duration(tt.jitterStartSeconds) * time.Second\n\t\t\texpectedResultTime := start.Add(tt.expectedResultSeconds)\n\t\t\tbackoffTime := start.Add(backoff)\n\t\t\tassert.WithinDuration(t, expectedResultTime, backoffTime, delta,\n\t\t\t\t\"The test specs are %v and the expected result in seconds is between %s and %s\",\n\t\t\t\ttt, tt.expectedResultSeconds, tt.expectedResultSeconds+delta)\n\t\t\tif expectedResultTime == backoffTime {\n\t\t\t\texactCount++\n\t\t\t}\n\n\t\t\t// Also check next X cron times\n\t\t\tcaseCount := 5\n\t\t\tfor i := 1; i < caseCount; i++ {\n\t\t\t\tstartTime := expectedResultTime\n\n\t\t\t\tbackoff, err := GetBackoffForNextSchedule(sched, startTime, startTime, tt.jitterStartSeconds, types.CronOverlapPolicySkipped)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\texpectedResultTime := startTime.Add(tt.expectedResultSeconds2)\n\t\t\t\tbackoffTime := startTime.Add(backoff)\n\t\t\t\tassert.WithinDuration(t, expectedResultTime, backoffTime, delta,\n\t\t\t\t\t\"Iteration %d: The test specs are %v and the expected result in seconds is between %s and %s\",\n\t\t\t\t\ti, tt, tt.expectedResultSeconds, tt.expectedResultSeconds+delta)\n\t\t\t\tif expectedResultTime == backoffTime {\n\t\t\t\t\texactCount++\n\t\t\t\t}\n\n\t\t\t}\n\n\t\t\t// If jitter is > 0, we want to detect whether jitter is being applied - BUT we don't want the test\n\t\t\t// to be flaky if the code randomly chooses a jitter of 0, so we try to have enough data points by\n\t\t\t// checking the next X cron times AND by choosing a jitter thats not super low.\n\n\t\t\tif tt.jitterStartSeconds > 0 && exactCount == caseCount {\n\t\t\t\t// Test to make sure a jitter test case sometimes doesn't get exact values\n\t\t\t\tt.Fatalf(\"FAILED to jitter properly? Test specs = %v\\n\", tt)\n\t\t\t} else if tt.jitterStartSeconds == 0 && exactCount != caseCount {\n\t\t\t\t// Test to make sure a non-jitter test case always gets exact values\n\t\t\t\tt.Fatalf(\"Jittered when we weren't supposed to? Test specs = %v\\n\", tt)\n\t\t\t}\n\n\t\t})\n\t}\n}\n\nfunc TestCronWithoutBufferOneCronWorkflow(t *testing.T) {\n\tvar bufferOneCronWorkflowTests = []struct {\n\t\tstartTime       string\n\t\tcloseTime       string\n\t\texpectedBackoff time.Duration\n\t\tdescription     string\n\t}{\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-17T11:00:00+00:00\",\n\t\t\texpectedBackoff: time.Hour * 23,\n\t\t\tdescription:     \"Next schedule is next day at 10:00\",\n\t\t},\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-18T10:45:00+00:00\",\n\t\t\texpectedBackoff: time.Hour*23 + time.Minute*15,\n\t\t\tdescription:     \"Skip that run and schedule at next possible time\",\n\t\t},\n\t}\n\n\tfor idx, tt := range bufferOneCronWorkflowTests {\n\t\tt.Run(fmt.Sprintf(\"%d_%s\", idx, tt.description), func(t *testing.T) {\n\t\t\tstart, err := time.Parse(time.RFC3339, tt.startTime)\n\t\t\trequire.NoError(t, err)\n\t\t\tclose, err := time.Parse(time.RFC3339, tt.closeTime)\n\t\t\trequire.NoError(t, err)\n\n\t\t\tsched, err := ValidateSchedule(\"0 10 * * *\")\n\t\t\trequire.NoError(t, err)\n\n\t\t\tbackoff, err := GetBackoffForNextSchedule(sched, start, close, 0, types.CronOverlapPolicySkipped)\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tt.expectedBackoff, backoff,\n\t\t\t\t\"Test case %d failed: %s\\nStart: %s\\nClose: %s\",\n\t\t\t\tidx, tt.description, tt.startTime, tt.closeTime)\n\t\t})\n\t}\n}\n\nfunc TestCronWithBufferOneCronWorkflow(t *testing.T) {\n\tvar bufferOneCronWorkflowTests = []struct {\n\t\tstartTime       string\n\t\tcloseTime       string\n\t\texpectedBackoff time.Duration\n\t\tdescription     string\n\t}{\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-17T11:00:00+00:00\",\n\t\t\texpectedBackoff: time.Hour * 23,\n\t\t\tdescription:     \"Next schedule is next day at 10:00\",\n\t\t},\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-18T10:45:00+00:00\",\n\t\t\texpectedBackoff: 0,\n\t\t\tdescription:     \"Start immediately after previous close time\",\n\t\t},\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-17T10:30:00+00:00\",\n\t\t\texpectedBackoff: time.Hour*23 + time.Minute*30,\n\t\t\tdescription:     \"Close time is after schedule time\",\n\t\t},\n\t\t{\n\t\t\tstartTime:       \"2018-12-17T10:00:00+00:00\",\n\t\t\tcloseTime:       \"2018-12-30T10:45:00+00:00\",\n\t\t\texpectedBackoff: 0,\n\t\t\tdescription:     \"Start immediately after previous close time even if previous one skipped multiple runs\",\n\t\t},\n\t}\n\n\tfor idx, tt := range bufferOneCronWorkflowTests {\n\t\tt.Run(fmt.Sprintf(\"%d_%s\", idx, tt.description), func(t *testing.T) {\n\t\t\tstart, err := time.Parse(time.RFC3339, tt.startTime)\n\t\t\trequire.NoError(t, err)\n\t\t\tclose, err := time.Parse(time.RFC3339, tt.closeTime)\n\t\t\trequire.NoError(t, err)\n\n\t\t\tsched, err := ValidateSchedule(\"0 10 * * *\")\n\t\t\trequire.NoError(t, err)\n\n\t\t\tbackoff, err := GetBackoffForNextSchedule(sched, start, close, 0, types.CronOverlapPolicyBufferOne)\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tt.expectedBackoff, backoff,\n\t\t\t\t\"Test case %d failed: %s\\nStart: %s\\nClose: %s\",\n\t\t\t\tidx, tt.description, tt.startTime, tt.closeTime)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/backoff/jitter.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"math/rand\"\n\t\"time\"\n)\n\n// JitDuration return random duration from (1-coefficient)*duration to (1+coefficient)*duration, inclusive, exclusive\nfunc JitDuration(duration time.Duration, coefficient float64) time.Duration {\n\tvalidateCoefficient(coefficient)\n\n\treturn time.Duration(JitInt64(duration.Nanoseconds(), coefficient))\n}\n\n// JitInt64 return random number from (1-coefficient)*input to (1+coefficient)*input, inclusive, exclusive\nfunc JitInt64(input int64, coefficient float64) int64 {\n\tvalidateCoefficient(coefficient)\n\tif coefficient == 0 || input == 0 {\n\t\treturn input\n\t}\n\n\tbase := int64(float64(input) * (1 - coefficient))\n\taddon := rand.Int63n(2 * (input - base))\n\treturn base + addon\n}\n\n// JitFloat64 return random number from (1-coefficient)*input to (1+coefficient)*input, inclusive, exclusive\nfunc JitFloat64(input float64, coefficient float64) float64 {\n\tvalidateCoefficient(coefficient)\n\tif coefficient == 0 || input == 0 {\n\t\treturn input\n\t}\n\n\tbase := input * (1 - coefficient)\n\taddon := rand.Float64() * 2 * (input - base)\n\treturn base + addon\n}\n\nfunc validateCoefficient(coefficient float64) {\n\tif coefficient < 0 || coefficient > 1 {\n\t\tpanic(\"coefficient cannot be < 0 or > 1\")\n\t}\n}\n"
  },
  {
    "path": "common/backoff/jitter_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"log\"\n\t\"math/rand\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tjitterSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestJitterSuite(t *testing.T) {\n\ts := new(jitterSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *jitterSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\nfunc (s *jitterSuite) TestJitInt64() {\n\tinput := int64(1048576)\n\tcoefficient := float64(0.25)\n\tlowerBound := int64(float64(input) * (1 - coefficient))\n\tupperBound := int64(float64(input) * (1 + coefficient))\n\n\tfor i := 0; i < 1048576; i++ {\n\t\tresult := JitInt64(input, coefficient)\n\t\ts.True(result >= lowerBound)\n\t\ts.True(result < upperBound)\n\t}\n}\n\nfunc (s *jitterSuite) TestJitInt64WithZeroCoefficient() {\n\tfor i := 0; i < 1048576; i++ {\n\t\tinput := rand.Int63()\n\t\ts.Equal(input, JitInt64(input, 0))\n\t}\n}\n\nfunc (s *jitterSuite) TestJitInt64WithZeroInput() {\n\ts.Equal(int64(0), JitInt64(0, 0.5))\n}\n\nfunc (s *jitterSuite) TestJitFloat64() {\n\tinput := float64(1048576.1048576)\n\tcoefficient := float64(0.16)\n\tlowerBound := float64(input) * (1 - coefficient)\n\tupperBound := float64(input) * (1 + coefficient)\n\n\tfor i := 0; i < 1048576; i++ {\n\t\tresult := JitFloat64(input, coefficient)\n\t\ts.True(result >= lowerBound)\n\t\ts.True(result < upperBound)\n\t}\n}\n\nfunc (s *jitterSuite) TestJitFloat64WithZeroCoefficient() {\n\tfor i := 0; i < 1048576; i++ {\n\t\tinput := rand.Float64()\n\t\ts.Equal(input, JitFloat64(input, 0))\n\t}\n}\n\nfunc (s *jitterSuite) TestJitFloat64WithZeroInput() {\n\ts.Equal(float64(0), JitFloat64(0, 0.5))\n}\n\nfunc (s *jitterSuite) TestJitDuration() {\n\tinput := time.Duration(1099511627776)\n\tcoefficient := float64(0.1)\n\tlowerBound := time.Duration(int64(float64(input.Nanoseconds()) * (1 - coefficient)))\n\tupperBound := time.Duration(int64(float64(input.Nanoseconds()) * (1 + coefficient)))\n\n\tfor i := 0; i < 1048576; i++ {\n\t\tresult := JitDuration(input, coefficient)\n\t\ts.True(result >= lowerBound)\n\t\ts.True(result < upperBound)\n\t}\n}\n\nfunc (s *jitterSuite) TestJitDurationWithZeroCoefficient() {\n\tfor i := 0; i < 1048576; i++ {\n\t\tinput := time.Duration(rand.Int63())\n\t\ts.Equal(input, JitDuration(input, 0))\n\t}\n}\n"
  },
  {
    "path": "common/backoff/retry.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype retryCountKeyType string\n\nconst retryCountKey = retryCountKeyType(\"retryCount\")\n\nvar errCauseOperationTimeout = errors.New(\"operation timeout\")\n\ntype (\n\t// Operation to retry\n\tOperation func(ctx context.Context) error\n\n\t// IsRetryable handler can be used to exclude certain errors during retry\n\tIsRetryable func(error) bool\n\n\t// ConcurrentRetrier is used for client-side throttling. It determines whether to\n\t// throttle outgoing traffic in case downstream backend server rejects\n\t// requests due to out-of-quota or server busy errors.\n\tConcurrentRetrier struct {\n\t\tsync.Mutex\n\t\tretrier      Retrier // Backoff retrier\n\t\tfailureCount int64   // Number of consecutive failures seen\n\t}\n\n\t// ThrottleRetryOption sets the options of ThrottleRetry\n\tThrottleRetryOption func(*ThrottleRetry)\n\n\t// ThrottleRetry is used to run operation with retry and also avoid throttling dependencies\n\tThrottleRetry struct {\n\t\tpolicy           RetryPolicy\n\t\tisRetryable      IsRetryable\n\t\tthrottlePolicy   RetryPolicy\n\t\tisThrottle       IsRetryable\n\t\tclock            clock.TimeSource\n\t\toperationTimeout time.Duration\n\t\texpireContext    bool\n\t}\n)\n\n// Throttle Sleep if there were failures since the last success call.\nfunc (c *ConcurrentRetrier) Throttle() {\n\tc.throttleInternal()\n}\n\nfunc (c *ConcurrentRetrier) throttleInternal() time.Duration {\n\tnext := done\n\n\t// Check if we have failure count.\n\tfailureCount := c.failureCount\n\tif failureCount > 0 {\n\t\tdefer c.Unlock()\n\t\tc.Lock()\n\t\tif c.failureCount > 0 {\n\t\t\tnext = c.retrier.NextBackOff()\n\t\t}\n\t}\n\n\tif next != done {\n\t\ttime.Sleep(next)\n\t}\n\n\treturn next\n}\n\n// Succeeded marks client request succeeded.\nfunc (c *ConcurrentRetrier) Succeeded() {\n\tdefer c.Unlock()\n\tc.Lock()\n\tc.failureCount = 0\n\tc.retrier.Reset()\n}\n\n// Failed marks client request failed because backend is busy.\nfunc (c *ConcurrentRetrier) Failed() {\n\tdefer c.Unlock()\n\tc.Lock()\n\tc.failureCount++\n}\n\n// NewConcurrentRetrier returns an instance of concurrent backoff retrier.\nfunc NewConcurrentRetrier(retryPolicy RetryPolicy) *ConcurrentRetrier {\n\tretrier := NewRetrier(retryPolicy, clock.NewRealTimeSource())\n\treturn &ConcurrentRetrier{retrier: retrier}\n}\n\n// NewThrottleRetry returns a retry handler with given options\nfunc NewThrottleRetry(opts ...ThrottleRetryOption) *ThrottleRetry {\n\tretryPolicy := NewExponentialRetryPolicy(50 * time.Millisecond)\n\tretryPolicy.SetMaximumInterval(2 * time.Second)\n\tthrottlePolicy := NewExponentialRetryPolicy(time.Second)\n\tthrottlePolicy.SetMaximumInterval(10 * time.Second)\n\tthrottlePolicy.SetExpirationInterval(NoInterval)\n\ttr := &ThrottleRetry{\n\t\tpolicy: retryPolicy,\n\t\tisRetryable: func(_ error) bool {\n\t\t\treturn false\n\t\t},\n\t\tthrottlePolicy: throttlePolicy,\n\t\tisThrottle: func(err error) bool {\n\t\t\t_, ok := err.(*types.ServiceBusyError)\n\t\t\treturn ok\n\t\t},\n\t\tclock: clock.NewRealTimeSource(),\n\t}\n\tfor _, opt := range opts {\n\t\topt(tr)\n\t}\n\treturn tr\n}\n\n// WithRetryPolicy returns a setter setting the retry policy of ThrottleRetry\nfunc WithRetryPolicy(policy RetryPolicy) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.policy = policy\n\t}\n}\n\n// WithThrottlePolicy returns setter setting the retry policy when operation returns throttle error\nfunc WithThrottlePolicy(throttlePolicy RetryPolicy) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.throttlePolicy = throttlePolicy\n\t}\n}\n\n// WithRetryableError returns a setter setting the retryable error of ThrottleRetry\nfunc WithRetryableError(isRetryable IsRetryable) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.isRetryable = isRetryable\n\t}\n}\n\n// WithThrottleError returns a setter setting the throttle error of ThrottleRetry\nfunc WithThrottleError(isThrottle IsRetryable) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.isThrottle = isThrottle\n\t}\n}\n\n// WithOperationTimeout establishes a timeout for each attempt of the Operation, running each in a child context that\n// will be cancelled after the specified operationTimeout.\n// Failures caused by the operationTimeout are considered retryable so long as the Operation returns an error that\n// is/wraps a context.DeadlineExceeded.\n// By default, there is no deadline for each operation.\nfunc WithOperationTimeout(operationTimeout time.Duration) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.operationTimeout = operationTimeout\n\t}\n}\n\n// WithContextExpiration causes the ThrottleRetry to run operations within a child context with a deadline equivalent\n// to the RetryPolicy.Expiration. This ensures that across all attempts, the operations may not exceed the deadline\n// of the RetryPolicy.\n// By default, the RetryPolicy is only enforced by not starting additional attempts once it has expired.\nfunc WithContextExpiration() ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.expireContext = true\n\t}\n}\n\nfunc WithClock(clock clock.TimeSource) ThrottleRetryOption {\n\treturn func(tr *ThrottleRetry) {\n\t\ttr.clock = clock\n\t}\n}\n\n// Do function can be used to wrap any call with retry logic\nfunc (tr *ThrottleRetry) Do(ctx context.Context, op Operation) error {\n\tvar prevErr error\n\tvar err error\n\tvar next time.Duration\n\n\tr := NewRetrier(tr.policy, tr.clock)\n\tt := NewRetrier(tr.throttlePolicy, tr.clock)\n\t// If enabled and the RetryPolicy has an expiration, enforce it via context timeout\n\tif expirationInterval := tr.policy.Expiration(); tr.expireContext && expirationInterval > 0 {\n\t\tvar cancel context.CancelFunc\n\t\tctx, cancel = context.WithTimeout(ctx, expirationInterval)\n\t\tdefer cancel()\n\t}\n\tretryCount := 0\n\tfor {\n\t\t// record the previous error before an operation\n\t\tprevErr = err\n\n\t\t// Avoid shadowing err\n\t\tvar attemptTimedOut bool\n\t\tattemptTimedOut, err = tr.attempt(ctx, retryCount, op)\n\t\t// operation completed successfully. No need to retry.\n\t\tif err == nil {\n\t\t\treturn nil\n\t\t}\n\t\tretryCount++\n\n\t\t// Check if the error is retryable\n\t\t// Attempts timing out is considered retryable\n\t\tif !attemptTimedOut && !tr.isRetryable(err) {\n\t\t\t// The returned error will be preferred to a previous one if one exists. That's because the\n\t\t\t// very last error is very likely a timeout error, and it's not useful for logging/troubleshooting\n\t\t\tif prevErr != nil {\n\t\t\t\treturn prevErr\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\tif next = r.NextBackOff(); next == done {\n\t\t\tif prevErr != nil {\n\t\t\t\treturn prevErr\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\t// check if the error is a throttle error\n\t\tif tr.isThrottle(err) {\n\t\t\tthrottleBackOff := t.NextBackOff()\n\t\t\tif throttleBackOff != done && throttleBackOff > next {\n\t\t\t\tnext = throttleBackOff\n\t\t\t}\n\t\t}\n\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tif prevErr != nil {\n\t\t\t\treturn prevErr\n\t\t\t}\n\t\t\treturn err\n\t\tcase <-tr.clock.After(next):\n\t\t}\n\t}\n}\n\nfunc (tr *ThrottleRetry) attempt(ctx context.Context, retryCount int, op Operation) (bool, error) {\n\t// update context with retry count\n\tctx = context.WithValue(ctx, retryCountKey, retryCount)\n\t// If configured with an operation timeout, set it on the context\n\tif tr.operationTimeout > 0 {\n\t\t// Avoid shadowing ctx\n\t\tvar cancel context.CancelFunc\n\t\tctx, cancel = context.WithTimeoutCause(ctx, tr.operationTimeout, errCauseOperationTimeout)\n\t\tdefer cancel()\n\t}\n\topErr := op(ctx)\n\t// Confirm that the context was cancelled by the above timeout\n\t// Validating the cause ensures that any other Context cancellation (incoming timeout, explicit cancel) doesn't\n\t// get treated as an attempt timeout.\n\t// Validating the returned error is/wraps a DeadlineExceeded adds confidence that it was caused by the context\n\t// timing out.\n\tif cause := context.Cause(ctx); errors.Is(cause, errCauseOperationTimeout) && errors.Is(opErr, context.DeadlineExceeded) {\n\t\treturn true, opErr\n\t}\n\treturn false, opErr\n}\n\n// IgnoreErrors can be used as IsRetryable handler for Retry function to exclude certain errors from the retry list\nfunc IgnoreErrors(errorsToExclude []error) func(error) bool {\n\treturn func(err error) bool {\n\t\tfor _, errorToExclude := range errorsToExclude {\n\t\t\tif err == errorToExclude {\n\t\t\t\treturn false\n\t\t\t}\n\t\t}\n\n\t\treturn true\n\t}\n}\n"
  },
  {
    "path": "common/backoff/retry_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tRetrySuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n\n\tsomeError struct{}\n)\n\nfunc TestRetrySuite(t *testing.T) {\n\tsuite.Run(t, new(RetrySuite))\n}\n\nfunc (s *RetrySuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *RetrySuite) TestRetrySuccess() {\n\ti := 0\n\top := func(ctx context.Context) error {\n\t\ti++\n\n\t\tif i == 5 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(10)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.NoError(err)\n\ts.Equal(5, i)\n}\n\nfunc (s *RetrySuite) TestRetryFailed() {\n\ti := 0\n\top := func(ctx context.Context) error {\n\t\ti++\n\n\t\tif i == 7 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(5)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.Error(err)\n}\n\nfunc (s *RetrySuite) TestRetryFailedReturnPreviousError() {\n\ti := 0\n\n\tthe5thError := fmt.Errorf(\"this is the error of the 5th attempt(4th retry attempt)\")\n\top := func(ctx context.Context) error {\n\t\ti++\n\t\tif i == 5 {\n\t\t\treturn the5thError\n\t\t}\n\t\tif i == 7 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\t// Note that this is retry attempts(maybe it should be renamed to SetMaximumRetryAttempts),\n\t// so the total attempts is 5+1=6\n\tpolicy.SetMaximumAttempts(5)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.Error(err)\n\ts.Equal(6, i)\n\ts.Equal(the5thError, err)\n}\n\nfunc (s *RetrySuite) TestIsRetryableSuccess() {\n\ti := 0\n\top := func(ctx context.Context) error {\n\t\ti++\n\n\t\tif i == 5 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn &someError{}\n\t}\n\n\tisRetryable := func(err error) bool {\n\t\tif _, ok := err.(*someError); ok {\n\t\t\treturn true\n\t\t}\n\n\t\treturn false\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(10)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(isRetryable),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.NoError(err, \"Retry count: %v\", i)\n\ts.Equal(5, i)\n}\n\nfunc (s *RetrySuite) TestIsRetryableFailure() {\n\ti := 0\n\top := func(ctx context.Context) error {\n\t\ti++\n\n\t\tif i == 5 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(10)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(IgnoreErrors([]error{&someError{}})),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.Error(err)\n\ts.Equal(1, i)\n}\n\nfunc (s *RetrySuite) TestRetryExpired() {\n\ti := 0\n\top := func(ctx context.Context) error {\n\t\ti++\n\t\ttime.Sleep(time.Second)\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(10 * time.Millisecond)\n\tpolicy.SetExpirationInterval(NoInterval)\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\tdefer cancel()\n\terr := throttleRetry.Do(ctx, op)\n\ts.Error(err)\n\ts.Equal(&someError{}, err)\n\ts.Equal(1, i)\n}\n\nfunc (s *RetrySuite) TestRetryExpiredReturnPreviousError() {\n\ti := 0\n\tprevErr := fmt.Errorf(\"previousError\")\n\top := func(ctx context.Context) error {\n\t\ti++\n\t\tif i == 1 {\n\t\t\treturn prevErr\n\t\t}\n\t\ttime.Sleep(time.Second)\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(10 * time.Millisecond)\n\tpolicy.SetExpirationInterval(NoInterval)\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)\n\tdefer cancel()\n\terr := throttleRetry.Do(ctx, op)\n\ts.Error(err)\n\ts.Equal(prevErr, err)\n\ts.Equal(2, i)\n}\n\nfunc (s *RetrySuite) TestRetryThrottled() {\n\ti := 0\n\tthrottleErr := fmt.Errorf(\"throttled\")\n\top := func(ctx context.Context) error {\n\t\ti++\n\t\tif i == 1 {\n\t\t\treturn throttleErr\n\t\t}\n\t\treturn nil\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(time.Millisecond)\n\tpolicy.SetExpirationInterval(NoInterval)\n\tthrottlePolicy := NewExponentialRetryPolicy(time.Second)\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t\tWithThrottlePolicy(throttlePolicy),\n\t\tWithThrottleError(func(e error) bool { return e == throttleErr }),\n\t)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)\n\tdefer cancel()\n\terr := throttleRetry.Do(ctx, op)\n\ts.Error(err)\n\ts.Equal(throttleErr, err)\n\ts.Equal(1, i) // Because retry expires before next retry\n}\n\nfunc (s *RetrySuite) TestConcurrentRetrier() {\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(10 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(4)\n\n\t// Basic checks\n\tretrier := NewConcurrentRetrier(policy)\n\tretrier.Failed()\n\ts.Equal(int64(1), retrier.failureCount)\n\tretrier.Succeeded()\n\ts.Equal(int64(0), retrier.failureCount)\n\tsleepDuration := retrier.throttleInternal()\n\ts.Equal(done, sleepDuration)\n\n\t// Multiple count check.\n\tretrier.Failed()\n\tretrier.Failed()\n\ts.Equal(int64(2), retrier.failureCount)\n\t// Verify valid sleep times.\n\tch := make(chan time.Duration, 3)\n\tgo func() {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\tch <- retrier.throttleInternal()\n\t\t}\n\t}()\n\tfor i := 0; i < 3; i++ {\n\t\tval := <-ch\n\t\tfmt.Printf(\"Duration: %d\\n\", val)\n\t\ts.True(val > 0)\n\t}\n\tretrier.Succeeded()\n\ts.Equal(int64(0), retrier.failureCount)\n\t// Verify we don't have any sleep times.\n\tgo func() {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\tch <- retrier.throttleInternal()\n\t\t}\n\t}()\n\tfor i := 0; i < 3; i++ {\n\t\tval := <-ch\n\t\tfmt.Printf(\"Duration: %d\\n\", val)\n\t\ts.Equal(done, val)\n\t}\n}\n\nfunc (s *RetrySuite) TestRetryCountInContext() {\n\tretryCounts := make([]int, 0)\n\top := func(ctx context.Context) error {\n\t\tretryCount := ctx.Value(retryCountKey).(int)\n\t\tretryCounts = append(retryCounts, retryCount)\n\t\tif retryCount == 2 {\n\t\t\treturn nil\n\t\t}\n\t\treturn &someError{}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\tpolicy.SetMaximumAttempts(10)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\tWithRetryableError(func(_ error) bool { return true }),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.NoError(err)\n\ts.Equal([]int{0, 1, 2}, retryCounts)\n}\n\nfunc (s *RetrySuite) TestContextDeadline() {\n\tcases := []struct {\n\t\tname            string\n\t\tpolicy          func() RetryPolicy\n\t\tenforceDeadline bool\n\t\topts            []ThrottleRetryOption\n\t\texpectedMin     time.Duration\n\t\texpectedMax     time.Duration\n\t}{\n\t\t{\n\t\t\tname: \"no deadline\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(NoInterval)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: true,\n\t\t\texpectedMin:     0,\n\t\t\texpectedMax:     0,\n\t\t},\n\t\t{\n\t\t\tname: \"don't enforce deadline\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(time.Hour)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: false,\n\t\t\texpectedMin:     0,\n\t\t\texpectedMax:     0,\n\t\t},\n\t\t{\n\t\t\tname: \"expiration interval\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(time.Hour)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: true,\n\t\t\texpectedMin:     50 * time.Minute,\n\t\t\texpectedMax:     70 * time.Minute,\n\t\t},\n\t\t{\n\t\t\tname: \"operation timeout\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(NoInterval)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: true,\n\t\t\topts: []ThrottleRetryOption{\n\t\t\t\tWithOperationTimeout(time.Hour),\n\t\t\t},\n\t\t\texpectedMin: 50 * time.Minute,\n\t\t\texpectedMax: 70 * time.Minute,\n\t\t},\n\t\t{\n\t\t\tname: \"expiration interval lower, take lower\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(time.Hour)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: true,\n\t\t\topts: []ThrottleRetryOption{\n\t\t\t\tWithOperationTimeout(10 * time.Hour),\n\t\t\t},\n\t\t\texpectedMin: 50 * time.Minute,\n\t\t\texpectedMax: 70 * time.Minute,\n\t\t},\n\t\t{\n\t\t\tname: \"operation timeout lower, take lower\",\n\t\t\tpolicy: func() RetryPolicy {\n\t\t\t\tpolicy := NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetExpirationInterval(10 * time.Hour)\n\t\t\t\treturn policy\n\t\t\t},\n\t\t\tenforceDeadline: true,\n\t\t\topts: []ThrottleRetryOption{\n\t\t\t\tWithOperationTimeout(time.Hour),\n\t\t\t},\n\t\t\texpectedMin: 50 * time.Minute,\n\t\t\texpectedMax: 70 * time.Minute,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\tvar opts []ThrottleRetryOption\n\t\t\topts = append(opts, WithRetryPolicy(tc.policy()))\n\t\t\tif tc.enforceDeadline {\n\t\t\t\topts = append(opts, WithContextExpiration())\n\t\t\t}\n\t\t\tif len(tc.opts) > 0 {\n\t\t\t\topts = append(opts, tc.opts...)\n\t\t\t}\n\t\t\tthrottleRetry := NewThrottleRetry(opts...)\n\t\t\tvar result time.Duration\n\n\t\t\terr := throttleRetry.Do(context.Background(), func(ctx context.Context) error {\n\t\t\t\tif deadline, ok := ctx.Deadline(); ok {\n\t\t\t\t\tresult = deadline.Sub(time.Now())\n\t\t\t\t} else {\n\t\t\t\t\tresult = 0\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t})\n\t\t\ts.NoError(err, \"somehow failed\")\n\t\t\ts.GreaterOrEqual(result, tc.expectedMin)\n\t\t\ts.LessOrEqual(result, tc.expectedMax)\n\t\t})\n\t}\n}\n\nfunc (s *RetrySuite) TestOperationTimeoutIsRetryable() {\n\tattempts := 0\n\top := func(ctx context.Context) error {\n\t\tretryCount := ctx.Value(retryCountKey).(int)\n\t\tattempts = max(attempts, retryCount)\n\t\t// Return nil, indicating success\n\t\tif retryCount == 3 {\n\t\t\treturn nil\n\t\t}\n\t\t// Return the DeadlineExceeded err\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn ctx.Err()\n\t\t}\n\t}\n\n\tpolicy := NewExponentialRetryPolicy(time.Nanosecond)\n\tpolicy.SetMaximumInterval(time.Nanosecond)\n\tpolicy.SetMaximumAttempts(10)\n\n\tthrottleRetry := NewThrottleRetry(\n\t\tWithRetryPolicy(policy),\n\t\t// Operation timeout is retryable anyway\n\t\tWithRetryableError(func(_ error) bool { return false }),\n\t\t// Set a timeout such that it will instantly be canceled every time\n\t\tWithOperationTimeout(time.Nanosecond),\n\t)\n\n\terr := throttleRetry.Do(context.Background(), op)\n\ts.NoError(err, \"somehow failed\")\n\ts.Equal(3, attempts)\n}\n\nfunc (e *someError) Error() string {\n\treturn \"Some Error\"\n}\n"
  },
  {
    "path": "common/backoff/retrypolicy.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"math\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nconst (\n\t// NoInterval represents Maximim interval\n\tNoInterval                      = 0\n\tdone              time.Duration = -1\n\tnoMaximumAttempts               = 0\n\n\tdefaultBackoffCoefficient = 2.0\n\tdefaultMaximumInterval    = 10 * time.Second\n\tdefaultExpirationInterval = time.Minute\n\tdefaultMaximumAttempts    = noMaximumAttempts\n)\n\ntype (\n\t// RetryPolicy is the API which needs to be implemented by various retry policy implementations\n\tRetryPolicy interface {\n\t\tComputeNextDelay(elapsedTime time.Duration, numAttempts int) time.Duration\n\t\t// Expiration returns the maximum time allowed by the RetryPolicy. A value of NoInterval (0) indicates that the\n\t\t// RetryPolicy may take an unbounded amount of time.\n\t\tExpiration() time.Duration\n\t}\n\n\t// Retrier manages the state of retry operation\n\tRetrier interface {\n\t\tNextBackOff() time.Duration\n\t\tReset()\n\t}\n\n\t// ExponentialRetryPolicy provides the implementation for retry policy using a coefficient to compute the next delay.\n\t// Formula used to compute the next delay is: initialInterval * math.Pow(backoffCoefficient, currentAttempt)\n\tExponentialRetryPolicy struct {\n\t\tinitialInterval    time.Duration\n\t\tbackoffCoefficient float64\n\t\tmaximumInterval    time.Duration\n\t\texpirationInterval time.Duration\n\t\tmaximumAttempts    int\n\t}\n\n\t// MultiPhasesRetryPolicy implements a policy that first use one policy to get next delay,\n\t// and once expired use the next policy for the following retry.\n\t// It can achieve fast retries in first phase then slowly retires in second phase.\n\t// The supported retry policy is ExponentialRetryPolicy.\n\t// To have the correct next delay, set the maximumAttempts in the non-final policy.\n\tMultiPhasesRetryPolicy struct {\n\t\tpolicies []*ExponentialRetryPolicy\n\t}\n\n\tretrierImpl struct {\n\t\tpolicy         RetryPolicy\n\t\tclock          clock.TimeSource\n\t\tcurrentAttempt int\n\t\tstartTime      time.Time\n\t}\n)\n\n// NewExponentialRetryPolicy returns an instance of ExponentialRetryPolicy using the provided initialInterval\nfunc NewExponentialRetryPolicy(initialInterval time.Duration) *ExponentialRetryPolicy {\n\tp := &ExponentialRetryPolicy{\n\t\tinitialInterval:    initialInterval,\n\t\tbackoffCoefficient: defaultBackoffCoefficient,\n\t\tmaximumInterval:    defaultMaximumInterval,\n\t\texpirationInterval: defaultExpirationInterval,\n\t\tmaximumAttempts:    defaultMaximumAttempts,\n\t}\n\n\treturn p\n}\n\n// NewMultiPhasesRetryPolicy creates MultiPhasesRetryPolicy\nfunc NewMultiPhasesRetryPolicy(policies ...*ExponentialRetryPolicy) *MultiPhasesRetryPolicy {\n\n\tfor i := 0; i < len(policies)-1; i++ {\n\t\tif policies[i].maximumAttempts == noMaximumAttempts {\n\t\t\tpanic(\"Non final retry policy in MultiPhasesRetryPolicy need to set maximum attempts\")\n\t\t}\n\t}\n\treturn &MultiPhasesRetryPolicy{\n\t\tpolicies: policies,\n\t}\n}\n\n// NewRetrier is used for creating a new instance of Retrier\nfunc NewRetrier(policy RetryPolicy, clock clock.TimeSource) Retrier {\n\tif policy == nil {\n\t\tpanic(\"Retry policy cannot be nil.\")\n\t}\n\tif clock == nil {\n\t\tpanic(\"Retry clock cannot be nil.\")\n\t}\n\n\treturn &retrierImpl{\n\t\tpolicy:         policy,\n\t\tclock:          clock,\n\t\tstartTime:      clock.Now(),\n\t\tcurrentAttempt: 0,\n\t}\n}\n\n// SetInitialInterval sets the initial interval used by ExponentialRetryPolicy for the very first retry\n// All later retries are computed using the following formula:\n// initialInterval * math.Pow(backoffCoefficient, currentAttempt)\nfunc (p *ExponentialRetryPolicy) SetInitialInterval(initialInterval time.Duration) {\n\tp.initialInterval = initialInterval\n}\n\n// SetBackoffCoefficient sets the coefficient used by ExponentialRetryPolicy to compute next delay for each retry\n// All retries are computed using the following formula:\n// initialInterval * math.Pow(backoffCoefficient, currentAttempt)\nfunc (p *ExponentialRetryPolicy) SetBackoffCoefficient(backoffCoefficient float64) {\n\tp.backoffCoefficient = backoffCoefficient\n}\n\n// SetMaximumInterval sets the maximum interval for each retry\nfunc (p *ExponentialRetryPolicy) SetMaximumInterval(maximumInterval time.Duration) {\n\tp.maximumInterval = maximumInterval\n}\n\n// SetExpirationInterval sets the absolute expiration interval for all retries\nfunc (p *ExponentialRetryPolicy) SetExpirationInterval(expirationInterval time.Duration) {\n\tp.expirationInterval = expirationInterval\n}\n\n// SetMaximumAttempts sets the maximum number of retry attempts\nfunc (p *ExponentialRetryPolicy) SetMaximumAttempts(maximumAttempts int) {\n\tp.maximumAttempts = maximumAttempts\n}\n\n// ComputeNextDelay returns the next delay interval.  This is used by Retrier to delay calling the operation again\nfunc (p *ExponentialRetryPolicy) ComputeNextDelay(elapsedTime time.Duration, numAttempts int) time.Duration {\n\t// Check to see if we ran out of maximum number of attempts\n\tif p.maximumAttempts != noMaximumAttempts && numAttempts >= p.maximumAttempts {\n\t\treturn done\n\t}\n\n\t// Stop retrying after expiration interval is elapsed\n\tif p.expirationInterval != NoInterval && elapsedTime > p.expirationInterval {\n\t\treturn done\n\t}\n\n\tnextInterval := float64(p.initialInterval) * math.Pow(p.backoffCoefficient, float64(numAttempts))\n\t// Disallow retries if initialInterval is negative or nextInterval overflows\n\tif nextInterval <= 0 {\n\t\treturn done\n\t}\n\tif p.maximumInterval != NoInterval {\n\t\tnextInterval = math.Min(nextInterval, float64(p.maximumInterval))\n\t}\n\n\tif p.expirationInterval != NoInterval {\n\t\tremainingTime := float64(math.Max(0, float64(p.expirationInterval-elapsedTime)))\n\t\tnextInterval = math.Min(remainingTime, nextInterval)\n\t}\n\n\t// Bail out if the next interval is smaller than initial retry interval\n\tnextDuration := time.Duration(nextInterval)\n\tif nextDuration < p.initialInterval {\n\t\treturn done\n\t}\n\n\t// add jitter to avoid global synchronization\n\tjitterPortion := int(0.2 * nextInterval)\n\t// Prevent overflow\n\tif jitterPortion < 1 {\n\t\tjitterPortion = 1\n\t}\n\tnextInterval = nextInterval*0.8 + float64(rand.Intn(jitterPortion))\n\n\treturn time.Duration(nextInterval)\n}\n\nfunc (p *ExponentialRetryPolicy) Expiration() time.Duration {\n\treturn p.expirationInterval\n}\n\n// ComputeNextDelay returns the next delay interval.\nfunc (tp MultiPhasesRetryPolicy) ComputeNextDelay(elapsedTime time.Duration, numAttempts int) time.Duration {\n\tpreviousStageRetryCount := 0\n\tfor _, policy := range tp.policies {\n\t\tnextInterval := policy.ComputeNextDelay(elapsedTime, numAttempts-previousStageRetryCount)\n\t\tif nextInterval != done {\n\t\t\treturn nextInterval\n\t\t}\n\t\tpreviousStageRetryCount += policy.maximumAttempts\n\t}\n\treturn done\n}\n\nfunc (tp MultiPhasesRetryPolicy) Expiration() time.Duration {\n\tvar sum time.Duration\n\tfor _, policy := range tp.policies {\n\t\texp := policy.Expiration()\n\t\tif exp == NoInterval {\n\t\t\treturn NoInterval\n\t\t}\n\t\tsum += exp\n\t}\n\treturn sum\n}\n\n// Reset will set the Retrier into initial state\nfunc (r *retrierImpl) Reset() {\n\tr.startTime = r.clock.Now()\n\tr.currentAttempt = 0\n}\n\n// NextBackOff returns the next delay interval.  This is used by Retry to delay calling the operation again\nfunc (r *retrierImpl) NextBackOff() time.Duration {\n\tnextInterval := r.policy.ComputeNextDelay(r.getElapsedTime(), r.currentAttempt)\n\n\t// Now increment the current attempt\n\tr.currentAttempt++\n\treturn nextInterval\n}\n\nfunc (r *retrierImpl) getElapsedTime() time.Duration {\n\treturn r.clock.Now().Sub(r.startTime)\n}\n"
  },
  {
    "path": "common/backoff/retrypolicy_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage backoff\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\tRetryPolicySuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestRetryPolicySuite(t *testing.T) {\n\tsuite.Run(t, new(RetryPolicySuite))\n}\n\nfunc (s *RetryPolicySuite) SetupTest() {\n\trand.Seed(int64(time.Now().Nanosecond()))\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *RetryPolicySuite) TestExponentialBackoff() {\n\tpolicy := createPolicy(time.Second)\n\tpolicy.SetMaximumInterval(10 * time.Second)\n\n\texpectedResult := []time.Duration{1, 2, 4, 8, 10}\n\tfor i, d := range expectedResult {\n\t\texpectedResult[i] = d * time.Second\n\t}\n\n\tr, _ := createRetrier(policy)\n\tfor _, expected := range expectedResult {\n\t\tmin, max := getNextBackoffRange(expected)\n\t\tnext := r.NextBackOff()\n\t\ts.True(next >= min, \"NextBackoff too low\")\n\t\ts.True(next < max, \"NextBackoff too high\")\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestNumberOfAttempts() {\n\tpolicy := createPolicy(time.Second)\n\tpolicy.SetMaximumAttempts(5)\n\n\tr, _ := createRetrier(policy)\n\tvar next time.Duration\n\tfor i := 0; i < 6; i++ {\n\t\tnext = r.NextBackOff()\n\t}\n\n\ts.Equal(done, next)\n}\n\n// Test to make sure relative maximum interval for each retry is honoured\nfunc (s *RetryPolicySuite) TestMaximumInterval() {\n\tpolicy := createPolicy(time.Second)\n\tpolicy.SetMaximumInterval(10 * time.Second)\n\n\texpectedResult := []time.Duration{1, 2, 4, 8, 10, 10, 10, 10, 10, 10}\n\tfor i, d := range expectedResult {\n\t\texpectedResult[i] = d * time.Second\n\t}\n\n\tr, _ := createRetrier(policy)\n\tfor _, expected := range expectedResult {\n\t\tmin, max := getNextBackoffRange(expected)\n\t\tnext := r.NextBackOff()\n\t\ts.True(next >= min, \"NextBackoff too low\")\n\t\ts.True(next < max, \"NextBackoff too high\")\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestBackoffCoefficient() {\n\tpolicy := createPolicy(2 * time.Second)\n\tpolicy.SetBackoffCoefficient(1.0)\n\n\tr, _ := createRetrier(policy)\n\tmin, max := getNextBackoffRange(2 * time.Second)\n\tfor i := 0; i < 10; i++ {\n\t\tnext := r.NextBackOff()\n\t\ts.True(next >= min, \"NextBackoff too low\")\n\t\ts.True(next < max, \"NextBackoff too high\")\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestExpirationInterval() {\n\tpolicy := createPolicy(2 * time.Second)\n\tpolicy.SetExpirationInterval(5 * time.Minute)\n\n\tr, clock := createRetrier(policy)\n\tclock.Advance(6 * time.Minute)\n\tnext := r.NextBackOff()\n\n\ts.Equal(done, next)\n}\n\nfunc (s *RetryPolicySuite) TestExpirationOverflow() {\n\tpolicy := createPolicy(2 * time.Second)\n\tpolicy.SetExpirationInterval(5 * time.Second)\n\n\tr, clock := createRetrier(policy)\n\tnext := r.NextBackOff()\n\tmin, max := getNextBackoffRange(2 * time.Second)\n\ts.True(next >= min, \"NextBackoff too low\")\n\ts.True(next < max, \"NextBackoff too high\")\n\n\tclock.Advance(2 * time.Second)\n\n\tnext = r.NextBackOff()\n\tmin, max = getNextBackoffRange(3 * time.Second)\n\ts.True(next >= min, \"NextBackoff too low\")\n\ts.True(next < max, \"NextBackoff too high\")\n}\n\nfunc (s *RetryPolicySuite) TestDefaultPublishRetryPolicy() {\n\tpolicy := NewExponentialRetryPolicy(50 * time.Millisecond)\n\tpolicy.SetExpirationInterval(time.Minute)\n\tpolicy.SetMaximumInterval(10 * time.Second)\n\n\tr, clock := createRetrier(policy)\n\texpectedResult := []time.Duration{\n\t\t50 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t400 * time.Millisecond,\n\t\t800 * time.Millisecond,\n\t\t1600 * time.Millisecond,\n\t\t3200 * time.Millisecond,\n\t\t6400 * time.Millisecond,\n\t\t10000 * time.Millisecond,\n\t\t10000 * time.Millisecond,\n\t\t10000 * time.Millisecond,\n\t\t10000 * time.Millisecond,\n\t\t6000 * time.Millisecond,\n\t\t1250 * time.Millisecond,\n\t\tdone,\n\t}\n\n\ttotal := 0\n\tfor i, expected := range expectedResult {\n\t\tnext := r.NextBackOff()\n\t\ttotal += int(next)\n\t\tif expected == done {\n\t\t\ts.Equal(done, next, \"backoff not done yet!!!\")\n\t\t} else {\n\t\t\tmin, _ := getNextBackoffRange(expected)\n\t\t\ts.True(next >= min, \"iteration %d: NextBackoff too low: actual: %v, expected: %v\", i, next, expected)\n\t\t\t// s.True(next < max, \"NextBackoff too high: actual: %v, expected: %v\", next, expected)\n\t\t\tclock.Advance(expected)\n\t\t}\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestNoMaxAttempts() {\n\tpolicy := createPolicy(50 * time.Millisecond)\n\tpolicy.SetExpirationInterval(time.Minute)\n\tpolicy.SetMaximumInterval(10 * time.Second)\n\n\tr, clock := createRetrier(policy)\n\tfor i := 0; i < 100; i++ {\n\t\tnext := r.NextBackOff()\n\t\t// print(\"Iter: \", i, \", Next Backoff: \", next.String(), \"\\n\")\n\t\ts.True(next > 0 || next == done, \"Unexpected value for next retry duration: %v\", next)\n\t\tclock.Advance(next)\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestUnbounded() {\n\tpolicy := createPolicy(50 * time.Millisecond)\n\n\tr, clock := createRetrier(policy)\n\tfor i := 0; i < 100; i++ {\n\t\tnext := r.NextBackOff()\n\t\t// print(\"Iter: \", i, \", Next Backoff: \", next.String(), \"\\n\")\n\t\ts.True(next > 0 || next == done, \"Unexpected value for next retry duration: %v\", next)\n\t\tclock.Advance(next)\n\t}\n}\n\nfunc (s *RetryPolicySuite) TestMultiPhasesRetryPolicy() {\n\tfirstPolicy := NewExponentialRetryPolicy(50 * time.Millisecond)\n\tfirstPolicy.SetMaximumAttempts(3)\n\tsecondPolicy := NewExponentialRetryPolicy(2 * time.Second)\n\tsecondPolicy.SetMaximumInterval(128 * time.Second)\n\tsecondPolicy.SetExpirationInterval(5 * time.Minute)\n\tpolicy := NewMultiPhasesRetryPolicy(firstPolicy, secondPolicy)\n\n\tr, clock := createRetrier(policy)\n\texpectedResult := []time.Duration{\n\t\t50 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t2 * time.Second,\n\t\t4 * time.Second,\n\t\t8 * time.Second,\n\t\t16 * time.Second,\n\t\t32 * time.Second,\n\t\t64 * time.Second,\n\t\t128 * time.Second,\n\t\t45650 * time.Millisecond,\n\t\tdone,\n\t}\n\tfor _, expected := range expectedResult {\n\t\tnext := r.NextBackOff()\n\t\tif expected == done {\n\t\t\ts.Equal(done, next, \"backoff not done yet!!!\")\n\t\t} else {\n\t\t\tmin, max := getNextBackoffRange(expected)\n\t\t\ts.True(next >= min, \"NextBackoff too low: actual: %v, expected: %v\", next, expected)\n\t\t\ts.True(next < max, \"NextBackoff too high: actual: %v, expected: %v\", next, expected)\n\t\t\tclock.Advance(expected)\n\t\t}\n\t}\n}\n\nfunc createPolicy(initialInterval time.Duration) *ExponentialRetryPolicy {\n\tpolicy := NewExponentialRetryPolicy(initialInterval)\n\tpolicy.SetBackoffCoefficient(2)\n\tpolicy.SetMaximumInterval(NoInterval)\n\tpolicy.SetExpirationInterval(NoInterval)\n\tpolicy.SetMaximumAttempts(noMaximumAttempts)\n\n\treturn policy\n}\n\nfunc createRetrier(policy RetryPolicy) (Retrier, clock.MockedTimeSource) {\n\tclock := clock.NewMockedTimeSourceAt(time.Time{})\n\treturn NewRetrier(policy, clock), clock\n}\n\nfunc getNextBackoffRange(duration time.Duration) (time.Duration, time.Duration) {\n\trangeMin := time.Duration(0.8 * float64(duration))\n\treturn rangeMin, duration\n}\n"
  },
  {
    "path": "common/blobstore/client.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage blobstore\n\nimport \"context\"\n\n//go:generate mockgen -package=$GOPACKAGE -destination=client_mock.go -self_package=github.com/uber/cadence/common/blobstore github.com/uber/cadence/common/blobstore Client\n\ntype (\n\t// Client defines the interface to a blobstore client.\n\tClient interface {\n\t\tPut(context.Context, *PutRequest) (*PutResponse, error)\n\t\tGet(context.Context, *GetRequest) (*GetResponse, error)\n\t\tExists(context.Context, *ExistsRequest) (*ExistsResponse, error)\n\t\tDelete(context.Context, *DeleteRequest) (*DeleteResponse, error)\n\t\tIsRetryableError(error) bool\n\t}\n\n\t// PutRequest is the request to Put\n\tPutRequest struct {\n\t\tKey  string\n\t\tBlob Blob\n\t}\n\n\t// PutResponse is the response from Put\n\tPutResponse struct{}\n\n\t// GetRequest is the request to Get\n\tGetRequest struct {\n\t\tKey string\n\t}\n\n\t// GetResponse is the response from Get\n\tGetResponse struct {\n\t\tBlob Blob\n\t}\n\n\t// ExistsRequest is the request to Exists\n\tExistsRequest struct {\n\t\tKey string\n\t}\n\n\t// ExistsResponse is the response from Exists\n\tExistsResponse struct {\n\t\tExists bool\n\t}\n\n\t// DeleteRequest is the request to Delete\n\tDeleteRequest struct {\n\t\tKey string\n\t}\n\n\t// DeleteResponse is the response from Delete\n\tDeleteResponse struct{}\n\n\t// Blob defines a blob which can be stored and fetched from blobstore\n\tBlob struct {\n\t\tTags map[string]string\n\t\tBody []byte\n\t}\n)\n"
  },
  {
    "path": "common/blobstore/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/blobstore (interfaces: Client)\n//\n// Generated by this command:\n//\n//\tmockgen -package=blobstore -destination=client_mock.go -self_package=github.com/uber/cadence/common/blobstore github.com/uber/cadence/common/blobstore Client\n//\n\n// Package blobstore is a generated GoMock package.\npackage blobstore\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// Delete mocks base method.\nfunc (m *MockClient) Delete(arg0 context.Context, arg1 *DeleteRequest) (*DeleteResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Delete\", arg0, arg1)\n\tret0, _ := ret[0].(*DeleteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Delete indicates an expected call of Delete.\nfunc (mr *MockClientMockRecorder) Delete(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Delete\", reflect.TypeOf((*MockClient)(nil).Delete), arg0, arg1)\n}\n\n// Exists mocks base method.\nfunc (m *MockClient) Exists(arg0 context.Context, arg1 *ExistsRequest) (*ExistsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Exists\", arg0, arg1)\n\tret0, _ := ret[0].(*ExistsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Exists indicates an expected call of Exists.\nfunc (mr *MockClientMockRecorder) Exists(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Exists\", reflect.TypeOf((*MockClient)(nil).Exists), arg0, arg1)\n}\n\n// Get mocks base method.\nfunc (m *MockClient) Get(arg0 context.Context, arg1 *GetRequest) (*GetResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Get\", arg0, arg1)\n\tret0, _ := ret[0].(*GetResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Get indicates an expected call of Get.\nfunc (mr *MockClientMockRecorder) Get(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Get\", reflect.TypeOf((*MockClient)(nil).Get), arg0, arg1)\n}\n\n// IsRetryableError mocks base method.\nfunc (m *MockClient) IsRetryableError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsRetryableError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsRetryableError indicates an expected call of IsRetryableError.\nfunc (mr *MockClientMockRecorder) IsRetryableError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsRetryableError\", reflect.TypeOf((*MockClient)(nil).IsRetryableError), arg0)\n}\n\n// Put mocks base method.\nfunc (m *MockClient) Put(arg0 context.Context, arg1 *PutRequest) (*PutResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Put\", arg0, arg1)\n\tret0, _ := ret[0].(*PutResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Put indicates an expected call of Put.\nfunc (mr *MockClientMockRecorder) Put(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Put\", reflect.TypeOf((*MockClient)(nil).Put), arg0, arg1)\n}\n"
  },
  {
    "path": "common/blobstore/filestore/client.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"os\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\ntype (\n\tclient struct {\n\t\toutputDirectory string\n\t}\n)\n\n// NewFilestoreClient constructs a blobstore backed by local file system\nfunc NewFilestoreClient(cfg *config.FileBlobstore) (blobstore.Client, error) {\n\tif cfg == nil {\n\t\treturn nil, errors.New(\"file blobstore config is nil\")\n\t}\n\tif len(cfg.OutputDirectory) == 0 {\n\t\treturn nil, errors.New(\"output directory not given for file blobstore\")\n\t}\n\toutputDirectory := cfg.OutputDirectory\n\texists, err := util.DirectoryExists(outputDirectory)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !exists {\n\t\tif err := util.MkdirAll(outputDirectory, os.FileMode(0766)); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn &client{\n\t\toutputDirectory: outputDirectory,\n\t}, nil\n}\n\n// Put stores a blob\nfunc (c *client) Put(_ context.Context, request *blobstore.PutRequest) (resp *blobstore.PutResponse, err error) {\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tos.Remove(c.bodyPath(request.Key))\n\t\t\tos.Remove(c.tagsPath(request.Key))\n\t\t}\n\t}()\n\tif err := util.WriteFile(c.bodyPath(request.Key), request.Blob.Body, os.FileMode(0666)); err != nil {\n\t\treturn nil, err\n\t}\n\ttagsData, err := json.Marshal(request.Blob.Tags)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif err := util.WriteFile(c.tagsPath(request.Key), tagsData, os.FileMode(0666)); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &blobstore.PutResponse{}, nil\n}\n\n// Get fetches a blob\nfunc (c *client) Get(_ context.Context, request *blobstore.GetRequest) (*blobstore.GetResponse, error) {\n\tdata, err := util.ReadFile(c.bodyPath(request.Key))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttagsData, err := util.ReadFile(c.tagsPath(request.Key))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttags := make(map[string]string)\n\tif err := json.Unmarshal(tagsData, &tags); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &blobstore.GetResponse{\n\t\tBlob: blobstore.Blob{\n\t\t\tBody: data,\n\t\t\tTags: tags,\n\t\t},\n\t}, nil\n}\n\n// Exists determines if a blob exists\nfunc (c *client) Exists(_ context.Context, request *blobstore.ExistsRequest) (*blobstore.ExistsResponse, error) {\n\texists, err := util.FileExists(c.bodyPath(request.Key))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &blobstore.ExistsResponse{\n\t\tExists: exists,\n\t}, nil\n}\n\n// Delete deletes a blob\nfunc (c *client) Delete(_ context.Context, request *blobstore.DeleteRequest) (*blobstore.DeleteResponse, error) {\n\tif err := os.Remove(c.bodyPath(request.Key)); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := os.Remove(c.tagsPath(request.Key)); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &blobstore.DeleteResponse{}, nil\n}\n\n// IsRetryableError returns true if the error is retryable false otherwise\nfunc (c *client) IsRetryableError(err error) bool {\n\treturn false\n}\n\nfunc (c *client) bodyPath(key string) string {\n\treturn fmt.Sprintf(\"%v/%v\", c.outputDirectory, key)\n}\n\nfunc (c *client) tagsPath(key string) string {\n\treturn fmt.Sprintf(\"%v/.%v.tags\", c.outputDirectory, key)\n}\n"
  },
  {
    "path": "common/blobstore/filestore/client_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage filestore\n\nimport (\n\t\"context\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/util\"\n)\n\ntype ClientSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestClientSuite(t *testing.T) {\n\tsuite.Run(t, new(ClientSuite))\n}\n\nfunc (s *ClientSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *ClientSuite) TestNewFilestoreClient_InvalidConfig() {\n\t_, err := NewFilestoreClient(nil)\n\ts.Error(err)\n\t_, err = NewFilestoreClient(&config.FileBlobstore{})\n\ts.Error(err)\n}\n\nfunc (s *ClientSuite) TestNewFilestoreClient_DirectoryAlreadyExists() {\n\tname := s.T().TempDir()\n\tc, err := NewFilestoreClient(&config.FileBlobstore{OutputDirectory: name})\n\ts.NoError(err)\n\ts.Equal(name, c.(*client).outputDirectory)\n}\n\nfunc (s *ClientSuite) TestNewFilestoreClient_DirectoryNotAlreadyExists() {\n\tname := s.T().TempDir()\n\tos.RemoveAll(name)\n\texists, err := util.DirectoryExists(name)\n\ts.NoError(err)\n\ts.False(exists)\n\tc, err := NewFilestoreClient(&config.FileBlobstore{OutputDirectory: name})\n\ts.NoError(err)\n\ts.Equal(name, c.(*client).outputDirectory)\n\texists, err = util.DirectoryExists(name)\n\ts.NoError(err)\n\ts.True(exists)\n\tos.RemoveAll(name)\n}\n\nfunc (s *ClientSuite) TestCrudOperations() {\n\tname := s.T().TempDir()\n\tc, err := NewFilestoreClient(&config.FileBlobstore{OutputDirectory: name})\n\ts.NoError(err)\n\tctx := context.Background()\n\n\t// put three blobs in blobstore\n\tkey1 := uuid.New()\n\tkey2 := uuid.New()\n\tkey3 := uuid.New()\n\tblob1 := blobstore.Blob{\n\t\tTags: nil,\n\t\tBody: []byte{1, 2, 3},\n\t}\n\tblob2 := blobstore.Blob{\n\t\tTags: map[string]string{\"key1\": \"value1\"},\n\t\tBody: nil,\n\t}\n\tblob3 := blobstore.Blob{\n\t\tTags: map[string]string{\"key1\": \"value1\", \"key2\": \"value2\"},\n\t\tBody: []byte{1, 2, 3, 4, 5},\n\t}\n\t_, err = c.Put(ctx, &blobstore.PutRequest{\n\t\tKey:  key1,\n\t\tBlob: blob1,\n\t})\n\ts.NoError(err)\n\t_, err = c.Put(ctx, &blobstore.PutRequest{\n\t\tKey:  key2,\n\t\tBlob: blob2,\n\t})\n\ts.NoError(err)\n\t_, err = c.Put(ctx, &blobstore.PutRequest{\n\t\tKey:  key3,\n\t\tBlob: blob3,\n\t})\n\ts.NoError(err)\n\n\t// get the blobs back\n\tget1, err := c.Get(ctx, &blobstore.GetRequest{Key: key1})\n\ts.NoError(err)\n\ts.Nil(get1.Blob.Tags)\n\ts.Equal([]byte{1, 2, 3}, get1.Blob.Body)\n\tget2, err := c.Get(ctx, &blobstore.GetRequest{Key: key2})\n\ts.NoError(err)\n\ts.Equal(map[string]string{\"key1\": \"value1\"}, get2.Blob.Tags)\n\ts.Empty(get2.Blob.Body)\n\tget3, err := c.Get(ctx, &blobstore.GetRequest{Key: key3})\n\ts.NoError(err)\n\ts.Equal(map[string]string{\"key1\": \"value1\", \"key2\": \"value2\"}, get3.Blob.Tags)\n\ts.Equal([]byte{1, 2, 3, 4, 5}, get3.Blob.Body)\n\n\t// confirm all the blobs exist\n\texists1, err := c.Exists(ctx, &blobstore.ExistsRequest{Key: key1})\n\ts.NoError(err)\n\ts.True(exists1.Exists)\n\texists2, err := c.Exists(ctx, &blobstore.ExistsRequest{Key: key2})\n\ts.NoError(err)\n\ts.True(exists2.Exists)\n\texists3, err := c.Exists(ctx, &blobstore.ExistsRequest{Key: key3})\n\ts.NoError(err)\n\ts.True(exists3.Exists)\n\n\t// delete a blob and confirm no longer can get and that no longer exists\n\t_, err = c.Delete(ctx, &blobstore.DeleteRequest{Key: key1})\n\ts.NoError(err)\n\texists1, err = c.Exists(ctx, &blobstore.ExistsRequest{Key: key1})\n\ts.NoError(err)\n\ts.False(exists1.Exists)\n\tget1, err = c.Get(ctx, &blobstore.GetRequest{Key: key1})\n\ts.Error(err)\n\ts.Nil(get1)\n}\n"
  },
  {
    "path": "common/blobstore/retryable_client.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage blobstore\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n)\n\ntype (\n\tretryableClient struct {\n\t\tclient        Client\n\t\tthrottleRetry *backoff.ThrottleRetry\n\t}\n)\n\n// NewRetryableClient constructs a blobstorre client which retries transient errors.\nfunc NewRetryableClient(client Client, policy backoff.RetryPolicy) Client {\n\treturn &retryableClient{\n\t\tclient: client,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(client.IsRetryableError),\n\t\t),\n\t}\n}\n\nfunc (c *retryableClient) Put(ctx context.Context, req *PutRequest) (*PutResponse, error) {\n\tvar resp *PutResponse\n\tvar err error\n\top := func(ctx context.Context) error {\n\t\tresp, err = c.client.Put(ctx, req)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *retryableClient) Get(ctx context.Context, req *GetRequest) (*GetResponse, error) {\n\tvar resp *GetResponse\n\tvar err error\n\top := func(ctx context.Context) error {\n\t\tresp, err = c.client.Get(ctx, req)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *retryableClient) Exists(ctx context.Context, req *ExistsRequest) (*ExistsResponse, error) {\n\tvar resp *ExistsResponse\n\tvar err error\n\top := func(ctx context.Context) error {\n\t\tresp, err = c.client.Exists(ctx, req)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *retryableClient) Delete(ctx context.Context, req *DeleteRequest) (*DeleteResponse, error) {\n\tvar resp *DeleteResponse\n\tvar err error\n\top := func(ctx context.Context) error {\n\t\tresp, err = c.client.Delete(ctx, req)\n\t\treturn err\n\t}\n\terr = c.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (c *retryableClient) IsRetryableError(err error) bool {\n\treturn c.client.IsRetryableError(err)\n}\n"
  },
  {
    "path": "common/blobstore/retryable_client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage blobstore\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n)\n\nfunc runCRUDTest(\n\tt *testing.T,\n\tretryPolicy backoff.RetryPolicy,\n\tretryableError bool,\n\treq, resp any,\n\texpectFn func(*MockClient, any, any),\n\tcallFn func(Client, context.Context, any) (any, error),\n\tassertFn func(*testing.T, any, any, any, error),\n) {\n\tmockClient := NewMockClient(gomock.NewController(t))\n\tthrottleRetryOptions := []backoff.ThrottleRetryOption{\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t}\n\tif retryableError {\n\t\tthrottleRetryOptions = append(throttleRetryOptions, backoff.WithRetryableError(mockClient.IsRetryableError))\n\t}\n\tclient := &retryableClient{\n\t\tclient:        mockClient,\n\t\tthrottleRetry: backoff.NewThrottleRetry(throttleRetryOptions...),\n\t}\n\n\texpectFn(mockClient, req, resp)\n\tresult, err := callFn(client, context.Background(), req)\n\tassertFn(t, req, resp, result, err)\n}\n\nfunc TestRetryableClient(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tretryPolicy    backoff.RetryPolicy\n\t\tretryableError bool\n\t\treq            any\n\t\tresp           any\n\t\texpectFn       func(*MockClient, any, any)\n\t\tcallFn         func(Client, context.Context, any) (any, error)\n\t\tassertFn       func(*testing.T, any, any, any, error)\n\t}{\n\t\t{\n\t\t\tname:           \"Put\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(0),\n\t\t\tretryableError: false,\n\t\t\treq:            &PutRequest{},\n\t\t\tresp:           &PutResponse{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tm.EXPECT().Put(gomock.Any(), req.(*PutRequest)).Return(resp.(*PutResponse), nil).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Put(ctx, req.(*PutRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, resp.(*PutResponse), result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Get\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(0),\n\t\t\tretryableError: false,\n\t\t\treq:            &GetRequest{},\n\t\t\tresp:           &GetResponse{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tm.EXPECT().Get(gomock.Any(), req.(*GetRequest)).Return(resp.(*GetResponse), nil).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Get(ctx, req.(*GetRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, resp.(*GetResponse), result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Exists\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(0),\n\t\t\tretryableError: false,\n\t\t\treq:            &ExistsRequest{},\n\t\t\tresp:           &ExistsResponse{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tm.EXPECT().Exists(gomock.Any(), req.(*ExistsRequest)).Return(resp.(*ExistsResponse), nil).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Exists(ctx, req.(*ExistsRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, resp.(*ExistsResponse), result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Delete\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(0),\n\t\t\tretryableError: false,\n\t\t\treq:            &DeleteRequest{},\n\t\t\tresp:           &DeleteResponse{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tm.EXPECT().Delete(gomock.Any(), req.(*DeleteRequest)).Return(resp.(*DeleteResponse), nil).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Delete(ctx, req.(*DeleteRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, resp.(*DeleteResponse), result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"RetryOnError\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(1),\n\t\t\tretryableError: true,\n\t\t\treq:            &PutRequest{},\n\t\t\tresp:           &PutResponse{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tretryableError := errors.New(\"retryable error\")\n\t\t\t\tm.EXPECT().Put(gomock.Any(), req.(*PutRequest)).Return(nil, retryableError).Times(1)\n\t\t\t\tm.EXPECT().IsRetryableError(retryableError).Return(true).Times(1)\n\t\t\t\tm.EXPECT().Put(gomock.Any(), req.(*PutRequest)).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Put(ctx, req.(*PutRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.NoError(t, err, \"Expected no error on successful retry\")\n\t\t\t\tassert.Equal(t, resp.(*PutResponse), result, \"Expected the response to match\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"NotRetryOnError\",\n\t\t\tretryPolicy:    backoff.NewExponentialRetryPolicy(0),\n\t\t\tretryableError: true,\n\t\t\treq:            &PutRequest{},\n\t\t\texpectFn: func(m *MockClient, req, resp any) {\n\t\t\t\tnonRetryableError := errors.New(\"non-retryable error\")\n\t\t\t\tm.EXPECT().Put(gomock.Any(), req.(*PutRequest)).Return(nil, nonRetryableError).Times(1)\n\t\t\t\tm.EXPECT().IsRetryableError(nonRetryableError).Return(false).Times(1)\n\t\t\t},\n\t\t\tcallFn: func(c Client, ctx context.Context, req any) (any, error) {\n\t\t\t\treturn c.Put(ctx, req.(*PutRequest))\n\t\t\t},\n\t\t\tassertFn: func(t *testing.T, req any, resp any, result any, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\trunCRUDTest(\n\t\t\t\tt,\n\t\t\t\ttc.retryPolicy,\n\t\t\t\ttc.retryableError,\n\t\t\t\ttc.req,\n\t\t\t\ttc.resp,\n\t\t\t\ttc.expectFn,\n\t\t\t\ttc.callFn,\n\t\t\t\ttc.assertFn,\n\t\t\t)\n\t\t})\n\t}\n}\n\nfunc TestRetryableClient_IsRetryableError(t *testing.T) {\n\tmockClient := NewMockClient(gomock.NewController(t))\n\tclient := &retryableClient{\n\t\tclient: mockClient,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(backoff.NewExponentialRetryPolicy(0)),\n\t\t\tbackoff.WithRetryableError(mockClient.IsRetryableError),\n\t\t),\n\t}\n\n\tretryableError := errors.New(\"retryable error\")\n\tmockClient.EXPECT().IsRetryableError(retryableError).Return(true).Times(1)\n\n\tisRetryableError := client.IsRetryableError(retryableError)\n\tassert.True(t, isRetryableError, \"Expected error to be retryable\")\n}\n"
  },
  {
    "path": "common/cache/ack_cache.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cache\n\nimport (\n\t\"container/heap\"\n\t\"errors\"\n\t\"math\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nvar (\n\t// ErrAckCacheFull is returned when the cache is at capacity and cannot accept new items\n\tErrAckCacheFull = errors.New(\"ack cache is full\")\n\t// ErrAlreadyAcked is returned when trying to add an item with sequence ID that was already acknowledged\n\tErrAlreadyAcked = errors.New(\"sequence ID already acknowledged\")\n)\n\n// AckCacheItem represents an item that can be stored in an AckCache.\n// Items must have a sequence ID for ordering.\ntype AckCacheItem interface {\n\t// GetSequenceID returns the sequence identifier for this item.\n\t// Items are ordered by sequence ID and acknowledged up to a sequence level.\n\tGetSequenceID() int64\n}\n\n// AckCache is a bounded cache that stores items in sequence ID order.\n// Items are removed via acknowledgment - calling Ack(level) removes all items\n// with sequence ID <= level. When the cache reaches capacity, new items are\n// rejected rather than evicting existing items.\n//\n// This cache is useful for scenarios like:\n// - Buffering out-of-order messages that need acknowledgment\n// - Caching items that must be processed in sequence\n// - Storing items with backpressure (reject rather than evict)\n//\n// The cache is thread-safe for concurrent readers and writers.\ntype AckCache[T AckCacheItem] interface {\n\t// Put stores an item in the cache with the specified size. Returns ErrAckCacheFull if the cache\n\t// is at capacity, or ErrAlreadyAcked if the item's sequence ID has already\n\t// been acknowledged. Silently ignores duplicate sequence IDs.\n\tPut(item T, size uint64) error\n\n\t// Get retrieves an item by its sequence ID. Returns the zero value of T\n\t// if the item is not found.\n\tGet(sequenceID int64) T\n\n\t// Ack acknowledges all items with sequence ID <= level, removing them\n\t// from the cache. Returns the total size of items that were removed and the count of items removed.\n\t// This is the primary mechanism for freeing cache space.\n\tAck(level int64) (uint64, int)\n\n\t// Size returns the current total byte size of all items in the cache.\n\tSize() uint64\n\n\t// Count returns the current number of items in the cache.\n\tCount() int\n}\n\n// BoundedAckCache is a heap-based implementation of AckCache that maintains\n// items in sequence ID order and rejects new items when capacity limits are reached.\ntype BoundedAckCache[T AckCacheItem] struct {\n\tmu sync.Mutex\n\n\tmaxCount dynamicproperties.IntPropertyFn\n\tmaxSize  dynamicproperties.IntPropertyFn\n\n\t// Heap maintains items in sequence ID order for efficient acknowledgment\n\torder sequenceHeap[T]\n\t// Map provides O(1) lookup by sequence ID\n\tcache map[int64]T\n\n\t// Track acknowledgment level and current usage\n\tlastAck  int64\n\tcurrSize uint64\n\n\tlogger        log.Logger\n\tbudgetManager Manager\n\tcacheID       string\n}\n\n// NewBoundedAckCache creates a new bounded ack cache with the specified capacity limits.\n//\n// Parameters:\n//   - maxCount: maximum number of items (dynamic property)\n//   - maxSize: maximum total byte size (dynamic property)\n//   - logger: optional logger for diagnostics (can be nil)\n//   - budgetManager: optional budget manager for host-level capacity tracking (can be nil)\n//   - cacheID: cache identifier for budget manager (required if budgetManager is provided)\n//\n// The cache will reject new items when either limit would be exceeded.\n// If a budget manager is provided, Put and Ack operations will automatically\n// reserve and release capacity through the budget manager.\nfunc NewBoundedAckCache[T AckCacheItem](\n\tmaxCount dynamicproperties.IntPropertyFn,\n\tmaxSize dynamicproperties.IntPropertyFn,\n\tlogger log.Logger,\n\tbudgetManager Manager,\n\tcacheID string,\n) AckCache[T] {\n\tinitialCount := maxCount()\n\treturn &BoundedAckCache[T]{\n\t\tmaxCount:      maxCount,\n\t\tmaxSize:       maxSize,\n\t\torder:         make(sequenceHeap[T], 0, initialCount),\n\t\tcache:         make(map[int64]T, initialCount),\n\t\tlogger:        logger,\n\t\tbudgetManager: budgetManager,\n\t\tcacheID:       cacheID,\n\t}\n}\n\n// Put stores an item in the cache with the specified size.\nfunc (c *BoundedAckCache[T]) Put(item T, size uint64) error {\n\tc.mu.Lock()\n\tdefer c.mu.Unlock()\n\n\tsequenceID := item.GetSequenceID()\n\n\t// Reject items that have already been acknowledged\n\tif c.lastAck >= sequenceID {\n\t\treturn ErrAlreadyAcked\n\t}\n\n\t// Silently ignore duplicate sequence IDs\n\tif _, exists := c.cache[sequenceID]; exists {\n\t\treturn nil\n\t}\n\n\t// Check capacity limits\n\tif (len(c.order) >= c.maxCount()) ||\n\t\t(c.currSize+size >= uint64(c.maxSize())) ||\n\t\t(c.currSize > math.MaxUint64-size) {\n\t\treturn ErrAckCacheFull\n\t}\n\n\tif c.budgetManager != nil {\n\t\treturn c.budgetManager.ReserveWithCallback(c.cacheID, size, 1, func() error {\n\t\t\treturn c.putInternal(item, sequenceID, size)\n\t\t})\n\t}\n\n\treturn c.putInternal(item, sequenceID, size)\n}\n\n// Get retrieves an item by sequence ID.\nfunc (c *BoundedAckCache[T]) Get(sequenceID int64) T {\n\tc.mu.Lock()\n\tdefer c.mu.Unlock()\n\n\treturn c.cache[sequenceID]\n}\n\n// Ack acknowledges all items with sequence ID <= level and returns total freed size and count of items removed.\nfunc (c *BoundedAckCache[T]) Ack(level int64) (uint64, int) {\n\tc.mu.Lock()\n\tdefer c.mu.Unlock()\n\n\tif c.budgetManager != nil {\n\t\tvar freedSize uint64\n\t\tvar removedCount int64\n\t\terr := c.budgetManager.ReleaseWithCallback(c.cacheID, func() (uint64, int64, error) {\n\t\t\tfreedSize, removedCount = c.ackInternal(level)\n\t\t\treturn freedSize, removedCount, nil\n\t\t})\n\t\tif err != nil {\n\t\t\treturn 0, 0\n\t\t}\n\t\treturn freedSize, int(removedCount)\n\t}\n\n\tfreedSize, removedCount := c.ackInternal(level)\n\treturn freedSize, int(removedCount)\n}\n\n// Size returns current total byte size.\nfunc (c *BoundedAckCache[T]) Size() uint64 {\n\tc.mu.Lock()\n\tdefer c.mu.Unlock()\n\n\treturn c.currSize\n}\n\n// Count returns current number of items.\nfunc (c *BoundedAckCache[T]) Count() int {\n\tc.mu.Lock()\n\tdefer c.mu.Unlock()\n\n\treturn len(c.order)\n}\n\n// putInternal adds an item to the cache. Caller must hold the lock.\nfunc (c *BoundedAckCache[T]) putInternal(item T, sequenceID int64, size uint64) error {\n\tc.cache[sequenceID] = item\n\theap.Push(&c.order, heapItem[T]{sequenceID: sequenceID, size: size})\n\tc.currSize += size\n\treturn nil\n}\n\n// ackInternal removes all items with sequence ID <= level. Caller must hold the lock.\nfunc (c *BoundedAckCache[T]) ackInternal(level int64) (uint64, int64) {\n\tvar freedSize uint64\n\tvar removedCount int64\n\n\tfor c.order.Len() > 0 && c.order.Peek().sequenceID <= level {\n\t\titem := heap.Pop(&c.order).(heapItem[T])\n\t\tdelete(c.cache, item.sequenceID)\n\t\tc.currSize -= item.size\n\t\tfreedSize += item.size\n\t\tremovedCount++\n\t}\n\n\tc.lastAck = level\n\treturn freedSize, removedCount\n}\n\n// heapItem represents an item in the sequence heap\ntype heapItem[T AckCacheItem] struct {\n\tsequenceID int64\n\tsize       uint64\n}\n\n// sequenceHeap implements heap.Interface for maintaining items in sequence order\ntype sequenceHeap[T AckCacheItem] []heapItem[T]\n\nfunc (h sequenceHeap[T]) Len() int           { return len(h) }\nfunc (h sequenceHeap[T]) Less(i, j int) bool { return h[i].sequenceID < h[j].sequenceID }\nfunc (h sequenceHeap[T]) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }\n\nfunc (h *sequenceHeap[T]) Push(x interface{}) {\n\t*h = append(*h, x.(heapItem[T]))\n}\n\nfunc (h *sequenceHeap[T]) Pop() interface{} {\n\told := *h\n\tn := len(old)\n\titem := old[n-1]\n\t*h = old[0 : n-1]\n\treturn item\n}\n\nfunc (h *sequenceHeap[T]) Peek() heapItem[T] {\n\treturn (*h)[0]\n}\n"
  },
  {
    "path": "common/cache/ack_cache_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cache\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\n// testItem implements AckCacheItem for testing\ntype testItem struct {\n\tsequenceID int64\n\tdata       string\n\tsize       uint64\n}\n\nfunc (t *testItem) GetSequenceID() int64 {\n\treturn t.sequenceID\n}\n\nfunc (t *testItem) ByteSize() uint64 {\n\tif t.size > 0 {\n\t\treturn t.size\n\t}\n\t// Default size based on data length\n\treturn uint64(len(t.data)) + 16 // 16 bytes overhead\n}\n\nfunc newTestItem(sequenceID int64, data string) *testItem {\n\treturn &testItem{\n\t\tsequenceID: sequenceID,\n\t\tdata:       data,\n\t}\n}\n\nfunc newTestItemWithSize(sequenceID int64, data string, size uint64) *testItem {\n\treturn &testItem{\n\t\tsequenceID: sequenceID,\n\t\tdata:       data,\n\t\tsize:       size,\n\t}\n}\n\nfunc TestBoundedAckCache_BasicOperations(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),   // max count\n\t\tdynamicproperties.GetIntPropertyFn(1000), // max size\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\t// Test empty cache\n\tassert.Equal(t, uint64(0), cache.Size())\n\tassert.Equal(t, 0, cache.Count())\n\tassert.Nil(t, cache.Get(100))\n\n\t// Test adding items\n\titem1 := newTestItem(100, \"item1\")\n\titem2 := newTestItem(200, \"item2\")\n\titem3 := newTestItem(150, \"item3\") // out of order\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, item1.ByteSize(), cache.Size())\n\tassert.Equal(t, 1, cache.Count())\n\tassert.Equal(t, item1, cache.Get(100))\n\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\tassert.Equal(t, item1.ByteSize()+item2.ByteSize(), cache.Size())\n\tassert.Equal(t, 2, cache.Count())\n\tassert.Equal(t, item2, cache.Get(200))\n\n\t// Test out-of-order insertion\n\trequire.NoError(t, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, item1.ByteSize()+item2.ByteSize()+item3.ByteSize(), cache.Size())\n\tassert.Equal(t, 3, cache.Count())\n\tassert.Equal(t, item3, cache.Get(150))\n\n\t// Test duplicate insertion (should be silently ignored)\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, 3, cache.Count()) // No change\n}\n\nfunc TestBoundedAckCache_Acknowledgment(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\titem1 := newTestItem(100, \"item1\")\n\titem2 := newTestItem(200, \"item2\")\n\titem3 := newTestItem(300, \"item3\")\n\titem4 := newTestItem(150, \"item4\")\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\trequire.NoError(t, cache.Put(item3, item3.ByteSize()))\n\trequire.NoError(t, cache.Put(item4, item4.ByteSize()))\n\n\ttotalSize := item1.ByteSize() + item2.ByteSize() + item3.ByteSize() + item4.ByteSize()\n\tassert.Equal(t, totalSize, cache.Size())\n\tassert.Equal(t, 4, cache.Count())\n\n\t// Ack at level 0 - should not remove anything\n\tfreedSize, removedCount := cache.Ack(0)\n\tassert.Equal(t, uint64(0), freedSize)\n\tassert.Equal(t, 0, removedCount)\n\tassert.Equal(t, 4, cache.Count())\n\tassert.Equal(t, totalSize, cache.Size())\n\n\t// Ack at level 100 - should remove item1\n\tfreedSize, removedCount = cache.Ack(100)\n\tassert.Equal(t, item1.ByteSize(), freedSize)\n\tassert.Equal(t, 1, removedCount)\n\tassert.Equal(t, 3, cache.Count())\n\tassert.Equal(t, totalSize-item1.ByteSize(), cache.Size())\n\tassert.Nil(t, cache.Get(100))\n\tassert.Equal(t, item4, cache.Get(150))\n\tassert.Equal(t, item2, cache.Get(200))\n\tassert.Equal(t, item3, cache.Get(300))\n\n\t// Ack at level 200 - should remove item4 and item2\n\tfreedSize, removedCount = cache.Ack(200)\n\tassert.Equal(t, item4.ByteSize()+item2.ByteSize(), freedSize)\n\tassert.Equal(t, 2, removedCount)\n\tassert.Equal(t, 1, cache.Count())\n\tassert.Equal(t, item3.ByteSize(), cache.Size())\n\tassert.Nil(t, cache.Get(100))\n\tassert.Nil(t, cache.Get(150))\n\tassert.Nil(t, cache.Get(200))\n\tassert.Equal(t, item3, cache.Get(300))\n\n\t// Ack at level 500 - should remove everything\n\tfreedSize, removedCount = cache.Ack(500)\n\tassert.Equal(t, item3.ByteSize(), freedSize)\n\tassert.Equal(t, 1, removedCount)\n\tassert.Equal(t, 0, cache.Count())\n\tassert.Equal(t, uint64(0), cache.Size())\n\tassert.Nil(t, cache.Get(300))\n}\n\nfunc TestBoundedAckCache_AlreadyAcked(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\titem1 := newTestItem(100, \"item1\")\n\titem2 := newTestItem(200, \"item2\")\n\titem3 := newTestItem(50, \"item3\") // lower than ack level\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\n\t// Ack at level 150\n\t_, _ = cache.Ack(150)\n\tassert.Equal(t, 1, cache.Count())\n\tassert.Nil(t, cache.Get(100))\n\tassert.Equal(t, item2, cache.Get(200))\n\n\t// Try to add item with sequence ID lower than ack level\n\tassert.Equal(t, ErrAlreadyAcked, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, 1, cache.Count()) // No change\n\n\t// Try to add item1 again (already acked)\n\tassert.Equal(t, ErrAlreadyAcked, cache.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, 1, cache.Count()) // No change\n}\n\nfunc TestBoundedAckCache_CountLimit(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(3),    // max count = 3\n\t\tdynamicproperties.GetIntPropertyFn(1000), // large size limit\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\titem1 := newTestItem(100, \"item1\")\n\titem2 := newTestItem(200, \"item2\")\n\titem3 := newTestItem(300, \"item3\")\n\titem4 := newTestItem(400, \"item4\")\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\trequire.NoError(t, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, 3, cache.Count())\n\n\t// Fourth item should be rejected\n\tassert.Equal(t, ErrAckCacheFull, cache.Put(item4, item4.ByteSize()))\n\tassert.Equal(t, 3, cache.Count())\n\tassert.Nil(t, cache.Get(400))\n\n\t// After acking, should be able to add more\n\t_, _ = cache.Ack(200)\n\tassert.Equal(t, 1, cache.Count())\n\trequire.NoError(t, cache.Put(item4, item4.ByteSize()))\n\tassert.Equal(t, 2, cache.Count())\n\tassert.Equal(t, item4, cache.Get(400))\n}\n\nfunc TestBoundedAckCache_SizeLimit(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),  // large count limit\n\t\tdynamicproperties.GetIntPropertyFn(100), // max size = 100\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\titem1 := newTestItemWithSize(100, \"item1\", 40)\n\titem2 := newTestItemWithSize(200, \"item2\", 40)\n\titem3 := newTestItemWithSize(300, \"item3\", 30) // would exceed size limit\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\tassert.Equal(t, uint64(80), cache.Size())\n\tassert.Equal(t, 2, cache.Count())\n\n\t// Third item should be rejected due to size\n\tassert.Equal(t, ErrAckCacheFull, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, uint64(80), cache.Size())\n\tassert.Equal(t, 2, cache.Count())\n\tassert.Nil(t, cache.Get(300))\n\n\t// After acking, should be able to add more\n\t_, _ = cache.Ack(150)\n\tassert.Equal(t, uint64(40), cache.Size())\n\tassert.Equal(t, 1, cache.Count())\n\trequire.NoError(t, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, uint64(70), cache.Size())\n\tassert.Equal(t, 2, cache.Count())\n}\n\nfunc TestBoundedAckCache_DynamicLimits(t *testing.T) {\n\tmaxCount := dynamicproperties.GetIntPropertyFn(2)\n\tmaxSize := dynamicproperties.GetIntPropertyFn(100)\n\n\tcache := NewBoundedAckCache[*testItem](maxCount, maxSize, log.NewNoop(), nil, \"\")\n\n\titem1 := newTestItem(100, \"item1\")\n\titem2 := newTestItem(200, \"item2\")\n\titem3 := newTestItem(300, \"item3\")\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\tassert.Equal(t, ErrAckCacheFull, cache.Put(item3, item3.ByteSize()))\n\n\t// Increase count limit dynamically\n\tmaxCount = dynamicproperties.GetIntPropertyFn(5)\n\tcache.(*BoundedAckCache[*testItem]).maxCount = maxCount\n\n\t// Now the third item should be accepted\n\trequire.NoError(t, cache.Put(item3, item3.ByteSize()))\n\tassert.Equal(t, 3, cache.Count())\n}\n\nfunc TestBoundedAckCache_ConcurrentOperations(t *testing.T) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\t// This test ensures no race conditions with concurrent access\n\tdone := make(chan bool, 3)\n\n\t// Writer goroutine\n\tgo func() {\n\t\tdefer func() { done <- true }()\n\t\tfor i := 0; i < 100; i++ {\n\t\t\titem := newTestItem(int64(i*10), \"item\")\n\t\t\tcache.Put(item, item.ByteSize())\n\t\t}\n\t}()\n\n\t// Reader goroutine\n\tgo func() {\n\t\tdefer func() { done <- true }()\n\t\tfor i := 0; i < 100; i++ {\n\t\t\tcache.Get(int64(i * 10))\n\t\t\tcache.Size()\n\t\t\tcache.Count()\n\t\t}\n\t}()\n\n\t// Acker goroutine\n\tgo func() {\n\t\tdefer func() { done <- true }()\n\t\tfor i := 0; i < 10; i++ {\n\t\t\t_, _ = cache.Ack(int64(i * 100))\n\t\t}\n\t}()\n\n\t// Wait for all goroutines\n\tfor i := 0; i < 3; i++ {\n\t\t<-done\n\t}\n\n\t// Final ack to clean up\n\t_, _ = cache.Ack(1000)\n\tassert.Equal(t, 0, cache.Count())\n\tassert.Equal(t, uint64(0), cache.Size())\n}\n\nfunc TestBoundedAckCache_WithBudgetManager(t *testing.T) {\n\tbudgetMgr := NewBudgetManager(\n\t\t\"test-budget\",\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tdynamicproperties.GetIntPropertyFn(100),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\tlog.NewNoop(),\n\t\tnil,\n\t)\n\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tbudgetMgr,\n\t\t\"cache1\",\n\t)\n\n\titem1 := newTestItemWithSize(100, \"item1\", 100)\n\titem2 := newTestItemWithSize(200, \"item2\", 150)\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, uint64(100), budgetMgr.UsedBytes())\n\tassert.Equal(t, int64(1), budgetMgr.UsedCount())\n\n\trequire.NoError(t, cache.Put(item2, item2.ByteSize()))\n\tassert.Equal(t, uint64(250), budgetMgr.UsedBytes())\n\tassert.Equal(t, int64(2), budgetMgr.UsedCount())\n\n\tfreedSize, freedCount := cache.Ack(100)\n\tassert.Equal(t, uint64(100), freedSize)\n\tassert.Equal(t, 1, freedCount)\n\tassert.Equal(t, uint64(150), budgetMgr.UsedBytes())\n\tassert.Equal(t, int64(1), budgetMgr.UsedCount())\n\n\tfreedSize, freedCount = cache.Ack(200)\n\tassert.Equal(t, uint64(150), freedSize)\n\tassert.Equal(t, 1, freedCount)\n\tassert.Equal(t, uint64(0), budgetMgr.UsedBytes())\n\tassert.Equal(t, int64(0), budgetMgr.UsedCount())\n}\n\nfunc TestBoundedAckCache_WithBudgetManager_BudgetExceeded(t *testing.T) {\n\tbudgetMgr := NewBudgetManager(\n\t\t\"test-budget\",\n\t\tdynamicproperties.GetIntPropertyFn(200),\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\tlog.NewNoop(),\n\t\tnil,\n\t)\n\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tbudgetMgr,\n\t\t\"cache1\",\n\t)\n\n\titem1 := newTestItemWithSize(100, \"item1\", 150)\n\titem2 := newTestItemWithSize(200, \"item2\", 100)\n\n\trequire.NoError(t, cache.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, uint64(150), budgetMgr.UsedBytes())\n\n\terr := cache.Put(item2, item2.ByteSize())\n\tassert.Equal(t, ErrBytesBudgetExceeded, err)\n\tassert.Equal(t, uint64(150), budgetMgr.UsedBytes())\n\tassert.Equal(t, int64(1), budgetMgr.UsedCount())\n\tassert.Equal(t, 1, cache.Count())\n\tassert.Nil(t, cache.Get(200))\n}\n\nfunc TestBoundedAckCache_WithBudgetManager_MultipleCaches(t *testing.T) {\n\tbudgetMgr := NewBudgetManager(\n\t\t\"test-budget\",\n\t\tdynamicproperties.GetIntPropertyFn(500),\n\t\tdynamicproperties.GetIntPropertyFn(100),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\tlog.NewNoop(),\n\t\tnil,\n\t)\n\n\tcache1 := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tbudgetMgr,\n\t\t\"cache1\",\n\t)\n\n\tcache2 := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tlog.NewNoop(),\n\t\tbudgetMgr,\n\t\t\"cache2\",\n\t)\n\n\titem1 := newTestItemWithSize(100, \"item1\", 200)\n\titem2 := newTestItemWithSize(200, \"item2\", 200)\n\n\trequire.NoError(t, cache1.Put(item1, item1.ByteSize()))\n\tassert.Equal(t, uint64(200), budgetMgr.UsedBytes())\n\n\trequire.NoError(t, cache2.Put(item2, item2.ByteSize()))\n\tassert.Equal(t, uint64(400), budgetMgr.UsedBytes())\n\n\tfreedSize, freedCount := cache1.Ack(100)\n\tassert.Equal(t, uint64(200), freedSize)\n\tassert.Equal(t, 1, freedCount)\n\tassert.Equal(t, uint64(200), budgetMgr.UsedBytes())\n\n\tfreedSize, freedCount = cache2.Ack(200)\n\tassert.Equal(t, uint64(200), freedSize)\n\tassert.Equal(t, 1, freedCount)\n\tassert.Equal(t, uint64(0), budgetMgr.UsedBytes())\n}\n\nfunc BenchmarkBoundedAckCache(b *testing.B) {\n\tcache := NewBoundedAckCache[*testItem](\n\t\tdynamicproperties.GetIntPropertyFn(10000),\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\t\"\",\n\t)\n\n\t// Pre-populate cache\n\tfor i := 0; i < 5000; i++ {\n\t\titem := newTestItem(int64(i*100), \"benchmark_item\")\n\t\tcache.Put(item, item.ByteSize())\n\t}\n\n\tb.ResetTimer()\n\tfor n := 0; n < b.N; n++ {\n\t\tsequenceID := int64(n * 100)\n\t\tcache.Get(sequenceID)\n\t\titem := newTestItem(int64(n*100+500000), \"new_item\")\n\t\tcache.Put(item, item.ByteSize())\n\t\tif n%100 == 0 {\n\t\t\t_, _ = cache.Ack(sequenceID)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/cache/budget.go",
    "content": "package cache\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// Manager implements a generic host-scoped budget manager suitable for\n// both non-evicting caches (admission/backpressure) and evicting caches (e.g., LRU).\n//\n// Fair Share Logic:\n// The budget manager implements a two-tier soft cap system to fairly distribute capacity across\n// multiple active caches while preventing any single cache from monopolizing resources:\n//   - Free space tier: (threshold * capacity) - shared by all caches on a first-come basis\n//   - Fair share tier: ((1 - threshold) * capacity) / activeCaches - allocated per active cache\n//\n// An active cache is one that currently has non-zero usage (usedBytes > 0 or usedCount > 0).\n// This ensures that when multiple caches are active, each gets a guaranteed fair share of capacity,\n// preventing scenarios where one cache consumes all available space and starves others.\n//\n// There are two admission modes available:\n//   - AdmissionOptimistic => add-then-undo on failure; may briefly overshoot limits\n//   - AdmissionStrict     => pre-check with CAS; never overshoots limits\n//\n// Note: Soft cap admission is always optimistic regardless of the configured mode.\n//\n// There are two modes available for reclaiming memory from the cache:\n//   - ReserveOrReclaimSelfRelease    => reclaim does its own Release() calls (per-item or per-batch).\n//   - ReserveOrReclaimManagerRelease => reclaim returns totals; manager calls Release() once.\n//\n// Pick the reclaim mode that fits the cache's eviction style (self-release vs manager-release)\n// and admission mode based on whether temporary overshooting is acceptable (Optimistic vs Strict).\n//\n// How to use it:\n//\n// Manager is a means to track cache size across multiple caches at the Module or Host level.\n// It solves the problem of accounting memory for multiple caches working with a limited total amount of memory\n// in a centralized way, allowing each cache to operate independently while respecting global limits.\n//\n// The cache implementation should use the callback-based methods to ensure proper budget tracking\n// with automatic cleanup on errors.\n//\n// Example usage - Adding an item to the cache:\n//\n//\tfunc (c *myCache) Put(key, value interface{}) error {\n//\t\titemSizeBytes := calculateSize(value)\n//\t\treturn c.budgetMgr.ReserveWithCallback(c.cacheID, itemSizeBytes, 1, func() error {\n//\t\t\treturn c.cache.Put(key, value)\n//\t\t})\n//\t}\n//\n// Example usage - Removing an item from the cache:\n//\n//\tfunc (c *myCache) Delete(key interface{}) error {\n//\t\treturn c.budgetMgr.ReleaseWithCallback(c.cacheID, func() (uint64, int64, error) {\n//\t\t\tbytes, count, err := c.cache.Delete(key)\n//\t\t\treturn bytes, count, err\n//\t\t})\n//\t}\n//\n// Example usage - Adding with automatic eviction (manager-release pattern):\n//\n//\tfunc (c *myCache) PutWithEviction(ctx context.Context, key, value interface{}) error {\n//\t\titemSizeBytes := calculateSize(value)\n//\t\treturn c.budgetMgr.ReserveOrReclaimManagerReleaseWithCallback(\n//\t\t\tctx, c.cacheID, itemSizeBytes, 1, true,\n//\t\t\tfunc(needBytes uint64, needCount int64) (uint64, int64) {\n//\t\t\t\treturn c.cache.EvictLRU(needBytes, needCount)\n//\t\t\t},\n//\t\t\tfunc() error {\n//\t\t\t\treturn c.cache.Put(key, value)\n//\t\t\t},\n//\t\t)\n//\t}\n//\n// Example usage - Adding with automatic eviction (self-release pattern):\n//\n//\tfunc (c *myCache) PutWithEviction(ctx context.Context, key, value interface{}) error {\n//\t\titemSizeBytes := calculateSize(value)\n//\t\treturn c.budgetMgr.ReserveOrReclaimSelfReleaseWithCallback(\n//\t\t\tctx, c.cacheID, itemSizeBytes, 1, true,\n//\t\t\tfunc(needBytes uint64, needCount int64) {\n//\t\t\t\t// Evict items one-by-one until we've freed enough\n//\t\t\t\tvar freedBytes, freedCount uint64, int64\n//\t\t\t\tfor freedBytes < needBytes || freedCount < needCount {\n//\t\t\t\t\tevictedBytes, evictedCount, err := c.cache.EvictOldest()\n//\t\t\t\t\tif err != nil {\n//\t\t\t\t\t\tbreak\n//\t\t\t\t\t}\n//\t\t\t\t\tc.budgetMgr.ReleaseForCache(c.cacheID, evictedBytes, evictedCount)\n//\t\t\t\t\tfreedBytes += evictedBytes\n//\t\t\t\t\tfreedCount += evictedCount\n//\t\t\t\t}\n//\t\t\t},\n//\t\t\tfunc() error {\n//\t\t\t\treturn c.cache.Put(key, value)\n//\t\t\t},\n//\t\t)\n//\t}\ntype Manager interface {\n\t// Callback-based reserve methods (reserve -> callback -> release on error)\n\n\t// ReserveWithCallback reserves capacity, executes callback, releases on callback error.\n\tReserveWithCallback(cacheID string, nBytes uint64, nCount int64, callback func() error) error\n\t// ReserveBytesWithCallback reserves bytes, executes callback, releases on callback error.\n\tReserveBytesWithCallback(cacheID string, nBytes uint64, callback func() error) error\n\t// ReserveCountWithCallback reserves count, executes callback, releases on callback error.\n\tReserveCountWithCallback(cacheID string, nCount int64, callback func() error) error\n\n\t// Callback-based release methods (callback -> release on success)\n\n\t// ReleaseWithCallback executes callback, releases capacity if callback succeeds.\n\tReleaseWithCallback(cacheID string, callback func() (nBytes uint64, nCount int64, err error)) error\n\t// ReleaseBytesWithCallback executes callback, releases bytes if callback succeeds.\n\tReleaseBytesWithCallback(cacheID string, callback func() (freedBytes uint64, err error)) error\n\t// ReleaseCountWithCallback executes callback, releases count if callback succeeds.\n\tReleaseCountWithCallback(cacheID string, callback func() (freedCount int64, err error)) error\n\n\t// Callback-based reclaim methods\n\n\t// ReserveOrReclaimSelfReleaseWithCallback reserves/reclaims capacity, executes callback, releases on callback error.\n\t// todo: make sure to test before implementing caches that use this, for example LRU caches. Tests have only been done in unit tests\n\tReserveOrReclaimSelfReleaseWithCallback(ctx context.Context, cacheID string, nBytes uint64, nCount int64, retriable bool, reclaim ReclaimSelfRelease, callback func() error) error\n\n\t// ReserveOrReclaimManagerReleaseWithCallback reserves/reclaims capacity, executes callback, releases on callback error.\n\t// todo: make sure to test before implementing caches that use this, for example LRU caches. Tests have only been done in unit tests\n\tReserveOrReclaimManagerReleaseWithCallback(ctx context.Context, cacheID string, nBytes uint64, nCount int64, retriable bool, reclaim ReclaimManagerRelease, callback func() error) error\n\n\t// UsedBytes returns current used bytes.\n\tUsedBytes() uint64\n\t// UsedCount returns current used count.\n\tUsedCount() int64\n\t// CapacityBytes returns the current effective bytes capacity (math.MaxUint64 means \"unlimited\").\n\tCapacityBytes() uint64\n\t// CapacityCount returns the current effective count capacity (math.MaxInt64 means \"unlimited\").\n\tCapacityCount() int64\n\t// Stop stops the metrics ticker and releases resources.\n\tStop()\n}\n\ntype AdmissionMode uint8\n\nconst (\n\tAdmissionOptimistic AdmissionMode = iota // add-then-undo; may overshoot briefly\n\tAdmissionStrict                          // CAS pre-check; never overshoots\n)\n\nconst (\n\tbudgetTypeBytes = \"bytes\"\n\tbudgetTypeCount = \"count\"\n)\n\nvar (\n\tErrBytesBudgetExceeded        = errors.New(\"bytes budget exceeded\")\n\tErrCountBudgetExceeded        = errors.New(\"count budget exceeded\")\n\tErrBytesSoftCapExceeded       = errors.New(\"bytes soft cap threshold exceeded\")\n\tErrCountSoftCapExceeded       = errors.New(\"count soft cap threshold exceeded\")\n\tErrOverflow                   = errors.New(\"budget counter overflow\")\n\tErrInvalidValue               = errors.New(\"invalid negative reserve value\")\n\tErrInsufficientUsageToReclaim = errors.New(\"insufficient usage to reclaim\")\n\tErrInvalidRequest             = errors.New(\"invalid request: both additionalBytes and additionalCount are zero\")\n)\n\ntype ReclaimSelfRelease func(needBytes uint64, needCount int64)\n\n// ReclaimManagerRelease is used when the manager should call Release once with totals.\n// IMPORTANT: Do NOT call mgr.Release(...) inside this callback, or you will double-release.\ntype ReclaimManagerRelease func(needBytes uint64, needCount int64) (freedBytes uint64, freedCount int64)\n\n// CapEnforcementResult contains the result of capacity enforcement (both hard and soft cap) for a cache\ntype CapEnforcementResult struct {\n\tFreeBytes      uint64 // bytes allocated from free capacity (on success only)\n\tFairShareBytes uint64 // bytes allocated from fair share capacity (on success only)\n\tFreeCount      int64  // count allocated from free capacity (on success only)\n\tFairShareCount int64  // count allocated from fair share capacity (on success only)\n\tAvailableBytes uint64 // total bytes available for this cache (min of soft cap and hard cap constraints)\n\tAvailableCount int64  // total count available for this cache (min of soft cap and hard cap constraints)\n}\n\n// Usage tracks usage for a specific cache\ntype Usage struct {\n\tmu sync.Mutex // protects all fields below\n\n\tusedBytes uint64 // total bytes used by this cache\n\tusedCount int64  // total count used by this cache\n\n\t// Capacity type breakdown for fairness strategies\n\tfairShareCapacityBytes uint64 // bytes from (1-threshold) portion\n\tfreeCapacityBytes      uint64 // bytes from threshold portion\n\tfairShareCapacityCount int64  // count from (1-threshold) portion\n\tfreeCapacityCount      int64  // count from threshold portion\n}\n\ntype manager struct {\n\tname                        string\n\tmaxBytes                    dynamicproperties.IntPropertyFn\n\tmaxCount                    dynamicproperties.IntPropertyFn\n\treclaimBackoff              time.Duration\n\tadmission                   AdmissionMode\n\tscope                       metrics.Scope\n\tlogger                      log.Logger // Rate-limited logger (1 RPS)\n\tenforcementSoftCapThreshold dynamicproperties.FloatPropertyFn\n\n\tusedBytes uint64 // atomic - total usage across all caches\n\tusedCount int64  // atomic - total usage across all caches\n\n\t// Per-cache usage tracking using sync.Map for lock-free operations\n\tcacheUsage sync.Map // map[string]*Usage\n\n\t// Optimistic tracking of active caches (usage > 0)\n\tactiveCacheCount int64 // atomic\n\n\t// Metrics ticker\n\tmetricsTicker *time.Ticker\n\tmetricsStop   chan struct{}\n}\n\n// NewBudgetManager creates a new Manager.\n//\n// Parameters:\n//   - name: manager name for identification\n//   - maxBytes: bytes capacity semantics:\n//     < 0  => enforcement disabled (unlimited; still tracked)\n//     == 0 => caching fully disabled (deny all reserves)\n//     > 0  => enforcement enabled with that capacity\n//   - maxCount: count capacity semantics (same as maxBytes)\n//   - admission: AdmissionOptimistic (add-then-undo) or AdmissionStrict (CAS pre-check)\n//   - reclaimBackoff: optional sleep between reclaim attempts (defaults to ~100µs if zero)\n//   - scope: metrics scope for emitting metrics (can be nil)\n//   - logger: logger for diagnostic messages (can be nil)\n//   - softCapThreshold: optional percentage (0.0-1.0) defining how capacity is split between two tiers:\n//   - Free space tier: (threshold * capacity) - shared by all caches\n//   - Fair share tier: ((1 - threshold) * capacity) / activeCaches - allocated per cache\n//     The fair share capacity is equal to the fair share capacity ((1 - threshold) * capacity) divided by the number\n//     of active caches (caches with usage > 0).\nfunc NewBudgetManager(\n\tname string,\n\tmaxBytes dynamicproperties.IntPropertyFn,\n\tmaxCount dynamicproperties.IntPropertyFn,\n\tadmission AdmissionMode,\n\treclaimBackoff time.Duration,\n\tscope metrics.Scope,\n\tlogger log.Logger,\n\tsoftCapThreshold dynamicproperties.FloatPropertyFn, // nil defaults to 1.0 (disabled)\n) Manager {\n\tif scope == nil {\n\t\tscope = metrics.NoopScope\n\t}\n\tif maxBytes == nil {\n\t\tmaxBytes = dynamicproperties.GetIntPropertyFn(-1) // unlimited if maxBytes is not provided\n\t}\n\tif maxCount == nil {\n\t\tmaxCount = dynamicproperties.GetIntPropertyFn(-1) // unlimited if maxCount is not provided\n\t}\n\tif softCapThreshold == nil {\n\t\tsoftCapThreshold = dynamicproperties.GetFloatPropertyFn(1.0) // no threshold if softCapThreshold is not provided\n\t}\n\tif admission == 0 {\n\t\tadmission = AdmissionOptimistic\n\t}\n\n\t// Create throttled logger (1 log per second)\n\tvar throttledLogger log.Logger\n\tif logger != nil {\n\t\tthrottledLogger = log.NewThrottledLogger(logger, dynamicproperties.GetIntPropertyFn(1))\n\t}\n\n\tlogger.Info(\"Budget manager started\", tag.Name(name))\n\n\tmgr := &manager{\n\t\tname:                        name,\n\t\tmaxBytes:                    maxBytes,\n\t\tmaxCount:                    maxCount,\n\t\treclaimBackoff:              reclaimBackoff,\n\t\tadmission:                   admission,\n\t\tscope:                       scope.Tagged(metrics.BudgetManagerNameTag(name)),\n\t\tlogger:                      throttledLogger,\n\t\tenforcementSoftCapThreshold: softCapThreshold,\n\t\tmetricsTicker:               time.NewTicker(10 * time.Second),\n\t\tmetricsStop:                 make(chan struct{}),\n\t}\n\n\t// Emit initial metrics\n\tmgr.updateMetrics()\n\n\t// Start metrics update goroutine\n\tgo mgr.metricsLoop()\n\n\treturn mgr\n}\n\n// Stop stops the metrics ticker and releases resources\nfunc (m *manager) Stop() {\n\tm.metricsTicker.Stop()\n\tclose(m.metricsStop)\n}\n\n// metricsLoop periodically updates metrics every 10 seconds\nfunc (m *manager) metricsLoop() {\n\tfor {\n\t\tselect {\n\t\tcase <-m.metricsTicker.C:\n\t\t\tm.updateMetrics()\n\t\tcase <-m.metricsStop:\n\t\t\treturn\n\t\t}\n\t}\n}\n\n// updateMetrics emits current state metrics\nfunc (m *manager) updateMetrics() {\n\t// Skip metrics if both capacity limits are disabled (set to 0)\n\tif m.maxBytes() == 0 && m.maxCount() == 0 {\n\t\treturn\n\t}\n\n\t// Emit capacity metrics\n\tcapacityBytes := m.CapacityBytes()\n\t// Only emit if not unlimited\n\tif capacityBytes != math.MaxUint64 {\n\t\tm.scope.UpdateGauge(metrics.BudgetManagerCapacityBytes, float64(capacityBytes))\n\t}\n\n\tcapacityCount := m.CapacityCount()\n\t// Only emit if not unlimited\n\tif capacityCount != math.MaxInt64 {\n\t\tm.scope.UpdateGauge(metrics.BudgetManagerCapacityCount, float64(capacityCount))\n\t}\n\n\t// Emit usage metrics\n\tm.scope.UpdateGauge(metrics.BudgetManagerUsedBytes, float64(m.UsedBytes()))\n\tm.scope.UpdateGauge(metrics.BudgetManagerUsedCount, float64(m.UsedCount()))\n\n\t// Emit soft threshold\n\tm.scope.UpdateGauge(metrics.BudgetManagerSoftThreshold, m.enforcementSoftCapThreshold())\n\n\t// Emit active cache count\n\tm.scope.UpdateGauge(metrics.BudgetManagerActiveCacheCount, float64(atomic.LoadInt64(&m.activeCacheCount)))\n}\n\n// emitHardCapExceeded logs when hard capacity limit is exceeded and increments counter\nfunc (m *manager) emitHardCapExceeded(cacheID, budgetType string, requested, available uint64) {\n\t// Skip metrics if both capacity limits are disabled (set to 0)\n\tif m.maxBytes() == 0 && m.maxCount() == 0 {\n\t\treturn\n\t}\n\n\tif m.scope != nil {\n\t\tm.scope.IncCounter(metrics.BudgetManagerHardCapExceeded)\n\t}\n\n\tif m.logger != nil {\n\t\tm.logger.Debug(\"Hard capacity limit exceeded\",\n\t\t\ttag.Name(m.name),\n\t\t\ttag.CacheID(cacheID),\n\t\t\ttag.Value(budgetType),\n\t\t\ttag.Counter(int(requested)),\n\t\t\ttag.Number(int64(available)),\n\t\t)\n\t}\n}\n\n// emitSoftCapExceeded logs when soft cap is exceeded and increments counter\nfunc (m *manager) emitSoftCapExceeded(cacheID, budgetType string, requested, available uint64) {\n\t// Skip metrics if both capacity limits are disabled (set to 0)\n\tif m.maxBytes() == 0 && m.maxCount() == 0 {\n\t\treturn\n\t}\n\n\tif m.scope != nil {\n\t\tm.scope.IncCounter(metrics.BudgetManagerSoftCapExceeded)\n\t}\n\n\tif m.logger != nil {\n\t\tm.logger.Debug(\"Soft capacity limit exceeded\",\n\t\t\ttag.Name(m.name),\n\t\t\ttag.CacheID(cacheID),\n\t\t\ttag.Value(budgetType),\n\t\t\ttag.Counter(int(requested)),\n\t\t\ttag.Number(int64(available)),\n\t\t)\n\t}\n}\n\n// emitCapacityExceeded emits appropriate metrics based on the error type\nfunc (m *manager) emitCapacityExceeded(cacheID string, err error, requestedBytes uint64, requestedCount int64, capResult CapEnforcementResult) {\n\t// Skip metrics if both capacity limits are disabled (set to 0)\n\tif m.maxBytes() == 0 && m.maxCount() == 0 {\n\t\treturn\n\t}\n\n\tswitch err {\n\tcase ErrBytesBudgetExceeded:\n\t\tm.emitHardCapExceeded(cacheID, budgetTypeBytes, requestedBytes, capResult.AvailableBytes)\n\tcase ErrCountBudgetExceeded:\n\t\tm.emitHardCapExceeded(cacheID, budgetTypeCount, uint64(requestedCount), uint64(capResult.AvailableCount))\n\tcase ErrBytesSoftCapExceeded:\n\t\tm.emitSoftCapExceeded(cacheID, budgetTypeBytes, requestedBytes, capResult.AvailableBytes)\n\tcase ErrCountSoftCapExceeded:\n\t\tm.emitSoftCapExceeded(cacheID, budgetTypeCount, uint64(requestedCount), uint64(capResult.AvailableCount))\n\t}\n}\n\n// Cache-aware Reserve methods implementing two-tier soft cap logic\n\nfunc (m *manager) ReserveForCache(cacheID string, nBytes uint64, nCount int64) error {\n\t// Check capacity constraints (both hard and soft cap)\n\tcapResult, err := m.enforceCapForCache(cacheID, nBytes, nCount)\n\tif err != nil {\n\t\tm.emitCapacityExceeded(cacheID, err, nBytes, nCount, capResult)\n\t\treturn err\n\t}\n\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\n\tif nBytes > 0 {\n\t\tif err := m.reserveBytes(nBytes); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tcacheUsage.usedBytes += nBytes\n\t\tcacheUsage.fairShareCapacityBytes += capResult.FairShareBytes\n\t\tcacheUsage.freeCapacityBytes += capResult.FreeBytes\n\t}\n\tif nCount > 0 {\n\t\tif err := m.reserveCount(nCount); err != nil {\n\t\t\tif nBytes > 0 {\n\t\t\t\tm.releaseBytes(nBytes)\n\t\t\t\tcacheUsage.usedBytes -= nBytes\n\t\t\t\tcacheUsage.fairShareCapacityBytes -= capResult.FairShareBytes\n\t\t\t\tcacheUsage.freeCapacityBytes -= capResult.FreeBytes\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\tcacheUsage.usedCount += nCount\n\t\tcacheUsage.fairShareCapacityCount += capResult.FairShareCount\n\t\tcacheUsage.freeCapacityCount += capResult.FreeCount\n\t}\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif !wasActive && isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, 1)\n\t}\n\n\treturn nil\n}\n\nfunc (m *manager) ReserveBytesForCache(cacheID string, nBytes uint64) error {\n\t// Check capacity constraints (both hard and soft cap)\n\tcapResult, err := m.enforceCapForCache(cacheID, nBytes, 0)\n\tif err != nil {\n\t\tm.emitCapacityExceeded(cacheID, err, nBytes, 0, capResult)\n\t\treturn err\n\t}\n\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\n\tif err := m.reserveBytes(nBytes); err != nil {\n\t\treturn err\n\t}\n\n\tcacheUsage.usedBytes += nBytes\n\tcacheUsage.fairShareCapacityBytes += capResult.FairShareBytes\n\tcacheUsage.freeCapacityBytes += capResult.FreeBytes\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif !wasActive && isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, 1)\n\t}\n\n\treturn nil\n}\n\nfunc (m *manager) ReserveCountForCache(cacheID string, nCount int64) error {\n\tif nCount < 0 {\n\t\tif m.logger != nil {\n\t\t\tm.logger.Error(\"Invalid negative count value in ReserveCountForCache\",\n\t\t\t\ttag.CacheID(cacheID),\n\t\t\t\ttag.Key(\"requested\"), tag.Value(nCount),\n\t\t\t)\n\t\t}\n\t\treturn ErrInvalidValue\n\t}\n\t// Check capacity constraints (both hard and soft cap)\n\tcapResult, err := m.enforceCapForCache(cacheID, 0, nCount)\n\tif err != nil {\n\t\tm.emitCapacityExceeded(cacheID, err, 0, nCount, capResult)\n\t\treturn err\n\t}\n\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\n\tif err := m.reserveCount(nCount); err != nil {\n\t\treturn err\n\t}\n\n\tcacheUsage.usedCount += nCount\n\tcacheUsage.fairShareCapacityCount += capResult.FairShareCount\n\tcacheUsage.freeCapacityCount += capResult.FreeCount\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif !wasActive && isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, 1)\n\t}\n\n\treturn nil\n}\n\nfunc (m *manager) reserveBytes(nBytes uint64) error {\n\tswitch m.admission {\n\tcase AdmissionStrict:\n\t\treturn m.reserveBytesStrict(nBytes)\n\tdefault:\n\t\treturn m.reserveBytesOptimistic(nBytes)\n\t}\n}\n\nfunc (m *manager) reserveCount(nCount int64) error {\n\tswitch m.admission {\n\tcase AdmissionStrict:\n\t\treturn m.reserveCountStrict(nCount)\n\tdefault:\n\t\treturn m.reserveCountOptimistic(nCount)\n\t}\n}\n\nfunc (m *manager) reserveBytesStrict(nBytes uint64) error {\n\tcapB := m.CapacityBytes()\n\tfor {\n\t\told := atomic.LoadUint64(&m.usedBytes)\n\n\t\tif old > math.MaxUint64-nBytes {\n\t\t\tif m.logger != nil {\n\t\t\t\tm.logger.Error(\"Bytes budget overflow detected in strict mode\",\n\t\t\t\t\ttag.Key(\"requested\"), tag.Value(nBytes),\n\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(old),\n\t\t\t\t)\n\t\t\t}\n\t\t\treturn ErrOverflow\n\t\t}\n\t\tif old+nBytes > capB {\n\t\t\tm.emitHardCapExceeded(\"\", budgetTypeBytes, old+nBytes, capB)\n\t\t\treturn ErrBytesBudgetExceeded\n\t\t}\n\t\tif atomic.CompareAndSwapUint64(&m.usedBytes, old, old+nBytes) {\n\t\t\treturn nil\n\t\t}\n\t\t// retry on contention\n\t}\n}\n\nfunc (m *manager) reserveBytesOptimistic(nBytes uint64) error {\n\tcapB := m.CapacityBytes()\n\tnewVal := atomic.AddUint64(&m.usedBytes, nBytes)\n\tif newVal < nBytes { // wrap-around\n\t\tm.releaseBytes(nBytes)\n\t\tif m.logger != nil {\n\t\t\tm.logger.Debug(\"Bytes budget overflow detected\",\n\t\t\t\ttag.Key(\"requested\"), tag.Value(nBytes),\n\t\t\t\ttag.Key(\"previous-used\"), tag.Value(newVal-nBytes),\n\t\t\t)\n\t\t}\n\t\treturn ErrOverflow\n\t}\n\tif newVal <= capB {\n\t\treturn nil\n\t}\n\tm.releaseBytes(nBytes)\n\tm.emitHardCapExceeded(\"\", budgetTypeBytes, newVal, capB)\n\treturn ErrBytesBudgetExceeded\n}\n\nfunc (m *manager) reserveCountStrict(nCount int64) error {\n\tcapC := m.CapacityCount()\n\tfor {\n\t\told := atomic.LoadInt64(&m.usedCount)\n\n\t\tif old > math.MaxInt64-nCount {\n\t\t\tif m.logger != nil {\n\t\t\t\tm.logger.Error(\"Count budget overflow detected in strict mode\",\n\t\t\t\t\ttag.Key(\"requested\"), tag.Value(nCount),\n\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(old),\n\t\t\t\t)\n\t\t\t}\n\t\t\treturn ErrOverflow\n\t\t}\n\t\tif old+nCount > capC {\n\t\t\tm.emitHardCapExceeded(\"\", budgetTypeCount, uint64(old+nCount), uint64(capC))\n\t\t\treturn ErrCountBudgetExceeded\n\t\t}\n\t\tif atomic.CompareAndSwapInt64(&m.usedCount, old, old+nCount) {\n\t\t\treturn nil\n\t\t}\n\t\t// retry on contention\n\t}\n}\n\nfunc (m *manager) reserveCountOptimistic(nCount int64) error {\n\tcapC := m.CapacityCount()\n\tnewVal := atomic.AddInt64(&m.usedCount, nCount)\n\tif newVal < 0 { // wrap-around\n\t\tm.releaseCount(nCount)\n\t\treturn ErrOverflow\n\t}\n\tif newVal <= capC {\n\t\treturn nil\n\t}\n\tm.releaseCount(nCount)\n\tm.emitHardCapExceeded(\"\", budgetTypeCount, uint64(newVal), uint64(capC))\n\treturn ErrCountBudgetExceeded\n}\n\n// reclaimNeeds holds the calculated reclaim requirements\ntype reclaimNeeds struct {\n\tneedBytes uint64\n\tneedCount int64\n}\n\n// tryReserveOrCalculateReclaimNeeds attempts to reserve capacity immediately.\n// If capacity is not available, it calculates how much the cache needs to reclaim\n// and validates that the cache has enough usage to satisfy the reclaim requirement.\n// Returns:\n// - nil, nil if reservation succeeded immediately\n// - reclaimNeeds, nil if reclaim is needed and cache has sufficient usage\n// - nil, error if reservation failed and cannot be satisfied via reclaim\nfunc (m *manager) tryReserveOrCalculateReclaimNeeds(cacheID string, nBytes uint64, nCount int64) (*reclaimNeeds, error) {\n\t// Check capacity constraints (both soft and hard cap)\n\tcapResult, err := m.enforceCapForCache(cacheID, nBytes, nCount)\n\tif err == nil {\n\t\t// Capacity available - try to reserve\n\t\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err == nil {\n\t\t\treturn nil, nil\n\t\t}\n\t\t// lost the race; caller should retry\n\t\treturn nil, err\n\t}\n\n\t// Calculate reclaim amounts based on available capacity\n\tallowedBytes := capResult.AvailableBytes\n\tallowedCount := capResult.AvailableCount\n\tneedB := nBytes - allowedBytes\n\tneedC := nCount - allowedCount\n\n\t// Check if this cache can reclaim enough to satisfy the constraint\n\tcacheUsage := m.getCacheUsage(cacheID)\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\tcurrentCacheBytes := cacheUsage.usedBytes\n\tcurrentCacheCount := cacheUsage.usedCount\n\n\tif (currentCacheBytes+allowedBytes < nBytes) || (currentCacheCount+allowedCount < nCount) {\n\t\t// Cache doesn't have enough to reclaim\n\t\treturn nil, ErrInsufficientUsageToReclaim\n\t}\n\n\treturn &reclaimNeeds{\n\t\tneedBytes: needB,\n\t\tneedCount: needC,\n\t}, nil\n}\n\nfunc (m *manager) ReserveOrReclaimSelfRelease(\n\tctx context.Context,\n\tcacheID string,\n\tnBytes uint64,\n\tnCount int64,\n\tretriable bool,\n\treclaim ReclaimSelfRelease,\n) error {\n\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err == nil {\n\t\treturn nil\n\t} else if !retriable {\n\t\treturn err // single-shot; no reclaim\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn ctx.Err()\n\t\tdefault:\n\t\t}\n\n\t\tneeds, err := m.tryReserveOrCalculateReclaimNeeds(cacheID, nBytes, nCount)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif needs == nil {\n\t\t\t// Reservation succeeded immediately\n\t\t\treturn nil\n\t\t}\n\n\t\tif reclaim != nil && (needs.needBytes > 0 || needs.needCount > 0) {\n\t\t\t// The manager uses only atomic operations (no locks), so the cache\n\t\t\t// can safely take its own locks (e.g., for protecting internal data structures\n\t\t\t// during eviction) without risk of deadlock. The cache must call ReleaseForCache\n\t\t\t// separately after evicting items.\n\t\t\treclaim(needs.needBytes, needs.needCount)\n\n\t\t\t// Fast path: try to consume immediately after reclaim before yielding.\n\t\t\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err == nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t}\n\n\t\tm.yield() // tiny backoff to avoid busy spin\n\t}\n}\n\nfunc (m *manager) ReserveOrReclaimManagerRelease(\n\tctx context.Context,\n\tcacheID string,\n\tnBytes uint64,\n\tnCount int64,\n\tretriable bool,\n\treclaim ReclaimManagerRelease,\n) error {\n\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err == nil {\n\t\treturn nil\n\t} else if !retriable {\n\t\treturn err\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn ctx.Err() // propagate context cancellation/deadline\n\t\tdefault:\n\t\t}\n\n\t\tneeds, err := m.tryReserveOrCalculateReclaimNeeds(cacheID, nBytes, nCount)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif needs == nil {\n\t\t\t// Reservation succeeded immediately\n\t\t\treturn nil\n\t\t}\n\n\t\tif reclaim != nil && (needs.needBytes > 0 || needs.needCount > 0) {\n\t\t\tfb, fc := reclaim(needs.needBytes, needs.needCount)\n\t\t\tif fb > 0 || fc > 0 {\n\t\t\t\tm.ReleaseForCache(cacheID, fb, fc)\n\n\t\t\t\t// Fast-path: try to admit immediately after releasing\n\t\t\t\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err == nil {\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tm.yield() // small backoff to avoid busy spin\n\t}\n}\n\nfunc (m *manager) yield() {\n\tif m.reclaimBackoff > 0 {\n\t\ttime.Sleep(m.reclaimBackoff)\n\t\treturn\n\t}\n\ttime.Sleep(100 * time.Microsecond)\n}\n\n// Cache-aware Release methods\n\nfunc (m *manager) ReleaseForCache(cacheID string, nBytes uint64, nCount int64) {\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\n\tif nBytes > 0 {\n\t\tm.releaseBytes(nBytes)\n\t\t// Update cache-specific usage tracking with over-release protection\n\t\tif nBytes > cacheUsage.usedBytes {\n\t\t\tif m.logger != nil {\n\t\t\t\tm.logger.Debug(\"Cache bytes over-release detected\",\n\t\t\t\t\ttag.CacheID(cacheID),\n\t\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nBytes),\n\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(cacheUsage.usedBytes),\n\t\t\t\t)\n\t\t\t}\n\t\t\tcacheUsage.usedBytes = 0\n\t\t\tcacheUsage.fairShareCapacityBytes = 0\n\t\t\tcacheUsage.freeCapacityBytes = 0\n\t\t} else {\n\t\t\tcacheUsage.usedBytes -= nBytes\n\t\t\t// Update capacity type breakdown - subtract from fairShare first, then free\n\t\t\tremainingBytes := nBytes\n\t\t\tfairShareToSubtract := min(remainingBytes, cacheUsage.fairShareCapacityBytes)\n\t\t\tif fairShareToSubtract > 0 {\n\t\t\t\tcacheUsage.fairShareCapacityBytes -= fairShareToSubtract\n\t\t\t\tremainingBytes -= fairShareToSubtract\n\t\t\t}\n\t\t\tif remainingBytes > 0 {\n\t\t\t\tcacheUsage.freeCapacityBytes -= remainingBytes\n\t\t\t}\n\t\t}\n\t}\n\tif nCount > 0 {\n\t\tm.releaseCount(nCount)\n\t\t// Update cache-specific usage tracking with over-release protection\n\t\tif nCount > cacheUsage.usedCount {\n\t\t\tif m.logger != nil {\n\t\t\t\tm.logger.Debug(\"Cache count over-release detected\",\n\t\t\t\t\ttag.CacheID(cacheID),\n\t\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nCount),\n\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(cacheUsage.usedCount),\n\t\t\t\t)\n\t\t\t}\n\t\t\tcacheUsage.usedCount = 0\n\t\t\tcacheUsage.fairShareCapacityCount = 0\n\t\t\tcacheUsage.freeCapacityCount = 0\n\t\t} else {\n\t\t\tcacheUsage.usedCount -= nCount\n\t\t\t// Update capacity type breakdown - subtract from fairShare first, then free\n\t\t\tremainingCount := nCount\n\t\t\tfairShareCountToSubtract := min(remainingCount, cacheUsage.fairShareCapacityCount)\n\t\t\tif fairShareCountToSubtract > 0 {\n\t\t\t\tcacheUsage.fairShareCapacityCount -= fairShareCountToSubtract\n\t\t\t\tremainingCount -= fairShareCountToSubtract\n\t\t\t}\n\t\t\tif remainingCount > 0 {\n\t\t\t\tcacheUsage.freeCapacityCount -= remainingCount\n\t\t\t}\n\t\t}\n\t}\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif wasActive && !isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, -1)\n\t}\n}\n\nfunc (m *manager) ReleaseBytesForCache(cacheID string, nBytes uint64) {\n\tm.releaseBytes(nBytes)\n\t// Update cache-specific usage tracking\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\t// Update cache-specific usage tracking with over-release protection\n\tif nBytes > cacheUsage.usedBytes {\n\t\tif m.logger != nil {\n\t\t\tm.logger.Debug(\"Cache bytes over-release detected\",\n\t\t\t\ttag.CacheID(cacheID),\n\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nBytes),\n\t\t\t\ttag.Key(\"current-used\"), tag.Value(cacheUsage.usedBytes),\n\t\t\t)\n\t\t}\n\t\tcacheUsage.usedBytes = 0\n\t\tcacheUsage.fairShareCapacityBytes = 0\n\t\tcacheUsage.freeCapacityBytes = 0\n\t} else {\n\t\tcacheUsage.usedBytes -= nBytes\n\t\t// Update capacity type breakdown - subtract from fairShare first, then free\n\t\tremainingBytes := nBytes\n\t\tfairShareToSubtract := min(remainingBytes, cacheUsage.fairShareCapacityBytes)\n\t\tif fairShareToSubtract > 0 {\n\t\t\tcacheUsage.fairShareCapacityBytes -= fairShareToSubtract\n\t\t\tremainingBytes -= fairShareToSubtract\n\t\t}\n\t\tif remainingBytes > 0 {\n\t\t\tcacheUsage.freeCapacityBytes -= remainingBytes\n\t\t}\n\t}\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif wasActive && !isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, -1)\n\t}\n}\n\nfunc (m *manager) ReleaseCountForCache(cacheID string, nCount int64) {\n\tm.releaseCount(nCount)\n\t// Update cache-specific usage tracking\n\tcacheUsage := m.getCacheUsage(cacheID)\n\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\twasActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\t// Update cache-specific usage tracking with over-release protection\n\tif nCount > cacheUsage.usedCount {\n\t\tif m.logger != nil {\n\t\t\tm.logger.Debug(\"Cache count over-release detected\",\n\t\t\t\ttag.CacheID(cacheID),\n\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nCount),\n\t\t\t\ttag.Key(\"current-used\"), tag.Value(cacheUsage.usedCount),\n\t\t\t)\n\t\t}\n\t\tcacheUsage.usedCount = 0\n\t\tcacheUsage.fairShareCapacityCount = 0\n\t\tcacheUsage.freeCapacityCount = 0\n\t} else {\n\t\tcacheUsage.usedCount -= nCount\n\t\t// Update capacity type breakdown - subtract from fairShare first, then free\n\t\tremainingCount := nCount\n\t\tfairShareCountToSubtract := min(remainingCount, cacheUsage.fairShareCapacityCount)\n\t\tif fairShareCountToSubtract > 0 {\n\t\t\tcacheUsage.fairShareCapacityCount -= fairShareCountToSubtract\n\t\t\tremainingCount -= fairShareCountToSubtract\n\t\t}\n\t\tif remainingCount > 0 {\n\t\t\tcacheUsage.freeCapacityCount -= remainingCount\n\t\t}\n\t}\n\n\t// Update active cache count while still holding the lock to prevent races\n\tisActive := cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tif wasActive && !isActive {\n\t\tatomic.AddInt64(&m.activeCacheCount, -1)\n\t}\n\n}\n\nfunc (m *manager) releaseBytes(nBytes uint64) {\n\tfor {\n\t\told := atomic.LoadUint64(&m.usedBytes)\n\t\tvar newVal uint64\n\t\tif nBytes > old {\n\t\t\t// Over-release detected - clamp to 0 to prevent uint64 wraparound\n\t\t\tnewVal = 0\n\t\t\tif atomic.CompareAndSwapUint64(&m.usedBytes, old, newVal) {\n\t\t\t\tif m.logger != nil {\n\t\t\t\t\tm.logger.Debug(\"Bytes over-release detected\",\n\t\t\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nBytes),\n\t\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(old),\n\t\t\t\t\t)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\t\t} else {\n\t\t\tnewVal = old - nBytes\n\t\t\tif atomic.CompareAndSwapUint64(&m.usedBytes, old, newVal) {\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\t// retry on contention\n\t}\n}\n\nfunc (m *manager) releaseCount(nCount int64) {\n\tif nCount <= 0 {\n\t\treturn\n\t}\n\tfor {\n\t\told := atomic.LoadInt64(&m.usedCount)\n\t\tvar newVal int64\n\t\tif nCount > old {\n\t\t\t// Over-release detected - clamp to 0\n\t\t\tnewVal = 0\n\t\t\tif atomic.CompareAndSwapInt64(&m.usedCount, old, newVal) {\n\t\t\t\tif m.logger != nil {\n\t\t\t\t\tm.logger.Debug(\"Count over-release detected\",\n\t\t\t\t\t\ttag.Key(\"requested-release\"), tag.Value(nCount),\n\t\t\t\t\t\ttag.Key(\"current-used\"), tag.Value(old),\n\t\t\t\t\t)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\t\t} else {\n\t\t\tnewVal = old - nCount\n\t\t\tif atomic.CompareAndSwapInt64(&m.usedCount, old, newVal) {\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\t// retry on contention\n\t}\n}\n\nfunc (m *manager) ReserveWithCallback(cacheID string, nBytes uint64, nCount int64, callback func() error) error {\n\tif err := m.ReserveForCache(cacheID, nBytes, nCount); err != nil {\n\t\treturn err\n\t}\n\tif err := callback(); err != nil {\n\t\tm.ReleaseForCache(cacheID, nBytes, nCount)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (m *manager) ReserveBytesWithCallback(cacheID string, nBytes uint64, callback func() error) error {\n\tif err := m.ReserveBytesForCache(cacheID, nBytes); err != nil {\n\t\treturn err\n\t}\n\tif err := callback(); err != nil {\n\t\tm.ReleaseBytesForCache(cacheID, nBytes)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (m *manager) ReserveCountWithCallback(cacheID string, nCount int64, callback func() error) error {\n\tif err := m.ReserveCountForCache(cacheID, nCount); err != nil {\n\t\treturn err\n\t}\n\tif err := callback(); err != nil {\n\t\tm.ReleaseCountForCache(cacheID, nCount)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (m *manager) ReleaseWithCallback(cacheID string, callback func() (freedBytes uint64, freedCount int64, err error)) error {\n\tfreedBytes, freedCount, err := callback()\n\tif err != nil {\n\t\treturn err\n\t}\n\tm.ReleaseForCache(cacheID, freedBytes, freedCount)\n\treturn nil\n}\n\nfunc (m *manager) ReleaseBytesWithCallback(cacheID string, callback func() (freedBytes uint64, err error)) error {\n\tfreedBytes, err := callback()\n\tif err != nil {\n\t\treturn err\n\t}\n\tm.ReleaseBytesForCache(cacheID, freedBytes)\n\treturn nil\n}\n\nfunc (m *manager) ReleaseCountWithCallback(cacheID string, callback func() (freedCount int64, err error)) error {\n\tfreedCount, err := callback()\n\tif err != nil {\n\t\treturn err\n\t}\n\tm.ReleaseCountForCache(cacheID, freedCount)\n\treturn nil\n}\n\nfunc (m *manager) ReserveOrReclaimSelfReleaseWithCallback(ctx context.Context, cacheID string, nBytes uint64, nCount int64, retriable bool, reclaim ReclaimSelfRelease, callback func() error) error {\n\tif err := m.ReserveOrReclaimSelfRelease(ctx, cacheID, nBytes, nCount, retriable, reclaim); err != nil {\n\t\treturn err\n\t}\n\tif err := callback(); err != nil {\n\t\tm.ReleaseForCache(cacheID, nBytes, nCount)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (m *manager) ReserveOrReclaimManagerReleaseWithCallback(ctx context.Context, cacheID string, nBytes uint64, nCount int64, retriable bool, reclaim ReclaimManagerRelease, callback func() error) error {\n\tif err := m.ReserveOrReclaimManagerRelease(ctx, cacheID, nBytes, nCount, retriable, reclaim); err != nil {\n\t\treturn err\n\t}\n\tif err := callback(); err != nil {\n\t\tm.ReleaseForCache(cacheID, nBytes, nCount)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (m *manager) UsedBytes() uint64 { return atomic.LoadUint64(&m.usedBytes) }\nfunc (m *manager) UsedCount() int64  { return atomic.LoadInt64(&m.usedCount) }\n\nfunc (m *manager) CapacityBytes() uint64 {\n\tv := m.maxBytes()\n\tif v < 0 {\n\t\treturn math.MaxUint64\n\t}\n\treturn uint64(v) // includes 0 as a valid, enforced cap\n}\n\nfunc (m *manager) CapacityCount() int64 {\n\tv := m.maxCount()\n\tif v < 0 {\n\t\treturn math.MaxInt64\n\t}\n\t// clamp just in case\n\tif v > math.MaxInt64 {\n\t\treturn math.MaxInt64\n\t}\n\treturn int64(v) // includes 0 as a valid, enforced cap\n}\n\nfunc (m *manager) AvailableBytes() uint64 {\n\tc := m.CapacityBytes()\n\tif c == 0 {\n\t\treturn 0\n\t}\n\tu := m.UsedBytes()\n\tif u >= c {\n\t\treturn 0\n\t}\n\treturn c - u\n}\nfunc (m *manager) AvailableCount() int64 {\n\tc := m.CapacityCount()\n\tif c == 0 {\n\t\treturn 0\n\t}\n\tu := m.UsedCount()\n\tif u >= c {\n\t\treturn 0\n\t}\n\treturn c - u\n}\n\n// getCacheUsage returns the CacheUsage for a given cacheID, creating one if it doesn't exist\nfunc (m *manager) getCacheUsage(cacheID string) *Usage {\n\tif value, ok := m.cacheUsage.Load(cacheID); ok {\n\t\treturn value.(*Usage)\n\t}\n\n\t// Create new cache usage entry\n\tnewUsage := &Usage{}\n\tactual, _ := m.cacheUsage.LoadOrStore(cacheID, newUsage)\n\treturn actual.(*Usage)\n}\n\n// getActiveCacheCount returns the current count of active caches\nfunc (m *manager) getActiveCacheCount() int64 {\n\treturn atomic.LoadInt64(&m.activeCacheCount)\n}\n\n// getCacheState safely reads the current state of a cache with defer unlock\nfunc (m *manager) getCacheState(cacheID string) (isActive bool, fairShareBytes uint64, fairShareCount int64) {\n\tcacheUsage := m.getCacheUsage(cacheID)\n\tcacheUsage.mu.Lock()\n\tdefer cacheUsage.mu.Unlock()\n\n\tisActive = cacheUsage.usedBytes > 0 || cacheUsage.usedCount > 0\n\tfairShareBytes = cacheUsage.fairShareCapacityBytes\n\tfairShareCount = cacheUsage.fairShareCapacityCount\n\treturn\n}\n\n// enforceCapForCache implements both hard and soft cap logic\nfunc (m *manager) enforceCapForCache(cacheID string, additionalBytes uint64, additionalCount int64) (CapEnforcementResult, error) {\n\t// Check if caching is completely disabled (zero capacity)\n\tif m.CapacityBytes() == 0 || m.CapacityCount() == 0 {\n\t\tvar err error\n\t\tif m.CapacityBytes() == 0 {\n\t\t\terr = ErrBytesBudgetExceeded\n\t\t} else {\n\t\t\terr = ErrCountBudgetExceeded\n\t\t}\n\t\treturn CapEnforcementResult{\n\t\t\tFreeBytes:      0,\n\t\t\tFairShareBytes: 0,\n\t\t\tFreeCount:      0,\n\t\t\tFairShareCount: 0,\n\t\t\tAvailableBytes: 0,\n\t\t\tAvailableCount: 0,\n\t\t}, err\n\t}\n\n\t// Reject requests with zero bytes AND zero count\n\tif additionalBytes == 0 && additionalCount == 0 {\n\t\treturn CapEnforcementResult{\n\t\t\tFreeBytes:      0,\n\t\t\tFairShareBytes: 0,\n\t\t\tFreeCount:      0,\n\t\t\tFairShareCount: 0,\n\t\t\tAvailableBytes: 0,\n\t\t\tAvailableCount: 0,\n\t\t}, ErrInvalidRequest\n\t}\n\n\t// Get hard cap constraints\n\thardCapAvailableBytes := m.AvailableBytes()\n\thardCapAvailableCount := m.AvailableCount()\n\n\t// Check soft cap constraints\n\tmanagerThreshold := m.enforcementSoftCapThreshold()\n\tif managerThreshold >= 0 && managerThreshold < 1 {\n\t\t// Soft caps enabled - calculate soft cap constraints\n\t\tthresholdBytes := uint64(float64(m.CapacityBytes()) * managerThreshold)\n\t\tthresholdCount := int64(float64(m.CapacityCount()) * managerThreshold)\n\n\t\t// Calculate available free space\n\t\tfreeSpaceBytes := uint64(0)\n\t\tif thresholdBytes > m.UsedBytes() {\n\t\t\tfreeSpaceBytes = thresholdBytes - m.UsedBytes()\n\t\t}\n\t\tfreeSpaceCount := int64(0)\n\t\tif thresholdCount > m.UsedCount() {\n\t\t\tfreeSpaceCount = thresholdCount - m.UsedCount()\n\t\t}\n\n\t\t// Calculate per-cache fair share limits\n\t\tactiveCaches := m.getActiveCacheCount()\n\n\t\t// Check if this cache is currently inactive but will become active\n\t\tcacheIsActive, currentFairShareBytes, currentFairShareCount := m.getCacheState(cacheID)\n\n\t\tif !cacheIsActive {\n\t\t\t// Cache will become active, include it in the count\n\t\t\tactiveCaches++\n\t\t} else if activeCaches == 0 {\n\t\t\t// Cache is already active, but global counter hasn't been updated yet.\n\t\t\t// This happens because we update activeCacheCount under per-cache lock,\n\t\t\t// but read it without any lock. Ensure we count at least this active cache.\n\t\t\tactiveCaches = 1\n\t\t}\n\n\t\tfairShareCapacityBytes := uint64(float64(m.CapacityBytes()) * (1.0 - managerThreshold))\n\t\tfairShareCapacityCount := int64(float64(m.CapacityCount()) * (1.0 - managerThreshold))\n\t\tfairSharePerCacheBytes := fairShareCapacityBytes / uint64(activeCaches)\n\t\tfairSharePerCacheCount := fairShareCapacityCount / activeCaches\n\n\t\t// Calculate available fair share capacity for this cache\n\t\tfairShareAvailableBytes := uint64(0)\n\t\tif fairSharePerCacheBytes > currentFairShareBytes {\n\t\t\tfairShareAvailableBytes = fairSharePerCacheBytes - currentFairShareBytes\n\t\t}\n\t\tfairShareAvailableCount := int64(0)\n\t\tif fairSharePerCacheCount > currentFairShareCount {\n\t\t\tfairShareAvailableCount = fairSharePerCacheCount - currentFairShareCount\n\t\t}\n\n\t\t// Total soft cap available is free space + fair share for this cache\n\t\tsoftCapAvailableBytes := freeSpaceBytes + fairShareAvailableBytes\n\t\tsoftCapAvailableCount := freeSpaceCount + fairShareAvailableCount\n\n\t\t// Available capacity is minimum of soft cap and hard cap constraints\n\t\tavailableBytes := min(softCapAvailableBytes, hardCapAvailableBytes)\n\t\tavailableCount := min(softCapAvailableCount, hardCapAvailableCount)\n\n\t\t// Check if the request can be satisfied\n\t\tif additionalBytes <= availableBytes && additionalCount <= availableCount {\n\t\t\t// Success - calculate allocation breakdown\n\t\t\tfreeBytesUsage := min(additionalBytes, freeSpaceBytes)\n\t\t\tfairShareBytesUsage := additionalBytes - freeBytesUsage\n\t\t\tfreeCountUsage := min(additionalCount, freeSpaceCount)\n\t\t\tfairShareCountUsage := additionalCount - freeCountUsage\n\n\t\t\treturn CapEnforcementResult{\n\t\t\t\tFreeBytes:      freeBytesUsage,\n\t\t\t\tFairShareBytes: fairShareBytesUsage,\n\t\t\t\tFreeCount:      freeCountUsage,\n\t\t\t\tFairShareCount: fairShareCountUsage,\n\t\t\t\tAvailableBytes: availableBytes,\n\t\t\t\tAvailableCount: availableCount,\n\t\t\t}, nil\n\t\t}\n\n\t\t// Determine which constraint failed\n\t\tvar err error\n\t\tif additionalBytes > hardCapAvailableBytes || additionalCount > hardCapAvailableCount {\n\t\t\t// Hard cap constraint failed\n\t\t\tif additionalBytes > hardCapAvailableBytes {\n\t\t\t\terr = ErrBytesBudgetExceeded\n\t\t\t} else {\n\t\t\t\terr = ErrCountBudgetExceeded\n\t\t\t}\n\t\t} else {\n\t\t\t// Soft cap constraint failed\n\t\t\tif additionalBytes > softCapAvailableBytes {\n\t\t\t\terr = ErrBytesSoftCapExceeded\n\t\t\t} else {\n\t\t\t\terr = ErrCountSoftCapExceeded\n\t\t\t}\n\t\t}\n\n\t\treturn CapEnforcementResult{\n\t\t\tFreeBytes:      0, // No allocation on failure\n\t\t\tFairShareBytes: 0, // No allocation on failure\n\t\t\tFreeCount:      0, // No allocation on failure\n\t\t\tFairShareCount: 0, // No allocation on failure\n\t\t\tAvailableBytes: availableBytes,\n\t\t\tAvailableCount: availableCount,\n\t\t}, err\n\t}\n\n\t// Soft caps disabled - only check hard cap\n\tavailableBytes := hardCapAvailableBytes\n\tavailableCount := hardCapAvailableCount\n\n\tif additionalBytes <= hardCapAvailableBytes && additionalCount <= hardCapAvailableCount {\n\t\t// Success - all allocation goes to fair share when soft caps disabled\n\t\treturn CapEnforcementResult{\n\t\t\tFreeBytes:      0, // No free space allocation when soft caps disabled\n\t\t\tFairShareBytes: additionalBytes,\n\t\t\tFreeCount:      0, // No free space allocation when soft caps disabled\n\t\t\tFairShareCount: additionalCount,\n\t\t\tAvailableBytes: availableBytes,\n\t\t\tAvailableCount: availableCount,\n\t\t}, nil\n\t}\n\n\t// Hard cap exceeded\n\tvar err error\n\tif additionalBytes > hardCapAvailableBytes {\n\t\terr = ErrBytesBudgetExceeded\n\t} else {\n\t\terr = ErrCountBudgetExceeded\n\t}\n\n\treturn CapEnforcementResult{\n\t\tFreeBytes:      0, // No allocation on failure\n\t\tFairShareBytes: 0, // No allocation on failure\n\t\tFreeCount:      0, // No allocation on failure\n\t\tFairShareCount: 0, // No allocation on failure\n\t\tAvailableBytes: availableBytes,\n\t\tAvailableCount: availableCount,\n\t}, err\n}\n"
  },
  {
    "path": "common/cache/budget_test.go",
    "content": "package cache\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\tmetricsmocks \"github.com/uber/cadence/common/metrics/mocks\"\n)\n\nfunc TestBudgetManager_ReserveForCache(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tcapacityBytes     int\n\t\tcapacityCount     int\n\t\tsoftCapThreshold  float64\n\t\trequestBytes      uint64\n\t\trequestCount      int64\n\t\tcacheID           string\n\t\texpectedError     error\n\t\texpectedUsedBytes uint64\n\t\texpectedUsedCount int64\n\t\tdescription       string\n\t}{\n\t\t{\n\t\t\tname:              \"successful reserve with sufficient capacity\",\n\t\t\tcapacityBytes:     1000,\n\t\t\tcapacityCount:     100,\n\t\t\tsoftCapThreshold:  1.0, // disabled\n\t\t\trequestBytes:      100,\n\t\t\trequestCount:      10,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     nil,\n\t\t\texpectedUsedBytes: 100,\n\t\t\texpectedUsedCount: 10,\n\t\t\tdescription:       \"Should successfully reserve when capacity is available\",\n\t\t},\n\t\t{\n\t\t\tname:              \"hard cap exceeded - bytes\",\n\t\t\tcapacityBytes:     100,\n\t\t\tcapacityCount:     100,\n\t\t\tsoftCapThreshold:  1.0, // disabled\n\t\t\trequestBytes:      200,\n\t\t\trequestCount:      10,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     ErrBytesBudgetExceeded,\n\t\t\texpectedUsedBytes: 0,\n\t\t\texpectedUsedCount: 0,\n\t\t\tdescription:       \"Should fail when requesting more bytes than hard cap\",\n\t\t},\n\t\t{\n\t\t\tname:              \"hard cap exceeded - count\",\n\t\t\tcapacityBytes:     1000,\n\t\t\tcapacityCount:     10,\n\t\t\tsoftCapThreshold:  1.0, // disabled\n\t\t\trequestBytes:      100,\n\t\t\trequestCount:      20,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     ErrCountBudgetExceeded,\n\t\t\texpectedUsedBytes: 0,\n\t\t\texpectedUsedCount: 0,\n\t\t\tdescription:       \"Should fail when requesting more count than hard cap\",\n\t\t},\n\t\t{\n\t\t\tname:              \"zero capacity - bytes\",\n\t\t\tcapacityBytes:     0,\n\t\t\tcapacityCount:     100,\n\t\t\tsoftCapThreshold:  1.0,\n\t\t\trequestBytes:      10,\n\t\t\trequestCount:      1,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     ErrBytesBudgetExceeded,\n\t\t\texpectedUsedBytes: 0,\n\t\t\texpectedUsedCount: 0,\n\t\t\tdescription:       \"Should fail when bytes capacity is zero\",\n\t\t},\n\t\t{\n\t\t\tname:              \"zero capacity - count\",\n\t\t\tcapacityBytes:     1000,\n\t\t\tcapacityCount:     0,\n\t\t\tsoftCapThreshold:  1.0,\n\t\t\trequestBytes:      10,\n\t\t\trequestCount:      1,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     ErrCountBudgetExceeded,\n\t\t\texpectedUsedBytes: 0,\n\t\t\texpectedUsedCount: 0,\n\t\t\tdescription:       \"Should fail when count capacity is zero\",\n\t\t},\n\t\t{\n\t\t\tname:              \"soft cap - free space allocation\",\n\t\t\tcapacityBytes:     1000,\n\t\t\tcapacityCount:     100,\n\t\t\tsoftCapThreshold:  0.5, // 500 bytes free, 500 bytes fair share\n\t\t\trequestBytes:      100,\n\t\t\trequestCount:      10,\n\t\t\tcacheID:           \"cache1\",\n\t\t\texpectedError:     nil,\n\t\t\texpectedUsedBytes: 100,\n\t\t\texpectedUsedCount: 10,\n\t\t\tdescription:       \"Should successfully allocate from free space when under threshold\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tdynamicproperties.GetFloatPropertyFn(tt.softCapThreshold),\n\t\t\t)\n\n\t\t\terr := mgr.ReserveWithCallback(tt.cacheID, tt.requestBytes, tt.requestCount, func() error {\n\t\t\t\treturn nil\n\t\t\t})\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedUsedBytes, mgr.UsedBytes(), \"Used bytes mismatch\")\n\t\t\tassert.Equal(t, tt.expectedUsedCount, mgr.UsedCount(), \"Used count mismatch\")\n\t\t})\n\t}\n}\n\n// Test soft cap with multiple caches - fair share enforcement\nfunc TestBudgetManager_SoftCapFairShare(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\tcapacityBytes    int\n\t\tcapacityCount    int\n\t\tsoftCapThreshold float64\n\t\tsetup            func(mgr Manager)\n\t\trequestBytes     uint64\n\t\trequestCount     int64\n\t\tcacheID          string\n\t\texpectedError    error\n\t\tdescription      string\n\t}{\n\t\t{\n\t\t\tname:             \"request within fair share - single active cache\",\n\t\t\tcapacityBytes:    1000,\n\t\t\tcapacityCount:    100,\n\t\t\tsoftCapThreshold: 0.5, // 500 free, 500 fair share\n\t\t\tsetup: func(mgr Manager) {\n\t\t\t\t// Fill up the free space (threshold)\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\t\t\t\t// Now cache1 tries to allocate more, should use fair share\n\t\t\t},\n\t\t\trequestBytes:  300, // Within fair share for cache1 (500/1 = 500 available)\n\t\t\trequestCount:  30,\n\t\t\tcacheID:       \"cache1\",\n\t\t\texpectedError: nil, // Should succeed because fair share allows it (cache1 is only active cache)\n\t\t\tdescription:   \"Single active cache should get full fair share capacity\",\n\t\t},\n\t\t{\n\t\t\tname:             \"request within fair share - multiple caches fair share\",\n\t\t\tcapacityBytes:    1000,\n\t\t\tcapacityCount:    100,\n\t\t\tsoftCapThreshold: 0.5,\n\t\t\tsetup: func(mgr Manager) {\n\t\t\t\t// Fill up the free space\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\t\t\t\t// Activate cache2 with some fair share usage\n\t\t\t\tmgr.ReserveWithCallback(\"cache2\", 100, 10, func() error { return nil })\n\t\t\t\t// Now 2 active caches, fair share = 500/2 = 250 per cache\n\t\t\t},\n\t\t\trequestBytes:  120, // Should be within fair share for cache2 (250 - 100 existing = 150 available)\n\t\t\trequestCount:  10,  // Should be within fair share for cache2 (25 - 10 existing = 15 available)\n\t\t\tcacheID:       \"cache2\",\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should succeed because cache is within per-cache fair share with multiple active caches\",\n\t\t},\n\t\t{\n\t\t\tname:             \"soft cap exceeded - multiple caches fair share\",\n\t\t\tcapacityBytes:    1000,\n\t\t\tcapacityCount:    100,\n\t\t\tsoftCapThreshold: 0.5,\n\t\t\tsetup: func(mgr Manager) {\n\t\t\t\t// Fill up the free space\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\t\t\t\t// Activate cache2 with some fair share usage\n\t\t\t\tmgr.ReserveWithCallback(\"cache2\", 100, 10, func() error { return nil })\n\t\t\t\t// Now 2 active caches, fair share = 500/2 = 250 per cache\n\t\t\t},\n\t\t\trequestBytes:  200, // cache2 already has 100, requesting 200 more = 300 total > 250 fair share\n\t\t\trequestCount:  20,\n\t\t\tcacheID:       \"cache2\",\n\t\t\texpectedError: ErrBytesSoftCapExceeded,\n\t\t\tdescription:   \"Should fail when exceeding per-cache fair share with multiple active caches\",\n\t\t},\n\t\t{\n\t\t\tname:             \"soft cap exceeded - count fair share violation\",\n\t\t\tcapacityBytes:    1000,\n\t\t\tcapacityCount:    100,\n\t\t\tsoftCapThreshold: 0.5,\n\t\t\tsetup: func(mgr Manager) {\n\t\t\t\t// Fill up the free space\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\t\t\t\t// Activate cache2\n\t\t\t\tmgr.ReserveWithCallback(\"cache2\", 100, 10, func() error { return nil })\n\t\t\t\t// Now 2 active caches, fair share count = 50/2 = 25 per cache\n\t\t\t},\n\t\t\trequestBytes:  50, // bytes within limit\n\t\t\trequestCount:  20, // cache2 already has 10, requesting 20 more = 30 total > 25 fair share\n\t\t\tcacheID:       \"cache2\",\n\t\t\texpectedError: ErrCountSoftCapExceeded,\n\t\t\tdescription:   \"Should fail when exceeding per-cache count fair share\",\n\t\t},\n\t\t{\n\t\t\tname:             \"soft cap exceeded - bytes fair share violation\",\n\t\t\tcapacityBytes:    1000,\n\t\t\tcapacityCount:    100,\n\t\t\tsoftCapThreshold: 0.5,\n\t\t\tsetup: func(mgr Manager) {\n\t\t\t\t// Fill up the free space\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\t\t\t\t// Activate cache2\n\t\t\t\tmgr.ReserveWithCallback(\"cache2\", 100, 10, func() error { return nil })\n\t\t\t\t// Now 2 active caches, fair share count = 50/2 = 25 per cache\n\t\t\t},\n\t\t\trequestBytes:  200, // cache2 already has 100, requesting 200 more = 300 total > 250 fair share\n\t\t\trequestCount:  1,   // count within limit\n\t\t\tcacheID:       \"cache2\",\n\t\t\texpectedError: ErrBytesSoftCapExceeded,\n\t\t\tdescription:   \"Should fail when exceeding per-cache bytes fair share\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tdynamicproperties.GetFloatPropertyFn(tt.softCapThreshold),\n\t\t\t)\n\n\t\t\tif tt.setup != nil {\n\t\t\t\ttt.setup(mgr)\n\t\t\t}\n\n\t\t\terr := mgr.ReserveWithCallback(tt.cacheID, tt.requestBytes, tt.requestCount, func() error { return nil })\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Test releasing capacity from fair share tracking\nfunc TestBudgetManager_ReleaseFairShareTracking(t *testing.T) {\n\tmgr := NewBudgetManager(\n\t\t\"test\",\n\t\tdynamicproperties.GetIntPropertyFn(1000), // 1000 bytes capacity\n\t\tdynamicproperties.GetIntPropertyFn(100),  // 100 count capacity\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(t),\n\t\tdynamicproperties.GetFloatPropertyFn(0.5), // 500 free, 500 fair share\n\t).(*manager)\n\n\t// Fill up the free space (500 bytes)\n\terr := mgr.ReserveWithCallback(\"cache1\", 500, 50, func() error { return nil })\n\tassert.NoError(t, err, \"Should reserve free space successfully\")\n\n\t// Now allocate from fair share (cache1 uses fair share portion)\n\terr = mgr.ReserveWithCallback(\"cache1\", 200, 20, func() error { return nil })\n\tassert.NoError(t, err, \"Should reserve from fair share successfully\")\n\n\t// Total usage: 700 bytes (500 free + 200 fair share)\n\tassert.Equal(t, uint64(700), mgr.UsedBytes(), \"Total used should be 700\")\n\n\t// Check fair share tracking for cache1\n\tcacheUsage := mgr.getCacheUsage(\"cache1\")\n\tfairShareBytes := cacheUsage.fairShareCapacityBytes\n\tassert.Equal(t, uint64(200), fairShareBytes, \"Fair share bytes should be 200\")\n\n\t// Release some capacity (100 bytes) - should deduct from fair share first\n\tmgr.ReleaseForCache(\"cache1\", 100, 10)\n\n\t// Total usage should be 600 (not going to zero)\n\tassert.Equal(t, uint64(600), mgr.UsedBytes(), \"Total used should be 600 after release\")\n\tassert.Equal(t, int64(60), mgr.UsedCount(), \"Total count should be 60 after release\")\n\n\t// Fair share tracking should be reduced to 100\n\tfairShareBytes = cacheUsage.fairShareCapacityBytes\n\tassert.Equal(t, uint64(100), fairShareBytes, \"Fair share bytes should be reduced to 100\")\n\n\t// Release more (150 bytes) - should deduct the remaining 100 from fair share, then 50 from free\n\tmgr.ReleaseForCache(\"cache1\", 150, 15)\n\n\t// Total usage should be 450\n\tassert.Equal(t, uint64(450), mgr.UsedBytes(), \"Total used should be 450 after second release\")\n\tassert.Equal(t, int64(45), mgr.UsedCount(), \"Total count should be 45 after second release\")\n\n\t// Fair share tracking should be 0\n\tfairShareBytes = cacheUsage.fairShareCapacityBytes\n\tassert.Equal(t, uint64(0), fairShareBytes, \"Fair share bytes should be 0 after releasing all fair share\")\n}\n\n// Test individual Reserve/Release methods for bytes and count\nfunc TestBudgetManager_IndividualReserveRelease(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\toperations    func(mgr Manager) error\n\t\texpectedError error\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"ReserveBytesForCache success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveBytesWithCallback(\"cache1\", 100, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should successfully reserve bytes only\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveBytesForCache exceeds capacity\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveBytesWithCallback(\"cache1\", 200, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\tdescription:   \"Should fail when bytes exceed capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveCountForCache success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveCountWithCallback(\"cache1\", 10, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should successfully reserve count only\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveCountForCache exceeds capacity\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 10,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveCountWithCallback(\"cache1\", 20, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrCountBudgetExceeded,\n\t\t\tdescription:   \"Should fail when count exceeds capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseBytesForCache\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveBytesWithCallback(\"cache1\", 100, func() error { return nil })\n\t\t\t\tmgr.ReleaseBytesWithCallback(\"cache1\", func() (uint64, error) { return 50, nil })\n\t\t\t\tif mgr.UsedBytes() != 50 {\n\t\t\t\t\treturn assert.AnError\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should release bytes correctly\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseCountForCache\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveCountWithCallback(\"cache1\", 10, func() error { return nil })\n\t\t\t\tmgr.ReleaseCountWithCallback(\"cache1\", func() (int64, error) { return 5, nil })\n\t\t\t\tif mgr.UsedCount() != 5 {\n\t\t\t\t\treturn assert.AnError\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should release count correctly\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveCountForCache with negative value\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveCountWithCallback(\"cache1\", -10, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrInvalidValue,\n\t\t\tdescription:   \"Should reject negative count values\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\terr := tt.operations(mgr)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Test Strict admission mode\nfunc TestBudgetManager_StrictAdmission(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\toperations    func(mgr Manager) error\n\t\texpectedError error\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"strict mode - reserve within capacity\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache1\", 100, 10, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should successfully reserve in strict mode when capacity available\",\n\t\t},\n\t\t{\n\t\t\tname:          \"strict mode - bytes exceed capacity\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache1\", 200, 10, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\tdescription:   \"Should fail in strict mode when bytes exceed capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"strict mode - count exceed capacity\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 10,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache1\", 100, 20, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrCountBudgetExceeded,\n\t\t\tdescription:   \"Should fail in strict mode when count exceeds capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"strict mode - prevents temporary overshoot\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\t// First reserve should succeed\n\t\t\t\tif err := mgr.ReserveWithCallback(\"cache1\", 60, 60, func() error { return nil }); err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\t// Second reserve should fail (would exceed if allowed)\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache2\", 50, 50, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\tdescription:   \"Strict mode should prevent any overshoot attempts\",\n\t\t},\n\t\t{\n\t\t\tname:          \"strict mode - negative count value\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperations: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveCountWithCallback(\"cache1\", -10, func() error { return nil })\n\t\t\t},\n\t\t\texpectedError: ErrInvalidValue,\n\t\t\tdescription:   \"Strict mode should reject negative count values\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionStrict, // Use strict mode\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\terr := tt.operations(mgr)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Test rollback behavior when partial reservations fail\nfunc TestBudgetManager_Rollback(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\tadmissionMode AdmissionMode\n\t\toperations    func(mgr Manager) (error, uint64, int64)\n\t\texpectedError error\n\t\texpectedBytes uint64\n\t\texpectedCount int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"rollback bytes when count fails\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 10,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperations: func(mgr Manager) (error, uint64, int64) {\n\t\t\t\terr := mgr.ReserveWithCallback(\"cache1\", 100, 20, func() error { return nil })\n\t\t\t\treturn err, mgr.UsedBytes(), mgr.UsedCount()\n\t\t\t},\n\t\t\texpectedError: ErrCountBudgetExceeded,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should rollback bytes when count reservation fails\",\n\t\t},\n\t\t{\n\t\t\tname:          \"no rollback needed when bytes fail first\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 1000,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperations: func(mgr Manager) (error, uint64, int64) {\n\t\t\t\terr := mgr.ReserveWithCallback(\"cache1\", 200, 10, func() error { return nil })\n\t\t\t\treturn err, mgr.UsedBytes(), mgr.UsedCount()\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"No rollback needed when bytes fail before count is reserved\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\ttt.admissionMode,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\terr, usedBytes, usedCount := tt.operations(mgr)\n\n\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\tassert.Equal(t, tt.expectedBytes, usedBytes, \"Used bytes should match expected after rollback\")\n\t\t\tassert.Equal(t, tt.expectedCount, usedCount, \"Used count should match expected\")\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_BytesOverflow(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes uint64\n\t\tcapacityCount int64\n\t\tadmission     AdmissionMode\n\t\tsetupUsage    uint64\n\t\treserveAmount uint64\n\t\texpectedError error\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"strict mode - overflow detection\",\n\t\t\tcapacityBytes: math.MaxUint64,\n\t\t\tcapacityCount: 100,\n\t\t\tadmission:     AdmissionStrict,\n\t\t\tsetupUsage:    math.MaxUint64 - 100,\n\t\t\treserveAmount: 150,\n\t\t\texpectedError: ErrOverflow,\n\t\t\tdescription:   \"Strict mode should detect overflow when old + n > MaxUint64\",\n\t\t},\n\t\t{\n\t\t\tname:          \"optimistic mode - overflow detection via wraparound\",\n\t\t\tcapacityBytes: math.MaxUint64,\n\t\t\tcapacityCount: 100,\n\t\t\tadmission:     AdmissionOptimistic,\n\t\t\tsetupUsage:    math.MaxUint64 - 50,\n\t\t\treserveAmount: 75,\n\t\t\texpectedError: ErrOverflow,\n\t\t\tdescription:   \"Optimistic mode should detect wraparound when newVal < n\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(int(tt.capacityBytes)),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(int(tt.capacityCount)),\n\t\t\t\ttt.admission,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tm := mgr.(*manager)\n\t\t\tatomic.StoreUint64(&m.usedBytes, tt.setupUsage)\n\n\t\t\terr := m.reserveBytes(tt.reserveAmount)\n\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_ReserveOrReclaimSelfRelease(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\tretriable     bool\n\t\trequestBytes  uint64\n\t\trequestCount  int64\n\t\tsetupUsage    func(mgr Manager)\n\t\treclaimFunc   ReclaimSelfRelease\n\t\texpectedError error\n\t\texpectedBytes uint64\n\t\texpectedCount int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"immediate success - no reclaim needed\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  100,\n\t\t\trequestCount:  10,\n\t\t\tsetupUsage:    nil,\n\t\t\treclaimFunc:   nil,\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should succeed immediately when capacity is available\",\n\t\t},\n\t\t{\n\t\t\tname:          \"non-retriable failure\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     false,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"other_cache\", 600, 60, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc:   nil,\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\texpectedBytes: 600,\n\t\t\texpectedCount: 60,\n\t\t\tdescription:   \"Should fail immediately when retriable is false\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reclaim with self-release\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) {\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should succeed after reclaim callback (cache releases its own items)\",\n\t\t},\n\t\t{\n\t\t\tname:          \"context cancellation\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 1000, 100, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) {\n\t\t\t},\n\t\t\texpectedError: context.Canceled,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should return context error when context is cancelled\",\n\t\t},\n\t\t{\n\t\t\tname:          \"insufficient cache budget for reclaim\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  900,\n\t\t\trequestCount:  90,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 100, 10, func() error { return nil })\n\t\t\t\tmgr.ReserveWithCallback(\"other_cache\", 200, 20, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) {\n\t\t\t},\n\t\t\texpectedError: ErrInsufficientUsageToReclaim,\n\t\t\texpectedBytes: 300,\n\t\t\texpectedCount: 30,\n\t\t\tdescription:   \"Should fail when cache doesn't have enough to reclaim\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tif tt.setupUsage != nil {\n\t\t\t\ttt.setupUsage(mgr)\n\t\t\t}\n\n\t\t\tvar reclaimCalled bool\n\t\t\tvar reclaimFunc ReclaimSelfRelease\n\t\t\tif tt.reclaimFunc != nil {\n\t\t\t\treclaimFunc = func(needBytes uint64, needCount int64) {\n\t\t\t\t\treclaimCalled = true\n\t\t\t\t\tmgr.ReleaseWithCallback(\"cache1\", func() (uint64, int64, error) {\n\t\t\t\t\t\treturn needBytes, needCount, nil\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tctx := context.Background()\n\t\t\tif tt.expectedError == context.Canceled {\n\t\t\t\tvar cancel context.CancelFunc\n\t\t\t\tctx, cancel = context.WithCancel(ctx)\n\t\t\t\tcancel()\n\t\t\t}\n\n\t\t\terr := mgr.(*manager).ReserveOrReclaimSelfReleaseWithCallback(ctx, \"cache1\", tt.requestBytes, tt.requestCount, tt.retriable, reclaimFunc, func() error { return nil })\n\n\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\tif tt.reclaimFunc != nil && tt.expectedError == nil {\n\t\t\t\tassert.True(t, reclaimCalled, \"Reclaim function should be called\")\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedBytes, mgr.UsedBytes(), \"Used bytes mismatch\")\n\t\t\tassert.Equal(t, tt.expectedCount, mgr.UsedCount(), \"Used count mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_ReserveOrReclaimManagerRelease(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\tretriable     bool\n\t\trequestBytes  uint64\n\t\trequestCount  int64\n\t\tsetupUsage    func(mgr Manager)\n\t\treclaimFunc   ReclaimManagerRelease\n\t\texpectedError error\n\t\texpectedBytes uint64\n\t\texpectedCount int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"immediate success - no reclaim needed\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  100,\n\t\t\trequestCount:  10,\n\t\t\tsetupUsage:    nil,\n\t\t\treclaimFunc:   nil,\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should succeed immediately when capacity is available\",\n\t\t},\n\t\t{\n\t\t\tname:          \"non-retriable failure\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     false,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"other_cache\", 600, 60, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc:   nil,\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\texpectedBytes: 600,\n\t\t\texpectedCount: 60,\n\t\t\tdescription:   \"Should fail immediately when retriable is false\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reclaim with manager release\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\treturn needBytes, needCount\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should succeed after reclaim callback (manager releases items)\",\n\t\t},\n\t\t{\n\t\t\tname:          \"context cancellation\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 1000, 100, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\treturn needBytes, needCount\n\t\t\t},\n\t\t\texpectedError: context.Canceled,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should return context error when context is cancelled\",\n\t\t},\n\t\t{\n\t\t\tname:          \"insufficient cache budget for reclaim\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  900,\n\t\t\trequestCount:  90,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 100, 10, func() error { return nil })\n\t\t\t\tmgr.ReserveWithCallback(\"other_cache\", 200, 20, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\treturn needBytes, needCount\n\t\t\t},\n\t\t\texpectedError: ErrInsufficientUsageToReclaim,\n\t\t\texpectedBytes: 300,\n\t\t\texpectedCount: 30,\n\t\t\tdescription:   \"Should fail when cache doesn't have enough to reclaim\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reclaim returns zero\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tretriable:     true,\n\t\t\trequestBytes:  500,\n\t\t\trequestCount:  50,\n\t\t\tsetupUsage: func(mgr Manager) {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t},\n\t\t\treclaimFunc: func(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\treturn 0, 0\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\texpectedBytes: 600,\n\t\t\texpectedCount: 60,\n\t\t\tdescription:   \"Should keep retrying when reclaim returns zero\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tif tt.setupUsage != nil {\n\t\t\t\ttt.setupUsage(mgr)\n\t\t\t}\n\n\t\t\tvar reclaimCalled bool\n\t\t\tvar reclaimFunc ReclaimManagerRelease\n\t\t\tif tt.reclaimFunc != nil {\n\t\t\t\treclaimFunc = func(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\t\treclaimCalled = true\n\t\t\t\t\treturn tt.reclaimFunc(needBytes, needCount)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tctx := context.Background()\n\t\t\tif tt.expectedError == context.Canceled {\n\t\t\t\tvar cancel context.CancelFunc\n\t\t\t\tctx, cancel = context.WithCancel(ctx)\n\t\t\t\tcancel()\n\t\t\t} else if tt.name == \"reclaim returns zero\" {\n\t\t\t\tvar cancel context.CancelFunc\n\t\t\t\tctx, cancel = context.WithTimeout(ctx, 10*time.Millisecond)\n\t\t\t\tdefer cancel()\n\t\t\t}\n\n\t\t\terr := mgr.(*manager).ReserveOrReclaimManagerReleaseWithCallback(ctx, \"cache1\", tt.requestBytes, tt.requestCount, tt.retriable, reclaimFunc, func() error { return nil })\n\n\t\t\tif tt.name == \"reclaim returns zero\" {\n\t\t\t\tassert.Error(t, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t}\n\t\t\tif tt.reclaimFunc != nil && tt.expectedError == nil {\n\t\t\t\tassert.True(t, reclaimCalled, \"Reclaim function should be called\")\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedBytes, mgr.UsedBytes(), \"Used bytes mismatch\")\n\t\t\tassert.Equal(t, tt.expectedCount, mgr.UsedCount(), \"Used count mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_CapacityCount(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tmaxCount      int\n\t\texpectedCount int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"positive capacity\",\n\t\t\tmaxCount:      100,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should return the configured count capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"zero capacity\",\n\t\t\tmaxCount:      0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should return 0 when capacity is 0\",\n\t\t},\n\t\t{\n\t\t\tname:          \"negative capacity (unlimited)\",\n\t\t\tmaxCount:      -1,\n\t\t\texpectedCount: math.MaxInt64,\n\t\t\tdescription:   \"Should return MaxInt64 when capacity is negative (unlimited)\",\n\t\t},\n\t\t{\n\t\t\tname:          \"nil maxCount defaults to unlimited\",\n\t\t\tmaxCount:      -2,\n\t\t\texpectedCount: math.MaxInt64,\n\t\t\tdescription:   \"Should return MaxInt64 when maxCount is nil\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar maxCountFn dynamicproperties.IntPropertyFn\n\t\t\tif tt.maxCount == -2 {\n\t\t\t\tmaxCountFn = nil\n\t\t\t} else {\n\t\t\t\tmaxCountFn = dynamicproperties.GetIntPropertyFn(tt.maxCount)\n\t\t\t}\n\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\t\t\tmaxCountFn,\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tassert.Equal(t, tt.expectedCount, mgr.CapacityCount(), tt.description)\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_AvailableBytes(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tusedBytes     uint64\n\t\texpectedAvail uint64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"full capacity available\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tusedBytes:     0,\n\t\t\texpectedAvail: 1000,\n\t\t\tdescription:   \"Should return full capacity when nothing is used\",\n\t\t},\n\t\t{\n\t\t\tname:          \"partial capacity available\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tusedBytes:     300,\n\t\t\texpectedAvail: 700,\n\t\t\tdescription:   \"Should return remaining capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"no capacity available - fully used\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tusedBytes:     1000,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when fully used\",\n\t\t},\n\t\t{\n\t\t\tname:          \"no capacity available - over capacity\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tusedBytes:     1200,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when used exceeds capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"zero capacity\",\n\t\t\tcapacityBytes: 0,\n\t\t\tusedBytes:     0,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when capacity is 0\",\n\t\t},\n\t\t{\n\t\t\tname:          \"nil maxBytes defaults to unlimited\",\n\t\t\tcapacityBytes: -2,\n\t\t\tusedBytes:     1000,\n\t\t\texpectedAvail: math.MaxUint64 - 1000,\n\t\t\tdescription:   \"Should return MaxUint64 - used when maxBytes is nil\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar maxBytesFn dynamicproperties.IntPropertyFn\n\t\t\tif tt.capacityBytes == -2 {\n\t\t\t\tmaxBytesFn = nil\n\t\t\t} else {\n\t\t\t\tmaxBytesFn = dynamicproperties.GetIntPropertyFn(tt.capacityBytes)\n\t\t\t}\n\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tmaxBytesFn,\n\t\t\t\tdynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tif tt.usedBytes > 0 {\n\t\t\t\tatomic.StoreUint64(&mgr.(*manager).usedBytes, tt.usedBytes)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedAvail, mgr.(*manager).AvailableBytes(), tt.description)\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_AvailableCount(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityCount int\n\t\tusedCount     int64\n\t\texpectedAvail int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"full capacity available\",\n\t\t\tcapacityCount: 100,\n\t\t\tusedCount:     0,\n\t\t\texpectedAvail: 100,\n\t\t\tdescription:   \"Should return full capacity when nothing is used\",\n\t\t},\n\t\t{\n\t\t\tname:          \"partial capacity available\",\n\t\t\tcapacityCount: 100,\n\t\t\tusedCount:     30,\n\t\t\texpectedAvail: 70,\n\t\t\tdescription:   \"Should return remaining capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"no capacity available - fully used\",\n\t\t\tcapacityCount: 100,\n\t\t\tusedCount:     100,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when fully used\",\n\t\t},\n\t\t{\n\t\t\tname:          \"no capacity available - over capacity\",\n\t\t\tcapacityCount: 100,\n\t\t\tusedCount:     120,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when used exceeds capacity\",\n\t\t},\n\t\t{\n\t\t\tname:          \"zero capacity\",\n\t\t\tcapacityCount: 0,\n\t\t\tusedCount:     0,\n\t\t\texpectedAvail: 0,\n\t\t\tdescription:   \"Should return 0 when capacity is 0\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tif tt.usedCount > 0 {\n\t\t\t\tatomic.StoreInt64(&mgr.(*manager).usedCount, tt.usedCount)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedAvail, mgr.(*manager).AvailableCount(), tt.description)\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_InternalReserveMethods(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\tadmissionMode AdmissionMode\n\t\toperation     func(*manager) error\n\t\texpectedError error\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"reserveBytesStrict - capacity exceeded\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveBytesStrict(200)\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\tdescription:   \"Should fail when bytes exceed capacity in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveBytesStrict - success\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveBytesStrict(50)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should succeed when bytes are within capacity in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveBytesStrict - overflow detection\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\tm.usedBytes = math.MaxUint64 - 100\n\t\t\t\treturn m.reserveBytesStrict(150)\n\t\t\t},\n\t\t\texpectedError: ErrOverflow,\n\t\t\tdescription:   \"Should detect overflow in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountStrict - capacity exceeded\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 10,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveCountStrict(20)\n\t\t\t},\n\t\t\texpectedError: ErrCountBudgetExceeded,\n\t\t\tdescription:   \"Should fail when count exceeds capacity in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountStrict - success\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 10,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveCountStrict(5)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should succeed when count is within capacity in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountStrict - overflow detection\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 1000,\n\t\t\tadmissionMode: AdmissionStrict,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\tm.usedCount = math.MaxInt64 - 10\n\t\t\t\treturn m.reserveCountStrict(20)\n\t\t\t},\n\t\t\texpectedError: ErrOverflow,\n\t\t\tdescription:   \"Should detect overflow in strict mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveBytesOptimistic - capacity exceeded\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveBytesOptimistic(200)\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\tdescription:   \"Should fail when bytes exceed capacity in optimistic mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveBytesOptimistic - success\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveBytesOptimistic(50)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should succeed when bytes are within capacity in optimistic mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountOptimistic - capacity exceeded\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 10,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveCountOptimistic(20)\n\t\t\t},\n\t\t\texpectedError: ErrCountBudgetExceeded,\n\t\t\tdescription:   \"Should fail when count exceeds capacity in optimistic mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountOptimistic - success\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 10,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\treturn m.reserveCountOptimistic(5)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\tdescription:   \"Should succeed when count is within capacity in optimistic mode\",\n\t\t},\n\t\t{\n\t\t\tname:          \"reserveCountOptimistic - overflow wraparound\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 1000,\n\t\t\tadmissionMode: AdmissionOptimistic,\n\t\t\toperation: func(m *manager) error {\n\t\t\t\tm.usedCount = math.MaxInt64 - 10\n\t\t\t\treturn m.reserveCountOptimistic(20)\n\t\t\t},\n\t\t\texpectedError: ErrOverflow,\n\t\t\tdescription:   \"Should detect wraparound overflow in optimistic mode\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\ttt.admissionMode,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\tm := mgr.(*manager)\n\t\t\terr := tt.operation(m)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err, tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_CallbackMethods(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcapacityBytes int\n\t\tcapacityCount int\n\t\toperation     func(mgr Manager) error\n\t\texpectedError error\n\t\texpectedBytes uint64\n\t\texpectedCount int64\n\t\tdescription   string\n\t}{\n\t\t{\n\t\t\tname:          \"ReserveWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReserveWithCallback(\"cache1\", 100, 10, func() error {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should reserve and call callback on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveWithCallback - callback error releases\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache1\", 100, 10, func() error {\n\t\t\t\t\treturn errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should release on callback error\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveWithCallback - reserve failure\",\n\t\t\tcapacityBytes: 100,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveWithCallback(\"cache1\", 200, 10, func() error {\n\t\t\t\t\treturn errors.New(\"should not be called\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: ErrBytesBudgetExceeded,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should not call callback when reserve fails\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveBytesWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReserveBytesWithCallback(\"cache1\", 100, func() error {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should reserve bytes and call callback on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveBytesWithCallback - callback error releases\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveBytesWithCallback(\"cache1\", 100, func() error {\n\t\t\t\t\treturn errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should release bytes on callback error\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveCountWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReserveCountWithCallback(\"cache1\", 10, func() error {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should reserve count and call callback on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveCountWithCallback - callback error releases\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\treturn mgr.ReserveCountWithCallback(\"cache1\", 10, func() error {\n\t\t\t\t\treturn errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should release count on callback error\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 100, 10, func() error { return nil })\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReleaseWithCallback(\"cache1\", func() (uint64, int64, error) {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn 100, 10, nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should call callback and release on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseWithCallback - callback error does not release\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 100, 10, func() error { return nil })\n\t\t\t\treturn mgr.ReleaseWithCallback(\"cache1\", func() (uint64, int64, error) {\n\t\t\t\t\treturn 0, 0, errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should not release when callback fails\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseBytesWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveBytesWithCallback(\"cache1\", 100, func() error { return nil })\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReleaseBytesWithCallback(\"cache1\", func() (uint64, error) {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn 100, nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should call callback and release bytes on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseBytesWithCallback - callback error does not release\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveBytesWithCallback(\"cache1\", 100, func() error { return nil })\n\t\t\t\treturn mgr.ReleaseBytesWithCallback(\"cache1\", func() (uint64, error) {\n\t\t\t\t\treturn 0, errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 100,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should not release bytes when callback fails\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseCountWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveCountWithCallback(\"cache1\", 10, func() error { return nil })\n\t\t\t\tcalled := false\n\t\t\t\terr := mgr.ReleaseCountWithCallback(\"cache1\", func() (int64, error) {\n\t\t\t\t\tcalled = true\n\t\t\t\t\treturn 10, nil\n\t\t\t\t})\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 0,\n\t\t\tdescription:   \"Should call callback and release count on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReleaseCountWithCallback - callback error does not release\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveCountWithCallback(\"cache1\", 10, func() error { return nil })\n\t\t\t\treturn mgr.ReleaseCountWithCallback(\"cache1\", func() (int64, error) {\n\t\t\t\t\treturn 0, errors.New(\"callback failed\")\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 0,\n\t\t\texpectedCount: 10,\n\t\t\tdescription:   \"Should not release count when callback fails\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveOrReclaimSelfReleaseWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t\tcalled := false\n\t\t\t\tcallbackCalled := false\n\t\t\t\terr := mgr.(*manager).ReserveOrReclaimSelfReleaseWithCallback(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\t\"cache1\",\n\t\t\t\t\t500,\n\t\t\t\t\t50,\n\t\t\t\t\ttrue,\n\t\t\t\t\tfunc(needBytes uint64, needCount int64) {\n\t\t\t\t\t\tcalled = true\n\t\t\t\t\t\tmgr.ReleaseWithCallback(\"cache1\", func() (uint64, int64, error) { return needBytes, needCount, nil })\n\t\t\t\t\t},\n\t\t\t\t\tfunc() error {\n\t\t\t\t\t\tcallbackCalled = true\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"reclaim not called\")\n\t\t\t\t}\n\t\t\t\tif !callbackCalled {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should reclaim, reserve and call callback on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveOrReclaimSelfReleaseWithCallback - callback error releases\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t\treturn mgr.(*manager).ReserveOrReclaimSelfReleaseWithCallback(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\t\"cache1\",\n\t\t\t\t\t500,\n\t\t\t\t\t50,\n\t\t\t\t\ttrue,\n\t\t\t\t\tfunc(needBytes uint64, needCount int64) {\n\t\t\t\t\t\tmgr.ReleaseWithCallback(\"cache1\", func() (uint64, int64, error) { return needBytes, needCount, nil })\n\t\t\t\t\t},\n\t\t\t\t\tfunc() error {\n\t\t\t\t\t\treturn errors.New(\"callback failed\")\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 500,\n\t\t\texpectedCount: 50,\n\t\t\tdescription:   \"Should release on callback error\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveOrReclaimManagerReleaseWithCallback - success\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t\tcalled := false\n\t\t\t\tcallbackCalled := false\n\t\t\t\terr := mgr.(*manager).ReserveOrReclaimManagerReleaseWithCallback(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\t\"cache1\",\n\t\t\t\t\t500,\n\t\t\t\t\t50,\n\t\t\t\t\ttrue,\n\t\t\t\t\tfunc(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\t\t\tcalled = true\n\t\t\t\t\t\treturn needBytes, needCount\n\t\t\t\t\t},\n\t\t\t\t\tfunc() error {\n\t\t\t\t\t\tcallbackCalled = true\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t\tif !called {\n\t\t\t\t\treturn errors.New(\"reclaim not called\")\n\t\t\t\t}\n\t\t\t\tif !callbackCalled {\n\t\t\t\t\treturn errors.New(\"callback not called\")\n\t\t\t\t}\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedBytes: 1000,\n\t\t\texpectedCount: 100,\n\t\t\tdescription:   \"Should reclaim, reserve and call callback on success\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ReserveOrReclaimManagerReleaseWithCallback - callback error releases\",\n\t\t\tcapacityBytes: 1000,\n\t\t\tcapacityCount: 100,\n\t\t\toperation: func(mgr Manager) error {\n\t\t\t\tmgr.ReserveWithCallback(\"cache1\", 600, 60, func() error { return nil })\n\t\t\t\treturn mgr.(*manager).ReserveOrReclaimManagerReleaseWithCallback(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\t\"cache1\",\n\t\t\t\t\t500,\n\t\t\t\t\t50,\n\t\t\t\t\ttrue,\n\t\t\t\t\tfunc(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\t\t\treturn needBytes, needCount\n\t\t\t\t\t},\n\t\t\t\t\tfunc() error {\n\t\t\t\t\t\treturn errors.New(\"callback failed\")\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"callback failed\"),\n\t\t\texpectedBytes: 500,\n\t\t\texpectedCount: 50,\n\t\t\tdescription:   \"Should release on callback error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmgr := NewBudgetManager(\n\t\t\t\t\"test\",\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityBytes),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(tt.capacityCount),\n\t\t\t\tAdmissionOptimistic,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tnil,\n\t\t\t)\n\n\t\t\terr := tt.operation(mgr)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Error(t, err, tt.description)\n\t\t\t\tassert.Equal(t, tt.expectedError.Error(), err.Error(), tt.description)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, tt.description)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedBytes, mgr.UsedBytes(), \"Used bytes mismatch\")\n\t\t\tassert.Equal(t, tt.expectedCount, mgr.UsedCount(), \"Used count mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestBudgetManager_ConcurrentCorrectness(t *testing.T) {\n\tmgr := NewBudgetManager(\n\t\t\"test\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(t),\n\t\tnil,\n\t)\n\n\tconst (\n\t\tnumGoroutines = 100\n\t\tnumOperations = 1000\n\t\treserveBytes  = 100\n\t\treserveCount  = 1\n\t)\n\n\t// Track total operations for verification\n\tvar totalReservedBytes atomic.Uint64\n\tvar totalReservedCount atomic.Int64\n\tvar totalReleasedBytes atomic.Uint64\n\tvar totalReleasedCount atomic.Int64\n\n\t// Run concurrent reserve/release operations\n\t// Each goroutine will reserve some and release some (but not all)\n\tvar wg sync.WaitGroup\n\twg.Add(numGoroutines)\n\n\tfor g := 0; g < numGoroutines; g++ {\n\t\tgo func(goroutineID int) {\n\t\t\tdefer wg.Done()\n\t\t\tcacheID := fmt.Sprintf(\"cache%d\", goroutineID%10)\n\n\t\t\tfor i := 0; i < numOperations; i++ {\n\t\t\t\t// Reserve\n\t\t\t\terr := mgr.(*manager).ReserveForCache(cacheID, reserveBytes, reserveCount)\n\t\t\t\tif err == nil {\n\t\t\t\t\ttotalReservedBytes.Add(reserveBytes)\n\t\t\t\t\ttotalReservedCount.Add(reserveCount)\n\n\t\t\t\t\t// Release only half the time to maintain non-zero state\n\t\t\t\t\tif i%2 == 0 {\n\t\t\t\t\t\tmgr.(*manager).ReleaseForCache(cacheID, reserveBytes, reserveCount)\n\t\t\t\t\t\ttotalReleasedBytes.Add(reserveBytes)\n\t\t\t\t\t\ttotalReleasedCount.Add(reserveCount)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}(g)\n\t}\n\n\twg.Wait()\n\n\t// Verify invariants without assuming zero state\n\t// 1. Manager's used bytes should equal total reserved - total released\n\texpectedBytes := totalReservedBytes.Load() - totalReleasedBytes.Load()\n\texpectedCount := totalReservedCount.Load() - totalReleasedCount.Load()\n\n\tassert.Equal(t, expectedBytes, mgr.UsedBytes(),\n\t\t\"Used bytes should equal total reserved minus total released\")\n\tassert.Equal(t, expectedCount, mgr.UsedCount(),\n\t\t\"Used count should equal total reserved minus total released\")\n\n\t// 2. Available capacity should equal total - used\n\tassert.Equal(t, uint64(1000000)-expectedBytes, mgr.(*manager).AvailableBytes(),\n\t\t\"Available bytes should equal capacity minus used\")\n\tassert.Equal(t, int64(100000)-expectedCount, mgr.(*manager).AvailableCount(),\n\t\t\"Available count should equal capacity minus used\")\n\n\t// 3. Active cache count should be > 0 since we have unreleased capacity\n\tif expectedBytes > 0 || expectedCount > 0 {\n\t\tassert.Greater(t, mgr.(*manager).getActiveCacheCount(), int64(0),\n\t\t\t\"Should have active caches when capacity is in use\")\n\t}\n\n\t// 4. Log final state for debugging\n\tt.Logf(\"Final state: reserved=%d bytes, released=%d bytes, used=%d bytes, active_caches=%d\",\n\t\ttotalReservedBytes.Load(), totalReleasedBytes.Load(), mgr.UsedBytes(),\n\t\tmgr.(*manager).getActiveCacheCount())\n}\n\nfunc TestBudgetManager_ConcurrentSoftCapCorrectness(t *testing.T) {\n\tmgr := NewBudgetManager(\n\t\t\"test\",\n\t\tdynamicproperties.GetIntPropertyFn(10000), // Smaller capacity\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(t),\n\t\tdynamicproperties.GetFloatPropertyFn(0.5), // 50% soft cap = 5000 bytes free\n\t)\n\n\tconst (\n\t\tnumGoroutines = 50\n\t\tnumOperations = 100\n\t\treserveBytes  = 200\n\t\treserveCount  = 2\n\t)\n\n\tvar totalReservedBytes atomic.Uint64\n\tvar totalReservedCount atomic.Int64\n\tvar totalReleasedBytes atomic.Uint64\n\tvar totalReleasedCount atomic.Int64\n\tvar totalSuccessful atomic.Int64\n\tvar totalFailed atomic.Int64\n\n\tvar wg sync.WaitGroup\n\twg.Add(numGoroutines)\n\n\t// Try to reserve concurrently - soft cap should cause some failures\n\t// Release only half to maintain non-zero state\n\tfor g := 0; g < numGoroutines; g++ {\n\t\tgo func(goroutineID int) {\n\t\t\tdefer wg.Done()\n\t\t\tcacheID := fmt.Sprintf(\"cache%d\", goroutineID%10)\n\n\t\t\tfor i := 0; i < numOperations; i++ {\n\t\t\t\terr := mgr.(*manager).ReserveForCache(cacheID, reserveBytes, reserveCount)\n\t\t\t\tif err == nil {\n\t\t\t\t\ttotalSuccessful.Add(1)\n\t\t\t\t\ttotalReservedBytes.Add(reserveBytes)\n\t\t\t\t\ttotalReservedCount.Add(reserveCount)\n\n\t\t\t\t\t// Release only half the time to maintain non-zero state\n\t\t\t\t\tif i%2 == 0 {\n\t\t\t\t\t\tmgr.(*manager).ReleaseForCache(cacheID, reserveBytes, reserveCount)\n\t\t\t\t\t\ttotalReleasedBytes.Add(reserveBytes)\n\t\t\t\t\t\ttotalReleasedCount.Add(reserveCount)\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\ttotalFailed.Add(1)\n\t\t\t\t}\n\t\t\t}\n\t\t}(g)\n\t}\n\n\twg.Wait()\n\n\t// Verify invariants without assuming zero state\n\texpectedBytes := totalReservedBytes.Load() - totalReleasedBytes.Load()\n\texpectedCount := totalReservedCount.Load() - totalReleasedCount.Load()\n\n\tassert.Equal(t, expectedBytes, mgr.UsedBytes(),\n\t\t\"Used bytes should equal total reserved minus total released\")\n\tassert.Equal(t, expectedCount, mgr.UsedCount(),\n\t\t\"Used count should equal total reserved minus total released\")\n\n\t// Available capacity should equal total - used\n\tassert.Equal(t, uint64(10000)-expectedBytes, mgr.(*manager).AvailableBytes(),\n\t\t\"Available bytes should equal capacity minus used\")\n\tassert.Equal(t, int64(1000)-expectedCount, mgr.(*manager).AvailableCount(),\n\t\t\"Available count should equal capacity minus used\")\n\n\t// Active cache count should be > 0 since we have unreleased capacity\n\tif expectedBytes > 0 || expectedCount > 0 {\n\t\tassert.Greater(t, mgr.(*manager).getActiveCacheCount(), int64(0),\n\t\t\t\"Should have active caches when capacity is in use\")\n\t\tassert.NotZero(t, mgr.UsedBytes(), \"Should have non-zero bytes in use\")\n\t\tassert.NotZero(t, mgr.UsedCount(), \"Should have non-zero count in use\")\n\t}\n\n\t// Log statistics\n\tt.Logf(\"Final state: reserved=%d bytes, released=%d bytes, used=%d bytes, active_caches=%d\",\n\t\ttotalReservedBytes.Load(), totalReleasedBytes.Load(), mgr.UsedBytes(),\n\t\tmgr.(*manager).getActiveCacheCount())\n\tt.Logf(\"Operations: successful=%d, failed=%d, total=%d\",\n\t\ttotalSuccessful.Load(), totalFailed.Load(),\n\t\ttotalSuccessful.Load()+totalFailed.Load())\n\n\t// At least some operations should have succeeded\n\tassert.Greater(t, totalSuccessful.Load(), int64(0),\n\t\t\"Some operations should succeed\")\n\t// Soft cap should have caused some failures\n\tassert.Greater(t, totalFailed.Load(), int64(0),\n\t\t\"Soft cap should cause some failures\")\n}\n\nfunc TestBudgetManager_ConcurrentActiveCacheCount(t *testing.T) {\n\tmgr := NewBudgetManager(\n\t\t\"test\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(t),\n\t\tdynamicproperties.GetFloatPropertyFn(0.5),\n\t)\n\n\tconst numCaches = 20\n\n\t// Activate all caches\n\tfor i := 0; i < numCaches; i++ {\n\t\tcacheID := fmt.Sprintf(\"cache%d\", i)\n\t\terr := mgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t\tassert.NoError(t, err)\n\t}\n\n\t// Verify active cache count\n\tassert.Equal(t, int64(numCaches), mgr.(*manager).getActiveCacheCount(),\n\t\t\"All caches should be active\")\n\n\t// Deactivate all caches concurrently\n\tvar wg sync.WaitGroup\n\twg.Add(numCaches)\n\tfor i := 0; i < numCaches; i++ {\n\t\tgo func(cacheID string) {\n\t\t\tdefer wg.Done()\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 100, 1)\n\t\t}(fmt.Sprintf(\"cache%d\", i))\n\t}\n\twg.Wait()\n\n\t// Verify all caches are inactive\n\tassert.Equal(t, int64(0), mgr.(*manager).getActiveCacheCount(),\n\t\t\"All caches should be inactive after release\")\n\tassert.Equal(t, uint64(0), mgr.UsedBytes(),\n\t\t\"All bytes should be released\")\n}\n\nfunc BenchmarkBudgetManager_ReserveRelease(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 100, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_ReserveBytes(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.(*manager).ReserveBytesForCache(cacheID, 100)\n\t\t\tmgr.(*manager).ReleaseBytesForCache(cacheID, 100)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_ReserveCount(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.(*manager).ReserveCountForCache(cacheID, 1)\n\t\t\tmgr.(*manager).ReleaseCountForCache(cacheID, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_ReserveWithCallback(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.ReserveWithCallback(cacheID, 100, 1, func() error {\n\t\t\t\treturn nil\n\t\t\t})\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 100, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_ReleaseWithCallback(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tfor i := 0; i < 1000; i++ {\n\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\tmgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t}\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.ReleaseWithCallback(cacheID, func() (uint64, int64, error) {\n\t\t\t\treturn 100, 1, nil\n\t\t\t})\n\t\t\tmgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_ReserveOrReclaimSelfRelease(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(10000),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tmgr.(*manager).ReserveForCache(\"cache1\", 9000, 900)\n\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\tmgr.(*manager).ReserveOrReclaimSelfRelease(\n\t\t\tcontext.Background(),\n\t\t\t\"cache1\",\n\t\t\t2000,\n\t\t\t200,\n\t\t\ttrue,\n\t\t\tfunc(needBytes uint64, needCount int64) {\n\t\t\t\tmgr.(*manager).ReleaseForCache(\"cache1\", needBytes, needCount)\n\t\t\t},\n\t\t)\n\t\tmgr.(*manager).ReleaseForCache(\"cache1\", 2000, 200)\n\t}\n}\n\nfunc BenchmarkBudgetManager_ReserveOrReclaimManagerRelease(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(10000),\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tmgr.(*manager).ReserveForCache(\"cache1\", 9000, 900)\n\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\tmgr.(*manager).ReserveOrReclaimManagerRelease(\n\t\t\tcontext.Background(),\n\t\t\t\"cache1\",\n\t\t\t2000,\n\t\t\t200,\n\t\t\ttrue,\n\t\t\tfunc(needBytes uint64, needCount int64) (uint64, int64) {\n\t\t\t\treturn needBytes, needCount\n\t\t\t},\n\t\t)\n\t\tmgr.(*manager).ReleaseForCache(\"cache1\", 2000, 200)\n\t}\n}\n\nfunc BenchmarkBudgetManager_MultipleCaches(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tnumCaches := 100\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%numCaches))\n\t\t\tmgr.(*manager).ReserveForCache(cacheID, 1000, 10)\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 1000, 10)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_StrictMode(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionStrict,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := \"cache\" + string(rune(i%10))\n\t\t\tmgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 100, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_SoftCap(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(1000000),\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tdynamicproperties.GetFloatPropertyFn(0.5),\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\ti := 0\n\t\tfor pb.Next() {\n\t\t\tcacheID := fmt.Sprintf(\"cache%d\", i%10)\n\t\t\tmgr.(*manager).ReserveForCache(cacheID, 100, 1)\n\t\t\tmgr.(*manager).ReleaseForCache(cacheID, 100, 1)\n\t\t\ti++\n\t\t}\n\t})\n}\n\nfunc BenchmarkBudgetManager_HighContention(b *testing.B) {\n\tmgr := NewBudgetManager(\n\t\t\"benchmark\",\n\t\tdynamicproperties.GetIntPropertyFn(100000),\n\t\tdynamicproperties.GetIntPropertyFn(10000),\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tnil,\n\t\ttestlogger.New(b),\n\t\tnil,\n\t)\n\n\tb.ResetTimer()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\tfor pb.Next() {\n\t\t\tmgr.(*manager).ReserveForCache(\"shared-cache\", 100, 1)\n\t\t\tmgr.(*manager).ReleaseForCache(\"shared-cache\", 100, 1)\n\t\t}\n\t})\n}\n\nfunc TestBudgetManager_DisabledManager_NoMetricsEmitted(t *testing.T) {\n\t// Create a mock scope that will track all metric calls\n\tmockScope := &metricsmocks.Scope{}\n\t// Mock the Tagged call which returns itself\n\tmockScope.On(\"Tagged\", mock.Anything).Return(mockScope)\n\n\t// Create manager with BOTH capacities disabled (set to 0)\n\tmgr := NewBudgetManager(\n\t\t\"test-disabled-manager\",\n\t\tdynamicproperties.GetIntPropertyFn(0), // maxBytes = 0 (disabled)\n\t\tdynamicproperties.GetIntPropertyFn(0), // maxCount = 0 (disabled)\n\t\tAdmissionOptimistic,\n\t\t0,\n\t\tmockScope,\n\t\ttestlogger.New(t),\n\t\tdynamicproperties.GetFloatPropertyFn(0.8),\n\t)\n\tdefer mgr.Stop()\n\n\t// Give time for any initial metrics to be emitted\n\ttime.Sleep(100 * time.Millisecond)\n\n\t// Try to reserve capacity (should fail but shouldn't emit metrics)\n\terr := mgr.(*manager).ReserveForCache(\"cache1\", 100, 1)\n\tassert.Error(t, err)\n\n\t// Try to release capacity (should not emit metrics)\n\tmgr.(*manager).ReleaseForCache(\"cache1\", 100, 1)\n\n\t// Wait a bit to ensure no async metrics are emitted\n\ttime.Sleep(100 * time.Millisecond)\n\n\t// Verify NO metrics were emitted (no calls to UpdateGauge or IncCounter)\n\tmockScope.AssertNotCalled(t, \"UpdateGauge\")\n\tmockScope.AssertNotCalled(t, \"IncCounter\")\n}\n"
  },
  {
    "path": "common/cache/cache.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/cache\n\npackage cache\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// A Cache is a generalized interface to a cache.  See cache.LRU for a specific\n// implementation (bounded cache with LRU eviction)\ntype Cache interface {\n\t// Get retrieves an element based on a key, returning nil if the element\n\t// does not exist\n\tGet(key interface{}) interface{}\n\n\t// Put adds an element to the cache, returning the previous element\n\tPut(key interface{}, value interface{}) interface{}\n\n\t// PutIfNotExist puts a value associated with a given key if it does not exist\n\tPutIfNotExist(key interface{}, value interface{}) (interface{}, error)\n\n\t// Delete deletes an element in the cache\n\tDelete(key interface{})\n\n\t// Release decrements the ref count of a pinned element. If the ref count\n\t// drops to 0, the element can be evicted from the cache.\n\tRelease(key interface{})\n\n\t// Iterator returns the iterator of the cache\n\tIterator() Iterator\n\n\t// Size returns the number of entries currently stored in the Cache\n\tSize() int\n}\n\n// Options control the behavior of the cache\ntype Options struct {\n\t// TTL controls the time-to-live for a given cache entry.  Cache entries that\n\t// are older than the TTL will not be returned.\n\tTTL time.Duration\n\n\t// InitialCapacity controls the initial capacity of the cache\n\tInitialCapacity int\n\n\t// Pin prevents in-use objects from getting evicted.\n\tPin bool\n\n\t// RemovedFunc is an optional function called when an element\n\t// is scheduled for deletion\n\tRemovedFunc RemovedFunc\n\n\t// MaxCount controls the max capacity of the cache\n\t// It is required option if MaxSize is not provided\n\tMaxCount int\n\n\t// MaxSize is an optional flag, but it has to be used along with a value that implements Sizeable() interface\n\t// to control the max size in bytes of the cache\n\t// It is required option if MaxCount is not provided\n\tMaxSize dynamicproperties.IntPropertyFn\n\n\t// ActivelyEvict will evict items that has expired TTL at every operation in the cache\n\t// This can be expensive if a lot of items expire at the same time\n\t// Should be used when it's important for memory that the expired items are evicted as soon as possible\n\t// If not set expired items will be evicted when one of these happens\n\t// - when the cache is full\n\t// - when the item is accessed\n\tActivelyEvict bool\n\n\t// TimeSource is used to get the current time\n\t// It is optional and defaults to clock.NewRealTimeSource()\n\tTimeSource clock.TimeSource\n\n\t// IsSizeBased is an optional flag to indicate if the cache is size based\n\t// It's default is false, but if set to true, the cache will evict items based on item size instead of count\n\t// But the item HAS to be able to cast as a Sizeable interface otherwise the cache will fail\n\tIsSizeBased dynamicproperties.BoolPropertyFn\n\n\t// MetricsScope is used to emit metrics for internals of the cache\n\tMetricsScope metrics.Scope\n\n\t// Logger is used to emit logs for internals of the cache\n\tLogger log.Logger\n\n\t// Deprecated: GetCacheItemSizeFunc is a function called upon adding the item to update the cache size.\n\t// It returns 0 by default, assuming the cache is just count based\n\t// It is required option if MaxCount is not provided\n\tGetCacheItemSizeFunc GetCacheItemSizeFunc\n}\n\n// SimpleOptions provides options that can be used to configure SimpleCache\ntype SimpleOptions struct {\n\t// InitialCapacity controls the initial capacity of the cache\n\tInitialCapacity int\n\n\t// RemovedFunc is an optional function called when an element\n\t// is scheduled for deletion\n\tRemovedFunc RemovedFunc\n}\n\n// RemovedFunc is a type for notifying applications when an item is\n// scheduled for removal from the Cache. If f is a function with the\n// appropriate signature and i is the interface{} scheduled for\n// deletion, Cache calls go f(i)\ntype RemovedFunc func(interface{})\n\n// Iterator represents the interface for cache iterators\ntype Iterator interface {\n\t// Close closes the iterator\n\t// and releases any allocated resources\n\tClose()\n\t// HasNext return true if there is more items to be returned\n\tHasNext() bool\n\t// Next return the next item\n\tNext() Entry\n}\n\n// Entry represents a key-value entry within the map\ntype Entry interface {\n\t// Key represents the key\n\tKey() interface{}\n\t// Value represents the value\n\tValue() interface{}\n\t// CreateTime represents the time when the entry is created\n\tCreateTime() time.Time\n}\n\n// GetCacheItemSizeFunc returns the cache item size in bytes\ntype GetCacheItemSizeFunc func(interface{}) uint64\n\n// DomainMetricsScopeCache represents a interface for mapping domainID and scopeIdx to metricsScope\ntype DomainMetricsScopeCache interface {\n\t// Get retrieves metrics scope for a domainID and scopeIdx\n\tGet(domainID string, scopeIdx metrics.ScopeIdx) (metrics.Scope, bool)\n\t// Put adds metrics scope for a domainID and scopeIdx\n\tPut(domainID string, scopeIdx metrics.ScopeIdx, metricsScope metrics.Scope)\n\n\tcommon.Daemon\n}\n\n// Sizeable is a interface that implements ByteSize() function\ntype Sizeable interface {\n\t// ByteSize returns an approximate size of the object in bytes\n\tByteSize() uint64\n}\n"
  },
  {
    "path": "common/cache/domainCache.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination domainCache_mock.go -self_package github.com/uber/cadence/common/cache\n\npackage cache\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"hash/fnv\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ReplicationPolicy is the domain's replication policy,\n// derived from domain's replication config\ntype ReplicationPolicy int\n\nconst (\n\t// ReplicationPolicyOneCluster indicate that workflows does not need to be replicated\n\t// applicable to local domain & global domain with one cluster\n\tReplicationPolicyOneCluster ReplicationPolicy = 0\n\t// ReplicationPolicyMultiCluster indicate that workflows need to be replicated\n\tReplicationPolicyMultiCluster ReplicationPolicy = 1\n)\n\nconst (\n\tdomainCacheInitialSize        = 10 * 1024\n\tdomainCacheMinRefreshInterval = 1 * time.Second\n\t// DomainCacheRefreshInterval domain cache refresh interval\n\tDomainCacheRefreshInterval = 10 * time.Second\n\tdomainCacheRefreshPageSize = 200\n\n\tdomainCachePersistenceTimeout = 3 * time.Second\n\n\tdomainCacheInitialized int32 = 0\n\tdomainCacheStarted     int32 = 1\n\tdomainCacheStopped     int32 = 2\n)\n\ntype (\n\t// PrepareCallbackFn is function to be called before CallbackFn is called,\n\t// it is guaranteed that PrepareCallbackFn and CallbackFn pair will be both called or non will be called\n\tPrepareCallbackFn func()\n\n\t// CallbackFn is function to be called when the domain cache entries are changed\n\t// it is guaranteed that  CallbackFn pair will be both called or non will be called\n\tCallbackFn func(updatedDomains []*DomainCacheEntry)\n\n\t// CatchUpFn is a function to execute PrepareCallbackFn and/or CallbackFn during domain callback registration,\n\t// allowing the caller to catch up with the latest domain changes\n\tCatchUpFn func(domainCache DomainCache, prepareCallback PrepareCallbackFn, callback CallbackFn)\n\n\t// DomainCache is used the cache domain information and configuration to avoid making too many calls to cassandra.\n\t// This cache is mainly used by frontend for resolving domain names to domain uuids which are used throughout the\n\t// system.  Each domain entry is kept in the cache for one hour but also has an expiry of 10 seconds.  This results\n\t// in updating the domain entry every 10 seconds but in the case of a cassandra failure we can still keep on serving\n\t// requests using the stale entry from cache upto an hour\n\tDomainCache interface {\n\t\tcommon.Daemon\n\t\tRegisterDomainChangeCallback(id string, catchUpFn CatchUpFn, prepareCallback PrepareCallbackFn, callback CallbackFn)\n\t\tUnregisterDomainChangeCallback(id string)\n\t\tGetDomain(name string) (*DomainCacheEntry, error)\n\t\tGetDomainByID(id string) (*DomainCacheEntry, error)\n\t\tGetDomainID(name string) (string, error)\n\t\tGetDomainName(id string) (string, error)\n\t\tGetAllDomain() map[string]*DomainCacheEntry\n\t\tGetCacheSize() (sizeOfCacheByName int64, sizeOfCacheByID int64)\n\t}\n\n\tDefaultDomainCache struct {\n\t\tstatus          int32\n\t\tshutdownChan    chan struct{}\n\t\tclusterGroup    string\n\t\tclusterMetadata cluster.Metadata\n\t\tcacheNameToID   *atomic.Value\n\t\tcacheByID       *atomic.Value\n\t\tdomainManager   persistence.DomainManager\n\t\ttimeSource      clock.TimeSource\n\t\tscope           metrics.Scope\n\t\tlogger          log.Logger\n\n\t\t// refresh lock is used to guarantee at most one\n\t\t// coroutine is doing domain refreshment\n\t\trefreshLock     sync.Mutex\n\t\tlastRefreshTime time.Time\n\t\t// This is debug field to emit callback count\n\t\tlastCallbackEmitTime time.Time\n\n\t\tcallbackLock     sync.Mutex\n\t\tprepareCallbacks map[string]PrepareCallbackFn\n\t\tcallbacks        map[string]CallbackFn\n\n\t\tthrottleRetry *backoff.ThrottleRetry\n\n\t\t// ctx and cancel are used to control the lifecycle of background operations\n\t\tctx    context.Context\n\t\tcancel context.CancelFunc\n\t}\n\n\t// DomainCacheEntries is DomainCacheEntry slice\n\tDomainCacheEntries []*DomainCacheEntry\n\n\t// DomainCacheEntry contains the info and config for a domain\n\tDomainCacheEntry struct {\n\t\tmu                          sync.RWMutex\n\t\tinfo                        *persistence.DomainInfo\n\t\tconfig                      *persistence.DomainConfig\n\t\treplicationConfig           *persistence.DomainReplicationConfig\n\t\tconfigVersion               int64\n\t\tfailoverVersion             int64\n\t\tisGlobalDomain              bool\n\t\tfailoverNotificationVersion int64\n\t\tpreviousFailoverVersion     int64\n\t\tfailoverEndTime             *int64\n\t\tnotificationVersion         int64\n\t\tinitialized                 bool\n\n\t\t// list of active clusters for active-active domains. initialized from replication config.\n\t\tactiveClusters []string\n\t}\n)\n\ntype DomainCacheOption func(*DefaultDomainCache)\n\nfunc WithTimeSource(timeSource clock.TimeSource) DomainCacheOption {\n\treturn func(cache *DefaultDomainCache) {\n\t\tif timeSource != nil {\n\t\t\tcache.timeSource = timeSource\n\t\t}\n\t}\n}\n\n// NewDomainCache creates a new instance of cache for holding onto domain information to reduce the load on persistence\nfunc NewDomainCache(\n\tdomainManager persistence.DomainManager,\n\tmetadata cluster.Metadata,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\topts ...DomainCacheOption,\n) *DefaultDomainCache {\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreateDomainCacheRetryPolicy()),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tcache := &DefaultDomainCache{\n\t\tstatus:           domainCacheInitialized,\n\t\tshutdownChan:     make(chan struct{}),\n\t\tclusterGroup:     getClusterGroupIdentifier(metadata),\n\t\tclusterMetadata:  metadata,\n\t\tcacheNameToID:    &atomic.Value{},\n\t\tcacheByID:        &atomic.Value{},\n\t\tdomainManager:    domainManager,\n\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\tscope:            metricsClient.Scope(metrics.DomainCacheScope),\n\t\tlogger:           logger,\n\t\tprepareCallbacks: make(map[string]PrepareCallbackFn),\n\t\tcallbacks:        make(map[string]CallbackFn),\n\t\tthrottleRetry:    throttleRetry,\n\t\tctx:              ctx,\n\t\tcancel:           cancel,\n\t}\n\tcache.cacheNameToID.Store(newDomainCache())\n\tcache.cacheByID.Store(newDomainCache())\n\n\tfor _, opt := range opts {\n\t\topt(cache)\n\t}\n\n\treturn cache\n}\n\nfunc getClusterGroupIdentifier(metadata cluster.Metadata) string {\n\tvar clusters []string\n\tfor cluster := range metadata.GetEnabledClusterInfo() {\n\t\tclusters = append(clusters, cluster)\n\t}\n\tsort.Strings(clusters)\n\treturn strings.Join(clusters, \"_\")\n}\n\nfunc newDomainCache() Cache {\n\treturn NewSimple(&SimpleOptions{\n\t\tInitialCapacity: domainCacheInitialSize,\n\t})\n}\n\n// NewGlobalDomainCacheEntryForTest returns an entry with test data\nfunc NewGlobalDomainCacheEntryForTest(\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\trepConfig *persistence.DomainReplicationConfig,\n\tfailoverVersion int64,\n) *DomainCacheEntry {\n\n\treturn &DomainCacheEntry{\n\t\tinfo:              info,\n\t\tconfig:            config,\n\t\tisGlobalDomain:    true,\n\t\treplicationConfig: repConfig,\n\t\tfailoverVersion:   failoverVersion,\n\t}\n}\n\n// NewLocalDomainCacheEntryForTest returns an entry with test data\nfunc NewLocalDomainCacheEntryForTest(\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\ttargetCluster string,\n) *DomainCacheEntry {\n\n\treturn &DomainCacheEntry{\n\t\tinfo:           info,\n\t\tconfig:         config,\n\t\tisGlobalDomain: false,\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: targetCluster,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: targetCluster}},\n\t\t},\n\t\tfailoverVersion: constants.EmptyVersion,\n\t}\n}\n\n// NewDomainCacheEntryForTest returns an entry with test data\nfunc NewDomainCacheEntryForTest(\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\tisGlobalDomain bool,\n\trepConfig *persistence.DomainReplicationConfig,\n\tfailoverVersion int64,\n\tfailoverEndtime *int64,\n\tfailoverNotificationVersion int64,\n\tpreviousFailoverVersion int64,\n\tnotificationVersion int64,\n) *DomainCacheEntry {\n\treturn &DomainCacheEntry{\n\t\tinfo:                        info,\n\t\tconfig:                      config,\n\t\tisGlobalDomain:              isGlobalDomain,\n\t\treplicationConfig:           repConfig,\n\t\tfailoverVersion:             failoverVersion,\n\t\tfailoverEndTime:             failoverEndtime,\n\t\tfailoverNotificationVersion: failoverNotificationVersion,\n\t\tpreviousFailoverVersion:     previousFailoverVersion,\n\t\tnotificationVersion:         notificationVersion,\n\t\tactiveClusters:              getActiveClusters(repConfig),\n\t}\n}\n\nfunc (c *DefaultDomainCache) GetCacheSize() (sizeOfCacheByName int64, sizeOfCacheByID int64) {\n\treturn int64(c.cacheByID.Load().(Cache).Size()), int64(c.cacheNameToID.Load().(Cache).Size())\n}\n\n// Start starts the background refresh of domain\nfunc (c *DefaultDomainCache) Start() {\n\tif !atomic.CompareAndSwapInt32(&c.status, domainCacheInitialized, domainCacheStarted) {\n\t\treturn\n\t}\n\n\t// initialize the cache by initial scan\n\terr := c.refreshDomains()\n\tif err != nil {\n\t\tc.logger.Fatal(\"Unable to initialize domain cache\", tag.Error(err))\n\t}\n\tgo c.refreshLoop()\n}\n\n// Stop stops background refresh of domain\nfunc (c *DefaultDomainCache) Stop() {\n\tif !atomic.CompareAndSwapInt32(&c.status, domainCacheStarted, domainCacheStopped) {\n\t\treturn\n\t}\n\tc.cancel() // Cancel the context first\n\tclose(c.shutdownChan)\n}\n\nfunc (c *DefaultDomainCache) GetAllDomain() map[string]*DomainCacheEntry {\n\tresult := make(map[string]*DomainCacheEntry)\n\tite := c.cacheByID.Load().(Cache).Iterator()\n\tdefer ite.Close()\n\n\tfor ite.HasNext() {\n\t\tentry := ite.Next()\n\t\tid := entry.Key().(string)\n\t\tdomainCacheEntry := entry.Value().(*DomainCacheEntry)\n\t\tdomainCacheEntry.mu.RLock()\n\t\tdup := domainCacheEntry.duplicate()\n\t\tdomainCacheEntry.mu.RUnlock()\n\t\tresult[id] = dup\n\t}\n\treturn result\n}\n\n// RegisterDomainChangeCallback set a domain change callback\n// WARN: the beforeCallback function will be triggered by domain cache when holding the domain cache lock,\n// make sure the callback function will not call domain cache again in case of dead lock\n// afterCallback will be invoked when NOT holding the domain cache lock.\nfunc (c *DefaultDomainCache) RegisterDomainChangeCallback(\n\tid string,\n\tcatchUpFn CatchUpFn,\n\tprepareCallback PrepareCallbackFn,\n\tcallback CallbackFn,\n) {\n\n\tc.callbackLock.Lock()\n\tc.prepareCallbacks[id] = prepareCallback\n\tc.callbacks[id] = callback\n\tc.callbackLock.Unlock()\n\n\tcatchUpFn(c, prepareCallback, callback)\n\n}\n\n// UnregisterDomainChangeCallback delete a domain failover callback\nfunc (c *DefaultDomainCache) UnregisterDomainChangeCallback(\n\tid string,\n) {\n\n\tc.callbackLock.Lock()\n\tdefer c.callbackLock.Unlock()\n\n\tdelete(c.prepareCallbacks, id)\n\tdelete(c.callbacks, id)\n}\n\n// GetDomain retrieves the information from the cache if it exists, otherwise retrieves the information from metadata\n// store and writes it to the cache with an expiry before returning back\nfunc (c *DefaultDomainCache) GetDomain(\n\tname string,\n) (*DomainCacheEntry, error) {\n\n\tif name == \"\" {\n\t\treturn nil, &types.BadRequestError{Message: \"Domain name is empty\"}\n\t}\n\n\treturn c.getDomain(name)\n}\n\n// GetDomainByID retrieves the information from the cache if it exists, otherwise retrieves the information from metadata\n// store and writes it to the cache with an expiry before returning back\nfunc (c *DefaultDomainCache) GetDomainByID(\n\tid string,\n) (*DomainCacheEntry, error) {\n\n\tif id == \"\" {\n\t\treturn nil, &types.BadRequestError{Message: \"DomainID is empty.\"}\n\t}\n\treturn c.getDomainByID(id, true)\n}\n\n// GetDomainID retrieves domainID by using GetDomain\nfunc (c *DefaultDomainCache) GetDomainID(\n\tname string,\n) (string, error) {\n\n\tentry, err := c.GetDomain(name)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn entry.info.ID, nil\n}\n\n// GetDomainName returns domain name given the domain id\nfunc (c *DefaultDomainCache) GetDomainName(\n\tid string,\n) (string, error) {\n\n\tentry, err := c.getDomainByID(id, false)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn entry.info.Name, nil\n}\n\nfunc (c *DefaultDomainCache) refreshLoop() {\n\ttimer := c.timeSource.NewTicker(DomainCacheRefreshInterval)\n\tdefer timer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-c.shutdownChan:\n\t\t\treturn\n\t\tcase <-timer.Chan():\n\t\t\terr := c.refreshDomains()\n\t\t\tif err != nil {\n\t\t\t\tc.logger.Error(\"Error refreshing domain cache\", tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tc.logger.Debug(\"Domain cache refreshed\")\n\t\t}\n\t}\n}\n\nfunc (c *DefaultDomainCache) refreshDomains() error {\n\tc.refreshLock.Lock()\n\tdefer c.refreshLock.Unlock()\n\treturn c.throttleRetry.Do(c.ctx, c.refreshDomainsLocked)\n}\n\n// this function only refresh the domains in the v2 table\n// the domains in the v1 table will be refreshed if cache is stale\nfunc (c *DefaultDomainCache) refreshDomainsLocked(ctx context.Context) error {\n\tnow := c.timeSource.Now()\n\tif now.Sub(c.lastRefreshTime) < domainCacheMinRefreshInterval {\n\t\treturn nil\n\t}\n\n\t// first load the metadata record, then load domains\n\t// this can guarantee that domains in the cache are not updated more than metadata record\n\tctx, cancel := context.WithTimeout(ctx, domainCachePersistenceTimeout)\n\tdefer cancel()\n\tmetadata, err := c.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar token []byte\n\trequest := &persistence.ListDomainsRequest{PageSize: domainCacheRefreshPageSize}\n\tvar domains DomainCacheEntries\n\tcontinuePage := true\n\n\tfor continuePage {\n\t\tctx, cancel := context.WithTimeout(ctx, domainCachePersistenceTimeout)\n\t\trequest.NextPageToken = token\n\t\tresponse, err := c.domainManager.ListDomains(ctx, request)\n\t\tcancel()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttoken = response.NextPageToken\n\t\tfor _, domain := range response.Domains {\n\t\t\tdomains = append(domains, c.buildEntryFromRecord(domain))\n\t\t}\n\t\tcontinuePage = len(token) != 0\n\t}\n\n\t// we mush apply the domain change by order\n\t// since history shard have to update the shard info\n\t// with domain change version.\n\tsort.Sort(domains)\n\tvar updatedEntries []*DomainCacheEntry\n\n\t// make a copy of the existing domain cache, so we can calculate diff and do compare and swap\n\tnewCacheNameToID := newDomainCache()\n\tnewCacheByID := newDomainCache()\n\tfor _, domain := range c.GetAllDomain() {\n\t\tnewCacheNameToID.Put(domain.info.Name, domain.info.ID)\n\t\tnewCacheByID.Put(domain.info.ID, domain)\n\t}\n\nUpdateLoop:\n\tfor _, domain := range domains {\n\t\tif domain.notificationVersion >= metadata.NotificationVersion {\n\t\t\t// this guarantee that domain change events before the\n\t\t\t// domainNotificationVersion is loaded into the cache.\n\n\t\t\t// the domain change events after the domainNotificationVersion\n\t\t\t// will be loaded into cache in the next refresh\n\t\t\tc.logger.Info(\"Domain notification is not less than than metadata notification version\", tag.WorkflowDomainName(domain.GetInfo().Name))\n\t\t\tbreak UpdateLoop\n\t\t}\n\t\ttriggerCallback, nextEntry, err := c.updateIDToDomainCache(newCacheByID, domain.info.ID, domain)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tc.scope.Tagged(\n\t\t\tmetrics.DomainTag(nextEntry.info.Name),\n\t\t\tmetrics.DomainTypeTag(nextEntry.isGlobalDomain),\n\t\t\tmetrics.ClusterGroupTag(c.clusterGroup),\n\t\t\tmetrics.ActiveClusterTag(nextEntry.replicationConfig.ActiveClusterName),\n\t\t\tmetrics.IsActiveActiveDomainTag(nextEntry.replicationConfig.IsActiveActive()),\n\t\t).UpdateGauge(metrics.ActiveClusterGauge, 1)\n\n\t\tc.updateNameToIDCache(newCacheNameToID, nextEntry.info.Name, nextEntry.info.ID)\n\n\t\tif triggerCallback {\n\t\t\tupdatedEntries = append(updatedEntries, nextEntry)\n\t\t}\n\t}\n\n\t// NOTE: READ REF BEFORE MODIFICATION\n\t// ref: historyEngine.go registerDomainFailoverCallback function\n\tc.callbackLock.Lock()\n\tdefer c.callbackLock.Unlock()\n\tc.triggerDomainChangePrepareCallbackLocked()\n\tc.cacheByID.Store(newCacheByID)\n\tc.cacheNameToID.Store(newCacheNameToID)\n\tc.triggerDomainChangeCallbackLocked(updatedEntries)\n\n\t// only update last refresh time when refresh succeeded\n\tc.lastRefreshTime = now\n\tif now.Sub(c.lastCallbackEmitTime) > 30*time.Minute {\n\t\tc.lastCallbackEmitTime = now\n\t\tc.scope.AddCounter(metrics.DomainCacheCallbacksCount, int64(len(c.callbacks)))\n\t\tc.scope.RecordHistogramDuration(metrics.DomainCacheUpdateLatency, c.timeSource.Now().Sub(now))\n\t}\n\n\treturn nil\n}\n\nfunc (c *DefaultDomainCache) checkDomainExists(\n\tname string,\n\tid string,\n) error {\n\tctx, cancel := context.WithTimeout(context.Background(), domainCachePersistenceTimeout)\n\tdefer cancel()\n\n\t_, err := c.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: name, ID: id})\n\treturn err\n}\n\nfunc (c *DefaultDomainCache) updateNameToIDCache(\n\tcacheNameToID Cache,\n\tname string,\n\tid string,\n) {\n\n\tcacheNameToID.Put(name, id)\n}\n\nfunc (c *DefaultDomainCache) updateIDToDomainCache(\n\tcacheByID Cache,\n\tid string,\n\trecord *DomainCacheEntry,\n) (bool, *DomainCacheEntry, error) {\n\telem, err := cacheByID.PutIfNotExist(id, &DomainCacheEntry{})\n\tif err != nil {\n\t\treturn false, nil, err\n\t}\n\tentry := elem.(*DomainCacheEntry)\n\n\tentry.mu.Lock()\n\tdefer entry.mu.Unlock()\n\n\t// initialized will be true when the entry contains valid data\n\ttriggerCallback := entry.initialized && record.notificationVersion > entry.notificationVersion\n\n\tentry.info = record.info\n\tentry.config = record.config\n\tentry.replicationConfig = record.replicationConfig\n\tentry.configVersion = record.configVersion\n\tentry.failoverVersion = record.failoverVersion\n\tentry.isGlobalDomain = record.isGlobalDomain\n\tentry.failoverNotificationVersion = record.failoverNotificationVersion\n\tentry.previousFailoverVersion = record.previousFailoverVersion\n\tentry.failoverEndTime = record.failoverEndTime\n\tentry.notificationVersion = record.notificationVersion\n\tentry.initialized = record.initialized\n\tentry.activeClusters = record.activeClusters\n\treturn triggerCallback, entry.duplicate(), nil\n}\n\n// getDomain retrieves the information from the cache if it exists, otherwise retrieves the information from metadata\n// store and writes it to the cache with an expiry before returning back\nfunc (c *DefaultDomainCache) getDomain(\n\tname string,\n) (*DomainCacheEntry, error) {\n\n\tid, cacheHit := c.cacheNameToID.Load().(Cache).Get(name).(string)\n\tif cacheHit {\n\t\treturn c.getDomainByID(id, true)\n\t}\n\n\tif err := c.checkDomainExists(name, \"\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\tc.refreshLock.Lock()\n\tdefer c.refreshLock.Unlock()\n\tid, cacheHit = c.cacheNameToID.Load().(Cache).Get(name).(string)\n\tif cacheHit {\n\t\treturn c.getDomainByID(id, true)\n\t}\n\tif err := c.refreshDomainsLocked(context.Background()); err != nil {\n\t\treturn nil, err\n\t}\n\tid, cacheHit = c.cacheNameToID.Load().(Cache).Get(name).(string)\n\tif cacheHit {\n\t\treturn c.getDomainByID(id, true)\n\t}\n\t// impossible case\n\treturn nil, &types.InternalServiceError{Message: \"DefaultDomainCache encounter case where domain exists but cannot be loaded\"}\n}\n\n// getDomainByID retrieves the information from the cache if it exists, otherwise retrieves the information from metadata\n// store and writes it to the cache with an expiry before returning back\nfunc (c *DefaultDomainCache) getDomainByID(\n\tid string,\n\tdeepCopy bool,\n) (*DomainCacheEntry, error) {\n\n\tvar result *DomainCacheEntry\n\tdefer func() {\n\t\tif result != nil {\n\t\t\tc.logger.Debugf(\"GetDomainByID returning domain %s, failoverVersion: %d\", result.info.Name, result.failoverVersion)\n\t\t}\n\t}()\n\tentry, cacheHit := c.cacheByID.Load().(Cache).Get(id).(*DomainCacheEntry)\n\tif cacheHit {\n\t\tentry.mu.RLock()\n\t\tresult = entry\n\t\tif deepCopy {\n\t\t\tresult = entry.duplicate()\n\t\t}\n\t\tentry.mu.RUnlock()\n\t\treturn result, nil\n\t}\n\n\tif err := c.checkDomainExists(\"\", id); err != nil {\n\t\treturn nil, err\n\t}\n\n\tc.refreshLock.Lock()\n\tdefer c.refreshLock.Unlock()\n\tentry, cacheHit = c.cacheByID.Load().(Cache).Get(id).(*DomainCacheEntry)\n\tif cacheHit {\n\t\tentry.mu.RLock()\n\t\tresult = entry\n\t\tif deepCopy {\n\t\t\tresult = entry.duplicate()\n\t\t}\n\t\tentry.mu.RUnlock()\n\t\treturn result, nil\n\t}\n\tif err := c.refreshDomainsLocked(context.Background()); err != nil {\n\t\treturn nil, err\n\t}\n\tentry, cacheHit = c.cacheByID.Load().(Cache).Get(id).(*DomainCacheEntry)\n\tif cacheHit {\n\t\tentry.mu.RLock()\n\t\tresult = entry\n\t\tif deepCopy {\n\t\t\tresult = entry.duplicate()\n\t\t}\n\t\tentry.mu.RUnlock()\n\t\treturn result, nil\n\t}\n\t// impossible case\n\treturn nil, &types.InternalServiceError{Message: \"DefaultDomainCache encounter case where domain exists but cannot be loaded\"}\n}\n\nfunc (c *DefaultDomainCache) triggerDomainChangePrepareCallbackLocked() {\n\tsw := c.scope.StartTimer(metrics.DomainCachePrepareCallbacksLatency)\n\tdefer sw.Stop()\n\n\tfor _, prepareCallback := range c.prepareCallbacks {\n\t\tprepareCallback()\n\t}\n}\n\nfunc (c *DefaultDomainCache) triggerDomainChangeCallbackLocked(nextDomains []*DomainCacheEntry) {\n\tsw := c.scope.StartTimer(metrics.DomainCacheCallbacksLatency)\n\tdefer sw.Stop()\n\n\tc.logger.Debug(\"Domain change callbacks are going to triggered\", tag.Number(int64(len(nextDomains))))\n\tfor _, callback := range c.callbacks {\n\t\tcallback(nextDomains)\n\t}\n\tc.logger.Debug(\"Domain change callbacks are completed\", tag.Number(int64(len(nextDomains))))\n}\n\nfunc (c *DefaultDomainCache) buildEntryFromRecord(\n\trecord *persistence.GetDomainResponse,\n) *DomainCacheEntry {\n\n\t// this is a shallow copy, but since the record is generated by persistence\n\t// and only accessible here, it would be fine\n\treturn &DomainCacheEntry{\n\t\tinfo:                        record.Info,\n\t\tconfig:                      record.Config,\n\t\treplicationConfig:           record.ReplicationConfig,\n\t\tconfigVersion:               record.ConfigVersion,\n\t\tfailoverVersion:             record.FailoverVersion,\n\t\tisGlobalDomain:              record.IsGlobalDomain,\n\t\tfailoverNotificationVersion: record.FailoverNotificationVersion,\n\t\tpreviousFailoverVersion:     record.PreviousFailoverVersion,\n\t\tfailoverEndTime:             record.FailoverEndTime,\n\t\tnotificationVersion:         record.NotificationVersion,\n\t\tinitialized:                 true,\n\t\tactiveClusters:              getActiveClusters(record.ReplicationConfig),\n\t}\n}\n\nfunc copyResetBinary(bins types.BadBinaries) types.BadBinaries {\n\tnewbins := make(map[string]*types.BadBinaryInfo, len(bins.Binaries))\n\tfor k, v := range bins.Binaries {\n\t\tnewbins[k] = v\n\t}\n\treturn types.BadBinaries{\n\t\tBinaries: newbins,\n\t}\n}\n\nfunc (entry *DomainCacheEntry) duplicate() *DomainCacheEntry {\n\t// this is a deep copy\n\tresult := &DomainCacheEntry{}\n\tresult.info = &persistence.DomainInfo{\n\t\tID:          entry.info.ID,\n\t\tName:        entry.info.Name,\n\t\tStatus:      entry.info.Status,\n\t\tDescription: entry.info.Description,\n\t\tOwnerEmail:  entry.info.OwnerEmail,\n\t}\n\tresult.info.Data = map[string]string{}\n\tfor k, v := range entry.info.Data {\n\t\tresult.info.Data[k] = v\n\t}\n\tresult.config = &persistence.DomainConfig{\n\t\tRetention:                entry.config.Retention,\n\t\tEmitMetric:               entry.config.EmitMetric,\n\t\tHistoryArchivalStatus:    entry.config.HistoryArchivalStatus,\n\t\tHistoryArchivalURI:       entry.config.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus: entry.config.VisibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    entry.config.VisibilityArchivalURI,\n\t\tBadBinaries:              copyResetBinary(entry.config.BadBinaries),\n\t\tAsyncWorkflowConfig:      entry.config.AsyncWorkflowConfig.DeepCopy(),\n\t\tIsolationGroups:          entry.config.IsolationGroups.DeepCopy(),\n\t}\n\tresult.replicationConfig = &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: entry.replicationConfig.ActiveClusterName,\n\t}\n\tfor _, clusterCfg := range entry.replicationConfig.Clusters {\n\t\tc := *clusterCfg\n\t\tresult.replicationConfig.Clusters = append(result.replicationConfig.Clusters, &c)\n\t}\n\tresult.replicationConfig.ActiveClusters = entry.replicationConfig.ActiveClusters.DeepCopy()\n\tresult.configVersion = entry.configVersion\n\tresult.failoverVersion = entry.failoverVersion\n\tresult.isGlobalDomain = entry.isGlobalDomain\n\tresult.failoverNotificationVersion = entry.failoverNotificationVersion\n\tresult.previousFailoverVersion = entry.previousFailoverVersion\n\tresult.failoverEndTime = entry.failoverEndTime\n\tresult.notificationVersion = entry.notificationVersion\n\tresult.initialized = entry.initialized\n\tresult.activeClusters = entry.activeClusters\n\treturn result\n}\n\n// GetInfo return the domain info\nfunc (entry *DomainCacheEntry) GetInfo() *persistence.DomainInfo {\n\treturn entry.info\n}\n\n// GetConfig return the domain config\nfunc (entry *DomainCacheEntry) GetConfig() *persistence.DomainConfig {\n\treturn entry.config\n}\n\n// GetReplicationConfig return the domain replication config\nfunc (entry *DomainCacheEntry) GetReplicationConfig() *persistence.DomainReplicationConfig {\n\treturn entry.replicationConfig\n}\n\n// GetConfigVersion return the domain config version\nfunc (entry *DomainCacheEntry) GetConfigVersion() int64 {\n\treturn entry.configVersion\n}\n\n// GetFailoverVersion return the domain failover version\nfunc (entry *DomainCacheEntry) GetFailoverVersion() int64 {\n\treturn entry.failoverVersion\n}\n\n// IsGlobalDomain return whether the domain is a global domain\nfunc (entry *DomainCacheEntry) IsGlobalDomain() bool {\n\treturn entry.isGlobalDomain\n}\n\n// GetFailoverNotificationVersion return the global notification version of when failover happened\nfunc (entry *DomainCacheEntry) GetFailoverNotificationVersion() int64 {\n\treturn entry.failoverNotificationVersion\n}\n\n// GetNotificationVersion return the global notification version of when domain changed\nfunc (entry *DomainCacheEntry) GetNotificationVersion() int64 {\n\treturn entry.notificationVersion\n}\n\n// GetPreviousFailoverVersion return the last domain failover version\nfunc (entry *DomainCacheEntry) GetPreviousFailoverVersion() int64 {\n\treturn entry.previousFailoverVersion\n}\n\n// GetFailoverEndTime return the failover end time\nfunc (entry *DomainCacheEntry) GetFailoverEndTime() *int64 {\n\treturn entry.failoverEndTime\n}\n\n// GetActiveClusterInfoByClusterAttribute return the active cluster info for a given cluster attribute\n// if clusterAttribute is nil, return the domain-level active cluster info and true\n// if the clusterAttribute exists, return the active cluster info of the clusterAttribute and true\n// if the clusterAttribute is not found, return false\nfunc (entry *DomainCacheEntry) GetActiveClusterInfoByClusterAttribute(clusterAttribute *types.ClusterAttribute) (*types.ActiveClusterInfo, bool) {\n\tif clusterAttribute == nil {\n\t\treturn &types.ActiveClusterInfo{\n\t\t\tActiveClusterName: entry.GetReplicationConfig().ActiveClusterName,\n\t\t\tFailoverVersion:   entry.GetFailoverVersion(),\n\t\t}, true\n\t}\n\tif entry.replicationConfig.ActiveClusters == nil {\n\t\treturn nil, false\n\t}\n\tif entry.replicationConfig.ActiveClusters.AttributeScopes == nil {\n\t\treturn nil, false\n\t}\n\tscope, ok := entry.replicationConfig.ActiveClusters.AttributeScopes[clusterAttribute.Scope]\n\tif !ok {\n\t\treturn nil, false\n\t}\n\tinfo, ok := scope.ClusterAttributes[clusterAttribute.Name]\n\tif !ok {\n\t\treturn nil, false\n\t}\n\treturn &types.ActiveClusterInfo{\n\t\tActiveClusterName: info.ActiveClusterName,\n\t\tFailoverVersion:   info.FailoverVersion,\n\t}, true\n}\n\n// NewDomainNotActiveError return a domain not active error\n// currentCluster is the current cluster\n// activeCluster is the active cluster which is either domain's active cluster or it's inferred from workflow task version\nfunc (entry *DomainCacheEntry) NewDomainNotActiveError(currentCluster, activeCluster string) *types.DomainNotActiveError {\n\tif entry.GetReplicationConfig().IsActiveActive() {\n\t\treturn &types.DomainNotActiveError{\n\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\"Domain: %s is active in cluster(s): %v, while current cluster %s is a standby cluster. Operation active cluster: %s\",\n\t\t\t\tentry.GetInfo().Name,\n\t\t\t\tentry.activeClusters,\n\t\t\t\tcurrentCluster,\n\t\t\t\tactiveCluster,\n\t\t\t),\n\t\t\tDomainName:     entry.GetInfo().Name,\n\t\t\tCurrentCluster: currentCluster,\n\t\t\tActiveCluster:  activeCluster,\n\t\t\tActiveClusters: entry.activeClusters,\n\t\t}\n\t}\n\n\treturn &types.DomainNotActiveError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Domain: %s is active in cluster: %s, while current cluster %s is a standby cluster.\",\n\t\t\tentry.GetInfo().Name,\n\t\t\tactiveCluster,\n\t\t\tcurrentCluster,\n\t\t),\n\t\tDomainName:     entry.GetInfo().Name,\n\t\tCurrentCluster: currentCluster,\n\t\tActiveCluster:  activeCluster,\n\t}\n}\n\n// IsActive return whether the domain is active in the current cluster,\n// - for local domain, it is always active\n// - for global domain, it is active if it is not pending active and the domain's active cluster is the current cluster or if the domain is active-active and the active cluster of one of the cluster attributes is the current cluster\n// TODO(active-active): for active-active domains, we should review this logic because now workflows can be active in different clusters based on the cluster attribute.\n// We should also revisit its usage in history service.\nfunc (entry *DomainCacheEntry) IsActiveIn(currentCluster string) bool {\n\tif !entry.IsGlobalDomain() {\n\t\t// domain is not a global domain, meaning domain is always \"active\" within each cluster\n\t\treturn true\n\t}\n\n\tif entry.IsDomainPendingActive() {\n\t\treturn false\n\t}\n\n\tactiveCluster := entry.GetReplicationConfig().ActiveClusterName\n\tif currentCluster == activeCluster {\n\t\treturn true\n\t}\n\n\tif activeClusters := entry.GetReplicationConfig().ActiveClusters; activeClusters != nil {\n\t\tfor _, scope := range activeClusters.AttributeScopes {\n\t\t\tfor _, cl := range scope.ClusterAttributes {\n\t\t\t\tif cl.ActiveClusterName == currentCluster {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn false\n}\n\n// IsDomainPendingActive returns whether the domain is in pending active state\nfunc (entry *DomainCacheEntry) IsDomainPendingActive() bool {\n\tif !entry.isGlobalDomain {\n\t\t// domain is not a global domain, meaning domain can never be in pending active state\n\t\treturn false\n\t}\n\treturn entry.failoverEndTime != nil\n}\n\n// GetReplicationPolicy return the derived workflow replication policy\nfunc (entry *DomainCacheEntry) GetReplicationPolicy() ReplicationPolicy {\n\t// frontend guarantee that the clusters always contains the active domain, so if the # of clusters is 1\n\t// then we do not need to send out any events for replication\n\tif entry.isGlobalDomain && len(entry.replicationConfig.Clusters) > 1 {\n\t\treturn ReplicationPolicyMultiCluster\n\t}\n\treturn ReplicationPolicyOneCluster\n}\n\n// HasReplicationCluster returns true if the domain has replication in the cluster\nfunc (entry *DomainCacheEntry) HasReplicationCluster(clusterName string) bool {\n\tfor _, cluster := range entry.GetReplicationConfig().Clusters {\n\t\tif cluster.ClusterName == clusterName {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// Len return length\nfunc (t DomainCacheEntries) Len() int {\n\treturn len(t)\n}\n\n// Swap implements sort.Interface.\nfunc (t DomainCacheEntries) Swap(i, j int) {\n\tt[i], t[j] = t[j], t[i]\n}\n\n// Less implements sort.Interface\nfunc (t DomainCacheEntries) Less(i, j int) bool {\n\treturn t[i].notificationVersion < t[j].notificationVersion\n}\n\n// CreateDomainCacheEntry create a cache entry with domainName\nfunc CreateDomainCacheEntry(\n\tdomainName string,\n) *DomainCacheEntry {\n\n\treturn &DomainCacheEntry{info: &persistence.DomainInfo{Name: domainName}}\n}\n\n// SampleRetentionKey is key to specify sample retention\nvar SampleRetentionKey = \"sample_retention_days\"\n\n// SampleRateKey is key to specify sample rate\nvar SampleRateKey = \"sample_retention_rate\"\n\n// GetRetentionDays returns retention in days for given workflow\nfunc (entry *DomainCacheEntry) GetRetentionDays(\n\tworkflowID string,\n) int32 {\n\n\tif entry.IsSampledForLongerRetention(workflowID) {\n\t\tif sampledRetentionValue, ok := entry.info.Data[SampleRetentionKey]; ok {\n\t\t\tsampledRetentionDays, err := strconv.Atoi(sampledRetentionValue)\n\t\t\tif err != nil || sampledRetentionDays < int(entry.config.Retention) {\n\t\t\t\treturn entry.config.Retention\n\t\t\t}\n\t\t\treturn int32(sampledRetentionDays)\n\t\t}\n\t}\n\treturn entry.config.Retention\n}\n\n// IsSampledForLongerRetentionEnabled return whether sample for longer retention is enabled or not\nfunc (entry *DomainCacheEntry) IsSampledForLongerRetentionEnabled(\n\tworkflowID string,\n) bool {\n\n\t_, ok := entry.info.Data[SampleRateKey]\n\treturn ok\n}\n\n// IsSampledForLongerRetention return should given workflow been sampled or not\nfunc (entry *DomainCacheEntry) IsSampledForLongerRetention(\n\tworkflowID string,\n) bool {\n\n\tif sampledRateValue, ok := entry.info.Data[SampleRateKey]; ok {\n\t\tsampledRate, err := strconv.ParseFloat(sampledRateValue, 64)\n\t\tif err != nil {\n\t\t\treturn false\n\t\t}\n\n\t\th := fnv.New32a()\n\t\t_, err = h.Write([]byte(workflowID))\n\t\tif err != nil {\n\t\t\treturn false\n\t\t}\n\t\thash := h.Sum32()\n\n\t\tr := float64(hash%1000) / float64(1000) // use 1000 so we support one decimal rate like 1.5%.\n\t\tif r < sampledRate {                    // sampled\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// IsDeprecatedOrDeleted This function checks the domain status to see if the domain has been deprecated or deleted.\nfunc (entry *DomainCacheEntry) IsDeprecatedOrDeleted() bool {\n\tif entry.info.Status == persistence.DomainStatusDeprecated || entry.info.Status == persistence.DomainStatusDeleted {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc getActiveClusters(replicationConfig *persistence.DomainReplicationConfig) []string {\n\tif !replicationConfig.IsActiveActive() {\n\t\treturn nil\n\t}\n\t// TODO(active-active): Replace with `GetAllClusters` once we remove regions\n\tactiveClusters := make([]string, 0, len(replicationConfig.ActiveClusters.AttributeScopes))\n\tfor _, scope := range replicationConfig.ActiveClusters.AttributeScopes {\n\t\tfor _, cl := range scope.ClusterAttributes {\n\t\t\tactiveClusters = append(activeClusters, cl.ActiveClusterName)\n\t\t}\n\t}\n\tsort.Strings(activeClusters)\n\treturn activeClusters\n}\n"
  },
  {
    "path": "common/cache/domainCacheNoOp.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\npackage cache\n\ntype noOpDomainCache struct{}\n\nfunc (c *noOpDomainCache) GetCacheSize() (sizeOfCacheByName int64, sizeOfCacheByID int64) {\n\treturn 0, 0\n}\n\nfunc (c *noOpDomainCache) GetAllDomain() map[string]*DomainCacheEntry {\n\treturn map[string]*DomainCacheEntry{}\n}\n\nfunc (c *noOpDomainCache) RegisterDomainChangeCallback(\n\tid string,\n\tcatchUpFn CatchUpFn,\n\tprepareCallback PrepareCallbackFn,\n\tcallback CallbackFn,\n) {\n}\n\nfunc (c *noOpDomainCache) UnregisterDomainChangeCallback(\n\tid string,\n) {\n}\n\nfunc (c *noOpDomainCache) GetDomain(\n\tname string,\n) (*DomainCacheEntry, error) {\n\treturn &DomainCacheEntry{}, nil\n}\n\nfunc (c *noOpDomainCache) GetDomainByID(\n\tid string,\n) (*DomainCacheEntry, error) {\n\treturn &DomainCacheEntry{}, nil\n}\n\nfunc (c *noOpDomainCache) GetDomainID(\n\tname string,\n) (string, error) {\n\treturn \"\", nil\n}\n\nfunc (c *noOpDomainCache) GetDomainName(\n\tid string,\n) (string, error) {\n\treturn \"\", nil\n}\n\nfunc (c *noOpDomainCache) getDomain(\n\tname string,\n) (*DomainCacheEntry, error) {\n\treturn &DomainCacheEntry{}, nil\n}\n\nfunc (c *noOpDomainCache) getDomainByID(\n\tid string,\n\tdeepCopy bool,\n) (*DomainCacheEntry, error) {\n\treturn &DomainCacheEntry{}, nil\n}\n\nfunc NewNoOpDomainCache() DomainCache {\n\treturn &noOpDomainCache{}\n}\n\nfunc (c *noOpDomainCache) Start() {}\n\nfunc (c *noOpDomainCache) Stop() {}\n"
  },
  {
    "path": "common/cache/domainCache_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: domainCache.go\n//\n// Generated by this command:\n//\n//\tmockgen -package cache -source domainCache.go -destination domainCache_mock.go -self_package github.com/uber/cadence/common/cache\n//\n\n// Package cache is a generated GoMock package.\npackage cache\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockDomainCache is a mock of DomainCache interface.\ntype MockDomainCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainCacheMockRecorder is the mock recorder for MockDomainCache.\ntype MockDomainCacheMockRecorder struct {\n\tmock *MockDomainCache\n}\n\n// NewMockDomainCache creates a new mock instance.\nfunc NewMockDomainCache(ctrl *gomock.Controller) *MockDomainCache {\n\tmock := &MockDomainCache{ctrl: ctrl}\n\tmock.recorder = &MockDomainCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainCache) EXPECT() *MockDomainCacheMockRecorder {\n\treturn m.recorder\n}\n\n// GetAllDomain mocks base method.\nfunc (m *MockDomainCache) GetAllDomain() map[string]*DomainCacheEntry {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllDomain\")\n\tret0, _ := ret[0].(map[string]*DomainCacheEntry)\n\treturn ret0\n}\n\n// GetAllDomain indicates an expected call of GetAllDomain.\nfunc (mr *MockDomainCacheMockRecorder) GetAllDomain() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllDomain\", reflect.TypeOf((*MockDomainCache)(nil).GetAllDomain))\n}\n\n// GetCacheSize mocks base method.\nfunc (m *MockDomainCache) GetCacheSize() (int64, int64) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCacheSize\")\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(int64)\n\treturn ret0, ret1\n}\n\n// GetCacheSize indicates an expected call of GetCacheSize.\nfunc (mr *MockDomainCacheMockRecorder) GetCacheSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCacheSize\", reflect.TypeOf((*MockDomainCache)(nil).GetCacheSize))\n}\n\n// GetDomain mocks base method.\nfunc (m *MockDomainCache) GetDomain(name string) (*DomainCacheEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomain\", name)\n\tret0, _ := ret[0].(*DomainCacheEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomain indicates an expected call of GetDomain.\nfunc (mr *MockDomainCacheMockRecorder) GetDomain(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomain\", reflect.TypeOf((*MockDomainCache)(nil).GetDomain), name)\n}\n\n// GetDomainByID mocks base method.\nfunc (m *MockDomainCache) GetDomainByID(id string) (*DomainCacheEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainByID\", id)\n\tret0, _ := ret[0].(*DomainCacheEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainByID indicates an expected call of GetDomainByID.\nfunc (mr *MockDomainCacheMockRecorder) GetDomainByID(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainByID\", reflect.TypeOf((*MockDomainCache)(nil).GetDomainByID), id)\n}\n\n// GetDomainID mocks base method.\nfunc (m *MockDomainCache) GetDomainID(name string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainID\", name)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainID indicates an expected call of GetDomainID.\nfunc (mr *MockDomainCacheMockRecorder) GetDomainID(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainID\", reflect.TypeOf((*MockDomainCache)(nil).GetDomainID), name)\n}\n\n// GetDomainName mocks base method.\nfunc (m *MockDomainCache) GetDomainName(id string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainName\", id)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainName indicates an expected call of GetDomainName.\nfunc (mr *MockDomainCacheMockRecorder) GetDomainName(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainName\", reflect.TypeOf((*MockDomainCache)(nil).GetDomainName), id)\n}\n\n// RegisterDomainChangeCallback mocks base method.\nfunc (m *MockDomainCache) RegisterDomainChangeCallback(id string, catchUpFn CatchUpFn, prepareCallback PrepareCallbackFn, callback CallbackFn) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RegisterDomainChangeCallback\", id, catchUpFn, prepareCallback, callback)\n}\n\n// RegisterDomainChangeCallback indicates an expected call of RegisterDomainChangeCallback.\nfunc (mr *MockDomainCacheMockRecorder) RegisterDomainChangeCallback(id, catchUpFn, prepareCallback, callback any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterDomainChangeCallback\", reflect.TypeOf((*MockDomainCache)(nil).RegisterDomainChangeCallback), id, catchUpFn, prepareCallback, callback)\n}\n\n// Start mocks base method.\nfunc (m *MockDomainCache) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockDomainCacheMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockDomainCache)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockDomainCache) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockDomainCacheMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockDomainCache)(nil).Stop))\n}\n\n// UnregisterDomainChangeCallback mocks base method.\nfunc (m *MockDomainCache) UnregisterDomainChangeCallback(id string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UnregisterDomainChangeCallback\", id)\n}\n\n// UnregisterDomainChangeCallback indicates an expected call of UnregisterDomainChangeCallback.\nfunc (mr *MockDomainCacheMockRecorder) UnregisterDomainChangeCallback(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UnregisterDomainChangeCallback\", reflect.TypeOf((*MockDomainCache)(nil).UnregisterDomainChangeCallback), id)\n}\n"
  },
  {
    "path": "common/cache/domainCache_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cache\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdomainCacheSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tmetadataMgr *mocks.MetadataManager\n\n\t\tdomainCache *DefaultDomainCache\n\t\tlogger      log.Logger\n\t}\n)\n\nfunc TestDomainCacheSuite(t *testing.T) {\n\ts := new(domainCacheSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainCacheSuite) SetupSuite() {\n}\n\nfunc (s *domainCacheSuite) TearDownSuite() {\n\n}\n\nfunc (s *domainCacheSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.logger = testlogger.New(s.Suite.T())\n\n\ts.metadataMgr = &mocks.MetadataManager{}\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.domainCache = NewDomainCache(s.metadataMgr, cluster.GetTestClusterMetadata(true), metricsClient, s.logger)\n\n\ts.domainCache.timeSource = clock.NewMockedTimeSource()\n}\n\nfunc (s *domainCacheSuite) TearDownTest() {\n\ts.domainCache.Stop()\n\ts.metadataMgr.AssertExpectations(s.T())\n}\n\nfunc (s *domainCacheSuite) TestListDomain() {\n\tdomainNotificationVersion := int64(0)\n\tdomainRecord1 := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"some random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             true,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tentry1 := s.buildEntryFromRecord(domainRecord1)\n\tdomainNotificationVersion++\n\n\tdomainRecord2 := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"another random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 2,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tentry2 := s.buildEntryFromRecord(domainRecord2)\n\tdomainNotificationVersion++\n\n\tdomainRecord3 := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"yet another random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 3,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\t// there is no domainNotificationVersion++ here\n\t// this is to test that if new domain change event happen during the pagination,\n\t// new change will not be loaded to domain cache\n\n\tpageToken := []byte(\"some random page token\")\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil)\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord1},\n\t\tNextPageToken: pageToken,\n\t}, nil).Once()\n\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: pageToken,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord2, domainRecord3},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\t// load domains\n\ts.domainCache.Start()\n\tdefer s.domainCache.Stop()\n\n\tentryByName1, err := s.domainCache.GetDomain(domainRecord1.Info.Name)\n\ts.Nil(err)\n\ts.Equal(entry1, entryByName1)\n\tentryByID1, err := s.domainCache.GetDomainByID(domainRecord1.Info.ID)\n\ts.Nil(err)\n\ts.Equal(entry1, entryByID1)\n\n\tentryByName2, err := s.domainCache.GetDomain(domainRecord2.Info.Name)\n\ts.Nil(err)\n\ts.Equal(entry2, entryByName2)\n\tentryByID2, err := s.domainCache.GetDomainByID(domainRecord2.Info.ID)\n\ts.Nil(err)\n\ts.Equal(entry2, entryByID2)\n\n\tallDomains := s.domainCache.GetAllDomain()\n\ts.Equal(map[string]*DomainCacheEntry{\n\t\tentry1.GetInfo().ID: entry1,\n\t\tentry2.GetInfo().ID: entry2,\n\t}, allDomains)\n}\n\nfunc (s *domainCacheSuite) TestGetDomain_NonLoaded_GetByName() {\n\tdomainNotificationVersion := int64(999999) // make this notification version really large for test\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil)\n\tdomainRecord := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"some random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\"abc\": {\n\t\t\t\t\t\tReason:          \"test reason\",\n\t\t\t\t\t\tOperator:        \"test operator\",\n\t\t\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t}\n\tentry := s.buildEntryFromRecord(domainRecord)\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: entry.info.Name}).Return(domainRecord, nil).Once()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\tentryByName, err := s.domainCache.GetDomain(domainRecord.Info.Name)\n\ts.Nil(err)\n\ts.Equal(entry, entryByName)\n\tentryByName, err = s.domainCache.GetDomain(domainRecord.Info.Name)\n\ts.Nil(err)\n\ts.Equal(entry, entryByName)\n}\n\nfunc (s *domainCacheSuite) TestGetDomain_NonLoaded_GetByID() {\n\tdomainNotificationVersion := int64(999999) // make this notification version really large for test\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil)\n\tdomainRecord := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"some random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t}\n\tentry := s.buildEntryFromRecord(domainRecord)\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{ID: entry.info.ID}).Return(domainRecord, nil).Once()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\tentryByID, err := s.domainCache.GetDomainByID(domainRecord.Info.ID)\n\ts.Nil(err)\n\ts.Equal(entry, entryByID)\n\tentryByID, err = s.domainCache.GetDomainByID(domainRecord.Info.ID)\n\ts.Nil(err)\n\ts.Equal(entry, entryByID)\n}\n\nfunc Test_IsActiveIn(t *testing.T) {\n\ttests := []struct {\n\t\tmsg              string\n\t\tisGlobalDomain   bool\n\t\tcurrentCluster   string\n\t\tactiveCluster    string\n\t\tactiveClusters   *types.ActiveClusters\n\t\tfailoverDeadline *int64\n\t\texpectIsActive   bool\n\t}{\n\t\t{\n\t\t\tmsg:            \"local domain\",\n\t\t\tisGlobalDomain: false,\n\t\t\texpectIsActive: true,\n\t\t},\n\t\t{\n\t\t\tmsg:              \"global pending active domain\",\n\t\t\tisGlobalDomain:   true,\n\t\t\tfailoverDeadline: common.Int64Ptr(time.Now().Unix()),\n\t\t\texpectIsActive:   false,\n\t\t},\n\t\t{\n\t\t\tmsg:            \"global domain on active cluster\",\n\t\t\tisGlobalDomain: true,\n\t\t\tcurrentCluster: \"A\",\n\t\t\tactiveCluster:  \"A\",\n\t\t\texpectIsActive: true,\n\t\t},\n\t\t{\n\t\t\tmsg:            \"global domain on passive cluster\",\n\t\t\tisGlobalDomain: true,\n\t\t\tcurrentCluster: \"A\",\n\t\t\tactiveCluster:  \"B\",\n\t\t\texpectIsActive: false,\n\t\t},\n\t\t{\n\t\t\tmsg:            \"active-active domain on active cluster\",\n\t\t\tisGlobalDomain: true,\n\t\t\tcurrentCluster: \"A\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {ActiveClusterName: \"A\"},\n\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"B\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectIsActive: true,\n\t\t},\n\t\t{\n\t\t\tmsg:            \"active-active domain on domain level active cluster\",\n\t\t\tisGlobalDomain: true,\n\t\t\tcurrentCluster: \"A\",\n\t\t\tactiveCluster:  \"A\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {ActiveClusterName: \"B\"},\n\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"B\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectIsActive: true,\n\t\t},\n\t\t{\n\t\t\tmsg:            \"active-active domain on passive cluster\",\n\t\t\tisGlobalDomain: true,\n\t\t\tcurrentCluster: \"C\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {ActiveClusterName: \"A\"},\n\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"B\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectIsActive: false,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.msg, func(t *testing.T) {\n\t\t\tdomain := NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{Name: \"test-domain\"},\n\t\t\t\tnil,\n\t\t\t\ttt.isGlobalDomain,\n\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: tt.activeCluster,\n\t\t\t\t\tActiveClusters:    tt.activeClusters,\n\t\t\t\t},\n\t\t\t\t0,\n\t\t\t\ttt.failoverDeadline,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t)\n\n\t\t\tisActive := domain.IsActiveIn(tt.currentCluster)\n\t\t\tassert.Equal(t, tt.expectIsActive, isActive)\n\t\t})\n\t}\n}\n\nfunc (s *domainCacheSuite) TestRegisterCallback_CatchUp() {\n\tprepareCallbackInvoked := false\n\tcallBackInvoked := false\n\tvar entriesNotification []*DomainCacheEntry\n\n\ts.domainCache.RegisterDomainChangeCallback(\n\t\t\"0\",\n\t\tfunc(_ DomainCache, prepareCallback PrepareCallbackFn, callback CallbackFn) {\n\t\t\tprepareCallback()\n\t\t\tcallback(entriesNotification)\n\t\t},\n\t\tfunc() {\n\t\t\tprepareCallbackInvoked = true\n\t\t},\n\t\tfunc(nextDomains []*DomainCacheEntry) {\n\t\t\tcallBackInvoked = true\n\t\t},\n\t)\n\n\ts.True(prepareCallbackInvoked)\n\ts.True(callBackInvoked)\n}\n\nfunc (s *domainCacheSuite) TestUpdateCache_TriggerCallBack() {\n\tdomainNotificationVersion := int64(0)\n\tdomainRecord1Old := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"some random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:               10,\n\t\tFailoverVersion:             11,\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tdomainNotificationVersion++\n\n\tdomainRecord2Old := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: uuid.New(), Name: \"another random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 2,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:               20,\n\t\tFailoverVersion:             21,\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tdomainNotificationVersion++\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil).Once()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord1Old, domainRecord2Old},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\t// load domains\n\ts.Nil(s.domainCache.refreshDomains())\n\n\tdomainRecord2New := &persistence.GetDomainResponse{\n\t\tInfo:   &*domainRecord2Old.Info,\n\t\tConfig: &*domainRecord2Old.Config,\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName, // only this changed\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:               domainRecord2Old.ConfigVersion,\n\t\tFailoverVersion:             domainRecord2Old.FailoverVersion + 1,\n\t\tFailoverNotificationVersion: domainNotificationVersion,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tentry2New := s.buildEntryFromRecord(domainRecord2New)\n\tdomainNotificationVersion++\n\n\tdomainRecord1New := &persistence.GetDomainResponse{ // only the description changed\n\t\tInfo:   &persistence.DomainInfo{ID: domainRecord1Old.Info.ID, Name: domainRecord1Old.Info.Name, Description: \"updated description\", Data: make(map[string]string)},\n\t\tConfig: &*domainRecord2Old.Config,\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:               domainRecord1Old.ConfigVersion + 1,\n\t\tFailoverVersion:             domainRecord1Old.FailoverVersion,\n\t\tFailoverNotificationVersion: domainRecord1Old.FailoverNotificationVersion,\n\t\tNotificationVersion:         domainNotificationVersion,\n\t}\n\tentry1New := s.buildEntryFromRecord(domainRecord1New)\n\tdomainNotificationVersion++\n\n\tprepareCallbackInvoked := false\n\tvar entriesNew []*DomainCacheEntry\n\ts.domainCache.RegisterDomainChangeCallback(\n\t\t\"0\",\n\t\tfunc(domainCache DomainCache, prepareCallback PrepareCallbackFn, callback CallbackFn) {},\n\t\tfunc() {\n\t\t\tprepareCallbackInvoked = true\n\t\t},\n\t\tfunc(nextDomains []*DomainCacheEntry) {\n\t\t\tentriesNew = nextDomains\n\t\t},\n\t)\n\ts.False(prepareCallbackInvoked)\n\ts.Empty(entriesNew)\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil).Once()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecord1New, domainRecord2New},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\ts.domainCache.timeSource.(clock.MockedTimeSource).Advance(domainCacheMinRefreshInterval)\n\ts.Nil(s.domainCache.refreshDomains())\n\n\t// the order matters here: the record 2 got updated first, thus with a lower notification version\n\t// the record 1 got updated later, thus a higher notification version.\n\t// making sure notifying from lower to higher version helps the shard to keep track the\n\t// domain change events\n\ts.True(prepareCallbackInvoked)\n\ts.Equal([]*DomainCacheEntry{entry2New, entry1New}, entriesNew)\n}\n\nfunc (s *domainCacheSuite) TestGetTriggerListAndUpdateCache_ConcurrentAccess() {\n\tdomainNotificationVersion := int64(999999) // make this notification version really large for test\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: domainNotificationVersion}, nil)\n\tid := uuid.New()\n\tdomainRecordOld := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: id, Name: \"some random domain name\", Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t}},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:   0,\n\t\tFailoverVersion: 0,\n\t}\n\tentryOld := s.buildEntryFromRecord(domainRecordOld)\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{ID: id}).Return(domainRecordOld, nil).Maybe()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, &persistence.ListDomainsRequest{\n\t\tPageSize:      domainCacheRefreshPageSize,\n\t\tNextPageToken: nil,\n\t}).Return(&persistence.ListDomainsResponse{\n\t\tDomains:       []*persistence.GetDomainResponse{domainRecordOld},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\tcoroutineCountGet := 1000\n\twaitGroup := &sync.WaitGroup{}\n\tstartChan := make(chan struct{})\n\ttestGetFn := func() {\n\t\t<-startChan\n\t\tentryNew, err := s.domainCache.GetDomainByID(id)\n\t\ts.Nil(err)\n\t\t// make the config version the same so we can easily compare those\n\t\tentryNew.configVersion = 0\n\t\tentryNew.failoverVersion = 0\n\t\ts.Equal(entryOld, entryNew)\n\t\twaitGroup.Done()\n\t}\n\n\tfor i := 0; i < coroutineCountGet; i++ {\n\t\twaitGroup.Add(1)\n\t\tgo testGetFn()\n\t}\n\tclose(startChan)\n\twaitGroup.Wait()\n}\n\nfunc (s *domainCacheSuite) TestGetCacheSize() {\n\ttestCache := newDomainCache()\n\ttestCache.Put(\"testDomainID\", &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{ID: \"testDomainID\", Name: \"testDomain\"},\n\t})\n\n\ts.domainCache.cacheByID.Store(testCache)\n\n\ts.domainCache.cacheNameToID.Store(testCache)\n\n\tbyName, byID := s.domainCache.GetCacheSize()\n\n\ts.Equal(int64(1), byName)\n\ts.Equal(int64(1), byID)\n}\n\nfunc (s *domainCacheSuite) TestStart_Stop() {\n\tmockTimeSource := clock.NewMockedTimeSource()\n\ts.domainCache.timeSource = mockTimeSource\n\n\ts.domainCache.lastRefreshTime = mockTimeSource.Now()\n\n\tdomainID := uuid.New()\n\tdomainName := \"some random domain name\"\n\n\ts.Equal(domainCacheInitialized, s.domainCache.status)\n\n\ts.Equal(0, len(s.domainCache.GetAllDomain()))\n\n\ts.domainCache.Start()\n\n\t// testing noop\n\ts.domainCache.Start()\n\n\tmockTimeSource.BlockUntil(1)\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: 4}, nil).Once()\n\n\tdomainResponse := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{ID: domainID, Name: domainName, Data: make(map[string]string)},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t},\n\t\tFailoverVersion:     124,\n\t\tNotificationVersion: 3,\n\t}\n\n\tlistDomainsResponse := &persistence.ListDomainsResponse{\n\t\tDomains: []*persistence.GetDomainResponse{domainResponse},\n\t}\n\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, mock.Anything).Return(listDomainsResponse, nil).Once()\n\n\tmockTimeSource.Advance(DomainCacheRefreshInterval)\n\n\ts.Equal(domainCacheStarted, s.domainCache.status)\n\n\t// need to wait for the go routine to make progress and update the cache\n\ttime.Sleep(200 * time.Millisecond)\n\n\tallDomains := s.domainCache.GetAllDomain()\n\ts.Equal(1, len(allDomains))\n\ts.Equal(domainID, allDomains[domainID].GetInfo().ID)\n\ts.Equal(int64(124), allDomains[domainID].GetFailoverVersion())\n\n\ts.domainCache.Stop()\n\ts.Equal(domainCacheStopped, s.domainCache.status)\n}\n\nfunc (s *domainCacheSuite) TestStart_Error() {\n\tmockLogger := log.NewMockLogger(gomock.NewController(s.T()))\n\ts.domainCache.logger = mockLogger\n\n\ts.Equal(domainCacheInitialized, s.domainCache.status)\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(nil, assert.AnError).Once()\n\tmockLogger.EXPECT().Fatal(\"Unable to initialize domain cache\", gomock.Any()).Times(1)\n\n\ts.domainCache.Start()\n}\n\nfunc (s *domainCacheSuite) TestUnregisterDomainChangeCallback() {\n\ts.domainCache.prepareCallbacks = map[string]PrepareCallbackFn{\n\t\t\"1\": func() {},\n\t}\n\ts.domainCache.callbacks = map[string]CallbackFn{\n\t\t\"1\": func([]*DomainCacheEntry) {},\n\t}\n\n\ts.domainCache.UnregisterDomainChangeCallback(\"1\")\n\ts.Empty(s.domainCache.prepareCallbacks)\n\ts.Empty(s.domainCache.callbacks)\n}\n\nfunc (s *domainCacheSuite) TestGetDomain_Error() {\n\tentry, err := s.domainCache.GetDomain(\"\")\n\ts.Nil(entry)\n\ts.ErrorContains(err, \"Domain name is empty\")\n}\n\nfunc (s *domainCacheSuite) TestGetDomainByID_Error() {\n\tentry, err := s.domainCache.GetDomainByID(\"\")\n\ts.Nil(entry)\n\ts.ErrorContains(err, \"DomainID is empty.\")\n}\n\nfunc (s *domainCacheSuite) TestGetDomainID() {\n\tentry, err := s.domainCache.GetDomainID(\"\")\n\ts.Empty(entry)\n\ts.ErrorContains(err, \"Domain name is empty\")\n\n\tdomainName := \"testDomain\"\n\tdomainID := \"testDomainID\"\n\n\ttestCache := newDomainCache()\n\tdomainEntry := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t},\n\t\tinitialized: true,\n\t}\n\ttestCache.Put(domainName, domainID)\n\ttestCache.Put(domainID, domainEntry)\n\n\ts.domainCache.cacheByID.Store(testCache)\n\n\ts.domainCache.cacheNameToID.Store(testCache)\n\n\tentry, err = s.domainCache.GetDomainID(domainName)\n\n\ts.Nil(err)\n\ts.Equal(domainID, entry)\n}\n\nfunc (s *domainCacheSuite) TestGetDomainName() {\n\tdomainName := \"testDomain\"\n\tdomainID := \"testDomainID\"\n\n\ttestCache := newDomainCache()\n\tdomainEntry := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t},\n\t\tinitialized: true,\n\t}\n\ttestCache.Put(domainName, domainID)\n\ttestCache.Put(domainID, domainEntry)\n\n\ts.domainCache.cacheByID.Store(testCache)\n\n\ts.domainCache.cacheNameToID.Store(testCache)\n\n\tentry, err := s.domainCache.GetDomainName(domainID)\n\n\ts.Nil(err)\n\ts.Equal(domainName, entry)\n}\n\nfunc (s *domainCacheSuite) TestGetDomainName_Error() {\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: \"\", ID: \"\"}).Return(nil, assert.AnError).Once()\n\n\tentry, err := s.domainCache.GetDomainName(\"\")\n\ts.Empty(entry)\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_updateIDToDomainCache_Error() {\n\tdomainCacheEntry := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{ID: \"testDomainID\", Name: \"testDomain\"},\n\t}\n\tnewCache := NewMockCache(gomock.NewController(s.T()))\n\n\tnewCache.EXPECT().PutIfNotExist(\"testDomainID\", &DomainCacheEntry{}).Return(false, assert.AnError).Times(1)\n\n\ttriggerCallback, entry, err := s.domainCache.updateIDToDomainCache(newCache, \"testDomainID\", domainCacheEntry)\n\n\ts.False(triggerCallback)\n\ts.Nil(entry)\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_getDomain_Error_checkDomainExists() {\n\tdomainName := \"testDomain\"\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: domainName, ID: \"\"}).Return(nil, assert.AnError).Once()\n\n\tentry, err := s.domainCache.getDomain(domainName)\n\n\ts.Nil(entry)\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_getDomain_Error_refreshDomainsLocked() {\n\tdomainName := \"testDomain\"\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: domainName, ID: \"\"}).Return(nil, nil).Once()\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(nil, assert.AnError).Once()\n\n\tentry, err := s.domainCache.getDomain(domainName)\n\n\ts.Nil(entry)\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_getDomain_cacheHitAfterRefreshLockLocked() {\n\tdomainName := \"testDomain\"\n\tdomainID := \"testDomainID\"\n\n\ttestCache := newDomainCache()\n\tdomainEntry := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{ID: domainID, Name: domainName, Data: map[string]string{\"k1\": \"v1\"}},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t},\n\t\tinitialized: true,\n\t}\n\n\ttestCache.Put(domainName, domainID)\n\ttestCache.Put(domainID, domainEntry)\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\n\tgo func() {\n\t\tdefer s.domainCache.refreshLock.Unlock()\n\t\t// to test the cache hit after refresh lock is locked, need to ensure that the domain cache is added after the first cache hit check\n\t\t// force a lock to ensure that the code will block on the lock, wait for the first cache hit check, add the domain cache\n\t\t// and then release the lock\n\t\ts.domainCache.refreshLock.Lock()\n\t\twg.Done()\n\t\ttime.Sleep(200 * time.Millisecond)\n\t\ts.domainCache.cacheByID.Store(testCache)\n\t\ts.domainCache.cacheNameToID.Store(testCache)\n\t}()\n\n\twg.Wait()\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: domainName, ID: \"\"}).Return(nil, nil).Once()\n\n\tentry, err := s.domainCache.getDomain(domainName)\n\n\ts.Nil(err)\n\t// because the code makes deep copies, it's not possible to compare all the pointers directly\n\ts.Equal(domainEntry.info.Name, entry.info.Name)\n\ts.Equal(domainEntry.info.ID, entry.info.ID)\n\ts.Equal(1, len(entry.info.Data))\n\ts.Equal(\"v1\", entry.info.Data[\"k1\"])\n}\n\nfunc (s *domainCacheSuite) Test_getDomainByID_refreshDomainsLockedError() {\n\tdomainID := \"testDomainID\"\n\n\ts.metadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{Name: \"\", ID: domainID}).Return(nil, nil).Once()\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(nil, assert.AnError).Once()\n\n\tentry, err := s.domainCache.getDomainByID(domainID, false)\n\n\ts.Nil(entry)\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_refreshLoop_domainCacheRefreshedError() {\n\tmockedTimeSource := clock.NewMockedTimeSource()\n\n\ts.domainCache.timeSource = mockedTimeSource\n\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(nil, assert.AnError).Once()\n\n\tgo func() {\n\t\tmockedTimeSource.BlockUntil(1)\n\t\tmockedTimeSource.Advance(DomainCacheRefreshInterval)\n\t\ts.domainCache.shutdownChan <- struct{}{}\n\t}()\n\n\ts.domainCache.refreshLoop()\n}\n\nfunc (s *domainCacheSuite) Test_refreshDomainsLocked_IntervalTooShort() {\n\tmockedTimeSource := clock.NewMockedTimeSource()\n\n\ts.domainCache.timeSource = mockedTimeSource\n\n\ts.domainCache.lastRefreshTime = mockedTimeSource.Now()\n\tctx := context.Background()\n\n\terr := s.domainCache.refreshDomainsLocked(ctx)\n\ts.NoError(err)\n}\n\nfunc (s *domainCacheSuite) Test_refreshDomains_ListDomainsNonRetryableError() {\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: 0}, nil).Once()\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, mock.Anything).Return(nil, assert.AnError).Once()\n\n\terr := s.domainCache.refreshDomains()\n\ts.ErrorIs(err, assert.AnError)\n}\n\nfunc (s *domainCacheSuite) Test_refreshDomains_ListDomainsRetryableError() {\n\tretryableError := &types.ServiceBusyError{\n\t\tMessage: \"Service is busy\",\n\t}\n\n\t// We expect the metadataMgr to be called twice, once for the initial attempt and once for the retry\n\ts.metadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: 0}, nil).Times(2)\n\n\t// First time return retryable error\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, mock.Anything).Return(nil, retryableError).Once()\n\n\t// Second time return non-retryable error\n\ts.metadataMgr.On(\"ListDomains\", mock.Anything, mock.Anything).Return(nil, assert.AnError).Once()\n\n\terr := s.domainCache.refreshDomains()\n\n\t// We expect the error to be the first error\n\ts.ErrorIs(err, retryableError)\n}\n\nfunc (s *domainCacheSuite) TestDomainCacheEntry_Getters() {\n\tgen := testdatagen.New(s.T())\n\n\tentry := DomainCacheEntry{}\n\n\tgen.Fuzz(&entry)\n\n\ts.Equal(entry.info, entry.GetInfo())\n\ts.Equal(entry.config, entry.GetConfig())\n\ts.Equal(entry.replicationConfig, entry.GetReplicationConfig())\n\ts.Equal(entry.failoverNotificationVersion, entry.GetFailoverNotificationVersion())\n\ts.Equal(entry.notificationVersion, entry.GetNotificationVersion())\n\ts.Equal(entry.failoverVersion, entry.GetFailoverVersion())\n\ts.Equal(entry.previousFailoverVersion, entry.GetPreviousFailoverVersion())\n\ts.Equal(entry.failoverEndTime, entry.GetFailoverEndTime())\n\ts.Equal(entry.isGlobalDomain, entry.IsGlobalDomain())\n\ts.Equal(entry.configVersion, entry.GetConfigVersion())\n}\n\nfunc (s *domainCacheSuite) TestDomainCacheEntry_IsDomainPendingActive() {\n\t//\tLocal domain\n\tentry := CreateDomainCacheEntry(\"domainName\")\n\n\ts.False(entry.IsDomainPendingActive())\n\n\t//\tGlobal domain not pending active\n\n\tentry.isGlobalDomain = true\n\n\ts.False(entry.IsDomainPendingActive())\n\n\t//\tGlobal domain pending active\n\n\tentry.failoverEndTime = common.Int64Ptr(time.Now().Unix() + 100)\n\n\ts.True(entry.IsDomainPendingActive())\n}\n\nfunc (s *domainCacheSuite) TestDomainCacheEntry_GetReplicationPolicy() {\n\tentry := &DomainCacheEntry{\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"active\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: \"active\"},\n\t\t\t\t{ClusterName: \"standby\"},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.Equal(ReplicationPolicyOneCluster, entry.GetReplicationPolicy())\n\n\tentry.isGlobalDomain = true\n\n\ts.Equal(ReplicationPolicyMultiCluster, entry.GetReplicationPolicy())\n}\n\nfunc (s *domainCacheSuite) TestDomainCacheEntry_HasReplicationCluster() {\n\tentry := &DomainCacheEntry{\n\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"active\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: \"active\"},\n\t\t\t\t{ClusterName: \"standby\"},\n\t\t\t},\n\t\t},\n\t}\n\n\ts.True(entry.HasReplicationCluster(\"active\"))\n\ts.True(entry.HasReplicationCluster(\"standby\"))\n\ts.False(entry.HasReplicationCluster(\"other\"))\n}\n\nfunc (s *domainCacheSuite) TestDomainCacheEntry_IsDeprecatedOrDeleted() {\n\tentry := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{\n\t\t\tStatus: persistence.DomainStatusDeprecated,\n\t\t},\n\t}\n\n\ts.True(entry.IsDeprecatedOrDeleted())\n\n\tentry.info.Status = persistence.DomainStatusDeleted\n\n\ts.True(entry.IsDeprecatedOrDeleted())\n\n\tentry.info.Status = persistence.DomainStatusRegistered\n\n\ts.False(entry.IsDeprecatedOrDeleted())\n}\n\nfunc (s *domainCacheSuite) buildEntryFromRecord(record *persistence.GetDomainResponse) *DomainCacheEntry {\n\tnewEntry := &DomainCacheEntry{}\n\tnewEntry.info = &*record.Info\n\tnewEntry.config = &*record.Config\n\tnewEntry.replicationConfig = &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: record.ReplicationConfig.ActiveClusterName,\n\t}\n\tfor _, testCluster := range record.ReplicationConfig.Clusters {\n\t\tnewEntry.replicationConfig.Clusters = append(newEntry.replicationConfig.Clusters, &*testCluster)\n\t}\n\tnewEntry.configVersion = record.ConfigVersion\n\tnewEntry.failoverVersion = record.FailoverVersion\n\tnewEntry.isGlobalDomain = record.IsGlobalDomain\n\tnewEntry.failoverNotificationVersion = record.FailoverNotificationVersion\n\tnewEntry.notificationVersion = record.NotificationVersion\n\tnewEntry.initialized = true\n\treturn newEntry\n}\n\nfunc Test_GetRetentionDays(t *testing.T) {\n\td := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{\n\t\t\tData: make(map[string]string),\n\t\t},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tRetention: 7,\n\t\t},\n\t}\n\td.info.Data[SampleRetentionKey] = \"30\"\n\td.info.Data[SampleRateKey] = \"0\"\n\n\twid := uuid.New()\n\trd := d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(7), rd)\n\n\td.info.Data[SampleRateKey] = \"1\"\n\trd = d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(30), rd)\n\n\td.info.Data[SampleRetentionKey] = \"invalid-value\"\n\trd = d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(7), rd) // fallback to normal retention\n\n\td.info.Data[SampleRetentionKey] = \"30\"\n\td.info.Data[SampleRateKey] = \"invalid-value\"\n\trd = d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(7), rd) // fallback to normal retention\n\n\twid = \"3aef42a8-db0a-4a3b-b8b7-9829d74b4ebf\"\n\td.info.Data[SampleRetentionKey] = \"30\"\n\td.info.Data[SampleRateKey] = \"0.8\"\n\trd = d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(7), rd) // fallback to normal retention\n\td.info.Data[SampleRateKey] = \"0.9\"\n\trd = d.GetRetentionDays(wid)\n\trequire.Equal(t, int32(30), rd)\n}\n\nfunc Test_IsSampledForLongerRetentionEnabled(t *testing.T) {\n\td := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{\n\t\t\tData: make(map[string]string),\n\t\t},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tRetention: 7,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t}\n\twid := uuid.New()\n\trequire.False(t, d.IsSampledForLongerRetentionEnabled(wid))\n\td.info.Data[SampleRetentionKey] = \"30\"\n\td.info.Data[SampleRateKey] = \"0\"\n\trequire.True(t, d.IsSampledForLongerRetentionEnabled(wid))\n}\n\nfunc Test_IsSampledForLongerRetention(t *testing.T) {\n\td := &DomainCacheEntry{\n\t\tinfo: &persistence.DomainInfo{\n\t\t\tData: make(map[string]string),\n\t\t},\n\t\tconfig: &persistence.DomainConfig{\n\t\t\tRetention: 7,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t},\n\t\t},\n\t}\n\twid := uuid.New()\n\trequire.False(t, d.IsSampledForLongerRetention(wid))\n\n\td.info.Data[SampleRetentionKey] = \"30\"\n\td.info.Data[SampleRateKey] = \"0\"\n\trequire.False(t, d.IsSampledForLongerRetention(wid))\n\n\td.info.Data[SampleRateKey] = \"1\"\n\trequire.True(t, d.IsSampledForLongerRetention(wid))\n\n\td.info.Data[SampleRateKey] = \"invalid-value\"\n\trequire.False(t, d.IsSampledForLongerRetention(wid))\n}\n\nfunc Test_WithTimeSource(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmetadataMgr := &mocks.MetadataManager{}\n\n\ttimeSource := clock.NewRealTimeSource()\n\tdomainCache := NewDomainCache(metadataMgr, cluster.GetTestClusterMetadata(true), metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), log.NewNoop(), WithTimeSource(timeSource))\n\n\tassert.Equal(t, timeSource, domainCache.timeSource)\n}\n\nfunc Test_NewLocalDomainCacheEntryForTest(t *testing.T) {\n\tdomain := NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"test-domain\"}, nil, \"targetCluster\")\n\tassert.False(t, domain.IsGlobalDomain())\n}\n\nfunc Test_NewDomainNotActiveError(t *testing.T) {\n\ttests := []struct {\n\t\tmsg            string\n\t\tdomain         *DomainCacheEntry\n\t\tcurrentCluster string\n\t\tactiveCluster  string\n\t\texpectedErr    *types.DomainNotActiveError\n\t}{\n\t\t{\n\t\t\tmsg:            \"local domain\",\n\t\t\tdomain:         NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"test-domain\"}, nil, \"targetCluster\"),\n\t\t\tcurrentCluster: \"currentCluster\",\n\t\t\tactiveCluster:  \"targetCluster\",\n\t\t\texpectedErr: &types.DomainNotActiveError{\n\t\t\t\tMessage:        \"Domain: test-domain is active in cluster: targetCluster, while current cluster currentCluster is a standby cluster.\",\n\t\t\t\tDomainName:     \"test-domain\",\n\t\t\t\tCurrentCluster: \"currentCluster\",\n\t\t\t\tActiveCluster:  \"targetCluster\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:            \"active-active domain\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tactiveCluster:  \"cluster2\",\n\t\t\tdomain: NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{Name: \"test-domain\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t),\n\t\t\texpectedErr: &types.DomainNotActiveError{\n\t\t\t\tMessage:        \"Domain: test-domain is active in cluster(s): [cluster1 cluster2], while current cluster cluster1 is a standby cluster. Operation active cluster: cluster2\",\n\t\t\t\tDomainName:     \"test-domain\",\n\t\t\t\tCurrentCluster: \"cluster1\",\n\t\t\t\tActiveClusters: []string{\"cluster1\", \"cluster2\"},\n\t\t\t\tActiveCluster:  \"cluster2\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.msg, func(t *testing.T) {\n\t\t\terr := tt.domain.NewDomainNotActiveError(tt.currentCluster, tt.activeCluster)\n\t\t\tassert.Equal(t, tt.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_getActiveClusters(t *testing.T) {\n\ttests := []struct {\n\t\tmsg                    string\n\t\treplicationConfig      *persistence.DomainReplicationConfig\n\t\texpectedActiveClusters []string\n\t}{\n\t\t{\n\t\t\tmsg: \"active-passive domain\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: \"active\",\n\t\t\t},\n\t\t\texpectedActiveClusters: nil,\n\t\t},\n\t\t{\n\t\t\tmsg: \"active-active domain\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveClusters: []string{\"cluster1\", \"cluster2\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.msg, func(t *testing.T) {\n\t\t\tactiveClusters := getActiveClusters(tt.replicationConfig)\n\t\t\tassert.Equal(t, tt.expectedActiveClusters, activeClusters)\n\t\t})\n\t}\n}\n\nfunc Test_GetActiveClusterInfoByClusterAttribute(t *testing.T) {\n\ttests := []struct {\n\t\tname                  string\n\t\tentry                 *DomainCacheEntry\n\t\tclusterAttribute      *types.ClusterAttribute\n\t\texpectedActiveCluster *types.ActiveClusterInfo\n\t\texpectedFound         bool\n\t}{\n\t\t{\n\t\t\tname: \"nil cluster attribute - returns domain-level active cluster info\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: nil,\n\t\t\texpectedActiveCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\tFailoverVersion:   100,\n\t\t\t},\n\t\t\texpectedFound: true,\n\t\t},\n\t\t{\n\t\t\tname: \"nil active clusters - returns false\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters:    nil,\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: nil,\n\t\t\texpectedFound:         false,\n\t\t},\n\t\t{\n\t\t\tname: \"nil attribute scopes - returns false\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: nil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: nil,\n\t\t\texpectedFound:         false,\n\t\t},\n\t\t{\n\t\t\tname: \"scope not found - returns false\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\", // different scope\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: nil,\n\t\t\texpectedFound:         false,\n\t\t},\n\t\t{\n\t\t\tname: \"attribute not found in scope - returns false\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\", // different name\n\t\t\t},\n\t\t\texpectedActiveCluster: nil,\n\t\t\texpectedFound:         false,\n\t\t},\n\t\t{\n\t\t\tname: \"successful lookup - returns cluster attribute info\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster-west\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster-east\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   250,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"cluster-west\",\n\t\t\t\tFailoverVersion:   300,\n\t\t\t},\n\t\t\texpectedFound: true,\n\t\t},\n\t\t{\n\t\t\tname: \"multiple scopes - successful lookup in specific scope\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"region-cluster-west\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"dc-cluster-west\",\n\t\t\t\t\t\t\t\t\t\tFailoverVersion:   400,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"datacenter\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: \"dc-cluster-west\",\n\t\t\t\tFailoverVersion:   400,\n\t\t\t},\n\t\t\texpectedFound: true,\n\t\t},\n\t\t{\n\t\t\tname: \"empty cluster attributes map in scope - returns false\",\n\t\t\tentry: &DomainCacheEntry{\n\t\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"domain-active-cluster\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfailoverVersion: 100,\n\t\t\t},\n\t\t\tclusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"us-west\",\n\t\t\t},\n\t\t\texpectedActiveCluster: nil,\n\t\t\texpectedFound:         false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tactualActiveCluster, actualFound := tt.entry.GetActiveClusterInfoByClusterAttribute(tt.clusterAttribute)\n\n\t\t\tassert.Equal(t, tt.expectedFound, actualFound)\n\t\t\tassert.Equal(t, tt.expectedActiveCluster, actualActiveCluster)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/cache/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: cache.go\n//\n// Generated by this command:\n//\n//\tmockgen -package cache -source cache.go -destination interface_mock.go -self_package github.com/uber/cadence/common/cache\n//\n\n// Package cache is a generated GoMock package.\npackage cache\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n)\n\n// MockCache is a mock of Cache interface.\ntype MockCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockCacheMockRecorder is the mock recorder for MockCache.\ntype MockCacheMockRecorder struct {\n\tmock *MockCache\n}\n\n// NewMockCache creates a new mock instance.\nfunc NewMockCache(ctrl *gomock.Controller) *MockCache {\n\tmock := &MockCache{ctrl: ctrl}\n\tmock.recorder = &MockCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCache) EXPECT() *MockCacheMockRecorder {\n\treturn m.recorder\n}\n\n// Delete mocks base method.\nfunc (m *MockCache) Delete(key any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Delete\", key)\n}\n\n// Delete indicates an expected call of Delete.\nfunc (mr *MockCacheMockRecorder) Delete(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Delete\", reflect.TypeOf((*MockCache)(nil).Delete), key)\n}\n\n// Get mocks base method.\nfunc (m *MockCache) Get(key any) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Get\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Get indicates an expected call of Get.\nfunc (mr *MockCacheMockRecorder) Get(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Get\", reflect.TypeOf((*MockCache)(nil).Get), key)\n}\n\n// Iterator mocks base method.\nfunc (m *MockCache) Iterator() Iterator {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Iterator\")\n\tret0, _ := ret[0].(Iterator)\n\treturn ret0\n}\n\n// Iterator indicates an expected call of Iterator.\nfunc (mr *MockCacheMockRecorder) Iterator() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Iterator\", reflect.TypeOf((*MockCache)(nil).Iterator))\n}\n\n// Put mocks base method.\nfunc (m *MockCache) Put(key, value any) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Put\", key, value)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Put indicates an expected call of Put.\nfunc (mr *MockCacheMockRecorder) Put(key, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Put\", reflect.TypeOf((*MockCache)(nil).Put), key, value)\n}\n\n// PutIfNotExist mocks base method.\nfunc (m *MockCache) PutIfNotExist(key, value any) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PutIfNotExist\", key, value)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PutIfNotExist indicates an expected call of PutIfNotExist.\nfunc (mr *MockCacheMockRecorder) PutIfNotExist(key, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PutIfNotExist\", reflect.TypeOf((*MockCache)(nil).PutIfNotExist), key, value)\n}\n\n// Release mocks base method.\nfunc (m *MockCache) Release(key any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Release\", key)\n}\n\n// Release indicates an expected call of Release.\nfunc (mr *MockCacheMockRecorder) Release(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Release\", reflect.TypeOf((*MockCache)(nil).Release), key)\n}\n\n// Size mocks base method.\nfunc (m *MockCache) Size() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Size\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Size indicates an expected call of Size.\nfunc (mr *MockCacheMockRecorder) Size() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Size\", reflect.TypeOf((*MockCache)(nil).Size))\n}\n\n// MockIterator is a mock of Iterator interface.\ntype MockIterator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockIteratorMockRecorder\n\tisgomock struct{}\n}\n\n// MockIteratorMockRecorder is the mock recorder for MockIterator.\ntype MockIteratorMockRecorder struct {\n\tmock *MockIterator\n}\n\n// NewMockIterator creates a new mock instance.\nfunc NewMockIterator(ctrl *gomock.Controller) *MockIterator {\n\tmock := &MockIterator{ctrl: ctrl}\n\tmock.recorder = &MockIteratorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockIterator) EXPECT() *MockIteratorMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockIterator) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockIteratorMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockIterator)(nil).Close))\n}\n\n// HasNext mocks base method.\nfunc (m *MockIterator) HasNext() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasNext\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasNext indicates an expected call of HasNext.\nfunc (mr *MockIteratorMockRecorder) HasNext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasNext\", reflect.TypeOf((*MockIterator)(nil).HasNext))\n}\n\n// Next mocks base method.\nfunc (m *MockIterator) Next() Entry {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Next\")\n\tret0, _ := ret[0].(Entry)\n\treturn ret0\n}\n\n// Next indicates an expected call of Next.\nfunc (mr *MockIteratorMockRecorder) Next() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Next\", reflect.TypeOf((*MockIterator)(nil).Next))\n}\n\n// MockEntry is a mock of Entry interface.\ntype MockEntry struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEntryMockRecorder\n\tisgomock struct{}\n}\n\n// MockEntryMockRecorder is the mock recorder for MockEntry.\ntype MockEntryMockRecorder struct {\n\tmock *MockEntry\n}\n\n// NewMockEntry creates a new mock instance.\nfunc NewMockEntry(ctrl *gomock.Controller) *MockEntry {\n\tmock := &MockEntry{ctrl: ctrl}\n\tmock.recorder = &MockEntryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEntry) EXPECT() *MockEntryMockRecorder {\n\treturn m.recorder\n}\n\n// CreateTime mocks base method.\nfunc (m *MockEntry) CreateTime() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTime\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// CreateTime indicates an expected call of CreateTime.\nfunc (mr *MockEntryMockRecorder) CreateTime() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTime\", reflect.TypeOf((*MockEntry)(nil).CreateTime))\n}\n\n// Key mocks base method.\nfunc (m *MockEntry) Key() any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Key\")\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Key indicates an expected call of Key.\nfunc (mr *MockEntryMockRecorder) Key() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Key\", reflect.TypeOf((*MockEntry)(nil).Key))\n}\n\n// Value mocks base method.\nfunc (m *MockEntry) Value() any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Value\")\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Value indicates an expected call of Value.\nfunc (mr *MockEntryMockRecorder) Value() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Value\", reflect.TypeOf((*MockEntry)(nil).Value))\n}\n\n// MockDomainMetricsScopeCache is a mock of DomainMetricsScopeCache interface.\ntype MockDomainMetricsScopeCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainMetricsScopeCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainMetricsScopeCacheMockRecorder is the mock recorder for MockDomainMetricsScopeCache.\ntype MockDomainMetricsScopeCacheMockRecorder struct {\n\tmock *MockDomainMetricsScopeCache\n}\n\n// NewMockDomainMetricsScopeCache creates a new mock instance.\nfunc NewMockDomainMetricsScopeCache(ctrl *gomock.Controller) *MockDomainMetricsScopeCache {\n\tmock := &MockDomainMetricsScopeCache{ctrl: ctrl}\n\tmock.recorder = &MockDomainMetricsScopeCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainMetricsScopeCache) EXPECT() *MockDomainMetricsScopeCacheMockRecorder {\n\treturn m.recorder\n}\n\n// Get mocks base method.\nfunc (m *MockDomainMetricsScopeCache) Get(domainID string, scopeIdx metrics.ScopeIdx) (metrics.Scope, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Get\", domainID, scopeIdx)\n\tret0, _ := ret[0].(metrics.Scope)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// Get indicates an expected call of Get.\nfunc (mr *MockDomainMetricsScopeCacheMockRecorder) Get(domainID, scopeIdx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Get\", reflect.TypeOf((*MockDomainMetricsScopeCache)(nil).Get), domainID, scopeIdx)\n}\n\n// Put mocks base method.\nfunc (m *MockDomainMetricsScopeCache) Put(domainID string, scopeIdx metrics.ScopeIdx, metricsScope metrics.Scope) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Put\", domainID, scopeIdx, metricsScope)\n}\n\n// Put indicates an expected call of Put.\nfunc (mr *MockDomainMetricsScopeCacheMockRecorder) Put(domainID, scopeIdx, metricsScope any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Put\", reflect.TypeOf((*MockDomainMetricsScopeCache)(nil).Put), domainID, scopeIdx, metricsScope)\n}\n\n// Start mocks base method.\nfunc (m *MockDomainMetricsScopeCache) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockDomainMetricsScopeCacheMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockDomainMetricsScopeCache)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockDomainMetricsScopeCache) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockDomainMetricsScopeCacheMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockDomainMetricsScopeCache)(nil).Stop))\n}\n\n// MockSizeable is a mock of Sizeable interface.\ntype MockSizeable struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSizeableMockRecorder\n\tisgomock struct{}\n}\n\n// MockSizeableMockRecorder is the mock recorder for MockSizeable.\ntype MockSizeableMockRecorder struct {\n\tmock *MockSizeable\n}\n\n// NewMockSizeable creates a new mock instance.\nfunc NewMockSizeable(ctrl *gomock.Controller) *MockSizeable {\n\tmock := &MockSizeable{ctrl: ctrl}\n\tmock.recorder = &MockSizeableMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSizeable) EXPECT() *MockSizeableMockRecorder {\n\treturn m.recorder\n}\n\n// ByteSize mocks base method.\nfunc (m *MockSizeable) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockSizeableMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockSizeable)(nil).ByteSize))\n}\n"
  },
  {
    "path": "common/cache/lru.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cache\n\nimport (\n\t\"container/list\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nvar (\n\t// ErrCacheFull is returned if Put fails due to cache being filled with pinned elements\n\tErrCacheFull = errors.New(\"Cache capacity is fully occupied with pinned elements\")\n\n\t// ErrEntryTooBig is returned if the entry is too big to be cached\n\tErrEntryTooBig = errors.New(\"Entry is too big to be cached\")\n)\n\n// upper limit to prevent infinite growing\nconst cacheCountLimit = 1 << 25\n\n// default size limit for size based cache if not defined (1 GB)\nconst cacheDefaultSizeLimit = 1 << 30\n\n// lru is a concurrent fixed size cache that evicts elements in lru order\ntype (\n\tlru struct {\n\t\tmut           sync.Mutex\n\t\tbyAccess      *list.List\n\t\tbyKey         map[interface{}]*list.Element\n\t\tmaxCount      int\n\t\tttl           time.Duration\n\t\tpin           bool\n\t\trmFunc        RemovedFunc\n\t\tsizeFunc      GetCacheItemSizeFunc\n\t\tmaxSize       dynamicproperties.IntPropertyFn\n\t\tcurrSize      uint64\n\t\tsizeByKey     map[interface{}]uint64\n\t\tisSizeBased   dynamicproperties.BoolPropertyFn\n\t\tactivelyEvict bool\n\t\t// We use this instead of time.Now() in order to make testing easier\n\t\ttimeSource   clock.TimeSource\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t\twarnOnce     sync.Once\n\t\terrorOnce    sync.Once\n\t}\n\n\titeratorImpl struct {\n\t\tlru        *lru\n\t\tcreateTime time.Time\n\t\tnextItem   *list.Element\n\t}\n\n\tentryImpl struct {\n\t\tkey        interface{}\n\t\tcreateTime time.Time\n\t\tvalue      interface{}\n\t\trefCount   int\n\t}\n)\n\n// Close closes the iterator\nfunc (it *iteratorImpl) Close() {\n\tit.lru.mut.Unlock()\n}\n\n// HasNext return true if there is more items to be returned\nfunc (it *iteratorImpl) HasNext() bool {\n\treturn it.nextItem != nil\n}\n\n// Next return the next item\nfunc (it *iteratorImpl) Next() Entry {\n\tif it.nextItem == nil {\n\t\tpanic(\"LRU cache iterator Next called when there is no next item\")\n\t}\n\n\tentry := it.nextItem.Value.(*entryImpl)\n\tit.nextItem = it.nextItem.Next()\n\t// make a copy of the entry so there will be no concurrent access to this entry\n\tentry = &entryImpl{\n\t\tkey:        entry.key,\n\t\tvalue:      entry.value,\n\t\tcreateTime: entry.createTime,\n\t}\n\tit.prepareNext()\n\treturn entry\n}\n\nfunc (it *iteratorImpl) prepareNext() {\n\tfor it.nextItem != nil {\n\t\tentry := it.nextItem.Value.(*entryImpl)\n\t\tif it.lru.isEntryExpired(entry, it.createTime) {\n\t\t\tnextItem := it.nextItem.Next()\n\t\t\tit.lru.deleteInternal(it.nextItem)\n\t\t\tit.nextItem = nextItem\n\t\t} else {\n\t\t\treturn\n\t\t}\n\t}\n}\n\n// Iterator returns an iterator to the map. This map\n// does not use re-entrant locks, so access or modification\n// to the map during iteration can cause a dead lock.\nfunc (c *lru) Iterator() Iterator {\n\tc.mut.Lock()\n\titerator := &iteratorImpl{\n\t\tlru:        c,\n\t\tcreateTime: c.timeSource.Now(),\n\t\tnextItem:   c.byAccess.Front(),\n\t}\n\titerator.prepareNext()\n\treturn iterator\n}\n\nfunc (entry *entryImpl) Key() interface{} {\n\treturn entry.key\n}\n\nfunc (entry *entryImpl) Value() interface{} {\n\treturn entry.value\n}\n\nfunc (entry *entryImpl) CreateTime() time.Time {\n\treturn entry.createTime\n}\n\n// New creates a new cache with the given options\nfunc New(opts *Options) Cache {\n\tif opts == nil || (opts.MaxCount <= 0 && opts.MaxSize() <= 0) {\n\t\tpanic(\"Either MaxCount (count based) or \" +\n\t\t\t\"MaxSize must be provided for the LRU cache\")\n\t}\n\n\ttimeSource := opts.TimeSource\n\tif timeSource == nil {\n\t\ttimeSource = clock.NewRealTimeSource()\n\t}\n\n\tcache := &lru{\n\t\tbyAccess:      list.New(),\n\t\tbyKey:         make(map[interface{}]*list.Element, opts.InitialCapacity),\n\t\tttl:           opts.TTL,\n\t\tpin:           opts.Pin,\n\t\trmFunc:        opts.RemovedFunc,\n\t\tactivelyEvict: opts.ActivelyEvict,\n\t\ttimeSource:    timeSource,\n\t\tlogger:        opts.Logger,\n\t\tisSizeBased:   opts.IsSizeBased,\n\t\tmetricsScope:  opts.MetricsScope,\n\t}\n\n\tif cache.logger == nil {\n\t\tcache.logger = log.NewNoop()\n\t}\n\n\tif cache.metricsScope == nil {\n\t\tcache.metricsScope = metrics.NoopScope\n\t}\n\n\tif opts.IsSizeBased == nil {\n\t\tcache.isSizeBased = dynamicproperties.GetBoolPropertyFn(false)\n\t} else {\n\t\tcache.isSizeBased = opts.IsSizeBased\n\t}\n\n\tcache.sizeFunc = opts.GetCacheItemSizeFunc\n\tcache.maxSize = opts.MaxSize\n\tif cache.maxSize == nil {\n\t\t// If maxSize is not defined for size-based cache, set default to cacheCountLimit\n\t\tcache.maxSize = dynamicproperties.GetIntPropertyFn(cacheDefaultSizeLimit)\n\t}\n\tcache.sizeByKey = make(map[interface{}]uint64, opts.InitialCapacity)\n\tcache.maxCount = opts.MaxCount\n\n\tcache.logger.Info(\"LRU cache initialized\",\n\t\ttag.Value(map[string]interface{}{\n\t\t\t\"isSizeBased\": cache.isSizeBased(),\n\t\t\t\"maxCount\":    cache.maxCount,\n\t\t\t\"maxSize\":     cache.maxSize(),\n\t\t}),\n\t)\n\n\treturn cache\n}\n\n// Get retrieves the value stored under the given key\nfunc (c *lru) Get(key interface{}) interface{} {\n\tc.mut.Lock()\n\tdefer c.mut.Unlock()\n\n\tc.evictExpiredItems()\n\n\telement := c.byKey[key]\n\tif element == nil {\n\t\tc.metricsScope.IncCounter(metrics.BaseCacheMiss)\n\t\treturn nil\n\t}\n\n\tentry := element.Value.(*entryImpl)\n\n\tif c.isEntryExpired(entry, c.timeSource.Now()) {\n\t\t// Entry has expired\n\t\tc.deleteInternal(element)\n\t\tc.metricsScope.IncCounter(metrics.BaseCacheMiss)\n\t\treturn nil\n\t}\n\n\tif c.pin {\n\t\tentry.refCount++\n\t}\n\tc.byAccess.MoveToFront(element)\n\tc.metricsScope.IncCounter(metrics.BaseCacheHit)\n\treturn entry.value\n}\n\n// Put puts a new value associated with a given key, returning the existing value (if present)\nfunc (c *lru) Put(key interface{}, value interface{}) interface{} {\n\tif c.pin {\n\t\tpanic(\"Cannot use Put API in Pin mode. Use Delete and PutIfNotExist if necessary\")\n\t}\n\tval, _ := c.putInternal(key, value, true)\n\treturn val\n}\n\n// PutIfNotExist puts a value associated with a given key if it does not exist\nfunc (c *lru) PutIfNotExist(key interface{}, value interface{}) (interface{}, error) {\n\texisting, err := c.putInternal(key, value, false)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif existing == nil {\n\t\t// This is a new value\n\t\treturn value, err\n\t}\n\n\treturn existing, err\n}\n\n// Delete deletes a key, value pair associated with a key\nfunc (c *lru) Delete(key interface{}) {\n\tc.mut.Lock()\n\tdefer c.mut.Unlock()\n\n\tc.evictExpiredItems()\n\n\telement := c.byKey[key]\n\tif element != nil {\n\t\tc.deleteInternal(element)\n\t}\n}\n\n// Release decrements the ref count of a pinned element.\nfunc (c *lru) Release(key interface{}) {\n\tc.mut.Lock()\n\tdefer c.mut.Unlock()\n\n\telt, ok := c.byKey[key]\n\tif !ok {\n\t\treturn\n\t}\n\tentry := elt.Value.(*entryImpl)\n\tentry.refCount--\n}\n\n// Size returns the number of entries currently in the lru, useful if cache is not full\nfunc (c *lru) Size() int {\n\tc.mut.Lock()\n\tdefer c.mut.Unlock()\n\n\tc.evictExpiredItems()\n\n\treturn len(c.byKey)\n}\n\n// evictExpiredItems evicts all items in the cache which are expired\nfunc (c *lru) evictExpiredItems() {\n\tif !c.activelyEvict {\n\t\treturn // do nothing if activelyEvict is not set\n\t}\n\n\tnow := c.timeSource.Now()\n\tfor elt := c.byAccess.Back(); elt != nil; elt = c.byAccess.Back() {\n\t\tif !c.isEntryExpired(elt.Value.(*entryImpl), now) {\n\t\t\t// List is sorted by item age, so we can stop as soon as we found first non expired item.\n\t\t\tbreak\n\t\t}\n\t\tc.deleteInternal(elt)\n\t}\n}\n\n// Put puts a new value associated with a given key, returning the existing value (if present)\n// allowUpdate flag is used to control overwrite behavior if the value exists\nfunc (c *lru) putInternal(key interface{}, value interface{}, allowUpdate bool) (interface{}, error) {\n\tvalueSize := uint64(1)\n\tsizeableValue, ok := value.(Sizeable)\n\n\tc.mut.Lock()\n\tdefer c.mut.Unlock()\n\n\tif !ok {\n\t\tc.warnOnce.Do(func() {\n\t\t\tc.logger.Warn(fmt.Sprintf(\"Cache is strictly count-based because value %T does not implement sizable\", value))\n\t\t})\n\t} else {\n\t\tvalueSize = sizeableValue.ByteSize()\n\t}\n\n\tc.evictExpiredItems()\n\n\telement := c.byKey[key]\n\tif element != nil {\n\t\tentry := element.Value.(*entryImpl)\n\t\tif c.isEntryExpired(entry, c.timeSource.Now()) {\n\t\t\t// Entry has expired\n\t\t\tc.deleteInternal(element)\n\t\t} else {\n\t\t\t// replace the value\n\t\t\texisting := entry.value\n\t\t\tif allowUpdate {\n\t\t\t\tif c.isCacheFull() {\n\t\t\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheFullCounter)\n\t\t\t\t}\n\t\t\t\tfor c.isCacheFull() {\n\t\t\t\t\t// Find the oldest unpinned item to evict\n\t\t\t\t\toldest := c.byAccess.Back()\n\t\t\t\t\tfor oldest != nil {\n\t\t\t\t\t\tentry := oldest.Value.(*entryImpl)\n\t\t\t\t\t\tif entry.refCount == 0 {\n\t\t\t\t\t\t\t// Found an unpinned item, evict it\n\t\t\t\t\t\t\tc.deleteInternal(oldest)\n\t\t\t\t\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheEvictCounter)\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t\toldest = oldest.Prev()\n\t\t\t\t\t}\n\t\t\t\t\tif oldest == nil {\n\t\t\t\t\t\t// All items are pinned, can't evict anything\n\t\t\t\t\t\treturn existing, ErrCacheFull\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tc.updateSizeOnDelete(key)\n\t\t\t\tc.updateSizeOnAdd(key, valueSize)\n\t\t\t\tentry.value = value\n\t\t\t\tif c.ttl != 0 {\n\t\t\t\t\tentry.createTime = c.timeSource.Now()\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tc.byAccess.MoveToFront(element)\n\t\t\tif c.pin {\n\t\t\t\tentry.refCount++\n\t\t\t}\n\t\t\treturn existing, nil\n\t\t}\n\t}\n\n\t// add the value if possible\n\tentry := &entryImpl{\n\t\tkey:   key,\n\t\tvalue: value,\n\t}\n\n\tif c.pin {\n\t\tentry.refCount++\n\t}\n\n\tif c.ttl != 0 {\n\t\tentry.createTime = c.timeSource.Now()\n\t}\n\n\t// ensuring that the cache has at least one spot for the new entry\n\t// different logic between count and size approach\n\tif c.isSizeBased() {\n\t\tif valueSize > uint64(c.maxSize()) {\n\t\t\t// value is too big to be cached, we also don't want to evict everyone else\n\t\t\t// TODO: we should handle this logic in the caller\n\t\t\treturn nil, ErrEntryTooBig\n\t\t}\n\t\tc.byKey[key] = c.byAccess.PushFront(entry)\n\t\tc.updateSizeOnAdd(key, valueSize)\n\t\tif c.isCacheFull() {\n\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheFullCounter)\n\t\t}\n\t\tfor c.isCacheFull() {\n\t\t\t// Find the oldest unpinned item to evict\n\t\t\toldest := c.byAccess.Back()\n\t\t\tfor oldest != nil {\n\t\t\t\tentry := oldest.Value.(*entryImpl)\n\t\t\t\tif entry.refCount == 0 {\n\t\t\t\t\t// Found an unpinned item, evict it\n\t\t\t\t\tc.deleteInternal(oldest)\n\t\t\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheEvictCounter)\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\toldest = oldest.Prev()\n\t\t\t}\n\t\t\tif oldest == nil {\n\t\t\t\t// All items are pinned, can't evict anything\n\t\t\t\tc.deleteInternal(c.byAccess.Front())\n\t\t\t\treturn nil, ErrCacheFull\n\t\t\t}\n\t\t}\n\t} else {\n\t\tc.byKey[key] = c.byAccess.PushFront(entry)\n\t\tc.updateSizeOnAdd(key, valueSize)\n\t\tif c.isCacheFull() {\n\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheFullCounter)\n\t\t}\n\t\tfor c.isCacheFull() {\n\t\t\t// Find the oldest unpinned item to evict\n\t\t\toldest := c.byAccess.Back()\n\t\t\tfor oldest != nil {\n\t\t\t\tentry := oldest.Value.(*entryImpl)\n\t\t\t\tif entry.refCount <= 0 {\n\t\t\t\t\t// Found an unpinned item, evict it\n\t\t\t\t\tc.deleteInternal(oldest)\n\t\t\t\t\tc.metricsScope.IncCounter(metrics.BaseCacheEvictCounter)\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\toldest = oldest.Prev()\n\t\t\t}\n\t\t\tif oldest == nil {\n\t\t\t\t// All items are pinned, can't evict anything\n\t\t\t\tc.deleteInternal(c.byAccess.Front())\n\t\t\t\treturn nil, ErrCacheFull\n\t\t\t}\n\t\t}\n\t}\n\treturn nil, nil\n}\n\nfunc (c *lru) deleteInternal(element *list.Element) {\n\tentry := c.byAccess.Remove(element).(*entryImpl)\n\tif c.rmFunc != nil {\n\t\tgo c.rmFunc(entry.value)\n\t}\n\tdelete(c.byKey, entry.key)\n\tc.updateSizeOnDelete(entry.key)\n}\n\nfunc (c *lru) isEntryExpired(entry *entryImpl, currentTime time.Time) bool {\n\treturn entry.refCount == 0 && !entry.createTime.IsZero() && currentTime.After(entry.createTime.Add(c.ttl))\n}\n\nfunc (c *lru) isCacheFull() bool {\n\tcount := len(c.byKey)\n\tif c.isSizeBased() {\n\t\tif c.maxSize() == 0 {\n\t\t\t// we don't want to stop caching if maxSize is misconfigured to 0, we will use cacheDefaultSizeLimit instead\n\t\t\t// BUT we need to warn users for this config\n\t\t\tc.errorOnce.Do(func() {\n\t\t\t\tc.logger.Error(fmt.Sprintf(\"Cache size is misconfigured to 0 for value type %T, please fix config\", c.byKey[0].Value.(*entryImpl).value))\n\t\t\t})\n\t\t\treturn c.currSize > uint64(cacheDefaultSizeLimit) || count > cacheCountLimit\n\t\t}\n\t\treturn c.currSize > uint64(c.maxSize()) || count > cacheCountLimit\n\t}\n\treturn count > c.maxCount || count > cacheCountLimit\n}\n\nfunc (c *lru) updateSizeOnAdd(key interface{}, valueSize uint64) {\n\tc.sizeByKey[key] = valueSize\n\t// the int overflow should not happen here\n\tc.currSize += uint64(valueSize)\n\tc.emitSizeOnUpdate()\n\n}\n\nfunc (c *lru) updateSizeOnDelete(key interface{}) {\n\tc.currSize -= uint64(c.sizeByKey[key])\n\tc.emitSizeOnUpdate()\n\tdelete(c.sizeByKey, key)\n}\n\nfunc (c *lru) emitSizeOnUpdate() {\n\tc.metricsScope.UpdateGauge(metrics.BaseCacheByteSize, float64(c.currSize))\n\tc.metricsScope.UpdateGauge(metrics.BaseCacheByteSizeLimitGauge, float64(c.maxSize()))\n\tc.metricsScope.UpdateGauge(metrics.BaseCacheCount, float64(len(c.byKey)))\n\tc.metricsScope.UpdateGauge(metrics.BaseCacheCountLimitGauge, float64(c.maxCount))\n}\n"
  },
  {
    "path": "common/cache/lru_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cache\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype keyType struct {\n\tdummyString string\n\tdummyInt    int\n}\n\nfunc TestLRU(t *testing.T) {\n\tcache := New(&Options{MaxCount: 5})\n\n\tcache.Put(\"A\", \"Foo\")\n\tassert.Equal(t, \"Foo\", cache.Get(\"A\"))\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 1, cache.Size())\n\n\tcache.Put(\"B\", \"Bar\")\n\tcache.Put(\"C\", \"Cid\")\n\tcache.Put(\"D\", \"Delt\")\n\tassert.Equal(t, 4, cache.Size())\n\n\tassert.Equal(t, \"Bar\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Cid\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delt\", cache.Get(\"D\"))\n\n\tcache.Put(\"A\", \"Foo2\")\n\tassert.Equal(t, \"Foo2\", cache.Get(\"A\"))\n\n\tcache.Put(\"E\", \"Epsi\")\n\tassert.Equal(t, \"Epsi\", cache.Get(\"E\"))\n\tassert.Equal(t, \"Foo2\", cache.Get(\"A\"))\n\n\t// Access C, D is now LRU\n\tcache.Get(\"C\")\n\tcache.Put(\"F\", \"Felp\")\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 5, cache.Size())\n\n\tcache.Delete(\"C\")\n\tassert.Nil(t, cache.Get(\"C\"))\n}\n\nfunc TestGenerics(t *testing.T) {\n\tkey := keyType{\n\t\tdummyString: \"some random key\",\n\t\tdummyInt:    59,\n\t}\n\tvalue := \"some random value\"\n\n\tcache := New(&Options{MaxCount: 5})\n\tcache.Put(key, value)\n\n\tassert.Equal(t, value, cache.Get(key))\n\tassert.Equal(t, value, cache.Get(keyType{\n\t\tdummyString: \"some random key\",\n\t\tdummyInt:    59,\n\t}))\n\tassert.Nil(t, cache.Get(keyType{\n\t\tdummyString: \"some other random key\",\n\t\tdummyInt:    56,\n\t}))\n}\n\nfunc TestLRUWithTTL(t *testing.T) {\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.UnixMilli(0))\n\tcache := New(&Options{\n\t\tMaxCount:   5,\n\t\tTTL:        time.Millisecond * 100,\n\t\tTimeSource: mockTimeSource,\n\t}).(*lru)\n\n\tcache.Put(\"A\", \"foo\")\n\tassert.Equal(t, \"foo\", cache.Get(\"A\"))\n\n\tmockTimeSource.Advance(time.Millisecond * 300)\n\n\tassert.Nil(t, cache.Get(\"A\"))\n\tassert.Equal(t, 0, cache.Size())\n}\n\nfunc TestLRUCacheConcurrentAccess(t *testing.T) {\n\tcache := New(&Options{MaxCount: 5})\n\tvalues := map[string]string{\n\t\t\"A\": \"foo\",\n\t\t\"B\": \"bar\",\n\t\t\"C\": \"zed\",\n\t\t\"D\": \"dank\",\n\t\t\"E\": \"ezpz\",\n\t}\n\n\tfor k, v := range values {\n\t\tcache.Put(k, v)\n\t}\n\n\tstart := make(chan struct{})\n\tvar wg sync.WaitGroup\n\tfor i := 0; i < 20; i++ {\n\t\twg.Add(2)\n\n\t\t// concurrent get and put\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\n\t\t\t<-start\n\n\t\t\tfor j := 0; j < 1000; j++ {\n\t\t\t\tcache.Get(\"A\")\n\t\t\t\tcache.Put(\"A\", \"fooo\")\n\t\t\t}\n\t\t}()\n\n\t\t// concurrent iteration\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\n\t\t\t<-start\n\n\t\t\tfor j := 0; j < 50; j++ {\n\t\t\t\tvar result []Entry\n\t\t\t\tit := cache.Iterator()\n\t\t\t\tfor it.HasNext() {\n\t\t\t\t\tentry := it.Next()\n\t\t\t\t\tresult = append(result, entry) //nolint:staticcheck\n\t\t\t\t}\n\t\t\t\tit.Close()\n\t\t\t}\n\t\t}()\n\t}\n\n\tclose(start)\n\twg.Wait()\n}\n\nfunc TestRemoveFunc(t *testing.T) {\n\tch := make(chan bool)\n\tcache := New(&Options{\n\t\tMaxCount: 5,\n\t\tRemovedFunc: func(i interface{}) {\n\t\t\t_, ok := i.(*testing.T)\n\t\t\tassert.True(t, ok)\n\t\t\tch <- true\n\t\t},\n\t})\n\n\tcache.Put(\"testing\", t)\n\tcache.Delete(\"testing\")\n\tassert.Nil(t, cache.Get(\"testing\"))\n\n\ttimeout := time.NewTimer(time.Millisecond * 300)\n\tselect {\n\tcase b := <-ch:\n\t\tassert.True(t, b)\n\tcase <-timeout.C:\n\t\tt.Error(\"RemovedFunc did not send true on channel ch\")\n\t}\n}\n\nfunc TestRemovedFuncWithTTL(t *testing.T) {\n\tch := make(chan bool)\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.UnixMilli(0))\n\tcache := New(&Options{\n\t\tMaxCount: 5,\n\t\tTTL:      time.Millisecond * 50,\n\t\tRemovedFunc: func(i interface{}) {\n\t\t\t_, ok := i.(*testing.T)\n\t\t\tassert.True(t, ok)\n\t\t\tch <- true\n\t\t},\n\t\tTimeSource: mockTimeSource,\n\t}).(*lru)\n\n\tcache.Put(\"A\", t)\n\tassert.Equal(t, t, cache.Get(\"A\"))\n\n\tmockTimeSource.Advance(time.Millisecond * 100)\n\n\tassert.Nil(t, cache.Get(\"A\"))\n\n\tselect {\n\tcase b := <-ch:\n\t\tassert.True(t, b)\n\tcase <-mockTimeSource.After(100 * time.Millisecond):\n\t\tt.Error(\"RemovedFunc did not send true on channel ch\")\n\t}\n}\n\nfunc TestRemovedFuncWithTTL_Pin(t *testing.T) {\n\tch := make(chan bool)\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.UnixMilli(0))\n\tcache := New(&Options{\n\t\tMaxCount: 5,\n\t\tTTL:      time.Millisecond * 50,\n\t\tPin:      true,\n\t\tRemovedFunc: func(i interface{}) {\n\t\t\t_, ok := i.(*testing.T)\n\t\t\tassert.True(t, ok)\n\t\t\tch <- true\n\t\t},\n\t\tTimeSource: mockTimeSource,\n\t}).(*lru)\n\n\t_, err := cache.PutIfNotExist(\"A\", t)\n\tassert.NoError(t, err)\n\tassert.Equal(t, t, cache.Get(\"A\"))\n\tmockTimeSource.Advance(time.Millisecond * 100)\n\tassert.Equal(t, t, cache.Get(\"A\"))\n\t// release 3 time since put if not exist also increase the counter\n\tcache.Release(\"A\")\n\tcache.Release(\"A\")\n\tcache.Release(\"A\")\n\tassert.Nil(t, cache.Get(\"A\"))\n\n\tselect {\n\tcase b := <-ch:\n\t\tassert.True(t, b)\n\tcase <-mockTimeSource.After(300 * time.Millisecond):\n\t\tt.Error(\"RemovedFunc did not send true on channel ch\")\n\t}\n}\n\nfunc TestIterator(t *testing.T) {\n\texpected := map[string]string{\n\t\t\"A\": \"Alpha\",\n\t\t\"B\": \"Beta\",\n\t\t\"G\": \"Gamma\",\n\t\t\"D\": \"Delta\",\n\t}\n\n\tcache := New(&Options{MaxCount: 5})\n\n\tfor k, v := range expected {\n\t\tcache.Put(k, v)\n\t}\n\n\tactual := map[string]string{}\n\n\tit := cache.Iterator()\n\tfor it.HasNext() {\n\t\tentry := it.Next()\n\t\tactual[entry.Key().(string)] = entry.Value().(string)\n\t}\n\tit.Close()\n\tassert.Equal(t, expected, actual)\n\n\tit = cache.Iterator()\n\tfor i := 0; i < len(expected); i++ {\n\t\tentry := it.Next()\n\t\tactual[entry.Key().(string)] = entry.Value().(string)\n\t}\n\tit.Close()\n\tassert.Equal(t, expected, actual)\n}\n\n// Move the struct definition and method outside the test function\ntype sizeableValue struct {\n\tval  string\n\tsize uint64\n}\n\nfunc (s sizeableValue) ByteSize() uint64 {\n\treturn s.size\n}\n\nfunc TestLRU_SizeBased_SizeExceeded(t *testing.T) {\n\tcache := New(&Options{\n\t\tIsSizeBased: dynamicproperties.GetBoolPropertyFn(true),\n\t\tMaxSize:     dynamicproperties.GetIntPropertyFn(15),\n\t})\n\n\tfooValue := sizeableValue{val: \"Foo\", size: 5}\n\tcache.Put(\"A\", fooValue)\n\tassert.Equal(t, fooValue, cache.Get(\"A\"))\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 1, cache.Size())\n\n\tbarValue := sizeableValue{val: \"Bar\", size: 5}\n\tcidValue := sizeableValue{val: \"Cid\", size: 5}\n\tdeltValue := sizeableValue{val: \"Delt\", size: 5}\n\n\tcache.Put(\"B\", barValue)\n\tcache.Put(\"C\", cidValue)\n\tcache.Put(\"D\", deltValue)\n\tassert.Nil(t, cache.Get(\"A\"))\n\tassert.Equal(t, 3, cache.Size())\n\n\tassert.Equal(t, barValue, cache.Get(\"B\"))\n\tassert.Equal(t, cidValue, cache.Get(\"C\"))\n\tassert.Equal(t, deltValue, cache.Get(\"D\"))\n\n\tfoo2Value := sizeableValue{val: \"Foo2\", size: 5}\n\tcache.Put(\"A\", foo2Value)\n\tassert.Equal(t, foo2Value, cache.Get(\"A\"))\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 3, cache.Size())\n\n\t// Put large value to evict the rest in a loop\n\tepsiValue := sizeableValue{val: \"Epsi\", size: 15}\n\tcache.Put(\"E\", epsiValue)\n\tassert.Nil(t, cache.Get(\"C\"))\n\tassert.Equal(t, epsiValue, cache.Get(\"E\"))\n\tassert.Nil(t, cache.Get(\"A\"))\n\tassert.Equal(t, 1, cache.Size())\n\n\t// Put large value greater than maxSize but should not evict anything\n\tmepsiValue := sizeableValue{val: \"Mepsi\", size: 25}\n\tcache.Put(\"M\", mepsiValue)\n\tassert.Nil(t, cache.Get(\"M\"))\n\tassert.Equal(t, 1, cache.Size())\n}\n\nfunc TestLRU_SizeBased_CountExceeded(t *testing.T) {\n\tcache := New(&Options{\n\t\tMaxCount:    5,\n\t\tIsSizeBased: dynamicproperties.GetBoolPropertyFn(true),\n\t\tMaxSize:     dynamicproperties.GetIntPropertyFn(10000),\n\t})\n\n\tfooValue := sizeableValue{val: \"Foo\", size: 5}\n\tcache.Put(\"A\", fooValue)\n\tassert.Equal(t, fooValue, cache.Get(\"A\"))\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 1, cache.Size())\n\n\tbarValue := sizeableValue{val: \"Bar\", size: 5}\n\tcidValue := sizeableValue{val: \"Cid\", size: 5}\n\tdeltValue := sizeableValue{val: \"Delt\", size: 5}\n\n\tcache.Put(\"B\", barValue)\n\tcache.Put(\"C\", cidValue)\n\tcache.Put(\"D\", deltValue)\n\tassert.Equal(t, 4, cache.Size())\n\n\tassert.Equal(t, barValue, cache.Get(\"B\"))\n\tassert.Equal(t, cidValue, cache.Get(\"C\"))\n\tassert.Equal(t, deltValue, cache.Get(\"D\"))\n\n\tfoo2Value := sizeableValue{val: \"Foo2\", size: 5}\n\tcache.Put(\"A\", foo2Value)\n\tassert.Equal(t, foo2Value, cache.Get(\"A\"))\n\tassert.Equal(t, 4, cache.Size())\n\n\tepsiValue := sizeableValue{val: \"Epsi\", size: 5}\n\tcache.Put(\"E\", epsiValue)\n\tassert.Equal(t, barValue, cache.Get(\"B\"))\n\tassert.Equal(t, epsiValue, cache.Get(\"E\"))\n\tassert.Equal(t, foo2Value, cache.Get(\"A\"))\n\tassert.Equal(t, 5, cache.Size())\n}\n\nfunc TestLRU_EvictWhileSwitchToSizeBased(t *testing.T) {\n\tsizeBased := true\n\n\t// Create a function literal that can be implicitly coerced to BoolPropertyFn\n\tcache := New(&Options{\n\t\tMaxCount:    2,\n\t\tIsSizeBased: func(...dynamicproperties.FilterOption) bool { return sizeBased },\n\t\tMaxSize:     dynamicproperties.GetIntPropertyFn(10000),\n\t})\n\n\tfooValue := sizeableValue{val: \"Foo\", size: 5}\n\tbarValue := sizeableValue{val: \"Bar\", size: 5}\n\tcidValue := sizeableValue{val: \"Cid\", size: 5}\n\tdeltValue := sizeableValue{val: \"Delt\", size: 5}\n\tcache.Put(\"A\", fooValue)\n\tcache.Put(\"B\", barValue)\n\tcache.Put(\"C\", cidValue)\n\tcache.Put(\"D\", deltValue)\n\tassert.Equal(t, 4, cache.Size())\n\tassert.Equal(t, fooValue, cache.Get(\"A\"))\n\tassert.Equal(t, barValue, cache.Get(\"B\"))\n\tassert.Equal(t, cidValue, cache.Get(\"C\"))\n\tassert.Equal(t, deltValue, cache.Get(\"D\"))\n\n\t// Change the sizeBased flag to false\n\tsizeBased = false\n\n\techoValue := sizeableValue{val: \"Echo\", size: 5}\n\tcache.Put(\"E\", echoValue)\n\tassert.Equal(t, deltValue, cache.Get(\"D\"))\n\tassert.Equal(t, echoValue, cache.Get(\"E\"))\n\tassert.Equal(t, 2, cache.Size())\n}\n\nfunc TestPanicMaxCountAndSizeNotProvided(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r == nil {\n\t\t\tt.Errorf(\"The LRU was initialized without panic\")\n\t\t}\n\t}()\n\n\tNew(&Options{\n\t\tTTL: time.Millisecond * 100,\n\t\tGetCacheItemSizeFunc: func(interface{}) uint64 {\n\t\t\treturn 5\n\t\t},\n\t})\n}\n\nfunc TestPanicMaxCountAndSizeFuncNotProvided(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r == nil {\n\t\t\tt.Errorf(\"The LRU was initialized without panic\")\n\t\t}\n\t}()\n\n\tNew(&Options{\n\t\tTTL:     time.Millisecond * 100,\n\t\tMaxSize: dynamicproperties.GetIntPropertyFn(0),\n\t})\n}\n\nfunc TestPanicOptionsIsNil(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r == nil {\n\t\t\tt.Errorf(\"The LRU was initialized without panic\")\n\t\t}\n\t}()\n\n\tNew(nil)\n}\n\nfunc TestEvictItemsPastTimeToLive_ActivelyEvict(t *testing.T) {\n\t// Create the cache with a TTL of 75s\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.UnixMilli(0))\n\tcache, ok := New(&Options{\n\t\tMaxCount:      5,\n\t\tTTL:           time.Second * 75,\n\t\tActivelyEvict: true,\n\t\tTimeSource:    mockTimeSource,\n\t}).(*lru)\n\trequire.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", 1)\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", 2)\n\trequire.NoError(t, err)\n\n\t// Nothing is expired after 50s\n\tmockTimeSource.Advance(time.Second * 50)\n\tassert.Equal(t, 2, cache.Size())\n\n\t_, err = cache.PutIfNotExist(\"C\", 3)\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", 4)\n\trequire.NoError(t, err)\n\n\t// No time has passed, so still nothing is expired\n\tassert.Equal(t, 4, cache.Size())\n\n\t// Advance time to 100s, so A and B should be expired\n\tmockTimeSource.Advance(time.Second * 50)\n\tassert.Equal(t, 2, cache.Size())\n\n\t// Advance time to 150s, so C and D should be expired as well\n\tmockTimeSource.Advance(time.Second * 50)\n\tassert.Equal(t, 0, cache.Size())\n}\n\nfunc TestLRU_PutInternal_EvictUnpinnedInMiddle(t *testing.T) {\n\tcache, ok := New(&Options{\n\t\tMaxCount: 5,\n\t\tPin:      true,\n\t}).(*lru)\n\trequire.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", \"Alpha\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", \"Beta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"C\", \"Charlie\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", \"Delta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"E\", \"Echo\")\n\trequire.NoError(t, err)\n\n\t// Verify all items are present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// release C and D twice\n\tcache.Release(\"C\")\n\tcache.Release(\"C\")\n\tcache.Release(\"D\")\n\tcache.Release(\"D\")\n\n\t// Try to add a new item - should evict unpinned items C and D\n\t_, err = cache.PutIfNotExist(\"F\", \"Foxtrot\")\n\trequire.NoError(t, err)\n\n\t// Verify pinned items are still present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\tassert.Equal(t, \"Foxtrot\", cache.Get(\"F\"))\n\n\t// Verify only C was evicted\n\tassert.Nil(t, cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\n}\n\nfunc TestLRU_PutInternal_EvictUnpinnedFront(t *testing.T) {\n\tcache, ok := New(&Options{\n\t\tMaxCount: 5,\n\t\tPin:      true,\n\t}).(*lru)\n\trequire.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", \"Alpha\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", \"Beta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"C\", \"Charlie\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", \"Delta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"E\", \"Echo\")\n\trequire.NoError(t, err)\n\n\t// Verify all items are present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// move C to the front\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\n\t// release C twice\n\tcache.Release(\"C\")\n\tcache.Release(\"C\")\n\n\t// Try to add a new item - should evict unpinned item C\n\t_, err = cache.PutIfNotExist(\"F\", \"Foxtrot\")\n\trequire.NoError(t, err)\n\n\t// Verify pinned items are still present\n\tassert.Equal(t, \"Foxtrot\", cache.Get(\"F\"))\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Verify only C was evicted\n\tassert.Nil(t, cache.Get(\"C\"))\n}\n\nfunc TestLRU_PutInternal_EvictUnpinnedBack(t *testing.T) {\n\tcache, ok := New(&Options{\n\t\tMaxCount: 5,\n\t\tPin:      true,\n\t}).(*lru)\n\trequire.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", \"Alpha\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", \"Beta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"C\", \"Charlie\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", \"Delta\")\n\trequire.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"E\", \"Echo\")\n\trequire.NoError(t, err)\n\n\t// Verify all items are present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// release A twice\n\tcache.Release(\"A\")\n\tcache.Release(\"A\")\n\n\t// Try to add a new item - should evict unpinned item A\n\t_, err = cache.PutIfNotExist(\"F\", \"Foxtrot\")\n\trequire.NoError(t, err)\n\n\t// Verify pinned items are still present\n\tassert.Equal(t, \"Foxtrot\", cache.Get(\"F\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Verify A was evicted\n\tassert.Nil(t, cache.Get(\"A\"))\n}\n\nfunc TestLRU_PutInternal_AllPinned(t *testing.T) {\n\tcache, ok := New(&Options{\n\t\tMaxCount: 5,\n\t\tPin:      true,\n\t}).(*lru)\n\tassert.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", \"Alpha\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", \"Beta\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"C\", \"Charlie\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", \"Delta\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"E\", \"Echo\")\n\tassert.NoError(t, err)\n\n\t// Verify all items are present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Try to add a new item - should not evict anything\n\t_, err = cache.PutIfNotExist(\"F\", \"Foxtrot\")\n\tassert.Error(t, err)\n\n\t// Verify pinned items are still present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Verify F was never added\n\tassert.Nil(t, cache.Get(\"F\"))\n}\n\nfunc TestLRU_PutInternal_Unpinned(t *testing.T) {\n\tcache, ok := New(&Options{\n\t\tMaxCount: 5,\n\t\tPin:      false,\n\t}).(*lru)\n\tassert.True(t, ok)\n\n\t_, err := cache.PutIfNotExist(\"A\", \"Alpha\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"B\", \"Beta\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"C\", \"Charlie\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"D\", \"Delta\")\n\tassert.NoError(t, err)\n\t_, err = cache.PutIfNotExist(\"E\", \"Echo\")\n\tassert.NoError(t, err)\n\n\t// Verify all items are present\n\tassert.Equal(t, \"Alpha\", cache.Get(\"A\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Try to add a new item - should just evict A\n\t_, err = cache.PutIfNotExist(\"F\", \"Foxtrot\")\n\tassert.NoError(t, err)\n\n\tassert.Equal(t, \"Foxtrot\", cache.Get(\"F\"))\n\tassert.Equal(t, \"Beta\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Charlie\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delta\", cache.Get(\"D\"))\n\tassert.Equal(t, \"Echo\", cache.Get(\"E\"))\n\n\t// Verify A was evicted\n\tassert.Nil(t, cache.Get(\"A\"))\n}\n"
  },
  {
    "path": "common/cache/metricsScopeCache.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cache\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst flushBufferedMetricsScopeDuration = 10 * time.Second\n\ntype (\n\tmetricsScopeMap map[string]map[metrics.ScopeIdx]metrics.Scope\n\n\tbuffer struct {\n\t\tsync.RWMutex\n\t\tbufferMap metricsScopeMap\n\t}\n\n\tdomainMetricsScopeCache struct {\n\t\tstatus        int32\n\t\tbuffer        *buffer\n\t\tcache         atomic.Value\n\t\tcloseCh       chan struct{}\n\t\tflushDuration time.Duration\n\t\ttimeSource    clock.TimeSource\n\t}\n)\n\n// NewDomainMetricsScopeCache constructs a new domainMetricsScopeCache\nfunc NewDomainMetricsScopeCache() DomainMetricsScopeCache {\n\n\tmc := &domainMetricsScopeCache{\n\t\tbuffer: &buffer{\n\t\t\tbufferMap: make(metricsScopeMap),\n\t\t},\n\t\tcloseCh:       make(chan struct{}),\n\t\tflushDuration: flushBufferedMetricsScopeDuration,\n\t\ttimeSource:    clock.NewRealTimeSource(),\n\t}\n\n\tmc.cache.Store(make(metricsScopeMap))\n\treturn mc\n}\n\nfunc (c *domainMetricsScopeCache) flushBufferedMetricsScope(flushDuration time.Duration) {\n\tticker := c.timeSource.NewTicker(flushDuration)\n\tdefer ticker.Stop()\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tc.buffer.Lock()\n\t\t\tif len(c.buffer.bufferMap) > 0 {\n\t\t\t\tscopeMap := make(metricsScopeMap)\n\n\t\t\t\tdata := c.cache.Load().(metricsScopeMap)\n\t\t\t\t// Copy everything over after atomic load\n\t\t\t\tfor key, val := range data {\n\t\t\t\t\tscopeMap[key] = map[metrics.ScopeIdx]metrics.Scope{}\n\t\t\t\t\tfor k, v := range val {\n\t\t\t\t\t\tscopeMap[key][k] = v\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Copy from buffered array\n\t\t\t\tfor key, val := range c.buffer.bufferMap {\n\t\t\t\t\tif _, ok := scopeMap[key]; !ok {\n\t\t\t\t\t\tscopeMap[key] = map[metrics.ScopeIdx]metrics.Scope{}\n\t\t\t\t\t}\n\t\t\t\t\tfor k, v := range val {\n\t\t\t\t\t\tscopeMap[key][k] = v\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tc.cache.Store(scopeMap)\n\t\t\t\tc.buffer.bufferMap = make(metricsScopeMap)\n\t\t\t}\n\t\t\tc.buffer.Unlock()\n\n\t\tcase <-c.closeCh:\n\t\t\treturn\n\t\t}\n\t}\n}\n\n// Get retrieves scope for domainID and scopeIdx\nfunc (c *domainMetricsScopeCache) Get(domainID string, scopeIdx metrics.ScopeIdx) (metrics.Scope, bool) {\n\tdata := c.cache.Load().(metricsScopeMap)\n\n\tif data == nil {\n\t\treturn nil, false\n\t}\n\n\tm, ok := data[domainID]\n\tif !ok {\n\t\treturn nil, false\n\t}\n\tmetricsScope, ok := m[scopeIdx]\n\n\treturn metricsScope, ok\n}\n\n// Put puts map of domainID and scopeIdx to metricsScope\nfunc (c *domainMetricsScopeCache) Put(domainID string, scopeIdx metrics.ScopeIdx, scope metrics.Scope) {\n\tc.buffer.Lock()\n\tdefer c.buffer.Unlock()\n\n\tif c.buffer.bufferMap[domainID] == nil {\n\t\tc.buffer.bufferMap[domainID] = map[metrics.ScopeIdx]metrics.Scope{}\n\t}\n\tc.buffer.bufferMap[domainID][scopeIdx] = scope\n}\n\nfunc (c *domainMetricsScopeCache) Start() {\n\tif !atomic.CompareAndSwapInt32(&c.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tgo c.flushBufferedMetricsScope(c.flushDuration)\n}\n\nfunc (c *domainMetricsScopeCache) Stop() {\n\tif !atomic.CompareAndSwapInt32(&c.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(c.closeCh)\n}\n"
  },
  {
    "path": "common/cache/metricsScopeCache_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cache\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype domainMetricsCacheSuite struct {\n\tsuite.Suite\n\t*require.Assertions\n\n\tmetricsClient metrics.Client\n\tmetricsCache  DomainMetricsScopeCache\n}\n\nfunc TestDomainMetricsCacheSuite(t *testing.T) {\n\ts := new(domainMetricsCacheSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainMetricsCacheSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.Frontend, metrics.HistogramMigration{})\n\n\tmetricsCache := NewDomainMetricsScopeCache().(*domainMetricsScopeCache)\n\tmetricsCache.flushDuration = 100 * time.Millisecond\n\ts.metricsCache = metricsCache\n\n\ts.metricsCache.Start()\n}\n\nfunc (s *domainMetricsCacheSuite) TearDownTest() {\n\ts.metricsCache.Stop()\n}\n\nfunc (s *domainMetricsCacheSuite) TestGetMetricsScope() {\n\tvar found bool\n\n\ttests := []struct {\n\t\tscopeID  metrics.ScopeIdx\n\t\tdomainID string\n\t}{\n\t\t{1, \"A\"},\n\t\t{2, \"B\"},\n\t\t{1, \"C\"},\n\t}\n\n\tfor _, t := range tests {\n\t\tmockMetricsScope := s.metricsClient.Scope(t.scopeID)\n\t\ts.metricsCache.Put(t.domainID, t.scopeID, mockMetricsScope)\n\t}\n\n\ttime.Sleep(110 * time.Millisecond)\n\n\tmetricsScope, found := s.metricsCache.Get(\"A\", 1)\n\ttestMetricsScope := s.metricsClient.Scope(1)\n\ts.Equal(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n\n\t_, found = s.metricsCache.Get(\"B\", 2)\n\ts.Equal(found, true)\n\n\tmetricsScope, found = s.metricsCache.Get(\"C\", 1)\n\ttestMetricsScope = s.metricsClient.Scope(3)\n\ts.NotEqual(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n\n\tmetricsScope, found = s.metricsCache.Get(\"D\", 3)\n\ttestMetricsScope = s.metricsClient.Scope(3)\n\ts.NotEqual(testMetricsScope, metricsScope)\n\ts.Equal(found, false)\n}\n\nfunc (s *domainMetricsCacheSuite) TestGetMetricsScopeMultipleFlushLoop() {\n\tvar found bool\n\n\ttests := []struct {\n\t\tscopeID  metrics.ScopeIdx\n\t\tdomainID string\n\t}{\n\t\t{1, \"A\"},\n\t\t{2, \"B\"},\n\t\t{1, \"C\"},\n\t\t{5, \"D\"},\n\t\t{3, \"E\"},\n\t}\n\n\tfor i := 0; i < 3; i++ {\n\t\tt := tests[i]\n\t\tmockMetricsScope := s.metricsClient.Scope(t.scopeID)\n\t\ts.metricsCache.Put(t.domainID, t.scopeID, mockMetricsScope)\n\t}\n\n\ttime.Sleep(110 * time.Millisecond)\n\n\tfor i := 3; i < len(tests); i++ {\n\t\tt := tests[i]\n\t\tmockMetricsScope := s.metricsClient.Scope(t.scopeID)\n\t\ts.metricsCache.Put(t.domainID, t.scopeID, mockMetricsScope)\n\t}\n\n\tmetricsScope, found := s.metricsCache.Get(\"A\", 1)\n\ttestMetricsScope := s.metricsClient.Scope(1)\n\ts.Equal(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n\n\t_, found = s.metricsCache.Get(\"B\", 2)\n\ts.Equal(found, true)\n\n\tmetricsScope, found = s.metricsCache.Get(\"C\", 1)\n\ttestMetricsScope = s.metricsClient.Scope(3)\n\ts.NotEqual(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n\n\tmetricsScope, found = s.metricsCache.Get(\"D\", 5)\n\ttestMetricsScope = s.metricsClient.Scope(5)\n\ts.NotEqual(testMetricsScope, metricsScope)\n\ts.Equal(found, false)\n\n\tmetricsScope, found = s.metricsCache.Get(\"E\", 3)\n\ttestMetricsScope = s.metricsClient.Scope(3)\n\ts.NotEqual(testMetricsScope, metricsScope)\n\ts.Equal(found, false)\n\n\ttime.Sleep(200 * time.Millisecond)\n\n\tmetricsScope, found = s.metricsCache.Get(\"D\", 5)\n\ttestMetricsScope = s.metricsClient.Scope(5)\n\ts.Equal(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n\n\tmetricsScope, found = s.metricsCache.Get(\"E\", 3)\n\ttestMetricsScope = s.metricsClient.Scope(3)\n\ts.Equal(testMetricsScope, metricsScope)\n\ts.Equal(found, true)\n}\n\nfunc (s *domainMetricsCacheSuite) TestConcurrentMetricsScopeAccess() {\n\n\tch := make(chan struct{})\n\tvar wg sync.WaitGroup\n\tvar metricsScope, testMetricsScope metrics.Scope\n\tvar found bool\n\n\tfor i := 0; i < 1000; i++ {\n\t\twg.Add(1)\n\t\t// concurrent get and put\n\t\tgo func(scopeIdx metrics.ScopeIdx) {\n\t\t\tdefer wg.Done()\n\n\t\t\t<-ch\n\n\t\t\ts.metricsCache.Get(\"test_domain\", scopeIdx)\n\t\t\ts.metricsCache.Put(\"test_domain\", scopeIdx, s.metricsClient.Scope(metrics.ScopeIdx(int(scopeIdx)%int(metrics.NumServices))))\n\t\t}(metrics.ScopeIdx(i))\n\t}\n\n\tclose(ch)\n\twg.Wait()\n\n\ttime.Sleep(120 * time.Millisecond)\n\n\tfor i := 0; i < 1000; i++ {\n\t\tmetricsScope, found = s.metricsCache.Get(\"test_domain\", metrics.ScopeIdx(i))\n\t\ttestMetricsScope = s.metricsClient.Scope(metrics.ScopeIdx(i % int(metrics.NumServices)))\n\n\t\ts.Equal(true, found)\n\t\ts.Equal(testMetricsScope, metricsScope)\n\t}\n}\n"
  },
  {
    "path": "common/cache/simple.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cache\n\nimport (\n\t\"container/list\"\n\t\"sync\"\n\t\"time\"\n)\n\nvar (\n\t// DummyCreateTime is the create time used by all entries in the cache.\n\tDummyCreateTime = time.Time{}\n)\n\ntype (\n\tsimple struct {\n\t\tsync.RWMutex\n\t\taccessMap   map[interface{}]*list.Element\n\t\titerateList *list.List\n\t\trmFunc      RemovedFunc\n\t}\n\n\tsimpleItr struct {\n\t\tsimple   *simple\n\t\tnextItem *list.Element\n\t}\n\n\tsimpleEntry struct {\n\t\tkey   interface{}\n\t\tvalue interface{}\n\t}\n)\n\n// Close closes the iterator\nfunc (it *simpleItr) Close() {\n\tit.simple.RUnlock()\n}\n\n// HasNext return true if there is more items to be returned\nfunc (it *simpleItr) HasNext() bool {\n\treturn it.nextItem != nil\n}\n\n// Next returns the next item\nfunc (it *simpleItr) Next() Entry {\n\tif it.nextItem == nil {\n\t\tpanic(\"Simple cache iterator Next called when there is no next item\")\n\t}\n\n\tentry := it.nextItem.Value.(*simpleEntry)\n\tit.nextItem = it.nextItem.Next()\n\t// make a copy of the entry so there will be no concurrent access to this entry\n\tentry = &simpleEntry{\n\t\tkey:   entry.key,\n\t\tvalue: entry.value,\n\t}\n\treturn entry\n}\n\nfunc (e *simpleEntry) Key() interface{} {\n\treturn e.key\n}\n\nfunc (e *simpleEntry) Value() interface{} {\n\treturn e.value\n}\n\n// CreateTime is not implemented for simple cache entries\nfunc (e *simpleEntry) CreateTime() time.Time {\n\treturn DummyCreateTime\n}\n\n// NewSimple creates a new simple cache with given options.\n// Simple cache will never evict entries and it will never reorder the elements.\n// Simple cache also does not have the concept of pinning that LRU cache has.\n// Internally simple cache uses a RWMutex instead of the exclusive Mutex that LRU cache uses.\n// The RWMutex makes simple cache readable by many threads without introducing lock contention.\nfunc NewSimple(opts *SimpleOptions) Cache {\n\tif opts == nil {\n\t\topts = &SimpleOptions{}\n\t}\n\treturn &simple{\n\t\titerateList: list.New(),\n\t\taccessMap:   make(map[interface{}]*list.Element, opts.InitialCapacity),\n\t\trmFunc:      opts.RemovedFunc,\n\t}\n}\n\n// Get retrieves the value stored under the given key\nfunc (c *simple) Get(key interface{}) interface{} {\n\tc.RLock()\n\tdefer c.RUnlock()\n\n\telement := c.accessMap[key]\n\tif element == nil {\n\t\treturn nil\n\t}\n\treturn element.Value.(*simpleEntry).Value()\n}\n\n// Put puts a new value associated with a given key, returning the existing value (if present).\nfunc (c *simple) Put(key interface{}, value interface{}) interface{} {\n\tc.Lock()\n\tdefer c.Unlock()\n\texisting := c.putInternal(key, value, true)\n\treturn existing\n}\n\n// PutIfNotExist puts a value associated with a given key if it does not exist\nfunc (c *simple) PutIfNotExist(key interface{}, value interface{}) (interface{}, error) {\n\tc.Lock()\n\tdefer c.Unlock()\n\texisting := c.putInternal(key, value, false)\n\tif existing == nil {\n\t\t// This is a new value\n\t\treturn value, nil\n\t}\n\treturn existing, nil\n}\n\n// Delete deletes a key, value pair associated with a key\nfunc (c *simple) Delete(key interface{}) {\n\tc.Lock()\n\tdefer c.Unlock()\n\n\telement := c.accessMap[key]\n\tif element == nil {\n\t\treturn\n\t}\n\tentry := c.iterateList.Remove(element).(*simpleEntry)\n\tif c.rmFunc != nil {\n\t\tgo c.rmFunc(entry.value)\n\t}\n\tdelete(c.accessMap, entry.key)\n}\n\n// Release does nothing for simple cache\nfunc (c *simple) Release(_ interface{}) {}\n\n// Size returns the number of entries currently in the cache\nfunc (c *simple) Size() int {\n\tc.RLock()\n\tdefer c.RUnlock()\n\n\treturn len(c.accessMap)\n}\n\nfunc (c *simple) Iterator() Iterator {\n\tc.RLock()\n\titerator := &simpleItr{\n\t\tsimple:   c,\n\t\tnextItem: c.iterateList.Front(),\n\t}\n\treturn iterator\n}\n\nfunc (c *simple) putInternal(key interface{}, value interface{}, allowUpdate bool) interface{} {\n\telt := c.accessMap[key]\n\tif elt != nil {\n\t\tentry := elt.Value.(*simpleEntry)\n\t\texisting := entry.value\n\t\tif allowUpdate {\n\t\t\tentry.value = value\n\t\t}\n\t\treturn existing\n\t}\n\tentry := &simpleEntry{\n\t\tkey:   key,\n\t\tvalue: value,\n\t}\n\tc.accessMap[key] = c.iterateList.PushFront(entry)\n\treturn nil\n}\n"
  },
  {
    "path": "common/cache/simple_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cache\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestSimple(t *testing.T) {\n\tcache := NewSimple(nil)\n\n\tcache.Put(\"A\", \"Foo\")\n\tassert.Equal(t, \"Foo\", cache.Get(\"A\"))\n\tassert.Nil(t, cache.Get(\"B\"))\n\tassert.Equal(t, 1, cache.Size())\n\n\tcache.Put(\"B\", \"Bar\")\n\tcache.Put(\"C\", \"Cid\")\n\tcache.Put(\"D\", \"Delt\")\n\tassert.Equal(t, 4, cache.Size())\n\n\tassert.Equal(t, \"Bar\", cache.Get(\"B\"))\n\tassert.Equal(t, \"Cid\", cache.Get(\"C\"))\n\tassert.Equal(t, \"Delt\", cache.Get(\"D\"))\n\n\tcache.Put(\"A\", \"Foo2\")\n\tassert.Equal(t, \"Foo2\", cache.Get(\"A\"))\n\n\tcache.Put(\"E\", \"Epsi\")\n\tassert.Equal(t, \"Epsi\", cache.Get(\"E\"))\n\tassert.Equal(t, \"Foo2\", cache.Get(\"A\"))\n\n\tcache.Delete(\"A\")\n\tassert.Nil(t, cache.Get(\"A\"))\n}\n\nfunc TestSimpleGenerics(t *testing.T) {\n\tkey := keyType{\n\t\tdummyString: \"some random key\",\n\t\tdummyInt:    59,\n\t}\n\tvalue := \"some random value\"\n\n\tcache := NewSimple(nil)\n\tcache.Put(key, value)\n\n\tassert.Equal(t, value, cache.Get(key))\n\tassert.Equal(t, value, cache.Get(keyType{\n\t\tdummyString: \"some random key\",\n\t\tdummyInt:    59,\n\t}))\n\tassert.Nil(t, cache.Get(keyType{\n\t\tdummyString: \"some other random key\",\n\t\tdummyInt:    56,\n\t}))\n}\n\nfunc TestSimpleCacheConcurrentAccess(t *testing.T) {\n\tcache := NewSimple(nil)\n\tvalues := map[string]string{\n\t\t\"A\": \"foo\",\n\t\t\"B\": \"bar\",\n\t\t\"C\": \"zed\",\n\t\t\"D\": \"dank\",\n\t\t\"E\": \"ezpz\",\n\t}\n\n\tfor k, v := range values {\n\t\tcache.Put(k, v)\n\t}\n\n\tstart := make(chan struct{})\n\tvar wg sync.WaitGroup\n\tfor i := 0; i < 20; i++ {\n\t\twg.Add(2)\n\n\t\t// concurrent get and put\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\n\t\t\t<-start\n\n\t\t\tfor j := 0; j < 1000; j++ {\n\t\t\t\tcache.Get(\"A\")\n\t\t\t\tcache.Put(\"A\", \"fooo\")\n\t\t\t}\n\t\t}()\n\n\t\t// concurrent iteration\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\n\t\t\t<-start\n\n\t\t\tfor j := 0; j < 50; j++ {\n\t\t\t\tvar result []Entry\n\t\t\t\tit := cache.Iterator()\n\t\t\t\tfor it.HasNext() {\n\t\t\t\t\tentry := it.Next()\n\t\t\t\t\tresult = append(result, entry) //nolint:staticcheck\n\t\t\t\t}\n\t\t\t\tit.Close()\n\t\t\t}\n\t\t}()\n\t}\n\n\tclose(start)\n\twg.Wait()\n}\n\nfunc TestSimpleRemoveFunc(t *testing.T) {\n\tch := make(chan bool)\n\tcache := NewSimple(&SimpleOptions{\n\t\tRemovedFunc: func(i interface{}) {\n\t\t\t_, ok := i.(*testing.T)\n\t\t\tassert.True(t, ok)\n\t\t\tch <- true\n\t\t},\n\t})\n\n\tcache.Put(\"testing\", t)\n\tcache.Delete(\"testing\")\n\tassert.Nil(t, cache.Get(\"testing\"))\n\n\ttimeout := time.NewTimer(time.Millisecond * 300)\n\tselect {\n\tcase b := <-ch:\n\t\tassert.True(t, b)\n\tcase <-timeout.C:\n\t\tt.Error(\"RemovedFunc did not send true on channel ch\")\n\t}\n}\n\nfunc TestSimpleIterator(t *testing.T) {\n\texpected := map[string]string{\n\t\t\"A\": \"Alpha\",\n\t\t\"B\": \"Beta\",\n\t\t\"G\": \"Gamma\",\n\t\t\"D\": \"Delta\",\n\t}\n\n\tcache := NewSimple(nil)\n\n\tfor k, v := range expected {\n\t\tcache.Put(k, v)\n\t}\n\n\tactual := map[string]string{}\n\n\tit := cache.Iterator()\n\tfor it.HasNext() {\n\t\tentry := it.Next()\n\t\tactual[entry.Key().(string)] = entry.Value().(string)\n\t}\n\tit.Close()\n\tassert.Equal(t, expected, actual)\n\n\tit = cache.Iterator()\n\tfor i := 0; i < len(expected); i++ {\n\t\tentry := it.Next()\n\t\tactual[entry.Key().(string)] = entry.Value().(string)\n\t}\n\tit.Close()\n\tassert.Equal(t, expected, actual)\n}\n"
  },
  {
    "path": "common/checksum/crc.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage checksum\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"fmt\"\n\t\"hash/crc32\"\n\n\t\"github.com/uber/cadence/common/codec\"\n)\n\n// GenerateCRC32 generates an IEEE crc32 checksum on the\n// serilized byte array of the given thrift object. The\n// serialization proto used will be of type thriftRW\nfunc GenerateCRC32(\n\tpayload codec.ThriftObject,\n\tpayloadVersion int,\n) (Checksum, error) {\n\n\tencoder := codec.NewThriftRWEncoder()\n\tpayloadBytes, err := encoder.Encode(payload)\n\tif err != nil {\n\t\treturn Checksum{}, err\n\t}\n\n\tcrc := crc32.ChecksumIEEE(payloadBytes)\n\tchecksum := make([]byte, 4)\n\tbinary.BigEndian.PutUint32(checksum, crc)\n\treturn Checksum{\n\t\tValue:   checksum,\n\t\tVersion: payloadVersion,\n\t\tFlavor:  FlavorIEEECRC32OverThriftBinary,\n\t}, nil\n}\n\n// Verify verifies that the checksum generated from the\n// given thrift object matches the specified expected checksum\n// Return ErrMismatch when checksums mismatch\nfunc Verify(\n\tpayload codec.ThriftObject,\n\tchecksum Checksum,\n) error {\n\n\tif !checksum.Flavor.IsValid() || checksum.Flavor != FlavorIEEECRC32OverThriftBinary {\n\t\treturn fmt.Errorf(\"unknown checksum flavor %v\", checksum.Flavor)\n\t}\n\n\texpected, err := GenerateCRC32(payload, checksum.Version)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !bytes.Equal(expected.Value, checksum.Value) {\n\t\treturn ErrMismatch\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/checksum/crc_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage checksum\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n)\n\nfunc TestCRC32OverThrift(t *testing.T) {\n\t// note: do not use a struct with map since\n\t// iteration order is not guaranteed in Go and\n\t// so, each call to thrift encode will result in\n\t// different set of serialized bytes\n\tobj := &shared.WorkflowExecutionInfo{\n\t\tExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: common.StringPtr(uuid.New()),\n\t\t\tRunId:      common.StringPtr(uuid.New()),\n\t\t},\n\t\tStartTime:     common.Int64Ptr(time.Now().UnixNano()),\n\t\tHistoryLength: common.Int64Ptr(550),\n\t}\n\n\tparallism := 10\n\tloopCount := 100\n\tsuccessCount := int64(0)\n\n\tstartC := make(chan struct{})\n\tdoneWG := sync.WaitGroup{}\n\tdoneWG.Add(parallism)\n\n\tfor i := 0; i < parallism; i++ {\n\t\tgo func() {\n\t\t\tdefer doneWG.Done()\n\t\t\t<-startC\n\t\t\tfor count := 0; count < loopCount; count++ {\n\t\t\t\tcsum, err := GenerateCRC32(obj, 1)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tif err := Verify(obj, csum); err != nil {\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tatomic.AddInt64(&successCount, 1)\n\t\t\t}\n\t\t}()\n\t}\n\n\tclose(startC)\n\tsuccess := common.AwaitWaitGroup(&doneWG, time.Second)\n\tassert.True(t, success, \"timed out waiting for goroutines to finish\")\n\tassert.Equal(t, int64(parallism*loopCount), successCount)\n}\n"
  },
  {
    "path": "common/checksum/ctc_benchmark_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage checksum\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n)\n\nfunc BenchmarkGenerateCRC32(b *testing.B) {\n\tobj := &shared.WorkflowExecutionInfo{\n\t\tExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: common.StringPtr(uuid.New()),\n\t\t\tRunId:      common.StringPtr(uuid.New()),\n\t\t},\n\t\tStartTime:     common.Int64Ptr(time.Now().UnixNano()),\n\t\tHistoryLength: common.Int64Ptr(550),\n\t}\n\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\tchecksum, _ := GenerateCRC32(obj, 1)\n\t\t_ = Verify(obj, checksum)\n\t}\n}\n"
  },
  {
    "path": "common/checksum/defs.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage checksum\n\nimport \"errors\"\n\ntype (\n\t// Checksum represents a checksum value along\n\t// with associated metadata\n\tChecksum struct {\n\t\t// Version represents version of the payload from\n\t\tVersion int\n\t\t// which this checksum was derived\n\t\tFlavor Flavor\n\t\t// Value is the checksum value\n\t\tValue []byte\n\t}\n\n\t// Flavor is an enum type that represents the type of checksum\n\tFlavor int\n)\n\nconst (\n\t// FlavorUnknown represents an unknown/uninitialized checksum flavor\n\tFlavorUnknown Flavor = iota\n\t// FlavorIEEECRC32OverThriftBinary represents crc32 checksum generated over thriftRW serialized payload\n\tFlavorIEEECRC32OverThriftBinary\n\tmaxFlavors\n)\n\n// ErrMismatch indicates a checksum verification failure due to\n// a derived checksum not being equal to expected checksum\nvar ErrMismatch = errors.New(\"checksum mismatch error\")\n\n// IsValid returns true if the checksum flavor is valid\nfunc (f Flavor) IsValid() bool {\n\treturn f > FlavorUnknown && f < maxFlavors\n}\n"
  },
  {
    "path": "common/client/versionChecker.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination versionChecker_mock.go -self_package github.com/uber/cadence/common/client github.com/uber/cadence/common/client VersionChecker\n\npackage client\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/gogo/protobuf/jsonpb\"\n\t\"github.com/hashicorp/go-version\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// GoSDK is the header value for common.ClientImplHeaderName indicating a go sdk client\n\tGoSDK = \"uber-go\"\n\t// JavaSDK is the header value for common.ClientImplHeaderName indicating a java sdk client\n\tJavaSDK = \"uber-java\"\n\t// CLI is the header value for common.ClientImplHeaderName indicating a cli client\n\tCLI = \"cli\"\n\n\t// SupportedGoSDKVersion indicates the highest go sdk version server will accept requests from\n\tSupportedGoSDKVersion = \"1.7.0\"\n\t// SupportedJavaSDKVersion indicates the highest java sdk version server will accept requests from\n\tSupportedJavaSDKVersion = \"1.5.0\"\n\t// SupportedCLIVersion indicates the highest cli version server will accept requests from\n\tSupportedCLIVersion = \"1.7.0\"\n\n\t// StickyQueryUnknownImplConstraints indicates the minimum client version of an unknown client type which supports StickyQuery\n\tStickyQueryUnknownImplConstraints = \"1.0.0\"\n\t// GoWorkerStickyQueryVersion indicates the minimum client version of go worker which supports StickyQuery\n\tGoWorkerStickyQueryVersion = \"1.0.0\"\n\t// JavaWorkerStickyQueryVersion indicates the minimum client version of the java worker which supports StickyQuery\n\tJavaWorkerStickyQueryVersion = \"1.0.0\"\n\t// GoWorkerConsistentQueryVersion indicates the minimum client version of the go worker which supports ConsistentQuery\n\tGoWorkerConsistentQueryVersion = \"1.5.0\"\n\t// JavaWorkerRawHistoryQueryVersion indicates the minimum client version of the java worker which supports RawHistoryQuery\n\tJavaWorkerRawHistoryQueryVersion = \"1.3.0\"\n\t// JavaWorkerConsistentQueryVersion indicates the minimum client version of the java worker which supports ConsistentQuery\n\tJavaWorkerConsistentQueryVersion = \"1.5.0\"\n\t// GoWorkerRawHistoryQueryVersion indicates the minimum client version of the go worker which supports RawHistoryQuery\n\tGoWorkerRawHistoryQueryVersion = \"1.6.0\"\n\t// CLIRawHistoryQueryVersion indicates the minimum CLI version of the go worker which supports RawHistoryQuery\n\t// Note: cli uses go client feature version\n\tCLIRawHistoryQueryVersion = \"1.6.0\"\n\t// Go Client version that supports WorkflowExecutionAlreadyCompleted Error\n\tCLIWorkflowAlreadyCompletedVersion = \"1.7.0\"\n\t// Go Client version that supports WorkflowExecutionAlreadyCompleted Error\n\tGoWorkerWorkflowAlreadyCompletedVersion = \"1.7.0\"\n\t// Java Client version that supports WorkflowExecutionAlreadyCompleted Error\n\tJavaWorkflowAlreadyCompletedVersion = \"1.4.0\"\n\n\tstickyQuery                   = \"sticky-query\"\n\tconsistentQuery               = \"consistent-query\"\n\trawHistoryQuery               = \"send-raw-workflow-history\"\n\tworkflowAlreadyCompletedError = \"workflow-already-completed\"\n)\n\nvar (\n\t// ErrUnknownFeature indicates that requested feature is not known by version checker\n\tErrUnknownFeature = &types.BadRequestError{Message: \"Unknown feature\"}\n\n\t// DefaultCLIFeatureFlags is the default FeatureFlags used by Cadence CLI\n\tDefaultCLIFeatureFlags = apiv1.FeatureFlags{\n\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t}\n)\n\ntype (\n\t// VersionChecker is used to check client/server compatibility and client's capabilities\n\tVersionChecker interface {\n\t\tClientSupported(ctx context.Context, enableClientVersionCheck bool) error\n\n\t\tSupportsStickyQuery(clientImpl string, clientFeatureVersion string) error\n\t\tSupportsConsistentQuery(clientImpl string, clientFeatureVersion string) error\n\t\tSupportsRawHistoryQuery(clientImpl string, clientFeatureVersion string) error\n\t\tSupportsWorkflowAlreadyCompletedError(clientImpl string, clientFeatureVersion string, featureFlags apiv1.FeatureFlags) error\n\t}\n\n\tversionChecker struct {\n\t\tsupportedFeatures                 map[string]map[string]version.Constraints\n\t\tsupportedClients                  map[string]version.Constraints\n\t\tstickyQueryUnknownImplConstraints version.Constraints\n\t}\n)\n\n// FeatureFlagsHeader returns the serialized version of the FeatureFlags\nfunc FeatureFlagsHeader(featureFlags apiv1.FeatureFlags) string {\n\tvar marshaler jsonpb.Marshaler\n\tserialized, err := marshaler.MarshalToString(&featureFlags)\n\tif err != nil {\n\t\t// fail open and return empty feature flags\n\t\treturn \"\"\n\t}\n\treturn serialized\n}\n\n// GetFeatureFlagsFromHeader returns FeatureFlags from yarpc headers\nfunc GetFeatureFlagsFromHeader(call *yarpc.Call) apiv1.FeatureFlags {\n\tfeatureFlagsSerialized := call.Header(common.ClientFeatureFlagsHeaderName)\n\n\tvar featureFlags apiv1.FeatureFlags\n\terr := jsonpb.UnmarshalString(featureFlagsSerialized, &featureFlags)\n\tif err != nil {\n\t\t// fail open and return empty feature flags\n\t\treturn apiv1.FeatureFlags{}\n\t}\n\n\treturn featureFlags\n}\n\n// NewVersionChecker constructs a new VersionChecker\nfunc NewVersionChecker() VersionChecker {\n\tsupportedFeatures := map[string]map[string]version.Constraints{\n\t\tGoSDK: {\n\t\t\tstickyQuery:                   mustNewConstraint(fmt.Sprintf(\">=%v\", GoWorkerStickyQueryVersion)),\n\t\t\tconsistentQuery:               mustNewConstraint(fmt.Sprintf(\">=%v\", GoWorkerConsistentQueryVersion)),\n\t\t\trawHistoryQuery:               mustNewConstraint(fmt.Sprintf(\">=%v\", GoWorkerRawHistoryQueryVersion)),\n\t\t\tworkflowAlreadyCompletedError: mustNewConstraint(fmt.Sprintf(\">=%v\", GoWorkerWorkflowAlreadyCompletedVersion)),\n\t\t},\n\t\tJavaSDK: {\n\t\t\tstickyQuery:                   mustNewConstraint(fmt.Sprintf(\">=%v\", JavaWorkerStickyQueryVersion)),\n\t\t\tconsistentQuery:               mustNewConstraint(fmt.Sprintf(\">=%v\", JavaWorkerConsistentQueryVersion)),\n\t\t\trawHistoryQuery:               mustNewConstraint(fmt.Sprintf(\">=%v\", JavaWorkerRawHistoryQueryVersion)),\n\t\t\tworkflowAlreadyCompletedError: mustNewConstraint(fmt.Sprintf(\">=%v\", JavaWorkflowAlreadyCompletedVersion)),\n\t\t},\n\t\tCLI: {\n\t\t\trawHistoryQuery:               mustNewConstraint(fmt.Sprintf(\">=%v\", CLIRawHistoryQueryVersion)),\n\t\t\tworkflowAlreadyCompletedError: mustNewConstraint(fmt.Sprintf(\">=%v\", CLIWorkflowAlreadyCompletedVersion)),\n\t\t},\n\t}\n\tsupportedClients := map[string]version.Constraints{\n\t\tGoSDK:   mustNewConstraint(fmt.Sprintf(\"<=%v\", SupportedGoSDKVersion)),\n\t\tJavaSDK: mustNewConstraint(fmt.Sprintf(\"<=%v\", SupportedJavaSDKVersion)),\n\t\tCLI:     mustNewConstraint(fmt.Sprintf(\"<=%v\", SupportedCLIVersion)),\n\t}\n\treturn &versionChecker{\n\t\tsupportedFeatures:                 supportedFeatures,\n\t\tsupportedClients:                  supportedClients,\n\t\tstickyQueryUnknownImplConstraints: mustNewConstraint(fmt.Sprintf(\">=%v\", StickyQueryUnknownImplConstraints)),\n\t}\n}\n\n// ClientSupported returns an error if client is unsupported, nil otherwise.\n// In case client version lookup fails assume the client is supported.\nfunc (vc *versionChecker) ClientSupported(ctx context.Context, enableClientVersionCheck bool) error {\n\tif !enableClientVersionCheck {\n\t\treturn nil\n\t}\n\n\tcall := yarpc.CallFromContext(ctx)\n\tclientFeatureVersion := call.Header(common.FeatureVersionHeaderName)\n\tclientImpl := call.Header(common.ClientImplHeaderName)\n\n\tif clientFeatureVersion == \"\" {\n\t\treturn nil\n\t}\n\tsupportedVersions, ok := vc.supportedClients[clientImpl]\n\tif !ok {\n\t\treturn nil\n\t}\n\tversion, err := version.NewVersion(clientFeatureVersion)\n\tif err != nil || !supportedVersions.Check(version) {\n\t\treturn &types.ClientVersionNotSupportedError{FeatureVersion: clientFeatureVersion, ClientImpl: clientImpl, SupportedVersions: supportedVersions.String()}\n\t}\n\n\treturn nil\n}\n\n// SupportsStickyQuery returns error if sticky query is not supported otherwise nil.\n// In case client version lookup fails assume the client does not support feature.\nfunc (vc *versionChecker) SupportsStickyQuery(clientImpl string, clientFeatureVersion string) error {\n\treturn vc.featureSupported(clientImpl, clientFeatureVersion, stickyQuery)\n}\n\n// SupportsConsistentQuery returns error if consistent query is not supported otherwise nil.\n// In case client version lookup fails assume the client does not support feature.\nfunc (vc *versionChecker) SupportsConsistentQuery(clientImpl string, clientFeatureVersion string) error {\n\n\treturn vc.featureSupported(clientImpl, clientFeatureVersion, consistentQuery)\n}\n\n// SupportsRawHistoryQuery returns error if raw history query is not supported otherwise nil.\n// In case client version lookup fails assume the client does not support feature.\nfunc (vc *versionChecker) SupportsRawHistoryQuery(clientImpl string, clientFeatureVersion string) error {\n\treturn vc.featureSupported(clientImpl, clientFeatureVersion, rawHistoryQuery)\n}\n\n// Returns error if workflowAlreadyCompletedError is not supported otherwise nil.\n// In case client version lookup fails assume the client does not support feature.\n// NOTE: Enabling this error will break customer code handling the workflow errors in their workflow\nfunc (vc *versionChecker) SupportsWorkflowAlreadyCompletedError(clientImpl string, clientFeatureVersion string, featureFlags apiv1.FeatureFlags) error {\n\tif featureFlags.WorkflowExecutionAlreadyCompletedErrorEnabled {\n\t\treturn vc.featureSupported(clientImpl, clientFeatureVersion, workflowAlreadyCompletedError)\n\t}\n\treturn &shared.FeatureNotEnabledError{FeatureFlag: \"WorkflowExecutionAlreadyCompletedErrorEnabled\"}\n}\n\nfunc (vc *versionChecker) featureSupported(clientImpl string, clientFeatureVersion string, feature string) error {\n\t// Some older clients may not provide clientImpl.\n\t// If this is the case special handling needs to be done to maintain backwards compatibility.\n\t// This can be removed after it is sure there are no existing clients which do not provide clientImpl in RPC headers.\n\tif clientImpl == \"\" {\n\t\tswitch feature {\n\t\tcase consistentQuery:\n\t\tcase rawHistoryQuery, workflowAlreadyCompletedError:\n\t\t\treturn &types.ClientVersionNotSupportedError{FeatureVersion: clientFeatureVersion}\n\t\tcase stickyQuery:\n\t\t\tversion, err := version.NewVersion(clientFeatureVersion)\n\t\t\tif err != nil {\n\t\t\t\treturn &types.ClientVersionNotSupportedError{FeatureVersion: clientFeatureVersion}\n\t\t\t}\n\t\t\tif !vc.stickyQueryUnknownImplConstraints.Check(version) {\n\t\t\t\treturn &types.ClientVersionNotSupportedError{FeatureVersion: clientFeatureVersion, SupportedVersions: vc.stickyQueryUnknownImplConstraints.String()}\n\t\t\t}\n\t\t\treturn nil\n\t\tdefault:\n\t\t\treturn ErrUnknownFeature\n\t\t}\n\t}\n\tif clientFeatureVersion == \"\" {\n\t\treturn &types.ClientVersionNotSupportedError{ClientImpl: clientImpl, FeatureVersion: clientFeatureVersion}\n\t}\n\timplMap, ok := vc.supportedFeatures[clientImpl]\n\tif !ok {\n\t\treturn &types.ClientVersionNotSupportedError{ClientImpl: clientImpl, FeatureVersion: clientFeatureVersion}\n\t}\n\tsupportedVersions, ok := implMap[feature]\n\tif !ok {\n\t\treturn &types.ClientVersionNotSupportedError{ClientImpl: clientImpl, FeatureVersion: clientFeatureVersion}\n\t}\n\tversion, err := version.NewVersion(clientFeatureVersion)\n\tif err != nil {\n\t\treturn &types.ClientVersionNotSupportedError{FeatureVersion: clientFeatureVersion, ClientImpl: clientImpl, SupportedVersions: supportedVersions.String()}\n\t}\n\tif !supportedVersions.Check(version) {\n\t\treturn &types.ClientVersionNotSupportedError{ClientImpl: clientImpl, FeatureVersion: clientFeatureVersion, SupportedVersions: supportedVersions.String()}\n\t}\n\treturn nil\n}\n\nfunc mustNewConstraint(v string) version.Constraints {\n\tconstraint, err := version.NewConstraint(v)\n\tif err != nil {\n\t\tpanic(\"invalid version constraint \" + v)\n\t}\n\treturn constraint\n}\n"
  },
  {
    "path": "common/client/versionChecker_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/client (interfaces: VersionChecker)\n//\n// Generated by this command:\n//\n//\tmockgen -package client -destination versionChecker_mock.go -self_package github.com/uber/cadence/common/client github.com/uber/cadence/common/client VersionChecker\n//\n\n// Package client is a generated GoMock package.\npackage client\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockVersionChecker is a mock of VersionChecker interface.\ntype MockVersionChecker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVersionCheckerMockRecorder\n\tisgomock struct{}\n}\n\n// MockVersionCheckerMockRecorder is the mock recorder for MockVersionChecker.\ntype MockVersionCheckerMockRecorder struct {\n\tmock *MockVersionChecker\n}\n\n// NewMockVersionChecker creates a new mock instance.\nfunc NewMockVersionChecker(ctrl *gomock.Controller) *MockVersionChecker {\n\tmock := &MockVersionChecker{ctrl: ctrl}\n\tmock.recorder = &MockVersionCheckerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVersionChecker) EXPECT() *MockVersionCheckerMockRecorder {\n\treturn m.recorder\n}\n\n// ClientSupported mocks base method.\nfunc (m *MockVersionChecker) ClientSupported(ctx context.Context, enableClientVersionCheck bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ClientSupported\", ctx, enableClientVersionCheck)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ClientSupported indicates an expected call of ClientSupported.\nfunc (mr *MockVersionCheckerMockRecorder) ClientSupported(ctx, enableClientVersionCheck any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ClientSupported\", reflect.TypeOf((*MockVersionChecker)(nil).ClientSupported), ctx, enableClientVersionCheck)\n}\n\n// SupportsConsistentQuery mocks base method.\nfunc (m *MockVersionChecker) SupportsConsistentQuery(clientImpl, clientFeatureVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsConsistentQuery\", clientImpl, clientFeatureVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SupportsConsistentQuery indicates an expected call of SupportsConsistentQuery.\nfunc (mr *MockVersionCheckerMockRecorder) SupportsConsistentQuery(clientImpl, clientFeatureVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsConsistentQuery\", reflect.TypeOf((*MockVersionChecker)(nil).SupportsConsistentQuery), clientImpl, clientFeatureVersion)\n}\n\n// SupportsRawHistoryQuery mocks base method.\nfunc (m *MockVersionChecker) SupportsRawHistoryQuery(clientImpl, clientFeatureVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsRawHistoryQuery\", clientImpl, clientFeatureVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SupportsRawHistoryQuery indicates an expected call of SupportsRawHistoryQuery.\nfunc (mr *MockVersionCheckerMockRecorder) SupportsRawHistoryQuery(clientImpl, clientFeatureVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsRawHistoryQuery\", reflect.TypeOf((*MockVersionChecker)(nil).SupportsRawHistoryQuery), clientImpl, clientFeatureVersion)\n}\n\n// SupportsStickyQuery mocks base method.\nfunc (m *MockVersionChecker) SupportsStickyQuery(clientImpl, clientFeatureVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsStickyQuery\", clientImpl, clientFeatureVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SupportsStickyQuery indicates an expected call of SupportsStickyQuery.\nfunc (mr *MockVersionCheckerMockRecorder) SupportsStickyQuery(clientImpl, clientFeatureVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsStickyQuery\", reflect.TypeOf((*MockVersionChecker)(nil).SupportsStickyQuery), clientImpl, clientFeatureVersion)\n}\n\n// SupportsWorkflowAlreadyCompletedError mocks base method.\nfunc (m *MockVersionChecker) SupportsWorkflowAlreadyCompletedError(clientImpl, clientFeatureVersion string, featureFlags apiv1.FeatureFlags) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsWorkflowAlreadyCompletedError\", clientImpl, clientFeatureVersion, featureFlags)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SupportsWorkflowAlreadyCompletedError indicates an expected call of SupportsWorkflowAlreadyCompletedError.\nfunc (mr *MockVersionCheckerMockRecorder) SupportsWorkflowAlreadyCompletedError(clientImpl, clientFeatureVersion, featureFlags any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsWorkflowAlreadyCompletedError\", reflect.TypeOf((*MockVersionChecker)(nil).SupportsWorkflowAlreadyCompletedError), clientImpl, clientFeatureVersion, featureFlags)\n}\n"
  },
  {
    "path": "common/client/versionChecker_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage client\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/encoding\"\n\t\"go.uber.org/yarpc/api/transport\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tVersionCheckerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestVersionCheckerSuite(t *testing.T) {\n\tsuite.Run(t, new(VersionCheckerSuite))\n}\n\nfunc (s *VersionCheckerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *VersionCheckerSuite) TestClientSupported() {\n\ttestCases := []struct {\n\t\tcallContext              context.Context\n\t\tenableClientVersionCheck bool\n\t\texpectErr                bool\n\t}{\n\t\t{\n\t\t\tenableClientVersionCheck: false,\n\t\t\texpectErr:                false,\n\t\t},\n\t\t{\n\t\t\tcallContext:              context.Background(),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                false,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(\"unknown-client\", \"0.0.0\"),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                false,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(GoSDK, \"malformed-version\"),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                true,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(GoSDK, s.getHigherVersion(SupportedGoSDKVersion)),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                true,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(JavaSDK, s.getHigherVersion(SupportedJavaSDKVersion)),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                true,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(CLI, s.getHigherVersion(SupportedCLIVersion)),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                true,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(GoSDK, SupportedGoSDKVersion),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                false,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(JavaSDK, SupportedJavaSDKVersion),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                false,\n\t\t},\n\t\t{\n\t\t\tcallContext:              s.constructCallContext(CLI, SupportedCLIVersion),\n\t\t\tenableClientVersionCheck: true,\n\t\t\texpectErr:                false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tversionChecker := NewVersionChecker()\n\t\terr := versionChecker.ClientSupported(tc.callContext, tc.enableClientVersionCheck)\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t\ts.IsType(&types.ClientVersionNotSupportedError{}, err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t}\n}\n\nfunc (s *VersionCheckerSuite) TestSupportsStickyQuery() {\n\ttestCases := []struct {\n\t\tclientImpl           string\n\t\tclientFeatureVersion string\n\t\texpectErr            bool\n\t}{\n\t\t{\n\t\t\tclientImpl: \"\",\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"\",\n\t\t\tclientFeatureVersion: \"0.9.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"\",\n\t\t\tclientFeatureVersion: \"1.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl: GoSDK,\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"unknown\",\n\t\t\tclientFeatureVersion: \"0.0.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"malformed-feature-version\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: GoWorkerStickyQueryVersion,\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: JavaWorkerStickyQueryVersion,\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"0.9.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"0.9.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"2.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"2.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvc := NewVersionChecker()\n\t\tif tc.expectErr {\n\t\t\terr := vc.SupportsStickyQuery(tc.clientImpl, tc.clientFeatureVersion)\n\t\t\ts.Error(err)\n\t\t\ts.IsType(&types.ClientVersionNotSupportedError{}, err)\n\t\t} else {\n\t\t\ts.NoError(vc.SupportsStickyQuery(tc.clientImpl, tc.clientFeatureVersion))\n\t\t}\n\t}\n}\n\nfunc (s *VersionCheckerSuite) TestSupportsConsistentQuery() {\n\ttestCases := []struct {\n\t\tclientImpl           string\n\t\tclientFeatureVersion string\n\t\texpectErr            bool\n\t}{\n\t\t{\n\t\t\tclientImpl: \"\",\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"\",\n\t\t\tclientFeatureVersion: GoWorkerConsistentQueryVersion,\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl: GoSDK,\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"unknown\",\n\t\t\tclientFeatureVersion: \"0.0.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.4.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"malformed-feature-version\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: GoWorkerConsistentQueryVersion,\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"1.4.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"2.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvc := NewVersionChecker()\n\t\tif tc.expectErr {\n\t\t\terr := vc.SupportsConsistentQuery(tc.clientImpl, tc.clientFeatureVersion)\n\t\t\ts.Error(err)\n\t\t\ts.IsType(&types.ClientVersionNotSupportedError{}, err)\n\t\t} else {\n\t\t\ts.NoError(vc.SupportsConsistentQuery(tc.clientImpl, tc.clientFeatureVersion))\n\t\t}\n\t}\n}\n\nfunc (s *VersionCheckerSuite) TestSupportsRawHistoryQuery() {\n\ttestCases := []struct {\n\t\tclientImpl           string\n\t\tclientFeatureVersion string\n\t\texpectErr            bool\n\t}{\n\t\t{\n\t\t\tclientImpl: \"\",\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"\",\n\t\t\tclientFeatureVersion: GoWorkerRawHistoryQueryVersion,\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl: GoSDK,\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"unknown\",\n\t\t\tclientFeatureVersion: \"0.0.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.5.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.7.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.2.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"malformed-feature-version\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: GoWorkerRawHistoryQueryVersion,\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"1.4.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"2.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           CLI,\n\t\t\tclientFeatureVersion: \"1.5.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           CLI,\n\t\t\tclientFeatureVersion: \"1.7.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvc := NewVersionChecker()\n\t\tif tc.expectErr {\n\t\t\terr := vc.SupportsRawHistoryQuery(tc.clientImpl, tc.clientFeatureVersion)\n\t\t\ts.Error(err)\n\t\t\ts.IsType(&types.ClientVersionNotSupportedError{}, err)\n\t\t} else {\n\t\t\ts.NoError(vc.SupportsRawHistoryQuery(tc.clientImpl, tc.clientFeatureVersion))\n\t\t}\n\t}\n}\n\nfunc (s *VersionCheckerSuite) TestSupportsWorkflowAlreadyCompletedError() {\n\ttestCases := []struct {\n\t\tclientImpl           string\n\t\tclientFeatureVersion string\n\t\texpectErr            bool\n\t}{\n\t\t{\n\t\t\tclientImpl: \"\",\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"\",\n\t\t\tclientFeatureVersion: GoWorkerWorkflowAlreadyCompletedVersion,\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl: GoSDK,\n\t\t\texpectErr:  true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           \"unknown\",\n\t\t\tclientFeatureVersion: \"0.0.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.4.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           JavaSDK,\n\t\t\tclientFeatureVersion: \"1.3.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"malformed-feature-version\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: GoWorkerWorkflowAlreadyCompletedVersion,\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"1.4.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           GoSDK,\n\t\t\tclientFeatureVersion: \"2.0.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           CLI,\n\t\t\tclientFeatureVersion: \"1.5.0\",\n\t\t\texpectErr:            true,\n\t\t},\n\t\t{\n\t\t\tclientImpl:           CLI,\n\t\t\tclientFeatureVersion: \"1.7.0\",\n\t\t\texpectErr:            false,\n\t\t},\n\t}\n\n\tfeatureFlags := apiv1.FeatureFlags{\n\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvc := NewVersionChecker()\n\t\tif tc.expectErr {\n\t\t\terr := vc.SupportsWorkflowAlreadyCompletedError(tc.clientImpl, tc.clientFeatureVersion, featureFlags)\n\t\t\ts.Error(err)\n\t\t\ts.IsType(&types.ClientVersionNotSupportedError{}, err)\n\t\t} else {\n\t\t\ts.NoError(vc.SupportsWorkflowAlreadyCompletedError(tc.clientImpl, tc.clientFeatureVersion, featureFlags))\n\t\t}\n\t}\n}\n\nfunc (s *VersionCheckerSuite) getHigherVersion(version string) string {\n\tsplit := strings.Split(version, \".\")\n\ts.Len(split, 3)\n\treturn fmt.Sprintf(\"%v.%v.%v\", split[0], split[1], split[2][0]-'0'+1)\n}\n\nfunc (s *VersionCheckerSuite) constructCallContext(clientImpl string, featureVersion string) context.Context {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.ClientImplHeaderName, clientImpl).With(common.FeatureVersionHeaderName, featureVersion),\n\t})\n\ts.NoError(err)\n\treturn ctx\n}\n\nfunc TestGetFeatureFlagsFromHeader(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tfeatureFlagsValue string // use \"\" to omit header\n\t\twant              apiv1.FeatureFlags\n\t}{\n\t\t{\n\t\t\tname:              \"no header\",\n\t\t\tfeatureFlagsValue: \"\",\n\t\t\twant:              apiv1.FeatureFlags{},\n\t\t},\n\t\t{\n\t\t\tname:              \"valid JSON with WorkflowExecutionAlreadyCompletedErrorEnabled true\",\n\t\t\tfeatureFlagsValue: `{\"WorkflowExecutionAlreadyCompletedErrorEnabled\":true}`,\n\t\t\twant: apiv1.FeatureFlags{\n\t\t\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"valid JSON with WorkflowExecutionAlreadyCompletedErrorEnabled false\",\n\t\t\tfeatureFlagsValue: `{\"WorkflowExecutionAlreadyCompletedErrorEnabled\":false}`,\n\t\t\twant: apiv1.FeatureFlags{\n\t\t\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: false,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"invalid JSON returns empty\",\n\t\t\tfeatureFlagsValue: `{invalid}`,\n\t\t\twant:              apiv1.FeatureFlags{},\n\t\t},\n\t\t{\n\t\t\tname:              \"empty JSON object\",\n\t\t\tfeatureFlagsValue: `{}`,\n\t\t\twant:              apiv1.FeatureFlags{},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctx := context.Background()\n\t\t\tctx, inboundCall := encoding.NewInboundCall(ctx)\n\t\t\theaders := transport.NewHeaders()\n\t\t\tif tt.featureFlagsValue != \"\" {\n\t\t\t\theaders = headers.With(common.ClientFeatureFlagsHeaderName, tt.featureFlagsValue)\n\t\t\t}\n\t\t\terr := inboundCall.ReadFromRequest(&transport.Request{Headers: headers})\n\t\t\trequire.NoError(t, err)\n\t\t\tcall := yarpc.CallFromContext(ctx)\n\t\t\tgot := GetFeatureFlagsFromHeader(call)\n\t\t\trequire.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGetFeatureFlagsFromHeader_RoundTrip(t *testing.T) {\n\tflags := apiv1.FeatureFlags{\n\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t}\n\tserialized := FeatureFlagsHeader(flags)\n\trequire.NotEmpty(t, serialized)\n\n\tctx := context.Background()\n\tctx, inboundCall := encoding.NewInboundCall(ctx)\n\terr := inboundCall.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.ClientFeatureFlagsHeaderName, serialized),\n\t})\n\trequire.NoError(t, err)\n\tcall := yarpc.CallFromContext(ctx)\n\tgot := GetFeatureFlagsFromHeader(call)\n\trequire.Equal(t, flags, got)\n}\n"
  },
  {
    "path": "common/clock/clockfx/clockfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clockfx\n\nimport (\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\n// Module provides real time source for fx application.\nvar Module = fx.Module(\"clock\", fx.Provide(clock.NewRealTimeSource))\n"
  },
  {
    "path": "common/clock/event_timer_gate.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"sync\"\n\t\"time\"\n)\n\ntype (\n\t// EventTimerGate is a TimerGate in which the time is retrieved from external events via SetCurrentTime method.\n\tEventTimerGate interface {\n\t\tTimerGate\n\t\t// SetCurrentTime set the current time, and additionally fire the fire chan\n\t\t// if new \"current\" time is after the fire time, return true if\n\t\t// \"current\" is actually updated\n\t\tSetCurrentTime(t time.Time) bool\n\t}\n\n\teventTimerGateImpl struct {\n\t\t// the channel which will be used to proxy the fired timer\n\t\tfireChan chan time.Time\n\n\t\t// lock for timer and fire time\n\t\tsync.RWMutex\n\t\t// view of current time\n\t\tcurrentTime time.Time\n\t\t// time which timer will fire\n\t\tfireTime time.Time\n\t}\n)\n\n// NewEventTimerGate create a new event timer gate instance\nfunc NewEventTimerGate(currentTime time.Time) EventTimerGate {\n\ttimer := &eventTimerGateImpl{\n\t\tcurrentTime: currentTime,\n\t\tfireTime:    time.Time{},\n\t\tfireChan:    make(chan time.Time, 1),\n\t}\n\treturn timer\n}\n\nfunc (t *eventTimerGateImpl) Chan() <-chan time.Time {\n\treturn t.fireChan\n}\n\nfunc (t *eventTimerGateImpl) FireAfter(now time.Time) bool {\n\tt.RLock()\n\tdefer t.RUnlock()\n\n\tactive := t.currentTime.Before(t.fireTime)\n\treturn active && t.fireTime.After(now)\n}\n\nfunc (t *eventTimerGateImpl) Update(nextTime time.Time) bool {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tactive := t.currentTime.Before(t.fireTime)\n\tif active {\n\t\tif t.fireTime.Before(nextTime) {\n\t\t\t// current time < next wake up time < next time\n\t\t\treturn false\n\t\t}\n\n\t\tif t.currentTime.Before(nextTime) {\n\t\t\t// current time < next time <= next wake up time\n\t\t\tt.fireTime = nextTime\n\t\t\treturn true\n\t\t}\n\n\t\t// next time <= current time < next wake up time\n\t\tt.fireTime = nextTime\n\t\tt.fire(t.currentTime)\n\t\treturn true\n\t}\n\n\t// this means the timer, before stopped, has already fired / never active\n\tif !t.currentTime.Before(nextTime) {\n\t\t// next time is <= current time, need to fire immediately\n\t\t// whether to update next wake up time or not is irrelevent\n\t\tt.fire(t.currentTime)\n\t} else {\n\t\t// next time > current time\n\t\tt.fireTime = nextTime\n\t}\n\treturn true\n}\n\nfunc (t *eventTimerGateImpl) Stop() {\n\tt.Lock()\n\tdefer t.Unlock()\n\tt.fireTime = time.Time{}\n}\n\nfunc (t *eventTimerGateImpl) SetCurrentTime(currentTime time.Time) bool {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tif !t.currentTime.Before(currentTime) {\n\t\t// new current time is <= current time\n\t\treturn false\n\t}\n\n\t// NOTE: do not update the current time now\n\tif !t.currentTime.Before(t.fireTime) {\n\t\t// current time already >= next wakeup time\n\t\t// avoid duplicate fire\n\t\tt.currentTime = currentTime\n\t\treturn true\n\t}\n\n\tt.currentTime = currentTime\n\tif !t.currentTime.Before(t.fireTime) {\n\t\tt.fire(t.currentTime)\n\t}\n\treturn true\n}\n\nfunc (t *eventTimerGateImpl) fire(fireTime time.Time) {\n\tselect {\n\tcase t.fireChan <- fireTime:\n\t\t// timer successfully triggered\n\tdefault:\n\t\t// timer already triggered, pass\n\t}\n}\n"
  },
  {
    "path": "common/clock/event_timer_gate_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\teventTimerGateSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcurrentTime    time.Time\n\t\teventTimerGate EventTimerGate\n\t}\n)\n\nfunc TestEventTimerGeteSuite(t *testing.T) {\n\ts := new(eventTimerGateSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *eventTimerGateSuite) SetupSuite() {\n\n}\n\nfunc (s *eventTimerGateSuite) TearDownSuite() {\n\n}\n\nfunc (s *eventTimerGateSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.currentTime = time.Now().Add(-10 * time.Minute)\n\ts.eventTimerGate = NewEventTimerGate(s.currentTime)\n}\n\nfunc (s *eventTimerGateSuite) TearDownTest() {\n\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFire() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.eventTimerGate.Update(newTimer)\n\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.eventTimerGate.SetCurrentTime(deadline)\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFireAfterUpdate_Active_Updated_BeforeNow() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(5 * time.Second)\n\tupdatedNewTimer := now.Add(-1 * time.Second)\n\tdeadline := now.Add(3 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.True(s.eventTimerGate.Update(updatedNewTimer))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFireAfterUpdate_Active_Updated() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(5 * time.Second)\n\tupdatedNewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(3 * time.Second)\n\ts.eventTimerGate.Update(newTimer)\n\ts.True(s.eventTimerGate.Update(updatedNewTimer))\n\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.eventTimerGate.SetCurrentTime(updatedNewTimer)\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFireAfterUpdate_Active_NotUpdated() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(1 * time.Second)\n\tupdatedNewTimer := now.Add(3 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.eventTimerGate.Update(newTimer)\n\ts.False(s.eventTimerGate.Update(updatedNewTimer))\n\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.eventTimerGate.SetCurrentTime(updatedNewTimer)\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFireAfterUpdate_NotActive_Updated() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(-5 * time.Second)\n\tupdatedNewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\t// this is to drain existing signal\n\t<-s.eventTimerGate.Chan()\n\n\t// test setup up complete\n\n\ts.True(s.eventTimerGate.Update(updatedNewTimer))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.eventTimerGate.SetCurrentTime(updatedNewTimer)\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerFireAfterUpdate_NotActive_NotUpdated() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(-5 * time.Second)\n\tupdatedNewTimer := now.Add(-1 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\t// this is to drain existing signal\n\t<-s.eventTimerGate.Chan()\n\n\t// test setup up complete\n\n\ts.True(s.eventTimerGate.Update(updatedNewTimer))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire when new timer is in the past\")\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerSetCurrentTime_NoUpdate() {\n\tnow := s.currentTime\n\tnewCurrentTime := now.Add(-1 * time.Second)\n\ts.False(s.eventTimerGate.SetCurrentTime(newCurrentTime))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire\")\n\tdefault:\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerSetCurrentTime_Update_TimerAlreadyFired() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(-1 * time.Second)\n\tnewCurrentTime := now.Add(1 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\t// this is to drain existing signal\n\t<-s.eventTimerGate.Chan()\n\n\t// test setup up complete\n\n\ts.True(s.eventTimerGate.SetCurrentTime(newCurrentTime))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire\")\n\tdefault:\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerSetCurrentTime_Update_TimerNotFired() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(2 * time.Second)\n\tnewCurrentTime := now.Add(1 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\ts.True(s.eventTimerGate.SetCurrentTime(newCurrentTime))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire\")\n\tdefault:\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerSetCurrentTime_Update_TimerFired() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(2 * time.Second)\n\tnewCurrentTime := now.Add(2 * time.Second)\n\n\ts.eventTimerGate.Update(newTimer)\n\ts.True(s.eventTimerGate.SetCurrentTime(newCurrentTime))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer should fire\")\n\t}\n\n\t// should fire only once\n\tnewCurrentTime = newCurrentTime.Add(1 * time.Second)\n\ts.True(s.eventTimerGate.SetCurrentTime(newCurrentTime))\n\tselect {\n\tcase <-s.eventTimerGate.Chan():\n\t\ts.Fail(\"timer should not fire\")\n\tdefault:\n\t}\n}\n\nfunc (s *eventTimerGateSuite) TestTimerWillFire_Zero() {\n\t// this test is to validate initial notification will trigger a scan of timer\n\ts.eventTimerGate.Update(time.Time{})\n\ts.False(s.eventTimerGate.FireAfter(time.Now()))\n}\n\nfunc (s *eventTimerGateSuite) TestTimerWillFire_Active() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(2 * time.Second)\n\ttimeBeforeNewTimer := now.Add(1 * time.Second)\n\ttimeAfterNewimer := now.Add(3 * time.Second)\n\ts.eventTimerGate.Update(newTimer)\n\ts.True(s.eventTimerGate.FireAfter(timeBeforeNewTimer))\n\ts.False(s.eventTimerGate.FireAfter(timeAfterNewimer))\n}\n\nfunc (s *eventTimerGateSuite) TestTimerWillFire_NotActive() {\n\tnow := s.currentTime\n\tnewTimer := now.Add(-2 * time.Second)\n\ttimeBeforeTimer := now.Add(-3 * time.Second)\n\ttimeAfterTimer := now.Add(1 * time.Second)\n\ts.eventTimerGate.Update(newTimer)\n\ts.False(s.eventTimerGate.FireAfter(timeBeforeTimer))\n\ts.False(s.eventTimerGate.FireAfter(timeAfterTimer))\n}\n"
  },
  {
    "path": "common/clock/ratelimiter.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package=$GOPACKAGE -destination=ratelimiter_mock.go github.com/uber/cadence/common/clock Reservation\n\npackage clock\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"golang.org/x/time/rate\"\n)\n\ntype (\n\t// Ratelimiter is a test-friendly version of [golang.org/x/time/rate.Limiter],\n\t// which can be backed by any TimeSource.  The changes summarize as:\n\t//   - Time is never allowed to rewind, and does not advance when canceling,\n\t//     fixing core issues with [golang.org/x/time/rate.Limiter] and greatly\n\t//     improving the chances of successfully canceling a reserved token in our\n\t//     usage (from \"almost literally never\" to \"likely 99%+\").\n\t//   - Reservations are simplified and do not allow waiting, but are MUCH more\n\t//     likely to return tokens when canceled or not allowed.\n\t//   - All MethodAt APIs have been excluded as we do not use them, and they would\n\t//     have to bypass the contained TimeSource, which seems error-prone.\n\t//   - All `MethodN` APIs (specifically the \">1 event\" parts) have been excluded\n\t//     because we do not use them, but they seem fairly easy to add if needed later.\n\tRatelimiter interface {\n\t\t// Allow returns true if an event can be performed now, i.e. there\n\t\t// are burst-able tokens available to use.\n\t\t//\n\t\t// To perform events at a steady rate, use Wait instead.\n\t\tAllow() bool\n\t\t// Burst returns the maximum burst size allowed by this Ratelimiter.\n\t\t// A Burst of zero returns false from Allow, unless its Limit is infinity.\n\t\t//\n\t\t// To see the number of burstable tokens available now, use Tokens.\n\t\tBurst() int\n\t\t// Limit returns the maximum overall rate of events that are allowed.\n\t\tLimit() rate.Limit\n\t\t// Reserve claims a single Allow() call that can be canceled if not used.\n\t\t// Check Reservation.Allow() to see if the event is allowed.\n\t\t//\n\t\t// This Reservation does not support waiting, and needs to have\n\t\t// Reservation.Used(true/false) called as soon as possible to have the\n\t\t// best chance of returning its unused token.\n\t\tReserve() Reservation\n\t\t// SetBurst sets the Burst value\n\t\tSetBurst(newBurst int)\n\t\t// SetLimit sets the Limit value\n\t\tSetLimit(newLimit rate.Limit)\n\n\t\tSetLimitAndBurst(newLimit rate.Limit, newBurst int)\n\t\t// Tokens returns the number of immediately-allowable events when called.\n\t\t// Values >= 1 will lead to Allow returning true.\n\t\t//\n\t\t// These values are NOT able to guarantee that a later call will succeed\n\t\t// or fail, as other calls to Allow or Reserve may occur before you are\n\t\t// able to act on the returned value.\n\t\t// It is essentially only suitable for monitoring or test use, and calling\n\t\t// it may lead to failing to cancel reserved tokens (as it advances time).\n\t\tTokens() float64\n\t\t// Wait blocks until one token is available (and consumes it), or it returns\n\t\t// an error and does not consume any tokens if:\n\t\t//   - the context is canceled\n\t\t//   - a token will never become available (burst == 0)\n\t\t//   - a token is not expected to become available in time\n\t\t//     (short deadline and/or many pending reservations or waiters)\n\t\t//\n\t\t// If an error is returned, regardless of the reason, you may NOT proceed\n\t\t// with the limited event you were waiting for.\n\t\tWait(ctx context.Context) error\n\t}\n\n\t// Reservation is a simplified and test-friendly version of [golang.org/x/time/rate.Reservation]\n\t// that only allows checking if it is successful or not, and (possibly) returning unused tokens.\n\t//\n\t// This Reservation MUST have Allow() checked and marked as Used(true/false)\n\t// as soon as possible, or an unused token is unlikely to be returned.\n\t//\n\t// Recommended use is either by using defer:\n\t//\n\t// \tfunc allow() (allowed bool) {\n\t//     \tr := ratelimiter.Reserve()\n\t// \t\t// defer in a closure to read the most-recent value, not the value when deferred\n\t// \t\tdefer func() { r.Used(allowed) }()\n\t// \t\tallowed = r.Allow()\n\t// \t\tif !allowed {\n\t//\t\t\treturn false // also marks the reservation as not-used\n\t// \t\t}\n\t// \t\tif somethingElseAllowsIt() {\n\t// \t\t\t// allow the event, and mark the reservation as used so the token is consumed\n\t//\t\t\treturn true\n\t// \t\t}\n\t// \t\t// cancel the reservation and possibly return the token\n\t// \t\treturn false\n\t// \t}\n\t//\n\t// Or directly calling Used when appropriate:\n\t//\n\t// \tfunc allow() bool {\n\t//     \tr := ratelimiter.Reserve()\n\t// \t\tallowed := r.Allow()\n\t// \t\tif !allowed {\n\t// \t\t\tr.Used(false) // mark the reservation as not-used\n\t//\t\t\treturn false\n\t// \t\t}\n\t// \t\tif somethingElseAllowsIt() {\n\t// \t\t\t// allow the event, and mark the reservation as used\n\t// \t\t\tr.Used(true)\n\t//\t\t\treturn true\n\t// \t\t}\n\t// \t\tr.Used(false) // mark the reservation as not-used, and restore the unused token\n\t// \t\treturn false\n\t// \t}\n\tReservation interface {\n\t\t// Allow returns true if a request should be allowed.\n\t\t//\n\t\t// This may be called at any time and it will return the same value each time,\n\t\t// but it becomes less correct as time passes, so you are expected to call it ~immediately.\n\t\t//\n\t\t// As soon as possible, also call Used(true/false) to consume / return the reserved token.\n\t\t//\n\t\t// This is equivalent to `OK() == true && DelayFrom(reservationTime) == 0`\n\t\t// with [rate.Reservation].\n\t\tAllow() bool\n\n\t\t// Used marks this Reservation as either used or not-used, and it must be called\n\t\t// once on every Reservation:\n\t\t//  - If true, the event is assumed to be allowed, and the Ratelimiter token\n\t\t//    will remain consumed.\n\t\t//  - If false, the Reservation will be rolled back, possibly restoring a Ratelimiter token.\n\t\t//    This is equivalent to calling Cancel() on a [rate.Reservation], but it is more likely\n\t\t//    to return the token.\n\t\tUsed(wasUsed bool)\n\t}\n\n\tratelimiter struct {\n\t\t// important design note: the lock MUST be held both while acquiring AND while handling\n\t\t// `timesource.Now()`.  otherwise time will not be monotonically handled due to waiting\n\t\t// on mutexes, and the *rate.Limiter internal time may go backwards and misbehave.\n\t\ttimesource TimeSource\n\t\tlatestNow  time.Time // updated on each call, never allowed to rewind\n\t\tlimiter    *rate.Limiter\n\t\tmut        sync.Mutex\n\t}\n\n\t// a reservation that allows an immediate call, and can be canceled\n\tallowedReservation struct {\n\t\tres *rate.Reservation\n\t\t// reservedAt is used to cancel at the reservation time, restoring tokens\n\t\t// if (and only if) no other time-advancing calls have interleaved\n\t\treservedAt time.Time\n\t\t// needs access to the parent-wrapped-ratelimiter to ensure time is not rewound\n\t\tlimiter *ratelimiter\n\t}\n\t// a reservation that does NOT allow an immediate call.\n\t// the *rate.Reservation has already been canceled, so this type is trivial.\n\tdeniedReservation struct{}\n)\n\nvar (\n\t_ Ratelimiter = (*ratelimiter)(nil)\n\t_ Reservation = (*allowedReservation)(nil)\n\t_ Reservation = deniedReservation{}\n)\n\n// err constants to keep allocations low\nvar (\n\t// ErrCannotWait is used as the common base for two internal reasons why Ratelimiter.Wait call cannot\n\t// succeed, without directly exposing the reason except as an explanatory string.\n\t//\n\t// Callers should only ever care about ErrCannotWait to be resistant to racing behavior.\n\t// Or perhaps ideally: not care about the reason at all beyond \"not a ctx.Err()\".\n\tErrCannotWait             = fmt.Errorf(\"ratelimiter.Wait() cannot be satisfied\")\n\terrWaitLongerThanDeadline = fmt.Errorf(\"would wait longer than ctx deadline: %w\", ErrCannotWait)\n\terrWaitZeroBurst          = fmt.Errorf(\"zero burst will never allow: %w\", ErrCannotWait)\n)\n\nfunc NewRatelimiter(lim rate.Limit, burst int) Ratelimiter {\n\treturn &ratelimiter{\n\t\ttimesource: NewRealTimeSource(),\n\t\tlimiter:    rate.NewLimiter(lim, burst),\n\t\t// intentionally zero, matches rate.Limiter and helps fill the bucket if it is changed before any use.\n\t\tlatestNow: time.Time{},\n\t}\n}\n\nfunc NewRateLimiterWithTimeSource(ts TimeSource, lim rate.Limit, burst int) Ratelimiter {\n\treturn &ratelimiter{\n\t\ttimesource: ts,\n\t\tlimiter:    rate.NewLimiter(lim, burst),\n\t\t// intentionally zero, matches rate.Limiter and helps fill the bucket if it is changed before any use.\n\t\tlatestNow: time.Time{},\n\t}\n}\n\n// lockNow updates the current \"now\" time, and ensures it never rolls back.\n// This should be equivalent to setting `r.timesource.Now()`, outside of tests\n// that abuse mocked time and Advance with negative values.\n//\n// The lock MUST be held until all time-accepting methods are called on the\n// underlying rate.Limiter, as otherwise the internal \"now\" value may rewind\n// and cause undefined behavior.\nfunc (r *ratelimiter) lockNow() (now time.Time, unlock func()) {\n\tr.mut.Lock()\n\tnewNow := r.timesource.Now() // caution: must be after acquiring the lock\n\tif newNow.After(r.latestNow) {\n\t\t// this should always be true because `time.Now()` is monotonic, and it is\n\t\t// always acquired inside the lock (so values are strictly ordered and\n\t\t// not arbitrarily delayed).\n\t\t//\n\t\t// that means e.g. lockReservedNow should not ever receive a value newer than\n\t\t// the Now we just acquired, so r.latestNow should never be newer either, so\n\t\t// this branch is always selected (unless time is equal).\n\t\t//\n\t\t// that said: it still shouldn't be allowed to rewind, so it's checked.\n\t\tr.latestNow = newNow\n\t}\n\treturn r.latestNow, r.mut.Unlock\n}\n\n// lockReservedNow gets the correct \"now\" time to use, ensuring it never rolls back.\n// This is intended to be used by reservations, to allow them to return claimed tokens\n// after arbitrary waits, as long as no other calls have interleaved.\n//\n// The returned value MUST be used instead of the passed time: it may be the same or newer.\n// The passed value may rewind the rate.Limiter's internal time record, and cause undefined behavior.\n//\n// The lock MUST be held until all time-accepting methods are called on the\n// underlying rate.Limiter, as otherwise the internal \"now\" value may rewind\n// and cause undefined behavior.\nfunc (r *ratelimiter) lockReservedNow(tryNow time.Time) (now time.Time, unlock func()) {\n\tr.mut.Lock()\n\tif tryNow.After(r.latestNow) {\n\t\t// this should never occur!\n\t\t//\n\t\t// reservations and their reserved-time are gathered behind a lock, and time is monotonic.\n\t\t// that means that a reserved-time should ALWAYS be the same or older than any other time\n\t\t// the ratelimiter sees -> its time would have to go backwards.\n\t\t//\n\t\t// that said, if this DOES happen, it must still maintain the newest time.\n\t\t// this should only occur in tests that move time backwards, unless there's\n\t\t// an implementation bug in the ratelimiter or Go's time.\n\t\tr.latestNow = tryNow\n\t}\n\treturn r.latestNow, r.mut.Unlock\n}\n\nfunc (r *ratelimiter) Allow() bool {\n\tnow, unlock := r.lockNow()\n\tdefer unlock()\n\treturn r.limiter.AllowN(now, 1)\n}\n\nfunc (r *ratelimiter) Reserve() Reservation {\n\tnow, unlock := r.lockNow()\n\tdefer unlock()\n\tres := r.limiter.ReserveN(now, 1)\n\tif res.OK() && res.DelayFrom(now) == 0 {\n\t\t// usable token, return a cancel-able handler in case it's not used\n\t\treturn &allowedReservation{\n\t\t\tres:        res,\n\t\t\treservedAt: now,\n\t\t\tlimiter:    r,\n\t\t}\n\t}\n\t// unusable, immediately cancel it to get rid of the future-reservation\n\t// since we don't allow waiting for it.\n\t//\n\t// this always succeeds and returns the token because nothing is allowed to\n\t// interleave, unlike if it is done asynchronously, so it returns MANY more\n\t// tokens when contended.\n\tres.CancelAt(now)\n\treturn deniedReservation{}\n}\n\nfunc (r *ratelimiter) Burst() int {\n\t// rate.Limiter.Burst does not currently advance time,\n\t// so this does not need to hold the ratelimiter lock.\n\t// if this changes, use lockNow.\n\treturn r.limiter.Burst()\n}\n\nfunc (r *ratelimiter) Limit() rate.Limit {\n\t// rate.Limiter.Limit does not currently advance time,\n\t// so this does not need to hold the ratelimiter lock.\n\t// if this changes, use lockNow.\n\treturn r.limiter.Limit()\n}\n\nfunc (r *ratelimiter) SetBurst(newBurst int) {\n\tr.mut.Lock()\n\tdefer r.mut.Unlock()\n\t// setting burst/limit does not advance time, unlike the underlying limiter.\n\t//\n\t// this allows calling them in any order, and the next request\n\t// will fill the token bucket to match elapsed time.\n\t//\n\t// this prefers new burst/limit values over past values,\n\t// as they are assumed to be \"better\", and in particular ensures the first\n\t// time-advancing call fills with the full values (starting from 0 time, like\n\t// the underlying limiter does).\n\tr.limiter.SetBurstAt(r.latestNow, newBurst)\n}\n\nfunc (r *ratelimiter) SetLimit(newLimit rate.Limit) {\n\tr.mut.Lock()\n\tdefer r.mut.Unlock()\n\t// setting burst/limit does not advance time, unlike the underlying limiter.\n\t//\n\t// this allows calling them in any order, and the next request\n\t// will fill the token bucket to match elapsed time.\n\t//\n\t// this prefers new burst/limit values over past values,\n\t// as they are assumed to be \"better\", and in particular ensures the first\n\t// time-advancing call fills with the full values (starting from 0 time, like\n\t// the underlying limiter does).\n\tr.limiter.SetLimitAt(r.latestNow, newLimit)\n}\n\nfunc (r *ratelimiter) SetLimitAndBurst(newLimit rate.Limit, newBurst int) {\n\tr.mut.Lock()\n\tdefer r.mut.Unlock()\n\t// See SetLimit and SetBurst for more information\n\tr.limiter.SetLimitAt(r.latestNow, newLimit)\n\tr.limiter.SetBurstAt(r.latestNow, newBurst)\n}\n\nfunc (r *ratelimiter) Tokens() float64 {\n\tnow, unlock := r.lockNow()\n\tdefer unlock()\n\treturn r.limiter.TokensAt(now)\n}\n\nfunc (r *ratelimiter) Wait(ctx context.Context) (err error) {\n\tif err := ctx.Err(); err != nil {\n\t\t// canceled contexts imply that the limited-work will not be done,\n\t\t// so do not allow any tokens to be consumed.\n\t\t// rate.Limiter also behaves this way in Wait.\n\t\treturn err\n\t}\n\n\tnow, unlock := r.lockNow()\n\tunlockOnce := doOnce(unlock)\n\tdefer unlockOnce() // unlock if panic or returned early with no err\n\n\tres := r.limiter.ReserveN(now, 1)\n\tif !res.OK() {\n\t\t// !OK can only mean \"impossible to wait\" with `ReserveN`.\n\t\t// Since there is no deadline passed to the limiter (the reservation\n\t\t// contains the delay), and we only ever request 1 token, this can only\n\t\t// mean a burst of 0 (and non-infinite rate.Limit).\n\t\tres.CancelAt(now) // unused token, return it while locked\n\t\treturn errWaitZeroBurst\n\t}\n\tdelay := res.DelayFrom(now)\n\tif delay == 0 {\n\t\treturn nil // available token, allow it\n\t}\n\tif deadline, ok := ctx.Deadline(); ok && now.Add(delay).After(deadline) {\n\t\tres.CancelAt(now)                // unused token, return it while locked\n\t\treturn errWaitLongerThanDeadline // don't wait for a known failure\n\t}\n\n\tunlockOnce() // unlock before waiting\n\n\tdefer func() {\n\t\tif err != nil {\n\t\t\t// err return means \"not allowed\", so cancel the reservation.\n\t\t\t//\n\t\t\t// note that this makes a separate call to get the current time:\n\t\t\t// this is intentional, as time has likely passed while waiting.\n\t\t\t//\n\t\t\t// if the cancellation happened before the time-to-act (the delay),\n\t\t\t// the reservation will still be successfully rolled back.\n\n\t\t\t// (re)-acquire the latest now value.\n\t\t\t//\n\t\t\t// it should not be advanced to the \"real\" now, to improve the chance\n\t\t\t// that the token will be restored, but if there was any waiting\n\t\t\t// it's pretty likely that other calls occurred while unlocked and\n\t\t\t// the most-recently-observed time must be maintained.\n\t\t\tnow, unlock := r.lockReservedNow(now)\n\t\t\tdefer unlock()\n\t\t\tres.CancelAt(now)\n\t\t}\n\t}()\n\n\t// wait for cancellation or the waiter's turn.\n\t// re-acquiring Now() here is valid and may shorten the delay, but may add\n\t// more branches and isn't strictly necessary as sleeps aren't precise anyway.\n\ttimer := r.timesource.NewTimer(delay)\n\tdefer timer.Stop()\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\tcase <-timer.Chan():\n\t\treturn nil\n\t}\n}\n\nfunc (r *allowedReservation) Used(used bool) {\n\tif !used {\n\t\tnow, unlock := r.limiter.lockReservedNow(r.reservedAt)\n\t\t// lock must be held while canceling, because it can affect the limiter's time\n\t\tdefer unlock()\n\n\t\t// if no calls interleaved, this will be the same as the reservation time,\n\t\t// and the cancellation will be able to roll back an immediately-available call.\n\t\t//\n\t\t// otherwise, the ratelimiter's time has been advanced, and this will fail to\n\t\t// restore any tokens.\n\t\tr.res.CancelAt(now)\n\t}\n\t// if used, do nothing.\n\t//\n\t// this method largely exists so it can be mocked and asserted,\n\t// or customized by an implementation that tracks allowed/rejected metrics.\n}\n\nfunc (r *allowedReservation) Allow() bool {\n\t// this is equivalent to `r.res.OK() && r.res.DelayFrom(r.reservedAt) == 0`,\n\t// but this is known to be `true` when it is first reserved and `r.reservedAt`\n\t// does not ever change, so it can be skipped here.\n\treturn true\n}\n\nfunc (r deniedReservation) Used(used bool) {\n\t// reservation already canceled, nothing to roll back\n}\n\nfunc (r deniedReservation) Allow() bool {\n\treturn false\n}\n\n// a small sync.Once-alike for single-threaded use.\n//\n// Ratelimiter.Wait benchmarks perform measurably better with this vs sync.Once,\n// though not by a large margin.\nfunc doOnce(f func()) func() {\n\tcalled := false\n\treturn func() {\n\t\tif called {\n\t\t\treturn\n\t\t}\n\t\tcalled = true\n\t\tf()\n\t}\n}\n"
  },
  {
    "path": "common/clock/ratelimiter_bench_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"go.uber.org/atomic\"\n\t\"golang.org/x/time/rate\"\n)\n\n/*\nBenchmark data can be seen in the testdata folder, for relevant machines.\n\nInterpretation:\n  - for essentially all operations, mostly regardless of sequential / parallel / how parallel,\n    the added latency is ~50%, and even extremes are less than double.\n    this seems entirely tolerable and safe to use.\n  - \"reserve and cancel\" sequentially is actually *faster* with the wrapper, probably due to fewer time advancements.\n    with parallel usage, real vs wrapper is basically a tie.\n  - mock-time ratelimiter is faster than real-time in essentially all cases, as you would hope.\n    though not by much, unless waiting was part of the test.\n  - real *rate.Limiter instances have some pathological behavior that... might disqualify them from safe use tbh.\n    essentially this summarizes as \"time.Now() is not globally monotonic\", and it can lead to significant over and under limiting.\n\nThe real *rate.Limiter instances show these quirks during parallel tests, due to time rewinding and jumping forward repeatedly\nfrom the *rate.Limiter's point of view (as it has an internal last-Now value protected by a mutex):\n 1. `Allow()`-like calls (both literally `Allow()`, and `Reserve()` followed by sometimes `reservation.Cancel()`)\n    can allow both significantly more or significantly fewer calls than they should.\n 2. `Wait()` can allow calls through at a faster rate than it should.  At peak, a bit over 2x.\n    (and probably slower, but sleeping is not guaranteed to wake up at a precise time so would still be correct behavior)\n\nEssentially, when one call sets the time to \"now\", and an out-of-sync call sets it to \"now-1ms\" (time is acquired outside\nthe limiter's lock, and they may make progress out of order), a token may be restored when a time-consistent limiter would\nnot do so.\n\nYou can recreate this by hand by feeding a ratelimiter \"now\" and \"now+1s\" (>=2x the limit) randomly, and watching what it\nallows / what the value of Tokens() is as time passes.  Doing this *literally* by hand reproduces it, e.g. pressing enter\non a command-line loop that does this - high frequency is not necessary, just illogical \"time travel\" outside tight bounds.\n\nThe first can be seen in m1_mac_go1.21.txt by comparing these tests:\n  - BenchmarkLimiter/allow/parallel/real-4 (note the near-1µs time per allow, which serial calls match)\n  - BenchmarkLimiter/allow/parallel/real-8 (time-per-allow is now 231ns, over 4x more allowed than it should)\n  - BenchmarkLimiter/reserve-canceling-every-3/parallel/real-2 (1.5ms per allow, around 1,000x fewer allowed than it should)\n  - BenchmarkLimiter/reserve-canceling-every-3/parallel/real-8 (450ns per allow, around 2x more than it should)\n\n\"real with pinned time\" behaves similarly, for similar reasons: canceling can time-travel.\npinned time does make the behavior when called on a single goroutine roughly perfect though.\n\nAnd the second can be seen in m1_mac_go1.21.txt by comparing these tests:\n  - BenchmarkLimiter/wait/parallel/1µs_rate/real (almost exactly 1µs per iteration)\n  - BenchmarkLimiter/wait/parallel/1µs_rate/real-8 (~500ns, twice as fast as it should allow)\n\nI have not tried to verify Wait's cause by hand, but it certainly seems like time-thrashing explains it as well.\nThis is also supported by the wrapper's time-locking *completely* eliminating this flaw, as all iterations take\nalmost exactly 1µs as they should.\n\n---\n\nAll machines I've tried it on behave similarly, though whether one parallel test triggers misbehavior or not\nis fairly random.  Higher parallelism triggers it more, and it seems *fairly* likely that at least one misbehaves\nin a full `-cpu 1,2,4,8,...[>=cores]` suite, and sometimes many more.\n*/\nfunc BenchmarkLimiter(b *testing.B) {\n\tconst (\n\t\tnormalLimit      = time.Microsecond // pretty arbitrary but has a good blend of allow and deny\n\t\tburst            = 1000             // arbitrary, but more interesting than 1, and better matches how we use limiters (rate/s == burst)\n\t\tallowedPeriodFmt = \"time per allow: %9s\"\n\t)\n\n\ttype runType func(b *testing.B, each func(int) bool)\n\n\t// runs a callback in a sequential benchmark, and tracks allowed/denied metrics\n\tvar runSerial runType = func(b *testing.B, each func(int) bool) {\n\t\tb.Helper()\n\t\tallowed, denied := 0, 0\n\t\tfor i := 0; i < b.N; i++ {\n\t\t\tif each(i) {\n\t\t\t\tallowed++\n\t\t\t} else {\n\t\t\t\tdenied++\n\t\t\t}\n\t\t}\n\t\tas := fmt.Sprintf(\"allowed: %v,\", allowed)\n\t\tds := fmt.Sprintf(\"denied: %v,\", denied)\n\t\tallowedPeriod := fmt.Sprintf(allowedPeriodFmt, \"n/a\")\n\t\tif allowed > 0 {\n\t\t\tallowedPeriod = fmt.Sprintf(allowedPeriodFmt, fmt.Sprint(b.Elapsed()/time.Duration(allowed)))\n\t\t}\n\t\tb.Logf(\"%-20s %-20s %s\", as, ds, allowedPeriod) // allows controlling whitespace better\n\t}\n\t// runs a callback in a parallel benchmark, and tracks allowed/denied metrics\n\tvar runParallel runType = func(b *testing.B, each func(int) bool) {\n\t\tb.Helper()\n\t\tvar allowed, denied atomic.Int64\n\t\tb.RunParallel(func(pb *testing.PB) {\n\t\t\tn := 0\n\t\t\tfor pb.Next() {\n\t\t\t\tif each(n) {\n\t\t\t\t\tallowed.Inc()\n\t\t\t\t} else {\n\t\t\t\t\tdenied.Inc()\n\t\t\t\t}\n\t\t\t\tn++\n\t\t\t}\n\t\t})\n\t\tas := fmt.Sprintf(\"allowed: %v,\", allowed.Load())\n\t\tds := fmt.Sprintf(\"denied: %v,\", denied.Load())\n\t\tallowedPeriod := fmt.Sprintf(allowedPeriodFmt, \"n/a\")\n\t\tif allowed.Load() > 0 {\n\t\t\tallowedPeriod = fmt.Sprintf(allowedPeriodFmt, fmt.Sprint(b.Elapsed()/time.Duration(allowed.Load())))\n\t\t}\n\t\tb.Logf(\"%-20s %-20s %s\", as, ds, allowedPeriod) // allows controlling whitespace better\n\t}\n\n\tboth := map[string]runType{\n\t\t\"serial\":   runSerial,\n\t\t\"parallel\": runParallel,\n\t}\n\n\t// Allow is a significant amount of our ratelimiter usage,\n\t// so this should probably take top priority.\n\tb.Run(\"allow\", func(b *testing.B) {\n\t\tfor name, runner := range both {\n\t\t\tb.Run(name, func(b *testing.B) {\n\t\t\t\tb.Run(\"real\", func(b *testing.B) {\n\t\t\t\t\t// very fast to jump back and forth, rather than slamming into \"deny\"\n\t\t\t\t\trl := rate.NewLimiter(rate.Every(normalLimit), burst)\n\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\treturn rl.Allow()\n\t\t\t\t\t})\n\t\t\t\t})\n\t\t\t\tb.Run(\"wrapped\", func(b *testing.B) {\n\t\t\t\t\trl := NewRatelimiter(rate.Every(normalLimit), burst)\n\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\treturn rl.Allow()\n\t\t\t\t\t})\n\t\t\t\t})\n\t\t\t\tb.Run(\"mocked timesource\", func(b *testing.B) {\n\t\t\t\t\tts := NewMockedTimeSource()\n\t\t\t\t\trl := NewRateLimiterWithTimeSource(ts, rate.Every(normalLimit), burst)\n\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\t// adjusted by eye, to try to very roughly match the above values for the final runs.\n\t\t\t\t\t\t// anything not extremely higher or lower is probably fine.\n\t\t\t\t\t\tts.Advance(normalLimit / 5)\n\t\t\t\t\t\treturn rl.Allow()\n\t\t\t\t\t})\n\t\t\t\t})\n\t\t\t})\n\t\t}\n\t})\n\n\t// Reserve is used when tiering ratelimiters, which is only done in a few\n\t// primarily-user-facing limits that aren't super high perf need...\n\t// ... BUT Reserve is used in the wrapper's Wait, so this serves to separate\n\t// out its cost from the Wait benchmarks.\n\tcancelNth := 3\n\tb.Run(fmt.Sprintf(\"reserve-canceling-every-%v\", cancelNth), func(b *testing.B) {\n\t\tfor name, runner := range both {\n\t\t\tb.Run(name, func(b *testing.B) {\n\t\t\t\t/*\n\t\t\t\t\tCAUTION: in parallel, these real ratelimiters run quickly, but note how\n\t\t\t\t\tfrequently requests are allowed.  Both more and less than intended, sometimes\n\t\t\t\t\tby multiple orders of magnitude.\n\n\t\t\t\t\texample:\n\t\t\t\t\t\tBenchmarkLimiter/reserve-canceling-every-3/parallel/real-2\n\t\t\t\t\t\t\tratelimiter_test.go:136: allowed: 752,        denied: 7519799,     time per allow: 1.558596ms\n\t\t\t\t\t\t\t                                                         1,500x too few! ----------^\n\t\t\t\t\t\tBenchmarkLimiter/reserve-canceling-every-3/parallel/real-8\n\t\t\t\t\t\t\tratelimiter_test.go:136: allowed: 2651622,    denied: 1415746,     time per allow:     449ns\n\t\t\t\t\t\t\t                                                            2x too many! --------------^\n\n\t\t\t\t\tthis occurs because the ratelimiter gathers `time.Now()` before locking, which\n\t\t\t\t\tleads to *handling* time in a non-monotonic way depending on what goroutines\n\t\t\t\t\tacquire the lock / in which order.\n\n\t\t\t\t\tthis is one of the reasons this wrapper was built.\n\t\t\t\t*/\n\t\t\t\tb.Run(\"real\", func(b *testing.B) {\n\t\t\t\t\trl := rate.NewLimiter(rate.Every(normalLimit), burst)\n\t\t\t\t\trunner(b, func(i int) (allowed bool) {\n\t\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\t\tallowed = r.OK() && r.Delay() == 0\n\t\t\t\t\t\tcanceled := i%cancelNth == 0\n\t\t\t\t\t\tallowed = allowed && !canceled\n\t\t\t\t\t\tif !allowed {\n\t\t\t\t\t\t\t// all not-allowed calls must be canceled, as they will not be \"used\".\n\t\t\t\t\t\t\t// otherwise they will still be \"used\", and tokens will go negative,\n\t\t\t\t\t\t\t// possibly far into the negatives.\n\t\t\t\t\t\t\tr.Cancel()\n\t\t\t\t\t\t\treturn\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn\n\t\t\t\t\t})\n\t\t\t\t})\n\t\t\t\tb.Run(\"real pinned time\", func(b *testing.B) {\n\t\t\t\t\trl := rate.NewLimiter(rate.Every(normalLimit), burst)\n\t\t\t\t\trunner(b, func(i int) (allowed bool) {\n\t\t\t\t\t\t// expected to be faster as the limiter's time does not\n\t\t\t\t\t\t// advance as often, and \"now\" is only gathered once.\n\t\t\t\t\t\t//\n\t\t\t\t\t\t// but this is not correct with concurrent use, so it's\n\t\t\t\t\t\t// purely synthetic and serves as a lower bound only.\n\t\t\t\t\t\tnow := time.Now()\n\t\t\t\t\t\tr := rl.ReserveN(now, 1)\n\t\t\t\t\t\tallowed = r.OK() && r.DelayFrom(now) == 0\n\t\t\t\t\t\tcanceled := i%cancelNth == 0\n\t\t\t\t\t\tallowed = allowed && !canceled\n\t\t\t\t\t\tif !allowed {\n\t\t\t\t\t\t\tr.CancelAt(now)\n\t\t\t\t\t\t\treturn\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn\n\t\t\t\t\t})\n\t\t\t\t})\n\n\t\t\t\t// calls Reserve on the wrapped limiter and sometimes cancels, sequentially or in parallel\n\t\t\t\trunWrapped := func(b *testing.B, rl Ratelimiter, runner runType, advance func()) {\n\t\t\t\t\tb.Helper()\n\t\t\t\t\trunner(b, func(i int) (allowed bool) {\n\t\t\t\t\t\t// not b.Helper because no logs occur inside here, and it has runtime cost\n\t\t\t\t\t\tif advance != nil {\n\t\t\t\t\t\t\tadvance() // for mock time\n\t\t\t\t\t\t}\n\t\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\t\tallowed = r.Allow()\n\t\t\t\t\t\tcanceled := i%cancelNth == 0\n\t\t\t\t\t\tallowed = allowed && !canceled\n\t\t\t\t\t\tr.Used(allowed)\n\t\t\t\t\t\treturn\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\t\tb.Run(\"wrapped\", func(b *testing.B) {\n\t\t\t\t\trl := NewRatelimiter(rate.Every(normalLimit), burst)\n\t\t\t\t\trunWrapped(b, rl, runner, nil)\n\t\t\t\t})\n\t\t\t\tb.Run(\"mocked timesource\", func(b *testing.B) {\n\t\t\t\t\tts := NewMockedTimeSource()\n\t\t\t\t\trl := NewRateLimiterWithTimeSource(ts, rate.Every(normalLimit), burst)\n\t\t\t\t\trunWrapped(b, rl, runner, func() {\n\t\t\t\t\t\t// any value should work, but smaller than normalLimit revealed\n\t\t\t\t\t\t// some issues in the past, where time-thrashing would lead to\n\t\t\t\t\t\t// extremely low allowed calls.\n\t\t\t\t\t\t//\n\t\t\t\t\t\t// this is now resolved with synchronous cancels when reservations\n\t\t\t\t\t\t// are not synchronously allowed, but it seems worth keeping to\n\t\t\t\t\t\t// reveal regressions.\n\t\t\t\t\t\tts.Advance(normalLimit / 10)\n\t\t\t\t\t})\n\t\t\t\t})\n\t\t\t})\n\t\t}\n\t})\n\n\t// Wait is implemented quite differently between real and wrapped.\n\t// For the most part our use is in batches, where queueing is expected.\n\t// This makes benchmarks kinda synthetic.  Queueing won't run any faster than the intended rate.\n\t//\n\t// Still, the allowed events per second should be similar (when limited).\n\t//\n\t// Unfortunately there are some rate.Limiter misbehaviors and a very high chance\n\t// of CPU latency impacting this test, so this is not asserted, only logged.\n\tb.Run(\"wait\", func(b *testing.B) {\n\t\tfor name, runner := range both {\n\t\t\tb.Run(name, func(b *testing.B) {\n\t\t\t\tdurations := []time.Duration{\n\t\t\t\t\t// try an insanely fast refresh, to see maximum throughput with \"zero\" waiting.\n\t\t\t\t\t// intentionally avoiding 1 nanosecond in case it gets special-cased, like 0 and math.MaxInt64.\n\t\t\t\t\t2 * time.Nanosecond,\n\t\t\t\t\t// try a rate that should contend and wait heavily.\n\t\t\t\t\t// this SHOULD be how long the per-iteration time is (when not mocked)... but read comments below.\n\t\t\t\t\tnormalLimit,\n\t\t\t\t}\n\t\t\t\tfor _, dur := range durations {\n\t\t\t\t\tb.Run(fmt.Sprintf(\"%v rate\", dur), func(b *testing.B) {\n\t\t\t\t\t\tlimit := rate.Every(dur)\n\t\t\t\t\t\t// set a wait timeout long enough to allow ~all attempts.\n\t\t\t\t\t\t//\n\t\t\t\t\t\t// b.Run seems to target <1s, but very fast or very slow things can extend it,\n\t\t\t\t\t\t// and up to about 2s seems normal for these.\n\t\t\t\t\t\t// because of that, 5s occasionally timed out, but 10s seems fine.\n\t\t\t\t\t\ttimeout := 20 * time.Second\n\t\t\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), timeout)\n\t\t\t\t\t\tdefer cancel()\n\n\t\t\t\t\t\tb.Run(\"real\", func(b *testing.B) {\n\t\t\t\t\t\t\t// sometimes misbehaves, allowing too many or too few requests through!\n\t\t\t\t\t\t\trl := rate.NewLimiter(limit, burst)\n\t\t\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\t\t\treturn rl.Wait(ctx) == nil\n\t\t\t\t\t\t\t})\n\t\t\t\t\t\t})\n\t\t\t\t\t\tb.Run(\"wrapped\", func(b *testing.B) {\n\t\t\t\t\t\t\trl := NewRatelimiter(limit, burst)\n\t\t\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\t\t\treturn rl.Wait(ctx) == nil\n\t\t\t\t\t\t\t})\n\t\t\t\t\t\t})\n\t\t\t\t\t\tb.Run(\"mocked timesource\", func(b *testing.B) {\n\t\t\t\t\t\t\tts := NewMockedTimeSource()\n\t\t\t\t\t\t\trl := NewRateLimiterWithTimeSource(ts, limit, burst)\n\t\t\t\t\t\t\tctx := &timesourceContext{\n\t\t\t\t\t\t\t\tts:       ts,\n\t\t\t\t\t\t\t\tdeadline: time.Now().Add(timeout),\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t// make sure tests don't block super long even if they misbehave,\n\t\t\t\t\t\t\t// by canceling the context after enough real-world time has passed.\n\t\t\t\t\t\t\tdefer time.AfterFunc(timeout, func() {\n\t\t\t\t\t\t\t\tts.Advance(timeout)\n\t\t\t\t\t\t\t}).Stop()\n\t\t\t\t\t\t\trunner(b, func(i int) bool {\n\t\t\t\t\t\t\t\t// advance enough to restore a token so it does not block forever.\n\t\t\t\t\t\t\t\t// \"mock-wait times out when it should\" is asserted in the tests.\n\t\t\t\t\t\t\t\tts.Advance(dur)\n\t\t\t\t\t\t\t\treturn rl.Wait(ctx) == nil\n\t\t\t\t\t\t\t})\n\t\t\t\t\t\t})\n\n\t\t\t\t\t\t// should be true if the round took longer than timeout,\n\t\t\t\t\t\t// as the AfterFunc will fire -> advance time -> close context -> then the round finishes.\n\t\t\t\t\t\tif ctx.Err() != nil {\n\t\t\t\t\t\t\tb.Errorf(\"benchmark likely invalid, did not complete before context timeout (%v)\", timeout)\n\t\t\t\t\t\t}\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\t})\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "common/clock/ratelimiter_comparison_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"golang.org/x/sync/errgroup\"\n\t\"golang.org/x/time/rate\"\n)\n\nconst (\n\t// amount of time between each conceptual \"tick\" of the test's clocks, both real and fake.\n\t// sadly even 10ms has some excessive latency a few % of the time, particularly during the Wait check.\n\tfuzzGranularity = 100 * time.Millisecond\n\n\t// keep running tests until this amount of time has passed, or a failure has occurred.\n\t// this could be a static count of tests to run, but a target duration\n\t// is a bit less fiddly.  \"many attempts\" is the goal, not any specific number.\n\t//\n\t// alternatively, tests have a deadline, this could run until near that deadline.\n\t// but right now that's not fine-tuned per package, so it's quite long.\n\tfuzzDuration = 8 * time.Second // mostly stays under 10s, feels reasonable\n\n\t// debug-level output is very noisy on successful runs, so hide it by default.\n\t// using a custom seed will also set this to true.\n\tfuzzDebugLog   = false\n\tfuzzCustomSeed = 0 // override with non-zero seed to run just one test with that seed\n)\n\nfunc TestAgainstRealRatelimit(t *testing.T) {\n\t// make sure to skip this test when checking coverage, as coverage it generates is not stable.\n\t// t.Skip(\"skipped to check coverage\")\n\n\tconst (\n\t\t// number of [fuzzGranularity] events to trigger per round,\n\t\t// and also the number of [fuzzGranularity] periods before a token is added.\n\t\t//\n\t\t// each test takes at least fuzzGranularity*events*rounds, so keep it kinda small.\n\t\tevents = 10\n\t\t// number of rounds of \"N events\" to try\n\t\trounds = 2\n\n\t\t// less than events to give Wait a better chance of waiting,\n\t\t// but anything up to (events*rounds)/2 will have some effect.\n\t\tmaxBurst = events / 2\n\t)\n\n\t// The mock ratelimiter should behave the same as a real ratelimiter in all cases.\n\t//\n\t// So fuzz test it: throw random calls at non-overlapping times that are a bit\n\t// spaced out (to prevent noise due to busy CPUs) and make sure all impls agree.\n\t//\n\t// If a test fails, please verify by hand with the failing seed.\n\t// Enabling debug logs can help show what's happening and what *should* be happening,\n\t// though there is a chance it's just due to CPU contention.\n\tt.Run(\"fuzz\", func(t *testing.T) {\n\t\tdeadline := time.Now().Add(fuzzDuration)\n\t\tfor testnum := 0; !t.Failed() && time.Now().Before(deadline); testnum++ {\n\t\t\tif testnum > 0 && fuzzCustomSeed != 0 {\n\t\t\t\t// already ran the seeded test, stop early.\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tt.Run(fmt.Sprintf(\"attempt %v\", testnum), func(t *testing.T) {\n\t\t\t\tt.Cleanup(func() {\n\t\t\t\t\tif t.Failed() {\n\t\t\t\t\t\t/*\n\t\t\t\t\t\t\tIs this test becoming too slow or too noisy?\n\n\t\t\t\t\t\t\tThe easiest \"likely to actually work\" fix is probably going to\n\t\t\t\t\t\t\trequire detecting excessive lag, and retrying instead of failing.\n\n\t\t\t\t\t\t\tThat could miss some real flaws if they are racy by nature, but\n\t\t\t\t\t\t\tseems like it might be good enough.\n\t\t\t\t\t\t*/\n\t\t\t\t\t\tt.Logf(\"---- CAUTION ----\")\n\t\t\t\t\t\tt.Logf(\"these tests are randomized by design, a random failure may be a real flaw!\")\n\t\t\t\t\t\tt.Logf(\"please replay with the failing seed (set `fuzzCustomSeed`) and check detailed output to see what the behavior should be.\")\n\t\t\t\t\t\tt.Logf(\"also try setting `fuzzDebugLog` to false to true to show Tokens() progression, this can help explain many issues.\")\n\t\t\t\t\t\tt.Logf(\"\")\n\t\t\t\t\t\tt.Logf(\"if you are intentionally making changes, be sure to run a few hundred rounds to make sure your changes are stable,\")\n\t\t\t\t\t\tt.Logf(\"and in particular make sure an 'Advancing mock time ...' event occurs and times match, as that is a bit rare\")\n\t\t\t\t\t\tt.Logf(\"---- CAUTION ----\")\n\t\t\t\t\t}\n\t\t\t\t})\n\n\t\t\t\t// parallel saves a fair bit of time but introduces a lot of CPU contention\n\t\t\t\t// and that leads to a moderate amount of flakiness.\n\t\t\t\t//\n\t\t\t\t// unfortunately not recommended here.\n\t\t\t\t// t.Parallel()\n\n\t\t\t\tseed := time.Now().UnixNano()\n\t\t\t\tif fuzzCustomSeed != 0 {\n\t\t\t\t\tseed = fuzzCustomSeed // override seed to test a specific scenario\n\t\t\t\t}\n\t\t\t\trng := rand.New(rand.NewSource(seed))\n\t\t\t\tt.Logf(\"rand seed: %v\", seed)\n\t\t\t\tround := makeRandomSchedule(t, rng, rounds, events)\n\t\t\t\tburst := rng.Intn(maxBurst) // zero is not very interesting, but allowed\n\t\t\t\t// waits do not usually need to pause and it's tough to assert randomly.\n\t\t\t\t// any waited periods are still asserted to be similar though.\n\t\t\t\t// the expected-wait case is tested separately below.\n\t\t\t\tnoMinWait := time.Duration(0)\n\t\t\t\tassertRatelimitersBehaveSimilarly(t, burst, noMinWait, round)\n\t\t\t})\n\t\t}\n\t})\n\n\t// also check edge cases that random tests only rarely or never hit:\n\tt.Run(\"edge cases\", func(t *testing.T) {\n\t\t// Wait at the end of a round until a token recovers at the start of the next.\n\t\tt.Run(\"wait must wait similarly\", func(t *testing.T) {\n\t\t\tschedule := [][]string{\n\t\t\t\t{\"a\", \"_\", \"_\", \"w\"}, // consume the only token, then wait with 0.75 tokens.\n\t\t\t\t{\"_\", \"_\", \"_\", \"_\"}, // first slot must be _ to recover a token and unblock the wait.\n\t\t\t}\n\t\t\tlogSchedule(t, schedule)\n\t\t\tminWait := fuzzGranularity / 2 // not waiting will fail this test\n\t\t\tburst := 1\n\t\t\tassertRatelimitersBehaveSimilarly(t, burst, minWait, schedule)\n\t\t})\n\t\tt.Run(\"canceled contexts do not consume tokens\", func(t *testing.T) {\n\t\t\tactual := rate.NewLimiter(rate.Every(time.Second), 1)\n\t\t\twrapped := NewRatelimiter(rate.Every(time.Second), 1)\n\n\t\t\tcanceledCtx, cancel := context.WithCancel(context.Background())\n\t\t\tcancel()\n\n\t\t\tactualErr := actual.Wait(canceledCtx)\n\t\t\tactualTokens := actual.Tokens()\n\n\t\t\twrappedErr := wrapped.Wait(canceledCtx)\n\t\t\twrappedTokens := wrapped.Tokens()\n\n\t\t\tassert.ErrorIs(t, actualErr, context.Canceled)\n\t\t\tassert.ErrorIs(t, wrappedErr, context.Canceled)\n\n\t\t\tassert.Equal(t, 1.0, actualTokens, \"rate.Limiter should still have a token available\")\n\t\t\tassert.Equal(t, 1.0, wrappedTokens, \"wrapped should still have a token available\")\n\t\t})\n\t\tt.Run(\"cancel while waiting returns tokens\", func(t *testing.T) {\n\t\t\t// expect to recover 0.1 token after sleep\n\t\t\tactual := rate.NewLimiter(rate.Every(10*fuzzGranularity), 1)\n\t\t\twrapped := NewRatelimiter(rate.Every(10*fuzzGranularity), 1)\n\n\t\t\tactual.Allow()\n\t\t\twrapped.Allow()\n\n\t\t\tt.Logf(\"tokens in real: %0.5f, actual: %0.5f\", actual.Tokens(), wrapped.Tokens())\n\t\t\tassert.InDeltaf(t, 0, actual.Tokens(), 0.1, \"rate.Limiter should have near zero tokens\")\n\t\t\tassert.InDeltaf(t, 0, wrapped.Tokens(), 0.1, \"wrapped should have near zero tokens\")\n\n\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\tdefer cancel()\n\t\t\tgo func() {\n\t\t\t\ttime.Sleep(fuzzGranularity)\n\t\t\t\tcancel()\n\t\t\t}()\n\n\t\t\tvar actualDur, wrappedDur time.Duration\n\t\t\tvar actualErr, wrappedErr error\n\n\t\t\tvar wg sync.WaitGroup\n\t\t\twg.Add(2)\n\t\t\tgo func() {\n\t\t\t\tdefer wg.Done()\n\t\t\t\tstart := time.Now()\n\t\t\t\tactualErr = actual.Wait(ctx)\n\t\t\t\tactualDur = time.Since(start).Round(time.Millisecond)\n\t\t\t}()\n\t\t\tgo func() {\n\t\t\t\tdefer wg.Done()\n\t\t\t\tstart := time.Now()\n\t\t\t\twrappedErr = wrapped.Wait(ctx)\n\t\t\t\twrappedDur = time.Since(start).Round(time.Millisecond)\n\t\t\t}()\n\t\t\twg.Wait()\n\n\t\t\tassert.ErrorIs(t, actualErr, context.Canceled)\n\t\t\tassert.ErrorIs(t, wrappedErr, context.Canceled)\n\n\t\t\t// in particular, this would go to -1 if the token was not successfully returned.\n\t\t\t// that can happen if canceling near the wait time elapsing, but that would take a full second in this test.\n\t\t\tt.Logf(\"tokens in real: %0.5f, actual: %0.5f\", actual.Tokens(), wrapped.Tokens())\n\t\t\tassert.InDeltaf(t, 0.1, actual.Tokens(), 0.1, \"rate.Limiter should have returned token when canceled, and the 1/10th sleep should remain\")\n\t\t\tassert.InDeltaf(t, 0.1, wrapped.Tokens(), 0.1, \"wrapped should have returned token when canceled, and the 1/10th sleep should remain\")\n\n\t\t\tt.Logf(\"waiting times for real: %v, actual: %v\", actualDur, wrappedDur)\n\t\t\tassert.InDeltaf(t, actualDur, fuzzGranularity, float64(fuzzGranularity/2), \"rate.Limiter should have waited until canceled\")\n\t\t\tassert.InDeltaf(t, wrappedDur, fuzzGranularity, float64(fuzzGranularity/2), \"wrapped should have waited until canceled\")\n\t\t})\n\t})\n\n}\n\n/*\nGenerate a random schedule for testing behavior.\n\nSchedules look like:\n\n\t[r _ a w _ _ _ _ _ w]\n\t[_ c _ _ _ _ w _ _ _]\n\nWhich means that the first round will:\n  - reserve()\n  - do nothing\n  - allow()\n  - wait() (may extend to next event, so the next one is always \"_\")\n  - ...\n\nBy the end of that first array it'll reach the end of the `rate.Every` time, and\nmay recover tokens during the next round's runtime.  So the second round will:\n  - do nothing, just refresh .1 while waiting ==> will have 1.0 tokens regenerated\n    since the test began, as it was reserved earlier (if non-zero burst).\n    (this could be immediately consumed by last round's final wait)\n  - reserve but cancel, restoring the token (if any), and maybe refresh .1 token\n  - do nothing, maybe refresh .1 more token...\n  - ... and try to wait on the 7th event.\n\nNon-blank events do not overlap between rounds to avoid triggering a race\nbetween the old consumed tokens being refreshed and a new call consuming them,\nas these make for very flaky tests unless time is mocked.\n*/\nfunc makeRandomSchedule(t *testing.T, rng *rand.Rand, rounds, events int) [][]string {\n\t// generate some non-colliding \"A == perform an Allow call\" rounds.\n\tschedule := make([][]string, rounds) // using string so it prints nicely\n\tfor i := 0; i < rounds; i++ {\n\t\tround := make([]string, events)\n\t\tfor j := 0; j < events; j++ {\n\t\t\tround[j] = \"_\" // \"do nothing\" marker\n\t\t}\n\t\tschedule[i] = round\n\t}\n\tskipWait := false\n\tfor i := 0; i < len(schedule[0]); i++ {\n\t\tif skipWait {\n\t\t\tskipWait = false\n\t\t\tcontinue\n\t\t}\n\t\t// pick one to set to true, or none (1 chance for none)\n\t\tset := rng.Intn(len(schedule) + 1)\n\t\tif set < len(schedule) {\n\t\t\t// 1/3rd of them are waits\n\t\t\tswitch rng.Intn(4) {\n\t\t\tcase 0:\n\t\t\t\tschedule[set][i] = \"a\" // allow\n\t\t\tcase 1:\n\t\t\t\tschedule[set][i] = \"r\" // reserve\n\t\t\tcase 2:\n\t\t\t\tschedule[set][i] = \"c\" // reserve but cancel\n\t\t\tcase 3:\n\t\t\t\tschedule[set][i] = \"w\" // wait\n\t\t\t\t// waits can pause for an event and a half, so skip the next event to avoid overlapping\n\t\t\t\tskipWait = true\n\t\t\tdefault:\n\t\t\t\tpanic(\"missing a case\")\n\t\t\t}\n\t\t}\n\t}\n\tlogSchedule(t, schedule)\n\n\treturn schedule\n}\n\nfunc logSchedule(t *testing.T, schedule [][]string) {\n\tt.Log(\"round setup:\")\n\tfor _, round := range schedule {\n\t\tt.Logf(\"\\t%v\", round)\n\t}\n}\n\nfunc debugLogf(t *testing.T, format string, args ...interface{}) {\n\t// by default be silent, but log if checking a specific seed or force-enabled.\n\tif fuzzDebugLog || fuzzCustomSeed != 0 {\n\t\tt.Helper()\n\t\tt.Logf(format, args...)\n\t}\n}\n\nfunc assertRatelimitersBehaveSimilarly(t *testing.T, burst int, minWaitDuration time.Duration, schedule [][]string) {\n\trounds, eventsPerRound := len(schedule), len(schedule[0])\n\tlimit := rate.Every(fuzzGranularity * time.Duration(eventsPerRound)) // refresh after each full round\n\tt.Logf(\"limit: %v, burst: %v\", fuzzGranularity, burst)\n\n\tinitialNow := time.Now().Round(time.Millisecond) // easier to read when logging/debugging when rounded\n\tts := NewMockedTimeSourceAt(initialNow)\n\tcompressedTS := NewMockedTimeSourceAt(initialNow)\n\n\tactual := rate.NewLimiter(limit, burst)\n\twrapped := NewRatelimiter(limit, burst)\n\tmocked := NewRateLimiterWithTimeSource(ts, limit, burst)\n\tcompressed := NewRateLimiterWithTimeSource(compressedTS, limit, burst)\n\n\t// record \"to be executed\" closures on the time-compressed ratelimiter too,\n\t// so it can be checked at a sped up rate.\n\t// nil values mean \"nothing to do\".\n\tcompressedReplay := make([][]func(t *testing.T), rounds)\n\tfor i := range compressedReplay {\n\t\tcompressedReplay[i] = make([]func(t *testing.T), eventsPerRound)\n\t}\n\n\tticker := time.NewTicker(fuzzGranularity)\n\tdefer ticker.Stop()\n\tfor round := range schedule {\n\t\tround := round // for closure\n\t\tfor event := range schedule[round] {\n\t\t\tevent := event // for closure\n\t\t\t<-ticker.C\n\t\t\tts.Advance(fuzzGranularity) // may also be advanced inside wait\n\t\t\tdebugLogf(t, \"Tokens before round, real: %0.2f, wrapped: %0.2f, mocked: %0.2f\", actual.Tokens(), wrapped.Tokens(), mocked.Tokens())\n\n\t\t\tswitch schedule[round][event] {\n\t\t\tcase \"a\":\n\t\t\t\t// call Allow on everything\n\t\t\t\ta, w, m := actual.Allow(), wrapped.Allow(), mocked.Allow()\n\t\t\t\tassertAllowsAgree(t, \"Allow\", round, event, a, w, m, nil)\n\t\t\t\tcompressedReplay[round][event] = func(t *testing.T) {\n\t\t\t\t\tc := compressed.Allow()\n\t\t\t\t\tassertAllowsAgree(t, \"Allow\", round, event, a, w, m, &c)\n\t\t\t\t}\n\t\t\tcase \"r\":\n\t\t\t\t// call Reserve on everything, don't cancel any we acquired successfully\n\t\t\t\tnow := time.Now()\n\t\t\t\t_a, _w, _m := actual.ReserveN(now, 1), wrapped.Reserve(), mocked.Reserve()\n\t\t\t\ta, w, m := _a.OK() && _a.DelayFrom(now) == 0, _w.Allow(), _m.Allow()\n\n\t\t\t\t// un-reserve any that were not successful, we don't leave them dangling.\n\t\t\t\tif !a {\n\t\t\t\t\t_a.CancelAt(now)\n\t\t\t\t}\n\t\t\t\t_w.Used(w)\n\t\t\t\t_m.Used(m)\n\t\t\t\tassertAllowsAgree(t, \"Reserve\", round, event, a, w, m, nil)\n\t\t\t\tcompressedReplay[round][event] = func(t *testing.T) {\n\t\t\t\t\t_c := compressed.Reserve()\n\t\t\t\t\tc := _c.Allow()\n\t\t\t\t\t_c.Used(c)\n\t\t\t\t\tassertAllowsAgree(t, \"Reserve\", round, event, a, w, m, &c)\n\t\t\t\t}\n\t\t\tcase \"c\":\n\t\t\t\t// call Reserve on everything, and cancel them all\n\t\t\t\tnow := time.Now() // needed for the real ratelimiter to cancel successfully like the wrapper does\n\t\t\t\t_a, _w, _m := actual.ReserveN(now, 1), wrapped.Reserve(), mocked.Reserve()\n\t\t\t\t_a.CancelAt(now)\n\t\t\t\t_w.Used(false)\n\t\t\t\t_m.Used(false)\n\t\t\t\tt.Logf(\"round[%v][%v] Cancel:     canceled\", round, event)\n\t\t\t\tcompressedReplay[round][event] = func(t *testing.T) {\n\t\t\t\t\tcompressed.Reserve().Used(false)\n\t\t\t\t\tt.Logf(\"round[%v][%v] Cancel:     canceled\", round, event)\n\t\t\t\t}\n\t\t\tcase \"w\":\n\t\t\t\t/*\n\t\t\t\t\tTry a brief Wait on everything.\n\n\t\t\t\t\tctx should expire in the middle of the *next* event, so all three possibilities are exercised:\n\t\t\t\t\t  - sometimes unblocks immediately because a token is available\n\t\t\t\t\t  - sometimes waits because a token will become available soon\n\t\t\t\t\t  - sometimes unblocks immediately because a token will not become available in time\n\n\t\t\t\t\tyou can see waits logged like this, they're somewhat rare though.  otherwise wait times are ~0s:\n\t\t\t\t\t\tregular:\n\t\t\t\t\t\t\tratelimiter_test.go:402: Wait elapsed: 20ms, wrapped err: <nil>\n\t\t\t\t\t\t\tratelimiter_test.go:639: Advancing mock time by 20ms\n\t\t\t\t\t\t\tratelimiter_test.go:402: Wait elapsed: 20ms, mocked err: <nil>\n\t\t\t\t\t\t\tratelimiter_test.go:402: Wait elapsed: 20ms, actual err: <nil>\n\t\t\t\t\t\t\tratelimiter_test.go:655: round[0][9] Wait:       actual: true,  wrapped: true,  mocked: true,\n\t\t\t\t\t\t\tratelimiter_test.go:455:             Wait time:  actual: 20ms,  wrapped: 20ms,  mocked: 20ms,\n\t\t\t\t\t\tcompressed:\n\t\t\t\t\t\t\tratelimiter_test.go:672: Advancing mock time by 20ms\n\t\t\t\t\t\t\tratelimiter_test.go:402: Wait elapsed: 20ms, compressed err: <nil>\n\t\t\t\t\t\t\tratelimiter_test.go:684: round[0][9] Wait:       actual: true,  wrapped: true,  mocked: true,  compressed: true\n\t\t\t\t\t\t\tratelimiter_test.go:455:             Wait time:  actual: 20ms,  wrapped: 20ms,  mocked: 20ms,  compressed: 20ms\n\t\t\t\t*/\n\t\t\t\ttimeout := fuzzGranularity + (fuzzGranularity / 2)\n\t\t\t\tstarted := time.Now() // intentionally gathered outside the goroutines, to reveal goroutine-starting lag\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), timeout)\n\n\t\t\t\ta, w, m := false, false, false\n\t\t\t\tvar aLatency, wLatency, mLatency time.Duration\n\t\t\t\tvar g errgroup.Group\n\t\t\t\tg.Go(func() error {\n\t\t\t\t\t_a := actual.Wait(ctx)\n\t\t\t\t\taLatency = time.Since(started).Round(time.Millisecond)\n\t\t\t\t\tdebugLogf(t, \"Wait elapsed: %v, actual err: %v\", aLatency, _a)\n\t\t\t\t\ta = _a == nil\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\tg.Go(func() error {\n\t\t\t\t\t_w := wrapped.Wait(ctx)\n\t\t\t\t\twLatency = time.Since(started).Round(time.Millisecond)\n\t\t\t\t\tdebugLogf(t, \"Wait elapsed: %v, wrapped err: %v\", wLatency, _w)\n\t\t\t\t\tw = _w == nil\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\tg.Go(func() error {\n\t\t\t\t\t// mocked time needs something to advance it, or it'll block until timeout...\n\t\t\t\t\t// but don't advance time if it returned immediately, the others won't wait either.\n\t\t\t\t\tdone := make(chan struct{})\n\t\t\t\t\tgo func() {\n\t\t\t\t\t\tselect {\n\t\t\t\t\t\tcase <-time.After(fuzzGranularity):\n\t\t\t\t\t\t\tt.Logf(\"Advancing mock time by %v\", fuzzGranularity)\n\t\t\t\t\t\t\tts.Advance(fuzzGranularity)\n\t\t\t\t\t\tcase <-done:\n\t\t\t\t\t\t\t// do nothing, it is not waiting\n\t\t\t\t\t\t}\n\t\t\t\t\t}()\n\n\t\t\t\t\t_m := mocked.Wait(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t\tmLatency = time.Since(started).Round(time.Millisecond)\n\t\t\t\t\tdebugLogf(t, \"Wait elapsed: %v, mocked err: %v\", mLatency, _m)\n\t\t\t\t\tm = _m == nil\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t\t_ = g.Wait()\n\n\t\t\t\tassertAllowsAgree(t, \"Wait\", round, event, a, w, m, nil)\n\t\t\t\tassertLatenciesSimilar(t, minWaitDuration, timeout, aLatency, wLatency, mLatency, nil)\n\t\t\t\tcompressedReplay[round][event] = func(t *testing.T) {\n\t\t\t\t\t// need a mocked-time context, or the real deadline will not match the mocked deadline\n\t\t\t\t\tctx := &timesourceContext{\n\t\t\t\t\t\tts:       compressedTS,\n\t\t\t\t\t\tdeadline: compressedTS.Now().Add(fuzzGranularity + (fuzzGranularity / 2)),\n\t\t\t\t\t}\n\n\t\t\t\t\tdone := make(chan struct{})\n\t\t\t\t\tgo func() {\n\t\t\t\t\t\t// wait until blocked inside Wait\n\t\t\t\t\t\tcompressedTS.BlockUntil(1)\n\t\t\t\t\t\t// sleep briefly to encourage close(done) if racing, or blocked on something else\n\t\t\t\t\t\ttime.Sleep(fuzzGranularity / 10)\n\t\t\t\t\t\tselect {\n\t\t\t\t\t\tcase <-done:\n\t\t\t\t\t\t\t// do nothing, Wait did not block correctly.\n\t\t\t\t\t\t\t// this goroutine may leak if no blockers occur. it's \"fine\".\n\t\t\t\t\t\tdefault:\n\t\t\t\t\t\t\t// blocked inside Wait, advance time so it unblocks\n\t\t\t\t\t\t\tt.Logf(\"Advancing mock time by %v\", fuzzGranularity)\n\t\t\t\t\t\t\tcompressedTS.Advance(fuzzGranularity)\n\t\t\t\t\t\t}\n\t\t\t\t\t}()\n\n\t\t\t\t\tcompressedStarted := compressedTS.Now()\n\t\t\t\t\t_c := compressed.Wait(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t\tcLatency := time.Since(started).Round(time.Millisecond)\n\t\t\t\t\tcLatency = compressedTS.Since(compressedStarted)\n\t\t\t\t\tdebugLogf(t, \"Wait elapsed: %v, compressed err: %v\", cLatency, _c)\n\t\t\t\t\tc := _c == nil\n\t\t\t\t\tassertAllowsAgree(t, \"Wait\", round, event, a, w, m, &c)\n\t\t\t\t\tassertLatenciesSimilar(t, minWaitDuration, timeout, aLatency, wLatency, mLatency, &cLatency)\n\t\t\t\t}\n\t\t\t\tcancel()\n\t\t\tcase \"_\":\n\t\t\t\t// do nothing\n\t\t\tdefault:\n\t\t\t\tpanic(\"unhandled case: \" + schedule[round][event])\n\t\t\t}\n\n\t\t\tdebugLogf(t, \"Tokens after round, real: %0.2f, wrapped: %0.2f, mocked: %0.2f\", actual.Tokens(), wrapped.Tokens(), mocked.Tokens())\n\t\t}\n\t}\n\tt.Run(\"compressed time\", func(t *testing.T) {\n\t\t// and now replay the compressed ratelimiter and make sure it matches too,\n\t\t// as time-compressed must behave the same as real-time.\n\t\t//\n\t\t// this is primarily intended to detect cases where real-time is accidentally used,\n\t\t// as ~zero time actually passes, which is quite different from the above tests.\n\t\t//\n\t\t// it's not perfect, but it does eventually notice such bugs, as you can see by\n\t\t// changing literally any timesource.Now() calls into time.Now() and running\n\t\t// tests a few times.\n\t\t// depending on the change it might not notice in most tests, but eventually a\n\t\t// problematic combination is triggered and can be replayed with the logged seed.\n\t\tfor round := range schedule {\n\t\t\tfor event := range schedule[round] {\n\t\t\t\tcompressedTS.Advance(fuzzGranularity)\n\t\t\t\tdebugLogf(t, \"Tokens before compressed round: %0.2f\", compressed.Tokens())\n\t\t\t\treplay := compressedReplay[round][event]\n\t\t\t\tif replay != nil {\n\t\t\t\t\treplay(t)\n\t\t\t\t}\n\t\t\t\tdebugLogf(t, \"Tokens after compressed round: %0.2f\", compressed.Tokens())\n\t\t\t}\n\t\t}\n\t})\n}\n\n// broken out to consts to help keep padding the same, for easier visual scanning\nconst (\n\tsimilarBehaviorLogComponents  = \"%-11s %-14s %-15s %-14s\" // \"{what:} {actual} {wrapped} {mocked}\" with padding\n\tsimilarBehaviorLogNormalRound = \"round[%v][%v] \" + similarBehaviorLogComponents\n)\n\nfunc assertAllowsAgree(t *testing.T, what string, round, event int, actual, wrapped, mocked bool, compressed *bool) {\n\tt.Helper()\n\tsWhat := what + \":\"\n\tsActual := fmt.Sprintf(\"actual: %v,\", actual)\n\tsWrapped := fmt.Sprintf(\"wrapped: %v,\", wrapped)\n\tsMocked := fmt.Sprintf(\"mocked: %v,\", mocked)\n\t// format strings should share padding so output aligns and it's easy to read.\n\t// unfortunately only strings do this nicely.s\n\tif compressed != nil {\n\t\tsCompressed := \"compressed: \" + fmt.Sprint(*compressed)\n\t\tt.Logf(similarBehaviorLogNormalRound+\" %s\", round, event, sWhat, sActual, sWrapped, sMocked, sCompressed)\n\t\tassert.True(t, actual == wrapped && wrapped == mocked && mocked == *compressed, \"ratelimiters disagree\")\n\t} else {\n\t\tt.Logf(similarBehaviorLogNormalRound, round, event, sWhat, sActual, sWrapped, sMocked)\n\t\tassert.True(t, actual == wrapped && wrapped == mocked, \"ratelimiters disagree\")\n\t}\n}\n\nfunc assertLatenciesSimilar(t *testing.T, mustBeGreaterThan, mustBeLessThan, actual, wrapped, mocked time.Duration, compressed *time.Duration) {\n\tt.Helper()\n\t// none are currently expected to wait, but if they do, they must not wait the full timeout.\n\t// this also helps handle small cpu stutter, as some may wait 1ms or so.\n\tmaxLatency := maxDur(actual, wrapped, mocked)\n\tminLatency := minDur(actual, wrapped, mocked)\n\tcompressedString := \"\"\n\tif compressed != nil {\n\t\tmaxLatency = maxDur(maxLatency, *compressed)\n\t\tminLatency = minDur(minLatency, *compressed)\n\t\tcompressedString = fmt.Sprintf(\"compressed: %v\", *compressed)\n\t}\n\tassert.GreaterOrEqualf(t, minLatency, mustBeGreaterThan, \"minimum latency must be within bounds, else waits did not wait long enough\")\n\tassert.LessOrEqualf(t, maxLatency, mustBeLessThan, \"maximum latency must not exceed timeout, it implies wait did not unblock when it should\")\n\n\t// log format tries to align with assertAllowsAgree\n\tindent := strings.Repeat(\" \", len(\"round[1][1]\"))\n\tsActual := fmt.Sprintf(\"actual: %v,\", actual)\n\tsWrapped := fmt.Sprintf(\"wrapped: %v,\", wrapped)\n\tsMocked := fmt.Sprintf(\"mocked: %v,\", mocked)\n\tt.Logf(\"%s \"+similarBehaviorLogComponents+\" %s\",\n\t\tindent, \"Wait time:\", sActual, sWrapped, sMocked, compressedString)\n\n\t// asserting is... tough.  long-ish noise latencies are somewhat common,\n\t// sometimes even exceeding 10ms to schedule a goroutine.\n\t//\n\t// if this becomes too noisy, just switch to logging, and check by hand\n\t// when making changes.\n\tif minLatency < fuzzGranularity/2 {\n\t\t// \"no wait\" test, assert that they are all under this fuzzGranularity\n\t\tassertNoWait := func(what string, wait time.Duration) {\n\t\t\tt.Helper()\n\t\t\tassert.LessOrEqual(t, wait, fuzzGranularity/2, \"%v ratelimiter waited too long\", what)\n\t\t}\n\t\tassertNoWait(\"actual\", actual)\n\t\tassertNoWait(\"wrapped\", wrapped)\n\t\tassertNoWait(\"mocked\", mocked)\n\t\tif compressed != nil {\n\t\t\tassertNoWait(\"compressed\", *compressed)\n\t\t}\n\t} else {\n\t\t// \"wait\" test, assert that they all waited, and none waited significantly longer\n\t\tassertWaited := func(what string, wait time.Duration) {\n\t\t\tt.Helper()\n\t\t\tassert.True(t,\n\t\t\t\twait > fuzzGranularity/2 && wait < fuzzGranularity+(fuzzGranularity/2),\n\t\t\t\t\"%v waited an incorrect amount of time, should be %v < %v <%v\",\n\t\t\t\twhat, fuzzGranularity/2, wait, fuzzGranularity+(fuzzGranularity/2))\n\t\t}\n\t\tassertWaited(\"actual\", actual)\n\t\tassertWaited(\"wrapped\", wrapped)\n\t\tassertWaited(\"mocked\", mocked)\n\t\tif compressed != nil {\n\t\t\tassertWaited(\"compressed\", *compressed)\n\t\t}\n\t}\n}\n\nfunc maxDur(d time.Duration, ds ...time.Duration) time.Duration {\n\tfor _, tmp := range ds {\n\t\tif tmp > d {\n\t\t\td = tmp\n\t\t}\n\t}\n\treturn d\n}\nfunc minDur(d time.Duration, ds ...time.Duration) time.Duration {\n\tfor _, tmp := range ds {\n\t\tif tmp < d {\n\t\t\td = tmp\n\t\t}\n\t}\n\treturn d\n}\n"
  },
  {
    "path": "common/clock/ratelimiter_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/clock (interfaces: Reservation)\n//\n// Generated by this command:\n//\n//\tmockgen -package=clock -destination=ratelimiter_mock.go github.com/uber/cadence/common/clock Reservation\n//\n\n// Package clock is a generated GoMock package.\npackage clock\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockReservation is a mock of Reservation interface.\ntype MockReservation struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReservationMockRecorder\n\tisgomock struct{}\n}\n\n// MockReservationMockRecorder is the mock recorder for MockReservation.\ntype MockReservationMockRecorder struct {\n\tmock *MockReservation\n}\n\n// NewMockReservation creates a new mock instance.\nfunc NewMockReservation(ctrl *gomock.Controller) *MockReservation {\n\tmock := &MockReservation{ctrl: ctrl}\n\tmock.recorder = &MockReservationMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockReservation) EXPECT() *MockReservationMockRecorder {\n\treturn m.recorder\n}\n\n// Allow mocks base method.\nfunc (m *MockReservation) Allow() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Allow\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Allow indicates an expected call of Allow.\nfunc (mr *MockReservationMockRecorder) Allow() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Allow\", reflect.TypeOf((*MockReservation)(nil).Allow))\n}\n\n// Used mocks base method.\nfunc (m *MockReservation) Used(wasUsed bool) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Used\", wasUsed)\n}\n\n// Used indicates an expected call of Used.\nfunc (mr *MockReservationMockRecorder) Used(wasUsed any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Used\", reflect.TypeOf((*MockReservation)(nil).Used), wasUsed)\n}\n"
  },
  {
    "path": "common/clock/ratelimiter_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/atomic\"\n\t\"golang.org/x/sync/errgroup\"\n\t\"golang.org/x/time/rate\"\n)\n\nfunc TestRatelimiter(t *testing.T) {\n\tt.Parallel()\n\n\tfor _, name := range []string{\"real\", \"mocked\"} {\n\t\tname := name\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tts := func() MockedTimeSource { return nil }\n\t\t\tif name == \"mocked\" {\n\t\t\t\tts = NewMockedTimeSource\n\t\t\t}\n\t\t\t// needs to be a bit coarser than the comparison tests, because time\n\t\t\t// is less tightly controlled / mock time relies on a background loop.\n\t\t\tgranularity := 100 * time.Millisecond\n\t\t\tassertRatelimiterBasicsWork(t, ts, granularity)\n\t\t})\n\t}\n}\n\n// check relatively basic properties of the ratelimiter wrapper,\n// and make sure that it behaves the same with both real time (makeTimesource returns nil)\n// and with mocked time (simulated by advancing mock-time in a background goroutine).\n//\n// broken out because it's just very deeply indented otherwise.\nfunc assertRatelimiterBasicsWork(t *testing.T, makeTimesource func() MockedTimeSource, granularity time.Duration) {\n\t// all \"events\" take place at 2*granularity\n\tevent := granularity * 2\n\n\tmakeLimiter := func(ts MockedTimeSource, limit rate.Limit, burst int) (rl Ratelimiter, sleep func(time.Duration)) {\n\t\tif ts == nil {\n\t\t\treturn NewRatelimiter(limit, burst), time.Sleep\n\t\t}\n\t\treturn NewRateLimiterWithTimeSource(ts, limit, burst), ts.Advance\n\t}\n\n\tnow := func(ts MockedTimeSource) time.Time {\n\t\tif ts == nil {\n\t\t\treturn time.Now()\n\t\t}\n\t\treturn ts.Now()\n\t}\n\n\t// creates a context that will time out with the passed timesource, or real time if nil\n\tcontextWithTimeout := func(ts TimeSource, dur time.Duration) (ctx context.Context, cancel context.CancelFunc) {\n\t\tif ts == nil {\n\t\t\t// could be a timesourceContext with a real timesource,\n\t\t\t// but I'm avoiding that until it's fleshed out fully and tested separately.\n\t\t\treturn context.WithTimeout(context.Background(), dur)\n\t\t}\n\t\tctx = &timesourceContext{\n\t\t\tts:       ts,\n\t\t\tdeadline: ts.Now().Add(dur),\n\t\t}\n\t\treturn ctx, func() {}\n\t}\n\n\t// advances mock time in the background until stopped, at [granularity/10] per millisecond tick.\n\t// currently this leads to tests that are about 10x faster than real time, but \"faster\" is not\n\t// a goal in this context.\n\t//\n\t// \"different\" is however good for making sure we don't rely on real time somehow,\n\t// so try to keep it either faster or slower, whatever makes for a decently non-flaky suite.\n\t//\n\t// flawed ratelimiter behavior when using mock time should also cause the fuzz test to fail\n\t// eventually, so failures here are, hopefully, only due to flawed tests or excessive noise.\n\tadvanceTime := func(ts MockedTimeSource) (cleanup func()) {\n\t\tif ts == nil {\n\t\t\treturn func() {} // nothing to do\n\t\t}\n\t\tstop, done := make(chan struct{}), make(chan struct{})\n\t\tgo func() {\n\t\t\tdefer close(done)\n\t\t\t// this is not a great way to do mock-time tests, but it does automatically\n\t\t\t// work in just about all cases, without needing customization.\n\t\t\t//\n\t\t\t// ideally, in other kinds of tests, you should wait for ts.BlockUntil(..)\n\t\t\t// or some other event, and then advance time semi-precisely, and fall back\n\t\t\t// to this strategy only when that isn't usable for some reason.\n\t\t\t//\n\t\t\t// one extremely nice quality of mocked time though, even when done like this:\n\t\t\t// time simply pauses while debugging, and nothing \"artificially\" times out\n\t\t\t// due to being blocked, regardless of how long you let things sit.\n\t\t\tticker := time.NewTicker(time.Millisecond)\n\t\t\tdefer ticker.Stop()\n\t\t\tfor {\n\t\t\t\tselect {\n\t\t\t\tcase <-stop:\n\t\t\t\t\treturn\n\t\t\t\tcase <-ticker.C:\n\t\t\t\t\tts.Advance(granularity / 10)\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t\treturn func() {\n\t\t\tclose(stop)\n\t\t\t<-done\n\t\t}\n\t}\n\n\tt.Run(\"sets burst and limit\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\t// should set on construction\n\t\trl, _ := makeLimiter(makeTimesource(), rate.Every(time.Second), 3)\n\t\tassert.EqualValues(t, 3, rl.Burst())\n\t\tassert.Equal(t, rate.Every(time.Second), rl.Limit())\n\n\t\t// and when calling the setters\n\t\trl.SetBurst(5)\n\t\trl.SetLimit(rate.Every(time.Millisecond))\n\t\tassert.EqualValues(t, 5, rl.Burst())\n\t\tassert.Equal(t, rate.Every(time.Millisecond), rl.Limit())\n\t})\n\tt.Run(\"limits requests to within available burst tokens\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\trl, _ := makeLimiter(makeTimesource(), rate.Every(time.Second), 3)\n\t\tassert.EqualValues(t, 3, rl.Tokens(), \"should have tokens to allow exactly 3 calls, and not exceed burst of 3\")\n\t\tassert.True(t, rl.Allow())\n\t\tassert.InDeltaf(t, 2, rl.Tokens(), 0.1, \"Allow should consume a token\")\n\t\tassert.True(t, rl.Allow())\n\t\tassert.InDelta(t, 1, rl.Tokens(), 0.1)\n\t\tassert.True(t, rl.Allow())\n\t\tassert.InDelta(t, 0, rl.Tokens(), 0.1)\n\t\tassert.False(t, rl.Allow(), \"should not allow after burstable tokens are consumed\")\n\t\tassert.InDelta(t, 0, rl.Tokens(), 0.1)\n\t\tassert.False(t, rl.Allow())\n\t})\n\tt.Run(\"recovers tokens as time passes\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\trl, sleep := makeLimiter(makeTimesource(), rate.Every(event), 1)\n\t\tassert.True(t, rl.Allow())\n\t\tassert.False(t, rl.Allow())\n\t\tsleep(event)\n\t\tassert.True(t, rl.Allow(), \"should have recovered one token\")\n\t\tassert.False(t, rl.Allow())\n\t})\n\tt.Run(\"waits until tokens are available\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tts := makeTimesource()\n\t\trl, _ := makeLimiter(ts, rate.Every(event), 1)\n\t\tstarted := now(ts) // must be collected before Allow, or sleep time could legitimately be lower than the rate\n\t\tassert.True(t, rl.Allow())\n\t\tassert.False(t, rl.Allow())\n\n\t\tstopTime := advanceTime(ts)\n\t\tctx, cancel := contextWithTimeout(ts, event)\n\t\tdefer cancel()\n\t\terr := rl.Wait(ctx)\n\t\tstopTime()\n\n\t\tassert.NoError(t, err, \"Wait should have waited and then allowed the event\")\n\t\telapsed := now(ts).Sub(started)\n\t\tassert.Truef(t, elapsed >= event-granularity && elapsed < event+granularity,\n\t\t\t\"elapsed time outside bounds, should be %v <= %v < %v\",\n\t\t\tevent-granularity, elapsed, event+granularity)\n\n\t\tassert.False(t, rl.Allow(), \"Wait must consume a token\")\n\t})\n\tt.Run(\"reserve\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\t// simple stuff\n\t\tfor name, tc := range map[string]struct {\n\t\t\tdo          func(t *testing.T, rl Ratelimiter, sleep func(time.Duration))\n\t\t\ttokenChange float64\n\t\t}{\n\t\t\t\"consumes a token\": {\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, sleep func(time.Duration)) {\n\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\tassert.True(t, r.Allow())\n\t\t\t\t\tr.Used(true)\n\t\t\t\t},\n\t\t\t\ttokenChange: -1,\n\t\t\t},\n\t\t\t\"restores a token when canceling\": {\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, sleep func(time.Duration)) {\n\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\tassert.True(t, r.Allow())\n\t\t\t\t\tr.Used(false)\n\t\t\t\t},\n\t\t\t\ttokenChange: 0,\n\t\t\t},\n\t\t\t\"does not restore if there are interleaving calls\": {\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, sleep func(time.Duration)) {\n\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\tassert.True(t, r.Allow())\n\t\t\t\t\tsleep(1) // advance time by any amount\n\t\t\t\t\trl.Allow()\n\t\t\t\t\tr.Used(false)\n\t\t\t\t},\n\t\t\t\ttokenChange: -2, // reservation did not restore back to -1\n\t\t\t},\n\t\t\t\"does not restore if Tokens is called in before canceling\": {\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, sleep func(time.Duration)) {\n\t\t\t\t\t// to make a possibly-surprising quirk explicit:\n\t\t\t\t\t// tokens advances the limiter's \"now\" value, which means\n\t\t\t\t\t// canceling an allowed token does not work.\n\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\tassert.True(t, r.Allow())\n\t\t\t\t\tsleep(1)    // advance time by any amount\n\t\t\t\t\trl.Tokens() // advance time internally\n\t\t\t\t\tr.Used(false)\n\t\t\t\t},\n\t\t\t\ttokenChange: -1, // -1 due to Tokens call\n\t\t\t},\n\t\t\t\"does not go negative\": {\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, sleep func(time.Duration)) {\n\t\t\t\t\trl.Allow()\n\t\t\t\t\trl.Allow()\n\t\t\t\t\tassert.InDeltaf(t, 0, rl.Tokens(), 0.1, \"should have consumed all tokens\")\n\t\t\t\t\tr := rl.Reserve()\n\t\t\t\t\tassert.False(t, r.Allow())\n\t\t\t\t\tr.Used(false)\n\t\t\t\t},\n\t\t\t\ttokenChange: -2, // would be -3 if it went negative\n\t\t\t},\n\t\t} {\n\t\t\tname, tc := name, tc\n\t\t\tt.Run(name, func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tts := makeTimesource()\n\t\t\t\trl, sleep := makeLimiter(ts, rate.Every(event), 2)\n\n\t\t\t\tbefore := rl.Tokens()\n\t\t\t\tstopTime := advanceTime(ts)\n\t\t\t\ttc.do(t, rl, sleep)\n\t\t\t\tstopTime()\n\t\t\t\tafter := rl.Tokens()\n\n\t\t\t\tassert.InDeltaf(t, tc.tokenChange, after-before, 0.1, \"tokens should have changed from %v to %v, but was %v\", before, before+tc.tokenChange, after)\n\t\t\t})\n\t\t}\n\t})\n\tt.Run(\"wait\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\twait := func(ts MockedTimeSource, periods float64, rl Ratelimiter) int64 {\n\t\t\tctx, cancel := contextWithTimeout(ts, time.Duration(periods*float64(event)))\n\t\t\tdefer cancel()\n\t\t\tif rl.Wait(ctx) == nil {\n\t\t\t\treturn 1\n\t\t\t}\n\t\t\treturn 0\n\t\t}\n\t\tparWait := func(ts MockedTimeSource, periods float64, count *atomic.Int64, rl Ratelimiter) {\n\t\t\tif wait(ts, periods, rl) == 1 {\n\t\t\t\tcount.Inc()\n\t\t\t}\n\t\t}\n\t\tfor name, tc := range map[string]struct {\n\t\t\tdrainFirst       bool\n\t\t\tdo               func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64\n\t\t\tlatency          float64 // num of [granularity]s that should elapse\n\t\t\tallowed          int64\n\t\t\ttokenChange      float64\n\t\t\tallowLowerTokens bool // concurrent waits cannot be guaranteed to return tokens, negatives are possible\n\t\t}{\n\t\t\t\"allows immediately with free tokens\": {\n\t\t\t\tdrainFirst: false,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\treturn wait(ts, 3, rl) // anything longer than 1\n\t\t\t\t},\n\t\t\t\tallowed:     1,\n\t\t\t\ttokenChange: -1,\n\t\t\t},\n\t\t\t\"fails immediately with too-short timeout\": {\n\t\t\t\tdrainFirst: true,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\treturn wait(ts, 0.1, rl)\n\t\t\t\t},\n\t\t\t\tlatency: 0,\n\t\t\t\tallowed: 0,\n\t\t\t},\n\t\t\t\"waits for one token\": {\n\t\t\t\tdrainFirst: true,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\treturn wait(ts, 2, rl)\n\t\t\t\t},\n\t\t\t\tlatency: 1,\n\t\t\t\tallowed: 1,\n\t\t\t},\n\t\t\t\"concurrently waits for 2 tokens\": {\n\t\t\t\tdrainFirst: true,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\tcount := atomic.Int64{}\n\t\t\t\t\tvar g errgroup.Group\n\t\t\t\t\tfor i := 0; i < 2; i++ {\n\t\t\t\t\t\tg.Go(func() error {\n\t\t\t\t\t\t\tparWait(ts, 2.5, &count, rl) // slightly longer than 2\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t\tg.Wait()\n\t\t\t\t\treturn count.Load()\n\t\t\t\t},\n\t\t\t\tlatency: 2,\n\t\t\t\tallowed: 2,\n\t\t\t},\n\t\t\t\"one wins and one fails when concurrently waiting\": {\n\t\t\t\tdrainFirst: true,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\tcount := atomic.Int64{}\n\t\t\t\t\tvar g errgroup.Group\n\t\t\t\t\tfor i := 0; i < 2; i++ {\n\t\t\t\t\t\tg.Go(func() error {\n\t\t\t\t\t\t\tparWait(ts, 1.5, &count, rl) // slightly longer than 1\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t\tg.Wait()\n\t\t\t\t\t// only one can win\n\t\t\t\t\treturn count.Load()\n\t\t\t\t},\n\t\t\t\tlatency:          1,\n\t\t\t\tallowed:          1,\n\t\t\t\tallowLowerTokens: true, // tokens are not guaranteed to be returned\n\t\t\t},\n\t\t\t\"immediately drains 2, waits for 2, fails 2\": {\n\t\t\t\tdrainFirst: false,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\tcount := atomic.Int64{}\n\t\t\t\t\tvar g errgroup.Group\n\t\t\t\t\tfor i := 0; i < 6; i++ {\n\t\t\t\t\t\tg.Go(func() error {\n\t\t\t\t\t\t\tparWait(ts, 2.5, &count, rl) // slightly longer than 2, so 2 waiters succeed\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t\tg.Wait()\n\t\t\t\t\treturn count.Load()\n\t\t\t\t},\n\t\t\t\tlatency:          2,\n\t\t\t\tallowed:          2,\n\t\t\t\ttokenChange:      -2,\n\t\t\t\tallowLowerTokens: true, // tokens are not guaranteed to be returned\n\t\t\t},\n\t\t\t\"cancel while waiting\": {\n\t\t\t\tdrainFirst: true,\n\t\t\t\tdo: func(t *testing.T, rl Ratelimiter, ts MockedTimeSource) int64 {\n\t\t\t\t\tvar wg sync.WaitGroup\n\t\t\t\t\tcount := atomic.Int64{}\n\t\t\t\t\twg.Add(2)\n\t\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), event*3/2)\n\t\t\t\t\tgo func() {\n\t\t\t\t\t\tdefer wg.Done()\n\t\t\t\t\t\t// a bit too much of a mess to thread this through nicely\n\t\t\t\t\t\tif ts == nil {\n\t\t\t\t\t\t\t<-time.After(event / 2) // cancel half way through the wait\n\t\t\t\t\t\t\tcancel()\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tts.AfterFunc(event/2, func() {\n\t\t\t\t\t\t\t\tcancel()\n\t\t\t\t\t\t\t})\n\t\t\t\t\t\t}\n\t\t\t\t\t}()\n\t\t\t\t\tgo func() {\n\t\t\t\t\t\tdefer wg.Done()\n\t\t\t\t\t\tif rl.Wait(ctx) == nil {\n\t\t\t\t\t\t\tcount.Inc()\n\t\t\t\t\t\t}\n\t\t\t\t\t}()\n\t\t\t\t\twg.Wait()\n\t\t\t\t\treturn count.Load()\n\t\t\t\t},\n\t\t\t\tallowed:     0,\n\t\t\t\tlatency:     0.5, // unfortunately allows 0 currently, but token change should ensure time passed\n\t\t\t\ttokenChange: 0.5, // only waited for half an event -> recovered half a token\n\t\t\t},\n\t\t} {\n\t\t\tname, tc := name, tc\n\t\t\tt.Run(name, func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tts := makeTimesource()\n\t\t\t\trl, _ := makeLimiter(ts, rate.Every(granularity*2), 2)\n\t\t\t\tif tc.drainFirst {\n\t\t\t\t\tassert.True(t, rl.Allow())\n\t\t\t\t\tassert.True(t, rl.Allow())\n\t\t\t\t\tassert.False(t, rl.Allow())\n\t\t\t\t}\n\n\t\t\t\tbefore := rl.Tokens()\n\t\t\t\tstart := now(ts)\n\t\t\t\tstopTime := advanceTime(ts)\n\t\t\t\ttc.do(t, rl, ts)\n\t\t\t\tstopTime()\n\t\t\t\tafter := rl.Tokens()\n\t\t\t\telapsed := now(ts).Sub(start)\n\t\t\t\ttarget := time.Duration(tc.latency * float64(granularity*2))\n\n\t\t\t\tif tc.allowLowerTokens {\n\t\t\t\t\t// some tests can reasonably fail to restore a used token, so they can go lower than would be ideal\n\t\t\t\t\tassert.True(t, after-before <= tc.tokenChange+0.5, \"tokens should have changed from %0.2f to ~%0.2f (or lower), but was %0.2f -> %0.2f\", before, before+tc.tokenChange, before, after)\n\t\t\t\t} else {\n\t\t\t\t\t// tokens should be fairly precise (cpu noise can make this fuzzy)\n\t\t\t\t\tassert.InDeltaf(t, tc.tokenChange, after-before, 0.5, \"tokens should have changed from %0.2f to %0.2f (+/- 0.5 for cpu noise), but was %0.2f -> %0.2f\", before, before+tc.tokenChange, before, after)\n\t\t\t\t}\n\t\t\t\tassert.True(t,\n\t\t\t\t\telapsed > (target-granularity) && elapsed < (target+granularity),\n\t\t\t\t\t\"should have taken %v < (actual) %v < %v\",\n\t\t\t\t\ttarget-granularity, elapsed, target+granularity,\n\t\t\t\t)\n\t\t\t})\n\t\t}\n\t})\n}\n\nfunc TestRatelimiterCoverage(t *testing.T) {\n\tt.Run(\"time paradox\", func(t *testing.T) {\n\t\t// this test's behavior should not be possible, as any reservation with\n\t\t// a time in the future must have been reserved in the future, which\n\t\t// would have advanced the internal latestNow to match that future-time.\n\t\trl := NewRatelimiter(rate.Every(time.Second), 1)\n\t\timpl := rl.(*ratelimiter)\n\t\tr := rl.Reserve()\n\t\tlatestNow := impl.latestNow\n\t\trequire.True(t, r.Allow())\n\t\trimpl := r.(*allowedReservation)\n\t\t// cheat, put the reservation into the future.\n\t\trimpl.reservedAt = time.Now().Add(time.Second)\n\t\trimpl.Used(false)\n\t\tassert.True(t, latestNow.Before(impl.latestNow), \"ratelimiter-internal time should have advanced despite impossible sequence\")\n\t})\n\tt.Run(\"zero burst\", func(t *testing.T) {\n\t\t// we do not currently make use of this, but it's allowed by the API.\n\t\trl := NewRatelimiter(rate.Every(time.Second), 0)\n\t\tctx, cancel := context.WithTimeout(context.Background(), time.Hour) // much longer than the rate\n\t\tdefer cancel()\n\t\tstarted := time.Now()\n\t\tassert.Error(t, rl.Wait(ctx), \"0 burst should never allow events, so it should fail immediately\")\n\t\telapsed := time.Since(started)\n\t\tassert.Less(t, elapsed, time.Second/10, \"Wait should have returned almost immediately on impossible waits\")\n\t})\n\tt.Run(\"mock limiter constructor\", func(t *testing.T) {\n\t\t// covered by fuzz testing, but this gets it to 100% without fuzz.\n\t\t_ = NewRateLimiterWithTimeSource(NewMockedTimeSource(), 1, 1)\n\t})\n}\n\n// context which becomes Done() based on a TimeSource instead of real time\ntype timesourceContext struct {\n\t// does not contain a parent context as we currently have no need,\n\t// but a \"real\" one would for forwarding Value and deadline lookups.\n\tts       TimeSource\n\tdeadline time.Time\n\tmut      sync.Mutex\n}\n\nfunc (t *timesourceContext) Deadline() (deadline time.Time, ok bool) {\n\tt.mut.Lock()\n\tdefer t.mut.Unlock()\n\treturn t.deadline, !t.deadline.IsZero()\n}\n\nfunc (t *timesourceContext) Done() <-chan struct{} {\n\t// not currently expected to be used, but it would look like this:\n\tt.mut.Lock()\n\tdefer t.mut.Unlock()\n\tc := make(chan struct{})\n\tdelay := t.deadline.Sub(t.ts.Now())\n\tt.ts.AfterFunc(delay, func() {\n\t\t// this stack may leak if time is not advanced past it in tests.\n\t\tclose(c)\n\t})\n\treturn c\n}\n\nfunc (t *timesourceContext) Err() error {\n\tt.mut.Lock()\n\tdefer t.mut.Unlock()\n\tif t.ts.Now().After(t.deadline) {\n\t\treturn context.DeadlineExceeded\n\t}\n\treturn nil\n}\n\nfunc (t *timesourceContext) Value(key any) any {\n\tpanic(\"unimplemented\")\n}\n\nvar _ context.Context = (*timesourceContext)(nil)\n"
  },
  {
    "path": "common/clock/real_timer_benchmark_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"testing\"\n\t\"time\"\n)\n\nfunc BenchmarkRealTimerGate(b *testing.B) {\n\ttimer := NewTimerGate(NewRealTimeSource())\n\n\tfor i := 0; i < b.N; i++ {\n\t\ttimer.Update(time.Now())\n\t}\n}\n"
  },
  {
    "path": "common/clock/sustain.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport \"time\"\n\n// Sustain tracks whether a boolean value is consistently true over a dynamic duration. It does this by recording the earliest\n// time that it received a true value, and clearing that timestamp any time that a false value is encountered.\n// The timestamp is only initialized when a true datapoint is accepted.\ntype Sustain struct {\n\tstarted  time.Time\n\tsource   TimeSource\n\tduration func() time.Duration\n}\n\nfunc NewSustain(source TimeSource, duration func() time.Duration) Sustain {\n\treturn Sustain{\n\t\tsource:   source,\n\t\tduration: duration,\n\t}\n}\n\n// Check accepts a datapoint and returns true if the condition has been sustained. For example, if the duration was\n// 60s, and a true datapoint was accepted every second, it would return false until 60s had elapsed from the first datapoint\n// and then subsequently return true.\nfunc (s *Sustain) Check(value bool) bool {\n\tif value {\n\t\tnow := s.source.Now()\n\t\tif s.started.IsZero() {\n\t\t\ts.started = now\n\t\t}\n\t\tif now.Sub(s.started) >= s.duration() {\n\t\t\treturn true\n\t\t}\n\t} else {\n\t\ts.Reset()\n\t}\n\treturn false\n}\n\n// CheckAndReset accepts a datapoint and returns true if the condition has been sustained.\n// If the condition has been sustained the timestamp is set to the current time so that it will be considered sustained\n// again until after the duration again elapses.\n// For example, if the duration was 60s, and a true datapoint was accepted every second, it would return true once every 60s and\n// otherwise return false\nfunc (s *Sustain) CheckAndReset(value bool) bool {\n\tif value {\n\t\tnow := s.source.Now()\n\t\tif s.started.IsZero() {\n\t\t\ts.started = now\n\t\t}\n\t\tif now.Sub(s.started) >= s.duration() {\n\t\t\ts.started = now\n\t\t\treturn true\n\t\t}\n\t} else {\n\t\ts.Reset()\n\t}\n\treturn false\n}\n\n// Reset clears the datapoints that the Sustain has received. It is equivalent to providing a false datapoint\nfunc (s *Sustain) Reset() {\n\ts.started = time.Time{}\n}\n"
  },
  {
    "path": "common/clock/sustain_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype check struct {\n\tseconds int\n\tvalue   bool\n}\n\nfunc TestCheckAndReset(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tduration time.Duration\n\t\tcalls    []check\n\t\texpected []bool\n\t}{\n\t\t{\n\t\t\tname:     \"simple case\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{10, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"intermediate successes\",\n\t\t\tduration: 10 * time.Second,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"resets after success\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{10, true},\n\t\t\t\t{0, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t\tfalse,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"resets after false\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{1, false},\n\t\t\t\t{1, true},\n\t\t\t\t{9, true},\n\t\t\t\t{1, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"duration = 0\",\n\t\t\tduration: 0,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tclock := NewMockedTimeSource()\n\t\t\tsus := NewSustain(clock, func() time.Duration {\n\t\t\t\treturn tc.duration\n\t\t\t})\n\t\t\trequire.Equal(t, len(tc.calls), len(tc.expected))\n\t\t\tfor i, c := range tc.calls {\n\t\t\t\texpected := tc.expected[i]\n\t\t\t\tclock.Advance(time.Duration(c.seconds) * time.Second)\n\t\t\t\tactual := sus.CheckAndReset(c.value)\n\t\t\t\tassert.Equal(t, expected, actual, \"check %d\", i)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCheck(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tduration time.Duration\n\t\tcalls    []check\n\t\texpected []bool\n\t}{\n\t\t{\n\t\t\tname:     \"simple case\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{10, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"intermediate successes\",\n\t\t\tduration: 10 * time.Second,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t\t{2, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"stays after success\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{10, true},\n\t\t\t\t{0, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"resets after false\",\n\t\t\tduration: time.Second * 10,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t\t{1, false},\n\t\t\t\t{1, true},\n\t\t\t\t{9, true},\n\t\t\t\t{1, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\tfalse,\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"duration = 0\",\n\t\t\tduration: 0,\n\t\t\tcalls: []check{\n\t\t\t\t{0, true},\n\t\t\t},\n\t\t\texpected: []bool{\n\t\t\t\ttrue,\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tclock := NewMockedTimeSource()\n\t\t\tsus := NewSustain(clock, func() time.Duration {\n\t\t\t\treturn tc.duration\n\t\t\t})\n\t\t\trequire.Equal(t, len(tc.calls), len(tc.expected))\n\t\t\tfor i, c := range tc.calls {\n\t\t\t\texpected := tc.expected[i]\n\t\t\t\tclock.Advance(time.Duration(c.seconds) * time.Second)\n\t\t\t\tactual := sus.Check(c.value)\n\t\t\t\tassert.Equal(t, expected, actual, \"check %d\", i)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/clock/testdata/amd_linux_go1.22.txt",
    "content": "// On a giant cloud machine with 96 cores (shared machine, possibly noisy but I tried to pick a quiet period and reran until no obvious outliers):\n\n❯ go version\ngo version go1.22.1 linux/amd64\n\n❯ go test -bench . -test.run xxx -cpu 1,2,4,8,32,128 .\ngoos: linux\ngoarch: amd64\npkg: github.com/uber/cadence/common/clock\ncpu: AMD EPYC 7B13\nBenchmarkLimiter/allow/serial/real             \t10606200\t       113.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   25.57µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     258ns\n    ratelimiter_test.go:116: allowed: 2125,       denied: 7875,        time per allow:     531ns\n    ratelimiter_test.go:116: allowed: 114089,     denied: 885911,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1203727,    denied: 9402473,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-2           \t10615315\t       113.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   16.63µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     263ns\n    ratelimiter_test.go:116: allowed: 2144,       denied: 7856,        time per allow:     536ns\n    ratelimiter_test.go:116: allowed: 113999,     denied: 886001,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1210150,    denied: 9405165,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-4           \t10230478\t       113.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   44.18µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     395ns\n    ratelimiter_test.go:116: allowed: 2161,       denied: 7839,        time per allow:     542ns\n    ratelimiter_test.go:116: allowed: 118221,     denied: 881779,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1161201,    denied: 9069277,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-8           \t10573476\t       113.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   31.64µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     326ns\n    ratelimiter_test.go:116: allowed: 2130,       denied: 7870,        time per allow:     534ns\n    ratelimiter_test.go:116: allowed: 114434,     denied: 885566,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1200860,    denied: 9372616,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-32          \t10499926\t       113.6 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.54µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     753ns\n    ratelimiter_test.go:116: allowed: 2132,       denied: 7868,        time per allow:     540ns\n    ratelimiter_test.go:116: allowed: 115211,     denied: 884789,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1193395,    denied: 9306531,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-128         \t10586818\t       113.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   32.12µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     337ns\n    ratelimiter_test.go:116: allowed: 2126,       denied: 7874,        time per allow:     535ns\n    ratelimiter_test.go:116: allowed: 114293,     denied: 885707,      time per allow:     991ns\n    ratelimiter_test.go:116: allowed: 1201247,    denied: 9385571,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped          \t 7489098\t       162.3 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   29.53µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     276ns\n    ratelimiter_test.go:116: allowed: 2589,       denied: 7411,        time per allow:     616ns\n    ratelimiter_test.go:116: allowed: 161196,     denied: 838804,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1216236,    denied: 6272862,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-2        \t 7691356\t       155.2 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   36.03µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     634ns\n    ratelimiter_test.go:116: allowed: 2569,       denied: 7431,        time per allow:     620ns\n    ratelimiter_test.go:116: allowed: 156978,     denied: 843022,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1194731,    denied: 6496625,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-4        \t 7584391\t       157.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   35.56µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     350ns\n    ratelimiter_test.go:116: allowed: 2592,       denied: 7408,        time per allow:     623ns\n    ratelimiter_test.go:116: allowed: 159164,     denied: 840836,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1197000,    denied: 6387391,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-8        \t 7750646\t       155.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   35.15µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     393ns\n    ratelimiter_test.go:116: allowed: 2696,       denied: 7304,        time per allow:     635ns\n    ratelimiter_test.go:116: allowed: 155786,     denied: 844214,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1205406,    denied: 6545240,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-32       \t 7739062\t       154.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   31.51µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     384ns\n    ratelimiter_test.go:116: allowed: 2571,       denied: 7429,        time per allow:     614ns\n    ratelimiter_test.go:116: allowed: 156018,     denied: 843982,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1196945,    denied: 6542117,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-128      \t 7452973\t       157.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   34.23µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     368ns\n    ratelimiter_test.go:116: allowed: 2880,       denied: 7120,        time per allow:     659ns\n    ratelimiter_test.go:116: allowed: 161964,     denied: 838036,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1173671,    denied: 6279302,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/mocked_timesource             \t 9300446\t       130.8 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   25.28µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     247ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     399ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     641ns\n    ratelimiter_test.go:116: allowed: 1861089,    denied: 7439357,     time per allow:     653ns\nBenchmarkLimiter/allow/serial/mocked_timesource-2           \t 9821623\t       120.8 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   14.53µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     288ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     411ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     607ns\n    ratelimiter_test.go:116: allowed: 1965324,    denied: 7856299,     time per allow:     603ns\nBenchmarkLimiter/allow/serial/mocked_timesource-4           \t 9834667\t       122.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   31.39µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     375ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     439ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     606ns\n    ratelimiter_test.go:116: allowed: 1967933,    denied: 7866734,     time per allow:     613ns\nBenchmarkLimiter/allow/serial/mocked_timesource-8           \t 8564794\t       128.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   28.07µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     598ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     483ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     696ns\n    ratelimiter_test.go:116: allowed: 1713958,    denied: 6850836,     time per allow:     642ns\nBenchmarkLimiter/allow/serial/mocked_timesource-32          \t 9999607\t       121.6 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   19.14µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     371ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     538ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     596ns\n    ratelimiter_test.go:116: allowed: 2000921,    denied: 7998686,     time per allow:     607ns\nBenchmarkLimiter/allow/serial/mocked_timesource-128         \t 9952394\t       121.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:  26.591µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     307ns\n    ratelimiter_test.go:116: allowed: 2999,       denied: 7001,        time per allow:     451ns\n    ratelimiter_test.go:116: allowed: 200999,     denied: 799001,      time per allow:     599ns\n    ratelimiter_test.go:116: allowed: 1991478,    denied: 7960916,     time per allow:     607ns\nBenchmarkLimiter/allow/parallel/real                        \t10205740\t       117.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  155.67µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     256ns\n    ratelimiter_test.go:138: allowed: 2167,       denied: 7833,        time per allow:     541ns\n    ratelimiter_test.go:138: allowed: 118534,     denied: 881466,      time per allow:     991ns\n    ratelimiter_test.go:138: allowed: 1199932,    denied: 9005808,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/real-2                      \t 9410480\t       135.1 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    57.2µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     421ns\n    ratelimiter_test.go:138: allowed: 2461,       denied: 7539,        time per allow:     597ns\n    ratelimiter_test.go:138: allowed: 155677,     denied: 844323,      time per allow:     992ns\n    ratelimiter_test.go:138: allowed: 995318,     denied: 6771672,     time per allow:     995ns\n    ratelimiter_test.go:138: allowed: 1276963,    denied: 8133517,     time per allow:     995ns\nBenchmarkLimiter/allow/parallel/real-4                      \t11752964\t       124.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   40.32µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     494ns\n    ratelimiter_test.go:138: allowed: 3014,       denied: 6986,        time per allow:     534ns\n    ratelimiter_test.go:138: allowed: 106056,     denied: 893944,      time per allow:     962ns\n    ratelimiter_test.go:138: allowed: 1497530,    denied: 10255434,    time per allow:     977ns\nBenchmarkLimiter/allow/parallel/real-8                      \t 6841635\t       170.6 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  65.389µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     462ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     223ns\n    ratelimiter_test.go:138: allowed: 965185,     denied: 34815,       time per allow:     235ns\n    ratelimiter_test.go:138: allowed: 3257342,    denied: 2028365,     time per allow:     284ns\n    ratelimiter_test.go:138: allowed: 3769827,    denied: 3071808,     time per allow:     309ns\nBenchmarkLimiter/allow/parallel/real-32                     \t 3867985\t       311.2 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   77.99µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     780ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     335ns\n    ratelimiter_test.go:138: allowed: 999737,     denied: 263,         time per allow:     310ns\n    ratelimiter_test.go:138: allowed: 3866778,    denied: 1207,        time per allow:     311ns\nBenchmarkLimiter/allow/parallel/real-128                    \t 3643759\t       322.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   140.1µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.225µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     342ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     329ns\n    ratelimiter_test.go:138: allowed: 3643553,    denied: 206,         time per allow:     322ns\nBenchmarkLimiter/allow/parallel/wrapped                     \t 7118395\t       166.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow: 138.249µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     455ns\n    ratelimiter_test.go:138: allowed: 2588,       denied: 7412,        time per allow:     616ns\n    ratelimiter_test.go:138: allowed: 169513,     denied: 830487,      time per allow:     994ns\n    ratelimiter_test.go:138: allowed: 1189205,    denied: 5929190,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-2                   \t 6918020\t       172.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   49.24µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.206µs\n    ratelimiter_test.go:138: allowed: 2713,       denied: 7287,        time per allow:     640ns\n    ratelimiter_test.go:138: allowed: 174391,     denied: 825609,      time per allow:     994ns\n    ratelimiter_test.go:138: allowed: 1197180,    denied: 5720840,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-4                   \t 6418586\t       189.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   35.67µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     870ns\n    ratelimiter_test.go:138: allowed: 2960,       denied: 7040,        time per allow:     669ns\n    ratelimiter_test.go:138: allowed: 187814,     denied: 812186,      time per allow:     994ns\n    ratelimiter_test.go:138: allowed: 1218744,    denied: 5199842,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-8                   \t 5855096\t       209.8 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   61.67µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.039µs\n    ratelimiter_test.go:138: allowed: 2878,       denied: 7122,        time per allow:     660ns\n    ratelimiter_test.go:138: allowed: 205806,     denied: 794194,      time per allow:     995ns\n    ratelimiter_test.go:138: allowed: 1229495,    denied: 4625601,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-32                  \t 4693095\t       285.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   46.33µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.325µs\n    ratelimiter_test.go:138: allowed: 3945,       denied: 6055,        time per allow:     760ns\n    ratelimiter_test.go:138: allowed: 256641,     denied: 743359,      time per allow:     996ns\n    ratelimiter_test.go:138: allowed: 1342685,    denied: 3350410,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-128                 \t 3859207\t       321.0 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  118.79µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.673µs\n    ratelimiter_test.go:138: allowed: 3188,       denied: 6812,        time per allow:     698ns\n    ratelimiter_test.go:138: allowed: 311877,     denied: 688123,      time per allow:     996ns\n    ratelimiter_test.go:138: allowed: 1239635,    denied: 2619572,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/mocked_timesource           \t 8854556\t       134.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   160.4µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     332ns\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:     395ns\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:     674ns\n    ratelimiter_test.go:138: allowed: 1771911,    denied: 7082645,     time per allow:     671ns\nBenchmarkLimiter/allow/parallel/mocked_timesource-2         \t 8279631\t       151.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   65.36µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     697ns\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:     618ns\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:     720ns\n    ratelimiter_test.go:138: allowed: 1656926,    denied: 6622705,     time per allow:     757ns\nBenchmarkLimiter/allow/parallel/mocked_timesource-4         \t 6189608\t       214.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   52.91µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     882ns\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:     741ns\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:     964ns\n    ratelimiter_test.go:138: allowed: 1238921,    denied: 4950687,     time per allow:   1.073µs\nBenchmarkLimiter/allow/parallel/mocked_timesource-8         \t 3798624\t       333.2 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    62.1µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     757ns\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:   1.229µs\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:   1.571µs\n    ratelimiter_test.go:138: allowed: 760724,     denied: 3037900,     time per allow:   1.663µs\nBenchmarkLimiter/allow/parallel/mocked_timesource-32        \t 3555322\t       384.3 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   61.69µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     737ns\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:   1.268µs\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:   1.679µs\n    ratelimiter_test.go:138: allowed: 712064,     denied: 2843258,     time per allow:   1.918µs\nBenchmarkLimiter/allow/parallel/mocked_timesource-128       \t 3936543\t       305.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  160.99µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.424µs\n    ratelimiter_test.go:138: allowed: 2999,       denied: 7001,        time per allow:   1.717µs\n    ratelimiter_test.go:138: allowed: 200999,     denied: 799001,      time per allow:   1.516µs\n    ratelimiter_test.go:138: allowed: 788308,     denied: 3148235,     time per allow:   1.525µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real             \t 3808436\t       314.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:     665ns\n    ratelimiter_test.go:138: allowed: 2173,       denied: 7827,        time per allow:   1.364µs\n    ratelimiter_test.go:138: allowed: 154369,     denied: 845631,      time per allow:    2.04µs\n    ratelimiter_test.go:138: allowed: 576628,     denied: 3231808,     time per allow:   2.075µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-2           \t 3794132\t       350.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-2\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.116µs\n    ratelimiter_test.go:138: allowed: 2542,       denied: 7458,        time per allow:   1.064µs\n    ratelimiter_test.go:138: allowed: 241564,     denied: 758436,      time per allow:   1.309µs\n    ratelimiter_test.go:138: allowed: 1008362,    denied: 2785770,     time per allow:    1.32µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-4           \t 3830700\t       309.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-4\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.112µs\n    ratelimiter_test.go:138: allowed: 5345,       denied: 4655,        time per allow:     552ns\n    ratelimiter_test.go:138: allowed: 532052,     denied: 467948,      time per allow:     588ns\n    ratelimiter_test.go:138: allowed: 2016383,    denied: 1814317,     time per allow:     588ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-8           \t 3584832\t       343.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-8\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 64,         denied: 36,          time per allow:   3.189µs\n    ratelimiter_test.go:138: allowed: 6665,       denied: 3335,        time per allow:     533ns\n    ratelimiter_test.go:138: allowed: 666662,     denied: 333338,      time per allow:     502ns\n    ratelimiter_test.go:138: allowed: 2389749,    denied: 1195083,     time per allow:     515ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-32          \t 2424110\t       513.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-32\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:    1.58µs\n    ratelimiter_test.go:138: allowed: 6666,       denied: 3334,        time per allow:     651ns\n    ratelimiter_test.go:138: allowed: 666655,     denied: 333345,      time per allow:     742ns\n    ratelimiter_test.go:138: allowed: 1616063,    denied: 808047,      time per allow:     770ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-128         \t 2093779\t       541.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-128\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.696µs\n    ratelimiter_test.go:138: allowed: 6623,       denied: 3377,        time per allow:     778ns\n    ratelimiter_test.go:138: allowed: 666621,     denied: 333379,      time per allow:     859ns\n    ratelimiter_test.go:138: allowed: 1395828,    denied: 697951,      time per allow:     812ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time             \t 7565743\t       160.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:     514ns\n    ratelimiter_test.go:138: allowed: 2560,       denied: 7440,        time per allow:     612ns\n    ratelimiter_test.go:138: allowed: 159529,     denied: 840471,      time per allow:     993ns\n    ratelimiter_test.go:138: allowed: 1211482,    denied: 6354261,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-2           \t 5678437\t       200.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-2\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.277µs\n    ratelimiter_test.go:138: allowed: 5500,       denied: 4500,        time per allow:     477ns\n    ratelimiter_test.go:138: allowed: 329228,     denied: 670772,      time per allow:     641ns\n    ratelimiter_test.go:138: allowed: 1771314,    denied: 3907123,     time per allow:     643ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-4           \t 5894012\t       204.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-4\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 65,         denied: 35,          time per allow:   1.045µs\n    ratelimiter_test.go:138: allowed: 6666,       denied: 3334,        time per allow:     341ns\n    ratelimiter_test.go:138: allowed: 608898,     denied: 391102,      time per allow:     334ns\n    ratelimiter_test.go:138: allowed: 3627237,    denied: 2266775,     time per allow:     332ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-8           \t 4104926\t       245.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-8\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.087µs\n    ratelimiter_test.go:138: allowed: 6666,       denied: 3334,        time per allow:     419ns\n    ratelimiter_test.go:138: allowed: 666664,     denied: 333336,      time per allow:     438ns\n    ratelimiter_test.go:138: allowed: 2736617,    denied: 1368309,     time per allow:     367ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-32          \t 2771666\t       441.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-32\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 63,         denied: 37,          time per allow:   1.475µs\n    ratelimiter_test.go:138: allowed: 6655,       denied: 3345,        time per allow:     693ns\n    ratelimiter_test.go:138: allowed: 666666,     denied: 333334,      time per allow:     649ns\n    ratelimiter_test.go:138: allowed: 1847767,    denied: 923899,      time per allow:     661ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-128         \t 2853688\t       433.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-128\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   3.034µs\n    ratelimiter_test.go:138: allowed: 6610,       denied: 3390,        time per allow:     618ns\n    ratelimiter_test.go:138: allowed: 666626,     denied: 333374,      time per allow:     630ns\n    ratelimiter_test.go:138: allowed: 1902376,    denied: 951312,      time per allow:     650ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped                      \t 4191378\t       292.4 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:    1.19µs\n    ratelimiter_test.go:138: allowed: 3852,       denied: 6148,        time per allow:     745ns\n    ratelimiter_test.go:138: allowed: 287256,     denied: 712744,      time per allow:     996ns\n    ratelimiter_test.go:138: allowed: 1226213,    denied: 2965165,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-2                    \t 4106936\t       306.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-2\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.504µs\n    ratelimiter_test.go:138: allowed: 4018,       denied: 5982,        time per allow:     758ns\n    ratelimiter_test.go:138: allowed: 292848,     denied: 707152,      time per allow:     997ns\n    ratelimiter_test.go:138: allowed: 1256480,    denied: 2850456,     time per allow:       1µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-4                    \t 3492135\t       331.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-4\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 65,         denied: 35,          time per allow:   1.614µs\n    ratelimiter_test.go:138: allowed: 6049,       denied: 3951,        time per allow:     844ns\n    ratelimiter_test.go:138: allowed: 342411,     denied: 657589,      time per allow:   1.003µs\n    ratelimiter_test.go:138: allowed: 1150167,    denied: 2341968,     time per allow:   1.004µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-8                    \t 3619214\t       319.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-8\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 62,         denied: 38,          time per allow:   1.637µs\n    ratelimiter_test.go:138: allowed: 4660,       denied: 5340,        time per allow:     826ns\n    ratelimiter_test.go:138: allowed: 328449,     denied: 671551,      time per allow:   1.009µs\n    ratelimiter_test.go:138: allowed: 1138671,    denied: 2480543,     time per allow:   1.013µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-32                   \t 3563878\t       352.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-32\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 58,         denied: 42,          time per allow:   2.592µs\n    ratelimiter_test.go:138: allowed: 5067,       denied: 4933,        time per allow:     862ns\n    ratelimiter_test.go:138: allowed: 332094,     denied: 667906,      time per allow:   1.013µs\n    ratelimiter_test.go:138: allowed: 1229827,    denied: 2334051,     time per allow:   1.022µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-128                  \t 2696084\t       456.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-128\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 54,         denied: 46,          time per allow:   4.346µs\n    ratelimiter_test.go:138: allowed: 4590,       denied: 5410,        time per allow:     855ns\n    ratelimiter_test.go:138: allowed: 421401,     denied: 578599,      time per allow:   1.056µs\n    ratelimiter_test.go:138: allowed: 1164520,    denied: 1531564,     time per allow:   1.055µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource            \t 4711965\t       265.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.099µs\n    ratelimiter_test.go:138: allowed: 1999,       denied: 8001,        time per allow:   1.256µs\n    ratelimiter_test.go:138: allowed: 100999,     denied: 899001,      time per allow:   2.521µs\n    ratelimiter_test.go:138: allowed: 472196,     denied: 4239769,     time per allow:   2.646µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-2          \t 4082649\t       298.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-2\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.448µs\n    ratelimiter_test.go:138: allowed: 1995,       denied: 8005,        time per allow:   1.502µs\n    ratelimiter_test.go:138: allowed: 100853,     denied: 899147,      time per allow:   2.914µs\n    ratelimiter_test.go:138: allowed: 408711,     denied: 3673938,     time per allow:   2.978µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-4          \t 3961304\t       281.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-4\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:   1.933µs\n    ratelimiter_test.go:138: allowed: 1986,       denied: 8014,        time per allow:   2.219µs\n    ratelimiter_test.go:138: allowed: 100626,     denied: 899374,      time per allow:    3.01µs\n    ratelimiter_test.go:138: allowed: 395937,     denied: 3565367,     time per allow:    2.82µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-8          \t 3600322\t       347.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-8\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 63,         denied: 37,          time per allow:   2.513µs\n    ratelimiter_test.go:138: allowed: 1935,       denied: 8065,        time per allow:   2.113µs\n    ratelimiter_test.go:138: allowed: 100133,     denied: 899867,      time per allow:   3.328µs\n    ratelimiter_test.go:138: allowed: 358211,     denied: 3242111,     time per allow:   3.487µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-32         \t 3664026\t       320.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-32\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 50,         denied: 50,          time per allow:   3.632µs\n    ratelimiter_test.go:138: allowed: 1952,       denied: 8048,        time per allow:   2.115µs\n    ratelimiter_test.go:138: allowed: 100075,     denied: 899925,      time per allow:   3.272µs\n    ratelimiter_test.go:138: allowed: 364912,     denied: 3299114,     time per allow:   3.216µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-128        \t 3662464\t       331.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-128\n    ratelimiter_test.go:138: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:138: allowed: 66,         denied: 34,          time per allow:    1.47µs\n    ratelimiter_test.go:138: allowed: 1961,       denied: 8039,        time per allow:   1.737µs\n    ratelimiter_test.go:138: allowed: 99630,      denied: 900370,      time per allow:   3.288µs\n    ratelimiter_test.go:138: allowed: 362096,     denied: 3300368,     time per allow:   3.353µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real                           \t 3881887\t       310.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     549ns\n    ratelimiter_test.go:116: allowed: 2175,       denied: 7825,        time per allow:   1.331µs\n    ratelimiter_test.go:116: allowed: 149973,     denied: 850027,      time per allow:   2.061µs\n    ratelimiter_test.go:116: allowed: 587105,     denied: 3294782,     time per allow:   2.055µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-2                         \t 4098498\t       292.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-2\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     551ns\n    ratelimiter_test.go:116: allowed: 2114,       denied: 7886,        time per allow:   1.348µs\n    ratelimiter_test.go:116: allowed: 136905,     denied: 863095,      time per allow:   2.138µs\n    ratelimiter_test.go:116: allowed: 558606,     denied: 3539892,     time per allow:   2.144µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-4                         \t 4114698\t       291.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-4\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     932ns\n    ratelimiter_test.go:116: allowed: 2148,       denied: 7852,        time per allow:   1.365µs\n    ratelimiter_test.go:116: allowed: 135966,     denied: 864034,      time per allow:   2.144µs\n    ratelimiter_test.go:116: allowed: 558237,     denied: 3556461,     time per allow:   2.151µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-8                         \t 4095794\t       292.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-8\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     548ns\n    ratelimiter_test.go:116: allowed: 2228,       denied: 7772,        time per allow:   1.332µs\n    ratelimiter_test.go:116: allowed: 138214,     denied: 861786,      time per allow:   2.119µs\n    ratelimiter_test.go:116: allowed: 562747,     denied: 3533047,     time per allow:   2.129µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-32                        \t 4068602\t       294.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-32\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     673ns\n    ratelimiter_test.go:116: allowed: 2095,       denied: 7905,        time per allow:   1.363µs\n    ratelimiter_test.go:116: allowed: 140538,     denied: 859462,      time per allow:   2.098µs\n    ratelimiter_test.go:116: allowed: 564976,     denied: 3503626,     time per allow:   2.122µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-128                       \t 3990260\t       295.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-128\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:   1.239µs\n    ratelimiter_test.go:116: allowed: 2184,       denied: 7816,        time per allow:   1.355µs\n    ratelimiter_test.go:116: allowed: 144378,     denied: 855622,      time per allow:   2.082µs\n    ratelimiter_test.go:116: allowed: 556167,     denied: 3434093,     time per allow:   2.117µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time               \t 7891219\t       154.4 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     354ns\n    ratelimiter_test.go:116: allowed: 2459,       denied: 7541,        time per allow:     594ns\n    ratelimiter_test.go:116: allowed: 153042,     denied: 846958,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1219275,    denied: 6671944,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-2             \t 7883937\t       153.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-2\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     470ns\n    ratelimiter_test.go:116: allowed: 2565,       denied: 7435,        time per allow:     613ns\n    ratelimiter_test.go:116: allowed: 153151,     denied: 846849,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1212499,    denied: 6671438,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-4             \t 7756005\t       155.4 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-4\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     393ns\n    ratelimiter_test.go:116: allowed: 2477,       denied: 7523,        time per allow:     599ns\n    ratelimiter_test.go:116: allowed: 155664,     denied: 844336,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1206085,    denied: 6549920,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-8             \t 7378713\t       158.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-8\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     408ns\n    ratelimiter_test.go:116: allowed: 2484,       denied: 7516,        time per allow:     600ns\n    ratelimiter_test.go:116: allowed: 163573,     denied: 836427,      time per allow:     994ns\n    ratelimiter_test.go:116: allowed: 1166468,    denied: 6212245,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-32            \t 7624196\t       155.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-32\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     462ns\n    ratelimiter_test.go:116: allowed: 2488,       denied: 7512,        time per allow:     601ns\n    ratelimiter_test.go:116: allowed: 158338,     denied: 841662,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1186770,    denied: 6437426,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-128           \t 7776684\t       153.1 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-128\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     444ns\n    ratelimiter_test.go:116: allowed: 2488,       denied: 7512,        time per allow:     602ns\n    ratelimiter_test.go:116: allowed: 155239,     denied: 844761,      time per allow:     993ns\n    ratelimiter_test.go:116: allowed: 1191589,    denied: 6585095,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped                        \t 4251330\t       286.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     560ns\n    ratelimiter_test.go:116: allowed: 3599,       denied: 6401,        time per allow:     723ns\n    ratelimiter_test.go:116: allowed: 283228,     denied: 716772,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1220218,    denied: 3031112,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-2                      \t 4503495\t       263.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-2\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:   1.426µs\n    ratelimiter_test.go:116: allowed: 4001,       denied: 5999,        time per allow:     755ns\n    ratelimiter_test.go:116: allowed: 267416,     denied: 732584,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1186958,    denied: 3316537,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-4                      \t 4751306\t       261.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-4\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:   1.246µs\n    ratelimiter_test.go:116: allowed: 3583,       denied: 6417,        time per allow:     723ns\n    ratelimiter_test.go:116: allowed: 253528,     denied: 746472,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1242170,    denied: 3509136,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-8                      \t 4372312\t       259.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-8\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:   1.262µs\n    ratelimiter_test.go:116: allowed: 3813,       denied: 6187,        time per allow:     740ns\n    ratelimiter_test.go:116: allowed: 275414,     denied: 724586,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1135920,    denied: 3236392,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-32                     \t 4645706\t       256.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-32\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     838ns\n    ratelimiter_test.go:116: allowed: 3748,       denied: 6252,        time per allow:     739ns\n    ratelimiter_test.go:116: allowed: 259264,     denied: 740736,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1191440,    denied: 3454266,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-128                    \t 4710174\t       271.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-128\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     856ns\n    ratelimiter_test.go:116: allowed: 3704,       denied: 6296,        time per allow:     733ns\n    ratelimiter_test.go:116: allowed: 255734,     denied: 744266,      time per allow:     996ns\n    ratelimiter_test.go:116: allowed: 1280804,    denied: 3429370,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource              \t 4923451\t       259.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     698ns\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.203µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.413µs\n    ratelimiter_test.go:116: allowed: 493344,     denied: 4430107,     time per allow:   2.585µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-2            \t 5527620\t       220.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-2\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     865ns\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.321µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.148µs\n    ratelimiter_test.go:116: allowed: 553761,     denied: 4973859,     time per allow:   2.203µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-4            \t 5319728\t       225.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-4\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     532ns\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.162µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.233µs\n    ratelimiter_test.go:116: allowed: 532972,     denied: 4786756,     time per allow:   2.245µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-8            \t 5014401\t       251.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-8\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:     888ns\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.217µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.369µs\n    ratelimiter_test.go:116: allowed: 502439,     denied: 4511962,     time per allow:   2.504µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-32           \t 4355498\t       256.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-32\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:    1.19µs\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.388µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.727µs\n    ratelimiter_test.go:116: allowed: 436549,     denied: 3918949,     time per allow:    2.56µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-128          \t 4664314\t       264.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-128\n    ratelimiter_test.go:116: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:116: allowed: 66,         denied: 34,          time per allow:   1.458µs\n    ratelimiter_test.go:116: allowed: 1999,       denied: 8001,        time per allow:   1.817µs\n    ratelimiter_test.go:116: allowed: 100999,     denied: 899001,      time per allow:   2.547µs\n    ratelimiter_test.go:116: allowed: 467431,     denied: 4196883,     time per allow:   2.641µs\nBenchmarkLimiter/wait/serial/2ns_rate/real                                       \t 1721722\t       735.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:    33.8µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     831ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     654ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     696ns\n    ratelimiter_test.go:116: allowed: 1721722,    denied: 0,           time per allow:     735ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-2                                     \t 1980352\t       650.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   40.31µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.036µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     662ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     605ns\n    ratelimiter_test.go:116: allowed: 1980352,    denied: 0,           time per allow:     650ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-4                                     \t 2035285\t       583.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   25.06µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     942ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     643ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     589ns\n    ratelimiter_test.go:116: allowed: 2035285,    denied: 0,           time per allow:     583ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-8                                     \t 1996000\t       629.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.88µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.231µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     687ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     601ns\n    ratelimiter_test.go:116: allowed: 1996000,    denied: 0,           time per allow:     629ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-32                                    \t 1934670\t       613.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:    24.9µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.185µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     666ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     620ns\n    ratelimiter_test.go:116: allowed: 1934670,    denied: 0,           time per allow:     613ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-128                                   \t 1978052\t       613.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   27.87µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.681µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     695ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     606ns\n    ratelimiter_test.go:116: allowed: 1978052,    denied: 0,           time per allow:     613ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped                                    \t 1572546\t       777.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   18.33µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     695ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     850ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     763ns\n    ratelimiter_test.go:116: allowed: 1572546,    denied: 0,           time per allow:     777ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-2                                  \t 1827685\t       663.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   46.22µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     890ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     692ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     656ns\n    ratelimiter_test.go:116: allowed: 1827685,    denied: 0,           time per allow:     663ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-4                                  \t 1847516\t       634.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:  26.329µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     919ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     732ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     649ns\n    ratelimiter_test.go:116: allowed: 1847516,    denied: 0,           time per allow:     634ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-8                                  \t 1892594\t       648.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   24.38µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     853ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     730ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     633ns\n    ratelimiter_test.go:116: allowed: 1892594,    denied: 0,           time per allow:     648ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-32                                 \t 1866420\t       643.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.23µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.197µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     729ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     642ns\n    ratelimiter_test.go:116: allowed: 1866420,    denied: 0,           time per allow:     643ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-128                                \t 1860192\t       665.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.49µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.444µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     738ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     645ns\n    ratelimiter_test.go:116: allowed: 1860192,    denied: 0,           time per allow:     665ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource                          \t 3480636\t       338.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:  20.771µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     621ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     296ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     344ns\n    ratelimiter_test.go:116: allowed: 3480636,    denied: 0,           time per allow:     337ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-2                        \t 4025022\t       310.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   11.83µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.034µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     295ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     298ns\n    ratelimiter_test.go:116: allowed: 4025022,    denied: 0,           time per allow:     310ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-4                        \t 3879685\t       311.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   30.59µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     798ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     329ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     309ns\n    ratelimiter_test.go:116: allowed: 3879685,    denied: 0,           time per allow:     310ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-8                        \t 3954216\t       302.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   15.32µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     522ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     561ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     303ns\n    ratelimiter_test.go:116: allowed: 3954216,    denied: 0,           time per allow:     302ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-32                       \t 3992751\t       311.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   15.98µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.062µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     298ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     300ns\n    ratelimiter_test.go:116: allowed: 3992751,    denied: 0,           time per allow:     311ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-128                      \t 3994308\t       305.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   17.84µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     567ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     305ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     300ns\n    ratelimiter_test.go:116: allowed: 3994308,    denied: 0,           time per allow:     305ns\nBenchmarkLimiter/wait/serial/1µs_rate/real                                       \t 1000000\t      1030 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   36.19µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     749ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     913ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:    1.03µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-2                                     \t 1000000\t      1008 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   21.62µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.063µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     982ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.007µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-4                                     \t 1000000\t      1013 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.92µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:    1.27µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     912ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.012µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-8                                     \t 1000000\t      1010 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   18.09µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.997µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:    1.01µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-32                                    \t 1000000\t      1023 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:    18.8µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   2.092µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     917ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.022µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-128                                   \t 1000000\t      1012 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   25.05µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     970ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     914ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.012µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped                                    \t 1153466\t      1029 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   29.46µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.157µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:   1.557µs\n    ratelimiter_test.go:116: allowed: 768654,     denied: 0,           time per allow:    1.04µs\n    ratelimiter_test.go:116: allowed: 1153466,    denied: 0,           time per allow:   1.028µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-2                                  \t 1000000\t      1010 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:  21.451µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     851ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     910ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.009µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-4                                  \t 1000000\t      1008 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   19.57µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.042µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     923ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.007µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-8                                  \t  985587\t      1016 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   20.13µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   2.425µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:   1.214µs\n    ratelimiter_test.go:116: allowed: 985587,     denied: 0,           time per allow:   1.016µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-32                                 \t 1000000\t      1009 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   37.11µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.453µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     913ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.009µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-128                                \t 1000000\t      1008 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:  21.431µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:   1.652µs\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     938ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:   1.008µs\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource                          \t 3296186\t       350.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   16.94µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     425ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     287ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     364ns\n    ratelimiter_test.go:116: allowed: 3296186,    denied: 0,           time per allow:     350ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-2                        \t 3759115\t       305.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-2\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   23.92µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     825ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     415ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     319ns\n    ratelimiter_test.go:116: allowed: 3759115,    denied: 0,           time per allow:     305ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-4                        \t 3970958\t       295.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-4\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   15.64µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     566ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     303ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     302ns\n    ratelimiter_test.go:116: allowed: 3970958,    denied: 0,           time per allow:     295ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-8                        \t 4056230\t       301.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-8\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   14.15µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     632ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     310ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     295ns\n    ratelimiter_test.go:116: allowed: 4056230,    denied: 0,           time per allow:     301ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-32                       \t 4013587\t       303.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-32\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   15.94µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     706ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     304ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     298ns\n    ratelimiter_test.go:116: allowed: 4013587,    denied: 0,           time per allow:     303ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-128                      \t 3777873\t       300.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-128\n    ratelimiter_test.go:116: allowed: 1,          denied: 0,           time per allow:   16.33µs\n    ratelimiter_test.go:116: allowed: 100,        denied: 0,           time per allow:     880ns\n    ratelimiter_test.go:116: allowed: 10000,      denied: 0,           time per allow:     309ns\n    ratelimiter_test.go:116: allowed: 1000000,    denied: 0,           time per allow:     317ns\n    ratelimiter_test.go:116: allowed: 3777873,    denied: 0,           time per allow:     300ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real                                     \t 1606105\t       743.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   88.94µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     910ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     693ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     747ns\n    ratelimiter_test.go:138: allowed: 1606105,    denied: 0,           time per allow:     743ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-2                                   \t 2684506\t       594.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   32.24µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     611ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     481ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     446ns\n    ratelimiter_test.go:138: allowed: 2684506,    denied: 0,           time per allow:     594ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-4                                   \t 2008572\t       607.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   41.57µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.732µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     557ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     597ns\n    ratelimiter_test.go:138: allowed: 2008572,    denied: 0,           time per allow:     607ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-8                                   \t 2382729\t       570.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   59.23µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.044µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     681ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     607ns\n    ratelimiter_test.go:138: allowed: 1975958,    denied: 0,           time per allow:     503ns\n    ratelimiter_test.go:138: allowed: 2382729,    denied: 0,           time per allow:     570ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-32                                  \t 1412924\t       856.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  135.94µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.709µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     800ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     849ns\n    ratelimiter_test.go:138: allowed: 1412924,    denied: 0,           time per allow:     856ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-128                                 \t 1227754\t       974.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   86.34µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.269µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     975ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     977ns\n    ratelimiter_test.go:138: allowed: 1227754,    denied: 0,           time per allow:     974ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped                                  \t 1599991\t       780.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  101.84µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.018µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:   1.044µs\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     749ns\n    ratelimiter_test.go:138: allowed: 1599991,    denied: 0,           time per allow:     780ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-2                                \t 1609435\t       728.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    59.3µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.021µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     869ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     745ns\n    ratelimiter_test.go:138: allowed: 1609435,    denied: 0,           time per allow:     728ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-4                                \t 1874695\t       693.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    36.2µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:    1.42µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     688ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     640ns\n    ratelimiter_test.go:138: allowed: 1874695,    denied: 0,           time per allow:     693ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-8                                \t 1358745\t       907.4 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  105.66µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.552µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     681ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     883ns\n    ratelimiter_test.go:138: allowed: 1358745,    denied: 0,           time per allow:     907ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-32                               \t 1000000\t      1170 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  159.77µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.318µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:   1.045µs\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:   1.169µs\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-128                              \t  876781\t      1165 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow: 143.331µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   3.529µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:   1.365µs\n    ratelimiter_test.go:138: allowed: 876781,     denied: 0,           time per allow:   1.165µs\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource                        \t 2869887\t       360.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  123.22µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     553ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     318ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     418ns\n    ratelimiter_test.go:138: allowed: 2869887,    denied: 0,           time per allow:     360ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-2                      \t 2966229\t       460.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   53.44µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     789ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     346ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     404ns\n    ratelimiter_test.go:138: allowed: 2966229,    denied: 0,           time per allow:     460ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-4                      \t 2462452\t       560.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   68.69µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.374µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     599ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     487ns\n    ratelimiter_test.go:138: allowed: 2462452,    denied: 0,           time per allow:     560ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-8                      \t 1838635\t       639.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   71.33µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   3.539µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     865ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     652ns\n    ratelimiter_test.go:138: allowed: 1838635,    denied: 0,           time per allow:     639ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-32                     \t 1778676\t       689.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   82.09µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.509µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     795ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     674ns\n    ratelimiter_test.go:138: allowed: 1778676,    denied: 0,           time per allow:     689ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-128                    \t 1928882\t       584.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  187.11µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   3.614µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     656ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     622ns\n    ratelimiter_test.go:138: allowed: 1928882,    denied: 0,           time per allow:     584ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real                                     \t 1000000\t      1025 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  190.51µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     917ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     982ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:   1.025µs\nBenchmarkLimiter/wait/parallel/1µs_rate/real-2                                   \t 1314758\t       929.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    56.4µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:    1.13µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     774ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     912ns\n    ratelimiter_test.go:138: allowed: 1314758,    denied: 0,           time per allow:     929ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-4                                   \t 2196931\t       560.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   39.84µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.296µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     524ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     546ns\n    ratelimiter_test.go:138: allowed: 2196931,    denied: 0,           time per allow:     560ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-8                                   \t 1964460\t       589.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   60.08µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.316µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     562ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     610ns\n    ratelimiter_test.go:138: allowed: 1964460,    denied: 0,           time per allow:     589ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-32                                  \t 1600582\t       799.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   61.42µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:    1.34µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     770ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     749ns\n    ratelimiter_test.go:138: allowed: 1600582,    denied: 0,           time per allow:     799ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-128                                 \t 1240540\t       982.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  171.79µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.062µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     865ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     967ns\n    ratelimiter_test.go:138: allowed: 1240540,    denied: 0,           time per allow:     982ns\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped                                  \t 1000000\t      1030 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  104.34µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     875ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:       1µs\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:    1.03µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-2                                \t 1000000\t      1006 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    55.1µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.365µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     960ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:   1.006µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-4                                \t 1000000\t      1007 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  57.911µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.387µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:   1.006µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-8                                \t 1000000\t      1009 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:    44.7µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.105µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     929ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:   1.008µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-32                               \t 1000000\t      1021 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   48.83µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   2.674µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:   1.078µs\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:    1.02µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-128                              \t 1000000\t      1081 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:  115.78µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   3.145µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:   1.023µs\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:    1.08µs\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource                        \t 3340384\t       366.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   80.65µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     594ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     291ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     359ns\n    ratelimiter_test.go:138: allowed: 3340384,    denied: 0,           time per allow:     366ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-2                      \t 3445034\t       368.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-2\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   43.04µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:     892ns\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     460ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     348ns\n    ratelimiter_test.go:138: allowed: 3445034,    denied: 0,           time per allow:     368ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-4                      \t 2553564\t       473.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-4\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   37.44µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.371µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     568ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     469ns\n    ratelimiter_test.go:138: allowed: 2553564,    denied: 0,           time per allow:     473ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-8                      \t 1935856\t       607.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-8\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   68.55µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.884µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     717ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     619ns\n    ratelimiter_test.go:138: allowed: 1935856,    denied: 0,           time per allow:     607ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-32                     \t 1874793\t       575.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-32\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow:   61.94µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   3.209µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     741ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     640ns\n    ratelimiter_test.go:138: allowed: 1874793,    denied: 0,           time per allow:     575ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-128                    \t 1944561\t       599.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-128\n    ratelimiter_test.go:138: allowed: 1,          denied: 0,           time per allow: 179.579µs\n    ratelimiter_test.go:138: allowed: 100,        denied: 0,           time per allow:   1.716µs\n    ratelimiter_test.go:138: allowed: 10000,      denied: 0,           time per allow:     595ns\n    ratelimiter_test.go:138: allowed: 1000000,    denied: 0,           time per allow:     617ns\n    ratelimiter_test.go:138: allowed: 1944561,    denied: 0,           time per allow:     598ns\nPASS\nok  \tgithub.com/uber/cadence/common/clock\t239.738s\n"
  },
  {
    "path": "common/clock/testdata/m1_mac_go1.21.txt",
    "content": "// On an M1 Pro Mac, 8 cores (4 high, 4 low):\n\n❯ go version\ngo version go1.21.9 darwin/arm64\n\n❯ go test -bench . -test.run xxx -cpu 1,2,4,8,32 .\ngoos: darwin\ngoarch: arm64\npkg: github.com/uber/cadence/common/clock\nBenchmarkLimiter/allow/serial/real            \t13040902\t        77.94 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  94.833µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     264ns\n    ratelimiter_test.go:114: allowed: 2461,       denied: 7539,        time per allow:     602ns\n    ratelimiter_test.go:114: allowed: 92982,      denied: 907018,      time per allow:     989ns\n    ratelimiter_test.go:114: allowed: 1017345,    denied: 12023557,    time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-2          \t15258253\t        78.99 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   12.75µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     148ns\n    ratelimiter_test.go:114: allowed: 1780,       denied: 8220,        time per allow:     441ns\n    ratelimiter_test.go:114: allowed: 79617,      denied: 920383,      time per allow:     987ns\n    ratelimiter_test.go:114: allowed: 1206105,    denied: 14052148,    time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-4          \t15349381\t        81.67 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  13.875µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     274ns\n    ratelimiter_test.go:114: allowed: 1798,       denied: 8202,        time per allow:     446ns\n    ratelimiter_test.go:114: allowed: 79149,      denied: 920851,      time per allow:     987ns\n    ratelimiter_test.go:114: allowed: 1238332,    denied: 14111049,    time per allow:   1.012µs\nBenchmarkLimiter/allow/serial/real-8          \t15338174\t        77.89 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  10.875µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     210ns\n    ratelimiter_test.go:114: allowed: 1793,       denied: 8207,        time per allow:     445ns\n    ratelimiter_test.go:114: allowed: 79210,      denied: 920790,      time per allow:     987ns\n    ratelimiter_test.go:114: allowed: 1195634,    denied: 14142540,    time per allow:     999ns\nBenchmarkLimiter/allow/serial/real-32         \t15305487\t        77.85 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/real-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  16.334µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     361ns\n    ratelimiter_test.go:114: allowed: 1780,       denied: 8220,        time per allow:     441ns\n    ratelimiter_test.go:114: allowed: 79367,      denied: 920633,      time per allow:     987ns\n    ratelimiter_test.go:114: allowed: 1192482,    denied: 14113005,    time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped         \t10489392\t       116.3 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  16.625µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     232ns\n    ratelimiter_test.go:114: allowed: 2192,       denied: 7808,        time per allow:     546ns\n    ratelimiter_test.go:114: allowed: 115384,     denied: 884616,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1220378,    denied: 9269014,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-2       \t10763562\t       112.3 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  12.583µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     180ns\n    ratelimiter_test.go:114: allowed: 2155,       denied: 7845,        time per allow:     537ns\n    ratelimiter_test.go:114: allowed: 112471,     denied: 887529,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1209814,    denied: 9553748,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-4       \t10674687\t       111.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.834µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     195ns\n    ratelimiter_test.go:114: allowed: 2152,       denied: 7848,        time per allow:     537ns\n    ratelimiter_test.go:114: allowed: 113396,     denied: 886604,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1193581,    denied: 9481106,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-8       \t10722634\t       111.9 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   9.083µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     253ns\n    ratelimiter_test.go:114: allowed: 2148,       denied: 7852,        time per allow:     535ns\n    ratelimiter_test.go:114: allowed: 112895,     denied: 887105,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1201251,    denied: 9521383,     time per allow:     999ns\nBenchmarkLimiter/allow/serial/wrapped-32      \t10658955\t       116.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/wrapped-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  39.166µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     260ns\n    ratelimiter_test.go:114: allowed: 2158,       denied: 7842,        time per allow:     541ns\n    ratelimiter_test.go:114: allowed: 113562,     denied: 886438,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1231568,    denied: 9427387,     time per allow:   1.007µs\nBenchmarkLimiter/allow/serial/mocked_timesource            \t12198974\t       103.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:      17µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     142ns\n    ratelimiter_test.go:114: allowed: 2999,       denied: 7001,        time per allow:     329ns\n    ratelimiter_test.go:114: allowed: 200999,     denied: 799001,      time per allow:     489ns\n    ratelimiter_test.go:114: allowed: 2440794,    denied: 9758180,     time per allow:     516ns\nBenchmarkLimiter/allow/serial/mocked_timesource-2          \t12352620\t        96.20 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.584µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     600ns\n    ratelimiter_test.go:114: allowed: 2999,       denied: 7001,        time per allow:     320ns\n    ratelimiter_test.go:114: allowed: 200999,     denied: 799001,      time per allow:     483ns\n    ratelimiter_test.go:114: allowed: 2471523,    denied: 9881097,     time per allow:     480ns\nBenchmarkLimiter/allow/serial/mocked_timesource-4          \t12307213\t        96.72 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   7.167µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     377ns\n    ratelimiter_test.go:114: allowed: 2999,       denied: 7001,        time per allow:     323ns\n    ratelimiter_test.go:114: allowed: 200999,     denied: 799001,      time per allow:     485ns\n    ratelimiter_test.go:114: allowed: 2462442,    denied: 9844771,     time per allow:     483ns\nBenchmarkLimiter/allow/serial/mocked_timesource-8          \t12366156\t        96.46 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.291µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     217ns\n    ratelimiter_test.go:114: allowed: 2999,       denied: 7001,        time per allow:     331ns\n    ratelimiter_test.go:114: allowed: 200999,     denied: 799001,      time per allow:     482ns\n    ratelimiter_test.go:114: allowed: 2474231,    denied: 9891925,     time per allow:     482ns\nBenchmarkLimiter/allow/serial/mocked_timesource-32         \t12335895\t        96.76 ns/op\n--- BENCH: BenchmarkLimiter/allow/serial/mocked_timesource-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  18.792µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     324ns\n    ratelimiter_test.go:114: allowed: 2999,       denied: 7001,        time per allow:     318ns\n    ratelimiter_test.go:114: allowed: 200999,     denied: 799001,      time per allow:     483ns\n    ratelimiter_test.go:114: allowed: 2468178,    denied: 9867717,     time per allow:     483ns\nBenchmarkLimiter/allow/parallel/real                       \t14847120\t        80.67 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  54.083µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     164ns\n    ratelimiter_test.go:136: allowed: 1809,       denied: 8191,        time per allow:     450ns\n    ratelimiter_test.go:136: allowed: 81791,      denied: 918209,      time per allow:     987ns\n    ratelimiter_test.go:136: allowed: 1198579,    denied: 13648541,    time per allow:     999ns\nBenchmarkLimiter/allow/parallel/real-2                     \t11729421\t        91.31 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  23.167µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:   1.147µs\n    ratelimiter_test.go:136: allowed: 1880,       denied: 8120,        time per allow:     476ns\n    ratelimiter_test.go:136: allowed: 103270,     denied: 896730,      time per allow:     990ns\n    ratelimiter_test.go:136: allowed: 1072367,    denied: 10657054,    time per allow:     998ns\nBenchmarkLimiter/allow/parallel/real-4                     \t 9522936\t       120.2 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  21.334µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     604ns\n    ratelimiter_test.go:136: allowed: 3052,       denied: 6948,        time per allow:     534ns\n    ratelimiter_test.go:136: allowed: 127279,     denied: 872721,      time per allow:     989ns\n    ratelimiter_test.go:136: allowed: 1148459,    denied: 8374477,     time per allow:     996ns\nBenchmarkLimiter/allow/parallel/real-8                     \t 6435804\t       170.1 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  13.209µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     299ns\n    ratelimiter_test.go:136: allowed: 3708,       denied: 6292,        time per allow:     450ns\n    ratelimiter_test.go:136: allowed: 572085,     denied: 427915,      time per allow:     325ns\n    ratelimiter_test.go:136: allowed: 3756916,    denied: 2678888,     time per allow:     291ns\nBenchmarkLimiter/allow/parallel/real-32                    \t 4673179\t       235.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/real-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  25.458µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     482ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     255ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     256ns\n    ratelimiter_test.go:136: allowed: 3639089,    denied: 1034090,     time per allow:     302ns\nBenchmarkLimiter/allow/parallel/wrapped                    \t10528147\t       116.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  31.833µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     206ns\n    ratelimiter_test.go:136: allowed: 2121,       denied: 7879,        time per allow:     529ns\n    ratelimiter_test.go:136: allowed: 114956,     denied: 885044,      time per allow:     991ns\n    ratelimiter_test.go:136: allowed: 1227259,    denied: 9300888,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-2                  \t 7672480\t       155.8 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  18.958µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     219ns\n    ratelimiter_test.go:136: allowed: 2663,       denied: 7337,        time per allow:     629ns\n    ratelimiter_test.go:136: allowed: 157380,     denied: 842620,      time per allow:     993ns\n    ratelimiter_test.go:136: allowed: 1196181,    denied: 6476299,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-4                  \t 4856625\t       244.5 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  22.083µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     298ns\n    ratelimiter_test.go:136: allowed: 3294,       denied: 6706,        time per allow:     700ns\n    ratelimiter_test.go:136: allowed: 248057,     denied: 751943,      time per allow:     996ns\n    ratelimiter_test.go:136: allowed: 1188577,    denied: 3668048,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-8                  \t 4135431\t       279.6 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  15.541µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     347ns\n    ratelimiter_test.go:136: allowed: 3936,       denied: 6064,        time per allow:     748ns\n    ratelimiter_test.go:136: allowed: 291147,     denied: 708853,      time per allow:     996ns\n    ratelimiter_test.go:136: allowed: 1157034,    denied: 2978397,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/wrapped-32                 \t 2850973\t       447.8 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/wrapped-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  22.959µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     384ns\n    ratelimiter_test.go:136: allowed: 5235,       denied: 4765,        time per allow:     811ns\n    ratelimiter_test.go:136: allowed: 421876,     denied: 578124,      time per allow:     997ns\n    ratelimiter_test.go:136: allowed: 1277734,    denied: 1573239,     time per allow:     999ns\nBenchmarkLimiter/allow/parallel/mocked_timesource          \t11951947\t       104.4 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  39.042µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     179ns\n    ratelimiter_test.go:136: allowed: 2999,       denied: 7001,        time per allow:     331ns\n    ratelimiter_test.go:136: allowed: 200999,     denied: 799001,      time per allow:     499ns\n    ratelimiter_test.go:136: allowed: 2391389,    denied: 9560558,     time per allow:     522ns\nBenchmarkLimiter/allow/parallel/mocked_timesource-2        \t 7483614\t       158.0 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  21.125µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     254ns\n    ratelimiter_test.go:136: allowed: 2999,       denied: 7001,        time per allow:     559ns\n    ratelimiter_test.go:136: allowed: 200999,     denied: 799001,      time per allow:   1.062µs\n    ratelimiter_test.go:136: allowed: 1125014,    denied: 4495057,     time per allow:     801ns\n    ratelimiter_test.go:136: allowed: 1497722,    denied: 5985892,     time per allow:     789ns\nBenchmarkLimiter/allow/parallel/mocked_timesource-4        \t 6092762\t       194.1 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:   15.75µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     233ns\n    ratelimiter_test.go:136: allowed: 2999,       denied: 7001,        time per allow:     627ns\n    ratelimiter_test.go:136: allowed: 200999,     denied: 799001,      time per allow:     979ns\n    ratelimiter_test.go:136: allowed: 1219552,    denied: 4873210,     time per allow:     969ns\nBenchmarkLimiter/allow/parallel/mocked_timesource-8        \t 4215762\t       283.1 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  21.375µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     292ns\n    ratelimiter_test.go:136: allowed: 2999,       denied: 7001,        time per allow:     872ns\n    ratelimiter_test.go:136: allowed: 200999,     denied: 799001,      time per allow:   1.416µs\n    ratelimiter_test.go:136: allowed: 844152,     denied: 3371610,     time per allow:   1.413µs\nBenchmarkLimiter/allow/parallel/mocked_timesource-32       \t 3159067\t       382.7 ns/op\n--- BENCH: BenchmarkLimiter/allow/parallel/mocked_timesource-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  22.625µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     227ns\n    ratelimiter_test.go:136: allowed: 2999,       denied: 7001,        time per allow:   1.093µs\n    ratelimiter_test.go:136: allowed: 200999,     denied: 799001,      time per allow:   1.889µs\n    ratelimiter_test.go:136: allowed: 632813,     denied: 2526254,     time per allow:    1.91µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real     \t 5593612\t       217.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     308ns\n    ratelimiter_test.go:114: allowed: 1922,       denied: 8078,        time per allow:   1.046µs\n    ratelimiter_test.go:114: allowed: 141910,     denied: 858090,      time per allow:   1.511µs\n    ratelimiter_test.go:114: allowed: 795043,     denied: 4798569,     time per allow:   1.527µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-2   \t 5838519\t       208.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-2\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     627ns\n    ratelimiter_test.go:114: allowed: 1955,       denied: 8045,        time per allow:   1.077µs\n    ratelimiter_test.go:114: allowed: 135241,     denied: 864759,      time per allow:   1.519µs\n    ratelimiter_test.go:114: allowed: 798897,     denied: 5039622,     time per allow:   1.519µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-4   \t 5828708\t       206.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-4\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     389ns\n    ratelimiter_test.go:114: allowed: 2013,       denied: 7987,        time per allow:   1.002µs\n    ratelimiter_test.go:114: allowed: 137317,     denied: 862683,      time per allow:   1.499µs\n    ratelimiter_test.go:114: allowed: 798393,     denied: 5030315,     time per allow:   1.506µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-8   \t 5810233\t       206.4 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-8\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     397ns\n    ratelimiter_test.go:114: allowed: 1873,       denied: 8127,        time per allow:   1.087µs\n    ratelimiter_test.go:114: allowed: 135457,     denied: 864543,      time per allow:   1.524µs\n    ratelimiter_test.go:114: allowed: 798038,     denied: 5012195,     time per allow:   1.503µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real-32  \t 5801004\t       206.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real-32\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     368ns\n    ratelimiter_test.go:114: allowed: 2044,       denied: 7956,        time per allow:    1.01µs\n    ratelimiter_test.go:114: allowed: 137067,     denied: 862933,      time per allow:   1.509µs\n    ratelimiter_test.go:114: allowed: 797823,     denied: 5003181,     time per allow:   1.501µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time            \t 9924230\t       121.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     252ns\n    ratelimiter_test.go:114: allowed: 2163,       denied: 7837,        time per allow:     540ns\n    ratelimiter_test.go:114: allowed: 121894,     denied: 878106,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1201932,    denied: 8722298,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-2          \t 9902736\t       120.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-2\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     241ns\n    ratelimiter_test.go:114: allowed: 2194,       denied: 7806,        time per allow:     549ns\n    ratelimiter_test.go:114: allowed: 122146,     denied: 877854,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1198147,    denied: 8704589,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-4          \t 9917348\t       120.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-4\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     315ns\n    ratelimiter_test.go:114: allowed: 2226,       denied: 7774,        time per allow:     554ns\n    ratelimiter_test.go:114: allowed: 121970,     denied: 878030,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1200230,    denied: 8717118,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-8          \t 9918499\t       120.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-8\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     553ns\n    ratelimiter_test.go:114: allowed: 2174,       denied: 7826,        time per allow:     543ns\n    ratelimiter_test.go:114: allowed: 121959,     denied: 878041,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1199866,    denied: 8718633,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-32         \t 9919794\t       121.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/real_pinned_time-32\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     460ns\n    ratelimiter_test.go:114: allowed: 2166,       denied: 7834,        time per allow:     541ns\n    ratelimiter_test.go:114: allowed: 121939,     denied: 878061,      time per allow:     991ns\n    ratelimiter_test.go:114: allowed: 1201433,    denied: 8718361,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped                     \t 6231513\t       194.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     448ns\n    ratelimiter_test.go:114: allowed: 2823,       denied: 7177,        time per allow:     647ns\n    ratelimiter_test.go:114: allowed: 193547,     denied: 806453,      time per allow:     994ns\n    ratelimiter_test.go:114: allowed: 1213880,    denied: 5017633,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-2                   \t 6742875\t       176.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-2\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     368ns\n    ratelimiter_test.go:114: allowed: 2900,       denied: 7100,        time per allow:     657ns\n    ratelimiter_test.go:114: allowed: 178948,     denied: 821052,      time per allow:     994ns\n    ratelimiter_test.go:114: allowed: 1187742,    denied: 5555133,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-4                   \t 6685393\t       177.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-4\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     436ns\n    ratelimiter_test.go:114: allowed: 2833,       denied: 7167,        time per allow:     650ns\n    ratelimiter_test.go:114: allowed: 180478,     denied: 819522,      time per allow:     994ns\n    ratelimiter_test.go:114: allowed: 1187807,    denied: 5497586,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-8                   \t 6632632\t       177.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-8\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     381ns\n    ratelimiter_test.go:114: allowed: 2850,       denied: 7150,        time per allow:     651ns\n    ratelimiter_test.go:114: allowed: 181903,     denied: 818097,      time per allow:     994ns\n    ratelimiter_test.go:114: allowed: 1179238,    denied: 5453394,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-32                  \t 6718569\t       178.1 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/wrapped-32\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     620ns\n    ratelimiter_test.go:114: allowed: 2749,       denied: 7251,        time per allow:     637ns\n    ratelimiter_test.go:114: allowed: 179568,     denied: 820432,      time per allow:     994ns\n    ratelimiter_test.go:114: allowed: 1197318,    denied: 5521251,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource           \t 6727784\t       174.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     799ns\n    ratelimiter_test.go:114: allowed: 1999,       denied: 8001,        time per allow:     856ns\n    ratelimiter_test.go:114: allowed: 100999,     denied: 899001,      time per allow:   1.765µs\n    ratelimiter_test.go:114: allowed: 673778,     denied: 6054006,     time per allow:   1.739µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-2         \t 7773721\t       154.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-2\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     395ns\n    ratelimiter_test.go:114: allowed: 1999,       denied: 8001,        time per allow:     825ns\n    ratelimiter_test.go:114: allowed: 100999,     denied: 899001,      time per allow:   1.528µs\n    ratelimiter_test.go:114: allowed: 778371,     denied: 6995350,     time per allow:   1.544µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-4         \t 7683728\t       156.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-4\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     366ns\n    ratelimiter_test.go:114: allowed: 1999,       denied: 8001,        time per allow:     811ns\n    ratelimiter_test.go:114: allowed: 100999,     denied: 899001,      time per allow:   1.546µs\n    ratelimiter_test.go:114: allowed: 769372,     denied: 6914356,     time per allow:   1.558µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-8         \t 7686458\t       163.4 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-8\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     385ns\n    ratelimiter_test.go:114: allowed: 1999,       denied: 8001,        time per allow:     839ns\n    ratelimiter_test.go:114: allowed: 100999,     denied: 899001,      time per allow:   1.545µs\n    ratelimiter_test.go:114: allowed: 769645,     denied: 6916813,     time per allow:   1.631µs\nBenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-32        \t 7580811\t       157.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/serial/mocked_timesource-32\n    ratelimiter_test.go:114: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:114: allowed: 66,         denied: 34,          time per allow:     387ns\n    ratelimiter_test.go:114: allowed: 1999,       denied: 8001,        time per allow:     796ns\n    ratelimiter_test.go:114: allowed: 100999,     denied: 899001,      time per allow:   1.567µs\n    ratelimiter_test.go:114: allowed: 759080,     denied: 6821731,     time per allow:    1.57µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real                      \t 5510388\t       218.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     462ns\n    ratelimiter_test.go:136: allowed: 1908,       denied: 8092,        time per allow:   1.077µs\n    ratelimiter_test.go:136: allowed: 134580,     denied: 865420,      time per allow:   1.618µs\n    ratelimiter_test.go:136: allowed: 774155,     denied: 4736233,     time per allow:   1.554µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-2                    \t 7520551\t       155.9 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-2\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     914ns\n    ratelimiter_test.go:136: allowed: 823,        denied: 9177,        time per allow:   1.925µs\n    ratelimiter_test.go:136: allowed: 767,        denied: 999233,      time per allow:  208.02µs\n    ratelimiter_test.go:136: allowed: 752,        denied: 7519799,     time per allow: 1.558596ms\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-4                    \t 4660968\t       254.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-4\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     707ns\n    ratelimiter_test.go:136: allowed: 3525,       denied: 6475,        time per allow:     749ns\n    ratelimiter_test.go:136: allowed: 202434,     denied: 797566,      time per allow:   1.271µs\n    ratelimiter_test.go:136: allowed: 897507,     denied: 3763461,     time per allow:   1.322µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-8                    \t 4067368\t       293.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-8\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     558ns\n    ratelimiter_test.go:136: allowed: 6615,       denied: 3385,        time per allow:     409ns\n    ratelimiter_test.go:136: allowed: 661672,     denied: 338328,      time per allow:     445ns\n    ratelimiter_test.go:136: allowed: 2651622,    denied: 1415746,     time per allow:     449ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real-32                   \t 2719041\t       407.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real-32\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     791ns\n    ratelimiter_test.go:136: allowed: 6666,       denied: 3334,        time per allow:     593ns\n    ratelimiter_test.go:136: allowed: 661514,     denied: 338486,      time per allow:     667ns\n    ratelimiter_test.go:136: allowed: 1812685,    denied: 906356,      time per allow:     611ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time          \t 9683659\t       124.3 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     343ns\n    ratelimiter_test.go:136: allowed: 2179,       denied: 7821,        time per allow:     542ns\n    ratelimiter_test.go:136: allowed: 124893,     denied: 875107,      time per allow:     992ns\n    ratelimiter_test.go:136: allowed: 1205125,    denied: 8478534,     time per allow:     999ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-2        \t 8225566\t       146.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-2\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     309ns\n    ratelimiter_test.go:136: allowed: 2680,       denied: 7320,        time per allow:     583ns\n    ratelimiter_test.go:136: allowed: 143199,     denied: 856801,      time per allow:   1.018µs\n    ratelimiter_test.go:136: allowed: 1169292,    denied: 7056274,     time per allow:   1.027µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-4        \t 5578884\t       222.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-4\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     989ns\n    ratelimiter_test.go:136: allowed: 5098,       denied: 4902,        time per allow:     420ns\n    ratelimiter_test.go:136: allowed: 365764,     denied: 634236,      time per allow:     588ns\n    ratelimiter_test.go:136: allowed: 2127002,    denied: 3451882,     time per allow:     584ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-8        \t 5841618\t       207.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-8\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 65,         denied: 35,          time per allow:     430ns\n    ratelimiter_test.go:136: allowed: 6664,       denied: 3336,        time per allow:     376ns\n    ratelimiter_test.go:136: allowed: 633158,     denied: 366842,      time per allow:     324ns\n    ratelimiter_test.go:136: allowed: 3671798,    denied: 2169820,     time per allow:     330ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-32       \t 3719898\t       309.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/real_pinned_time-32\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     506ns\n    ratelimiter_test.go:136: allowed: 6648,       denied: 3352,        time per allow:     346ns\n    ratelimiter_test.go:136: allowed: 666666,     denied: 333334,      time per allow:     483ns\n    ratelimiter_test.go:136: allowed: 2479932,    denied: 1239966,     time per allow:     464ns\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped                   \t 6123994\t       207.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     404ns\n    ratelimiter_test.go:136: allowed: 2810,       denied: 7190,        time per allow:     646ns\n    ratelimiter_test.go:136: allowed: 196910,     denied: 803090,      time per allow:     995ns\n    ratelimiter_test.go:136: allowed: 1227907,    denied: 4896087,     time per allow:   1.032µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-2                 \t 5416012\t       228.0 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-2\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:   1.486µs\n    ratelimiter_test.go:136: allowed: 3350,       denied: 6650,        time per allow:     711ns\n    ratelimiter_test.go:136: allowed: 221339,     denied: 778661,      time per allow:       1µs\n    ratelimiter_test.go:136: allowed: 1230035,    denied: 4185977,     time per allow:   1.004µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-4                 \t 4629619\t       260.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-4\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     994ns\n    ratelimiter_test.go:136: allowed: 3573,       denied: 6427,        time per allow:     778ns\n    ratelimiter_test.go:136: allowed: 252008,     denied: 747992,      time per allow:   1.028µs\n    ratelimiter_test.go:136: allowed: 1167444,    denied: 3462175,     time per allow:   1.031µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-8                 \t 4268707\t       283.6 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-8\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     561ns\n    ratelimiter_test.go:136: allowed: 4390,       denied: 5610,        time per allow:     845ns\n    ratelimiter_test.go:136: allowed: 287705,     denied: 712295,      time per allow:   1.219µs\n    ratelimiter_test.go:136: allowed: 901533,     denied: 2517796,     time per allow:   1.066µs\n    ratelimiter_test.go:136: allowed: 1133229,    denied: 3135478,     time per allow:   1.068µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-32                \t 2863960\t       444.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/wrapped-32\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     755ns\n    ratelimiter_test.go:136: allowed: 6306,       denied: 3694,        time per allow:     942ns\n    ratelimiter_test.go:136: allowed: 390106,     denied: 609894,      time per allow:   1.074µs\n    ratelimiter_test.go:136: allowed: 1178002,    denied: 1685958,     time per allow:    1.08µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource         \t 6881562\t       176.7 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     395ns\n    ratelimiter_test.go:136: allowed: 1999,       denied: 8001,        time per allow:     795ns\n    ratelimiter_test.go:136: allowed: 100999,     denied: 899001,      time per allow:   1.726µs\n    ratelimiter_test.go:136: allowed: 689156,     denied: 6192406,     time per allow:   1.764µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-2       \t 5739153\t       209.1 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-2\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:   1.123µs\n    ratelimiter_test.go:136: allowed: 1986,       denied: 8014,        time per allow:   1.146µs\n    ratelimiter_test.go:136: allowed: 100400,     denied: 899600,      time per allow:   2.082µs\n    ratelimiter_test.go:136: allowed: 570975,     denied: 5168178,     time per allow:   2.101µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-4       \t 4748774\t       252.8 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-4\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     511ns\n    ratelimiter_test.go:136: allowed: 1961,       denied: 8039,        time per allow:   1.445µs\n    ratelimiter_test.go:136: allowed: 97522,      denied: 902478,      time per allow:   2.591µs\n    ratelimiter_test.go:136: allowed: 459408,     denied: 4289366,     time per allow:   2.613µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-8       \t 4080715\t       290.5 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-8\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     474ns\n    ratelimiter_test.go:136: allowed: 1929,       denied: 8071,        time per allow:   1.568µs\n    ratelimiter_test.go:136: allowed: 93842,      denied: 906158,      time per allow:   3.133µs\n    ratelimiter_test.go:136: allowed: 379398,     denied: 3701317,     time per allow:   3.124µs\nBenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-32      \t 3506800\t       351.2 ns/op\n--- BENCH: BenchmarkLimiter/reserve-canceling-every-3/parallel/mocked_timesource-32\n    ratelimiter_test.go:136: allowed: 0,          denied: 1,           time per allow:       n/a\n    ratelimiter_test.go:136: allowed: 66,         denied: 34,          time per allow:     931ns\n    ratelimiter_test.go:136: allowed: 1925,       denied: 8075,        time per allow:   2.409µs\n    ratelimiter_test.go:136: allowed: 90771,      denied: 909229,      time per allow:   3.769µs\n    ratelimiter_test.go:136: allowed: 317064,     denied: 3189736,     time per allow:   3.884µs\nBenchmarkLimiter/wait/serial/2ns_rate/real                                    \t 2271784\t       482.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  12.833µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     475ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     479ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     528ns\n    ratelimiter_test.go:114: allowed: 2271784,    denied: 0,           time per allow:     482ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-2                                  \t 3010875\t       403.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  14.583µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     524ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     434ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     398ns\n    ratelimiter_test.go:114: allowed: 3010875,    denied: 0,           time per allow:     402ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-4                                  \t 3046982\t       388.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  11.667µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     462ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     405ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     393ns\n    ratelimiter_test.go:114: allowed: 3046982,    denied: 0,           time per allow:     388ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-8                                  \t 2938251\t       399.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  13.459µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     491ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     408ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     408ns\n    ratelimiter_test.go:114: allowed: 2938251,    denied: 0,           time per allow:     399ns\nBenchmarkLimiter/wait/serial/2ns_rate/real-32                                 \t 2992760\t       394.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/real-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  11.083µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     493ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     415ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     400ns\n    ratelimiter_test.go:114: allowed: 2992760,    denied: 0,           time per allow:     393ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped                                 \t 2425740\t       506.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  21.791µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     653ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     397ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     494ns\n    ratelimiter_test.go:114: allowed: 2425740,    denied: 0,           time per allow:     506ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-2                               \t 2937199\t       407.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  10.875µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     454ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     446ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     408ns\n    ratelimiter_test.go:114: allowed: 2937199,    denied: 0,           time per allow:     407ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-4                               \t 3006384\t       394.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   9.167µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     506ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     424ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     399ns\n    ratelimiter_test.go:114: allowed: 3006384,    denied: 0,           time per allow:     394ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-8                               \t 2951150\t       426.7 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  13.084µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     526ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     404ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     406ns\n    ratelimiter_test.go:114: allowed: 2951150,    denied: 0,           time per allow:     426ns\nBenchmarkLimiter/wait/serial/2ns_rate/wrapped-32                              \t 2990496\t       403.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/wrapped-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  10.125µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     515ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     401ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     401ns\n    ratelimiter_test.go:114: allowed: 2990496,    denied: 0,           time per allow:     402ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource                       \t 5608531\t       214.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:     8.5µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     273ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     184ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     213ns\n    ratelimiter_test.go:114: allowed: 5608531,    denied: 0,           time per allow:     214ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-2                     \t 6328546\t       189.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  10.709µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     582ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     193ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 6328546,    denied: 0,           time per allow:     189ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-4                     \t 6209492\t       189.4 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:    13.5µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     346ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     193ns\n    ratelimiter_test.go:114: allowed: 6209492,    denied: 0,           time per allow:     189ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-8                     \t 6340005\t       196.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  22.041µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     339ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     202ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 6340005,    denied: 0,           time per allow:     196ns\nBenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-32                    \t 6319386\t       189.4 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/2ns_rate/mocked_timesource-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   13.75µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     311ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 6319386,    denied: 0,           time per allow:     189ns\nBenchmarkLimiter/wait/serial/1µs_rate/real                                    \t 1201171\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  17.084µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     502ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201171,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/real-2                                  \t 1000000\t      1037 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.542µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     581ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:   1.037µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-4                                  \t 1201134\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.083µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     555ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201134,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/real-8                                  \t 1000000\t      1043 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  11.709µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     505ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:   1.042µs\nBenchmarkLimiter/wait/serial/1µs_rate/real-32                                 \t 1201106\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/real-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  13.417µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     549ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201106,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped                                 \t 1201178\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   10.75µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     865ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201178,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-2                               \t 1000000\t      1049 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  32.458µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     503ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:   1.048µs\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-4                               \t 1201174\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.084µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     557ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201174,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-8                               \t 1201167\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   9.458µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     714ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201167,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/wrapped-32                              \t 1201171\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/wrapped-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   8.125µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     513ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     900ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:114: allowed: 1201171,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource                       \t 5592026\t       236.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   9.041µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     271ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     191ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     214ns\n    ratelimiter_test.go:114: allowed: 5592026,    denied: 0,           time per allow:     236ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-2                     \t 6447422\t       186.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-2\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  21.375µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     355ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     188ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     186ns\n    ratelimiter_test.go:114: allowed: 6447422,    denied: 0,           time per allow:     186ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-4                     \t 6362046\t       186.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-4\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:   11.75µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     374ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     193ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     188ns\n    ratelimiter_test.go:114: allowed: 6362046,    denied: 0,           time per allow:     186ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-8                     \t 6338952\t       189.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-8\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:  24.583µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     283ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     187ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     189ns\n    ratelimiter_test.go:114: allowed: 6338952,    denied: 0,           time per allow:     189ns\nBenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-32                    \t 6387135\t       187.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/serial/1µs_rate/mocked_timesource-32\n    ratelimiter_test.go:114: allowed: 1,          denied: 0,           time per allow:      13µs\n    ratelimiter_test.go:114: allowed: 100,        denied: 0,           time per allow:     350ns\n    ratelimiter_test.go:114: allowed: 10000,      denied: 0,           time per allow:     192ns\n    ratelimiter_test.go:114: allowed: 1000000,    denied: 0,           time per allow:     187ns\n    ratelimiter_test.go:114: allowed: 6387135,    denied: 0,           time per allow:     187ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real                                  \t 2533497\t       495.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  63.791µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     505ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     506ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     473ns\n    ratelimiter_test.go:136: allowed: 2533497,    denied: 0,           time per allow:     495ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-2                                \t 4293474\t       281.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  19.958µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     608ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     291ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     279ns\n    ratelimiter_test.go:136: allowed: 4293474,    denied: 0,           time per allow:     281ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-4                                \t 3730892\t       321.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  20.541µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     635ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     342ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     321ns\n    ratelimiter_test.go:136: allowed: 3730892,    denied: 0,           time per allow:     321ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-8                                \t 2580943\t       466.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  19.667µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     551ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     451ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     464ns\n    ratelimiter_test.go:136: allowed: 2580943,    denied: 0,           time per allow:     466ns\nBenchmarkLimiter/wait/parallel/2ns_rate/real-32                               \t 1901340\t       614.3 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/real-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  38.334µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     619ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     644ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     631ns\n    ratelimiter_test.go:136: allowed: 1901340,    denied: 0,           time per allow:     614ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped                               \t 2373084\t       537.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  30.208µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     495ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     551ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     505ns\n    ratelimiter_test.go:136: allowed: 2373084,    denied: 0,           time per allow:     537ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-2                             \t 3986494\t       302.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  21.083µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     455ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     320ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     301ns\n    ratelimiter_test.go:136: allowed: 3986494,    denied: 0,           time per allow:     302ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-4                             \t 3463202\t       346.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  38.709µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     753ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     374ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     346ns\n    ratelimiter_test.go:136: allowed: 3463202,    denied: 0,           time per allow:     346ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-8                             \t 1935219\t       651.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  17.083µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     461ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     644ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     620ns\n    ratelimiter_test.go:136: allowed: 1935219,    denied: 0,           time per allow:     651ns\nBenchmarkLimiter/wait/parallel/2ns_rate/wrapped-32                            \t 1308968\t       907.9 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/wrapped-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  25.458µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     562ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     955ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     916ns\n    ratelimiter_test.go:136: allowed: 1308968,    denied: 0,           time per allow:     907ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource                     \t 5477860\t       221.1 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  35.083µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     313ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     196ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     219ns\n    ratelimiter_test.go:136: allowed: 5477860,    denied: 0,           time per allow:     221ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-2                   \t 4676773\t       256.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  31.334µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     410ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     303ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     256ns\n    ratelimiter_test.go:136: allowed: 4676773,    denied: 0,           time per allow:     256ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-4                   \t 4149928\t       285.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  22.875µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     375ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     337ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     289ns\n    ratelimiter_test.go:136: allowed: 4149928,    denied: 0,           time per allow:     284ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-8                   \t 3410410\t       351.4 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  16.459µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     400ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     357ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     351ns\n    ratelimiter_test.go:136: allowed: 3410410,    denied: 0,           time per allow:     351ns\nBenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-32                  \t 2806215\t       442.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/2ns_rate/mocked_timesource-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  32.125µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     457ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     398ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     427ns\n    ratelimiter_test.go:136: allowed: 2806215,    denied: 0,           time per allow:     442ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real                                  \t 1201167\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  43.708µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     543ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:136: allowed: 1201167,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-2                                \t 1215512\t       987.4 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  14.625µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     479ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     883ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     987ns\n    ratelimiter_test.go:136: allowed: 1215512,    denied: 0,           time per allow:     987ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-4                                \t 1280577\t       926.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  14.875µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     432ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     846ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     937ns\n    ratelimiter_test.go:136: allowed: 1280577,    denied: 0,           time per allow:     926ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-8                                \t 2569482\t       456.5 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  24.917µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     395ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     428ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     467ns\n    ratelimiter_test.go:136: allowed: 2569482,    denied: 0,           time per allow:     456ns\nBenchmarkLimiter/wait/parallel/1µs_rate/real-32                               \t 1904204\t       597.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/real-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  34.334µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     475ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     562ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     630ns\n    ratelimiter_test.go:136: allowed: 1904204,    denied: 0,           time per allow:     597ns\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped                               \t 1201110\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  81.292µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     587ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:136: allowed: 1201110,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-2                             \t 1000000\t      1045 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  18.166µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     797ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:   1.044µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-4                             \t 1201161\t      1007 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  16.666µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     702ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:136: allowed: 1201161,    denied: 0,           time per allow:   1.006µs\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-8                             \t 1201149\t       999.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  20.417µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     921ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     901ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     999ns\n    ratelimiter_test.go:136: allowed: 1201149,    denied: 0,           time per allow:     999ns\nBenchmarkLimiter/wait/parallel/1µs_rate/wrapped-32                            \t 1000000\t      1006 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/wrapped-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  25.416µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     570ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     971ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:   1.006µs\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource                     \t 5436198\t       231.0 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  39.084µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     281ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     196ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     220ns\n    ratelimiter_test.go:136: allowed: 5436198,    denied: 0,           time per allow:     231ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-2                   \t 4571574\t       262.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-2\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  14.334µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     554ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     257ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     262ns\n    ratelimiter_test.go:136: allowed: 4571574,    denied: 0,           time per allow:     262ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-4                   \t 4076700\t       292.6 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-4\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:  16.291µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     583ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     338ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     294ns\n    ratelimiter_test.go:136: allowed: 4076700,    denied: 0,           time per allow:     292ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-8                   \t 2949456\t       356.2 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-8\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:    19.5µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     347ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     313ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     406ns\n    ratelimiter_test.go:136: allowed: 2949456,    denied: 0,           time per allow:     356ns\nBenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-32                  \t 2807648\t       433.8 ns/op\n--- BENCH: BenchmarkLimiter/wait/parallel/1µs_rate/mocked_timesource-32\n    ratelimiter_test.go:136: allowed: 1,          denied: 0,           time per allow:   53.75µs\n    ratelimiter_test.go:136: allowed: 100,        denied: 0,           time per allow:     431ns\n    ratelimiter_test.go:136: allowed: 10000,      denied: 0,           time per allow:     539ns\n    ratelimiter_test.go:136: allowed: 1000000,    denied: 0,           time per allow:     427ns\n    ratelimiter_test.go:136: allowed: 2807648,    denied: 0,           time per allow:     433ns\nPASS\nok  \tgithub.com/uber/cadence/common/clock\t199.030s\n"
  },
  {
    "path": "common/clock/time_source.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage clock\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/jonboulle/clockwork\"\n)\n\ntype (\n\t// TimeSource provides an interface that packages can use instead of directly using\n\t// the [time] module, so that chronology-related behavior can be tested.\n\tTimeSource interface {\n\t\tAfter(d time.Duration) <-chan time.Time\n\t\tSleep(d time.Duration)\n\t\tSleepWithContext(ctx context.Context, d time.Duration) error\n\t\tNow() time.Time\n\t\tSince(t time.Time) time.Duration\n\t\tNewTicker(d time.Duration) Ticker\n\t\tNewTimer(d time.Duration) Timer\n\t\tAfterFunc(d time.Duration, f func()) Timer\n\t\tContextWithTimeout(context.Context, time.Duration) (context.Context, context.CancelFunc)\n\t\tContextWithDeadline(context.Context, time.Time) (context.Context, context.CancelFunc)\n\t}\n\n\t// Ticker provides an interface which can be used instead of directly using\n\t// [time.Ticker]. The real-time ticker t provides ticks through t.C which\n\t// becomes t.Chan() to make this channel requirement definable in this\n\t// interface.\n\tTicker interface {\n\t\tChan() <-chan time.Time\n\t\tReset(d time.Duration)\n\t\tStop()\n\t}\n\n\t// Timer provides an interface which can be used instead of directly using\n\t// [time.Timer]. The real-time timer t provides events through t.C which becomes\n\t// t.Chan() to make this channel requirement definable in this interface.\n\tTimer interface {\n\t\tChan() <-chan time.Time\n\t\tReset(d time.Duration) bool\n\t\tStop() bool\n\t}\n\n\t// clock serves real wall-clock time\n\tclock struct {\n\t\tclockwork.Clock\n\t}\n\n\t// fakeClock serves fake controlled time\n\tfakeClock struct {\n\t\t*clockwork.FakeClock\n\t}\n\n\t// MockedTimeSource provides an interface for a clock which can be manually advanced\n\t// through time.\n\t//\n\t// MockedTimeSource maintains a list of \"waiters,\" which consists of all callers\n\t// waiting on the underlying clock (i.e. Tickers and Timers including callers of\n\t// Sleep or After). Users can call BlockUntil to block until the clock has an\n\t// expected number of waiters.\n\tMockedTimeSource interface {\n\t\tTimeSource\n\t\t// Advance advances the FakeClock to a new point in time, ensuring any existing\n\t\t// waiters are notified appropriately before returning.\n\t\tAdvance(d time.Duration)\n\t\t// BlockUntil blocks until the FakeClock has at least the given number\n\t\t// of waiters running at the same time.\n\t\t//\n\t\t// Waiters are either time.Sleep, time.After[Func], time.Ticker, or time.Timer,\n\t\t// and they decrement the counter when they complete or are stopped.\n\t\tBlockUntil(waiters int)\n\t}\n)\n\n// NewRealTimeSource returns a time source that servers\n// real wall clock time\nfunc NewRealTimeSource() TimeSource {\n\treturn &clock{\n\t\tClock: clockwork.NewRealClock(),\n\t}\n}\n\n// NewMockedTimeSourceAt returns a time source that servers\n// fake controlled time. The initial time of the MockedTimeSource will be the given time.\nfunc NewMockedTimeSourceAt(t time.Time) MockedTimeSource {\n\treturn &fakeClock{\n\t\tFakeClock: clockwork.NewFakeClockAt(t),\n\t}\n}\n\nfunc (r *clock) ContextWithTimeout(ctx context.Context, d time.Duration) (context.Context, context.CancelFunc) {\n\treturn context.WithTimeout(ctx, d)\n}\n\nfunc (r *clock) ContextWithDeadline(ctx context.Context, t time.Time) (context.Context, context.CancelFunc) {\n\treturn context.WithDeadline(ctx, t)\n}\n\nfunc (r *clock) NewTicker(d time.Duration) Ticker {\n\treturn r.Clock.NewTicker(d)\n}\n\nfunc (r *clock) NewTimer(d time.Duration) Timer {\n\treturn r.Clock.NewTimer(d)\n}\n\nfunc (r *clock) AfterFunc(d time.Duration, f func()) Timer {\n\treturn r.Clock.AfterFunc(d, f)\n}\n\nfunc (r *clock) SleepWithContext(ctx context.Context, duration time.Duration) error {\n\tselect {\n\tcase <-r.After(duration):\n\t\treturn nil\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\t}\n}\n\n// NewMockedTimeSource returns a time source that servers\n// fake controlled time\nfunc NewMockedTimeSource() MockedTimeSource {\n\treturn &fakeClock{\n\t\tFakeClock: clockwork.NewFakeClock(),\n\t}\n}\n\nfunc (c *fakeClock) NewTicker(d time.Duration) Ticker {\n\treturn c.FakeClock.NewTicker(d)\n}\n\nfunc (c *fakeClock) NewTimer(d time.Duration) Timer {\n\treturn c.FakeClock.NewTimer(d)\n}\n\nfunc (c *fakeClock) AfterFunc(d time.Duration, f func()) Timer {\n\treturn c.FakeClock.AfterFunc(d, f)\n}\n\nfunc (c *fakeClock) ContextWithTimeout(ctx context.Context, duration time.Duration) (context.Context, context.CancelFunc) {\n\treturn clockwork.WithTimeout(ctx, c.FakeClock, duration)\n}\n\nfunc (c *fakeClock) ContextWithDeadline(ctx context.Context, deadline time.Time) (context.Context, context.CancelFunc) {\n\treturn clockwork.WithDeadline(ctx, c.FakeClock, deadline)\n}\n\nfunc (c *fakeClock) SleepWithContext(ctx context.Context, duration time.Duration) error {\n\tselect {\n\tcase <-c.After(duration):\n\t\treturn nil\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\t}\n}\n"
  },
  {
    "path": "common/clock/timer_gate.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package=$GOPACKAGE -destination=timer_gate_mock.go github.com/uber/cadence/common/clock TimerGate\npackage clock\n\nimport (\n\t\"sync\"\n\t\"time\"\n)\n\ntype (\n\t// TimerGate is a timer interface which fires a timer at a given timestamp.\n\tTimerGate interface {\n\t\t// Chan return the channel which will be fired when time is up\n\t\tChan() <-chan time.Time\n\t\t// FireAfter check will the timer get fired after a certain time\n\t\tFireAfter(t time.Time) bool\n\t\t// Update update the timer gate, return true if update is a success\n\t\t// success means timer is idle or timer is set with a sooner time to fire\n\t\tUpdate(t time.Time) bool\n\t\t// Close shutdown the timer\n\t\tStop()\n\t}\n\n\ttimerGateImpl struct {\n\t\tsync.RWMutex\n\t\ttimeSource TimeSource\n\t\ttimer      Timer\n\t\tfireTime   time.Time\n\t}\n)\n\nfunc NewTimerGate(timeSource TimeSource) TimerGate {\n\tt := &timerGateImpl{\n\t\ttimer:      timeSource.NewTimer(0),\n\t\ttimeSource: timeSource,\n\t}\n\t// the timer should be stopped when initialized\n\tif !t.timer.Stop() {\n\t\t// drain the existing signal if exist\n\t\t// TODO: the drain can be removed after go 1.23\n\t\t<-t.timer.Chan()\n\t}\n\treturn t\n}\n\nfunc (t *timerGateImpl) Chan() <-chan time.Time {\n\treturn t.timer.Chan()\n}\n\nfunc (t *timerGateImpl) FireAfter(now time.Time) bool {\n\tt.RLock()\n\tdefer t.RUnlock()\n\treturn t.fireTime.After(now)\n}\n\nfunc (t *timerGateImpl) Stop() {\n\tt.Lock()\n\tdefer t.Unlock()\n\tt.fireTime = time.Time{}\n\tt.timer.Stop()\n\t// TODO: the drain can be removed after go 1.23\n\tselect {\n\tcase <-t.timer.Chan():\n\tdefault:\n\t}\n}\n\nfunc (t *timerGateImpl) Update(fireTime time.Time) bool {\n\tt.Lock()\n\tdefer t.Unlock()\n\tif t.timer.Stop() {\n\t\tif t.fireTime.Before(fireTime) {\n\t\t\t// this means the timer, before stopped, is active && next wake up time do not have to be updated\n\t\t\tt.timer.Reset(t.fireTime.Sub(t.timeSource.Now()))\n\t\t\treturn false\n\t\t}\n\t}\n\t// TODO: the drain can be removed after go 1.23\n\tselect {\n\tcase <-t.timer.Chan():\n\tdefault:\n\t}\n\t// this means the timer, before stopped, is active && next wake up time has to be updated\n\t// or this means the timer, before stopped, is already fired / never active\n\tt.fireTime = fireTime\n\tt.timer.Reset(fireTime.Sub(t.timeSource.Now()))\n\t// Notifies caller that next notification is reset to fire at passed in 'next' visibility time\n\treturn true\n}\n"
  },
  {
    "path": "common/clock/timer_gate_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/clock (interfaces: TimerGate)\n//\n// Generated by this command:\n//\n//\tmockgen -package=clock -destination=timer_gate_mock.go github.com/uber/cadence/common/clock TimerGate\n//\n\n// Package clock is a generated GoMock package.\npackage clock\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockTimerGate is a mock of TimerGate interface.\ntype MockTimerGate struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTimerGateMockRecorder\n\tisgomock struct{}\n}\n\n// MockTimerGateMockRecorder is the mock recorder for MockTimerGate.\ntype MockTimerGateMockRecorder struct {\n\tmock *MockTimerGate\n}\n\n// NewMockTimerGate creates a new mock instance.\nfunc NewMockTimerGate(ctrl *gomock.Controller) *MockTimerGate {\n\tmock := &MockTimerGate{ctrl: ctrl}\n\tmock.recorder = &MockTimerGateMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTimerGate) EXPECT() *MockTimerGateMockRecorder {\n\treturn m.recorder\n}\n\n// Chan mocks base method.\nfunc (m *MockTimerGate) Chan() <-chan time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Chan\")\n\tret0, _ := ret[0].(<-chan time.Time)\n\treturn ret0\n}\n\n// Chan indicates an expected call of Chan.\nfunc (mr *MockTimerGateMockRecorder) Chan() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Chan\", reflect.TypeOf((*MockTimerGate)(nil).Chan))\n}\n\n// FireAfter mocks base method.\nfunc (m *MockTimerGate) FireAfter(t time.Time) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FireAfter\", t)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// FireAfter indicates an expected call of FireAfter.\nfunc (mr *MockTimerGateMockRecorder) FireAfter(t any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FireAfter\", reflect.TypeOf((*MockTimerGate)(nil).FireAfter), t)\n}\n\n// Stop mocks base method.\nfunc (m *MockTimerGate) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockTimerGateMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockTimerGate)(nil).Stop))\n}\n\n// Update mocks base method.\nfunc (m *MockTimerGate) Update(t time.Time) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Update\", t)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Update indicates an expected call of Update.\nfunc (mr *MockTimerGateMockRecorder) Update(t any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Update\", reflect.TypeOf((*MockTimerGate)(nil).Update), t)\n}\n"
  },
  {
    "path": "common/clock/timer_gate_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clock\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\ttimerGateSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\ttimerGate TimerGate\n\t}\n)\n\nfunc TestTimerGeteSuite(t *testing.T) {\n\ts := new(timerGateSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerGateSuite) SetupSuite() {\n\n}\n\nfunc (s *timerGateSuite) TearDownSuite() {\n\n}\n\nfunc (s *timerGateSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.timerGate = NewTimerGate(NewRealTimeSource())\n}\n\nfunc (s *timerGateSuite) TearDownTest() {\n\ts.timerGate.Stop()\n}\n\nfunc (s *timerGateSuite) TestTimerFire() {\n\tnow := time.Now()\n\tnewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire before test deadline\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerFireAfterUpdate_Active_Updated_BeforeNow() {\n\tnow := time.Now()\n\tnewTimer := now.Add(9 * time.Second)\n\tupdatedNewTimer := now.Add(-1 * time.Second)\n\tdeadline := now.Add(3 * time.Second)\n\n\ts.True(s.timerGate.Update(newTimer))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\t\ts.Fail(\"timer should not fire when current time not updated\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\ts.True(s.timerGate.Update(updatedNewTimer))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire before test deadline\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerFireAfterUpdate_Active_Updated() {\n\tnow := time.Now()\n\tnewTimer := now.Add(5 * time.Second)\n\tupdatedNewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(3 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\ts.True(s.timerGate.Update(updatedNewTimer))\n\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire before test deadline\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerFireAfterUpdate_Active_NotUpdated() {\n\tnow := time.Now()\n\tnewTimer := now.Add(1 * time.Second)\n\tupdatedNewTimer := now.Add(3 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\ts.False(s.timerGate.Update(updatedNewTimer))\n\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire before test deadline\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerFireAfterUpdate_NotActive_Updated() {\n\tnow := time.Now()\n\tnewTimer := now.Add(-5 * time.Second)\n\tupdatedNewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(3 * time.Second)\n\n\ts.True(s.timerGate.Update(newTimer))\n\t// this is to drain existing signal\n\t<-s.timerGate.Chan()\n\n\t// test setup up complete\n\n\ts.True(s.timerGate.Update(updatedNewTimer))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire before test deadline\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerWillFire_Zero() {\n\t// this test is to validate initial notification will trigger a scan of timer\n\ts.True(s.timerGate.Update(time.Time{}))\n\ts.False(s.timerGate.FireAfter(time.Now()))\n\n\tselect { // this is to drain existing signal\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(time.Second).C:\n\t\ts.Fail(\"timer should fire\")\n\t}\n\n\tnow := time.Now()\n\tnewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire\")\n\t}\n\ts.True(s.timerGate.Update(time.Time{}))\n\tselect { // this is to drain existing signal\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(time.Second).C:\n\t\ts.Fail(\"timer should fire\")\n\t}\n\n\tnow = time.Now()\n\tnewTimer = now.Add(1 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\ts.True(s.timerGate.Update(time.Time{}))\n\tselect { // this is to drain existing signal\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(time.Second).C:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerFireAfterStopAndUpdate() {\n\tnow := time.Now()\n\tnewTimer := now.Add(1 * time.Second)\n\tdeadline := now.Add(2 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\ts.True(s.timerGate.FireAfter(newTimer.Add(-1 * time.Nanosecond)))\n\ts.timerGate.Stop()\n\ts.False(s.timerGate.FireAfter(newTimer.Add(-1 * time.Nanosecond)))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\t\ts.Fail(\"timer should be cancelled\")\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t}\n\n\tnow = time.Now()\n\tnewTimer = now.Add(1 * time.Second)\n\tdeadline = now.Add(2 * time.Second)\n\ts.True(s.timerGate.Update(newTimer))\n\ts.True(s.timerGate.FireAfter(newTimer.Add(-1 * time.Nanosecond)))\n\tselect {\n\tcase <-s.timerGate.Chan():\n\tcase <-time.NewTimer(deadline.Sub(now)).C:\n\t\ts.Fail(\"timer should fire\")\n\t}\n}\n\nfunc (s *timerGateSuite) TestTimerWillFire() {\n\tnow := time.Now()\n\tnewTimer := now.Add(2 * time.Second)\n\ttimeBeforeNewTimer := now.Add(1 * time.Second)\n\ttimeAfterNewTimer := now.Add(3 * time.Second)\n\ts.timerGate.Update(newTimer)\n\ts.True(s.timerGate.FireAfter(timeBeforeNewTimer))\n\ts.False(s.timerGate.FireAfter(timeAfterNewTimer))\n\ts.timerGate.Stop()\n\ts.False(s.timerGate.FireAfter(timeBeforeNewTimer))\n}\n"
  },
  {
    "path": "common/cluster/metadata.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cluster\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// Metadata provides information about clusters\n\tMetadata struct {\n\t\tlog     log.Logger\n\t\tmetrics metrics.Scope\n\n\t\t// failoverVersionIncrement is the increment of each cluster's version when failover happen\n\t\tfailoverVersionIncrement int64\n\t\t// primaryClusterName is the name of the primary cluster, only the primary cluster can register / update domain\n\t\t// all clusters can do domain failover\n\t\tprimaryClusterName string\n\t\t// currentClusterName is the name of the current cluster\n\t\tcurrentClusterName string\n\t\t// allClusters contains all cluster info\n\t\tallClusters map[string]config.ClusterInformation\n\t\t// enabledClusters contains enabled info\n\t\tenabledClusters map[string]config.ClusterInformation\n\t\t// remoteClusters contains enabled and remote info\n\t\tremoteClusters map[string]config.ClusterInformation\n\t\t// versionToClusterName contains all initial version -> corresponding cluster name\n\t\tversionToClusterName map[int64]string\n\t\t// allows for a new failover version migration\n\t\tuseNewFailoverVersionOverride dynamicproperties.BoolPropertyFnWithDomainFilter\n\t}\n)\n\n// NewMetadata create a new instance of Metadata\nfunc NewMetadata(\n\tclusterGroupMetadata config.ClusterGroupMetadata,\n\tuseMinFailoverVersionOverrideConfig dynamicproperties.BoolPropertyFnWithDomainFilter,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) Metadata {\n\tversionToClusterName := make(map[int64]string)\n\tfor clusterName, info := range clusterGroupMetadata.ClusterGroup {\n\t\tversionToClusterName[info.InitialFailoverVersion] = clusterName\n\t}\n\n\t// We never use disable clusters, filter them out on start\n\tenabledClusters := map[string]config.ClusterInformation{}\n\tfor cluster, info := range clusterGroupMetadata.ClusterGroup {\n\t\tif info.Enabled {\n\t\t\tenabledClusters[cluster] = info\n\t\t}\n\t}\n\n\t// Precompute remote clusters, they are used in multiple places\n\tremoteClusters := map[string]config.ClusterInformation{}\n\tfor cluster, info := range enabledClusters {\n\t\tif cluster != clusterGroupMetadata.CurrentClusterName {\n\t\t\tremoteClusters[cluster] = info\n\t\t}\n\t}\n\n\tm := Metadata{\n\t\tlog:                           logger,\n\t\tmetrics:                       metricsClient.Scope(metrics.ClusterMetadataScope),\n\t\tfailoverVersionIncrement:      clusterGroupMetadata.FailoverVersionIncrement,\n\t\tprimaryClusterName:            clusterGroupMetadata.PrimaryClusterName,\n\t\tcurrentClusterName:            clusterGroupMetadata.CurrentClusterName,\n\t\tallClusters:                   clusterGroupMetadata.ClusterGroup,\n\t\tenabledClusters:               enabledClusters,\n\t\tremoteClusters:                remoteClusters,\n\t\tversionToClusterName:          versionToClusterName,\n\t\tuseNewFailoverVersionOverride: useMinFailoverVersionOverrideConfig,\n\t}\n\n\tm.log.Info(\"cluster metadata created\",\n\t\ttag.Dynamic(\"primary-cluster-name\", m.primaryClusterName),\n\t\ttag.Dynamic(\"current-cluster-name\", m.currentClusterName),\n\t\ttag.Dynamic(\"failover-version-increment\", m.failoverVersionIncrement),\n\t)\n\n\treturn m\n}\n\n// GetNextFailoverVersion returns the next valid FailoverVersion for a domain\nfunc (m Metadata) GetNextFailoverVersion(targetClusterName string, currentFailoverVersion int64, domainName string) int64 {\n\tinitialFailoverVersion := m.getInitialFailoverVersion(targetClusterName, domainName)\n\tfailoverVersion := currentFailoverVersion/m.failoverVersionIncrement*m.failoverVersionIncrement + initialFailoverVersion\n\tif failoverVersion < currentFailoverVersion {\n\t\treturn failoverVersion + m.failoverVersionIncrement\n\t}\n\treturn failoverVersion\n}\n\n// IsVersionFromSameCluster return true if the new version is used for the same cluster\nfunc (m Metadata) IsVersionFromSameCluster(version1 int64, version2 int64) bool {\n\tv1Server, err := m.resolveServerName(version1)\n\tif err != nil {\n\t\t// preserving old behaviour however, this should never occur\n\t\tm.metrics.IncCounter(metrics.ClusterMetadataFailureToResolveCounter)\n\t\tm.log.Error(\"could not resolve an incoming version\", tag.Dynamic(\"failover-version\", version1))\n\t\treturn false\n\t}\n\tv2Server, err := m.resolveServerName(version2)\n\tif err != nil {\n\t\tm.log.Error(\"could not resolve an incoming version\", tag.Dynamic(\"failover-version\", version2))\n\t\treturn false\n\t}\n\treturn v1Server == v2Server\n}\n\nfunc (m Metadata) IsPrimaryCluster() bool {\n\treturn m.primaryClusterName == m.currentClusterName\n}\n\n// GetCurrentClusterName return the current cluster name\nfunc (m Metadata) GetCurrentClusterName() string {\n\treturn m.currentClusterName\n}\n\n// GetAllClusterInfo return all cluster info\nfunc (m Metadata) GetAllClusterInfo() map[string]config.ClusterInformation {\n\treturn m.allClusters\n}\n\n// GetEnabledClusterInfo return enabled cluster info\nfunc (m Metadata) GetEnabledClusterInfo() map[string]config.ClusterInformation {\n\treturn m.enabledClusters\n}\n\n// GetRemoteClusterInfo return enabled AND remote cluster info\nfunc (m Metadata) GetRemoteClusterInfo() map[string]config.ClusterInformation {\n\treturn m.remoteClusters\n}\n\n// ClusterNameForFailoverVersion return the corresponding cluster name for a given failover version\nfunc (m Metadata) ClusterNameForFailoverVersion(failoverVersion int64) (string, error) {\n\tif failoverVersion == constants.EmptyVersion {\n\t\treturn m.currentClusterName, nil\n\t}\n\tserver, err := m.resolveServerName(failoverVersion)\n\tif err != nil {\n\t\tm.metrics.IncCounter(metrics.ClusterMetadataResolvingFailoverVersionCounter)\n\t\treturn \"\", fmt.Errorf(\"failed to resolve failover version to a cluster: %v\", err)\n\t}\n\treturn server, nil\n}\n\n// gets the initial failover version for a cluster / domain\n// along with some helpers for a migration - should it be necessary\nfunc (m Metadata) getInitialFailoverVersion(cluster string, domainName string) int64 {\n\tinfo, ok := m.allClusters[cluster]\n\tif !ok {\n\t\tpanic(fmt.Sprintf(\n\t\t\t\"Unknown cluster name: %v with given cluster initial failover version map: %v.\",\n\t\t\tcluster,\n\t\t\tm.allClusters,\n\t\t))\n\t}\n\n\t// if using the minFailover Version during a cluster config, then return this from config\n\t// (assuming it's safe to do so). This is not the normal state of things and intended only\n\t// for when migrating versions.\n\tusingNewFailoverVersion := m.useNewFailoverVersionOverride(domainName)\n\tif usingNewFailoverVersion && info.NewInitialFailoverVersion != nil {\n\t\tm.log.Debug(\"using new failover version for cluster\", tag.ClusterName(cluster), tag.WorkflowDomainName(domainName))\n\t\tm.metrics.IncCounter(metrics.ClusterMetadataGettingMinFailoverVersionCounter)\n\t\treturn *info.NewInitialFailoverVersion\n\t}\n\t// default behaviour - return the initial failover version - a marker to\n\t// identify the cluster for all counters\n\tm.log.Debug(\"getting failover version for cluster\", tag.ClusterName(cluster), tag.WorkflowDomainName(domainName))\n\tm.metrics.IncCounter(metrics.ClusterMetadataGettingFailoverVersionCounter)\n\treturn info.InitialFailoverVersion\n}\n\n// resolves the server name from a version number. Better to use this\n// than to check versionToClusterName directly, as this also falls back to catch\n// when there's a migration NewInitialFailoverVersion\nfunc (m Metadata) resolveServerName(originalVersion int64) (string, error) {\n\tversion := originalVersion % m.failoverVersionIncrement\n\t// attempt a lookup first\n\tserver, ok := m.versionToClusterName[version]\n\tif ok {\n\t\treturn server, nil\n\t}\n\n\t// else fall back on checking for new failover versions\n\tfor name, cluster := range m.allClusters {\n\t\tif cluster.NewInitialFailoverVersion != nil && *cluster.NewInitialFailoverVersion == version {\n\t\t\treturn name, nil\n\t\t}\n\t}\n\n\tm.metrics.IncCounter(metrics.ClusterMetadataFailureToResolveCounter)\n\treturn \"\", fmt.Errorf(\"could not resolve failover version: %d\", originalVersion)\n}\n"
  },
  {
    "path": "common/cluster/metadata_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cluster\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"testing\"\n\t\"testing/quick\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestMetadataBehaviour(t *testing.T) {\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst clusterName3 = \"c3\"\n\tconst initialFailoverVersionC3 = 4\n\n\tconst failoverVersionIncrement = 100\n\n\ttests := map[string]struct {\n\t\tfailoverCluster string\n\t\tcurrentVersion  int64\n\t\texpectedOut     int64\n\t}{\n\t\t\"a new domain, created in c1 already - a failover to c1 where it already is should have no effect\": {\n\t\t\tfailoverCluster: clusterName1,\n\t\t\tcurrentVersion:  0,\n\t\t\texpectedOut:     0,\n\t\t},\n\t\t\"a new domain, created in c1 already - a failover to c2 should set the failover version to be based on c2\": {\n\t\t\tfailoverCluster: clusterName2,\n\t\t\tcurrentVersion:  0,\n\t\t\texpectedOut:     2,\n\t\t},\n\t\t\"a failover to c3 should set the failover version to be based on c3\": {\n\t\t\tfailoverCluster: clusterName3,\n\t\t\tcurrentVersion:  2,\n\t\t\texpectedOut:     4,\n\t\t},\n\t\t\"a subsequent failover back to c1 should increment the failover version by failoverVersionIncrement\": {\n\t\t\tfailoverCluster: clusterName1,\n\t\t\tcurrentVersion:  2,\n\t\t\texpectedOut:     100,\n\t\t},\n\t\t\"when the current failover version matches the target cluster it should not increment the failover version\": {\n\t\t\tfailoverCluster: clusterName1,\n\t\t\tcurrentVersion:  100,\n\t\t\texpectedOut:     100,\n\t\t},\n\t\t\"and a subsequent fail back over to c2\": {\n\t\t\tfailoverCluster: clusterName2,\n\t\t\tcurrentVersion:  100,\n\t\t\texpectedOut:     102,\n\t\t},\n\t\t\"and a subsequent fail back over to c1 should skip over c3\": {\n\t\t\tfailoverCluster: clusterName1,\n\t\t\tcurrentVersion:  102,\n\t\t\texpectedOut:     200,\n\t\t},\n\t\t\"Ensuring the behaviour of getNextFailoverVersion can handle negative numbers 1\": {\n\t\t\tfailoverCluster: clusterName1,\n\t\t\tcurrentVersion:  -1,\n\t\t\texpectedOut:     0,\n\t\t},\n\t\t\"Ensuring the behaviour of getNextFailoverVersion can handle negative numbers 2\": {\n\t\t\tfailoverCluster: clusterName2,\n\t\t\tcurrentVersion:  -1,\n\t\t\texpectedOut:     2,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tm := Metadata{\n\t\t\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\t\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\t\t\tclusterName1: {\n\t\t\t\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t\t\t\t},\n\t\t\t\t\tclusterName2: {\n\t\t\t\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t\t\t\t},\n\t\t\t\t\tclusterName3: {\n\t\t\t\t\t\tInitialFailoverVersion: initialFailoverVersionC3,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tversionToClusterName: map[int64]string{\n\t\t\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t\t\t\tinitialFailoverVersionC3: clusterName3,\n\t\t\t\t},\n\t\t\t\tuseNewFailoverVersionOverride: func(domain string) bool { return false },\n\t\t\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\t\t\tlog:                           testlogger.New(t),\n\t\t\t}\n\t\t\tassert.Equal(t, td.expectedOut, m.GetNextFailoverVersion(td.failoverCluster, td.currentVersion, \"a domain\"), name)\n\t\t})\n\t}\n}\n\nfunc TestFailoverVersionLogicIsMonotonic(t *testing.T) {\n\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tvar minFailoverVersionC1 = int64(1)\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tvar minFailoverVersionC2 = int64(3)\n\n\tconst failoverVersionIncrement = 100\n\tconst someDomainMigrating = \"a domain migrating\"\n\tconst someDomainNotMigrating = \"a domain not migrating\"\n\n\tm := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\tclusterName1: {\n\t\t\t\tInitialFailoverVersion:    initialFailoverVersionC1,\n\t\t\t\tNewInitialFailoverVersion: &minFailoverVersionC1,\n\t\t\t},\n\t\t\tclusterName2: {\n\t\t\t\tInitialFailoverVersion:    initialFailoverVersionC2,\n\t\t\t\tNewInitialFailoverVersion: &minFailoverVersionC2,\n\t\t\t},\n\t\t},\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return someDomainMigrating == domain },\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tlog:                           testlogger.New(t),\n\t}\n\n\tcurrent := int64(0)\n\titerations := 0\n\n\tfor iterations < 4000 {\n\t\tnext := m.GetNextFailoverVersion(clusterName2, current, someDomainNotMigrating)\n\t\tassert.Truef(t, next > current, \"current: %d, next %d\", current, next)\n\t\tcurrent = next\n\n\t\tnext = m.GetNextFailoverVersion(clusterName1, current, someDomainNotMigrating)\n\t\tassert.Truef(t, next > current, \"current: %d, next %d\", current, next)\n\t\tcurrent = next\n\n\t\tnext = m.GetNextFailoverVersion(clusterName1, current, someDomainMigrating)\n\t\tassert.Truef(t, next > current, \"current: %d, next %d\", current, next)\n\t\tcurrent = next\n\t\titerations++\n\t}\n}\n\nfunc TestResolvingClusterVersion(t *testing.T) {\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tvar newInitialFailoverVersionC1 = int64(1)\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tvar newInitialFailoverVersionC2 = int64(3)\n\tconst failoverVersionIncrement = 100\n\n\ttests := map[string]struct {\n\t\tinput          int64\n\t\texpectedOutput string\n\t\texpectedErr    error\n\t}{\n\t\t\"first cluster, normal version\": {\n\t\t\tinput:          initialFailoverVersionC1,\n\t\t\texpectedOutput: clusterName1,\n\t\t},\n\t\t\"first cluster, minFailover version\": {\n\t\t\tinput:          newInitialFailoverVersionC1,\n\t\t\texpectedOutput: clusterName1,\n\t\t},\n\t\t\"second cluster, normal version\": {\n\t\t\tinput:          initialFailoverVersionC2,\n\t\t\texpectedOutput: clusterName2,\n\t\t},\n\t\t\"second cluster, minFailover version\": {\n\t\t\tinput:          newInitialFailoverVersionC2,\n\t\t\texpectedOutput: clusterName2,\n\t\t},\n\n\t\t\"first cluster, normal version - higher versions\": {\n\t\t\tinput:          200,\n\t\t\texpectedOutput: clusterName1,\n\t\t},\n\t\t\"first cluster, minFailover version - higher versions\": {\n\t\t\tinput:          101,\n\t\t\texpectedOutput: clusterName1,\n\t\t},\n\t\t\"second cluster, normal initialFailover version - higher versions\": {\n\t\t\tinput:          302,\n\t\t\texpectedOutput: clusterName2,\n\t\t},\n\t\t\"second cluster, minFailover version - higher versions\": {\n\t\t\tinput:          403,\n\t\t\texpectedOutput: clusterName2,\n\t\t},\n\n\t\t\"invalid input\": {\n\t\t\tinput:       599,\n\t\t\texpectedErr: fmt.Errorf(\"could not resolve failover version: 599\"),\n\t\t},\n\t}\n\n\tm := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\tclusterName1: {\n\t\t\t\tInitialFailoverVersion:    initialFailoverVersionC1,\n\t\t\t\tNewInitialFailoverVersion: &newInitialFailoverVersionC1,\n\t\t\t},\n\t\t\tclusterName2: {\n\t\t\t\tInitialFailoverVersion:    initialFailoverVersionC2,\n\t\t\t\tNewInitialFailoverVersion: &newInitialFailoverVersionC2,\n\t\t\t},\n\t\t},\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tmetrics: metrics.NewNoopMetricsClient().Scope(0),\n\t\tlog:     log.NewNoop(),\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(*testing.T) {\n\t\t\tout, err := m.resolveServerName(td.input)\n\t\t\tassert.Equal(t, td.expectedOutput, out)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestIsPartOfTheSameCluster(t *testing.T) {\n\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\n\ttests := map[string]struct {\n\t\tv1                             int64\n\t\tv2                             int64\n\t\tuseMinFailoverVersionAllowance func(string) bool\n\t\texpectedResult                 bool\n\t}{\n\t\t\"normal case v1 - is from the same cluster\": {\n\t\t\tv1:             0,\n\t\t\tv2:             0,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t\"normal case v2 - is from a different cluster\": {\n\t\t\tv1:             0,\n\t\t\tv2:             2,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t\"normal case v3 - is from the same cluster\": {\n\t\t\tv1:             0,\n\t\t\tv2:             100,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t\"normal case v4 - is from a different cluster\": {\n\t\t\tv1:             0,\n\t\t\tv2:             102,\n\t\t\texpectedResult: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(*testing.T) {\n\t\t\tm := Metadata{\n\t\t\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\t\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\t\t\tclusterName1: {\n\t\t\t\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t\t\t\t},\n\t\t\t\t\tclusterName2: {\n\t\t\t\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tversionToClusterName: map[int64]string{\n\t\t\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t\t\t},\n\t\t\t\tlog:     log.NewNoop(),\n\t\t\t\tmetrics: metrics.NewNoopMetricsClient().Scope(0),\n\t\t\t}\n\n\t\t\tassert.Equal(t, td.expectedResult, m.IsVersionFromSameCluster(td.v1, td.v2), name)\n\t\t})\n\n\t}\n}\n\nfunc TestIsPartOfTheSameClusterAPIFixing(t *testing.T) {\n\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\n\ttests := []struct {\n\t\tv0       int64\n\t\tv1       int64\n\t\texpected bool\n\t}{\n\t\t{v0: 0, v1: 0, expected: true},\n\t\t{v0: 0, v1: 1, expected: false},\n\t\t{v0: 0, v1: 2, expected: false},\n\t\t{v0: 0, v1: 3, expected: false},\n\t\t{v0: 0, v1: 4, expected: false},\n\t\t{v0: 0, v1: 5, expected: false},\n\t\t{v0: 0, v1: 6, expected: false},\n\t\t{v0: 0, v1: 7, expected: false},\n\t\t{v0: 0, v1: 8, expected: false},\n\t\t{v0: 0, v1: 9, expected: false},\n\t\t{v0: 0, v1: 10, expected: false},\n\t\t{v0: 0, v1: 11, expected: false},\n\t\t{v0: 0, v1: 12, expected: false},\n\t\t{v0: 0, v1: 13, expected: false},\n\t\t{v0: 0, v1: 14, expected: false},\n\t\t{v0: 0, v1: 15, expected: false},\n\t\t{v0: 0, v1: 16, expected: false},\n\t\t{v0: 0, v1: 17, expected: false},\n\t\t{v0: 0, v1: 18, expected: false},\n\t\t{v0: 0, v1: 19, expected: false},\n\t\t{v0: 0, v1: 20, expected: false},\n\t\t{v0: 0, v1: 21, expected: false},\n\t\t{v0: 0, v1: 22, expected: false},\n\t\t{v0: 0, v1: 23, expected: false},\n\t\t{v0: 0, v1: 24, expected: false},\n\t\t{v0: 0, v1: 25, expected: false},\n\t\t{v0: 0, v1: 26, expected: false},\n\t\t{v0: 0, v1: 27, expected: false},\n\t\t{v0: 0, v1: 28, expected: false},\n\t\t{v0: 0, v1: 29, expected: false},\n\t\t{v0: 0, v1: 30, expected: false},\n\t\t{v0: 0, v1: 31, expected: false},\n\t\t{v0: 0, v1: 32, expected: false},\n\t\t{v0: 0, v1: 33, expected: false},\n\t\t{v0: 0, v1: 34, expected: false},\n\t\t{v0: 0, v1: 35, expected: false},\n\t\t{v0: 0, v1: 36, expected: false},\n\t\t{v0: 0, v1: 37, expected: false},\n\t\t{v0: 0, v1: 38, expected: false},\n\t\t{v0: 0, v1: 39, expected: false},\n\t\t{v0: 0, v1: 40, expected: false},\n\t\t{v0: 0, v1: 41, expected: false},\n\t\t{v0: 0, v1: 42, expected: false},\n\t\t{v0: 0, v1: 43, expected: false},\n\t\t{v0: 0, v1: 44, expected: false},\n\t\t{v0: 0, v1: 45, expected: false},\n\t\t{v0: 0, v1: 46, expected: false},\n\t\t{v0: 0, v1: 47, expected: false},\n\t\t{v0: 0, v1: 48, expected: false},\n\t\t{v0: 0, v1: 49, expected: false},\n\t\t{v0: 0, v1: 50, expected: false},\n\t\t{v0: 0, v1: 51, expected: false},\n\t\t{v0: 0, v1: 52, expected: false},\n\t\t{v0: 0, v1: 53, expected: false},\n\t\t{v0: 0, v1: 54, expected: false},\n\t\t{v0: 0, v1: 55, expected: false},\n\t\t{v0: 0, v1: 56, expected: false},\n\t\t{v0: 0, v1: 57, expected: false},\n\t\t{v0: 0, v1: 58, expected: false},\n\t\t{v0: 0, v1: 59, expected: false},\n\t\t{v0: 0, v1: 60, expected: false},\n\t\t{v0: 0, v1: 61, expected: false},\n\t\t{v0: 0, v1: 62, expected: false},\n\t\t{v0: 0, v1: 63, expected: false},\n\t\t{v0: 0, v1: 64, expected: false},\n\t\t{v0: 0, v1: 65, expected: false},\n\t\t{v0: 0, v1: 66, expected: false},\n\t\t{v0: 0, v1: 67, expected: false},\n\t\t{v0: 0, v1: 68, expected: false},\n\t\t{v0: 0, v1: 69, expected: false},\n\t\t{v0: 0, v1: 70, expected: false},\n\t\t{v0: 0, v1: 71, expected: false},\n\t\t{v0: 0, v1: 72, expected: false},\n\t\t{v0: 0, v1: 73, expected: false},\n\t\t{v0: 0, v1: 74, expected: false},\n\t\t{v0: 0, v1: 75, expected: false},\n\t\t{v0: 0, v1: 76, expected: false},\n\t\t{v0: 0, v1: 77, expected: false},\n\t\t{v0: 0, v1: 78, expected: false},\n\t\t{v0: 0, v1: 79, expected: false},\n\t\t{v0: 0, v1: 80, expected: false},\n\t\t{v0: 0, v1: 81, expected: false},\n\t\t{v0: 0, v1: 82, expected: false},\n\t\t{v0: 0, v1: 83, expected: false},\n\t\t{v0: 0, v1: 84, expected: false},\n\t\t{v0: 0, v1: 85, expected: false},\n\t\t{v0: 0, v1: 86, expected: false},\n\t\t{v0: 0, v1: 87, expected: false},\n\t\t{v0: 0, v1: 88, expected: false},\n\t\t{v0: 0, v1: 89, expected: false},\n\t\t{v0: 0, v1: 90, expected: false},\n\t\t{v0: 0, v1: 91, expected: false},\n\t\t{v0: 0, v1: 92, expected: false},\n\t\t{v0: 0, v1: 93, expected: false},\n\t\t{v0: 0, v1: 94, expected: false},\n\t\t{v0: 0, v1: 95, expected: false},\n\t\t{v0: 0, v1: 96, expected: false},\n\t\t{v0: 0, v1: 97, expected: false},\n\t\t{v0: 0, v1: 98, expected: false},\n\t\t{v0: 0, v1: 99, expected: false},\n\t\t{v0: 0, v1: 100, expected: true},\n\t\t{v0: 0, v1: 101, expected: false},\n\t\t{v0: 0, v1: 102, expected: false},\n\t\t{v0: 0, v1: 103, expected: false},\n\t\t{v0: 0, v1: 104, expected: false},\n\t\t{v0: 0, v1: 105, expected: false},\n\t\t{v0: 0, v1: 106, expected: false},\n\t\t{v0: 0, v1: 107, expected: false},\n\t\t{v0: 0, v1: 108, expected: false},\n\t\t{v0: 0, v1: 109, expected: false},\n\t\t{v0: 0, v1: 110, expected: false},\n\t\t{v0: 0, v1: 111, expected: false},\n\t\t{v0: 0, v1: 112, expected: false},\n\t\t{v0: 0, v1: 113, expected: false},\n\t\t{v0: 0, v1: 114, expected: false},\n\t\t{v0: 0, v1: 115, expected: false},\n\t\t{v0: 0, v1: 116, expected: false},\n\t\t{v0: 0, v1: 117, expected: false},\n\t\t{v0: 0, v1: 118, expected: false},\n\t\t{v0: 0, v1: 119, expected: false},\n\t\t{v0: 0, v1: 120, expected: false},\n\t\t{v0: 0, v1: 121, expected: false},\n\t\t{v0: 0, v1: 122, expected: false},\n\t\t{v0: 0, v1: 123, expected: false},\n\t\t{v0: 0, v1: 124, expected: false},\n\t\t{v0: 0, v1: 125, expected: false},\n\t\t{v0: 0, v1: 126, expected: false},\n\t\t{v0: 0, v1: 127, expected: false},\n\t\t{v0: 0, v1: 128, expected: false},\n\t\t{v0: 0, v1: 129, expected: false},\n\t\t{v0: 0, v1: 130, expected: false},\n\t\t{v0: 0, v1: 131, expected: false},\n\t\t{v0: 0, v1: 132, expected: false},\n\t\t{v0: 0, v1: 133, expected: false},\n\t\t{v0: 0, v1: 134, expected: false},\n\t\t{v0: 0, v1: 135, expected: false},\n\t\t{v0: 0, v1: 136, expected: false},\n\t\t{v0: 0, v1: 137, expected: false},\n\t\t{v0: 0, v1: 138, expected: false},\n\t\t{v0: 0, v1: 139, expected: false},\n\t\t{v0: 0, v1: 140, expected: false},\n\t\t{v0: 0, v1: 141, expected: false},\n\t\t{v0: 0, v1: 142, expected: false},\n\t\t{v0: 0, v1: 143, expected: false},\n\t\t{v0: 0, v1: 144, expected: false},\n\t\t{v0: 0, v1: 145, expected: false},\n\t\t{v0: 0, v1: 146, expected: false},\n\t\t{v0: 0, v1: 147, expected: false},\n\t\t{v0: 0, v1: 148, expected: false},\n\t\t{v0: 0, v1: 149, expected: false},\n\t\t{v0: 0, v1: 150, expected: false},\n\t\t{v0: 0, v1: 151, expected: false},\n\t\t{v0: 0, v1: 152, expected: false},\n\t\t{v0: 0, v1: 153, expected: false},\n\t\t{v0: 0, v1: 154, expected: false},\n\t\t{v0: 0, v1: 155, expected: false},\n\t\t{v0: 0, v1: 156, expected: false},\n\t\t{v0: 0, v1: 157, expected: false},\n\t\t{v0: 0, v1: 158, expected: false},\n\t\t{v0: 0, v1: 159, expected: false},\n\t\t{v0: 0, v1: 160, expected: false},\n\t\t{v0: 0, v1: 161, expected: false},\n\t\t{v0: 0, v1: 162, expected: false},\n\t\t{v0: 0, v1: 163, expected: false},\n\t\t{v0: 0, v1: 164, expected: false},\n\t\t{v0: 0, v1: 165, expected: false},\n\t\t{v0: 0, v1: 166, expected: false},\n\t\t{v0: 0, v1: 167, expected: false},\n\t\t{v0: 0, v1: 168, expected: false},\n\t\t{v0: 0, v1: 169, expected: false},\n\t\t{v0: 0, v1: 170, expected: false},\n\t\t{v0: 0, v1: 171, expected: false},\n\t\t{v0: 0, v1: 172, expected: false},\n\t\t{v0: 0, v1: 173, expected: false},\n\t\t{v0: 0, v1: 174, expected: false},\n\t\t{v0: 0, v1: 175, expected: false},\n\t\t{v0: 0, v1: 176, expected: false},\n\t\t{v0: 0, v1: 177, expected: false},\n\t\t{v0: 0, v1: 178, expected: false},\n\t\t{v0: 0, v1: 179, expected: false},\n\t\t{v0: 0, v1: 180, expected: false},\n\t\t{v0: 0, v1: 181, expected: false},\n\t\t{v0: 0, v1: 182, expected: false},\n\t\t{v0: 0, v1: 183, expected: false},\n\t\t{v0: 0, v1: 184, expected: false},\n\t\t{v0: 0, v1: 185, expected: false},\n\t\t{v0: 0, v1: 186, expected: false},\n\t\t{v0: 0, v1: 187, expected: false},\n\t\t{v0: 0, v1: 188, expected: false},\n\t\t{v0: 0, v1: 189, expected: false},\n\t\t{v0: 0, v1: 190, expected: false},\n\t\t{v0: 0, v1: 191, expected: false},\n\t\t{v0: 0, v1: 192, expected: false},\n\t\t{v0: 0, v1: 193, expected: false},\n\t\t{v0: 0, v1: 194, expected: false},\n\t\t{v0: 0, v1: 195, expected: false},\n\t\t{v0: 0, v1: 196, expected: false},\n\t\t{v0: 0, v1: 197, expected: false},\n\t\t{v0: 0, v1: 198, expected: false},\n\t\t{v0: 0, v1: 199, expected: false},\n\t\t{v0: 0, v1: 200, expected: true},\n\t}\n\n\tm := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\tclusterName1: {\n\t\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t\t},\n\t\t\tclusterName2: {\n\t\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t\t},\n\t\t},\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return false },\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tlog:                           log.NewNoop(),\n\t}\n\n\tfor i := range tests {\n\t\tt.Run(fmt.Sprintf(\"%d\", i), func(t *testing.T) {\n\t\t\tassert.Equal(t, tests[i].expected, m.IsVersionFromSameCluster(tests[i].v0, tests[i].v1))\n\t\t})\n\t}\n}\n\nfunc TestClusterNameForFailoverVersion(t *testing.T) {\n\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\n\ttests := []struct {\n\t\tv0       int64\n\t\tv1       int64\n\t\texpected bool\n\t}{\n\t\t{v0: 0, v1: 0, expected: true},\n\t\t{v0: 0, v1: 1, expected: false},\n\t\t{v0: 0, v1: 2, expected: false},\n\t\t{v0: 0, v1: 3, expected: false},\n\t\t{v0: 0, v1: 4, expected: false},\n\t\t{v0: 0, v1: 5, expected: false},\n\t\t{v0: 0, v1: 6, expected: false},\n\t\t{v0: 0, v1: 7, expected: false},\n\t\t{v0: 0, v1: 8, expected: false},\n\t\t{v0: 0, v1: 9, expected: false},\n\t\t{v0: 0, v1: 10, expected: false},\n\t\t{v0: 0, v1: 11, expected: false},\n\t\t{v0: 0, v1: 12, expected: false},\n\t\t{v0: 0, v1: 13, expected: false},\n\t\t{v0: 0, v1: 14, expected: false},\n\t\t{v0: 0, v1: 15, expected: false},\n\t\t{v0: 0, v1: 16, expected: false},\n\t\t{v0: 0, v1: 17, expected: false},\n\t\t{v0: 0, v1: 18, expected: false},\n\t\t{v0: 0, v1: 19, expected: false},\n\t\t{v0: 0, v1: 20, expected: false},\n\t\t{v0: 0, v1: 21, expected: false},\n\t\t{v0: 0, v1: 22, expected: false},\n\t\t{v0: 0, v1: 23, expected: false},\n\t\t{v0: 0, v1: 24, expected: false},\n\t\t{v0: 0, v1: 25, expected: false},\n\t\t{v0: 0, v1: 26, expected: false},\n\t\t{v0: 0, v1: 27, expected: false},\n\t\t{v0: 0, v1: 28, expected: false},\n\t\t{v0: 0, v1: 29, expected: false},\n\t\t{v0: 0, v1: 30, expected: false},\n\t\t{v0: 0, v1: 31, expected: false},\n\t\t{v0: 0, v1: 32, expected: false},\n\t\t{v0: 0, v1: 33, expected: false},\n\t\t{v0: 0, v1: 34, expected: false},\n\t\t{v0: 0, v1: 35, expected: false},\n\t\t{v0: 0, v1: 36, expected: false},\n\t\t{v0: 0, v1: 37, expected: false},\n\t\t{v0: 0, v1: 38, expected: false},\n\t\t{v0: 0, v1: 39, expected: false},\n\t\t{v0: 0, v1: 40, expected: false},\n\t\t{v0: 0, v1: 41, expected: false},\n\t\t{v0: 0, v1: 42, expected: false},\n\t\t{v0: 0, v1: 43, expected: false},\n\t\t{v0: 0, v1: 44, expected: false},\n\t\t{v0: 0, v1: 45, expected: false},\n\t\t{v0: 0, v1: 46, expected: false},\n\t\t{v0: 0, v1: 47, expected: false},\n\t\t{v0: 0, v1: 48, expected: false},\n\t\t{v0: 0, v1: 49, expected: false},\n\t\t{v0: 0, v1: 50, expected: false},\n\t\t{v0: 0, v1: 51, expected: false},\n\t\t{v0: 0, v1: 52, expected: false},\n\t\t{v0: 0, v1: 53, expected: false},\n\t\t{v0: 0, v1: 54, expected: false},\n\t\t{v0: 0, v1: 55, expected: false},\n\t\t{v0: 0, v1: 56, expected: false},\n\t\t{v0: 0, v1: 57, expected: false},\n\t\t{v0: 0, v1: 58, expected: false},\n\t\t{v0: 0, v1: 59, expected: false},\n\t\t{v0: 0, v1: 60, expected: false},\n\t\t{v0: 0, v1: 61, expected: false},\n\t\t{v0: 0, v1: 62, expected: false},\n\t\t{v0: 0, v1: 63, expected: false},\n\t\t{v0: 0, v1: 64, expected: false},\n\t\t{v0: 0, v1: 65, expected: false},\n\t\t{v0: 0, v1: 66, expected: false},\n\t\t{v0: 0, v1: 67, expected: false},\n\t\t{v0: 0, v1: 68, expected: false},\n\t\t{v0: 0, v1: 69, expected: false},\n\t\t{v0: 0, v1: 70, expected: false},\n\t\t{v0: 0, v1: 71, expected: false},\n\t\t{v0: 0, v1: 72, expected: false},\n\t\t{v0: 0, v1: 73, expected: false},\n\t\t{v0: 0, v1: 74, expected: false},\n\t\t{v0: 0, v1: 75, expected: false},\n\t\t{v0: 0, v1: 76, expected: false},\n\t\t{v0: 0, v1: 77, expected: false},\n\t\t{v0: 0, v1: 78, expected: false},\n\t\t{v0: 0, v1: 79, expected: false},\n\t\t{v0: 0, v1: 80, expected: false},\n\t\t{v0: 0, v1: 81, expected: false},\n\t\t{v0: 0, v1: 82, expected: false},\n\t\t{v0: 0, v1: 83, expected: false},\n\t\t{v0: 0, v1: 84, expected: false},\n\t\t{v0: 0, v1: 85, expected: false},\n\t\t{v0: 0, v1: 86, expected: false},\n\t\t{v0: 0, v1: 87, expected: false},\n\t\t{v0: 0, v1: 88, expected: false},\n\t\t{v0: 0, v1: 89, expected: false},\n\t\t{v0: 0, v1: 90, expected: false},\n\t\t{v0: 0, v1: 91, expected: false},\n\t\t{v0: 0, v1: 92, expected: false},\n\t\t{v0: 0, v1: 93, expected: false},\n\t\t{v0: 0, v1: 94, expected: false},\n\t\t{v0: 0, v1: 95, expected: false},\n\t\t{v0: 0, v1: 96, expected: false},\n\t\t{v0: 0, v1: 97, expected: false},\n\t\t{v0: 0, v1: 98, expected: false},\n\t\t{v0: 0, v1: 99, expected: false},\n\t\t{v0: 0, v1: 100, expected: true},\n\t\t{v0: 0, v1: 101, expected: false},\n\t\t{v0: 0, v1: 102, expected: false},\n\t\t{v0: 0, v1: 103, expected: false},\n\t\t{v0: 0, v1: 104, expected: false},\n\t\t{v0: 0, v1: 105, expected: false},\n\t\t{v0: 0, v1: 106, expected: false},\n\t\t{v0: 0, v1: 107, expected: false},\n\t\t{v0: 0, v1: 108, expected: false},\n\t\t{v0: 0, v1: 109, expected: false},\n\t\t{v0: 0, v1: 110, expected: false},\n\t\t{v0: 0, v1: 111, expected: false},\n\t\t{v0: 0, v1: 112, expected: false},\n\t\t{v0: 0, v1: 113, expected: false},\n\t\t{v0: 0, v1: 114, expected: false},\n\t\t{v0: 0, v1: 115, expected: false},\n\t\t{v0: 0, v1: 116, expected: false},\n\t\t{v0: 0, v1: 117, expected: false},\n\t\t{v0: 0, v1: 118, expected: false},\n\t\t{v0: 0, v1: 119, expected: false},\n\t\t{v0: 0, v1: 120, expected: false},\n\t\t{v0: 0, v1: 121, expected: false},\n\t\t{v0: 0, v1: 122, expected: false},\n\t\t{v0: 0, v1: 123, expected: false},\n\t\t{v0: 0, v1: 124, expected: false},\n\t\t{v0: 0, v1: 125, expected: false},\n\t\t{v0: 0, v1: 126, expected: false},\n\t\t{v0: 0, v1: 127, expected: false},\n\t\t{v0: 0, v1: 128, expected: false},\n\t\t{v0: 0, v1: 129, expected: false},\n\t\t{v0: 0, v1: 130, expected: false},\n\t\t{v0: 0, v1: 131, expected: false},\n\t\t{v0: 0, v1: 132, expected: false},\n\t\t{v0: 0, v1: 133, expected: false},\n\t\t{v0: 0, v1: 134, expected: false},\n\t\t{v0: 0, v1: 135, expected: false},\n\t\t{v0: 0, v1: 136, expected: false},\n\t\t{v0: 0, v1: 137, expected: false},\n\t\t{v0: 0, v1: 138, expected: false},\n\t\t{v0: 0, v1: 139, expected: false},\n\t\t{v0: 0, v1: 140, expected: false},\n\t\t{v0: 0, v1: 141, expected: false},\n\t\t{v0: 0, v1: 142, expected: false},\n\t\t{v0: 0, v1: 143, expected: false},\n\t\t{v0: 0, v1: 144, expected: false},\n\t\t{v0: 0, v1: 145, expected: false},\n\t\t{v0: 0, v1: 146, expected: false},\n\t\t{v0: 0, v1: 147, expected: false},\n\t\t{v0: 0, v1: 148, expected: false},\n\t\t{v0: 0, v1: 149, expected: false},\n\t\t{v0: 0, v1: 150, expected: false},\n\t\t{v0: 0, v1: 151, expected: false},\n\t\t{v0: 0, v1: 152, expected: false},\n\t\t{v0: 0, v1: 153, expected: false},\n\t\t{v0: 0, v1: 154, expected: false},\n\t\t{v0: 0, v1: 155, expected: false},\n\t\t{v0: 0, v1: 156, expected: false},\n\t\t{v0: 0, v1: 157, expected: false},\n\t\t{v0: 0, v1: 158, expected: false},\n\t\t{v0: 0, v1: 159, expected: false},\n\t\t{v0: 0, v1: 160, expected: false},\n\t\t{v0: 0, v1: 161, expected: false},\n\t\t{v0: 0, v1: 162, expected: false},\n\t\t{v0: 0, v1: 163, expected: false},\n\t\t{v0: 0, v1: 164, expected: false},\n\t\t{v0: 0, v1: 165, expected: false},\n\t\t{v0: 0, v1: 166, expected: false},\n\t\t{v0: 0, v1: 167, expected: false},\n\t\t{v0: 0, v1: 168, expected: false},\n\t\t{v0: 0, v1: 169, expected: false},\n\t\t{v0: 0, v1: 170, expected: false},\n\t\t{v0: 0, v1: 171, expected: false},\n\t\t{v0: 0, v1: 172, expected: false},\n\t\t{v0: 0, v1: 173, expected: false},\n\t\t{v0: 0, v1: 174, expected: false},\n\t\t{v0: 0, v1: 175, expected: false},\n\t\t{v0: 0, v1: 176, expected: false},\n\t\t{v0: 0, v1: 177, expected: false},\n\t\t{v0: 0, v1: 178, expected: false},\n\t\t{v0: 0, v1: 179, expected: false},\n\t\t{v0: 0, v1: 180, expected: false},\n\t\t{v0: 0, v1: 181, expected: false},\n\t\t{v0: 0, v1: 182, expected: false},\n\t\t{v0: 0, v1: 183, expected: false},\n\t\t{v0: 0, v1: 184, expected: false},\n\t\t{v0: 0, v1: 185, expected: false},\n\t\t{v0: 0, v1: 186, expected: false},\n\t\t{v0: 0, v1: 187, expected: false},\n\t\t{v0: 0, v1: 188, expected: false},\n\t\t{v0: 0, v1: 189, expected: false},\n\t\t{v0: 0, v1: 190, expected: false},\n\t\t{v0: 0, v1: 191, expected: false},\n\t\t{v0: 0, v1: 192, expected: false},\n\t\t{v0: 0, v1: 193, expected: false},\n\t\t{v0: 0, v1: 194, expected: false},\n\t\t{v0: 0, v1: 195, expected: false},\n\t\t{v0: 0, v1: 196, expected: false},\n\t\t{v0: 0, v1: 197, expected: false},\n\t\t{v0: 0, v1: 198, expected: false},\n\t\t{v0: 0, v1: 199, expected: false},\n\t\t{v0: 0, v1: 200, expected: true},\n\t}\n\n\tm := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters: map[string]config.ClusterInformation{\n\t\t\tclusterName1: {\n\t\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t\t},\n\t\t\tclusterName2: {\n\t\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t\t},\n\t\t},\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return false },\n\t\tlog:                           log.NewNoop(),\n\t}\n\n\tfor i := range tests {\n\t\tt.Run(fmt.Sprintf(\"%d\", i), func(t *testing.T) {\n\t\t\tassert.Equal(t, tests[i].expected, m.IsVersionFromSameCluster(tests[i].v0, tests[i].v1))\n\t\t})\n\t}\n}\n\nfunc TestServerResolution(t *testing.T) {\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\tconst domainToMigrate = \"some domain\"\n\n\tallClusters := map[string]config.ClusterInformation{\n\t\tclusterName1: {\n\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t},\n\t\tclusterName2: {\n\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t},\n\t}\n\n\timpl := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters:              allClusters,\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return domain == domainToMigrate },\n\t\tlog:                           log.NewNoop(),\n\t}\n\n\terr := quick.Check(func(currentFOVersion int64, migrateDomain bool) bool {\n\t\tvar nextFailoverVersion int64\n\t\tif migrateDomain {\n\t\t\tfo := impl.GetNextFailoverVersion(clusterName1, currentFOVersion, domainToMigrate)\n\t\t\tnextFailoverVersion = fo\n\t\t} else {\n\t\t\tfo := impl.GetNextFailoverVersion(clusterName1, currentFOVersion, \"a different domain\")\n\t\t\tnextFailoverVersion = fo\n\t\t}\n\t\t// do a round-trip\n\t\tclusterNameResolved, err := impl.ClusterNameForFailoverVersion(nextFailoverVersion)\n\t\trequire.NoError(t, err)\n\t\treturn clusterName1 == clusterNameResolved\n\t}, &quick.Config{})\n\tassert.NoError(t, err)\n}\n\n// the point of this test is to assert that there's no clumsy errors, and\n// in an unmigrated state, the old implementation for getting versions and the new\n// one are equal\nfunc TestNoChangesInUnmigratedState(t *testing.T) {\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\n\tallClusters := map[string]config.ClusterInformation{\n\t\tclusterName1: {\n\t\t\tInitialFailoverVersion: initialFailoverVersionC1,\n\t\t},\n\t\tclusterName2: {\n\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t},\n\t}\n\n\toldImpl := func(cluster string, currentFailoverVersion int64) int64 {\n\t\tinfo, ok := allClusters[cluster]\n\t\tif !ok {\n\t\t\tpanic(fmt.Sprintf(\n\t\t\t\t\"Unknown cluster name: %v with given cluster initial failover version map: %v.\",\n\t\t\t\tcluster,\n\t\t\t\tallClusters,\n\t\t\t))\n\t\t}\n\t\tfailoverVersion := currentFailoverVersion/failoverVersionIncrement*failoverVersionIncrement + info.InitialFailoverVersion\n\t\tif failoverVersion < currentFailoverVersion {\n\t\t\treturn failoverVersion + failoverVersionIncrement\n\t\t}\n\t\treturn failoverVersion\n\t}\n\n\tnewImpl := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters:              allClusters,\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return false },\n\t\tlog:                           log.NewNoop(),\n\t}\n\n\terr := quick.CheckEqual(func(currVersion int64) int64 {\n\t\t// partially apply the cluster-name, since the fuzzer trying to guess that is pointless\n\t\treturn oldImpl(clusterName2, currVersion)\n\t}, func(currentVersion int64) int64 {\n\t\treturn newImpl.GetNextFailoverVersion(clusterName2, currentVersion, \"some domain\") // all domains are turned off\n\t},\n\t\t&quick.Config{})\n\tassert.NoError(t, err)\n}\n\n// the point of this test is to assert that there's no clumsy errors, and\n// in an unmigrated state, the old implementation for getting versions and the new\n// one are equal\nfunc TestFailoverVersionResolution(t *testing.T) {\n\tconst clusterName1 = \"c1\"\n\tconst initialFailoverVersionC1 = 0\n\tvar newFailoverVersionC1 = int64(1)\n\tconst clusterName2 = \"c2\"\n\tconst initialFailoverVersionC2 = 2\n\tconst failoverVersionIncrement = 100\n\n\tallClusters := map[string]config.ClusterInformation{\n\t\tclusterName1: {\n\t\t\tInitialFailoverVersion:    initialFailoverVersionC1,\n\t\t\tNewInitialFailoverVersion: &newFailoverVersionC1,\n\t\t},\n\t\tclusterName2: {\n\t\t\tInitialFailoverVersion: initialFailoverVersionC2,\n\t\t},\n\t}\n\n\tsut := Metadata{\n\t\tfailoverVersionIncrement: failoverVersionIncrement,\n\t\tallClusters:              allClusters,\n\t\tversionToClusterName: map[int64]string{\n\t\t\tinitialFailoverVersionC1: clusterName1,\n\t\t\tinitialFailoverVersionC2: clusterName2,\n\t\t},\n\t\tmetrics:                       metrics.NewNoopMetricsClient().Scope(0),\n\t\tuseNewFailoverVersionOverride: func(domain string) bool { return false },\n\t\tlog:                           log.NewNoop(),\n\t}\n\n\ttests := map[string]struct {\n\t\tin          int64\n\t\texpectedOut string\n\t\texpectedErr error\n\t}{\n\t\t\"normal failover version\": {\n\t\t\tin:          initialFailoverVersionC1,\n\t\t\texpectedOut: clusterName1,\n\t\t},\n\t\t\"New failover version\": {\n\t\t\tin:          initialFailoverVersionC1,\n\t\t\texpectedOut: clusterName1,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tout, err := sut.ClusterNameForFailoverVersion(td.in)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, td.expectedOut, out)\n\t\t})\n\t}\n}\n\nfunc TestGetters(t *testing.T) {\n\tm := NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\"cluster0\": {\n\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t},\n\t\t\t\t\"cluster1\": {\n\t\t\t\t\tInitialFailoverVersion: 3,\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t},\n\t\t\t\t\"cluster2\": {\n\t\t\t\t\tInitialFailoverVersion: 5,\n\t\t\t\t\tEnabled:                false,\n\t\t\t\t},\n\t\t\t},\n\t\t\tFailoverVersionIncrement: 100,\n\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n\n\tassert.True(t, m.IsPrimaryCluster())\n\tassert.Equal(t, \"cluster0\", m.GetCurrentClusterName())\n\n\t// do existence checks\n\tassert.Equal(t, []string{\"cluster0\", \"cluster1\", \"cluster2\"}, keysOfClusterInfoMap(m.GetAllClusterInfo()))\n\tassert.Equal(t, []string{\"cluster0\", \"cluster1\"}, keysOfClusterInfoMap(m.GetEnabledClusterInfo()))\n\tassert.Equal(t, []string{\"cluster1\"}, keysOfClusterInfoMap(m.GetRemoteClusterInfo()))\n}\n\nfunc keysOfClusterInfoMap(m map[string]config.ClusterInformation) []string {\n\tkeys := make([]string, 0, len(m))\n\tfor k := range m {\n\t\tkeys = append(keys, k)\n\t}\n\tsort.Strings(keys)\n\treturn keys\n}\n"
  },
  {
    "path": "common/cluster/metadata_test_utils.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cluster\n\nimport (\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\tcommonMetrics \"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\t// TestCurrentClusterInitialFailoverVersion is initial failover version for current cluster\n\tTestCurrentClusterInitialFailoverVersion = int64(0)\n\t// TestAlternativeClusterInitialFailoverVersion is initial failover version for alternative cluster\n\tTestAlternativeClusterInitialFailoverVersion = int64(1)\n\t// TestDisabledClusterInitialFailoverVersion is initial failover version for disabled cluster\n\tTestDisabledClusterInitialFailoverVersion = int64(2)\n\t// TestFailoverVersionIncrement is failover version increment used for test\n\tTestFailoverVersionIncrement = int64(10)\n\t// TestCurrentClusterName is current cluster used for test\n\tTestCurrentClusterName = \"active\"\n\t// TestAlternativeClusterName is alternative cluster used for test\n\tTestAlternativeClusterName = \"standby\"\n\t// TestDisabledClusterName is disabled cluster used for test\n\tTestDisabledClusterName = \"disabled\"\n\t// TestRegion1 is region1 used for test\n\tTestRegion1 = \"region1\"\n\t// TestRegion2 is region2 used for test\n\tTestRegion2 = \"region2\"\n\t// TestCurrentClusterFrontendAddress is the ip port address of current cluster\n\tTestCurrentClusterFrontendAddress = \"127.0.0.1:7104\"\n\t// TestAlternativeClusterFrontendAddress is the ip port address of alternative cluster\n\tTestAlternativeClusterFrontendAddress = \"127.0.0.1:8104\"\n\t// TestClusterXDCTransport is the RPC transport used for XDC traffic <tchannel|grpc>\n\tTestClusterXDCTransport = \"grpc\"\n)\n\nvar (\n\t// TestAllClusterNames is the all cluster names used for test\n\tTestAllClusterNames = []string{TestCurrentClusterName, TestAlternativeClusterName}\n\t// TestAllClusterInfo is the same as above, just convenient for test mocking\n\tTestAllClusterInfo = map[string]config.ClusterInformation{\n\t\tTestCurrentClusterName: {\n\t\t\tEnabled:                true,\n\t\t\tInitialFailoverVersion: TestCurrentClusterInitialFailoverVersion,\n\t\t\tRPCName:                service.Frontend,\n\t\t\tRPCAddress:             TestCurrentClusterFrontendAddress,\n\t\t\tRPCTransport:           TestClusterXDCTransport,\n\t\t},\n\t\tTestAlternativeClusterName: {\n\t\t\tEnabled:                true,\n\t\t\tInitialFailoverVersion: TestAlternativeClusterInitialFailoverVersion,\n\t\t\tRPCName:                service.Frontend,\n\t\t\tRPCAddress:             TestAlternativeClusterFrontendAddress,\n\t\t\tRPCTransport:           TestClusterXDCTransport,\n\t\t},\n\t\tTestDisabledClusterName: {\n\t\t\tEnabled:                false,\n\t\t\tInitialFailoverVersion: TestDisabledClusterInitialFailoverVersion,\n\t\t},\n\t}\n\n\t// TestSingleDCAllClusterNames is the all cluster names used for test\n\tTestSingleDCAllClusterNames = []string{TestCurrentClusterName}\n\t// TestSingleDCClusterInfo is the same as above, just convenient for test mocking\n\tTestSingleDCClusterInfo = map[string]config.ClusterInformation{\n\t\tTestCurrentClusterName: {\n\t\t\tEnabled:                true,\n\t\t\tInitialFailoverVersion: TestCurrentClusterInitialFailoverVersion,\n\t\t\tRPCName:                service.Frontend,\n\t\t\tRPCAddress:             TestCurrentClusterFrontendAddress,\n\t\t\tRPCTransport:           TestClusterXDCTransport,\n\t\t},\n\t}\n\n\t// TestActiveClusterMetadata is metadata for an active cluster\n\tTestActiveClusterMetadata = NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: TestFailoverVersionIncrement,\n\t\t\tPrimaryClusterName:       TestCurrentClusterName,\n\t\t\tCurrentClusterName:       TestCurrentClusterName,\n\t\t\tClusterGroup:             TestAllClusterInfo,\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tcommonMetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n\n\t// TestPassiveClusterMetadata is metadata for a passive cluster\n\tTestPassiveClusterMetadata = NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: TestFailoverVersionIncrement,\n\t\t\tPrimaryClusterName:       TestCurrentClusterName,\n\t\t\tCurrentClusterName:       TestAlternativeClusterName,\n\t\t\tClusterGroup:             TestAllClusterInfo,\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tcommonMetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n)\n\n// GetTestClusterMetadata return an cluster metadata instance, which is initialized\nfunc GetTestClusterMetadata(isPrimaryCluster bool) Metadata {\n\tprimaryClusterName := TestCurrentClusterName\n\tif !isPrimaryCluster {\n\t\tprimaryClusterName = TestAlternativeClusterName\n\t}\n\n\treturn NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: TestFailoverVersionIncrement,\n\t\t\tPrimaryClusterName:       primaryClusterName,\n\t\t\tCurrentClusterName:       TestCurrentClusterName,\n\t\t\tClusterGroup:             TestAllClusterInfo,\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tcommonMetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n}\n"
  },
  {
    "path": "common/cluster/utils.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cluster\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// GetOrUseDefaultActiveCluster return the current cluster name or use the input if valid\nfunc GetOrUseDefaultActiveCluster(currentClusterName string, activeClusterName string) string {\n\tif len(activeClusterName) == 0 {\n\t\treturn currentClusterName\n\t}\n\treturn activeClusterName\n}\n\n// GetOrUseDefaultClusters return the current cluster or use the input if valid\nfunc GetOrUseDefaultClusters(currentClusterName string, clusters []*persistence.ClusterReplicationConfig) []*persistence.ClusterReplicationConfig {\n\tif len(clusters) == 0 {\n\t\treturn []*persistence.ClusterReplicationConfig{\n\t\t\t&persistence.ClusterReplicationConfig{\n\t\t\t\tClusterName: currentClusterName,\n\t\t\t},\n\t\t}\n\t}\n\treturn clusters\n}\n"
  },
  {
    "path": "common/cluster/utils_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cluster\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestGetOrUseDefaultActiveCluster(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tcurrentCluster  string\n\t\tactiveCluster   string\n\t\texpectedCluster string\n\t}{\n\t\t{\n\t\t\tname:            \"empty active cluster\",\n\t\t\tcurrentCluster:  \"cluster1\",\n\t\t\tactiveCluster:   \"\",\n\t\t\texpectedCluster: \"cluster1\",\n\t\t},\n\t\t{\n\t\t\tname:            \"non-empty active cluster\",\n\t\t\tcurrentCluster:  \"cluster1\",\n\t\t\tactiveCluster:   \"cluster2\",\n\t\t\texpectedCluster: \"cluster2\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := GetOrUseDefaultActiveCluster(tt.currentCluster, tt.activeCluster)\n\t\t\tassert.Equal(t, tt.expectedCluster, got)\n\t\t})\n\t}\n}\n\nfunc TestGetOrUseDefaultClusters(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\tcurrentCluster   string\n\t\tclusters         []*persistence.ClusterReplicationConfig\n\t\texpectedClusters []*persistence.ClusterReplicationConfig\n\t}{\n\t\t{\n\t\t\tname:           \"empty clusters\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tclusters:       []*persistence.ClusterReplicationConfig{},\n\t\t\texpectedClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t&persistence.ClusterReplicationConfig{\n\t\t\t\t\tClusterName: \"cluster1\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"non-empty clusters\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tclusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t&persistence.ClusterReplicationConfig{\n\t\t\t\t\tClusterName: \"cluster2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t&persistence.ClusterReplicationConfig{\n\t\t\t\t\tClusterName: \"cluster2\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := GetOrUseDefaultClusters(tt.currentCluster, tt.clusters)\n\t\t\tassert.Equal(t, tt.expectedClusters, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/codec/gob/gob.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gob\n\nimport (\n\t\"bytes\"\n\t\"encoding/gob\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n)\n\nvar errEmptyArgument = errors.New(\"length of input argument is 0\")\n\n// Encoder is wrapper of gob encoder/decoder\ntype Encoder struct{}\n\n// NewGobEncoder create new Encoder\nfunc NewGobEncoder() *Encoder {\n\treturn &Encoder{}\n}\n\n// Encode one or more objects to binary\nfunc (gobEncoder *Encoder) Encode(value ...interface{}) ([]byte, error) {\n\tif len(value) == 0 {\n\t\treturn nil, errEmptyArgument\n\t}\n\n\tvar buf bytes.Buffer\n\tenc := gob.NewEncoder(&buf)\n\tfor i, obj := range value {\n\t\tif err := enc.Encode(obj); err != nil {\n\t\t\treturn nil, fmt.Errorf(\n\t\t\t\t\"unable to encode argument: %d, %v, with gob error: %v\", i, reflect.TypeOf(obj), err)\n\t\t}\n\t}\n\treturn buf.Bytes(), nil\n}\n\n// Decode binary to one or more objects\nfunc (gobEncoder *Encoder) Decode(input []byte, valuePtr ...interface{}) error {\n\tif len(valuePtr) == 0 {\n\t\treturn errEmptyArgument\n\t}\n\n\tdec := gob.NewDecoder(bytes.NewBuffer(input))\n\tfor i, obj := range valuePtr {\n\t\tif err := dec.Decode(obj); err != nil {\n\t\t\treturn fmt.Errorf(\n\t\t\t\t\"unable to decode argument: %d, %v, with gob error: %v\", i, reflect.TypeOf(obj), err)\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/codec/gob/gob_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gob\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype testStruct struct {\n\tDomain     string\n\tWorkflowID string\n\tRunID      string\n\tStartTime  int64\n}\n\nfunc TestGobEncoder(t *testing.T) {\n\tencoder := NewGobEncoder()\n\n\tdomain := \"test-domain\"\n\twid := uuid.New()\n\trid := uuid.New()\n\tstartTime := time.Now().UnixNano()\n\n\t// test encode and decode 1 object\n\tmsg := &testStruct{\n\t\tDomain:     domain,\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t\tStartTime:  startTime,\n\t}\n\tpayload, err := encoder.Encode(msg)\n\trequire.NoError(t, err)\n\tvar decoded *testStruct\n\terr = encoder.Decode(payload, &decoded)\n\trequire.NoError(t, err)\n\trequire.Equal(t, msg, decoded)\n\n\t// test encode and decode 2 objects\n\tmsg2 := \"test-string\"\n\tpayload, err = encoder.Encode(msg2, msg)\n\trequire.NoError(t, err)\n\tvar decoded2 string\n\terr = encoder.Decode(payload, &decoded2, &decoded)\n\trequire.NoError(t, err)\n\trequire.Equal(t, msg, decoded)\n\trequire.Equal(t, msg2, decoded2)\n\n\t// test encode and decode 0 object\n\t_, err = encoder.Encode()\n\trequire.Error(t, err)\n\terr = encoder.Decode(payload)\n\trequire.Error(t, err)\n}\n"
  },
  {
    "path": "common/codec/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination interfaces_mock.go -self_package github.com/uber/cadence/common/codec github.com/uber/cadence/common/codec BinaryEncoder\n\npackage codec\n\nimport (\n\t\"go.uber.org/thriftrw/protocol/stream\"\n\t\"go.uber.org/thriftrw/wire\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// BinaryEncoder represent the encoder which can serialize or deserialize object\n\tBinaryEncoder interface {\n\t\tEncode(obj ThriftObject) ([]byte, error)\n\t\tDecode(payload []byte, val ThriftObject) error\n\t}\n\n\t// ThriftObject represents a thrift object\n\tThriftObject interface {\n\t\tFromWire(w wire.Value) error\n\t\tToWire() (wire.Value, error)\n\t\tEncode(stream.Writer) error\n\t\tDecode(stream.Reader) error\n\t}\n)\n\nconst (\n\t// used by thriftrw binary codec\n\tpreambleVersion0 byte = 0x59\n)\n\nvar (\n\t// MissingBinaryEncodingVersion indicate that the encoding version is missing\n\tMissingBinaryEncodingVersion = &types.BadRequestError{Message: \"Missing binary encoding version.\"}\n\t// InvalidBinaryEncodingVersion indicate that the encoding version is incorrect\n\tInvalidBinaryEncodingVersion = &types.BadRequestError{Message: \"Invalid binary encoding version.\"}\n\t// MsgPayloadNotThriftEncoded indicate message is not thrift encoded\n\tMsgPayloadNotThriftEncoded = &types.BadRequestError{Message: \"Message payload is not thrift encoded.\"}\n)\n"
  },
  {
    "path": "common/codec/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/codec (interfaces: BinaryEncoder)\n//\n// Generated by this command:\n//\n//\tmockgen -package codec -destination interfaces_mock.go -self_package github.com/uber/cadence/common/codec github.com/uber/cadence/common/codec BinaryEncoder\n//\n\n// Package codec is a generated GoMock package.\npackage codec\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockBinaryEncoder is a mock of BinaryEncoder interface.\ntype MockBinaryEncoder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockBinaryEncoderMockRecorder\n\tisgomock struct{}\n}\n\n// MockBinaryEncoderMockRecorder is the mock recorder for MockBinaryEncoder.\ntype MockBinaryEncoderMockRecorder struct {\n\tmock *MockBinaryEncoder\n}\n\n// NewMockBinaryEncoder creates a new mock instance.\nfunc NewMockBinaryEncoder(ctrl *gomock.Controller) *MockBinaryEncoder {\n\tmock := &MockBinaryEncoder{ctrl: ctrl}\n\tmock.recorder = &MockBinaryEncoderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockBinaryEncoder) EXPECT() *MockBinaryEncoderMockRecorder {\n\treturn m.recorder\n}\n\n// Decode mocks base method.\nfunc (m *MockBinaryEncoder) Decode(payload []byte, val ThriftObject) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Decode\", payload, val)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Decode indicates an expected call of Decode.\nfunc (mr *MockBinaryEncoderMockRecorder) Decode(payload, val any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Decode\", reflect.TypeOf((*MockBinaryEncoder)(nil).Decode), payload, val)\n}\n\n// Encode mocks base method.\nfunc (m *MockBinaryEncoder) Encode(obj ThriftObject) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Encode\", obj)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Encode indicates an expected call of Encode.\nfunc (mr *MockBinaryEncoderMockRecorder) Encode(obj any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Encode\", reflect.TypeOf((*MockBinaryEncoder)(nil).Encode), obj)\n}\n"
  },
  {
    "path": "common/codec/version0_thriftrw.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage codec\n\nimport (\n\t\"bytes\"\n\n\t\"go.uber.org/thriftrw/protocol/binary\"\n)\n\ntype (\n\t// ThriftRWEncoder is an implementation using thrift rw for binary encoding / decoding\n\t// NOTE: this encoder only works for thrift struct\n\tThriftRWEncoder struct {\n\t}\n)\n\nvar _ BinaryEncoder = (*ThriftRWEncoder)(nil)\n\n// NewThriftRWEncoder generate a new ThriftRWEncoder\nfunc NewThriftRWEncoder() *ThriftRWEncoder {\n\treturn &ThriftRWEncoder{}\n}\n\n// Encode encode the object\nfunc (t *ThriftRWEncoder) Encode(obj ThriftObject) ([]byte, error) {\n\tif obj == nil {\n\t\treturn nil, MsgPayloadNotThriftEncoded\n\t}\n\tvar writer bytes.Buffer\n\t// use the first byte to version the serialization\n\terr := writer.WriteByte(preambleVersion0)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tsw := binary.Default.Writer(&writer)\n\tdefer sw.Close()\n\tif err := obj.Encode(sw); err != nil {\n\t\treturn nil, err\n\t}\n\treturn writer.Bytes(), nil\n}\n\n// Decode decode the object\nfunc (t *ThriftRWEncoder) Decode(b []byte, val ThriftObject) error {\n\tif len(b) < 1 {\n\t\treturn MissingBinaryEncodingVersion\n\t}\n\n\tversion := b[0]\n\tif version != preambleVersion0 {\n\t\treturn InvalidBinaryEncodingVersion\n\t}\n\n\treader := bytes.NewReader(b[1:])\n\tsr := binary.Default.Reader(reader)\n\treturn val.Decode(sr)\n}\n"
  },
  {
    "path": "common/codec/version0_thriftrw_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage codec\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/multierr\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n)\n\ntype (\n\tthriftRWEncoderSuite struct {\n\t\tsuite.Suite\n\t\tencoder *ThriftRWEncoder\n\t}\n)\n\nvar (\n\tthriftObject = &workflow.HistoryEvent{\n\t\tVersion:   int64Ptr(1234),\n\t\tEventId:   int64Ptr(130),\n\t\tTimestamp: int64Ptr(112345132134),\n\t\tEventType: workflow.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr(),\n\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: &workflow.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain: stringPtr(\"some random target domain\"),\n\t\t\tWorkflowExecution: &workflow.WorkflowExecution{\n\t\t\t\tWorkflowId: stringPtr(\"some random target workflow ID\"),\n\t\t\t\tRunId:      stringPtr(\"some random target run ID\"),\n\t\t\t},\n\t\t\tChildWorkflowOnly: boolPtr(true),\n\t\t\tControl:           []byte(\"some random control\"),\n\t\t},\n\t}\n\tthriftEncodedBinary = []byte{\n\t\t89, 10, 0, 10, 0, 0, 0, 0, 0, 0, 0, 130, 10, 0, 20, 0, 0, 0, 26, 40, 74, 172, 102,\n\t\t8, 0, 30, 0, 0, 0, 23, 10, 0, 35, 0, 0, 0, 0, 0, 0, 4, 210, 12, 1, 44, 11, 0, 20,\n\t\t0, 0, 0, 25, 115, 111, 109, 101, 32, 114, 97, 110, 100, 111, 109, 32, 116, 97, 114,\n\t\t103, 101, 116, 32, 100, 111, 109, 97, 105, 110, 12, 0, 30, 11, 0, 10, 0, 0, 0, 30,\n\t\t115, 111, 109, 101, 32, 114, 97, 110, 100, 111, 109, 32, 116, 97, 114, 103, 101,\n\t\t116, 32, 119, 111, 114, 107, 102, 108, 111, 119, 32, 73, 68, 11, 0, 20, 0, 0, 0, 25,\n\t\t115, 111, 109, 101, 32, 114, 97, 110, 100, 111, 109, 32, 116, 97, 114, 103, 101, 116,\n\t\t32, 114, 117, 110, 32, 73, 68, 0, 11, 0, 40, 0, 0, 0, 19, 115, 111, 109, 101, 32, 114,\n\t\t97, 110, 100, 111, 109, 32, 99, 111, 110, 116, 114, 111, 108, 2, 0, 50, 1, 0, 0,\n\t}\n)\n\nfunc TestThriftRWEncoderSuite(t *testing.T) {\n\ts := new(thriftRWEncoderSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *thriftRWEncoderSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\ts.encoder = NewThriftRWEncoder()\n}\n\nfunc (s *thriftRWEncoderSuite) TestEncode() {\n\tbinary, err := s.encoder.Encode(thriftObject)\n\ts.Nil(err)\n\ts.Equal(thriftEncodedBinary, binary)\n}\n\nfunc (s *thriftRWEncoderSuite) TestEncodeConcurrent() {\n\tvar wg sync.WaitGroup\n\tcount := 200\n\terrs := make([]error, count)\n\twg.Add(count)\n\tfor i := 0; i < count; i++ {\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\t\t\tbinary, err := s.encoder.Encode(thriftObject)\n\t\t\tif err != nil {\n\t\t\t\terrs[idx] = err\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(thriftEncodedBinary, binary); diff != \"\" {\n\t\t\t\terrs[idx] = fmt.Errorf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t\t\treturn\n\t\t\t}\n\t\t}(i)\n\t}\n\twg.Wait()\n\ts.NoError(multierr.Combine(errs...))\n}\n\nfunc (s *thriftRWEncoderSuite) TestDecode() {\n\tvar val workflow.HistoryEvent\n\terr := s.encoder.Decode(thriftEncodedBinary, &val)\n\ts.Nil(err)\n\tval.Equals(thriftObject)\n}\n\nfunc (s *thriftRWEncoderSuite) TestDecode_MissingVersion() {\n\tbinary := []byte{}\n\tvar val workflow.HistoryEvent\n\terr := s.encoder.Decode(binary, &val)\n\ts.Equal(MissingBinaryEncodingVersion, err)\n}\n\nfunc (s *thriftRWEncoderSuite) TestDecode_InvalidVersion() {\n\tbinary := []byte{}\n\tbinary = append(binary, thriftEncodedBinary...)\n\tbinary[0] = preambleVersion0 - 1\n\n\tvar val workflow.HistoryEvent\n\terr := s.encoder.Decode(binary, &val)\n\ts.Equal(InvalidBinaryEncodingVersion, err)\n}\n\nfunc int64Ptr(v int64) *int64 {\n\treturn &v\n}\n\nfunc stringPtr(v string) *string {\n\treturn &v\n}\n\nfunc boolPtr(v bool) *bool {\n\treturn &v\n}\n"
  },
  {
    "path": "common/collection/channelPriorityQueue.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport \"fmt\"\n\nconst numPriorities = 2\n\n// channelPriorityQueue is a priority queue built using channels\ntype channelPriorityQueue struct {\n\tchannels   []chan interface{}\n\tshutdownCh chan struct{}\n}\n\n// ChannelPriorityQueue is an interface for a priority queue\ntype ChannelPriorityQueue interface {\n\tAdd(priority int, item interface{}) bool\n\tRemove() (interface{}, bool)\n\tClose()\n}\n\n// NewChannelPriorityQueue returns a ChannelPriorityQueue\nfunc NewChannelPriorityQueue(queueSize int) ChannelPriorityQueue {\n\tchannels := make([]chan interface{}, numPriorities)\n\tfor i := range channels {\n\t\tchannels[i] = make(chan interface{}, queueSize)\n\t}\n\treturn &channelPriorityQueue{\n\t\tchannels:   channels,\n\t\tshutdownCh: make(chan struct{}),\n\t}\n}\n\n// Add adds an item to a channel in the queue. This is blocking and waits for\n// the queue to get empty if it is full. Returns false if the queue is closed.\nfunc (c *channelPriorityQueue) Add(priority int, item interface{}) bool {\n\tif priority >= numPriorities {\n\t\tpanic(fmt.Sprintf(\"trying to add item with invalid priority %v, queue only supports %v priorities\", priority, numPriorities))\n\t}\n\tselect {\n\tcase c.channels[priority] <- item:\n\tcase <-c.shutdownCh:\n\t\treturn false\n\t}\n\treturn true\n}\n\n// Remove removes an item from the priority queue. This is blocking till an\n// element becomes available in the priority queue\nfunc (c *channelPriorityQueue) Remove() (interface{}, bool) {\n\t// pick from highest priority if exists\n\tselect {\n\tcase item, ok := <-c.channels[0]:\n\t\treturn item, ok\n\tcase <-c.shutdownCh:\n\t\treturn nil, false\n\tdefault:\n\t}\n\n\t// blocking select from all priorities\n\tvar item interface{}\n\tvar ok bool\n\tselect {\n\tcase item, ok = <-c.channels[0]:\n\tcase item, ok = <-c.channels[1]:\n\tcase <-c.shutdownCh:\n\t}\n\treturn item, ok\n}\n\n// Destroy - destroys the channel priority queue\nfunc (c *channelPriorityQueue) Close() {\n\tclose(c.shutdownCh)\n}\n"
  },
  {
    "path": "common/collection/channelPriorityQueue_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestChannelPriorityQueue(t *testing.T) {\n\tqueue := NewChannelPriorityQueue(1)\n\n\tshutdown := queue.Add(1, 20)\n\tassert.True(t, shutdown)\n\n\tshutdown = queue.Add(0, 10)\n\tassert.True(t, shutdown)\n\n\titem, ok := queue.Remove()\n\tassert.Equal(t, 10, item)\n\tassert.True(t, ok)\n\n\titem, ok = queue.Remove()\n\tassert.Equal(t, 20, item)\n\tassert.True(t, ok)\n\n\tqueue.Close()\n\n\t// once we close the channel we should get shutdown at least once\n\tfor {\n\t\tshutdown = queue.Add(1, 20)\n\t\tif !shutdown {\n\t\t\tbreak\n\t\t}\n\t}\n\n\titem, ok = queue.Remove()\n\tassert.Nil(t, item)\n\tassert.False(t, ok)\n}\n\nfunc BenchmarkChannelPriorityQueue(b *testing.B) {\n\tqueue := NewChannelPriorityQueue(100)\n\n\tfor i := 0; i < 10; i++ {\n\t\tgo sendChannelQueue(queue)\n\t}\n\n\tfor n := 0; n < b.N; n++ {\n\t\tqueue.Remove()\n\t}\n}\n\nfunc sendChannelQueue(queue ChannelPriorityQueue) {\n\tfor {\n\t\tpriority := rand.Int() % numPriorities\n\t\tqueue.Add(priority, struct{}{})\n\t}\n}\n"
  },
  {
    "path": "common/collection/concurrent_priority_queue.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport \"sync\"\n\ntype (\n\tconcurrentPriorityQueueImpl[T any] struct {\n\t\tsync.RWMutex\n\t\tpriorityQueue Queue[T]\n\t}\n)\n\n// NewConcurrentPriorityQueue create a new concurrent priority queue\nfunc NewConcurrentPriorityQueue[T any](compareLess func(this T, other T) bool) Queue[T] {\n\treturn &concurrentPriorityQueueImpl[T]{\n\t\tpriorityQueue: NewPriorityQueue(compareLess),\n\t}\n}\n\n// Peek returns the top item of the priority queue\nfunc (pq *concurrentPriorityQueueImpl[T]) Peek() (T, error) {\n\tpq.RLock()\n\tdefer pq.RUnlock()\n\n\treturn pq.priorityQueue.Peek()\n}\n\n// Add push an item to priority queue\nfunc (pq *concurrentPriorityQueueImpl[T]) Add(item T) {\n\tpq.Lock()\n\tdefer pq.Unlock()\n\n\tpq.priorityQueue.Add(item)\n}\n\n// Remove pop an item from priority queue\nfunc (pq *concurrentPriorityQueueImpl[T]) Remove() (T, error) {\n\tpq.Lock()\n\tdefer pq.Unlock()\n\n\treturn pq.priorityQueue.Remove()\n}\n\n// IsEmpty indicate if the priority queue is empty\nfunc (pq *concurrentPriorityQueueImpl[T]) IsEmpty() bool {\n\tpq.RLock()\n\tdefer pq.RUnlock()\n\n\treturn pq.priorityQueue.IsEmpty()\n}\n\n// Len return the size of the queue\nfunc (pq *concurrentPriorityQueueImpl[T]) Len() int {\n\tpq.RLock()\n\tdefer pq.RUnlock()\n\n\treturn pq.priorityQueue.Len()\n}\n"
  },
  {
    "path": "common/collection/concurrent_queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"errors\"\n\t\"sync\"\n)\n\ntype (\n\tconcurrentQueueImpl[T any] struct {\n\t\tsync.RWMutex\n\t\titems []T\n\t}\n)\n\n// NewConcurrentQueue creates a new concurrent queue\nfunc NewConcurrentQueue[T any]() Queue[T] {\n\treturn &concurrentQueueImpl[T]{\n\t\titems: make([]T, 0, 1000),\n\t}\n}\n\nfunc (q *concurrentQueueImpl[T]) Peek() (T, error) {\n\tq.RLock()\n\tdefer q.RUnlock()\n\n\tvar item T\n\tif q.isEmptyLocked() {\n\t\treturn item, errors.New(\"queue is empty\")\n\t}\n\treturn q.items[0], nil\n}\n\nfunc (q *concurrentQueueImpl[T]) Add(item T) {\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tq.items = append(q.items, item)\n}\n\nfunc (q *concurrentQueueImpl[T]) Remove() (T, error) {\n\tq.Lock()\n\tdefer q.Unlock()\n\tvar item T\n\tif q.isEmptyLocked() {\n\t\treturn item, errors.New(\"queue is empty\")\n\t}\n\n\titem = q.items[0]\n\tq.items = q.items[1:]\n\n\treturn item, nil\n}\n\nfunc (q *concurrentQueueImpl[T]) IsEmpty() bool {\n\tq.RLock()\n\tdefer q.RUnlock()\n\n\treturn q.isEmptyLocked()\n}\n\nfunc (q *concurrentQueueImpl[T]) Len() int {\n\tq.RLock()\n\tdefer q.RUnlock()\n\n\treturn len(q.items)\n}\n\nfunc (q *concurrentQueueImpl[T]) isEmptyLocked() bool {\n\treturn len(q.items) == 0\n}\n"
  },
  {
    "path": "common/collection/concurrent_queue_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tconcurrentQueueSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tconcurrentQueue *concurrentQueueImpl[int]\n\t}\n)\n\nfunc TestConcurrentQueueSuite(t *testing.T) {\n\ts := new(concurrentQueueSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *concurrentQueueSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.concurrentQueue = NewConcurrentQueue[int]().(*concurrentQueueImpl[int])\n}\n\nfunc (s *concurrentQueueSuite) TearDownTest() {\n}\n\nfunc (s *concurrentQueueSuite) TestAddAndRemove() {\n\ts.Equal(0, s.concurrentQueue.Len())\n\ts.True(s.concurrentQueue.IsEmpty())\n\t_, err := s.concurrentQueue.Peek()\n\ts.Error(err)\n\t_, err = s.concurrentQueue.Remove()\n\ts.Error(err)\n\n\tnumItems := 100\n\titems := make([]int, 0, numItems)\n\tfor i := 0; i != 100; i++ {\n\t\tnum := rand.Int()\n\t\titems = append(items, num)\n\t\ts.concurrentQueue.Add(num)\n\t\ts.Equal(i+1, s.concurrentQueue.Len())\n\t}\n\ts.False(s.concurrentQueue.IsEmpty())\n\tnum, err := s.concurrentQueue.Peek()\n\ts.NoError(err)\n\ts.Equal(items[0], num)\n\n\tfor i := 0; i != 100; i++ {\n\t\tnum, err := s.concurrentQueue.Remove()\n\t\ts.NoError(err)\n\t\ts.Equal(items[i], num)\n\t\ts.Equal(numItems-i-1, s.concurrentQueue.Len())\n\t}\n\ts.True(s.concurrentQueue.IsEmpty())\n\t_, err = s.concurrentQueue.Peek()\n\ts.Error(err)\n\t_, err = s.concurrentQueue.Remove()\n\ts.Error(err)\n}\n\nfunc (s *concurrentQueueSuite) TestMultipleProducer() {\n\tconcurrency := 10\n\tnumItemsPerProducer := 10\n\n\tvar wg sync.WaitGroup\n\tfor i := 0; i != concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j != numItemsPerProducer; j++ {\n\t\t\t\ts.concurrentQueue.Add(rand.Int())\n\t\t\t}\n\t\t}()\n\t}\n\twg.Wait()\n\n\texpectedLength := concurrency * numItemsPerProducer\n\ts.Equal(expectedLength, s.concurrentQueue.Len())\n\ts.False(s.concurrentQueue.IsEmpty())\n\tfor i := 0; i != expectedLength; i++ {\n\t\t_, _ = s.concurrentQueue.Remove()\n\t}\n}\n\nfunc BenchmarkConcurrentQueue(b *testing.B) {\n\tqueue := NewConcurrentQueue[testTask]()\n\n\tfor i := 0; i < 100; i++ {\n\t\tgo send(queue)\n\t}\n\n\tfor n := 0; n < b.N; n++ {\n\t\tremove(queue)\n\t}\n}\n"
  },
  {
    "path": "common/collection/concurrent_tx_map.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n)\n\nconst (\n\t// nShards represents the number of shards\n\t// At any given point of time, there can only\n\t// be nShards number of concurrent writers to\n\t// the map at max\n\tnShards = 32\n)\n\ntype (\n\n\t// ShardedConcurrentTxMap is an implementation of\n\t// ConcurrentMap that internally uses multiple\n\t// sharded maps to increase parallelism\n\tShardedConcurrentTxMap struct {\n\t\tshards     [nShards]mapShard\n\t\thashfn     HashFunc\n\t\tsize       int32\n\t\tinitialCap int\n\t}\n\n\t// mapIteratorImpl represents an iterator type\n\t// for the concurrent map.\n\tmapIteratorImpl struct {\n\t\tstopCh chan struct{}\n\t\tdataCh chan *MapEntry\n\t}\n\n\t// mapShard represents a single instance\n\t// of thread safe map\n\tmapShard struct {\n\t\tsync.RWMutex\n\t\titems map[interface{}]interface{}\n\t}\n)\n\n// NewShardedConcurrentTxMap returns an instance of ShardedConcurrentMap\n//\n// ShardedConcurrentMap is a thread safe map that maintains upto nShards\n// number of maps internally to allow nShards writers to be acive at the\n// same time. This map *does not* use re-entrant locks, so access to the\n// map during iterator can cause a dead lock.\n//\n// @param initialSz\n//\n//\tThe initial size for the map\n//\n// @param hashfn\n//\n//\tThe hash function to use for sharding\nfunc NewShardedConcurrentTxMap(initialCap int, hashfn HashFunc) ConcurrentTxMap {\n\tcmap := new(ShardedConcurrentTxMap)\n\tcmap.hashfn = hashfn\n\tcmap.initialCap = max(nShards, initialCap/nShards)\n\treturn cmap\n}\n\n// Get returns the value corresponding to the key, if it exist\nfunc (cmap *ShardedConcurrentTxMap) Get(key interface{}) (interface{}, bool) {\n\tshard := cmap.getShard(key)\n\tvar ok bool\n\tvar value interface{}\n\tshard.RLock()\n\tif shard.items != nil {\n\t\tvalue, ok = shard.items[key]\n\t}\n\tshard.RUnlock()\n\treturn value, ok\n}\n\n// Contains returns true if the key exist and false otherwise\nfunc (cmap *ShardedConcurrentTxMap) Contains(key interface{}) bool {\n\t_, ok := cmap.Get(key)\n\treturn ok\n}\n\n// Put records the given key value mapping. Overwrites previous values\nfunc (cmap *ShardedConcurrentTxMap) Put(key interface{}, value interface{}) {\n\tshard := cmap.getShard(key)\n\tshard.Lock()\n\tcmap.lazyInitShard(shard)\n\t_, ok := shard.items[key]\n\tif !ok {\n\t\tatomic.AddInt32(&cmap.size, 1)\n\t}\n\tshard.items[key] = value\n\tshard.Unlock()\n}\n\n// PutIfNotExist records the mapping, if there is no mapping for this key already\n// Returns true if the mapping was recorded, false otherwise\nfunc (cmap *ShardedConcurrentTxMap) PutIfNotExist(key interface{}, value interface{}) bool {\n\tshard := cmap.getShard(key)\n\tvar ok bool\n\tshard.Lock()\n\tcmap.lazyInitShard(shard)\n\t_, ok = shard.items[key]\n\tif !ok {\n\t\tshard.items[key] = value\n\t\tatomic.AddInt32(&cmap.size, 1)\n\t}\n\tshard.Unlock()\n\treturn !ok\n}\n\n// Remove deletes the given key from the map\nfunc (cmap *ShardedConcurrentTxMap) Remove(key interface{}) {\n\tshard := cmap.getShard(key)\n\tshard.Lock()\n\tcmap.lazyInitShard(shard)\n\t_, ok := shard.items[key]\n\tif ok {\n\t\tdelete(shard.items, key)\n\t\tatomic.AddInt32(&cmap.size, -1)\n\t}\n\tshard.Unlock()\n}\n\n// GetAndDo returns the value corresponding to the key, and apply fn to key value before return value\n// return (value, value exist or not, error when evaluation fn)\nfunc (cmap *ShardedConcurrentTxMap) GetAndDo(key interface{}, fn ActionFunc) (interface{}, bool, error) {\n\tshard := cmap.getShard(key)\n\tvar value interface{}\n\tvar ok bool\n\tvar err error\n\tshard.Lock()\n\tif shard.items != nil {\n\t\tvalue, ok = shard.items[key]\n\t\tif ok {\n\t\t\terr = fn(key, value)\n\t\t}\n\t}\n\tshard.Unlock()\n\treturn value, ok, err\n}\n\n// PutOrDo put the key value in the map, if key does not exists, otherwise, call fn with existing key and value\n// return (value, fn evaluated or not, error when evaluation fn)\nfunc (cmap *ShardedConcurrentTxMap) PutOrDo(key interface{}, value interface{}, fn ActionFunc) (interface{}, bool, error) {\n\tshard := cmap.getShard(key)\n\tvar err error\n\tshard.Lock()\n\tcmap.lazyInitShard(shard)\n\tv, ok := shard.items[key]\n\tif !ok {\n\t\tshard.items[key] = value\n\t\tv = value\n\t\tatomic.AddInt32(&cmap.size, 1)\n\t} else {\n\t\terr = fn(key, v)\n\t}\n\tshard.Unlock()\n\treturn v, ok, err\n}\n\n// RemoveIf deletes the given key from the map if fn return true\nfunc (cmap *ShardedConcurrentTxMap) RemoveIf(key interface{}, fn PredicateFunc) bool {\n\tshard := cmap.getShard(key)\n\tvar removed bool\n\tshard.Lock()\n\tif shard.items != nil {\n\t\tvalue, ok := shard.items[key]\n\t\tif ok && fn(key, value) {\n\t\t\tremoved = true\n\t\t\tdelete(shard.items, key)\n\t\t\tatomic.AddInt32(&cmap.size, -1)\n\t\t}\n\t}\n\tshard.Unlock()\n\treturn removed\n}\n\nfunc newMapIterator() *mapIteratorImpl {\n\treturn &mapIteratorImpl{\n\t\tdataCh: make(chan *MapEntry, 8),\n\t\tstopCh: make(chan struct{}),\n\t}\n}\n\n// Close closes the iterator\nfunc (it *mapIteratorImpl) Close() {\n\tclose(it.stopCh)\n}\n\n// Entries returns a channel of map entries\nfunc (it *mapIteratorImpl) Entries() <-chan *MapEntry {\n\treturn it.dataCh\n}\n\n// Iter returns an iterator to the map. This map\n// does not use re-entrant locks, so access or modification\n// to the map during iteration can cause a dead lock.\nfunc (cmap *ShardedConcurrentTxMap) Iter() MapIterator {\n\n\titerator := newMapIterator()\n\n\tgo func(iterator *mapIteratorImpl) {\n\t\tfor i := 0; i < nShards; i++ {\n\t\t\tcmap.shards[i].RLock()\n\t\t\tfor k, v := range cmap.shards[i].items {\n\t\t\t\tentry := &MapEntry{Key: k, Value: v}\n\t\t\t\tselect {\n\t\t\t\tcase iterator.dataCh <- entry:\n\t\t\t\tcase <-iterator.stopCh:\n\t\t\t\t\tcmap.shards[i].RUnlock()\n\t\t\t\t\tclose(iterator.dataCh)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\t\t\tcmap.shards[i].RUnlock()\n\t\t}\n\t\tclose(iterator.dataCh)\n\t}(iterator)\n\n\treturn iterator\n}\n\n// Len returns the number of items in the map\nfunc (cmap *ShardedConcurrentTxMap) Len() int {\n\treturn int(atomic.LoadInt32(&cmap.size))\n}\n\nfunc (cmap *ShardedConcurrentTxMap) getShard(key interface{}) *mapShard {\n\tshardIdx := cmap.hashfn(key) % nShards\n\treturn &cmap.shards[shardIdx]\n}\n\nfunc (cmap *ShardedConcurrentTxMap) lazyInitShard(shard *mapShard) {\n\tif shard.items == nil {\n\t\tshard.items = make(map[interface{}]interface{}, cmap.initialCap)\n\t}\n}\n"
  },
  {
    "path": "common/collection/concurrent_tx_map_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"errors\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tConcurrentTxMapSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n\tboolType bool\n\tintType  int\n)\n\nfunc TestConcurrentTxMapSuite(t *testing.T) {\n\tsuite.Run(t, new(ConcurrentTxMapSuite))\n}\n\nfunc (s *ConcurrentTxMapSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *ConcurrentTxMapSuite) TestLen() {\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\n\tkey1 := \"0001\"\n\ttestMap.Put(key1, boolType(true))\n\ts.Equal(1, testMap.Len(), \"Wrong concurrent map size\")\n\n\ttestMap.Put(key1, boolType(false))\n\ts.Equal(1, testMap.Len(), \"Wrong concurrent map size\")\n\n\tkey2 := \"0002\"\n\ttestMap.Put(key2, boolType(false))\n\ts.Equal(2, testMap.Len(), \"Wrong concurrent map size\")\n\n\ttestMap.PutIfNotExist(key2, boolType(false))\n\ts.Equal(2, testMap.Len(), \"Wrong concurrent map size\")\n\n\ttestMap.Remove(key2)\n\ts.Equal(1, testMap.Len(), \"Wrong concurrent map size\")\n\n\ttestMap.Remove(key2)\n\ts.Equal(1, testMap.Len(), \"Wrong concurrent map size\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestGetAndDo() {\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\tkey := uuid.New()\n\tvar value intType\n\tfnApplied := false\n\n\tinterf, ok, err := testMap.GetAndDo(key, func(key interface{}, value interface{}) error {\n\t\tfnApplied = true\n\t\treturn nil\n\t})\n\ts.Nil(interf, \"GetAndDo should return nil when key not found\")\n\ts.Nil(err, \"GetAndDo should return nil when function not applied\")\n\ts.False(ok, \"GetAndDo should return false when key not found\")\n\ts.False(fnApplied, \"GetAndDo should not apply function when key not exixts\")\n\n\tvalue = intType(1)\n\ttestMap.Put(key, &value)\n\tinterf, ok, err = testMap.GetAndDo(key, func(key interface{}, value interface{}) error {\n\t\tfnApplied = true\n\t\tintValue := value.(*intType)\n\t\t*intValue++\n\t\treturn errors.New(\"some err\")\n\t})\n\n\tvalue1 := interf.(*intType)\n\ts.Equal(*(value1), intType(2))\n\ts.NotNil(err, \"GetAndDo should return non nil when function applied\")\n\ts.True(ok, \"GetAndDo should return true when key found\")\n\ts.True(fnApplied, \"GetAndDo should apply function when key exixts\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestPutOrDo() {\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\tkey := uuid.New()\n\tvar value intType\n\tfnApplied := false\n\n\tvalue = intType(1)\n\tinterf, ok, err := testMap.PutOrDo(key, &value, func(key interface{}, value interface{}) error {\n\t\tfnApplied = true\n\t\treturn errors.New(\"some err\")\n\t})\n\tvalueRetuern := interf.(*intType)\n\ts.Equal(value, *valueRetuern)\n\ts.Nil(err, \"PutOrDo should return nil when function not applied\")\n\ts.False(ok, \"PutOrDo should return false when function not applied\")\n\ts.False(fnApplied, \"PutOrDo should not apply function when key not exixts\")\n\n\tanotherValue := intType(111)\n\tinterf, ok, err = testMap.PutOrDo(key, &anotherValue, func(key interface{}, value interface{}) error {\n\t\tfnApplied = true\n\t\tintValue := value.(*intType)\n\t\t*intValue++\n\t\treturn errors.New(\"some err\")\n\t})\n\tvalueRetuern = interf.(*intType)\n\ts.Equal(value, *valueRetuern)\n\ts.NotNil(err, \"PutOrDo should return non nil when function applied\")\n\ts.True(ok, \"PutOrDo should return true when function applied\")\n\ts.True(fnApplied, \"PutOrDo should apply function when key exixts\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestRemoveIf() {\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\tkey := uuid.New()\n\tvalue := intType(1)\n\ttestMap.Put(key, &value)\n\n\tremoved := testMap.RemoveIf(key, func(key interface{}, value interface{}) bool {\n\t\tintValue := value.(*intType)\n\t\treturn *intValue == intType(2)\n\t})\n\ts.Equal(1, testMap.Len(), \"TestRemoveIf should only entry if condition is met\")\n\ts.False(removed, \"TestRemoveIf should return false if key is not deleted\")\n\n\tremoved = testMap.RemoveIf(key, func(key interface{}, value interface{}) bool {\n\t\tintValue := value.(*intType)\n\t\treturn *intValue == intType(1)\n\t})\n\ts.Equal(0, testMap.Len(), \"TestRemoveIf should only entry if condition is met\")\n\ts.True(removed, \"TestRemoveIf should return true if key is deleted\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestGetAfterPut() {\n\n\tcountMap := make(map[string]int)\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\n\tfor i := 0; i < 1024; i++ {\n\t\tkey := uuid.New()\n\t\tcountMap[key] = 0\n\t\ttestMap.Put(key, boolType(true))\n\t}\n\n\tfor k := range countMap {\n\t\tv, ok := testMap.Get(k)\n\t\tboolValue := v.(boolType)\n\t\ts.True(ok, \"Get after put failed\")\n\t\ts.True(bool(boolValue), \"Wrong value returned from map\")\n\t}\n\n\ts.Equal(len(countMap), testMap.Len(), \"Size() returned wrong value\")\n\n\tit := testMap.Iter()\n\tfor entry := range it.Entries() {\n\t\tcountMap[entry.Key.(string)]++\n\t}\n\tit.Close()\n\n\tfor _, v := range countMap {\n\t\ts.Equal(1, v, \"Iterator test failed\")\n\t}\n\n\tfor k := range countMap {\n\t\ttestMap.Remove(k)\n\t}\n\n\ts.Equal(0, testMap.Len(), \"Map returned non-zero size after deleting all entries\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestPutIfNotExist() {\n\ttestMap := NewShardedConcurrentTxMap(1, UUIDHashCode)\n\tkey := uuid.New()\n\tok := testMap.PutIfNotExist(key, boolType(true))\n\ts.True(ok, \"PutIfNotExist failed to insert item\")\n\tok = testMap.PutIfNotExist(key, boolType(true))\n\ts.False(ok, \"PutIfNotExist invariant failed\")\n}\n\nfunc (s *ConcurrentTxMapSuite) TestMapConcurrency() {\n\tnKeys := 1024\n\tkeys := make([]string, nKeys)\n\tfor i := 0; i < nKeys; i++ {\n\t\tkeys[i] = uuid.New()\n\t}\n\n\tvar total int32\n\tvar startWG sync.WaitGroup\n\tvar doneWG sync.WaitGroup\n\ttestMap := NewShardedConcurrentTxMap(1024, UUIDHashCode)\n\n\tstartWG.Add(1)\n\n\tfor i := 0; i < 10; i++ {\n\n\t\tdoneWG.Add(1)\n\n\t\tgo func() {\n\t\t\tstartWG.Wait()\n\t\t\tfor n := 0; n < nKeys; n++ {\n\t\t\t\tval := intType(rand.Int())\n\t\t\t\tif testMap.PutIfNotExist(keys[n], val) {\n\t\t\t\t\tatomic.AddInt32(&total, int32(val))\n\t\t\t\t\t_, ok := testMap.Get(keys[n])\n\t\t\t\t\ts.True(ok, \"Concurrency Get test failed\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tdoneWG.Done()\n\t\t}()\n\t}\n\n\tstartWG.Done()\n\tdoneWG.Wait()\n\n\ts.Equal(nKeys, testMap.Len(), \"Wrong concurrent map size\")\n\n\tvar gotTotal int32\n\tfor i := 0; i < nKeys; i++ {\n\t\tv, ok := testMap.Get(keys[i])\n\t\ts.True(ok, \"Get failed to find previously inserted key\")\n\t\tintVal := v.(intType)\n\t\tgotTotal += int32(intVal)\n\t}\n\n\ts.Equal(total, gotTotal, \"Concurrent put test failed, wrong sum of values inserted\")\n}\n"
  },
  {
    "path": "common/collection/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\ntype (\n\t// Queue is the interface for queue\n\tQueue[T any] interface {\n\t\t// Peek returns the first item of the queue\n\t\tPeek() (T, error)\n\t\t// Add push an item to the queue\n\t\tAdd(item T)\n\t\t// Remove pop an item from the queue\n\t\tRemove() (T, error)\n\t\t// IsEmpty indicate if the queue is empty\n\t\tIsEmpty() bool\n\t\t// Len return the size of the queue\n\t\tLen() int\n\t}\n\n\t// HashFunc represents a hash function for string\n\tHashFunc func(interface{}) uint32\n\n\t// ActionFunc take a key and value, do calculation and return err\n\tActionFunc func(key interface{}, value interface{}) error\n\t// PredicateFunc take a key and value, do calculation and return boolean\n\tPredicateFunc func(key interface{}, value interface{}) bool\n\n\t// ConcurrentTxMap is a generic interface for any implementation of a dictionary\n\t// or a key value lookup table that is thread safe, and providing functionality\n\t// to modify key / value pair inside within a transaction\n\tConcurrentTxMap interface {\n\t\t// Get returns the value for the given key\n\t\tGet(key interface{}) (interface{}, bool)\n\t\t// Contains returns true if the key exist and false otherwise\n\t\tContains(key interface{}) bool\n\t\t// Put records the mapping from given key to value\n\t\tPut(key interface{}, value interface{})\n\t\t// PutIfNotExist records the key value mapping only\n\t\t// if the mapping does not already exist\n\t\tPutIfNotExist(key interface{}, value interface{}) bool\n\t\t// Remove deletes the key from the map\n\t\tRemove(key interface{})\n\t\t// GetAndDo returns the value corresponding to the key, and apply fn to key value before return value\n\t\t// return (value, value exist or not, error when evaluation fn)\n\t\tGetAndDo(key interface{}, fn ActionFunc) (interface{}, bool, error)\n\t\t// PutOrDo put the key value in the map, if key does not exists, otherwise, call fn with existing key and value\n\t\t// return (value, fn evaluated or not, error when evaluation fn)\n\t\tPutOrDo(key interface{}, value interface{}, fn ActionFunc) (interface{}, bool, error)\n\t\t// RemoveIf deletes the given key from the map if fn return true\n\t\t// return whether the key is removed or not\n\t\tRemoveIf(key interface{}, fn PredicateFunc) bool\n\t\t// Iter returns an iterator to the map\n\t\tIter() MapIterator\n\t\t// Len returns the number of items in the map\n\t\tLen() int\n\t}\n\n\t// MapIterator represents the interface for map iterators\n\tMapIterator interface {\n\t\t// Close closes the iterator\n\t\t// and releases any allocated resources\n\t\tClose()\n\t\t// Entries returns a channel of MapEntry\n\t\t// objects that can be used in a range loop\n\t\tEntries() <-chan *MapEntry\n\t}\n\n\t// MapEntry represents a key-value entry within the map\n\tMapEntry struct {\n\t\t// Key represents the key\n\t\tKey interface{}\n\t\t// Value represents the value\n\t\tValue interface{}\n\t}\n)\n\nconst (\n\t// UUIDStringLength is the length of an UUID represented as a hex string\n\tUUIDStringLength = 36 // xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx\n)\n"
  },
  {
    "path": "common/collection/iterator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\ntype (\n\t// Iterator represents the interface for iterator\n\tIterator interface {\n\t\t// HasNext return whether this iterator has next value\n\t\tHasNext() bool\n\t\t// Next returns the next item and error\n\t\tNext() (interface{}, error)\n\t}\n)\n"
  },
  {
    "path": "common/collection/ordered_map.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"container/list\"\n\t\"sync\"\n)\n\ntype (\n\t// OrderedMap is an interface for a dictionary which during iteration\n\t// return all values in their insertion order\n\tOrderedMap interface {\n\t\t// Get returns the value for the given key\n\t\tGet(key interface{}) (interface{}, bool)\n\t\t// Contains returns true if the key exist and false otherwise\n\t\tContains(key interface{}) bool\n\t\t// Put records the mapping from given key to value\n\t\tPut(key interface{}, value interface{})\n\t\t// Remove deletes the key from the map\n\t\tRemove(key interface{})\n\t\t// Iter returns an iterator to the map which iterates over map entries\n\t\t// in insertion order\n\t\tIter() MapIterator\n\t\t// Len returns the number of items in the map\n\t\tLen() int\n\t}\n\n\torderedMapValue struct {\n\t\tvalue interface{}\n\n\t\tlistElement *list.Element\n\t}\n\n\torderedMap struct {\n\t\titems map[interface{}]orderedMapValue\n\t\tlist  *list.List\n\t}\n\n\tconcurrentOrderedMap struct {\n\t\tsync.RWMutex\n\n\t\torderedMap *orderedMap\n\t}\n)\n\n// NewOrderedMap creates a new OrderedMap\n// implementation has O(1) complexity for all methods\nfunc NewOrderedMap() OrderedMap {\n\treturn &orderedMap{\n\t\titems: make(map[interface{}]orderedMapValue),\n\t\tlist:  list.New(),\n\t}\n}\n\nfunc (m *orderedMap) Get(key interface{}) (interface{}, bool) {\n\tif mapValue, ok := m.items[key]; ok {\n\t\treturn mapValue.value, true\n\t}\n\n\treturn nil, false\n}\n\nfunc (m *orderedMap) Contains(key interface{}) bool {\n\t_, ok := m.items[key]\n\treturn ok\n}\n\nfunc (m *orderedMap) Put(key interface{}, value interface{}) {\n\t// remove existing key if there's one\n\tm.Remove(key)\n\n\tlistElement := m.list.PushBack(&MapEntry{\n\t\tKey:   key,\n\t\tValue: value,\n\t})\n\tm.items[key] = orderedMapValue{\n\t\tvalue:       value,\n\t\tlistElement: listElement,\n\t}\n}\n\nfunc (m *orderedMap) Remove(key interface{}) {\n\tmapValue, ok := m.items[key]\n\tif !ok {\n\t\treturn\n\t}\n\n\tm.list.Remove(mapValue.listElement)\n\tdelete(m.items, key)\n}\n\nfunc (m *orderedMap) Iter() MapIterator {\n\titerator := newMapIterator()\n\n\tgo orderedMapIteratorLoop(nil, iterator, m.list)\n\n\treturn iterator\n}\n\nfunc (m *orderedMap) Len() int {\n\treturn len(m.items)\n}\n\n// NewConcurrentOrderedMap creates a new thread safe OrderedMap\nfunc NewConcurrentOrderedMap() OrderedMap {\n\treturn &concurrentOrderedMap{\n\t\torderedMap: NewOrderedMap().(*orderedMap),\n\t}\n}\n\nfunc (m *concurrentOrderedMap) Get(key interface{}) (interface{}, bool) {\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\treturn m.orderedMap.Get(key)\n}\n\nfunc (m *concurrentOrderedMap) Contains(key interface{}) bool {\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\treturn m.orderedMap.Contains(key)\n}\n\nfunc (m *concurrentOrderedMap) Put(key interface{}, value interface{}) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tm.orderedMap.Put(key, value)\n}\n\nfunc (m *concurrentOrderedMap) Remove(key interface{}) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tm.orderedMap.Remove(key)\n}\n\n// Iter returns an iterator to the map.\n// NOTE that any modification to the map during iteration\n// will lead to a dead lock.\nfunc (m *concurrentOrderedMap) Iter() MapIterator {\n\titerator := newMapIterator()\n\n\tgo orderedMapIteratorLoop(&m.RWMutex, iterator, m.orderedMap.list)\n\n\treturn iterator\n}\n\nfunc (m *concurrentOrderedMap) Len() int {\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\treturn m.orderedMap.Len()\n}\n\nfunc orderedMapIteratorLoop(\n\tmutex *sync.RWMutex,\n\titerator *mapIteratorImpl,\n\tlist *list.List,\n) {\n\tif mutex != nil {\n\t\tmutex.RLock()\n\t\tdefer mutex.RUnlock()\n\t}\n\n\tfor element := list.Front(); element != nil; element = element.Next() {\n\t\tselect {\n\t\tcase iterator.dataCh <- element.Value.(*MapEntry):\n\t\t\t// no-op\n\t\tcase <-iterator.stopCh:\n\t\t\tclose(iterator.dataCh)\n\t\t\treturn\n\t\t}\n\t}\n\tclose(iterator.dataCh)\n}\n"
  },
  {
    "path": "common/collection/ordered_map_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\torderedMapSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\torderedMap OrderedMap\n\t}\n)\n\nfunc TestOrderedMapSuite(t *testing.T) {\n\ts := new(orderedMapSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *orderedMapSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.orderedMap = NewOrderedMap()\n}\n\nfunc (s *orderedMapSuite) TestBasicOperation() {\n\tkey := \"0001\"\n\tvalue1 := boolType(true)\n\tvalue2 := intType(1234)\n\n\ts.Equal(0, s.orderedMap.Len())\n\n\t// remove a non-exist key\n\ts.orderedMap.Remove(key)\n\n\t// get a non-exist key\n\tv, ok := s.orderedMap.Get(key)\n\ts.False(ok)\n\ts.Nil(v)\n\ts.False(s.orderedMap.Contains(key))\n\n\ts.orderedMap.Put(key, value1)\n\tv, ok = s.orderedMap.Get(key)\n\ts.True(ok)\n\ts.Equal(value1, v)\n\ts.True(s.orderedMap.Contains(key))\n\ts.Equal(1, s.orderedMap.Len())\n\n\ts.orderedMap.Put(key, value2)\n\tv, ok = s.orderedMap.Get(key)\n\ts.True(ok)\n\ts.Equal(value2, v)\n\ts.True(s.orderedMap.Contains(key))\n\ts.Equal(1, s.orderedMap.Len())\n\n\ts.orderedMap.Remove(key)\n\ts.False(s.orderedMap.Contains(key))\n\ts.Equal(0, s.orderedMap.Len())\n}\n\nfunc (s *orderedMapSuite) TestIterator() {\n\tnumItems := 10\n\tfor i := 0; i != numItems; i++ {\n\t\ts.orderedMap.Put(i, intType(i))\n\t}\n\n\titer := s.orderedMap.Iter()\n\tnumIterated := 0\n\tfor entry := range iter.Entries() {\n\t\ts.Equal(numIterated, entry.Key.(int))\n\t\ts.Equal(intType(numIterated), entry.Value.(intType))\n\t\tnumIterated++\n\t}\n\titer.Close()\n\n\t// remove all even numbers\n\tfor i := 0; i != numItems; i++ {\n\t\tif i%2 == 0 {\n\t\t\ts.orderedMap.Remove(i)\n\t\t}\n\t}\n\t// add more odd numbers at the end\n\tfor i := numItems; i != 2*numItems; i++ {\n\t\tif i%2 == 1 {\n\t\t\ts.orderedMap.Put(i, intType(i))\n\t\t}\n\t}\n\n\titer = s.orderedMap.Iter()\n\tnumIterated = 0\n\tfor entry := range iter.Entries() {\n\t\ts.Equal(2*numIterated+1, entry.Key.(int))\n\t\ts.Equal(intType(2*numIterated+1), entry.Value.(intType))\n\t\tnumIterated++\n\t}\n\titer.Close()\n}\n"
  },
  {
    "path": "common/collection/pagingIterator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\ntype (\n\t// PaginationFn is the function which get a page of results\n\tPaginationFn func(paginationToken []byte) ([]interface{}, []byte, error)\n\n\t// PagingIteratorImpl is the implementation of PagingIterator\n\tPagingIteratorImpl struct {\n\t\tpaginationFn      PaginationFn\n\t\tpageToken         []byte\n\t\tpageErr           error\n\t\tpageItems         []interface{}\n\t\tnextPageItemIndex int\n\t}\n)\n\n// NewPagingIterator create a new paging iterator\n// TODO: this implementation should be removed in favor of pagination/iterator.go\nfunc NewPagingIterator(paginationFn PaginationFn) Iterator {\n\titer := &PagingIteratorImpl{\n\t\tpaginationFn:      paginationFn,\n\t\tpageToken:         nil,\n\t\tpageErr:           nil,\n\t\tpageItems:         nil,\n\t\tnextPageItemIndex: 0,\n\t}\n\titer.getNextPage() // this will initialize the paging iterator\n\treturn iter\n}\n\n// HasNext return whether has next item or err\nfunc (iter *PagingIteratorImpl) HasNext() bool {\n\t// pagination encounters error\n\tif iter.pageErr != nil {\n\t\treturn true\n\t}\n\n\t// still have local cached item to return\n\tif iter.nextPageItemIndex < len(iter.pageItems) {\n\t\treturn true\n\t}\n\n\tif len(iter.pageToken) != 0 {\n\t\titer.getNextPage()\n\t\treturn iter.HasNext()\n\t}\n\n\treturn false\n}\n\n// Next return next item or err\nfunc (iter *PagingIteratorImpl) Next() (interface{}, error) {\n\tif !iter.HasNext() {\n\t\tpanic(\"HistoryEventIterator Next() called without checking HasNext()\")\n\t}\n\n\tif iter.pageErr != nil {\n\t\terr := iter.pageErr\n\t\titer.pageErr = nil\n\t\treturn nil, err\n\t}\n\n\t// we have cached events\n\tif iter.nextPageItemIndex < len(iter.pageItems) {\n\t\tindex := iter.nextPageItemIndex\n\t\titer.nextPageItemIndex++\n\t\treturn iter.pageItems[index], nil\n\t}\n\n\tpanic(\"HistoryEventIterator Next() should return either a history event or a err\")\n}\n\nfunc (iter *PagingIteratorImpl) getNextPage() {\n\titems, token, err := iter.paginationFn(iter.pageToken)\n\tif err == nil {\n\t\titer.pageItems = items\n\t\titer.pageToken = token\n\t\titer.pageErr = nil\n\t} else {\n\t\titer.pageItems = nil\n\t\titer.pageToken = nil\n\t\titer.pageErr = err\n\t}\n\titer.nextPageItemIndex = 0\n}\n"
  },
  {
    "path": "common/collection/pagingIterator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"errors\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tpagingIteratorSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestPagingIteratorSuite(t *testing.T) {\n\ts := new(pagingIteratorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *pagingIteratorSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\n}\n\nfunc (s *pagingIteratorSuite) TearDownSuite() {\n\n}\n\nfunc (s *pagingIteratorSuite) SetupTest() {\n\n}\n\nfunc (s *pagingIteratorSuite) TearDownTest() {\n\n}\n\nfunc (s *pagingIteratorSuite) TestIteration_NoErr() {\n\tphase := 0\n\toutputs := [][]interface{}{\n\t\t[]interface{}{1, 2, 3, 4, 5},\n\t\t[]interface{}{},\n\t\t[]interface{}{6},\n\t\t[]interface{}{},\n\t}\n\ttokens := [][]byte{\n\t\t[]byte(\"some random token 1\"),\n\t\t[]byte(\"some random token 2\"),\n\t\t[]byte(\"some random token 3\"),\n\t\t[]byte(nil),\n\t}\n\tpagingFn := func(token []byte) ([]interface{}, []byte, error) {\n\t\tswitch phase {\n\t\tcase 0:\n\t\t\ts.Equal(0, len(token))\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn outputs[phase], tokens[phase], nil\n\t\tcase 1:\n\t\t\ts.Equal(tokens[0], token)\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn outputs[phase], tokens[phase], nil\n\t\tcase 2:\n\t\t\ts.Equal(tokens[1], token)\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn outputs[phase], tokens[phase], nil\n\t\tcase 3:\n\t\t\ts.Equal(tokens[2], token)\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn outputs[phase], tokens[phase], nil\n\t\tdefault:\n\t\t\tpanic(\"should not reach here during test\")\n\t\t}\n\t}\n\n\tresult := []int{}\n\tite := NewPagingIterator(pagingFn)\n\tfor ite.HasNext() {\n\t\titem, err := ite.Next()\n\t\ts.Nil(err)\n\t\tnum, ok := item.(int)\n\t\ts.True(ok)\n\t\tresult = append(result, num)\n\t}\n\ts.Equal([]int{1, 2, 3, 4, 5, 6}, result)\n}\n\nfunc (s *pagingIteratorSuite) TestIteration_Err_Beginging() {\n\tphase := 0\n\tite := NewPagingIterator(func(token []byte) ([]interface{}, []byte, error) {\n\t\tswitch phase {\n\t\tcase 0:\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn nil, nil, errors.New(\"some random error\")\n\t\tdefault:\n\t\t\tpanic(\"should not reach here during test\")\n\t\t}\n\t})\n\n\ts.True(ite.HasNext())\n\titem, err := ite.Next()\n\ts.Nil(item)\n\ts.NotNil(err)\n\ts.False(ite.HasNext())\n}\n\nfunc (s *pagingIteratorSuite) TestIteration_Err_NotBegining() {\n\n\tphase := 0\n\toutputs := [][]interface{}{\n\t\t[]interface{}{1, 2, 3, 4, 5},\n\t}\n\ttokens := [][]byte{\n\t\t[]byte(\"some random token 1\"),\n\t}\n\tpagingFn := func(token []byte) ([]interface{}, []byte, error) {\n\t\tswitch phase {\n\t\tcase 0:\n\t\t\ts.Equal(0, len(token))\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn outputs[phase], tokens[phase], nil\n\t\tcase 1:\n\t\t\ts.Equal(tokens[0], token)\n\t\t\tdefer func() { phase++ }()\n\t\t\treturn nil, nil, errors.New(\"some random error\")\n\t\tdefault:\n\t\t\tpanic(\"should not reach here during test\")\n\t\t}\n\t}\n\n\tresult := []int{}\n\tite := NewPagingIterator(pagingFn)\n\tfor ite.HasNext() {\n\t\titem, err := ite.Next()\n\t\tif err != nil {\n\t\t\tbreak\n\t\t}\n\t\tnum, ok := item.(int)\n\t\ts.True(ok)\n\t\tresult = append(result, num)\n\t}\n\ts.Equal([]int{1, 2, 3, 4, 5}, result)\n}\n"
  },
  {
    "path": "common/collection/priority_queue.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"container/heap\"\n\t\"errors\"\n)\n\ntype (\n\tpriorityQueueImpl[T any] struct {\n\t\tcompareLess func(this T, other T) bool\n\t\titems       []T\n\t}\n)\n\n// NewPriorityQueue create a new priority queue\n// with the provided list of items.\n// PriorityQueue will take ownership of the passed in items,\n// so caller should stop modifying it.\n// The complexity is O(n) where n is the number of items\nfunc NewPriorityQueue[T any](\n\tcompareLess func(this T, other T) bool,\n\titems ...T,\n) Queue[T] {\n\tpq := &priorityQueueImpl[T]{\n\t\tcompareLess: compareLess,\n\t\titems:       items,\n\t}\n\tif len(pq.items) > 0 {\n\t\theap.Init(pq)\n\t}\n\treturn pq\n}\n\n// Peek returns the top item of the priority queue\nfunc (pq *priorityQueueImpl[T]) Peek() (T, error) {\n\tvar item T\n\tif pq.IsEmpty() {\n\t\treturn item, errors.New(\"queue is empty\")\n\t}\n\treturn pq.items[0], nil\n}\n\n// Add push an item to priority queue\nfunc (pq *priorityQueueImpl[T]) Add(item T) {\n\theap.Push(pq, item)\n}\n\n// Remove pop an item from priority queue\nfunc (pq *priorityQueueImpl[T]) Remove() (T, error) {\n\tvar item T\n\tif pq.IsEmpty() {\n\t\treturn item, errors.New(\"queue is empty\")\n\t}\n\treturn heap.Pop(pq).(T), nil\n}\n\n// IsEmpty indicate if the priority queue is empty\nfunc (pq *priorityQueueImpl[T]) IsEmpty() bool {\n\treturn pq.Len() == 0\n}\n\n// below are the functions used by heap.Interface and go internal heap implementation\n\n// Len implements sort.Interface\nfunc (pq *priorityQueueImpl[T]) Len() int {\n\treturn len(pq.items)\n}\n\n// Less implements sort.Interface\nfunc (pq *priorityQueueImpl[T]) Less(i, j int) bool {\n\treturn pq.compareLess(pq.items[i], pq.items[j])\n}\n\n// Swap implements sort.Interface\nfunc (pq *priorityQueueImpl[T]) Swap(i, j int) {\n\tpq.items[i], pq.items[j] = pq.items[j], pq.items[i]\n}\n\n// Push push an item to priority queue, used by go internal heap implementation\nfunc (pq *priorityQueueImpl[T]) Push(item interface{}) {\n\tpq.items = append(pq.items, item.(T))\n}\n\n// Pop pop an item from priority queue, used by go internal heap implementation\nfunc (pq *priorityQueueImpl[T]) Pop() interface{} {\n\tpqItem := pq.items[pq.Len()-1]\n\tpq.items = pq.items[0 : pq.Len()-1]\n\treturn pqItem\n}\n"
  },
  {
    "path": "common/collection/priority_queue_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"math/rand\"\n\t\"runtime\"\n\t\"sort\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tPriorityQueueSuite struct {\n\t\tsuite.Suite\n\t\tpq Queue[testPriorityQueueItem]\n\t}\n\n\ttestPriorityQueueItem struct {\n\t\tvalue int\n\t}\n)\n\nfunc testPriorityQueueItemCompareLess(this testPriorityQueueItem, that testPriorityQueueItem) bool {\n\treturn this.value < that.value\n}\n\nfunc TestPriorityQueueSuite(t *testing.T) {\n\tsuite.Run(t, new(PriorityQueueSuite))\n}\n\nfunc (s *PriorityQueueSuite) SetupTest() {\n\ts.pq = NewPriorityQueue(testPriorityQueueItemCompareLess)\n}\n\nfunc (s *PriorityQueueSuite) TestInsertAndPop() {\n\ts.pq.Add(testPriorityQueueItem{10})\n\ts.pq.Add(testPriorityQueueItem{3})\n\ts.pq.Add(testPriorityQueueItem{5})\n\ts.pq.Add(testPriorityQueueItem{4})\n\ts.pq.Add(testPriorityQueueItem{1})\n\ts.pq.Add(testPriorityQueueItem{16})\n\ts.pq.Add(testPriorityQueueItem{-10})\n\n\texpected := []int{-10, 1, 3, 4, 5, 10, 16}\n\tresult := []int{}\n\n\tfor !s.pq.IsEmpty() {\n\t\titem, err := s.pq.Remove()\n\t\ts.NoError(err)\n\t\tresult = append(result, item.value)\n\t}\n\ts.Equal(expected, result)\n\n\ts.pq.Add(testPriorityQueueItem{1000})\n\ts.pq.Add(testPriorityQueueItem{1233})\n\ts.pq.Remove() // remove 1000\n\ts.pq.Add(testPriorityQueueItem{4})\n\ts.pq.Add(testPriorityQueueItem{18})\n\ts.pq.Add(testPriorityQueueItem{192})\n\ts.pq.Add(testPriorityQueueItem{255})\n\ts.pq.Remove() // remove 4\n\ts.pq.Remove() // remove 18\n\ts.pq.Add(testPriorityQueueItem{59})\n\ts.pq.Add(testPriorityQueueItem{727})\n\n\texpected = []int{59, 192, 255, 727, 1233}\n\tresult = []int{}\n\n\tfor !s.pq.IsEmpty() {\n\t\titem, err := s.pq.Remove()\n\t\ts.NoError(err)\n\t\tresult = append(result, item.value)\n\t}\n\ts.Equal(expected, result)\n}\n\nfunc (s *PriorityQueueSuite) TestRandomNumber() {\n\tfor round := 0; round < 1000; round++ {\n\n\t\texpected := []int{}\n\t\tresult := []int{}\n\t\tfor i := 0; i < 1000; i++ {\n\t\t\tnum := rand.Int()\n\t\t\ts.pq.Add(testPriorityQueueItem{num})\n\t\t\texpected = append(expected, num)\n\t\t}\n\t\tsort.Ints(expected)\n\n\t\tfor !s.pq.IsEmpty() {\n\t\t\titem, err := s.pq.Remove()\n\t\t\ts.NoError(err)\n\t\t\tresult = append(result, item.value)\n\t\t}\n\t\ts.Equal(expected, result)\n\t}\n}\n\ntype testTask struct {\n\tid       string\n\tpriority int\n}\n\nfunc BenchmarkConcurrentPriorityQueue(b *testing.B) {\n\tqueue := NewConcurrentPriorityQueue(func(this testTask, other testTask) bool {\n\t\treturn this.priority < other.priority\n\t})\n\n\tfor i := 0; i < 100; i++ {\n\t\tgo send(queue)\n\t}\n\n\tfor n := 0; n < b.N; n++ {\n\t\tremove(queue)\n\t}\n}\n\nfunc remove(queue Queue[testTask]) {\n\tfor queue.IsEmpty() {\n\t\truntime.Gosched()\n\t}\n\tqueue.Remove()\n}\n\nfunc send(queue Queue[testTask]) {\n\tfor {\n\t\tt := testTask{\n\t\t\tid:       \"abc\",\n\t\t\tpriority: rand.Int() % numPriorities,\n\t\t}\n\t\tqueue.Add(t)\n\t}\n}\n"
  },
  {
    "path": "common/collection/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage collection\n\nimport (\n\t\"encoding/binary\"\n\t\"encoding/hex\"\n)\n\n// UUIDHashCode is a hash function for hashing string uuid\n// if the uuid is malformed, then the hash function always\n// returns 0 as the hash value\nfunc UUIDHashCode(input interface{}) uint32 {\n\tkey, ok := input.(string)\n\tif !ok {\n\t\treturn 0\n\t}\n\tif len(key) != UUIDStringLength {\n\t\treturn 0\n\t}\n\t// Use the first 4 bytes of the uuid as the hash\n\tb, err := hex.DecodeString(key[:8])\n\tif err != nil {\n\t\treturn 0\n\t}\n\treturn binary.BigEndian.Uint32(b)\n}\n"
  },
  {
    "path": "common/config/archival.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\n// Validate validates the archival config\nfunc (a *Archival) Validate(domainDefaults *ArchivalDomainDefaults) error {\n\tif a.History.Status == constants.ArchivalEnabled {\n\t\tif domainDefaults.History.URI == \"\" || a.History.Provider == nil {\n\t\t\treturn errors.New(\"invalid history archival config, must provide domainDefaults.History.URI and Provider\")\n\t\t}\n\t} else {\n\t\tif a.History.EnableRead {\n\t\t\treturn errors.New(\"invalid history archival config, cannot EnableRead when archival is disabled\")\n\t\t}\n\t}\n\n\tif a.Visibility.Status == constants.ArchivalEnabled {\n\t\tif domainDefaults.Visibility.URI == \"\" || a.Visibility.Provider == nil {\n\t\t\treturn errors.New(\"invalid visibility archival config, must provide domainDefaults.Visibility.URI and Provider\")\n\t\t}\n\t} else {\n\t\tif a.Visibility.EnableRead {\n\t\t\treturn errors.New(\"invalid visibility archival config, cannot EnableRead when archival is disabled\")\n\t\t}\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/config/archival_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\nfunc defaultFilestoreConfig(t *testing.T) *YamlNode {\n\tnode, err := ToYamlNode(&FilestoreArchiver{\n\t\tFileMode: \"044\",\n\t})\n\trequire.NoError(t, err)\n\treturn node\n}\n\n// History archival\n\nfunc TestValidEnabledHistoryArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tHistory: HistoryArchival{\n\t\t\tStatus: constants.ArchivalEnabled,\n\t\t\tProvider: HistoryArchiverProvider{\n\t\t\t\tFilestoreConfig: defaultFilestoreConfig(t),\n\t\t\t},\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{\n\t\tHistory: HistoryArchivalDomainDefaults{\n\t\t\tURI: \"/var/tmp\",\n\t\t},\n\t})\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidHEnabledHistoryArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tHistory: HistoryArchival{\n\t\t\tStatus: constants.ArchivalEnabled,\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.Error(t, err)\n}\n\nfunc TestValidDisabledHistoryArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tHistory: HistoryArchival{\n\t\t\tProvider: HistoryArchiverProvider{\n\t\t\t\tFilestoreConfig: defaultFilestoreConfig(t),\n\t\t\t},\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidDisabledHistoryArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tHistory: HistoryArchival{\n\t\t\tEnableRead: true,\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.Error(t, err)\n}\n\nfunc TestValidEmptyHistoryArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tHistory: HistoryArchival{},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.NoError(t, err)\n}\n\n// Visibility archival\n\nfunc TestValidEnabledVisibilityArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tVisibility: VisibilityArchival{\n\t\t\tStatus: constants.ArchivalEnabled,\n\t\t\tProvider: VisibilityArchiverProvider{\n\t\t\t\tFilestoreConfig: defaultFilestoreConfig(t),\n\t\t\t},\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{\n\t\tVisibility: VisibilityArchivalDomainDefaults{\n\t\t\tURI: \"/var/tmp\",\n\t\t},\n\t})\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidHEnabledVisibilityArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tVisibility: VisibilityArchival{\n\t\t\tStatus: constants.ArchivalEnabled,\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.Error(t, err)\n}\n\nfunc TestValidDisabledVisibilityArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tVisibility: VisibilityArchival{\n\t\t\tProvider: VisibilityArchiverProvider{\n\t\t\t\tFilestoreConfig: defaultFilestoreConfig(t),\n\t\t\t},\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidDisabledVisibilityArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tVisibility: VisibilityArchival{\n\t\t\tEnableRead: true,\n\t\t},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.Error(t, err)\n}\n\nfunc TestValidEmptyVisibilityArchivalConfig(t *testing.T) {\n\tarchival := Archival{\n\t\tVisibility: VisibilityArchival{},\n\t}\n\terr := archival.Validate(&ArchivalDomainDefaults{})\n\trequire.NoError(t, err)\n}\n"
  },
  {
    "path": "common/config/authorization.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/golang-jwt/jwt/v5\"\n)\n\ntype (\n\tAuthorization struct {\n\t\tOAuthAuthorizer OAuthAuthorizer `yaml:\"oauthAuthorizer\"`\n\t\tNoopAuthorizer  NoopAuthorizer  `yaml:\"noopAuthorizer\"`\n\t}\n\n\tNoopAuthorizer struct {\n\t\tEnable bool `yaml:\"enable\"`\n\t}\n\n\tOAuthAuthorizer struct {\n\t\tEnable bool `yaml:\"enable\"`\n\t\t// Max of TTL in the claim\n\t\tMaxJwtTTL int64 `yaml:\"maxJwtTTL\"`\n\t\t// Credentials to verify/create the JWT using public/private keys\n\t\tJwtCredentials *JwtCredentials `yaml:\"jwtCredentials\"`\n\t\t// Provider\n\t\tProvider *OAuthProvider `yaml:\"provider\"`\n\t}\n\n\tJwtCredentials struct {\n\t\t// support: RS256 (RSA using SHA256)\n\t\tAlgorithm string `yaml:\"algorithm\"`\n\t\t// Public Key Path for verifying JWT token passed in from external clients\n\t\tPublicKey string `yaml:\"publicKey\"`\n\t}\n\n\t// OAuthProvider is used to validate tokens provided by 3rd party Identity Provider service\n\tOAuthProvider struct {\n\t\tJWKSURL             string `yaml:\"jwksURL\"`\n\t\tGroupsAttributePath string `yaml:\"groupsAttributePath\"`\n\t\tAdminAttributePath  string `yaml:\"adminAttributePath\"`\n\t}\n)\n\n// Validate validates the persistence config\nfunc (a *Authorization) Validate() error {\n\tif a.OAuthAuthorizer.Enable && a.NoopAuthorizer.Enable {\n\t\treturn fmt.Errorf(\"[AuthorizationConfig] More than one authorizer is enabled\")\n\t}\n\n\tif a.OAuthAuthorizer.Enable {\n\t\tif err := a.validateOAuth(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (a *Authorization) validateOAuth() error {\n\toauthConfig := a.OAuthAuthorizer\n\n\tif oauthConfig.MaxJwtTTL <= 0 {\n\t\treturn fmt.Errorf(\"[OAuthConfig] MaxTTL must be greater than 0\")\n\t}\n\n\tif oauthConfig.JwtCredentials == nil && oauthConfig.Provider == nil {\n\t\treturn errors.New(\"jwtCredentials or provider must be provided\")\n\t}\n\n\tif oauthConfig.JwtCredentials != nil {\n\t\tif oauthConfig.JwtCredentials.PublicKey == \"\" {\n\t\t\treturn fmt.Errorf(\"[OAuthConfig] PublicKey can't be empty\")\n\t\t}\n\n\t\tif oauthConfig.JwtCredentials.Algorithm != jwt.SigningMethodRS256.Name {\n\t\t\treturn fmt.Errorf(\"[OAuthConfig] The only supported Algorithm is RS256\")\n\t\t}\n\t}\n\n\tif oauthConfig.Provider != nil {\n\t\tif oauthConfig.Provider.JWKSURL == \"\" {\n\t\t\treturn fmt.Errorf(\"[OAuthConfig] JWKSURL is not set\")\n\t\t}\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/config/authorization_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestMultipleAuthEnabled(t *testing.T) {\n\tcfg := Authorization{\n\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\tEnable: true,\n\t\t},\n\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\tEnable: true,\n\t\t},\n\t}\n\n\terr := cfg.Validate()\n\tassert.EqualError(t, err, \"[AuthorizationConfig] More than one authorizer is enabled\")\n}\n\nfunc TestTTLIsZero(t *testing.T) {\n\tcfg := Authorization{\n\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\tEnable:         true,\n\t\t\tJwtCredentials: &JwtCredentials{},\n\t\t\tMaxJwtTTL:      0,\n\t\t},\n\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t}\n\n\terr := cfg.Validate()\n\tassert.EqualError(t, err, \"[OAuthConfig] MaxTTL must be greater than 0\")\n}\n\nfunc TestPublicKeyIsEmpty(t *testing.T) {\n\tcfg := Authorization{\n\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\tEnable: true,\n\t\t\tJwtCredentials: &JwtCredentials{\n\t\t\t\tAlgorithm: \"\",\n\t\t\t\tPublicKey: \"\",\n\t\t\t},\n\t\t\tMaxJwtTTL: 1000000,\n\t\t},\n\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t}\n\n\terr := cfg.Validate()\n\tassert.EqualError(t, err, \"[OAuthConfig] PublicKey can't be empty\")\n}\n\nfunc TestAlgorithmIsInvalid(t *testing.T) {\n\tcfg := Authorization{\n\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\tEnable: true,\n\t\t\tJwtCredentials: &JwtCredentials{\n\t\t\t\tAlgorithm: \"SHA256\",\n\t\t\t\tPublicKey: \"public\",\n\t\t\t},\n\t\t\tMaxJwtTTL: 1000000,\n\t\t},\n\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t}\n\n\terr := cfg.Validate()\n\tassert.EqualError(t, err, \"[OAuthConfig] The only supported Algorithm is RS256\")\n}\n\nfunc TestCorrectValidation(t *testing.T) {\n\tcfg := Authorization{\n\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\tEnable: true,\n\t\t\tJwtCredentials: &JwtCredentials{\n\t\t\t\tAlgorithm: \"RS256\",\n\t\t\t\tPublicKey: \"public\",\n\t\t\t},\n\t\t\tMaxJwtTTL: 1000000,\n\t\t},\n\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t}\n\n\terr := cfg.Validate()\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "common/config/cluster.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"log\"\n\n\t\"go.uber.org/multierr\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n)\n\ntype (\n\t// ClusterGroupMetadata contains all the clusters participating in a replication group(aka XDC/GlobalDomain)\n\tClusterGroupMetadata struct {\n\t\t// FailoverVersionIncrement is the increment of each cluster version when failover happens\n\t\t// It decides the maximum number clusters in this replication groups\n\t\tFailoverVersionIncrement int64 `yaml:\"failoverVersionIncrement\"`\n\t\t// PrimaryClusterName is the primary cluster name, only the primary cluster can register / update domain\n\t\t// all clusters can do domain failover\n\t\tPrimaryClusterName string `yaml:\"primaryClusterName\"`\n\t\t// Deprecated: please use PrimaryClusterName\n\t\tMasterClusterName string `yaml:\"masterClusterName\"`\n\t\t// CurrentClusterName is the name of the cluster of current deployment\n\t\tCurrentClusterName string `yaml:\"currentClusterName\"`\n\t\t// ClusterRedirectionPolicy contains the cluster redirection policy for global domains\n\t\tClusterRedirectionPolicy *ClusterRedirectionPolicy `yaml:\"clusterRedirectionPolicy\"`\n\t\t// ClusterGroup contains information for each cluster within the replication group\n\t\t// Key is the clusterName\n\t\tClusterGroup map[string]ClusterInformation `yaml:\"clusterGroup\"`\n\t\t// Deprecated: please use ClusterGroup\n\t\tClusterInformation map[string]ClusterInformation `yaml:\"clusterInformation\"`\n\t}\n\n\t// ClusterInformation contains the information about each cluster participating in cross DC\n\tClusterInformation struct {\n\t\tEnabled bool `yaml:\"enabled\"`\n\t\t// InitialFailoverVersion is the identifier of each cluster. 0 <= the value < failoverVersionIncrement\n\t\tInitialFailoverVersion int64 `yaml:\"initialFailoverVersion\"`\n\t\t// NewInitialFailoverVersion is a new failover version for an initialFailoverVersion migration\n\t\t// for when it's necessary to migrate between two values.\n\t\t//\n\t\t// this is a pointer to imply optionality, it's an optional field and its lack\n\t\t// is indicated by a nil pointer. Zero is a valid field\n\t\tNewInitialFailoverVersion *int64 `yaml:\"newInitialFailoverVersion\"`\n\t\t// RPCName indicate the remote service name\n\t\tRPCName string `yaml:\"rpcName\"`\n\t\t// Address indicate the remote service address(Host:Port). Host can be DNS name.\n\t\t// For currentCluster, it's usually the same as publicClient.hostPort\n\t\tRPCAddress string `yaml:\"rpcAddress\" validate:\"nonzero\"`\n\t\t// RPCTransport specifies transport to use for replication traffic.\n\t\t// Allowed values: tchannel|grpc\n\t\t// Default: tchannel\n\t\tRPCTransport string `yaml:\"rpcTransport\"`\n\t\t// AuthorizationProvider contains the information to authorize the cluster\n\t\tAuthorizationProvider AuthorizationProvider `yaml:\"authorizationProvider\"`\n\t\t// TLS configures client TLS/SSL authentication for connections to this cluster\n\t\tTLS TLS `yaml:\"tls\"`\n\t}\n\n\tAuthorizationProvider struct {\n\t\t// Enable indicates if the auth provider is enabled\n\t\tEnable bool `yaml:\"enable\"`\n\t\t// Type auth provider type\n\t\tType string `yaml:\"type\"` // only supports OAuthAuthorization\n\t\t// PrivateKey is the private key path\n\t\tPrivateKey string `yaml:\"privateKey\"`\n\t}\n)\n\n// Validate validates ClusterGroupMetadata\nfunc (m *ClusterGroupMetadata) Validate() error {\n\tif m == nil {\n\t\treturn errors.New(\"ClusterGroupMetadata cannot be empty\")\n\t}\n\n\tvar errs error\n\n\tif len(m.PrimaryClusterName) == 0 {\n\t\terrs = multierr.Append(errs, errors.New(\"primary cluster name is empty\"))\n\t}\n\tif len(m.CurrentClusterName) == 0 {\n\t\terrs = multierr.Append(errs, errors.New(\"current cluster name is empty\"))\n\t}\n\tif m.FailoverVersionIncrement == 0 {\n\t\terrs = multierr.Append(errs, errors.New(\"version increment is 0\"))\n\t}\n\tif len(m.ClusterGroup) == 0 {\n\t\terrs = multierr.Append(errs, errors.New(\"empty cluster group\"))\n\t}\n\tif _, ok := m.ClusterGroup[m.PrimaryClusterName]; len(m.PrimaryClusterName) > 0 && !ok {\n\t\terrs = multierr.Append(errs, errors.New(\"primary cluster is not specified in the cluster group\"))\n\t}\n\tif _, ok := m.ClusterGroup[m.CurrentClusterName]; len(m.CurrentClusterName) > 0 && !ok {\n\t\terrs = multierr.Append(errs, errors.New(\"current cluster is not specified in the cluster group\"))\n\t}\n\n\tversionToClusterName := make(map[int64]string)\n\tfor clusterName, info := range m.ClusterGroup {\n\t\tif len(clusterName) == 0 {\n\t\t\terrs = multierr.Append(errs, errors.New(\"cluster with empty name defined\"))\n\t\t}\n\t\tversionToClusterName[info.InitialFailoverVersion] = clusterName\n\n\t\tif m.FailoverVersionIncrement <= info.InitialFailoverVersion || info.InitialFailoverVersion < 0 {\n\t\t\terrs = multierr.Append(errs, fmt.Errorf(\n\t\t\t\t\"cluster %s: version increment %v is smaller than initial version: %v\",\n\t\t\t\tclusterName,\n\t\t\t\tm.FailoverVersionIncrement,\n\t\t\t\tinfo.InitialFailoverVersion,\n\t\t\t))\n\t\t}\n\n\t\tif info.Enabled && (len(info.RPCName) == 0 || len(info.RPCAddress) == 0) {\n\t\t\terrs = multierr.Append(errs, fmt.Errorf(\"cluster %v: rpc name / address is empty\", clusterName))\n\t\t}\n\t\tif info.RPCTransport != tchannel.TransportName && info.RPCTransport != grpc.TransportName {\n\t\t\terrs = multierr.Append(errs, fmt.Errorf(\"cluster %v: rpc transport must %v or %v\",\n\t\t\t\tclusterName, tchannel.TransportName, grpc.TransportName))\n\t\t}\n\t}\n\tif len(versionToClusterName) != len(m.ClusterGroup) {\n\t\terrs = multierr.Append(errs, errors.New(\"initial versions of the cluster group have duplicates\"))\n\t}\n\n\treturn errs\n}\n\n// FillDefaults populates default values for unspecified fields\nfunc (m *ClusterGroupMetadata) FillDefaults() {\n\tif m == nil {\n\t\tlog.Fatal(\"ClusterGroupMetadata cannot be empty\")\n\t}\n\n\t// TODO: remove this after 0.23 and mention a breaking change in config.\n\tif len(m.PrimaryClusterName) == 0 && len(m.MasterClusterName) != 0 {\n\t\tm.PrimaryClusterName = m.MasterClusterName\n\t\tlog.Println(\"[WARN] masterClusterName config is deprecated. Please replace it with primaryClusterName.\")\n\t}\n\n\t// TODO: remove this after 0.23 and mention a breaking change in config.\n\tif len(m.ClusterGroup) == 0 && len(m.ClusterInformation) != 0 {\n\t\tm.ClusterGroup = m.ClusterInformation\n\t\tlog.Println(\"[WARN] clusterInformation config is deprecated. Please replace it with clusterGroup.\")\n\t}\n\n\tfor name, cluster := range m.ClusterGroup {\n\t\tif cluster.RPCName == \"\" {\n\t\t\t// filling RPCName with a default value if empty\n\t\t\tcluster.RPCName = \"cadence-frontend\"\n\t\t}\n\t\tif cluster.RPCTransport == \"\" {\n\t\t\tcluster.RPCTransport = tchannel.TransportName\n\t\t}\n\t\tm.ClusterGroup[name] = cluster\n\t}\n}\n"
  },
  {
    "path": "common/config/cluster_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestClusterGroupMetadataDefaults(t *testing.T) {\n\tconfig := ClusterGroupMetadata{\n\t\tMasterClusterName: \"active\",\n\t\tClusterInformation: map[string]ClusterInformation{\n\t\t\t\"active\": {},\n\t\t},\n\t}\n\n\tconfig.FillDefaults()\n\n\tassert.Equal(t, \"active\", config.PrimaryClusterName)\n\tassert.Equal(t, \"cadence-frontend\", config.ClusterGroup[\"active\"].RPCName)\n\tassert.Equal(t, \"tchannel\", config.ClusterGroup[\"active\"].RPCTransport)\n}\n\nfunc TestClusterGroupMetadataValidate(t *testing.T) {\n\tmodify := func(initial *ClusterGroupMetadata, modify func(metadata *ClusterGroupMetadata)) *ClusterGroupMetadata {\n\t\tmodify(initial)\n\t\treturn initial\n\t}\n\n\ttests := []struct {\n\t\tmsg    string\n\t\tconfig *ClusterGroupMetadata\n\t\terr    string\n\t}{\n\t\t{\n\t\t\tmsg:    \"valid\",\n\t\t\tconfig: validClusterGroupMetadata(),\n\t\t},\n\t\t{\n\t\t\tmsg:    \"empty\",\n\t\t\tconfig: nil,\n\t\t\terr:    \"ClusterGroupMetadata cannot be empty\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"primary cluster name is empty\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.PrimaryClusterName = \"\"\n\t\t\t}),\n\t\t\terr: \"primary cluster name is empty\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"current cluster name is empty\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.CurrentClusterName = \"\"\n\t\t\t}),\n\t\t\terr: \"current cluster name is empty\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"primary cluster is not specified in the cluster group\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.PrimaryClusterName = \"non-existing\"\n\t\t\t}),\n\t\t\terr: \"primary cluster is not specified in the cluster group\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"current cluster is not specified in the cluster group\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.CurrentClusterName = \"non-existing\"\n\t\t\t}),\n\t\t\terr: \"current cluster is not specified in the cluster group\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"version increment is 0\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.FailoverVersionIncrement = 0\n\t\t\t}),\n\t\t\terr: \"version increment is 0\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"empty cluster group\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.ClusterGroup = nil\n\t\t\t}),\n\t\t\terr: \"empty cluster group\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"cluster with empty name defined\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.ClusterGroup[\"\"] = ClusterInformation{}\n\t\t\t}),\n\t\t\terr: \"cluster with empty name defined\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"increment smaller than initial version\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tm.FailoverVersionIncrement = 1\n\t\t\t}),\n\t\t\terr: \"cluster standby: version increment 1 is smaller than initial version: 2\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"empty rpc name\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tactive := m.ClusterGroup[\"active\"]\n\t\t\t\tactive.RPCName = \"\"\n\t\t\t\tm.ClusterGroup[\"active\"] = active\n\t\t\t}),\n\t\t\terr: \"cluster active: rpc name / address is empty\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"empty rpc address\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tactive := m.ClusterGroup[\"active\"]\n\t\t\t\tactive.RPCAddress = \"\"\n\t\t\t\tm.ClusterGroup[\"active\"] = active\n\t\t\t}),\n\t\t\terr: \"cluster active: rpc name / address is empty\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"invalid rpc transport\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tactive := m.ClusterGroup[\"active\"]\n\t\t\t\tactive.RPCTransport = \"invalid\"\n\t\t\t\tm.ClusterGroup[\"active\"] = active\n\t\t\t}),\n\t\t\terr: \"cluster active: rpc transport must tchannel or grpc\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"initial version duplicated\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\tstandby := m.ClusterGroup[\"standby\"]\n\t\t\t\tstandby.InitialFailoverVersion = 0\n\t\t\t\tm.ClusterGroup[\"standby\"] = standby\n\t\t\t}),\n\t\t\terr: \"initial versions of the cluster group have duplicates\",\n\t\t},\n\t\t{\n\t\t\tmsg: \"multiple errors\",\n\t\t\tconfig: modify(validClusterGroupMetadata(), func(m *ClusterGroupMetadata) {\n\t\t\t\t*m = ClusterGroupMetadata{}\n\t\t\t}),\n\t\t\terr: \"primary cluster name is empty; current cluster name is empty; version increment is 0; empty cluster group\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.msg, func(t *testing.T) {\n\t\t\terr := tt.config.Validate()\n\t\t\tif tt.err == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.Contains(t, err.Error(), tt.err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc validClusterGroupMetadata() *ClusterGroupMetadata {\n\treturn &ClusterGroupMetadata{\n\t\tFailoverVersionIncrement: 10,\n\t\tPrimaryClusterName:       \"active\",\n\t\tCurrentClusterName:       \"standby\",\n\t\tClusterGroup: map[string]ClusterInformation{\n\t\t\t\"active\": {\n\t\t\t\tEnabled:                true,\n\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\tRPCName:                \"cadence-frontend\",\n\t\t\t\tRPCAddress:             \"localhost:7833\",\n\t\t\t\tRPCTransport:           \"grpc\",\n\t\t\t},\n\t\t\t\"standby\": {\n\t\t\t\tEnabled:                true,\n\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\tRPCName:                \"cadence-frontend\",\n\t\t\t\tRPCAddress:             \"localhost:7833\",\n\t\t\t\tRPCTransport:           \"grpc\",\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/config/config.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"log\"\n\t\"regexp\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally/m3\"\n\t\"github.com/uber-go/tally/prometheus\"\n\tyarpctls \"go.uber.org/yarpc/api/transport/tls\"\n\t\"gopkg.in/yaml.v2\" // CAUTION: go.uber.org/config does not support yaml.v3\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\tc \"github.com/uber/cadence/common/dynamicconfig/configstore/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tringpopprovider \"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\tsdconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\ntype (\n\t// Config contains the configuration for a set of cadence services\n\tConfig struct {\n\t\t// Ringpop is the ringpop related configuration\n\t\tRingpop ringpopprovider.Config `yaml:\"ringpop\"`\n\t\t// Membership is used to configure peer provider plugin\n\t\tMembership Membership `yaml:\"membership\"`\n\t\t// Persistence contains the configuration for cadence datastores\n\t\tPersistence Persistence `yaml:\"persistence\"`\n\t\t// Log is the logging config\n\t\tLog Logger `yaml:\"log\"`\n\t\t// ClusterGroupMetadata is the config containing all valid clusters and active cluster\n\t\tClusterGroupMetadata *ClusterGroupMetadata `yaml:\"clusterGroupMetadata\"`\n\t\t// Deprecated: please use ClusterGroupMetadata\n\t\tClusterMetadata *ClusterGroupMetadata `yaml:\"clusterMetadata\"`\n\t\t// Deprecated: please use ClusterRedirectionPolicy under ClusterGroupMetadata\n\t\tDCRedirectionPolicy *ClusterRedirectionPolicy `yaml:\"dcRedirectionPolicy\"`\n\t\t// Services is a map of service name to service config items\n\t\tServices map[string]Service `yaml:\"services\"`\n\t\t// Kafka is the config for connecting to kafka\n\t\tKafka KafkaConfig `yaml:\"kafka\"`\n\t\t// Archival is the config for archival\n\t\tArchival Archival `yaml:\"archival\"`\n\t\t// PublicClient is config for sys worker service connecting to cadence frontend\n\t\tPublicClient PublicClient `yaml:\"publicClient\"`\n\t\t// DynamicConfigClient is the config for setting up the file based dynamic config client\n\t\t// Filepath would be relative to the root directory when the path wasn't absolute.\n\t\t// Included for backwards compatibility, please transition to DynamicConfig\n\t\t// If both are specified, DynamicConig will be used.\n\t\tDynamicConfigClient dynamicconfig.FileBasedClientConfig `yaml:\"dynamicConfigClient\"`\n\t\t// DynamicConfig is the config for setting up all dynamic config clients\n\t\t// Allows for changes in client without needing code change\n\t\tDynamicConfig DynamicConfig `yaml:\"dynamicconfig\"`\n\t\t// DomainDefaults is the default config for every domain\n\t\tDomainDefaults DomainDefaults `yaml:\"domainDefaults\"`\n\t\t// Blobstore is the config for setting up blobstore\n\t\tBlobstore Blobstore `yaml:\"blobstore\"`\n\t\t// Authorization is the config for setting up authorization\n\t\tAuthorization Authorization `yaml:\"authorization\"`\n\t\t// HeaderForwardingRules defines which inbound headers to include or exclude on outbound calls\n\t\tHeaderForwardingRules []HeaderRule `yaml:\"headerForwardingRules\"`\n\t\t// Note: This is not implemented yet. It's coming in the next release.\n\t\t// AsyncWorkflowQueues is the config for predefining async workflow queue(s)\n\t\t// To use Async APIs for a domain first specify the queue using Admin API.\n\t\t// Either refer to one of the predefined queues in this config or alternatively specify the queue details inline in the API call.\n\t\tAsyncWorkflowQueues map[string]AsyncWorkflowQueueProvider `yaml:\"asyncWorkflowQueues\"`\n\t\t// ShardDistributorClient is the config for shard distributor client\n\t\t// Shard distributor is used to distribute shards across multiple cadence service instances\n\t\t// Note: This is not recommended for use, it's still experimental\n\t\tShardDistributorClient ShardDistributorClient `yaml:\"shardDistributorClient\"`\n\n\t\t// ShardDistribution is a config for the shard distributor leader election component that allows to run a single process per region and manage shard namespaces.\n\t\tShardDistribution sdconfig.ShardDistribution `yaml:\"shardDistribution\"`\n\n\t\t// ShardDistributorMatchingConfig is the config for shard distributor executor client in matching service\n\t\tShardDistributorMatchingConfig clientcommon.Config `yaml:\"shard-distributor-matching\"`\n\n\t\t// Histograms controls timer vs histogram metric emission while they are being migrated.\n\t\t//\n\t\t// Timers will eventually be dropped, and this config will be validation-only (e.g. to error if any explicitly request timers).\n\t\tHistograms metrics.HistogramMigration `yaml:\"histograms\"`\n\t}\n\n\t// Membership holds peer provider configuration.\n\tMembership struct {\n\t\tProvider PeerProvider `yaml:\"provider\"`\n\t}\n\n\t// PeerProvider is provider config. Contents depends on plugin in use\n\tPeerProvider map[string]*YamlNode\n\n\tHeaderRule struct {\n\t\tAdd   bool // if false, matching headers are removed if previously matched.\n\t\tMatch *regexp.Regexp\n\t}\n\n\tDynamicConfig struct {\n\t\tClient      string                              `yaml:\"client\"`\n\t\tConfigStore c.ClientConfig                      `yaml:\"configstore\"`\n\t\tFileBased   dynamicconfig.FileBasedClientConfig `yaml:\"filebased\"`\n\t}\n\n\t// Service contains the service specific config items\n\tService struct {\n\t\t// TChannel is the tchannel configuration\n\t\tRPC RPC `yaml:\"rpc\"`\n\t\t// Metrics is the metrics subsystem configuration\n\t\tMetrics Metrics `yaml:\"metrics\"`\n\t\t// PProf is the PProf configuration\n\t\tPProf PProf `yaml:\"pprof\"`\n\t}\n\n\t// PProf contains the rpc config items\n\tPProf struct {\n\t\t// Port is the port on which the PProf will bind to\n\t\tPort int `yaml:\"port\"`\n\t\t// Host is the host on which the PProf will bind to, default to `localhost`\n\t\tHost string `yaml:\"host\"`\n\t}\n\n\t// RPC contains the rpc config items\n\tRPC struct {\n\t\t// Port is the port  on which the Thrift TChannel will bind to\n\t\tPort uint16 `yaml:\"port\"`\n\t\t// GRPCPort is the port on which the grpc listener will bind to\n\t\tGRPCPort uint16 `yaml:\"grpcPort\"`\n\t\t// BindOnLocalHost is true if localhost is the bind address\n\t\tBindOnLocalHost bool `yaml:\"bindOnLocalHost\"`\n\t\t// BindOnIP can be used to bind service on specific ip (eg. `0.0.0.0`) -\n\t\t// check net.ParseIP for supported syntax, only IPv4 is supported,\n\t\t// mutually exclusive with `BindOnLocalHost` option\n\t\tBindOnIP string `yaml:\"bindOnIP\"`\n\t\t// DisableLogging disables all logging for rpc\n\t\tDisableLogging bool `yaml:\"disableLogging\"`\n\t\t// LogLevel is the desired log level\n\t\tLogLevel string `yaml:\"logLevel\"`\n\t\t// GRPCMaxMsgSize allows overriding default (4MB) message size for gRPC\n\t\tGRPCMaxMsgSize int `yaml:\"grpcMaxMsgSize\"`\n\t\t// TLS allows configuring optional TLS/SSL authentication on the server (only on gRPC port)\n\t\tTLS TLS `yaml:\"tls\"`\n\t\t// HTTP keeps configuration for exposed HTTP API\n\t\tHTTP *HTTP `yaml:\"http\"`\n\t}\n\n\t// HTTP API configuration\n\tHTTP struct {\n\t\t// Port for listening HTTP requests\n\t\tPort uint16 `yaml:\"port\"`\n\t\t// List of RPC procedures available to call using HTTP\n\t\tProcedures []string `yaml:\"procedures\"`\n\t\t// TLS allows configuring TLS/SSL for HTTP requests\n\t\tTLS TLS `yaml:\"tls\"`\n\t\t// Mode represents the TLS mode of the transport.\n\t\t// Available modes: disabled, permissive, enforced\n\t\tTLSMode yarpctls.Mode `yaml:\"TLSMode\"`\n\t}\n\n\t// Blobstore contains the config for blobstore\n\tBlobstore struct {\n\t\tFilestore *FileBlobstore `yaml:\"filestore\"`\n\t}\n\n\t// FileBlobstore contains the config for a file backed blobstore\n\tFileBlobstore struct {\n\t\tOutputDirectory string `yaml:\"outputDirectory\"`\n\t}\n\n\t// Persistence contains the configuration for data store / persistence layer\n\tPersistence struct {\n\t\t// DefaultStore is the name of the default data store to use\n\t\tDefaultStore string `yaml:\"defaultStore\" validate:\"nonzero\"`\n\t\t// VisibilityStore is the name of the datastore to be used for visibility records\n\t\t// Must provide one of VisibilityStore and AdvancedVisibilityStore\n\t\tVisibilityStore string `yaml:\"visibilityStore\"`\n\t\t// AdvancedVisibilityStore is the name of the datastore to be used for visibility records\n\t\t// Must provide one of VisibilityStore and AdvancedVisibilityStore\n\t\tAdvancedVisibilityStore string `yaml:\"advancedVisibilityStore\"`\n\t\t// HistoryMaxConns is the desired number of conns to history store. Value specified\n\t\t// here overrides the MaxConns config specified as part of datastore\n\t\t// Deprecated: This value is not used\n\t\tHistoryMaxConns int `yaml:\"historyMaxConns\"`\n\t\t// EnablePersistenceLatencyHistogramMetrics is to enable latency histogram metrics for persistence layer\n\t\tEnablePersistenceLatencyHistogramMetrics bool `yaml:\"enablePersistenceLatencyHistogramMetrics\"`\n\t\t// NumHistoryShards is the desired number of history shards. It's for computing the historyShardID from workflowID into [0, NumHistoryShards)\n\t\t// Therefore, the value cannot be changed once set.\n\t\t// TODO This config doesn't belong here, needs refactoring\n\t\tNumHistoryShards int `yaml:\"numHistoryShards\" validate:\"nonzero\"`\n\t\t// DataStores contains the configuration for all datastores\n\t\tDataStores map[string]DataStore `yaml:\"datastores\"`\n\t\t// TODO: move dynamic config out of static config\n\t\t// TransactionSizeLimit is the largest allowed transaction size\n\t\tTransactionSizeLimit dynamicproperties.IntPropertyFn `yaml:\"-\" json:\"-\"`\n\t\t// TODO: move dynamic config out of static config\n\t\t// ErrorInjectionRate is the the rate for injecting random error\n\t\tErrorInjectionRate dynamicproperties.FloatPropertyFn `yaml:\"-\" json:\"-\"`\n\t}\n\n\t// DataStore is the configuration for a single datastore\n\tDataStore struct {\n\t\t// Cassandra contains the config for a cassandra datastore\n\t\t// Deprecated: please use NoSQL instead, the structure is backward-compatible\n\t\tCassandra *Cassandra `yaml:\"cassandra\"`\n\t\t// SQL contains the config for a SQL based datastore\n\t\tSQL *SQL `yaml:\"sql\"`\n\t\t// NoSQL contains the config for a NoSQL based datastore\n\t\tNoSQL *NoSQL `yaml:\"nosql\"`\n\t\t// ShardedNoSQL contains the config for a collection of NoSQL datastores that are used as a single datastore\n\t\tShardedNoSQL *ShardedNoSQL `yaml:\"shardedNosql\"`\n\t\t// ElasticSearch contains the config for a ElasticSearch datastore\n\t\tElasticSearch *ElasticSearchConfig `yaml:\"elasticsearch\"`\n\t\t// Pinot contains the config for a Pinot datastore\n\t\tPinot *PinotVisibilityConfig `yaml:\"pinot\"`\n\t}\n\n\t// Cassandra contains configuration to connect to Cassandra cluster\n\t// Deprecated: please use NoSQL instead, the structure is backward-compatible\n\tCassandra = NoSQL\n\n\t// NoSQL contains configuration to connect to NoSQL Database cluster\n\tNoSQL struct {\n\t\t// PluginName is the name of NoSQL plugin, default is \"cassandra\". Supported values: cassandra\n\t\tPluginName string `yaml:\"pluginName\"`\n\t\t// Hosts is a csv of cassandra endpoints\n\t\tHosts string `yaml:\"hosts\" validate:\"nonzero\"`\n\t\t// Port is the cassandra port used for connection by gocql client\n\t\tPort int `yaml:\"port\"`\n\t\t// User is the cassandra user used for authentication by gocql client\n\t\tUser string `yaml:\"user\"`\n\t\t// Password is the cassandra password used for authentication by gocql client\n\t\tPassword string `yaml:\"password\"`\n\t\t// AllowedAuthenticators informs the cassandra client to expect a custom authenticator\n\t\tAllowedAuthenticators []string `yaml:\"allowedAuthenticators\"`\n\t\t// Keyspace is the cassandra keyspace\n\t\tKeyspace string `yaml:\"keyspace\"`\n\t\t// Region is the region filter arg for cassandra\n\t\tRegion string `yaml:\"region\"`\n\t\t// Datacenter is the data center filter arg for cassandra\n\t\tDatacenter string `yaml:\"datacenter\"`\n\t\t// MaxConns is the max number of connections to this datastore for a single keyspace\n\t\tMaxConns int `yaml:\"maxConns\"`\n\t\t// TLS configuration\n\t\tTLS *TLS `yaml:\"tls\"`\n\t\t// ProtoVersion\n\t\tProtoVersion int `yaml:\"protoVersion\"`\n\t\t// ConnectTimeout defines duration for initial dial\n\t\tConnectTimeout time.Duration `yaml:\"connectTimeout\"`\n\t\t// Timout is a connection timeout\n\t\tTimeout time.Duration `yaml:\"timeout\"`\n\t\t// Consistency defines default consistency level\n\t\tConsistency string `yaml:\"consistency\"`\n\t\t// SerialConsistency sets the consistency for the serial part of queries\n\t\tSerialConsistency string `yaml:\"serialConsistency\"`\n\t\t// ConnectAttributes is a set of key-value attributes as a supplement/extension to the above common fields\n\t\t// Use it ONLY when a configure is too specific to a particular NoSQL database that should not be in the common struct\n\t\t// Otherwise please add new fields to the struct for better documentation\n\t\t// If being used in any database, update this comment here to make it clear\n\t\tConnectAttributes map[string]string `yaml:\"connectAttributes\"`\n\t\t// HostSelectionPolicy sets gocql policy for selecting host for a query\n\t\t// Available selections are: \"tokenaware,roundrobin\", \"hostpool-epsilon-greedy\", \"roundrobin\"\n\t\tHostSelectionPolicy string `yaml:\"hostSelectionPolicy\"`\n\t}\n\n\t// ShardedNoSQL contains configuration to connect to a set of NoSQL Database clusters in a sharded manner\n\tShardedNoSQL struct {\n\t\t// DefaultShard is the DB shard where the non-sharded tables (ie. cluster metadata) are stored\n\t\tDefaultShard string `yaml:\"defaultShard\"`\n\t\t// ShardingPolicy is the configuration for the sharding strategy used\n\t\tShardingPolicy ShardingPolicy `yaml:\"shardingPolicy\"`\n\t\t// Connections is the collection of NoSQL DB plugins that are used to connect to the shard\n\t\tConnections map[string]DBShardConnection `yaml:\"connections\"`\n\t}\n\n\t// ShardingPolicy contains configuration for physical DB sharding\n\tShardingPolicy struct {\n\t\t// HistoryShardMapping defines the ranges of history shards stored by each DB shard. Ranges listed here *MUST*\n\t\t// be continuous and non-overlapping, such that the first range in the list starts from Shard 0, each following\n\t\t// range starts with <prevRange.End> + 1, and the last range ends with <NumHistoryHosts>-1.\n\t\tHistoryShardMapping []HistoryShardRange `yaml:\"historyShardMapping\"`\n\t\t// TaskListHashing defines the parameters needed for shard ownership calculation based on hashing\n\t\tTaskListHashing TasklistHashing `yaml:\"taskListHashing\"`\n\t}\n\n\t// HistoryShardRange contains configuration for one NoSQL DB Shard\n\tHistoryShardRange struct {\n\t\t// Start defines the inclusive lower bound for the history shard range\n\t\tStart int `yaml:\"start\"`\n\t\t// End defines the exclusive upper bound for the history shard range\n\t\tEnd int `yaml:\"end\"`\n\t\t// Shard defines the shard that owns this range\n\t\tShard string `yaml:\"shard\"`\n\t}\n\n\tTasklistHashing struct {\n\t\t// ShardOrder defines the order of shards to be used when hashing tasklists to shards\n\t\tShardOrder []string `yaml:\"shardOrder\"`\n\t}\n\n\t// DBShardConnection contains configuration for one NoSQL DB Shard\n\tDBShardConnection struct {\n\t\t// NoSQLPlugin is the NoSQL plugin used for connecting to the DB shard\n\t\tNoSQLPlugin *NoSQL `yaml:\"nosqlPlugin\"`\n\t}\n\n\t// SQL is the configuration for connecting to a SQL backed datastore\n\tSQL struct {\n\t\t// User is the username to be used for the conn\n\t\t// If useMultipleDatabases, must be empty and provide it via multipleDatabasesConfig instead\n\t\tUser string `yaml:\"user\"`\n\t\t// Password is the password corresponding to the user name\n\t\t// If useMultipleDatabases, must be empty and provide it via multipleDatabasesConfig instead\n\t\tPassword string `yaml:\"password\"`\n\t\t// PluginName is the name of SQL plugin\n\t\tPluginName string `yaml:\"pluginName\" validate:\"nonzero\"`\n\t\t// DatabaseName is the name of SQL database to connect to\n\t\t// If useMultipleDatabases, must be empty and provide it via multipleDatabasesConfig instead\n\t\t// Required if not useMultipleDatabases\n\t\tDatabaseName string `yaml:\"databaseName\"`\n\t\t// ConnectAddr is the remote addr of the database\n\t\t// If useMultipleDatabases, must be empty and provide it via multipleDatabasesConfig instead\n\t\t// Required if not useMultipleDatabases\n\t\tConnectAddr string `yaml:\"connectAddr\"`\n\t\t// ConnectProtocol is the protocol that goes with the ConnectAddr ex - tcp, unix\n\t\tConnectProtocol string `yaml:\"connectProtocol\"`\n\t\t// ConnectAttributes is a set of key-value attributes to be sent as part of connect data_source_name url\n\t\tConnectAttributes map[string]string `yaml:\"connectAttributes\"`\n\t\t// MaxConns the max number of connections to this datastore\n\t\tMaxConns int `yaml:\"maxConns\"`\n\t\t// MaxIdleConns is the max number of idle connections to this datastore\n\t\tMaxIdleConns int `yaml:\"maxIdleConns\"`\n\t\t// MaxConnLifetime is the maximum time a connection can be alive\n\t\tMaxConnLifetime time.Duration `yaml:\"maxConnLifetime\"`\n\t\t// NumShards is the number of DB shards in a sharded sql database. Default is 1 for single SQL database setup.\n\t\t// It's for computing a shardID value of [0,NumShards) to decide which shard of DB to query.\n\t\t// Relationship with NumHistoryShards, both values cannot be changed once set in the same cluster,\n\t\t// and the historyShardID value calculated from NumHistoryShards will be calculated using this NumShards to get a dbShardID\n\t\tNumShards int `yaml:\"nShards\"`\n\t\t// TLS is the configuration for TLS connections\n\t\tTLS *TLS `yaml:\"tls\"`\n\t\t// EncodingType is the configuration for the type of encoding used for sql blobs\n\t\tEncodingType string `yaml:\"encodingType\"`\n\t\t// DecodingTypes is the configuration for all the sql blob decoding types which need to be supported\n\t\t// DecodingTypes should not be removed unless there are no blobs in database with the encoding type\n\t\tDecodingTypes []string `yaml:\"decodingTypes\"`\n\t\t// UseMultipleDatabases enables using multiple databases as a sharding SQL database, default is false\n\t\t// When enabled, connection will be established using MultipleDatabasesConfig in favor of single values\n\t\t// of  User, Password, DatabaseName, ConnectAddr.\n\t\tUseMultipleDatabases bool `yaml:\"useMultipleDatabases\"`\n\t\t// Required when UseMultipleDatabases is true\n\t\t// the length of the list should be exactly the same as NumShards\n\t\tMultipleDatabasesConfig []MultipleDatabasesConfigEntry `yaml:\"multipleDatabasesConfig\"`\n\t}\n\n\t// MultipleDatabasesConfigEntry is an entry for MultipleDatabasesConfig to connect to a single SQL database\n\tMultipleDatabasesConfigEntry struct {\n\t\t// User is the username to be used for the conn\n\t\tUser string `yaml:\"user\"`\n\t\t// Password is the password corresponding to the user name\n\t\tPassword string `yaml:\"password\"`\n\t\t// DatabaseName is the name of SQL database to connect to\n\t\tDatabaseName string `yaml:\"databaseName\" validate:\"nonzero\"`\n\t\t// ConnectAddr is the remote addr of the database\n\t\tConnectAddr string `yaml:\"connectAddr\" validate:\"nonzero\"`\n\t}\n\n\t// CustomDatastoreConfig is the configuration for connecting to a custom datastore that is not supported by cadence core\n\t// TODO can we remove it?\n\tCustomDatastoreConfig struct {\n\t\t// Name of the custom datastore\n\t\tName string `yaml:\"name\"`\n\t\t// Options is a set of key-value attributes that can be used by AbstractDatastoreFactory implementation\n\t\tOptions map[string]string `yaml:\"options\"`\n\t}\n\n\t// Replicator describes the configuration of replicator\n\t// TODO can we remove it?\n\tReplicator struct{}\n\n\t// Logger contains the config items for logger\n\tLogger struct {\n\t\t// Stdout is true then the output needs to goto standard out\n\t\t// By default this is false and output will go to standard error\n\t\tStdout bool `yaml:\"stdout\"`\n\t\t// Level is the desired log level\n\t\tLevel string `yaml:\"level\"`\n\t\t// OutputFile is the path to the log output file\n\t\t// Stdout must be false, otherwise Stdout will take precedence\n\t\tOutputFile string `yaml:\"outputFile\"`\n\t\t// LevelKey is the desired log level, defaults to \"level\"\n\t\tLevelKey string `yaml:\"levelKey\"`\n\t\t// Encoding decides the format, supports \"console\" and \"json\".\n\t\t// \"json\" will print the log in JSON format(better for machine), while \"console\" will print in plain-text format(more human friendly)\n\t\t// Default is \"json\"\n\t\tEncoding string `yaml:\"encoding\"`\n\t}\n\n\t// ClusterRedirectionPolicy contains the frontend datacenter redirection policy\n\t// When using XDC (global domain) feature to failover a domain from one cluster to another one, client may call the passive cluster to start /signal workflows etc.\n\t// To have a seamless failover experience, cluster should use this forwarding option to forward those APIs to the active cluster.\n\tClusterRedirectionPolicy struct {\n\t\t// Support \"noop\", \"selected-apis-forwarding\" and \"all-domain-apis-forwarding\", default (when empty) is \"noop\"\n\t\t//\n\t\t// 1) \"noop\" will not do any forwarding.\n\t\t//\n\t\t// 2) \"all-domain-apis-forwarding\" will forward all domain specific APIs(worker and non worker) if the current active domain is\n\t\t// the same as \"allDomainApisForwardingTargetCluster\"( or \"allDomainApisForwardingTargetCluster\" is empty), otherwise it fallbacks to \"selected-apis-forwarding\".\n\t\t//\n\t\t// 3) \"selected-apis-forwarding\" will forward all non-worker APIs including\n\t\t// 1. StartWorkflowExecution\n\t\t// 2. SignalWithStartWorkflowExecution\n\t\t// 3. SignalWorkflowExecution\n\t\t// 4. RequestCancelWorkflowExecution\n\t\t// 5. TerminateWorkflowExecution\n\t\t// 6. QueryWorkflow\n\t\t// 7. ResetWorkflow\n\t\t//\n\t\t// 4) \"selected-apis-forwarding-v2\" will forward all of \"selected-apis-forwarding\", and also activity responses\n\t\t// and heartbeats, but not other worker APIs.\n\t\t//\n\t\t// \"selected-apis-forwarding(-v2)\" and \"all-domain-apis-forwarding\" can work with EnableDomainNotActiveAutoForwarding dynamicconfig to select certain domains using the policy.\n\t\t//\n\t\t// Usage recommendation: when enabling XDC(global domain) feature, either \"all-domain-apis-forwarding\" or \"selected-apis-forwarding(-v2)\" should be used to ensure seamless domain failover(high availability)\n\t\t// Depending on the cost of cross cluster calls:\n\t\t//\n\t\t// 1) If the network communication overhead is high(e.g., clusters are in remote datacenters of different region), then should use \"selected-apis-forwarding(-v2)\".\n\t\t// But you must ensure a different set of workers with the same workflow & activity code are connected to each Cadence cluster.\n\t\t//\n\t\t// 2) If the network communication overhead is low (e.g. in the same datacenter, mostly for cluster migration usage), then you can use \"all-domain-apis-forwarding\". Then only one set of\n\t\t// workflow & activity worker connected of one of the Cadence cluster is enough as all domain APIs are forwarded. See more details in documentation of cluster migration section.\n\t\t// Usually \"allDomainApisForwardingTargetCluster\" should be empty(default value) except for very rare cases: you have more than two clusters and some are in a remote region but some are in local region.\n\t\tPolicy string `yaml:\"policy\"`\n\t\t// A supplement for \"all-domain-apis-forwarding\" policy. It decides how the policy fallback to  \"selected-apis-forwarding\" policy.\n\t\t// If this is not empty, and current domain is not active in the value of allDomainApisForwardingTargetCluster, then the policy will fallback to \"selected-apis-forwarding\" policy.\n\t\t// Default is empty, meaning that all requests will not fallback.\n\t\tAllDomainApisForwardingTargetCluster string `yaml:\"allDomainApisForwardingTargetCluster\"`\n\t\t// Not being used, but we have to keep it so that config loading is not broken\n\t\tToDC string `yaml:\"toDC\"`\n\t}\n\n\t// Metrics contains the config items for metrics subsystem\n\tMetrics struct {\n\t\t// M3 is the configuration for m3 metrics reporter\n\t\tM3 *m3.Configuration `yaml:\"m3\"`\n\t\t// Statsd is the configuration for statsd reporter\n\t\tStatsd *Statsd `yaml:\"statsd\"`\n\t\t// Prometheus is the configuration for prometheus reporter\n\t\t// Some documentation below because the tally library is missing it:\n\t\t// In this configuration, default timerType is \"histogram\", alternatively \"summary\" is also supported.\n\t\t// In some cases, summary is better. Choose it wisely.\n\t\t// For histogram, default buckets are defined in https://github.com/uber/cadence/blob/master/common/metrics/tally/prometheus/buckets.go#L34\n\t\t// For summary, default objectives are defined in https://github.com/uber-go/tally/blob/137973e539cd3589f904c23d0b3a28c579fd0ae4/prometheus/reporter.go#L70\n\t\t// You can customize the buckets/objectives if the default is not good enough.\n\t\tPrometheus *prometheus.Configuration `yaml:\"prometheus\"`\n\t\t// Tags is the set of key-value pairs to be reported\n\t\t// as part of every metric\n\t\tTags map[string]string `yaml:\"tags\"`\n\t\t// Prefix sets the prefix to all outgoing metrics\n\t\tPrefix string `yaml:\"prefix\"`\n\t\t// ReportingInterval is the interval of metrics reporter\n\t\tReportingInterval time.Duration `yaml:\"reportingInterval\"` // defaults to 1s\n\t}\n\n\t// Statsd contains the config items for statsd metrics reporter\n\tStatsd struct {\n\t\t// The host and port of the statsd server\n\t\tHostPort string `yaml:\"hostPort\" validate:\"nonzero\"`\n\t\t// The prefix to use in reporting to statsd\n\t\tPrefix string `yaml:\"prefix\" validate:\"nonzero\"`\n\t\t// FlushInterval is the maximum interval for sending packets.\n\t\t// If it is not specified, it defaults to 1 second.\n\t\tFlushInterval time.Duration `yaml:\"flushInterval\"`\n\t\t// FlushBytes specifies the maximum udp packet size you wish to send.\n\t\t// If FlushBytes is unspecified, it defaults  to 1432 bytes, which is\n\t\t// considered safe for local traffic.\n\t\tFlushBytes int `yaml:\"flushBytes\"`\n\t}\n\n\t// Archival contains the config for archival\n\tArchival struct {\n\t\t// History is the config for the history archival\n\t\tHistory HistoryArchival `yaml:\"history\"`\n\t\t// Visibility is the config for visibility archival\n\t\tVisibility VisibilityArchival `yaml:\"visibility\"`\n\t}\n\n\t// HistoryArchival contains the config for history archival\n\tHistoryArchival struct {\n\t\t// Status is the status of history archival either: enabled, disabled, or paused\n\t\tStatus string `yaml:\"status\"`\n\t\t// EnableRead whether history can be read from archival\n\t\tEnableRead bool `yaml:\"enableRead\"`\n\t\t// Provider contains the config for all history archivers\n\t\tProvider HistoryArchiverProvider `yaml:\"provider\"`\n\t}\n\n\t// HistoryArchiverProvider contains the config for all history archivers.\n\t//\n\t// Because archivers support external plugins, so there is no fundamental structure expected,\n\t// but a top-level key per named store plugin is required, and will be used to select the\n\t// config for a plugin as it is initialized.\n\t//\n\t// Config keys and structures expected in the main default binary include:\n\t//  - FilestoreConfig: [*FilestoreArchiver], used with provider scheme [github.com/uber/cadence/common/archiver/filestore.URIScheme]\n\t//  - S3storeConfig: [*S3Archiver], used with provider scheme [github.com/uber/cadence/common/archiver/s3store.URIScheme]\n\t//  - \"gstorage\" via [github.com/uber/cadence/common/archiver/gcloud.ConfigKey]: [github.com/uber/cadence/common/archiver/gcloud.Config], used with provider scheme \"gs\" [github.com/uber/cadence/common/archiver/gcloud.URIScheme]\n\t//\n\t// For handling hardcoded config, see ToYamlNode.\n\tHistoryArchiverProvider map[string]*YamlNode\n\n\t// VisibilityArchival contains the config for visibility archival\n\tVisibilityArchival struct {\n\t\t// Status is the status of visibility archival either: enabled, disabled, or paused\n\t\tStatus string `yaml:\"status\"`\n\t\t// EnableRead whether visibility can be read from archival\n\t\tEnableRead bool `yaml:\"enableRead\"`\n\t\t// Provider contains the config for all visibility archivers\n\t\tProvider VisibilityArchiverProvider `yaml:\"provider\"`\n\t}\n\n\t// VisibilityArchiverProvider contains the config for all visibility archivers.\n\t//\n\t// Because archivers support external plugins, so there is no fundamental structure expected,\n\t// but a top-level key per named store plugin is required, and will be used to select the\n\t// config for a plugin as it is initialized.\n\t//\n\t// Config keys and structures expected in the main default binary include:\n\t//  - FilestoreConfig: [*FilestoreArchiver], used with provider scheme [github.com/uber/cadence/common/archiver/filestore.URIScheme]\n\t//  - S3storeConfig: [*S3Archiver], used with provider scheme [github.com/uber/cadence/common/archiver/s3store.URIScheme]\n\t//  - \"gstorage\" via [github.com/uber/cadence/common/archiver/gcloud.ConfigKey]: [github.com/uber/cadence/common/archiver/gcloud.Config], used with provider scheme \"gs\" [github.com/uber/cadence/common/archiver/gcloud.URIScheme]\n\t//\n\t// For handling hardcoded config, see ToYamlNode.\n\tVisibilityArchiverProvider map[string]*YamlNode\n\n\t// FilestoreArchiver contain the config for filestore archiver\n\tFilestoreArchiver struct {\n\t\tFileMode string `yaml:\"fileMode\"`\n\t\tDirMode  string `yaml:\"dirMode\"`\n\t}\n\n\t// S3Archiver contains the config for S3 archiver\n\tS3Archiver struct {\n\t\tRegion           string  `yaml:\"region\"`\n\t\tEndpoint         *string `yaml:\"endpoint\"`\n\t\tS3ForcePathStyle bool    `yaml:\"s3ForcePathStyle\"`\n\t}\n\n\t// PublicClient is config for connecting to cadence frontend\n\tPublicClient struct {\n\t\t// HostPort is the host port to connect on. Host can be DNS name\n\t\t// Default to currentCluster's RPCAddress in ClusterInformation\n\t\tHostPort string `yaml:\"hostPort\"`\n\t\t// Transport is the tranport to use when communicating using the SDK client.\n\t\t// Defaults to:\n\t\t// - currentCluster's RPCTransport in ClusterInformation (if HostPort is not provided)\n\t\t// - grpc (if HostPort is provided)\n\t\tTransport string `yaml:\"transport\"`\n\t\t// interval to refresh DNS. Default to 10s\n\t\tRefreshInterval time.Duration `yaml:\"RefreshInterval\"`\n\t}\n\n\t// DomainDefaults is the default config for each domain\n\tDomainDefaults struct {\n\t\t// Archival is the default archival config for each domain\n\t\tArchival ArchivalDomainDefaults `yaml:\"archival\"`\n\t}\n\n\t// ArchivalDomainDefaults is the default archival config for each domain\n\tArchivalDomainDefaults struct {\n\t\t// History is the domain default history archival config for each domain\n\t\tHistory HistoryArchivalDomainDefaults `yaml:\"history\"`\n\t\t// Visibility is the domain default visibility archival config for each domain\n\t\tVisibility VisibilityArchivalDomainDefaults `yaml:\"visibility\"`\n\t}\n\n\t// HistoryArchivalDomainDefaults is the default history archival config for each domain\n\tHistoryArchivalDomainDefaults struct {\n\t\t// Status is the domain default status of history archival: enabled or disabled\n\t\tStatus string `yaml:\"status\"`\n\t\t// URI is the domain default URI for history archiver\n\t\tURI string `yaml:\"URI\"`\n\t}\n\n\t// VisibilityArchivalDomainDefaults is the default visibility archival config for each domain\n\tVisibilityArchivalDomainDefaults struct {\n\t\t// Status is the domain default status of visibility archival: enabled or disabled\n\t\tStatus string `yaml:\"status\"`\n\t\t// URI is the domain default URI for visibility archiver\n\t\tURI string `yaml:\"URI\"`\n\t}\n\n\t// ShardDistributorClient contains the config items for shard distributor\n\tShardDistributorClient struct {\n\t\t// The host and port of the shard distributor server\n\t\tHostPort string `yaml:\"hostPort\"`\n\t}\n\n\t// YamlNode is a lazy-unmarshaler, because *yaml.Node only exists in gopkg.in/yaml.v3, not v2,\n\t// and go.uber.org/config currently uses only v2.\n\tYamlNode struct {\n\t\tunmarshal func(out any) error\n\t}\n\n\t// AsyncWorkflowQueueProvider contains the config for an async workflow queue.\n\t// Type is the implementation type of the queue provider.\n\t// Config is the configuration for the queue provider.\n\t// Config types and structures expected in the main default binary include:\n\t// - type: \"kafka\", config: [*github.com/uber/cadence/common/asyncworkflow/queue/kafka.QueueConfig]]]\n\tAsyncWorkflowQueueProvider struct {\n\t\tType   string    `yaml:\"type\"`\n\t\tConfig *YamlNode `yaml:\"config\"`\n\t}\n)\n\nconst (\n\t// NonShardedStoreName is the shard name used for singular (non-sharded) stores\n\tNonShardedStoreName = \"NonShardedStore\"\n\n\tFilestoreConfig = \"filestore\"\n\tS3storeConfig   = \"s3store\"\n)\n\nvar _ yaml.Unmarshaler = (*YamlNode)(nil)\n\nfunc (y *YamlNode) UnmarshalYAML(unmarshal func(interface{}) error) error {\n\ty.unmarshal = unmarshal\n\treturn nil\n}\n\nfunc (y *YamlNode) Decode(out any) error {\n\tif y == nil {\n\t\treturn nil\n\t}\n\treturn y.unmarshal(out)\n}\n\n// ToYamlNode is a bit of a hack to get a *yaml.Node for config-parsing compatibility purposes.\n// There is probably a better way to achieve this with yaml-loading compatibility, but this is at least fairly simple.\nfunc ToYamlNode(input any) (*YamlNode, error) {\n\tdata, err := yaml.Marshal(input)\n\tif err != nil {\n\t\t// should be extremely unlikely, unless yaml marshaling is customized\n\t\treturn nil, fmt.Errorf(\"could not serialize data to yaml: %w\", err)\n\t}\n\tvar out *YamlNode\n\terr = yaml.Unmarshal(data, &out)\n\tif err != nil {\n\t\t// should not be possible\n\t\treturn nil, fmt.Errorf(\"could not deserialize to yaml node: %w\", err)\n\t}\n\treturn out, nil\n}\n\nfunc (n *NoSQL) ConvertToShardedNoSQLConfig() *ShardedNoSQL {\n\tconnections := make(map[string]DBShardConnection)\n\tconnections[NonShardedStoreName] = DBShardConnection{\n\t\tNoSQLPlugin: n,\n\t}\n\n\treturn &ShardedNoSQL{\n\t\tDefaultShard: NonShardedStoreName,\n\t\tConnections:  connections,\n\t}\n}\n\n// ValidateAndFillDefaults validates this config and fills default values if needed\nfunc (c *Config) ValidateAndFillDefaults() error {\n\tc.fillDefaults()\n\treturn c.validate()\n}\n\nfunc (c *Config) validate() error {\n\tif err := c.Persistence.Validate(); err != nil {\n\t\treturn err\n\t}\n\tif err := c.ClusterGroupMetadata.Validate(); err != nil {\n\t\treturn err\n\t}\n\tif err := c.Archival.Validate(&c.DomainDefaults.Archival); err != nil {\n\t\treturn err\n\t}\n\n\treturn c.Authorization.Validate()\n}\n\nfunc (c *Config) fillDefaults() {\n\tc.Persistence.FillDefaults()\n\n\t// TODO: remove this at the point when we decided to make some breaking changes in config.\n\tif c.ClusterGroupMetadata == nil && c.ClusterMetadata != nil {\n\t\tc.ClusterGroupMetadata = c.ClusterMetadata\n\t\tlog.Println(\"[WARN] clusterMetadata config is deprecated. Please replace it with clusterGroupMetadata.\")\n\t}\n\n\tc.ClusterGroupMetadata.FillDefaults()\n\n\t// filling publicClient with current cluster's RPC address if empty\n\tif c.PublicClient.HostPort == \"\" && c.ClusterGroupMetadata != nil {\n\t\tname := c.ClusterGroupMetadata.CurrentClusterName\n\t\tcurrentCluster := c.ClusterGroupMetadata.ClusterGroup[name]\n\t\tc.PublicClient.HostPort = currentCluster.RPCAddress\n\t\tc.PublicClient.Transport = currentCluster.RPCTransport\n\n\t}\n\tif c.PublicClient.Transport == \"\" {\n\t\tc.PublicClient.Transport = \"grpc\"\n\t}\n\n\tif c.ClusterGroupMetadata.ClusterRedirectionPolicy == nil && c.DCRedirectionPolicy != nil {\n\t\tlog.Println(\"[WARN] dcRedirectionPolicy config is deprecated. Please replace it with clusterRedirectionPolicy.\")\n\t\tc.ClusterGroupMetadata.ClusterRedirectionPolicy = c.DCRedirectionPolicy\n\t}\n}\n\n// String converts the config object into a string\nfunc (c *Config) String() string {\n\tout, _ := json.MarshalIndent(c, \"\", \"    \")\n\treturn string(out)\n}\n\nfunc (c *Config) GetServiceConfig(serviceName string) (Service, error) {\n\tshortName := service.ShortName(serviceName)\n\tserviceConfig, ok := c.Services[shortName]\n\tif !ok {\n\t\treturn Service{}, fmt.Errorf(\"no config section for service: %s\", shortName)\n\t}\n\treturn serviceConfig, nil\n}\n"
  },
  {
    "path": "common/config/config_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"net/url\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\tuconfig \"go.uber.org/config\"\n\t\"gopkg.in/validator.v2\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestToString(t *testing.T) {\n\tvar cfg Config\n\terr := Load(\"\", \"../../config\", \"\", &cfg)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, cfg.String())\n}\n\nfunc TestFillingDefaultSQLEncodingDecodingTypes(t *testing.T) {\n\tcfg := &Config{\n\t\tPersistence: Persistence{\n\t\t\tDataStores: map[string]DataStore{\n\t\t\t\t\"sql\": {\n\t\t\t\t\tSQL: &SQL{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tClusterGroupMetadata: &ClusterGroupMetadata{},\n\t}\n\tcfg.fillDefaults()\n\tassert.Equal(t, string(constants.EncodingTypeThriftRW), cfg.Persistence.DataStores[\"sql\"].SQL.EncodingType)\n\tassert.Equal(t, []string{string(constants.EncodingTypeThriftRW)}, cfg.Persistence.DataStores[\"sql\"].SQL.DecodingTypes)\n}\n\nfunc getValidMultipleDatabasseConfig() *Config {\n\tmetadata := validClusterGroupMetadata()\n\tcfg := &Config{\n\t\tClusterGroupMetadata: metadata,\n\t\tPersistence: Persistence{\n\t\t\tDefaultStore:            \"default\",\n\t\t\tAdvancedVisibilityStore: \"esv7\",\n\t\t\tDataStores: map[string]DataStore{\n\t\t\t\t\"default\": {\n\t\t\t\t\tSQL: &SQL{\n\t\t\t\t\t\tPluginName:           \"fake\",\n\t\t\t\t\t\tConnectProtocol:      \"tcp\",\n\t\t\t\t\t\tNumShards:            2,\n\t\t\t\t\t\tUseMultipleDatabases: true,\n\t\t\t\t\t\tMultipleDatabasesConfig: []MultipleDatabasesConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tDatabaseName: \"db1\",\n\t\t\t\t\t\t\t\tConnectAddr:  \"192.168.0.1:3306\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tDatabaseName: \"db2\",\n\t\t\t\t\t\t\t\tConnectAddr:  \"192.168.0.2:3306\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"esv7\": {\n\t\t\t\t\tElasticSearch: &ElasticSearchConfig{\n\t\t\t\t\t\tVersion: \"v7\",\n\t\t\t\t\t\tURL: url.URL{Scheme: \"http\",\n\t\t\t\t\t\t\tHost: \"127.0.0.1:9200\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\"visibility\": \"cadence-visibility-dev\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t// no sql or nosql, should be populated from cassandra\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\treturn cfg\n}\n\nfunc TestValidMultipleDatabaseConfig(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidMultipleDatabaseConfig_useBasicVisibility(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tcfg.Persistence.VisibilityStore = \"basic\"\n\tcfg.Persistence.DataStores[\"basic\"] = DataStore{\n\t\tSQL: &SQL{},\n\t}\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: multipleSQLDatabases can only be used with advanced visibility only\")\n}\n\nfunc TestInvalidMultipleDatabaseConfig_wrongNumDBShards(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tsqlds := cfg.Persistence.DataStores[\"default\"]\n\tsqlds.SQL.NumShards = 3\n\tcfg.Persistence.DataStores[\"default\"] = sqlds\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: nShards must be greater than one and equal to the length of multipleDatabasesConfig\")\n}\n\nfunc TestInvalidMultipleDatabaseConfig_nonEmptySQLUser(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tsqlds := cfg.Persistence.DataStores[\"default\"]\n\tsqlds.SQL.User = \"user\"\n\tcfg.Persistence.DataStores[\"default\"] = sqlds\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: user can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n}\n\nfunc TestInvalidMultipleDatabaseConfig_nonEmptySQLPassword(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tsqlds := cfg.Persistence.DataStores[\"default\"]\n\tsqlds.SQL.Password = \"pw\"\n\tcfg.Persistence.DataStores[\"default\"] = sqlds\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: password can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n}\n\nfunc TestInvalidMultipleDatabaseConfig_nonEmptySQLDatabaseName(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tsqlds := cfg.Persistence.DataStores[\"default\"]\n\tsqlds.SQL.DatabaseName = \"db\"\n\tcfg.Persistence.DataStores[\"default\"] = sqlds\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: databaseName can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n}\n\nfunc TestInvalidMultipleDatabaseConfig_nonEmptySQLConnAddr(t *testing.T) {\n\tcfg := getValidMultipleDatabasseConfig()\n\tsqlds := cfg.Persistence.DataStores[\"default\"]\n\tsqlds.SQL.ConnectAddr = \"127.0.0.1:3306\"\n\tcfg.Persistence.DataStores[\"default\"] = sqlds\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.EqualError(t, err, \"sql persistence config: connectAddr can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n}\n\nfunc TestConfigFallbacks(t *testing.T) {\n\tmetadata := validClusterGroupMetadata()\n\tcfg := &Config{\n\t\tServices: map[string]Service{\n\t\t\t\"frontend\": {\n\t\t\t\tRPC: RPC{\n\t\t\t\t\tPort: 7900,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tClusterGroupMetadata: metadata,\n\t\tPersistence: Persistence{\n\t\t\tDefaultStore:    \"default\",\n\t\t\tVisibilityStore: \"cass\",\n\t\t\tDataStores: map[string]DataStore{\n\t\t\t\t\"default\": {\n\t\t\t\t\tSQL: &SQL{\n\t\t\t\t\t\tPluginName:      \"fake\",\n\t\t\t\t\t\tConnectProtocol: \"tcp\",\n\t\t\t\t\t\tConnectAddr:     \"192.168.0.1:3306\",\n\t\t\t\t\t\tDatabaseName:    \"db1\",\n\t\t\t\t\t\tNumShards:       0, // default value, should be changed\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"cass\": {\n\t\t\t\t\tCassandra: &NoSQL{\n\t\t\t\t\t\tHosts: \"127.0.0.1\",\n\t\t\t\t\t},\n\t\t\t\t\t// no sql or nosql, should be populated from cassandra\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.NoError(t, err) // sanity check, must be valid or the later tests are potentially useless\n\n\tassert.NotEmpty(t, cfg.Persistence.DataStores[\"cass\"].Cassandra, \"cassandra config should remain after update\")\n\tassert.NotEmpty(t, cfg.Persistence.DataStores[\"cass\"].NoSQL, \"nosql config should contain cassandra config / not be empty\")\n\tassert.NotZero(t, cfg.Persistence.DataStores[\"default\"].SQL.NumShards, \"num shards should be nonzero\")\n\tassert.Equal(t, \"localhost:7833\", cfg.PublicClient.HostPort)\n}\n\nfunc TestConfigErrorInAuthorizationConfig(t *testing.T) {\n\tcfg := &Config{\n\t\tAuthorization: Authorization{\n\t\t\tOAuthAuthorizer: OAuthAuthorizer{\n\t\t\t\tEnable: true,\n\t\t\t},\n\t\t\tNoopAuthorizer: NoopAuthorizer{\n\t\t\t\tEnable: true,\n\t\t\t},\n\t\t},\n\t\tClusterGroupMetadata: &ClusterGroupMetadata{},\n\t}\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.Error(t, err)\n}\n\nfunc TestGetServiceConfig(t *testing.T) {\n\tcfg := Config{}\n\t_, err := cfg.GetServiceConfig(service.Frontend)\n\tassert.EqualError(t, err, \"no config section for service: frontend\")\n\n\tcfg = Config{Services: map[string]Service{\"frontend\": {RPC: RPC{GRPCPort: 123}}}}\n\tsvc, err := cfg.GetServiceConfig(service.Frontend)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, svc)\n}\n\nfunc getValidShardedNoSQLConfig() *Config {\n\tmetadata := validClusterGroupMetadata()\n\tcfg := &Config{\n\t\tClusterGroupMetadata: metadata,\n\t\tPersistence: Persistence{\n\t\t\tNumHistoryShards:        2,\n\t\t\tDefaultStore:            \"default\",\n\t\t\tAdvancedVisibilityStore: \"visibility\",\n\t\t\tDataStores: map[string]DataStore{\n\t\t\t\t\"default\": {\n\t\t\t\t\tShardedNoSQL: &ShardedNoSQL{\n\t\t\t\t\t\tDefaultShard: \"shard-1\",\n\t\t\t\t\t\tShardingPolicy: ShardingPolicy{\n\t\t\t\t\t\t\tHistoryShardMapping: []HistoryShardRange{\n\t\t\t\t\t\t\t\tHistoryShardRange{\n\t\t\t\t\t\t\t\t\tStart: 0,\n\t\t\t\t\t\t\t\t\tEnd:   1,\n\t\t\t\t\t\t\t\t\tShard: \"shard-1\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tHistoryShardRange{\n\t\t\t\t\t\t\t\t\tStart: 1,\n\t\t\t\t\t\t\t\t\tEnd:   2,\n\t\t\t\t\t\t\t\t\tShard: \"shard-2\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tTaskListHashing: TasklistHashing{\n\t\t\t\t\t\t\t\tShardOrder: []string{\n\t\t\t\t\t\t\t\t\t\"shard-1\",\n\t\t\t\t\t\t\t\t\t\"shard-2\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConnections: map[string]DBShardConnection{\n\t\t\t\t\t\t\t\"shard-1\": {\n\t\t\t\t\t\t\t\tNoSQLPlugin: &NoSQL{\n\t\t\t\t\t\t\t\t\tPluginName: \"cassandra\",\n\t\t\t\t\t\t\t\t\tHosts:      \"127.0.0.1\",\n\t\t\t\t\t\t\t\t\tKeyspace:   \"unit-test\",\n\t\t\t\t\t\t\t\t\tPort:       1234,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"shard-2\": {\n\t\t\t\t\t\t\t\tNoSQLPlugin: &NoSQL{\n\t\t\t\t\t\t\t\t\tPluginName: \"cassandra\",\n\t\t\t\t\t\t\t\t\tHosts:      \"127.0.0.1\",\n\t\t\t\t\t\t\t\t\tKeyspace:   \"unit-test\",\n\t\t\t\t\t\t\t\t\tPort:       5678,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"visibility\": {\n\t\t\t\t\tElasticSearch: &ElasticSearchConfig{\n\t\t\t\t\t\tVersion: \"v7\",\n\t\t\t\t\t\tURL: url.URL{Scheme: \"http\",\n\t\t\t\t\t\t\tHost: \"127.0.0.1:9200\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\"visibility\": \"cadence-visibility-dev\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\treturn cfg\n}\n\nfunc TestValidShardedNoSQLConfig(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.NoError(t, err)\n}\n\nfunc TestInvalidShardedNoSQLConfig_MultipleConfigTypes(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.NoSQL = &NoSQL{}\n\tcfg.Persistence.DataStores[\"default\"] = store\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"must provide exactly one type of config, but provided 2\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_MissingDefaultShard(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.DefaultShard = \"\"\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"defaultShard can not be empty\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_UnknownDefaultShard(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tdelete(store.ShardedNoSQL.Connections, store.ShardedNoSQL.DefaultShard)\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"defaultShard (shard-1) is not defined in connections list\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_HistoryShardingUnordered(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[0].Start = 1\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[0].End = 2\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[1].Start = 0\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[1].End = 1\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Non-continuous history shard range\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_HistoryShardingOverlapping(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[0].End = 2 // 0-2 overlaps with 1-2\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Non-continuous history shard range\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_HistoryShardingMissingFirstShard(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping = store.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[1:]\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Non-continuous history shard range\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_HistoryShardingMissingLastShard(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.Persistence.NumHistoryShards = 3 // config only specifies shards 0 and 1, so this is invalid\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Last history shard found in the config is 1 while the max is 2\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_HistoryShardingRefersToUnknownConnection(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.ShardingPolicy.HistoryShardMapping[0].Shard = \"unknown-shard-name\"\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Unknown history shard name\")\n}\n\nfunc TestInvalidShardedNoSQLConfig_TasklistShardingRefersToUnknownConnection(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tstore := cfg.Persistence.DataStores[\"default\"]\n\tstore.ShardedNoSQL.ShardingPolicy.TaskListHashing.ShardOrder[1] = \"unknown-shard-name\"\n\n\terr := cfg.ValidateAndFillDefaults()\n\trequire.ErrorContains(t, err, \"Unknown tasklist shard name\")\n}\n\nfunc TestHistogramMigrationConfig(t *testing.T) {\n\tt.Run(\"valid\", func(t *testing.T) {\n\t\torig := metrics.HistogramMigrationMetrics\n\t\tt.Cleanup(func() {\n\t\t\tmetrics.HistogramMigrationMetrics = orig\n\t\t})\n\t\tmetrics.HistogramMigrationMetrics = map[string]struct{}{\n\t\t\t\"key1\": {},\n\t\t\t\"key2\": {},\n\t\t\t\"key3\": {},\n\t\t\t\"key4\": {},\n\t\t}\n\n\t\t// intentionally avoiding the full config struct, as it has other required config\n\t\tyaml, err := uconfig.NewYAML(uconfig.RawSource(strings.NewReader(`\ndefault: histogram\nnames:\n  key1: true\n  key2: false\n  key3:\n`)))\n\t\trequire.NoError(t, err)\n\n\t\tvar cfg metrics.HistogramMigration\n\t\terr = yaml.Get(uconfig.Root).Populate(&cfg)\n\t\trequire.NoError(t, err)\n\n\t\terr = validator.Validate(cfg)\n\t\trequire.NoError(t, err)\n\n\t\tcheck := func(key string, timer, histogram bool) {\n\t\t\tassert.Equalf(t, timer, cfg.EmitTimer(key), \"wrong value for EmitTimer(%q)\", key)\n\t\t\tassert.Equalf(t, histogram, cfg.EmitHistogram(key), \"wrong value for EmitHistogram(%q)\", key)\n\t\t}\n\t\tcheck(\"key1\", true, true)\n\t\tcheck(\"key2\", false, false)\n\t\tcheck(\"key3\", false, false) // the type's default mode == false.  not truly intended behavior, but it's weird config so it's fine.\n\t\tcheck(\"key4\", false, true)  // configured default == histogram\n\t\tcheck(\"key5\", true, true)   // not migrating = always emitted\n\t\tif t.Failed() {\n\t\t\tt.Logf(\"config: %#v\", cfg)\n\t\t}\n\t})\n\tt.Run(\"invalid default\", func(t *testing.T) {\n\t\tyaml, err := uconfig.NewYAML(uconfig.RawSource(strings.NewReader(`\ndefault: xyz\n`)))\n\t\trequire.NoError(t, err)\n\n\t\tvar cfg metrics.HistogramMigration\n\t\terr = yaml.Get(uconfig.Root).Populate(&cfg)\n\t\tassert.ErrorContains(t, err, `unsupported histogram migration mode \"xyz\", must be \"timer\", \"histogram\", or \"both\"`)\n\t})\n\tt.Run(\"invalid key\", func(t *testing.T) {\n\t\torig := metrics.HistogramMigrationMetrics\n\t\tt.Cleanup(func() {\n\t\t\tmetrics.HistogramMigrationMetrics = orig\n\t\t})\n\t\tmetrics.HistogramMigrationMetrics = map[string]struct{}{\n\t\t\t\"key1\": {},\n\t\t}\n\t\tyaml, err := uconfig.NewYAML(uconfig.RawSource(strings.NewReader(`\nnames:\n  key1: xyz\n`)))\n\t\trequire.NoError(t, err)\n\n\t\tvar cfg metrics.HistogramMigration\n\t\terr = yaml.Get(uconfig.Root).Populate(&cfg)\n\t\tassert.ErrorContains(t, err, \"cannot unmarshal !!str `xyz` into bool\")\n\t})\n\tt.Run(\"nonexistent key\", func(t *testing.T) {\n\t\tyaml, err := uconfig.NewYAML(uconfig.RawSource(strings.NewReader(`\nnames:\n  definitely_does_not_exist: true\n`)))\n\t\trequire.NoError(t, err)\n\n\t\tvar cfg metrics.HistogramMigration\n\t\terr = yaml.Get(uconfig.Root).Populate(&cfg)\n\t\tassert.ErrorContains(t, err, `unknown histogram-migration metric name \"definitely_does_not_exist\"`)\n\t})\n}\n"
  },
  {
    "path": "common/config/elasticsearch.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"net/url\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/credentials\"\n\t\"github.com/aws/aws-sdk-go/aws/session\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\nvar errAWSSigningCredential = fmt.Errorf(\"must provide exactly one type of credential, EnvironmentCredential or StaticCredential\")\n\n// ElasticSearchConfig for connecting to ElasticSearch\ntype (\n\tElasticSearchConfig struct {\n\t\tURL     url.URL           `yaml:\"url\"`     //nolint:govet\n\t\tIndices map[string]string `yaml:\"indices\"` //nolint:govet\n\t\t// supporting v6 and v7. Default to v6 if empty.\n\t\tVersion string `yaml:\"version\"` //nolint:govet\n\t\t// optional username to communicate with ElasticSearch\n\t\tUsername string `yaml:\"username\"` //nolint:govet\n\t\t// optional password to communicate with ElasticSearch\n\t\tPassword string `yaml:\"password\"` //nolint:govet\n\t\t// optional to disable sniff, according to issues on Github,\n\t\t// Sniff could cause issue like \"no Elasticsearch node available\"\n\t\tDisableSniff bool `yaml:\"disableSniff\"`\n\t\t// optional to disable health check\n\t\tDisableHealthCheck bool `yaml:\"disableHealthCheck\"`\n\t\t// optional to use AWS signing client\n\t\t// See more info https://github.com/olivere/elastic/wiki/Using-with-AWS-Elasticsearch-Service\n\t\tAWSSigning AWSSigning `yaml:\"awsSigning\"`\n\t\t// optional to use Signed Certificates over https\n\t\tTLS TLS `yaml:\"tls\"`\n\t\t// optional to add custom headers\n\t\tCustomHeaders map[string]string   `yaml:\"customHeaders,omitempty\"`\n\t\tMigration     VisibilityMigration `yaml:\"migration\"`\n\t\t// optional, will use default consumer name if not provided\n\t\t// default consumerName is topic + \"-consumer\"\n\t\tConsumerName string `yaml:\"consumerName\"`\n\t}\n\n\t// AWSSigning contains config to enable signing,\n\t// Must provide either StaticCredential or EnvironmentCredential\n\tAWSSigning struct {\n\t\tEnable                bool                      `yaml:\"enable\"`\n\t\tStaticCredential      *AWSStaticCredential      `yaml:\"staticCredential\"`\n\t\tEnvironmentCredential *AWSEnvironmentCredential `yaml:\"environmentCredential\"`\n\t}\n\n\t// AWSStaticCredential to create a static credentials value provider.\n\t// SessionToken is only required for temporary security credentials retrieved via STS,\n\t// otherwise an empty string can be passed for this parameter.\n\t// See more in https://github.com/aws/aws-sdk-go/blob/master/aws/credentials/static_provider.go#L21\n\tAWSStaticCredential struct {\n\t\tAccessKey    string `yaml:\"accessKey\"`\n\t\tRegion       string `yaml:\"region\"`\n\t\tSecretKey    string `yaml:\"secretKey\"`\n\t\tSessionToken string `yaml:\"sessionToken\"`\n\t}\n\n\t// AWSEnvironmentCredential will make a new Session created from SDK defaults, config files,\n\t// environment, and user provided config files.\n\t// See more in https://github.com/aws/aws-sdk-go/blob/3974dd034387fbc7cf09c8cd2400787ce07f3285/aws/session/session.go#L147\n\tAWSEnvironmentCredential struct {\n\t\tRegion string `yaml:\"region\"`\n\t}\n\n\tVisibilityMigration struct {\n\t\tEnabled bool `yaml:\"enabled\"`\n\t}\n)\n\n// GetVisibilityIndex return visibility index name\nfunc (cfg *ElasticSearchConfig) GetVisibilityIndex() string {\n\treturn cfg.Indices[constants.VisibilityAppName]\n}\n\n// SetUsernamePassword set the username/password into URL\n// It is a bit tricky here because url.URL doesn't expose the username/password in the struct\n// because of the security concern.\nfunc (cfg *ElasticSearchConfig) SetUsernamePassword() {\n\tif cfg.Username != \"\" {\n\t\tcfg.URL.User = url.UserPassword(cfg.Username, cfg.Password)\n\t}\n}\n\nfunc (a AWSSigning) Validate() error {\n\tif a.EnvironmentCredential == nil && a.StaticCredential == nil {\n\t\treturn errAWSSigningCredential\n\t}\n\tif a.EnvironmentCredential != nil && a.StaticCredential != nil {\n\t\treturn errAWSSigningCredential\n\t}\n\n\tif a.EnvironmentCredential != nil && len(a.EnvironmentCredential.Region) == 0 {\n\t\treturn errors.New(\"missing region in environmentCredential\")\n\t}\n\n\tif a.StaticCredential != nil && len(a.StaticCredential.Region) == 0 {\n\t\treturn errors.New(\"missing region in staticCredential\")\n\t}\n\n\treturn nil\n}\n\nfunc (a AWSSigning) GetCredentials() (*credentials.Credentials, *string, error) {\n\tif err := a.Validate(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\t// refer to https://github.com/olivere/elastic/blob/release-branch.v7/recipes/aws-connect-v4/main.go\n\tif a.EnvironmentCredential != nil {\n\t\tsess, err := session.NewSession(&aws.Config{Region: &a.EnvironmentCredential.Region})\n\t\tif err != nil {\n\t\t\treturn nil, nil, fmt.Errorf(\"creating aws session: %w\", err)\n\t\t}\n\n\t\treturn sess.Config.Credentials, sess.Config.Region, nil\n\t}\n\n\tawsCredentials := credentials.NewStaticCredentials(\n\t\ta.StaticCredential.AccessKey,\n\t\ta.StaticCredential.SecretKey,\n\t\ta.StaticCredential.SessionToken,\n\t)\n\n\treturn awsCredentials, &a.StaticCredential.Region, nil\n}\n\n// GetCustomHeader returns the header for the specified key\nfunc (cfg *ElasticSearchConfig) GetCustomHeader(headerKey string) string {\n\tif headerValue, ok := cfg.CustomHeaders[headerKey]; ok {\n\t\treturn headerValue\n\t}\n\treturn \"\"\n}\n"
  },
  {
    "path": "common/config/elasticsearch_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestAWSSigning_ValidateEmpty(t *testing.T) {\n\n\ttests := []struct {\n\t\tmsg    string\n\t\tconfig AWSSigning\n\t\terr    error\n\t}{\n\t\t{\n\t\t\tmsg: \"Empty config should error\",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tStaticCredential:      nil,\n\t\t\t\tEnvironmentCredential: nil,\n\t\t\t},\n\t\t\terr: errAWSSigningCredential,\n\t\t},\n\t\t{\n\t\t\tmsg: \"error when both config sections are provided\",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tEnable:                false,\n\t\t\t\tStaticCredential:      &AWSStaticCredential{},\n\t\t\t\tEnvironmentCredential: &AWSEnvironmentCredential{},\n\t\t\t},\n\t\t\terr: errAWSSigningCredential,\n\t\t},\n\t\t{\n\t\t\tmsg: \"StaticCredential must have region set\",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tEnable:                false,\n\t\t\t\tStaticCredential:      &AWSStaticCredential{},\n\t\t\t\tEnvironmentCredential: nil,\n\t\t\t},\n\t\t\terr: errors.New(\"missing region in staticCredential\"),\n\t\t},\n\t\t{\n\t\t\tmsg: \"EnvironmentCredential must have region set\",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tEnable:                false,\n\t\t\t\tStaticCredential:      nil,\n\t\t\t\tEnvironmentCredential: &AWSEnvironmentCredential{},\n\t\t\t},\n\t\t\terr: errors.New(\"missing region in environmentCredential\"),\n\t\t},\n\t\t{\n\t\t\tmsg: \"Valid StaticCredential config should have no error \",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tEnable:                false,\n\t\t\t\tStaticCredential:      &AWSStaticCredential{Region: \"region1\"},\n\t\t\t\tEnvironmentCredential: nil,\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tmsg: \"Valid EnvironmentCredential config should have no error\",\n\t\t\tconfig: AWSSigning{\n\t\t\t\tEnable:                false,\n\t\t\t\tStaticCredential:      nil,\n\t\t\t\tEnvironmentCredential: &AWSEnvironmentCredential{Region: \"region1\"},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tassert.Equal(t, tc.err, tc.config.Validate(), tc.msg)\n\t}\n\n}\n\nfunc TestGetCustomHeader(t *testing.T) {\n\n\ttests := []struct {\n\t\tconfig   ElasticSearchConfig\n\t\theader   string\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tconfig: ElasticSearchConfig{\n\t\t\t\tCustomHeaders: map[string]string{\n\t\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\theader:   \"key1\",\n\t\t\texpected: \"value1\",\n\t\t},\n\t\t{\n\t\t\tconfig: ElasticSearchConfig{\n\t\t\t\tCustomHeaders: map[string]string{\n\t\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\theader:   \"key2\",\n\t\t\texpected: \"\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tval := tc.config.GetCustomHeader(tc.header)\n\t\tassert.Equal(t, val, tc.expected)\n\t}\n\n}\n"
  },
  {
    "path": "common/config/fx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// Module returns a config.Provider that could be used byother components.\nvar Module = fx.Module(\"configfx\",\n\tfx.Provide(New),\n)\n\ntype Context struct {\n\tEnvironment string\n\tZone        string\n}\n\n// Params defines the dependencies of the configfx module.\ntype Params struct {\n\tfx.In\n\n\tService string `name:\"service\"`\n\n\tContext   Context\n\tLookupEnv LookupEnvFunc `optional:\"true\"`\n\n\tConfigDir string `name:\"config-dir\"`\n\n\tLifecycle fx.Lifecycle `optional:\"true\"` // required for strict mode\n}\n\n// Result defines the objects that the configfx module provides.\ntype Result struct {\n\tfx.Out\n\n\tConfig        Config\n\tServiceConfig Service\n\tHistograms    metrics.HistogramMigration\n}\n\n// LookupEnvFunc returns the value of the environment variable given by key.\n// It should behave the same as `os.LookupEnv`. If a function returns false,\n// an environment variable is looked up using `os.LookupEnv`.\ntype LookupEnvFunc func(key string) (string, bool)\n\n// New exports functionality similar to Module, but allows the caller to wrap\n// or modify Result. Most users should use Module instead.\nfunc New(p Params) (Result, error) {\n\tlookupFun := os.LookupEnv\n\tif p.LookupEnv != nil {\n\t\tlookupFun = func(key string) (string, bool) {\n\t\t\tif result, ok := p.LookupEnv(key); ok {\n\t\t\t\treturn result, true\n\t\t\t}\n\t\t\treturn lookupFun(key)\n\t\t}\n\t}\n\n\tvar cfg Config\n\terr := Load(p.Context.Environment, p.ConfigDir, p.Context.Zone, &cfg)\n\tif err != nil {\n\t\treturn Result{}, fmt.Errorf(\"load config: %w\", err)\n\t}\n\n\tcfg.fillDefaults()\n\n\tsvcCfg, err := cfg.GetServiceConfig(p.Service)\n\tif err != nil {\n\t\treturn Result{}, fmt.Errorf(\"get service config: %w\", err)\n\t}\n\n\tp.Lifecycle.Append(fx.StartHook(cfg.validate))\n\n\treturn Result{\n\t\tConfig:        cfg,\n\t\tServiceConfig: svcCfg,\n\t\tHistograms:    cfg.Histograms,\n\t}, nil\n}\n"
  },
  {
    "path": "common/config/kafkaConfig.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport \"fmt\"\n\ntype (\n\t// KafkaConfig describes the configuration needed to connect to all kafka clusters\n\tKafkaConfig struct {\n\t\tTLS      TLS                      `yaml:\"tls\"`\n\t\tSASL     SASL                     `yaml:\"sasl\"`\n\t\tClusters map[string]ClusterConfig `yaml:\"clusters\"`\n\t\tTopics   map[string]TopicConfig   `yaml:\"topics\"`\n\t\t// Applications describes the applications that will use the Kafka topics\n\t\tApplications map[string]TopicList `yaml:\"applications\"`\n\t\tVersion      string               `yaml:\"version\"`\n\t}\n\n\t// ClusterConfig describes the configuration for a single Kafka cluster\n\tClusterConfig struct {\n\t\tBrokers []string `yaml:\"brokers\"`\n\t}\n\n\t// TopicConfig describes the mapping from topic to Kafka cluster\n\tTopicConfig struct {\n\t\tCluster string `yaml:\"cluster\"`\n\t\t// Properties map describes whether the topic properties, such as whether it is secure\n\t\tProperties map[string]any `yaml:\"properties,omitempty\"`\n\t}\n\n\t// TopicList describes the topic names for each cluster\n\tTopicList struct {\n\t\tTopic    string `yaml:\"topic\"`\n\t\tDLQTopic string `yaml:\"dlq-topic\"`\n\t}\n)\n\n// Validate will validate config for kafka\nfunc (k *KafkaConfig) Validate(checkApp bool) {\n\tif len(k.Clusters) == 0 {\n\t\tpanic(\"Empty Kafka Cluster Config\")\n\t}\n\tif len(k.Topics) == 0 {\n\t\tpanic(\"Empty Topics Config\")\n\t}\n\n\tvalidateTopicsFn := func(topic string) {\n\t\tif topic == \"\" {\n\t\t\tpanic(\"Empty Topic Name\")\n\t\t} else if topicConfig, ok := k.Topics[topic]; !ok {\n\t\t\tpanic(fmt.Sprintf(\"Missing Topic Config for Topic %v\", topic))\n\t\t} else if clusterConfig, ok := k.Clusters[topicConfig.Cluster]; !ok {\n\t\t\tpanic(fmt.Sprintf(\"Missing Kafka Cluster Config for Cluster %v\", topicConfig.Cluster))\n\t\t} else if len(clusterConfig.Brokers) == 0 {\n\t\t\tpanic(fmt.Sprintf(\"Missing Kafka Brokers Config for Cluster %v\", topicConfig.Cluster))\n\t\t}\n\t}\n\n\tif checkApp {\n\t\tif len(k.Applications) == 0 {\n\t\t\tpanic(\"Empty Applications Config\")\n\t\t}\n\t\tfor _, topics := range k.Applications {\n\t\t\tvalidateTopicsFn(topics.Topic)\n\t\t\tvalidateTopicsFn(topics.DLQTopic)\n\t\t}\n\t}\n}\n\n// GetKafkaClusterForTopic gets cluster from topic\nfunc (k *KafkaConfig) GetKafkaClusterForTopic(topic string) string {\n\treturn k.Topics[topic].Cluster\n}\n\n// GetBrokersForKafkaCluster gets broker from cluster\nfunc (k *KafkaConfig) GetBrokersForKafkaCluster(kafkaCluster string) []string {\n\treturn k.Clusters[kafkaCluster].Brokers\n}\n\n// GetTopicsForApplication gets topic from application\nfunc (k *KafkaConfig) GetTopicsForApplication(app string) TopicList {\n\treturn k.Applications[app]\n}\n\n// GetKafkaPropertiesForTopic gets properties from topic\nfunc (k *KafkaConfig) GetKafkaPropertyForTopic(topic string, property string) any {\n\ttopicConfig, ok := k.Topics[topic]\n\tif !ok || topicConfig.Properties == nil {\n\t\t// No properties for the specified topic in the config\n\t\treturn nil\n\t}\n\n\t// retrieve the property from the topic properties\n\tpropertyValue, ok := topicConfig.Properties[property]\n\tif !ok {\n\t\t// Property not found\n\t\treturn nil\n\t}\n\n\treturn propertyValue\n}\n"
  },
  {
    "path": "common/config/loader.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\n\tuconfig \"go.uber.org/config\"\n\t\"gopkg.in/validator.v2\"\n)\n\nconst (\n\t// EnvKeyRoot the environment variable key for runtime root dir\n\tEnvKeyRoot = \"CADENCE_ROOT\"\n\t// EnvKeyConfigDir the environment variable key for config dir\n\tEnvKeyConfigDir = \"CADENCE_CONFIG_DIR\"\n\t// EnvKeyEnvironment is the environment variable key for environment\n\tEnvKeyEnvironment = \"CADENCE_ENVIRONMENT\"\n\t// EnvKeyAvailabilityZone is the environment variable key for AZ\n\tEnvKeyAvailabilityZone = \"CADENCE_AVAILABILTY_ZONE\"\n)\n\nconst (\n\tbaseFile         = \"base.yaml\"\n\tenvDevelopment   = \"development\"\n\tdefaultConfigDir = \"config\"\n)\n\n// Load loads the configuration from a set of\n// yaml config files found in the config directory\n//\n// The loader first fetches the set of files matching\n// a pre-determined naming convention, then sorts\n// them by hierarchy order and after that, simply\n// loads the files one after another with the\n// key/values in the later files overriding the key/values\n// in the earlier files\n//\n// The hierarchy is as follows from lowest to highest\n//\n//\tbase.yaml\n//\t    env.yaml   -- environment is one of the input params ex-development\n//\t      env_az.yaml -- zone is another input param\nfunc Load(env string, configDir string, zone string, config interface{}) error {\n\n\tif len(env) == 0 {\n\t\tenv = envDevelopment\n\t}\n\n\tif len(configDir) == 0 {\n\t\tconfigDir = defaultConfigDir\n\t}\n\n\tfiles, err := getConfigFiles(env, configDir, zone)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to get config files: %w\", err)\n\t}\n\n\tlog.Printf(\"Loading configFiles=%v\\n\", files)\n\n\tvar options []uconfig.YAMLOption\n\tfor _, f := range files {\n\t\toptions = append(options, uconfig.File(f))\n\t}\n\n\t// expand env variables declared in .yaml files\n\toptions = append(options, uconfig.Expand(os.LookupEnv))\n\n\tyaml, err := uconfig.NewYAML(options...)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to create yaml parser: %w\", err)\n\t}\n\n\terr = yaml.Get(uconfig.Root).Populate(config)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to populate config: %w\", err)\n\t}\n\n\terr = validator.Validate(config)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to validate config: %w\", err)\n\t}\n\treturn nil\n}\n\n// getConfigFiles returns the list of config files to\n// process in the hierarchy order\nfunc getConfigFiles(env string, configDir string, zone string) ([]string, error) {\n\n\tcandidates := []string{\n\t\tpath(configDir, baseFile),\n\t\tpath(configDir, file(env, \"yaml\")),\n\t}\n\n\tif len(zone) > 0 {\n\t\tf := file(concat(env, zone), \"yaml\")\n\t\tcandidates = append(candidates, path(configDir, f))\n\t}\n\n\tvar result []string\n\n\tfor _, c := range candidates {\n\t\tif _, err := os.Stat(c); err != nil {\n\t\t\tcontinue\n\t\t}\n\t\tresult = append(result, c)\n\t}\n\n\tif len(result) == 0 {\n\t\treturn nil, fmt.Errorf(\"no config files found within %v\", configDir)\n\t}\n\n\treturn result, nil\n}\n\nfunc concat(a, b string) string {\n\treturn a + \"_\" + b\n}\n\nfunc file(name string, suffix string) string {\n\treturn name + \".\" + suffix\n}\n\nfunc path(dir string, file string) string {\n\treturn dir + \"/\" + file\n}\n"
  },
  {
    "path": "common/config/loader_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"io/ioutil\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tLoaderSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\t}\n\n\titemsConfig struct {\n\t\tItem1 string `yaml:\"item1\"`\n\t\tItem2 string `yaml:\"item2\"`\n\t}\n\n\ttestConfig struct {\n\t\tItems itemsConfig `yaml:\"items\"`\n\t}\n)\n\nfunc TestLoaderSuite(t *testing.T) {\n\tsuite.Run(t, new(LoaderSuite))\n}\n\nfunc (s *LoaderSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *LoaderSuite) TestBaseYaml() {\n\n\tdir := s.T().TempDir()\n\n\ts.createFile(dir, \"base.yaml\", `\nitems:\n  item1: base1\n  item2: base2`)\n\n\tenvs := []string{\"\", \"prod\"}\n\tzones := []string{\"\", \"us-east-1a\"}\n\n\tfor _, env := range envs {\n\t\tfor _, zone := range zones {\n\t\t\tvar cfg testConfig\n\t\t\terr := Load(env, dir, zone, &cfg)\n\t\t\ts.Nil(err)\n\t\t\ts.Equal(\"base1\", cfg.Items.Item1)\n\t\t\ts.Equal(\"base2\", cfg.Items.Item2)\n\t\t}\n\t}\n}\n\nfunc (s *LoaderSuite) TestHierarchy() {\n\n\tdir := s.T().TempDir()\n\n\ts.createFile(dir, \"base.yaml\", `\nitems:\n  item1: base1\n  item2: base2`)\n\n\ts.createFile(dir, \"development.yaml\", `\nitems:\n  item1: development1`)\n\n\ts.createFile(dir, \"prod.yaml\", `\nitems:\n  item1: prod1`)\n\n\ts.createFile(dir, \"prod_dca.yaml\", `\nitems:\n  item1: prod_dca1`)\n\n\ttestCases := []struct {\n\t\tenv   string\n\t\tzone  string\n\t\titem1 string\n\t\titem2 string\n\t}{\n\t\t{\"\", \"\", \"development1\", \"base2\"},\n\t\t{\"\", \"dca\", \"development1\", \"base2\"},\n\t\t{\"\", \"pdx\", \"development1\", \"base2\"},\n\t\t{\"development\", \"\", \"development1\", \"base2\"},\n\t\t{\"development\", \"dca\", \"development1\", \"base2\"},\n\t\t{\"development\", \"pdx\", \"development1\", \"base2\"},\n\t\t{\"prod\", \"\", \"prod1\", \"base2\"},\n\t\t{\"prod\", \"dca\", \"prod_dca1\", \"base2\"},\n\t\t{\"prod\", \"pdx\", \"prod1\", \"base2\"},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvar cfg testConfig\n\t\terr := Load(tc.env, dir, tc.zone, &cfg)\n\t\ts.Nil(err)\n\t\ts.Equal(tc.item1, cfg.Items.Item1)\n\t\ts.Equal(tc.item2, cfg.Items.Item2)\n\t}\n}\n\nfunc (s *LoaderSuite) TestInvalidPath() {\n\tvar cfg testConfig\n\terr := Load(\"prod\", \"\", \"\", &cfg)\n\ts.NotNil(err)\n}\n\nfunc (s *LoaderSuite) createFile(dir string, file string, content string) {\n\terr := ioutil.WriteFile(path(dir, file), []byte(content), fileMode)\n\ts.Nil(err)\n}\n"
  },
  {
    "path": "common/config/log.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n)\n\nconst fileMode = os.FileMode(0644)\n\n// NewZapLogger builds and returns a new\n// Zap logger for this logging configuration\nfunc (cfg *Logger) NewZapLogger() (*zap.Logger, error) {\n\tlevelKey := cfg.LevelKey\n\tif levelKey == \"\" {\n\t\tlevelKey = \"level\"\n\t}\n\n\tencodeConfig := zapcore.EncoderConfig{\n\t\tTimeKey:        \"ts\",\n\t\tLevelKey:       levelKey,\n\t\tNameKey:        \"logger\",\n\t\tCallerKey:      \"\", // we use our own caller, check common/log/logger.go\n\t\tMessageKey:     \"msg\",\n\t\tStacktraceKey:  \"stacktrace\",\n\t\tLineEnding:     zapcore.DefaultLineEnding,\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\tEncodeDuration: zapcore.SecondsDurationEncoder,\n\t\tEncodeCaller:   nil,\n\t}\n\n\toutputPath := \"stderr\"\n\tif len(cfg.OutputFile) > 0 {\n\t\toutputPath = cfg.OutputFile\n\t\tif cfg.Stdout {\n\t\t\toutputPath = \"stdout\"\n\t\t}\n\t}\n\n\tencoding := \"json\"\n\tif cfg.Encoding != \"\" {\n\t\tif cfg.Encoding == \"json\" || cfg.Encoding == \"console\" {\n\t\t\tencoding = cfg.Encoding\n\t\t} else {\n\t\t\treturn nil, fmt.Errorf(\"invalid encoding for log, only supporting json or console\")\n\t\t}\n\t}\n\n\tconfig := zap.Config{\n\t\tLevel:            zap.NewAtomicLevelAt(parseZapLevel(cfg.Level)),\n\t\tDevelopment:      false,\n\t\tSampling:         nil, // consider exposing this to config for our external customer\n\t\tEncoding:         encoding,\n\t\tEncoderConfig:    encodeConfig,\n\t\tOutputPaths:      []string{outputPath},\n\t\tErrorOutputPaths: []string{outputPath},\n\t}\n\treturn config.Build()\n}\n\nfunc parseZapLevel(level string) zapcore.Level {\n\tswitch strings.ToLower(level) {\n\tcase \"debug\":\n\t\treturn zap.DebugLevel\n\tcase \"info\":\n\t\treturn zap.InfoLevel\n\tcase \"warn\":\n\t\treturn zap.WarnLevel\n\tcase \"error\":\n\t\treturn zap.ErrorLevel\n\tcase \"fatal\":\n\t\treturn zap.FatalLevel\n\tdefault:\n\t\treturn zap.InfoLevel\n\t}\n}\n"
  },
  {
    "path": "common/config/log_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/zap\"\n)\n\ntype LogSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestLogSuite(t *testing.T) {\n\tsuite.Run(t, new(LogSuite))\n}\n\nfunc (s *LogSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *LogSuite) TestParseLogLevel() {\n\ts.Equal(zap.DebugLevel, parseZapLevel(\"debug\"))\n\ts.Equal(zap.InfoLevel, parseZapLevel(\"info\"))\n\ts.Equal(zap.WarnLevel, parseZapLevel(\"warn\"))\n\ts.Equal(zap.ErrorLevel, parseZapLevel(\"error\"))\n\ts.Equal(zap.FatalLevel, parseZapLevel(\"fatal\"))\n\ts.Equal(zap.InfoLevel, parseZapLevel(\"unknown\"))\n}\n\nfunc (s *LogSuite) TestNewLogger() {\n\n\tdir := s.T().TempDir()\n\n\tconfig := &Logger{\n\t\tLevel:      \"info\",\n\t\tOutputFile: dir + \"/test.log\",\n\t}\n\n\tlog, _ := config.NewZapLogger()\n\ts.NotNil(log)\n\t_, err := os.Stat(dir + \"/test.log\")\n\ts.Nil(err)\n}\n"
  },
  {
    "path": "common/config/metrics.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"time\"\n\n\t\"github.com/cactus/go-statsd-client/statsd\"\n\tprom \"github.com/m3db/prometheus_client_golang/prometheus\"\n\t\"github.com/uber-go/tally\"\n\t\"github.com/uber-go/tally/prometheus\"\n\ttallystatsdreporter \"github.com/uber-go/tally/statsd\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmprom \"github.com/uber/cadence/common/metrics/tally/prometheus\"\n\tstatsdreporter \"github.com/uber/cadence/common/metrics/tally/statsd\"\n)\n\nconst (\n\t_defaultReportingInterval = time.Second\n)\n\n// tally sanitizer options that satisfy both Prometheus and M3 restrictions.\n// This will rename metrics at the tally emission level, so metrics name we\n// use maybe different from what gets emitted. In the current implementation\n// it will replace - and . with _\n// We should still ensure that the base metrics are prometheus compatible,\n// but this is necessary as the same prom client initialization is used by\n// our system workflows.\nvar (\n\tsafeCharacters = []rune{'_'}\n\n\tsanitizeOptions = tally.SanitizeOptions{\n\t\tNameCharacters: tally.ValidCharacters{\n\t\t\tRanges:     tally.AlphanumericRange,\n\t\t\tCharacters: safeCharacters,\n\t\t},\n\t\tKeyCharacters: tally.ValidCharacters{\n\t\t\tRanges:     tally.AlphanumericRange,\n\t\t\tCharacters: safeCharacters,\n\t\t},\n\t\tValueCharacters: tally.ValidCharacters{\n\t\t\tRanges:     tally.AlphanumericRange,\n\t\t\tCharacters: safeCharacters,\n\t\t},\n\t\tReplacementCharacter: tally.DefaultReplacementCharacter,\n\t}\n)\n\n// NewScope builds a new tally scope for this metrics configuration\n// Only one reporter type is allowed\nfunc (c *Metrics) NewScope(logger log.Logger, service string) tally.Scope {\n\tif c.ReportingInterval <= 0 {\n\t\tc.ReportingInterval = _defaultReportingInterval\n\t}\n\trootScope := tally.NoopScope\n\tif c.M3 != nil {\n\t\trootScope = c.newM3Scope(logger)\n\t}\n\tif c.Statsd != nil {\n\t\tif rootScope != tally.NoopScope {\n\t\t\tlogger.Fatal(\"error creating metric reporter: cannot have more than one types of metric configuration\")\n\t\t}\n\t\trootScope = c.newStatsdScope(logger)\n\t}\n\tif c.Prometheus != nil {\n\t\tif rootScope != tally.NoopScope {\n\t\t\tlogger.Fatal(\"error creating metric reporter: cannot have more than one types of metric configuration\")\n\t\t}\n\t\trootScope = c.newPrometheusScope(logger)\n\t}\n\trootScope = rootScope.Tagged(map[string]string{metrics.CadenceServiceTagName: service})\n\treturn rootScope\n}\n\n// newM3Scope returns a new m3 scope with\n// a default reporting interval of a second\nfunc (c *Metrics) newM3Scope(logger log.Logger) tally.Scope {\n\treporter, err := c.M3.NewReporter()\n\tif err != nil {\n\t\tlogger.Fatal(\"error creating m3 reporter\", tag.Error(err))\n\t}\n\tscopeOpts := tally.ScopeOptions{\n\t\tTags:           c.Tags,\n\t\tCachedReporter: reporter,\n\t\tPrefix:         c.Prefix,\n\t}\n\tscope, _ := tally.NewRootScope(scopeOpts, c.ReportingInterval)\n\treturn scope\n}\n\n// newM3Scope returns a new statsd scope with\n// a default reporting interval of a second\nfunc (c *Metrics) newStatsdScope(logger log.Logger) tally.Scope {\n\tconfig := c.Statsd\n\tif len(config.HostPort) == 0 {\n\t\treturn tally.NoopScope\n\t}\n\tstatter, err := statsd.NewClientWithConfig(&statsd.ClientConfig{\n\t\tAddress:       config.HostPort,\n\t\tPrefix:        config.Prefix,\n\t\tUseBuffered:   true,\n\t\tFlushInterval: config.FlushInterval,\n\t\tFlushBytes:    config.FlushBytes,\n\t})\n\tif err != nil {\n\t\tlogger.Fatal(\"error creating statsd client\", tag.Error(err))\n\t}\n\t// NOTE: according to ( https://github.com/uber-go/tally )Tally's statsd implementation doesn't support tagging.\n\t// Therefore, we implement Tally interface to have a statsd reporter that can support tagging\n\treporter := statsdreporter.NewReporter(statter, tallystatsdreporter.Options{})\n\tscopeOpts := tally.ScopeOptions{\n\t\tTags:     c.Tags,\n\t\tReporter: reporter,\n\t\tPrefix:   c.Prefix,\n\t}\n\tscope, _ := tally.NewRootScope(scopeOpts, c.ReportingInterval)\n\treturn scope\n}\n\n// newPrometheusScope returns a new prometheus scope with\n// a default reporting interval of a second\nfunc (c *Metrics) newPrometheusScope(logger log.Logger) tally.Scope {\n\tif len(c.Prometheus.DefaultHistogramBuckets) == 0 {\n\t\tc.Prometheus.DefaultHistogramBuckets = mprom.DefaultHistogramBuckets()\n\t}\n\treporter, err := c.Prometheus.NewReporter(\n\t\tprometheus.ConfigurationOptions{\n\t\t\tRegistry: prom.NewRegistry(),\n\t\t\tOnError: func(err error) {\n\t\t\t\tlogger.Warn(\"error in prometheus reporter\", tag.Error(err))\n\t\t\t},\n\t\t},\n\t)\n\tif err != nil {\n\t\tlogger.Fatal(\"error creating prometheus reporter\", tag.Error(err))\n\t}\n\tscopeOpts := tally.ScopeOptions{\n\t\tTags:            c.Tags,\n\t\tCachedReporter:  reporter,\n\t\tSeparator:       prometheus.DefaultSeparator,\n\t\tSanitizeOptions: &sanitizeOptions,\n\t\tPrefix:          c.Prefix,\n\t}\n\tscope, _ := tally.NewRootScope(scopeOpts, c.ReportingInterval)\n\treturn scope\n}\n"
  },
  {
    "path": "common/config/metrics_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"github.com/uber-go/tally/m3\"\n\t\"github.com/uber-go/tally/prometheus\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype MetricsSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestMetricsSuite(t *testing.T) {\n\tsuite.Run(t, new(MetricsSuite))\n}\n\nfunc (s *MetricsSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *MetricsSuite) TestStatsd() {\n\tstatsd := &Statsd{\n\t\tHostPort: \"127.0.0.1:8125\",\n\t\tPrefix:   \"testStatsd\",\n\t}\n\n\tconfig := new(Metrics)\n\tconfig.Statsd = statsd\n\tscope := config.NewScope(testlogger.New(s.T()), \"test\")\n\ts.NotNil(scope)\n}\n\nfunc (s *MetricsSuite) TestM3() {\n\tm3 := &m3.Configuration{\n\t\tHostPort: \"127.0.0.1:8125\",\n\t\tService:  \"testM3\",\n\t\tEnv:      \"devel\",\n\t}\n\tconfig := new(Metrics)\n\tconfig.M3 = m3\n\tscope := config.NewScope(testlogger.New(s.T()), \"test\")\n\ts.NotNil(scope)\n}\n\nfunc (s *MetricsSuite) TestPrometheus() {\n\tprom := &prometheus.Configuration{\n\t\tOnError:       \"panic\",\n\t\tTimerType:     \"histogram\",\n\t\tListenAddress: \"127.0.0.1:0\",\n\t}\n\tconfig := new(Metrics)\n\tconfig.Prometheus = prom\n\tscope := config.NewScope(testlogger.New(s.T()), \"test\")\n\ts.NotNil(scope)\n}\n\nfunc (s *MetricsSuite) TestNoop() {\n\tconfig := &Metrics{}\n\tscope := config.NewScope(testlogger.New(s.T()), \"test\")\n\ts.Equal(tally.NoopScope.Tagged(map[string]string{metrics.CadenceServiceTagName: \"test\"}), scope)\n}\n"
  },
  {
    "path": "common/config/persistence.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\nconst (\n\t// StoreTypeSQL refers to sql based storage as persistence store\n\tStoreTypeSQL = \"sql\"\n\t// StoreTypeCassandra refers to cassandra as persistence store\n\tStoreTypeCassandra = \"cassandra\"\n)\n\n// DefaultStoreType returns the storeType for the default persistence store\nfunc (c *Persistence) DefaultStoreType() string {\n\tif c.DataStores[c.DefaultStore].SQL != nil {\n\t\treturn StoreTypeSQL\n\t}\n\treturn StoreTypeCassandra\n}\n\n// FillDefaults populates default values for unspecified fields in persistence config\nfunc (c *Persistence) FillDefaults() {\n\tfor k, store := range c.DataStores {\n\t\tif store.Cassandra != nil && store.NoSQL == nil {\n\t\t\t// for backward-compatibility\n\t\t\tstore.NoSQL = store.Cassandra\n\t\t\tstore.NoSQL.PluginName = \"cassandra\"\n\t\t}\n\n\t\tif store.SQL != nil {\n\t\t\t// filling default encodingType/decodingTypes for SQL persistence\n\t\t\tif store.SQL.EncodingType == \"\" {\n\t\t\t\tstore.SQL.EncodingType = string(constants.EncodingTypeThriftRW)\n\t\t\t}\n\t\t\tif len(store.SQL.DecodingTypes) == 0 {\n\t\t\t\tstore.SQL.DecodingTypes = []string{\n\t\t\t\t\tstring(constants.EncodingTypeThriftRW),\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif store.SQL.NumShards == 0 {\n\t\t\t\tstore.SQL.NumShards = 1\n\t\t\t}\n\t\t}\n\n\t\t// write changes back to DataStores, as ds is a value object\n\t\tc.DataStores[k] = store\n\t}\n}\n\n// Validate validates the persistence config\nfunc (c *Persistence) Validate() error {\n\tdbStoreKeys := []string{c.DefaultStore}\n\n\tuseAdvancedVisibilityOnly := false\n\tif _, ok := c.DataStores[c.VisibilityStore]; ok {\n\t\tdbStoreKeys = append(dbStoreKeys, c.VisibilityStore)\n\t} else {\n\t\tif _, ok := c.DataStores[c.AdvancedVisibilityStore]; !ok {\n\t\t\treturn fmt.Errorf(\"must provide one of VisibilityStore and AdvancedVisibilityStore\")\n\t\t}\n\t\tuseAdvancedVisibilityOnly = true\n\t}\n\n\tfor _, st := range dbStoreKeys {\n\t\tds, ok := c.DataStores[st]\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"persistence config: missing config for datastore %v\", st)\n\t\t}\n\t\tif ds.Cassandra != nil && ds.NoSQL != nil && ds.Cassandra != ds.NoSQL {\n\t\t\treturn fmt.Errorf(\"persistence config: datastore %v: only one of Cassandra or NoSQL can be specified\", st)\n\t\t}\n\t\tconfigCount := 0\n\t\tif ds.NoSQL != nil {\n\t\t\tconfigCount++\n\t\t}\n\t\tif ds.ShardedNoSQL != nil {\n\t\t\tconfigCount++\n\t\t}\n\t\tif ds.SQL != nil {\n\t\t\tconfigCount++\n\t\t}\n\t\tif configCount != 1 {\n\t\t\treturn fmt.Errorf(\"persistence config: datastore %v: must provide exactly one type of config, but provided %d\", st, configCount)\n\t\t}\n\t\tif ds.SQL != nil {\n\t\t\tswitch {\n\t\t\tcase ds.SQL.UseMultipleDatabases:\n\t\t\t\tif !useAdvancedVisibilityOnly {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: multipleSQLDatabases can only be used with advanced visibility only\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.DatabaseName != \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: databaseName can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.ConnectAddr != \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: connectAddr can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.User != \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: user can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.Password != \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: password can only be configured in multipleDatabasesConfig when UseMultipleDatabases is true\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.NumShards <= 1 || len(ds.SQL.MultipleDatabasesConfig) != ds.SQL.NumShards {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: nShards must be greater than one and equal to the length of multipleDatabasesConfig\")\n\t\t\t\t}\n\t\t\t\tfor _, entry := range ds.SQL.MultipleDatabasesConfig {\n\t\t\t\t\tif entry.DatabaseName == \"\" {\n\t\t\t\t\t\treturn fmt.Errorf(\"sql multipleDatabasesConfig persistence config: databaseName can not be empty\")\n\t\t\t\t\t}\n\t\t\t\t\tif entry.ConnectAddr == \"\" {\n\t\t\t\t\t\treturn fmt.Errorf(\"sql multipleDatabasesConfig persistence config: connectAddr can not be empty\")\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t// SQLite plugin doesn't require ConnectAddr and DatabaseName\n\t\t\tcase ds.SQL.PluginName != \"sqlite\":\n\t\t\t\tif ds.SQL.DatabaseName == \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: databaseName can not be empty\")\n\t\t\t\t}\n\t\t\t\tif ds.SQL.ConnectAddr == \"\" {\n\t\t\t\t\treturn fmt.Errorf(\"sql persistence config: connectAddr can not be empty\")\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tif ds.ShardedNoSQL != nil {\n\t\t\tcfg := ds.ShardedNoSQL\n\t\t\tconnections := cfg.Connections\n\n\t\t\t// validate default shard\n\t\t\tif cfg.DefaultShard == \"\" {\n\t\t\t\treturn fmt.Errorf(\"ShardedNosql config: defaultShard can not be empty\")\n\t\t\t}\n\t\t\tif _, found := connections[cfg.DefaultShard]; !found {\n\t\t\t\treturn fmt.Errorf(\n\t\t\t\t\t\"ShardedNosql config: defaultShard (%v) is not defined in connections list\", cfg.DefaultShard)\n\t\t\t}\n\n\t\t\t// validate history sharding\n\t\t\thistoryShardMapping := cfg.ShardingPolicy.HistoryShardMapping\n\t\t\tcurrentShardID := 0\n\t\t\tfor _, shardRange := range historyShardMapping {\n\t\t\t\tif _, found := connections[shardRange.Shard]; !found {\n\t\t\t\t\treturn fmt.Errorf(\"ShardedNosql config: Unknown history shard name: %v\", shardRange.Shard)\n\t\t\t\t}\n\t\t\t\tif shardRange.Start != currentShardID {\n\t\t\t\t\treturn fmt.Errorf(\"ShardedNosql config: Non-continuous history shard range %v (%v) found while expecting %v\",\n\t\t\t\t\t\tshardRange.Start,\n\t\t\t\t\t\tshardRange.Shard,\n\t\t\t\t\t\tcurrentShardID,\n\t\t\t\t\t)\n\t\t\t\t}\n\t\t\t\tcurrentShardID = shardRange.End\n\t\t\t}\n\t\t\tif currentShardID != c.NumHistoryShards {\n\t\t\t\treturn fmt.Errorf(\"ShardedNosql config: Last history shard found in the config is %v while the max is %v\",\n\t\t\t\t\tcurrentShardID-1,\n\t\t\t\t\tc.NumHistoryShards-1,\n\t\t\t\t)\n\t\t\t}\n\n\t\t\t// validate tasklist sharding\n\t\t\ttasklistShards := cfg.ShardingPolicy.TaskListHashing.ShardOrder\n\t\t\tfor _, shardName := range tasklistShards {\n\t\t\t\tif _, found := connections[shardName]; !found {\n\t\t\t\t\treturn fmt.Errorf(\"ShardedNosql config: Unknown tasklist shard name: %v\", shardName)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// IsAdvancedVisibilityConfigExist returns whether user specified advancedVisibilityStore in config\nfunc (c *Persistence) IsAdvancedVisibilityConfigExist() bool {\n\treturn len(c.AdvancedVisibilityStore) != 0\n}\n"
  },
  {
    "path": "common/config/pinot.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\n// PinotVisibilityConfig for connecting to Pinot\ntype (\n\tPinotVisibilityConfig struct {\n\t\tCluster     string              `yaml:\"cluster\"`     //nolint:govet\n\t\tBroker      string              `yaml:\"broker\"`      //nolint:govet\n\t\tTable       string              `yaml:\"table\"`       //nolint:govet\n\t\tServiceName string              `yaml:\"serviceName\"` //nolint:govet\n\t\tMigration   VisibilityMigration `yaml:\"migration\"`   //nolint:govet\n\t}\n)\n"
  },
  {
    "path": "common/config/pprof.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\t\"net/http\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\n\t// DO NOT REMOVE THE LINE BELOW\n\t_ \"net/http/pprof\"\n)\n\ntype (\n\t// PProfInitializerImpl initialize the pprof based on config\n\tPProfInitializerImpl struct {\n\t\tPProf  *PProf\n\t\tLogger log.Logger\n\t}\n)\n\nconst (\n\tpprofNotInitialized int32 = 0\n\tpprofInitialized    int32 = 1\n)\n\n// the pprof should only be initialized once per process\n// otherwise, the caller / worker will experience weird issue\nvar pprofStatus = pprofNotInitialized\n\n// NewInitializer create a new instance of PProf Initializer\nfunc (cfg *PProf) NewInitializer(logger log.Logger) *PProfInitializerImpl {\n\treturn &PProfInitializerImpl{\n\t\tPProf:  cfg,\n\t\tLogger: logger,\n\t}\n}\n\n// Start the pprof based on config\nfunc (initializer *PProfInitializerImpl) Start() error {\n\tport := initializer.PProf.Port\n\tif port == 0 {\n\t\tinitializer.Logger.Info(\"PProf not started due to port not set\")\n\t\treturn nil\n\t}\n\n\thost := initializer.PProf.Host\n\tif host == \"\" {\n\t\thost = \"localhost\"\n\t}\n\n\tif atomic.CompareAndSwapInt32(&pprofStatus, pprofNotInitialized, pprofInitialized) {\n\t\tgo func() {\n\t\t\tinitializer.Logger.Info(\"PProf listen on \", tag.Dynamic(\"pprof-host\", host), tag.Port(port))\n\t\t\terr := http.ListenAndServe(net.JoinHostPort(host, fmt.Sprint(port)), nil)\n\t\t\tif err != nil {\n\t\t\t\tinitializer.Logger.Error(\"listen and serve err\", tag.Error(err))\n\t\t\t}\n\t\t}()\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/config/sasl.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\ntype (\n\t// SASL describe SASL configuration (for Kafka)\n\tSASL struct {\n\t\tEnabled   bool   `yaml:\"enabled\"` // false as default\n\t\tUser      string `yaml:\"user\"`\n\t\tPassword  string `yaml:\"password\"`\n\t\tAlgorithm string `yaml:\"algorithm\"` // plain, sha512 or sha256\n\t}\n)\n"
  },
  {
    "path": "common/config/tls.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"crypto/tls\"\n\t\"crypto/x509\"\n\t\"os\"\n)\n\ntype (\n\t// TLS describe TLS configuration\n\tTLS struct {\n\t\tEnabled bool `yaml:\"enabled\"`\n\n\t\t// For Postgres(https://www.postgresql.org/docs/9.1/libpq-ssl.html) and MySQL\n\t\t// default to require if Enable is true.\n\t\t// For MySQL: https://github.com/go-sql-driver/mysql , it also can be set in ConnectAttributes, default is tls-custom\n\t\tSSLMode string `yaml:\"sslmode\"`\n\n\t\t// CertPath and KeyPath are optional depending on server\n\t\t// config, but both fields must be omitted to avoid using a\n\t\t// client certificate\n\t\tCertFile string `yaml:\"certFile\"`\n\t\tKeyFile  string `yaml:\"keyFile\"`\n\n\t\tCaFile  string   `yaml:\"caFile\"` // optional depending on server config\n\t\tCaFiles []string `yaml:\"caFiles\"`\n\t\t// If you want to verify the hostname and server cert (like a wildcard for cass cluster) then you should turn this on\n\t\t// This option is basically the inverse of InSecureSkipVerify\n\t\t// See InSecureSkipVerify in http://golang.org/pkg/crypto/tls/ for more info\n\t\tEnableHostVerification bool `yaml:\"enableHostVerification\"`\n\n\t\t// Set RequireClientAuth to true if mutual TLS is desired.\n\t\t// In this mode, client will need to present their certificate which is signed by CA\n\t\t// that is specified with server \"caFile\"/\"caFiles\" or stored in server host level CA store.\n\t\tRequireClientAuth bool `yaml:\"requireClientAuth\"`\n\n\t\tServerName string `yaml:\"serverName\"`\n\t}\n)\n\n// ToTLSConfig converts Cadence TLS config to crypto/tls.Config\nfunc (config TLS) ToTLSConfig() (*tls.Config, error) {\n\tif !config.Enabled {\n\t\treturn nil, nil\n\t}\n\n\t// Setup base TLS config\n\t// EnableHostVerification is a secure flag vs insecureSkipVerify is insecure so inverse the value\n\ttlsConfig := &tls.Config{\n\t\tInsecureSkipVerify: !config.EnableHostVerification,\n\t}\n\n\t// Setup server name\n\tif config.ServerName != \"\" {\n\t\ttlsConfig.ServerName = config.ServerName\n\t}\n\n\t// Load CA certs\n\tcaFiles := config.CaFiles\n\tif config.CaFile != \"\" {\n\t\tcaFiles = append(caFiles, config.CaFile)\n\t}\n\n\tif len(caFiles) > 0 {\n\t\tcaCertPool := x509.NewCertPool()\n\t\tfor _, caFile := range caFiles {\n\t\t\tcaCert, err := os.ReadFile(caFile)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tcaCertPool.AppendCertsFromPEM(caCert)\n\t\t}\n\t\ttlsConfig.RootCAs = caCertPool\n\t}\n\n\t// Enable mutual TLS\n\tif config.RequireClientAuth {\n\t\ttlsConfig.ClientAuth = tls.RequireAndVerifyClientCert\n\t\ttlsConfig.ClientCAs = tlsConfig.RootCAs\n\t}\n\n\t// Load client cert\n\tif config.CertFile != \"\" && config.KeyFile != \"\" {\n\t\tcert, err := tls.LoadX509KeyPair(config.CertFile, config.KeyFile)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ttlsConfig.Certificates = []tls.Certificate{cert}\n\t}\n\n\treturn tlsConfig, nil\n}\n"
  },
  {
    "path": "common/constants/constants.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage constants\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/shadower\"\n)\n\nconst (\n\t// FirstEventID is the id of the first event in the history\n\tFirstEventID int64 = 1\n\t// EmptyEventID is the id of the empty event\n\tEmptyEventID int64 = -23\n\t// EmptyVersion is used as the default value for failover version when no value is provided\n\tEmptyVersion int64 = -24\n\t// EndEventID is the id of the end event, here we use the int64 max\n\tEndEventID int64 = 1<<63 - 1\n\t// BufferedEventID is the id of the buffered event\n\tBufferedEventID int64 = -123\n\t// EmptyEventTaskID is uninitialized id of the task id within event\n\tEmptyEventTaskID int64 = -1234\n\t// TransientEventID is the id of the transient event\n\tTransientEventID int64 = -124\n\t// FirstBlobPageToken is the page token identifying the first blob for each history archival\n\tFirstBlobPageToken = 1\n\t// LastBlobNextPageToken is the next page token on the last blob for each history archival\n\tLastBlobNextPageToken = -1\n\t// InclusiveEndMessageID is the id of the end message, here we use the int64 max -1 because in some place we need to convert it to exclusive end message id\n\tInclusiveEndMessageID int64 = 1<<63 - 2\n\t// EmptyMessageID is the default start message ID for replication level\n\tEmptyMessageID = -1\n\t// InitialPreviousFailoverVersion is the initial previous failover version\n\tInitialPreviousFailoverVersion int64 = -1\n)\n\nconst (\n\t// EmptyUUID is the placeholder for UUID when it's empty\n\tEmptyUUID = \"emptyUuid\"\n)\n\n// Data encoding types\nconst (\n\tEncodingTypeJSON           EncodingType = \"json\"\n\tEncodingTypeThriftRW       EncodingType = \"thriftrw\"\n\tEncodingTypeThriftRWSnappy EncodingType = \"thriftrw_snappy\"\n\tEncodingTypeGob            EncodingType = \"gob\"\n\tEncodingTypeUnknown        EncodingType = \"unknow\"\n\tEncodingTypeEmpty          EncodingType = \"\"\n\tEncodingTypeProto          EncodingType = \"proto3\"\n)\n\ntype (\n\t// EncodingType is an enum that represents various data encoding types\n\tEncodingType string\n)\n\n// MaxTaskTimeout is maximum task timeout allowed. 366 days in seconds\nconst MaxTaskTimeout = 31622400\n\nconst (\n\t// GetHistoryMaxPageSize is the max page size for get history\n\tGetHistoryMaxPageSize = 1000\n\t// ReadDLQMessagesPageSize is the max page size for read DLQ messages\n\tReadDLQMessagesPageSize = 1000\n)\n\nconst (\n\t// VisibilityAppName is used to find kafka topics and ES indexName for visibility\n\tVisibilityAppName      = \"visibility\"\n\tPinotVisibilityAppName = \"pinot-visibility\"\n)\n\nconst (\n\t// ESVisibilityStoreName is used to find es advanced visibility store\n\tESVisibilityStoreName = \"es-visibility\"\n\t// PinotVisibilityStoreName is used to find pinot advanced visibility store\n\tPinotVisibilityStoreName = \"pinot-visibility\"\n\t// OSVisibilityStoreName is used to find opensearch advanced visibility store\n\tOSVisibilityStoreName = \"os-visibility\"\n)\n\n// This was flagged by salus as potentially hardcoded credentials. This is a false positive by the scanner and should be\n// disregarded.\n// #nosec\nconst (\n\t// SystemGlobalDomainName is global domain name for cadence system workflows running globally\n\tSystemGlobalDomainName = \"cadence-system-global\"\n\t// SystemDomainID is domain id for all cadence system workflows\n\tSystemDomainID = \"32049b68-7872-4094-8e63-d0dd59896a83\"\n\t// SystemLocalDomainName is domain name for cadence system workflows running in local cluster\n\tSystemLocalDomainName = \"cadence-system\"\n\t// SystemDomainRetentionDays is retention config for all cadence system workflows\n\tSystemDomainRetentionDays = 7\n\t// BatcherDomainID is domain id for batcher local domain\n\tBatcherDomainID = \"3116607e-419b-4783-85fc-47726a4c3fe9\"\n\t// BatcherLocalDomainName is domain name for batcher workflows running in local cluster\n\t// Batcher cannot use SystemLocalDomain because auth\n\tBatcherLocalDomainName = \"cadence-batcher\"\n\t// ShadowerDomainID is domain id for workflow shadower local domain\n\tShadowerDomainID = \"59c51119-1b41-4a28-986d-d6e377716f82\"\n\t// ShadowerLocalDomainName\n\tShadowerLocalDomainName = shadower.LocalDomainName\n)\n\nconst (\n\t// MinLongPollTimeout is the minimum context timeout for long poll API, below which\n\t// the request won't be processed\n\tMinLongPollTimeout = time.Second * 2\n\t// CriticalLongPollTimeout is a threshold for the context timeout passed into long poll API,\n\t// below which a warning will be logged\n\tCriticalLongPollTimeout = time.Second * 20\n)\n\nconst (\n\t// DefaultTransactionSizeLimit is the largest allowed transaction size to persistence\n\tDefaultTransactionSizeLimit = 14 * 1024 * 1024\n)\n\nconst (\n\t// DefaultIDLengthWarnLimit is the warning length for various ID types\n\tDefaultIDLengthWarnLimit = 128\n\t// DefaultIDLengthErrorLimit is the maximum length allowed for various ID types\n\tDefaultIDLengthErrorLimit = 1000\n)\n\nconst (\n\t// ArchivalEnabled is the status for enabling archival\n\tArchivalEnabled = \"enabled\"\n\t// ArchivalDisabled is the status for disabling archival\n\tArchivalDisabled = \"disabled\"\n\t// ArchivalPaused is the status for pausing archival\n\tArchivalPaused = \"paused\"\n)\n\n// dynamic config AdvancedVisibility write/read mode\nconst (\n\t// AdvancedVisibilityModeOff means do not use advanced visibility store\n\tAdvancedVisibilityModeOff = \"off\"\n\t// AdvancedVisibilityModeES means ElasticSearch visibility mode\n\tVisibilityModeES = \"es\"\n\t// AdvancedVisibilityModePinot means Pinot visibility mode\n\tVisibilityModePinot = \"pinot\"\n\t// AdvancedVisibilityModeOS means OpenSearch visibility mode\n\tVisibilityModeOS = \"os\"\n\t// AdvancedVisibilityModeDB means db visibility mode\n\tVisibilityModeDB = \"db\"\n)\n\nconst (\n\tESPersistenceName    = \"elasticsearch\"\n\tPinotPersistenceName = \"pinot\"\n\tDBPersistenceName    = \"db\"\n)\n\nconst (\n\t// DomainDataKeyForManagedFailover is key of DomainData for managed failover\n\tDomainDataKeyForManagedFailover = \"IsManagedByCadence\"\n\t// DomainDataKeyForPreferredCluster is the key of DomainData for domain rebalance\n\tDomainDataKeyForPreferredCluster = \"PreferredCluster\"\n\t// DomainDataKeyForFailoverHistory is the key of DomainData for failover history\n\tDomainDataKeyForFailoverHistory = \"FailoverHistory\"\n\t// DomainDataKeyForReadGroups stores which groups have read permission of the domain API\n\tDomainDataKeyForReadGroups = \"READ_GROUPS\"\n\t// DomainDataKeyForWriteGroups stores which groups have write permission of the domain API\n\tDomainDataKeyForWriteGroups = \"WRITE_GROUPS\"\n\t// DomainDataKeyForProcessGroups stores which groups have process permission of the domain API\n\tDomainDataKeyForProcessGroups = \"PROCESS_GROUPS\"\n)\n\ntype (\n\t// TaskType is the enum for representing different task types\n\tTaskType int\n)\n\nconst (\n\t// TaskTypeTransfer is the task type for transfer task\n\t// starting from 2 here to be consistent with the row type define for cassandra\n\t// TODO: we can remove +2 from the following definition\n\t// we don't have to make them consistent with cassandra definition\n\t// there's also no row type for sql or other nosql persistence implementation\n\tTaskTypeTransfer TaskType = iota + 2\n\t// TaskTypeTimer is the task type for timer task\n\tTaskTypeTimer\n\t// TaskTypeReplication is the task type for replication task\n\tTaskTypeReplication\n\t// Deprecated: TaskTypeCrossCluster is the task type for cross cluster task\n\t// as of June 2024, this feature is no longer supported. Keeping the enum here\n\t// to avoid future reuse of the ID and/or confusion\n\tTaskTypeCrossCluster TaskType = 6\n)\n\nconst (\n\t// DefaultESAnalyzerPause controls if we want to dynamically pause the analyzer\n\tDefaultESAnalyzerPause = false\n\t// DefaultESAnalyzerTimeWindow controls how many days to go back for ElasticSearch Analyzer\n\tDefaultESAnalyzerTimeWindow = time.Hour * 24 * 30\n\t// DefaultESAnalyzerMaxNumDomains controls how many domains to check\n\tDefaultESAnalyzerMaxNumDomains = 500\n\t// DefaultESAnalyzerMaxNumWorkflowTypes controls how many workflow types per domain to check\n\tDefaultESAnalyzerMaxNumWorkflowTypes = 100\n\t// DefaultESAnalyzerNumWorkflowsToRefresh controls how many workflows per workflow type should be refreshed\n\tDefaultESAnalyzerNumWorkflowsToRefresh = 100\n\t// DefaultESAnalyzerBufferWaitTime controls min time required to consider a worklow stuck\n\tDefaultESAnalyzerBufferWaitTime = time.Minute * 30\n\t// DefaultESAnalyzerMinNumWorkflowsForAvg controls how many workflows to have at least to rely on workflow run time avg per type\n\tDefaultESAnalyzerMinNumWorkflowsForAvg = 100\n\t// DefaultESAnalyzerLimitToTypes controls if we want to limit ESAnalyzer only to some workflow types\n\tDefaultESAnalyzerLimitToTypes = \"\"\n\t// DefaultESAnalyzerEnableAvgDurationBasedChecks controls if we want to enable avg duration based refreshes\n\tDefaultESAnalyzerEnableAvgDurationBasedChecks = false\n\t// DefaultESAnalyzerLimitToDomains controls if we want to limit ESAnalyzer only to some domains\n\tDefaultESAnalyzerLimitToDomains = \"\"\n\t// DefaultESAnalyzerWorkflowDurationWarnThreshold defines warning threshold for a workflow duration\n\tDefaultESAnalyzerWorkflowDurationWarnThresholds = \"\"\n)\n\n// StickyTaskConditionFailedErrorMsg error msg for sticky task ConditionFailedError\nconst StickyTaskConditionFailedErrorMsg = \"StickyTaskConditionFailedError\"\n\n// MemoKeyForOperator is the memo key for operator\nconst MemoKeyForOperator = \"operator\"\n\n// ReservedTaskListPrefix is the required naming prefix for any task list partition other than partition 0\nconst ReservedTaskListPrefix = \"/__cadence_sys/\"\n\ntype (\n\t// VisibilityOperation is an enum that represents visibility message types\n\tVisibilityOperation string\n)\n\n// Enum for visibility message type\nconst (\n\tRecordStarted          VisibilityOperation = \"RecordStarted\"\n\tRecordClosed           VisibilityOperation = \"RecordClosed\"\n\tUpsertSearchAttributes VisibilityOperation = \"UpsertSearchAttributes\"\n)\n\nconst (\n\tnumBitsPerLevel = 3\n)\n\nconst (\n\t// HighPriorityClass is the priority class for high priority tasks\n\tHighPriorityClass = iota << numBitsPerLevel\n\t// DefaultPriorityClass is the priority class for default priority tasks\n\tDefaultPriorityClass\n\t// LowPriorityClass is the priority class for low priority tasks\n\tLowPriorityClass\n)\n\nconst (\n\t// HighPrioritySubclass is the priority subclass for high priority tasks\n\tHighPrioritySubclass = iota\n\t// DefaultPrioritySubclass is the priority subclass for high priority tasks\n\tDefaultPrioritySubclass\n\t// LowPrioritySubclass is the priority subclass for high priority tasks\n\tLowPrioritySubclass\n)\n\nconst (\n\t// DefaultHistoryMaxAutoResetPoints is the default maximum number for auto reset points\n\tDefaultHistoryMaxAutoResetPoints = 20\n)\n\nconst (\n\t// WorkflowIDRateLimitReason is the reason set in ServiceBusyError when workflow ID rate limit is exceeded\n\tWorkflowIDRateLimitReason = \"external-workflow-id-rate-limit\"\n)\n\ntype (\n\t// FailoverType is the enum for representing different failover types\n\tFailoverType int\n)\n\nconst (\n\tFailoverTypeForce FailoverType = iota + 1\n\tFailoverTypeGrace\n)\n\nfunc (v FailoverType) String() string {\n\tswitch v {\n\tcase FailoverTypeForce:\n\t\treturn \"Force\"\n\tcase FailoverTypeGrace:\n\t\treturn \"Grace\"\n\tdefault:\n\t\treturn \"Unknown\"\n\t}\n}\n\nconst (\n\tStringSizeOverheadBytes = 16\n)\n\n// GetTaskPriority returns priority given a task's priority class and subclass\nfunc GetTaskPriority(\n\tclass int,\n\tsubClass int,\n) int {\n\treturn class | subClass\n}\n\n// GRPCConnectionClosingError is the error message returned when a gRPC client connection is closing\nconst GRPCConnectionClosingError = \"grpc: the client connection is closing\"\n"
  },
  {
    "path": "common/convert.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\n\ts \"github.com/uber/cadence/.gen/go/shared\"\n)\n\n// Returns a pointer to the given value.\n// todo (david.porter) Remove the remaining helpers here which are now obsolete\nfunc Ptr[T any](v T) *T {\n\treturn &v\n}\n\n// IntPtr makes a copy and returns the pointer to an int.\nfunc IntPtr(v int) *int {\n\treturn &v\n}\n\n// Int16Ptr makes a copy and returns the pointer to an int16.\nfunc Int16Ptr(v int16) *int16 {\n\treturn &v\n}\n\n// Int32Ptr makes a copy and returns the pointer to an int32.\nfunc Int32Ptr(v int32) *int32 {\n\treturn &v\n}\n\n// Int64Ptr makes a copy and returns the pointer to an int64.\nfunc Int64Ptr(v int64) *int64 {\n\treturn &v\n}\n\n// Uint32Ptr makes a copy and returns the pointer to a uint32.\nfunc Uint32Ptr(v uint32) *uint32 {\n\treturn &v\n}\n\n// Uint64Ptr makes a copy and returns the pointer to a uint64.\nfunc Uint64Ptr(v uint64) *uint64 {\n\treturn &v\n}\n\n// Float64Ptr makes a copy and returns the pointer to an int64.\nfunc Float64Ptr(v float64) *float64 {\n\treturn &v\n}\n\n// BoolPtr makes a copy and returns the pointer to a bool.\nfunc BoolPtr(v bool) *bool {\n\treturn &v\n}\n\n// StringPtr makes a copy and returns the pointer to a string.\nfunc StringPtr(v string) *string {\n\treturn &v\n}\n\n// TimeNowNanosPtr returns an int64 ptr to current time in unix nanos\nfunc TimeNowNanosPtr() *int64 {\n\tv := time.Now().UnixNano()\n\treturn &v\n}\n\n// TimePtr makes a copy and returns the pointer to a time\nfunc TimePtr(v time.Time) *time.Time {\n\treturn &v\n}\n\n// DurationPtr makes a copy and returns the pointer to a duration\nfunc DurationPtr(v time.Duration) *time.Duration {\n\treturn &v\n}\n\n// TaskListPtr makes a copy and returns the pointer to a TaskList.\nfunc TaskListPtr(v s.TaskList) *s.TaskList {\n\treturn &v\n}\n\n// ActivityTypePtr makes a copy and returns the pointer to a ActivityType.\nfunc ActivityTypePtr(v s.ActivityType) *s.ActivityType {\n\treturn &v\n}\n\n// DecisionTypePtr makes a copy and returns the pointer to a DecisionType.\nfunc DecisionTypePtr(t s.DecisionType) *s.DecisionType {\n\treturn &t\n}\n\n// ParentClosePolicyPtr makes a copy and returns the pointer to a DecisionType.\nfunc ParentClosePolicyPtr(t s.ParentClosePolicy) *s.ParentClosePolicy {\n\treturn &t\n}\n\n// EventTypePtr makes a copy and returns the pointer to a EventType.\nfunc EventTypePtr(t s.EventType) *s.EventType {\n\treturn &t\n}\n\n// WorkflowTypePtr makes a copy and returns the pointer to a WorkflowType.\nfunc WorkflowTypePtr(t s.WorkflowType) *s.WorkflowType {\n\treturn &t\n}\n\n// TimeoutTypePtr makes a copy and returns the pointer to a TimeoutType.\nfunc TimeoutTypePtr(t s.TimeoutType) *s.TimeoutType {\n\treturn &t\n}\n\n// TaskListKindPtr makes a copy and returns the pointer to a TaskListKind.\nfunc TaskListKindPtr(t s.TaskListKind) *s.TaskListKind {\n\treturn &t\n}\n\n// TaskListTypePtr makes a copy and returns the pointer to a TaskListKind.\nfunc TaskListTypePtr(t s.TaskListType) *s.TaskListType {\n\treturn &t\n}\n\n// DecisionTaskFailedCausePtr makes a copy and returns the pointer to a DecisionTaskFailedCause.\nfunc DecisionTaskFailedCausePtr(t s.DecisionTaskFailedCause) *s.DecisionTaskFailedCause {\n\treturn &t\n}\n\n// CancelExternalWorkflowExecutionFailedCausePtr makes a copy and returns the pointer to a CancelExternalWorkflowExecutionFailedCause.\nfunc CancelExternalWorkflowExecutionFailedCausePtr(t s.CancelExternalWorkflowExecutionFailedCause) *s.CancelExternalWorkflowExecutionFailedCause {\n\treturn &t\n}\n\n// SignalExternalWorkflowExecutionFailedCausePtr makes a copy and returns the pointer to a SignalExternalWorkflowExecutionFailedCause.\nfunc SignalExternalWorkflowExecutionFailedCausePtr(t s.SignalExternalWorkflowExecutionFailedCause) *s.SignalExternalWorkflowExecutionFailedCause {\n\treturn &t\n}\n\n// ChildWorkflowExecutionFailedCausePtr makes a copy and returns the pointer to a ChildWorkflowExecutionFailedCause.\nfunc ChildWorkflowExecutionFailedCausePtr(t s.ChildWorkflowExecutionFailedCause) *s.ChildWorkflowExecutionFailedCause {\n\treturn &t\n}\n\n// ArchivalStatusPtr makes a copy and returns the pointer to an ArchivalStatus.\nfunc ArchivalStatusPtr(t s.ArchivalStatus) *s.ArchivalStatus {\n\treturn &t\n}\n\n// ClientArchivalStatusPtr makes a copy and returns the pointer to a client ArchivalStatus.\nfunc ClientArchivalStatusPtr(t shared.ArchivalStatus) *shared.ArchivalStatus {\n\treturn &t\n}\n\n// QueryResultTypePtr makes a copy and returns the pointer to a QueryResultType\nfunc QueryResultTypePtr(t s.QueryResultType) *s.QueryResultType {\n\treturn &t\n}\n\n// QueryRejectConditionPtr makes a copy and returns the pointer to a QueryRejectCondition\nfunc QueryRejectConditionPtr(t s.QueryRejectCondition) *s.QueryRejectCondition {\n\treturn &t\n}\n\n// QueryConsistencyLevelPtr makes a copy and returns the pointer to a QueryConsistencyLevel\nfunc QueryConsistencyLevelPtr(t s.QueryConsistencyLevel) *s.QueryConsistencyLevel {\n\treturn &t\n}\n\n// QueryTaskCompletedTypePtr makes a copy and returns the pointer to a QueryTaskCompletedType\nfunc QueryTaskCompletedTypePtr(t s.QueryTaskCompletedType) *s.QueryTaskCompletedType {\n\treturn &t\n}\n\n// StringDefault returns value if string pointer is set otherwise default value of string\nfunc StringDefault(v *string) string {\n\tvar defaultString string\n\tif v == nil {\n\t\treturn defaultString\n\t}\n\treturn *v\n}\n\n// Int32Default returns value if int32 pointer is set otherwise default value of int32\nfunc Int32Default(v *int32) int32 {\n\tvar defaultInt32 int32\n\tif v == nil {\n\t\treturn defaultInt32\n\t}\n\treturn *v\n}\n\n// Int64Default returns value if int64 pointer is set otherwise default value of int64\nfunc Int64Default(v *int64) int64 {\n\tvar defaultInt64 int64\n\tif v == nil {\n\t\treturn defaultInt64\n\t}\n\treturn *v\n}\n\n// BoolDefault returns value if bool pointer is set otherwise default value of bool\nfunc BoolDefault(v *bool) bool {\n\tvar defaultBool bool\n\tif v == nil {\n\t\treturn defaultBool\n\t}\n\treturn *v\n}\n"
  },
  {
    "path": "common/convert_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestPtr(t *testing.T) {\n\n\taString := \"a string\"\n\taStruct := struct{}{}\n\tanInterface := interface{}(nil)\n\n\tt.Run(\"a string should be converted to a pointer of a string\", func(t *testing.T) {\n\t\tassert.Equal(t, &aString, Ptr(aString))\n\t})\n\n\tt.Run(\"a struct should be converted to a pointer of a string\", func(t *testing.T) {\n\t\tassert.Equal(t, &aStruct, Ptr(aStruct))\n\t})\n\n\tt.Run(\"a interface should be converted to a pointer of a string\", func(t *testing.T) {\n\t\tassert.Equal(t, &anInterface, Ptr(anInterface))\n\t})\n}\n"
  },
  {
    "path": "common/ctxutils/ctxutils.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ctxutils\n\nimport (\n\t\"context\"\n\t\"sync\"\n)\n\n// WithPropagatedContextCancel returns a copy of parent which is cancelled whenever\n// it is cancelled itself (with the returned cancel func) or cancelCtx is cancelled\n// you need to call cancel func to avoid potential goroutine leak\nfunc WithPropagatedContextCancel(parent context.Context, cancelCtx context.Context) (context.Context, context.CancelFunc) {\n\tdone := cancelCtx.Done()\n\tif done == nil {\n\t\treturn parent, func() {}\n\t}\n\n\tchildWithCancel, cancel := context.WithCancel(parent)\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\n\tgo func() {\n\t\tdefer wg.Done()\n\n\t\tselect {\n\t\tcase <-done:\n\t\t\tcancel()\n\t\tcase <-childWithCancel.Done():\n\t\t}\n\t}()\n\n\treturn childWithCancel, func() {\n\t\tcancel()\n\t\twg.Wait()\n\t}\n}\n"
  },
  {
    "path": "common/ctxutils/ctxutils_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ctxutils\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n)\n\nfunc TestWithPropagatedContextCancel(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tcancelCtx, cancelFn := context.WithCancel(context.Background())\n\n\tctx, stopper := WithPropagatedContextCancel(context.Background(), cancelCtx)\n\tdefer stopper()\n\n\tcancelFn() // cancel should be propagated to \"chained\" ctx\n\n\ttestTimeout := time.NewTimer(10 * time.Second)\n\tcancelledOriginalContext := false\n\tselect {\n\tcase <-ctx.Done():\n\t\tcancelledOriginalContext = true\n\tcase <-testTimeout.C:\n\t\tt.Error(\"test timed out\")\n\t}\n\n\trequire.True(t, cancelledOriginalContext)\n}\n\nfunc TestWithPropagatedContextCancelWorksWhenContextHasNoCancellation(t *testing.T) {\n\tassert.NotPanics(t, func() {\n\t\tWithPropagatedContextCancel(context.Background(), context.Background())\n\t})\n\n\tgoleak.VerifyNone(t)\n}\n\nfunc TestWithPropagatedContextCancelWorksWhenParentContextIsCancelled(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tcancelCtx1, cancelFn1 := context.WithCancel(context.Background())\n\tcancelCtx2, _ := context.WithCancel(context.Background())\n\n\tctx, stopper := WithPropagatedContextCancel(cancelCtx1, cancelCtx2)\n\tdefer stopper()\n\n\t// in this case there is no need to wait for the cancelCtx to cancel\n\t// since the parent context is closed before\n\tcancelFn1()\n\n\ttestTimeout := time.NewTimer(10 * time.Second)\n\tselect {\n\tcase <-ctx.Done():\n\tcase <-testTimeout.C:\n\t\tt.Error(\"sanity check #1 - we expect ctx to be cancelled\")\n\t}\n\n\tselect {\n\tcase <-cancelCtx1.Done():\n\tcase <-testTimeout.C:\n\t\tt.Error(\"sanity check #2 - we expect ctx to be cancelled\")\n\t}\n}\n\nfunc TestWithPropagatedContextSanityCheckOriginalContextNeverAffected(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\toriginalCtx, _ := context.WithCancel(context.Background())\n\tcancelCtx, cancelFn := context.WithCancel(context.Background())\n\n\tctx, stopper := WithPropagatedContextCancel(originalCtx, cancelCtx)\n\n\tcancelFn() // cancel should be propagated to \"chained\" ctx\n\tstopper()\n\n\ttestTimeout := time.NewTimer(10 * time.Second)\n\n\t// cancellation of ctx is expected\n\tselect {\n\tcase <-ctx.Done():\n\t\tbreak\n\tcase <-testTimeout.C:\n\t\tt.Error(\"test timed out\")\n\t}\n\n\t// ... but cancellation of originalCtx is not\n\t// actually this is guaranteed by Go (contexts are immutable), but let's check anyway\n\tselect {\n\tcase <-originalCtx.Done():\n\t\tt.Error(\"originalCtx should not be cancelled\")\n\tdefault:\n\t}\n}\n"
  },
  {
    "path": "common/daemon.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nconst (\n\t// used for background threads\n\n\t// DaemonStatusInitialized coroutine pool initialized\n\tDaemonStatusInitialized int32 = 0\n\t// DaemonStatusStarted coroutine pool started\n\tDaemonStatusStarted int32 = 1\n\t// DaemonStatusStopped coroutine pool stopped\n\tDaemonStatusStopped int32 = 2\n)\n\ntype (\n\t// Daemon is the base interfaces implemented by\n\t// background tasks within Cadence\n\tDaemon interface {\n\t\tStart()\n\t\tStop()\n\t}\n)\n"
  },
  {
    "path": "common/definition/indexedKeys.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage definition\n\nimport \"github.com/uber/cadence/common/types\"\n\n// valid indexed fields on ES\nconst (\n\tDomainID               = \"DomainID\"\n\tWorkflowID             = \"WorkflowID\"\n\tRunID                  = \"RunID\"\n\tWorkflowType           = \"WorkflowType\"\n\tStartTime              = \"StartTime\"\n\tExecutionTime          = \"ExecutionTime\"\n\tCloseTime              = \"CloseTime\"\n\tCloseStatus            = \"CloseStatus\"\n\tHistoryLength          = \"HistoryLength\"\n\tEncoding               = \"Encoding\"\n\tKafkaKey               = \"KafkaKey\"\n\tBinaryChecksums        = \"BinaryChecksums\"\n\tTaskList               = \"TaskList\"\n\tClusterAttributeScope  = \"ClusterAttributeScope\"\n\tClusterAttributeName   = \"ClusterAttributeName\"\n\tIsCron                 = \"IsCron\"\n\tNumClusters            = \"NumClusters\"\n\tUpdateTime             = \"UpdateTime\"\n\tCronSchedule           = \"CronSchedule\"\n\tExecutionStatus        = \"ExecutionStatus\"\n\tScheduledExecutionTime = \"ScheduledExecutionTime\"\n\tCustomDomain           = \"CustomDomain\" // to support batch workflow\n\tOperator               = \"Operator\"     // to support batch workflow\n\n\tCustomStringField    = \"CustomStringField\"\n\tCustomKeywordField   = \"CustomKeywordField\"\n\tCustomIntField       = \"CustomIntField\"\n\tCustomBoolField      = \"CustomBoolField\"\n\tCustomDoubleField    = \"CustomDoubleField\"\n\tCustomDatetimeField  = \"CustomDatetimeField\"\n\tCadenceChangeVersion = \"CadenceChangeVersion\"\n)\n\nconst (\n\t// Memo is valid non-indexed fields on ES\n\tMemo = \"Memo\"\n\t// Attr is prefix of custom search attributes\n\tAttr = \"Attr\"\n\t// HeaderFormat is the format of context headers in search attributes\n\tHeaderFormat = \"Header_%s\"\n)\n\n// defaultIndexedKeys defines all searchable keys\nvar defaultIndexedKeys = createDefaultIndexedKeys()\n\nfunc createDefaultIndexedKeys() map[string]interface{} {\n\tdefaultIndexedKeys := map[string]interface{}{\n\t\tCustomStringField:    types.IndexedValueTypeString,\n\t\tCustomKeywordField:   types.IndexedValueTypeKeyword,\n\t\tCustomIntField:       types.IndexedValueTypeInt,\n\t\tCustomBoolField:      types.IndexedValueTypeBool,\n\t\tCustomDoubleField:    types.IndexedValueTypeDouble,\n\t\tCustomDatetimeField:  types.IndexedValueTypeDatetime,\n\t\tCadenceChangeVersion: types.IndexedValueTypeKeyword,\n\t\tBinaryChecksums:      types.IndexedValueTypeKeyword,\n\t\tCustomDomain:         types.IndexedValueTypeString,\n\t\tOperator:             types.IndexedValueTypeString,\n\t}\n\tfor k, v := range systemIndexedKeys {\n\t\tdefaultIndexedKeys[k] = v\n\t}\n\treturn defaultIndexedKeys\n}\n\n// GetDefaultIndexedKeys return default valid indexed keys\nfunc GetDefaultIndexedKeys() map[string]interface{} {\n\treturn defaultIndexedKeys\n}\n\n// systemIndexedKeys is Cadence created visibility keys\nvar systemIndexedKeys = map[string]interface{}{\n\tDomainID:               types.IndexedValueTypeKeyword,\n\tWorkflowID:             types.IndexedValueTypeKeyword,\n\tRunID:                  types.IndexedValueTypeKeyword,\n\tWorkflowType:           types.IndexedValueTypeKeyword,\n\tStartTime:              types.IndexedValueTypeInt,\n\tExecutionTime:          types.IndexedValueTypeInt,\n\tCloseTime:              types.IndexedValueTypeInt,\n\tCloseStatus:            types.IndexedValueTypeInt,\n\tHistoryLength:          types.IndexedValueTypeInt,\n\tTaskList:               types.IndexedValueTypeKeyword,\n\tIsCron:                 types.IndexedValueTypeBool,\n\tNumClusters:            types.IndexedValueTypeInt,\n\tUpdateTime:             types.IndexedValueTypeInt,\n\tCronSchedule:           types.IndexedValueTypeKeyword,\n\tExecutionStatus:        types.IndexedValueTypeInt,\n\tScheduledExecutionTime: types.IndexedValueTypeInt,\n\tClusterAttributeScope:  types.IndexedValueTypeKeyword,\n\tClusterAttributeName:   types.IndexedValueTypeKeyword,\n}\n\n// IsSystemIndexedKey return true is key is system added\nfunc IsSystemIndexedKey(key string) bool {\n\t_, ok := systemIndexedKeys[key]\n\treturn ok\n}\n\n// IsSystemBoolKey return true is key is system added bool key\nfunc IsSystemBoolKey(key string) bool {\n\treturn systemIndexedKeys[key] == types.IndexedValueTypeBool\n}\n"
  },
  {
    "path": "common/definition/indexedKeys_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage definition\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestIsSystemBoolKey(t *testing.T) {\n\ttests := []struct {\n\t\tkey      string\n\t\texpected bool\n\t}{\n\t\t{\"IsCron\", true},\n\t\t{\"StartTime\", false},\n\t}\n\n\tfor _, test := range tests {\n\t\tactualResult := IsSystemBoolKey(test.key)\n\t\tassert.Equal(t, test.expected, actualResult)\n\t}\n}\n"
  },
  {
    "path": "common/definition/resourceDeduplication.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage definition\n\nimport \"fmt\"\n\nconst (\n\tresourceIDTemplate       = \"%v::%v\"\n\teventReappliedIDTemplate = \"%v::%v::%v\"\n)\n\ntype (\n\t// DeduplicationID uses to generate id for deduplication\n\tDeduplicationID interface {\n\t\tGetID() string\n\t}\n)\n\n// Deduplication id type\nconst (\n\teventReappliedID = iota\n)\n\n// Deduplication resource struct\ntype (\n\t// EventReappliedID is the deduplication resource for reapply event\n\tEventReappliedID struct {\n\t\tid string\n\t}\n)\n\n// NewEventReappliedID returns EventReappliedID resource\nfunc NewEventReappliedID(\n\trunID string,\n\teventID int64,\n\tversion int64,\n) EventReappliedID {\n\n\tnewID := fmt.Sprintf(\n\t\teventReappliedIDTemplate,\n\t\trunID,\n\t\teventID,\n\t\tversion,\n\t)\n\treturn EventReappliedID{\n\t\tid: newID,\n\t}\n}\n\n// GetID returns id of EventReappliedID\nfunc (e EventReappliedID) GetID() string {\n\treturn e.id\n}\n\n// GenerateDeduplicationKey generates deduplication key\nfunc GenerateDeduplicationKey(\n\tresource DeduplicationID,\n) string {\n\n\tswitch resource.(type) {\n\tcase EventReappliedID:\n\t\treturn generateKey(eventReappliedID, resource.GetID())\n\tdefault:\n\t\tpanic(\"unsupported deduplication key\")\n\t}\n}\n\nfunc generateKey(resourceType int32, id string) string {\n\treturn fmt.Sprintf(resourceIDTemplate, resourceType, id)\n}\n"
  },
  {
    "path": "common/definition/resourceDeduplication_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage definition\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tresourceDeduplicationSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestResourceDeduplicationSuite(t *testing.T) {\n\ts := new(resourceDeduplicationSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *resourceDeduplicationSuite) TestGenerateKey() {\n\tresourceType := int32(1)\n\tid := \"id\"\n\tkey := generateKey(resourceType, id)\n\ts.Equal(fmt.Sprintf(\"%v::%v\", resourceType, id), key)\n}\n\nfunc (s *resourceDeduplicationSuite) TestGenerateDeduplicationKey() {\n\trunID := \"runID\"\n\teventID := int64(1)\n\tversion := int64(2)\n\tresource := NewEventReappliedID(runID, eventID, version)\n\tkey := GenerateDeduplicationKey(resource)\n\ts.Equal(fmt.Sprintf(\"%v::%v::%v::%v\", eventReappliedID, runID, eventID, version), key)\n}\n\nfunc (s *resourceDeduplicationSuite) TestEventReappliedID() {\n\trunID := \"runID\"\n\teventID := int64(1)\n\tversion := int64(2)\n\tresource := NewEventReappliedID(runID, eventID, version)\n\ts.Equal(fmt.Sprintf(\"%v::%v::%v\", runID, eventID, version), resource.GetID())\n}\n"
  },
  {
    "path": "common/definition/workflowIdentifier.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage definition\n\nimport (\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype (\n\t// WorkflowIdentifier is the combinations which represent a workflow\n\tWorkflowIdentifier struct {\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t}\n)\n\n// Size calculates the size in bytes of the WorkflowIdentifier struct.\nfunc (wi *WorkflowIdentifier) ByteSize() uint64 {\n\t// Calculate the size of strings in bytes, we assume that all those fields are using ASCII which is 1 byte per char\n\tsize := len(wi.DomainID) + len(wi.WorkflowID) + len(wi.RunID)\n\t// Each string internally holds a reference pointer and a length, which are 8 bytes each\n\tstringOverhead := 3 * constants.StringSizeOverheadBytes\n\treturn uint64(size + stringOverhead)\n}\n\n// NewWorkflowIdentifier create a new WorkflowIdentifier\nfunc NewWorkflowIdentifier(domainID string, workflowID string, runID string) WorkflowIdentifier {\n\treturn WorkflowIdentifier{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n}\n"
  },
  {
    "path": "common/definition/workflowidentifier_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage definition\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\n// TestWorkflowIdentifierSize verifies the Size method of WorkflowIdentifier.\nfunc TestWorkflowIdentifierSize(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\twi       WorkflowIdentifier\n\t\texpected uint64\n\t}{\n\t\t{\n\t\t\tname:     \"non-empty fields\",\n\t\t\twi:       NewWorkflowIdentifier(\"domain\", \"workflow\", \"run\"),\n\t\t\texpected: uint64(len(\"domain\") + len(\"workflow\") + len(\"run\") + 3*16),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty fields\",\n\t\t\twi:       NewWorkflowIdentifier(\"\", \"\", \"\"),\n\t\t\texpected: uint64(3 * 16),\n\t\t},\n\t\t{\n\t\t\tname:     \"short fields\",\n\t\t\twi:       NewWorkflowIdentifier(\"a\", \"b\", \"c\"),\n\t\t\texpected: uint64(3 + 3*16),\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tsize := test.wi.ByteSize()\n\t\t\tassert.Equal(t, test.expected, size)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/archivalConfigStateMachine.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport \"github.com/uber/cadence/common/types\"\n\n// domainArchivalConfigStateMachine is only used by domainHandler.\n// It is simply meant to simplify the logic around archival domain state changes.\n// Logically this class can be thought of as part of domainHandler.\n\ntype (\n\t// ArchivalState represents the state of archival config\n\t// the only invalid state is {URI=\"\", status=enabled}\n\t// once URI is set it is immutable\n\tArchivalState struct {\n\t\tStatus types.ArchivalStatus\n\t\tURI    string\n\t}\n\n\t// ArchivalEvent represents a change request to archival config state\n\t// the only restriction placed on events is that defaultURI is not empty\n\t// status can be nil, enabled, or disabled (nil indicates no update by user is being attempted)\n\tArchivalEvent struct {\n\t\tdefaultURI string\n\t\tURI        string\n\t\tstatus     *types.ArchivalStatus\n\t}\n)\n\n// the following errors represents impossible code states that should never occur\nvar (\n\terrInvalidState            = &types.BadRequestError{Message: \"Encountered illegal state: archival is enabled but URI is not set (should be impossible)\"}\n\terrInvalidEvent            = &types.BadRequestError{Message: \"Encountered illegal event: default URI is not set (should be impossible)\"}\n\terrCannotHandleStateChange = &types.BadRequestError{Message: \"Encountered current state and event that cannot be handled (should be impossible)\"}\n\terrURIUpdate               = &types.BadRequestError{Message: \"Cannot update existing archival URI\"}\n)\n\nfunc neverEnabledState() *ArchivalState {\n\treturn &ArchivalState{\n\t\tURI:    \"\",\n\t\tStatus: types.ArchivalStatusDisabled,\n\t}\n}\n\nfunc (e *ArchivalEvent) validate() error {\n\tif len(e.defaultURI) == 0 {\n\t\treturn errInvalidEvent\n\t}\n\treturn nil\n}\n\nfunc (s *ArchivalState) validate() error {\n\tif s.Status == types.ArchivalStatusEnabled && len(s.URI) == 0 {\n\t\treturn errInvalidState\n\t}\n\treturn nil\n}\n\nfunc (s *ArchivalState) getNextState(\n\te *ArchivalEvent,\n\tURIValidationFunc func(URI string) error,\n) (nextState *ArchivalState, changed bool, err error) {\n\tdefer func() {\n\t\t// ensure that any existing URI name was not mutated\n\t\tif nextState != nil && len(s.URI) != 0 && s.URI != nextState.URI {\n\t\t\tnextState = nil\n\t\t\tchanged = false\n\t\t\terr = errCannotHandleStateChange\n\t\t\treturn\n\t\t}\n\n\t\t// ensure that next state is valid\n\t\tif nextState != nil {\n\t\t\tif nextStateErr := nextState.validate(); nextStateErr != nil {\n\t\t\t\tnextState = nil\n\t\t\t\tchanged = false\n\t\t\t\terr = nextStateErr\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\n\t\tif nextState != nil && nextState.URI != \"\" {\n\t\t\tif validateURIErr := URIValidationFunc(nextState.URI); validateURIErr != nil {\n\t\t\t\tnextState = nil\n\t\t\t\tchanged = false\n\t\t\t\terr = validateURIErr\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n\n\tif s == nil || e == nil {\n\t\treturn nil, false, errCannotHandleStateChange\n\t}\n\tif err := s.validate(); err != nil {\n\t\treturn nil, false, err\n\t}\n\tif err := e.validate(); err != nil {\n\t\treturn nil, false, err\n\t}\n\n\t/**\n\tAt this point state and event are both non-nil and valid.\n\n\tState can be any one of the following:\n\t{status=enabled,  URI=\"foo\"}\n\t{status=disabled, URI=\"foo\"}\n\t{status=disabled, URI=\"\"}\n\n\tEvent can be any one of the following:\n\t{status=enabled,  URI=\"foo\", defaultURI=\"bar\"}\n\t{status=enabled,  URI=\"\",    defaultURI=\"bar\"}\n\t{status=disabled, URI=\"foo\", defaultURI=\"bar\"}\n\t{status=disabled, URI=\"\",    defaultURI=\"bar\"}\n\t{status=nil,      URI=\"foo\", defaultURI=\"bar\"}\n\t{status=nil,      URI=\"\",    defaultURI=\"bar\"}\n\t*/\n\n\tstateURISet := len(s.URI) != 0\n\teventURISet := len(e.URI) != 0\n\n\t// factor this case out to ensure that URI is immutable\n\tif stateURISet && eventURISet && s.URI != e.URI {\n\t\treturn nil, false, errURIUpdate\n\t}\n\n\t// state 1\n\tif s.Status == types.ArchivalStatusEnabled && stateURISet {\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusDisabled,\n\t\t\t\tURI:    s.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && !eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusDisabled,\n\t\t\t\tURI:    s.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status == nil && eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status == nil && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t}\n\n\t// state 2\n\tif s.Status == types.ArchivalStatusDisabled && stateURISet {\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tURI:    s.URI,\n\t\t\t\tStatus: types.ArchivalStatusEnabled,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && !eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusEnabled,\n\t\t\t\tURI:    s.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status == nil && eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status == nil && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t}\n\n\t// state 3\n\tif s.Status == types.ArchivalStatusDisabled && !stateURISet {\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusEnabled,\n\t\t\t\tURI:    e.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusEnabled && !eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusEnabled,\n\t\t\t\tURI:    e.defaultURI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusDisabled,\n\t\t\t\tURI:    e.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status != nil && *e.status == types.ArchivalStatusDisabled && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t\tif e.status == nil && eventURISet {\n\t\t\treturn &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusDisabled,\n\t\t\t\tURI:    e.URI,\n\t\t\t}, true, nil\n\t\t}\n\t\tif e.status == nil && !eventURISet {\n\t\t\treturn s, false, nil\n\t\t}\n\t}\n\treturn nil, false, errCannotHandleStateChange\n}\n"
  },
  {
    "path": "common/domain/archivalConfigStateMachine_test.go",
    "content": "// Copyright (c) 2023 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestArchivalStateGetNextState(t *testing.T) {\n\tenabled := types.ArchivalStatusEnabled\n\tdisabled := types.ArchivalStatusDisabled\n\t// Mock URI validation function that always succeeds\n\tsuccessfulValidation := func(URI string) error { return nil }\n\n\ttests := []struct {\n\t\tname            string\n\t\tcurrentState    *ArchivalState\n\t\tevent           *ArchivalEvent\n\t\tvalidateURI     func(string) error\n\t\texpectedState   *ArchivalState\n\t\texpectedChanged bool\n\t\texpectedError   error\n\t}{\n\t\t{\n\t\t\tname: \"Enable archival with valid URI, no change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Enable archival without setting a URI - triggers validation error\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: disabled,\n\t\t\t\tURI:    \"\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"\",\n\t\t\t\tURI:        \"\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errInvalidEvent, // Expecting error due to enabling archival without a URI\n\t\t},\n\n\t\t{\n\t\t\tname: \"Archival disabled with URI, enabling with same URI\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: disabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: true,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Archival disabled without URI, enabling with default URI\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: disabled,\n\t\t\t\tURI:    \"\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://default-uri.com\"},\n\t\t\texpectedChanged: true,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Event with empty defaultURI\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errInvalidEvent,\n\t\t},\n\t\t{\n\t\t\tname: \"Immutable URI change attempt\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tURI:        \"http://new-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errURIUpdate,\n\t\t},\n\t\t{\n\t\t\tname:         \"Current state is nil\",\n\t\t\tcurrentState: nil,\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errCannotHandleStateChange,\n\t\t},\n\t\t{\n\t\t\tname: \"Event is nil\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent:           nil,\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errCannotHandleStateChange,\n\t\t},\n\t\t{\n\t\t\tname:            \"State and event nil\",\n\t\t\tcurrentState:    nil,\n\t\t\tevent:           nil,\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errCannotHandleStateChange,\n\t\t},\n\t\t{\n\t\t\tname: \"Unrecognized ArchivalStatus in event\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     types.ArchivalStatus(999).Ptr(), // Assuming 999 is an unrecognized status\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errCannotHandleStateChange,\n\t\t},\n\t\t{\n\t\t\tname: \"Starting from an invalid state configuration\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"\", // Invalid state: enabled but no URI\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   nil,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errInvalidState,\n\t\t},\n\t\t{\n\t\t\tname: \"URI validation failure after passing immutability check\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tstatus:     &enabled,\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI: func(URI string) error {\n\t\t\t\t// Simulating a validation failure for any URI, focusing on triggering validation logic inside defer\n\t\t\t\treturn &types.BadRequestError{Message: \"URI validation failed due to invalid format\"}\n\t\t\t},\n\t\t\texpectedState:   nil,                                                                            // Expecting nil as the validation failure should prevent state transition\n\t\t\texpectedChanged: false,                                                                          // Expecting false as no change should occur due to validation failure\n\t\t\texpectedError:   &types.BadRequestError{Message: \"URI validation failed due to invalid format\"}, // Specific error from URI validation\n\t\t},\n\t\t{\n\t\t\tname: \"Trigger defer logic without URI change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: types.ArchivalStatusEnabled,\n\t\t\t\tURI:    \"http://existing-uri.com\", // Keeping URI same to avoid errURIUpdate.\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tstatus:     &enabled,\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI: func(URI string) error {\n\t\t\t\treturn errCannotHandleStateChange\n\t\t\t},\n\t\t\texpectedState:   nil, // Adjusting expectation based on understanding of defer logic.\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   errCannotHandleStateChange, // Expecting this validation failure.\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tnextState, changed, err := tt.currentState.getNextState(tt.event, tt.validateURI)\n\t\t\tassert.Equal(t, tt.expectedState, nextState)\n\t\t\tassert.Equal(t, tt.expectedChanged, changed)\n\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestNeverEnabledState(t *testing.T) {\n\texpectedState := &ArchivalState{\n\t\tURI:    \"\",\n\t\tStatus: types.ArchivalStatusDisabled,\n\t}\n\n\tactualState := neverEnabledState()\n\n\tassert.Equal(t, expectedState, actualState, \"The returned state does not match the expected never enabled state\")\n}\n\nfunc TestArchivalStateGetNextState_State1Scenarios(t *testing.T) {\n\tenabled := types.ArchivalStatusEnabled\n\tdisabled := types.ArchivalStatusDisabled\n\t// Mock URI validation function that always succeeds\n\tsuccessfulValidation := func(URI string) error { return nil }\n\n\ttests := []struct {\n\t\tname            string\n\t\tcurrentState    *ArchivalState\n\t\tevent           *ArchivalEvent\n\t\tvalidateURI     func(string) error\n\t\texpectedState   *ArchivalState\n\t\texpectedChanged bool\n\t\texpectedError   error\n\t}{\n\t\t{\n\t\t\tname: \"Enabled status without event URI set - no change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     &enabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Disabled status with event URI set - state change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tstatus:     &disabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: disabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: true,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Disabled status without event URI set - state change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tstatus:     &disabled,\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: disabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: true,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"No status with event URI set - no change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\tURI:        \"http://existing-uri.com\", // URI set but no status change attempted.\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"No status without event URI set - no change\",\n\t\t\tcurrentState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t\t// No URI and no status change attempted.\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   &ArchivalState{Status: enabled, URI: \"http://existing-uri.com\"},\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tnextState, changed, err := tt.currentState.getNextState(tt.event, tt.validateURI)\n\t\t\tassert.Equal(t, tt.expectedState, nextState)\n\t\t\tassert.Equal(t, tt.expectedChanged, changed)\n\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestArchivalStateGetNextState_State2Scenarios(t *testing.T) {\n\tdisabled := types.ArchivalStatusDisabled\n\tenabled := types.ArchivalStatusEnabled\n\tsuccessfulValidation := func(URI string) error { return nil }\n\n\tcurrentStateWithURI := &ArchivalState{\n\t\tStatus: disabled,\n\t\tURI:    \"http://existing-uri.com\",\n\t}\n\n\ttests := []struct {\n\t\tname            string\n\t\tevent           *ArchivalEvent\n\t\tvalidateURI     func(string) error\n\t\texpectedState   *ArchivalState\n\t\texpectedChanged bool\n\t\texpectedError   error\n\t}{\n\t\t{\n\t\t\tname: \"Enable archival without event URI - state change to enabled with existing URI\",\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tstatus:     &enabled,\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI: successfulValidation,\n\t\t\texpectedState: &ArchivalState{\n\t\t\t\tStatus: enabled,\n\t\t\t\tURI:    \"http://existing-uri.com\",\n\t\t\t},\n\t\t\texpectedChanged: true,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Disable archival with event URI set - no change\",\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tstatus:     &disabled,\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   currentStateWithURI,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Disable archival without event URI - no change\",\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tstatus:     &disabled,\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   currentStateWithURI,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Event URI set without status change - no change\",\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tURI:        \"http://existing-uri.com\",\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   currentStateWithURI,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"No event URI or status change - no change\",\n\t\t\tevent: &ArchivalEvent{\n\t\t\t\tdefaultURI: \"http://default-uri.com\",\n\t\t\t},\n\t\t\tvalidateURI:     successfulValidation,\n\t\t\texpectedState:   currentStateWithURI,\n\t\t\texpectedChanged: false,\n\t\t\texpectedError:   nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tnextState, changed, err := currentStateWithURI.getNextState(tt.event, tt.validateURI)\n\t\t\tassert.Equal(t, tt.expectedState, nextState)\n\t\t\tassert.Equal(t, tt.expectedChanged, changed)\n\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/attrValidator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// AttrValidatorImpl is domain attr validator\n\tAttrValidatorImpl struct {\n\t\tclusterMetadata  cluster.Metadata\n\t\tminRetentionDays int32\n\t}\n)\n\n// newAttrValidator create a new domain attr validator\nfunc newAttrValidator(\n\tclusterMetadata cluster.Metadata,\n\tminRetentionDays int32,\n) *AttrValidatorImpl {\n\n\treturn &AttrValidatorImpl{\n\t\tclusterMetadata:  clusterMetadata,\n\t\tminRetentionDays: minRetentionDays,\n\t}\n}\n\nfunc (d *AttrValidatorImpl) validateLocalDomainUpdateRequest(updateRequest *types.UpdateDomainRequest) error {\n\tif updateRequest.ActiveClusterName != nil || updateRequest.ActiveClusters != nil || updateRequest.Clusters != nil {\n\t\treturn errLocalDomainsCannotFailover\n\t}\n\tif updateRequest.FailoverTimeoutInSeconds != nil {\n\t\treturn errLocalDomainsCannotFailover\n\t}\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateDomainConfig(config *persistence.DomainConfig) error {\n\tif config.Retention < int32(d.minRetentionDays) {\n\t\treturn errInvalidRetentionPeriod\n\t}\n\tif config.HistoryArchivalStatus == types.ArchivalStatusEnabled && len(config.HistoryArchivalURI) == 0 {\n\t\treturn errInvalidArchivalConfig\n\t}\n\tif config.VisibilityArchivalStatus == types.ArchivalStatusEnabled && len(config.VisibilityArchivalURI) == 0 {\n\t\treturn errInvalidArchivalConfig\n\t}\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateDomainReplicationConfigForLocalDomain(\n\treplicationConfig *persistence.DomainReplicationConfig,\n) error {\n\n\tactiveCluster := replicationConfig.ActiveClusterName\n\tclusters := replicationConfig.Clusters\n\n\tif err := d.validateClusterName(activeCluster); err != nil {\n\t\treturn err\n\t}\n\tfor _, clusterConfig := range clusters {\n\t\tif err := d.validateClusterName(clusterConfig.ClusterName); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif activeCluster != d.clusterMetadata.GetCurrentClusterName() {\n\t\treturn &types.BadRequestError{Message: \"Invalid local domain active cluster\"}\n\t}\n\n\tif len(clusters) != 1 || clusters[0].ClusterName != activeCluster {\n\t\treturn &types.BadRequestError{Message: \"Invalid local domain clusters\"}\n\t}\n\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateDomainReplicationConfigForGlobalDomain(\n\treplicationConfig *persistence.DomainReplicationConfig,\n) error {\n\t// TODO: https://github.com/uber/cadence/issues/4345 add checking for \"pending active\" as well\n\t// Right now we only have checking if clusters to remove are \"current active cluster\" in this method.\n\t// However, there could be edge cases that a cluster is in \"pending active\" state during graceful failover.\n\t// It's better to do this check so that people won't make mistake.\n\t// However, this is not critical -- even this happens, they can add the active cluster back\n\n\tactiveCluster := replicationConfig.ActiveClusterName\n\tclusters := replicationConfig.Clusters\n\tactiveClusters := replicationConfig.ActiveClusters\n\n\tif activeCluster == \"\" {\n\t\treturn errActiveClusterNameRequired\n\t}\n\n\tfor _, clusterConfig := range clusters {\n\t\tif err := d.validateClusterName(clusterConfig.ClusterName); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tisInClusters := func(clusterName string) bool {\n\t\tfor _, clusterConfig := range clusters {\n\t\t\tif clusterConfig.ClusterName == clusterName {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t\treturn false\n\t}\n\n\tif err := d.validateClusterName(activeCluster); err != nil {\n\t\treturn err\n\t}\n\n\tif !isInClusters(activeCluster) {\n\t\treturn errActiveClusterNotInClusters\n\t}\n\t// For active-active domains, also validate that all clusters in AttributeScopes are valid\n\tif activeClusters != nil && activeClusters.AttributeScopes != nil {\n\t\tfor _, scope := range activeClusters.AttributeScopes {\n\t\t\tfor _, cluster := range scope.ClusterAttributes {\n\t\t\t\tif err := d.validateClusterName(cluster.ActiveClusterName); err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t\tif !isInClusters(cluster.ActiveClusterName) {\n\t\t\t\t\treturn errActiveClusterNotInClusters\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateDomainReplicationConfigClustersDoesNotRemove(\n\tclustersOld []*persistence.ClusterReplicationConfig,\n\tclustersNew []*persistence.ClusterReplicationConfig,\n) error {\n\n\tclusterNamesOld := make(map[string]bool)\n\tfor _, clusterConfig := range clustersOld {\n\t\tclusterNamesOld[clusterConfig.ClusterName] = true\n\t}\n\tclusterNamesNew := make(map[string]bool)\n\tfor _, clusterConfig := range clustersNew {\n\t\tclusterNamesNew[clusterConfig.ClusterName] = true\n\t}\n\n\tif len(clusterNamesNew) < len(clusterNamesOld) {\n\t\treturn errCannotRemoveClustersFromDomain\n\t}\n\n\tfor clusterName := range clusterNamesOld {\n\t\tif _, ok := clusterNamesNew[clusterName]; !ok {\n\t\t\treturn errCannotRemoveClustersFromDomain\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateClusterName(\n\tclusterName string,\n) error {\n\n\tif _, ok := d.clusterMetadata.GetEnabledClusterInfo()[clusterName]; !ok {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\"Invalid cluster name: %v\",\n\t\t\tclusterName,\n\t\t)}\n\t}\n\treturn nil\n}\n\nfunc (d *AttrValidatorImpl) validateActiveActiveDomainReplicationConfig(\n\tactiveClusters *types.ActiveClusters,\n) error {\n\n\tif activeClusters == nil || activeClusters.AttributeScopes == nil {\n\t\treturn nil\n\t}\n\n\tclusters := d.clusterMetadata.GetEnabledClusterInfo()\n\n\tfor _, scopeData := range activeClusters.AttributeScopes {\n\t\tfor _, activeCluster := range scopeData.ClusterAttributes {\n\t\t\t_, ok := clusters[activeCluster.ActiveClusterName]\n\t\t\tif !ok {\n\t\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\t\t\"Invalid active cluster name: %v\",\n\t\t\t\t\tactiveCluster.ActiveClusterName,\n\t\t\t\t)}\n\t\t\t}\n\t\t\tif activeCluster.FailoverVersion < 0 {\n\t\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\t\t\"invalid failover version: %d\",\n\t\t\t\t\tactiveCluster.FailoverVersion,\n\t\t\t\t)}\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/domain/attrValidator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tattrValidatorSuite struct {\n\t\tsuite.Suite\n\n\t\tminRetentionDays int\n\t\tvalidator        *AttrValidatorImpl\n\t}\n)\n\nfunc TestAttrValidatorSuite(t *testing.T) {\n\ts := new(attrValidatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *attrValidatorSuite) SetupSuite() {\n}\n\nfunc (s *attrValidatorSuite) TearDownSuite() {\n}\n\nfunc (s *attrValidatorSuite) SetupTest() {\n\ts.minRetentionDays = 1\n\ts.validator = newAttrValidator(cluster.TestActiveClusterMetadata, int32(s.minRetentionDays))\n}\n\nfunc (s *attrValidatorSuite) TearDownTest() {\n}\n\nfunc (s *attrValidatorSuite) TestValidateDomainConfig() {\n\ttestCases := []struct {\n\t\ttestDesc           string\n\t\tretention          int32\n\t\thistoryArchival    types.ArchivalStatus\n\t\thistoryURI         string\n\t\tvisibilityArchival types.ArchivalStatus\n\t\tvisibilityURI      string\n\t\texpectedErr        error\n\t}{\n\t\t{\n\t\t\ttestDesc:           \"valid retention and archival settings\",\n\t\t\tretention:          int32(s.minRetentionDays + 1),\n\t\t\thistoryArchival:    types.ArchivalStatusEnabled,\n\t\t\thistoryURI:         \"testScheme://history\",\n\t\t\tvisibilityArchival: types.ArchivalStatusEnabled,\n\t\t\tvisibilityURI:      \"testScheme://visibility\",\n\t\t\texpectedErr:        nil,\n\t\t},\n\t\t{\n\t\t\ttestDesc:    \"invalid retention period\",\n\t\t\tretention:   int32(s.minRetentionDays - 1),\n\t\t\texpectedErr: errInvalidRetentionPeriod,\n\t\t},\n\t\t{\n\t\t\ttestDesc:        \"enabled history archival without URI\",\n\t\t\tretention:       int32(s.minRetentionDays + 1),\n\t\t\thistoryArchival: types.ArchivalStatusEnabled,\n\t\t\texpectedErr:     errInvalidArchivalConfig,\n\t\t},\n\t\t{\n\t\t\ttestDesc:           \"enabled visibility archival without URI\",\n\t\t\tretention:          int32(s.minRetentionDays + 1),\n\t\t\tvisibilityArchival: types.ArchivalStatusEnabled,\n\t\t\texpectedErr:        errInvalidArchivalConfig,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.testDesc, func() {\n\t\t\terr := s.validator.validateDomainConfig(&persistence.DomainConfig{\n\t\t\t\tRetention:                tc.retention,\n\t\t\t\tHistoryArchivalStatus:    tc.historyArchival,\n\t\t\t\tHistoryArchivalURI:       tc.historyURI,\n\t\t\t\tVisibilityArchivalStatus: tc.visibilityArchival,\n\t\t\t\tVisibilityArchivalURI:    tc.visibilityURI,\n\t\t\t})\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\ts.ErrorIs(err, tc.expectedErr)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *attrValidatorSuite) TestValidateConfigRetentionPeriod() {\n\ttestCases := []struct {\n\t\tretentionPeriod int32\n\t\texpectedErr     error\n\t}{\n\t\t{\n\t\t\tretentionPeriod: 10,\n\t\t\texpectedErr:     nil,\n\t\t},\n\t\t{\n\t\t\tretentionPeriod: 0,\n\t\t\texpectedErr:     errInvalidRetentionPeriod,\n\t\t},\n\t\t{\n\t\t\tretentionPeriod: -3,\n\t\t\texpectedErr:     errInvalidRetentionPeriod,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tactualErr := s.validator.validateDomainConfig(\n\t\t\t&persistence.DomainConfig{Retention: tc.retentionPeriod},\n\t\t)\n\t\ts.Equal(tc.expectedErr, actualErr)\n\t}\n}\n\nfunc (s *attrValidatorSuite) TestClusterName() {\n\terr := s.validator.validateClusterName(\"some random foo bar\")\n\ts.IsType(&types.BadRequestError{}, err)\n\n\terr = s.validator.validateClusterName(cluster.TestCurrentClusterName)\n\ts.NoError(err)\n\n\terr = s.validator.validateClusterName(cluster.TestAlternativeClusterName)\n\ts.NoError(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateDomainReplicationConfigForLocalDomain() {\n\terr := s.validator.validateDomainReplicationConfigForLocalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\terr = s.validator.validateDomainReplicationConfigForLocalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\terr = s.validator.validateDomainReplicationConfigForLocalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\terr = s.validator.validateDomainReplicationConfigForLocalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateDomainReplicationConfigForGlobalDomain() {\n\terr := s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\t// When ActiveClusterName is not provided, and ActiveClusters are not provided, it should return an error\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t// When ActiveClusterName and ActiveClusters are provided, it should not return an error\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\t// When ActiveClusterName is not provided, and ActiveClusters are provided, it should return an error\n\terr = s.validator.validateDomainReplicationConfigForGlobalDomain(\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t)\n\ts.Error(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateDomainReplicationConfigClustersDoesNotRemove() {\n\terr := s.validator.validateDomainReplicationConfigClustersDoesNotRemove(\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\terr = s.validator.validateDomainReplicationConfigClustersDoesNotRemove(\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t},\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t)\n\ts.NoError(err)\n\n\terr = s.validator.validateDomainReplicationConfigClustersDoesNotRemove(\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\terr = s.validator.validateDomainReplicationConfigClustersDoesNotRemove(\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t},\n\t\t[]*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t},\n\t)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc TestValidateActiveActiveDomainReplicationConfig(t *testing.T) {\n\t// Define test cluster names\n\tconst (\n\t\tclusterA = \"cluster-a\"\n\t\tclusterB = \"cluster-b\"\n\t)\n\n\t// Create explicit cluster metadata for the test\n\tclusterMetadata := cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: 10,\n\t\t\tPrimaryClusterName:       clusterA,\n\t\t\tCurrentClusterName:       clusterA,\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\tclusterA: {\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t},\n\t\t\t\tclusterB: {\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n\n\tvalidator := newAttrValidator(clusterMetadata, 1)\n\n\ttestCases := []struct {\n\t\tname           string\n\t\tactiveClusters *types.ActiveClusters\n\t\texpectedErr    bool\n\t\terrType        interface{}\n\t}{\n\t\t{\n\t\t\tname: \"invalid cluster in AttributeScopes\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"city\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"invalid-cluster\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: true,\n\t\t\terrType:     &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"empty ActiveClusters - all maps nil\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: nil,\n\t\t\t},\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"empty ActiveClusters - maps initialized but empty\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t\t\t},\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"an invalid failover version should return an error\",\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"city\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: clusterA,\n\t\t\t\t\t\t\t\tFailoverVersion:   -1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\terr := validator.validateActiveActiveDomainReplicationConfig(tc.activeClusters)\n\t\t\tif tc.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.errType != nil {\n\t\t\t\t\tassert.IsType(t, tc.errType, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/audit/audit.go",
    "content": "package audit\n"
  },
  {
    "path": "common/domain/clusterattributes.go",
    "content": "package domain\n\nimport (\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc mergeActiveActiveScopes(existingDomain *types.ActiveClusters, incomingTask *types.ActiveClusters) (result *types.ActiveClusters, isChanged bool) {\n\n\tif existingDomain == nil && incomingTask == nil {\n\t\treturn nil, false\n\t}\n\n\t// new active clusters are being added\n\tif existingDomain == nil && incomingTask != nil {\n\t\treturn incomingTask, true\n\t}\n\n\t// existing data is not being mutated\n\tif incomingTask == nil && existingDomain != nil {\n\t\treturn existingDomain, false\n\t}\n\n\tmergedActiveClusters := &types.ActiveClusters{\n\t\tAttributeScopes: make(map[string]types.ClusterAttributeScope),\n\t}\n\n\tfor scope, existingScopeData := range existingDomain.AttributeScopes {\n\t\tincomingScope, presentInIncoming := incomingTask.AttributeScopes[scope]\n\t\tif presentInIncoming {\n\t\t\tmerged, isCh := mergeScope(existingScopeData, incomingScope)\n\t\t\tisChanged = isChanged || isCh\n\t\t\tmergedActiveClusters.AttributeScopes[scope] = merged\n\t\t} else {\n\t\t\tmergedActiveClusters.AttributeScopes[scope] = existingScopeData\n\t\t}\n\t}\n\n\tfor newScope, newScopeData := range incomingTask.AttributeScopes {\n\t\t_, existsAlready := mergedActiveClusters.AttributeScopes[newScope]\n\t\tif !existsAlready {\n\t\t\tmergedActiveClusters.AttributeScopes[newScope] = newScopeData\n\t\t\tisChanged = true\n\t\t}\n\t}\n\n\treturn mergedActiveClusters, isChanged\n}\n\nfunc mergeScope(existing types.ClusterAttributeScope, incoming types.ClusterAttributeScope) (merged types.ClusterAttributeScope, isChanged bool) {\n\n\tmerged.ClusterAttributes = make(map[string]types.ActiveClusterInfo)\n\n\tfor existingAttr := range existing.ClusterAttributes {\n\t\tincomingAtt, presentInNew := incoming.ClusterAttributes[existingAttr]\n\t\tif presentInNew {\n\t\t\tif incomingAtt.FailoverVersion > existing.ClusterAttributes[existingAttr].FailoverVersion {\n\t\t\t\tmerged.ClusterAttributes[existingAttr] = incomingAtt\n\t\t\t\tisChanged = true\n\t\t\t} else {\n\t\t\t\tmerged.ClusterAttributes[existingAttr] = existing.ClusterAttributes[existingAttr]\n\t\t\t}\n\t\t} else {\n\t\t\tmerged.ClusterAttributes[existingAttr] = existing.ClusterAttributes[existingAttr]\n\t\t}\n\t}\n\n\tfor newAttr, newAttrClusterInfo := range incoming.ClusterAttributes {\n\t\t_, existsInOld := existing.ClusterAttributes[newAttr]\n\t\tif !existsInOld {\n\t\t\tmerged.ClusterAttributes[newAttr] = newAttrClusterInfo\n\t\t\tisChanged = true\n\t\t}\n\t}\n\n\treturn merged, isChanged\n}\n"
  },
  {
    "path": "common/domain/clusterattributes_test.go",
    "content": "package domain\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDomainTask_ActiveActiveMerge(t *testing.T) {\n\ttests := map[string]struct {\n\t\tdomainOperation        types.DomainOperation\n\t\tisGlobalDomain         bool\n\t\tactiveClusters         *types.ActiveClusters\n\t\texpectPublish          bool\n\t\texpectedActiveClusters *types.ActiveClusters\n\t}{\n\t\t\"active-active domain with location and region scopes\": {\n\t\t\tdomainOperation: types.DomainOperationUpdate,\n\t\t\tisGlobalDomain:  true,\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectPublish: true,\n\t\t\texpectedActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"active-active domain create operation\": {\n\t\t\tdomainOperation: types.DomainOperationCreate,\n\t\t\tisGlobalDomain:  true,\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectPublish: true,\n\t\t\texpectedActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"non-global domain should not publish\": {\n\t\t\tdomainOperation: types.DomainOperationUpdate,\n\t\t\tisGlobalDomain:  false,\n\t\t\tactiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectPublish: false,\n\t\t},\n\t\t\"nil active clusters\": {\n\t\t\tdomainOperation:        types.DomainOperationUpdate,\n\t\t\tisGlobalDomain:         true,\n\t\t\tactiveClusters:         nil,\n\t\t\texpectPublish:          true,\n\t\t\texpectedActiveClusters: nil,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tkafkaProducer := &mocks.KafkaProducer{}\n\t\t\tdomainReplicator := NewDomainReplicator(\n\t\t\t\tkafkaProducer,\n\t\t\t\ttestlogger.New(t),\n\t\t\t).(*domainReplicatorImpl)\n\n\t\t\t// Setup test data\n\t\t\ttaskType := types.ReplicationTaskTypeDomain\n\t\t\tid := uuid.New()\n\t\t\tdomainName := \"test-active-active-domain\"\n\t\t\tstatus := types.DomainStatusRegistered\n\t\t\tdescription := \"test domain\"\n\t\t\townerEmail := \"test@example.com\"\n\t\t\tdata := map[string]string{\"k\": \"v\"}\n\t\t\tretention := int32(10)\n\t\t\temitMetric := true\n\t\t\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\t\t\thistoryArchivalURI := \"test-history-uri\"\n\t\t\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\t\t\tvisibilityArchivalURI := \"test-visibility-uri\"\n\t\t\tclusterActive := \"clusterA\"\n\t\t\tclusterStandby := \"clusterB\"\n\t\t\tconfigVersion := int64(5)\n\t\t\tfailoverVersion := int64(100)\n\t\t\tpreviousFailoverVersion := int64(50)\n\t\t\tclusters := []*p.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: clusterActive},\n\t\t\t\t{ClusterName: clusterStandby},\n\t\t\t}\n\n\t\t\tinfo := &p.DomainInfo{\n\t\t\t\tID:          id,\n\t\t\t\tName:        domainName,\n\t\t\t\tStatus:      p.DomainStatusRegistered,\n\t\t\t\tDescription: description,\n\t\t\t\tOwnerEmail:  ownerEmail,\n\t\t\t\tData:        data,\n\t\t\t}\n\t\t\tconfig := &p.DomainConfig{\n\t\t\t\tRetention:                retention,\n\t\t\t\tEmitMetric:               emitMetric,\n\t\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t\t}\n\t\t\treplicationConfig := &p.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: clusterActive,\n\t\t\t\tClusters:          clusters,\n\t\t\t\tActiveClusters:    tc.activeClusters,\n\t\t\t}\n\n\t\t\tif tc.expectPublish {\n\t\t\t\tkafkaProducer.On(\"Publish\", mock.Anything, &types.ReplicationTask{\n\t\t\t\t\tTaskType: &taskType,\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tDomainOperation: &tc.domainOperation,\n\t\t\t\t\t\tID:              id,\n\t\t\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\t\t\tName:        domainName,\n\t\t\t\t\t\t\tStatus:      &status,\n\t\t\t\t\t\t\tDescription: description,\n\t\t\t\t\t\t\tOwnerEmail:  ownerEmail,\n\t\t\t\t\t\t\tData:        data,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\t\t\t\t\tEmitMetric:                             emitMetric,\n\t\t\t\t\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\t\t\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\t\t\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\t\t\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\t\tActiveClusterName: clusterActive,\n\t\t\t\t\t\t\tClusters:          domainReplicator.convertClusterReplicationConfigToThrift(clusters),\n\t\t\t\t\t\t\tActiveClusters:    tc.expectedActiveClusters,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfigVersion:           configVersion,\n\t\t\t\t\t\tFailoverVersion:         failoverVersion,\n\t\t\t\t\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil).Once()\n\t\t\t}\n\n\t\t\terr := domainReplicator.HandleTransmissionTask(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.domainOperation,\n\t\t\t\tinfo,\n\t\t\t\tconfig,\n\t\t\t\treplicationConfig,\n\t\t\t\tconfigVersion,\n\t\t\t\tfailoverVersion,\n\t\t\t\tpreviousFailoverVersion,\n\t\t\t\ttc.isGlobalDomain,\n\t\t\t)\n\n\t\t\tassert.NoError(t, err)\n\t\t\tkafkaProducer.AssertExpectations(t)\n\t\t})\n\t}\n}\n\nfunc TestActiveActiveDomainUpdates_Merge(t *testing.T) {\n\n\t// worked example:\n\t//\n\t// Cluster A - Initial Failover Version: 0 - us-east-1\n\t// Cluster B - Initial Failover Version: 1 - us-west-1\n\t// Cluster C - Initial Failover Version: 2 - us-west-2\n\t// Cluster D - Initial Failover Version: 3 - us-west-2\n\n\ttests := map[string]struct {\n\t\tcurrentActiveClusters *types.ActiveClusters\n\t\trequest               *types.ActiveClusters\n\n\t\texpectedResult    *types.ActiveClusters\n\t\texpectedIsChanged bool\n\t}{\n\n\t\t\"simple case - incoming location failover out of us-east-1 - failing over some cities to cluster C - the result should indicate the failover has occurred\": {\n\t\t\tcurrentActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// incoming location failover out of us-east-1 - failing over some cities to cluster C\n\t\t\trequest: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Denver\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: true,\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\n\t\t// this simulates a stale domain update being sent through the replication queue\n\t\t// in this case, the local data is more recent and should win-out\n\t\t\"incoming location failover out of us-east-1 - failing over some cities to cluster C - the local data is more recent - the result should drop stale cluster-attributes from the replication message\": {\n\t\t\tcurrentActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 101},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// incoming location failover out of us-east-1 - failing over some cities to cluster C\n\t\t\trequest: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Denver\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: false,\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 101},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t// This simulates the case when there's either been very fast changes or a network split\n\t\t// and at least two clusters have updated their local domain-data in the mean time and\n\t\t// at for the same amount of times. Ie it's pretty unlikely, though possible.\n\t\t// in this case, the arbitrary initial failover version is used as a deterministic tie-break\n\t\t\"conflict resolution - both incoming and local data have been modified - employ tie-break - the result with the higher failover version should win\": {\n\t\t\tcurrentActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 101},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 103},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// incoming location failover out of us-east-1 - failing all locations to cluster C\n\t\t\trequest: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: true,\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t// Prague and Denver are failing over to cluster C - they should be updated to the new cluster\n\t\t\t\t\t\t\t// the somewhat arbitrary initial failover version is used as a tie-break\n\t\t\t\t\t\t\t// and clusterC is chosen\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 102},\n\t\t\t\t\t\t\t// Sao Paulo is not failing over - it should remain at the same cluster\n\t\t\t\t\t\t\t\"Sao Paulo\": {ActiveClusterName: \"clusterD\", FailoverVersion: 103},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\n\t\t\"simple case - no updates - no change\": {\n\t\t\tcurrentActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\trequest: &types.ActiveClusters{},\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"Prague\":      {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"Denver\":      {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"Maarstricht\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t\t\"Sao Paulo\":   {ActiveClusterName: \"clusterD\", FailoverVersion: 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"clusterA\", FailoverVersion: 0},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"clusterB\", FailoverVersion: 1},\n\t\t\t\t\t\t\t\"us-west-2\": {ActiveClusterName: \"clusterC\", FailoverVersion: 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: false,\n\t\t},\n\t}\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult, isChanged := mergeActiveActiveScopes(td.currentActiveClusters, td.request)\n\t\t\tassert.Equal(t, td.expectedResult, result)\n\t\t\tassert.Equal(t, td.expectedIsChanged, isChanged)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/dlq_message_handler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination dlq_message_handler_mock.go\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// DLQMessageHandler is the interface handles domain DLQ messages\n\tDLQMessageHandler interface {\n\t\tcommon.Daemon\n\n\t\tCount(ctx context.Context, forceFetch bool) (int64, error)\n\t\tRead(ctx context.Context, lastMessageID int64, pageSize int, pageToken []byte) ([]*types.ReplicationTask, []byte, error)\n\t\tPurge(ctx context.Context, lastMessageID int64) error\n\t\tMerge(ctx context.Context, lastMessageID int64, pageSize int, pageToken []byte) ([]byte, error)\n\t}\n\n\tdlqMessageHandlerImpl struct {\n\t\treplicationHandler ReplicationTaskExecutor\n\t\treplicationQueue   ReplicationQueue\n\t\tlogger             log.Logger\n\t\tmetricsClient      metrics.Client\n\t\ttimeSrc            clock.TimeSource\n\t\tdone               chan struct{}\n\t\tstatus             int32\n\n\t\tmu        sync.Mutex\n\t\tlastCount int64\n\t}\n)\n\n// NewDLQMessageHandler returns a DLQTaskHandler instance\nfunc NewDLQMessageHandler(\n\treplicationHandler ReplicationTaskExecutor,\n\treplicationQueue ReplicationQueue,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n) DLQMessageHandler {\n\treturn &dlqMessageHandlerImpl{\n\t\treplicationHandler: replicationHandler,\n\t\treplicationQueue:   replicationQueue,\n\t\tlogger:             logger,\n\t\tmetricsClient:      metricsClient,\n\t\ttimeSrc:            timeSource,\n\t\tdone:               make(chan struct{}),\n\t\tlastCount:          -1,\n\t}\n}\n\n// Start starts the DLQ handler\nfunc (d *dlqMessageHandlerImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&d.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tgo d.emitDLQSizeMetricsLoop()\n\td.logger.Info(\"Domain DLQ handler started.\")\n}\n\n// Stop stops the DLQ handler\nfunc (d *dlqMessageHandlerImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&d.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\td.logger.Info(\"Domain DLQ handler shutting down.\")\n\tclose(d.done)\n}\n\n// Count counts domain replication DLQ messages\nfunc (d *dlqMessageHandlerImpl) Count(ctx context.Context, forceFetch bool) (int64, error) {\n\tif forceFetch || d.lastCount == -1 {\n\t\tif err := d.fetchAndEmitDLQSize(ctx); err != nil {\n\t\t\treturn 0, err\n\t\t}\n\t}\n\treturn d.lastCount, nil\n}\n\n// ReadMessages reads domain replication DLQ messages\nfunc (d *dlqMessageHandlerImpl) Read(\n\tctx context.Context,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*types.ReplicationTask, []byte, error) {\n\n\tackLevel, err := d.replicationQueue.GetDLQAckLevel(ctx)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\treturn d.replicationQueue.GetMessagesFromDLQ(\n\t\tctx,\n\t\tackLevel,\n\t\tlastMessageID,\n\t\tpageSize,\n\t\tpageToken,\n\t)\n}\n\n// PurgeMessages purges domain replication DLQ messages\nfunc (d *dlqMessageHandlerImpl) Purge(\n\tctx context.Context,\n\tlastMessageID int64,\n) error {\n\n\tackLevel, err := d.replicationQueue.GetDLQAckLevel(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := d.replicationQueue.RangeDeleteMessagesFromDLQ(\n\t\tctx,\n\t\tackLevel,\n\t\tlastMessageID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := d.replicationQueue.UpdateDLQAckLevel(\n\t\tctx,\n\t\tlastMessageID,\n\t); err != nil {\n\t\td.logger.Error(\"Failed to update DLQ ack level after purging messages\", tag.Error(err))\n\t}\n\n\treturn nil\n}\n\n// MergeMessages merges domain replication DLQ messages\nfunc (d *dlqMessageHandlerImpl) Merge(\n\tctx context.Context,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]byte, error) {\n\n\tackLevel, err := d.replicationQueue.GetDLQAckLevel(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmessages, token, err := d.replicationQueue.GetMessagesFromDLQ(\n\t\tctx,\n\t\tackLevel,\n\t\tlastMessageID,\n\t\tpageSize,\n\t\tpageToken,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar ackedMessageID int64\n\tfor _, message := range messages {\n\t\tdomainTask := message.GetDomainTaskAttributes()\n\t\tif domainTask == nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: \"Encounter non domain replication task in domain replication queue.\"}\n\t\t}\n\n\t\t// TODO:\n\t\tif err := d.replicationHandler.Execute(\n\t\t\tdomainTask,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tackedMessageID = message.SourceTaskID\n\t}\n\n\tif err := d.replicationQueue.RangeDeleteMessagesFromDLQ(\n\t\tctx,\n\t\tackLevel,\n\t\tackedMessageID,\n\t); err != nil {\n\t\td.logger.Error(\"failed to delete merged tasks on merging domain DLQ message\", tag.Error(err))\n\t\treturn nil, err\n\t}\n\tif err := d.replicationQueue.UpdateDLQAckLevel(ctx, ackedMessageID); err != nil {\n\t\td.logger.Error(\"failed to update ack level on merging domain DLQ message\", tag.Error(err))\n\t}\n\n\treturn token, nil\n}\n\nfunc (d *dlqMessageHandlerImpl) emitDLQSizeMetricsLoop() {\n\tticker := d.timeSrc.NewTicker(queueSizeQueryInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-d.done:\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\terr := d.fetchAndEmitDLQSize(context.Background())\n\t\t\tif err != nil {\n\t\t\t\td.logger.Warn(\"Failed to get DLQ size.\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (d *dlqMessageHandlerImpl) fetchAndEmitDLQSize(ctx context.Context) error {\n\tsize, err := d.replicationQueue.GetDLQSize(ctx)\n\tif err != nil {\n\t\td.metricsClient.Scope(metrics.DomainReplicationQueueScope).IncCounter(metrics.DomainReplicationQueueSizeErrorCount)\n\t\treturn err\n\t}\n\n\td.metricsClient.Scope(metrics.DomainReplicationQueueScope).UpdateGauge(metrics.DomainReplicationQueueSizeGauge, float64(size))\n\n\td.mu.Lock()\n\td.lastCount = size\n\td.mu.Unlock()\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/domain/dlq_message_handler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: dlq_message_handler.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source dlq_message_handler.go -destination dlq_message_handler_mock.go\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockDLQMessageHandler is a mock of DLQMessageHandler interface.\ntype MockDLQMessageHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDLQMessageHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockDLQMessageHandlerMockRecorder is the mock recorder for MockDLQMessageHandler.\ntype MockDLQMessageHandlerMockRecorder struct {\n\tmock *MockDLQMessageHandler\n}\n\n// NewMockDLQMessageHandler creates a new mock instance.\nfunc NewMockDLQMessageHandler(ctrl *gomock.Controller) *MockDLQMessageHandler {\n\tmock := &MockDLQMessageHandler{ctrl: ctrl}\n\tmock.recorder = &MockDLQMessageHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDLQMessageHandler) EXPECT() *MockDLQMessageHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// Count mocks base method.\nfunc (m *MockDLQMessageHandler) Count(ctx context.Context, forceFetch bool) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Count\", ctx, forceFetch)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Count indicates an expected call of Count.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Count(ctx, forceFetch any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Count\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Count), ctx, forceFetch)\n}\n\n// Merge mocks base method.\nfunc (m *MockDLQMessageHandler) Merge(ctx context.Context, lastMessageID int64, pageSize int, pageToken []byte) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Merge\", ctx, lastMessageID, pageSize, pageToken)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Merge indicates an expected call of Merge.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Merge(ctx, lastMessageID, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Merge\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Merge), ctx, lastMessageID, pageSize, pageToken)\n}\n\n// Purge mocks base method.\nfunc (m *MockDLQMessageHandler) Purge(ctx context.Context, lastMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Purge\", ctx, lastMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Purge indicates an expected call of Purge.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Purge(ctx, lastMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Purge\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Purge), ctx, lastMessageID)\n}\n\n// Read mocks base method.\nfunc (m *MockDLQMessageHandler) Read(ctx context.Context, lastMessageID int64, pageSize int, pageToken []byte) ([]*types.ReplicationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Read\", ctx, lastMessageID, pageSize, pageToken)\n\tret0, _ := ret[0].([]*types.ReplicationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// Read indicates an expected call of Read.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Read(ctx, lastMessageID, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Read\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Read), ctx, lastMessageID, pageSize, pageToken)\n}\n\n// Start mocks base method.\nfunc (m *MockDLQMessageHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockDLQMessageHandler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockDLQMessageHandlerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockDLQMessageHandler)(nil).Stop))\n}\n"
  },
  {
    "path": "common/domain/dlq_message_handler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdlqMessageHandlerSuite struct {\n\t\tsuite.Suite\n\n\t\t*require.Assertions\n\t\tcontroller *gomock.Controller\n\n\t\tmockReplicationTaskExecutor *MockReplicationTaskExecutor\n\t\tmockReplicationQueue        *MockReplicationQueue\n\t\tdlqMessageHandler           *dlqMessageHandlerImpl\n\t}\n)\n\nfunc TestDLQMessageHandlerSuite(t *testing.T) {\n\ts := new(dlqMessageHandlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *dlqMessageHandlerSuite) SetupSuite() {\n}\n\nfunc (s *dlqMessageHandlerSuite) TearDownSuite() {\n\n}\n\nfunc (s *dlqMessageHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockReplicationTaskExecutor = NewMockReplicationTaskExecutor(s.controller)\n\ts.mockReplicationQueue = NewMockReplicationQueue(s.controller)\n\n\tlogger := testlogger.New(s.Suite.T())\n\ts.dlqMessageHandler = NewDLQMessageHandler(\n\t\ts.mockReplicationTaskExecutor,\n\t\ts.mockReplicationQueue,\n\t\tlogger,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tclock.NewMockedTimeSource(),\n\t).(*dlqMessageHandlerImpl)\n}\n\nfunc (s *dlqMessageHandlerSuite) TearDownTest() {\n}\n\nfunc (s *dlqMessageHandlerSuite) TestReadMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:     types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID: 1,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\n\tresp, token, err := s.dlqMessageHandler.Read(context.Background(), lastMessageID, pageSize, pageToken)\n\n\ts.NoError(err)\n\ts.Equal(tasks, resp)\n\ts.Nil(token)\n}\n\nfunc TestDLQMessageHandler_Start(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(m *MockReplicationQueue)\n\t\tinitialStatus  int32\n\t\texpectedStatus int32\n\t}{\n\t\t{\n\t\t\tname: \"Should start when initialized\",\n\t\t\tsetupMocks: func(m *MockReplicationQueue) {\n\t\t\t\tm.EXPECT().GetDLQSize(gomock.Any()).Return(int64(1), nil).AnyTimes()\n\t\t\t},\n\t\t\tinitialStatus:  common.DaemonStatusInitialized,\n\t\t\texpectedStatus: common.DaemonStatusStarted,\n\t\t},\n\t\t{\n\t\t\tname: \"Should not start when already started\",\n\t\t\tsetupMocks: func(m *MockReplicationQueue) {\n\t\t\t\t// No calls expected since the handler should recognize it's already started\n\t\t\t},\n\t\t\tinitialStatus:  common.DaemonStatusStarted,\n\t\t\texpectedStatus: common.DaemonStatusStarted,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := NewMockReplicationQueue(ctrl)\n\t\t\tmockReplicationTaskExecutor := NewMockReplicationTaskExecutor(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockedTime := clock.NewMockedTimeSource()\n\t\t\ttc.setupMocks(mockQueue)\n\n\t\t\tdoneChan := make(chan struct{})\n\t\t\thandler := &dlqMessageHandlerImpl{\n\t\t\t\treplicationHandler: mockReplicationTaskExecutor,\n\t\t\t\treplicationQueue:   mockQueue,\n\t\t\t\tmetricsClient:      metrics.NewNoopMetricsClient(),\n\t\t\t\tlogger:             mockLogger,\n\t\t\t\tdone:               doneChan,\n\t\t\t\tstatus:             tc.initialStatus,\n\t\t\t\ttimeSrc:            mockedTime,\n\t\t\t}\n\n\t\t\thandler.Start()\n\t\t\tmockedTime.Advance(5 * time.Minute)\n\t\t\tdefer handler.Stop()\n\t\t\tassert.Equal(t, tc.expectedStatus, atomic.LoadInt32(&handler.status))\n\t\t})\n\t}\n}\n\nfunc (s *dlqMessageHandlerSuite) TestStop() {\n\ttests := []struct {\n\t\tname           string\n\t\tinitialStatus  int32\n\t\texpectedStatus int32\n\t\tshouldStop     bool\n\t}{\n\t\t{\n\t\t\tname:           \"Should stop when started\",\n\t\t\tinitialStatus:  common.DaemonStatusStarted,\n\t\t\texpectedStatus: common.DaemonStatusStopped,\n\t\t\tshouldStop:     true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Should not stop when not started\",\n\t\t\tinitialStatus:  common.DaemonStatusInitialized,\n\t\t\texpectedStatus: common.DaemonStatusInitialized,\n\t\t\tshouldStop:     false,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\ts.Run(test.name, func() {\n\t\t\tatomic.StoreInt32(&s.dlqMessageHandler.status, test.initialStatus)\n\t\t\ts.dlqMessageHandler.Stop()\n\t\t\ts.Equal(test.expectedStatus, atomic.LoadInt32(&s.dlqMessageHandler.status))\n\t\t})\n\t}\n}\n\nfunc (s *dlqMessageHandlerSuite) TestCount() {\n\ttests := []struct {\n\t\tname          string\n\t\tforceFetch    bool\n\t\tlastCount     int64\n\t\tfetchSize     int64\n\t\tfetchError    error\n\t\texpectedCount int64\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname:          \"Force fetch with error\",\n\t\t\tforceFetch:    true,\n\t\t\tlastCount:     10,\n\t\t\tfetchSize:     0,\n\t\t\tfetchError:    fmt.Errorf(\"fetch error\"),\n\t\t\texpectedCount: 0,\n\t\t\texpectedError: fmt.Errorf(\"fetch error\"),\n\t\t},\n\t\t{\n\t\t\tname:          \"Force fetch with success\",\n\t\t\tforceFetch:    true,\n\t\t\tlastCount:     10,\n\t\t\tfetchSize:     20,\n\t\t\tfetchError:    nil,\n\t\t\texpectedCount: 20,\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname:          \"No fetch needed\",\n\t\t\tforceFetch:    false,\n\t\t\tlastCount:     30,\n\t\t\texpectedCount: 30,\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\ts.Run(test.name, func() {\n\t\t\ts.mockReplicationQueue.EXPECT().GetDLQSize(gomock.Any()).Return(test.fetchSize, test.fetchError).MaxTimes(1)\n\t\t\ts.dlqMessageHandler.lastCount = test.lastCount\n\t\t\tcount, err := s.dlqMessageHandler.Count(context.Background(), test.forceFetch)\n\t\t\ts.Equal(test.expectedCount, count)\n\t\t\tif test.expectedError != nil {\n\t\t\t\ts.Equal(test.expectedError, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *dlqMessageHandlerSuite) TestReadMessages_ThrowErrorOnGetDLQAckLevel() {\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:     types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID: 1,\n\t\t},\n\t}\n\ttestError := fmt.Errorf(\"test\")\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(int64(-1), testError).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(tasks, nil, nil).Times(0)\n\n\t_, _, err := s.dlqMessageHandler.Read(context.Background(), lastMessageID, pageSize, pageToken)\n\n\ts.Equal(testError, err)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestReadMessages_ThrowErrorOnReadMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\n\ttestError := fmt.Errorf(\"test\")\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(nil, nil, testError).Times(1)\n\n\t_, _, err := s.dlqMessageHandler.Read(context.Background(), lastMessageID, pageSize, pageToken)\n\n\ts.Equal(testError, err)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestPurgeMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), lastMessageID).Return(nil).Times(1)\n\terr := s.dlqMessageHandler.Purge(context.Background(), lastMessageID)\n\n\ts.NoError(err)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestPurgeMessages_ThrowErrorOnGetDLQAckLevel() {\n\tlastMessageID := int64(20)\n\ttestError := fmt.Errorf(\"test\")\n\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(int64(-1), testError).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).Times(0)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Times(0)\n\terr := s.dlqMessageHandler.Purge(context.Background(), lastMessageID)\n\n\ts.Equal(testError, err)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestPurgeMessages_ThrowErrorOnPurgeMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\ttestError := fmt.Errorf(\"test\")\n\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID).Return(testError).Times(1)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Times(0)\n\terr := s.dlqMessageHandler.Purge(context.Background(), lastMessageID)\n\n\ts.Equal(testError, err)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\tmessageID := int64(11)\n\n\tdomainAttribute := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID,\n\t\t\tDomainTaskAttributes: domainAttribute,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), messageID).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), ackLevel, messageID).Return(nil).Times(1)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.NoError(err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_ThrowErrorOnGetDLQAckLevel() {\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\tmessageID := int64(11)\n\ttestError := fmt.Errorf(\"test\")\n\tdomainAttribute := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID,\n\t\t\tDomainTaskAttributes: domainAttribute,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(int64(-1), testError).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(tasks, nil, nil).Times(0)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(gomock.Any()).Times(0)\n\ts.mockReplicationQueue.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Any()).Times(0)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Times(0)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.Equal(testError, err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_ThrowErrorOnGetDLQMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\ttestError := fmt.Errorf(\"test\")\n\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(nil, nil, testError).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(gomock.Any()).Times(0)\n\ts.mockReplicationQueue.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Any()).Times(0)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Times(0)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.Equal(testError, err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_ThrowErrorOnHandleReceivingTask() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\tmessageID1 := int64(11)\n\tmessageID2 := int64(12)\n\ttestError := fmt.Errorf(\"test\")\n\tdomainAttribute1 := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\tdomainAttribute2 := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID1,\n\t\t\tDomainTaskAttributes: domainAttribute1,\n\t\t},\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID2,\n\t\t\tDomainTaskAttributes: domainAttribute2,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute1).Return(nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute2).Return(testError).Times(1)\n\ts.mockReplicationQueue.EXPECT().DeleteMessageFromDLQ(gomock.Any(), messageID2).Times(0)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.Equal(testError, err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_ThrowErrorOnDeleteMessages() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\tmessageID1 := int64(11)\n\tmessageID2 := int64(12)\n\ttestError := fmt.Errorf(\"test\")\n\tdomainAttribute1 := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\tdomainAttribute2 := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID1,\n\t\t\tDomainTaskAttributes: domainAttribute1,\n\t\t},\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID2,\n\t\t\tDomainTaskAttributes: domainAttribute2,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute1).Return(nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute2).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), ackLevel, messageID2).Return(testError).Times(1)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.Error(err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_IgnoreErrorOnUpdateDLQAckLevel() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\tmessageID := int64(11)\n\ttestError := fmt.Errorf(\"test\")\n\tdomainAttribute := &types.DomainTaskAttributes{\n\t\tID: uuid.New(),\n\t}\n\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\tSourceTaskID:         messageID,\n\t\t\tDomainTaskAttributes: domainAttribute,\n\t\t},\n\t}\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\ts.mockReplicationTaskExecutor.EXPECT().Execute(domainAttribute).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), ackLevel, messageID).Return(nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().UpdateDLQAckLevel(gomock.Any(), messageID).Return(testError).Times(1)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\ts.NoError(err)\n\ts.Nil(token)\n}\n\nfunc (s *dlqMessageHandlerSuite) TestMergeMessages_NonDomainTask() {\n\tackLevel := int64(10)\n\tlastMessageID := int64(20)\n\tpageSize := 100\n\tpageToken := []byte{}\n\n\t// Create a task that mimics a non-domain replication task by not setting DomainTaskAttributes\n\ttasks := []*types.ReplicationTask{\n\t\t{\n\t\t\tTaskType:     types.ReplicationTaskTypeDomain.Ptr(), // Still set to domain but no attributes\n\t\t\tSourceTaskID: 1,\n\t\t},\n\t}\n\n\ts.mockReplicationQueue.EXPECT().GetDLQAckLevel(gomock.Any()).Return(ackLevel, nil).Times(1)\n\ts.mockReplicationQueue.EXPECT().GetMessagesFromDLQ(gomock.Any(), ackLevel, lastMessageID, pageSize, pageToken).\n\t\tReturn(tasks, nil, nil).Times(1)\n\n\ttoken, err := s.dlqMessageHandler.Merge(context.Background(), lastMessageID, pageSize, pageToken)\n\n\ts.NotNil(err)\n\ts.IsType(&types.InternalServiceError{}, err)\n\ts.Equal(\"Encounter non domain replication task in domain replication queue.\", err.Error())\n\ts.Nil(token)\n}\n"
  },
  {
    "path": "common/domain/errors.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport \"github.com/uber/cadence/common/types\"\n\nvar (\n\t// err indicating that this cluster is not the primary, so cannot do domain registration or update\n\terrNotPrimaryCluster                   = &types.BadRequestError{Message: \"Cluster is not primary cluster, cannot do domain registration or domain update.\"}\n\terrCannotRemoveClustersFromDomain      = &types.BadRequestError{Message: \"Cannot remove existing replicated clusters from a domain.\"}\n\terrActiveClusterNotInClusters          = &types.BadRequestError{Message: \"Active cluster is not contained in all clusters.\"}\n\terrCannotDoDomainFailoverAndUpdate     = &types.BadRequestError{Message: \"Cannot set active cluster to current cluster when other parameters are set.\"}\n\terrCannotDoGracefulFailoverFromCluster = &types.BadRequestError{Message: \"Cannot start the graceful failover from a to-be-passive cluster.\"}\n\terrGracefulFailoverInActiveCluster     = &types.BadRequestError{Message: \"Cannot start the graceful failover from an active cluster to an active cluster.\"}\n\terrOngoingGracefulFailover             = &types.BadRequestError{Message: \"Cannot start concurrent graceful failover.\"}\n\terrInvalidGracefulFailover             = &types.BadRequestError{Message: \"Cannot start graceful failover without updating active cluster or in local domain.\"}\n\terrInvalidFailoverNoChangeDetected     = &types.BadRequestError{Message: \"a failover was requested, but there was no change detected, the configuration was not updated\"}\n\terrActiveClusterNameRequired           = &types.BadRequestError{Message: \"ActiveClusterName is required for all global domains.\"}\n\terrLocalDomainsCannotFailover          = &types.BadRequestError{Message: \"Local domains cannot perform failovers or change replication configuration\"}\n\n\terrInvalidRetentionPeriod = &types.BadRequestError{Message: \"A valid retention period is not set on request.\"}\n\terrInvalidArchivalConfig  = &types.BadRequestError{Message: \"Invalid to enable archival without specifying a uri.\"}\n)\n"
  },
  {
    "path": "common/domain/failover_watcher.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination failover_watcher_mock.go\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nconst (\n\tupdateDomainRetryInitialInterval = 50 * time.Millisecond\n\tupdateDomainRetryCoefficient     = 2.0\n\tupdateDomainMaxRetry             = 3\n)\n\ntype (\n\t// FailoverWatcher handles failover operation on domain entities\n\tFailoverWatcher interface {\n\t\tcommon.Daemon\n\t}\n\n\tfailoverWatcherImpl struct {\n\t\tstatus          int32\n\t\tshutdownChan    chan struct{}\n\t\trefreshInterval dynamicproperties.DurationPropertyFn\n\t\trefreshJitter   dynamicproperties.FloatPropertyFn\n\t\tretryPolicy     backoff.RetryPolicy\n\n\t\tdomainManager persistence.DomainManager\n\t\tdomainCache   cache.DomainCache\n\t\ttimeSource    clock.TimeSource\n\t\tscope         metrics.Scope\n\t\tlogger        log.Logger\n\t}\n)\n\nvar _ FailoverWatcher = (*failoverWatcherImpl)(nil)\n\n// NewFailoverWatcher initializes domain failover processor\nfunc NewFailoverWatcher(\n\tdomainCache cache.DomainCache,\n\tdomainManager persistence.DomainManager,\n\ttimeSource clock.TimeSource,\n\trefreshInterval dynamicproperties.DurationPropertyFn,\n\trefreshJitter dynamicproperties.FloatPropertyFn,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) FailoverWatcher {\n\n\tretryPolicy := &backoff.ExponentialRetryPolicy{}\n\tretryPolicy.SetInitialInterval(updateDomainRetryInitialInterval)\n\tretryPolicy.SetBackoffCoefficient(updateDomainRetryCoefficient)\n\tretryPolicy.SetMaximumAttempts(updateDomainMaxRetry)\n\n\treturn &failoverWatcherImpl{\n\t\tstatus:          common.DaemonStatusInitialized,\n\t\tshutdownChan:    make(chan struct{}),\n\t\trefreshInterval: refreshInterval,\n\t\trefreshJitter:   refreshJitter,\n\t\tretryPolicy:     retryPolicy,\n\t\tdomainCache:     domainCache,\n\t\tdomainManager:   domainManager,\n\t\ttimeSource:      timeSource,\n\t\tscope:           metricsClient.Scope(metrics.DomainFailoverScope),\n\t\tlogger:          logger,\n\t}\n}\n\nfunc (p *failoverWatcherImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tgo p.refreshDomainLoop()\n\n\tp.logger.Info(\"Domain failover processor started.\")\n}\n\nfunc (p *failoverWatcherImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(p.shutdownChan)\n\tp.logger.Info(\"Domain failover processor stop.\")\n}\n\nfunc (p *failoverWatcherImpl) refreshDomainLoop() {\n\n\ttimer := p.timeSource.NewTimer(backoff.JitDuration(\n\t\tp.refreshInterval(),\n\t\tp.refreshJitter(),\n\t))\n\tdefer timer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.shutdownChan:\n\t\t\treturn\n\t\tcase <-timer.Chan():\n\t\t\tdomains := p.domainCache.GetAllDomain()\n\t\t\tfor _, domain := range domains {\n\t\t\t\tp.handleFailoverTimeout(domain)\n\t\t\t\tselect {\n\t\t\t\tcase <-p.shutdownChan:\n\t\t\t\t\tp.logger.Debug(\"Stop refresh domain as the processing is stopping.\")\n\t\t\t\t\treturn\n\t\t\t\tdefault:\n\t\t\t\t}\n\t\t\t}\n\n\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\tp.refreshInterval(),\n\t\t\t\tp.refreshJitter(),\n\t\t\t))\n\t\t}\n\t}\n}\n\nfunc (p *failoverWatcherImpl) handleFailoverTimeout(\n\tdomain *cache.DomainCacheEntry,\n) {\n\n\tfailoverEndTime := domain.GetFailoverEndTime()\n\tif domain.IsDomainPendingActive() && p.timeSource.Now().After(time.Unix(0, *failoverEndTime)) {\n\t\tdomainID := domain.GetInfo().ID\n\t\t// force failover the domain without setting the failover timeout\n\t\tif err := CleanPendingActiveState(\n\t\t\tp.domainManager,\n\t\t\tdomainID,\n\t\t\tdomain.GetFailoverVersion(),\n\t\t\tp.retryPolicy,\n\t\t); err != nil {\n\t\t\tp.logger.Error(\"Failed to update pending-active domain to active\", tag.WorkflowDomainID(domainID), tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\tp.scope.Tagged(metrics.DomainTag(domain.GetInfo().Name)).IncCounter(metrics.GracefulFailoverFailure)\n\t}\n}\n\n// CleanPendingActiveState removes the pending active state from the domain\nfunc CleanPendingActiveState(\n\tdomainManager persistence.DomainManager,\n\tdomainID string,\n\tfailoverVersion int64,\n\tpolicy backoff.RetryPolicy,\n) error {\n\n\t// must get the metadata (notificationVersion) first\n\t// this version can be regarded as the lock on the v2 domain table\n\t// and since we do not know which table will return the domain afterwards\n\t// this call has to be made\n\tmetadata, err := domainManager.GetMetadata(context.Background())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"getting metadata: %w\", err)\n\t}\n\n\tgetResponse, err := domainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{ID: domainID})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"getting domain: %w\", err)\n\t}\n\tlocalFailoverVersion := getResponse.FailoverVersion\n\tisGlobalDomain := getResponse.IsGlobalDomain\n\tgracefulFailoverEndTime := getResponse.FailoverEndTime\n\n\tif isGlobalDomain && gracefulFailoverEndTime != nil && failoverVersion == localFailoverVersion {\n\t\t// if the domain is still pending active and the failover versions are the same, clean the state\n\t\tupdateReq := &persistence.UpdateDomainRequest{\n\t\t\tInfo:                        getResponse.Info,\n\t\t\tConfig:                      getResponse.Config,\n\t\t\tReplicationConfig:           getResponse.ReplicationConfig,\n\t\t\tConfigVersion:               getResponse.ConfigVersion,\n\t\t\tFailoverVersion:             localFailoverVersion,\n\t\t\tFailoverNotificationVersion: getResponse.FailoverNotificationVersion,\n\t\t\tFailoverEndTime:             nil,\n\t\t\tNotificationVersion:         metadata.NotificationVersion,\n\t\t}\n\t\top := func(ctx context.Context) error {\n\t\t\treturn domainManager.UpdateDomain(ctx, updateReq)\n\t\t}\n\t\tthrottleRetry := backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(isUpdateDomainRetryable),\n\t\t)\n\t\tif err := throttleRetry.Do(context.Background(), op); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc isUpdateDomainRetryable(\n\terr error,\n) bool {\n\treturn true\n}\n"
  },
  {
    "path": "common/domain/failover_watcher_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: failover_watcher.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source failover_watcher.go -destination failover_watcher_mock.go\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockFailoverWatcher is a mock of FailoverWatcher interface.\ntype MockFailoverWatcher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFailoverWatcherMockRecorder\n\tisgomock struct{}\n}\n\n// MockFailoverWatcherMockRecorder is the mock recorder for MockFailoverWatcher.\ntype MockFailoverWatcherMockRecorder struct {\n\tmock *MockFailoverWatcher\n}\n\n// NewMockFailoverWatcher creates a new mock instance.\nfunc NewMockFailoverWatcher(ctrl *gomock.Controller) *MockFailoverWatcher {\n\tmock := &MockFailoverWatcher{ctrl: ctrl}\n\tmock.recorder = &MockFailoverWatcherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFailoverWatcher) EXPECT() *MockFailoverWatcherMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockFailoverWatcher) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockFailoverWatcherMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockFailoverWatcher)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockFailoverWatcher) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockFailoverWatcherMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockFailoverWatcher)(nil).Stop))\n}\n"
  },
  {
    "path": "common/domain/failover_watcher_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"errors\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tfailoverWatcherSuite struct {\n\t\tsuite.Suite\n\n\t\t*require.Assertions\n\t\tcontroller *gomock.Controller\n\n\t\tmockDomainCache *cache.MockDomainCache\n\t\ttimeSource      clock.TimeSource\n\t\tmockMetadataMgr *mocks.MetadataManager\n\t\twatcher         *failoverWatcherImpl\n\t}\n)\n\nfunc TestFailoverWatcherSuite(t *testing.T) {\n\ts := new(failoverWatcherSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *failoverWatcherSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\nfunc (s *failoverWatcherSuite) TearDownSuite() {\n}\n\nfunc (s *failoverWatcherSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.timeSource = clock.NewMockedTimeSource()\n\ts.mockMetadataMgr = &mocks.MetadataManager{}\n\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: 1,\n\t}, nil)\n\n\tlogger := testlogger.New(s.T())\n\tscope := tally.NewTestScope(\"failover_test\", nil)\n\tmetricsClient := metrics.NewClient(scope, metrics.Frontend, metrics.HistogramMigration{})\n\ts.watcher = NewFailoverWatcher(\n\t\ts.mockDomainCache,\n\t\ts.mockMetadataMgr,\n\t\ts.timeSource,\n\t\tdynamicproperties.GetDurationPropertyFn(10*time.Second),\n\t\tdynamicproperties.GetFloatPropertyFn(0.2),\n\t\tmetricsClient,\n\t\tlogger,\n\t).(*failoverWatcherImpl)\n}\n\nfunc (s *failoverWatcherSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.watcher.Stop()\n}\n\nfunc (s *failoverWatcherSuite) TestCleanPendingActiveState() {\n\tdomainName := uuid.New()\n\tinfo := &persistence.DomainInfo{\n\t\tID:          domainName,\n\t\tName:        domainName,\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"some random description\",\n\t\tOwnerEmail:  \"some random email\",\n\t\tData:        nil,\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention:  1,\n\t\tEmitMetric: true,\n\t}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: \"active\"},\n\t\t},\n\t}\n\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainName,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             1,\n\t\tFailoverNotificationVersion: 1,\n\t\tFailoverEndTime:             nil,\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\n\t// does not have failover end time\n\terr := CleanPendingActiveState(s.mockMetadataMgr, domainName, 1, s.watcher.retryPolicy)\n\ts.NoError(err)\n\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainName,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             1,\n\t\tFailoverNotificationVersion: 1,\n\t\tFailoverEndTime:             common.Int64Ptr(1),\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\n\t// does not match failover versions\n\terr = CleanPendingActiveState(s.mockMetadataMgr, domainName, 5, s.watcher.retryPolicy)\n\ts.NoError(err)\n\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, &persistence.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             nil,\n\t\tNotificationVersion:         1,\n\t}).Return(nil).Times(1)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainName,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             common.Int64Ptr(1),\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\n\terr = CleanPendingActiveState(s.mockMetadataMgr, domainName, 2, s.watcher.retryPolicy)\n\ts.NoError(err)\n}\n\nfunc (s *failoverWatcherSuite) TestHandleFailoverTimeout() {\n\tdomainName := uuid.New()\n\tinfo := &persistence.DomainInfo{\n\t\tID:          domainName,\n\t\tName:        domainName,\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"some random description\",\n\t\tOwnerEmail:  \"some random email\",\n\t\tData:        nil,\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention:  1,\n\t\tEmitMetric: true,\n\t}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: \"active\"},\n\t\t},\n\t}\n\tendtime := common.Int64Ptr(s.timeSource.Now().UnixNano() - 1)\n\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainName,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             1,\n\t\tFailoverNotificationVersion: 1,\n\t\tFailoverEndTime:             endtime,\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, &persistence.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             1,\n\t\tFailoverNotificationVersion: 1,\n\t\tFailoverEndTime:             nil,\n\t\tNotificationVersion:         1,\n\t}).Return(nil).Times(1)\n\n\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\tinfo,\n\t\tdomainConfig,\n\t\ttrue,\n\t\treplicationConfig,\n\t\t1,\n\t\tendtime,\n\t\t0, 0, 0,\n\t)\n\ts.watcher.handleFailoverTimeout(domainEntry)\n}\n\nfunc (s *failoverWatcherSuite) TestStart() {\n\ts.Assertions.Equal(common.DaemonStatusInitialized, s.watcher.status)\n\ts.watcher.Start()\n\ts.Assertions.Equal(common.DaemonStatusStarted, s.watcher.status)\n\n\t// Verify that calling Start again does not change the status\n\ts.watcher.Start()\n\ts.Assertions.Equal(common.DaemonStatusStarted, s.watcher.status)\n\ts.watcher.Stop()\n}\n\nfunc (s *failoverWatcherSuite) TestIsUpdateDomainRetryable() {\n\ttestCases := []struct {\n\t\tname      string\n\t\tinputErr  error\n\t\twantRetry bool\n\t}{\n\t\t{\"nil error\", nil, true},\n\t\t{\"non-nil error\", errors.New(\"some error\"), true},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\tretry := isUpdateDomainRetryable(tc.inputErr)\n\t\t\ts.Equal(tc.wantRetry, retry)\n\t\t})\n\t}\n}\n\nfunc (s *failoverWatcherSuite) TestRefreshDomainLoop() {\n\n\tdomainName := \"testDomain\"\n\tdomainID := uuid.New()\n\tfailoverEndTime := common.Int64Ptr(time.Now().Add(-time.Hour).UnixNano()) // 1 hour in the past\n\tmockTimeSource, _ := s.timeSource.(clock.MockedTimeSource)\n\n\tdomainInfo := &persistence.DomainInfo{ID: domainID, Name: domainName}\n\tdomainConfig := &persistence.DomainConfig{Retention: 1, EmitMetric: true}\n\treplicationConfig := &persistence.DomainReplicationConfig{ActiveClusterName: \"active\", Clusters: []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}}}\n\tdomainEntry := cache.NewDomainCacheEntryForTest(domainInfo, domainConfig, true, replicationConfig, 1, failoverEndTime, 0, 0, 0)\n\n\tdomainsMap := map[string]*cache.DomainCacheEntry{domainID: domainEntry}\n\ts.mockDomainCache.EXPECT().GetAllDomain().Return(domainsMap).AnyTimes()\n\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil).Maybe()\n\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.AnythingOfType(\"*persistence.GetDomainRequest\")).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        domainInfo,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             1,\n\t\tFailoverNotificationVersion: 1,\n\t\tFailoverEndTime:             failoverEndTime,\n\t\tNotificationVersion:         1,\n\t}, nil).Once()\n\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\ts.watcher.Start()\n\n\t// Delay to allow loop to start\n\ttime.Sleep(1 * time.Second)\n\tmockTimeSource.Advance(12 * time.Second)\n\t// Now stop the watcher, which should trigger the shutdown case in refreshDomainLoop\n\ts.watcher.Stop()\n\n\t// Enough time for shutdown process to complete\n\ttime.Sleep(1 * time.Second)\n\n\ts.mockMetadataMgr.AssertExpectations(s.T())\n}\n"
  },
  {
    "path": "common/domain/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination handler_mock.go\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"regexp\"\n\t\"time\"\n\n\tguuid \"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\terrDomainUpdateTooFrequent = &types.ServiceBusyError{Message: \"Domain update too frequent.\"}\n\terrInvalidDomainName       = &types.BadRequestError{Message: \"Domain name can only include alphanumeric and dash characters.\"}\n)\n\ntype (\n\t// Handler is the domain operation handler\n\tHandler interface {\n\t\tDeleteDomain(\n\t\t\tctx context.Context,\n\t\t\tdeleteRequest *types.DeleteDomainRequest,\n\t\t) error\n\t\tDeprecateDomain(\n\t\t\tctx context.Context,\n\t\t\tdeprecateRequest *types.DeprecateDomainRequest,\n\t\t) error\n\t\tDescribeDomain(\n\t\t\tctx context.Context,\n\t\t\tdescribeRequest *types.DescribeDomainRequest,\n\t\t) (*types.DescribeDomainResponse, error)\n\t\tListDomains(\n\t\t\tctx context.Context,\n\t\t\tlistRequest *types.ListDomainsRequest,\n\t\t) (*types.ListDomainsResponse, error)\n\t\tRegisterDomain(\n\t\t\tctx context.Context,\n\t\t\tregisterRequest *types.RegisterDomainRequest,\n\t\t) error\n\t\tUpdateDomain(\n\t\t\tctx context.Context,\n\t\t\tupdateRequest *types.UpdateDomainRequest,\n\t\t) (*types.UpdateDomainResponse, error)\n\t\tFailoverDomain(\n\t\t\tctx context.Context,\n\t\t\tfailoverRequest *types.FailoverDomainRequest,\n\t\t) (*types.FailoverDomainResponse, error)\n\t\tUpdateIsolationGroups(\n\t\t\tctx context.Context,\n\t\t\tupdateRequest types.UpdateDomainIsolationGroupsRequest,\n\t\t) error\n\t\tUpdateAsyncWorkflowConfiguraton(\n\t\t\tctx context.Context,\n\t\t\tupdateRequest types.UpdateDomainAsyncWorkflowConfiguratonRequest,\n\t\t) error\n\t}\n\n\t// handlerImpl is the domain operation handler implementation\n\thandlerImpl struct {\n\t\tdomainManager       persistence.DomainManager\n\t\tdomainAuditManager  persistence.DomainAuditManager\n\t\tclusterMetadata     cluster.Metadata\n\t\tdomainReplicator    Replicator\n\t\tdomainAttrValidator *AttrValidatorImpl\n\t\tarchivalMetadata    archiver.ArchivalMetadata\n\t\tarchiverProvider    provider.ArchiverProvider\n\t\ttimeSource          clock.TimeSource\n\t\tconfig              Config\n\t\tlogger              log.Logger\n\t}\n\n\t// Config is the domain config for domain handler\n\tConfig struct {\n\t\tMinRetentionDays         dynamicproperties.IntPropertyFn\n\t\tMaxRetentionDays         dynamicproperties.IntPropertyFn\n\t\tRequiredDomainDataKeys   dynamicproperties.MapPropertyFn\n\t\tMaxBadBinaryCount        dynamicproperties.IntPropertyFnWithDomainFilter\n\t\tFailoverCoolDown         dynamicproperties.DurationPropertyFnWithDomainFilter\n\t\tFailoverHistoryMaxSize   dynamicproperties.IntPropertyFnWithDomainFilter\n\t\tEnableDomainAuditLogging dynamicproperties.BoolPropertyFn\n\t}\n\n\t// FailoverEvent is the failover information to be stored for each failover event in domain data\n\tFailoverEvent struct {\n\t\tEventTime time.Time `json:\"eventTime\"`\n\n\t\t// active-passive domain failover\n\t\tFromCluster  string `json:\"fromCluster,omitempty\"`\n\t\tToCluster    string `json:\"toCluster,omitempty\"`\n\t\tFailoverType string `json:\"failoverType,omitempty\"`\n\t\tReason       string `json:\"reason,omitempty\"`\n\t}\n\n\t// FailoverHistory is the history of failovers for a domain limited by the FailoverHistoryMaxSize config\n\tFailoverHistory struct {\n\t\tFailoverEvents []FailoverEvent\n\t}\n)\n\nvar _ Handler = (*handlerImpl)(nil)\n\n// NewHandler create a new domain handler\nfunc NewHandler(\n\tconfig Config,\n\tlogger log.Logger,\n\tdomainManager persistence.DomainManager,\n\tdomainAuditManager persistence.DomainAuditManager,\n\tclusterMetadata cluster.Metadata,\n\tdomainReplicator Replicator,\n\tarchivalMetadata archiver.ArchivalMetadata,\n\tarchiverProvider provider.ArchiverProvider,\n\ttimeSource clock.TimeSource,\n) Handler {\n\treturn &handlerImpl{\n\t\tlogger:              logger,\n\t\tdomainManager:       domainManager,\n\t\tdomainAuditManager:  domainAuditManager,\n\t\tclusterMetadata:     clusterMetadata,\n\t\tdomainReplicator:    domainReplicator,\n\t\tdomainAttrValidator: newAttrValidator(clusterMetadata, int32(config.MinRetentionDays())),\n\t\tarchivalMetadata:    archivalMetadata,\n\t\tarchiverProvider:    archiverProvider,\n\t\ttimeSource:          timeSource,\n\t\tconfig:              config,\n\t}\n}\n\n// RegisterDomain register a new domain\nfunc (d *handlerImpl) RegisterDomain(\n\tctx context.Context,\n\tregisterRequest *types.RegisterDomainRequest,\n) error {\n\n\t// cluster global domain enabled\n\tif !d.clusterMetadata.IsPrimaryCluster() && registerRequest.GetIsGlobalDomain() {\n\t\treturn errNotPrimaryCluster\n\t}\n\n\t// first check if the name is already registered as the local domain\n\t_, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: registerRequest.GetName()})\n\tswitch err.(type) {\n\tcase nil:\n\t\t// domain already exists, cannot proceed\n\t\treturn &types.DomainAlreadyExistsError{Message: \"Domain already exists.\"}\n\tcase *types.EntityNotExistsError:\n\t\t// domain does not exists, proceeds\n\tdefault:\n\t\t// other err\n\t\treturn err\n\t}\n\n\t// input validation on domain name\n\tmatchedRegex, err := regexp.MatchString(\"^[a-zA-Z0-9-]+$\", registerRequest.GetName())\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !matchedRegex {\n\t\treturn errInvalidDomainName\n\t}\n\n\tactiveClusterName := d.clusterMetadata.GetCurrentClusterName()\n\t// input validation on cluster names\n\tif registerRequest.ActiveClusterName != \"\" {\n\t\tactiveClusterName = registerRequest.GetActiveClusterName()\n\t}\n\tclusters := []*persistence.ClusterReplicationConfig{}\n\tfor _, clusterConfig := range registerRequest.Clusters {\n\t\tclusterName := clusterConfig.GetClusterName()\n\t\tclusters = append(clusters, &persistence.ClusterReplicationConfig{ClusterName: clusterName})\n\t}\n\tclusters = cluster.GetOrUseDefaultClusters(activeClusterName, clusters)\n\n\tcurrentHistoryArchivalState := neverEnabledState()\n\tnextHistoryArchivalState := currentHistoryArchivalState\n\tclusterHistoryArchivalConfig := d.archivalMetadata.GetHistoryConfig()\n\tif clusterHistoryArchivalConfig.ClusterConfiguredForArchival() {\n\t\tarchivalEvent, err := d.toArchivalRegisterEvent(\n\t\t\tregisterRequest.HistoryArchivalStatus,\n\t\t\tregisterRequest.GetHistoryArchivalURI(),\n\t\t\tclusterHistoryArchivalConfig.GetDomainDefaultStatus(),\n\t\t\tclusterHistoryArchivalConfig.GetDomainDefaultURI(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tnextHistoryArchivalState, _, err = currentHistoryArchivalState.getNextState(archivalEvent, d.validateHistoryArchivalURI)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcurrentVisibilityArchivalState := neverEnabledState()\n\tnextVisibilityArchivalState := currentVisibilityArchivalState\n\tclusterVisibilityArchivalConfig := d.archivalMetadata.GetVisibilityConfig()\n\tif clusterVisibilityArchivalConfig.ClusterConfiguredForArchival() {\n\t\tarchivalEvent, err := d.toArchivalRegisterEvent(\n\t\t\tregisterRequest.VisibilityArchivalStatus,\n\t\t\tregisterRequest.GetVisibilityArchivalURI(),\n\t\t\tclusterVisibilityArchivalConfig.GetDomainDefaultStatus(),\n\t\t\tclusterVisibilityArchivalConfig.GetDomainDefaultURI(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tnextVisibilityArchivalState, _, err = currentVisibilityArchivalState.getNextState(archivalEvent, d.validateVisibilityArchivalURI)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\teventID, err := guuid.NewV7()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tinfo := &persistence.DomainInfo{\n\t\tID:          eventID.String(),\n\t\tName:        registerRequest.GetName(),\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tOwnerEmail:  registerRequest.GetOwnerEmail(),\n\t\tDescription: registerRequest.GetDescription(),\n\t\tData:        registerRequest.Data,\n\t}\n\tconfig := &persistence.DomainConfig{\n\t\tRetention:                registerRequest.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\tEmitMetric:               registerRequest.GetEmitMetric(),\n\t\tHistoryArchivalStatus:    nextHistoryArchivalState.Status,\n\t\tHistoryArchivalURI:       nextHistoryArchivalState.URI,\n\t\tVisibilityArchivalStatus: nextVisibilityArchivalState.Status,\n\t\tVisibilityArchivalURI:    nextVisibilityArchivalState.URI,\n\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t}\n\n\tactiveClusters, err := d.activeClustersFromRegisterRequest(registerRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: activeClusterName,\n\t\tClusters:          clusters,\n\t\tActiveClusters:    activeClusters,\n\t}\n\tisGlobalDomain := registerRequest.GetIsGlobalDomain()\n\n\tif err := d.domainAttrValidator.validateDomainConfig(config); err != nil {\n\t\treturn err\n\t}\n\tif isGlobalDomain {\n\t\tif err := d.domainAttrValidator.validateDomainReplicationConfigForGlobalDomain(\n\t\t\treplicationConfig,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\tif err := d.domainAttrValidator.validateDomainReplicationConfigForLocalDomain(\n\t\t\treplicationConfig,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tfailoverVersion := constants.EmptyVersion\n\tif registerRequest.GetIsGlobalDomain() {\n\t\tfailoverVersion = d.clusterMetadata.GetNextFailoverVersion(activeClusterName, 0, registerRequest.Name)\n\t}\n\n\tdomainRequest := &persistence.CreateDomainRequest{\n\t\tInfo:              info,\n\t\tConfig:            config,\n\t\tReplicationConfig: replicationConfig,\n\t\tIsGlobalDomain:    isGlobalDomain,\n\t\tConfigVersion:     0,\n\t\tFailoverVersion:   failoverVersion,\n\t\tLastUpdatedTime:   d.timeSource.Now().UnixNano(),\n\t}\n\n\tdomainResponse, err := d.domainManager.CreateDomain(ctx, domainRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif domainRequest.IsGlobalDomain {\n\t\terr = d.domainReplicator.HandleTransmissionTask(\n\t\t\tctx,\n\t\t\ttypes.DomainOperationCreate,\n\t\t\tdomainRequest.Info,\n\t\t\tdomainRequest.Config,\n\t\t\tdomainRequest.ReplicationConfig,\n\t\t\tdomainRequest.ConfigVersion,\n\t\t\tdomainRequest.FailoverVersion,\n\t\t\tconstants.InitialPreviousFailoverVersion,\n\t\t\tdomainRequest.IsGlobalDomain,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\td.logger.Info(\"Register domain succeeded\",\n\t\ttag.WorkflowDomainName(registerRequest.GetName()),\n\t\ttag.WorkflowDomainID(domainResponse.ID),\n\t)\n\n\t// Construct GetDomainResponse for audit log\n\tdomainStateAfterCreate := &persistence.GetDomainResponse{\n\t\tInfo:              domainRequest.Info,\n\t\tConfig:            domainRequest.Config,\n\t\tReplicationConfig: domainRequest.ReplicationConfig,\n\t\tIsGlobalDomain:    domainRequest.IsGlobalDomain,\n\t\tConfigVersion:     domainRequest.ConfigVersion,\n\t\tFailoverVersion:   domainRequest.FailoverVersion,\n\t\tLastUpdatedTime:   domainRequest.LastUpdatedTime,\n\t}\n\n\terr = d.updateDomainAuditLog(ctx, nil, domainStateAfterCreate, persistence.DomainAuditOperationTypeCreate, \"domain created\")\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\n// ListDomains list all domains\nfunc (d *handlerImpl) ListDomains(\n\tctx context.Context,\n\tlistRequest *types.ListDomainsRequest,\n) (*types.ListDomainsResponse, error) {\n\n\tpageSize := 100\n\tif listRequest.GetPageSize() != 0 {\n\t\tpageSize = int(listRequest.GetPageSize())\n\t}\n\n\tresp, err := d.domainManager.ListDomains(ctx, &persistence.ListDomainsRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: listRequest.NextPageToken,\n\t})\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomains := []*types.DescribeDomainResponse{}\n\tfor _, domain := range resp.Domains {\n\t\tdesc := &types.DescribeDomainResponse{\n\t\t\tIsGlobalDomain:  domain.IsGlobalDomain,\n\t\t\tFailoverVersion: domain.FailoverVersion,\n\t\t}\n\t\tdesc.DomainInfo, desc.Configuration, desc.ReplicationConfiguration = d.createResponse(domain.Info, domain.Config, domain.ReplicationConfig)\n\t\tdomains = append(domains, desc)\n\t}\n\n\tresponse := &types.ListDomainsResponse{\n\t\tDomains:       domains,\n\t\tNextPageToken: resp.NextPageToken,\n\t}\n\n\treturn response, nil\n}\n\n// DescribeDomain describe the domain\nfunc (d *handlerImpl) DescribeDomain(\n\tctx context.Context,\n\tdescribeRequest *types.DescribeDomainRequest,\n) (*types.DescribeDomainResponse, error) {\n\n\t// TODO, we should migrate the non global domain to new table, see #773\n\treq := &persistence.GetDomainRequest{\n\t\tName: describeRequest.GetName(),\n\t\tID:   describeRequest.GetUUID(),\n\t}\n\tresp, err := d.domainManager.GetDomain(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresponse := &types.DescribeDomainResponse{\n\t\tIsGlobalDomain:  resp.IsGlobalDomain,\n\t\tFailoverVersion: resp.FailoverVersion,\n\t}\n\tif resp.FailoverEndTime != nil {\n\t\tresponse.FailoverInfo = &types.FailoverInfo{\n\t\t\tFailoverVersion: resp.FailoverVersion,\n\t\t\t// This reflects that last domain update time. If there is a domain config update, this won't be accurate.\n\t\t\tFailoverStartTimestamp:  resp.LastUpdatedTime,\n\t\t\tFailoverExpireTimestamp: *resp.FailoverEndTime,\n\t\t}\n\t}\n\tresponse.DomainInfo, response.Configuration, response.ReplicationConfiguration = d.createResponse(resp.Info, resp.Config, resp.ReplicationConfig)\n\treturn response, nil\n}\n\n// UpdateDomain update the domain\nfunc (d *handlerImpl) UpdateDomain(\n\tctx context.Context,\n\tupdateRequest *types.UpdateDomainRequest,\n) (*types.UpdateDomainResponse, error) {\n\n\t// must get the metadata (notificationVersion) first\n\t// this version can be regarded as the lock on the v2 domain table\n\t// and since we do not know which table will return the domain afterwards\n\t// this call has to be made\n\tmetadata, err := d.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\tcurrentDomainState, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tisGlobalDomain := currentDomainState.IsGlobalDomain\n\n\tif !isGlobalDomain {\n\t\treturn d.updateLocalDomain(ctx, updateRequest, currentDomainState, notificationVersion)\n\t}\n\n\tif updateRequest.IsAFailoverRequest() {\n\t\treturn d.handleFailoverRequest(ctx, updateRequest, currentDomainState, notificationVersion)\n\t}\n\n\treturn d.updateGlobalDomainConfiguration(ctx, updateRequest, currentDomainState, notificationVersion)\n}\n\n// All domain updates are throttled by the cool down time (incorrecty called 'failover' cool down).\n// The guard is an anti-flapping measure.\nfunc (d *handlerImpl) ensureUpdateOrFailoverCooldown(currentDomainState *persistence.GetDomainResponse) error {\n\tlastUpdatedTime := time.Unix(0, currentDomainState.LastUpdatedTime)\n\tnow := d.timeSource.Now()\n\tif lastUpdatedTime.Add(d.config.FailoverCoolDown(currentDomainState.Info.Name)).After(now) {\n\t\td.logger.Debugf(\"Domain was last updated at %v, failoverCoolDown: %v, current time: %v.\", lastUpdatedTime, d.config.FailoverCoolDown(currentDomainState.Info.Name), now)\n\t\treturn errDomainUpdateTooFrequent\n\t}\n\treturn nil\n}\n\n// For global domains only, this is assumed to be invoked when\n// the incoming request is either specifying an active-cluster parameter to update\n// or active_clusters in the case of a AA domain\nfunc (d *handlerImpl) handleFailoverRequest(ctx context.Context,\n\tupdateRequest *types.UpdateDomainRequest,\n\tcurrentState *persistence.GetDomainResponse,\n\tnotificationVersion int64,\n) (*types.UpdateDomainResponse, error) {\n\n\t// intendedDomainState will be modified\n\t// into the intended shape by the functions here\n\tintendedDomainState := currentState.DeepCopy()\n\n\tisGlobalDomain := currentState.IsGlobalDomain\n\n\tcurrentActiveCluster := currentState.ReplicationConfig.ActiveClusterName\n\twasActiveActive := currentState.ReplicationConfig.IsActiveActive()\n\tnow := d.timeSource.Now()\n\n\t// by default, we assume failovers are of type force\n\tfailoverType := constants.FailoverTypeForce\n\n\tvar activeClusterChanged bool\n\tvar configurationChanged bool\n\n\t// will be set to the domain notification version after the update\n\tintendedDomainState.FailoverNotificationVersion = types.UndefinedFailoverVersion\n\t// not used except for graceful failover requests, but specifically set to -1\n\t// so as to be explicitly undefined\n\tintendedDomainState.PreviousFailoverVersion = constants.InitialPreviousFailoverVersion\n\n\t// by default, we assume a force failover and that any preexisting graceful failover state is invalidated\n\t// if there's a duration of failover time to occur (such as in graceful failover) this will be re-set.\n\t// But if there's an existing graceful failover and a subsequent force\n\t// we want to ensure that it'll be ended immmediately.\n\tintendedDomainState.FailoverEndTime = nil\n\n\t// Update replication config\n\treplicationCfg, replicationConfigChanged, activeClusterChanged, err := d.updateReplicationConfig(\n\t\tcurrentState.Info.Name,\n\t\tintendedDomainState.ReplicationConfig,\n\t\tupdateRequest,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !activeClusterChanged && !replicationConfigChanged {\n\t\treturn nil, errInvalidFailoverNoChangeDetected\n\t}\n\tintendedDomainState.ReplicationConfig = replicationCfg\n\n\terr = d.ensureUpdateOrFailoverCooldown(currentState)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// if the failover 'graceful' - as indicated as having a FailoverTimeoutInSeconds,\n\t// then we set some additional parameters for the graceful failover\n\tif updateRequest.FailoverTimeoutInSeconds != nil {\n\t\tgracefulFailoverEndTime, previousFailoverVersion, err := d.handleGracefulFailover(\n\t\t\tupdateRequest,\n\t\t\tintendedDomainState.ReplicationConfig,\n\t\t\tcurrentActiveCluster,\n\t\t\tcurrentState.FailoverEndTime,\n\t\t\tcurrentState.FailoverVersion,\n\t\t\tactiveClusterChanged,\n\t\t\tisGlobalDomain,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tfailoverType = constants.FailoverTypeGrace\n\t\tintendedDomainState.FailoverEndTime = gracefulFailoverEndTime\n\t\tintendedDomainState.PreviousFailoverVersion = previousFailoverVersion\n\t}\n\n\t// replication config is a subset of config,\n\tconfigurationChanged = replicationConfigChanged\n\n\tif err = d.domainAttrValidator.validateDomainConfig(intendedDomainState.Config); err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = d.validateDomainReplicationConfigForFailover(intendedDomainState.ReplicationConfig, configurationChanged, activeClusterChanged)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// increment the in the configuration fencing token to ensure that configurations\n\t// are applied in order\n\tif configurationChanged {\n\t\tintendedDomainState.ConfigVersion++\n\t}\n\n\tintendedDomainState.FailoverVersion = d.clusterMetadata.GetNextFailoverVersion(\n\t\tintendedDomainState.ReplicationConfig.ActiveClusterName,\n\t\tcurrentState.FailoverVersion,\n\t\tupdateRequest.Name,\n\t)\n\n\t// this is an intended failover step, to capture the current cluster's domain configuration\n\t// (represented by the notification-version counter) and set it when running failover to allow\n\t// systems like graceful failover to dedup failover processes and callbacks\n\tintendedDomainState.FailoverNotificationVersion = notificationVersion\n\n\tisActiveActive := intendedDomainState.ReplicationConfig.IsActiveActive()\n\n\tif wasActiveActive || isActiveActive {\n\t\t// if the domain was ever active-active, we bump the failover-version at the domain level\n\t\t// to ensure that it gets incremented every single time, even though the failover may be only\n\t\t// at the cluster-attribute level (ie the FailoverVersion may just go up by the failover-increment\n\t\t// but otherwise be a noop - the actual active cluster may not have changed)).\n\t\t//\n\t\t// The reason for this is caution and simplicity at the time of writing:\n\t\t// It's harmless to bump and this simplifies thinking about failover events\n\t\t// through out the rest of the Cadence codebase. We can reliably assume that every failover\n\t\t// event that has meaningful changes only needs to subscribe to this fencing token.\n\t\t// to detect changes.\n\t\t//\n\t\t// Any parts of the system (domain-callbacks, domain cache etc) that watch for\n\t\t// failover events and may trigger some action or cache invalidation will be watching\n\t\t// the domain-level failver counter for changes. Therefore, by bumping it even for\n\t\t// cases where it isn't changing, we ensure all these other subprocesses will\n\t\t// take\n\t\tintendedDomainState.FailoverVersion = d.clusterMetadata.GetNextFailoverVersion(\n\t\t\tintendedDomainState.ReplicationConfig.ActiveClusterName,\n\t\t\tcurrentState.FailoverVersion+1,\n\t\t\tupdateRequest.Name,\n\t\t)\n\t}\n\n\t// the domain-data table is only updated for active-passive domain failovers\n\t// as a historical backwards compatibility measure.\n\t// Going forward, any history use-cases should rely on the FailoverHistory endpoint which\n\t// supports all failover types and more than a handful of entries.\n\tif !wasActiveActive && !isActiveActive {\n\t\terr = updateFailoverHistoryInDomainData(intendedDomainState.Info, d.config, NewFailoverEvent(\n\t\t\tnow,\n\t\t\tfailoverType,\n\t\t\t&currentActiveCluster,\n\t\t\tupdateRequest.ActiveClusterName,\n\t\t\tupdateRequest.FailoverReason,\n\t\t))\n\t\tif err != nil {\n\t\t\td.logger.Warn(\"failed to update failover history\", tag.Error(err))\n\t\t}\n\t}\n\n\tupdateReq := createUpdateRequest(\n\t\tintendedDomainState.Info,\n\t\tintendedDomainState.Config,\n\t\tintendedDomainState.ReplicationConfig,\n\t\tintendedDomainState.ConfigVersion,\n\t\tintendedDomainState.FailoverVersion,\n\t\tintendedDomainState.FailoverNotificationVersion,\n\t\tintendedDomainState.FailoverEndTime,\n\t\tintendedDomainState.PreviousFailoverVersion,\n\t\tnow,\n\t\tnotificationVersion,\n\t)\n\n\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif err = d.domainReplicator.HandleTransmissionTask(\n\t\tctx,\n\t\ttypes.DomainOperationUpdate,\n\t\tintendedDomainState.Info,\n\t\tintendedDomainState.Config,\n\t\tintendedDomainState.ReplicationConfig,\n\t\tintendedDomainState.ConfigVersion,\n\t\tintendedDomainState.FailoverVersion,\n\t\tintendedDomainState.PreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\tresponse := &types.UpdateDomainResponse{\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tFailoverVersion: intendedDomainState.FailoverVersion,\n\t}\n\tresponse.DomainInfo, response.Configuration, response.ReplicationConfiguration = d.createResponse(intendedDomainState.Info, intendedDomainState.Config, intendedDomainState.ReplicationConfig)\n\n\terr = d.updateDomainAuditLog(ctx, currentState, intendedDomainState, persistence.DomainAuditOperationTypeFailover, \"domain failover\")\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\td.logger.Info(\"Failover request succeeded\",\n\t\ttag.WorkflowDomainName(intendedDomainState.Info.Name),\n\t\ttag.WorkflowDomainID(intendedDomainState.Info.ID),\n\t)\n\treturn response, nil\n}\n\nfunc (d *handlerImpl) updateDomainAuditLog(ctx context.Context,\n\tcurrentState *persistence.GetDomainResponse,\n\tintendedDomainState *persistence.GetDomainResponse,\n\toperationType persistence.DomainAuditOperationType,\n\tcomment string,\n) error {\n\n\tif d.domainAuditManager == nil {\n\t\treturn nil\n\t}\n\n\tif !d.config.EnableDomainAuditLogging() {\n\t\treturn nil\n\t}\n\n\t// Must be a UUID v7, since we need a time value as well\n\teventID, err := guuid.NewV7()\n\tif err != nil {\n\t\treturn err\n\t}\n\t// the creation time is used in the database as a partition but passed around\n\t// embedded in the eventUUID for ergonomics. This means that users wishing\n\t// to get an audit entry by ID do not need to know the creation time in advance\n\t// since it's embedded in the sorting-values of the UUID.\n\tcreationTime := time.Unix(eventID.Time().UnixTime())\n\n\t_, err = d.domainAuditManager.CreateDomainAuditLog(ctx, &persistence.CreateDomainAuditLogRequest{\n\t\tDomainID:      intendedDomainState.GetInfo().GetID(),\n\t\tEventID:       eventID.String(),\n\t\tCreatedTime:   creationTime,\n\t\tStateBefore:   currentState,\n\t\tStateAfter:    intendedDomainState,\n\t\tOperationType: operationType,\n\t\tComment:       comment,\n\t})\n\tif err != nil {\n\t\td.logger.Error(\"Failed to create domain audit log\",\n\t\t\ttag.WorkflowDomainID(intendedDomainState.GetInfo().GetID()),\n\t\t\ttag.Error(err),\n\t\t)\n\t\t// Log the error but don't fail the operation - audit logging is best effort\n\t\t// to avoid breaking critical domain operations\n\t}\n\treturn nil\n}\n\n// updateGlobalDomainConfiguration handles the update of a global domain configuration\n// this excludes failover/active_cluster/active_clusters updates. They are grouped under\n// forms of failover\nfunc (d *handlerImpl) updateGlobalDomainConfiguration(ctx context.Context,\n\tupdateRequest *types.UpdateDomainRequest,\n\tcurrentDomainState *persistence.GetDomainResponse,\n\tnotificationVersion int64,\n) (*types.UpdateDomainResponse, error) {\n\n\t// intendedDomainState will be modified\n\t// into the intended shape by the functions here\n\tintendedDomainState := currentDomainState.DeepCopy()\n\n\tconfigVersion := currentDomainState.ConfigVersion\n\tfailoverVersion := currentDomainState.FailoverVersion\n\tisGlobalDomain := currentDomainState.IsGlobalDomain\n\n\tnow := d.timeSource.Now()\n\n\t// whether history archival config changed\n\thistoryArchivalConfigChanged := false\n\t// whether visibility archival config changed\n\tvisibilityArchivalConfigChanged := false\n\t// whether active cluster is changed\n\tactiveClusterChanged := false\n\t// whether anything other than active cluster is changed\n\tconfigurationChanged := false\n\n\t// Update history archival state\n\thistoryArchivalConfigChanged, err := d.updateHistoryArchivalState(intendedDomainState.Config, updateRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update visibility archival state\n\tvisibilityArchivalConfigChanged, err = d.updateVisibilityArchivalState(intendedDomainState.Config, updateRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update domain info\n\tinfo, domainInfoChanged := d.updateDomainInfo(\n\t\tupdateRequest,\n\t\tintendedDomainState.Info,\n\t)\n\n\t// Update domain config\n\tconfig, domainConfigChanged, err := d.updateDomainConfiguration(\n\t\tupdateRequest.GetName(),\n\t\tintendedDomainState.Config,\n\t\tupdateRequest,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update domain bad binary\n\tconfig, deleteBinaryChanged, err := d.updateDeleteBadBinary(\n\t\tconfig,\n\t\tupdateRequest.DeleteBadBinary,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update replication config\n\treplicationConfig, replicationConfigChanged, activeClusterChanged, err := d.updateReplicationConfig(\n\t\tintendedDomainState.Info.Name,\n\t\tintendedDomainState.ReplicationConfig,\n\t\tupdateRequest,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfigurationChanged = historyArchivalConfigChanged || visibilityArchivalConfigChanged || domainInfoChanged || domainConfigChanged || deleteBinaryChanged || replicationConfigChanged\n\n\tif err = d.domainAttrValidator.validateDomainConfig(config); err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = d.validateGlobalDomainReplicationConfigForUpdateDomain(replicationConfig, configurationChanged, activeClusterChanged)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = d.ensureUpdateOrFailoverCooldown(currentDomainState)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif configurationChanged || activeClusterChanged {\n\n\t\t// set the versions\n\t\tif configurationChanged {\n\t\t\tconfigVersion++\n\t\t}\n\n\t\tupdateReq := createUpdateRequest(\n\t\t\tinfo,\n\t\t\tconfig,\n\t\t\treplicationConfig,\n\t\t\tconfigVersion,\n\t\t\tfailoverVersion,\n\t\t\tcurrentDomainState.FailoverNotificationVersion,\n\t\t\tintendedDomainState.FailoverEndTime,\n\t\t\tintendedDomainState.PreviousFailoverVersion,\n\t\t\tnow,\n\t\t\tnotificationVersion,\n\t\t)\n\n\t\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tif err = d.domainReplicator.HandleTransmissionTask(\n\t\tctx,\n\t\ttypes.DomainOperationUpdate,\n\t\tinfo,\n\t\tconfig,\n\t\treplicationConfig,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tintendedDomainState.PreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\tresponse := &types.UpdateDomainResponse{\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\tresponse.DomainInfo, response.Configuration, response.ReplicationConfiguration = d.createResponse(info, config, replicationConfig)\n\n\t// Construct GetDomainResponse for audit log with the final updated values\n\tdomainStateAfterUpdate := &persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      config,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              isGlobalDomain,\n\t\tConfigVersion:               configVersion,\n\t\tFailoverVersion:             failoverVersion,\n\t\tFailoverNotificationVersion: intendedDomainState.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     intendedDomainState.PreviousFailoverVersion,\n\t\tFailoverEndTime:             intendedDomainState.FailoverEndTime,\n\t\tLastUpdatedTime:             now.UnixNano(),\n\t\tNotificationVersion:         notificationVersion,\n\t}\n\n\terr = d.updateDomainAuditLog(ctx, currentDomainState, domainStateAfterUpdate, persistence.DomainAuditOperationTypeUpdate, \"domain updated\")\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\td.logger.Info(\"Update domain succeeded\",\n\t\ttag.WorkflowDomainName(info.Name),\n\t\ttag.WorkflowDomainID(info.ID),\n\t)\n\treturn response, nil\n}\n\nfunc (d *handlerImpl) updateLocalDomain(ctx context.Context,\n\tupdateRequest *types.UpdateDomainRequest,\n\tcurrentState *persistence.GetDomainResponse,\n\tnotificationVersion int64,\n) (*types.UpdateDomainResponse, error) {\n\n\terr := d.domainAttrValidator.validateLocalDomainUpdateRequest(updateRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// whether history archival config changed\n\thistoryArchivalConfigChanged := false\n\t// whether visibility archival config changed\n\tvisibilityArchivalConfigChanged := false\n\t// whether anything other than active cluster is changed\n\tconfigurationChanged := false\n\n\tintendedDomainState := currentState.DeepCopy()\n\n\tconfigVersion := currentState.ConfigVersion\n\n\tnow := d.timeSource.Now()\n\n\t// Update history archival state\n\thistoryArchivalConfigChanged, err = d.updateHistoryArchivalState(intendedDomainState.Config, updateRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update visibility archival state\n\tvisibilityArchivalConfigChanged, err = d.updateVisibilityArchivalState(intendedDomainState.Config, updateRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update domain info\n\tinfo, domainInfoChanged := d.updateDomainInfo(\n\t\tupdateRequest,\n\t\tintendedDomainState.Info,\n\t)\n\n\t// Update domain config\n\tconfig, domainConfigChanged, err := d.updateDomainConfiguration(\n\t\tupdateRequest.GetName(),\n\t\tintendedDomainState.Config,\n\t\tupdateRequest,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Update domain bad binary\n\tconfig, deleteBinaryChanged, err := d.updateDeleteBadBinary(\n\t\tconfig,\n\t\tupdateRequest.DeleteBadBinary,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfigurationChanged = historyArchivalConfigChanged || visibilityArchivalConfigChanged || domainInfoChanged || domainConfigChanged || deleteBinaryChanged\n\n\tif err = d.domainAttrValidator.validateDomainConfig(config); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err = d.domainAttrValidator.validateDomainReplicationConfigForLocalDomain(\n\t\tintendedDomainState.ReplicationConfig,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif configurationChanged {\n\t\t// set the versions\n\t\tif configurationChanged {\n\t\t\tconfigVersion = intendedDomainState.ConfigVersion + 1\n\t\t}\n\n\t\tupdateReq := createUpdateRequest(\n\t\t\tinfo,\n\t\t\tconfig,\n\t\t\tintendedDomainState.ReplicationConfig,\n\t\t\tconfigVersion,\n\t\t\tintendedDomainState.FailoverVersion,\n\t\t\tintendedDomainState.FailoverNotificationVersion,\n\t\t\tintendedDomainState.FailoverEndTime,\n\t\t\tintendedDomainState.PreviousFailoverVersion,\n\t\t\tnow,\n\t\t\tnotificationVersion,\n\t\t)\n\n\t\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\terr = d.updateDomainAuditLog(ctx, currentState, intendedDomainState, persistence.DomainAuditOperationTypeUpdate, \"domain updated\")\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tresponse := &types.UpdateDomainResponse{\n\t\tIsGlobalDomain:  false,\n\t\tFailoverVersion: intendedDomainState.FailoverVersion,\n\t}\n\tresponse.DomainInfo, response.Configuration, response.ReplicationConfiguration = d.createResponse(info, config, intendedDomainState.ReplicationConfig)\n\n\treturn response, nil\n}\n\n// FailoverDomain handles failover of the domain to a different cluster\nfunc (d *handlerImpl) FailoverDomain(\n\tctx context.Context,\n\tfailoverRequest *types.FailoverDomainRequest,\n) (*types.FailoverDomainResponse, error) {\n\n\tmetadata, err := d.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\n\tcurrentDomainState, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: failoverRequest.GetDomainName()})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = d.validateDomainFailoverRequest(failoverRequest, currentDomainState)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresponse, err := d.handleFailoverRequest(\n\t\tctx,\n\t\tfailoverRequest.ToUpdateDomainRequest(),\n\t\tcurrentDomainState,\n\t\tnotificationVersion,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response.ToFailoverDomainResponse(), nil\n}\n\n// DeleteDomain deletes a domain\nfunc (d *handlerImpl) DeleteDomain(\n\tctx context.Context,\n\tdeleteRequest *types.DeleteDomainRequest,\n) error {\n\tgetResponse, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: deleteRequest.GetName()})\n\tif err != nil {\n\t\treturn err\n\t}\n\tisGlobalDomain := getResponse.IsGlobalDomain\n\tif isGlobalDomain && !d.clusterMetadata.IsPrimaryCluster() {\n\t\treturn errNotPrimaryCluster\n\t}\n\n\tdeleteReq := &persistence.DeleteDomainByNameRequest{\n\t\tName: getResponse.Info.Name,\n\t}\n\terr = d.domainManager.DeleteDomainByName(ctx, deleteReq)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif isGlobalDomain {\n\t\tif err := d.domainReplicator.HandleTransmissionTask(\n\t\t\tctx,\n\t\t\ttypes.DomainOperationDelete,\n\t\t\tgetResponse.Info,\n\t\t\tgetResponse.Config,\n\t\t\tgetResponse.ReplicationConfig,\n\t\t\tgetResponse.ConfigVersion,\n\t\t\tgetResponse.FailoverVersion,\n\t\t\tgetResponse.PreviousFailoverVersion,\n\t\t\tisGlobalDomain,\n\t\t); err != nil {\n\t\t\treturn fmt.Errorf(\"unable to delete a domain in replica cluster: %v\", err)\n\t\t}\n\t}\n\n\terr = d.updateDomainAuditLog(ctx, getResponse, nil, persistence.DomainAuditOperationTypeDelete, \"domain deleted\")\n\tif err != nil {\n\t\treturn err\n\t}\n\n\td.logger.Info(\"Delete domain succeeded\",\n\t\ttag.WorkflowDomainName(getResponse.Info.Name),\n\t\ttag.WorkflowDomainID(getResponse.Info.ID),\n\t)\n\treturn nil\n}\n\n// DeprecateDomain deprecates a domain\nfunc (d *handlerImpl) DeprecateDomain(\n\tctx context.Context,\n\tdeprecateRequest *types.DeprecateDomainRequest,\n) error {\n\n\t// must get the metadata (notificationVersion) first\n\t// this version can be regarded as the lock on the v2 domain table\n\t// and since we do not know which table will return the domain afterwards\n\t// this call has to be made\n\tmetadata, err := d.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\tgetResponse, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: deprecateRequest.GetName()})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tisGlobalDomain := getResponse.IsGlobalDomain\n\tif isGlobalDomain && !d.clusterMetadata.IsPrimaryCluster() {\n\t\treturn errNotPrimaryCluster\n\t}\n\tgetResponse.ConfigVersion = getResponse.ConfigVersion + 1\n\tgetResponse.Info.Status = persistence.DomainStatusDeprecated\n\n\tupdateReq := createUpdateRequest(\n\t\tgetResponse.Info,\n\t\tgetResponse.Config,\n\t\tgetResponse.ReplicationConfig,\n\t\tgetResponse.ConfigVersion,\n\t\tgetResponse.FailoverVersion,\n\t\tgetResponse.FailoverNotificationVersion,\n\t\tgetResponse.FailoverEndTime,\n\t\tgetResponse.PreviousFailoverVersion,\n\t\td.timeSource.Now(),\n\t\tnotificationVersion,\n\t)\n\n\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif isGlobalDomain {\n\t\tif err := d.domainReplicator.HandleTransmissionTask(\n\t\t\tctx,\n\t\t\ttypes.DomainOperationUpdate,\n\t\t\tgetResponse.Info,\n\t\t\tgetResponse.Config,\n\t\t\tgetResponse.ReplicationConfig,\n\t\t\tgetResponse.ConfigVersion,\n\t\t\tgetResponse.FailoverVersion,\n\t\t\tgetResponse.PreviousFailoverVersion,\n\t\t\tisGlobalDomain,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\td.logger.Info(\"DeprecateDomain domain succeeded\",\n\t\ttag.WorkflowDomainName(getResponse.Info.Name),\n\t\ttag.WorkflowDomainID(getResponse.Info.ID),\n\t)\n\n\tdomainStateAfterDeprecate := &persistence.GetDomainResponse{\n\t\tInfo: getResponse.Info,\n\t}\n\terr = d.updateDomainAuditLog(ctx, getResponse, domainStateAfterDeprecate, persistence.DomainAuditOperationTypeDeprecate, \"domain deprecated\")\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// UpdateIsolationGroups is used for draining and undraining of isolation-groups for a domain.\n// Like the isolation-group API, this controller expects Upsert semantics for\n// isolation-groups and does not modify any other domain information.\n//\n// Isolation-groups are regional in their configuration scope, so it's expected that this upsert\n// includes configuration for both clusters every time.\n//\n// The update is handled like other domain updates in that they expected to be replicated. So\n// unlike the global isolation-group API it shouldn't be necessary to call\nfunc (d *handlerImpl) UpdateIsolationGroups(\n\tctx context.Context,\n\tupdateRequest types.UpdateDomainIsolationGroupsRequest,\n) error {\n\n\t// must get the metadata (notificationVersion) first\n\t// this version can be regarded as the lock on the v2 domain table\n\t// and since we do not know which table will return the domain afterwards\n\t// this call has to be made\n\tmetadata, err := d.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\n\tif updateRequest.IsolationGroups == nil {\n\t\treturn fmt.Errorf(\"invalid request, isolationGroup configuration must be set: Got: %v\", updateRequest)\n\t}\n\n\tcurrentDomainConfig, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.Domain})\n\tif err != nil {\n\t\treturn err\n\t}\n\tif currentDomainConfig.Config == nil {\n\t\treturn fmt.Errorf(\"unable to load config for domain as expected\")\n\t}\n\n\tconfigVersion := currentDomainConfig.ConfigVersion\n\tlastUpdatedTime := time.Unix(0, currentDomainConfig.LastUpdatedTime)\n\n\t// Check the failover cool down time\n\tif lastUpdatedTime.Add(d.config.FailoverCoolDown(currentDomainConfig.Info.Name)).After(d.timeSource.Now()) {\n\t\treturn errDomainUpdateTooFrequent\n\t}\n\n\tif !d.clusterMetadata.IsPrimaryCluster() && currentDomainConfig.IsGlobalDomain {\n\t\treturn errNotPrimaryCluster\n\t}\n\n\tconfigVersion++\n\tlastUpdatedTime = d.timeSource.Now()\n\n\t// Mutate the domain config to perform the isolation-group update\n\tcurrentDomainConfig.Config.IsolationGroups = updateRequest.IsolationGroups\n\n\tupdateReq := createUpdateRequest(\n\t\tcurrentDomainConfig.Info,\n\t\tcurrentDomainConfig.Config,\n\t\tcurrentDomainConfig.ReplicationConfig,\n\t\tconfigVersion,\n\t\tcurrentDomainConfig.FailoverVersion,\n\t\tcurrentDomainConfig.FailoverNotificationVersion,\n\t\tcurrentDomainConfig.FailoverEndTime,\n\t\tcurrentDomainConfig.PreviousFailoverVersion,\n\t\tlastUpdatedTime,\n\t\tnotificationVersion,\n\t)\n\n\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif currentDomainConfig.IsGlobalDomain {\n\t\t// One might reasonably wonder what value there is in replication of isolation-group information - info which is\n\t\t// regional and therefore of no value to the other region?\n\t\t// Probably not a lot, in and of itself, however, the isolation-group information is stored\n\t\t// in the domain configuration fields in the domain tables. Access and updates to those records is\n\t\t// done through a replicated mechanism with explicit versioning and conflict resolution.\n\t\t// Therefore, in order to avoid making an already complex mechanisim much more difficult to understand,\n\t\t// the data is replicated in the same way so as to try and make things less confusing when both codepaths\n\t\t// are updating the table:\n\t\t// - versions like the confiugration version are updated in the same manner\n\t\t// - the last-updated timestamps are updated in the same manner\n\t\tif err := d.domainReplicator.HandleTransmissionTask(\n\t\t\tctx,\n\t\t\ttypes.DomainOperationUpdate,\n\t\t\tcurrentDomainConfig.Info,\n\t\t\tcurrentDomainConfig.Config,\n\t\t\tcurrentDomainConfig.ReplicationConfig,\n\t\t\tconfigVersion,\n\t\t\tcurrentDomainConfig.FailoverVersion,\n\t\t\tcurrentDomainConfig.PreviousFailoverVersion,\n\t\t\tcurrentDomainConfig.IsGlobalDomain,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\td.logger.Info(\"isolation group update succeeded\",\n\t\ttag.WorkflowDomainName(currentDomainConfig.Info.Name),\n\t\ttag.WorkflowDomainID(currentDomainConfig.Info.ID),\n\t)\n\treturn nil\n}\n\nfunc (d *handlerImpl) UpdateAsyncWorkflowConfiguraton(\n\tctx context.Context,\n\tupdateRequest types.UpdateDomainAsyncWorkflowConfiguratonRequest,\n) error {\n\t// must get the metadata (notificationVersion) first\n\t// this version can be regarded as the lock on the v2 domain table\n\t// and since we do not know which table will return the domain afterwards\n\t// this call has to be made\n\tmetadata, err := d.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\n\tcurrentDomainConfig, err := d.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.Domain})\n\tif err != nil {\n\t\treturn err\n\t}\n\tif currentDomainConfig.Config == nil {\n\t\treturn fmt.Errorf(\"unable to load config for domain as expected\")\n\t}\n\n\tconfigVersion := currentDomainConfig.ConfigVersion\n\tlastUpdatedTime := time.Unix(0, currentDomainConfig.LastUpdatedTime)\n\n\t// Check the failover cool down time\n\tif lastUpdatedTime.Add(d.config.FailoverCoolDown(currentDomainConfig.Info.Name)).After(d.timeSource.Now()) {\n\t\treturn errDomainUpdateTooFrequent\n\t}\n\n\tif !d.clusterMetadata.IsPrimaryCluster() && currentDomainConfig.IsGlobalDomain {\n\t\treturn errNotPrimaryCluster\n\t}\n\n\tconfigVersion++\n\tlastUpdatedTime = d.timeSource.Now()\n\n\t// Mutate the domain config to perform the async wf config update\n\tif updateRequest.Configuration == nil {\n\t\t// this is a delete request so empty all the fields\n\t\tcurrentDomainConfig.Config.AsyncWorkflowConfig = types.AsyncWorkflowConfiguration{}\n\t} else {\n\t\tcurrentDomainConfig.Config.AsyncWorkflowConfig = *updateRequest.Configuration\n\t}\n\n\td.logger.Debug(\"async workflow queue config update\", tag.Dynamic(\"config\", currentDomainConfig))\n\n\tupdateReq := createUpdateRequest(\n\t\tcurrentDomainConfig.Info,\n\t\tcurrentDomainConfig.Config,\n\t\tcurrentDomainConfig.ReplicationConfig,\n\t\tconfigVersion,\n\t\tcurrentDomainConfig.FailoverVersion,\n\t\tcurrentDomainConfig.FailoverNotificationVersion,\n\t\tcurrentDomainConfig.FailoverEndTime,\n\t\tcurrentDomainConfig.PreviousFailoverVersion,\n\t\tlastUpdatedTime,\n\t\tnotificationVersion,\n\t)\n\n\terr = d.domainManager.UpdateDomain(ctx, &updateReq)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif currentDomainConfig.IsGlobalDomain {\n\t\tif err := d.domainReplicator.HandleTransmissionTask(\n\t\t\tctx,\n\t\t\ttypes.DomainOperationUpdate,\n\t\t\tcurrentDomainConfig.Info,\n\t\t\tcurrentDomainConfig.Config,\n\t\t\tcurrentDomainConfig.ReplicationConfig,\n\t\t\tconfigVersion,\n\t\t\tcurrentDomainConfig.FailoverVersion,\n\t\t\tcurrentDomainConfig.PreviousFailoverVersion,\n\t\t\tcurrentDomainConfig.IsGlobalDomain,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\terr = d.updateDomainAuditLog(ctx, currentDomainConfig, currentDomainConfig, persistence.DomainAuditOperationTypeUpdate, \"async workflow queue config update\")\n\tif err != nil {\n\t\treturn err\n\t}\n\n\td.logger.Info(\"async workflow queue config update succeeded\",\n\t\ttag.WorkflowDomainName(currentDomainConfig.Info.Name),\n\t\ttag.WorkflowDomainID(currentDomainConfig.Info.ID),\n\t)\n\treturn nil\n}\n\nfunc (d *handlerImpl) createResponse(\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\treplicationConfig *persistence.DomainReplicationConfig,\n) (*types.DomainInfo, *types.DomainConfiguration, *types.DomainReplicationConfiguration) {\n\n\tinfoResult := &types.DomainInfo{\n\t\tName:        info.Name,\n\t\tStatus:      getDomainStatus(info),\n\t\tDescription: info.Description,\n\t\tOwnerEmail:  info.OwnerEmail,\n\t\tData:        info.Data,\n\t\tUUID:        info.ID,\n\t}\n\n\tconfigResult := &types.DomainConfiguration{\n\t\tEmitMetric:                             config.EmitMetric,\n\t\tWorkflowExecutionRetentionPeriodInDays: config.Retention,\n\t\tHistoryArchivalStatus:                  config.HistoryArchivalStatus.Ptr(),\n\t\tHistoryArchivalURI:                     config.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:               config.VisibilityArchivalStatus.Ptr(),\n\t\tVisibilityArchivalURI:                  config.VisibilityArchivalURI,\n\t\tBadBinaries:                            &config.BadBinaries,\n\t\tIsolationGroups:                        &config.IsolationGroups,\n\t\tAsyncWorkflowConfig:                    &config.AsyncWorkflowConfig,\n\t}\n\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor _, cluster := range replicationConfig.Clusters {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: cluster.ClusterName,\n\t\t})\n\t}\n\n\treplicationConfigResult := &types.DomainReplicationConfiguration{\n\t\tActiveClusterName: replicationConfig.ActiveClusterName,\n\t\tClusters:          clusters,\n\t\tActiveClusters:    replicationConfig.ActiveClusters,\n\t}\n\n\treturn infoResult, configResult, replicationConfigResult\n}\n\nfunc (d *handlerImpl) mergeBadBinaries(\n\told map[string]*types.BadBinaryInfo,\n\tnew map[string]*types.BadBinaryInfo,\n\tcreateTimeNano int64,\n) types.BadBinaries {\n\n\tif old == nil {\n\t\told = map[string]*types.BadBinaryInfo{}\n\t}\n\tfor k, v := range new {\n\t\tv.CreatedTimeNano = common.Int64Ptr(createTimeNano)\n\t\told[k] = v\n\t}\n\treturn types.BadBinaries{\n\t\tBinaries: old,\n\t}\n}\n\nfunc (d *handlerImpl) mergeDomainData(\n\told map[string]string,\n\tnew map[string]string,\n) map[string]string {\n\n\tif old == nil {\n\t\told = map[string]string{}\n\t}\n\tfor k, v := range new {\n\t\told[k] = v\n\t}\n\treturn old\n}\n\nfunc (d *handlerImpl) toArchivalRegisterEvent(\n\tstatus *types.ArchivalStatus,\n\tURI string,\n\tdefaultStatus types.ArchivalStatus,\n\tdefaultURI string,\n) (*ArchivalEvent, error) {\n\n\tevent := &ArchivalEvent{\n\t\tstatus:     status,\n\t\tURI:        URI,\n\t\tdefaultURI: defaultURI,\n\t}\n\tif event.status == nil {\n\t\tevent.status = defaultStatus.Ptr()\n\t}\n\tif err := event.validate(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (d *handlerImpl) toArchivalUpdateEvent(\n\tstatus *types.ArchivalStatus,\n\tURI string,\n\tdefaultURI string,\n) (*ArchivalEvent, error) {\n\n\tevent := &ArchivalEvent{\n\t\tstatus:     status,\n\t\tURI:        URI,\n\t\tdefaultURI: defaultURI,\n\t}\n\tif err := event.validate(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (d *handlerImpl) validateHistoryArchivalURI(URIString string) error {\n\tURI, err := archiver.NewURI(URIString)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tarchiver, err := d.archiverProvider.GetHistoryArchiver(URI.Scheme(), service.Frontend)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn archiver.ValidateURI(URI)\n}\n\nfunc (d *handlerImpl) validateVisibilityArchivalURI(URIString string) error {\n\tURI, err := archiver.NewURI(URIString)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tarchiver, err := d.archiverProvider.GetVisibilityArchiver(URI.Scheme(), service.Frontend)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn archiver.ValidateURI(URI)\n}\n\nfunc (d *handlerImpl) getHistoryArchivalState(\n\tconfig *persistence.DomainConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (*ArchivalState, bool, error) {\n\n\tcurrentHistoryArchivalState := &ArchivalState{\n\t\tStatus: config.HistoryArchivalStatus,\n\t\tURI:    config.HistoryArchivalURI,\n\t}\n\tclusterHistoryArchivalConfig := d.archivalMetadata.GetHistoryConfig()\n\n\tif clusterHistoryArchivalConfig.ClusterConfiguredForArchival() {\n\t\tarchivalEvent, err := d.toArchivalUpdateEvent(\n\t\t\tupdateRequest.HistoryArchivalStatus,\n\t\t\tupdateRequest.GetHistoryArchivalURI(),\n\t\t\tclusterHistoryArchivalConfig.GetDomainDefaultURI(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn currentHistoryArchivalState, false, err\n\t\t}\n\t\treturn currentHistoryArchivalState.getNextState(archivalEvent, d.validateHistoryArchivalURI)\n\t}\n\treturn currentHistoryArchivalState, false, nil\n}\n\nfunc (d *handlerImpl) updateHistoryArchivalState(\n\tconfig *persistence.DomainConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (bool, error) {\n\thistoryArchivalState, changed, err := d.getHistoryArchivalState(config, updateRequest)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tif changed {\n\t\tconfig.HistoryArchivalStatus = historyArchivalState.Status\n\t\tconfig.HistoryArchivalURI = historyArchivalState.URI\n\t}\n\n\treturn changed, nil\n}\n\nfunc (d *handlerImpl) getVisibilityArchivalState(\n\tconfig *persistence.DomainConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (*ArchivalState, bool, error) {\n\tcurrentVisibilityArchivalState := &ArchivalState{\n\t\tStatus: config.VisibilityArchivalStatus,\n\t\tURI:    config.VisibilityArchivalURI,\n\t}\n\tclusterVisibilityArchivalConfig := d.archivalMetadata.GetVisibilityConfig()\n\tif clusterVisibilityArchivalConfig.ClusterConfiguredForArchival() {\n\t\tarchivalEvent, err := d.toArchivalUpdateEvent(\n\t\t\tupdateRequest.VisibilityArchivalStatus,\n\t\t\tupdateRequest.GetVisibilityArchivalURI(),\n\t\t\tclusterVisibilityArchivalConfig.GetDomainDefaultURI(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn currentVisibilityArchivalState, false, err\n\t\t}\n\t\treturn currentVisibilityArchivalState.getNextState(archivalEvent, d.validateVisibilityArchivalURI)\n\t}\n\treturn currentVisibilityArchivalState, false, nil\n}\n\nfunc (d *handlerImpl) updateVisibilityArchivalState(\n\tconfig *persistence.DomainConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (bool, error) {\n\tvisibilityArchivalState, changed, err := d.getVisibilityArchivalState(\n\t\tconfig,\n\t\tupdateRequest,\n\t)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tif changed {\n\t\tconfig.VisibilityArchivalStatus = visibilityArchivalState.Status\n\t\tconfig.VisibilityArchivalURI = visibilityArchivalState.URI\n\t}\n\n\treturn changed, nil\n}\n\nfunc (d *handlerImpl) updateDomainInfo(\n\tupdateRequest *types.UpdateDomainRequest,\n\tcurrentDomainInfo *persistence.DomainInfo,\n) (*persistence.DomainInfo, bool) {\n\n\tisDomainUpdated := false\n\tif updateRequest.Description != nil {\n\t\tisDomainUpdated = true\n\t\tcurrentDomainInfo.Description = *updateRequest.Description\n\t}\n\tif updateRequest.OwnerEmail != nil {\n\t\tisDomainUpdated = true\n\t\tcurrentDomainInfo.OwnerEmail = *updateRequest.OwnerEmail\n\t}\n\tif updateRequest.Data != nil {\n\t\tisDomainUpdated = true\n\t\t// only do merging\n\t\tcurrentDomainInfo.Data = d.mergeDomainData(currentDomainInfo.Data, updateRequest.Data)\n\t}\n\treturn currentDomainInfo, isDomainUpdated\n}\n\nfunc (d *handlerImpl) updateDomainConfiguration(\n\tdomainName string,\n\tconfig *persistence.DomainConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (*persistence.DomainConfig, bool, error) {\n\n\tisConfigChanged := false\n\tif updateRequest.EmitMetric != nil {\n\t\tisConfigChanged = true\n\t\tconfig.EmitMetric = *updateRequest.EmitMetric\n\t}\n\tif updateRequest.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\tisConfigChanged = true\n\t\tconfig.Retention = *updateRequest.WorkflowExecutionRetentionPeriodInDays\n\t}\n\tif updateRequest.BadBinaries != nil {\n\t\tmaxLength := d.config.MaxBadBinaryCount(domainName)\n\t\t// only do merging\n\t\tconfig.BadBinaries = d.mergeBadBinaries(config.BadBinaries.Binaries, updateRequest.BadBinaries.Binaries, d.timeSource.Now().UnixNano())\n\t\tif len(config.BadBinaries.Binaries) > maxLength {\n\t\t\treturn config, isConfigChanged, &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"Total resetBinaries cannot exceed the max limit: %v\", maxLength),\n\t\t\t}\n\t\t}\n\t}\n\treturn config, isConfigChanged, nil\n}\n\nfunc (d *handlerImpl) updateDeleteBadBinary(\n\tconfig *persistence.DomainConfig,\n\tdeleteBadBinary *string,\n) (*persistence.DomainConfig, bool, error) {\n\n\tif deleteBadBinary != nil {\n\t\t_, ok := config.BadBinaries.Binaries[*deleteBadBinary]\n\t\tif !ok {\n\t\t\treturn config, false, &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"Bad binary checksum %v doesn't exists.\", *deleteBadBinary),\n\t\t\t}\n\t\t}\n\t\tdelete(config.BadBinaries.Binaries, *deleteBadBinary)\n\t\treturn config, true, nil\n\t}\n\treturn config, false, nil\n}\n\n// updateReplicationConfig is the function which takes the input request and current state and edits and returns it\n// to the desired state by merging the request values with the current state.\n// replicationConfigChanged being turned on will trigger an increment in the configVersion.\n// activeClusterChanged indicates a failover is happening and a failover version is to be incremented\nfunc (d *handlerImpl) updateReplicationConfig(\n\tdomainName string,\n\tconfig *persistence.DomainReplicationConfig,\n\tupdateRequest *types.UpdateDomainRequest,\n) (\n\tmutatedCfg *persistence.DomainReplicationConfig,\n\treplicationConfigChanged bool,\n\tactiveClusterChanged bool,\n\terr error,\n) {\n\n\tif len(updateRequest.Clusters) != 0 {\n\t\treplicationConfigChanged = true\n\t\tclustersNew := []*persistence.ClusterReplicationConfig{}\n\t\tfor _, clusterConfig := range updateRequest.Clusters {\n\t\t\tclustersNew = append(clustersNew, &persistence.ClusterReplicationConfig{\n\t\t\t\tClusterName: clusterConfig.GetClusterName(),\n\t\t\t})\n\t\t}\n\n\t\tif err := d.domainAttrValidator.validateDomainReplicationConfigClustersDoesNotRemove(\n\t\t\tconfig.Clusters,\n\t\t\tclustersNew,\n\t\t); err != nil {\n\t\t\td.logger.Warn(\"removing replica clusters from domain replication group\", tag.Error(err))\n\t\t}\n\t\tconfig.Clusters = clustersNew\n\t}\n\n\tif updateRequest.ActiveClusterName != nil {\n\t\tactiveClusterChanged = true\n\t\tconfig.ActiveClusterName = *updateRequest.ActiveClusterName\n\t}\n\n\terr = d.domainAttrValidator.validateActiveActiveDomainReplicationConfig(updateRequest.ActiveClusters)\n\tif err != nil {\n\t\treturn nil, false, false, err\n\t}\n\n\tif updateRequest != nil && updateRequest.ActiveClusters != nil && updateRequest.ActiveClusters.AttributeScopes != nil {\n\t\tresult, isCh := d.buildActiveActiveClusterScopesFromUpdateRequest(updateRequest, config, domainName)\n\t\tif isCh {\n\n\t\t\tif config.ActiveClusters == nil {\n\t\t\t\tconfig.ActiveClusters = &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: result.AttributeScopes,\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconfig.ActiveClusters.AttributeScopes = result.AttributeScopes\n\t\t\tactiveClusterChanged = true\n\t\t\treplicationConfigChanged = true\n\t\t}\n\t}\n\n\treturn config, replicationConfigChanged, activeClusterChanged, nil\n}\n\nfunc (d *handlerImpl) handleGracefulFailover(\n\tupdateRequest *types.UpdateDomainRequest,\n\treplicationConfig *persistence.DomainReplicationConfig,\n\tcurrentActiveCluster string,\n\tgracefulFailoverEndTime *int64,\n\tfailoverVersion int64,\n\tactiveClusterChanged bool,\n\tisGlobalDomain bool,\n) (*int64, int64, error) {\n\t// must update active cluster on a global domain\n\tif !activeClusterChanged || !isGlobalDomain || replicationConfig.IsActiveActive() {\n\t\treturn nil, 0, errInvalidGracefulFailover\n\t}\n\t// must start with the passive -> active cluster\n\tif replicationConfig.ActiveClusterName != d.clusterMetadata.GetCurrentClusterName() {\n\t\treturn nil, 0, errCannotDoGracefulFailoverFromCluster\n\t}\n\tif replicationConfig.ActiveClusterName == currentActiveCluster {\n\t\treturn nil, 0, errGracefulFailoverInActiveCluster\n\t}\n\t// cannot have concurrent failover\n\tif gracefulFailoverEndTime != nil {\n\t\treturn nil, 0, errOngoingGracefulFailover\n\t}\n\tendTime := d.timeSource.Now().Add(time.Duration(updateRequest.GetFailoverTimeoutInSeconds()) * time.Second).UnixNano()\n\tpreviousFailoverVersion := failoverVersion\n\n\treturn &endTime, previousFailoverVersion, nil\n}\n\nfunc (d *handlerImpl) validateGlobalDomainReplicationConfigForUpdateDomain(\n\treplicationConfig *persistence.DomainReplicationConfig,\n\tconfigurationChanged bool,\n\tactiveClusterChanged bool,\n) error {\n\tvar err error\n\tif err = d.domainAttrValidator.validateDomainReplicationConfigForGlobalDomain(\n\t\treplicationConfig,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif configurationChanged && activeClusterChanged && !replicationConfig.IsActiveActive() {\n\t\treturn errCannotDoDomainFailoverAndUpdate\n\t}\n\n\tif !activeClusterChanged && !d.clusterMetadata.IsPrimaryCluster() {\n\t\treturn errNotPrimaryCluster\n\t}\n\treturn nil\n}\n\n// validateDomainFailoverRequest is the handler method for the FailoverDomain\n// handler. It can receive request to failover domains in a variety of combinations\n// including invalid ones.\nfunc (d *handlerImpl) validateDomainFailoverRequest(\n\trequest *types.FailoverDomainRequest,\n\tcurrentDomainState *persistence.GetDomainResponse,\n) error {\n\tif request == nil {\n\t\treturn &types.BadRequestError{Message: \"Request cannot be nil\"}\n\t}\n\n\tif request.DomainName == \"\" {\n\t\treturn &types.BadRequestError{Message: \"DomainName cannot be empty\"}\n\t}\n\n\tif !currentDomainState.IsGlobalDomain {\n\t\treturn errLocalDomainsCannotFailover\n\t}\n\n\tif request.ActiveClusters == nil && request.DomainActiveClusterName == nil {\n\t\treturn &types.BadRequestError{Message: \"Domain's ActiveClusterName or ActiveClusters must be set to failover the domain\"}\n\t}\n\treturn nil\n}\n\n// validateDomainReplicationConfigForFailover is to check if the replication config\n// is valid and sane for failovers. It is only for global domains\nfunc (d *handlerImpl) validateDomainReplicationConfigForFailover(\n\treplicationConfig *persistence.DomainReplicationConfig,\n\tconfigurationChanged bool,\n\tactiveClusterChanged bool,\n) error {\n\t// todo (add any additional failover validation here)\n\treturn d.validateGlobalDomainReplicationConfigForUpdateDomain(replicationConfig, configurationChanged, activeClusterChanged)\n}\n\nfunc (d *handlerImpl) activeClustersFromRegisterRequest(registerRequest *types.RegisterDomainRequest) (*types.ActiveClusters, error) {\n\tif !registerRequest.GetIsGlobalDomain() || registerRequest.ActiveClusters == nil {\n\t\t// local or active-passive domain\n\t\treturn nil, nil\n\t}\n\n\tclusters := d.clusterMetadata.GetAllClusterInfo()\n\n\tactiveClustersScopes := make(map[string]types.ClusterAttributeScope)\n\n\t// Handle AttributeScopes from the request\n\tif registerRequest.ActiveClusters != nil && registerRequest.ActiveClusters.AttributeScopes != nil {\n\t\tfor scope, scopeData := range registerRequest.ActiveClusters.AttributeScopes {\n\t\t\tnewScopeData := types.ClusterAttributeScope{\n\t\t\t\tClusterAttributes: make(map[string]types.ActiveClusterInfo),\n\t\t\t}\n\t\t\tfor attribute, clusterInfo := range scopeData.ClusterAttributes {\n\t\t\t\tclusterMetadata, ok := clusters[clusterInfo.ActiveClusterName]\n\t\t\t\tif !ok {\n\t\t\t\t\treturn nil, &types.BadRequestError{\n\t\t\t\t\t\tMessage: fmt.Sprintf(\"Cluster %v not found. Domain cannot be registered in this cluster for scope %q and attribute %q\", clusterInfo.ActiveClusterName, scope, attribute),\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tnewScopeData.ClusterAttributes[attribute] = types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: clusterInfo.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   clusterMetadata.InitialFailoverVersion,\n\t\t\t\t}\n\t\t\t}\n\t\t\tactiveClustersScopes[scope] = newScopeData\n\t\t}\n\t}\n\n\tif len(activeClustersScopes) == 0 {\n\t\treturn nil, nil\n\t}\n\n\treturn &types.ActiveClusters{\n\t\tAttributeScopes: activeClustersScopes,\n\t}, nil\n}\n\nfunc getDomainStatus(info *persistence.DomainInfo) *types.DomainStatus {\n\tswitch info.Status {\n\tcase persistence.DomainStatusRegistered:\n\t\tv := types.DomainStatusRegistered\n\t\treturn &v\n\tcase persistence.DomainStatusDeprecated:\n\t\tv := types.DomainStatusDeprecated\n\t\treturn &v\n\tcase persistence.DomainStatusDeleted:\n\t\tv := types.DomainStatusDeleted\n\t\treturn &v\n\t}\n\n\treturn nil\n}\n\n// Maps fields onto an updateDomain Request\n// it's really important that this explicitly calls out each field to ensure no fields get missed or dropped\nfunc createUpdateRequest(\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\treplicationConfig *persistence.DomainReplicationConfig,\n\tconfigVersion int64,\n\tfailoverVersion int64,\n\tfailoverNotificationVersion int64,\n\tfailoverEndTime *int64,\n\tpreviousFailoverVersion int64,\n\tlastUpdatedTime time.Time,\n\tnotificationVersion int64,\n) persistence.UpdateDomainRequest {\n\treturn persistence.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      config,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               configVersion,\n\t\tFailoverVersion:             failoverVersion,\n\t\tFailoverNotificationVersion: failoverNotificationVersion,\n\t\tFailoverEndTime:             failoverEndTime,\n\t\tPreviousFailoverVersion:     previousFailoverVersion,\n\t\tLastUpdatedTime:             lastUpdatedTime.UnixNano(),\n\t\tNotificationVersion:         notificationVersion,\n\t}\n}\n\nfunc updateFailoverHistoryInDomainData(\n\tinfo *persistence.DomainInfo,\n\tconfig Config,\n\tfailoverEvent FailoverEvent,\n) error {\n\tdata := info.Data\n\tif info.Data == nil {\n\t\tdata = make(map[string]string)\n\t}\n\n\tvar failoverHistory []FailoverEvent\n\t_ = json.Unmarshal([]byte(data[constants.DomainDataKeyForFailoverHistory]), &failoverHistory)\n\n\tfailoverHistory = append([]FailoverEvent{failoverEvent}, failoverHistory...)\n\n\t// Truncate the history to the max size\n\tfailoverHistoryJSON, err := json.Marshal(failoverHistory[:min(config.FailoverHistoryMaxSize(info.Name), len(failoverHistory))])\n\tif err != nil {\n\t\treturn err\n\t}\n\tdata[constants.DomainDataKeyForFailoverHistory] = string(failoverHistoryJSON)\n\n\tinfo.Data = data\n\n\treturn nil\n}\n\nfunc NewFailoverEvent(\n\teventTime time.Time,\n\tfailoverType constants.FailoverType,\n\tfromCluster *string,\n\ttoCluster *string,\n\treason *string,\n) FailoverEvent {\n\tres := FailoverEvent{\n\t\tEventTime:    eventTime,\n\t\tFailoverType: failoverType.String(),\n\t}\n\tif fromCluster != nil {\n\t\tres.FromCluster = *fromCluster\n\t}\n\tif toCluster != nil {\n\t\tres.ToCluster = *toCluster\n\t}\n\tif reason != nil {\n\t\tres.Reason = *reason\n\t}\n\treturn res\n}\n\nfunc (d *handlerImpl) buildActiveActiveClusterScopesFromUpdateRequest(updateRequest *types.UpdateDomainRequest, config *persistence.DomainReplicationConfig, domainName string) (out *types.ActiveClusters, isChanged bool) {\n\tvar existing *types.ActiveClusters\n\tif config != nil && config.ActiveClusters != nil {\n\t\texisting = config.ActiveClusters\n\t}\n\n\tif updateRequest.ActiveClusters == nil || updateRequest.ActiveClusters.AttributeScopes == nil {\n\t\treturn existing, false\n\t}\n\n\t// ensure a failover version is set for the incoming request\n\tfor scope, scopeData := range updateRequest.ActiveClusters.AttributeScopes {\n\t\tfor attribute, activeCluster := range scopeData.ClusterAttributes {\n\n\t\t\tcurrentFailoverVersion := types.UndefinedFailoverVersion\n\t\t\tif config != nil && config.ActiveClusters != nil {\n\t\t\t\tfo, err := config.ActiveClusters.GetFailoverVersionForAttribute(scope, attribute)\n\t\t\t\tif err == nil {\n\t\t\t\t\tcurrentFailoverVersion = fo\n\t\t\t\t}\n\t\t\t}\n\t\t\tnextFailoverVersion := d.clusterMetadata.GetNextFailoverVersion(activeCluster.ActiveClusterName, currentFailoverVersion, domainName)\n\n\t\t\tactiveCluster.FailoverVersion = nextFailoverVersion\n\t\t\tscopeData.ClusterAttributes[attribute] = activeCluster\n\t\t}\n\t}\n\n\t// if there's no existing active cluster info, use what's in the request\n\tif existing == nil {\n\t\treturn updateRequest.ActiveClusters, true\n\t}\n\n\t// if, on the other hand, there are existing active clusters, merge them with the incoming request\n\tresult, isChanged := mergeActiveActiveScopes(config.ActiveClusters, updateRequest.ActiveClusters)\n\tif isChanged {\n\t\treturn result, isChanged\n\t}\n\n\treturn config.ActiveClusters, false\n}\n"
  },
  {
    "path": "common/domain/handler_MasterCluster_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdomainHandlerGlobalDomainEnabledPrimaryClusterSuite struct {\n\t\t*persistencetests.TestBase\n\n\t\tminRetentionDays       int\n\t\tmaxBadBinaryCount      int\n\t\tfailoverHistoryMaxSize int\n\t\tdomainManager          persistence.DomainManager\n\t\tmockProducer           *mocks.KafkaProducer\n\t\tmockDomainReplicator   Replicator\n\t\tarchivalMetadata       archiver.ArchivalMetadata\n\t\tmockArchiverProvider   *provider.MockArchiverProvider\n\t\tmockDomainAuditManager persistence.DomainAuditManager\n\n\t\thandler *handlerImpl\n\t}\n)\n\nfunc TestDomainHandlerGlobalDomainEnabledPrimaryClusterSuite(t *testing.T) {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\n\ts := new(domainHandlerGlobalDomainEnabledPrimaryClusterSuite)\n\ts.setupTestBase(t)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) setupTestBase(t *testing.T) {\n\tsqliteTestBaseOptions := sqlite.GetTestClusterOption()\n\tsqliteTestBaseOptions.ClusterMetadata = cluster.GetTestClusterMetadata(true)\n\ts.TestBase = persistencetests.NewTestBaseWithSQL(t, sqliteTestBaseOptions)\n\ts.Setup()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TearDownSuite() {\n\ts.TestBase.TearDownWorkflowStore()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) SetupTest() {\n\ts.setupTestBase(s.T())\n\n\tlogger := s.Logger\n\tdcCollection := dc.NewCollection(dc.NewNopClient(), logger)\n\ts.minRetentionDays = 1\n\ts.maxBadBinaryCount = 10\n\ts.failoverHistoryMaxSize = 5\n\ts.domainManager = s.TestBase.DomainManager\n\ts.mockProducer = &mocks.KafkaProducer{}\n\ts.mockDomainReplicator = NewDomainReplicator(s.mockProducer, logger)\n\ts.mockDomainAuditManager = persistence.NewMockDomainAuditManager(s.Controller)\n\ts.archivalMetadata = archiver.NewArchivalMetadata(\n\t\tdcCollection,\n\t\t\"\",\n\t\tfalse,\n\t\t\"\",\n\t\tfalse,\n\t\t&config.ArchivalDomainDefaults{},\n\t)\n\ts.mockArchiverProvider = &provider.MockArchiverProvider{}\n\tdomainConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(s.minRetentionDays),\n\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(s.maxBadBinaryCount),\n\t\tFailoverCoolDown:         dynamicproperties.GetDurationPropertyFnFilteredByDomain(0 * time.Second),\n\t\tFailoverHistoryMaxSize:   dynamicproperties.GetIntPropertyFilteredByDomain(s.failoverHistoryMaxSize),\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\ts.handler = NewHandler(\n\t\tdomainConfig,\n\t\tlogger,\n\t\ts.domainManager,\n\t\ts.mockDomainAuditManager,\n\t\ts.ClusterMetadata,\n\t\ts.mockDomainReplicator,\n\t\ts.archivalMetadata,\n\t\ts.mockArchiverProvider,\n\t\tclock.NewMockedTimeSource(),\n\t).(*handlerImpl)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TearDownTest() {\n\ts.mockProducer.AssertExpectations(s.T())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestRegisterGetDomain_LocalDomain_InvalidCluster() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := cluster.TestAlternativeClusterName\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: activeClusterName,\n\t\t},\n\t}\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := false\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestRegisterGetDomain_LocalDomain_AllDefault() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := false\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\tretention := int32(1)\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: \"\",\n\t\tOwnerEmail:  \"\",\n\t\tData:        nil,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             true,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          clusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(constants.EmptyVersion, resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestRegisterGetDomain_LocalDomain_NoDefault() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := cluster.TestCurrentClusterName\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: activeClusterName,\n\t\t},\n\t}\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := false\n\n\tvar expectedClusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\texpectedClusters = append(expectedClusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: description,\n\t\tOwnerEmail:  email,\n\t\tData:        data,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             emitMetric,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          expectedClusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(constants.EmptyVersion, resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateGetDomain_LocalDomain_NoAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\tisGlobalDomain := false\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      s.ClusterMetadata.GetCurrentClusterName(),\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(constants.EmptyVersion, failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateGetDomain_LocalDomain_AllAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := false\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t})\n\ts.Nil(err)\n\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(constants.EmptyVersion, failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            common.StringPtr(description),\n\t\tOwnerEmail:                             common.StringPtr(email),\n\t\tData:                                   data,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(retention),\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestDeprecateGetDomain_LocalDomain() {\n\tdomainName := s.getRandomDomainName()\n\tdomain := s.setupLocalDomain(domainName)\n\n\terr := s.handler.DeprecateDomain(context.Background(), &types.DeprecateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\n\texpectedResp := domain\n\texpectedResp.DomainInfo.Status = types.DomainStatusDeprecated.Ptr()\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tassertDomainEqual(s.Suite, getResp, expectedResp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestRegisterGetDomain_GlobalDomain_AllDefault() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := true\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\tretention := int32(1)\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: \"\",\n\t\tOwnerEmail:  \"\",\n\t\tData:        nil,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             true,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          clusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(s.ClusterMetadata.GetCurrentClusterName(), 0, \"some-domaain\"), resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestRegisterGetDomain_GlobalDomain_NoDefault() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: description,\n\t\tOwnerEmail:  email,\n\t\tData:        data,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             emitMetric,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: activeClusterName,\n\t\tClusters:          clusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"), resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_NoAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tisGlobalDomain := true\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: activeClusterName,\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"), failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_AllAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tisGlobalDomain := true\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t})\n\ts.Nil(err)\n\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: activeClusterName,\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"), failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            common.StringPtr(description),\n\t\tOwnerEmail:                             common.StringPtr(email),\n\t\tData:                                   data,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(retention),\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tActiveClusterName:                      nil,\n\t\tClusters:                               clusters,\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_Failover() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tprevActiveClusterName := \"\"\n\tnextActiveClusterName := s.ClusterMetadata.GetCurrentClusterName()\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != nextActiveClusterName {\n\t\t\tprevActiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(prevActiveClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tisGlobalDomain := true\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      prevActiveClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tvar failoverHistory []FailoverEvent\n\tfailoverHistory = append(failoverHistory, FailoverEvent{\n\t\tEventTime:    s.handler.timeSource.Now(),\n\t\tFromCluster:  prevActiveClusterName,\n\t\tToCluster:    nextActiveClusterName,\n\t\tFailoverType: constants.FailoverType(constants.FailoverTypeForce).String(),\n\t})\n\tfailoverHistoryJSON, _ := json.Marshal(failoverHistory)\n\tdata[constants.DomainDataKeyForFailoverHistory] = string(failoverHistoryJSON)\n\n\tfnTest := func(\n\t\tinfo *types.DomainInfo,\n\t\tconfig *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration,\n\t\tisGlobalDomain bool,\n\t\tfailoverVersion int64) {\n\n\t\ts.T().Helper()\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: nextActiveClusterName,\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(\n\t\t\tnextActiveClusterName,\n\t\t\ts.ClusterMetadata.GetNextFailoverVersion(prevActiveClusterName, 0, \"some-domain\"),\n\t\t\t\"some-domain\"), failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:              domainName,\n\t\tActiveClusterName: common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestUpdateDomain_CoolDown() {\n\tdomainConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(s.minRetentionDays),\n\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(s.maxBadBinaryCount),\n\t\tFailoverCoolDown:         dynamicproperties.GetDurationPropertyFnFilteredByDomain(10000 * time.Second),\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\ts.handler = NewHandler(\n\t\tdomainConfig,\n\t\ts.Logger,\n\t\ts.domainManager,\n\t\ts.mockDomainAuditManager,\n\t\ts.ClusterMetadata,\n\t\ts.mockDomainReplicator,\n\t\ts.archivalMetadata,\n\t\ts.mockArchiverProvider,\n\t\tclock.NewRealTimeSource(),\n\t).(*handlerImpl)\n\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := true\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\tretention := int32(1)\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: \"\",\n\t\tOwnerEmail:  \"\",\n\t\tData:        nil,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             true,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          clusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(s.ClusterMetadata.GetCurrentClusterName(), 0, \"some-domain\"), resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n\n\t_, err = s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:        domainName,\n\t\tDescription: common.StringPtr(\"test1\"),\n\t})\n\ts.Error(err)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) TestDeprecateGetDomain_GlobalDomain() {\n\tdomainName := s.getRandomDomainName()\n\tdomain := s.setupGlobalDomain(domainName)\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := s.handler.DeprecateDomain(context.Background(), &types.DeprecateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\n\texpectedResp := domain\n\texpectedResp.DomainInfo.Status = types.DomainStatusDeprecated.Ptr()\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tassertDomainEqual(s.Suite, getResp, expectedResp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) getRandomDomainName() string {\n\treturn \"domain\" + uuid.New()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) setupLocalDomain(domainName string) *types.DescribeDomainResponse {\n\treturn setupLocalDomain(s.Suite, s.handler, s.ClusterMetadata, domainName)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledPrimaryClusterSuite) setupGlobalDomain(domainName string) *types.DescribeDomainResponse {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\treturn setupGlobalDomain(s.Suite, s.handler, s.ClusterMetadata, domainName)\n}\n\nfunc setupGlobalDomain(s *suite.Suite, handler *handlerImpl, clusterMetadata cluster.Metadata, domainName string) *types.DescribeDomainResponse {\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range clusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != clusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tisGlobalDomain := true\n\n\terr := handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tgetResp, err := handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\treturn getResp\n}\n\nfunc setupLocalDomain(s *suite.Suite, handler *handlerImpl, clusterMetadata cluster.Metadata, domainName string) *types.DescribeDomainResponse {\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(clusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\terr := handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      clusterMetadata.GetCurrentClusterName(),\n\t\tData:                                   data,\n\t})\n\ts.Nil(err)\n\tgetResp, err := handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\treturn getResp\n}\n\nfunc assertDomainEqual(s *suite.Suite, actual, expected *types.DescribeDomainResponse) {\n\ts.NotEmpty(actual.DomainInfo.GetUUID())\n\texpected.DomainInfo.UUID = actual.DomainInfo.GetUUID()\n\ts.Equal(expected, actual)\n}\n"
  },
  {
    "path": "common/domain/handler_NotMasterCluster_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdomainHandlerGlobalDomainEnabledNotPrimaryClusterSuite struct {\n\t\t*persistencetests.TestBase\n\n\t\tminRetentionDays       int\n\t\tmaxBadBinaryCount      int\n\t\tfailoverHistoryMaxSize int\n\t\tdomainManager          persistence.DomainManager\n\t\tmockProducer           *mocks.KafkaProducer\n\t\tmockDomainReplicator   Replicator\n\t\tarchivalMetadata       archiver.ArchivalMetadata\n\t\tmockArchiverProvider   *provider.MockArchiverProvider\n\t\tmockDomainAuditManager persistence.DomainAuditManager\n\n\t\thandler *handlerImpl\n\t}\n)\n\nfunc TestDomainHandlerGlobalDomainEnabledNotPrimaryClusterSuite(t *testing.T) {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\n\ts := new(domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite)\n\ts.setupTestBase(t)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) setupTestBase(t *testing.T) {\n\tsqliteTestBaseOptions := sqlite.GetTestClusterOption()\n\tsqliteTestBaseOptions.ClusterMetadata = cluster.GetTestClusterMetadata(false)\n\ts.TestBase = persistencetests.NewTestBaseWithSQL(t, sqliteTestBaseOptions)\n\ts.Setup()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TearDownSuite() {\n\ts.TestBase.TearDownWorkflowStore()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) SetupTest() {\n\ts.setupTestBase(s.T())\n\n\tlogger := s.Logger\n\tdcCollection := dc.NewCollection(dc.NewNopClient(), logger)\n\ts.minRetentionDays = 1\n\ts.maxBadBinaryCount = 10\n\ts.failoverHistoryMaxSize = 5\n\ts.domainManager = s.TestBase.DomainManager\n\ts.mockProducer = &mocks.KafkaProducer{}\n\ts.mockDomainReplicator = NewDomainReplicator(s.mockProducer, logger)\n\ts.mockDomainAuditManager = persistence.NewMockDomainAuditManager(s.Controller)\n\ts.archivalMetadata = archiver.NewArchivalMetadata(\n\t\tdcCollection,\n\t\t\"\",\n\t\tfalse,\n\t\t\"\",\n\t\tfalse,\n\t\t&config.ArchivalDomainDefaults{},\n\t)\n\ts.mockArchiverProvider = &provider.MockArchiverProvider{}\n\tdomainConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(s.minRetentionDays),\n\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(s.maxBadBinaryCount),\n\t\tFailoverCoolDown:         dynamicproperties.GetDurationPropertyFnFilteredByDomain(0 * time.Second),\n\t\tFailoverHistoryMaxSize:   dynamicproperties.GetIntPropertyFilteredByDomain(s.failoverHistoryMaxSize),\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\ts.handler = NewHandler(\n\t\tdomainConfig,\n\t\tlogger,\n\t\ts.domainManager,\n\t\ts.mockDomainAuditManager,\n\t\ts.ClusterMetadata,\n\t\ts.mockDomainReplicator,\n\t\ts.archivalMetadata,\n\t\ts.mockArchiverProvider,\n\t\tclock.NewMockedTimeSource(),\n\t).(*handlerImpl)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TearDownTest() {\n\ts.mockProducer.AssertExpectations(s.T())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestRegisterGetDomain_LocalDomain_AllDefault() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := false\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\tretention := int32(1)\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: \"\",\n\t\tOwnerEmail:  \"\",\n\t\tData:        nil,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             true,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          clusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(constants.EmptyVersion, resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestRegisterGetDomain_LocalDomain_NoDefault() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := cluster.TestCurrentClusterName\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: activeClusterName,\n\t\t},\n\t}\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := false\n\n\tvar expectedClusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\texpectedClusters = append(expectedClusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\n\ts.NotEmpty(resp.DomainInfo.GetUUID())\n\tresp.DomainInfo.UUID = \"\"\n\ts.Equal(&types.DomainInfo{\n\t\tName:        domainName,\n\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\tDescription: description,\n\t\tOwnerEmail:  email,\n\t\tData:        data,\n\t\tUUID:        \"\",\n\t}, resp.DomainInfo)\n\ts.Equal(&types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             emitMetric,\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     \"\",\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  \"\",\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t}, resp.Configuration)\n\ts.Equal(&types.DomainReplicationConfiguration{\n\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\tClusters:          expectedClusters,\n\t}, resp.ReplicationConfiguration)\n\ts.Equal(constants.EmptyVersion, resp.GetFailoverVersion())\n\ts.Equal(isGlobalDomain, resp.GetIsGlobalDomain())\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestUpdateGetDomain_LocalDomain_NoAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\tisGlobalDomain := false\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      s.ClusterMetadata.GetCurrentClusterName(),\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.Nil(err)\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(constants.EmptyVersion, failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestUpdateGetDomain_LocalDomain_AllAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := false\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t})\n\ts.Nil(err)\n\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: s.ClusterMetadata.GetCurrentClusterName(),\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(constants.EmptyVersion, failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            common.StringPtr(description),\n\t\tOwnerEmail:                             common.StringPtr(email),\n\t\tData:                                   data,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(retention),\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestDeprecateGetDomain_LocalDomain() {\n\tdomainName := s.getRandomDomainName()\n\tdomain := s.setupLocalDomain(domainName)\n\n\terr := s.handler.DeprecateDomain(context.Background(), &types.DeprecateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.Nil(err)\n\n\texpectedResp := domain\n\texpectedResp.DomainInfo.Status = types.DomainStatusDeprecated.Ptr()\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tassertDomainEqual(s.Suite, getResp, expectedResp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestRegisterGetDomain_GlobalDomain_AllDefault() {\n\tdomainName := s.getRandomDomainName()\n\tisGlobalDomain := true\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\ts.Equal(1, len(clusters))\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:           domainName,\n\t\tIsGlobalDomain: isGlobalDomain,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.IsType(&types.EntityNotExistsError{}, err)\n\ts.Nil(resp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestRegisterGetDomain_GlobalDomain_NoDefault() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             email,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tData:                                   data,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n\n\tresp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.IsType(&types.EntityNotExistsError{}, err)\n\ts.Nil(resp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_NoAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := \"\"\n\tclusters := []*persistence.ClusterReplicationConfig{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &persistence.ClusterReplicationConfig{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\t_, err := s.DomainManager.CreateDomain(context.Background(), &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        domainName,\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: activeClusterName,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tConfigVersion:   0,\n\t\tFailoverVersion: s.ClusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"),\n\t})\n\ts.Nil(err)\n\n\tresp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n\ts.Nil(resp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_AllAttrSet() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := \"\"\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tclustersDB := []*persistence.ClusterReplicationConfig{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t\tclustersDB = append(clustersDB, &persistence.ClusterReplicationConfig{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\ts.True(len(clustersDB) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\t_, err := s.DomainManager.CreateDomain(context.Background(), &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        domainName,\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: \"\",\n\t\t\tOwnerEmail:  \"\",\n\t\t\tData:        map[string]string{},\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                0,\n\t\t\tEmitMetric:               false,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: activeClusterName,\n\t\t\tClusters:          clustersDB,\n\t\t},\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tConfigVersion:   0,\n\t\tFailoverVersion: s.ClusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"),\n\t})\n\ts.Nil(err)\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            common.StringPtr(description),\n\t\tOwnerEmail:                             common.StringPtr(email),\n\t\tData:                                   data,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(retention),\n\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tActiveClusterName:                      nil,\n\t\tClusters:                               clusters,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n\ts.Nil(updateResp)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestUpdateGetDomain_GlobalDomain_Failover() {\n\tdomainName := s.getRandomDomainName()\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tprevActiveClusterName := \"\"\n\tnextActiveClusterName := s.ClusterMetadata.GetCurrentClusterName()\n\tclusters := []*types.ClusterReplicationConfiguration{}\n\tclustersDB := []*persistence.ClusterReplicationConfig{}\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tprevActiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t\tclustersDB = append(clustersDB, &persistence.ClusterReplicationConfig{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(prevActiveClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\ts.True(len(clustersDB) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\t_, err := s.DomainManager.CreateDomain(context.Background(), &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        domainName,\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: prevActiveClusterName,\n\t\t\tClusters:          clustersDB,\n\t\t},\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tConfigVersion:   0,\n\t\tFailoverVersion: s.ClusterMetadata.GetNextFailoverVersion(prevActiveClusterName, 0, \"some-domain\"),\n\t})\n\ts.Nil(err)\n\n\tvar failoverHistory []FailoverEvent\n\tfailoverHistory = append(failoverHistory, FailoverEvent{EventTime: s.handler.timeSource.Now(), FromCluster: prevActiveClusterName, ToCluster: nextActiveClusterName, FailoverType: constants.FailoverType(constants.FailoverTypeForce).String()})\n\tfailoverHistoryJSON, _ := json.Marshal(failoverHistory)\n\tdata[constants.DomainDataKeyForFailoverHistory] = string(failoverHistoryJSON)\n\n\tfnTest := func(info *types.DomainInfo, config *types.DomainConfiguration,\n\t\treplicationConfig *types.DomainReplicationConfiguration, isGlobalDomain bool, failoverVersion int64) {\n\t\ts.NotEmpty(info.GetUUID())\n\t\tinfo.UUID = \"\"\n\t\ts.Equal(&types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t\tUUID:        \"\",\n\t\t}, info)\n\t\ts.Equal(&types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t}, config)\n\t\ts.Equal(&types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: nextActiveClusterName,\n\t\t\tClusters:          clusters,\n\t\t}, replicationConfig)\n\t\ts.Equal(s.ClusterMetadata.GetNextFailoverVersion(\n\t\t\tnextActiveClusterName,\n\t\t\ts.ClusterMetadata.GetNextFailoverVersion(prevActiveClusterName, 0, \"some-domain\"),\n\t\t\t\"some-domain\"), failoverVersion)\n\t\ts.Equal(isGlobalDomain, isGlobalDomain)\n\t}\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\tupdateResp, err := s.handler.UpdateDomain(context.Background(), &types.UpdateDomainRequest{\n\t\tName:              domainName,\n\t\tActiveClusterName: common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tupdateResp.DomainInfo,\n\t\tupdateResp.Configuration,\n\t\tupdateResp.ReplicationConfiguration,\n\t\tupdateResp.GetIsGlobalDomain(),\n\t\tupdateResp.GetFailoverVersion(),\n\t)\n\n\tgetResp, err := s.handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\tfnTest(\n\t\tgetResp.DomainInfo,\n\t\tgetResp.Configuration,\n\t\tgetResp.ReplicationConfiguration,\n\t\tgetResp.GetIsGlobalDomain(),\n\t\tgetResp.GetFailoverVersion(),\n\t)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) TestDeprecateGetDomain_GlobalDomain() {\n\tdomainName := s.getRandomDomainName()\n\ts.setupGlobalDomainWithMetadataManager(domainName)\n\n\terr := s.handler.DeprecateDomain(context.Background(), &types.DeprecateDomainRequest{\n\t\tName: domainName,\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) getRandomDomainName() string {\n\treturn \"domain\" + uuid.New()\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) setupLocalDomain(domainName string) *types.DescribeDomainResponse {\n\treturn setupLocalDomain(s.Suite, s.handler, s.ClusterMetadata, domainName)\n}\n\nfunc (s *domainHandlerGlobalDomainEnabledNotPrimaryClusterSuite) setupGlobalDomainWithMetadataManager(domainName string) *types.DescribeDomainResponse {\n\treturn setupGlobalDomainWithMetadataManager(s.Suite, s.handler, s.ClusterMetadata, s.DomainManager, domainName)\n}\n\nfunc setupGlobalDomainWithMetadataManager(s *suite.Suite, handler *handlerImpl, clusterMetadata cluster.Metadata, domainManager persistence.DomainManager, domainName string) *types.DescribeDomainResponse {\n\tdescription := \"some random description\"\n\temail := \"some random email\"\n\tretention := int32(7)\n\temitMetric := true\n\tactiveClusterName := \"\"\n\tclusters := []*persistence.ClusterReplicationConfig{}\n\tfor clusterName := range clusterMetadata.GetAllClusterInfo() {\n\t\tif clusterName != clusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName = clusterName\n\t\t}\n\t\tclusters = append(clusters, &persistence.ClusterReplicationConfig{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.True(len(activeClusterName) > 0)\n\ts.True(len(clusters) > 1)\n\tdata := map[string]string{\"some random key\": \"some random value\"}\n\tisGlobalDomain := true\n\n\t_, err := domainManager.CreateDomain(context.Background(), &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        domainName,\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  email,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: activeClusterName,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tConfigVersion:   0,\n\t\tFailoverVersion: clusterMetadata.GetNextFailoverVersion(activeClusterName, 0, \"some-domain\"),\n\t})\n\ts.Nil(err)\n\n\tgetResp, err := handler.DescribeDomain(context.Background(), &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\ts.Nil(err)\n\treturn getResp\n}\n"
  },
  {
    "path": "common/domain/handler_integration_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdomainHandlerCommonSuite struct {\n\t\t*persistencetests.TestBase\n\n\t\tminRetentionDays       int\n\t\tmaxBadBinaryCount      int\n\t\tfailoverHistoryMaxSize int\n\n\t\tdomainManager          persistence.DomainManager\n\t\tmockProducer           *mocks.KafkaProducer\n\t\tmockDomainReplicator   Replicator\n\t\tarchivalMetadata       archiver.ArchivalMetadata\n\t\tmockArchiverProvider   *provider.MockArchiverProvider\n\t\tmockDomainAuditManager persistence.DomainAuditManager\n\n\t\thandler *handlerImpl\n\t}\n)\n\nvar nowInt64 = time.Now().UnixNano()\n\nfunc TestDomainHandlerCommonSuite(t *testing.T) {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\n\ts := new(domainHandlerCommonSuite)\n\ts.setupTestBase(t)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainHandlerCommonSuite) setupTestBase(t *testing.T) {\n\tsqliteTestBaseOptions := sqlite.GetTestClusterOption()\n\tsqliteTestBaseOptions.ClusterMetadata = cluster.GetTestClusterMetadata(true)\n\ts.TestBase = persistencetests.NewTestBaseWithSQL(t, sqliteTestBaseOptions)\n\ts.Setup()\n}\n\nfunc (s *domainHandlerCommonSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\nfunc (s *domainHandlerCommonSuite) SetupTest() {\n\ts.setupTestBase(s.T())\n\n\tlogger := s.Logger\n\tdcCollection := dc.NewCollection(dc.NewNopClient(), logger)\n\ts.minRetentionDays = 1\n\ts.maxBadBinaryCount = 10\n\ts.failoverHistoryMaxSize = 5\n\ts.domainManager = s.TestBase.DomainManager\n\ts.mockProducer = &mocks.KafkaProducer{}\n\ts.mockDomainReplicator = NewDomainReplicator(s.mockProducer, logger)\n\ts.mockDomainAuditManager = persistence.NewMockDomainAuditManager(s.Controller)\n\ts.archivalMetadata = archiver.NewArchivalMetadata(\n\t\tdcCollection,\n\t\t\"\",\n\t\tfalse,\n\t\t\"\",\n\t\tfalse,\n\t\t&config.ArchivalDomainDefaults{},\n\t)\n\ts.mockArchiverProvider = provider.NewMockArchiverProvider(s.Controller)\n\tdomainConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(s.minRetentionDays),\n\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(s.maxBadBinaryCount),\n\t\tFailoverCoolDown:         dynamicproperties.GetDurationPropertyFnFilteredByDomain(0 * time.Second),\n\t\tFailoverHistoryMaxSize:   dynamicproperties.GetIntPropertyFilteredByDomain(s.failoverHistoryMaxSize),\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\ts.handler = NewHandler(\n\t\tdomainConfig,\n\t\tlogger,\n\t\ts.domainManager,\n\t\ts.mockDomainAuditManager,\n\t\ts.ClusterMetadata,\n\t\ts.mockDomainReplicator,\n\t\ts.archivalMetadata,\n\t\ts.mockArchiverProvider,\n\t\tclock.NewRealTimeSource(),\n\t).(*handlerImpl)\n}\n\nfunc (s *domainHandlerCommonSuite) TearDownTest() {\n\ts.mockProducer.AssertExpectations(s.T())\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeDomainData_Overriding() {\n\tout := s.handler.mergeDomainData(\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v0\",\n\t\t},\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v2\",\n\t\t},\n\t)\n\n\tassert.Equal(s.T(), map[string]string{\n\t\t\"k0\": \"v2\",\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeDomainData_Adding() {\n\tout := s.handler.mergeDomainData(\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v0\",\n\t\t},\n\t\tmap[string]string{\n\t\t\t\"k1\": \"v2\",\n\t\t},\n\t)\n\n\tassert.Equal(s.T(), map[string]string{\n\t\t\"k0\": \"v0\",\n\t\t\"k1\": \"v2\",\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeDomainData_Merging() {\n\tout := s.handler.mergeDomainData(\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v0\",\n\t\t},\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v1\",\n\t\t\t\"k1\": \"v2\",\n\t\t},\n\t)\n\n\tassert.Equal(s.T(), map[string]string{\n\t\t\"k0\": \"v1\",\n\t\t\"k1\": \"v2\",\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeDomainData_Nil() {\n\tout := s.handler.mergeDomainData(\n\t\tnil,\n\t\tmap[string]string{\n\t\t\t\"k0\": \"v1\",\n\t\t\t\"k1\": \"v2\",\n\t\t},\n\t)\n\n\tassert.Equal(s.T(), map[string]string{\n\t\t\"k0\": \"v1\",\n\t\t\"k1\": \"v2\",\n\t}, out)\n}\n\n// test merging bad binaries\nfunc (s *domainHandlerCommonSuite) TestMergeBadBinaries_Overriding() {\n\tout := s.handler.mergeBadBinaries(\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason0\"},\n\t\t},\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason2\"},\n\t\t}, nowInt64,\n\t)\n\n\tassert.Equal(s.T(), types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason2\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t},\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeBadBinaries_Adding() {\n\tout := s.handler.mergeBadBinaries(\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason0\"},\n\t\t},\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k1\": {Reason: \"reason2\"},\n\t\t}, nowInt64,\n\t)\n\n\texpected := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason0\"},\n\t\t\t\"k1\": {Reason: \"reason2\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t},\n\t}\n\tassert.Equal(s.T(), expected, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeBadBinaries_Merging() {\n\tout := s.handler.mergeBadBinaries(\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason0\"},\n\t\t},\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason1\"},\n\t\t\t\"k1\": {Reason: \"reason2\"},\n\t\t}, nowInt64,\n\t)\n\n\tassert.Equal(s.T(), types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason1\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t\t\"k1\": {Reason: \"reason2\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t},\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestMergeBadBinaries_Nil() {\n\tout := s.handler.mergeBadBinaries(\n\t\tnil,\n\t\tmap[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason1\"},\n\t\t\t\"k1\": {Reason: \"reason2\"},\n\t\t}, nowInt64,\n\t)\n\n\tassert.Equal(s.T(), types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"k0\": {Reason: \"reason1\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t\t\"k1\": {Reason: \"reason2\", CreatedTimeNano: common.Int64Ptr(nowInt64)},\n\t\t},\n\t}, out)\n}\n\nfunc (s *domainHandlerCommonSuite) TestListDomain() {\n\tdomainName1 := s.getRandomDomainName()\n\tdescription1 := \"some random description 1\"\n\temail1 := \"some random email 1\"\n\tretention1 := int32(1)\n\temitMetric1 := true\n\tdata1 := map[string]string{\"some random key 1\": \"some random value 1\"}\n\tisGlobalDomain1 := false\n\tactiveClusterName1 := s.ClusterMetadata.GetCurrentClusterName()\n\tvar cluster1 []*types.ClusterReplicationConfiguration\n\tfor _, replicationConfig := range cluster.GetOrUseDefaultClusters(s.ClusterMetadata.GetCurrentClusterName(), nil) {\n\t\tcluster1 = append(cluster1, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: replicationConfig.ClusterName,\n\t\t})\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName1,\n\t\tDescription:                            description1,\n\t\tOwnerEmail:                             email1,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention1,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric1),\n\t\tData:                                   data1,\n\t\tIsGlobalDomain:                         isGlobalDomain1,\n\t})\n\ts.Nil(err)\n\n\tdomainName2 := s.getRandomDomainName()\n\tdescription2 := \"some random description 2\"\n\temail2 := \"some random email 2\"\n\tretention2 := int32(2)\n\temitMetric2 := false\n\tdata2 := map[string]string{\"some random key 2\": \"some random value 2\"}\n\tisGlobalDomain2 := true\n\tactiveClusterName2 := \"\"\n\tvar cluster2 []*types.ClusterReplicationConfiguration\n\tfor clusterName := range s.ClusterMetadata.GetEnabledClusterInfo() {\n\t\tif clusterName != s.ClusterMetadata.GetCurrentClusterName() {\n\t\t\tactiveClusterName2 = clusterName\n\t\t}\n\t\tcluster2 = append(cluster2, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: clusterName,\n\t\t})\n\t}\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\terr = s.handler.RegisterDomain(context.Background(), &types.RegisterDomainRequest{\n\t\tName:                                   domainName2,\n\t\tDescription:                            description2,\n\t\tOwnerEmail:                             email2,\n\t\tWorkflowExecutionRetentionPeriodInDays: retention2,\n\t\tEmitMetric:                             common.BoolPtr(emitMetric2),\n\t\tClusters:                               cluster2,\n\t\tActiveClusterName:                      activeClusterName2,\n\t\tData:                                   data2,\n\t\tIsGlobalDomain:                         isGlobalDomain2,\n\t})\n\ts.Nil(err)\n\n\tdomains := map[string]*types.DescribeDomainResponse{}\n\tpagesize := int32(1)\n\tvar token []byte\n\tfor doPaging := true; doPaging; doPaging = len(token) > 0 {\n\t\tresp, err := s.handler.ListDomains(context.Background(), &types.ListDomainsRequest{\n\t\t\tPageSize:      pagesize,\n\t\t\tNextPageToken: token,\n\t\t})\n\t\ts.Nil(err)\n\t\ttoken = resp.NextPageToken\n\t\ts.True(len(resp.Domains) <= int(pagesize))\n\t\tif len(resp.Domains) > 0 {\n\t\t\ts.NotEmpty(resp.Domains[0].DomainInfo.GetUUID())\n\t\t\tresp.Domains[0].DomainInfo.UUID = \"\"\n\t\t\tdomains[resp.Domains[0].DomainInfo.GetName()] = resp.Domains[0]\n\t\t}\n\t}\n\tdelete(domains, constants.SystemLocalDomainName)\n\n\ts.Equal(map[string]*types.DescribeDomainResponse{\n\t\tdomainName1: {\n\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\tName:        domainName1,\n\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\tDescription: description1,\n\t\t\t\tOwnerEmail:  email1,\n\t\t\t\tData:        data1,\n\t\t\t\tUUID:        \"\",\n\t\t\t},\n\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: retention1,\n\t\t\t\tEmitMetric:                             emitMetric1,\n\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t\t},\n\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusterName: activeClusterName1,\n\t\t\t\tClusters:          cluster1,\n\t\t\t},\n\t\t\tFailoverVersion: constants.EmptyVersion,\n\t\t\tIsGlobalDomain:  isGlobalDomain1,\n\t\t},\n\t\tdomainName2: {\n\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\tName:        domainName2,\n\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\tDescription: description2,\n\t\t\t\tOwnerEmail:  email2,\n\t\t\t\tData:        data2,\n\t\t\t\tUUID:        \"\",\n\t\t\t},\n\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: retention2,\n\t\t\t\tEmitMetric:                             emitMetric2,\n\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t\t},\n\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusterName: activeClusterName2,\n\t\t\t\tClusters:          cluster2,\n\t\t\t},\n\t\t\tFailoverVersion: s.ClusterMetadata.GetNextFailoverVersion(activeClusterName2, 0, \"some-domain\"),\n\t\t\tIsGlobalDomain:  isGlobalDomain2,\n\t\t},\n\t}, domains)\n}\n\nfunc (s *domainHandlerCommonSuite) TestRegisterDomain_InvalidRetentionPeriod() {\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   \"random-domain-name\",\n\t\tDescription:                            \"random domain name\",\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(0),\n\t\tIsGlobalDomain:                         false,\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.Equal(errInvalidRetentionPeriod, err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestRegisterDomain_InvalidDomainName() {\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   \"random-domain name!^\",\n\t\tDescription:                            \"a domain with bad naming\",\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(3),\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.Equal(errInvalidDomainName, err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_InvalidRetentionPeriod() {\n\tdomain := \"random-domain-name\"\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         false,\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                                   domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(int32(-1)),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.Equal(errInvalidRetentionPeriod, err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_GracefulFailover_Success() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"standby\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: s.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\tresp1, _ := s.domainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{\n\t\tName: domain,\n\t})\n\ts.Equal(\"standby\", resp1.ReplicationConfig.ActiveClusterName)\n\ts.Equal(cluster.TestAlternativeClusterInitialFailoverVersion, resp1.FailoverVersion)\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     domain,\n\t\tActiveClusterName:        common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t\tFailoverTimeoutInSeconds: common.Int32Ptr(100),\n\t}\n\tresp, err := s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.NoError(err)\n\tresp2, err := s.domainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{\n\t\tID: resp.GetDomainInfo().GetUUID(),\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp2.FailoverEndTime)\n\ts.Equal(cluster.TestFailoverVersionIncrement, resp2.FailoverVersion)\n\ts.Equal(cluster.TestAlternativeClusterInitialFailoverVersion, resp2.PreviousFailoverVersion)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_GracefulFailover_NotCurrentActiveCluster() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"active\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: \"active\"},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     domain,\n\t\tActiveClusterName:        common.StringPtr(\"standby\"),\n\t\tFailoverTimeoutInSeconds: common.Int32Ptr(100),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.Error(err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_GracefulFailover_OngoingFailover() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"standby\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: s.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     domain,\n\t\tActiveClusterName:        common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t\tFailoverTimeoutInSeconds: common.Int32Ptr(100),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.NoError(err)\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.Error(err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_GracefulFailover_NoUpdateActiveCluster() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"standby\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: s.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     domain,\n\t\tOwnerEmail:               common.StringPtr(\"test\"),\n\t\tFailoverTimeoutInSeconds: common.Int32Ptr(100),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.Error(err)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_GracefulFailover_After_ForceFailover() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Times(3)\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"standby\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: s.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\t// Start graceful failover\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     domain,\n\t\tActiveClusterName:        common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t\tFailoverTimeoutInSeconds: common.Int32Ptr(100),\n\t}\n\tresp, err := s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.NoError(err)\n\n\t// Force failover\n\tupdateRequest = &types.UpdateDomainRequest{\n\t\tName:              domain,\n\t\tActiveClusterName: common.StringPtr(s.ClusterMetadata.GetCurrentClusterName()),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.NoError(err)\n\tresp2, err := s.domainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{\n\t\tID: resp.GetDomainInfo().GetUUID(),\n\t})\n\ts.NoError(err)\n\ts.Nil(resp2.FailoverEndTime)\n}\n\nfunc (s *domainHandlerCommonSuite) TestUpdateDomain_ForceFailover_SameActiveCluster() {\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Twice()\n\tdomain := uuid.New()\n\tregisterRequest := &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(10),\n\t\tIsGlobalDomain:                         true,\n\t\tActiveClusterName:                      \"standby\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{ClusterName: s.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t{ClusterName: \"standby\"},\n\t\t},\n\t}\n\terr := s.handler.RegisterDomain(context.Background(), registerRequest)\n\ts.NoError(err)\n\n\t// Start graceful failover\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:              domain,\n\t\tActiveClusterName: common.StringPtr(\"standby\"),\n\t}\n\t_, err = s.handler.UpdateDomain(context.Background(), updateRequest)\n\ts.NoError(err)\n}\n\nfunc (s *domainHandlerCommonSuite) getRandomDomainName() string {\n\treturn \"domain\" + uuid.New()\n}\n\nfunc TestHandlerImpl_UpdateIsolationGroups(t *testing.T) {\n\n\tt0 := int64(1565914445 * time.Second)\n\tt1 := int64(1685914445 * time.Second)\n\tmockedTimeSourceAt := clock.NewMockedTimeSourceAt(time.Unix(0, t1))\n\n\tinfo := persistence.DomainInfo{\n\t\tID:   \"10CF5859-C5CC-4CCC-888E-631F84D53F57\",\n\t\tName: \"test\",\n\t}\n\n\tdomainConfig := persistence.DomainConfig{\n\t\tRetention:       10,\n\t\tIsolationGroups: nil,\n\t}\n\treplicationConfig := persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"cluster-1\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: \"cluster-1\"},\n\t\t},\n\t}\n\n\tig := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateDrained},\n\t}\n\n\ttests := map[string]struct {\n\t\tin                types.UpdateDomainIsolationGroupsRequest\n\t\tmanagerAffordance func(m *persistence.MockDomainManager)\n\t\texpectedErr       error\n\t}{\n\t\t\"successful update\": {\n\t\t\tin: types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain:          \"test\",\n\t\t\t\tIsolationGroups: ig,\n\t\t\t},\n\t\t\tmanagerAffordance: func(m *persistence.MockDomainManager) {\n\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{\n\t\t\t\t\tNotificationVersion: 3,\n\t\t\t\t}, nil)\n\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"test\"}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:                        &info,\n\t\t\t\t\tConfig:                      &domainConfig,\n\t\t\t\t\tReplicationConfig:           &replicationConfig,\n\t\t\t\t\tIsGlobalDomain:              false,\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             1,\n\t\t\t\t\tFailoverNotificationVersion: 1,\n\t\t\t\t\tPreviousFailoverVersion:     1,\n\t\t\t\t\tFailoverEndTime:             &t0,\n\t\t\t\t\tLastUpdatedTime:             t0,\n\t\t\t\t\tNotificationVersion:         1,\n\t\t\t\t}, nil)\n\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &info,\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:       10,\n\t\t\t\t\t\tIsolationGroups: ig,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig:           &replicationConfig,\n\t\t\t\t\tConfigVersion:               2,\n\t\t\t\t\tFailoverVersion:             1,\n\t\t\t\t\tFailoverNotificationVersion: 1,\n\t\t\t\t\tPreviousFailoverVersion:     1,\n\t\t\t\t\tFailoverEndTime:             &t0,\n\t\t\t\t\tLastUpdatedTime:             t1,\n\t\t\t\t\tNotificationVersion:         3,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tdomainMgrMock := persistence.NewMockDomainManager(ctrl)\n\t\t\ttd.managerAffordance(domainMgrMock)\n\n\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\treplicator := NewDomainReplicator(producer, testlogger.New(t))\n\n\t\t\thandler := handlerImpl{\n\t\t\t\tdomainManager:       domainMgrMock,\n\t\t\t\tclusterMetadata:     cluster.Metadata{},\n\t\t\t\tdomainReplicator:    replicator,\n\t\t\t\tdomainAttrValidator: nil,\n\t\t\t\tarchivalMetadata:    nil,\n\t\t\t\tarchiverProvider:    nil,\n\t\t\t\ttimeSource:          mockedTimeSourceAt,\n\t\t\t\tconfig: Config{\n\t\t\t\t\tMinRetentionDays:  func(opts ...dynamicproperties.FilterOption) int { return 0 },\n\t\t\t\t\tMaxBadBinaryCount: func(string) int { return 3 },\n\t\t\t\t\tFailoverCoolDown:  func(string) time.Duration { return time.Second },\n\t\t\t\t},\n\t\t\t\tlogger: testlogger.New(t),\n\t\t\t}\n\n\t\t\terr := handler.UpdateIsolationGroups(context.TODO(), td.in)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestHandlerImpl_UpdateAsyncWorkflowConfiguraton(t *testing.T) {\n\tt0 := int64(1565914445 * time.Second)\n\tt1 := int64(1685914445 * time.Second)\n\tmockedTimeSourceAt := clock.NewMockedTimeSourceAt(time.Unix(0, t1))\n\n\tinfo := persistence.DomainInfo{\n\t\tID:   \"10CF5859-C5CC-4CCC-888E-631F84D53F57\",\n\t\tName: \"test\",\n\t}\n\n\tdomainConfig := persistence.DomainConfig{\n\t\tRetention: 10,\n\t}\n\treplicationConfig := persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"cluster-1\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: \"cluster-1\"},\n\t\t},\n\t}\n\n\tasyncWFCfg := types.AsyncWorkflowConfiguration{\n\t\tEnabled:             true,\n\t\tPredefinedQueueName: \"queue-name\",\n\t\tQueueType:           \"kafka\",\n\t\tQueueConfig: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         []byte(`{\"key\":\"value\"}`),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tin                types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\tmanagerAffordance func(m *persistence.MockDomainManager)\n\t\texpectedErr       error\n\t}{\n\t\t\"successful update\": {\n\t\t\tin: types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain:        \"test\",\n\t\t\t\tConfiguration: &asyncWFCfg,\n\t\t\t},\n\t\t\tmanagerAffordance: func(m *persistence.MockDomainManager) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{\n\t\t\t\t\tNotificationVersion: 3,\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"test\"}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:                        &info,\n\t\t\t\t\tConfig:                      &domainConfig,\n\t\t\t\t\tReplicationConfig:           &replicationConfig,\n\t\t\t\t\tIsGlobalDomain:              false,\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             1,\n\t\t\t\t\tFailoverNotificationVersion: 1,\n\t\t\t\t\tPreviousFailoverVersion:     1,\n\t\t\t\t\tFailoverEndTime:             &t0,\n\t\t\t\t\tLastUpdatedTime:             t0,\n\t\t\t\t\tNotificationVersion:         1,\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &info,\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:           10,\n\t\t\t\t\t\tAsyncWorkflowConfig: asyncWFCfg,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig:           &replicationConfig,\n\t\t\t\t\tConfigVersion:               2,\n\t\t\t\t\tFailoverVersion:             1,\n\t\t\t\t\tFailoverNotificationVersion: 1,\n\t\t\t\t\tPreviousFailoverVersion:     1,\n\t\t\t\t\tFailoverEndTime:             &t0,\n\t\t\t\t\tLastUpdatedTime:             t1,\n\t\t\t\t\tNotificationVersion:         3,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tdomainMgrMock := persistence.NewMockDomainManager(ctrl)\n\t\t\ttd.managerAffordance(domainMgrMock)\n\n\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\treplicator := NewDomainReplicator(producer, testlogger.New(t))\n\n\t\t\thandler := handlerImpl{\n\t\t\t\tdomainManager:    domainMgrMock,\n\t\t\t\tclusterMetadata:  cluster.Metadata{},\n\t\t\t\tdomainReplicator: replicator,\n\t\t\t\ttimeSource:       mockedTimeSourceAt,\n\t\t\t\tconfig: Config{\n\t\t\t\t\tMinRetentionDays:  func(opts ...dynamicproperties.FilterOption) int { return 0 },\n\t\t\t\t\tMaxBadBinaryCount: func(string) int { return 3 },\n\t\t\t\t\tFailoverCoolDown:  func(string) time.Duration { return time.Second },\n\t\t\t\t},\n\t\t\t\tlogger: testlogger.New(t),\n\t\t\t}\n\n\t\t\terr := handler.UpdateAsyncWorkflowConfiguraton(context.Background(), td.in)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/handler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: handler.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source handler.go -destination handler_mock.go\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockHandler) DeleteDomain(ctx context.Context, deleteRequest *types.DeleteDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, deleteRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockHandlerMockRecorder) DeleteDomain(ctx, deleteRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockHandler)(nil).DeleteDomain), ctx, deleteRequest)\n}\n\n// DeprecateDomain mocks base method.\nfunc (m *MockHandler) DeprecateDomain(ctx context.Context, deprecateRequest *types.DeprecateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeprecateDomain\", ctx, deprecateRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeprecateDomain indicates an expected call of DeprecateDomain.\nfunc (mr *MockHandlerMockRecorder) DeprecateDomain(ctx, deprecateRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeprecateDomain\", reflect.TypeOf((*MockHandler)(nil).DeprecateDomain), ctx, deprecateRequest)\n}\n\n// DescribeDomain mocks base method.\nfunc (m *MockHandler) DescribeDomain(ctx context.Context, describeRequest *types.DescribeDomainRequest) (*types.DescribeDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeDomain\", ctx, describeRequest)\n\tret0, _ := ret[0].(*types.DescribeDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeDomain indicates an expected call of DescribeDomain.\nfunc (mr *MockHandlerMockRecorder) DescribeDomain(ctx, describeRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeDomain\", reflect.TypeOf((*MockHandler)(nil).DescribeDomain), ctx, describeRequest)\n}\n\n// FailoverDomain mocks base method.\nfunc (m *MockHandler) FailoverDomain(ctx context.Context, failoverRequest *types.FailoverDomainRequest) (*types.FailoverDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FailoverDomain\", ctx, failoverRequest)\n\tret0, _ := ret[0].(*types.FailoverDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FailoverDomain indicates an expected call of FailoverDomain.\nfunc (mr *MockHandlerMockRecorder) FailoverDomain(ctx, failoverRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailoverDomain\", reflect.TypeOf((*MockHandler)(nil).FailoverDomain), ctx, failoverRequest)\n}\n\n// ListDomains mocks base method.\nfunc (m *MockHandler) ListDomains(ctx context.Context, listRequest *types.ListDomainsRequest) (*types.ListDomainsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListDomains\", ctx, listRequest)\n\tret0, _ := ret[0].(*types.ListDomainsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDomains indicates an expected call of ListDomains.\nfunc (mr *MockHandlerMockRecorder) ListDomains(ctx, listRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDomains\", reflect.TypeOf((*MockHandler)(nil).ListDomains), ctx, listRequest)\n}\n\n// RegisterDomain mocks base method.\nfunc (m *MockHandler) RegisterDomain(ctx context.Context, registerRequest *types.RegisterDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RegisterDomain\", ctx, registerRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RegisterDomain indicates an expected call of RegisterDomain.\nfunc (mr *MockHandlerMockRecorder) RegisterDomain(ctx, registerRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterDomain\", reflect.TypeOf((*MockHandler)(nil).RegisterDomain), ctx, registerRequest)\n}\n\n// UpdateAsyncWorkflowConfiguraton mocks base method.\nfunc (m *MockHandler) UpdateAsyncWorkflowConfiguraton(ctx context.Context, updateRequest types.UpdateDomainAsyncWorkflowConfiguratonRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAsyncWorkflowConfiguraton\", ctx, updateRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAsyncWorkflowConfiguraton indicates an expected call of UpdateAsyncWorkflowConfiguraton.\nfunc (mr *MockHandlerMockRecorder) UpdateAsyncWorkflowConfiguraton(ctx, updateRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAsyncWorkflowConfiguraton\", reflect.TypeOf((*MockHandler)(nil).UpdateAsyncWorkflowConfiguraton), ctx, updateRequest)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockHandler) UpdateDomain(ctx context.Context, updateRequest *types.UpdateDomainRequest) (*types.UpdateDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, updateRequest)\n\tret0, _ := ret[0].(*types.UpdateDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockHandlerMockRecorder) UpdateDomain(ctx, updateRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockHandler)(nil).UpdateDomain), ctx, updateRequest)\n}\n\n// UpdateIsolationGroups mocks base method.\nfunc (m *MockHandler) UpdateIsolationGroups(ctx context.Context, updateRequest types.UpdateDomainIsolationGroupsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateIsolationGroups\", ctx, updateRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateIsolationGroups indicates an expected call of UpdateIsolationGroups.\nfunc (mr *MockHandlerMockRecorder) UpdateIsolationGroups(ctx, updateRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateIsolationGroups\", reflect.TypeOf((*MockHandler)(nil).UpdateIsolationGroups), ctx, updateRequest)\n}\n"
  },
  {
    "path": "common/domain/handler_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"net/url\"\n\t\"reflect\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\n// newTestHandler creates a new instance of the handler with mocked dependencies for testing.\nfunc newTestHandler(t *testing.T, ctrl *gomock.Controller, domainManager *persistence.MockDomainManager, primaryCluster bool, domainReplicator Replicator) Handler {\n\tmockDC := dynamicconfig.NewCollection(dynamicconfig.NewNopClient(), log.NewNoop())\n\tdomainDefaults := &config.ArchivalDomainDefaults{\n\t\tHistory: config.HistoryArchivalDomainDefaults{\n\t\t\tStatus: \"Disabled\",\n\t\t\tURI:    \"https://history.example.com\",\n\t\t},\n\t\tVisibility: config.VisibilityArchivalDomainDefaults{\n\t\t\tStatus: \"Disabled\",\n\t\t\tURI:    \"https://visibility.example.com\",\n\t\t},\n\t}\n\tarchivalMetadata := archiver.NewArchivalMetadata(mockDC, \"Enabled\", true, \"Enabled\", true, domainDefaults)\n\ttestConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(1),\n\t\tMaxRetentionDays:         dynamicproperties.GetIntPropertyFn(5),\n\t\tRequiredDomainDataKeys:   nil,\n\t\tMaxBadBinaryCount:        func(string) int { return 3 },\n\t\tFailoverCoolDown:         func(string) time.Duration { return time.Second },\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(true),\n\t}\n\n\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(ctrl)\n\tmockDomainAuditManager.EXPECT().CreateDomainAuditLog(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainAuditLogResponse{EventID: \"test-event-id\"}, nil).AnyTimes()\n\tmockDomainAuditManager.EXPECT().Close().AnyTimes()\n\n\treturn NewHandler(\n\t\ttestConfig,\n\t\tlog.NewNoop(),\n\t\tdomainManager,\n\t\tmockDomainAuditManager,\n\t\tcluster.GetTestClusterMetadata(primaryCluster),\n\t\tdomainReplicator,\n\t\tarchivalMetadata,\n\t\tprovider.NewArchiverProvider(nil, nil),\n\t\tclock.NewMockedTimeSource(),\n\t)\n}\n\nfunc TestRegisterDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\trequest          *types.RegisterDomainRequest\n\t\tisPrimaryCluster bool\n\t\tmockSetup        func(*persistence.MockDomainManager, *MockReplicator, *types.RegisterDomainRequest)\n\t\twantErr          bool\n\t\texpectedErr      error\n\t}{\n\t\t{\n\t\t\tname:             \"success\",\n\t\t\tisPrimaryCluster: true,\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{ID: \"test-domain-id\"}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:             \"domain already exists\",\n\t\t\tisPrimaryCluster: true,\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"existing-domain\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(&persistence.GetDomainResponse{}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:             \"global domain registration on non-primary cluster\",\n\t\t\tisPrimaryCluster: false,\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"global-test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *persistence.GetDomainRequest) (*persistence.GetDomainResponse, error) {\n\t\t\t\t\t\tif req.Name == \"global-test-domain\" {\n\t\t\t\t\t\t\treturn nil, &types.EntityNotExistsError{}\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil, errors.New(\"unexpected domain name\")\n\t\t\t\t\t}).AnyTimes()\n\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: errNotPrimaryCluster,\n\t\t},\n\t\t{\n\t\t\tname: \"unexpected error on domain lookup\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"test-domain-with-lookup-error\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tunexpectedErr := &types.InternalServiceError{Message: \"Internal server error.\"}\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, unexpectedErr)\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.InternalServiceError{},\n\t\t},\n\t\t{\n\t\t\tname: \"domain name does not match regex\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"invalid_domain!\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: errInvalidDomainName,\n\t\t},\n\t\t{\n\t\t\tname: \"specify active cluster name\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"test-domain-with-active-cluster\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\tActiveClusterName:                      \"active\",\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, req *persistence.CreateDomainRequest) (*persistence.CreateDomainResponse, error) {\n\t\t\t\t\t\treturn &persistence.CreateDomainResponse{ID: \"test-domain-id\"}, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"specify clusters including an invalid one\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName: \"test-domain-with-clusters\",\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: \"valid-cluster-1\"},\n\t\t\t\t\t{ClusterName: \"invalid-cluster\"},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"invalid history archival configuration\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                  \"test-domain-invalid-archival-config\",\n\t\t\t\tHistoryArchivalStatus: types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\tHistoryArchivalURI:    \"invalid-uri\",\n\t\t\t\tIsGlobalDomain:        true,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &url.Error{},\n\t\t},\n\t\t{\n\t\t\tname: \"error during domain creation\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"domain-creation-error\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 2,\n\t\t\t\tIsGlobalDomain:                         false,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"creation failed\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain with replication task\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"global-domain-with-replication\",\n\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, replicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{ID: \"domain-id\"}, nil)\n\t\t\t\treplicator.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationCreate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), commonconstants.InitialPreviousFailoverVersion, true).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain replication task failure\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"global-domain-replication-failure\",\n\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{ID: \"domain-id\"}, nil)\n\t\t\t\tmockReplicator.EXPECT().HandleTransmissionTask(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.DomainOperationCreate,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tcommonconstants.InitialPreviousFailoverVersion,\n\t\t\t\t\ttrue,\n\t\t\t\t).Return(errors.New(\"replication task failed\"))\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: errors.New(\"replication task failed\"),\n\t\t},\n\t\t{\n\t\t\tname: \"visibility archival URI triggers parsing/validation error and not able to reach next state\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:                     \"domain-with-invalid-visibility-uri\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\tVisibilityArchivalURI:    \"invalid-visibility-uri\",\n\t\t\t\tIsGlobalDomain:           true,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &url.Error{},\n\t\t},\n\t\t{\n\t\t\tname: \"local domain with invalid replication configuration: non-global domain\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:              \"local-invalid-replication\",\n\t\t\t\tIsGlobalDomain:    false,\n\t\t\t\tActiveClusterName: \"current-cluster\",\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: \"non-current-cluster\"},\n\t\t\t\t\t{ClusterName: \"non-current-cluster2\"},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Invalid local domain active cluster\"},\n\t\t},\n\t\t{\n\t\t\tname: \"local domain with invalid replication configuration: global domain\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:              \"global-invalid-replication\",\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tActiveClusterName: \"current-cluster\",\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: \"non-current-cluster2\"},\n\t\t\t\t\t{ClusterName: \"non-current-cluster3\"},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Invalid local domain active cluster\"},\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain with an invalid cluster in request\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"active-active-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: \"invalid-cluster\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain with an active cluster in request that doesn't exist in domain's clusters list\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"active-active-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\texpectedErr: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain successfully registered with explicit ActiveClusterName\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:              \"active-active-domain\",\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{ID: \"test-domain-id\"}, nil)\n\t\t\t\tmockReplicator.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationCreate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), commonconstants.InitialPreviousFailoverVersion, true).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain successfully registered without explicit ActiveClusterName (uses current cluster)\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"active-active-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tmockSetup: func(mockDomainMgr *persistence.MockDomainManager, mockReplicator *MockReplicator, request *types.RegisterDomainRequest) {\n\t\t\t\tmockDomainMgr.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: request.Name}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDomainMgr.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{ID: \"test-domain-id\"}, nil)\n\t\t\t\tmockReplicator.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationCreate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), commonconstants.InitialPreviousFailoverVersion, true).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, tc.isPrimaryCluster, mockReplicator)\n\n\t\t\ttc.mockSetup(mockDomainMgr, mockReplicator, tc.request)\n\n\t\t\terr := handler.RegisterDomain(context.Background(), tc.request)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\n\t\t\t\tif tc.expectedErr != nil {\n\t\t\t\t\tassert.IsType(t, tc.expectedErr, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListDomains(t *testing.T) {\n\n\tvar configIsolationGroup types.IsolationGroupConfiguration\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *types.ListDomainsRequest\n\t\tsetupMocks    func(mockDomainManager *persistence.MockDomainManager)\n\t\texpectedResp  *types.ListDomainsResponse\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - List domains\",\n\t\t\trequest: &types.ListDomainsRequest{\n\t\t\t\tPageSize: 2,\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager) {\n\t\t\t\tresp := &persistence.ListDomainsResponse{\n\t\t\t\t\tDomains: []*persistence.GetDomainResponse{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\t\tID:   \"domainID1\",\n\t\t\t\t\t\t\t\tName: \"domainName1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\t\tID:   \"domainID2\",\n\t\t\t\t\t\t\t\tName: \"domainName2\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}\n\t\t\t\tmockDomainManager.EXPECT().ListDomains(gomock.Any(), &persistence.ListDomainsRequest{\n\t\t\t\t\tPageSize:      2,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}).Return(resp, nil)\n\t\t\t},\n\t\t\texpectedResp: &types.ListDomainsResponse{\n\t\t\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\t\tName:        \"domainName1\",\n\t\t\t\t\t\t\tUUID:        \"domainID1\",\n\t\t\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\t\t\tDescription: \"\",\n\t\t\t\t\t\t\tOwnerEmail:  \"\",\n\t\t\t\t\t\t\tData:        nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tEmitMetric:               false,\n\t\t\t\t\t\t\tBadBinaries:              &types.BadBinaries{Binaries: nil},\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\t\t\t\tIsolationGroups:          &configIsolationGroup,\n\t\t\t\t\t\t\tAsyncWorkflowConfig:      &types.AsyncWorkflowConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\t\tActiveClusterName: \"\",\n\t\t\t\t\t\t\tClusters:          []*types.ClusterReplicationConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\t\tName:        \"domainName2\",\n\t\t\t\t\t\t\tUUID:        \"domainID2\",\n\t\t\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\t\t\tDescription: \"\",\n\t\t\t\t\t\t\tOwnerEmail:  \"\",\n\t\t\t\t\t\t\tData:        nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tEmitMetric:               false,\n\t\t\t\t\t\t\tBadBinaries:              &types.BadBinaries{Binaries: nil},\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\t\t\t\tIsolationGroups:          &configIsolationGroup,\n\t\t\t\t\t\t\tAsyncWorkflowConfig:      &types.AsyncWorkflowConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\t\tActiveClusterName: \"\",\n\t\t\t\t\t\t\tClusters:          []*types.ClusterReplicationConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - Persistence error\",\n\t\t\trequest: &types.ListDomainsRequest{\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager) {\n\t\t\t\tmockDomainManager.EXPECT().ListDomains(gomock.Any(), &persistence.ListDomainsRequest{\n\t\t\t\t\tPageSize:      10,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}).Return(nil, errors.New(\"persistence error\"))\n\t\t\t},\n\t\t\texpectedResp:  nil,\n\t\t\texpectedError: errors.New(\"persistence error\"),\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tcontroller := gomock.NewController(t)\n\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\tmockReplicator := NewMockReplicator(controller)\n\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator)\n\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttest.setupMocks(mockDomainMgr)\n\n\t\t\tresp, err := handler.ListDomains(context.Background(), test.request)\n\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), test.expectedError.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.EqualValues(t, test.expectedResp, resp)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc TestHandler_DescribeDomain(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tmockDomainManager := persistence.NewMockDomainManager(controller)\n\tmockReplicator := NewMockReplicator(controller)\n\n\thandler := newTestHandler(t, controller, mockDomainManager, true, mockReplicator)\n\n\tdomainName := \"test-domain\"\n\tdomainID := \"test-domain-id\"\n\tisGlobalDomain := true\n\tfailoverVersion := int64(123)\n\tlastUpdatedTime := time.Now().UnixNano()\n\tfailoverEndTime := lastUpdatedTime + int64(time.Hour)\n\tvar configIsolationGroup types.IsolationGroupConfiguration\n\n\tdomainInfo := &persistence.DomainInfo{\n\t\tID:          domainID,\n\t\tName:        domainName,\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"Test Domain\",\n\t\tOwnerEmail:  \"test@uber.com\",\n\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention: 24,\n\t}\n\tdomainReplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active\",\n\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t}\n\n\tmockDomainManager.EXPECT().\n\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: domainName}).\n\t\tReturn(&persistence.GetDomainResponse{\n\t\t\tInfo:              domainInfo,\n\t\t\tConfig:            domainConfig,\n\t\t\tReplicationConfig: domainReplicationConfig,\n\t\t\tIsGlobalDomain:    isGlobalDomain,\n\t\t\tFailoverVersion:   failoverVersion,\n\t\t\tFailoverEndTime:   &failoverEndTime,\n\t\t\tLastUpdatedTime:   lastUpdatedTime,\n\t\t}, nil)\n\n\tdescribeRequest := &types.DescribeDomainRequest{Name: &domainName}\n\texpectedResponse := &types.DescribeDomainResponse{\n\t\tIsGlobalDomain:  isGlobalDomain,\n\t\tFailoverVersion: failoverVersion,\n\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\tFailoverVersion:         failoverVersion,\n\t\t\tFailoverStartTimestamp:  lastUpdatedTime,\n\t\t\tFailoverExpireTimestamp: failoverEndTime,\n\t\t},\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:        domainName,\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: \"Test Domain\",\n\t\t\tOwnerEmail:  \"test@uber.com\",\n\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\tUUID:        domainID,\n\t\t},\n\t\tConfiguration: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: 24,\n\t\t\tEmitMetric:                             false,\n\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: nil},\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusDisabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"\",\n\t\t\tIsolationGroups:                        &configIsolationGroup,\n\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t},\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: \"active\",\n\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t{ClusterName: \"active\"},\n\t\t\t\t{ClusterName: \"standby\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tresp, err := handler.DescribeDomain(context.Background(), describeRequest)\n\tassert.NoError(t, err)\n\tassert.EqualValues(t, expectedResponse, resp)\n\n\tmockDomainManager.EXPECT().\n\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\tReturn(nil, types.EntityNotExistsError{}).\n\t\tTimes(1)\n\n\t_, err = handler.DescribeDomain(context.Background(), describeRequest)\n\n\tassert.Error(t, err)\n\tassert.IsType(t, types.EntityNotExistsError{}, err)\n}\n\nfunc TestHandler_DeprecateDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tdomainName     string\n\t\tsetupMocks     func(m *persistence.MockDomainManager, r *MockReplicator)\n\t\tisGlobalDomain bool\n\t\tprimaryCluster bool\n\t\texpectedErr    error\n\t}{\n\t\t{\n\t\t\tname:       \"success - deprecate local domain\",\n\t\t\tdomainName: \"local-domain\",\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\t// Mock calls\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"local-domain\"}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"local-domain\", Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention: 7,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    false,\n\t\t\t\t}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\tisGlobalDomain: false,\n\t\t\tprimaryCluster: true,\n\t\t\texpectedErr:    nil,\n\t\t},\n\t\t{\n\t\t\tname:       \"failure - domain not found\",\n\t\t\tdomainName: \"non-existent-domain\",\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"non-existent-domain\"}).Return(nil, errors.New(\"domain not found\"))\n\t\t\t},\n\t\t\tisGlobalDomain: false,\n\t\t\tprimaryCluster: false,\n\t\t\texpectedErr:    errors.New(\"domain not found\"),\n\t\t},\n\t\t{\n\t\t\tname:       \"failure - get metadata error\",\n\t\t\tdomainName: \"test-domain\",\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(nil, errors.New(\"metadata error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"metadata error\"),\n\t\t},\n\t\t{\n\t\t\tname:           \"failure - not primary cluster for global domain\",\n\t\t\tdomainName:     \"global-domain\",\n\t\t\tisGlobalDomain: true,\n\t\t\tprimaryCluster: false,\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"global-domain\"}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"global-domain\", Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention: 7,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedErr: errNotPrimaryCluster,\n\t\t},\n\t\t{\n\t\t\tname:           \"failure - update domain error\",\n\t\t\tdomainName:     \"test-domain\",\n\t\t\tprimaryCluster: true,\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"test-domain\", Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention: 7,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tFailoverVersion:   123,\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t}, nil)\n\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.AssignableToTypeOf(&persistence.UpdateDomainRequest{})).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, req *persistence.UpdateDomainRequest) error {\n\t\t\t\t\t\tif req.Info.Name != \"test-domain\" || req.ConfigVersion != 2 || req.Info.Status != persistence.DomainStatusDeprecated {\n\t\t\t\t\t\t\treturn errors.New(\"unexpected UpdateDomainRequest\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn errors.New(\"update domain error\")\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"update domain error\"),\n\t\t},\n\t\t{\n\t\t\tname:           \"failure - replicator error for global domain\",\n\t\t\tdomainName:     \"global-domain\",\n\t\t\tisGlobalDomain: true,\n\t\t\tprimaryCluster: true,\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:           &persistence.DomainInfo{Status: persistence.DomainStatusDeprecated},\n\t\t\t\t\tIsGlobalDomain: true,\n\t\t\t\t\tConfigVersion:  1}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tr.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationUpdate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), true).Return(errors.New(\"replicator error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"replicator error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainManager, tc.primaryCluster, mockReplicator)\n\t\t\ttc.setupMocks(mockDomainManager, mockReplicator)\n\n\t\t\terr := handler.DeprecateDomain(context.Background(), &types.DeprecateDomainRequest{Name: tc.domainName})\n\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.EqualError(t, err, tc.expectedErr.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandler_UpdateIsolationGroups(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\tdomain           string\n\t\tisolationGroups  types.IsolationGroupConfiguration\n\t\tisPrimaryCluster bool\n\t\tsetupMocks       func(m *persistence.MockDomainManager, r *MockReplicator)\n\t\texpectedErr      error\n\t}{\n\t\t{\n\t\t\tname:   \"Successful Update for Local Domain\",\n\t\t\tdomain: \"local-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {Name: \"group1\", State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"local-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName: \"local-domain\",\n\t\t\t\t\t\tID:   \"domainID\",\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                7,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{\"group1\": {Name: \"group1\", State: types.IsolationGroupStateHealthy}},\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalURI:       \"https://test.history\",\n\t\t\t\t\t\tVisibilityArchivalURI:    \"https://test.visibility\",\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: false},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: time.Date(2024, 1, 1, 1, 1, 1, 1, time.UTC).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:   \"Successful Update for Global Domain\",\n\t\t\tdomain: \"global-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {State: types.IsolationGroupStateDrained},\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"global-domain\"},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention: 30,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tConfigVersion:     int64(1),\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\tLastUpdatedTime:   time.Date(2024, 1, 1, 1, 1, 1, 1, time.UTC).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tr.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationUpdate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), true).Return(nil)\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure Due to Invalid Request (nil Isolation Groups)\",\n\t\t\tdomain: \"domain-with-nil-isolation-groups\",\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"invalid request, isolationGroup configuration must be set: Got: {domain-with-nil-isolation-groups map[]}\"),\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure Due to Domain Not Found\",\n\t\t\tdomain: \"nonexistent-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"domain not found\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"domain not found\"),\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure Due to UpdateDomain Error\",\n\t\t\tdomain: \"domain-with-update-error\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:              &persistence.DomainInfo{Name: \"domain-with-update-error\"},\n\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    false,\n\t\t\t\t}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"update error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"update error\"),\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure Due to Replication Error (for Global Domains)\",\n\t\t\tdomain: \"global-domain-with-replication-error\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tisPrimaryCluster: true,\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:              &persistence.DomainInfo{Name: \"global-domain-with-replication-error\"},\n\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t}, nil)\n\t\t\t\tm.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tr.EXPECT().HandleTransmissionTask(gomock.Any(), types.DomainOperationUpdate, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), true).Return(errors.New(\"replication error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"replication error\"),\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure due to GetMetadata error\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(nil, errors.New(\"metadata error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"metadata error\"),\n\t\t},\n\t\t{\n\t\t\tname:            \"Failure due to nil domain config\",\n\t\t\tdomain:          \"test-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:              &persistence.DomainInfo{Name: \"test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig:            nil,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    false,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"unable to load config for domain as expected\"),\n\t\t},\n\t\t{\n\t\t\tname:   \"Failure due to failover cool-down\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {Name: \"group1\", State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tlastUpdatedWithinCoolDownPeriod := time.Now().UnixNano() - (500 * int64(time.Millisecond))\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                7,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalURI:       \"https://test.history\",\n\t\t\t\t\t\tVisibilityArchivalURI:    \"https://test.visibility\",\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: false},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: lastUpdatedWithinCoolDownPeriod,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedErr: errDomainUpdateTooFrequent,\n\t\t},\n\t\t{\n\t\t\tname:   \"Global domain update from non-primary cluster\",\n\t\t\tdomain: \"global-domain\",\n\t\t\tisolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"group1\": {Name: \"group1\", State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\tsetupMocks: func(m *persistence.MockDomainManager, r *MockReplicator) {\n\t\t\t\tm.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tm.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"global-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{Name: \"global-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention: 7,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedErr: errNotPrimaryCluster,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainManager, tc.isPrimaryCluster, mockReplicator)\n\t\t\ttc.setupMocks(mockDomainManager, mockReplicator)\n\n\t\t\terr := handler.UpdateIsolationGroups(context.Background(), types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain:          tc.domain,\n\t\t\t\tIsolationGroups: tc.isolationGroups,\n\t\t\t})\n\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedErr, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandler_UpdateAsyncWorkflowConfiguraton(t *testing.T) {\n\ttestDomain := \"testDomain\"\n\n\tcreateDomainConfig := func(asyncEnabled bool) *persistence.DomainConfig {\n\t\treturn &persistence.DomainConfig{\n\t\t\tRetention:                10,\n\t\t\tEmitMetric:               true,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalURI:       \"https://history.example.com\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"https://visibility.example.com\",\n\t\t\tBadBinaries:              types.BadBinaries{},\n\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: asyncEnabled},\n\t\t}\n\t}\n\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator)\n\t\tisPrimaryCluster bool\n\t\trequest          *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\texpectedErr      error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: testDomain}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:            &persistence.DomainInfo{ID: \"domainID\", Name: testDomain},\n\t\t\t\t\tConfig:          createDomainConfig(false),\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 1,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-10 * time.Minute).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tmockDomainManager.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, req *persistence.UpdateDomainRequest) error {\n\t\t\t\t\t\texpectedReq := &persistence.UpdateDomainRequest{\n\t\t\t\t\t\t\tInfo:                        &persistence.DomainInfo{ID: \"domainID\", Name: testDomain},\n\t\t\t\t\t\t\tConfig:                      createDomainConfig(true),\n\t\t\t\t\t\t\tConfigVersion:               2,\n\t\t\t\t\t\t\tFailoverVersion:             1,\n\t\t\t\t\t\t\tFailoverNotificationVersion: 0,\n\t\t\t\t\t\t\tLastUpdatedTime:             req.LastUpdatedTime,\n\t\t\t\t\t\t\tNotificationVersion:         1,\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif !reflect.DeepEqual(req, expectedReq) {\n\t\t\t\t\t\t\treturn fmt.Errorf(\"unexpected UpdateDomainRequest: got %+v, want %+v\", req, expectedReq)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: testDomain,\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"fail to get metadata\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(nil, fmt.Errorf(\"metadata error\"))\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: testDomain,\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"metadata error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"fail to get domain\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"get domain error\"))\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: testDomain,\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"get domain error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"fail due to nil domain config\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: testDomain}).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{ID: \"domainID\", Name: testDomain},\n\t\t\t\t\tConfig: nil,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: testDomain,\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"unable to load config for domain as expected\"),\n\t\t},\n\t\t{\n\t\t\tname: \"fail due to update too frequent\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tlastUpdatedWithinCoolDownPeriod := time.Now().UnixNano() - (500 * int64(time.Millisecond))\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(false),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: lastUpdatedWithinCoolDownPeriod,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: errDomainUpdateTooFrequent,\n\t\t},\n\t\t{\n\t\t\tname:             \"fail due to non-primary cluster update for global domain\",\n\t\t\tisPrimaryCluster: false,\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"global-test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"global-test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(false),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-24 * time.Hour).UnixNano(),\n\t\t\t\t}, nil)\n\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"global-test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: errNotPrimaryCluster,\n\t\t},\n\t\t{\n\t\t\tname: \"configuration delete request\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(true),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-24 * time.Hour).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tmockDomainManager.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, req *persistence.UpdateDomainRequest) error {\n\t\t\t\t\t\tif req.Config.AsyncWorkflowConfig.Enabled != false {\n\t\t\t\t\t\t\treturn fmt.Errorf(\"unexpected UpdateDomainRequest for delete configuration\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain:        \"test-domain\",\n\t\t\t\tConfiguration: nil,\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"fail on domain manager UpdateDomain call\",\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(false),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-24 * time.Hour).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tmockDomainManager.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"update domain error\"))\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"update domain error\"),\n\t\t},\n\t\t{\n\t\t\tname:             \"global domain update triggers replication\",\n\t\t\tisPrimaryCluster: true,\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"global-test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"global-test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(false),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-24 * time.Hour).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tmockDomainManager.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, req *persistence.UpdateDomainRequest) error {\n\t\t\t\t\t\tif req.Config.AsyncWorkflowConfig.Enabled != true {\n\t\t\t\t\t\t\treturn fmt.Errorf(\"unexpected UpdateDomainRequest for global domain update\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t\tmockReplicator.EXPECT().HandleTransmissionTask(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttrue,\n\t\t\t\t).Return(nil)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"global-test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:             \"global domain update replication error\",\n\t\t\tisPrimaryCluster: true,\n\t\t\tsetupMocks: func(mockDomainManager *persistence.MockDomainManager, mockReplicator *MockReplicator) {\n\t\t\t\tmockDomainManager.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Eq(&persistence.GetDomainRequest{Name: \"global-test-domain\"})).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo:   &persistence.DomainInfo{Name: \"global-test-domain\", ID: \"domainID\"},\n\t\t\t\t\tConfig: createDomainConfig(false),\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: \"active\"}, {ClusterName: \"standby\"}},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:   1,\n\t\t\t\t\tFailoverVersion: 123,\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: time.Now().Add(-24 * time.Hour).UnixNano(),\n\t\t\t\t}, nil)\n\t\t\t\tmockDomainManager.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, req *persistence.UpdateDomainRequest) error {\n\t\t\t\t\t\t// Validate the UpdateDomainRequest for global domain update\n\t\t\t\t\t\tif req.Config.AsyncWorkflowConfig.Enabled != true {\n\t\t\t\t\t\t\treturn fmt.Errorf(\"unexpected UpdateDomainRequest for global domain update\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t\tmockReplicator.EXPECT().HandleTransmissionTask(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttrue,\n\t\t\t\t).Return(fmt.Errorf(\"replication error\"))\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"global-test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"replication error\"),\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\tmockReplicator := NewMockReplicator(ctrl)\n\t\t\thandler := newTestHandler(t, ctrl, mockDomainManager, test.isPrimaryCluster, mockReplicator)\n\t\t\ttest.setupMocks(mockDomainManager, mockReplicator)\n\n\t\t\terr := handler.UpdateAsyncWorkflowConfiguraton(context.Background(), *test.request)\n\t\t\tif test.expectedErr != nil {\n\t\t\t\tassert.EqualError(t, err, test.expectedErr.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandler_UpdateDomain(t *testing.T) {\n\tctx := context.Background()\n\tmaxLength := 1\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(\n\t\t\tdomainManager *persistence.MockDomainManager,\n\t\t\tupdateRequest *types.UpdateDomainRequest,\n\t\t\tarchivalMetadata *archiver.MockArchivalMetadata,\n\t\t\ttimeSource clock.MockedTimeSource,\n\t\t\tdomainReplicator *MockReplicator,\n\t\t)\n\t\trequest  *types.UpdateDomainRequest\n\t\tresponse func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse\n\t\terr      error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - global domain force failover\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tfailoverData, _ := json.Marshal([]FailoverEvent{{\n\t\t\t\t\tEventTime:    timeSource.Now(),\n\t\t\t\t\tFromCluster:  cluster.TestCurrentClusterName,\n\t\t\t\t\tToCluster:    cluster.TestAlternativeClusterName,\n\t\t\t\t\tFailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeForce).String(),\n\t\t\t\t}})\n\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\t\"FailoverHistory\": string(failoverData),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion: commonconstants.InitialPreviousFailoverVersion,\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion,\n\t\t\t\t\tFailoverVersion:         cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\texpectedUpdateRequest.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion,\n\t\t\t\t\t\tcluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\tcommonconstants.InitialPreviousFailoverVersion,\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\tdata, _ := json.Marshal([]FailoverEvent{{EventTime: timeSource.Now(), FromCluster: cluster.TestCurrentClusterName, ToCluster: cluster.TestAlternativeClusterName, FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeForce).String()}})\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion/cluster.TestFailoverVersionIncrement*cluster.TestFailoverVersionIncrement + cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tData:   map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(data)},\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - global domain grace failover\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: 1,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tdata, _ := json.Marshal([]FailoverEvent{{EventTime: timeSource.Now(), FromCluster: cluster.TestAlternativeClusterName, ToCluster: cluster.TestCurrentClusterName, FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}})\n\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\tcommonconstants.DomainDataKeyForFailoverHistory: string(data),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion: 1,\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion,\n\t\t\t\t\tFailoverVersion:         10,\n\t\t\t\t\tFailoverEndTime:         common.Ptr(timeSource.Now().Add(time.Duration(10) * time.Second).UnixNano()),\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\texpectedUpdateRequest.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion,\n\t\t\t\t\t\tint64(10),\n\t\t\t\t\t\tint64(1),\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:                     constants.TestDomainName,\n\t\t\t\tActiveClusterName:        common.Ptr(cluster.TestCurrentClusterName),\n\t\t\t\tFailoverTimeoutInSeconds: common.Int32Ptr(10),\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\tdata, _ := json.Marshal([]FailoverEvent{{EventTime: timeSource.Now(), FromCluster: cluster.TestAlternativeClusterName, ToCluster: cluster.TestCurrentClusterName, FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}})\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestFailoverVersionIncrement,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tData:   map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(data)},\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - global domain config change\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData:        nil,\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               false,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig:       domainResponse.ReplicationConfig,\n\t\t\t\t\tPreviousFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion + 1,\n\t\t\t\t\tFailoverVersion:         cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\texpectedUpdateRequest.Config,\n\t\t\t\t\t\tdomainResponse.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion+1,\n\t\t\t\t\t\tcluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\tcluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:       constants.TestDomainName,\n\t\t\t\tEmitMetric: common.Ptr(false),\n\t\t\t},\n\t\t\tresponse: func(_ clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             false,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - active-active domain active clusters change\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tvar data map[string]string\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData:        data,\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\t// previously it was alternative cluster with failover version 1.\n\t\t\t\t\t\t\t\t\t\t\t// failover version should be the next failover version of the new cluster\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion + cluster.TestFailoverVersionIncrement,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion: -1, // this is not applicable to active-active domain\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion + 1,\n\t\t\t\t\tFailoverVersion:         cluster.TestCurrentClusterInitialFailoverVersion + cluster.TestFailoverVersionIncrement, // this is incremented to indicate there was a change in replication config\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\texpectedUpdateRequest.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion+1,\n\t\t\t\t\t\tcluster.TestCurrentClusterInitialFailoverVersion+cluster.TestFailoverVersionIncrement,\n\t\t\t\t\t\tint64(-1), // previous failover version is not applicable to active-active domain\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: { // failover region2 to region1\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   123123123123, // this number will be ignored and replaced with appropriate failover version\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion + cluster.TestFailoverVersionIncrement,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion + cluster.TestFailoverVersionIncrement,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - active-passive to active-active migration\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\t// once migrated to active clusters, this version should be carried over to new active clusters config\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion + 2*cluster.TestFailoverVersionIncrement,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tvar data map[string]string\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData:        data,\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName, // should be left as is\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion: -1, // this is not applicable to active-active domain\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion + 1,\n\t\t\t\t\tFailoverVersion:         cluster.TestCurrentClusterInitialFailoverVersion + 3*cluster.TestFailoverVersionIncrement, // this is incremented to indicate there was a change in replication config\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\texpectedUpdateRequest.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion+1,\n\t\t\t\t\t\tcluster.TestCurrentClusterInitialFailoverVersion+3*cluster.TestFailoverVersionIncrement,\n\t\t\t\t\t\tint64(-1), // previous failover version is not applicable to active-active domain\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion + 3*cluster.TestFailoverVersionIncrement,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - local domain force failover - shoudl not be able to failover a local domain\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestCurrentClusterName),\n\t\t\t},\n\t\t\tresponse: func(_ clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  false,\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion/cluster.TestFailoverVersionIncrement*cluster.TestFailoverVersionIncrement + cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\terr: errLocalDomainsCannotFailover,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetMetadata error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, _ *types.UpdateDomainRequest, _ *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(nil, errors.New(\"get-metadata-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"get-metadata-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetDomain error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, _ *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).Return(nil, errors.New(\"get-domain-error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t},\n\t\t\terr: errors.New(\"get-domain-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - updateHistoryArchivalState error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalEnabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalEnabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalEnabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t},\n\t\t\terr: errInvalidEvent,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - updateVisibilityArchivalState error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfigHistory := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalConfigVisibility := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalEnabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalEnabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalEnabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfigHistory).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfigVisibility).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t},\n\t\t\terr: errInvalidEvent,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - updateDomainConfiguration error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tLastUpdatedTime:   time.Now().UnixNano(),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t\tBadBinaries: &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\tReason: \"test-reason\",\n\t\t\t\t\t},\n\t\t\t\t\t\"bad-binary-2\": {\n\t\t\t\t\t\tReason: \"test-reason-2\",\n\t\t\t\t\t},\n\t\t\t\t}},\n\t\t\t},\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"Total resetBinaries cannot exceed the max limit: %v\", maxLength),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - updateDeleteBadBinary error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:            constants.TestDomainName,\n\t\t\t\tDeleteBadBinary: common.Ptr(\"bad-binary\"),\n\t\t\t},\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"Bad binary checksum %v doesn't exists.\", \"bad-binary\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - handleGracefulFailover error in the case of a global domain - it should return an error to the user\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:                     constants.TestDomainName,\n\t\t\t\tFailoverTimeoutInSeconds: common.Int32Ptr(1),\n\t\t\t},\n\t\t\terr: errInvalidFailoverNoChangeDetected,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - validateDomainConfig error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t},\n\t\t\terr: errInvalidRetentionPeriod,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - validateDomainReplicationConfigForUpdateDomain error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, _ clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: \"bad-cluster\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\t\tRetention: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIsGlobalDomain: true,\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName: constants.TestDomainName,\n\t\t\t},\n\t\t\terr: &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\t\"Invalid cluster name: %v\",\n\t\t\t\t\"bad-cluster\",\n\t\t\t)},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - updateChangesForUpdateDomain error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\t\tRetention: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(-time.Hour)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\terr: errDomainUpdateTooFrequent,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - domainManager.UpdateDomain error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, _ *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\t\tRetention: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, gomock.Any()).Return(errors.New(\"update-domain-error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\terr: errors.New(\"update-domain-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - active-active domain failover with explicit ActiveClusterName uses correct cluster for failover version\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tvar data map[string]string\n\t\t\t\texpectedUpdateRequest := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\t\t\tDescription: domainResponse.Info.Description,\n\t\t\t\t\t\tOwnerEmail:  domainResponse.Info.OwnerEmail,\n\t\t\t\t\t\tData:        data,\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion: -1,\n\t\t\t\t\tConfigVersion:           domainResponse.ConfigVersion,\n\t\t\t\t\tFailoverVersion:         cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\tLastUpdatedTime:         timeSource.Now().UnixNano(),\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, expectedUpdateRequest).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\texpectedUpdateRequest.Info,\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\texpectedUpdateRequest.ReplicationConfig,\n\t\t\t\t\t\tdomainResponse.ConfigVersion,\n\t\t\t\t\t\tcluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\tint64(-1),\n\t\t\t\t\t\tdomainResponse.IsGlobalDomain,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.UpdateDomainResponse {\n\t\t\t\treturn &types.UpdateDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - HandleTransmissionTask error\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.UpdateDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetName()}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\t\tRetention: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tarchivalConfig := archiver.NewArchivalConfig(\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\t\t\t\tfalse,\n\t\t\t\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t\tcommonconstants.ArchivalDisabled,\n\t\t\t\t\t\"\")\n\t\t\t\tarchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\t\t\t\tarchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, gomock.Any()).Return(nil).Times(1)\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"handle-transmission-task-error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tName:              constants.TestDomainName,\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\terr: errors.New(\"handle-transmission-task-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\tmockReplicator := NewMockReplicator(ctrl)\n\t\t\tmockArchivalMetadata := &archiver.MockArchivalMetadata{}\n\t\t\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(ctrl)\n\n\t\t\ttestConfig := Config{\n\t\t\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(1),\n\t\t\t\tMaxRetentionDays:         dynamicproperties.GetIntPropertyFn(5),\n\t\t\t\tRequiredDomainDataKeys:   nil,\n\t\t\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(maxLength),\n\t\t\t\tFailoverCoolDown:         func(string) time.Duration { return time.Second },\n\t\t\t\tFailoverHistoryMaxSize:   dynamicproperties.GetIntPropertyFilteredByDomain(5),\n\t\t\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t}\n\n\t\t\tclusterMetadata := cluster.GetTestClusterMetadata(true)\n\t\t\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Unix(1761769472, 0))\n\n\t\t\thandler := handlerImpl{\n\t\t\t\tdomainManager:       mockDomainManager,\n\t\t\t\tclusterMetadata:     clusterMetadata,\n\t\t\t\tdomainReplicator:    mockReplicator,\n\t\t\t\tdomainAttrValidator: newAttrValidator(clusterMetadata, int32(testConfig.MinRetentionDays())),\n\t\t\t\tarchivalMetadata:    mockArchivalMetadata,\n\t\t\t\tarchiverProvider:    provider.NewArchiverProvider(nil, nil),\n\t\t\t\ttimeSource:          mockTimeSource,\n\t\t\t\tconfig:              testConfig,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t\tdomainAuditManager:  mockDomainAuditManager,\n\t\t\t}\n\n\t\t\t// For all tests in this suite, audit log succeeds\n\t\t\tmockDomainAuditManager.EXPECT().\n\t\t\t\tCreateDomainAuditLog(gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(&persistence.CreateDomainAuditLogResponse{EventID: \"test-event-id\"}, nil).\n\t\t\t\tAnyTimes()\n\n\t\t\ttc.setupMock(mockDomainManager, tc.request, mockArchivalMetadata, mockTimeSource, mockReplicator)\n\n\t\t\tresponse, err := handler.UpdateDomain(ctx, tc.request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, response)\n\t\t\t\tassert.Equal(t, tc.response(mockTimeSource), response)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandler_UpdateDomain_AuditLogFailureDoesNotPropagate(t *testing.T) {\n\t// This test verifies that audit log write failures do not prevent domain updates from succeeding.\n\t// Audit logging is best-effort only and should not block critical domain operations.\n\n\tctx := context.Background()\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\tmockReplicator := NewMockReplicator(ctrl)\n\tmockArchivalMetadata := &archiver.MockArchivalMetadata{}\n\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(ctrl)\n\n\ttestConfig := Config{\n\t\tMinRetentionDays:         dynamicproperties.GetIntPropertyFn(1),\n\t\tMaxRetentionDays:         dynamicproperties.GetIntPropertyFn(5),\n\t\tRequiredDomainDataKeys:   nil,\n\t\tMaxBadBinaryCount:        dynamicproperties.GetIntPropertyFilteredByDomain(1),\n\t\tFailoverCoolDown:         func(string) time.Duration { return time.Second },\n\t\tFailoverHistoryMaxSize:   dynamicproperties.GetIntPropertyFilteredByDomain(5),\n\t\tEnableDomainAuditLogging: dynamicproperties.GetBoolPropertyFn(true),\n\t}\n\n\tclusterMetadata := cluster.GetTestClusterMetadata(true)\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Unix(1761769472, 0))\n\n\thandler := handlerImpl{\n\t\tdomainManager:       mockDomainManager,\n\t\tclusterMetadata:     clusterMetadata,\n\t\tdomainReplicator:    mockReplicator,\n\t\tdomainAttrValidator: newAttrValidator(clusterMetadata, int32(testConfig.MinRetentionDays())),\n\t\tarchivalMetadata:    mockArchivalMetadata,\n\t\tarchiverProvider:    provider.NewArchiverProvider(nil, nil),\n\t\ttimeSource:          mockTimeSource,\n\t\tconfig:              testConfig,\n\t\tlogger:              log.NewNoop(),\n\t\tdomainAuditManager:  mockDomainAuditManager,\n\t}\n\n\tdomainResponse := &persistence.GetDomainResponse{\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tEmitMetric:               true,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t},\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tName:        constants.TestDomainName,\n\t\t\tID:          constants.TestDomainID,\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: \"original-description\",\n\t\t},\n\t\tIsGlobalDomain:  true,\n\t\tLastUpdatedTime: mockTimeSource.Now().UnixNano(),\n\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t}\n\n\t// Set up mocks\n\tmockDomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\tmockDomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: constants.TestDomainName}).\n\t\tReturn(domainResponse, nil).Times(1)\n\n\tarchivalConfig := archiver.NewArchivalConfig(\n\t\tcommonconstants.ArchivalDisabled,\n\t\tdynamicproperties.GetStringPropertyFn(commonconstants.ArchivalDisabled),\n\t\tfalse,\n\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t\tcommonconstants.ArchivalDisabled,\n\t\t\"\")\n\tmockArchivalMetadata.On(\"GetHistoryConfig\").Return(archivalConfig).Times(1)\n\tmockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archivalConfig).Times(1)\n\n\tmockTimeSource.Advance(time.Hour)\n\n\tmockDomainManager.EXPECT().UpdateDomain(ctx, gomock.Any()).Return(nil).Times(1)\n\tmockReplicator.EXPECT().\n\t\tHandleTransmissionTask(\n\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t).Return(nil).Times(1)\n\n\t// Critical assertion: Set up audit log to FAIL\n\tmockDomainAuditManager.EXPECT().\n\t\tCreateDomainAuditLog(gomock.Any(), gomock.Any()).\n\t\tReturn(nil, errors.New(\"audit log database unavailable\")).\n\t\tTimes(1)\n\n\t// Execute the update\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:        constants.TestDomainName,\n\t\tDescription: common.Ptr(\"updated-description\"),\n\t}\n\n\tresponse, err := handler.UpdateDomain(ctx, updateRequest)\n\n\t// Assert: Domain update should succeed despite audit log failure\n\tassert.NoError(t, err, \"UpdateDomain should succeed even when audit log write fails\")\n\tassert.NotNil(t, response, \"Response should not be nil\")\n\n\t// Verify the response contains the updated information\n\tassert.Equal(t, true, response.IsGlobalDomain)\n\tassert.Equal(t, cluster.TestCurrentClusterInitialFailoverVersion, response.FailoverVersion)\n\tassert.Equal(t, constants.TestDomainName, response.DomainInfo.Name)\n\tassert.Equal(t, constants.TestDomainID, response.DomainInfo.UUID)\n\tassert.Equal(t, \"updated-description\", response.DomainInfo.Description)\n\tassert.Equal(t, types.DomainStatusRegistered, *response.DomainInfo.Status)\n}\n\nfunc TestUpdateDomainInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\trequest           *types.UpdateDomainRequest\n\t\tchanged           bool\n\t\tupdatedDomainInfo *persistence.DomainInfo\n\t}{\n\t\t{\n\t\t\tname: \"Success case - new domain info\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tDescription: common.Ptr(\"new-description\"),\n\t\t\t\tOwnerEmail:  common.Ptr(\"new-email\"),\n\t\t\t\tData:        map[string]string{\"new-key\": \"new-value\"},\n\t\t\t},\n\t\t\tchanged: true,\n\t\t\tupdatedDomainInfo: &persistence.DomainInfo{\n\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"new-description\",\n\t\t\t\tOwnerEmail:  \"new-email\",\n\t\t\t\tData:        map[string]string{\"key\": \"value\", \"new-key\": \"new-value\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"Success case - no new domain info in request\",\n\t\t\trequest: &types.UpdateDomainRequest{},\n\t\t\tupdatedDomainInfo: &persistence.DomainInfo{\n\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"some-description\",\n\t\t\t\tOwnerEmail:  \"some-email\",\n\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tdomainInfo := &persistence.DomainInfo{\n\t\t\t\tID:          constants.TestDomainID,\n\t\t\t\tName:        constants.TestDomainName,\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"some-description\",\n\t\t\t\tOwnerEmail:  \"some-email\",\n\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t}\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator)\n\n\t\t\tupdatedDomainInfo, changed := (*handlerImpl).updateDomainInfo(handler.(*handlerImpl), tc.request, domainInfo)\n\n\t\t\tassert.Equal(t, tc.changed, changed)\n\t\t\tassert.Equal(t, tc.updatedDomainInfo, updatedDomainInfo)\n\t\t})\n\t}\n}\n\nfunc TestUpdateDomainConfiguration(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                string\n\t\trequest             *types.UpdateDomainRequest\n\t\tchanged             bool\n\t\tupdatedDomainConfig func(now int64) *persistence.DomainConfig\n\t\terr                 error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - new domain config\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tEmitMetric:                             common.Ptr(false),\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(3),\n\t\t\t\tBadBinaries: &types.BadBinaries{\n\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\t\tReason: \"test-reason\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tchanged: true,\n\t\t\tupdatedDomainConfig: func(now int64) *persistence.DomainConfig {\n\t\t\t\treturn &persistence.DomainConfig{\n\t\t\t\t\tRetention:                3,\n\t\t\t\t\tEmitMetric:               false,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\t\t\tReason:          \"test-reason\",\n\t\t\t\t\t\t\t\tCreatedTimeNano: common.Ptr(now),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tIsolationGroups:     types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - new domain config - only new bad binaries\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tBadBinaries: &types.BadBinaries{\n\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\t\tReason: \"test-reason\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tchanged: false,\n\t\t\tupdatedDomainConfig: func(now int64) *persistence.DomainConfig {\n\t\t\t\treturn &persistence.DomainConfig{\n\t\t\t\t\tRetention:                1,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\t\t\tReason:          \"test-reason\",\n\t\t\t\t\t\t\t\tCreatedTimeNano: common.Ptr(now),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tIsolationGroups:     types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - new domain config has bad binaries greater than max length\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tEmitMetric:                             common.Ptr(false),\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(3),\n\t\t\t\tBadBinaries: &types.BadBinaries{\n\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\t\tReason: \"test-reason\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"bad-binary-2\": {\n\t\t\t\t\t\t\tReason: \"test-reason-2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"bad-binary-3\": {\n\t\t\t\t\t\t\tReason: \"test-reason-3\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"bad-binary-4\": {\n\t\t\t\t\t\t\tReason: \"test-reason-4\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"Total resetBinaries cannot exceed the max limit: %v\", 3),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator)\n\n\t\t\tcfg := &persistence.DomainConfig{\n\t\t\t\tRetention:                1,\n\t\t\t\tEmitMetric:               true,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t}\n\n\t\t\tnow := handler.(*handlerImpl).timeSource.Now().UnixNano()\n\n\t\t\tupdatedDomainConfig, changed, err := (*handlerImpl).updateDomainConfiguration(handler.(*handlerImpl), constants.TestDomainName, cfg, tc.request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.changed, changed)\n\t\t\t\tassert.Equal(t, tc.updatedDomainConfig(now), updatedDomainConfig)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDeleteBadBinary(t *testing.T) {\n\tnow := time.Now().UnixNano()\n\n\ttestCases := []struct {\n\t\tname                string\n\t\tdeleteBadBinary     *string\n\t\tchanged             bool\n\t\tupdatedDomainConfig *persistence.DomainConfig\n\t\terr                 error\n\t}{\n\t\t{\n\t\t\tname:            \"Success case - deleteBadBinary not nil\",\n\t\t\tdeleteBadBinary: common.Ptr(\"bad-binary\"),\n\t\t\tchanged:         true,\n\t\t\tupdatedDomainConfig: &persistence.DomainConfig{\n\t\t\t\tRetention:                1,\n\t\t\t\tEmitMetric:               true,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - deleteBadBinary nil\",\n\t\t\tupdatedDomainConfig: &persistence.DomainConfig{\n\t\t\t\tRetention:                1,\n\t\t\t\tEmitMetric:               true,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\tBadBinaries: types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\tReason:          \"test-reason\",\n\t\t\t\t\t\tCreatedTimeNano: &now,\n\t\t\t\t\t},\n\t\t\t\t}},\n\t\t\t\tIsolationGroups:     types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"Error case - deleteBadBinary not in config.BadBinaries.Binaries\",\n\t\t\tdeleteBadBinary: common.Ptr(\"bad-binary-2\"),\n\t\t\terr:             &types.BadRequestError{Message: \"Bad binary checksum bad-binary-2 doesn't exists.\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator)\n\n\t\t\tcfg := &persistence.DomainConfig{\n\t\t\t\tRetention:                1,\n\t\t\t\tEmitMetric:               true,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\tBadBinaries: types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\"bad-binary\": {\n\t\t\t\t\t\tReason:          \"test-reason\",\n\t\t\t\t\t\tCreatedTimeNano: &now,\n\t\t\t\t\t},\n\t\t\t\t}},\n\t\t\t\tIsolationGroups:     types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t}\n\n\t\t\tupdatedDomainConfig, changed, err := (*handlerImpl).updateDeleteBadBinary(handler.(*handlerImpl), cfg, tc.deleteBadBinary)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.changed, changed)\n\t\t\t\tassert.Equal(t, tc.updatedDomainConfig, updatedDomainConfig)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateReplicationConfig(t *testing.T) {\n\tcfg := func() *persistence.DomainReplicationConfig {\n\t\treturn &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t}\n\t}\n\n\tactiveActiveCfg := func() *persistence.DomainReplicationConfig {\n\t\treturn &persistence.DomainReplicationConfig{\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t}\n\n\ttestCases := []struct {\n\t\tname                     string\n\t\trequest                  *types.UpdateDomainRequest\n\t\tcurrentReplicationConfig *persistence.DomainReplicationConfig\n\t\tupdatedReplicationConfig *persistence.DomainReplicationConfig\n\t\tclusterUpdated           bool\n\t\tactiveClusterUpdated     bool\n\t}{\n\t\t{\n\t\t\tname:                     \"Success case - no change\",\n\t\t\trequest:                  &types.UpdateDomainRequest{},\n\t\t\tcurrentReplicationConfig: cfg(),\n\t\t\tupdatedReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - cluster and activeCluster updated\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t\tClusters:          []*types.ClusterReplicationConfiguration{{ClusterName: cluster.TestDisabledClusterName}, {ClusterName: cluster.TestAlternativeClusterName}, {ClusterName: cluster.TestCurrentClusterName}},\n\t\t\t},\n\t\t\tcurrentReplicationConfig: cfg(),\n\t\t\tupdatedReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestDisabledClusterName}, {ClusterName: cluster.TestAlternativeClusterName}, {ClusterName: cluster.TestCurrentClusterName}},\n\t\t\t},\n\t\t\tclusterUpdated:       true,\n\t\t\tactiveClusterUpdated: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - cluster and activeCluster updated with warning\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t\tClusters:          []*types.ClusterReplicationConfiguration{{ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tcurrentReplicationConfig: cfg(),\n\t\t\tupdatedReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tclusterUpdated:       true,\n\t\t\tactiveClusterUpdated: true,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain - update cluster of region2\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\t// changing this from alternative cluster to current cluster\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   99999, // this will be ignored\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tcurrentReplicationConfig: activeActiveCfg(),\n\t\t\tupdatedReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion + cluster.TestFailoverVersionIncrement,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tclusterUpdated:       true,\n\t\t\tactiveClusterUpdated: true,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain - add a new cluster in a new region\",\n\t\t\trequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region3\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tcurrentReplicationConfig: activeActiveCfg(),\n\t\t\tupdatedReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\tcluster.TestRegion1: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tcluster.TestRegion2: {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"region3\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tclusterUpdated:       true,\n\t\t\tactiveClusterUpdated: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator).(*handlerImpl)\n\n\t\t\tupdatedReplicationConfig, clusterUpdated, activeClusterUpdated, err := handler.updateReplicationConfig(\n\t\t\t\tconstants.TestDomainName,\n\t\t\t\ttc.currentReplicationConfig,\n\t\t\t\ttc.request,\n\t\t\t)\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tc.clusterUpdated, clusterUpdated, \"cluster-updated field was %v when it was expected to be %v\", clusterUpdated, tc.clusterUpdated)\n\t\t\tassert.Equal(t, tc.activeClusterUpdated, activeClusterUpdated, \"active-cluster-updated field was %v when it was expected to be %v\", activeClusterUpdated, tc.activeClusterUpdated)\n\t\t\tassert.Equal(t, tc.updatedReplicationConfig, updatedReplicationConfig, \"replication-config-updated was flagged as %v when it was expected to be %v\", updatedReplicationConfig, tc.updatedReplicationConfig)\n\t\t})\n\t}\n}\n\nfunc TestHandleGracefulFailover(t *testing.T) {\n\tfailoverTimeoutInSeconds := int32(1)\n\tfailoverVersion := int64(3)\n\n\ttestCases := []struct {\n\t\tname                           string\n\t\treplicationConfig              *persistence.DomainReplicationConfig\n\t\tcurrentActiveCluster           string\n\t\tgracefulFailoverEndTime        *int64\n\t\tactiveClusterChange            bool\n\t\tisGlobalDomain                 bool\n\t\tupdatedGracefulFailoverEndTime func(now time.Time) *int64\n\t\terr                            error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tcurrentActiveCluster: cluster.TestAlternativeClusterName,\n\t\t\tactiveClusterChange:  true,\n\t\t\tisGlobalDomain:       true,\n\t\t\tupdatedGracefulFailoverEndTime: func(now time.Time) *int64 {\n\t\t\t\treturn common.Ptr(now.Add(time.Duration(failoverTimeoutInSeconds) * time.Second).UnixNano())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                \"Error case - activeClusterChange is false\",\n\t\t\tactiveClusterChange: false,\n\t\t\tisGlobalDomain:      true,\n\t\t\terr:                 errInvalidGracefulFailover,\n\t\t},\n\t\t{\n\t\t\tname:                \"Error case - isGlobalDomain is false\",\n\t\t\tactiveClusterChange: true,\n\t\t\tisGlobalDomain:      false,\n\t\t\terr:                 errInvalidGracefulFailover,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - replication ActiveClusterName is different from clusterMetadata currentClusterName\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tactiveClusterChange: true,\n\t\t\tisGlobalDomain:      true,\n\t\t\terr:                 errCannotDoGracefulFailoverFromCluster,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - replication ActiveClusterName is the same as target currentActiveCluster\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tcurrentActiveCluster: cluster.TestCurrentClusterName,\n\t\t\tactiveClusterChange:  true,\n\t\t\tisGlobalDomain:       true,\n\t\t\terr:                  errGracefulFailoverInActiveCluster,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - ongoing failover, cannot have concurrent failover\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t},\n\t\t\tcurrentActiveCluster:    cluster.TestAlternativeClusterName,\n\t\t\tactiveClusterChange:     true,\n\t\t\tisGlobalDomain:          true,\n\t\t\tgracefulFailoverEndTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\terr:                     errOngoingGracefulFailover,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDomainMgr := persistence.NewMockDomainManager(controller)\n\t\t\tmockReplicator := NewMockReplicator(controller)\n\n\t\t\thandler := newTestHandler(t, controller, mockDomainMgr, true, mockReplicator)\n\n\t\t\tnow := handler.(*handlerImpl).timeSource.Now()\n\n\t\t\trequest := &types.UpdateDomainRequest{\n\t\t\t\tFailoverTimeoutInSeconds: common.Int32Ptr(failoverTimeoutInSeconds),\n\t\t\t}\n\n\t\t\tgracefulFailoverEndTime, previousFailoverVersion, err := (*handlerImpl).handleGracefulFailover(\n\t\t\t\thandler.(*handlerImpl),\n\t\t\t\trequest,\n\t\t\t\ttc.replicationConfig,\n\t\t\t\ttc.currentActiveCluster,\n\t\t\t\ttc.gracefulFailoverEndTime,\n\t\t\t\tfailoverVersion,\n\t\t\t\ttc.activeClusterChange,\n\t\t\t\ttc.isGlobalDomain,\n\t\t\t)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.updatedGracefulFailoverEndTime(now), gracefulFailoverEndTime)\n\t\t\t\tassert.Equal(t, failoverVersion, previousFailoverVersion)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateFailoverHistory(t *testing.T) {\n\tnow := time.Now()\n\tfromCluster := \"fromCluster\"\n\ttoCluster := \"toCluster\"\n\tfailoverType := commonconstants.FailoverType(commonconstants.FailoverTypeForce)\n\tfailoverHistoryMaxSize := 5\n\n\ttestCases := []struct {\n\t\tname             string\n\t\tdomainInfo       func() *persistence.DomainInfo\n\t\tnewFailoverEvent FailoverEvent\n\t\tresponse         func() string\n\t\terr              error\n\t}{\n\t\t{\n\t\t\tname:       \"Success case - DomainInfo data is nil\",\n\t\t\tdomainInfo: func() *persistence.DomainInfo { return &persistence.DomainInfo{} },\n\t\t\tnewFailoverEvent: FailoverEvent{\n\t\t\t\tEventTime:    now,\n\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\tToCluster:    toCluster,\n\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t},\n\t\t\tresponse: func() string {\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: now, FromCluster: fromCluster, ToCluster: toCluster, FailoverType: failoverType.String()}}\n\t\t\t\tjsonResp, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn string(jsonResp)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"Success case - FailoverHistory is nil\",\n\t\t\tdomainInfo: func() *persistence.DomainInfo { return &persistence.DomainInfo{Data: map[string]string{}} },\n\t\t\tnewFailoverEvent: FailoverEvent{\n\t\t\t\tEventTime:    now,\n\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\tToCluster:    toCluster,\n\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t},\n\t\t\tresponse: func() string {\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: now, FromCluster: fromCluster, ToCluster: toCluster, FailoverType: failoverType.String()}}\n\t\t\t\tjsonResp, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn string(jsonResp)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - FailoverHistory is not nil\",\n\t\t\tdomainInfo: func() *persistence.DomainInfo {\n\t\t\t\teventTime := time.Date(2021, 1, 1, 1, 1, 1, 1, time.UTC)\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: eventTime, FromCluster: \"fromCluster1\", ToCluster: \"toCluster1\", FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}}\n\t\t\t\tfailoverHistoryJSON, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn &persistence.DomainInfo{Data: map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(failoverHistoryJSON)}}\n\t\t\t},\n\t\t\tnewFailoverEvent: FailoverEvent{\n\t\t\t\tEventTime:    now,\n\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\tToCluster:    toCluster,\n\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t},\n\t\t\tresponse: func() string {\n\t\t\t\teventTime := time.Date(2021, 1, 1, 1, 1, 1, 1, time.UTC)\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: eventTime, FromCluster: \"fromCluster1\", ToCluster: \"toCluster1\", FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}}\n\t\t\t\tfailoverHistory = append([]FailoverEvent{{EventTime: now, FromCluster: fromCluster, ToCluster: toCluster, FailoverType: failoverType.String()}}, failoverHistory...)\n\t\t\t\tjsonResp, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn string(jsonResp)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - active-passive to active-active\",\n\t\t\tdomainInfo: func() *persistence.DomainInfo {\n\t\t\t\teventTime := time.Date(2021, 1, 1, 1, 1, 1, 1, time.UTC)\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: eventTime, FromCluster: \"fromCluster1\", ToCluster: \"toCluster1\", FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}}\n\t\t\t\tfailoverHistoryJSON, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn &persistence.DomainInfo{Data: map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(failoverHistoryJSON)}}\n\t\t\t},\n\t\t\tnewFailoverEvent: FailoverEvent{\n\t\t\t\tEventTime:    now,\n\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\tToCluster:    \"\",\n\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t},\n\t\t\tresponse: func() string {\n\t\t\t\teventTime := time.Date(2021, 1, 1, 1, 1, 1, 1, time.UTC)\n\t\t\t\tfailoverHistory := []FailoverEvent{{EventTime: eventTime, FromCluster: \"fromCluster1\", ToCluster: \"toCluster1\", FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()}}\n\t\t\t\tfailoverHistory = append([]FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventTime:    now,\n\t\t\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\t\t\tToCluster:    \"\",\n\t\t\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t\t\t}}, failoverHistory...)\n\t\t\t\tjsonResp, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn string(jsonResp)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - FailoverHistory is at max size\",\n\t\t\tdomainInfo: func() *persistence.DomainInfo {\n\t\t\t\tvar failoverHistory []FailoverEvent\n\t\t\t\tfor i := 0; i < failoverHistoryMaxSize; i++ {\n\t\t\t\t\teventTime := time.Date(2021, 1, i, 1, 1, 1, 1, time.UTC)\n\t\t\t\t\tfailoverHistory = append(failoverHistory, FailoverEvent{EventTime: eventTime, FromCluster: \"fromCluster\" + strconv.Itoa(i), ToCluster: \"toCluster\" + strconv.Itoa(i), FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()})\n\t\t\t\t}\n\t\t\t\tfailoverHistoryJSON, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn &persistence.DomainInfo{Data: map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(failoverHistoryJSON)}}\n\t\t\t},\n\t\t\tnewFailoverEvent: FailoverEvent{\n\t\t\t\tEventTime:    now,\n\t\t\t\tFromCluster:  fromCluster,\n\t\t\t\tToCluster:    toCluster,\n\t\t\t\tFailoverType: failoverType.String(),\n\t\t\t},\n\t\t\tresponse: func() string {\n\t\t\t\tvar failoverHistory []FailoverEvent\n\t\t\t\tfor i := 0; i < 5; i++ {\n\t\t\t\t\teventTime := time.Date(2021, 1, i, 1, 1, 1, 1, time.UTC)\n\t\t\t\t\tfailoverHistory = append(failoverHistory, FailoverEvent{EventTime: eventTime, FromCluster: \"fromCluster\" + strconv.Itoa(i), ToCluster: \"toCluster\" + strconv.Itoa(i), FailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeGrace).String()})\n\t\t\t\t}\n\t\t\t\tfailoverHistory = append([]FailoverEvent{{EventTime: now, FromCluster: fromCluster, ToCluster: toCluster, FailoverType: failoverType.String()}}, failoverHistory[:(5-1)]...)\n\t\t\t\tjsonResp, _ := json.Marshal(failoverHistory)\n\t\t\t\treturn string(jsonResp)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcfg := Config{\n\t\t\t\tFailoverHistoryMaxSize: func(domain string) int {\n\t\t\t\t\treturn failoverHistoryMaxSize\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tdomainInfo := tc.domainInfo()\n\t\t\terr := updateFailoverHistoryInDomainData(domainInfo, cfg, tc.newFailoverEvent)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, domainInfo.Data)\n\t\t\t\tassert.NotNil(t, domainInfo.Data[commonconstants.DomainDataKeyForFailoverHistory])\n\t\t\t\tassert.Equal(t, tc.response(), domainInfo.Data[commonconstants.DomainDataKeyForFailoverHistory])\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandler_FailoverDomain(t *testing.T) {\n\tctx := context.Background()\n\tmaxLength := 1\n\n\tclusterA := \"cluster-a\"\n\tclusterAInitialFailoverVersion := int64(1)\n\n\tclusterB := \"cluster-b\"\n\tclusterBInitialFailoverVersion := int64(2)\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(\n\t\t\tdomainManager *persistence.MockDomainManager,\n\t\t\tupdateRequest *types.FailoverDomainRequest,\n\t\t\tarchivalMetadata *archiver.MockArchivalMetadata,\n\t\t\ttimeSource clock.MockedTimeSource,\n\t\t\tdomainReplicator *MockReplicator,\n\t\t)\n\t\trequest  *types.FailoverDomainRequest\n\t\tresponse func(timeSource clock.MockedTimeSource) *types.FailoverDomainResponse\n\t\terr      error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - active/passive domain - global domain force failover - failing over from cluster A to cluster B\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.FailoverDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: clusterA,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: clusterA}, {ClusterName: clusterB}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverVersion: clusterAInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{\n\t\t\t\t\tNotificationVersion: 15,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetDomainName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour)\n\n\t\t\t\tfailoverHistoryJSON, _ := json.Marshal([]FailoverEvent{\n\t\t\t\t\t{EventTime: timeSource.Now(),\n\t\t\t\t\t\tFromCluster:  clusterA,\n\t\t\t\t\t\tToCluster:    clusterB,\n\t\t\t\t\t\tFailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeForce).String()},\n\t\t\t\t})\n\n\t\t\t\tupdateExpectation := &persistence.UpdateDomainRequest{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\tcommonconstants.DomainDataKeyForFailoverHistory: string(failoverHistoryJSON),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: domainResponse.Config,\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: clusterB,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: clusterA}, {ClusterName: clusterB}},\n\t\t\t\t\t},\n\t\t\t\t\tPreviousFailoverVersion:     commonconstants.InitialPreviousFailoverVersion,\n\t\t\t\t\tConfigVersion:               domainResponse.ConfigVersion,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tLastUpdatedTime:             timeSource.Now().UnixNano(),\n\t\t\t\t\tFailoverNotificationVersion: 15,\n\t\t\t\t\tNotificationVersion:         15,\n\t\t\t\t}\n\n\t\t\t\tdomainManager.EXPECT().UpdateDomain(ctx, updateExpectation).Return(nil).Times(1)\n\n\t\t\t\tdomainReplicator.EXPECT().\n\t\t\t\t\tHandleTransmissionTask(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\ttypes.DomainOperationUpdate,\n\t\t\t\t\t\t&persistence.DomainInfo{\n\t\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\t\tcommonconstants.DomainDataKeyForFailoverHistory: string(failoverHistoryJSON),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tdomainResponse.Config,\n\t\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: clusterB,\n\t\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t\t{ClusterName: clusterA}, {ClusterName: clusterB}},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tdomainResponse.ConfigVersion,\n\t\t\t\t\t\tclusterBInitialFailoverVersion,\n\t\t\t\t\t\tcommonconstants.InitialPreviousFailoverVersion,\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.FailoverDomainRequest{\n\t\t\t\tDomainName:              constants.TestDomainName,\n\t\t\t\tDomainActiveClusterName: common.Ptr(clusterB),\n\t\t\t},\n\t\t\tresponse: func(timeSource clock.MockedTimeSource) *types.FailoverDomainResponse {\n\t\t\t\tdata, _ := json.Marshal([]FailoverEvent{\n\t\t\t\t\t{EventTime: timeSource.Now(),\n\t\t\t\t\t\tFromCluster:  clusterA,\n\t\t\t\t\t\tToCluster:    clusterB,\n\t\t\t\t\t\tFailoverType: commonconstants.FailoverType(commonconstants.FailoverTypeForce).String()},\n\t\t\t\t})\n\t\t\t\treturn &types.FailoverDomainResponse{\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tFailoverVersion: clusterBInitialFailoverVersion,\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tUUID:   constants.TestDomainID,\n\t\t\t\t\t\tData:   map[string]string{commonconstants.DomainDataKeyForFailoverHistory: string(data)},\n\t\t\t\t\t\tStatus: common.Ptr(types.DomainStatusRegistered),\n\t\t\t\t\t},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\t\tHistoryArchivalStatus:                  common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tVisibilityArchivalStatus:               common.Ptr(types.ArchivalStatusDisabled),\n\t\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\t\tActiveClusterName: clusterB,\n\t\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t\t{ClusterName: clusterA}, {ClusterName: clusterB},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - domain not found\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.FailoverDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetDomainName()}).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{Message: \"Domain not found\"}).Times(1)\n\t\t\t},\n\t\t\trequest: &types.FailoverDomainRequest{\n\t\t\t\tDomainName:              constants.TestDomainName,\n\t\t\t\tDomainActiveClusterName: common.Ptr(cluster.TestAlternativeClusterName),\n\t\t\t},\n\t\t\terr: &types.EntityNotExistsError{Message: \"Domain not found\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - update too frequent\",\n\t\t\tsetupMock: func(domainManager *persistence.MockDomainManager, updateRequest *types.FailoverDomainRequest, archivalMetadata *archiver.MockArchivalMetadata, timeSource clock.MockedTimeSource, domainReplicator *MockReplicator) {\n\t\t\t\tdomainResponse := &persistence.GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.DomainConfig{\n\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t\t},\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tName:   constants.TestDomainName,\n\t\t\t\t\t\tID:     constants.TestDomainID,\n\t\t\t\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\t\tLastUpdatedTime: timeSource.Now().UnixNano(), // Set to current time to trigger cool down\n\t\t\t\t\tFailoverVersion: cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t}\n\t\t\t\tdomainManager.EXPECT().GetMetadata(ctx).Return(&persistence.GetMetadataResponse{}, nil).Times(1)\n\t\t\t\tdomainManager.EXPECT().GetDomain(ctx, &persistence.GetDomainRequest{Name: updateRequest.GetDomainName()}).\n\t\t\t\t\tReturn(domainResponse, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &types.FailoverDomainRequest{\n\t\t\t\tDomainName:              constants.TestDomainName,\n\t\t\t\tDomainActiveClusterName: common.Ptr(cluster.TestCurrentClusterName),\n\t\t\t},\n\t\t\terr: errDomainUpdateTooFrequent,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\tmockReplicator := NewMockReplicator(ctrl)\n\t\t\tmockArchivalMetadata := &archiver.MockArchivalMetadata{}\n\n\t\t\ttestConfig := Config{\n\t\t\t\tMinRetentionDays:       dynamicproperties.GetIntPropertyFn(1),\n\t\t\t\tMaxRetentionDays:       dynamicproperties.GetIntPropertyFn(5),\n\t\t\t\tRequiredDomainDataKeys: nil,\n\t\t\t\tMaxBadBinaryCount:      dynamicproperties.GetIntPropertyFilteredByDomain(maxLength),\n\t\t\t\tFailoverCoolDown:       func(string) time.Duration { return time.Second },\n\t\t\t\tFailoverHistoryMaxSize: dynamicproperties.GetIntPropertyFilteredByDomain(5),\n\t\t\t}\n\n\t\t\tclusterMetadata := cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 100,\n\t\t\t\t\tPrimaryClusterName:       clusterA,\n\t\t\t\t\tCurrentClusterName:       clusterA,\n\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\tclusterA: {\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: clusterAInitialFailoverVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tclusterB: {\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: clusterBInitialFailoverVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfunc(d string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Unix(1730419200, 0))\n\n\t\t\thandler := handlerImpl{\n\t\t\t\tdomainManager:       mockDomainManager,\n\t\t\t\tclusterMetadata:     clusterMetadata,\n\t\t\t\tdomainReplicator:    mockReplicator,\n\t\t\t\tdomainAttrValidator: newAttrValidator(clusterMetadata, int32(testConfig.MinRetentionDays())),\n\t\t\t\tarchivalMetadata:    mockArchivalMetadata,\n\t\t\t\tarchiverProvider:    provider.NewArchiverProvider(nil, nil),\n\t\t\t\ttimeSource:          mockTimeSource,\n\t\t\t\tconfig:              testConfig,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\ttc.setupMock(mockDomainManager, tc.request, mockArchivalMetadata, mockTimeSource, mockReplicator)\n\n\t\t\tresponse, err := handler.FailoverDomain(ctx, tc.request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, response)\n\t\t\t\tassert.Equal(t, tc.response(mockTimeSource), response)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestBuildActiveActiveClustersFromUpdateRequest(t *testing.T) {\n\n\ttestsCases := map[string]struct {\n\t\tupdateRequest          *types.UpdateDomainRequest\n\t\tconfig                 *persistence.DomainReplicationConfig\n\t\tdomainName             string\n\t\thandler                *handlerImpl\n\t\texpectedActiveClusters *types.ActiveClusters\n\t\texpectedIsChanged      bool\n\t}{\n\t\t\"Success case - ActiveClusters - failover event - where there's existing cluster attributes and we expect them to be incremented\": {\n\t\t\tupdateRequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"location\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\", // this is expected to be a failvoer to cluster C from the existing A\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tconfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"location\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\tFailoverVersion:   102,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: true,\n\t\t},\n\t\t\"Success case - ActiveClusters - where there is the introduction of cluster attributes for the first time - we should see that these results are reflected\": {\n\t\t\tupdateRequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"location\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\t\t// failover version can be absent\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\t\t// failover version can be absent\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\t\t// failover version can be absent\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: true,\n\t\t},\n\t\t\"Success case - ActiveClusters - where there existing cluster attributes. These should be merged\": {\n\t\t\tupdateRequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"location\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\t\t// failover version can be absent\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tconfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"location\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"nyc\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"tokyo\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterC\",\n\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"morocco\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedIsChanged: true,\n\t\t},\n\t\t\"Success case - AttributeScopes is nil\": {\n\t\t\tupdateRequest: &types.UpdateDomainRequest{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveClusters: nil,\n\t\t\texpectedIsChanged:      false,\n\t\t},\n\t}\n\n\tfor name, tc := range testsCases {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\n\t\t\tmetadata := cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 100,\n\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\t\"clusterA\": {\n\t\t\t\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"clusterB\": {\n\t\t\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"clusterC\": {\n\t\t\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"clusterD\": {\n\t\t\t\t\t\t\tInitialFailoverVersion: 3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfunc(d string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\thandler := handlerImpl{\n\t\t\t\tdomainManager:    mockDomainManager,\n\t\t\t\tclusterMetadata:  metadata,\n\t\t\t\tarchiverProvider: provider.NewArchiverProvider(nil, nil),\n\t\t\t\ttimeSource:       mockTimeSource,\n\t\t\t\tlogger:           log.NewNoop(),\n\t\t\t}\n\n\t\t\tactiveClusters, isChanged := handler.buildActiveActiveClusterScopesFromUpdateRequest(tc.updateRequest, tc.config, tc.domainName)\n\t\t\tassert.Equal(t, tc.expectedActiveClusters, activeClusters)\n\t\t\tassert.Equal(t, tc.expectedIsChanged, isChanged)\n\t\t})\n\t}\n}\n\nfunc TestActiveClustersFromRegisterRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\trequest         *types.RegisterDomainRequest\n\t\texpectedResult  *types.ActiveClusters\n\t\texpectedErr     error\n\t\tclusterMetadata func() cluster.Metadata\n\t}{\n\t\t{\n\t\t\tname: \"local domain returns nil\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: false,\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"global domain with no active cluster data returns nil\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new AttributeScopes with valid clusters\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new AttributeScopes with invalid cluster\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"invalid-cluster\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    &types.BadRequestError{},\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple scopes with multiple attributes\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestAlternativeClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   cluster.TestCurrentClusterInitialFailoverVersion,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty ActiveClusters with non-nil AttributeScopes\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.GetTestClusterMetadata(true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"custom cluster metadata with different failover versions\",\n\t\t\trequest: &types.RegisterDomainRequest{\n\t\t\t\tName:           \"test-domain\",\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"clusterA\"},\n\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: \"clusterB\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterA\",\n\t\t\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"region2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"clusterB\",\n\t\t\t\t\t\t\t\tFailoverVersion:   20,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t\tclusterMetadata: func() cluster.Metadata {\n\t\t\t\treturn cluster.NewMetadata(\n\t\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\t\tFailoverVersionIncrement: 100,\n\t\t\t\t\t\tPrimaryClusterName:       \"clusterA\",\n\t\t\t\t\t\tCurrentClusterName:       \"clusterA\",\n\t\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\t\t\"clusterA\": {\n\t\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\t\tInitialFailoverVersion: 10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"clusterB\": {\n\t\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\t\tInitialFailoverVersion: 20,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tfunc(d string) bool { return false },\n\t\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\t\tlog.NewNoop(),\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thandler := &handlerImpl{\n\t\t\t\tclusterMetadata: tc.clusterMetadata(),\n\t\t\t\tlogger:          log.NewNoop(),\n\t\t\t}\n\n\t\t\tresult, err := handler.activeClustersFromRegisterRequest(tc.request)\n\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.IsType(t, tc.expectedErr, err)\n\t\t\t\tif badReqErr, ok := tc.expectedErr.(*types.BadRequestError); ok {\n\t\t\t\t\tresultErr, ok := err.(*types.BadRequestError)\n\t\t\t\t\tassert.True(t, ok)\n\t\t\t\t\tif badReqErr.Message != \"\" {\n\t\t\t\t\t\tassert.Equal(t, badReqErr.Message, resultErr.Message)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateDomainReplicationConfigForFailover(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\treplicationConfig    *persistence.DomainReplicationConfig\n\t\tisGlobalDomain       bool\n\t\tconfigurationChanged bool\n\t\tactiveClusterChanged bool\n\t\tisPrimaryCluster     bool\n\t\texpectedErr          error\n\t}{\n\t\t{\n\t\t\tname: \"global domain with valid config on primary cluster - no changes\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: false,\n\t\t\tactiveClusterChanged: false,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain config change only on primary cluster\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: true,\n\t\t\tactiveClusterChanged: false,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain active cluster change only on primary cluster\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: false,\n\t\t\tactiveClusterChanged: true,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain config change on non-primary cluster should fail\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: true,\n\t\t\tactiveClusterChanged: false,\n\t\t\tisPrimaryCluster:     false,\n\t\t\texpectedErr:          errNotPrimaryCluster,\n\t\t},\n\t\t{\n\t\t\tname: \"global active-passive domain cannot change both config and active cluster\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: true,\n\t\t\tactiveClusterChanged: true,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          errCannotDoDomainFailoverAndUpdate,\n\t\t},\n\t\t{\n\t\t\tname: \"global active-active domain can change both config and active cluster\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: true,\n\t\t\tactiveClusterChanged: true,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"global active-active domain with AttributeScopes can change both\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {ActiveClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t\t\t\t\"dc2\": {ActiveClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: true,\n\t\t\tactiveClusterChanged: true,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"global domain with invalid cluster name\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: \"invalid-cluster\",\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: \"invalid-cluster\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: false,\n\t\t\tactiveClusterChanged: false,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"global domain with no clusters should fail\",\n\t\t\treplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{},\n\t\t\t},\n\t\t\tisGlobalDomain:       true,\n\t\t\tconfigurationChanged: false,\n\t\t\tactiveClusterChanged: false,\n\t\t\tisPrimaryCluster:     true,\n\t\t\texpectedErr:          &types.BadRequestError{},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tclusterMetadata := cluster.GetTestClusterMetadata(tc.isPrimaryCluster)\n\t\t\thandler := &handlerImpl{\n\t\t\t\tclusterMetadata:     clusterMetadata,\n\t\t\t\tdomainAttrValidator: newAttrValidator(clusterMetadata, 1),\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\terr := handler.validateDomainReplicationConfigForFailover(\n\t\t\t\ttc.replicationConfig,\n\t\t\t\ttc.configurationChanged,\n\t\t\t\ttc.activeClusterChanged,\n\t\t\t)\n\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.IsType(t, tc.expectedErr, err)\n\t\t\t\tif tc.expectedErr == errNotPrimaryCluster {\n\t\t\t\t\tassert.Equal(t, errNotPrimaryCluster, err)\n\t\t\t\t} else if tc.expectedErr == errCannotDoDomainFailoverAndUpdate {\n\t\t\t\t\tassert.Equal(t, errCannotDoDomainFailoverAndUpdate, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/replicationTaskExecutor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination replicationTaskHandler_mock.go\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\tguuid \"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// ErrEmptyDomainReplicationTask is the error to indicate empty replication task\n\tErrEmptyDomainReplicationTask = &types.BadRequestError{Message: \"empty domain replication task\"}\n\t// ErrInvalidDomainOperation is the error to indicate empty domain operation attribute\n\tErrInvalidDomainOperation = &types.BadRequestError{Message: \"invalid domain operation attribute\"}\n\t// ErrInvalidDomainID is the error to indicate empty rID attribute\n\tErrInvalidDomainID = &types.BadRequestError{Message: \"invalid domain ID attribute\"}\n\t// ErrInvalidDomainInfo is the error to indicate empty info attribute\n\tErrInvalidDomainInfo = &types.BadRequestError{Message: \"invalid domain info attribute\"}\n\t// ErrInvalidDomainConfig is the error to indicate empty config attribute\n\tErrInvalidDomainConfig = &types.BadRequestError{Message: \"invalid domain config attribute\"}\n\t// ErrInvalidDomainReplicationConfig is the error to indicate empty replication config attribute\n\tErrInvalidDomainReplicationConfig = &types.BadRequestError{Message: \"invalid domain replication config attribute\"}\n\t// ErrInvalidDomainStatus is the error to indicate invalid domain status\n\tErrInvalidDomainStatus = &types.BadRequestError{Message: \"invalid domain status attribute\"}\n\t// ErrNameUUIDCollision is the error to indicate domain name / UUID collision\n\tErrNameUUIDCollision = &types.BadRequestError{Message: \"domain replication encounter name / UUID collision\"}\n)\n\nconst (\n\tdefaultDomainRepliationTaskContextTimeout = 5 * time.Second\n)\n\n// NOTE: the counterpart of domain replication transmission logic is in service/fropntend package\n\ntype (\n\t// ReplicationTaskExecutor is the interface which is to execute domain replication task\n\tReplicationTaskExecutor interface {\n\t\tExecute(task *types.DomainTaskAttributes) error\n\t}\n\n\tdomainReplicationTaskExecutorImpl struct {\n\t\tdomainManager            persistence.DomainManager\n\t\ttimeSource               clock.TimeSource\n\t\tlogger                   log.Logger\n\t\tdomainAuditManager       persistence.DomainAuditManager\n\t\tenableDomainAuditLogging dynamicproperties.BoolPropertyFn\n\t}\n)\n\n// NewReplicationTaskExecutor create a new instance of domain replicator\nfunc NewReplicationTaskExecutor(\n\tdomainManager persistence.DomainManager,\n\tdomainAuditManager persistence.DomainAuditManager,\n\ttimeSource clock.TimeSource,\n\tlogger log.Logger,\n\tenableDomainAuditLogging dynamicproperties.BoolPropertyFn,\n) ReplicationTaskExecutor {\n\treturn &domainReplicationTaskExecutorImpl{\n\t\tdomainManager:            domainManager,\n\t\ttimeSource:               timeSource,\n\t\tlogger:                   logger,\n\t\tdomainAuditManager:       domainAuditManager,\n\t\tenableDomainAuditLogging: enableDomainAuditLogging,\n\t}\n}\n\n// Execute handles receiving of the domain replication task\nfunc (h *domainReplicationTaskExecutorImpl) Execute(task *types.DomainTaskAttributes) error {\n\tctx, cancel := context.WithTimeout(context.Background(), defaultDomainRepliationTaskContextTimeout)\n\tdefer cancel()\n\n\tif err := h.validateDomainReplicationTask(task); err != nil {\n\t\treturn err\n\t}\n\n\tswitch task.GetDomainOperation() {\n\tcase types.DomainOperationCreate:\n\t\treturn h.handleDomainCreationReplicationTask(ctx, task)\n\tcase types.DomainOperationUpdate:\n\t\treturn h.handleDomainUpdateReplicationTask(ctx, task)\n\tcase types.DomainOperationDelete:\n\t\treturn h.handleDomainDeleteReplicationTask(ctx, task)\n\tdefault:\n\t\treturn ErrInvalidDomainOperation\n\t}\n}\n\n// handleDomainCreationReplicationTask handles the domain creation replication task\nfunc (h *domainReplicationTaskExecutorImpl) handleDomainCreationReplicationTask(ctx context.Context, task *types.DomainTaskAttributes) error {\n\t// task already validated\n\tstatus, err := h.convertDomainStatusFromThrift(task.Info.Status)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\trequest := &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          task.GetID(),\n\t\t\tName:        task.Info.GetName(),\n\t\t\tStatus:      status,\n\t\t\tDescription: task.Info.GetDescription(),\n\t\t\tOwnerEmail:  task.Info.GetOwnerEmail(),\n\t\t\tData:        task.Info.Data,\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                task.Config.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\t\tEmitMetric:               task.Config.GetEmitMetric(),\n\t\t\tHistoryArchivalStatus:    task.Config.GetHistoryArchivalStatus(),\n\t\t\tHistoryArchivalURI:       task.Config.GetHistoryArchivalURI(),\n\t\t\tVisibilityArchivalStatus: task.Config.GetVisibilityArchivalStatus(),\n\t\t\tVisibilityArchivalURI:    task.Config.GetVisibilityArchivalURI(),\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: task.ReplicationConfig.GetActiveClusterName(),\n\t\t\tClusters:          h.convertClusterReplicationConfigFromThrift(task.ReplicationConfig.Clusters),\n\t\t\tActiveClusters:    task.ReplicationConfig.GetActiveClusters(),\n\t\t},\n\t\tIsGlobalDomain:  true, // local domain will not be replicated\n\t\tConfigVersion:   task.GetConfigVersion(),\n\t\tFailoverVersion: task.GetFailoverVersion(),\n\t\tLastUpdatedTime: h.timeSource.Now().UnixNano(),\n\t}\n\n\t_, err = h.domainManager.CreateDomain(ctx, request)\n\tif err != nil {\n\t\t// SQL and Cassandra handle domain UUID collision differently\n\t\t// here, whenever seeing a error replicating a domain\n\t\t// do a check if there is a name / UUID collision\n\n\t\trecordExists := true\n\t\tresp, getErr := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\t\tName: task.Info.GetName(),\n\t\t})\n\t\tswitch getErr.(type) {\n\t\tcase nil:\n\t\t\tif resp.Info.ID != task.GetID() {\n\t\t\t\treturn ErrNameUUIDCollision\n\t\t\t}\n\t\tcase *types.EntityNotExistsError:\n\t\t\t// no check is necessary\n\t\t\trecordExists = false\n\t\tdefault:\n\t\t\t// return the original err\n\t\t\treturn err\n\t\t}\n\n\t\tresp, getErr = h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\t\tID: task.GetID(),\n\t\t})\n\t\tswitch getErr.(type) {\n\t\tcase nil:\n\t\t\tif resp.Info.Name != task.Info.GetName() {\n\t\t\t\treturn ErrNameUUIDCollision\n\t\t\t}\n\t\tcase *types.EntityNotExistsError:\n\t\t\t// no check is necessary\n\t\t\trecordExists = false\n\t\tdefault:\n\t\t\t// return the original err\n\t\t\treturn err\n\t\t}\n\n\t\tif recordExists {\n\t\t\t// name -> id & id -> name check pass, this is duplication request\n\t\t\treturn nil\n\t\t}\n\t\treturn err\n\t}\n\n\tresp, getErr := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\tID: task.GetID(),\n\t})\n\tif getErr != nil {\n\t\treturn fmt.Errorf(\"failed to get domain while trying to create domain audit log: %w\", getErr)\n\t}\n\th.createDomainAuditLog(\n\t\tctx,\n\t\ttask,\n\t\tpersistence.DomainAuditOperationTypeCreate,\n\t\tnil,\n\t\tresp,\n\t)\n\n\treturn nil\n}\n\n// handleDomainUpdateReplicationTask handles the domain update replication task\nfunc (h *domainReplicationTaskExecutorImpl) handleDomainUpdateReplicationTask(ctx context.Context, task *types.DomainTaskAttributes) error {\n\t// task already validated\n\tstatus, err := h.convertDomainStatusFromThrift(task.Info.Status)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// first we need to get the current notification version since we need to it for conditional update\n\tmetadata, err := h.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\th.logger.Error(\"Error getting metadata while handling replication task\", tag.Error(err))\n\t\treturn err\n\t}\n\tnotificationVersion := metadata.NotificationVersion\n\n\t// plus, we need to check whether the config version is <= the config version set in the input\n\t// plus, we need to check whether the failover version is <= the failover version set in the input\n\toriginalDomainState, err := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\tID: task.GetID(),\n\t})\n\tintendedDomainState := originalDomainState.DeepCopy()\n\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t// this can happen if the create domain replication task is to processed.\n\t\t\t// e.g. new cluster which does not have anything\n\t\t\treturn h.handleDomainCreationReplicationTask(ctx, task)\n\t\t}\n\t\th.logger.Error(\"Domain update failed, error in fetching domain\", tag.Error(err))\n\t\treturn err\n\t}\n\n\trecordUpdated := false\n\trequest := &persistence.UpdateDomainRequest{\n\t\tInfo:                        intendedDomainState.Info,\n\t\tConfig:                      intendedDomainState.Config,\n\t\tReplicationConfig:           intendedDomainState.ReplicationConfig,\n\t\tConfigVersion:               intendedDomainState.ConfigVersion,\n\t\tFailoverVersion:             intendedDomainState.FailoverVersion,\n\t\tFailoverNotificationVersion: intendedDomainState.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     intendedDomainState.PreviousFailoverVersion,\n\t\tNotificationVersion:         notificationVersion,\n\t\tLastUpdatedTime:             h.timeSource.Now().UnixNano(),\n\t}\n\n\tif intendedDomainState.ConfigVersion < task.GetConfigVersion() {\n\t\trecordUpdated = true\n\t\trequest.Info = &persistence.DomainInfo{\n\t\t\tID:          task.GetID(),\n\t\t\tName:        task.Info.GetName(),\n\t\t\tStatus:      status,\n\t\t\tDescription: task.Info.GetDescription(),\n\t\t\tOwnerEmail:  task.Info.GetOwnerEmail(),\n\t\t\tData:        task.Info.Data,\n\t\t}\n\t\trequest.Config = &persistence.DomainConfig{\n\t\t\tRetention:                task.Config.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\t\tEmitMetric:               task.Config.GetEmitMetric(),\n\t\t\tHistoryArchivalStatus:    task.Config.GetHistoryArchivalStatus(),\n\t\t\tHistoryArchivalURI:       task.Config.GetHistoryArchivalURI(),\n\t\t\tVisibilityArchivalStatus: task.Config.GetVisibilityArchivalStatus(),\n\t\t\tVisibilityArchivalURI:    task.Config.GetVisibilityArchivalURI(),\n\t\t\tIsolationGroups:          task.Config.GetIsolationGroupsConfiguration(),\n\t\t\tAsyncWorkflowConfig:      task.Config.GetAsyncWorkflowConfiguration(),\n\t\t}\n\t\tif task.Config.GetBadBinaries() != nil {\n\t\t\trequest.Config.BadBinaries = *task.Config.GetBadBinaries()\n\t\t}\n\t\trequest.ReplicationConfig.Clusters = h.convertClusterReplicationConfigFromThrift(task.ReplicationConfig.Clusters)\n\t\trequest.ConfigVersion = task.GetConfigVersion()\n\t}\n\n\tif originalDomainState.FailoverVersion < task.GetFailoverVersion() {\n\t\trecordUpdated = true\n\t\trequest.ReplicationConfig.ActiveClusterName = task.ReplicationConfig.GetActiveClusterName()\n\t\trequest.ReplicationConfig.ActiveClusters = task.ReplicationConfig.GetActiveClusters()\n\t\trequest.FailoverVersion = task.GetFailoverVersion()\n\t\trequest.FailoverNotificationVersion = notificationVersion\n\t\trequest.PreviousFailoverVersion = task.GetPreviousFailoverVersion()\n\t} else if !originalDomainState.ReplicationConfig.IsActiveActive() {\n\t\th.logger.Warn(\"the existing failover version was more recent, indicating that the domain replication message was out of date and is consequently being dropped\",\n\t\t\ttag.WorkflowDomainName(originalDomainState.Info.Name),\n\t\t\ttag.FailoverVersion(originalDomainState.FailoverVersion),\n\t\t\ttag.FailoverVersion(task.GetFailoverVersion()))\n\t}\n\n\tif intendedDomainState.ReplicationConfig.IsActiveActive() || task.ReplicationConfig.IsActiveActive() {\n\t\tmergedActiveClusters, aaChanged := mergeActiveActiveScopes(intendedDomainState.ReplicationConfig.ActiveClusters, task.ReplicationConfig.ActiveClusters)\n\t\tif aaChanged {\n\t\t\trecordUpdated = true\n\t\t\trequest.ReplicationConfig.ActiveClusters = mergedActiveClusters\n\t\t}\n\t}\n\n\tif !recordUpdated {\n\t\th.logger.Warn(\"no record updated while handling domain update replication task\",\n\t\t\ttag.WorkflowDomainName(task.Info.GetName()),\n\t\t\ttag.WorkflowDomainID(task.GetID()))\n\t\treturn nil\n\t}\n\n\terr = h.domainManager.UpdateDomain(ctx, request)\n\tif err != nil {\n\t\th.logger.Error(\"failed to update domain while handling domain update replication task\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowDomainName(task.Info.GetName()),\n\t\t\ttag.WorkflowDomainID(task.GetID()))\n\t\treturn err\n\t}\n\tafterUpdate, getErr := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\tID: task.GetID(),\n\t})\n\tif getErr != nil {\n\t\treturn fmt.Errorf(\"failed to get domain while trying to create domain audit log for update: %w\", getErr)\n\t}\n\n\t// relying on the fact that for both failovers and the failover of cluster-attibutes\n\t// within the domain, in both instances the failover version will be incremented, indicating\n\t// this is failover type update.\n\tif intendedDomainState.FailoverVersion < afterUpdate.FailoverVersion {\n\t\th.createDomainAuditLog(\n\t\t\tctx,\n\t\t\ttask,\n\t\t\tpersistence.DomainAuditOperationTypeFailover,\n\t\t\toriginalDomainState,\n\t\t\tafterUpdate,\n\t\t)\n\t\treturn nil\n\t}\n\th.createDomainAuditLog(\n\t\tctx,\n\t\ttask,\n\t\tpersistence.DomainAuditOperationTypeUpdate,\n\t\toriginalDomainState,\n\t\tafterUpdate,\n\t)\n\treturn nil\n}\n\n// handleDomainDeleteReplicationTask handles the domain delete replication task\nfunc (h *domainReplicationTaskExecutorImpl) handleDomainDeleteReplicationTask(ctx context.Context, task *types.DomainTaskAttributes) error {\n\trequest := &persistence.DeleteDomainByNameRequest{\n\t\tName: task.Info.GetName(),\n\t}\n\n\t// ignoring error since this might be already deleted\n\tgetDomainResp, _ := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\tName: task.Info.GetName(),\n\t})\n\n\terr := h.domainManager.DeleteDomainByName(ctx, request)\n\tif err != nil {\n\t\t_, err := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\t\tName: task.Info.GetName(),\n\t\t})\n\n\t\tvar entityNotExistsError *types.EntityNotExistsError\n\t\tif errors.As(err, &entityNotExistsError) {\n\t\t\treturn nil\n\t\t}\n\t\treturn err\n\t}\n\n\th.createDomainAuditLog(\n\t\tctx,\n\t\ttask,\n\t\tpersistence.DomainAuditOperationTypeDelete,\n\t\tgetDomainResp,\n\t\tnil,\n\t)\n\treturn nil\n}\n\nfunc (h *domainReplicationTaskExecutorImpl) validateDomainReplicationTask(task *types.DomainTaskAttributes) error {\n\tif task == nil {\n\t\treturn ErrEmptyDomainReplicationTask\n\t}\n\n\tif task.DomainOperation == nil {\n\t\treturn ErrInvalidDomainOperation\n\t} else if task.ID == \"\" {\n\t\treturn ErrInvalidDomainID\n\t} else if task.Info == nil {\n\t\treturn ErrInvalidDomainInfo\n\t} else if task.Config == nil {\n\t\treturn ErrInvalidDomainConfig\n\t} else if task.ReplicationConfig == nil {\n\t\treturn ErrInvalidDomainReplicationConfig\n\t}\n\treturn nil\n}\n\nfunc (h *domainReplicationTaskExecutorImpl) convertClusterReplicationConfigFromThrift(\n\tinput []*types.ClusterReplicationConfiguration) []*persistence.ClusterReplicationConfig {\n\toutput := []*persistence.ClusterReplicationConfig{}\n\tfor _, cluster := range input {\n\t\tclusterName := cluster.GetClusterName()\n\t\toutput = append(output, &persistence.ClusterReplicationConfig{ClusterName: clusterName})\n\t}\n\treturn output\n}\n\nfunc (h *domainReplicationTaskExecutorImpl) convertDomainStatusFromThrift(input *types.DomainStatus) (int, error) {\n\tif input == nil {\n\t\treturn 0, ErrInvalidDomainStatus\n\t}\n\n\tswitch *input {\n\tcase types.DomainStatusRegistered:\n\t\treturn persistence.DomainStatusRegistered, nil\n\tcase types.DomainStatusDeprecated:\n\t\treturn persistence.DomainStatusDeprecated, nil\n\tdefault:\n\t\treturn 0, ErrInvalidDomainStatus\n\t}\n}\n\nfunc (h *domainReplicationTaskExecutorImpl) createDomainAuditLog(\n\tctx context.Context,\n\ttask *types.DomainTaskAttributes,\n\toperationType persistence.DomainAuditOperationType,\n\tstateBefore *persistence.GetDomainResponse,\n\tstateAfter *persistence.GetDomainResponse,\n) {\n\tif !h.enableDomainAuditLogging() {\n\t\treturn\n\t}\n\teventID, err := guuid.NewV7()\n\tif err != nil {\n\t\th.logger.Error(\"failed to generate event ID while creating domain audit log\", tag.Error(err))\n\t\treturn\n\t}\n\tcreationTime := time.Unix(eventID.Time().UnixTime())\n\t_, err = h.domainAuditManager.CreateDomainAuditLog(ctx, &persistence.CreateDomainAuditLogRequest{\n\t\tEventID:       eventID.String(),\n\t\tDomainID:      task.GetID(),\n\t\tOperationType: operationType,\n\t\tCreatedTime:   creationTime,\n\t\tStateBefore:   stateBefore,\n\t\tStateAfter:    stateAfter,\n\t\tIdentity:      \"replication task executor\",\n\t\tComment:       fmt.Sprintf(\"replicated domain operation %s for domain %s\", operationType.String(), task.Info.GetName()),\n\t})\n\tif err != nil {\n\t\th.logger.Error(\"failed to create domain audit log while creating domain audit log\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowDomainName(task.Info.GetName()),\n\t\t\ttag.WorkflowDomainID(task.GetID()))\n\t}\n}\n"
  },
  {
    "path": "common/domain/replicationTaskExecutor_integration_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tdomainReplicationTaskExecutorSuite struct {\n\t\t*persistencetests.TestBase\n\t\tdomainReplicator *domainReplicationTaskExecutorImpl\n\t}\n)\n\nfunc TestDomainReplicationTaskExecutorSuite(t *testing.T) {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\n\ts := new(domainReplicationTaskExecutorSuite)\n\ts.setupTestBase(t)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) setupTestBase(t *testing.T) {\n\tsqliteTestBaseOptions := sqlite.GetTestClusterOption()\n\ts.TestBase = persistencetests.NewTestBaseWithSQL(t, sqliteTestBaseOptions)\n\ts.Setup()\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) SetupTest() {\n\ts.setupTestBase(s.T())\n\n\tdomainAuditManager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\tif err != nil {\n\t\ts.T().Fatalf(\"Failed to create domain audit manager: %v\", err)\n\t}\n\n\t// Disable audit logging for integration tests as SQLite doesn't fully support the audit manager\n\tenableAuditLogging := func(...dynamicproperties.FilterOption) bool { return false }\n\n\ts.domainReplicator = NewReplicationTaskExecutor(\n\t\ts.DomainManager,\n\t\tdomainAuditManager,\n\t\tclock.NewRealTimeSource(),\n\t\ts.Logger,\n\t\tenableAuditLogging,\n\t).(*domainReplicationTaskExecutorImpl)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TearDownTest() {\n\ts.TearDownWorkflowStore()\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_RegisterDomainTask_NameUUIDCollision() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\ttask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\n\terr := s.domainReplicator.Execute(task)\n\ts.Nil(err)\n\n\ttask.ID = uuid.New()\n\ttask.Info.Name = name\n\terr = s.domainReplicator.Execute(task)\n\ts.NotNil(err)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\ttask.ID = id\n\ttask.Info.Name = \"other random domain test name\"\n\terr = s.domainReplicator.Execute(task)\n\ts.NotNil(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_RegisterDomainTask() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\ttask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(task)\n\ts.Nil(err)\n\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{ID: id})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusRegistered, resp.Info.Status)\n\ts.Equal(description, resp.Info.Description)\n\ts.Equal(ownerEmail, resp.Info.OwnerEmail)\n\ts.Equal(data, resp.Info.Data)\n\ts.Equal(retention, resp.Config.Retention)\n\ts.Equal(emitMetric, resp.Config.EmitMetric)\n\ts.Equal(historyArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(historyArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(visibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(visibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(clusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(clusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(configVersion, resp.ConfigVersion)\n\ts.Equal(failoverVersion, resp.FailoverVersion)\n\ts.Equal(int64(0), resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n\n\t// handle duplicated task\n\terr = s.domainReplicator.Execute(task)\n\ts.Nil(err)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_UpdateDomainTask_DomainNotExist() {\n\toperation := types.DomainOperationUpdate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(12)\n\tfailoverVersion := int64(59)\n\tdomainData := map[string]string{\"k1\": \"v1\", \"k2\": \"v2\"}\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tupdateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        domainData,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(updateTask)\n\ts.Nil(err)\n\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusRegistered, resp.Info.Status)\n\ts.Equal(description, resp.Info.Description)\n\ts.Equal(ownerEmail, resp.Info.OwnerEmail)\n\ts.Equal(domainData, resp.Info.Data)\n\ts.Equal(retention, resp.Config.Retention)\n\ts.Equal(emitMetric, resp.Config.EmitMetric)\n\ts.Equal(historyArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(historyArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(visibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(visibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(clusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(clusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(configVersion, resp.ConfigVersion)\n\ts.Equal(failoverVersion, resp.FailoverVersion)\n\ts.Equal(int64(0), resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_UpdateDomainTask_UpdateConfig_UpdateActiveCluster() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tcreateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\n\terr := s.domainReplicator.Execute(createTask)\n\ts.Nil(err)\n\n\t// success update case\n\tupdateOperation := types.DomainOperationUpdate\n\tupdateStatus := types.DomainStatusDeprecated\n\tupdateDescription := \"other random domain test description\"\n\tupdateOwnerEmail := \"other random domain test owner\"\n\tupdatedData := map[string]string{\"k\": \"v1\"}\n\tupdateRetention := int32(122)\n\tupdateEmitMetric := true\n\tupdateHistoryArchivalStatus := types.ArchivalStatusDisabled\n\tupdateHistoryArchivalURI := \"some updated history archival uri\"\n\tupdateVisibilityArchivalStatus := types.ArchivalStatusDisabled\n\tupdateVisibilityArchivalURI := \"some updated visibility archival uri\"\n\tupdateClusterActive := \"other random active cluster name\"\n\tupdateClusterStandby := \"other random standby cluster name\"\n\tupdateConfigVersion := configVersion + 1\n\tupdateFailoverVersion := failoverVersion + 1\n\tupdateClusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: updateClusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: updateClusterStandby,\n\t\t},\n\t}\n\tupdateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &updateOperation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &updateStatus,\n\t\t\tDescription: updateDescription,\n\t\t\tOwnerEmail:  updateOwnerEmail,\n\t\t\tData:        updatedData,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: updateRetention,\n\t\t\tEmitMetric:                             updateEmitMetric,\n\t\t\tHistoryArchivalStatus:                  updateHistoryArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     updateHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus:               updateVisibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  updateVisibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tConfigVersion:   updateConfigVersion,\n\t\tFailoverVersion: updateFailoverVersion,\n\t}\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(updateTask)\n\ts.Nil(err)\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusDeprecated, resp.Info.Status)\n\ts.Equal(updateDescription, resp.Info.Description)\n\ts.Equal(updateOwnerEmail, resp.Info.OwnerEmail)\n\ts.Equal(updatedData, resp.Info.Data)\n\ts.Equal(updateRetention, resp.Config.Retention)\n\ts.Equal(updateEmitMetric, resp.Config.EmitMetric)\n\ts.Equal(updateHistoryArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(updateHistoryArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(updateVisibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(updateVisibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(updateClusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(updateClusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(updateConfigVersion, resp.ConfigVersion)\n\ts.Equal(updateFailoverVersion, resp.FailoverVersion)\n\ts.Equal(notificationVersion, resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_UpdateDomainTask_UpdateConfig_NoUpdateActiveCluster() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusDisabled\n\thistoryArchivalURI := \"\"\n\tvisibilityArchivalStatus := types.ArchivalStatusDisabled\n\tvisibilityArchivalURI := \"\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tcreateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\n\terr := s.domainReplicator.Execute(createTask)\n\ts.Nil(err)\n\n\t// success update case\n\tupdateOperation := types.DomainOperationUpdate\n\tupdateStatus := types.DomainStatusDeprecated\n\tupdateDescription := \"other random domain test description\"\n\tupdateOwnerEmail := \"other random domain test owner\"\n\tupdateData := map[string]string{\"k\": \"v2\"}\n\tupdateRetention := int32(122)\n\tupdateEmitMetric := true\n\tupdateHistoryArchivalStatus := types.ArchivalStatusEnabled\n\tupdateHistoryArchivalURI := \"some updated history archival uri\"\n\tupdateVisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tupdateVisibilityArchivalURI := \"some updated visibility archival uri\"\n\tupdateClusterActive := \"other random active cluster name\"\n\tupdateClusterStandby := \"other random standby cluster name\"\n\tupdateConfigVersion := configVersion + 1\n\tupdateFailoverVersion := failoverVersion - 1\n\tupdateClusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: updateClusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: updateClusterStandby,\n\t\t},\n\t}\n\tupdateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &updateOperation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &updateStatus,\n\t\t\tDescription: updateDescription,\n\t\t\tOwnerEmail:  updateOwnerEmail,\n\t\t\tData:        updateData,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: updateRetention,\n\t\t\tEmitMetric:                             updateEmitMetric,\n\t\t\tHistoryArchivalStatus:                  updateHistoryArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     updateHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus:               updateVisibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  updateVisibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tConfigVersion:           updateConfigVersion,\n\t\tFailoverVersion:         updateFailoverVersion,\n\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t}\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(updateTask)\n\ts.Nil(err)\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusDeprecated, resp.Info.Status)\n\ts.Equal(updateDescription, resp.Info.Description)\n\ts.Equal(updateOwnerEmail, resp.Info.OwnerEmail)\n\ts.Equal(updateData, resp.Info.Data)\n\ts.Equal(updateRetention, resp.Config.Retention)\n\ts.Equal(updateEmitMetric, resp.Config.EmitMetric)\n\ts.Equal(updateHistoryArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(updateHistoryArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(updateVisibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(updateVisibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(clusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(updateClusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(updateConfigVersion, resp.ConfigVersion)\n\ts.Equal(failoverVersion, resp.FailoverVersion)\n\ts.Equal(constants.InitialPreviousFailoverVersion, resp.PreviousFailoverVersion)\n\ts.Equal(int64(0), resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_UpdateDomainTask_NoUpdateConfig_UpdateActiveCluster() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tcreateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:           configVersion,\n\t\tFailoverVersion:         failoverVersion,\n\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t}\n\n\terr := s.domainReplicator.Execute(createTask)\n\ts.Nil(err)\n\tresp1, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp1)\n\ts.Equal(id, resp1.Info.ID)\n\ts.Equal(name, resp1.Info.Name)\n\ts.Equal(persistence.DomainStatusRegistered, resp1.Info.Status)\n\ts.Equal(description, resp1.Info.Description)\n\ts.Equal(ownerEmail, resp1.Info.OwnerEmail)\n\ts.Equal(data, resp1.Info.Data)\n\ts.Equal(retention, resp1.Config.Retention)\n\ts.Equal(emitMetric, resp1.Config.EmitMetric)\n\ts.Equal(historyArchivalStatus, resp1.Config.HistoryArchivalStatus)\n\ts.Equal(historyArchivalURI, resp1.Config.HistoryArchivalURI)\n\ts.Equal(visibilityArchivalStatus, resp1.Config.VisibilityArchivalStatus)\n\ts.Equal(visibilityArchivalURI, resp1.Config.VisibilityArchivalURI)\n\ts.Equal(clusterActive, resp1.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(clusters), resp1.ReplicationConfig.Clusters)\n\ts.Equal(configVersion, resp1.ConfigVersion)\n\ts.Equal(failoverVersion, resp1.FailoverVersion)\n\ts.Equal(constants.InitialPreviousFailoverVersion, resp1.PreviousFailoverVersion)\n\n\t// success update case\n\tupdateOperation := types.DomainOperationUpdate\n\tupdateStatus := types.DomainStatusDeprecated\n\tupdateDescription := \"other random domain test description\"\n\tupdateOwnerEmail := \"other random domain test owner\"\n\tupdatedData := map[string]string{\"k\": \"v2\"}\n\tupdateRetention := int32(122)\n\tupdateEmitMetric := true\n\tupdateClusterActive := \"other random active cluster name\"\n\tupdateClusterStandby := \"other random standby cluster name\"\n\tupdateConfigVersion := configVersion - 1\n\tupdateFailoverVersion := failoverVersion + 1\n\tupdateClusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: updateClusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: updateClusterStandby,\n\t\t},\n\t}\n\tupdateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &updateOperation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &updateStatus,\n\t\t\tDescription: updateDescription,\n\t\t\tOwnerEmail:  updateOwnerEmail,\n\t\t\tData:        updatedData,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: updateRetention,\n\t\t\tEmitMetric:                             updateEmitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tConfigVersion:           updateConfigVersion,\n\t\tFailoverVersion:         updateFailoverVersion,\n\t\tPreviousFailoverVersion: failoverVersion,\n\t}\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(updateTask)\n\ts.Nil(err)\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusRegistered, resp.Info.Status)\n\ts.Equal(description, resp.Info.Description)\n\ts.Equal(ownerEmail, resp.Info.OwnerEmail)\n\ts.Equal(data, resp.Info.Data)\n\ts.Equal(retention, resp.Config.Retention)\n\ts.Equal(emitMetric, resp.Config.EmitMetric)\n\ts.Equal(historyArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(historyArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(visibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(visibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(updateClusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(clusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(configVersion, resp.ConfigVersion)\n\ts.Equal(updateFailoverVersion, resp.FailoverVersion)\n\ts.Equal(notificationVersion, resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n\ts.Equal(failoverVersion, resp.PreviousFailoverVersion)\n}\n\nfunc (s *domainReplicationTaskExecutorSuite) TestExecute_UpdateDomainTask_NoUpdateConfig_NoUpdateActiveCluster() {\n\toperation := types.DomainOperationCreate\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tcreateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &operation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  ownerEmail,\n\t\t\tData:        data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\tEmitMetric:                             emitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tConfigVersion:   configVersion,\n\t\tFailoverVersion: failoverVersion,\n\t}\n\tmetadata, err := s.DomainManager.GetMetadata(context.Background())\n\ts.Nil(err)\n\tnotificationVersion := metadata.NotificationVersion\n\terr = s.domainReplicator.Execute(createTask)\n\ts.Nil(err)\n\n\t// success update case\n\tupdateOperation := types.DomainOperationUpdate\n\tupdateStatus := types.DomainStatusDeprecated\n\tupdateDescription := \"other random domain test description\"\n\tupdateOwnerEmail := \"other random domain test owner\"\n\tupdatedData := map[string]string{\"k\": \"v2\"}\n\tupdateRetention := int32(122)\n\tupdateEmitMetric := true\n\tupdateClusterActive := \"other random active cluster name\"\n\tupdateClusterStandby := \"other random standby cluster name\"\n\tupdateConfigVersion := configVersion - 1\n\tupdateFailoverVersion := failoverVersion - 1\n\tupdateClusters := []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: updateClusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: updateClusterStandby,\n\t\t},\n\t}\n\tupdateTask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &updateOperation,\n\t\tID:              id,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        name,\n\t\t\tStatus:      &updateStatus,\n\t\t\tDescription: updateDescription,\n\t\t\tOwnerEmail:  updateOwnerEmail,\n\t\t\tData:        updatedData,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: updateRetention,\n\t\t\tEmitMetric:                             updateEmitMetric,\n\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tConfigVersion:           updateConfigVersion,\n\t\tFailoverVersion:         updateFailoverVersion,\n\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t}\n\terr = s.domainReplicator.Execute(updateTask)\n\ts.Nil(err)\n\tresp, err := s.DomainManager.GetDomain(context.Background(), &persistence.GetDomainRequest{Name: name})\n\ts.Nil(err)\n\ts.NotNil(resp)\n\ts.Equal(id, resp.Info.ID)\n\ts.Equal(name, resp.Info.Name)\n\ts.Equal(persistence.DomainStatusRegistered, resp.Info.Status)\n\ts.Equal(description, resp.Info.Description)\n\ts.Equal(ownerEmail, resp.Info.OwnerEmail)\n\ts.Equal(data, resp.Info.Data)\n\ts.Equal(retention, resp.Config.Retention)\n\ts.Equal(emitMetric, resp.Config.EmitMetric)\n\ts.Equal(historyArchivalStatus, resp.Config.HistoryArchivalStatus)\n\ts.Equal(historyArchivalURI, resp.Config.HistoryArchivalURI)\n\ts.Equal(visibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\ts.Equal(visibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\ts.Equal(clusterActive, resp.ReplicationConfig.ActiveClusterName)\n\ts.Equal(s.domainReplicator.convertClusterReplicationConfigFromThrift(clusters), resp.ReplicationConfig.Clusters)\n\ts.Equal(configVersion, resp.ConfigVersion)\n\ts.Equal(failoverVersion, resp.FailoverVersion)\n\ts.Equal(constants.InitialPreviousFailoverVersion, resp.PreviousFailoverVersion)\n\ts.Equal(int64(0), resp.FailoverNotificationVersion)\n\ts.Equal(notificationVersion, resp.NotificationVersion)\n}\n"
  },
  {
    "path": "common/domain/replicationTaskExecutor_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDomainReplicationTaskExecutor_Execute(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tsetupMock func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager)\n\t\ttask      *types.DomainTaskAttributes\n\t\twantErr   bool\n\t\terrType   interface{}\n\t}{\n\t\t{\n\t\t\tname: \"Validate Domain Task - Empty Task\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\t// No setup required as the task itself is nil, triggering the validation error\n\t\t\t},\n\t\t\ttask:    nil,\n\t\t\twantErr: true,\n\t\t\terrType: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Create Domain Task - Valid\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainResponse{ID: \"validDomainID\"}, nil).\n\t\t\t\t\tTimes(1)\n\n\t\t\t\t// Expect GetDomain for audit log\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{ID: \"validDomainID\"}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tID:   \"validDomainID\",\n\t\t\t\t\t\t\tName: \"validDomain\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).\n\t\t\t\t\tTimes(1)\n\n\t\t\t\t// Expect audit log creation\n\t\t\t\tmockAuditManager.EXPECT().\n\t\t\t\t\tCreateDomainAuditLog(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainAuditLogResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperationCreate.Ptr(),\n\t\t\t\tID:              \"validDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"validDomain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"A valid domain\",\n\t\t\t\t\tOwnerEmail:  \"owner@example.com\",\n\t\t\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"test://visibility\",\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"activeClusterName\"},\n\t\t\t\t\t\t{ClusterName: \"standbyClusterName\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\"region2\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"standbyClusterName\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tConfigVersion:   1,\n\t\t\t\tFailoverVersion: 1,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Create Domain Task - Name UUID Collision\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\t// call to GetDomain simulates a name collision by returning a different domain ID\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{Name: \"collisionDomain\"}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{Info: &persistence.DomainInfo{ID: uuid.New()}}, nil).\n\t\t\t\t\tTimes(1)\n\n\t\t\t\t// Expect CreateDomain to be called, which should result in a collision error\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrNameUUIDCollision).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperationCreate.Ptr(),\n\t\t\t\tID:              uuid.New(),\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"collisionDomain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"A domain with UUID collision\",\n\t\t\t\t\tOwnerEmail:  \"owner@example.com\",\n\t\t\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"test://visibility\",\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"activeClusterName\"},\n\t\t\t\t\t\t{ClusterName: \"standbyClusterName\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tConfigVersion:   1,\n\t\t\t\tFailoverVersion: 1,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\terrType: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Update Domain Task - Valid Update\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetMetadataResponse{NotificationVersion: 123}, nil).\n\t\t\t\t\tTimes(1)\n\n\t\t\t\t// Mock GetDomain to simulate domain fetch before update (using ID)\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{ID: \"existingDomainID\"}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"existingDomainID\", Name: \"existingDomainName\"},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\tFailoverVersion:   50,\n\t\t\t\t\t}, nil).Times(1)\n\n\t\t\t\t// Mock UpdateDomain to simulate a successful domain update\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tUpdateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\n\t\t\t\t// Mock GetDomain after update for audit log\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{ID: \"existingDomainID\"}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"existingDomainID\", Name: \"existingDomainName\"},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t\tConfigVersion:     2,\n\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t}, nil).Times(1)\n\n\t\t\t\t// Expect audit log creation\n\t\t\t\tmockAuditManager.EXPECT().\n\t\t\t\t\tCreateDomainAuditLog(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainAuditLogResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperationUpdate.Ptr(),\n\t\t\t\tID:              \"existingDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"existingDomainName\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"Updated description\",\n\t\t\t\t\tOwnerEmail:  \"updatedOwner@example.com\",\n\t\t\t\t\tData:        map[string]string{\"updatedKey\": \"updatedValue\"},\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{},\n\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"activeClusterName\"},\n\t\t\t\t\t\t{ClusterName: \"standbyClusterName\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tConfigVersion:   2,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Invalid Domain Operation\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\t// No mock setup is required as the operation should not proceed to any database calls\n\t\t\t},\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\t// Set up a task without a valid DomainOperation or with an unrecognized operation\n\t\t\t\tDomainOperation: nil, // Assuming this would not match any specific case\n\t\t\t\tID:              \"invalidOperationDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"invalidOperationDomain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"Domain with invalid operation\",\n\t\t\t\t\tOwnerEmail:  \"owner@example.com\",\n\t\t\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\t\t},\n\t\t\t\tConfig:            &types.DomainConfiguration{},\n\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\terrType: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Unsupported Domain Operation\",\n\t\t\tsetupMock: func(mockDomainManager persistence.MockDomainManager, mockAuditManager persistence.MockDomainAuditManager) {\n\t\t\t\t// No mock setup is needed as the operation should immediately return an error\n\t\t\t},\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperation(999).Ptr(), // Assuming 999 is not a valid operation\n\t\t\t\tID:              \"someDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"someDomain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"A domain with an unsupported operation\",\n\t\t\t\t\tOwnerEmail:  \"email@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"uri://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"uri://visibility\",\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"activeCluster\",\n\t\t\t\t\tClusters:          []*types.ClusterReplicationConfiguration{{ClusterName: \"activeCluster\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\terrType: &types.BadRequestError{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(ctrl)\n\t\t\tmockTimeSource := clock.NewRealTimeSource()\n\t\t\tmockLogger := log.NewNoop()\n\t\t\tenableAuditLogging := func(...dynamicproperties.FilterOption) bool { return true }\n\n\t\t\texecutor := NewReplicationTaskExecutor(mockDomainManager, mockDomainAuditManager, mockTimeSource, mockLogger, enableAuditLogging).(*domainReplicationTaskExecutorImpl)\n\t\t\ttt.setupMock(*mockDomainManager, *mockDomainAuditManager)\n\t\t\terr := executor.Execute(tt.task)\n\t\t\tif tt.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.IsType(t, tt.errType, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc domainCreationTask() *types.DomainTaskAttributes {\n\treturn &types.DomainTaskAttributes{\n\t\tDomainOperation: types.DomainOperationCreate.Ptr(),\n\t\tID:              \"testDomainID\",\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        \"testDomain\",\n\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\tDescription: \"This is a test domain\",\n\t\t\tOwnerEmail:  \"owner@test.com\",\n\t\t\tData:        map[string]string{\"key1\": \"value1\"}, // Arbitrary domain metadata\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: 10,\n\t\t\tEmitMetric:                             true,\n\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\tHistoryArchivalURI:                     \"test://history/archival\",\n\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusEnabled.Ptr(),\n\t\t\tVisibilityArchivalURI:                  \"test://visibility/archival\",\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: \"activeClusterName\",\n\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t{\n\t\t\t\t\tClusterName: \"activeClusterName\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tClusterName: \"standbyClusterName\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tConfigVersion:           1,\n\t\tFailoverVersion:         1,\n\t\tPreviousFailoverVersion: 0,\n\t}\n}\n\nfunc TestHandleDomainCreationReplicationTask(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\ttask      *types.DomainTaskAttributes\n\t\tsetup     func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager)\n\t\twantError bool\n\t}{\n\t\t{\n\t\t\tname: \"Successful Domain Creation\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainResponse{ID: \"testDomainID\"}, nil)\n\n\t\t\t\t// Expect GetDomain for audit log\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{ID: \"testDomainID\"}).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tID:   \"testDomainID\",\n\t\t\t\t\t\t\tName: \"testDomain\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\n\t\t\t\t// Expect audit log creation\n\t\t\t\tmockAuditManager.EXPECT().\n\t\t\t\t\tCreateDomainAuditLog(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainAuditLogResponse{}, nil)\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Generic Error During Domain Creation\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, types.InternalServiceError{Message: \"an internal error\"}).\n\t\t\t\t\tTimes(1)\n\n\t\t\t\t// Since CreateDomain failed, handleDomainCreationReplicationTask check for domain existence by name and ID\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}). // Simulate that no domain exists with the given name/ID\n\t\t\t\t\tAnyTimes()\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Name/UUID Collision - EntityNotExistsError\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrNameUUIDCollision).Times(1)\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{}).AnyTimes()\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Immediate Error Return from CreateDomain\",\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, types.InternalServiceError{Message: \"internal error\"}).\n\t\t\t\t\tTimes(1)\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrInvalidDomainStatus).\n\t\t\t\t\tAnyTimes()\n\t\t\t},\n\t\t\ttask:      domainCreationTask(),\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain Creation with Nil Status\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperationCreate.Ptr(),\n\t\t\t\tID:              \"testDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"testDomain\",\n\t\t\t\t\t// Status is intentionally left as nil to trigger the error\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\t// No need to set up a mock for CreateDomain as the call should not reach this point\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain Creation with Unrecognized Status\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tDomainOperation: types.DomainOperationCreate.Ptr(),\n\t\t\t\tID:              \"testDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:   \"testDomain\",\n\t\t\t\t\tStatus: types.DomainStatus(999).Ptr(), // Assuming 999 is an unrecognized status\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\t// As before, no need for mock setup for CreateDomain\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Unexpected Error Type from GetDomain Leads to Default Error Handling\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrInvalidDomainStatus).Times(1)\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"unexpected error\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Successful GetDomain with Name/UUID Mismatch\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrNameUUIDCollision).AnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{ID: \"testDomainID\", Name: \"mismatchName\"},\n\t\t\t\t\t}, nil).AnyTimes()\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Domain Creation with Unhandled Error\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"unhandled error\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Handle Domain Creation - Unexpected Error from GetDomain\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"test error\")).Times(1)\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}).Times(1)\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"unexpected error\")).Times(1)\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Duplicate Domain Creation With Same ID and Name\",\n\t\t\ttask: domainCreationTask(),\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, ErrNameUUIDCollision).Times(1)\n\n\t\t\t\t// Setup GetDomain to return matching ID and Name, indicating no actual conflict\n\t\t\t\t// This setup ensures that recordExists becomes true\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{ID: \"testDomainID\", Name: \"testDomain\"},\n\t\t\t\t\t}, nil).Times(2)\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)) // Fixed time\n\t\t\tenableAuditLogging := func(...dynamicproperties.FilterOption) bool { return true }\n\n\t\t\texecutor := &domainReplicationTaskExecutorImpl{\n\t\t\t\tdomainManager:            mockDomainManager,\n\t\t\t\tdomainAuditManager:       mockDomainAuditManager,\n\t\t\t\tlogger:                   mockLogger,\n\t\t\t\ttimeSource:               mockTimeSource,\n\t\t\t\tenableDomainAuditLogging: enableAuditLogging,\n\t\t\t}\n\n\t\t\ttt.setup(mockDomainManager, mockDomainAuditManager)\n\t\t\terr := executor.handleDomainCreationReplicationTask(context.Background(), tt.task)\n\t\t\tif tt.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandleDomainUpdateReplicationTask(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\ttask    *types.DomainTaskAttributes\n\t\twantErr bool\n\t\tsetup   func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager)\n\t}{\n\t\t{\n\t\t\tname: \"Convert Status Error\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tStatus: func() *types.DomainStatus { var ds types.DomainStatus = 999; return &ds }(), // Invalid status to trigger conversion error\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tsetup:   func(dm *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {},\n\t\t},\n\t\t{\n\t\t\tname:    \"Error Fetching Metadata\",\n\t\t\ttask:    domainCreationTask(),\n\t\t\twantErr: true,\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"Error getting metadata while handling replication task\")).\n\t\t\t\t\tAnyTimes()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetDomain Returns General Error\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"someDomain\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"general error\")).AnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetMetadataResponse{}, nil).AnyTimes()\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetDomain Returns EntityNotExistsError - Triggers Domain Creation\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tID: \"nonexistentDomainID\",\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"nonexistentDomain\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &persistence.GetDomainRequest{\n\t\t\t\t\t\tID: \"nonexistentDomainID\",\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}).AnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CreateDomainResponse{\n\t\t\t\t\t\tID: \"nonexistentDomainID\",\n\t\t\t\t\t}, nil).\n\t\t\t\t\tAnyTimes()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Record Not Updated then return nil\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName:   \"testDomain\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\tBadBinaries: &types.BadBinaries{\n\t\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\t\t\"checksum1\": {\n\t\t\t\t\t\t\t\tReason:          \"reasontest\",\n\t\t\t\t\t\t\t\tOperator:        \"operatortest\",\n\t\t\t\t\t\t\t\tCreatedTimeNano: func() *int64 { var ct int64 = 12345; return &ct }(),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil).AnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{\n\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"testDomainID\", Name: \"testDomain\"},\n\t\t\t\t\t\tConfig:            &persistence.DomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\t\t\t\t}, nil).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tUpdateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Update Domain with BadBinaries Set\",\n\t\t\ttask: &types.DomainTaskAttributes{\n\t\t\t\tInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"existingDomainName\",\n\t\t\t\t},\n\t\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\t\tBadBinaries: &types.BadBinaries{\n\t\t\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tsetup: func(mockDomainManager *persistence.MockDomainManager, mockAuditManager *persistence.MockDomainAuditManager) {\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetMetadataResponse{NotificationVersion: 1}, nil).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetDomainResponse{}, nil).\n\t\t\t\t\tAnyTimes()\n\n\t\t\t\tmockDomainManager.EXPECT().\n\t\t\t\t\tUpdateDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).\n\t\t\t\t\tAnyTimes()\n\t\t\t},\n\t\t},\n\t}\n\tassert := assert.New(t)\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDomainManager := persistence.NewMockDomainManager(mockCtrl)\n\t\t\tmockDomainAuditManager := persistence.NewMockDomainAuditManager(mockCtrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)) // Fixed time\n\t\t\tenableAuditLogging := func(...dynamicproperties.FilterOption) bool { return true }\n\n\t\t\texecutor := &domainReplicationTaskExecutorImpl{\n\t\t\t\tdomainManager:            mockDomainManager,\n\t\t\t\tdomainAuditManager:       mockDomainAuditManager,\n\t\t\t\tlogger:                   mockLogger,\n\t\t\t\ttimeSource:               mockTimeSource,\n\t\t\t\tenableDomainAuditLogging: enableAuditLogging,\n\t\t\t}\n\t\t\ttt.setup(mockDomainManager, mockDomainAuditManager)\n\n\t\t\terr := executor.handleDomainUpdateReplicationTask(context.Background(), tt.task)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(err, \"Expected an error for test case: %s\", tt.name)\n\t\t\t} else {\n\t\t\t\tassert.NoError(err, \"Expected no error for test case: %s\", tt.name)\n\t\t\t}\n\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/replicationTaskHandler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: replicationTaskExecutor.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source replicationTaskExecutor.go -destination replicationTaskHandler_mock.go\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockReplicationTaskExecutor is a mock of ReplicationTaskExecutor interface.\ntype MockReplicationTaskExecutor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReplicationTaskExecutorMockRecorder\n\tisgomock struct{}\n}\n\n// MockReplicationTaskExecutorMockRecorder is the mock recorder for MockReplicationTaskExecutor.\ntype MockReplicationTaskExecutorMockRecorder struct {\n\tmock *MockReplicationTaskExecutor\n}\n\n// NewMockReplicationTaskExecutor creates a new mock instance.\nfunc NewMockReplicationTaskExecutor(ctrl *gomock.Controller) *MockReplicationTaskExecutor {\n\tmock := &MockReplicationTaskExecutor{ctrl: ctrl}\n\tmock.recorder = &MockReplicationTaskExecutorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockReplicationTaskExecutor) EXPECT() *MockReplicationTaskExecutorMockRecorder {\n\treturn m.recorder\n}\n\n// Execute mocks base method.\nfunc (m *MockReplicationTaskExecutor) Execute(task *types.DomainTaskAttributes) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\", task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockReplicationTaskExecutorMockRecorder) Execute(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockReplicationTaskExecutor)(nil).Execute), task)\n}\n"
  },
  {
    "path": "common/domain/replication_queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination replication_queue_mock.go -self_package github.com/uber/common/persistence\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nconst (\n\tpurgeInterval                 = 5 * time.Minute\n\tqueueSizeQueryInterval        = 5 * time.Minute\n\tlocalDomainReplicationCluster = \"domainReplication\"\n)\n\nvar _ ReplicationQueue = (*replicationQueueImpl)(nil)\n\n// NewReplicationQueue creates a new ReplicationQueue instance\nfunc NewReplicationQueue(\n\tqueue persistence.QueueManager,\n\tclusterName string,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) ReplicationQueue {\n\treturn &replicationQueueImpl{\n\t\tqueue:         queue,\n\t\tclusterName:   clusterName,\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t\tencoder:       codec.NewThriftRWEncoder(),\n\t\tdone:          make(chan bool),\n\t\tstatus:        common.DaemonStatusInitialized,\n\t}\n}\n\ntype (\n\treplicationQueueImpl struct {\n\t\tqueue         persistence.QueueManager\n\t\tclusterName   string\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t\tencoder       codec.BinaryEncoder\n\t\tdone          chan bool\n\t\tstatus        int32\n\t}\n\n\t// ReplicationQueue is used to publish and list domain replication tasks\n\tReplicationQueue interface {\n\t\tcommon.Daemon\n\t\tPublish(ctx context.Context, message interface{}) error\n\t\tPublishToDLQ(ctx context.Context, message interface{}) error\n\t\tGetReplicationMessages(ctx context.Context, lastMessageID int64, maxCount int) ([]*types.ReplicationTask, int64, error)\n\t\tUpdateAckLevel(ctx context.Context, lastProcessedMessageID int64, clusterName string) error\n\t\tGetAckLevels(ctx context.Context) (map[string]int64, error)\n\t\tGetMessagesFromDLQ(ctx context.Context, firstMessageID int64, lastMessageID int64, pageSize int, pageToken []byte) ([]*types.ReplicationTask, []byte, error)\n\t\tUpdateDLQAckLevel(ctx context.Context, lastProcessedMessageID int64) error\n\t\tGetDLQAckLevel(ctx context.Context) (int64, error)\n\t\tRangeDeleteMessagesFromDLQ(ctx context.Context, firstMessageID int64, lastMessageID int64) error\n\t\tDeleteMessageFromDLQ(ctx context.Context, messageID int64) error\n\t\tGetDLQSize(ctx context.Context) (int64, error)\n\t}\n)\n\nfunc (q *replicationQueueImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tgo q.purgeProcessor()\n}\n\nfunc (q *replicationQueueImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(q.done)\n}\n\nfunc (q *replicationQueueImpl) Publish(\n\tctx context.Context,\n\tmessage interface{},\n) error {\n\ttask, ok := message.(*types.ReplicationTask)\n\tif !ok {\n\t\treturn errors.New(\"wrong message type\")\n\t}\n\n\tbytes, err := q.encoder.Encode(thrift.FromReplicationTask(task))\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to encode message: %v\", err)\n\t}\n\treturn q.queue.EnqueueMessage(ctx, &persistence.EnqueueMessageRequest{\n\t\tMessagePayload: bytes,\n\t})\n}\n\nfunc (q *replicationQueueImpl) PublishToDLQ(\n\tctx context.Context,\n\tmessage interface{},\n) error {\n\ttask, ok := message.(*types.ReplicationTask)\n\tif !ok {\n\t\treturn errors.New(\"wrong message type\")\n\t}\n\n\tbytes, err := q.encoder.Encode(thrift.FromReplicationTask(task))\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to encode message: %v\", err)\n\t}\n\n\treturn q.queue.EnqueueMessageToDLQ(ctx, &persistence.EnqueueMessageToDLQRequest{\n\t\tMessagePayload: bytes,\n\t})\n}\n\nfunc (q *replicationQueueImpl) GetReplicationMessages(\n\tctx context.Context,\n\tlastMessageID int64,\n\tmaxCount int,\n) ([]*types.ReplicationTask, int64, error) {\n\n\tresp, err := q.queue.ReadMessages(ctx, &persistence.ReadMessagesRequest{\n\t\tLastMessageID: lastMessageID,\n\t\tMaxCount:      maxCount,\n\t})\n\tif err != nil {\n\t\treturn nil, lastMessageID, err\n\t}\n\n\tvar replicationTasks []*types.ReplicationTask\n\tfor _, message := range resp.Messages {\n\t\tvar replicationTask replicator.ReplicationTask\n\t\terr := q.encoder.Decode(message.Payload, &replicationTask)\n\t\tif err != nil {\n\t\t\treturn nil, lastMessageID, fmt.Errorf(\"failed to decode task: %v\", err)\n\t\t}\n\n\t\tlastMessageID = message.ID\n\t\treplicationTasks = append(replicationTasks, thrift.ToReplicationTask(&replicationTask))\n\t}\n\n\tq.logger.Debugf(\"Returning %d domain replication tasks. lastMessageID: %d\", len(replicationTasks), lastMessageID)\n\treturn replicationTasks, lastMessageID, nil\n}\n\nfunc (q *replicationQueueImpl) UpdateAckLevel(\n\tctx context.Context,\n\tlastProcessedMessageID int64,\n\tclusterName string,\n) error {\n\n\tif err := q.queue.UpdateAckLevel(ctx, &persistence.UpdateAckLevelRequest{\n\t\tMessageID:   lastProcessedMessageID,\n\t\tClusterName: clusterName,\n\t}); err != nil {\n\t\treturn fmt.Errorf(\"failed to update ack level: %v\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (q *replicationQueueImpl) GetAckLevels(\n\tctx context.Context,\n) (map[string]int64, error) {\n\tresp, err := q.queue.GetAckLevels(ctx, &persistence.GetAckLevelsRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp.AckLevels, nil\n}\n\nfunc (q *replicationQueueImpl) GetMessagesFromDLQ(\n\tctx context.Context,\n\tfirstMessageID int64,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*types.ReplicationTask, []byte, error) {\n\n\tresp, err := q.queue.ReadMessagesFromDLQ(ctx, &persistence.ReadMessagesFromDLQRequest{\n\t\tFirstMessageID: firstMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t\tPageSize:       pageSize,\n\t\tPageToken:      pageToken,\n\t})\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tvar replicationTasks []*types.ReplicationTask\n\tfor _, message := range resp.Messages {\n\t\tvar replicationTask replicator.ReplicationTask\n\t\terr := q.encoder.Decode(message.Payload, &replicationTask)\n\t\tif err != nil {\n\t\t\treturn nil, nil, fmt.Errorf(\"failed to decode dlq task: %v\", err)\n\t\t}\n\n\t\t// Overwrite to local cluster message id\n\t\treplicationTask.SourceTaskId = common.Int64Ptr(int64(message.ID))\n\t\treplicationTasks = append(replicationTasks, thrift.ToReplicationTask(&replicationTask))\n\t}\n\n\treturn replicationTasks, resp.NextPageToken, nil\n}\n\nfunc (q *replicationQueueImpl) UpdateDLQAckLevel(\n\tctx context.Context,\n\tlastProcessedMessageID int64,\n) error {\n\treturn q.queue.UpdateDLQAckLevel(ctx, &persistence.UpdateDLQAckLevelRequest{\n\t\tMessageID:   lastProcessedMessageID,\n\t\tClusterName: localDomainReplicationCluster,\n\t})\n}\n\nfunc (q *replicationQueueImpl) GetDLQAckLevel(\n\tctx context.Context,\n) (int64, error) {\n\tdlqMetadataResp, err := q.queue.GetDLQAckLevels(ctx, &persistence.GetDLQAckLevelsRequest{})\n\tif err != nil {\n\t\treturn constants.EmptyMessageID, err\n\t}\n\n\tackLevel, ok := dlqMetadataResp.AckLevels[localDomainReplicationCluster]\n\tif !ok {\n\t\treturn constants.EmptyMessageID, nil\n\t}\n\treturn ackLevel, nil\n}\n\nfunc (q *replicationQueueImpl) RangeDeleteMessagesFromDLQ(\n\tctx context.Context,\n\tfirstMessageID int64,\n\tlastMessageID int64,\n) error {\n\treturn q.queue.RangeDeleteMessagesFromDLQ(ctx, &persistence.RangeDeleteMessagesFromDLQRequest{\n\t\tFirstMessageID: firstMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t})\n}\n\nfunc (q *replicationQueueImpl) DeleteMessageFromDLQ(\n\tctx context.Context,\n\tmessageID int64,\n) error {\n\n\treturn q.queue.DeleteMessageFromDLQ(ctx, &persistence.DeleteMessageFromDLQRequest{\n\t\tMessageID: messageID,\n\t})\n}\n\nfunc (q *replicationQueueImpl) GetDLQSize(ctx context.Context) (int64, error) {\n\tresp, err := q.queue.GetDLQSize(ctx, &persistence.GetDLQSizeRequest{})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn resp.Size, nil\n}\n\nfunc (q *replicationQueueImpl) purgeAckedMessages() error {\n\tackLevelByCluster, err := q.GetAckLevels(context.Background())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to purge messages: %v\", err)\n\t}\n\n\tif len(ackLevelByCluster) == 0 {\n\t\treturn nil\n\t}\n\n\tminAckLevel := int64(math.MaxInt64)\n\tfor _, ackLevel := range ackLevelByCluster {\n\t\tif ackLevel < minAckLevel {\n\t\t\tminAckLevel = ackLevel\n\t\t}\n\t}\n\n\tif minAckLevel != int64(math.MaxInt64) {\n\t\tif err = q.queue.DeleteMessagesBefore(context.Background(), &persistence.DeleteMessagesBeforeRequest{\n\t\t\tMessageID: minAckLevel,\n\t\t}); err != nil {\n\t\t\treturn fmt.Errorf(\"failed to purge messages: %v\", err)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (q *replicationQueueImpl) purgeProcessor() {\n\tticker := time.NewTicker(purgeInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-q.done:\n\t\t\treturn\n\t\tcase <-ticker.C:\n\t\t\tif err := q.purgeAckedMessages(); err != nil {\n\t\t\t\tq.logger.Warn(\"Failed to purge acked domain replication messages.\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/domain/replication_queue_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: replication_queue.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source replication_queue.go -destination replication_queue_mock.go -self_package github.com/uber/common/persistence\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockReplicationQueue is a mock of ReplicationQueue interface.\ntype MockReplicationQueue struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReplicationQueueMockRecorder\n\tisgomock struct{}\n}\n\n// MockReplicationQueueMockRecorder is the mock recorder for MockReplicationQueue.\ntype MockReplicationQueueMockRecorder struct {\n\tmock *MockReplicationQueue\n}\n\n// NewMockReplicationQueue creates a new mock instance.\nfunc NewMockReplicationQueue(ctrl *gomock.Controller) *MockReplicationQueue {\n\tmock := &MockReplicationQueue{ctrl: ctrl}\n\tmock.recorder = &MockReplicationQueueMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockReplicationQueue) EXPECT() *MockReplicationQueueMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteMessageFromDLQ mocks base method.\nfunc (m *MockReplicationQueue) DeleteMessageFromDLQ(ctx context.Context, messageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessageFromDLQ\", ctx, messageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessageFromDLQ indicates an expected call of DeleteMessageFromDLQ.\nfunc (mr *MockReplicationQueueMockRecorder) DeleteMessageFromDLQ(ctx, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessageFromDLQ\", reflect.TypeOf((*MockReplicationQueue)(nil).DeleteMessageFromDLQ), ctx, messageID)\n}\n\n// GetAckLevels mocks base method.\nfunc (m *MockReplicationQueue) GetAckLevels(ctx context.Context) (map[string]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevels\", ctx)\n\tret0, _ := ret[0].(map[string]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAckLevels indicates an expected call of GetAckLevels.\nfunc (mr *MockReplicationQueueMockRecorder) GetAckLevels(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevels\", reflect.TypeOf((*MockReplicationQueue)(nil).GetAckLevels), ctx)\n}\n\n// GetDLQAckLevel mocks base method.\nfunc (m *MockReplicationQueue) GetDLQAckLevel(ctx context.Context) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQAckLevel\", ctx)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQAckLevel indicates an expected call of GetDLQAckLevel.\nfunc (mr *MockReplicationQueueMockRecorder) GetDLQAckLevel(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQAckLevel\", reflect.TypeOf((*MockReplicationQueue)(nil).GetDLQAckLevel), ctx)\n}\n\n// GetDLQSize mocks base method.\nfunc (m *MockReplicationQueue) GetDLQSize(ctx context.Context) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQSize\", ctx)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQSize indicates an expected call of GetDLQSize.\nfunc (mr *MockReplicationQueueMockRecorder) GetDLQSize(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQSize\", reflect.TypeOf((*MockReplicationQueue)(nil).GetDLQSize), ctx)\n}\n\n// GetMessagesFromDLQ mocks base method.\nfunc (m *MockReplicationQueue) GetMessagesFromDLQ(ctx context.Context, firstMessageID, lastMessageID int64, pageSize int, pageToken []byte) ([]*types.ReplicationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesFromDLQ\", ctx, firstMessageID, lastMessageID, pageSize, pageToken)\n\tret0, _ := ret[0].([]*types.ReplicationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetMessagesFromDLQ indicates an expected call of GetMessagesFromDLQ.\nfunc (mr *MockReplicationQueueMockRecorder) GetMessagesFromDLQ(ctx, firstMessageID, lastMessageID, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesFromDLQ\", reflect.TypeOf((*MockReplicationQueue)(nil).GetMessagesFromDLQ), ctx, firstMessageID, lastMessageID, pageSize, pageToken)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockReplicationQueue) GetReplicationMessages(ctx context.Context, lastMessageID int64, maxCount int) ([]*types.ReplicationTask, int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", ctx, lastMessageID, maxCount)\n\tret0, _ := ret[0].([]*types.ReplicationTask)\n\tret1, _ := ret[1].(int64)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockReplicationQueueMockRecorder) GetReplicationMessages(ctx, lastMessageID, maxCount any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockReplicationQueue)(nil).GetReplicationMessages), ctx, lastMessageID, maxCount)\n}\n\n// Publish mocks base method.\nfunc (m *MockReplicationQueue) Publish(ctx context.Context, message any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Publish\", ctx, message)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Publish indicates an expected call of Publish.\nfunc (mr *MockReplicationQueueMockRecorder) Publish(ctx, message any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Publish\", reflect.TypeOf((*MockReplicationQueue)(nil).Publish), ctx, message)\n}\n\n// PublishToDLQ mocks base method.\nfunc (m *MockReplicationQueue) PublishToDLQ(ctx context.Context, message any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PublishToDLQ\", ctx, message)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PublishToDLQ indicates an expected call of PublishToDLQ.\nfunc (mr *MockReplicationQueueMockRecorder) PublishToDLQ(ctx, message any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PublishToDLQ\", reflect.TypeOf((*MockReplicationQueue)(nil).PublishToDLQ), ctx, message)\n}\n\n// RangeDeleteMessagesFromDLQ mocks base method.\nfunc (m *MockReplicationQueue) RangeDeleteMessagesFromDLQ(ctx context.Context, firstMessageID, lastMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessagesFromDLQ\", ctx, firstMessageID, lastMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteMessagesFromDLQ indicates an expected call of RangeDeleteMessagesFromDLQ.\nfunc (mr *MockReplicationQueueMockRecorder) RangeDeleteMessagesFromDLQ(ctx, firstMessageID, lastMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessagesFromDLQ\", reflect.TypeOf((*MockReplicationQueue)(nil).RangeDeleteMessagesFromDLQ), ctx, firstMessageID, lastMessageID)\n}\n\n// Start mocks base method.\nfunc (m *MockReplicationQueue) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockReplicationQueueMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockReplicationQueue)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockReplicationQueue) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockReplicationQueueMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockReplicationQueue)(nil).Stop))\n}\n\n// UpdateAckLevel mocks base method.\nfunc (m *MockReplicationQueue) UpdateAckLevel(ctx context.Context, lastProcessedMessageID int64, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevel\", ctx, lastProcessedMessageID, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAckLevel indicates an expected call of UpdateAckLevel.\nfunc (mr *MockReplicationQueueMockRecorder) UpdateAckLevel(ctx, lastProcessedMessageID, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevel\", reflect.TypeOf((*MockReplicationQueue)(nil).UpdateAckLevel), ctx, lastProcessedMessageID, clusterName)\n}\n\n// UpdateDLQAckLevel mocks base method.\nfunc (m *MockReplicationQueue) UpdateDLQAckLevel(ctx context.Context, lastProcessedMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDLQAckLevel\", ctx, lastProcessedMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDLQAckLevel indicates an expected call of UpdateDLQAckLevel.\nfunc (mr *MockReplicationQueueMockRecorder) UpdateDLQAckLevel(ctx, lastProcessedMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDLQAckLevel\", reflect.TypeOf((*MockReplicationQueue)(nil).UpdateDLQAckLevel), ctx, lastProcessedMessageID)\n}\n"
  },
  {
    "path": "common/domain/replication_queue_test.go",
    "content": "// Copyright (c) 2023 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/binary\"\n\t\"errors\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tpreambleVersion0 byte = 0x59\n)\n\nfunc TestReplicationQueueImpl_Start(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tinitialStatus  int32\n\t\texpectedStatus int32\n\t\tshouldStart    bool\n\t}{\n\t\t{\n\t\t\tname:           \"Should start when initialized\",\n\t\t\tinitialStatus:  common.DaemonStatusInitialized,\n\t\t\texpectedStatus: common.DaemonStatusStarted,\n\t\t\tshouldStart:    true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Should not start when already started\",\n\t\t\tinitialStatus:  common.DaemonStatusStarted,\n\t\t\texpectedStatus: common.DaemonStatusStarted,\n\t\t\tshouldStart:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t)).(*replicationQueueImpl)\n\t\t\tatomic.StoreInt32(&rq.status, tt.initialStatus)\n\n\t\t\trq.Start()\n\t\t\tdefer rq.Stop()\n\t\t\tassert.Equal(t, tt.expectedStatus, atomic.LoadInt32(&rq.status))\n\n\t\t\tif tt.shouldStart {\n\t\t\t\tselect {\n\t\t\t\tcase <-rq.done:\n\t\t\t\t\tt.Error(\"purgeProcessor should not have stopped\")\n\t\t\t\tcase <-time.After(time.Millisecond):\n\t\t\t\t\t// expected, as the purgeProcessor should still be running\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicationQueueImpl_Stop(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tinitialStatus  int32\n\t\texpectedStatus int32\n\t\tshouldStop     bool\n\t}{\n\t\t{\n\t\t\tname:           \"Should stop when started\",\n\t\t\tinitialStatus:  common.DaemonStatusStarted,\n\t\t\texpectedStatus: common.DaemonStatusStopped,\n\t\t\tshouldStop:     true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Should not stop when not started\",\n\t\t\tinitialStatus:  common.DaemonStatusInitialized,\n\t\t\texpectedStatus: common.DaemonStatusInitialized,\n\t\t\tshouldStop:     false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t)).(*replicationQueueImpl)\n\t\t\tatomic.StoreInt32(&rq.status, tt.initialStatus)\n\n\t\t\trq.Stop()\n\t\t\tassert.Equal(t, tt.expectedStatus, atomic.LoadInt32(&rq.status))\n\n\t\t\tif tt.shouldStop {\n\t\t\t\tselect {\n\t\t\t\tcase <-rq.done:\n\t\t\t\t\t// expected channel closed\n\t\t\t\tdefault:\n\t\t\t\t\tt.Error(\"done channel should be closed\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicationQueueImpl_Publish(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\ttask      *types.ReplicationTask\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful publish\",\n\t\t\ttask:    &types.ReplicationTask{},\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().EnqueueMessage(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"publish fails\",\n\t\t\ttask:    &types.ReplicationTask{},\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().EnqueueMessage(gomock.Any(), gomock.Any()).Return(errors.New(\"enqueue error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.Publish(context.Background(), tt.task)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicationQueueImpl_PublishToDLQ(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\ttask      *types.ReplicationTask\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful publish to DLQ\",\n\t\t\ttask:    &types.ReplicationTask{},\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().EnqueueMessageToDLQ(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"publish to DLQ fails\",\n\t\t\ttask:    &types.ReplicationTask{},\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().EnqueueMessageToDLQ(gomock.Any(), gomock.Any()).Return(errors.New(\"enqueue to DLQ error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.PublishToDLQ(context.Background(), tt.task)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetReplicationMessages(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(mockQueueManager *persistence.MockQueueManager)\n\t\texpectedTasks  int\n\t\texpectedLastID int64\n\t\texpectError    bool\n\t}{\n\t\t{\n\t\t\tname: \"handles empty message list\",\n\t\t\tsetupMocks: func(mockQueueManager *persistence.MockQueueManager) {\n\t\t\t\tmockQueueManager.EXPECT().\n\t\t\t\t\tReadMessages(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadMessagesResponse{Messages: nil}, nil)\n\t\t\t},\n\t\t\texpectedTasks:  0,\n\t\t\texpectedLastID: 0,\n\t\t\texpectError:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"decodes single message correctly\",\n\t\t\tsetupMocks: func(mockQueueManager *persistence.MockQueueManager) {\n\t\t\t\t// Setup mock to return one encoded message\n\t\t\t\tencodedMessage, _ := mockEncodeReplicationTask(123)\n\t\t\t\tmockQueueManager.EXPECT().\n\t\t\t\t\tReadMessages(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadMessagesResponse{Messages: []*persistence.QueueMessage{{ID: 1, Payload: encodedMessage}}}, nil)\n\t\t\t},\n\t\t\texpectedTasks:  1,\n\t\t\texpectedLastID: 1,\n\t\t\texpectError:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"decodes multiple messages correctly\",\n\t\t\tsetupMocks: func(mockQueueManager *persistence.MockQueueManager) {\n\t\t\t\t// Setup mock to return multiple encoded messages\n\t\t\t\tencodedMessage1, _ := mockEncodeReplicationTask(123)\n\t\t\t\tencodedMessage2, _ := mockEncodeReplicationTask(456)\n\t\t\t\tmockQueueManager.EXPECT().\n\t\t\t\t\tReadMessages(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadMessagesResponse{Messages: []*persistence.QueueMessage{\n\t\t\t\t\t\t{ID: 1, Payload: encodedMessage1},\n\t\t\t\t\t\t{ID: 2, Payload: encodedMessage2},\n\t\t\t\t\t}}, nil)\n\t\t\t},\n\t\t\texpectedTasks:  2,\n\t\t\texpectedLastID: 2,\n\t\t\texpectError:    false,\n\t\t},\n\t\t{\n\t\t\tname:           \"read messages fails\",\n\t\t\texpectedLastID: 100,\n\t\t\texpectError:    true,\n\t\t\tsetupMocks: func(mockQueueManager *persistence.MockQueueManager) {\n\t\t\t\tmockQueueManager.EXPECT().ReadMessages(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"read error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueManager := persistence.NewMockQueueManager(ctrl)\n\t\t\ttc.setupMocks(mockQueueManager)\n\t\t\treplicationQueue := NewReplicationQueue(mockQueueManager, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttasks, lastID, err := replicationQueue.GetReplicationMessages(context.Background(), 0, 10)\n\n\t\t\tif tc.expectError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedTasks, len(tasks))\n\t\t\t\tassert.Equal(t, tc.expectedLastID, lastID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateAckLevel(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tlastID    int64\n\t\tcluster   string\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful ack level update\",\n\t\t\tlastID:  100,\n\t\t\tcluster: \"testCluster\",\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().UpdateAckLevel(gomock.Any(), gomock.Eq(&persistence.UpdateAckLevelRequest{\n\t\t\t\t\tMessageID:   int64(100),\n\t\t\t\t\tClusterName: \"testCluster\",\n\t\t\t\t})).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ack level update fails\",\n\t\t\tlastID:  100,\n\t\t\tcluster: \"testCluster\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().UpdateAckLevel(gomock.Any(), gomock.Eq(&persistence.UpdateAckLevelRequest{\n\t\t\t\t\tMessageID:   int64(100),\n\t\t\t\t\tClusterName: \"testCluster\",\n\t\t\t\t})).Return(errors.New(\"update error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.UpdateAckLevel(context.Background(), tt.lastID, tt.cluster)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicationQueueImpl_GetAckLevels(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\twant      map[string]int64\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful ack levels retrieval\",\n\t\t\twant:    map[string]int64{\"testCluster\": 100},\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{\"testCluster\": 100}}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ack levels retrieval fails\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"retrieval error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\tgot, err := rq.GetAckLevels(context.Background())\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mockEncodeReplicationTask(sourceTaskID int64) ([]byte, error) {\n\tvar buf bytes.Buffer\n\tbuf.WriteByte(preambleVersion0)\n\tbinary.Write(&buf, binary.BigEndian, sourceTaskID)\n\treturn buf.Bytes(), nil\n}\n\nfunc TestGetMessagesFromDLQ(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tfirstID   int64\n\t\tlastID    int64\n\t\tpageSize  int\n\t\tpageToken []byte\n\t\ttaskID    int64\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"successful message retrieval\",\n\t\t\tfirstID:   100,\n\t\t\tlastID:    200,\n\t\t\tpageSize:  10,\n\t\t\tpageToken: []byte(\"token\"),\n\t\t\ttaskID:    12345,\n\t\t\twantErr:   false,\n\t\t},\n\t\t{\n\t\t\tname:      \"read messages fails\",\n\t\t\tfirstID:   100,\n\t\t\tlastID:    200,\n\t\t\tpageSize:  10,\n\t\t\tpageToken: []byte(\"token\"),\n\t\t\twantErr:   true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\n\t\t\tif !tt.wantErr {\n\t\t\t\tencodedData, _ := mockEncodeReplicationTask(tt.taskID)\n\t\t\t\tmessages := []*persistence.QueueMessage{\n\t\t\t\t\t{ID: 1, Payload: encodedData},\n\t\t\t\t}\n\t\t\t\tmockQueue.EXPECT().ReadMessagesFromDLQ(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadMessagesFromDLQResponse{Messages: messages, NextPageToken: []byte(\"nextToken\")}, nil)\n\t\t\t} else {\n\t\t\t\tmockQueue.EXPECT().ReadMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"read error\"))\n\t\t\t}\n\n\t\t\treplicationTasks, token, err := rq.GetMessagesFromDLQ(context.Background(), tt.firstID, tt.lastID, tt.pageSize, tt.pageToken)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, replicationTasks, 1, \"Expected one replication task to be returned\")\n\t\t\t\tassert.Equal(t, []byte(\"nextToken\"), token, \"Expected token to match 'nextToken'\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDLQAckLevel(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tlastID    int64\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful DLQ ack level update\",\n\t\t\tlastID:  100,\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Eq(&persistence.UpdateDLQAckLevelRequest{\n\t\t\t\t\tMessageID:   int64(100),\n\t\t\t\t\tClusterName: \"domainReplication\",\n\t\t\t\t})).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"DLQ ack level update fails\",\n\t\t\tlastID:  100,\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Eq(&persistence.UpdateDLQAckLevelRequest{\n\t\t\t\t\tMessageID:   int64(100),\n\t\t\t\t\tClusterName: \"domainReplication\",\n\t\t\t\t})).Return(errors.New(\"update DLQ ack level error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.UpdateDLQAckLevel(context.Background(), tt.lastID)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDLQAckLevel(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\twant      int64\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful DLQ ack level retrieval\",\n\t\t\twant:    100,\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().GetDLQAckLevels(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetDLQAckLevelsResponse{AckLevels: map[string]int64{\"domainReplication\": 100}}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"DLQ ack level retrieval fails\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().GetDLQAckLevels(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"get DLQ ack level error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\tgot, err := rq.GetDLQAckLevel(context.Background())\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteMessagesFromDLQ(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tfirstID   int64\n\t\tlastID    int64\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successful range delete from DLQ\",\n\t\t\tfirstID: 10,\n\t\t\tlastID:  20,\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Eq(&persistence.RangeDeleteMessagesFromDLQRequest{\n\t\t\t\t\tFirstMessageID: int64(10),\n\t\t\t\t\tLastMessageID:  int64(20),\n\t\t\t\t})).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"range delete from DLQ fails\",\n\t\t\tfirstID: 10,\n\t\t\tlastID:  20,\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Eq(&persistence.RangeDeleteMessagesFromDLQRequest{\n\t\t\t\t\tFirstMessageID: int64(10),\n\t\t\t\t\tLastMessageID:  int64(20),\n\t\t\t\t})).Return(errors.New(\"range delete error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.RangeDeleteMessagesFromDLQ(context.Background(), tt.firstID, tt.lastID)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessageFromDLQ(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tmessageID int64\n\t\twantErr   bool\n\t\tsetupMock func(q *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:      \"successful delete from DLQ\",\n\t\t\tmessageID: 15,\n\t\t\twantErr:   false,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Eq(&persistence.DeleteMessageFromDLQRequest{\n\t\t\t\t\tMessageID: int64(15),\n\t\t\t\t})).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"delete from DLQ fails\",\n\t\t\tmessageID: 15,\n\t\t\twantErr:   true,\n\t\t\tsetupMock: func(q *persistence.MockQueueManager) {\n\t\t\t\tq.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Eq(&persistence.DeleteMessageFromDLQRequest{\n\t\t\t\t\tMessageID: int64(15),\n\t\t\t\t})).Return(errors.New(\"delete error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueue := persistence.NewMockQueueManager(ctrl)\n\t\t\trq := NewReplicationQueue(mockQueue, \"testCluster\", nil, testlogger.New(t))\n\t\t\ttt.setupMock(mockQueue)\n\t\t\terr := rq.DeleteMessageFromDLQ(context.Background(), tt.messageID)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDLQSize(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\twantSize  int64\n\t\twantErr   bool\n\t\tsetupMock func(m *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:     \"returns correct size for non-empty DLQ\",\n\t\t\twantSize: 10,\n\t\t\twantErr:  false,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQSizeResponse{Size: int64(10)}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"returns zero for empty DLQ\",\n\t\t\twantSize: 0,\n\t\t\twantErr:  false,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQSizeResponse{Size: int64(0)}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"propagates error from underlying queue\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"database error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueManager := persistence.NewMockQueueManager(ctrl)\n\t\t\ttt.setupMock(mockQueueManager)\n\t\t\tq := &replicationQueueImpl{queue: mockQueueManager, logger: testlogger.New(t)}\n\t\t\tsize, err := q.GetDLQSize(context.Background())\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.wantSize, size)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPurgeAckedMessages(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\twantErr   bool\n\t\tsetupMock func(m *persistence.MockQueueManager)\n\t}{\n\t\t{\n\t\t\tname:    \"successfully purges messages\",\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{\"clusterA\": 5}}, nil)\n\t\t\t\tm.EXPECT().DeleteMessagesBefore(gomock.Any(), gomock.Eq(&persistence.DeleteMessagesBeforeRequest{\n\t\t\t\t\tMessageID: int64(5),\n\t\t\t\t})).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"does nothing when no ack levels\",\n\t\t\twantErr: false,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{}}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"error on GetAckLevels\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"database error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"error on DeleteMessagesBefore\",\n\t\t\twantErr: true,\n\t\t\tsetupMock: func(m *persistence.MockQueueManager) {\n\t\t\t\tm.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{\"clusterA\": 5}}, nil)\n\t\t\t\tm.EXPECT().DeleteMessagesBefore(gomock.Any(), gomock.Eq(&persistence.DeleteMessagesBeforeRequest{\n\t\t\t\t\tMessageID: int64(5),\n\t\t\t\t})).Return(errors.New(\"delete error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueManager := persistence.NewMockQueueManager(ctrl)\n\t\t\ttt.setupMock(mockQueueManager)\n\t\t\tq := &replicationQueueImpl{queue: mockQueueManager, logger: testlogger.New(t)}\n\t\t\terr := q.purgeAckedMessages()\n\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/domain/transmissionTaskHandler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination transmissionTaskHandler_mock.go\n\npackage domain\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// NOTE: the counterpart of domain replication receiving logic is in service/worker package\n\ntype (\n\t// Replicator is the interface which can replicate the domain\n\tReplicator interface {\n\t\tHandleTransmissionTask(\n\t\t\tctx context.Context,\n\t\t\tdomainOperation types.DomainOperation,\n\t\t\tinfo *persistence.DomainInfo,\n\t\t\tconfig *persistence.DomainConfig,\n\t\t\treplicationConfig *persistence.DomainReplicationConfig,\n\t\t\tconfigVersion int64,\n\t\t\tfailoverVersion int64,\n\t\t\tpreviousFailoverVersion int64,\n\t\t\tisGlobalDomainEnabled bool,\n\t\t) error\n\t}\n\n\tdomainReplicatorImpl struct {\n\t\treplicationMessageSink messaging.Producer\n\t\tlogger                 log.Logger\n\t}\n)\n\n// NewDomainReplicator create a new instance of domain replicator\nfunc NewDomainReplicator(replicationMessageSink messaging.Producer, logger log.Logger) Replicator {\n\treturn &domainReplicatorImpl{\n\t\treplicationMessageSink: replicationMessageSink,\n\t\tlogger:                 logger,\n\t}\n}\n\n// HandleTransmissionTask handle transmission of the domain replication task\nfunc (domainReplicator *domainReplicatorImpl) HandleTransmissionTask(\n\tctx context.Context,\n\tdomainOperation types.DomainOperation,\n\tinfo *persistence.DomainInfo,\n\tconfig *persistence.DomainConfig,\n\treplicationConfig *persistence.DomainReplicationConfig,\n\tconfigVersion int64,\n\tfailoverVersion int64,\n\tpreviousFailoverVersion int64,\n\tisGlobalDomainEnabled bool,\n) error {\n\n\tif !isGlobalDomainEnabled {\n\t\tdomainReplicator.logger.Warn(\"Should not replicate non global domain\", tag.WorkflowDomainID(info.ID))\n\t\treturn nil\n\t}\n\n\tstatus, err := domainReplicator.convertDomainStatusToThrift(info.Status)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttaskType := types.ReplicationTaskTypeDomain\n\ttask := &types.DomainTaskAttributes{\n\t\tDomainOperation: &domainOperation,\n\t\tID:              info.ID,\n\t\tInfo: &types.DomainInfo{\n\t\t\tName:        info.Name,\n\t\t\tStatus:      status,\n\t\t\tDescription: info.Description,\n\t\t\tOwnerEmail:  info.OwnerEmail,\n\t\t\tData:        info.Data,\n\t\t},\n\t\tConfig: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: config.Retention,\n\t\t\tEmitMetric:                             config.EmitMetric,\n\t\t\tHistoryArchivalStatus:                  config.HistoryArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     config.HistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus:               config.VisibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  config.VisibilityArchivalURI,\n\t\t\tBadBinaries:                            &config.BadBinaries,\n\t\t\tIsolationGroups:                        &config.IsolationGroups,\n\t\t\tAsyncWorkflowConfig:                    &config.AsyncWorkflowConfig,\n\t\t},\n\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: replicationConfig.ActiveClusterName,\n\t\t\tClusters:          domainReplicator.convertClusterReplicationConfigToThrift(replicationConfig.Clusters),\n\t\t\tActiveClusters:    replicationConfig.ActiveClusters,\n\t\t},\n\t\tConfigVersion:           configVersion,\n\t\tFailoverVersion:         failoverVersion,\n\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t}\n\n\treturn domainReplicator.replicationMessageSink.Publish(\n\t\tctx,\n\t\t&types.ReplicationTask{\n\t\t\tTaskType:             &taskType,\n\t\t\tDomainTaskAttributes: task,\n\t\t})\n}\n\nfunc (domainReplicator *domainReplicatorImpl) convertClusterReplicationConfigToThrift(\n\tinput []*persistence.ClusterReplicationConfig,\n) []*types.ClusterReplicationConfiguration {\n\toutput := []*types.ClusterReplicationConfiguration{}\n\tfor _, cluster := range input {\n\t\toutput = append(output, &types.ClusterReplicationConfiguration{ClusterName: cluster.ClusterName})\n\t}\n\treturn output\n}\n\nfunc (domainReplicator *domainReplicatorImpl) convertDomainStatusToThrift(input int) (*types.DomainStatus, error) {\n\tswitch input {\n\tcase persistence.DomainStatusRegistered:\n\t\toutput := types.DomainStatusRegistered\n\t\treturn &output, nil\n\tcase persistence.DomainStatusDeprecated:\n\t\toutput := types.DomainStatusDeprecated\n\t\treturn &output, nil\n\tdefault:\n\t\treturn nil, ErrInvalidDomainStatus\n\t}\n}\n"
  },
  {
    "path": "common/domain/transmissionTaskHandler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: transmissionTaskHandler.go\n//\n// Generated by this command:\n//\n//\tmockgen -package domain -source transmissionTaskHandler.go -destination transmissionTaskHandler_mock.go\n//\n\n// Package domain is a generated GoMock package.\npackage domain\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockReplicator is a mock of Replicator interface.\ntype MockReplicator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReplicatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockReplicatorMockRecorder is the mock recorder for MockReplicator.\ntype MockReplicatorMockRecorder struct {\n\tmock *MockReplicator\n}\n\n// NewMockReplicator creates a new mock instance.\nfunc NewMockReplicator(ctrl *gomock.Controller) *MockReplicator {\n\tmock := &MockReplicator{ctrl: ctrl}\n\tmock.recorder = &MockReplicatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockReplicator) EXPECT() *MockReplicatorMockRecorder {\n\treturn m.recorder\n}\n\n// HandleTransmissionTask mocks base method.\nfunc (m *MockReplicator) HandleTransmissionTask(ctx context.Context, domainOperation types.DomainOperation, info *persistence.DomainInfo, config *persistence.DomainConfig, replicationConfig *persistence.DomainReplicationConfig, configVersion, failoverVersion, previousFailoverVersion int64, isGlobalDomainEnabled bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleTransmissionTask\", ctx, domainOperation, info, config, replicationConfig, configVersion, failoverVersion, previousFailoverVersion, isGlobalDomainEnabled)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// HandleTransmissionTask indicates an expected call of HandleTransmissionTask.\nfunc (mr *MockReplicatorMockRecorder) HandleTransmissionTask(ctx, domainOperation, info, config, replicationConfig, configVersion, failoverVersion, previousFailoverVersion, isGlobalDomainEnabled any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleTransmissionTask\", reflect.TypeOf((*MockReplicator)(nil).HandleTransmissionTask), ctx, domainOperation, info, config, replicationConfig, configVersion, failoverVersion, previousFailoverVersion, isGlobalDomainEnabled)\n}\n"
  },
  {
    "path": "common/domain/transmissionTaskHandler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage domain\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\ttransmissionTaskSuite struct {\n\t\tsuite.Suite\n\t\tdomainReplicator *domainReplicatorImpl\n\t\tkafkaProducer    *mocks.KafkaProducer\n\t}\n)\n\nfunc TestTransmissionTaskSuite(t *testing.T) {\n\ts := new(transmissionTaskSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transmissionTaskSuite) SetupSuite() {\n}\n\nfunc (s *transmissionTaskSuite) TearDownSuite() {\n\n}\n\nfunc (s *transmissionTaskSuite) SetupTest() {\n\ts.kafkaProducer = &mocks.KafkaProducer{}\n\ts.domainReplicator = NewDomainReplicator(\n\t\ts.kafkaProducer,\n\t\ttestlogger.New(s.Suite.T()),\n\t).(*domainReplicatorImpl)\n}\n\nfunc (s *transmissionTaskSuite) TearDownTest() {\n\ts.kafkaProducer.AssertExpectations(s.T())\n}\n\nfunc (s *transmissionTaskSuite) TestHandleTransmissionTask_RegisterDomainTask_IsGlobalDomain() {\n\ttaskType := types.ReplicationTaskTypeDomain\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusRegistered\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tdomainOperation := types.DomainOperationCreate\n\tinfo := &p.DomainInfo{\n\t\tID:          id,\n\t\tName:        name,\n\t\tStatus:      p.DomainStatusRegistered,\n\t\tDescription: description,\n\t\tOwnerEmail:  ownerEmail,\n\t\tData:        data,\n\t}\n\tconfig := &p.DomainConfig{\n\t\tRetention:                retention,\n\t\tEmitMetric:               emitMetric,\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t}\n\treplicationConfig := &p.DomainReplicationConfig{\n\t\tActiveClusterName: clusterActive,\n\t\tClusters:          clusters,\n\t}\n\tisGlobalDomain := true\n\n\ts.kafkaProducer.On(\"Publish\", mock.Anything, &types.ReplicationTask{\n\t\tTaskType: &taskType,\n\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\tDomainOperation: &domainOperation,\n\t\t\tID:              id,\n\t\t\tInfo: &types.DomainInfo{\n\t\t\t\tName:        name,\n\t\t\t\tStatus:      &status,\n\t\t\t\tDescription: description,\n\t\t\t\tOwnerEmail:  ownerEmail,\n\t\t\t\tData:        data,\n\t\t\t},\n\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\t\tEmitMetric:                             emitMetric,\n\t\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{},\n\t\t\t},\n\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusterName: clusterActive,\n\t\t\t\tClusters:          s.domainReplicator.convertClusterReplicationConfigToThrift(clusters),\n\t\t\t},\n\t\t\tConfigVersion:           configVersion,\n\t\t\tFailoverVersion:         failoverVersion,\n\t\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t\t},\n\t}).Return(nil).Once()\n\n\terr := s.domainReplicator.HandleTransmissionTask(\n\t\tcontext.Background(),\n\t\tdomainOperation,\n\t\tinfo,\n\t\tconfig,\n\t\treplicationConfig,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tpreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t)\n\ts.Nil(err)\n}\n\nfunc (s *transmissionTaskSuite) TestHandleTransmissionTask_RegisterDomainTask_NotGlobalDomain() {\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tdomainOperation := types.DomainOperationCreate\n\tinfo := &p.DomainInfo{\n\t\tID:          id,\n\t\tName:        name,\n\t\tStatus:      p.DomainStatusRegistered,\n\t\tDescription: description,\n\t\tOwnerEmail:  ownerEmail,\n\t\tData:        data,\n\t}\n\tconfig := &p.DomainConfig{\n\t\tRetention:                retention,\n\t\tEmitMetric:               emitMetric,\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\tBadBinaries:              types.BadBinaries{},\n\t}\n\treplicationConfig := &p.DomainReplicationConfig{\n\t\tActiveClusterName: clusterActive,\n\t\tClusters:          clusters,\n\t}\n\tisGlobalDomain := false\n\n\terr := s.domainReplicator.HandleTransmissionTask(\n\t\tcontext.Background(),\n\t\tdomainOperation,\n\t\tinfo,\n\t\tconfig,\n\t\treplicationConfig,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tpreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t)\n\ts.Nil(err)\n}\n\nfunc (s *transmissionTaskSuite) TestHandleTransmissionTask_UpdateDomainTask_IsGlobalDomain() {\n\ttaskType := types.ReplicationTaskTypeDomain\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tstatus := types.DomainStatusDeprecated\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tdomainOperation := types.DomainOperationUpdate\n\tinfo := &p.DomainInfo{\n\t\tID:          id,\n\t\tName:        name,\n\t\tStatus:      p.DomainStatusDeprecated,\n\t\tDescription: description,\n\t\tOwnerEmail:  ownerEmail,\n\t\tData:        data,\n\t}\n\tconfig := &p.DomainConfig{\n\t\tRetention:                retention,\n\t\tEmitMetric:               emitMetric,\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateDrained},\n\t\t},\n\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\tEnabled:             true,\n\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\tQueueType:           \"kafka\",\n\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         []byte(`{\"cluster\": \"queue1\"}`),\n\t\t\t},\n\t\t},\n\t}\n\treplicationConfig := &p.DomainReplicationConfig{\n\t\tActiveClusterName: clusterActive,\n\t\tClusters:          clusters,\n\t}\n\tisGlobalDomain := true\n\n\ts.kafkaProducer.On(\"Publish\", mock.Anything, &types.ReplicationTask{\n\t\tTaskType: &taskType,\n\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\tDomainOperation: &domainOperation,\n\t\t\tID:              id,\n\t\t\tInfo: &types.DomainInfo{\n\t\t\t\tName:        name,\n\t\t\t\tStatus:      &status,\n\t\t\t\tDescription: description,\n\t\t\t\tOwnerEmail:  ownerEmail,\n\t\t\t\tData:        data,\n\t\t\t},\n\t\t\tConfig: &types.DomainConfiguration{\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: retention,\n\t\t\t\tEmitMetric:                             emitMetric,\n\t\t\t\tHistoryArchivalStatus:                  historyArchivalStatus.Ptr(),\n\t\t\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\t\t\tVisibilityArchivalStatus:               visibilityArchivalStatus.Ptr(),\n\t\t\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\tIsolationGroups: &types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateDrained},\n\t\t\t\t},\n\t\t\t\tAsyncWorkflowConfig: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\tQueueType:           \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"cluster\": \"queue1\"}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tReplicationConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusterName: clusterActive,\n\t\t\t\tClusters:          s.domainReplicator.convertClusterReplicationConfigToThrift(clusters),\n\t\t\t},\n\t\t\tConfigVersion:           configVersion,\n\t\t\tFailoverVersion:         failoverVersion,\n\t\t\tPreviousFailoverVersion: previousFailoverVersion,\n\t\t},\n\t}).Return(nil).Once()\n\n\terr := s.domainReplicator.HandleTransmissionTask(\n\t\tcontext.Background(),\n\t\tdomainOperation,\n\t\tinfo,\n\t\tconfig,\n\t\treplicationConfig,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tpreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t)\n\ts.Nil(err)\n}\n\nfunc (s *transmissionTaskSuite) TestHandleTransmissionTask_UpdateDomainTask_NotGlobalDomain() {\n\tid := uuid.New()\n\tname := \"some random domain test name\"\n\tdescription := \"some random test description\"\n\townerEmail := \"some random test owner\"\n\tdata := map[string]string{\"k\": \"v\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"some random history archival uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"some random visibility archival uri\"\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(59)\n\tpreviousFailoverVersion := int64(55)\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tdomainOperation := types.DomainOperationUpdate\n\tinfo := &p.DomainInfo{\n\t\tID:          id,\n\t\tName:        name,\n\t\tStatus:      p.DomainStatusDeprecated,\n\t\tDescription: description,\n\t\tOwnerEmail:  ownerEmail,\n\t\tData:        data,\n\t}\n\tconfig := &p.DomainConfig{\n\t\tRetention:                retention,\n\t\tEmitMetric:               emitMetric,\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t}\n\treplicationConfig := &p.DomainReplicationConfig{\n\t\tActiveClusterName: clusterActive,\n\t\tClusters:          clusters,\n\t}\n\tisGlobalDomain := false\n\n\terr := s.domainReplicator.HandleTransmissionTask(\n\t\tcontext.Background(),\n\t\tdomainOperation,\n\t\tinfo,\n\t\tconfig,\n\t\treplicationConfig,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tpreviousFailoverVersion,\n\t\tisGlobalDomain,\n\t)\n\ts.Nil(err)\n}\n"
  },
  {
    "path": "common/dynamicconfig/clientInterface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination clientInterface_mock.go -self_package github.com/uber/cadence/common/dynamicconfig\n\npackage dynamicconfig\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tConfigStoreClient = \"configstore\"\n\tFileBasedClient   = \"filebased\"\n\tInMemoryClient    = \"memory\"\n\tNopClient         = \"nop\"\n)\n\n// Client allows fetching values from a dynamic configuration system NOTE: This does not have async\n// options right now. In the interest of keeping it minimal, we can add when requirement arises.\ntype Client interface {\n\tGetValue(name dynamicproperties.Key) (interface{}, error)\n\tGetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error)\n\n\tGetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error)\n\tGetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error)\n\tGetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error)\n\tGetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error)\n\tGetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error)\n\tGetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error)\n\tGetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error)\n\t// UpdateValue takes value as map and updates by overriding. It doesn't support update with filters.\n\tUpdateValue(name dynamicproperties.Key, value interface{}) error\n\tRestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error\n\tListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error)\n}\n\nvar NotFoundError = &types.EntityNotExistsError{\n\tMessage: \"unable to find key\",\n}\n"
  },
  {
    "path": "common/dynamicconfig/clientInterface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: clientInterface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package dynamicconfig -source clientInterface.go -destination clientInterface_mock.go -self_package github.com/uber/cadence/common/dynamicconfig\n//\n\n// Package dynamicconfig is a generated GoMock package.\npackage dynamicconfig\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tdynamicproperties \"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// GetBoolValue mocks base method.\nfunc (m *MockClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]any) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBoolValue\", name, filters)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetBoolValue indicates an expected call of GetBoolValue.\nfunc (mr *MockClientMockRecorder) GetBoolValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBoolValue\", reflect.TypeOf((*MockClient)(nil).GetBoolValue), name, filters)\n}\n\n// GetDurationValue mocks base method.\nfunc (m *MockClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]any) (time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDurationValue\", name, filters)\n\tret0, _ := ret[0].(time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDurationValue indicates an expected call of GetDurationValue.\nfunc (mr *MockClientMockRecorder) GetDurationValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDurationValue\", reflect.TypeOf((*MockClient)(nil).GetDurationValue), name, filters)\n}\n\n// GetFloatValue mocks base method.\nfunc (m *MockClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]any) (float64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFloatValue\", name, filters)\n\tret0, _ := ret[0].(float64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetFloatValue indicates an expected call of GetFloatValue.\nfunc (mr *MockClientMockRecorder) GetFloatValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFloatValue\", reflect.TypeOf((*MockClient)(nil).GetFloatValue), name, filters)\n}\n\n// GetIntValue mocks base method.\nfunc (m *MockClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]any) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIntValue\", name, filters)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetIntValue indicates an expected call of GetIntValue.\nfunc (mr *MockClientMockRecorder) GetIntValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIntValue\", reflect.TypeOf((*MockClient)(nil).GetIntValue), name, filters)\n}\n\n// GetListValue mocks base method.\nfunc (m *MockClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]any) ([]any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetListValue\", name, filters)\n\tret0, _ := ret[0].([]any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetListValue indicates an expected call of GetListValue.\nfunc (mr *MockClientMockRecorder) GetListValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetListValue\", reflect.TypeOf((*MockClient)(nil).GetListValue), name, filters)\n}\n\n// GetMapValue mocks base method.\nfunc (m *MockClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]any) (map[string]any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMapValue\", name, filters)\n\tret0, _ := ret[0].(map[string]any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMapValue indicates an expected call of GetMapValue.\nfunc (mr *MockClientMockRecorder) GetMapValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMapValue\", reflect.TypeOf((*MockClient)(nil).GetMapValue), name, filters)\n}\n\n// GetStringValue mocks base method.\nfunc (m *MockClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]any) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStringValue\", name, filters)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStringValue indicates an expected call of GetStringValue.\nfunc (mr *MockClientMockRecorder) GetStringValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStringValue\", reflect.TypeOf((*MockClient)(nil).GetStringValue), name, filters)\n}\n\n// GetValue mocks base method.\nfunc (m *MockClient) GetValue(name dynamicproperties.Key) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetValue\", name)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetValue indicates an expected call of GetValue.\nfunc (mr *MockClientMockRecorder) GetValue(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetValue\", reflect.TypeOf((*MockClient)(nil).GetValue), name)\n}\n\n// GetValueWithFilters mocks base method.\nfunc (m *MockClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]any) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetValueWithFilters\", name, filters)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetValueWithFilters indicates an expected call of GetValueWithFilters.\nfunc (mr *MockClientMockRecorder) GetValueWithFilters(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetValueWithFilters\", reflect.TypeOf((*MockClient)(nil).GetValueWithFilters), name, filters)\n}\n\n// ListValue mocks base method.\nfunc (m *MockClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListValue\", name)\n\tret0, _ := ret[0].([]*types.DynamicConfigEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListValue indicates an expected call of ListValue.\nfunc (mr *MockClientMockRecorder) ListValue(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListValue\", reflect.TypeOf((*MockClient)(nil).ListValue), name)\n}\n\n// RestoreValue mocks base method.\nfunc (m *MockClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RestoreValue\", name, filters)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RestoreValue indicates an expected call of RestoreValue.\nfunc (mr *MockClientMockRecorder) RestoreValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestoreValue\", reflect.TypeOf((*MockClient)(nil).RestoreValue), name, filters)\n}\n\n// UpdateValue mocks base method.\nfunc (m *MockClient) UpdateValue(name dynamicproperties.Key, value any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateValue\", name, value)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateValue indicates an expected call of UpdateValue.\nfunc (mr *MockClientMockRecorder) UpdateValue(name, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateValue\", reflect.TypeOf((*MockClient)(nil).UpdateValue), name, value)\n}\n"
  },
  {
    "path": "common/dynamicconfig/config/testConfig.yaml",
    "content": "admin.HeaderForwardingRules:\n- value:\n  - Add: true\n    Match: (?i)key\n  constraints: {}\nfrontend.validSearchAttributes:\n- value:\n    DomainID: 1\n  constraints: {}\ntestGetBoolPropertyKey:\n- value: false\n  constraints: {}\n- value: true\n  constraints:\n    domainName: global-samples-domain\n- value: true\n  constraints:\n    domainName: samples-domain\ntestGetDurationPropertyKey:\n- value: 1m\n  constraints: {}\n- value: wrong duration string\n  constraints:\n    domainName: samples-domain\n    taskListName: longIdleTimeTasklist\n- value: 2\n  constraints:\n    domainName: samples-domain\ntestGetFloat64PropertyKey:\n- value: 12\n  constraints: {}\n- value: wrong type\n  constraints:\n    domainName: samples-domain\ntestGetIntPropertyKey:\n- value: 1000\n  constraints: {}\n- value: 1000.1\n  constraints:\n    domainName: global-samples-domain\ntestGetMapPropertyKey:\n- value:\n    key1: \"1\"\n    key2: 1\n    key3:\n    - false\n    - key4: true\n      key5: 2.1\n  constraints: {}\n- value: \"1\"\n  constraints:\n    taskListName: random tasklist\ntestGetStringPropertyKey:\n- value: some random string\n  constraints: {}\n- value: constrained-string\n  constraints:\n    taskListName: random tasklist\n"
  },
  {
    "path": "common/dynamicconfig/config.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\terrCountLogThreshold = 1000\n)\n\n// NewCollection creates a new collection\nfunc NewCollection(\n\tclient Client,\n\tlogger log.Logger,\n\tfilterOptions ...dynamicproperties.FilterOption,\n) *Collection {\n\n\treturn &Collection{\n\t\tclient:        client,\n\t\tlogger:        logger,\n\t\terrCount:      -1,\n\t\tfilterOptions: filterOptions,\n\t}\n}\n\n// Collection wraps dynamic config client with a closure so that across the code, the config values\n// can be directly accessed by calling the function without propagating the client everywhere in\n// code\ntype Collection struct {\n\tclient        Client\n\tlogger        log.Logger\n\terrCount      int64\n\tfilterOptions []dynamicproperties.FilterOption\n}\n\nfunc (c *Collection) logError(\n\tkey dynamicproperties.Key,\n\tfilters map[dynamicproperties.Filter]interface{},\n\terr error,\n) {\n\terrCount := atomic.AddInt64(&c.errCount, 1)\n\tif errCount%errCountLogThreshold == 0 {\n\t\t// log only every 'x' errors to reduce mem allocs and to avoid log noise\n\t\tfilteredKey := getFilteredKeyAsString(key, filters)\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\tc.logger.Debug(\"dynamic config not set, use default value\", tag.Key(filteredKey))\n\t\t} else {\n\t\t\tc.logger.Warn(\"Failed to fetch key from dynamic config\", tag.Key(filteredKey), tag.Error(err))\n\t\t}\n\t}\n}\n\n// GetProperty gets a interface property and returns defaultValue if property is not found\nfunc (c *Collection) GetProperty(key dynamicproperties.Key) dynamicproperties.PropertyFn {\n\treturn func() interface{} {\n\t\tval, err := c.client.GetValue(key)\n\t\tif err != nil {\n\t\t\tc.logError(key, nil, err)\n\t\t\treturn key.DefaultValue()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetIntProperty gets property and asserts that it's an integer\nfunc (c *Collection) GetIntProperty(key dynamicproperties.IntKey) dynamicproperties.IntPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) int {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetIntValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultInt()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetIntPropertyFilteredByDomain gets property with domain filter and asserts that it's an integer\nfunc (c *Collection) GetIntPropertyFilteredByDomain(key dynamicproperties.IntKey) dynamicproperties.IntPropertyFnWithDomainFilter {\n\treturn func(domain string) int {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainFilter(domain))\n\t\tval, err := c.client.GetIntValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultInt()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetIntPropertyFilteredByWorkflowType gets property with workflow type filter and asserts that it's an integer\nfunc (c *Collection) GetIntPropertyFilteredByWorkflowType(key dynamicproperties.IntKey) dynamicproperties.IntPropertyFnWithWorkflowTypeFilter {\n\treturn func(domainName string, workflowType string) int {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domainName),\n\t\t\tdynamicproperties.WorkflowTypeFilter(workflowType),\n\t\t)\n\t\tval, err := c.client.GetIntValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultInt()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationPropertyFilteredByWorkflowType gets property with workflow type filter and asserts that it's a duration\nfunc (c *Collection) GetDurationPropertyFilteredByWorkflowType(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFnWithWorkflowTypeFilter {\n\treturn func(domainName string, workflowType string) time.Duration {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domainName),\n\t\t\tdynamicproperties.WorkflowTypeFilter(workflowType),\n\t\t)\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetIntPropertyFilteredByTaskListInfo gets property with taskListInfo as filters and asserts that it's an integer\nfunc (c *Collection) GetIntPropertyFilteredByTaskListInfo(key dynamicproperties.IntKey) dynamicproperties.IntPropertyFnWithTaskListInfoFilters {\n\treturn func(domain string, taskList string, taskType int) int {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domain),\n\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\tdynamicproperties.TaskTypeFilter(taskType),\n\t\t)\n\t\tval, err := c.client.GetIntValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultInt()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetIntPropertyFilteredByShardID gets property with shardID as filter and asserts that it's an integer\nfunc (c *Collection) GetIntPropertyFilteredByShardID(key dynamicproperties.IntKey) dynamicproperties.IntPropertyFnWithShardIDFilter {\n\treturn func(shardID int) int {\n\t\tfilters := c.toFilterMap(dynamicproperties.ShardIDFilter(shardID))\n\t\tval, err := c.client.GetIntValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultInt()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolPropertyFilteredByShardID gets property with shardID as filter and asserts that it's a bool\nfunc (c *Collection) GetBoolPropertyFilteredByShardID(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFnWithShardIDFilter {\n\treturn func(shardID int) bool {\n\t\tfilters := c.toFilterMap(dynamicproperties.ShardIDFilter(shardID))\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetFloat64Property gets property and asserts that it's a float64\nfunc (c *Collection) GetFloat64Property(key dynamicproperties.FloatKey) dynamicproperties.FloatPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) float64 {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetFloatValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultFloat()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetFloat64PropertyFilteredByShardID gets property with shardID filter and asserts that it's a float64\nfunc (c *Collection) GetFloat64PropertyFilteredByShardID(key dynamicproperties.FloatKey) dynamicproperties.FloatPropertyFnWithShardIDFilter {\n\treturn func(shardID int) float64 {\n\t\tfilters := c.toFilterMap(dynamicproperties.ShardIDFilter(shardID))\n\t\tval, err := c.client.GetFloatValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultFloat()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetFloatPropertyFilteredByTaskListInfo gets property with taskListInfo as filters and asserts that it's a float64\nfunc (c *Collection) GetFloat64PropertyFilteredByTaskListInfo(key dynamicproperties.FloatKey) dynamicproperties.FloatPropertyFnWithTaskListInfoFilters {\n\treturn func(domain string, taskList string, taskType int) float64 {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domain),\n\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\tdynamicproperties.TaskTypeFilter(taskType),\n\t\t)\n\t\tval, err := c.client.GetFloatValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultFloat()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetFloat64PropertyFilteredByNamespace gets property with domain filter and asserts that it's a float64\nfunc (c *Collection) GetFloat64PropertyFilteredByNamespace(key dynamicproperties.FloatKey) dynamicproperties.Float64PropertyFnWithNamespaceFilters {\n\treturn func(namespace string) float64 {\n\t\tfilters := c.toFilterMap(dynamicproperties.NamespaceFilter(namespace))\n\t\tval, err := c.client.GetFloatValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultFloat()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationProperty gets property and asserts that it's a duration\nfunc (c *Collection) GetDurationProperty(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) time.Duration {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationPropertyFilteredByDomain gets property with domain filter and asserts that it's a duration\nfunc (c *Collection) GetDurationPropertyFilteredByDomain(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFnWithDomainFilter {\n\treturn func(domain string) time.Duration {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainFilter(domain))\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationPropertyFilteredByDomainID gets property with domainID filter and asserts that it's a duration\nfunc (c *Collection) GetDurationPropertyFilteredByDomainID(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFnWithDomainIDFilter {\n\treturn func(domainID string) time.Duration {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainIDFilter(domainID))\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationPropertyFilteredByTaskListInfo gets property with taskListInfo as filters and asserts that it's a duration\nfunc (c *Collection) GetDurationPropertyFilteredByTaskListInfo(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFnWithTaskListInfoFilters {\n\treturn func(domain string, taskList string, taskType int) time.Duration {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domain),\n\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\tdynamicproperties.TaskTypeFilter(taskType),\n\t\t)\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetDurationPropertyFilteredByShardID gets property with shardID id as filter and asserts that it's a duration\nfunc (c *Collection) GetDurationPropertyFilteredByShardID(key dynamicproperties.DurationKey) dynamicproperties.DurationPropertyFnWithShardIDFilter {\n\treturn func(shardID int) time.Duration {\n\t\tfilters := c.toFilterMap(dynamicproperties.ShardIDFilter(shardID))\n\t\tval, err := c.client.GetDurationValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultDuration()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolProperty gets property and asserts that it's an bool\nfunc (c *Collection) GetBoolProperty(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) bool {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetStringProperty gets property and asserts that it's an string\nfunc (c *Collection) GetStringProperty(key dynamicproperties.StringKey) dynamicproperties.StringPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) string {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetStringValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultString()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetMapProperty gets property and asserts that it's a map\nfunc (c *Collection) GetMapProperty(key dynamicproperties.MapKey) dynamicproperties.MapPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetMapValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultMap()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetMapPropertyFilteredByDomain gets property with domain filter and asserts that it's a map\nfunc (c *Collection) GetMapPropertyFilteredByDomain(key dynamicproperties.MapKey) dynamicproperties.MapPropertyFnWithDomainFilter {\n\treturn func(domain string) map[string]interface{} {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainFilter(domain))\n\t\tval, err := c.client.GetMapValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultMap()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetStringPropertyFilteredByDomain gets property with domain filter and asserts that it's a string\nfunc (c *Collection) GetStringPropertyFilteredByDomain(key dynamicproperties.StringKey) dynamicproperties.StringPropertyFnWithDomainFilter {\n\treturn func(domain string) string {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainFilter(domain))\n\t\tval, err := c.client.GetStringValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultString()\n\t\t}\n\t\treturn val\n\t}\n}\n\nfunc (c *Collection) GetStringPropertyFilteredByTaskListInfo(key dynamicproperties.StringKey) dynamicproperties.StringPropertyFnWithTaskListInfoFilters {\n\treturn func(domain string, taskList string, taskType int) string {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domain),\n\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\tdynamicproperties.TaskTypeFilter(taskType),\n\t\t)\n\t\tval, err := c.client.GetStringValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultString()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetStringPropertyFilteredByNamespace gets property with domain filter and asserts that it's a string\nfunc (c *Collection) GetStringPropertyFilteredByNamespace(key dynamicproperties.StringKey) dynamicproperties.StringPropertyFnWithNamespaceFilters {\n\treturn func(namespace string) string {\n\t\tfilters := c.toFilterMap(dynamicproperties.NamespaceFilter(namespace))\n\t\tval, err := c.client.GetStringValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultString()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolPropertyFilteredByDomain gets property with domain filter and asserts that it's a bool\nfunc (c *Collection) GetBoolPropertyFilteredByDomain(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFnWithDomainFilter {\n\treturn func(domain string) bool {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainFilter(domain))\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolPropertyFilteredByDomainID gets property with domainID filter and asserts that it's a bool\nfunc (c *Collection) GetBoolPropertyFilteredByDomainID(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFnWithDomainIDFilter {\n\treturn func(domainID string) bool {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainIDFilter(domainID))\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolPropertyFilteredByDomainIDAndWorkflowID gets property with domainID and workflowID filters and asserts that it's a bool\nfunc (c *Collection) GetBoolPropertyFilteredByDomainIDAndWorkflowID(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFnWithDomainIDAndWorkflowIDFilter {\n\treturn func(domainID string, workflowID string) bool {\n\t\tfilters := c.toFilterMap(dynamicproperties.DomainIDFilter(domainID), dynamicproperties.WorkflowIDFilter(workflowID))\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\n// GetBoolPropertyFilteredByTaskListInfo gets property with taskListInfo as filters and asserts that it's an bool\nfunc (c *Collection) GetBoolPropertyFilteredByTaskListInfo(key dynamicproperties.BoolKey) dynamicproperties.BoolPropertyFnWithTaskListInfoFilters {\n\treturn func(domain string, taskList string, taskType int) bool {\n\t\tfilters := c.toFilterMap(\n\t\t\tdynamicproperties.DomainFilter(domain),\n\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\tdynamicproperties.TaskTypeFilter(taskType),\n\t\t)\n\t\tval, err := c.client.GetBoolValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultBool()\n\t\t}\n\t\treturn val\n\t}\n}\n\nfunc (c *Collection) GetListProperty(key dynamicproperties.ListKey) dynamicproperties.ListPropertyFn {\n\treturn func(opts ...dynamicproperties.FilterOption) []interface{} {\n\t\tfilters := c.toFilterMap(opts...)\n\t\tval, err := c.client.GetListValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultList()\n\t\t}\n\t\treturn val\n\t}\n}\n\nfunc (c *Collection) GetStringPropertyFilteredByRatelimitKey(key dynamicproperties.StringKey) dynamicproperties.StringPropertyWithRatelimitKeyFilter {\n\treturn func(ratelimitKey string) string {\n\t\tfilters := c.toFilterMap(dynamicproperties.RatelimitKeyFilter(ratelimitKey))\n\t\tval, err := c.client.GetStringValue(\n\t\t\tkey,\n\t\t\tfilters,\n\t\t)\n\t\tif err != nil {\n\t\t\tc.logError(key, filters, err)\n\t\t\treturn key.DefaultString()\n\t\t}\n\t\treturn val\n\t}\n}\n\nfunc (c *Collection) toFilterMap(opts ...dynamicproperties.FilterOption) map[dynamicproperties.Filter]interface{} {\n\tl := len(opts)\n\tm := make(map[dynamicproperties.Filter]interface{}, l)\n\tfor _, opt := range opts {\n\t\topt(m)\n\t}\n\tfor _, opt := range c.filterOptions {\n\t\topt(m)\n\t}\n\treturn m\n}\n\nfunc getFilteredKeyAsString(\n\tkey dynamicproperties.Key,\n\tfilters map[dynamicproperties.Filter]interface{},\n) string {\n\tvar sb strings.Builder\n\tsb.WriteString(key.String())\n\tfor filter := dynamicproperties.UnknownFilter + 1; filter < dynamicproperties.LastFilterTypeForTest; filter++ {\n\t\tif value, ok := filters[filter]; ok {\n\t\t\tsb.WriteString(fmt.Sprintf(\",%v:%v\", filter.String(), value))\n\t\t}\n\t}\n\treturn sb.String()\n}\n"
  },
  {
    "path": "common/dynamicconfig/config_benchmark_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nfunc BenchmarkGetIntProperty(b *testing.B) {\n\tclient := NewInMemoryClient()\n\tcln := NewCollection(client, log.NewNoop())\n\tkey := dynamicproperties.MatchingMaxTaskBatchSize\n\tfor i := 0; i < b.N; i++ {\n\t\tsize := cln.GetIntProperty(key)\n\t\tassert.Equal(b, 100, size())\n\t}\n}\n"
  },
  {
    "path": "common/dynamicconfig/config_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype configSuite struct {\n\tsuite.Suite\n\tclient *inMemoryClient\n\tcln    *Collection\n}\n\nfunc TestConfigSuite(t *testing.T) {\n\ts := new(configSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *configSuite) SetupTest() {\n\ts.client = NewInMemoryClient().(*inMemoryClient)\n\tlogger := log.NewNoop()\n\ts.cln = NewCollection(s.client, logger)\n}\n\nfunc (s *configSuite) TestGetProperty() {\n\tkey := dynamicproperties.TestGetStringPropertyKey\n\tvalue := s.cln.GetProperty(key)\n\ts.Equal(key.DefaultValue(), value())\n\ts.client.SetValue(key, \"b\")\n\ts.Equal(\"b\", value())\n}\n\nfunc (s *configSuite) TestGetStringProperty() {\n\tkey := dynamicproperties.TestGetStringPropertyKey\n\tvalue := s.cln.GetStringProperty(key)\n\ts.Equal(key.DefaultValue(), value())\n\ts.client.SetValue(key, \"b\")\n\ts.Equal(\"b\", value())\n}\n\nfunc (s *configSuite) TestGetIntProperty() {\n\tkey := dynamicproperties.TestGetIntPropertyKey\n\tvalue := s.cln.GetIntProperty(key)\n\ts.Equal(key.DefaultInt(), value())\n\ts.client.SetValue(key, 50)\n\ts.Equal(50, value())\n}\n\nfunc (s *configSuite) TestGetIntPropertyFilteredByDomain() {\n\tkey := dynamicproperties.TestGetIntPropertyFilteredByDomainKey\n\tdomain := \"testDomain\"\n\tvalue := s.cln.GetIntPropertyFilteredByDomain(key)\n\ts.Equal(key.DefaultInt(), value(domain))\n\ts.client.SetValue(key, 50)\n\ts.Equal(50, value(domain))\n}\n\nfunc (s *configSuite) TestGetIntPropertyFilteredByWorkflowType() {\n\tkey := dynamicproperties.TestGetIntPropertyFilteredByWorkflowTypeKey\n\tdomain := \"testDomain\"\n\tworkflowType := \"testWorkflowType\"\n\tvalue := s.cln.GetIntPropertyFilteredByWorkflowType(key)\n\ts.Equal(key.DefaultInt(), value(domain, workflowType))\n\ts.client.SetValue(key, 50)\n\ts.Equal(50, value(domain, workflowType))\n}\n\nfunc (s *configSuite) TestGetIntPropertyFilteredByShardID() {\n\tkey := dynamicproperties.TestGetIntPropertyFilteredByShardIDKey\n\tshardID := 1\n\tvalue := s.cln.GetIntPropertyFilteredByShardID(key)\n\ts.Equal(key.DefaultInt(), value(shardID))\n\ts.client.SetValue(key, 10)\n\ts.Equal(10, value(shardID))\n}\n\nfunc (s *configSuite) TestGetStringPropertyFnWithDomainFilter() {\n\tkey := dynamicproperties.DefaultEventEncoding\n\tdomain := \"testDomain\"\n\tvalue := s.cln.GetStringPropertyFilteredByDomain(key)\n\ts.Equal(key.DefaultString(), value(domain))\n\ts.client.SetValue(key, \"efg\")\n\ts.Equal(\"efg\", value(domain))\n}\n\nfunc (s *configSuite) TestGetStringPropertyFnByTaskListInfo() {\n\tkey := dynamicproperties.TasklistLoadBalancerStrategy\n\tdomain := \"testDomain\"\n\ttaskList := \"testTaskList\"\n\ttaskType := 0\n\tvalue := s.cln.GetStringPropertyFilteredByTaskListInfo(key)\n\ts.Equal(key.DefaultString(), value(domain, taskList, taskType))\n\ts.client.SetValue(key, \"round-robin\")\n\ts.Equal(\"round-robin\", value(domain, taskList, taskType))\n}\n\nfunc (s *configSuite) TestGetStringPropertyFnWithNamespaceFilter() {\n\tkey := dynamicproperties.ShardDistributorMigrationMode\n\tnamespace := \"testService\"\n\tvalue := s.cln.GetStringPropertyFilteredByNamespace(key)\n\ts.Equal(key.DefaultString(), value(namespace))\n\ts.client.SetValue(key, \"efg\")\n\ts.Equal(\"efg\", value(namespace))\n}\n\nfunc (s *configSuite) TestGetStringPropertyFilteredByRatelimitKey() {\n\tkey := dynamicproperties.FrontendGlobalRatelimiterMode\n\tratelimitKey := \"user:testDomain\"\n\tvalue := s.cln.GetStringPropertyFilteredByRatelimitKey(key)\n\ts.Equal(key.DefaultString(), value(ratelimitKey))\n\ts.client.SetValue(key, \"fake-mode\")\n\ts.Equal(\"fake-mode\", value(ratelimitKey))\n}\n\nfunc (s *configSuite) TestGetIntPropertyFilteredByTaskListInfo() {\n\tkey := dynamicproperties.TestGetIntPropertyFilteredByTaskListInfoKey\n\tdomain := \"testDomain\"\n\ttaskList := \"testTaskList\"\n\ttaskType := 0\n\tvalue := s.cln.GetIntPropertyFilteredByTaskListInfo(key)\n\ts.Equal(key.DefaultInt(), value(domain, taskList, taskType))\n\ts.client.SetValue(key, 50)\n\ts.Equal(50, value(domain, taskList, taskType))\n}\n\nfunc (s *configSuite) TestGetFloat64Property() {\n\tkey := dynamicproperties.TestGetFloat64PropertyKey\n\tvalue := s.cln.GetFloat64Property(key)\n\ts.Equal(key.DefaultFloat(), value())\n\ts.client.SetValue(key, 0.01)\n\ts.Equal(0.01, value())\n}\n\nfunc (s *configSuite) TestGetFloat64PropertyFilteredByShardID() {\n\tkey := dynamicproperties.TestGetFloat64PropertyFilteredByShardIDKey\n\tshardID := 1\n\tvalue := s.cln.GetFloat64PropertyFilteredByShardID(key)\n\ts.Equal(key.DefaultFloat(), value(shardID))\n\ts.client.SetValue(key, 0.01)\n\ts.Equal(0.01, value(shardID))\n}\n\nfunc (s *configSuite) TestGetBoolProperty() {\n\tkey := dynamicproperties.TestGetBoolPropertyKey\n\tvalue := s.cln.GetBoolProperty(key)\n\ts.Equal(key.DefaultBool(), value())\n\ts.client.SetValue(key, false)\n\ts.Equal(false, value())\n}\n\nfunc (s *configSuite) TestGetBoolPropertyFilteredByDomainID() {\n\tkey := dynamicproperties.TestGetBoolPropertyFilteredByDomainIDKey\n\tdomainID := \"testDomainID\"\n\tvalue := s.cln.GetBoolPropertyFilteredByDomainID(key)\n\ts.Equal(key.DefaultBool(), value(domainID))\n\ts.client.SetValue(key, false)\n\ts.Equal(false, value(domainID))\n}\n\nfunc (s *configSuite) TestGetBoolPropertyFilteredByDomain() {\n\tkey := dynamicproperties.TestGetBoolPropertyFilteredByDomainKey\n\tdomain := \"testDomain\"\n\tvalue := s.cln.GetBoolPropertyFilteredByDomain(key)\n\ts.Equal(key.DefaultBool(), value(domain))\n\ts.client.SetValue(key, true)\n\ts.Equal(true, value(domain))\n}\n\nfunc (s *configSuite) TestGetBoolPropertyFilteredByDomainIDAndWorkflowID() {\n\tkey := dynamicproperties.TestGetBoolPropertyFilteredByDomainIDAndWorkflowIDKey\n\tdomainID := \"testDomainID\"\n\tworkflowID := \"testWorkflowID\"\n\tvalue := s.cln.GetBoolPropertyFilteredByDomainIDAndWorkflowID(key)\n\ts.Equal(key.DefaultBool(), value(domainID, workflowID))\n\ts.client.SetValue(key, true)\n\ts.Equal(true, value(domainID, workflowID))\n}\n\nfunc (s *configSuite) TestGetBoolPropertyFilteredByTaskListInfo() {\n\tkey := dynamicproperties.TestGetBoolPropertyFilteredByTaskListInfoKey\n\tdomain := \"testDomain\"\n\ttaskList := \"testTaskList\"\n\ttaskType := 0\n\tvalue := s.cln.GetBoolPropertyFilteredByTaskListInfo(key)\n\ts.Equal(key.DefaultBool(), value(domain, taskList, taskType))\n\ts.client.SetValue(key, true)\n\ts.Equal(true, value(domain, taskList, taskType))\n}\n\nfunc (s *configSuite) TestGetBoolPropertyFilteredByShardID() {\n\tkey := dynamicproperties.TestGetBoolPropertyFilteredByShardIDKey\n\tshardID := 1\n\tvalue := s.cln.GetBoolPropertyFilteredByShardID(key)\n\ts.Equal(key.DefaultBool(), value(shardID))\n\ts.client.SetValue(key, true)\n\ts.Equal(true, value(shardID))\n}\n\nfunc (s *configSuite) TestGetDurationProperty() {\n\tkey := dynamicproperties.TestGetDurationPropertyKey\n\tvalue := s.cln.GetDurationProperty(key)\n\ts.Equal(key.DefaultDuration(), value())\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value())\n}\n\nfunc (s *configSuite) TestGetDurationPropertyFilteredByDomain() {\n\tkey := dynamicproperties.TestGetDurationPropertyFilteredByDomainKey\n\tdomain := \"testDomain\"\n\tvalue := s.cln.GetDurationPropertyFilteredByDomain(key)\n\ts.Equal(key.DefaultDuration(), value(domain))\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value(domain))\n}\n\nfunc (s *configSuite) TestGetDurationPropertyFilteredByDomainID() {\n\tkey := dynamicproperties.TestGetDurationPropertyFilteredByDomainIDKey\n\tdomain := \"testDomainID\"\n\tvalue := s.cln.GetDurationPropertyFilteredByDomainID(key)\n\ts.Equal(key.DefaultDuration(), value(domain))\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value(domain))\n}\n\nfunc (s *configSuite) TestGetDurationPropertyFilteredByShardID() {\n\tkey := dynamicproperties.TestGetDurationPropertyFilteredByShardID\n\tshardID := 1\n\tvalue := s.cln.GetDurationPropertyFilteredByShardID(key)\n\ts.Equal(key.DefaultDuration(), value(shardID))\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value(shardID))\n}\n\nfunc (s *configSuite) TestGetDurationPropertyFilteredByTaskListInfo() {\n\tkey := dynamicproperties.TestGetDurationPropertyFilteredByTaskListInfoKey\n\tdomain := \"testDomain\"\n\ttaskList := \"testTaskList\"\n\ttaskType := 0\n\tvalue := s.cln.GetDurationPropertyFilteredByTaskListInfo(key)\n\ts.Equal(key.DefaultDuration(), value(domain, taskList, taskType))\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value(domain, taskList, taskType))\n}\n\nfunc (s *configSuite) TestGetDurationPropertyFilteredByWorkflowType() {\n\tkey := dynamicproperties.TestGetDurationPropertyFilteredByWorkflowTypeKey\n\tdomain := \"testDomain\"\n\tworkflowType := \"testWorkflowType\"\n\tvalue := s.cln.GetDurationPropertyFilteredByWorkflowType(key)\n\ts.Equal(key.DefaultDuration(), value(domain, workflowType))\n\ts.client.SetValue(key, time.Minute)\n\ts.Equal(time.Minute, value(domain, workflowType))\n}\n\nfunc (s *configSuite) TestGetMapProperty() {\n\tkey := dynamicproperties.TestGetMapPropertyKey\n\tval := map[string]interface{}{\n\t\t\"testKey\": 123,\n\t}\n\tvalue := s.cln.GetMapProperty(key)\n\ts.Equal(key.DefaultMap(), value())\n\tval[\"testKey\"] = \"321\"\n\ts.client.SetValue(key, val)\n\ts.Equal(val, value())\n\ts.Equal(\"321\", value()[\"testKey\"])\n}\n\nfunc (s *configSuite) TestGetMapPropertyFilteredByDomain() {\n\tkey := dynamicproperties.TestGetMapPropertyKey\n\tdomain := \"testDomain\"\n\tval := map[string]interface{}{\n\t\t\"testKey\": 123,\n\t}\n\tvalue := s.cln.GetMapPropertyFilteredByDomain(key)\n\ts.Equal(key.DefaultMap(), value(domain))\n\tval[\"testKey\"] = \"321\"\n\ts.client.SetValue(key, val)\n\ts.Equal(val, value(domain))\n\ts.Equal(\"321\", value(domain)[\"testKey\"])\n}\n\nfunc (s *configSuite) TestGetListProperty() {\n\tkey := dynamicproperties.TestGetListPropertyKey\n\tarr := []interface{}{}\n\tvalue := s.cln.GetListProperty(key)\n\ts.Equal(key.DefaultList(), value())\n\tarr = append(arr, 1)\n\ts.client.SetValue(key, arr)\n\ts.Equal(1, len(value()))\n\ts.Equal(1, value()[0])\n}\n\nfunc (s *configSuite) TestUpdateConfig() {\n\tkey := dynamicproperties.TestGetBoolPropertyKey\n\tvalue := s.cln.GetBoolProperty(key)\n\terr := s.client.UpdateValue(key, false)\n\ts.NoError(err)\n\ts.Equal(false, value())\n\terr = s.client.UpdateValue(key, true)\n\ts.NoError(err)\n\ts.Equal(true, value())\n}\n\nfunc TestDynamicConfigKeyIsMapped(t *testing.T) {\n\tfor i := dynamicproperties.UnknownIntKey + 1; i < dynamicproperties.LastIntKey; i++ {\n\t\tkey, ok := dynamicproperties.IntKeys[i]\n\t\trequire.True(t, ok, \"missing IntKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty IntKey: %d\", i)\n\t}\n\tfor i := dynamicproperties.UnknownBoolKey + 1; i < dynamicproperties.LastBoolKey; i++ {\n\t\tkey, ok := dynamicproperties.BoolKeys[i]\n\t\trequire.True(t, ok, \"missing BoolKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty BoolKey: %d\", i)\n\t}\n\tfor i := dynamicproperties.UnknownFloatKey + 1; i < dynamicproperties.LastFloatKey; i++ {\n\t\tkey, ok := dynamicproperties.FloatKeys[i]\n\t\trequire.True(t, ok, \"missing FloatKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty FloatKey: %d\", i)\n\t}\n\tfor i := dynamicproperties.UnknownStringKey + 1; i < dynamicproperties.LastStringKey; i++ {\n\t\tkey, ok := dynamicproperties.StringKeys[i]\n\t\trequire.True(t, ok, \"missing StringKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty StringKey: %d\", i)\n\t}\n\tfor i := dynamicproperties.UnknownDurationKey + 1; i < dynamicproperties.LastDurationKey; i++ {\n\t\tkey, ok := dynamicproperties.DurationKeys[i]\n\t\trequire.True(t, ok, \"missing DurationKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty DurationKey: %d\", i)\n\t}\n\tfor i := dynamicproperties.UnknownMapKey + 1; i < dynamicproperties.LastMapKey; i++ {\n\t\tkey, ok := dynamicproperties.MapKeys[i]\n\t\trequire.True(t, ok, \"missing MapKey: %d\", i)\n\t\trequire.NotEmpty(t, key, \"empty MapKey: %d\", i)\n\t}\n}\n"
  },
  {
    "path": "common/dynamicconfig/configstore/config/config.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport \"time\"\n\n// This package is necessary to avoid import cycle as config_store_client imports common/config\n// while common/config imports this ClientConfig definition\n\n// ClientConfig is the config for the config store based dynamic config client.\n// It specifies how often the cached config should be updated by checking underlying database.\ntype ClientConfig struct {\n\tPollInterval        time.Duration `yaml:\"pollInterval\"`\n\tUpdateRetryAttempts int           `yaml:\"updateRetryAttempts\"`\n\tFetchTimeout        time.Duration `yaml:\"FetchTimeout\"`\n\tUpdateTimeout       time.Duration `yaml:\"UpdateTimeout\"`\n}\n"
  },
  {
    "path": "common/dynamicconfig/configstore/config_store_client.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage configstore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig\"\n\tcsc \"github.com/uber/cadence/common/dynamicconfig/configstore/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination configstore_mock.go -self_package github.com/uber/cadence/common/dynamicconfig/configstore\n\nvar _ dc.Client = (*configStoreClient)(nil)\n\n// Client is a stateful config store\ntype Client interface {\n\tcommon.Daemon\n\tdc.Client\n}\n\nconst (\n\tconfigStoreMinPollInterval = time.Second * 2\n)\n\nvar defaultConfigValues = &csc.ClientConfig{\n\tPollInterval:        time.Second * 10,\n\tUpdateRetryAttempts: 1,\n\tFetchTimeout:        2,\n\tUpdateTimeout:       2,\n}\n\ntype configStoreClient struct {\n\tstatus             int32\n\tconfigStoreType    persistence.ConfigType\n\tvalues             atomic.Value\n\tlastUpdatedTime    time.Time\n\tconfig             *csc.ClientConfig\n\tconfigStoreManager persistence.ConfigStoreManager\n\tdoneCh             chan struct{}\n\tlogger             log.Logger\n}\n\ntype cacheEntry struct {\n\tcacheVersion  int64\n\tschemaVersion int64\n\tdcEntries     map[string]*types.DynamicConfigEntry\n}\n\n// NewConfigStoreClient creates a config store client\nfunc NewConfigStoreClient(\n\tclientCfg *csc.ClientConfig,\n\tpersistenceCfg *config.Persistence,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconfigType persistence.ConfigType,\n) (Client, error) {\n\tif persistenceCfg == nil {\n\t\treturn nil, errors.New(\"persistence cfg is nil\")\n\t}\n\n\tds, ok := persistenceCfg.DataStores[persistenceCfg.DefaultStore]\n\tif !ok {\n\t\treturn nil, errors.New(\"default persistence config missing\")\n\t}\n\n\tif err := validateClientConfig(clientCfg); err != nil {\n\t\tlogger.Warn(\"invalid ClientConfig values, using default values\")\n\t\tclientCfg = defaultConfigValues\n\t}\n\n\tclient, err := newConfigStoreClient(clientCfg, &ds, logger, metricsClient, configType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\terr = client.startUpdate()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn client, nil\n}\n\nfunc newConfigStoreClient(\n\tclientCfg *csc.ClientConfig,\n\tds *config.DataStore,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconfigType persistence.ConfigType,\n) (*configStoreClient, error) {\n\tvar store persistence.ConfigStore\n\tvar err error\n\tswitch {\n\tcase ds.ShardedNoSQL != nil:\n\t\tstore, err = nosql.NewNoSQLConfigStore(*ds.ShardedNoSQL, logger, metricsClient, nil)\n\tcase ds.NoSQL != nil:\n\t\tstore, err = nosql.NewNoSQLConfigStore(*ds.NoSQL.ConvertToShardedNoSQLConfig(), logger, metricsClient, nil)\n\tcase ds.SQL != nil:\n\t\tvar db sqlplugin.DB\n\t\tdb, err = sql.NewSQLDB(ds.SQL)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstore, err = sql.NewSQLConfigStore(db, logger, nil)\n\tdefault:\n\t\treturn nil, errors.New(\"both NoSQL and SQL store are not provided\")\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdoneCh := make(chan struct{})\n\tclient := &configStoreClient{\n\t\tstatus:             common.DaemonStatusStarted,\n\t\tconfig:             clientCfg,\n\t\tdoneCh:             doneCh,\n\t\tconfigStoreManager: persistence.NewConfigStoreManagerImpl(store, logger),\n\t\tlogger:             logger,\n\t\tconfigStoreType:    configType,\n\t}\n\n\treturn client, nil\n}\n\nfunc (csc *configStoreClient) startUpdate() error {\n\tif err := csc.update(); err != nil {\n\t\treturn err\n\t}\n\tgo func() {\n\t\tticker := time.NewTicker(csc.config.PollInterval)\n\t\tfor {\n\t\t\tselect {\n\t\t\tcase <-ticker.C:\n\t\t\t\terr := csc.update()\n\t\t\t\tif err != nil {\n\t\t\t\t\tcsc.logger.Error(\"Failed to update dynamic config\", tag.Error(err))\n\t\t\t\t}\n\t\t\tcase <-csc.doneCh:\n\t\t\t\tticker.Stop()\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n\treturn nil\n}\n\nfunc (csc *configStoreClient) GetValue(name dynamicproperties.Key) (interface{}, error) {\n\treturn csc.getValueWithFilters(name, nil, name.DefaultValue())\n}\n\nfunc (csc *configStoreClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error) {\n\treturn csc.getValueWithFilters(name, filters, name.DefaultValue())\n}\n\nfunc (csc *configStoreClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error) {\n\tdefaultValue := name.DefaultInt()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tfloatVal, ok := val.(float64)\n\tif !ok {\n\t\treturn defaultValue, errors.New(\"value type is not int\")\n\t}\n\n\tif floatVal != math.Trunc(floatVal) {\n\t\treturn defaultValue, errors.New(\"value type is not int\")\n\t}\n\n\treturn int(floatVal), nil\n}\n\nfunc (csc *configStoreClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error) {\n\tdefaultValue := name.DefaultFloat()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif floatVal, ok := val.(float64); ok {\n\t\treturn floatVal, nil\n\t}\n\treturn defaultValue, errors.New(\"value type is not float64\")\n}\n\nfunc (csc *configStoreClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error) {\n\tdefaultValue := name.DefaultBool()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif boolVal, ok := val.(bool); ok {\n\t\treturn boolVal, nil\n\t}\n\treturn defaultValue, errors.New(\"value type is not bool\")\n}\n\nfunc (csc *configStoreClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error) {\n\tdefaultValue := name.DefaultString()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif stringVal, ok := val.(string); ok {\n\t\treturn stringVal, nil\n\t}\n\treturn defaultValue, errors.New(\"value type is not string\")\n}\n\n// Note that all number types (ex: ints) will be returned as float64.\n// It is the caller's responsibility to convert based on their context for value type.\nfunc (csc *configStoreClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error) {\n\tdefaultValue := name.DefaultMap()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\tif mapVal, ok := val.(map[string]interface{}); ok {\n\t\treturn mapVal, nil\n\t}\n\treturn defaultValue, errors.New(\"value type is not map\")\n}\n\nfunc (csc *configStoreClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error) {\n\tdefaultValue := name.DefaultDuration()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tvar durVal time.Duration\n\tswitch v := val.(type) {\n\tcase string:\n\t\tdurVal, err = time.ParseDuration(v)\n\t\tif err != nil {\n\t\t\treturn defaultValue, errors.New(\"value string encoding cannot be parsed into duration\")\n\t\t}\n\tcase time.Duration:\n\t\tdurVal = v\n\tdefault:\n\t\treturn defaultValue, errors.New(\"value type is not duration\")\n\t}\n\n\treturn durVal, nil\n}\nfunc (csc *configStoreClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error) {\n\tdefaultValue := name.DefaultList()\n\tval, err := csc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\tif listVal, ok := val.([]interface{}); ok {\n\t\treturn listVal, nil\n\t}\n\treturn defaultValue, errors.New(\"value type is not list\")\n}\n\nfunc (csc *configStoreClient) UpdateValue(name dynamicproperties.Key, value interface{}) error {\n\tdcValues, ok := value.([]*types.DynamicConfigValue)\n\tif !ok && value != nil {\n\t\treturn errors.New(\"invalid value\")\n\t}\n\treturn csc.updateValue(name, dcValues, csc.config.UpdateRetryAttempts)\n}\n\nfunc (csc *configStoreClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error {\n\t// if empty filter provided, update fallback value.\n\t// if u want to remove entire entry, just do update value with empty\n\tloaded := csc.values.Load()\n\tif loaded == nil {\n\t\treturn dc.NotFoundError\n\t}\n\tcurrentCached := loaded.(cacheEntry)\n\n\tif currentCached.dcEntries == nil {\n\t\treturn dc.NotFoundError\n\t}\n\n\tval, ok := currentCached.dcEntries[name.String()]\n\tif !ok {\n\t\treturn dc.NotFoundError\n\t}\n\n\tnewValues := make([]*types.DynamicConfigValue, 0, len(val.Values))\n\tif filters == nil {\n\t\tfor _, dcValue := range val.Values {\n\t\t\tif dcValue.Filters != nil || len(dcValue.Filters) != 0 {\n\t\t\t\tnewValues = append(newValues, dcValue.Copy())\n\t\t\t}\n\t\t}\n\t} else {\n\t\tfor _, dcValue := range val.Values {\n\t\t\tif !matchFilters(dcValue, filters) || dcValue.Filters == nil || len(dcValue.Filters) == 0 {\n\t\t\t\tnewValues = append(newValues, dcValue.Copy())\n\t\t\t}\n\t\t}\n\t}\n\n\treturn csc.updateValue(name, newValues, csc.config.UpdateRetryAttempts)\n}\n\nfunc (csc *configStoreClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\tvar resList []*types.DynamicConfigEntry\n\n\tloaded := csc.values.Load()\n\tif loaded == nil {\n\t\treturn nil, nil\n\t}\n\tcurrentCached := loaded.(cacheEntry)\n\n\tif currentCached.dcEntries == nil {\n\t\treturn nil, nil\n\t}\n\tlistAll := false\n\tif name == nil {\n\t\t// if key is not specified, return all entries\n\t\tlistAll = true\n\t} else if _, ok := currentCached.dcEntries[name.String()]; !ok {\n\t\t// if key is not known, return all entries\n\t\tlistAll = true\n\t}\n\tif listAll {\n\t\t// if key is not known/specified, return all entries\n\t\tresList = make([]*types.DynamicConfigEntry, 0, len(currentCached.dcEntries))\n\t\tfor _, entry := range currentCached.dcEntries {\n\t\t\tresList = append(resList, entry.Copy())\n\t\t}\n\t} else {\n\t\t// if key is known, return just that specific entry\n\t\tresList = make([]*types.DynamicConfigEntry, 0, 1)\n\t\tresList = append(resList, currentCached.dcEntries[name.String()])\n\t}\n\n\treturn resList, nil\n}\n\nfunc (csc *configStoreClient) Stop() {\n\tif !atomic.CompareAndSwapInt32(&csc.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(csc.doneCh)\n\tcsc.configStoreManager.Close()\n}\n\nfunc (csc *configStoreClient) Start() {\n\terr := csc.startUpdate()\n\tif err != nil {\n\t\tcsc.logger.Fatal(\"could not start config store\", tag.Error(err))\n\t}\n\tif !atomic.CompareAndSwapInt32(&csc.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n}\n\nfunc (csc *configStoreClient) updateValue(name dynamicproperties.Key, dcValues []*types.DynamicConfigValue, retryAttempts int) error {\n\t// since values are not unique, no way to know if you are trying to update a specific value\n\t// or if you want to add another of the same value with different filters.\n\t// UpdateValue will replace everything associated with dc key.\n\tfor _, dcValue := range dcValues {\n\t\tif err := validateKeyDataBlobPair(name, dcValue.Value); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tloaded := csc.values.Load()\n\tvar currentCached cacheEntry\n\tif loaded == nil {\n\t\tcurrentCached = cacheEntry{\n\t\t\tcacheVersion:  0,\n\t\t\tschemaVersion: 0,\n\t\t\tdcEntries:     map[string]*types.DynamicConfigEntry{},\n\t\t}\n\t} else {\n\t\tcurrentCached = loaded.(cacheEntry)\n\t}\n\n\tkeyName := name.String()\n\tvar newEntries []*types.DynamicConfigEntry\n\n\texistingEntry, entryExists := currentCached.dcEntries[keyName]\n\n\tif len(dcValues) == 0 {\n\t\tnewEntries = make([]*types.DynamicConfigEntry, 0, len(currentCached.dcEntries))\n\n\t\tfor _, entry := range currentCached.dcEntries {\n\t\t\tif entryExists && entry == existingEntry {\n\t\t\t\tcontinue\n\t\t\t} else {\n\t\t\t\tnewEntries = append(newEntries, entry.Copy())\n\t\t\t}\n\t\t}\n\t} else {\n\t\tif entryExists {\n\t\t\tnewEntries = make([]*types.DynamicConfigEntry, 0, len(currentCached.dcEntries))\n\t\t} else {\n\t\t\tnewEntries = make([]*types.DynamicConfigEntry, 0, len(currentCached.dcEntries)+1)\n\t\t\tnewEntries = append(newEntries,\n\t\t\t\t&types.DynamicConfigEntry{\n\t\t\t\t\tName:   keyName,\n\t\t\t\t\tValues: dcValues,\n\t\t\t\t})\n\t\t}\n\n\t\tfor _, entry := range currentCached.dcEntries {\n\t\t\tif entryExists && entry.Name == keyName {\n\t\t\t\tnewEntries = append(newEntries,\n\t\t\t\t\t&types.DynamicConfigEntry{\n\t\t\t\t\t\tName:   keyName,\n\t\t\t\t\t\tValues: dcValues,\n\t\t\t\t\t})\n\t\t\t} else {\n\t\t\t\tnewEntries = append(newEntries, entry.Copy())\n\t\t\t}\n\t\t}\n\t}\n\n\tnewSnapshot := &persistence.DynamicConfigSnapshot{\n\t\tVersion: currentCached.cacheVersion + 1,\n\t\tValues: &types.DynamicConfigBlob{\n\t\t\tSchemaVersion: currentCached.schemaVersion,\n\t\t\tEntries:       newEntries,\n\t\t},\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), csc.config.UpdateTimeout)\n\tdefer cancel()\n\n\terr := csc.configStoreManager.UpdateDynamicConfig(\n\t\tctx,\n\t\t&persistence.UpdateDynamicConfigRequest{\n\t\t\tSnapshot: newSnapshot,\n\t\t}, csc.configStoreType,\n\t)\n\n\tselect {\n\tcase <-ctx.Done():\n\t\t// potentially we can retry on timeout\n\t\treturn errors.New(\"timeout error on update\")\n\tdefault:\n\t\tif err != nil {\n\t\t\tif _, ok := err.(*persistence.ConditionFailedError); ok && retryAttempts > 0 {\n\t\t\t\t// fetch new config and retry\n\t\t\t\terr := csc.update()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn csc.updateValue(name, dcValues, retryAttempts-1)\n\t\t\t}\n\n\t\t\tif retryAttempts == 0 {\n\t\t\t\treturn errors.New(\"ran out of retry attempts on update\")\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n}\n\nfunc (csc *configStoreClient) update() error {\n\tctx, cancel := context.WithTimeout(context.Background(), csc.config.FetchTimeout)\n\tdefer cancel()\n\n\tres, err := csc.configStoreManager.FetchDynamicConfig(ctx, csc.configStoreType)\n\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn errors.New(\"timeout error on fetch\")\n\tdefault:\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to fetch dynamic config snapshot %v\", err)\n\t\t}\n\n\t\tif res != nil && res.Snapshot != nil {\n\t\t\tdefer func() {\n\t\t\t\tcsc.lastUpdatedTime = time.Now()\n\t\t\t}()\n\n\t\t\treturn csc.storeValues(res.Snapshot)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (csc *configStoreClient) storeValues(snapshot *persistence.DynamicConfigSnapshot) error {\n\t// Converting the list of dynamic config entries into a map for better lookup performance\n\tvar dcEntryMap map[string]*types.DynamicConfigEntry\n\tif snapshot.Values.Entries == nil {\n\t\tdcEntryMap = nil\n\t} else {\n\t\tdcEntryMap = make(map[string]*types.DynamicConfigEntry)\n\t\tfor _, entry := range snapshot.Values.Entries {\n\t\t\tdcEntryMap[entry.Name] = entry\n\t\t}\n\t}\n\n\tcsc.values.Store(cacheEntry{\n\t\tcacheVersion:  snapshot.Version,\n\t\tschemaVersion: snapshot.Values.SchemaVersion,\n\t\tdcEntries:     dcEntryMap,\n\t})\n\tcsc.logger.Debug(\"Updated dynamic config\")\n\treturn nil\n}\n\nfunc (csc *configStoreClient) getValueWithFilters(key dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}, defaultValue interface{}) (interface{}, error) {\n\tkeyName := key.String()\n\tloaded := csc.values.Load()\n\tif loaded == nil {\n\t\treturn defaultValue, nil\n\t}\n\tcached := loaded.(cacheEntry)\n\tfound := false\n\n\tif entry, ok := cached.dcEntries[keyName]; ok && entry != nil {\n\t\tfor _, dcValue := range entry.Values {\n\t\t\tif len(dcValue.Filters) == 0 {\n\t\t\t\tparsedVal, err := convertFromDataBlob(dcValue.Value)\n\n\t\t\t\tif err == nil {\n\t\t\t\t\tdefaultValue = parsedVal\n\t\t\t\t\tfound = true\n\t\t\t\t}\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tif matchFilters(dcValue, filters) {\n\t\t\t\treturn convertFromDataBlob(dcValue.Value)\n\t\t\t}\n\t\t}\n\t}\n\tif found {\n\t\treturn defaultValue, nil\n\t}\n\treturn defaultValue, dc.NotFoundError\n}\n\nfunc matchFilters(dcValue *types.DynamicConfigValue, filters map[dynamicproperties.Filter]interface{}) bool {\n\tif len(dcValue.Filters) > len(filters) {\n\t\treturn false\n\t}\n\n\tfor _, valueFilter := range dcValue.Filters {\n\t\tfilterKey := dynamicproperties.ParseFilter(valueFilter.Name)\n\t\tif filters[filterKey] == nil {\n\t\t\treturn false\n\t\t}\n\n\t\trequestValue, err := convertFromDataBlob(valueFilter.Value)\n\t\tif err != nil || filters[filterKey] != requestValue {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc validateClientConfig(config *csc.ClientConfig) error {\n\tif config == nil {\n\t\treturn errors.New(\"no config found for config store based dynamic config client\")\n\t}\n\tif config.PollInterval < configStoreMinPollInterval {\n\t\treturn fmt.Errorf(\"poll interval should be at least %v\", configStoreMinPollInterval)\n\t}\n\tif config.UpdateRetryAttempts < 0 {\n\t\treturn errors.New(\"UpdateRetryAttempts must be non-negative\")\n\t}\n\tif config.FetchTimeout <= 0 {\n\t\treturn errors.New(\"FetchTimeout must be positive\")\n\t}\n\tif config.UpdateTimeout <= 0 {\n\t\treturn errors.New(\"UpdateTimeout must be positive\")\n\t}\n\treturn nil\n}\n\nfunc convertFromDataBlob(blob *types.DataBlob) (interface{}, error) {\n\tswitch *blob.EncodingType {\n\tcase types.EncodingTypeJSON:\n\t\tvar v interface{}\n\t\terr := json.Unmarshal(blob.Data, &v)\n\t\treturn v, err\n\tdefault:\n\t\treturn nil, errors.New(\"unsupported blob encoding\")\n\t}\n}\n\nfunc validateKeyDataBlobPair(key dynamicproperties.Key, blob *types.DataBlob) error {\n\tvalue, err := convertFromDataBlob(blob)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = fmt.Errorf(\"key value pair mismatch, key type: %T, value type: %T\", key, value)\n\tswitch key.(type) {\n\tcase dynamicproperties.IntKey:\n\t\tif _, ok := value.(int); !ok {\n\t\t\tfloatVal, ok := value.(float64)\n\t\t\tif !ok { // int can be decoded as float64\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif floatVal != math.Trunc(floatVal) {\n\t\t\t\treturn errors.New(\"value type is not int\")\n\t\t\t}\n\t\t}\n\tcase dynamicproperties.BoolKey:\n\t\tif _, ok := value.(bool); !ok {\n\t\t\treturn err\n\t\t}\n\tcase dynamicproperties.FloatKey:\n\t\tif _, ok := value.(float64); !ok {\n\t\t\treturn err\n\t\t}\n\tcase dynamicproperties.StringKey:\n\t\tif _, ok := value.(string); !ok {\n\t\t\treturn err\n\t\t}\n\tcase dynamicproperties.DurationKey:\n\t\tif _, ok := value.(time.Duration); !ok {\n\t\t\tdurationStr, ok := value.(string)\n\t\t\tif !ok {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif _, err = time.ParseDuration(durationStr); err != nil {\n\t\t\t\treturn errors.New(\"value string encoding cannot be parsed into duration\")\n\t\t\t}\n\t\t}\n\tcase dynamicproperties.MapKey:\n\t\tif _, ok := value.(map[string]interface{}); !ok {\n\t\t\treturn err\n\t\t}\n\tcase dynamicproperties.ListKey:\n\t\tif _, ok := value.([]interface{}); !ok {\n\t\t\treturn err\n\t\t}\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown key type: %T\", key)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/dynamicconfig/configstore/config_store_client_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage configstore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tc \"github.com/uber/cadence/common/dynamicconfig/configstore/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tretryAttempts = 2\n)\n\ntype configStoreClientSuite struct {\n\tsuite.Suite\n\t*require.Assertions\n\tclient         *configStoreClient\n\tmockManager    *p.MockConfigStoreManager\n\tmockController *gomock.Controller\n\tdoneCh         chan struct{}\n}\n\nvar snapshot1 *p.DynamicConfigSnapshot\n\nfunc TestConfigStoreClientSuite(t *testing.T) {\n\ts := new(configStoreClientSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *configStoreClientSuite) SetupSuite() {\n\ts.doneCh = make(chan struct{})\n\ts.mockController = gomock.NewController(s.T())\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(s.mockController)\n\tmockPlugin.EXPECT().\n\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(nil, nil).AnyTimes()\n\tnosql.RegisterPlugin(\"cassandra\", mockPlugin)\n}\n\nfunc (s *configStoreClientSuite) TearDownSuite() {\n\tclose(s.doneCh)\n}\n\nfunc (s *configStoreClientSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\tsnapshot1 = &p.DynamicConfigSnapshot{\n\t\tVersion: 1,\n\t\tValues: &types.DynamicConfigBlob{\n\t\t\tSchemaVersion: 1,\n\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetBoolPropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(false),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"global-samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetIntPropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(1000),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(1000.1),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"global-samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetFloat64PropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(12),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"wrong type\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetStringPropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"some random string\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"constrained-string\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"random tasklist\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetMapPropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData: jsonMarshalHelper(map[string]interface{}{\n\t\t\t\t\t\t\t\t\t\"key1\": \"1\",\n\t\t\t\t\t\t\t\t\t\"key2\": 1,\n\t\t\t\t\t\t\t\t\t\"key3\": []interface{}{\n\t\t\t\t\t\t\t\t\t\tfalse,\n\t\t\t\t\t\t\t\t\t\tmap[string]interface{}{\n\t\t\t\t\t\t\t\t\t\t\t\"key4\": true,\n\t\t\t\t\t\t\t\t\t\t\t\"key5\": 2.1,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"1\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"random tasklist\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName: dynamicproperties.TestGetDurationPropertyKey.String(),\n\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"1m\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"wrong duration string\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"longIdleTimeTaskList\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(2),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tconnections := make(map[string]config.DBShardConnection)\n\tconnections[config.NonShardedStoreName] = config.DBShardConnection{\n\t\tNoSQLPlugin: &config.NoSQL{\n\t\t\tPluginName: \"cassandra\",\n\t\t},\n\t}\n\n\tvar err error\n\ts.client, err = newConfigStoreClient(\n\t\t&c.ClientConfig{\n\t\t\tPollInterval:        time.Second * 2,\n\t\t\tUpdateRetryAttempts: retryAttempts,\n\t\t\tFetchTimeout:        time.Second * 1,\n\t\t\tUpdateTimeout:       time.Second * 1,\n\t\t},\n\t\t&config.DataStore{\n\t\t\tShardedNoSQL: &config.ShardedNoSQL{\n\t\t\t\tDefaultShard: config.NonShardedStoreName,\n\t\t\t\tConnections:  connections,\n\t\t\t},\n\t\t}, log.NewNoop(), metrics.NewNoopMetricsClient(), p.DynamicConfig)\n\ts.Require().NoError(err)\n\n\ts.mockManager = p.NewMockConfigStoreManager(s.mockController)\n\ts.client.configStoreManager = s.mockManager\n}\n\nfunc defaultTestSetup(s *configStoreClientSuite) {\n\ts.mockManager.EXPECT().\n\t\tFetchDynamicConfig(gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.FetchDynamicConfigResponse{\n\t\t\tSnapshot: snapshot1,\n\t\t}, nil).\n\t\tAnyTimes()\n\terr := s.client.startUpdate()\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestGetValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetValue(dynamicproperties.TestGetBoolPropertyKey)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetValue_NonExistKey() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetValue(dynamicproperties.MaxRetentionDays)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.MaxRetentionDays.DefaultInt(), v)\n\tv, err = s.client.GetValue(dynamicproperties.EnableVisibilitySampling)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.EnableVisibilitySampling.DefaultBool(), v)\n\tv, err = s.client.GetValue(dynamicproperties.FrontendErrorInjectionRate)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.FrontendErrorInjectionRate.DefaultFloat(), v)\n\tv, err = s.client.GetValue(dynamicproperties.WriteVisibilityStoreName)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.WriteVisibilityStoreName.DefaultString(), v)\n\tv, err = s.client.GetValue(dynamicproperties.FrontendShutdownDrainDuration)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.FrontendShutdownDrainDuration.DefaultDuration(), v)\n\tv, err = s.client.GetValue(dynamicproperties.RequiredDomainDataKeys)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.RequiredDomainDataKeys.DefaultMap(), v)\n}\n\nfunc (s *configStoreClientSuite) TestGetValueWithFilters() {\n\tdefaultTestSetup(s)\n\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"global-samples-domain\",\n\t}\n\n\tv, err := s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(true, v)\n\n\tfilters = map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"non-exist-domain\",\n\t}\n\tv, err = s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(false, v)\n\n\tfilters = map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\tdynamicproperties.TaskListName: \"non-exist-tasklist\",\n\t}\n\tv, err = s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(true, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetValueWithFilters_UnknownFilter() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:    \"global-samples-domain1\",\n\t\tdynamicproperties.UnknownFilter: \"unknown-filter1\",\n\t}\n\tv, err := s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetIntValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(1000, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetIntValue_FilterNotMatch() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(1000, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetIntValue_WrongType() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"global-samples-domain\",\n\t}\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetIntPropertyKey.DefaultInt(), v)\n}\n\nfunc (s *configStoreClientSuite) TestGetFloatValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetFloatValue(dynamicproperties.TestGetFloat64PropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(12.0, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetFloatValue_WrongType() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetFloatValue(dynamicproperties.TestGetFloat64PropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetFloat64PropertyKey.DefaultFloat(), v)\n}\n\nfunc (s *configStoreClientSuite) TestGetBoolValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetBoolValue(dynamicproperties.TestGetBoolPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetStringValue() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.TaskListName: \"random tasklist\",\n\t}\n\tv, err := s.client.GetStringValue(dynamicproperties.TestGetStringPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(\"constrained-string\", v)\n}\n\nfunc (s *configStoreClientSuite) TestGetMapValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetMapValue(dynamicproperties.TestGetMapPropertyKey, nil)\n\ts.NoError(err)\n\texpectedVal := map[string]interface{}{\n\t\t\"key1\": \"1\",\n\t\t\"key2\": float64(1),\n\t\t\"key3\": []interface{}{\n\t\t\tfalse,\n\t\t\tmap[string]interface{}{\n\t\t\t\t\"key4\": true,\n\t\t\t\t\"key5\": 2.1,\n\t\t\t},\n\t\t},\n\t}\n\ts.Equal(expectedVal, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetMapValue_WrongType() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.TaskListName: \"random tasklist\",\n\t}\n\tv, err := s.client.GetMapValue(dynamicproperties.TestGetMapPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetMapPropertyKey.DefaultMap(), v)\n}\n\nfunc (s *configStoreClientSuite) TestGetDurationValue() {\n\tdefaultTestSetup(s)\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(time.Minute, v)\n}\n\nfunc (s *configStoreClientSuite) TestGetDurationValue_NotStringRepresentation() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetDurationPropertyKey.DefaultDuration(), v)\n}\n\nfunc (s *configStoreClientSuite) TestGetDurationValue_ParseFailed() {\n\tdefaultTestSetup(s)\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\tdynamicproperties.TaskListName: \"longIdleTimeTaskList\",\n\t}\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetDurationPropertyKey.DefaultDuration(), v)\n}\n\nfunc (s *configStoreClientSuite) TestValidateConfig_InvalidConfig() {\n\terr := validateClientConfig(\n\t\t&c.ClientConfig{\n\t\t\tPollInterval:        time.Second * 1,\n\t\t\tUpdateRetryAttempts: 0,\n\t\t\tFetchTimeout:        time.Second * 3,\n\t\t\tUpdateTimeout:       time.Second * 4,\n\t\t},\n\t)\n\ts.Error(err)\n\n\terr = validateClientConfig(\n\t\t&c.ClientConfig{\n\t\t\tPollInterval:        time.Second * 2,\n\t\t\tUpdateRetryAttempts: -1,\n\t\t\tFetchTimeout:        time.Second * 2,\n\t\t\tUpdateTimeout:       time.Second * 2,\n\t\t},\n\t)\n\ts.Error(err)\n\n\terr = validateClientConfig(\n\t\t&c.ClientConfig{\n\t\t\tPollInterval:        time.Second * 2,\n\t\t\tUpdateRetryAttempts: 0,\n\t\t\tFetchTimeout:        time.Second * 0,\n\t\t\tUpdateTimeout:       time.Second * 0,\n\t\t},\n\t)\n\ts.Error(err)\n\n\terr = validateClientConfig(\n\t\t&c.ClientConfig{\n\t\t\tPollInterval:        time.Second * 2,\n\t\t\tUpdateRetryAttempts: 1,\n\t\t\tFetchTimeout:        time.Second * 1,\n\t\t\tUpdateTimeout:       time.Second * 0,\n\t\t},\n\t)\n\ts.Error(err)\n}\n\nfunc (s *configStoreClientSuite) TestMatchFilters() {\n\ttestCases := []struct {\n\t\tv       *types.DynamicConfigValue\n\t\tfilters map[dynamicproperties.Filter]interface{}\n\t\tmatched bool\n\t}{\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue:   nil,\n\t\t\t\tFilters: nil,\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName: \"some random domain\",\n\t\t\t},\n\t\t\tmatched: true,\n\t\t},\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue: nil,\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"some key\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"some value\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue: nil,\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName: \"some random domain\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue: nil,\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"sample-task-list\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: true,\n\t\t},\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue: nil,\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"some-other-filter\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"sample-task-list\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &types.DynamicConfigValue{\n\t\t\t\tValue: nil,\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t{\n\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         jsonMarshalHelper(\"samples-domain\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t}\n\n\tfor index, tc := range testCases {\n\t\tmatched := matchFilters(tc.v, tc.filters)\n\t\ts.Equal(tc.matched, matched, fmt.Sprintf(\"Test case %v failed\", index))\n\t}\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_NilOverwrite() {\n\tdefaultTestSetup(s)\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, request *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\tif request.Snapshot.Values.Entries[0].Name != dynamicproperties.TestGetBoolPropertyKey.String() {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn errors.New(\"entry not removed\")\n\t\t}).AnyTimes()\n\n\terr := s.client.UpdateValue(dynamicproperties.TestGetBoolPropertyKey, nil)\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_NoRetrySuccess() {\n\tdefaultTestSetup(s)\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), EqSnapshotVersion(2), p.DynamicConfig).\n\t\tReturn(nil).MaxTimes(1)\n\n\tvalues := []*types.DynamicConfigValue{\n\t\t{\n\t\t\tValue: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t},\n\t\t\tFilters: nil,\n\t\t},\n\t}\n\n\terr := s.client.UpdateValue(dynamicproperties.TestGetBoolPropertyKey, values)\n\ts.NoError(err)\n\n\tsnapshot2 := snapshot1\n\tsnapshot2.Values.Entries[0].Values = values\n\ts.mockManager.EXPECT().\n\t\tFetchDynamicConfig(gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.FetchDynamicConfigResponse{\n\t\t\tSnapshot: snapshot2,\n\t\t}, nil).MaxTimes(1)\n\n\terr = s.client.update()\n\ts.NoError(err)\n\n\tv, err := s.client.GetValue(dynamicproperties.TestGetBoolPropertyKey)\n\ts.NoError(err)\n\ts.Equal(true, v)\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_SuccessNewKey() {\n\tvalues := []*types.DynamicConfigValue{\n\t\t{\n\t\t\tValue: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t},\n\t\t\tFilters: nil,\n\t\t},\n\t}\n\n\ts.mockManager.EXPECT().\n\t\tFetchDynamicConfig(gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.FetchDynamicConfigResponse{\n\t\t\tSnapshot: &p.DynamicConfigSnapshot{\n\t\t\t\tVersion: 1,\n\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\tEntries:       nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}, nil).\n\t\tAnyTimes()\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, request *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\ts.Equal(1, len(request.Snapshot.Values.Entries))\n\t\t\ts.Equal(request.Snapshot.Values.Entries[0].Values, values)\n\t\t\treturn nil\n\t\t}).AnyTimes()\n\n\ts.client.update()\n\terr := s.client.UpdateValue(dynamicproperties.TestGetBoolPropertyKey, values)\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_RetrySuccess() {\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), EqSnapshotVersion(2), p.DynamicConfig).\n\t\tReturn(&p.ConditionFailedError{}).AnyTimes()\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), EqSnapshotVersion(3), p.DynamicConfig).\n\t\tReturn(nil).AnyTimes()\n\n\tsnapshot1.Version = 2\n\ts.mockManager.EXPECT().\n\t\tFetchDynamicConfig(gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.FetchDynamicConfigResponse{\n\t\t\tSnapshot: snapshot1,\n\t\t}, nil).AnyTimes()\n\n\ts.client.update()\n\n\terr := s.client.UpdateValue(dynamicproperties.TestGetBoolPropertyKey, []*types.DynamicConfigValue{})\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_RetryFailure() {\n\tdefaultTestSetup(s)\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.ConditionFailedError{}).MaxTimes(retryAttempts + 1)\n\n\terr := s.client.UpdateValue(dynamicproperties.TestGetFloat64PropertyKey, []*types.DynamicConfigValue{})\n\ts.Error(err)\n}\n\nfunc (s *configStoreClientSuite) TestUpdateValue_Timeout() {\n\tdefaultTestSetup(s)\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, _ *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\ttime.Sleep(2 * time.Second)\n\t\t\treturn nil\n\t\t}).AnyTimes()\n\n\terr := s.client.UpdateValue(dynamicproperties.TestGetDurationPropertyKey, []*types.DynamicConfigValue{})\n\ts.Error(err)\n}\n\nfunc (s *configStoreClientSuite) TestRestoreValue_NoFilter() {\n\tdefaultTestSetup(s)\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, request *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\tfor _, entry := range request.Snapshot.Values.Entries {\n\t\t\t\tif entry.Name == dynamicproperties.TestGetBoolPropertyKey.String() {\n\t\t\t\t\tfor _, value := range entry.Values {\n\t\t\t\t\t\ts.Equal(value.Value.Data, jsonMarshalHelper(true))\n\t\t\t\t\t\tif value.Filters == nil {\n\t\t\t\t\t\t\treturn errors.New(\"fallback value not restored\")\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn nil\n\t\t}).AnyTimes()\n\n\terr := s.client.RestoreValue(dynamicproperties.TestGetBoolPropertyKey, nil)\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestRestoreValue_FilterNoMatch() {\n\tdefaultTestSetup(s)\n\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, request *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\tfor _, resEntry := range request.Snapshot.Values.Entries {\n\t\t\t\tfor _, oriEntry := range snapshot1.Values.Entries {\n\t\t\t\t\tif oriEntry.Name == resEntry.Name {\n\t\t\t\t\t\ts.Equal(resEntry.Values, oriEntry.Values)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn nil\n\t\t}).AnyTimes()\n\n\tnoMatchFilter := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"unknown-domain\",\n\t}\n\n\terr := s.client.RestoreValue(dynamicproperties.TestGetBoolPropertyKey, noMatchFilter)\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestRestoreValue_FilterMatch() {\n\tdefaultTestSetup(s)\n\ts.mockManager.EXPECT().\n\t\tUpdateDynamicConfig(gomock.Any(), gomock.Any(), p.DynamicConfig).\n\t\tDoAndReturn(func(_ context.Context, request *p.UpdateDynamicConfigRequest, cfgType p.ConfigType) error {\n\t\t\tfor _, resEntry := range request.Snapshot.Values.Entries {\n\t\t\t\tif resEntry.Name == dynamicproperties.TestGetBoolPropertyKey.String() {\n\t\t\t\t\ts.Equal(2, len(resEntry.Values))\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn nil\n\t\t}).AnyTimes()\n\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\n\terr := s.client.RestoreValue(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n}\n\nfunc (s *configStoreClientSuite) TestListValues() {\n\tdefaultTestSetup(s)\n\tval, err := s.client.ListValue(nil)\n\ts.NoError(err)\n\tfor _, resEntry := range val {\n\t\tfor _, oriEntry := range snapshot1.Values.Entries {\n\t\t\tif oriEntry.Name == resEntry.Name {\n\t\t\t\ts.Equal(resEntry.Values, oriEntry.Values)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *configStoreClientSuite) TestListValues_EmptyCache() {\n\ts.mockManager.EXPECT().\n\t\tFetchDynamicConfig(gomock.Any(), p.DynamicConfig).\n\t\tReturn(&p.FetchDynamicConfigResponse{\n\t\t\tSnapshot: &p.DynamicConfigSnapshot{\n\t\t\t\tVersion: 1,\n\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\tEntries:       nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}, nil).\n\t\tMaxTimes(1)\n\n\ts.client.update()\n\n\tval, err := s.client.ListValue(nil)\n\ts.NoError(err)\n\ts.Nil(val)\n}\n\nfunc (s *configStoreClientSuite) TestValidateKeyDataBlobPair() {\n\ttests := []struct {\n\t\tname    string\n\t\tkey     dynamicproperties.Key\n\t\tblob    *types.DataBlob\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"valid int key\",\n\t\t\tkey:  dynamicproperties.TestGetIntPropertyKey,\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(10),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid int key - wrong type\",\n\t\t\tkey:  dynamicproperties.TestGetIntPropertyKey,\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"valid bool key\",\n\t\t\tkey:  dynamicproperties.TestGetBoolPropertyKey,\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(true),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid bool key - wrong type\",\n\t\t\tkey:  dynamicproperties.TestGetBoolPropertyKey,\n\t\t\tblob: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonMarshalHelper(\"true\"),\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ts.Run(tc.name, func() {\n\t\t\terr := validateKeyDataBlobPair(tc.key, tc.blob)\n\t\t\tif tc.wantErr {\n\t\t\t\ts.Require().Error(err, \"Expected an error for case: %s\", tc.name)\n\t\t\t} else {\n\t\t\t\ts.Require().NoError(err, \"Expected no error for case: %s\", tc.name)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *configStoreClientSuite) TestNewConfigStoreClient_NilPersistenceConfig() {\n\t_, err := NewConfigStoreClient(&c.ClientConfig{}, nil, log.NewNoop(), metrics.NewNoopMetricsClient(), p.DynamicConfig)\n\ts.Require().Error(err, \"should fail when persistence config is nil\")\n\ts.Require().EqualError(err, \"persistence cfg is nil\")\n}\n\nfunc (s *configStoreClientSuite) TestNewConfigStoreClient_MissingDefaultPersistenceConfig() {\n\tpersistenceCfg := &config.Persistence{\n\t\tDataStores: map[string]config.DataStore{},\n\t}\n\t_, err := NewConfigStoreClient(&c.ClientConfig{}, persistenceCfg, log.NewNoop(), metrics.NewNoopMetricsClient(), p.DynamicConfig)\n\ts.Require().Error(err, \"should fail when default persistence config is missing\")\n\ts.Require().EqualError(err, \"default persistence config missing\")\n}\n\nfunc (s *configStoreClientSuite) TestNewConfigStoreClient_InvalidClientConfig() {\n\tpersistenceCfg := &config.Persistence{\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"default\": {},\n\t\t},\n\t\tDefaultStore: \"default\",\n\t}\n\tclientCfg := &c.ClientConfig{\n\t\tPollInterval: time.Millisecond,\n\t}\n\tlogger := log.NewNoop()\n\t_, err := NewConfigStoreClient(clientCfg, persistenceCfg, logger, metrics.NewNoopMetricsClient(), p.DynamicConfig)\n\ts.Require().Error(err, \"should fail when client config is invalid\")\n}\n\nfunc jsonMarshalHelper(v interface{}) []byte {\n\tdata, _ := json.Marshal(v)\n\treturn data\n}\n\ntype eqSnapshotVersionMatcher struct {\n\tversion int64\n}\n\nfunc (e eqSnapshotVersionMatcher) Matches(x interface{}) bool {\n\targ, ok := x.(*p.UpdateDynamicConfigRequest)\n\tif !ok {\n\t\treturn false\n\t}\n\treturn e.version == arg.Snapshot.Version\n}\n\nfunc (e eqSnapshotVersionMatcher) String() string {\n\treturn fmt.Sprintf(\"Version match %d.\\n\", e.version)\n}\n\nfunc EqSnapshotVersion(version int64) gomock.Matcher {\n\treturn eqSnapshotVersionMatcher{version}\n}\n"
  },
  {
    "path": "common/dynamicconfig/configstore/configstore_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: config_store_client.go\n//\n// Generated by this command:\n//\n//\tmockgen -package configstore -source config_store_client.go -destination configstore_mock.go -self_package github.com/uber/cadence/common/dynamicconfig/configstore\n//\n\n// Package configstore is a generated GoMock package.\npackage configstore\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tdynamicproperties \"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// GetBoolValue mocks base method.\nfunc (m *MockClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]any) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBoolValue\", name, filters)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetBoolValue indicates an expected call of GetBoolValue.\nfunc (mr *MockClientMockRecorder) GetBoolValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBoolValue\", reflect.TypeOf((*MockClient)(nil).GetBoolValue), name, filters)\n}\n\n// GetDurationValue mocks base method.\nfunc (m *MockClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]any) (time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDurationValue\", name, filters)\n\tret0, _ := ret[0].(time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDurationValue indicates an expected call of GetDurationValue.\nfunc (mr *MockClientMockRecorder) GetDurationValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDurationValue\", reflect.TypeOf((*MockClient)(nil).GetDurationValue), name, filters)\n}\n\n// GetFloatValue mocks base method.\nfunc (m *MockClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]any) (float64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFloatValue\", name, filters)\n\tret0, _ := ret[0].(float64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetFloatValue indicates an expected call of GetFloatValue.\nfunc (mr *MockClientMockRecorder) GetFloatValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFloatValue\", reflect.TypeOf((*MockClient)(nil).GetFloatValue), name, filters)\n}\n\n// GetIntValue mocks base method.\nfunc (m *MockClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]any) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIntValue\", name, filters)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetIntValue indicates an expected call of GetIntValue.\nfunc (mr *MockClientMockRecorder) GetIntValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIntValue\", reflect.TypeOf((*MockClient)(nil).GetIntValue), name, filters)\n}\n\n// GetListValue mocks base method.\nfunc (m *MockClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]any) ([]any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetListValue\", name, filters)\n\tret0, _ := ret[0].([]any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetListValue indicates an expected call of GetListValue.\nfunc (mr *MockClientMockRecorder) GetListValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetListValue\", reflect.TypeOf((*MockClient)(nil).GetListValue), name, filters)\n}\n\n// GetMapValue mocks base method.\nfunc (m *MockClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]any) (map[string]any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMapValue\", name, filters)\n\tret0, _ := ret[0].(map[string]any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMapValue indicates an expected call of GetMapValue.\nfunc (mr *MockClientMockRecorder) GetMapValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMapValue\", reflect.TypeOf((*MockClient)(nil).GetMapValue), name, filters)\n}\n\n// GetStringValue mocks base method.\nfunc (m *MockClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]any) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStringValue\", name, filters)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStringValue indicates an expected call of GetStringValue.\nfunc (mr *MockClientMockRecorder) GetStringValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStringValue\", reflect.TypeOf((*MockClient)(nil).GetStringValue), name, filters)\n}\n\n// GetValue mocks base method.\nfunc (m *MockClient) GetValue(name dynamicproperties.Key) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetValue\", name)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetValue indicates an expected call of GetValue.\nfunc (mr *MockClientMockRecorder) GetValue(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetValue\", reflect.TypeOf((*MockClient)(nil).GetValue), name)\n}\n\n// GetValueWithFilters mocks base method.\nfunc (m *MockClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]any) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetValueWithFilters\", name, filters)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetValueWithFilters indicates an expected call of GetValueWithFilters.\nfunc (mr *MockClientMockRecorder) GetValueWithFilters(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetValueWithFilters\", reflect.TypeOf((*MockClient)(nil).GetValueWithFilters), name, filters)\n}\n\n// ListValue mocks base method.\nfunc (m *MockClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListValue\", name)\n\tret0, _ := ret[0].([]*types.DynamicConfigEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListValue indicates an expected call of ListValue.\nfunc (mr *MockClientMockRecorder) ListValue(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListValue\", reflect.TypeOf((*MockClient)(nil).ListValue), name)\n}\n\n// RestoreValue mocks base method.\nfunc (m *MockClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RestoreValue\", name, filters)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RestoreValue indicates an expected call of RestoreValue.\nfunc (mr *MockClientMockRecorder) RestoreValue(name, filters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestoreValue\", reflect.TypeOf((*MockClient)(nil).RestoreValue), name, filters)\n}\n\n// Start mocks base method.\nfunc (m *MockClient) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockClientMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockClient)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockClient) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockClientMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockClient)(nil).Stop))\n}\n\n// UpdateValue mocks base method.\nfunc (m *MockClient) UpdateValue(name dynamicproperties.Key, value any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateValue\", name, value)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateValue indicates an expected call of UpdateValue.\nfunc (mr *MockClientMockRecorder) UpdateValue(name, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateValue\", reflect.TypeOf((*MockClient)(nil).UpdateValue), name, value)\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicconfigfx/fx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicconfigfx\n\nimport (\n\t\"path/filepath\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// Module provides fx options for dynamic config initialization\nvar Module = fx.Options(fx.Provide(New))\n\n// Params required to build a new dynamic config.\ntype Params struct {\n\tfx.In\n\n\tCfg           config.Config\n\tLogger        log.Logger\n\tMetricsClient metrics.Client\n\tRootDir       string `name:\"root-dir\"`\n\n\tLifecycle fx.Lifecycle\n}\n\ntype Result struct {\n\tfx.Out\n\n\tClient     dynamicconfig.Client\n\tCollection *dynamicconfig.Collection\n}\n\n// New creates dynamicconfig.Client from the configuration\nfunc New(p Params) Result {\n\tstopped := make(chan struct{})\n\n\tif p.Cfg.DynamicConfig.Client == \"\" {\n\t\tp.Cfg.DynamicConfigClient.Filepath = constructPathIfNeed(p.RootDir, p.Cfg.DynamicConfigClient.Filepath)\n\t} else {\n\t\tp.Cfg.DynamicConfig.FileBased.Filepath = constructPathIfNeed(p.RootDir, p.Cfg.DynamicConfig.FileBased.Filepath)\n\t}\n\n\tp.Lifecycle.Append(fx.StopHook(func() {\n\t\tclose(stopped)\n\t}))\n\n\tvar res dynamicconfig.Client\n\n\tvar err error\n\tif p.Cfg.DynamicConfig.Client == \"\" {\n\t\tp.Logger.Warn(\"falling back to legacy file based dynamicClientConfig\")\n\t\tres, err = dynamicconfig.NewFileBasedClient(&p.Cfg.DynamicConfigClient, p.Logger, stopped)\n\t} else {\n\t\tswitch p.Cfg.DynamicConfig.Client {\n\t\tcase dynamicconfig.ConfigStoreClient:\n\t\t\tp.Logger.Info(\"initialising ConfigStore dynamic config client\")\n\t\t\tres, err = configstore.NewConfigStoreClient(\n\t\t\t\t&p.Cfg.DynamicConfig.ConfigStore,\n\t\t\t\t&p.Cfg.Persistence,\n\t\t\t\tp.Logger,\n\t\t\t\tp.MetricsClient,\n\t\t\t\tpersistence.DynamicConfig,\n\t\t\t)\n\t\tcase dynamicconfig.FileBasedClient:\n\t\t\tp.Logger.Info(\"initialising File Based dynamic config client\")\n\t\t\tres, err = dynamicconfig.NewFileBasedClient(&p.Cfg.DynamicConfig.FileBased, p.Logger, stopped)\n\t\t}\n\t}\n\n\tif res == nil {\n\t\tp.Logger.Info(\"initialising NOP dynamic config client\")\n\t\tres = dynamicconfig.NewNopClient()\n\t} else if err != nil {\n\t\tp.Logger.Error(\"creating dynamic config client failed, using no-op config client instead\", tag.Error(err))\n\t\tres = dynamicconfig.NewNopClient()\n\t}\n\n\tclusterGroupMetadata := p.Cfg.ClusterGroupMetadata\n\tdc := dynamicconfig.NewCollection(\n\t\tres,\n\t\tp.Logger,\n\t\tdynamicproperties.ClusterNameFilter(clusterGroupMetadata.CurrentClusterName),\n\t)\n\n\treturn Result{\n\t\tClient:     res,\n\t\tCollection: dc,\n\t}\n}\n\n// constructPathIfNeed would append the dir as the root dir\n// when the file wasn't absolute path.\nfunc constructPathIfNeed(dir string, file string) string {\n\tif !filepath.IsAbs(file) {\n\t\treturn dir + \"/\" + file\n\t}\n\treturn file\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicconfigfx/fx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicconfigfx\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestModule(t *testing.T) {\n\tapp := fxtest.New(t,\n\t\ttestlogger.Module(t),\n\t\tfx.Provide(\n\t\t\tfunc() config.Config {\n\t\t\t\treturn config.Config{\n\t\t\t\t\tClusterGroupMetadata: &config.ClusterGroupMetadata{},\n\t\t\t\t}\n\t\t\t},\n\t\t\tfunc() fxRoot {\n\t\t\t\treturn fxRoot{\n\t\t\t\t\tRootDir: \"../../../\",\n\t\t\t\t}\n\t\t\t},\n\t\t\tmetrics.NewNoopMetricsClient,\n\t\t),\n\t\tModule,\n\t\tfx.Invoke(func(c dynamicconfig.Client) {}),\n\t)\n\tapp.RequireStart().RequireStop()\n}\n\ntype fxRoot struct {\n\tfx.Out\n\n\tRootDir string `name:\"root-dir\"`\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/config_mock.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"time\"\n)\n\n// These mock functions are for tests to use config properties that are dynamic\n\n// GetIntPropertyFn returns value as IntPropertyFn\nfunc GetIntPropertyFn(value int) func(opts ...FilterOption) int {\n\treturn func(...FilterOption) int { return value }\n}\n\n// GetIntPropertyFilteredByDomain returns values as IntPropertyFnWithDomainFilters\nfunc GetIntPropertyFilteredByDomain(value int) func(domain string) int {\n\treturn func(domain string) int { return value }\n}\n\n// GetIntPropertyFilteredByTaskListInfo returns value as IntPropertyFnWithTaskListInfoFilters\nfunc GetIntPropertyFilteredByTaskListInfo(value int) func(domain string, taskList string, taskType int) int {\n\treturn func(domain string, taskList string, taskType int) int { return value }\n}\n\n// GetIntPropertyFilteredByShardID returns values as IntPropertyFnWithShardIDFilter\nfunc GetIntPropertyFilteredByShardID(value int) func(shardID int) int {\n\treturn func(shardID int) int { return value }\n}\n\n// GetIntPropertyFilteredByWorkflowType returns values as IntPropertyFnWithWorkflowTypeFilters\nfunc GetIntPropertyFilteredByWorkflowType(value int) func(domainName string, workflowType string) int {\n\treturn func(domainName string, workflowType string) int { return value }\n}\n\n// GetDurationPropertyFilteredByWorkflowType returns values as IntPropertyFnWithWorkflowTypeFilters\nfunc GetDurationPropertyFilteredByWorkflowType(value time.Duration) func(domainName string, workflowType string) time.Duration {\n\treturn func(domainName string, workflowType string) time.Duration { return value }\n}\n\n// GetFloatPropertyFn returns value as FloatPropertyFn\nfunc GetFloatPropertyFn(value float64) func(opts ...FilterOption) float64 {\n\treturn func(...FilterOption) float64 { return value }\n}\n\n// GetBoolPropertyFn returns value as BoolPropertyFn\nfunc GetBoolPropertyFn(value bool) func(opts ...FilterOption) bool {\n\treturn func(...FilterOption) bool { return value }\n}\n\n// GetBoolPropertyFnFilteredByDomain returns value as BoolPropertyFnWithDomainFilters\nfunc GetBoolPropertyFnFilteredByDomain(value bool) func(domain string) bool {\n\treturn func(domain string) bool { return value }\n}\n\n// GetBoolPropertyFnFilteredByDomainID returns value as BoolPropertyFnWithDomainIDFilters\nfunc GetBoolPropertyFnFilteredByDomainID(value bool) func(domainID string) bool {\n\treturn func(domainID string) bool { return value }\n}\n\n// GetBoolPropertyFilteredByTaskListInfo returns value as BoolPropertyFnWithTaskListInfoFilters\nfunc GetBoolPropertyFilteredByTaskListInfo(value bool) func(domain string, taskList string, taskType int) bool {\n\treturn func(domain string, taskList string, taskType int) bool { return value }\n}\n\n// GetDurationPropertyFnFilteredByDomain returns value as DurationPropertyFnFilteredByDomain\nfunc GetDurationPropertyFnFilteredByDomain(value time.Duration) func(domain string) time.Duration {\n\treturn func(domain string) time.Duration { return value }\n}\n\n// GetDurationPropertyFn returns value as DurationPropertyFn\nfunc GetDurationPropertyFn(value time.Duration) func(opts ...FilterOption) time.Duration {\n\treturn func(...FilterOption) time.Duration { return value }\n}\n\n// GetDurationPropertyFnFilteredByTaskListInfo returns value as DurationPropertyFnWithTaskListInfoFilters\nfunc GetDurationPropertyFnFilteredByTaskListInfo(value time.Duration) func(domain string, taskList string, taskType int) time.Duration {\n\treturn func(domain string, taskList string, taskType int) time.Duration { return value }\n}\n\n// GetDurationPropertyFnFilteredByShardID returns value as DurationPropertyFnWithShardIDFilter\nfunc GetDurationPropertyFnFilteredByShardID(value time.Duration) func(shardID int) time.Duration {\n\treturn func(shardID int) time.Duration { return value }\n}\n\n// GetStringPropertyFn returns value as StringPropertyFn\nfunc GetStringPropertyFn(value string) func(opts ...FilterOption) string {\n\treturn func(...FilterOption) string { return value }\n}\n\n// GetStringPropertyFnFilteredByDomain returns value as StringPropertyFnWithDomainFilters\nfunc GetStringPropertyFnFilteredByDomain(value string) func(domain string) string {\n\treturn func(domain string) string { return value }\n}\n\n// GetMapPropertyFn returns value as MapPropertyFn\nfunc GetMapPropertyFn(value map[string]interface{}) func(opts ...FilterOption) map[string]interface{} {\n\treturn func(...FilterOption) map[string]interface{} { return value }\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/constants.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n)\n\ntype (\n\t// DynamicInt defines the properties for a dynamic config with int value type\n\tDynamicInt struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue int\n\t\tFilters      []Filter\n\t}\n\n\tDynamicBool struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue bool\n\t\tFilters      []Filter\n\t}\n\n\tDynamicFloat struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue float64\n\t\tFilters      []Filter\n\t}\n\n\tDynamicString struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue string\n\t\tFilters      []Filter\n\t}\n\n\tDynamicDuration struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue time.Duration\n\t\tFilters      []Filter\n\t}\n\n\tDynamicMap struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue map[string]interface{}\n\t\tFilters      []Filter\n\t}\n\n\tDynamicList struct {\n\t\tKeyName      string\n\t\tDescription  string\n\t\tDefaultValue []interface{}\n\t\tFilters      []Filter\n\t}\n\n\tIntKey      int\n\tBoolKey     int\n\tFloatKey    int\n\tStringKey   int\n\tDurationKey int\n\tMapKey      int\n\tListKey     int\n\n\tKey interface {\n\t\tString() string\n\t\tDescription() string\n\t\tDefaultValue() interface{}\n\t\t// Filters is used to identify what filters a DynamicConfig key may have.\n\t\t// For example, CLI tool uses this to figure out all domain specific configurations for migration validation.\n\t\tFilters() []Filter\n\t}\n)\n\n// ListAllProductionKeys returns all key used in production\nfunc ListAllProductionKeys() []Key {\n\tresult := make([]Key, 0, len(IntKeys)+len(BoolKeys)+len(FloatKeys)+len(StringKeys)+len(DurationKeys)+len(MapKeys))\n\tfor i := TestGetIntPropertyFilteredByTaskListInfoKey + 1; i < LastIntKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetBoolPropertyFilteredByTaskListInfoKey + 1; i < LastBoolKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetFloat64PropertyKey + 1; i < LastFloatKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetStringPropertyKey + 1; i < LastStringKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetDurationPropertyFilteredByTaskListInfoKey + 1; i < LastDurationKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetMapPropertyKey + 1; i < LastMapKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\tfor i := TestGetListPropertyKey + 1; i < LastListKey; i++ {\n\t\tresult = append(result, i)\n\t}\n\treturn result\n}\n\nfunc GetKeyFromKeyName(keyName string) (Key, error) {\n\tkeyVal, ok := _keyNames[keyName]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"invalid dynamic config key name: %s\", keyName)\n\t}\n\treturn keyVal, nil\n}\n\n// GetAllKeys returns a copy of all configuration keys with all details\nfunc GetAllKeys() map[string]Key {\n\tresult := make(map[string]Key, len(_keyNames))\n\tfor k, v := range _keyNames {\n\t\tresult[k] = v\n\t}\n\treturn result\n}\n\nfunc ValidateKeyValuePair(key Key, value interface{}) error {\n\terr := fmt.Errorf(\"key value pair mismatch, key type: %T, value type: %T\", key, value)\n\tswitch key.(type) {\n\tcase IntKey:\n\t\tif _, ok := value.(int); !ok {\n\t\t\treturn err\n\t\t}\n\tcase BoolKey:\n\t\tif _, ok := value.(bool); !ok {\n\t\t\treturn err\n\t\t}\n\tcase FloatKey:\n\t\tif _, ok := value.(float64); !ok {\n\t\t\treturn err\n\t\t}\n\tcase StringKey:\n\t\tif _, ok := value.(string); !ok {\n\t\t\treturn err\n\t\t}\n\tcase DurationKey:\n\t\tif _, ok := value.(time.Duration); !ok {\n\t\t\treturn err\n\t\t}\n\tcase MapKey:\n\t\tif _, ok := value.(map[string]interface{}); !ok {\n\t\t\treturn err\n\t\t}\n\tcase ListKey:\n\t\tif _, ok := value.([]interface{}); !ok {\n\t\t\treturn err\n\t\t}\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown key type: %T\", key)\n\t}\n\treturn nil\n}\n\nfunc (k IntKey) String() string {\n\treturn IntKeys[k].KeyName\n}\n\nfunc (k IntKey) Description() string {\n\treturn IntKeys[k].Description\n}\n\nfunc (k IntKey) DefaultValue() interface{} {\n\treturn IntKeys[k].DefaultValue\n}\n\nfunc (k IntKey) DefaultInt() int {\n\treturn IntKeys[k].DefaultValue\n}\n\nfunc (k IntKey) Filters() []Filter {\n\treturn IntKeys[k].Filters\n}\n\nfunc (k BoolKey) String() string {\n\treturn BoolKeys[k].KeyName\n}\n\nfunc (k BoolKey) Description() string {\n\treturn BoolKeys[k].Description\n}\n\nfunc (k BoolKey) DefaultValue() interface{} {\n\treturn BoolKeys[k].DefaultValue\n}\n\nfunc (k BoolKey) DefaultBool() bool {\n\treturn BoolKeys[k].DefaultValue\n}\n\nfunc (k BoolKey) Filters() []Filter {\n\treturn BoolKeys[k].Filters\n}\n\nfunc (k FloatKey) String() string {\n\treturn FloatKeys[k].KeyName\n}\n\nfunc (k FloatKey) Description() string {\n\treturn FloatKeys[k].Description\n}\n\nfunc (k FloatKey) DefaultValue() interface{} {\n\treturn FloatKeys[k].DefaultValue\n}\n\nfunc (k FloatKey) DefaultFloat() float64 {\n\treturn FloatKeys[k].DefaultValue\n}\n\nfunc (k FloatKey) Filters() []Filter {\n\treturn FloatKeys[k].Filters\n}\n\nfunc (k StringKey) String() string {\n\treturn StringKeys[k].KeyName\n}\n\nfunc (k StringKey) Description() string {\n\treturn StringKeys[k].Description\n}\n\nfunc (k StringKey) DefaultValue() interface{} {\n\treturn StringKeys[k].DefaultValue\n}\n\nfunc (k StringKey) DefaultString() string {\n\treturn StringKeys[k].DefaultValue\n}\n\nfunc (k StringKey) Filters() []Filter {\n\treturn StringKeys[k].Filters\n}\n\nfunc (k DurationKey) String() string {\n\treturn DurationKeys[k].KeyName\n}\n\nfunc (k DurationKey) Description() string {\n\treturn DurationKeys[k].Description\n}\n\nfunc (k DurationKey) DefaultValue() interface{} {\n\treturn DurationKeys[k].DefaultValue\n}\n\nfunc (k DurationKey) DefaultDuration() time.Duration {\n\treturn DurationKeys[k].DefaultValue\n}\n\nfunc (k DurationKey) Filters() []Filter {\n\treturn DurationKeys[k].Filters\n}\n\nfunc (k MapKey) String() string {\n\treturn MapKeys[k].KeyName\n}\n\nfunc (k MapKey) Description() string {\n\treturn MapKeys[k].Description\n}\n\nfunc (k MapKey) DefaultValue() interface{} {\n\treturn MapKeys[k].DefaultValue\n}\n\nfunc (k MapKey) DefaultMap() map[string]interface{} {\n\treturn MapKeys[k].DefaultValue\n}\n\nfunc (k MapKey) Filters() []Filter {\n\treturn MapKeys[k].Filters\n}\n\nfunc (k ListKey) String() string {\n\treturn ListKeys[k].KeyName\n}\n\nfunc (k ListKey) Description() string {\n\treturn ListKeys[k].Description\n}\n\nfunc (k ListKey) DefaultValue() interface{} {\n\treturn ListKeys[k].DefaultValue\n}\n\nfunc (k ListKey) DefaultList() []interface{} {\n\treturn ListKeys[k].DefaultValue\n}\n\nfunc (k ListKey) Filters() []Filter {\n\treturn ListKeys[k].Filters\n}\n\n// UnlimitedRPS represents an integer to use for \"unlimited\" RPS values.\n//\n// Since our ratelimiters do int/float conversions, and zero or negative values\n// result in not allowing any requests, math.MaxInt is unsafe:\n//\n//\tint(float64(math.MaxInt)) // -9223372036854775808\n//\n// Much higher values are possible, but we can't handle 2 billion RPS, this is good enough.\nconst UnlimitedRPS = math.MaxInt32\n\n/***\n* !!!Important!!!\n* For developer: Make sure to add/maintain the comment in the right format: usage, keyName, and default value\n* So that our go-docs can have the full [documentation](https://pkg.go.dev/github.com/uber/cadence@v0.19.1/common/service/dynamicconfig#Key).\n***/\nconst (\n\tUnknownIntKey IntKey = iota\n\n\t// key for tests\n\tTestGetIntPropertyKey\n\tTestGetIntPropertyFilteredByDomainKey\n\tTestGetIntPropertyFilteredByWorkflowTypeKey\n\tTestGetIntPropertyFilteredByTaskListInfoKey\n\tTestGetIntPropertyFilteredByShardIDKey\n\n\t// key for common & admin\n\n\t// TransactionSizeLimit is the maximum allowed size in bytes for a single persistence transaction when appending history events\n\t// KeyName: system.transactionSizeLimit\n\t// Value type: Int\n\t// Default value: 14680064 (14*1024*1024, ~14MB)\n\t// Allowed filters: N/A\n\tTransactionSizeLimit\n\t// MaxRetentionDays is the maximum allowed retention period in days for workflow history after workflow close for all domains\n\t// KeyName: system.maxRetentionDays\n\t// Value type: Int\n\t// Default value: 30\n\t// Allowed filters: N/A\n\tMaxRetentionDays\n\t// MinRetentionDays is the minimum allowed retention period in days for workflow history after workflow close for all domains\n\t// KeyName: system.minRetentionDays\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tMinRetentionDays\n\t// MaxDecisionStartToCloseSeconds is the maximum allowed value for decision start to close timeout in seconds\n\t// KeyName: system.maxDecisionStartToCloseSeconds\n\t// Value type: Int\n\t// Default value: 240\n\t// Allowed filters: N/A\n\tMaxDecisionStartToCloseSeconds\n\t// BlobSizeLimitError is the per event blob size limit\n\t// KeyName: limit.blobSize.error\n\t// Value type: Int\n\t// Default value: 2 * 1024 * 1024\n\t// Allowed filters: N/A\n\tBlobSizeLimitError\n\t// BlobSizeLimitWarn is the per event blob size limit for warning\n\t// KeyName: limit.blobSize.warn\n\t// Value type: Int\n\t// Default value: 262144 (256*1024)\n\t// Allowed filters: DomainName\n\tBlobSizeLimitWarn\n\t// HistorySizeLimitError is the per workflow execution history size limit\n\t// KeyName: limit.historySize.error\n\t// Value type: Int\n\t// Default value: 209715200 (200*1024*1024)\n\t// Allowed filters: DomainName\n\tHistorySizeLimitError\n\t// HistorySizeLimitWarn is the per workflow execution history size limit for warning\n\t// KeyName: limit.historySize.warn\n\t// Value type: Int\n\t// Default value: 52428800 (50*1024*1024)\n\t// Allowed filters: DomainName\n\tHistorySizeLimitWarn\n\t// HistoryCountLimitError is the per workflow execution history event count limit\n\t// KeyName: limit.historyCount.error\n\t// Value type: Int\n\t// Default value: 204800 (200*1024)\n\t// Allowed filters: DomainName\n\tHistoryCountLimitError\n\t// HistoryCountLimitWarn is the per workflow execution history event count limit for warning\n\t// KeyName: limit.historyCount.warn\n\t// Value type: Int\n\t// Default value: 51200 (50*1024)\n\t// Allowed filters: DomainName\n\tHistoryCountLimitWarn\n\t// PendingActivitiesCountLimitError is the limit of how many pending activities a workflow can have at a point in time\n\t// KeyName: limit.pendingActivityCount.error\n\t// Value type: Int\n\t// Default value: 1024\n\t// Allowed filters: N/A\n\tPendingActivitiesCountLimitError\n\t// PendingActivitiesCountLimitWarn is the limit of how many activities a workflow can have before a warning is logged\n\t// KeyName: limit.pendingActivityCount.warn\n\t// Value type: Int\n\t// Default value: 512\n\t// Allowed filters: N/A\n\tPendingActivitiesCountLimitWarn\n\t// DomainNameMaxLength is the length limit for domain name\n\t// KeyName: limit.domainNameLength\n\t// Value type: Int\n\t// Default value: common.DefaultIDLengthErrorLimit (1000)\n\t// Allowed filters: DomainName\n\tDomainNameMaxLength\n\t// IdentityMaxLength is the length limit for identity\n\t// KeyName: limit.identityLength\n\t// Value type: Int\n\t// Default value: 1000 ( see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tIdentityMaxLength\n\t// WorkflowIDMaxLength is the length limit for workflowID\n\t// KeyName: limit.workflowIDLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tWorkflowIDMaxLength\n\t// SignalNameMaxLength is the length limit for signal name\n\t// KeyName: limit.signalNameLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tSignalNameMaxLength\n\t// WorkflowTypeMaxLength is the length limit for workflow type\n\t// KeyName: limit.workflowTypeLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tWorkflowTypeMaxLength\n\t// RequestIDMaxLength is the length limit for requestID\n\t// KeyName: limit.requestIDLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tRequestIDMaxLength\n\t// TaskListNameMaxLength is the length limit for task list name\n\t// KeyName: limit.taskListNameLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tTaskListNameMaxLength\n\t// ActivityIDMaxLength is the length limit for activityID\n\t// KeyName: limit.activityIDLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tActivityIDMaxLength\n\t// ActivityTypeMaxLength is the length limit for activity type\n\t// KeyName: limit.activityTypeLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tActivityTypeMaxLength\n\t// MarkerNameMaxLength is the length limit for marker name\n\t// KeyName: limit.markerNameLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tMarkerNameMaxLength\n\t// TimerIDMaxLength is the length limit for timerID\n\t// KeyName: limit.timerIDLength\n\t// Value type: Int\n\t// Default value: 1000 (see common.DefaultIDLengthErrorLimit)\n\t// Allowed filters: DomainName\n\tTimerIDMaxLength\n\t// MaxIDLengthWarnLimit is the warn length limit for various IDs, including: Domain, TaskList, WorkflowID, ActivityID, TimerID, WorkflowType, ActivityType, SignalName, MarkerName, ErrorReason/FailureReason/CancelCause, Identity, RequestID\n\t// KeyName: limit.maxIDWarnLength\n\t// Value type: Int\n\t// Default value: 128 (see common.DefaultIDLengthWarnLimit)\n\t// Allowed filters: N/A\n\tMaxIDLengthWarnLimit\n\n\t// key for frontend\n\n\t// FrontendPersistenceMaxQPS is the max qps frontend host can query DB\n\t// KeyName: frontend.persistenceMaxQPS\n\t// Value type: Int\n\t// Default value: 2000\n\t// Allowed filters: N/A\n\tFrontendPersistenceMaxQPS\n\t// FrontendPersistenceGlobalMaxQPS is the max qps frontend cluster can query DB\n\t// KeyName: frontend.persistenceGlobalMaxQPS\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tFrontendPersistenceGlobalMaxQPS\n\t// FrontendVisibilityMaxPageSize is default max size for ListWorkflowExecutions in one page\n\t// KeyName: frontend.visibilityMaxPageSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: DomainName\n\tFrontendVisibilityMaxPageSize\n\t// FrontendVisibilityListMaxQPS is max qps frontend can list open/close workflows\n\t// KeyName: frontend.visibilityListMaxQPS\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: DomainName\n\t// deprecated: never used for ratelimiting, only sampling-based failure injection, and only on database-based visibility\n\tFrontendVisibilityListMaxQPS\n\t// FrontendESVisibilityListMaxQPS is max qps frontend can list open/close workflows from ElasticSearch\n\t// KeyName: frontend.esVisibilityListMaxQPS\n\t// Value type: Int\n\t// Default value: 30\n\t// Allowed filters: DomainName\n\t// deprecated: never read from, all ES reads and writes erroneously use PersistenceMaxQPS\n\tFrontendESVisibilityListMaxQPS\n\t// FrontendESIndexMaxResultWindow is ElasticSearch index setting max_result_window\n\t// KeyName: frontend.esIndexMaxResultWindow\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tFrontendESIndexMaxResultWindow\n\t// FrontendHistoryMaxPageSize is default max size for GetWorkflowExecutionHistory in one page\n\t// KeyName: frontend.historyMaxPageSize\n\t// Value type: Int\n\t// Default value: 1000 (see common.GetHistoryMaxPageSize)\n\t// Allowed filters: DomainName\n\tFrontendHistoryMaxPageSize\n\t// FrontendUserRPS is used to limit \"user\" requests (StartWorkflow, Signal, etc)\n\t// per frontend instance (across all domains, or for non-domain-related requests),\n\t// and is mostly intended to protect against excessive single-host load.\n\t//\n\t// KeyName: frontend.rps\n\t// Value type: Int\n\t// Default value: 1200\n\t// Allowed filters: N/A\n\tFrontendUserRPS\n\t// FrontendWorkerRPS is used to limit \"worker\" requests (PollFor...Task, RespondTask..., etc)\n\t// per frontend instance (across all domains, or for non-domain-related requests),\n\t// and is mostly intended to protect against excessive single-host load.\n\t//\n\t// KeyName: frontend.workerrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: N/A\n\tFrontendWorkerRPS\n\t// FrontendVisibilityRPS is used to limit \"visibility\" requests (ListWorkflow* and similar)\n\t// per frontend instance (across all domains, or for non-domain-related requests),\n\t// and is mostly intended to protect against excessive single-host load.\n\t//\n\t// KeyName: frontend.visibilityrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: N/A\n\tFrontendVisibilityRPS\n\t// FrontendAsyncRPS is used to limit \"async\" requests (StartWorkflowAsync, etc for many \"user\" APIs)\n\t// per frontend instance (across all domains, or for non-domain-related requests),\n\t// and is mostly intended to protect against excessive single-host load.\n\t//\n\t// KeyName: frontend.asyncrps\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tFrontendAsyncRPS\n\t// FrontendMaxDomainUserRPSPerInstance is used to limit \"user\" requests (StartWorkflow, Signal, etc)\n\t// per domain per frontend instance, and is mostly intended to protect against excessive single-host load.\n\t//\n\t// This limit applies along-side FrontendGlobalDomainUserRPS: both must be allowed to allow a request.\n\t//\n\t// KeyName: frontend.domainrps\n\t// Value type: Int\n\t// Default value: 1200\n\t// Allowed filters: DomainName\n\tFrontendMaxDomainUserRPSPerInstance\n\t// FrontendMaxDomainWorkerRPSPerInstance is used to limit \"worker\" requests (PollFor...Task, RespondTask..., etc)\n\t// per domain per frontend instance, and is mostly intended to protect against excessive single-host load.\n\t//\n\t// This limit applies along-side FrontendGlobalDomainWorkerRPS: both must be allowed to allow a request.\n\t//\n\t// KeyName: frontend.domainworkerrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: DomainName\n\tFrontendMaxDomainWorkerRPSPerInstance\n\t// FrontendMaxDomainVisibilityRPSPerInstance is used to limit \"visibility\" requests (ListWorkflow* and similar)\n\t// per domain per frontend instance, and is mostly intended to protect against excessive single-host load.\n\t//\n\t// This limit applies along-side FrontendGlobalDomainVisibilityRPS: both must be allowed to allow a request.\n\t//\n\t// KeyName: frontend.domainvisibilityrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: DomainName\n\tFrontendMaxDomainVisibilityRPSPerInstance\n\t// FrontendMaxDomainAsyncRPSPerInstance is used to limit \"async\" requests (StartWorkflowAsync, etc for many \"user\" APIs)\n\t// per frontend instance, and is mostly intended to protect against excessive single-host load.\n\t//\n\t// This limit applies along-side FrontendGlobalDomainAsyncRPS: both must be allowed to allow a request.\n\t//\n\t// KeyName: frontend.domainasyncrps\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: DomainName\n\tFrontendMaxDomainAsyncRPSPerInstance\n\t// FrontendGlobalDomainUserRPS is used to limit \"user\" requests (StartWorkflow, Signal, etc)\n\t// per domain to a target RPS that is shared across the entire cluster.\n\t//\n\t// Currently, there are two ways this is achieved, which can be selected by FrontendGlobalRatelimiterMode\n\t// with a \"user:\" key prefix:\n\t//   1. \"local\", where the configured RPS is split evenly across all frontend hosts in the cluster.\n\t//      This works well if your load is roughly evenly distributed.\n\t//   2. \"global\", where frontend hosts share load information with each other, to adjust to imbalanced load.\n\t//      This works well if your load is very imbalanced, e.g. one domain tends to contact a subset of frontend hosts much more than others.\n\t//\n\t// KeyName: frontend.globalDomainrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS (0 triggers a fallback to per-instance-RPS, generally avoid)\n\t// Allowed filters: DomainName\n\tFrontendGlobalDomainUserRPS\n\t// FrontendGlobalDomainWorkerRPS is used to limit \"worker\" requests (PollFor...Task, RespondTask..., etc)\n\t// per domain to a target RPS that is shared across the entire cluster.\n\t//\n\t// Currently, there are two ways this is achieved, which can be selected by FrontendGlobalRatelimiterMode\n\t// with a \"worker:\" key prefix:\n\t//   1. \"local\", where the configured RPS is split evenly across all frontend hosts in the cluster.\n\t//      This works well if your load is roughly evenly distributed.\n\t//   2. \"global\", where frontend hosts share load information with each other, to adjust to imbalanced load.\n\t//      This works well if your load is very imbalanced, e.g. one domain tends to contact a subset of frontend hosts much more than others.\n\t//\n\t// KeyName: frontend.globalDomainWorkerrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS (0 triggers a fallback to per-instance-RPS, generally avoid)\n\t// Allowed filters: DomainName\n\tFrontendGlobalDomainWorkerRPS\n\t// FrontendGlobalDomainVisibilityRPS is used to limit \"visibility\" requests (ListWorkflow* and similar)\n\t// per domain to a target RPS that is shared across the entire cluster.\n\t//\n\t// Currently, there are two ways this is achieved, which can be selected by FrontendGlobalRatelimiterMode\n\t// with a \"visibility:\" key prefix:\n\t//   1. \"local\", where the configured RPS is split evenly across all frontend hosts in the cluster.\n\t//      This works well if your load is roughly evenly distributed.\n\t//   2. \"global\", where frontend hosts share load information with each other, to adjust to imbalanced load.\n\t//      This works well if your load is very imbalanced, e.g. one domain tends to contact a subset of frontend hosts much more than others.\n\t//\n\t// KeyName: frontend.globalDomainVisibilityrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS (0 triggers a fallback to per-instance-RPS, generally avoid)\n\t// Allowed filters: DomainName\n\tFrontendGlobalDomainVisibilityRPS\n\t// FrontendGlobalDomainAsyncRPS is used to limit \"async\" requests (StartWorkflowAsync, etc for many \"user\" APIs)\n\t// per domain to a target RPS that is shared across the entire cluster.\n\t//\n\t// Currently, there are two ways this is achieved, which can be selected by FrontendGlobalRatelimiterMode\n\t// with a \"async:\" key prefix:\n\t//   1. \"local\", where the configured RPS is split evenly across all frontend hosts in the cluster.\n\t//      This works well if your load is roughly evenly distributed.\n\t//   2. \"global\", where frontend hosts share load information with each other, to adjust to imbalanced load.\n\t//      This works well if your load is very imbalanced, e.g. one domain tends to contact a subset of frontend hosts much more than others.\n\t//\n\t// KeyName: frontend.globalDomainAsyncrps\n\t// Value type: Int\n\t// Default value: 100000\n\t// Allowed filters: DomainName\n\tFrontendGlobalDomainAsyncRPS\n\t// FrontendDecisionResultCountLimit is max number of decisions per RespondDecisionTaskCompleted request\n\t// KeyName: frontend.decisionResultCountLimit\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tFrontendDecisionResultCountLimit\n\t// FrontendThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\n\t// KeyName: frontend.throttledLogRPS\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: N/A\n\tFrontendThrottledLogRPS\n\t// FrontendMaxBadBinaries is the max number of bad binaries in domain config\n\t// KeyName: frontend.maxBadBinaries\n\t// Value type: Int\n\t// Default value: 10 (see domain.MaxBadBinaries)\n\t// Allowed filters: DomainName\n\tFrontendMaxBadBinaries\n\t// SearchAttributesNumberOfKeysLimit is the limit of number of keys\n\t// KeyName: frontend.searchAttributesNumberOfKeysLimit\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: DomainName\n\tSearchAttributesNumberOfKeysLimit\n\t// SearchAttributesSizeOfValueLimit is the size limit of each value\n\t// KeyName: frontend.searchAttributesSizeOfValueLimit\n\t// Value type: Int\n\t// Default value: 2048 (2*1024)\n\t// Allowed filters: DomainName\n\tSearchAttributesSizeOfValueLimit\n\t// SearchAttributesTotalSizeLimit is the size limit of the whole map\n\t// KeyName: frontend.searchAttributesTotalSizeLimit\n\t// Value type: Int\n\t// Default value: 40960 (40*1024)\n\t// Allowed filters: DomainName\n\tSearchAttributesTotalSizeLimit\n\t// VisibilityArchivalQueryMaxPageSize is the maximum page size for a visibility archival query\n\t// KeyName: frontend.visibilityArchivalQueryMaxPageSize\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tVisibilityArchivalQueryMaxPageSize\n\t// FrontendFailoverHistoryMaxSize is the maximum size for the number of failover event records in a domain failover history\n\t// KeyName: frontend.failoverHistoryMaxSize\n\t// Value type: Int\n\t// Default value: 5\n\t// Allowed filters: DomainName\n\tFrontendFailoverHistoryMaxSize\n\n\t// key for matching\n\n\t// MatchingUserRPS is request rate per second for each matching host\n\t// KeyName: matching.rps\n\t// Value type: Int\n\t// Default value: 1200\n\t// Allowed filters: N/A\n\tMatchingUserRPS\n\t// MatchingWorkerRPS is background-processing request rate per second for each matching host\n\t// KeyName: matching.workerrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: N/A\n\tMatchingWorkerRPS\n\t// MatchingDomainUserRPS is request rate per domain per second for each matching host\n\t// KeyName: matching.domainrps\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMatchingDomainUserRPS\n\t// MatchingDomainWorkerRPS is background-processing request rate per domain per second for each matching host\n\t// KeyName: matching.domainworkerrps\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: N/A\n\tMatchingDomainWorkerRPS\n\t// MatchingPersistenceMaxQPS is the max qps matching host can query DB\n\t// KeyName: matching.persistenceMaxQPS\n\t// Value type: Int\n\t// Default value: 3000\n\t// Allowed filters: N/A\n\tMatchingPersistenceMaxQPS\n\t// MatchingPersistenceGlobalMaxQPS is the max qps matching cluster can query DB\n\t// KeyName: matching.persistenceGlobalMaxQPS\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMatchingPersistenceGlobalMaxQPS\n\t// MatchingMinTaskThrottlingBurstSize is the minimum burst size for task list throttling\n\t// KeyName: matching.minTaskThrottlingBurstSize\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingMinTaskThrottlingBurstSize\n\t// MatchingGetTasksBatchSize is the maximum batch size to fetch from the task buffer\n\t// KeyName: matching.getTasksBatchSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingGetTasksBatchSize\n\t// MatchingOutstandingTaskAppendsThreshold is the threshold for outstanding task appends\n\t// KeyName: matching.outstandingTaskAppendsThreshold\n\t// Value type: Int\n\t// Default value: 250\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingOutstandingTaskAppendsThreshold\n\t// MatchingMaxTaskBatchSize is max batch size for task writer\n\t// KeyName: matching.maxTaskBatchSize\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingMaxTaskBatchSize\n\t// MatchingMaxTaskDeleteBatchSize is the max batch size for range deletion of tasks\n\t// KeyName: matching.maxTaskDeleteBatchSize\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingMaxTaskDeleteBatchSize\n\t// MatchingThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\n\t// KeyName: matching.throttledLogRPS\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: N/A\n\tMatchingThrottledLogRPS\n\t// MatchingNumTasklistWritePartitions is the number of write partitions for a task list\n\t// KeyName: matching.numTasklistWritePartitions\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingNumTasklistWritePartitions\n\t// MatchingNumTasklistReadPartitions is the number of read partitions for a task list\n\t// KeyName: matching.numTasklistReadPartitions\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingNumTasklistReadPartitions\n\t// MatchingForwarderMaxOutstandingPolls is the max number of inflight polls from the forwarder\n\t// KeyName: matching.forwarderMaxOutstandingPolls\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingForwarderMaxOutstandingPolls\n\t// MatchingForwarderMaxOutstandingTasks is the max number of inflight addTask/queryTask from the forwarder\n\t// KeyName: matching.forwarderMaxOutstandingTasks\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingForwarderMaxOutstandingTasks\n\t// MatchingForwarderMaxRatePerSecond is the max rate at which add/query can be forwarded\n\t// KeyName: matching.forwarderMaxRatePerSecond\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingForwarderMaxRatePerSecond\n\t// MatchingForwarderMaxChildrenPerNode is the max number of children per node in the task list partition tree\n\t// KeyName: matching.forwarderMaxChildrenPerNode\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingForwarderMaxChildrenPerNode\n\t// MatchingReadRangeSize is the read range size for the task reader\n\t// KeyName: matching.readRangeSize\n\t// Value type: Int\n\t// Default value: 50000\n\t// Allowed filters: N/A\n\tMatchingReadRangeSize\n\t// MatchingPartitionUpscaleRPS is the threshold of adding tasks RPS per partition to trigger upscale\n\t// KeyName: matching.partitionUpscaleRPS\n\t// Value type: Int\n\t// Default value: 200\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingPartitionUpscaleRPS\n\t// MatchingIsolationGroupsPerPartition is the target number of isolation groups to assign to each partition\n\t// KeyName: matching.isolationGroupsPerPartition\n\t// Value type: Int\n\t// Default value: 2\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIsolationGroupsPerPartition\n\t// MatchingPercentageOnboardedToShardManager is the percentage of task lists that will be onboarded to the shard manager.\n\t// KeyName: matching.percentageOnboardedToShardManager\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMatchingPercentageOnboardedToShardManager\n\n\t// key for history\n\n\t// HistoryRPS is request rate per second for each history host\n\t// KeyName: history.rps\n\t// Value type: Int\n\t// Default value: 3000\n\t// Allowed filters: N/A\n\tHistoryRPS\n\t// HistoryPersistenceMaxQPS is the max qps history host can query DB\n\t// KeyName: history.persistenceMaxQPS\n\t// Value type: Int\n\t// Default value: 9000\n\t// Allowed filters: N/A\n\tHistoryPersistenceMaxQPS\n\t// HistoryPersistenceGlobalMaxQPS is the max qps history cluster can query DB\n\t// KeyName: history.persistenceGlobalMaxQPS\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tHistoryPersistenceGlobalMaxQPS\n\t// HistoryVisibilityOpenMaxQPS is max qps one history host can write visibility open_executions\n\t// KeyName: history.historyVisibilityOpenMaxQPS\n\t// Value type: Int\n\t// Default value: 300\n\t// Allowed filters: DomainName\n\tHistoryVisibilityOpenMaxQPS\n\t// HistoryVisibilityClosedMaxQPS is max qps one history host can write visibility closed_executions\n\t// KeyName: history.historyVisibilityClosedMaxQPS\n\t// Value type: Int\n\t// Default value: 300\n\t// Allowed filters: DomainName\n\tHistoryVisibilityClosedMaxQPS\n\t// HistoryCacheInitialSize is initial size of history cache\n\t// KeyName: history.cacheInitialSize\n\t// Value type: Int\n\t// Default value: 128\n\t// Allowed filters: N/A\n\tHistoryCacheInitialSize\n\t// HistoryCacheMaxSize is max size of history cache\n\t// KeyName: history.cacheMaxSize\n\t// Value type: Int\n\t// Default value: 512\n\t// Allowed filters: N/A\n\tHistoryCacheMaxSize\n\t// ExecutionCacheMaxByteSize is the max byte size of history cache\n\t// KeyName: history.executionCacheMaxByteSize\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tExecutionCacheMaxByteSize\n\t// EventsCacheInitialCount is initial count of events cache\n\t// KeyName: history.eventsCacheInitialSize\n\t// Value type: Int\n\t// Default value: 128\n\t// Allowed filters: N/A\n\tEventsCacheInitialCount\n\t// EventsCacheMaxCount is max count of events cache\n\t// KeyName: history.eventsCacheMaxSize\n\t// Value type: Int\n\t// Default value: 512\n\t// Allowed filters: N/A\n\tEventsCacheMaxCount\n\t// EventsCacheMaxSize is max size of events cache in bytes\n\t// KeyName: history.eventsCacheMaxSizeInBytes\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tEventsCacheMaxSize\n\t// EventsCacheGlobalInitialCount is initial count of global events cache\n\t// KeyName: history.eventsCacheGlobalInitialSize\n\t// Value type: Int\n\t// Default value: 4096\n\t// Allowed filters: N/A\n\tEventsCacheGlobalInitialCount\n\t// EventsCacheGlobalMaxCount is max count of global events cache\n\t// KeyName: history.eventsCacheGlobalMaxSize\n\t// Value type: Int\n\t// Default value: 131072\n\t// Allowed filters: N/A\n\tEventsCacheGlobalMaxCount\n\t// AcquireShardConcurrency is number of goroutines that can be used to acquire shards in the shard controller.\n\t// KeyName: history.acquireShardConcurrency\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tAcquireShardConcurrency\n\t// TaskProcessRPS is the task processing rate per second for each domain\n\t// KeyName: history.taskProcessRPS\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: DomainName\n\tTaskProcessRPS\n\t// TaskSchedulerType is the task scheduler type for priority task processor\n\t// KeyName: history.taskSchedulerType\n\t// Value type: Int enum(1 for SchedulerTypeFIFO, 2 for SchedulerTypeWRR(weighted round robin scheduler implementation))\n\t// Default value: 2 (task.SchedulerTypeWRR)\n\t// Allowed filters: N/A\n\tTaskSchedulerType\n\t// TaskSchedulerWorkerCount is the number of workers per host in task scheduler\n\t// KeyName: history.taskSchedulerWorkerCount\n\t// Value type: Int\n\t// Default value: 200\n\t// Allowed filters: N/A\n\tTaskSchedulerWorkerCount\n\t// TaskSchedulerQueueSize is the size of task channel for host level task scheduler\n\t// KeyName: history.taskSchedulerQueueSize\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tTaskSchedulerQueueSize\n\t// TaskSchedulerDispatcherCount is the number of task dispatcher in task scheduler (only applies to host level task scheduler)\n\t// KeyName: history.taskSchedulerDispatcherCount\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tTaskSchedulerDispatcherCount\n\t// TaskSchedulerGlobalDomainRPS is the task scheduling domain rate limit per second for the whole Cadence cluster\n\t// KeyName: history.taskSchedulerGlobalDomainRPS\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: DomainName\n\tTaskSchedulerGlobalDomainRPS\n\t// TaskCriticalRetryCount is the critical retry count for background tasks\n\t// when task attempt exceeds this threshold:\n\t// - task attempt metrics and additional error logs will be emitted\n\t// - task priority will be lowered\n\t// KeyName: history.taskCriticalRetryCount\n\t// Value type: Int\n\t// Default value: 50\n\t// Allowed filters: N/A\n\tTaskCriticalRetryCount\n\t// QueueProcessorSplitMaxLevel is the max processing queue level\n\t// KeyName: history.queueProcessorSplitMaxLevel\n\t// Value type: Int\n\t// Default value: 2 // 3 levels, start from 0\n\t// Allowed filters: N/A\n\tQueueProcessorSplitMaxLevel\n\t// QueueMaxPendingTaskCount is the max number of pending tasks in the queue\n\t// KeyName: history.queueMaxPendingTaskCount\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tQueueMaxPendingTaskCount\n\t// QueueCriticalPendingTaskCount is the critical pending task count for the queue, which is supposed to be less than QueueMaxPendingTaskCount\n\t// KeyName: history.queueCriticalPendingTaskCount\n\t// Value type: Int\n\t// Default value: 9000\n\t// Allowed filters: N/A\n\tQueueCriticalPendingTaskCount\n\t// TimerTaskBatchSize is batch size for timer processor to process tasks\n\t// KeyName: history.timerTaskBatchSize\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tTimerTaskBatchSize\n\t// TimerTaskDeleteBatchSize is batch size for timer processor to delete timer tasks\n\t// KeyName: history.timerTaskDeleteBatchSize\n\t// Value type: Int\n\t// Default value: 4000\n\t// Allowed filters: N/A\n\tTimerTaskDeleteBatchSize\n\t// TimerProcessorGetFailureRetryCount is retry count for timer processor get failure operation\n\t// KeyName: history.timerProcessorGetFailureRetryCount\n\t// Value type: Int\n\t// Default value: 5\n\t// Allowed filters: N/A\n\tTimerProcessorGetFailureRetryCount\n\t// TimerProcessorCompleteTimerFailureRetryCount is retry count for timer processor complete timer operation\n\t// KeyName: history.timerProcessorCompleteTimerFailureRetryCount\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: N/A\n\tTimerProcessorCompleteTimerFailureRetryCount\n\t// TimerProcessorFailoverMaxPollRPS is max poll rate per second for timer processor\n\t// KeyName: history.timerProcessorFailoverMaxPollRPS\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tTimerProcessorFailoverMaxPollRPS\n\t// TimerProcessorMaxPollRPS is max poll rate per second for timer processor\n\t// KeyName: history.timerProcessorMaxPollRPS\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: N/A\n\tTimerProcessorMaxPollRPS\n\t// TimerProcessorMaxRedispatchQueueSize is the threshold of the number of tasks in the redispatch queue for timer processor\n\t// KeyName: history.timerProcessorMaxRedispatchQueueSize\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tTimerProcessorMaxRedispatchQueueSize\n\t// TimerProcessorHistoryArchivalSizeLimit is the max history size for inline archival\n\t// KeyName: history.timerProcessorHistoryArchivalSizeLimit\n\t// Value type: Int\n\t// Default value: 500*1024\n\t// Allowed filters: N/A\n\tTimerProcessorHistoryArchivalSizeLimit\n\n\t// TransferTaskBatchSize is batch size for transferQueueProcessor\n\t// KeyName: history.transferTaskBatchSize\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tTransferTaskBatchSize\n\t// TransferTaskDeleteBatchSize is batch size for transferQueueProcessor to delete transfer tasks\n\t// KeyName: history.transferTaskDeleteBatchSize\n\t// Value type: Int\n\t// Default value: 4000\n\t// Allowed filters: N/A\n\tTransferTaskDeleteBatchSize\n\t// TransferProcessorFailoverMaxPollRPS is max poll rate per second for transferQueueProcessor\n\t// KeyName: history.transferProcessorFailoverMaxPollRPS\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tTransferProcessorFailoverMaxPollRPS\n\t// TransferProcessorMaxPollRPS is max poll rate per second for transferQueueProcessor\n\t// KeyName: history.transferProcessorMaxPollRPS\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: N/A\n\tTransferProcessorMaxPollRPS\n\t// TransferProcessorCompleteTransferFailureRetryCount is times of retry for failure\n\t// KeyName: history.transferProcessorCompleteTransferFailureRetryCount\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: N/A\n\tTransferProcessorCompleteTransferFailureRetryCount\n\t// TransferProcessorMaxRedispatchQueueSize is the threshold of the number of tasks in the redispatch queue for transferQueueProcessor\n\t// KeyName: history.transferProcessorMaxRedispatchQueueSize\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tTransferProcessorMaxRedispatchQueueSize\n\t// ReplicatorTaskBatchSize is batch size for ReplicatorProcessor\n\t// KeyName: history.replicatorTaskBatchSize\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tReplicatorTaskBatchSize\n\t// ReplicatorMaxTaskBatchSize is the maximum batch size for ReplicatorProcessor\n\t// KeyName: history.replicatorMaxTaskBatchSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: ShardID, ClusterName\n\tReplicatorMaxTaskBatchSize\n\t// ReplicatorMinTaskBatchSize is the minimum batch size for ReplicatorProcessor\n\t// KeyName: history.replicatorMinTaskBatchSize\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: ShardID, ClusterName\n\tReplicatorMinTaskBatchSize\n\t// ReplicatorTaskBatchStepCount is how many batch size values between ReplicatorMinTaskBatchSize and ReplicatorMaxTaskBatchSize are used\n\t// \tby ReplicatorProcessor to adjust the batch size of the replication tasks. Less value will make the adjustment more aggressive.\n\t//  cannot be less than 3, otherwise only ReplicatorMinTaskBatchSize and ReplicatorMaxTaskBatchSize will be used.\n\t// KeyName: history.replicatorTaskBatchStepCount\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: ShardID, ClusterName\n\tReplicatorTaskBatchStepCount\n\t// ReplicatorTaskDeleteBatchSize is batch size for ReplicatorProcessor to delete replication tasks\n\t// KeyName: history.replicatorTaskDeleteBatchSize\n\t// Value type: Int\n\t// Default value: 4000\n\t// Allowed filters: N/A\n\tReplicatorTaskDeleteBatchSize\n\t// HistoryNodeDeleteBatchSize is batch size for deleting history nodes\n\t// KeyName: history.historyNodeDeleteBatchSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tHistoryNodeDeleteBatchSize\n\t// ReplicatorReadTaskMaxRetryCount is the number of read replication task retry time\n\t// KeyName: history.replicatorReadTaskMaxRetryCount\n\t// Value type: Int\n\t// Default value: 3\n\t// Allowed filters: N/A\n\tReplicatorReadTaskMaxRetryCount\n\t// ReplicatorCacheCapacity is the capacity of replication cache in number of tasks\n\t// KeyName: history.replicatorCacheCapacity\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tReplicatorCacheCapacity\n\t// ReplicatorCacheMaxSize is the max size of the replication cache in bytes\n\t// KeyName: history.replicatorCacheMaxSize\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tReplicatorCacheMaxSize\n\t// ReplicationBudgetManagerMaxSizeBytes is the max size of the replication budget manager cache in bytes\n\t// KeyName: history.replicationBudgetManagerMaxSizeBytes\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tReplicationBudgetManagerMaxSizeBytes\n\t// ReplicationBudgetManagerMaxSizeCount is the max count of the replication budget manager cache\n\t// KeyName: history.replicationBudgetManagerMaxSizeCount\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tReplicationBudgetManagerMaxSizeCount\n\n\t// MaximumBufferedEventsBatch is max number of buffer event in mutable state\n\t// KeyName: history.maximumBufferedEventsBatch\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tMaximumBufferedEventsBatch\n\t// MaximumSignalsPerExecution is max number of signals supported by single execution\n\t// KeyName: history.maximumSignalsPerExecution\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: DomainName\n\tMaximumSignalsPerExecution\n\t// NumArchiveSystemWorkflows is key for number of archive system workflows running in total\n\t// KeyName: history.numArchiveSystemWorkflows\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tNumArchiveSystemWorkflows\n\t// ArchiveRequestRPS is the rate limit on the number of archive request per second\n\t// KeyName: history.archiveRequestRPS\n\t// Value type: Int\n\t// Default value: 300 // should be much smaller than frontend RPS\n\t// Allowed filters: N/A\n\tArchiveRequestRPS\n\t// ArchiveInlineHistoryRPS is the (per instance) rate limit on the number of inline history archival attempts per second\n\t// KeyName: history.archiveInlineHistoryRPS\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tArchiveInlineHistoryRPS\n\t// ArchiveInlineHistoryGlobalRPS is the global rate limit on the number of inline history archival attempts per second\n\t// KeyName: history.archiveInlineHistoryGlobalRPS\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tArchiveInlineHistoryGlobalRPS\n\t// ArchiveInlineVisibilityRPS is the (per instance) rate limit on the number of inline visibility archival attempts per second\n\t// KeyName: history.archiveInlineVisibilityRPS\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tArchiveInlineVisibilityRPS\n\t// ArchiveInlineVisibilityGlobalRPS is the global rate limit on the number of inline visibility archival attempts per second\n\t// KeyName: history.archiveInlineVisibilityGlobalRPS\n\t// Value type: Int\n\t// Default value: 10000\n\t// Allowed filters: N/A\n\tArchiveInlineVisibilityGlobalRPS\n\t// HistoryMaxAutoResetPoints is the key for max number of auto reset points stored in mutableState\n\t// KeyName: history.historyMaxAutoResetPoints\n\t// Value type: Int\n\t// Default value: DefaultHistoryMaxAutoResetPoints\n\t// Allowed filters: DomainName\n\tHistoryMaxAutoResetPoints\n\t// ParentClosePolicyThreshold decides that parent close policy will be processed by sys workers(if enabled) if the number of children is greater than or equal to this threshold\n\t// KeyName: history.parentClosePolicyThreshold\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: DomainName\n\tParentClosePolicyThreshold\n\t// ParentClosePolicyBatchSize is the batch size of parent close policy processed by sys workers\n\t// KeyName: history.parentClosePolicyBatchSize\n\t// Value type: Int\n\t// Default value: 200\n\t// Allowed filters: DomainName\n\tParentClosePolicyBatchSize\n\t// NumParentClosePolicySystemWorkflows is key for number of parentClosePolicy system workflows running in total\n\t// KeyName: history.numParentClosePolicySystemWorkflows\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: N/A\n\tNumParentClosePolicySystemWorkflows\n\t// HistoryThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\n\t// KeyName: history.throttledLogRPS\n\t// Value type: Int\n\t// Default value: 4\n\t// Allowed filters: N/A\n\tHistoryThrottledLogRPS\n\t// DecisionRetryCriticalAttempts is decision attempt threshold for logging and emiting metrics\n\t// KeyName: history.decisionRetryCriticalAttempts\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: N/A\n\tDecisionRetryCriticalAttempts\n\t// DecisionRetryMaxAttempts is the max limit for decision retry attempts. 0 indicates infinite number of attempts.\n\t// KeyName: history.decisionRetryMaxAttempts\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: DomainName\n\tDecisionRetryMaxAttempts\n\t// NormalDecisionScheduleToStartMaxAttempts is the maximum decision attempt for creating a scheduleToStart timeout\n\t// timer for normal (non-sticky) decision\n\t// KeyName: history.normalDecisionScheduleToStartMaxAttempts\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tNormalDecisionScheduleToStartMaxAttempts\n\t// MaxBufferedQueryCount indicates the maximum number of queries which can be buffered at a given time for a single workflow\n\t// KeyName: history.MaxBufferedQueryCount\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tMaxBufferedQueryCount\n\t// MutableStateChecksumGenProbability is the probability [0-100] that checksum will be generated for mutable state\n\t// KeyName: history.mutableStateChecksumGenProbability\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tMutableStateChecksumGenProbability\n\t// MutableStateChecksumVerifyProbability is the probability [0-100] that checksum will be verified for mutable state\n\t// KeyName: history.mutableStateChecksumVerifyProbability\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tMutableStateChecksumVerifyProbability\n\t// TaskSchedulerMigrationRatio is the ratio of task that is migrated to new scheduler\n\t// KeyName: history.taskSchedulerMigrationRatio\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tTaskSchedulerMigrationRatio\n\t// MaxActivityCountDispatchByDomain max # of activity tasks to dispatch to matching before creating transfer tasks. This is an performance optimization to skip activity scheduling efforts.\n\t// KeyName: history.activityDispatchForSyncMatchCountByDomain\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tMaxActivityCountDispatchByDomain\n\n\t// key for history replication\n\n\t// ReplicationTaskFetcherParallelism determines how many go routines we spin up for fetching tasks\n\t// KeyName: history.ReplicationTaskFetcherParallelism\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tReplicationTaskFetcherParallelism\n\t// ReplicationTaskProcessorErrorRetryMaxAttempts is the max retry attempts for applying replication tasks\n\t// KeyName: history.ReplicationTaskProcessorErrorRetryMaxAttempts\n\t// Value type: Int\n\t// Default value: 10\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorErrorRetryMaxAttempts\n\n\t// WorkflowIDExternalRPS is the rate limit per workflowID for external calls\n\t// KeyName: history.workflowIDExternalRPS\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: DomainName\n\tWorkflowIDExternalRPS\n\t// WorkflowIDInternalRPS is the rate limit per workflowID for internal calls\n\t// KeyName: history.workflowIDInternalRPS\n\t// Value type: Int\n\t// Default value: UnlimitedRPS\n\t// Allowed filters: DomainName\n\tWorkflowIDInternalRPS\n\n\t// key for worker\n\n\t// WorkerPersistenceMaxQPS is the max qps worker host can query DB\n\t// KeyName: worker.persistenceMaxQPS\n\t// Value type: Int\n\t// Default value: 500\n\t// Allowed filters: N/A\n\tWorkerPersistenceMaxQPS\n\t// WorkerPersistenceGlobalMaxQPS is the max qps worker cluster can query DB\n\t// KeyName: worker.persistenceGlobalMaxQPS\n\t// Value type: Int\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tWorkerPersistenceGlobalMaxQPS\n\t// WorkerIndexerConcurrency is the max concurrent messages to be processed at any given time\n\t// KeyName: worker.indexerConcurrency\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tWorkerIndexerConcurrency\n\t// WorkerESProcessorNumOfWorkers is num of workers for esProcessor\n\t// KeyName: worker.ESProcessorNumOfWorkers\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\tWorkerESProcessorNumOfWorkers\n\t// WorkerESProcessorBulkActions is max number of requests in bulk for esProcessor\n\t// KeyName: worker.ESProcessorBulkActions\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tWorkerESProcessorBulkActions\n\t// WorkerESProcessorBulkSize is max total size of bulk in bytes for esProcessor\n\t// KeyName: worker.ESProcessorBulkSize\n\t// Value type: Int\n\t// Default value: 2<<24 // 16MB\n\t// Allowed filters: N/A\n\tWorkerESProcessorBulkSize\n\t// WorkerArchiverConcurrency is controls the number of coroutines handling archival work per archival workflow\n\t// KeyName: worker.ArchiverConcurrency\n\t// Value type: Int\n\t// Default value: 50\n\t// Allowed filters: N/A\n\tWorkerArchiverConcurrency\n\t// WorkerArchivalsPerIteration is controls the number of archivals handled in each iteration of archival workflow\n\t// KeyName: worker.ArchivalsPerIteration\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tWorkerArchivalsPerIteration\n\t// WorkerThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\n\t// KeyName: worker.throttledLogRPS\n\t// Value type: Int\n\t// Default value: 20\n\t// Allowed filters: N/A\n\tWorkerThrottledLogRPS\n\t// ScannerPersistenceMaxQPS is the maximum rate of persistence calls from worker.Scanner\n\t// KeyName: worker.scannerPersistenceMaxQPS\n\t// Value type: Int\n\t// Default value: 5\n\t// Allowed filters: N/A\n\tScannerPersistenceMaxQPS\n\t// ScannerGetOrphanTasksPageSize is the maximum number of orphans to delete in one batch\n\t// KeyName: worker.scannerGetOrphanTasksPageSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tScannerGetOrphanTasksPageSize\n\t// ScannerBatchSizeForTasklistHandler is for: 1. max number of tasks to query per call(get tasks for tasklist) in the scavenger handler. 2. The scavenger then uses the return to decide if a tasklist can be deleted. It's better to keep it a relatively high number to let it be more efficient.\n\t// KeyName: worker.scannerBatchSizeForTasklistHandler\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tScannerBatchSizeForTasklistHandler\n\t// ScannerMaxTasksProcessedPerTasklistJob is the number of tasks to process for a tasklist in each workflow run\n\t// KeyName: worker.scannerMaxTasksProcessedPerTasklistJob\n\t// Value type: Int\n\t// Default value: 256\n\t// Allowed filters: N/A\n\tScannerMaxTasksProcessedPerTasklistJob\n\t// ConcreteExecutionsScannerConcurrency indicates the concurrency of concrete execution scanner\n\t// KeyName: worker.executionsScannerConcurrency\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerConcurrency\n\t// ConcreteExecutionsScannerBlobstoreFlushThreshold indicates the flush threshold of blobstore in concrete execution scanner\n\t// KeyName: worker.executionsScannerBlobstoreFlushThreshold\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerBlobstoreFlushThreshold\n\t// ConcreteExecutionsScannerActivityBatchSize indicates the batch size of scanner activities\n\t// KeyName: worker.executionsScannerActivityBatchSize\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerActivityBatchSize\n\t// ConcreteExecutionsScannerPersistencePageSize indicates the page size of execution persistence fetches in concrete execution scanner\n\t// KeyName: worker.executionsScannerPersistencePageSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerPersistencePageSize\n\t// CurrentExecutionsScannerConcurrency indicates the concurrency of current executions scanner\n\t// KeyName: worker.currentExecutionsConcurrency\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerConcurrency\n\t// CurrentExecutionsScannerBlobstoreFlushThreshold indicates the flush threshold of blobstore in current executions scanner\n\t// KeyName: worker.currentExecutionsBlobstoreFlushThreshold\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerBlobstoreFlushThreshold\n\t// CurrentExecutionsScannerActivityBatchSize indicates the batch size of scanner activities\n\t// KeyName: worker.currentExecutionsActivityBatchSize\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerActivityBatchSize\n\t// CurrentExecutionsScannerPersistencePageSize indicates the page size of execution persistence fetches in current executions scanner\n\t// KeyName: worker.currentExecutionsPersistencePageSize\n\t// Value type: INt\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerPersistencePageSize\n\t// TimersScannerConcurrency is the concurrency of timers scanner\n\t// KeyName: worker.timersScannerConcurrency\n\t// Value type: Int\n\t// Default value: 5\n\t// Allowed filters: N/A\n\tTimersScannerConcurrency\n\t// TimersScannerPersistencePageSize is the page size of timers persistence fetches in timers scanner\n\t// KeyName: worker.timersScannerPersistencePageSize\n\t// Value type: Int\n\t// Default value: 1000\n\t// Allowed filters: N/A\n\tTimersScannerPersistencePageSize\n\t// TimersScannerBlobstoreFlushThreshold is threshold to flush blob store\n\t// KeyName: worker.timersScannerBlobstoreFlushThreshold\n\t// Value type: Int\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tTimersScannerBlobstoreFlushThreshold\n\t// TimersScannerActivityBatchSize is TimersScannerActivityBatchSize\n\t// KeyName: worker.timersScannerActivityBatchSize\n\t// Value type: Int\n\t// Default value: 25\n\t// Allowed filters: N/A\n\tTimersScannerActivityBatchSize\n\t// TimersScannerPeriodStart is interval start for fetching scheduled timers\n\t// KeyName: worker.timersScannerPeriodStart\n\t// Value type: Int\n\t// Default value: 24\n\t// Allowed filters: N/A\n\tTimersScannerPeriodStart\n\t// TimersScannerPeriodEnd is interval end for fetching scheduled timers\n\t// KeyName: worker.timersScannerPeriodEnd\n\t// Value type: Int\n\t// Default value: 3\n\t// Allowed filters: N/A\n\tTimersScannerPeriodEnd\n\t// ESAnalyzerMaxNumDomains defines how many domains to check\n\t// KeyName: worker.ESAnalyzerMaxNumDomains\n\t// Value type: int\n\t// Default value: 500\n\tESAnalyzerMaxNumDomains\n\t// ESAnalyzerMaxNumWorkflowTypes defines how many workflow types to check per domain\n\t// KeyName: worker.ESAnalyzerMaxNumWorkflowTypes\n\t// Value type: int\n\t// Default value: 100\n\tESAnalyzerMaxNumWorkflowTypes\n\t// ESAnalyzerNumWorkflowsToRefresh controls how many workflows per workflow type should be refreshed per workflow type\n\t// KeyName: worker.ESAnalyzerNumWorkflowsToRefresh\n\t// Value type: Int\n\t// Default value: 100\n\tESAnalyzerNumWorkflowsToRefresh\n\t// ESAnalyzerMinNumWorkflowsForAvg controls how many workflows to have at least to rely on workflow run time avg per type\n\t// KeyName: worker.ESAnalyzerMinNumWorkflowsForAvg\n\t// Value type: Int\n\t// Default value: 100\n\tESAnalyzerMinNumWorkflowsForAvg\n\n\t// VisibilityArchivalQueryMaxRangeInDays is the maximum number of days for a visibility archival query\n\t// KeyName: frontend.visibilityArchivalQueryMaxRangeInDays\n\t// Value type: Int\n\t// Default value: 60\n\t// Allowed filters: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tVisibilityArchivalQueryMaxRangeInDays\n\t// VisibilityArchivalQueryMaxQPS is the timeout for a visibility archival query\n\t// KeyName: frontend.visibilityArchivalQueryMaxQPS\n\t// Value type: Int\n\t// Default value: 1\n\t// Allowed filters: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tVisibilityArchivalQueryMaxQPS\n\n\t// WorkflowDeletionJitterRange defines the duration in minutes for workflow close tasks jittering\n\t// KeyName: system.workflowDeletionJitterRange\n\t// Value type: Int\n\t// Default value: 60\n\tWorkflowDeletionJitterRange\n\n\t// SampleLoggingRate defines the rate we want sampled logs to be logged at\n\t// KeyName: system.sampleLoggingRate\n\t// Value type: Int\n\t// Default value: 100\n\tSampleLoggingRate\n\t// LargeShardHistorySizeMetricThreshold defines the threshold for what constitutes a large history storage size to alert on\n\t// KeyName: system.largeShardHistorySizeMetricThreshold\n\t// Value type: Int\n\t// Default value: 10485760 (10mb)\n\tLargeShardHistorySizeMetricThreshold\n\t// LargeShardHistoryEventMetricThreshold defines the threshold for what constitutes a large history event size to alert on\n\t// KeyName: system.largeShardHistoryEventMetricThreshold\n\t// Value type: Int\n\t// Default value: 50 * 1024\n\tLargeShardHistoryEventMetricThreshold\n\t// LargeShardHistoryBlobMetricThreshold defines the threshold for what constitutes a large history blob size to alert on\n\t// KeyName: system.largeShardHistoryBlobMetricThreshold\n\t// Value type: Int\n\t// Default value: 262144 (1/4mb)\n\tLargeShardHistoryBlobMetricThreshold\n\n\t// IsolationGroupStateUpdateRetryAttempts\n\t// KeyName: system.isolationGroupStateUpdateRetryAttempts\n\t// Value type: int\n\t// Default value: 2\n\tIsolationGroupStateUpdateRetryAttempts\n\n\t// DeleteHistoryEventContextTimeout in seconds\n\t// KeyName: system.deleteHistoryEventContextTimeout\n\t// Value type: Int\n\t// Default value: 30\n\tDeleteHistoryEventContextTimeout\n\n\t// QueueMaxVirtualQueueCount is the max number of virtual queues\n\t// KeyName: history.queueMaxVirtualQueueCount\n\t// Value type: Int\n\t// Default value: 2\n\t// Allowed filters: N/A\n\tQueueMaxVirtualQueueCount\n\n\t// ShardDistributorMaxEtcdTxnOps is the maximum number of operations per etcd transaction.\n\t// etcd enforces a server-side limit (--max-txn-ops, default 128).\n\t// This value must not exceed the etcd cluster's configured limit.\n\t// KeyName: shardDistributor.maxEtcdTxnOps\n\t// Value type: Int\n\t// Default value: 128\n\t// Allowed filters: N/A\n\tShardDistributorMaxEtcdTxnOps\n\n\t// LastIntKey must be the last one in this const group\n\tLastIntKey\n)\n\nconst (\n\tUnknownBoolKey BoolKey = iota\n\n\t// key for tests\n\tTestGetBoolPropertyKey\n\tTestGetBoolPropertyFilteredByDomainIDKey\n\tTestGetBoolPropertyFilteredByTaskListInfoKey\n\tTestGetBoolPropertyFilteredByDomainKey\n\tTestGetBoolPropertyFilteredByDomainIDAndWorkflowIDKey\n\tTestGetBoolPropertyFilteredByShardIDKey\n\n\t// key for common & admin\n\n\t// EnableVisibilitySampling is key for enable visibility sampling for basic(DB based) visibility\n\t// KeyName: system.enableVisibilitySampling\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableVisibilitySampling\n\t// EnableReadFromClosedExecutionV2 is key for enable read from cadence_visibility.closed_executions_v2\n\t// KeyName: system.enableReadFromClosedExecutionV2\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableReadFromClosedExecutionV2\n\t// EnableLogCustomerQueryParameter is key for enable log customer query parameters\n\t// KeyName: system.enableLogCustomerQueryParameter\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableLogCustomerQueryParameter\n\t// EmitShardDiffLog is whether emit the shard diff log\n\t// KeyName: history.emitShardDiffLog\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEmitShardDiffLog\n\t// DisableListVisibilityByFilter is config to disable list open/close workflow using filter\n\t// KeyName: frontend.disableListVisibilityByFilter\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tDisableListVisibilityByFilter\n\t// EnableReadFromHistoryArchival is key for enabling reading history from archival store\n\t// KeyName: system.enableReadFromHistoryArchival\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableReadFromHistoryArchival\n\t// EnableReadFromVisibilityArchival is key for enabling reading visibility from archival store to override the value from static config.\n\t// KeyName: system.enableReadFromVisibilityArchival\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableReadFromVisibilityArchival\n\t// EnableDomainNotActiveAutoForwarding decides requests form which domain will be forwarded to active cluster if domain is not active in current cluster.\n\t// Only when \"selected-api-forwarding\" or \"all-domain-apis-forwarding\" is the policy in ClusterRedirectionPolicy(in static config).\n\t// If the policy is \"noop\"(default) this flag is not doing anything.\n\t// KeyName: system.enableDomainNotActiveAutoForwarding\n\t// Value type: Bool\n\t// Default value: true (meaning all domains are allowed to use the policy specified in static config)\n\t// Allowed filters: DomainName\n\tEnableDomainNotActiveAutoForwarding\n\t// EnableGracefulFailover is whether enabling graceful failover\n\t// KeyName: system.enableGracefulFailover\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableGracefulFailover\n\t// DisallowQuery is the key to disallow query for a domain\n\t// KeyName: system.disallowQuery\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tDisallowQuery\n\t// EnableDebugMode is for enabling debugging components, logs and metrics\n\t// KeyName: system.enableDebugMode\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableDebugMode\n\t// EnableGRPCOutbound is the key for enabling outbound GRPC traffic\n\t// KeyName: system.enableGRPCOutbound\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableGRPCOutbound\n\t// EnableSQLAsyncTransaction is the key for enabling async transaction\n\t// KeyName: system.enableSQLAsyncTransaction\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableSQLAsyncTransaction\n\t// EnableConnectionRetainingDirectChooser is the key for enabling connection retaining direct yarpc chooser\n\t// KeyName: system.enableConnectionRetainingDirectChooser\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableConnectionRetainingDirectChooser\n\t// EnableDomainAuditLogging enables audit logging for a domain to the domain audit log table\n\t// KeyName: system.enableDomainAuditLogging\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableDomainAuditLogging\n\n\t// key for frontend\n\n\t// EnableClientVersionCheck is enables client version check for frontend\n\t// KeyName: frontend.enableClientVersionCheck\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableClientVersionCheck\n\t// SendRawWorkflowHistory is whether to enable raw history retrieving\n\t// KeyName: frontend.sendRawWorkflowHistory\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tSendRawWorkflowHistory\n\t// FrontendEmitSignalNameMetricsTag enables emitting signal name tag in metrics in frontend client\n\t// KeyName: frontend.emitSignalNameMetricsTag\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tFrontendEmitSignalNameMetricsTag\n\t// EnableQueryAttributeValidation enables validation of queries' search attributes against the dynamic config whitelist\n\t// Keyname: frontend.enableQueryAttributeValidation\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableQueryAttributeValidation\n\n\t// key for matching\n\n\t// MatchingEnableSyncMatch is to enable sync match\n\t// KeyName: matching.enableSyncMatch\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnableSyncMatch\n\t// MatchingEnableTaskInfoLogByDomainID is enables info level logs for decision/activity task based on the request domainID\n\t// KeyName: matching.enableTaskInfoLogByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tMatchingEnableTaskInfoLogByDomainID\n\t// MatchingEnableTasklistGuardAgainstOwnershipShardLoss\n\t// enables guards to prevent tasklists from processing if there is any detection that the host\n\t// no longer is active or owns the shard\n\t// KeyName: matching.enableTasklistGuardAgainstOwnershipLoss\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tMatchingEnableTasklistGuardAgainstOwnershipShardLoss\n\t// MatchingEnableStandbyTaskCompletion is to enable completion of tasks in the domain's passive side\n\t// KeyName: matching.enableStandbyTaskCompletion\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnableStandbyTaskCompletion\n\n\t// MatchingEnableGetNumberOfPartitionsFromCache is to enable getting number of partitions from cache instead of dynamic config\n\t// KeyName: matching.enableGetNumberOfPartitionsFromCache\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnableGetNumberOfPartitionsFromCache\n\t// MatchingEnableAdaptiveScaler is to enable adaptive task list scaling\n\t// KeyName: matching.enableAdaptiveScaler\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnableAdaptiveScaler\n\t// MatchingEnablePartitionEmptyCheck enables using TaskListStatus.empty to check if a partition is empty\n\t// KeyName: matching.enablePartitionEmptyCheck\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnablePartitionEmptyCheck\n\t// MatchingEnableReturnAllTaskListKinds returns TaskLists of all kinds when GetTaskListsByDomain is called. Useful in testing Cadence\n\t// KeyName: matching.matchingReturnAllTaskListKinds\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tMatchingEnableReturnAllTaskListKinds\n\n\t// key for history\n\n\t// EventsCacheGlobalEnable is enables global cache over all history shards\n\t// KeyName: history.eventsCacheGlobalEnable\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEventsCacheGlobalEnable\n\t// QueueProcessorEnableSplit indicates whether processing queue split policy should be enabled\n\t// KeyName: history.queueProcessorEnableSplit\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tQueueProcessorEnableSplit\n\t// QueueProcessorEnableRandomSplitByDomainID indicates whether random queue split policy should be enabled for a domain\n\t// KeyName: history.queueProcessorEnableRandomSplitByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tQueueProcessorEnableRandomSplitByDomainID\n\t// QueueProcessorEnablePendingTaskSplitByDomainID indicates whether pending task split policy should be enabled\n\t// KeyName: history.queueProcessorEnablePendingTaskSplitByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tQueueProcessorEnablePendingTaskSplitByDomainID\n\t// QueueProcessorEnableStuckTaskSplitByDomainID indicates whether stuck task split policy should be enabled\n\t// KeyName: history.queueProcessorEnableStuckTaskSplitByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tQueueProcessorEnableStuckTaskSplitByDomainID\n\t// QueueProcessorEnablePersistQueueStates indicates whether processing queue states should be persisted\n\t// KeyName: history.queueProcessorEnablePersistQueueStates\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tQueueProcessorEnablePersistQueueStates\n\t// QueueProcessorEnableLoadQueueStates indicates whether processing queue states should be loaded\n\t// KeyName: history.queueProcessorEnableLoadQueueStates\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tQueueProcessorEnableLoadQueueStates\n\t// QueueProcessorEnableGracefulSyncShutdown indicates whether processing queue should be shutdown gracefully & synchronously\n\t// KeyName: history.queueProcessorEnableGracefulSyncShutdown\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tQueueProcessorEnableGracefulSyncShutdown\n\t// TransferProcessorEnableValidator is whether validator should be enabled for transferQueueProcessor\n\t// KeyName: history.transferProcessorEnableValidator\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tTransferProcessorEnableValidator\n\t// TaskSchedulerEnableRateLimiter indicates whether the task scheduler rate limiter is enabled\n\t// KeyName: history.taskSchedulerEnableRateLimiter\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tTaskSchedulerEnableRateLimiter\n\t// TaskSchedulerEnableRateLimiterShadowMode indicates whether the task scheduler rate limiter is in shadow mode\n\t// KeyName: history.taskSchedulerEnableRateLimiterShadowMode\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName\n\tTaskSchedulerEnableRateLimiterShadowMode\n\t// TaskSchedulerEnableMigration indicates whether the task scheduler migration is enabled\n\t// KeyName: history.taskSchedulerEnableMigration\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tTaskSchedulerEnableMigration\n\t// EnableAdminProtection is whether to enable admin checking\n\t// KeyName: history.enableAdminProtection\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableAdminProtection\n\t// EnableParentClosePolicy is whether to  ParentClosePolicy\n\t// KeyName: history.enableParentClosePolicy\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName\n\tEnableParentClosePolicy\n\t// EnableDropStuckTaskByDomainID is whether stuck timer/transfer task should be dropped for a domain\n\t// KeyName: history.DropStuckTaskByDomain\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tEnableDropStuckTaskByDomainID\n\t// EnableConsistentQuery indicates if consistent query is enabled for the cluster\n\t// KeyName: history.EnableConsistentQuery\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableConsistentQuery\n\t// EnableConsistentQueryByDomain indicates if consistent query is enabled for a domain\n\t// KeyName: history.EnableConsistentQueryByDomain\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableConsistentQueryByDomain\n\t// EnableContextHeaderInVisibility is key for enable context header in visibility\n\t// KeyName: history.enableContextHeaderInVisibility\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableContextHeaderInVisibility\n\t// EnableCrossClusterOperationsForDomain indicates if cross cluster operations can be scheduled for a domain\n\t// KeyName: history.enableCrossClusterOperations\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableCrossClusterOperationsForDomain\n\t// EnableHistoryCorruptionCheck enables additional sanity check for corrupted history. This allows early catches of DB corruptions but potiantally increased latency.\n\t// KeyName: history.enableHistoryCorruptionCheck\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableHistoryCorruptionCheck\n\t// EnableActivityLocalDispatchByDomain is allows worker to dispatch activity tasks through local tunnel after decisions are made. This is an performance optimization to skip activity scheduling efforts\n\t// KeyName: history.enableActivityLocalDispatchByDomain\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName\n\tEnableActivityLocalDispatchByDomain\n\t// HistoryEnableTaskInfoLogByDomainID is enables info level logs for decision/activity task based on the request domainID\n\t// KeyName: history.enableTaskInfoLogByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tHistoryEnableTaskInfoLogByDomainID\n\t// EnableReplicationTaskGeneration is the flag to control replication generation\n\t// KeyName: history.enableReplicationTaskGeneration\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainID, WorkflowID\n\tEnableReplicationTaskGeneration\n\t// UseNewInitialFailoverVersion is a switch to issue a failover version based on the minFailoverVersion\n\t// rather than the default initialFailoverVersion. USed as a per-domain migration switch\n\t// KeyName: history.useNewInitialFailoverVersion\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tUseNewInitialFailoverVersion\n\t// EnableRecordWorkflowExecutionUninitialized enables record workflow execution uninitialized state in ElasticSearch\n\t// KeyName: history.EnableRecordWorkflowExecutionUninitialized\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName\n\tEnableRecordWorkflowExecutionUninitialized\n\t// EnableCleanupOrphanedHistoryBranchOnWorkflowCreation enables cleanup of orphaned history branches when CreateWorkflowExecution fails\n\t// KeyName: history.enableCleanupOrphanedHistoryBranchOnWorkflowCreation\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableCleanupOrphanedHistoryBranchOnWorkflowCreation\n\t// AllowArchivingIncompleteHistory will continue on when seeing some error like history mutated(usually caused by database consistency issues)\n\t// KeyName: worker.AllowArchivingIncompleteHistory\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tAllowArchivingIncompleteHistory\n\t// EnableCleaningOrphanTaskInTasklistScavenger indicates if enabling the scanner to clean up orphan tasks\n\t// Only implemented for single SQL database. TODO https://github.com/uber/cadence/issues/4064 for supporting multiple/sharded SQL database and NoSQL\n\t// KeyName: worker.enableCleaningOrphanTaskInTasklistScavenger\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableCleaningOrphanTaskInTasklistScavenger\n\t// TaskListScannerEnabled indicates if task list scanner should be started as part of worker.Scanner\n\t// KeyName: worker.taskListScannerEnabled\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tTaskListScannerEnabled\n\t// HistoryScannerEnabled indicates if history scanner should be started as part of worker.Scanner\n\t// KeyName: worker.historyScannerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tHistoryScannerEnabled\n\t// ConcreteExecutionsScannerEnabled indicates if executions scanner should be started as part of worker.Scanner\n\t// KeyName: worker.executionsScannerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerEnabled\n\t// ConcreteExecutionsScannerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\n\t// KeyName: worker.executionsScannerInvariantCollectionMutableState\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerInvariantCollectionMutableState\n\t// ConcreteExecutionsScannerInvariantCollectionHistory indicates if history invariant checks should be run\n\t// KeyName: worker.executionsScannerInvariantCollectionHistory\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerInvariantCollectionHistory\n\t// ConcreteExecutionsFixerInvariantCollectionStale indicates if the stale workflow invariant should be run\n\t// KeyName: worker.executionsFixerInvariantCollectionStale\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tConcreteExecutionsFixerInvariantCollectionStale\n\t// ConcreteExecutionsFixerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\n\t// KeyName: worker.executionsFixerInvariantCollectionMutableState\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tConcreteExecutionsFixerInvariantCollectionMutableState\n\t// ConcreteExecutionsFixerInvariantCollectionHistory indicates if history invariant checks should be run\n\t// KeyName: worker.executionsFixerInvariantCollectionHistory\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tConcreteExecutionsFixerInvariantCollectionHistory\n\t// ConcreteExecutionsScannerInvariantCollectionStale indicates if the stale workflow invariant should be run\n\t// KeyName: worker.executionsScannerInvariantCollectionStale\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tConcreteExecutionsScannerInvariantCollectionStale\n\t// CurrentExecutionsScannerEnabled indicates if current executions scanner should be started as part of worker.Scanner\n\t// KeyName: worker.currentExecutionsScannerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerEnabled\n\t// CurrentExecutionsScannerInvariantCollectionHistory indicates if history invariant checks should be run\n\t// KeyName: worker.currentExecutionsScannerInvariantCollectionHistory\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerInvariantCollectionHistory\n\t// CurrentExecutionsScannerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\n\t// KeyName: worker.currentExecutionsInvariantCollectionMutableState\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tCurrentExecutionsScannerInvariantCollectionMutableState\n\t// EnableBatcher is decides whether start batcher in our worker\n\t// KeyName: worker.enableBatcher\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableBatcher\n\t// EnableScheduler decides whether to start the scheduler worker for cron-based scheduling\n\t// KeyName: worker.enableScheduler\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableScheduler\n\t// EnableParentClosePolicyWorker decides whether or not enable system workers for processing parent close policy task\n\t// KeyName: system.enableParentClosePolicyWorker\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableParentClosePolicyWorker\n\t// EnableESAnalyzer decides whether to enable system workers for processing ElasticSearch Analyzer\n\t// KeyName: system.enableESAnalyzer\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableESAnalyzer\n\t// EnableAsyncWorkflowConsumption decides whether to enable system workers for processing async workflows\n\t// KeyName: worker.enableAsyncWorkflowConsumption\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableAsyncWorkflowConsumption\n\n\t// EnableStickyQuery indicates if sticky query should be enabled per domain\n\t// KeyName: system.enableStickyQuery\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: DomainName\n\tEnableStickyQuery\n\t// EnableFailoverManager indicates if failover manager is enabled\n\t// KeyName: system.enableFailoverManager\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tEnableFailoverManager\n\t// ConcreteExecutionFixerDomainAllow is which domains are allowed to be fixed by concrete fixer workflow\n\t// KeyName: worker.concreteExecutionFixerDomainAllow\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tConcreteExecutionFixerDomainAllow\n\t// CurrentExecutionFixerDomainAllow is which domains are allowed to be fixed by current fixer workflow\n\t// KeyName: worker.currentExecutionFixerDomainAllow\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tCurrentExecutionFixerDomainAllow\n\t// TimersScannerEnabled is if timers scanner should be started as part of worker.Scanner\n\t// KeyName: worker.timersScannerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tTimersScannerEnabled\n\t// TimersFixerEnabled is if timers fixer should be started as part of worker.Scanner\n\t// KeyName: worker.timersFixerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tTimersFixerEnabled\n\t// TimersFixerDomainAllow is which domains are allowed to be fixed by timer fixer workflow\n\t// KeyName: worker.timersFixerDomainAllow\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tTimersFixerDomainAllow\n\t// ConcreteExecutionFixerEnabled is if concrete execution fixer workflow is enabled\n\t// KeyName: worker.concreteExecutionFixerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tConcreteExecutionFixerEnabled\n\t// CurrentExecutionFixerEnabled is if current execution fixer workflow is enabled\n\t// KeyName: worker.currentExecutionFixerEnabled\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tCurrentExecutionFixerEnabled\n\n\t// EnableAuthorization is the key to enable authorization for a domain, only for extension binary:\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableAuthorization\n\t// EnableServiceAuthorization is the key to enable authorization for a service, only for extension binary:\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableServiceAuthorization\n\t// EnableServiceAuthorizationLogOnly is the key to enable authorization logging for a service, only for extension binary:\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableServiceAuthorizationLogOnly\n\t// ESAnalyzerPause defines if we want to dynamically pause the analyzer workflow\n\t// KeyName: worker.ESAnalyzerPause\n\t// Value type: bool\n\t// Default value: false\n\tESAnalyzerPause\n\t// EnableArchivalCompression indicates whether blobs are compressed before they are archived\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableArchivalCompression\n\t// ESAnalyzerEnableAvgDurationBasedChecks controls if we want to enable avg duration based task refreshes\n\t// KeyName: worker.ESAnalyzerEnableAvgDurationBasedChecks\n\t// Value type: Bool\n\t// Default value: false\n\tESAnalyzerEnableAvgDurationBasedChecks\n\n\t// Lockdown defines if we want to allow failovers of domains to this cluster\n\t// KeyName: system.Lockdown\n\t// Value type: bool\n\t// Default value: false\n\tLockdown\n\n\t// EnablePendingActivityValidation is feature flag if pending activity count validation is enabled\n\t// KeyName: limit.pendingActivityCount.enabled\n\t// Value type: bool\n\t// Default value: false\n\tEnablePendingActivityValidation\n\n\t// EnableCassandraAllConsistencyLevelDelete uses all consistency level for Cassandra delete operations\n\t// KeyName: system.enableCassandraAllConsistencyLevelDelete\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableCassandraAllConsistencyLevelDelete\n\n\t// EnableTasklistIsolation Is a feature to enable subdivision of workflows by units called 'isolation-groups'\n\t// and to control their movement and blast radius. This has some nontrivial operational overheads in management\n\t// and a good understanding of poller distribution, so probably not worth enabling unless it's well understood.\n\t// KeyName: system.enableTasklistIsolation\n\t// Value type: bool\n\t// Default value: false\n\tEnableTasklistIsolation\n\n\t// EnablePartitionIsolationGroupAssignment enables assigning isolation groups to individual TaskList partitions\n\t// KeyName: matching.enablePartitionIsolationGroupAssignment\n\t// Value type: bool\n\t// Default value: false\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tEnablePartitionIsolationGroupAssignment\n\n\t// EnableShardIDMetrics turns on or off shardId metrics\n\t// KeyName: system.enableShardIDMetrics\n\t// Value type: Bool\n\t// Default value: true\n\tEnableShardIDMetrics\n\n\t// EnableTimerDebugLogByDomainID enables log for debugging timer task issue by domain\n\t// KeyName: history.enableTimerDebugLogByDomainID\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainID\n\tEnableTimerDebugLogByDomainID\n\n\t// EnableRetryForChecksumFailure enables retry if mutable state checksum verification fails\n\t// KeyName: history.enableMutableStateChecksumFailureRetry\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableRetryForChecksumFailure\n\n\t// EnableStrongIdempotency enables strong idempotency for APIs\n\t// KeyName: history.enableStrongIdempotency\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableStrongIdempotency\n\n\t// EnableStrongIdempotencySanityCheck enables sanity check for strong idempotency\n\t// KeyName: history.enableStrongIdempotencySanityCheck\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableStrongIdempotencySanityCheck\n\n\t// MatchingEnableClientAutoConfig enables auto config for domain and tasklist on client (worker) side\n\t// KeyName: matching.enableClientAutoConfig\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingEnableClientAutoConfig\n\n\t// EnableNoSQLHistoryTaskDualWriteMode is to enable dual write of history events\n\t// KeyName: history.enableNoSQLHistoryTaskDualWrite\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableNoSQLHistoryTaskDualWriteMode\n\t// ReadNoSQLHistoryTaskFromDataBlob is to read history tasks from data blob\n\t// KeyName: history.readNoSQLHistoryTaskFromDataBlob\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tReadNoSQLHistoryTaskFromDataBlob\n\t// ReadNoSQLShardFromDataBlob is to read shards from data blob\n\t// KeyName: history.readNoSQLShardFromDataBlob\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tReadNoSQLShardFromDataBlob\n\t// EnableSizeBasedHistoryExecutionCache is the feature flag to enable size based cache for execution cache\n\t// KeyName: history.enableSizeBasedHistoryExecutionCache\n\t// Value type: Bool\n\t// Default value: false\n\tEnableSizeBasedHistoryExecutionCache\n\n\t// EnableSizeBasedHistoryEventCache is the feature flag to enable size based cache for event cache\n\t// KeyName: history.enableSizeBasedHistoryEventCache\n\t// Value type: Bool\n\t// Default value: false\n\tEnableSizeBasedHistoryEventCache\n\n\t// DisableTransferFailoverQueue is to disable transfer failover queue\n\t// KeyName: history.disableTransferFailoverQueue\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tDisableTransferFailoverQueue\n\t// DisableTimerFailoverQueue is to disable timer failover queue\n\t// KeyName: history.disableTimerFailoverQueue\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tDisableTimerFailoverQueue\n\n\t// EnableTransferQueueV2 is to enable transfer queue v2\n\t// KeyName: history.enableTransferQueueV2\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: ShardID\n\tEnableTransferQueueV2\n\t// EnableTimerQueueV2 is to enable timer queue v2\n\t// KeyName: history.enableTimerQueueV2\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: ShardID\n\tEnableTimerQueueV2\n\t// EnableTransferQueueV2PendingTaskCountAlert is to enable transfer queue v2 pending task count alert\n\t// KeyName: history.enableTransferQueueV2PendingTaskCountAlert\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: ShardID\n\tEnableTransferQueueV2PendingTaskCountAlert\n\t// EnableTimerQueueV2PendingTaskCountAlert is to enable timer queue v2 pending task count alert\n\t// KeyName: history.enableTimerQueueV2PendingTaskCountAlert\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: ShardID\n\tEnableTimerQueueV2PendingTaskCountAlert\n\n\t// EnableActiveClusterSelectionPolicyInStartWorkflow is to enable active cluster selection policy in start workflow requests for a domain\n\t// KeyName: frontend.enableActiveClusterSelectionPolicyInStartWorkflow\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableActiveClusterSelectionPolicyInStartWorkflow\n\n\t// EnforceDecisionTaskAttempts is the key for enforcing decision retry attempts limit in case of timeouts.\n\t// KeyName: history.enforceDecisionTaskAttempts\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnforceDecisionTaskAttempts\n\n\t// MatchingExcludeShortLivedTaskListsFromShardManager excludes short-lived task lists (e.g. bits task lists and sticky task lists)\n\t// from using the shard manager to handle these shards. These short-lived task lists are assigned using hash_ring.\n\t// KeyName: matching.excludeShortLivedTaskListsFromShardManager\n\t// Value type: Bool\n\t// Default value: true\n\t// Allowed filters: N/A\n\tMatchingExcludeShortLivedTaskListsFromShardManager\n\n\t// EnableHierarchicalWeightedRoundRobinTaskScheduler is to enable hierarchical weighted round robin task scheduler\n\t// KeyName: history.enableHierarchicalWeightedRoundRobinTaskScheduler\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: N/A\n\tEnableHierarchicalWeightedRoundRobinTaskScheduler\n\n\t// EnableTaskListAwareTaskSchedulerByDomain is to enable task list aware task scheduler by domain\n\t// KeyName: history.enableTaskListAwareTaskSchedulerByDomain\n\t// Value type: Bool\n\t// Default value: false\n\t// Allowed filters: DomainName\n\tEnableTaskListAwareTaskSchedulerByDomain\n\n\t// LastBoolKey must be the last one in this const group\n\tLastBoolKey\n)\n\nconst (\n\tUnknownFloatKey FloatKey = iota\n\n\t// key for tests\n\tTestGetFloat64PropertyKey\n\tTestGetFloat64PropertyFilteredByShardIDKey\n\n\t// key for common & admin\n\n\t// PersistenceErrorInjectionRate is rate for injecting random error in persistence\n\t// KeyName: system.persistenceErrorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tPersistenceErrorInjectionRate\n\t// AdminErrorInjectionRate is the rate for injecting random error in admin client\n\t// KeyName: admin.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tAdminErrorInjectionRate\n\n\t// key for frontend\n\n\t// DomainFailoverRefreshTimerJitterCoefficient is the jitter for domain failover refresh timer jitter\n\t// KeyName: frontend.domainFailoverRefreshTimerJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.1\n\t// Allowed filters: N/A\n\tDomainFailoverRefreshTimerJitterCoefficient\n\t// FrontendErrorInjectionRate is rate for injecting random error in frontend client\n\t// KeyName: frontend.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tFrontendErrorInjectionRate\n\n\t// key for matching\n\n\t// MatchingErrorInjectionRate is rate for injecting random error in matching client\n\t// KeyName: matching.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMatchingErrorInjectionRate\n\n\t// key for history\n\n\t// QueueProcessorRandomSplitProbability is the probability for a domain to be split to a new processing queue\n\t// KeyName: history.queueProcessorRandomSplitProbability\n\t// Value type: Float64\n\t// Default value: 0.01\n\t// Allowed filters: N/A\n\tQueueProcessorRandomSplitProbability\n\t// QueueProcessorPollBackoffIntervalJitterCoefficient is backoff interval jitter coefficient\n\t// KeyName: history.queueProcessorPollBackoffIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tQueueProcessorPollBackoffIntervalJitterCoefficient\n\t// TimerProcessorUpdateAckIntervalJitterCoefficient is the update interval jitter coefficient\n\t// KeyName: history.timerProcessorUpdateAckIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTimerProcessorUpdateAckIntervalJitterCoefficient\n\t// TimerProcessorMaxPollIntervalJitterCoefficient is the max poll interval jitter coefficient\n\t// KeyName: history.timerProcessorMaxPollIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTimerProcessorMaxPollIntervalJitterCoefficient\n\t// TimerProcessorSplitQueueIntervalJitterCoefficient is the split processing queue interval jitter coefficient\n\t// KeyName: history.timerProcessorSplitQueueIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTimerProcessorSplitQueueIntervalJitterCoefficient\n\t// TransferProcessorMaxPollIntervalJitterCoefficient is the max poll interval jitter coefficient\n\t// KeyName: history.transferProcessorMaxPollIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTransferProcessorMaxPollIntervalJitterCoefficient\n\t// TransferProcessorSplitQueueIntervalJitterCoefficient is the split processing queue interval jitter coefficient\n\t// KeyName: history.transferProcessorSplitQueueIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTransferProcessorSplitQueueIntervalJitterCoefficient\n\t// TransferProcessorUpdateAckIntervalJitterCoefficient is the update interval jitter coefficient\n\t// KeyName: history.transferProcessorUpdateAckIntervalJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tTransferProcessorUpdateAckIntervalJitterCoefficient\n\t// ReplicationTaskProcessorCleanupJitterCoefficient is the jitter for cleanup timer\n\t// KeyName: history.ReplicationTaskProcessorCleanupJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorCleanupJitterCoefficient\n\t// ReplicationTaskProcessorStartWaitJitterCoefficient is the jitter for batch start wait timer\n\t// KeyName: history.ReplicationTaskProcessorStartWaitJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.9\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorStartWaitJitterCoefficient\n\t// ReplicationTaskProcessorHostQPS is the qps of task processing rate limiter on host level\n\t// KeyName: history.ReplicationTaskProcessorHostQPS\n\t// Value type: Float64\n\t// Default value: 1500\n\t// Allowed filters: N/A\n\tReplicationTaskProcessorHostQPS\n\t// ReplicationTaskProcessorShardQPS is the qps of task processing rate limiter on shard level\n\t// KeyName: history.ReplicationTaskProcessorShardQPS\n\t// Value type: Float64\n\t// Default value: 5\n\t// Allowed filters: N/A\n\tReplicationTaskProcessorShardQPS\n\t// ReplicationTaskGenerationQPS is the wait time between each replication task generation qps\n\t// KeyName: history.ReplicationTaskGenerationQPS\n\t// Value type: Float64\n\t// Default value: 100\n\t// Allowed filters: N/A\n\tReplicationTaskGenerationQPS\n\t// MutableStateChecksumInvalidateBefore is the epoch timestamp before which all checksums are to be discarded\n\t// KeyName: history.mutableStateChecksumInvalidateBefore\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMutableStateChecksumInvalidateBefore\n\t// NotifyFailoverMarkerTimerJitterCoefficient is the jitter for failover marker notifier timer\n\t// KeyName: history.NotifyFailoverMarkerTimerJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tNotifyFailoverMarkerTimerJitterCoefficient\n\t// HistoryErrorInjectionRate is rate for injecting random error in history client\n\t// KeyName: history.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tHistoryErrorInjectionRate\n\t// ReplicationBudgetManagerSoftCapThreshold is the soft cap threshold for the replication budget manager cache (0.0 to 1.0)\n\t// KeyName: history.replicationBudgetManagerSoftCapThreshold\n\t// Value type: Float64\n\t// Default value: 1.0\n\t// Allowed filters: N/A\n\tReplicationBudgetManagerSoftCapThreshold\n\t// ReplicationTaskFetcherTimerJitterCoefficient is the jitter for fetcher timer\n\t// KeyName: history.ReplicationTaskFetcherTimerJitterCoefficient\n\t// Value type: Float64\n\t// Default value: 0.15\n\t// Allowed filters: N/A\n\tReplicationTaskFetcherTimerJitterCoefficient\n\t// WorkerDeterministicConstructionCheckProbability controls the probability of running a deterministic construction check for any given archival\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tWorkerDeterministicConstructionCheckProbability\n\t// WorkerBlobIntegrityCheckProbability controls the probability of running an integrity check for any given archival\n\t// KeyName: N/A\n\t// Default value: N/A\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tWorkerBlobIntegrityCheckProbability\n\n\t// HistoryGlobalRatelimiterNewDataWeight defines how much weight to give each host's newest data, per update.  Must be between 0 and 1, higher values match new values more closely after a single update.\n\t// KeyName: history.globalRatelimiterNewDataWeight\n\t// Value type: Float64\n\t// Default value: 0.5\n\tHistoryGlobalRatelimiterNewDataWeight\n\n\tMatchingPartitionDownscaleFactor\n\n\t// MatchingOverrideTaskListRPS is the RPS override for a specific TaskList.\n\t// When set to a non-zero value, this overrides the RPS value that pollers specify.\n\t// KeyName: matching.overrideTaskListRps\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: DomainName, TaskListName, TaskType\n\tMatchingOverrideTaskListRPS\n\n\t// Key for shard distributor\n\n\t// ShardDistributorErrorInjectionRate is rate for injecting random error in shard distributor client\n\t// KeyName: sharddistributor.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tShardDistributorErrorInjectionRate\n\n\t// ShardDistributorExecutorErrorInjectionRate is rate for injecting random error in shard distributor executor client\n\t// KeyName: sharddistributorexecutor.errorInjectionRate\n\t// Value type: Float64\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tShardDistributorExecutorErrorInjectionRate\n\n\t// ShardDistributorLoadBalancingNaiveMaxDeviation is max deviation between the coldest and hottest executors\n\t// in naive load balancing mode\n\t//\n\t// KeyName: shardDistributor.loadBalancingNaive.maxDeviation\n\t// Value type: Float64\n\t// Default value: 2.0\n\t// Allowed filters: namespace\n\tShardDistributorLoadBalancingNaiveMaxDeviation\n\n\t// LastFloatKey must be the last one in this const group\n\tLastFloatKey\n)\n\nconst (\n\tUnknownStringKey StringKey = iota\n\n\t// key for tests\n\tTestGetStringPropertyKey\n\n\t// key for common & admin\n\n\t// WriteVisibilityStoreName is key for how to write to advanced visibility. Usually using with ReadVisibilityStoreName\n\t// KeyName: system.writeVisibilityStoreName\n\t// Value type: String enum: string [\"es\",\"os\",\"pinot\",\"db\"]\n\t// Default value: \"es\"\n\t// Allowed filters: N/A\n\tWriteVisibilityStoreName\n\t// ReadVisibilityStoreName is key to identify which store to read visibility data from\n\t// KeyName: system.readVisibilityStoreName\n\t// Value type: String enum: \"db\" or \"es\" or \"pinot\" or \"os\"\n\t// Default value: \"es\"\n\t// Allowed filters: DomainName\n\tReadVisibilityStoreName\n\t// HistoryArchivalStatus is key for the status of history archival to override the value from static config.\n\t// KeyName: system.historyArchivalStatus\n\t// Value type: string enum: \"enabled\" or \"disabled\"\n\t// Default value: \"enabled\"\n\t// Allowed filters: N/A\n\tHistoryArchivalStatus\n\t// VisibilityArchivalStatus is key for the status of visibility archival to override the value from static config.\n\t// KeyName: system.visibilityArchivalStatus\n\t// Value type: string enum: \"enabled\" or \"disabled\"\n\t// Default value: \"enabled\"\n\t// Allowed filters: N/A\n\tVisibilityArchivalStatus\n\t// DefaultEventEncoding is the encoding type for history events\n\t// KeyName: history.defaultEventEncoding\n\t// Value type: String\n\t// Default value: string(constants.EncodingTypeThriftRW)\n\t// Allowed filters: DomainName\n\tDefaultEventEncoding\n\t// AdminOperationToken is the token to pass admin checking\n\t// KeyName: history.adminOperationToken\n\t// Value type: String\n\t// Default value: common.DefaultAdminOperationToken\n\t// Allowed filters: N/A\n\tAdminOperationToken\n\t// ESAnalyzerLimitToTypes controls if we want to limit ESAnalyzer only to some workflow types\n\t// KeyName: worker.ESAnalyzerLimitToTypes\n\t// Value type: String\n\t// Default value: \"\" => means no limitation\n\tESAnalyzerLimitToTypes\n\t// ESAnalyzerLimitToDomains controls if we want to limit ESAnalyzer only to some domains\n\t// KeyName: worker.ESAnalyzerLimitToDomains\n\t// Value type: String\n\t// Default value: \"\" => means no limitation\n\tESAnalyzerLimitToDomains\n\t// ESAnalyzerWorkflowDurationWarnThresholds defines the warning execution thresholds for workflow types\n\t// KeyName: worker.ESAnalyzerWorkflowDurationWarnThresholds\n\t// Value type: string [{\"DomainName\":\"<domain>\", \"WorkflowType\":\"<workflowType>\", \"Threshold\":\"<duration>\", \"Refresh\":<shouldRefresh>, \"MaxNumWorkflows\":<maxNumber>}]\n\t// Default value: \"\"\n\tESAnalyzerWorkflowDurationWarnThresholds\n\t// ESAnalyzerWorkflowVersionMetricDomains defines the domains we want to emit wf version metrics on\n\t// KeyName: worker.ESAnalyzerWorkflowVersionMetricDomains\n\t// Value type: string [\"test-domain\",\"test-domain2\"]\n\t// Default value: \"\"\n\tESAnalyzerWorkflowVersionMetricDomains\n\t// ESAnalyzerWorkflowTypeMetricDomains defines the domains we want to emit wf type metrics on\n\t// KeyName: worker.ESAnalyzerWorkflowTypeMetricDomains\n\t// Value type: string [\"test-domain\",\"test-domain2\"]\n\t// Default value: \"\"\n\tESAnalyzerWorkflowTypeMetricDomains\n\n\t// FrontendGlobalRatelimiterMode controls what keys use global vs fallback behavior,\n\t// and whether shadowing is enabled.  This is only available for frontend usage for now.\n\t//\n\t//   - \"disabled\" stops usage-tracking and all Update requests, in an attempt to be as close to \"do not use at all\" as possible.\n\t//   - \"local\" uses the new limiters with call tracking and metrics, but forces local-only behavior and does not submit usage data to aggregators.\n\t//   - \"global\" uses the new global-load-balanced logic (though it may decide to use a local-fallback internally, and this is not prevented)\n\t//   - \"x-shadow-y\" means \"use x, and shadow all calls to y but ignore the result\".\n\t//     this calls both, tracks and emits both metrics, and can be used to \"warm\" either limiter's in-memory state before switching.\n\t//\n\t// These values can be seen as constants of github.com/uber/cadence/common/quotas/global/collection.keyMode\n\t//\n\t// KeyName: frontend.globalRatelimiterMode\n\t// Value type: string enum: \"disabled\", \"local\", \"global\", \"local-shadow-global\", or \"global-shadow-local\"\n\t// Default value: \"disabled\"\n\t// Allowed filters: RatelimitKey (on global key, e.g. prefixed by collection name)\n\tFrontendGlobalRatelimiterMode\n\t// EnableAuthorizationV2 is the key to enable authorization v2 for a domain, only for extension binary:\n\t// KeyName: system.enableAuthorizationV2\n\t// Value type: string [\"disabled\",\"shadow\",\"enabled\"]\n\t// Default value: \"disabled\"\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableAuthorizationV2\n\tTasklistLoadBalancerStrategy\n\n\t// EnableAdminAuthorization is the key to enable authorization for admin operations, only for extension binary:\n\t// KeyName: system.enableAdminAuthorization\n\t// Value type: string [\"disabled\",\"shadow\",\"enabled\"]\n\t// Default value: \"disabled\"\n\t// TODO: https://github.com/uber/cadence/issues/3861\n\t// Note: not currently used in open-source\n\tEnableAdminAuthorization\n\n\t// MatchingShardDistributionMode is the mode of shard distribution for matching, we currently have four modes, we _highly_\n\t// recommend using hash_ring while the shard distributor is still in development.\n\t//\n\t// - \"hash_ring\" means that the shards are distributed using a consistent hash ring, in particular using the ringpop library\n\t// - \"shard_distributor\" means that the shards are distributed using the _highly experimental_ shard distributor service\n\t// - \"hash_ring-shadow-shard_distributor\" means that the shards are distrubuted using the hash ring, but shadowed by the shard distributor\n\t// - \"shard_distributor-shadow-hash_ring\" means that the shards are distrubuted using the shard distributor, but shadowed by the hash ring\n\t//\n\t// KeyName: matching.shardDistributionMode\n\t// Value type: string enum: \"hash_ring\" or \"shard_distributor\"\n\t// Default value: \"hash_ring\"\n\tMatchingShardDistributionMode\n\n\t// SerializationEncoding is the encoding type for blobs\n\t// KeyName: history.serializationEncoding\n\t// Value type: String\n\t// Default value: \"thriftrw\"\n\tSerializationEncoding\n\n\t// ShardDistributorMigrationMode is the mode the at represent the state of the migration to rely on shard distributor for the sharding mechanism\n\t//\n\t// \"invalid\" invalid mode for the migration, not expected to be used\n\t// \"local_pass\" the executor library is integrated but no external call to the SD happening\n\t// \"local_pass_shadow\" heartbeat calls to the SD to update the sharding state in SD\n\t// \"distributed_pass\" the local sharding mechanism is sent to SD, returned by SD and applied in the onboarded service\n\t// \"onboarded\" the sharding logic in SD is used\n\t//\n\t// KeyName: shardDistributor.migrationMode\n\t// Value type: String\n\t// Default value: onboarded\n\t// Allowed filters: namespace\n\tShardDistributorMigrationMode\n\n\t// ShardDistributorLoadBalancingMode is the load balancing mode for the shard distributor\n\t// Depending on the mode, the shard distributor will use different ways to distribute the shards\n\t//\n\t// * \"naive\" \t- mode assigns shards to the least loaded hosts without considering the existing shard distribution\n\t// * \"greedy\" \t- mode balances the load across all hosts while minimizing shard movements and uses shard statistics to make better decisions\n\t//\n\t// KeyName: shardDistributor.loadBalancingMode\n\t// Value type: String\n\t// Default value: \"naive\"\n\t// Allowed filters: namespace\n\tShardDistributorLoadBalancingMode\n\n\t// LastStringKey must be the last one in this const group\n\tLastStringKey\n)\n\nconst (\n\tUnknownDurationKey DurationKey = iota\n\n\t// key for tests\n\tTestGetDurationPropertyKey\n\tTestGetDurationPropertyFilteredByDomainKey\n\tTestGetDurationPropertyFilteredByTaskListInfoKey\n\tTestGetDurationPropertyFilteredByWorkflowTypeKey\n\tTestGetDurationPropertyFilteredByDomainIDKey\n\tTestGetDurationPropertyFilteredByShardID\n\n\t// FrontendShutdownDrainDuration is the duration of traffic drain during shutdown\n\t// KeyName: frontend.shutdownDrainDuration\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tFrontendShutdownDrainDuration\n\t// FrontendWarmupDuration is the duration before a frontend host reports its status as healthy\n\t// KeyName: frontend.warmupDuration\n\t// Value type: Duration\n\t// Default value: 30s\n\t// Allowed filters: N/A\n\tFrontendWarmupDuration\n\t// FrontendFailoverCoolDown is duration between two domain failvoers\n\t// KeyName: frontend.failoverCoolDown\n\t// Value type: Duration\n\t// Default value: 1m (one minute, see domain.FailoverCoolDown)\n\t// Allowed filters: DomainName\n\tFrontendFailoverCoolDown\n\t// DomainFailoverRefreshInterval is the domain failover refresh timer\n\t// KeyName: frontend.domainFailoverRefreshInterval\n\t// Value type: Duration\n\t// Default value: 10s (10*time.Second)\n\t// Allowed filters: N/A\n\tDomainFailoverRefreshInterval\n\t// GlobalRatelimiterUpdateInterval controls how frequently ratelimiter usage information is submitted to aggregators.\n\t// This value is shared between limiting and aggregating hosts (frontend and history).\n\t// KeyName: frontend.globalRatelimiterUpdateInterval\n\t// Value type: Duration\n\t// Default value: 3 seconds\n\tGlobalRatelimiterUpdateInterval\n\t// FrontendMaxWorkerPollDelay is the maximum duration a worker poll request (PollForActivityTask, PollForDecisionTask)\n\t// will wait for a rate limit token before being rejected. This setting doesn't completely control how\n\t// long a request can take - the request will complete after the minimum of the configured request timeout\n\t// KeyName: frontend.maxWorkerPollDelay\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: DomainName\n\tFrontendMaxWorkerPollDelay\n\n\t// MatchingLongPollExpirationInterval is the long poll expiration interval in the matching service\n\t// KeyName: matching.longPollExpirationInterval\n\t// Value type: Duration\n\t// Default value: time.Minute\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingLongPollExpirationInterval\n\t// MatchingUpdateAckInterval is the interval for update ack\n\t// KeyName: matching.updateAckInterval\n\t// Value type: Duration\n\t// Default value: 1m (1*time.Minute)\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingUpdateAckInterval\n\t// MatchingIdleTasklistCheckInterval is the IdleTasklistCheckInterval\n\t// KeyName: matching.idleTasklistCheckInterval\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIdleTasklistCheckInterval\n\t// MaxTasklistIdleTime is the max time tasklist being idle\n\t// KeyName: matching.maxTasklistIdleTime\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMaxTasklistIdleTime\n\t// MatchingShutdownDrainDuration is the duration of traffic drain during shutdown\n\t// KeyName: matching.shutdownDrainDuration\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tMatchingShutdownDrainDuration\n\t// MatchingActivityTaskSyncMatchWaitTime is the amount of time activity task will wait to be sync matched\n\t// KeyName: matching.activityTaskSyncMatchWaitTime\n\t// Value type: Duration\n\t// Default value: 100ms\n\t// Allowed filters: DomainName\n\tMatchingActivityTaskSyncMatchWaitTime\n\n\t// MatchingPartitionUpscaleSustainedDuration is the sustained period to wait before upscaling the number of partitions\n\t// KeyName: matching.partitionUpscaleSustainedDuration\n\t// Value type: Duration\n\t// Default value: 1m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingPartitionUpscaleSustainedDuration\n\t// MatchingPartitionDownscaleSustainedDuration is the sustained period to wait before downscaling the number of partitions\n\t// KeyName: matching.partitionDownscaleSustainedDuration\n\t// Value type: Duration\n\t// Default value: 2m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingPartitionDownscaleSustainedDuration\n\t// MatchingAdaptiveScalerUpdateInterval is the internal for adaptive scaler to update\n\t// KeyName: matching.adaptiveScalerUpdateInterval\n\t// Value type: Duration\n\t// Default value: 15s\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingAdaptiveScalerUpdateInterval\n\t// MatchingQPSTrackerInterval is the interval for qps tracker's loop. Changes are not reflected until service restart\n\t// KeyName: matching.qpsTrackerInterval\n\t// Value type: Duration\n\t// Default value: 10s\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingQPSTrackerInterval\n\n\t// MatchingIsolationGroupUpscaleSustainedDuration is the sustained period to wait before upscaling the number of partitions an isolation group is assigned to\n\t// KeyName: matching.isolationGroupUpscaleSustainedDuration\n\t// Value type: Duration\n\t// Default value: 1m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIsolationGroupUpscaleSustainedDuration\n\t// MatchingIsolationGroupDownscaleSustainedDuration is the sustained period to wait before downscaling the number of partitions an isolation group is assigned to\n\t// KeyName: matching.isolationGroupDownscaleSustainedDuration\n\t// Value type: Duration\n\t// Default value: 2m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIsolationGroupDownscaleSustainedDuration\n\t// MatchingIsolationGroupHasPollersSustainedDuration is the sustained period to wait before considering an isolation group as an active and assigning partitions to it\n\t// KeyName: matching.isolationGroupHasPollersSustainedDuration\n\t// Value type: Duration\n\t// Default value: 1m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIsolationGroupHasPollersSustainedDuration\n\t// MatchingIsolationGroupNoPollersSustainedDuration is the sustained period to wait before considering an isolation group as inactive and unassigning all partitions from it\n\t// KeyName: matching.isolationGroupNoPollersSustainedDuration\n\t// Value type: Duration\n\t// Default value: 1m\n\t// Allowed filters: DomainName,TasklistName,TaskType\n\tMatchingIsolationGroupNoPollersSustainedDuration\n\n\t// HistoryLongPollExpirationInterval is the long poll expiration interval in the history service\n\t// KeyName: history.longPollExpirationInterval\n\t// Value type: Duration\n\t// Default value: 20s( time.Second*20)\n\t// Allowed filters: DomainName\n\tHistoryLongPollExpirationInterval\n\t// HistoryCacheTTL is TTL of history cache\n\t// KeyName: history.cacheTTL\n\t// Value type: Duration\n\t// Default value: 1h (time.Hour)\n\t// Allowed filters: N/A\n\tHistoryCacheTTL\n\t// HistoryShutdownDrainDuration is the duration of traffic drain during shutdown\n\t// KeyName: history.shutdownDrainDuration\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tHistoryShutdownDrainDuration\n\t// EventsCacheTTL is TTL of events cache\n\t// KeyName: history.eventsCacheTTL\n\t// Value type: Duration\n\t// Default value: 1h (time.Hour)\n\t// Allowed filters: N/A\n\tEventsCacheTTL\n\t// AcquireShardInterval is interval that timer used to acquire shard\n\t// KeyName: history.acquireShardInterval\n\t// Value type: Duration\n\t// Default value: 1m (time.Minute)\n\t// Allowed filters: N/A\n\tAcquireShardInterval\n\t// StandbyClusterDelay is the artificial delay added to standby cluster's view of active cluster's time\n\t// KeyName: history.standbyClusterDelay\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: N/A\n\tStandbyClusterDelay\n\t// StandbyTaskMissingEventsResendDelay is the amount of time standby cluster's will wait (if events are missing)before calling remote for missing events\n\t// KeyName: history.standbyTaskMissingEventsResendDelay\n\t// Value type: Duration\n\t// Default value: 15m (15*time.Minute)\n\t// Allowed filters: N/A\n\tStandbyTaskMissingEventsResendDelay\n\t// StandbyTaskMissingEventsDiscardDelay is the amount of time standby cluster's will wait (if events are missing)before discarding the task\n\t// KeyName: history.standbyTaskMissingEventsDiscardDelay\n\t// Value type: Duration\n\t// Default value: 25m (25*time.Minute)\n\t// Allowed filters: N/A\n\tStandbyTaskMissingEventsDiscardDelay\n\t// ActiveTaskRedispatchInterval is the active task redispatch interval\n\t// KeyName: history.activeTaskRedispatchInterval\n\t// Value type: Duration\n\t// Default value: 5s (5*time.Second)\n\t// Allowed filters: N/A\n\tActiveTaskRedispatchInterval\n\t// StandbyTaskRedispatchInterval is the standby task redispatch interval\n\t// KeyName: history.standbyTaskRedispatchInterval\n\t// Value type: Duration\n\t// Default value: 30s (30*time.Second)\n\t// Allowed filters: N/A\n\tStandbyTaskRedispatchInterval\n\t// StandbyTaskReReplicationContextTimeout is the context timeout for standby task re-replication\n\t// KeyName: history.standbyTaskReReplicationContextTimeout\n\t// Value type: Duration\n\t// Default value: 3m (3*time.Minute)\n\t// Allowed filters: DomainID\n\tStandbyTaskReReplicationContextTimeout\n\t// ResurrectionCheckMinDelay is the minimal timer processing delay before scanning history to see\n\t// if there's a resurrected timer/activity\n\t// KeyName: history.resurrectionCheckMinDelay\n\t// Value type: Duration\n\t// Default value: 24h (24*time.Hour)\n\t// Allowed filters: DomainName\n\tResurrectionCheckMinDelay\n\t// QueueProcessorSplitLookAheadDurationByDomainID is the look ahead duration when spliting a domain to a new processing queue\n\t// KeyName: history.queueProcessorSplitLookAheadDurationByDomainID\n\t// Value type: Duration\n\t// Default value: 20m (20*time.Minute)\n\t// Allowed filters: DomainID\n\tQueueProcessorSplitLookAheadDurationByDomainID\n\t// QueueProcessorPollBackoffInterval is the backoff duration when queue processor is throttled\n\t// KeyName: history.queueProcessorPollBackoffInterval\n\t// Value type: Duration\n\t// Default value: 5s (5*time.Second)\n\t// Allowed filters: N/A\n\tQueueProcessorPollBackoffInterval\n\t// VirtualSliceForceAppendInterval is the duration forcing a new virtual slice to be appended to the root virtual queue instead of being merged. It has 2 benefits: First, virtual slices won't grow infinitely, task loading for that slice can complete and its scope can be shrinked. Second, when we need to unload a virtual slice to free memory, we won't unload too many tasks.\n\t// KeyName: history.virtualSliceForceAppendInterval\n\t// Value type: Duration\n\t// Default value: 5m\n\t// Allowed filters: N/A\n\tVirtualSliceForceAppendInterval\n\t// TimerProcessorUpdateAckInterval is update interval for timer processor\n\t// KeyName: history.timerProcessorUpdateAckInterval\n\t// Value type: Duration\n\t// Default value: 30s (30*time.Second)\n\t// Allowed filters: N/A\n\tTimerProcessorUpdateAckInterval\n\t// TimerProcessorCompleteTimerInterval is complete timer interval for timer processor\n\t// KeyName: history.timerProcessorCompleteTimerInterval\n\t// Value type: Duration\n\t// Default value: 60s (60*time.Second)\n\t// Allowed filters: N/A\n\tTimerProcessorCompleteTimerInterval\n\t// TimerProcessorFailoverMaxStartJitterInterval is the max jitter interval for starting timer\n\t// failover queue processing. The actual jitter interval used will be a random duration between\n\t// 0 and the max interval so that timer failover queue across different shards won't start at\n\t// the same time\n\t// KeyName: history.timerProcessorFailoverMaxStartJitterInterval\n\t// Value type: Duration\n\t// Default value: 0s (0*time.Second)\n\t// Allowed filters: N/A\n\tTimerProcessorFailoverMaxStartJitterInterval\n\t// TimerProcessorMaxPollInterval is max poll interval for timer processor\n\t// KeyName: history.timerProcessorMaxPollInterval\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: N/A\n\tTimerProcessorMaxPollInterval\n\t// TimerProcessorSplitQueueInterval is the split processing queue interval for timer processor\n\t// KeyName: history.timerProcessorSplitQueueInterval\n\t// Value type: Duration\n\t// Default value: 1m (1*time.Minute)\n\t// Allowed filters: N/A\n\tTimerProcessorSplitQueueInterval\n\t// TimerProcessorArchivalTimeLimit is the upper time limit for inline history archival\n\t// KeyName: history.timerProcessorArchivalTimeLimit\n\t// Value type: Duration\n\t// Default value: 2s (2*time.Second)\n\t// Allowed filters: N/A\n\tTimerProcessorArchivalTimeLimit\n\t// TimerProcessorMaxTimeShift is the max shift timer processor can have\n\t// KeyName: history.timerProcessorMaxTimeShift\n\t// Value type: Duration\n\t// Default value: 1s (1*time.Second)\n\t// Allowed filters: N/A\n\tTimerProcessorMaxTimeShift\n\t// TransferProcessorFailoverMaxStartJitterInterval is the max jitter interval for starting transfer\n\t// failover queue processing. The actual jitter interval used will be a random duration between\n\t// 0 and the max interval so that timer failover queue across different shards won't start at\n\t// the same time\n\t// KeyName: history.transferProcessorFailoverMaxStartJitterInterval\n\t// Value type: Duration\n\t// Default value: 0s (0*time.Second)\n\t// Allowed filters: N/A\n\tTransferProcessorFailoverMaxStartJitterInterval\n\t// TransferProcessorMaxPollInterval is max poll interval for transferQueueProcessor\n\t// KeyName: history.transferProcessorMaxPollInterval\n\t// Value type: Duration\n\t// Default value: 1m (1*time.Minute)\n\t// Allowed filters: N/A\n\tTransferProcessorMaxPollInterval\n\t// TransferProcessorSplitQueueInterval is the split processing queue interval for transferQueueProcessor\n\t// KeyName: history.transferProcessorSplitQueueInterval\n\t// Value type: Duration\n\t// Default value: 1m (1*time.Minute)\n\t// Allowed filters: N/A\n\tTransferProcessorSplitQueueInterval\n\t// TransferProcessorUpdateAckInterval is update interval for transferQueueProcessor\n\t// KeyName: history.transferProcessorUpdateAckInterval\n\t// Value type: Duration\n\t// Default value: 30s (30*time.Second)\n\t// Allowed filters: N/A\n\tTransferProcessorUpdateAckInterval\n\t// TransferProcessorCompleteTransferInterval is complete timer interval for transferQueueProcessor\n\t// KeyName: history.transferProcessorCompleteTransferInterval\n\t// Value type: Duration\n\t// Default value: 60s (60*time.Second)\n\t// Allowed filters: N/A\n\tTransferProcessorCompleteTransferInterval\n\t// TransferProcessorValidationInterval is interval for performing transfer queue validation\n\t// KeyName: history.transferProcessorValidationInterval\n\t// Value type: Duration\n\t// Default value: 30s (30*time.Second)\n\t// Allowed filters: N/A\n\tTransferProcessorValidationInterval\n\t// TransferProcessorVisibilityArchivalTimeLimit is the upper time limit for archiving visibility records\n\t// KeyName: history.transferProcessorVisibilityArchivalTimeLimit\n\t// Value type: Duration\n\t// Default value: 400ms (400*time.Millisecond)\n\t// Allowed filters: N/A\n\tTransferProcessorVisibilityArchivalTimeLimit\n\t// ReplicatorUpperLatency indicates the max allowed replication latency between clusters\n\t// KeyName: history.replicatorUpperLatency\n\t// Value type: Duration\n\t// Default value: 40s (40 * time.Second)\n\t// Allowed filters: N/A\n\tReplicatorUpperLatency\n\t// ShardUpdateMinInterval is the minimal time interval which the shard info can be updated\n\t// KeyName: history.shardUpdateMinInterval\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: N/A\n\tShardUpdateMinInterval\n\t// ShardSyncMinInterval is the minimal time interval which the shard info should be sync to remote\n\t// KeyName: history.shardSyncMinInterval\n\t// Value type: Duration\n\t// Default value: 5m (5*time.Minute)\n\t// Allowed filters: N/A\n\tShardSyncMinInterval\n\t// StickyTTL is to expire a sticky tasklist if no update more than this duration\n\t// KeyName: history.stickyTTL\n\t// Value type: Duration\n\t// Default value: time.Hour*24*365\n\t// Allowed filters: DomainName\n\tStickyTTL\n\t// DecisionHeartbeatTimeout is for decision heartbeat\n\t// KeyName: history.decisionHeartbeatTimeout\n\t// Value type: Duration\n\t// Default value: 30m (time.Minute*30)\n\t// Allowed filters: DomainName\n\tDecisionHeartbeatTimeout\n\t// NormalDecisionScheduleToStartTimeout is scheduleToStart timeout duration for normal (non-sticky) decision task\n\t// KeyName: history.normalDecisionScheduleToStartTimeout\n\t// Value type: Duration\n\t// Default value: time.Minute*5\n\t// Allowed filters: DomainName\n\tNormalDecisionScheduleToStartTimeout\n\t// NotifyFailoverMarkerInterval is determines the frequency to notify failover marker\n\t// KeyName: history.NotifyFailoverMarkerInterval\n\t// Value type: Duration\n\t// Default value: 5s (5*time.Second)\n\t// Allowed filters: N/A\n\tNotifyFailoverMarkerInterval\n\t// ActivityMaxScheduleToStartTimeoutForRetry is maximum value allowed when overwritting the schedule to start timeout for activities with retry policy\n\t// KeyName: history.activityMaxScheduleToStartTimeoutForRetry\n\t// Value type: Duration\n\t// Default value: 30m (30*time.Minute)\n\t// Allowed filters: DomainName\n\tActivityMaxScheduleToStartTimeoutForRetry\n\t// ReplicationTaskFetcherAggregationInterval determines how frequently the fetch requests are sent\n\t// KeyName: history.ReplicationTaskFetcherAggregationInterval\n\t// Value type: Duration\n\t// Default value: 2s (2 * time.Second)\n\t// Allowed filters: N/A\n\tReplicationTaskFetcherAggregationInterval\n\t// ReplicationTaskFetcherErrorRetryWait is the wait time when fetcher encounters error\n\t// KeyName: history.ReplicationTaskFetcherErrorRetryWait\n\t// Value type: Duration\n\t// Default value: time.Second\n\t// Allowed filters: N/A\n\tReplicationTaskFetcherErrorRetryWait\n\t// ReplicationTaskFetcherServiceBusyWait is the wait time when fetcher encounters service busy error\n\t// KeyName: history.ReplicationTaskFetcherServiceBusyWait\n\t// Value type: Duration\n\t// Default value: 60s (60 * time.Second)\n\t// Allowed filters: N/A\n\tReplicationTaskFetcherServiceBusyWait\n\t// ReplicationTaskProcessorErrorRetryWait is the initial retry wait when we see errors in applying replication tasks\n\t// KeyName: history.ReplicationTaskProcessorErrorRetryWait\n\t// Value type: Duration\n\t// Default value: 50ms (50*time.Millisecond)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorErrorRetryWait\n\t// ReplicationTaskProcessorErrorSecondRetryWait is the initial retry wait for the second phase retry\n\t// KeyName: history.ReplicationTaskProcessorErrorSecondRetryWait\n\t// Value type: Duration\n\t// Default value: 5s (5* time.Second)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorErrorSecondRetryWait\n\t// ReplicationTaskProcessorErrorSecondRetryMaxWait is the max wait time for the second phase retry\n\t// KeyName: history.ReplicationTaskProcessorErrorSecondRetryMaxWait\n\t// Value type: Duration\n\t// Default value: 30s (30 * time.Second)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorErrorSecondRetryMaxWait\n\t// ReplicationTaskProcessorErrorSecondRetryExpiration is the expiration duration for the second phase retry\n\t// KeyName: history.ReplicationTaskProcessorErrorSecondRetryExpiration\n\t// Value type: Duration\n\t// Default value: 5m (5* time.Minute)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorErrorSecondRetryExpiration\n\t// ReplicationTaskProcessorNoTaskInitialWait is the wait time when not ask is returned\n\t// KeyName: history.ReplicationTaskProcessorNoTaskInitialWait\n\t// Value type: Duration\n\t// Default value: 2s (2* time.Second)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorNoTaskInitialWait\n\t// ReplicationTaskProcessorCleanupInterval determines how frequently the cleanup replication queue\n\t// KeyName: history.ReplicationTaskProcessorCleanupInterval\n\t// Value type: Duration\n\t// Default value: 1m (1* time.Minute)\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorCleanupInterval\n\t// ReplicationTaskProcessorStartWait is the wait time before each task processing batch\n\t// KeyName: history.ReplicationTaskProcessorStartWait\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: ShardID\n\tReplicationTaskProcessorStartWait\n\t// ReplicationTaskProcessorLatencyLogThreshold is the threshold of whether history will log history replication latency\n\t// KeyName: history.ReplicationTaskProcessorLatencyLogThreshold\n\t// Value type: Duration\n\t// Default value: 0\n\t// Allowed filters: N/A\n\tReplicationTaskProcessorLatencyLogThreshold\n\t// WorkerESProcessorFlushInterval is flush interval for esProcessor\n\t// KeyName: worker.ESProcessorFlushInterval\n\t// Value type: Duration\n\t// Default value: 1s (1*time.Second)\n\t// Allowed filters: N/A\n\tWorkerESProcessorFlushInterval\n\t// WorkerTimeLimitPerArchivalIteration is controls the time limit of each iteration of archival workflow\n\t// KeyName: worker.TimeLimitPerArchivalIteration\n\t// Value type: Duration\n\t// Default value: archiver.MaxArchivalIterationTimeout()\n\t// Allowed filters: N/A\n\tWorkerTimeLimitPerArchivalIteration\n\t// WorkerReplicationTaskMaxRetryDuration is the max retry duration for any task\n\t// KeyName: worker.replicationTaskMaxRetryDuration\n\t// Value type: Duration\n\t// Default value: 10m (time.Minute*10)\n\t// Allowed filters: N/A\n\tWorkerReplicationTaskMaxRetryDuration\n\t// ESAnalyzerTimeWindow defines the time window ElasticSearch Analyzer will consider while taking workflow averages\n\t// KeyName: worker.ESAnalyzerTimeWindow\n\t// Value type: Duration\n\t// Default value: 30 days\n\tESAnalyzerTimeWindow\n\t// ESAnalyzerBufferWaitTime controls min time required to consider a worklow stuck\n\t// KeyName: worker.ESAnalyzerBufferWaitTime\n\t// Value type: Duration\n\t// Default value: 30 minutes\n\tESAnalyzerBufferWaitTime\n\t// IsolationGroupStateRefreshInterval\n\t// KeyName: system.isolationGroupStateRefreshInterval\n\t// Value type: Duration\n\t// Default value: 30 seconds\n\tIsolationGroupStateRefreshInterval\n\t// IsolationGroupStateFetchTimeout is the dynamic config DB fetch timeout value\n\t// KeyName: system.isolationGroupStateFetchTimeout\n\t// Value type: Duration\n\t// Default value: 30 seconds\n\tIsolationGroupStateFetchTimeout\n\t// IsolationGroupStateUpdateTimeout is the dynamic config DB update timeout value\n\t// KeyName: system.isolationGroupStateUpdateTimeout\n\t// Value type: Duration\n\t// Default value: 30 seconds\n\tIsolationGroupStateUpdateTimeout\n\n\t// AsyncTaskDispatchTimeout is the timeout of dispatching tasks for async match\n\t// KeyName: matching.asyncTaskDispatchTimeout\n\t// Value type: Duration\n\t// Default value: 3 seconds\n\t// Allowed filters: domainName, taskListName, taskListType\n\tAsyncTaskDispatchTimeout\n\n\t// AppendTaskTimeout is the timeout of appending tasks to persistence.\n\t// KeyName: matching.appendTaskTimeout\n\t// Value type: Duration\n\t// Default value: 5 seconds\n\t// Allowed filters: domainName, taskListName, taskListType\n\tAppendTaskTimeout\n\n\t// HistoryGlobalRatelimiterDecayAfter defines how long to wait for an update before considering a host's data \"possibly gone\", causing its weight to gradually decline.\n\t// KeyName: history.globalRatelimiterDecayAfter\n\t// Value type: Duration\n\t// Default value: 6 seconds\n\tHistoryGlobalRatelimiterDecayAfter\n\t// HistoryGlobalRatelimiterGCAfter defines how long to wait until a host's data is considered entirely useless, e.g. host has likely disappeared, its weight is very low, and the data can be deleted.\n\t// KeyName: history.globalRatelimiterGCAfter\n\t// Value type: Duration\n\t// Default value: 30 seconds\n\tHistoryGlobalRatelimiterGCAfter\n\n\t// LocalPollWaitTime is the wait time for a poller to wait before considering request forwarding\n\t// KeyName: matching.localPollWaitTime\n\t// Value type: Duration\n\t// Default value: 10ms\n\t// Allowed filters: domainName, taskListName, taskListType\n\tLocalPollWaitTime\n\n\t// LocalTaskWaitTime is the wait time for a task to wait before considering task forwarding\n\t// KeyName: matching.localTaskWaitTime\n\t// Value type: Duration\n\t// Default value: 10ms\n\t// Allowed filters: domainName, taskListName, taskListType\n\tLocalTaskWaitTime\n\n\t// TaskIsolationDuration is the time period for which we attempt to respect tasklist isolation before allowing any poller to process the task\n\t// KeyName: matching.taskIsolationDuration\n\t// Value type: Duration\n\t// Default value: 200ms\n\t// Allowed filters: domainName, taskListName, taskListType\n\tTaskIsolationDuration\n\n\t// TaskIsolationPollerWindow is the time period for which pollers are remembered when deciding whether to skip tasklist isolation due to unpolled isolation groups.\n\t// KeyName: matching.taskIsolationPollerWindow\n\t// Value type: Duration\n\t// Default value: 2s\n\t// Allowed filters: domainName, taskListName, taskListType\n\tTaskIsolationPollerWindow\n\n\t// DomainAuditLogTTL is the TTL for domain audit log entries\n\t// KeyName: system.domainAuditLogTTL\n\t// Value type: Duration\n\t// Default value: 365 days (1 year)\n\t// Allowed filters: DomainID\n\tDomainAuditLogTTL\n\n\t// LastDurationKey must be the last one in this const group\n\tLastDurationKey\n)\n\nconst (\n\tUnknownMapKey MapKey = iota\n\n\t// key for tests\n\tTestGetMapPropertyKey\n\n\t// key for common & admin\n\n\t// RequiredDomainDataKeys is the key for the list of data keys required in domain registration\n\t// KeyName: system.requiredDomainDataKeys\n\t// Value type: Map\n\t// Default value: nil\n\t// Allowed filters: N/A\n\tRequiredDomainDataKeys\n\n\t// key for frontend\n\n\t// ValidSearchAttributes is legal indexed keys that can be used in list APIs. When overriding, ensure to include the existing default attributes of the current release\n\t// KeyName: frontend.validSearchAttributes\n\t// Value type: Map\n\t// Default value: the default attributes of this release version, see definition.GetDefaultIndexedKeys()\n\t// Allowed filters: N/A\n\tValidSearchAttributes\n\n\t// key for history\n\n\t// TaskSchedulerRoundRobinWeights is the priority weight for weighted round robin task scheduler\n\t// KeyName: history.taskSchedulerRoundRobinWeight\n\t// Value type: Map\n\t// Default value: please see common.ConvertIntMapToDynamicConfigMapProperty(DefaultTaskPriorityWeight) in code base\n\t// Allowed filters: N/A\n\tTaskSchedulerRoundRobinWeights\n\t// TaskSchedulerDomainRoundRobinWeights is the priority round robin weights for domains\n\t// KeyName: history.taskSchedulerDomainRoundRobinWeight\n\t// Value type: Map\n\t// Default value: see common.ConvertIntMapToDynamicConfigMapProperty(DefaultTaskSchedulerRoundRobinWeights) in code base\n\t// Allowed filters: DomainName\n\tTaskSchedulerDomainRoundRobinWeights\n\t// QueueProcessorPendingTaskSplitThreshold is the threshold for the number of pending tasks per domain\n\t// KeyName: history.queueProcessorPendingTaskSplitThreshold\n\t// Value type: Map\n\t// Default value: see common.ConvertIntMapToDynamicConfigMapProperty(DefaultPendingTaskSplitThreshold) in code base\n\t// Allowed filters: N/A\n\tQueueProcessorPendingTaskSplitThreshold\n\t// QueueProcessorStuckTaskSplitThreshold is the threshold for the number of attempts of a task\n\t// KeyName: history.queueProcessorStuckTaskSplitThreshold\n\t// Value type: Map\n\t// Default value: see common.ConvertIntMapToDynamicConfigMapProperty(DefaultStuckTaskSplitThreshold) in code base\n\t// Allowed filters: N/A\n\tQueueProcessorStuckTaskSplitThreshold\n\n\t// PinotOptimizedQueryColumns is the list of search attributes that can be used in pinot optimized query\n\t// KeyName: frontend.pinotOptimizedQueryColumns\n\t// Value type: Map\n\t// Default value: empty map\n\t// Allowed filters: N/A\n\tPinotOptimizedQueryColumns\n\n\t// SearchAttributesHiddenValueKeys is the list of search attributes that values should be hidden\n\t// KeyName: frontend.searchAttributesHiddenValueKeys\n\t// Value type: Map\n\t// Default value: empty map\n\t// Allowed filters: N/A\n\tSearchAttributesHiddenValueKeys\n\n\t// LastMapKey must be the last one in this const group\n\tLastMapKey\n)\n\nconst (\n\tUnknownListKey ListKey = iota\n\tTestGetListPropertyKey\n\n\t// AllIsolationGroups is the list of all possible isolation groups in a service\n\t// KeyName: system.allIsolationGroups\n\t// Value type: []string\n\t// Default value: N/A\n\t// Allowed filters: N/A\n\tAllIsolationGroups\n\n\t// RateLimiterBypassCallerTypes defines which caller types bypass rate limiters (both frontend and persistence)\n\t// KeyName: system.rateLimiterBypassCallerTypes\n\t// Value type: []string\n\t// Default value: empty list\n\t// Allowed filters: N/A\n\tRateLimiterBypassCallerTypes\n\n\t// HeaderForwardingRules defines which headers are forwarded from inbound calls to outbound.\n\t// This value is only loaded at startup.\n\t//\n\t// Regexes and header names are used as-is, you are strongly encouraged to use `(?i)` to make your regex case-insensitive.\n\t//\n\t// KeyName: admin.HeaderForwardingRules\n\t// Value type: []rpc.HeaderRule or an []interface{} containing `map[string]interface{}{\"Add\":bool,\"Match\":string}` values.\n\t// Default value: forward all headers.  (this is a problematic value, and it will be changing as we reduce to a list of known values)\n\tHeaderForwardingRules\n\n\tLastListKey\n)\n\n// DefaultIsolationGroupConfigStoreManagerGlobalMapping is the dynamic config value for isolation groups\n// Note: This is not typically used for normal dynamic config (type 0), but instead\n// it's used only for IsolationGroup config (type 1).\n// KeyName: system.defaultIsolationGroupConfigStoreManagerGlobalMapping\nconst DefaultIsolationGroupConfigStoreManagerGlobalMapping ListKey = -1 // This is a hack to put it in a different list due to it being a different config type\n\nvar IntKeys = map[IntKey]DynamicInt{\n\tTestGetIntPropertyKey: {\n\t\tKeyName:      \"testGetIntPropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetIntPropertyFilteredByDomainKey: {\n\t\tKeyName:      \"testGetIntPropertyFilteredByDomainKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetIntPropertyFilteredByTaskListInfoKey: {\n\t\tKeyName:      \"testGetIntPropertyFilteredByTaskListInfoKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetIntPropertyFilteredByWorkflowTypeKey: {\n\t\tKeyName:      \"testGetIntPropertyFilteredByWorkflowTypeKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetIntPropertyFilteredByShardIDKey: {\n\t\tKeyName:      \"testGetIntPropertyFilteredByShardIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t\tFilters:      nil,\n\t},\n\tTransactionSizeLimit: {\n\t\tKeyName:      \"system.transactionSizeLimit\",\n\t\tDescription:  \"TransactionSizeLimit is the largest allowed transaction size to persistence\",\n\t\tDefaultValue: 14680064,\n\t},\n\tMaxRetentionDays: {\n\t\tKeyName:      \"system.maxRetentionDays\",\n\t\tDescription:  \"MaxRetentionDays is the maximum allowed retention days for domain\",\n\t\tDefaultValue: 30,\n\t},\n\tMinRetentionDays: {\n\t\tKeyName:      \"system.minRetentionDays\",\n\t\tDescription:  \"MinRetentionDays is the minimal allowed retention days for domain\",\n\t\tDefaultValue: 1,\n\t},\n\tMaxDecisionStartToCloseSeconds: {\n\t\tKeyName:      \"system.maxDecisionStartToCloseSeconds\",\n\t\tDescription:  \"MaxDecisionStartToCloseSeconds is the maximum allowed value for decision start to close timeout in seconds\",\n\t\tDefaultValue: 240,\n\t},\n\tBlobSizeLimitError: {\n\t\tKeyName:      \"limit.blobSize.error\",\n\t\tDescription:  \"BlobSizeLimitError is the per event blob size limit\",\n\t\tDefaultValue: 2 * 1024 * 1024,\n\t},\n\tBlobSizeLimitWarn: {\n\t\tKeyName:      \"limit.blobSize.warn\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"BlobSizeLimitWarn is the per event blob size limit for warning\",\n\t\tDefaultValue: 256 * 1024,\n\t},\n\tHistorySizeLimitError: {\n\t\tKeyName:      \"limit.historySize.error\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistorySizeLimitError is the per workflow execution history size limit\",\n\t\tDefaultValue: 200 * 1024 * 1024,\n\t},\n\tHistorySizeLimitWarn: {\n\t\tKeyName:      \"limit.historySize.warn\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistorySizeLimitWarn is the per workflow execution history size limit for warning\",\n\t\tDefaultValue: 50 * 1024 * 1024,\n\t},\n\tHistoryCountLimitError: {\n\t\tKeyName:      \"limit.historyCount.error\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryCountLimitError is the per workflow execution history event count limit\",\n\t\tDefaultValue: 200 * 1024,\n\t},\n\tHistoryCountLimitWarn: {\n\t\tKeyName:      \"limit.historyCount.warn\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryCountLimitWarn is the per workflow execution history event count limit for warning\",\n\t\tDefaultValue: 50 * 1024,\n\t},\n\tPendingActivitiesCountLimitError: {\n\t\tKeyName:      \"limit.pendingActivityCount.error\",\n\t\tDescription:  \"PendingActivitiesCountLimitError is the limit of how many pending activities a workflow can have at a point in time\",\n\t\tDefaultValue: 1024,\n\t},\n\tPendingActivitiesCountLimitWarn: {\n\t\tKeyName:      \"limit.pendingActivityCount.warn\",\n\t\tDescription:  \"PendingActivitiesCountLimitWarn is the limit of how many activities a workflow can have before a warning is logged\",\n\t\tDefaultValue: 512,\n\t},\n\tDomainNameMaxLength: {\n\t\tKeyName:      \"limit.domainNameLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DomainNameMaxLength is the length limit for domain name\",\n\t\tDefaultValue: 1000,\n\t},\n\tIdentityMaxLength: {\n\t\tKeyName:      \"limit.identityLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"IdentityMaxLength is the length limit for identity\",\n\t\tDefaultValue: 1000,\n\t},\n\tWorkflowIDMaxLength: {\n\t\tKeyName:      \"limit.workflowIDLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"WorkflowIDMaxLength is the length limit for workflowID\",\n\t\tDefaultValue: 1000,\n\t},\n\tSignalNameMaxLength: {\n\t\tKeyName:      \"limit.signalNameLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"SignalNameMaxLength is the length limit for signal name\",\n\t\tDefaultValue: 1000,\n\t},\n\tWorkflowTypeMaxLength: {\n\t\tKeyName:      \"limit.workflowTypeLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"WorkflowTypeMaxLength is the length limit for workflow type\",\n\t\tDefaultValue: 1000,\n\t},\n\tRequestIDMaxLength: {\n\t\tKeyName:      \"limit.requestIDLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"RequestIDMaxLength is the length limit for requestID\",\n\t\tDefaultValue: 1000,\n\t},\n\tTaskListNameMaxLength: {\n\t\tKeyName:      \"limit.taskListNameLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TaskListNameMaxLength is the length limit for task list name\",\n\t\tDefaultValue: 1000,\n\t},\n\tActivityIDMaxLength: {\n\t\tKeyName:      \"limit.activityIDLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ActivityIDMaxLength is the length limit for activityID\",\n\t\tDefaultValue: 1000,\n\t},\n\tActivityTypeMaxLength: {\n\t\tKeyName:      \"limit.activityTypeLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ActivityTypeMaxLength is the length limit for activity type\",\n\t\tDefaultValue: 1000,\n\t},\n\tMarkerNameMaxLength: {\n\t\tKeyName:      \"limit.markerNameLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"MarkerNameMaxLength is the length limit for marker name\",\n\t\tDefaultValue: 1000,\n\t},\n\tTimerIDMaxLength: {\n\t\tKeyName:      \"limit.timerIDLength\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TimerIDMaxLength is the length limit for timerID\",\n\t\tDefaultValue: 1000,\n\t},\n\tMaxIDLengthWarnLimit: {\n\t\tKeyName:      \"limit.maxIDWarnLength\",\n\t\tDescription:  \"MaxIDLengthWarnLimit is the warn length limit for various IDs, including: Domain, TaskList, WorkflowID, ActivityID, TimerID, WorkflowType, ActivityType, SignalName, MarkerName, ErrorReason/FailureReason/CancelCause, Identity, RequestID\",\n\t\tDefaultValue: 128,\n\t},\n\tFrontendPersistenceMaxQPS: {\n\t\tKeyName:      \"frontend.persistenceMaxQPS\",\n\t\tDescription:  \"FrontendPersistenceMaxQPS is the max qps frontend host can query DB\",\n\t\tDefaultValue: 2000,\n\t},\n\tFrontendPersistenceGlobalMaxQPS: {\n\t\tKeyName:      \"frontend.persistenceGlobalMaxQPS\",\n\t\tDescription:  \"FrontendPersistenceGlobalMaxQPS is the max qps frontend cluster can query DB\",\n\t\tDefaultValue: 0,\n\t},\n\tFrontendVisibilityMaxPageSize: {\n\t\tKeyName:      \"frontend.visibilityMaxPageSize\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendVisibilityMaxPageSize is default max size for ListWorkflowExecutions in one page\",\n\t\tDefaultValue: 1000,\n\t},\n\tFrontendVisibilityListMaxQPS: {\n\t\tKeyName: \"frontend.visibilityListMaxQPS\",\n\t\tDescription: \"deprecated: never used for ratelimiting, only sampling-based failure injection, and only on database-based visibility.\\n\" +\n\t\t\t\"FrontendVisibilityListMaxQPS is max qps frontend can list open/close workflows\",\n\t\tDefaultValue: 10,\n\t},\n\tFrontendESVisibilityListMaxQPS: {\n\t\tKeyName: \"frontend.esVisibilityListMaxQPS\",\n\t\tDescription: \"deprecated: never read from, all ES reads and writes erroneously use PersistenceMaxQPS.\\n\" +\n\t\t\t\"FrontendESVisibilityListMaxQPS is max qps frontend can list open/close workflows from ElasticSearch\",\n\t\tDefaultValue: 30,\n\t},\n\tFrontendESIndexMaxResultWindow: {\n\t\tKeyName:      \"frontend.esIndexMaxResultWindow\",\n\t\tDescription:  \"FrontendESIndexMaxResultWindow is ElasticSearch index setting max_result_window\",\n\t\tDefaultValue: 10000,\n\t},\n\tFrontendHistoryMaxPageSize: {\n\t\tKeyName:      \"frontend.historyMaxPageSize\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendHistoryMaxPageSize is default max size for GetWorkflowExecutionHistory in one page\",\n\t\tDefaultValue: 1000,\n\t},\n\tFrontendUserRPS: {\n\t\tKeyName:      \"frontend.rps\",\n\t\tDescription:  \"FrontendUserRPS is workflow rate limit per second\",\n\t\tDefaultValue: 1200,\n\t},\n\tFrontendWorkerRPS: {\n\t\tKeyName:      \"frontend.workerrps\",\n\t\tDescription:  \"FrontendWorkerRPS is background-processing workflow rate limit per second\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendVisibilityRPS: {\n\t\tKeyName:      \"frontend.visibilityrps\",\n\t\tDescription:  \"FrontendVisibilityRPS is the global workflow List*WorkflowExecutions request rate limit per second\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendAsyncRPS: {\n\t\tKeyName:      \"frontend.asyncrps\",\n\t\tDescription:  \"FrontendAsyncRPS is the async workflow request rate limit per second\",\n\t\tDefaultValue: 10000,\n\t},\n\tFrontendMaxDomainUserRPSPerInstance: {\n\t\tKeyName:      \"frontend.domainrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxDomainUserRPSPerInstance is workflow domain rate limit per second\",\n\t\tDefaultValue: 1200,\n\t},\n\tFrontendMaxDomainWorkerRPSPerInstance: {\n\t\tKeyName:      \"frontend.domainworkerrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxDomainWorkerRPSPerInstance is background-processing workflow domain rate limit per second\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendMaxDomainVisibilityRPSPerInstance: {\n\t\tKeyName:      \"frontend.domainvisibilityrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxDomainVisibilityRPSPerInstance is the per-instance List*WorkflowExecutions request rate limit per second\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendMaxDomainAsyncRPSPerInstance: {\n\t\tKeyName:      \"frontend.domainasyncrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxDomainAsyncRPSPerInstance is the per-instance async workflow request rate limit per second\",\n\t\tDefaultValue: 10000,\n\t},\n\tFrontendGlobalDomainUserRPS: {\n\t\tKeyName:      \"frontend.globalDomainrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendGlobalDomainUserRPS is workflow domain rate limit per second for the whole Cadence cluster\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendGlobalDomainWorkerRPS: {\n\t\tKeyName:      \"frontend.globalDomainWorkerrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendGlobalDomainWorkerRPS is background-processing workflow domain rate limit per second for the whole Cadence cluster\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendGlobalDomainVisibilityRPS: {\n\t\tKeyName:      \"frontend.globalDomainVisibilityrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendGlobalDomainVisibilityRPS is the per-domain List*WorkflowExecutions request rate limit per second\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tFrontendGlobalDomainAsyncRPS: {\n\t\tKeyName:      \"frontend.globalDomainAsyncrps\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendGlobalDomainAsyncRPS is the per-domain async workflow request rate limit per second\",\n\t\tDefaultValue: 100000,\n\t},\n\tFrontendDecisionResultCountLimit: {\n\t\tKeyName:      \"frontend.decisionResultCountLimit\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendDecisionResultCountLimit is max number of decisions per RespondDecisionTaskCompleted request\",\n\t\tDefaultValue: 0,\n\t},\n\tFrontendThrottledLogRPS: {\n\t\tKeyName:      \"frontend.throttledLogRPS\",\n\t\tDescription:  \"FrontendThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\",\n\t\tDefaultValue: 20,\n\t},\n\tFrontendMaxBadBinaries: {\n\t\tKeyName:      \"frontend.maxBadBinaries\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxBadBinaries is the max number of bad binaries in domain config\",\n\t\tDefaultValue: 10,\n\t},\n\tSearchAttributesNumberOfKeysLimit: {\n\t\tKeyName:      \"frontend.searchAttributesNumberOfKeysLimit\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"SearchAttributesNumberOfKeysLimit is the limit of number of keys\",\n\t\tDefaultValue: 100,\n\t},\n\tSearchAttributesSizeOfValueLimit: {\n\t\tKeyName:      \"frontend.searchAttributesSizeOfValueLimit\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"SearchAttributesSizeOfValueLimit is the size limit of each value\",\n\t\tDefaultValue: 2048,\n\t},\n\tSearchAttributesTotalSizeLimit: {\n\t\tKeyName:      \"frontend.searchAttributesTotalSizeLimit\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"SearchAttributesTotalSizeLimit is the size limit of the whole map\",\n\t\tDefaultValue: 40 * 1024,\n\t},\n\tVisibilityArchivalQueryMaxPageSize: {\n\t\tKeyName:      \"frontend.visibilityArchivalQueryMaxPageSize\",\n\t\tDescription:  \"VisibilityArchivalQueryMaxPageSize is the maximum page size for a visibility archival query\",\n\t\tDefaultValue: 10000,\n\t},\n\tFrontendFailoverHistoryMaxSize: {\n\t\tKeyName:      \"frontend.failoverHistoryMaxSize\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendFailoverHistoryMaxSize is the maximum size for the number of failover event records in a domain failover history\",\n\t\tDefaultValue: 5,\n\t},\n\tMatchingUserRPS: {\n\t\tKeyName:      \"matching.rps\",\n\t\tDescription:  \"MatchingUserRPS is request rate per second for each matching host\",\n\t\tDefaultValue: 1200,\n\t},\n\tMatchingWorkerRPS: {\n\t\tKeyName:      \"matching.workerrps\",\n\t\tDescription:  \"MatchingWorkerRPS is background-processing request rate per second for each matching host\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tMatchingDomainUserRPS: {\n\t\tKeyName:      \"matching.domainrps\",\n\t\tDescription:  \"MatchingDomainUserRPS is request rate per domain per second for each matching host\",\n\t\tDefaultValue: 0,\n\t},\n\tMatchingDomainWorkerRPS: {\n\t\tKeyName:      \"matching.domainworkerrps\",\n\t\tDescription:  \"MatchingDomainWorkerRPS is background-processing request rate per domain per second for each matching host\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tMatchingPersistenceMaxQPS: {\n\t\tKeyName:      \"matching.persistenceMaxQPS\",\n\t\tDescription:  \"MatchingPersistenceMaxQPS is the max qps matching host can query DB\",\n\t\tDefaultValue: 3000,\n\t},\n\tMatchingPersistenceGlobalMaxQPS: {\n\t\tKeyName:      \"matching.persistenceGlobalMaxQPS\",\n\t\tDescription:  \"MatchingPersistenceGlobalMaxQPS is the max qps matching cluster can query DB\",\n\t\tDefaultValue: 0,\n\t},\n\tMatchingMinTaskThrottlingBurstSize: {\n\t\tKeyName:      \"matching.minTaskThrottlingBurstSize\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingMinTaskThrottlingBurstSize is the minimum burst size for task list throttling\",\n\t\tDefaultValue: 1,\n\t},\n\tMatchingGetTasksBatchSize: {\n\t\tKeyName:      \"matching.getTasksBatchSize\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingGetTasksBatchSize is the maximum batch size to fetch from the task buffer\",\n\t\tDefaultValue: 1000,\n\t},\n\tMatchingOutstandingTaskAppendsThreshold: {\n\t\tKeyName:      \"matching.outstandingTaskAppendsThreshold\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingOutstandingTaskAppendsThreshold is the threshold for outstanding task appends\",\n\t\tDefaultValue: 250,\n\t},\n\tMatchingMaxTaskBatchSize: {\n\t\tKeyName:      \"matching.maxTaskBatchSize\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingMaxTaskBatchSize is max batch size for task writer\",\n\t\tDefaultValue: 100,\n\t},\n\tMatchingMaxTaskDeleteBatchSize: {\n\t\tKeyName:      \"matching.maxTaskDeleteBatchSize\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingMaxTaskDeleteBatchSize is the max batch size for range deletion of tasks\",\n\t\tDefaultValue: 100,\n\t},\n\tMatchingThrottledLogRPS: {\n\t\tKeyName:      \"matching.throttledLogRPS\",\n\t\tDescription:  \"MatchingThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\",\n\t\tDefaultValue: 20,\n\t},\n\tMatchingNumTasklistWritePartitions: {\n\t\tKeyName:      \"matching.numTasklistWritePartitions\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingNumTasklistWritePartitions is the number of write partitions for a task list\",\n\t\tDefaultValue: 1,\n\t},\n\tMatchingNumTasklistReadPartitions: {\n\t\tKeyName:      \"matching.numTasklistReadPartitions\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingNumTasklistReadPartitions is the number of read partitions for a task list\",\n\t\tDefaultValue: 1,\n\t},\n\tMatchingForwarderMaxOutstandingPolls: {\n\t\tKeyName:      \"matching.forwarderMaxOutstandingPolls\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingForwarderMaxOutstandingPolls is the max number of inflight polls from the forwarder\",\n\t\tDefaultValue: 1,\n\t},\n\tMatchingForwarderMaxOutstandingTasks: {\n\t\tKeyName:      \"matching.forwarderMaxOutstandingTasks\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingForwarderMaxOutstandingTasks is the max number of inflight addTask/queryTask from the forwarder\",\n\t\tDefaultValue: 1,\n\t},\n\tMatchingForwarderMaxRatePerSecond: {\n\t\tKeyName:      \"matching.forwarderMaxRatePerSecond\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingForwarderMaxRatePerSecond is the max rate at which add/query can be forwarded\",\n\t\tDefaultValue: 10,\n\t},\n\tMatchingForwarderMaxChildrenPerNode: {\n\t\tKeyName:      \"matching.forwarderMaxChildrenPerNode\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingForwarderMaxChildrenPerNode is the max number of children per node in the task list partition tree\",\n\t\tDefaultValue: 20,\n\t},\n\tMatchingReadRangeSize: {\n\t\tKeyName:      \"matching.readRangeSize\",\n\t\tDescription:  \"MatchingReadRangeSize is the read range size for the task reader\",\n\t\tDefaultValue: 50000,\n\t},\n\tMatchingPartitionUpscaleRPS: {\n\t\tKeyName:      \"matching.partitionUpscaleRPS\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingPartitionUpscaleRPS is the threshold of adding tasks RPS per partition to trigger upscale\",\n\t\tDefaultValue: 200,\n\t},\n\tMatchingIsolationGroupsPerPartition: {\n\t\tKeyName:      \"matching.isolationGroupsPerPartition\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIsolationGroupsPerPartition is the target number of isolation groups to assign to each partition\",\n\t\tDefaultValue: 2,\n\t},\n\tMatchingPercentageOnboardedToShardManager: {\n\t\tKeyName:      \"matching.percentageOnboardedToShardManager\",\n\t\tDescription:  \"MatchingPercentageOnboardedToShardManager is the percentage of task lists that will be onboarded to the shard manager\",\n\t\tDefaultValue: 0,\n\t},\n\tHistoryRPS: {\n\t\tKeyName:      \"history.rps\",\n\t\tDescription:  \"HistoryRPS is request rate per second for each history host\",\n\t\tDefaultValue: 3000,\n\t},\n\tHistoryPersistenceMaxQPS: {\n\t\tKeyName:      \"history.persistenceMaxQPS\",\n\t\tDescription:  \"HistoryPersistenceMaxQPS is the max qps history host can query DB\",\n\t\tDefaultValue: 9000,\n\t},\n\tHistoryPersistenceGlobalMaxQPS: {\n\t\tKeyName:      \"history.persistenceGlobalMaxQPS\",\n\t\tDescription:  \"HistoryPersistenceGlobalMaxQPS is the max qps history cluster can query DB\",\n\t\tDefaultValue: 0,\n\t},\n\tHistoryVisibilityOpenMaxQPS: {\n\t\tKeyName:      \"history.historyVisibilityOpenMaxQPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryVisibilityOpenMaxQPS is max qps one history host can write visibility open_executions\",\n\t\tDefaultValue: 300,\n\t},\n\tHistoryVisibilityClosedMaxQPS: {\n\t\tKeyName:      \"history.historyVisibilityClosedMaxQPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryVisibilityClosedMaxQPS is max qps one history host can write visibility closed_executions\",\n\t\tDefaultValue: 300,\n\t},\n\tHistoryCacheInitialSize: {\n\t\tKeyName:      \"history.cacheInitialSize\",\n\t\tDescription:  \"HistoryCacheInitialSize is initial size of history cache\",\n\t\tDefaultValue: 128,\n\t},\n\tHistoryCacheMaxSize: {\n\t\tKeyName:      \"history.cacheMaxSize\",\n\t\tDescription:  \"HistoryCacheMaxSize is max size of history cache\",\n\t\tDefaultValue: 512,\n\t},\n\tExecutionCacheMaxByteSize: {\n\t\tKeyName:      \"history.executionCacheMaxByteSize\",\n\t\tDescription:  \"ExecutionCacheMaxByteSize is max size of execution cache in bytes\",\n\t\tDefaultValue: 0,\n\t},\n\tEventsCacheInitialCount: {\n\t\tKeyName:      \"history.eventsCacheInitialSize\",\n\t\tDescription:  \"EventsCacheInitialCount is initial count of events cache\",\n\t\tDefaultValue: 128,\n\t},\n\tEventsCacheMaxCount: {\n\t\tKeyName:      \"history.eventsCacheMaxSize\",\n\t\tDescription:  \"EventsCacheMaxCount is max count of events cache\",\n\t\tDefaultValue: 512,\n\t},\n\tEventsCacheMaxSize: {\n\t\tKeyName:      \"history.eventsCacheMaxSizeInBytes\",\n\t\tDescription:  \"EventsCacheMaxSize is max size of events cache in bytes\",\n\t\tDefaultValue: 0,\n\t},\n\tEventsCacheGlobalInitialCount: {\n\t\tKeyName:      \"history.eventsCacheGlobalInitialSize\",\n\t\tDescription:  \"EventsCacheGlobalInitialCount is initial count of global events cache\",\n\t\tDefaultValue: 4096,\n\t},\n\tEventsCacheGlobalMaxCount: {\n\t\tKeyName:      \"history.eventsCacheGlobalMaxSize\",\n\t\tDescription:  \"EventsCacheGlobalMaxCount is max count of global events cache\",\n\t\tDefaultValue: 131072,\n\t},\n\tAcquireShardConcurrency: {\n\t\tKeyName:      \"history.acquireShardConcurrency\",\n\t\tDescription:  \"AcquireShardConcurrency is number of goroutines that can be used to acquire shards in the shard controller.\",\n\t\tDefaultValue: 1,\n\t},\n\tTaskProcessRPS: {\n\t\tKeyName:      \"history.taskProcessRPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TaskProcessRPS is the task processing rate per second for each domain\",\n\t\tDefaultValue: 1000,\n\t},\n\tTaskSchedulerType: {\n\t\tKeyName:      \"history.taskSchedulerType\",\n\t\tDescription:  \"TaskSchedulerType is the task scheduler type for priority task processor\",\n\t\tDefaultValue: 2, // int(task.SchedulerTypeWRR),\n\t},\n\tTaskSchedulerWorkerCount: {\n\t\tKeyName:      \"history.taskSchedulerWorkerCount\",\n\t\tDescription:  \"TaskSchedulerWorkerCount is the number of workers per host in task scheduler\",\n\t\tDefaultValue: 200,\n\t},\n\tTaskSchedulerQueueSize: {\n\t\tKeyName:      \"history.taskSchedulerQueueSize\",\n\t\tDescription:  \"TaskSchedulerQueueSize is the size of task channel for host level task scheduler\",\n\t\tDefaultValue: 10000,\n\t},\n\tTaskSchedulerDispatcherCount: {\n\t\tKeyName:      \"history.taskSchedulerDispatcherCount\",\n\t\tDescription:  \"TaskSchedulerDispatcherCount is the number of task dispatcher in task scheduler (only applies to host level task scheduler)\",\n\t\tDefaultValue: 1,\n\t},\n\tTaskSchedulerGlobalDomainRPS: {\n\t\tKeyName:      \"history.taskSchedulerGlobalDomainRPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TaskSchedulerGlobalDomainRPS is the task scheduling domain rate limit per second for the whole Cadence cluster\",\n\t\tDefaultValue: 1000,\n\t},\n\tTaskCriticalRetryCount: {\n\t\tKeyName:      \"history.taskCriticalRetryCount\",\n\t\tDescription:  \"TaskCriticalRetryCount is the critical retry count for background tasks, when task attempt exceeds this threshold:- task attempt metrics and additional error logs will be emitted- task priority will be lowered\",\n\t\tDefaultValue: 50,\n\t},\n\tQueueProcessorSplitMaxLevel: {\n\t\tKeyName:      \"history.queueProcessorSplitMaxLevel\",\n\t\tDescription:  \"QueueProcessorSplitMaxLevel is the max processing queue level\",\n\t\tDefaultValue: 2, // 3 levels, start from 0\n\t},\n\tQueueMaxPendingTaskCount: {\n\t\tKeyName:      \"history.queueMaxPendingTaskCount\",\n\t\tDescription:  \"QueueMaxPendingTaskCount is the max number of pending tasks in the queue\",\n\t\tDefaultValue: 10000,\n\t},\n\tQueueCriticalPendingTaskCount: {\n\t\tKeyName:      \"history.queueCriticalPendingTaskCount\",\n\t\tDescription:  \"QueueCriticalPendingTaskCount is the critical pending task count for the queue, which is supposed to be less than QueueMaxPendingTaskCount\",\n\t\tDefaultValue: 9000,\n\t},\n\tTimerTaskBatchSize: {\n\t\tKeyName:      \"history.timerTaskBatchSize\",\n\t\tDescription:  \"TimerTaskBatchSize is batch size for timer processor to process tasks\",\n\t\tDefaultValue: 100,\n\t},\n\tTimerTaskDeleteBatchSize: {\n\t\tKeyName:      \"history.timerTaskDeleteBatchSize\",\n\t\tDescription:  \"TimerTaskDeleteBatchSize is batch size for timer processor to delete timer tasks\",\n\t\tDefaultValue: 4000,\n\t},\n\tTimerProcessorGetFailureRetryCount: {\n\t\tKeyName:      \"history.timerProcessorGetFailureRetryCount\",\n\t\tDescription:  \"TimerProcessorGetFailureRetryCount is retry count for timer processor get failure operation\",\n\t\tDefaultValue: 5,\n\t},\n\tTimerProcessorCompleteTimerFailureRetryCount: {\n\t\tKeyName:      \"history.timerProcessorCompleteTimerFailureRetryCount\",\n\t\tDescription:  \"TimerProcessorCompleteTimerFailureRetryCount is retry count for timer processor complete timer operation\",\n\t\tDefaultValue: 10,\n\t},\n\tTimerProcessorFailoverMaxPollRPS: {\n\t\tKeyName:      \"history.timerProcessorFailoverMaxPollRPS\",\n\t\tDescription:  \"TimerProcessorFailoverMaxPollRPS is max poll rate per second for timer processor\",\n\t\tDefaultValue: 1,\n\t},\n\tTimerProcessorMaxPollRPS: {\n\t\tKeyName:      \"history.timerProcessorMaxPollRPS\",\n\t\tDescription:  \"TimerProcessorMaxPollRPS is max poll rate per second for timer processor\",\n\t\tDefaultValue: 20,\n\t},\n\tTimerProcessorMaxRedispatchQueueSize: {\n\t\tKeyName:      \"history.timerProcessorMaxRedispatchQueueSize\",\n\t\tDescription:  \"TimerProcessorMaxRedispatchQueueSize is the threshold of the number of tasks in the redispatch queue for timer processor\",\n\t\tDefaultValue: 10000,\n\t},\n\tTimerProcessorHistoryArchivalSizeLimit: {\n\t\tKeyName:      \"history.timerProcessorHistoryArchivalSizeLimit\",\n\t\tDescription:  \"TimerProcessorHistoryArchivalSizeLimit is the max history size for inline archival\",\n\t\tDefaultValue: 500 * 1024,\n\t},\n\tTransferTaskBatchSize: {\n\t\tKeyName:      \"history.transferTaskBatchSize\",\n\t\tDescription:  \"TransferTaskBatchSize is batch size for transferQueueProcessor\",\n\t\tDefaultValue: 100,\n\t},\n\tTransferTaskDeleteBatchSize: {\n\t\tKeyName:      \"history.transferTaskDeleteBatchSize\",\n\t\tDescription:  \"TransferTaskDeleteBatchSize is batch size for transferQueueProcessor to delete transfer tasks\",\n\t\tDefaultValue: 4000,\n\t},\n\tTransferProcessorFailoverMaxPollRPS: {\n\t\tKeyName:      \"history.transferProcessorFailoverMaxPollRPS\",\n\t\tDescription:  \"TransferProcessorFailoverMaxPollRPS is max poll rate per second for transferQueueProcessor\",\n\t\tDefaultValue: 1,\n\t},\n\tTransferProcessorMaxPollRPS: {\n\t\tKeyName:      \"history.transferProcessorMaxPollRPS\",\n\t\tDescription:  \"TransferProcessorMaxPollRPS is max poll rate per second for transferQueueProcessor\",\n\t\tDefaultValue: 20,\n\t},\n\tTransferProcessorCompleteTransferFailureRetryCount: {\n\t\tKeyName:      \"history.transferProcessorCompleteTransferFailureRetryCount\",\n\t\tDescription:  \"TransferProcessorCompleteTransferFailureRetryCount is times of retry for failure\",\n\t\tDefaultValue: 10,\n\t},\n\tTransferProcessorMaxRedispatchQueueSize: {\n\t\tKeyName:      \"history.transferProcessorMaxRedispatchQueueSize\",\n\t\tDescription:  \"TransferProcessorMaxRedispatchQueueSize is the threshold of the number of tasks in the redispatch queue for transferQueueProcessor\",\n\t\tDefaultValue: 10000,\n\t},\n\tReplicatorTaskBatchSize: {\n\t\tKeyName:      \"history.replicatorTaskBatchSize\",\n\t\tDescription:  \"ReplicatorTaskBatchSize is batch size for ReplicatorProcessor\",\n\t\tDefaultValue: 25,\n\t},\n\tReplicatorMinTaskBatchSize: {\n\t\tKeyName:      \"history.replicatorMinTaskBatchSize\",\n\t\tDescription:  \"ReplicatorMinTaskBatchSize is minimum batch size for ReplicatorProcessor\",\n\t\tFilters:      []Filter{ShardID, ClusterName},\n\t\tDefaultValue: 1,\n\t},\n\tReplicatorMaxTaskBatchSize: {\n\t\tKeyName:      \"history.replicatorMaxTaskBatchSize\",\n\t\tDescription:  \"ReplicatorMaxTaskBatchSize is maximum size for ReplicatorProcessor\",\n\t\tFilters:      []Filter{ShardID, ClusterName},\n\t\tDefaultValue: 1000,\n\t},\n\tReplicatorTaskBatchStepCount: {\n\t\tKeyName:      \"history.replicatorTaskBatchStepCount\",\n\t\tDescription:  \"ReplicatorTaskBatchStepCount is how many batch size values between ReplicatorMinTaskBatchSize and ReplicatorMaxTaskBatchSize are used by ReplicatorProcessor to adjust the batch size of the replication tasks. Less value will make the adjustment more aggressive. It cannot be less than 3, otherwise only ReplicatorMinTaskBatchSize and ReplicatorMaxTaskBatchSize will be used.\",\n\t\tFilters:      []Filter{ShardID, ClusterName},\n\t\tDefaultValue: 10,\n\t},\n\tReplicatorTaskDeleteBatchSize: {\n\t\tKeyName:      \"history.replicatorTaskDeleteBatchSize\",\n\t\tDescription:  \"ReplicatorTaskDeleteBatchSize is batch size for ReplicatorProcessor to delete replication tasks\",\n\t\tDefaultValue: 4000,\n\t},\n\tHistoryNodeDeleteBatchSize: {\n\t\tKeyName:      \"history.historyNodeDeleteBatchSize\",\n\t\tDescription:  \"HistoryNodeDeleteBatchSize is batch size for deleting history nodes\",\n\t\tDefaultValue: 1000,\n\t},\n\tReplicatorReadTaskMaxRetryCount: {\n\t\tKeyName:      \"history.replicatorReadTaskMaxRetryCount\",\n\t\tDescription:  \"ReplicatorReadTaskMaxRetryCount is the number of read replication task retry time\",\n\t\tDefaultValue: 3,\n\t},\n\tReplicatorCacheCapacity: {\n\t\tKeyName:      \"history.replicatorCacheCapacity\",\n\t\tDescription:  \"ReplicatorCacheCapacity is the capacity of replication cache in number of tasks\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicatorCacheMaxSize: {\n\t\tKeyName:      \"history.replicatorCacheMaxSize\",\n\t\tDescription:  \"ReplicatorCacheMaxSize is the max size of the replication cache in bytes\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicationBudgetManagerMaxSizeBytes: {\n\t\tKeyName:      \"history.replicationBudgetManagerMaxSizeBytes\",\n\t\tDescription:  \"ReplicationBudgetManagerMaxSizeBytes is the max size of the replication budget manager cache in bytes\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicationBudgetManagerMaxSizeCount: {\n\t\tKeyName:      \"history.replicationBudgetManagerMaxSizeCount\",\n\t\tDescription:  \"ReplicationBudgetManagerMaxSizeCount is the max count of the replication budget manager cache\",\n\t\tDefaultValue: 0,\n\t},\n\tMaximumBufferedEventsBatch: {\n\t\tKeyName:      \"history.maximumBufferedEventsBatch\",\n\t\tDescription:  \"MaximumBufferedEventsBatch is max number of buffer event in mutable state\",\n\t\tDefaultValue: 100,\n\t},\n\tMaximumSignalsPerExecution: {\n\t\tKeyName:      \"history.maximumSignalsPerExecution\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"MaximumSignalsPerExecution is max number of signals supported by single execution\",\n\t\tDefaultValue: 10000, // 10K signals should big enough given workflow execution has 200K history lengh limit. It needs to be non-zero to protect continueAsNew from infinit loop\n\t},\n\tNumArchiveSystemWorkflows: {\n\t\tKeyName:      \"history.numArchiveSystemWorkflows\",\n\t\tDescription:  \"NumArchiveSystemWorkflows is key for number of archive system workflows running in total\",\n\t\tDefaultValue: 1000,\n\t},\n\tArchiveRequestRPS: {\n\t\tKeyName:      \"history.archiveRequestRPS\",\n\t\tDescription:  \"ArchiveRequestRPS is the rate limit on the number of archive request per second\",\n\t\tDefaultValue: 300, // should be much smaller than frontend RPS\n\t},\n\tArchiveInlineHistoryRPS: {\n\t\tKeyName:      \"history.archiveInlineHistoryRPS\",\n\t\tDescription:  \"ArchiveInlineHistoryRPS is the (per instance) rate limit on the number of inline history archival attempts per second\",\n\t\tDefaultValue: 1000,\n\t},\n\tArchiveInlineHistoryGlobalRPS: {\n\t\tKeyName:      \"history.archiveInlineHistoryGlobalRPS\",\n\t\tDescription:  \"ArchiveInlineHistoryGlobalRPS is the global rate limit on the number of inline history archival attempts per second\",\n\t\tDefaultValue: 10000,\n\t},\n\tArchiveInlineVisibilityRPS: {\n\t\tKeyName:      \"history.archiveInlineVisibilityRPS\",\n\t\tDescription:  \"ArchiveInlineVisibilityRPS is the (per instance) rate limit on the number of inline visibility archival attempts per second\",\n\t\tDefaultValue: 1000,\n\t},\n\tArchiveInlineVisibilityGlobalRPS: {\n\t\tKeyName:      \"history.archiveInlineVisibilityGlobalRPS\",\n\t\tDescription:  \"ArchiveInlineVisibilityGlobalRPS is the global rate limit on the number of inline visibility archival attempts per second\",\n\t\tDefaultValue: 10000,\n\t},\n\tHistoryMaxAutoResetPoints: {\n\t\tKeyName:      \"history.historyMaxAutoResetPoints\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryMaxAutoResetPoints is the key for max number of auto reset points stored in mutableState\",\n\t\tDefaultValue: 20,\n\t},\n\tParentClosePolicyThreshold: {\n\t\tKeyName:      \"history.parentClosePolicyThreshold\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ParentClosePolicyThreshold decides that parent close policy will be processed by sys workers(if enabled) if the number of children is greater than or equal to this threshold\",\n\t\tDefaultValue: 10,\n\t},\n\tParentClosePolicyBatchSize: {\n\t\tKeyName:      \"history.parentClosePolicyBatchSize\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ParentClosePolicyBatchSize is the batch size of parent close policy processed by sys workers\",\n\t\tDefaultValue: 200,\n\t},\n\tNumParentClosePolicySystemWorkflows: {\n\t\tKeyName:      \"history.numParentClosePolicySystemWorkflows\",\n\t\tDescription:  \"NumParentClosePolicySystemWorkflows is key for number of parentClosePolicy system workflows running in total\",\n\t\tDefaultValue: 10,\n\t},\n\tHistoryThrottledLogRPS: {\n\t\tKeyName:      \"history.throttledLogRPS\",\n\t\tDescription:  \"HistoryThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\",\n\t\tDefaultValue: 4,\n\t},\n\tDecisionRetryCriticalAttempts: {\n\t\tKeyName:      \"history.decisionRetryCriticalAttempts\",\n\t\tDescription:  \"DecisionRetryCriticalAttempts is decision attempt threshold for logging and emiting metrics\",\n\t\tDefaultValue: 10,\n\t},\n\tDecisionRetryMaxAttempts: {\n\t\tKeyName:      \"history.decisionRetryMaxAttempts\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DecisionRetryMaxAttempts is the max limit for decision retry attempts. 0 indicates infinite number of attempts.\",\n\t\tDefaultValue: 1000,\n\t},\n\tNormalDecisionScheduleToStartMaxAttempts: {\n\t\tKeyName:      \"history.normalDecisionScheduleToStartMaxAttempts\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"NormalDecisionScheduleToStartMaxAttempts is the maximum decision attempt for creating a scheduleToStart timeout timer for normal (non-sticky) decision\",\n\t\tDefaultValue: 0,\n\t},\n\tMaxBufferedQueryCount: {\n\t\tKeyName:      \"history.MaxBufferedQueryCount\",\n\t\tDescription:  \"MaxBufferedQueryCount indicates the maximum number of queries which can be buffered at a given time for a single workflow\",\n\t\tDefaultValue: 1,\n\t},\n\tMutableStateChecksumGenProbability: {\n\t\tKeyName:      \"history.mutableStateChecksumGenProbability\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"MutableStateChecksumGenProbability is the probability [0-100] that checksum will be generated for mutable state\",\n\t\tDefaultValue: 0,\n\t},\n\tMutableStateChecksumVerifyProbability: {\n\t\tKeyName:      \"history.mutableStateChecksumVerifyProbability\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"MutableStateChecksumVerifyProbability is the probability [0-100] that checksum will be verified for mutable state\",\n\t\tDefaultValue: 0,\n\t},\n\tTaskSchedulerMigrationRatio: {\n\t\tKeyName:      \"history.taskSchedulerMigrationRatio\",\n\t\tDescription:  \"DEPRECATED: TaskSchedulerMigrationRatio is the ratio of task that is migrated to new scheduler\",\n\t\tDefaultValue: 0,\n\t},\n\tMaxActivityCountDispatchByDomain: {\n\t\tKeyName:      \"history.maxActivityCountDispatchByDomain\",\n\t\tDescription:  \"MaxActivityCountDispatchByDomain max # of activity tasks to dispatch to matching before creating transfer tasks. This is an performance optimization to skip activity scheduling efforts.\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicationTaskFetcherParallelism: {\n\t\tKeyName:      \"history.ReplicationTaskFetcherParallelism\",\n\t\tDescription:  \"ReplicationTaskFetcherParallelism determines how many go routines we spin up for fetching tasks\",\n\t\tDefaultValue: 1,\n\t},\n\tReplicationTaskProcessorErrorRetryMaxAttempts: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorErrorRetryMaxAttempts\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorErrorRetryMaxAttempts is the max retry attempts for applying replication tasks\",\n\t\tDefaultValue: 10,\n\t},\n\tWorkflowIDExternalRPS: {\n\t\tKeyName:      \"history.workflowIDExternalRPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"WorkflowIDExternalRPS is the rate limit per workflowID for external calls\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tWorkflowIDInternalRPS: {\n\t\tKeyName:      \"history.workflowIDInternalRPS\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"WorkflowIDInternalRPS is the rate limit per workflowID for internal calls\",\n\t\tDefaultValue: UnlimitedRPS,\n\t},\n\tWorkerPersistenceMaxQPS: {\n\t\tKeyName:      \"worker.persistenceMaxQPS\",\n\t\tDescription:  \"WorkerPersistenceMaxQPS is the max qps worker host can query DB\",\n\t\tDefaultValue: 500,\n\t},\n\tWorkerPersistenceGlobalMaxQPS: {\n\t\tKeyName:      \"worker.persistenceGlobalMaxQPS\",\n\t\tDescription:  \"WorkerPersistenceGlobalMaxQPS is the max qps worker cluster can query DB\",\n\t\tDefaultValue: 0,\n\t},\n\tWorkerIndexerConcurrency: {\n\t\tKeyName:      \"worker.indexerConcurrency\",\n\t\tDescription:  \"WorkerIndexerConcurrency is the max concurrent messages to be processed at any given time\",\n\t\tDefaultValue: 1000,\n\t},\n\tWorkerESProcessorNumOfWorkers: {\n\t\tKeyName:      \"worker.ESProcessorNumOfWorkers\",\n\t\tDescription:  \"WorkerESProcessorNumOfWorkers is num of workers for esProcessor\",\n\t\tDefaultValue: 1,\n\t},\n\tWorkerESProcessorBulkActions: {\n\t\tKeyName:      \"worker.ESProcessorBulkActions\",\n\t\tDescription:  \"WorkerESProcessorBulkActions is max number of requests in bulk for esProcessor\",\n\t\tDefaultValue: 1000,\n\t},\n\tWorkerESProcessorBulkSize: {\n\t\tKeyName:      \"worker.ESProcessorBulkSize\",\n\t\tDescription:  \"WorkerESProcessorBulkSize is max total size of bulk in bytes for esProcessor\",\n\t\tDefaultValue: 2 << 24, // 16MB\n\t},\n\tWorkerArchiverConcurrency: {\n\t\tKeyName:      \"worker.ArchiverConcurrency\",\n\t\tDescription:  \"WorkerArchiverConcurrency is controls the number of coroutines handling archival work per archival workflow\",\n\t\tDefaultValue: 50,\n\t},\n\tWorkerArchivalsPerIteration: {\n\t\tKeyName:      \"worker.ArchivalsPerIteration\",\n\t\tDescription:  \"WorkerArchivalsPerIteration is controls the number of archivals handled in each iteration of archival workflow\",\n\t\tDefaultValue: 1000,\n\t},\n\tWorkerThrottledLogRPS: {\n\t\tKeyName:      \"worker.throttledLogRPS\",\n\t\tDescription:  \"WorkerThrottledLogRPS is the rate limit on number of log messages emitted per second for throttled logger\",\n\t\tDefaultValue: 20,\n\t},\n\tScannerPersistenceMaxQPS: {\n\t\tKeyName:      \"worker.scannerPersistenceMaxQPS\",\n\t\tDescription:  \"ScannerPersistenceMaxQPS is the maximum rate of persistence calls from worker.Scanner\",\n\t\tDefaultValue: 5,\n\t},\n\tScannerGetOrphanTasksPageSize: {\n\t\tKeyName:      \"worker.scannerGetOrphanTasksPageSize\",\n\t\tDescription:  \"ScannerGetOrphanTasksPageSize is the maximum number of orphans to delete in one batch\",\n\t\tDefaultValue: 1000,\n\t},\n\tScannerBatchSizeForTasklistHandler: {\n\t\tKeyName:      \"worker.scannerBatchSizeForTasklistHandler\",\n\t\tDescription:  \"ScannerBatchSizeForTasklistHandler is for: 1. max number of tasks to query per call(get tasks for tasklist) in the scavenger handler. 2. The scavenger then uses the return to decide if a tasklist can be deleted. It's better to keep it a relatively high number to let it be more efficient.\",\n\t\tDefaultValue: 1000,\n\t},\n\tScannerMaxTasksProcessedPerTasklistJob: {\n\t\tKeyName:      \"worker.scannerMaxTasksProcessedPerTasklistJob\",\n\t\tDescription:  \"ScannerMaxTasksProcessedPerTasklistJob is the number of tasks to process for a tasklist in each workflow run\",\n\t\tDefaultValue: 256,\n\t},\n\tConcreteExecutionsScannerConcurrency: {\n\t\tKeyName:      \"worker.executionsScannerConcurrency\",\n\t\tDescription:  \"ConcreteExecutionsScannerConcurrency indicates the concurrency of concrete execution scanner\",\n\t\tDefaultValue: 25,\n\t},\n\tConcreteExecutionsScannerBlobstoreFlushThreshold: {\n\t\tKeyName:      \"worker.executionsScannerBlobstoreFlushThreshold\",\n\t\tDescription:  \"ConcreteExecutionsScannerBlobstoreFlushThreshold indicates the flush threshold of blobstore in concrete execution scanner\",\n\t\tDefaultValue: 100,\n\t},\n\tConcreteExecutionsScannerActivityBatchSize: {\n\t\tKeyName:      \"worker.executionsScannerActivityBatchSize\",\n\t\tDescription:  \"ConcreteExecutionsScannerActivityBatchSize indicates the batch size of scanner activities\",\n\t\tDefaultValue: 25,\n\t},\n\tConcreteExecutionsScannerPersistencePageSize: {\n\t\tKeyName:      \"worker.executionsScannerPersistencePageSize\",\n\t\tDescription:  \"ConcreteExecutionsScannerPersistencePageSize indicates the page size of execution persistence fetches in concrete execution scanner\",\n\t\tDefaultValue: 1000,\n\t},\n\tCurrentExecutionsScannerConcurrency: {\n\t\tKeyName:      \"worker.currentExecutionsConcurrency\",\n\t\tDescription:  \"CurrentExecutionsScannerConcurrency indicates the concurrency of current executions scanner\",\n\t\tDefaultValue: 25,\n\t},\n\tCurrentExecutionsScannerBlobstoreFlushThreshold: {\n\t\tKeyName:      \"worker.currentExecutionsBlobstoreFlushThreshold\",\n\t\tDescription:  \"CurrentExecutionsScannerBlobstoreFlushThreshold indicates the flush threshold of blobstore in current executions scanner\",\n\t\tDefaultValue: 100,\n\t},\n\tCurrentExecutionsScannerActivityBatchSize: {\n\t\tKeyName:      \"worker.currentExecutionsActivityBatchSize\",\n\t\tDescription:  \"CurrentExecutionsScannerActivityBatchSize indicates the batch size of scanner activities\",\n\t\tDefaultValue: 25,\n\t},\n\tCurrentExecutionsScannerPersistencePageSize: {\n\t\tKeyName:      \"worker.currentExecutionsPersistencePageSize\",\n\t\tDescription:  \"CurrentExecutionsScannerPersistencePageSize indicates the page size of execution persistence fetches in current executions scanner\",\n\t\tDefaultValue: 1000,\n\t},\n\tTimersScannerConcurrency: {\n\t\tKeyName:      \"worker.timersScannerConcurrency\",\n\t\tDescription:  \"TimersScannerConcurrency is the concurrency of timers scanner\",\n\t\tDefaultValue: 5,\n\t},\n\tTimersScannerPersistencePageSize: {\n\t\tKeyName:      \"worker.timersScannerPersistencePageSize\",\n\t\tDescription:  \"TimersScannerPersistencePageSize is the page size of timers persistence fetches in timers scanner\",\n\t\tDefaultValue: 1000,\n\t},\n\tTimersScannerBlobstoreFlushThreshold: {\n\t\tKeyName:      \"worker.timersScannerBlobstoreFlushThreshold\",\n\t\tDescription:  \"TimersScannerBlobstoreFlushThreshold is threshold to flush blob store\",\n\t\tDefaultValue: 100,\n\t},\n\tTimersScannerActivityBatchSize: {\n\t\tKeyName:      \"worker.timersScannerActivityBatchSize\",\n\t\tDescription:  \"TimersScannerActivityBatchSize is TimersScannerActivityBatchSize\",\n\t\tDefaultValue: 25,\n\t},\n\tTimersScannerPeriodStart: {\n\t\tKeyName:      \"worker.timersScannerPeriodStart\",\n\t\tDescription:  \"TimersScannerPeriodStart is interval start for fetching scheduled timers\",\n\t\tDefaultValue: 24,\n\t},\n\tTimersScannerPeriodEnd: {\n\t\tKeyName:      \"worker.timersScannerPeriodEnd\",\n\t\tDescription:  \"TimersScannerPeriodEnd is interval end for fetching scheduled timers\",\n\t\tDefaultValue: 3,\n\t},\n\tESAnalyzerMaxNumDomains: {\n\t\tKeyName:      \"worker.ESAnalyzerMaxNumDomains\",\n\t\tDescription:  \"ESAnalyzerMaxNumDomains defines how many domains to check\",\n\t\tDefaultValue: 500,\n\t},\n\tESAnalyzerMaxNumWorkflowTypes: {\n\t\tKeyName:      \"worker.ESAnalyzerMaxNumWorkflowTypes\",\n\t\tDescription:  \"ESAnalyzerMaxNumWorkflowTypes defines how many workflow types to check per domain\",\n\t\tDefaultValue: 100,\n\t},\n\tESAnalyzerNumWorkflowsToRefresh: {\n\t\tKeyName:      \"worker.ESAnalyzerNumWorkflowsToRefresh\",\n\t\tDescription:  \"ESAnalyzerNumWorkflowsToRefresh controls how many workflows per workflow type should be refreshed per workflow type\",\n\t\tDefaultValue: 100,\n\t},\n\tESAnalyzerMinNumWorkflowsForAvg: {\n\t\tKeyName:      \"worker.ESAnalyzerMinNumWorkflowsForAvg\",\n\t\tDescription:  \"ESAnalyzerMinNumWorkflowsForAvg controls how many workflows to have at least to rely on workflow run time avg per type\",\n\t\tDefaultValue: 100,\n\t},\n\tVisibilityArchivalQueryMaxRangeInDays: {\n\t\tKeyName:      \"frontend.visibilityArchivalQueryMaxRangeInDays\",\n\t\tDescription:  \"VisibilityArchivalQueryMaxRangeInDays is the maximum number of days for a visibility archival query\",\n\t\tDefaultValue: 60,\n\t},\n\tVisibilityArchivalQueryMaxQPS: {\n\t\tKeyName:      \"frontend.visibilityArchivalQueryMaxQPS\",\n\t\tDescription:  \"VisibilityArchivalQueryMaxQPS is the timeout for a visibility archival query\",\n\t\tDefaultValue: 1,\n\t},\n\tWorkflowDeletionJitterRange: {\n\t\tKeyName:      \"system.workflowDeletionJitterRange\",\n\t\tDescription:  \"WorkflowDeletionJitterRange defines the duration in minutes for workflow close tasks jittering\",\n\t\tDefaultValue: 60,\n\t},\n\tSampleLoggingRate: {\n\t\tKeyName:      \"system.sampleLoggingRate\",\n\t\tDescription:  \"The rate for which sampled logs are logged at. 100 means 1/100 is logged\",\n\t\tDefaultValue: 100,\n\t},\n\tLargeShardHistorySizeMetricThreshold: {\n\t\tKeyName:      \"system.largeShardHistorySizeMetricThreshold\",\n\t\tDescription:  \"defines the threshold for what consititutes a large history size to alert on, default is 10mb\",\n\t\tDefaultValue: 10485760,\n\t},\n\tLargeShardHistoryEventMetricThreshold: {\n\t\tKeyName:      \"system.largeShardHistoryEventMetricThreshold\",\n\t\tDescription:  \"defines the threshold for what consititutes a large history event length to alert on, default is 50k\",\n\t\tDefaultValue: 50 * 1024,\n\t},\n\tLargeShardHistoryBlobMetricThreshold: {\n\t\tKeyName:      \"system.largeShardHistoryBlobMetricThreshold\",\n\t\tDescription:  \"defines the threshold for what consititutes a large history blob write to alert on, default is 1/4mb\",\n\t\tDefaultValue: 262144,\n\t},\n\tIsolationGroupStateUpdateRetryAttempts: {\n\t\tKeyName:      \"system.isolationGroupStateUpdateRetryAttempts\",\n\t\tDescription:  \"The number of attempts to push Isolation group configuration to the config store\",\n\t\tDefaultValue: 2,\n\t},\n\tDeleteHistoryEventContextTimeout: {\n\t\tKeyName:      \"system.deleteHistoryEventContextTimeout\",\n\t\tDescription:  \"This is the number of seconds allowed for a deleteHistoryEvent task to the database\",\n\t\tDefaultValue: 30,\n\t},\n\tQueueMaxVirtualQueueCount: {\n\t\tKeyName:      \"history.queueMaxVirtualQueueCount\",\n\t\tDescription:  \"QueueMaxVirtualQueueCount is the max number of virtual queues\",\n\t\tDefaultValue: 2,\n\t},\n\tShardDistributorMaxEtcdTxnOps: {\n\t\tKeyName:      \"shardDistributor.maxEtcdTxnOps\",\n\t\tDescription:  \"ShardDistributorMaxEtcdTxnOps is the maximum number of operations per etcd transaction, must not exceed the etcd cluster's configured --max-txn-ops limit\",\n\t\tDefaultValue: 128,\n\t},\n}\n\nvar BoolKeys = map[BoolKey]DynamicBool{\n\tTestGetBoolPropertyKey: {\n\t\tKeyName:      \"testGetBoolPropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t},\n\tTestGetBoolPropertyFilteredByDomainIDKey: {\n\t\tKeyName:      \"testGetBoolPropertyFilteredByDomainIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t},\n\tTestGetBoolPropertyFilteredByTaskListInfoKey: {\n\t\tKeyName:      \"testGetBoolPropertyFilteredByTaskListInfoKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t},\n\tTestGetBoolPropertyFilteredByDomainKey: {\n\t\tKeyName:      \"testGetBoolPropertyFilteredByDomainKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t\tFilters:      nil,\n\t},\n\tTestGetBoolPropertyFilteredByDomainIDAndWorkflowIDKey: {\n\t\tKeyName:      \"testGetBoolPropertyFilteredByDomainIDandWorkflowIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t\tFilters:      nil,\n\t},\n\tTestGetBoolPropertyFilteredByShardIDKey: {\n\t\tKeyName:      \"testGetBoolPropertyFilteredByShardIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: false,\n\t\tFilters:      []Filter{ShardID},\n\t},\n\tEnableVisibilitySampling: {\n\t\tKeyName:      \"system.enableVisibilitySampling\",\n\t\tDescription:  \"EnableVisibilitySampling is key for enable visibility sampling for basic(DB based) visibility\",\n\t\tDefaultValue: false, // ...\n\t\tFilters:      nil,\n\t},\n\tEnableReadFromClosedExecutionV2: {\n\t\tKeyName:      \"system.enableReadFromClosedExecutionV2\",\n\t\tDescription:  \"EnableReadFromClosedExecutionV2 is key for enable read from cadence_visibility.closed_executions_v2\",\n\t\tDefaultValue: false,\n\t},\n\tEnableLogCustomerQueryParameter: {\n\t\tKeyName:      \"system.enableLogCustomerQueryParameter\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableLogCustomerQueryParameter is key for enable log customer query parameters\",\n\t\tDefaultValue: false,\n\t},\n\tEmitShardDiffLog: {\n\t\tKeyName:      \"history.emitShardDiffLog\",\n\t\tDescription:  \"EmitShardDiffLog is whether emit the shard diff log\",\n\t\tDefaultValue: false,\n\t},\n\tEnableRecordWorkflowExecutionUninitialized: {\n\t\tKeyName:      \"history.enableRecordWorkflowExecutionUninitialized\",\n\t\tDescription:  \"EnableRecordWorkflowExecutionUninitialized enables record workflow execution uninitialized state in ElasticSearch\",\n\t\tDefaultValue: false,\n\t},\n\tEnableCleanupOrphanedHistoryBranchOnWorkflowCreation: {\n\t\tKeyName:      \"history.enableCleanupOrphanedHistoryBranchOnWorkflowCreation\",\n\t\tDescription:  \"EnableCleanupOrphanedHistoryBranchOnWorkflowCreation enables cleanup of orphaned history branches when CreateWorkflowExecution fails\",\n\t\tDefaultValue: true,\n\t},\n\tDisableListVisibilityByFilter: {\n\t\tKeyName:      \"frontend.disableListVisibilityByFilter\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DisableListVisibilityByFilter is config to disable list open/close workflow using filter\",\n\t\tDefaultValue: false,\n\t},\n\tEnableReadFromHistoryArchival: {\n\t\tKeyName:      \"system.enableReadFromHistoryArchival\",\n\t\tDescription:  \"EnableReadFromHistoryArchival is key for enabling reading history from archival store\",\n\t\tDefaultValue: true,\n\t},\n\tEnableReadFromVisibilityArchival: {\n\t\tKeyName:      \"system.enableReadFromVisibilityArchival\",\n\t\tDescription:  \"EnableReadFromVisibilityArchival is key for enabling reading visibility from archival store to override the value from static config.\",\n\t\tDefaultValue: true,\n\t},\n\tEnableDomainNotActiveAutoForwarding: {\n\t\tKeyName:      \"system.enableDomainNotActiveAutoForwarding\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableDomainNotActiveAutoForwarding decides requests form which domain will be forwarded to active cluster if domain is not active in current cluster. Only when selected-api-forwarding or all-domain-apis-forwarding is the policy in ClusterRedirectionPolicy(in static config). If the policy is noop(default) this flag is not doing anything.\",\n\t\tDefaultValue: true,\n\t},\n\tEnableGracefulFailover: {\n\t\tKeyName:      \"system.enableGracefulFailover\",\n\t\tDescription:  \"EnableGracefulFailover is whether enabling graceful failover\",\n\t\tDefaultValue: true,\n\t},\n\tEnableDomainAuditLogging: {\n\t\tKeyName:      \"system.enableDomainAuditLogging\",\n\t\tDescription:  \"EnableDomainAuditLogging enables audit logging for a domain to the domain audit log table\",\n\t\tDefaultValue: false,\n\t},\n\tDisallowQuery: {\n\t\tKeyName:      \"system.disallowQuery\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DisallowQuery is the key to disallow query for a domain\",\n\t\tDefaultValue: false,\n\t},\n\tEnableDebugMode: {\n\t\tKeyName:      \"system.enableDebugMode\",\n\t\tDescription:  \"EnableDebugMode is for enabling debugging components, logs and metrics\",\n\t\tDefaultValue: false,\n\t},\n\tEnableGRPCOutbound: {\n\t\tKeyName:      \"system.enableGRPCOutbound\",\n\t\tDescription:  \"EnableGRPCOutbound is the key for enabling outbound GRPC traffic\",\n\t\tDefaultValue: true,\n\t},\n\tEnableSQLAsyncTransaction: {\n\t\tKeyName:      \"system.enableSQLAsyncTransaction\",\n\t\tDescription:  \"EnableSQLAsyncTransaction is the key for enabling async transaction\",\n\t\tDefaultValue: false,\n\t},\n\tEnableConnectionRetainingDirectChooser: {\n\t\tKeyName:      \"system.enableConnectionRetainingDirectChooser\",\n\t\tDescription:  \"EnableConnectionRetainingDirectChooser is the key for enabling connection retaining direct chooser\",\n\t\tDefaultValue: false,\n\t},\n\tEnableClientVersionCheck: {\n\t\tKeyName:      \"frontend.enableClientVersionCheck\",\n\t\tDescription:  \"EnableClientVersionCheck is enables client version check for frontend\",\n\t\tDefaultValue: false,\n\t},\n\tSendRawWorkflowHistory: {\n\t\tKeyName:      \"frontend.sendRawWorkflowHistory\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"SendRawWorkflowHistory is whether to enable raw history retrieving\",\n\t\tDefaultValue: false,\n\t},\n\tFrontendEmitSignalNameMetricsTag: {\n\t\tKeyName:      \"frontend.emitSignalNameMetricsTag\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendEmitSignalNameMetricsTag enables emitting signal name tag in metrics in frontend client\",\n\t\tDefaultValue: false,\n\t},\n\tEnableQueryAttributeValidation: {\n\t\tKeyName:      \"frontend.enableQueryAttributeValidation\",\n\t\tDescription:  \"EnableQueryAttributeValidation enables validation of queries' search attributes against the dynamic config whitelist\",\n\t\tDefaultValue: true,\n\t},\n\tMatchingEnableSyncMatch: {\n\t\tKeyName:      \"matching.enableSyncMatch\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnableSyncMatch is to enable sync match\",\n\t\tDefaultValue: true,\n\t},\n\tMatchingEnableTaskInfoLogByDomainID: {\n\t\tKeyName:      \"matching.enableTaskInfoLogByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"MatchingEnableTaskInfoLogByDomainID is enables info level logs for decision/activity task based on the request domainID\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnableTasklistGuardAgainstOwnershipShardLoss: {\n\t\tKeyName:      \"matching.enableTasklistGuardAgainstOwnershipLoss\",\n\t\tDescription:  \"allows guards to ensure that tasklists don't continue processing if there's signal that they've lost ownership\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnableGetNumberOfPartitionsFromCache: {\n\t\tKeyName:      \"matching.enableGetNumberOfPartitionsFromCache\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnableGetNumberOfPartitionsFromCache is to enable getting number of partitions from cache instead of dynamic config\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnableStandbyTaskCompletion: {\n\t\tKeyName:      \"matching.enableStandbyTaskCompletion\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnableStandbyTaskCompletion is to enable completion of tasks in the domain's passive side\",\n\t\tDefaultValue: true,\n\t},\n\tMatchingEnableAdaptiveScaler: {\n\t\tKeyName:      \"matching.enableAdaptiveScaler\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnableAdaptiveScaler is to enable adaptive task list scaling\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnablePartitionEmptyCheck: {\n\t\tKeyName:      \"matching.enablePartitionEmptyCheck\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnablePartitionEmptyCheck enables using TaskListStatus.empty to check if a partition is empty\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnableReturnAllTaskListKinds: {\n\t\tKeyName:      \"matching.matchingReturnAllTaskListKinds\",\n\t\tDescription:  \"Returns TaskLists of all kinds when GetTaskListsByDomain is called. Useful in testing Cadence\",\n\t\tDefaultValue: false,\n\t},\n\tEventsCacheGlobalEnable: {\n\t\tKeyName:      \"history.eventsCacheGlobalEnable\",\n\t\tDescription:  \"EventsCacheGlobalEnable is enables global cache over all history shards\",\n\t\tDefaultValue: false,\n\t},\n\tQueueProcessorEnableSplit: {\n\t\tKeyName:      \"history.queueProcessorEnableSplit\",\n\t\tDescription:  \"QueueProcessorEnableSplit indicates whether processing queue split policy should be enabled\",\n\t\tDefaultValue: false,\n\t},\n\tQueueProcessorEnableRandomSplitByDomainID: {\n\t\tKeyName:      \"history.queueProcessorEnableRandomSplitByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"QueueProcessorEnableRandomSplitByDomainID indicates whether random queue split policy should be enabled for a domain\",\n\t\tDefaultValue: false,\n\t},\n\tQueueProcessorEnablePendingTaskSplitByDomainID: {\n\t\tKeyName:      \"history.queueProcessorEnablePendingTaskSplitByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"ueueProcessorEnablePendingTaskSplitByDomainID indicates whether pending task split policy should be enabled\",\n\t\tDefaultValue: false,\n\t},\n\tQueueProcessorEnableStuckTaskSplitByDomainID: {\n\t\tKeyName:      \"history.queueProcessorEnableStuckTaskSplitByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"QueueProcessorEnableStuckTaskSplitByDomainID indicates whether stuck task split policy should be enabled\",\n\t\tDefaultValue: false,\n\t},\n\tQueueProcessorEnablePersistQueueStates: {\n\t\tKeyName:      \"history.queueProcessorEnablePersistQueueStates\",\n\t\tDescription:  \"QueueProcessorEnablePersistQueueStates indicates whether processing queue states should be persisted\",\n\t\tDefaultValue: true,\n\t},\n\tQueueProcessorEnableLoadQueueStates: {\n\t\tKeyName:      \"history.queueProcessorEnableLoadQueueStates\",\n\t\tDescription:  \"QueueProcessorEnableLoadQueueStates indicates whether processing queue states should be loaded\",\n\t\tDefaultValue: true,\n\t},\n\tQueueProcessorEnableGracefulSyncShutdown: {\n\t\tKeyName:      \"history.queueProcessorEnableGracefulSyncShutdown\",\n\t\tDescription:  \"QueueProcessorEnableGracefulSyncShutdown indicates whether processing queue should be shutdown gracefully & synchronously\",\n\t\tDefaultValue: false,\n\t},\n\tTransferProcessorEnableValidator: {\n\t\tKeyName:      \"history.transferProcessorEnableValidator\",\n\t\tDescription:  \"TransferProcessorEnableValidator is whether validator should be enabled for transferQueueProcessor\",\n\t\tDefaultValue: false,\n\t},\n\tTaskSchedulerEnableRateLimiter: {\n\t\tKeyName:      \"history.taskSchedulerEnableRateLimiter\",\n\t\tDescription:  \"TaskSchedulerEnableRateLimiter indicates whether the task scheduler rate limiter is enabled\",\n\t\tDefaultValue: false,\n\t},\n\tTaskSchedulerEnableRateLimiterShadowMode: {\n\t\tKeyName:      \"history.taskSchedulerEnableRateLimiterShadowMode\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TaskSchedulerEnableRateLimiterShadowMode indicates whether the task scheduler rate limiter is in shadow mode\",\n\t\tDefaultValue: true,\n\t},\n\tTaskSchedulerEnableMigration: {\n\t\tKeyName:      \"history.taskSchedulerEnableMigration\",\n\t\tDescription:  \"DEPRECATED: TaskSchedulerEnableMigration indicates whether the task scheduler migration is enabled\",\n\t\tDefaultValue: false,\n\t},\n\tEnableAdminProtection: {\n\t\tKeyName:      \"history.enableAdminProtection\",\n\t\tDescription:  \"EnableAdminProtection is whether to enable admin checking\",\n\t\tDefaultValue: false,\n\t},\n\tEnableParentClosePolicy: {\n\t\tKeyName:      \"history.enableParentClosePolicy\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableParentClosePolicy is whether to  ParentClosePolicy\",\n\t\tDefaultValue: true,\n\t},\n\tEnableDropStuckTaskByDomainID: {\n\t\tKeyName:      \"history.DropStuckTaskByDomain\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"EnableDropStuckTaskByDomainID is whether stuck timer/transfer task should be dropped for a domain\",\n\t\tDefaultValue: false,\n\t},\n\tEnableConsistentQuery: {\n\t\tKeyName:      \"history.EnableConsistentQuery\",\n\t\tDescription:  \"EnableConsistentQuery indicates if consistent query is enabled for the cluster\",\n\t\tDefaultValue: true,\n\t},\n\tEnableConsistentQueryByDomain: {\n\t\tKeyName:      \"history.EnableConsistentQueryByDomain\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableConsistentQueryByDomain indicates if consistent query is enabled for a domain\",\n\t\tDefaultValue: false,\n\t},\n\tEnableContextHeaderInVisibility: {\n\t\tKeyName:      \"history.enableContextHeaderInVisibility\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableContextHeaderInVisibility is key for enable context header in visibility\",\n\t\tDefaultValue: false,\n\t},\n\tEnableCrossClusterOperationsForDomain: {\n\t\tKeyName:      \"history.enableCrossClusterOperations\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableCrossClusterOperationsForDomain indicates if cross cluster operations can be scheduled for a domain\",\n\t\tDefaultValue: false,\n\t},\n\tEnableHistoryCorruptionCheck: {\n\t\tKeyName:      \"history.enableHistoryCorruptionCheck\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableHistoryCorruptionCheck enables additional sanity check for corrupted history. This allows early catches of DB corruptions but potiantally increased latency.\",\n\t\tDefaultValue: false,\n\t},\n\tEnableActivityLocalDispatchByDomain: {\n\t\tKeyName:      \"history.enableActivityLocalDispatchByDomain\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableActivityLocalDispatchByDomain is allows worker to dispatch activity tasks through local tunnel after decisions are made. This is an performance optimization to skip activity scheduling efforts\",\n\t\tDefaultValue: true,\n\t},\n\tHistoryEnableTaskInfoLogByDomainID: {\n\t\tKeyName:      \"history.enableTaskInfoLogByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"HistoryEnableTaskInfoLogByDomainID is enables info level logs for decision/activity task based on the request domainID\",\n\t\tDefaultValue: false,\n\t},\n\tEnableReplicationTaskGeneration: {\n\t\tKeyName:      \"history.enableReplicationTaskGeneration\",\n\t\tFilters:      []Filter{DomainID, WorkflowID},\n\t\tDescription:  \"EnableReplicationTaskGeneration is the flag to control replication generation\",\n\t\tDefaultValue: true,\n\t},\n\tUseNewInitialFailoverVersion: {\n\t\tKeyName:      \"history.useNewInitialFailoverVersion\",\n\t\tDescription:  \"use the minInitialFailover version\",\n\t\tDefaultValue: false,\n\t},\n\tAllowArchivingIncompleteHistory: {\n\t\tKeyName:      \"worker.AllowArchivingIncompleteHistory\",\n\t\tDescription:  \"AllowArchivingIncompleteHistory will continue on when seeing some error like history mutated(usually caused by database consistency issues)\",\n\t\tDefaultValue: false,\n\t},\n\tEnableCleaningOrphanTaskInTasklistScavenger: {\n\t\tKeyName:      \"worker.enableCleaningOrphanTaskInTasklistScavenger\",\n\t\tDescription:  \"EnableCleaningOrphanTaskInTasklistScavenger indicates if enabling the scanner to clean up orphan tasks\",\n\t\tDefaultValue: false,\n\t},\n\tTaskListScannerEnabled: {\n\t\tKeyName:      \"worker.taskListScannerEnabled\",\n\t\tDescription:  \"TaskListScannerEnabled indicates if task list scanner should be started as part of worker.Scanner\",\n\t\tDefaultValue: true,\n\t},\n\tHistoryScannerEnabled: {\n\t\tKeyName:      \"worker.historyScannerEnabled\",\n\t\tDescription:  \"HistoryScannerEnabled indicates if history scanner should be started as part of worker.Scanner\",\n\t\tDefaultValue: false,\n\t},\n\tConcreteExecutionsScannerEnabled: {\n\t\tKeyName:      \"worker.executionsScannerEnabled\",\n\t\tDescription:  \"ConcreteExecutionsScannerEnabled indicates if executions scanner should be started as part of worker.Scanner\",\n\t\tDefaultValue: false,\n\t},\n\tConcreteExecutionsScannerInvariantCollectionMutableState: {\n\t\tKeyName:      \"worker.executionsScannerInvariantCollectionMutableState\",\n\t\tDescription:  \"ConcreteExecutionsScannerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tConcreteExecutionsScannerInvariantCollectionHistory: {\n\t\tKeyName:      \"worker.executionsScannerInvariantCollectionHistory\",\n\t\tDescription:  \"ConcreteExecutionsScannerInvariantCollectionHistory indicates if history invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tConcreteExecutionsScannerInvariantCollectionStale: {\n\t\tKeyName:      \"worker.executionsScannerInvariantCollectionStale\",\n\t\tDescription:  \"ConcreteExecutionsScannerInvariantCollectionStale indicates if the stale-workflow invariant should be run\",\n\t\tDefaultValue: false, // may be enabled after further verification, but for now it's a bit too risky to enable by default\n\t},\n\tConcreteExecutionsFixerInvariantCollectionMutableState: {\n\t\tKeyName:      \"worker.executionsFixerInvariantCollectionMutableState\",\n\t\tDescription:  \"ConcreteExecutionsFixerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tConcreteExecutionsFixerInvariantCollectionHistory: {\n\t\tKeyName:      \"worker.executionsFixerInvariantCollectionHistory\",\n\t\tDescription:  \"ConcreteExecutionsFixerInvariantCollectionHistory indicates if history invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tConcreteExecutionsFixerInvariantCollectionStale: {\n\t\tKeyName:      \"worker.executionsFixerInvariantCollectionStale\",\n\t\tDescription:  \"ConcreteExecutionsFixerInvariantCollectionStale indicates if the stale-workflow invariant should be run\",\n\t\tDefaultValue: false, // may be enabled after further verification, but for now it's a bit too risky to enable by default\n\t},\n\tCurrentExecutionsScannerEnabled: {\n\t\tKeyName:      \"worker.currentExecutionsScannerEnabled\",\n\t\tDescription:  \"CurrentExecutionsScannerEnabled indicates if current executions scanner should be started as part of worker.Scanner\",\n\t\tDefaultValue: false,\n\t},\n\tCurrentExecutionsScannerInvariantCollectionHistory: {\n\t\tKeyName:      \"worker.currentExecutionsScannerInvariantCollectionHistory\",\n\t\tDescription:  \"CurrentExecutionsScannerInvariantCollectionHistory indicates if history invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tCurrentExecutionsScannerInvariantCollectionMutableState: {\n\t\tKeyName:      \"worker.currentExecutionsInvariantCollectionMutableState\",\n\t\tDescription:  \"CurrentExecutionsScannerInvariantCollectionMutableState indicates if mutable state invariant checks should be run\",\n\t\tDefaultValue: true,\n\t},\n\tEnableBatcher: {\n\t\tKeyName:      \"worker.enableBatcher\",\n\t\tDescription:  \"EnableBatcher is decides whether start batcher in our worker\",\n\t\tDefaultValue: true,\n\t},\n\tEnableScheduler: {\n\t\tKeyName:      \"worker.enableScheduler\",\n\t\tDescription:  \"EnableScheduler decides whether to start the scheduler worker for cron-based scheduling\",\n\t\tDefaultValue: false,\n\t},\n\tEnableParentClosePolicyWorker: {\n\t\tKeyName:      \"system.enableParentClosePolicyWorker\",\n\t\tDescription:  \"EnableParentClosePolicyWorker decides whether or not enable system workers for processing parent close policy task\",\n\t\tDefaultValue: true,\n\t},\n\tEnableESAnalyzer: {\n\t\tKeyName:      \"system.enableESAnalyzer\",\n\t\tDescription:  \"EnableESAnalyzer decides whether to enable system workers for processing ElasticSearch Analyzer\",\n\t\tDefaultValue: false,\n\t},\n\tEnableAsyncWorkflowConsumption: {\n\t\tKeyName:      \"worker.enableAsyncWorkflowConsumption\",\n\t\tDescription:  \"EnableAsyncWorkflowConsumption decides whether to enable async workflows\",\n\t\tDefaultValue: true,\n\t},\n\tEnableStickyQuery: {\n\t\tKeyName:      \"system.enableStickyQuery\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableStickyQuery indicates if sticky query should be enabled per domain\",\n\t\tDefaultValue: true,\n\t},\n\tEnableFailoverManager: {\n\t\tKeyName:      \"system.enableFailoverManager\",\n\t\tDescription:  \"EnableFailoverManager indicates if failover manager is enabled\",\n\t\tDefaultValue: true,\n\t},\n\tConcreteExecutionFixerDomainAllow: {\n\t\tKeyName:      \"worker.concreteExecutionFixerDomainAllow\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ConcreteExecutionFixerDomainAllow is which domains are allowed to be fixed by concrete fixer workflow\",\n\t\tDefaultValue: false,\n\t},\n\tCurrentExecutionFixerDomainAllow: {\n\t\tKeyName:      \"worker.currentExecutionFixerDomainAllow\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"CurrentExecutionFixerDomainAllow is which domains are allowed to be fixed by current fixer workflow\",\n\t\tDefaultValue: false,\n\t},\n\tTimersScannerEnabled: {\n\t\tKeyName:      \"worker.timersScannerEnabled\",\n\t\tDescription:  \"TimersScannerEnabled is if timers scanner should be started as part of worker.Scanner\",\n\t\tDefaultValue: false,\n\t},\n\tTimersFixerEnabled: {\n\t\tKeyName:      \"worker.timersFixerEnabled\",\n\t\tDescription:  \"TimersFixerEnabled is if timers fixer should be started as part of worker.Scanner\",\n\t\tDefaultValue: false,\n\t},\n\tTimersFixerDomainAllow: {\n\t\tKeyName:      \"worker.timersFixerDomainAllow\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"TimersFixerDomainAllow is which domains are allowed to be fixed by timer fixer workflow\",\n\t\tDefaultValue: false,\n\t},\n\tConcreteExecutionFixerEnabled: {\n\t\tKeyName:      \"worker.concreteExecutionFixerEnabled\",\n\t\tDescription:  \"ConcreteExecutionFixerEnabled is if concrete execution fixer workflow is enabled\",\n\t\tDefaultValue: false,\n\t},\n\tCurrentExecutionFixerEnabled: {\n\t\tKeyName:      \"worker.currentExecutionFixerEnabled\",\n\t\tDescription:  \"CurrentExecutionFixerEnabled is if current execution fixer workflow is enabled\",\n\t\tDefaultValue: false,\n\t},\n\tEnableAuthorization: {\n\t\tKeyName:      \"system.enableAuthorization\",\n\t\tDescription:  \"EnableAuthorization is the key to enable authorization for a domain, only for extension binary:\",\n\t\tDefaultValue: false,\n\t},\n\tEnableTasklistIsolation: {\n\t\tKeyName:      \"system.enableTasklistIsolation\",\n\t\tDescription:  \"EnableTasklistIsolation is a feature to enable isolation-groups for a domain. Should not be enabled without a deep understanding of this feature\",\n\t\tDefaultValue: false,\n\t},\n\tEnableServiceAuthorization: {\n\t\tKeyName:      \"system.enableServiceAuthorization\",\n\t\tDescription:  \"EnableServiceAuthorization is the key to enable authorization for a service, only for extension binary:\",\n\t\tDefaultValue: false,\n\t},\n\tEnableServiceAuthorizationLogOnly: {\n\t\tKeyName:      \"system.enableServiceAuthorizationLogOnly\",\n\t\tDescription:  \"EnableServiceAuthorizationLogOnly is the key to enable authorization logging for a service, only for extension binary:\",\n\t\tDefaultValue: false,\n\t},\n\tESAnalyzerPause: {\n\t\tKeyName:      \"worker.ESAnalyzerPause\",\n\t\tDescription:  \"ESAnalyzerPause defines if we want to dynamically pause the analyzer workflow\",\n\t\tDefaultValue: false,\n\t},\n\tEnableArchivalCompression: {\n\t\tKeyName:      \"worker.EnableArchivalCompression\",\n\t\tDescription:  \"EnableArchivalCompression indicates whether blobs are compressed before they are archived\",\n\t\tDefaultValue: false,\n\t},\n\tESAnalyzerEnableAvgDurationBasedChecks: {\n\t\tKeyName:      \"worker.ESAnalyzerEnableAvgDurationBasedChecks\",\n\t\tDescription:  \"ESAnalyzerEnableAvgDurationBasedChecks controls if we want to enable avg duration based task refreshes\",\n\t\tDefaultValue: false,\n\t},\n\tLockdown: {\n\t\tKeyName:      \"system.Lockdown\",\n\t\tDescription:  \"Lockdown defines if we want to allow failovers of domains to this cluster\",\n\t\tDefaultValue: false,\n\t},\n\tEnablePendingActivityValidation: {\n\t\tKeyName:      \"limit.pendingActivityCount.enabled\",\n\t\tDescription:  \"Enables pending activity count limiting/validation\",\n\t\tDefaultValue: false,\n\t},\n\tEnableCassandraAllConsistencyLevelDelete: {\n\t\tKeyName:      \"system.enableCassandraAllConsistencyLevelDelete\",\n\t\tDescription:  \"Uses all consistency level for Cassandra delete operations\",\n\t\tDefaultValue: false,\n\t},\n\tEnableShardIDMetrics: {\n\t\tKeyName:      \"system.enableShardIDMetrics\",\n\t\tDescription:  \"Enable shardId metrics in persistence client\",\n\t\tDefaultValue: true,\n\t},\n\tEnableTimerDebugLogByDomainID: {\n\t\tKeyName:      \"history.enableTimerDebugLogByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"Enable log for debugging timer task issue by domain\",\n\t\tDefaultValue: false,\n\t},\n\tEnableRetryForChecksumFailure: {\n\t\tKeyName:      \"history.enableMutableStateChecksumFailureRetry\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableRetryForChecksumFailure enables retry if mutable state checksum verification fails\",\n\t\tDefaultValue: false,\n\t},\n\tEnableStrongIdempotency: {\n\t\tKeyName:      \"history.enableStrongIdempotency\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableStrongIdempotency enables strong idempotency for APIs\",\n\t\tDefaultValue: false,\n\t},\n\tEnableStrongIdempotencySanityCheck: {\n\t\tKeyName:      \"history.enableStrongIdempotencySanityCheck\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnableStrongIdempotencySanityCheck enables sanity check for strong idempotency\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingEnableClientAutoConfig: {\n\t\tKeyName:      \"matching.enableClientAutoConfig\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingEnableClientAutoConfig is to enable auto config on worker side\",\n\t\tDefaultValue: false,\n\t},\n\tEnablePartitionIsolationGroupAssignment: {\n\t\tKeyName:      \"matching.enablePartitionIsolationGroupAssignment\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"EnablePartitionIsolationGroupAssignment enables assigning isolation groups to individual TaskList partitions\",\n\t\tDefaultValue: false,\n\t},\n\tEnableNoSQLHistoryTaskDualWriteMode: {\n\t\tKeyName:      \"history.enableNoSQLHistoryTaskDualWrite\",\n\t\tDescription:  \"EnableHistoryTaskDualWrite is to enable dual write of history events\",\n\t\tDefaultValue: false,\n\t},\n\tReadNoSQLHistoryTaskFromDataBlob: {\n\t\tKeyName:      \"history.readNoSQLHistoryTaskFromDataBlob\",\n\t\tDescription:  \"ReadNoSQLHistoryTaskFromDataBlob is to read history tasks from data blob\",\n\t\tDefaultValue: false,\n\t},\n\tReadNoSQLShardFromDataBlob: {\n\t\tKeyName:      \"history.readNoSQLShardFromDataBlob\",\n\t\tDescription:  \"ReadNoSQLShardFromDataBlob is to read shards from data blob\",\n\t\tDefaultValue: true,\n\t},\n\tEnableSizeBasedHistoryExecutionCache: {\n\t\tKeyName:      \"history.enableSizeBasedHistoryExecutionCache\",\n\t\tDescription:  \"EnableSizeBasedHistoryExecutionCache is to enable size based history execution cache\",\n\t\tDefaultValue: false,\n\t},\n\tEnableSizeBasedHistoryEventCache: {\n\t\tKeyName:      \"history.enableSizeBasedHistoryEventCache\",\n\t\tDescription:  \"EnableSizeBasedHistoryEventCache is to enable size based history event cache\",\n\t\tDefaultValue: false,\n\t},\n\tDisableTransferFailoverQueue: {\n\t\tKeyName:      \"history.disableTransferFailoverQueue\",\n\t\tDescription:  \"DisableTransferFailoverQueue is to disable transfer failover queue\",\n\t\tDefaultValue: false,\n\t},\n\tDisableTimerFailoverQueue: {\n\t\tKeyName:      \"history.disableTimerFailoverQueue\",\n\t\tDescription:  \"DisableTimerFailoverQueue is to disable timer failover queue\",\n\t\tDefaultValue: false,\n\t},\n\tEnableTransferQueueV2: {\n\t\tKeyName:      \"history.enableTransferQueueV2\",\n\t\tDescription:  \"EnableTransferQueueV2 is to enable transfer queue v2\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDefaultValue: false,\n\t},\n\tEnableTimerQueueV2: {\n\t\tKeyName:      \"history.enableTimerQueueV2\",\n\t\tDescription:  \"EnableTimerQueueV2 is to enable timer queue v2\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDefaultValue: false,\n\t},\n\tEnableTransferQueueV2PendingTaskCountAlert: {\n\t\tKeyName:      \"history.enableTransferQueueV2PendingTaskCountAlert\",\n\t\tDescription:  \"EnableTransferQueueV2PendingTaskCountAlert is to enable transfer queue v2 pending task count alert\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDefaultValue: false,\n\t},\n\tEnableTimerQueueV2PendingTaskCountAlert: {\n\t\tKeyName:      \"history.enableTimerQueueV2PendingTaskCountAlert\",\n\t\tDescription:  \"EnableTimerQueueV2PendingTaskCountAlert is to enable timer queue v2 pending task count alert\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDefaultValue: false,\n\t},\n\tEnableActiveClusterSelectionPolicyInStartWorkflow: {\n\t\tKeyName:      \"frontend.enableActiveClusterSelectionPolicyInStartWorkflow\",\n\t\tDescription:  \"EnableActiveClusterSelectionPolicyInStartWorkflow is to enable active cluster selection policy in start workflow requests for a domain\",\n\t\tDefaultValue: false,\n\t\tFilters:      []Filter{DomainName},\n\t},\n\tEnforceDecisionTaskAttempts: {\n\t\tKeyName:      \"history.enforceDecisionTaskAttempts\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"EnforceDecisionTaskAttempts is the key for enforcing decision retry attempts limit in case of timeouts\",\n\t\tDefaultValue: false,\n\t},\n\tMatchingExcludeShortLivedTaskListsFromShardManager: {\n\t\tKeyName:      \"matching.excludeShortLivedTaskListsFromShardManager\",\n\t\tDescription:  \"MatchingExcludeShortLivedTaskListsFromShardManager excludes short-lived task lists (e.g. bits task lists and sticky task lists) from the shard manager\",\n\t\tDefaultValue: true,\n\t},\n\tEnableHierarchicalWeightedRoundRobinTaskScheduler: {\n\t\tKeyName:      \"history.enableHierarchicalWeightedRoundRobinTaskScheduler\",\n\t\tDescription:  \"EnableHierarchicalWeightedRoundRobinTaskScheduler is to enable hierarchical weighted round robin task scheduler\",\n\t\tDefaultValue: false,\n\t},\n\tEnableTaskListAwareTaskSchedulerByDomain: {\n\t\tKeyName:      \"history.enableTaskListAwareTaskSchedulerByDomain\",\n\t\tDescription:  \"EnableTaskListAwareTaskSchedulerByDomain is to enable task list aware task scheduler by domain\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDefaultValue: false,\n\t},\n}\n\nvar FloatKeys = map[FloatKey]DynamicFloat{\n\tTestGetFloat64PropertyKey: {\n\t\tKeyName:      \"testGetFloat64PropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetFloat64PropertyFilteredByShardIDKey: {\n\t\tKeyName:      \"testGetFloat64PropertyFilteredByShardIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t\tFilters:      nil,\n\t},\n\tPersistenceErrorInjectionRate: {\n\t\tKeyName:      \"system.persistenceErrorInjectionRate\",\n\t\tDescription:  \"PersistenceErrorInjectionRate is rate for injecting random error in persistence\",\n\t\tDefaultValue: 0,\n\t},\n\tAdminErrorInjectionRate: {\n\t\tKeyName:      \"admin.errorInjectionRate\",\n\t\tDescription:  \"dminErrorInjectionRate is the rate for injecting random error in admin client\",\n\t\tDefaultValue: 0,\n\t},\n\tDomainFailoverRefreshTimerJitterCoefficient: {\n\t\tKeyName:      \"frontend.domainFailoverRefreshTimerJitterCoefficient\",\n\t\tDescription:  \"DomainFailoverRefreshTimerJitterCoefficient is the jitter for domain failover refresh timer jitter\",\n\t\tDefaultValue: 0.1,\n\t},\n\tFrontendErrorInjectionRate: {\n\t\tKeyName:      \"frontend.errorInjectionRate\",\n\t\tDescription:  \"FrontendErrorInjectionRate is rate for injecting random error in frontend client\",\n\t\tDefaultValue: 0,\n\t},\n\tMatchingErrorInjectionRate: {\n\t\tKeyName:      \"matching.errorInjectionRate\",\n\t\tDescription:  \"MatchingErrorInjectionRate is rate for injecting random error in matching client\",\n\t\tDefaultValue: 0,\n\t},\n\tQueueProcessorRandomSplitProbability: {\n\t\tKeyName:      \"history.queueProcessorRandomSplitProbability\",\n\t\tDescription:  \"QueueProcessorRandomSplitProbability is the probability for a domain to be split to a new processing queue\",\n\t\tDefaultValue: 0.01,\n\t},\n\tQueueProcessorPollBackoffIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.queueProcessorPollBackoffIntervalJitterCoefficient\",\n\t\tDescription:  \"QueueProcessorPollBackoffIntervalJitterCoefficient is backoff interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTimerProcessorUpdateAckIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.timerProcessorUpdateAckIntervalJitterCoefficient\",\n\t\tDescription:  \"TimerProcessorUpdateAckIntervalJitterCoefficient is the update interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTimerProcessorMaxPollIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.timerProcessorMaxPollIntervalJitterCoefficient\",\n\t\tDescription:  \"TimerProcessorMaxPollIntervalJitterCoefficient is the max poll interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTimerProcessorSplitQueueIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.timerProcessorSplitQueueIntervalJitterCoefficient\",\n\t\tDescription:  \"TimerProcessorSplitQueueIntervalJitterCoefficient is the split processing queue interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTransferProcessorMaxPollIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.transferProcessorMaxPollIntervalJitterCoefficient\",\n\t\tDescription:  \"TransferProcessorMaxPollIntervalJitterCoefficient is the max poll interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTransferProcessorSplitQueueIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.transferProcessorSplitQueueIntervalJitterCoefficient\",\n\t\tDescription:  \"TransferProcessorSplitQueueIntervalJitterCoefficient is the split processing queue interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tTransferProcessorUpdateAckIntervalJitterCoefficient: {\n\t\tKeyName:      \"history.transferProcessorUpdateAckIntervalJitterCoefficient\",\n\t\tDescription:  \"TransferProcessorUpdateAckIntervalJitterCoefficient is the update interval jitter coefficient\",\n\t\tDefaultValue: 0.15,\n\t},\n\tReplicationTaskProcessorCleanupJitterCoefficient: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorCleanupJitterCoefficient\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorCleanupJitterCoefficient is the jitter for cleanup timer\",\n\t\tDefaultValue: 0.15,\n\t},\n\tReplicationTaskProcessorStartWaitJitterCoefficient: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorStartWaitJitterCoefficient\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorStartWaitJitterCoefficient is the jitter for batch start wait timer\",\n\t\tDefaultValue: 0.9,\n\t},\n\tReplicationTaskProcessorHostQPS: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorHostQPS\",\n\t\tDescription:  \"ReplicationTaskProcessorHostQPS is the qps of task processing rate limiter on host level\",\n\t\tDefaultValue: 1500,\n\t},\n\tReplicationTaskProcessorShardQPS: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorShardQPS\",\n\t\tDescription:  \"ReplicationTaskProcessorShardQPS is the qps of task processing rate limiter on shard level\",\n\t\tDefaultValue: 5,\n\t},\n\tReplicationTaskGenerationQPS: {\n\t\tKeyName:      \"history.ReplicationTaskGenerationQPS\",\n\t\tDescription:  \"ReplicationTaskGenerationQPS is the wait time between each replication task generation qps\",\n\t\tDefaultValue: 100,\n\t},\n\tMutableStateChecksumInvalidateBefore: {\n\t\tKeyName:      \"history.mutableStateChecksumInvalidateBefore\",\n\t\tDescription:  \"MutableStateChecksumInvalidateBefore is the epoch timestamp before which all checksums are to be discarded\",\n\t\tDefaultValue: 0,\n\t},\n\tNotifyFailoverMarkerTimerJitterCoefficient: {\n\t\tKeyName:      \"history.NotifyFailoverMarkerTimerJitterCoefficient\",\n\t\tDescription:  \"NotifyFailoverMarkerTimerJitterCoefficient is the jitter for failover marker notifier timer\",\n\t\tDefaultValue: 0.15,\n\t},\n\tHistoryErrorInjectionRate: {\n\t\tKeyName:      \"history.errorInjectionRate\",\n\t\tDescription:  \"HistoryErrorInjectionRate is rate for injecting random error in history client\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicationBudgetManagerSoftCapThreshold: {\n\t\tKeyName:      \"history.replicationBudgetManagerSoftCapThreshold\",\n\t\tDescription:  \"ReplicationBudgetManagerSoftCapThreshold is the soft cap threshold for the replication budget manager cache (0.0 to 1.0)\",\n\t\tDefaultValue: 1.0,\n\t},\n\tReplicationTaskFetcherTimerJitterCoefficient: {\n\t\tKeyName:      \"history.ReplicationTaskFetcherTimerJitterCoefficient\",\n\t\tDescription:  \"ReplicationTaskFetcherTimerJitterCoefficient is the jitter for fetcher timer\",\n\t\tDefaultValue: 0.15,\n\t},\n\tWorkerDeterministicConstructionCheckProbability: {\n\t\tKeyName:      \"worker.DeterministicConstructionCheckProbability\",\n\t\tDescription:  \"WorkerDeterministicConstructionCheckProbability controls the probability of running a deterministic construction check for any given archival\",\n\t\tDefaultValue: 0.002,\n\t},\n\tWorkerBlobIntegrityCheckProbability: {\n\t\tKeyName:      \"worker.BlobIntegrityCheckProbability\",\n\t\tDescription:  \"WorkerBlobIntegrityCheckProbability controls the probability of running an integrity check for any given archival\",\n\t\tDefaultValue: 0.002,\n\t},\n\tHistoryGlobalRatelimiterNewDataWeight: {\n\t\tKeyName:      \"history.globalRatelimiterNewDataWeight\",\n\t\tDescription:  \"HistoryGlobalRatelimiterNewDataWeight defines how much weight to give each host's newest data, per update.  Must be between 0 and 1, higher values match new values more closely after a single update\",\n\t\tDefaultValue: 0.5,\n\t},\n\tMatchingPartitionDownscaleFactor: {\n\t\tKeyName:      \"matching.partitionDownscaleFactor\",\n\t\tDescription:  \"MatchingPartitionDownscaleFactor introduces hysteresis to prevent oscillation by setting a lower QPS threshold for downscaling, ensuring partitions are only removed when the load decreases significantly below the capacity of fewer partitions.\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDefaultValue: 0.75,\n\t},\n\tMatchingOverrideTaskListRPS: {\n\t\tKeyName:      \"matching.overrideTaskListRps\",\n\t\tDescription:  \"MatchingOverrideTaskListRPS is the RPS override for a specific TaskList. When set to a non-zero value, this overrides the RPS value that pollers specify. By default (0), the pollers' specified RPS is respected.\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDefaultValue: 0,\n\t},\n\tShardDistributorErrorInjectionRate: {\n\t\tKeyName:      \"sharddistributor.errorInjectionRate\",\n\t\tDescription:  \"ShardDistributorInjectionRate is rate for injecting random error in shard distributor client\",\n\t\tDefaultValue: 0,\n\t},\n\tShardDistributorExecutorErrorInjectionRate: {\n\t\tKeyName:      \"sharddistributorexecutor.errorInjectionRate\",\n\t\tDescription:  \"ShardDistributorExecutorInjectionRate is rate for injecting random error in shard distributor executor client\",\n\t\tDefaultValue: 0,\n\t},\n\n\tShardDistributorLoadBalancingNaiveMaxDeviation: {\n\t\tKeyName:      \"shardDistributor.loadBalancingNaive.maxDeviation\",\n\t\tDescription:  \"ShardDistributorLoadBalancingNaiveMaxDeviation is max deviation between the coldest and hottest executors in naive load balancing mode\",\n\t\tDefaultValue: 2.0,\n\t\tFilters:      []Filter{Namespace},\n\t},\n}\n\nvar StringKeys = map[StringKey]DynamicString{\n\tTestGetStringPropertyKey: {\n\t\tKeyName:      \"testGetStringPropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: \"\",\n\t},\n\tWriteVisibilityStoreName: {\n\t\tKeyName:      \"system.writeVisibilityStoreName\",\n\t\tDescription:  \"WriteVisibilityStoreName is key for how to write to advanced visibility. The default option is es, which can be used for seamless migration from db visibility to advanced visibility, usually using with ReadVisibilityStoreName\",\n\t\tDefaultValue: \"es\",\n\t},\n\tHistoryArchivalStatus: {\n\t\tKeyName:      \"system.historyArchivalStatus\",\n\t\tDescription:  \"HistoryArchivalStatus is key for the status of history archival to override the value from static config.\",\n\t\tDefaultValue: \"enabled\",\n\t},\n\tVisibilityArchivalStatus: {\n\t\tKeyName:      \"system.visibilityArchivalStatus\",\n\t\tDescription:  \"VisibilityArchivalStatus is key for the status of visibility archival to override the value from static config.\",\n\t\tDefaultValue: \"enabled\",\n\t},\n\tDefaultEventEncoding: {\n\t\tKeyName:      \"history.defaultEventEncoding\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DefaultEventEncoding is the encoding type for history events\",\n\t\tDefaultValue: string(constants.EncodingTypeThriftRW),\n\t},\n\tAdminOperationToken: {\n\t\tKeyName:      \"history.adminOperationToken\",\n\t\tDescription:  \"AdminOperationToken is the token to pass admin checking\",\n\t\tDefaultValue: \"CadenceTeamONLY\",\n\t},\n\tESAnalyzerLimitToTypes: {\n\t\tKeyName:      \"worker.ESAnalyzerLimitToTypes\",\n\t\tDescription:  \"ESAnalyzerLimitToTypes controls if we want to limit ESAnalyzer only to some workflow types\",\n\t\tDefaultValue: \"\",\n\t},\n\tESAnalyzerLimitToDomains: {\n\t\tKeyName:      \"worker.ESAnalyzerLimitToDomains\",\n\t\tDescription:  \"ESAnalyzerLimitToDomains controls if we want to limit ESAnalyzer only to some domains\",\n\t\tDefaultValue: \"\",\n\t},\n\tESAnalyzerWorkflowDurationWarnThresholds: {\n\t\tKeyName:      \"worker.ESAnalyzerWorkflowDurationWarnThresholds\",\n\t\tDescription:  \"ESAnalyzerWorkflowDurationWarnThresholds defines the warning execution thresholds for workflow types\",\n\t\tDefaultValue: \"\",\n\t},\n\tESAnalyzerWorkflowVersionMetricDomains: {\n\t\tKeyName:      \"worker.ESAnalyzerWorkflowVersionMetricDomains\",\n\t\tDescription:  \"ESAnalyzerWorkflowDurationWarnThresholds defines the domains we want to emit wf version metrics on\",\n\t\tDefaultValue: \"\",\n\t},\n\tESAnalyzerWorkflowTypeMetricDomains: {\n\t\tKeyName:      \"worker.ESAnalyzerWorkflowTypeMetricDomains\",\n\t\tDescription:  \"ESAnalyzerWorkflowDurationWarnThresholds defines the domains we want to emit wf version metrics on\",\n\t\tDefaultValue: \"\",\n\t},\n\tFrontendGlobalRatelimiterMode: {\n\t\tKeyName:      \"frontend.globalRatelimiterMode\",\n\t\tDescription:  \"FrontendGlobalRatelimiterMode defines which mode a global key should be in, per key, to make gradual changes to ratelimiter algorithms\",\n\t\tDefaultValue: \"disabled\",\n\t\tFilters:      []Filter{RatelimitKey},\n\t},\n\tEnableAuthorizationV2: {\n\t\tKeyName:      \"system.enableAuthorizationV2\",\n\t\tDescription:  \"EnableAuthorizationV2 is the key to enable authorization v2 for a domain, only for extension binary:\",\n\t\tDefaultValue: \"disabled\", // available options: \"disabled\",\"shadow\",\"enabled\"\n\t},\n\tTasklistLoadBalancerStrategy: {\n\t\tKeyName:      \"system.tasklistLoadBalancerStrategy\",\n\t\tDescription:  \"TasklistLoadBalancerStrategy is the key for tasklist load balancer strategy\",\n\t\tDefaultValue: \"weighted\", // available options: \"random, round-robin, weighted\"\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t},\n\tEnableAdminAuthorization: {\n\t\tKeyName:      \"system.enableAdminAuthorization\",\n\t\tDescription:  \"EnableAdminAuthorization is the key to enable authorization for admin operations, only for extension of authorizer implementation\",\n\t\tDefaultValue: \"disabled\", // available options: \"disabled\",\"shadow\",\"enabled\"\n\t},\n\tReadVisibilityStoreName: {\n\t\tKeyName:      \"system.readVisibilityStoreName\",\n\t\tDescription:  \"ReadVisibilityStoreName is key to identify which store to read visibility data from\",\n\t\tDefaultValue: \"es\",\n\t\tFilters:      []Filter{DomainName},\n\t},\n\tMatchingShardDistributionMode: {\n\t\tKeyName:      \"matching.shardDistributionMode\",\n\t\tDescription:  \"MatchingShardDistributionMode defines which shard distribution mode should be used\",\n\t\tDefaultValue: \"hash_ring\",\n\t},\n\tSerializationEncoding: {\n\t\tKeyName:      \"history.serializationEncoding\",\n\t\tDescription:  \"SerializationEncoding is the encoding type for blobs\",\n\t\tDefaultValue: string(constants.EncodingTypeThriftRW),\n\t},\n\tShardDistributorMigrationMode: {\n\t\tKeyName:      \"shardDistributor.migrationMode\",\n\t\tDescription:  \"ShardDistributorMigrationMode is the mode the at represent the state of the migration to rely on shard distributor for the sharding mechanism\",\n\t\tDefaultValue: \"onboarded\",\n\t\tFilters:      []Filter{Namespace},\n\t},\n\tShardDistributorLoadBalancingMode: {\n\t\tKeyName:      \"shardDistributor.loadBalancingMode\",\n\t\tDescription:  \"ShardDistributorLoadBalancingMode is the load balancing mode for the shard distributor. Depending on the mode, the shard distributor will use different ways to distribute the shards\",\n\t\tDefaultValue: \"naive\",\n\t},\n}\n\nvar DurationKeys = map[DurationKey]DynamicDuration{\n\tTestGetDurationPropertyKey: {\n\t\tKeyName:      \"testGetDurationPropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetDurationPropertyFilteredByDomainKey: {\n\t\tKeyName:      \"testGetDurationPropertyFilteredByDomainKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetDurationPropertyFilteredByTaskListInfoKey: {\n\t\tKeyName:      \"testGetDurationPropertyFilteredByTaskListInfoKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t},\n\tTestGetDurationPropertyFilteredByWorkflowTypeKey: {\n\t\tKeyName:      \"testGetDurationPropertyFilteredByWorkflowTypeKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t\tFilters:      nil,\n\t},\n\tTestGetDurationPropertyFilteredByDomainIDKey: {\n\t\tKeyName:      \"testGetDurationPropertyFilteredByDomainIDKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t\tFilters:      nil,\n\t},\n\tTestGetDurationPropertyFilteredByShardID: {\n\t\tKeyName:      \"testGetDurationPropertyFilteredByShardID\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: 0,\n\t\tFilters:      nil,\n\t},\n\tFrontendShutdownDrainDuration: {\n\t\tKeyName:      \"frontend.shutdownDrainDuration\",\n\t\tDescription:  \"FrontendShutdownDrainDuration is the duration of traffic drain during shutdown\",\n\t\tDefaultValue: 0,\n\t},\n\tFrontendWarmupDuration: {\n\t\tKeyName:      \"frontend.warmupDuration\",\n\t\tDescription:  \"FrontendWarmupDuration is the duration before a frontend host reports its status as healthy\",\n\t\tDefaultValue: 30 * time.Second,\n\t},\n\tFrontendFailoverCoolDown: {\n\t\tKeyName:      \"frontend.failoverCoolDown\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendFailoverCoolDown is duration between two domain failvoers\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tDomainFailoverRefreshInterval: {\n\t\tKeyName:      \"frontend.domainFailoverRefreshInterval\",\n\t\tDescription:  \"DomainFailoverRefreshInterval is the domain failover refresh timer\",\n\t\tDefaultValue: time.Second * 10,\n\t},\n\tGlobalRatelimiterUpdateInterval: {\n\t\tKeyName:      \"frontend.globalRatelimiterUpdateInterval\",\n\t\tDescription:  \"GlobalRatelimiterUpdateInterval defines how often each global ratelimiter collection submits load information, and the expected update rate in aggregators (used to determine when hosts are lost)\",\n\t\tDefaultValue: 3 * time.Second,\n\t},\n\tFrontendMaxWorkerPollDelay: {\n\t\tKeyName:      \"frontend.maxWorkerPollDelay\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"FrontendMaxWorkerPollDelay is the maximum duration a worker poll request will wait for a rate limit token before being rejected\",\n\t\tDefaultValue: 0,\n\t},\n\tMatchingLongPollExpirationInterval: {\n\t\tKeyName:      \"matching.longPollExpirationInterval\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingLongPollExpirationInterval is the long poll expiration interval in the matching service\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingUpdateAckInterval: {\n\t\tKeyName:      \"matching.updateAckInterval\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingUpdateAckInterval is the interval for update ack\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingIdleTasklistCheckInterval: {\n\t\tKeyName:      \"matching.idleTasklistCheckInterval\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIdleTasklistCheckInterval is the IdleTasklistCheckInterval\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tMaxTasklistIdleTime: {\n\t\tKeyName:      \"matching.maxTasklistIdleTime\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MaxTasklistIdleTime is the max time tasklist being idle\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tMatchingShutdownDrainDuration: {\n\t\tKeyName:      \"matching.shutdownDrainDuration\",\n\t\tDescription:  \"MatchingShutdownDrainDuration is the duration of traffic drain during shutdown\",\n\t\tDefaultValue: 0,\n\t},\n\tMatchingActivityTaskSyncMatchWaitTime: {\n\t\tKeyName:      \"matching.activityTaskSyncMatchWaitTime\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"MatchingActivityTaskSyncMatchWaitTime is the amount of time activity task will wait to be sync matched\",\n\t\tDefaultValue: time.Millisecond * 50,\n\t},\n\tMatchingPartitionUpscaleSustainedDuration: {\n\t\tKeyName:      \"matching.partitionUpscaleSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingPartitionUpscaleSustainedDuration is the sustained period to wait before upscaling the number of partitions\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingPartitionDownscaleSustainedDuration: {\n\t\tKeyName:      \"matching.partitionDownscaleSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingPartitionDownscaleSustainedDuration is the sustained period to wait before downscaling the number of partitions\",\n\t\tDefaultValue: 2 * time.Minute,\n\t},\n\tMatchingIsolationGroupUpscaleSustainedDuration: {\n\t\tKeyName:      \"matching.isolationGroupUpscaleSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIsolationGroupUpscaleSustainedDuration is the sustained period to wait before upscaling the number of partitions an isolation group is assigned to\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingIsolationGroupDownscaleSustainedDuration: {\n\t\tKeyName:      \"matching.isolationGroupDownscaleSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIsolationGroupDownscaleSustainedDuration is the sustained period to wait before downscaling the number of partitions an isolation group is assigned to\",\n\t\tDefaultValue: 2 * time.Minute,\n\t},\n\tMatchingIsolationGroupHasPollersSustainedDuration: {\n\t\tKeyName:      \"matching.isolationGroupHasPollersSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIsolationGroupHasPollersSustainedDuration is the sustained period to wait before considering an isolation group as an active and assigning partitions to it\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingIsolationGroupNoPollersSustainedDuration: {\n\t\tKeyName:      \"matching.isolationGroupNoPollersSustainedDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingIsolationGroupNoPollersSustainedDuration is the sustained period to wait before considering an isolation group as inactive and unassigning all partitions from it\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tMatchingAdaptiveScalerUpdateInterval: {\n\t\tKeyName:      \"matching.adaptiveScalerUpdateInterval\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingAdaptiveScalerUpdateInterval is the internal for adaptive scaler to update\",\n\t\tDefaultValue: time.Second * 15,\n\t},\n\tMatchingQPSTrackerInterval: {\n\t\tKeyName:      \"matching.qpsTrackerInterval\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"MatchingQPSTrackerInterval is the interval for qps tracker's loop. Changes are not reflected until service restart\",\n\t\tDefaultValue: time.Second * 10,\n\t},\n\tHistoryLongPollExpirationInterval: {\n\t\tKeyName:      \"history.longPollExpirationInterval\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"HistoryLongPollExpirationInterval is the long poll expiration interval in the history service\",\n\t\tDefaultValue: time.Second * 20, // history client: client/history/client.go set the client timeout 20s\n\t},\n\tHistoryCacheTTL: {\n\t\tKeyName:      \"history.cacheTTL\",\n\t\tDescription:  \"HistoryCacheTTL is TTL of history cache\",\n\t\tDefaultValue: time.Hour,\n\t},\n\tHistoryShutdownDrainDuration: {\n\t\tKeyName:      \"history.shutdownDrainDuration\",\n\t\tDescription:  \"HistoryShutdownDrainDuration is the duration of traffic drain during shutdown\",\n\t\tDefaultValue: 0,\n\t},\n\tEventsCacheTTL: {\n\t\tKeyName:      \"history.eventsCacheTTL\",\n\t\tDescription:  \"EventsCacheTTL is TTL of events cache\",\n\t\tDefaultValue: time.Hour,\n\t},\n\tAcquireShardInterval: {\n\t\tKeyName:      \"history.acquireShardInterval\",\n\t\tDescription:  \"AcquireShardInterval is interval that timer used to acquire shard\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tStandbyClusterDelay: {\n\t\tKeyName:      \"history.standbyClusterDelay\",\n\t\tDescription:  \"StandbyClusterDelay is the artificial delay added to standby cluster's view of active cluster's time\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tStandbyTaskMissingEventsResendDelay: {\n\t\tKeyName:      \"history.standbyTaskMissingEventsResendDelay\",\n\t\tDescription:  \"StandbyTaskMissingEventsResendDelay is the amount of time standby cluster's will wait (if events are missing)before calling remote for missing events\",\n\t\tDefaultValue: time.Minute * 15,\n\t},\n\tStandbyTaskMissingEventsDiscardDelay: {\n\t\tKeyName:      \"history.standbyTaskMissingEventsDiscardDelay\",\n\t\tDescription:  \"StandbyTaskMissingEventsDiscardDelay is the amount of time standby cluster's will wait (if events are missing)before discarding the task\",\n\t\tDefaultValue: time.Minute * 25,\n\t},\n\tActiveTaskRedispatchInterval: {\n\t\tKeyName:      \"history.activeTaskRedispatchInterval\",\n\t\tDescription:  \"ActiveTaskRedispatchInterval is the active task redispatch interval\",\n\t\tDefaultValue: time.Second * 5,\n\t},\n\tStandbyTaskRedispatchInterval: {\n\t\tKeyName:      \"history.standbyTaskRedispatchInterval\",\n\t\tDescription:  \"StandbyTaskRedispatchInterval is the standby task redispatch interval\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tStandbyTaskReReplicationContextTimeout: {\n\t\tKeyName:      \"history.standbyTaskReReplicationContextTimeout\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"StandbyTaskReReplicationContextTimeout is the context timeout for standby task re-replication\",\n\t\tDefaultValue: time.Minute * 3,\n\t},\n\tResurrectionCheckMinDelay: {\n\t\tKeyName:      \"history.resurrectionCheckMinDelay\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ResurrectionCheckMinDelay is the minimal timer processing delay before scanning history to see if there's a resurrected timer/activity\",\n\t\tDefaultValue: time.Hour * 24,\n\t},\n\tQueueProcessorSplitLookAheadDurationByDomainID: {\n\t\tKeyName:      \"history.queueProcessorSplitLookAheadDurationByDomainID\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"QueueProcessorSplitLookAheadDurationByDomainID is the look ahead duration when spliting a domain to a new processing queue\",\n\t\tDefaultValue: time.Minute * 20,\n\t},\n\tQueueProcessorPollBackoffInterval: {\n\t\tKeyName:      \"history.queueProcessorPollBackoffInterval\",\n\t\tDescription:  \"QueueProcessorPollBackoffInterval is the backoff duration when queue processor is throttled\",\n\t\tDefaultValue: time.Second * 5,\n\t},\n\tVirtualSliceForceAppendInterval: {\n\t\tKeyName:      \"history.virtualSliceForceAppendInterval\",\n\t\tDescription:  \"VirtualSliceForceAppendInterval is the duration forcing a new virtual slice to be appended to the root virtual queue instead of being merged. It has 2 benefits: First, virtual slices won't grow infinitely, task loading for that slice can complete and its scope can be shrinked. Second, when we need to unload a virtual slice to free memory, we won't unload too many tasks.\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tTimerProcessorUpdateAckInterval: {\n\t\tKeyName:      \"history.timerProcessorUpdateAckInterval\",\n\t\tDescription:  \"TimerProcessorUpdateAckInterval is update interval for timer processor\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tTimerProcessorCompleteTimerInterval: {\n\t\tKeyName:      \"history.timerProcessorCompleteTimerInterval\",\n\t\tDescription:  \"TimerProcessorCompleteTimerInterval is complete timer interval for timer processor\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tTimerProcessorFailoverMaxStartJitterInterval: {\n\t\tKeyName:      \"history.timerProcessorFailoverMaxStartJitterInterval\",\n\t\tDescription:  \"TimerProcessorFailoverMaxStartJitterInterval is the max jitter interval for starting timer failover queue processing. The actual jitter interval used will be a random duration between 0 and the max interval so that timer failover queue across different shards won't start at the same time\",\n\t\tDefaultValue: 0,\n\t},\n\tTimerProcessorMaxPollInterval: {\n\t\tKeyName:      \"history.timerProcessorMaxPollInterval\",\n\t\tDescription:  \"TimerProcessorMaxPollInterval is max poll interval for timer processor\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tTimerProcessorSplitQueueInterval: {\n\t\tKeyName:      \"history.timerProcessorSplitQueueInterval\",\n\t\tDescription:  \"TimerProcessorSplitQueueInterval is the split processing queue interval for timer processor\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tTimerProcessorArchivalTimeLimit: {\n\t\tKeyName:      \"history.timerProcessorArchivalTimeLimit\",\n\t\tDescription:  \"TimerProcessorArchivalTimeLimit is the upper time limit for inline history archival\",\n\t\tDefaultValue: time.Second * 2,\n\t},\n\tTimerProcessorMaxTimeShift: {\n\t\tKeyName:      \"history.timerProcessorMaxTimeShift\",\n\t\tDescription:  \"TimerProcessorMaxTimeShift is the max shift timer processor can have\",\n\t\tDefaultValue: time.Second,\n\t},\n\tTransferProcessorFailoverMaxStartJitterInterval: {\n\t\tKeyName:      \"history.transferProcessorFailoverMaxStartJitterInterval\",\n\t\tDescription:  \"TransferProcessorFailoverMaxStartJitterInterval is the max jitter interval for starting transfer failover queue processing. The actual jitter interval used will be a random duration between 0 and the max interval so that timer failover queue across different shards won't start at the same time\",\n\t\tDefaultValue: 0,\n\t},\n\tTransferProcessorMaxPollInterval: {\n\t\tKeyName:      \"history.transferProcessorMaxPollInterval\",\n\t\tDescription:  \"TransferProcessorMaxPollInterval is max poll interval for transferQueueProcessor\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tTransferProcessorSplitQueueInterval: {\n\t\tKeyName:      \"history.transferProcessorSplitQueueInterval\",\n\t\tDescription:  \"TransferProcessorSplitQueueInterval is the split processing queue interval for transferQueueProcessor\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tTransferProcessorUpdateAckInterval: {\n\t\tKeyName:      \"history.transferProcessorUpdateAckInterval\",\n\t\tDescription:  \"TransferProcessorUpdateAckInterval is update interval for transferQueueProcessor\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tTransferProcessorCompleteTransferInterval: {\n\t\tKeyName:      \"history.transferProcessorCompleteTransferInterval\",\n\t\tDescription:  \"TransferProcessorCompleteTransferInterval is complete timer interval for transferQueueProcessor\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tTransferProcessorValidationInterval: {\n\t\tKeyName:      \"history.transferProcessorValidationInterval\",\n\t\tDescription:  \"TransferProcessorValidationInterval is interval for performing transfer queue validation\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tTransferProcessorVisibilityArchivalTimeLimit: {\n\t\tKeyName:      \"history.transferProcessorVisibilityArchivalTimeLimit\",\n\t\tDescription:  \"TransferProcessorVisibilityArchivalTimeLimit is the upper time limit for archiving visibility records\",\n\t\tDefaultValue: time.Millisecond * 400,\n\t},\n\tReplicatorUpperLatency: {\n\t\tKeyName:      \"history.replicatorUpperLatency\",\n\t\tDescription:  \"ReplicatorUpperLatency indicates the max allowed replication latency between clusters\",\n\t\tDefaultValue: time.Second * 40,\n\t},\n\tShardUpdateMinInterval: {\n\t\tKeyName:      \"history.shardUpdateMinInterval\",\n\t\tDescription:  \"ShardUpdateMinInterval is the minimal time interval which the shard info can be updated\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tShardSyncMinInterval: {\n\t\tKeyName:      \"history.shardSyncMinInterval\",\n\t\tDescription:  \"ShardSyncMinInterval is the minimal time interval which the shard info should be sync to remote\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tStickyTTL: {\n\t\tKeyName:      \"history.stickyTTL\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"StickyTTL is to expire a sticky tasklist if no update more than this duration\",\n\t\tDefaultValue: time.Hour * 24 * 365,\n\t},\n\tDecisionHeartbeatTimeout: {\n\t\tKeyName:      \"history.decisionHeartbeatTimeout\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"DecisionHeartbeatTimeout is for decision heartbeat\",\n\t\tDefaultValue: time.Minute * 30, // about 30m\n\t},\n\tNormalDecisionScheduleToStartTimeout: {\n\t\tKeyName:      \"history.normalDecisionScheduleToStartTimeout\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"NormalDecisionScheduleToStartTimeout is scheduleToStart timeout duration for normal (non-sticky) decision task\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tNotifyFailoverMarkerInterval: {\n\t\tKeyName:      \"history.NotifyFailoverMarkerInterval\",\n\t\tDescription:  \"NotifyFailoverMarkerInterval is determines the frequency to notify failover marker\",\n\t\tDefaultValue: time.Second * 5,\n\t},\n\tActivityMaxScheduleToStartTimeoutForRetry: {\n\t\tKeyName:      \"history.activityMaxScheduleToStartTimeoutForRetry\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDescription:  \"ActivityMaxScheduleToStartTimeoutForRetry is maximum value allowed when overwritting the schedule to start timeout for activities with retry policy\",\n\t\tDefaultValue: time.Minute * 30,\n\t},\n\tReplicationTaskFetcherAggregationInterval: {\n\t\tKeyName:      \"history.ReplicationTaskFetcherAggregationInterval\",\n\t\tDescription:  \"ReplicationTaskFetcherAggregationInterval determines how frequently the fetch requests are sent\",\n\t\tDefaultValue: time.Second * 2,\n\t},\n\tReplicationTaskFetcherErrorRetryWait: {\n\t\tKeyName:      \"history.ReplicationTaskFetcherErrorRetryWait\",\n\t\tDescription:  \"ReplicationTaskFetcherErrorRetryWait is the wait time when fetcher encounters error\",\n\t\tDefaultValue: time.Second,\n\t},\n\tReplicationTaskFetcherServiceBusyWait: {\n\t\tKeyName:      \"history.ReplicationTaskFetcherServiceBusyWait\",\n\t\tDescription:  \"ReplicationTaskFetcherServiceBusyWait is the wait time when fetcher encounters service busy error\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tReplicationTaskProcessorErrorRetryWait: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorErrorRetryWait\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorErrorRetryWait is the initial retry wait when we see errors in applying replication tasks\",\n\t\tDefaultValue: time.Millisecond * 50,\n\t},\n\tReplicationTaskProcessorErrorSecondRetryWait: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorErrorSecondRetryWait\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorErrorSecondRetryWait is the initial retry wait for the second phase retry\",\n\t\tDefaultValue: time.Second * 5,\n\t},\n\tReplicationTaskProcessorErrorSecondRetryMaxWait: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorErrorSecondRetryMaxWait\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorErrorSecondRetryMaxWait is the max wait time for the second phase retry\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tReplicationTaskProcessorErrorSecondRetryExpiration: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorErrorSecondRetryExpiration\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorErrorSecondRetryExpiration is the expiration duration for the second phase retry\",\n\t\tDefaultValue: time.Minute * 5,\n\t},\n\tReplicationTaskProcessorNoTaskInitialWait: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorNoTaskInitialWait\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorNoTaskInitialWait is the wait time when not ask is returned\",\n\t\tDefaultValue: time.Second * 2,\n\t},\n\tReplicationTaskProcessorCleanupInterval: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorCleanupInterval\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorCleanupInterval determines how frequently the cleanup replication queue\",\n\t\tDefaultValue: time.Minute,\n\t},\n\tReplicationTaskProcessorStartWait: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorStartWait\",\n\t\tFilters:      []Filter{ShardID},\n\t\tDescription:  \"ReplicationTaskProcessorStartWait is the wait time before each task processing batch\",\n\t\tDefaultValue: 0,\n\t},\n\tReplicationTaskProcessorLatencyLogThreshold: {\n\t\tKeyName:      \"history.ReplicationTaskProcessorLatencyLogThreshold\",\n\t\tDescription:  \"ReplicationTaskProcessorLatencyLogThreshold is is the threshold of whether history will log history replication latency.\",\n\t\tDefaultValue: 0,\n\t},\n\tWorkerESProcessorFlushInterval: {\n\t\tKeyName:      \"worker.ESProcessorFlushInterval\",\n\t\tDescription:  \"WorkerESProcessorFlushInterval is flush interval for esProcessor\",\n\t\tDefaultValue: time.Second,\n\t},\n\tWorkerTimeLimitPerArchivalIteration: {\n\t\tKeyName:      \"worker.TimeLimitPerArchivalIteration\",\n\t\tDescription:  \"WorkerTimeLimitPerArchivalIteration is controls the time limit of each iteration of archival workflow\",\n\t\tDefaultValue: time.Hour * 24 * 15,\n\t},\n\tWorkerReplicationTaskMaxRetryDuration: {\n\t\tKeyName:      \"worker.replicationTaskMaxRetryDuration\",\n\t\tDescription:  \"WorkerReplicationTaskMaxRetryDuration is the max retry duration for any task\",\n\t\tDefaultValue: time.Minute * 10,\n\t},\n\tESAnalyzerTimeWindow: {\n\t\tKeyName:      \"worker.ESAnalyzerTimeWindow\",\n\t\tDescription:  \"ESAnalyzerTimeWindow defines the time window ElasticSearch Analyzer will consider while taking workflow averages\",\n\t\tDefaultValue: time.Hour * 24 * 30,\n\t},\n\tIsolationGroupStateRefreshInterval: {\n\t\tKeyName:      \"system.isolationGroupStateRefreshInterval\",\n\t\tDescription:  \"the frequency by which the IsolationGroupState handler will poll configuration\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tIsolationGroupStateFetchTimeout: {\n\t\tKeyName:      \"system.IsolationGroupStateFetchTimeout\",\n\t\tDescription:  \"IsolationGroupStateFetchTimeout is the dynamic config DB fetch timeout value\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tIsolationGroupStateUpdateTimeout: {\n\t\tKeyName:      \"system.IsolationGroupStateUpdateTimeout\",\n\t\tDescription:  \"IsolationGroupStateFetchTimeout is the dynamic config DB update timeout value\",\n\t\tDefaultValue: time.Second * 30,\n\t},\n\tESAnalyzerBufferWaitTime: {\n\t\tKeyName:      \"worker.ESAnalyzerBufferWaitTime\",\n\t\tDescription:  \"ESAnalyzerBufferWaitTime controls min time required to consider a worklow stuck\",\n\t\tDefaultValue: time.Minute * 30,\n\t},\n\tAsyncTaskDispatchTimeout: {\n\t\tKeyName:      \"matching.asyncTaskDispatchTimeout\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"AsyncTaskDispatchTimeout is the timeout of dispatching tasks for async match\",\n\t\tDefaultValue: time.Second * 3,\n\t},\n\tAppendTaskTimeout: {\n\t\tKeyName:      \"matching.appendTaskTimeout\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"AppendTaskTimeout is the timeout of appending tasks to persistence.\",\n\t\tDefaultValue: time.Second * 5,\n\t},\n\tHistoryGlobalRatelimiterDecayAfter: {\n\t\tKeyName:      \"history.globalRatelimiterDecayAfter\",\n\t\tDescription:  \"HistoryGlobalRatelimiterDecayAfter defines how long to wait for an update before considering a host's data \\\"possibly gone\\\", causing its weight to gradually decline.\",\n\t\tDefaultValue: 6 * time.Second,\n\t},\n\tHistoryGlobalRatelimiterGCAfter: {\n\t\tKeyName:      \"history.globalRatelimiterGCAfter\",\n\t\tDescription:  \"HistoryGlobalRatelimiterGCAfter defines how long to wait until a host's data is considered entirely useless, e.g. host has likely disappeared, its weight is very low, and the data can be deleted.\",\n\t\tDefaultValue: 30 * time.Second,\n\t},\n\tLocalPollWaitTime: {\n\t\tKeyName:      \"matching.localPollWaitTime\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"LocalPollWaitTime is the time a poller waits before considering request forwarding.\",\n\t\tDefaultValue: time.Millisecond * 10,\n\t},\n\tLocalTaskWaitTime: {\n\t\tKeyName:      \"matching.localTaskWaitTime\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"LocalTaskWaitTime is the time a task waits for a poller to arrive before considering task forwarding\",\n\t\tDefaultValue: time.Millisecond * 10,\n\t},\n\tTaskIsolationDuration: {\n\t\tKeyName:      \"matching.taskIsolationDuration\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"TaskIsolationDuration is the time period for which we attempt to respect tasklist isolation before allowing any poller to process the task\",\n\t\tDefaultValue: time.Millisecond * 200,\n\t},\n\tTaskIsolationPollerWindow: {\n\t\tKeyName:      \"matching.taskIsolationPollerWindow\",\n\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\tDescription:  \"TaskIsolationDuration is the time period for which we attempt to respect tasklist isolation before allowing any poller to process the task\",\n\t\tDefaultValue: time.Second * 2,\n\t},\n\tDomainAuditLogTTL: {\n\t\tKeyName:      \"system.domainAuditLogTTL\",\n\t\tFilters:      []Filter{DomainID},\n\t\tDescription:  \"DomainAuditLogTTL is the TTL for domain audit log entries\",\n\t\tDefaultValue: time.Hour * 24 * 365, // 1 year\n\t},\n}\n\nvar MapKeys = map[MapKey]DynamicMap{\n\tTestGetMapPropertyKey: {\n\t\tKeyName:      \"testGetMapPropertyKey\",\n\t\tDescription:  \"\",\n\t\tDefaultValue: nil,\n\t},\n\tRequiredDomainDataKeys: {\n\t\tKeyName:      \"system.requiredDomainDataKeys\",\n\t\tDescription:  \"RequiredDomainDataKeys is the key for the list of data keys required in domain registration\",\n\t\tDefaultValue: nil,\n\t},\n\tValidSearchAttributes: {\n\t\tKeyName:      \"frontend.validSearchAttributes\",\n\t\tDescription:  \"ValidSearchAttributes is legal indexed keys that can be used in list APIs. When overriding, ensure to include the existing default attributes of the current release\",\n\t\tDefaultValue: definition.GetDefaultIndexedKeys(),\n\t},\n\tTaskSchedulerRoundRobinWeights: {\n\t\tKeyName:      \"history.taskSchedulerRoundRobinWeight\",\n\t\tDescription:  \"TaskSchedulerRoundRobinWeights is the priority weight for weighted round robin task scheduler\",\n\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(DefaultTaskSchedulerRoundRobinWeights),\n\t},\n\tTaskSchedulerDomainRoundRobinWeights: {\n\t\tKeyName:      \"history.taskSchedulerDomainRoundRobinWeight\",\n\t\tDescription:  \"TaskSchedulerDomainRoundRobinWeights is the priority round robin weights for domains\",\n\t\tFilters:      []Filter{DomainName},\n\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(DefaultTaskSchedulerRoundRobinWeights),\n\t},\n\tQueueProcessorPendingTaskSplitThreshold: {\n\t\tKeyName:      \"history.queueProcessorPendingTaskSplitThreshold\",\n\t\tDescription:  \"QueueProcessorPendingTaskSplitThreshold is the threshold for the number of pending tasks per domain\",\n\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(map[int]int{0: 1000, 1: 10000}),\n\t},\n\tQueueProcessorStuckTaskSplitThreshold: {\n\t\tKeyName:      \"history.queueProcessorStuckTaskSplitThreshold\",\n\t\tDescription:  \"QueueProcessorStuckTaskSplitThreshold is the threshold for the number of attempts of a task\",\n\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(map[int]int{0: 100, 1: 10000}),\n\t},\n\tPinotOptimizedQueryColumns: {\n\t\tKeyName:      \"frontend.pinotOptimizedQueryColumns\",\n\t\tDescription:  \"PinotOptimizedQueryColumns is the list of search attributes that can be used in pinot optimized query\",\n\t\tDefaultValue: map[string]interface{}{},\n\t},\n\tSearchAttributesHiddenValueKeys: {\n\t\tKeyName:      \"frontend.searchAttributesHiddenValueKeys\",\n\t\tDescription:  \"SearchAttributesHiddenValueKeys is the list of search attributes that values should be hidden\",\n\t\tDefaultValue: map[string]interface{}{},\n\t},\n}\n\nvar ListKeys = map[ListKey]DynamicList{\n\tAllIsolationGroups: {\n\t\tKeyName:     \"system.allIsolationGroups\",\n\t\tDescription: \"A list of all the isolation groups in a system\",\n\t},\n\tRateLimiterBypassCallerTypes: {\n\t\tKeyName:      \"system.rateLimiterBypassCallerTypes\",\n\t\tDescription:  \"List of caller types that bypass rate limiters (both frontend and persistence)\",\n\t\tDefaultValue: []interface{}{},\n\t},\n\tDefaultIsolationGroupConfigStoreManagerGlobalMapping: {\n\t\tKeyName: \"system.defaultIsolationGroupConfigStoreManagerGlobalMapping\",\n\t\tDescription: \"A configuration store for global isolation groups - used in isolation-group config only, not normal dynamic config.\" +\n\t\t\t\"Not intended for use in normal dynamic config\",\n\t},\n\tHeaderForwardingRules: {\n\t\tKeyName: \"admin.HeaderForwardingRules\", // make a new scope for global?\n\t\tDescription: \"Only loaded at startup.  \" +\n\t\t\t\"A list of rpc.HeaderRule values that define which headers to include or exclude for all requests, applied in order.  \" +\n\t\t\t\"Regexes and header names are used as-is, you are strongly encouraged to use `(?i)` to make your regex case-insensitive.\",\n\t\tDefaultValue: []interface{}{\n\t\t\t// historical behavior: include literally everything.\n\t\t\t// this alone is quite problematic, and is strongly recommended against.\n\t\t\tmap[string]interface{}{ // config imports dynamicconfig, sadly\n\t\t\t\t\"Add\":   true,\n\t\t\t\t\"Match\": \"\",\n\t\t\t},\n\t\t},\n\t},\n}\n\nvar _keyNames map[string]Key\n\nfunc init() {\n\tpanicIfKeyInvalid := func(name string, key Key) {\n\t\tif name == \"\" {\n\t\t\tpanic(fmt.Sprintf(\"empty keyName: %T, %v\", key, key))\n\t\t}\n\t\tif _, ok := _keyNames[name]; ok {\n\t\t\tpanic(fmt.Sprintf(\"duplicate keyName: %v\", name))\n\t\t}\n\t}\n\t_keyNames = make(map[string]Key)\n\tfor k, v := range IntKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range BoolKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range FloatKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range StringKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range DurationKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range MapKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n\tfor k, v := range ListKeys {\n\t\tpanicIfKeyInvalid(v.KeyName, k)\n\t\t_keyNames[v.KeyName] = k\n\t}\n}\n\nvar (\n\tDefaultTaskSchedulerRoundRobinWeights = map[int]int{\n\t\tconstants.GetTaskPriority(constants.HighPriorityClass, constants.DefaultPrioritySubclass):    500,\n\t\tconstants.GetTaskPriority(constants.DefaultPriorityClass, constants.DefaultPrioritySubclass): 20,\n\t\tconstants.GetTaskPriority(constants.LowPriorityClass, constants.DefaultPrioritySubclass):     5,\n\t}\n)\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/constants_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype constantSuite struct {\n\tsuite.Suite\n}\n\nfunc TestConstantSuite(t *testing.T) {\n\tsuite.Run(t, new(constantSuite))\n}\n\nfunc (s *constantSuite) TestListAllProductionKeys() {\n\t// check if we given enough capacity\n\ttestResult := ListAllProductionKeys()\n\ts.GreaterOrEqual(len(IntKeys)+len(BoolKeys)+len(FloatKeys)+len(StringKeys)+len(DurationKeys)+len(MapKeys), len(testResult))\n\ts.Equal(TestGetIntPropertyFilteredByTaskListInfoKey+1, testResult[0])\n}\n\nfunc (s *constantSuite) TestGetKeyFromKeyName() {\n\tokKeyName := \"system.transactionSizeLimit\"\n\tokResult, err := GetKeyFromKeyName(okKeyName)\n\ts.NoError(err)\n\ts.Equal(TransactionSizeLimit, okResult)\n\n\tnotOkKeyName := \"system.transactionSizeLimit1\"\n\tnotOkResult, err := GetKeyFromKeyName(notOkKeyName)\n\ts.Error(err)\n\ts.Nil(notOkResult)\n}\n\nfunc (s *constantSuite) TestGetAllKeys() {\n\ttestResult := GetAllKeys()\n\ts.Equal(len(IntKeys)+len(BoolKeys)+len(FloatKeys)+len(StringKeys)+len(DurationKeys)+len(MapKeys)+len(ListKeys), len(testResult))\n\ts.Equal(_keyNames[\"testGetIntPropertyKey\"], testResult[\"testGetIntPropertyKey\"])\n\ts.NotEqual(_keyNames[\"testGetIntPropertyKey\"], testResult[\"testGetIntPropertyFilteredByTaskListInfoKey\"])\n}\n\ntype NewKey int\n\nfunc (k NewKey) String() string {\n\treturn \"NewKey\"\n}\n\nfunc (k NewKey) Description() string {\n\treturn \"NewKey is a new key\"\n}\n\nfunc (k NewKey) DefaultValue() interface{} {\n\treturn 0\n}\n\nfunc (k NewKey) Filters() []Filter {\n\treturn nil\n}\n\nfunc (s *constantSuite) TestValidateKeyValuePair() {\n\tnewKeyError := ValidateKeyValuePair(NewKey(0), 0)\n\ts.Error(newKeyError)\n\tintKeyError := ValidateKeyValuePair(TestGetIntPropertyKey, \"0\")\n\ts.Error(intKeyError)\n\tboolKeyError := ValidateKeyValuePair(TestGetBoolPropertyKey, 0)\n\ts.Error(boolKeyError)\n\tfloatKeyError := ValidateKeyValuePair(TestGetFloat64PropertyKey, 0)\n\ts.Error(floatKeyError)\n\tstringKeyError := ValidateKeyValuePair(TestGetStringPropertyKey, 0)\n\ts.Error(stringKeyError)\n\tdurationKeyError := ValidateKeyValuePair(TestGetDurationPropertyKey, 0)\n\ts.Error(durationKeyError)\n\tmapKeyError := ValidateKeyValuePair(TestGetMapPropertyKey, 0)\n\ts.Error(mapKeyError)\n\tlistKeyError := ValidateKeyValuePair(TestGetListPropertyKey, 0)\n\ts.Error(listKeyError)\n}\n\nfunc (s *constantSuite) TestIntKey() {\n\ttestIntKeys := map[string]struct {\n\t\tkey                  IntKey\n\t\texpectedString       string\n\t\texpectedDefaultValue int\n\t\texpectedDescription  string\n\t\texpectedFilters      []Filter\n\t}{\n\t\t\"TestGetIntPropertyKey\": {\n\t\t\tkey:                  TestGetIntPropertyKey,\n\t\t\texpectedString:       \"testGetIntPropertyKey\",\n\t\t\texpectedDescription:  \"\",\n\t\t\texpectedDefaultValue: 0,\n\t\t\texpectedFilters:      nil,\n\t\t},\n\t\t\"TransactionSizeLimit\": {\n\t\t\tkey:                  TransactionSizeLimit,\n\t\t\texpectedString:       \"system.transactionSizeLimit\",\n\t\t\texpectedDescription:  \"TransactionSizeLimit is the largest allowed transaction size to persistence\",\n\t\t\texpectedDefaultValue: 14680064,\n\t\t},\n\t\t\"BlobSizeLimitWarn\": {\n\t\t\tkey:                  BlobSizeLimitWarn,\n\t\t\texpectedString:       \"limit.blobSize.warn\",\n\t\t\texpectedDescription:  \"BlobSizeLimitWarn is the per event blob size limit for warning\",\n\t\t\texpectedDefaultValue: 256 * 1024,\n\t\t\texpectedFilters:      []Filter{DomainName},\n\t\t},\n\t}\n\n\tfor _, value := range testIntKeys {\n\t\ts.Equal(value.expectedString, value.key.String())\n\t\ts.Equal(value.expectedDefaultValue, value.key.DefaultValue())\n\t\ts.Equal(value.expectedDescription, value.key.Description())\n\t\ts.Equal(value.expectedFilters, value.key.Filters())\n\t\ts.Equal(value.expectedDefaultValue, value.key.DefaultInt())\n\t}\n}\n\nfunc (s *constantSuite) TestBoolKey() {\n\ttestBoolKeys := map[string]struct {\n\t\tkey                  BoolKey\n\t\texpectedString       string\n\t\texpectedDefaultValue bool\n\t\texpectedDescription  string\n\t\texpectedFilters      []Filter\n\t}{\n\t\t\"TestGetBoolPropertyKey\": {\n\t\t\tkey:                  TestGetBoolPropertyKey,\n\t\t\texpectedString:       \"testGetBoolPropertyKey\",\n\t\t\texpectedDescription:  \"\",\n\t\t\texpectedDefaultValue: false,\n\t\t\texpectedFilters:      nil,\n\t\t},\n\t\t\"FrontendEmitSignalNameMetricsTag\": {\n\t\t\tkey:                  FrontendEmitSignalNameMetricsTag,\n\t\t\texpectedString:       \"frontend.emitSignalNameMetricsTag\",\n\t\t\texpectedDescription:  \"FrontendEmitSignalNameMetricsTag enables emitting signal name tag in metrics in frontend client\",\n\t\t\texpectedDefaultValue: false,\n\t\t\texpectedFilters:      []Filter{DomainName},\n\t\t},\n\t}\n\n\tfor _, value := range testBoolKeys {\n\t\ts.Equal(value.expectedString, value.key.String())\n\t\ts.Equal(value.expectedDefaultValue, value.key.DefaultValue())\n\t\ts.Equal(value.expectedDescription, value.key.Description())\n\t\ts.Equal(value.expectedFilters, value.key.Filters())\n\t\ts.Equal(value.expectedDefaultValue, value.key.DefaultBool())\n\t}\n}\n\nfunc (s *constantSuite) TestFloatKey() {\n\ttestFloatKeys := map[string]struct {\n\t\tkey          FloatKey\n\t\tKeyName      string\n\t\tFilters      []Filter\n\t\tDescription  string\n\t\tDefaultValue float64\n\t}{\n\t\t\"TestGetFloat64PropertyKey\": {\n\t\t\tkey:          TestGetFloat64PropertyKey,\n\t\t\tKeyName:      \"testGetFloat64PropertyKey\",\n\t\t\tDescription:  \"\",\n\t\t\tDefaultValue: 0,\n\t\t},\n\t\t\"DomainFailoverRefreshTimerJitterCoefficient\": {\n\t\t\tkey:          DomainFailoverRefreshTimerJitterCoefficient,\n\t\t\tKeyName:      \"frontend.domainFailoverRefreshTimerJitterCoefficient\",\n\t\t\tDescription:  \"DomainFailoverRefreshTimerJitterCoefficient is the jitter for domain failover refresh timer jitter\",\n\t\t\tDefaultValue: 0.1,\n\t\t},\n\t\t\"ReplicationTaskProcessorStartWaitJitterCoefficient\": {\n\t\t\tkey:          ReplicationTaskProcessorStartWaitJitterCoefficient,\n\t\t\tKeyName:      \"history.ReplicationTaskProcessorStartWaitJitterCoefficient\",\n\t\t\tFilters:      []Filter{ShardID},\n\t\t\tDescription:  \"ReplicationTaskProcessorStartWaitJitterCoefficient is the jitter for batch start wait timer\",\n\t\t\tDefaultValue: 0.9,\n\t\t},\n\t}\n\n\tfor _, value := range testFloatKeys {\n\t\ts.Equal(value.KeyName, value.key.String())\n\t\ts.Equal(value.DefaultValue, value.key.DefaultValue())\n\t\ts.Equal(value.Description, value.key.Description())\n\t\ts.Equal(value.Filters, value.key.Filters())\n\t\ts.Equal(value.DefaultValue, value.key.DefaultFloat())\n\t}\n}\n\nfunc (s *constantSuite) TestStringKey() {\n\ttestStringKeys := map[string]struct {\n\t\tKey          StringKey\n\t\tKeyName      string\n\t\tFilters      []Filter\n\t\tDescription  string\n\t\tDefaultValue string\n\t}{\n\t\t\"TestGetStringPropertyKey\": {\n\t\t\tKey:          TestGetStringPropertyKey,\n\t\t\tKeyName:      \"testGetStringPropertyKey\",\n\t\t\tDescription:  \"\",\n\t\t\tDefaultValue: \"\",\n\t\t},\n\t\t\"HistoryArchivalStatus\": {\n\t\t\tKey:          HistoryArchivalStatus,\n\t\t\tKeyName:      \"system.historyArchivalStatus\",\n\t\t\tDescription:  \"HistoryArchivalStatus is key for the status of history archival to override the value from static config.\",\n\t\t\tDefaultValue: \"enabled\",\n\t\t},\n\t\t\"DefaultEventEncoding\": {\n\t\t\tKey:          DefaultEventEncoding,\n\t\t\tKeyName:      \"history.defaultEventEncoding\",\n\t\t\tFilters:      []Filter{DomainName},\n\t\t\tDescription:  \"DefaultEventEncoding is the encoding type for history events\",\n\t\t\tDefaultValue: string(constants.EncodingTypeThriftRW),\n\t\t},\n\t\t\"ReadVisibilityStoreName\": {\n\t\t\tKey:          ReadVisibilityStoreName,\n\t\t\tKeyName:      \"system.readVisibilityStoreName\",\n\t\t\tFilters:      []Filter{DomainName},\n\t\t\tDescription:  \"ReadVisibilityStoreName is key to identify which store to read visibility data from\",\n\t\t\tDefaultValue: \"es\",\n\t\t},\n\t\t\"ShardDistributorMigrationMode\": {\n\t\t\tKey:          ShardDistributorMigrationMode,\n\t\t\tKeyName:      \"shardDistributor.migrationMode\",\n\t\t\tFilters:      []Filter{Namespace},\n\t\t\tDescription:  \"ShardDistributorMigrationMode is the mode the at represent the state of the migration to rely on shard distributor for the sharding mechanism\",\n\t\t\tDefaultValue: \"onboarded\",\n\t\t},\n\t}\n\n\tfor _, value := range testStringKeys {\n\t\ts.Equal(value.KeyName, value.Key.String())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultValue())\n\t\ts.Equal(value.Description, value.Key.Description())\n\t\ts.Equal(value.Filters, value.Key.Filters())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultString())\n\t}\n}\n\nfunc (s *constantSuite) TestDurationKey() {\n\ttestDurationKeys := map[string]struct {\n\t\tKey          DurationKey\n\t\tKeyName      string\n\t\tFilters      []Filter\n\t\tDescription  string\n\t\tDefaultValue time.Duration\n\t}{\n\t\t\"TestGetDurationPropertyKey\": {\n\t\t\tKey:          TestGetDurationPropertyKey,\n\t\t\tKeyName:      \"testGetDurationPropertyKey\",\n\t\t\tDescription:  \"\",\n\t\t\tDefaultValue: 0,\n\t\t},\n\t\t\"FrontendFailoverCoolDown\": {\n\t\t\tKey:          FrontendFailoverCoolDown,\n\t\t\tKeyName:      \"frontend.failoverCoolDown\",\n\t\t\tFilters:      []Filter{DomainName},\n\t\t\tDescription:  \"FrontendFailoverCoolDown is duration between two domain failvoers\",\n\t\t\tDefaultValue: time.Minute,\n\t\t},\n\t\t\"MatchingIdleTasklistCheckInterval\": {\n\t\t\tKey:          MatchingIdleTasklistCheckInterval,\n\t\t\tKeyName:      \"matching.idleTasklistCheckInterval\",\n\t\t\tFilters:      []Filter{DomainName, TaskListName, TaskType},\n\t\t\tDescription:  \"MatchingIdleTasklistCheckInterval is the IdleTasklistCheckInterval\",\n\t\t\tDefaultValue: time.Minute * 5,\n\t\t},\n\t}\n\n\tfor _, value := range testDurationKeys {\n\t\ts.Equal(value.KeyName, value.Key.String())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultValue())\n\t\ts.Equal(value.Description, value.Key.Description())\n\t\ts.Equal(value.Filters, value.Key.Filters())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultDuration())\n\t}\n}\n\nfunc (s *constantSuite) TestMapKey() {\n\ttestMapKeys := map[string]struct {\n\t\tKey          MapKey\n\t\tKeyName      string\n\t\tFilters      []Filter\n\t\tDescription  string\n\t\tDefaultValue map[string]interface{}\n\t}{\n\t\t\"TestGetMapPropertyKey\": {\n\t\t\tKey:          TestGetMapPropertyKey,\n\t\t\tKeyName:      \"testGetMapPropertyKey\",\n\t\t\tDescription:  \"\",\n\t\t\tDefaultValue: nil,\n\t\t},\n\t\t\"TaskSchedulerRoundRobinWeights\": {\n\t\t\tKey:         TaskSchedulerRoundRobinWeights,\n\t\t\tKeyName:     \"history.taskSchedulerRoundRobinWeight\",\n\t\t\tDescription: \"TaskSchedulerRoundRobinWeights is the priority weight for weighted round robin task scheduler\",\n\t\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(map[int]int{\n\t\t\t\tconstants.GetTaskPriority(constants.HighPriorityClass, constants.DefaultPrioritySubclass):    500,\n\t\t\t\tconstants.GetTaskPriority(constants.DefaultPriorityClass, constants.DefaultPrioritySubclass): 20,\n\t\t\t\tconstants.GetTaskPriority(constants.LowPriorityClass, constants.DefaultPrioritySubclass):     5,\n\t\t\t}),\n\t\t},\n\t\t\"QueueProcessorStuckTaskSplitThreshold\": {\n\t\t\tKey:          QueueProcessorStuckTaskSplitThreshold,\n\t\t\tKeyName:      \"history.queueProcessorStuckTaskSplitThreshold\",\n\t\t\tDescription:  \"QueueProcessorStuckTaskSplitThreshold is the threshold for the number of attempts of a task\",\n\t\t\tDefaultValue: ConvertIntMapToDynamicConfigMapProperty(map[int]int{0: 100, 1: 10000}),\n\t\t},\n\t}\n\n\tfor _, value := range testMapKeys {\n\t\ts.Equal(value.KeyName, value.Key.String())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultValue())\n\t\ts.Equal(value.Description, value.Key.Description())\n\t\ts.Equal(value.Filters, value.Key.Filters())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultMap())\n\t}\n}\n\nfunc (s *constantSuite) TestListKey() {\n\ttestListKeys := map[string]struct {\n\t\tKey          ListKey\n\t\tKeyName      string\n\t\tFilters      []Filter\n\t\tDescription  string\n\t\tDefaultValue []interface{}\n\t}{\n\t\t\"DefaultIsolationGroupConfigStoreManagerGlobalMapping\": {\n\t\t\tKey:     DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\tKeyName: \"system.defaultIsolationGroupConfigStoreManagerGlobalMapping\",\n\t\t\tDescription: \"A configuration store for global isolation groups - used in isolation-group config only, not normal dynamic config.\" +\n\t\t\t\t\"Not intended for use in normal dynamic config\",\n\t\t},\n\t\t\"HeaderForwardingRules\": {\n\t\t\tKey:     HeaderForwardingRules,\n\t\t\tKeyName: \"admin.HeaderForwardingRules\",\n\t\t\tDescription: \"Only loaded at startup.  \" +\n\t\t\t\t\"A list of rpc.HeaderRule values that define which headers to include or exclude for all requests, applied in order.  \" +\n\t\t\t\t\"Regexes and header names are used as-is, you are strongly encouraged to use `(?i)` to make your regex case-insensitive.\",\n\t\t\tDefaultValue: []interface{}{\n\t\t\t\tmap[string]interface{}{\n\t\t\t\t\t\"Add\":   true,\n\t\t\t\t\t\"Match\": \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, value := range testListKeys {\n\t\ts.Equal(value.KeyName, value.Key.String())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultValue())\n\t\ts.Equal(value.Description, value.Key.Description())\n\t\ts.Equal(value.Filters, value.Key.Filters())\n\t\ts.Equal(value.DefaultValue, value.Key.DefaultList())\n\t}\n}\n\nfunc TestDynamicConfigFilterTypeIsMapped(t *testing.T) {\n\trequire.Equal(t, int(LastFilterTypeForTest), len(filters))\n\tfor i := UnknownFilter; i < LastFilterTypeForTest; i++ {\n\t\trequire.NotEmpty(t, filters[i])\n\t}\n}\n\nfunc TestDynamicConfigFilterTypeIsParseable(t *testing.T) {\n\tallFilters := map[Filter]int{}\n\tfor idx, filterString := range filters { // TestDynamicConfigFilterTypeIsMapped ensures this is a complete list\n\t\t// all filter-strings must parse to unique filters\n\t\tparsed := ParseFilter(filterString)\n\t\tprev, ok := allFilters[parsed]\n\t\tassert.False(t, ok, \"%q is already mapped to the same filter type as %q\", filterString, filters[prev])\n\t\tallFilters[parsed] = idx\n\n\t\t// otherwise, only \"unknown\" should map to \"unknown\".\n\t\t// ParseFilter should probably be re-implemented to simply use a map that is shared with the definitions\n\t\t// so values cannot get out of sync, but for now this is just asserting what is currently built.\n\t\tif idx == 0 {\n\t\t\tassert.Equalf(t, UnknownFilter, ParseFilter(filterString), \"first filter string should have parsed as unknown: %v\", filterString)\n\t\t\t// unknown filter string is likely safe to change and then should be updated here, but otherwise this ensures the logic isn't entirely position-dependent.\n\t\t\trequire.Equalf(t, \"unknownFilter\", filterString, \"expected first filter to be 'unknownFilter', but it was %v\", filterString)\n\t\t} else {\n\t\t\tassert.NotEqualf(t, UnknownFilter, ParseFilter(filterString), \"failed to parse filter: %s, make sure it is in ParseFilter's switch statement\", filterString)\n\t\t}\n\t}\n}\n\nfunc TestDynamicConfigFilterStringsCorrectly(t *testing.T) {\n\tfor _, filterString := range filters {\n\t\t// filter-string-parsing and the resulting filter's String() must match\n\t\tparsed := ParseFilter(filterString)\n\t\tassert.Equal(t, filterString, parsed.String(), \"filters need to String() correctly as some impls rely on it\")\n\t}\n\t// should not be possible normally, but improper casting could trigger it\n\tbadFilter := Filter(len(filters))\n\tassert.Equal(t, UnknownFilter.String(), badFilter.String(), \"filters with indexes outside the list of known strings should String() to the unknown filter type\")\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/definitions.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n)\n\n// PropertyFn is a wrapper to get property from dynamic config\ntype PropertyFn func() interface{}\n\n// IntPropertyFn is a wrapper to get int property from dynamic config\ntype IntPropertyFn func(opts ...FilterOption) int\n\n// IntPropertyFnWithDomainFilter is a wrapper to get int property from dynamic config with domain as filter\ntype IntPropertyFnWithDomainFilter func(domain string) int\n\n// IntPropertyFnWithTaskListInfoFilters is a wrapper to get int property from dynamic config with three filters: domain, taskList, taskType\ntype IntPropertyFnWithTaskListInfoFilters func(domain string, taskList string, taskType int) int\n\n// IntPropertyFnWithShardIDFilter is a wrapper to get int property from dynamic config with shardID as filter\ntype IntPropertyFnWithShardIDFilter func(shardID int) int\n\n// FloatPropertyFn is a wrapper to get float property from dynamic config\ntype FloatPropertyFn func(opts ...FilterOption) float64\n\n// FloatPropertyFnWithShardIDFilter is a wrapper to get float property from dynamic config with shardID as filter\ntype FloatPropertyFnWithShardIDFilter func(shardID int) float64\n\n// FloatPropertyFnWithTaskListInfoFilters is a wrapper to get duration property from dynamic config  with three filters: domain, taskList, taskType\ntype FloatPropertyFnWithTaskListInfoFilters func(domain string, taskList string, taskType int) float64\n\n// Float64PropertyFnWithNamespaceFilters is a wrapper to get string property from dynamic config with namespace as filter\ntype Float64PropertyFnWithNamespaceFilters func(namespace string) float64\n\n// DurationPropertyFn is a wrapper to get duration property from dynamic config\ntype DurationPropertyFn func(opts ...FilterOption) time.Duration\n\n// DurationPropertyFnWithDomainFilter is a wrapper to get duration property from dynamic config with domain as filter\ntype DurationPropertyFnWithDomainFilter func(domain string) time.Duration\n\n// DurationPropertyFnWithDomainIDFilter is a wrapper to get duration property from dynamic config with domainID as filter\ntype DurationPropertyFnWithDomainIDFilter func(domainID string) time.Duration\n\n// DurationPropertyFnWithTaskListInfoFilters is a wrapper to get duration property from dynamic config  with three filters: domain, taskList, taskType\ntype DurationPropertyFnWithTaskListInfoFilters func(domain string, taskList string, taskType int) time.Duration\n\n// DurationPropertyFnWithShardIDFilter is a wrapper to get duration property from dynamic config with shardID as filter\ntype DurationPropertyFnWithShardIDFilter func(shardID int) time.Duration\n\n// BoolPropertyFn is a wrapper to get bool property from dynamic config\ntype BoolPropertyFn func(opts ...FilterOption) bool\n\n// StringPropertyFn is a wrapper to get string property from dynamic config\ntype StringPropertyFn func(opts ...FilterOption) string\n\n// MapPropertyFn is a wrapper to get map property from dynamic config\ntype MapPropertyFn func(opts ...FilterOption) map[string]interface{}\n\n// MapPropertyFnWithDomainFilter is a wrapper to get map property from dynamic config with domainName as filter\ntype MapPropertyFnWithDomainFilter func(domain string) map[string]interface{}\n\n// StringPropertyFnWithDomainFilter is a wrapper to get string property from dynamic config\ntype StringPropertyFnWithDomainFilter func(domain string) string\n\n// StringPropertyFnWithTaskListInfoFilters is a wrapper to get string property from dynamic config with domainID as filter\ntype StringPropertyFnWithTaskListInfoFilters func(domain string, taskList string, taskType int) string\n\n// StringPropertyFnWithNamespaceFilters is a wrapper to get string property from dynamic config with namespace as filter\ntype StringPropertyFnWithNamespaceFilters func(namespace string) string\n\n// BoolPropertyFnWithDomainFilter is a wrapper to get bool property from dynamic config with domain as filter\ntype BoolPropertyFnWithDomainFilter func(domain string) bool\n\n// BoolPropertyFnWithDomainIDFilter is a wrapper to get bool property from dynamic config with domainID as filter\ntype BoolPropertyFnWithDomainIDFilter func(domainID string) bool\n\n// BoolPropertyFnWithDomainIDAndWorkflowIDFilter is a wrapper to get bool property from dynamic config with domainID and workflowID as filter\ntype BoolPropertyFnWithDomainIDAndWorkflowIDFilter func(domainID string, workflowID string) bool\n\n// BoolPropertyFnWithTaskListInfoFilters is a wrapper to get bool property from dynamic config with three filters: domain, taskList, taskType\ntype BoolPropertyFnWithTaskListInfoFilters func(domain string, taskList string, taskType int) bool\n\n// BoolPropertyFnWithShardIDFilter is a wrapper to get bool property from dynamic config with shardID as filter\ntype BoolPropertyFnWithShardIDFilter func(shardID int) bool\n\n// IntPropertyFnWithWorkflowTypeFilter is a wrapper to get int property from dynamic config with domain as filter\ntype IntPropertyFnWithWorkflowTypeFilter func(domainName string, workflowType string) int\n\n// DurationPropertyFnWithWorkflowTypeFilter is a wrapper to get duration property from dynamic config with domain as filter\ntype DurationPropertyFnWithWorkflowTypeFilter func(domainName string, workflowType string) time.Duration\n\n// ListPropertyFn is a wrapper to get a list property from dynamic config\ntype ListPropertyFn func(opts ...FilterOption) []interface{}\n\n// StringPropertyWithRatelimitKeyFilter is a wrapper to get strings (currently global ratelimiter modes) per global ratelimit key\ntype StringPropertyWithRatelimitKeyFilter func(globalRatelimitKey string) string\n\n// StringPropertyWithNamespaceFilter is a wrapper to get strings per namespace\ntype StringPropertyWithNamespaceFilter func(namespace string) string\n\nfunc (f IntPropertyFn) AsFloat64(opts ...FilterOption) func() float64 {\n\treturn func() float64 { return float64(f(opts...)) }\n}\n\nfunc (f IntPropertyFnWithDomainFilter) AsFloat64(domain string) func() float64 {\n\treturn func() float64 { return float64(f(domain)) }\n}\n\nfunc (f FloatPropertyFn) AsFloat64(opts ...FilterOption) func() float64 {\n\treturn func() float64 { return float64(f(opts...)) }\n}\n\n// ConvertIntMapToDynamicConfigMapProperty converts a map whose key value type are both int to\n// a map value that is compatible with dynamic config's map property\nfunc ConvertIntMapToDynamicConfigMapProperty(\n\tintMap map[int]int,\n) map[string]interface{} {\n\tdcValue := make(map[string]interface{})\n\tfor key, value := range intMap {\n\t\tdcValue[strconv.Itoa(key)] = value\n\t}\n\treturn dcValue\n}\n\n// ConvertDynamicConfigMapPropertyToIntMap convert a map property from dynamic config to a map\n// whose type for both key and value are int\nfunc ConvertDynamicConfigMapPropertyToIntMap(dcValue map[string]interface{}) (map[int]int, error) {\n\tintMap := make(map[int]int)\n\tfor key, value := range dcValue {\n\t\tintKey, err := strconv.Atoi(strings.TrimSpace(key))\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to convert key %v, error: %v\", key, err)\n\t\t}\n\n\t\tvar intValue int\n\t\tswitch value := value.(type) {\n\t\tcase float64:\n\t\t\tintValue = int(value)\n\t\tcase int:\n\t\t\tintValue = value\n\t\tcase int32:\n\t\t\tintValue = int(value)\n\t\tcase int64:\n\t\t\tintValue = int(value)\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"unknown value %v with type %T\", value, value)\n\t\t}\n\t\tintMap[intKey] = intValue\n\t}\n\treturn intMap, nil\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/filter.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Filter represents a filter on the dynamic config key\ntype Filter int\n\nfunc (f Filter) String() string {\n\tif f > UnknownFilter && int(f) < len(filters) {\n\t\treturn filters[f]\n\t}\n\treturn filters[UnknownFilter]\n}\n\nfunc ParseFilter(filterName string) Filter {\n\tswitch filterName {\n\tcase \"domainName\":\n\t\treturn DomainName\n\tcase \"domainID\":\n\t\treturn DomainID\n\tcase \"taskListName\":\n\t\treturn TaskListName\n\tcase \"taskType\":\n\t\treturn TaskType\n\tcase \"shardID\":\n\t\treturn ShardID\n\tcase \"clusterName\":\n\t\treturn ClusterName\n\tcase \"workflowID\":\n\t\treturn WorkflowID\n\tcase \"workflowType\":\n\t\treturn WorkflowType\n\tcase \"ratelimitKey\":\n\t\treturn RatelimitKey\n\tcase \"namespace\":\n\t\treturn Namespace\n\tdefault:\n\t\treturn UnknownFilter\n\t}\n}\n\nvar filters = []string{\n\t\"unknownFilter\",\n\t\"domainName\",\n\t\"domainID\",\n\t\"taskListName\",\n\t\"taskType\",\n\t\"shardID\",\n\t\"clusterName\",\n\t\"workflowID\",\n\t\"workflowType\",\n\t\"ratelimitKey\",\n\t\"namespace\",\n}\n\nconst (\n\tUnknownFilter Filter = iota\n\t// DomainName is the domain name\n\tDomainName\n\t// DomainID is the domain id\n\tDomainID\n\t// TaskListName is the tasklist name\n\tTaskListName\n\t// TaskType is the task type (0:Decision, 1:Activity)\n\tTaskType\n\t// ShardID is the shard id\n\tShardID\n\t// ClusterName is the cluster name in a multi-region setup\n\tClusterName\n\t// WorkflowID is the workflow id\n\tWorkflowID\n\t// WorkflowType is the workflow type name\n\tWorkflowType\n\t// RatelimitKey is the global ratelimit key (not a local key name)\n\tRatelimitKey\n\t// Namespace is the entity of independent shard distribution mechanism\n\tNamespace\n\n\t// LastFilterTypeForTest must be the last one in this const group for testing purpose\n\tLastFilterTypeForTest\n)\n\n// FilterOption is used to provide filters for dynamic config keys\ntype FilterOption func(filterMap map[Filter]interface{})\n\n// TaskListFilter filters by task list name\nfunc TaskListFilter(name string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[TaskListName] = name\n\t}\n}\n\n// DomainFilter filters by domain name\nfunc DomainFilter(name string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[DomainName] = name\n\t}\n}\n\n// DomainIDFilter filters by domain id\nfunc DomainIDFilter(domainID string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[DomainID] = domainID\n\t}\n}\n\n// TaskTypeFilter filters by task type\nfunc TaskTypeFilter(taskType int) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[TaskType] = taskType\n\t}\n}\n\n// ShardIDFilter filters by shard id\nfunc ShardIDFilter(shardID int) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[ShardID] = shardID\n\t}\n}\n\n// ClusterNameFilter filters by cluster name\nfunc ClusterNameFilter(clusterName string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[ClusterName] = clusterName\n\t}\n}\n\n// WorkflowIDFilter filters by workflowID\nfunc WorkflowIDFilter(workflowID string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[WorkflowID] = workflowID\n\t}\n}\n\n// WorkflowType filters by workflow type name\nfunc WorkflowTypeFilter(name string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[WorkflowType] = name\n\t}\n}\n\n// RatelimitKeyFilter filters on global ratelimiter keys (via the global name, not local names)\nfunc RatelimitKeyFilter(key string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[RatelimitKey] = key\n\t}\n}\n\n// NamespaceFilter filters by namespace\nfunc NamespaceFilter(namespace string) FilterOption {\n\treturn func(filterMap map[Filter]interface{}) {\n\t\tfilterMap[Namespace] = namespace\n\t}\n}\n\n// ToGetDynamicConfigFilterRequest generates a GetDynamicConfigRequest object\n// by converting filters to DynamicConfigFilter objects and setting values\nfunc ToGetDynamicConfigFilterRequest(configName string, filters []FilterOption) *types.GetDynamicConfigRequest {\n\tfilterMap := make(map[Filter]interface{}, len(filters))\n\tfor _, opt := range filters {\n\t\topt(filterMap)\n\t}\n\tvar dcFilters []*types.DynamicConfigFilter\n\tfor f, entity := range filterMap {\n\t\tfilter := &types.DynamicConfigFilter{\n\t\t\tName: f.String(),\n\t\t}\n\n\t\tdata, err := json.Marshal(entity)\n\t\tif err != nil {\n\t\t\tfmt.Errorf(\"could not marshall entity: %s\", err)\n\t\t}\n\n\t\tencodingType := types.EncodingTypeJSON\n\t\tfilter.Value = &types.DataBlob{\n\t\t\tEncodingType: &encodingType,\n\t\t\tData:         data,\n\t\t}\n\n\t\tdcFilters = append(dcFilters, filter)\n\t}\n\n\trequest := &types.GetDynamicConfigRequest{\n\t\tConfigName: configName,\n\t\tFilters:    dcFilters,\n\t}\n\n\treturn request\n}\n"
  },
  {
    "path": "common/dynamicconfig/dynamicproperties/utils_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicproperties\n\nimport (\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestConvertDynamicConfigMapPropertyToIntMap(t *testing.T) {\n\tdcValue := make(map[string]interface{})\n\tfor idx, value := range []interface{}{int(0), int32(1), int64(2), float64(3.0)} {\n\t\tdcValue[strconv.Itoa(idx)] = value\n\t}\n\n\tintMap, err := ConvertDynamicConfigMapPropertyToIntMap(dcValue)\n\trequire.NoError(t, err)\n\trequire.Len(t, intMap, 4)\n\tfor i := 0; i != 4; i++ {\n\t\trequire.Equal(t, i, intMap[i])\n\t}\n}\n"
  },
  {
    "path": "common/dynamicconfig/file_based_client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ Client = (*fileBasedClient)(nil)\n\nconst (\n\tminPollInterval = time.Second * 5\n\tfileMode        = 0644 // used for update config file\n)\n\ntype constrainedValue struct {\n\tValue       interface{}\n\tConstraints map[string]interface{}\n}\n\n// FileBasedClientConfig is the config for the file based dynamic config client.\n// It specifies where the config file is stored and how often the config should be\n// updated by checking the config file again.\ntype FileBasedClientConfig struct {\n\tFilepath     string        `yaml:\"filepath\"`\n\tPollInterval time.Duration `yaml:\"pollInterval\"`\n}\n\ntype fileBasedClient struct {\n\tvalues          atomic.Value\n\tlastUpdatedTime time.Time\n\tconfig          *FileBasedClientConfig\n\tdoneCh          chan struct{}\n\tlogger          log.Logger\n}\n\n// NewFileBasedClient creates a file based client.\nfunc NewFileBasedClient(config *FileBasedClientConfig, logger log.Logger, doneCh chan struct{}) (Client, error) {\n\tif err := validateConfig(config); err != nil {\n\t\treturn nil, err\n\t}\n\n\tclient := &fileBasedClient{\n\t\tconfig: config,\n\t\tdoneCh: doneCh,\n\t\tlogger: logger,\n\t}\n\tif err := client.update(); err != nil {\n\t\treturn nil, err\n\t}\n\tgo func() {\n\t\tticker := time.NewTicker(client.config.PollInterval)\n\t\tfor {\n\t\t\tselect {\n\t\t\tcase <-ticker.C:\n\t\t\t\terr := client.update()\n\t\t\t\tif err != nil {\n\t\t\t\t\tclient.logger.Error(\"Failed to update dynamic config\", tag.Error(err))\n\t\t\t\t}\n\t\t\tcase <-client.doneCh:\n\t\t\t\tticker.Stop()\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n\treturn client, nil\n}\n\nfunc (fc *fileBasedClient) GetValue(name dynamicproperties.Key) (interface{}, error) {\n\treturn fc.getValueWithFilters(name, nil, name.DefaultValue())\n}\n\nfunc (fc *fileBasedClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error) {\n\treturn fc.getValueWithFilters(name, filters, name.DefaultValue())\n}\n\nfunc (fc *fileBasedClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error) {\n\tdefaultValue := name.DefaultInt()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif intVal, ok := val.(int); ok {\n\t\treturn intVal, nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not int but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error) {\n\tdefaultValue := name.DefaultFloat()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif floatVal, ok := val.(float64); ok {\n\t\treturn floatVal, nil\n\t} else if intVal, ok := val.(int); ok {\n\t\treturn float64(intVal), nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not float64 but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error) {\n\tdefaultValue := name.DefaultBool()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif boolVal, ok := val.(bool); ok {\n\t\treturn boolVal, nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not bool but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error) {\n\tdefaultValue := name.DefaultString()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tif stringVal, ok := val.(string); ok {\n\t\treturn stringVal, nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not string but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error) {\n\tdefaultValue := name.DefaultMap()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\tif mapVal, ok := val.(map[string]interface{}); ok {\n\t\treturn mapVal, nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not map but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error) {\n\tdefaultValue := name.DefaultDuration()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\n\tdurationString, ok := val.(string)\n\tif !ok {\n\t\treturn defaultValue, fmt.Errorf(\"value type is not string but is: %T\", val)\n\t}\n\n\tdurationVal, err := time.ParseDuration(durationString)\n\tif err != nil {\n\t\treturn defaultValue, fmt.Errorf(\"failed to parse duration: %v\", err)\n\t}\n\treturn durationVal, nil\n}\n\nfunc (fc *fileBasedClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error) {\n\tdefaultValue := name.DefaultList()\n\tval, err := fc.getValueWithFilters(name, filters, defaultValue)\n\tif err != nil {\n\t\treturn defaultValue, err\n\t}\n\tif listVal, ok := val.([]interface{}); ok {\n\t\treturn listVal, nil\n\t}\n\treturn defaultValue, fmt.Errorf(\"value type is not list but is: %T\", val)\n}\n\nfunc (fc *fileBasedClient) UpdateValue(name dynamicproperties.Key, value interface{}) error {\n\tif err := dynamicproperties.ValidateKeyValuePair(name, value); err != nil {\n\t\treturn err\n\t}\n\tkeyName := name.String()\n\tcurrentValues := make(map[string][]*constrainedValue)\n\n\tconfContent, err := ioutil.ReadFile(fc.config.Filepath)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to read dynamic config file %v: %v\", fc.config.Filepath, err)\n\t}\n\n\tif err = yaml.Unmarshal(confContent, currentValues); err != nil {\n\t\treturn fmt.Errorf(\"failed to decode dynamic config %v\", err)\n\t}\n\n\tcVal := &constrainedValue{\n\t\tValue: value,\n\t}\n\tcurrentValues[keyName] = []*constrainedValue{cVal}\n\tnewBytes, _ := yaml.Marshal(currentValues)\n\n\terr = ioutil.WriteFile(fc.config.Filepath, newBytes, fileMode)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to write config file, err: %v\", err)\n\t}\n\n\treturn fc.storeValues(currentValues)\n}\n\nfunc (fc *fileBasedClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error {\n\treturn errors.New(\"not supported for file based client\")\n}\n\nfunc (fc *fileBasedClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\treturn nil, errors.New(\"not supported for file based client\")\n}\n\nfunc (fc *fileBasedClient) update() error {\n\tdefer func() {\n\t\tfc.lastUpdatedTime = time.Now()\n\t}()\n\n\tnewValues := make(map[string][]*constrainedValue)\n\n\tinfo, err := os.Stat(fc.config.Filepath)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to get status of dynamic config file: %v\", err)\n\t}\n\tif !info.ModTime().After(fc.lastUpdatedTime) {\n\t\treturn nil\n\t}\n\n\tconfContent, err := ioutil.ReadFile(fc.config.Filepath)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to read dynamic config file %v: %v\", fc.config.Filepath, err)\n\t}\n\n\tif err = yaml.Unmarshal(confContent, newValues); err != nil {\n\t\treturn fmt.Errorf(\"failed to decode dynamic config %v\", err)\n\t}\n\n\treturn fc.storeValues(newValues)\n}\n\nfunc (fc *fileBasedClient) storeValues(newValues map[string][]*constrainedValue) error {\n\t// yaml will unmarshal map into map[interface{}]interface{} instead of map[string]interface{}\n\t// manually convert key type to string for all values here\n\t// We don't need to convert constraints as their type can't be map. If user does use a map as filter\n\t// value, it won't match anyway.\n\tfor _, s := range newValues {\n\t\tfor _, cv := range s {\n\t\t\tvar err error\n\t\t\tcv.Value, err = convertKeyTypeToString(cv.Value)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t}\n\n\tfc.values.Store(newValues)\n\tfc.logger.Info(\"Updated dynamic config\")\n\treturn nil\n}\n\nfunc (fc *fileBasedClient) getValueWithFilters(key dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}, defaultValue interface{}) (interface{}, error) {\n\tkeyName := key.String()\n\tvalues := fc.values.Load().(map[string][]*constrainedValue)\n\tfound := false\n\tfor _, constrainedValue := range values[keyName] {\n\t\tif len(constrainedValue.Constraints) == 0 {\n\t\t\t// special handling for default value (value without any constraints)\n\t\t\tdefaultValue = constrainedValue.Value\n\t\t\tfound = true\n\t\t\tcontinue\n\t\t}\n\t\tif match(constrainedValue, filters) {\n\t\t\treturn constrainedValue.Value, nil\n\t\t}\n\t}\n\tif !found {\n\t\treturn defaultValue, NotFoundError\n\t}\n\treturn defaultValue, nil\n}\n\n// match will return true if the constraints matches the filters or any subsets\nfunc match(v *constrainedValue, filters map[dynamicproperties.Filter]interface{}) bool {\n\tif len(v.Constraints) > len(filters) {\n\t\treturn false\n\t}\n\n\tfor constrain, constrainedValue := range v.Constraints {\n\t\tconstrainKey := dynamicproperties.ParseFilter(constrain)\n\t\tif filters[constrainKey] == nil || filters[constrainKey] != constrainedValue {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc convertKeyTypeToString(v interface{}) (interface{}, error) {\n\tswitch v := v.(type) {\n\tcase map[interface{}]interface{}:\n\t\treturn convertKeyTypeToStringMap(v)\n\tcase []interface{}:\n\t\treturn convertKeyTypeToStringSlice(v)\n\tdefault:\n\t\treturn v, nil\n\t}\n}\n\nfunc convertKeyTypeToStringMap(m map[interface{}]interface{}) (map[string]interface{}, error) {\n\tstringKeyMap := make(map[string]interface{})\n\tfor key, value := range m {\n\t\tstringKey, ok := key.(string)\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(\"type of key %v is not string\", key)\n\t\t}\n\t\tconvertedValue, err := convertKeyTypeToString(value)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstringKeyMap[stringKey] = convertedValue\n\t}\n\treturn stringKeyMap, nil\n}\n\nfunc convertKeyTypeToStringSlice(s []interface{}) ([]interface{}, error) {\n\tstringKeySlice := make([]interface{}, len(s))\n\tfor idx, value := range s {\n\t\tconvertedValue, err := convertKeyTypeToString(value)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstringKeySlice[idx] = convertedValue\n\t}\n\treturn stringKeySlice, nil\n}\n\nfunc validateConfig(config *FileBasedClientConfig) error {\n\tif config == nil {\n\t\treturn errors.New(\"no config found for file based dynamic config client\")\n\t}\n\tif _, err := os.Stat(config.Filepath); err != nil {\n\t\treturn fmt.Errorf(\"error checking dynamic config file at path %s, error: %v\", config.Filepath, err)\n\t}\n\n\t// check if poll interval needs to be adjusted\n\tif config.PollInterval < minPollInterval {\n\t\tconfig.PollInterval = minPollInterval\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/dynamicconfig/file_based_client_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype fileBasedClientSuite struct {\n\tsuite.Suite\n\t*require.Assertions\n\tclient Client\n\tdoneCh chan struct{}\n}\n\nfunc TestFileBasedClientSuite(t *testing.T) {\n\ts := new(fileBasedClientSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *fileBasedClientSuite) SetupSuite() {\n\tvar err error\n\ts.doneCh = make(chan struct{})\n\ts.client, err = NewFileBasedClient(&FileBasedClientConfig{\n\t\tFilepath:     \"config/testConfig.yaml\",\n\t\tPollInterval: time.Second * 5,\n\t}, log.NewNoop(), s.doneCh)\n\ts.Require().NoError(err)\n}\n\nfunc (s *fileBasedClientSuite) TearDownSuite() {\n\tclose(s.doneCh)\n}\n\nfunc (s *fileBasedClientSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *fileBasedClientSuite) TestGetValue() {\n\tv, err := s.client.GetValue(dynamicproperties.TestGetBoolPropertyKey)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetValue_NonExistKey() {\n\tv, err := s.client.GetValue(dynamicproperties.EnableVisibilitySampling)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.EnableVisibilitySampling.DefaultBool(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetValueWithFilters() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"global-samples-domain\",\n\t}\n\tv, err := s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(true, v)\n\n\tfilters = map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"non-exist-domain\",\n\t}\n\tv, err = s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(false, v)\n\n\tfilters = map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\tdynamicproperties.TaskListName: \"non-exist-tasklist\",\n\t}\n\tv, err = s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(true, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetValueWithFilters_UnknownFilter() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:    \"global-samples-domain1\",\n\t\tdynamicproperties.UnknownFilter: \"unknown-filter1\",\n\t}\n\tv, err := s.client.GetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetIntValue() {\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(1000, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetIntValue_FilterNotMatch() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(1000, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetIntValue_WrongType() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"global-samples-domain\",\n\t}\n\tv, err := s.client.GetIntValue(dynamicproperties.TestGetIntPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetIntPropertyKey.DefaultInt(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetFloatValue() {\n\tv, err := s.client.GetFloatValue(dynamicproperties.TestGetFloat64PropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(12.0, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetFloatValue_WrongType() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetFloatValue(dynamicproperties.TestGetFloat64PropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetFloat64PropertyKey.DefaultFloat(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetBoolValue() {\n\tv, err := s.client.GetBoolValue(dynamicproperties.TestGetBoolPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(false, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetStringValue() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.TaskListName: \"random tasklist\",\n\t}\n\tv, err := s.client.GetStringValue(dynamicproperties.TestGetStringPropertyKey, filters)\n\ts.NoError(err)\n\ts.Equal(\"constrained-string\", v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetMapValue() {\n\tv, err := s.client.GetMapValue(dynamicproperties.TestGetMapPropertyKey, nil)\n\ts.NoError(err)\n\texpectedVal := map[string]interface{}{\n\t\t\"key1\": \"1\",\n\t\t\"key2\": 1,\n\t\t\"key3\": []interface{}{\n\t\t\tfalse,\n\t\t\tmap[string]interface{}{\n\t\t\t\t\"key4\": true,\n\t\t\t\t\"key5\": 2.1,\n\t\t\t},\n\t\t},\n\t}\n\ts.Equal(expectedVal, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetMapValue_WrongType() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.TaskListName: \"random tasklist\",\n\t}\n\tv, err := s.client.GetMapValue(dynamicproperties.TestGetMapPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetMapPropertyKey.DefaultMap(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetDurationValue() {\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, nil)\n\ts.NoError(err)\n\ts.Equal(time.Minute, v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetDurationValue_NotStringRepresentation() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName: \"samples-domain\",\n\t}\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetDurationPropertyKey.DefaultDuration(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestGetDurationValue_ParseFailed() {\n\tfilters := map[dynamicproperties.Filter]interface{}{\n\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\tdynamicproperties.TaskListName: \"longIdleTimeTasklist\",\n\t}\n\tv, err := s.client.GetDurationValue(dynamicproperties.TestGetDurationPropertyKey, filters)\n\ts.Error(err)\n\ts.Equal(dynamicproperties.TestGetDurationPropertyKey.DefaultDuration(), v)\n}\n\nfunc (s *fileBasedClientSuite) TestValidateConfig_ConfigNotExist() {\n\t_, err := NewFileBasedClient(nil, nil, nil)\n\ts.Error(err)\n}\n\nfunc (s *fileBasedClientSuite) TestValidateConfig_FileNotExist() {\n\t_, err := NewFileBasedClient(&FileBasedClientConfig{\n\t\tFilepath:     \"file/not/exist.yaml\",\n\t\tPollInterval: time.Second * 10,\n\t}, nil, nil)\n\ts.Error(err)\n}\n\nfunc (s *fileBasedClientSuite) TestValidateConfig_ShortPollInterval() {\n\tcfg := &FileBasedClientConfig{\n\t\tFilepath:     \"config/testConfig.yaml\",\n\t\tPollInterval: time.Second,\n\t}\n\t_, err := NewFileBasedClient(cfg, log.NewNoop(), nil)\n\ts.NoError(err)\n\ts.Equal(minPollInterval, cfg.PollInterval, \"fallback to default poll interval\")\n\n}\n\nfunc (s *fileBasedClientSuite) TestMatch() {\n\ttestCases := []struct {\n\t\tv       *constrainedValue\n\t\tfilters map[dynamicproperties.Filter]interface{}\n\t\tmatched bool\n\t}{\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName: \"some random domain\",\n\t\t\t},\n\t\t\tmatched: true,\n\t\t},\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{\"some key\": \"some value\"},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{\"domainName\": \"samples-domain\"},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName: \"some random domain\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{\n\t\t\t\t\t\"domainName\":   \"samples-domain\",\n\t\t\t\t\t\"taskListName\": \"sample-task-list\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: true,\n\t\t},\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{\n\t\t\t\t\t\"domainName\":        \"samples-domain\",\n\t\t\t\t\t\"some-other-filter\": \"sample-task-list\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.DomainName:   \"samples-domain\",\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t\t{\n\t\t\tv: &constrainedValue{\n\t\t\t\tConstraints: map[string]interface{}{\n\t\t\t\t\t\"domainName\": \"samples-domain\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tfilters: map[dynamicproperties.Filter]interface{}{\n\t\t\t\tdynamicproperties.TaskListName: \"sample-task-list\",\n\t\t\t},\n\t\t\tmatched: false,\n\t\t},\n\t}\n\n\tfor index, tc := range testCases {\n\t\tmatched := match(tc.v, tc.filters)\n\t\ts.Equal(tc.matched, matched, fmt.Sprintf(\"Test case %v failved\", index))\n\t}\n}\n\nfunc (s *fileBasedClientSuite) TestUpdateConfig() {\n\tclient := s.client.(*fileBasedClient)\n\tkey := dynamicproperties.ValidSearchAttributes\n\n\t// pre-check existing config\n\tcurrent, err := client.GetMapValue(key, nil)\n\ts.NoError(err)\n\tcurrentDomainVal, ok := current[\"DomainID\"]\n\ts.True(ok)\n\ts.Equal(1, currentDomainVal)\n\t_, ok = current[\"WorkflowID\"]\n\ts.False(ok)\n\n\t// update config\n\tv := map[string]interface{}{\n\t\t\"WorkflowID\": 1,\n\t\t\"DomainID\":   2,\n\t}\n\terr = client.UpdateValue(key, v)\n\ts.NoError(err)\n\n\t// verify update result\n\tcurrent, err = client.GetMapValue(key, nil)\n\ts.NoError(err)\n\tcurrentDomainVal, ok = current[\"DomainID\"]\n\ts.True(ok)\n\ts.Equal(2, currentDomainVal)\n\tcurrentWorkflowIDVal, ok := current[\"WorkflowID\"]\n\ts.True(ok)\n\ts.Equal(1, currentWorkflowIDVal)\n\n\t// revert test file back\n\tv = map[string]interface{}{\n\t\t\"DomainID\": 1,\n\t}\n\terr = client.UpdateValue(key, v)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "common/dynamicconfig/inMemoryClient.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"errors\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype inMemoryClient struct {\n\tsync.RWMutex\n\n\tglobalValues map[dynamicproperties.Key]interface{}\n}\n\n// NewInMemoryClient creates a new in memory dynamic config client for testing purpose\nfunc NewInMemoryClient() Client {\n\treturn &inMemoryClient{\n\t\tglobalValues: make(map[dynamicproperties.Key]interface{}),\n\t}\n}\n\nfunc (mc *inMemoryClient) SetValue(key dynamicproperties.Key, value interface{}) {\n\tmc.Lock()\n\tdefer mc.Unlock()\n\n\tmc.globalValues[key] = value\n}\n\nfunc (mc *inMemoryClient) GetValue(key dynamicproperties.Key) (interface{}, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[key]; ok {\n\t\treturn val, nil\n\t}\n\treturn key.DefaultValue(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\treturn mc.GetValue(name)\n}\n\nfunc (mc *inMemoryClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(int), nil\n\t}\n\treturn name.DefaultInt(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(float64), nil\n\t}\n\treturn name.DefaultFloat(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(bool), nil\n\t}\n\treturn name.DefaultBool(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(string), nil\n\t}\n\treturn name.DefaultString(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(map[string]interface{}), nil\n\t}\n\treturn name.DefaultMap(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.(time.Duration), nil\n\t}\n\treturn name.DefaultDuration(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error) {\n\tmc.RLock()\n\tdefer mc.RUnlock()\n\n\tif val, ok := mc.globalValues[name]; ok {\n\t\treturn val.([]interface{}), nil\n\t}\n\treturn name.DefaultList(), NotFoundError\n}\n\nfunc (mc *inMemoryClient) UpdateValue(key dynamicproperties.Key, value interface{}) error {\n\tif err := dynamicproperties.ValidateKeyValuePair(key, value); err != nil {\n\t\treturn err\n\t}\n\tmc.SetValue(key, value)\n\treturn nil\n}\n\nfunc (mc *inMemoryClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error {\n\treturn errors.New(\"not supported for in-memory client\")\n}\n\nfunc (mc *inMemoryClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\treturn nil, errors.New(\"not supported for in-memory client\")\n}\n"
  },
  {
    "path": "common/dynamicconfig/nopClient.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamicconfig\n\nimport (\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// nopClient is a dummy implements of dynamicconfig Client interface, all operations will always return default values.\ntype nopClient struct{}\n\nfunc (mc *nopClient) GetValue(name dynamicproperties.Key) (interface{}, error) {\n\treturn nil, NotFoundError\n}\n\nfunc (mc *nopClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error) {\n\treturn nil, NotFoundError\n}\n\nfunc (mc *nopClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error) {\n\treturn name.DefaultInt(), NotFoundError\n}\n\nfunc (mc *nopClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error) {\n\treturn name.DefaultFloat(), NotFoundError\n}\n\nfunc (mc *nopClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error) {\n\tif filters[dynamicproperties.DomainName] == \"TestRawHistoryDomain\" {\n\t\treturn true, NotFoundError\n\t}\n\treturn name.DefaultBool(), NotFoundError\n}\n\nfunc (mc *nopClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error) {\n\treturn name.DefaultString(), NotFoundError\n}\n\nfunc (mc *nopClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error) {\n\treturn name.DefaultMap(), NotFoundError\n}\n\nfunc (mc *nopClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error) {\n\treturn name.DefaultDuration(), NotFoundError\n}\n\nfunc (mc *nopClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error) {\n\treturn name.DefaultList(), NotFoundError\n}\n\nfunc (mc *nopClient) UpdateValue(name dynamicproperties.Key, value interface{}) error {\n\treturn errors.New(\"not supported for nop client\")\n}\n\nfunc (mc *nopClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error {\n\treturn errors.New(\"not supported for nop client\")\n}\n\nfunc (mc *nopClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\treturn nil, errors.New(\"not supported for nop client\")\n}\n\n// NewNopClient creates a nop client\nfunc NewNopClient() Client {\n\treturn &nopClient{}\n}\n\n// NewNopCollection creates a new nop collection\nfunc NewNopCollection() *Collection {\n\treturn NewCollection(&nopClient{}, log.NewNoop())\n}\n"
  },
  {
    "path": "common/dynamicconfig/quotas/dynamicratelimiterfactory.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// NewSimpleDynamicRateLimiterFactory creates a new LimiterFactory which creates\n// a new DynamicRateLimiter for each domain, the RPS for the DynamicRateLimiter is given by the dynamic config\nfunc NewSimpleDynamicRateLimiterFactory(rps dynamicproperties.IntPropertyFnWithDomainFilter) quotas.LimiterFactory {\n\treturn dynamicRateLimiterFactory{\n\t\trps: rps,\n\t}\n}\n\ntype dynamicRateLimiterFactory struct {\n\trps dynamicproperties.IntPropertyFnWithDomainFilter\n}\n\n// GetLimiter returns a new Limiter for the given domain\nfunc (f dynamicRateLimiterFactory) GetLimiter(domain string) quotas.Limiter {\n\treturn quotas.NewDynamicRateLimiter(func() float64 { return float64(f.rps(domain)) })\n}\n"
  },
  {
    "path": "common/dynamicconfig/quotas/dynamicratelimiterfactory_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"golang.org/x/time/rate\"\n)\n\nfunc TestNewSimpleDynamicRateLimiterFactory(t *testing.T) {\n\tconst _testDomain = \"test-domain\"\n\n\tfactory := NewSimpleDynamicRateLimiterFactory(func(domain string) int {\n\t\tassert.Equal(t, _testDomain, domain)\n\t\treturn 100\n\t})\n\n\tlimiter := factory.GetLimiter(_testDomain)\n\n\tassert.Equal(t, rate.Limit(100), limiter.Limit())\n}\n"
  },
  {
    "path": "common/dynamicconfig/quotas/fallbackdynamicratelimiterfactory.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// NewFallbackDynamicRateLimiterFactory is used to create a Limiter for a given domain\n// the created Limiter will use the primary dynamic config if it is set\n// otherwise it will use the secondary dynamic config\nfunc NewFallbackDynamicRateLimiterFactory(\n\tprimary dynamicproperties.IntPropertyFnWithDomainFilter,\n\tsecondary dynamicproperties.IntPropertyFn,\n) quotas.LimiterFactory {\n\treturn fallbackDynamicRateLimiterFactory{\n\t\tprimary:   primary,\n\t\tsecondary: secondary,\n\t}\n}\n\ntype fallbackDynamicRateLimiterFactory struct {\n\tprimary dynamicproperties.IntPropertyFnWithDomainFilter\n\t// secondary is used when primary is not set\n\tsecondary dynamicproperties.IntPropertyFn\n}\n\n// GetLimiter returns a new Limiter for the given domain\nfunc (f fallbackDynamicRateLimiterFactory) GetLimiter(domain string) quotas.Limiter {\n\treturn quotas.NewDynamicRateLimiter(func() float64 {\n\t\tif primaryLimit := f.primary(domain); primaryLimit > 0 {\n\t\t\treturn float64(primaryLimit)\n\t\t}\n\t\treturn float64(f.secondary())\n\t})\n}\n"
  },
  {
    "path": "common/dynamicconfig/quotas/fallbackdynamicratelimiterfactory_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\nfunc TestNewFallbackDynamicRateLimiterFactory(t *testing.T) {\n\tfactory := NewFallbackDynamicRateLimiterFactory(\n\t\tfunc(string) int { return 2 },\n\t\tfunc(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t)\n\n\tlimiter := factory.GetLimiter(\"TestDomainName\")\n\n\t// The limiter should accept 2 requests per second\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, false, limiter.Allow())\n}\n\nfunc TestNewFallbackDynamicRateLimiterFactoryFallback(t *testing.T) {\n\tfactory := NewFallbackDynamicRateLimiterFactory(\n\t\tfunc(string) int { return 0 },\n\t\tfunc(opts ...dynamicproperties.FilterOption) int { return 2 },\n\t)\n\n\tlimiter := factory.GetLimiter(\"TestDomainName\")\n\n\t// The limiter should accept 2 requests per second\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, false, limiter.Allow())\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/backoff.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage bulk\n\nimport (\n\t\"math\"\n\t\"math/rand\"\n\t\"time\"\n)\n\n// exponentialBackoff implements the simple exponential backoff described by\n// Douglas Thain at http://dthain.blogspot.de/2009/02/exponential-backoff-in-distributed.html.\n// TODO https://github.com/uber/cadence/issues/3676\ntype exponentialBackoff struct {\n\tinitialTimeout float64 // initial timeout (in msec)\n\tfactor         float64 // exponential factor (e.g. 2)\n\tmaximumTimeout float64 // maximum timeout (in msec)\n}\n\nvar _ GenericBackoff = (*exponentialBackoff)(nil)\n\n// NewExponentialBackoff returns a exponentialBackoff backoff policy.\n// Use initialTimeout to set the first/minimal interval\n// and maxTimeout to set the maximum wait interval.\nfunc NewExponentialBackoff(initialTimeout, maxTimeout time.Duration) GenericBackoff {\n\treturn &exponentialBackoff{\n\t\tinitialTimeout: float64(int64(initialTimeout / time.Millisecond)),\n\t\tfactor:         2.0,\n\t\tmaximumTimeout: float64(int64(maxTimeout / time.Millisecond)),\n\t}\n}\n\n// Next implements BackoffFunc for exponentialBackoff.\nfunc (b *exponentialBackoff) Next(retry int) (time.Duration, bool) {\n\tr := 1.0 + rand.Float64() // random number in [1..2]\n\tm := math.Min(r*b.initialTimeout*math.Pow(b.factor, float64(retry)), b.maximumTimeout)\n\tif m >= b.maximumTimeout {\n\t\treturn 0, false\n\t}\n\td := time.Duration(int64(m)) * time.Millisecond\n\treturn d, true\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/bulk.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage bulk\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\nconst UnknownStatusCode = -1\n\ntype GenericBulkableRequestType int\n\nconst (\n\tBulkableIndexRequest GenericBulkableRequestType = iota\n\tBulkableDeleteRequest\n\tBulkableCreateRequest\n)\n\ntype (\n\t// GenericBulkProcessor is a bulk processor\n\tGenericBulkProcessor interface {\n\t\tStart(ctx context.Context) error\n\t\tStop() error\n\t\tClose() error\n\t\tAdd(request *GenericBulkableAddRequest)\n\t\tFlush() error\n\t}\n\n\t// BulkProcessorParameters holds all required and optional parameters for executing bulk service\n\tBulkProcessorParameters struct {\n\t\tName          string\n\t\tNumOfWorkers  int\n\t\tBulkActions   int\n\t\tBulkSize      int\n\t\tFlushInterval time.Duration\n\t\tBackoff       GenericBackoff\n\t\tBeforeFunc    GenericBulkBeforeFunc\n\t\tAfterFunc     GenericBulkAfterFunc\n\t}\n\n\t// GenericBackoff allows callers to implement their own Backoff strategy.\n\tGenericBackoff interface {\n\t\t// Next implements a BackoffFunc.\n\t\tNext(retry int) (time.Duration, bool)\n\t}\n\n\t// GenericBulkBeforeFunc defines the signature of callbacks that are executed\n\t// before a commit to Elasticsearch.\n\tGenericBulkBeforeFunc func(executionId int64, requests []GenericBulkableRequest)\n\n\t// GenericBulkAfterFunc defines the signature of callbacks that are executed\n\t// after a commit to Elasticsearch. The err parameter signals an error.\n\tGenericBulkAfterFunc func(executionId int64, requests []GenericBulkableRequest, response *GenericBulkResponse, err *GenericError)\n\n\t// GenericBulkableRequest is a generic interface to bulkable requests.\n\tGenericBulkableRequest interface {\n\t\tString() string\n\t\tSource() ([]string, error)\n\t}\n\n\t// GenericBulkableAddRequest a struct to hold a bulk request\n\tGenericBulkableAddRequest struct {\n\t\tIndex       string\n\t\tType        string\n\t\tID          string\n\t\tVersionType string\n\t\tVersion     int64\n\t\t// request types can be index, delete or create\n\t\tRequestType GenericBulkableRequestType\n\t\t// should be nil if IsDelete is true\n\t\tDoc interface{}\n\t}\n\n\t// GenericBulkResponse is generic struct of bulk response\n\tGenericBulkResponse struct {\n\t\tTook   int                                   `json:\"took,omitempty\"`\n\t\tErrors bool                                  `json:\"errors,omitempty\"`\n\t\tItems  []map[string]*GenericBulkResponseItem `json:\"items,omitempty\"`\n\t}\n\n\t// GenericError encapsulates error status and details returned from Elasticsearch.\n\tGenericError struct {\n\t\tStatus  int   `json:\"status\"`\n\t\tDetails error `json:\"error,omitempty\"`\n\t}\n\n\t// GenericBulkResponseItem is the result of a single bulk request.\n\tGenericBulkResponseItem struct {\n\t\tIndex   string `json:\"_index,omitempty\"`\n\t\tType    string `json:\"_type,omitempty\"`\n\t\tID      string `json:\"_id,omitempty\"`\n\t\tVersion int64  `json:\"_version,omitempty\"`\n\t\tResult  string `json:\"result,omitempty\"`\n\t\tStatus  int    `json:\"status,omitempty\"`\n\t\t// the error details\n\t\tError interface{}\n\t}\n)\n"
  },
  {
    "path": "common/elasticsearch/bulk/bulk_delete_request.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage bulk\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n)\n\nvar _ GenericBulkableRequest = (*BulkDeleteRequest)(nil)\n\n// BulkDeleteRequest is a request to remove a document from Elasticsearch.\n//\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-bulk.html\n// for details.\ntype BulkDeleteRequest struct {\n\tindex       string\n\ttyp         string\n\tid          string\n\trouting     string\n\tversion     int64\n\tversionType string // default is \"internal\"\n\n\tsource []string\n}\n\ntype bulkDeleteRequestCommand map[string]bulkDeleteRequestCommandOp\n\ntype bulkDeleteRequestCommandOp struct {\n\tIndex       string `json:\"_index,omitempty\"`\n\tID          string `json:\"_id,omitempty\"`\n\tVersion     int64  `json:\"version,omitempty\"`\n\tVersionType string `json:\"version_type,omitempty\"`\n}\n\n// NewBulkDeleteRequest returns a new BulkDeleteRequest.\nfunc NewBulkDeleteRequest() *BulkDeleteRequest {\n\treturn &BulkDeleteRequest{}\n}\n\n// Index specifies the Elasticsearch index to use for this delete request.\n// If unspecified, the index set on the BulkService will be used.\nfunc (r *BulkDeleteRequest) Index(index string) *BulkDeleteRequest {\n\tr.index = index\n\tr.source = nil\n\treturn r\n}\n\n// Type specifies the Elasticsearch type to use for this delete request.\n// If unspecified, the type set on the BulkService will be used.\nfunc (r *BulkDeleteRequest) Type(typ string) *BulkDeleteRequest {\n\tr.typ = typ\n\tr.source = nil\n\treturn r\n}\n\n// ID specifies the identifier of the document to delete.\nfunc (r *BulkDeleteRequest) ID(id string) *BulkDeleteRequest {\n\tr.id = id\n\tr.source = nil\n\treturn r\n}\n\n// Version indicates the version to be deleted as part of an optimistic\n// concurrency model.\nfunc (r *BulkDeleteRequest) Version(version int64) *BulkDeleteRequest {\n\tr.version = version\n\tr.source = nil\n\treturn r\n}\n\n// VersionType can be \"internal\" (default), \"external\", \"external_gte\",\n// or \"external_gt\".\nfunc (r *BulkDeleteRequest) VersionType(versionType string) *BulkDeleteRequest {\n\tr.versionType = versionType\n\tr.source = nil\n\treturn r\n}\n\n// String returns the on-wire representation of the delete request,\n// concatenated as a single string.\nfunc (r *BulkDeleteRequest) String() string {\n\tlines, err := r.Source()\n\tif err != nil {\n\t\treturn fmt.Sprintf(\"error: %v\", err)\n\t}\n\treturn strings.Join(lines, \"\\n\")\n}\n\n// Source returns the on-wire representation of the delete request,\n// split into an action-and-meta-data line and an (optional) source line.\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-bulk.html\n// for details.\nfunc (r *BulkDeleteRequest) Source() ([]string, error) {\n\tif r.source != nil {\n\t\treturn r.source, nil\n\t}\n\tcommand := bulkDeleteRequestCommand{\n\t\t\"delete\": bulkDeleteRequestCommandOp{\n\t\t\tIndex:       r.index,\n\t\t\tID:          r.id,\n\t\t\tVersion:     r.version,\n\t\t\tVersionType: r.versionType,\n\t\t},\n\t}\n\n\tvar err error\n\tvar body []byte\n\n\tbody, err = json.Marshal(command)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tlines := []string{string(body)}\n\tr.source = lines\n\n\treturn lines, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/bulk_index_request.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage bulk\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n)\n\nvar _ GenericBulkableRequest = (*BulkIndexRequest)(nil)\n\ntype BulkIndexRequest struct {\n\tindex       string\n\ttyp         string\n\tid          string\n\topType      string\n\tversion     *int64\n\tversionType string // default is \"internal\"\n\tdoc         interface{}\n\n\tsource []string\n}\n\ntype bulkIndexRequestCommand map[string]bulkIndexRequestCommandOp\n\ntype bulkIndexRequestCommandOp struct {\n\tIndex       string `json:\"_index,omitempty\"`\n\tID          string `json:\"_id,omitempty\"`\n\tVersion     *int64 `json:\"version,omitempty\"`\n\tVersionType string `json:\"version_type,omitempty\"`\n}\n\n// NewBulkIndexRequest returns a new BulkIndexRequest.\n// The operation type is \"index\" by default.\nfunc NewBulkIndexRequest() *BulkIndexRequest {\n\treturn &BulkIndexRequest{\n\t\topType: \"index\",\n\t}\n}\n\n// Index specifies the Elasticsearch index to use for this index request.\n// If unspecified, the index set on the BulkService will be used.\nfunc (r *BulkIndexRequest) Index(index string) *BulkIndexRequest {\n\tr.index = index\n\tr.source = nil\n\treturn r\n}\n\n// Type specifies the Elasticsearch type to use for this index request.\n// If unspecified, the type set on the BulkService will be used.\nfunc (r *BulkIndexRequest) Type(typ string) *BulkIndexRequest {\n\tr.typ = typ\n\tr.source = nil\n\treturn r\n}\n\n// ID specifies the identifier of the document to index.\nfunc (r *BulkIndexRequest) ID(id string) *BulkIndexRequest {\n\tr.id = id\n\tr.source = nil\n\treturn r\n}\n\n// OpType specifies if this request should follow create-only or upsert\n// behavior. This follows the OpType of the standard document index API.\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-index_.html#operation-type\n// for details.\nfunc (r *BulkIndexRequest) OpType(opType string) *BulkIndexRequest {\n\tr.opType = opType\n\tr.source = nil\n\treturn r\n}\n\n// Version indicates the version of the document as part of an optimistic\n// concurrency model.\nfunc (r *BulkIndexRequest) Version(version int64) *BulkIndexRequest {\n\tr.version = &version\n\tr.source = nil\n\treturn r\n}\n\n// VersionType specifies how versions are created. It can be e.g. internal,\n// external, external_gte, or force.\n//\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-index_.html#index-versioning\n// for details.\nfunc (r *BulkIndexRequest) VersionType(versionType string) *BulkIndexRequest {\n\tr.versionType = versionType\n\tr.source = nil\n\treturn r\n}\n\n// Doc specifies the document to index.\nfunc (r *BulkIndexRequest) Doc(doc interface{}) *BulkIndexRequest {\n\tr.doc = doc\n\tr.source = nil\n\treturn r\n}\n\n// String returns the on-wire representation of the index request,\n// concatenated as a single string.\nfunc (r *BulkIndexRequest) String() string {\n\tlines, err := r.Source()\n\tif err != nil {\n\t\treturn fmt.Sprintf(\"error: %v\", err)\n\t}\n\treturn strings.Join(lines, \"\\n\")\n}\n\n// Source returns the on-wire representation of the index request,\n// split into an action-and-meta-data line and an (optional) source line.\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-bulk.html\n// for details.\nfunc (r *BulkIndexRequest) Source() ([]string, error) {\n\t// { \"index\" : { \"_index\" : \"test\", \"_type\" : \"type1\", \"_id\" : \"1\" } }\n\t// { \"field1\" : \"value1\" }\n\n\tif r.source != nil {\n\t\treturn r.source, nil\n\t}\n\n\tlines := make([]string, 2)\n\n\tindexCommand := bulkIndexRequestCommandOp{\n\t\tIndex:       r.index,\n\t\tID:          r.id,\n\t\tVersion:     r.version,\n\t\tVersionType: r.versionType,\n\t}\n\tcommand := bulkIndexRequestCommand{\n\t\tr.opType: indexCommand,\n\t}\n\n\tvar err error\n\tvar body []byte\n\tbody, err = json.Marshal(command)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tlines[0] = string(body)\n\n\t// \"field1\" ...\n\tif r.doc != nil {\n\t\tbody, err := json.Marshal(r.doc)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tlines[1] = string(body)\n\t} else {\n\t\tlines[1] = \"{}\"\n\t}\n\n\tr.source = lines\n\treturn lines, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/bulk_update_request.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage bulk\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n)\n\nvar _ GenericBulkableRequest = (*BulkUpdateRequest)(nil)\n\n// BulkUpdateRequest is a request to update a document in Elasticsearch.\n//\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-bulk.html\n// for details.\ntype BulkUpdateRequest struct {\n\tindex string\n\tid    string\n\n\tversion     int64\n\tversionType string // default is \"internal\"\n\tdoc         interface{}\n\n\tsource []string\n}\n\ntype bulkUpdateRequestCommand map[string]bulkUpdateRequestCommandOp\n\ntype bulkUpdateRequestCommandOp struct {\n\tIndex       string `json:\"_index,omitempty\"`\n\tID          string `json:\"_id,omitempty\"`\n\tVersion     int64  `json:\"version,omitempty\"`\n\tVersionType string `json:\"version_type,omitempty\"`\n}\n\ntype bulkUpdateRequestCommandData struct {\n\tDoc    interface{} `json:\"doc,omitempty\"`\n\tUpsert interface{} `json:\"upsert,omitempty\"`\n\tSource *bool       `json:\"_source,omitempty\"`\n}\n\n// NewBulkUpdateRequest returns a new BulkUpdateRequest.\nfunc NewBulkUpdateRequest() *BulkUpdateRequest {\n\treturn &BulkUpdateRequest{}\n}\n\n// Index specifies the Elasticsearch index to use for this update request.\n// If unspecified, the index set on the BulkService will be used.\nfunc (r *BulkUpdateRequest) Index(index string) *BulkUpdateRequest {\n\tr.index = index\n\tr.source = nil\n\treturn r\n}\n\n// ID specifies the identifier of the document to update.\nfunc (r *BulkUpdateRequest) ID(id string) *BulkUpdateRequest {\n\tr.id = id\n\tr.source = nil\n\treturn r\n}\n\n// Version indicates the version of the document as part of an optimistic\n// concurrency model.\nfunc (r *BulkUpdateRequest) Version(version int64) *BulkUpdateRequest {\n\tr.version = version\n\tr.source = nil\n\treturn r\n}\n\n// VersionType can be \"internal\" (default), \"external\", \"external_gte\",\n// or \"external_gt\".\nfunc (r *BulkUpdateRequest) VersionType(versionType string) *BulkUpdateRequest {\n\tr.versionType = versionType\n\tr.source = nil\n\treturn r\n}\n\n// Doc specifies the updated document.\nfunc (r *BulkUpdateRequest) Doc(doc interface{}) *BulkUpdateRequest {\n\tr.doc = doc\n\tr.source = nil\n\treturn r\n}\n\n// String returns the on-wire representation of the update request,\n// concatenated as a single string.\nfunc (r *BulkUpdateRequest) String() string {\n\tlines, err := r.Source()\n\tif err != nil {\n\t\treturn fmt.Sprintf(\"error: %v\", err)\n\t}\n\treturn strings.Join(lines, \"\\n\")\n}\n\n// Source returns the on-wire representation of the update request,\n// split into an action-and-meta-data line and an (optional) source line.\n// See https://www.elastic.co/guide/en/elasticsearch/reference/7.0/docs-bulk.html\n// for details.\nfunc (r *BulkUpdateRequest) Source() ([]string, error) {\n\t// { \"update\" : { \"_index\" : \"test\", \"_type\" : \"type1\", \"_id\" : \"1\", ... } }\n\t// { \"doc\" : { \"field1\" : \"value1\", ... } }\n\t// or\n\t// { \"update\" : { \"_index\" : \"test\", \"_type\" : \"type1\", \"_id\" : \"1\", ... } }\n\t// { \"script\" : { ... } }\n\n\tif r.source != nil {\n\t\treturn r.source, nil\n\t}\n\n\tlines := make([]string, 2)\n\n\t// \"update\" ...\n\tupdateCommand := bulkUpdateRequestCommandOp{\n\t\tIndex:       r.index,\n\t\tID:          r.id,\n\t\tVersion:     r.version,\n\t\tVersionType: r.versionType,\n\t}\n\tcommand := bulkUpdateRequestCommand{\n\t\t\"update\": updateCommand,\n\t}\n\n\tvar err error\n\tvar body []byte\n\n\tbody, err = json.Marshal(command)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tlines[0] = string(body)\n\n\t// 2nd line: {\"doc\" : { ... }} or {\"script\": {...}}\n\tdata := bulkUpdateRequestCommandData{\n\t\tDoc: r.doc,\n\t}\n\n\t// encoding/json\n\tbody, err = json.Marshal(data)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tlines[1] = string(body)\n\n\tr.source = lines\n\treturn lines, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/mocks/GenericBulkProcessor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v0.0.0-dev. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tbulk \"github.com/uber/cadence/common/elasticsearch/bulk\"\n)\n\n// GenericBulkProcessor is an autogenerated mock type for the GenericBulkProcessor type\ntype GenericBulkProcessor struct {\n\tmock.Mock\n}\n\n// Add provides a mock function with given fields: request\nfunc (_m *GenericBulkProcessor) Add(request *bulk.GenericBulkableAddRequest) {\n\t_m.Called(request)\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *GenericBulkProcessor) Close() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Flush provides a mock function with given fields:\nfunc (_m *GenericBulkProcessor) Flush() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Start provides a mock function with given fields: ctx\nfunc (_m *GenericBulkProcessor) Start(ctx context.Context) error {\n\tret := _m.Called(ctx)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context) error); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Stop provides a mock function with given fields:\nfunc (_m *GenericBulkProcessor) Stop() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/elasticsearch/bulk/mocks/GenericBulkableRequest.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport mock \"github.com/stretchr/testify/mock\"\n\n// GenericBulkableRequest is an autogenerated mock type for the GenericBulkableRequest type\ntype GenericBulkableRequest struct {\n\tmock.Mock\n}\n\n// Source provides a mock function with given fields:\nfunc (_m *GenericBulkableRequest) Source() ([]string, error) {\n\tret := _m.Called()\n\n\tvar r0 []string\n\tif rf, ok := ret.Get(0).(func() []string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).([]string)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func() error); ok {\n\t\tr1 = rf()\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// String provides a mock function with given fields:\nfunc (_m *GenericBulkableRequest) String() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/elasticsearch/client/client.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage client\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n)\n\n// Client is a generic ES client implementation.\n// This interface allows to use different Elasticsearch and OpenSearch versions\n// without exposing implementation details and structs\ntype Client interface {\n\t// ClearScroll clears the search context and results for a scrolling search.\n\tClearScroll(ctx context.Context, scrollID string) error\n\t// Count returns number of document matches by given query\n\tCount(ctx context.Context, index, body string) (int64, error)\n\t// CreateIndex creates index with given name\n\tCreateIndex(ctx context.Context, index string) error\n\t// IsNotFoundError checks if error is a \"not found\"\n\tIsNotFoundError(err error) bool\n\t// PutMapping updates Client with new field mapping\n\tPutMapping(ctx context.Context, index, body string) error\n\t// RunBulkProcessor starts bulk indexing processor\n\t// @TODO consider to extract Bulk Processor as a separate entity\n\tRunBulkProcessor(ctx context.Context, p *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error)\n\t// Scroll retrieves the next batch of results for a scrolling search.\n\tScroll(ctx context.Context, index, body, scrollID string) (*Response, error)\n\t// Search returns Elasticsearch hit bytes and additional metadata\n\tSearch(ctx context.Context, index, body string) (*Response, error)\n}\n\n// Response is used to pass data retrieved from Elasticsearch/OpenSearch to upper layer\ntype Response struct {\n\tTookInMillis int64\n\tTotalHits    int64\n\tHits         *SearchHits\n\tAggregations map[string]json.RawMessage\n\tSort         []interface{}\n\tScrollID     string\n}\n\n// SearchHits specifies the list of search hits.\ntype SearchHits struct {\n\tTotalHits *TotalHits   // total number of hits found\n\tHits      []*SearchHit // the actual hits returned\n}\n\n// TotalHits specifies total number of hits and its relation\ntype TotalHits struct {\n\tValue int64 // value of the total hit count\n}\n\n// SearchHit is a single hit.\ntype SearchHit struct {\n\tIndex  string          // index name\n\tID     string          // external or internal\n\tSort   []interface{}   // sort information\n\tSource json.RawMessage // stored document source\n}\n"
  },
  {
    "path": "common/elasticsearch/client/os2/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage os2\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/aws/aws-sdk-go/aws\"\n\t\"github.com/aws/aws-sdk-go/aws/session\"\n\t\"github.com/opensearch-project/opensearch-go/v4\"\n\tosapi \"github.com/opensearch-project/opensearch-go/v4/opensearchapi\"\n\t\"github.com/opensearch-project/opensearch-go/v4/opensearchtransport\"\n\trequestsigner \"github.com/opensearch-project/opensearch-go/v4/signer/aws\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/elasticsearch/client\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype (\n\t// OS2 implements Client\n\tOS2 struct {\n\t\tclient  *osapi.Client\n\t\tlogger  log.Logger\n\t\tdecoder *NumberDecoder\n\t}\n\n\terrorDetails struct {\n\t\tType   string `json:\"type\"`\n\t\tReason string `json:\"reason\"`\n\t\tIndex  string `json:\"index,omitempty\"`\n\t}\n\n\t// response holds data retrieved from OpenSearch\n\tresponse struct {\n\t\tTookInMillis int64 `json:\"took,omitempty\"`\n\t\tHits         *searchHits\n\t\tAggregations map[string]json.RawMessage `json:\"aggregations,omitempty\"`\n\t\tSort         []interface{}              `json:\"sort,omitempty\"` // sort information\n\t\tScrollID     string                     `json:\"_scroll_id,omitempty\"`\n\t}\n\n\t// searchHits specifies the list of search hits.\n\tsearchHits struct {\n\t\tTotalHits *totalHits   `json:\"total,omitempty\"` // total number of hits found\n\t\tHits      []*searchHit `json:\"hits,omitempty\"`  // the actual hits returned\n\t}\n\n\t// totalHits specifies total number of hits and its relation\n\ttotalHits struct {\n\t\tValue int64 `json:\"value\"` // value of the total hit count\n\t}\n\n\t// searchHit is a single hit.\n\tsearchHit struct {\n\t\tIndex  string          `json:\"_index,omitempty\"`  // index name\n\t\tID     string          `json:\"_id,omitempty\"`     // external or internal\n\t\tSort   []any           `json:\"sort,omitempty\"`    // sort information\n\t\tSource json.RawMessage `json:\"_source,omitempty\"` // stored document source\n\t}\n\n\tconvertLogger struct {\n\t\tlogger log.Logger\n\t}\n)\n\nvar _ opensearchtransport.Logger = (*convertLogger)(nil)\n\nfunc (c convertLogger) LogRoundTrip(request *http.Request, h *http.Response, err error, t time.Time, duration time.Duration) error {\n\t// req and resp bodies must not be touched because we have not enabled them, and doing so might affect the request\n\tif err != nil {\n\t\t// possible future enhancement: bulk failures are MUCH more relevant than query failures like timeouts.\n\t\t// this can probably be figured out by checking the request URL, if the volume proves too high.\n\t\tc.logger.Error(\n\t\t\t\"opensearch request failed\",\n\t\t\ttag.Error(err),\n\t\t\ttag.Dynamic(\"request_uri\", request.URL.String()),\n\t\t\ttag.Dynamic(\"request_method\", request.Method),\n\t\t\ttag.Dynamic(\"response_code\", h.StatusCode),\n\t\t\ttag.Duration(duration),\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc (c convertLogger) RequestBodyEnabled() bool  { return false }\nfunc (c convertLogger) ResponseBodyEnabled() bool { return false }\n\n// NewClient returns a new implementation of GenericClient\nfunc NewClient(\n\tconnectConfig *config.ElasticSearchConfig,\n\tlogger log.Logger,\n\ttlsClient *http.Client,\n) (*OS2, error) {\n\n\tosconfig := osapi.Config{\n\t\tClient: opensearch.Config{\n\t\t\tAddresses:    []string{connectConfig.URL.String()},\n\t\t\tMaxRetries:   5,\n\t\t\tRetryBackoff: func(i int) time.Duration { return time.Duration(i) * 100 * time.Millisecond },\n\t\t\tLogger:       &convertLogger{logger: logger},\n\t\t},\n\t}\n\n\tif len(connectConfig.CustomHeaders) > 0 {\n\t\tosconfig.Client.Header = http.Header{}\n\n\t\tfor key, value := range connectConfig.CustomHeaders {\n\t\t\tosconfig.Client.Header.Set(key, value)\n\t\t}\n\t}\n\n\t// DiscoverNodesOnStart is false by default. Turn it on only when disable sniff is set to False in ES config\n\tif !connectConfig.DisableSniff {\n\t\tosconfig.Client.DiscoverNodesOnStart = true\n\t}\n\n\tif connectConfig.AWSSigning.Enable {\n\t\tcredentials, region, err := connectConfig.AWSSigning.GetCredentials()\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"getting aws credentials: %w\", err)\n\t\t}\n\n\t\tsessionOptions := session.Options{\n\t\t\tConfig: aws.Config{\n\t\t\t\tRegion:      region,\n\t\t\t\tCredentials: credentials,\n\t\t\t},\n\t\t}\n\n\t\tsigner, err := requestsigner.NewSigner(sessionOptions)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"creating aws signer: %w\", err)\n\t\t}\n\n\t\tosconfig.Client.Signer = signer\n\t}\n\n\tif tlsClient != nil {\n\t\tosconfig.Client.Transport = tlsClient.Transport\n\t\tlogger.Info(\"Using TLS client\")\n\t}\n\n\tosClient, err := osapi.NewClient(osconfig)\n\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"creating OpenSearch client: %w\", err)\n\t}\n\n\t// initial health check\n\tctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)\n\tdefer cancel()\n\tresp, err := osClient.Ping(ctx, nil /*PingReq*/)\n\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"OpenSearch client unable to ping: %w\", err)\n\t}\n\n\tif resp.IsError() {\n\t\treturn nil, fmt.Errorf(\"OpenSearch client received error on ping: %s\", resp)\n\t}\n\n\treturn &OS2{\n\t\tclient:  osClient,\n\t\tlogger:  logger,\n\t\tdecoder: &NumberDecoder{},\n\t}, nil\n}\n\nfunc (c *OS2) IsNotFoundError(err error) bool {\n\n\tvar clientErr *opensearch.StructError\n\tif errors.As(err, &clientErr) {\n\t\treturn clientErr.Status == http.StatusNotFound\n\t}\n\treturn false\n}\n\nfunc (c *OS2) PutMapping(ctx context.Context, index, body string) error {\n\n\treq := osapi.MappingPutReq{\n\t\tIndices: []string{index},\n\t\tBody:    strings.NewReader(body),\n\t}\n\n\t_, err := c.client.Indices.Mapping.Put(ctx, req)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"OpenSearch PutMapping: %w\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (c *OS2) CreateIndex(ctx context.Context, index string) error {\n\treq := osapi.IndicesCreateReq{\n\t\tIndex: index,\n\t}\n\n\t_, err := c.client.Indices.Create(ctx, req)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (c *OS2) Count(ctx context.Context, index, query string) (int64, error) {\n\n\treq := &osapi.IndicesCountReq{\n\t\tIndices: []string{index},\n\t\tBody:    strings.NewReader(query),\n\t}\n\tresp, err := c.client.Indices.Count(ctx, req)\n\tif err != nil {\n\t\treturn 0, fmt.Errorf(\"OpenSearch Count: %w\", err)\n\t}\n\n\treturn int64(resp.Count), nil\n}\n\nfunc (c *OS2) ClearScroll(ctx context.Context, scrollID string) error {\n\t_, err := c.client.Scroll.Delete(ctx, osapi.ScrollDeleteReq{\n\t\tScrollIDs: []string{scrollID},\n\t})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"OpenSearch ClearScroll: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc (c *OS2) Scroll(ctx context.Context, index, body, scrollID string) (*client.Response, error) {\n\n\tvar scrollResp *osapi.ScrollGetResp\n\tvar searchResp *osapi.SearchResp\n\tvar osResponse response\n\tvar respBody io.ReadCloser\n\tvar searchErr error\n\t// handle scroll id get call\n\tif len(scrollID) != 0 {\n\t\tscrollResp, searchErr = c.client.Scroll.Get(ctx, osapi.ScrollGetReq{\n\t\t\tScrollID: scrollID,\n\t\t\tParams: osapi.ScrollGetParams{\n\t\t\t\tScroll: time.Minute,\n\t\t\t\t// do not set scroll ID here as it will be added to the params and scroll ID can be excessively long\n\t\t\t},\n\t\t})\n\t\tif searchErr != nil {\n\t\t\treturn nil, fmt.Errorf(\"opensearch scroll search error: %w\", searchErr)\n\t\t}\n\t\tif scrollResp.Inspect().Response == nil {\n\t\t\treturn nil, fmt.Errorf(\"OpenSearch scroll search response nil\")\n\t\t}\n\t\trespBody = scrollResp.Inspect().Response.Body\n\n\t} else {\n\t\t// when scrollID is not passed, it is normal search request\n\t\tsearchResp, searchErr = c.client.Search(ctx, &osapi.SearchReq{\n\t\t\tIndices: []string{index},\n\t\t\tBody:    strings.NewReader(body),\n\t\t\tParams: osapi.SearchParams{\n\t\t\t\tScroll: time.Minute,\n\t\t\t},\n\t\t})\n\t\tif searchErr != nil {\n\t\t\treturn nil, fmt.Errorf(\"opensearch scroll search error: %w\", searchErr)\n\t\t}\n\t\tif searchResp.Inspect().Response == nil {\n\t\t\treturn nil, fmt.Errorf(\"OpenSearch scroll search response nil\")\n\t\t}\n\t\trespBody = searchResp.Inspect().Response.Body\n\t}\n\n\tbodyBytes, err := io.ReadAll(respBody)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read scroll search response body: %w\", err)\n\t}\n\tif err := c.decoder.Decode(bytes.NewReader(bodyBytes), &osResponse); err != nil && !errors.Is(err, io.EOF) {\n\t\treturn nil, fmt.Errorf(\"decoding OpenSearch scroll response to Response: %w\", err)\n\t}\n\n\tvar totalHits int64\n\tvar hits []*client.SearchHit\n\t// no more hits\n\tif osResponse.Hits == nil || len(osResponse.Hits.Hits) == 0 {\n\t\treturn &client.Response{\n\t\t\tScrollID:     osResponse.ScrollID,\n\t\t\tTookInMillis: osResponse.TookInMillis,\n\t\t\tHits:         &client.SearchHits{Hits: hits},\n\t\t}, io.EOF\n\t}\n\n\tfor _, h := range osResponse.Hits.Hits {\n\t\thits = append(hits, &client.SearchHit{Source: h.Source})\n\t}\n\n\tif osResponse.Hits.TotalHits != nil {\n\t\ttotalHits = osResponse.Hits.TotalHits.Value\n\t}\n\n\treturn &client.Response{\n\t\tTookInMillis: osResponse.TookInMillis,\n\t\tTotalHits:    totalHits,\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tAggregations: osResponse.Aggregations,\n\t\tScrollID:     osResponse.ScrollID,\n\t}, nil\n\n}\n\nfunc (c *OS2) Search(ctx context.Context, index, body string) (*client.Response, error) {\n\n\tresp, err := c.client.Search(ctx, &osapi.SearchReq{\n\t\tIndices: []string{index},\n\t\tBody:    strings.NewReader(body),\n\t})\n\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"OpenSearch Search error: %w\", err)\n\t}\n\tif resp.Inspect().Response == nil {\n\t\treturn nil, fmt.Errorf(\"OpenSearch search response nil\")\n\t}\n\tvar osResponse response\n\tbodyBytes, err := io.ReadAll(resp.Inspect().Response.Body)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read OpenSearch scroll search response body: %w\", err)\n\t}\n\tif err := c.decoder.Decode(bytes.NewReader(bodyBytes), &osResponse); err != nil && !errors.Is(err, io.EOF) {\n\t\treturn nil, fmt.Errorf(\"decoding Opensearch result to Response: %w\", err)\n\t}\n\tvar hits []*client.SearchHit\n\tvar sort []interface{}\n\tvar totalHits int64\n\n\tif osResponse.Hits != nil && osResponse.Hits.TotalHits != nil {\n\t\ttotalHits = osResponse.Hits.TotalHits.Value\n\t\tfor _, h := range osResponse.Hits.Hits {\n\t\t\tsort = h.Sort\n\t\t\thits = append(hits, &client.SearchHit{Source: h.Source})\n\t\t}\n\t}\n\n\treturn &client.Response{\n\t\tTookInMillis: osResponse.TookInMillis,\n\t\tTotalHits:    totalHits,\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tAggregations: osResponse.Aggregations,\n\t\tSort:         sort,\n\t}, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/client/os2/client_bulk.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage os2\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"io\"\n\n\t\"github.com/opensearch-project/opensearch-go/v4/opensearchapi\"\n\t\"github.com/opensearch-project/opensearch-go/v4/opensearchutil\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// metricsRequest is used for a callback/metrics needs\nvar metricsRequest = []bulk.GenericBulkableRequest{&bulk.BulkIndexRequest{}}\n\ntype bulkProcessor struct {\n\tprocessor opensearchutil.BulkIndexer\n\tbefore    bulk.GenericBulkBeforeFunc\n\tafter     bulk.GenericBulkAfterFunc\n\tlogger    log.Logger\n}\n\nfunc (v *bulkProcessor) Start(ctx context.Context) error {\n\treturn nil\n}\n\nfunc (v *bulkProcessor) Stop() error {\n\treturn v.Close()\n}\nfunc (v *bulkProcessor) Close() error {\n\treturn v.processor.Close(context.Background())\n}\n\nfunc (v *bulkProcessor) Add(request *bulk.GenericBulkableAddRequest) {\n\n\treq := opensearchutil.BulkIndexerItem{\n\t\tIndex:      request.Index,\n\t\tDocumentID: request.ID,\n\t}\n\tvar callBackRequest bulk.GenericBulkableRequest\n\tswitch request.RequestType {\n\tcase bulk.BulkableDeleteRequest:\n\t\treq.Action = \"delete\"\n\t\tcallBackRequest = bulk.NewBulkDeleteRequest().\n\t\t\tID(request.ID).\n\t\t\tIndex(request.Index)\n\tcase bulk.BulkableIndexRequest:\n\t\tbody, err := json.Marshal(request.Doc)\n\t\tif err != nil {\n\t\t\tv.logger.Error(\"marshal bulk index request doc\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\treq.Action = \"index\"\n\t\treq.Version = &request.Version\n\t\treq.VersionType = &request.VersionType\n\t\treq.Body = bytes.NewReader(body)\n\t\tcallBackRequest = bulk.NewBulkIndexRequest().\n\t\t\tID(request.ID).\n\t\t\tIndex(request.Index).\n\t\t\tVersion(request.Version).\n\t\t\tVersionType(request.VersionType).Doc(request.Doc)\n\tcase bulk.BulkableCreateRequest:\n\t\tbody, err := json.Marshal(request.Doc)\n\t\tif err != nil {\n\t\t\tv.logger.Error(\"marshal bulk create request doc\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\tversionType := \"internal\"\n\t\treq.Action = \"create\"\n\t\treq.VersionType = &versionType\n\t\treq.Body = bytes.NewReader(body)\n\t\tcallBackRequest = bulk.NewBulkIndexRequest().ID(request.ID).\n\t\t\tIndex(request.Index).\n\t\t\tVersion(request.Version).\n\t\t\tVersionType(versionType).Doc(request.Doc)\n\n\t}\n\n\treq.OnFailure = func(ctx context.Context, item opensearchutil.BulkIndexerItem, res opensearchapi.BulkRespItem, err error) {\n\t\tv.processCallback(0, callBackRequest, req, res, false, err)\n\t}\n\n\treq.OnSuccess = func(ctx context.Context, item opensearchutil.BulkIndexerItem, res opensearchapi.BulkRespItem) {\n\t\tv.processCallback(0, callBackRequest, req, res, true, nil)\n\t}\n\tif err := v.processor.Add(context.Background(), req); err != nil {\n\t\tv.logger.Error(\"adding bulk request to OpenSearch\", tag.Error(err))\n\t}\n}\n\n// processCallback processes both success and failure scenarios.\nfunc (v *bulkProcessor) processCallback(index int64, callBackRequest bulk.GenericBulkableRequest, req opensearchutil.BulkIndexerItem, res opensearchapi.BulkRespItem, onSuccess bool, err error) {\n\tv.before(0, metricsRequest)\n\n\tgr := []bulk.GenericBulkableRequest{callBackRequest}\n\n\tit := bulk.GenericBulkResponseItem{\n\t\tIndex:   res.Index,\n\t\tID:      res.ID,\n\t\tVersion: int64(res.Version),\n\t\tStatus:  res.Status,\n\t\tError:   res.Error,\n\t}\n\n\tif onSuccess {\n\t\tgbr := bulk.GenericBulkResponse{\n\t\t\tTook:   0,\n\t\t\tErrors: false,\n\t\t\tItems: []map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t{req.Action: &it},\n\t\t\t},\n\t\t}\n\n\t\tv.after(0, gr, &gbr, nil /*No errors here*/)\n\t} else {\n\t\tgbr := bulk.GenericBulkResponse{\n\t\t\tTook:   0,\n\t\t\tErrors: res.Error != nil && (res.Error.Type != \"\" || res.Status > 201),\n\t\t\tItems: []map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t{req.Action: &it},\n\t\t\t},\n\t\t}\n\n\t\tge := bulk.GenericError{\n\t\t\tStatus:  res.Status,\n\t\t\tDetails: err,\n\t\t}\n\n\t\tv.after(0, gr, &gbr, &ge)\n\t}\n}\n\nfunc (v *bulkProcessor) Flush() error {\n\t// Opensearch client doesn't provide flush method\n\treturn nil\n}\n\nfunc (c *OS2) RunBulkProcessor(_ context.Context, parameters *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error) {\n\tprocessor, err := opensearchutil.NewBulkIndexer(opensearchutil.BulkIndexerConfig{\n\t\tClient:        c.client,\n\t\tFlushInterval: parameters.FlushInterval,\n\t\tNumWorkers:    parameters.NumOfWorkers,\n\t})\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &bulkProcessor{\n\t\tprocessor: processor,\n\t\tbefore:    parameters.BeforeFunc,\n\t\tafter:     parameters.AfterFunc,\n\t\tlogger:    c.logger,\n\t}, nil\n}\n\n// NumberDecoder uses json.NewDecoder, with UseNumber() enabled, from\n// the Go standard library to decode JSON data.\ntype NumberDecoder struct{}\n\nfunc (u *NumberDecoder) UnmarshalFromReader(reader io.Reader, response *opensearchapi.BulkRespItem) error {\n\tdec := json.NewDecoder(reader)\n\tdec.UseNumber()\n\treturn dec.Decode(response)\n}\n\nfunc (u *NumberDecoder) Decode(reader io.Reader, v interface{}) error {\n\tdec := json.NewDecoder(reader)\n\tdec.UseNumber()\n\treturn dec.Decode(v)\n}\n"
  },
  {
    "path": "common/elasticsearch/client/os2/client_bulk_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage os2\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"io\"\n\t\"testing\"\n\n\t\"github.com/opensearch-project/opensearch-go/v4/opensearchapi\"\n\t\"github.com/opensearch-project/opensearch-go/v4/opensearchutil\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestStart(t *testing.T) {\n\tctx := context.Background()\n\tprocessor := &bulkProcessor{}\n\tif err := processor.Start(ctx); err != nil {\n\t\tt.Errorf(\"Start() error = %v, wantErr %v\", err, nil)\n\t}\n}\n\nfunc TestStopAndClose(t *testing.T) {\n\tosClient, testServer := getSecureMockOS2Client(t, nil, false)\n\tdefer testServer.Close()\n\n\tparams := bulk.BulkProcessorParameters{\n\t\tFlushInterval: 1,\n\t\tNumOfWorkers:  1,\n\t}\n\tprocessor, err := osClient.RunBulkProcessor(context.Background(), &params)\n\tassert.NoError(t, err)\n\n\tif err := processor.Stop(); err != nil {\n\t\tt.Errorf(\"Stop() error = %v, wantErr %v\", err, nil)\n\t}\n}\n\nfunc TestFlush(t *testing.T) {\n\tprocessor := &bulkProcessor{}\n\tif err := processor.Flush(); err != nil {\n\t\tt.Errorf(\"Flush() error = %v, wantErr %v\", err, nil)\n\t}\n}\n\nfunc TestAddDeleteRequest(t *testing.T) {\n\ttestcases := []struct {\n\t\tname      string\n\t\tinput     bulk.GenericBulkableAddRequest\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname: \"delete request\",\n\t\t\tinput: bulk.GenericBulkableAddRequest{\n\t\t\t\tRequestType: bulk.BulkableDeleteRequest,\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersion:     1,\n\t\t\t\tVersionType: \"external\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"index request success\",\n\t\t\tinput: bulk.GenericBulkableAddRequest{\n\t\t\t\tRequestType: bulk.BulkableIndexRequest,\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tDoc:         map[string]interface{}{\"field\": \"value\"},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"create request success\",\n\t\t\tinput: bulk.GenericBulkableAddRequest{\n\t\t\t\tRequestType: bulk.BulkableCreateRequest,\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tDoc:         map[string]interface{}{\"field\": \"value\"},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"create request failure\",\n\t\t\tinput: bulk.GenericBulkableAddRequest{\n\t\t\t\tRequestType: bulk.BulkableCreateRequest,\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tDoc:         func() {},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"index request failure\",\n\t\t\tinput: bulk.GenericBulkableAddRequest{\n\t\t\t\tRequestType: bulk.BulkableIndexRequest,\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tDoc:         func() {},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testcases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tosClient, testServer := getSecureMockOS2Client(t, nil, false)\n\t\t\tdefer testServer.Close()\n\n\t\t\tparams := bulk.BulkProcessorParameters{\n\t\t\t\tFlushInterval: 1,\n\t\t\t\tNumOfWorkers:  1,\n\t\t\t}\n\t\t\tprocessor, err := osClient.RunBulkProcessor(context.Background(), &params)\n\t\t\tassert.NoError(t, err)\n\t\t\tprocessor.Add(&tc.input)\n\t\t\tprocessor.Stop()\n\t\t})\n\t}\n}\n\nfunc TestProcessCallback(t *testing.T) {\n\tosClient, testServer := getSecureMockOS2Client(t, nil, false)\n\tdefer testServer.Close()\n\tprocessor, _ := opensearchutil.NewBulkIndexer(opensearchutil.BulkIndexerConfig{\n\t\tClient:        osClient.client,\n\t\tFlushInterval: 1,\n\t\tNumWorkers:    1,\n\t})\n\tbulkProcessor := &bulkProcessor{\n\t\tprocessor: processor,\n\t\tbefore:    beforeFunc,\n\t\tafter:     func(int64, []bulk.GenericBulkableRequest, *bulk.GenericBulkResponse, *bulk.GenericError) {},\n\t}\n\n\tcallBackRequest := bulk.NewBulkIndexRequest().ID(\"test-id\").Index(\"test-index\").Doc(map[string]interface{}{\"field\": \"value\"})\n\treq := opensearchutil.BulkIndexerItem{Index: \"test-index\", DocumentID: \"test-id\"}\n\tres := opensearchapi.BulkRespItem{Index: \"test-index\", ID: \"test-id\", Status: 200}\n\n\ttests := []struct {\n\t\tname        string\n\t\tonSuccess   bool\n\t\terrorPassed error\n\t\texpectError bool\n\t}{\n\t\t{\"Success Callback\", true, nil, false},\n\t\t{\"Failure Callback\", false, errors.New(\"test error\"), true},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tbulkProcessor.processCallback(0, callBackRequest, req, res, test.onSuccess, test.errorPassed)\n\t\t})\n\t}\n}\n\nfunc beforeFunc(executionID int64, requests []bulk.GenericBulkableRequest) {}\n\nfunc TestRunBulkProcessor(t *testing.T) {\n\tosClient, testServer := getSecureMockOS2Client(t, nil, false)\n\tdefer testServer.Close()\n\n\ttestcases := []struct {\n\t\tname      string\n\t\tinput     bulk.BulkProcessorParameters\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname: \"normal\",\n\t\t\tinput: bulk.BulkProcessorParameters{\n\t\t\t\tFlushInterval: 1,\n\t\t\t\tNumOfWorkers:  1,\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testcases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tprocessor, err := osClient.RunBulkProcessor(context.Background(), &tc.input)\n\t\t\tif tc.expectErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, processor)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUnmarshalFromReader(t *testing.T) {\n\t// test unmarshalFromReader\n\ttestcases := []struct {\n\t\tname      string\n\t\tinput     string\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname:      \"normal\",\n\t\t\tinput:     `{\"status\":200}`,\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testcases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecoder := &NumberDecoder{}\n\t\t\treader := createReaderFromString(tc.input)\n\n\t\t\tvar response opensearchapi.BulkRespItem\n\t\t\terr := decoder.UnmarshalFromReader(reader, &response)\n\t\t\tif tc.expectErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDecode(t *testing.T) {\n\t// test decode\n\ttestcases := []struct {\n\t\tname      string\n\t\tinput     string\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname:      \"normal\",\n\t\t\tinput:     `{\"status\":200}`,\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testcases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecoder := &NumberDecoder{}\n\t\t\treader := createReaderFromString(tc.input)\n\n\t\t\tvar response opensearchapi.BulkRespItem\n\t\t\terr := decoder.Decode(reader, &response)\n\t\t\tif tc.expectErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Function to create a sample reader from a string\nfunc createReaderFromString(s string) io.Reader {\n\treturn bytes.NewBufferString(s)\n}\n\n// MockProcessor is a mock of the processor that will capture the requests.\ntype MockProcessor struct {\n\tmock.Mock\n}\n\nfunc (m *MockProcessor) Add(ctx context.Context, req opensearchutil.BulkIndexerItem) error {\n\targs := m.Called(ctx, req)\n\treturn args.Error(0)\n}\n\nfunc (m *MockProcessor) Close(ctx context.Context) error {\n\targs := m.Called(ctx)\n\treturn args.Error(0)\n}\n\nfunc (m *MockProcessor) Stats() opensearchutil.BulkIndexerStats {\n\targs := m.Called()\n\treturn args.Get(0).(opensearchutil.BulkIndexerStats)\n}\n\nfunc TestBulkProcessorPayload(t *testing.T) {\n\tmockProcessor := new(MockProcessor)\n\tlogger := testlogger.New(t)\n\tv := &bulkProcessor{\n\t\tprocessor: mockProcessor,\n\t\tlogger:    logger,\n\t}\n\n\trequest := &bulk.GenericBulkableAddRequest{\n\t\tIndex:       \"test-index\",\n\t\tID:          \"123\",\n\t\tVersion:     1,\n\t\tVersionType: \"external\",\n\t\tRequestType: bulk.BulkableIndexRequest,\n\t\tDoc: map[string]interface{}{\n\t\t\t\"field1\": \"value1\",\n\t\t},\n\t}\n\n\texpectedBody, err := json.Marshal(request.Doc)\n\tassert.NoError(t, err)\n\texpectedReq := opensearchutil.BulkIndexerItem{\n\t\tIndex:       \"test-index\",\n\t\tDocumentID:  \"123\",\n\t\tAction:      \"index\",\n\t\tVersion:     &request.Version,\n\t\tVersionType: &request.VersionType,\n\t\tBody:        bytes.NewReader(expectedBody),\n\t}\n\n\tmockProcessor.On(\"Add\", mock.Anything, mock.Anything).Run(func(args mock.Arguments) {\n\t\tactualReq := args.Get(1).(opensearchutil.BulkIndexerItem)\n\n\t\tassert.Equal(t, expectedReq.Index, actualReq.Index)\n\t\tassert.Equal(t, expectedReq.DocumentID, actualReq.DocumentID)\n\t\tassert.Equal(t, expectedReq.Action, actualReq.Action)\n\t\tassert.Equal(t, expectedReq.Version, actualReq.Version)\n\t\tassert.Equal(t, expectedReq.VersionType, actualReq.VersionType)\n\n\t\tvar expectedDoc, actualDoc map[string]interface{}\n\t\tassert.NoError(t, json.NewDecoder(expectedReq.Body).Decode(&expectedDoc))\n\t\tassert.NoError(t, json.NewDecoder(actualReq.Body).Decode(&actualDoc))\n\t\tassert.Equal(t, expectedDoc, actualDoc)\n\n\t}).Return(nil)\n\n\tv.Add(request)\n\tmockProcessor.AssertExpectations(t)\n}\n"
  },
  {
    "path": "common/elasticsearch/client/os2/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage os2\n\nimport (\n\t\"context\"\n\t\"crypto/tls\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"net/url\"\n\t\"testing\"\n\n\t\"github.com/opensearch-project/opensearch-go/v4\"\n\tosapi \"github.com/opensearch-project/opensearch-go/v4/opensearchapi\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\ntype MockTransport struct{}\n\nfunc (m *MockTransport) Perform(req *http.Request) (*http.Response, error) {\n\t// Simulate a network or connection error\n\treturn nil, fmt.Errorf(\"forced connection error\")\n}\n\nfunc TestNewClient(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestServer := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif r.URL.Path == \"/\" {\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{ \"status\": \"green\" }`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t}))\n\tdefer testServer.Close()\n\turl, err := url.Parse(testServer.URL)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\tbadURL, err := url.Parse(\"http://nonexistent.elasticsearch.server:9200\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\ttests := []struct {\n\t\tname        string\n\t\tconfig      *config.ElasticSearchConfig\n\t\thandlerFunc http.HandlerFunc\n\t\texpectedErr bool\n\t}{\n\t\t{\n\t\t\tname: \"without aws signing config\",\n\t\t\tconfig: &config.ElasticSearchConfig{\n\t\t\t\tURL:          *url,\n\t\t\t\tDisableSniff: false,\n\t\t\t\tCustomHeaders: map[string]string{\n\t\t\t\t\t\"key\": \"value\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"with wrong aws signing config\",\n\t\t\tconfig: &config.ElasticSearchConfig{\n\t\t\t\tURL:          *badURL,\n\t\t\t\tDisableSniff: false,\n\t\t\t\tAWSSigning: config.AWSSigning{\n\t\t\t\t\tEnable: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: true, //will fail to ping os sever\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tclient, err := NewClient(tt.config, logger, testServer.Client())\n\n\t\t\tif !tt.expectedErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, client)\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateIndex(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\thandler   http.HandlerFunc\n\t\texpectErr bool\n\t\tsecure    bool\n\t}{\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.Method == \"PUT\" && r.URL.Path == \"/test-index\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\"acknowledged\": true, \"shards_acknowledged\": true, \"index\": \"test-index\"}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: false,\n\t\t\tsecure:    true,\n\t\t},\n\t\t{\n\t\t\tname: \"error case\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.NotFound(w, r)\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t\tsecure:    true,\n\t\t},\n\t\t{\n\t\t\tname: \"not valid config\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.NotFound(w, r)\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t\tsecure:    false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tt.handler, tt.secure)\n\t\t\tdefer testServer.Close()\n\n\t\t\terr := os2Client.CreateIndex(context.Background(), \"test-index\")\n\t\t\tif tt.expectErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getSecureMockOS2Client(t *testing.T, handler http.HandlerFunc, secure bool) (*OS2, *httptest.Server) {\n\ttestServer := httptest.NewTLSServer(handler)\n\tosConfig := osapi.Config{\n\t\tClient: opensearch.Config{\n\t\t\tAddresses: []string{testServer.URL},\n\t\t},\n\t}\n\tif secure {\n\t\tosConfig.Client.Transport = &http.Transport{\n\t\t\tTLSClientConfig: &tls.Config{\n\t\t\t\tInsecureSkipVerify: true,\n\t\t\t},\n\t\t}\n\t}\n\n\tclient, err := osapi.NewClient(osConfig)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to create open search client: %v\", err)\n\t}\n\tmockClient := &OS2{\n\t\tclient:  client,\n\t\tlogger:  testlogger.New(t),\n\t\tdecoder: &NumberDecoder{},\n\t}\n\tassert.NoError(t, err)\n\treturn mockClient, testServer\n}\n\nfunc TestPutMapping(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\thandler     http.HandlerFunc\n\t\tindex       string\n\t\tbody        string\n\t\texpectedErr bool\n\t}{\n\t\t{\n\t\t\tname: \"Successful PutMapping\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tw.Write([]byte(`{\"acknowledged\": true}`))\n\t\t\t},\n\t\t\tindex:       \"testIndex\",\n\t\t\tbody:        `{\"properties\": {\"field\": {\"type\": \"text\"}}}`,\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Failed PutMapping\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t},\n\t\t\tindex:       \"nonExistentIndex\",\n\t\t\tbody:        `{\"properties\": {\"field\": {\"type\": \"text\"}}}`,\n\t\t\texpectedErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\n\t\t\terr := os2Client.PutMapping(context.Background(), tc.index, tc.body)\n\n\t\t\tif tc.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPutMappingError(t *testing.T) {\n\thandler := func(w http.ResponseWriter, r *http.Request) {\n\t\tw.WriteHeader(http.StatusOK)\n\t}\n\n\tos2Client, testServer := getSecureMockOS2Client(t, http.HandlerFunc(handler), true)\n\tdefer testServer.Close()\n\tos2Client.client.Client.Transport = &MockTransport{}\n\terr := os2Client.PutMapping(context.Background(), \"testIndex\", `{\"properties\": {\"field\": {\"type\": \"text\"}}}`)\n\tassert.Error(t, err)\n}\n\nfunc TestIsNotFoundError(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\thandler  http.HandlerFunc\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname: \"Other error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.Error(w, \"Bad Request\", http.StatusBadRequest)\n\t\t\t}),\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"NotFound error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\tjson.NewEncoder(w).Encode(map[string]interface{}{\n\t\t\t\t\t\"error\": map[string]interface{}{\n\t\t\t\t\t\t\"type\": \"index_not_found_exception\",\n\t\t\t\t\t},\n\t\t\t\t\t\"status\": 404,\n\t\t\t\t})\n\t\t\t}),\n\t\t\texpected: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\t\t\terr := os2Client.CreateIndex(context.Background(), \"testIndex\")\n\t\t\tres := os2Client.IsNotFoundError(err)\n\t\t\tassert.Equal(t, tc.expected, res)\n\t\t})\n\t}\n}\n\nfunc TestCount(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\thandler       http.HandlerFunc\n\t\tindex         string\n\t\tquery         string\n\t\texpectedCount int64\n\t\texpectError   bool\n\t}{\n\t\t{\n\t\t\tname: \"Successful Count\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tfmt.Fprintln(w, `{\"count\": 42}`)\n\t\t\t},\n\t\t\tindex:         \"testIndex\",\n\t\t\tquery:         \"{}\",\n\t\t\texpectedCount: 42,\n\t\t\texpectError:   false,\n\t\t},\n\t\t{\n\t\t\tname: \"OpenSearch Error\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tfmt.Fprintln(w, `{\"error\": \"Internal Server Error\"}`)\n\t\t\t},\n\t\t\tindex:       \"testIndex\",\n\t\t\tquery:       \"{}\",\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Decoding Error\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tfmt.Fprintln(w, `{\"count\": \"should be an int64\"}`)\n\t\t\t},\n\t\t\tindex:       \"testIndex\",\n\t\t\tquery:       \"{}\",\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\n\t\t\tcount, err := os2Client.Count(context.Background(), tc.index, tc.query)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedCount, count)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScroll(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\tscrollID         string\n\t\thandler          http.HandlerFunc\n\t\texpectError      bool\n\t\texpectedScrollID string // Add more fields as needed for assertions\n\t}{\n\t\t{\n\t\t\tname:     \"Initial Search Request\",\n\t\t\tscrollID: \"\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tfmt.Fprintln(w, `{\"_scroll_id\": \"scrollID123\", \"took\": 10, \"hits\": {\"total\": {\"value\": 2}, \"hits\": [{\"_source\": {\"field1\": \"value1\"}}]}}`)\n\t\t\t},\n\t\t\texpectError:      false,\n\t\t\texpectedScrollID: \"scrollID123\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Subsequent Scroll Request\",\n\t\t\tscrollID: \"existingScrollID\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tfmt.Fprintln(w, `{\"_scroll_id\": \"scrollID456\", \"took\": 5, \"hits\": {\"total\": {\"value\": 1}, \"hits\": [{\"_source\": {\"field2\": \"value2\"}}]}}`)\n\t\t\t},\n\t\t\texpectError:      false,\n\t\t\texpectedScrollID: \"scrollID456\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Error Response\",\n\t\t\tscrollID: \"\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tfmt.Fprintln(w, `{\"error\": \"Internal Server Error\"}`)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"No More Hits\",\n\t\t\tscrollID: \"someScrollID\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tfmt.Fprintln(w, `{\"_scroll_id\": \"scrollIDNoHits\", \"took\": 5, \"hits\": {\"hits\": []}}`)\n\t\t\t},\n\t\t\texpectError:      false,\n\t\t\texpectedScrollID: \"scrollIDNoHits\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\n\t\t\tresp, err := os2Client.Scroll(context.Background(), \"testIndex\", \"{}\", tc.scrollID)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else if tc.name == \"No More Hits\" {\n\t\t\t\tassert.Equal(t, io.EOF, err, \"Expected io.EOF error for no more hits\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, tc.expectedScrollID, resp.ScrollID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestClearScroll(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tscrollID      string\n\t\thandler       http.HandlerFunc\n\t\texpectedError bool\n\t}{\n\t\t{\n\t\t\tname:     \"Successful Scroll Clear\",\n\t\t\tscrollID: \"testScrollID\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tw.Write([]byte(`{\"succeeded\": true}`))\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"OpenSearch Server Error\",\n\t\t\tscrollID: \"testScrollID\",\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tfmt.Fprintln(w, `{\"error\": {\"root_cause\": [{\"type\": \"internal_server_error\",\"reason\": \"Internal server error\"}],\"type\": \"internal_server_error\",\"reason\": \"Internal server error\"}}`)\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\n\t\t\terr := os2Client.ClearScroll(context.Background(), tc.scrollID)\n\n\t\t\tif tc.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSearch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tindex         string\n\t\tbody          string\n\t\thandler       http.HandlerFunc\n\t\texpectedError bool\n\t\texpectedHits  int\n\t}{\n\t\t{\n\t\t\tname:  \"Successful Search\",\n\t\t\tindex: \"testIndex\",\n\t\t\tbody:  `{\"query\": {\"match_all\": {}}}`,\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tfmt.Fprintln(w, `{\"took\": 10, \"hits\": {\"total\": {\"value\": 2}, \"hits\": [{\"_source\": {\"field\": \"value\"}, \"sort\": [1750950124525781262, \"test sort val\"]}, {\"_source\": {\"field\": \"another value\"}, \"sort\": [1750950124525781269, \"test sort val 2\"]}]}}`)\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\texpectedHits:  2,\n\t\t},\n\t\t{\n\t\t\tname:  \"OpenSearch Error\",\n\t\t\tindex: \"testIndex\",\n\t\t\tbody:  `{\"query\": {\"match_all\": {}}}`,\n\t\t\thandler: func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t\tfmt.Fprintln(w, `{\"error\": \"Bad request\"}`)\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tos2Client, testServer := getSecureMockOS2Client(t, tc.handler, true)\n\t\t\tdefer testServer.Close()\n\n\t\t\tresp, err := os2Client.Search(context.Background(), tc.index, tc.body)\n\n\t\t\tif tc.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Len(t, resp.Hits.Hits, tc.expectedHits)\n\t\t\t\tassert.Equal(t, resp.Sort[0], json.Number(\"1750950124525781269\"))\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v6/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage v6\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/elasticsearch/client\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ElasticV6 implements Client\n\tElasticV6 struct {\n\t\tclient *elastic.Client\n\t\tlogger log.Logger\n\t}\n\tconvertlogger func(msg string, tags ...tag.Tag)\n)\n\nvar _ elastic.Logger = (convertlogger)(nil)\n\nfunc (c convertlogger) Printf(format string, v ...interface{}) {\n\tc(fmt.Sprintf(format, v...))\n}\n\nfunc (c *ElasticV6) IsNotFoundError(err error) bool {\n\treturn elastic.IsNotFound(err)\n}\n\n// NewV6Client returns a new implementation of GenericClient\nfunc NewV6Client(\n\tconnectConfig *config.ElasticSearchConfig,\n\tlogger log.Logger,\n\ttlsClient *http.Client,\n\tawsSigningClient *http.Client,\n) (*ElasticV6, error) {\n\tclientOptFuncs := []elastic.ClientOptionFunc{\n\t\telastic.SetURL(connectConfig.URL.String()),\n\t\telastic.SetRetrier(elastic.NewBackoffRetrier(elastic.NewExponentialBackoff(128*time.Millisecond, 513*time.Millisecond))),\n\t\telastic.SetDecoder(&elastic.NumberDecoder{}), // critical to ensure decode of int64 won't lose precise)\n\t\t// elastic.SetInfoLog(convertlogger(logger.Info)),   // logs start/stop of components, and every successful request's code and timing.  probably too noisy.\n\t\telastic.SetErrorLog(convertlogger(logger.Error)), // logs errors when they occur, sounds likely not noisy\n\t}\n\tif connectConfig.DisableSniff {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetSniff(false))\n\t}\n\tif connectConfig.DisableHealthCheck {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHealthcheck(false))\n\t}\n\n\tif awsSigningClient != nil {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHttpClient(awsSigningClient))\n\t}\n\n\tif tlsClient != nil {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHttpClient(tlsClient))\n\t}\n\n\tclient, err := elastic.NewClient(clientOptFuncs...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ElasticV6{\n\t\tclient: client,\n\t\tlogger: logger,\n\t}, nil\n}\n\nfunc (c *ElasticV6) PutMapping(ctx context.Context, index, body string) error {\n\t_, err := c.client.PutMapping().Index(index).Type(\"_doc\").BodyString(body).Do(ctx)\n\treturn err\n}\n\nfunc (c *ElasticV6) CreateIndex(ctx context.Context, index string) error {\n\t_, err := c.client.CreateIndex(index).Do(ctx)\n\treturn err\n}\n\nfunc (c *ElasticV6) Count(ctx context.Context, index, query string) (int64, error) {\n\treturn c.client.Count(index).BodyString(query).Do(ctx)\n}\n\nfunc (c *ElasticV6) ClearScroll(ctx context.Context, scrollID string) error {\n\treturn elastic.NewScrollService(c.client).ScrollId(scrollID).Clear(ctx)\n}\nfunc (c *ElasticV6) Scroll(ctx context.Context, index, body, scrollID string) (*client.Response, error) {\n\tscrollService := elastic.NewScrollService(c.client)\n\tvar esResult *elastic.SearchResult\n\tvar err error\n\n\t// we are not returning error immediately here, as result + error combination is possible\n\tif len(scrollID) == 0 {\n\t\tesResult, err = scrollService.Index(index).Body(body).Do(ctx)\n\t} else {\n\t\tesResult, err = scrollService.ScrollId(scrollID).Do(ctx)\n\t}\n\n\tif esResult == nil {\n\t\treturn nil, err\n\t}\n\n\tvar hits []*client.SearchHit\n\tif esResult.Hits != nil {\n\t\tfor _, h := range esResult.Hits.Hits {\n\t\t\tif h.Source != nil {\n\t\t\t\thits = append(hits, &client.SearchHit{Source: *h.Source})\n\t\t\t}\n\t\t}\n\t}\n\n\tresult := &client.Response{\n\t\tTookInMillis: esResult.TookInMillis,\n\t\tTotalHits:    esResult.TotalHits(),\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tScrollID:     esResult.ScrollId,\n\t}\n\n\tif len(esResult.Aggregations) > 0 {\n\t\tresult.Aggregations = make(map[string]json.RawMessage, len(esResult.Aggregations))\n\t\tfor key, agg := range esResult.Aggregations {\n\t\t\tif agg != nil {\n\t\t\t\tresult.Aggregations[key] = *agg\n\t\t\t}\n\t\t}\n\t}\n\n\treturn result, err\n}\n\nfunc (c *ElasticV6) Search(ctx context.Context, index, body string) (*client.Response, error) {\n\tesResult, err := c.client.Search(index).Source(body).Do(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif esResult.Error != nil {\n\t\treturn nil, types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ElasticSearch Error: %#v\", esResult.Error),\n\t\t}\n\t} else if esResult.TimedOut {\n\t\treturn nil, types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ElasticSearch Error: Request timed out: %v ms\", esResult.TookInMillis),\n\t\t}\n\t}\n\n\tvar sort []interface{}\n\tvar hits []*client.SearchHit\n\n\tif esResult != nil && esResult.Hits != nil {\n\t\tfor _, h := range esResult.Hits.Hits {\n\t\t\tif h.Source != nil {\n\t\t\t\thits = append(hits, &client.SearchHit{Source: *h.Source})\n\t\t\t}\n\t\t\tsort = h.Sort\n\t\t}\n\t}\n\n\tresult := &client.Response{\n\t\tTookInMillis: esResult.TookInMillis,\n\t\tTotalHits:    esResult.TotalHits(),\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tSort:         sort,\n\t}\n\n\tif len(esResult.Aggregations) > 0 {\n\t\tresult.Aggregations = make(map[string]json.RawMessage, len(esResult.Aggregations))\n\t\tfor key, agg := range esResult.Aggregations {\n\t\t\tif agg != nil {\n\t\t\t\tresult.Aggregations[key] = *agg\n\t\t\t}\n\t\t}\n\t}\n\n\treturn result, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v6/client_bulk.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v6\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\n// bulkProcessorParametersV6 holds all required and optional parameters for executing bulk service\ntype bulkProcessorParametersV6 struct {\n\tName          string\n\tNumOfWorkers  int\n\tBulkActions   int\n\tBulkSize      int\n\tFlushInterval time.Duration\n\tBackoff       elastic.Backoff\n\tBeforeFunc    elastic.BulkBeforeFunc\n\tAfterFunc     elastic.BulkAfterFunc\n}\n\ntype v6BulkProcessor struct {\n\tprocessor *elastic.BulkProcessor\n\tlogger    log.Logger\n}\n\nfunc (v *v6BulkProcessor) Start(ctx context.Context) error {\n\treturn v.processor.Start(ctx)\n}\n\nfunc (v *v6BulkProcessor) Stop() error {\n\treturn v.processor.Stop()\n}\n\nfunc (v *v6BulkProcessor) Close() error {\n\treturn v.processor.Close()\n}\n\nfunc (v *v6BulkProcessor) Add(request *bulk.GenericBulkableAddRequest) {\n\tvar req elastic.BulkableRequest\n\tswitch request.RequestType {\n\tcase bulk.BulkableDeleteRequest:\n\t\treq = elastic.NewBulkDeleteRequest().\n\t\t\tIndex(request.Index).\n\t\t\tType(request.Type).\n\t\t\tId(request.ID)\n\tcase bulk.BulkableIndexRequest:\n\t\treq = elastic.NewBulkIndexRequest().\n\t\t\tIndex(request.Index).\n\t\t\tType(request.Type).\n\t\t\tId(request.ID).\n\t\t\tVersionType(request.VersionType).\n\t\t\tVersion(request.Version).\n\t\t\tDoc(request.Doc)\n\tcase bulk.BulkableCreateRequest:\n\t\t// for bulk create request still calls the bulk index method\n\t\t// with providing operation type\n\t\treq = elastic.NewBulkIndexRequest().\n\t\t\tOpType(\"create\").\n\t\t\tIndex(request.Index).\n\t\t\tType(request.Type).\n\t\t\tId(request.ID).\n\t\t\tVersionType(\"internal\").\n\t\t\tDoc(request.Doc)\n\t}\n\tv.processor.Add(req)\n}\n\nfunc (v *v6BulkProcessor) Flush() error {\n\treturn v.processor.Flush()\n}\n\nfunc (c *ElasticV6) runBulkProcessor(ctx context.Context, p *bulkProcessorParametersV6) (*v6BulkProcessor, error) {\n\tprocessor, err := c.client.BulkProcessor().\n\t\tName(p.Name).\n\t\tWorkers(p.NumOfWorkers).\n\t\tBulkActions(p.BulkActions).\n\t\tBulkSize(p.BulkSize).\n\t\tFlushInterval(p.FlushInterval).\n\t\tBackoff(p.Backoff).\n\t\tBefore(p.BeforeFunc).\n\t\tAfter(p.AfterFunc).\n\t\tDo(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &v6BulkProcessor{\n\t\tprocessor: processor,\n\t}, nil\n}\n\nfunc (c *ElasticV6) RunBulkProcessor(ctx context.Context, parameters *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error) {\n\tbeforeFunc := func(executionId int64, requests []elastic.BulkableRequest) {\n\t\tparameters.BeforeFunc(executionId, fromV6ToGenericBulkableRequests(requests))\n\t}\n\n\tafterFunc := func(executionId int64, requests []elastic.BulkableRequest, response *elastic.BulkResponse, err error) {\n\t\tparameters.AfterFunc(\n\t\t\texecutionId,\n\t\t\tfromV6ToGenericBulkableRequests(requests),\n\t\t\tfromV6ToGenericBulkResponse(response),\n\t\t\tconvertV6ErrorToGenericError(err))\n\t}\n\n\treturn c.runBulkProcessor(ctx, &bulkProcessorParametersV6{\n\t\tName:          parameters.Name,\n\t\tNumOfWorkers:  parameters.NumOfWorkers,\n\t\tBulkActions:   parameters.BulkActions,\n\t\tBulkSize:      parameters.BulkSize,\n\t\tFlushInterval: parameters.FlushInterval,\n\t\tBackoff:       parameters.Backoff,\n\t\tBeforeFunc:    beforeFunc,\n\t\tAfterFunc:     afterFunc,\n\t})\n}\n\nfunc convertV6ErrorToGenericError(err error) *bulk.GenericError {\n\tif err == nil {\n\t\treturn nil\n\t}\n\tstatus := bulk.UnknownStatusCode\n\tswitch e := err.(type) {\n\tcase *elastic.Error:\n\t\tstatus = e.Status\n\t}\n\treturn &bulk.GenericError{\n\t\tStatus:  status,\n\t\tDetails: err,\n\t}\n}\n\nfunc fromV6ToGenericBulkResponse(response *elastic.BulkResponse) *bulk.GenericBulkResponse {\n\tif response == nil {\n\t\treturn &bulk.GenericBulkResponse{}\n\t}\n\treturn &bulk.GenericBulkResponse{\n\t\tTook:   response.Took,\n\t\tErrors: response.Errors,\n\t\tItems:  fromV6ToGenericBulkResponseItemMaps(response.Items),\n\t}\n}\n\nfunc fromV6ToGenericBulkResponseItemMaps(items []map[string]*elastic.BulkResponseItem) []map[string]*bulk.GenericBulkResponseItem {\n\tvar gitems []map[string]*bulk.GenericBulkResponseItem\n\tfor _, it := range items {\n\t\tgitems = append(gitems, fromV6ToGenericBulkResponseItemMap(it))\n\t}\n\treturn gitems\n}\n\nfunc fromV6ToGenericBulkResponseItemMap(m map[string]*elastic.BulkResponseItem) map[string]*bulk.GenericBulkResponseItem {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tgm := make(map[string]*bulk.GenericBulkResponseItem, len(m))\n\tfor k, v := range m {\n\t\tgm[k] = fromV6ToGenericBulkResponseItem(v)\n\t}\n\treturn gm\n}\n\nfunc fromV6ToGenericBulkResponseItem(v *elastic.BulkResponseItem) *bulk.GenericBulkResponseItem {\n\treturn &bulk.GenericBulkResponseItem{\n\t\tIndex:   v.Index,\n\t\tType:    v.Type,\n\t\tID:      v.Id,\n\t\tVersion: v.Version,\n\t\tResult:  v.Result,\n\t\tStatus:  v.Status,\n\t}\n}\n\nfunc fromV6ToGenericBulkableRequests(requests []elastic.BulkableRequest) []bulk.GenericBulkableRequest {\n\tvar v6Reqs []bulk.GenericBulkableRequest\n\tfor _, req := range requests {\n\t\tv6Reqs = append(v6Reqs, req)\n\t}\n\treturn v6Reqs\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v6/client_bulk_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v6\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestConvertV6ErrorToGenericError(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\terr      error\n\t\texpected *bulk.GenericError\n\t}{\n\t\t{\n\t\t\tname:     \"nil error\",\n\t\t\terr:      nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"non-elasticsearch error\",\n\t\t\terr:  errors.New(\"generic error\"),\n\t\t\texpected: &bulk.GenericError{\n\t\t\t\tStatus:  bulk.UnknownStatusCode,\n\t\t\t\tDetails: errors.New(\"generic error\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"elasticsearch error\",\n\t\t\terr: &elastic.Error{\n\t\t\t\tStatus: 404,\n\t\t\t\tDetails: &elastic.ErrorDetails{\n\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &bulk.GenericError{\n\t\t\t\tStatus: 404,\n\t\t\t\tDetails: &elastic.Error{\n\t\t\t\t\tStatus: 404,\n\t\t\t\t\tDetails: &elastic.ErrorDetails{\n\t\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := convertV6ErrorToGenericError(tt.err)\n\n\t\t\tif tt.expected == nil {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tassert.Equal(t, tt.expected.Status, result.Status)\n\n\t\t\t\tif expectedDetails, ok := tt.expected.Details.(*elastic.Error); ok {\n\t\t\t\t\tresultDetails, _ := result.Details.(*elastic.Error)\n\t\t\t\t\tassert.NotNil(t, resultDetails)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Status, resultDetails.Status)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Details.Type, resultDetails.Details.Type)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Details.Reason, resultDetails.Details.Reason)\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, tt.expected.Details.Error(), result.Details.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromV6toGenericBulkResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tresponse       *elastic.BulkResponse\n\t\texpectNil      bool\n\t\texpectedTook   int\n\t\texpectedErrors bool\n\t}{\n\t\t{\n\t\t\tname:           \"nil response\",\n\t\t\tresponse:       nil,\n\t\t\texpectNil:      false,\n\t\t\texpectedTook:   0,\n\t\t\texpectedErrors: false,\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil response\",\n\t\t\tresponse: &elastic.BulkResponse{\n\t\t\t\tTook:   100,\n\t\t\t\tErrors: true,\n\t\t\t\tItems: []map[string]*elastic.BulkResponseItem{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"index\": &elastic.BulkResponseItem{\n\t\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t\t\tError:  nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectNil:      false,\n\t\t\texpectedTook:   100,\n\t\t\texpectedErrors: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := fromV6ToGenericBulkResponse(tt.response)\n\n\t\t\tif tt.expectNil {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tassert.Equal(t, tt.expectedTook, result.Took)\n\t\t\t\tassert.Equal(t, tt.expectedErrors, result.Errors)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromV6ToGenericBulkResponseItemMaps(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tv6items  []map[string]*elastic.BulkResponseItem\n\t\texpected []map[string]*bulk.GenericBulkResponseItem\n\t}{\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tv6items: []map[string]*elastic.BulkResponseItem{\n\t\t\t\t{\n\t\t\t\t\t\"index\": &elastic.BulkResponseItem{\n\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\t\"update\": &elastic.BulkResponseItem{\n\t\t\t\t\t\tStatus: 404,\n\t\t\t\t\t\tError: &elastic.ErrorDetails{\n\t\t\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t{\n\t\t\t\t\t\"index\": {\n\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\t\"update\": {\n\t\t\t\t\t\tStatus: 404,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"nil case\",\n\t\t\tv6items:  nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgenericItems := fromV6ToGenericBulkResponseItemMaps(tt.v6items)\n\t\t\tassert.Len(t, genericItems, len(tt.expected), \"The lengths of actual and expected slices should match.\")\n\t\t\tassert.Equal(t, tt.expected, genericItems)\n\t\t})\n\t}\n}\n\nfunc TestFromV6ToGenericBulkResponseItemMap(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tv6item   map[string]*elastic.BulkResponseItem\n\t\texpected map[string]*bulk.GenericBulkResponseItem\n\t}{\n\t\t{\n\t\t\tname:     \"nil case\",\n\t\t\tv6item:   nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tv6item: map[string]*elastic.BulkResponseItem{\n\t\t\t\t\"index\": {\n\t\t\t\t\tIndex:   \"test_index\",\n\t\t\t\t\tType:    \"test_type\",\n\t\t\t\t\tId:      \"1\",\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tResult:  \"created\",\n\t\t\t\t\tStatus:  201,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t\"index\": {\n\t\t\t\t\tIndex:   \"test_index\",\n\t\t\t\t\tType:    \"test_type\",\n\t\t\t\t\tID:      \"1\",\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tResult:  \"created\",\n\t\t\t\t\tStatus:  201,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgenericItems := fromV6ToGenericBulkResponseItemMap(tt.v6item)\n\t\t\tassert.Equal(t, tt.expected, genericItems)\n\t\t})\n\t}\n}\n\nfunc TestFromV6ToGenericBulkResponseItem(t *testing.T) {\n\telasticItem := &elastic.BulkResponseItem{\n\t\tIndex:   \"test_index\",\n\t\tType:    \"test_type\",\n\t\tId:      \"1\",\n\t\tVersion: 1,\n\t\tResult:  \"created\",\n\t\tStatus:  201,\n\t}\n\texpectedGenericItem := &bulk.GenericBulkResponseItem{\n\t\tIndex:   \"test_index\",\n\t\tType:    \"test_type\",\n\t\tID:      \"1\",\n\t\tVersion: 1,\n\t\tResult:  \"created\",\n\t\tStatus:  201,\n\t}\n\n\tresult := fromV6ToGenericBulkResponseItem(elasticItem)\n\tassert.Equal(t, expectedGenericItem, result)\n}\n\nfunc TestFromV6ToGenericBulkableRequests(t *testing.T) {\n\tmockRequests := []elastic.BulkableRequest{\n\t\telastic.NewBulkIndexRequest().Index(\"index1\").Type(\"_doc\").Id(\"1\").Doc(map[string]interface{}{\"field\": \"value1\"}),\n\t\telastic.NewBulkIndexRequest().Index(\"index2\").Type(\"_doc\").Id(\"2\").Doc(map[string]interface{}{\"field\": \"value2\"}),\n\t}\n\n\tgenericRequests := fromV6ToGenericBulkableRequests(mockRequests)\n\tassert.Len(t, genericRequests, len(mockRequests))\n}\n\nfunc getV6BulkProcessor(t *testing.T) *v6BulkProcessor {\n\treturn &v6BulkProcessor{\n\t\tprocessor: &elastic.BulkProcessor{},\n\t\tlogger:    testlogger.New(t),\n\t}\n}\n\nfunc TestProcessorFunc(t *testing.T) {\n\tprocessor := getV6BulkProcessor(t)\n\terr := processor.Start(context.Background())\n\tassert.NoError(t, err)\n\n\terr = processor.Flush()\n\tassert.NoError(t, err)\n\n\terr = processor.Stop()\n\tassert.NoError(t, err)\n\n\terr = processor.Close()\n\tassert.NoError(t, err)\n}\n\nfunc TestProcessorAdd(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\trequest   *bulk.GenericBulkableAddRequest\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname:      \"bulk add nil case\",\n\t\t\trequest:   nil,\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk add normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableIndexRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk create normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableCreateRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk add normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableDeleteRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\telasticV6, testServer := getMockClient(t, nil)\n\tdefer testServer.Close()\n\tsvc := elasticV6.client.BulkProcessor().\n\t\tName(\"FlushInterval-1\").\n\t\tWorkers(2).\n\t\tBulkActions(-1).\n\t\tBulkSize(-1)\n\n\tp, err := svc.Do(context.Background())\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\tprocessor := v6BulkProcessor{\n\t\tprocessor: p,\n\t\tlogger:    testlogger.New(t),\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif tt.expectErr {\n\t\t\t\tassert.Panics(t, func() { processor.Add(tt.request) }, \"The method should panic\")\n\t\t\t} else {\n\t\t\t\tassert.NotPanics(t, func() { processor.Add(tt.request) }, \"The method should not panic\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRunBulkProcessor(t *testing.T) {\n\ttests := []struct {\n\t\tname   string\n\t\tparams *bulk.BulkProcessorParameters\n\t}{\n\t\t{\n\t\t\tname: \"successful initialization\",\n\t\t\tparams: &bulk.BulkProcessorParameters{\n\t\t\t\tName:          \"test-processor\",\n\t\t\t\tNumOfWorkers:  5,\n\t\t\t\tBulkActions:   1000,\n\t\t\t\tBulkSize:      2 * 1024 * 1024, // 2MB\n\t\t\t\tFlushInterval: 30 * time.Second,\n\t\t\t\tBackoff:       elastic.NewExponentialBackoff(10*time.Millisecond, 8*time.Second),\n\t\t\t\tBeforeFunc: func(executionId int64, requests []bulk.GenericBulkableRequest) {\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\telasticV6, testServer := getMockClient(t, nil)\n\t\t\tdefer testServer.Close()\n\t\t\tresult, err := elasticV6.RunBulkProcessor(context.Background(), tt.params)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v6/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v6\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"net/url\"\n\t\"testing\"\n\n\t\"github.com/olivere/elastic\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestNewV6Client(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestServer := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tw.WriteHeader(http.StatusOK)\n\t}))\n\tdefer testServer.Close()\n\turl, err := url.Parse(testServer.URL)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\n\tconnectionConfig := &config.ElasticSearchConfig{\n\t\tURL:                *url,\n\t\tDisableSniff:       true,\n\t\tDisableHealthCheck: true,\n\t}\n\tsharedClient := testServer.Client()\n\tclient, err := NewV6Client(connectionConfig, logger, sharedClient, sharedClient)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, client)\n\n\t// failed case due to an unreachable Elasticsearch server\n\tbadURL, err := url.Parse(\"http://nonexistent.elasticsearch.server:9200\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\tconnectionConfig.DisableHealthCheck = false\n\tconnectionConfig.URL = *badURL\n\t_, err = NewV6Client(connectionConfig, logger, nil, nil)\n\tassert.Error(t, err)\n}\n\nfunc TestCreateIndex(t *testing.T) {\n\tvar handlerCalled bool\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\thandlerCalled = true\n\t\tif r.URL.Path == \"/testIndex\" && r.Method == \"PUT\" {\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"acknowledged\": true}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV6, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV6.CreateIndex(context.Background(), \"testIndex\")\n\tassert.True(t, handlerCalled, \"Expected handler to be called\")\n\tassert.NoError(t, err)\n}\n\nfunc TestPutMapping(t *testing.T) {\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif r.URL.Path == \"/testIndex/_mapping/_doc\" && r.Method == \"PUT\" {\n\t\t\tbody, err := io.ReadAll(r.Body)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"Failed to read request body: %v\", err)\n\t\t\t}\n\t\t\tdefer r.Body.Close()\n\t\t\tvar receivedMapping map[string]interface{}\n\t\t\tif err := json.Unmarshal(body, &receivedMapping); err != nil {\n\t\t\t\tt.Fatalf(\"Failed to unmarshal request body: %v\", err)\n\t\t\t}\n\n\t\t\t// Define expected mapping structurally\n\t\t\texpectedMapping := map[string]interface{}{\n\t\t\t\t\"properties\": map[string]interface{}{\n\t\t\t\t\t\"title\": map[string]interface{}{\n\t\t\t\t\t\t\"type\": \"text\",\n\t\t\t\t\t},\n\t\t\t\t\t\"publish_date\": map[string]interface{}{\n\t\t\t\t\t\t\"type\": \"date\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Compare structurally\n\t\t\tif !assert.Equal(t, expectedMapping, receivedMapping) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"acknowledged\": true}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV6, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV6.PutMapping(context.Background(), \"testIndex\", `{\n        \"properties\": {\n            \"title\": {\n                \"type\": \"text\"\n            },\n            \"publish_date\": {\n                \"type\": \"date\"\n            }\n        }\n    }`)\n\tassert.NoError(t, err)\n}\n\nfunc TestCount(t *testing.T) {\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif r.URL.Path == \"/testIndex/_count\" && r.Method == \"POST\" {\n\t\t\tbody, err := io.ReadAll(r.Body)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"Failed to read request body: %v\", err)\n\t\t\t}\n\t\t\tdefer r.Body.Close()\n\t\t\texpectedQuery := `{\"query\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}`\n\t\t\tif string(body) != expectedQuery {\n\t\t\t\tt.Fatalf(\"Expected query %s, got %s\", expectedQuery, body)\n\t\t\t}\n\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"count\": 42}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV6, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\tcount, err := elasticV6.Count(context.Background(), \"testIndex\", `{\"query\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}`)\n\tassert.NoError(t, err)\n\tassert.Equal(t, int64(42), count)\n}\n\nfunc TestSearch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tquery     string\n\t\texpected  map[string]interface{}\n\t\texpectErr bool\n\t\texpectAgg bool\n\t\tindex     string\n\t\thandler   http.HandlerFunc\n\t}{\n\t\t{\n\t\t\tname:  \"normal case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"sort\": [1]\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"elasticsearch error\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"error\": {\n\t\t\t\t\t\t\t\"root_cause\": [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\"type\": \"index_not_found_exception\",\n\t\t\t\t\t\t\t\t\t\"reason\": \"no such index\",\n\t\t\t\t\t\t\t\t\t\"resource.type\": \"index_or_alias\",\n\t\t\t\t\t\t\t\t\t\"resource.id\": \"testIndex\",\n\t\t\t\t\t\t\t\t\t\"index_uuid\": \"_na_\",\n\t\t\t\t\t\t\t\t\t\"index\": \"testIndex\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t\"type\": \"index_not_found_exception\",\n\t\t\t\t\t\t\t\"reason\": \"no such index\",\n\t\t\t\t\t\t\t\"resource.type\": \"index_or_alias\",\n\t\t\t\t\t\t\t\"resource.id\": \"testIndex\",\n\t\t\t\t\t\t\t\"index_uuid\": \"_na_\",\n\t\t\t\t\t\t\t\"index\": \"testIndex\"\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"status\": 404\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"elasticsearch timeout\",\n\t\t\tquery:     `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex:     \"testIndex\",\n\t\t\texpectErr: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK) // Assuming Elasticsearch returns HTTP 200 for timeouts with an indication in the body\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 30,\n\t\t\t\t\t\t\"timed_out\": true,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 0,\n\t\t\t\t\t\t\t\"hits\": []\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t\t{\n\t\t\tname:  \"elasticsearch aggregations\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t\texpectAgg: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"sample_agg\": {\n\t\t\t\t\t\t\t\t\"value\": 42\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t\t{\n\t\t\tname:      \"elasticsearch non exist index\",\n\t\t\tquery:     `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex:     \"test_failure\",\n\t\t\texpectErr: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV6, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\tresp, err := elasticV6.Search(context.Background(), tc.index, tc.query)\n\t\t\tif !tc.expectErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\t// Verify the response details\n\t\t\t\tassert.Equal(t, int64(5), resp.TookInMillis)\n\t\t\t\tassert.Equal(t, int64(1), resp.TotalHits)\n\t\t\t\tassert.NotNil(t, resp.Hits)\n\t\t\t\tassert.Len(t, resp.Hits.Hits, 1)\n\n\t\t\t\tvar actual map[string]interface{}\n\t\t\t\tif err := json.Unmarshal([]byte(string(resp.Hits.Hits[0].Source)), &actual); err != nil {\n\t\t\t\t\tt.Fatalf(\"Failed to unmarshal actual JSON: %v\", err)\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tc.expected, actual)\n\n\t\t\t\tif tc.expectAgg {\n\t\t\t\t\t// Verify the response includes the expected aggregations\n\t\t\t\t\tassert.NotNil(t, resp.Aggregations, \"Aggregations should not be nil\")\n\t\t\t\t\tassert.Contains(t, resp.Aggregations, \"sample_agg\", \"Aggregations should contain 'sample_agg'\")\n\n\t\t\t\t\t// Additional assertions can be made to verify the contents of the aggregation\n\t\t\t\t\tsampleAgg := resp.Aggregations[\"sample_agg\"]\n\t\t\t\t\tvar aggResult map[string]interface{}\n\t\t\t\t\terr = json.Unmarshal(sampleAgg, &aggResult)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.Equal(t, float64(42), aggResult[\"value\"], \"Aggregation 'sample_agg' should have a value of 42\")\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScroll(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tquery     string\n\t\texpected  map[string]interface{}\n\t\texpectErr bool\n\t\tindex     string\n\t\thandler   http.HandlerFunc\n\t\tscrollID  string\n\t}{\n\t\t{\n\t\t\tname:  \"normal case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"sample_agg\": {\n\t\t\t\t\t\t\t\t\"value\": 42\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"error case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t\tscrollID:  \"1\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV6, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\tresp, err := elasticV6.Scroll(context.Background(), tc.index, tc.query, tc.scrollID)\n\t\t\tif !tc.expectErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualSource map[string]interface{}\n\t\t\t\terr := json.Unmarshal(resp.Hits.Hits[0].Source, &actualSource)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, actualSource)\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestClearScroll(t *testing.T) {\n\tvar handlerCalled bool\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\thandlerCalled = true\n\t\tif r.Method == \"DELETE\" && r.URL.Path == \"/_search/scroll\" {\n\t\t\t// Simulate a successful clear scroll response\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tresponse := `{\n\t\t\t\t\"succeeded\": true,\n\t\t\t\t\"num_freed\": 1\n\t\t\t}`\n\t\t\tw.Write([]byte(response))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV6, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV6.ClearScroll(context.Background(), \"scrollID\")\n\tassert.True(t, handlerCalled, \"Expected handler to be called\")\n\tassert.NoError(t, err)\n}\n\nfunc TestIsNotFoundError(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\thandler  http.HandlerFunc\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname: \"NotFound error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.NotFound(w, r)\n\t\t\t}),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Other error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.Error(w, \"Bad Request\", http.StatusBadRequest)\n\t\t\t}),\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV6, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\terr := elasticV6.CreateIndex(context.Background(), \"testIndex\")\n\t\t\tres := elasticV6.IsNotFoundError(err)\n\t\t\tassert.Equal(t, tc.expected, res)\n\t\t})\n\t}\n}\n\nfunc getMockClient(t *testing.T, handler http.HandlerFunc) (ElasticV6, *httptest.Server) {\n\ttestServer := httptest.NewTLSServer(handler)\n\tmockClient, err := elastic.NewClient(\n\t\telastic.SetURL(testServer.URL),\n\t\telastic.SetSniff(false),\n\t\telastic.SetHealthcheck(false),\n\t\telastic.SetHttpClient(testServer.Client()))\n\tassert.NoError(t, err)\n\treturn ElasticV6{\n\t\tclient: mockClient,\n\t}, testServer\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v7/client.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage v7\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic/v7\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/elasticsearch/client\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ElasticV7 implements ES7\n\tElasticV7 struct {\n\t\tclient *elastic.Client\n\t\tlogger log.Logger\n\t}\n\tconvertlogger func(msg string, tags ...tag.Tag)\n)\n\nvar _ elastic.Logger = (convertlogger)(nil)\n\nfunc (c convertlogger) Printf(format string, v ...interface{}) {\n\tc(fmt.Sprintf(format, v...))\n}\n\n// NewV7Client returns a new implementation of GenericClient\nfunc NewV7Client(\n\tconnectConfig *config.ElasticSearchConfig,\n\tlogger log.Logger,\n\ttlsClient *http.Client,\n\tawsSigningClient *http.Client,\n) (*ElasticV7, error) {\n\tclientOptFuncs := []elastic.ClientOptionFunc{\n\t\telastic.SetURL(connectConfig.URL.String()),\n\t\telastic.SetRetrier(elastic.NewBackoffRetrier(elastic.NewExponentialBackoff(128*time.Millisecond, 513*time.Millisecond))),\n\t\telastic.SetDecoder(&elastic.NumberDecoder{}), // critical to ensure decode of int64 won't lose precise\n\t\t// elastic.SetInfoLog(convertlogger(logger.Info)),   // logs start/stop of components, and every successful request's code and timing.  probably too noisy.\n\t\telastic.SetErrorLog(convertlogger(logger.Error)), // logs errors when they occur, sounds likely not noisy\n\t}\n\tif connectConfig.DisableSniff {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetSniff(false))\n\t}\n\tif connectConfig.DisableHealthCheck {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHealthcheck(false))\n\t}\n\n\tif awsSigningClient != nil {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHttpClient(awsSigningClient))\n\t}\n\n\tif tlsClient != nil {\n\t\tclientOptFuncs = append(clientOptFuncs, elastic.SetHttpClient(tlsClient))\n\t}\n\n\tclient, err := elastic.NewClient(clientOptFuncs...)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ElasticV7{\n\t\tclient: client,\n\t\tlogger: logger,\n\t}, nil\n}\n\nfunc (c *ElasticV7) IsNotFoundError(err error) bool {\n\treturn elastic.IsNotFound(err)\n}\n\nfunc (c *ElasticV7) PutMapping(ctx context.Context, index, body string) error {\n\t_, err := c.client.PutMapping().Index(index).BodyString(body).Do(ctx)\n\treturn err\n}\nfunc (c *ElasticV7) CreateIndex(ctx context.Context, index string) error {\n\t_, err := c.client.CreateIndex(index).Do(ctx)\n\treturn err\n}\nfunc (c *ElasticV7) Count(ctx context.Context, index, query string) (int64, error) {\n\treturn c.client.Count(index).BodyString(query).Do(ctx)\n}\n\nfunc (c *ElasticV7) ClearScroll(ctx context.Context, scrollID string) error {\n\treturn elastic.NewScrollService(c.client).ScrollId(scrollID).Clear(ctx)\n}\n\nfunc (c *ElasticV7) Search(ctx context.Context, index string, body string) (*client.Response, error) {\n\tesResult, err := c.client.Search(index).Source(body).Do(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif esResult.Error != nil {\n\t\treturn nil, types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ElasticSearch Error: %#v\", esResult.Error),\n\t\t}\n\t} else if esResult.TimedOut {\n\t\treturn nil, types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ElasticSearch Error: Request timed out: %v ms\", esResult.TookInMillis),\n\t\t}\n\t}\n\n\tvar sort []interface{}\n\tvar hits []*client.SearchHit\n\n\tif esResult != nil && esResult.Hits != nil {\n\t\tfor _, h := range esResult.Hits.Hits {\n\t\t\thits = append(hits, &client.SearchHit{Source: h.Source})\n\t\t\tsort = h.Sort\n\t\t}\n\t}\n\n\treturn &client.Response{\n\t\tTookInMillis: esResult.TookInMillis,\n\t\tTotalHits:    esResult.TotalHits(),\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tAggregations: esResult.Aggregations,\n\t\tSort:         sort,\n\t}, nil\n\n}\n\nfunc (c *ElasticV7) Scroll(ctx context.Context, index, body, scrollID string) (*client.Response, error) {\n\tscrollService := elastic.NewScrollService(c.client)\n\tvar esResult *elastic.SearchResult\n\tvar err error\n\n\t// we are not returning error immediately here, as result + error combination is possible\n\tif len(scrollID) == 0 {\n\t\tesResult, err = scrollService.Index(index).Body(body).Do(ctx)\n\t} else {\n\t\tesResult, err = scrollService.ScrollId(scrollID).Do(ctx)\n\t}\n\n\tif esResult == nil {\n\t\treturn nil, err\n\t}\n\n\tvar hits []*client.SearchHit\n\tif esResult.Hits != nil {\n\t\tfor _, h := range esResult.Hits.Hits {\n\t\t\thits = append(hits, &client.SearchHit{Source: h.Source})\n\t\t}\n\t}\n\n\treturn &client.Response{\n\t\tTookInMillis: esResult.TookInMillis,\n\t\tTotalHits:    esResult.TotalHits(),\n\t\tHits:         &client.SearchHits{Hits: hits},\n\t\tAggregations: esResult.Aggregations,\n\t\tScrollID:     esResult.ScrollId,\n\t}, err\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v7/client_bulk.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v7\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic/v7\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n)\n\ntype // bulkProcessorParametersV7 holds all required and optional parameters for executing bulk service\nbulkProcessorParametersV7 struct {\n\tName          string\n\tNumOfWorkers  int\n\tBulkActions   int\n\tBulkSize      int\n\tFlushInterval time.Duration\n\tBackoff       elastic.Backoff\n\tBeforeFunc    elastic.BulkBeforeFunc\n\tAfterFunc     elastic.BulkAfterFunc\n}\n\ntype v7BulkProcessor struct {\n\tprocessor *elastic.BulkProcessor\n}\n\nfunc (v *v7BulkProcessor) Start(ctx context.Context) error {\n\treturn v.processor.Start(ctx)\n}\n\nfunc (v *v7BulkProcessor) Stop() error {\n\treturn v.processor.Stop()\n}\n\nfunc (v *v7BulkProcessor) Close() error {\n\treturn v.processor.Close()\n}\n\nfunc (v *v7BulkProcessor) Add(request *bulk.GenericBulkableAddRequest) {\n\tvar req elastic.BulkableRequest\n\tswitch request.RequestType {\n\tcase bulk.BulkableDeleteRequest:\n\t\treq = elastic.NewBulkDeleteRequest().\n\t\t\tIndex(request.Index).\n\t\t\tId(request.ID)\n\tcase bulk.BulkableIndexRequest:\n\t\treq = elastic.NewBulkIndexRequest().\n\t\t\tIndex(request.Index).\n\t\t\tId(request.ID).\n\t\t\tVersionType(request.VersionType).\n\t\t\tVersion(request.Version).\n\t\t\tDoc(request.Doc)\n\tcase bulk.BulkableCreateRequest:\n\t\t// for bulk create request still calls the bulk index method\n\t\t// with providing operation type\n\t\treq = elastic.NewBulkIndexRequest().\n\t\t\tOpType(\"create\").\n\t\t\tIndex(request.Index).\n\t\t\tId(request.ID).\n\t\t\tVersionType(\"internal\").\n\t\t\tDoc(request.Doc)\n\t}\n\tv.processor.Add(req)\n}\n\nfunc (v *v7BulkProcessor) Flush() error {\n\treturn v.processor.Flush()\n}\n\nfunc (c *ElasticV7) runBulkProcessor(ctx context.Context, p *bulkProcessorParametersV7) (*v7BulkProcessor, error) {\n\tprocessor, err := c.client.BulkProcessor().\n\t\tName(p.Name).\n\t\tWorkers(p.NumOfWorkers).\n\t\tBulkActions(p.BulkActions).\n\t\tBulkSize(p.BulkSize).\n\t\tFlushInterval(p.FlushInterval).\n\t\tBackoff(p.Backoff).\n\t\tBefore(p.BeforeFunc).\n\t\tAfter(p.AfterFunc).\n\t\tDo(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &v7BulkProcessor{\n\t\tprocessor: processor,\n\t}, nil\n}\n\nfunc fromV7ToGenericBulkResponse(response *elastic.BulkResponse) *bulk.GenericBulkResponse {\n\tif response == nil {\n\t\treturn &bulk.GenericBulkResponse{}\n\t}\n\treturn &bulk.GenericBulkResponse{\n\t\tTook:   response.Took,\n\t\tErrors: response.Errors,\n\t\tItems:  fromV7ToGenericBulkResponseItemMaps(response.Items),\n\t}\n}\n\nfunc fromV7ToGenericBulkResponseItemMaps(items []map[string]*elastic.BulkResponseItem) []map[string]*bulk.GenericBulkResponseItem {\n\tvar gitems []map[string]*bulk.GenericBulkResponseItem\n\tfor _, it := range items {\n\t\tgitems = append(gitems, fromV7ToGenericBulkResponseItemMap(it))\n\t}\n\treturn gitems\n}\n\nfunc fromV7ToGenericBulkResponseItemMap(m map[string]*elastic.BulkResponseItem) map[string]*bulk.GenericBulkResponseItem {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tgm := make(map[string]*bulk.GenericBulkResponseItem, len(m))\n\tfor k, v := range m {\n\t\tgm[k] = fromV7ToGenericBulkResponseItem(v)\n\t}\n\treturn gm\n}\n\nfunc fromV7ToGenericBulkResponseItem(v *elastic.BulkResponseItem) *bulk.GenericBulkResponseItem {\n\treturn &bulk.GenericBulkResponseItem{\n\t\tIndex:   v.Index,\n\t\tType:    v.Type,\n\t\tID:      v.Id,\n\t\tVersion: v.Version,\n\t\tResult:  v.Result,\n\t\tStatus:  v.Status,\n\t}\n}\n\nfunc fromV7ToGenericBulkableRequests(requests []elastic.BulkableRequest) []bulk.GenericBulkableRequest {\n\tvar v7Reqs []bulk.GenericBulkableRequest\n\tfor _, req := range requests {\n\t\tv7Reqs = append(v7Reqs, req)\n\t}\n\treturn v7Reqs\n}\n\nfunc (c *ElasticV7) RunBulkProcessor(ctx context.Context, parameters *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error) {\n\tbeforeFunc := func(executionId int64, requests []elastic.BulkableRequest) {\n\t\tparameters.BeforeFunc(executionId, fromV7ToGenericBulkableRequests(requests))\n\t}\n\n\tafterFunc := func(executionId int64, requests []elastic.BulkableRequest, response *elastic.BulkResponse, err error) {\n\t\tparameters.AfterFunc(\n\t\t\texecutionId,\n\t\t\tfromV7ToGenericBulkableRequests(requests),\n\t\t\tfromV7ToGenericBulkResponse(response),\n\t\t\tconvertV7ErrorToGenericError(err))\n\t}\n\n\treturn c.runBulkProcessor(ctx, &bulkProcessorParametersV7{\n\t\tName:          parameters.Name,\n\t\tNumOfWorkers:  parameters.NumOfWorkers,\n\t\tBulkActions:   parameters.BulkActions,\n\t\tBulkSize:      parameters.BulkSize,\n\t\tFlushInterval: parameters.FlushInterval,\n\t\tBackoff:       parameters.Backoff,\n\t\tBeforeFunc:    beforeFunc,\n\t\tAfterFunc:     afterFunc,\n\t})\n}\nfunc convertV7ErrorToGenericError(err error) *bulk.GenericError {\n\tif err == nil {\n\t\treturn nil\n\t}\n\tstatus := bulk.UnknownStatusCode\n\tswitch e := err.(type) {\n\tcase *elastic.Error:\n\t\tstatus = e.Status\n\t}\n\treturn &bulk.GenericError{\n\t\tStatus:  status,\n\t\tDetails: err,\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v7/client_bulk_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v7\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\telastic \"github.com/olivere/elastic/v7\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n)\n\nfunc TestFromV7toGenericBulkResponse(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tresponse       *elastic.BulkResponse\n\t\texpectNil      bool\n\t\texpectedTook   int\n\t\texpectedErrors bool\n\t}{\n\t\t{\n\t\t\tname:           \"nil response\",\n\t\t\tresponse:       nil,\n\t\t\texpectNil:      false,\n\t\t\texpectedTook:   0,\n\t\t\texpectedErrors: false,\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil response\",\n\t\t\tresponse: &elastic.BulkResponse{\n\t\t\t\tTook:   100,\n\t\t\t\tErrors: true,\n\t\t\t\tItems: []map[string]*elastic.BulkResponseItem{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"index\": &elastic.BulkResponseItem{\n\t\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t\t\tError:  nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectNil:      false,\n\t\t\texpectedTook:   100,\n\t\t\texpectedErrors: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := fromV7ToGenericBulkResponse(tt.response)\n\n\t\t\tif tt.expectNil {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tassert.Equal(t, tt.expectedTook, result.Took)\n\t\t\t\tassert.Equal(t, tt.expectedErrors, result.Errors)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromV7ToGenericBulkResponseItemMaps(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tv7items  []map[string]*elastic.BulkResponseItem\n\t\texpected []map[string]*bulk.GenericBulkResponseItem\n\t}{\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tv7items: []map[string]*elastic.BulkResponseItem{\n\t\t\t\t{\n\t\t\t\t\t\"index\": &elastic.BulkResponseItem{\n\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\t\"update\": &elastic.BulkResponseItem{\n\t\t\t\t\t\tStatus: 404,\n\t\t\t\t\t\tError: &elastic.ErrorDetails{\n\t\t\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t{\n\t\t\t\t\t\"index\": {\n\t\t\t\t\t\tStatus: 200,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\t\"update\": {\n\t\t\t\t\t\tStatus: 404,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"nil case\",\n\t\t\tv7items:  nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgenericItems := fromV7ToGenericBulkResponseItemMaps(tt.v7items)\n\t\t\tassert.Len(t, genericItems, len(tt.expected), \"The lengths of actual and expected slices should match.\")\n\t\t\tassert.Equal(t, tt.expected, genericItems)\n\t\t})\n\t}\n}\n\nfunc TestFromV7ToGenericBulkResponseItemMap(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tv7item   map[string]*elastic.BulkResponseItem\n\t\texpected map[string]*bulk.GenericBulkResponseItem\n\t}{\n\t\t{\n\t\t\tname:     \"nil case\",\n\t\t\tv7item:   nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tv7item: map[string]*elastic.BulkResponseItem{\n\t\t\t\t\"index\": {\n\t\t\t\t\tIndex:   \"test_index\",\n\t\t\t\t\tType:    \"test_type\",\n\t\t\t\t\tId:      \"1\",\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tResult:  \"created\",\n\t\t\t\t\tStatus:  201,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]*bulk.GenericBulkResponseItem{\n\t\t\t\t\"index\": {\n\t\t\t\t\tIndex:   \"test_index\",\n\t\t\t\t\tType:    \"test_type\",\n\t\t\t\t\tID:      \"1\",\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tResult:  \"created\",\n\t\t\t\t\tStatus:  201,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgenericItems := fromV7ToGenericBulkResponseItemMap(tt.v7item)\n\t\t\tassert.Equal(t, tt.expected, genericItems)\n\t\t})\n\t}\n}\n\nfunc TestFromV7ToGenericBulkResponseItem(t *testing.T) {\n\telasticItem := &elastic.BulkResponseItem{\n\t\tIndex:   \"test_index\",\n\t\tType:    \"test_type\",\n\t\tId:      \"1\",\n\t\tVersion: 1,\n\t\tResult:  \"created\",\n\t\tStatus:  201,\n\t}\n\texpectedGenericItem := &bulk.GenericBulkResponseItem{\n\t\tIndex:   \"test_index\",\n\t\tType:    \"test_type\",\n\t\tID:      \"1\",\n\t\tVersion: 1,\n\t\tResult:  \"created\",\n\t\tStatus:  201,\n\t}\n\n\tresult := fromV7ToGenericBulkResponseItem(elasticItem)\n\tassert.Equal(t, expectedGenericItem, result)\n}\n\nfunc TestFromV7ToGenericBulkableRequests(t *testing.T) {\n\tmockRequests := []elastic.BulkableRequest{\n\t\telastic.NewBulkIndexRequest().Index(\"index1\").Type(\"_doc\").Id(\"1\").Doc(map[string]interface{}{\"field\": \"value1\"}),\n\t\telastic.NewBulkIndexRequest().Index(\"index2\").Type(\"_doc\").Id(\"2\").Doc(map[string]interface{}{\"field\": \"value2\"}),\n\t}\n\n\tgenericRequests := fromV7ToGenericBulkableRequests(mockRequests)\n\tassert.Len(t, genericRequests, len(mockRequests))\n}\n\nfunc getV7BulkProcessor() *v7BulkProcessor {\n\treturn &v7BulkProcessor{\n\t\tprocessor: &elastic.BulkProcessor{},\n\t}\n}\n\nfunc TestProcessorFunc(t *testing.T) {\n\tprocessor := getV7BulkProcessor()\n\terr := processor.Start(context.Background())\n\tassert.NoError(t, err)\n\n\terr = processor.Flush()\n\tassert.NoError(t, err)\n\n\terr = processor.Stop()\n\tassert.NoError(t, err)\n\n\terr = processor.Close()\n\tassert.NoError(t, err)\n}\n\nfunc TestProcessorAdd(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\trequest   *bulk.GenericBulkableAddRequest\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tname:      \"bulk add nil case\",\n\t\t\trequest:   nil,\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk add normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableIndexRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk create normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableCreateRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"bulk add normal case\",\n\t\t\trequest: &bulk.GenericBulkableAddRequest{\n\t\t\t\tIndex:       \"test-index\",\n\t\t\t\tRequestType: bulk.BulkableDeleteRequest,\n\t\t\t\tID:          \"test-id\",\n\t\t\t\tVersionType: \"internal\",\n\t\t\t\tType:        \"test-type\",\n\t\t\t\tVersion:     int64(1),\n\t\t\t\tDoc:         \"\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\telasticV7, testServer := getMockClient(t, nil)\n\tdefer testServer.Close()\n\tsvc := elasticV7.client.BulkProcessor().\n\t\tName(\"FlushInterval-1\").\n\t\tWorkers(2).\n\t\tBulkActions(-1).\n\t\tBulkSize(-1)\n\n\tp, err := svc.Do(context.Background())\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\tprocessor := v7BulkProcessor{\n\t\tprocessor: p,\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif tt.expectErr {\n\t\t\t\tassert.Panics(t, func() { processor.Add(tt.request) }, \"The method should panic\")\n\t\t\t} else {\n\t\t\t\tassert.NotPanics(t, func() { processor.Add(tt.request) }, \"The method should not panic\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRunBulkProcessor(t *testing.T) {\n\ttests := []struct {\n\t\tname   string\n\t\tparams *bulk.BulkProcessorParameters\n\t}{\n\t\t{\n\t\t\tname: \"successful initialization\",\n\t\t\tparams: &bulk.BulkProcessorParameters{\n\t\t\t\tName:          \"test-processor\",\n\t\t\t\tNumOfWorkers:  5,\n\t\t\t\tBulkActions:   1000,\n\t\t\t\tBulkSize:      2 * 1024 * 1024, // 2MB\n\t\t\t\tFlushInterval: 30 * time.Second,\n\t\t\t\tBackoff:       elastic.NewExponentialBackoff(10*time.Millisecond, 8*time.Second),\n\t\t\t\tBeforeFunc: func(executionId int64, requests []bulk.GenericBulkableRequest) {\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\telasticV7, testServer := getMockClient(t, nil)\n\t\t\tdefer testServer.Close()\n\t\t\tresult, err := elasticV7.RunBulkProcessor(context.Background(), tt.params)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, result)\n\t\t})\n\t}\n}\n\nfunc TestConvertV7ErrorToGenericError(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\terr      error\n\t\texpected *bulk.GenericError\n\t}{\n\t\t{\n\t\t\tname:     \"nil error\",\n\t\t\terr:      nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"non-elasticsearch error\",\n\t\t\terr:  errors.New(\"generic error\"),\n\t\t\texpected: &bulk.GenericError{\n\t\t\t\tStatus:  bulk.UnknownStatusCode,\n\t\t\t\tDetails: errors.New(\"generic error\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"elasticsearch error\",\n\t\t\terr: &elastic.Error{\n\t\t\t\tStatus: 404,\n\t\t\t\tDetails: &elastic.ErrorDetails{\n\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &bulk.GenericError{\n\t\t\t\tStatus: 404,\n\t\t\t\tDetails: &elastic.Error{\n\t\t\t\t\tStatus: 404,\n\t\t\t\t\tDetails: &elastic.ErrorDetails{\n\t\t\t\t\t\tType:   \"index_not_found_exception\",\n\t\t\t\t\t\tReason: \"no such index\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := convertV7ErrorToGenericError(tt.err)\n\n\t\t\tif tt.expected == nil {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tassert.Equal(t, tt.expected.Status, result.Status)\n\n\t\t\t\tif expectedDetails, ok := tt.expected.Details.(*elastic.Error); ok {\n\t\t\t\t\tresultDetails, _ := result.Details.(*elastic.Error)\n\t\t\t\t\tassert.NotNil(t, resultDetails)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Status, resultDetails.Status)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Details.Type, resultDetails.Details.Type)\n\t\t\t\t\tassert.Equal(t, expectedDetails.Details.Reason, resultDetails.Details.Reason)\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, tt.expected.Details.Error(), result.Details.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/client/v7/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage v7\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"net/url\"\n\t\"testing\"\n\n\telastic \"github.com/olivere/elastic/v7\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestNewV7Client(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestServer := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tw.WriteHeader(http.StatusOK)\n\t}))\n\tdefer testServer.Close()\n\turl, err := url.Parse(testServer.URL)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\n\tconnectionConfig := &config.ElasticSearchConfig{\n\t\tURL:                *url,\n\t\tDisableSniff:       true,\n\t\tDisableHealthCheck: true,\n\t}\n\tsharedClient := testServer.Client()\n\tclient, err := NewV7Client(connectionConfig, logger, sharedClient, sharedClient)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, client)\n\n\t// failed case due to an unreachable Elasticsearch server\n\tbadURL, err := url.Parse(\"http://nonexistent.elasticsearch.server:9200\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse bad URL: %v\", err)\n\t}\n\tconnectionConfig.DisableHealthCheck = false\n\tconnectionConfig.URL = *badURL\n\t_, err = NewV7Client(connectionConfig, logger, nil, nil)\n\tassert.Error(t, err)\n}\n\nfunc TestCreateIndex(t *testing.T) {\n\tvar handlerCalled bool\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\thandlerCalled = true\n\t\tif r.URL.Path == \"/testIndex\" && r.Method == \"PUT\" {\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"acknowledged\": true}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV7, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV7.CreateIndex(context.Background(), \"testIndex\")\n\tassert.True(t, handlerCalled, \"Expected handler to be called\")\n\tassert.NoError(t, err)\n}\n\nfunc TestPutMapping(t *testing.T) {\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif r.URL.Path == \"/testIndex/_mapping\" && r.Method == \"PUT\" {\n\t\t\tbody, err := io.ReadAll(r.Body)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"Failed to read request body: %v\", err)\n\t\t\t}\n\t\t\tdefer r.Body.Close()\n\t\t\tvar receivedMapping map[string]interface{}\n\t\t\tif err := json.Unmarshal(body, &receivedMapping); err != nil {\n\t\t\t\tt.Fatalf(\"Failed to unmarshal request body: %v\", err)\n\t\t\t}\n\n\t\t\t// Define expected mapping structurally\n\t\t\texpectedMapping := map[string]interface{}{\n\t\t\t\t\"properties\": map[string]interface{}{\n\t\t\t\t\t\"title\": map[string]interface{}{\n\t\t\t\t\t\t\"type\": \"text\",\n\t\t\t\t\t},\n\t\t\t\t\t\"publish_date\": map[string]interface{}{\n\t\t\t\t\t\t\"type\": \"date\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Compare structurally\n\t\t\tif !assert.Equal(t, expectedMapping, receivedMapping) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"acknowledged\": true}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV7, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV7.PutMapping(context.Background(), \"testIndex\", `{\n        \"properties\": {\n            \"title\": {\n                \"type\": \"text\"\n            },\n            \"publish_date\": {\n                \"type\": \"date\"\n            }\n        }\n    }`)\n\tassert.NoError(t, err)\n}\n\nfunc TestCount(t *testing.T) {\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\tif r.URL.Path == \"/testIndex/_count\" && r.Method == \"POST\" {\n\t\t\tbody, err := io.ReadAll(r.Body)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"Failed to read request body: %v\", err)\n\t\t\t}\n\t\t\tdefer r.Body.Close()\n\t\t\texpectedQuery := `{\"query\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}`\n\t\t\tif string(body) != expectedQuery {\n\t\t\t\tt.Fatalf(\"Expected query %s, got %s\", expectedQuery, body)\n\t\t\t}\n\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tw.Write([]byte(`{\"count\": 42}`))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV7, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\tcount, err := elasticV7.Count(context.Background(), \"testIndex\", `{\"query\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}`)\n\tassert.NoError(t, err)\n\tassert.Equal(t, int64(42), count)\n}\n\nfunc TestSearch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tquery     string\n\t\texpected  map[string]interface{}\n\t\texpectErr bool\n\t\texpectAgg bool\n\t\tindex     string\n\t\thandler   http.HandlerFunc\n\t}{\n\t\t{\n\t\t\tname:  \"normal case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"sort\": [1]\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"elasticsearch error\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"error\": {\n\t\t\t\t\t\t\t\"root_cause\": [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\"type\": \"index_not_found_exception\",\n\t\t\t\t\t\t\t\t\t\"reason\": \"no such index\",\n\t\t\t\t\t\t\t\t\t\"resource.type\": \"index_or_alias\",\n\t\t\t\t\t\t\t\t\t\"resource.id\": \"testIndex\",\n\t\t\t\t\t\t\t\t\t\"index_uuid\": \"_na_\",\n\t\t\t\t\t\t\t\t\t\"index\": \"testIndex\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\t\"type\": \"index_not_found_exception\",\n\t\t\t\t\t\t\t\"reason\": \"no such index\",\n\t\t\t\t\t\t\t\"resource.type\": \"index_or_alias\",\n\t\t\t\t\t\t\t\"resource.id\": \"testIndex\",\n\t\t\t\t\t\t\t\"index_uuid\": \"_na_\",\n\t\t\t\t\t\t\t\"index\": \"testIndex\"\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"status\": 404\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"elasticsearch timeout\",\n\t\t\tquery:     `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex:     \"testIndex\",\n\t\t\texpectErr: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK) // Assuming Elasticsearch returns HTTP 200 for timeouts with an indication in the body\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 30,\n\t\t\t\t\t\t\"timed_out\": true,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 0,\n\t\t\t\t\t\t\t\"hits\": []\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t\t{\n\t\t\tname:  \"elasticsearch aggregations\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t\texpectAgg: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"sample_agg\": {\n\t\t\t\t\t\t\t\t\"value\": 42\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t\t{\n\t\t\tname:      \"elasticsearch non exist index\",\n\t\t\tquery:     `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex:     \"test_failure\",\n\t\t\texpectErr: true,\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV7, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\tresp, err := elasticV7.Search(context.Background(), tc.index, tc.query)\n\t\t\tif !tc.expectErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\t// Verify the response details\n\t\t\t\tassert.Equal(t, int64(5), resp.TookInMillis)\n\t\t\t\tassert.Equal(t, int64(1), resp.TotalHits)\n\t\t\t\tassert.NotNil(t, resp.Hits)\n\t\t\t\tassert.Len(t, resp.Hits.Hits, 1)\n\n\t\t\t\tvar actual map[string]interface{}\n\t\t\t\tif err := json.Unmarshal([]byte(string(resp.Hits.Hits[0].Source)), &actual); err != nil {\n\t\t\t\t\tt.Fatalf(\"Failed to unmarshal actual JSON: %v\", err)\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tc.expected, actual)\n\n\t\t\t\tif tc.expectAgg {\n\t\t\t\t\t// Verify the response includes the expected aggregations\n\t\t\t\t\tassert.NotNil(t, resp.Aggregations, \"Aggregations should not be nil\")\n\t\t\t\t\tassert.Contains(t, resp.Aggregations, \"sample_agg\", \"Aggregations should contain 'sample_agg'\")\n\n\t\t\t\t\t// Additional assertions can be made to verify the contents of the aggregation\n\t\t\t\t\tsampleAgg := resp.Aggregations[\"sample_agg\"]\n\t\t\t\t\tvar aggResult map[string]interface{}\n\t\t\t\t\terr = json.Unmarshal(sampleAgg, &aggResult)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.Equal(t, float64(42), aggResult[\"value\"], \"Aggregation 'sample_agg' should have a value of 42\")\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScroll(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tquery     string\n\t\texpected  map[string]interface{}\n\t\texpectErr bool\n\t\tindex     string\n\t\thandler   http.HandlerFunc\n\t\tscrollID  string\n\t}{\n\t\t{\n\t\t\tname:  \"normal case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"WorkflowID\": \"test-workflow-id\",\n\t\t\t},\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tif r.URL.Path == \"/testIndex/_search\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"took\": 5,\n\t\t\t\t\t\t\"timed_out\": false,\n\t\t\t\t\t\t\"hits\": {\n\t\t\t\t\t\t\t\"total\": 1,\n\t\t\t\t\t\t\t\"hits\": [{\n\t\t\t\t\t\t\t\t\"_source\": {\n\t\t\t\t\t\t\t\t\t\"WorkflowID\": \"test-workflow-id\"\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t}]\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"sample_agg\": {\n\t\t\t\t\t\t\t\t\"value\": 42\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"error case\",\n\t\t\tquery: `{\"query\":{\"bool\":{\"must\":{\"match\":{\"WorkflowID\":\"test-workflow-id\"}}}}}`,\n\t\t\tindex: \"testIndex\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t}),\n\t\t\texpectErr: true,\n\t\t\tscrollID:  \"1\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV7, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\tresp, err := elasticV7.Scroll(context.Background(), tc.index, tc.query, tc.scrollID)\n\t\t\tif !tc.expectErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualSource map[string]interface{}\n\t\t\t\terr := json.Unmarshal(resp.Hits.Hits[0].Source, &actualSource)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, actualSource)\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestClearScroll(t *testing.T) {\n\tvar handlerCalled bool\n\thandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\thandlerCalled = true\n\t\tif r.Method == \"DELETE\" && r.URL.Path == \"/_search/scroll\" {\n\t\t\t// Simulate a successful clear scroll response\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\tresponse := `{\n\t\t\t\t\"succeeded\": true,\n\t\t\t\t\"num_freed\": 1\n\t\t\t}`\n\t\t\tw.Write([]byte(response))\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t}\n\t})\n\telasticV7, testServer := getMockClient(t, handler)\n\tdefer testServer.Close()\n\terr := elasticV7.ClearScroll(context.Background(), \"scrollID\")\n\tassert.True(t, handlerCalled, \"Expected handler to be called\")\n\tassert.NoError(t, err)\n}\n\nfunc TestIsNotFoundError(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\thandler  http.HandlerFunc\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname: \"NotFound error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.NotFound(w, r)\n\t\t\t}),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Other error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thttp.Error(w, \"Bad Request\", http.StatusBadRequest)\n\t\t\t}),\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\telasticV7, testServer := getMockClient(t, tc.handler)\n\t\t\tdefer testServer.Close()\n\t\t\terr := elasticV7.CreateIndex(context.Background(), \"testIndex\")\n\t\t\tres := elasticV7.IsNotFoundError(err)\n\t\t\tassert.Equal(t, tc.expected, res)\n\t\t})\n\t}\n}\n\nfunc getMockClient(t *testing.T, handler http.HandlerFunc) (ElasticV7, *httptest.Server) {\n\ttestServer := httptest.NewTLSServer(handler)\n\tmockClient, err := elastic.NewClient(\n\t\telastic.SetURL(testServer.URL),\n\t\telastic.SetSniff(false),\n\t\telastic.SetHealthcheck(false),\n\t\telastic.SetHttpClient(testServer.Client()))\n\tassert.NoError(t, err)\n\treturn ElasticV7{\n\t\tclient: mockClient,\n\t}, testServer\n}\n"
  },
  {
    "path": "common/elasticsearch/client.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"math\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/elasticsearch/client\"\n\t\"github.com/uber/cadence/common/elasticsearch/query\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nvar _ GenericClient = (*ESClient)(nil)\n\ntype ESClient struct {\n\tClient client.Client\n\tLogger log.Logger\n}\n\nfunc (c *ESClient) RunBulkProcessor(ctx context.Context, p *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error) {\n\treturn c.Client.RunBulkProcessor(ctx, p)\n}\n\nfunc (c *ESClient) CreateIndex(ctx context.Context, index string) error {\n\treturn c.Client.CreateIndex(ctx, index)\n}\n\nfunc (c *ESClient) IsNotFoundError(err error) bool {\n\treturn c.Client.IsNotFoundError(err)\n}\n\nfunc (c *ESClient) Search(ctx context.Context, request *SearchRequest) (*SearchResponse, error) {\n\ttoken, err := GetNextPageToken(request.ListRequest.NextPageToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsearchResult, err := c.getSearchResult(\n\t\tctx,\n\t\trequest.Index,\n\t\trequest.ListRequest,\n\t\trequest.MatchQuery,\n\t\trequest.IsOpen,\n\t\ttoken,\n\t)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn c.getListWorkflowExecutionsResponse(searchResult, token, request.ListRequest.PageSize, request.MaxResultWindow, request.Filter)\n}\n\nfunc (c *ESClient) SearchByQuery(ctx context.Context, request *SearchByQueryRequest) (*SearchResponse, error) {\n\ttoken, err := GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsearchResult, err := c.Client.Search(ctx, request.Index, request.Query)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn c.getListWorkflowExecutionsResponse(searchResult, token, request.PageSize, request.MaxResultWindow, request.Filter)\n}\n\nfunc (c *ESClient) SearchRaw(ctx context.Context, index, query string) (*RawResponse, error) {\n\tresponse, err := c.Client.Search(ctx, index, query)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresult := RawResponse{\n\t\tTookInMillis: response.TookInMillis,\n\t\tHits: SearchHits{\n\t\t\tTotalHits: response.TotalHits,\n\t\t\tHits:      c.esHitsToExecutions(response.Hits, nil /* no filter */),\n\t\t},\n\t\tAggregations: response.Aggregations,\n\t}\n\n\treturn &result, nil\n\n}\n\nfunc (c *ESClient) esHitsToExecutions(eshits *client.SearchHits, filter IsRecordValidFilter) []*p.InternalVisibilityWorkflowExecutionInfo {\n\tvar hits = make([]*p.InternalVisibilityWorkflowExecutionInfo, 0)\n\tif eshits != nil && len(eshits.Hits) > 0 {\n\t\tfor _, hit := range eshits.Hits {\n\t\t\tworkflowExecutionInfo := c.convertSearchResultToVisibilityRecord(hit)\n\t\t\tif filter == nil || filter(workflowExecutionInfo) {\n\t\t\t\thits = append(hits, workflowExecutionInfo)\n\t\t\t}\n\t\t}\n\t}\n\treturn hits\n}\n\nfunc (c *ESClient) ScanByQuery(ctx context.Context, request *ScanByQueryRequest) (*SearchResponse, error) {\n\tvar err error\n\ttoken, err := GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsearchResult, err := c.Client.Scroll(ctx, request.Index, request.Query, token.ScrollID)\n\n\tisLastPage := false\n\tif err == io.EOF { // no more result\n\t\tisLastPage = true\n\t\tif err := c.Client.ClearScroll(ctx, searchResult.ScrollID); err != nil {\n\t\t\tc.Logger.Warn(\"scroll clear failed\", tag.Error(err))\n\t\t}\n\t} else if err != nil && isNodeUnavailableError(err) { // scroll ID is no longer valid and points to a node that is unavailable. Query with fresh scroll\n\t\tc.Logger.Warn(\"scroll node not available for id, retrying with fresh scroll\",\n\t\t\ttag.Dynamic(\"scrollID\", token.ScrollID),\n\t\t\ttag.Error(err))\n\t\t// fresh scroll\n\t\tres, scrollErr := c.Client.Scroll(ctx, request.Index, request.Query, \"\")\n\t\tif scrollErr != nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"ScanByQuery failed after fresh scroll. Error: %v\", err),\n\t\t\t}\n\t\t}\n\t\t// set fresh scroll result to search result\n\t\tsearchResult = res\n\n\t} else if err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ScanByQuery failed. Error: %v\", err),\n\t\t}\n\t}\n\n\tresponse := &p.InternalListWorkflowExecutionsResponse{}\n\tif searchResult == nil {\n\t\treturn response, nil\n\t}\n\n\tresponse.Executions = c.esHitsToExecutions(searchResult.Hits, nil /* no filter */)\n\n\tif len(searchResult.Hits.Hits) == request.PageSize && !isLastPage {\n\t\tnextPageToken, err := SerializePageToken(&ElasticVisibilityPageToken{ScrollID: searchResult.ScrollID})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresponse.NextPageToken = make([]byte, len(nextPageToken))\n\t\tcopy(response.NextPageToken, nextPageToken)\n\t}\n\n\treturn response, nil\n}\n\nfunc (c *ESClient) SearchForOneClosedExecution(ctx context.Context, index string, request *SearchForOneClosedExecutionRequest) (*SearchForOneClosedExecutionResponse, error) {\n\tmatchDomainQuery := query.NewMatchQuery(DomainID, request.DomainUUID)\n\texistClosedStatusQuery := query.NewExistsQuery(CloseStatus)\n\tmatchWorkflowIDQuery := query.NewMatchQuery(WorkflowID, request.Execution.GetWorkflowID())\n\tboolQuery := query.NewBoolQuery().Must(matchDomainQuery).Must(existClosedStatusQuery).Must(matchWorkflowIDQuery)\n\trid := request.Execution.GetRunID()\n\tif rid != \"\" {\n\t\tmatchRunIDQuery := query.NewMatchQuery(RunID, rid)\n\t\tboolQuery = boolQuery.Must(matchRunIDQuery)\n\t}\n\n\tqb := query.NewBuilder()\n\tbody, err := qb.Query(boolQuery).String()\n\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"getting body from query builder: %w\", err)\n\t}\n\n\tsearchResult, err := c.Client.Search(ctx, index, body)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"SearchForOneClosedExecution failed. Error: %v\", err),\n\t\t}\n\t}\n\n\tresponse := &p.InternalGetClosedWorkflowExecutionResponse{}\n\tactualHits := searchResult.Hits.Hits\n\tif len(actualHits) == 0 {\n\t\treturn response, nil\n\t}\n\tresponse.Execution = c.convertSearchResultToVisibilityRecord(actualHits[0])\n\n\treturn response, nil\n}\n\nfunc (c *ESClient) CountByQuery(ctx context.Context, index, query string) (int64, error) {\n\treturn c.Client.Count(ctx, index, query)\n}\n\nfunc (c *ESClient) PutMapping(ctx context.Context, index, root, key, valueType string) error {\n\tmapping := buildPutMappingBody(root, key, valueType)\n\tbody, err := json.Marshal(mapping)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn c.Client.PutMapping(ctx, index, string(body))\n}\n\nfunc (c *ESClient) getListWorkflowExecutionsResponse(\n\tsearchHits *client.Response,\n\ttoken *ElasticVisibilityPageToken,\n\tpageSize int,\n\tmaxResultWindow int,\n\tisRecordValid func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\n\tresponse := &p.InternalListWorkflowExecutionsResponse{}\n\tnumOfActualHits := len(searchHits.Hits.Hits)\n\tresponse.Executions = make([]*p.InternalVisibilityWorkflowExecutionInfo, 0)\n\tfor i := 0; i < numOfActualHits; i++ {\n\t\tworkflowExecutionInfo := c.convertSearchResultToVisibilityRecord(searchHits.Hits.Hits[i])\n\t\tif isRecordValid == nil || isRecordValid(workflowExecutionInfo) {\n\t\t\t// for old APIs like ListOpenWorkflowExecutions, we added 1 ms to range query to overcome ES limitation\n\t\t\t// (see getSearchResult function), but manually dropped records beyond request range here.\n\t\t\tresponse.Executions = append(response.Executions, workflowExecutionInfo)\n\t\t}\n\t}\n\n\tif numOfActualHits == pageSize { // this means the response is not the last page\n\t\tvar nextPageToken []byte\n\t\tvar err error\n\n\t\t// ES Search API support pagination using From and PageSize, but has limit that From+PageSize cannot exceed a threshold\n\t\t// to retrieve deeper pages, use ES SearchAfter\n\t\tif searchHits.TotalHits <= int64(maxResultWindow-pageSize) { // use ES Search From+Size\n\t\t\tnextPageToken, err = SerializePageToken(&ElasticVisibilityPageToken{From: token.From + numOfActualHits})\n\t\t} else { // use ES Search After\n\t\t\tvar sortVal interface{}\n\t\t\tsortVal = searchHits.Sort[0]\n\t\t\ttieBreaker := searchHits.Sort[1].(string)\n\t\t\tnextPageToken, err = SerializePageToken(&ElasticVisibilityPageToken{SortValue: sortVal, TieBreaker: tieBreaker})\n\t\t}\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tresponse.NextPageToken = make([]byte, len(nextPageToken))\n\t\tcopy(response.NextPageToken, nextPageToken)\n\t}\n\n\treturn response, nil\n}\n\nfunc (c *ESClient) convertSearchResultToVisibilityRecord(hit *client.SearchHit) *p.InternalVisibilityWorkflowExecutionInfo {\n\tvar source *VisibilityRecord\n\terr := json.Unmarshal(hit.Source, &source)\n\tif err != nil { // log and skip error\n\t\treturn nil\n\t}\n\n\trecord := &p.InternalVisibilityWorkflowExecutionInfo{\n\t\tDomainID:               source.DomainID,\n\t\tWorkflowType:           source.WorkflowType,\n\t\tWorkflowID:             source.WorkflowID,\n\t\tRunID:                  source.RunID,\n\t\tTypeName:               source.WorkflowType,\n\t\tStartTime:              time.Unix(0, source.StartTime),\n\t\tExecutionTime:          time.Unix(0, source.ExecutionTime),\n\t\tMemo:                   p.NewDataBlob(source.Memo, constants.EncodingType(source.Encoding)),\n\t\tTaskList:               source.TaskList,\n\t\tIsCron:                 source.IsCron,\n\t\tNumClusters:            source.NumClusters,\n\t\tClusterAttributeScope:  source.ClusterAttributeScope,\n\t\tClusterAttributeName:   source.ClusterAttributeName,\n\t\tSearchAttributes:       source.Attr,\n\t\tCronSchedule:           source.CronSchedule,\n\t\tExecutionStatus:        types.WorkflowExecutionStatus(source.ExecutionStatus),\n\t\tScheduledExecutionTime: time.Unix(0, source.ScheduledExecutionTime),\n\t}\n\tif source.UpdateTime != 0 {\n\t\trecord.UpdateTime = time.Unix(0, source.UpdateTime)\n\t}\n\tif source.CloseTime != 0 {\n\t\trecord.CloseTime = time.Unix(0, source.CloseTime)\n\t\trecord.Status = thrift.ToWorkflowExecutionCloseStatus(&source.CloseStatus)\n\t\trecord.HistoryLength = source.HistoryLength\n\t}\n\n\treturn record\n}\n\nfunc (c *ESClient) getSearchResult(\n\tctx context.Context,\n\tindex string,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n\tmatchQuery *query.MatchQuery,\n\tisOpen bool,\n\ttoken *ElasticVisibilityPageToken,\n) (*client.Response, error) {\n\n\t// always match domain id\n\tboolQuery := query.NewBoolQuery().Must(query.NewMatchQuery(DomainID, request.DomainUUID))\n\n\tif matchQuery != nil {\n\t\tboolQuery = boolQuery.Must(matchQuery)\n\t}\n\t// ElasticSearch v6 is unable to precisely compare time, have to manually add resolution 1ms to time range.\n\t// Also has to use string instead of int64 to avoid data conversion issue,\n\t// 9223372036854775807 to 9223372036854776000 (long overflow)\n\tif request.LatestTime.UnixNano() > math.MaxInt64-oneMicroSecondInNano { // prevent latestTime overflow\n\t\trequest.LatestTime = time.Unix(0, math.MaxInt64-oneMicroSecondInNano)\n\t}\n\tif request.EarliestTime.UnixNano() < math.MinInt64+oneMicroSecondInNano { // prevent earliestTime overflow\n\t\trequest.EarliestTime = time.Unix(0, math.MinInt64+oneMicroSecondInNano)\n\t}\n\n\tvar rangeQueryField string\n\texistClosedStatusQuery := query.NewExistsQuery(CloseStatus)\n\tif isOpen {\n\t\trangeQueryField = StartTime\n\t\tboolQuery = boolQuery.MustNot(existClosedStatusQuery)\n\t} else {\n\t\tboolQuery = boolQuery.Must(existClosedStatusQuery)\n\t\trangeQueryField = CloseTime\n\t}\n\n\tearliestTimeStr := strconv.FormatInt(request.EarliestTime.UnixNano()-oneMicroSecondInNano, 10)\n\tlatestTimeStr := strconv.FormatInt(request.LatestTime.UnixNano()+oneMicroSecondInNano, 10)\n\trangeQuery := query.NewRangeQuery(rangeQueryField).Gte(earliestTimeStr).Lte(latestTimeStr)\n\tboolQuery = boolQuery.Filter(rangeQuery)\n\n\tqb := query.NewBuilder()\n\tqb.Query(boolQuery).From(token.From).Size(request.PageSize)\n\n\tqb.Sortby(\n\t\tquery.NewFieldSort(rangeQueryField).Desc(),\n\t\tquery.NewFieldSort(RunID).Desc(),\n\t)\n\n\tif ShouldSearchAfter(token) {\n\t\tqb.SearchAfter(token.SortValue, token.TieBreaker)\n\t}\n\n\tbody, err := qb.String()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"getting body from query builder: %w\", err)\n\t}\n\n\treturn c.Client.Search(ctx, index, body)\n}\n\nfunc buildPutMappingBody(root, key, valueType string) map[string]interface{} {\n\tbody := make(map[string]interface{})\n\tif len(root) != 0 {\n\t\tbody[\"properties\"] = map[string]interface{}{\n\t\t\troot: map[string]interface{}{\n\t\t\t\t\"properties\": map[string]interface{}{\n\t\t\t\t\tkey: map[string]interface{}{\n\t\t\t\t\t\t\"type\": valueType,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t} else {\n\t\tbody[\"properties\"] = map[string]interface{}{\n\t\t\tkey: map[string]interface{}{\n\t\t\t\t\"type\": valueType,\n\t\t\t},\n\t\t}\n\t}\n\treturn body\n}\n\n// Helper to check if error is node unavailable error from OpenSearch\nfunc isNodeUnavailableError(err error) bool {\n\tif err == nil {\n\t\treturn false\n\t}\n\terrMsg := err.Error()\n\treturn strings.Contains(errMsg, \"node\") && strings.Contains(errMsg, \"not available\")\n}\n"
  },
  {
    "path": "common/elasticsearch/common.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nconst (\n\t// TODO https://github.com/uber/cadence/issues/3686\n\toneMicroSecondInNano = int64(time.Microsecond / time.Nanosecond)\n\n\tesDocIDDelimiter = \"~\"\n\tesDocType        = \"_doc\"\n\tesDocIDSizeLimit = 512\n)\n\n// Build Http Client with TLS\nfunc buildTLSHTTPClient(config config.TLS) (*http.Client, error) {\n\ttlsConfig, err := config.ToTLSConfig()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Setup HTTPS client\n\ttransport := &http.Transport{TLSClientConfig: tlsConfig}\n\ttlsClient := &http.Client{Transport: transport}\n\n\treturn tlsClient, nil\n}\n\nfunc GetESDocIDSizeLimit() int {\n\treturn esDocIDSizeLimit\n}\n\nfunc GetESDocType() string {\n\treturn esDocType\n}\n\nfunc GetESDocDelimiter() string {\n\treturn esDocIDDelimiter\n}\n\nfunc GenerateDocID(wid, rid string) string {\n\treturn wid + esDocIDDelimiter + rid\n}\n"
  },
  {
    "path": "common/elasticsearch/defs.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport \"github.com/uber/cadence/.gen/go/indexer\"\n\n// All legal fields allowed in elastic search index\nconst (\n\tDomainID               = \"DomainID\"\n\tWorkflowID             = \"WorkflowID\"\n\tRunID                  = \"RunID\"\n\tWorkflowType           = \"WorkflowType\"\n\tStartTime              = \"StartTime\"\n\tExecutionTime          = \"ExecutionTime\"\n\tCloseTime              = \"CloseTime\"\n\tCloseStatus            = \"CloseStatus\"\n\tHistoryLength          = \"HistoryLength\"\n\tMemo                   = \"Memo\"\n\tEncoding               = \"Encoding\"\n\tTaskList               = \"TaskList\"\n\tIsCron                 = \"IsCron\"\n\tNumClusters            = \"NumClusters\"\n\tClusterAttributeScope  = \"ClusterAttributeScope\"\n\tClusterAttributeName   = \"ClusterAttributeName\"\n\tVisibilityOperation    = \"VisibilityOperation\"\n\tUpdateTime             = \"UpdateTime\"\n\tShardID                = \"ShardID\"\n\tCronSchedule           = \"CronSchedule\"\n\tExecutionStatus        = \"ExecutionStatus\"\n\tScheduledExecutionTime = \"ScheduledExecutionTime\"\n)\n\n// Supported field types\nvar (\n\tFieldTypeString = indexer.FieldTypeString\n\tFieldTypeInt    = indexer.FieldTypeInt\n\tFieldTypeBool   = indexer.FieldTypeBool\n\tFieldTypeBinary = indexer.FieldTypeBinary\n)\n"
  },
  {
    "path": "common/elasticsearch/esql/README.md",
    "content": "# ESQL: Translate SQL to Elasticsearch DSL\nUse SQL to query Elasticsearch. ES V6 compatible.\n\n## Supported features\n- [x] =, !=, <, >, <=, >=, <>, ()\n- [x] comparison with arithmetics (e.g. colA + 1 < colB * 2)\n- [x] arithmetic operators: +, -, *, /, %, >>, <<, ()\n- [x] AND, OR, NOT\n- [x] LIKE, IN, REGEX, IS NULL, BETWEEN\n- [x] LIMIT, SIZE, DISTINCT\n- [x] COUNT, COUNT(DISTINCT)\n- [x] AVG, MAX, MIN, SUM\n- [x] GROUP BY, ORDER BY\n- [x] HAVING\n- [x] query key value macro (see usage)\n- [x] pagination (search after)\n- [ ] pagination for aggregation\n- [ ] functions\n- [ ] JOIN\n- [ ] nested queries\n\n\n## Usage\nPlease refer to code and comments in `esql.go`. `esql.go` contains all the apis that an outside user needs.\n### Basic Usage\n~~~~go\nsql := `SELECT COUNT(*), MAX(colA) FROM myTable WHERE colB < 10 GROUP BY colC HAVING COUNT(*) > 20`\ne := NewESql()\ndsl, _, err := e.ConvertPretty(sql)    // convert sql to dsl\nif err == nil {\n    fmt.Println(dsl)\n}\n~~~~\n### Custom Query Macro\nESQL support API `ProcessQueryKey` to register custom policy for colName replacement. It accepts 2 functions, the first function determines whether a colName is to be replaced, the second specifies how to do the replacement. Use case: user has custom field `field`, but to resolve confict, server stores the field as `Custom.field`. `ProcessQueryKey` API can automatically do the conversion.\n\nESQL support API `ProcessQueryValue` to register custom policy for value processing. It accepts 2 functions, the first function determines whether a value of a colName is to be processed, the second specifies how to do the processing. Use case: user want to query time in readable format, but server stores time as an integer (unix nano). `ProcessQueryValue` API can automatically do the conversion.\n\nBelow shows an example.\n~~~~go\nsql := \"SELECT colA FROM myTable WHERE colB < 10 AND dateTime = '2015-01-01T02:59:59Z'\"\n// custom policy that change colName like \"col..\" to \"myCol..\"\nfunc myKeyFilter(colName string) bool {\n    return strings.HasPrefix(colName, \"col\")\n}\nfunc myKeyProcess(colName string) (string, error) {\n    return \"myCol\"+colName[3:], nil\n}\n// custom policy that convert formatted time string to unix nano\nfunc myValueFilter(colName string) bool {\n    return strings.Contains(colName, \"Time\") || strings.Contains(colName, \"time\")\n}\nfunc myValueProcess(timeStr string) (string, error) {\n    // convert formatted time string to unix nano integer\n    parsedTime, _ := time.Parse(defaultDateTimeFormat, timeStr)\n    return fmt.Sprintf(\"%v\", parsedTime.UnixNano()), nil\n}\n// with the 2 policies , converted dsl is equivalent to\n// \"SELECT myColA FROM myTable WHERE myColB < 10 AND dateTime = '1561678568048000000'\n// in which the time is in unix nano format\ne := NewESql()\ne.ProcessQueryKey(myKeyFilter, myKeyProcess)            // set up macro for key\ne.ProcessQueryValue(myValueFilter, myValueProcess)      // set up macro for value\ndsl, _, err := e.ConvertPretty(sql)                     // convert sql to dsl\nif err == nil {\n    fmt.Println(dsl)\n}\n~~~~\n### Pagination\nESQL support 2 kinds of pagination: FROM keyword and ES search_after.\n- FROM keyword: the same as SQL syntax. Be careful, **ES only support a page smaller than 10k**, if your offset is large than 10k, search_after is necessary.\n- search_after: Once you know the paging tokens, just feed them to `Convert` or `ConvertPretty` API in order.\n\nBelow shows an example.\n~~~~go\n// first page\nsql_page1 := \"SELECT * FROM myTable ORDER BY colA, colB LIMIT 10\"\ne := NewESql()\ndsl_page1, sortFields, err := e.ConvertPretty(sql_page1)\n\n// second page\n// 1. Use FROM to retrieve the 2nd page\nsql_page2_FROM := \"SELECT * FROM myTable ORDER BY colA, colB LIMIT 10 FROM 10\"\ndsl_page2_FROM, sortFields, err := e.ConvertPretty(sql_page2_FROM)\n\n// 2. Use search_after to retrieve the 2nd page\n// we can use sortFields and the query result from page 1 to get the page tokens\nsql_page2_search_after := sql_page1\npage_token_colA := \"123\"\npage_token_colB := \"bbc\"\ndsl_page2_search_after, sortFields, err := e.ConvertPretty(sql_page2_search_after, page_colA, page_colB)\n~~~~\n\nFor Cadence usage, refer to [this link](cadenceDevReadme.md).\n\n\n## ES V2.x vs ES V6.5\n|Item|ES V2.x|ES v6.5|\n|:-:|:-:|:-:|\n|missing check|{\"missing\": {\"field\": \"xxx\"}}|{\"must_not\": {\"exist\": {\"field\": \"xxx\"}}}|\n|group by multiple columns|nested \"aggs\" field|\"composite\" flattened grouping|\n\n\n## Attentions\n- If you want to apply aggregation on some fields, they should not be in type `text` in ES\n- `COUNT(colName)` will include documents w/ null values in that column in ES SQL API, while in esql we exclude null valued documents\n- ES SQL API and esql do not support `SELECT DISTINCT`, a workaround is to query something like `SELECT * FROM table GROUP BY colName`\n- ES SQL API does not support `ORDER BY aggregation`, esql support it by applying bucket_sort\n- ES SQL API does not support `HAVING aggregation` that not show up in `SELECT`, esql support it\n- To use regex query, the column should be `keyword` type, otherwise the regex is applied to all the terms produced by tokenizer from the original text rather than the original text itself\n- Comparison with arithmetics can be potentially slow since it uses scripting query and thus is not able to take advantage of reverse index. For binary operators, please refer to [this link](https://www.elastic.co/guide/en/elasticsearch/painless/6.5/painless-operators.html) on the precedence. We don't support all of them.\n- Comparison with arithmetics does not support date type\n\n\n## Acknowledgement\nThis project is originated from [elasticsql](https://github.com/cch123/elasticsql). Table below shows the improvement.\n\n|Item|detail|\n|:-:|:-:|\n|comparison|support comparison with arithmetics of different columns|\n|keyword IS|support standard SQL keywords IS NULL, IS NOT NULL for missing check|\n|keyword NOT|support NOT, convert NOT recursively since elasticsearch's must_not is not the same as boolean operator NOT in sql|\n|keyword LIKE|using \"wildcard\" tag, support SQL wildcard '%' and '_'|\n|keyword REGEX|using \"regexp\" tag, support standard regular expression syntax|\n|keyword GROUP BY|using \"composite\" tag to flatten multiple grouping|\n|keyword ORDER BY|using \"bucket_sort\" to support order by aggregation functions|\n|keyword HAVING|using \"bucket_selector\" and painless scripting language to support HAVING|\n|aggregations|allow introducing aggregation functions from all HAVING, SELECT, ORDER BY|\n|column name filtering|allow user pass an white list, when the sql query tries to select column out side white list, refuse the converting|\n|column name replacing|allow user pass an function as initializing parameter, the matched column name will be replaced upon the policy|\n|query value replacing|allow user pass an function as initializing parameter, query value will be processed by such function if the column name matched in filter function|\n|pagination|also return the sorting fields for future search after usage|\n|optimization|using \"filter\" tag rather than \"must\" tag to avoid scoring analysis and save time|\n|optimization|no redundant {\"bool\": {\"filter\": xxx}} wrapped|all queries wrapped by {\"bool\": {\"filter\": xxx}}|\n|optimization|does not return document contents in aggregation query|\n|optimization|only return fields user specifies after SELECT|\n"
  },
  {
    "path": "common/elasticsearch/esql/aggregation.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\nfunc (e *ESql) convertAgg(sel sqlparser.Select) (dsl string, err error) {\n\tif len(sel.GroupBy) == 0 && sel.Having != nil {\n\t\terr = fmt.Errorf(`esql: HAVING used without GROUP BY`)\n\t\treturn \"\", err\n\t}\n\n\tcolNameSetGroupBy := make(map[string]int)\n\tvar dslGroupBy string\n\tif len(sel.GroupBy) != 0 {\n\t\tdslGroupBy, colNameSetGroupBy, err = e.convertGroupByExpr(sel.GroupBy)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t}\n\taggFuncExprSlice, colNameSlice, aggNameSlice, err := e.extractSelectedExpr(sel.SelectExprs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// verify don't select col name out side agg group name\n\tif err = e.checkSelGroupByCompatibility(colNameSlice, colNameSetGroupBy, aggNameSlice); err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// explanations for getAggSelect, getAggOrderBy, getAggHaving:\n\t// user can introduce aggregation functions from SELECT, ORDER BY and HAVING, for each different\n\t// aggregation functions, we need to add a tag for it in \"aggs\" field, which let ES to do the calculation\n\t// each aggregation's query body is in the form of \"<tag>: {\"<agg function name>\": {\"field\": \"<colName>\"}}\n\t//\n\t// <tag> is generated by us, the convention in esql is tag = <agg function name>_<colName> to prevent dup tag name\n\t// <agg function name> can be sum, max, min, count, avg\n\t// <colName> is the field that agg apply to\n\t//\n\t// however, for each source, there can be dups, we don't want to introduce duplicate tags\n\t// aggTagSet, aggTagOrderBySet, aggTagHavingSet are used to resolve dups, each of them is a map[string]int\n\t// which maps the tag string to an offset integer which indicates the position of this tag in\n\t// the corresponding aggxxxSlice\n\t//\n\t// aggNamexxxSlice stores agg functions names, aggTargetxxxSlice stores colNames, aggTagxxxSlice stores the tags\n\t// they are used to generate final json query\n\n\t// handle selected aggregation functions\n\taggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet, err := e.getAggSelect(aggFuncExprSlice)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// handle order by aggregation functions\n\taggNameOrderBySlice, aggTargetOrderBySlice, aggTagOrderBySlice, aggDirOrderBySlice, aggTagOrderBySet, err := e.getAggOrderBy(sel.OrderBy)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// handle having aggregation functions\n\tscript, aggNameHavingSlice, aggTargetHavingSlice, aggTagHavingSlice, aggTagHavingSet, err := e.getAggHaving(sel.Having)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// add necessary aggregations originated from order by and having\n\tfor tag, i := range aggTagOrderBySet {\n\t\tif _, exist := aggTagSet[tag]; !exist {\n\t\t\taggTagSet[tag] = len(aggTagSet)\n\t\t\taggNameSlice = append(aggNameSlice, aggNameOrderBySlice[i])\n\t\t\taggTargetSlice = append(aggTargetSlice, aggTargetOrderBySlice[i])\n\t\t\taggTagSlice = append(aggTagSlice, aggTagOrderBySlice[i])\n\t\t}\n\t}\n\tfor tag, i := range aggTagHavingSet {\n\t\tif _, exist := aggTagSet[tag]; !exist {\n\t\t\taggTagSet[tag] = len(aggTagSet)\n\t\t\taggNameSlice = append(aggNameSlice, aggNameHavingSlice[i])\n\t\t\taggTargetSlice = append(aggTargetSlice, aggTargetHavingSlice[i])\n\t\t\taggTagSlice = append(aggTagSlice, aggTagHavingSlice[i])\n\t\t}\n\t}\n\n\t// generate inside aggs field\n\tvar dslAgg string\n\tif len(aggTagSlice) > 0 {\n\t\tvar dslAggSlice []string\n\t\tfor i, tag := range aggTagSlice {\n\t\t\tif tag != \"_count\" {\n\t\t\t\tdslAgg := fmt.Sprintf(`\"%v\": {\"%v\": {\"field\": \"%v\"}}`, tag, aggNameSlice[i], aggTargetSlice[i])\n\t\t\t\tdslAggSlice = append(dslAggSlice, dslAgg)\n\t\t\t}\n\t\t}\n\t\tif len(aggTagOrderBySlice) > 0 {\n\t\t\tvar dslOrderSlice []string\n\t\t\tfor i, tag := range aggTagOrderBySlice {\n\t\t\t\tdslOrder := fmt.Sprintf(`{\"%v\": {\"order\": \"%v\"}}`, tag, aggDirOrderBySlice[i])\n\t\t\t\tdslOrderSlice = append(dslOrderSlice, dslOrder)\n\t\t\t}\n\t\t\tdslAggOrder := strings.Join(dslOrderSlice, \",\")\n\t\t\tdslAggOrder = fmt.Sprintf(`\"bucket_sort\": {\"bucket_sort\": {\"sort\": [%v], \"size\": %v}}`, dslAggOrder, e.bucketNumber)\n\t\t\tdslAggSlice = append(dslAggSlice, dslAggOrder)\n\t\t}\n\t\tif script != \"\" {\n\t\t\tvar bucketPathSlice []string\n\t\t\tfor tag := range aggTagHavingSet {\n\t\t\t\tbucketPathSlice = append(bucketPathSlice, fmt.Sprintf(`\"%v\": \"%v\"`, tag, tag))\n\t\t\t}\n\t\t\tbucketPathStr := strings.Join(bucketPathSlice, \",\")\n\t\t\tbucketFilterStr := fmt.Sprintf(`\"having\": {\"bucket_selector\": {\"buckets_path\": {%v}, \"script\": \"%v\"}}`, bucketPathStr, script)\n\t\t\tdslAggSlice = append(dslAggSlice, bucketFilterStr)\n\t\t}\n\t\tdslAgg = \"{\" + strings.Join(dslAggSlice, \",\") + \"}\"\n\t}\n\n\t// generate final dsl for aggs field\n\t// here \"groupby\" is just a tag and can be any unreserved word\n\tif len(dslGroupBy) == 0 && len(aggTagSlice) == 0 {\n\t\tdsl = \"\"\n\t} else if len(aggTagSlice) == 0 {\n\t\tdsl = fmt.Sprintf(`{\"groupby\": {%v}}`, dslGroupBy)\n\t} else if len(dslGroupBy) == 0 {\n\t\tdsl = dslAgg\n\t} else {\n\t\tdsl = fmt.Sprintf(`{\"groupby\": {%v, \"aggs\": %v}}`, dslGroupBy, dslAgg)\n\t}\n\treturn dsl, nil\n}\n\nfunc (e *ESql) checkSelGroupByCompatibility(colNameSlice []string, colNameGroupBy map[string]int, aggNameSlice []string) error {\n\tfor _, aggName := range aggNameSlice {\n\t\tcolNameGroupBy[aggName] = 1\n\t}\n\tif len(colNameGroupBy) == 0 {\n\t\treturn nil\n\t}\n\tfor _, colNameStr := range colNameSlice {\n\t\tif _, exist := colNameGroupBy[colNameStr]; !exist {\n\t\t\terr := fmt.Errorf(`esql: select column %v that not in group by`, colNameStr)\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (e *ESql) getAggOrderBy(orderBy sqlparser.OrderBy) ([]string, []string, []string, []string, map[string]int, error) {\n\tvar aggNameSlice, aggTargetSlice, aggDirSlice, aggTagSlice []string\n\taggTagDirSet := make(map[string]string) // tag -> asc / desc\n\taggTagSet := make(map[string]int)       // tag -> offset, for compatiblity checking\n\n\taggCnt := 0\n\tfor _, orderExpr := range orderBy {\n\t\tswitch orderExpr.Expr.(type) {\n\t\tcase *sqlparser.FuncExpr:\n\t\t\taggCnt++\n\t\t\tfuncExpr := orderExpr.Expr.(*sqlparser.FuncExpr)\n\t\t\taggNameStr := strings.ToLower(funcExpr.Name.String())\n\t\t\t// ? should we convert funcExpr.Exprs to colname?\n\t\t\taggTargetStr := sqlparser.String(funcExpr.Exprs)\n\t\t\taggTargetStr = strings.Trim(aggTargetStr, \"`\")\n\t\t\taggTargetStr, err := e.keyProcess(aggTargetStr)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, nil, nil, nil, nil, err\n\t\t\t}\n\t\t\tvar aggTagStr string\n\t\t\tswitch aggNameStr {\n\t\t\tcase \"count\":\n\t\t\t\t// no need to handle count(*) since the size of bucket is always returned\n\t\t\t\tif aggTargetStr == \"*\" {\n\t\t\t\t\taggTagStr = \"_count\"\n\t\t\t\t} else if funcExpr.Distinct {\n\t\t\t\t\taggTagStr = aggNameStr + \"_distinct_\" + aggTargetStr\n\t\t\t\t\taggNameStr = \"cardinality\"\n\t\t\t\t} else {\n\t\t\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\t\t\t\taggNameStr = \"value_count\"\n\t\t\t\t}\n\t\t\tcase \"avg\", \"sum\", \"min\", \"max\":\n\t\t\t\tif funcExpr.Distinct {\n\t\t\t\t\terr := fmt.Errorf(`esql: aggregation function %v w/ DISTINCT not supported`, aggNameStr)\n\t\t\t\t\treturn nil, nil, nil, nil, nil, err\n\t\t\t\t}\n\t\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\t\tdefault:\n\t\t\t\terr := fmt.Errorf(`esql: aggregation function %v not supported`, aggNameStr)\n\t\t\t\treturn nil, nil, nil, nil, nil, err\n\t\t\t}\n\t\t\tif dir, exist := aggTagDirSet[aggTagStr]; exist {\n\t\t\t\tif dir != orderExpr.Direction {\n\t\t\t\t\terr := fmt.Errorf(`esql: order by aggregation direction conflict`)\n\t\t\t\t\treturn nil, nil, nil, nil, nil, err\n\t\t\t\t}\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\taggTagStr = strings.Replace(aggTagStr, \".\", \"_\", -1)\n\t\t\taggTagDirSet[aggTagStr] = orderExpr.Direction\n\t\t\taggTagSet[aggTagStr] = len(aggTagSet)\n\t\t\taggNameSlice = append(aggNameSlice, aggNameStr)\n\t\t\taggTargetSlice = append(aggTargetSlice, aggTargetStr)\n\t\t\taggTagSlice = append(aggTagSlice, aggTagStr)\n\t\t\taggDirSlice = append(aggDirSlice, orderExpr.Direction)\n\t\tdefault:\n\t\t}\n\t}\n\n\tif aggCnt > 0 && aggCnt < len(orderBy) {\n\t\terr := fmt.Errorf(`esql: mix order by agg functions and column names`)\n\t\treturn nil, nil, nil, nil, nil, err\n\t}\n\treturn aggNameSlice, aggTargetSlice, aggTagSlice, aggDirSlice, aggTagSet, nil\n}\n\nfunc (e *ESql) getAggSelect(exprs []*sqlparser.FuncExpr) ([]string, []string, []string, map[string]int, error) {\n\tvar aggNameSlice, aggTargetSlice, aggTagSlice []string\n\taggTagSet := make(map[string]int) // tag -> offset, for compatibility checking\n\n\tfor _, funcExpr := range exprs {\n\t\taggNameStr := strings.ToLower(funcExpr.Name.String())\n\t\taggTargetStr := sqlparser.String(funcExpr.Exprs)\n\t\taggTargetStr = strings.Trim(aggTargetStr, \"`\")\n\t\taggTargetStr, err := e.keyProcess(aggTargetStr)\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, nil, err\n\t\t}\n\n\t\tvar aggTagStr string\n\t\tswitch aggNameStr {\n\t\tcase \"count\":\n\t\t\t// no need to handle count(*) since the size of bucket is always returned\n\t\t\tif aggTargetStr == \"*\" {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif funcExpr.Distinct {\n\t\t\t\taggTagStr = aggNameStr + \"_distinct_\" + aggTargetStr\n\t\t\t\taggNameStr = \"cardinality\"\n\t\t\t} else {\n\t\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\t\t\t// * ES SQL translate API just ignore non DISTINCT COUNT since the count of a bucket is always\n\t\t\t\t// * returned. However, we don't want count null value of a certain field, as a result we count\n\t\t\t\t// * documents w/ non-null value of the target field by \"value_count\" keyword\n\t\t\t\taggNameStr = \"value_count\"\n\t\t\t}\n\t\tcase \"avg\", \"sum\", \"min\", \"max\":\n\t\t\tif funcExpr.Distinct {\n\t\t\t\terr := fmt.Errorf(`esql: aggregation function %v w/ DISTINCT not supported`, aggNameStr)\n\t\t\t\treturn nil, nil, nil, nil, err\n\t\t\t}\n\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\tdefault:\n\t\t\terr := fmt.Errorf(`esql: aggregation function %v not supported`, aggNameStr)\n\t\t\treturn nil, nil, nil, nil, err\n\t\t}\n\t\tif _, exist := aggTagSet[aggTagStr]; exist {\n\t\t\tcontinue\n\t\t}\n\t\taggTagStr = strings.Replace(aggTagStr, \".\", \"_\", -1)\n\t\taggTagSet[aggTagStr] = len(aggTagSet)\n\t\taggNameSlice = append(aggNameSlice, aggNameStr)\n\t\taggTargetSlice = append(aggTargetSlice, aggTargetStr)\n\t\taggTagSlice = append(aggTagSlice, aggTagStr)\n\t}\n\n\treturn aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet, nil\n}\n\nfunc (e *ESql) extractSelectedExpr(expr sqlparser.SelectExprs) ([]*sqlparser.FuncExpr, []string, []string, error) {\n\tvar aggFuncExprSlice []*sqlparser.FuncExpr\n\tvar colNameSlice, aggNameSlice []string\n\tfor _, selectExpr := range expr {\n\t\t// from sqlparser's definition, we need to first convert the selectExpr to AliasedExpr\n\t\t// and then check whether AliasedExpr is a FuncExpr or just ColName\n\t\tswitch selectExpr.(type) {\n\t\tcase *sqlparser.AliasedExpr:\n\t\t\taliasedExpr := selectExpr.(*sqlparser.AliasedExpr)\n\t\t\tswitch aliasedExpr.Expr.(type) {\n\t\t\tcase *sqlparser.FuncExpr:\n\t\t\t\tfuncExpr := aliasedExpr.Expr.(*sqlparser.FuncExpr)\n\t\t\t\taggFuncExprSlice = append(aggFuncExprSlice, funcExpr)\n\t\t\t\taggNameSlice = append(aggNameSlice, sqlparser.String(funcExpr.Exprs))\n\t\t\tcase *sqlparser.ColName:\n\t\t\t\tlhs := aliasedExpr.Expr.(*sqlparser.ColName)\n\t\t\t\tlhsStr, err := e.convertColName(lhs)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, nil, nil, err\n\t\t\t\t}\n\t\t\t\tcolNameSlice = append(colNameSlice, lhsStr)\n\t\t\tdefault:\n\t\t\t\terr := fmt.Errorf(`esql: %T not supported in select body`, aliasedExpr)\n\t\t\t\treturn nil, nil, nil, err\n\t\t\t}\n\t\tdefault:\n\t\t}\n\t}\n\treturn aggFuncExprSlice, colNameSlice, aggNameSlice, nil\n}\n\nfunc (e *ESql) convertGroupByExpr(expr sqlparser.GroupBy) (dsl string, colNameSet map[string]int, err error) {\n\tvar groupByStrSlice []string\n\tcolNameSet = make(map[string]int)\n\tfor _, groupByExpr := range expr {\n\t\tswitch groupByItem := groupByExpr.(type) {\n\t\tcase *sqlparser.ColName:\n\t\t\tcolNameStr, err := e.convertColName(groupByItem)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", nil, err\n\t\t\t}\n\t\t\tif _, exist := colNameSet[colNameStr]; !exist {\n\t\t\t\tcolNameSet[colNameStr] = 1\n\t\t\t\tgroupByStr := fmt.Sprintf(`{\"group_%v\": {\"terms\": {\"field\": \"%v\", \"missing_bucket\": true}}}`, colNameStr, colNameStr)\n\t\t\t\tgroupByStrSlice = append(groupByStrSlice, groupByStr)\n\t\t\t}\n\t\tdefault:\n\t\t\terr = fmt.Errorf(`esql: GROUP BY %T not supported`, groupByExpr)\n\t\t\treturn \"\", nil, err\n\t\t}\n\t}\n\tdsl = strings.Join(groupByStrSlice, \",\")\n\tdsl = fmt.Sprintf(`\"composite\": {\"size\": %v, \"sources\": [%v]}`, e.bucketNumber, dsl)\n\treturn dsl, colNameSet, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/cadenceDevReadme.md",
    "content": "# ESQL Cadence Usage\n\n## Motivation\nCurrently [Cadence](https://github.com/cadence-workflow/cadence) is using [elasticsql](https://github.com/cch123/elasticsql) to translate sql query. However it only support up to ES V2.x while Cadence is using ES V6.x. Beyond that, Cadence has some specific requirements that not supported by elasticsql yet.\n\nCurrent Cadence query request processing steps are listed below:\n- generate SQL from query\n- use elasticsql to translate SQL to DSL\n- ES V6.x does not support \"missing\" field, convert \"missing\" to \"bool\",\"must_not\",\"exist\" for ExecutionTime query if any\n- complete \"range\" field for ExecutionTime query by adding {\"gt\": 0}\n- add domain query\n- key whitelist filtering\n- delete some useless field like \"from\", \"size\"\n- modify sorting field (add workflow id as sorting tie breaker)\n- setup search after for pagination\n\nESQL aims at dealing all these addtional processing steps and providing an api to generate DSL in one step for visibility usage in Cadence.\n\n## Usage\nESQL has convert functions specific for cadence usage. Please refer to `cadencesql.go`. Below shows an example.\nAttention: to use cadence version api, `SetCadence()` must be called at initialzation.\n~~~~go\nsql := \"SELECT colA FROM myTable WHERE colB < 10 AND dateTime = '2015-01-01T02:59:59Z'\"\ndomainID := \"CadenceSampleDomain\"\n// custom policy that change colName like \"col..\" to \"myCol..\"\nfunc myKeyFilter(colName string) bool {\n    return strings.HasPrefix(colName, \"col\")\n}\nfunc myKeyProcess(colName string) (string, error) {\n    return \"myCol\"+colName[3:], nil\n}\n// custom policy that convert formatted time string to unix nano\nfunc myValueFilter(colName string) bool {\n    return strings.Contains(colName, \"Time\") || strings.Contains(colName, \"time\")\n}\nfunc myValueProcess(timeStr string) (string, error) {\n    // convert formatted time string to unix nano integer\n    parsedTime, _ := time.Parse(defaultDateTimeFormat, timeStr)\n    return fmt.Sprintf(\"%v\", parsedTime.UnixNano()), nil\n}\n// with the 2 policies , converted dsl is equivalent to\n// \"SELECT myColA FROM myTable WHERE myColB < 10 AND dateTime = '1561678568048000000'\n// in which the time is in unix nano format\ne := NewESql()\ne.SetCadence()\ne.ProcessQueryKey(myKeyFilter, myKeyProcess)         // set up filtering policy\ne.ProcessQueryValue(myValueFilter, myValueProcess)     // set up process policy\ndsl, _, err := e.ConvertPrettyCadence(sql, domainID)             // convert sql to dsl\nif err == nil {\n    fmt.Println(dsl)\n}\n~~~~\n\n## Testing\nTo setup local testing environment:\n- start cassandra service locally. Please refer to [Cadence](https://github.com/cadence-workflow/cadence) readme.\n- start zookeeper and kafka service locally. Here is a [referecne](https://kafka.apache.org/quickstart).\n- start elasticsearch and kibana service locally.\n- start a cadence worker by `./bin/helloworld -m worker` under cadence directory.\n- start cadence service locally. Please refer to [Cadence](https://github.com/cadence-workflow/cadence) readme.\n"
  },
  {
    "path": "common/elasticsearch/esql/cadenceSpecial.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\nfunc (e *ESql) addCadenceSort(orderBySlice []string, sortFields []string) ([]string, []string, error) {\n\tswitch len(orderBySlice) {\n\tcase 0: // if unsorted, use default sorting\n\t\tcadenceOrderStartTime := fmt.Sprintf(`{\"%v\": \"%v\"}`, StartTime, StartTimeOrder)\n\t\torderBySlice = append(orderBySlice, cadenceOrderStartTime)\n\t\tsortFields = append(sortFields, StartTime)\n\tcase 1: // user should not use tieBreaker to sort\n\t\tif sortFields[0] == TieBreaker {\n\t\t\terr := fmt.Errorf(\"esql: Cadence does not allow user sort by RunID\")\n\t\t\treturn nil, nil, err\n\t\t}\n\tdefault:\n\t\terr := fmt.Errorf(\"esql: Cadence only allow 1 custom sort field\")\n\t\treturn nil, nil, err\n\t}\n\n\t// add tie breaker\n\tcadenceOrderTieBreaker := fmt.Sprintf(`{\"%v\": \"%v\"}`, TieBreaker, TieBreakerOrder)\n\torderBySlice = append(orderBySlice, cadenceOrderTieBreaker)\n\tsortFields = append(sortFields, TieBreaker)\n\treturn orderBySlice, sortFields, nil\n}\n\nfunc (e *ESql) addCadenceDomainTimeQuery(sel sqlparser.Select, domainID string, dslMap map[string]interface{}) {\n\tvar domainIDQuery string\n\tif domainID != \"\" {\n\t\tdomainIDQuery = fmt.Sprintf(`{\"term\": {\"%v\": \"%v\"}}`, DomainID, domainID)\n\t}\n\tif sel.Where == nil {\n\t\tif domainID != \"\" {\n\t\t\tdslMap[\"query\"] = domainIDQuery\n\t\t}\n\t} else {\n\t\tif domainID != \"\" {\n\t\t\tdomainIDQuery = domainIDQuery + \",\"\n\t\t}\n\t\tif strings.Contains(fmt.Sprintf(\"%v\", dslMap[\"query\"]), ExecutionTime) {\n\t\t\texecutionTimeBound := fmt.Sprintf(`{\"range\": {\"%v\": {\"gte\": \"0\"}}}`, ExecutionTime)\n\t\t\tdslMap[\"query\"] = fmt.Sprintf(`{\"bool\": {\"filter\": [%v %v, %v]}}`, domainIDQuery, executionTimeBound, dslMap[\"query\"])\n\t\t} else {\n\t\t\tdslMap[\"query\"] = fmt.Sprintf(`{\"bool\": {\"filter\": [%v %v]}}`, domainIDQuery, dslMap[\"query\"])\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/cadencesql.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\n// SetCadence ... specify whether do special handling for cadence visibility\n// should not be called if there is potential race condition\n// should not be called by non-cadence user\nfunc (e *ESql) SetCadence(cadenceArg bool) {\n\te.cadence = cadenceArg\n}\n\n// ConvertPrettyCadence ...\n// convert sql to es dsl, for cadence usage\nfunc (e *ESql) ConvertPrettyCadence(sql string, domainID string, pagination ...interface{}) (dsl string, sortFields []string, err error) {\n\tdsl, sortFields, err = e.ConvertCadence(sql, domainID, pagination...)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\n\tvar prettifiedDSLBytes bytes.Buffer\n\terr = json.Indent(&prettifiedDSLBytes, []byte(dsl), \"\", \"  \")\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\treturn prettifiedDSLBytes.String(), sortFields, err\n}\n\n// ConvertCadence ...\n// convert sql to es dsl, for cadence usage\nfunc (e *ESql) ConvertCadence(sql string, domainID string, pagination ...interface{}) (dsl string, sortFields []string, err error) {\n\tif !e.cadence {\n\t\terr = fmt.Errorf(`esql: cadence option not turned on`)\n\t\treturn \"\", nil, err\n\t}\n\tstmt, err := sqlparser.Parse(sql)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\n\t// sql valid, start to handle\n\tswitch stmt := stmt.(type) {\n\tcase *sqlparser.Select:\n\t\tdsl, sortFields, err = e.convertSelect(*(stmt), domainID, pagination...)\n\tdefault:\n\t\terr = fmt.Errorf(`esql: Queries other than select not supported`)\n\t}\n\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\treturn dsl, sortFields, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/esql.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\n// ProcessFunc ...\n// esql use ProcessFunc to process query key value macro\ntype ProcessFunc func(string) (string, error)\n\n// FilterFunc ...\n// esql use FilterFunc to determine whether a target is to be processed by ProcessFunc\n// only those FilterFunc(colName) == true will be processed\ntype FilterFunc func(string) bool\n\n// ESql ...\n// ESql is used to hold necessary information that required in parsing\ntype ESql struct {\n\tfilterKey    FilterFunc  // select the column we want to process key macro\n\tfilterValue  FilterFunc  // select the column we want to process value macro\n\tprocessKey   ProcessFunc // if selected by filterKey, change the query name\n\tprocessValue ProcessFunc // if selected by filterValue, change the query value\n\tcadence      bool\n\tpageSize     int\n\tbucketNumber int\n}\n\n// NewESql ... return a new default ESql\nfunc NewESql() *ESql {\n\treturn &ESql{\n\t\tpageSize:     DefaultPageSize,\n\t\tbucketNumber: DefaultBucketNumber,\n\t\tcadence:      false,\n\t\tprocessKey:   nil,\n\t\tprocessValue: nil,\n\t\tfilterKey:    nil,\n\t\tfilterValue:  nil,\n\t}\n}\n\n// ProcessQueryKey ... set up user specified column name processing policy\n// should not be called if there is potential race condition\nfunc (e *ESql) ProcessQueryKey(filterArg FilterFunc, processArg ProcessFunc) {\n\te.filterKey = filterArg\n\te.processKey = processArg\n}\n\n// ProcessQueryValue ... set up user specified column value processing policy\n// should not be called if there is potential race condition\nfunc (e *ESql) ProcessQueryValue(filterArg FilterFunc, processArg ProcessFunc) {\n\te.filterValue = filterArg\n\te.processValue = processArg\n}\n\n// SetPageSize ... set the number of documents returned in a non-aggregation query\n// should not be called if there is potential race condition\nfunc (e *ESql) SetPageSize(pageSizeArg int) {\n\te.pageSize = pageSizeArg\n}\n\n// SetBucketNum ... set the number of bucket returned in an aggregation query\n// should not be called if there is potential race condition\nfunc (e *ESql) SetBucketNum(bucketNumArg int) {\n\te.bucketNumber = bucketNumArg\n}\n\n// ConvertPretty ...\n// Transform sql to elasticsearch dsl, and prettify the output json\n//\n// usage:\n//   - dsl, sortField, err := e.ConvertPretty(sql, pageParam1, pageParam2, ...)\n//\n// arguments:\n//   - sql: the sql query needs conversion in string format\n//   - pagination: variadic arguments that indicates es search_after for pagination\n//\n// return values:\n//   - dsl: the elasticsearch dsl json style string\n//   - sortField: string array that contains all column names used for sorting. useful for pagination.\n//   - err: contains err information\nfunc (e *ESql) ConvertPretty(sql string, pagination ...interface{}) (dsl string, sortField []string, err error) {\n\tdsl, sortField, err = e.Convert(sql, pagination...)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\n\tvar prettifiedDSLBytes bytes.Buffer\n\terr = json.Indent(&prettifiedDSLBytes, []byte(dsl), \"\", \"  \")\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\treturn prettifiedDSLBytes.String(), sortField, err\n}\n\n// Convert ...\n// Transform sql to elasticsearch dsl string\n//\n// usage:\n//   - dsl, sortField, err := e.Convert(sql, pageParam1, pageParam2, ...)\n//\n// arguments:\n//   - sql: the sql query needs conversion in string format\n//   - pagination: variadic arguments that indicates es search_after\n//\n// return values:\n//   - dsl: the elasticsearch dsl json style string\n//   - sortField: string array that contains all column names used for sorting. useful for pagination.\n//   - err: contains err information\nfunc (e *ESql) Convert(sql string, pagination ...interface{}) (dsl string, sortField []string, err error) {\n\tstmt, err := sqlparser.Parse(sql)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\n\t// sql valid, start to handle\n\tswitch stmt := stmt.(type) {\n\tcase *sqlparser.Select:\n\t\tdsl, sortField, err = e.convertSelect(*(stmt), \"\", pagination...)\n\tdefault:\n\t\terr = fmt.Errorf(`esql: Queries other than select not supported`)\n\t}\n\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\treturn dsl, sortField, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/esql_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or `SELl\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n)\n\nvar sqls = []string{\n\t`SELECT * FROM test1`,\n\t`SELECT * FROM test1 ORDER BY colE, colD DESC LIMIT 10 OFFSET 4`,\n\t`SELECT * FROM test1 WHERE colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE colB = colBB`,\n\t`SELECT * FROM test1 WHERE colB = 'ab' AND colB = colBB`,\n\t`SELECT * FROM test1 WHERE colB = 'ab' AND colB != colBB`,\n\t`SELECT * FROM test1 WHERE colD = 10`,\n\t`SELECT * FROM test1 WHERE ( NOT colD = 10)`,\n\t`SELECT * FROM test1 WHERE ( NOT (colD = 10))`,\n\t`SELECT * FROM test1 WHERE colD != 10 ORDER BY colD ASC LIMIT 14`,\n\t`SELECT * FROM test1 WHERE ( NOT colD != 10)`,\n\t`SELECT * FROM test1 WHERE colD + 1 > 9`,\n\t`SELECT * FROM test1 WHERE colB = colBB AND 2 * colD > 8+2`,\n\t`SELECT * FROM test1 WHERE colB + 'c' = colA`,\n\t`SELECT * FROM test1 WHERE colB + \"c\" = colA`,\n\t\"SELECT * FROM test1 WHERE `colB` + 'c' = colA\",\n\t`SELECT * FROM test1 WHERE ((colD != 10))`,\n\t`SELECT * FROM test1 WHERE colB = 'ab' AND ExecutionTime = 2016 AND colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE colB = 'ab' OR colD = 10`,\n\t`SELECT * FROM test1 WHERE colD != 10 AND colB = 'bc' OR colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE colD != 10 AND NOT colB = 'bc' OR NOT (colB = 'ab' AND NOT colE < 10)`,\n\t`SELECT * FROM test1 WHERE colD != 10 AND (colB = 'bc' OR colB = 'ab')`,\n\t`SELECT * FROM test1 WHERE (colD != 10 AND colB = 'bc') OR colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE ((colD != 10) AND (colB = 'bc' OR colB = 'ab'))`,\n\t`SELECT * FROM test1 WHERE NOT colD != 10 ORDER BY colE DESC, colD DESC`,\n\t`SELECT * FROM test1 WHERE NOT NOT colD != 10`,\n\t`SELECT * FROM test1 WHERE NOT NOT NOT colD != 10 ORDER BY colE, colD ASC`,\n\t`SELECT * FROM test1 WHERE NOT colD != 10 AND colB = 'bc' OR colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE colD != 10 AND NOT (colB = 'bc' OR colB = 'ab')`,\n\t`SELECT * FROM test1 WHERE (colD != 10 AND colB = 'bc') OR NOT colB = 'ab'`,\n\t`SELECT * FROM test1 WHERE NOT ((colD != 10) AND (NOT colB = 'bc' OR colB = 'ab'))`,\n\t`SELECT * FROM test1 WHERE colE > 3 AND colD <= 15`,\n\t`SELECT * FROM test1 WHERE colE < 5 OR colD >= 17 ORDER BY colE DESC, colD ASC`,\n\t`SELECT * FROM test1 WHERE NOT (colE >= 5 AND colD < 17)`,\n\t`SELECT * FROM test1 WHERE NOT colE >= 5 AND colD < 17`,\n\t`SELECT * FROM test1 WHERE colE <= 9 OR colD >= 6`,\n\t`SELECT * FROM test1 WHERE NOT (colE > 9 AND colD < 6)`,\n\t`SELECT * FROM test1 WHERE colE > 0 OR colD <= 21.000`,\n\t`SELECT colC FROM test1 WHERE ExecutionTime IS NULL ORDER BY colD`,\n\t`SELECT * FROM test1 WHERE colB IS NOT NULL ORDER BY colE`,\n\t`SELECT * FROM test1 WHERE ExecutionTime IS NULL AND colD IS NOT NULL`,\n\t`SELECT * FROM test1 WHERE NOT ExecutionTime IS NULL OR colD IS NOT NULL`,\n\t`SELECT * FROM test1 WHERE colB = 'ab' AND (ExecutionTime IS NULL OR colD IS NOT NULL)`,\n\t`SELECT * FROM test1 WHERE NOT ExecutionTime IS NULL`,\n\t`SELECT * FROM test1 WHERE NOT ExecutionTime IS NOT NULL`,\n\t`SELECT ExecutionTime FROM test1 WHERE NOT colE BETWEEN 4 AND 15`,\n\t`SELECT * FROM test1 WHERE colE BETWEEN 3 AND 12`,\n\t`SELECT * FROM test1 WHERE NOT colE BETWEEN 3 AND 15 AND colD < 9 OR NOT colB != 'aa'`,\n\t`SELECT colA FROM test1 WHERE colB IN ('aa', 'ab', 'bb')`,\n\t`SELECT ExecutionTime FROM test1 WHERE colB NOT IN ('ab', 'bb') AND ExecutionTime IS NOT NULL`,\n\t`SELECT * FROM test1 WHERE colB NOT IN ('ab', 'bb') AND NOT (colE > 8 OR NOT colD <> 10)`,\n\t`SELECT colB FROM test1 GROUP BY colB ORDER BY colB`,\n\t`SELECT colB, colA FROM test1 GROUP BY colB, colB, colA, colA`,\n\t`SELECT colB FROM test1 WHERE colE > 6 and ExecutionTime IS NOT NULL GROUP BY colB`,\n\t`SELECT * FROM test1 WHERE colC REGEXP '[ab]{3} a{2}[ab] b+'`,\n\t`SELECT * FROM test1 WHERE colB LIKE '_a_' OR colB LIKE 'b%'`,\n\t`SELECT colB, colA FROM test1 GROUP BY colB, colA`,\n\t`SELECT COUNT(DISTINCT colB) FROM test1`,\n\t`SELECT COUNT(DISTINCT colB), COUNT(colB) FROM test1`,\n\t`SELECT COUNT(DISTINCT colB), COUNT(colB), COUNT(DISTINCT colB), COUNT(colB) FROM test1`,\n\t`SELECT COUNT(colA), COUNT(colB) FROM test1`,\n\t`SELECT COUNT(*), COUNT(colB) FROM test1`,\n\t`SELECT COUNT(DISTINCT colB), AVG(colE) FROM test1`,\n\t`SELECT COUNT(*) FROM test1`,\n\t`SELECT COUNT(colB) FROM test1`,\n\t`SELECT MIN(colE) FROM test1`,\n\t`SELECT colA AS a FROM test1`,\n\t\"SELECT `colA` FROM `test1` WHERE `colB` != 'ab'\",\n\t\"SELECT `colA` FROM `test1` WHERE `colB` != \\\"ab\\\"\",\n\t\"SELECT `colA` FROM `test1` WHERE `colE` BETWEEN 2 AND 10\",\n\t`SELECT COUNT(colB), AVG(colD), MAX(colE) FROM test1`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB`,\n\t`SELECT AVG(colE), COUNT(*), colB FROM test1 GROUP BY colB, colA`,\n\t`SELECT AVG(colE), COUNT(ExecutionTime) FROM test1 GROUP BY colB, colA`,\n\t`SELECT AVG(colE), COUNT(colA) FROM test1 GROUP BY colB, colA`,\n\t`SELECT COUNT(DISTINCT colE) FROM test1 GROUP BY colB, colA`,\n\t`SELECT MAX(colD), AVG(colE), COUNT(colA) FROM test1 GROUP BY colB, colA`,\n\t`SELECT MAX(colD), AVG(colE), COUNT(colA) FROM test1 WHERE ExecutionTime IS NOT NULL GROUP BY colB, colA`,\n\t`SELECT MAX(colD), AVG(colE), MIN(colD) FROM test1 WHERE ExecutionTime IS NOT NULL AND NOT colD < 2 GROUP BY colB, colA`,\n\t`SELECT MAX(colD), AVG(colE), MIN(colD) FROM test1 WHERE colE > 1 GROUP BY colB, colA`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY AVG(colE)`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY AVG(colE) DESC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY AVG(colE) ASC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY COUNT(*) ASC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY COUNT(colA) ASC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY COUNT(colA) ASC, COUNT(colA) ASC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY COUNT(DISTINCT colA) DESC`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY COUNT(DISTINCT colA) DESC, COUNT(colA), COUNT(*) DESC`,\n\t`SELECT AVG(colE), COUNT(colE), COUNT(DISTINCT colE) FROM test1 GROUP BY colB ORDER BY COUNT(colE), COUNT(DISTINCT colE) ASC`,\n\t`SELECT COUNT(DISTINCT colA) FROM test1 GROUP BY colB ORDER BY COUNT(colA), COUNT(*), COUNT(DISTINCT colA)`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING MAX(colD) > 4`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING COUNT(colD) > 4`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING COUNT(*) > 4`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING COUNT(DISTINCT colA) > 2`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING MAX(colD) > MIN(colE)`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING COUNT(DISTINCT colA) > MIN(colE)`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB HAVING MAX(colD) > COUNT(colD)`,\n\t`SELECT AVG(colE) FROM test1 WHERE colD > 2 GROUP BY colB HAVING MAX(colD) > COUNT(colD) OR MAX(colD) < AVG(colE) AND COUNT(colD) = COUNT(colD) OR COUNT(colD) <> MAX(colD)`,\n\t`SELECT COUNT(DISTINCT colB) FROM test1 GROUP BY colB`,\n\t`SELECT AVG(colE) FROM test1 GROUP BY colB ORDER BY MAX(colD)`,\n\t`SELECT * FROM test1 GROUP BY colB HAVING COUNT(*) > COUNT(colA)`,\n\t`SELECT * FROM test1 GROUP BY colB HAVING NOT (COUNT(*) > COUNT(colA))`,\n\t`SELECT * FROM test1 GROUP BY colB HAVING COUNT(*) BETWEEN 0 AND 50`,\n}\n\nvar dsls = []string{\n\t`{\"size\": 1000}`,\n\t`{\"size\": 10,\"from\": 4,\"sort\": [{\"colE\": \"asc\"},{\"colD\": \"desc\"}]}`,\n\t`{\"query\": {\"term\": {\"colB\": \"ab\"}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value == doc['colBB'].value\"}}}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"term\": {\"colB\": \"ab\"}},{\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value == doc['colBB'].value\"}}}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"term\": {\"colB\": \"ab\"}},{\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value !== doc['colBB'].value\"}}}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"term\": {\"colD\": \"10\"}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},\"size\": 1000}`,\n\t`{\"size\": 14,\"sort\": [{\"colD\": \"asc\"}],\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}}}`,\n\t`{\"query\": {\"term\": {\"colD\": \"10\"}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colD'].value + 1 > 9\"}}}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value == doc['colBB'].value\"}}}}},{\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"2 * doc['colD'].value > 8 + 2\"}}}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value + 'c' == doc['colA'].value\"}}}}},\"size\": 1000}`,\n\t`{\"size\": 1000,\"query\": {\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value + 'c' == doc['colA'].value\"}}}}}}`,\n\t`{\"query\": {\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"doc['colB'].value + 'c' == doc['colA'].value\"}}}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"term\": {\"colB\": \"ab\"}},{\"term\": {\"ExecutionTime\": \"2016\"}},{\"term\": {\"colB\": \"ab\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"term\": {\"colB\": \"ab\"}},{\"term\": {\"colD\": \"10\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"term\": {\"colB\": \"bc\"}}]}},{\"term\": {\"colB\": \"ab\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"bc\"}}}}]}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}},{\"range\": {\"colE\": {\"lt\": \"10\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"bool\": {\"should\": [{\"term\": {\"colB\": \"bc\"}},{\"term\": {\"colB\": \"ab\"}}]}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"term\": {\"colB\": \"bc\"}}]}},{\"term\": {\"colB\": \"ab\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"bool\": {\"should\": [{\"term\": {\"colB\": \"bc\"}},{\"term\": {\"colB\": \"ab\"}}]}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"term\": {\"colD\": \"10\"}},\"size\": 1000,\"sort\": [{\"colE\": \"desc\"},{\"colD\": \"desc\"}]}`,\n\t`{\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},\"size\": 1000}`,\n\t`{\"query\": {\"term\": {\"colD\": \"10\"}},\"size\": 1000,\"sort\": [{\"colE\": \"asc\"},{\"colD\": \"asc\"}]}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"term\": {\"colD\": \"10\"}},{\"term\": {\"colB\": \"bc\"}}]}},{\"term\": {\"colB\": \"ab\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"bc\"}}}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}},{\"term\": {\"colB\": \"bc\"}}]}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"term\": {\"colD\": \"10\"}},{\"bool\": {\"filter\": [{\"term\": {\"colB\": \"bc\"}},{\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}}]}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"range\": {\"colE\": {\"gt\": \"3\"}}},{\"range\": {\"colD\": {\"lte\": \"15\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"range\": {\"colE\": {\"lt\": \"5\"}}},{\"range\": {\"colD\": {\"gte\": \"17\"}}}]}},\"size\": 1000,\"sort\": [{\"colE\": \"desc\"},{\"colD\": \"asc\"}]}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"range\": {\"colE\": {\"lt\": \"5\"}}},{\"range\": {\"colD\": {\"gte\": \"17\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"range\": {\"colE\": {\"lt\": \"5\"}}},{\"range\": {\"colD\": {\"lt\": \"17\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"range\": {\"colE\": {\"lte\": \"9\"}}},{\"range\": {\"colD\": {\"gte\": \"6\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"range\": {\"colE\": {\"lte\": \"9\"}}},{\"range\": {\"colD\": {\"gte\": \"6\"}}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"range\": {\"colE\": {\"gt\": \"0\"}}},{\"range\": {\"colD\": {\"lte\": \"21.000\"}}}]}},\"size\": 1000}`,\n\t`{\"size\": 1000,\"sort\": [{\"colD\": \"asc\"}],\"query\": {\"bool\": {\"must_not\": {\"exists\": {\"field\": \"ExecutionTime\"}}}},\"_source\": {\"includes\": [\"colC\"]}}`,\n\t`{\"query\": {\"exists\": {\"field\": \"colB\"}},\"size\": 1000,\"sort\": [{\"colE\": \"asc\"}]}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"exists\": {\"field\": \"ExecutionTime\"}}}},{\"exists\": {\"field\": \"colD\"}}]}},\"size\": 1000}`,\n\t`{\"size\": 1000,\"query\": {\"bool\": {\"should\": [{\"exists\": {\"field\": \"ExecutionTime\"}},{\"exists\": {\"field\": \"colD\"}}]}}}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"term\": {\"colB\": \"ab\"}},{\"bool\": {\"should\": [{\"bool\": {\"must_not\": {\"exists\": {\"field\": \"ExecutionTime\"}}}},{\"exists\": {\"field\": \"colD\"}}]}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"exists\": {\"field\": \"ExecutionTime\"}},\"size\": 1000}`,\n\t`{\"size\": 1000,\"query\": {\"bool\": {\"must_not\": {\"exists\": {\"field\": \"ExecutionTime\"}}}}}`,\n\t`{\"query\": {\"bool\": {\"must_not\": [{\"range\": {\"colE\": {\"gte\": \"4\", \"lte\": \"15\"}}}]}},\"_source\": {\"includes\": [\"ExecutionTime\"]},\"size\": 1000}`,\n\t`{\"query\": {\"range\": {\"colE\": {\"gte\": \"3\", \"lte\": \"12\"}}},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"should\": [{\"bool\": {\"filter\": [{\"bool\": {\"must_not\": [{\"range\": {\"colE\": {\"gte\": \"3\", \"lte\": \"15\"}}}]}},{\"range\": {\"colD\": {\"lt\": \"9\"}}}]}},{\"term\": {\"colB\": \"aa\"}}]}},\"size\": 1000}`,\n\t`{\"query\": {\"terms\": {\"colB\": [\"aa\", \"ab\", \"bb\"]}},\"_source\": {\"includes\": [\"colA\"]},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"terms\": {\"colB\": [\"ab\", \"bb\"]}}}},{\"exists\": {\"field\": \"ExecutionTime\"}}]}},\"_source\": {\"includes\": [\"ExecutionTime\"]},\"size\": 1000}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"bool\": {\"must_not\": {\"terms\": {\"colB\": [\"ab\", \"bb\"]}}}},{\"range\": {\"colE\": {\"lte\": \"8\"}}},{\"bool\": {\"must_not\": {\"term\": {\"colD\": \"10\"}}}}]}},\"size\": 1000}`,\n\t`{\"_source\": {\"includes\": [\"colB\"]},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}}},\"size\": 0,\"_source\": {\"includes\": [\"colB\", \"colA\"]}}`,\n\t`{\"query\": {\"bool\": {\"filter\": [{\"range\": {\"colE\": {\"gt\": \"6\"}}},{\"exists\": {\"field\": \"ExecutionTime\"}}]}},\"_source\": {\"includes\": [\"colB\"]},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}}},\"size\": 0}`,\n\t`{\"query\": {\"regexp\": {\"colC\": \"[ab]{3} a{2}[ab] b+\"}},\"size\": 1000}`,\n\t`{\"size\": 1000,\"query\": {\"bool\": {\"should\": [{\"wildcard\": {\"colB\": {\"wildcard\": \"?a?\"}}},{\"wildcard\": {\"colB\": {\"wildcard\": \"b*\"}}}]}}}`,\n\t`{\"_source\": {\"includes\": [\"colB\", \"colA\"]},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}}},\"size\": 0}`,\n\t`{\"size\": 0,\"aggs\": {\"count_distinct_colB\": {\"cardinality\": {\"field\": \"colB\"}}}}`,\n\t`{\"size\": 0,\"aggs\": {\"count_distinct_colB\": {\"cardinality\": {\"field\": \"colB\"}},\"count_colB\": {\"value_count\": {\"field\": \"colB\"}}}}`,\n\t`{\"aggs\": {\"count_distinct_colB\": {\"cardinality\": {\"field\": \"colB\"}},\"count_colB\": {\"value_count\": {\"field\": \"colB\"}}},\"size\": 0}`,\n\t`{\"size\": 0,\"aggs\": {\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"count_colB\": {\"value_count\": {\"field\": \"colB\"}}}}`,\n\t`{\"aggs\": {\"count_colB\": {\"value_count\": {\"field\": \"colB\"}}},\"size\": 0}`,\n\t`{\"size\": 0,\"aggs\": {\"count_distinct_colB\": {\"cardinality\": {\"field\": \"colB\"}},\"avg_colE\": {\"avg\": {\"field\": \"colE\"}}}}`,\n\t`{\"size\": 0}`,\n\t`{\"aggs\": {\"count_colB\": {\"value_count\": {\"field\": \"colB\"}}},\"size\": 0}`,\n\t`{\"aggs\": {\"min_colE\": {\"min\": {\"field\": \"colE\"}}},\"size\": 0}`,\n\t`{\"_source\": {\"includes\": [\"colA\"]},\"size\": 1000}`,\n\t`{\"_source\": {\"includes\": [\"colA\"]},\"size\": 1000,\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}}}`,\n\t`{\"query\": {\"bool\": {\"must_not\": {\"term\": {\"colB\": \"ab\"}}}},\"_source\": {\"includes\": [\"colA\"]},\"size\": 1000}`,\n\t`{\"_source\": {\"includes\": [\"colA\"]},\"size\": 1000,\"query\": {\"range\": {\"colE\": {\"gte\": \"2\", \"lte\": \"10\"}}}}`,\n\t`{\"aggs\": {\"count_colB\": {\"value_count\": {\"field\": \"colB\"}},\"avg_colD\": {\"avg\": {\"field\": \"colD\"}},\"max_colE\": {\"max\": {\"field\": \"colE\"}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}}}}},\"size\": 0}`,\n\t`{\"_source\": {\"includes\": [\"colB\"]},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_ExecutionTime\": {\"value_count\": {\"field\": \"ExecutionTime\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"count_distinct_colE\": {\"cardinality\": {\"field\": \"colE\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}}}}},\"size\": 0}`,\n\t`{\"query\": {\"exists\": {\"field\": \"ExecutionTime\"}},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"min_colD\": {\"min\": {\"field\": \"colD\"}}}}},\"size\": 0,\"query\": {\"bool\": {\"filter\": [{\"exists\": {\"field\": \"ExecutionTime\"}},{\"range\": {\"colD\": {\"gte\": \"2\"}}}]}}}`,\n\t`{\"query\": {\"range\": {\"colE\": {\"gt\": \"1\"}}},\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}},{\"group_colA\": {\"terms\": {\"field\": \"colA\", \"missing_bucket\": true}}}]}, \"aggs\": {\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"min_colD\": {\"min\": {\"field\": \"colD\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"avg_colE\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"avg_colE\": {\"order\": \"desc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"avg_colE\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"_count\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_colA\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_colA\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"size\": 0,\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_distinct_colA\": {\"cardinality\": {\"field\": \"colA\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_distinct_colA\": {\"order\": \"desc\"}}], \"size\": 1000}}}}}}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_distinct_colA\": {\"cardinality\": {\"field\": \"colA\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_distinct_colA\": {\"order\": \"desc\"}},{\"count_colA\": {\"order\": \"asc\"}},{\"_count\": {\"order\": \"desc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colE\": {\"value_count\": {\"field\": \"colE\"}},\"count_distinct_colE\": {\"cardinality\": {\"field\": \"colE\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_colE\": {\"order\": \"asc\"}},{\"count_distinct_colE\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"count_distinct_colA\": {\"cardinality\": {\"field\": \"colA\"}},\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"count_colA\": {\"order\": \"asc\"}},{\"_count\": {\"order\": \"asc\"}},{\"count_distinct_colA\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"max_colD\": \"max_colD\"}, \"script\": \"params.max_colD > 4\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_colD\": {\"value_count\": {\"field\": \"colD\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"count_colD\": \"count_colD\"}, \"script\": \"params.count_colD > 4\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"_count\": \"_count\"}, \"script\": \"params._count > 4\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_distinct_colA\": {\"cardinality\": {\"field\": \"colA\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"count_distinct_colA\": \"count_distinct_colA\"}, \"script\": \"params.count_distinct_colA > 2\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"min_colE\": {\"min\": {\"field\": \"colE\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"max_colD\": \"max_colD\",\"min_colE\": \"min_colE\"}, \"script\": \"params.max_colD > params.min_colE\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"count_distinct_colA\": {\"cardinality\": {\"field\": \"colA\"}},\"min_colE\": {\"min\": {\"field\": \"colE\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"count_distinct_colA\": \"count_distinct_colA\",\"min_colE\": \"min_colE\"}, \"script\": \"params.count_distinct_colA > params.min_colE\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"count_colD\": {\"value_count\": {\"field\": \"colD\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"count_colD\": \"count_colD\",\"max_colD\": \"max_colD\"}, \"script\": \"params.max_colD > params.count_colD\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"count_colD\": {\"value_count\": {\"field\": \"colD\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"avg_colE\": \"avg_colE\",\"max_colD\": \"max_colD\",\"count_colD\": \"count_colD\"}, \"script\": \"params.max_colD > params.count_colD || params.max_colD < params.avg_colE && params.count_colD == params.count_colD || params.count_colD !== params.max_colD\"}}}}},\"size\": 0,\"query\": {\"range\": {\"colD\": {\"gt\": \"2\"}}}}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"count_distinct_colB\": {\"cardinality\": {\"field\": \"colB\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"avg_colE\": {\"avg\": {\"field\": \"colE\"}},\"max_colD\": {\"max\": {\"field\": \"colD\"}},\"bucket_sort\": {\"bucket_sort\": {\"sort\": [{\"max_colD\": {\"order\": \"asc\"}}], \"size\": 1000}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"_count\": \"_count\",\"count_colA\": \"count_colA\"}, \"script\": \"params._count > params.count_colA\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"count_colA\": {\"value_count\": {\"field\": \"colA\"}},\"having\": {\"bucket_selector\": {\"buckets_path\": {\"_count\": \"_count\",\"count_colA\": \"count_colA\"}, \"script\": \"!(params._count > params.count_colA)\"}}}}},\"size\": 0}`,\n\t`{\"aggs\": {\"groupby\": {\"composite\": {\"size\": 1000, \"sources\": [{\"group_colB\": {\"terms\": {\"field\": \"colB\", \"missing_bucket\": true}}}]}, \"aggs\": {\"having\": {\"bucket_selector\": {\"buckets_path\": {\"_count\": \"_count\"}, \"script\": \"(params._count >= 0 && params._count <= 50)\"}}}}},\"size\": 0}`,\n}\n\nfunc TestUnit(t *testing.T) {\n\te := NewESql()\n\tfor i, sql := range sqls {\n\t\tfmt.Printf(\"test %dth query ...\\n\", i+1)\n\t\tdsl, _, err := e.Convert(sql)\n\t\tif err != nil {\n\t\t\tt.Errorf(\"%vth query fails: %v\", i+1, err)\n\t\t\treturn\n\t\t}\n\t\tvar dslMap, dslMapRef map[string]interface{}\n\t\terr = json.Unmarshal([]byte(dsl), &dslMap)\n\t\tif err != nil {\n\t\t\tt.Errorf(\"%vth query fails\", i+1)\n\t\t\treturn\n\t\t}\n\t\terr = json.Unmarshal([]byte(dsls[i]), &dslMapRef)\n\t\tif err != nil {\n\t\t\tt.Errorf(\"%vth query reference fails\", i+1)\n\t\t\treturn\n\t\t}\n\t\tif !reflect.DeepEqual(dslMap, dslMapRef) {\n\t\t\tt.Errorf(\"%vth query does not match\", i+1)\n\t\t\treturn\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/globals.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport \"github.com/xwb1989/sqlparser\"\n\n// used for invert operator when NOT is specified\nvar oppositeOperator = map[string]string{\n\t\"=\":                     \"!=\",\n\t\"!=\":                    \"=\",\n\t\"<\":                     \">=\",\n\t\"<=\":                    \">\",\n\t\">\":                     \"<=\",\n\t\">=\":                    \"<\",\n\t\"<>\":                    \"=\",\n\t\"in\":                    \"not in\",\n\t\"like\":                  \"not like\",\n\t\"regexp\":                \"not regexp\",\n\t\"not in\":                \"in\",\n\t\"not like\":              \"like\",\n\t\"not regexp\":            \"regexp\",\n\tsqlparser.IsNullStr:     sqlparser.IsNotNullStr,\n\tsqlparser.IsNotNullStr:  sqlparser.IsNullStr,\n\tsqlparser.BetweenStr:    sqlparser.NotBetweenStr,\n\tsqlparser.NotBetweenStr: sqlparser.BetweenStr,\n}\n\n// used for convert SQL operator to painless operator in HAVING expression\nvar op2PainlessOp = map[string]string{\n\t\"=\":  \"==\",\n\t\"!=\": \"!==\",\n\t\"<\":  \"<\",\n\t\"<=\": \"<=\",\n\t\">\":  \">\",\n\t\">=\": \">=\",\n\t\"<>\": \"!==\",\n}\n\nvar opBinaryExpr = map[string]string{\n\t\"|\":  \"|\",\n\t\"&\":  \"&\",\n\t\"^\":  \"^\",\n\t\"+\":  \"+\",\n\t\"-\":  \"-\",\n\t\"*\":  \"*\",\n\t\"/\":  \"/\",\n\t\"%\":  \"%\",\n\t\">>\": \">>\",\n\t\"<<\": \"<<\",\n}\n\n// default sizes and identifiers used in cadence visibility\nconst (\n\tDefaultPageSize      = 1000\n\tDefaultBucketNumber  = 1000\n\tESDefaultMaxPageSize = 10000\n\tTieBreaker           = \"RunID\"\n\tRunID                = \"RunID\"\n\tStartTime            = \"StartTime\"\n\tDomainID             = \"DomainID\"\n\tWorkflowID           = \"WorkflowID\"\n\tExecutionTime        = \"ExecutionTime\"\n\tTieBreakerOrder      = \"desc\"\n\tStartTimeOrder       = \"desc\"\n)\n"
  },
  {
    "path": "common/elasticsearch/esql/having.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\nfunc (e *ESql) getAggHaving(having *sqlparser.Where) (string, []string, []string, []string, map[string]int, error) {\n\tvar aggNameSlice, aggTargetSlice, aggTagSlice []string\n\taggTagSet := make(map[string]int)\n\tvar script string\n\tvar err error\n\tif having != nil {\n\t\tscript, err = e.convertHavingExpr(having.Expr, &aggNameSlice, &aggTargetSlice, &aggTagSlice, aggTagSet)\n\t\tif err != nil {\n\t\t\treturn \"\", nil, nil, nil, nil, err\n\t\t}\n\t}\n\treturn script, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet, nil\n}\n\nfunc (e *ESql) convertHavingExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\tswitch expr.(type) {\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn e.convertHavingComparisionExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tcase *sqlparser.AndExpr:\n\t\treturn e.convertHavingAndExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tcase *sqlparser.OrExpr:\n\t\treturn e.convertHavingOrExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tcase *sqlparser.NotExpr:\n\t\treturn e.convertHavingNotExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tcase *sqlparser.ParenExpr:\n\t\treturn e.convertHavingParenExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tcase *sqlparser.RangeCond:\n\t\treturn e.convertHavingBetweenExpr(expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tdefault:\n\t\terr := fmt.Errorf(`esql: %T expression in HAVING no supported`, expr)\n\t\treturn \"\", err\n\t}\n}\n\nfunc (e *ESql) convertHavingBetweenExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\trangeCond := expr.(*sqlparser.RangeCond)\n\tlhs := rangeCond.Left\n\tfrom, to := rangeCond.From, rangeCond.To\n\tvar expr1 sqlparser.Expr = &sqlparser.ComparisonExpr{Left: lhs, Right: from, Operator: \">=\"}\n\tvar expr2 sqlparser.Expr = &sqlparser.ComparisonExpr{Left: lhs, Right: to, Operator: \"<=\"}\n\tvar expr3 sqlparser.Expr = &sqlparser.AndExpr{Left: expr1, Right: expr2}\n\n\tscript, err := e.convertHavingAndExpr(expr3, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// here parenthesis is to deal with the case when an not(!) operator out side\n\t// if no parenthesis, NOT xxx BETWEEN a and b -> !xxx > a && xxx < b\n\tscript = fmt.Sprintf(`(%v)`, script)\n\treturn script, nil\n}\n\nfunc (e *ESql) convertHavingAndExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\tandExpr := expr.(*sqlparser.AndExpr)\n\tleftExpr := andExpr.Left\n\trightExpr := andExpr.Right\n\tscriptLeft, err := e.convertHavingExpr(leftExpr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tscriptRight, err := e.convertHavingExpr(rightExpr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(`%v && %v`, scriptLeft, scriptRight), nil\n}\n\nfunc (e *ESql) convertHavingOrExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\torExpr := expr.(*sqlparser.OrExpr)\n\tleftExpr := orExpr.Left\n\trightExpr := orExpr.Right\n\tscriptLeft, err := e.convertHavingExpr(leftExpr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tscriptRight, err := e.convertHavingExpr(rightExpr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(`%v || %v`, scriptLeft, scriptRight), nil\n}\n\nfunc (e *ESql) convertHavingParenExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\tparenExpr := expr.(*sqlparser.ParenExpr)\n\tscript, err := e.convertHavingExpr(parenExpr.Expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(`(%v)`, script), nil\n}\n\nfunc (e *ESql) convertHavingNotExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\tnotExpr := expr.(*sqlparser.NotExpr)\n\tscript, err := e.convertHavingExpr(notExpr.Expr, aggNameSlice, aggTargetSlice, aggTagSlice, aggTagSet)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(`!%v`, script), nil\n}\n\nfunc (e *ESql) convertHavingComparisionExpr(expr sqlparser.Expr, aggNameSlice *[]string, aggTargetSlice *[]string,\n\taggTagSlice *[]string, aggTagSet map[string]int) (string, error) {\n\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\tvar funcExprs []*sqlparser.FuncExpr\n\n\tif _, exist := op2PainlessOp[comparisonExpr.Operator]; !exist {\n\t\terr := fmt.Errorf(`esql: %s operator not supported in having comparison clause`, comparisonExpr.Operator)\n\t\treturn \"\", err\n\t}\n\t// convert SQL operator format to equivalent painless operator\n\top := op2PainlessOp[comparisonExpr.Operator]\n\n\t// lhs\n\tleftFuncExpr, ok := comparisonExpr.Left.(*sqlparser.FuncExpr)\n\tif !ok {\n\t\terr := fmt.Errorf(\"esql: found %v in HAVING which is not aggregation function\", sqlparser.String(comparisonExpr.Left))\n\t\treturn \"\", err\n\t}\n\tfuncExprs = append(funcExprs, leftFuncExpr)\n\n\t// rhs, can be a value or an aggregation function\n\tvar rhsStr, script string\n\tswitch comparisonExpr.Right.(type) {\n\tcase *sqlparser.SQLVal:\n\t\trhsStr = sqlparser.String(comparisonExpr.Right)\n\t\trhsStr = strings.Trim(rhsStr, `'`)\n\tcase *sqlparser.FuncExpr:\n\t\trightFuncExpr := comparisonExpr.Right.(*sqlparser.FuncExpr)\n\t\tfuncExprs = append(funcExprs, rightFuncExpr)\n\tdefault:\n\t\terr := fmt.Errorf(\"esql: %T in HAVING rhs not supported\", comparisonExpr.Right)\n\t\treturn \"\", err\n\t}\n\n\tfor _, funcExpr := range funcExprs {\n\t\taggNameStr := strings.ToLower(funcExpr.Name.String())\n\t\taggTargetStr := sqlparser.String(funcExpr.Exprs)\n\t\taggTargetStr = strings.Trim(aggTargetStr, \"`\")\n\t\taggTargetStr, err := e.keyProcess(aggTargetStr)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\n\t\tvar aggTagStr string\n\t\tswitch aggNameStr {\n\t\tcase \"count\":\n\t\t\tif aggTargetStr == \"*\" {\n\t\t\t\taggTagStr = \"_count\"\n\t\t\t} else if funcExpr.Distinct {\n\t\t\t\taggTagStr = aggNameStr + \"_distinct_\" + aggTargetStr\n\t\t\t\taggNameStr = \"cardinality\"\n\t\t\t} else {\n\t\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\t\t\taggNameStr = \"value_count\"\n\t\t\t}\n\t\tcase \"avg\", \"sum\", \"min\", \"max\":\n\t\t\tif funcExpr.Distinct {\n\t\t\t\terr := fmt.Errorf(`esql: HAVING: aggregation function %v w/ DISTINCT not supported`, aggNameStr)\n\t\t\t\treturn \"\", err\n\t\t\t}\n\t\t\taggTagStr = aggNameStr + \"_\" + aggTargetStr\n\t\tdefault:\n\t\t\terr := fmt.Errorf(`esql: HAVING: aggregation function %v not supported`, aggNameStr)\n\t\t\treturn \"\", err\n\t\t}\n\t\taggTagStr = strings.Replace(aggTagStr, \".\", \"_\", -1)\n\t\taggTagSet[aggTagStr] = len(*aggNameSlice)\n\t\t*aggNameSlice = append(*aggNameSlice, aggNameStr)\n\t\t*aggTargetSlice = append(*aggTargetSlice, aggTargetStr)\n\t\t*aggTagSlice = append(*aggTagSlice, aggTagStr)\n\t}\n\n\tn := len(*aggTagSlice)\n\tif rhsStr == \"\" {\n\t\tscript = fmt.Sprintf(`params.%v %v params.%v`, (*aggTagSlice)[n-2], op, (*aggTagSlice)[n-1])\n\t} else {\n\t\tscript = fmt.Sprintf(`params.%v %v %v`, (*aggTagSlice)[n-1], op, rhsStr)\n\t}\n\treturn script, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/scriptQuery.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\nfunc (e *ESql) convertToScript(expr sqlparser.Expr) (script string, err error) {\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.ColName:\n\t\tscript, err = e.convertColName(expr)\n\t\tscript = fmt.Sprintf(`doc['%v'].value`, script)\n\tcase *sqlparser.SQLVal:\n\t\tscript, err = e.convertValExpr(expr, true)\n\tcase *sqlparser.BinaryExpr:\n\t\tscript, err = e.convertBinaryExpr(expr)\n\tcase *sqlparser.ParenExpr:\n\t\tscript, err = e.convertToScript(expr.Expr)\n\t\tscript = fmt.Sprintf(`(%v)`, script)\n\tdefault:\n\t\terr = fmt.Errorf(\"esql: invalid expression type for scripting\")\n\t}\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn script, nil\n}\n\nfunc (e *ESql) convertBinaryExpr(expr sqlparser.Expr) (string, error) {\n\tvar script, lhsStr, rhsStr string\n\tvar err error\n\tbinExpr, ok := expr.(*sqlparser.BinaryExpr)\n\tif !ok {\n\t\terr = fmt.Errorf(\"esql: invalid binary expression\")\n\t\treturn \"\", err\n\t}\n\tlhsExpr, rhsExpr := binExpr.Left, binExpr.Right\n\top, ok := opBinaryExpr[binExpr.Operator]\n\tif !ok {\n\t\terr = fmt.Errorf(\"esql: not supported binary expression operator\")\n\t\treturn \"\", err\n\t}\n\n\tlhsStr, err = e.convertToScript(lhsExpr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\trhsStr, err = e.convertToScript(rhsExpr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tscript = fmt.Sprintf(`%v %v %v`, lhsStr, op, rhsStr)\n\treturn script, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/esql/select.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esql\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/xwb1989/sqlparser\"\n)\n\nfunc (e *ESql) convertSelect(sel sqlparser.Select, domainID string, pagination ...interface{}) (dsl string, sortField []string, err error) {\n\tif sel.Distinct != \"\" {\n\t\terr := fmt.Errorf(`esql: SELECT DISTINCT not supported. use GROUP BY instead`)\n\t\treturn \"\", nil, err\n\t}\n\n\tvar rootParent sqlparser.Expr\n\t// a map that contains the main components of a query\n\tdslMap := make(map[string]interface{})\n\n\t// handle WHERE keyword\n\tif sel.Where != nil {\n\t\tdslQuery, err := e.convertWhereExpr(sel.Where.Expr, rootParent)\n\t\tif err != nil {\n\t\t\treturn \"\", nil, err\n\t\t}\n\t\tdslMap[\"query\"] = dslQuery\n\t}\n\t// cadence special handling: add domain ID query and time query bounds\n\tif e.cadence {\n\t\te.addCadenceDomainTimeQuery(sel, domainID, dslMap)\n\t}\n\n\t// handle FROM keyword, currently only support 1 target table\n\tif len(sel.From) != 1 {\n\t\tif len(sel.From) == 0 {\n\t\t\terr = fmt.Errorf(\"esql: invalid from expressino: no from expression specified\")\n\t\t} else {\n\t\t\terr = fmt.Errorf(\"esql: join not supported\")\n\t\t}\n\t\treturn \"\", nil, err\n\t}\n\n\t// handle SELECT keyword\n\t_, selectedColNameSlice, aggNameSlice, err := e.extractSelectedExpr(sel.SelectExprs)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\tif len(selectedColNameSlice) > 0 {\n\t\tcolNames := `\"` + strings.Join(selectedColNameSlice, `\", \"`) + `\"`\n\t\tdslMap[\"_source\"] = fmt.Sprintf(`{\"includes\": [%v]}`, colNames)\n\t}\n\n\t// handle all aggregations, including GROUP BY, SELECT <agg function>, ORDER BY <agg function>, HAVING\n\tdslAgg, err := e.convertAgg(sel)\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\tif dslAgg != \"\" || len(aggNameSlice) > 0 {\n\t\tif dslAgg != \"\" {\n\t\t\tdslMap[\"aggs\"] = dslAgg\n\t\t}\n\t\t// do not return document contents if this is an aggregation query\n\t\tdslMap[\"size\"] = 0\n\t} else {\n\t\t// handle LIMIT and OFFSET keyword, these 2 keywords only works in non-aggregation query\n\t\tdslMap[\"size\"] = e.pageSize\n\t\tif sel.Limit != nil {\n\t\t\tif sel.Limit.Offset != nil {\n\t\t\t\tdslMap[\"from\"] = sqlparser.String(sel.Limit.Offset)\n\t\t\t}\n\t\t\tdslMap[\"size\"] = sqlparser.String(sel.Limit.Rowcount)\n\t\t}\n\t\t// handle pagination\n\t\tvar searchAfterSlice []string\n\t\tfor _, v := range pagination {\n\t\t\tswitch v.(type) {\n\t\t\tcase int:\n\t\t\t\tsearchAfterSlice = append(searchAfterSlice, fmt.Sprintf(`%v`, v))\n\t\t\tdefault:\n\t\t\t\tsearchAfterSlice = append(searchAfterSlice, fmt.Sprintf(`\"%v\"`, v))\n\t\t\t}\n\t\t}\n\t\tif len(searchAfterSlice) > 0 {\n\t\t\tsearchAfterStr := strings.Join(searchAfterSlice, \",\")\n\t\t\tdslMap[\"search_after\"] = fmt.Sprintf(`[%v]`, searchAfterStr)\n\t\t}\n\t}\n\n\t// handle ORDER BY <column name>\n\t// if it is an aggregate query, no point to order\n\tif _, exist := dslMap[\"aggs\"]; !exist && len(aggNameSlice) == 0 {\n\t\tvar orderBySlice []string\n\t\tfor _, orderExpr := range sel.OrderBy {\n\t\t\tvar colNameStr string\n\t\t\tif colName, ok := orderExpr.Expr.(*sqlparser.ColName); ok {\n\t\t\t\tcolNameStr, err = e.convertColName(colName)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn \"\", nil, err\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\terr := fmt.Errorf(`esql: mix order by aggregations and column names`)\n\t\t\t\treturn \"\", nil, err\n\t\t\t}\n\t\t\tcolNameStr = strings.Trim(colNameStr, \"`\")\n\t\t\torderByStr := fmt.Sprintf(`{\"%v\": \"%v\"}`, colNameStr, orderExpr.Direction)\n\t\t\torderBySlice = append(orderBySlice, orderByStr)\n\t\t\tsortField = append(sortField, colNameStr)\n\t\t}\n\t\t// cadence special handling: add runID as sorting tie breaker\n\t\tif e.cadence {\n\t\t\torderBySlice, sortField, err = e.addCadenceSort(orderBySlice, sortField)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", nil, err\n\t\t\t}\n\t\t}\n\t\tif len(orderBySlice) > 0 {\n\t\t\tdslMap[\"sort\"] = fmt.Sprintf(\"[%v]\", strings.Join(orderBySlice, \",\"))\n\t\t}\n\t}\n\n\t// generate the final json query\n\tvar dslQuerySlice []string\n\tfor tag, content := range dslMap {\n\t\tdslQuerySlice = append(dslQuerySlice, fmt.Sprintf(`\"%v\": %v`, tag, content))\n\t}\n\tdsl = \"{\" + strings.Join(dslQuerySlice, \",\") + \"}\"\n\treturn dsl, sortField, nil\n}\n\nfunc (e *ESql) convertWhereExpr(expr sqlparser.Expr, parent sqlparser.Expr) (string, error) {\n\tvar err error\n\tif expr == nil {\n\t\terr = fmt.Errorf(\"esql: invalid where expression, where expression should not be nil\")\n\t\treturn \"\", err\n\t}\n\n\tswitch expr.(type) {\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn e.convertComparisionExpr(expr, parent, false)\n\tcase *sqlparser.AndExpr:\n\t\treturn e.convertAndExpr(expr, parent)\n\tcase *sqlparser.OrExpr:\n\t\treturn e.convertOrExpr(expr, parent)\n\tcase *sqlparser.ParenExpr:\n\t\treturn e.convertParenExpr(expr, parent)\n\tcase *sqlparser.NotExpr:\n\t\treturn e.convertNotExpr(expr, parent)\n\tcase *sqlparser.RangeCond:\n\t\treturn e.convertBetweenExpr(expr, parent, true, true, false)\n\tcase *sqlparser.IsExpr:\n\t\treturn e.convertIsExpr(expr, parent, false)\n\tdefault:\n\t\terr = fmt.Errorf(`esql: %T expression not supported in WHERE clause`, expr)\n\t\treturn \"\", err\n\t}\n}\n\nfunc (e *ESql) convertBetweenExpr(expr sqlparser.Expr, parent sqlparser.Expr, fromInclusive bool, toInclusive bool, not bool) (string, error) {\n\trangeCond := expr.(*sqlparser.RangeCond)\n\tlhs, ok := rangeCond.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\terr := fmt.Errorf(\"esql: invalid range column name\")\n\t\treturn \"\", err\n\t}\n\tlhsStr, err := e.convertColName(lhs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tfromStr := strings.Trim(sqlparser.String(rangeCond.From), `'`)\n\ttoStr := strings.Trim(sqlparser.String(rangeCond.To), `'`)\n\top := rangeCond.Operator\n\tif not {\n\t\top = oppositeOperator[op]\n\t}\n\n\tgt := \"gte\"\n\tlt := \"lte\"\n\tif !fromInclusive {\n\t\tgt = \"gt\"\n\t}\n\tif !toInclusive {\n\t\tlt = \"lt\"\n\t}\n\n\tdsl := fmt.Sprintf(`{\"range\": {\"%v\": {\"%v\": \"%v\", \"%v\": \"%v\"}}}`, lhsStr, gt, fromStr, lt, toStr)\n\tif op == sqlparser.NotBetweenStr {\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": [%v]}}`, dsl)\n\t}\n\treturn dsl, nil\n}\n\nfunc (e *ESql) convertParenExpr(expr sqlparser.Expr, parent sqlparser.Expr) (string, error) {\n\texprInside := expr.(*sqlparser.ParenExpr).Expr\n\treturn e.convertWhereExpr(exprInside, expr)\n}\n\n// * dsl must_not is not an equivalent to sql NOT, should convert the inside expression accordingly\nfunc (e *ESql) convertNotExpr(expr sqlparser.Expr, parent sqlparser.Expr) (string, error) {\n\tnotExpr := expr.(*sqlparser.NotExpr)\n\texprInside := notExpr.Expr\n\tswitch exprInside := (exprInside).(type) {\n\tcase *sqlparser.NotExpr:\n\t\texpr1 := exprInside\n\t\texpr2 := expr1.Expr\n\t\treturn e.convertWhereExpr(expr2, parent)\n\tcase *sqlparser.AndExpr:\n\t\texpr1 := exprInside\n\t\tvar exprLeft sqlparser.Expr = &sqlparser.NotExpr{Expr: expr1.Left}\n\t\tvar exprRight sqlparser.Expr = &sqlparser.NotExpr{Expr: expr1.Right}\n\t\tvar expr2 sqlparser.Expr = &sqlparser.OrExpr{Left: exprLeft, Right: exprRight}\n\t\treturn e.convertOrExpr(expr2, parent)\n\tcase *sqlparser.OrExpr:\n\t\texpr1 := exprInside\n\t\tvar exprLeft sqlparser.Expr = &sqlparser.NotExpr{Expr: expr1.Left}\n\t\tvar exprRight sqlparser.Expr = &sqlparser.NotExpr{Expr: expr1.Right}\n\t\tvar expr2 sqlparser.Expr = &sqlparser.AndExpr{Left: exprLeft, Right: exprRight}\n\t\treturn e.convertAndExpr(expr2, parent)\n\tcase *sqlparser.ParenExpr:\n\t\texpr1 := exprInside\n\t\texprBody := expr1.Expr\n\t\tvar expr2 sqlparser.Expr = &sqlparser.NotExpr{Expr: exprBody}\n\t\treturn e.convertNotExpr(expr2, parent)\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn e.convertComparisionExpr(exprInside, parent, true)\n\tcase *sqlparser.IsExpr:\n\t\treturn e.convertIsExpr(exprInside, parent, true)\n\tcase *sqlparser.RangeCond:\n\t\treturn e.convertBetweenExpr(exprInside, parent, true, true, true)\n\tdefault:\n\t\terr := fmt.Errorf(\"esql: %T expression not supported\", exprInside)\n\t\treturn \"\", err\n\t}\n}\n\nfunc (e *ESql) convertAndExpr(expr sqlparser.Expr, parent sqlparser.Expr) (string, error) {\n\tandExpr := expr.(*sqlparser.AndExpr)\n\tlhsExpr := andExpr.Left\n\trhsExpr := andExpr.Right\n\n\tlhsStr, err := e.convertWhereExpr(lhsExpr, expr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\trhsStr, err := e.convertWhereExpr(rhsExpr, expr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tvar dsl string\n\tif lhsStr == \"\" || rhsStr == \"\" {\n\t\tdsl = lhsStr + rhsStr\n\t} else {\n\t\tdsl = lhsStr + `,` + rhsStr\n\t}\n\n\t// merge chained AND expression\n\tif _, ok := parent.(*sqlparser.AndExpr); ok {\n\t\treturn dsl, nil\n\t}\n\treturn fmt.Sprintf(`{\"bool\": {\"filter\": [%v]}}`, dsl), nil\n}\n\nfunc (e *ESql) convertOrExpr(expr sqlparser.Expr, parent sqlparser.Expr) (string, error) {\n\torExpr := expr.(*sqlparser.OrExpr)\n\tlhsExpr := orExpr.Left\n\trhsExpr := orExpr.Right\n\n\tlhsStr, err := e.convertWhereExpr(lhsExpr, expr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\trhsStr, err := e.convertWhereExpr(rhsExpr, expr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tvar dsl string\n\tif lhsStr == \"\" || rhsStr == \"\" {\n\t\tdsl = lhsStr + rhsStr\n\t} else {\n\t\tdsl = lhsStr + `,` + rhsStr\n\t}\n\n\t// merge chained OR expression\n\tif _, ok := parent.(*sqlparser.OrExpr); ok {\n\t\treturn dsl, nil\n\t}\n\treturn fmt.Sprintf(`{\"bool\": {\"should\": [%v]}}`, dsl), nil\n}\n\nfunc (e *ESql) convertIsExpr(expr sqlparser.Expr, parent sqlparser.Expr, not bool) (string, error) {\n\tisExpr := expr.(*sqlparser.IsExpr)\n\tlhs, ok := isExpr.Expr.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", fmt.Errorf(\"esql: is expression only support colname missing check\")\n\t}\n\tlhsStr, err := e.convertColName(lhs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tdsl := \"\"\n\top := isExpr.Operator\n\tif not {\n\t\tif _, exist := oppositeOperator[op]; !exist {\n\t\t\terr := fmt.Errorf(\"esql: is expression only support is null and is not null\")\n\t\t\treturn \"\", err\n\t\t}\n\t\top = oppositeOperator[op]\n\t}\n\tswitch op {\n\tcase sqlparser.IsNullStr:\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": {\"exists\": {\"field\": \"%v\"}}}}`, lhsStr)\n\tcase sqlparser.IsNotNullStr:\n\t\tdsl = fmt.Sprintf(`{\"exists\": {\"field\": \"%v\"}}`, lhsStr)\n\tdefault:\n\t\treturn \"\", fmt.Errorf(\"esql: is expression only support is null and is not null\")\n\t}\n\treturn dsl, nil\n}\n\nfunc (e *ESql) convertComparisionExpr(expr sqlparser.Expr, parent sqlparser.Expr, not bool) (string, error) {\n\t// extract lhs, and check lhs is a colName\n\tvar err error\n\tscriptQuery := false\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\tlhsExpr, rhsExpr := comparisonExpr.Left, comparisonExpr.Right\n\tvar lhsStr, rhsStr, dsl string\n\t// get operator\n\top := comparisonExpr.Operator\n\tif not {\n\t\tif _, exist := oppositeOperator[op]; !exist {\n\t\t\terr := fmt.Errorf(`esql: %s operator not supported in comparison clause`, comparisonExpr.Operator)\n\t\t\treturn \"\", err\n\t\t}\n\t\top = oppositeOperator[op]\n\t}\n\n\tif _, ok := lhsExpr.(*sqlparser.ColName); !ok {\n\t\tscriptQuery = true\n\t}\n\tswitch rhsExpr.(type) {\n\tcase *sqlparser.SQLVal, sqlparser.ValTuple:\n\t\trhsStr, err = e.convertValExpr(rhsExpr, false)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\trhsStr, err = e.valueProcess(lhsStr, rhsStr)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\tdefault:\n\t\tscriptQuery = true\n\t}\n\n\t// use painless scripting query here\n\tif scriptQuery {\n\t\tlhsStr, err = e.convertToScript(lhsExpr)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\trhsStr, err = e.convertToScript(rhsExpr)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\top, ok := op2PainlessOp[op]\n\t\tif !ok {\n\t\t\terr = fmt.Errorf(\"esql: not supported painless operator\")\n\t\t\treturn \"\", err\n\t\t}\n\t\tscript := fmt.Sprintf(`%v %v %v`, lhsStr, op, rhsStr)\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"filter\": {\"script\": {\"script\": {\"source\": \"%v\"}}}}}`, script)\n\t\treturn dsl, nil\n\t}\n\n\tlhs := lhsExpr.(*sqlparser.ColName)\n\tlhsStr, err = e.convertColName(lhs)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// generate dsl according to operator\n\tswitch op {\n\tcase \"=\":\n\t\tdsl = fmt.Sprintf(`{\"term\": {\"%v\": \"%v\"}}`, lhsStr, rhsStr)\n\tcase \"<\":\n\t\tdsl = fmt.Sprintf(`{\"range\": {\"%v\": {\"lt\": \"%v\"}}}`, lhsStr, rhsStr)\n\tcase \"<=\":\n\t\tdsl = fmt.Sprintf(`{\"range\": {\"%v\": {\"lte\": \"%v\"}}}`, lhsStr, rhsStr)\n\tcase \">\":\n\t\tdsl = fmt.Sprintf(`{\"range\": {\"%v\": {\"gt\": \"%v\"}}}`, lhsStr, rhsStr)\n\tcase \">=\":\n\t\tdsl = fmt.Sprintf(`{\"range\": {\"%v\": {\"gte\": \"%v\"}}}`, lhsStr, rhsStr)\n\tcase \"<>\", \"!=\":\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": {\"term\": {\"%v\": \"%v\"}}}}`, lhsStr, rhsStr)\n\tcase \"in\":\n\t\trhsStr = strings.Replace(rhsStr, `'`, `\"`, -1)\n\t\trhsStr = strings.Trim(rhsStr, \"(\")\n\t\trhsStr = strings.Trim(rhsStr, \")\")\n\t\tdsl = fmt.Sprintf(`{\"terms\": {\"%v\": [%v]}}`, lhsStr, rhsStr)\n\tcase \"not in\":\n\t\trhsStr = strings.Replace(rhsStr, `'`, `\"`, -1)\n\t\trhsStr = strings.Trim(rhsStr, \"(\")\n\t\trhsStr = strings.Trim(rhsStr, \")\")\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": {\"terms\": {\"%v\": [%v]}}}}`, lhsStr, rhsStr)\n\tcase \"like\":\n\t\trhsStr = strings.Replace(rhsStr, `_`, `?`, -1)\n\t\trhsStr = strings.Replace(rhsStr, `%`, `*`, -1)\n\t\tdsl = fmt.Sprintf(`{\"wildcard\": {\"%v\": {\"wildcard\": \"%v\"}}}`, lhsStr, rhsStr)\n\tcase \"not like\":\n\t\trhsStr = strings.Replace(rhsStr, `_`, `?`, -1)\n\t\trhsStr = strings.Replace(rhsStr, `%`, `*`, -1)\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": {\"wildcard\": {\"%v\": {\"wildcard\": \"%v\"}}}}}`, lhsStr, rhsStr)\n\tcase \"regexp\":\n\t\tdsl = fmt.Sprintf(`{\"regexp\": {\"%v\": \"%v\"}}`, lhsStr, rhsStr)\n\tcase \"not regexp\":\n\t\tdsl = fmt.Sprintf(`{\"bool\": {\"must_not\": {\"regexp\": {\"%v\": \"%v\"}}}}`, lhsStr, rhsStr)\n\tdefault:\n\t\terr := fmt.Errorf(`esql: %s operator not supported in comparison clause`, comparisonExpr.Operator)\n\t\treturn \"\", err\n\t}\n\treturn dsl, nil\n}\n\nfunc (e *ESql) convertValExpr(expr sqlparser.Expr, script bool) (dsl string, err error) {\n\tswitch expr.(type) {\n\tcase *sqlparser.SQLVal:\n\t\tdsl = sqlparser.String(expr)\n\t\tif !script {\n\t\t\tdsl = strings.Trim(dsl, `'`)\n\t\t}\n\t// ValTuple is not a pointer from sqlparser\n\tcase sqlparser.ValTuple:\n\t\tdsl = sqlparser.String(expr)\n\tdefault:\n\t\terr = fmt.Errorf(\"esql: not supported rhs expression %T\", expr)\n\t\treturn \"\", err\n\t}\n\treturn dsl, nil\n}\n\nfunc (e *ESql) convertColName(colName *sqlparser.ColName) (string, error) {\n\t// here we garuantee colName is of type *ColName\n\tcolNameStr := sqlparser.String(colName)\n\treplacedColNameStr, err := e.keyProcess(colNameStr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treplacedColNameStr = strings.Replace(replacedColNameStr, \"`\", \"\", -1)\n\treturn replacedColNameStr, nil\n}\n\nfunc (e *ESql) keyProcess(target string) (string, error) {\n\tif e.filterKey != nil && e.filterKey(target) && e.processKey != nil {\n\t\ttarget, err := e.processKey(target)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\treturn target, nil\n\t}\n\treturn target, nil\n}\n\nfunc (e *ESql) valueProcess(colName string, value string) (string, error) {\n\tif e.filterValue != nil && e.filterValue(colName) && e.processValue != nil {\n\t\tvalue, err := e.processValue(value)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\treturn value, nil\n\t}\n\treturn value, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/interfaces.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\n\tesaws \"github.com/olivere/elastic/aws/v4\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\tesc \"github.com/uber/cadence/common/elasticsearch/client\"\n\t\"github.com/uber/cadence/common/elasticsearch/client/os2\"\n\tv6 \"github.com/uber/cadence/common/elasticsearch/client/v6\"\n\tv7 \"github.com/uber/cadence/common/elasticsearch/client/v7\"\n\t\"github.com/uber/cadence/common/elasticsearch/query\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\tp \"github.com/uber/cadence/common/persistence\"\n)\n\n// NewGenericClient create a ES client\nfunc NewGenericClient(\n\tconnectConfig *config.ElasticSearchConfig,\n\tlogger log.Logger,\n) (*ESClient, error) {\n\tif connectConfig.Version == \"\" {\n\t\tconnectConfig.Version = \"v6\"\n\t}\n\tvar tlsClient *http.Client\n\tvar signingAWSClient *http.Client\n\n\tif connectConfig.AWSSigning.Enable {\n\t\tcreds, region, err := connectConfig.AWSSigning.GetCredentials()\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"getting AWS credentials: %w\", err)\n\t\t}\n\n\t\tsigningAWSClient = esaws.NewV4SigningClient(creds, *region)\n\t}\n\n\tif connectConfig.TLS.Enabled {\n\t\tvar err error\n\t\ttlsClient, err = buildTLSHTTPClient(connectConfig.TLS)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tvar esClient esc.Client\n\tvar err error\n\n\t// this logger is only used for client-internal purposes, e.g. request errors, which are generally lack detailed context.\n\t// it is currently (unfortunately) shared between both queries and bulk-updates.\n\tclientLogger := logger.WithTags(tag.ComponentESVisibilityClient)\n\n\tswitch connectConfig.Version {\n\tcase \"v6\":\n\t\tesClient, err = v6.NewV6Client(connectConfig, clientLogger, tlsClient, signingAWSClient)\n\tcase \"v7\":\n\t\tesClient, err = v7.NewV7Client(connectConfig, clientLogger, tlsClient, signingAWSClient)\n\tcase \"os2\":\n\t\tesClient, err = os2.NewClient(connectConfig, clientLogger, tlsClient)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"not supported ElasticSearch version: %v\", connectConfig.Version)\n\t}\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ESClient{\n\t\tClient: esClient,\n\t\tLogger: logger,\n\t}, nil\n}\n\ntype (\n\t// GenericClient is a generic interface for all versions of ElasticSearch clients\n\tGenericClient interface {\n\t\t// Search API is only for supporting various List[Open/Closed]WorkflowExecutions(ByXyz).\n\t\t// Use SearchByQuery or ScanByQuery for generic purpose searching.\n\t\tSearch(ctx context.Context, request *SearchRequest) (*SearchResponse, error)\n\t\t// SearchByQuery is the generic purpose searching\n\t\tSearchByQuery(ctx context.Context, request *SearchByQueryRequest) (*SearchResponse, error)\n\t\t// SearchRaw is for searching with raw json. Returns RawResult object which is subset of ESv6 and ESv7 response\n\t\tSearchRaw(ctx context.Context, index, query string) (*RawResponse, error)\n\t\t// ScanByQuery is also generic purpose searching, but implemented with ScrollService of ElasticSearch,\n\t\t// which is more performant for pagination, but comes with some limitation of in-parallel requests.\n\t\tScanByQuery(ctx context.Context, request *ScanByQueryRequest) (*SearchResponse, error)\n\t\t// TODO remove it in https://github.com/uber/cadence/issues/3682\n\t\tSearchForOneClosedExecution(ctx context.Context, index string, request *SearchForOneClosedExecutionRequest) (*SearchForOneClosedExecutionResponse, error)\n\t\t// CountByQuery is for returning the count of workflow executions that match the query\n\t\tCountByQuery(ctx context.Context, index, query string) (int64, error)\n\n\t\t// RunBulkProcessor returns a processor for adding/removing docs into ElasticSearch index\n\t\tRunBulkProcessor(ctx context.Context, p *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error)\n\n\t\t// PutMapping adds new field type to the index\n\t\tPutMapping(ctx context.Context, index, root, key, valueType string) error\n\t\t// CreateIndex creates a new index\n\t\tCreateIndex(ctx context.Context, index string) error\n\n\t\tIsNotFoundError(err error) bool\n\t}\n\n\t// SearchRequest is request for Search\n\tSearchRequest struct {\n\t\tIndex           string\n\t\tListRequest     *p.InternalListWorkflowExecutionsRequest\n\t\tIsOpen          bool\n\t\tFilter          IsRecordValidFilter\n\t\tMatchQuery      *query.MatchQuery\n\t\tMaxResultWindow int\n\t}\n\n\t// SearchByQueryRequest is request for SearchByQuery\n\tSearchByQueryRequest struct {\n\t\tIndex           string\n\t\tQuery           string\n\t\tNextPageToken   []byte\n\t\tPageSize        int\n\t\tFilter          IsRecordValidFilter\n\t\tMaxResultWindow int\n\t}\n\n\t// ScanByQueryRequest is request for SearchByQuery\n\tScanByQueryRequest struct {\n\t\tIndex         string\n\t\tQuery         string\n\t\tNextPageToken []byte\n\t\tPageSize      int\n\t}\n\n\t// SearchResponse is a response to Search, SearchByQuery and ScanByQuery\n\tSearchResponse = p.InternalListWorkflowExecutionsResponse\n\n\t// SearchForOneClosedExecutionRequest is request for SearchForOneClosedExecution\n\tSearchForOneClosedExecutionRequest = p.InternalGetClosedWorkflowExecutionRequest\n\n\t// SearchForOneClosedExecutionResponse is response for SearchForOneClosedExecution\n\tSearchForOneClosedExecutionResponse = p.InternalGetClosedWorkflowExecutionResponse\n\n\t// VisibilityRecord is a struct of doc for deserialization\n\tVisibilityRecord struct {\n\t\tWorkflowID             string\n\t\tRunID                  string\n\t\tWorkflowType           string\n\t\tDomainID               string\n\t\tStartTime              int64\n\t\tExecutionTime          int64\n\t\tCloseTime              int64\n\t\tCloseStatus            workflow.WorkflowExecutionCloseStatus\n\t\tHistoryLength          int64\n\t\tMemo                   []byte\n\t\tEncoding               string\n\t\tTaskList               string\n\t\tIsCron                 bool\n\t\tNumClusters            int16\n\t\tClusterAttributeScope  string\n\t\tClusterAttributeName   string\n\t\tUpdateTime             int64\n\t\tAttr                   map[string]interface{}\n\t\tCronSchedule           string\n\t\tExecutionStatus        int32\n\t\tScheduledExecutionTime int64\n\t}\n\n\tSearchHits struct {\n\t\tTotalHits int64\n\t\tHits      []*p.InternalVisibilityWorkflowExecutionInfo\n\t}\n\n\tRawResponse struct {\n\t\tTookInMillis int64\n\t\tHits         SearchHits\n\t\tAggregations map[string]json.RawMessage\n\t}\n\n\t// IsRecordValidFilter is a function to filter visibility records\n\tIsRecordValidFilter func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool\n)\n"
  },
  {
    "path": "common/elasticsearch/mocks/GenericClient.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\telasticsearch \"github.com/uber/cadence/common/elasticsearch\"\n\tbulk \"github.com/uber/cadence/common/elasticsearch/bulk\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// GenericClient is an autogenerated mock type for the GenericClient type\ntype GenericClient struct {\n\tmock.Mock\n}\n\n// CountByQuery provides a mock function with given fields: ctx, index, query\nfunc (_m *GenericClient) CountByQuery(ctx context.Context, index string, query string) (int64, error) {\n\tret := _m.Called(ctx, index, query)\n\n\tvar r0 int64\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string) int64); ok {\n\t\tr0 = rf(ctx, index, query)\n\t} else {\n\t\tr0 = ret.Get(0).(int64)\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, string, string) error); ok {\n\t\tr1 = rf(ctx, index, query)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateIndex provides a mock function with given fields: ctx, index\nfunc (_m *GenericClient) CreateIndex(ctx context.Context, index string) error {\n\tret := _m.Called(ctx, index)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, string) error); ok {\n\t\tr0 = rf(ctx, index)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// IsNotFoundError provides a mock function with given fields: err\nfunc (_m *GenericClient) IsNotFoundError(err error) bool {\n\tret := _m.Called(err)\n\n\tvar r0 bool\n\tif rf, ok := ret.Get(0).(func(error) bool); ok {\n\t\tr0 = rf(err)\n\t} else {\n\t\tr0 = ret.Get(0).(bool)\n\t}\n\n\treturn r0\n}\n\n// PutMapping provides a mock function with given fields: ctx, index, root, key, valueType\nfunc (_m *GenericClient) PutMapping(ctx context.Context, index string, root string, key string, valueType string) error {\n\tret := _m.Called(ctx, index, root, key, valueType)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string, string, string) error); ok {\n\t\tr0 = rf(ctx, index, root, key, valueType)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// RunBulkProcessor provides a mock function with given fields: ctx, p\nfunc (_m *GenericClient) RunBulkProcessor(ctx context.Context, p *bulk.BulkProcessorParameters) (bulk.GenericBulkProcessor, error) {\n\tret := _m.Called(ctx, p)\n\n\tvar r0 bulk.GenericBulkProcessor\n\tif rf, ok := ret.Get(0).(func(context.Context, *bulk.BulkProcessorParameters) bulk.GenericBulkProcessor); ok {\n\t\tr0 = rf(ctx, p)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(bulk.GenericBulkProcessor)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *bulk.BulkProcessorParameters) error); ok {\n\t\tr1 = rf(ctx, p)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ScanByQuery provides a mock function with given fields: ctx, request\nfunc (_m *GenericClient) ScanByQuery(ctx context.Context, request *elasticsearch.ScanByQueryRequest) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.InternalListWorkflowExecutionsResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *elasticsearch.ScanByQueryRequest) *persistence.InternalListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.InternalListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *elasticsearch.ScanByQueryRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Search provides a mock function with given fields: ctx, request\nfunc (_m *GenericClient) Search(ctx context.Context, request *elasticsearch.SearchRequest) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.InternalListWorkflowExecutionsResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *elasticsearch.SearchRequest) *persistence.InternalListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.InternalListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *elasticsearch.SearchRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\nfunc (_m *GenericClient) SearchRaw(ctx context.Context, index string, query string) (*elasticsearch.RawResponse, error) {\n\tret := _m.Called(ctx, index, query)\n\n\tvar r0 *elasticsearch.RawResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string) *elasticsearch.RawResponse); ok {\n\t\tr0 = rf(ctx, index, query)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*elasticsearch.RawResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, string, string) error); ok {\n\t\tr1 = rf(ctx, index, query)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// SearchByQuery provides a mock function with given fields: ctx, request\nfunc (_m *GenericClient) SearchByQuery(ctx context.Context, request *elasticsearch.SearchByQueryRequest) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.InternalListWorkflowExecutionsResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *elasticsearch.SearchByQueryRequest) *persistence.InternalListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.InternalListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *elasticsearch.SearchByQueryRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// SearchForOneClosedExecution provides a mock function with given fields: ctx, index, request\nfunc (_m *GenericClient) SearchForOneClosedExecution(ctx context.Context, index string, request *persistence.InternalGetClosedWorkflowExecutionRequest) (*persistence.InternalGetClosedWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, index, request)\n\n\tvar r0 *persistence.InternalGetClosedWorkflowExecutionResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, string, *persistence.InternalGetClosedWorkflowExecutionRequest) *persistence.InternalGetClosedWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, index, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.InternalGetClosedWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, string, *persistence.InternalGetClosedWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, index, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n"
  },
  {
    "path": "common/elasticsearch/page_token.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ElasticVisibilityPageToken holds the paging token for ElasticSearch\n\tElasticVisibilityPageToken struct {\n\t\t// for ES API From+Size\n\t\tFrom int\n\t\t// for ES API searchAfter\n\t\tSortValue  interface{}\n\t\tTieBreaker string // runID\n\t\t// for ES scroll API\n\t\tScrollID string\n\t}\n)\n\n// DeserializePageToken return the structural token\nfunc DeserializePageToken(data []byte) (*ElasticVisibilityPageToken, error) {\n\tvar token ElasticVisibilityPageToken\n\tdec := json.NewDecoder(bytes.NewReader(data))\n\tdec.UseNumber()\n\terr := dec.Decode(&token)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"unable to deserialize page token. err: %v\", err),\n\t\t}\n\t}\n\treturn &token, nil\n}\n\n// SerializePageToken return the token blob\nfunc SerializePageToken(token *ElasticVisibilityPageToken) ([]byte, error) {\n\tdata, err := json.Marshal(token)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"unable to serialize page token. err: %v\", err),\n\t\t}\n\t}\n\treturn data, nil\n}\n\n// GetNextPageToken returns the structural token with nil handling\nfunc GetNextPageToken(token []byte) (*ElasticVisibilityPageToken, error) {\n\tvar result *ElasticVisibilityPageToken\n\tvar err error\n\tif len(token) > 0 {\n\t\tresult, err = DeserializePageToken(token)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tresult = &ElasticVisibilityPageToken{}\n\t}\n\treturn result, nil\n}\n\n// ShouldSearchAfter decides if should search after\nfunc ShouldSearchAfter(token *ElasticVisibilityPageToken) bool {\n\treturn token.TieBreaker != \"\"\n}\n"
  },
  {
    "path": "common/elasticsearch/query/bool_query.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\ntype Query interface {\n\t// Source returns the JSON-serializable query request.\n\tSource() (interface{}, error)\n}\n\ntype BoolQuery struct {\n\tmustClauses    []Query\n\tmustNotClauses []Query\n\tfilterClauses  []Query\n}\n\n// Creates a new bool query.\nfunc NewBoolQuery() *BoolQuery {\n\treturn &BoolQuery{\n\t\tmustClauses:    make([]Query, 0),\n\t\tmustNotClauses: make([]Query, 0),\n\t\tfilterClauses:  make([]Query, 0),\n\t}\n}\n\nfunc (q *BoolQuery) Must(queries ...Query) *BoolQuery {\n\tq.mustClauses = append(q.mustClauses, queries...)\n\treturn q\n}\n\nfunc (q *BoolQuery) MustNot(queries ...Query) *BoolQuery {\n\tq.mustNotClauses = append(q.mustNotClauses, queries...)\n\treturn q\n}\n\nfunc (q *BoolQuery) Filter(filters ...Query) *BoolQuery {\n\tq.filterClauses = append(q.filterClauses, filters...)\n\treturn q\n}\n\nfunc (q *BoolQuery) Source() (interface{}, error) {\n\tquery := make(map[string]interface{})\n\n\tboolClause := make(map[string]interface{})\n\tquery[\"bool\"] = boolClause\n\n\t// must\n\tif len(q.mustClauses) == 1 {\n\t\tsrc, err := q.mustClauses[0].Source()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tboolClause[\"must\"] = src\n\t} else if len(q.mustClauses) > 1 {\n\t\tvar clauses []interface{}\n\t\tfor _, subQuery := range q.mustClauses {\n\t\t\tsrc, err := subQuery.Source()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tclauses = append(clauses, src)\n\t\t}\n\t\tboolClause[\"must\"] = clauses\n\t}\n\n\t// must_not\n\tif len(q.mustNotClauses) == 1 {\n\t\tsrc, err := q.mustNotClauses[0].Source()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tboolClause[\"must_not\"] = src\n\t} else if len(q.mustNotClauses) > 1 {\n\t\tvar clauses []interface{}\n\t\tfor _, subQuery := range q.mustNotClauses {\n\t\t\tsrc, err := subQuery.Source()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tclauses = append(clauses, src)\n\t\t}\n\t\tboolClause[\"must_not\"] = clauses\n\t}\n\n\t// filter\n\tif len(q.filterClauses) == 1 {\n\t\tsrc, err := q.filterClauses[0].Source()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tboolClause[\"filter\"] = src\n\t} else if len(q.filterClauses) > 1 {\n\t\tvar clauses []interface{}\n\t\tfor _, subQuery := range q.filterClauses {\n\t\t\tsrc, err := subQuery.Source()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tclauses = append(clauses, src)\n\t\t}\n\t\tboolClause[\"filter\"] = clauses\n\t}\n\n\treturn query, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/bool_query_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n)\n\nfunc TestBoolQuery(t *testing.T) {\n\tq := NewBoolQuery()\n\tq = q.MustNot(NewRangeQuery(\"age\").From(10).To(20))\n\tsrc, err := q.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"bool\":{\"must_not\":{\"range\":{\"age\":{\"from\":10,\"include_lower\":true,\"include_upper\":true,\"to\":20}}}}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/query/builder.go",
    "content": "package query\n\nimport \"encoding/json\"\n\n// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\ntype Builder struct {\n\tquery                 Query         // query\n\tfrom                  int           // from\n\tsize                  int           // size\n\tsorters               []Sorter      // sort\n\tsearchAfterSortValues []interface{} // search_after\n\n}\n\nfunc NewBuilder() *Builder {\n\treturn &Builder{\n\t\tfrom: -1,\n\t\tsize: -1,\n\t}\n}\n\nfunc (b *Builder) Query(query Query) *Builder {\n\tb.query = query\n\treturn b\n}\n\nfunc (b *Builder) From(from int) *Builder {\n\tb.from = from\n\treturn b\n}\n\nfunc (b *Builder) Sortby(sorters ...Sorter) *Builder {\n\tb.sorters = sorters\n\treturn b\n}\n\nfunc (b *Builder) Size(size int) *Builder {\n\tb.size = size\n\treturn b\n}\n\nfunc (b *Builder) SearchAfter(v ...interface{}) *Builder {\n\tb.searchAfterSortValues = v\n\treturn b\n}\n\n// Source returns the serializable JSON for the source builder.\nfunc (b *Builder) Source() (interface{}, error) {\n\tsource := make(map[string]interface{})\n\n\tif b.from != -1 {\n\t\tsource[\"from\"] = b.from\n\t}\n\tif b.size != -1 {\n\t\tsource[\"size\"] = b.size\n\t}\n\n\tif b.query != nil {\n\t\tsrc, err := b.query.Source()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tsource[\"query\"] = src\n\t}\n\tif len(b.sorters) > 0 {\n\t\tvar sortarr []interface{}\n\t\tfor _, sorter := range b.sorters {\n\t\t\tsrc, err := sorter.Source()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tsortarr = append(sortarr, src)\n\t\t}\n\t\tsource[\"sort\"] = sortarr\n\t}\n\n\tif len(b.searchAfterSortValues) > 0 {\n\t\tsource[\"search_after\"] = b.searchAfterSortValues\n\t}\n\n\treturn source, nil\n}\n\nfunc (b *Builder) String() (string, error) {\n\tsource, err := b.Source()\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tmarshaled, err := json.Marshal(source)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\treturn string(marshaled), nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/builder_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/olivere/elastic/v7\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestQueryBuilder(t *testing.T) {\n\tqb := NewBuilder()\n\tqb.Query(NewExistsQuery(\"user\"))\n\tqb.Size(10)\n\tqb.From(100)\n\tqb.Sortby(NewFieldSort(\"StartDate\"))\n\tsrc, err := qb.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"from\":100,\"query\":{\"exists\":{\"field\":\"user\"}},\"size\":10,\"sort\":[{\"StartDate\":{\"order\":\"asc\"}}]}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n\nfunc TestBuilderAgainsESv7(t *testing.T) {\n\tqb := NewBuilder()\n\tqb.Query(NewExistsQuery(\"user\"))\n\tqb.Size(10)\n\tqb.Sortby(NewFieldSort(\"runid\").Desc())\n\tqb.Query(NewBoolQuery().Must(NewMatchQuery(\"domainID\", \"uuid\"))).SearchAfter(\"sortval\", \"tiebraker\")\n\tqbs, err := qb.Source()\n\tassert.NoError(t, err)\n\n\tsearchSource := elastic.NewSearchSource().\n\t\tQuery(elastic.NewExistsQuery(\"user\")).\n\t\tSize(10).\n\t\tSortBy(elastic.NewFieldSort(\"runid\").Desc()).\n\t\tQuery(elastic.NewBoolQuery().Must(elastic.NewMatchQuery(\"domainID\", \"uuid\"))).SearchAfter(\"sortval\", \"tiebraker\")\n\n\tsss, err := searchSource.Source()\n\tassert.NoError(t, err)\n\n\tassert.Equal(t, sss, qbs, \"ESv7 and local QueryBuilder should produce the same query\")\n}\n"
  },
  {
    "path": "common/elasticsearch/query/exists_query.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\n// ExistsQuery is a query that only matches on documents that the field\n// has a value in them.\n//\n// For more details, see:\n// https://www.elastic.co/guide/en/elasticsearch/reference/6.8/query-dsl-exists-query.html\ntype ExistsQuery struct {\n\tname      string\n\tqueryName string\n}\n\n// NewExistsQuery creates and initializes a new exists query.\nfunc NewExistsQuery(name string) *ExistsQuery {\n\treturn &ExistsQuery{\n\t\tname: name,\n\t}\n}\n\n// QueryName sets the query name for the filter that can be used\n// when searching for matched queries per hit.\nfunc (q *ExistsQuery) QueryName(queryName string) *ExistsQuery {\n\tq.queryName = queryName\n\treturn q\n}\n\n// Source returns the JSON serializable content for this query.\nfunc (q *ExistsQuery) Source() (interface{}, error) {\n\t// {\n\t//   \"exists\" : {\n\t//     \"field\" : \"user\"\n\t//   }\n\t// }\n\n\tquery := make(map[string]interface{})\n\tparams := make(map[string]interface{})\n\tquery[\"exists\"] = params\n\n\tparams[\"field\"] = q.name\n\tif q.queryName != \"\" {\n\t\tparams[\"_name\"] = q.queryName\n\t}\n\n\treturn query, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/exists_query_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n)\n\nfunc TestExistsQuery(t *testing.T) {\n\tq := NewExistsQuery(\"user\")\n\tsrc, err := q.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"exists\":{\"field\":\"user\"}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/query/match_query.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\n// MatchQuery is a family of queries that accepts text/numerics/dates,\n// analyzes them, and constructs a query.\n//\n// To create a new MatchQuery, use NewMatchQuery. To create specific types\n// of queries, e.g. a match_phrase query, use NewMatchPhrQuery(...).Type(\"phrase\"),\n// or use one of the shortcuts e.g. NewMatchPhraseQuery(...).\n//\n// For more details, see\n// https://www.elastic.co/guide/en/elasticsearch/reference/7.0/query-dsl-match-query.html\ntype MatchQuery struct {\n\tname                string\n\ttext                interface{}\n\toperator            string // or / and\n\tanalyzer            string\n\tboost               *float64\n\tfuzziness           string\n\tprefixLength        *int\n\tmaxExpansions       *int\n\tminimumShouldMatch  string\n\tfuzzyRewrite        string\n\tlenient             *bool\n\tfuzzyTranspositions *bool\n\tzeroTermsQuery      string\n\tcutoffFrequency     *float64\n\tqueryName           string\n}\n\n// NewMatchQuery creates and initializes a new MatchQuery.\nfunc NewMatchQuery(name string, text interface{}) *MatchQuery {\n\treturn &MatchQuery{name: name, text: text}\n}\n\n// Operator sets the operator to use when using a boolean query.\n// Can be \"AND\" or \"OR\" (default).\nfunc (q *MatchQuery) Operator(operator string) *MatchQuery {\n\tq.operator = operator\n\treturn q\n}\n\n// Analyzer explicitly sets the analyzer to use. It defaults to use explicit\n// mapping config for the field, or, if not set, the default search analyzer.\nfunc (q *MatchQuery) Analyzer(analyzer string) *MatchQuery {\n\tq.analyzer = analyzer\n\treturn q\n}\n\n// Fuzziness sets the fuzziness when evaluated to a fuzzy query type.\n// Defaults to \"AUTO\".\nfunc (q *MatchQuery) Fuzziness(fuzziness string) *MatchQuery {\n\tq.fuzziness = fuzziness\n\treturn q\n}\n\n// PrefixLength sets the length of a length of common (non-fuzzy)\n// prefix for fuzzy match queries. It must be non-negative.\nfunc (q *MatchQuery) PrefixLength(prefixLength int) *MatchQuery {\n\tq.prefixLength = &prefixLength\n\treturn q\n}\n\n// MaxExpansions is used with fuzzy or prefix type queries. It specifies\n// the number of term expansions to use. It defaults to unbounded so that\n// its recommended to set it to a reasonable value for faster execution.\nfunc (q *MatchQuery) MaxExpansions(maxExpansions int) *MatchQuery {\n\tq.maxExpansions = &maxExpansions\n\treturn q\n}\n\n// CutoffFrequency can be a value in [0..1] (or an absolute number >=1).\n// It represents the maximum treshold of a terms document frequency to be\n// considered a low frequency term.\nfunc (q *MatchQuery) CutoffFrequency(cutoff float64) *MatchQuery {\n\tq.cutoffFrequency = &cutoff\n\treturn q\n}\n\n// MinimumShouldMatch sets the optional minimumShouldMatch value to\n// apply to the query.\nfunc (q *MatchQuery) MinimumShouldMatch(minimumShouldMatch string) *MatchQuery {\n\tq.minimumShouldMatch = minimumShouldMatch\n\treturn q\n}\n\n// FuzzyRewrite sets the fuzzy_rewrite parameter controlling how the\n// fuzzy query will get rewritten.\nfunc (q *MatchQuery) FuzzyRewrite(fuzzyRewrite string) *MatchQuery {\n\tq.fuzzyRewrite = fuzzyRewrite\n\treturn q\n}\n\n// FuzzyTranspositions sets whether transpositions are supported in\n// fuzzy queries.\n//\n// The default metric used by fuzzy queries to determine a match is\n// the Damerau-Levenshtein distance formula which supports transpositions.\n// Setting transposition to false will\n// * switch to classic Levenshtein distance.\n// * If not set, Damerau-Levenshtein distance metric will be used.\nfunc (q *MatchQuery) FuzzyTranspositions(fuzzyTranspositions bool) *MatchQuery {\n\tq.fuzzyTranspositions = &fuzzyTranspositions\n\treturn q\n}\n\n// Lenient specifies whether format based failures will be ignored.\nfunc (q *MatchQuery) Lenient(lenient bool) *MatchQuery {\n\tq.lenient = &lenient\n\treturn q\n}\n\n// ZeroTermsQuery can be \"all\" or \"none\".\nfunc (q *MatchQuery) ZeroTermsQuery(zeroTermsQuery string) *MatchQuery {\n\tq.zeroTermsQuery = zeroTermsQuery\n\treturn q\n}\n\n// Boost sets the boost to apply to this query.\nfunc (q *MatchQuery) Boost(boost float64) *MatchQuery {\n\tq.boost = &boost\n\treturn q\n}\n\n// QueryName sets the query name for the filter that can be used when\n// searching for matched filters per hit.\nfunc (q *MatchQuery) QueryName(queryName string) *MatchQuery {\n\tq.queryName = queryName\n\treturn q\n}\n\n// Source returns JSON for the function score query.\nfunc (q *MatchQuery) Source() (interface{}, error) {\n\t// {\"match\":{\"name\":{\"query\":\"value\",\"type\":\"boolean/phrase\"}}}\n\tsource := make(map[string]interface{})\n\n\tmatch := make(map[string]interface{})\n\tsource[\"match\"] = match\n\n\tquery := make(map[string]interface{})\n\tmatch[q.name] = query\n\n\tquery[\"query\"] = q.text\n\n\tif q.operator != \"\" {\n\t\tquery[\"operator\"] = q.operator\n\t}\n\tif q.analyzer != \"\" {\n\t\tquery[\"analyzer\"] = q.analyzer\n\t}\n\tif q.fuzziness != \"\" {\n\t\tquery[\"fuzziness\"] = q.fuzziness\n\t}\n\tif q.prefixLength != nil {\n\t\tquery[\"prefix_length\"] = *q.prefixLength\n\t}\n\tif q.maxExpansions != nil {\n\t\tquery[\"max_expansions\"] = *q.maxExpansions\n\t}\n\tif q.minimumShouldMatch != \"\" {\n\t\tquery[\"minimum_should_match\"] = q.minimumShouldMatch\n\t}\n\tif q.fuzzyRewrite != \"\" {\n\t\tquery[\"fuzzy_rewrite\"] = q.fuzzyRewrite\n\t}\n\tif q.lenient != nil {\n\t\tquery[\"lenient\"] = *q.lenient\n\t}\n\tif q.fuzzyTranspositions != nil {\n\t\tquery[\"fuzzy_transpositions\"] = *q.fuzzyTranspositions\n\t}\n\tif q.zeroTermsQuery != \"\" {\n\t\tquery[\"zero_terms_query\"] = q.zeroTermsQuery\n\t}\n\tif q.cutoffFrequency != nil {\n\t\tquery[\"cutoff_frequency\"] = *q.cutoffFrequency\n\t}\n\tif q.boost != nil {\n\t\tquery[\"boost\"] = *q.boost\n\t}\n\tif q.queryName != \"\" {\n\t\tquery[\"_name\"] = q.queryName\n\t}\n\n\treturn source, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/match_query_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n)\n\nfunc TestMatchQuery(t *testing.T) {\n\tq := NewMatchQuery(\"message\", \"this is a test\")\n\tsrc, err := q.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"match\":{\"message\":{\"query\":\"this is a test\"}}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n\nfunc TestMatchQueryWithOptions(t *testing.T) {\n\tq := NewMatchQuery(\"message\", \"this is a test\").Analyzer(\"whitespace\").Operator(\"or\").Boost(2.5)\n\tsrc, err := q.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"match\":{\"message\":{\"analyzer\":\"whitespace\",\"boost\":2.5,\"operator\":\"or\",\"query\":\"this is a test\"}}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/query/range_query.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\n// RangeQuery matches documents with fields that have terms within a certain range.\n//\n// For details, see\n// https://www.elastic.co/guide/en/elasticsearch/reference/7.0/query-dsl-range-query.html\ntype RangeQuery struct {\n\tname         string\n\tfrom         interface{}\n\tto           interface{}\n\tincludeLower bool\n\tincludeUpper bool\n}\n\n// NewRangeQuery creates and initializes a new RangeQuery.\nfunc NewRangeQuery(name string) *RangeQuery {\n\treturn &RangeQuery{name: name, includeLower: true, includeUpper: true}\n}\n\n// From indicates the from part of the RangeQuery.\n// Use nil to indicate an unbounded from part.\nfunc (q *RangeQuery) From(from interface{}) *RangeQuery {\n\tq.from = from\n\treturn q\n}\n\n// Gt indicates a greater-than value for the from part.\n// Use nil to indicate an unbounded from part.\nfunc (q *RangeQuery) Gt(from interface{}) *RangeQuery {\n\tq.from = from\n\tq.includeLower = false\n\treturn q\n}\n\n// Gte indicates a greater-than-or-equal value for the from part.\n// Use nil to indicate an unbounded from part.\nfunc (q *RangeQuery) Gte(from interface{}) *RangeQuery {\n\tq.from = from\n\tq.includeLower = true\n\treturn q\n}\n\n// To indicates the to part of the RangeQuery.\n// Use nil to indicate an unbounded to part.\nfunc (q *RangeQuery) To(to interface{}) *RangeQuery {\n\tq.to = to\n\treturn q\n}\n\n// Lt indicates a less-than value for the to part.\n// Use nil to indicate an unbounded to part.\nfunc (q *RangeQuery) Lt(to interface{}) *RangeQuery {\n\tq.to = to\n\tq.includeUpper = false\n\treturn q\n}\n\n// Lte indicates a less-than-or-equal value for the to part.\n// Use nil to indicate an unbounded to part.\nfunc (q *RangeQuery) Lte(to interface{}) *RangeQuery {\n\tq.to = to\n\tq.includeUpper = true\n\treturn q\n}\n\n// IncludeLower indicates whether the lower bound should be included or not.\n// Defaults to true.\nfunc (q *RangeQuery) IncludeLower(includeLower bool) *RangeQuery {\n\tq.includeLower = includeLower\n\treturn q\n}\n\n// IncludeUpper indicates whether the upper bound should be included or not.\n// Defaults to true.\nfunc (q *RangeQuery) IncludeUpper(includeUpper bool) *RangeQuery {\n\tq.includeUpper = includeUpper\n\treturn q\n}\n\n// Source returns JSON for the query.\nfunc (q *RangeQuery) Source() (interface{}, error) {\n\tsource := make(map[string]interface{})\n\n\trangeQ := make(map[string]interface{})\n\tsource[\"range\"] = rangeQ\n\n\tparams := make(map[string]interface{})\n\trangeQ[q.name] = params\n\n\tparams[\"from\"] = q.from\n\tparams[\"to\"] = q.to\n\tparams[\"include_lower\"] = q.includeLower\n\tparams[\"include_upper\"] = q.includeUpper\n\n\treturn source, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/range_query_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n)\n\nfunc TestRangeQuery(t *testing.T) {\n\tq := NewRangeQuery(\"postDate\").From(\"2010-03-01\").To(\"2010-04-01\")\n\tsrc, err := q.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"range\":{\"postDate\":{\"from\":\"2010-03-01\",\"include_lower\":true,\"include_upper\":true,\"to\":\"2010-04-01\"}}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/query/sort.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\ntype Sorter interface {\n\tSource() (interface{}, error)\n}\n\n// FieldSort sorts by a given field.\ntype FieldSort struct {\n\tSorter\n\tfieldName string\n\tascending bool\n}\n\n// NewFieldSort creates a new FieldSort.\nfunc NewFieldSort(fieldName string) *FieldSort {\n\treturn &FieldSort{\n\t\tfieldName: fieldName,\n\t\tascending: true,\n\t}\n}\n\n// FieldName specifies the name of the field to be used for sorting.\nfunc (s *FieldSort) FieldName(fieldName string) *FieldSort {\n\ts.fieldName = fieldName\n\treturn s\n}\n\n// Order defines whether sorting ascending (default) or descending.\nfunc (s *FieldSort) Order(ascending bool) *FieldSort {\n\ts.ascending = ascending\n\treturn s\n}\n\n// Asc sets ascending sort order.\nfunc (s *FieldSort) Asc() *FieldSort {\n\ts.ascending = true\n\treturn s\n}\n\n// Desc sets descending sort order.\nfunc (s *FieldSort) Desc() *FieldSort {\n\ts.ascending = false\n\treturn s\n}\n\n// Source returns the JSON-serializable data.\nfunc (s *FieldSort) Source() (interface{}, error) {\n\tsource := make(map[string]interface{})\n\tx := make(map[string]interface{})\n\tsource[s.fieldName] = x\n\tif s.ascending {\n\t\tx[\"order\"] = \"asc\"\n\t} else {\n\t\tx[\"order\"] = \"desc\"\n\t}\n\n\treturn source, nil\n}\n"
  },
  {
    "path": "common/elasticsearch/query/sort_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n)\n\nfunc TestFieldSort(t *testing.T) {\n\tbuilder := NewFieldSort(\"grade\")\n\tsrc, err := builder.Source()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tdata, err := json.Marshal(src)\n\tif err != nil {\n\t\tt.Fatalf(\"marshaling to JSON failed: %v\", err)\n\t}\n\tgot := string(data)\n\texpected := `{\"grade\":{\"order\":\"asc\"}}`\n\tif got != expected {\n\t\tt.Errorf(\"expected\\n%s\\n,got:\\n%s\", expected, got)\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/validator/queryValidator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to qvom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, qvETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage validator\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// VisibilityQueryValidator for sql query validation\ntype VisibilityQueryValidator struct {\n\tvalidSearchAttributes          dynamicproperties.MapPropertyFn\n\tenableQueryAttributeValidation dynamicproperties.BoolPropertyFn\n}\n\n// NewQueryValidator create VisibilityQueryValidator\nfunc NewQueryValidator(\n\tvalidSearchAttributes dynamicproperties.MapPropertyFn,\n\tenableQueryAttributeValidation dynamicproperties.BoolPropertyFn) *VisibilityQueryValidator {\n\treturn &VisibilityQueryValidator{\n\t\tvalidSearchAttributes:          validSearchAttributes,\n\t\tenableQueryAttributeValidation: enableQueryAttributeValidation,\n\t}\n}\n\n// ValidateQuery validates that search attributes in the query are legal.\n// Adds attr prefix for customized fields and returns modified query.\nfunc (qv *VisibilityQueryValidator) ValidateQuery(whereClause string) (string, error) {\n\tif len(whereClause) != 0 {\n\t\t// Build a placeholder query that allows us to easily parse the contents of the where clause.\n\t\t// IMPORTANT: This query is never executed, it is just used to parse and validate whereClause\n\t\tvar placeholderQuery string\n\t\twhereClause := strings.TrimSpace(whereClause)\n\t\t// #nosec\n\t\tif common.IsJustOrderByClause(whereClause) { // just order by\n\t\t\tplaceholderQuery = fmt.Sprintf(\"SELECT * FROM dummy %s\", whereClause)\n\t\t} else {\n\t\t\tplaceholderQuery = fmt.Sprintf(\"SELECT * FROM dummy WHERE %s\", whereClause)\n\t\t}\n\n\t\tstmt, err := sqlparser.Parse(placeholderQuery)\n\t\tif err != nil {\n\t\t\treturn \"\", &types.BadRequestError{Message: \"Invalid query.\"}\n\t\t}\n\n\t\tsel, ok := stmt.(*sqlparser.Select)\n\t\tif !ok {\n\t\t\treturn \"\", &types.BadRequestError{Message: \"Invalid select query.\"}\n\t\t}\n\t\tbuf := sqlparser.NewTrackedBuffer(nil)\n\t\t// validate where expr\n\t\tif sel.Where != nil {\n\t\t\terr = qv.validateWhereExpr(sel.Where.Expr)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", &types.BadRequestError{Message: err.Error()}\n\t\t\t}\n\t\t\tsel.Where.Expr.Format(buf)\n\t\t}\n\t\t// validate order by\n\t\terr = qv.validateOrderByExpr(sel.OrderBy)\n\t\tif err != nil {\n\t\t\treturn \"\", &types.BadRequestError{Message: err.Error()}\n\t\t}\n\t\tsel.OrderBy.Format(buf)\n\n\t\treturn buf.String(), nil\n\t}\n\treturn whereClause, nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateWhereExpr(expr sqlparser.Expr) error {\n\tif expr == nil {\n\t\treturn nil\n\t}\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.AndExpr, *sqlparser.OrExpr:\n\t\treturn qv.validateAndOrExpr(expr)\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn qv.validateComparisonExpr(expr)\n\tcase *sqlparser.RangeCond:\n\t\treturn qv.validateRangeExpr(expr)\n\tcase *sqlparser.ParenExpr:\n\t\treturn qv.validateWhereExpr(expr.Expr)\n\tdefault:\n\t\treturn errors.New(\"invalid where clause\")\n\t}\n\n}\n\nfunc (qv *VisibilityQueryValidator) validateAndOrExpr(expr sqlparser.Expr) error {\n\tvar leftExpr sqlparser.Expr\n\tvar rightExpr sqlparser.Expr\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.AndExpr:\n\t\tleftExpr = expr.Left\n\t\trightExpr = expr.Right\n\tcase *sqlparser.OrExpr:\n\t\tleftExpr = expr.Left\n\t\trightExpr = expr.Right\n\t}\n\n\tif err := qv.validateWhereExpr(leftExpr); err != nil {\n\t\treturn err\n\t}\n\treturn qv.validateWhereExpr(rightExpr)\n}\n\nfunc (qv *VisibilityQueryValidator) validateComparisonExpr(expr sqlparser.Expr) error {\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\tcolName, ok := comparisonExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn errors.New(\"invalid comparison expression\")\n\t}\n\tcolNameStr := colName.Name.String()\n\tif !qv.isValidSearchAttributes(colNameStr) {\n\t\treturn fmt.Errorf(\"invalid search attribute %q\", colNameStr)\n\t}\n\n\tif !definition.IsSystemIndexedKey(colNameStr) { // add search attribute prefix\n\t\tcomparisonExpr.Left = &sqlparser.ColName{\n\t\t\tMetadata:  colName.Metadata,\n\t\t\tName:      sqlparser.NewColIdent(definition.Attr + \".\" + colNameStr),\n\t\t\tQualifier: colName.Qualifier,\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateRangeExpr(expr sqlparser.Expr) error {\n\trangeCond := expr.(*sqlparser.RangeCond)\n\tcolName, ok := rangeCond.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn errors.New(\"invalid range expression\")\n\t}\n\tcolNameStr := colName.Name.String()\n\n\tif !qv.isValidSearchAttributes(colNameStr) {\n\t\treturn fmt.Errorf(\"invalid search attribute %q\", colNameStr)\n\t}\n\n\tif !definition.IsSystemIndexedKey(colNameStr) { // add search attribute prefix\n\t\trangeCond.Left = &sqlparser.ColName{\n\t\t\tMetadata:  colName.Metadata,\n\t\t\tName:      sqlparser.NewColIdent(definition.Attr + \".\" + colNameStr),\n\t\t\tQualifier: colName.Qualifier,\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateOrderByExpr(orderBy sqlparser.OrderBy) error {\n\tfor _, orderByExpr := range orderBy {\n\t\tcolName, ok := orderByExpr.Expr.(*sqlparser.ColName)\n\t\tif !ok {\n\t\t\treturn errors.New(\"invalid order by expression\")\n\t\t}\n\t\tcolNameStr := colName.Name.String()\n\t\tif qv.isValidSearchAttributes(colNameStr) {\n\t\t\tif !definition.IsSystemIndexedKey(colNameStr) { // add search attribute prefix\n\t\t\t\torderByExpr.Expr = &sqlparser.ColName{\n\t\t\t\t\tMetadata:  colName.Metadata,\n\t\t\t\t\tName:      sqlparser.NewColIdent(definition.Attr + \".\" + colNameStr),\n\t\t\t\t\tQualifier: colName.Qualifier,\n\t\t\t\t}\n\t\t\t}\n\t\t} else {\n\t\t\treturn errors.New(\"invalid order by attribute\")\n\t\t}\n\t}\n\treturn nil\n}\n\n// isValidSearchAttributes return true if key is registered\nfunc (qv *VisibilityQueryValidator) isValidSearchAttributes(key string) bool {\n\tif qv.enableQueryAttributeValidation() {\n\t\tvalidAttr := qv.validSearchAttributes()\n\t\t_, isValidKey := validAttr[key]\n\t\treturn isValidKey\n\t}\n\treturn true\n}\n"
  },
  {
    "path": "common/elasticsearch/validator/queryValidator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to qvom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, qvETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage validator\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestValidateQuery(t *testing.T) {\n\ttests := []struct {\n\t\tmsg       string\n\t\tquery     string\n\t\tvalidated string\n\t\terr       string\n\t\tdcValid   map[string]interface{}\n\t}{\n\t\t{\n\t\t\tmsg:       \"empty query\",\n\t\t\tquery:     \"\",\n\t\t\tvalidated: \"\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"simple query\",\n\t\t\tquery:     \"WorkflowID = 'wid'\",\n\t\t\tvalidated: \"WorkflowID = 'wid'\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"custom field\",\n\t\t\tquery:     \"CustomStringField = 'custom'\",\n\t\t\tvalidated: \"`Attr.CustomStringField` = 'custom'\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"complex query\",\n\t\t\tquery:     \"WorkflowID = 'wid' and ((CustomStringField = 'custom') or CustomIntField between 1 and 10)\",\n\t\t\tvalidated: \"WorkflowID = 'wid' and ((`Attr.CustomStringField` = 'custom') or `Attr.CustomIntField` between 1 and 10)\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid SQL\",\n\t\t\tquery: \"Invalid SQL\",\n\t\t\terr:   \"Invalid query.\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid where expression\",\n\t\t\tquery: \"InvalidWhereExpr\",\n\t\t\terr:   \"invalid where clause\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid comparison\",\n\t\t\tquery: \"WorkflowID = 'wid' and 1 < 2\",\n\t\t\terr:   \"invalid comparison expression\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid range\",\n\t\t\tquery: \"1 between 1 and 2 or WorkflowID = 'wid'\",\n\t\t\terr:   \"invalid range expression\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid search attribute in comparison\",\n\t\t\tquery: \"Invalid = 'a' and 1 < 2\",\n\t\t\terr:   \"invalid search attribute \\\"Invalid\\\"\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid search attribute in range\",\n\t\t\tquery: \"Invalid between 1 and 2 or WorkflowID = 'wid'\",\n\t\t\terr:   \"invalid search attribute \\\"Invalid\\\"\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"only order by\",\n\t\t\tquery:     \"order by CloseTime desc\",\n\t\t\tvalidated: \" order by CloseTime desc\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"only order by search attribute\",\n\t\t\tquery:     \"order by CustomIntField desc\",\n\t\t\tvalidated: \" order by `Attr.CustomIntField` desc\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"condition + order by\",\n\t\t\tquery:     \"WorkflowID = 'wid' order by CloseTime desc\",\n\t\t\tvalidated: \"WorkflowID = 'wid' order by CloseTime desc\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid order by attribute\",\n\t\t\tquery: \"order by InvalidField desc\",\n\t\t\terr:   \"invalid order by attribute\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"invalid order by attribute expr\",\n\t\t\tquery: \"order by 123\",\n\t\t\terr:   \"invalid order by expression\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"security SQL injection - with another statement\",\n\t\t\tquery: \"WorkflowID = 'wid'; SELECT * FROM important_table;\",\n\t\t\terr:   \"Invalid query.\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"security SQL injection - with always true expression\",\n\t\t\tquery: \"WorkflowID = 'wid' and (RunID = 'rid' or 1 = 1)\",\n\t\t\terr:   \"invalid comparison expression\",\n\t\t},\n\t\t{\n\t\t\tmsg:   \"security SQL injection - with union\",\n\t\t\tquery: \"WorkflowID = 'wid' union select * from dummy\",\n\t\t\terr:   \"Invalid select query.\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"valid custom search attribute\",\n\t\t\tquery:     \"CustomStringField = 'value'\",\n\t\t\tvalidated: \"`Attr.CustomStringField` = 'value'\",\n\t\t},\n\t\t{\n\t\t\tmsg:       \"custom search attribute can contain underscore\",\n\t\t\tquery:     \"Custom_Field = 'value'\",\n\t\t\tvalidated: \"`Attr.Custom_Field` = 'value'\",\n\t\t\tdcValid: map[string]interface{}{\n\t\t\t\t\"Custom_Field\": types.IndexedValueTypeString,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:       \"custom search attribute can contain number\",\n\t\t\tquery:     \"Custom_0 = 'value'\",\n\t\t\tvalidated: \"`Attr.Custom_0` = 'value'\",\n\t\t\tdcValid: map[string]interface{}{\n\t\t\t\t\"Custom_0\": types.IndexedValueTypeString,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:   \"customg search attribute cannot contain dot\",\n\t\t\tquery: \"Custom.Field = 'value'\",\n\t\t\terr:   \"invalid search attribute \\\"Field\\\"\",\n\t\t\tdcValid: map[string]interface{}{\n\t\t\t\t\"Custom.Field\": types.IndexedValueTypeString,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:   \"custom search attribute cannot contain dash\",\n\t\t\tquery: \"Custom-Field = 'value'\",\n\t\t\terr:   \"invalid comparison expression\",\n\t\t\tdcValid: map[string]interface{}{\n\t\t\t\t\"Custom-Field\": types.IndexedValueTypeString,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:       \"custom string search attribute support not like\",\n\t\t\tquery:     \"CustomStringField not like 'value'\",\n\t\t\tvalidated: \"`Attr.CustomStringField` not like 'value'\",\n\t\t\tdcValid: map[string]interface{}{\n\t\t\t\t\"CustomStringField\": types.IndexedValueTypeString,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.msg, func(t *testing.T) {\n\t\t\tvalidSearchAttr := func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\t\t\tvalid := definition.GetDefaultIndexedKeys()\n\t\t\t\tfor k, v := range tt.dcValid {\n\t\t\t\t\tvalid[k] = v\n\t\t\t\t}\n\t\t\t\treturn valid\n\t\t\t}\n\t\t\tvalidateSearchAttr := dynamicproperties.GetBoolPropertyFn(true)\n\t\t\tqv := NewQueryValidator(validSearchAttr, validateSearchAttr)\n\t\t\tvalidated, err := qv.ValidateQuery(tt.query)\n\t\t\tif err != nil {\n\t\t\t\tassert.Equal(t, tt.err, err.Error())\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.validated, validated)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/elasticsearch/validator/searchAttrValidator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to qvom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, qvETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage validator\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// SearchAttributesValidator is used to validate search attributes\ntype SearchAttributesValidator struct {\n\tlogger log.Logger\n\n\tenableQueryAttributeValidation    dynamicproperties.BoolPropertyFn\n\tvalidSearchAttributes             dynamicproperties.MapPropertyFn\n\tsearchAttributesNumberOfKeysLimit dynamicproperties.IntPropertyFnWithDomainFilter\n\tsearchAttributesSizeOfValueLimit  dynamicproperties.IntPropertyFnWithDomainFilter\n\tsearchAttributesTotalSizeLimit    dynamicproperties.IntPropertyFnWithDomainFilter\n}\n\n// NewSearchAttributesValidator create SearchAttributesValidator\nfunc NewSearchAttributesValidator(\n\tlogger log.Logger,\n\tenableQueryAttributeValidation dynamicproperties.BoolPropertyFn,\n\tvalidSearchAttributes dynamicproperties.MapPropertyFn,\n\tsearchAttributesNumberOfKeysLimit dynamicproperties.IntPropertyFnWithDomainFilter,\n\tsearchAttributesSizeOfValueLimit dynamicproperties.IntPropertyFnWithDomainFilter,\n\tsearchAttributesTotalSizeLimit dynamicproperties.IntPropertyFnWithDomainFilter,\n) *SearchAttributesValidator {\n\treturn &SearchAttributesValidator{\n\t\tlogger:                            logger,\n\t\tenableQueryAttributeValidation:    enableQueryAttributeValidation,\n\t\tvalidSearchAttributes:             validSearchAttributes,\n\t\tsearchAttributesNumberOfKeysLimit: searchAttributesNumberOfKeysLimit,\n\t\tsearchAttributesSizeOfValueLimit:  searchAttributesSizeOfValueLimit,\n\t\tsearchAttributesTotalSizeLimit:    searchAttributesTotalSizeLimit,\n\t}\n}\n\n// ValidateSearchAttributes validate search attributes are valid for writing and not exceed limits\nfunc (sv *SearchAttributesValidator) ValidateSearchAttributes(input *types.SearchAttributes, domain string) error {\n\tif input == nil {\n\t\treturn nil\n\t}\n\n\t// verify: number of keys <= limit\n\tfields := input.GetIndexedFields()\n\tlengthOfFields := len(fields)\n\tif lengthOfFields > sv.searchAttributesNumberOfKeysLimit(domain) {\n\t\tsv.logger.WithTags(tag.Number(int64(lengthOfFields)), tag.WorkflowDomainName(domain)).\n\t\t\tError(\"number of keys in search attributes exceed limit\")\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"number of keys %d exceed limit\", lengthOfFields)}\n\t}\n\n\ttotalSize := 0\n\n\tvalidateAttrFn := sv.enableQueryAttributeValidation\n\tvalidateAttr := true\n\tif validateAttrFn != nil {\n\t\tvalidateAttr = validateAttrFn()\n\t}\n\tvalidAttr := sv.validSearchAttributes()\n\tfor key, val := range fields {\n\t\tif validateAttr {\n\t\t\t// verify: key is whitelisted\n\t\t\tif !sv.isValidSearchAttributesKey(validAttr, key) {\n\t\t\t\tsv.logger.WithTags(tag.ESKey(key), tag.WorkflowDomainName(domain)).\n\t\t\t\t\tError(\"invalid search attribute key\")\n\t\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"%s is not a valid search attribute key\", key)}\n\t\t\t}\n\t\t\t// verify: value has the correct type\n\t\t\tif !sv.isValidSearchAttributesValue(validAttr, key, val) {\n\t\t\t\tsv.logger.WithTags(tag.ESKey(key), tag.ESValue(val), tag.WorkflowDomainName(domain)).\n\t\t\t\t\tError(\"invalid search attribute value\")\n\t\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"%s is not a valid search attribute value for key %s\", val, key)}\n\t\t\t}\n\t\t}\n\t\t// verify: key is not system reserved\n\t\tif definition.IsSystemIndexedKey(key) {\n\t\t\tsv.logger.WithTags(tag.ESKey(key), tag.WorkflowDomainName(domain)).\n\t\t\t\tError(\"illegal update of system reserved attribute\")\n\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"%s is read-only Cadence reservered attribute\", key)}\n\t\t}\n\t\t// verify: size of single value <= limit\n\t\tif len(val) > sv.searchAttributesSizeOfValueLimit(domain) {\n\t\t\tsv.logger.WithTags(tag.ESKey(key), tag.Number(int64(len(val))), tag.WorkflowDomainName(domain)).\n\t\t\t\tError(\"value size of search attribute exceed limit\")\n\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"size limit exceed for key %s\", key)}\n\t\t}\n\t\ttotalSize += len(key) + len(val)\n\t}\n\n\t// verify: total size <= limit\n\tif totalSize > sv.searchAttributesTotalSizeLimit(domain) {\n\t\tsv.logger.WithTags(tag.Number(int64(totalSize)), tag.WorkflowDomainName(domain)).\n\t\t\tError(\"total size of search attributes exceed limit\")\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"total size %d exceed limit\", totalSize)}\n\t}\n\n\treturn nil\n}\n\n// isValidSearchAttributesKey return true if key is registered\nfunc (sv *SearchAttributesValidator) isValidSearchAttributesKey(\n\tvalidAttr map[string]interface{},\n\tkey string,\n) bool {\n\t_, isValidKey := validAttr[key]\n\treturn isValidKey\n}\n\n// isValidSearchAttributesValue return true if value has the correct representation for the attribute key\nfunc (sv *SearchAttributesValidator) isValidSearchAttributesValue(\n\tvalidAttr map[string]interface{},\n\tkey string,\n\tvalue []byte,\n) bool {\n\tvalueType := common.ConvertIndexedValueTypeToInternalType(validAttr[key], sv.logger)\n\t_, err := common.DeserializeSearchAttributeValue(value, valueType)\n\treturn err == nil\n}\n"
  },
  {
    "path": "common/elasticsearch/validator/searchAttrValidator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to qvom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, qvETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage validator\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype searchAttributesValidatorSuite struct {\n\tsuite.Suite\n}\n\nfunc TestSearchAttributesValidatorSuite(t *testing.T) {\n\ts := new(searchAttributesValidatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *searchAttributesValidatorSuite) TestValidateSearchAttributes() {\n\tnumOfKeysLimit := 2\n\tsizeOfValueLimit := 5\n\tsizeOfTotalLimit := 20\n\n\tvalidator := NewSearchAttributesValidator(log.NewNoop(),\n\t\tdynamicproperties.GetBoolPropertyFn(true),\n\t\tdynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tdynamicproperties.GetIntPropertyFilteredByDomain(numOfKeysLimit),\n\t\tdynamicproperties.GetIntPropertyFilteredByDomain(sizeOfValueLimit),\n\t\tdynamicproperties.GetIntPropertyFilteredByDomain(sizeOfTotalLimit))\n\n\tdomain := \"domain\"\n\tvar attr *types.SearchAttributes\n\n\terr := validator.ValidateSearchAttributes(attr, domain)\n\ts.Nil(err)\n\n\tfields := map[string][]byte{\n\t\t\"CustomIntField\": []byte(`1`),\n\t}\n\tattr = &types.SearchAttributes{\n\t\tIndexedFields: fields,\n\t}\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Nil(err)\n\n\tfields = map[string][]byte{\n\t\t\"CustomIntField\":     []byte(`1`),\n\t\t\"CustomKeywordField\": []byte(`\"keyword\"`),\n\t\t\"CustomBoolField\":    []byte(`true`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(\"number of keys 3 exceed limit\", err.Error())\n\n\tfields = map[string][]byte{\n\t\t\"InvalidKey\": []byte(`\"1\"`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(`InvalidKey is not a valid search attribute key`, err.Error())\n\n\tfields = map[string][]byte{\n\t\t\"CustomStringField\": []byte(`\"1\"`),\n\t\t\"CustomBoolField\":   []byte(`123`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(`123 is not a valid search attribute value for key CustomBoolField`, err.Error())\n\n\tfields = map[string][]byte{\n\t\t\"CustomIntField\": []byte(`[1,2]`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.NoError(err)\n\n\tfields = map[string][]byte{\n\t\t\"StartTime\": []byte(`1`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(`StartTime is read-only Cadence reservered attribute`, err.Error())\n\n\tfields = map[string][]byte{\n\t\t\"CustomKeywordField\": []byte(`\"123456\"`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(`size limit exceed for key CustomKeywordField`, err.Error())\n\n\tfields = map[string][]byte{\n\t\t\"CustomKeywordField\": []byte(`\"123\"`),\n\t\t\"CustomStringField\":  []byte(`\"12\"`),\n\t}\n\tattr.IndexedFields = fields\n\terr = validator.ValidateSearchAttributes(attr, domain)\n\ts.Equal(`total size 44 exceed limit`, err.Error())\n}\n"
  },
  {
    "path": "common/errors/fake_errors.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage errors\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math/rand\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// ErrFakeServiceBusy is a fake service busy error.\n\tErrFakeServiceBusy = &types.ServiceBusyError{Message: \"Fake Service Busy Error.\"}\n\t// ErrFakeInternalService is a fake internal service error.\n\tErrFakeInternalService = &types.InternalServiceError{Message: \"Fake Internal Service Error.\"}\n\t// ErrFakeTimeout is a fake timeout error.\n\tErrFakeTimeout = context.DeadlineExceeded\n\t// ErrFakeUnhandled is a fake unhandled error.\n\tErrFakeUnhandled = errors.New(\"fake unhandled error\")\n)\n\nvar (\n\tfakeErrors = []error{\n\t\tErrFakeServiceBusy,\n\t\tErrFakeInternalService,\n\t\tErrFakeTimeout,\n\t\tErrFakeUnhandled,\n\t}\n)\n\n// ShouldForwardCall determines if the call should be forward to the underlying\n// client given the fake error generated\nfunc ShouldForwardCall(\n\terr error,\n) bool {\n\tif err == nil {\n\t\treturn true\n\t}\n\n\tif err == ErrFakeTimeout || err == ErrFakeUnhandled {\n\t\t// forward the call with 50% chance\n\t\treturn rand.Intn(2) == 0\n\t}\n\n\treturn false\n}\n\n// GenerateFakeError generates a random fake error\nfunc GenerateFakeError(\n\terrorRate float64,\n) error {\n\tif rand.Float64() < errorRate {\n\t\treturn fakeErrors[rand.Intn(len(fakeErrors))]\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/errors/internal_failure_error.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage errors\n\ntype (\n\t// InternalFailureError represents unexpected case happening or a code bug\n\tInternalFailureError struct {\n\t\tMsg string\n\t}\n)\n\n// NewInternalFailureError return internal failure error\nfunc NewInternalFailureError(msg string) *InternalFailureError {\n\treturn &InternalFailureError{Msg: msg}\n}\n\nfunc (e *InternalFailureError) Error() string {\n\treturn e.Msg\n}\n"
  },
  {
    "path": "common/errors/peer_hostname_error.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage errors\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n)\n\n// PeerHostnameError wraps an error with peer hostname information\ntype PeerHostnameError struct {\n\tPeerHostname string\n\tWrappedError error\n}\n\n// Error implements the error interface\nfunc (e *PeerHostnameError) Error() string {\n\treturn fmt.Sprintf(\"peer hostname: %s, error: %v\", e.PeerHostname, e.WrappedError)\n}\n\n// Unwrap implements the error unwrapping interface\nfunc (e *PeerHostnameError) Unwrap() error {\n\treturn e.WrappedError\n}\n\n// NewPeerHostnameError creates a new PeerHostnameError\nfunc NewPeerHostnameError(err error, peer string) error {\n\tif err == nil {\n\t\treturn nil\n\t}\n\tif peer == \"\" {\n\t\treturn err\n\t}\n\treturn &PeerHostnameError{\n\t\tPeerHostname: peer,\n\t\tWrappedError: err,\n\t}\n}\n\n// ExtractPeerHostname extracts the peer hostname from a wrapped error\n// Returns the hostname and the original unwrapped error\nfunc ExtractPeerHostname(err error) (string, error) {\n\tif err == nil {\n\t\treturn \"\", nil\n\t}\n\tvar peerErr *PeerHostnameError\n\tcurrent := err\n\tif errors.As(current, &peerErr) {\n\t\treturn peerErr.PeerHostname, peerErr.WrappedError\n\t}\n\treturn \"\", err\n}\n"
  },
  {
    "path": "common/errors/peer_hostname_error_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage errors\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestNewPeerHostnameError(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\terr      error\n\t\tpeer     string\n\t\twantErr  bool\n\t\twantPeer string\n\t}{\n\t\t{\n\t\t\tname:     \"nil error returns nil\",\n\t\t\terr:      nil,\n\t\t\tpeer:     \"host1\",\n\t\t\twantErr:  false,\n\t\t\twantPeer: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"empty peer returns original error\",\n\t\t\terr:      errors.New(\"original error\"),\n\t\t\tpeer:     \"\",\n\t\t\twantErr:  true,\n\t\t\twantPeer: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"wraps error with peer\",\n\t\t\terr:      errors.New(\"original error\"),\n\t\t\tpeer:     \"host1\",\n\t\t\twantErr:  true,\n\t\t\twantPeer: \"host1\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := NewPeerHostnameError(tt.err, tt.peer)\n\t\t\tif !tt.wantErr {\n\t\t\t\tassert.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.NotNil(t, got)\n\t\t\tif tt.wantPeer == \"\" {\n\t\t\t\tassert.Equal(t, tt.err, got)\n\t\t\t} else {\n\t\t\t\tvar peerErr *PeerHostnameError\n\t\t\t\tassert.True(t, errors.As(got, &peerErr))\n\t\t\t\tassert.Equal(t, tt.wantPeer, peerErr.PeerHostname)\n\t\t\t\tassert.Equal(t, tt.err, peerErr.WrappedError)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestExtractPeerHostname(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\terr      error\n\t\twantPeer string\n\t\twantErr  error\n\t}{\n\t\t{\n\t\t\tname:     \"nil error\",\n\t\t\terr:      nil,\n\t\t\twantPeer: \"\",\n\t\t\twantErr:  nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"non-peer error\",\n\t\t\terr:      errors.New(\"some error\"),\n\t\t\twantPeer: \"\",\n\t\t\twantErr:  errors.New(\"some error\"),\n\t\t},\n\t\t{\n\t\t\tname:     \"peer error\",\n\t\t\terr:      NewPeerHostnameError(errors.New(\"original error\"), \"host1\"),\n\t\t\twantPeer: \"host1\",\n\t\t\twantErr:  errors.New(\"original error\"),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tpeer, err := ExtractPeerHostname(tt.err)\n\t\t\tassert.Equal(t, tt.wantPeer, peer)\n\t\t\tassert.Equal(t, tt.wantErr, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/errors/tasklist_not_owned_by_host_error.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errors\n\nimport \"fmt\"\n\nvar _ error = &TaskListNotOwnedByHostError{}\n\ntype TaskListNotOwnedByHostError struct {\n\tOwnedByIdentity string\n\tMyIdentity      string\n\tTasklistName    string\n}\n\nfunc (m *TaskListNotOwnedByHostError) Error() string {\n\treturn fmt.Sprintf(\"task list is not owned by this host: OwnedBy: %s, Me: %s, Tasklist: %s\",\n\t\tm.OwnedByIdentity, m.MyIdentity, m.TasklistName)\n}\n\nfunc NewTaskListNotOwnedByHostError(ownedByIdentity string, myIdentity string, tasklistName string) *TaskListNotOwnedByHostError {\n\treturn &TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: ownedByIdentity,\n\t\tMyIdentity:      myIdentity,\n\t\tTasklistName:    tasklistName,\n\t}\n}\n"
  },
  {
    "path": "common/future/future.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage future\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"sync/atomic\"\n)\n\ntype (\n\t// Future represents a value that will be available in the future\n\tFuture interface {\n\t\tGet(ctx context.Context, valuePtr interface{}) error\n\t\tIsReady() bool\n\t}\n\n\t// Settable is for setting value and error for the corresponding Future\n\t// The Set method can only be called once. Later calls will result in a\n\t// panic since the value and error has already been set.\n\tSettable interface {\n\t\tSet(interface{}, error)\n\t}\n\n\tfutureImpl struct {\n\t\tvalue   interface{}\n\t\terr     error\n\t\treadyCh chan struct{}\n\t\tstatus  int32\n\t}\n)\n\nconst (\n\tvalueNotSet = iota\n\tvalueSet\n)\n\n// NewFuture creates a new Future and the Settable for setting its value\nfunc NewFuture() (Future, Settable) {\n\tfuture := &futureImpl{\n\t\treadyCh: make(chan struct{}),\n\t\tstatus:  valueNotSet,\n\t}\n\treturn future, future\n}\n\nfunc (f *futureImpl) Get(\n\tctx context.Context,\n\tvaluePtr interface{},\n) error {\n\tif err := ctx.Err(); err != nil {\n\t\t// if the given context is invalid,\n\t\t// guarantee to return an error\n\t\treturn err\n\t}\n\n\tselect {\n\tcase <-f.readyCh:\n\t\treturn f.populateValue(valuePtr)\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\t}\n}\n\nfunc (f *futureImpl) IsReady() bool {\n\tselect {\n\tcase <-f.readyCh:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc (f *futureImpl) Set(\n\tvalue interface{},\n\terr error,\n) {\n\tif !atomic.CompareAndSwapInt32(&f.status, valueNotSet, valueSet) {\n\t\tpanic(\"future has already been set\")\n\t}\n\n\tf.value = value\n\tf.err = err\n\tclose(f.readyCh)\n}\n\nfunc (f *futureImpl) populateValue(\n\tvaluePtr interface{},\n) (err error) {\n\tdefer func() {\n\t\tif p := recover(); p != nil {\n\t\t\terr = fmt.Errorf(\"failed to populate valuePtr: %v\", p)\n\t\t}\n\t}()\n\n\tif f.err != nil || f.value == nil || valuePtr == nil {\n\t\treturn f.err\n\t}\n\n\trf := reflect.ValueOf(valuePtr)\n\tif rf.Type().Kind() != reflect.Ptr {\n\t\treturn errors.New(\"valuePtr parameter is not a pointer\")\n\t}\n\n\tfv := reflect.ValueOf(f.value)\n\tif fv.IsValid() {\n\t\trf.Elem().Set(fv)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/future/future_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage future\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tfutureSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t}\n\n\ttestType struct {\n\t\tintField    int\n\t\tstringField string\n\t\ttimeField   time.Time\n\t}\n)\n\nfunc TestFutureSuite(t *testing.T) {\n\ts := new(futureSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *futureSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *futureSuite) TestFutureSetGet() {\n\tvar intVal int\n\tvar listVal []string\n\tvar mapVal map[string]struct{}\n\tvar testTypeVal testType\n\n\ttestCases := []struct {\n\t\tfutureValue interface{}\n\t\tfutureErr   error\n\t\tvaluePtr    interface{}\n\t\texpectErr   bool\n\t}{\n\t\t{\n\t\t\tfutureValue: testType{\n\t\t\t\tintField:    101,\n\t\t\t\tstringField: \"a\",\n\t\t\t\ttimeField:   time.Now(),\n\t\t\t},\n\t\t\tvaluePtr: &testTypeVal,\n\t\t},\n\t\t{\n\t\t\tfutureValue: int(101),\n\t\t\tvaluePtr:    &intVal,\n\t\t},\n\t\t{\n\t\t\tfutureValue: []string{\"a\", \"b\", \"c\"},\n\t\t\tvaluePtr:    &listVal,\n\t\t},\n\t\t{\n\t\t\tfutureValue: nil,\n\t\t\tvaluePtr:    &listVal,\n\t\t},\n\t\t{\n\t\t\tfutureValue: map[string]struct{}{\"a\": {}, \"b\": {}},\n\t\t\tvaluePtr:    &mapVal,\n\t\t},\n\t\t{\n\t\t\tfutureValue: map[string]struct{}{\"a\": {}, \"b\": {}},\n\t\t\tvaluePtr:    &intVal,\n\t\t\texpectErr:   true,\n\t\t},\n\t\t{\n\t\t\tfutureValue: int(101),\n\t\t\tvaluePtr:    101,\n\t\t\texpectErr:   true,\n\t\t},\n\t\t{\n\t\t\tfutureValue: nil,\n\t\t\tfutureErr:   errors.New(\"some random value\"),\n\t\t\tvaluePtr:    &intVal,\n\t\t\texpectErr:   true,\n\t\t},\n\t\t{\n\t\t\tfutureValue: time.Now(),\n\t\t\tvaluePtr:    nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tfuture, settable := NewFuture()\n\t\tsettable.Set(tc.futureValue, tc.futureErr)\n\n\t\terr := future.Get(context.Background(), tc.valuePtr)\n\t\tif tc.expectErr {\n\t\t\tif tc.futureErr != nil {\n\t\t\t\ts.Equal(tc.futureErr, err)\n\t\t\t}\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\n\t\ts.NoError(err)\n\t\tif tc.valuePtr != nil {\n\t\t\tswitch tc.futureValue.(type) {\n\t\t\tcase int:\n\t\t\t\ts.Equal(tc.futureValue, intVal)\n\t\t\tcase []string:\n\t\t\t\ts.Equal(tc.futureValue, listVal)\n\t\t\tcase map[string]struct{}:\n\t\t\t\ts.Equal(tc.futureValue, mapVal)\n\t\t\tcase testType:\n\t\t\t\ts.Equal(tc.futureValue, testTypeVal)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *futureSuite) TestFutureGet_ContextErr() {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*100)\n\tdefer cancel()\n\n\tfuture, settable := NewFuture()\n\terr := future.Get(ctx, nil)\n\ts.Error(err)\n\n\tsettable.Set(\"some random value\", nil)\n\terr = future.Get(ctx, nil)\n\ts.Error(err)\n}\n\nfunc (s *futureSuite) TestFutureIsReady() {\n\tfuture, settable := NewFuture()\n\ts.False(future.IsReady())\n\n\tsettable.Set(nil, errors.New(\"some random error\"))\n\ts.True(future.IsReady())\n}\n\nfunc (s *futureSuite) TestFutureDoubleSet() {\n\t_, settable := NewFuture()\n\tsettable.Set(\"some random value\", nil)\n\n\ts.Panics(func() {\n\t\tsettable.Set(nil, errors.New(\"some random error\"))\n\t})\n}\n"
  },
  {
    "path": "common/headers.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nimport \"github.com/uber/cadence/common/types\"\n\nconst (\n\t// LibraryVersionHeaderName refers to the name of the\n\t// tchannel / http header that contains the client\n\t// library version\n\tLibraryVersionHeaderName = \"cadence-client-library-version\"\n\n\t// FeatureVersionHeaderName refers to the name of the\n\t// tchannel / http header that contains the client\n\t// feature version\n\t// the feature version sent from client represents the\n\t// feature set of the cadence client library support.\n\t// This can be used for client capibility check, on\n\t// Cadence server, for backward compatibility\n\tFeatureVersionHeaderName = \"cadence-client-feature-version\"\n\n\t// Header name to pass the feature flags from customer to client or server\n\tClientFeatureFlagsHeaderName = \"cadence-client-feature-flags\"\n\n\t// ClientImplHeaderName refers to the name of the\n\t// header that contains the client implementation\n\tClientImplHeaderName = \"cadence-client-name\"\n\t// AuthorizationTokenHeaderName refers to the jwt token in the request\n\tAuthorizationTokenHeaderName = \"cadence-authorization\"\n\n\t// AutoforwardingClusterHeaderName refers to the name of the header that contains the source cluster of the auto-forwarding request\n\tAutoforwardingClusterHeaderName = \"cadence-forwarding-cluster\"\n\n\t// PartitionConfigHeaderName refers to the name of the header that contains the json encoded partition config of the request\n\tPartitionConfigHeaderName = \"cadence-workflow-partition-config\"\n\t// IsolationGroupHeaderName refers to the name of the header that contains the isolation group of the client\n\tIsolationGroupHeaderName = \"cadence-worker-isolation-group\"\n\n\t// ClientIsolationGroupHeaderName refers to the name of the header that contains the isolation group which the client request is from\n\tClientIsolationGroupHeaderName = \"cadence-client-isolation-group\"\n\n\t// CallerTypeHeaderName refers to the name of the header that contains the caller type (CLI, UI, SDK, internal, etc.)\n\tCallerTypeHeaderName = types.CallerTypeHeaderName\n)\n"
  },
  {
    "path": "common/isolationgroup/context.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroup\n\nimport \"context\"\n\nconst (\n\tGroupKey         = \"isolation-group\"\n\tOriginalGroupKey = \"original-isolation-group\"\n\tWorkflowIDKey    = \"wf-id\"\n)\n\ntype configKey struct{}\n\ntype isolationGroupKey struct{}\n\n// ConfigFromContext retrieves infomation about the partition config of the context\n// which is used for tasklist isolation\nfunc ConfigFromContext(ctx context.Context) map[string]string {\n\tval, ok := ctx.Value(configKey{}).(map[string]string)\n\tif !ok {\n\t\treturn nil\n\t}\n\treturn val\n}\n\n// ContextWithConfig stores the partition config of tasklist isolation into the given context\nfunc ContextWithConfig(ctx context.Context, partitionConfig map[string]string) context.Context {\n\treturn context.WithValue(ctx, configKey{}, partitionConfig)\n}\n\n// IsolationGroupFromContext retrieves the isolation group from the given context,\n// which is used to identify which isolation group the poller is from\nfunc IsolationGroupFromContext(ctx context.Context) string {\n\tval, ok := ctx.Value(isolationGroupKey{}).(string)\n\tif !ok {\n\t\treturn \"\"\n\t}\n\treturn val\n}\n\n// ContextWithIsolationGroup stores the isolation group into the given context\nfunc ContextWithIsolationGroup(ctx context.Context, isolationGroup string) context.Context {\n\treturn context.WithValue(ctx, isolationGroupKey{}, isolationGroup)\n}\n"
  },
  {
    "path": "common/isolationgroup/context_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroup\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestContext(t *testing.T) {\n\tpartitionConfig := map[string]string{\"test\": \"abc\"}\n\tassert.Equal(t, partitionConfig, ConfigFromContext(ContextWithConfig(context.Background(), partitionConfig)))\n\n\tisolationGroup := \"zone1\"\n\tassert.Equal(t, isolationGroup, IsolationGroupFromContext(ContextWithIsolationGroup(context.Background(), isolationGroup)))\n}\n"
  },
  {
    "path": "common/isolationgroup/defaultisolationgroupstate/state.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage defaultisolationgroupstate\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype defaultIsolationGroupStateHandler struct {\n\tstatus                     int32\n\tdone                       chan struct{}\n\tlog                        log.Logger\n\tdomainCache                cache.DomainCache\n\tglobalIsolationGroupDrains dynamicconfig.Client\n\tconfig                     defaultConfig\n\tmetricsClient              metrics.Client\n}\n\n// NewDefaultIsolationGroupStateWatcherWithConfigStoreClient Is a constructor which allows passing in the dynamic config client\nfunc NewDefaultIsolationGroupStateWatcherWithConfigStoreClient(\n\tlogger log.Logger,\n\tdc *dynamicconfig.Collection,\n\tdomainCache cache.DomainCache,\n\tcfgStoreClient dynamicconfig.Client, // can be nil, which means global drain is unsupported\n\tmetricsClient metrics.Client,\n\tgetIsolationGroups func() []string,\n) (isolationgroup.State, error) {\n\tstopChan := make(chan struct{})\n\n\tconfig := defaultConfig{\n\t\tIsolationGroupEnabled: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableTasklistIsolation),\n\t\tAllIsolationGroups:    getIsolationGroups,\n\t}\n\n\treturn &defaultIsolationGroupStateHandler{\n\t\tdone:                       stopChan,\n\t\tdomainCache:                domainCache,\n\t\tglobalIsolationGroupDrains: cfgStoreClient,\n\t\tstatus:                     common.DaemonStatusInitialized,\n\t\tlog:                        logger,\n\t\tconfig:                     config,\n\t\tmetricsClient:              metricsClient,\n\t}, nil\n}\n\nfunc (z *defaultIsolationGroupStateHandler) IsDrained(ctx context.Context, domain string, isolationGroup string) (bool, error) {\n\tstate, err := z.get(ctx, domain)\n\tif err != nil {\n\t\treturn false, fmt.Errorf(\"could not determine if drained: %w\", err)\n\t}\n\treturn isDrained(isolationGroup, state.Global, state.Domain), nil\n}\n\nfunc (z *defaultIsolationGroupStateHandler) IsDrainedByDomainID(ctx context.Context, domainID string, isolationGroup string) (bool, error) {\n\tdomain, err := z.domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn false, fmt.Errorf(\"could not determine if drained: %w\", err)\n\t}\n\treturn z.IsDrained(ctx, domain.GetInfo().Name, isolationGroup)\n}\n\n// Start the state handler\nfunc (z *defaultIsolationGroupStateHandler) Start() {\n\tif !atomic.CompareAndSwapInt32(&z.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n}\n\nfunc (z *defaultIsolationGroupStateHandler) Stop() {\n\tif z == nil {\n\t\treturn\n\t}\n\tif !atomic.CompareAndSwapInt32(&z.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(z.done)\n}\n\n// Get the statue of a isolationGroup, with respect to both domain and global drains. Domain-specific drains override global config\n// will return nil, nil when it is not enabled\nfunc (z *defaultIsolationGroupStateHandler) get(ctx context.Context, domain string) (*isolationGroups, error) {\n\tif !z.config.IsolationGroupEnabled(domain) {\n\t\treturn &isolationGroups{}, nil\n\t}\n\n\tdomainData, err := z.domainCache.GetDomain(domain)\n\tif err != nil || domainData == nil {\n\t\treturn nil, fmt.Errorf(\"could not resolve domain in isolationGroup handler: %w\", err)\n\t}\n\n\tdomainCfg := domainData.GetConfig()\n\tvar domainState types.IsolationGroupConfiguration\n\tif domainCfg != nil && domainCfg.IsolationGroups != nil {\n\t\tdomainState = domainCfg.IsolationGroups\n\t}\n\tig := &isolationGroups{\n\t\tDomain: domainState,\n\t}\n\n\tif z.globalIsolationGroupDrains != nil {\n\t\tglobalCfg, err := z.globalIsolationGroupDrains.GetListValue(dynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping, nil)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"could not resolve global drains in %w\", err)\n\t\t}\n\n\t\tglobalState, err := isolationgroupapi.MapDynamicConfigResponse(globalCfg)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"could not resolve global drains in isolationGroup handler: %w\", err)\n\t\t}\n\t\tig.Global = globalState\n\t}\n\n\treturn ig, nil\n}\n\nfunc isDrained(isolationGroup string, global types.IsolationGroupConfiguration, domain types.IsolationGroupConfiguration) bool {\n\tglobalCfg, hasGlobalConfig := global[isolationGroup]\n\tdomainCfg, hasDomainConfig := domain[isolationGroup]\n\tif hasGlobalConfig {\n\t\tif globalCfg.State == types.IsolationGroupStateDrained {\n\t\t\treturn true\n\t\t}\n\t}\n\tif hasDomainConfig {\n\t\tif domainCfg.State == types.IsolationGroupStateDrained {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/isolationgroup/defaultisolationgroupstate/state_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage defaultisolationgroupstate\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestIsDrainedHandler(t *testing.T) {\n\n\tvalidInput := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t}\n\n\tvalidCfg, _ := isolationgroupapi.MapUpdateGlobalIsolationGroupsRequest(validInput)\n\n\tvalidCfgData := validCfg[0].Value.GetData()\n\tdynamicConfigResponse := []interface{}{}\n\tjson.Unmarshal(validCfgData, &dynamicConfigResponse)\n\n\ttests := map[string]struct {\n\t\trequestIsolationgroup string\n\t\tdcAffordance          func(client *dynamicconfig.MockClient)\n\t\tdomainAffordance      func(mock *cache.MockDomainCache)\n\t\tcfg                   defaultConfig\n\t\texpected              bool\n\t\texpectedErr           error\n\t}{\n\t\t\"normal case - no drains present - no configuration specifying a drain - feature is enabled\": {\n\t\t\trequestIsolationgroup: \"zone-3\", // no config specified for this\n\t\t\tcfg: defaultConfig{\n\t\t\t\tIsolationGroupEnabled: func(string) bool { return true },\n\t\t\t\tAllIsolationGroups:    func() []string { return []string{\"zone-1\", \"zone-2\", \"zone-3\"} },\n\t\t\t},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {\n\t\t\t\tclient.EXPECT().GetListValue(\n\t\t\t\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\t\t\tgomock.Any(), // covering the mapping in the mapper unit-test instead\n\t\t\t\t).Return(dynamicConfigResponse, nil)\n\t\t\t},\n\t\t\tdomainAffordance: func(mock *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"domain-id\", Name: \"domain\"}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\t\t\t\tmock.EXPECT().GetDomainByID(\"domain-id\").Return(domainResponse, nil)\n\t\t\t\tmock.EXPECT().GetDomain(\"domain\").Return(domainResponse, nil)\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t\"normal case - drains present - feature is enabled\": {\n\t\t\trequestIsolationgroup: \"zone-2\", // no config specified for this\n\t\t\tcfg: defaultConfig{\n\t\t\t\tIsolationGroupEnabled: func(string) bool { return true },\n\t\t\t\tAllIsolationGroups:    func() []string { return []string{\"zone-1\", \"zone-2\", \"zone-3\"} },\n\t\t\t},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {\n\t\t\t\tclient.EXPECT().GetListValue(\n\t\t\t\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\t\t\tgomock.Any(), // covering the mapping in the mapper unit-test instead\n\t\t\t\t).Return(dynamicConfigResponse, nil)\n\t\t\t},\n\t\t\tdomainAffordance: func(mock *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"domain-id\", Name: \"domain\"}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\t\t\t\tmock.EXPECT().GetDomainByID(\"domain-id\").Return(domainResponse, nil)\n\t\t\t\tmock.EXPECT().GetDomain(\"domain\").Return(domainResponse, nil)\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t\"normal case - feature is disabled\": {\n\t\t\trequestIsolationgroup: \"zone-2\", // no config specified for this\n\t\t\tcfg: defaultConfig{\n\t\t\t\tIsolationGroupEnabled: func(string) bool { return false },\n\t\t\t\tAllIsolationGroups:    func() []string { return []string{\"zone-1\", \"zone-2\", \"zone-3\"} },\n\t\t\t},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {},\n\t\t\tdomainAffordance: func(mock *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"domain-id\", Name: \"domain\"}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\t\t\t\tmock.EXPECT().GetDomainByID(\"domain-id\").Return(domainResponse, nil)\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmockCtl := gomock.NewController(t)\n\t\t\tdcMock := dynamicconfig.NewMockClient(mockCtl)\n\t\t\tdomaincacheMock := cache.NewMockDomainCache(mockCtl)\n\t\t\ttd.dcAffordance(dcMock)\n\t\t\ttd.domainAffordance(domaincacheMock)\n\t\t\thandler := defaultIsolationGroupStateHandler{\n\t\t\t\tlog:                        testlogger.New(t),\n\t\t\t\tglobalIsolationGroupDrains: dcMock,\n\t\t\t\tdomainCache:                domaincacheMock,\n\t\t\t\tconfig:                     td.cfg,\n\t\t\t}\n\t\t\tres, err := handler.IsDrainedByDomainID(context.Background(), \"domain-id\", td.requestIsolationgroup)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestIsDrained(t *testing.T) {\n\n\tigA := \"isolationGroupA\"\n\tigB := \"isolationGroupB\"\n\n\tisolationGroupsAllHealthy := types.IsolationGroupConfiguration{\n\t\tigA: {\n\t\t\tName:  igA,\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t\tigB: {\n\t\t\tName:  igB,\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n\n\tisolationGroupsOneDrain := types.IsolationGroupConfiguration{\n\t\tigA: {\n\t\t\tName:  igA,\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tglobalIGCfg    types.IsolationGroupConfiguration\n\t\tdomainIGCfg    types.IsolationGroupConfiguration\n\t\tisolationGroup string\n\t\texpected       bool\n\t}{\n\t\t\"default behaviour - no drains - isolationGroup is specified\": {\n\t\t\tglobalIGCfg:    isolationGroupsAllHealthy,\n\t\t\tdomainIGCfg:    isolationGroupsAllHealthy,\n\t\t\tisolationGroup: igA,\n\t\t\texpected:       false,\n\t\t},\n\t\t\"default behaviour - no drains - isolationGroup is not specified\": {\n\t\t\tglobalIGCfg:    isolationGroupsAllHealthy,\n\t\t\tdomainIGCfg:    isolationGroupsAllHealthy,\n\t\t\tisolationGroup: \"some-not-specified-drain\",\n\t\t\texpected:       false,\n\t\t},\n\t\t\"default behaviour - globalDrain\": {\n\t\t\tglobalIGCfg:    isolationGroupsOneDrain,\n\t\t\tdomainIGCfg:    isolationGroupsAllHealthy,\n\t\t\tisolationGroup: igA,\n\t\t\texpected:       true,\n\t\t},\n\t\t\"default behaviour - domainDrain\": {\n\t\t\tglobalIGCfg:    isolationGroupsAllHealthy,\n\t\t\tdomainIGCfg:    isolationGroupsOneDrain,\n\t\t\tisolationGroup: igA,\n\t\t\texpected:       true,\n\t\t},\n\t\t\"default behaviour - both \": {\n\t\t\tglobalIGCfg:    isolationGroupsOneDrain,\n\t\t\tdomainIGCfg:    isolationGroupsOneDrain,\n\t\t\tisolationGroup: igA,\n\t\t\texpected:       true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, isDrained(td.isolationGroup, td.globalIGCfg, td.domainIGCfg))\n\t\t})\n\t}\n}\n\nfunc TestIsolationGroupStateMapping(t *testing.T) {\n\n\tz1 := types.IsolationGroupPartition{\n\t\tName:  \"zone-1\",\n\t\tState: types.IsolationGroupStateHealthy,\n\t}\n\n\tz2 := types.IsolationGroupPartition{\n\t\tName:  \"zone-2\",\n\t\tState: types.IsolationGroupStateDrained,\n\t}\n\n\t// JSON serialization inside the dynamic config library makes a mess of things because\n\t// it doesn't have any type information. So mimicing this type information loss to ensure\n\t// any field name types or similar serialization quirks are picked up by the test\n\tzMarshalled, _ := json.Marshal([]types.IsolationGroupPartition{z1, z2})\n\tvar rawIsolationGroupDataMarshalled []interface{}\n\tjson.Unmarshal(zMarshalled, &rawIsolationGroupDataMarshalled)\n\n\ttests := map[string]struct {\n\t\tin          []interface{}\n\t\texpected    types.IsolationGroupConfiguration\n\t\texpectedErr error\n\t}{\n\t\t\"valid mapping\": {\n\t\t\tin: rawIsolationGroupDataMarshalled,\n\t\t\texpected: map[string]types.IsolationGroupPartition{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty mapping\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"invalid mapping 1\": {\n\t\t\tin:          []interface{}{\"invalid\"},\n\t\t\texpectedErr: errors.New(\"failed parse a dynamic config entry, map[], (got invalid)\"),\n\t\t},\n\t\t\"invalid mapping 2\": {\n\t\t\tin:          []interface{}{`{\"Name\": \"some zone\", \"State\": \"not the right type\"}`},\n\t\t\texpectedErr: errors.New(`failed parse a dynamic config entry, map[], (got {\"Name\": \"some zone\", \"State\": \"not the right type\"})`),\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres, err := isolationgroupapi.MapDynamicConfigResponse(td.in)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestUpdateRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin          types.IsolationGroupConfiguration\n\t\texpected    []*types.DynamicConfigValue\n\t\texpectedErr error\n\t}{\n\t\t\"valid mapping\": {\n\t\t\tin: types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*types.DynamicConfigValue{\n\t\t\t\t{\n\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`[{\"Name\":\"zone-1\",\"State\":1},{\"Name\":\"zone-2\",\"State\":2}]`),\n\t\t\t\t\t},\n\t\t\t\t\tFilters: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty mapping\": {\n\t\t\tin: types.IsolationGroupConfiguration{},\n\t\t\texpected: []*types.DynamicConfigValue{\n\t\t\t\t{\n\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`[]`),\n\t\t\t\t\t},\n\t\t\t\t\tFilters: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres, err := isolationgroupapi.MapUpdateGlobalIsolationGroupsRequest(td.in)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestNewDefaultIsolationGroupStateWatcherWithConfigStoreClient(t *testing.T) {\n\tdc := dynamicconfig.NewNopCollection()\n\tdomainCache := cache.NewNoOpDomainCache()\n\tclient := metrics.NewNoopMetricsClient()\n\tig := func() []string { return nil }\n\tNewDefaultIsolationGroupStateWatcherWithConfigStoreClient(\n\t\tlog.NewNoop(),\n\t\tdc,\n\t\tdomainCache,\n\t\tnil,\n\t\tclient,\n\t\tig,\n\t)\n}\n\nfunc TestIsolationGroupShutdown(t *testing.T) {\n\tvar v defaultIsolationGroupStateHandler\n\tassert.NotPanics(t, func() {\n\t\tv.Stop()\n\t})\n}\n"
  },
  {
    "path": "common/isolationgroup/defaultisolationgroupstate/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage defaultisolationgroupstate\n\nimport (\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// IsolationGroups is an internal convenience return type of a collection of IsolationGroup configurations\ntype isolationGroups struct {\n\tGlobal types.IsolationGroupConfiguration\n\tDomain types.IsolationGroupConfiguration\n}\n\n// defaultConfig values for the partitioning library for segmenting portions of workflows into isolation-groups - a resiliency\n// concept meant to help move workflows around and away from failure zones.\ntype defaultConfig struct {\n\t// IsolationGroupEnabled is a domain-based configuration value for whether this feature is enabled at all\n\tIsolationGroupEnabled dynamicproperties.BoolPropertyFnWithDomainFilter\n\t// AllIsolationGroups is all the possible isolation-groups available for a region\n\tAllIsolationGroups func() []string\n}\n"
  },
  {
    "path": "common/isolationgroup/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroup\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination isolation_group_mock.go -self_package github.com/uber/cadence/common/isolationgroup\n\n// State is a heavily cached in-memory library for returning the state of what zones are healthy or\n// drained presently. It may return an inclusive (allow-list based) or an exclusive (deny-list based) set of IsolationGroups\n// depending on the implementation.\ntype State interface {\n\tcommon.Daemon\n\t// IsDrained answers the question - \"is this particular isolationGroup drained?\". Used by startWorkflow calls\n\t// and similar sync frontend calls to make routing decisions\n\tIsDrained(ctx context.Context, Domain string, IsolationGroup string) (bool, error)\n\tIsDrainedByDomainID(ctx context.Context, DomainID string, IsolationGroup string) (bool, error)\n}\n"
  },
  {
    "path": "common/isolationgroup/isolation_group_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package isolationgroup -source interface.go -destination isolation_group_mock.go -self_package github.com/uber/cadence/common/isolationgroup\n//\n\n// Package isolationgroup is a generated GoMock package.\npackage isolationgroup\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockState is a mock of State interface.\ntype MockState struct {\n\tctrl     *gomock.Controller\n\trecorder *MockStateMockRecorder\n\tisgomock struct{}\n}\n\n// MockStateMockRecorder is the mock recorder for MockState.\ntype MockStateMockRecorder struct {\n\tmock *MockState\n}\n\n// NewMockState creates a new mock instance.\nfunc NewMockState(ctrl *gomock.Controller) *MockState {\n\tmock := &MockState{ctrl: ctrl}\n\tmock.recorder = &MockStateMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockState) EXPECT() *MockStateMockRecorder {\n\treturn m.recorder\n}\n\n// IsDrained mocks base method.\nfunc (m *MockState) IsDrained(ctx context.Context, Domain, IsolationGroup string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDrained\", ctx, Domain, IsolationGroup)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsDrained indicates an expected call of IsDrained.\nfunc (mr *MockStateMockRecorder) IsDrained(ctx, Domain, IsolationGroup any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDrained\", reflect.TypeOf((*MockState)(nil).IsDrained), ctx, Domain, IsolationGroup)\n}\n\n// IsDrainedByDomainID mocks base method.\nfunc (m *MockState) IsDrainedByDomainID(ctx context.Context, DomainID, IsolationGroup string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDrainedByDomainID\", ctx, DomainID, IsolationGroup)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsDrainedByDomainID indicates an expected call of IsDrainedByDomainID.\nfunc (mr *MockStateMockRecorder) IsDrainedByDomainID(ctx, DomainID, IsolationGroup any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDrainedByDomainID\", reflect.TypeOf((*MockState)(nil).IsDrainedByDomainID), ctx, DomainID, IsolationGroup)\n}\n\n// Start mocks base method.\nfunc (m *MockState) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockStateMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockState)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockState) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockStateMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockState)(nil).Stop))\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/domain-api-handlers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (z *handlerImpl) GetDomainState(ctx context.Context, request types.GetDomainIsolationGroupsRequest) (*types.GetDomainIsolationGroupsResponse, error) {\n\tres, err := z.domainHandler.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: &request.Domain,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif res == nil || res.Configuration == nil || res.Configuration.IsolationGroups == nil {\n\t\treturn &types.GetDomainIsolationGroupsResponse{}, nil\n\t}\n\treturn &types.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: *res.Configuration.IsolationGroups,\n\t}, nil\n}\n\n// UpdateDomainState is the read operation for updating a domain's isolation-groups\n// todo (david.porter) delete this handler and use domain-handler directly\nfunc (z *handlerImpl) UpdateDomainState(ctx context.Context, request types.UpdateDomainIsolationGroupsRequest) error {\n\terr := z.domainHandler.UpdateIsolationGroups(ctx, request)\n\treturn err\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/domain-api-handlers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\nfunc TestUpdateDomainState(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin                      types.UpdateDomainIsolationGroupsRequest\n\t\tdomainHandlerAffordance func(h *domain.MockHandler)\n\t\texpectedErr             error\n\t}{\n\t\t\"updating normal value\": {\n\t\t\tin: types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateHealthy},\n\t\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: types.IsolationGroupStateDrained},\n\t\t\t\t}},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().UpdateIsolationGroups(gomock.Any(), types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateHealthy},\n\t\t\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: types.IsolationGroupStateDrained},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t\"empty value - ie removing isolation groups\": {\n\t\t\tin: types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain:          \"domain\",\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().UpdateIsolationGroups(gomock.Any(), types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain:          \"domain\",\n\t\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t\"error state - nil value - should not happen\": {\n\t\t\tin: types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain:          \"domain\",\n\t\t\t\tIsolationGroups: nil,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\tvar ig types.IsolationGroupConfiguration\n\t\t\t\th.EXPECT().UpdateIsolationGroups(gomock.Any(), types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain:          \"domain\",\n\t\t\t\t\tIsolationGroups: ig, // nil\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t\"error returned\": {\n\t\t\tin: types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain:          \"domain\",\n\t\t\t\tIsolationGroups: nil,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\tvar ig types.IsolationGroupConfiguration\n\t\t\t\th.EXPECT().UpdateIsolationGroups(gomock.Any(), types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain:          \"domain\",\n\t\t\t\t\tIsolationGroups: ig,\n\t\t\t\t}).Return(assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: assert.AnError,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdcMock := dynamicconfig.NewMockClient(gomock.NewController(t))\n\t\t\tdomainHandlerMock := domain.NewMockHandler(ctrl)\n\t\t\ttd.domainHandlerAffordance(domainHandlerMock)\n\t\t\thandler := handlerImpl{\n\t\t\t\tlog:                        testlogger.New(t),\n\t\t\t\tdomainHandler:              domainHandlerMock,\n\t\t\t\tglobalIsolationGroupDrains: dcMock,\n\t\t\t}\n\t\t\terr := handler.UpdateDomainState(context.Background(), td.in)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestGetDomainState(t *testing.T) {\n\n\tvalidCfg := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n\n\tdomainName := \"domain\"\n\n\ttests := map[string]struct {\n\t\tin                      types.GetDomainIsolationGroupsRequest\n\t\texpected                *types.GetDomainIsolationGroupsResponse\n\t\tdomainHandlerAffordance func(h *domain.MockHandler)\n\t\texpectedErr             error\n\t}{\n\t\t\"normal value being returned - valid config present\": {\n\t\t\tin: types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tIsolationGroups: &validCfg,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsResponse{IsolationGroups: validCfg},\n\t\t},\n\t\t\"normal value being returned - no config present - should just return empty\": {\n\t\t\tin: types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsResponse{},\n\t\t},\n\t\t\"normal value being returned - no config present - should just return nil - 2\": {\n\t\t\tin: types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tIsolationGroups: &types.IsolationGroupConfiguration{},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t},\n\t\t\"an error was returned\": {\n\t\t\tin: types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\th.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: assert.AnError,\n\t\t},\n\t\t\"invalid input - nil input\": {\n\t\t\tin: types.GetDomainIsolationGroupsRequest{},\n\t\t\tdomainHandlerAffordance: func(h *domain.MockHandler) {\n\t\t\t\tn := \"\"\n\t\t\t\th.EXPECT().\n\t\t\t\t\tDescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\t\tName: &n,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil, nil)\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsResponse{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdomainHandlerMock := domain.NewMockHandler(ctrl)\n\t\t\ttd.domainHandlerAffordance(domainHandlerMock)\n\t\t\thandler := handlerImpl{\n\t\t\t\tlog:           testlogger.New(t),\n\t\t\t\tdomainHandler: domainHandlerMock,\n\t\t\t}\n\t\t\tres, err := handler.GetDomainState(context.Background(), td.in)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tif td.expectedErr != nil {\n\t\t\t\tassert.Equal(t, td.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/global-api-handlers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (z *handlerImpl) UpdateGlobalState(ctx context.Context, in types.UpdateGlobalIsolationGroupsRequest) error {\n\tif z.globalIsolationGroupDrains == nil {\n\t\treturn &types.BadRequestError{\"global isolation group drain is not supported in this cluster\"}\n\t}\n\tmappedInput, err := MapUpdateGlobalIsolationGroupsRequest(in.IsolationGroups)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn z.globalIsolationGroupDrains.UpdateValue(\n\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\tmappedInput,\n\t)\n}\n\nfunc (z *handlerImpl) GetGlobalState(ctx context.Context) (*types.GetGlobalIsolationGroupsResponse, error) {\n\tif z.globalIsolationGroupDrains == nil {\n\t\treturn nil, &types.BadRequestError{\"global isolation group drain is not supported in this cluster\"}\n\t}\n\tres, err := z.globalIsolationGroupDrains.GetListValue(dynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping, nil)\n\tif err != nil {\n\t\tvar e types.EntityNotExistsError\n\t\tif errors.As(err, &e) {\n\t\t\treturn &types.GetGlobalIsolationGroupsResponse{}, nil\n\t\t}\n\t\treturn nil, fmt.Errorf(\"failed to get global isolation groups from datastore: %w\", err)\n\t}\n\tresp, err := MapDynamicConfigResponse(res)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to get global isolation groups from datastore: %w\", err)\n\t}\n\treturn &types.GetGlobalIsolationGroupsResponse{IsolationGroups: resp}, nil\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/global-api-handlers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestUpdateGlobalState(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin           types.UpdateGlobalIsolationGroupsRequest\n\t\tdcAffordance func(client *dynamicconfig.MockClient)\n\t\texpectedErr  error\n\t}{\n\t\t\"updating normal value\": {\n\t\t\tin: types.UpdateGlobalIsolationGroupsRequest{IsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateHealthy},\n\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: types.IsolationGroupStateDrained},\n\t\t\t}},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {\n\t\t\t\tclient.EXPECT().UpdateValue(\n\t\t\t\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\t\t\tgomock.Any(), // covering the mapping in the mapper unit-test instead\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tdcMock := dynamicconfig.NewMockClient(gomock.NewController(t))\n\t\t\ttd.dcAffordance(dcMock)\n\t\t\thandler := handlerImpl{\n\t\t\t\tlog:                        testlogger.New(t),\n\t\t\t\tglobalIsolationGroupDrains: dcMock,\n\t\t\t}\n\t\t\terr := handler.UpdateGlobalState(context.Background(), td.in)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestGetGlobalState(t *testing.T) {\n\n\tvalidInput := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n\n\tvalidCfg, _ := MapUpdateGlobalIsolationGroupsRequest(validInput)\n\n\tvalidCfgData := validCfg[0].Value.GetData()\n\tdynamicConfigResponse := []interface{}{}\n\tjson.Unmarshal(validCfgData, &dynamicConfigResponse)\n\n\ttests := map[string]struct {\n\t\tin           types.GetGlobalIsolationGroupsRequest\n\t\tdcAffordance func(client *dynamicconfig.MockClient)\n\t\texpected     *types.GetGlobalIsolationGroupsResponse\n\t\texpectedErr  error\n\t}{\n\t\t\"updating normal value\": {\n\t\t\tin: types.GetGlobalIsolationGroupsRequest{},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {\n\t\t\t\tclient.EXPECT().GetListValue(\n\t\t\t\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(dynamicConfigResponse, nil)\n\t\t\t},\n\t\t\texpected: &types.GetGlobalIsolationGroupsResponse{IsolationGroups: validInput},\n\t\t},\n\t\t\"an error was returned\": {\n\t\t\tin: types.GetGlobalIsolationGroupsRequest{},\n\t\t\tdcAffordance: func(client *dynamicconfig.MockClient) {\n\t\t\t\tclient.EXPECT().GetListValue(\n\t\t\t\t\tdynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, errors.New(\"an error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"failed to get global isolation groups from datastore: an error\"),\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tdcMock := dynamicconfig.NewMockClient(gomock.NewController(t))\n\t\t\ttd.dcAffordance(dcMock)\n\t\t\thandler := handlerImpl{\n\t\t\t\tlog:                        testlogger.New(t),\n\t\t\t\tglobalIsolationGroupDrains: dcMock,\n\t\t\t}\n\t\t\tres, err := handler.GetGlobalState(context.Background())\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tif td.expectedErr != nil {\n\t\t\t\t// only compare strings because full wrapping makes it fiddly otherwise\n\t\t\t\tassert.Equal(t, td.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype handlerImpl struct {\n\tlog                        log.Logger\n\tglobalIsolationGroupDrains dynamicconfig.Client\n\tdomainHandler              domain.Handler\n}\n\nfunc New(log log.Logger, igConfigStore dynamicconfig.Client, dh domain.Handler) Handler {\n\treturn &handlerImpl{\n\t\tlog:                        log,\n\t\tglobalIsolationGroupDrains: igConfigStore,\n\t\tdomainHandler:              dh,\n\t}\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination isolation_handler_mock.go -self_package github.com/uber/cadence/common/isolationgroup/isolationgrouphandler\n\ntype Handler interface {\n\t// GetGlobalState returns the current configuration of isolation-groups which apply to all domains\n\tGetGlobalState(ctx context.Context) (*types.GetGlobalIsolationGroupsResponse, error)\n\t// UpdateGlobalState updates isolation-groups which apply to all domains\n\tUpdateGlobalState(ctx context.Context, state types.UpdateGlobalIsolationGroupsRequest) error\n\t// GetDomainState is the read operation for getting the current state of a domain's isolation-groups\n\tGetDomainState(ctx context.Context, request types.GetDomainIsolationGroupsRequest) (*types.GetDomainIsolationGroupsResponse, error)\n\t// UpdateDomainState is the read operation for updating a domain's isolation-groups\n\tUpdateDomainState(ctx context.Context, state types.UpdateDomainIsolationGroupsRequest) error\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/isolation_handler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package isolationgroupapi -source interface.go -destination isolation_handler_mock.go -self_package github.com/uber/cadence/common/isolationgroup/isolationgrouphandler\n//\n\n// Package isolationgroupapi is a generated GoMock package.\npackage isolationgroupapi\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// GetDomainState mocks base method.\nfunc (m *MockHandler) GetDomainState(ctx context.Context, request types.GetDomainIsolationGroupsRequest) (*types.GetDomainIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainState\", ctx, request)\n\tret0, _ := ret[0].(*types.GetDomainIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainState indicates an expected call of GetDomainState.\nfunc (mr *MockHandlerMockRecorder) GetDomainState(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainState\", reflect.TypeOf((*MockHandler)(nil).GetDomainState), ctx, request)\n}\n\n// GetGlobalState mocks base method.\nfunc (m *MockHandler) GetGlobalState(ctx context.Context) (*types.GetGlobalIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetGlobalState\", ctx)\n\tret0, _ := ret[0].(*types.GetGlobalIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetGlobalState indicates an expected call of GetGlobalState.\nfunc (mr *MockHandlerMockRecorder) GetGlobalState(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetGlobalState\", reflect.TypeOf((*MockHandler)(nil).GetGlobalState), ctx)\n}\n\n// UpdateDomainState mocks base method.\nfunc (m *MockHandler) UpdateDomainState(ctx context.Context, state types.UpdateDomainIsolationGroupsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainState\", ctx, state)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomainState indicates an expected call of UpdateDomainState.\nfunc (mr *MockHandlerMockRecorder) UpdateDomainState(ctx, state any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainState\", reflect.TypeOf((*MockHandler)(nil).UpdateDomainState), ctx, state)\n}\n\n// UpdateGlobalState mocks base method.\nfunc (m *MockHandler) UpdateGlobalState(ctx context.Context, state types.UpdateGlobalIsolationGroupsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateGlobalState\", ctx, state)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateGlobalState indicates an expected call of UpdateGlobalState.\nfunc (mr *MockHandlerMockRecorder) UpdateGlobalState(ctx, state any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateGlobalState\", reflect.TypeOf((*MockHandler)(nil).UpdateGlobalState), ctx, state)\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/mappers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc MapDynamicConfigResponse(in []interface{}) (out types.IsolationGroupConfiguration, err error) {\n\tif in == nil {\n\t\treturn nil, nil\n\t}\n\n\tout = make(types.IsolationGroupConfiguration, len(in))\n\tfor _, v := range in {\n\t\tv1, ok := v.(map[string]interface{})\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(\"failed parse a dynamic config entry, %v, (got %v)\", v1, v)\n\t\t}\n\t\tn, okName := v1[\"Name\"]\n\t\ts, okState := v1[\"State\"]\n\t\tif !okState || !okName {\n\t\t\treturn nil, fmt.Errorf(\"failed parse a dynamic config entry, %v, (got %v)\", v1, v)\n\t\t}\n\t\tnS, okStr := n.(string)\n\t\tsI, okI := s.(float64)\n\t\tif !okStr || !okI {\n\t\t\treturn nil, fmt.Errorf(\"failed parse a dynamic config entry, %v, (got %v)\", v1, v)\n\t\t}\n\t\tout[nS] = types.IsolationGroupPartition{\n\t\t\tName:  nS,\n\t\t\tState: types.IsolationGroupState(sI),\n\t\t}\n\t}\n\treturn out, nil\n}\n\nfunc MapUpdateGlobalIsolationGroupsRequest(in types.IsolationGroupConfiguration) ([]*types.DynamicConfigValue, error) {\n\tjsonData, err := json.Marshal(in.ToPartitionList())\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to marshal input for dynamic config: %w\", err)\n\t}\n\tout := []*types.DynamicConfigValue{\n\t\t&types.DynamicConfigValue{\n\t\t\tValue: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         jsonData,\n\t\t\t},\n\t\t},\n\t}\n\treturn out, nil\n}\n\nfunc MapAllIsolationGroupsResponse(in []interface{}) ([]string, error) {\n\tvar allIsolationGroups []string\n\tfor k := range in {\n\t\tv, ok := in[k].(string)\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(\"failed to get all-isolation-groups response from dynamic config: got %v (%T)\", in[k], in[k])\n\t\t}\n\t\tallIsolationGroups = append(allIsolationGroups, v)\n\t}\n\treturn allIsolationGroups, nil\n}\n"
  },
  {
    "path": "common/isolationgroup/isolationgroupapi/mappers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage isolationgroupapi\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestMapAllIsolationGroupStates(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin          []interface{}\n\t\texpected    []string\n\t\texpectedErr error\n\t}{\n\t\t\"valid mapping\": {\n\t\t\tin:       []interface{}{\"zone-1\", \"zone-2\", \"zone-3\"},\n\t\t\texpected: []string{\"zone-1\", \"zone-2\", \"zone-3\"},\n\t\t},\n\t\t\"invalid mapping\": {\n\t\t\tin:          []interface{}{1, 2, 3},\n\t\t\texpectedErr: errors.New(\"failed to get all-isolation-groups response from dynamic config: got 1 (int)\"),\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres, err := MapAllIsolationGroupsResponse(td.in)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/json_task_token_serializer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nimport \"encoding/json\"\n\ntype (\n\tjsonTaskTokenSerializer struct{}\n)\n\n// NewJSONTaskTokenSerializer creates a new instance of TaskTokenSerializer\nfunc NewJSONTaskTokenSerializer() TaskTokenSerializer {\n\treturn &jsonTaskTokenSerializer{}\n}\n\nfunc (j *jsonTaskTokenSerializer) Serialize(token *TaskToken) ([]byte, error) {\n\tdata, err := json.Marshal(token)\n\n\treturn data, err\n}\n\nfunc (j *jsonTaskTokenSerializer) Deserialize(data []byte) (*TaskToken, error) {\n\tvar token TaskToken\n\terr := json.Unmarshal(data, &token)\n\n\treturn &token, err\n}\n\nfunc (j *jsonTaskTokenSerializer) SerializeQueryTaskToken(token *QueryTaskToken) ([]byte, error) {\n\tdata, err := json.Marshal(token)\n\n\treturn data, err\n}\n\nfunc (j *jsonTaskTokenSerializer) DeserializeQueryTaskToken(data []byte) (*QueryTaskToken, error) {\n\tvar token QueryTaskToken\n\terr := json.Unmarshal(data, &token)\n\n\treturn &token, err\n}\n"
  },
  {
    "path": "common/json_task_token_serializer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestJsonTaskTokenSerializer_Token_RoundTrip(t *testing.T) {\n\ttoken := TaskToken{\n\t\tDomainID:        \"test-domain\",\n\t\tWorkflowID:      \"test-workflow-id\",\n\t\tWorkflowType:    \"test-workflow-type\",\n\t\tRunID:           \"test-run-id\",\n\t\tScheduleID:      1,\n\t\tScheduleAttempt: 1,\n\t\tActivityID:      \"test-activity-id\",\n\t\tActivityType:    \"test-activity-type\",\n\t}\n\n\tserializer := NewJSONTaskTokenSerializer()\n\n\tserialized, err := serializer.Serialize(&token)\n\trequire.NoError(t, err)\n\tdeserializedToken, err := serializer.Deserialize(serialized)\n\trequire.NoError(t, err)\n\trequire.Equal(t, token, *deserializedToken)\n}\n\nfunc TestNewJSONTaskTokenSerializer_QueryToken_Roundtrip(t *testing.T) {\n\ttoken := QueryTaskToken{\n\t\tDomainID:   \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t\tTaskList:   \"test-task-list\",\n\t\tTaskID:     \"test-task-id\",\n\t}\n\tserializer := NewJSONTaskTokenSerializer()\n\n\tserialized, err := serializer.SerializeQueryTaskToken(&token)\n\trequire.NoError(t, err)\n\tdeserializedToken, err := serializer.DeserializeQueryTaskToken(serialized)\n\trequire.NoError(t, err)\n\trequire.Equal(t, token, *deserializedToken)\n}\n"
  },
  {
    "path": "common/locks/lock.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage locks\n\nimport (\n\t\"context\"\n\t\"sync\"\n)\n\ntype (\n\t// Mutex accepts a context in its Lock method.\n\t// It blocks the goroutine until either the lock is acquired or the context\n\t// is closed.\n\tMutex interface {\n\t\tLock(context.Context) error\n\t\tUnlock()\n\t}\n)\n\ntype impl struct {\n\tch chan struct{}\n}\n\nfunc NewMutex() Mutex {\n\tch := make(chan struct{}, 1)\n\tch <- struct{}{}\n\treturn &impl{\n\t\tch: ch,\n\t}\n}\n\nfunc (m *impl) Lock(ctx context.Context) error {\n\tselect {\n\tcase <-m.ch:\n\t\treturn nil\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\t}\n}\n\nfunc (m *impl) Unlock() {\n\tselect {\n\tcase m.ch <- struct{}{}:\n\tdefault:\n\t\t// reaching this branch means the mutex was not locked.\n\t\t// this will intentionally crash the process, and print stack traces.\n\t\t//\n\t\t// it's not really possible to do this by hand (`fatal` is private),\n\t\t// and other common options like `log.Fatal` don't print stacks / don't\n\t\t// print all stacks / have loads of minor inconsistencies.\n\t\t//\n\t\t// regardless, what we want is to mimic mutexes when wrongly unlocked,\n\t\t// so just use the mutex implementation to guarantee it's the same.\n\t\tvar m sync.Mutex\n\t\tm.Unlock()\n\t}\n}\n"
  },
  {
    "path": "common/locks/lock_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage locks\n\nimport (\n\t\"context\"\n\t\"os\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n)\n\nfunc TestBasicLocking(t *testing.T) {\n\tlock := NewMutex()\n\terr1 := lock.Lock(context.Background())\n\trequire.Nil(t, err1)\n\tlock.Unlock()\n}\n\nfunc TestExpiredContext(t *testing.T) {\n\tlock := NewMutex()\n\terr1 := lock.Lock(context.Background())\n\trequire.Nil(t, err1)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Minute)\n\tcancel()\n\terr2 := lock.Lock(ctx)\n\trequire.NotNil(t, err2)\n\trequire.Equal(t, err2, ctx.Err())\n\n\tlock.Unlock()\n\n\terr3 := lock.Lock(context.Background())\n\trequire.Nil(t, err3)\n\tlock.Unlock()\n}\n\nfunc BenchmarkLock(b *testing.B) {\n\tl := NewMutex()\n\tctx := context.Background()\n\tfor n := 0; n < b.N; n++ {\n\t\t_ = l.Lock(ctx)\n\t\tl.Unlock()\n\t}\n}\n\nfunc BenchmarkStdlibLock(b *testing.B) {\n\tvar m sync.Mutex\n\tfor n := 0; n < b.N; n++ {\n\t\tm.Lock()\n\t\tm.Unlock()\n\t}\n}\n\nfunc BenchmarkParallelLock(b *testing.B) {\n\tl := NewMutex()\n\tctx := context.Background()\n\tb.RunParallel(func(pb *testing.PB) {\n\t\tfor pb.Next() {\n\t\t\t_ = l.Lock(ctx)\n\t\t\tl.Unlock()\n\t\t}\n\t})\n}\n\nfunc BenchmarkParallelStdlibLock(b *testing.B) {\n\tvar m sync.Mutex\n\tb.RunParallel(func(pb *testing.PB) {\n\t\tfor pb.Next() {\n\t\t\tm.Lock()\n\t\t\tm.Unlock()\n\t\t}\n\t})\n}\n\nfunc TestShouldBeFatal(t *testing.T) {\n\tt.Run(\"stdlib\", func(t *testing.T) {\n\t\tfatalTest(t)\n\t\tvar m sync.Mutex\n\t\tm.Unlock()\n\t})\n\tt.Run(\"custom\", func(t *testing.T) {\n\t\tfatalTest(t)\n\t\tNewMutex().Unlock()\n\t})\n}\n\nfunc fatalTest(t *testing.T) {\n\tt.Helper()\n\tif os.Getenv(\"RUN_FATAL_TESTS\") != \"true\" {\n\t\tt.Skipf(\n\t\t\t\"This test intentionally crashes the process, and cannot be tested normally.\\n\"+\n\t\t\t\t\"Check output manually when running this test on its own:\\n\"+\n\t\t\t\t\"\\tRUN_FATAL_TESTS=true go test -test.run %q %v\",\n\t\t\tt.Name(), \"github.com/uber/cadence/common/locks\",\n\t\t)\n\t} else {\n\t\tt.Cleanup(func() {\n\t\t\tt.Fatal(\"process should have crashed, and this should be unreachable\")\n\t\t})\n\t}\n}\n\nfunc TestConcurrently(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tconst (\n\t\ttryDuration = time.Millisecond // long enough to ensure at least one acquire\n\t\ttries       = 1000\n\t)\n\n\t// attempt to cancel all attempts within a reasonable amount of time\n\tctx, cancel := context.WithTimeout(context.Background(), tryDuration*tries*10)\n\tdefer cancel()\n\tsuccess, err := int32(0), int32(0)\n\n\tm := NewMutex()\n\tvar wg sync.WaitGroup\n\tfor i := 0; i < tries; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tctx, cancel := context.WithTimeout(ctx, tryDuration)\n\t\t\tdefer cancel()\n\t\t\tif m.Lock(ctx) == nil {\n\t\t\t\tatomic.AddInt32(&success, 1)\n\t\t\t\ttime.Sleep(tryDuration / 10) // sleep for a fraction of the time, to guarantee some timeouts\n\t\t\t\tm.Unlock()\n\t\t\t} else {\n\t\t\t\tatomic.AddInt32(&err, 1)\n\t\t\t}\n\t\t}()\n\t}\n\twg.Wait()\n\n\tt.Logf(\"success: %v, err: %v\", success, err)\n\tassert.Greater(t, success, int32(1), \"should have successfully acquired more than once, as once could be a fluke\")\n\tassert.NotZero(t, err, \"should have timed out at least once\")\n}\n"
  },
  {
    "path": "common/log/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport \"github.com/uber/cadence/common/log/tag\"\n\n//go:generate mockgen -package=$GOPACKAGE -destination=logger_mock.go -self_package=github.com/uber/cadence/common/log github.com/uber/cadence/common/log Logger\n\n// Logger is our abstraction for logging\n// Usage examples:\n//\n//\t import \"github.com/uber/cadence/common/log/tag\"\n//\t 1) logger = logger.WithTags(\n//\t         tag.WorkflowNextEventID( 123),\n//\t         tag.WorkflowActionWorkflowStarted,\n//\t         tag.WorkflowDomainID(\"test-domain-id\"))\n//\t    logger.Info(\"hello world\")\n//\t 2) logger.Info(\"hello world\",\n//\t         tag.WorkflowNextEventID( 123),\n//\t         tag.WorkflowActionWorkflowStarted,\n//\t         tag.WorkflowDomainID(\"test-domain-id\"))\n//\t\t   )\n//\t Note: msg should be static, it is not recommended to use fmt.Sprintf() for msg.\n//\t       Anything dynamic should be tagged.\ntype Logger interface {\n\tDebugf(msg string, args ...any)\n\tDebug(msg string, tags ...tag.Tag)\n\tInfo(msg string, tags ...tag.Tag)\n\tWarn(msg string, tags ...tag.Tag)\n\tError(msg string, tags ...tag.Tag)\n\tFatal(msg string, tags ...tag.Tag)\n\tWithTags(tags ...tag.Tag) Logger\n\tSampleInfo(msg string, sampleRate int, tags ...tag.Tag)\n\tDebugOn() bool\n\t// Helper returns a logger that will skip one more level in stack trace. This is helpful for layered architecture, when you want to point to a business logic error, instead of pointing to the wrapped generated level.\n\tHelper() Logger\n}\n\ntype noop struct{}\n\n// NewNoop return a noop logger\nfunc NewNoop() Logger {\n\treturn &noop{}\n}\n\nfunc (n *noop) Debugf(msg string, args ...any)                         {}\nfunc (n *noop) Debug(msg string, tags ...tag.Tag)                      {}\nfunc (n *noop) Info(msg string, tags ...tag.Tag)                       {}\nfunc (n *noop) Warn(msg string, tags ...tag.Tag)                       {}\nfunc (n *noop) Error(msg string, tags ...tag.Tag)                      {}\nfunc (n *noop) Fatal(msg string, tags ...tag.Tag)                      {}\nfunc (n *noop) SampleInfo(msg string, sampleRate int, tags ...tag.Tag) {}\nfunc (n *noop) WithTags(tags ...tag.Tag) Logger {\n\treturn n\n}\nfunc (n *noop) DebugOn() bool {\n\treturn true\n}\nfunc (n *noop) Helper() Logger { return &noop{} }\n"
  },
  {
    "path": "common/log/logfx/fx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage logfx\n\nimport (\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\n// Module provides fx options to initialize logger from configuration.\nvar Module = fx.Options(\n\tfx.Provide(zapBuilder),\n\tModuleWithoutZap,\n)\n\n// ModuleWithoutZap provides fx options to initialize logger from existing zap logger, which might be provided outside.\n// This is useful for monorepos or any application that uses centralize log configuration like Uber monorepo.\nvar ModuleWithoutZap = fx.Options(\n\tfx.Provide(log.NewLogger))\n\ntype zapBuilderParams struct {\n\tfx.In\n\n\tCfg config.Config\n}\n\nfunc zapBuilder(p zapBuilderParams) (*zap.Logger, error) {\n\tlogger, err := p.Cfg.Log.NewZapLogger()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn logger, nil\n}\n"
  },
  {
    "path": "common/log/logfx/fx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage logfx\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nfunc TestLogFx(t *testing.T) {\n\tapp := fxtest.New(t,\n\t\tfx.Provide(func() config.Config {\n\t\t\treturn config.Config{}\n\t\t}),\n\t\tModule,\n\t\tfx.Invoke(func(logger log.Logger) {\n\t\t\tlogger.Info(\"hello world\")\n\t\t}),\n\t)\n\tapp.RequireStart().RequireStop()\n}\n\nfunc TestLogFxWithExternalZapLogger(t *testing.T) {\n\tapp := fxtest.New(t,\n\t\tfx.Provide(func() *zap.Logger {\n\t\t\treturn zaptest.NewLogger(t)\n\t\t}),\n\t\tModuleWithoutZap,\n\t\tfx.Invoke(func(logger log.Logger) {\n\t\t\tlogger.Info(\"hello world\")\n\t\t}))\n\tapp.RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "common/log/logger.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\t\"path/filepath\"\n\t\"runtime\"\n\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype loggerImpl struct {\n\tzapLogger     *zap.Logger\n\tskip          int\n\tsampleLocalFn func(int) bool\n}\n\nconst (\n\tskipForDefaultLogger = 3\n\t// we put a default message when it is empty so that the log can be searchable/filterable\n\tdefaultMsgForEmpty = \"none\"\n)\n\nvar defaultSampleFn = func(i int) bool { return rand.Intn(i) == 0 }\n\n// NewLogger returns a new logger\nfunc NewLogger(zapLogger *zap.Logger, opts ...Option) Logger {\n\timpl := &loggerImpl{\n\t\t// Skip wrapped logger from the stack trace so that zap logs will point to the caller.\n\t\tzapLogger:     zapLogger.WithOptions(zap.AddCallerSkip(1)),\n\t\tskip:          skipForDefaultLogger,\n\t\tsampleLocalFn: defaultSampleFn,\n\t}\n\tfor _, opt := range opts {\n\t\topt(impl)\n\t}\n\n\treturn impl\n}\n\nfunc caller(skip int) string {\n\t_, path, lineno, ok := runtime.Caller(skip)\n\tif !ok {\n\t\treturn \"\"\n\t}\n\treturn fmt.Sprintf(\"%v:%v\", filepath.Base(path), lineno)\n}\n\nfunc (lg *loggerImpl) buildFieldsWithCallat(tags []tag.Tag) []zap.Field {\n\tfs := lg.buildFields(tags)\n\tfs = append(fs, zap.String(tag.LoggingCallAtKey, caller(lg.skip)))\n\treturn fs\n}\n\nfunc (lg *loggerImpl) buildFields(tags []tag.Tag) []zap.Field {\n\tfs := make([]zap.Field, 0, len(tags))\n\tfor _, t := range tags {\n\t\tf := t.Field()\n\t\tif f.Key == \"\" {\n\t\t\t// ignore empty field(which can be constructed manually)\n\t\t\tcontinue\n\t\t}\n\t\tfs = append(fs, f)\n\n\t\tif obj, ok := f.Interface.(zapcore.ObjectMarshaler); ok && f.Type == zapcore.ErrorType {\n\t\t\tfs = append(fs, zap.Object(f.Key+\"-details\", obj))\n\t\t}\n\t}\n\treturn fs\n}\n\nfunc setDefaultMsg(msg string) string {\n\tif msg == \"\" {\n\t\treturn defaultMsgForEmpty\n\t}\n\treturn msg\n}\n\nfunc (lg *loggerImpl) Debugf(msg string, args ...any) {\n\tif !lg.DebugOn() {\n\t\treturn\n\t}\n\n\tfields := lg.buildFieldsWithCallat(nil)\n\tlg.zapLogger.Debug(setDefaultMsg(fmt.Sprintf(msg, args...)), fields...)\n}\n\nfunc (lg *loggerImpl) Debug(msg string, tags ...tag.Tag) {\n\tif !lg.DebugOn() {\n\t\treturn\n\t}\n\n\tfields := lg.buildFieldsWithCallat(tags)\n\tlg.zapLogger.Debug(msg, fields...)\n}\n\nfunc (lg *loggerImpl) Info(msg string, tags ...tag.Tag) {\n\tmsg = setDefaultMsg(msg)\n\tfields := lg.buildFieldsWithCallat(tags)\n\tlg.zapLogger.Info(msg, fields...)\n}\n\nfunc (lg *loggerImpl) Warn(msg string, tags ...tag.Tag) {\n\tmsg = setDefaultMsg(msg)\n\tfields := lg.buildFieldsWithCallat(tags)\n\tlg.zapLogger.Warn(msg, fields...)\n}\n\nfunc (lg *loggerImpl) Error(msg string, tags ...tag.Tag) {\n\tmsg = setDefaultMsg(msg)\n\tfields := lg.buildFieldsWithCallat(tags)\n\tlg.zapLogger.Error(msg, fields...)\n}\n\nfunc (lg *loggerImpl) Fatal(msg string, tags ...tag.Tag) {\n\tmsg = setDefaultMsg(msg)\n\tfields := lg.buildFieldsWithCallat(tags)\n\tlg.zapLogger.Fatal(msg, fields...)\n}\n\nfunc (lg *loggerImpl) WithTags(tags ...tag.Tag) Logger {\n\tfields := lg.buildFields(tags)\n\tzapLogger := lg.zapLogger.With(fields...)\n\treturn &loggerImpl{\n\t\tzapLogger:     zapLogger,\n\t\tskip:          lg.skip,\n\t\tsampleLocalFn: lg.sampleLocalFn,\n\t}\n}\n\n// DebugOn checks if debug level is on.\nfunc (lg *loggerImpl) DebugOn() bool {\n\treturn lg.zapLogger.Check(zap.DebugLevel, \"test\") != nil\n}\n\nfunc (lg *loggerImpl) SampleInfo(msg string, sampleRate int, tags ...tag.Tag) {\n\tif lg.sampleLocalFn(sampleRate) {\n\t\tmsg = setDefaultMsg(msg)\n\t\tfields := lg.buildFieldsWithCallat(tags)\n\t\tlg.zapLogger.Info(msg, fields...)\n\t}\n}\n\nfunc (lg *loggerImpl) Helper() Logger {\n\treturn &loggerImpl{\n\t\t// AddCallerSkip increments the value so we don't need to track previous level.\n\t\tzapLogger:     lg.zapLogger.WithOptions(zap.AddCallerSkip(1)),\n\t\tskip:          lg.skip,\n\t\tsampleLocalFn: lg.sampleLocalFn,\n\t}\n}\n"
  },
  {
    "path": "common/log/logger_bench_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\npackage log\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n/**\nlonger@~/gocode/src/github.com/uber/cadence/common/log:(log)$ go test -v -bench=. | egrep \"(Bench)|(ns/op)\"\nBenchmarkZapLoggerWithFields-8      \t{\"level\":\"info\",\"ts\":1555094254.794006,\"caller\":\"log/logger_bench_test.go:21\",\"msg\":\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\"wf-schedule-id\":0,\"cluster-name\":\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\",\"wf-domain-name\":\"test-domain-name\"}\n  200000\t      8609 ns/op\nBenchmarkLoggerWithFields-8         \t{\"level\":\"info\",\"ts\":1555094256.608516,\"msg\":\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\"wf-schedule-id\":0,\"cluster-name\":\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\",\"wf-domain-name\":\"test-domain-name\",\"logging-call-at\":\"logger_bench_test.go:36\"}\n  200000\t      8773 ns/op\nBenchmarkZapLoggerWithoutFields-8   \t{\"level\":\"info\",\"ts\":1555094258.4521542,\"caller\":\"log/logger_bench_test.go:49\",\"msg\":\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\"wf-schedule-id\":0,\"cluster-name\":\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\",\"wf-domain-name\":\"test-domain-name\"}\n  200000\t      8535 ns/op\nBenchmarkLoggerWithoutFields-8      \t{\"level\":\"info\",\"ts\":1555094260.2499342,\"msg\":\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\"wf-domain-name\":\"test-domain-name\",\"wf-schedule-id\":0,\"cluster-name\":\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\",\"logging-call-at\":\"logger_bench_test.go:64\"}\n  200000\t      8998 ns/op\n*/\n\nfunc BenchmarkZapLoggerWithFields(b *testing.B) {\n\tzapLogger, err := buildZapLoggerForZap()\n\tif err != nil {\n\t\tb.Fail()\n\t}\n\n\tfor i := 0; i < b.N; i++ {\n\t\tlg := zapLogger.With(zap.Int64(\"wf-schedule-id\", int64(i)), zap.String(\"cluster-name\", \"this is a very long value: 1234567890 1234567890 1234567890 1234567890\"))\n\t\tlg.Info(\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\n\t\t\tzap.String(\"wf-domain-name\", \"test-domain-name\"))\n\t}\n\n}\n\nfunc BenchmarkLoggerWithFields(b *testing.B) {\n\tzapLogger, err := buildZapLoggerForDefault()\n\tif err != nil {\n\t\tb.Fail()\n\t}\n\tlogger := NewLogger(zapLogger)\n\n\tfor i := 0; i < b.N; i++ {\n\t\tlg := logger.WithTags(tag.WorkflowScheduleID(int64(i)), tag.ClusterName(\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\"))\n\t\tlg.Info(\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\n\t\t\ttag.WorkflowDomainName(\"test-domain-name\"))\n\t}\n\n}\n\nfunc BenchmarkZapLoggerWithoutFields(b *testing.B) {\n\tzapLogger, err := buildZapLoggerForZap()\n\tif err != nil {\n\t\tb.Fail()\n\t}\n\n\tfor i := 0; i < b.N; i++ {\n\t\tzapLogger.Info(\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\n\t\t\tzap.Int64(\"wf-schedule-id\", int64(i)), zap.String(\"cluster-name\", \"this is a very long value: 1234567890 1234567890 1234567890 1234567890\"),\n\t\t\tzap.String(\"wf-domain-name\", \"test-domain-name\"))\n\t}\n\n}\n\nfunc BenchmarkLoggerWithoutFields(b *testing.B) {\n\tzapLogger, err := buildZapLoggerForDefault()\n\tif err != nil {\n\t\tb.Fail()\n\t}\n\tlogger := NewLogger(zapLogger)\n\n\tfor i := 0; i < b.N; i++ {\n\t\tlogger.Info(\"msg to print log, 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890 1234567890\",\n\t\t\ttag.WorkflowDomainName(\"test-domain-name\"),\n\t\t\ttag.WorkflowScheduleID(int64(i)), tag.ClusterName(\"this is a very long value: 1234567890 1234567890 1234567890 1234567890\"))\n\t}\n}\n\nfunc buildZapLoggerForZap() (*zap.Logger, error) {\n\tencConfig := zap.NewProductionEncoderConfig()\n\tcfg := zap.Config{\n\t\tLevel:            zap.NewAtomicLevelAt(zap.InfoLevel),\n\t\tDevelopment:      false,\n\t\tSampling:         nil,\n\t\tEncoding:         \"json\",\n\t\tEncoderConfig:    encConfig,\n\t\tOutputPaths:      []string{\"stderr\"},\n\t\tErrorOutputPaths: []string{\"stderr\"},\n\t}\n\treturn cfg.Build()\n}\n\n// only difference here is we disable caller since we already have another one\nfunc buildZapLoggerForDefault() (*zap.Logger, error) {\n\tencConfig := zapcore.EncoderConfig{\n\t\tTimeKey:        \"ts\",\n\t\tLevelKey:       \"level\",\n\t\tNameKey:        \"logger\",\n\t\tCallerKey:      \"\", // disable caller here since we already have another one\n\t\tMessageKey:     \"msg\",\n\t\tStacktraceKey:  \"stacktrace\",\n\t\tLineEnding:     zapcore.DefaultLineEnding,\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.EpochTimeEncoder,\n\t\tEncodeDuration: zapcore.SecondsDurationEncoder,\n\t\tEncodeCaller:   nil,\n\t}\n\tcfg := zap.Config{\n\t\tLevel:            zap.NewAtomicLevelAt(zap.InfoLevel),\n\t\tDevelopment:      false,\n\t\tSampling:         nil,\n\t\tEncoding:         \"json\",\n\t\tEncoderConfig:    encConfig,\n\t\tOutputPaths:      []string{\"stderr\"},\n\t\tErrorOutputPaths: []string{\"stderr\"},\n\t}\n\treturn cfg.Build()\n}\n"
  },
  {
    "path": "common/log/logger_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/log (interfaces: Logger)\n//\n// Generated by this command:\n//\n//\tmockgen -package=log -destination=logger_mock.go -self_package=github.com/uber/cadence/common/log github.com/uber/cadence/common/log Logger\n//\n\n// Package log is a generated GoMock package.\npackage log\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttag \"github.com/uber/cadence/common/log/tag\"\n)\n\n// MockLogger is a mock of Logger interface.\ntype MockLogger struct {\n\tctrl     *gomock.Controller\n\trecorder *MockLoggerMockRecorder\n\tisgomock struct{}\n}\n\n// MockLoggerMockRecorder is the mock recorder for MockLogger.\ntype MockLoggerMockRecorder struct {\n\tmock *MockLogger\n}\n\n// NewMockLogger creates a new mock instance.\nfunc NewMockLogger(ctrl *gomock.Controller) *MockLogger {\n\tmock := &MockLogger{ctrl: ctrl}\n\tmock.recorder = &MockLoggerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockLogger) EXPECT() *MockLoggerMockRecorder {\n\treturn m.recorder\n}\n\n// Debug mocks base method.\nfunc (m *MockLogger) Debug(msg string, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Debug\", varargs...)\n}\n\n// Debug indicates an expected call of Debug.\nfunc (mr *MockLoggerMockRecorder) Debug(msg any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Debug\", reflect.TypeOf((*MockLogger)(nil).Debug), varargs...)\n}\n\n// DebugOn mocks base method.\nfunc (m *MockLogger) DebugOn() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DebugOn\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// DebugOn indicates an expected call of DebugOn.\nfunc (mr *MockLoggerMockRecorder) DebugOn() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DebugOn\", reflect.TypeOf((*MockLogger)(nil).DebugOn))\n}\n\n// Debugf mocks base method.\nfunc (m *MockLogger) Debugf(msg string, args ...any) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Debugf\", varargs...)\n}\n\n// Debugf indicates an expected call of Debugf.\nfunc (mr *MockLoggerMockRecorder) Debugf(msg any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Debugf\", reflect.TypeOf((*MockLogger)(nil).Debugf), varargs...)\n}\n\n// Error mocks base method.\nfunc (m *MockLogger) Error(msg string, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Error\", varargs...)\n}\n\n// Error indicates an expected call of Error.\nfunc (mr *MockLoggerMockRecorder) Error(msg any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Error\", reflect.TypeOf((*MockLogger)(nil).Error), varargs...)\n}\n\n// Fatal mocks base method.\nfunc (m *MockLogger) Fatal(msg string, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Fatal\", varargs...)\n}\n\n// Fatal indicates an expected call of Fatal.\nfunc (mr *MockLoggerMockRecorder) Fatal(msg any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Fatal\", reflect.TypeOf((*MockLogger)(nil).Fatal), varargs...)\n}\n\n// Helper mocks base method.\nfunc (m *MockLogger) Helper() Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Helper\")\n\tret0, _ := ret[0].(Logger)\n\treturn ret0\n}\n\n// Helper indicates an expected call of Helper.\nfunc (mr *MockLoggerMockRecorder) Helper() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Helper\", reflect.TypeOf((*MockLogger)(nil).Helper))\n}\n\n// Info mocks base method.\nfunc (m *MockLogger) Info(msg string, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Info\", varargs...)\n}\n\n// Info indicates an expected call of Info.\nfunc (mr *MockLoggerMockRecorder) Info(msg any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Info\", reflect.TypeOf((*MockLogger)(nil).Info), varargs...)\n}\n\n// SampleInfo mocks base method.\nfunc (m *MockLogger) SampleInfo(msg string, sampleRate int, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg, sampleRate}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"SampleInfo\", varargs...)\n}\n\n// SampleInfo indicates an expected call of SampleInfo.\nfunc (mr *MockLoggerMockRecorder) SampleInfo(msg, sampleRate any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg, sampleRate}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SampleInfo\", reflect.TypeOf((*MockLogger)(nil).SampleInfo), varargs...)\n}\n\n// Warn mocks base method.\nfunc (m *MockLogger) Warn(msg string, tags ...tag.Tag) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{msg}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Warn\", varargs...)\n}\n\n// Warn indicates an expected call of Warn.\nfunc (mr *MockLoggerMockRecorder) Warn(msg any, tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{msg}, tags...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Warn\", reflect.TypeOf((*MockLogger)(nil).Warn), varargs...)\n}\n\n// WithTags mocks base method.\nfunc (m *MockLogger) WithTags(tags ...tag.Tag) Logger {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range tags {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"WithTags\", varargs...)\n\tret0, _ := ret[0].(Logger)\n\treturn ret0\n}\n\n// WithTags indicates an expected call of WithTags.\nfunc (mr *MockLoggerMockRecorder) WithTags(tags ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WithTags\", reflect.TypeOf((*MockLogger)(nil).WithTags), tags...)\n}\n"
  },
  {
    "path": "common/log/logger_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"runtime\"\n\t\"strconv\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\nconst anyNum = \"[0-9]+\"\n\nfunc TestLoggers(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname          string\n\t\tloggerFactory func(zapLogger *zap.Logger) Logger\n\t}{\n\t\t{\n\t\t\tname: \"normal\",\n\t\t\tloggerFactory: func(zapLogger *zap.Logger) Logger {\n\t\t\t\treturn NewLogger(zapLogger)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"throttled\",\n\t\t\tloggerFactory: func(zapLogger *zap.Logger) Logger {\n\t\t\t\treturn NewThrottledLogger(NewLogger(zapLogger), dynamicproperties.GetIntPropertyFn(1))\n\t\t\t},\n\t\t},\n\t\t// Unfortunately, replay logger is impossible to test because it requires a workflow context, which is not exposed by go client.\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tfor _, scenario := range []struct {\n\t\t\t\tname     string\n\t\t\t\tfn       func(logger Logger)\n\t\t\t\texpected string\n\t\t\t}{\n\t\t\t\t{\n\t\t\t\t\tname: \"debug\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Debug(\"test debug\")\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"debug\",\"msg\":\"test debug\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"info\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Info(\"test info\")\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"info\",\"msg\":\"test info\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"warn\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Warn(\"test warn\")\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"warn\",\"msg\":\"test warn\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"error\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Error(\"test error\")\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"error\",\"msg\":\"test error\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"debugf\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Debugf(\"test %v\", \"debug\")\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"debug\",\"msg\":\"test debug\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"with tags\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.WithTags(tag.Error(fmt.Errorf(\"test error\"))).Info(\"test info\", tag.WorkflowActionWorkflowStarted)\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"info\",\"msg\":\"test info\",\"error\":\"test error\",\"wf-action\":\"add-workflow-started-event\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"empty message\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Info(\"\", tag.WorkflowActionWorkflowStarted)\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"info\",\"msg\":\"` + defaultMsgForEmpty + `\",\"wf-action\":\"add-workflow-started-event\",\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"error with details\",\n\t\t\t\t\tfn: func(logger Logger) {\n\t\t\t\t\t\tlogger.Error(\"oh no\", tag.Error(&testError{\"workflow123\"}))\n\t\t\t\t\t},\n\t\t\t\t\texpected: `{\"level\":\"error\",\"msg\":\"oh no\",\"error\":\"test error\",\"error-details\":{\"workflow-id\":\"workflow123\"},\"logging-call-at\":\"logger_test.go:%s\"}`,\n\t\t\t\t},\n\t\t\t} {\n\t\t\t\tt.Run(scenario.name, func(t *testing.T) {\n\t\t\t\t\tbuf := &strings.Builder{}\n\t\t\t\t\tzapLogger := zap.New(zapcore.NewCore(zapcore.NewJSONEncoder(zapcore.EncoderConfig{\n\t\t\t\t\t\tMessageKey:     \"msg\",\n\t\t\t\t\t\tLevelKey:       \"level\",\n\t\t\t\t\t\tNameKey:        \"logger\",\n\t\t\t\t\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\t\t\t\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\t\t\t\t\tEncodeDuration: zapcore.StringDurationEncoder,\n\t\t\t\t\t}), zapcore.AddSync(buf), zap.DebugLevel))\n\t\t\t\t\tlogger := tc.loggerFactory(zapLogger)\n\t\t\t\t\tscenario.fn(logger)\n\t\t\t\t\tout := strings.TrimRight(buf.String(), \"\\n\")\n\t\t\t\t\texpected := fmt.Sprintf(scenario.expected, anyNum)\n\t\t\t\t\tassert.Regexp(t, expected, out)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLoggerCallAt(t *testing.T) {\n\tbuf := &strings.Builder{}\n\tzapLogger := zap.New(zapcore.NewCore(zapcore.NewJSONEncoder(zapcore.EncoderConfig{\n\t\tMessageKey:     \"msg\",\n\t\tLevelKey:       \"level\",\n\t\tNameKey:        \"logger\",\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\tEncodeDuration: zapcore.StringDurationEncoder,\n\t}), zapcore.AddSync(buf), zap.DebugLevel))\n\n\tlogger := NewLogger(zapLogger)\n\tpreCaller := caller(1)\n\tlogger.WithTags(tag.Error(fmt.Errorf(\"test error\"))).Info(\"test info\", tag.WorkflowActionWorkflowStarted)\n\n\tout := strings.TrimRight(buf.String(), \"\\n\")\n\tsps := strings.Split(preCaller, \":\")\n\tpar, err := strconv.Atoi(sps[1])\n\tassert.Nil(t, err)\n\tlineNum := fmt.Sprintf(\"%v\", par+1)\n\tassert.Equal(t, `{\"level\":\"info\",\"msg\":\"test info\",\"error\":\"test error\",\"wf-action\":\"add-workflow-started-event\",\"logging-call-at\":\"logger_test.go:`+lineNum+`\"}`, out)\n\n}\n\nfunc TestLogger_Sampled(t *testing.T) {\n\tbuf := &strings.Builder{}\n\tzapLogger := zap.New(zapcore.NewCore(zapcore.NewJSONEncoder(zapcore.EncoderConfig{\n\t\tMessageKey:     \"msg\",\n\t\tLevelKey:       \"level\",\n\t\tNameKey:        \"logger\",\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\tEncodeDuration: zapcore.StringDurationEncoder,\n\t}), zapcore.AddSync(buf), zap.DebugLevel))\n\tlogger := NewLogger(zapLogger, WithSampleFunc(func(i int) bool {\n\t\treturn true\n\t}))\n\n\t// No Assertion here, since default logger uses random function.\n\t// But we check that it is\n\tlogger.SampleInfo(\"test info\", 1, tag.WorkflowActionWorkflowStarted)\n\tout := strings.TrimRight(buf.String(), \"\\n\")\n\tassert.Regexp(t, `{\"level\":\"info\",\"msg\":\"test info\",\"wf-action\":\"add-workflow-started-event\",\"logging-call-at\":\"logger_test.go:`+anyNum+`\"}`, out)\n}\n\nfunc TestDebugOn(t *testing.T) {\n\tbuf := &strings.Builder{}\n\tl := zap.NewAtomicLevel()\n\tzapLogger := zap.New(zapcore.NewCore(zapcore.NewJSONEncoder(zapcore.EncoderConfig{\n\t\tMessageKey:     \"msg\",\n\t\tLevelKey:       \"level\",\n\t\tNameKey:        \"logger\",\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\tEncodeDuration: zapcore.StringDurationEncoder,\n\t}), zapcore.AddSync(buf), l))\n\n\tlogger := NewLogger(zapLogger)\n\n\t// Set level to debug and check if debugOn is true\n\tl.SetLevel(zap.DebugLevel)\n\tassert.True(t, logger.DebugOn())\n\n\t// Set level to info and check if debugOn is false\n\tl.SetLevel(zap.InfoLevel)\n\tassert.False(t, logger.DebugOn())\n\n\t// Set level to debug again and check if debugOn is true\n\tl.SetLevel(zap.DebugLevel)\n\tassert.True(t, logger.DebugOn())\n}\n\nfunc TestHelper(t *testing.T) {\n\t// Test without Helper() first\n\tt.Run(\"WithoutHelper\", func(t *testing.T) {\n\t\tbuf := &strings.Builder{}\n\t\tzapLogger := createTestZapLogger(buf)\n\t\tlogger := NewLogger(zapLogger)\n\n\t\texpectedLine := helperFuncWithoutHelper(logger)\n\n\t\tout := strings.TrimRight(buf.String(), \"\\n\")\n\n\t\tvar logEntry map[string]interface{}\n\t\tif err := json.Unmarshal([]byte(out), &logEntry); err != nil {\n\t\t\tt.Fatalf(\"Failed to parse log output as JSON: %v\", err)\n\t\t}\n\n\t\tlogCaller, ok := logEntry[\"caller\"].(string)\n\t\tif !ok {\n\t\t\tt.Fatal(\"Log entry does not contain caller information\")\n\t\t}\n\n\t\trequire.Contains(t, logCaller, \"logger_test.go\")\n\n\t\t// Get the line from the caller string\n\t\tparts := strings.Split(logCaller, \":\")\n\t\tif len(parts) != 2 {\n\t\t\tt.Fatalf(\"Unexpected caller format: %s\", logCaller)\n\t\t}\n\n\t\t// Extract line number\n\t\tlineStr := parts[1]\n\t\tlineNum, err := strconv.Atoi(lineStr)\n\t\tif err != nil {\n\t\t\tt.Fatalf(\"Failed to parse line number from %s: %v\", logCaller, err)\n\t\t}\n\n\t\t// With Helper(), the caller should be near the line where helper was called\n\t\t// Allow a small range of lines to avoid brittle tests\n\t\tif lineNum < expectedLine {\n\t\t\tt.Errorf(\"Caller line number (%d) is not near the calling line (%d)\", lineNum, expectedLine)\n\t\t}\n\t})\n\n\t// Test with Helper()\n\tt.Run(\"WithHelper\", func(t *testing.T) {\n\t\tbuf := &strings.Builder{}\n\t\tzapLogger := createTestZapLogger(buf)\n\t\tlogger := NewLogger(zapLogger)\n\n\t\t// Line number where the helper is called from the test\n\t\tcallerLine := line() // Get current line number\n\t\thelperFuncWithHelper(logger)\n\n\t\tout := strings.TrimRight(buf.String(), \"\\n\")\n\n\t\tvar logEntry map[string]interface{}\n\t\tif err := json.Unmarshal([]byte(out), &logEntry); err != nil {\n\t\t\tt.Fatalf(\"Failed to parse log output as JSON: %v\", err)\n\t\t}\n\n\t\tlogCaller, ok := logEntry[\"caller\"].(string)\n\t\tif !ok {\n\t\t\tt.Fatal(\"Log entry does not contain caller information\")\n\t\t}\n\n\t\trequire.Contains(t, logCaller, \"logger_test.go\")\n\n\t\t// Get the line from the caller string\n\t\tparts := strings.Split(logCaller, \":\")\n\t\tif len(parts) != 2 {\n\t\t\tt.Fatalf(\"Unexpected caller format: %s\", logCaller)\n\t\t}\n\n\t\t// Extract line number\n\t\tlineStr := parts[1]\n\t\tlineNum, err := strconv.Atoi(lineStr)\n\t\tif err != nil {\n\t\t\tt.Fatalf(\"Failed to parse line number from %s: %v\", logCaller, err)\n\t\t}\n\n\t\t// With Helper(), the caller should be near the line where helper was called\n\t\t// Allow a small range of lines to avoid brittle tests\n\t\tif lineNum < callerLine+1 {\n\t\t\tt.Errorf(\"Caller line number (%d) is not near the calling line (%d)\", lineNum, callerLine)\n\t\t}\n\t})\n}\n\ntype testError struct {\n\tWorkflowID string\n}\n\nfunc (e testError) Error() string { return \"test error\" }\nfunc (e testError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"workflow-id\", e.WorkflowID)\n\treturn nil\n}\n\n// Helper function to create the zap logger\nfunc createTestZapLogger(buf *strings.Builder) *zap.Logger {\n\tl := zap.NewAtomicLevel()\n\treturn zap.New(zapcore.NewCore(zapcore.NewJSONEncoder(zapcore.EncoderConfig{\n\t\tTimeKey:        \"ts\",\n\t\tLevelKey:       \"level\",\n\t\tNameKey:        \"logger\",\n\t\tCallerKey:      \"caller\",\n\t\tMessageKey:     \"msg\",\n\t\tStacktraceKey:  \"stacktrace\",\n\t\tLineEnding:     zapcore.DefaultLineEnding,\n\t\tEncodeLevel:    zapcore.LowercaseLevelEncoder,\n\t\tEncodeTime:     zapcore.ISO8601TimeEncoder,\n\t\tEncodeDuration: zapcore.SecondsDurationEncoder,\n\t\tEncodeCaller:   zapcore.ShortCallerEncoder,\n\t}), zapcore.AddSync(buf), l), zap.AddStacktrace(zapcore.ErrorLevel), zap.AddCaller())\n}\n\nfunc helperFuncWithoutHelper(l Logger) (lineNum int) {\n\tlineNum = line() + 1\n\tl.Error(\"test info\", tag.WorkflowActionWorkflowStarted)\n\treturn lineNum\n}\n\nfunc helperFuncWithHelper(l Logger) {\n\tl.Helper().Error(\"test info\", tag.WorkflowActionWorkflowStarted)\n}\n\n// Helper function to get current line number\nfunc line() int {\n\t_, _, line, _ := runtime.Caller(1)\n\treturn line\n}\n"
  },
  {
    "path": "common/log/options.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage log\n\n// Option is used to set options for the logger.\ntype Option func(impl *loggerImpl)\n\n// WithSampleFunc sets the sampling function for the logger.\nfunc WithSampleFunc(fn func(int) bool) Option {\n\treturn func(impl *loggerImpl) {\n\t\timpl.sampleLocalFn = fn\n\t}\n}\n"
  },
  {
    "path": "common/log/panic.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport (\n\t\"fmt\"\n\t\"runtime/debug\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// CapturePanic is used to capture panic, it will log the panic and also return the error through pointer.\n// If the panic value is not error then a default error is returned\n// We have to use pointer is because in golang: \"recover return nil if was not called directly by a deferred function.\"\n// And we have to set the returned error otherwise our handler will return nil as error which is incorrect\n// errPanic MUST be the result from calling recover, which MUST be done in a single level deep\n// deferred function. The usual way of calling this is:\n// - defer func() { log.CapturePanic(recover(), logger, &err) }()\nfunc CapturePanic(errPanic interface{}, logger Logger, retError *error) {\n\tif errPanic != nil {\n\t\terr, ok := errPanic.(error)\n\t\tif !ok {\n\t\t\terr = fmt.Errorf(\"panic object is not error: %#v\", errPanic)\n\t\t}\n\n\t\tst := string(debug.Stack())\n\n\t\t// This function is called in deferred block and is all over the place.\n\t\t// We want the log to point to the line of panic, not this line, or stack of the defer function.\n\t\tlogger.Helper().Helper().Error(\"Panic is captured\", tag.SysStackTrace(st), tag.Error(err))\n\n\t\tif retError != nil {\n\t\t\t*retError = err\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/log/replay.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype replayLogger struct {\n\tlogger            Logger\n\tctx               workflow.Context\n\tenableLogInReplay bool\n}\n\nconst skipForReplayLogger = skipForDefaultLogger + 1\n\n// NewReplayLogger creates a logger which is aware of cadence's replay mode\nfunc NewReplayLogger(logger Logger, ctx workflow.Context, enableLogInReplay bool) Logger {\n\tlg, ok := logger.(*loggerImpl)\n\tif ok {\n\t\tlogger = &loggerImpl{\n\t\t\tzapLogger:     lg.zapLogger,\n\t\t\tskip:          skipForReplayLogger,\n\t\t\tsampleLocalFn: lg.sampleLocalFn,\n\t\t}\n\t} else {\n\t\tlogger.Warn(\"ReplayLogger may not emit callat tag correctly because the logger passed in is not loggerImpl\")\n\t}\n\treturn &replayLogger{\n\t\tlogger:            logger,\n\t\tctx:               ctx,\n\t\tenableLogInReplay: enableLogInReplay,\n\t}\n}\n\nfunc (r *replayLogger) Debugf(msg string, args ...any) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\n\tr.logger.Debugf(fmt.Sprintf(msg, args...))\n}\n\nfunc (r *replayLogger) Debug(msg string, tags ...tag.Tag) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\tr.logger.Debug(msg, tags...)\n}\n\nfunc (r *replayLogger) Info(msg string, tags ...tag.Tag) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\tr.logger.Info(msg, tags...)\n}\n\nfunc (r *replayLogger) Warn(msg string, tags ...tag.Tag) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\tr.logger.Warn(msg, tags...)\n}\n\nfunc (r *replayLogger) Error(msg string, tags ...tag.Tag) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\tr.logger.Error(msg, tags...)\n}\n\nfunc (r *replayLogger) Fatal(msg string, tags ...tag.Tag) {\n\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\treturn\n\t}\n\tr.logger.Fatal(msg, tags...)\n}\n\nfunc (r *replayLogger) SampleInfo(msg string, sampleRate int, tags ...tag.Tag) {\n\tif rand.Intn(sampleRate) == 0 {\n\t\tif workflow.IsReplaying(r.ctx) && !r.enableLogInReplay {\n\t\t\treturn\n\t\t}\n\t\tr.logger.Info(msg, tags...)\n\t}\n}\n\nfunc (r *replayLogger) DebugOn() bool {\n\treturn r.logger.DebugOn()\n}\n\nfunc (r *replayLogger) WithTags(tags ...tag.Tag) Logger {\n\treturn &replayLogger{\n\t\tlogger:            r.logger.WithTags(tags...),\n\t\tctx:               r.ctx,\n\t\tenableLogInReplay: r.enableLogInReplay,\n\t}\n}\n\nfunc (r *replayLogger) Helper() Logger {\n\treturn &replayLogger{\n\t\tlogger:            r.logger.Helper(),\n\t\tctx:               r.ctx,\n\t\tenableLogInReplay: r.enableLogInReplay,\n\t}\n}\n"
  },
  {
    "path": "common/log/tag/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tag\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/zap\"\n)\n\n// Tag is the interface for logging system\ntype Tag struct {\n\t// keep this field private\n\tfield zap.Field\n}\n\n// Field returns a zap field\nfunc (t *Tag) Field() zap.Field {\n\treturn t.field\n}\n\nfunc newStringTag(key string, value string) Tag {\n\treturn Tag{\n\t\tfield: zap.String(key, value),\n\t}\n}\n\nfunc newStringsTag(key string, value []string) Tag {\n\treturn Tag{\n\t\tfield: zap.Strings(key, value),\n\t}\n}\n\nfunc newInt64(key string, value int64) Tag {\n\treturn Tag{\n\t\tfield: zap.Int64(key, value),\n\t}\n}\n\nfunc newInt(key string, value int) Tag {\n\treturn Tag{\n\t\tfield: zap.Int(key, value),\n\t}\n}\n\nfunc newInt32(key string, value int32) Tag {\n\treturn Tag{\n\t\tfield: zap.Int32(key, value),\n\t}\n}\n\nfunc newBoolTag(key string, value bool) Tag {\n\treturn Tag{\n\t\tfield: zap.Bool(key, value),\n\t}\n}\n\nfunc newErrorTag(key string, value error) Tag {\n\t// NOTE zap already chosen \"error\" as key\n\treturn Tag{\n\t\tfield: zap.Error(value),\n\t}\n}\n\nfunc newDurationTag(key string, value time.Duration) Tag {\n\treturn Tag{\n\t\tfield: zap.Duration(key, value),\n\t}\n}\n\nfunc newTimeTag(key string, value time.Time) Tag {\n\treturn Tag{\n\t\tfield: zap.Time(key, value),\n\t}\n}\n\nfunc newObjectTag(key string, value interface{}) Tag {\n\treturn Tag{\n\t\tfield: zap.String(key, fmt.Sprintf(\"%v\", value)),\n\t}\n}\n\nfunc newPredefinedStringTag(key string, value string) Tag {\n\treturn Tag{\n\t\tfield: zap.String(key, value),\n\t}\n}\n\nfunc newPredefinedDynamicTag(key string, value interface{}) Tag {\n\treturn Tag{\n\t\tfield: zap.Any(key, value),\n\t}\n}\n\nfunc newFloat64Tag(key string, value float64) Tag {\n\treturn Tag{\n\t\tfield: zap.Float64(key, value),\n\t}\n}\n"
  },
  {
    "path": "common/log/tag/tags.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tag\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// All logging tags are defined in this file.\n// To help finding available tags, we recommend that all tags to be categorized and placed in the corresponding section.\n// We currently have those categories:\n//   0. Common tags that can't be categorized(or belong to more than one)\n//   1. Workflow: these tags are information that are useful to our customer, like workflow-id/run-id/task-list/...\n//   2. System : these tags are internal information which usually cannot be understood by our customers,\n\n// /////////////////  Common tags defined here ///////////////////\n\n// Error returns tag for Error\nfunc Error(err error) Tag {\n\treturn newErrorTag(\"error\", err)\n}\n\n// ClusterName returns tag for ClusterName\nfunc ClusterName(clusterName string) Tag {\n\treturn newStringTag(\"cluster-name\", clusterName)\n}\n\n// Region returns tag for Region\nfunc Region(region string) Tag {\n\treturn newStringTag(\"region\", region)\n}\n\n// ActiveClusterName returns tag for ActiveClusterName.\nfunc ActiveClusterName(activeClusterName string) Tag {\n\treturn newStringTag(\"active-cluster-name\", activeClusterName)\n}\n\n// ActiveClusterChangeType returns tag for ActiveClusterChangeType.\nfunc ActiveClusterChangeType(changeType string) Tag {\n\treturn newStringTag(\"active-active-change-type\", changeType)\n}\n\n// Timestamp returns tag for Timestamp\nfunc Timestamp(timestamp time.Time) Tag {\n\treturn newTimeTag(\"timestamp\", timestamp)\n}\n\nfunc EarliestTime(time int64) Tag {\n\treturn newInt64(\"earliest-time\", time)\n}\n\nfunc LatestTime(time int64) Tag {\n\treturn newInt64(\"latest-time\", time)\n}\n\n// /////////////////  Workflow tags defined here: ( wf is short for workflow) ///////////////////\n\n// WorkflowAction returns tag for WorkflowAction\nfunc workflowAction(action string) Tag {\n\treturn newPredefinedStringTag(\"wf-action\", action)\n}\n\n// WorkflowListFilterType returns tag for WorkflowListFilterType\nfunc workflowListFilterType(listFilterType string) Tag {\n\treturn newPredefinedStringTag(\"wf-list-filter-type\", listFilterType)\n}\n\n// general\n\n// ClientImpl returns tag for ClientImpl\nfunc ClientImpl(clientImpl string) Tag {\n\treturn newStringTag(\"client-impl\", clientImpl)\n}\n\n// ClientFeatureVersion returns tag for ClientFeatureVersion\nfunc ClientFeatureVersion(clientFeatureVersion string) Tag {\n\treturn newStringTag(\"client-feature-version\", clientFeatureVersion)\n}\n\n// WorkflowError returns tag for WorkflowError\nfunc WorkflowError(error error) Tag {\n\treturn newErrorTag(\"wf-error\", error)\n}\n\n// WorkflowTimeoutType returns tag for WorkflowTimeoutType\nfunc WorkflowTimeoutType(timeoutType int64) Tag {\n\treturn newInt64(\"wf-timeout-type\", timeoutType)\n}\n\n// ActivityTimeoutType returns tag for ActivityTimeoutType\nfunc ActivityTimeoutType(timerType shared.TimeoutType) Tag {\n\treturn newStringTag(\"activity-timer-type\", timerType.String())\n}\n\n// WorkflowPollContextTimeout returns tag for WorkflowPollContextTimeout\nfunc WorkflowPollContextTimeout(pollContextTimeout time.Duration) Tag {\n\treturn newDurationTag(\"wf-poll-context-timeout\", pollContextTimeout)\n}\n\n// WorkflowHandlerName returns tag for WorkflowHandlerName\nfunc WorkflowHandlerName(handlerName string) Tag {\n\treturn newStringTag(\"wf-handler-name\", handlerName)\n}\n\n// WorkflowID returns tag for WorkflowID\nfunc WorkflowID(workflowID string) Tag {\n\treturn newStringTag(\"wf-id\", workflowID)\n}\n\n// TargetWorkflowID returns tag for target WorkflowID\nfunc TargetWorkflowID(workflowID string) Tag {\n\treturn newStringTag(\"target-wf-id\", workflowID)\n}\n\n// WorkflowType returns tag for WorkflowType\nfunc WorkflowType(wfType string) Tag {\n\treturn newStringTag(\"wf-type\", wfType)\n}\n\n// WorkflowSignalName returns tag for WorkflowSignalName\nfunc WorkflowSignalName(signalName string) Tag {\n\treturn newStringTag(\"wf-signal-name\", signalName)\n}\n\n// WorkflowRequestID returns tag for WorkflowRequestID\nfunc WorkflowRequestID(requestID string) Tag {\n\treturn newStringTag(\"wf-request-id\", requestID)\n}\n\n// WorkflowState returns tag for WorkflowState\nfunc WorkflowState(s int) Tag {\n\treturn newInt(\"wf-state\", s)\n}\n\n// WorkflowRunID returns tag for WorkflowRunID\nfunc WorkflowRunID(runID string) Tag {\n\treturn newStringTag(\"wf-run-id\", runID)\n}\n\n// TargetWorkflowRunID returns tag for target  WorkflowRunID\nfunc TargetWorkflowRunID(runID string) Tag {\n\treturn newStringTag(\"target-wf-run-id\", runID)\n}\n\n// WorkflowResetBaseRunID returns tag for WorkflowResetBaseRunID\nfunc WorkflowResetBaseRunID(runID string) Tag {\n\treturn newStringTag(\"wf-reset-base-run-id\", runID)\n}\n\n// WorkflowResetNewRunID returns tag for WorkflowResetNewRunID\nfunc WorkflowResetNewRunID(runID string) Tag {\n\treturn newStringTag(\"wf-reset-new-run-id\", runID)\n}\n\n// WorkflowBinaryChecksum returns tag for WorkflowBinaryChecksum\nfunc WorkflowBinaryChecksum(cs string) Tag {\n\treturn newStringTag(\"wf-binary-checksum\", cs)\n}\n\n// WorkflowActivityID returns tag for WorkflowActivityID\nfunc WorkflowActivityID(id string) Tag {\n\treturn newStringTag(\"wf-activity-id\", id)\n}\n\n// WorkflowTimerID returns tag for WorkflowTimerID\nfunc WorkflowTimerID(id string) Tag {\n\treturn newStringTag(\"wf-timer-id\", id)\n}\n\n// WorkflowBeginningRunID returns tag for WorkflowBeginningRunID\nfunc WorkflowBeginningRunID(beginningRunID string) Tag {\n\treturn newStringTag(\"wf-beginning-run-id\", beginningRunID)\n}\n\n// WorkflowEndingRunID returns tag for WorkflowEndingRunID\nfunc WorkflowEndingRunID(endingRunID string) Tag {\n\treturn newStringTag(\"wf-ending-run-id\", endingRunID)\n}\n\n// WorkflowDecisionTimeoutSeconds returns tag for WorkflowDecisionTimeoutSeconds\nfunc WorkflowDecisionTimeoutSeconds(s int32) Tag {\n\treturn newInt32(\"wf-decision-timeout\", s)\n}\n\n// QueryID returns tag for QueryID\nfunc QueryID(queryID string) Tag {\n\treturn newStringTag(\"query-id\", queryID)\n}\n\n// BlobSizeViolationOperation returns tag for BlobSizeViolationOperation\nfunc BlobSizeViolationOperation(operation string) Tag {\n\treturn newStringTag(\"blob-size-violation-operation\", operation)\n}\n\n// WorkflowCronSchedule returns a tag to report a workflow's cron schedule\nfunc WorkflowCronSchedule(schedule string) Tag {\n\treturn newStringTag(\"wf-cron-schedule\", schedule)\n}\n\n// WorkflowCloseStatus returns a tag to report a workflow's close status\nfunc WorkflowCloseStatus(status int) Tag {\n\treturn newInt(\"close-status\", status)\n}\n\n// IsWorkflowOpen returns a tag to report a workflow is open or not\nfunc IsWorkflowOpen(isOpen bool) Tag {\n\treturn newBoolTag(\"is-workflow-open\", isOpen)\n}\n\n// WorkflowTerminationReason returns a tag to report a workflow's termination reason\nfunc WorkflowTerminationReason(reason string) Tag {\n\treturn newStringTag(\"wf-termination-reason\", reason)\n}\n\nfunc Duration(duration time.Duration) Tag {\n\treturn newDurationTag(\"duration\", duration)\n}\n\n// domain related\n\n// WorkflowDomainID returns tag for WorkflowDomainID\nfunc WorkflowDomainID(domainID string) Tag {\n\treturn newStringTag(\"wf-domain-id\", domainID)\n}\n\n// TargetWorkflowDomainID returns tag for target WorkflowDomainID\nfunc TargetWorkflowDomainID(domainID string) Tag {\n\treturn newStringTag(\"target-wf-domain-id\", domainID)\n}\n\n// WorkflowDomainName returns tag for WorkflowDomainName\nfunc WorkflowDomainName(domainName string) Tag {\n\treturn newStringTag(\"wf-domain-name\", domainName)\n}\n\n// WorkflowDomainIDs returns tag for WorkflowDomainIDs\nfunc WorkflowDomainIDs(domainIDs interface{}) Tag {\n\treturn newObjectTag(\"wf-domain-ids\", domainIDs)\n}\n\n// OperationName returns tag for OperationName\nfunc OperationName(operationName string) Tag {\n\treturn newStringTag(\"operation-name\", operationName)\n}\n\n// history event ID related\n\n// WorkflowEventID returns tag for WorkflowEventID\nfunc WorkflowEventID(eventID int64) Tag {\n\treturn newInt64(\"wf-history-event-id\", eventID)\n}\n\n// WorkflowScheduleID returns tag for WorkflowScheduleID\nfunc WorkflowScheduleID(scheduleID int64) Tag {\n\treturn newInt64(\"wf-schedule-id\", scheduleID)\n}\n\n// WorkflowStartedID returns tag for WorkflowStartedID\nfunc WorkflowStartedID(id int64) Tag {\n\treturn newInt64(\"wf-started-id\", id)\n}\n\n// WorkflowInitiatedID returns tag for WorkflowInitiatedID\nfunc WorkflowInitiatedID(id int64) Tag {\n\treturn newInt64(\"wf-initiated-id\", id)\n}\n\n// WorkflowFirstEventID returns tag for WorkflowFirstEventID\nfunc WorkflowFirstEventID(firstEventID int64) Tag {\n\treturn newInt64(\"wf-first-event-id\", firstEventID)\n}\n\n// WorkflowNextEventID returns tag for WorkflowNextEventID\nfunc WorkflowNextEventID(nextEventID int64) Tag {\n\treturn newInt64(\"wf-next-event-id\", nextEventID)\n}\n\n// WorkflowResetNextEventID returns tag for WorkflowResetNextEventID\nfunc WorkflowResetNextEventID(resetNextEventID int64) Tag {\n\treturn newInt64(\"wf-reset-next-event-id\", resetNextEventID)\n}\n\nfunc WorkflowExternalEntityType(externalEntityType string) Tag {\n\treturn newStringTag(\"wf-external-entity-type\", externalEntityType)\n}\n\nfunc WorkflowExternalEntityKey(externalEntityKey string) Tag {\n\treturn newStringTag(\"wf-external-entity-key\", externalEntityKey)\n}\n\n// history tree\n\n// WorkflowTreeID returns tag for WorkflowTreeID\nfunc WorkflowTreeID(treeID string) Tag {\n\treturn newStringTag(\"wf-tree-id\", treeID)\n}\n\n// WorkflowBranchID returns tag for WorkflowBranchID\nfunc WorkflowBranchID(branchID string) Tag {\n\treturn newStringTag(\"wf-branch-id\", branchID)\n}\n\n// workflow task\n\n// WorkflowDecisionType returns tag for WorkflowDecisionType\nfunc WorkflowDecisionType(decisionType int64) Tag {\n\treturn newInt64(\"wf-decision-type\", decisionType)\n}\n\n// WorkflowQueryType returns tag for WorkflowQueryType\nfunc WorkflowQueryType(qt string) Tag {\n\treturn newStringTag(\"wf-query-type\", qt)\n}\n\n// WorkflowDecisionFailCause returns tag for WorkflowDecisionFailCause\nfunc WorkflowDecisionFailCause(decisionFailCause int64) Tag {\n\treturn newInt64(\"wf-decision-fail-cause\", decisionFailCause)\n}\n\n// WorkflowTaskListType returns tag for WorkflowTaskListType\nfunc WorkflowTaskListType(taskListType int) Tag {\n\treturn newInt(\"wf-task-list-type\", taskListType)\n}\n\n// WorkflowTaskListKind returns tag for WorkflowTaskListKind\nfunc WorkflowTaskListKind(taskListKind int32) Tag {\n\treturn newInt32(\"wf-task-list-kind\", taskListKind)\n}\n\n// WorkflowTaskListName returns tag for WorkflowTaskListName\nfunc WorkflowTaskListName(taskListName string) Tag {\n\treturn newStringTag(\"wf-task-list-name\", taskListName)\n}\n\n// size limit\n\n// WorkflowSize returns tag for WorkflowSize\nfunc WorkflowSize(workflowSize int64) Tag {\n\treturn newInt64(\"wf-size\", workflowSize)\n}\n\n// WorkflowSignalCount returns tag for SignalCount\nfunc WorkflowSignalCount(signalCount int32) Tag {\n\treturn newInt32(\"wf-signal-count\", signalCount)\n}\n\n// WorkflowHistorySize returns tag for HistorySize\nfunc WorkflowHistorySize(historySize int) Tag {\n\treturn newInt(\"wf-history-size\", historySize)\n}\n\n// WorkflowHistorySizeBytes returns tag for HistorySizeBytes\nfunc WorkflowHistorySizeBytes(historySizeBytes int) Tag {\n\treturn newInt(\"wf-history-size-bytes\", historySizeBytes)\n}\n\n// WorkflowEventCount returns tag for EventCount\nfunc WorkflowEventCount(eventCount int) Tag {\n\treturn newInt(\"wf-event-count\", eventCount)\n}\n\nfunc WorkflowEventType(eventType string) Tag {\n\treturn newStringTag(\"wf-event-type\", eventType)\n}\n\n// /////////////////  System tags defined here:  ///////////////////\n// Tags with pre-define values\n\n// component returns tag for component\nfunc component(component string) Tag {\n\treturn newPredefinedStringTag(\"component\", component)\n}\n\n// lifecycle returns tag for lifecycle\nfunc lifecycle(lifecycle string) Tag {\n\treturn newPredefinedStringTag(\"lifecycle\", lifecycle)\n}\n\n// storeOperation returns tag for storeOperation\nfunc storeOperation(storeOperation string) Tag {\n\treturn newPredefinedStringTag(\"store-operation\", storeOperation)\n}\n\n// clientOperation returns tag for clientOperation\nfunc clientOperation(clientOperation string) Tag {\n\treturn newPredefinedStringTag(\"client-operation\", clientOperation)\n}\n\n// operationResult returns tag for operationResult\nfunc operationResult(operationResult string) Tag {\n\treturn newPredefinedStringTag(\"operation-result\", operationResult)\n}\n\n// errorType returns tag for errorType\nfunc errorType(errorType string) Tag {\n\treturn newPredefinedStringTag(\"error-type\", errorType)\n}\n\n// shardupdate returns tag for shardupdate\nfunc shardupdate(shardupdate string) Tag {\n\treturn newPredefinedStringTag(\"shard-update\", shardupdate)\n}\n\n// idType returns tag for idType\nfunc idType(idType string) Tag {\n\treturn newPredefinedStringTag(\"id-type\", idType)\n}\n\n// queueType returns tag for queueType\nfunc queueType(queueType string) Tag {\n\treturn newPredefinedStringTag(\"queue-type\", queueType)\n}\n\n// general\n\n// Service returns tag for Service\nfunc Service(sv string) Tag {\n\treturn newStringTag(\"service\", sv)\n}\n\n// DestService returns tag for destination service\nfunc DestService(sv string) Tag {\n\treturn newStringTag(\"dest-service\", sv)\n}\n\n// Addresses returns tag for Addresses\nfunc Addresses(ads []string) Tag {\n\treturn newObjectTag(\"addresses\", ads)\n}\n\n// Subscriber returns tag for Subscriber\nfunc Subscriber(subscriber string) Tag {\n\treturn newStringTag(\"subscriber\", subscriber)\n}\n\n// Address return tag for Address\nfunc Address(ad string) Tag {\n\treturn newStringTag(\"address\", ad)\n}\n\n// Env return tag for runtime environment\nfunc Env(env string) Tag {\n\treturn newStringTag(\"env\", env)\n}\n\n// Key returns tag for Key\nfunc Key(k string) Tag {\n\treturn newStringTag(\"key\", k)\n}\n\n// Name returns tag for Name\nfunc Name(k string) Tag {\n\treturn newStringTag(\"name\", k)\n}\n\n// Mode returns tag for Mode\nfunc Mode(mode string) Tag {\n\treturn newStringTag(\"mode\", mode)\n}\n\n// Value returns tag for Value\nfunc Value(v interface{}) Tag {\n\treturn newObjectTag(\"value\", v)\n}\n\n// Reason returns tag for Reason\nfunc Reason(reason string) Tag {\n\treturn newStringTag(\"reason\", reason)\n}\n\n// ValueType returns tag for ValueType\nfunc ValueType(v interface{}) Tag {\n\treturn newStringTag(\"value-type\", fmt.Sprintf(\"%T\", v))\n}\n\n// DefaultValue returns tag for DefaultValue\nfunc DefaultValue(v interface{}) Tag {\n\treturn newObjectTag(\"default-value\", v)\n}\n\n// Port returns tag for Port\nfunc Port(p int) Tag {\n\treturn newInt(\"port\", p)\n}\n\n// CursorTimestamp returns tag for CursorTimestamp\nfunc CursorTimestamp(timestamp time.Time) Tag {\n\treturn newTimeTag(\"cursor-timestamp\", timestamp)\n}\n\n// MetricScope returns tag for MetricScope\nfunc MetricScope(metricScope int) Tag {\n\treturn newInt(\"metric-scope\", metricScope)\n}\n\n// StoreType returns tag for StoreType\nfunc StoreType(storeType string) Tag {\n\treturn newPredefinedStringTag(\"store-type\", storeType)\n}\n\n// StoreError returns tag for StoreError\nfunc StoreError(storeErr error) Tag {\n\treturn newErrorTag(\"store-error\", storeErr)\n}\n\n// StoreShard returns tag for StoreShard\nfunc StoreShard(storeShard string) Tag {\n\treturn newPredefinedStringTag(\"store-shard\", storeShard)\n}\n\n// ClientError returns tag for ClientError\nfunc ClientError(clientErr error) Tag {\n\treturn newErrorTag(\"client-error\", clientErr)\n}\n\n// DetailInfo returns tag for DetailInfo\nfunc DetailInfo(i string) Tag {\n\treturn newStringTag(\"detail-info\", i)\n}\n\n// Counter returns tag for Counter\nfunc Counter(c int) Tag {\n\treturn newInt(\"counter\", c)\n}\n\n// Number returns tag for Number\nfunc Number(n int64) Tag {\n\treturn newInt64(\"number\", n)\n}\n\n// NextNumber returns tag for NextNumber\nfunc NextNumber(n int64) Tag {\n\treturn newInt64(\"next-number\", n)\n}\n\n// Bool returns tag for Bool\nfunc Bool(b bool) Tag {\n\treturn newBoolTag(\"bool\", b)\n}\n\n/* Tags for logging manual access */\n\n// RequestCaller returns tag for caller (the name of the service making this request)\nfunc RequestCaller(callerName string) Tag {\n\treturn newStringTag(\"request-caller\", callerName)\n}\n\n// ActorType returns type of the actor (service or user) making this request\nfunc ActorType(actorType string) Tag {\n\treturn newStringTag(\"actor-type\", actorType)\n}\n\n// ActorID returns tag for the actor ID\nfunc ActorID(actorID string) Tag {\n\treturn newStringTag(\"actor-id\", actorID)\n}\n\n// ActorEmail returns tag for the actor's email address\nfunc ActorEmail(actorEmail string) Tag {\n\treturn newStringTag(\"actor-email\", actorEmail)\n}\n\nfunc IsShadowModeEnabled(isShadow bool) Tag {\n\treturn newBoolTag(\"is-shadow-mode-enabled\", isShadow)\n}\n\n// HandlerCall returns tag for the API name of a request\nfunc HandlerCall(handlerCall string) Tag {\n\treturn newStringTag(\"handler-call\", handlerCall)\n}\n\n// RequestBody returns the tag for the API request body\nfunc RequestBody(requestBody string) Tag {\n\treturn newStringTag(\"request-body\", requestBody)\n}\n\n// RequestType return tag for the type of request (internal, external)\nfunc RequestType(requestType string) Tag {\n\treturn newStringTag(\"request-type\", requestType)\n}\n\n// history engine shard\n\n// ShardID returns tag for ShardID\nfunc ShardID(shardID int) Tag {\n\treturn newInt(\"shard-id\", shardID)\n}\n\n// ShardTime returns tag for ShardTime\nfunc ShardTime(shardTime interface{}) Tag {\n\treturn newObjectTag(\"shard-time\", shardTime)\n}\n\n// ShardReplicationAck returns tag for ShardReplicationAck\nfunc ShardReplicationAck(shardReplicationAck int64) Tag {\n\treturn newInt64(\"shard-replication-ack\", shardReplicationAck)\n}\n\n// ShardReplicationToken returns information about a particular replication request\nfunc ShardReplicationToken(token interface{}) Tag {\n\treturn newObjectTag(\"shard-replication-token\", token)\n}\n\n// PreviousShardRangeID returns tag for PreviousShardRangeID\nfunc PreviousShardRangeID(id int64) Tag {\n\treturn newInt64(\"previous-shard-range-id\", id)\n}\n\n// ShardRangeID returns tag for ShardRangeID\nfunc ShardRangeID(id int64) Tag {\n\treturn newInt64(\"shard-range-id\", id)\n}\n\n// ReadLevel returns tag for ReadLevel\nfunc ReadLevel(lv int64) Tag {\n\treturn newInt64(\"read-level\", lv)\n}\n\n// MinLevel returns tag for MinLevel\nfunc MinLevel(lv int64) Tag {\n\treturn newInt64(\"min-level\", lv)\n}\n\n// MaxLevel returns tag for MaxLevel\nfunc MaxLevel(lv int64) Tag {\n\treturn newInt64(\"max-level\", lv)\n}\n\n// ShardTransferAcks returns tag for ShardTransferAcks\nfunc ShardTransferAcks(shardTransferAcks interface{}) Tag {\n\treturn newObjectTag(\"shard-transfer-acks\", shardTransferAcks)\n}\n\n// ShardTimerAcks returns tag for ShardTimerAcks\nfunc ShardTimerAcks(shardTimerAcks interface{}) Tag {\n\treturn newObjectTag(\"shard-timer-acks\", shardTimerAcks)\n}\n\n// ShardCrossClusterAcks returns tag for ShardCrossClusterAcks\nfunc ShardCrossClusterAcks(shardCrossClusterAcks interface{}) Tag {\n\treturn newObjectTag(\"shard-cross-cluster-acks\", shardCrossClusterAcks)\n}\n\n// task queue processor\n\n// QueueLevel returns tag for QueueLevel\nfunc QueueLevel(level int) Tag {\n\treturn newInt(\"queue-level\", level)\n}\n\n// PreviousQueueLevel returns tag for PreviousQueueLevel\nfunc PreviousQueueLevel(level int) Tag {\n\treturn newInt(\"previous-queue-level\", level)\n}\n\n// QueueSplitPolicyType returns tag for QueueSplitPolicyType\nfunc QueueSplitPolicyType(policyType int) Tag {\n\treturn newInt(\"split-policy-type\", policyType)\n}\n\n// TaskID returns tag for TaskID\nfunc TaskID(taskID int64) Tag {\n\treturn newInt64(\"queue-task-id\", taskID)\n}\n\n// TaskType returns tag for TaskType for queue processor\nfunc TaskType(taskType int) Tag {\n\treturn newInt(\"queue-task-type\", taskType)\n}\n\n// TaskVisibilityTimestamp returns tag for task visibilityTimestamp\nfunc TaskVisibilityTimestamp(timestamp int64) Tag {\n\treturn newInt64(\"queue-task-visibility-timestamp\", timestamp)\n}\n\n// NumberProcessed returns tag for NumberProcessed\nfunc NumberProcessed(n int) Tag {\n\treturn newInt(\"number-processed\", n)\n}\n\n// NumberDeleted returns tag for NumberDeleted\nfunc NumberDeleted(n int) Tag {\n\treturn newInt(\"number-deleted\", n)\n}\n\n// TimerTaskStatus returns tag for TimerTaskStatus\nfunc TimerTaskStatus(timerTaskStatus int32) Tag {\n\treturn newInt32(\"timer-task-status\", timerTaskStatus)\n}\n\n// retry\n\n// Attempt returns tag for Attempt\nfunc Attempt(attempt int32) Tag {\n\treturn newInt32(\"attempt\", attempt)\n}\n\n// AttemptCount returns tag for AttemptCount\nfunc AttemptCount(attemptCount int) Tag {\n\treturn newInt(\"attempt-count\", attemptCount)\n}\n\n// AttemptStart returns tag for AttemptStart\nfunc AttemptStart(attemptStart time.Time) Tag {\n\treturn newTimeTag(\"attempt-start\", attemptStart)\n}\n\n// AttemptEnd returns tag for AttemptEnd\nfunc AttemptEnd(attemptEnd time.Time) Tag {\n\treturn newTimeTag(\"attempt-end\", attemptEnd)\n}\n\n// ScheduleAttempt returns tag for ScheduleAttempt\nfunc ScheduleAttempt(scheduleAttempt int64) Tag {\n\treturn newInt64(\"schedule-attempt\", scheduleAttempt)\n}\n\n// ElasticSearch\n\n// ESRequest returns tag for ESRequest\nfunc ESRequest(ESRequest string) Tag {\n\treturn newStringTag(\"es-request\", ESRequest)\n}\n\n// ESResponseStatus returns tag for ESResponse status\nfunc ESResponseStatus(status int) Tag {\n\treturn newInt(\"es-response-status\", status)\n}\n\n// ESResponseError returns tag for ESResponse error\nfunc ESResponseError(msg string) Tag {\n\treturn newStringTag(\"es-response-error\", msg)\n}\n\n// ESKey returns tag for ESKey\nfunc ESKey(ESKey string) Tag {\n\treturn newStringTag(\"es-mapping-key\", ESKey)\n}\n\n// ESValue returns tag for ESValue\nfunc ESValue(ESValue []byte) Tag {\n\t// convert value to string type so that the value logged is human readable\n\treturn newStringTag(\"es-mapping-value\", string(ESValue))\n}\n\n// ESConfig returns tag for ESConfig\nfunc ESConfig(c interface{}) Tag {\n\treturn newObjectTag(\"es-config\", c)\n}\n\n// ESField returns tag for ESField\nfunc ESField(ESField string) Tag {\n\treturn newStringTag(\"es-field\", ESField)\n}\n\n// ESDocID returns tag for ESDocID\nfunc ESDocID(id string) Tag {\n\treturn newStringTag(\"es-doc-id\", id)\n}\n\n// ESAggregationID returns tag for ESDocID\nfunc ESAggregationID(id string) Tag {\n\treturn newStringTag(\"es-agg-id\", id)\n}\n\n// LoggingCallAtKey is reserved tag\nconst LoggingCallAtKey = \"logging-call-at\"\n\n// SysStackTrace returns tag for SysStackTrace\nfunc SysStackTrace(stackTrace string) Tag {\n\treturn newStringTag(\"sys-stack-trace\", stackTrace)\n}\n\n// Kafka related\n\n// KafkaTopicName returns tag for TopicName\nfunc KafkaTopicName(topicName string) Tag {\n\treturn newStringTag(\"kafka-topic-name\", topicName)\n}\n\n// KafkaConsumerName returns tag for ConsumerName\nfunc KafkaConsumerName(consumerName string) Tag {\n\treturn newStringTag(\"kafka-consumer-name\", consumerName)\n}\n\n// KafkaPartition returns tag for Partition\nfunc KafkaPartition(partition int32) Tag {\n\treturn newInt32(\"kafka-partition\", partition)\n}\n\n// KafkaPartitionKey returns tag for PartitionKey\nfunc KafkaPartitionKey(partitionKey interface{}) Tag {\n\treturn newObjectTag(\"kafka-partition-key\", partitionKey)\n}\n\n// KafkaOffset returns tag for Offset\nfunc KafkaOffset(offset int64) Tag {\n\treturn newInt64(\"kafka-offset\", offset)\n}\n\n// TokenLastEventID returns tag for TokenLastEventID\nfunc TokenLastEventID(id int64) Tag {\n\treturn newInt64(\"token-last-event-id\", id)\n}\n\n// /////////////////  XDC tags defined here: xdc- ///////////////////\n\n// SourceCluster returns tag for SourceCluster\nfunc SourceCluster(sourceCluster string) Tag {\n\treturn newStringTag(\"xdc-source-cluster\", sourceCluster)\n}\n\nfunc RemoteCluster(remoteCluster string) Tag {\n\treturn newStringTag(\"xdc-remote-cluster\", remoteCluster)\n}\n\n// PrevActiveCluster returns tag for PrevActiveCluster\nfunc PrevActiveCluster(prevActiveCluster string) Tag {\n\treturn newStringTag(\"xdc-prev-active-cluster\", prevActiveCluster)\n}\n\n// FailoverMsg returns tag for FailoverMsg\nfunc FailoverMsg(failoverMsg string) Tag {\n\treturn newStringTag(\"xdc-failover-msg\", failoverMsg)\n}\n\n// FailoverVersion returns tag for Version\nfunc FailoverVersion(version int64) Tag {\n\treturn newInt64(\"xdc-failover-version\", version)\n}\n\n// CurrentVersion returns tag for CurrentVersion\nfunc CurrentVersion(currentVersion int64) Tag {\n\treturn newInt64(\"xdc-current-version\", currentVersion)\n}\n\n// IncomingVersion returns tag for IncomingVersion\nfunc IncomingVersion(incomingVersion int64) Tag {\n\treturn newInt64(\"xdc-incoming-version\", incomingVersion)\n}\n\n// ReplicationInfo returns tag for ReplicationInfo\nfunc ReplicationInfo(replicationInfo interface{}) Tag {\n\treturn newObjectTag(\"xdc-replication-info\", replicationInfo)\n}\n\n// FirstEventVersion returns tag for FirstEventVersion\nfunc FirstEventVersion(version int64) Tag {\n\treturn newInt64(\"xdc-first-event-version\", version)\n}\n\n// LastEventVersion returns tag for LastEventVersion\nfunc LastEventVersion(version int64) Tag {\n\treturn newInt64(\"xdc-last-event-version\", version)\n}\n\n// TokenLastEventVersion returns tag for TokenLastEventVersion\nfunc TokenLastEventVersion(version int64) Tag {\n\treturn newInt64(\"xdc-token-last-event-version\", version)\n}\n\n// ResponseSize returns tag for ResponseSize\nfunc ResponseSize(size int) Tag {\n\treturn newInt(\"response-size\", size)\n}\n\n// ResponseTotalSize returns tag for ResponseTotalSize\nfunc ResponseTotalSize(size int) Tag {\n\treturn newInt(\"response-total-size\", size)\n}\n\n// ResponseMaxSize returns tag for ResponseMaxSize\nfunc ResponseMaxSize(size int) Tag {\n\treturn newInt(\"response-max-size\", size)\n}\n\n// ReplicationMessagesTotalSize returns tag for ReplicationMessagesTotalSize\n// Should be used to indicate the final size of types.ReplicationMessages\nfunc ReplicationMessagesTotalSize(size int) Tag {\n\treturn newInt(\"replication-messages-total-size\", size)\n}\n\n// ReplicationMessagesMaxSize returns tag for ReplicationMessagesMaxSize\n// Should be used to indicate maximum allowed size of types.ReplicationMessages\nfunc ReplicationMessagesMaxSize(size int) Tag {\n\treturn newInt(\"replication-messages-max-size\", size)\n}\n\n// ReplicationTaskID returns tag for ReplicationTaskID\n// Should be used to indicate id of a types.ReplicationTask\nfunc ReplicationTaskID(id int64) Tag {\n\treturn newInt64(\"replication-task-id\", id)\n}\n\n// ReplicationTaskCreationTime returns tag for ReplicationTaskCreationTime\n// Should be used to indicate CreationTime of a types.ReplicationTask\nfunc ReplicationTaskCreationTime(creationTime *int64) Tag {\n\tif creationTime == nil {\n\t\treturn newStringTag(\"replication-task-creation-time\", \"nil\")\n\t}\n\treturn newInt64(\"replication-task-creation-time\", *creationTime)\n}\n\n// ReplicationTaskBatchSize returns tag for task batch size\n// Should be used to indicate used batch size for replication task processing\nfunc ReplicationTaskBatchSize(batchSize int) Tag {\n\treturn newInt(\"replication-task-batch-size\", batchSize)\n}\n\n// /////////////////  Archival tags defined here: archival- ///////////////////\n// archival request tags\n\n// ArchivalCallerServiceName returns tag for the service name calling archival client\nfunc ArchivalCallerServiceName(callerServiceName string) Tag {\n\treturn newStringTag(\"archival-caller-service-name\", callerServiceName)\n}\n\n// ArchivalArchiveAttemptedInline returns tag for whether archival is attempted inline before signal is sent.\nfunc ArchivalArchiveAttemptedInline(archiveInline bool) Tag {\n\treturn newBoolTag(\"archival-archive-attempted-inline\", archiveInline)\n}\n\n// ArchivalRequestDomainID returns tag for RequestDomainID\nfunc ArchivalRequestDomainID(requestDomainID string) Tag {\n\treturn newStringTag(\"archival-request-domain-id\", requestDomainID)\n}\n\n// ArchivalRequestDomainName returns tag for RequestDomainName\nfunc ArchivalRequestDomainName(requestDomainName string) Tag {\n\treturn newStringTag(\"archival-request-domain-name\", requestDomainName)\n}\n\n// ArchivalRequestWorkflowID returns tag for RequestWorkflowID\nfunc ArchivalRequestWorkflowID(requestWorkflowID string) Tag {\n\treturn newStringTag(\"archival-request-workflow-id\", requestWorkflowID)\n}\n\n// ArchvialRequestWorkflowType returns tag for RequestWorkflowType\nfunc ArchvialRequestWorkflowType(requestWorkflowType string) Tag {\n\treturn newStringTag(\"archival-request-workflow-type\", requestWorkflowType)\n}\n\n// ArchivalRequestRunID returns tag for RequestRunID\nfunc ArchivalRequestRunID(requestRunID string) Tag {\n\treturn newStringTag(\"archival-request-run-id\", requestRunID)\n}\n\n// ArchivalRequestBranchToken returns tag for RequestBranchToken\nfunc ArchivalRequestBranchToken(requestBranchToken []byte) Tag {\n\treturn newObjectTag(\"archival-request-branch-token\", requestBranchToken)\n}\n\n// ArchivalRequestNextEventID returns tag for RequestNextEventID\nfunc ArchivalRequestNextEventID(requestNextEventID int64) Tag {\n\treturn newInt64(\"archival-request-next-event-id\", requestNextEventID)\n}\n\n// ArchivalRequestCloseFailoverVersion returns tag for RequestCloseFailoverVersion\nfunc ArchivalRequestCloseFailoverVersion(requestCloseFailoverVersion int64) Tag {\n\treturn newInt64(\"archival-request-close-failover-version\", requestCloseFailoverVersion)\n}\n\n// ArchivalRequestCloseTimestamp returns tag for RequestCloseTimestamp\nfunc ArchivalRequestCloseTimestamp(requestCloseTimeStamp int64) Tag {\n\treturn newInt64(\"archival-request-close-timestamp\", requestCloseTimeStamp)\n}\n\n// ArchivalRequestCloseStatus returns tag for RequestCloseStatus\nfunc ArchivalRequestCloseStatus(requestCloseStatus string) Tag {\n\treturn newStringTag(\"archival-request-close-status\", requestCloseStatus)\n}\n\n// ArchivalURI returns tag for Archival URI\nfunc ArchivalURI(URI string) Tag {\n\treturn newStringTag(\"archival-URI\", URI)\n}\n\n// ArchivalArchiveFailReason returns tag for ArchivalArchiveFailReason\nfunc ArchivalArchiveFailReason(archiveFailReason string) Tag {\n\treturn newStringTag(\"archival-archive-fail-reason\", archiveFailReason)\n}\n\n// ArchivalDeleteHistoryFailReason returns tag for ArchivalDeleteHistoryFailReason\nfunc ArchivalDeleteHistoryFailReason(deleteHistoryFailReason string) Tag {\n\treturn newStringTag(\"archival-delete-history-fail-reason\", deleteHistoryFailReason)\n}\n\n// ArchivalVisibilityQuery returns tag for the query for getting archived visibility record\nfunc ArchivalVisibilityQuery(query string) Tag {\n\treturn newStringTag(\"archival-visibility-query\", query)\n}\n\n// The following logger tags are only used by internal archiver implemention.\n// TODO: move them to internal repo once cadence plugin model is in place.\n\n// ArchivalBlobKey returns tag for BlobKey\nfunc ArchivalBlobKey(blobKey string) Tag {\n\treturn newStringTag(\"archival-blob-key\", blobKey)\n}\n\n// ArchivalDeterministicConstructionCheckFailReason returns tag for ArchivalDeterministicConstructionCheckFailReason\nfunc ArchivalDeterministicConstructionCheckFailReason(deterministicConstructionCheckFailReason string) Tag {\n\treturn newStringTag(\"archival-deterministic-construction-check-fail-reason\", deterministicConstructionCheckFailReason)\n}\n\n// ArchivalNonDeterministicBlobKey returns tag for randomly generated NonDeterministicBlobKey\nfunc ArchivalNonDeterministicBlobKey(nondeterministicBlobKey string) Tag {\n\treturn newStringTag(\"archival-non-deterministic-blob-key\", nondeterministicBlobKey)\n}\n\n// ArchivalBlobIntegrityCheckFailReason returns tag for ArchivalBlobIntegrityCheckFailReason\nfunc ArchivalBlobIntegrityCheckFailReason(blobIntegrityCheckFailReason string) Tag {\n\treturn newStringTag(\"archival-blob-integrity-check-fail-reason\", blobIntegrityCheckFailReason)\n}\n\n// ArchivalBlobstoreContextTimeout returns tag for ArchivalBlobstoreContextTimeout\nfunc ArchivalBlobstoreContextTimeout(blobstoreContextTimeout time.Duration) Tag {\n\treturn newDurationTag(\"archival-blobstore-context-timeout\", blobstoreContextTimeout)\n}\n\n// VisibilityQuery returns tag for the query for getting visibility records\nfunc VisibilityQuery(query string) Tag {\n\treturn newStringTag(\"visibility-query\", query)\n}\n\n// MembershipChangeEvent is a predefined tag for when logging hashring change events,\n// expected to be of type membership.ChangeEvent\nfunc MembershipChangeEvent(event interface{}) Tag {\n\treturn newPredefinedDynamicTag(\"membership-change-event\", event)\n}\n\n// Dynamic Uses reflection based logging for arbitrary values\n// for not very performant logging\nfunc Dynamic(key string, v interface{}) Tag {\n\treturn newPredefinedDynamicTag(key, v)\n}\n\nfunc IsolationGroup(group string) Tag {\n\treturn newStringTag(\"isolation-group\", group)\n}\n\nfunc TaskLatency(duration time.Duration) Tag {\n\treturn newDurationTag(\"task-latency\", duration)\n}\n\nfunc IsolationDuration(duration time.Duration) Tag {\n\treturn newDurationTag(\"isolation-duration\", duration)\n}\n\nfunc PartitionConfig(p map[string]string) Tag {\n\treturn newObjectTag(\"partition-config\", p)\n}\n\nfunc PollerGroups(pollers []string) Tag {\n\treturn newObjectTag(\"poller-isolation-groups\", pollers)\n}\n\nfunc FallbackIsolationGroup(group string) Tag {\n\treturn newStringTag(\"fallback-isolation-group\", group)\n}\n\nfunc PollerGroupsConfiguration(pollers types.IsolationGroupConfiguration) Tag {\n\treturn newObjectTag(\"poller-isolation-groups\", pollers.ToPartitionList())\n}\n\nfunc WorkflowIDCacheSize(size int) Tag {\n\treturn newInt(\"workflow-id-cache-size\", size)\n}\n\nfunc AsyncWFQueueID(queueID string) Tag {\n\treturn newStringTag(\"async-wf-queue-id\", queueID)\n}\n\nfunc AsyncWFRequestType(requestType string) Tag {\n\treturn newStringTag(\"async-wf-request-type\", requestType)\n}\n\nfunc GlobalRatelimiterKey(globalKey string) Tag {\n\treturn newStringTag(\"global-ratelimit-key\", globalKey)\n}\nfunc GlobalRatelimiterKeyMode(mode string) Tag {\n\treturn newStringTag(\"global-ratelimit-key-mode\", mode)\n}\nfunc GlobalRatelimiterIdleCount(count int) Tag {\n\treturn newInt(\"global-ratelimit-key-idle-count\", count)\n}\nfunc GlobalRatelimiterCollectionName(name string) Tag {\n\treturn newStringTag(\"global-ratelimit-collection\", name)\n}\nfunc GlobalRatelimiterPeer(peer string) Tag {\n\treturn newStringTag(\"global-ratelimit-peer\", peer)\n}\n\nfunc CurrentQPS(qps float64) Tag {\n\treturn newFloat64Tag(\"current-qps\", qps)\n}\n\nfunc NumReadPartitions(n int) Tag {\n\treturn newInt(\"num-read-partitions\", n)\n}\n\nfunc NumWritePartitions(n int) Tag {\n\treturn newInt(\"num-write-partitions\", n)\n}\n\nfunc ReadChanged(b bool) Tag {\n\treturn newBoolTag(\"read-changed\", b)\n}\n\nfunc WriteChanged(b bool) Tag {\n\treturn newBoolTag(\"write-changed\", b)\n}\n\nfunc IsolationChanged(b bool) Tag {\n\treturn newBoolTag(\"isolation-changed\", b)\n}\n\nfunc CurrentNumReadPartitions(n int) Tag {\n\treturn newInt(\"current-num-read-partitions\", n)\n}\n\nfunc CurrentNumWritePartitions(n int) Tag {\n\treturn newInt(\"current-num-write-partitions\", n)\n}\n\nfunc PartitionUpscaleThreshold(qps float64) Tag {\n\treturn newFloat64Tag(\"partition-upscale-threshold\", qps)\n}\n\nfunc PartitionDownscaleThreshold(qps float64) Tag {\n\treturn newFloat64Tag(\"partition-downscale-threshold\", qps)\n}\n\nfunc PartitionDownscaleFactor(qps float64) Tag {\n\treturn newFloat64Tag(\"partition-downscale-factor\", qps)\n}\n\nfunc MatchingTaskID(id int64) Tag {\n\treturn newInt64(\"matching-task-id\", id)\n}\n\nfunc MatchingTaskScheduleID(id int64) Tag {\n\treturn newInt64(\"matching-task-schedule-id\", id)\n}\n\nfunc DecisionTaskState(state int32) Tag {\n\treturn newInt32(\"decision-task-state\", state)\n}\n\nfunc ActivityTaskState(state int32) Tag {\n\treturn newInt32(\"activity-task-state\", state)\n}\n\nfunc ShardNamespace(name string) Tag {\n\treturn newStringTag(\"shard-namespace\", name)\n}\n\nfunc ShardExecutor(ID string) Tag {\n\treturn newStringTag(\"shard-executor\", ID)\n}\n\nfunc ShardExecutors(executorIDs []string) Tag {\n\treturn newStringsTag(\"shard-executors\", executorIDs)\n}\n\nfunc ShardKey(shardKey string) Tag {\n\treturn newStringTag(\"shard-key\", shardKey)\n}\n\nfunc ShardStatus(status string) Tag {\n\treturn newStringTag(\"shard-status\", status)\n}\nfunc ShardLoad(load string) Tag {\n\treturn newStringTag(\"shard-load\", load)\n}\n\nfunc ElectionDelay(t time.Duration) Tag {\n\treturn newDurationTag(\"election-delay\", t)\n}\n\nfunc WorkflowContextLockLatency(duration time.Duration) Tag {\n\treturn newDurationTag(\"workflow-context-lock-latency\", duration)\n}\n\n// DynamicConfigLinearIteratorSpec is a predefined tag to log dynamic config linear iterator spec\nfunc DynamicConfigLinearIteratorSpec(spec interface{}) Tag {\n\treturn newObjectTag(\"dynamic-config-linear-iterator-spec\", spec)\n}\n\nfunc HashRingResult(addr string) Tag {\n\treturn newStringTag(\"hashring-result\", addr)\n}\n\nfunc ShardDistributorResult(addr string) Tag {\n\treturn newStringTag(\"shard-distributor-result\", addr)\n}\n\n// PeerHostname returns a tag for peer hostname\nfunc PeerHostname(hostname string) Tag {\n\treturn newStringTag(\"peer-hostname\", hostname)\n}\n\n// PendingTaskCount returns a tag for pending task count\nfunc PendingTaskCount(count int) Tag {\n\treturn newInt(\"pending-task-count\", count)\n}\n\n// MaxTaskCount returns a tag for max task count\nfunc MaxTaskCount(count int) Tag {\n\treturn newInt(\"max-task-count\", count)\n}\n\n// VirtualQueueID returns a tag for virtual queue id\nfunc VirtualQueueID(id int64) Tag {\n\treturn newInt64(\"virtual-queue-id\", id)\n}\n\nfunc AlertType(alertType int) Tag {\n\treturn newInt(\"alert-type\", alertType)\n}\n\n// CacheID returns a tag for cache identifier\nfunc CacheID(cacheID string) Tag {\n\treturn newStringTag(\"cache-id\", cacheID)\n}\n\nfunc DomainAuditOperationType(operationType fmt.Stringer) Tag {\n\treturn newStringTag(\"domain-audit-operation-type\", operationType.String())\n}\n"
  },
  {
    "path": "common/log/tag/values.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tag\n\n// Pre-defined values for TagWorkflowAction\nvar (\n\t// workflow start / finish\n\tWorkflowActionWorkflowStarted       = workflowAction(\"add-workflow-started-event\")\n\tWorkflowActionWorkflowCanceled      = workflowAction(\"add-workflow-canceled-event\")\n\tWorkflowActionWorkflowCompleted     = workflowAction(\"add-workflow-completed--event\")\n\tWorkflowActionWorkflowFailed        = workflowAction(\"add-workflow-failed-event\")\n\tWorkflowActionWorkflowTimeout       = workflowAction(\"add-workflow-timeout-event\")\n\tWorkflowActionWorkflowTerminated    = workflowAction(\"add-workflow-terminated-event\")\n\tWorkflowActionWorkflowContinueAsNew = workflowAction(\"add-workflow-continue-as-new-event\")\n\n\t// workflow cancellation / sign\n\tWorkflowActionWorkflowCancelRequested        = workflowAction(\"add-workflow-cancel-requested-event\")\n\tWorkflowActionWorkflowSignaled               = workflowAction(\"add-workflow-signaled-event\")\n\tWorkflowActionWorkflowRecordMarker           = workflowAction(\"add-workflow-marker-record-event\")\n\tWorkflowActionUpsertWorkflowSearchAttributes = workflowAction(\"add-workflow-upsert-search-attributes-event\")\n\n\t// decision\n\tWorkflowActionDecisionTaskScheduled = workflowAction(\"add-decisiontask-scheduled-event\")\n\tWorkflowActionDecisionTaskStarted   = workflowAction(\"add-decisiontask-started-event\")\n\tWorkflowActionDecisionTaskCompleted = workflowAction(\"add-decisiontask-completed-event\")\n\tWorkflowActionDecisionTaskTimedOut  = workflowAction(\"add-decisiontask-timedout-event\")\n\tWorkflowActionDecisionTaskFailed    = workflowAction(\"add-decisiontask-failed-event\")\n\n\t// in memory decision\n\tWorkflowActionInMemoryDecisionTaskScheduled = workflowAction(\"add-in-memory-decisiontask-scheduled\")\n\tWorkflowActionInMemoryDecisionTaskStarted   = workflowAction(\"add-in-memory-decisiontask-started\")\n\n\t// activity\n\tWorkflowActionActivityTaskScheduled       = workflowAction(\"add-activitytask-scheduled-event\")\n\tWorkflowActionActivityTaskStarted         = workflowAction(\"add-activitytask-started-event\")\n\tWorkflowActionActivityTaskCompleted       = workflowAction(\"add-activitytask-completed-event\")\n\tWorkflowActionActivityTaskFailed          = workflowAction(\"add-activitytask-failed-event\")\n\tWorkflowActionActivityTaskTimedOut        = workflowAction(\"add-activitytask-timed-event\")\n\tWorkflowActionActivityTaskCanceled        = workflowAction(\"add-activitytask-canceled-event\")\n\tWorkflowActionActivityTaskCancelRequested = workflowAction(\"add-activitytask-cancel-requested-event\")\n\tWorkflowActionActivityTaskCancelFailed    = workflowAction(\"add-activitytask-cancel-failed-event\")\n\tWorkflowActionActivityTaskRetry           = workflowAction(\"add-activitytask-retry-event\")\n\n\t// timer\n\tWorkflowActionTimerStarted      = workflowAction(\"add-timer-started-event\")\n\tWorkflowActionTimerFired        = workflowAction(\"add-timer-fired-event\")\n\tWorkflowActionTimerCanceled     = workflowAction(\"add-timer-canceled-event\")\n\tWorkflowActionTimerCancelFailed = workflowAction(\"add-timer-cancel-failed-event\")\n\n\t// child workflow start / finish\n\tWorkflowActionChildWorkflowInitiated        = workflowAction(\"add-childworkflow-initiated-event\")\n\tWorkflowActionChildWorkflowStarted          = workflowAction(\"add-childworkflow-started-event\")\n\tWorkflowActionChildWorkflowInitiationFailed = workflowAction(\"add-childworkflow-initiation-failed-event\")\n\tWorkflowActionChildWorkflowCanceled         = workflowAction(\"add-childworkflow-canceled-event\")\n\tWorkflowActionChildWorkflowCompleted        = workflowAction(\"add-childworkflow-completed-event\")\n\tWorkflowActionChildWorkflowFailed           = workflowAction(\"add-childworkflow-failed-event\")\n\tWorkflowActionChildWorkflowTerminated       = workflowAction(\"add-childworkflow-terminated-event\")\n\tWorkflowActionChildWorkflowTimedOut         = workflowAction(\"add-childworkflow-timedout-event\")\n\n\t// external workflow cancellation\n\tWorkflowActionExternalWorkflowCancelInitiated = workflowAction(\"add-externalworkflow-cancel-initiated-event\")\n\tWorkflowActionExternalWorkflowCancelRequested = workflowAction(\"add-externalworkflow-cancel-requested-event\")\n\tWorkflowActionExternalWorkflowCancelFailed    = workflowAction(\"add-externalworkflow-cancel-failed-event\")\n\n\t// external workflow signal\n\tWorkflowActionExternalWorkflowSignalInitiated = workflowAction(\"add-externalworkflow-signal-initiated-event\")\n\tWorkflowActionExternalWorkflowSignalRequested = workflowAction(\"add-externalworkflow-signal-requested-event\")\n\tWorkflowActionExternalWorkflowSignalFailed    = workflowAction(\"add-externalworkflow-signal-failed-event\")\n\n\tWorkflowActionUnknown = workflowAction(\"add-unknown-event\")\n)\n\n// Pre-defined values for TagWorkflowListFilterType\nvar (\n\tWorkflowListWorkflowFilterByID     = workflowListFilterType(\"WID\")\n\tWorkflowListWorkflowFilterByType   = workflowListFilterType(\"WType\")\n\tWorkflowListWorkflowFilterByStatus = workflowListFilterType(\"status\")\n)\n\n// Pre-defined values for TagSysComponent\nvar (\n\tComponentTaskList                         = component(\"tasklist\")\n\tComponentHistoryEngine                    = component(\"history-engine\")\n\tComponentHistoryCache                     = component(\"history-cache\")\n\tComponentDecisionHandler                  = component(\"decision-handler\")\n\tComponentEventsCache                      = component(\"events-cache\")\n\tComponentTransferQueue                    = component(\"transfer-queue-processor\")\n\tComponentTransferQueueV2                  = component(\"transfer-queue-processor-v2\")\n\tComponentTimerQueue                       = component(\"timer-queue-processor\")\n\tComponentTimerQueueV2                     = component(\"timer-queue-processor-v2\")\n\tComponentTimerBuilder                     = component(\"timer-builder\")\n\tComponentReplicatorQueue                  = component(\"replicator-queue-processor\")\n\tComponentShardController                  = component(\"shard-controller\")\n\tComponentShard                            = component(\"shard\")\n\tComponentShardItem                        = component(\"shard-item\")\n\tComponentShardEngine                      = component(\"shard-engine\")\n\tComponentMatchingEngine                   = component(\"matching-engine\")\n\tComponentReplicator                       = component(\"replicator\")\n\tComponentReplicationTaskProcessor         = component(\"replication-task-processor\")\n\tComponentReplicationAckManager            = component(\"replication-ack-manager\")\n\tComponentReplicationCacheManager          = component(\"replication-cache-manager\")\n\tComponentReplicationDynamicTaskBatchSizer = component(\"replication-dynamic-task-batch-sizer\")\n\tComponentHistoryReplicator                = component(\"history-replicator\")\n\tComponentHistoryResender                  = component(\"history-resender\")\n\tComponentIndexer                          = component(\"indexer\")\n\tComponentIndexerProcessor                 = component(\"indexer-processor\")\n\tComponentIndexerESProcessor               = component(\"indexer-es-processor\")\n\tComponentESVisibilityClient               = component(\"visibility-client\") // used for client-internal logs, e.g. request errors\n\tComponentESVisibilityManager              = component(\"es-visibility-manager\")\n\tComponentArchiver                         = component(\"archiver\")\n\tComponentBatcher                          = component(\"batcher\")\n\tComponentScheduler                        = component(\"scheduler\")\n\tComponentWorker                           = component(\"worker\")\n\tComponentServiceResolver                  = component(\"service-resolver\")\n\tComponentFailoverCoordinator              = component(\"failover-coordinator\")\n\tComponentFailoverMarkerNotifier           = component(\"failover-marker-notifier\")\n\tComponentCrossClusterQueueProcessor       = component(\"cross-cluster-queue-processor\")\n\tComponentCrossClusterTaskProcessor        = component(\"cross-cluster-task-processor\")\n\tComponentCrossClusterTaskFetcher          = component(\"cross-cluster-task-fetcher\")\n\tComponentShardScanner                     = component(\"shardscanner-scanner\")\n\tComponentShardFixer                       = component(\"shardscanner-fixer\")\n\tComponentPinotVisibilityManager           = component(\"pinot-visibility-manager\")\n\tComponentAsyncWFConsumptionManager        = component(\"async-wf-consumption-manager\")\n\tComponentGlobalRatelimiter                = component(\"global-ratelimiter\")\n\tComponentMapQ                             = component(\"mapq\")\n\tComponentMapQTree                         = component(\"mapq-tree\")\n\tComponentMapQTreeNode                     = component(\"mapq-tree-node\")\n\tComponentRPCFactory                       = component(\"rpc-factory\")\n\tComponentTaskListAdaptiveScaler           = component(\"task-list-adaptive-scaler\")\n\tComponentActiveClusterManager             = component(\"active-cluster-manager\")\n\tComponentMembershipResolver               = component(\"membership-resolver\")\n\tComponentHashring                         = component(\"hashring\")\n\tComponentNamespaceManager                 = component(\"shard-namespace-manager\")\n\tComponentLeaderElection                   = component(\"shard-leader-election\")\n\tComponentLeaderProcessor                  = component(\"shard-leader-processor\")\n)\n\n// Predefined values for QueueTypes\nvar (\n\tQueueTypeActive  = queueType(\"active\")\n\tQueueTypeStandby = queueType(\"standby\")\n)\n\n// Pre-defined values for TagSysLifecycle\nvar (\n\tLifeCycleStarting         = lifecycle(\"Starting\")\n\tLifeCycleStarted          = lifecycle(\"Started\")\n\tLifeCycleStopping         = lifecycle(\"Stopping\")\n\tLifeCycleStopped          = lifecycle(\"Stopped\")\n\tLifeCycleStopTimedout     = lifecycle(\"StopTimedout\")\n\tLifeCycleStartFailed      = lifecycle(\"StartFailed\")\n\tLifeCycleStopFailed       = lifecycle(\"StopFailed\")\n\tLifeCycleProcessingFailed = lifecycle(\"ProcessingFailed\")\n)\n\n// Pre-defined values for SysErrorType\nvar (\n\tErrorTypeInvalidHistoryAction         = errorType(\"InvalidHistoryAction\")\n\tErrorTypeInvalidQueryTask             = errorType(\"InvalidQueryTask\")\n\tErrorTypeQueryTaskFailed              = errorType(\"QueryTaskFailed\")\n\tErrorTypePersistentStoreError         = errorType(\"PersistentStoreError\")\n\tErrorTypeHistorySerializationError    = errorType(\"HistorySerializationError\")\n\tErrorTypeHistoryDeserializationError  = errorType(\"HistoryDeserializationError\")\n\tErrorTypeDuplicateTask                = errorType(\"DuplicateTask\")\n\tErrorTypeMultipleCompletionDecisions  = errorType(\"MultipleCompletionDecisions\")\n\tErrorTypeDuplicateTransferTask        = errorType(\"DuplicateTransferTask\")\n\tErrorTypeDecisionFailed               = errorType(\"DecisionFailed\")\n\tErrorTypeInvalidMutableStateAction    = errorType(\"InvalidMutableStateAction\")\n\tErrorTypeInvalidMemDecisionTaskAction = errorType(\"InvalidMemDecisionTaskAction\")\n)\n\n// Pre-defined values for SysShardUpdate\nvar (\n\t// Shard context events\n\tValueShardRangeUpdated            = shardupdate(\"ShardRangeUpdated\")\n\tValueShardAllocateTimerBeforeRead = shardupdate(\"ShardAllocateTimerBeforeRead\")\n\tValueRingMembershipChangedEvent   = shardupdate(\"RingMembershipChangedEvent\")\n)\n\n// Pre-defined values for OperationResult\nvar (\n\tOperationFailed   = operationResult(\"OperationFailed\")\n\tOperationStuck    = operationResult(\"OperationStuck\")\n\tOperationCritical = operationResult(\"OperationCritical\")\n)\n\n// Pre-defined values for TagSysStoreOperation\nvar (\n\tStoreOperationCreateShard = storeOperation(\"create-shard\")\n\tStoreOperationGetShard    = storeOperation(\"get-shard\")\n\tStoreOperationUpdateShard = storeOperation(\"update-shard\")\n\n\tStoreOperationCreateWorkflowExecution            = storeOperation(\"create-wf-execution\")\n\tStoreOperationGetWorkflowExecution               = storeOperation(\"get-wf-execution\")\n\tStoreOperationUpdateWorkflowExecution            = storeOperation(\"update-wf-execution\")\n\tStoreOperationConflictResolveWorkflowExecution   = storeOperation(\"conflict-resolve-wf-execution\")\n\tStoreOperationResetWorkflowExecution             = storeOperation(\"reset-wf-execution\")\n\tStoreOperationDeleteWorkflowExecution            = storeOperation(\"delete-wf-execution\")\n\tStoreOperationDeleteCurrentWorkflowExecution     = storeOperation(\"delete-current-wf-execution\")\n\tStoreOperationGetCurrentExecution                = storeOperation(\"get-current-execution\")\n\tStoreOperationListCurrentExecution               = storeOperation(\"list-current-execution\")\n\tStoreOperationIsWorkflowExecutionExists          = storeOperation(\"is-wf-execution-exists\")\n\tStoreOperationListConcreteExecution              = storeOperation(\"list-concrete-execution\")\n\tStoreOperationGetTransferTasks                   = storeOperation(\"get-transfer-tasks\")\n\tStoreOperationGetCrossClusterTasks               = storeOperation(\"get-cross-cluster-tasks\")\n\tStoreOperationGetReplicationTasks                = storeOperation(\"get-replication-tasks\")\n\tStoreOperationCompleteTransferTask               = storeOperation(\"complete-transfer-task\")\n\tStoreOperationCompleteCrossClusterTask           = storeOperation(\"complete-cross-cluster-task\")\n\tStoreOperationCompleteReplicationTask            = storeOperation(\"complete-replication-task\")\n\tStoreOperationPutReplicationTaskToDLQ            = storeOperation(\"put-replication-task-to-dlq\")\n\tStoreOperationGetReplicationTasksFromDLQ         = storeOperation(\"get-replication-tasks-from-dlq\")\n\tStoreOperationGetReplicationDLQSize              = storeOperation(\"get-replication-dlq-size\")\n\tStoreOperationDeleteReplicationTaskFromDLQ       = storeOperation(\"delete-replication-task-from-dlq\")\n\tStoreOperationRangeDeleteReplicationTaskFromDLQ  = storeOperation(\"range-delete-replication-task-from-dlq\")\n\tStoreOperationCreateFailoverMarkerTasks          = storeOperation(\"createFailoverMarkerTasks\")\n\tStoreOperationGetTimerIndexTasks                 = storeOperation(\"get-timer-index-tasks\")\n\tStoreOperationCompleteTimerTask                  = storeOperation(\"complete-timer-task\")\n\tStoreOperationGetHistoryTasks                    = storeOperation(\"get-history-tasks\")\n\tStoreOperationCompleteHistoryTask                = storeOperation(\"complete-history-task\")\n\tStoreOperationRangeCompleteHistoryTask           = storeOperation(\"range-complete-history-task\")\n\tStoreOperationGetActiveClusterSelectionPolicy    = storeOperation(\"get-active-cluster-selection-policy\")\n\tStoreOperationDeleteActiveClusterSelectionPolicy = storeOperation(\"delete-active-cluster-selection-policy\")\n\n\tStoreOperationCreateTasks           = storeOperation(\"create-tasks\")\n\tStoreOperationGetTasks              = storeOperation(\"get-tasks\")\n\tStoreOperationGetOrphanTasks        = storeOperation(\"get-orphan-tasks\")\n\tStoreOperationCompleteTask          = storeOperation(\"complete-task\")\n\tStoreOperationCompleteTasksLessThan = storeOperation(\"complete-tasks-less-than\")\n\tStoreOperationLeaseTaskList         = storeOperation(\"lease-task-list\")\n\tStoreOperationGetTaskList           = storeOperation(\"get-task-list\")\n\tStoreOperationUpdateTaskList        = storeOperation(\"update-task-list\")\n\tStoreOperationListTaskList          = storeOperation(\"list-task-list\")\n\tStoreOperationDeleteTaskList        = storeOperation(\"delete-task-list\")\n\tStoreOperationGetTaskListSize       = storeOperation(\"get-task-list-size\")\n\tStoreOperationStopTaskList          = storeOperation(\"stop-task-list\")\n\n\tStoreOperationCreateDomain       = storeOperation(\"create-domain\")\n\tStoreOperationGetDomain          = storeOperation(\"get-domain\")\n\tStoreOperationUpdateDomain       = storeOperation(\"update-domain\")\n\tStoreOperationDeleteDomain       = storeOperation(\"delete-domain\")\n\tStoreOperationDeleteDomainByName = storeOperation(\"delete-domain-by-name\")\n\tStoreOperationListDomains        = storeOperation(\"list-domains\")\n\tStoreOperationGetMetadata        = storeOperation(\"get-metadata\")\n\n\tStoreOperationRecordWorkflowExecutionStarted           = storeOperation(\"record-wf-execution-started\")\n\tStoreOperationRecordWorkflowExecutionClosed            = storeOperation(\"record-wf-execution-closed\")\n\tStoreOperationUpsertWorkflowExecution                  = storeOperation(\"upsert-wf-execution\")\n\tStoreOperationListOpenWorkflowExecutions               = storeOperation(\"list-open-wf-executions\")\n\tStoreOperationListClosedWorkflowExecutions             = storeOperation(\"list-closed-wf-executions\")\n\tStoreOperationListOpenWorkflowExecutionsByType         = storeOperation(\"list-open-wf-executions-by-type\")\n\tStoreOperationListClosedWorkflowExecutionsByType       = storeOperation(\"list-closed-wf-executions-by-type\")\n\tStoreOperationListOpenWorkflowExecutionsByWorkflowID   = storeOperation(\"list-open-wf-executions-by-wfID\")\n\tStoreOperationListClosedWorkflowExecutionsByWorkflowID = storeOperation(\"list-closed-wf-executions-by-wfID\")\n\tStoreOperationListClosedWorkflowExecutionsByStatus     = storeOperation(\"list-closed-wf-executions-by-status\")\n\tStoreOperationGetClosedWorkflowExecution               = storeOperation(\"get-closed-wf-execution\")\n\tStoreOperationVisibilityDeleteWorkflowExecution        = storeOperation(\"vis-delete-wf-execution\")\n\tStoreOperationListWorkflowExecutions                   = storeOperation(\"list-wf-executions\")\n\tStoreOperationScanWorkflowExecutions                   = storeOperation(\"scan-wf-executions\")\n\tStoreOperationCountWorkflowExecutions                  = storeOperation(\"count-wf-executions\")\n\tStoreOperationDeleteUninitializedWorkflowExecution     = storeOperation(\"delete-uninitialized-wf-execution\")\n\tStoreOperationRecordWorkflowExecutionUninitialized     = storeOperation(\"record-wf-execution-uninitialized\")\n\n\tStoreOperationAppendHistoryNodes        = storeOperation(\"append-history-nodes\")\n\tStoreOperationReadHistoryBranch         = storeOperation(\"read-history-branch\")\n\tStoreOperationReadHistoryBranchByBatch  = storeOperation(\"read-history-branch-by-batch\")\n\tStoreOperationReadRawHistoryBranch      = storeOperation(\"read-raw-history-branch\")\n\tStoreOperationForkHistoryBranch         = storeOperation(\"fork-history-branch\")\n\tStoreOperationDeleteHistoryBranch       = storeOperation(\"delete-history-branch\")\n\tStoreOperationGetHistoryTree            = storeOperation(\"get-history-tree\")\n\tStoreOperationGetAllHistoryTreeBranches = storeOperation(\"get-all-history-tree-branches\")\n\n\tStoreOperationEnqueueMessage             = storeOperation(\"enqueue-message\")\n\tStoreOperationReadMessages               = storeOperation(\"read-messages\")\n\tStoreOperationUpdateAckLevel             = storeOperation(\"update-ack-level\")\n\tStoreOperationGetAckLevels               = storeOperation(\"get-ack-levels\")\n\tStoreOperationDeleteMessagesBefore       = storeOperation(\"delete-messages-before\")\n\tStoreOperationEnqueueMessageToDLQ        = storeOperation(\"enqueue-message-to-dlq\")\n\tStoreOperationReadMessagesFromDLQ        = storeOperation(\"read-messages-from-dlq\")\n\tStoreOperationRangeDeleteMessagesFromDLQ = storeOperation(\"range-delete-messages-from-dlq\")\n\tStoreOperationUpdateDLQAckLevel          = storeOperation(\"update-dlq-ack-level\")\n\tStoreOperationGetDLQAckLevels            = storeOperation(\"get-dlq-ack-levels\")\n\tStoreOperationGetDLQSize                 = storeOperation(\"get-dlq-size\")\n\tStoreOperationDeleteMessageFromDLQ       = storeOperation(\"delete-message-from-dlq\")\n\n\tStoreOperationFetchDynamicConfig  = storeOperation(\"fetch-dynamic-config\")\n\tStoreOperationUpdateDynamicConfig = storeOperation(\"update-dynamic-config\")\n)\n\n// Pre-defined values for TagSysClientOperation\nvar (\n\tAdminClientOperationAddSearchAttribute                    = clientOperation(\"admin-add-search-attribute\")\n\tAdminClientOperationDescribeHistoryHost                   = clientOperation(\"admin-describe-history-host\")\n\tAdminClientOperationDescribeShardDistribution             = clientOperation(\"admin-shard-list\")\n\tAdminClientOperationRemoveTask                            = clientOperation(\"admin-remove-task\")\n\tAdminClientOperationCloseShard                            = clientOperation(\"admin-close-shard\")\n\tAdminClientOperationResetQueue                            = clientOperation(\"admin-reset-queue\")\n\tAdminClientOperationDescribeQueue                         = clientOperation(\"admin-describe-queue\")\n\tAdminClientOperationDescribeWorkflowExecution             = clientOperation(\"admin-describe-wf-execution\")\n\tAdminClientOperationGetWorkflowExecutionRawHistoryV2      = clientOperation(\"admin-get-wf-execution-raw-history-v2\")\n\tAdminClientOperationDescribeCluster                       = clientOperation(\"admin-describe-cluster\")\n\tAdminClientOperationGetReplicationMessages                = clientOperation(\"admin-get-replication-messsages\")\n\tAdminClientOperationGetDomainReplicationMessages          = clientOperation(\"admin-get-domain-replication-messsages\")\n\tAdminClientOperationGetDLQReplicationMessages             = clientOperation(\"admin-get-dlq-replication-messsages\")\n\tAdminClientOperationReapplyEvents                         = clientOperation(\"admin-reapply-events\")\n\tAdminClientOperationCountDLQMessages                      = clientOperation(\"admin-count-dlq-messsages\")\n\tAdminClientOperationDeleteWorkflow                        = clientOperation(\"admin-delete-workflow\")\n\tAdminClientOperationReadDLQMessages                       = clientOperation(\"admin-read-dlq-messsages\")\n\tAdminClientOperationPurgeDLQMessages                      = clientOperation(\"admin-purge-dlq-messsages\")\n\tAdminClientOperationMergeDLQMessages                      = clientOperation(\"admin-merge-dlq-messsages\")\n\tAdminClientOperationRefreshWorkflowTasks                  = clientOperation(\"admin-refresh-wf-tasks\")\n\tAdminClientOperationResendReplicationTasks                = clientOperation(\"admin-resend-replication-tasks\")\n\tAdminClientOperationGetCrossClusterTasks                  = clientOperation(\"admin-get-cross-cluster-tasks\")\n\tAdminClientOperationRespondCrossClusterTasksCompleted     = clientOperation(\"admin-respond-cross-cluster-tasks-completed\")\n\tAdminClientOperationGetDynamicConfig                      = clientOperation(\"admin-get-dynamic-config\")\n\tAdminClientOperationUpdateDynamicConfig                   = clientOperation(\"admin-update-dynamic-config\")\n\tAdminClientOperationRestoreDynamicConfig                  = clientOperation(\"admin-restore-dynamic-config\")\n\tAdminClientOperationListDynamicConfig                     = clientOperation(\"admin-list-dynamic-config\")\n\tAdminClientOperationMaintainCorruptWorkflow               = clientOperation(\"admin-maintain-corrupt-workflow\")\n\tAdminClientOperationUpdateGlobalIsolationGroups           = clientOperation(\"admin-update-global-isolation-groups\")\n\tAdminClientOperationGetGlobalIsolationGroups              = clientOperation(\"admin-get-global-isolation-groups\")\n\tAdminClientOperationUpdateDomainIsolationGroups           = clientOperation(\"admin-update-domain-isolation-groups\")\n\tAdminClientOperationGetDomainIsolationGroups              = clientOperation(\"admin-get-domain-isolation-groups\")\n\tAdminClientOperationGetDomainAsyncWorkflowConfiguraton    = clientOperation(\"admin-get-domain-async-workflow-configuration\")\n\tAdminClientOperationUpdateDomainAsyncWorkflowConfiguraton = clientOperation(\"admin-update-domain-async-workflow-configuration\")\n\tAdminDeleteWorkflow                                       = clientOperation(\"admin-delete-workflow\")\n\tMaintainCorruptWorkflow                                   = clientOperation(\"maintain-corrupt-workflow\")\n\tAdminClientOperationUpdateTaskListPartitionConfig         = clientOperation(\"admin-update-task-list-partition-config\")\n\n\tFrontendClientOperationDeleteDomain                          = clientOperation(\"frontend-delete-domain\")\n\tFrontendClientOperationDeprecateDomain                       = clientOperation(\"frontend-deprecate-domain\")\n\tFrontendClientOperationDescribeDomain                        = clientOperation(\"frontend-describe-domain\")\n\tFrontendClientOperationDescribeTaskList                      = clientOperation(\"frontend-describe-task-list\")\n\tFrontendClientOperationDescribeWorkflowExecution             = clientOperation(\"frontend-describe-wf-execution\")\n\tFrontendClientOperationDiagnoseWorkflowExecution             = clientOperation(\"frontend-diagnose-wf-execution\")\n\tFrontendClientOperationGetWorkflowExecutionHistory           = clientOperation(\"frontend-get-wf-execution-history\")\n\tFrontendClientOperationListArchivedWorkflowExecutions        = clientOperation(\"frontend-list-archived-wf-executions\")\n\tFrontendClientOperationListClosedWorkflowExecutions          = clientOperation(\"frontend-list-closed-wf-executions\")\n\tFrontendClientOperationListDomains                           = clientOperation(\"frontend-list-domains\")\n\tFrontendClientOperationListOpenWorkflowExecutions            = clientOperation(\"frontend-list-open-wf-executions\")\n\tFrontendClientOperationListWorkflowExecutions                = clientOperation(\"frontend-list-wf-executions\")\n\tFrontendClientOperationScanWorkflowExecutions                = clientOperation(\"frontend-scan-wf-executions\")\n\tFrontendClientOperationCountWorkflowExecutions               = clientOperation(\"frontend-count-wf-executions\")\n\tFrontendClientOperationGetSearchAttributes                   = clientOperation(\"frontend-get-search-attributes\")\n\tFrontendClientOperationPollForActivityTask                   = clientOperation(\"frontend-poll-for-activity-task\")\n\tFrontendClientOperationPollForDecisionTask                   = clientOperation(\"frontend-poll-for-decision-task\")\n\tFrontendClientOperationQueryWorkflow                         = clientOperation(\"frontend-query-workflow\")\n\tFrontendClientOperationRecordActivityTaskHeartbeat           = clientOperation(\"frontend-record-activity-heartbeat\")\n\tFrontendClientOperationRecordActivityTaskHeartbeatByID       = clientOperation(\"frontend-record-activity-heartbeat-by-id\")\n\tFrontendClientOperationRegisterDomain                        = clientOperation(\"frontend-register-domain\")\n\tFrontendClientOperationRequestCancelWorkflowExecution        = clientOperation(\"frontend-request-cancel-wf-execution\")\n\tFrontendClientOperationResetStickyTaskList                   = clientOperation(\"frontend-reset-sticky-task-list\")\n\tFrontendClientOperationResetWorkflowExecution                = clientOperation(\"frontend-reset-wf-execution\")\n\tFrontendClientOperationRefreshWorkflowTasks                  = clientOperation(\"frontend-refresh-wf-tasks\")\n\tFrontendClientOperationRespondActivityTaskCanceled           = clientOperation(\"frontend-respond-activity-task-canceled\")\n\tFrontendClientOperationRespondActivityTaskCanceledByID       = clientOperation(\"frontend-respond-activity-task-canceled-by-id\")\n\tFrontendClientOperationRespondActivityTaskCompleted          = clientOperation(\"frontend-respond-activity-task-completed\")\n\tFrontendClientOperationRespondActivityTaskCompletedByID      = clientOperation(\"frontend-respond-activity-task-completed-by-id\")\n\tFrontendClientOperationRespondActivityTaskFailed             = clientOperation(\"frontend-respond-activity-task-failed\")\n\tFrontendClientOperationRespondActivityTaskFailedByID         = clientOperation(\"frontend-respond-activity-task-failed-by-id\")\n\tFrontendClientOperationRespondDecisionTaskCompleted          = clientOperation(\"frontend-respond-decision-task-completed\")\n\tFrontendClientOperationRespondDecisionTaskFailed             = clientOperation(\"frontend-respond-decision-task-failed\")\n\tFrontendClientOperationRespondQueryTaskCompleted             = clientOperation(\"frontend-respond-query-task-completed\")\n\tFrontendClientOperationRestartWorkflowExecution              = clientOperation(\"frontend-restart-wf-execution\")\n\tFrontendClientOperationSignalWithStartWorkflowExecution      = clientOperation(\"frontend-signal-with-start-wf-execution\")\n\tFrontendClientOperationSignalWithStartWorkflowExecutionAsync = clientOperation(\"frontend-signal-with-start-wf-execution-async\")\n\tFrontendClientOperationSignalWorkflowExecution               = clientOperation(\"frontend-signal-wf-execution\")\n\tFrontendClientOperationStartWorkflowExecution                = clientOperation(\"frontend-start-wf-execution\")\n\tFrontendClientOperationStartWorkflowExecutionAsync           = clientOperation(\"frontend-start-wf-execution-async\")\n\tFrontendClientOperationTerminateWorkflowExecution            = clientOperation(\"frontend-terminate-wf-execution\")\n\tFrontendClientOperationUpdateDomain                          = clientOperation(\"frontend-update-domain\")\n\tFrontendClientOperationFailoverDomain                        = clientOperation(\"frontend-failover-domain\")\n\tFrontendClientOperationListFailoverHistory                   = clientOperation(\"frontend-list-failover-history\")\n\tFrontendClientOperationGetClusterInfo                        = clientOperation(\"frontend-get-cluster-info\")\n\tFrontendClientOperationListTaskListPartitions                = clientOperation(\"frontend-list-task-list-partitions\")\n\tFrontendClientOperationGetTaskListsByDomain                  = clientOperation(\"frontend-get-task-list-for-domain\")\n\n\tHistoryClientOperationStartWorkflowExecution            = clientOperation(\"history-start-wf-execution\")\n\tHistoryClientOperationDescribeHistoryHost               = clientOperation(\"history-describe-history-host\")\n\tHistoryClientOperationCloseShard                        = clientOperation(\"history-close-shard\")\n\tHistoryClientOperationResetQueue                        = clientOperation(\"history-reset-queue\")\n\tHistoryClientOperationDescribeQueue                     = clientOperation(\"history-describe-queue\")\n\tHistoryClientOperationRemoveTask                        = clientOperation(\"history-remove-task\")\n\tHistoryClientOperationDescribeMutableState              = clientOperation(\"history-describe-mutable-state\")\n\tHistoryClientOperationGetMutableState                   = clientOperation(\"history-get-mutable-state\")\n\tHistoryClientOperationPollMutableState                  = clientOperation(\"history-poll-mutable-state\")\n\tHistoryClientOperationResetStickyTaskList               = clientOperation(\"history-reset-task-list\")\n\tHistoryClientOperationDescribeWorkflowExecution         = clientOperation(\"history-describe-wf-execution\")\n\tHistoryClientOperationRecordDecisionTaskStarted         = clientOperation(\"history-record-decision-task-started\")\n\tHistoryClientOperationRecordActivityTaskStarted         = clientOperation(\"history-record-activity-task-started\")\n\tHistoryClientOperationRecordDecisionTaskCompleted       = clientOperation(\"history-record-decision-task-completed\")\n\tHistoryClientOperationRecordDecisionTaskFailed          = clientOperation(\"history-record-decision-task-failed\")\n\tHistoryClientOperationRecordActivityTaskCompleted       = clientOperation(\"history-record-activity-task-completed\")\n\tHistoryClientOperationRecordActivityTaskFailed          = clientOperation(\"history-record-activity-task-failed\")\n\tHistoryClientOperationRecordActivityTaskCanceled        = clientOperation(\"history-record-activity-task-canceled\")\n\tHistoryClientOperationRecordActivityTaskHeartbeat       = clientOperation(\"history-record-activity-task-heartbeat\")\n\tHistoryClientOperationRequestCancelWorkflowExecution    = clientOperation(\"history-request-cancel-wf-execution\")\n\tHistoryClientOperationSignalWorkflowExecution           = clientOperation(\"history-signal-wf-execution\")\n\tHistoryClientOperationSignalWithStartWorkflowExecution  = clientOperation(\"history-signal-with-start-wf-execution\")\n\tHistoryClientOperationRemoveSignalMutableState          = clientOperation(\"history-remove-signal-mutable-state\")\n\tHistoryClientOperationTerminateWorkflowExecution        = clientOperation(\"history-terminate-wf-execution\")\n\tHistoryClientOperationResetWorkflowExecution            = clientOperation(\"history-reset-wf-execution\")\n\tHistoryClientOperationScheduleDecisionTask              = clientOperation(\"history-schedule-decision-task\")\n\tHistoryClientOperationRecordChildExecutionCompleted     = clientOperation(\"history-record-child-execution-completed\")\n\tHistoryClientOperationReplicateEventsV2                 = clientOperation(\"history-replicate-events-v2\")\n\tHistoryClientOperationSyncShardStatus                   = clientOperation(\"history-sync-shard-status\")\n\tHistoryClientOperationSyncActivity                      = clientOperation(\"history-sync-activity\")\n\tHistoryClientOperationGetReplicationMessages            = clientOperation(\"history-get-replication-messages\")\n\tHistoryClientOperationGetDLQReplicationMessages         = clientOperation(\"history-get-dlq-replication-messages\")\n\tHistoryClientOperationQueryWorkflow                     = clientOperation(\"history-query-wf\")\n\tHistoryClientOperationReapplyEvents                     = clientOperation(\"history-reapply-events\")\n\tHistoryClientOperationCountDLQMessages                  = clientOperation(\"history-count-dlq-messages\")\n\tHistoryClientOperationReadDLQMessages                   = clientOperation(\"history-read-dlq-messages\")\n\tHistoryClientOperationPurgeDLQMessages                  = clientOperation(\"history-purge-dlq-messages\")\n\tHistoryClientOperationMergeDLQMessages                  = clientOperation(\"history-merge-dlq-messages\")\n\tHistoryClientOperationRefreshWorkflowTasks              = clientOperation(\"history-refresh-wf-tasks\")\n\tHistoryClientOperationNotifyFailoverMarkers             = clientOperation(\"history-notify-failover-markers\")\n\tHistoryClientOperationGetCrossClusterTasks              = clientOperation(\"history-get-cross-cluster-tasks\")\n\tHistoryClientOperationRespondCrossClusterTasksCompleted = clientOperation(\"history-respond-cross-cluster-tasks-completed\")\n\tHistoryClientOperationGetFailoverInfo                   = clientOperation(\"history-get-failover-info\")\n\tHistoryClientOperationRespondActivityTaskCanceled       = clientOperation(\"history-respond-activity-task-canceled\")\n\tHistoryClientOperationRespondActivityTaskCompleted      = clientOperation(\"history-respond-activity-task-completed\")\n\tHistoryClientOperationRespondActivityTaskFailed         = clientOperation(\"history-respond-activity-task-failed\")\n\tHistoryClientOperationRespondDecisionTaskCompleted      = clientOperation(\"history-respond-decision-task-completed\")\n\tHistoryClientOperationRespondDecisionTaskFailed         = clientOperation(\"history-respond-decision-task-failed\")\n\tHistoryClientOperationRatelimitUpdate                   = clientOperation(\"history-ratelimit-update\")\n\n\tMatchingClientOperationAddActivityTask                = clientOperation(\"matching-add-activity-task\")\n\tMatchingClientOperationAddDecisionTask                = clientOperation(\"matching-add-decision-task\")\n\tMatchingClientOperationPollForActivityTask            = clientOperation(\"matching-poll-for-activity-task\")\n\tMatchingClientOperationPollForDecisionTask            = clientOperation(\"matching-poll-for-decision-task\")\n\tMatchingClientOperationQueryWorkflow                  = clientOperation(\"matching-query-wf\")\n\tMatchingClientOperationQueryTaskCompleted             = clientOperation(\"matching-query-task-completed\")\n\tMatchingClientOperationCancelOutstandingPoll          = clientOperation(\"matching-cancel-outstanding-poll\")\n\tMatchingClientOperationDescribeTaskList               = clientOperation(\"matching-describe-task-list\")\n\tMatchingClientOperationListTaskListPartitions         = clientOperation(\"matching-list-task-list-partitions\")\n\tMatchingClientOperationGetTaskListsByDomain           = clientOperation(\"matching-get-task-list-for-domain\")\n\tMatchingClientOperationRespondQueryTaskCompleted      = clientOperation(\"matching-respond-query-task-completed\")\n\tMatchingClientOperationUpdateTaskListPartitionConfig  = clientOperation(\"matching-update-task-list-partition-config\")\n\tMatchingClientOperationRefreshTaskListPartitionConfig = clientOperation(\"matching-refresh-task-list-partition-config\")\n\n\tShardDistributorClientOperationGetShardOwner       = clientOperation(\"shard-distributor-get-shard-owner\")\n\tShardDistributorClientOperationWatchNamespaceState = clientOperation(\"shard-distributor-watch-namespace-state\")\n\tShardDistributorExecutorClientOperationHeartbeat   = clientOperation(\"shard-distributor-executor-heartbeat\")\n)\n\n// Pre-defined values for TagIDType\nvar (\n\tIDTypeDomainName   = idType(\"domainName\")\n\tIDTypeIdentity     = idType(\"identity\")\n\tIDTypeWorkflowID   = idType(\"workflowID\")\n\tIDTypeSignalName   = idType(\"signalName\")\n\tIDTypeWorkflowType = idType(\"workflowType\")\n\tIDTypeRequestID    = idType(\"requestID\")\n\tIDTypeTaskListName = idType(\"taskListName\")\n\tIDTypeActivityID   = idType(\"activityID\")\n\tIDTypeActivityType = idType(\"activityType\")\n\tIDTypeMarkerName   = idType(\"markerName\")\n\tIDTypeTimerID      = idType(\"timerID\")\n)\n"
  },
  {
    "path": "common/log/testlogger/fx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testlogger\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/fx\"\n)\n\n// Module allows to push testlogger for tests.\nfunc Module(t *testing.T) fx.Option {\n\treturn fx.Options(\n\t\tfx.Provide(func() TestingT { return TestingT(t) }),\n\t\tfx.Provide(New),\n\t)\n}\n"
  },
  {
    "path": "common/log/testlogger/fx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testlogger\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\nfunc TestModule(t *testing.T) {\n\tapp := fxtest.New(t, Module(t), fx.Invoke(func(logger log.Logger) {}))\n\tapp.RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "common/log/testlogger/testlogger.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testlogger\n\nimport (\n\t\"fmt\"\n\t\"slices\"\n\t\"strings\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/atomic\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n\t\"go.uber.org/zap/zaptest\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype TestingT interface {\n\tzaptest.TestingT\n\tCleanup(func()) // not currently part of zaptest.TestingT\n}\n\n// New is a helper to create new development logger in unit test\nfunc New(t TestingT) log.Logger {\n\t// test logger that emits all logs (none dropped / sample func never returns false)\n\treturn log.NewLogger(NewZap(t), log.WithSampleFunc(func(int) bool { return true }))\n}\n\n// NewZap makes a new test-oriented logger that prevents bad-lifecycle logs from failing tests.\nfunc NewZap(t TestingT) *zap.Logger {\n\t/*\n\t\tHORRIBLE HACK due to async shutdown, both in our code and in libraries (e.g. gocql):\n\t\tnormally, logs produced after a test finishes will *intentionally* fail the test and/or\n\t\tcause data to race on the test's internal `t.done` field.\n\n\t\tthat's a good thing, it reveals possibly-dangerously-flawed lifecycle management.\n\n\t\tunfortunately some of our code and some libraries do not have good lifecycle management,\n\t\tand this cannot easily be patched from the outside.\n\n\t\tso this logger cheats: after a test completes, it logs to stderr rather than TestingT.\n\t\tEVERY ONE of these logs is bad and we should not produce them, but it's causing many\n\t\totherwise-useful tests to be flaky, and that's a larger interruption than is useful.\n\t*/\n\tlogAfterComplete, err := zap.NewDevelopment()\n\trequire.NoError(t, err, \"could not build a fallback zap logger\")\n\treplaced := &fallbackTestCore{\n\t\tt:         t,\n\t\tfallback:  logAfterComplete.Core(),\n\t\ttesting:   zaptest.NewLogger(t).Core(),\n\t\tcompleted: &atomic.Bool{},\n\t}\n\n\tt.Cleanup(replaced.UseFallback) // switch to fallback before ending the test\n\n\treturn zap.New(replaced)\n}\n\n// NewObserved makes a new test logger that both logs to `t` and collects logged\n// events for asserting in tests.\nfunc NewObserved(t TestingT) (log.Logger, *observer.ObservedLogs) {\n\tobsCore, obs := observer.New(zapcore.DebugLevel)\n\tz := NewZap(t)\n\tz = z.WithOptions(zap.WrapCore(func(core zapcore.Core) zapcore.Core {\n\t\treturn zapcore.NewTee(core, obsCore)\n\t}))\n\tl := log.NewLogger(z, log.WithSampleFunc(func(int) bool { return true }))\n\treturn l, obs\n}\n\ntype fallbackTestCore struct {\n\tt         TestingT\n\tfallback  zapcore.Core\n\ttesting   zapcore.Core\n\tcompleted *atomic.Bool\n}\n\nvar _ zapcore.Core = (*fallbackTestCore)(nil)\n\nfunc (f *fallbackTestCore) UseFallback() {\n\tf.completed.Store(true)\n}\n\nfunc (f *fallbackTestCore) Enabled(level zapcore.Level) bool {\n\tif f.completed.Load() {\n\t\treturn f.fallback.Enabled(level)\n\t}\n\treturn f.testing.Enabled(level)\n}\n\nfunc (f *fallbackTestCore) With(fields []zapcore.Field) zapcore.Core {\n\t// need to copy and defer, else the returned core will be used at an\n\t// arbitrarily later point in time, possibly after the test has completed.\n\treturn &fallbackTestCore{\n\t\tt:         f.t,\n\t\tfallback:  f.fallback.With(fields),\n\t\ttesting:   f.testing.With(fields),\n\t\tcompleted: f.completed,\n\t}\n}\n\nfunc (f *fallbackTestCore) Check(entry zapcore.Entry, checked *zapcore.CheckedEntry) *zapcore.CheckedEntry {\n\t// see other Check impls, all look similar.\n\t// this defers the \"where to log\" decision to Write, as `f` is the core that will write.\n\tif f.fallback.Enabled(entry.Level) {\n\t\treturn checked.AddCore(entry, f)\n\t}\n\treturn checked // do not add any cores\n}\n\nfunc (f *fallbackTestCore) Write(entry zapcore.Entry, fields []zapcore.Field) error {\n\tif f.completed.Load() {\n\t\tentry.Message = fmt.Sprintf(\"COULD FAIL TEST %q, logged too late: %v\", f.t.Name(), entry.Message)\n\n\t\thasStack := slices.ContainsFunc(fields, func(field zapcore.Field) bool {\n\t\t\t// no specific stack-trace type, so just look for probable fields.\n\t\t\treturn strings.Contains(strings.ToLower(field.Key), \"stack\")\n\t\t})\n\t\tif !hasStack {\n\t\t\tfields = append(fields, zap.Stack(\"log_stack\"))\n\t\t}\n\t\treturn f.fallback.Write(entry, fields)\n\t}\n\treturn f.testing.Write(entry, fields)\n}\n\nfunc (f *fallbackTestCore) Sync() error {\n\tif f.completed.Load() {\n\t\treturn f.fallback.Sync()\n\t}\n\treturn f.testing.Sync()\n}\n\nvar _ zapcore.Core = (*fallbackTestCore)(nil)\n"
  },
  {
    "path": "common/log/testlogger/testlogger_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testlogger\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\nvar (\n\tdone   = make(chan struct{})\n\tlogged = make(chan struct{})\n)\n\nfunc TestMain(m *testing.M) {\n\tcode := m.Run()\n\t// ensure synchronization between t.done and t.logf, else this test is extremely flaky.\n\t// for details see: https://github.com/golang/go/issues/67701\n\tclose(done)\n\tselect {\n\tcase <-logged:\n\t\tos.Exit(code)\n\tcase <-time.After(time.Second): // should be MUCH faster\n\t\t_, _ = fmt.Fprintln(os.Stderr, \"timed out waiting for test to log\")\n\t\tos.Exit(1)\n\t}\n}\n\n// Unfortunately a moderate hack, to work around our faulty lifecycle management,\n// and some libraries with issues as well.\n// Ideally this test WOULD fail, but that's much harder to assert \"safely\".\nfunc TestLoggerShouldNotFailIfLoggedLate(t *testing.T) {\n\torigLogger := New(t)\n\t// if With does not defer core selection, this will fail the test\n\t// by sending the logs to t.Logf\n\twithLogger := origLogger.WithTags(tag.ActorID(\"testing\")) // literally any tag\n\torigLogger.Info(\"before is fine, orig\")\n\twithLogger.Info(\"before is fine, with\")\n\tgo func() {\n\t\t<-done\n\t\torigLogger.Info(\"too late, orig\")\n\t\twithLogger.Info(\"too late, with\")\n\t\tclose(logged)\n\t}()\n}\n\nfunc TestSubtestShouldNotFail(t *testing.T) {\n\t// when complete, a subtest's too-late logs just get pushed to the parent,\n\t// and do not fail any tests.  they only fail when no running parent exists.\n\t//\n\t// if Go changes this behavior, this test could fail, otherwise AFAICT it\n\t// should be stable.\n\tassertDoesNotFail := func(name string, setup, log func(t *testing.T)) {\n\t\t// need to wrap in something that will out-live the \"real\" test,\n\t\t// to ensure there is a running parent test to push logs toward.\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// same setup as TestMain but contained within this sub-test\n\t\t\tvar (\n\t\t\t\tdone   = make(chan struct{})\n\t\t\t\tlogged = make(chan struct{})\n\t\t\t)\n\t\t\tt.Run(\"inner\", func(t *testing.T) {\n\t\t\t\tsetup(t)\n\t\t\t\tgo func() {\n\t\t\t\t\t<-done\n\t\t\t\t\t// despite being too late, the parent test is still running\n\t\t\t\t\t// so this does not fail the test.\n\t\t\t\t\tlog(t)\n\t\t\t\t\tclose(logged)\n\t\t\t\t}()\n\t\t\t\ttime.AfterFunc(10*time.Millisecond, func() {\n\t\t\t\t\tclose(done)\n\t\t\t\t})\n\t\t\t})\n\t\t\t<-logged\n\t\t})\n\t}\n\n\tassertDoesNotFail(\"real\", func(t *testing.T) {\n\t\t// no setup needed\n\t}, func(t *testing.T) {\n\t\tt.Logf(\"too late but allowed\")\n\t})\n\n\tvar l log.Logger\n\tassertDoesNotFail(\"wrapped\", func(t *testing.T) {\n\t\tl = New(t)\n\t}, func(t *testing.T) {\n\t\tl.Info(\"too late but allowed\")\n\t})\n}\n\nfunc TestObserver(t *testing.T) {\n\tl, obs := NewObserved(t)\n\tl.Info(\"a log\")\n\tl.Info(\"some unrelated log\")\n\tassert.Len(t, obs.FilterMessage(\"a log\").All(), 1, \"did not find a log that was logged\")\n}\n"
  },
  {
    "path": "common/log/throttle.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage log\n\nimport (\n\t\"math/rand\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\ntype throttledLogger struct {\n\trps     int32\n\tlimiter quotas.Limiter\n\tlog     Logger\n}\n\nconst skipForThrottleLogger = 6\n\n// NewThrottledLogger returns an implementation of logger that throttles the\n// log messages being emitted. The underlying implementation uses a token bucket\n// ratelimiter and stops emitting logs once the bucket runs out of tokens\n//\n// Fatal/Panic logs are always emitted without any throttling\nfunc NewThrottledLogger(logger Logger, rps dynamicproperties.IntPropertyFn) Logger {\n\tvar log Logger\n\tlg, ok := logger.(*loggerImpl)\n\tif ok {\n\t\tlog = &loggerImpl{\n\t\t\tzapLogger:     lg.zapLogger,\n\t\t\tskip:          skipForThrottleLogger,\n\t\t\tsampleLocalFn: lg.sampleLocalFn,\n\t\t}\n\t} else {\n\t\tlogger.Warn(\"ReplayLogger may not emit callat tag correctly because the logger passed in is not loggerImpl\")\n\t\tlog = logger\n\t}\n\n\trate := rps()\n\tlimiter := quotas.NewDynamicRateLimiter(func() float64 {\n\t\treturn float64(rps())\n\t})\n\ttl := &throttledLogger{\n\t\tlimiter: limiter,\n\t\trps:     int32(rate),\n\t\tlog:     log,\n\t}\n\treturn tl\n}\n\nfunc (tl *throttledLogger) Debugf(msg string, args ...any) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Debugf(msg, args...)\n\t})\n}\n\nfunc (tl *throttledLogger) Debug(msg string, tags ...tag.Tag) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Debug(msg, tags...)\n\t})\n}\n\nfunc (tl *throttledLogger) Info(msg string, tags ...tag.Tag) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Info(msg, tags...)\n\t})\n}\n\nfunc (tl *throttledLogger) Warn(msg string, tags ...tag.Tag) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Warn(msg, tags...)\n\t})\n}\n\nfunc (tl *throttledLogger) Error(msg string, tags ...tag.Tag) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Error(msg, tags...)\n\t})\n}\n\nfunc (tl *throttledLogger) Fatal(msg string, tags ...tag.Tag) {\n\ttl.rateLimit(func() {\n\t\ttl.log.Fatal(msg, tags...)\n\t})\n}\n\nfunc (tl *throttledLogger) SampleInfo(msg string, sampleRate int, tags ...tag.Tag) {\n\tif rand.Intn(sampleRate) == 0 {\n\t\ttl.rateLimit(func() {\n\t\t\ttl.log.Info(msg, tags...)\n\t\t})\n\t}\n}\n\nfunc (tl *throttledLogger) DebugOn() bool {\n\treturn tl.log.DebugOn()\n}\n\n// Return a logger with the specified key-value pairs set, to be included in a subsequent normal logging call\nfunc (tl *throttledLogger) WithTags(tags ...tag.Tag) Logger {\n\tresult := &throttledLogger{\n\t\trps:     atomic.LoadInt32(&tl.rps),\n\t\tlimiter: tl.limiter,\n\t\tlog:     tl.log.WithTags(tags...),\n\t}\n\treturn result\n}\n\nfunc (tl *throttledLogger) rateLimit(f func()) {\n\tif ok := tl.limiter.Allow(); ok {\n\t\tf()\n\t}\n}\n\nfunc (tl *throttledLogger) Helper() Logger {\n\tresult := &throttledLogger{\n\t\trps:     atomic.LoadInt32(&tl.rps),\n\t\tlimiter: tl.limiter,\n\t\tlog:     tl.log.Helper(),\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/mapq/README.md",
    "content": "# MAPQ: Multi-tenant, Auto-partitioned, Persistent Queue\n\nNOTE: This component is WIP.\n\n## Overview\n\nMAPQ is a new queue framework (introduced in June 2024), aiming to unify Cadence's internal task/request queues. The existing implementations for these applications are cumbersome and maintenance-heavy, with significant overlap and limited extensibility.\nMAPQ will address the challenges of scalability, throughput, consistency, and ordering guarantees required by these diverse needs.\n\n\nChallenges and Motivation\n- History Task Queues: These queues are poorly understood and difficult to maintain, owing to the departure of their original developer and the non-maintainable state of the code. The design struggles with burst loads from timer/child workflow cases, requiring introduction of more granular task types and automated partitioning that the current system cannot support without extensive refactoring.\n- Matching Task Lists: These are basic FIFO queues with some advanced features like sticky task lists, zonal isolation groups and partitioning. The most pressing issue is auto partitioning to reduce operational overhead.\n- Async Request Queues: Initially integrated with Kafka topics as the request queue. Initial testing faced challenges like complex provisioning, inability to dynamically create topics/register consumers, poor visibility into the requests in the queue and difficult to tweak alerts. Async APIs are already designed with pluggable queue implementation already so swapping Kafka with something else will not be tricky.\n\n\n### Goals\n\nMAPQ will provide a solution tailored to meet the following goals:\n\n- Multi-Tenancy: Guarantees fair access to resources for each tenant based on predefined quotas.\n- Auto-Partitioning: Dynamically adjusts partitions based on specified fine-grained policies, supporting both random and deterministic message distribution across physically or virtually partitioned queues.\n- Burst-Protection: Detects incoming message spikes and mitigates by utilizing dynamic auto-partitioning.\n- Skew-Protection: Detects incoming message skews for given partition keys and mitigates by utilizing dynamic auto-partitioning.\n- Advanced Partitioning Policies: Executes on a tree-like partitioning policy to support various levels of partition key hierarchies and strategies.\n- Persistent: Ensures message durability via pluggable persistent layer.\n- Delivery Guarantees: Guarantees at least once delivery.\n\n\n### Problems with Existing Queues in Cadence\n\nHistory Queues:\n\n- Lack of granular partitioning and inextensibility of history queues make it difficult to address following pathological scenarios:\n- Task prioritization: Background tasks like workflow deletion timer tasks share the same queue and consume from the same “processing budget” as other high priority tasks such as user created timers. This is because all timer tasks for a given shard are managed by a single queue.\n- Multi tenancy: Tasks of the same type (e.g. all timers) are managed by a single queue and a noisy domain can drastically regress the experience of other domains. It is not possible to write tasks of a specific domain(s) to a separate queue and adjust read/write qps. Current queue granularity ends at task type (timer or transfer).\n- Burst cases: Bursts of timers or child workflows are known issues that Cadence has no answers to. These bursts usually cause timeouts and may also impact processing of other domains’ tasks.\n\n## High Level Design\n\nMAPQ uses a tree data structure where nodes route incoming messages to child nodes. Nodes can be splitted/merged based on given policies. All leaf nodes are at the same level. Leaf nodes are the actual “queues” where messages are written to/read from via a provided persistent layer plugin.\n\nThe routing key per level, partitioning/departitioning strategy, RPS limits and other options are provided to MAPQ during initialization as a tree-like policy. It contains per level defaults and per-node (identified via path from root) overrides.\n\nOnce initialized the tree will have a minimal number of nodes provided in the policy but it respects policies for not-yet-existing nodes. Since MAPQ supports auto-partitioning there will be new nodes added/removed and it accepts providing policies for such nodes. For example, you might want to partition by domain only for bursty domains and allocate them specific RPS.\n\n\n#### Tree structure with policies\n\n![MAPQ partitioned queue tree](../../docs/images/mapq_partitioned_queue_tree_example.png)\n\n\n#### Initialization and Object Hierarcy\n\n![MAPQ initialization](../../docs/images/mapq_initialization.png)\n\n\n#### Enqueue Flow\n\n![MAPQ enqueue flow](../../docs/images/mapq_enqueue_flow.png)\n\n\n#### Dispatch Flow\n\n![MAPQ enqueue flow](../../docs/images/mapq_dispatch_flow.png)\n"
  },
  {
    "path": "common/mapq/client_impl.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mapq\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mapq/tree\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype clientImpl struct {\n\tlogger          log.Logger\n\tscope           metrics.Scope\n\tpersister       types.Persister\n\tconsumerFactory types.ConsumerFactory\n\ttree            *tree.QueueTree\n\tpartitions      []string\n\tpolicies        []types.NodePolicy\n}\n\nfunc (c *clientImpl) Start(ctx context.Context) error {\n\tc.logger.Info(\"Starting MAPQ client\")\n\terr := c.tree.Start(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tc.logger.Info(\"Started MAPQ client\")\n\treturn nil\n}\n\nfunc (c *clientImpl) Stop(ctx context.Context) error {\n\tc.logger.Info(\"Stopping MAPQ client\")\n\n\t// Stop the tree which will stop the dispatchers\n\tif err := c.tree.Stop(ctx); err != nil {\n\t\treturn fmt.Errorf(\"failed to stop tree: %w\", err)\n\t}\n\n\t// stop the consumer factory which will stop the consumers\n\terr := c.consumerFactory.Stop(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to stop consumer factory: %w\", err)\n\t}\n\n\tc.logger.Info(\"Stopped MAPQ client\")\n\treturn nil\n}\n\nfunc (c *clientImpl) Enqueue(ctx context.Context, items []types.Item) ([]types.ItemToPersist, error) {\n\treturn c.tree.Enqueue(ctx, items)\n}\n\nfunc (c *clientImpl) Ack(context.Context, types.Item) error {\n\treturn errors.New(\"not implemented\")\n}\n\nfunc (c *clientImpl) Nack(context.Context, types.Item) error {\n\treturn errors.New(\"not implemented\")\n}\n"
  },
  {
    "path": "common/mapq/client_impl_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mapq\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestNew(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\ttests := []struct {\n\t\tname    string\n\t\topts    []Options\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\topts: []Options{\n\t\t\t\tWithPersister(types.NewMockPersister(ctrl)),\n\t\t\t\tWithConsumerFactory(types.NewMockConsumerFactory(ctrl)),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"no persister\",\n\t\t\twantErr: true,\n\t\t\topts: []Options{\n\t\t\t\tWithConsumerFactory(types.NewMockConsumerFactory(ctrl)),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"no consumer factoru\",\n\t\t\twantErr: true,\n\t\t\topts: []Options{\n\t\t\t\tWithPersister(types.NewMockPersister(ctrl)),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlogger := testlogger.New(t)\n\t\t\tscope := metrics.NoopScope\n\t\t\tcl, err := New(logger, scope, tc.opts...)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"New() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t_, ok := cl.(*clientImpl)\n\t\t\tif !ok {\n\t\t\t\tt.Errorf(\"New() = %T, want *clientImpl\", cl)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestStartStop(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tconsumerFactory := types.NewMockConsumerFactory(ctrl)\n\tconsumer := types.NewMockConsumer(ctrl)\n\tconsumerFactory.EXPECT().Stop(gomock.Any()).Return(nil).Times(1)\n\tconsumerFactory.EXPECT().New(gomock.Any()).Return(consumer, nil).Times(1)\n\topts := []Options{\n\t\tWithPersister(types.NewMockPersister(ctrl)),\n\t\tWithConsumerFactory(consumerFactory),\n\t}\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tcl, err := New(logger, scope, opts...)\n\tif err != nil {\n\t\tt.Fatalf(\"New() error: %v\", err)\n\t}\n\n\tcl.Start(context.Background())\n\tdefer cl.Stop(context.Background())\n}\n\nfunc TestAck(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tconsumerFactory := types.NewMockConsumerFactory(ctrl)\n\tconsumer := types.NewMockConsumer(ctrl)\n\tconsumerFactory.EXPECT().Stop(gomock.Any()).Return(nil).Times(1)\n\tconsumerFactory.EXPECT().New(gomock.Any()).Return(consumer, nil).Times(1)\n\topts := []Options{\n\t\tWithPersister(types.NewMockPersister(ctrl)),\n\t\tWithConsumerFactory(consumerFactory),\n\t}\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tcl, err := New(logger, scope, opts...)\n\tif err != nil {\n\t\tt.Fatalf(\"New() error: %v\", err)\n\t}\n\n\tcl.Start(context.Background())\n\tdefer cl.Stop(context.Background())\n\n\terr = cl.Ack(context.Background(), nil)\n\tif err == nil || err.Error() != \"not implemented\" {\n\t\tt.Errorf(\"Ack() error: %q, want %q\", err, \"not implemented\")\n\t}\n}\n\nfunc TestNack(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tconsumerFactory := types.NewMockConsumerFactory(ctrl)\n\tconsumer := types.NewMockConsumer(ctrl)\n\tconsumerFactory.EXPECT().Stop(gomock.Any()).Return(nil).Times(1)\n\tconsumerFactory.EXPECT().New(gomock.Any()).Return(consumer, nil).Times(1)\n\topts := []Options{\n\t\tWithPersister(types.NewMockPersister(ctrl)),\n\t\tWithConsumerFactory(consumerFactory),\n\t}\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tcl, err := New(logger, scope, opts...)\n\tif err != nil {\n\t\tt.Fatalf(\"New() error: %v\", err)\n\t}\n\n\tcl.Start(context.Background())\n\tdefer cl.Stop(context.Background())\n\n\terr = cl.Nack(context.Background(), nil)\n\tif err == nil || err.Error() != \"not implemented\" {\n\t\tt.Errorf(\"Ack() error: %q, want %q\", err, \"not implemented\")\n\t}\n}\n"
  },
  {
    "path": "common/mapq/dispatcher/dispatcher.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dispatcher\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n)\n\ntype Dispatcher struct {\n\tconsumer  types.Consumer\n\tctx       context.Context\n\tcancelCtx context.CancelFunc\n\twg        sync.WaitGroup\n}\n\nfunc New(c types.Consumer) *Dispatcher {\n\tctx, cancelCtx := context.WithCancel(context.Background())\n\treturn &Dispatcher{\n\t\tconsumer:  c,\n\t\tctx:       ctx,\n\t\tcancelCtx: cancelCtx,\n\t}\n}\n\nfunc (d *Dispatcher) Start(ctx context.Context) error {\n\td.wg.Add(1)\n\tgo d.run()\n\treturn nil\n}\n\nfunc (d *Dispatcher) Stop(ctx context.Context) error {\n\td.cancelCtx()\n\ttimeout := 10 * time.Second\n\tif dl, ok := ctx.Deadline(); ok {\n\t\ttimeout = time.Until(dl)\n\t}\n\tif !common.AwaitWaitGroup(&d.wg, timeout) {\n\t\treturn fmt.Errorf(\"failed to stop dispatcher in %v\", timeout)\n\t}\n\treturn nil\n}\n\nfunc (d *Dispatcher) run() {\n\tdefer d.wg.Done()\n\t// TODO: implement\n}\n"
  },
  {
    "path": "common/mapq/dispatcher/dispatcher_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage dispatcher\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"go.uber.org/goleak\"\n)\n\nfunc TestStartStop(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\td := New(nil)\n\terr := d.Start(context.Background())\n\tif err != nil {\n\t\tt.Fatalf(\"Start() failed: %v\", err)\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)\n\tdefer cancel()\n\terr = d.Stop(ctx)\n\tif err != nil {\n\t\tt.Fatalf(\"Stop() failed: %v\", err)\n\t}\n}\n"
  },
  {
    "path": "common/mapq/example_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mapq\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestExample(t *testing.T) {\n\tpersister := &InMemoryPersister{}\n\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tcl, err := New(\n\t\tlogger,\n\t\tscope,\n\t\tWithConsumerFactory(&NoOpConsumerFactory{}),\n\t\tWithPersister(persister),\n\t\tWithPartitions([]string{\"type\", \"sub-type\", \"domain\"}),\n\t\tWithPolicies([]types.NodePolicy{\n\t\t\t// level 0: default policy for root (splitted by type)\n\t\t\t{\n\t\t\t\tPath: \"*\",\n\t\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\"timer\", \"transfer\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// level 1: default policy (splitted by sub-type)\n\t\t\t{\n\t\t\t\tPath: \"*/.\",\n\t\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// level 1: timer node\n\t\t\t{\n\t\t\t\tPath: \"*/timer\",\n\t\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\n\t\t\t\t\t\tpersistence.TaskTypeDeleteHistoryEvent,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// level 1: transfer node\n\t\t\t{\n\t\t\t\tPath: \"*/transfer\",\n\t\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\n\t\t\t\t\t\tpersistence.TransferTaskTypeStartChildExecution,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// level 2: nodes per <type, sub-type> pairs\n\t\t\t// - default 1000 RPS for per sub-type node\n\t\t\t// - split by domain. predefined split for d3 domain\n\t\t\t{\n\t\t\t\tPath: \"*/./.\",\n\t\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\"d3\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\t// override for timer delete history event:\n\t\t\t// - only allow 50 RPS\n\t\t\t// - disable split policy\n\t\t\t{\n\t\t\t\tPath:           \"*/timer/4\",\n\t\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 50},\n\t\t\t\tSplitPolicy:    &types.SplitPolicy{Disabled: true},\n\t\t\t},\n\t\t\t// override for start child execution\n\t\t\t// - only allow 10 RPS\n\t\t\t// - disable split policy\n\t\t\t{\n\t\t\t\tPath:           \"*/transfer/4\",\n\t\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 10},\n\t\t\t\tSplitPolicy:    &types.SplitPolicy{Disabled: true},\n\t\t\t},\n\t\t\t// level 3: default policy for all nodes at this level. nodes per <type, sub-type, domain> pairs\n\t\t\t// - only allow 100 rps\n\t\t\t// - disable split policy\n\t\t\t{\n\t\t\t\tPath:           \"*/././.\",\n\t\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 100},\n\t\t\t\tSplitPolicy:    &types.SplitPolicy{Disabled: true},\n\t\t\t},\n\t\t\t// level 3: override policy for catch-all nodes at this level (all domains that don't have a specific node)\n\t\t\t// this policy will override the 100 RPS policy defined above to give more RPS to catch-all nodes\n\t\t\t{\n\t\t\t\tPath:           \"*/././*\",\n\t\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 1000},\n\t\t\t\tSplitPolicy:    &types.SplitPolicy{Disabled: true},\n\t\t\t},\n\t\t}),\n\t)\n\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\tctx := context.Background()\n\tif err := cl.Start(ctx); err != nil {\n\t\tpanic(err)\n\t}\n\tdefer cl.Stop(ctx)\n\n\t_, err = cl.Enqueue(context.Background(), []types.Item{\n\t\tnewTimerItem(\"d1\", time.Now(), persistence.TaskTypeDecisionTimeout),\n\t\tnewTimerItem(\"d1\", time.Now(), persistence.TaskTypeActivityTimeout),\n\t\tnewTimerItem(\"d1\", time.Now(), persistence.TaskTypeUserTimer),\n\t\tnewTimerItem(\"d3\", time.Now(), persistence.TaskTypeUserTimer),\n\t\tnewTimerItem(\"d3\", time.Now(), persistence.TaskTypeUserTimer),\n\t})\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\tif len(persister.items) != 5 {\n\t\tpanic(fmt.Errorf(\"expected 5 items in persister, got %v\", len(persister.items)))\n\t}\n\n\t_, err = cl.Enqueue(context.Background(), []types.Item{\n\t\tnewTransferItem(\"d2\", 1, persistence.TransferTaskTypeDecisionTask),\n\t\tnewTransferItem(\"d2\", 2, persistence.TransferTaskTypeActivityTask),\n\t\tnewTransferItem(\"d2\", 3, persistence.TransferTaskTypeStartChildExecution),\n\t\tnewTransferItem(\"d2\", 4, persistence.TransferTaskTypeStartChildExecution),\n\t\tnewTransferItem(\"d2\", 5, persistence.TransferTaskTypeStartChildExecution),\n\t})\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\tif len(persister.items) != 10 {\n\t\tpanic(fmt.Errorf(\"expected 10 items in persister, got %v\", len(persister.items)))\n\t}\n}\n\nvar _ types.ConsumerFactory = (*NoOpConsumerFactory)(nil)\n\ntype NoOpConsumerFactory struct{}\n\nfunc (f *NoOpConsumerFactory) New(types.ItemPartitions) (types.Consumer, error) {\n\treturn &NoOpConsumer{}, nil\n}\nfunc (f *NoOpConsumerFactory) Stop(context.Context) error {\n\treturn nil\n}\n\ntype NoOpConsumer struct{}\n\nfunc (c *NoOpConsumer) Start(context.Context) error {\n\treturn nil\n}\n\nfunc (c *NoOpConsumer) Stop(context.Context) error {\n\treturn nil\n}\n\nfunc (c *NoOpConsumer) Process(ctx context.Context, item types.Item) error {\n\tfmt.Printf(\"processing item: %v\\n\", item)\n\treturn nil\n}\n\ntype InMemoryPersister struct {\n\titems   []types.ItemToPersist\n\toffsets *types.Offsets\n}\n\nfunc (p *InMemoryPersister) Persist(ctx context.Context, items []types.ItemToPersist) error {\n\tfmt.Printf(\"persisting %v items\\n\", len(items))\n\tfor _, item := range items {\n\t\tpartitionsKV := map[string]any{}\n\t\tactualKV := map[string]any{}\n\t\tfor _, k := range item.GetPartitionKeys() {\n\t\t\tpartitionsKV[k] = item.GetPartitionValue(k)\n\t\t\tactualKV[k] = item.GetAttribute(k)\n\t\t}\n\t\tfmt.Printf(\"item attributes: %v, partitions: %v\\n\", actualKV, partitionsKV)\n\t}\n\tp.items = append(p.items, items...)\n\treturn nil\n}\n\nfunc (p *InMemoryPersister) GetOffsets(context.Context) (*types.Offsets, error) {\n\treturn p.offsets, nil\n}\n\nfunc (p *InMemoryPersister) CommitOffsets(ctx context.Context, offsets *types.Offsets) error {\n\tfmt.Printf(\"committing offsets: %v\\n\", offsets)\n\tp.offsets = offsets\n\treturn nil\n}\n\n// Fetch(ctx context.Context, partitions ItemPartitions, pageInfo PageInfo) ([]Item, error)\nfunc (p *InMemoryPersister) Fetch(ctx context.Context, partitions types.ItemPartitions, pageInfo types.PageInfo) ([]types.Item, error) {\n\treturn nil, nil\n}\n\nfunc newTimerItem(domain string, t time.Time, timerType int) types.Item {\n\tswitch timerType {\n\tcase persistence.TaskTypeDecisionTimeout:\n\tcase persistence.TaskTypeActivityTimeout:\n\tcase persistence.TaskTypeUserTimer:\n\tcase persistence.TaskTypeWorkflowTimeout:\n\tcase persistence.TaskTypeDeleteHistoryEvent:\n\tcase persistence.TaskTypeActivityRetryTimer:\n\tcase persistence.TaskTypeWorkflowBackoffTimer:\n\tdefault:\n\t\tpanic(fmt.Errorf(\"unknown timer type: %v\", timerType))\n\t}\n\n\treturn &timerItem{\n\t\tt:         t,\n\t\ttimerType: timerType,\n\t\tdomain:    domain,\n\t}\n}\n\ntype timerItem struct {\n\tt         time.Time\n\ttimerType int\n\tdomain    string\n}\n\nfunc (t *timerItem) String() string {\n\treturn fmt.Sprintf(\"timerItem{timerType: %v, time: %v}\", t.timerType, t.t)\n}\n\nfunc (t *timerItem) Offset() int64 {\n\treturn t.t.UnixNano()\n}\n\nfunc (t *timerItem) GetAttribute(key string) any {\n\tswitch key {\n\tcase \"type\":\n\t\treturn \"timer\"\n\tcase \"sub-type\":\n\t\treturn t.timerType\n\tcase \"domain\":\n\t\treturn t.domain\n\tdefault:\n\t\tpanic(fmt.Errorf(\"unknown key: %v\", key))\n\t}\n}\n\nfunc newTransferItem(domain string, taskID int64, transferType int) types.Item {\n\tswitch transferType {\n\tcase persistence.TransferTaskTypeActivityTask:\n\tcase persistence.TransferTaskTypeDecisionTask:\n\tcase persistence.TransferTaskTypeCloseExecution:\n\tcase persistence.TransferTaskTypeCancelExecution:\n\tcase persistence.TransferTaskTypeSignalExecution:\n\tcase persistence.TransferTaskTypeStartChildExecution:\n\tcase persistence.TransferTaskTypeRecordWorkflowStarted:\n\tcase persistence.TransferTaskTypeResetWorkflow:\n\tcase persistence.TransferTaskTypeUpsertWorkflowSearchAttributes:\n\tcase persistence.TransferTaskTypeRecordWorkflowClosed:\n\tcase persistence.TransferTaskTypeRecordChildExecutionCompleted:\n\tcase persistence.TransferTaskTypeApplyParentClosePolicy:\n\tdefault:\n\t\tpanic(fmt.Errorf(\"unknown transfer type: %v\", transferType))\n\t}\n\n\treturn &transferItem{\n\t\tdomain:       domain,\n\t\ttaskID:       taskID,\n\t\ttransferType: transferType,\n\t}\n}\n\ntype transferItem struct {\n\ttaskID       int64\n\ttransferType int\n\tdomain       string\n}\n\nfunc (t *transferItem) String() string {\n\treturn fmt.Sprintf(\"transferItem{transferType: %v, taskID: %v}\", t.transferType, t.taskID)\n}\n\nfunc (t *transferItem) Offset() int64 {\n\treturn t.taskID\n}\n\nfunc (t *transferItem) GetAttribute(key string) any {\n\tswitch key {\n\tcase \"type\":\n\t\treturn \"transfer\"\n\tcase \"sub-type\":\n\t\treturn t.transferType\n\tcase \"domain\":\n\t\treturn t.domain\n\tdefault:\n\t\tpanic(fmt.Errorf(\"unknown key: %v\", key))\n\t}\n}\n"
  },
  {
    "path": "common/mapq/mapq.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mapq\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/mapq/tree\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype Options func(*clientImpl)\n\nfunc WithPersister(p types.Persister) Options {\n\treturn func(c *clientImpl) {\n\t\tc.persister = p\n\t}\n}\n\nfunc WithConsumerFactory(cf types.ConsumerFactory) Options {\n\treturn func(c *clientImpl) {\n\t\tc.consumerFactory = cf\n\t}\n}\n\n// WithPartitions sets the partition keys for each level.\n// MAPQ creates a tree with depth = len(partitions)\nfunc WithPartitions(partitions []string) Options {\n\treturn func(c *clientImpl) {\n\t\tc.partitions = partitions\n\t}\n}\n\n// WithPolicies sets the policies for the MAPQ instance.\n// Policies can be defined for nodes at a specific level or nodes with specific path.\n//\n// Path conventions:\n// - \"*\" -> represents the root node at level 0\n// - \"*/.\" matches with all nodes at level 1\n// - \"*/*\" represents the catch-all node at level 1\n// - \"*/xyz\" represents a specific node at level 1 whose partition value is xyz\n// - \"*/./.\" matches with all nodes at level 2\n// - \"*/xyz/.\" matches with all nodes at level 2 whose parent is xyz node\n// - \"*/xyz/*\" represents the catch-all node at level 2 whose parent is xyz node\n// - \"*/xyz/abc\" represents a specific node at level 2 whose level 2 attribute value is abc and parent is xyz node\nfunc WithPolicies(policies []types.NodePolicy) Options {\n\treturn func(c *clientImpl) {\n\t\tc.policies = policies\n\t}\n}\n\nfunc New(logger log.Logger, scope metrics.Scope, opts ...Options) (types.Client, error) {\n\tc := &clientImpl{\n\t\tlogger: logger.WithTags(tag.ComponentMapQ),\n\t\tscope:  scope,\n\t}\n\n\tfor _, opt := range opts {\n\t\topt(c)\n\t}\n\n\tif c.persister == nil {\n\t\treturn nil, fmt.Errorf(\"persister is required. Use WithPersister option to set it\")\n\t}\n\n\tif c.consumerFactory == nil {\n\t\treturn nil, fmt.Errorf(\"consumer factory is required. Use WithConsumerFactory option to set it\")\n\t}\n\n\ttree, err := tree.New(logger, scope, c.partitions, c.policies, c.persister, c.consumerFactory)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tc.tree = tree\n\tc.logger.Info(\"MAPQ client created\",\n\t\ttag.Dynamic(\"partitions\", c.partitions),\n\t\ttag.Dynamic(\"policies\", c.policies),\n\t\ttag.Dynamic(\"tree\", c.tree.String()),\n\t)\n\n\treturn c, nil\n}\n"
  },
  {
    "path": "common/mapq/tree/queue_tree.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tree\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// QueueTree is a tree structure that represents the queue structure for MAPQ\ntype QueueTree struct {\n\toriginalLogger  log.Logger\n\tlogger          log.Logger\n\tscope           metrics.Scope\n\tpartitions      []string\n\tpolicyCol       types.NodePolicyCollection\n\tpersister       types.Persister\n\tconsumerFactory types.ConsumerFactory\n\troot            *QueueTreeNode\n}\n\nfunc New(\n\tlogger log.Logger,\n\tscope metrics.Scope,\n\tpartitions []string,\n\tpolicies []types.NodePolicy,\n\tpersister types.Persister,\n\tconsumerFactory types.ConsumerFactory,\n) (*QueueTree, error) {\n\tt := &QueueTree{\n\t\toriginalLogger:  logger,\n\t\tlogger:          logger.WithTags(tag.ComponentMapQTree),\n\t\tscope:           scope,\n\t\tpartitions:      partitions,\n\t\tpolicyCol:       types.NewNodePolicyCollection(policies),\n\t\tpersister:       persister,\n\t\tconsumerFactory: consumerFactory,\n\t}\n\n\treturn t, t.init()\n}\n\n// Start the dispatchers for all leaf nodes\nfunc (t *QueueTree) Start(ctx context.Context) error {\n\tt.logger.Info(\"Starting MAPQ tree\", tag.Dynamic(\"tree\", t.String()))\n\terr := t.root.Start(ctx, t.consumerFactory, nil, map[string]any{})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to start root node: %w\", err)\n\t}\n\n\tt.logger.Info(\"Started MAPQ tree\")\n\treturn nil\n}\n\n// Stop the dispatchers for all leaf nodes\nfunc (t *QueueTree) Stop(ctx context.Context) error {\n\tt.logger.Info(\"Stopping MAPQ tree\", tag.Dynamic(\"tree\", t.String()))\n\n\terr := t.root.Stop(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to stop nodes: %w\", err)\n\t}\n\n\tt.logger.Info(\"Stopped MAPQ tree\")\n\treturn nil\n}\n\nfunc (t *QueueTree) String() string {\n\tvar sb strings.Builder\n\tvar nodes []*QueueTreeNode\n\tnodes = append(nodes, t.root)\n\tfor len(nodes) > 0 {\n\t\tnode := nodes[0]\n\t\tnodes = nodes[1:]\n\t\tsb.WriteString(node.String())\n\t\tsb.WriteString(\"\\n\")\n\t\tfor _, child := range node.Children {\n\t\t\tnodes = append(nodes, child)\n\t\t}\n\t}\n\n\treturn sb.String()\n}\n\nfunc (t *QueueTree) Enqueue(ctx context.Context, items []types.Item) ([]types.ItemToPersist, error) {\n\tif t.root == nil {\n\t\treturn nil, fmt.Errorf(\"root node is nil\")\n\t}\n\n\tvar itemsToPersist []types.ItemToPersist\n\tfor _, item := range items {\n\t\titemToPersist, err := t.root.Enqueue(ctx, item, nil, map[string]any{})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\titemsToPersist = append(itemsToPersist, itemToPersist)\n\t}\n\n\treturn itemsToPersist, t.persister.Persist(ctx, itemsToPersist)\n}\n\nfunc (t *QueueTree) init() error {\n\tt.root = &QueueTreeNode{\n\t\tPath:     \"*\", // Root node\n\t\tChildren: map[any]*QueueTreeNode{},\n\t}\n\n\tif err := t.root.Init(t.originalLogger, t.scope, t.policyCol, t.partitions); err != nil {\n\t\treturn fmt.Errorf(\"failed to initialize node: %w\", err)\n\t}\n\n\t// Create tree nodes with catch-all nodes at all levels and predefined splits.\n\t// There will be len(partitions) levels in the tree.\n\terr := t.constructInitialNodes(t.root)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to construct initial tree: %w\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (t *QueueTree) constructInitialNodes(n *QueueTreeNode) error {\n\tnodeLevel := nodeLevel(n.Path)\n\tif nodeLevel == len(t.partitions) { // reached the leaf level\n\t\treturn nil\n\t}\n\n\tif n.Children[\"*\"] != nil { // catch-all node already exists\n\t\treturn nil\n\t}\n\n\t_, err := n.addChild(\"*\", t.policyCol, t.partitions)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tfor _, child := range n.Children {\n\t\tif err := t.constructInitialNodes(child); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc nodeLevel(path string) int {\n\treturn len(strings.Split(path, \"/\")) - 1\n}\n"
  },
  {
    "path": "common/mapq/tree/queue_tree_node.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tree\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/mapq/dispatcher\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// QueueTreeNode represents a node in the queue tree\ntype QueueTreeNode struct {\n\t// originalLogger is the logger passed in during creation. No node specific tags are added to this logger and it should be passed to child nodes\n\toriginalLogger log.Logger\n\n\t// logger is the logger for this node. It has the node specific tags added to it\n\tlogger log.Logger\n\tscope  metrics.Scope\n\n\t// The path to the node\n\tPath string\n\n\t// The partition key used by this node to decide which child to enqueue the item.\n\t// Partition key of a node is the attribute key of child node.\n\tPartitionKey string\n\n\t// The attribute key used to create this node by parent\n\tAttributeKey string\n\n\t// The attribute value used to create this node by parent\n\tAttributeVal any\n\n\t// The policy for this node. It's merged policy from all policies that match this node\n\tNodePolicy types.NodePolicy\n\n\t// Children by attribute key\n\t// \"*\" is a special key that represents the default/fallback child queue\n\t// If there's no children then the node is considered leaf node\n\tChildren map[any]*QueueTreeNode\n\n\t// The dispatcher for this node. Only leaf nodes have dispatcher\n\tDispatcher *dispatcher.Dispatcher\n}\n\nfunc (n *QueueTreeNode) Start(\n\tctx context.Context,\n\tconsumerFactory types.ConsumerFactory,\n\tpartitions []string,\n\tpartitionMap map[string]any,\n) error {\n\tn.logger.Info(\"Starting node\", tag.Dynamic(\"node\", n.String()))\n\n\t// If there are no children then this is a leaf node\n\tif len(n.Children) == 0 {\n\t\tn.logger.Info(\"Creating consumer and starting a new dispatcher for leaf node\")\n\t\tc, err := consumerFactory.New(types.NewItemPartitions(partitions, partitionMap))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\td := dispatcher.New(c)\n\t\tif err := d.Start(ctx); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tn.Dispatcher = d\n\t\treturn nil\n\t}\n\n\tfor _, child := range n.Children {\n\t\tpartitionMap[n.PartitionKey] = child.AttributeVal\n\t\terr := child.Start(ctx, consumerFactory, partitions, partitionMap)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to start child %s: %w\", child.Path, err)\n\t\t}\n\t}\n\n\tn.logger.Info(\"Started node\")\n\treturn nil\n}\n\nfunc (n *QueueTreeNode) Stop(ctx context.Context) error {\n\tn.logger.Info(\"Stopping node\")\n\n\tif n.Dispatcher != nil { // leaf node\n\t\treturn n.Dispatcher.Stop(ctx)\n\t}\n\n\tfor _, child := range n.Children {\n\t\tif err := child.Stop(ctx); err != nil {\n\t\t\treturn fmt.Errorf(\"failed to stop child %s: %w\", child.Path, err)\n\t\t}\n\t}\n\n\tn.logger.Info(\"Stopped node\")\n\treturn nil\n}\n\nfunc (n *QueueTreeNode) Enqueue(\n\tctx context.Context,\n\titem types.Item,\n\tpartitions []string,\n\tpartitionMap map[string]any,\n) (types.ItemToPersist, error) {\n\t// If there are no children then this is a leaf node\n\tif len(n.Children) == 0 {\n\t\treturn types.NewItemToPersist(item, types.NewItemPartitions(partitions, partitionMap)), nil\n\t}\n\n\t// Add the attribute value to queueNodePathParts\n\tpartitionVal := item.GetAttribute(n.PartitionKey)\n\tpartitions = append(partitions, n.PartitionKey)\n\tpartitionMap[n.PartitionKey] = partitionVal\n\n\tchild, ok := n.Children[partitionVal]\n\tif !ok {\n\t\t// TODO: thread safety missing\n\t\tchild, ok = n.Children[\"*\"]\n\t\tpartitionMap[n.PartitionKey] = \"*\"\n\t\tif !ok {\n\t\t\t// catch-all nodes are created during initalization so this should never happen\n\t\t\treturn nil, fmt.Errorf(\"no child found for attribute %v in node %v\", partitionVal, n.Path)\n\t\t}\n\t}\n\n\treturn child.Enqueue(ctx, item, partitions, partitionMap)\n}\n\nfunc (n *QueueTreeNode) String() string {\n\treturn fmt.Sprintf(\"QueueTreeNode{Path: %q, AttributeKey: %v, AttributeVal: %v, NodePolicy: %s, Num Children: %d}\", n.Path, n.AttributeKey, n.AttributeVal, n.NodePolicy, len(n.Children))\n}\n\nfunc (n *QueueTreeNode) Init(logger log.Logger, scope metrics.Scope, policyCol types.NodePolicyCollection, partitions []string) error {\n\tn.originalLogger = logger\n\tn.logger = logger.WithTags(tag.ComponentMapQTreeNode, tag.Dynamic(\"path\", n.Path))\n\tn.scope = scope\n\n\t// Get the merged policy for this node\n\tpolicy, err := policyCol.GetMergedPolicyForNode(n.Path)\n\tif err != nil {\n\t\treturn err\n\t}\n\tn.NodePolicy = policy\n\n\t// Set partition key of the node\n\tnodeLevel := nodeLevel(n.Path)\n\tif nodeLevel < len(partitions) {\n\t\tn.PartitionKey = partitions[nodeLevel]\n\t}\n\n\t// Create predefined children nodes\n\treturn n.addPredefinedSplits(policyCol, partitions)\n}\n\nfunc (n *QueueTreeNode) addChild(attrVal any, policyCol types.NodePolicyCollection, partitions []string) (*QueueTreeNode, error) {\n\tpath := fmt.Sprintf(\"%s/%v\", n.Path, attrVal)\n\tch := &QueueTreeNode{\n\t\tPath:         path,\n\t\tAttributeKey: n.PartitionKey,\n\t\tAttributeVal: attrVal,\n\t\tChildren:     map[any]*QueueTreeNode{},\n\t}\n\n\tif err := ch.Init(n.originalLogger, n.scope, policyCol, partitions); err != nil {\n\t\treturn nil, err\n\t}\n\n\tn.Children[attrVal] = ch\n\treturn ch, nil\n}\n\nfunc (n *QueueTreeNode) addPredefinedSplits(policyCol types.NodePolicyCollection, partitions []string) error {\n\tif n.NodePolicy.SplitPolicy == nil || len(n.NodePolicy.SplitPolicy.PredefinedSplits) == 0 {\n\t\treturn nil\n\t}\n\n\tif nodeLevel(n.Path) >= len(partitions) {\n\t\treturn fmt.Errorf(\"predefined split is defined for a leaf level node %s\", n.Path)\n\t}\n\n\tfor _, split := range n.NodePolicy.SplitPolicy.PredefinedSplits {\n\n\t\t_, err := n.addChild(split, policyCol, partitions)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/mapq/tree/queue_tree_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tree\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mapq/types\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestStartStop(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tconsumerFactory := types.NewMockConsumerFactory(ctrl)\n\tconsumer := types.NewMockConsumer(ctrl)\n\t// 7 consumers are created in the test tree for each leaf node defined by getTestPolicies()\n\t// - */timer/deletehistory/*\n\t// - */timer/*/domain1\n\t// - */timer/*/*\n\t// - */transfer/*/domain1\n\t// - */transfer/*/*\n\t// - */*/*/domain1\n\t// - */*/*/*\n\tconsumerFactory.EXPECT().New(gomock.Any()).Return(consumer, nil).Times(7)\n\n\ttree, err := New(\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopScope,\n\t\t[]string{\"type\", \"sub-type\", \"domain\"},\n\t\tgetTestPolicies(),\n\t\ttypes.NewMockPersister(ctrl),\n\t\tconsumerFactory,\n\t)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to create queue tree: %v\", err)\n\t}\n\n\tif err := tree.Start(context.Background()); err != nil {\n\t\tt.Fatalf(\"failed to start queue tree: %v\", err)\n\t}\n\n\tif err := tree.Stop(context.Background()); err != nil {\n\t\tt.Fatalf(\"failed to stop queue tree: %v\", err)\n\t}\n}\n\nfunc TestEnqueue(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tpolicies           []types.NodePolicy\n\t\tleafNodeCount      int\n\t\titems              []types.Item\n\t\tpersistErr         error\n\t\twantItemsToPersist []types.ItemToPersist\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname:          \"success\",\n\t\t\tpolicies:      getTestPolicies(),\n\t\t\tleafNodeCount: 7,\n\t\t\titems: []types.Item{\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain1\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain2\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain3\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain1\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain2\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain3\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain1\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain2\"}),\n\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain3\"}),\n\t\t\t},\n\t\t\twantItemsToPersist: func() []types.ItemToPersist {\n\t\t\t\tresult := make([]types.ItemToPersist, 9)\n\t\t\t\tpartitions := []string{\"type\", \"sub-type\", \"domain\"}\n\n\t\t\t\t// deletehistory timers should go to the leaf node \"*/timer/deletehistory/*\"\n\t\t\t\tresult[0] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain1\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"*\"}),\n\t\t\t\t)\n\t\t\t\tresult[1] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain2\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"*\"}),\n\t\t\t\t)\n\t\t\t\tresult[2] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"domain3\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"deletehistory\", \"domain\": \"*\"}),\n\t\t\t\t)\n\n\t\t\t\t// activitytimeout timers either goes to domain1 specific leaf node (\"*/timer/*/domain1\") or the catch all leaf node \"*/timer/*/*\"\n\t\t\t\tresult[3] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain1\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"*\", \"domain\": \"domain1\"}),\n\t\t\t\t)\n\t\t\t\tresult[4] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain2\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"*\", \"domain\": \"*\"}),\n\t\t\t\t)\n\t\t\t\tresult[5] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"timer\", \"sub-type\": \"activitytimeout\", \"domain\": \"domain3\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"timer\", \"sub-type\": \"*\", \"domain\": \"*\"}),\n\t\t\t\t)\n\n\t\t\t\t// transfer tasks either goes to domain1 specific leaf node (\"*/transfer/*/domain1\") or the catch all leaf node \"*/transfer/*/*\"\n\t\t\t\tresult[6] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain1\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"transfer\", \"sub-type\": \"*\", \"domain\": \"domain1\"}),\n\t\t\t\t)\n\t\t\t\tresult[7] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain2\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"transfer\", \"sub-type\": \"*\", \"domain\": \"*\"}),\n\t\t\t\t)\n\t\t\t\tresult[8] = types.NewItemToPersist(\n\t\t\t\t\tmockItem(t, map[string]any{\"type\": \"transfer\", \"sub-type\": \"activity\", \"domain\": \"domain3\"}),\n\t\t\t\t\ttypes.NewItemPartitions(partitions, map[string]any{\"type\": \"transfer\", \"sub-type\": \"*\", \"domain\": \"*\"}),\n\t\t\t\t)\n\n\t\t\t\treturn result\n\t\t\t}(),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\n\t\t\tdefer goleak.VerifyNone(t)\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tconsumerFactory := types.NewMockConsumerFactory(ctrl)\n\t\t\tconsumer := types.NewMockConsumer(ctrl)\n\t\t\t// consumers will be creeted for each leaf node\n\t\t\tconsumerFactory.EXPECT().New(gomock.Any()).Return(consumer, nil).Times(tc.leafNodeCount)\n\n\t\t\tvar gotItemsToPersistByPersister []types.ItemToPersist\n\t\t\tpersister := types.NewMockPersister(ctrl)\n\t\t\tpersister.EXPECT().Persist(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, itemsToPersist []types.ItemToPersist) error {\n\t\t\t\tgotItemsToPersistByPersister = itemsToPersist\n\t\t\t\treturn tc.persistErr\n\t\t\t})\n\n\t\t\ttree, err := New(\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tmetrics.NoopScope,\n\t\t\t\t[]string{\"type\", \"sub-type\", \"domain\"},\n\t\t\t\tgetTestPolicies(),\n\t\t\t\tpersister,\n\t\t\t\tconsumerFactory,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"failed to create queue tree: %v\", err)\n\t\t\t}\n\n\t\t\tif err := tree.Start(context.Background()); err != nil {\n\t\t\t\tt.Fatalf(\"failed to start queue tree: %v\", err)\n\t\t\t}\n\t\t\tdefer tree.Stop(context.Background())\n\n\t\t\treturnedItemsToPersist, err := tree.Enqueue(context.Background(), tc.items)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Enqueue() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif got, want := len(gotItemsToPersistByPersister), len(tc.wantItemsToPersist); got != want {\n\t\t\t\tt.Fatalf(\"Items to persist count mismatch: gotItemsToPersistByPersister=%v, want=%v\", got, want)\n\t\t\t}\n\n\t\t\tif got, want := len(returnedItemsToPersist), len(tc.wantItemsToPersist); got != want {\n\t\t\t\tt.Fatalf(\"Items to persist count mismatch: returnedItemsToPersist=%v, want=%v\", got, want)\n\t\t\t}\n\n\t\t\tfor i := range gotItemsToPersistByPersister {\n\t\t\t\tif diff := cmp.Diff(tc.wantItemsToPersist[i].String(), gotItemsToPersistByPersister[i].String()); diff != \"\" {\n\t\t\t\t\tt.Errorf(\"%d - gotItemsToPersistByPersister mismatch (-want +got):\\n%s\", i, diff)\n\t\t\t\t}\n\n\t\t\t\tif diff := cmp.Diff(tc.wantItemsToPersist[i].String(), returnedItemsToPersist[i].String()); diff != \"\" {\n\t\t\t\t\tt.Errorf(\"%d - returnedItemsToPersist mismatch (-want +got):\\n%s\", i, diff)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mockItem(t *testing.T, attributes map[string]any) types.Item {\n\titem := types.NewMockItem(gomock.NewController(t))\n\titem.EXPECT().GetAttribute(gomock.Any()).DoAndReturn(func(key string) any {\n\t\treturn attributes[key]\n\t}).AnyTimes()\n\titem.EXPECT().String().Return(\"mockitem\").AnyTimes()\n\treturn item\n}\n\nfunc getTestPolicies() []types.NodePolicy {\n\treturn []types.NodePolicy{\n\t\t{\n\t\t\tPath: \"*\", // level 0\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"timer\", \"transfer\"},\n\t\t\t},\n\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 100},\n\t\t},\n\t\t{\n\t\t\tPath:           \"*/.\", // level 1 default policy\n\t\t\tSplitPolicy:    &types.SplitPolicy{},\n\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 50},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/timer\", // level 1 timer node\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"deletehistory\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/./.\", // level 2 default policy\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"domain1\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/timer/deletehistory\", // level 2 deletehistory timer node policy\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 5},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/././*\", // level 3 default catch-all node policy\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 1000},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/././domain1\", // level 3 domain node policy\n\t\t\tSplitPolicy: &types.SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &types.DispatchPolicy{DispatchRPS: 42},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/mapq/types/client.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"context\"\n)\n\ntype Client interface {\n\t// Enqueue adds an item to the queue\n\tEnqueue(context.Context, []Item) ([]ItemToPersist, error)\n\n\t// Ack marks the item as processed.\n\t// Out of order acks are supported.\n\t// Acking an item that has not been dequeued will have no effect.\n\t// Queue's committed offset is updated to the last ack'ed item's offset periodically until there's a gap (un-acked item).\n\t// In other words, all items up to the committed offset are ack'ed.\n\t// Ack'ed item might still be returned from Dequeue if its offset is higher than last committed offset before process restarts.\n\tAck(context.Context, Item) error\n\n\t// Nack negatively acknowledges an item in the queue\n\t// Nack'ing an already ack'ed item will have no effect.\n\tNack(context.Context, Item) error\n\n\t// Start the client. It will\n\t// - fetch the last committed offsets from the persister,\n\t// - start corresponding consumers\n\t// - dispatch items starting from those offsets.\n\tStart(context.Context) error\n\n\t// Stop the client. It will\n\t// - stop all consumers\n\t// - persist the last committed offsets\n\tStop(context.Context) error\n}\n"
  },
  {
    "path": "common/mapq/types/consumer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport \"context\"\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination consumer_mock.go -package types github.com/uber/cadence/common/mapq/types ConsumerFactory,Consumer\n\ntype ConsumerFactory interface {\n\t// New creates a new consumer with the given partitions or returns an existing consumer\n\t// to process the given partitions\n\t// Consumer lifecycle is managed by the factory so the returned consumer must be started.\n\tNew(ItemPartitions) (Consumer, error)\n\n\t// Stop stops all consumers created by this factory\n\tStop(context.Context) error\n}\n\ntype Consumer interface {\n\tStart(context.Context) error\n\tStop(context.Context) error\n\tProcess(context.Context, Item) error\n}\n"
  },
  {
    "path": "common/mapq/types/consumer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: consumer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package types -source consumer.go -destination consumer_mock.go -package types github.com/uber/cadence/common/mapq/types ConsumerFactory,Consumer\n//\n\n// Package types is a generated GoMock package.\npackage types\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockConsumerFactory is a mock of ConsumerFactory interface.\ntype MockConsumerFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConsumerFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockConsumerFactoryMockRecorder is the mock recorder for MockConsumerFactory.\ntype MockConsumerFactoryMockRecorder struct {\n\tmock *MockConsumerFactory\n}\n\n// NewMockConsumerFactory creates a new mock instance.\nfunc NewMockConsumerFactory(ctrl *gomock.Controller) *MockConsumerFactory {\n\tmock := &MockConsumerFactory{ctrl: ctrl}\n\tmock.recorder = &MockConsumerFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConsumerFactory) EXPECT() *MockConsumerFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// New mocks base method.\nfunc (m *MockConsumerFactory) New(arg0 ItemPartitions) (Consumer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"New\", arg0)\n\tret0, _ := ret[0].(Consumer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// New indicates an expected call of New.\nfunc (mr *MockConsumerFactoryMockRecorder) New(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"New\", reflect.TypeOf((*MockConsumerFactory)(nil).New), arg0)\n}\n\n// Stop mocks base method.\nfunc (m *MockConsumerFactory) Stop(arg0 context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Stop\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockConsumerFactoryMockRecorder) Stop(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockConsumerFactory)(nil).Stop), arg0)\n}\n\n// MockConsumer is a mock of Consumer interface.\ntype MockConsumer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConsumerMockRecorder\n\tisgomock struct{}\n}\n\n// MockConsumerMockRecorder is the mock recorder for MockConsumer.\ntype MockConsumerMockRecorder struct {\n\tmock *MockConsumer\n}\n\n// NewMockConsumer creates a new mock instance.\nfunc NewMockConsumer(ctrl *gomock.Controller) *MockConsumer {\n\tmock := &MockConsumer{ctrl: ctrl}\n\tmock.recorder = &MockConsumerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConsumer) EXPECT() *MockConsumerMockRecorder {\n\treturn m.recorder\n}\n\n// Process mocks base method.\nfunc (m *MockConsumer) Process(arg0 context.Context, arg1 Item) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Process\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Process indicates an expected call of Process.\nfunc (mr *MockConsumerMockRecorder) Process(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Process\", reflect.TypeOf((*MockConsumer)(nil).Process), arg0, arg1)\n}\n\n// Start mocks base method.\nfunc (m *MockConsumer) Start(arg0 context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockConsumerMockRecorder) Start(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockConsumer)(nil).Start), arg0)\n}\n\n// Stop mocks base method.\nfunc (m *MockConsumer) Stop(arg0 context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Stop\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockConsumerMockRecorder) Stop(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockConsumer)(nil).Stop), arg0)\n}\n"
  },
  {
    "path": "common/mapq/types/item.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination item_mock.go -package types github.com/uber/cadence/common/mapq/types Item\n\nimport \"fmt\"\n\ntype Item interface {\n\t// GetAttribute returns the value of the attribute key.\n\t// Value can be any type. It's up to the client to interpret the value.\n\t// Attribute keys used as partition keys will be converted to string because they are used as node\n\t// identifiers in the queue tree.\n\tGetAttribute(key string) any\n\n\t// Offset returns the offset of the item in the queue. e.g. monotonically increasing sequence number or a timestamp\n\tOffset() int64\n\n\t// String returns a human friendly representation of the item for logging purposes\n\tString() string\n}\n\ntype ItemToPersist interface {\n\tItem\n\tItemPartitions\n}\n\nfunc NewItemToPersist(item Item, itemPartitions ItemPartitions) ItemToPersist {\n\treturn &defaultItemToPersist{\n\t\titem:           item,\n\t\titemPartitions: itemPartitions,\n\t}\n}\n\ntype ItemPartitions interface {\n\t// GetPartitionKeys returns the partition keys ordered by their level in the tree\n\tGetPartitionKeys() []string\n\n\t// GetPartitionValue returns the partition value to determine the target queue\n\t// e.g.\n\t//  Below example demonstrates that item is in a catch-all queue for sub-type\n\t//  \tItem.GetAttribute(\"sub-type\") returns \"timer\"\n\t//  \tItemPartitions.GetPartitionValue(\"sub-type\") returns \"*\"\n\t//\n\tGetPartitionValue(key string) any\n\n\t// String returns a human friendly representation of the item for logging purposes\n\tString() string\n}\n\nfunc NewItemPartitions(partitionKeys []string, partitionMap map[string]any) ItemPartitions {\n\treturn &defaultItemPartitions{\n\t\tpartitionKeys: partitionKeys,\n\t\tpartitionMap:  partitionMap,\n\t}\n}\n\ntype defaultItemPartitions struct {\n\tpartitionKeys []string\n\tpartitionMap  map[string]any\n}\n\nfunc (i *defaultItemPartitions) GetPartitionKeys() []string {\n\treturn i.partitionKeys\n}\n\nfunc (i *defaultItemPartitions) GetPartitionValue(key string) any {\n\treturn i.partitionMap[key]\n}\n\nfunc (i *defaultItemPartitions) String() string {\n\treturn fmt.Sprintf(\"ItemPartitions{partitionKeys:%v, partitionMap:%v}\", i.partitionKeys, i.partitionMap)\n}\n\ntype defaultItemToPersist struct {\n\titem           Item\n\titemPartitions ItemPartitions\n}\n\nfunc (i *defaultItemToPersist) String() string {\n\treturn fmt.Sprintf(\"ItemToPersist{item:%v, itemPartitions:%v}\", i.item, i.itemPartitions)\n}\n\nfunc (i *defaultItemToPersist) Offset() int64 {\n\treturn i.item.Offset()\n}\n\nfunc (i *defaultItemToPersist) GetAttribute(key string) any {\n\treturn i.item.GetAttribute(key)\n}\n\nfunc (i *defaultItemToPersist) GetPartitionKeys() []string {\n\treturn i.itemPartitions.GetPartitionKeys()\n}\n\nfunc (i *defaultItemToPersist) GetPartitionValue(key string) any {\n\treturn i.itemPartitions.GetPartitionValue(key)\n}\n"
  },
  {
    "path": "common/mapq/types/item_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: item.go\n//\n// Generated by this command:\n//\n//\tmockgen -package types -source item.go -destination item_mock.go -package types github.com/uber/cadence/common/mapq/types Item\n//\n\n// Package types is a generated GoMock package.\npackage types\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockItem is a mock of Item interface.\ntype MockItem struct {\n\tctrl     *gomock.Controller\n\trecorder *MockItemMockRecorder\n\tisgomock struct{}\n}\n\n// MockItemMockRecorder is the mock recorder for MockItem.\ntype MockItemMockRecorder struct {\n\tmock *MockItem\n}\n\n// NewMockItem creates a new mock instance.\nfunc NewMockItem(ctrl *gomock.Controller) *MockItem {\n\tmock := &MockItem{ctrl: ctrl}\n\tmock.recorder = &MockItemMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockItem) EXPECT() *MockItemMockRecorder {\n\treturn m.recorder\n}\n\n// GetAttribute mocks base method.\nfunc (m *MockItem) GetAttribute(key string) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAttribute\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// GetAttribute indicates an expected call of GetAttribute.\nfunc (mr *MockItemMockRecorder) GetAttribute(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAttribute\", reflect.TypeOf((*MockItem)(nil).GetAttribute), key)\n}\n\n// Offset mocks base method.\nfunc (m *MockItem) Offset() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Offset\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// Offset indicates an expected call of Offset.\nfunc (mr *MockItemMockRecorder) Offset() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Offset\", reflect.TypeOf((*MockItem)(nil).Offset))\n}\n\n// String mocks base method.\nfunc (m *MockItem) String() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"String\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// String indicates an expected call of String.\nfunc (mr *MockItemMockRecorder) String() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"String\", reflect.TypeOf((*MockItem)(nil).String))\n}\n\n// MockItemToPersist is a mock of ItemToPersist interface.\ntype MockItemToPersist struct {\n\tctrl     *gomock.Controller\n\trecorder *MockItemToPersistMockRecorder\n\tisgomock struct{}\n}\n\n// MockItemToPersistMockRecorder is the mock recorder for MockItemToPersist.\ntype MockItemToPersistMockRecorder struct {\n\tmock *MockItemToPersist\n}\n\n// NewMockItemToPersist creates a new mock instance.\nfunc NewMockItemToPersist(ctrl *gomock.Controller) *MockItemToPersist {\n\tmock := &MockItemToPersist{ctrl: ctrl}\n\tmock.recorder = &MockItemToPersistMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockItemToPersist) EXPECT() *MockItemToPersistMockRecorder {\n\treturn m.recorder\n}\n\n// GetAttribute mocks base method.\nfunc (m *MockItemToPersist) GetAttribute(key string) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAttribute\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// GetAttribute indicates an expected call of GetAttribute.\nfunc (mr *MockItemToPersistMockRecorder) GetAttribute(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAttribute\", reflect.TypeOf((*MockItemToPersist)(nil).GetAttribute), key)\n}\n\n// GetPartitionKeys mocks base method.\nfunc (m *MockItemToPersist) GetPartitionKeys() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionKeys\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetPartitionKeys indicates an expected call of GetPartitionKeys.\nfunc (mr *MockItemToPersistMockRecorder) GetPartitionKeys() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionKeys\", reflect.TypeOf((*MockItemToPersist)(nil).GetPartitionKeys))\n}\n\n// GetPartitionValue mocks base method.\nfunc (m *MockItemToPersist) GetPartitionValue(key string) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionValue\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// GetPartitionValue indicates an expected call of GetPartitionValue.\nfunc (mr *MockItemToPersistMockRecorder) GetPartitionValue(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionValue\", reflect.TypeOf((*MockItemToPersist)(nil).GetPartitionValue), key)\n}\n\n// Offset mocks base method.\nfunc (m *MockItemToPersist) Offset() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Offset\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// Offset indicates an expected call of Offset.\nfunc (mr *MockItemToPersistMockRecorder) Offset() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Offset\", reflect.TypeOf((*MockItemToPersist)(nil).Offset))\n}\n\n// String mocks base method.\nfunc (m *MockItemToPersist) String() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"String\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// String indicates an expected call of String.\nfunc (mr *MockItemToPersistMockRecorder) String() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"String\", reflect.TypeOf((*MockItemToPersist)(nil).String))\n}\n\n// MockItemPartitions is a mock of ItemPartitions interface.\ntype MockItemPartitions struct {\n\tctrl     *gomock.Controller\n\trecorder *MockItemPartitionsMockRecorder\n\tisgomock struct{}\n}\n\n// MockItemPartitionsMockRecorder is the mock recorder for MockItemPartitions.\ntype MockItemPartitionsMockRecorder struct {\n\tmock *MockItemPartitions\n}\n\n// NewMockItemPartitions creates a new mock instance.\nfunc NewMockItemPartitions(ctrl *gomock.Controller) *MockItemPartitions {\n\tmock := &MockItemPartitions{ctrl: ctrl}\n\tmock.recorder = &MockItemPartitionsMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockItemPartitions) EXPECT() *MockItemPartitionsMockRecorder {\n\treturn m.recorder\n}\n\n// GetPartitionKeys mocks base method.\nfunc (m *MockItemPartitions) GetPartitionKeys() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionKeys\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetPartitionKeys indicates an expected call of GetPartitionKeys.\nfunc (mr *MockItemPartitionsMockRecorder) GetPartitionKeys() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionKeys\", reflect.TypeOf((*MockItemPartitions)(nil).GetPartitionKeys))\n}\n\n// GetPartitionValue mocks base method.\nfunc (m *MockItemPartitions) GetPartitionValue(key string) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPartitionValue\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// GetPartitionValue indicates an expected call of GetPartitionValue.\nfunc (mr *MockItemPartitionsMockRecorder) GetPartitionValue(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPartitionValue\", reflect.TypeOf((*MockItemPartitions)(nil).GetPartitionValue), key)\n}\n\n// String mocks base method.\nfunc (m *MockItemPartitions) String() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"String\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// String indicates an expected call of String.\nfunc (mr *MockItemPartitionsMockRecorder) String() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"String\", reflect.TypeOf((*MockItemPartitions)(nil).String))\n}\n"
  },
  {
    "path": "common/mapq/types/item_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n)\n\nfunc TestNewItemToPersist(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\titem := NewMockItem(ctrl)\n\titemStr := \"###item###\"\n\titem.EXPECT().String().Return(itemStr).Times(1)\n\titem.EXPECT().GetAttribute(\"attr1\").Return(\"value1\").Times(1)\n\titem.EXPECT().GetAttribute(\"attr2\").Return(\"value2\").Times(1)\n\n\tpartitions := []string{\"attr1\", \"attr2\"}\n\titemPartitions := NewItemPartitions(\n\t\tpartitions,\n\t\tmap[string]any{\n\t\t\t\"attr1\": \"*\",\n\t\t\t\"attr2\": \"value2\",\n\t\t},\n\t)\n\n\titemToPersist := NewItemToPersist(item, itemPartitions)\n\tif itemToPersist == nil {\n\t\tt.Fatal(\"itemToPersist is nil\")\n\t}\n\n\tif got := itemToPersist.GetAttribute(\"attr1\"); got != \"value1\" {\n\t\tt.Errorf(\"itemToPersist.GetAttribute(attr1) = %v, want %v\", got, \"value1\")\n\t}\n\tif got := itemToPersist.GetAttribute(\"attr2\"); got != \"value2\" {\n\t\tt.Errorf(\"itemToPersist.GetAttribute(attr2) = %v, want %v\", got, \"value2\")\n\t}\n\n\tgotPartitions := itemToPersist.GetPartitionKeys()\n\tif diff := cmp.Diff(partitions, gotPartitions); diff != \"\" {\n\t\tt.Fatalf(\"Partition keys mismatch (-want +got):\\n%s\", diff)\n\t}\n\tif got := itemToPersist.GetPartitionValue(\"attr1\"); got != \"*\" {\n\t\tt.Errorf(\"itemToPersist.GetPartitionValue(attr1) = %v, want %v\", got, \"*\")\n\t}\n\tif got := itemToPersist.GetPartitionValue(\"attr2\"); got != \"value2\" {\n\t\tt.Errorf(\"itemToPersist.GetPartitionValue(attr2) = %v, want %v\", got, \"value2\")\n\t}\n\n\titemToPersistStr := itemToPersist.String()\n\titemPartitionsStr := itemPartitions.String()\n\tif !strings.Contains(itemToPersistStr, itemPartitionsStr) {\n\t\tt.Errorf(\"itemToPersist.String() = %v, want to contain %v\", itemToPersistStr, itemPartitionsStr)\n\t}\n\tif !strings.Contains(itemToPersistStr, itemStr) {\n\t\tt.Errorf(\"itemToPersist.String() = %v, want to contain %v\", itemToPersistStr, itemStr)\n\t}\n}\n"
  },
  {
    "path": "common/mapq/types/offsets.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\n// Offsets encapsulates the whole queue tree state including the offsets of each leaf node\ntype Offsets struct {\n\t// TODO: complete\n}\n"
  },
  {
    "path": "common/mapq/types/persister.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport \"context\"\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination persister_mock.go -package types github.com/uber/cadence/common/mapq/types Persister\n\ntype Persister interface {\n\tPersist(ctx context.Context, items []ItemToPersist) error\n\tGetOffsets(ctx context.Context) (*Offsets, error)\n\tCommitOffsets(ctx context.Context, offsets *Offsets) error\n\tFetch(ctx context.Context, partitions ItemPartitions, pageInfo PageInfo) ([]Item, error)\n}\n\ntype PageInfo struct {\n\t// TODO: define ack levels, page size, etc.\n}\n"
  },
  {
    "path": "common/mapq/types/persister_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: persister.go\n//\n// Generated by this command:\n//\n//\tmockgen -package types -source persister.go -destination persister_mock.go -package types github.com/uber/cadence/common/mapq/types Persister\n//\n\n// Package types is a generated GoMock package.\npackage types\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPersister is a mock of Persister interface.\ntype MockPersister struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPersisterMockRecorder\n\tisgomock struct{}\n}\n\n// MockPersisterMockRecorder is the mock recorder for MockPersister.\ntype MockPersisterMockRecorder struct {\n\tmock *MockPersister\n}\n\n// NewMockPersister creates a new mock instance.\nfunc NewMockPersister(ctrl *gomock.Controller) *MockPersister {\n\tmock := &MockPersister{ctrl: ctrl}\n\tmock.recorder = &MockPersisterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPersister) EXPECT() *MockPersisterMockRecorder {\n\treturn m.recorder\n}\n\n// CommitOffsets mocks base method.\nfunc (m *MockPersister) CommitOffsets(ctx context.Context, offsets *Offsets) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CommitOffsets\", ctx, offsets)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CommitOffsets indicates an expected call of CommitOffsets.\nfunc (mr *MockPersisterMockRecorder) CommitOffsets(ctx, offsets any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CommitOffsets\", reflect.TypeOf((*MockPersister)(nil).CommitOffsets), ctx, offsets)\n}\n\n// Fetch mocks base method.\nfunc (m *MockPersister) Fetch(ctx context.Context, partitions ItemPartitions, pageInfo PageInfo) ([]Item, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Fetch\", ctx, partitions, pageInfo)\n\tret0, _ := ret[0].([]Item)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Fetch indicates an expected call of Fetch.\nfunc (mr *MockPersisterMockRecorder) Fetch(ctx, partitions, pageInfo any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Fetch\", reflect.TypeOf((*MockPersister)(nil).Fetch), ctx, partitions, pageInfo)\n}\n\n// GetOffsets mocks base method.\nfunc (m *MockPersister) GetOffsets(ctx context.Context) (*Offsets, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOffsets\", ctx)\n\tret0, _ := ret[0].(*Offsets)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOffsets indicates an expected call of GetOffsets.\nfunc (mr *MockPersisterMockRecorder) GetOffsets(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOffsets\", reflect.TypeOf((*MockPersister)(nil).GetOffsets), ctx)\n}\n\n// Persist mocks base method.\nfunc (m *MockPersister) Persist(ctx context.Context, items []ItemToPersist) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Persist\", ctx, items)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Persist indicates an expected call of Persist.\nfunc (mr *MockPersisterMockRecorder) Persist(ctx, items any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Persist\", reflect.TypeOf((*MockPersister)(nil).Persist), ctx, items)\n}\n"
  },
  {
    "path": "common/mapq/types/policy.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n)\n\ntype DispatchPolicy struct {\n\t// DispatchRPS is the rate limit for items dequeued from the node to be pushed to processors.\n\t// All nodes inherit the DispatchRPS from the parent node as is (not distributed to children).\n\t// If parent has 100 rps limit, then all curent and to-be-created children will have 100 rps limit.\n\tDispatchRPS int64 `json:\"dispatchRPS,omitempty\"`\n\n\t// Concurrency is the maximum number of items to be processed concurrently.\n\tConcurrency int `json:\"concurrency,omitempty\"`\n\n\t// TODO: define retry policy\n}\n\nfunc (dp DispatchPolicy) String() string {\n\treturn fmt.Sprintf(\"DispatchPolicy{DispatchRPS:%d, Concurrency:%d}\", dp.DispatchRPS, dp.Concurrency)\n}\n\ntype SplitPolicy struct {\n\t// Disabled is used to disable the split policy for the node.\n\tDisabled bool `json:\"disabled,omitempty\"`\n\n\t// PredefinedSplits is a list of predefined splits for the attribute key\n\t// Child nodes for these attributes will be created during initialization\n\tPredefinedSplits []any `json:\"predefinedSplits,omitempty\"`\n}\n\nfunc (sp SplitPolicy) String() string {\n\treturn fmt.Sprintf(\"SplitPolicy{Disabled:%v, PredefinedSplits:%v}\", sp.Disabled, sp.PredefinedSplits)\n}\n\ntype NodePolicy struct {\n\t// The path to the node\n\t// Root node has empty path \"\".\n\t// \"/\" is used as path separator.\n\t// \"*\" means the policy applies to the special catch-all node\n\t// \".\" means the policy applies to all nodes in the specified level except the catch-all node\n\tPath string `json:\"path,omitempty\"`\n\n\tSplitPolicy *SplitPolicy `json:\"splitPolicy,omitempty\"`\n\n\t// DispatchPolicy is enforced at the leaf node level.\n\tDispatchPolicy *DispatchPolicy `json:\"dispatchPolicy,omitempty\"`\n}\n\n// Merge merges two NodePolicy objects by marshalling/unmarshalling them.\n// Any field in the other policy will override the field in the current policy.\nfunc (np NodePolicy) Merge(other NodePolicy) (NodePolicy, error) {\n\tmarshalled1, err := json.Marshal(np)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\tvar m1 map[string]any\n\terr = json.Unmarshal(marshalled1, &m1)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\tmarshalled2, err := json.Marshal(other)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\tvar m2 map[string]any\n\terr = json.Unmarshal(marshalled2, &m2)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\tfor k, v2 := range m2 {\n\t\tm1[k] = v2\n\t}\n\n\tmergedMarshalled, err := json.Marshal(m1)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\tvar merged NodePolicy\n\terr = json.Unmarshal(mergedMarshalled, &merged)\n\tif err != nil {\n\t\treturn NodePolicy{}, err\n\t}\n\n\treturn merged, nil\n}\n\nfunc (np NodePolicy) String() string {\n\treturn fmt.Sprintf(\"NodePolicy{Path:%v, DispatchPolicy:%s, SplitPolicy:%s}\", np.Path, np.DispatchPolicy, np.SplitPolicy)\n}\n"
  },
  {
    "path": "common/mapq/types/policy_collection.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"strings\"\n)\n\ntype NodePolicyCollection struct {\n\t// policies is sorted by level and generic to special\n\tpolicies []NodePolicy\n}\n\nfunc NewNodePolicyCollection(policies []NodePolicy) NodePolicyCollection {\n\ttrimPaths(policies)\n\tsortPolicies(policies)\n\treturn NodePolicyCollection{\n\t\tpolicies: policies,\n\t}\n}\n\nfunc (npc NodePolicyCollection) GetMergedPolicyForNode(path string) (NodePolicy, error) {\n\tresult := NodePolicy{}\n\tfor _, policy := range npc.policies {\n\t\tpathParts := strings.Split(path, \"/\")\n\t\tplcPathParts := strings.Split(policy.Path, \"/\")\n\t\tlength := len(pathParts)\n\t\tif len(plcPathParts) <= length {\n\t\t\tlength = len(plcPathParts)\n\t\t} else {\n\t\t\t// if policy path is longer than the node path then it can't be a match\n\t\t\tcontinue\n\t\t}\n\n\t\tmatch := true\n\t\tfor i := 0; i < length; i++ {\n\t\t\tpathPart := pathParts[i]\n\t\t\tplcPathPart := plcPathParts[i]\n\t\t\tif pathPart == \".\" || plcPathPart == \".\" || pathPart == plcPathPart {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tmatch = false\n\t\t\tbreak\n\t\t}\n\n\t\tif !match {\n\t\t\tcontinue\n\t\t}\n\t\tvar err error\n\t\tresult, err = result.Merge(policy)\n\t\tif err != nil {\n\t\t\treturn result, fmt.Errorf(\"failed to merge policies: %w\", err)\n\t\t}\n\t}\n\n\t// set the path in the result\n\tresult.Path = path\n\treturn result, nil\n}\n\nfunc (npc NodePolicyCollection) GetPolicies() []NodePolicy {\n\t// return a copy of the slice so the order is not messed up\n\tpolicies := make([]NodePolicy, len(npc.policies))\n\tcopy(policies, npc.policies)\n\treturn policies\n}\n\nfunc trimPaths(policies []NodePolicy) {\n\tfor i := range policies {\n\t\tpolicies[i].Path = strings.TrimRight(policies[i].Path, \"/\")\n\t}\n}\n\nfunc sortPolicies(policies []NodePolicy) {\n\t// sort policies by level and generic to special\n\tsort.Slice(policies, func(i, j int) bool {\n\t\tparts1 := strings.Split(policies[i].Path, \"/\")\n\t\tparts2 := strings.Split(policies[j].Path, \"/\")\n\t\tl1 := len(parts1)\n\t\tl2 := len(parts2)\n\t\tif l1 != l2 {\n\t\t\treturn l1 < l2\n\t\t}\n\n\t\tpathPartPrecedence := map[any]int{\n\t\t\t\".\": 0,\n\t\t\t\"*\": 1,\n\t\t}\n\t\tfor partIdx := 0; partIdx < l1; partIdx++ {\n\t\t\tp1, ok := pathPartPrecedence[parts1[partIdx]]\n\t\t\tif !ok {\n\t\t\t\tp1 = 2\n\t\t\t}\n\t\t\tp2, ok := pathPartPrecedence[parts2[partIdx]]\n\t\t\tif !ok {\n\t\t\t\tp2 = 2\n\t\t\t}\n\n\t\t\tif p1 != p2 {\n\t\t\t\treturn p1 < p2\n\t\t\t}\n\t\t}\n\n\t\t// If parts have same precedence then just return the order\n\t\treturn i < j\n\t})\n}\n"
  },
  {
    "path": "common/mapq/types/policy_collection_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n)\n\nconst (\n\tfuzzCustomSeed = 0\n)\n\nfunc TestGetPolicies(t *testing.T) {\n\tpolicies := getTestPolicies()\n\twantOrderedPaths := make([]string, len(policies))\n\tfor i, p := range policies {\n\t\twantOrderedPaths[i] = p.Path\n\t}\n\n\tshufflePolicies(policies)\n\tnpc := NewNodePolicyCollection(policies)\n\n\tgotPolicies := npc.GetPolicies()\n\tif len(gotPolicies) != len(policies) {\n\t\tt.Fatalf(\"Policies count mismatch, got %v, want %v\", len(gotPolicies), len(policies))\n\t}\n\n\tgotPaths := make([]string, len(policies))\n\tfor i, p := range gotPolicies {\n\t\tt.Logf(\"%d - %s\", i, p)\n\t\tgotPaths[i] = p.Path\n\t}\n\n\tif diff := cmp.Diff(wantOrderedPaths, gotPaths); diff != \"\" {\n\t\tt.Fatalf(\"Policies not sorted as expected (-want +got):\\n%s\", diff)\n\t}\n}\n\nfunc TestGetMergedPolicyforNode(t *testing.T) {\n\tnpc := NewNodePolicyCollection(getTestPolicies())\n\n\ttests := []struct {\n\t\tname string\n\t\tpath string\n\t\twant NodePolicy\n\t}{\n\t\t{\n\t\t\tname: \"root node\",\n\t\t\tpath: \"*\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\"timer\", \"transfer\"},\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 100},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 1 catch-all node\",\n\t\t\tpath: \"*/*\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath:           \"*/*\",\n\t\t\t\tSplitPolicy:    &SplitPolicy{},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 50},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 1 timer node\",\n\t\t\tpath: \"*/timer\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/timer\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\"deletehistory\"},\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 50},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 2 catch all node\",\n\t\t\tpath: \"*/./*\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/./*\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tPredefinedSplits: []any{\"domain1\"},\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 50},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 2 deletehistory timer node\",\n\t\t\tpath: \"*/timer/deletehistory\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/timer/deletehistory\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tDisabled: true,\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 5},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 3 catch-all node\",\n\t\t\tpath: \"*/*/*/*\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/*/*/*\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tDisabled: true,\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 1000},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 3 domain1 node for activitytimeout\",\n\t\t\tpath: \"*/timer/activitytimeout/domain1\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/timer/activitytimeout/domain1\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tDisabled: true,\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 42},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"level 3 domain1 node for childwfcompleted\",\n\t\t\tpath: \"*/transfer/childwfcompleted/domain1\",\n\t\t\twant: NodePolicy{\n\t\t\t\tPath: \"*/transfer/childwfcompleted/domain1\",\n\t\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\t\tDisabled: true,\n\t\t\t\t},\n\t\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 42},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot, err := npc.GetMergedPolicyForNode(tc.path)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"failed to get merged policy for node %v: %v\", tc.path, err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.want, got); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Policy mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// getTestPolicies returns a set of test policies for testing\n// It intentionally returns the policies in a sorted order to test the sorting logic\nfunc getTestPolicies() []NodePolicy {\n\treturn []NodePolicy{\n\t\t{\n\t\t\tPath: \"*\", // level 0\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"timer\", \"transfer\"},\n\t\t\t},\n\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 100},\n\t\t},\n\t\t{\n\t\t\tPath:           \"*/.\", // level 1 default policy\n\t\t\tSplitPolicy:    &SplitPolicy{},\n\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 50},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/timer\", // level 1 timer node\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"deletehistory\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/./.\", // level 2 default policy\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tPredefinedSplits: []any{\"domain1\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/timer/deletehistory\", // level 2 deletehistory timer node policy\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 5},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/././*\", // level 3 default catch-all node policy\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 1000},\n\t\t},\n\t\t{\n\t\t\tPath: \"*/././domain1\", // level 3 domain node policy\n\t\t\tSplitPolicy: &SplitPolicy{\n\t\t\t\tDisabled: true,\n\t\t\t},\n\t\t\tDispatchPolicy: &DispatchPolicy{DispatchRPS: 42},\n\t\t},\n\t}\n}\n\nfunc shufflePolicies(policies []NodePolicy) {\n\tseed := time.Now().UnixNano()\n\tif fuzzCustomSeed != 0 {\n\t\tseed = fuzzCustomSeed // override seed to test a specific scenario\n\t}\n\trng := rand.New(rand.NewSource(seed))\n\trng.Shuffle(len(policies), func(i, j int) {\n\t\tpolicies[i], policies[j] = policies[j], policies[i]\n\t})\n}\n"
  },
  {
    "path": "common/membership/hashring.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage membership\n\nimport (\n\t\"fmt\"\n\t\"slices\"\n\t\"sort\"\n\t\"strings\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/dgryski/go-farm\"\n\t\"github.com/uber/ringpop-go/hashring\"\n\t\"github.com/uber/ringpop-go/membership\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination peerprovider_mock.go github.com/uber/cadence/common/membership PeerProvider\n\n// ErrInsufficientHosts is thrown when there are not enough hosts to serve the request\nvar ErrInsufficientHosts = &types.InternalServiceError{Message: \"Not enough hosts to serve the request\"}\n\nconst (\n\tminRefreshInternal     = time.Second\n\tdefaultRefreshInterval = 2 * time.Second\n\treplicaPoints          = 100\n)\n\n// PeerProvider is used to retrieve membership information from provider\ntype PeerProvider interface {\n\tcommon.Daemon\n\n\tGetMembers(service string) ([]HostInfo, error)\n\tWhoAmI() (HostInfo, error)\n\tSelfEvict() error\n\tSubscribe(name string, handler func(ChangedEvent)) error\n}\n\ntype Ring struct {\n\tstatus       int32\n\tservice      string\n\tpeerProvider PeerProvider\n\trefreshChan  chan struct{}\n\tshutdownCh   chan struct{}\n\tshutdownWG   sync.WaitGroup\n\ttimeSource   clock.TimeSource\n\tscope        metrics.Scope\n\tlogger       log.Logger\n\n\tvalue atomic.Value // this stores the current hashring\n\n\tmembers struct {\n\t\tsync.RWMutex\n\t\trefreshed time.Time\n\t\tkeys      map[string]HostInfo // for mapping ip:port to HostInfo\n\t}\n\n\tsubscribers struct {\n\t\tsync.RWMutex\n\t\tkeys map[string]chan<- *ChangedEvent\n\t}\n}\n\nfunc NewHashring(\n\tservice string,\n\tprovider PeerProvider,\n\ttimeSource clock.TimeSource,\n\tlogger log.Logger,\n\tscope metrics.Scope,\n) *Ring {\n\tr := &Ring{\n\t\tstatus:       common.DaemonStatusInitialized,\n\t\tservice:      service,\n\t\tpeerProvider: provider,\n\t\tshutdownCh:   make(chan struct{}),\n\t\trefreshChan:  make(chan struct{}, 1),\n\t\ttimeSource:   timeSource,\n\t\tlogger:       logger.WithTags(tag.ComponentHashring),\n\t\tscope:        scope,\n\t}\n\n\tr.members.keys = make(map[string]HostInfo)\n\tr.subscribers.keys = make(map[string]chan<- *ChangedEvent)\n\n\tr.value.Store(emptyHashring())\n\treturn r\n}\n\nfunc emptyHashring() *hashring.HashRing {\n\treturn hashring.New(farm.Fingerprint32, replicaPoints)\n}\n\n// Start starts the hashring\nfunc (r *Ring) Start() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&r.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\tr.logger.Info(\"Starting hashring\", tag.ComponentHashring)\n\tdefer r.logger.Info(\"Started hashring\", tag.ComponentHashring)\n\n\tif err := r.peerProvider.Subscribe(r.service, r.handleUpdates); err != nil {\n\t\tr.logger.Fatal(\"subscribing to peer provider\", tag.Error(err))\n\t}\n\n\tif err := r.Refresh(); err != nil {\n\t\tr.logger.Fatal(\"failed to start service resolver\", tag.Error(err))\n\t}\n\n\tr.shutdownWG.Add(1)\n\tgo r.refreshRingWorker()\n}\n\n// Stop stops the resolver\nfunc (r *Ring) Stop() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&r.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\tr.logger.Info(\"Stopping hashring\", tag.ComponentHashring)\n\tdefer r.logger.Info(\"Stopped hashring\", tag.ComponentHashring)\n\n\tr.peerProvider.Stop()\n\tr.value.Store(emptyHashring())\n\n\tr.subscribers.Lock()\n\tr.subscribers.keys = make(map[string]chan<- *ChangedEvent)\n\tr.subscribers.Unlock()\n\tclose(r.shutdownCh)\n\n\tif success := common.AwaitWaitGroup(&r.shutdownWG, time.Minute); !success {\n\t\tr.logger.Warn(\"service resolver timed out on shutdown.\")\n\t}\n}\n\n// LookupRaw finds the host address in the ring responsible for serving the given key\n// without converting it to HostInfo\nfunc (r *Ring) LookupRaw(key string) (string, error) {\n\taddr, found := r.ring().Lookup(key)\n\tif !found {\n\t\tr.signalSelf()\n\t\treturn \"\", ErrInsufficientHosts\n\t}\n\treturn addr, nil\n}\n\n// Lookup finds the host in the ring responsible for serving the given key\nfunc (r *Ring) Lookup(key string) (HostInfo, error) {\n\taddr, err := r.LookupRaw(key)\n\tif err != nil {\n\t\treturn HostInfo{}, err\n\t}\n\n\treturn r.AddressToHost(addr)\n}\n\nfunc (r *Ring) AddressToHost(addr string) (HostInfo, error) {\n\tr.members.RLock()\n\tdefer r.members.RUnlock()\n\thost, ok := r.members.keys[addr]\n\tif !ok {\n\t\treturn HostInfo{}, fmt.Errorf(\"host not found in member keys, host: %q\", addr)\n\t}\n\treturn host, nil\n}\n\n// Subscribe registers callback watcher. Services can use this to be informed about membership changes\nfunc (r *Ring) Subscribe(watcher string, notifyChannel chan<- *ChangedEvent) error {\n\tr.subscribers.Lock()\n\tdefer r.subscribers.Unlock()\n\n\t_, ok := r.subscribers.keys[watcher]\n\tif ok {\n\t\treturn fmt.Errorf(\"watcher %q is already subscribed\", watcher)\n\t}\n\n\tr.subscribers.keys[watcher] = notifyChannel\n\treturn nil\n}\n\nfunc (r *Ring) handleUpdates(event ChangedEvent) {\n\tr.signalSelf()\n}\n\nfunc (r *Ring) signalSelf() {\n\tvar event struct{}\n\tselect {\n\tcase r.refreshChan <- event:\n\tdefault: // channel already has an event, don't block\n\t}\n}\n\nfunc (r *Ring) notifySubscribers(msg ChangedEvent) {\n\tr.subscribers.Lock()\n\tdefer r.subscribers.Unlock()\n\n\tfor name, ch := range r.subscribers.keys {\n\t\tselect {\n\t\tcase ch <- &msg:\n\t\tdefault:\n\t\t\tr.logger.Warn(\"subscriber notification failed\", tag.Name(name))\n\t\t}\n\t}\n}\n\n// Unsubscribe removes subscriber\nfunc (r *Ring) Unsubscribe(name string) error {\n\tr.subscribers.Lock()\n\tdefer r.subscribers.Unlock()\n\tdelete(r.subscribers.keys, name)\n\treturn nil\n}\n\n// MemberCount returns number of hosts in a ring\nfunc (r *Ring) MemberCount() int {\n\treturn r.ring().ServerCount()\n}\n\nfunc (r *Ring) Members() []HostInfo {\n\tservers := r.ring().Servers()\n\n\tvar hosts = make([]HostInfo, 0, len(servers))\n\tr.members.RLock()\n\tdefer r.members.RUnlock()\n\tfor _, s := range servers {\n\t\thost, ok := r.members.keys[s]\n\t\tif !ok {\n\t\t\tr.logger.Warn(\"host not found in hashring keys\", tag.Address(s))\n\t\t\tcontinue\n\t\t}\n\t\thosts = append(hosts, host)\n\t}\n\n\treturn hosts\n}\n\nfunc (r *Ring) Refresh() error {\n\tif r.members.refreshed.After(r.timeSource.Now().Add(-minRefreshInternal)) {\n\t\t// refreshed too frequently\n\t\tr.logger.Debug(\"refresh skipped, refreshed too frequently\")\n\t\treturn nil\n\t}\n\n\tmembers, err := r.peerProvider.GetMembers(r.service)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"getting members from peer provider: %w\", err)\n\t}\n\n\tnewMembersMap := r.makeMembersMap(members)\n\n\tdiff := r.diffMembers(newMembersMap)\n\tif diff.Empty() {\n\t\treturn nil\n\t}\n\n\tring := emptyHashring()\n\tring.AddMembers(castToMembers(members)...)\n\tr.value.Store(ring)\n\t// sort members for deterministic order in the logs\n\tsort.Slice(members, func(i, j int) bool { return members[i].addr < members[j].addr })\n\tr.logger.Info(\"refreshed ring members\", tag.Value(members), tag.Counter(len(members)), tag.Service(r.service))\n\n\tr.updateMembersMap(newMembersMap)\n\n\tr.emitHashIdentifier()\n\tr.notifySubscribers(diff)\n\n\tr.logger.Info(\"notified subscribers\", tag.Service(r.service))\n\n\treturn nil\n}\n\nfunc (r *Ring) updateMembersMap(newMembers map[string]HostInfo) {\n\tr.members.Lock()\n\tdefer r.members.Unlock()\n\n\tr.members.keys = newMembers\n\tr.members.refreshed = r.timeSource.Now()\n}\n\nfunc (r *Ring) refreshRingWorker() {\n\tdefer r.shutdownWG.Done()\n\n\trefreshTicker := r.timeSource.NewTicker(defaultRefreshInterval)\n\tdefer refreshTicker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-r.shutdownCh:\n\t\t\treturn\n\t\tcase <-r.refreshChan: // local signal or signal from provider\n\t\t\tif err := r.Refresh(); err != nil {\n\t\t\t\tr.logger.Error(\"failed to refresh ring\", tag.Error(err))\n\t\t\t}\n\t\tcase <-refreshTicker.Chan(): // periodically force refreshing membership\n\t\t\tr.signalSelf()\n\t\t}\n\t}\n}\n\nfunc (r *Ring) ring() *hashring.HashRing {\n\treturn r.value.Load().(*hashring.HashRing)\n}\n\nfunc (r *Ring) emitHashIdentifier() float64 {\n\tmembers := r.Members()\n\tself, err := r.peerProvider.WhoAmI()\n\tif err != nil {\n\t\tr.logger.Error(\"Observed a problem looking up self from the membership provider while emitting hash identifier metrics\", tag.Error(err))\n\t\tself = HostInfo{\n\t\t\tidentity: \"unknown\",\n\t\t}\n\t}\n\n\tsort.Slice(members, func(i int, j int) bool {\n\t\treturn members[i].addr > members[j].addr\n\t})\n\tvar sb strings.Builder\n\tfor i := range members {\n\t\tsb.WriteString(members[i].addr)\n\t\tsb.WriteString(\"\\n\")\n\t}\n\thashedView := farm.Hash32([]byte(sb.String()))\n\t// Trimming the metric because collisions are unlikely and I didn't want to use the full Float64\n\t// in-case it overflowed something. The number itself is meaningless, so additional precision\n\t// doesn't really give any advantage, besides reducing the risk of collision\n\ttrimmedForMetric := float64(hashedView % 1000)\n\tr.logger.Debug(\"Hashring view\",\n\t\ttag.Dynamic(\"hashring-view\", sb.String()),\n\t\ttag.Dynamic(\"trimmed-hash-id\", trimmedForMetric),\n\t\ttag.Service(r.service),\n\t\ttag.Dynamic(\"self-addr\", self.addr),\n\t\ttag.Dynamic(\"self-identity\", self.identity),\n\t\ttag.Dynamic(\"self-ip\", self.ip),\n\t)\n\tr.scope.Tagged(\n\t\tmetrics.ServiceTag(r.service),\n\t\tmetrics.HostTag(self.identity),\n\t).UpdateGauge(metrics.HashringViewIdentifier, trimmedForMetric)\n\treturn trimmedForMetric\n}\n\nfunc (r *Ring) makeMembersMap(members []HostInfo) map[string]HostInfo {\n\tmembersMap := make(map[string]HostInfo, len(members))\n\tfor _, m := range members {\n\t\tmembersMap[m.GetAddress()] = m\n\t}\n\treturn membersMap\n}\n\nfunc (r *Ring) diffMembers(newMembers map[string]HostInfo) ChangedEvent {\n\tr.members.RLock()\n\tdefer r.members.RUnlock()\n\n\tvar combinedChange ChangedEvent\n\n\t// find newly added hosts\n\tfor addr := range newMembers {\n\t\tif _, found := r.members.keys[addr]; !found {\n\t\t\tcombinedChange.HostsAdded = append(combinedChange.HostsAdded, addr)\n\t\t}\n\t}\n\t// find removed hosts\n\tfor addr := range r.members.keys {\n\t\tif _, found := newMembers[addr]; !found {\n\t\t\tcombinedChange.HostsRemoved = append(combinedChange.HostsRemoved, addr)\n\t\t}\n\t}\n\n\t// order since it will most probably used in logs\n\tslices.Sort(combinedChange.HostsAdded)\n\tslices.Sort(combinedChange.HostsUpdated)\n\tslices.Sort(combinedChange.HostsRemoved)\n\treturn combinedChange\n}\n\nfunc castToMembers[T membership.Member](members []T) []membership.Member {\n\tresult := make([]membership.Member, 0, len(members))\n\tfor _, h := range members {\n\t\tresult = append(result, h)\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/membership/hashring_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"runtime\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nvar letters = []rune(\"abcdefghijklmnopqrstuvwxyz\")\n\nconst maxTestDuration = 5 * time.Second\n\nfunc randSeq(n int) string {\n\tb := make([]rune, n)\n\tfor i := range b {\n\t\tb[i] = letters[rand.Intn(len(letters))]\n\t}\n\treturn string(b)\n}\n\nfunc randomHostInfo(n int) []HostInfo {\n\tres := make([]HostInfo, 0, n)\n\tfor i := 0; i < n; i++ {\n\t\tres = append(res, NewDetailedHostInfo(randSeq(5), randSeq(12), PortMap{randSeq(3): 666}))\n\t}\n\treturn res\n}\n\nfunc TestDiffMemberMakesCorrectDiff(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurr           []HostInfo\n\t\tnew            []HostInfo\n\t\texpectedChange ChangedEvent\n\t}{\n\t\t{\n\t\t\tname:           \"empty and one added\",\n\t\t\tcurr:           []HostInfo{},\n\t\t\tnew:            []HostInfo{NewHostInfo(\"a\")},\n\t\t\texpectedChange: ChangedEvent{HostsAdded: []string{\"a\"}},\n\t\t},\n\t\t{\n\t\t\tname:           \"non-empty and added\",\n\t\t\tcurr:           []HostInfo{NewHostInfo(\"a\")},\n\t\t\tnew:            []HostInfo{NewHostInfo(\"a\"), NewHostInfo(\"b\")},\n\t\t\texpectedChange: ChangedEvent{HostsAdded: []string{\"b\"}},\n\t\t},\n\t\t{\n\t\t\tname:           \"empty and nothing has changed\",\n\t\t\tcurr:           []HostInfo{},\n\t\t\tnew:            []HostInfo{},\n\t\t\texpectedChange: ChangedEvent{},\n\t\t},\n\t\t{\n\t\t\tname:           \"multiple hosts, but no change\",\n\t\t\tcurr:           []HostInfo{NewHostInfo(\"a\"), NewHostInfo(\"b\"), NewHostInfo(\"c\")},\n\t\t\tnew:            []HostInfo{NewHostInfo(\"c\"), NewHostInfo(\"b\"), NewHostInfo(\"a\")},\n\t\t\texpectedChange: ChangedEvent{},\n\t\t},\n\n\t\t{\n\t\t\tname:           \"multiple hosts, add/delete\",\n\t\t\tcurr:           []HostInfo{NewHostInfo(\"a\"), NewHostInfo(\"b\"), NewHostInfo(\"c\")},\n\t\t\tnew:            []HostInfo{NewHostInfo(\"b\"), NewHostInfo(\"e\"), NewHostInfo(\"f\")},\n\t\t\texpectedChange: ChangedEvent{HostsRemoved: []string{\"a\", \"c\"}, HostsAdded: []string{\"e\", \"f\"}},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tr := &Ring{}\n\t\t\tcurrMembers := r.makeMembersMap(tt.curr)\n\t\t\tr.members.keys = currMembers\n\n\t\t\tcombinedChange := r.diffMembers(r.makeMembersMap(tt.new))\n\t\t\tassert.Equal(t, tt.expectedChange, combinedChange)\n\t\t})\n\t}\n}\n\ntype hashringTestData struct {\n\tt                *testing.T\n\tmockPeerProvider *MockPeerProvider\n\tmockTimeSource   clock.MockedTimeSource\n\thashRing         *Ring\n\tobservedLogs     *observer.ObservedLogs\n}\n\nfunc newHashringTestData(t *testing.T) *hashringTestData {\n\tvar td hashringTestData\n\n\tctrl := gomock.NewController(t)\n\ttd.t = t\n\ttd.mockPeerProvider = NewMockPeerProvider(ctrl)\n\ttd.mockTimeSource = clock.NewMockedTimeSourceAt(time.Now())\n\n\tlogger, observedLogs := testlogger.NewObserved(t)\n\ttd.observedLogs = observedLogs\n\n\ttd.hashRing = NewHashring(\n\t\t\"test-service\",\n\t\ttd.mockPeerProvider,\n\t\ttd.mockTimeSource,\n\t\tlogger,\n\t\tmetrics.NoopScope,\n\t)\n\n\treturn &td\n}\n\n// starts hashring' background work and verifies all the goroutines closed at the end\nfunc (td *hashringTestData) startHashRing() {\n\ttd.mockPeerProvider.EXPECT().Stop()\n\n\ttd.t.Cleanup(func() {\n\t\ttd.hashRing.Stop()\n\t\tgoleak.VerifyNone(td.t)\n\t})\n\n\ttd.hashRing.Start()\n}\n\nfunc (td *hashringTestData) bypassRefreshRatelimiter() {\n\ttd.hashRing.members.refreshed = time.Now().AddDate(0, 0, -1)\n}\n\nfunc TestFailedLookupWillAskProvider(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar wg sync.WaitGroup\n\twg.Add(2)\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").\n\t\tDo(func(string) {\n\t\t\t// we expect first call on hashring creation\n\t\t\t// the second call should be initiated by failed Lookup\n\t\t\twg.Done()\n\t\t}).Times(2)\n\n\ttd.startHashRing()\n\t_, err := td.hashRing.Lookup(\"a\")\n\tassert.Error(t, err)\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"Failed Lookup should lead to refresh\")\n}\n\nfunc TestFailingToSubscribeIsFatal(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttd := newHashringTestData(t)\n\n\t// we need to intercept logger calls, use mock\n\tmockLogger := log.NewMockLogger(gomock.NewController(t))\n\ttd.hashRing.logger = mockLogger\n\n\tmockLogger.EXPECT().Fatal(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(msg string, args ...any) {\n\t\t\t// we need to stop goroutine like log.Fatal() does with an entire program\n\t\t\truntime.Goexit()\n\t\t},\n\t).Times(1)\n\n\tmockLogger.EXPECT().Info(gomock.Any(), gomock.Any()).Times(2)\n\n\ttd.mockPeerProvider.EXPECT().\n\t\tSubscribe(gomock.Any(), gomock.Any()).\n\t\tReturn(errors.New(\"can't subscribe\"))\n\n\t// because we use runtime.Goexit() we need to call .Start in a separate goroutine\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\ttd.hashRing.Start()\n\t}()\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"must be finished - failed to subscribe\")\n}\n\nfunc TestHandleUpdatesNeverBlocks(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar wg sync.WaitGroup\n\tfor i := 0; i < 10; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\ttd.hashRing.handleUpdates(ChangedEvent{})\n\t\t\twg.Done()\n\t\t}()\n\t}\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"handleUpdates should never block\")\n}\n\nfunc TestHandlerSchedulesUpdates(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar wg sync.WaitGroup\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\twg.Done()\n\t\tfmt.Println(\"GetMembers called\")\n\t\treturn randomHostInfo(5), nil\n\t}).Times(2)\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\twg.Add(1) // we expect 1st GetMembers to be called during hashring start\n\ttd.startHashRing()\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"GetMembers must be called\")\n\n\twg.Add(1) // another call to GetMembers should happen because of handleUpdate\n\ttd.bypassRefreshRatelimiter()\n\ttd.hashRing.handleUpdates(ChangedEvent{})\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"GetMembers must be called again\")\n}\n\nfunc TestFailedRefreshLogsError(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar wg sync.WaitGroup\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\twg.Done()\n\t\treturn randomHostInfo(5), nil\n\t}).Times(1)\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\twg.Add(1) // we expect 1st GetMembers to be called during hashring start\n\ttd.startHashRing()\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"GetMembers must be called\")\n\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\twg.Done()\n\t\treturn nil, errors.New(\"GetMembers failed\")\n\t}).Times(1)\n\n\twg.Add(1) // another call to GetMembers should happen because of handleUpdate\n\ttd.bypassRefreshRatelimiter()\n\ttd.hashRing.handleUpdates(ChangedEvent{})\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration), \"GetMembers must be called again (and fail)\")\n\ttd.hashRing.Stop()\n\tassert.Equal(t, 1, td.observedLogs.FilterMessageSnippet(\"failed to refresh ring\").Len())\n}\n\nfunc TestRefreshUpdatesRingOnlyWhenRingHasChanged(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").Times(1).Return(randomHostInfo(3), nil)\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\t// Start will also call .Refresh()\n\ttd.startHashRing()\n\tupdatedAt := td.hashRing.members.refreshed\n\n\tassert.NoError(t, td.hashRing.Refresh())\n\tassert.Equal(t, updatedAt, td.hashRing.members.refreshed)\n}\n\nfunc TestRefreshWillNotifySubscribers(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar hostsToReturn []HostInfo\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\thostsToReturn = randomHostInfo(5)\n\t\treturn hostsToReturn, nil\n\t}).Times(2)\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\ttd.startHashRing()\n\n\tvar changeCh = make(chan *ChangedEvent, 2)\n\t// Check if multiple subscribers will get notified\n\tassert.NoError(t, td.hashRing.Subscribe(\"subscriber1\", changeCh))\n\tassert.NoError(t, td.hashRing.Subscribe(\"subscriber2\", changeCh))\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\tchangedEvent := <-changeCh\n\t\tchangedEvent2 := <-changeCh\n\t\tassert.NotEmpty(t, changedEvent, \"changed event should never be empty\")\n\t\tassert.NotEmpty(t, changedEvent2, \"changed event should never be empty\")\n\t}()\n\n\ttd.bypassRefreshRatelimiter()\n\ttd.hashRing.signalSelf()\n\n\t// wait until both subscribers will get notification\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration))\n\n\t// Test if internal members are updated\n\tassert.ElementsMatch(t, td.hashRing.Members(), hostsToReturn, \"members should contain just-added nodes\")\n}\n\nfunc TestSubscribersAreNotifiedPeriodically(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tvar hostsToReturn []HostInfo\n\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\t// we have to change members since subscribers are only notified on change\n\t\thostsToReturn = randomHostInfo(5)\n\t\treturn hostsToReturn, nil\n\t}).Times(2)\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\ttd.startHashRing()\n\n\tvar changeCh = make(chan *ChangedEvent, 1)\n\tassert.NoError(t, td.hashRing.Subscribe(\"subscriber1\", changeCh))\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\tevent := <-changeCh\n\t\tassert.NotEmpty(t, event, \"changed event should never be empty\")\n\t}()\n\n\ttd.mockTimeSource.BlockUntil(1)                   // we should wait until ticker(defaultRefreshInterval) is created\n\ttd.mockTimeSource.Advance(defaultRefreshInterval) // and only then to advance time\n\n\trequire.True(t, common.AwaitWaitGroup(&wg, maxTestDuration)) // wait until subscriber will get notification\n\n\t// Test if internal members are updated\n\tassert.ElementsMatch(t, td.hashRing.Members(), hostsToReturn, \"members should contain just-added nodes\")\n}\n\nfunc TestSubscribeIgnoresDuplicates(t *testing.T) {\n\tvar changeCh = make(chan *ChangedEvent)\n\ttd := newHashringTestData(t)\n\n\tassert.NoError(t, td.hashRing.Subscribe(\"test-service\", changeCh))\n\tassert.Error(t, td.hashRing.Subscribe(\"test-service\", changeCh))\n\tassert.Equal(t, 1, len(td.hashRing.subscribers.keys))\n}\n\nfunc TestUnsubcribeIgnoresDeletionOnEmpty(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tassert.Equal(t, 0, len(td.hashRing.subscribers.keys))\n\tassert.NoError(t, td.hashRing.Unsubscribe(\"test-service\"))\n\tassert.NoError(t, td.hashRing.Unsubscribe(\"test-service\"))\n\tassert.NoError(t, td.hashRing.Unsubscribe(\"test-service\"))\n}\n\nfunc TestUnsubcribeDeletes(t *testing.T) {\n\ttd := newHashringTestData(t)\n\tvar changeCh = make(chan *ChangedEvent)\n\n\tassert.Equal(t, 0, len(td.hashRing.subscribers.keys))\n\tassert.NoError(t, td.hashRing.Subscribe(\"testservice1\", changeCh))\n\tassert.Equal(t, 1, len(td.hashRing.subscribers.keys))\n\tassert.NoError(t, td.hashRing.Unsubscribe(\"test-service\"))\n\tassert.Equal(t, 1, len(td.hashRing.subscribers.keys))\n\tassert.NoError(t, td.hashRing.Unsubscribe(\"testservice1\"))\n\tassert.Equal(t, 0, len(td.hashRing.subscribers.keys))\n}\n\nfunc TestMemberCountReturnsNumber(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\tassert.Equal(t, 0, td.hashRing.MemberCount())\n\n\tring := emptyHashring()\n\tfor _, addr := range []string{\"127\", \"128\"} {\n\t\thost := NewHostInfo(addr)\n\t\tring.AddMembers(host)\n\t}\n\ttd.hashRing.value.Store(ring)\n\tassert.Equal(t, 2, td.hashRing.MemberCount())\n}\n\nfunc TestErrorIsPropagatedWhenProviderFails(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\ttd.mockPeerProvider.EXPECT().GetMembers(gomock.Any()).Return(nil, errors.New(\"provider failure\"))\n\n\tassert.ErrorContains(t, td.hashRing.Refresh(), \"provider failure\")\n}\n\nfunc TestStopWillStopProvider(t *testing.T) {\n\ttd := newHashringTestData(t)\n\n\ttd.mockPeerProvider.EXPECT().Stop().Times(1)\n\n\ttd.hashRing.status = common.DaemonStatusStarted\n\ttd.hashRing.Stop()\n}\n\nfunc TestLookupAndRefreshRaceCondition(t *testing.T) {\n\ttd := newHashringTestData(t)\n\tvar wg sync.WaitGroup\n\n\ttd.mockPeerProvider.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").AnyTimes().DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\treturn randomHostInfo(5), nil\n\t})\n\ttd.mockPeerProvider.EXPECT().WhoAmI().AnyTimes()\n\n\ttd.startHashRing()\n\twg.Add(2)\n\tgo func() {\n\t\tfor i := 0; i < 50; i++ {\n\t\t\t_, _ = td.hashRing.Lookup(\"a\")\n\t\t}\n\t\twg.Done()\n\t}()\n\tgo func() {\n\t\tfor i := 0; i < 50; i++ {\n\t\t\ttd.bypassRefreshRatelimiter()\n\t\t\tassert.NoError(t, td.hashRing.Refresh())\n\t\t}\n\t\twg.Done()\n\t}()\n\n\twg.Wait()\n}\n\nfunc TestEmitHashringView(t *testing.T) {\n\ttests := map[string]struct {\n\t\thosts          []HostInfo\n\t\tlookuperr      error\n\t\tselfInfo       HostInfo\n\t\tselfErr        error\n\t\texpectedResult float64\n\t}{\n\t\t\"example one - sorted set 1 - the output should be some random hashed value\": {\n\t\t\thosts: []HostInfo{\n\t\t\t\t{addr: \"10.0.0.1:1234\", ip: \"10.0.0.1\", identity: \"host1\", portMap: nil},\n\t\t\t\t{addr: \"10.0.0.2:1234\", ip: \"10.0.0.2\", identity: \"host2\", portMap: nil},\n\t\t\t\t{addr: \"10.0.0.3:1234\", ip: \"10.0.0.3\", identity: \"host3\", portMap: nil},\n\t\t\t},\n\t\t\tselfInfo:       HostInfo{identity: \"host123\"},\n\t\t\texpectedResult: 835.0, // the number here is meaningless\n\t\t},\n\t\t\"example one - unsorted set 1 - the order of the hosts should not matter\": {\n\t\t\thosts: []HostInfo{\n\t\t\t\t{addr: \"10.0.0.1:1234\", ip: \"10.0.0.1\", identity: \"host1\", portMap: nil},\n\t\t\t\t{addr: \"10.0.0.3:1234\", ip: \"10.0.0.3\", identity: \"host3\", portMap: nil},\n\t\t\t\t{addr: \"10.0.0.2:1234\", ip: \"10.0.0.2\", identity: \"host2\", portMap: nil},\n\t\t\t},\n\t\t\tselfInfo:       HostInfo{identity: \"host123\"},\n\t\t\texpectedResult: 835.0, // the test here is that it's the same as test 1\n\t\t},\n\t\t\"example 2 - empty set\": {\n\t\t\thosts:          []HostInfo{},\n\t\t\tselfInfo:       HostInfo{identity: \"host123\"},\n\t\t\texpectedResult: 242.0, // meaningless hash value\n\t\t},\n\t\t\"example 3 - nil set\": {\n\t\t\thosts:          nil,\n\t\t\tselfInfo:       HostInfo{identity: \"host123\"},\n\t\t\texpectedResult: 242.0, // meaningless hash value\n\t\t},\n\t}\n\n\tfor testName, testInput := range tests {\n\t\tt.Run(testName, func(t *testing.T) {\n\t\t\ttd := newHashringTestData(t)\n\n\t\t\ttd.mockPeerProvider.EXPECT().GetMembers(\"test-service\").DoAndReturn(func(service string) ([]HostInfo, error) {\n\t\t\t\treturn testInput.hosts, testInput.lookuperr\n\t\t\t})\n\t\t\ttd.mockPeerProvider.EXPECT().WhoAmI().DoAndReturn(func() (HostInfo, error) {\n\t\t\t\treturn testInput.selfInfo, testInput.selfErr\n\t\t\t}).AnyTimes()\n\n\t\t\trequire.NoError(t, td.hashRing.Refresh())\n\t\t\tassert.Equal(t, testInput.expectedResult, td.hashRing.emitHashIdentifier())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/membership/hostinfo.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\t\"strconv\"\n\t\"strings\"\n)\n\nconst (\n\tPortTchannel = \"tchannel\"\n\tPortGRPC     = \"grpc\"\n)\n\n// PortMap is a map of port names to port numbers.\ntype PortMap map[string]uint16\n\n// HostInfo is a type that contains the info about a cadence host\ntype HostInfo struct {\n\taddr     string // ip:port returned by peer provider\n\tip       string // @todo should we set this to net.IP ?\n\tidentity string\n\tportMap  PortMap // ports host is listening to\n}\n\n// NewHostInfo creates a new HostInfo instance\nfunc NewHostInfo(addr string) HostInfo {\n\tip, _, _ := net.SplitHostPort(addr)\n\treturn HostInfo{\n\t\taddr: addr,\n\t\tip:   ip,\n\t}\n}\n\n// String formats a PortMap into a string of name:port pairs\nfunc (m PortMap) String() string {\n\tres := make([]string, 0, len(m))\n\tfor name, port := range m {\n\t\tres = append(res, fmt.Sprintf(\"%s:%d\", name, port))\n\t}\n\treturn strings.Join(res, \", \")\n}\n\n// NewDetailedHostInfo creates a new HostInfo instance with identity and portmap information\nfunc NewDetailedHostInfo(addr string, identity string, portMap PortMap) HostInfo {\n\tip, _, _ := net.SplitHostPort(addr)\n\treturn HostInfo{\n\t\taddr:     addr,\n\t\tip:       ip,\n\t\tidentity: identity,\n\t\tportMap:  portMap,\n\t}\n}\n\n// GetAddress returns the ip:port address\nfunc (hi HostInfo) GetAddress() string {\n\treturn hi.addr\n}\n\n// GetNamedAddress returns the ip:port address\nfunc (hi HostInfo) GetNamedAddress(port string) (string, error) {\n\tif port, set := hi.portMap[port]; set {\n\t\treturn net.JoinHostPort(hi.ip, strconv.Itoa(int(port))), nil\n\t}\n\n\treturn \"\", fmt.Errorf(\"port %q is not set for %+v\", port, hi)\n}\n\n// Belongs tells if ip:port is assigned to this member\nfunc (hi HostInfo) Belongs(address string) (bool, error) {\n\n\tif hi.addr == address {\n\t\treturn true, nil\n\t}\n\n\tip, port, err := net.SplitHostPort(address)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tif ip != hi.ip {\n\t\treturn false, nil\n\t}\n\n\tfor _, number := range hi.portMap {\n\t\tif port == strconv.Itoa(int(number)) {\n\t\t\treturn true, nil\n\t\t}\n\t}\n\treturn false, nil\n}\n\n// Identity implements ringpop's Membership interface\nfunc (hi HostInfo) Identity() string {\n\t// if identity is not set, return address\n\tif hi.identity == \"\" {\n\t\treturn hi.addr\n\t}\n\n\treturn hi.identity\n}\n\n// Label is a noop function to conform to ringpop hashring member interface\nfunc (hi HostInfo) Label(key string) (value string, has bool) {\n\treturn \"\", false\n}\n\n// SetLabel is a noop function to conform to ringpop hashring member interface\nfunc (hi HostInfo) SetLabel(key string, value string) {\n}\n\n// String will return a human-readable host details\nfunc (hi HostInfo) String() string {\n\treturn fmt.Sprintf(\"addr: %s, identity: %s, portMap: %s\", hi.addr, hi.identity, hi.portMap)\n}\n"
  },
  {
    "path": "common/membership/hostinfo_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestBelongs(t *testing.T) {\n\n\thost := HostInfo{}\n\n\tbelongs, err := host.Belongs(\"127.1\")\n\tassert.False(t, belongs, \"invalid host info data should result to false\")\n\tassert.Error(t, err)\n\n\thost2 := NewDetailedHostInfo(\"127.0.0.1:123\", \"dummy\", PortMap{})\n\tbelongs, err = host2.Belongs(\"127.0.0.1:123\")\n\tassert.True(t, belongs, \"match on address, port map might be empty\")\n\tassert.NoError(t, err)\n\n\thost3 := NewDetailedHostInfo(\"127.0.0.1:1234\", \"dummy\", PortMap{PortGRPC: 3333})\n\tbelongs, err = host3.Belongs(\"127.0.0.1:3333\")\n\tassert.True(t, belongs, \"portmap should be checked\")\n\tassert.NoError(t, err)\n\n\thost4 := NewDetailedHostInfo(\"127.0.0.1:1234\", \"dummy\", PortMap{PortGRPC: 3333})\n\tbelongs, err = host4.Belongs(\"127.0.0.2:3333\")\n\tassert.False(t, belongs, \"different IP, will result in false\")\n\tassert.NoError(t, err)\n\n\thost5 := NewDetailedHostInfo(\"127.0.0.1:1234\", \"dummy\", PortMap{PortGRPC: 3333})\n\tbelongs, err = host5.Belongs(\"127.0.0.1:3334\")\n\tassert.False(t, belongs, \"portmap has no such port, should return empty without an error\")\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "common/membership/membershipfx/membershipfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membershipfx\n\nimport (\n\t\"fmt\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// Module provides membership components for fx app.\nvar Module = fx.Module(\"membership\", fx.Provide(buildMembership))\n\ntype buildMembershipParams struct {\n\tfx.In\n\n\tClock         clock.TimeSource\n\tRPCFactory    rpc.Factory\n\tPeerProvider  membership.PeerProvider\n\tLogger        log.Logger\n\tMetricsClient metrics.Client\n\tLifecycle     fx.Lifecycle\n}\n\ntype buildMembershipResult struct {\n\tfx.Out\n\n\tRings    map[string]membership.SingleProvider\n\tResolver membership.Resolver\n}\n\nfunc buildMembership(params buildMembershipParams) (buildMembershipResult, error) {\n\trings := make(map[string]membership.SingleProvider)\n\tfor _, s := range service.ListWithRing {\n\t\trings[s] = membership.NewHashring(s, params.PeerProvider, params.Clock, params.Logger, params.MetricsClient.Scope(metrics.HashringScope))\n\t}\n\n\tresolver, err := membership.NewResolver(\n\t\tparams.PeerProvider,\n\t\tparams.MetricsClient,\n\t\tparams.Logger,\n\t\trings,\n\t)\n\tif err != nil {\n\t\treturn buildMembershipResult{}, fmt.Errorf(\"create resolver: %w\", err)\n\t}\n\n\tparams.Lifecycle.Append(fx.StartStopHook(startResolver(resolver, params.RPCFactory), resolver.Stop))\n\tparams.Lifecycle.Append(fx.StopHook(params.RPCFactory.Stop))\n\n\treturn buildMembershipResult{\n\t\tRings:    rings,\n\t\tResolver: resolver,\n\t}, nil\n}\n\nfunc startResolver(resolver membership.Resolver, rpcFactory rpc.Factory) func() error {\n\treturn func() error {\n\t\terr := rpcFactory.Start(resolver)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"start rpc factory: %w\", err)\n\t\t}\n\t\tresolver.Start()\n\t\treturn nil\n\t}\n}\n"
  },
  {
    "path": "common/membership/membershipfx/memebershipfx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membershipfx\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestFxStartStop(t *testing.T) {\n\tapp := fxtest.New(t, fx.Provide(func() appParams {\n\t\tctrl := gomock.NewController(t)\n\t\tprovider := membership.NewMockPeerProvider(ctrl)\n\t\tprovider.EXPECT().Start()\n\t\tprovider.EXPECT().Stop()\n\t\tfor _, s := range service.ListWithRing {\n\t\t\tprovider.EXPECT().Subscribe(s, gomock.Any()).Return(nil)\n\t\t\tprovider.EXPECT().GetMembers(s).Return([]membership.HostInfo{}, nil)\n\t\t\t// this is also called by every ring, but could be called multiple times.\n\t\t\tprovider.EXPECT().Stop()\n\t\t}\n\t\tfactory := rpc.NewMockFactory(ctrl)\n\t\tfactory.EXPECT().Start(gomock.Any()).Return(nil)\n\t\tfactory.EXPECT().Stop().Return(nil)\n\t\treturn appParams{\n\t\t\tClock:         clock.NewMockedTimeSource(),\n\t\t\tPeerProvider:  provider,\n\t\t\tLogger:        testlogger.New(t),\n\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\tRPCFactory:    factory,\n\t\t}\n\t}), Module, fx.Invoke(func(resolver membership.Resolver) {}))\n\tapp.RequireStart().RequireStop()\n}\n\ntype appParams struct {\n\tfx.Out\n\n\tClock         clock.TimeSource\n\tPeerProvider  membership.PeerProvider\n\tLogger        log.Logger\n\tMetricsClient metrics.Client\n\tRPCFactory    rpc.Factory\n}\n"
  },
  {
    "path": "common/membership/peerprovider_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: hashring.go\n//\n// Generated by this command:\n//\n//\tmockgen -package membership -source hashring.go -destination peerprovider_mock.go github.com/uber/cadence/common/membership PeerProvider\n//\n\n// Package membership is a generated GoMock package.\npackage membership\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPeerProvider is a mock of PeerProvider interface.\ntype MockPeerProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerProviderMockRecorder is the mock recorder for MockPeerProvider.\ntype MockPeerProviderMockRecorder struct {\n\tmock *MockPeerProvider\n}\n\n// NewMockPeerProvider creates a new mock instance.\nfunc NewMockPeerProvider(ctrl *gomock.Controller) *MockPeerProvider {\n\tmock := &MockPeerProvider{ctrl: ctrl}\n\tmock.recorder = &MockPeerProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerProvider) EXPECT() *MockPeerProviderMockRecorder {\n\treturn m.recorder\n}\n\n// GetMembers mocks base method.\nfunc (m *MockPeerProvider) GetMembers(service string) ([]HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMembers\", service)\n\tret0, _ := ret[0].([]HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMembers indicates an expected call of GetMembers.\nfunc (mr *MockPeerProviderMockRecorder) GetMembers(service any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMembers\", reflect.TypeOf((*MockPeerProvider)(nil).GetMembers), service)\n}\n\n// SelfEvict mocks base method.\nfunc (m *MockPeerProvider) SelfEvict() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelfEvict\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SelfEvict indicates an expected call of SelfEvict.\nfunc (mr *MockPeerProviderMockRecorder) SelfEvict() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelfEvict\", reflect.TypeOf((*MockPeerProvider)(nil).SelfEvict))\n}\n\n// Start mocks base method.\nfunc (m *MockPeerProvider) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockPeerProviderMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockPeerProvider)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockPeerProvider) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockPeerProviderMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockPeerProvider)(nil).Stop))\n}\n\n// Subscribe mocks base method.\nfunc (m *MockPeerProvider) Subscribe(name string, handler func(ChangedEvent)) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Subscribe\", name, handler)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockPeerProviderMockRecorder) Subscribe(name, handler any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockPeerProvider)(nil).Subscribe), name, handler)\n}\n\n// WhoAmI mocks base method.\nfunc (m *MockPeerProvider) WhoAmI() (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WhoAmI\")\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WhoAmI indicates an expected call of WhoAmI.\nfunc (mr *MockPeerProviderMockRecorder) WhoAmI() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WhoAmI\", reflect.TypeOf((*MockPeerProvider)(nil).WhoAmI))\n}\n"
  },
  {
    "path": "common/membership/resolver.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination resolver_mock.go -self_package github.com/uber/cadence/common/membership\n\n// Package membership provides service discovery and membership information mechanism\npackage membership\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\n\t// ChangedEvent describes a change in a membership ring\n\tChangedEvent struct {\n\t\tHostsAdded   []string\n\t\tHostsUpdated []string\n\t\tHostsRemoved []string\n\t}\n\n\t// Resolver provides membership information for all cadence services.\n\tResolver interface {\n\t\tcommon.Daemon\n\n\t\t// WhoAmI returns self host details.\n\t\t// To be consistent with peer provider, it is advised to use peer provider\n\t\t// to return this information\n\t\tWhoAmI() (HostInfo, error)\n\n\t\t// EvictSelf evicts this member from the membership ring. After this method is\n\t\t// called, other members should discover that this node is no longer part of the\n\t\t// ring.\n\t\t// This primitive is useful to carry out graceful host shutdown during deployments.\n\t\tEvictSelf() error\n\n\t\t// Lookup will return host which is an owner for provided key.\n\t\tLookup(service, key string) (HostInfo, error)\n\n\t\t// Subscribe adds a subscriber which will get detailed change data on the given\n\t\t// channel, whenever membership changes.\n\t\tSubscribe(service, name string, notifyChannel chan<- *ChangedEvent) error\n\n\t\t// Unsubscribe removes a subscriber for this service.\n\t\tUnsubscribe(service, name string) error\n\n\t\t// MemberCount returns host count in a service specific hashring\n\t\tMemberCount(service string) (int, error)\n\n\t\t// Members returns all host addresses in a service specific hashring\n\t\tMembers(service string) ([]HostInfo, error)\n\n\t\t// LookupByAddress returns Host which owns IP:port tuple\n\t\tLookupByAddress(service, address string) (HostInfo, error)\n\t}\n)\n\n// MultiringResolver uses ring-per-service for membership information\ntype MultiringResolver struct {\n\tmetrics metrics.Client\n\tlogger  log.Logger\n\tstatus  int32\n\n\tprovider PeerProvider\n\tmu       sync.Mutex\n\trings    map[string]SingleProvider\n}\n\nvar _ Resolver = (*MultiringResolver)(nil)\n\n// NewResolver builds hashrings for all services\nfunc NewResolver(\n\tprovider PeerProvider,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\trings map[string]SingleProvider,\n) (Resolver, error) {\n\trpo := &MultiringResolver{\n\t\tstatus:   common.DaemonStatusInitialized,\n\t\tprovider: provider,\n\t\trings:    rings,\n\t\tmetrics:  metricsClient,\n\t\tlogger:   logger.WithTags(tag.ComponentMembershipResolver),\n\t\tmu:       sync.Mutex{},\n\t}\n\n\treturn rpo, nil\n}\n\n// Start starts provider and all rings\nfunc (rpo *MultiringResolver) Start() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&rpo.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\trpo.logger.Info(\"Starting membership resolver\", tag.ComponentMembershipResolver)\n\tdefer rpo.logger.Info(\"Started membership resolver\", tag.ComponentMembershipResolver)\n\n\trpo.provider.Start()\n\n\trpo.mu.Lock()\n\tdefer rpo.mu.Unlock()\n\tfor _, ring := range rpo.rings {\n\t\tring.Start()\n\t}\n}\n\n// Stop stops all rings and membership provider\nfunc (rpo *MultiringResolver) Stop() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&rpo.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\trpo.logger.Info(\"Stopping membership resolver\", tag.ComponentMembershipResolver)\n\tdefer rpo.logger.Info(\"Stopped membership resolver\", tag.ComponentMembershipResolver)\n\n\trpo.mu.Lock()\n\tdefer rpo.mu.Unlock()\n\tfor _, ring := range rpo.rings {\n\t\tring.Stop()\n\t}\n\n\trpo.provider.Stop()\n}\n\n// WhoAmI asks to provide current instance address\nfunc (rpo *MultiringResolver) WhoAmI() (HostInfo, error) {\n\treturn rpo.provider.WhoAmI()\n}\n\n// EvictSelf is used to remove this host from membership ring\nfunc (rpo *MultiringResolver) EvictSelf() error {\n\treturn rpo.provider.SelfEvict()\n}\n\nfunc (rpo *MultiringResolver) getRing(service string) (SingleProvider, error) {\n\trpo.mu.Lock()\n\tdefer rpo.mu.Unlock()\n\tring, found := rpo.rings[service]\n\tif !found {\n\t\treturn nil, fmt.Errorf(\"service %q is not tracked by Resolver\", service)\n\t}\n\treturn ring, nil\n}\n\nfunc (rpo *MultiringResolver) Lookup(service string, key string) (HostInfo, error) {\n\tring, err := rpo.getRing(service)\n\tif err != nil {\n\t\treturn HostInfo{}, err\n\t}\n\treturn ring.Lookup(key)\n}\n\nfunc (rpo *MultiringResolver) Subscribe(service string, name string, notifyChannel chan<- *ChangedEvent) error {\n\tring, err := rpo.getRing(service)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn ring.Subscribe(name, notifyChannel)\n}\n\nfunc (rpo *MultiringResolver) Unsubscribe(service string, name string) error {\n\tring, err := rpo.getRing(service)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn ring.Unsubscribe(name)\n}\n\nfunc (rpo *MultiringResolver) Members(service string) ([]HostInfo, error) {\n\tring, err := rpo.getRing(service)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn ring.Members(), nil\n}\n\nfunc (rpo *MultiringResolver) LookupByAddress(service, address string) (HostInfo, error) {\n\tmembers, err := rpo.Members(service)\n\tif err != nil {\n\t\treturn HostInfo{}, err\n\t}\n\tfor _, m := range members {\n\t\tif belongs, err := m.Belongs(address); err == nil && belongs {\n\t\t\treturn m, nil\n\t\t}\n\t}\n\trpo.metrics.Scope(metrics.ResolverHostNotFoundScope).IncCounter(metrics.RingResolverError)\n\treturn HostInfo{}, fmt.Errorf(\"host not found in service %s: %s\", service, address)\n}\n\nfunc (rpo *MultiringResolver) MemberCount(service string) (int, error) {\n\tring, err := rpo.getRing(service)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn ring.MemberCount(), nil\n}\n\nfunc (ce *ChangedEvent) Empty() bool {\n\treturn len(ce.HostsAdded) == 0 && len(ce.HostsUpdated) == 0 && len(ce.HostsRemoved) == 0\n}\n"
  },
  {
    "path": "common/membership/resolver_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: resolver.go\n//\n// Generated by this command:\n//\n//\tmockgen -package membership -source resolver.go -destination resolver_mock.go -self_package github.com/uber/cadence/common/membership\n//\n\n// Package membership is a generated GoMock package.\npackage membership\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockResolver is a mock of Resolver interface.\ntype MockResolver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockResolverMockRecorder\n\tisgomock struct{}\n}\n\n// MockResolverMockRecorder is the mock recorder for MockResolver.\ntype MockResolverMockRecorder struct {\n\tmock *MockResolver\n}\n\n// NewMockResolver creates a new mock instance.\nfunc NewMockResolver(ctrl *gomock.Controller) *MockResolver {\n\tmock := &MockResolver{ctrl: ctrl}\n\tmock.recorder = &MockResolverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockResolver) EXPECT() *MockResolverMockRecorder {\n\treturn m.recorder\n}\n\n// EvictSelf mocks base method.\nfunc (m *MockResolver) EvictSelf() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"EvictSelf\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// EvictSelf indicates an expected call of EvictSelf.\nfunc (mr *MockResolverMockRecorder) EvictSelf() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"EvictSelf\", reflect.TypeOf((*MockResolver)(nil).EvictSelf))\n}\n\n// Lookup mocks base method.\nfunc (m *MockResolver) Lookup(service, key string) (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Lookup\", service, key)\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Lookup indicates an expected call of Lookup.\nfunc (mr *MockResolverMockRecorder) Lookup(service, key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Lookup\", reflect.TypeOf((*MockResolver)(nil).Lookup), service, key)\n}\n\n// LookupByAddress mocks base method.\nfunc (m *MockResolver) LookupByAddress(service, address string) (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LookupByAddress\", service, address)\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LookupByAddress indicates an expected call of LookupByAddress.\nfunc (mr *MockResolverMockRecorder) LookupByAddress(service, address any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LookupByAddress\", reflect.TypeOf((*MockResolver)(nil).LookupByAddress), service, address)\n}\n\n// MemberCount mocks base method.\nfunc (m *MockResolver) MemberCount(service string) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberCount\", service)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberCount indicates an expected call of MemberCount.\nfunc (mr *MockResolverMockRecorder) MemberCount(service any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberCount\", reflect.TypeOf((*MockResolver)(nil).MemberCount), service)\n}\n\n// Members mocks base method.\nfunc (m *MockResolver) Members(service string) ([]HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Members\", service)\n\tret0, _ := ret[0].([]HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Members indicates an expected call of Members.\nfunc (mr *MockResolverMockRecorder) Members(service any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Members\", reflect.TypeOf((*MockResolver)(nil).Members), service)\n}\n\n// Start mocks base method.\nfunc (m *MockResolver) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockResolverMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockResolver)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockResolver) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockResolverMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockResolver)(nil).Stop))\n}\n\n// Subscribe mocks base method.\nfunc (m *MockResolver) Subscribe(service, name string, notifyChannel chan<- *ChangedEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Subscribe\", service, name, notifyChannel)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockResolverMockRecorder) Subscribe(service, name, notifyChannel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockResolver)(nil).Subscribe), service, name, notifyChannel)\n}\n\n// Unsubscribe mocks base method.\nfunc (m *MockResolver) Unsubscribe(service, name string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Unsubscribe\", service, name)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Unsubscribe indicates an expected call of Unsubscribe.\nfunc (mr *MockResolverMockRecorder) Unsubscribe(service, name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unsubscribe\", reflect.TypeOf((*MockResolver)(nil).Unsubscribe), service, name)\n}\n\n// WhoAmI mocks base method.\nfunc (m *MockResolver) WhoAmI() (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WhoAmI\")\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WhoAmI indicates an expected call of WhoAmI.\nfunc (mr *MockResolverMockRecorder) WhoAmI() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WhoAmI\", reflect.TypeOf((*MockResolver)(nil).WhoAmI))\n}\n"
  },
  {
    "path": "common/membership/resolver_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nvar testServices = []string{\"test-worker\", \"test-services\"}\n\nfunc TestSubscribeIsCalledOnPeerProvider(t *testing.T) {\n\tr, pp := newTestResolver(t)\n\t_, err := r.getRing(\"test-worker\")\n\tassert.NoError(t, err)\n\n\t// After membership is started, we expect start, subscribe and GetMembers on PeerProvider\n\tpp.EXPECT().Start().Times(1)\n\tpp.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(len(testServices))\n\tpp.EXPECT().GetMembers(gomock.Any()).Times(len(testServices))\n\n\tr.Start()\n}\n\nfunc TestNewCreatesAllRings(t *testing.T) {\n\ta, _ := newTestResolver(t)\n\tassert.Equal(t, len(testServices), len(a.rings))\n\n}\n\nfunc TestMethodsAreRoutedToARing(t *testing.T) {\n\tvar changeCh = make(chan *ChangedEvent)\n\ta, pp := newTestResolver(t)\n\n\t// add members to this ring\n\thosts := []HostInfo{}\n\tfor _, addr := range []string{\"127\", \"128\"} {\n\t\thosts = append(hosts, NewHostInfo(addr))\n\t}\n\n\tpp.EXPECT().GetMembers(\"test-worker\").Return(hosts, nil).Times(1)\n\tpp.EXPECT().WhoAmI().AnyTimes()\n\n\tr, err := a.getRing(\"test-worker\")\n\tr.Refresh()\n\n\tassert.NoError(t, err)\n\n\thi, err := r.Lookup(\"key\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"127\", hi.GetAddress())\n\n\t// the same ring will be picked here\n\t_, err = a.Lookup(\"WRONG-RING-NAME\", \"key\")\n\tassert.Error(t, err)\n\n\tmembers, err := a.Members(\"test-worker\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, 2, len(members))\n\n\tnomembers, err := a.Members(\"WRONG-RING-NAME\")\n\tassert.Error(t, err)\n\tassert.Equal(t, 0, len(nomembers))\n\n\tmemcount, err := a.MemberCount(\"test-worker\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, 2, memcount)\n\n\tnomemcount, err := a.MemberCount(\"WRONG-RING-NAME\")\n\tassert.Error(t, err)\n\tassert.Equal(t, 0, nomemcount)\n\n\tserr := a.Subscribe(\"test-worker\", \"sub1\", changeCh)\n\tassert.NoError(t, serr)\n\n\tserr = a.Subscribe(\"WRONG-RING-NAME\", \"sub1\", changeCh)\n\tassert.Error(t, serr)\n\n\tserr = a.Unsubscribe(\"test-worker\", \"sub1\")\n\tassert.NoError(t, serr)\n\n\tserr = a.Unsubscribe(\"WRONG-RING-NAME\", \"sub1\")\n\tassert.Error(t, serr)\n\n}\n\nfunc TestNonExistingRingReturnsError(t *testing.T) {\n\ta, _ := newTestResolver(t)\n\t_, err := a.getRing(\"non-existing\")\n\tassert.Error(t, err)\n}\n\nfunc TestCallsAreForwardedToProvider(t *testing.T) {\n\ta, mockedPeer := newTestResolver(t)\n\n\tmockedPeer.EXPECT().WhoAmI().Times(1)\n\tmockedPeer.EXPECT().SelfEvict().Times(1)\n\tmockedPeer.EXPECT().Stop().Times(1)\n\n\ta.status = common.DaemonStatusStarted\n\ta.WhoAmI()\n\ta.EvictSelf()\n\ta.Stop()\n\n}\n\nfunc newTestResolver(t *testing.T) (*MultiringResolver, *MockPeerProvider) {\n\n\tctrl := gomock.NewController(t)\n\tpp := NewMockPeerProvider(ctrl)\n\n\trings := make(map[string]SingleProvider, len(testServices))\n\tfor _, testService := range testServices {\n\t\tring := NewHashring(testService, pp, clock.NewMockedTimeSource(), log.NewNoop(), metrics.NewNoopMetricsClient().Scope(metrics.HashringScope))\n\t\trings[testService] = ring\n\t}\n\n\tresolver, err := NewResolver(\n\t\tpp,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t\trings,\n\t)\n\trequire.NoError(t, err)\n\n\treturn resolver.(*MultiringResolver), pp\n}\n"
  },
  {
    "path": "common/membership/sharddistributorresolver.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n)\n\ntype ModeKey string\n\nconst (\n\t// ModeKeyHashRing represents the hash ring shard distribution mode\n\tModeKeyHashRing ModeKey = \"hash_ring\"\n\t// ModeKeyShardDistributor represents the shard distributor mode\n\tModeKeyShardDistributor ModeKey = \"shard_distributor\"\n\t// ModeKeyHashRingShadowShardDistributor represents hash ring mode with shard distributor shadow\n\tModeKeyHashRingShadowShardDistributor ModeKey = \"hash_ring-shadow-shard_distributor\"\n\t// ModeKeyShardDistributorShadowHashRing represents shard distributor mode with hash ring shadow\n\tModeKeyShardDistributorShadowHashRing ModeKey = \"shard_distributor-shadow-hash_ring\"\n)\n\ntype shardDistributorResolver struct {\n\tshardDistributionMode      dynamicproperties.StringPropertyFn\n\texcludeShortLivedTaskLists dynamicproperties.BoolPropertyFn\n\tpercentageOnboarded        dynamicproperties.IntPropertyFn\n\tspectator                  spectatorclient.Spectator\n\tring                       SingleProvider\n\tlogger                     log.Logger\n}\n\nfunc (s shardDistributorResolver) AddressToHost(owner string) (HostInfo, error) {\n\treturn s.ring.AddressToHost(owner)\n}\n\nfunc NewShardDistributorResolver(\n\tspectator spectatorclient.Spectator,\n\tshardDistributionMode dynamicproperties.StringPropertyFn,\n\texcludeShortLivedTaskLists dynamicproperties.BoolPropertyFn,\n\tpercentageOnboarded dynamicproperties.IntPropertyFn,\n\tring SingleProvider,\n\tlogger log.Logger,\n) SingleProvider {\n\treturn &shardDistributorResolver{\n\t\tspectator:                  spectator,\n\t\tshardDistributionMode:      shardDistributionMode,\n\t\texcludeShortLivedTaskLists: excludeShortLivedTaskLists,\n\t\tpercentageOnboarded:        percentageOnboarded,\n\t\tring:                       ring,\n\t\tlogger:                     logger,\n\t}\n}\n\nfunc (s shardDistributorResolver) Start() {\n\t// We do not need to start anything in the shard distributor, so just start the ring\n\ts.ring.Start()\n}\n\nfunc (s shardDistributorResolver) Stop() {\n\t// We do not need to stop anything in the shard distributor, so just stop the ring\n\ts.ring.Stop()\n}\n\nfunc (s shardDistributorResolver) Lookup(key string) (HostInfo, error) {\n\tif s.shardDistributionMode() != \"hash_ring\" && s.spectator == nil {\n\t\t// This will avoid panics when the shard distributor is not configured\n\t\ts.logger.Warn(\"No shard distributor client, defaulting to hash ring\", tag.Value(s.shardDistributionMode()))\n\n\t\treturn s.ring.Lookup(key)\n\t}\n\n\texcludeTaskList := TaskListExcludedFromShardDistributor(key, uint64(s.percentageOnboarded()), s.excludeShortLivedTaskLists())\n\tif excludeTaskList {\n\t\treturn s.ring.Lookup(key)\n\t}\n\n\tswitch ModeKey(s.shardDistributionMode()) {\n\tcase ModeKeyHashRing:\n\t\treturn s.ring.Lookup(key)\n\tcase ModeKeyShardDistributor:\n\t\treturn s.lookUpInShardDistributor(key)\n\tcase ModeKeyHashRingShadowShardDistributor:\n\t\thashRingResult, err := s.ring.Lookup(key)\n\t\tif err != nil {\n\t\t\treturn HostInfo{}, err\n\t\t}\n\t\t// Asynchronously lookup in shard distributor to avoid blocking the main thread\n\t\tgo func() {\n\t\t\tshardDistributorResult, err := s.lookUpInShardDistributor(key)\n\t\t\tif err != nil {\n\t\t\t\ts.logger.Warn(\"Failed to lookup in shard distributor shadow\", tag.Error(err))\n\t\t\t} else {\n\t\t\t\tlogDifferencesInHostInfo(s.logger, hashRingResult, shardDistributorResult)\n\t\t\t}\n\t\t}()\n\n\t\treturn hashRingResult, nil\n\tcase ModeKeyShardDistributorShadowHashRing:\n\t\tshardDistributorResult, err := s.lookUpInShardDistributor(key)\n\t\tif err != nil {\n\t\t\treturn HostInfo{}, err\n\t\t}\n\t\t// Asynchronously lookup in hash ring to avoid blocking the main thread\n\t\tgo func() {\n\t\t\thashRingResult, err := s.ring.Lookup(key)\n\t\t\tif err != nil {\n\t\t\t\ts.logger.Warn(\"Failed to lookup in hash ring shadow\", tag.Error(err))\n\t\t\t} else {\n\t\t\t\tlogDifferencesInHostInfo(s.logger, hashRingResult, shardDistributorResult)\n\t\t\t}\n\t\t}()\n\n\t\treturn shardDistributorResult, nil\n\t}\n\n\t// Default to hash ring\n\ts.logger.Warn(\"Unknown shard distribution mode, defaulting to hash ring\", tag.Value(s.shardDistributionMode()))\n\n\treturn s.ring.Lookup(key)\n}\n\nfunc (s shardDistributorResolver) Subscribe(name string, channel chan<- *ChangedEvent) error {\n\t// Shard distributor does not support subscription yet, so use the ring\n\treturn s.ring.Subscribe(name, channel)\n}\n\nfunc (s shardDistributorResolver) Unsubscribe(name string) error {\n\t// Shard distributor does not support subscription yet, so use the ring\n\treturn s.ring.Unsubscribe(name)\n}\n\nfunc (s shardDistributorResolver) Members() []HostInfo {\n\t// Shard distributor does not member tracking yet, so use the ring\n\treturn s.ring.Members()\n}\n\nfunc (s shardDistributorResolver) MemberCount() int {\n\t// Shard distributor does not member tracking yet, so use the ring\n\treturn s.ring.MemberCount()\n}\n\nfunc (s shardDistributorResolver) Refresh() error {\n\t// Shard distributor does not need refresh, so propagate to the ring\n\treturn s.ring.Refresh()\n}\n\n// TODO: cache the hostinfos, creating them on every request is relatively expensive (we need to do string parsing etc.)\nfunc (s shardDistributorResolver) lookUpInShardDistributor(key string) (HostInfo, error) {\n\tctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)\n\tdefer cancel()\n\towner, err := s.spectator.GetShardOwner(ctx, key)\n\tif err != nil {\n\t\treturn HostInfo{}, err\n\t}\n\n\tportMap := make(PortMap)\n\tfor _, portType := range []string{PortTchannel, PortGRPC} {\n\t\tportString, ok := owner.Metadata[portType]\n\t\tif !ok {\n\t\t\ts.logger.Warn(fmt.Sprintf(\"Port %s not found in metadata\", portType), tag.Value(owner))\n\t\t\tcontinue\n\t\t}\n\n\t\tport, err := strconv.ParseUint(portString, 10, 16)\n\t\tif err != nil {\n\t\t\ts.logger.Warn(fmt.Sprintf(\"Failed to convert %s port to int\", portType), tag.Error(err), tag.Value(owner))\n\t\t\tcontinue\n\t\t}\n\t\t// This is safe because the conversion above ensures the port is within the uint16 range\n\t\tportMap[portType] = uint16(port)\n\t}\n\n\thostaddress := owner.Metadata[\"hostIP\"]\n\tif hostaddress == \"\" {\n\t\treturn HostInfo{}, fmt.Errorf(\"hostIP not found in metadata\")\n\t}\n\n\taddress := net.JoinHostPort(hostaddress, strconv.Itoa(int(portMap[PortTchannel])))\n\n\thostInfo := NewDetailedHostInfo(address, owner.ExecutorID, portMap)\n\treturn hostInfo, nil\n}\n\nfunc logDifferencesInHostInfo(logger log.Logger, hashRingResult, shardDistributorResult HostInfo) {\n\tfor _, portName := range []string{PortTchannel, PortGRPC} {\n\t\thashRingAddr, err := hashRingResult.GetNamedAddress(portName)\n\t\tif err != nil {\n\t\t\tlogger.Warn(fmt.Sprintf(\"Failed to get hash ring %s address\", portName), tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\n\t\tshardDistributorAddr, err := shardDistributorResult.GetNamedAddress(portName)\n\t\tif err != nil {\n\t\t\tlogger.Warn(fmt.Sprintf(\"Failed to get shard distributor %s address\", portName), tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\n\t\tif hashRingAddr != shardDistributorAddr {\n\t\t\tlogger.Warn(fmt.Sprintf(\"Shadow lookup mismatch %s addresses\", portName),\n\t\t\t\ttag.HashRingResult(hashRingAddr),\n\t\t\t\ttag.ShardDistributorResult(shardDistributorAddr))\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/membership/sharddistributorresolver_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n)\n\nfunc TestShardDistributorResolver_Lookup_modeHashRing(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tresolver.shardDistributionMode = func(...dynamicproperties.FilterOption) string {\n\t\treturn string(ModeKeyHashRing)\n\t}\n\n\tring.EXPECT().Lookup(\"test-key\").Return(HostInfo{addr: \"test-addr\"}, nil)\n\n\thost, err := resolver.Lookup(\"test-key\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"test-addr\", host.addr)\n}\n\nfunc TestShardDistributorResolver_Lookup_modeShardDistributor(t *testing.T) {\n\tresolver, _, shardDistributorMock := newShardDistributorResolver(t)\n\tresolver.shardDistributionMode = func(...dynamicproperties.FilterOption) string {\n\t\treturn string(ModeKeyShardDistributor)\n\t}\n\n\tshardDistributorMock.EXPECT().GetShardOwner(gomock.Any(), \"test-key\").\n\t\tReturn(&spectatorclient.ShardOwner{\n\t\t\tExecutorID: \"test-owner\",\n\t\t\tMetadata: map[string]string{\n\t\t\t\t\"hostIP\":   \"127.0.0.1\",\n\t\t\t\t\"tchannel\": \"7933\",\n\t\t\t\t\"grpc\":     \"7833\",\n\t\t\t},\n\t\t}, nil)\n\n\thost, err := resolver.Lookup(\"test-key\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"127.0.0.1:7933\", host.addr)\n}\n\nfunc TestShardDistributorResolver_Lookup_modeHashRingShadowShardDistributor(t *testing.T) {\n\tresolver, ring, shardDistributorMock := newShardDistributorResolver(t)\n\tresolver.shardDistributionMode = func(...dynamicproperties.FilterOption) string {\n\t\treturn string(ModeKeyHashRingShadowShardDistributor)\n\t}\n\n\tcases := []struct {\n\t\tname                   string\n\t\thashRingAddr           string\n\t\thashRingError          error\n\t\tshardDistributorHostIP string\n\t\tshardDistributorError  error\n\t\texpectedLog            string\n\t}{\n\t\t{\n\t\t\tname:                   \"hash ring and shard distributor agree\",\n\t\t\thashRingAddr:           \"127.0.0.1:7933\",\n\t\t\tshardDistributorHostIP: \"127.0.0.1\",\n\t\t},\n\t\t{\n\t\t\tname:                   \"hash ring and shard distributor disagree\",\n\t\t\thashRingAddr:           \"127.0.0.1:7933\",\n\t\t\tshardDistributorHostIP: \"127.0.0.2\",\n\t\t\texpectedLog:            \"Shadow lookup mismatch\",\n\t\t},\n\t\t{\n\t\t\tname:                  \"shard distributor error\",\n\t\t\thashRingAddr:          \"127.0.0.1:7933\",\n\t\t\tshardDistributorError: assert.AnError,\n\t\t\texpectedLog:           \"Failed to lookup in shard distributor shadow\",\n\t\t},\n\t\t{\n\t\t\tname:          \"hash ring error\",\n\t\t\thashRingError: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlogger, logs := testlogger.NewObserved(t)\n\t\t\tresolver.logger = logger\n\n\t\t\tring.EXPECT().Lookup(\"test-key\").Return(NewDetailedHostInfo(\n\t\t\t\ttc.hashRingAddr,\n\t\t\t\t\"test-owner\",\n\t\t\t\tPortMap{PortTchannel: 7933, PortGRPC: 7833},\n\t\t\t), tc.hashRingError)\n\t\t\t// If the hash ring lookup fails, we should just bail out and not call the shard distributor\n\t\t\tif tc.hashRingError == nil {\n\t\t\t\tshardDistributorMock.EXPECT().GetShardOwner(gomock.Any(), \"test-key\").\n\t\t\t\t\tReturn(&spectatorclient.ShardOwner{\n\t\t\t\t\t\tExecutorID: \"test-owner\",\n\t\t\t\t\t\tMetadata: map[string]string{\n\t\t\t\t\t\t\t\"hostIP\":   tc.shardDistributorHostIP,\n\t\t\t\t\t\t\t\"tchannel\": \"7933\",\n\t\t\t\t\t\t\t\"grpc\":     \"7833\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, tc.shardDistributorError)\n\t\t\t}\n\n\t\t\thost, err := resolver.Lookup(\"test-key\")\n\t\t\tassert.Equal(t, err, tc.hashRingError)\n\n\t\t\tif tc.hashRingError == nil {\n\t\t\t\tassert.Equal(t, \"127.0.0.1:7933\", host.addr)\n\t\t\t}\n\n\t\t\t// Wait a bit for async shadow lookup to complete\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\tif tc.expectedLog != \"\" {\n\t\t\t\tif tc.expectedLog == \"Shadow lookup mismatch\" {\n\t\t\t\t\t// logDifferencesInHostInfo logs separately for tchannel and grpc ports\n\t\t\t\t\tassert.Equal(t, 2, logs.Len())\n\t\t\t\t} else {\n\t\t\t\t\t// Error cases only log once\n\t\t\t\t\tassert.Equal(t, 1, logs.Len())\n\t\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(tc.expectedLog).Len())\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, 0, logs.Len())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardDistributorResolver_Lookup_modeShardDistributorShadowHashRing(t *testing.T) {\n\tresolver, ring, shardDistributorMock := newShardDistributorResolver(t)\n\tresolver.shardDistributionMode = func(...dynamicproperties.FilterOption) string {\n\t\treturn string(ModeKeyShardDistributorShadowHashRing)\n\t}\n\n\tcases := []struct {\n\t\tname                   string\n\t\tshardDistributorHostIP string\n\t\tshardDistributorError  error\n\t\thashRingAddr           string\n\t\thashRingError          error\n\t\texpectedLog            string\n\t}{\n\t\t{\n\t\t\tname:                   \"shard distributor and hash ring agree\",\n\t\t\tshardDistributorHostIP: \"127.0.0.1\",\n\t\t\thashRingAddr:           \"127.0.0.1:7933\",\n\t\t},\n\t\t{\n\t\t\tname:                   \"shard distributor and hash ring disagree\",\n\t\t\tshardDistributorHostIP: \"127.0.0.1\",\n\t\t\thashRingAddr:           \"127.0.0.2:7933\",\n\t\t\texpectedLog:            \"Shadow lookup mismatch\",\n\t\t},\n\t\t{\n\t\t\tname:                   \"hash ring error\",\n\t\t\tshardDistributorHostIP: \"127.0.0.1\",\n\t\t\thashRingError:          assert.AnError,\n\t\t\texpectedLog:            \"Failed to lookup in hash ring shadow\",\n\t\t},\n\t\t{\n\t\t\tname:                  \"shard distributor error\",\n\t\t\tshardDistributorError: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlogger, logs := testlogger.NewObserved(t)\n\t\t\tresolver.logger = logger\n\n\t\t\tshardDistributorMock.EXPECT().GetShardOwner(gomock.Any(), \"test-key\").\n\t\t\t\tReturn(&spectatorclient.ShardOwner{\n\t\t\t\t\tExecutorID: \"test-owner\",\n\t\t\t\t\tMetadata: map[string]string{\n\t\t\t\t\t\t\"hostIP\":   tc.shardDistributorHostIP,\n\t\t\t\t\t\t\"tchannel\": \"7933\",\n\t\t\t\t\t\t\"grpc\":     \"7833\",\n\t\t\t\t\t},\n\t\t\t\t}, tc.shardDistributorError)\n\n\t\t\t// If the hash ring lookup fails, we should just bail out and not call the shard distributor\n\t\t\tif tc.shardDistributorError == nil {\n\t\t\t\tring.EXPECT().Lookup(\"test-key\").Return(NewDetailedHostInfo(\n\t\t\t\t\ttc.hashRingAddr,\n\t\t\t\t\t\"test-owner\",\n\t\t\t\t\tPortMap{PortTchannel: 7933, PortGRPC: 7833},\n\t\t\t\t), tc.hashRingError)\n\t\t\t}\n\n\t\t\thost, err := resolver.Lookup(\"test-key\")\n\t\t\tassert.Equal(t, err, tc.shardDistributorError)\n\n\t\t\tif tc.shardDistributorError == nil {\n\t\t\t\tassert.Equal(t, \"127.0.0.1:7933\", host.addr)\n\t\t\t}\n\n\t\t\t// Wait a bit for async shadow lookup to complete\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\tif tc.expectedLog != \"\" {\n\t\t\t\tif tc.expectedLog == \"Shadow lookup mismatch\" {\n\t\t\t\t\t// logDifferencesInHostInfo logs separately for tchannel and grpc ports\n\t\t\t\t\tassert.Equal(t, 2, logs.Len())\n\t\t\t\t} else {\n\t\t\t\t\t// Error cases only log once\n\t\t\t\t\tassert.Equal(t, 1, logs.Len())\n\t\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(tc.expectedLog).Len())\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, 0, logs.Len())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardDistributorResolver_Lookup_UnknownMode(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tresolver.shardDistributionMode = func(...dynamicproperties.FilterOption) string {\n\t\treturn \"unknown\"\n\t}\n\n\tring.EXPECT().Lookup(\"test-key\").Return(HostInfo{addr: \"test-addr\"}, nil)\n\n\thost, err := resolver.Lookup(\"test-key\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"test-addr\", host.addr)\n}\n\n/* Test all the simple proxies\n */\nfunc TestShardDistributorResolver_Start(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Start().Times(1)\n\tresolver.Start()\n}\n\nfunc TestShardDistributorResolver_Stop(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Stop().Times(1)\n\tresolver.Stop()\n}\n\nfunc TestShardDistributorResolver_Subscribe(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Subscribe(\"test-name\", gomock.Any()).Times(1)\n\tresolver.Subscribe(\"test-name\", nil)\n}\n\nfunc TestShardDistributorResolver_UnSubscribe(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Unsubscribe(\"test-name\").Times(1)\n\tresolver.Unsubscribe(\"test-name\")\n}\n\nfunc TestShardDistributorResolver_Members(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Members().Return(nil)\n\tresolver.Members()\n}\n\nfunc TestShardDistributorResolver_MemberCount(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().MemberCount().Return(0)\n\tresolver.MemberCount()\n}\n\nfunc TestShardDistributorResolver_Refresh(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().Refresh().Times(1)\n\tresolver.Refresh()\n}\n\nfunc TestShardDistributorResolver_AddressToHost(t *testing.T) {\n\tresolver, ring, _ := newShardDistributorResolver(t)\n\tring.EXPECT().AddressToHost(\"test\").Return(HostInfo{}, nil)\n\tresolver.AddressToHost(\"test\")\n}\n\nfunc TestShardDistributorResolver_Lookup_ExcludeShortLivedTaskLists(t *testing.T) {\n\tcases := []struct {\n\t\tname                       string\n\t\texcludeShortLivedTaskLists bool\n\t\ttaskListName               string\n\t\tmode                       ModeKey\n\t\texpectHashRing             bool\n\t}{\n\t\t{\n\t\t\tname:                       \"exclude enabled with UUID tasklist uses hash ring\",\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tmode:                       ModeKeyShardDistributor,\n\t\t\texpectHashRing:             true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"exclude enabled without UUID tasklist uses shard distributor\",\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t\ttaskListName:               \"my-regular-tasklist\",\n\t\t\tmode:                       ModeKeyShardDistributor,\n\t\t\texpectHashRing:             false,\n\t\t},\n\t\t{\n\t\t\tname:                       \"exclude disabled with UUID tasklist uses shard distributor\",\n\t\t\texcludeShortLivedTaskLists: false,\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tmode:                       ModeKeyShardDistributor,\n\t\t\texpectHashRing:             false,\n\t\t},\n\t\t{\n\t\t\tname:                       \"exclude enabled with UUID tasklist in hash ring mode still uses hash ring\",\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tmode:                       ModeKeyHashRing,\n\t\t\texpectHashRing:             true,\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tspectator := spectatorclient.NewMockSpectator(ctrl)\n\t\t\tring := NewMockSingleProvider(ctrl)\n\t\t\tlogger := log.NewNoop()\n\n\t\t\tresolver := NewShardDistributorResolver(\n\t\t\t\tspectator,\n\t\t\t\tdynamicproperties.GetStringPropertyFn(string(tc.mode)),\n\t\t\t\tdynamicproperties.GetBoolPropertyFn(tc.excludeShortLivedTaskLists),\n\t\t\t\tdynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tring,\n\t\t\t\tlogger,\n\t\t\t).(*shardDistributorResolver)\n\n\t\t\tif tc.expectHashRing {\n\t\t\t\tring.EXPECT().Lookup(tc.taskListName).Return(HostInfo{addr: \"hash-ring-addr\"}, nil)\n\t\t\t} else {\n\t\t\t\tspectator.EXPECT().GetShardOwner(gomock.Any(), tc.taskListName).\n\t\t\t\t\tReturn(&spectatorclient.ShardOwner{\n\t\t\t\t\t\tExecutorID: \"test-owner\",\n\t\t\t\t\t\tMetadata: map[string]string{\n\t\t\t\t\t\t\t\"hostIP\":   \"127.0.0.1\",\n\t\t\t\t\t\t\t\"tchannel\": \"7933\",\n\t\t\t\t\t\t\t\"grpc\":     \"7833\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t}\n\n\t\t\thost, err := resolver.Lookup(tc.taskListName)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif tc.expectHashRing {\n\t\t\t\tassert.Equal(t, \"hash-ring-addr\", host.addr)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, \"127.0.0.1:7933\", host.addr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc newShardDistributorResolver(t *testing.T) (*shardDistributorResolver, *MockSingleProvider, *spectatorclient.MockSpectator) {\n\tctrl := gomock.NewController(t)\n\tspectator := spectatorclient.NewMockSpectator(ctrl)\n\tshardDistributionMode := dynamicproperties.GetStringPropertyFn(\"\")\n\texcludeShortLivedTaskLists := dynamicproperties.GetBoolPropertyFn(false)\n\tpercentageOnboarded := dynamicproperties.GetIntPropertyFn(100)\n\tring := NewMockSingleProvider(ctrl)\n\tlogger := log.NewNoop()\n\n\tresolver := NewShardDistributorResolver(spectator, shardDistributionMode, excludeShortLivedTaskLists, percentageOnboarded, ring, logger).(*shardDistributorResolver)\n\n\treturn resolver, ring, spectator\n}\n"
  },
  {
    "path": "common/membership/singleprovider.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage membership\n\nimport \"github.com/uber/cadence/common\"\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=singleprovider_mock.go SingleProvider\n\ntype SingleProvider interface {\n\tcommon.Daemon\n\tLookup(key string) (HostInfo, error)\n\tSubscribe(name string, channel chan<- *ChangedEvent) error\n\tAddressToHost(owner string) (HostInfo, error)\n\tUnsubscribe(name string) error\n\tMembers() []HostInfo\n\tMemberCount() int\n\tRefresh() error\n}\n"
  },
  {
    "path": "common/membership/singleprovider_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: singleprovider.go\n//\n// Generated by this command:\n//\n//\tmockgen -package membership -source singleprovider.go -destination=singleprovider_mock.go SingleProvider\n//\n\n// Package membership is a generated GoMock package.\npackage membership\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockSingleProvider is a mock of SingleProvider interface.\ntype MockSingleProvider struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSingleProviderMockRecorder\n\tisgomock struct{}\n}\n\n// MockSingleProviderMockRecorder is the mock recorder for MockSingleProvider.\ntype MockSingleProviderMockRecorder struct {\n\tmock *MockSingleProvider\n}\n\n// NewMockSingleProvider creates a new mock instance.\nfunc NewMockSingleProvider(ctrl *gomock.Controller) *MockSingleProvider {\n\tmock := &MockSingleProvider{ctrl: ctrl}\n\tmock.recorder = &MockSingleProviderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSingleProvider) EXPECT() *MockSingleProviderMockRecorder {\n\treturn m.recorder\n}\n\n// AddressToHost mocks base method.\nfunc (m *MockSingleProvider) AddressToHost(owner string) (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddressToHost\", owner)\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddressToHost indicates an expected call of AddressToHost.\nfunc (mr *MockSingleProviderMockRecorder) AddressToHost(owner any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddressToHost\", reflect.TypeOf((*MockSingleProvider)(nil).AddressToHost), owner)\n}\n\n// Lookup mocks base method.\nfunc (m *MockSingleProvider) Lookup(key string) (HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Lookup\", key)\n\tret0, _ := ret[0].(HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Lookup indicates an expected call of Lookup.\nfunc (mr *MockSingleProviderMockRecorder) Lookup(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Lookup\", reflect.TypeOf((*MockSingleProvider)(nil).Lookup), key)\n}\n\n// MemberCount mocks base method.\nfunc (m *MockSingleProvider) MemberCount() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberCount\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// MemberCount indicates an expected call of MemberCount.\nfunc (mr *MockSingleProviderMockRecorder) MemberCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberCount\", reflect.TypeOf((*MockSingleProvider)(nil).MemberCount))\n}\n\n// Members mocks base method.\nfunc (m *MockSingleProvider) Members() []HostInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Members\")\n\tret0, _ := ret[0].([]HostInfo)\n\treturn ret0\n}\n\n// Members indicates an expected call of Members.\nfunc (mr *MockSingleProviderMockRecorder) Members() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Members\", reflect.TypeOf((*MockSingleProvider)(nil).Members))\n}\n\n// Refresh mocks base method.\nfunc (m *MockSingleProvider) Refresh() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Refresh\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Refresh indicates an expected call of Refresh.\nfunc (mr *MockSingleProviderMockRecorder) Refresh() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Refresh\", reflect.TypeOf((*MockSingleProvider)(nil).Refresh))\n}\n\n// Start mocks base method.\nfunc (m *MockSingleProvider) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockSingleProviderMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockSingleProvider)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockSingleProvider) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockSingleProviderMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockSingleProvider)(nil).Stop))\n}\n\n// Subscribe mocks base method.\nfunc (m *MockSingleProvider) Subscribe(name string, channel chan<- *ChangedEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Subscribe\", name, channel)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockSingleProviderMockRecorder) Subscribe(name, channel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockSingleProvider)(nil).Subscribe), name, channel)\n}\n\n// Unsubscribe mocks base method.\nfunc (m *MockSingleProvider) Unsubscribe(name string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Unsubscribe\", name)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Unsubscribe indicates an expected call of Unsubscribe.\nfunc (mr *MockSingleProviderMockRecorder) Unsubscribe(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unsubscribe\", reflect.TypeOf((*MockSingleProvider)(nil).Unsubscribe), name)\n}\n"
  },
  {
    "path": "common/membership/tasklist_differentiator.go",
    "content": "package membership\n\nimport (\n\t\"regexp\"\n\n\t\"github.com/dgryski/go-farm\"\n)\n\nconst uuidRegex = `[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}`\n\nvar uuidRegexp = regexp.MustCompile(uuidRegex)\n\nfunc TaskListExcludedFromShardDistributor(taskListName string, percentageOnboarded uint64, excludeShortLivedTaskLists bool) bool {\n\t// This regex checks if the task list name has a UUID, if it does we\n\t// consider it a short lived tasklist that will not be managed by the shard distributor.\n\texcludeShortLivedTaskLists = excludeShortLivedTaskLists && uuidRegexp.MatchString(taskListName)\n\treturn excludeShortLivedTaskLists || !belowPercentage(taskListName, percentageOnboarded)\n}\n\nfunc belowPercentage(taskListName string, percentageOnboarded uint64) bool {\n\thash := farm.Fingerprint64([]byte(taskListName))\n\treturn isbelowPercentage(hash, percentageOnboarded)\n}\n\nfunc isbelowPercentage(hash uint64, percentageOnboarded uint64) bool {\n\treturn hash%100 < percentageOnboarded\n}\n"
  },
  {
    "path": "common/membership/tasklist_differentiator_test.go",
    "content": "package membership\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestTaskListExcludedFromShardDistributor(t *testing.T) {\n\ttests := []struct {\n\t\tname                       string\n\t\ttaskListName               string\n\t\twant                       bool\n\t\tpercentageOnboarded        uint64\n\t\texcludeShortLivedTaskLists bool\n\t}{\n\t\t{\n\t\t\tname:                       \"task list with UUID - exclude enabled\",\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list with UUID - exclude disabled\",\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: false,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list with uppercase UUID - exclude enabled\",\n\t\t\ttaskListName:               \"tasklist-550E8400-E29B-41D4-A716-446655440000\",\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list name is UUID only - exclude enabled\",\n\t\t\ttaskListName:               \"550e8400-e29b-41d4-a716-446655440000\",\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list without UUID\",\n\t\t\ttaskListName:               \"my-task-list\",\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"empty task list name\",\n\t\t\ttaskListName:               \"\",\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list with partial UUID-like string\",\n\t\t\ttaskListName:               \"tasklist-550e8400-e29b\",\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list with UUID prefix\",\n\t\t\ttaskListName:               \"550e8400-e29b-41d4-a716-446655440000-suffix\",\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        100,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list below percentage\",\n\t\t\ttaskListName:               \"my-task-list\", // farm.Fingerprint64([]byte(\"my-task-list\"))%100 = 58\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        60,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list above percentage\",\n\t\t\ttaskListName:               \"my-task-list\", // farm.Fingerprint64([]byte(\"my-task-list\"))%100 = 58\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        50,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list below percentage - but UUID - exclude enabled\",\n\t\t\ttaskListName:               \"my-task-list-550e8400-e29b-41d4-a716-446655440003\", // farm.Fingerprint64([]byte(\"my-task-list-550e8400-e29b-41d4-a716-446655440003\"))%100 = 69\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        70,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list below percentage - UUID - exclude disabled\",\n\t\t\ttaskListName:               \"my-task-list-550e8400-e29b-41d4-a716-446655440003\", // farm.Fingerprint64([]byte(\"my-task-list-550e8400-e29b-41d4-a716-446655440003\"))%100 = 69\n\t\t\twant:                       false,\n\t\t\tpercentageOnboarded:        70,\n\t\t\texcludeShortLivedTaskLists: false,\n\t\t},\n\t\t{\n\t\t\tname:                       \"task list above percentage - but UUID\",\n\t\t\ttaskListName:               \"my-task-list-550e8400-e29b-41d4-a716-446655440003\", // farm.Fingerprint64([]byte(\"my-task-list-550e8400-e29b-41d4-a716-446655440003\"))%100 = 69\n\t\t\twant:                       true,\n\t\t\tpercentageOnboarded:        60,\n\t\t\texcludeShortLivedTaskLists: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := TaskListExcludedFromShardDistributor(tt.taskListName, tt.percentageOnboarded, tt.excludeShortLivedTaskLists)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestIsbelowPercentage(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\thash       uint64\n\t\tpercentage uint64\n\t\twant       bool\n\t}{\n\t\t{\n\t\t\tname:       \"hash and percentage are 0\",\n\t\t\thash:       0,\n\t\t\tpercentage: 0,\n\t\t\twant:       false,\n\t\t},\n\n\t\t{\n\t\t\tname:       \"hash is 0 and percentage is 1\",\n\t\t\thash:       0,\n\t\t\tpercentage: 1,\n\t\t\twant:       true,\n\t\t},\n\t\t{\n\t\t\tname:       \"hash is 100 and percentage is 1 (we wrap)\",\n\t\t\thash:       100,\n\t\t\tpercentage: 1,\n\t\t\twant:       true,\n\t\t},\n\t\t{\n\t\t\tname:       \"hash is same as percentage\",\n\t\t\thash:       33,\n\t\t\tpercentage: 33,\n\t\t\twant:       false,\n\t\t},\n\t\t{\n\t\t\tname:       \"hash is big\",\n\t\t\thash:       10000000000033,\n\t\t\tpercentage: 34,\n\t\t\twant:       true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := isbelowPercentage(tt.hash, tt.percentage)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/messaging/ackManager.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage messaging\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\n\t\"go.uber.org/atomic\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype ackManager struct {\n\tsync.RWMutex\n\toutstandingMessages map[int64]bool // key->itemID, value->(true for acked/completed, false->for non acked)\n\treadLevel           int64          // Maximum itemID inserted into outstandingMessages\n\tackLevel            int64          // Maximum itemID below which all messages are acked\n\tbacklogCounter      atomic.Int64\n\tlogIncontinuousErr  bool // emit error for itemID being incontinuous when consuming for potential bugs\n\tlogger              log.Logger\n}\n\n// NewAckManager returns a AckManager without monitoring the itemIDs continousness.\n// For example, our internal matching task queue doesn't guarantee it.\nfunc NewAckManager(logger log.Logger) AckManager {\n\treturn newAckManager(false, logger)\n}\n\n// NewContinuousAckManager returns a ContinuousAckManager\n// it will emit error logs for itemIDs being incontinuous\n// This is useful for some message queue system that guarantees continuousness\n// that we want to monitor it's behaving correctly\nfunc NewContinuousAckManager(logger log.Logger) AckManager {\n\treturn newAckManager(true, logger)\n}\n\nfunc newAckManager(logIncontinuousErr bool, logger log.Logger) AckManager {\n\treturn &ackManager{\n\t\tlogger:              logger,\n\t\toutstandingMessages: make(map[int64]bool),\n\t\treadLevel:           -1,\n\t\tackLevel:            -1,\n\t\tlogIncontinuousErr:  logIncontinuousErr,\n\t}\n}\n\n// Registers message as in-flight and moves read level to it. Messages can be added in increasing order of messageID only.\n// NOTE that ackManager assumes adding messages is in order\nfunc (m *ackManager) ReadItem(itemID int64) error {\n\tm.Lock()\n\tdefer m.Unlock()\n\tm.backlogCounter.Inc()\n\tif m.readLevel >= itemID {\n\t\treturn fmt.Errorf(\"next item ID is less than or equal to current read level. itemID %d, readLevel %d\", itemID, m.readLevel)\n\t}\n\tif _, ok := m.outstandingMessages[itemID]; ok {\n\t\treturn fmt.Errorf(\"already present in outstanding items but hasn't added itemID:%d\", itemID)\n\t}\n\tm.readLevel = itemID\n\tif m.ackLevel == -1 {\n\t\t// because of ordering, the first itemID is the minimum to ack\n\t\tm.ackLevel = itemID - 1\n\t\tm.logger.Info(\"this is the very first itemID being read in this ackManager\",\n\t\t\ttag.TaskID(itemID),\n\t\t)\n\t}\n\tm.outstandingMessages[itemID] = false // true is for acked\n\treturn nil\n}\n\nfunc (m *ackManager) AckItem(itemID int64) (ackLevel int64) {\n\tm.Lock()\n\tdefer m.Unlock()\n\tif completed, ok := m.outstandingMessages[itemID]; ok && !completed {\n\t\tm.outstandingMessages[itemID] = true\n\t\tm.backlogCounter.Dec()\n\t} else {\n\t\tm.logger.Warn(\"Duplicated completion for item\",\n\t\t\ttag.TaskID(itemID))\n\t}\n\n\t// Update ackLevel\n\tfor current := m.ackLevel + 1; current <= m.readLevel; current++ {\n\t\tif acked, ok := m.outstandingMessages[current]; ok {\n\t\t\tif acked {\n\t\t\t\tm.ackLevel = current\n\t\t\t\tdelete(m.outstandingMessages, current)\n\t\t\t} else {\n\t\t\t\treturn m.ackLevel\n\t\t\t}\n\t\t} else {\n\t\t\tif m.logIncontinuousErr {\n\t\t\t\tm.logger.Error(\"potential bug, an item is probably skipped when adding\", tag.TaskID(current))\n\t\t\t}\n\t\t}\n\t}\n\treturn m.ackLevel\n}\n\nfunc (m *ackManager) GetReadLevel() int64 {\n\tm.RLock()\n\tdefer m.RUnlock()\n\treturn m.readLevel\n}\n\nfunc (m *ackManager) SetReadLevel(readLevel int64) {\n\tm.Lock()\n\tdefer m.Unlock()\n\tm.readLevel = readLevel\n}\n\nfunc (m *ackManager) GetAckLevel() int64 {\n\tm.RLock()\n\tdefer m.RUnlock()\n\treturn m.ackLevel\n}\n\nfunc (m *ackManager) SetAckLevel(ackLevel int64) {\n\tm.Lock()\n\tdefer m.Unlock()\n\tif ackLevel > m.ackLevel {\n\t\tm.ackLevel = ackLevel\n\t}\n\tif ackLevel > m.readLevel {\n\t\tm.readLevel = ackLevel\n\t}\n}\n\nfunc (m *ackManager) GetBacklogCount() int64 {\n\treturn m.backlogCounter.Load()\n}\n"
  },
  {
    "path": "common/messaging/ackManager_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage messaging\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestAckManager(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tm := NewAckManager(logger)\n\tm.SetAckLevel(100)\n\tassert.EqualValues(t, 100, m.GetAckLevel())\n\tassert.EqualValues(t, 100, m.GetReadLevel())\n\tconst t1 = 200\n\tconst t2 = 220\n\tconst t3 = 320\n\tconst t4 = 340\n\tconst t5 = 360\n\n\terr := m.ReadItem(t1)\n\tassert.Nil(t, err)\n\tassert.EqualValues(t, 100, m.GetAckLevel())\n\tassert.EqualValues(t, t1, m.GetReadLevel())\n\n\terr = m.ReadItem(t2)\n\tassert.Nil(t, err)\n\tassert.EqualValues(t, 100, m.GetAckLevel())\n\tassert.EqualValues(t, t2, m.GetReadLevel())\n\n\tm.AckItem(t2)\n\tassert.EqualValues(t, 100, m.GetAckLevel())\n\tassert.EqualValues(t, t2, m.GetReadLevel())\n\n\tm.AckItem(t1)\n\tassert.EqualValues(t, t2, m.GetAckLevel())\n\tassert.EqualValues(t, t2, m.GetReadLevel())\n\n\tm.SetAckLevel(300)\n\tassert.EqualValues(t, 300, m.GetAckLevel())\n\tassert.EqualValues(t, 300, m.GetReadLevel())\n\n\terr = m.ReadItem(t3)\n\tassert.Nil(t, err)\n\tassert.EqualValues(t, 300, m.GetAckLevel())\n\tassert.EqualValues(t, t3, m.GetReadLevel())\n\n\terr = m.ReadItem(t4)\n\tassert.Nil(t, err)\n\tassert.EqualValues(t, 300, m.GetAckLevel())\n\tassert.EqualValues(t, t4, m.GetReadLevel())\n\n\tm.AckItem(t3)\n\tassert.EqualValues(t, t3, m.GetAckLevel())\n\tassert.EqualValues(t, t4, m.GetReadLevel())\n\n\tm.AckItem(t4)\n\tassert.EqualValues(t, t4, m.GetAckLevel())\n\tassert.EqualValues(t, t4, m.GetReadLevel())\n\n\tm.SetReadLevel(t5)\n\tassert.EqualValues(t, t5, m.GetReadLevel())\n}\n"
  },
  {
    "path": "common/messaging/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package messaging -source interface.go -destination client_mock.go\n//\n\n// Package messaging is a generated GoMock package.\npackage messaging\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// NewConsumer mocks base method.\nfunc (m *MockClient) NewConsumer(appName, consumerName string) (Consumer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewConsumer\", appName, consumerName)\n\tret0, _ := ret[0].(Consumer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewConsumer indicates an expected call of NewConsumer.\nfunc (mr *MockClientMockRecorder) NewConsumer(appName, consumerName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewConsumer\", reflect.TypeOf((*MockClient)(nil).NewConsumer), appName, consumerName)\n}\n\n// NewProducer mocks base method.\nfunc (m *MockClient) NewProducer(appName string) (Producer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewProducer\", appName)\n\tret0, _ := ret[0].(Producer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewProducer indicates an expected call of NewProducer.\nfunc (mr *MockClientMockRecorder) NewProducer(appName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewProducer\", reflect.TypeOf((*MockClient)(nil).NewProducer), appName)\n}\n\n// MockConsumer is a mock of Consumer interface.\ntype MockConsumer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConsumerMockRecorder\n\tisgomock struct{}\n}\n\n// MockConsumerMockRecorder is the mock recorder for MockConsumer.\ntype MockConsumerMockRecorder struct {\n\tmock *MockConsumer\n}\n\n// NewMockConsumer creates a new mock instance.\nfunc NewMockConsumer(ctrl *gomock.Controller) *MockConsumer {\n\tmock := &MockConsumer{ctrl: ctrl}\n\tmock.recorder = &MockConsumerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConsumer) EXPECT() *MockConsumerMockRecorder {\n\treturn m.recorder\n}\n\n// Messages mocks base method.\nfunc (m *MockConsumer) Messages() <-chan Message {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Messages\")\n\tret0, _ := ret[0].(<-chan Message)\n\treturn ret0\n}\n\n// Messages indicates an expected call of Messages.\nfunc (mr *MockConsumerMockRecorder) Messages() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Messages\", reflect.TypeOf((*MockConsumer)(nil).Messages))\n}\n\n// Start mocks base method.\nfunc (m *MockConsumer) Start() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockConsumerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockConsumer)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockConsumer) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockConsumerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockConsumer)(nil).Stop))\n}\n\n// MockMessage is a mock of Message interface.\ntype MockMessage struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMessageMockRecorder\n\tisgomock struct{}\n}\n\n// MockMessageMockRecorder is the mock recorder for MockMessage.\ntype MockMessageMockRecorder struct {\n\tmock *MockMessage\n}\n\n// NewMockMessage creates a new mock instance.\nfunc NewMockMessage(ctrl *gomock.Controller) *MockMessage {\n\tmock := &MockMessage{ctrl: ctrl}\n\tmock.recorder = &MockMessageMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMessage) EXPECT() *MockMessageMockRecorder {\n\treturn m.recorder\n}\n\n// Ack mocks base method.\nfunc (m *MockMessage) Ack() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Ack\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Ack indicates an expected call of Ack.\nfunc (mr *MockMessageMockRecorder) Ack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ack\", reflect.TypeOf((*MockMessage)(nil).Ack))\n}\n\n// Nack mocks base method.\nfunc (m *MockMessage) Nack() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Nack\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Nack indicates an expected call of Nack.\nfunc (mr *MockMessageMockRecorder) Nack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Nack\", reflect.TypeOf((*MockMessage)(nil).Nack))\n}\n\n// Offset mocks base method.\nfunc (m *MockMessage) Offset() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Offset\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// Offset indicates an expected call of Offset.\nfunc (mr *MockMessageMockRecorder) Offset() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Offset\", reflect.TypeOf((*MockMessage)(nil).Offset))\n}\n\n// Partition mocks base method.\nfunc (m *MockMessage) Partition() int32 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Partition\")\n\tret0, _ := ret[0].(int32)\n\treturn ret0\n}\n\n// Partition indicates an expected call of Partition.\nfunc (mr *MockMessageMockRecorder) Partition() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Partition\", reflect.TypeOf((*MockMessage)(nil).Partition))\n}\n\n// Value mocks base method.\nfunc (m *MockMessage) Value() []byte {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Value\")\n\tret0, _ := ret[0].([]byte)\n\treturn ret0\n}\n\n// Value indicates an expected call of Value.\nfunc (mr *MockMessageMockRecorder) Value() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Value\", reflect.TypeOf((*MockMessage)(nil).Value))\n}\n\n// MockProducer is a mock of Producer interface.\ntype MockProducer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProducerMockRecorder\n\tisgomock struct{}\n}\n\n// MockProducerMockRecorder is the mock recorder for MockProducer.\ntype MockProducerMockRecorder struct {\n\tmock *MockProducer\n}\n\n// NewMockProducer creates a new mock instance.\nfunc NewMockProducer(ctrl *gomock.Controller) *MockProducer {\n\tmock := &MockProducer{ctrl: ctrl}\n\tmock.recorder = &MockProducerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProducer) EXPECT() *MockProducerMockRecorder {\n\treturn m.recorder\n}\n\n// Publish mocks base method.\nfunc (m *MockProducer) Publish(ctx context.Context, message any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Publish\", ctx, message)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Publish indicates an expected call of Publish.\nfunc (mr *MockProducerMockRecorder) Publish(ctx, message any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Publish\", reflect.TypeOf((*MockProducer)(nil).Publish), ctx, message)\n}\n\n// MockCloseableProducer is a mock of CloseableProducer interface.\ntype MockCloseableProducer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCloseableProducerMockRecorder\n\tisgomock struct{}\n}\n\n// MockCloseableProducerMockRecorder is the mock recorder for MockCloseableProducer.\ntype MockCloseableProducerMockRecorder struct {\n\tmock *MockCloseableProducer\n}\n\n// NewMockCloseableProducer creates a new mock instance.\nfunc NewMockCloseableProducer(ctrl *gomock.Controller) *MockCloseableProducer {\n\tmock := &MockCloseableProducer{ctrl: ctrl}\n\tmock.recorder = &MockCloseableProducerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCloseableProducer) EXPECT() *MockCloseableProducerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockCloseableProducer) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockCloseableProducerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockCloseableProducer)(nil).Close))\n}\n\n// Publish mocks base method.\nfunc (m *MockCloseableProducer) Publish(ctx context.Context, message any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Publish\", ctx, message)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Publish indicates an expected call of Publish.\nfunc (mr *MockCloseableProducerMockRecorder) Publish(ctx, message any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Publish\", reflect.TypeOf((*MockCloseableProducer)(nil).Publish), ctx, message)\n}\n\n// MockAckManager is a mock of AckManager interface.\ntype MockAckManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockAckManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockAckManagerMockRecorder is the mock recorder for MockAckManager.\ntype MockAckManagerMockRecorder struct {\n\tmock *MockAckManager\n}\n\n// NewMockAckManager creates a new mock instance.\nfunc NewMockAckManager(ctrl *gomock.Controller) *MockAckManager {\n\tmock := &MockAckManager{ctrl: ctrl}\n\tmock.recorder = &MockAckManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockAckManager) EXPECT() *MockAckManagerMockRecorder {\n\treturn m.recorder\n}\n\n// AckItem mocks base method.\nfunc (m *MockAckManager) AckItem(id int64) int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AckItem\", id)\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// AckItem indicates an expected call of AckItem.\nfunc (mr *MockAckManagerMockRecorder) AckItem(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AckItem\", reflect.TypeOf((*MockAckManager)(nil).AckItem), id)\n}\n\n// GetAckLevel mocks base method.\nfunc (m *MockAckManager) GetAckLevel() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevel\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetAckLevel indicates an expected call of GetAckLevel.\nfunc (mr *MockAckManagerMockRecorder) GetAckLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevel\", reflect.TypeOf((*MockAckManager)(nil).GetAckLevel))\n}\n\n// GetBacklogCount mocks base method.\nfunc (m *MockAckManager) GetBacklogCount() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBacklogCount\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetBacklogCount indicates an expected call of GetBacklogCount.\nfunc (mr *MockAckManagerMockRecorder) GetBacklogCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBacklogCount\", reflect.TypeOf((*MockAckManager)(nil).GetBacklogCount))\n}\n\n// GetReadLevel mocks base method.\nfunc (m *MockAckManager) GetReadLevel() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReadLevel\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetReadLevel indicates an expected call of GetReadLevel.\nfunc (mr *MockAckManagerMockRecorder) GetReadLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReadLevel\", reflect.TypeOf((*MockAckManager)(nil).GetReadLevel))\n}\n\n// ReadItem mocks base method.\nfunc (m *MockAckManager) ReadItem(id int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadItem\", id)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReadItem indicates an expected call of ReadItem.\nfunc (mr *MockAckManagerMockRecorder) ReadItem(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadItem\", reflect.TypeOf((*MockAckManager)(nil).ReadItem), id)\n}\n\n// SetAckLevel mocks base method.\nfunc (m *MockAckManager) SetAckLevel(ackLevel int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetAckLevel\", ackLevel)\n}\n\n// SetAckLevel indicates an expected call of SetAckLevel.\nfunc (mr *MockAckManagerMockRecorder) SetAckLevel(ackLevel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetAckLevel\", reflect.TypeOf((*MockAckManager)(nil).SetAckLevel), ackLevel)\n}\n\n// SetReadLevel mocks base method.\nfunc (m *MockAckManager) SetReadLevel(readLevel int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetReadLevel\", readLevel)\n}\n\n// SetReadLevel indicates an expected call of SetReadLevel.\nfunc (mr *MockAckManagerMockRecorder) SetReadLevel(readLevel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetReadLevel\", reflect.TypeOf((*MockAckManager)(nil).SetReadLevel), readLevel)\n}\n"
  },
  {
    "path": "common/messaging/errors.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage messaging\n\nimport \"errors\"\n\nvar (\n\t// ErrMessageSizeLimit indicate that message is rejected by server due to size limitation\n\tErrMessageSizeLimit = errors.New(\"message was too large, server rejected it to avoid allocation error\")\n)\n"
  },
  {
    "path": "common/messaging/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination client_mock.go\n\npackage messaging\n\nimport \"context\"\n\ntype (\n\t// Client is the interface used to abstract out interaction with messaging system for replication\n\tClient interface {\n\t\tNewConsumer(appName, consumerName string) (Consumer, error)\n\t\tNewProducer(appName string) (Producer, error)\n\t}\n\n\t// Consumer is the unified interface for both internal and external kafka clients\n\tConsumer interface {\n\t\t// Start starts the consumer\n\t\tStart() error\n\t\t// Stop stops the consumer\n\t\tStop()\n\t\t// Messages return the message channel for this consumer\n\t\tMessages() <-chan Message\n\t}\n\n\t// Message is the unified interface for a Kafka message\n\tMessage interface {\n\t\t// Value is a mutable reference to the message's value\n\t\tValue() []byte\n\t\t// Partition is the ID of the partition from which the message was read.\n\t\tPartition() int32\n\t\t// Offset is the message's offset.\n\t\tOffset() int64\n\t\t// Ack marks the message as successfully processed.\n\t\tAck() error\n\t\t// Nack marks the message processing as failed and the message will be retried or sent to DLQ.\n\t\tNack() error\n\t}\n\n\t// Producer is the interface used to send replication tasks to other clusters through replicator\n\tProducer interface {\n\t\tPublish(ctx context.Context, message interface{}) error\n\t}\n\n\t// CloseableProducer is a Producer that can be closed\n\tCloseableProducer interface {\n\t\tProducer\n\t\tClose() error\n\t}\n\n\t// AckManager convert out of order acks into ackLevel movement.\n\tAckManager interface {\n\t\t// Read an item into backlog for processing for ack\n\t\tReadItem(id int64) error\n\t\t// Get current max ID from read items\n\t\tGetReadLevel() int64\n\t\t// Set current max ID from read items\n\t\tSetReadLevel(readLevel int64)\n\t\t// Mark an item as done processing, and remove from backlog\n\t\tAckItem(id int64) (ackLevel int64)\n\t\t// Get current max level that can safely ack\n\t\tGetAckLevel() int64\n\t\t// Set current max level that can safely ack\n\t\tSetAckLevel(ackLevel int64)\n\t\t// GetBacklogCount return the of items that are waiting for ack\n\t\tGetBacklogCount() int64\n\t}\n)\n"
  },
  {
    "path": "common/messaging/kafka/client_impl.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/IBM/sarama\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// This is a default implementation of Client interface which makes use of uber-go/kafka-client as consumer\n\tclientImpl struct {\n\t\tconfig        *config.KafkaConfig\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t}\n)\n\nvar _ messaging.Client = (*clientImpl)(nil)\n\n// NewKafkaClient is used to create an instance of KafkaClient\nfunc NewKafkaClient(\n\tkc *config.KafkaConfig,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\t_ tally.Scope,\n\tcheckApp bool,\n) messaging.Client {\n\tkc.Validate(checkApp)\n\n\t// mapping from cluster name to list of broker ip addresses\n\tbrokers := map[string][]string{}\n\tfor cluster, cfg := range kc.Clusters {\n\t\tbrokers[cluster] = cfg.Brokers\n\t\tfor i := range brokers[cluster] {\n\t\t\tif !strings.Contains(cfg.Brokers[i], \":\") {\n\t\t\t\tcfg.Brokers[i] += \":9092\"\n\t\t\t}\n\t\t}\n\t}\n\n\t// mapping from topic name to cluster that has that topic\n\ttopicClusterAssignment := map[string][]string{}\n\tfor topic, cfg := range kc.Topics {\n\t\ttopicClusterAssignment[topic] = []string{cfg.Cluster}\n\t}\n\n\treturn &clientImpl{\n\t\tconfig:        kc,\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t}\n}\n\n// NewConsumer is used to create a Kafka consumer\nfunc (c *clientImpl) NewConsumer(app, consumerName string) (messaging.Consumer, error) {\n\ttopics := c.config.GetTopicsForApplication(app)\n\t// All defaut values are copied from uber/kafka-clientImpl bo keep the same behavior\n\tkafkaVersion := c.config.Version\n\tif kafkaVersion == \"\" {\n\t\tkafkaVersion = \"0.10.2.0\"\n\t}\n\n\tversion, err := sarama.ParseKafkaVersion(kafkaVersion)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tsaramaConfig := sarama.NewConfig()\n\tsaramaConfig.Version = version\n\tsaramaConfig.Consumer.Fetch.Default = 30 * 1024 * 1024 // 30MB.\n\tsaramaConfig.Consumer.Return.Errors = true\n\tsaramaConfig.Consumer.Offsets.CommitInterval = time.Second\n\tsaramaConfig.Consumer.Offsets.Initial = sarama.OffsetOldest\n\tsaramaConfig.Consumer.MaxProcessingTime = 250 * time.Millisecond\n\n\terr = c.initAuth(saramaConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdlqProducer, err := c.newProducerByTopic(topics.DLQTopic)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tclusterName := c.config.GetKafkaClusterForTopic(topics.Topic)\n\tbrokers := c.config.GetBrokersForKafkaCluster(clusterName)\n\treturn NewKafkaConsumer(dlqProducer, brokers, topics.Topic, consumerName, saramaConfig, c.metricsClient, c.logger)\n}\n\n// NewProducer is used to create a Kafka producer\nfunc (c *clientImpl) NewProducer(app string) (messaging.Producer, error) {\n\ttopics := c.config.GetTopicsForApplication(app)\n\treturn c.newProducerByTopic(topics.Topic)\n}\n\nfunc (c *clientImpl) newProducerByTopic(topic string) (messaging.Producer, error) {\n\tkafkaClusterName := c.config.GetKafkaClusterForTopic(topic)\n\tbrokers := c.config.GetBrokersForKafkaCluster(kafkaClusterName)\n\n\tconfig := sarama.NewConfig()\n\tconfig.Producer.Return.Successes = true\n\terr := c.initAuth(config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tproducer, err := sarama.NewSyncProducer(brokers, config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif c.metricsClient != nil {\n\t\tc.logger.Info(\"Create producer with metricsClient\")\n\t\twithMetricsOpt := messaging.WithMetricTags(metrics.TopicTag(topic))\n\t\treturn messaging.NewMetricProducer(NewKafkaProducer(topic, producer, c.logger), c.metricsClient, withMetricsOpt), nil\n\t}\n\treturn NewKafkaProducer(topic, producer, c.logger), nil\n}\n\nfunc (c *clientImpl) initAuth(saramaConfig *sarama.Config) error {\n\ttlsConfig, err := c.config.TLS.ToTLSConfig()\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"Error creating Kafka TLS config %v\", err))\n\t}\n\n\t// TLS support\n\tsaramaConfig.Net.TLS.Enable = tlsConfig != nil\n\tsaramaConfig.Net.TLS.Config = tlsConfig\n\n\t// SASL support\n\tsaramaConfig.Net.SASL.Enable = c.config.SASL.Enabled\n\tsaramaConfig.Net.SASL.User = c.config.SASL.User\n\tsaramaConfig.Net.SASL.Password = c.config.SASL.Password\n\tsaramaConfig.Net.SASL.Handshake = true\n\n\tif c.config.SASL.Enabled {\n\t\tif c.config.SASL.Algorithm == \"sha512\" {\n\t\t\tsaramaConfig.Net.SASL.SCRAMClientGeneratorFunc = func() sarama.SCRAMClient {\n\t\t\t\treturn &authorization.XDGSCRAMClient{HashGeneratorFcn: authorization.SHA512}\n\t\t\t}\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypeSCRAMSHA512\n\t\t} else if c.config.SASL.Algorithm == \"sha256\" {\n\t\t\tsaramaConfig.Net.SASL.SCRAMClientGeneratorFunc = func() sarama.SCRAMClient {\n\t\t\t\treturn &authorization.XDGSCRAMClient{HashGeneratorFcn: authorization.SHA256}\n\t\t\t}\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypeSCRAMSHA256\n\t\t} else if c.config.SASL.Algorithm == \"plain\" {\n\t\t\tsaramaConfig.Net.SASL.Mechanism = sarama.SASLTypePlaintext\n\t\t} else {\n\t\t\treturn fmt.Errorf(\"invalid SHA algorithm %s: can be either sha256 or sha512\", c.config.SASL.Algorithm)\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/messaging/kafka/client_impl_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestNewKafkaClient(t *testing.T) {\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t)\n\ttestCases := []struct {\n\t\tname        string\n\t\tconfig      *config.KafkaConfig\n\t\tcheckApp    bool\n\t\texpectedErr string\n\t}{\n\t\t{\n\t\t\tname: \"Missing clusters\",\n\t\t\tconfig: &config.KafkaConfig{\n\t\t\t\tClusters: map[string]config.ClusterConfig{},\n\t\t\t},\n\t\t\tcheckApp:    true,\n\t\t\texpectedErr: \"Empty Kafka Cluster Config\",\n\t\t},\n\t\t{\n\t\t\tname: \"Missing topics\",\n\t\t\tconfig: &config.KafkaConfig{\n\t\t\t\tClusters: map[string]config.ClusterConfig{\n\t\t\t\t\t\"testCluster\": {\n\t\t\t\t\t\tBrokers: []string{\"testBrokers\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopics: map[string]config.TopicConfig{},\n\t\t\t},\n\t\t\tcheckApp:    true,\n\t\t\texpectedErr: \"Empty Topics Config\",\n\t\t},\n\t\t{\n\t\t\tname: \"Missing Applications\",\n\t\t\tconfig: &config.KafkaConfig{\n\t\t\t\tClusters: map[string]config.ClusterConfig{\n\t\t\t\t\t\"test-cluster\": {\n\t\t\t\t\t\tBrokers: []string{\"test-brokers\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopics: map[string]config.TopicConfig{\n\t\t\t\t\t\"test-topic\": {\n\t\t\t\t\t\tCluster: \"test-cluster\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tApplications: map[string]config.TopicList{},\n\t\t\t},\n\t\t\tcheckApp:    true,\n\t\t\texpectedErr: \"Empty Applications Config\",\n\t\t},\n\t\t{\n\t\t\tname: \"Missing topics config\",\n\t\t\tconfig: &config.KafkaConfig{\n\t\t\t\tClusters: map[string]config.ClusterConfig{\n\t\t\t\t\t\"test-cluster\": {\n\t\t\t\t\t\tBrokers: []string{\"test-brokers\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopics: map[string]config.TopicConfig{\n\t\t\t\t\t\"test-topic\": {\n\t\t\t\t\t\tCluster: \"test-cluster\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tApplications: map[string]config.TopicList{\n\t\t\t\t\t\"test-app\": {\n\t\t\t\t\t\tTopic:    \"test-topic\",\n\t\t\t\t\t\tDLQTopic: \"test-topic-dlq\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tcheckApp:    true,\n\t\t\texpectedErr: \"Missing Topic Config for Topic test-topic-dlq\",\n\t\t},\n\t\t{\n\t\t\tname: \"Normal Case\",\n\t\t\tconfig: &config.KafkaConfig{\n\t\t\t\tClusters: map[string]config.ClusterConfig{\n\t\t\t\t\t\"test-cluster\": {\n\t\t\t\t\t\tBrokers: []string{\"test-brokers\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTopics: map[string]config.TopicConfig{\n\t\t\t\t\t\"test-topic\": {\n\t\t\t\t\t\tCluster: \"test-cluster\",\n\t\t\t\t\t},\n\t\t\t\t\t\"test-topic-dlq\": {\n\t\t\t\t\t\tCluster: \"test-cluster\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tApplications: map[string]config.TopicList{\n\t\t\t\t\t\"test-app\": {\n\t\t\t\t\t\tTopic:    \"test-topic\",\n\t\t\t\t\t\tDLQTopic: \"test-topic-dlq\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tcheckApp: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdefer func() {\n\t\t\t\tif r := recover(); r != nil {\n\t\t\t\t\tassert.Equal(t, tc.expectedErr, r)\n\t\t\t\t}\n\t\t\t}()\n\t\t\tkafkaClient := NewKafkaClient(tc.config, metricsClient, logger, nil, tc.checkApp)\n\t\t\t// Type assert to *clientImpl to access struct fields\n\t\t\tclient, ok := kafkaClient.(*clientImpl)\n\t\t\tassert.True(t, ok, \"Expected kafkaClient to be of type *clientImpl\")\n\t\t\tassert.Equal(t, tc.config, client.config)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/messaging/kafka/consumer_impl.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/IBM/sarama\"\n\t\"golang.org/x/net/context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst rcvBufferSize = 2 * 1024\nconst dlqPublishTimeout = time.Minute\n\ntype (\n\t// a wrapper of sarama consumer group for our consumer interface\n\tconsumerImpl struct {\n\t\ttopic           string\n\t\tconsumerHandler *consumerHandlerImpl\n\t\tconsumerGroup   sarama.ConsumerGroup\n\t\tmsgChan         <-chan messaging.Message\n\t\twg              sync.WaitGroup\n\t\tcancelFunc      context.CancelFunc\n\n\t\tlogger log.Logger\n\t}\n\n\t// consumerHandlerImpl represents a Sarama consumer group consumer\n\t// It's for passing into sarama consumer group API\n\tconsumerHandlerImpl struct {\n\t\tsync.RWMutex\n\t\tdlqProducer messaging.Producer\n\n\t\ttopic          string\n\t\tcurrentSession sarama.ConsumerGroupSession\n\t\tmsgChan        chan<- messaging.Message\n\t\tmanager        *partitionAckManager\n\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t\tthrottleRetry *backoff.ThrottleRetry\n\t}\n\n\tmessageImpl struct {\n\t\tsaramaMsg *sarama.ConsumerMessage\n\t\tsession   sarama.ConsumerGroupSession\n\t\thandler   *consumerHandlerImpl\n\t\tlogger    log.Logger\n\t}\n)\n\nvar _ messaging.Message = (*messageImpl)(nil)\nvar _ messaging.Consumer = (*consumerImpl)(nil)\n\nfunc NewKafkaConsumer(\n\tdlqProducer messaging.Producer,\n\tbrokers []string,\n\ttopic string,\n\tconsumerName string,\n\tsaramaConfig *sarama.Config,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) (messaging.Consumer, error) {\n\tconsumerGroup, err := sarama.NewConsumerGroup(brokers, consumerName, saramaConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmsgChan := make(chan messaging.Message, rcvBufferSize)\n\tconsumerHandler := newConsumerHandlerImpl(dlqProducer, topic, msgChan, metricsClient, logger)\n\n\treturn &consumerImpl{\n\t\ttopic:           topic,\n\t\tconsumerHandler: consumerHandler,\n\t\tconsumerGroup:   consumerGroup,\n\t\tmsgChan:         msgChan,\n\t\tlogger:          logger,\n\t}, nil\n}\n\nfunc (c *consumerImpl) Start() error {\n\tctx, cancel := context.WithCancel(context.Background())\n\tc.cancelFunc = cancel\n\tc.wg.Add(1)\n\n\t// consumer loop\n\tgo func() {\n\t\tdefer c.wg.Done()\n\t\tfor {\n\t\t\t// `Consume` should be called inside an infinite loop, when a\n\t\t\t// server-side rebalance happens, the consumer session will need to be\n\t\t\t// recreated to get the new claims\n\t\t\tif err := c.consumerGroup.Consume(ctx, []string{c.topic}, c.consumerHandler); err != nil {\n\t\t\t\tc.logger.Error(\"Error from consumer: %v\", tag.Error(err))\n\t\t\t}\n\t\t\t// check if context was cancelled, signaling that the consumer should stop\n\t\t\tif ctx.Err() != nil {\n\t\t\t\tc.logger.Info(\"context was cancel, stopping consumer loop\")\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n\treturn nil\n}\n\n// Stop stops the consumer\nfunc (c *consumerImpl) Stop() {\n\tc.logger.Info(\"Stopping consumer\")\n\tc.cancelFunc()\n\tc.logger.Info(\"Waiting consumer goroutines to complete\")\n\tc.wg.Wait()\n\tc.logger.Info(\"Stopping consumer handler and group\")\n\tc.consumerHandler.stop()\n\tc.consumerGroup.Close()\n\tc.logger.Info(\"Stopped consumer\")\n}\n\n// Messages return the message channel for this consumer\nfunc (c *consumerImpl) Messages() <-chan messaging.Message {\n\treturn c.msgChan\n}\n\nfunc newConsumerHandlerImpl(\n\tdlqProducer messaging.Producer,\n\ttopic string,\n\tmsgChan chan<- messaging.Message,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) *consumerHandlerImpl {\n\treturn &consumerHandlerImpl{\n\t\tdlqProducer: dlqProducer,\n\t\ttopic:       topic,\n\t\tmsgChan:     msgChan,\n\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(common.CreateDlqPublishRetryPolicy()),\n\t\t\tbackoff.WithRetryableError(func(_ error) bool { return true }),\n\t\t),\n\t}\n}\n\n// Setup is run at the beginning of a new session, before ConsumeClaim\nfunc (h *consumerHandlerImpl) Setup(session sarama.ConsumerGroupSession) error {\n\th.Lock()\n\tdefer h.Unlock()\n\n\th.currentSession = session\n\th.manager = newPartitionAckManager(h.metricsClient, h.logger)\n\th.logger.Info(\"start consumer group session\", tag.Name(string(session.GenerationID())))\n\th.metricsClient.IncCounter(metrics.MessagingClientConsumerScope, metrics.KafkaConsumerSessionStart)\n\n\treturn nil\n}\n\nfunc (h *consumerHandlerImpl) getCurrentSession() sarama.ConsumerGroupSession {\n\th.RLock()\n\tdefer h.RUnlock()\n\n\treturn h.currentSession\n}\n\nfunc (h *consumerHandlerImpl) completeMessage(message *messageImpl, isAck bool) error {\n\th.RLock()\n\tdefer h.RUnlock()\n\n\tif !isAck {\n\t\top := func(ctx context.Context) error {\n\t\t\t// NOTE: current KafkaProducer is not taking use the this context, because saramaProducer doesn't support it\n\t\t\t// https://github.com/IBM/sarama/issues/1849\n\t\t\tctx, cancel := context.WithTimeout(ctx, dlqPublishTimeout)\n\t\t\terr := h.dlqProducer.Publish(ctx, message.saramaMsg)\n\t\t\tcancel()\n\t\t\treturn err\n\t\t}\n\t\terr := h.throttleRetry.Do(context.Background(), op)\n\t\tif err != nil {\n\t\t\th.metricsClient.IncCounter(metrics.MessagingClientConsumerScope, metrics.KafkaConsumerMessageNackDlqErr)\n\t\t\th.logger.Error(\"Fail to publish message to DLQ when nacking message, please take action!!\",\n\t\t\t\ttag.KafkaPartition(message.Partition()),\n\t\t\t\ttag.KafkaOffset(message.Offset()))\n\t\t} else {\n\t\t\th.logger.Warn(\"nack message and publish to DLQ\",\n\t\t\t\ttag.KafkaPartition(message.Partition()),\n\t\t\t\ttag.KafkaOffset(message.Offset()))\n\t\t}\n\t}\n\tackLevel, err := h.manager.CompleteMessage(message.Partition(), message.Offset(), isAck)\n\tif err != nil {\n\t\th.logger.Error(\"Failed to complete an message that hasn't been added to the partition\",\n\t\t\ttag.KafkaPartition(message.Partition()),\n\t\t\ttag.KafkaOffset(message.Offset()))\n\t\treturn err\n\t}\n\th.currentSession.MarkOffset(h.topic, message.Partition(), ackLevel+1, \"\")\n\treturn nil\n}\n\n// Cleanup is run at the end of a session, once all ConsumeClaim goroutines have exited\nfunc (h *consumerHandlerImpl) Cleanup(sarama.ConsumerGroupSession) error {\n\treturn nil\n}\n\n// ConsumeClaim must start a consumer loop of ConsumerGroupClaim's Messages().\nfunc (h *consumerHandlerImpl) ConsumeClaim(session sarama.ConsumerGroupSession, claim sarama.ConsumerGroupClaim) error {\n\th.RLock()\n\tdefer h.RUnlock()\n\n\t// NOTE: Do not move the code below to a goroutine.\n\t// The `ConsumeClaim` itself is called within a goroutine:\n\tfor message := range claim.Messages() {\n\t\th.manager.AddMessage(message.Partition, message.Offset)\n\t\th.msgChan <- &messageImpl{\n\t\t\tsaramaMsg: message,\n\t\t\tsession:   session,\n\t\t\thandler:   h,\n\t\t\tlogger:    h.logger,\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (h *consumerHandlerImpl) stop() {\n\tclose(h.msgChan)\n}\n\nfunc (m *messageImpl) Value() []byte {\n\treturn m.saramaMsg.Value\n}\n\nfunc (m *messageImpl) Partition() int32 {\n\treturn m.saramaMsg.Partition\n}\n\nfunc (m *messageImpl) Offset() int64 {\n\treturn m.saramaMsg.Offset\n}\n\nfunc (m *messageImpl) Ack() error {\n\tif m.isFromPreviousSession() {\n\t\treturn nil\n\t}\n\treturn m.handler.completeMessage(m, true)\n}\n\nfunc (m *messageImpl) Nack() error {\n\tif m.isFromPreviousSession() {\n\t\treturn nil\n\t}\n\treturn m.handler.completeMessage(m, false)\n}\n\nfunc (m *messageImpl) isFromPreviousSession() bool {\n\tif m.session.GenerationID() != m.handler.getCurrentSession().GenerationID() {\n\t\tm.logger.Warn(\"Skip message that is from a previous session\",\n\t\t\ttag.KafkaPartition(m.saramaMsg.Partition),\n\t\t\ttag.KafkaOffset(m.saramaMsg.Offset),\n\t\t\ttag.Name(string(m.session.GenerationID())),\n\t\t\ttag.Value(m.handler.getCurrentSession().GenerationID()),\n\t\t)\n\t\treturn true\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/messaging/kafka/consumer_impl_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/IBM/sarama\"\n\t\"github.com/IBM/sarama/mocks\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestNewConsumer(t *testing.T) {\n\tmockProducer := mocks.NewSyncProducer(t, nil)\n\tgroup := \"tests\"\n\tmockBroker := initMockBroker(t, group)\n\tdefer mockBroker.Close()\n\tbrokerAddr := []string{mockBroker.Addr()}\n\tkafkaConfig := &config.KafkaConfig{\n\t\tClusters: map[string]config.ClusterConfig{\n\t\t\t\"test-cluster\": {\n\t\t\t\tBrokers: brokerAddr,\n\t\t\t},\n\t\t},\n\t\tTopics: map[string]config.TopicConfig{\n\t\t\t\"test-topic\": {\n\t\t\t\tCluster: \"test-cluster\",\n\t\t\t},\n\t\t\t\"test-topic-dlq\": {\n\t\t\t\tCluster: \"test-cluster\",\n\t\t\t},\n\t\t},\n\t\tApplications: map[string]config.TopicList{\n\t\t\t\"test-app\": {\n\t\t\t\tTopic:    \"test-topic\",\n\t\t\t\tDLQTopic: \"test-topic-dlq\",\n\t\t\t},\n\t\t},\n\t}\n\ttopic := \"test-topic\"\n\tconsumerName := \"test-consumer\"\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t)\n\tkafkaProducer := NewKafkaProducer(topic, mockProducer, logger)\n\tclusterName := kafkaConfig.GetKafkaClusterForTopic(topic)\n\tbrokers := kafkaConfig.GetBrokersForKafkaCluster(clusterName)\n\tconsumer, err := NewKafkaConsumer(kafkaProducer, brokers, topic, consumerName,\n\t\tnil, metricsClient, logger)\n\tassert.NoError(t, err, \"An error was not expected but got %v\", err)\n\tassert.NotNil(t, consumer, \"Expected consumer but got nil\")\n\n\terr = consumer.Start()\n\tassert.NoError(t, err)\n\n\tconsumer.Stop()\n}\n\nfunc TestNewConsumerHandlerImpl(t *testing.T) {\n\ttopic := \"test-topic\"\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t)\n\tmockProducer := mocks.NewSyncProducer(t, nil)\n\tkafkaProducer := NewKafkaProducer(topic, mockProducer, logger)\n\tmsgChan := make(chan messaging.Message, 1)\n\tconsumerHandler := newConsumerHandlerImpl(kafkaProducer, topic, msgChan, metricsClient, logger)\n\n\tassert.NotNil(t, consumerHandler)\n\tassert.Equal(t, kafkaProducer, consumerHandler.dlqProducer)\n\tassert.Equal(t, topic, consumerHandler.topic)\n\t// Close the channel at the end of the test\n\tclose(msgChan)\n}\n\n// test multiple methods related to messageImpl since setup is repeated\nfunc TestMessageImpl(t *testing.T) {\n\ttopic := \"test-topic\"\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t)\n\tmockProducer := mocks.NewSyncProducer(t, nil)\n\tkafkaProducer := NewKafkaProducer(topic, mockProducer, logger)\n\tmsgChan := make(chan messaging.Message, 1)\n\tconsumerHandler := newConsumerHandlerImpl(kafkaProducer, topic, msgChan, metricsClient, logger)\n\tconsumerGroupSession := NewMockConsumerGroupSession(int32(1))\n\tpartition := int32(100)\n\toffset := int64(0)\n\tmsgImpl := &messageImpl{\n\t\tsaramaMsg: &sarama.ConsumerMessage{\n\t\t\tTopic:     topic,\n\t\t\tPartition: partition,\n\t\t\tOffset:    offset,\n\t\t},\n\t\tsession: consumerGroupSession,\n\t\thandler: consumerHandler,\n\t\tlogger:  logger,\n\t}\n\n\t// Ack message that is from a previous session\n\terr := msgImpl.handler.Setup(NewMockConsumerGroupSession(int32(2)))\n\tassert.NoError(t, err)\n\terr = msgImpl.Ack()\n\tassert.NoError(t, err)\n\n\t// normal case\n\terr = msgImpl.handler.Setup(NewMockConsumerGroupSession(int32(1)))\n\tassert.NoError(t, err)\n\tmsgImpl.handler.manager.AddMessage(partition, offset)\n\terr = msgImpl.Ack()\n\tassert.NoError(t, err)\n\n\t// Nack message that is from a previous session\n\terr = msgImpl.handler.Setup(NewMockConsumerGroupSession(int32(2)))\n\tassert.NoError(t, err)\n\terr = msgImpl.Nack()\n\tassert.NoError(t, err)\n\n\t// normal case\n\terr = msgImpl.handler.Setup(NewMockConsumerGroupSession(int32(1)))\n\tassert.NoError(t, err)\n\tmockProducer.ExpectSendMessageAndSucceed()\n\tmsgImpl.handler.manager.AddMessage(partition, offset)\n\terr = msgImpl.Nack()\n\tassert.NoError(t, err)\n\n\tclose(msgChan)\n}\n\n// MockConsumerGroupSession implements sarama.ConsumerGroupSession for testing purposes.\ntype MockConsumerGroupSession struct {\n\tclaims       map[string][]int32\n\tmemberID     string\n\tgenerationID int32\n\toffsets      map[string]map[int32]int64\n\tcommitCalled bool\n\tcontext      context.Context\n}\n\nfunc NewMockConsumerGroupSession(generationID int32) *MockConsumerGroupSession {\n\treturn &MockConsumerGroupSession{\n\t\tclaims:       map[string][]int32{},\n\t\tmemberID:     \"test-member\",\n\t\tgenerationID: generationID,\n\t\toffsets:      map[string]map[int32]int64{},\n\t\tcommitCalled: false,\n\t}\n}\n\nfunc (m *MockConsumerGroupSession) Claims() map[string][]int32 {\n\treturn m.claims\n}\n\nfunc (m *MockConsumerGroupSession) MemberID() string {\n\treturn m.memberID\n}\n\nfunc (m *MockConsumerGroupSession) GenerationID() int32 {\n\treturn m.generationID\n}\n\nfunc (m *MockConsumerGroupSession) MarkOffset(topic string, partition int32, offset int64, metadata string) {\n\t// not needed for testing\n}\n\nfunc (m *MockConsumerGroupSession) Commit() {\n\t// not needed for testing\n}\n\nfunc (m *MockConsumerGroupSession) ResetOffset(topic string, partition int32, offset int64, metadata string) {\n\t// not needed for testing\n}\n\nfunc (m *MockConsumerGroupSession) MarkMessage(msg *sarama.ConsumerMessage, metadata string) {\n\t// not needed for testing\n}\n\nfunc (m *MockConsumerGroupSession) Context() context.Context {\n\t// Return a context, you can use context.Background() or a custom context if needed\n\treturn m.context\n}\n\nfunc initMockBroker(t *testing.T, group string) *sarama.MockBroker {\n\ttopics := []string{\"test-topic\"}\n\tmockBroker := sarama.NewMockBroker(t, 0)\n\n\tmockBroker.SetHandlerByMap(map[string]sarama.MockResponse{\n\t\t\"MetadataRequest\": sarama.NewMockMetadataResponse(t).\n\t\t\tSetBroker(mockBroker.Addr(), mockBroker.BrokerID()).\n\t\t\tSetLeader(topics[0], 0, mockBroker.BrokerID()).\n\t\t\tSetController(mockBroker.BrokerID()),\n\t\t\"FindCoordinatorRequest\": sarama.NewMockFindCoordinatorResponse(t).\n\t\t\tSetCoordinator(sarama.CoordinatorGroup, group, mockBroker),\n\t})\n\treturn mockBroker\n}\n"
  },
  {
    "path": "common/messaging/kafka/partition_ack_manager.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"errors\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// Used to convert out of order acks into ackLevel movement,\n// assuming reading messages is in order and continuous(no skipping)\ntype partitionAckManager struct {\n\tsync.RWMutex\n\tackMgrs map[int32]messaging.AckManager // map from partition to its ackManager\n\tscopes  map[int32]metrics.Scope        // map from partition to its Scope\n\n\tmetricsClient metrics.Client\n\tlogger        log.Logger\n}\n\nfunc newPartitionAckManager(metricsClient metrics.Client, logger log.Logger) *partitionAckManager {\n\treturn &partitionAckManager{\n\t\tackMgrs: make(map[int32]messaging.AckManager),\n\t\tscopes:  make(map[int32]metrics.Scope),\n\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t}\n}\n\n// AddMessage mark a messageID as read and waiting for completion\nfunc (pam *partitionAckManager) AddMessage(partitionID int32, messageID int64) {\n\tvar err error\n\tpam.RLock()\n\tif am, ok := pam.ackMgrs[partitionID]; ok {\n\t\terr = am.ReadItem(messageID)\n\t\tpam.scopes[partitionID].IncCounter(metrics.KafkaConsumerMessageIn)\n\n\t\tpam.RUnlock()\n\t} else {\n\t\tpam.RUnlock()\n\t\tpam.Lock()\n\n\t\tpartitionLogger := pam.logger.WithTags(tag.KafkaPartition(partitionID))\n\t\tam := messaging.NewContinuousAckManager(partitionLogger)\n\t\tpam.ackMgrs[partitionID] = am\n\n\t\tscope := pam.metricsClient.Scope(metrics.MessagingClientConsumerScope, metrics.KafkaPartitionTag(partitionID))\n\t\tpam.scopes[partitionID] = scope\n\t\tscope.IncCounter(metrics.KafkaConsumerMessageIn)\n\n\t\terr = am.ReadItem(messageID)\n\t\tpam.Unlock()\n\t}\n\tif err != nil {\n\t\tpam.logger.Warn(\"potential bug when adding message to ackManager\", tag.Error(err), tag.KafkaPartition(partitionID), tag.TaskID(messageID))\n\t}\n}\n\n// CompleteMessage complete the message from ack/nack kafka message\nfunc (pam *partitionAckManager) CompleteMessage(partitionID int32, messageID int64, isAck bool) (ackLevel int64, err error) {\n\tpam.RLock()\n\tdefer pam.RUnlock()\n\tif am, ok := pam.ackMgrs[partitionID]; ok {\n\t\tackLevel = am.AckItem(messageID)\n\t\tif isAck {\n\t\t\tpam.scopes[partitionID].IncCounter(metrics.KafkaConsumerMessageAck)\n\t\t} else {\n\t\t\tpam.scopes[partitionID].IncCounter(metrics.KafkaConsumerMessageNack)\n\t\t}\n\t} else {\n\t\treturn -1, errors.New(\"Failed to complete an message that hasn't been added to the partition\")\n\t}\n\treturn ackLevel, nil\n}\n"
  },
  {
    "path": "common/messaging/kafka/partition_ack_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestAddMessage(t *testing.T) {\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t) // Mocked\n\tpam := newPartitionAckManager(metricsClient, logger)\n\tpartitionID := int32(1)\n\tmessageID := int64(100)\n\t// Test adding a message\n\tpam.AddMessage(partitionID, messageID)\n\n\t// Verify the message is added to the ack manager\n\tpam.RLock() // Read lock since we are only reading data\n\t_, ok := pam.ackMgrs[partitionID]\n\tpam.RUnlock()\n\n\tassert.True(t, ok, \"AckManager for partition %v was not created\", partitionID)\n}\n\nfunc TestCompleteMessage(t *testing.T) {\n\t// Setup\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\tlogger := testlogger.New(t)\n\tpam := newPartitionAckManager(metricsClient, logger)\n\n\tpartitionID := int32(1)\n\tmessageID := int64(100)\n\n\ttestCases := []struct {\n\t\tname        string\n\t\tpartitionID int32\n\t\tmessageID   int64\n\t\tisAck       bool\n\t\texpected    int64\n\t\thasErr      bool\n\t}{\n\t\t{\n\t\t\tname:        \"Acknowledge the message\",\n\t\t\tpartitionID: partitionID,\n\t\t\tmessageID:   messageID,\n\t\t\tisAck:       true,\n\t\t\texpected:    int64(100),\n\t\t\thasErr:      false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Not acknowledge the message\",\n\t\t\tpartitionID: partitionID,\n\t\t\tmessageID:   messageID,\n\t\t\tisAck:       false,\n\t\t\texpected:    int64(100),\n\t\t\thasErr:      false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Not exist partition\",\n\t\t\tpartitionID: partitionID + 1,\n\t\t\tmessageID:   messageID,\n\t\t\tisAck:       true,\n\t\t\texpected:    int64(-1),\n\t\t\thasErr:      true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Add a message first to simulate a real-world scenario, this will create ackMgr for partition\n\t\t\tpam.AddMessage(partitionID, messageID)\n\n\t\t\tackLevel, err := pam.CompleteMessage(tc.partitionID, tc.messageID, tc.isAck)\n\t\t\tassert.True(t, ackLevel == tc.expected, \"Test case %s failed: expected ackLevel %d, got %d\", tc.name, tc.expected, ackLevel)\n\n\t\t\tif tc.hasErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error but none was found\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"An error was not expected but got %v\", err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/messaging/kafka/producer_impl.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/IBM/sarama\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n)\n\ntype (\n\tproducerImpl struct {\n\t\ttopic      string\n\t\tproducer   sarama.SyncProducer\n\t\tmsgEncoder codec.BinaryEncoder\n\t\tlogger     log.Logger\n\t}\n)\n\nvar _ messaging.Producer = (*producerImpl)(nil)\n\n// NewKafkaProducer is used to create the Kafka based producer implementation\nfunc NewKafkaProducer(topic string, producer sarama.SyncProducer, logger log.Logger) messaging.Producer {\n\treturn &producerImpl{\n\t\ttopic:      topic,\n\t\tproducer:   producer,\n\t\tmsgEncoder: codec.NewThriftRWEncoder(),\n\t\tlogger:     logger.WithTags(tag.KafkaTopicName(topic)),\n\t}\n}\n\n// Publish is used to send messages to other clusters through Kafka topic\n// TODO implement context when https://github.com/IBM/sarama/issues/1849 is supported\nfunc (p *producerImpl) Publish(_ context.Context, msg interface{}) error {\n\tmessage, err := p.getProducerMessage(msg)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tpartition, offset, err := p.producer.SendMessage(message)\n\tif err != nil {\n\t\tp.logger.Warn(\"Failed to publish message to kafka\",\n\t\t\ttag.KafkaPartition(partition),\n\t\t\ttag.KafkaPartitionKey(message.Key),\n\t\t\ttag.KafkaOffset(offset),\n\t\t\ttag.Error(err))\n\t\treturn p.convertErr(err)\n\t}\n\n\treturn nil\n}\n\n// Close is used to close Kafka publisher\nfunc (p *producerImpl) Close() error {\n\treturn p.convertErr(p.producer.Close())\n}\n\nfunc (p *producerImpl) serializeThrift(input codec.ThriftObject) ([]byte, error) {\n\tpayload, err := p.msgEncoder.Encode(input)\n\tif err != nil {\n\t\tp.logger.Error(\"Failed to serialize thrift object\", tag.Error(err))\n\n\t\treturn nil, err\n\t}\n\n\treturn payload, nil\n}\n\nfunc (p *producerImpl) getProducerMessage(message interface{}) (*sarama.ProducerMessage, error) {\n\tswitch message := message.(type) {\n\tcase *indexer.Message:\n\t\tpayload, err := p.serializeThrift(message)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tmsg := &sarama.ProducerMessage{\n\t\t\tTopic: p.topic,\n\t\t\tKey:   sarama.StringEncoder(message.GetWorkflowID()),\n\t\t\tValue: sarama.ByteEncoder(payload),\n\t\t}\n\t\treturn msg, nil\n\tcase *sarama.ConsumerMessage:\n\t\tmsg := &sarama.ProducerMessage{\n\t\t\tTopic: p.topic,\n\t\t\tKey:   sarama.ByteEncoder(message.Key),\n\t\t\tValue: sarama.ByteEncoder(message.Value),\n\t\t}\n\t\treturn msg, nil\n\tcase *indexer.PinotMessage:\n\t\tmsg := &sarama.ProducerMessage{\n\t\t\tTopic: p.topic,\n\t\t\tKey:   sarama.StringEncoder(message.GetWorkflowID()),\n\t\t\tValue: sarama.ByteEncoder(message.GetPayload()),\n\t\t}\n\t\treturn msg, nil\n\tcase *sqlblobs.AsyncRequestMessage:\n\t\tpayload, err := p.serializeThrift(message)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tmsg := &sarama.ProducerMessage{\n\t\t\tTopic: p.topic,\n\t\t\tKey:   sarama.StringEncoder(message.GetPartitionKey()),\n\t\t\tValue: sarama.ByteEncoder(payload),\n\t\t}\n\t\treturn msg, nil\n\tdefault:\n\t\treturn nil, errors.New(\"unknown producer message type\")\n\t}\n}\n\nfunc (p *producerImpl) convertErr(err error) error {\n\tswitch err {\n\tcase sarama.ErrMessageSizeTooLarge:\n\t\treturn messaging.ErrMessageSizeLimit\n\tdefault:\n\t\treturn err\n\t}\n}\n"
  },
  {
    "path": "common/messaging/kafka/producer_impl_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage kafka\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/IBM/sarama/mocks\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestNewKafkaProducer(t *testing.T) {\n\t// Create a mock sarama.SyncProducer\n\tmockProducer := mocks.NewSyncProducer(t, nil)\n\tlogger := testlogger.New(t)\n\ttopic := \"test-topic\"\n\n\tkafkaProducer := NewKafkaProducer(topic, mockProducer, logger)\n\n\t// Type assert to *producerImpl to access struct fields\n\tproducedImpl, ok := kafkaProducer.(*producerImpl)\n\tassert.True(t, ok, \"Expected kafkaProducer to be of type *producerImpl\")\n\n\t// Assert that all fields are correctly set\n\tassert.Equal(t, topic, producedImpl.topic)\n\tassert.Equal(t, mockProducer, producedImpl.producer)\n\tassert.NotNil(t, producedImpl.logger)\n\tassert.NotNil(t, producedImpl.msgEncoder)\n}\n\nfunc TestPublish(t *testing.T) {\n\tmsgType := indexer.MessageTypeIndex\n\ttestCases := []struct {\n\t\tname    string\n\t\tmessage interface{}\n\t\thasErr  bool\n\t}{\n\t\t{\n\t\t\tname: \"Publish message succeeded\",\n\t\t\tmessage: &indexer.Message{\n\t\t\t\tMessageType: &msgType,\n\t\t\t\tDomainID:    common.StringPtr(\"test-domain-id\"),\n\t\t\t\tWorkflowID:  common.StringPtr(\"test-workflow-id\"),\n\t\t\t\tRunID:       common.StringPtr(\"test-workflow-run-id\"),\n\t\t\t},\n\t\t\thasErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Unrecognized message type\",\n\t\t\tmessage: \"This is not a recognized message type\",\n\t\t\thasErr:  true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Create a mock sarama.SyncProducer\n\t\t\tmockProducer := mocks.NewSyncProducer(t, nil)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttopic := \"test-topic\"\n\t\t\tctx := context.Background()\n\t\t\tkafkaProducer := NewKafkaProducer(topic, mockProducer, logger)\n\n\t\t\t// Only expect SendMessage to be called if no error is expected\n\t\t\tif !tc.hasErr {\n\t\t\t\tmockProducer.ExpectSendMessageAndSucceed()\n\t\t\t}\n\t\t\terr := kafkaProducer.Publish(ctx, tc.message)\n\n\t\t\tif !tc.hasErr {\n\t\t\t\tassert.NoError(t, err, \"An error was not expected but got %v\", err)\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err, \"Expected an error but none was found\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/messaging/metrics_producer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage messaging\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype MetricsProducer struct {\n\tproducer Producer\n\tscope    metrics.Scope\n\ttags     []metrics.Tag\n}\n\ntype MetricProducerOptions func(*MetricsProducer)\n\nfunc WithMetricTags(tags ...metrics.Tag) MetricProducerOptions {\n\treturn func(p *MetricsProducer) {\n\t\tp.tags = tags\n\t}\n}\n\n// NewMetricProducer creates a new instance of producer that emits metrics\nfunc NewMetricProducer(\n\tproducer Producer,\n\tmetricsClient metrics.Client,\n\topts ...MetricProducerOptions,\n) Producer {\n\tp := &MetricsProducer{\n\t\tproducer: producer,\n\t}\n\n\tfor _, opt := range opts {\n\t\topt(p)\n\t}\n\n\tp.scope = metricsClient.Scope(metrics.MessagingClientPublishScope, p.tags...)\n\treturn p\n}\n\nfunc (p *MetricsProducer) Publish(ctx context.Context, msg interface{}) error {\n\tp.scope.IncCounter(metrics.CadenceClientRequests)\n\n\tsw := p.scope.StartTimer(metrics.CadenceClientLatency)\n\terr := p.producer.Publish(ctx, msg)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.scope.IncCounter(metrics.CadenceClientFailures)\n\t}\n\treturn err\n}\n\nfunc (p *MetricsProducer) Close() error {\n\tif closeableProducer, ok := p.producer.(CloseableProducer); ok {\n\t\treturn closeableProducer.Close()\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/messaging/metrics_producer_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage messaging\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n)\n\nfunc TestPublish(t *testing.T) {\n\ttests := []struct {\n\t\tdesc                string\n\t\ttags                []metrics.Tag\n\t\tproducerFails       bool\n\t\tmetricsClientMockFn func() *mocks.Client\n\t}{\n\t\t{\n\t\t\tdesc:          \"success\",\n\t\t\tproducerFails: false,\n\t\t\ttags: []metrics.Tag{\n\t\t\t\tmetrics.TopicTag(\"test-topic-1\"),\n\t\t\t},\n\t\t\tmetricsClientMockFn: func() *mocks.Client {\n\t\t\t\tmetricsClient := &mocks.Client{}\n\t\t\t\tmetricsScope := &mocks.Scope{}\n\t\t\t\tmetricsClient.\n\t\t\t\t\tOn(\"Scope\", metrics.MessagingClientPublishScope, metrics.TopicTag(\"test-topic-1\")).\n\t\t\t\t\tReturn(metricsScope).\n\t\t\t\t\tOnce()\n\t\t\t\tmetricsScope.On(\"IncCounter\", metrics.CadenceClientRequests).Once()\n\n\t\t\t\tsw := metrics.NoopScope.StartTimer(-1)\n\t\t\t\tmetricsScope.On(\"StartTimer\", metrics.CadenceClientLatency).Return(sw).Once()\n\t\t\t\treturn metricsClient\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"failure\",\n\t\t\tproducerFails: true,\n\t\t\ttags: []metrics.Tag{\n\t\t\t\tmetrics.TopicTag(\"test-topic-2\"),\n\t\t\t},\n\t\t\tmetricsClientMockFn: func() *mocks.Client {\n\t\t\t\tmetricsClient := &mocks.Client{}\n\t\t\t\tmetricsScope := &mocks.Scope{}\n\t\t\t\tmetricsClient.\n\t\t\t\t\tOn(\"Scope\", metrics.MessagingClientPublishScope, metrics.TopicTag(\"test-topic-2\")).\n\t\t\t\t\tReturn(metricsScope).\n\t\t\t\t\tOnce()\n\t\t\t\tmetricsScope.On(\"IncCounter\", metrics.CadenceClientRequests).Once()\n\t\t\t\tmetricsScope.On(\"IncCounter\", metrics.CadenceClientFailures).Once()\n\n\t\t\t\tsw := metrics.NoopScope.StartTimer(-1)\n\t\t\t\tmetricsScope.On(\"StartTimer\", metrics.CadenceClientLatency).Return(sw).Once()\n\t\t\t\treturn metricsClient\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\t// setup\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockProducer := NewMockProducer(ctrl)\n\t\t\tmsg := \"custom-message\"\n\t\t\tif tc.producerFails {\n\t\t\t\tmockProducer.EXPECT().Publish(gomock.Any(), msg).Return(errors.New(\"publish failed\")).Times(1)\n\t\t\t} else {\n\t\t\t\tmockProducer.EXPECT().Publish(gomock.Any(), msg).Return(nil).Times(1)\n\t\t\t}\n\t\t\tmetricsClient := tc.metricsClientMockFn()\n\n\t\t\t// create producer and call publish\n\t\t\tp := NewMetricProducer(mockProducer, metricsClient, WithMetricTags(tc.tags...))\n\t\t\terr := p.Publish(context.Background(), msg)\n\n\t\t\t// validations\n\t\t\tif tc.producerFails != (err != nil) {\n\t\t\t\tt.Errorf(\"expected producer to fail: %v, got: %v\", tc.producerFails, err)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tmetricsClient.AssertExpectations(t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/messaging/mocks/message_mock.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mocks\n\nimport mock \"github.com/stretchr/testify/mock\"\n\n// Message is an autogenerated mock type for the Message type\ntype Message struct {\n\tmock.Mock\n}\n\n// Ack provides a mock function with given fields:\nfunc (_m *Message) Ack() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Nack provides a mock function with given fields:\nfunc (_m *Message) Nack() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Offset provides a mock function with given fields:\nfunc (_m *Message) Offset() int64 {\n\tret := _m.Called()\n\n\tvar r0 int64\n\tif rf, ok := ret.Get(0).(func() int64); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(int64)\n\t}\n\n\treturn r0\n}\n\n// Partition provides a mock function with given fields:\nfunc (_m *Message) Partition() int32 {\n\tret := _m.Called()\n\n\tvar r0 int32\n\tif rf, ok := ret.Get(0).(func() int32); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(int32)\n\t}\n\n\treturn r0\n}\n\n// Value provides a mock function with given fields:\nfunc (_m *Message) Value() []byte {\n\tret := _m.Called()\n\n\tvar r0 []byte\n\tif rf, ok := ret.Get(0).(func() []byte); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).([]byte)\n\t\t}\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/messaging/noop_producer.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage messaging\n\nimport \"context\"\n\ntype noopProducer struct{}\n\n// NewNoopProducer returns a no-op message producer\nfunc NewNoopProducer() Producer {\n\treturn &noopProducer{}\n}\n\nfunc (p noopProducer) Publish(\n\t_ context.Context,\n\t_ interface{},\n) error {\n\treturn nil\n}\n"
  },
  {
    "path": "common/metrics/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\n// ClientImpl is used for reporting metrics by various Cadence services\ntype ClientImpl struct {\n\t// parentReporter is the parent scope for the metrics\n\tparentScope tally.Scope\n\tchildScopes map[ScopeIdx]tally.Scope\n\tmetricDefs  map[MetricIdx]metricDefinition\n\tserviceIdx  ServiceIdx\n\tconfig      HistogramMigration\n}\n\n// NewClient creates and returns a new instance of\n// Client implementation\n// reporter holds the common tags for the service\n// serviceIdx indicates the service type in (InputhostIndex, ... StorageIndex)\nfunc NewClient(scope tally.Scope, serviceIdx ServiceIdx, config HistogramMigration) Client {\n\ttotalScopes := len(ScopeDefs[Common]) + len(ScopeDefs[serviceIdx])\n\tmetricsClient := &ClientImpl{\n\t\tparentScope: scope,\n\t\tchildScopes: make(map[ScopeIdx]tally.Scope, totalScopes),\n\t\tmetricDefs:  getMetricDefs(serviceIdx),\n\t\tserviceIdx:  serviceIdx,\n\t\tconfig:      config,\n\t}\n\n\tfor idx, def := range ScopeDefs[Common] {\n\t\tscopeTags := map[string]string{\n\t\t\tOperationTagName: def.operation,\n\t\t}\n\t\tmergeMapToRight(def.tags, scopeTags)\n\t\tmetricsClient.childScopes[idx] = scope.Tagged(scopeTags)\n\t}\n\n\tfor idx, def := range ScopeDefs[serviceIdx] {\n\t\tscopeTags := map[string]string{\n\t\t\tOperationTagName: def.operation,\n\t\t}\n\t\tmergeMapToRight(def.tags, scopeTags)\n\t\tmetricsClient.childScopes[idx] = scope.Tagged(scopeTags)\n\t}\n\n\treturn metricsClient\n}\n\n// IncCounter increments one for a counter and emits\n// to metrics backend\nfunc (m *ClientImpl) IncCounter(scope ScopeIdx, counterIdx MetricIdx) {\n\tname := string(m.metricDefs[counterIdx].metricName)\n\tm.childScopes[scope].Counter(name).Inc(1)\n}\n\n// AddCounter adds delta to the counter and\n// emits to the metrics backend\nfunc (m *ClientImpl) AddCounter(scope ScopeIdx, counterIdx MetricIdx, delta int64) {\n\tname := string(m.metricDefs[counterIdx].metricName)\n\tm.childScopes[scope].Counter(name).Inc(delta)\n}\n\n// StartTimer starts a timer for the given\n// metric name\nfunc (m *ClientImpl) StartTimer(scope ScopeIdx, timerIdx MetricIdx) tally.Stopwatch {\n\tname := string(m.metricDefs[timerIdx].metricName)\n\tif m.config.EmitTimer(name) {\n\t\treturn m.childScopes[scope].Timer(name).Start()\n\t}\n\treturn NoopStopwatch\n}\n\n// RecordTimer record and emit a timer for the given\n// metric name\nfunc (m *ClientImpl) RecordTimer(scope ScopeIdx, timerIdx MetricIdx, d time.Duration) {\n\tname := string(m.metricDefs[timerIdx].metricName)\n\tif m.config.EmitTimer(name) {\n\t\tm.childScopes[scope].Timer(name).Record(d)\n\t}\n}\n\n// RecordHistogramDuration record and emit a duration\nfunc (m *ClientImpl) RecordHistogramDuration(scope ScopeIdx, timerIdx MetricIdx, d time.Duration) {\n\tname := string(m.metricDefs[timerIdx].metricName)\n\tif m.config.EmitHistogram(name) {\n\t\tm.childScopes[scope].Histogram(name, m.getBuckets(timerIdx)).RecordDuration(d)\n\t}\n}\n\n// UpdateGauge reports Gauge type metric\nfunc (m *ClientImpl) UpdateGauge(scopeIdx ScopeIdx, gaugeIdx MetricIdx, value float64) {\n\tname := string(m.metricDefs[gaugeIdx].metricName)\n\tm.childScopes[scopeIdx].Gauge(name).Update(value)\n}\n\n// Scope return a new internal metrics scope that can be used to add additional\n// information to the metrics emitted\nfunc (m *ClientImpl) Scope(scope ScopeIdx, tags ...Tag) Scope {\n\tsc := m.childScopes[scope]\n\treturn newMetricsScope(sc, sc, m.metricDefs, false, m.config).Tagged(tags...)\n}\n\nfunc (m *ClientImpl) getBuckets(id MetricIdx) tally.Buckets {\n\tif m.metricDefs[id].buckets != nil {\n\t\treturn m.metricDefs[id].buckets\n\t}\n\treturn tally.DefaultBuckets\n}\n\nfunc getMetricDefs(serviceIdx ServiceIdx) map[MetricIdx]metricDefinition {\n\tdefs := make(map[MetricIdx]metricDefinition)\n\tfor idx, def := range MetricDefs[Common] {\n\t\tdefs[idx] = def\n\t}\n\n\tfor idx, def := range MetricDefs[serviceIdx] {\n\t\tdefs[idx] = def\n\t}\n\n\treturn defs\n}\n\nfunc mergeMapToRight(src map[string]string, dest map[string]string) {\n\tfor k, v := range src {\n\t\tdest[k] = v\n\t}\n}\n"
  },
  {
    "path": "common/metrics/config.go",
    "content": "package metrics\n\nimport \"fmt\"\n\ntype HistogramMigration struct {\n\tDefault HistogramMigrationMode `yaml:\"default\"`\n\t// Names maps \"metric name\" -> \"should it be emitted\".\n\t//\n\t// If a name/key does not exist, the default mode will be checked to determine\n\t// if a timer or histogram should be emitted.\n\t//\n\t// This is only checked for timers and histograms that are in HistogramMigrationMetrics.\n\tNames map[string]bool `yaml:\"names\"`\n}\n\nfunc (h *HistogramMigration) UnmarshalYAML(read func(any) error) error {\n\ttype tmpType HistogramMigration // without the custom unmarshaler\n\tvar tmp tmpType\n\tif err := read(&tmp); err != nil {\n\t\treturn err\n\t}\n\tfor k := range tmp.Names {\n\t\tif _, ok := HistogramMigrationMetrics[k]; !ok {\n\t\t\treturn fmt.Errorf(\n\t\t\t\t\"unknown histogram-migration metric name %q.  \"+\n\t\t\t\t\t\"if this is a valid name, add it to common/metrics.HistogramMigrationMetrics before starting the service\",\n\t\t\t\tk,\n\t\t\t)\n\t\t}\n\t}\n\t*h = HistogramMigration(tmp)\n\treturn nil\n}\n\n// HistogramMigrationMetrics contains all metric names being migrated, to prevent affecting\n// non-migration-related timers and histograms, and to catch metric name config\n// mistakes early on.\n//\n// It is public to allow Cadence operators to add to the collection before\n// loading config, in case they have any custom migrations to perform.\n// This is likely best done in an `init` func, to ensure it happens early enough\n// and does not race with config reading.\nvar HistogramMigrationMetrics = map[string]struct{}{\n\t\"task_attempt\":                          {},\n\t\"task_attempt_counts\":                   {},\n\t\"task_attempt_per_domain\":               {},\n\t\"task_attempt_per_domain_counts\":        {},\n\t\"task_latency\":                          {},\n\t\"task_latency_ns\":                       {},\n\t\"task_latency_per_domain\":               {},\n\t\"task_latency_per_domain_ns\":            {},\n\t\"task_latency_processing\":               {},\n\t\"task_latency_processing_ns\":            {},\n\t\"task_latency_queue\":                    {},\n\t\"task_latency_queue_ns\":                 {},\n\t\"task_latency_processing_per_domain\":    {},\n\t\"task_latency_processing_per_domain_ns\": {},\n\t\"task_latency_queue_per_domain\":         {},\n\t\"task_latency_queue_per_domain_ns\":      {},\n\n\t\"replication_tasks_lag\":                {},\n\t\"replication_tasks_lag_counts\":         {},\n\t\"replication_tasks_applied_latency\":    {},\n\t\"replication_tasks_applied_latency_ns\": {},\n\n\t\"cache_latency\":     {},\n\t\"cache_latency_ns\":  {},\n\t\"cache_size\":        {},\n\t\"cache_size_counts\": {},\n\n\t\"replication_task_latency\":    {},\n\t\"replication_task_latency_ns\": {},\n\n\t\"replication_tasks_returned\":             {},\n\t\"replication_tasks_returned_counts\":      {},\n\t\"replication_tasks_returned_diff\":        {},\n\t\"replication_tasks_returned_diff_counts\": {},\n\n\t\"replication_tasks_fetched\":        {},\n\t\"replication_tasks_fetched_counts\": {},\n\t\"replication_tasks_lag_raw\":        {},\n\t\"replication_tasks_lag_raw_counts\": {},\n}\n\nfunc (h HistogramMigration) EmitTimer(name string) bool {\n\tif _, ok := HistogramMigrationMetrics[name]; !ok {\n\t\treturn true\n\t}\n\temit, ok := h.Names[name]\n\tif ok {\n\t\treturn emit\n\t}\n\treturn h.Default.EmitTimer()\n}\nfunc (h HistogramMigration) EmitHistogram(name string) bool {\n\tif _, ok := HistogramMigrationMetrics[name]; !ok {\n\t\treturn true\n\t}\n\n\temit, ok := h.Names[name]\n\tif ok {\n\t\treturn emit\n\t}\n\treturn h.Default.EmitHistogram()\n}\n\n// HistogramMigrationMode is a pseudo-enum to provide unmarshalling config and helper methods.\n// It should only be created by YAML unmarshaling, or by getting it from the HistogramMigration map.\n// Zero values from the map are valid, they are just the default mode (NOT the configured default).\n//\n// By default / when not specified / when an empty string, it currently means \"timer\".\n// This will likely change when most or all timers have histograms available, and will\n// eventually be fully deprecated and removed.\ntype HistogramMigrationMode string\n\nfunc (h *HistogramMigrationMode) UnmarshalYAML(read func(any) error) error {\n\tvar value string\n\tif err := read(&value); err != nil {\n\t\treturn fmt.Errorf(\"cannot read histogram migration mode as a string: %w\", err)\n\t}\n\tswitch value {\n\tcase \"timer\", \"histogram\", \"both\":\n\t\t*h = HistogramMigrationMode(value)\n\tdefault:\n\t\treturn fmt.Errorf(`unsupported histogram migration mode %q, must be \"timer\", \"histogram\", or \"both\"`, value)\n\t}\n\treturn nil\n}\n\nfunc (h HistogramMigrationMode) EmitTimer() bool {\n\tswitch h {\n\tcase \"timer\", \"both\", \"\": // default == not specified == both\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc (h HistogramMigrationMode) EmitHistogram() bool {\n\tswitch h {\n\tcase \"histogram\", \"both\": // default == not specified == both\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "common/metrics/config_test.go",
    "content": "package metrics\n\nimport (\n\t\"testing\"\n\n\t\"golang.org/x/exp/maps\"\n)\n\nfunc TestHistogramMigrationMetricsExist(t *testing.T) {\n\tdup := maps.Clone(HistogramMigrationMetrics)\n\tfor _, serviceMetrics := range MetricDefs {\n\t\tfor _, def := range serviceMetrics {\n\t\t\tdelete(dup, def.metricName.String())\n\t\t}\n\t}\n\tif len(dup) != 0 {\n\t\tt.Error(\"HistogramMigrationMetrics contains metric names which do not exist:\", dup)\n\t}\n}\n"
  },
  {
    "path": "common/metrics/context.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport \"context\"\n\ntype contextTag string\n\nconst contextTagsKey = contextTag(\"metrics.Tags\")\n\nfunc TagContext(ctx context.Context, ctxTags ...Tag) context.Context {\n\ttags, ok := ctx.Value(contextTagsKey).([]Tag)\n\tif !ok {\n\t\ttags = []Tag{}\n\t}\n\ttags = append(tags, ctxTags...)\n\treturn context.WithValue(ctx, contextTagsKey, tags)\n}\n\nfunc GetContextTags(ctx context.Context) []Tag {\n\ttags, ok := ctx.Value(contextTagsKey).([]Tag)\n\tif !ok {\n\t\ttags = []Tag{}\n\t}\n\treturn tags\n}\n"
  },
  {
    "path": "common/metrics/context_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestContextTags(t *testing.T) {\n\tctx := context.Background()\n\tassert.Empty(t, GetContextTags(ctx))\n\n\ttag1 := DomainTag(\"domain\")\n\tctx = TagContext(ctx, tag1)\n\tassert.Equal(t, []Tag{tag1}, GetContextTags(ctx))\n\n\ttag2 := TransportTag(\"grpc\")\n\tctx = TagContext(ctx, tag2)\n\tassert.Equal(t, []Tag{tag1, tag2}, GetContextTags(ctx))\n\n\tctx1 := context.Background()\n\tctx1 = TagContext(ctx1, tag1, tag2)\n\tassert.Contains(t, GetContextTags(ctx1), tag1)\n\tassert.Contains(t, GetContextTags(ctx1), tag2)\n}\n"
  },
  {
    "path": "common/metrics/defs.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\n// types used/defined by the package\ntype (\n\t// MetricName is the name of the metric\n\tMetricName string\n\n\t// MetricType is the type of the metric\n\tMetricType int\n\n\t// metricDefinition contains the definition for a metric\n\tmetricDefinition struct {\n\t\tmetricType            MetricType    // metric type\n\t\tmetricName            MetricName    // metric name\n\t\tmetricRollupName      MetricName    // optional. if non-empty, this name must be used for rolled-up version of this metric\n\t\tbuckets               tally.Buckets // buckets if we are emitting histograms\n\t\texponentialBuckets    histogrammy[SubsettableHistogram]\n\t\tintExponentialBuckets histogrammy[IntSubsettableHistogram]\n\t}\n\n\t// scopeDefinition holds the tag definitions for a scope\n\tscopeDefinition struct {\n\t\toperation string            // 'operation' tag for scope\n\t\ttags      map[string]string // additional tags for scope\n\t}\n\n\t// ServiceIdx is an index that uniquely identifies the service\n\tServiceIdx int\n\n\t// ScopeIdx is an index that uniquely identifies an operation, which is required to form a new metrics scope\n\tScopeIdx int\n\n\t// MetricIdx is an index that uniquely identifies the metric definition\n\tMetricIdx int\n)\n\nfunc (s scopeDefinition) GetOperationString() string {\n\treturn s.operation\n}\n\n// MetricTypes which are supported\nconst (\n\tCounter MetricType = iota\n\tTimer\n\tGauge\n\tHistogram\n)\n\n// Service names for all services that emit metrics.\nconst (\n\tCommon ServiceIdx = iota\n\tFrontend\n\tHistory\n\tMatching\n\tWorker\n\tShardDistributor\n\n\tNumServices\n)\n\n// This package should hold all the metrics and tags for cadence\n// Note that to better support Prometheus, metric name and tag name\n// should match the regex [a-zA-Z_][a-zA-Z0-9_]*, tag value can be any Unicode characters.\n// See more https://prometheus.io/docs/concepts/data_model/#metric-names-and-labels\n\n// Common tags for all services\nconst (\n\tOperationTagName      = \"operation\"\n\tCadenceRoleTagName    = \"cadence_role\"\n\tCadenceServiceTagName = \"cadence_service\"\n\tStatsTypeTagName      = \"stats_type\"\n\tCacheTypeTagName      = \"cache_type\"\n)\n\n// Common tag values\nconst (\n\tHistoryClientRoleTagValue  = \"history_client\"\n\tMatchingClientRoleTagValue = \"matching_client\"\n\tFrontendClientRoleTagValue = \"frontend_client\"\n\tAdminClientRoleTagValue    = \"admin_client\"\n\tDCRedirectionRoleTagValue  = \"dc_redirection\"\n\tBlobstoreRoleTagValue      = \"blobstore\"\n\n\tSizeStatsTypeTagValue  = \"size\"\n\tCountStatsTypeTagValue = \"count\"\n\n\tMutableStateCacheTypeTagValue = \"mutablestate\"\n\tEventsCacheTypeTagValue       = \"events\"\n)\n\n// Common service base metrics\nconst (\n\tRestartCount         = \"restarts\"\n\tNumGoRoutinesGauge   = \"num_goroutines\"\n\tGoMaxProcsGauge      = \"gomaxprocs\"\n\tMemoryAllocatedGauge = \"memory_allocated\"\n\tMemoryHeapGauge      = \"memory_heap\"\n\tMemoryHeapIdleGauge  = \"memory_heapidle\"\n\tMemoryHeapInuseGauge = \"memory_heapinuse\"\n\tMemoryStackGauge     = \"memory_stack\"\n\tNumGCCounter         = \"memory_num_gc\"\n\tGcPauseMsTimer       = \"memory_gc_pause_ms\"\n)\n\n// ServiceMetrics are types for common service base metrics\nvar ServiceMetrics = map[MetricName]MetricType{\n\tRestartCount: Counter,\n}\n\n// GoRuntimeMetrics represent the runtime stats from go runtime\nvar GoRuntimeMetrics = map[MetricName]MetricType{\n\tNumGoRoutinesGauge:   Gauge,\n\tGoMaxProcsGauge:      Gauge,\n\tMemoryAllocatedGauge: Gauge,\n\tMemoryHeapGauge:      Gauge,\n\tMemoryHeapIdleGauge:  Gauge,\n\tMemoryHeapInuseGauge: Gauge,\n\tMemoryStackGauge:     Gauge,\n\tNumGCCounter:         Counter,\n\tGcPauseMsTimer:       Timer,\n}\n\n// Scopes enum\nconst (\n\t// -- Common Operation scopes --\n\n\t// PersistenceCreateShardScope tracks CreateShard calls made by service to persistence layer\n\tPersistenceCreateShardScope ScopeIdx = iota\n\t// PersistenceGetShardScope tracks GetShard calls made by service to persistence layer\n\tPersistenceGetShardScope\n\t// PersistenceUpdateShardScope tracks UpdateShard calls made by service to persistence layer\n\tPersistenceUpdateShardScope\n\t// PersistenceCreateWorkflowExecutionScope tracks CreateWorkflowExecution calls made by service to persistence layer\n\tPersistenceCreateWorkflowExecutionScope\n\t// PersistenceGetWorkflowExecutionScope tracks GetWorkflowExecution calls made by service to persistence layer\n\tPersistenceGetWorkflowExecutionScope\n\t// PersistenceUpdateWorkflowExecutionScope tracks UpdateWorkflowExecution calls made by service to persistence layer\n\tPersistenceUpdateWorkflowExecutionScope\n\t// PersistenceConflictResolveWorkflowExecutionScope tracks ConflictResolveWorkflowExecution calls made by service to persistence layer\n\tPersistenceConflictResolveWorkflowExecutionScope\n\t// PersistenceResetWorkflowExecutionScope tracks ResetWorkflowExecution calls made by service to persistence layer\n\tPersistenceResetWorkflowExecutionScope\n\t// PersistenceDeleteWorkflowExecutionScope tracks DeleteWorkflowExecution calls made by service to persistence layer\n\tPersistenceDeleteWorkflowExecutionScope\n\t// PersistenceDeleteCurrentWorkflowExecutionScope tracks DeleteCurrentWorkflowExecution calls made by service to persistence layer\n\tPersistenceDeleteCurrentWorkflowExecutionScope\n\t// PersistenceGetCurrentExecutionScope tracks GetCurrentExecution calls made by service to persistence layer\n\tPersistenceGetCurrentExecutionScope\n\t// PersistenceIsWorkflowExecutionExistsScope tracks IsWorkflowExecutionExists calls made by service to persistence layer\n\tPersistenceIsWorkflowExecutionExistsScope\n\t// PersistenceListCurrentExecutionsScope tracks ListCurrentExecutions calls made by service to persistence layer\n\tPersistenceListCurrentExecutionsScope\n\t// PersistenceListConcreteExecutionsScope tracks ListConcreteExecutions calls made by service to persistence layer\n\tPersistenceListConcreteExecutionsScope\n\t// PersistenceGetTransferTasksScope tracks GetTransferTasks calls made by service to persistence layer\n\tPersistenceGetTransferTasksScope\n\t// PersistenceCompleteTransferTaskScope tracks CompleteTransferTasks calls made by service to persistence layer\n\tPersistenceCompleteTransferTaskScope\n\t// PersistenceGetCrossClusterTasksScope tracks GetCrossClusterTasks calls made by service to persistence layer\n\tPersistenceGetCrossClusterTasksScope\n\t// PersistenceCompleteCrossClusterTaskScope tracks CompleteCrossClusterTasks calls made by service to persistence layer\n\tPersistenceCompleteCrossClusterTaskScope\n\t// PersistenceGetReplicationTasksScope tracks GetReplicationTasks calls made by service to persistence layer\n\tPersistenceGetReplicationTasksScope\n\t// PersistenceCompleteReplicationTaskScope tracks CompleteReplicationTasks calls made by service to persistence layer\n\tPersistenceCompleteReplicationTaskScope\n\t// PersistencePutReplicationTaskToDLQScope tracks PersistencePutReplicationTaskToDLQScope calls made by service to persistence layer\n\tPersistencePutReplicationTaskToDLQScope\n\t// PersistenceGetReplicationTasksFromDLQScope tracks PersistenceGetReplicationTasksFromDLQScope calls made by service to persistence layer\n\tPersistenceGetReplicationTasksFromDLQScope\n\t// PersistenceGetReplicationDLQSizeScope tracks PersistenceGetReplicationDLQSizeScope calls made by service to persistence layer\n\tPersistenceGetReplicationDLQSizeScope\n\t// PersistenceDeleteReplicationTaskFromDLQScope tracks PersistenceDeleteReplicationTaskFromDLQScope calls made by service to persistence layer\n\tPersistenceDeleteReplicationTaskFromDLQScope\n\t// PersistenceRangeDeleteReplicationTaskFromDLQScope tracks PersistenceRangeDeleteReplicationTaskFromDLQScope calls made by service to persistence layer\n\tPersistenceRangeDeleteReplicationTaskFromDLQScope\n\t// PersistenceCreateFailoverMarkerTasksScope tracks CreateFailoverMarkerTasks calls made by service to persistence layer\n\tPersistenceCreateFailoverMarkerTasksScope\n\t// PersistenceGetTimerIndexTasksScope tracks GetTimerIndexTasks calls made by service to persistence layer\n\tPersistenceGetTimerIndexTasksScope\n\t// PersistenceCompleteTimerTaskScope tracks CompleteTimerTasks calls made by service to persistence layer\n\tPersistenceCompleteTimerTaskScope\n\t// PersistenceGetHistoryTasksScope tracks GetHistoryTasks calls made by service to persistence layer\n\tPersistenceGetHistoryTasksScope\n\t// PersistenceCompleteHistoryTaskScope tracks CompleteHistoryTask calls made by service to persistence layer\n\tPersistenceCompleteHistoryTaskScope\n\t// PersistenceRangeCompleteHistoryTaskScope tracks RangeCompleteHistoryTask calls made by service to persistence layer\n\tPersistenceRangeCompleteHistoryTaskScope\n\t// PersistenceCreateTasksScope tracks CreateTask calls made by service to persistence layer\n\tPersistenceCreateTasksScope\n\t// PersistenceGetTasksScope tracks GetTasks calls made by service to persistence layer\n\tPersistenceGetTasksScope\n\t// PersistenceCompleteTaskScope tracks CompleteTask calls made by service to persistence layer\n\tPersistenceCompleteTaskScope\n\t// PersistenceCompleteTasksLessThanScope is the metric scope for persistence.TaskManager.PersistenceCompleteTasksLessThan API\n\tPersistenceCompleteTasksLessThanScope\n\t// PersistenceGetOrphanTasksScope is the metric scope for persistence.TaskManager.GetOrphanTasks API\n\tPersistenceGetOrphanTasksScope\n\t// PersistenceLeaseTaskListScope tracks LeaseTaskList calls made by service to persistence layer\n\tPersistenceLeaseTaskListScope\n\t// PersistenceGetTaskListScope tracks GetTaskList calls made by service to persistence layer\n\tPersistenceGetTaskListScope\n\t// PersistenceUpdateTaskListScope tracks PersistenceUpdateTaskListScope calls made by service to persistence layer\n\tPersistenceUpdateTaskListScope\n\t// PersistenceListTaskListScope is the metric scope for persistence.TaskManager.ListTaskList API\n\tPersistenceListTaskListScope\n\t// PersistenceDeleteTaskListScope is the metric scope for persistence.TaskManager.DeleteTaskList API\n\tPersistenceDeleteTaskListScope\n\t// PersistenceGetTaskListSizeScope is the metric scope for persistence.TaskManager.GetTaskListSize API\n\tPersistenceGetTaskListSizeScope\n\t// PersistenceAppendHistoryEventsScope tracks AppendHistoryEvents calls made by service to persistence layer\n\tPersistenceAppendHistoryEventsScope\n\t// PersistenceGetWorkflowExecutionHistoryScope tracks GetWorkflowExecutionHistory calls made by service to persistence layer\n\tPersistenceGetWorkflowExecutionHistoryScope\n\t// PersistenceDeleteWorkflowExecutionHistoryScope tracks DeleteWorkflowExecutionHistory calls made by service to persistence layer\n\tPersistenceDeleteWorkflowExecutionHistoryScope\n\t// PersistenceCreateDomainScope tracks CreateDomain calls made by service to persistence layer\n\tPersistenceCreateDomainScope\n\t// PersistenceGetDomainScope tracks GetDomain calls made by service to persistence layer\n\tPersistenceGetDomainScope\n\t// PersistenceUpdateDomainScope tracks UpdateDomain calls made by service to persistence layer\n\tPersistenceUpdateDomainScope\n\t// PersistenceDeleteDomainScope tracks DeleteDomain calls made by service to persistence layer\n\tPersistenceDeleteDomainScope\n\t// PersistenceDeleteDomainByNameScope tracks DeleteDomainByName calls made by service to persistence layer\n\tPersistenceDeleteDomainByNameScope\n\t// PersistenceListDomainsScope tracks DeleteDomainByName calls made by service to persistence layer\n\tPersistenceListDomainsScope\n\t// PersistenceGetMetadataScope tracks DeleteDomainByName calls made by service to persistence layer\n\tPersistenceGetMetadataScope\n\t// PersistenceRecordWorkflowExecutionStartedScope tracks RecordWorkflowExecutionStarted calls made by service to persistence layer\n\tPersistenceRecordWorkflowExecutionStartedScope\n\t// PersistenceRecordWorkflowExecutionClosedScope tracks RecordWorkflowExecutionClosed calls made by service to persistence layer\n\tPersistenceRecordWorkflowExecutionClosedScope\n\t// PersistenceRecordWorkflowExecutionUninitializedScope tracks RecordWorkflowExecutionUninitialized calls made by service to persistence layer\n\tPersistenceRecordWorkflowExecutionUninitializedScope\n\t// PersistenceUpsertWorkflowExecutionScope tracks UpsertWorkflowExecution calls made by service to persistence layer\n\tPersistenceUpsertWorkflowExecutionScope\n\t// PersistenceListOpenWorkflowExecutionsScope tracks ListOpenWorkflowExecutions calls made by service to persistence layer\n\tPersistenceListOpenWorkflowExecutionsScope\n\t// PersistenceListClosedWorkflowExecutionsScope tracks ListClosedWorkflowExecutions calls made by service to persistence layer\n\tPersistenceListClosedWorkflowExecutionsScope\n\t// PersistenceListOpenWorkflowExecutionsByTypeScope tracks ListOpenWorkflowExecutionsByType calls made by service to persistence layer\n\tPersistenceListOpenWorkflowExecutionsByTypeScope\n\t// PersistenceListClosedWorkflowExecutionsByTypeScope tracks ListClosedWorkflowExecutionsByType calls made by service to persistence layer\n\tPersistenceListClosedWorkflowExecutionsByTypeScope\n\t// PersistenceListOpenWorkflowExecutionsByWorkflowIDScope tracks ListOpenWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tPersistenceListOpenWorkflowExecutionsByWorkflowIDScope\n\t// PersistenceListClosedWorkflowExecutionsByWorkflowIDScope tracks ListClosedWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tPersistenceListClosedWorkflowExecutionsByWorkflowIDScope\n\t// PersistenceListClosedWorkflowExecutionsByStatusScope tracks ListClosedWorkflowExecutionsByStatus calls made by service to persistence layer\n\tPersistenceListClosedWorkflowExecutionsByStatusScope\n\t// PersistenceGetClosedWorkflowExecutionScope tracks GetClosedWorkflowExecution calls made by service to persistence layer\n\tPersistenceGetClosedWorkflowExecutionScope\n\t// PersistenceVisibilityDeleteWorkflowExecutionScope is the metrics scope for persistence.VisibilityManager.DeleteWorkflowExecution\n\tPersistenceVisibilityDeleteWorkflowExecutionScope\n\t// PersistenceDeleteUninitializedWorkflowExecutionScope tracks DeleteUninitializedWorkflowExecution calls made by service to persistence layer\n\tPersistenceDeleteUninitializedWorkflowExecutionScope\n\t// PersistenceListWorkflowExecutionsScope tracks ListWorkflowExecutions calls made by service to persistence layer\n\tPersistenceListWorkflowExecutionsScope\n\t// PersistenceScanWorkflowExecutionsScope tracks ScanWorkflowExecutions calls made by service to persistence layer\n\tPersistenceScanWorkflowExecutionsScope\n\t// PersistenceCountWorkflowExecutionsScope tracks CountWorkflowExecutions calls made by service to persistence layer\n\tPersistenceCountWorkflowExecutionsScope\n\t// PersistenceEnqueueMessageScope tracks Enqueue calls made by service to persistence layer\n\tPersistenceEnqueueMessageScope\n\t// PersistenceEnqueueMessageToDLQScope tracks Enqueue DLQ calls made by service to persistence layer\n\tPersistenceEnqueueMessageToDLQScope\n\t// PersistenceReadMessagesScope tracks ReadMessages calls made by service to persistence layer\n\tPersistenceReadMessagesScope\n\t// PersistenceReadMessagesFromDLQScope tracks ReadMessagesFromDLQ calls made by service to persistence layer\n\tPersistenceReadMessagesFromDLQScope\n\t// PersistenceDeleteMessagesBeforeScope tracks DeleteMessages calls made by service to persistence layer\n\tPersistenceDeleteMessagesBeforeScope\n\t// PersistenceDeleteMessageFromDLQScope tracks DeleteMessageFromDLQ calls made by service to persistence layer\n\tPersistenceDeleteMessageFromDLQScope\n\t// PersistenceRangeDeleteMessagesFromDLQScope tracks RangeDeleteMessagesFromDLQ calls made by service to persistence layer\n\tPersistenceRangeDeleteMessagesFromDLQScope\n\t// PersistenceUpdateAckLevelScope tracks UpdateAckLevel calls made by service to persistence layer\n\tPersistenceUpdateAckLevelScope\n\t// PersistenceGetAckLevelsScope tracks GetAckLevel calls made by service to persistence layer\n\tPersistenceGetAckLevelsScope\n\t// PersistenceUpdateDLQAckLevelScope tracks UpdateDLQAckLevel calls made by service to persistence layer\n\tPersistenceUpdateDLQAckLevelScope\n\t// PersistenceGetDLQAckLevelsScope tracks GetDLQAckLevel calls made by service to persistence layer\n\tPersistenceGetDLQAckLevelsScope\n\t// PersistenceGetDLQSizeScope tracks GetDLQSize calls made by service to persistence layer\n\tPersistenceGetDLQSizeScope\n\t// PersistenceFetchDynamicConfigScope tracks FetchDynamicConfig calls made by service to persistence layer\n\tPersistenceFetchDynamicConfigScope\n\t// PersistenceUpdateDynamicConfigScope tracks UpdateDynamicConfig calls made by service to persistence layer\n\tPersistenceUpdateDynamicConfigScope\n\t// PersistenceShardRequestCountScope tracks number of persistence calls made to each shard\n\tPersistenceShardRequestCountScope\n\t// PersistenceGetActiveClusterSelectionPolicyScope tracks GetActiveClusterSelectionPolicy calls made by service to persistence layer\n\tPersistenceGetActiveClusterSelectionPolicyScope\n\t// PersistenceDeleteActiveClusterSelectionPolicyScope tracks DeleteActiveClusterSelectionPolicy calls made by service to persistence layer\n\tPersistenceDeleteActiveClusterSelectionPolicyScope\n\n\t// ResolverHostNotFoundScope is a simple low level error indicating a lookup failed in the membership resolver\n\tResolverHostNotFoundScope\n\t// HashringScope is a metrics scope for emitting events for the service hashrhing\n\tHashringScope\n\t// HistoryClientStartWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientStartWorkflowExecutionScope\n\t// HistoryClientDescribeHistoryHostScope tracks RPC calls to history service\n\tHistoryClientDescribeHistoryHostScope\n\t// HistoryClientRemoveTaskScope tracks RPC calls to history service\n\tHistoryClientRemoveTaskScope\n\t// HistoryClientCloseShardScope tracks RPC calls to history service\n\tHistoryClientCloseShardScope\n\t// HistoryClientResetQueueScope tracks RPC calls to history service\n\tHistoryClientResetQueueScope\n\t// HistoryClientDescribeQueueScope tracks RPC calls to history service\n\tHistoryClientDescribeQueueScope\n\t// HistoryClientRecordActivityTaskHeartbeatScope tracks RPC calls to history service\n\tHistoryClientRecordActivityTaskHeartbeatScope\n\t// HistoryClientRespondDecisionTaskCompletedScope tracks RPC calls to history service\n\tHistoryClientRespondDecisionTaskCompletedScope\n\t// HistoryClientRespondDecisionTaskFailedScope tracks RPC calls to history service\n\tHistoryClientRespondDecisionTaskFailedScope\n\t// HistoryClientRespondActivityTaskCompletedScope tracks RPC calls to history service\n\tHistoryClientRespondActivityTaskCompletedScope\n\t// HistoryClientRespondActivityTaskFailedScope tracks RPC calls to history service\n\tHistoryClientRespondActivityTaskFailedScope\n\t// HistoryClientRespondActivityTaskCanceledScope tracks RPC calls to history service\n\tHistoryClientRespondActivityTaskCanceledScope\n\t// HistoryClientDescribeMutableStateScope tracks RPC calls to history service\n\tHistoryClientDescribeMutableStateScope\n\t// HistoryClientGetMutableStateScope tracks RPC calls to history service\n\tHistoryClientGetMutableStateScope\n\t// HistoryClientPollMutableStateScope tracks RPC calls to history service\n\tHistoryClientPollMutableStateScope\n\t// HistoryClientResetStickyTaskListScope tracks RPC calls to history service\n\tHistoryClientResetStickyTaskListScope\n\t// HistoryClientDescribeWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientDescribeWorkflowExecutionScope\n\t// HistoryClientRecordDecisionTaskStartedScope tracks RPC calls to history service\n\tHistoryClientRecordDecisionTaskStartedScope\n\t// HistoryClientRecordActivityTaskStartedScope tracks RPC calls to history service\n\tHistoryClientRecordActivityTaskStartedScope\n\t// HistoryClientRequestCancelWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientRequestCancelWorkflowExecutionScope\n\t// HistoryClientSignalWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientSignalWorkflowExecutionScope\n\t// HistoryClientSignalWithStartWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientSignalWithStartWorkflowExecutionScope\n\t// HistoryClientRemoveSignalMutableStateScope tracks RPC calls to history service\n\tHistoryClientRemoveSignalMutableStateScope\n\t// HistoryClientTerminateWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientTerminateWorkflowExecutionScope\n\t// HistoryClientResetWorkflowExecutionScope tracks RPC calls to history service\n\tHistoryClientResetWorkflowExecutionScope\n\t// HistoryClientScheduleDecisionTaskScope tracks RPC calls to history service\n\tHistoryClientScheduleDecisionTaskScope\n\t// HistoryClientRecordChildExecutionCompletedScope tracks RPC calls to history service\n\tHistoryClientRecordChildExecutionCompletedScope\n\t// HistoryClientSyncShardStatusScope tracks RPC calls to history service\n\tHistoryClientReplicateEventsV2Scope\n\t// HistoryClientReplicateRawEventsV2Scope tracks RPC calls to history service\n\tHistoryClientSyncShardStatusScope\n\t// HistoryClientSyncActivityScope tracks RPC calls to history service\n\tHistoryClientSyncActivityScope\n\t// HistoryClientGetReplicationTasksScope tracks RPC calls to history service\n\tHistoryClientGetReplicationTasksScope\n\t// HistoryClientGetDLQReplicationTasksScope tracks RPC calls to history service\n\tHistoryClientGetDLQReplicationTasksScope\n\t// HistoryClientQueryWorkflowScope tracks RPC calls to history service\n\tHistoryClientQueryWorkflowScope\n\t// HistoryClientReapplyEventsScope tracks RPC calls to history service\n\tHistoryClientReapplyEventsScope\n\t// HistoryClientCountDLQMessagesScope tracks RPC calls to history service\n\tHistoryClientCountDLQMessagesScope\n\t// HistoryClientReadDLQMessagesScope tracks RPC calls to history service\n\tHistoryClientReadDLQMessagesScope\n\t// HistoryClientPurgeDLQMessagesScope tracks RPC calls to history service\n\tHistoryClientPurgeDLQMessagesScope\n\t// HistoryClientMergeDLQMessagesScope tracks RPC calls to history service\n\tHistoryClientMergeDLQMessagesScope\n\t// HistoryClientRefreshWorkflowTasksScope tracks RPC calls to history service\n\tHistoryClientRefreshWorkflowTasksScope\n\t// HistoryClientNotifyFailoverMarkersScope tracks RPC calls to history service\n\tHistoryClientNotifyFailoverMarkersScope\n\t// HistoryClientGetCrossClusterTasksScope tracks RPC calls to history service\n\tHistoryClientGetCrossClusterTasksScope\n\t// HistoryClientRespondCrossClusterTasksCompletedScope tracks RPC calls to history service\n\tHistoryClientRespondCrossClusterTasksCompletedScope\n\t// HistoryClientGetFailoverInfoScope tracks RPC calls to history service\n\tHistoryClientGetFailoverInfoScope\n\t// HistoryClientGetDLQReplicationMessagesScope tracks RPC calls to history service\n\tHistoryClientGetDLQReplicationMessagesScope\n\t// HistoryClientGetReplicationMessagesScope tracks RPC calls to history service\n\tHistoryClientGetReplicationMessagesScope\n\t// HistoryClientWfIDCacheScope tracks workflow ID cache metrics\n\tHistoryClientWfIDCacheScope\n\t// HistoryClientRatelimitUpdateScope tracks global ratelimiter related calls to history service\n\tHistoryClientRatelimitUpdateScope\n\n\t// MatchingClientPollForDecisionTaskScope tracks RPC calls to matching service\n\tMatchingClientPollForDecisionTaskScope\n\t// MatchingClientPollForActivityTaskScope tracks RPC calls to matching service\n\tMatchingClientPollForActivityTaskScope\n\t// MatchingClientAddActivityTaskScope tracks RPC calls to matching service\n\tMatchingClientAddActivityTaskScope\n\t// MatchingClientAddDecisionTaskScope tracks RPC calls to matching service\n\tMatchingClientAddDecisionTaskScope\n\t// MatchingClientQueryWorkflowScope tracks RPC calls to matching service\n\tMatchingClientQueryWorkflowScope\n\t// MatchingClientRespondQueryTaskCompletedScope tracks RPC calls to matching service\n\tMatchingClientRespondQueryTaskCompletedScope\n\t// MatchingClientCancelOutstandingPollScope tracks RPC calls to matching service\n\tMatchingClientCancelOutstandingPollScope\n\t// MatchingClientDescribeTaskListScope tracks RPC calls to matching service\n\tMatchingClientDescribeTaskListScope\n\t// MatchingClientListTaskListPartitionsScope tracks RPC calls to matching service\n\tMatchingClientListTaskListPartitionsScope\n\t// MatchingClientGetTaskListsByDomainScope tracks RPC calls to matching service\n\tMatchingClientGetTaskListsByDomainScope\n\t// MatchingClientUpdateTaskListPartitionConfigScope tracks RPC calls to matching service\n\tMatchingClientUpdateTaskListPartitionConfigScope\n\t// MatchingClientRefreshTaskListPartitionConfigScope tracks RPC calls to matching service\n\tMatchingClientRefreshTaskListPartitionConfigScope\n\n\t// FrontendClientDeleteDomainScope tracks RPC calls to frontend service\n\tFrontendClientDeleteDomainScope\n\t// FrontendClientDeprecateDomainScope tracks RPC calls to frontend service\n\tFrontendClientDeprecateDomainScope\n\t// FrontendClientDescribeDomainScope tracks RPC calls to frontend service\n\tFrontendClientDescribeDomainScope\n\t// FrontendClientDescribeTaskListScope tracks RPC calls to frontend service\n\tFrontendClientDescribeTaskListScope\n\t// FrontendClientDescribeWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientDescribeWorkflowExecutionScope\n\t// FrontendClientDiagnoseWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientDiagnoseWorkflowExecutionScope\n\t// FrontendClientGetWorkflowExecutionHistoryScope tracks RPC calls to frontend service\n\tFrontendClientGetWorkflowExecutionHistoryScope\n\t// FrontendClientGetWorkflowExecutionRawHistoryScope tracks RPC calls to frontend service\n\tFrontendClientGetWorkflowExecutionRawHistoryScope\n\t// FrontendClientPollForWorkflowExecutionRawHistoryScope tracks RPC calls to frontend service\n\tFrontendClientPollForWorkflowExecutionRawHistoryScope\n\t// FrontendClientListArchivedWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientListArchivedWorkflowExecutionsScope\n\t// FrontendClientListClosedWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientListClosedWorkflowExecutionsScope\n\t// FrontendClientListDomainsScope tracks RPC calls to frontend service\n\tFrontendClientListDomainsScope\n\t// FrontendClientListOpenWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientListOpenWorkflowExecutionsScope\n\t// FrontendClientPollForActivityTaskScope tracks RPC calls to frontend service\n\tFrontendClientPollForActivityTaskScope\n\t// FrontendClientPollForDecisionTaskScope tracks RPC calls to frontend service\n\tFrontendClientPollForDecisionTaskScope\n\t// FrontendClientQueryWorkflowScope tracks RPC calls to frontend service\n\tFrontendClientQueryWorkflowScope\n\t// FrontendClientRecordActivityTaskHeartbeatScope tracks RPC calls to frontend service\n\tFrontendClientRecordActivityTaskHeartbeatScope\n\t// FrontendClientRecordActivityTaskHeartbeatByIDScope tracks RPC calls to frontend service\n\tFrontendClientRecordActivityTaskHeartbeatByIDScope\n\t// FrontendClientRegisterDomainScope tracks RPC calls to frontend service\n\tFrontendClientRegisterDomainScope\n\t// FrontendClientRequestCancelWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientRequestCancelWorkflowExecutionScope\n\t// FrontendClientResetStickyTaskListScope tracks RPC calls to frontend service\n\tFrontendClientResetStickyTaskListScope\n\t// FrontendClientRefreshWorkflowTasksScope tracks RPC calls to frontend service\n\tFrontendClientRefreshWorkflowTasksScope\n\t// FrontendClientResetWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientResetWorkflowExecutionScope\n\t// FrontendClientRespondActivityTaskCanceledScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskCanceledScope\n\t// FrontendClientRespondActivityTaskCanceledByIDScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskCanceledByIDScope\n\t// FrontendClientRespondActivityTaskCompletedScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskCompletedScope\n\t// FrontendClientRespondActivityTaskCompletedByIDScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskCompletedByIDScope\n\t// FrontendClientRespondActivityTaskFailedScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskFailedScope\n\t// FrontendClientRespondActivityTaskFailedByIDScope tracks RPC calls to frontend service\n\tFrontendClientRespondActivityTaskFailedByIDScope\n\t// FrontendClientRespondDecisionTaskCompletedScope tracks RPC calls to frontend service\n\tFrontendClientRespondDecisionTaskCompletedScope\n\t// FrontendClientRespondDecisionTaskFailedScope tracks RPC calls to frontend service\n\tFrontendClientRespondDecisionTaskFailedScope\n\t// FrontendClientRespondQueryTaskCompletedScope tracks RPC calls to frontend service\n\tFrontendClientRespondQueryTaskCompletedScope\n\t// FrontendClientSignalWithStartWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientSignalWithStartWorkflowExecutionScope\n\t// FrontendClientSignalWorkflowExecutionAsyncScope tracks RPC calls to frontend service\n\tFrontendClientSignalWithStartWorkflowExecutionAsyncScope\n\t// FrontendClientSignalWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientSignalWorkflowExecutionScope\n\t// FrontendClientStartWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientStartWorkflowExecutionScope\n\t// FrontendClientStartWorkflowExecutionAsyncScope tracks RPC calls to frontend service\n\tFrontendClientStartWorkflowExecutionAsyncScope\n\t// FrontendClientRestartWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientRestartWorkflowExecutionScope\n\t// FrontendClientTerminateWorkflowExecutionScope tracks RPC calls to frontend service\n\tFrontendClientTerminateWorkflowExecutionScope\n\t// FrontendClientUpdateDomainScope tracks RPC calls to frontend service\n\tFrontendClientUpdateDomainScope\n\t// FrontendClientFailoverDomainScope tracks RPC calls to frontend service\n\tFrontendClientFailoverDomainScope\n\t// FrontendClientListFailoverHistoryScope tracks RPC calls to frontend service\n\tFrontendClientListFailoverHistoryScope\n\t// FrontendClientListWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientListWorkflowExecutionsScope\n\t// FrontendClientScanWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientScanWorkflowExecutionsScope\n\t// FrontendClientCountWorkflowExecutionsScope tracks RPC calls to frontend service\n\tFrontendClientCountWorkflowExecutionsScope\n\t// FrontendClientGetSearchAttributesScope tracks RPC calls to frontend service\n\tFrontendClientGetSearchAttributesScope\n\t// FrontendClientGetReplicationTasksScope tracks RPC calls to frontend service\n\tFrontendClientGetReplicationTasksScope\n\t// FrontendClientGetDomainReplicationTasksScope tracks RPC calls to frontend service\n\tFrontendClientGetDomainReplicationTasksScope\n\t// FrontendClientGetDLQReplicationTasksScope tracks RPC calls to frontend service\n\tFrontendClientGetDLQReplicationTasksScope\n\t// FrontendClientReapplyEventsScope tracks RPC calls to frontend service\n\tFrontendClientReapplyEventsScope\n\t// FrontendClientGetClusterInfoScope tracks RPC calls to frontend\n\tFrontendClientGetClusterInfoScope\n\t// FrontendClientListTaskListPartitionsScope tracks RPC calls to frontend service\n\tFrontendClientListTaskListPartitionsScope\n\t// FrontendClientGetTaskListsByDomainScope tracks RPC calls to frontend service\n\tFrontendClientGetTaskListsByDomainScope\n\n\t// AdminClientAddSearchAttributeScope tracks RPC calls to admin service\n\tAdminClientAddSearchAttributeScope\n\t// AdminClientCloseShardScope tracks RPC calls to admin service\n\tAdminClientCloseShardScope\n\t// AdminClientRemoveTaskScope tracks RPC calls to admin service\n\tAdminClientRemoveTaskScope\n\t// AdminClientResetQueueScope tracks RPC calls to admin service\n\tAdminClientResetQueueScope\n\t// AdminClientDescribeQueueScope tracks RPC calls to admin service\n\tAdminClientDescribeQueueScope\n\t// AdminClientDescribeHistoryHostScope tracks RPC calls to admin service\n\tAdminClientDescribeHistoryHostScope\n\t// AdminClientDescribeShardDistributionScope tracks RPC calls to admin service\n\tAdminClientDescribeShardDistributionScope\n\t// AdminClientDescribeWorkflowExecutionScope tracks RPC calls to admin service\n\tAdminClientDescribeWorkflowExecutionScope\n\t// AdminClientGetWorkflowExecutionRawHistoryV2Scope tracks RPC calls to admin service\n\tAdminClientGetWorkflowExecutionRawHistoryV2Scope\n\t// AdminClientDescribeClusterScope tracks RPC calls to admin service\n\tAdminClientDescribeClusterScope\n\t// AdminClientCountDLQMessagesScope tracks RPC calls to admin service\n\tAdminClientCountDLQMessagesScope\n\t// AdminClientReadDLQMessagesScope tracks RPC calls to admin service\n\tAdminClientReadDLQMessagesScope\n\t// AdminClientPurgeDLQMessagesScope tracks RPC calls to admin service\n\tAdminClientPurgeDLQMessagesScope\n\t// AdminClientMergeDLQMessagesScope tracks RPC calls to admin service\n\tAdminClientMergeDLQMessagesScope\n\t// AdminClientRefreshWorkflowTasksScope tracks RPC calls to admin service\n\tAdminClientRefreshWorkflowTasksScope\n\t// AdminClientResendReplicationTasksScope tracks RPC calls to admin service\n\tAdminClientResendReplicationTasksScope\n\t// AdminClientGetCrossClusterTasksScope tracks RPC calls to Admin service\n\tAdminClientGetCrossClusterTasksScope\n\t// AdminClientRespondCrossClusterTasksCompletedScope tracks RPC calls to Admin service\n\tAdminClientRespondCrossClusterTasksCompletedScope\n\t// AdminClientGetDynamicConfigScope tracks RPC calls to admin service\n\tAdminClientGetDynamicConfigScope\n\t// AdminClientUpdateDynamicConfigScope tracks RPC calls to admin service\n\tAdminClientUpdateDynamicConfigScope\n\t// AdminClientRestoreDynamicConfigScope tracks RPC calls to admin service\n\tAdminClientRestoreDynamicConfigScope\n\t// AdminClientListDynamicConfigScope tracks RPC calls to admin service\n\tAdminClientListDynamicConfigScope\n\t// AdminClientGetGlobalIsolationGroupsScope is a request to get all the global isolation-groups\n\tAdminClientGetGlobalIsolationGroupsScope\n\t// AdminClientUpdateGlobalIsolationGroupsScope is a request to update the global isolation-groups\n\tAdminClientUpdateGlobalIsolationGroupsScope\n\t// AdminClientGetDomainIsolationGroupsScope is a request to get the domains' isolation groups\n\tAdminClientGetDomainIsolationGroupsScope\n\t// AdminClientUpdateDomainIsolationGroupsScope is a request to update the domains isolation-groups\n\tAdminClientUpdateDomainIsolationGroupsScope\n\t// AdminClientDeleteWorkflowScope is the metric scope for admin.DeleteWorkflow\n\tAdminClientDeleteWorkflowScope\n\t// AdminClientMaintainCorruptWorkflowScope is the metric scope for admin.MaintainCorruptWorkflow\n\tAdminClientMaintainCorruptWorkflowScope\n\t// AdminClientGetReplicationTasksScope is the metric scope for admin.GetReplicationTasks\n\tAdminClientGetReplicationTasksScope\n\t// AdminClientReapplyEventsScope is the metric scope for admin.ReapplyEvents\n\tAdminClientReapplyEventsScope\n\t// AdminClientGetDLQReplicationMessagesScope is the metric scope for admin.GetDLQReplicationMessages\n\tAdminClientGetDLQReplicationMessagesScope\n\t// AdminClientGetDomainReplicationMessagesScope is the metric scope for admin.GetDomainReplicationMessages\n\tAdminClientGetDomainReplicationMessagesScope\n\t// AdminClientGetReplicationMessagesScope is the metric scope for admin.GetReplicationMessages\n\tAdminClientGetReplicationMessagesScope\n\t// AdminClientGetWorkflowExecutionRawHistoryScope is the metric scope for admin.GetDomainAsyncWorkflow\n\tAdminClientGetDomainAsyncWorkflowConfiguratonScope\n\t// AdminClientGetWorkflowExecutionRawHistoryScope is the metric scope for admin.UpdateDomainAsyncWorkflowConfiguration\n\tAdminClientUpdateDomainAsyncWorkflowConfiguratonScope\n\t// AdminClientUpdateTaskListPartitionConfigScope is the metrics scope for admin.UpdateTaskListPartitionConfig\n\tAdminClientUpdateTaskListPartitionConfigScope\n\n\t// DCRedirectionDeleteDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionDeleteDomainScope\n\t// DCRedirectionDeleteDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionDeprecateDomainScope\n\t// DCRedirectionFailoverDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionFailoverDomainScope\n\t// DCRedirectionListFailoverHistoryScope tracks RPC calls for dc redirection\n\tDCRedirectionListFailoverHistoryScope\n\t// DCRedirectionDescribeDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionDescribeDomainScope\n\t// DCRedirectionDescribeTaskListScope tracks RPC calls for dc redirection\n\tDCRedirectionDescribeTaskListScope\n\t// DCRedirectionDescribeWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionDescribeWorkflowExecutionScope\n\t// DCRedirectionDiagnoseWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionDiagnoseWorkflowExecutionScope\n\t// DCRedirectionGetWorkflowExecutionHistoryScope tracks RPC calls for dc redirection\n\tDCRedirectionGetWorkflowExecutionHistoryScope\n\t// DCRedirectionGetWorkflowExecutionRawHistoryScope tracks RPC calls for dc redirection\n\tDCRedirectionGetWorkflowExecutionRawHistoryScope\n\t// DCRedirectionPollForWorklfowExecutionRawHistoryScope tracks RPC calls for dc redirection\n\tDCRedirectionPollForWorklfowExecutionRawHistoryScope\n\t// DCRedirectionListArchivedWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionListArchivedWorkflowExecutionsScope\n\t// DCRedirectionListClosedWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionListClosedWorkflowExecutionsScope\n\t// DCRedirectionListDomainsScope tracks RPC calls for dc redirection\n\tDCRedirectionListDomainsScope\n\t// DCRedirectionListOpenWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionListOpenWorkflowExecutionsScope\n\t// DCRedirectionListWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionListWorkflowExecutionsScope\n\t// DCRedirectionScanWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionScanWorkflowExecutionsScope\n\t// DCRedirectionCountWorkflowExecutionsScope tracks RPC calls for dc redirection\n\tDCRedirectionCountWorkflowExecutionsScope\n\t// DCRedirectionGetSearchAttributesScope tracks RPC calls for dc redirection\n\tDCRedirectionGetSearchAttributesScope\n\t// DCRedirectionPollForActivityTaskScope tracks RPC calls for dc redirection\n\tDCRedirectionPollForActivityTaskScope\n\t// DCRedirectionPollForDecisionTaskScope tracks RPC calls for dc redirection\n\tDCRedirectionPollForDecisionTaskScope\n\t// DCRedirectionQueryWorkflowScope tracks RPC calls for dc redirection\n\tDCRedirectionQueryWorkflowScope\n\t// DCRedirectionRecordActivityTaskHeartbeatScope tracks RPC calls for dc redirection\n\tDCRedirectionRecordActivityTaskHeartbeatScope\n\t// DCRedirectionRecordActivityTaskHeartbeatByIDScope tracks RPC calls for dc redirection\n\tDCRedirectionRecordActivityTaskHeartbeatByIDScope\n\t// DCRedirectionRegisterDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionRegisterDomainScope\n\t// DCRedirectionRequestCancelWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionRequestCancelWorkflowExecutionScope\n\t// DCRedirectionResetStickyTaskListScope tracks RPC calls for dc redirection\n\tDCRedirectionResetStickyTaskListScope\n\t// DCRedirectionResetWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionResetWorkflowExecutionScope\n\t// DCRedirectionRespondActivityTaskCanceledScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskCanceledScope\n\t// DCRedirectionRespondActivityTaskCanceledByIDScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskCanceledByIDScope\n\t// DCRedirectionRespondActivityTaskCompletedScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskCompletedScope\n\t// DCRedirectionRespondActivityTaskCompletedByIDScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskCompletedByIDScope\n\t// DCRedirectionRespondActivityTaskFailedScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskFailedScope\n\t// DCRedirectionRespondActivityTaskFailedByIDScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondActivityTaskFailedByIDScope\n\t// DCRedirectionRespondDecisionTaskCompletedScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondDecisionTaskCompletedScope\n\t// DCRedirectionRespondDecisionTaskFailedScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondDecisionTaskFailedScope\n\t// DCRedirectionRespondQueryTaskCompletedScope tracks RPC calls for dc redirection\n\tDCRedirectionRespondQueryTaskCompletedScope\n\t// DCRedirectionSignalWithStartWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionSignalWithStartWorkflowExecutionScope\n\t// DCRedirectionSignalWithStartWorkflowExecutionAsyncScope tracks RPC calls for dc redirection\n\tDCRedirectionSignalWithStartWorkflowExecutionAsyncScope\n\t// DCRedirectionSignalWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionSignalWorkflowExecutionScope\n\t// DCRedirectionStartWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionStartWorkflowExecutionScope\n\t// DCRedirectionStartWorkflowExecutionAsyncScope tracks RPC calls for dc redirection\n\tDCRedirectionStartWorkflowExecutionAsyncScope\n\t// DCRedirectionTerminateWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionTerminateWorkflowExecutionScope\n\t// DCRedirectionUpdateDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionUpdateDomainScope\n\t// DCRedirectionListTaskListPartitionsScope tracks RPC calls for dc redirection\n\tDCRedirectionListTaskListPartitionsScope\n\t// DCRedirectionGetTaskListsByDomainScope tracks RPC calls for dc redirection\n\tDCRedirectionGetTaskListsByDomainScope\n\t// DCRedirectionRefreshWorkflowTasksScope tracks RPC calls for dc redirection\n\tDCRedirectionRefreshWorkflowTasksScope\n\t// DCRedirectionRestartWorkflowExecutionScope tracks RPC calls for dc redirection\n\tDCRedirectionRestartWorkflowExecutionScope\n\t// DCRedirectionForwardingPolicyScope tracks cluster redirection decisions\n\tDCRedirectionForwardingPolicyScope\n\n\t// MessagingPublishScope tracks Publish calls made by service to messaging layer\n\tMessagingClientPublishScope\n\t// MessagingPublishBatchScope tracks Publish calls made by service to messaging layer\n\tMessagingClientPublishBatchScope\n\t// MessagingClientConsumerScope tracks the consumer activities\n\tMessagingClientConsumerScope\n\n\t// DomainCacheScope tracks domain cache callbacks\n\tDomainCacheScope\n\t// HistoryRereplicationByTransferTaskScope tracks history replication calls made by transfer task\n\tHistoryRereplicationByTransferTaskScope\n\t// HistoryRereplicationByTimerTaskScope tracks history replication calls made by timer task\n\tHistoryRereplicationByTimerTaskScope\n\t// HistoryRereplicationByHistoryReplicationScope tracks history replication calls made by history replication\n\tHistoryRereplicationByHistoryReplicationScope\n\t// HistoryRereplicationByHistoryMetadataReplicationScope tracks history replication calls made by history replication\n\tHistoryRereplicationByHistoryMetadataReplicationScope\n\t// HistoryRereplicationByActivityReplicationScope tracks history replication calls made by activity replication\n\tHistoryRereplicationByActivityReplicationScope\n\n\t// PersistenceAppendHistoryNodesScope tracks AppendHistoryNodes calls made by service to persistence layer\n\tPersistenceAppendHistoryNodesScope\n\t// PersistenceReadHistoryBranchScope tracks ReadHistoryBranch calls made by service to persistence layer\n\tPersistenceReadHistoryBranchScope\n\t// PersistenceReadHistoryBranchByBatchScope tracks ReadHistoryBranch calls made by service to persistence layer\n\tPersistenceReadHistoryBranchByBatchScope\n\t// PersistenceReadRawHistoryBranchScope tracks ReadHistoryBranch calls made by service to persistence layer\n\tPersistenceReadRawHistoryBranchScope\n\t// PersistenceForkHistoryBranchScope tracks ForkHistoryBranch calls made by service to persistence layer\n\tPersistenceForkHistoryBranchScope\n\t// PersistenceDeleteHistoryBranchScope tracks DeleteHistoryBranch calls made by service to persistence layer\n\tPersistenceDeleteHistoryBranchScope\n\t// PersistenceCompleteForkBranchScope tracks CompleteForkBranch calls made by service to persistence layer\n\tPersistenceCompleteForkBranchScope\n\t// PersistenceGetHistoryTreeScope tracks GetHistoryTree calls made by service to persistence layer\n\tPersistenceGetHistoryTreeScope\n\t// PersistenceGetAllHistoryTreeBranchesScope tracks GetHistoryTree calls made by service to persistence layer\n\tPersistenceGetAllHistoryTreeBranchesScope\n\n\t// ClusterMetadataArchivalConfigScope tracks ArchivalConfig calls to ClusterMetadata\n\tClusterMetadataArchivalConfigScope\n\n\t// ElasticsearchRecordWorkflowExecutionStartedScope tracks RecordWorkflowExecutionStarted calls made by service to persistence layer\n\tElasticsearchRecordWorkflowExecutionStartedScope\n\t// ElasticsearchRecordWorkflowExecutionClosedScope tracks RecordWorkflowExecutionClosed calls made by service to persistence layer\n\tElasticsearchRecordWorkflowExecutionClosedScope\n\t// ElasticsearchRecordWorkflowExecutionUninitializedScope tracks RecordWorkflowExecutionUninitialized calls made by service to persistence layer\n\tElasticsearchRecordWorkflowExecutionUninitializedScope\n\t// ElasticsearchUpsertWorkflowExecutionScope tracks UpsertWorkflowExecution calls made by service to persistence layer\n\tElasticsearchUpsertWorkflowExecutionScope\n\t// ElasticsearchListOpenWorkflowExecutionsScope tracks ListOpenWorkflowExecutions calls made by service to persistence layer\n\tElasticsearchListOpenWorkflowExecutionsScope\n\t// ElasticsearchListClosedWorkflowExecutionsScope tracks ListClosedWorkflowExecutions calls made by service to persistence layer\n\tElasticsearchListClosedWorkflowExecutionsScope\n\t// ElasticsearchListOpenWorkflowExecutionsByTypeScope tracks ListOpenWorkflowExecutionsByType calls made by service to persistence layer\n\tElasticsearchListOpenWorkflowExecutionsByTypeScope\n\t// ElasticsearchListClosedWorkflowExecutionsByTypeScope tracks ListClosedWorkflowExecutionsByType calls made by service to persistence layer\n\tElasticsearchListClosedWorkflowExecutionsByTypeScope\n\t// ElasticsearchListOpenWorkflowExecutionsByWorkflowIDScope tracks ListOpenWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tElasticsearchListOpenWorkflowExecutionsByWorkflowIDScope\n\t// ElasticsearchListClosedWorkflowExecutionsByWorkflowIDScope tracks ListClosedWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tElasticsearchListClosedWorkflowExecutionsByWorkflowIDScope\n\t// ElasticsearchListClosedWorkflowExecutionsByStatusScope tracks ListClosedWorkflowExecutionsByStatus calls made by service to persistence layer\n\tElasticsearchListClosedWorkflowExecutionsByStatusScope\n\t// ElasticsearchGetClosedWorkflowExecutionScope tracks GetClosedWorkflowExecution calls made by service to persistence layer\n\tElasticsearchGetClosedWorkflowExecutionScope\n\t// ElasticsearchListWorkflowExecutionsScope tracks ListWorkflowExecutions calls made by service to persistence layer\n\tElasticsearchListWorkflowExecutionsScope\n\t// ElasticsearchScanWorkflowExecutionsScope tracks ScanWorkflowExecutions calls made by service to persistence layer\n\tElasticsearchScanWorkflowExecutionsScope\n\t// ElasticsearchCountWorkflowExecutionsScope tracks CountWorkflowExecutions calls made by service to persistence layer\n\tElasticsearchCountWorkflowExecutionsScope\n\t// ElasticsearchDeleteWorkflowExecutionsScope tracks DeleteWorkflowExecution calls made by service to persistence layer\n\tElasticsearchDeleteWorkflowExecutionsScope\n\t// ElasticsearchDeleteUninitializedWorkflowExecutionsScope tracks DeleteUninitializedWorkflowExecution calls made by service to persistence layer\n\tElasticsearchDeleteUninitializedWorkflowExecutionsScope\n\n\t// PinotRecordWorkflowExecutionStartedScope tracks RecordWorkflowExecutionStarted calls made by service to persistence layer\n\tPinotRecordWorkflowExecutionStartedScope\n\t// PinotRecordWorkflowExecutionClosedScope tracks RecordWorkflowExecutionClosed calls made by service to persistence layer\n\tPinotRecordWorkflowExecutionClosedScope\n\t// PinotRecordWorkflowExecutionUninitializedScope tracks RecordWorkflowExecutionUninitialized calls made by service to persistence layer\n\tPinotRecordWorkflowExecutionUninitializedScope\n\t// PinotUpsertWorkflowExecutionScope tracks UpsertWorkflowExecution calls made by service to persistence layer\n\tPinotUpsertWorkflowExecutionScope\n\t// PinotListOpenWorkflowExecutionsScope tracks ListOpenWorkflowExecutions calls made by service to persistence layer\n\tPinotListOpenWorkflowExecutionsScope\n\t// PinotListClosedWorkflowExecutionsScope tracks ListClosedWorkflowExecutions calls made by service to persistence layer\n\tPinotListClosedWorkflowExecutionsScope\n\t// PinotListOpenWorkflowExecutionsByTypeScope tracks ListOpenWorkflowExecutionsByType calls made by service to persistence layer\n\tPinotListOpenWorkflowExecutionsByTypeScope\n\t// PinotListClosedWorkflowExecutionsByTypeScope tracks ListClosedWorkflowExecutionsByType calls made by service to persistence layer\n\tPinotListClosedWorkflowExecutionsByTypeScope\n\t// PinotListOpenWorkflowExecutionsByWorkflowIDScope tracks ListOpenWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tPinotListOpenWorkflowExecutionsByWorkflowIDScope\n\t// PinotListClosedWorkflowExecutionsByWorkflowIDScope tracks ListClosedWorkflowExecutionsByWorkflowID calls made by service to persistence layer\n\tPinotListClosedWorkflowExecutionsByWorkflowIDScope\n\t// PinotListClosedWorkflowExecutionsByStatusScope tracks ListClosedWorkflowExecutionsByStatus calls made by service to persistence layer\n\tPinotListClosedWorkflowExecutionsByStatusScope\n\t// PinotGetClosedWorkflowExecutionScope tracks GetClosedWorkflowExecution calls made by service to persistence layer\n\tPinotGetClosedWorkflowExecutionScope\n\t// PinotListWorkflowExecutionsScope tracks ListWorkflowExecutions calls made by service to persistence layer\n\tPinotListWorkflowExecutionsScope\n\t// PinotScanWorkflowExecutionsScope tracks ScanWorkflowExecutions calls made by service to persistence layer\n\tPinotScanWorkflowExecutionsScope\n\t// PinotCountWorkflowExecutionsScope tracks CountWorkflowExecutions calls made by service to persistence layer\n\tPinotCountWorkflowExecutionsScope\n\t// PinotDeleteWorkflowExecutionsScope tracks DeleteWorkflowExecution calls made by service to persistence layer\n\tPinotDeleteWorkflowExecutionsScope\n\t// PinotDeleteUninitializedWorkflowExecutionsScope tracks DeleteUninitializedWorkflowExecution calls made by service to persistence layer\n\tPinotDeleteUninitializedWorkflowExecutionsScope\n\n\t// SequentialTaskProcessingScope is used by sequential task processing logic\n\tSequentialTaskProcessingScope\n\t// ParallelTaskProcessingScope is used by parallel task processing logic\n\tParallelTaskProcessingScope\n\t// TaskSchedulerScope is used by task scheduler logic\n\tTaskSchedulerScope\n\t// TaskSchedulerRateLimiterScope is used by task scheduler rate limiter logic\n\tTaskSchedulerRateLimiterScope\n\n\t// HistoryEngineScope is used by history engine for areas that aren't covered by other, more specific scopes\n\tHistoryEngineScope\n\t// HistoryArchiverScope is used by history archivers\n\tHistoryArchiverScope\n\t// VisibilityArchiverScope is used by visibility archivers\n\tVisibilityArchiverScope\n\n\t// The following metrics are only used by internal archiver implemention.\n\t// TODO: move them to internal repo once cadence plugin model is in place.\n\n\t// BlobstoreClientUploadScope tracks Upload calls to blobstore\n\tBlobstoreClientUploadScope\n\t// BlobstoreClientDownloadScope tracks Download calls to blobstore\n\tBlobstoreClientDownloadScope\n\t// BlobstoreClientGetMetadataScope tracks GetMetadata calls to blobstore\n\tBlobstoreClientGetMetadataScope\n\t// BlobstoreClientExistsScope tracks Exists calls to blobstore\n\tBlobstoreClientExistsScope\n\t// BlobstoreClientDeleteScope tracks Delete calls to blobstore\n\tBlobstoreClientDeleteScope\n\t// BlobstoreClientDirectoryExistsScope tracks DirectoryExists calls to blobstore\n\tBlobstoreClientDirectoryExistsScope\n\n\t// DomainFailoverScope is used in domain failover processor\n\tDomainFailoverScope\n\t// DomainReplicationQueueScope is used in domainreplication queue\n\tDomainReplicationQueueScope\n\t// ClusterMetadataScope is used for the cluster metadata\n\tClusterMetadataScope\n\t// GetAvailableIsolationGroupsScope is the metric for the default partitioner's getIsolationGroups operation\n\tGetAvailableIsolationGroupsScope\n\t// TaskValidatorScope is the metric for the taskvalidator's workflow check operation.\n\tTaskValidatorScope\n\n\t// GlobalRatelimiter is the metrics scope for limiting-side common/quotas/global behavior\n\tGlobalRatelimiter\n\t// GlobalRatelimiterAggregator is the metrics scope for aggregator-side common/quotas/global behavior\n\tGlobalRatelimiterAggregator\n\n\t// P2PRPCPeerChooserScope is the metrics scope for P2P RPC peer chooser\n\tP2PRPCPeerChooserScope\n\n\t// PartitionConfigProviderScope is the metrics scope for Partition Config Provider\n\tPartitionConfigProviderScope\n\n\t// ShardDistributorClientGetShardOwnerScope tracks GetShardOwner calls made by service to shard distributor\n\tShardDistributorClientGetShardOwnerScope\n\n\t// ShardDistributorClientWatchNamespaceStateScope tracks WatchNamespaceState calls made by service to shard distributor\n\tShardDistributorClientWatchNamespaceStateScope\n\n\t// ShardDistributorExecutorClientHeartbeatScope tracks Heartbeat calls made by executor to shard distributor\n\tShardDistributorExecutorClientHeartbeatScope\n\n\t// LoadBalancerScope is the metrics scope for Round Robin load balancer\n\tLoadBalancerScope\n\n\t// ActiveClusterManager is the scope used by active cluster manager\n\tActiveClusterManager\n\n\t// ActiveClusterManagerWorkflowCacheScope is the scope used by active cluster manager's workflow cache\n\tActiveClusterManagerWorkflowCacheScope\n\n\tNumCommonScopes\n)\n\n// -- Operation scopes for Admin service --\nconst (\n\t// AdminDescribeHistoryHostScope is the metric scope for admin.AdminDescribeHistoryHostScope\n\tAdminDescribeHistoryHostScope = iota + NumCommonScopes\n\t// AdminDescribeClusterScope is the metric scope for admin.AdminDescribeClusterScope\n\tAdminDescribeClusterScope\n\t// AdminAddSearchAttributeScope is the metric scope for admin.AdminAddSearchAttributeScope\n\tAdminAddSearchAttributeScope\n\t// AdminDescribeWorkflowExecutionScope is the metric scope for admin.AdminDescribeWorkflowExecutionScope\n\tAdminDescribeWorkflowExecutionScope\n\t// AdminGetWorkflowExecutionRawHistoryScope is the metric scope for admin.GetWorkflowExecutionRawHistoryScope\n\tAdminGetWorkflowExecutionRawHistoryScope\n\t// AdminGetWorkflowExecutionRawHistoryV2Scope is the metric scope for admin.GetWorkflowExecutionRawHistoryScope\n\tAdminGetWorkflowExecutionRawHistoryV2Scope\n\t// AdminGetReplicationMessagesScope is the metric scope for admin.GetReplicationMessages\n\tAdminGetReplicationMessagesScope\n\t// AdminGetDomainReplicationMessagesScope is the metric scope for admin.GetDomainReplicationMessages\n\tAdminGetDomainReplicationMessagesScope\n\t// AdminGetDLQReplicationMessagesScope is the metric scope for admin.GetDLQReplicationMessages\n\tAdminGetDLQReplicationMessagesScope\n\t// AdminReapplyEventsScope is the metric scope for admin.ReapplyEvents\n\tAdminReapplyEventsScope\n\t// AdminRefreshWorkflowTasksScope is the metric scope for admin.RefreshWorkflowTasks\n\tAdminRefreshWorkflowTasksScope\n\t// AdminResendReplicationTasksScope is the metric scope for admin.ResendReplicationTasks\n\tAdminResendReplicationTasksScope\n\t// AdminRemoveTaskScope is the metric scope for admin.AdminRemoveTaskScope\n\tAdminRemoveTaskScope\n\t// AdminCloseShardScope is the metric scope for admin.AdminCloseShardScope\n\tAdminCloseShardScope\n\t// AdminResetQueueScope is the metric scope for admin.AdminResetQueueScope\n\tAdminResetQueueScope\n\t// AdminDescribeQueueScope is the metrics scope for admin.AdminDescribeQueueScope\n\tAdminDescribeQueueScope\n\t// AdminCountDLQMessagesScope is the metric scope for admin.AdminCountDLQMessagesScope\n\tAdminCountDLQMessagesScope\n\t// AdminReadDLQMessagesScope is the metric scope for admin.AdminReadDLQMessagesScope\n\tAdminReadDLQMessagesScope\n\t// AdminPurgeDLQMessagesScope is the metric scope for admin.AdminPurgeDLQMessagesScope\n\tAdminPurgeDLQMessagesScope\n\t// AdminMergeDLQMessagesScope is the metric scope for admin.AdminMergeDLQMessagesScope\n\tAdminMergeDLQMessagesScope\n\t// AdminDescribeShardDistributionScope is the metric scope for admin.DescribeShardDistribution\n\tAdminDescribeShardDistributionScope\n\t// AdminGetCrossClusterTasksScope is the metric scope for admin.GetCrossClusterTasks\n\tAdminGetCrossClusterTasksScope\n\t// AdminRespondCrossClusterTasksCompletedScope is the metric scope for admin.AdminRespondCrossClusterTasksCompleted\n\tAdminRespondCrossClusterTasksCompletedScope\n\t// AdminGetDynamicConfigScope is the metric scope for admin.GetDynamicConfig\n\tAdminGetDynamicConfigScope\n\t// AdminUpdateDynamicConfigScope is the metric scope for admin.UpdateDynamicConfig\n\tAdminUpdateDynamicConfigScope\n\t// AdminRestoreDynamicConfigScope is the metric scope for admin.RestoreDynamicConfig\n\tAdminRestoreDynamicConfigScope\n\t// AdminListDynamicConfigScope is the metric scope for admin.ListDynamicConfig\n\tAdminListDynamicConfigScope\n\t// AdminDeleteWorkflowScope is the metric scope for admin.DeleteWorkflow\n\tAdminDeleteWorkflowScope\n\t// GetGlobalIsolationGroups is the scope for getting global isolation groups\n\tGetGlobalIsolationGroups\n\t// UpdateGlobalIsolationGroups is the scope for getting global isolation groups\n\tUpdateGlobalIsolationGroups\n\t// GetDomainIsolationGroups is the scope for getting domain isolation groups\n\tGetDomainIsolationGroups\n\t// UpdateDomainIsolationGroups is the scope for getting domain isolation groups\n\tUpdateDomainIsolationGroups\n\t// GetDomainAsyncWorkflowConfiguraton is the scope for getting domain async workflow configuration\n\tGetDomainAsyncWorkflowConfiguraton\n\t// UpdateDomainAsyncWorkflowConfiguraton is the scope for updating domain async workflow configuration\n\tUpdateDomainAsyncWorkflowConfiguraton\n\t// UpdateTaskListPartitionConfig is the scope for update task list partition config\n\tUpdateTaskListPartitionConfig\n\n\tNumAdminScopes\n)\n\n// -- Operation scopes for Frontend service --\nconst (\n\t// FrontendRestartWorkflowExecutionScope is the metric for frontend.RestartWorkflowExecution\n\tFrontendRestartWorkflowExecutionScope = iota + NumAdminScopes\n\t// FrontendStartWorkflowExecutionScope is the metric scope for frontend.StartWorkflowExecution\n\tFrontendStartWorkflowExecutionScope\n\t// FrontendStartWorkflowExecutionAsyncScope is the metric scope for frontend.StartWorkflowExecutionAsync\n\tFrontendStartWorkflowExecutionAsyncScope\n\t// PollForDecisionTaskScope is the metric scope for frontend.PollForDecisionTask\n\tFrontendPollForDecisionTaskScope\n\t// FrontendPollForActivityTaskScope is the metric scope for frontend.PollForActivityTask\n\tFrontendPollForActivityTaskScope\n\t// FrontendRecordActivityTaskHeartbeatScope is the metric scope for frontend.RecordActivityTaskHeartbeat\n\tFrontendRecordActivityTaskHeartbeatScope\n\t// FrontendRecordActivityTaskHeartbeatByIDScope is the metric scope for frontend.RespondDecisionTaskCompleted\n\tFrontendRecordActivityTaskHeartbeatByIDScope\n\t// FrontendRespondDecisionTaskCompletedScope is the metric scope for frontend.RespondDecisionTaskCompleted\n\tFrontendRespondDecisionTaskCompletedScope\n\t// FrontendRespondDecisionTaskFailedScope is the metric scope for frontend.RespondDecisionTaskFailed\n\tFrontendRespondDecisionTaskFailedScope\n\t// FrontendRespondQueryTaskCompletedScope is the metric scope for frontend.RespondQueryTaskCompleted\n\tFrontendRespondQueryTaskCompletedScope\n\t// FrontendRespondActivityTaskCompletedScope is the metric scope for frontend.RespondActivityTaskCompleted\n\tFrontendRespondActivityTaskCompletedScope\n\t// FrontendRespondActivityTaskFailedScope is the metric scope for frontend.RespondActivityTaskFailed\n\tFrontendRespondActivityTaskFailedScope\n\t// FrontendRespondActivityTaskCanceledScope is the metric scope for frontend.RespondActivityTaskCanceled\n\tFrontendRespondActivityTaskCanceledScope\n\t// FrontendRespondActivityTaskCompletedScope is the metric scope for frontend.RespondActivityTaskCompletedByID\n\tFrontendRespondActivityTaskCompletedByIDScope\n\t// FrontendRespondActivityTaskFailedScope is the metric scope for frontend.RespondActivityTaskFailedByID\n\tFrontendRespondActivityTaskFailedByIDScope\n\t// FrontendRespondActivityTaskCanceledScope is the metric scope for frontend.RespondActivityTaskCanceledByID\n\tFrontendRespondActivityTaskCanceledByIDScope\n\t// FrontendGetWorkflowExecutionHistoryScope is the metric scope for frontend.GetWorkflowExecutionHistory\n\tFrontendGetWorkflowExecutionHistoryScope\n\t// FrontendGetWorkflowExecutionRawHistoryScope is the metric scope for frontend.GetWorkflowExecutionRawHistory\n\tFrontendGetWorkflowExecutionRawHistoryScope\n\t// FrontendPollForWorklfowExecutionRawHistoryScope is the metric scope for frontend.GetWorkflowExecutionRawHistory\n\tFrontendPollForWorklfowExecutionRawHistoryScope\n\t// FrontendSignalWorkflowExecutionScope is the metric scope for frontend.SignalWorkflowExecution\n\tFrontendSignalWorkflowExecutionScope\n\t// FrontendSignalWithStartWorkflowExecutionScope is the metric scope for frontend.SignalWithStartWorkflowExecution\n\tFrontendSignalWithStartWorkflowExecutionScope\n\t// FrontendSignalWithStartWorkflowExecutionAsyncScope is the metric scope for frontend.SignalWithStartWorkflowExecutionAsync\n\tFrontendSignalWithStartWorkflowExecutionAsyncScope\n\t// FrontendTerminateWorkflowExecutionScope is the metric scope for frontend.TerminateWorkflowExecution\n\tFrontendTerminateWorkflowExecutionScope\n\t// FrontendRequestCancelWorkflowExecutionScope is the metric scope for frontend.RequestCancelWorkflowExecution\n\tFrontendRequestCancelWorkflowExecutionScope\n\t// FrontendListArchivedWorkflowExecutionsScope is the metric scope for frontend.ListArchivedWorkflowExecutions\n\tFrontendListArchivedWorkflowExecutionsScope\n\t// FrontendListOpenWorkflowExecutionsScope is the metric scope for frontend.ListOpenWorkflowExecutions\n\tFrontendListOpenWorkflowExecutionsScope\n\t// FrontendListClosedWorkflowExecutionsScope is the metric scope for frontend.ListClosedWorkflowExecutions\n\tFrontendListClosedWorkflowExecutionsScope\n\t// FrontendListWorkflowExecutionsScope is the metric scope for frontend.ListWorkflowExecutions\n\tFrontendListWorkflowExecutionsScope\n\t// FrontendScanWorkflowExecutionsScope is the metric scope for frontend.ListWorkflowExecutions\n\tFrontendScanWorkflowExecutionsScope\n\t// FrontendCountWorkflowExecutionsScope is the metric scope for frontend.CountWorkflowExecutions\n\tFrontendCountWorkflowExecutionsScope\n\t// FrontendRegisterDomainScope is the metric scope for frontend.RegisterDomain\n\tFrontendRegisterDomainScope\n\t// FrontendDescribeDomainScope is the metric scope for frontend.DescribeDomain\n\tFrontendDescribeDomainScope\n\t// FrontendUpdateDomainScope is the metric scope for frontend.DescribeDomain\n\tFrontendUpdateDomainScope\n\t// FrontendDeleteDomainScope is the metric scope for frontend.DeleteDomain\n\tFrontendDeleteDomainScope\n\t// FrontendDeprecateDomainScope is the metric scope for frontend.DeprecateDomain\n\tFrontendDeprecateDomainScope\n\t// FrontendFailoverDomainScope is the metric scope for frontend.FailoverDomain\n\tFrontendFailoverDomainScope\n\t// FrontendListFailoverHistoryScope is the metric scope for frontend.ListFailoverHistory\n\tFrontendListFailoverHistoryScope\n\t// FrontendQueryWorkflowScope is the metric scope for frontend.QueryWorkflow\n\tFrontendQueryWorkflowScope\n\t// FrontendDescribeWorkflowExecutionScope is the metric scope for frontend.DescribeWorkflowExecution\n\tFrontendDescribeWorkflowExecutionScope\n\t// FrontendDiagnoseWorkflowExecutionScope is the metric scope for frontend.DescribeWorkflowExecution\n\tFrontendDiagnoseWorkflowExecutionScope\n\t// FrontendDescribeWorkflowExecutionStatusScope is a custom metric for more\n\t// rich details about workflow description calls, including workflow open/closed status\n\tFrontendDescribeWorkflowExecutionStatusScope\n\t// FrontendDescribeTaskListScope is the metric scope for frontend.DescribeTaskList\n\tFrontendDescribeTaskListScope\n\t// FrontendResetStickyTaskListScope is the metric scope for frontend.ResetStickyTaskList\n\tFrontendListTaskListPartitionsScope\n\t// FrontendGetTaskListsByDomainScope is the metric scope for frontend.ResetStickyTaskList\n\tFrontendGetTaskListsByDomainScope\n\t// FrontendRefreshWorkflowTasksScope is the metric scope for frontend.RefreshWorkflowTasks\n\tFrontendRefreshWorkflowTasksScope\n\t// FrontendResetStickyTaskListScope is the metric scope for frontend.ResetStickyTaskList\n\tFrontendResetStickyTaskListScope\n\t// FrontendListDomainsScope is the metric scope for frontend.ListDomain\n\tFrontendListDomainsScope\n\t// FrontendResetWorkflowExecutionScope is the metric scope for frontend.ResetWorkflowExecution\n\tFrontendResetWorkflowExecutionScope\n\t// FrontendGetSearchAttributesScope is the metric scope for frontend.GetSearchAttributes\n\tFrontendGetSearchAttributesScope\n\t// FrontendGetClusterInfoScope is the metric scope for frontend.GetClusterInfo\n\tFrontendGetClusterInfoScope\n\n\tNumFrontendScopes\n)\n\n// -- Operation scopes for History service --\nconst (\n\t// HistoryStartWorkflowExecutionScope tracks StartWorkflowExecution API calls received by service\n\tHistoryStartWorkflowExecutionScope = iota + NumFrontendScopes\n\t// HistoryRecordActivityTaskHeartbeatScope tracks RecordActivityTaskHeartbeat API calls received by service\n\tHistoryRecordActivityTaskHeartbeatScope\n\t// HistoryRespondDecisionTaskCompletedScope tracks RespondDecisionTaskCompleted API calls received by service\n\tHistoryRespondDecisionTaskCompletedScope\n\t// HistoryRespondDecisionTaskFailedScope tracks RespondDecisionTaskFailed API calls received by service\n\tHistoryRespondDecisionTaskFailedScope\n\t// HistoryRespondActivityTaskCompletedScope tracks RespondActivityTaskCompleted API calls received by service\n\tHistoryRespondActivityTaskCompletedScope\n\t// HistoryRespondActivityTaskFailedScope tracks RespondActivityTaskFailed API calls received by service\n\tHistoryRespondActivityTaskFailedScope\n\t// HistoryRespondActivityTaskCanceledScope tracks RespondActivityTaskCanceled API calls received by service\n\tHistoryRespondActivityTaskCanceledScope\n\t// HistoryResetQueueScope tracks ResetQueue API calls received by service\n\tHistoryResetQueueScope\n\t// HistoryDescribeQueueScope tracks DescribeQueue API calls received by service\n\tHistoryDescribeQueueScope\n\t// HistoryDescribeMutabelStateScope tracks DescribeMutableState API calls received by service\n\tHistoryDescribeMutabelStateScope\n\t// HistoryGetMutableStateScope tracks GetMutableState API calls received by service\n\tHistoryGetMutableStateScope\n\t// HistoryPollMutableStateScope tracks PollMutableState API calls received by service\n\tHistoryPollMutableStateScope\n\t// HistoryResetStickyTaskListScope tracks ResetStickyTaskList API calls received by service\n\tHistoryResetStickyTaskListScope\n\t// HistoryDescribeWorkflowExecutionScope tracks DescribeWorkflowExecution API calls received by service\n\tHistoryDescribeWorkflowExecutionScope\n\t// HistoryRecordDecisionTaskStartedScope tracks RecordDecisionTaskStarted API calls received by service\n\tHistoryRecordDecisionTaskStartedScope\n\t// HistoryRecordActivityTaskStartedScope tracks RecordActivityTaskStarted API calls received by service\n\tHistoryRecordActivityTaskStartedScope\n\t// HistorySignalWorkflowExecutionScope tracks SignalWorkflowExecution API calls received by service\n\tHistorySignalWorkflowExecutionScope\n\t// HistorySignalWithStartWorkflowExecutionScope tracks SignalWithStartWorkflowExecution API calls received by service\n\tHistorySignalWithStartWorkflowExecutionScope\n\t// HistoryRemoveSignalMutableStateScope tracks RemoveSignalMutableState API calls received by service\n\tHistoryRemoveSignalMutableStateScope\n\t// HistoryTerminateWorkflowExecutionScope tracks TerminateWorkflowExecution API calls received by service\n\tHistoryTerminateWorkflowExecutionScope\n\t// HistoryScheduleDecisionTaskScope tracks ScheduleDecisionTask API calls received by service\n\tHistoryScheduleDecisionTaskScope\n\t// HistoryRecordChildExecutionCompletedScope tracks CompleteChildExecution API calls received by service\n\tHistoryRecordChildExecutionCompletedScope\n\t// HistoryRequestCancelWorkflowExecutionScope tracks RequestCancelWorkflowExecution API calls received by service\n\tHistoryRequestCancelWorkflowExecutionScope\n\t// HistoryReplicateEventsScope tracks ReplicateEvents API calls received by service\n\tHistoryReplicateEventsScope\n\t// HistoryReplicateRawEventsScope tracks ReplicateEvents API calls received by service\n\tHistoryReplicateRawEventsScope\n\t// HistoryReplicateEventsV2Scope tracks ReplicateEvents API calls received by service\n\tHistoryReplicateEventsV2Scope\n\t// HistorySyncShardStatusScope tracks HistorySyncShardStatus API calls received by service\n\tHistorySyncShardStatusScope\n\t// HistorySyncActivityScope tracks HistoryActivity API calls received by service\n\tHistorySyncActivityScope\n\t// HistoryDescribeMutableStateScope tracks HistoryActivity API calls received by service\n\tHistoryDescribeMutableStateScope\n\t// GetReplicationMessages tracks GetReplicationMessages API calls received by service\n\tHistoryGetReplicationMessagesScope\n\t// HistoryGetDLQReplicationMessagesScope tracks GetReplicationMessages API calls received by service\n\tHistoryGetDLQReplicationMessagesScope\n\t// HistoryCountDLQMessagesScope tracks CountDLQMessages API calls received by service\n\tHistoryCountDLQMessagesScope\n\t// HistoryReadDLQMessagesScope tracks ReadDLQMessages API calls received by service\n\tHistoryReadDLQMessagesScope\n\t// HistoryPurgeDLQMessagesScope tracks PurgeDLQMessages API calls received by service\n\tHistoryPurgeDLQMessagesScope\n\t// HistoryMergeDLQMessagesScope tracks MergeDLQMessages API calls received by service\n\tHistoryMergeDLQMessagesScope\n\t// HistoryShardControllerScope is the scope used by shard controller\n\tHistoryShardControllerScope\n\t// HistoryReapplyEventsScope tracks ReapplyEvents API calls received by service\n\tHistoryReapplyEventsScope\n\t// HistoryRefreshWorkflowTasksScope tracks RefreshWorkflowTasks API calls received by service\n\tHistoryRefreshWorkflowTasksScope\n\t// HistoryNotifyFailoverMarkersScope is the scope used by notify failover marker API\n\tHistoryNotifyFailoverMarkersScope\n\t// HistoryGetCrossClusterTasksScope tracks GetCrossClusterTasks API calls received by service\n\tHistoryGetCrossClusterTasksScope\n\t// HistoryRespondCrossClusterTasksCompletedScope tracks RespondCrossClusterTasksCompleted API calls received by service\n\tHistoryRespondCrossClusterTasksCompletedScope\n\t// HistoryGetFailoverInfoScope tracks HistoryGetFailoverInfo API calls received by service\n\tHistoryGetFailoverInfoScope\n\t// HistoryRatelimitUpdateScope tracks RatelimitUpdate API calls received by the history service\n\tHistoryRatelimitUpdateScope\n\t// TaskPriorityAssignerScope is the scope used by all metric emitted by task priority assigner\n\tTaskPriorityAssignerScope\n\t// TransferQueueProcessorScope is the scope used by all metric emitted by transfer queue processor\n\tTransferQueueProcessorScope\n\t// TransferQueueProcessorV2Scope is the scope used by all metric emitted by transfer queue processor\n\tTransferQueueProcessorV2Scope\n\t// TransferActiveQueueProcessorScope is the scope used by all metric emitted by transfer queue processor\n\tTransferActiveQueueProcessorScope\n\t// TransferStandbyQueueProcessorScope is the scope used by all metric emitted by transfer queue processor\n\tTransferStandbyQueueProcessorScope\n\t// TransferActiveTaskActivityScope is the scope used for activity task processing by transfer queue processor\n\tTransferActiveTaskActivityScope\n\t// TransferActiveTaskDecisionScope is the scope used for decision task processing by transfer queue processor\n\tTransferActiveTaskDecisionScope\n\t// TransferActiveTaskCloseExecutionScope is the scope used for close execution task processing by transfer queue processor\n\tTransferActiveTaskCloseExecutionScope\n\t// TransferActiveTaskCancelExecutionScope is the scope used for cancel execution task processing by transfer queue processor\n\tTransferActiveTaskCancelExecutionScope\n\t// TransferActiveTaskSignalExecutionScope is the scope used for signal execution task processing by transfer queue processor\n\tTransferActiveTaskSignalExecutionScope\n\t// TransferActiveTaskStartChildExecutionScope is the scope used for start child execution task processing by transfer queue processor\n\tTransferActiveTaskStartChildExecutionScope\n\t// TransferActiveTaskRecordWorkflowStartedScope is the scope used for record workflow started task processing by transfer queue processor\n\tTransferActiveTaskRecordWorkflowStartedScope\n\t// TransferActiveTaskResetWorkflowScope is the scope used for record workflow started task processing by transfer queue processor\n\tTransferActiveTaskResetWorkflowScope\n\t// TransferActiveTaskUpsertWorkflowSearchAttributesScope is the scope used for upsert search attributes processing by transfer queue processor\n\tTransferActiveTaskUpsertWorkflowSearchAttributesScope\n\t// TransferActiveTaskRecordWorkflowClosedScope is the scope used for record workflow closed task processing by transfer queue processor\n\tTransferActiveTaskRecordWorkflowClosedScope\n\t// TransferActiveTaskRecordChildExecutionCompletedScope is the scope used for record child execution completed task processing by transfer queue processor\n\tTransferActiveTaskRecordChildExecutionCompletedScope\n\t// TransferActiveTaskApplyParentClosePolicyScope is the scope used for apply parent close policy task processing by transfer queue processor\n\tTransferActiveTaskApplyParentClosePolicyScope\n\t// TransferStandbyTaskResetWorkflowScope is the scope used for record workflow started task processing by transfer queue processor\n\tTransferStandbyTaskResetWorkflowScope\n\t// TransferStandbyTaskActivityScope is the scope used for activity task processing by transfer queue processor\n\tTransferStandbyTaskActivityScope\n\t// TransferStandbyTaskDecisionScope is the scope used for decision task processing by transfer queue processor\n\tTransferStandbyTaskDecisionScope\n\t// TransferStandbyTaskCloseExecutionScope is the scope used for close execution task processing by transfer queue processor\n\tTransferStandbyTaskCloseExecutionScope\n\t// TransferStandbyTaskCancelExecutionScope is the scope used for cancel execution task processing by transfer queue processor\n\tTransferStandbyTaskCancelExecutionScope\n\t// TransferStandbyTaskSignalExecutionScope is the scope used for signal execution task processing by transfer queue processor\n\tTransferStandbyTaskSignalExecutionScope\n\t// TransferStandbyTaskStartChildExecutionScope is the scope used for start child execution task processing by transfer queue processor\n\tTransferStandbyTaskStartChildExecutionScope\n\t// TransferStandbyTaskRecordWorkflowStartedScope is the scope used for record workflow started task processing by transfer queue processor\n\tTransferStandbyTaskRecordWorkflowStartedScope\n\t// TransferStandbyTaskUpsertWorkflowSearchAttributesScope is the scope used for upsert search attributes processing by transfer queue processor\n\tTransferStandbyTaskUpsertWorkflowSearchAttributesScope\n\t// TransferActiveTaskRecordWorkflowClosedScope is the scope used for record workflow closed task processing by transfer queue processor\n\tTransferStandbyTaskRecordWorkflowClosedScope\n\t// TransferActiveTaskRecordChildExecutionCompletedScope is the scope used for record child execution completed task processing by transfer queue processor\n\tTransferStandbyTaskRecordChildExecutionCompletedScope\n\t// TransferActiveTaskApplyParentClosePolicyScope is the scope used for apply parent close policy task processing by transfer queue processor\n\tTransferStandbyTaskApplyParentClosePolicyScope\n\t// TimerQueueProcessorScope is the scope used by all metric emitted by timer queue processor\n\tTimerQueueProcessorScope\n\t// TimerQueueProcessorV2Scope is the scope used by all metric emitted by timer queue processor\n\tTimerQueueProcessorV2Scope\n\t// TimerActiveQueueProcessorScope is the scope used by all metric emitted by timer queue processor\n\tTimerActiveQueueProcessorScope\n\t// TimerQueueProcessorScope is the scope used by all metric emitted by timer queue processor\n\tTimerStandbyQueueProcessorScope\n\t// TimerActiveTaskActivityTimeoutScope is the scope used by metric emitted by timer queue processor for processing activity timeouts\n\tTimerActiveTaskActivityTimeoutScope\n\t// TimerActiveTaskDecisionTimeoutScope is the scope used by metric emitted by timer queue processor for processing decision timeouts\n\tTimerActiveTaskDecisionTimeoutScope\n\t// TimerActiveTaskUserTimerScope is the scope used by metric emitted by timer queue processor for processing user timers\n\tTimerActiveTaskUserTimerScope\n\t// TimerActiveTaskWorkflowTimeoutScope is the scope used by metric emitted by timer queue processor for processing workflow timeouts.\n\tTimerActiveTaskWorkflowTimeoutScope\n\t// TimerActiveTaskActivityRetryTimerScope is the scope used by metric emitted by timer queue processor for processing retry task.\n\tTimerActiveTaskActivityRetryTimerScope\n\t// TimerActiveTaskWorkflowBackoffTimerScope is the scope used by metric emitted by timer queue processor for processing retry task.\n\tTimerActiveTaskWorkflowBackoffTimerScope\n\t// TimerActiveTaskDeleteHistoryEventScope is the scope used by metric emitted by timer queue processor for processing history event cleanup\n\tTimerActiveTaskDeleteHistoryEventScope\n\t// TimerStandbyTaskActivityTimeoutScope is the scope used by metric emitted by timer queue processor for processing activity timeouts\n\tTimerStandbyTaskActivityTimeoutScope\n\t// TimerStandbyTaskDecisionTimeoutScope is the scope used by metric emitted by timer queue processor for processing decision timeouts\n\tTimerStandbyTaskDecisionTimeoutScope\n\t// TimerStandbyTaskUserTimerScope is the scope used by metric emitted by timer queue processor for processing user timers\n\tTimerStandbyTaskUserTimerScope\n\t// TimerStandbyTaskWorkflowTimeoutScope is the scope used by metric emitted by timer queue processor for processing workflow timeouts.\n\tTimerStandbyTaskWorkflowTimeoutScope\n\t// TimerStandbyTaskActivityRetryTimerScope is the scope used by metric emitted by timer queue processor for processing retry task.\n\tTimerStandbyTaskActivityRetryTimerScope\n\t// TimerStandbyTaskDeleteHistoryEventScope is the scope used by metric emitted by timer queue processor for processing history event cleanup\n\tTimerStandbyTaskDeleteHistoryEventScope\n\t// TimerStandbyTaskWorkflowBackoffTimerScope is the scope used by metric emitted by timer queue processor for processing retry task.\n\tTimerStandbyTaskWorkflowBackoffTimerScope\n\t// CrossClusterQueueProcessorScope is the scope used by all metric emitted by cross cluster queue processor in the source cluster\n\tCrossClusterQueueProcessorScope\n\t// CrossClusterTaskProcessorScope is the scope used by all metric emitted by cross cluster task processor in the target cluster\n\tCrossClusterTaskProcessorScope\n\t// CrossClusterTaskFetcherScope is the scope used by all metrics emitted by cross cluster task fetcher in the target cluster\n\tCrossClusterTaskFetcherScope\n\t// CrossClusterSourceTaskStartChildExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing start child workflow task.\n\tCrossClusterSourceTaskStartChildExecutionScope\n\t// CrossClusterSourceTaskCancelExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing cancel workflow task.\n\tCrossClusterSourceTaskCancelExecutionScope\n\t// CrossClusterSourceTaskSignalExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing signal workflow task.\n\tCrossClusterSourceTaskSignalExecutionScope\n\t// CrossClusterSourceTaskRecordChildWorkflowExecutionCompleteScope is the scope used by metric emitted by cross cluster queue processor for recording child workflow completion task.\n\tCrossClusterSourceTaskRecordChildWorkflowExecutionCompleteScope\n\t// CrossClusterSourceTaskApplyParentClosePolicyScope is the scope used by metric emitted by cross cluster queue processor for processing applying parent close policy\n\tCrossClusterSourceTaskApplyParentClosePolicyScope\n\t// CrossClusterTargetTaskStartChildExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing start child workflow task.\n\tCrossClusterTargetTaskStartChildExecutionScope\n\t// CrossClusterTargetTaskCancelExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing cancel workflow task.\n\tCrossClusterTargetTaskCancelExecutionScope\n\t// CrossClusterTargetTaskSignalExecutionScope is the scope used by metric emitted by cross cluster queue processor for processing signal workflow task.\n\tCrossClusterTargetTaskSignalExecutionScope\n\t// CrossClusterTargetTaskRecordChildWorkflowExecutionCompleteScope is the scope used by metric emitted by cross cluster queue processor for recording child workflow completion task.\n\tCrossClusterTargetTaskRecordChildWorkflowExecutionCompleteScope\n\t// CrossClusterTargetTaskApplyParentClosePolicyScope is the scope used by metric emitted by cross cluster queue processor for processing applying parent close policy\n\tCrossClusterTargetTaskApplyParentClosePolicyScope\n\t// HistoryEventNotificationScope is the scope used by shard history event notification\n\tHistoryEventNotificationScope\n\t// ReplicatorQueueProcessorScope is the scope used by all metric emitted by replicator queue processor\n\tReplicatorQueueProcessorScope\n\t// ReplicatorCacheManagerScope is the scope used by all metric emitted by replicator cache manager\n\tReplicatorCacheManagerScope\n\t// ReplicatorTaskHistoryScope is the scope used for history task processing by replicator queue processor\n\tReplicatorTaskHistoryScope\n\t// ReplicatorTaskSyncActivityScope is the scope used for sync activity by replicator queue processor\n\tReplicatorTaskSyncActivityScope\n\t// ReplicateHistoryEventsScope is the scope used by historyReplicator API for applying events\n\tReplicateHistoryEventsScope\n\t// ReplicationMetricEmitterScope is the scope used by all metrics emitted by replication metric emitter\n\tReplicationMetricEmitterScope\n\t// ShardInfoScope is the scope used when updating shard info\n\tShardInfoScope\n\t// WorkflowContextScope is the scope used by WorkflowContext component\n\tWorkflowContextScope\n\t// HistoryCacheGetAndCreateScope is the scope used by history cache\n\tHistoryCacheGetAndCreateScope\n\t// HistoryCacheGetOrCreateScope is the scope used by history cache\n\tHistoryCacheGetOrCreateScope\n\t// HistoryCacheGetOrCreateCurrentScope is the scope used by history cache\n\tHistoryCacheGetOrCreateCurrentScope\n\t// HistoryCacheGetCurrentExecutionScope is the scope used by history cache for getting current execution\n\tHistoryCacheGetCurrentExecutionScope\n\t// EventsCacheGetEventScope is the scope used by events cache\n\tEventsCacheGetEventScope\n\t// EventsCachePutEventScope is the scope used by events cache\n\tEventsCachePutEventScope\n\t// EventsCacheGetFromStoreScope is the scope used by events cache\n\tEventsCacheGetFromStoreScope\n\t// ExecutionSizeStatsScope is the scope used for emiting workflow execution size related stats\n\tExecutionSizeStatsScope\n\t// ExecutionCountStatsScope is the scope used for emiting workflow execution count related stats\n\tExecutionCountStatsScope\n\t// SessionSizeStatsScope is the scope used for emiting session update size related stats\n\tSessionSizeStatsScope\n\t// SessionCountStatsScope is the scope used for emiting session update count related stats\n\tSessionCountStatsScope\n\t// HistoryResetWorkflowExecutionScope tracks ResetWorkflowExecution API calls received by service\n\tHistoryResetWorkflowExecutionScope\n\t// HistoryQueryWorkflowScope tracks QueryWorkflow API calls received by service\n\tHistoryQueryWorkflowScope\n\t// HistoryProcessDeleteHistoryEventScope tracks ProcessDeleteHistoryEvent processing calls\n\tHistoryProcessDeleteHistoryEventScope\n\t// WorkflowCompletionStatsScope tracks workflow completion updates\n\tWorkflowCompletionStatsScope\n\t// ArchiverClientScope is scope used by all metrics emitted by archiver.Client\n\tArchiverClientScope\n\t// ReplicationTaskFetcherScope is scope used by all metrics emitted by ReplicationTaskFetcher\n\tReplicationTaskFetcherScope\n\t// ReplicationTaskCleanupScope is scope used by all metrics emitted by ReplicationTaskProcessor cleanup\n\tReplicationTaskCleanupScope\n\t// ReplicationDLQStatsScope is scope used by all metrics emitted related to replication DLQ\n\tReplicationDLQStatsScope\n\t// FailoverMarkerScope is scope used by all metrics emitted related to failover marker\n\tFailoverMarkerScope\n\t// HistoryReplicationV2TaskScope is the scope used by history task replication processing\n\tHistoryReplicationV2TaskScope\n\t// SyncActivityTaskScope is the scope used by sync activity information processing\n\tSyncActivityTaskScope\n\t// LargeExecutionSizeShardScope is the scope to track large history size for hotshard detection\n\tLargeExecutionSizeShardScope\n\t// LargeExecutionCountShardScope is the scope to track large history count for hotshard detection\n\tLargeExecutionCountShardScope\n\t// LargeExecutionBlobShardScope is the scope to track large blobs for hotshard detection\n\tLargeExecutionBlobShardScope\n\t// HistoryExecutionCacheScope is the scope used by history execution cache\n\tHistoryExecutionCacheScope\n\t// HistoryWorkflowCacheScope is the scope used by history workflow cache\n\tHistoryWorkflowCacheScope\n\t// HistoryFlushBufferedEventsScope is the scope used by history when flushing buffered events\n\tHistoryFlushBufferedEventsScope\n\t// HistoryTaskSchedulerMigrationScope is the scope used by history task scheduler migration\n\tHistoryTaskSchedulerMigrationScope\n\n\tNumHistoryScopes\n)\n\n// -- Operation scopes for Matching service --\nconst (\n\t// PollForDecisionTaskScope tracks PollForDecisionTask API calls received by service\n\tMatchingPollForDecisionTaskScope = iota + NumHistoryScopes\n\t// PollForActivityTaskScope tracks PollForActivityTask API calls received by service\n\tMatchingPollForActivityTaskScope\n\t// MatchingAddActivityTaskScope tracks AddActivityTask API calls received by service\n\tMatchingAddActivityTaskScope\n\t// MatchingAddDecisionTaskScope tracks AddDecisionTask API calls received by service\n\tMatchingAddDecisionTaskScope\n\t// MatchingAddTaskScope tracks both AddActivityTask and AddDevisionTask API calls received by service\n\tMatchingAddTaskScope\n\t// MatchingTaskListMgrScope is the metrics scope for matching.TaskListManager component\n\tMatchingTaskListMgrScope\n\t// MatchingAdaptiveScalerScope is hte metrics scope for matching's Adaptive Scaler component\n\tMatchingAdaptiveScalerScope\n\t// MatchingQueryWorkflowScope tracks AddDecisionTask API calls received by service\n\tMatchingQueryWorkflowScope\n\t// MatchingRespondQueryTaskCompletedScope tracks AddDecisionTask API calls received by service\n\tMatchingRespondQueryTaskCompletedScope\n\t// MatchingCancelOutstandingPollScope tracks CancelOutstandingPoll API calls received by service\n\tMatchingCancelOutstandingPollScope\n\t// MatchingDescribeTaskListScope tracks DescribeTaskList API calls received by service\n\tMatchingDescribeTaskListScope\n\t// MatchingListTaskListPartitionsScope tracks ListTaskListPartitions API calls received by service\n\tMatchingListTaskListPartitionsScope\n\t// MatchingGetTaskListsByDomainScope tracks GetTaskListsByDomain API calls received by service\n\tMatchingGetTaskListsByDomainScope\n\t// MatchingUpdateTaskListPartitionConfigScope tracks UpdateTaskListPartitionConfig API calls received by service\n\tMatchingUpdateTaskListPartitionConfigScope\n\t// MatchingRefreshTaskListPartitionConfigScope tracks RefreshTaskListPartitionConfig API calls received by service\n\tMatchingRefreshTaskListPartitionConfigScope\n\n\tNumMatchingScopes\n)\n\n// -- Operation scopes for Worker service --\nconst (\n\t// ReplicationScope is the scope used by all metric emitted by replicator\n\tReplicatorScope = iota + NumMatchingScopes\n\t// DomainReplicationTaskScope is the scope used by domain task replication processing\n\tDomainReplicationTaskScope\n\t// ESProcessorScope is scope used by all metric emitted by esProcessor\n\tESProcessorScope\n\t// IndexProcessorScope is scope used by all metric emitted by index processor\n\tIndexProcessorScope\n\t// ArchiverDeleteHistoryActivityScope is scope used by all metrics emitted by archiver.DeleteHistoryActivity\n\tArchiverDeleteHistoryActivityScope\n\t// ArchiverUploadHistoryActivityScope is scope used by all metrics emitted by archiver.UploadHistoryActivity\n\tArchiverUploadHistoryActivityScope\n\t// ArchiverArchiveVisibilityActivityScope is scope used by all metrics emitted by archiver.ArchiveVisibilityActivity\n\tArchiverArchiveVisibilityActivityScope\n\t// ArchiverScope is scope used by all metrics emitted by archiver.Archiver\n\tArchiverScope\n\t// ArchiverPumpScope is scope used by all metrics emitted by archiver.Pump\n\tArchiverPumpScope\n\t// ArchiverArchivalWorkflowScope is scope used by all metrics emitted by archiver.ArchivalWorkflow\n\tArchiverArchivalWorkflowScope\n\t// TaskListScavengerScope is scope used by all metrics emitted by worker.tasklist.Scavenger module\n\tTaskListScavengerScope\n\t// ExecutionsScannerScope is scope used by all metrics emitted by worker.executions.Scanner module\n\tExecutionsScannerScope\n\t// ExecutionsFixerScope is the scope used by all metrics emitted by worker.executions.Fixer module\n\tExecutionsFixerScope\n\t// BatcherScope is scope used by all metrics emitted by worker.Batcher module\n\tBatcherScope\n\t// HistoryScavengerScope is scope used by all metrics emitted by worker.history.Scavenger module\n\tHistoryScavengerScope\n\t// ParentClosePolicyProcessorScope is scope used by all metrics emitted by worker.ParentClosePolicyProcessor\n\tParentClosePolicyProcessorScope\n\t// ShardScannerScope is scope used by all metrics emitted by worker.shardscanner module\n\tShardScannerScope\n\t// CheckDataCorruptionWorkflowScope is scope used by the data corruption workflow\n\tCheckDataCorruptionWorkflowScope\n\t// ESAnalyzerScope is scope used by ElasticSearch Analyzer (esanalyzer) workflow\n\tESAnalyzerScope\n\t// AsyncWorkflowConsumerScope is scope used by async workflow consumer\n\tAsyncWorkflowConsumerScope\n\t// DiagnosticsWorkflowScope is scope used by diagnostics workflow\n\tDiagnosticsWorkflowScope\n\n\tNumWorkerScopes\n)\n\n// -- Operation scopes for ShardDistributor service --\nconst (\n\t// ShardDistributorGetShardOwnerScope tracks GetShardOwner API calls received by service\n\tShardDistributorGetShardOwnerScope = iota + NumWorkerScopes\n\tShardDistributorWatchNamespaceStateScope\n\tShardDistributorHeartbeatScope\n\tShardDistributorAssignLoopScope\n\n\tShardDistributorStoreGetShardOwnerScope\n\tShardDistributorStoreAssignShardScope\n\tShardDistributorStoreAssignShardsScope\n\tShardDistributorStoreDeleteExecutorsScope\n\tShardDistributorStoreGetShardStatsScope\n\tShardDistributorStoreDeleteShardStatsScope\n\tShardDistributorStoreGetHeartbeatScope\n\tShardDistributorStoreGetExecutorScope\n\tShardDistributorStoreGetStateScope\n\tShardDistributorStoreRecordHeartbeatScope\n\tShardDistributorStoreSubscribeToExecutorStatusChangesScope\n\tShardDistributorStoreSubscribeToAssignmentChangesScope\n\tShardDistributorStoreDeleteAssignedStatesScope\n\n\t// The scope for the shard distributor executor\n\tShardDistributorExecutorScope\n\n\t// ShardDistributorWatchScope tracks etcd watch stream processing\n\tShardDistributorWatchScope\n\n\tNumShardDistributorScopes\n)\n\n// ScopeDefs record the scopes for all services\nvar ScopeDefs = map[ServiceIdx]map[ScopeIdx]scopeDefinition{\n\t// common scope Names\n\tCommon: {\n\t\tPersistenceCreateShardScope:                              {operation: \"CreateShard\"},\n\t\tPersistenceGetShardScope:                                 {operation: \"GetShard\"},\n\t\tPersistenceUpdateShardScope:                              {operation: \"UpdateShard\"},\n\t\tPersistenceCreateWorkflowExecutionScope:                  {operation: \"CreateWorkflowExecution\"},\n\t\tPersistenceGetWorkflowExecutionScope:                     {operation: \"GetWorkflowExecution\"},\n\t\tPersistenceUpdateWorkflowExecutionScope:                  {operation: \"UpdateWorkflowExecution\"},\n\t\tPersistenceConflictResolveWorkflowExecutionScope:         {operation: \"ConflictResolveWorkflowExecution\"},\n\t\tPersistenceResetWorkflowExecutionScope:                   {operation: \"ResetWorkflowExecution\"},\n\t\tPersistenceDeleteWorkflowExecutionScope:                  {operation: \"DeleteWorkflowExecution\"},\n\t\tPersistenceDeleteCurrentWorkflowExecutionScope:           {operation: \"DeleteCurrentWorkflowExecution\"},\n\t\tPersistenceGetCurrentExecutionScope:                      {operation: \"GetCurrentExecution\"},\n\t\tPersistenceIsWorkflowExecutionExistsScope:                {operation: \"IsWorkflowExecutionExists\"},\n\t\tPersistenceListCurrentExecutionsScope:                    {operation: \"ListCurrentExecutions\"},\n\t\tPersistenceListConcreteExecutionsScope:                   {operation: \"ListConcreteExecutions\"},\n\t\tPersistenceGetTransferTasksScope:                         {operation: \"GetTransferTasks\"},\n\t\tPersistenceCompleteTransferTaskScope:                     {operation: \"CompleteTransferTask\"},\n\t\tPersistenceGetCrossClusterTasksScope:                     {operation: \"GetCrossClusterTasks\"},\n\t\tPersistenceCompleteCrossClusterTaskScope:                 {operation: \"GetCrossClusterTasks\"},\n\t\tPersistenceGetReplicationTasksScope:                      {operation: \"GetReplicationTasks\"},\n\t\tPersistenceCompleteReplicationTaskScope:                  {operation: \"CompleteReplicationTask\"},\n\t\tPersistencePutReplicationTaskToDLQScope:                  {operation: \"PutReplicationTaskToDLQ\"},\n\t\tPersistenceGetReplicationTasksFromDLQScope:               {operation: \"GetReplicationTasksFromDLQ\"},\n\t\tPersistenceGetReplicationDLQSizeScope:                    {operation: \"GetReplicationDLQSize\"},\n\t\tPersistenceDeleteReplicationTaskFromDLQScope:             {operation: \"DeleteReplicationTaskFromDLQ\"},\n\t\tPersistenceRangeDeleteReplicationTaskFromDLQScope:        {operation: \"RangeDeleteReplicationTaskFromDLQ\"},\n\t\tPersistenceCreateFailoverMarkerTasksScope:                {operation: \"CreateFailoverMarkerTasks\"},\n\t\tPersistenceGetTimerIndexTasksScope:                       {operation: \"GetTimerIndexTasks\"},\n\t\tPersistenceCompleteTimerTaskScope:                        {operation: \"CompleteTimerTask\"},\n\t\tPersistenceGetHistoryTasksScope:                          {operation: \"GetHistoryTasks\"},\n\t\tPersistenceCompleteHistoryTaskScope:                      {operation: \"CompleteHistoryTask\"},\n\t\tPersistenceRangeCompleteHistoryTaskScope:                 {operation: \"RangeCompleteHistoryTask\"},\n\t\tPersistenceCreateTasksScope:                              {operation: \"CreateTask\"},\n\t\tPersistenceGetTasksScope:                                 {operation: \"GetTasks\"},\n\t\tPersistenceCompleteTaskScope:                             {operation: \"CompleteTask\"},\n\t\tPersistenceCompleteTasksLessThanScope:                    {operation: \"CompleteTasksLessThan\"},\n\t\tPersistenceGetOrphanTasksScope:                           {operation: \"GetOrphanTasks\"},\n\t\tPersistenceLeaseTaskListScope:                            {operation: \"LeaseTaskList\"},\n\t\tPersistenceGetTaskListScope:                              {operation: \"GetTaskList\"},\n\t\tPersistenceUpdateTaskListScope:                           {operation: \"UpdateTaskList\"},\n\t\tPersistenceListTaskListScope:                             {operation: \"ListTaskList\"},\n\t\tPersistenceDeleteTaskListScope:                           {operation: \"DeleteTaskList\"},\n\t\tPersistenceGetTaskListSizeScope:                          {operation: \"GetTaskListSize\"},\n\t\tPersistenceAppendHistoryEventsScope:                      {operation: \"AppendHistoryEvents\"},\n\t\tPersistenceGetWorkflowExecutionHistoryScope:              {operation: \"GetWorkflowExecutionHistory\"},\n\t\tPersistenceDeleteWorkflowExecutionHistoryScope:           {operation: \"DeleteWorkflowExecutionHistory\"},\n\t\tPersistenceCreateDomainScope:                             {operation: \"CreateDomain\"},\n\t\tPersistenceGetDomainScope:                                {operation: \"GetDomain\"},\n\t\tPersistenceUpdateDomainScope:                             {operation: \"UpdateDomain\"},\n\t\tPersistenceDeleteDomainScope:                             {operation: \"DeleteDomain\"},\n\t\tPersistenceDeleteDomainByNameScope:                       {operation: \"DeleteDomainByName\"},\n\t\tPersistenceListDomainsScope:                              {operation: \"ListDomain\"},\n\t\tPersistenceGetMetadataScope:                              {operation: \"GetMetadata\"},\n\t\tPersistenceRecordWorkflowExecutionStartedScope:           {operation: \"RecordWorkflowExecutionStarted\"},\n\t\tPersistenceRecordWorkflowExecutionClosedScope:            {operation: \"RecordWorkflowExecutionClosed\"},\n\t\tPersistenceRecordWorkflowExecutionUninitializedScope:     {operation: \"RecordWorkflowExecutionUninitialized\"},\n\t\tPersistenceUpsertWorkflowExecutionScope:                  {operation: \"UpsertWorkflowExecution\"},\n\t\tPersistenceListOpenWorkflowExecutionsScope:               {operation: \"ListOpenWorkflowExecutions\"},\n\t\tPersistenceListClosedWorkflowExecutionsScope:             {operation: \"ListClosedWorkflowExecutions\"},\n\t\tPersistenceListOpenWorkflowExecutionsByTypeScope:         {operation: \"ListOpenWorkflowExecutionsByType\"},\n\t\tPersistenceListClosedWorkflowExecutionsByTypeScope:       {operation: \"ListClosedWorkflowExecutionsByType\"},\n\t\tPersistenceListOpenWorkflowExecutionsByWorkflowIDScope:   {operation: \"ListOpenWorkflowExecutionsByWorkflowID\"},\n\t\tPersistenceListClosedWorkflowExecutionsByWorkflowIDScope: {operation: \"ListClosedWorkflowExecutionsByWorkflowID\"},\n\t\tPersistenceListClosedWorkflowExecutionsByStatusScope:     {operation: \"ListClosedWorkflowExecutionsByStatus\"},\n\t\tPersistenceGetClosedWorkflowExecutionScope:               {operation: \"GetClosedWorkflowExecution\"},\n\t\tPersistenceVisibilityDeleteWorkflowExecutionScope:        {operation: \"VisibilityDeleteWorkflowExecution\"},\n\t\tPersistenceDeleteUninitializedWorkflowExecutionScope:     {operation: \"VisibilityDeleteUninitializedWorkflowExecution\"},\n\t\tPersistenceListWorkflowExecutionsScope:                   {operation: \"ListWorkflowExecutions\"},\n\t\tPersistenceScanWorkflowExecutionsScope:                   {operation: \"ScanWorkflowExecutions\"},\n\t\tPersistenceCountWorkflowExecutionsScope:                  {operation: \"CountWorkflowExecutions\"},\n\t\tPersistenceAppendHistoryNodesScope:                       {operation: \"AppendHistoryNodes\"},\n\t\tPersistenceReadHistoryBranchScope:                        {operation: \"ReadHistoryBranch\"},\n\t\tPersistenceReadHistoryBranchByBatchScope:                 {operation: \"ReadHistoryBranch\"},\n\t\tPersistenceReadRawHistoryBranchScope:                     {operation: \"ReadHistoryBranch\"},\n\t\tPersistenceForkHistoryBranchScope:                        {operation: \"ForkHistoryBranch\"},\n\t\tPersistenceDeleteHistoryBranchScope:                      {operation: \"DeleteHistoryBranch\"},\n\t\tPersistenceCompleteForkBranchScope:                       {operation: \"CompleteForkBranch\"},\n\t\tPersistenceGetHistoryTreeScope:                           {operation: \"GetHistoryTree\"},\n\t\tPersistenceGetAllHistoryTreeBranchesScope:                {operation: \"GetAllHistoryTreeBranches\"},\n\t\tPersistenceEnqueueMessageScope:                           {operation: \"EnqueueMessage\"},\n\t\tPersistenceEnqueueMessageToDLQScope:                      {operation: \"EnqueueMessageToDLQ\"},\n\t\tPersistenceReadMessagesScope:                             {operation: \"ReadQueueMessages\"},\n\t\tPersistenceReadMessagesFromDLQScope:                      {operation: \"ReadQueueMessagesFromDLQ\"},\n\t\tPersistenceDeleteMessagesBeforeScope:                     {operation: \"DeleteQueueMessages\"},\n\t\tPersistenceDeleteMessageFromDLQScope:                     {operation: \"DeleteQueueMessageFromDLQ\"},\n\t\tPersistenceRangeDeleteMessagesFromDLQScope:               {operation: \"RangeDeleteMessagesFromDLQ\"},\n\t\tPersistenceUpdateAckLevelScope:                           {operation: \"UpdateAckLevel\"},\n\t\tPersistenceGetAckLevelsScope:                             {operation: \"GetAckLevel\"},\n\t\tPersistenceUpdateDLQAckLevelScope:                        {operation: \"UpdateDLQAckLevel\"},\n\t\tPersistenceGetDLQAckLevelsScope:                          {operation: \"GetDLQAckLevel\"},\n\t\tPersistenceGetDLQSizeScope:                               {operation: \"GetDLQSize\"},\n\t\tPersistenceFetchDynamicConfigScope:                       {operation: \"FetchDynamicConfig\"},\n\t\tPersistenceUpdateDynamicConfigScope:                      {operation: \"UpdateDynamicConfig\"},\n\t\tPersistenceShardRequestCountScope:                        {operation: \"ShardIdPersistenceRequest\"},\n\t\tPersistenceGetActiveClusterSelectionPolicyScope:          {operation: \"GetActiveClusterSelectionPolicy\"},\n\t\tPersistenceDeleteActiveClusterSelectionPolicyScope:       {operation: \"DeleteActiveClusterSelectionPolicy\"},\n\t\tResolverHostNotFoundScope:                                {operation: \"ResolverHostNotFound\"},\n\n\t\tClusterMetadataArchivalConfigScope: {operation: \"ArchivalConfig\"},\n\n\t\tHistoryClientStartWorkflowExecutionScope:            {operation: \"HistoryClientStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientDescribeHistoryHostScope:               {operation: \"HistoryClientDescribeHistoryHost\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRemoveTaskScope:                        {operation: \"HistoryClientRemoveTask\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientCloseShardScope:                        {operation: \"HistoryClientCloseShard\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientResetQueueScope:                        {operation: \"HistoryClientResetQueue\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientDescribeQueueScope:                     {operation: \"HistoryClientDescribeQueue\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRecordActivityTaskHeartbeatScope:       {operation: \"HistoryClientRecordActivityTaskHeartbeat\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondDecisionTaskCompletedScope:      {operation: \"HistoryClientRespondDecisionTaskCompleted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondDecisionTaskFailedScope:         {operation: \"HistoryClientRespondDecisionTaskFailed\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondActivityTaskCompletedScope:      {operation: \"HistoryClientRespondActivityTaskCompleted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondActivityTaskFailedScope:         {operation: \"HistoryClientRespondActivityTaskFailed\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondActivityTaskCanceledScope:       {operation: \"HistoryClientRespondActivityTaskCanceled\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientDescribeMutableStateScope:              {operation: \"HistoryClientDescribeMutableState\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetMutableStateScope:                   {operation: \"HistoryClientGetMutableState\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientPollMutableStateScope:                  {operation: \"HistoryClientPollMutableState\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientResetStickyTaskListScope:               {operation: \"HistoryClientResetStickyTaskList\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientDescribeWorkflowExecutionScope:         {operation: \"HistoryClientDescribeWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRecordDecisionTaskStartedScope:         {operation: \"HistoryClientRecordDecisionTaskStarted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRecordActivityTaskStartedScope:         {operation: \"HistoryClientRecordActivityTaskStarted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRequestCancelWorkflowExecutionScope:    {operation: \"HistoryClientRequestCancelWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientSignalWorkflowExecutionScope:           {operation: \"HistoryClientSignalWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientSignalWithStartWorkflowExecutionScope:  {operation: \"HistoryClientSignalWithStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRemoveSignalMutableStateScope:          {operation: \"HistoryClientRemoveSignalMutableState\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientTerminateWorkflowExecutionScope:        {operation: \"HistoryClientTerminateWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientResetWorkflowExecutionScope:            {operation: \"HistoryClientResetWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientScheduleDecisionTaskScope:              {operation: \"HistoryClientScheduleDecisionTask\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRecordChildExecutionCompletedScope:     {operation: \"HistoryClientRecordChildExecutionCompleted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientReplicateEventsV2Scope:                 {operation: \"HistoryClientReplicateEventsV2\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientSyncShardStatusScope:                   {operation: \"HistoryClientSyncShardStatus\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientSyncActivityScope:                      {operation: \"HistoryClientSyncActivity\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetReplicationTasksScope:               {operation: \"HistoryClientGetReplicationTasks\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetDLQReplicationTasksScope:            {operation: \"HistoryClientGetDLQReplicationTasks\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientQueryWorkflowScope:                     {operation: \"HistoryClientQueryWorkflow\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientReapplyEventsScope:                     {operation: \"HistoryClientReapplyEvents\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientCountDLQMessagesScope:                  {operation: \"HistoryClientCountDLQMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientReadDLQMessagesScope:                   {operation: \"HistoryClientReadDLQMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientPurgeDLQMessagesScope:                  {operation: \"HistoryClientPurgeDLQMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientMergeDLQMessagesScope:                  {operation: \"HistoryClientMergeDLQMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRefreshWorkflowTasksScope:              {operation: \"HistoryClientRefreshWorkflowTasks\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientNotifyFailoverMarkersScope:             {operation: \"HistoryClientNotifyFailoverMarkers\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetCrossClusterTasksScope:              {operation: \"HistoryClientGetCrossClusterTasks\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRespondCrossClusterTasksCompletedScope: {operation: \"HistoryClientRespondCrossClusterTasksCompleted\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetFailoverInfoScope:                   {operation: \"HistoryClientGetFailoverInfo\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetDLQReplicationMessagesScope:         {operation: \"HistoryClientGetDLQReplicationMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientGetReplicationMessagesScope:            {operation: \"HistoryClientGetReplicationMessages\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientWfIDCacheScope:                         {operation: \"HistoryClientWfIDCache\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\t\tHistoryClientRatelimitUpdateScope:                   {operation: \"HistoryClientRatelimitUpdate\", tags: map[string]string{CadenceRoleTagName: HistoryClientRoleTagValue}},\n\n\t\tMatchingClientPollForDecisionTaskScope:            {operation: \"MatchingClientPollForDecisionTask\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientPollForActivityTaskScope:            {operation: \"MatchingClientPollForActivityTask\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientAddActivityTaskScope:                {operation: \"MatchingClientAddActivityTask\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientAddDecisionTaskScope:                {operation: \"MatchingClientAddDecisionTask\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientQueryWorkflowScope:                  {operation: \"MatchingClientQueryWorkflow\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientRespondQueryTaskCompletedScope:      {operation: \"MatchingClientRespondQueryTaskCompleted\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientCancelOutstandingPollScope:          {operation: \"MatchingClientCancelOutstandingPoll\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientDescribeTaskListScope:               {operation: \"MatchingClientDescribeTaskList\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientListTaskListPartitionsScope:         {operation: \"MatchingClientListTaskListPartitions\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientGetTaskListsByDomainScope:           {operation: \"MatchingClientGetTaskListsByDomain\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientUpdateTaskListPartitionConfigScope:  {operation: \"MatchingClientUpdateTaskListPartitionConfig\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\t\tMatchingClientRefreshTaskListPartitionConfigScope: {operation: \"MatchingClientRefreshTaskListPartitionConfig\", tags: map[string]string{CadenceRoleTagName: MatchingClientRoleTagValue}},\n\n\t\tFrontendClientDeleteDomainScope:                          {operation: \"FrontendClientDeleteDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientDeprecateDomainScope:                       {operation: \"FrontendClientDeprecateDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientDescribeDomainScope:                        {operation: \"FrontendClientDescribeDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientDescribeTaskListScope:                      {operation: \"FrontendClientDescribeTaskList\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientDescribeWorkflowExecutionScope:             {operation: \"FrontendClientDescribeWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientDiagnoseWorkflowExecutionScope:             {operation: \"FrontendClientDiagnoseWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetWorkflowExecutionHistoryScope:           {operation: \"FrontendClientGetWorkflowExecutionHistory\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetWorkflowExecutionRawHistoryScope:        {operation: \"FrontendClientGetWorkflowExecutionRawHistory\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientPollForWorkflowExecutionRawHistoryScope:    {operation: \"FrontendClientPollForWorkflowExecutionRawHistory\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListArchivedWorkflowExecutionsScope:        {operation: \"FrontendClientListArchivedWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListClosedWorkflowExecutionsScope:          {operation: \"FrontendClientListClosedWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListDomainsScope:                           {operation: \"FrontendClientListDomains\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListOpenWorkflowExecutionsScope:            {operation: \"FrontendClientListOpenWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientPollForActivityTaskScope:                   {operation: \"FrontendClientPollForActivityTask\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientPollForDecisionTaskScope:                   {operation: \"FrontendClientPollForDecisionTask\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientQueryWorkflowScope:                         {operation: \"FrontendClientQueryWorkflow\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRecordActivityTaskHeartbeatScope:           {operation: \"FrontendClientRecordActivityTaskHeartbeat\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRecordActivityTaskHeartbeatByIDScope:       {operation: \"FrontendClientRecordActivityTaskHeartbeatByID\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRegisterDomainScope:                        {operation: \"FrontendClientRegisterDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRequestCancelWorkflowExecutionScope:        {operation: \"FrontendClientRequestCancelWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientResetStickyTaskListScope:                   {operation: \"FrontendClientResetStickyTaskList\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRefreshWorkflowTasksScope:                  {operation: \"FrontendClientRefreshWorkflowTasks\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientResetWorkflowExecutionScope:                {operation: \"FrontendClientResetWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskCanceledScope:           {operation: \"FrontendClientRespondActivityTaskCanceled\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskCanceledByIDScope:       {operation: \"FrontendClientRespondActivityTaskCanceledByID\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskCompletedScope:          {operation: \"FrontendClientRespondActivityTaskCompleted\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskCompletedByIDScope:      {operation: \"FrontendClientRespondActivityTaskCompletedByID\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskFailedScope:             {operation: \"FrontendClientRespondActivityTaskFailed\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondActivityTaskFailedByIDScope:         {operation: \"FrontendClientRespondActivityTaskFailedByID\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondDecisionTaskCompletedScope:          {operation: \"FrontendClientRespondDecisionTaskCompleted\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondDecisionTaskFailedScope:             {operation: \"FrontendClientRespondDecisionTaskFailed\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRespondQueryTaskCompletedScope:             {operation: \"FrontendClientRespondQueryTaskCompleted\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientSignalWithStartWorkflowExecutionScope:      {operation: \"FrontendClientSignalWithStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientSignalWithStartWorkflowExecutionAsyncScope: {operation: \"FrontendClientSignalWithStartWorkflowExecutionAsync\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientSignalWorkflowExecutionScope:               {operation: \"FrontendClientSignalWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientStartWorkflowExecutionScope:                {operation: \"FrontendClientStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientStartWorkflowExecutionAsyncScope:           {operation: \"FrontendClientStartWorkflowExecutionAsync\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientTerminateWorkflowExecutionScope:            {operation: \"FrontendClientTerminateWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientUpdateDomainScope:                          {operation: \"FrontendClientUpdateDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientFailoverDomainScope:                        {operation: \"FrontendClientFailoverDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListFailoverHistoryScope:                   {operation: \"FrontendClientListFailoverHistory\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListWorkflowExecutionsScope:                {operation: \"FrontendClientListWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientScanWorkflowExecutionsScope:                {operation: \"FrontendClientScanWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientCountWorkflowExecutionsScope:               {operation: \"FrontendClientCountWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetSearchAttributesScope:                   {operation: \"FrontendClientGetSearchAttributes\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetReplicationTasksScope:                   {operation: \"FrontendClientGetReplicationTasks\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetDomainReplicationTasksScope:             {operation: \"FrontendClientGetDomainReplicationTasks\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetDLQReplicationTasksScope:                {operation: \"FrontendClientGetDLQReplicationTasks\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientReapplyEventsScope:                         {operation: \"FrontendClientReapplyEvents\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetClusterInfoScope:                        {operation: \"FrontendClientGetClusterInfo\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientListTaskListPartitionsScope:                {operation: \"FrontendClientListTaskListPartitions\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientGetTaskListsByDomainScope:                  {operation: \"FrontendClientGetTaskListsByDomain\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\t\tFrontendClientRestartWorkflowExecutionScope:              {operation: \"FrontendClientRestartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: FrontendClientRoleTagValue}},\n\n\t\tAdminClientGetReplicationTasksScope:                   {operation: \"AdminClientGetReplicationTasks\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientAddSearchAttributeScope:                    {operation: \"AdminClientAddSearchAttribute\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDescribeHistoryHostScope:                   {operation: \"AdminClientDescribeHistoryHost\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDescribeShardDistributionScope:             {operation: \"AdminClientDescribeShardDistribution\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDescribeWorkflowExecutionScope:             {operation: \"AdminClientDescribeWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetWorkflowExecutionRawHistoryV2Scope:      {operation: \"AdminClientGetWorkflowExecutionRawHistoryV2\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDescribeClusterScope:                       {operation: \"AdminClientDescribeCluster\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientRefreshWorkflowTasksScope:                  {operation: \"AdminClientRefreshWorkflowTasks\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientResendReplicationTasksScope:                {operation: \"AdminClientResendReplicationTasks\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientCloseShardScope:                            {operation: \"AdminClientCloseShard\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientRemoveTaskScope:                            {operation: \"AdminClientRemoveTask\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientResetQueueScope:                            {operation: \"AdminClientResetQueue\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDescribeQueueScope:                         {operation: \"AdminClientDescribeQueue\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientCountDLQMessagesScope:                      {operation: \"AdminClientCountDLQMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientReadDLQMessagesScope:                       {operation: \"AdminClientReadDLQMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientPurgeDLQMessagesScope:                      {operation: \"AdminClientPurgeDLQMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientMergeDLQMessagesScope:                      {operation: \"AdminClientMergeDLQMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetCrossClusterTasksScope:                  {operation: \"AdminClientGetCrossClusterTasks\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientRespondCrossClusterTasksCompletedScope:     {operation: \"AdminClientRespondCrossClusterTasksCompleted\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetDynamicConfigScope:                      {operation: \"AdminClientGetDynamicConfig\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientUpdateDynamicConfigScope:                   {operation: \"AdminClientUpdateDynamicConfig\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientRestoreDynamicConfigScope:                  {operation: \"AdminClientRestoreDynamicConfig\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientListDynamicConfigScope:                     {operation: \"AdminClientListDynamicConfigScope\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetGlobalIsolationGroupsScope:              {operation: \"AdminClientGetGlobalIsolationGroups\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientUpdateGlobalIsolationGroupsScope:           {operation: \"AdminClientUpdateGlobalIsolationGroups\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetDomainIsolationGroupsScope:              {operation: \"AdminClientGetDomainIsolationGroups\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientUpdateDomainIsolationGroupsScope:           {operation: \"AdminClientUpdateDomainIsolationGroups\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientDeleteWorkflowScope:                        {operation: \"AdminClientDeleteWorkflow\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientMaintainCorruptWorkflowScope:               {operation: \"AdminClientMaintainCorruptWorkflow\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientReapplyEventsScope:                         {operation: \"AdminClientReapplyEvents\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetDLQReplicationMessagesScope:             {operation: \"AdminClientGetDLQReplicationMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetDomainReplicationMessagesScope:          {operation: \"AdminClientGetDomainReplicationMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetReplicationMessagesScope:                {operation: \"AdminClientGetReplicationMessages\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientGetDomainAsyncWorkflowConfiguratonScope:    {operation: \"AdminClientGetDomainAsyncWorkflowConfiguraton\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientUpdateDomainAsyncWorkflowConfiguratonScope: {operation: \"AdminClientUpdateDomainAsyncWorkflowConfiguraton\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\t\tAdminClientUpdateTaskListPartitionConfigScope:         {operation: \"AdminClientUpdateTaskListPartitionConfig\", tags: map[string]string{CadenceRoleTagName: AdminClientRoleTagValue}},\n\n\t\tDCRedirectionDeleteDomainScope:                          {operation: \"DCRedirectionDeleteDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionDeprecateDomainScope:                       {operation: \"DCRedirectionDeprecateDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionFailoverDomainScope:                        {operation: \"DCRedirectionFailoverDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListFailoverHistoryScope:                   {operation: \"DCRedirectionListFailoverHistory\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionDescribeDomainScope:                        {operation: \"DCRedirectionDescribeDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionDescribeTaskListScope:                      {operation: \"DCRedirectionDescribeTaskList\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionDescribeWorkflowExecutionScope:             {operation: \"DCRedirectionDescribeWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionDiagnoseWorkflowExecutionScope:             {operation: \"DCRedirectionDiagnoseWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionGetWorkflowExecutionHistoryScope:           {operation: \"DCRedirectionGetWorkflowExecutionHistory\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionGetWorkflowExecutionRawHistoryScope:        {operation: \"DCRedirectionGetWorkflowExecutionRawHistoryScope\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionPollForWorklfowExecutionRawHistoryScope:    {operation: \"DCRedirectionPollForWorklfowExecutionRawHistory\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListArchivedWorkflowExecutionsScope:        {operation: \"DCRedirectionListArchivedWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListClosedWorkflowExecutionsScope:          {operation: \"DCRedirectionListClosedWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListDomainsScope:                           {operation: \"DCRedirectionListDomains\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListOpenWorkflowExecutionsScope:            {operation: \"DCRedirectionListOpenWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListWorkflowExecutionsScope:                {operation: \"DCRedirectionListWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionScanWorkflowExecutionsScope:                {operation: \"DCRedirectionScanWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionCountWorkflowExecutionsScope:               {operation: \"DCRedirectionCountWorkflowExecutions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionGetSearchAttributesScope:                   {operation: \"DCRedirectionGetSearchAttributes\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionPollForActivityTaskScope:                   {operation: \"DCRedirectionPollForActivityTask\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionPollForDecisionTaskScope:                   {operation: \"DCRedirectionPollForDecisionTask\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionQueryWorkflowScope:                         {operation: \"DCRedirectionQueryWorkflow\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRecordActivityTaskHeartbeatScope:           {operation: \"DCRedirectionRecordActivityTaskHeartbeat\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRecordActivityTaskHeartbeatByIDScope:       {operation: \"DCRedirectionRecordActivityTaskHeartbeatByID\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRegisterDomainScope:                        {operation: \"DCRedirectionRegisterDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRequestCancelWorkflowExecutionScope:        {operation: \"DCRedirectionRequestCancelWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionResetStickyTaskListScope:                   {operation: \"DCRedirectionResetStickyTaskList\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionResetWorkflowExecutionScope:                {operation: \"DCRedirectionResetWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskCanceledScope:           {operation: \"DCRedirectionRespondActivityTaskCanceled\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskCanceledByIDScope:       {operation: \"DCRedirectionRespondActivityTaskCanceledByID\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskCompletedScope:          {operation: \"DCRedirectionRespondActivityTaskCompleted\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskCompletedByIDScope:      {operation: \"DCRedirectionRespondActivityTaskCompletedByID\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskFailedScope:             {operation: \"DCRedirectionRespondActivityTaskFailed\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondActivityTaskFailedByIDScope:         {operation: \"DCRedirectionRespondActivityTaskFailedByID\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondDecisionTaskCompletedScope:          {operation: \"DCRedirectionRespondDecisionTaskCompleted\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondDecisionTaskFailedScope:             {operation: \"DCRedirectionRespondDecisionTaskFailed\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRespondQueryTaskCompletedScope:             {operation: \"DCRedirectionRespondQueryTaskCompleted\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionSignalWithStartWorkflowExecutionScope:      {operation: \"DCRedirectionSignalWithStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionSignalWithStartWorkflowExecutionAsyncScope: {operation: \"DCRedirectionSignalWithStartWorkflowExecutionAsync\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionSignalWorkflowExecutionScope:               {operation: \"DCRedirectionSignalWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionStartWorkflowExecutionScope:                {operation: \"DCRedirectionStartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionStartWorkflowExecutionAsyncScope:           {operation: \"DCRedirectionStartWorkflowExecutionAsync\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionTerminateWorkflowExecutionScope:            {operation: \"DCRedirectionTerminateWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionUpdateDomainScope:                          {operation: \"DCRedirectionUpdateDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionListTaskListPartitionsScope:                {operation: \"DCRedirectionListTaskListPartitions\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionGetTaskListsByDomainScope:                  {operation: \"DCRedirectionGetTaskListsByDomain\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRefreshWorkflowTasksScope:                  {operation: \"DCRedirectionRefreshWorkflowTasks\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionRestartWorkflowExecutionScope:              {operation: \"DCRedirectionRestartWorkflowExecution\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\t\tDCRedirectionForwardingPolicyScope:                      {operation: \"DCRedirectionForwardingPolicy\", tags: map[string]string{CadenceRoleTagName: DCRedirectionRoleTagValue}},\n\n\t\tMessagingClientPublishScope:      {operation: \"MessagingClientPublish\"},\n\t\tMessagingClientPublishBatchScope: {operation: \"MessagingClientPublishBatch\"},\n\t\tMessagingClientConsumerScope:     {operation: \"MessagingClientConsumerScope\"},\n\n\t\tDomainCacheScope:                                      {operation: \"DomainCache\"},\n\t\tHistoryRereplicationByTransferTaskScope:               {operation: \"HistoryRereplicationByTransferTask\"},\n\t\tHistoryRereplicationByTimerTaskScope:                  {operation: \"HistoryRereplicationByTimerTask\"},\n\t\tHistoryRereplicationByHistoryReplicationScope:         {operation: \"HistoryRereplicationByHistoryReplication\"},\n\t\tHistoryRereplicationByHistoryMetadataReplicationScope: {operation: \"HistoryRereplicationByHistoryMetadataReplication\"},\n\t\tHistoryRereplicationByActivityReplicationScope:        {operation: \"HistoryRereplicationByActivityReplication\"},\n\n\t\tElasticsearchRecordWorkflowExecutionStartedScope:           {operation: \"RecordWorkflowExecutionStarted\"},\n\t\tElasticsearchRecordWorkflowExecutionClosedScope:            {operation: \"RecordWorkflowExecutionClosed\"},\n\t\tElasticsearchRecordWorkflowExecutionUninitializedScope:     {operation: \"RecordWorkflowExecutionUninitialized\"},\n\t\tElasticsearchUpsertWorkflowExecutionScope:                  {operation: \"UpsertWorkflowExecution\"},\n\t\tElasticsearchListOpenWorkflowExecutionsScope:               {operation: \"ListOpenWorkflowExecutions\"},\n\t\tElasticsearchListClosedWorkflowExecutionsScope:             {operation: \"ListClosedWorkflowExecutions\"},\n\t\tElasticsearchListOpenWorkflowExecutionsByTypeScope:         {operation: \"ListOpenWorkflowExecutionsByType\"},\n\t\tElasticsearchListClosedWorkflowExecutionsByTypeScope:       {operation: \"ListClosedWorkflowExecutionsByType\"},\n\t\tElasticsearchListOpenWorkflowExecutionsByWorkflowIDScope:   {operation: \"ListOpenWorkflowExecutionsByWorkflowID\"},\n\t\tElasticsearchListClosedWorkflowExecutionsByWorkflowIDScope: {operation: \"ListClosedWorkflowExecutionsByWorkflowID\"},\n\t\tElasticsearchListClosedWorkflowExecutionsByStatusScope:     {operation: \"ListClosedWorkflowExecutionsByStatus\"},\n\t\tElasticsearchGetClosedWorkflowExecutionScope:               {operation: \"GetClosedWorkflowExecution\"},\n\t\tElasticsearchListWorkflowExecutionsScope:                   {operation: \"ListWorkflowExecutions\"},\n\t\tElasticsearchScanWorkflowExecutionsScope:                   {operation: \"ScanWorkflowExecutions\"},\n\t\tElasticsearchCountWorkflowExecutionsScope:                  {operation: \"CountWorkflowExecutions\"},\n\t\tElasticsearchDeleteWorkflowExecutionsScope:                 {operation: \"DeleteWorkflowExecution\"},\n\t\tElasticsearchDeleteUninitializedWorkflowExecutionsScope:    {operation: \"DeleteUninitializedWorkflowExecution\"},\n\t\tPinotRecordWorkflowExecutionStartedScope:                   {operation: \"RecordWorkflowExecutionStarted\"},\n\t\tPinotRecordWorkflowExecutionClosedScope:                    {operation: \"RecordWorkflowExecutionClosed\"},\n\t\tPinotRecordWorkflowExecutionUninitializedScope:             {operation: \"RecordWorkflowExecutionUninitialized\"},\n\t\tPinotUpsertWorkflowExecutionScope:                          {operation: \"UpsertWorkflowExecution\"},\n\t\tPinotListOpenWorkflowExecutionsScope:                       {operation: \"ListOpenWorkflowExecutions\"},\n\t\tPinotListClosedWorkflowExecutionsScope:                     {operation: \"ListClosedWorkflowExecutions\"},\n\t\tPinotListOpenWorkflowExecutionsByTypeScope:                 {operation: \"ListOpenWorkflowExecutionsByType\"},\n\t\tPinotListClosedWorkflowExecutionsByTypeScope:               {operation: \"ListClosedWorkflowExecutionsByType\"},\n\t\tPinotListOpenWorkflowExecutionsByWorkflowIDScope:           {operation: \"ListOpenWorkflowExecutionsByWorkflowID\"},\n\t\tPinotListClosedWorkflowExecutionsByWorkflowIDScope:         {operation: \"ListClosedWorkflowExecutionsByWorkflowID\"},\n\t\tPinotListClosedWorkflowExecutionsByStatusScope:             {operation: \"ListClosedWorkflowExecutionsByStatus\"},\n\t\tPinotGetClosedWorkflowExecutionScope:                       {operation: \"GetClosedWorkflowExecution\"},\n\t\tPinotListWorkflowExecutionsScope:                           {operation: \"ListWorkflowExecutions\"},\n\t\tPinotScanWorkflowExecutionsScope:                           {operation: \"ScanWorkflowExecutions\"},\n\t\tPinotCountWorkflowExecutionsScope:                          {operation: \"CountWorkflowExecutions\"},\n\t\tPinotDeleteWorkflowExecutionsScope:                         {operation: \"DeleteWorkflowExecution\"},\n\t\tPinotDeleteUninitializedWorkflowExecutionsScope:            {operation: \"DeleteUninitializedWorkflowExecution\"},\n\t\tSequentialTaskProcessingScope:                              {operation: \"SequentialTaskProcessing\"},\n\t\tParallelTaskProcessingScope:                                {operation: \"ParallelTaskProcessing\"},\n\t\tTaskSchedulerScope:                                         {operation: \"TaskScheduler\"},\n\t\tTaskSchedulerRateLimiterScope:                              {operation: \"TaskSchedulerRateLimiter\"},\n\n\t\tHistoryEngineScope:      {operation: \"HistoryEngine\"},\n\t\tHistoryArchiverScope:    {operation: \"HistoryArchiver\"},\n\t\tVisibilityArchiverScope: {operation: \"VisibilityArchiver\"},\n\n\t\tBlobstoreClientUploadScope:          {operation: \"BlobstoreClientUpload\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\t\tBlobstoreClientDownloadScope:        {operation: \"BlobstoreClientDownload\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\t\tBlobstoreClientGetMetadataScope:     {operation: \"BlobstoreClientGetMetadata\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\t\tBlobstoreClientExistsScope:          {operation: \"BlobstoreClientExists\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\t\tBlobstoreClientDeleteScope:          {operation: \"BlobstoreClientDelete\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\t\tBlobstoreClientDirectoryExistsScope: {operation: \"BlobstoreClientDirectoryExists\", tags: map[string]string{CadenceRoleTagName: BlobstoreRoleTagValue}},\n\n\t\tGetAvailableIsolationGroupsScope: {operation: \"GetAvailableIsolationGroups\"},\n\n\t\tDomainFailoverScope:         {operation: \"DomainFailover\"},\n\t\tTaskValidatorScope:          {operation: \"TaskValidation\"},\n\t\tDomainReplicationQueueScope: {operation: \"DomainReplicationQueue\"},\n\t\tClusterMetadataScope:        {operation: \"ClusterMetadata\"},\n\t\tHashringScope:               {operation: \"Hashring\"},\n\n\t\t// currently used by both frontend and history, but may grow to other limiting-host-services.\n\t\tGlobalRatelimiter:           {operation: \"GlobalRatelimiter\"},\n\t\tGlobalRatelimiterAggregator: {operation: \"GlobalRatelimiterAggregator\"},\n\n\t\tP2PRPCPeerChooserScope:       {operation: \"P2PRPCPeerChooser\"},\n\t\tPartitionConfigProviderScope: {operation: \"PartitionConfigProvider\"},\n\n\t\tShardDistributorClientGetShardOwnerScope:       {operation: \"ShardDistributorClientGetShardOwner\"},\n\t\tShardDistributorClientWatchNamespaceStateScope: {operation: \"ShardDistributorClientWatchNamespaceState\"},\n\t\tShardDistributorExecutorClientHeartbeatScope:   {operation: \"ShardDistributorExecutorHeartbeat\"},\n\n\t\tLoadBalancerScope: {operation: \"RRLoadBalancer\"},\n\n\t\tActiveClusterManager:                   {operation: \"ActiveClusterManager\"},\n\t\tActiveClusterManagerWorkflowCacheScope: {operation: \"ActiveClusterManagerWorkflowCache\"},\n\t},\n\t// Frontend Scope Names\n\tFrontend: {\n\t\t// Admin API scope co-locates with frontend\n\t\tAdminRemoveTaskScope:                        {operation: \"AdminRemoveTask\"},\n\t\tAdminCloseShardScope:                        {operation: \"AdminCloseShard\"},\n\t\tAdminResetQueueScope:                        {operation: \"AdminResetQueue\"},\n\t\tAdminDescribeQueueScope:                     {operation: \"AdminDescribeQueue\"},\n\t\tAdminCountDLQMessagesScope:                  {operation: \"AdminCountDLQMessages\"},\n\t\tAdminReadDLQMessagesScope:                   {operation: \"AdminReadDLQMessages\"},\n\t\tAdminPurgeDLQMessagesScope:                  {operation: \"AdminPurgeDLQMessages\"},\n\t\tAdminMergeDLQMessagesScope:                  {operation: \"AdminMergeDLQMessages\"},\n\t\tAdminDescribeHistoryHostScope:               {operation: \"DescribeHistoryHost\"},\n\t\tAdminDescribeShardDistributionScope:         {operation: \"AdminShardList\"},\n\t\tAdminDescribeClusterScope:                   {operation: \"DescribeCluster\"},\n\t\tAdminAddSearchAttributeScope:                {operation: \"AddSearchAttribute\"},\n\t\tAdminDescribeWorkflowExecutionScope:         {operation: \"DescribeWorkflowExecution\"},\n\t\tAdminGetWorkflowExecutionRawHistoryScope:    {operation: \"GetWorkflowExecutionRawHistory\"},\n\t\tAdminGetWorkflowExecutionRawHistoryV2Scope:  {operation: \"GetWorkflowExecutionRawHistoryV2\"},\n\t\tAdminGetReplicationMessagesScope:            {operation: \"GetReplicationMessages\"},\n\t\tAdminGetDomainReplicationMessagesScope:      {operation: \"GetDomainReplicationMessages\"},\n\t\tAdminGetDLQReplicationMessagesScope:         {operation: \"AdminGetDLQReplicationMessages\"},\n\t\tAdminReapplyEventsScope:                     {operation: \"ReapplyEvents\"},\n\t\tAdminRefreshWorkflowTasksScope:              {operation: \"RefreshWorkflowTasks\"},\n\t\tAdminResendReplicationTasksScope:            {operation: \"ResendReplicationTasks\"},\n\t\tAdminGetCrossClusterTasksScope:              {operation: \"AdminGetCrossClusterTasks\"},\n\t\tAdminRespondCrossClusterTasksCompletedScope: {operation: \"AdminRespondCrossClusterTasksCompleted\"},\n\t\tAdminGetDynamicConfigScope:                  {operation: \"AdminGetDynamicConfig\"},\n\t\tAdminUpdateDynamicConfigScope:               {operation: \"AdminUpdateDynamicConfig\"},\n\t\tAdminRestoreDynamicConfigScope:              {operation: \"AdminRestoreDynamicConfig\"},\n\t\tAdminListDynamicConfigScope:                 {operation: \"AdminListDynamicConfig\"},\n\t\tAdminDeleteWorkflowScope:                    {operation: \"AdminDeleteWorkflow\"},\n\t\tGetGlobalIsolationGroups:                    {operation: \"GetGlobalIsolationGroups\"},\n\t\tUpdateGlobalIsolationGroups:                 {operation: \"UpdateGlobalIsolationGroups\"},\n\t\tGetDomainIsolationGroups:                    {operation: \"GetDomainIsolationGroups\"},\n\t\tUpdateDomainIsolationGroups:                 {operation: \"UpdateDomainIsolationGroups\"},\n\t\tGetDomainAsyncWorkflowConfiguraton:          {operation: \"GetDomainAsyncWorkflowConfiguraton\"},\n\t\tUpdateDomainAsyncWorkflowConfiguraton:       {operation: \"UpdateDomainAsyncWorkflowConfiguraton\"},\n\t\tUpdateTaskListPartitionConfig:               {operation: \"UpdateTaskListPartitionConfig\"},\n\n\t\tFrontendRestartWorkflowExecutionScope:              {operation: \"RestartWorkflowExecution\"},\n\t\tFrontendStartWorkflowExecutionScope:                {operation: \"StartWorkflowExecution\"},\n\t\tFrontendStartWorkflowExecutionAsyncScope:           {operation: \"StartWorkflowExecutionAsync\"},\n\t\tFrontendPollForDecisionTaskScope:                   {operation: \"PollForDecisionTask\"},\n\t\tFrontendPollForActivityTaskScope:                   {operation: \"PollForActivityTask\"},\n\t\tFrontendRecordActivityTaskHeartbeatScope:           {operation: \"RecordActivityTaskHeartbeat\"},\n\t\tFrontendRecordActivityTaskHeartbeatByIDScope:       {operation: \"RecordActivityTaskHeartbeatByID\"},\n\t\tFrontendRespondDecisionTaskCompletedScope:          {operation: \"RespondDecisionTaskCompleted\"},\n\t\tFrontendRespondDecisionTaskFailedScope:             {operation: \"RespondDecisionTaskFailed\"},\n\t\tFrontendRespondQueryTaskCompletedScope:             {operation: \"RespondQueryTaskCompleted\"},\n\t\tFrontendRespondActivityTaskCompletedScope:          {operation: \"RespondActivityTaskCompleted\"},\n\t\tFrontendRespondActivityTaskFailedScope:             {operation: \"RespondActivityTaskFailed\"},\n\t\tFrontendRespondActivityTaskCanceledScope:           {operation: \"RespondActivityTaskCanceled\"},\n\t\tFrontendRespondActivityTaskCompletedByIDScope:      {operation: \"RespondActivityTaskCompletedByID\"},\n\t\tFrontendRespondActivityTaskFailedByIDScope:         {operation: \"RespondActivityTaskFailedByID\"},\n\t\tFrontendRespondActivityTaskCanceledByIDScope:       {operation: \"RespondActivityTaskCanceledByID\"},\n\t\tFrontendGetWorkflowExecutionHistoryScope:           {operation: \"GetWorkflowExecutionHistory\"},\n\t\tFrontendGetWorkflowExecutionRawHistoryScope:        {operation: \"GetWorkflowExecutionRawHistory\"},\n\t\tFrontendPollForWorklfowExecutionRawHistoryScope:    {operation: \"PollForWorklfowExecutionRawHistory\"},\n\t\tFrontendSignalWorkflowExecutionScope:               {operation: \"SignalWorkflowExecution\"},\n\t\tFrontendSignalWithStartWorkflowExecutionScope:      {operation: \"SignalWithStartWorkflowExecution\"},\n\t\tFrontendSignalWithStartWorkflowExecutionAsyncScope: {operation: \"SignalWithStartWorkflowExecutionAsync\"},\n\t\tFrontendTerminateWorkflowExecutionScope:            {operation: \"TerminateWorkflowExecution\"},\n\t\tFrontendResetWorkflowExecutionScope:                {operation: \"ResetWorkflowExecution\"},\n\t\tFrontendRequestCancelWorkflowExecutionScope:        {operation: \"RequestCancelWorkflowExecution\"},\n\t\tFrontendListArchivedWorkflowExecutionsScope:        {operation: \"ListArchivedWorkflowExecutions\"},\n\t\tFrontendListOpenWorkflowExecutionsScope:            {operation: \"ListOpenWorkflowExecutions\"},\n\t\tFrontendListClosedWorkflowExecutionsScope:          {operation: \"ListClosedWorkflowExecutions\"},\n\t\tFrontendListWorkflowExecutionsScope:                {operation: \"ListWorkflowExecutions\"},\n\t\tFrontendScanWorkflowExecutionsScope:                {operation: \"ScanWorkflowExecutions\"},\n\t\tFrontendCountWorkflowExecutionsScope:               {operation: \"CountWorkflowExecutions\"},\n\t\tFrontendRegisterDomainScope:                        {operation: \"RegisterDomain\"},\n\t\tFrontendDescribeDomainScope:                        {operation: \"DescribeDomain\"},\n\t\tFrontendListDomainsScope:                           {operation: \"ListDomain\"},\n\t\tFrontendUpdateDomainScope:                          {operation: \"UpdateDomain\"},\n\t\tFrontendDeleteDomainScope:                          {operation: \"DeleteDomain\"},\n\t\tFrontendDeprecateDomainScope:                       {operation: \"DeprecateDomain\"},\n\t\tFrontendFailoverDomainScope:                        {operation: \"FailoverDomain\"},\n\t\tFrontendListFailoverHistoryScope:                   {operation: \"ListFailoverHistory\"},\n\t\tFrontendQueryWorkflowScope:                         {operation: \"QueryWorkflow\"},\n\t\tFrontendDescribeWorkflowExecutionScope:             {operation: \"DescribeWorkflowExecution\"},\n\t\tFrontendDiagnoseWorkflowExecutionScope:             {operation: \"DiagnoseWorkflowExecution\"},\n\t\tFrontendDescribeWorkflowExecutionStatusScope:       {operation: \"DescribeWorkflowExecutionStatus\"},\n\t\tFrontendListTaskListPartitionsScope:                {operation: \"FrontendListTaskListPartitions\"},\n\t\tFrontendGetTaskListsByDomainScope:                  {operation: \"FrontendGetTaskListsByDomain\"},\n\t\tFrontendRefreshWorkflowTasksScope:                  {operation: \"FrontendRefreshWorkflowTasks\"},\n\t\tFrontendDescribeTaskListScope:                      {operation: \"DescribeTaskList\"},\n\t\tFrontendResetStickyTaskListScope:                   {operation: \"ResetStickyTaskList\"},\n\t\tFrontendGetSearchAttributesScope:                   {operation: \"GetSearchAttributes\"},\n\t\tFrontendGetClusterInfoScope:                        {operation: \"GetClusterInfo\"},\n\t},\n\t// History Scope Names\n\tHistory: {\n\t\tHistoryStartWorkflowExecutionScope:                              {operation: \"StartWorkflowExecution\"},\n\t\tHistoryRecordActivityTaskHeartbeatScope:                         {operation: \"RecordActivityTaskHeartbeat\"},\n\t\tHistoryRespondDecisionTaskCompletedScope:                        {operation: \"RespondDecisionTaskCompleted\"},\n\t\tHistoryRespondDecisionTaskFailedScope:                           {operation: \"RespondDecisionTaskFailed\"},\n\t\tHistoryRespondActivityTaskCompletedScope:                        {operation: \"RespondActivityTaskCompleted\"},\n\t\tHistoryRespondActivityTaskFailedScope:                           {operation: \"RespondActivityTaskFailed\"},\n\t\tHistoryRespondActivityTaskCanceledScope:                         {operation: \"RespondActivityTaskCanceled\"},\n\t\tHistoryResetQueueScope:                                          {operation: \"ResetQueue\"},\n\t\tHistoryDescribeQueueScope:                                       {operation: \"DescribeQueue\"},\n\t\tHistoryDescribeMutabelStateScope:                                {operation: \"DescribeMutableState\"},\n\t\tHistoryGetMutableStateScope:                                     {operation: \"GetMutableState\"},\n\t\tHistoryPollMutableStateScope:                                    {operation: \"PollMutableState\"},\n\t\tHistoryResetStickyTaskListScope:                                 {operation: \"ResetStickyTaskListScope\"},\n\t\tHistoryDescribeWorkflowExecutionScope:                           {operation: \"DescribeWorkflowExecution\"},\n\t\tHistoryRecordDecisionTaskStartedScope:                           {operation: \"RecordDecisionTaskStarted\"},\n\t\tHistoryRecordActivityTaskStartedScope:                           {operation: \"RecordActivityTaskStarted\"},\n\t\tHistorySignalWorkflowExecutionScope:                             {operation: \"SignalWorkflowExecution\"},\n\t\tHistorySignalWithStartWorkflowExecutionScope:                    {operation: \"SignalWithStartWorkflowExecution\"},\n\t\tHistoryRemoveSignalMutableStateScope:                            {operation: \"RemoveSignalMutableState\"},\n\t\tHistoryTerminateWorkflowExecutionScope:                          {operation: \"TerminateWorkflowExecution\"},\n\t\tHistoryResetWorkflowExecutionScope:                              {operation: \"ResetWorkflowExecution\"},\n\t\tHistoryQueryWorkflowScope:                                       {operation: \"QueryWorkflow\"},\n\t\tHistoryProcessDeleteHistoryEventScope:                           {operation: \"ProcessDeleteHistoryEvent\"},\n\t\tHistoryScheduleDecisionTaskScope:                                {operation: \"ScheduleDecisionTask\"},\n\t\tHistoryRecordChildExecutionCompletedScope:                       {operation: \"RecordChildExecutionCompleted\"},\n\t\tHistoryRequestCancelWorkflowExecutionScope:                      {operation: \"RequestCancelWorkflowExecution\"},\n\t\tHistoryReplicateEventsScope:                                     {operation: \"ReplicateEvents\"},\n\t\tHistoryReplicateRawEventsScope:                                  {operation: \"ReplicateRawEvents\"},\n\t\tHistoryReplicateEventsV2Scope:                                   {operation: \"ReplicateEventsV2\"},\n\t\tHistorySyncShardStatusScope:                                     {operation: \"SyncShardStatus\"},\n\t\tHistorySyncActivityScope:                                        {operation: \"SyncActivity\"},\n\t\tHistoryDescribeMutableStateScope:                                {operation: \"DescribeMutableState\"},\n\t\tHistoryGetReplicationMessagesScope:                              {operation: \"GetReplicationMessages\"},\n\t\tHistoryGetDLQReplicationMessagesScope:                           {operation: \"GetDLQReplicationMessages\"},\n\t\tHistoryCountDLQMessagesScope:                                    {operation: \"CountDLQMessages\"},\n\t\tHistoryReadDLQMessagesScope:                                     {operation: \"ReadDLQMessages\"},\n\t\tHistoryPurgeDLQMessagesScope:                                    {operation: \"PurgeDLQMessages\"},\n\t\tHistoryMergeDLQMessagesScope:                                    {operation: \"MergeDLQMessages\"},\n\t\tHistoryShardControllerScope:                                     {operation: \"ShardController\"},\n\t\tHistoryReapplyEventsScope:                                       {operation: \"EventReapplication\"},\n\t\tHistoryRefreshWorkflowTasksScope:                                {operation: \"RefreshWorkflowTasks\"},\n\t\tHistoryNotifyFailoverMarkersScope:                               {operation: \"NotifyFailoverMarkers\"},\n\t\tHistoryGetCrossClusterTasksScope:                                {operation: \"GetCrossClusterTasks\"},\n\t\tHistoryRespondCrossClusterTasksCompletedScope:                   {operation: \"RespondCrossClusterTasksCompleted\"},\n\t\tHistoryGetFailoverInfoScope:                                     {operation: \"GetFailoverInfo\"},\n\t\tHistoryRatelimitUpdateScope:                                     {operation: \"RatelimitUpdate\"},\n\t\tTaskPriorityAssignerScope:                                       {operation: \"TaskPriorityAssigner\"},\n\t\tTransferQueueProcessorScope:                                     {operation: \"TransferQueueProcessor\"},\n\t\tTransferQueueProcessorV2Scope:                                   {operation: \"TransferQueueProcessorV2\"},\n\t\tTransferActiveQueueProcessorScope:                               {operation: \"TransferActiveQueueProcessor\"},\n\t\tTransferStandbyQueueProcessorScope:                              {operation: \"TransferStandbyQueueProcessor\"},\n\t\tTransferActiveTaskActivityScope:                                 {operation: \"TransferActiveTaskActivity\"},\n\t\tTransferActiveTaskDecisionScope:                                 {operation: \"TransferActiveTaskDecision\"},\n\t\tTransferActiveTaskCloseExecutionScope:                           {operation: \"TransferActiveTaskCloseExecution\"},\n\t\tTransferActiveTaskCancelExecutionScope:                          {operation: \"TransferActiveTaskCancelExecution\"},\n\t\tTransferActiveTaskSignalExecutionScope:                          {operation: \"TransferActiveTaskSignalExecution\"},\n\t\tTransferActiveTaskStartChildExecutionScope:                      {operation: \"TransferActiveTaskStartChildExecution\"},\n\t\tTransferActiveTaskRecordWorkflowStartedScope:                    {operation: \"TransferActiveTaskRecordWorkflowStarted\"},\n\t\tTransferActiveTaskResetWorkflowScope:                            {operation: \"TransferActiveTaskResetWorkflow\"},\n\t\tTransferActiveTaskUpsertWorkflowSearchAttributesScope:           {operation: \"TransferActiveTaskUpsertWorkflowSearchAttributes\"},\n\t\tTransferActiveTaskRecordWorkflowClosedScope:                     {operation: \"TransferActiveTaskRecordWorkflowClosed\"},\n\t\tTransferActiveTaskRecordChildExecutionCompletedScope:            {operation: \"TransferActiveTaskRecordChildExecutionCompleted\"},\n\t\tTransferActiveTaskApplyParentClosePolicyScope:                   {operation: \"TransferActiveTaskApplyParentClosePolicy\"},\n\t\tTransferStandbyTaskActivityScope:                                {operation: \"TransferStandbyTaskActivity\"},\n\t\tTransferStandbyTaskDecisionScope:                                {operation: \"TransferStandbyTaskDecision\"},\n\t\tTransferStandbyTaskCloseExecutionScope:                          {operation: \"TransferStandbyTaskCloseExecution\"},\n\t\tTransferStandbyTaskCancelExecutionScope:                         {operation: \"TransferStandbyTaskCancelExecution\"},\n\t\tTransferStandbyTaskSignalExecutionScope:                         {operation: \"TransferStandbyTaskSignalExecution\"},\n\t\tTransferStandbyTaskStartChildExecutionScope:                     {operation: \"TransferStandbyTaskStartChildExecution\"},\n\t\tTransferStandbyTaskRecordWorkflowStartedScope:                   {operation: \"TransferStandbyTaskRecordWorkflowStarted\"},\n\t\tTransferStandbyTaskResetWorkflowScope:                           {operation: \"TransferStandbyTaskResetWorkflow\"},\n\t\tTransferStandbyTaskUpsertWorkflowSearchAttributesScope:          {operation: \"TransferStandbyTaskUpsertWorkflowSearchAttributes\"},\n\t\tTransferStandbyTaskRecordWorkflowClosedScope:                    {operation: \"TransferStandbyTaskRecordWorkflowClosed\"},\n\t\tTransferStandbyTaskRecordChildExecutionCompletedScope:           {operation: \"TransferStandbyTaskRecordChildExecutionCompleted\"},\n\t\tTransferStandbyTaskApplyParentClosePolicyScope:                  {operation: \"TransferStandbyTaskApplyParentClosePolicy\"},\n\t\tTimerQueueProcessorScope:                                        {operation: \"TimerQueueProcessor\"},\n\t\tTimerQueueProcessorV2Scope:                                      {operation: \"TimerQueueProcessorV2\"},\n\t\tTimerActiveQueueProcessorScope:                                  {operation: \"TimerActiveQueueProcessor\"},\n\t\tTimerStandbyQueueProcessorScope:                                 {operation: \"TimerStandbyQueueProcessor\"},\n\t\tTimerActiveTaskActivityTimeoutScope:                             {operation: \"TimerActiveTaskActivityTimeout\"},\n\t\tTimerActiveTaskDecisionTimeoutScope:                             {operation: \"TimerActiveTaskDecisionTimeout\"},\n\t\tTimerActiveTaskUserTimerScope:                                   {operation: \"TimerActiveTaskUserTimer\"},\n\t\tTimerActiveTaskWorkflowTimeoutScope:                             {operation: \"TimerActiveTaskWorkflowTimeout\"},\n\t\tTimerActiveTaskActivityRetryTimerScope:                          {operation: \"TimerActiveTaskActivityRetryTimer\"},\n\t\tTimerActiveTaskWorkflowBackoffTimerScope:                        {operation: \"TimerActiveTaskWorkflowBackoffTimer\"},\n\t\tTimerActiveTaskDeleteHistoryEventScope:                          {operation: \"TimerActiveTaskDeleteHistoryEvent\"},\n\t\tTimerStandbyTaskActivityTimeoutScope:                            {operation: \"TimerStandbyTaskActivityTimeout\"},\n\t\tTimerStandbyTaskDecisionTimeoutScope:                            {operation: \"TimerStandbyTaskDecisionTimeout\"},\n\t\tTimerStandbyTaskUserTimerScope:                                  {operation: \"TimerStandbyTaskUserTimer\"},\n\t\tTimerStandbyTaskWorkflowTimeoutScope:                            {operation: \"TimerStandbyTaskWorkflowTimeout\"},\n\t\tTimerStandbyTaskActivityRetryTimerScope:                         {operation: \"TimerStandbyTaskActivityRetryTimer\"},\n\t\tTimerStandbyTaskWorkflowBackoffTimerScope:                       {operation: \"TimerStandbyTaskWorkflowBackoffTimer\"},\n\t\tTimerStandbyTaskDeleteHistoryEventScope:                         {operation: \"TimerStandbyTaskDeleteHistoryEvent\"},\n\t\tCrossClusterQueueProcessorScope:                                 {operation: \"CrossClusterQueueProcessor\"},\n\t\tCrossClusterTaskProcessorScope:                                  {operation: \"CrossClusterTaskProcessor\"},\n\t\tCrossClusterTaskFetcherScope:                                    {operation: \"CrossClusterTaskFetcher\"},\n\t\tCrossClusterSourceTaskStartChildExecutionScope:                  {operation: \"CrossClusterSourceTaskStartChildExecution\"},\n\t\tCrossClusterSourceTaskCancelExecutionScope:                      {operation: \"CrossClusterSourceTaskCancelExecution\"},\n\t\tCrossClusterSourceTaskSignalExecutionScope:                      {operation: \"CrossClusterSourceTaskSignalExecution\"},\n\t\tCrossClusterSourceTaskRecordChildWorkflowExecutionCompleteScope: {operation: \"CrossClusterSourceTaskTypeRecordChildWorkflowExecutionComplete\"},\n\t\tCrossClusterSourceTaskApplyParentClosePolicyScope:               {operation: \"CrossClusterSourceTaskTypeApplyParentClosePolicy\"},\n\t\tCrossClusterTargetTaskStartChildExecutionScope:                  {operation: \"CrossClusterTargetTaskStartChildExecution\"},\n\t\tCrossClusterTargetTaskCancelExecutionScope:                      {operation: \"CrossClusterTargetTaskCancelExecution\"},\n\t\tCrossClusterTargetTaskSignalExecutionScope:                      {operation: \"CrossClusterTargetTaskSignalExecution\"},\n\t\tCrossClusterTargetTaskRecordChildWorkflowExecutionCompleteScope: {operation: \"CrossClusterTargetTaskTypeRecordChildWorkflowExecutionComplete\"},\n\t\tCrossClusterTargetTaskApplyParentClosePolicyScope:               {operation: \"CrossClusterTargetTaskTypeApplyParentClosePolicy\"},\n\t\tHistoryEventNotificationScope:                                   {operation: \"HistoryEventNotification\"},\n\t\tReplicatorQueueProcessorScope:                                   {operation: \"ReplicatorQueueProcessor\"},\n\t\tReplicatorCacheManagerScope:                                     {operation: \"ReplicatorCacheManager\"},\n\t\tReplicatorTaskHistoryScope:                                      {operation: \"ReplicatorTaskHistory\"},\n\t\tReplicatorTaskSyncActivityScope:                                 {operation: \"ReplicatorTaskSyncActivity\"},\n\t\tReplicateHistoryEventsScope:                                     {operation: \"ReplicateHistoryEvents\"},\n\t\tReplicationMetricEmitterScope:                                   {operation: \"ReplicationMetricEmitter\"},\n\t\tShardInfoScope:                                                  {operation: \"ShardInfo\"},\n\t\tWorkflowContextScope:                                            {operation: \"WorkflowContext\"},\n\t\tHistoryCacheGetAndCreateScope:                                   {operation: \"HistoryCacheGetAndCreate\", tags: map[string]string{CacheTypeTagName: MutableStateCacheTypeTagValue}},\n\t\tHistoryCacheGetOrCreateScope:                                    {operation: \"HistoryCacheGetOrCreate\", tags: map[string]string{CacheTypeTagName: MutableStateCacheTypeTagValue}},\n\t\tHistoryCacheGetOrCreateCurrentScope:                             {operation: \"HistoryCacheGetOrCreateCurrent\", tags: map[string]string{CacheTypeTagName: MutableStateCacheTypeTagValue}},\n\t\tHistoryCacheGetCurrentExecutionScope:                            {operation: \"HistoryCacheGetCurrentExecution\", tags: map[string]string{CacheTypeTagName: MutableStateCacheTypeTagValue}},\n\t\tEventsCacheGetEventScope:                                        {operation: \"EventsCacheGetEvent\", tags: map[string]string{CacheTypeTagName: EventsCacheTypeTagValue}},\n\t\tEventsCachePutEventScope:                                        {operation: \"EventsCachePutEvent\", tags: map[string]string{CacheTypeTagName: EventsCacheTypeTagValue}},\n\t\tEventsCacheGetFromStoreScope:                                    {operation: \"EventsCacheGetFromStore\", tags: map[string]string{CacheTypeTagName: EventsCacheTypeTagValue}},\n\t\tExecutionSizeStatsScope:                                         {operation: \"ExecutionStats\", tags: map[string]string{StatsTypeTagName: SizeStatsTypeTagValue}},\n\t\tExecutionCountStatsScope:                                        {operation: \"ExecutionStats\", tags: map[string]string{StatsTypeTagName: CountStatsTypeTagValue}},\n\t\tSessionSizeStatsScope:                                           {operation: \"SessionStats\", tags: map[string]string{StatsTypeTagName: SizeStatsTypeTagValue}},\n\t\tSessionCountStatsScope:                                          {operation: \"SessionStats\", tags: map[string]string{StatsTypeTagName: CountStatsTypeTagValue}},\n\t\tWorkflowCompletionStatsScope:                                    {operation: \"CompletionStats\", tags: map[string]string{StatsTypeTagName: CountStatsTypeTagValue}},\n\t\tArchiverClientScope:                                             {operation: \"ArchiverClient\"},\n\t\tReplicationTaskFetcherScope:                                     {operation: \"ReplicationTaskFetcher\"},\n\t\tReplicationTaskCleanupScope:                                     {operation: \"ReplicationTaskCleanup\"},\n\t\tReplicationDLQStatsScope:                                        {operation: \"ReplicationDLQStats\"},\n\t\tFailoverMarkerScope:                                             {operation: \"FailoverMarker\"},\n\t\tHistoryReplicationV2TaskScope:                                   {operation: \"HistoryReplicationV2Task\"},\n\t\tSyncActivityTaskScope:                                           {operation: \"SyncActivityTask\"},\n\t\tLargeExecutionSizeShardScope:                                    {operation: \"LargeExecutionSizeShard\"},\n\t\tLargeExecutionCountShardScope:                                   {operation: \"LargeExecutionCountShard\"},\n\t\tLargeExecutionBlobShardScope:                                    {operation: \"LargeExecutionBlobShard\"},\n\t\tHistoryExecutionCacheScope:                                      {operation: \"HistoryExecutionCache\"},\n\t\tHistoryWorkflowCacheScope:                                       {operation: \"HistoryWorkflowCache\"},\n\t\tHistoryFlushBufferedEventsScope:                                 {operation: \"HistoryFlushBufferedEvents\"},\n\t\tHistoryTaskSchedulerMigrationScope:                              {operation: \"HistoryTaskSchedulerMigration\"},\n\t},\n\t// Matching Scope Names\n\tMatching: {\n\t\tMatchingPollForDecisionTaskScope:            {operation: \"PollForDecisionTask\"},\n\t\tMatchingPollForActivityTaskScope:            {operation: \"PollForActivityTask\"},\n\t\tMatchingAddActivityTaskScope:                {operation: \"AddActivityTask\"},\n\t\tMatchingAddDecisionTaskScope:                {operation: \"AddDecisionTask\"},\n\t\tMatchingAddTaskScope:                        {operation: \"AddTask\"},\n\t\tMatchingTaskListMgrScope:                    {operation: \"TaskListMgr\"},\n\t\tMatchingAdaptiveScalerScope:                 {operation: \"adaptivescaler\"},\n\t\tMatchingQueryWorkflowScope:                  {operation: \"QueryWorkflow\"},\n\t\tMatchingRespondQueryTaskCompletedScope:      {operation: \"RespondQueryTaskCompleted\"},\n\t\tMatchingCancelOutstandingPollScope:          {operation: \"CancelOutstandingPoll\"},\n\t\tMatchingDescribeTaskListScope:               {operation: \"DescribeTaskList\"},\n\t\tMatchingListTaskListPartitionsScope:         {operation: \"ListTaskListPartitions\"},\n\t\tMatchingGetTaskListsByDomainScope:           {operation: \"GetTaskListsByDomain\"},\n\t\tMatchingUpdateTaskListPartitionConfigScope:  {operation: \"UpdateTaskListPartitionConfig\"},\n\t\tMatchingRefreshTaskListPartitionConfigScope: {operation: \"RefreshTaskListPartitionConfig\"},\n\t},\n\t// Worker Scope Names\n\tWorker: {\n\t\tReplicatorScope:                        {operation: \"Replicator\"},\n\t\tDomainReplicationTaskScope:             {operation: \"DomainReplicationTask\"},\n\t\tESProcessorScope:                       {operation: \"ESProcessor\"},\n\t\tIndexProcessorScope:                    {operation: \"IndexProcessor\"},\n\t\tArchiverDeleteHistoryActivityScope:     {operation: \"ArchiverDeleteHistoryActivity\"},\n\t\tArchiverUploadHistoryActivityScope:     {operation: \"ArchiverUploadHistoryActivity\"},\n\t\tArchiverArchiveVisibilityActivityScope: {operation: \"ArchiverArchiveVisibilityActivity\"},\n\t\tArchiverScope:                          {operation: \"Archiver\"},\n\t\tArchiverPumpScope:                      {operation: \"ArchiverPump\"},\n\t\tArchiverArchivalWorkflowScope:          {operation: \"ArchiverArchivalWorkflow\"},\n\t\tTaskListScavengerScope:                 {operation: \"tasklistscavenger\"},\n\t\tExecutionsScannerScope:                 {operation: \"ExecutionsScanner\"},\n\t\tShardScannerScope:                      {operation: \"ShardScanner\"},\n\t\tCheckDataCorruptionWorkflowScope:       {operation: \"CheckDataCorruptionWorkflow\"},\n\t\tExecutionsFixerScope:                   {operation: \"ExecutionsFixer\"},\n\t\tHistoryScavengerScope:                  {operation: \"historyscavenger\"},\n\t\tBatcherScope:                           {operation: \"batcher\"},\n\t\tParentClosePolicyProcessorScope:        {operation: \"ParentClosePolicyProcessor\"},\n\t\tESAnalyzerScope:                        {operation: \"ESAnalyzer\"},\n\t\tAsyncWorkflowConsumerScope:             {operation: \"AsyncWorkflowConsumer\"},\n\t\tDiagnosticsWorkflowScope:               {operation: \"DiagnosticsWorkflow\"},\n\t},\n\tShardDistributor: {\n\t\tShardDistributorGetShardOwnerScope:                         {operation: \"GetShardOwner\"},\n\t\tShardDistributorWatchNamespaceStateScope:                   {operation: \"WatchNamespaceState\"},\n\t\tShardDistributorHeartbeatScope:                             {operation: \"ExecutorHeartbeat\"},\n\t\tShardDistributorAssignLoopScope:                            {operation: \"ShardAssignLoop\"},\n\t\tShardDistributorExecutorScope:                              {operation: \"Executor\"},\n\t\tShardDistributorStoreGetShardOwnerScope:                    {operation: \"StoreGetShardOwner\"},\n\t\tShardDistributorStoreAssignShardScope:                      {operation: \"StoreAssignShard\"},\n\t\tShardDistributorStoreAssignShardsScope:                     {operation: \"StoreAssignShards\"},\n\t\tShardDistributorStoreDeleteExecutorsScope:                  {operation: \"StoreDeleteExecutors\"},\n\t\tShardDistributorStoreGetShardStatsScope:                    {operation: \"StoreGetShardStats\"},\n\t\tShardDistributorStoreDeleteShardStatsScope:                 {operation: \"StoreDeleteShardStats\"},\n\t\tShardDistributorStoreGetHeartbeatScope:                     {operation: \"StoreGetHeartbeat\"},\n\t\tShardDistributorStoreGetExecutorScope:                      {operation: \"StoreGetExecutor\"},\n\t\tShardDistributorStoreGetStateScope:                         {operation: \"StoreGetState\"},\n\t\tShardDistributorStoreRecordHeartbeatScope:                  {operation: \"StoreRecordHeartbeat\"},\n\t\tShardDistributorStoreSubscribeToExecutorStatusChangesScope: {operation: \"StoreSubscribeToExecutorStatusChanges\"},\n\t\tShardDistributorStoreSubscribeToAssignmentChangesScope:     {operation: \"StoreSubscribeToAssignmentChanges\"},\n\t\tShardDistributorStoreDeleteAssignedStatesScope:             {operation: \"StoreDeleteAssignedStates\"},\n\t\tShardDistributorWatchScope:                                 {operation: \"Watch\"},\n\t},\n}\n\n// Common Metrics enum\nconst (\n\tCadenceRequests MetricIdx = iota\n\tCadenceFailures\n\tCadenceLatency\n\tCadenceErrBadRequestCounter\n\tCadenceErrDomainNotActiveCounter\n\tCadenceErrServiceBusyCounter\n\tCadenceErrEntityNotExistsCounter\n\tCadenceErrWorkflowExecutionAlreadyCompletedCounter\n\tCadenceErrExecutionAlreadyStartedCounter\n\tCadenceErrDomainAlreadyExistsCounter\n\tCadenceErrCancellationAlreadyRequestedCounter\n\tCadenceErrQueryFailedCounter\n\tCadenceErrLimitExceededCounter\n\tCadenceErrContextTimeoutCounter\n\tCadenceErrGRPCConnectionClosingCounter\n\tCadenceErrRetryTaskCounter\n\tCadenceErrBadBinaryCounter\n\tCadenceErrClientVersionNotSupportedCounter\n\tCadenceErrIncompleteHistoryCounter\n\tCadenceErrNonDeterministicCounter\n\tCadenceErrUnauthorizedCounter\n\tCadenceErrAuthorizeFailedCounter\n\tCadenceRequestsWithoutCallerType\n\tCadenceErrRemoteSyncMatchFailedCounter\n\tCadenceErrDomainNameExceededWarnLimit\n\tCadenceErrIdentityExceededWarnLimit\n\tCadenceErrWorkflowIDExceededWarnLimit\n\tCadenceErrSignalNameExceededWarnLimit\n\tCadenceErrWorkflowTypeExceededWarnLimit\n\tCadenceErrRequestIDExceededWarnLimit\n\tCadenceErrTaskListNameExceededWarnLimit\n\tCadenceErrActivityIDExceededWarnLimit\n\tCadenceErrActivityTypeExceededWarnLimit\n\tCadenceErrMarkerNameExceededWarnLimit\n\tCadenceErrTimerIDExceededWarnLimit\n\tPersistenceRequests\n\tPersistenceFailures\n\tPersistenceLatency\n\tPersistenceLatencyHistogram\n\tPersistenceErrShardExistsCounter\n\tPersistenceErrShardOwnershipLostCounter\n\tPersistenceErrConditionFailedCounter\n\tPersistenceErrCurrentWorkflowConditionFailedCounter\n\tPersistenceErrTimeoutCounter\n\tPersistenceErrBusyCounter\n\tPersistenceErrEntityNotExistsCounter\n\tPersistenceErrExecutionAlreadyStartedCounter\n\tPersistenceErrDomainAlreadyExistsCounter\n\tPersistenceErrBadRequestCounter\n\tPersistenceErrDuplicateRequestCounter\n\tPersistenceErrDBUnavailableCounter\n\tPersistenceSampledCounter\n\tPersistenceEmptyResponseCounter\n\tPersistenceResponseRowSize\n\tPersistenceResponsePayloadSize\n\n\tPersistenceRequestsPerDomain\n\tPersistenceRequestsPerShard\n\tPersistenceFailuresPerDomain\n\tPersistenceLatencyPerDomain\n\tPersistenceLatencyPerShard\n\tPersistenceErrShardExistsCounterPerDomain\n\tPersistenceErrShardOwnershipLostCounterPerDomain\n\tPersistenceErrConditionFailedCounterPerDomain\n\tPersistenceErrCurrentWorkflowConditionFailedCounterPerDomain\n\tPersistenceErrTimeoutCounterPerDomain\n\tPersistenceErrBusyCounterPerDomain\n\tPersistenceErrEntityNotExistsCounterPerDomain\n\tPersistenceErrExecutionAlreadyStartedCounterPerDomain\n\tPersistenceErrDomainAlreadyExistsCounterPerDomain\n\tPersistenceErrBadRequestCounterPerDomain\n\tPersistenceErrDuplicateRequestCounterPerDomain\n\tPersistenceErrDBUnavailableCounterPerDomain\n\tPersistenceSampledCounterPerDomain\n\tPersistenceEmptyResponseCounterPerDomain\n\n\tNoSQLShardStoreReadFromOriginalColumnCounter\n\tNoSQLShardStoreReadFromDataBlobCounter\n\n\tCadenceClientRequests\n\tCadenceClientFailures\n\tCadenceClientLatency\n\n\tCadenceTasklistRequests\n\n\tCadenceDcRedirectionClientRequests\n\tCadenceDcRedirectionClientFailures\n\tCadenceDcRedirectionClientLatency\n\n\tCadenceAuthorizationLatency\n\n\tDomainCachePrepareCallbacksLatency\n\tDomainCacheCallbacksLatency\n\tDomainCacheCallbacksCount\n\n\tHistorySize\n\tHistoryCount\n\tEventBlobSize\n\n\tEventBlobSizeExceedLimit\n\n\tDecisionResultCount\n\n\tArchivalConfigFailures\n\tActiveClusterGauge\n\n\tElasticsearchRequests\n\tElasticsearchFailures\n\tElasticsearchLatency\n\tElasticsearchErrBadRequestCounter\n\tElasticsearchErrBusyCounter\n\tElasticsearchRequestsPerDomain\n\tElasticsearchFailuresPerDomain\n\tElasticsearchLatencyPerDomain\n\tElasticsearchErrBadRequestCounterPerDomain\n\tElasticsearchErrBusyCounterPerDomain\n\n\tPinotRequests\n\tPinotFailures\n\tPinotLatency\n\tPinotErrBadRequestCounter\n\tPinotErrBusyCounter\n\tPinotRequestsPerDomain\n\tPinotFailuresPerDomain\n\tPinotLatencyPerDomain\n\tPinotErrBadRequestCounterPerDomain\n\tPinotErrBusyCounterPerDomain\n\n\tSequentialTaskSubmitRequest\n\tSequentialTaskSubmitRequestTaskQueueExist\n\tSequentialTaskSubmitRequestTaskQueueMissing\n\tSequentialTaskSubmitLatency\n\tSequentialTaskQueueSize\n\tSequentialTaskQueueProcessingLatency\n\tSequentialTaskTaskProcessingLatency\n\n\tParallelTaskSubmitRequest\n\tParallelTaskSubmitLatency\n\tParallelTaskTaskProcessingLatency\n\n\tPriorityTaskSubmitRequest\n\tPriorityTaskSubmitLatency\n\n\tKafkaConsumerMessageIn\n\tKafkaConsumerMessageAck\n\tKafkaConsumerMessageNack\n\tKafkaConsumerMessageNackDlqErr\n\tKafkaConsumerSessionStart\n\n\tDescribeWorkflowStatusCount\n\tDescribeWorkflowStatusError\n\n\tGracefulFailoverLatency\n\tGracefulFailoverFailure\n\n\tHistoryArchiverArchiveNonRetryableErrorCount\n\tHistoryArchiverArchiveTransientErrorCount\n\tHistoryArchiverArchiveSuccessCount\n\tHistoryArchiverHistoryMutatedCount\n\tHistoryArchiverTotalUploadSize\n\tHistoryArchiverHistorySize\n\tHistoryArchiverDuplicateArchivalsCount\n\n\t// The following metrics are only used by internal history archiver implemention.\n\t// TODO: move them to internal repo once cadence plugin model is in place.\n\tHistoryArchiverBlobExistsCount\n\tHistoryArchiverBlobSize\n\tHistoryArchiverRunningDeterministicConstructionCheckCount\n\tHistoryArchiverDeterministicConstructionCheckFailedCount\n\tHistoryArchiverRunningBlobIntegrityCheckCount\n\tHistoryArchiverBlobIntegrityCheckFailedCount\n\n\tVisibilityArchiverArchiveNonRetryableErrorCount\n\tVisibilityArchiverArchiveTransientErrorCount\n\tVisibilityArchiveSuccessCount\n\n\tMatchingClientForwardedCounter\n\tMatchingClientInvalidTaskListName\n\n\t// common metrics that are emitted per task list\n\tCadenceRequestsPerTaskList\n\tCadenceRequestsPerTaskListWithoutRollup\n\tCadenceFailuresPerTaskList\n\tCadenceLatencyPerTaskList\n\tCadenceErrBadRequestPerTaskListCounter\n\tCadenceErrDomainNotActivePerTaskListCounter\n\tCadenceErrServiceBusyPerTaskListCounter\n\tCadenceErrEntityNotExistsPerTaskListCounter\n\tCadenceErrExecutionAlreadyStartedPerTaskListCounter\n\tCadenceErrDomainAlreadyExistsPerTaskListCounter\n\tCadenceErrCancellationAlreadyRequestedPerTaskListCounter\n\tCadenceErrQueryFailedPerTaskListCounter\n\tCadenceErrLimitExceededPerTaskListCounter\n\tCadenceErrContextTimeoutPerTaskListCounter\n\tCadenceErrRetryTaskPerTaskListCounter\n\tCadenceErrBadBinaryPerTaskListCounter\n\tCadenceErrClientVersionNotSupportedPerTaskListCounter\n\tCadenceErrIncompleteHistoryPerTaskListCounter\n\tCadenceErrNonDeterministicPerTaskListCounter\n\tCadenceErrUnauthorizedPerTaskListCounter\n\tCadenceErrAuthorizeFailedPerTaskListCounter\n\tCadenceErrRemoteSyncMatchFailedPerTaskListCounter\n\tCadenceErrStickyWorkerUnavailablePerTaskListCounter\n\tCadenceErrReadOnlyPartitionPerTaskListCounter\n\tCadenceErrTaskListNotOwnedByHostPerTaskListCounter\n\n\tCadenceShardSuccessGauge\n\tCadenceShardFailureGauge\n\n\tDomainReplicationQueueSizeGauge\n\tDomainReplicationQueueSizeErrorCount\n\tDomainCacheUpdateLatency\n\n\tParentClosePolicyProcessorSuccess\n\tParentClosePolicyProcessorFailures\n\n\tValidatedWorkflowCount\n\n\tHashringViewIdentifier\n\n\tAsyncRequestPayloadSize\n\n\t// limiter-side metrics\n\tGlobalRatelimiterStartupUsageHistogram\n\tGlobalRatelimiterFailingUsageHistogram\n\tGlobalRatelimiterGlobalUsageHistogram\n\tGlobalRatelimiterUpdateLatency         // time spent performing all Update requests, per batch attempt. ideally well below update interval.\n\tGlobalRatelimiterAllowedRequestsCount  // per key/type usage\n\tGlobalRatelimiterRejectedRequestsCount // per key/type usage\n\tGlobalRatelimiterQuota                 // per-global-key quota information, emitted when a key is in us\n\n\t// aggregator-side metrics\n\tGlobalRatelimiterInitialized\n\tGlobalRatelimiterReinitialized\n\tGlobalRatelimiterUpdated\n\tGlobalRatelimiterDecayed\n\tGlobalRatelimiterLimitsQueried\n\tGlobalRatelimiterHostLimitsQueried\n\tGlobalRatelimiterRemovedLimits\n\tGlobalRatelimiterRemovedHostLimits\n\n\t// p2p rpc metrics\n\tP2PPeersCount\n\tP2PPeerAdded\n\tP2PPeerRemoved\n\t// task list partition config metrics\n\tTaskListPartitionConfigVersionGauge\n\tTaskListPartitionConfigNumReadGauge\n\tTaskListPartitionConfigNumWriteGauge\n\n\t// base cache metrics\n\tBaseCacheByteSize\n\tBaseCacheByteSizeLimitGauge\n\tBaseCacheHit\n\tBaseCacheMiss\n\tBaseCacheCount\n\tBaseCacheCountLimitGauge\n\tBaseCacheFullCounter\n\tBaseCacheEvictCounter\n\n\t// active cluster manager metrics\n\tActiveClusterManagerLookupRequestCount\n\tActiveClusterManagerLookupSuccessCount\n\tActiveClusterManagerLookupFailureCount\n\tActiveClusterManagerLookupLatency\n\n\t// cluster forwarding policy metrics\n\tClusterForwardingPolicyRequests\n\n\tRingResolverError\n\n\t// WorkflowExecutionHistoryAccess tracks the access to the workflow history\n\tWorkflowExecutionHistoryAccess\n\n\t// Budget manager metrics\n\tBudgetManagerCapacityBytes\n\tBudgetManagerCapacityCount\n\tBudgetManagerUsedBytes\n\tBudgetManagerUsedCount\n\tBudgetManagerSoftThreshold\n\tBudgetManagerActiveCacheCount\n\tBudgetManagerHardCapExceeded\n\tBudgetManagerSoftCapExceeded\n\n\tWeightedChannelPoolSizeGauge\n\n\tNumCommonMetrics // Needs to be last on this list for iota numbering\n)\n\n// History Metrics enum\nconst (\n\tTaskRequests = iota + NumCommonMetrics\n\tTaskLatency\n\tExponentialTaskLatency\n\tTaskFailures\n\tTaskDiscarded\n\tTaskAttemptTimer\n\tExponentialTaskAttemptCounts\n\tTaskStandbyRetryCounter\n\tTaskNotActiveCounter\n\tTaskLimitExceededCounter\n\tTaskBatchCompleteCounter\n\tTaskBatchCompleteFailure\n\tTaskProcessingLatency\n\tExponentialTaskProcessingLatency\n\tTaskQueueLatency\n\tExponentialTaskQueueLatency\n\tScheduleToStartHistoryQueueLatencyPerTaskList\n\tTaskRequestsOldScheduler\n\tTaskRequestsNewScheduler\n\tPendingTaskGauge\n\tReschedulerTaskCountGauge\n\tNewHistoryTaskCounter\n\n\tTaskRequestsPerDomain\n\tTaskLatencyPerDomain\n\tExponentialTaskLatencyPerDomain\n\tTaskFailuresPerDomain\n\tTaskWorkflowBusyPerDomain\n\tTaskDiscardedPerDomain\n\tTaskUnsupportedPerDomain\n\tTaskAttemptTimerPerDomain\n\tExponentialTaskAttemptCountsPerDomain\n\tTaskStandbyRetryCounterPerDomain\n\tTaskListNotOwnedByHostCounterPerDomain\n\tTaskPendingActiveCounterPerDomain\n\tTaskNotActiveCounterPerDomain\n\tTaskTargetNotActiveCounterPerDomain\n\tTaskLimitExceededCounterPerDomain\n\tTaskProcessingLatencyPerDomain\n\tExponentialTaskProcessingLatencyPerDomain\n\tTaskQueueLatencyPerDomain\n\tExponentialTaskQueueLatencyPerDomain\n\tTaskScheduleLatencyPerDomain\n\tTaskEnqueueToFetchLatency\n\tTransferTaskMissingEventCounterPerDomain\n\tReplicationTasksAppliedPerDomain\n\tWorkflowTerminateCounterPerDomain\n\tTaskSchedulerAllowedCounterPerDomain\n\tTaskSchedulerThrottledCounterPerDomain\n\n\tTaskRedispatchQueuePendingTasksTimer\n\n\tTransferTaskThrottledCounter\n\tTimerTaskThrottledCounter\n\tCrossClusterTaskThrottledCounter\n\n\tTransferTaskMissingEventCounter\n\n\tProcessingQueueNumTimer\n\tProcessingQueueMaxLevelTimer\n\tProcessingQueuePendingTaskSplitCounter\n\tProcessingQueueStuckTaskSplitCounter\n\tProcessingQueueSelectedDomainSplitCounter\n\tProcessingQueueRandomSplitCounter\n\tProcessingQueueThrottledCounter\n\tCorruptedHistoryTaskCounter\n\n\tQueueValidatorLostTaskCounter\n\tQueueValidatorDropTaskCounter\n\tQueueValidatorInvalidLoadCounter\n\tQueueValidatorValidationCounter\n\tQueueValidatorValidationFailure\n\n\tCrossClusterFetchLatency\n\tCrossClusterFetchRequests\n\tCrossClusterFetchFailures\n\tCrossClusterFetchServiceBusyFailures\n\tCrossClusterTaskRespondLatency\n\tCrossClusterTaskRespondRequests\n\tCrossClusterTaskRespondFailures\n\tCrossClusterTaskFetchedTimer\n\tCrossClusterTaskPendingTimer\n\n\tClusterMetadataFailureToResolveCounter\n\tClusterMetadataGettingMinFailoverVersionCounter\n\tClusterMetadataGettingFailoverVersionCounter\n\tClusterMetadataResolvingFailoverVersionCounter\n\tClusterMetadataResolvingMinFailoverVersionCounter\n\n\tActivityE2ELatency\n\tActivityLostCounter\n\tAckLevelUpdateCounter\n\tAckLevelUpdateFailedCounter\n\tDecisionTypeScheduleActivityCounter\n\tDecisionTypeScheduleActivityDispatchSucceedCounter\n\tDecisionTypeScheduleActivityDispatchCounter\n\tDecisionTypeCompleteWorkflowCounter\n\tDecisionTypeFailWorkflowCounter\n\tDecisionTypeCancelWorkflowCounter\n\tDecisionTypeStartTimerCounter\n\tDecisionTypeCancelActivityCounter\n\tDecisionTypeCancelTimerCounter\n\tDecisionTypeRecordMarkerCounter\n\tDecisionTypeCancelExternalWorkflowCounter\n\tDecisionTypeChildWorkflowCounter\n\tDecisionTypeContinueAsNewCounter\n\tDecisionTypeSignalExternalWorkflowCounter\n\tDecisionTypeUpsertWorkflowSearchAttributesCounter\n\tEmptyCompletionDecisionsCounter\n\tMultipleCompletionDecisionsCounter\n\tFailedDecisionsCounter\n\tDecisionAttemptTimer\n\tDecisionRetriesExceededCounter\n\tStaleMutableStateCounter\n\tDataInconsistentCounter\n\tDuplicateActivityTaskEventCounter\n\tTimerResurrectionCounter\n\tTimerProcessingDeletionTimerNoopDueToMutableStateNotLoading\n\tTimerProcessingDeletionTimerNoopDueToWFRunning\n\tActivityResurrectionCounter\n\tAutoResetPointsLimitExceededCounter\n\tAutoResetPointCorruptionCounter\n\tConcurrencyUpdateFailureCounter\n\tCadenceErrEventAlreadyStartedCounter\n\tCadenceErrShardOwnershipLostCounter\n\tHeartbeatTimeoutCounter\n\tScheduleToStartTimeoutCounter\n\tStartToCloseTimeoutCounter\n\tScheduleToCloseTimeoutCounter\n\tNewTimerCounter\n\tNewTimerNotifyCounter\n\tAcquireShardsCounter\n\tAcquireShardsLatency\n\tShardClosedCounter\n\tShardItemCreatedCounter\n\tShardItemRemovedCounter\n\tShardItemAcquisitionLatency\n\tShardInfoReplicationPendingTasksTimer\n\tShardInfoTransferActivePendingTasksTimer\n\tShardInfoTransferStandbyPendingTasksTimer\n\tShardInfoTimerActivePendingTasksTimer\n\tShardInfoTimerStandbyPendingTasksTimer\n\tShardInfoCrossClusterPendingTasksTimer\n\tShardInfoReplicationLagTimer\n\tShardInfoTransferLagTimer\n\tShardInfoTimerLagTimer\n\tShardInfoCrossClusterLagTimer\n\tShardInfoTransferDiffTimer\n\tShardInfoTimerDiffTimer\n\tShardInfoTransferFailoverInProgressTimer\n\tShardInfoTimerFailoverInProgressTimer\n\tShardInfoTransferFailoverLatencyTimer\n\tShardInfoTimerFailoverLatencyTimer\n\tSyncShardFromRemoteCounter\n\tSyncShardFromRemoteFailure\n\tMembershipChangedCounter\n\tNumShardsGauge\n\tGetEngineForShardErrorCounter\n\tGetEngineForShardLatency\n\tRemoveEngineForShardLatency\n\tCompleteDecisionWithStickyEnabledCounter\n\tCompleteDecisionWithStickyDisabledCounter\n\tDecisionHeartbeatTimeoutCounter\n\tHistoryEventNotificationQueueingLatency\n\tHistoryEventNotificationFanoutLatency\n\tHistoryEventNotificationInFlightMessageGauge\n\tHistoryEventNotificationFailDeliveryCount\n\tEmptyReplicationEventsCounter\n\tDuplicateReplicationEventsCounter\n\tStaleReplicationEventsCounter\n\tReplicationEventsSizeTimer\n\tBufferReplicationTaskTimer\n\tUnbufferReplicationTaskTimer\n\tHistoryConflictsCounter\n\tCompleteTaskFailedCounter\n\tCacheSize\n\tCacheSizeHistogram\n\tCacheRequests\n\tCacheFailures\n\tCacheLatency\n\tExponentialCacheLatency\n\tCacheHitCounter\n\tCacheMissCounter\n\tCacheFullCounter\n\tAcquireLockFailedCounter\n\tWorkflowContextCleared\n\tWorkflowContextLockLatency\n\tMutableStateSize\n\tExecutionInfoSize\n\tActivityInfoSize\n\tTimerInfoSize\n\tChildInfoSize\n\tSignalInfoSize\n\tBufferedEventsSize\n\tActivityInfoCount\n\tTimerInfoCount\n\tChildInfoCount\n\tSignalInfoCount\n\tRequestCancelInfoCount\n\tBufferedEventsCount\n\tTransferTasksCount\n\tTimerTasksCount\n\tCrossClusterTasksCount\n\tReplicationTasksCount\n\tDeleteActivityInfoCount\n\tDeleteTimerInfoCount\n\tDeleteChildInfoCount\n\tDeleteSignalInfoCount\n\tDeleteRequestCancelInfoCount\n\tWorkflowRetryBackoffTimerCount\n\tWorkflowCronBackoffTimerCount\n\tWorkflowCleanupDeleteCount\n\tWorkflowCleanupArchiveCount\n\tWorkflowCleanupNopCount\n\tWorkflowCleanupDeleteHistoryInlineCount\n\tWorkflowSuccessCount\n\tWorkflowCancelCount\n\tWorkflowFailedCount\n\tWorkflowTimeoutCount\n\tWorkflowTerminateCount\n\tWorkflowContinuedAsNew\n\tWorkflowCompletedUnknownType\n\t// WorkflowCreationFailedCleanupHaltedTimeoutCount is where the attempt to cleanup after wf start failure was halted due to a timeout making it uncertain if it's safe\n\tWorkflowCreationFailedCleanupHaltedTimeoutCount\n\t// WorkflowCreationFailedCleanupUnknownCount is where the attempt to cleanup after wf start failure was halted due to not having enough certainty\n\tWorkflowCreationFailedCleanupUnknownCount\n\t// WorkflowCreationFailedCleanupSuccessCount is where the attempt to cleanup after wf start failure was successful\n\tWorkflowCreationFailedCleanupSuccessCount\n\t// WorkflowCreationFailedCleanupFailureCount is where the attempt to cleanup after wf start failure also resulted in failure\n\tWorkflowCreationFailedCleanupFailureCount\n\tArchiverClientSendSignalCount\n\tArchiverClientSendSignalFailureCount\n\tArchiverClientHistoryRequestCount\n\tArchiverClientHistoryInlineArchiveAttemptCount\n\tArchiverClientHistoryInlineArchiveFailureCount\n\tArchiverClientHistoryInlineArchiveThrottledCount\n\tArchiverClientVisibilityRequestCount\n\tArchiverClientVisibilityInlineArchiveAttemptCount\n\tArchiverClientVisibilityInlineArchiveFailureCount\n\tArchiverClientVisibilityInlineArchiveThrottledCount\n\tArchiverClientSendSignalCountPerDomain\n\tArchiverClientSendSignalFailureCountPerDomain\n\tArchiverClientHistoryRequestCountPerDomain\n\tArchiverClientHistoryInlineArchiveAttemptCountPerDomain\n\tArchiverClientHistoryInlineArchiveFailureCountPerDomain\n\tArchiverClientHistoryInlineArchiveThrottledCountPerDomain\n\tArchiverClientVisibilityRequestCountPerDomain\n\tArchiverClientVisibilityInlineArchiveAttemptCountPerDomain\n\tArchiverClientVisibilityInlineArchiveFailureCountPerDomain\n\tArchiverClientVisibilityInlineArchiveThrottledCountPerDomain\n\tLastRetrievedMessageID\n\tLastProcessedMessageID\n\tReplicationLatency\n\tReplicationTasksApplied\n\tReplicationTasksFailed\n\tReplicationTasksLag\n\tReplicationTasksLagHistogram\n\tReplicationTasksLagRaw\n\tReplicationTasksLagRawHistogram\n\tReplicationTasksDelay\n\tReplicationTasksFetched\n\tReplicationTasksFetchedHistogram\n\tReplicationTasksReturned\n\tReplicationTasksReturnedHistogram\n\tReplicationTasksReturnedDiff\n\tReplicationTasksReturnedDiffHistogram\n\tReplicationTasksAppliedLatency\n\tReplicationTasksAppliedLatencyHistogram\n\tReplicationTasksBatchSize\n\tReplicationDynamicTaskBatchSizerDecision\n\tReplicationDLQFailed\n\tReplicationDLQMaxLevelGauge\n\tReplicationDLQAckLevelGauge\n\tReplicationDLQProbeFailed\n\tReplicationDLQSize\n\tReplicationDLQValidationFailed\n\tReplicationMessageTooLargePerShard\n\tGetReplicationMessagesForShardLatency\n\tGetDLQReplicationMessagesLatency\n\tEventReapplySkippedCount\n\tDirectQueryDispatchLatency\n\tDirectQueryDispatchStickyLatency\n\tDirectQueryDispatchNonStickyLatency\n\tDirectQueryDispatchStickySuccessCount\n\tDirectQueryDispatchNonStickySuccessCount\n\tDirectQueryDispatchClearStickinessLatency\n\tDirectQueryDispatchClearStickinessSuccessCount\n\tDirectQueryDispatchTimeoutBeforeNonStickyCount\n\tDecisionTaskQueryLatency\n\tConsistentQueryPerShard\n\tConsistentQueryTimeoutCount\n\tQueryBeforeFirstDecisionCount\n\tQueryBufferExceededCount\n\tQueryRegistryInvalidStateCount\n\tWorkerNotSupportsConsistentQueryCount\n\tDecisionStartToCloseTimeoutOverrideCount\n\tReplicationTaskCleanupCount\n\tReplicationTaskCleanupFailure\n\tReplicationTaskLatency\n\tExponentialReplicationTaskLatency\n\tExponentialReplicationTaskFetchLatency\n\tReplicationTasksFetchedSize\n\tMutableStateChecksumMismatch\n\tMutableStateChecksumInvalidated\n\tFailoverMarkerCount\n\tFailoverMarkerReplicationLatency\n\tFailoverMarkerInsertFailure\n\tFailoverMarkerNotificationFailure\n\tFailoverMarkerUpdateShardFailure\n\tFailoverMarkerCallbackCount\n\tHistoryFailoverCallbackCount\n\tWorkflowVersionCount\n\tWorkflowTypeCount\n\tWorkflowStartedCount\n\tLargeHistoryBlobCount\n\tLargeHistoryEventCount\n\tLargeHistorySizeCount\n\tUpdateWorkflowExecutionCount\n\tWorkflowIDCacheSizeGauge\n\tWorkflowIDCacheRequestsExternalRatelimitedCounter\n\tWorkflowIDCacheRequestsExternalMaxRequestsPerSecondsTimer\n\tWorkflowIDCacheRequestsInternalMaxRequestsPerSecondsTimer\n\tWorkflowIDCacheRequestsInternalRatelimitedCounter\n\tVirtualQueueCountGauge\n\tVirtualQueuePausedGauge\n\tVirtualQueueRunningGauge\n\n\tTaskRequestsPerTaskList\n\tExponentialTaskLatencyPerTaskList\n\tExponentialTaskProcessingLatencyPerTaskList\n\tExponentialTaskQueueLatencyPerTaskList\n\tExponentialTaskScheduleLatencyPerTaskList\n\n\tNumHistoryMetrics\n)\n\n// Matching metrics enum\nconst (\n\tPollSuccessPerTaskListCounter = iota + NumHistoryMetrics\n\tPollTimeoutPerTaskListCounter\n\tPollSuccessWithSyncPerTaskListCounter\n\tLeaseRequestPerTaskListCounter\n\tLeaseFailurePerTaskListCounter\n\tConditionFailedErrorPerTaskListCounter\n\tRespondQueryTaskFailedPerTaskListCounter\n\tSyncThrottlePerTaskListCounter\n\tBufferThrottlePerTaskListCounter\n\tBufferUnknownTaskDispatchError\n\tBufferIsolationGroupRedirectCounter\n\tBufferIsolationGroupRedirectFailureCounter\n\tBufferIsolationGroupMisconfiguredCounter\n\tSyncMatchLatencyPerTaskList\n\tAsyncMatchLatencyPerTaskList\n\tAsyncMatchDispatchLatencyPerTaskList\n\tAsyncMatchDispatchTimeoutCounterPerTaskList\n\tExpiredTasksPerTaskListCounter\n\tForwardedPerTaskListCounter\n\tForwardTaskCallsPerTaskList\n\tForwardTaskErrorsPerTaskList\n\tSyncMatchForwardTaskThrottleErrorPerTasklist\n\tAsyncMatchForwardTaskThrottleErrorPerTasklist\n\tForwardTaskLatencyPerTaskList\n\tForwardQueryCallsPerTaskList\n\tForwardQueryErrorsPerTaskList\n\tForwardQueryLatencyPerTaskList\n\tForwardPollCallsPerTaskList\n\tForwardPollErrorsPerTaskList\n\tForwardPollLatencyPerTaskList\n\tLocalToLocalMatchPerTaskListCounter\n\tLocalToRemoteMatchPerTaskListCounter\n\tRemoteToLocalMatchPerTaskListCounter\n\tRemoteToRemoteMatchPerTaskListCounter\n\tIsolationTaskMatchPerTaskListCounter\n\tIsolationSuccessPerTaskListCounter\n\tPollerPerTaskListCounter\n\tPollerInvalidIsolationGroupCounter\n\tTaskListPartitionUpdateFailedCounter\n\tTaskListManagersGauge\n\tTaskLagPerTaskListGauge\n\tTaskBacklogPerTaskListGauge\n\tTaskCountPerTaskListGauge\n\tRateLimitPerTaskListGauge\n\tSyncMatchLocalPollLatencyPerTaskList\n\tSyncMatchForwardPollLatencyPerTaskList\n\tAsyncMatchLocalPollCounterPerTaskList\n\tAsyncMatchLocalPollAttemptPerTaskList\n\tAsyncMatchLocalPollLatencyPerTaskList\n\tAsyncMatchForwardPollCounterPerTaskList\n\tAsyncMatchForwardPollAttemptPerTaskList\n\tAsyncMatchForwardPollLatencyPerTaskList\n\tAsyncMatchLocalPollAfterForwardFailedCounterPerTaskList\n\tAsyncMatchLocalPollAfterForwardFailedAttemptPerTaskList\n\tAsyncMatchLocalPollAfterForwardFailedLatencyPerTaskList\n\tPollLocalMatchLatencyPerTaskList\n\tPollForwardMatchLatencyPerTaskList\n\tPollLocalMatchAfterForwardFailedLatencyPerTaskList\n\tPollDecisionTaskAlreadyStartedCounterPerTaskList\n\tPollActivityTaskAlreadyStartedCounterPerTaskList\n\tTaskListReadWritePartitionMismatchGauge\n\tTaskListPollerPartitionMismatchGauge\n\tEstimatedAddTaskQPSGauge\n\tTaskListPartitionUpscaleThresholdGauge\n\tTaskListPartitionDownscaleThresholdGauge\n\tStandbyClusterTasksCompletedCounterPerTaskList\n\tStandbyClusterTasksNotStartedCounterPerTaskList\n\tStandbyClusterTasksCompletionFailurePerTaskList\n\tTaskIsolationLeakPerTaskList\n\tPartitionUpscale\n\tPartitionDownscale\n\tIsolationRebalance\n\tIsolationGroupPartitionsGauge\n\tIsolationGroupStartedPolling\n\tIsolationGroupStoppedPolling\n\tIsolationGroupUpscale\n\tIsolationGroupDownscale\n\tPartitionDrained\n\n\tNumMatchingMetrics\n)\n\n// Worker metrics enum\nconst (\n\tReplicatorMessages = iota + NumMatchingMetrics\n\tReplicatorFailures\n\tReplicatorMessagesDropped\n\tReplicatorLatency\n\tReplicatorDLQFailures\n\tESProcessorRequests\n\tESProcessorRetries\n\tESProcessorFailures\n\tESProcessorCorruptedData\n\tESProcessorProcessMsgLatency\n\tIndexProcessorCorruptedData\n\tIndexProcessorProcessMsgLatency\n\tArchiverNonRetryableErrorCount\n\tArchiverStartedCount\n\tArchiverStoppedCount\n\tArchiverCoroutineStartedCount\n\tArchiverCoroutineStoppedCount\n\tArchiverHandleHistoryRequestLatency\n\tArchiverHandleVisibilityRequestLatency\n\tArchiverUploadWithRetriesLatency\n\tArchiverDeleteWithRetriesLatency\n\tArchiverUploadFailedAllRetriesCount\n\tArchiverUploadSuccessCount\n\tArchiverDeleteFailedAllRetriesCount\n\tArchiverDeleteSuccessCount\n\tArchiverHandleVisibilityFailedAllRetiresCount\n\tArchiverHandleVisibilitySuccessCount\n\tArchiverBacklogSizeGauge\n\tArchiverPumpTimeoutCount\n\tArchiverPumpSignalThresholdCount\n\tArchiverPumpTimeoutWithoutSignalsCount\n\tArchiverPumpSignalChannelClosedCount\n\tArchiverWorkflowStartedCount\n\tArchiverNumPumpedRequestsCount\n\tArchiverNumHandledRequestsCount\n\tArchiverPumpedNotEqualHandledCount\n\tArchiverHandleAllRequestsLatency\n\tArchiverWorkflowStoppingCount\n\tTaskProcessedCount\n\tTaskDeletedCount\n\tTaskListProcessedCount\n\tTaskListDeletedCount\n\tTaskListOutstandingCount\n\tExecutionsOutstandingCount\n\tStartedCount\n\tStoppedCount\n\tExecutorTasksDeferredCount\n\tExecutorTasksDroppedCount\n\tBatcherProcessorSuccess\n\tBatcherProcessorFailures\n\tHistoryScavengerSuccessCount\n\tHistoryScavengerErrorCount\n\tHistoryScavengerSkipCount\n\tDomainReplicationEnqueueDLQCount\n\tScannerExecutionsGauge\n\tScannerCorruptedGauge\n\tScannerCheckFailedGauge\n\tScannerCorruptionByTypeGauge\n\tScannerCorruptedOpenExecutionGauge\n\tScannerShardSizeMaxGauge\n\tScannerShardSizeMedianGauge\n\tScannerShardSizeMinGauge\n\tScannerShardSizeNinetyGauge\n\tScannerShardSizeSeventyFiveGauge\n\tScannerShardSizeTwentyFiveGauge\n\tScannerShardSizeTenGauge\n\tShardScannerScan\n\tShardScannerFix\n\tDataCorruptionWorkflowCount\n\tDataCorruptionWorkflowFailure\n\tDataCorruptionWorkflowSuccessCount\n\tDataCorruptionWorkflowSkipCount\n\tESAnalyzerNumStuckWorkflowsDiscovered\n\tESAnalyzerNumStuckWorkflowsRefreshed\n\tESAnalyzerNumStuckWorkflowsFailedToRefresh\n\tESAnalyzerNumLongRunningWorkflows\n\tAsyncWorkflowConsumerCount\n\tAsyncWorkflowProcessMsgLatency\n\tAsyncWorkflowFailureCorruptMsgCount\n\tAsyncWorkflowFailureByFrontendCount\n\tAsyncWorkflowSuccessCount\n\tDiagnosticsWorkflowStartedCount\n\tDiagnosticsWorkflowSuccess\n\tDiagnosticsWorkflowExecutionLatency\n\n\tNumWorkerMetrics\n)\n\n// ShardDistributor metrics enum\nconst (\n\tShardDistributorRequests = iota + NumWorkerMetrics\n\tShardDistributorFailures\n\tShardDistributorLatency\n\tShardDistributorErrContextTimeoutCounter\n\tShardDistributorErrNamespaceNotFound\n\tShardDistributorErrShardNotFound\n\n\tShardDistributorAssignLoopNumRebalancedShards\n\tShardDistributorAssignLoopShardRebalanceLatency\n\tShardDistributorAssignLoopAttempts\n\tShardDistributorAssignLoopSuccess\n\tShardDistributorAssignLoopFail\n\n\tShardDistributorActiveShards\n\tShardDistributorTotalExecutors\n\tShardDistributorOldestExecutorHeartbeatLag\n\n\tShardDistributorStoreExecutorNotFound\n\tShardDistributorStoreFailuresPerNamespace\n\tShardDistributorStoreRequestsPerNamespace\n\tShardDistributorStoreLatencyHistogramPerNamespace\n\n\t// ShardDistributorShardAssignmentDistributionLatency measures the time taken between assignment of a shard\n\t// and the time it is fully distributed to executors\n\tShardDistributorShardAssignmentDistributionLatency\n\n\t// ShardDistributorShardHandoverLatency measures the time taken to hand over a shard from one executor to another\n\tShardDistributorShardHandoverLatency\n\n\t// ShardDistributorWatchProcessingLatency measures how long it takes to process a single WatchResponse\n\tShardDistributorWatchProcessingLatency\n\t// ShardDistributorWatchEventsReceived counts the total number of watch events received\n\tShardDistributorWatchEventsReceived\n\n\tNumShardDistributorMetrics\n)\n\n// MetricDefs record the metrics for all services\nvar MetricDefs = map[ServiceIdx]map[MetricIdx]metricDefinition{\n\tCommon: {\n\t\tCadenceRequests:                                              {metricName: \"cadence_requests\", metricType: Counter},\n\t\tCadenceFailures:                                              {metricName: \"cadence_errors\", metricType: Counter},\n\t\tCadenceLatency:                                               {metricName: \"cadence_latency\", metricType: Timer},\n\t\tCadenceErrBadRequestCounter:                                  {metricName: \"cadence_errors_bad_request\", metricType: Counter},\n\t\tCadenceErrDomainNotActiveCounter:                             {metricName: \"cadence_errors_domain_not_active\", metricType: Counter},\n\t\tCadenceErrServiceBusyCounter:                                 {metricName: \"cadence_errors_service_busy\", metricType: Counter},\n\t\tCadenceErrEntityNotExistsCounter:                             {metricName: \"cadence_errors_entity_not_exists\", metricType: Counter},\n\t\tCadenceErrWorkflowExecutionAlreadyCompletedCounter:           {metricName: \"cadence_errors_workflow_execution_already_completed\", metricType: Counter},\n\t\tCadenceErrExecutionAlreadyStartedCounter:                     {metricName: \"cadence_errors_execution_already_started\", metricType: Counter},\n\t\tCadenceErrDomainAlreadyExistsCounter:                         {metricName: \"cadence_errors_domain_already_exists\", metricType: Counter},\n\t\tCadenceErrCancellationAlreadyRequestedCounter:                {metricName: \"cadence_errors_cancellation_already_requested\", metricType: Counter},\n\t\tCadenceErrQueryFailedCounter:                                 {metricName: \"cadence_errors_query_failed\", metricType: Counter},\n\t\tCadenceErrLimitExceededCounter:                               {metricName: \"cadence_errors_limit_exceeded\", metricType: Counter},\n\t\tCadenceErrContextTimeoutCounter:                              {metricName: \"cadence_errors_context_timeout\", metricType: Counter},\n\t\tCadenceErrGRPCConnectionClosingCounter:                       {metricName: \"cadence_errors_grpc_connection_closing\", metricType: Counter},\n\t\tCadenceErrRetryTaskCounter:                                   {metricName: \"cadence_errors_retry_task\", metricType: Counter},\n\t\tCadenceErrBadBinaryCounter:                                   {metricName: \"cadence_errors_bad_binary\", metricType: Counter},\n\t\tCadenceErrClientVersionNotSupportedCounter:                   {metricName: \"cadence_errors_client_version_not_supported\", metricType: Counter},\n\t\tCadenceErrIncompleteHistoryCounter:                           {metricName: \"cadence_errors_incomplete_history\", metricType: Counter},\n\t\tCadenceErrNonDeterministicCounter:                            {metricName: \"cadence_errors_nondeterministic\", metricType: Counter},\n\t\tCadenceErrUnauthorizedCounter:                                {metricName: \"cadence_errors_unauthorized\", metricType: Counter},\n\t\tCadenceErrAuthorizeFailedCounter:                             {metricName: \"cadence_errors_authorize_failed\", metricType: Counter},\n\t\tCadenceRequestsWithoutCallerType:                             {metricName: \"cadence_requests_without_caller_type\", metricType: Counter},\n\t\tCadenceErrRemoteSyncMatchFailedCounter:                       {metricName: \"cadence_errors_remote_syncmatch_failed\", metricType: Counter},\n\t\tCadenceErrDomainNameExceededWarnLimit:                        {metricName: \"cadence_errors_domain_name_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrIdentityExceededWarnLimit:                          {metricName: \"cadence_errors_identity_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrWorkflowIDExceededWarnLimit:                        {metricName: \"cadence_errors_workflow_id_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrSignalNameExceededWarnLimit:                        {metricName: \"cadence_errors_signal_name_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrWorkflowTypeExceededWarnLimit:                      {metricName: \"cadence_errors_workflow_type_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrRequestIDExceededWarnLimit:                         {metricName: \"cadence_errors_request_id_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrTaskListNameExceededWarnLimit:                      {metricName: \"cadence_errors_task_list_name_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrActivityIDExceededWarnLimit:                        {metricName: \"cadence_errors_activity_id_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrActivityTypeExceededWarnLimit:                      {metricName: \"cadence_errors_activity_type_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrMarkerNameExceededWarnLimit:                        {metricName: \"cadence_errors_marker_name_exceeded_warn_limit\", metricType: Counter},\n\t\tCadenceErrTimerIDExceededWarnLimit:                           {metricName: \"cadence_errors_timer_id_exceeded_warn_limit\", metricType: Counter},\n\t\tPersistenceRequests:                                          {metricName: \"persistence_requests\", metricType: Counter},\n\t\tPersistenceFailures:                                          {metricName: \"persistence_errors\", metricType: Counter},\n\t\tPersistenceLatency:                                           {metricName: \"persistence_latency\", metricType: Timer},\n\t\tPersistenceLatencyHistogram:                                  {metricName: \"persistence_latency_histogram\", metricType: Histogram, buckets: PersistenceLatencyBuckets},\n\t\tPersistenceErrShardExistsCounter:                             {metricName: \"persistence_errors_shard_exists\", metricType: Counter},\n\t\tPersistenceErrShardOwnershipLostCounter:                      {metricName: \"persistence_errors_shard_ownership_lost\", metricType: Counter},\n\t\tPersistenceErrConditionFailedCounter:                         {metricName: \"persistence_errors_condition_failed\", metricType: Counter},\n\t\tPersistenceErrCurrentWorkflowConditionFailedCounter:          {metricName: \"persistence_errors_current_workflow_condition_failed\", metricType: Counter},\n\t\tPersistenceErrTimeoutCounter:                                 {metricName: \"persistence_errors_timeout\", metricType: Counter},\n\t\tPersistenceErrBusyCounter:                                    {metricName: \"persistence_errors_busy\", metricType: Counter},\n\t\tPersistenceErrEntityNotExistsCounter:                         {metricName: \"persistence_errors_entity_not_exists\", metricType: Counter},\n\t\tPersistenceErrExecutionAlreadyStartedCounter:                 {metricName: \"persistence_errors_execution_already_started\", metricType: Counter},\n\t\tPersistenceErrDomainAlreadyExistsCounter:                     {metricName: \"persistence_errors_domain_already_exists\", metricType: Counter},\n\t\tPersistenceErrBadRequestCounter:                              {metricName: \"persistence_errors_bad_request\", metricType: Counter},\n\t\tPersistenceErrDuplicateRequestCounter:                        {metricName: \"persistence_errors_duplicate_request\", metricType: Counter},\n\t\tPersistenceErrDBUnavailableCounter:                           {metricName: \"persistence_errors_db_unavailable\", metricType: Counter},\n\t\tPersistenceSampledCounter:                                    {metricName: \"persistence_sampled\", metricType: Counter},\n\t\tPersistenceEmptyResponseCounter:                              {metricName: \"persistence_empty_response\", metricType: Counter},\n\t\tPersistenceResponseRowSize:                                   {metricName: \"persistence_response_row_size\", metricType: Histogram, buckets: ResponseRowSizeBuckets},\n\t\tPersistenceResponsePayloadSize:                               {metricName: \"persistence_response_payload_size\", metricType: Histogram, buckets: ResponsePayloadSizeBuckets},\n\t\tPersistenceRequestsPerDomain:                                 {metricName: \"persistence_requests_per_domain\", metricRollupName: \"persistence_requests\", metricType: Counter},\n\t\tPersistenceRequestsPerShard:                                  {metricName: \"persistence_requests_per_shard\", metricType: Counter},\n\t\tPersistenceFailuresPerDomain:                                 {metricName: \"persistence_errors_per_domain\", metricRollupName: \"persistence_errors\", metricType: Counter},\n\t\tPersistenceLatencyPerDomain:                                  {metricName: \"persistence_latency_per_domain\", metricRollupName: \"persistence_latency\", metricType: Timer},\n\t\tPersistenceLatencyPerShard:                                   {metricName: \"persistence_latency_per_shard\", metricType: Timer},\n\t\tPersistenceErrShardExistsCounterPerDomain:                    {metricName: \"persistence_errors_shard_exists_per_domain\", metricRollupName: \"persistence_errors_shard_exists\", metricType: Counter},\n\t\tPersistenceErrShardOwnershipLostCounterPerDomain:             {metricName: \"persistence_errors_shard_ownership_lost_per_domain\", metricRollupName: \"persistence_errors_shard_ownership_lost\", metricType: Counter},\n\t\tPersistenceErrConditionFailedCounterPerDomain:                {metricName: \"persistence_errors_condition_failed_per_domain\", metricRollupName: \"persistence_errors_condition_failed\", metricType: Counter},\n\t\tPersistenceErrCurrentWorkflowConditionFailedCounterPerDomain: {metricName: \"persistence_errors_current_workflow_condition_failed_per_domain\", metricRollupName: \"persistence_errors_current_workflow_condition_failed\", metricType: Counter},\n\t\tPersistenceErrTimeoutCounterPerDomain:                        {metricName: \"persistence_errors_timeout_per_domain\", metricRollupName: \"persistence_errors_timeout\", metricType: Counter},\n\t\tPersistenceErrBusyCounterPerDomain:                           {metricName: \"persistence_errors_busy_per_domain\", metricRollupName: \"persistence_errors_busy\", metricType: Counter},\n\t\tPersistenceErrEntityNotExistsCounterPerDomain:                {metricName: \"persistence_errors_entity_not_exists_per_domain\", metricRollupName: \"persistence_errors_entity_not_exists\", metricType: Counter},\n\t\tPersistenceErrExecutionAlreadyStartedCounterPerDomain:        {metricName: \"persistence_errors_execution_already_started_per_domain\", metricRollupName: \"persistence_errors_execution_already_started\", metricType: Counter},\n\t\tPersistenceErrDomainAlreadyExistsCounterPerDomain:            {metricName: \"persistence_errors_domain_already_exists_per_domain\", metricRollupName: \"persistence_errors_domain_already_exists\", metricType: Counter},\n\t\tPersistenceErrBadRequestCounterPerDomain:                     {metricName: \"persistence_errors_bad_request_per_domain\", metricRollupName: \"persistence_errors_bad_request\", metricType: Counter},\n\t\tPersistenceErrDuplicateRequestCounterPerDomain:               {metricName: \"persistence_errors_duplicate_request_per_domain\", metricRollupName: \"persistence_errors_duplicate_request\", metricType: Counter},\n\t\tPersistenceErrDBUnavailableCounterPerDomain:                  {metricName: \"persistence_errors_db_unavailable_per_domain\", metricRollupName: \"persistence_errors_db_unavailable\", metricType: Counter},\n\t\tPersistenceSampledCounterPerDomain:                           {metricName: \"persistence_sampled_per_domain\", metricRollupName: \"persistence_sampled\", metricType: Counter},\n\t\tPersistenceEmptyResponseCounterPerDomain:                     {metricName: \"persistence_empty_response_per_domain\", metricRollupName: \"persistence_empty_response\", metricType: Counter},\n\t\tNoSQLShardStoreReadFromOriginalColumnCounter:                 {metricName: \"nosql_shard_store_read_from_original_column\", metricType: Counter},\n\t\tNoSQLShardStoreReadFromDataBlobCounter:                       {metricName: \"nosql_shard_store_read_from_data_blob\", metricType: Counter},\n\t\tCadenceClientRequests:                                        {metricName: \"cadence_client_requests\", metricType: Counter},\n\t\tCadenceClientFailures:                                        {metricName: \"cadence_client_errors\", metricType: Counter},\n\t\tCadenceClientLatency:                                         {metricName: \"cadence_client_latency\", metricType: Timer},\n\t\tCadenceTasklistRequests:                                      {metricName: \"cadence_tasklist_request\", metricType: Counter},\n\t\tCadenceDcRedirectionClientRequests:                           {metricName: \"cadence_client_requests_redirection\", metricType: Counter},\n\t\tCadenceDcRedirectionClientFailures:                           {metricName: \"cadence_client_errors_redirection\", metricType: Counter},\n\t\tCadenceDcRedirectionClientLatency:                            {metricName: \"cadence_client_latency_redirection\", metricType: Timer},\n\t\tCadenceAuthorizationLatency:                                  {metricName: \"cadence_authorization_latency\", metricType: Timer},\n\t\tDomainCachePrepareCallbacksLatency:                           {metricName: \"domain_cache_prepare_callbacks_latency\", metricType: Timer},\n\t\tDomainCacheCallbacksLatency:                                  {metricName: \"domain_cache_callbacks_latency\", metricType: Timer},\n\t\tDomainCacheCallbacksCount:                                    {metricName: \"domain_cache_callbacks_count\", metricType: Counter},\n\t\tHistorySize:                                                  {metricName: \"history_size\", metricType: Timer},\n\t\tHistoryCount:                                                 {metricName: \"history_count\", metricType: Timer},\n\t\tEventBlobSizeExceedLimit:                                     {metricName: \"blob_size_exceed_limit\", metricType: Counter},\n\t\tEventBlobSize:                                                {metricName: \"event_blob_size\", metricType: Timer},\n\t\tDecisionResultCount:                                          {metricName: \"decision_result_count\", metricType: Timer},\n\t\tArchivalConfigFailures:                                       {metricName: \"archivalconfig_failures\", metricType: Counter},\n\t\tActiveClusterGauge:                                           {metricName: \"active_cluster\", metricType: Gauge},\n\t\tElasticsearchRequests:                                        {metricName: \"elasticsearch_requests\", metricType: Counter},\n\t\tElasticsearchFailures:                                        {metricName: \"elasticsearch_errors\", metricType: Counter},\n\t\tElasticsearchLatency:                                         {metricName: \"elasticsearch_latency\", metricType: Timer},\n\t\tElasticsearchErrBadRequestCounter:                            {metricName: \"elasticsearch_errors_bad_request\", metricType: Counter},\n\t\tElasticsearchErrBusyCounter:                                  {metricName: \"elasticsearch_errors_busy\", metricType: Counter},\n\t\tElasticsearchRequestsPerDomain:                               {metricName: \"elasticsearch_requests_per_domain\", metricRollupName: \"elasticsearch_requests\", metricType: Counter},\n\t\tElasticsearchFailuresPerDomain:                               {metricName: \"elasticsearch_errors_per_domain\", metricRollupName: \"elasticsearch_errors\", metricType: Counter},\n\t\tElasticsearchLatencyPerDomain:                                {metricName: \"elasticsearch_latency_per_domain\", metricRollupName: \"elasticsearch_latency\", metricType: Timer},\n\t\tElasticsearchErrBadRequestCounterPerDomain:                   {metricName: \"elasticsearch_errors_bad_request_per_domain\", metricRollupName: \"elasticsearch_errors_bad_request\", metricType: Counter},\n\t\tElasticsearchErrBusyCounterPerDomain:                         {metricName: \"elasticsearch_errors_busy_per_domain\", metricRollupName: \"elasticsearch_errors_busy\", metricType: Counter},\n\t\tPinotRequests:                                                {metricName: \"pinot_requests\", metricType: Counter},\n\t\tPinotFailures:                                                {metricName: \"pinot_errors\", metricType: Counter},\n\t\tPinotLatency:                                                 {metricName: \"pinot_latency\", metricType: Timer},\n\t\tPinotErrBadRequestCounter:                                    {metricName: \"pinot_errors_bad_request\", metricType: Counter},\n\t\tPinotErrBusyCounter:                                          {metricName: \"pinot_errors_busy\", metricType: Counter},\n\t\tPinotRequestsPerDomain:                                       {metricName: \"pinot_requests_per_domain\", metricRollupName: \"pinot_requests\", metricType: Counter},\n\t\tPinotFailuresPerDomain:                                       {metricName: \"pinot_errors_per_domain\", metricRollupName: \"pinot_errors\", metricType: Counter},\n\t\tPinotLatencyPerDomain:                                        {metricName: \"pinot_latency_per_domain\", metricRollupName: \"pinot_latency\", metricType: Timer},\n\t\tPinotErrBadRequestCounterPerDomain:                           {metricName: \"pinot_errors_bad_request_per_domain\", metricRollupName: \"pinot_errors_bad_request\", metricType: Counter},\n\t\tPinotErrBusyCounterPerDomain:                                 {metricName: \"pinot_errors_busy_per_domain\", metricRollupName: \"pinot_errors_busy\", metricType: Counter},\n\t\tSequentialTaskSubmitRequest:                                  {metricName: \"sequentialtask_submit_request\", metricType: Counter},\n\t\tSequentialTaskSubmitRequestTaskQueueExist:                    {metricName: \"sequentialtask_submit_request_taskqueue_exist\", metricType: Counter},\n\t\tSequentialTaskSubmitRequestTaskQueueMissing:                  {metricName: \"sequentialtask_submit_request_taskqueue_missing\", metricType: Counter},\n\t\tSequentialTaskSubmitLatency:                                  {metricName: \"sequentialtask_submit_latency\", metricType: Timer},\n\t\tSequentialTaskQueueSize:                                      {metricName: \"sequentialtask_queue_size\", metricType: Timer},\n\t\tSequentialTaskQueueProcessingLatency:                         {metricName: \"sequentialtask_queue_processing_latency\", metricType: Timer},\n\t\tSequentialTaskTaskProcessingLatency:                          {metricName: \"sequentialtask_task_processing_latency\", metricType: Timer},\n\t\tParallelTaskSubmitRequest:                                    {metricName: \"paralleltask_submit_request\", metricType: Counter},\n\t\tParallelTaskSubmitLatency:                                    {metricName: \"paralleltask_submit_latency\", metricType: Timer},\n\t\tParallelTaskTaskProcessingLatency:                            {metricName: \"paralleltask_task_processing_latency\", metricType: Timer},\n\t\tPriorityTaskSubmitRequest:                                    {metricName: \"prioritytask_submit_request\", metricType: Counter},\n\t\tPriorityTaskSubmitLatency:                                    {metricName: \"prioritytask_submit_latency\", metricType: Timer},\n\t\tKafkaConsumerMessageIn:                                       {metricName: \"kafka_consumer_message_in\", metricType: Counter},\n\t\tKafkaConsumerMessageAck:                                      {metricName: \"kafka_consumer_message_ack\", metricType: Counter},\n\t\tKafkaConsumerMessageNack:                                     {metricName: \"kafka_consumer_message_nack\", metricType: Counter},\n\t\tKafkaConsumerMessageNackDlqErr:                               {metricName: \"kafka_consumer_message_nack_dlq_err\", metricType: Counter},\n\t\tKafkaConsumerSessionStart:                                    {metricName: \"kafka_consumer_session_start\", metricType: Counter},\n\t\tGracefulFailoverLatency:                                      {metricName: \"graceful_failover_latency\", metricType: Timer},\n\t\tGracefulFailoverFailure:                                      {metricName: \"graceful_failover_failures\", metricType: Counter},\n\n\t\tHistoryArchiverArchiveNonRetryableErrorCount:              {metricName: \"history_archiver_archive_non_retryable_error\", metricType: Counter},\n\t\tHistoryArchiverArchiveTransientErrorCount:                 {metricName: \"history_archiver_archive_transient_error\", metricType: Counter},\n\t\tHistoryArchiverArchiveSuccessCount:                        {metricName: \"history_archiver_archive_success\", metricType: Counter},\n\t\tHistoryArchiverHistoryMutatedCount:                        {metricName: \"history_archiver_history_mutated\", metricType: Counter},\n\t\tHistoryArchiverTotalUploadSize:                            {metricName: \"history_archiver_total_upload_size\", metricType: Timer},\n\t\tHistoryArchiverHistorySize:                                {metricName: \"history_archiver_history_size\", metricType: Timer},\n\t\tHistoryArchiverDuplicateArchivalsCount:                    {metricName: \"history_archiver_duplicate_archivals\", metricType: Counter},\n\t\tHistoryArchiverBlobExistsCount:                            {metricName: \"history_archiver_blob_exists\", metricType: Counter},\n\t\tHistoryArchiverBlobSize:                                   {metricName: \"history_archiver_blob_size\", metricType: Timer},\n\t\tHistoryArchiverRunningDeterministicConstructionCheckCount: {metricName: \"history_archiver_running_deterministic_construction_check\", metricType: Counter},\n\t\tHistoryArchiverDeterministicConstructionCheckFailedCount:  {metricName: \"history_archiver_deterministic_construction_check_failed\", metricType: Counter},\n\t\tHistoryArchiverRunningBlobIntegrityCheckCount:             {metricName: \"history_archiver_running_blob_integrity_check\", metricType: Counter},\n\t\tHistoryArchiverBlobIntegrityCheckFailedCount:              {metricName: \"history_archiver_blob_integrity_check_failed\", metricType: Counter},\n\t\tVisibilityArchiverArchiveNonRetryableErrorCount:           {metricName: \"visibility_archiver_archive_non_retryable_error\", metricType: Counter},\n\t\tVisibilityArchiverArchiveTransientErrorCount:              {metricName: \"visibility_archiver_archive_transient_error\", metricType: Counter},\n\t\tVisibilityArchiveSuccessCount:                             {metricName: \"visibility_archiver_archive_success\", metricType: Counter},\n\t\tMatchingClientForwardedCounter:                            {metricName: \"forwarded\", metricType: Counter},\n\t\tMatchingClientInvalidTaskListName:                         {metricName: \"invalid_task_list_name\", metricType: Counter},\n\n\t\t// per task list common metrics\n\t\tCadenceRequestsPerTaskList: {\n\t\t\tmetricName: \"cadence_requests_per_tl\", metricRollupName: \"cadence_requests\", metricType: Counter,\n\t\t},\n\t\tCadenceRequestsPerTaskListWithoutRollup: {\n\t\t\tmetricName: \"cadence_requests_per_tl\", metricType: Counter,\n\t\t},\n\t\tCadenceFailuresPerTaskList: {\n\t\t\tmetricName: \"cadence_errors_per_tl\", metricRollupName: \"cadence_errors\", metricType: Counter,\n\t\t},\n\t\tCadenceLatencyPerTaskList: {\n\t\t\tmetricName: \"cadence_latency_per_tl\", metricRollupName: \"cadence_latency\", metricType: Timer,\n\t\t},\n\t\tCadenceErrBadRequestPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_bad_request_per_tl\", metricRollupName: \"cadence_errors_bad_request\", metricType: Counter,\n\t\t},\n\t\tCadenceErrDomainNotActivePerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_domain_not_active_per_tl\", metricRollupName: \"cadence_errors_domain_not_active\", metricType: Counter,\n\t\t},\n\t\tCadenceErrServiceBusyPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_service_busy_per_tl\", metricRollupName: \"cadence_errors_service_busy\", metricType: Counter,\n\t\t},\n\t\tCadenceErrEntityNotExistsPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_entity_not_exists_per_tl\", metricRollupName: \"cadence_errors_entity_not_exists\", metricType: Counter,\n\t\t},\n\t\tCadenceErrExecutionAlreadyStartedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_execution_already_started_per_tl\", metricRollupName: \"cadence_errors_execution_already_started\", metricType: Counter,\n\t\t},\n\t\tCadenceErrDomainAlreadyExistsPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_domain_already_exists_per_tl\", metricRollupName: \"cadence_errors_domain_already_exists\", metricType: Counter,\n\t\t},\n\t\tCadenceErrCancellationAlreadyRequestedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_cancellation_already_requested_per_tl\", metricRollupName: \"cadence_errors_cancellation_already_requested\", metricType: Counter,\n\t\t},\n\t\tCadenceErrQueryFailedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_query_failed_per_tl\", metricRollupName: \"cadence_errors_query_failed\", metricType: Counter,\n\t\t},\n\t\tCadenceErrLimitExceededPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_limit_exceeded_per_tl\", metricRollupName: \"cadence_errors_limit_exceeded\", metricType: Counter,\n\t\t},\n\t\tCadenceErrContextTimeoutPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_context_timeout_per_tl\", metricRollupName: \"cadence_errors_context_timeout\", metricType: Counter,\n\t\t},\n\t\tCadenceErrRetryTaskPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_retry_task_per_tl\", metricRollupName: \"cadence_errors_retry_task\", metricType: Counter,\n\t\t},\n\t\tCadenceErrBadBinaryPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_bad_binary_per_tl\", metricRollupName: \"cadence_errors_bad_binary\", metricType: Counter,\n\t\t},\n\t\tCadenceErrClientVersionNotSupportedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_client_version_not_supported_per_tl\", metricRollupName: \"cadence_errors_client_version_not_supported\", metricType: Counter,\n\t\t},\n\t\tCadenceErrIncompleteHistoryPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_incomplete_history_per_tl\", metricRollupName: \"cadence_errors_incomplete_history\", metricType: Counter,\n\t\t},\n\t\tCadenceErrNonDeterministicPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_nondeterministic_per_tl\", metricRollupName: \"cadence_errors_nondeterministic\", metricType: Counter,\n\t\t},\n\t\tCadenceErrUnauthorizedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_unauthorized_per_tl\", metricRollupName: \"cadence_errors_unauthorized\", metricType: Counter,\n\t\t},\n\t\tCadenceErrAuthorizeFailedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_authorize_failed_per_tl\", metricRollupName: \"cadence_errors_authorize_failed\", metricType: Counter,\n\t\t},\n\t\tCadenceErrRemoteSyncMatchFailedPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_remote_syncmatch_failed_per_tl\", metricRollupName: \"cadence_errors_remote_syncmatch_failed\", metricType: Counter,\n\t\t},\n\t\tCadenceErrStickyWorkerUnavailablePerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_sticky_worker_unavailable_per_tl\", metricRollupName: \"cadence_errors_sticky_worker_unavailable_per_tl\", metricType: Counter,\n\t\t},\n\t\tCadenceErrReadOnlyPartitionPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_read_only_partition_per_tl\", metricRollupName: \"cadence_errors_read_only_partition\", metricType: Counter,\n\t\t},\n\t\tCadenceErrTaskListNotOwnedByHostPerTaskListCounter: {\n\t\t\tmetricName: \"cadence_errors_task_list_not_owned_by_host_per_tl\", metricRollupName: \"cadence_errors_task_list_not_owned_by_host\", metricType: Counter,\n\t\t},\n\t\tCadenceShardSuccessGauge:             {metricName: \"cadence_shard_success\", metricType: Gauge},\n\t\tCadenceShardFailureGauge:             {metricName: \"cadence_shard_failure\", metricType: Gauge},\n\t\tDomainReplicationQueueSizeGauge:      {metricName: \"domain_replication_queue_size\", metricType: Gauge},\n\t\tDomainReplicationQueueSizeErrorCount: {metricName: \"domain_replication_queue_failed\", metricType: Counter},\n\t\tDomainCacheUpdateLatency:             {metricName: \"domain_cache_update_latency\", metricType: Histogram, buckets: DomainCacheUpdateBuckets},\n\t\tParentClosePolicyProcessorSuccess:    {metricName: \"parent_close_policy_processor_requests\", metricType: Counter},\n\t\tParentClosePolicyProcessorFailures:   {metricName: \"parent_close_policy_processor_errors\", metricType: Counter},\n\n\t\tValidatedWorkflowCount:      {metricName: \"task_validator_count\", metricType: Counter},\n\t\tHashringViewIdentifier:      {metricName: \"hashring_view_identifier\", metricType: Counter},\n\t\tDescribeWorkflowStatusError: {metricName: \"describe_wf_error\", metricType: Counter},\n\t\tDescribeWorkflowStatusCount: {metricName: \"describe_wf_status\", metricType: Counter},\n\n\t\tAsyncRequestPayloadSize: {metricName: \"async_request_payload_size_per_domain\", metricRollupName: \"async_request_payload_size\", metricType: Timer},\n\n\t\tGlobalRatelimiterStartupUsageHistogram: {metricName: \"global_ratelimiter_startup_usage_histogram\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterFailingUsageHistogram: {metricName: \"global_ratelimiter_failing_usage_histogram\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterGlobalUsageHistogram:  {metricName: \"global_ratelimiter_global_usage_histogram\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterUpdateLatency:         {metricName: \"global_ratelimiter_update_latency\", metricType: Timer},\n\t\tGlobalRatelimiterAllowedRequestsCount:  {metricName: \"global_ratelimiter_allowed_requests\", metricType: Counter},\n\t\tGlobalRatelimiterRejectedRequestsCount: {metricName: \"global_ratelimiter_rejected_requests\", metricType: Counter},\n\t\tGlobalRatelimiterQuota:                 {metricName: \"global_ratelimiter_quota\", metricType: Gauge},\n\n\t\tGlobalRatelimiterInitialized:       {metricName: \"global_ratelimiter_initialized\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterReinitialized:     {metricName: \"global_ratelimiter_reinitialized\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterUpdated:           {metricName: \"global_ratelimiter_updated\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterDecayed:           {metricName: \"global_ratelimiter_decayed\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterLimitsQueried:     {metricName: \"global_ratelimiter_limits_queried\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterHostLimitsQueried: {metricName: \"global_ratelimiter_host_limits_queried\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterRemovedLimits:     {metricName: \"global_ratelimiter_removed_limits\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\t\tGlobalRatelimiterRemovedHostLimits: {metricName: \"global_ratelimiter_removed_host_limits\", metricType: Histogram, buckets: GlobalRatelimiterUsageHistogram},\n\n\t\tP2PPeersCount:                        {metricName: \"peers_count\", metricType: Gauge},\n\t\tP2PPeerAdded:                         {metricName: \"peer_added\", metricType: Counter},\n\t\tP2PPeerRemoved:                       {metricName: \"peer_removed\", metricType: Counter},\n\t\tTaskListPartitionConfigVersionGauge:  {metricName: \"task_list_partition_config_version\", metricType: Gauge},\n\t\tTaskListPartitionConfigNumReadGauge:  {metricName: \"task_list_partition_config_num_read\", metricType: Gauge},\n\t\tTaskListPartitionConfigNumWriteGauge: {metricName: \"task_list_partition_config_num_write\", metricType: Gauge},\n\n\t\tBaseCacheByteSize:           {metricName: \"cache_byte_size\", metricType: Gauge},\n\t\tBaseCacheByteSizeLimitGauge: {metricName: \"cache_byte_size_limit\", metricType: Gauge},\n\t\tBaseCacheHit:                {metricName: \"cache_hit\", metricType: Counter},\n\t\tBaseCacheMiss:               {metricName: \"cache_miss\", metricType: Counter},\n\t\tBaseCacheCount:              {metricName: \"cache_count\", metricType: Counter},\n\t\tBaseCacheCountLimitGauge:    {metricName: \"cache_count_limit\", metricType: Gauge},\n\t\tBaseCacheFullCounter:        {metricName: \"cache_full\", metricType: Counter},\n\t\tBaseCacheEvictCounter:       {metricName: \"cache_evict\", metricType: Counter},\n\n\t\tActiveClusterManagerLookupRequestCount: {metricName: \"active_cluster_manager_lookup_request_count\", metricType: Counter},\n\t\tActiveClusterManagerLookupSuccessCount: {metricName: \"active_cluster_manager_lookup_success_count\", metricType: Counter},\n\t\tActiveClusterManagerLookupFailureCount: {metricName: \"active_cluster_manager_lookup_failure_count\", metricType: Counter},\n\t\tActiveClusterManagerLookupLatency:      {metricName: \"active_cluster_manager_lookup_latency\", metricType: Histogram, buckets: ExponentialDurationBuckets},\n\n\t\tClusterForwardingPolicyRequests: {metricName: \"cluster_forwarding_policy_requests\", metricType: Counter},\n\n\t\tRingResolverError: {metricName: \"ring_resolver_error\", metricType: Counter},\n\n\t\tWorkflowExecutionHistoryAccess: {metricName: \"workflow_execution_history_access\", metricType: Gauge},\n\n\t\t// Budget manager metrics\n\t\tBudgetManagerCapacityBytes:    {metricName: \"budget_manager_capacity_bytes\", metricType: Gauge},\n\t\tBudgetManagerCapacityCount:    {metricName: \"budget_manager_capacity_count\", metricType: Gauge},\n\t\tBudgetManagerUsedBytes:        {metricName: \"budget_manager_used_bytes\", metricType: Gauge},\n\t\tBudgetManagerUsedCount:        {metricName: \"budget_manager_used_count\", metricType: Gauge},\n\t\tBudgetManagerSoftThreshold:    {metricName: \"budget_manager_soft_threshold\", metricType: Gauge},\n\t\tBudgetManagerActiveCacheCount: {metricName: \"budget_manager_active_cache_count\", metricType: Gauge},\n\t\tBudgetManagerHardCapExceeded:  {metricName: \"budget_manager_hard_cap_exceeded\", metricType: Counter},\n\t\tBudgetManagerSoftCapExceeded:  {metricName: \"budget_manager_soft_cap_exceeded\", metricType: Counter},\n\n\t\tWeightedChannelPoolSizeGauge: {metricName: \"weighted_channel_pool_size\", metricType: Gauge},\n\t},\n\tHistory: {\n\t\tTaskRequests:                                  {metricName: \"task_requests\", metricType: Counter},\n\t\tTaskLatency:                                   {metricName: \"task_latency\", metricType: Timer},\n\t\tExponentialTaskLatency:                        {metricName: \"task_latency_ns\", metricType: Histogram, exponentialBuckets: Low1ms100s},\n\t\tTaskAttemptTimer:                              {metricName: \"task_attempt\", metricType: Timer},\n\t\tExponentialTaskAttemptCounts:                  {metricName: \"task_attempt_counts\", metricType: Histogram, intExponentialBuckets: Mid1To16k},\n\t\tTaskFailures:                                  {metricName: \"task_errors\", metricType: Counter},\n\t\tTaskDiscarded:                                 {metricName: \"task_errors_discarded\", metricType: Counter},\n\t\tTaskStandbyRetryCounter:                       {metricName: \"task_errors_standby_retry_counter\", metricType: Counter},\n\t\tTaskNotActiveCounter:                          {metricName: \"task_errors_not_active_counter\", metricType: Counter},\n\t\tTaskLimitExceededCounter:                      {metricName: \"task_errors_limit_exceeded_counter\", metricType: Counter},\n\t\tTaskProcessingLatency:                         {metricName: \"task_latency_processing\", metricType: Timer},\n\t\tExponentialTaskProcessingLatency:              {metricName: \"task_latency_processing_ns\", metricType: Histogram, exponentialBuckets: Low1ms100s},\n\t\tTaskQueueLatency:                              {metricName: \"task_latency_queue\", metricType: Timer},\n\t\tExponentialTaskQueueLatency:                   {metricName: \"task_latency_queue_ns\", metricType: Histogram, exponentialBuckets: Mid1ms24h},\n\t\tScheduleToStartHistoryQueueLatencyPerTaskList: {metricName: \"schedule_to_start_history_queue_latency_per_tl\", metricType: Timer},\n\t\tTaskRequestsOldScheduler:                      {metricName: \"task_requests_old_scheduler\", metricType: Counter},\n\t\tTaskRequestsNewScheduler:                      {metricName: \"task_requests_new_scheduler\", metricType: Counter},\n\t\tPendingTaskGauge:                              {metricName: \"pending_task_gauge\", metricType: Gauge},\n\t\tReschedulerTaskCountGauge:                     {metricName: \"rescheduler_task_count\", metricType: Gauge},\n\t\tNewHistoryTaskCounter:                         {metricName: \"new_history_task_counter\", metricType: Counter},\n\n\t\t// per domain task metrics\n\n\t\tTaskRequestsPerDomain:                     {metricName: \"task_requests_per_domain\", metricRollupName: \"task_requests\", metricType: Counter},\n\t\tTaskLatencyPerDomain:                      {metricName: \"task_latency_per_domain\", metricRollupName: \"task_latency\", metricType: Timer},\n\t\tExponentialTaskLatencyPerDomain:           {metricName: \"task_latency_per_domain_ns\", metricType: Histogram, exponentialBuckets: Mid1ms24h},\n\t\tTaskAttemptTimerPerDomain:                 {metricName: \"task_attempt_per_domain\", metricRollupName: \"task_attempt\", metricType: Timer},\n\t\tExponentialTaskAttemptCountsPerDomain:     {metricName: \"task_attempt_per_domain_counts\", metricType: Histogram, intExponentialBuckets: Mid1To16k},\n\t\tTaskFailuresPerDomain:                     {metricName: \"task_errors_per_domain\", metricRollupName: \"task_errors\", metricType: Counter},\n\t\tTaskWorkflowBusyPerDomain:                 {metricName: \"task_errors_workflow_busy_per_domain\", metricRollupName: \"task_errors_workflow_busy\", metricType: Counter},\n\t\tTaskDiscardedPerDomain:                    {metricName: \"task_errors_discarded_per_domain\", metricRollupName: \"task_errors_discarded\", metricType: Counter},\n\t\tTaskUnsupportedPerDomain:                  {metricName: \"task_errors_unsupported_per_domain\", metricRollupName: \"task_errors_discarded\", metricType: Counter},\n\t\tTaskStandbyRetryCounterPerDomain:          {metricName: \"task_errors_standby_retry_counter_per_domain\", metricRollupName: \"task_errors_standby_retry_counter\", metricType: Counter},\n\t\tTaskListNotOwnedByHostCounterPerDomain:    {metricName: \"task_errors_task_list_not_owned_by_host_counter_per_domain\", metricRollupName: \"task_errors_task_list_not_owned_by_host_counter\", metricType: Counter},\n\t\tTaskPendingActiveCounterPerDomain:         {metricName: \"task_errors_pending_active_counter_per_domain\", metricRollupName: \"task_errors_pending_active_counter\", metricType: Counter},\n\t\tTaskNotActiveCounterPerDomain:             {metricName: \"task_errors_not_active_counter_per_domain\", metricRollupName: \"task_errors_not_active_counter\", metricType: Counter},\n\t\tTaskTargetNotActiveCounterPerDomain:       {metricName: \"task_errors_target_not_active_counter_per_domain\", metricRollupName: \"task_errors_target_not_active_counter\", metricType: Counter},\n\t\tTaskLimitExceededCounterPerDomain:         {metricName: \"task_errors_limit_exceeded_counter_per_domain\", metricRollupName: \"task_errors_limit_exceeded_counter\", metricType: Counter},\n\t\tTaskProcessingLatencyPerDomain:            {metricName: \"task_latency_processing_per_domain\", metricRollupName: \"task_latency_processing\", metricType: Timer},\n\t\tExponentialTaskProcessingLatencyPerDomain: {metricName: \"task_latency_processing_per_domain_ns\", metricType: Histogram, exponentialBuckets: Low1ms100s},\n\t\tTaskQueueLatencyPerDomain:                 {metricName: \"task_latency_queue_per_domain\", metricRollupName: \"task_latency_queue\", metricType: Timer},\n\t\tExponentialTaskQueueLatencyPerDomain:      {metricName: \"task_latency_queue_per_domain_ns\", metricType: Histogram, exponentialBuckets: Mid1ms24h},\n\t\tTaskScheduleLatencyPerDomain:              {metricName: \"task_latency_schedule_per_domain\", metricRollupName: \"task_latency_schedule\", metricType: Histogram, buckets: HistoryTaskLatencyBuckets},\n\t\tTaskEnqueueToFetchLatency:                 {metricName: \"task_latency_enqueue_to_fetch\", metricType: Histogram, buckets: HistoryTaskLatencyBuckets},\n\t\tTransferTaskMissingEventCounterPerDomain:  {metricName: \"transfer_task_missing_event_counter_per_domain\", metricRollupName: \"transfer_task_missing_event_counter\", metricType: Counter},\n\t\tReplicationTasksAppliedPerDomain:          {metricName: \"replication_tasks_applied_per_domain\", metricType: Counter},\n\t\tWorkflowTerminateCounterPerDomain:         {metricName: \"workflow_terminate_counter_per_domain\", metricRollupName: \"workflow_terminate_counter\", metricType: Counter},\n\t\tTaskSchedulerAllowedCounterPerDomain:      {metricName: \"task_scheduler_allowed_counter_per_domain\", metricRollupName: \"task_scheduler_allowed_counter\", metricType: Counter},\n\t\tTaskSchedulerThrottledCounterPerDomain:    {metricName: \"task_scheduler_throttled_counter_per_domain\", metricRollupName: \"task_scheduler_throttled_counter\", metricType: Counter},\n\n\t\t// per task list task metrics\n\n\t\tTaskRequestsPerTaskList:                     {metricName: \"task_requests_per_task_list\", metricType: Counter},\n\t\tExponentialTaskLatencyPerTaskList:           {metricName: \"task_latency_per_task_list_ns\", metricType: Histogram, exponentialBuckets: High1ms24h},\n\t\tExponentialTaskProcessingLatencyPerTaskList: {metricName: \"task_latency_processing_per_task_list_ns\", metricType: Histogram, exponentialBuckets: High1ms24h},\n\t\tExponentialTaskQueueLatencyPerTaskList:      {metricName: \"task_latency_queue_per_task_list_ns\", metricType: Histogram, exponentialBuckets: High1ms24h},\n\t\tExponentialTaskScheduleLatencyPerTaskList:   {metricName: \"task_latency_schedule_per_task_list_ns\", metricType: Histogram, exponentialBuckets: High1ms24h},\n\n\t\tTaskBatchCompleteCounter:                                     {metricName: \"task_batch_complete_counter\", metricType: Counter},\n\t\tTaskBatchCompleteFailure:                                     {metricName: \"task_batch_complete_error\", metricType: Counter},\n\t\tTaskRedispatchQueuePendingTasksTimer:                         {metricName: \"task_redispatch_queue_pending_tasks\", metricType: Timer},\n\t\tTransferTaskThrottledCounter:                                 {metricName: \"transfer_task_throttled_counter\", metricType: Counter},\n\t\tTimerTaskThrottledCounter:                                    {metricName: \"timer_task_throttled_counter\", metricType: Counter},\n\t\tCrossClusterTaskThrottledCounter:                             {metricName: \"cross_cluster_task_throttled_counter\", metricType: Counter},\n\t\tTransferTaskMissingEventCounter:                              {metricName: \"transfer_task_missing_event_counter\", metricType: Counter},\n\t\tProcessingQueueNumTimer:                                      {metricName: \"processing_queue_num\", metricType: Timer},\n\t\tProcessingQueueMaxLevelTimer:                                 {metricName: \"processing_queue_max_level\", metricType: Timer},\n\t\tProcessingQueuePendingTaskSplitCounter:                       {metricName: \"processing_queue_pending_task_split_counter\", metricType: Counter},\n\t\tProcessingQueueStuckTaskSplitCounter:                         {metricName: \"processing_queue_stuck_task_split_counter\", metricType: Counter},\n\t\tProcessingQueueSelectedDomainSplitCounter:                    {metricName: \"processing_queue_selected_domain_split_counter\", metricType: Counter},\n\t\tProcessingQueueRandomSplitCounter:                            {metricName: \"processing_queue_random_split_counter\", metricType: Counter},\n\t\tProcessingQueueThrottledCounter:                              {metricName: \"processing_queue_throttled_counter\", metricType: Counter},\n\t\tCorruptedHistoryTaskCounter:                                  {metricName: \"corrupted_history_task_counter\", metricType: Counter},\n\t\tQueueValidatorLostTaskCounter:                                {metricName: \"queue_validator_lost_task_counter\", metricType: Counter},\n\t\tQueueValidatorDropTaskCounter:                                {metricName: \"queue_validator_drop_task_counter\", metricType: Counter},\n\t\tQueueValidatorInvalidLoadCounter:                             {metricName: \"queue_validator_invalid_load_counter\", metricType: Counter},\n\t\tQueueValidatorValidationCounter:                              {metricName: \"queue_validator_validation_counter\", metricType: Counter},\n\t\tQueueValidatorValidationFailure:                              {metricName: \"queue_validator_validation_error\", metricType: Counter},\n\t\tCrossClusterFetchLatency:                                     {metricName: \"cross_cluster_fetch_latency\", metricType: Timer},\n\t\tCrossClusterFetchRequests:                                    {metricName: \"cross_cluster_fetch_requests\", metricType: Counter},\n\t\tCrossClusterFetchFailures:                                    {metricName: \"cross_cluster_fetch_errors\", metricType: Counter},\n\t\tCrossClusterFetchServiceBusyFailures:                         {metricName: \"cross_cluster_fetch_errors_service_busy\", metricType: Counter},\n\t\tCrossClusterTaskRespondLatency:                               {metricName: \"cross_cluster_respond_latency\", metricType: Timer},\n\t\tCrossClusterTaskRespondRequests:                              {metricName: \"cross_cluster_respond_requests\", metricType: Counter},\n\t\tCrossClusterTaskRespondFailures:                              {metricName: \"cross_cluster_fetch_errors\", metricType: Counter},\n\t\tCrossClusterTaskFetchedTimer:                                 {metricName: \"cross_cluster_task_fetched\", metricType: Timer},\n\t\tCrossClusterTaskPendingTimer:                                 {metricName: \"cross_cluster_task_pending\", metricType: Timer},\n\t\tClusterMetadataFailureToResolveCounter:                       {metricName: \"failed_to_resolve_failover_version\", metricType: Counter},\n\t\tClusterMetadataGettingMinFailoverVersionCounter:              {metricName: \"getting_min_failover_version_counter\", metricType: Counter},\n\t\tClusterMetadataGettingFailoverVersionCounter:                 {metricName: \"getting_failover_version_counter\", metricType: Counter},\n\t\tClusterMetadataResolvingFailoverVersionCounter:               {metricName: \"resolving_failover_version_counter\", metricType: Counter},\n\t\tClusterMetadataResolvingMinFailoverVersionCounter:            {metricName: \"resolving_min_failover_version_counter\", metricType: Counter},\n\t\tActivityE2ELatency:                                           {metricName: \"activity_end_to_end_latency\", metricType: Timer},\n\t\tActivityLostCounter:                                          {metricName: \"activity_lost\", metricType: Counter},\n\t\tAckLevelUpdateCounter:                                        {metricName: \"ack_level_update\", metricType: Counter},\n\t\tAckLevelUpdateFailedCounter:                                  {metricName: \"ack_level_update_failed\", metricType: Counter},\n\t\tDecisionTypeScheduleActivityCounter:                          {metricName: \"schedule_activity_decision\", metricType: Counter},\n\t\tDecisionTypeScheduleActivityDispatchSucceedCounter:           {metricName: \"schedule_activity_decision_sync_match_succeed\", metricType: Counter},\n\t\tDecisionTypeScheduleActivityDispatchCounter:                  {metricName: \"schedule_activity_decision_try_sync_match\", metricType: Counter},\n\t\tDecisionTypeCompleteWorkflowCounter:                          {metricName: \"complete_workflow_decision\", metricType: Counter},\n\t\tDecisionTypeFailWorkflowCounter:                              {metricName: \"fail_workflow_decision\", metricType: Counter},\n\t\tDecisionTypeCancelWorkflowCounter:                            {metricName: \"cancel_workflow_decision\", metricType: Counter},\n\t\tDecisionTypeStartTimerCounter:                                {metricName: \"start_timer_decision\", metricType: Counter},\n\t\tDecisionTypeCancelActivityCounter:                            {metricName: \"cancel_activity_decision\", metricType: Counter},\n\t\tDecisionTypeCancelTimerCounter:                               {metricName: \"cancel_timer_decision\", metricType: Counter},\n\t\tDecisionTypeRecordMarkerCounter:                              {metricName: \"record_marker_decision\", metricType: Counter},\n\t\tDecisionTypeCancelExternalWorkflowCounter:                    {metricName: \"cancel_external_workflow_decision\", metricType: Counter},\n\t\tDecisionTypeContinueAsNewCounter:                             {metricName: \"continue_as_new_decision\", metricType: Counter},\n\t\tDecisionTypeSignalExternalWorkflowCounter:                    {metricName: \"signal_external_workflow_decision\", metricType: Counter},\n\t\tDecisionTypeUpsertWorkflowSearchAttributesCounter:            {metricName: \"upsert_workflow_search_attributes_decision\", metricType: Counter},\n\t\tDecisionTypeChildWorkflowCounter:                             {metricName: \"child_workflow_decision\", metricType: Counter},\n\t\tEmptyCompletionDecisionsCounter:                              {metricName: \"empty_completion_decisions\", metricType: Counter},\n\t\tMultipleCompletionDecisionsCounter:                           {metricName: \"multiple_completion_decisions\", metricType: Counter},\n\t\tFailedDecisionsCounter:                                       {metricName: \"failed_decisions\", metricType: Counter},\n\t\tDecisionAttemptTimer:                                         {metricName: \"decision_attempt\", metricType: Timer},\n\t\tDecisionRetriesExceededCounter:                               {metricName: \"decision_retries_exceeded\", metricType: Counter},\n\t\tStaleMutableStateCounter:                                     {metricName: \"stale_mutable_state\", metricType: Counter},\n\t\tDataInconsistentCounter:                                      {metricName: \"data_inconsistent\", metricType: Counter},\n\t\tDuplicateActivityTaskEventCounter:                            {metricName: \"duplicate_activity_task_event\", metricType: Counter},\n\t\tTimerResurrectionCounter:                                     {metricName: \"timer_resurrection\", metricType: Counter},\n\t\tTimerProcessingDeletionTimerNoopDueToMutableStateNotLoading:  {metricName: \"timer_processing_skipping_deletion_due_to_missing_mutable_state\", metricType: Counter},\n\t\tTimerProcessingDeletionTimerNoopDueToWFRunning:               {metricName: \"timer_processing_skipping_deletion_due_to_running\", metricType: Counter},\n\t\tActivityResurrectionCounter:                                  {metricName: \"activity_resurrection\", metricType: Counter},\n\t\tAutoResetPointsLimitExceededCounter:                          {metricName: \"auto_reset_points_exceed_limit\", metricType: Counter},\n\t\tAutoResetPointCorruptionCounter:                              {metricName: \"auto_reset_point_corruption\", metricType: Counter},\n\t\tConcurrencyUpdateFailureCounter:                              {metricName: \"concurrency_update_failure\", metricType: Counter},\n\t\tCadenceErrShardOwnershipLostCounter:                          {metricName: \"cadence_errors_shard_ownership_lost\", metricType: Counter},\n\t\tCadenceErrEventAlreadyStartedCounter:                         {metricName: \"cadence_errors_event_already_started\", metricType: Counter},\n\t\tHeartbeatTimeoutCounter:                                      {metricName: \"heartbeat_timeout\", metricType: Counter},\n\t\tScheduleToStartTimeoutCounter:                                {metricName: \"schedule_to_start_timeout\", metricType: Counter},\n\t\tStartToCloseTimeoutCounter:                                   {metricName: \"start_to_close_timeout\", metricType: Counter},\n\t\tScheduleToCloseTimeoutCounter:                                {metricName: \"schedule_to_close_timeout\", metricType: Counter},\n\t\tNewTimerCounter:                                              {metricName: \"new_timer\", metricType: Counter},\n\t\tNewTimerNotifyCounter:                                        {metricName: \"new_timer_notifications\", metricType: Counter},\n\t\tAcquireShardsCounter:                                         {metricName: \"acquire_shards_count\", metricType: Counter},\n\t\tAcquireShardsLatency:                                         {metricName: \"acquire_shards_latency\", metricType: Timer},\n\t\tShardClosedCounter:                                           {metricName: \"shard_closed_count\", metricType: Counter},\n\t\tShardItemCreatedCounter:                                      {metricName: \"sharditem_created_count\", metricType: Counter},\n\t\tShardItemRemovedCounter:                                      {metricName: \"sharditem_removed_count\", metricType: Counter},\n\t\tShardItemAcquisitionLatency:                                  {metricName: \"sharditem_acquisition_latency\", metricType: Timer},\n\t\tShardInfoReplicationPendingTasksTimer:                        {metricName: \"shardinfo_replication_pending_task\", metricType: Timer},\n\t\tShardInfoTransferActivePendingTasksTimer:                     {metricName: \"shardinfo_transfer_active_pending_task\", metricType: Timer},\n\t\tShardInfoTransferStandbyPendingTasksTimer:                    {metricName: \"shardinfo_transfer_standby_pending_task\", metricType: Timer},\n\t\tShardInfoTimerActivePendingTasksTimer:                        {metricName: \"shardinfo_timer_active_pending_task\", metricType: Timer},\n\t\tShardInfoTimerStandbyPendingTasksTimer:                       {metricName: \"shardinfo_timer_standby_pending_task\", metricType: Timer},\n\t\tShardInfoCrossClusterPendingTasksTimer:                       {metricName: \"shardinfo_cross_cluster_pending_task\", metricType: Timer},\n\t\tShardInfoReplicationLagTimer:                                 {metricName: \"shardinfo_replication_lag\", metricType: Timer},\n\t\tShardInfoTransferLagTimer:                                    {metricName: \"shardinfo_transfer_lag\", metricType: Timer},\n\t\tShardInfoTimerLagTimer:                                       {metricName: \"shardinfo_timer_lag\", metricType: Timer},\n\t\tShardInfoCrossClusterLagTimer:                                {metricName: \"shardinfo_cross_cluster_lag\", metricType: Timer},\n\t\tShardInfoTransferDiffTimer:                                   {metricName: \"shardinfo_transfer_diff\", metricType: Timer},\n\t\tShardInfoTimerDiffTimer:                                      {metricName: \"shardinfo_timer_diff\", metricType: Timer},\n\t\tShardInfoTransferFailoverInProgressTimer:                     {metricName: \"shardinfo_transfer_failover_in_progress\", metricType: Timer},\n\t\tShardInfoTimerFailoverInProgressTimer:                        {metricName: \"shardinfo_timer_failover_in_progress\", metricType: Timer},\n\t\tShardInfoTransferFailoverLatencyTimer:                        {metricName: \"shardinfo_transfer_failover_latency\", metricType: Timer},\n\t\tShardInfoTimerFailoverLatencyTimer:                           {metricName: \"shardinfo_timer_failover_latency\", metricType: Timer},\n\t\tSyncShardFromRemoteCounter:                                   {metricName: \"syncshard_remote_count\", metricType: Counter},\n\t\tSyncShardFromRemoteFailure:                                   {metricName: \"syncshard_remote_failed\", metricType: Counter},\n\t\tMembershipChangedCounter:                                     {metricName: \"membership_changed_count\", metricType: Counter},\n\t\tNumShardsGauge:                                               {metricName: \"numshards_gauge\", metricType: Gauge},\n\t\tGetEngineForShardErrorCounter:                                {metricName: \"get_engine_for_shard_errors\", metricType: Counter},\n\t\tGetEngineForShardLatency:                                     {metricName: \"get_engine_for_shard_latency\", metricType: Timer},\n\t\tRemoveEngineForShardLatency:                                  {metricName: \"remove_engine_for_shard_latency\", metricType: Timer},\n\t\tCompleteDecisionWithStickyEnabledCounter:                     {metricName: \"complete_decision_sticky_enabled_count\", metricType: Counter},\n\t\tCompleteDecisionWithStickyDisabledCounter:                    {metricName: \"complete_decision_sticky_disabled_count\", metricType: Counter},\n\t\tDecisionHeartbeatTimeoutCounter:                              {metricName: \"decision_heartbeat_timeout_count\", metricType: Counter},\n\t\tHistoryEventNotificationQueueingLatency:                      {metricName: \"history_event_notification_queueing_latency\", metricType: Timer},\n\t\tHistoryEventNotificationFanoutLatency:                        {metricName: \"history_event_notification_fanout_latency\", metricType: Timer},\n\t\tHistoryEventNotificationInFlightMessageGauge:                 {metricName: \"history_event_notification_inflight_message_gauge\", metricType: Gauge},\n\t\tHistoryEventNotificationFailDeliveryCount:                    {metricName: \"history_event_notification_fail_delivery_count\", metricType: Counter},\n\t\tEmptyReplicationEventsCounter:                                {metricName: \"empty_replication_events\", metricType: Counter},\n\t\tDuplicateReplicationEventsCounter:                            {metricName: \"duplicate_replication_events\", metricType: Counter},\n\t\tStaleReplicationEventsCounter:                                {metricName: \"stale_replication_events\", metricType: Counter},\n\t\tReplicationEventsSizeTimer:                                   {metricName: \"replication_events_size\", metricType: Timer},\n\t\tBufferReplicationTaskTimer:                                   {metricName: \"buffer_replication_tasks\", metricType: Timer},\n\t\tUnbufferReplicationTaskTimer:                                 {metricName: \"unbuffer_replication_tasks\", metricType: Timer},\n\t\tHistoryConflictsCounter:                                      {metricName: \"history_conflicts\", metricType: Counter},\n\t\tCompleteTaskFailedCounter:                                    {metricName: \"complete_task_fail_count\", metricType: Counter},\n\t\tCacheSize:                                                    {metricName: \"cache_size\", metricType: Timer},\n\t\tCacheSizeHistogram:                                           {metricName: \"cache_size_counts\", metricType: Histogram, buckets: TaskCountBuckets},\n\t\tCacheRequests:                                                {metricName: \"cache_requests\", metricType: Counter},\n\t\tCacheFailures:                                                {metricName: \"cache_errors\", metricType: Counter},\n\t\tCacheLatency:                                                 {metricName: \"cache_latency\", metricType: Timer},\n\t\tExponentialCacheLatency:                                      {metricName: \"cache_latency_ns\", metricType: Histogram, exponentialBuckets: Low1ms100s},\n\t\tCacheHitCounter:                                              {metricName: \"cache_hit\", metricType: Counter},\n\t\tCacheMissCounter:                                             {metricName: \"cache_miss\", metricType: Counter},\n\t\tCacheFullCounter:                                             {metricName: \"cache_full\", metricType: Counter},\n\t\tAcquireLockFailedCounter:                                     {metricName: \"acquire_lock_failed\", metricType: Counter},\n\t\tWorkflowContextCleared:                                       {metricName: \"workflow_context_cleared\", metricType: Counter},\n\t\tWorkflowContextLockLatency:                                   {metricName: \"workflow_context_lock_latency\", metricType: Timer},\n\t\tMutableStateSize:                                             {metricName: \"mutable_state_size\", metricType: Timer},\n\t\tExecutionInfoSize:                                            {metricName: \"execution_info_size\", metricType: Timer},\n\t\tActivityInfoSize:                                             {metricName: \"activity_info_size\", metricType: Timer},\n\t\tTimerInfoSize:                                                {metricName: \"timer_info_size\", metricType: Timer},\n\t\tChildInfoSize:                                                {metricName: \"child_info_size\", metricType: Timer},\n\t\tSignalInfoSize:                                               {metricName: \"signal_info_size\", metricType: Timer},\n\t\tBufferedEventsSize:                                           {metricName: \"buffered_events_size\", metricType: Timer},\n\t\tActivityInfoCount:                                            {metricName: \"activity_info_count\", metricType: Timer},\n\t\tTimerInfoCount:                                               {metricName: \"timer_info_count\", metricType: Timer},\n\t\tChildInfoCount:                                               {metricName: \"child_info_count\", metricType: Timer},\n\t\tSignalInfoCount:                                              {metricName: \"signal_info_count\", metricType: Timer},\n\t\tRequestCancelInfoCount:                                       {metricName: \"request_cancel_info_count\", metricType: Timer},\n\t\tBufferedEventsCount:                                          {metricName: \"buffered_events_count\", metricType: Timer},\n\t\tDeleteActivityInfoCount:                                      {metricName: \"delete_activity_info\", metricType: Timer},\n\t\tDeleteTimerInfoCount:                                         {metricName: \"delete_timer_info\", metricType: Timer},\n\t\tDeleteChildInfoCount:                                         {metricName: \"delete_child_info\", metricType: Timer},\n\t\tDeleteSignalInfoCount:                                        {metricName: \"delete_signal_info\", metricType: Timer},\n\t\tDeleteRequestCancelInfoCount:                                 {metricName: \"delete_request_cancel_info\", metricType: Timer},\n\t\tWorkflowRetryBackoffTimerCount:                               {metricName: \"workflow_retry_backoff_timer\", metricType: Counter},\n\t\tWorkflowCronBackoffTimerCount:                                {metricName: \"workflow_cron_backoff_timer\", metricType: Counter},\n\t\tWorkflowCleanupDeleteCount:                                   {metricName: \"workflow_cleanup_delete\", metricType: Counter},\n\t\tWorkflowCleanupArchiveCount:                                  {metricName: \"workflow_cleanup_archive\", metricType: Counter},\n\t\tWorkflowCleanupNopCount:                                      {metricName: \"workflow_cleanup_nop\", metricType: Counter},\n\t\tWorkflowCleanupDeleteHistoryInlineCount:                      {metricName: \"workflow_cleanup_delete_history_inline\", metricType: Counter},\n\t\tWorkflowSuccessCount:                                         {metricName: \"workflow_success\", metricType: Counter},\n\t\tWorkflowCancelCount:                                          {metricName: \"workflow_cancel\", metricType: Counter},\n\t\tWorkflowFailedCount:                                          {metricName: \"workflow_failed\", metricType: Counter},\n\t\tWorkflowTimeoutCount:                                         {metricName: \"workflow_timeout\", metricType: Counter},\n\t\tWorkflowTerminateCount:                                       {metricName: \"workflow_terminate\", metricType: Counter},\n\t\tWorkflowContinuedAsNew:                                       {metricName: \"workflow_continued_as_new\", metricType: Counter},\n\t\tWorkflowCompletedUnknownType:                                 {metricName: \"workflow_completed_unknown_type\", metricType: Counter},\n\t\tWorkflowCreationFailedCleanupHaltedTimeoutCount:              {metricName: \"workflow_creation_failed_cleanup_halted_timeout_count\", metricType: Counter}, // where an attempt to cleanup after wf start failure was halted due to a timeout making it uncertain if it's safe\n\t\tWorkflowCreationFailedCleanupUnknownCount:                    {metricName: \"workflow_creation_failed_cleanup_unknown_count\", metricType: Counter},        // where an attempt to cleanup after wf start failure was halted due to not having enough certainty\n\t\tWorkflowCreationFailedCleanupSuccessCount:                    {metricName: \"workflow_creation_failed_cleanup_success_count\", metricType: Counter},        // where an attempt to cleanup after wf start failure failure was successful\n\t\tWorkflowCreationFailedCleanupFailureCount:                    {metricName: \"workflow_creation_failed_cleanup_failure_count\", metricType: Counter},        // where an attempt to cleanup in failure also resulted in failure\n\t\tArchiverClientSendSignalCount:                                {metricName: \"archiver_client_sent_signal\", metricType: Counter},\n\t\tArchiverClientSendSignalFailureCount:                         {metricName: \"archiver_client_send_signal_error\", metricType: Counter},\n\t\tArchiverClientHistoryRequestCount:                            {metricName: \"archiver_client_history_request\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveAttemptCount:               {metricName: \"archiver_client_history_inline_archive_attempt\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveFailureCount:               {metricName: \"archiver_client_history_inline_archive_failure\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveThrottledCount:             {metricName: \"archiver_client_history_inline_archive_throttled\", metricType: Counter},\n\t\tArchiverClientVisibilityRequestCount:                         {metricName: \"archiver_client_visibility_request\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveAttemptCount:            {metricName: \"archiver_client_visibility_inline_archive_attempt\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveFailureCount:            {metricName: \"archiver_client_visibility_inline_archive_failure\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveThrottledCount:          {metricName: \"archiver_client_visibility_inline_archive_throttled\", metricType: Counter},\n\t\tArchiverClientSendSignalCountPerDomain:                       {metricName: \"archiver_client_sent_signal_per_domain\", metricRollupName: \"archiver_client_sent_signal\", metricType: Counter},\n\t\tArchiverClientSendSignalFailureCountPerDomain:                {metricName: \"archiver_client_send_signal_error_per_domain\", metricRollupName: \"archiver_client_send_signal_error\", metricType: Counter},\n\t\tArchiverClientHistoryRequestCountPerDomain:                   {metricName: \"archiver_client_history_request_per_domain\", metricRollupName: \"archiver_client_history_request\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveAttemptCountPerDomain:      {metricName: \"archiver_client_history_inline_archive_attempt_per_domain\", metricRollupName: \"archiver_client_history_inline_archive_attempt\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveFailureCountPerDomain:      {metricName: \"archiver_client_history_inline_archive_failure_per_domain\", metricRollupName: \"archiver_client_history_inline_archive_failure\", metricType: Counter},\n\t\tArchiverClientHistoryInlineArchiveThrottledCountPerDomain:    {metricName: \"archiver_client_history_inline_archive_throttled_per_domain\", metricRollupName: \"archiver_client_history_inline_archive_throttled\", metricType: Counter},\n\t\tArchiverClientVisibilityRequestCountPerDomain:                {metricName: \"archiver_client_visibility_request_per_domain\", metricRollupName: \"archiver_client_visibility_request\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveAttemptCountPerDomain:   {metricName: \"archiver_client_visibility_inline_archive_attempt_per_domain\", metricRollupName: \"archiver_client_visibility_inline_archive_attempt\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveFailureCountPerDomain:   {metricName: \"archiver_client_visibility_inline_archive_failure_per_domain\", metricRollupName: \"archiver_client_visibility_inline_archive_failure\", metricType: Counter},\n\t\tArchiverClientVisibilityInlineArchiveThrottledCountPerDomain: {metricName: \"archiver_client_visibility_inline_archive_throttled_per_domain\", metricRollupName: \"archiver_client_visibility_inline_archive_throttled\", metricType: Counter},\n\t\tLastRetrievedMessageID:                                       {metricName: \"last_retrieved_message_id\", metricType: Gauge},\n\t\tLastProcessedMessageID:                                       {metricName: \"last_processed_message_id\", metricType: Gauge},\n\t\tReplicationLatency:                                           {metricName: \"replication_latency\", metricType: Gauge},\n\t\tReplicationTasksApplied:                                      {metricName: \"replication_tasks_applied\", metricType: Counter},\n\t\tReplicationTasksFailed:                                       {metricName: \"replication_tasks_failed\", metricType: Counter},\n\t\tReplicationTasksLag:                                          {metricName: \"replication_tasks_lag\", metricType: Timer},\n\t\tReplicationTasksLagHistogram:                                 {metricName: \"replication_tasks_lag_counts\", metricType: Histogram, buckets: TaskCountBuckets},\n\t\tReplicationTasksLagRaw:                                       {metricName: \"replication_tasks_lag_raw\", metricType: Timer},\n\t\tReplicationTasksLagRawHistogram:                              {metricName: \"replication_tasks_lag_raw_counts\", metricType: Histogram, buckets: TaskCountBuckets},\n\t\tReplicationTasksDelay:                                        {metricName: \"replication_tasks_delay\", metricType: Histogram, buckets: ReplicationTaskDelayBucket},\n\t\tReplicationTasksFetched:                                      {metricName: \"replication_tasks_fetched\", metricType: Timer},\n\t\tReplicationTasksFetchedHistogram:                             {metricName: \"replication_tasks_fetched_counts\", metricType: Histogram, buckets: ResponseRowSizeBuckets},\n\t\tReplicationTasksReturned:                                     {metricName: \"replication_tasks_returned\", metricType: Timer},\n\t\tReplicationTasksReturnedHistogram:                            {metricName: \"replication_tasks_returned_counts\", metricType: Histogram, buckets: ResponseRowSizeBuckets},\n\t\tReplicationTasksReturnedDiff:                                 {metricName: \"replication_tasks_returned_diff\", metricType: Timer},\n\t\tReplicationTasksReturnedDiffHistogram:                        {metricName: \"replication_tasks_returned_diff_counts\", metricType: Histogram, buckets: ResponseRowSizeBuckets},\n\t\tReplicationTasksAppliedLatency:                               {metricName: \"replication_tasks_applied_latency\", metricType: Timer},\n\t\tReplicationTasksAppliedLatencyHistogram:                      {metricName: \"replication_tasks_applied_latency_ns\", metricType: Histogram, exponentialBuckets: Low1ms100s},\n\t\tReplicationTasksBatchSize:                                    {metricName: \"replication_tasks_batch_size\", metricType: Gauge},\n\t\tReplicationDynamicTaskBatchSizerDecision:                     {metricName: \"replication_dynamic_task_batch_sizer_decision\", metricType: Counter},\n\t\tReplicationDLQFailed:                                         {metricName: \"replication_dlq_enqueue_failed\", metricType: Counter},\n\t\tReplicationDLQMaxLevelGauge:                                  {metricName: \"replication_dlq_max_level\", metricType: Gauge},\n\t\tReplicationDLQAckLevelGauge:                                  {metricName: \"replication_dlq_ack_level\", metricType: Gauge},\n\t\tReplicationDLQProbeFailed:                                    {metricName: \"replication_dlq_probe_failed\", metricType: Counter},\n\t\tReplicationDLQSize:                                           {metricName: \"replication_dlq_size\", metricType: Gauge},\n\t\tReplicationDLQValidationFailed:                               {metricName: \"replication_dlq_validation_failed\", metricType: Counter},\n\t\tReplicationMessageTooLargePerShard:                           {metricName: \"replication_message_too_large_per_shard\", metricType: Counter},\n\t\tGetReplicationMessagesForShardLatency:                        {metricName: \"get_replication_messages_for_shard\", metricType: Timer},\n\t\tGetDLQReplicationMessagesLatency:                             {metricName: \"get_dlq_replication_messages\", metricType: Timer},\n\t\tEventReapplySkippedCount:                                     {metricName: \"event_reapply_skipped_count\", metricType: Counter},\n\t\tDirectQueryDispatchLatency:                                   {metricName: \"direct_query_dispatch_latency\", metricType: Timer},\n\t\tDirectQueryDispatchStickyLatency:                             {metricName: \"direct_query_dispatch_sticky_latency\", metricType: Timer},\n\t\tDirectQueryDispatchNonStickyLatency:                          {metricName: \"direct_query_dispatch_non_sticky_latency\", metricType: Timer},\n\t\tDirectQueryDispatchStickySuccessCount:                        {metricName: \"direct_query_dispatch_sticky_success\", metricType: Counter},\n\t\tDirectQueryDispatchNonStickySuccessCount:                     {metricName: \"direct_query_dispatch_non_sticky_success\", metricType: Counter},\n\t\tDirectQueryDispatchClearStickinessLatency:                    {metricName: \"direct_query_dispatch_clear_stickiness_latency\", metricType: Timer},\n\t\tDirectQueryDispatchClearStickinessSuccessCount:               {metricName: \"direct_query_dispatch_clear_stickiness_success\", metricType: Counter},\n\t\tDirectQueryDispatchTimeoutBeforeNonStickyCount:               {metricName: \"direct_query_dispatch_timeout_before_non_sticky\", metricType: Counter},\n\t\tDecisionTaskQueryLatency:                                     {metricName: \"decision_task_query_latency\", metricType: Timer},\n\t\tConsistentQueryPerShard:                                      {metricName: \"consistent_query_per_shard\", metricType: Counter},\n\t\tConsistentQueryTimeoutCount:                                  {metricName: \"consistent_query_timeout\", metricType: Counter},\n\t\tQueryBeforeFirstDecisionCount:                                {metricName: \"query_before_first_decision\", metricType: Counter},\n\t\tQueryBufferExceededCount:                                     {metricName: \"query_buffer_exceeded\", metricType: Counter},\n\t\tQueryRegistryInvalidStateCount:                               {metricName: \"query_registry_invalid_state\", metricType: Counter},\n\t\tWorkerNotSupportsConsistentQueryCount:                        {metricName: \"worker_not_supports_consistent_query\", metricType: Counter},\n\t\tDecisionStartToCloseTimeoutOverrideCount:                     {metricName: \"decision_start_to_close_timeout_overrides\", metricType: Counter},\n\t\tReplicationTaskCleanupCount:                                  {metricName: \"replication_task_cleanup_count\", metricType: Counter},\n\t\tReplicationTaskCleanupFailure:                                {metricName: \"replication_task_cleanup_failed\", metricType: Counter},\n\t\tReplicationTaskLatency:                                       {metricName: \"replication_task_latency\", metricType: Timer},\n\t\tExponentialReplicationTaskLatency:                            {metricName: \"replication_task_latency_ns\", metricType: Histogram, exponentialBuckets: Mid1ms24h},\n\t\tExponentialReplicationTaskFetchLatency:                       {metricName: \"replication_task_fetch_latency_ns\", metricType: Histogram, exponentialBuckets: Mid1ms24h},\n\t\tReplicationTasksFetchedSize:                                  {metricName: \"replication_tasks_fetched_size\", metricType: Histogram, buckets: ResponseRowSizeBuckets},\n\t\tMutableStateChecksumMismatch:                                 {metricName: \"mutable_state_checksum_mismatch\", metricType: Counter},\n\t\tMutableStateChecksumInvalidated:                              {metricName: \"mutable_state_checksum_invalidated\", metricType: Counter},\n\t\tFailoverMarkerCount:                                          {metricName: \"failover_marker_count\", metricType: Counter},\n\t\tFailoverMarkerReplicationLatency:                             {metricName: \"failover_marker_replication_latency\", metricType: Timer},\n\t\tFailoverMarkerInsertFailure:                                  {metricName: \"failover_marker_insert_failures\", metricType: Counter},\n\t\tFailoverMarkerNotificationFailure:                            {metricName: \"failover_marker_notification_failures\", metricType: Counter},\n\t\tFailoverMarkerUpdateShardFailure:                             {metricName: \"failover_marker_update_shard_failures\", metricType: Counter},\n\t\tFailoverMarkerCallbackCount:                                  {metricName: \"failover_marker_callback_count\", metricType: Counter},\n\t\tHistoryFailoverCallbackCount:                                 {metricName: \"failover_callback_handler_count\", metricType: Counter},\n\t\tTransferTasksCount:                                           {metricName: \"transfer_tasks_count\", metricType: Timer},\n\t\tTimerTasksCount:                                              {metricName: \"timer_tasks_count\", metricType: Timer},\n\t\tCrossClusterTasksCount:                                       {metricName: \"cross_cluster_tasks_count\", metricType: Timer},\n\t\tReplicationTasksCount:                                        {metricName: \"replication_tasks_count\", metricType: Timer},\n\t\tWorkflowVersionCount:                                         {metricName: \"workflow_version_count\", metricType: Gauge},\n\t\tWorkflowTypeCount:                                            {metricName: \"workflow_type_count\", metricType: Gauge},\n\t\tWorkflowStartedCount:                                         {metricName: \"workflow_started_count\", metricType: Counter},\n\t\tLargeHistoryBlobCount:                                        {metricName: \"large_history_blob_count\", metricType: Counter},\n\t\tLargeHistoryEventCount:                                       {metricName: \"large_history_event_count\", metricType: Counter},\n\t\tLargeHistorySizeCount:                                        {metricName: \"large_history_size_count\", metricType: Counter},\n\t\tUpdateWorkflowExecutionCount:                                 {metricName: \"update_workflow_execution_count\", metricType: Counter},\n\t\tWorkflowIDCacheSizeGauge:                                     {metricName: \"workflow_id_cache_size\", metricType: Gauge},\n\t\tWorkflowIDCacheRequestsExternalRatelimitedCounter:            {metricName: \"workflow_id_external_requests_ratelimited\", metricType: Counter},\n\t\tWorkflowIDCacheRequestsExternalMaxRequestsPerSecondsTimer:    {metricName: \"workflow_id_external_requests_max_requests_per_seconds\", metricType: Timer},\n\t\tWorkflowIDCacheRequestsInternalMaxRequestsPerSecondsTimer:    {metricName: \"workflow_id_internal_requests_max_requests_per_seconds\", metricType: Timer},\n\t\tWorkflowIDCacheRequestsInternalRatelimitedCounter:            {metricName: \"workflow_id_internal_requests_ratelimited\", metricType: Counter},\n\t\tVirtualQueueCountGauge:                                       {metricName: \"virtual_queue_count\", metricType: Gauge},\n\t\tVirtualQueuePausedGauge:                                      {metricName: \"virtual_queue_paused\", metricType: Gauge},\n\t\tVirtualQueueRunningGauge:                                     {metricName: \"virtual_queue_running\", metricType: Gauge},\n\t},\n\tMatching: {\n\t\tPollSuccessPerTaskListCounter:                           {metricName: \"poll_success_per_tl\", metricRollupName: \"poll_success\"},\n\t\tPollTimeoutPerTaskListCounter:                           {metricName: \"poll_timeouts_per_tl\", metricRollupName: \"poll_timeouts\"},\n\t\tPollSuccessWithSyncPerTaskListCounter:                   {metricName: \"poll_success_sync_per_tl\", metricRollupName: \"poll_success_sync\"},\n\t\tLeaseRequestPerTaskListCounter:                          {metricName: \"lease_requests_per_tl\", metricRollupName: \"lease_requests\"},\n\t\tLeaseFailurePerTaskListCounter:                          {metricName: \"lease_failures_per_tl\", metricRollupName: \"lease_failures\"},\n\t\tConditionFailedErrorPerTaskListCounter:                  {metricName: \"condition_failed_errors_per_tl\", metricRollupName: \"condition_failed_errors\"},\n\t\tRespondQueryTaskFailedPerTaskListCounter:                {metricName: \"respond_query_failed_per_tl\", metricRollupName: \"respond_query_failed\"},\n\t\tSyncThrottlePerTaskListCounter:                          {metricName: \"sync_throttle_count_per_tl\", metricRollupName: \"sync_throttle_count\"},\n\t\tBufferThrottlePerTaskListCounter:                        {metricName: \"buffer_throttle_count_per_tl\", metricRollupName: \"buffer_throttle_count\"},\n\t\tBufferUnknownTaskDispatchError:                          {metricName: \"buffer_unknown_task_dispatch_error_per_tl\", metricRollupName: \"buffer_unknown_task_dispatch_error\"},\n\t\tBufferIsolationGroupRedirectCounter:                     {metricName: \"buffer_isolation_group_redirected_per_tl\", metricRollupName: \"buffer_isolation_group_redirected\"},\n\t\tBufferIsolationGroupRedirectFailureCounter:              {metricName: \"buffer_isolation_group_redirect_failure_per_tl\", metricRollupName: \"buffer_isolation_group_redirect_failure\"},\n\t\tBufferIsolationGroupMisconfiguredCounter:                {metricName: \"buffer_isolation_group_misconfigured_failure_per_tl\", metricRollupName: \"buffer_isolation_group_misconfigured_failure\"},\n\t\tExpiredTasksPerTaskListCounter:                          {metricName: \"tasks_expired_per_tl\", metricRollupName: \"tasks_expired\"},\n\t\tForwardedPerTaskListCounter:                             {metricName: \"forwarded_per_tl\", metricRollupName: \"forwarded\"},\n\t\tForwardTaskCallsPerTaskList:                             {metricName: \"forward_task_calls_per_tl\", metricRollupName: \"forward_task_calls\"},\n\t\tForwardTaskErrorsPerTaskList:                            {metricName: \"forward_task_errors_per_tl\", metricRollupName: \"forward_task_errors\"},\n\t\tSyncMatchForwardTaskThrottleErrorPerTasklist:            {metricName: \"sync_forward_task_throttle_errors_per_tl\", metricRollupName: \"sync_forward_task_throttle_errors\"},\n\t\tAsyncMatchForwardTaskThrottleErrorPerTasklist:           {metricName: \"async_forward_task_throttle_errors_per_tl\", metricRollupName: \"async_forward_task_throttle_errors\"},\n\t\tForwardQueryCallsPerTaskList:                            {metricName: \"forward_query_calls_per_tl\", metricRollupName: \"forward_query_calls\"},\n\t\tForwardQueryErrorsPerTaskList:                           {metricName: \"forward_query_errors_per_tl\", metricRollupName: \"forward_query_errors\"},\n\t\tForwardPollCallsPerTaskList:                             {metricName: \"forward_poll_calls_per_tl\", metricRollupName: \"forward_poll_calls\"},\n\t\tForwardPollErrorsPerTaskList:                            {metricName: \"forward_poll_errors_per_tl\", metricRollupName: \"forward_poll_errors\"},\n\t\tSyncMatchLatencyPerTaskList:                             {metricName: \"syncmatch_latency_per_tl\", metricRollupName: \"syncmatch_latency\", metricType: Timer},\n\t\tAsyncMatchLatencyPerTaskList:                            {metricName: \"asyncmatch_latency_per_tl\", metricRollupName: \"asyncmatch_latency\", metricType: Timer},\n\t\tAsyncMatchDispatchLatencyPerTaskList:                    {metricName: \"asyncmatch_dispatch_latency_per_tl\", metricRollupName: \"asyncmatch_dispatch_latency\", metricType: Timer},\n\t\tAsyncMatchDispatchTimeoutCounterPerTaskList:             {metricName: \"asyncmatch_dispatch_timeouts_per_tl\", metricRollupName: \"asyncmatch_dispatch_timeouts\"},\n\t\tForwardTaskLatencyPerTaskList:                           {metricName: \"forward_task_latency_per_tl\", metricRollupName: \"forward_task_latency\"},\n\t\tForwardQueryLatencyPerTaskList:                          {metricName: \"forward_query_latency_per_tl\", metricRollupName: \"forward_query_latency\"},\n\t\tForwardPollLatencyPerTaskList:                           {metricName: \"forward_poll_latency_per_tl\", metricRollupName: \"forward_poll_latency\"},\n\t\tLocalToLocalMatchPerTaskListCounter:                     {metricName: \"local_to_local_matches_per_tl\", metricRollupName: \"local_to_local_matches\"},\n\t\tLocalToRemoteMatchPerTaskListCounter:                    {metricName: \"local_to_remote_matches_per_tl\", metricRollupName: \"local_to_remote_matches\"},\n\t\tRemoteToLocalMatchPerTaskListCounter:                    {metricName: \"remote_to_local_matches_per_tl\", metricRollupName: \"remote_to_local_matches\"},\n\t\tRemoteToRemoteMatchPerTaskListCounter:                   {metricName: \"remote_to_remote_matches_per_tl\", metricRollupName: \"remote_to_remote_matches\"},\n\t\tIsolationTaskMatchPerTaskListCounter:                    {metricName: \"isolation_task_matches_per_tl\", metricType: Counter},\n\t\tIsolationSuccessPerTaskListCounter:                      {metricName: \"isolation_success_per_tl\", metricRollupName: \"isolation_success\"},\n\t\tPollerPerTaskListCounter:                                {metricName: \"poller_count_per_tl\", metricRollupName: \"poller_count\"},\n\t\tPollerInvalidIsolationGroupCounter:                      {metricName: \"poller_invalid_isolation_group_per_tl\", metricType: Counter},\n\t\tTaskListPartitionUpdateFailedCounter:                    {metricName: \"tasklist_partition_update_failed_per_tl\", metricType: Counter},\n\t\tTaskListManagersGauge:                                   {metricName: \"tasklist_managers\", metricType: Gauge},\n\t\tTaskLagPerTaskListGauge:                                 {metricName: \"task_lag_per_tl\", metricType: Gauge},\n\t\tTaskBacklogPerTaskListGauge:                             {metricName: \"task_backlog_per_tl\", metricType: Gauge},\n\t\tTaskCountPerTaskListGauge:                               {metricName: \"task_count_per_tl\", metricType: Gauge},\n\t\tRateLimitPerTaskListGauge:                               {metricName: \"rate_limit_per_tl\", metricType: Gauge},\n\t\tSyncMatchLocalPollLatencyPerTaskList:                    {metricName: \"syncmatch_local_poll_latency_per_tl\", metricRollupName: \"syncmatch_local_poll_latency\"},\n\t\tSyncMatchForwardPollLatencyPerTaskList:                  {metricName: \"syncmatch_forward_poll_latency_per_tl\", metricRollupName: \"syncmatch_forward_poll_latency\"},\n\t\tAsyncMatchLocalPollCounterPerTaskList:                   {metricName: \"asyncmatch_local_poll_per_tl\", metricRollupName: \"asyncmatch_local_poll\"},\n\t\tAsyncMatchLocalPollAttemptPerTaskList:                   {metricName: \"asyncmatch_local_poll_attempt_per_tl\", metricRollupName: \"asyncmatch_local_poll_attempt\", metricType: Timer},\n\t\tAsyncMatchLocalPollLatencyPerTaskList:                   {metricName: \"asyncmatch_local_poll_latency_per_tl\", metricRollupName: \"asyncmatch_local_poll_latency\"},\n\t\tAsyncMatchForwardPollCounterPerTaskList:                 {metricName: \"asyncmatch_forward_poll_per_tl\", metricRollupName: \"asyncmatch_forward_poll\"},\n\t\tAsyncMatchForwardPollAttemptPerTaskList:                 {metricName: \"asyncmatch_forward_poll_attempt_per_tl\", metricRollupName: \"asyncmatch_forward_poll_attempt\", metricType: Timer},\n\t\tAsyncMatchForwardPollLatencyPerTaskList:                 {metricName: \"asyncmatch_forward_poll_latency_per_tl\", metricRollupName: \"asyncmatch_forward_poll_latency\"},\n\t\tAsyncMatchLocalPollAfterForwardFailedCounterPerTaskList: {metricName: \"asyncmatch_local_poll_after_forward_failed_per_tl\", metricRollupName: \"asyncmatch_local_poll_after_forward_failed\"},\n\t\tAsyncMatchLocalPollAfterForwardFailedAttemptPerTaskList: {metricName: \"asyncmatch_local_poll_after_forward_failed_attempt_per_tl\", metricRollupName: \"asyncmatch_local_poll_after_forward_failed_attempt\", metricType: Timer},\n\t\tAsyncMatchLocalPollAfterForwardFailedLatencyPerTaskList: {metricName: \"asyncmatch_local_poll_after_forward_failed_latency_per_tl\", metricRollupName: \"asyncmatch_local_poll_after_forward_failed_latency\"},\n\t\tPollLocalMatchLatencyPerTaskList:                        {metricName: \"poll_local_match_latency_per_tl\", metricRollupName: \"poll_local_match_latency\", metricType: Timer},\n\t\tPollForwardMatchLatencyPerTaskList:                      {metricName: \"poll_forward_match_latency_per_tl\", metricRollupName: \"poll_forward_match_latency\", metricType: Timer},\n\t\tPollLocalMatchAfterForwardFailedLatencyPerTaskList:      {metricName: \"poll_local_match_after_forward_failed_latency_per_tl\", metricRollupName: \"poll_local_match_after_forward_failed_latency\", metricType: Timer},\n\t\tPollDecisionTaskAlreadyStartedCounterPerTaskList:        {metricName: \"poll_decision_task_already_started_per_tl\", metricType: Counter},\n\t\tPollActivityTaskAlreadyStartedCounterPerTaskList:        {metricName: \"poll_activity_task_already_started_per_tl\", metricType: Counter},\n\t\tTaskListReadWritePartitionMismatchGauge:                 {metricName: \"tasklist_read_write_partition_mismatch\", metricType: Gauge},\n\t\tTaskListPollerPartitionMismatchGauge:                    {metricName: \"tasklist_poller_partition_mismatch\", metricType: Gauge},\n\t\tEstimatedAddTaskQPSGauge:                                {metricName: \"estimated_add_task_qps_per_tl\", metricType: Gauge},\n\t\tTaskListPartitionUpscaleThresholdGauge:                  {metricName: \"tasklist_partition_upscale_threshold\", metricType: Gauge},\n\t\tTaskListPartitionDownscaleThresholdGauge:                {metricName: \"tasklist_partition_downscale_threshold\", metricType: Gauge},\n\t\tStandbyClusterTasksCompletedCounterPerTaskList:          {metricName: \"standby_cluster_tasks_completed_per_tl\", metricType: Counter},\n\t\tStandbyClusterTasksNotStartedCounterPerTaskList:         {metricName: \"standby_cluster_tasks_not_started_per_tl\", metricType: Counter},\n\t\tStandbyClusterTasksCompletionFailurePerTaskList:         {metricName: \"standby_cluster_tasks_completion_failure_per_tl\", metricType: Counter},\n\t\tTaskIsolationLeakPerTaskList:                            {metricName: \"task_isolation_leak_per_tl\", metricRollupName: \"task_isolation_leak\"},\n\t\tPartitionUpscale:                                        {metricName: \"partition_upscale_per_tl\", metricRollupName: \"partition_upscale\"},\n\t\tPartitionDownscale:                                      {metricName: \"partition_downscale_per_tl\", metricRollupName: \"partition_downscale\"},\n\t\tPartitionDrained:                                        {metricName: \"partition_drained_per_tl\", metricRollupName: \"partition_drained\"},\n\t\tIsolationRebalance:                                      {metricName: \"isolation_rebalance_per_tl\", metricRollupName: \"isolation_rebalance\"},\n\t\tIsolationGroupStartedPolling:                            {metricName: \"ig_started_polling_per_tl\", metricRollupName: \"ig_started_polling\"},\n\t\tIsolationGroupStoppedPolling:                            {metricName: \"ig_stopped_polling_per_tl\", metricRollupName: \"ig_stopped_polling\"},\n\t\tIsolationGroupUpscale:                                   {metricName: \"ig_upscale_per_tl\", metricRollupName: \"ig_upscale\"},\n\t\tIsolationGroupDownscale:                                 {metricName: \"ig_downscale_per_tl\", metricRollupName: \"ig_downscale\"},\n\t\tIsolationGroupPartitionsGauge:                           {metricName: \"ig_partitions_per_tl\", metricType: Gauge},\n\t},\n\tWorker: {\n\t\tReplicatorMessages:                            {metricName: \"replicator_messages\"},\n\t\tReplicatorFailures:                            {metricName: \"replicator_errors\"},\n\t\tReplicatorMessagesDropped:                     {metricName: \"replicator_messages_dropped\"},\n\t\tReplicatorLatency:                             {metricName: \"replicator_latency\"},\n\t\tReplicatorDLQFailures:                         {metricName: \"replicator_dlq_enqueue_fails\", metricType: Counter},\n\t\tESProcessorRequests:                           {metricName: \"es_processor_requests\"},\n\t\tESProcessorRetries:                            {metricName: \"es_processor_retries\"},\n\t\tESProcessorFailures:                           {metricName: \"es_processor_errors\"},\n\t\tESProcessorCorruptedData:                      {metricName: \"es_processor_corrupted_data\"},\n\t\tESProcessorProcessMsgLatency:                  {metricName: \"es_processor_process_msg_latency\", metricType: Timer},\n\t\tIndexProcessorCorruptedData:                   {metricName: \"index_processor_corrupted_data\"},\n\t\tIndexProcessorProcessMsgLatency:               {metricName: \"index_processor_process_msg_latency\", metricType: Timer},\n\t\tArchiverNonRetryableErrorCount:                {metricName: \"archiver_non_retryable_error\"},\n\t\tArchiverStartedCount:                          {metricName: \"archiver_started\"},\n\t\tArchiverStoppedCount:                          {metricName: \"archiver_stopped\"},\n\t\tArchiverCoroutineStartedCount:                 {metricName: \"archiver_coroutine_started\"},\n\t\tArchiverCoroutineStoppedCount:                 {metricName: \"archiver_coroutine_stopped\"},\n\t\tArchiverHandleHistoryRequestLatency:           {metricName: \"archiver_handle_history_request_latency\"},\n\t\tArchiverHandleVisibilityRequestLatency:        {metricName: \"archiver_handle_visibility_request_latency\"},\n\t\tArchiverUploadWithRetriesLatency:              {metricName: \"archiver_upload_with_retries_latency\"},\n\t\tArchiverDeleteWithRetriesLatency:              {metricName: \"archiver_delete_with_retries_latency\"},\n\t\tArchiverUploadFailedAllRetriesCount:           {metricName: \"archiver_upload_failed_all_retries\"},\n\t\tArchiverUploadSuccessCount:                    {metricName: \"archiver_upload_success\"},\n\t\tArchiverDeleteFailedAllRetriesCount:           {metricName: \"archiver_delete_failed_all_retries\"},\n\t\tArchiverDeleteSuccessCount:                    {metricName: \"archiver_delete_success\"},\n\t\tArchiverHandleVisibilityFailedAllRetiresCount: {metricName: \"archiver_handle_visibility_failed_all_retries\"},\n\t\tArchiverHandleVisibilitySuccessCount:          {metricName: \"archiver_handle_visibility_success\"},\n\t\tArchiverBacklogSizeGauge:                      {metricName: \"archiver_backlog_size\"},\n\t\tArchiverPumpTimeoutCount:                      {metricName: \"archiver_pump_timeout\"},\n\t\tArchiverPumpSignalThresholdCount:              {metricName: \"archiver_pump_signal_threshold\"},\n\t\tArchiverPumpTimeoutWithoutSignalsCount:        {metricName: \"archiver_pump_timeout_without_signals\"},\n\t\tArchiverPumpSignalChannelClosedCount:          {metricName: \"archiver_pump_signal_channel_closed\"},\n\t\tArchiverWorkflowStartedCount:                  {metricName: \"archiver_workflow_started\"},\n\t\tArchiverNumPumpedRequestsCount:                {metricName: \"archiver_num_pumped_requests\"},\n\t\tArchiverNumHandledRequestsCount:               {metricName: \"archiver_num_handled_requests\"},\n\t\tArchiverPumpedNotEqualHandledCount:            {metricName: \"archiver_pumped_not_equal_handled\"},\n\t\tArchiverHandleAllRequestsLatency:              {metricName: \"archiver_handle_all_requests_latency\"},\n\t\tArchiverWorkflowStoppingCount:                 {metricName: \"archiver_workflow_stopping\"},\n\t\tTaskProcessedCount:                            {metricName: \"task_processed\", metricType: Gauge},\n\t\tTaskDeletedCount:                              {metricName: \"task_deleted\", metricType: Gauge},\n\t\tTaskListProcessedCount:                        {metricName: \"tasklist_processed\", metricType: Gauge},\n\t\tTaskListDeletedCount:                          {metricName: \"tasklist_deleted\", metricType: Gauge},\n\t\tTaskListOutstandingCount:                      {metricName: \"tasklist_outstanding\", metricType: Gauge},\n\t\tExecutionsOutstandingCount:                    {metricName: \"executions_outstanding\", metricType: Gauge},\n\t\tStartedCount:                                  {metricName: \"started\", metricType: Counter},\n\t\tStoppedCount:                                  {metricName: \"stopped\", metricType: Counter},\n\t\tExecutorTasksDeferredCount:                    {metricName: \"executor_deferred\", metricType: Counter},\n\t\tExecutorTasksDroppedCount:                     {metricName: \"executor_dropped\", metricType: Counter},\n\t\tBatcherProcessorSuccess:                       {metricName: \"batcher_processor_requests\", metricType: Counter},\n\t\tBatcherProcessorFailures:                      {metricName: \"batcher_processor_errors\", metricType: Counter},\n\t\tHistoryScavengerSuccessCount:                  {metricName: \"scavenger_success\", metricType: Counter},\n\t\tHistoryScavengerErrorCount:                    {metricName: \"scavenger_errors\", metricType: Counter},\n\t\tHistoryScavengerSkipCount:                     {metricName: \"scavenger_skips\", metricType: Counter},\n\t\tDomainReplicationEnqueueDLQCount:              {metricName: \"domain_replication_dlq_enqueue_requests\", metricType: Counter},\n\t\tScannerExecutionsGauge:                        {metricName: \"scanner_executions\", metricType: Gauge},\n\t\tScannerCorruptedGauge:                         {metricName: \"scanner_corrupted\", metricType: Gauge},\n\t\tScannerCheckFailedGauge:                       {metricName: \"scanner_check_failed\", metricType: Gauge},\n\t\tScannerCorruptionByTypeGauge:                  {metricName: \"scanner_corruption_by_type\", metricType: Gauge},\n\t\tScannerCorruptedOpenExecutionGauge:            {metricName: \"scanner_corrupted_open_execution\", metricType: Gauge},\n\t\tScannerShardSizeMaxGauge:                      {metricName: \"scanner_shard_size_max\", metricType: Gauge},\n\t\tScannerShardSizeMedianGauge:                   {metricName: \"scanner_shard_size_median\", metricType: Gauge},\n\t\tScannerShardSizeMinGauge:                      {metricName: \"scanner_shard_size_min\", metricType: Gauge},\n\t\tScannerShardSizeNinetyGauge:                   {metricName: \"scanner_shard_size_ninety\", metricType: Gauge},\n\t\tScannerShardSizeSeventyFiveGauge:              {metricName: \"scanner_shard_size_seventy_five\", metricType: Gauge},\n\t\tScannerShardSizeTwentyFiveGauge:               {metricName: \"scanner_shard_size_twenty_five\", metricType: Gauge},\n\t\tScannerShardSizeTenGauge:                      {metricName: \"scanner_shard_size_ten\", metricType: Gauge},\n\t\tShardScannerScan:                              {metricName: \"shardscanner_scan\", metricType: Counter},\n\t\tShardScannerFix:                               {metricName: \"shardscanner_fix\", metricType: Counter},\n\t\tDataCorruptionWorkflowFailure:                 {metricName: \"data_corruption_workflow_failure\", metricType: Counter},\n\t\tDataCorruptionWorkflowSuccessCount:            {metricName: \"data_corruption_workflow_success\", metricType: Counter},\n\t\tDataCorruptionWorkflowCount:                   {metricName: \"data_corruption_workflow_count\", metricType: Counter},\n\t\tDataCorruptionWorkflowSkipCount:               {metricName: \"data_corruption_workflow_skips\", metricType: Counter},\n\t\tESAnalyzerNumStuckWorkflowsDiscovered:         {metricName: \"es_analyzer_num_stuck_workflows_discovered\", metricType: Counter},\n\t\tESAnalyzerNumStuckWorkflowsRefreshed:          {metricName: \"es_analyzer_num_stuck_workflows_refreshed\", metricType: Counter},\n\t\tESAnalyzerNumStuckWorkflowsFailedToRefresh:    {metricName: \"es_analyzer_num_stuck_workflows_failed_to_refresh\", metricType: Counter},\n\t\tESAnalyzerNumLongRunningWorkflows:             {metricName: \"es_analyzer_num_long_running_workflows\", metricType: Counter},\n\t\tAsyncWorkflowConsumerCount:                    {metricName: \"async_workflow_consumer_count\", metricType: Gauge},\n\t\tAsyncWorkflowProcessMsgLatency:                {metricName: \"async_workflow_process_msg_latency\", metricType: Timer},\n\t\tAsyncWorkflowFailureCorruptMsgCount:           {metricName: \"async_workflow_failure_corrupt_msg\", metricType: Counter},\n\t\tAsyncWorkflowFailureByFrontendCount:           {metricName: \"async_workflow_failure_by_frontend\", metricType: Counter},\n\t\tAsyncWorkflowSuccessCount:                     {metricName: \"async_workflow_success\", metricType: Counter},\n\t\tDiagnosticsWorkflowStartedCount:               {metricName: \"diagnostics_workflow_count\", metricType: Counter},\n\t\tDiagnosticsWorkflowSuccess:                    {metricName: \"diagnostics_workflow_success\", metricType: Counter},\n\t\tDiagnosticsWorkflowExecutionLatency:           {metricName: \"diagnostics_workflow_execution_latency\", metricType: Timer},\n\t},\n\tShardDistributor: {\n\t\tShardDistributorRequests:                        {metricName: \"shard_distributor_requests\", metricType: Counter},\n\t\tShardDistributorErrContextTimeoutCounter:        {metricName: \"shard_distributor_err_context_timeout\", metricType: Counter},\n\t\tShardDistributorFailures:                        {metricName: \"shard_distributor_failures\", metricType: Counter},\n\t\tShardDistributorLatency:                         {metricName: \"shard_distributor_latency\", metricType: Timer},\n\t\tShardDistributorErrNamespaceNotFound:            {metricName: \"shard_distributor_err_namespace_not_found\", metricType: Counter},\n\t\tShardDistributorErrShardNotFound:                {metricName: \"shard_distributor_err_shard_not_found\", metricType: Counter},\n\t\tShardDistributorAssignLoopShardRebalanceLatency: {metricName: \"shard_distrubutor_shard_assign_latency\", metricType: Histogram},\n\t\tShardDistributorAssignLoopNumRebalancedShards:   {metricName: \"shard_distributor_shard_assign_reassigned_shards\", metricType: Gauge},\n\t\tShardDistributorAssignLoopAttempts:              {metricName: \"shard_distrubutor_shard_assign_attempt\", metricType: Counter},\n\t\tShardDistributorAssignLoopSuccess:               {metricName: \"shard_distrubutor_shard_assign_success\", metricType: Counter},\n\t\tShardDistributorAssignLoopFail:                  {metricName: \"shard_distrubutor_shard_assign_fail\", metricType: Counter},\n\n\t\tShardDistributorActiveShards:               {metricName: \"shard_distributor_active_shards\", metricType: Gauge},\n\t\tShardDistributorTotalExecutors:             {metricName: \"shard_distributor_total_executors\", metricType: Gauge},\n\t\tShardDistributorOldestExecutorHeartbeatLag: {metricName: \"shard_distributor_oldest_executor_heartbeat_lag\", metricType: Gauge},\n\n\t\tShardDistributorStoreExecutorNotFound:             {metricName: \"shard_distributor_store_executor_not_found\", metricType: Counter},\n\t\tShardDistributorStoreFailuresPerNamespace:         {metricName: \"shard_distributor_store_failures_per_namespace\", metricType: Counter},\n\t\tShardDistributorStoreRequestsPerNamespace:         {metricName: \"shard_distributor_store_requests_per_namespace\", metricType: Counter},\n\t\tShardDistributorStoreLatencyHistogramPerNamespace: {metricName: \"shard_distributor_store_latency_histogram_per_namespace\", metricType: Histogram, buckets: ShardDistributorExecutorStoreLatencyBuckets},\n\n\t\tShardDistributorShardAssignmentDistributionLatency: {metricName: \"shard_distributor_shard_assignment_distribution_latency\", metricType: Histogram, buckets: ShardDistributorShardAssignmentLatencyBuckets},\n\t\tShardDistributorShardHandoverLatency:               {metricName: \"shard_distributor_shard_handover_latency\", metricType: Histogram, buckets: ShardDistributorShardAssignmentLatencyBuckets},\n\n\t\tShardDistributorWatchProcessingLatency: {metricName: \"shard_distributor_watch_processing_latency\", metricType: Histogram, buckets: Default1ms100s.buckets()},\n\t\tShardDistributorWatchEventsReceived:    {metricName: \"shard_distributor_watch_events_received\", metricType: Counter},\n\t},\n}\n\nvar (\n\t// PersistenceLatencyBuckets contains duration buckets for measuring persistence latency\n\tPersistenceLatencyBuckets = tally.DurationBuckets([]time.Duration{\n\t\t1 * time.Millisecond,\n\t\t2 * time.Millisecond,\n\t\t3 * time.Millisecond,\n\t\t4 * time.Millisecond,\n\t\t5 * time.Millisecond,\n\t\t6 * time.Millisecond,\n\t\t7 * time.Millisecond,\n\t\t8 * time.Millisecond,\n\t\t9 * time.Millisecond,\n\t\t10 * time.Millisecond,\n\t\t12 * time.Millisecond,\n\t\t15 * time.Millisecond,\n\t\t17 * time.Millisecond,\n\t\t20 * time.Millisecond,\n\t\t25 * time.Millisecond,\n\t\t30 * time.Millisecond,\n\t\t35 * time.Millisecond,\n\t\t40 * time.Millisecond,\n\t\t50 * time.Millisecond,\n\t\t60 * time.Millisecond,\n\t\t70 * time.Millisecond,\n\t\t80 * time.Millisecond,\n\t\t90 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t120 * time.Millisecond,\n\t\t150 * time.Millisecond,\n\t\t170 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t250 * time.Millisecond,\n\t\t300 * time.Millisecond,\n\t\t400 * time.Millisecond,\n\t\t500 * time.Millisecond,\n\t\t600 * time.Millisecond,\n\t\t700 * time.Millisecond,\n\t\t800 * time.Millisecond,\n\t\t900 * time.Millisecond,\n\t\t1 * time.Second,\n\t\t2 * time.Second,\n\t\t3 * time.Second,\n\t\t4 * time.Second,\n\t\t5 * time.Second,\n\t\t6 * time.Second,\n\t\t7 * time.Second,\n\t\t8 * time.Second,\n\t\t9 * time.Second,\n\t\t10 * time.Second,\n\t\t12 * time.Second,\n\t\t15 * time.Second,\n\t\t20 * time.Second,\n\t\t25 * time.Second,\n\t\t30 * time.Second,\n\t\t35 * time.Second,\n\t\t40 * time.Second,\n\t\t50 * time.Second,\n\t\t60 * time.Second,\n\t})\n\n\tShardDistributorExecutorStoreLatencyBuckets = tally.DurationBuckets([]time.Duration{\n\t\t0,\n\t\t5 * time.Millisecond,\n\t\t10 * time.Millisecond,\n\t\t25 * time.Millisecond,\n\t\t50 * time.Millisecond,\n\t\t75 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t120 * time.Millisecond,\n\t\t150 * time.Millisecond,\n\t\t170 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t250 * time.Millisecond,\n\t\t300 * time.Millisecond,\n\t\t400 * time.Millisecond,\n\t\t500 * time.Millisecond,\n\t\t600 * time.Millisecond,\n\t\t700 * time.Millisecond,\n\t\t800 * time.Millisecond,\n\t\t900 * time.Millisecond,\n\t\t1 * time.Second,\n\t\t2 * time.Second,\n\t\t3 * time.Second,\n\t\t4 * time.Second,\n\t\t5 * time.Second,\n\t\t6 * time.Second,\n\t\t7 * time.Second,\n\t\t8 * time.Second,\n\t\t9 * time.Second,\n\t\t10 * time.Second,\n\t\t12 * time.Second,\n\t\t15 * time.Second,\n\t\t20 * time.Second,\n\t\t25 * time.Second,\n\t\t30 * time.Second,\n\t\t35 * time.Second,\n\t\t40 * time.Second,\n\t\t50 * time.Second,\n\t\t60 * time.Second,\n\t})\n\n\tShardDistributorShardAssignmentLatencyBuckets = tally.DurationBuckets([]time.Duration{\n\t\t// ShardDistributorShardHandoverLatency for GracefulHandoverType should be within 0s and 1s\n\n\t\t0,\n\t\t50 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t300 * time.Millisecond,\n\t\t400 * time.Millisecond,\n\t\t500 * time.Millisecond,\n\t\t600 * time.Millisecond,\n\t\t700 * time.Millisecond,\n\t\t800 * time.Millisecond,\n\t\t900 * time.Millisecond,\n\n\t\t// ShardDistributorShardHandoverLatency for EmergencyHandoverType should be within 0s and 10s\n\t\t1 * time.Second,\n\t\t2 * time.Second,\n\t\t3 * time.Second,\n\t\t4 * time.Second,\n\t\t5 * time.Second,\n\t\t6 * time.Second,\n\t\t7 * time.Second,\n\t\t8 * time.Second,\n\t\t9 * time.Second,\n\t\t10 * time.Second,\n\n\t\t12 * time.Second,\n\t\t15 * time.Second,\n\t\t20 * time.Second,\n\t\t30 * time.Second,\n\t\t45 * time.Second,\n\n\t\t1 * time.Minute,\n\t\t2 * time.Minute,\n\t\t5 * time.Minute,\n\t\t10 * time.Minute,\n\t})\n\n\t// ReplicationTaskDelayBucket contains buckets for replication task delay\n\tReplicationTaskDelayBucket = tally.DurationBuckets([]time.Duration{\n\t\t0 * time.Second, // zero value is needed for the first bucket\n\t\t1 * time.Second,\n\t\t10 * time.Second,\n\t\t1 * time.Minute,\n\t\t5 * time.Minute,\n\t\t10 * time.Minute,\n\t\t30 * time.Minute,\n\t\t1 * time.Hour,\n\t\t2 * time.Hour,\n\t\t6 * time.Hour,\n\t\t12 * time.Hour,\n\t\t24 * time.Hour,\n\t\t36 * time.Hour,\n\t\t48 * time.Hour,\n\t\t72 * time.Hour,\n\t\t96 * time.Hour,\n\t\t120 * time.Hour,\n\t\t144 * time.Hour,\n\t\t168 * time.Hour, // one week\n\t})\n\n\tHistoryTaskLatencyBuckets = tally.DurationBuckets([]time.Duration{\n\t\t1 * time.Millisecond,\n\t\t5 * time.Millisecond,\n\t\t10 * time.Millisecond,\n\t\t20 * time.Millisecond,\n\t\t50 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t250 * time.Millisecond,\n\t\t500 * time.Millisecond,\n\t\t1 * time.Second,\n\t\t2 * time.Second,\n\t\t5 * time.Second,\n\t\t10 * time.Second,\n\t\t30 * time.Second,\n\t\t1 * time.Minute,\n\t\t5 * time.Minute,\n\t\t10 * time.Minute,\n\t\t1 * time.Hour,\n\t})\n)\n\n// GlobalRatelimiterUsageHistogram contains buckets for tracking how many ratelimiters are\n// in which various states (startup, healthy, failing, as well as aggregator-side quantities, deleted, etc).\n//\n// this is intended for coarse scale checking, not alerting, so the buckets\n// should be considered unstable and can be changed whenever desired.\nvar GlobalRatelimiterUsageHistogram = append(\n\ttally.ValueBuckets{0},                              // need an explicit 0 or zero is reported as 1\n\ttally.MustMakeExponentialValueBuckets(1, 2, 17)..., // 1..65536\n)\n\n// ResponseRowSizeBuckets contains buckets for tracking how many rows are returned per persistence operation\nvar ResponseRowSizeBuckets = append(\n\ttally.ValueBuckets{0},                              // need an explicit 0 or zero is reported as 1\n\ttally.MustMakeExponentialValueBuckets(1, 2, 17)..., // 1..65536\n)\n\n// TaskCountBuckets contains buckets for tracking task counts that can reach into the millions,\n// such as replication lag measured as task ID distance (2^20 = 1,048,576).\nvar TaskCountBuckets = append(\n\ttally.ValueBuckets{0},                              // need an explicit 0 or zero is reported as 1\n\ttally.MustMakeExponentialValueBuckets(1, 2, 21)..., // 1..1048576\n)\n\n// DomainCacheUpdateBuckets contain metric results for domain update operations\nvar DomainCacheUpdateBuckets = append(\n\ttally.ValueBuckets{0},                              // need an explicit 0 or zero is reported as 1\n\ttally.MustMakeExponentialValueBuckets(1, 2, 17)..., // 1..65536\n)\n\n// ResponsePayloadSizeBuckets contains buckets for tracking the size of the payload returned per persistence operation\nvar ResponsePayloadSizeBuckets = append(\n\ttally.ValueBuckets{0},                                 // need an explicit 0 or zero is reported as 1\n\ttally.MustMakeExponentialValueBuckets(1024, 2, 20)..., // 1kB..1GB\n)\n\n// ExponentialDurationBuckets is a set of exponential duration buckets\nvar ExponentialDurationBuckets = func() tally.DurationBuckets {\n\t// generate 79 buckets, starting from 1ms, with a factor of 2^0.25\n\tbuckets, err := tally.ExponentialDurationBuckets(1*time.Millisecond, math.Pow(2, 0.25), 79)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\t// add a 0 bucket to the beginning\n\tbuckets = append([]time.Duration{0}, buckets...)\n\treturn buckets\n}()\n\n// ErrorClass is an enum to help with classifying SLA vs. non-SLA errors (SLA = \"service level agreement\")\ntype ErrorClass uint8\n\nconst (\n\t// NoError indicates that there is no error (error should be nil)\n\tNoError = ErrorClass(iota)\n\t// UserError indicates that this is NOT an SLA-reportable error\n\tUserError\n\t// InternalError indicates that this is an SLA-reportable error\n\tInternalError\n)\n\n// Empty returns true if the metricName is an empty string\nfunc (mn MetricName) Empty() bool {\n\treturn mn == \"\"\n}\n\n// String returns string representation of this metric name\nfunc (mn MetricName) String() string {\n\treturn string(mn)\n}\n"
  },
  {
    "path": "common/metrics/defs_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"regexp\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nvar IsMetric = regexp.MustCompile(`^[a-z][a-z_]*$`).MatchString\n\nfunc TestScopeDefsMapped(t *testing.T) {\n\tfor i := PersistenceCreateShardScope; i < NumCommonScopes; i++ {\n\t\tkey, ok := ScopeDefs[Common][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := AdminDescribeHistoryHostScope; i < NumAdminScopes; i++ {\n\t\tkey, ok := ScopeDefs[Frontend][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := FrontendStartWorkflowExecutionScope; i < NumFrontendScopes; i++ {\n\t\tkey, ok := ScopeDefs[Frontend][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := HistoryStartWorkflowExecutionScope; i < NumHistoryScopes; i++ {\n\t\tkey, ok := ScopeDefs[History][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := MatchingPollForDecisionTaskScope; i < NumMatchingScopes; i++ {\n\t\tkey, ok := ScopeDefs[Matching][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := ReplicatorScope; i < NumWorkerScopes; i++ {\n\t\tkey, ok := ScopeDefs[Worker][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n\tfor i := ShardDistributorGetShardOwnerScope; i < NumShardDistributorScopes; i++ {\n\t\tkey, ok := ScopeDefs[ShardDistributor][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t\tfor tag := range key.tags {\n\t\t\tassert.True(t, IsMetric(tag), \"metric tags should conform to regex\")\n\t\t}\n\t}\n}\n\nfunc TestMetricDefsMapped(t *testing.T) {\n\tfor i := CadenceRequests; i < NumCommonMetrics; i++ {\n\t\tkey, ok := MetricDefs[Common][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t}\n\tfor i := TaskRequests; i < NumHistoryMetrics; i++ {\n\t\tkey, ok := MetricDefs[History][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t}\n\tfor i := PollSuccessPerTaskListCounter; i < NumMatchingMetrics; i++ {\n\t\tkey, ok := MetricDefs[Matching][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t}\n\tfor i := ReplicatorMessages; i < NumWorkerMetrics; i++ {\n\t\tkey, ok := MetricDefs[Worker][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t}\n\tfor i := ShardDistributorRequests; i < NumShardDistributorMetrics; i++ {\n\t\tkey, ok := MetricDefs[ShardDistributor][i]\n\t\trequire.True(t, ok)\n\t\trequire.NotEmpty(t, key)\n\t}\n}\n\nfunc TestMetricDefs(t *testing.T) {\n\tfor service, metrics := range MetricDefs {\n\t\tfor _, metricDef := range metrics {\n\t\t\tmatched := IsMetric(string(metricDef.metricName))\n\t\t\tassert.True(t, matched, fmt.Sprintf(\"Service: %v, metric_name: %v\", service, metricDef.metricName))\n\t\t}\n\t}\n}\n\n// \"index -> operation\" must be unique so they can be looked up without needing to know the service ID.\n//\n// Duplicate indexes with the same operation name are technically fine, but there doesn't seem to be any benefit in allowing it,\n// and it trivially ensures that all values have only one operation name.\nfunc TestOperationIndexesAreUnique(t *testing.T) {\n\tseen := make(map[ScopeIdx]bool)\n\tfor serviceIdx, serviceOps := range ScopeDefs {\n\t\tfor idx := range serviceOps {\n\t\t\tif seen[idx] {\n\t\t\t\tt.Error(\"duplicate operation index:\", idx, \"with name:\", serviceOps[idx].operation, \"in service:\", serviceIdx)\n\t\t\t}\n\t\t\tseen[idx] = true\n\n\t\t\t// to serve as documentation: operations are NOT unique due to dups across services.\n\t\t\t// this is probably fine, they're just used as metric tag values.\n\t\t\t// but not being able to prevent dups means it's possible we have unintentional name collisions.\n\t\t\t//\n\t\t\t// name := serviceOps[idx].operation\n\t\t\t// if seenNames[name] {\n\t\t\t// \tt.Error(\"duplicate operation string:\", name, \"at different indexes\")\n\t\t\t// }\n\t\t\t// seenNames[name] = true\n\t\t}\n\t}\n}\n\nfunc TestMetricsAreUnique(t *testing.T) {\n\t// Duplicate indexes is arguably fine, but there doesn't seem to be any benefit in allowing it.\n\tt.Run(\"indexes\", func(t *testing.T) {\n\t\tseen := make(map[MetricIdx]string)\n\t\tfor _, serviceMetrics := range MetricDefs {\n\t\t\tfor idx, met := range serviceMetrics {\n\t\t\t\tif prev, ok := seen[idx]; ok {\n\t\t\t\t\tt.Errorf(\"duplicate metric index %d, with names %q and %q\", idx, met.metricName.String(), prev)\n\t\t\t\t}\n\t\t\t\tseen[idx] = met.metricName.String()\n\t\t\t}\n\t\t}\n\t})\n\t// Duplicate names carry a high risk of causing different-tag collisions in Prometheus, and should not be allowed.\n\tt.Run(\"names\", func(t *testing.T) {\n\t\tseen := make(map[string]bool)\n\t\tfor _, serviceMetrics := range MetricDefs {\n\t\t\tfor _, met := range serviceMetrics {\n\t\t\t\tname := met.metricName.String()\n\t\t\t\tif seen[name] {\n\t\t\t\t\tswitch name {\n\t\t\t\t\tcase \"cache_full\", \"cache_miss\", \"cache_hit\", \"cadence_requests_per_tl\", \"cross_cluster_fetch_errors\":\n\t\t\t\t\t\tcontinue // known dup.  worth changing as some cause problems.\n\t\t\t\t\t}\n\t\t\t\t\tt.Errorf(\"duplicate metric name %q\", name)\n\t\t\t\t}\n\t\t\t\tseen[name] = true\n\t\t\t}\n\t\t}\n\t})\n}\n\nfunc TestHistogramSuffixes(t *testing.T) {\n\tfor _, serviceMetrics := range MetricDefs {\n\t\tfor _, def := range serviceMetrics {\n\t\t\tif def.intExponentialBuckets != nil {\n\t\t\t\tif !strings.HasSuffix(def.metricName.String(), \"_counts\") {\n\t\t\t\t\tt.Errorf(\"int-exponential-histogram metric %q should have suffix \\\"_counts\\\"\", def.metricName.String())\n\t\t\t\t}\n\t\t\t}\n\t\t\tif def.exponentialBuckets != nil {\n\t\t\t\tif !strings.HasSuffix(def.metricName.String(), \"_ns\") {\n\t\t\t\t\tt.Errorf(\"exponential-histogram metric %q should have suffix \\\"_ns\\\"\", def.metricName.String())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc TestExponentialDurationBuckets(t *testing.T) {\n\tfactor := math.Pow(2, 0.25)\n\tassert.Equal(t, 80, len(ExponentialDurationBuckets))\n\tassert.Equal(t, 0*time.Millisecond, ExponentialDurationBuckets[0], \"bucket[0] mismatch\")\n\tassert.Equal(t, 1*time.Millisecond, ExponentialDurationBuckets[1], \"bucket[1] mismatch\")\n\tassert.InDelta(t, bucketVal(1*time.Millisecond, factor, 2), ExponentialDurationBuckets[2], float64(time.Millisecond), \"bucket[2] mismatch\")\n\tassert.InDelta(t, bucketVal(1*time.Millisecond, factor, 79), ExponentialDurationBuckets[79], 0.1*float64(time.Second), \"bucket[79] mismatch\")\n}\n\nfunc bucketVal(start time.Duration, factor float64, bucket int) time.Duration {\n\tif bucket == 0 {\n\t\treturn 0\n\t}\n\treturn time.Duration(math.Pow(factor, float64(bucket-1)) * float64(start))\n}\n"
  },
  {
    "path": "common/metrics/histograms.go",
    "content": "package metrics\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"slices\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\n// Nearly all histograms should use pre-defined buckets here, to encourage consistency.\n//\n// When creating a new one, make sure to read makeSubsettableHistogram for context, and choose\n// values that communicate your *intent*, not the actual final value.  Add a test to show the\n// resulting buckets, so reviewers can see just how much / how little you chose.\n//\n// Similarly, name them more by their intent than their exact ranges, because \"it has just barely\n// enough buckets\" is almost always not *actually* enough when things misbehave.\n// Choosing by intent also helps make adjusting these values safer if needed (e.g. subsetting or\n// reducing the range), because we can be sure all intents match.\nvar (\n\t// Default1ms100s is our \"default\" set of buckets, targeting at least 1ms through 100s,\n\t// and is \"rounded up\" slightly to reach 80 buckets (100s needs 68 buckets),\n\t// plus multi-minute exceptions are common enough to support for the small additional cost\n\t// (this goes up to ~15m).\n\t//\n\t// That makes this *relatively* costly, but hopefully good enough for most metrics without\n\t// needing any careful thought.  If this proves incorrect, it will be down-scaled without\n\t// checking existing uses, so make sure to use something else if you require a certain level\n\t// of precision.\n\t//\n\t// If you need sub-millisecond precision, or substantially longer durations than about 1 minute,\n\t// consider using a different histogram.\n\tDefault1ms100s = makeSubsettableHistogram(2, time.Millisecond, 100*time.Second, 80)\n\n\t// Low1ms100s is a half-resolution version of Default1ms100s, intended for use any time the default\n\t// is expected to be more costly than we feel comfortable with.  Or for automatic down-scaling\n\t// at runtime via dynamic config, if that's ever built.\n\t//\n\t// This uses only 40 buckets, so it's likely good enough for most purposes, e.g. database operations\n\t// that might have high cardinality but should not exceed about a minute even in exceptional cases.\n\t// Reducing this to \"1ms to 10s\" seems reasonable for some cases, but that still needs ~32 buckets,\n\t// which isn't a big savings and reduces our ability to notice abnormally slow events.\n\tLow1ms100s = Default1ms100s.subsetTo(1)\n\n\t// High1ms24h covers things like activity latency, where ranges are very large but\n\t// \"longer than 1 day\" is not particularly worth being precise about.\n\t//\n\t// This uses a lot of buckets, so it must be used with relatively-low cardinality elsewhere,\n\t// and/or consider dual emitting (Mid1ms24h or lower) so overviews can be queried efficiently.\n\tHigh1ms24h = makeSubsettableHistogram(2, time.Millisecond, 24*time.Hour, 112)\n\n\t// Mid1ms24h is a one-scale-lower version of High1ms24h,\n\t// for use when we know it's too detailed to be worth emitting.\n\t//\n\t// This uses 56 buckets, half of High1ms24h's 112\n\tMid1ms24h = High1ms24h.subsetTo(1)\n\n\t// Mid1To16k is a histogram for small counters, like \"how many replication tasks did we receive\".\n\t//\n\t// This targets single-digits through ~10k with some buffer, and ends well below 1M.\n\t//\n\t// Note: we intentionally start at 8 (not 1). Starting at 1 causes duplicate bucket boundaries\n\t// due to float->duration truncation, which Prometheus rejects.\n\tMid1To16k = IntSubsettableHistogram(makeSubsettableHistogram(2, 8, 16384, 64))\n)\n\n// SubsettableHistogram is a duration-based histogram that can be subset to a lower scale\n// in a standardized, predictable way.  It is intentionally compatible with Prometheus and OTEL's\n// \"exponential histograms\": https://opentelemetry.io/docs/specs/otel/metrics/data-model/#exponentialhistogram\n//\n// These histograms MUST always have a \"_ns\" suffix in their name to avoid confusion with timers.\ntype SubsettableHistogram struct {\n\ttallyBuckets tally.DurationBuckets\n\n\tscale int\n}\n\n// IntSubsettableHistogram is a non-duration-based integer-distribution histogram, otherwise identical\n// to SubsettableHistogram but built as a separate type so you cannot pass the wrong one.\n//\n// These histograms MUST always have a \"_counts\" suffix in their name to avoid confusion with timers.\n// (more suffixes will likely be allowed as needed)\ntype IntSubsettableHistogram SubsettableHistogram\n\n// currently we have no apparent need for float-histograms,\n// as all our value ranges go from >=1 to many thousands, where\n// decimal precision is pointless and mostly just looks bad.\n//\n// if we ever need 0..1 precision in histograms, we can add them then.\n// \ttype FloatSubsettableHistogram struct {\n// \t\ttally.ValueBuckets\n// \t\tscale int\n// \t}\n\n// subsetTo takes an existing histogram and reduces its detail level.\n//\n// this can be replaced by simply creating a new histogram with the same args\n// as the original but half the length, but it's added because: 1) it works too,\n// and 2) to document how the process works, because we'll likely be doing this\n// at query time to reduce the level of detail in small / over-crowded graphs.\nfunc (s SubsettableHistogram) subsetTo(newScale int) SubsettableHistogram {\n\tif newScale >= s.scale {\n\t\tpanic(fmt.Sprintf(\"scale %v is not less than the current scale %v\", newScale, s.scale))\n\t}\n\tif newScale < 0 {\n\t\tpanic(fmt.Sprintf(\"negative scales (%v == greater than *2 per step) are possible, but do not have tests yet\", newScale))\n\t}\n\tdup := SubsettableHistogram{\n\t\ttallyBuckets: slices.Clone(s.tallyBuckets),\n\t\tscale:        s.scale,\n\t}\n\n\t// compress every other bucket per -1 scale\n\tfor dup.scale > newScale {\n\t\tif (len(dup.tallyBuckets)-2)%2 != 0 { // -2 for 0 and last-2^N\n\t\t\tpanic(fmt.Sprintf(\n\t\t\t\t\"cannot subset from scale %v to %v, %v-buckets is not divisible by 2\",\n\t\t\t\tdup.scale, dup.scale-1, len(dup.tallyBuckets)-2))\n\t\t}\n\t\tif len(dup.tallyBuckets) <= 3 {\n\t\t\t// at 3 buckets, there's just 0, start, end.\n\t\t\t//\n\t\t\t// this is well past the point of being useful,\n\t\t\t// and it means we might try to slice `[1:0]` or similar.\n\t\t\tpanic(fmt.Sprintf(\n\t\t\t\t\"not enough buckets to subset from scale %d to %d, only have %d: %v\",\n\t\t\t\tdup.scale, dup.scale-1, len(dup.tallyBuckets), dup.tallyBuckets))\n\t\t}\n\n\t\t// the first and last buckets will be kept, the rest will lose half per scale\n\t\tbucketsToCompress := dup.tallyBuckets[1 : len(dup.tallyBuckets)-2] // trim\n\n\t\thalf := make(tally.DurationBuckets, 0, (len(bucketsToCompress)/2)+2)\n\t\thalf = append(half, dup.tallyBuckets[0]) // keep the zero value intact\n\t\tfor i := 0; i < len(bucketsToCompress); i += 2 {\n\t\t\thalf = append(half, bucketsToCompress[i]) // keep the first, third, etc\n\t\t}\n\t\thalf = append(half, dup.tallyBuckets[len(dup.tallyBuckets)-1]) // keep the last 2^N intact\n\n\t\tdup.tallyBuckets = half\n\t\tdup.scale--\n\t}\n\treturn dup\n}\n\nfunc (i IntSubsettableHistogram) subsetTo(newScale int) IntSubsettableHistogram {\n\treturn IntSubsettableHistogram(SubsettableHistogram(i).subsetTo(newScale))\n}\n\nfunc (s SubsettableHistogram) tags() map[string]string {\n\treturn map[string]string{\n\t\t// use the duration string func, as \"1ms\" duration suffix is clearer than \"1000000\" for nanoseconds.\n\t\t// this also helps differentiate \"tracks time\" from \"may be a general value distribution\",\n\t\t// both visually when querying by hand and for any future automation (if needed).\n\t\t\"histogram_start\": s.start().String(),\n\t\t\"histogram_end\":   s.end().String(),\n\t\t\"histogram_scale\": strconv.Itoa(s.scale),\n\t}\n}\n\nfunc (i IntSubsettableHistogram) tags() map[string]string {\n\treturn map[string]string{\n\t\t\"histogram_start\": strconv.Itoa(int(i.start())),\n\t\t\"histogram_end\":   strconv.Itoa(int(i.end())),\n\t\t\"histogram_scale\": strconv.Itoa(i.scale),\n\t}\n}\n\n// makeSubsettableHistogram is a replacement for tally.MustMakeExponentialDurationBuckets,\n// tailored to make \"construct a range\" or \"ensure N buckets\" simpler for OTEL-compatible exponential histograms\n// (i.e. https://opentelemetry.io/docs/specs/otel/metrics/data-model/#exponentialhistogram).\n//\n// It ensures values are as precise as possible (preventing display-space-wasteful numbers like 3.99996ms),\n// that all histograms start with a 0 value (else erroneous negative values are impossible to notice),\n// and avoids some bugs with low values (tally.MustMakeExponentialDurationBuckets misbehaves for small numbers,\n// causing all buckets to == the minimum value).\n//\n// # To use\n//\n// Choose scale/start/end values that match your *intent*, i.e. the range you want to capture, and then choose\n// a length that EXCEEDS that by at least 2x, ideally 4x-10x, and ends at a highly-divisible-by-2 value.\n//\n// Length should ideally be a highly-divisible-by-2 value, to keep subsetting \"clean\" as long as possible,\n// i.e. the ranges of values in each bucket stays the same rather than having the second-to-last one cover\n// a smaller range.\n//\n// This length will be padded internally to add both a zero value and a next-power-of-2 value, to help keep\n// negative values identifiable, and keep the upper limit unchanged across all scales (as it cannot cleanly\n// subset, because it holds an infinite range of values).\n//\n// The exact length / exceeding / etc does not matter (just write a test so it's documented and can be reviewed),\n// it just serves to document your intent and to make sure that we have some head-room so we can understand how\n// wrongly we guessed at our needs if it's exceeded during an incident of some kind.  We've failed at this\n// multiple times in the past, it's worth paying a bit extra to be able to diagnose problems.\n//\n// For all histograms produced, add a test to print the concrete values, so they can be quickly checked when reading.\nfunc makeSubsettableHistogram(scale int, start, end time.Duration, length int) SubsettableHistogram {\n\tif scale < 0 || scale > 3 {\n\t\t// anything outside this range is currently not expected and probably a mistake,\n\t\t// but any value is technically sound.\n\t\tpanic(fmt.Sprintf(\"scale must be between 0 (grows by *2) and 3 (grows by *2^1/8), got scale: %v\", scale))\n\t}\n\tif start <= 0 {\n\t\tpanic(fmt.Sprintf(\"start must be greater than 0 or it will not grow exponentially, got %v\", start))\n\t}\n\tif start >= end {\n\t\tpanic(fmt.Sprintf(\"start must be less than end (%v < %v)\", start, end))\n\t}\n\tif length < 12 || length > 160 {\n\t\t// 160 is probably higher than we should consider, but going further is currently risking much too high costs.\n\t\t// if this changes, e.g. due to metrics optimizations, just adjust this validation.\n\t\t//\n\t\t// 12 is pretty arbitrary, I just don't expect us to ever make a histogram that small, so it's probably\n\t\t// from accidentally passing scale or something to the wrong argument.\n\t\tpanic(fmt.Sprintf(\"length must be between 12 < %d <=160\", length))\n\t}\n\t// make sure the number of buckets completes a \"full\" row,\n\t// to which we will add one more to reach the next start*2^N value.\n\t//\n\t// for a more visual example of what this means, see the logged strings in tests:\n\t// each \"row\" of values must be the same width, and it must have a single value in the last row.\n\t//\n\t// adding a couple buckets to ensure this is met costs very little, and ensures\n\t// subsetting combines values more consistently (not crossing rows) for longer.\n\tpowerOfTwoWidth := int(math.Pow(2, float64(scale))) // num of buckets needed to double a value\n\tmissing := length % powerOfTwoWidth\n\tif missing != 0 {\n\t\tpanic(fmt.Sprintf(`number of buckets must end at a power of 2 of the starting value.  `+\n\t\t\t`got %d, probably raise to %d`,\n\t\t\tlength, length+missing,\n\t\t))\n\t}\n\n\tvar buckets tally.DurationBuckets\n\tfor i := 0; i < length; i++ {\n\t\tbuckets = append(buckets, nextBucket(start, len(buckets), scale))\n\t}\n\t// the loop above has \"filled\" a full row, we need one more to reach the next power of 2.\n\tbuckets = append(buckets, nextBucket(start, len(buckets), scale))\n\n\tif last(buckets) < end*2 {\n\t\tpanic(fmt.Sprintf(\"not enough buckets (%d) to exceed the end target (%v) by at least 2x. \"+\n\t\t\t\"you are STRONGLY encouraged to include ~2x-10x more range than your intended end value, \"+\n\t\t\t\"preferably 4x+, because we almost always underestimate the ranges needed during incidents\",\n\t\t\tlength, last(buckets)),\n\t\t)\n\t}\n\n\treturn SubsettableHistogram{\n\t\ttallyBuckets: append(\n\t\t\t// always include a zero value at the beginning, so negative values are noticeable (\"-inf to 0\" bucket)\n\t\t\ttally.DurationBuckets{0},\n\t\t\tbuckets...,\n\t\t),\n\t\tscale: scale,\n\t}\n}\n\n// last-item-in-slice helper to eliminate some magic `-1`s\nfunc last[T any, X ~[]T](s X) T {\n\treturn s[len(s)-1]\n}\n\nfunc nextBucket(start time.Duration, num int, scale int) time.Duration {\n\t// calculating it from `start` each time reduces floating point error, ensuring \"clean\" multiples\n\t// at every power of 2 (and possibly others), instead of e.g. \"1ms ... 1.9999994ms\" which occurs\n\t// if you try to build from the previous value each time.\n\treturn time.Duration(\n\t\tfloat64(start) *\n\t\t\tmath.Pow(2, float64(num)/math.Pow(2, float64(scale))))\n}\n\nfunc (s SubsettableHistogram) histScale() int                 { return s.scale }\nfunc (s SubsettableHistogram) width() int                     { return int(math.Pow(2, float64(s.scale))) }\nfunc (s SubsettableHistogram) len() int                       { return len(s.tallyBuckets) }\nfunc (s SubsettableHistogram) start() time.Duration           { return s.tallyBuckets[1] }\nfunc (s SubsettableHistogram) end() time.Duration             { return s.tallyBuckets[len(s.tallyBuckets)-1] }\nfunc (s SubsettableHistogram) buckets() tally.DurationBuckets { return s.tallyBuckets }\nfunc (s SubsettableHistogram) print(to func(string, ...any)) {\n\tto(\"%v\\n\", s.tallyBuckets[0:1]) // zero value on its own row\n\tfor rowStart := 1; rowStart < s.len(); rowStart += s.width() {\n\t\tto(\"%v\\n\", s.tallyBuckets[rowStart:min(rowStart+s.width(), len(s.tallyBuckets))])\n\t}\n}\n\nfunc (i IntSubsettableHistogram) histScale() int       { return i.scale }\nfunc (i IntSubsettableHistogram) width() int           { return int(math.Pow(2, float64(i.scale))) }\nfunc (i IntSubsettableHistogram) len() int             { return len(i.tallyBuckets) }\nfunc (i IntSubsettableHistogram) start() time.Duration { return i.tallyBuckets[1] }\nfunc (i IntSubsettableHistogram) end() time.Duration {\n\treturn i.tallyBuckets[len(i.tallyBuckets)-1]\n}\nfunc (i IntSubsettableHistogram) buckets() tally.DurationBuckets { return i.tallyBuckets }\nfunc (i IntSubsettableHistogram) print(to func(string, ...any)) {\n\t// fairly unreadable as duration-strings, so convert to int by hand\n\tto(\"[%d]\\n\", int(i.tallyBuckets[0])) // zero value on its own row\n\tfor rowStart := 1; rowStart < i.len(); rowStart += i.width() {\n\t\tints := make([]int, 0, i.width())\n\t\tfor _, d := range i.tallyBuckets[rowStart:min(rowStart+i.width(), len(i.tallyBuckets))] {\n\t\t\tints = append(ints, int(d))\n\t\t}\n\t\tto(\"%v\\n\", ints)\n\t}\n}\n\nvar _ histogrammy[SubsettableHistogram] = SubsettableHistogram{}\nvar _ histogrammy[IntSubsettableHistogram] = IntSubsettableHistogram{}\n\n// internal utility/test methods, but could be exposed if there's a use for it\ntype histogrammy[T any] interface {\n\thistScale() int                 // exponential scale value.  0..3 inclusive.\n\twidth() int                     // number of values per power of 2 == how wide to print each row.  1, 2, 4, or 8.\n\tlen() int                       // number of buckets\n\tstart() time.Duration           // first non-zero bucket\n\tend() time.Duration             // last bucket\n\tbuckets() tally.DurationBuckets // access to all buckets\n\tsubsetTo(newScale int) T        // generic so specific types can be returned\n\ttags() map[string]string        // start, end, and scale tags that need to be implicitly added\n\n\tprint(to func(string, ...any)) // test-oriented printer\n}\n"
  },
  {
    "path": "common/metrics/histograms_test.go",
    "content": "package metrics\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestHistogramValues(t *testing.T) {\n\tt.Run(\"default_1ms_to_100s\", func(t *testing.T) {\n\t\tcheckHistogram(t, Default1ms100s, `\n[0s]\n[1ms 1.189207ms 1.414213ms 1.681792ms]\n[2ms 2.378414ms 2.828427ms 3.363585ms]\n[4ms 4.756828ms 5.656854ms 6.727171ms]\n[8ms 9.513656ms 11.313708ms 13.454342ms]\n[16ms 19.027313ms 22.627416ms 26.908685ms]\n[32ms 38.054627ms 45.254833ms 53.81737ms]\n[64ms 76.109255ms 90.509667ms 107.634741ms]\n[128ms 152.21851ms 181.019335ms 215.269482ms]\n[256ms 304.437021ms 362.038671ms 430.538964ms]\n[512ms 608.874042ms 724.077343ms 861.077929ms]\n[1.024s 1.217748085s 1.448154687s 1.722155858s]\n[2.048s 2.435496171s 2.896309375s 3.444311716s]\n[4.096s 4.870992343s 5.792618751s 6.888623433s]\n[8.192s 9.741984686s 11.585237502s 13.777246867s]\n[16.384s 19.483969372s 23.170475005s 27.554493735s]\n[32.768s 38.967938744s 46.340950011s 55.10898747s]\n[1m5.536s 1m17.935877488s 1m32.681900023s 1m50.21797494s]\n[2m11.072s 2m35.871754977s 3m5.363800047s 3m40.43594988s]\n[4m22.144s 5m11.743509955s 6m10.727600094s 7m20.87189976s]\n[8m44.288s 10m23.48701991s 12m21.455200189s 14m41.743799521s]\n[17m28.576s]\n`)\n\t})\n\tt.Run(\"low_1ms_to_100s\", func(t *testing.T) {\n\t\tcheckHistogram(t, Low1ms100s, `\n[0s]\n[1ms 1.414213ms]\n[2ms 2.828427ms]\n[4ms 5.656854ms]\n[8ms 11.313708ms]\n[16ms 22.627416ms]\n[32ms 45.254833ms]\n[64ms 90.509667ms]\n[128ms 181.019335ms]\n[256ms 362.038671ms]\n[512ms 724.077343ms]\n[1.024s 1.448154687s]\n[2.048s 2.896309375s]\n[4.096s 5.792618751s]\n[8.192s 11.585237502s]\n[16.384s 23.170475005s]\n[32.768s 46.340950011s]\n[1m5.536s 1m32.681900023s]\n[2m11.072s 3m5.363800047s]\n[4m22.144s 6m10.727600094s]\n[8m44.288s 12m21.455200189s]\n[17m28.576s]\n`)\n\t})\n\tt.Run(\"high_1ms_to_24h\", func(t *testing.T) {\n\t\tcheckHistogram(t, High1ms24h, `\n[0s]\n[1ms 1.189207ms 1.414213ms 1.681792ms]\n[2ms 2.378414ms 2.828427ms 3.363585ms]\n[4ms 4.756828ms 5.656854ms 6.727171ms]\n[8ms 9.513656ms 11.313708ms 13.454342ms]\n[16ms 19.027313ms 22.627416ms 26.908685ms]\n[32ms 38.054627ms 45.254833ms 53.81737ms]\n[64ms 76.109255ms 90.509667ms 107.634741ms]\n[128ms 152.21851ms 181.019335ms 215.269482ms]\n[256ms 304.437021ms 362.038671ms 430.538964ms]\n[512ms 608.874042ms 724.077343ms 861.077929ms]\n[1.024s 1.217748085s 1.448154687s 1.722155858s]\n[2.048s 2.435496171s 2.896309375s 3.444311716s]\n[4.096s 4.870992343s 5.792618751s 6.888623433s]\n[8.192s 9.741984686s 11.585237502s 13.777246867s]\n[16.384s 19.483969372s 23.170475005s 27.554493735s]\n[32.768s 38.967938744s 46.340950011s 55.10898747s]\n[1m5.536s 1m17.935877488s 1m32.681900023s 1m50.21797494s]\n[2m11.072s 2m35.871754977s 3m5.363800047s 3m40.43594988s]\n[4m22.144s 5m11.743509955s 6m10.727600094s 7m20.87189976s]\n[8m44.288s 10m23.48701991s 12m21.455200189s 14m41.743799521s]\n[17m28.576s 20m46.974039821s 24m42.910400378s 29m23.487599042s]\n[34m57.152s 41m33.948079642s 49m25.820800757s 58m46.975198084s]\n[1h9m54.304s 1h23m7.896159284s 1h38m51.641601515s 1h57m33.950396168s]\n[2h19m48.608s 2h46m15.792318568s 3h17m43.283203031s 3h55m7.900792337s]\n[4h39m37.216s 5h32m31.584637137s 6h35m26.566406062s 7h50m15.801584674s]\n[9h19m14.432s 11h5m3.169274274s 13h10m53.132812125s 15h40m31.603169349s]\n[18h38m28.864s 22h10m6.338548549s 26h21m46.265624251s 31h21m3.206338698s]\n[37h16m57.728s 44h20m12.677097099s 52h43m32.531248503s 62h42m6.412677396s]\n[74h33m55.456s]\n`)\n\t})\n\tt.Run(\"mid_1ms_24h\", func(t *testing.T) {\n\t\tcheckHistogram(t, Mid1ms24h, `\n[0s]\n[1ms 1.414213ms]\n[2ms 2.828427ms]\n[4ms 5.656854ms]\n[8ms 11.313708ms]\n[16ms 22.627416ms]\n[32ms 45.254833ms]\n[64ms 90.509667ms]\n[128ms 181.019335ms]\n[256ms 362.038671ms]\n[512ms 724.077343ms]\n[1.024s 1.448154687s]\n[2.048s 2.896309375s]\n[4.096s 5.792618751s]\n[8.192s 11.585237502s]\n[16.384s 23.170475005s]\n[32.768s 46.340950011s]\n[1m5.536s 1m32.681900023s]\n[2m11.072s 3m5.363800047s]\n[4m22.144s 6m10.727600094s]\n[8m44.288s 12m21.455200189s]\n[17m28.576s 24m42.910400378s]\n[34m57.152s 49m25.820800757s]\n[1h9m54.304s 1h38m51.641601515s]\n[2h19m48.608s 3h17m43.283203031s]\n[4h39m37.216s 6h35m26.566406062s]\n[9h19m14.432s 13h10m53.132812125s]\n[18h38m28.864s 26h21m46.265624251s]\n[37h16m57.728s 52h43m32.531248503s]\n[74h33m55.456s]\n`)\n\t})\n\tt.Run(\"mid_to_16k_ints\", func(t *testing.T) {\n\t\tcheckHistogram(t, Mid1To16k, `\n[0]\n[8 9 11 13]\n[16 19 22 26]\n[32 38 45 53]\n[64 76 90 107]\n[128 152 181 215]\n[256 304 362 430]\n[512 608 724 861]\n[1024 1217 1448 1722]\n[2048 2435 2896 3444]\n[4096 4870 5792 6888]\n[8192 9741 11585 13777]\n[16384 19483 23170 27554]\n[32768 38967 46340 55108]\n[65536 77935 92681 110217]\n[131072 155871 185363 220435]\n[262144 311743 370727 440871]\n[524288]\n`)\n\t})\n}\n\n// most histograms should pass this check, but fuzzy comparison is fine if needed for extreme cases.\nfunc checkHistogram[T any](t *testing.T, h histogrammy[T], expected string) {\n\tvar buf strings.Builder\n\th.print(func(s string, a ...any) {\n\t\tstr := fmt.Sprintf(s, a...)\n\t\tt.Logf(str)\n\t\tbuf.WriteString(str)\n\t})\n\tif strings.TrimSpace(expected) != strings.TrimSpace(buf.String()) {\n\t\tt.Error(\"histogram definition changed, update the test if this is intended\")\n\t}\n\n\tbuckets := h.buckets()\n\tassert.EqualValues(t, 0, buckets[0], \"first bucket should always be zero\")\n\tfor i := 1; i < len(buckets); i += h.width() {\n\t\tif i > 1 {\n\t\t\t// ensure good float math.\n\t\t\t//\n\t\t\t// this is *intentionally* doing exact comparisons, as floating point math with\n\t\t\t// human-friendly integers have precise power-of-2 multiples for a very long time.\n\t\t\t//\n\t\t\t// note that the equivalent tally buckets, e.g.:\n\t\t\t//     tally.MustMakeExponentialDurationBuckets(time.Millisecond, math.Pow(2, 1.0/4.0))\n\t\t\t// fails this test, and the logs produced show ugly e.g. 31.999942ms values.\n\t\t\t// tally also produces incorrect results if you start at e.g. 1,\n\t\t\t// as it just produces endless `1` values.\n\t\t\tassert.Equalf(t, buckets[i-h.width()]*2, buckets[i],\n\t\t\t\t\"current row's value (%v) is not a power of 2 greater than previous (%v), skewed / bad math?\",\n\t\t\t\tbuckets[i-h.width()], buckets[i])\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/metrics/interfaces.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\ntype (\n\t// Client is  the interface used to report metrics tally.\n\tClient interface {\n\t\t// IncCounter increments a counter metric\n\t\tIncCounter(scope ScopeIdx, counter MetricIdx)\n\t\t// AddCounter adds delta to the counter metric\n\t\tAddCounter(scope ScopeIdx, counter MetricIdx, delta int64)\n\t\t// StartTimer starts a timer for the given\n\t\t// metric name. Time will be recorded when stopwatch is stopped.\n\t\tStartTimer(scope ScopeIdx, timer MetricIdx) tally.Stopwatch\n\t\t// RecordTimer starts a timer for the given\n\t\t// metric name\n\t\tRecordTimer(scope ScopeIdx, timer MetricIdx, d time.Duration)\n\t\t// RecordHistogramDuration records a histogram duration value for the given\n\t\t// metric name\n\t\tRecordHistogramDuration(scope ScopeIdx, timer MetricIdx, d time.Duration)\n\t\t// UpdateGauge reports Gauge type absolute value metric\n\t\tUpdateGauge(scope ScopeIdx, gauge MetricIdx, value float64)\n\t\t// Scope return an internal scope that can be used to add additional\n\t\t// information to metrics\n\t\tScope(scope ScopeIdx, tags ...Tag) Scope\n\t}\n\n\t// Scope is an interface for metrics\n\tScope interface {\n\t\t// IncCounter increments a counter metric\n\t\tIncCounter(counter MetricIdx)\n\t\t// AddCounter adds delta to the counter metric\n\t\tAddCounter(counter MetricIdx, delta int64)\n\t\t// StartTimer starts a timer for the given metric name.\n\t\t// Time will be recorded when stopwatch is stopped.\n\t\tStartTimer(timer MetricIdx) Stopwatch\n\t\t// RecordTimer starts a timer for the given metric name\n\t\tRecordTimer(timer MetricIdx, d time.Duration)\n\t\t// RecordHistogramDuration records a histogram duration value for the given\n\t\t// metric name\n\t\tRecordHistogramDuration(timer MetricIdx, d time.Duration)\n\t\t// RecordHistogramValue records a histogram value for the given metric name\n\t\tRecordHistogramValue(timer MetricIdx, value float64)\n\t\t// ExponentialHistogram records a subsettable exponential histogram value for the given metric name\n\t\tExponentialHistogram(hist MetricIdx, d time.Duration)\n\t\t// IntExponentialHistogram records a subsettable exponential histogram value for the given metric name\n\t\tIntExponentialHistogram(hist MetricIdx, value int)\n\t\t// UpdateGauge reports Gauge type absolute value metric\n\t\tUpdateGauge(gauge MetricIdx, value float64)\n\t\t// Tagged return an internal scope that can be used to add additional\n\t\t// information to metrics\n\t\tTagged(tags ...Tag) Scope\n\t}\n)\n\nvar sanitizer = tally.NewSanitizer(tally.SanitizeOptions{\n\tNameCharacters:       tally.ValidCharacters{tally.AlphanumericRange, tally.UnderscoreCharacters},\n\tKeyCharacters:        tally.ValidCharacters{tally.AlphanumericRange, tally.UnderscoreCharacters},\n\tValueCharacters:      tally.ValidCharacters{tally.AlphanumericRange, tally.UnderscoreCharacters},\n\tReplacementCharacter: '_',\n})\n"
  },
  {
    "path": "common/metrics/metricsfx/metricsfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metricsfx\n\nimport (\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// Module provides metrics client for fx application.\nvar Module = fx.Module(\"metricsfx\",\n\tfx.Provide(buildClient))\n\n// ModuleForExternalScope provides metrics client for fx application when tally.Scope is created outside.\nvar ModuleForExternalScope = fx.Module(\"metricsfx\",\n\tfx.Provide(func(params serviceIdxParams) metrics.ServiceIdx {\n\t\treturn service.GetMetricsServiceIdx(params.ServiceFullName, params.Logger)\n\t}),\n\tfx.Provide(buildClientFromTally))\n\ntype clientParams struct {\n\tfx.In\n\n\tLogger          log.Logger\n\tServiceFullName string `name:\"service-full-name\"`\n\tSvcCfg          config.Service\n\tHistogramCfg    metrics.HistogramMigration\n}\n\ntype clientResult struct {\n\tfx.Out\n\n\tScope  tally.Scope\n\tClient metrics.Client\n}\n\nfunc buildClient(params clientParams) clientResult {\n\tscope := params.SvcCfg.Metrics.NewScope(params.Logger, params.ServiceFullName)\n\treturn clientResult{\n\t\tScope:  scope,\n\t\tClient: buildClientFromTally(scope, service.GetMetricsServiceIdx(params.ServiceFullName, params.Logger), params.HistogramCfg),\n\t}\n}\n\ntype serviceIdxParams struct {\n\tfx.In\n\n\tLogger          log.Logger\n\tServiceFullName string `name:\"service-full-name\"`\n}\n\nfunc buildClientFromTally(scope tally.Scope, serviceID metrics.ServiceIdx, hm metrics.HistogramMigration) metrics.Client {\n\treturn metrics.NewClient(scope, serviceID, hm)\n}\n"
  },
  {
    "path": "common/metrics/metricsfx/metricsfx_test.go",
    "content": "package metricsfx\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestModule(t *testing.T) {\n\tfxApp := fxtest.New(t,\n\t\ttestlogger.Module(t),\n\t\tfx.Provide(fx.Annotated{\n\t\t\tTarget: func() string { return service.Frontend },\n\t\t\tName:   \"service-full-name\"},\n\t\t\tfunc() config.Service {\n\t\t\t\treturn config.Service{}\n\t\t\t}),\n\t\tfx.Provide(func() metrics.HistogramMigration { return metrics.HistogramMigration{} }),\n\t\tModule,\n\t\tfx.Invoke(func(mc metrics.Client) {}))\n\tfxApp.RequireStart().RequireStop()\n}\n\nfunc TestModuleWithExternalScope(t *testing.T) {\n\tfxApp := fxtest.New(t,\n\t\ttestlogger.Module(t),\n\t\tfx.Provide(func() tally.Scope { return tally.NoopScope },\n\t\t\tfx.Annotated{\n\t\t\t\tTarget: func() string { return service.Frontend },\n\t\t\t\tName:   \"service-full-name\"}),\n\t\tfx.Provide(func() metrics.HistogramMigration { return metrics.HistogramMigration{} }),\n\t\tModuleForExternalScope,\n\t\tfx.Invoke(func(mc metrics.Client) {}))\n\tfxApp.RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "common/metrics/mocks/Client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n// Code generated by mockery 2.7.4. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\ttime \"time\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\ttally \"github.com/uber-go/tally\"\n\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n)\n\n// Client is an autogenerated mock type for the Client type\ntype Client struct {\n\tmock.Mock\n}\n\n// AddCounter provides a mock function with given fields: scope, counter, delta\nfunc (_m *Client) AddCounter(scope metrics.ScopeIdx, counter metrics.MetricIdx, delta int64) {\n\t_m.Called(scope, counter, delta)\n}\n\n// IncCounter provides a mock function with given fields: scope, counter\nfunc (_m *Client) IncCounter(scope metrics.ScopeIdx, counter metrics.MetricIdx) {\n\t_m.Called(scope, counter)\n}\n\n// RecordHistogramDuration provides a mock function with given fields: scope, timer, d\nfunc (_m *Client) RecordHistogramDuration(scope metrics.ScopeIdx, timer metrics.MetricIdx, d time.Duration) {\n\t_m.Called(scope, timer, d)\n}\n\n// RecordTimer provides a mock function with given fields: scope, timer, d\nfunc (_m *Client) RecordTimer(scope metrics.ScopeIdx, timer metrics.MetricIdx, d time.Duration) {\n\t_m.Called(scope, timer, d)\n}\n\n// Scope provides a mock function with given fields: scope, tags\nfunc (_m *Client) Scope(scope metrics.ScopeIdx, tags ...metrics.Tag) metrics.Scope {\n\t_va := make([]interface{}, len(tags))\n\tfor _i := range tags {\n\t\t_va[_i] = tags[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, scope)\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tvar r0 metrics.Scope\n\tif rf, ok := ret.Get(0).(func(metrics.ScopeIdx, ...metrics.Tag) metrics.Scope); ok {\n\t\tr0 = rf(scope, tags...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(metrics.Scope)\n\t\t}\n\t}\n\n\treturn r0\n}\n\n// StartTimer provides a mock function with given fields: scope, timer\nfunc (_m *Client) StartTimer(scope metrics.ScopeIdx, timer metrics.MetricIdx) tally.Stopwatch {\n\tret := _m.Called(scope, timer)\n\n\tvar r0 tally.Stopwatch\n\tif rf, ok := ret.Get(0).(func(metrics.ScopeIdx, metrics.MetricIdx) tally.Stopwatch); ok {\n\t\tr0 = rf(scope, timer)\n\t} else {\n\t\tr0 = ret.Get(0).(tally.Stopwatch)\n\t}\n\n\treturn r0\n}\n\n// UpdateGauge provides a mock function with given fields: scope, gauge, value\nfunc (_m *Client) UpdateGauge(scope metrics.ScopeIdx, gauge metrics.MetricIdx, value float64) {\n\t_m.Called(scope, gauge, value)\n}\n"
  },
  {
    "path": "common/metrics/mocks/Scope.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n// Code generated by mockery v1.0.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\ttime \"time\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n)\n\n// Scope is an autogenerated mock type for the Scope type\ntype Scope struct {\n\tmock.Mock\n}\n\n// AddCounter provides a mock function with given fields: counter, delta\nfunc (_m *Scope) AddCounter(counter metrics.MetricIdx, delta int64) {\n\t_m.Called(counter, delta)\n}\n\n// IncCounter provides a mock function with given fields: counter\nfunc (_m *Scope) IncCounter(counter metrics.MetricIdx) {\n\t_m.Called(counter)\n}\n\n// RecordHistogramDuration provides a mock function with given fields: timer, d\nfunc (_m *Scope) RecordHistogramDuration(timer metrics.MetricIdx, d time.Duration) {\n\t_m.Called(timer, d)\n}\n\n// RecordHistogramValue provides a mock function with given fields: timer, value\nfunc (_m *Scope) RecordHistogramValue(timer metrics.MetricIdx, value float64) {\n\t_m.Called(timer, value)\n}\n\nfunc (_m *Scope) ExponentialHistogram(hist metrics.MetricIdx, d time.Duration) {\n\t_m.Called(hist, d)\n}\n\nfunc (_m *Scope) IntExponentialHistogram(hist metrics.MetricIdx, value int) {\n\t_m.Called(hist, value)\n}\n\n// RecordTimer provides a mock function with given fields: timer, d\nfunc (_m *Scope) RecordTimer(timer metrics.MetricIdx, d time.Duration) {\n\t_m.Called(timer, d)\n}\n\n// StartTimer provides a mock function with given fields: timer\nfunc (_m *Scope) StartTimer(timer metrics.MetricIdx) metrics.Stopwatch {\n\tret := _m.Called(timer)\n\n\tvar r0 metrics.Stopwatch\n\tif rf, ok := ret.Get(0).(func(idx metrics.MetricIdx) metrics.Stopwatch); ok {\n\t\tr0 = rf(timer)\n\t} else {\n\t\tr0 = ret.Get(0).(metrics.Stopwatch)\n\t}\n\n\treturn r0\n}\n\n// Tagged provides a mock function with given fields: tags\nfunc (_m *Scope) Tagged(tags ...metrics.Tag) metrics.Scope {\n\t_va := make([]interface{}, len(tags))\n\tfor _i := range tags {\n\t\t_va[_i] = tags[_i]\n\t}\n\tvar _ca []interface{}\n\t_ca = append(_ca, _va...)\n\tret := _m.Called(_ca...)\n\n\tvar r0 metrics.Scope\n\tif rf, ok := ret.Get(0).(func(...metrics.Tag) metrics.Scope); ok {\n\t\tr0 = rf(tags...)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(metrics.Scope)\n\t\t}\n\t}\n\n\treturn r0\n}\n\n// UpdateGauge provides a mock function with given fields: gauge, value\nfunc (_m *Scope) UpdateGauge(gauge metrics.MetricIdx, value float64) {\n\t_m.Called(gauge, value)\n}\n"
  },
  {
    "path": "common/metrics/nop.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\nvar (\n\tNoopClient    Client = &noopClientImpl{}\n\tNoopScope     Scope  = &noopScopeImpl{}\n\tNoopStopwatch        = tally.NewStopwatch(time.Now(), &nopStopwatchRecorder{})\n)\n\ntype nopStopwatchRecorder struct{}\n\n// RecordStopwatch is a nop impl for replay mode\nfunc (n *nopStopwatchRecorder) RecordStopwatch(stopwatchStart time.Time) {}\n\n// NopStopwatch return a fake tally stop watch\nfunc NopStopwatch() tally.Stopwatch {\n\treturn tally.NewStopwatch(time.Now(), &nopStopwatchRecorder{})\n}\n\ntype noopClientImpl struct{}\n\nfunc (n noopClientImpl) IncCounter(scope ScopeIdx, counter MetricIdx) {}\n\nfunc (n noopClientImpl) AddCounter(scope ScopeIdx, counter MetricIdx, delta int64) {}\n\nfunc (n noopClientImpl) StartTimer(scope ScopeIdx, timer MetricIdx) tally.Stopwatch {\n\treturn NoopStopwatch\n}\n\nfunc (n noopClientImpl) RecordTimer(scope ScopeIdx, timer MetricIdx, d time.Duration) {}\n\nfunc (n *noopClientImpl) RecordHistogramDuration(scope ScopeIdx, timer MetricIdx, d time.Duration) {}\n\nfunc (n noopClientImpl) UpdateGauge(scope ScopeIdx, gauge MetricIdx, value float64) {}\n\nfunc (n noopClientImpl) Scope(scope ScopeIdx, tags ...Tag) Scope {\n\treturn NoopScope\n}\n\n// NewNoopMetricsClient initialize new no-op metrics client\nfunc NewNoopMetricsClient() Client {\n\treturn &noopClientImpl{}\n}\n\ntype noopScopeImpl struct{}\n\nfunc (n *noopScopeImpl) IncCounter(counter MetricIdx) {}\n\nfunc (n *noopScopeImpl) AddCounter(counter MetricIdx, delta int64) {}\n\nfunc (n *noopScopeImpl) StartTimer(timer MetricIdx) Stopwatch {\n\treturn NewTestStopwatch()\n}\n\nfunc (n *noopScopeImpl) RecordTimer(timer MetricIdx, d time.Duration) {}\n\nfunc (n *noopScopeImpl) RecordHistogramDuration(timer MetricIdx, d time.Duration) {}\n\nfunc (n *noopScopeImpl) RecordHistogramValue(timer MetricIdx, value float64) {}\n\nfunc (n *noopScopeImpl) ExponentialHistogram(hist MetricIdx, d time.Duration) {}\n\nfunc (n *noopScopeImpl) IntExponentialHistogram(hist MetricIdx, value int) {}\n\nfunc (n *noopScopeImpl) UpdateGauge(gauge MetricIdx, value float64) {}\n\nfunc (n *noopScopeImpl) Tagged(tags ...Tag) Scope {\n\treturn n\n}\n"
  },
  {
    "path": "common/metrics/runtime.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"runtime\"\n\t\"strconv\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\nvar (\n\t// Revision is the VCS revision associated with this build. Overridden using ldflags\n\t// at compile time. Example:\n\t// $ go build -ldflags \"-X github.com/uber/cadence/common/metrics.Revision=abcdef\" ...\n\t// see get-ldflags.sh for GIT_REVISION\n\tRevision = \"unknown\"\n\n\t// Branch is the VCS branch associated with this build.\n\tBranch = \"unknown\"\n\n\t// ReleaseVersion is the version associated with this build.\n\tReleaseVersion = \"unknown\"\n\n\t// BuildDate is the date this build was created.\n\tBuildDate = \"unknown\"\n\n\t// BuildTimeUnix is the seconds since epoch representing the date this build was created.\n\tBuildTimeUnix = \"0\"\n\n\t// goVersion is the current runtime version.\n\tgoVersion = runtime.Version()\n\t// cadenceVersion is the current version of cadence\n\tcadenceVersion = VersionString\n)\n\nconst (\n\t// buildInfoMetricName is the emitted build information metric's name.\n\tbuildInfoMetricName = \"build_information\"\n\n\t// buildAgeMetricName is the emitted build age metric's name.\n\tbuildAgeMetricName = \"build_age\"\n)\n\n// RuntimeMetricsReporter A struct containing the state of the RuntimeMetricsReporter.\ntype RuntimeMetricsReporter struct {\n\tscope          tally.Scope\n\tbuildInfoScope tally.Scope\n\treportInterval time.Duration\n\tstarted        int32\n\tquit           chan struct{}\n\tlogger         log.Logger\n\tlastNumGC      uint32\n\tbuildTime      time.Time\n}\n\n// NewRuntimeMetricsReporter Creates a new RuntimeMetricsReporter.\nfunc NewRuntimeMetricsReporter(\n\tscope tally.Scope,\n\treportInterval time.Duration,\n\tlogger log.Logger,\n\tinstanceID string,\n) *RuntimeMetricsReporter {\n\tconst (\n\t\tbase    = 10\n\t\tbitSize = 64\n\t)\n\tif len(instanceID) > 0 {\n\t\tscope = scope.Tagged(map[string]string{instance: instanceID})\n\t}\n\tvar memstats runtime.MemStats\n\truntime.ReadMemStats(&memstats)\n\trReporter := &RuntimeMetricsReporter{\n\t\tscope:          scope,\n\t\treportInterval: reportInterval,\n\t\tlogger:         logger,\n\t\tlastNumGC:      memstats.NumGC,\n\t\tquit:           make(chan struct{}),\n\t}\n\trReporter.buildInfoScope = scope.Tagged(\n\t\tmap[string]string{\n\t\t\trevisionTag:       Revision,\n\t\t\tbranchTag:         Branch,\n\t\t\tbuildDateTag:      BuildDate,\n\t\t\tbuildVersionTag:   ReleaseVersion,\n\t\t\tgoVersionTag:      goVersion,\n\t\t\tcadenceVersionTag: cadenceVersion,\n\t\t},\n\t)\n\tsec, err := strconv.ParseInt(BuildTimeUnix, base, bitSize)\n\tif err != nil || sec < 0 {\n\t\tsec = 0\n\t}\n\trReporter.buildTime = time.Unix(sec, 0)\n\treturn rReporter\n}\n\n// report Sends runtime metrics to the local metrics collector.\nfunc (r *RuntimeMetricsReporter) report() {\n\tvar memStats runtime.MemStats\n\truntime.ReadMemStats(&memStats)\n\n\tr.scope.Gauge(NumGoRoutinesGauge).Update(float64(runtime.NumGoroutine()))\n\tr.scope.Gauge(GoMaxProcsGauge).Update(float64(runtime.GOMAXPROCS(0)))\n\tr.scope.Gauge(MemoryAllocatedGauge).Update(float64(memStats.Alloc))\n\tr.scope.Gauge(MemoryHeapGauge).Update(float64(memStats.HeapAlloc))\n\tr.scope.Gauge(MemoryHeapIdleGauge).Update(float64(memStats.HeapIdle))\n\tr.scope.Gauge(MemoryHeapInuseGauge).Update(float64(memStats.HeapInuse))\n\tr.scope.Gauge(MemoryStackGauge).Update(float64(memStats.StackInuse))\n\n\t// memStats.NumGC is a perpetually incrementing counter (unless it wraps at 2^32)\n\tnum := memStats.NumGC\n\tlastNum := atomic.SwapUint32(&r.lastNumGC, num) // reset for the next iteration\n\tif delta := num - lastNum; delta > 0 {\n\t\tr.scope.Counter(NumGCCounter).Inc(int64(delta))\n\t\tif delta > 255 {\n\t\t\t// too many GCs happened, the timestamps buffer got wrapped around. Report only the last 256\n\t\t\tlastNum = num - 256\n\t\t}\n\t\tfor i := lastNum; i != num; i++ {\n\t\t\tpause := memStats.PauseNs[i%256]\n\t\t\tr.scope.Timer(GcPauseMsTimer).Record(time.Duration(pause))\n\t\t}\n\t}\n\n\t// report build info\n\tbuildInfoGauge := r.buildInfoScope.Gauge(buildInfoMetricName)\n\tbuildAgeGauge := r.buildInfoScope.Gauge(buildAgeMetricName)\n\tbuildInfoGauge.Update(1.0)\n\tbuildAgeGauge.Update(float64(time.Since(r.buildTime)))\n}\n\n// Start Starts the reporter thread that periodically emits metrics.\nfunc (r *RuntimeMetricsReporter) Start() {\n\tif !atomic.CompareAndSwapInt32(&r.started, 0, 1) {\n\t\treturn\n\t}\n\tgo func() {\n\t\tticker := time.NewTicker(r.reportInterval)\n\t\tfor {\n\t\t\tselect {\n\t\t\tcase <-ticker.C:\n\t\t\t\tr.report()\n\t\t\tcase <-r.quit:\n\t\t\t\tticker.Stop()\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n\tr.logger.Info(\"RuntimeMetricsReporter started\")\n}\n\n// Stop Stops reporting of runtime metrics. The reporter cannot be started again after it's been stopped.\nfunc (r *RuntimeMetricsReporter) Stop() {\n\tclose(r.quit)\n\tr.logger.Info(\"RuntimeMetricsReporter stopped\")\n}\n"
  },
  {
    "path": "common/metrics/scope.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\ntype metricsScope struct {\n\tscope          tally.Scope\n\trootScope      tally.Scope\n\tdefs           map[MetricIdx]metricDefinition\n\tisDomainTagged bool\n\tconfig         HistogramMigration\n}\n\nfunc newMetricsScope(\n\trootScope tally.Scope,\n\tscope tally.Scope,\n\tdefs map[MetricIdx]metricDefinition,\n\tisDomain bool,\n\tconfig HistogramMigration,\n) Scope {\n\treturn &metricsScope{\n\t\tscope:          scope,\n\t\trootScope:      rootScope,\n\t\tdefs:           defs,\n\t\tisDomainTagged: isDomain,\n\t\tconfig:         config,\n\t}\n}\n\nfunc (m *metricsScope) IncCounter(id MetricIdx) {\n\tm.AddCounter(id, 1)\n}\n\nfunc (m *metricsScope) AddCounter(id MetricIdx, delta int64) {\n\tdef := m.defs[id]\n\tm.scope.Counter(def.metricName.String()).Inc(delta)\n\tif !def.metricRollupName.Empty() {\n\t\tm.rootScope.Counter(def.metricRollupName.String()).Inc(delta)\n\t}\n}\n\nfunc (m *metricsScope) UpdateGauge(id MetricIdx, value float64) {\n\tdef := m.defs[id]\n\tm.scope.Gauge(def.metricName.String()).Update(value)\n\tif !def.metricRollupName.Empty() {\n\t\tm.scope.Gauge(def.metricRollupName.String()).Update(value)\n\t}\n}\n\nfunc (m *metricsScope) StartTimer(id MetricIdx) Stopwatch {\n\tdef := m.defs[id]\n\ttimer := m.scope.Timer(def.metricName.String())\n\tswitch {\n\tcase !def.metricRollupName.Empty():\n\t\tif m.config.EmitTimer(def.metricRollupName.String()) {\n\t\t\treturn NewStopwatch(timer, m.rootScope.Timer(def.metricRollupName.String()))\n\t\t}\n\tcase m.isDomainTagged:\n\t\tif m.config.EmitTimer(def.metricName.String()) {\n\t\t\ttimerAll := m.scope.Tagged(map[string]string{domain: allValue}).Timer(def.metricName.String())\n\t\t\treturn NewStopwatch(timer, timerAll)\n\t\t}\n\tdefault:\n\t\tif m.config.EmitTimer(def.metricName.String()) {\n\t\t\treturn NewStopwatch(timer)\n\t\t}\n\t}\n\treturn Stopwatch{} // noop\n}\n\nfunc (m *metricsScope) RecordTimer(id MetricIdx, d time.Duration) {\n\tdef := m.defs[id]\n\tif m.config.EmitTimer(def.metricName.String()) {\n\t\tm.scope.Timer(def.metricName.String()).Record(d)\n\t}\n\tswitch {\n\tcase !def.metricRollupName.Empty():\n\t\tif m.config.EmitTimer(def.metricRollupName.String()) {\n\t\t\tm.rootScope.Timer(def.metricRollupName.String()).Record(d)\n\t\t}\n\tcase m.isDomainTagged:\n\t\t// N.B. - Dual emit here so that we can see aggregate timer stats across all\n\t\t// domains along with the individual domains stats\n\t\tif m.config.EmitTimer(def.metricName.String()) {\n\t\t\tm.scope.Tagged(map[string]string{domain: allValue}).Timer(def.metricName.String()).Record(d)\n\t\t}\n\t}\n}\n\nfunc (m *metricsScope) RecordHistogramDuration(id MetricIdx, value time.Duration) {\n\tdef := m.defs[id]\n\tif m.config.EmitHistogram(def.metricName.String()) {\n\t\tm.scope.Histogram(def.metricName.String(), m.getBuckets(id)).RecordDuration(value)\n\t}\n\tif !def.metricRollupName.Empty() {\n\t\tif m.config.EmitHistogram(def.metricRollupName.String()) {\n\t\t\tm.rootScope.Histogram(def.metricRollupName.String(), m.getBuckets(id)).RecordDuration(value)\n\t\t}\n\t}\n}\n\nfunc (m *metricsScope) RecordHistogramValue(id MetricIdx, value float64) {\n\tdef := m.defs[id]\n\tif m.config.EmitHistogram(def.metricName.String()) {\n\t\tm.scope.Histogram(def.metricName.String(), m.getBuckets(id)).RecordValue(value)\n\t}\n\tif !def.metricRollupName.Empty() {\n\t\tif m.config.EmitHistogram(def.metricRollupName.String()) {\n\t\t\tm.rootScope.Histogram(def.metricRollupName.String(), m.getBuckets(id)).RecordValue(value)\n\t\t}\n\t}\n}\n\nfunc (m *metricsScope) ExponentialHistogram(id MetricIdx, value time.Duration) {\n\tdef := m.defs[id]\n\tif m.config.EmitHistogram(def.metricName.String()) {\n\t\tm.scope.Tagged(def.exponentialBuckets.tags()).Histogram(def.metricName.String(), def.exponentialBuckets.buckets()).RecordDuration(value)\n\t}\n}\n\nfunc (m *metricsScope) IntExponentialHistogram(id MetricIdx, value int) {\n\tdef := m.defs[id]\n\tif m.config.EmitHistogram(def.metricName.String()) {\n\t\tm.scope.Tagged(def.intExponentialBuckets.tags()).Histogram(def.metricName.String(), def.intExponentialBuckets.buckets()).RecordDuration(time.Duration(value))\n\t}\n}\n\nfunc (m *metricsScope) Tagged(tags ...Tag) Scope {\n\tdomainTagged := m.isDomainTagged\n\ttagMap := make(map[string]string, len(tags))\n\tfor _, tag := range tags {\n\t\tif isDomainTagged(tag) {\n\t\t\tdomainTagged = true\n\t\t}\n\t\ttagMap[tag.Key()] = tag.Value()\n\t}\n\treturn newMetricsScope(m.rootScope, m.scope.Tagged(tagMap), m.defs, domainTagged, m.config)\n}\n\nfunc (m *metricsScope) getBuckets(id MetricIdx) tally.Buckets {\n\tif m.defs[id].buckets != nil {\n\t\treturn m.defs[id].buckets\n\t}\n\treturn tally.DefaultBuckets\n}\n\nfunc isDomainTagged(tag Tag) bool {\n\treturn tag.Key() == domain && tag.Value() != allValue\n}\n"
  },
  {
    "path": "common/metrics/scope_test.go",
    "content": "package metrics\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n)\n\nfunc TestHistogramMode(t *testing.T) {\n\tts := tally.NewTestScope(\"\", nil)\n\tfindName := func(m MetricIdx) string {\n\t\tdef, ok := MetricDefs[Common][m]\n\t\tif ok {\n\t\t\treturn def.metricName.String()\n\t\t}\n\t\tdef, ok = MetricDefs[History][m]\n\t\tif ok {\n\t\t\treturn def.metricName.String()\n\t\t}\n\t\tt.Fatalf(\"MetricDef not found in common or history: %v\", m)\n\t\treturn \"unknown\"\n\t}\n\n\torig := HistogramMigrationMetrics\n\tt.Cleanup(func() {\n\t\tHistogramMigrationMetrics = orig\n\t})\n\n\tHistogramMigrationMetrics = map[string]struct{}{\n\t\tfindName(CadenceLatency):                            {},\n\t\tfindName(ExponentialReplicationTaskLatency):         {},\n\t\tfindName(PersistenceLatencyPerShard):                {},\n\t\tfindName(ExponentialTaskProcessingLatency):          {},\n\t\tfindName(PersistenceLatency):                        {},\n\t\tfindName(PersistenceLatencyHistogram):               {},\n\t\tfindName(TaskAttemptTimer):                          {},\n\t\tfindName(ExponentialTaskAttemptCounts):              {},\n\t\tfindName(TaskQueueLatency):                          {},\n\t\tfindName(ExponentialTaskQueueLatency):               {},\n\t\tfindName(TaskLatencyPerDomain):                      {},\n\t\tfindName(ExponentialTaskLatencyPerDomain):           {},\n\t\tfindName(TaskAttemptTimerPerDomain):                 {},\n\t\tfindName(ExponentialTaskAttemptCountsPerDomain):     {},\n\t\tfindName(TaskProcessingLatencyPerDomain):            {},\n\t\tfindName(ExponentialTaskProcessingLatencyPerDomain): {},\n\t\tfindName(TaskQueueLatencyPerDomain):                 {},\n\t\tfindName(ExponentialTaskQueueLatencyPerDomain):      {},\n\t}\n\n\tc := NewClient(ts, History, HistogramMigration{\n\t\t// Default: ..., left at default value\n\t\tNames: map[string]bool{\n\t\t\tfindName(CadenceLatency):                    true,  // timer type\n\t\t\tfindName(ExponentialReplicationTaskLatency): false, // histogram type\n\n\t\t\tfindName(PersistenceLatencyPerShard):       false, // timer type\n\t\t\tfindName(ExponentialTaskProcessingLatency): true,  // histogram type\n\t\t},\n\t})\n\tscope := c.Scope(HistoryDescribeQueueScope) // scope doesn't matter for this test\n\n\tscope.RecordTimer(CadenceLatency, time.Second)\n\tscope.ExponentialHistogram(ExponentialReplicationTaskLatency, 2*time.Second)\n\n\tscope.RecordTimer(PersistenceLatencyPerShard, 3*time.Second)\n\tscope.ExponentialHistogram(ExponentialTaskProcessingLatency, 4*time.Second)\n\n\t// unspecified -> default config\n\tscope.RecordTimer(PersistenceLatency, 5*time.Second)\n\tscope.RecordHistogramDuration(PersistenceLatencyHistogram, 6*time.Second)\n\n\t// not migrating -> always emit\n\tscope.RecordTimer(CadenceDcRedirectionClientLatency, 7*time.Second)\n\tscope.RecordHistogramDuration(GlobalRatelimiterStartupUsageHistogram, 8*time.Second)\n\n\ts := ts.Snapshot()\n\tfindMetric := func(idx MetricIdx) (timer, histogram bool) {\n\t\tname := findName(idx)\n\t\tfor _, v := range s.Timers() {\n\t\t\tif v.Name() == name {\n\t\t\t\tt.Logf(\"found timer: %v = %v\", v.Name(), v.Values())\n\t\t\t\ttimer = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tfor _, v := range s.Histograms() {\n\t\t\tif v.Name() == findName(idx) {\n\t\t\t\tnzDur := make(map[time.Duration]int64, 1)\n\t\t\t\tfor k, val := range v.Durations() {\n\t\t\t\t\tif val != 0 {\n\t\t\t\t\t\tnzDur[k] = val\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tnzVal := make(map[float64]int64, 1)\n\t\t\t\tfor k, val := range v.Values() {\n\t\t\t\t\tif val != 0 {\n\t\t\t\t\t\tnzVal[k] = val\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tt.Logf(\"found histogram: %v = %v (values: %v)\", v.Name(), nzDur, nzVal)\n\t\t\t\thistogram = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\treturn\n\t}\n\tassertFound := func(idx MetricIdx, timer, histogram bool) {\n\t\tname := findName(idx)\n\t\tfoundTimer, foundHistogram := findMetric(idx)\n\t\tassert.Equalf(t, foundTimer, timer, \"wrong timer behavior for %v\", name)\n\t\tassert.Equalf(t, foundHistogram, histogram, \"wrong histogram behavior for %v\", name)\n\t}\n\n\t// only the timer\n\tassertFound(CadenceLatency, true, false)\n\tassertFound(ExponentialReplicationTaskLatency, false, false)\n\t// only the histogram\n\tassertFound(PersistenceLatencyPerShard, false, false)\n\tassertFound(ExponentialTaskProcessingLatency, false, true)\n\t// timers only (via default)\n\tassertFound(PersistenceLatency, true, false)\n\tassertFound(PersistenceLatencyHistogram, false, false)\n\t// not migrating, the correct type should be emitted\n\tassertFound(CadenceDcRedirectionClientLatency, true, false)\n\tassertFound(GlobalRatelimiterStartupUsageHistogram, false, true)\n\n\t// when fixing: check logs!  you should see metrics with values for: 1, 4, 7, 8\n}\n"
  },
  {
    "path": "common/metrics/stopwatch.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\n// Stopwatch is a helper for simpler tracking of elapsed time, use the\n// Stop() method to report time elapsed since its created back to the\n// timer or histogram.\ntype Stopwatch struct {\n\tstart  time.Time\n\ttimers []tally.Timer\n}\n\n// NewStopwatch creates a new immutable stopwatch for recording the start\n// time to a stopwatch reporter.\nfunc NewStopwatch(timers ...tally.Timer) Stopwatch {\n\treturn Stopwatch{time.Now(), timers}\n}\n\n// NewTestStopwatch returns a new test stopwatch\nfunc NewTestStopwatch() Stopwatch {\n\treturn Stopwatch{time.Now(), []tally.Timer{}}\n}\n\n// Stop reports time elapsed since the stopwatch start to the recorder.\nfunc (sw Stopwatch) Stop() {\n\td := time.Since(sw.start)\n\tfor _, timer := range sw.timers {\n\t\ttimer.Record(d)\n\t}\n}\n"
  },
  {
    "path": "common/metrics/tags.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage metrics\n\nimport (\n\t\"regexp\"\n\t\"strconv\"\n)\n\nconst (\n\trevisionTag       = \"revision\"\n\tbranchTag         = \"branch\"\n\tbuildDateTag      = \"build_date\"\n\tbuildVersionTag   = \"build_version\"\n\tgoVersionTag      = \"go_version\"\n\tcadenceVersionTag = \"cadence_version\"\n\n\tinstance                  = \"instance\"\n\tdomain                    = \"domain\"\n\tdomainType                = \"domain_type\"\n\tclusterGroup              = \"cluster_group\"\n\tsourceCluster             = \"source_cluster\"\n\ttargetCluster             = \"target_cluster\"\n\tactiveCluster             = \"active_cluster\"\n\tisActiveActiveDomain      = \"is_active_active_domain\"\n\ttaskList                  = \"tasklist\"\n\ttaskListType              = \"tasklistType\"\n\ttaskListRootPartition     = \"tasklist_root_partition\"\n\tworkflowType              = \"workflowType\"\n\tactivityType              = \"activityType\"\n\tdecisionType              = \"decisionType\"\n\tinvariantType             = \"invariantType\"\n\tshardScannerScanResult    = \"shardscanner_scan_result\"\n\tshardScannerFixResult     = \"shardscanner_fix_result\"\n\tkafkaPartition            = \"kafkaPartition\"\n\ttransport                 = \"transport\"\n\tcaller                    = \"caller\"\n\tservice                   = \"service\"\n\tdestService               = \"dest_service\"\n\tsignalName                = \"signalName\"\n\tworkflowVersion           = \"workflow_version\"\n\tshardID                   = \"shard_id\"\n\tmatchingHost              = \"matching_host\"\n\thost                      = \"host\"\n\tpollerIsolationGroup      = \"poller_isolation_group\"\n\tasyncWFRequestType        = \"async_wf_request_type\"\n\tworkflowTerminationReason = \"workflow_termination_reason\"\n\tworkflowCloseStatus       = \"workflow_close_status\"\n\tisolationEnabled          = \"isolation_enabled\"\n\tisolationGroup            = \"isolation_group\"\n\tleakCause                 = \"leak_cause\"\n\ttopic                     = \"topic\"\n\tmode                      = \"mode\"\n\tisRetry                   = \"is_retry\"\n\tqueryConsistencyLevel     = \"query_consistency_level\"\n\tbudgetManagerName         = \"budget_manager_name\"\n\n\t// limiter-side tags\n\tglobalRatelimitKey            = \"global_ratelimit_key\"\n\tglobalRatelimitType           = \"global_ratelimit_type\"\n\tglobalRatelimitIsPrimary      = \"is_primary\"\n\tglobalRatelimitCollectionName = \"global_ratelimit_collection\"\n\n\tallValue     = \"all\"\n\tunknownValue = \"_unknown_\"\n)\n\nvar (\n\tsafeAlphaNumericStringRE = regexp.MustCompile(`[^a-zA-Z0-9]`)\n)\n\n// Tag is an interface to define metrics tags\ntype (\n\tTag interface {\n\t\tKey() string\n\t\tValue() string\n\t}\n\n\tsimpleMetric struct {\n\t\tkey   string\n\t\tvalue string\n\t}\n)\n\nfunc (s simpleMetric) Key() string   { return s.key }\nfunc (s simpleMetric) Value() string { return s.value }\n\nfunc metricWithUnknown(key, value string) Tag {\n\tif len(value) == 0 {\n\t\tvalue = unknownValue\n\t}\n\treturn simpleMetric{key: key, value: value}\n}\n\nfunc ShardIDTag(shardIDVal int) Tag {\n\treturn metricWithUnknown(shardID, strconv.Itoa(shardIDVal))\n}\n\n// DomainTag returns a new domain tag. For timers, this also ensures that we\n// dual emit the metric with the all tag. If a blank domain is provided then\n// this converts that to an unknown domain.\nfunc DomainTag(value string) Tag {\n\treturn metricWithUnknown(domain, value)\n}\n\n// DomainTypeTag returns a tag for domain type.\n// This allows differentiating between global/local domains.\nfunc DomainTypeTag(isGlobal bool) Tag {\n\tvar value string\n\tif isGlobal {\n\t\tvalue = \"global\"\n\t} else {\n\t\tvalue = \"local\"\n\t}\n\treturn simpleMetric{key: domainType, value: value}\n}\n\n// DomainUnknownTag returns a new domain:unknown tag-value\nfunc DomainUnknownTag() Tag {\n\treturn DomainTag(\"\")\n}\n\n// ClusterGroupTag return a new cluster group tag\nfunc ClusterGroupTag(value string) Tag {\n\treturn simpleMetric{key: clusterGroup, value: value}\n}\n\n// InstanceTag returns a new instance tag\nfunc InstanceTag(value string) Tag {\n\treturn simpleMetric{key: instance, value: value}\n}\n\n// SourceClusterTag returns a new source cluster tag.\nfunc SourceClusterTag(value string) Tag {\n\treturn metricWithUnknown(sourceCluster, value)\n}\n\n// TargetClusterTag returns a new target cluster tag.\nfunc TargetClusterTag(value string) Tag {\n\treturn metricWithUnknown(targetCluster, value)\n}\n\n// ActiveClusterTag returns a new active cluster type tag.\nfunc ActiveClusterTag(value string) Tag {\n\treturn metricWithUnknown(activeCluster, value)\n}\n\n// IsActiveActiveDomainTag returns a new is active active domain tag.\nfunc IsActiveActiveDomainTag(value bool) Tag {\n\treturn simpleMetric{key: isActiveActiveDomain, value: strconv.FormatBool(value)}\n}\n\n// TaskListTag returns a new task list tag.\nfunc TaskListTag(value string) Tag {\n\tif len(value) == 0 {\n\t\tvalue = unknownValue\n\t}\n\treturn simpleMetric{key: taskList, value: sanitizer.Value(value)}\n}\n\n// TaskListUnknownTag returns a new tasklist:unknown tag-value\nfunc TaskListUnknownTag() Tag {\n\treturn simpleMetric{key: taskList, value: unknownValue}\n}\n\n// TaskListTypeTag returns a new task list type tag.\nfunc TaskListTypeTag(value string) Tag {\n\treturn metricWithUnknown(taskListType, value)\n}\n\n// TaskListRootPartition returns a new task list root partition tag.\nfunc TaskListRootPartitionTag(value string) Tag {\n\tif len(value) == 0 {\n\t\tvalue = unknownValue\n\t}\n\treturn simpleMetric{key: taskListRootPartition, value: sanitizer.Value(value)}\n}\n\n// WorkflowTypeTag returns a new workflow type tag.\nfunc WorkflowTypeTag(value string) Tag {\n\treturn metricWithUnknown(workflowType, value)\n}\n\n// ActivityTypeTag returns a new activity type tag.\nfunc ActivityTypeTag(value string) Tag {\n\treturn metricWithUnknown(activityType, value)\n}\n\n// DecisionTypeTag returns a new decision type tag.\nfunc DecisionTypeTag(value string) Tag {\n\treturn metricWithUnknown(decisionType, value)\n}\n\n// ShardScannerScanResult returns a new shardscanner scan result type tag.\nfunc ShardScannerScanResult(value string) Tag {\n\treturn metricWithUnknown(shardScannerScanResult, value)\n}\n\n// ShardScannerFixResult returns a new shardscanner fix result type tag.\nfunc ShardScannerFixResult(value string) Tag {\n\treturn metricWithUnknown(shardScannerFixResult, value)\n}\n\n// InvariantTypeTag returns a new invariant type tag.\nfunc InvariantTypeTag(value string) Tag {\n\treturn metricWithUnknown(invariantType, value)\n}\n\n// KafkaPartitionTag returns a new KafkaPartition type tag.\nfunc KafkaPartitionTag(value int32) Tag {\n\treturn simpleMetric{key: kafkaPartition, value: strconv.Itoa(int(value))}\n}\n\n// TransportTag returns a new RPC Transport type tag.\nfunc TransportTag(value string) Tag {\n\treturn simpleMetric{key: transport, value: value}\n}\n\n// CallerTag returns a new RPC Caller type tag.\nfunc CallerTag(value string) Tag {\n\treturn simpleMetric{key: caller, value: value}\n}\n\n// ServiceTag returns a new service tag.\nfunc ServiceTag(value string) Tag {\n\treturn simpleMetric{key: service, value: value}\n}\n\n// DestServiceTag returns a new destination service tag.\nfunc DestServiceTag(value string) Tag {\n\treturn simpleMetric{key: destService, value: value}\n}\n\n// Hosttag emits the host identifier\nfunc HostTag(value string) Tag {\n\treturn metricWithUnknown(host, value)\n}\n\n// SignalNameTag returns a new SignalName tag\nfunc SignalNameTag(value string) Tag {\n\treturn metricWithUnknown(signalName, value)\n}\n\n// SignalNameAllTag returns a new SignalName tag with all value\nfunc SignalNameAllTag() Tag {\n\treturn metricWithUnknown(signalName, allValue)\n}\n\n// WorkflowVersionTag returns a new WorkflowVersion tag\nfunc WorkflowVersionTag(value string) Tag {\n\treturn metricWithUnknown(workflowVersion, value)\n}\n\nfunc MatchingHostTag(value string) Tag {\n\treturn metricWithUnknown(matchingHost, value)\n}\n\n// PollerIsolationGroupTag returns a new PollerIsolationGroup tag\nfunc PollerIsolationGroupTag(value string) Tag {\n\treturn metricWithUnknown(pollerIsolationGroup, value)\n}\n\n// AsyncWFRequestTypeTag returns a new AsyncWFRequestTypeTag tag\nfunc AsyncWFRequestTypeTag(value string) Tag {\n\treturn metricWithUnknown(asyncWFRequestType, value)\n}\n\n// GlobalRatelimiterKeyTag reports the local ratelimit key being used, e.g. \"domain-x\".\n// This will likely be ambiguous if it is not combined with the collection name,\n// but keeping this untouched helps keep the values template-friendly and correlate-able\n// in metrics dashboards and queries.\nfunc GlobalRatelimiterKeyTag(value string) Tag {\n\treturn simpleMetric{key: globalRatelimitKey, value: value}\n}\n\n// GlobalRatelimiterTypeTag reports the \"limit usage type\" being reported, e.g. global vs local\nfunc GlobalRatelimiterTypeTag(value string) Tag {\n\treturn simpleMetric{key: globalRatelimitType, value: value}\n}\n\nfunc GlobalRatelimiterIsPrimary(isPrimary bool) Tag {\n\treturn simpleMetric{key: globalRatelimitIsPrimary, value: strconv.FormatBool(isPrimary)}\n}\n\n// GlobalRatelimiterCollectionName is a namespacing tag to uniquely identify metrics\n// coming from the different ratelimiter collections (user, worker, visibility, async).\nfunc GlobalRatelimiterCollectionName(value string) Tag {\n\treturn simpleMetric{key: globalRatelimitCollectionName, value: value}\n}\n\n// WorkflowTerminationReasonTag reports the reason for workflow termination\nfunc WorkflowTerminationReasonTag(value string) Tag {\n\tvalue = safeAlphaNumericStringRE.ReplaceAllString(value, \"_\")\n\treturn simpleMetric{key: workflowTerminationReason, value: value}\n}\n\n// WorkflowCloseStatusTag is a stringified workflow status\nfunc WorkflowCloseStatusTag(value string) Tag {\n\tvalue = safeAlphaNumericStringRE.ReplaceAllString(value, \"_\")\n\treturn simpleMetric{key: workflowCloseStatus, value: value}\n}\n\nfunc IsolationGroupTag(group string) Tag {\n\treturn simpleMetric{key: isolationGroup, value: sanitizer.Value(group)}\n}\n\nfunc IsolationLeakCause(cause string) Tag {\n\treturn simpleMetric{key: leakCause, value: sanitizer.Value(cause)}\n}\n\n// IsolationEnabledTag returns whether isolation is enabled\nfunc IsolationEnabledTag(enabled bool) Tag {\n\treturn simpleMetric{key: isolationEnabled, value: strconv.FormatBool(enabled)}\n}\n\nfunc TopicTag(value string) Tag {\n\treturn metricWithUnknown(topic, value)\n}\n\nfunc ModeTag(value string) Tag {\n\treturn metricWithUnknown(mode, value)\n}\n\n// IsRetryTag returns a new is_retry tag.\nfunc IsRetryTag(retry bool) Tag {\n\treturn simpleMetric{key: isRetry, value: strconv.FormatBool(retry)}\n}\n\nfunc NamespaceTag(namespace string) Tag {\n\treturn metricWithUnknown(\"namespace\", namespace)\n}\n\nfunc NamespaceTypeTag(namespaceType string) Tag {\n\treturn metricWithUnknown(\"namespace_type\", namespaceType)\n}\n\nfunc HandoverTypeTag(handoverType string) Tag {\n\treturn metricWithUnknown(\"handover_type\", handoverType)\n}\n\nfunc ExecutorStatusTag(status string) Tag {\n\treturn metricWithUnknown(\"executor_status\", status)\n}\n\nfunc ShardDistributorWatchTypeTag(watchType string) Tag {\n\treturn metricWithUnknown(\"watch_type\", watchType)\n}\n\nfunc TaskCategoryTag(category string) Tag {\n\treturn metricWithUnknown(\"task_category\", category)\n}\n\n// ReasonTag returns a new reason tag\nfunc ReasonTag(reason string) Tag {\n\treturn metricWithUnknown(\"reason\", reason)\n}\n\n// DecisionTag returns a new decision tag\nfunc DecisionTag(decision string) Tag {\n\treturn metricWithUnknown(\"decision\", decision)\n}\n\n// ActiveClusterLookupFnTag returns a new active cluster lookup function tag.\nfunc ActiveClusterLookupFnTag(fn string) Tag {\n\treturn metricWithUnknown(\"fn\", fn)\n}\n\n// ActiveClusterSelectionStrategyTag returns a new active cluster selection strategy tag.\nfunc ActiveClusterSelectionStrategyTag(strategy string) Tag {\n\treturn metricWithUnknown(\"strategy\", strategy)\n}\n\n// QueryConsistencyLevelTag returns a new query consistency level tag.\nfunc QueryConsistencyLevelTag(level string) Tag {\n\treturn metricWithUnknown(queryConsistencyLevel, level)\n}\n\n// BudgetManagerNameTag returns a new budget manager name tag.\nfunc BudgetManagerNameTag(name string) Tag {\n\treturn metricWithUnknown(budgetManagerName, name)\n}\n"
  },
  {
    "path": "common/metrics/tally/prometheus/buckets.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage prometheus\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally/prometheus\"\n)\n\n// DefaultHistogramBuckets is the default histogram buckets used when\n// creating a new Histogram in the prometheus registry.\n// Cadence uses Histogram to report both latency and non-latency metrics(like history_count/history_size).\n// The buckets need to set correct boundary for both\n// See more in https://github.com/uber/cadence/issues/4006\nfunc DefaultHistogramBuckets() []prometheus.HistogramObjective {\n\n\t// For latency metrics, this is a nanoSecond.\n\t// For non-latency metrics, this is ONE.\n\t// divided by Second because of https://github.com/uber-go/tally/blob/04828d51c63b1d09b46824d7c0d7904b5eb1b3b6/prometheus/reporter.go#L183\n\toneOrNanoSecond := float64(time.Nanosecond) / float64(time.Second)\n\n\t// convenient units\n\tmicroSecOr1K := oneOrNanoSecond * 1000\n\tmilliSecOr1M := oneOrNanoSecond * 1000000\n\n\treturn []prometheus.HistogramObjective{\n\t\t{Upper: 5 * oneOrNanoSecond},\n\t\t{Upper: 10 * oneOrNanoSecond},\n\t\t{Upper: 50 * oneOrNanoSecond},\n\t\t{Upper: 100 * oneOrNanoSecond},\n\n\t\t{Upper: microSecOr1K},\n\t\t{Upper: 10 * microSecOr1K},\n\t\t{Upper: 50 * microSecOr1K},\n\t\t{Upper: 200 * microSecOr1K},\n\n\t\t// Below are exactly the same as https://github.com/uber-go/tally/blob/04828d51c63b1d09b46824d7c0d7904b5eb1b3b6/prometheus/reporter.go#L51\n\t\t// to ensure backward compatibility\n\t\t{Upper: milliSecOr1M},\n\t\t{Upper: 2 * milliSecOr1M},\n\t\t{Upper: 5 * milliSecOr1M},\n\t\t{Upper: 10 * milliSecOr1M},\n\t\t{Upper: 20 * milliSecOr1M},\n\t\t{Upper: 50 * milliSecOr1M},\n\t\t{Upper: 100 * milliSecOr1M},\n\t\t{Upper: 200 * milliSecOr1M},\n\t\t{Upper: 500 * milliSecOr1M},\n\t\t{Upper: 1000 * milliSecOr1M},\n\t\t{Upper: 2000 * milliSecOr1M},\n\t\t{Upper: 5000 * milliSecOr1M},\n\t\t{Upper: 10000 * milliSecOr1M},\n\t}\n}\n"
  },
  {
    "path": "common/metrics/tally/statsd/reporter.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage statsd\n\nimport (\n\t\"bytes\"\n\t\"sort\"\n\t\"time\"\n\n\t\"github.com/cactus/go-statsd-client/statsd\"\n\t\"github.com/uber-go/tally\"\n\ttallystatsdreporter \"github.com/uber-go/tally/statsd\"\n)\n\ntype cadenceTallyStatsdReporter struct {\n\t// Wrapper on top of \"github.com/uber-go/tally/statsd\"\n\ttallystatsd tally.StatsReporter\n}\n\nfunc (r *cadenceTallyStatsdReporter) metricNameWithTags(originalName string, tags map[string]string) string {\n\tvar keys []string\n\tfor k := range tags {\n\t\tkeys = append(keys, k)\n\t}\n\tsort.Strings(keys)\n\n\tvar buffer bytes.Buffer\n\tbuffer.WriteString(originalName)\n\n\tfor _, tk := range keys {\n\t\t// adding \".\" as delimiter so that it will show as different parts in Graphite/Grafana\n\t\tbuffer.WriteString(\".\" + tk + \".\" + tags[tk])\n\t}\n\n\treturn buffer.String()\n}\n\n// NewReporter is a wrapper on top of \"github.com/uber-go/tally/statsd\"\n// The purpose is to support tagging\n// The implementation is to append tags as metric name suffixes\nfunc NewReporter(statsd statsd.Statter, opts tallystatsdreporter.Options) tally.StatsReporter {\n\treturn &cadenceTallyStatsdReporter{\n\t\ttallystatsd: tallystatsdreporter.NewReporter(statsd, opts),\n\t}\n}\n\nfunc (r *cadenceTallyStatsdReporter) ReportCounter(name string, tags map[string]string, value int64) {\n\tnewName := r.metricNameWithTags(name, tags)\n\tr.tallystatsd.ReportCounter(newName, map[string]string{}, value)\n}\n\nfunc (r *cadenceTallyStatsdReporter) ReportGauge(name string, tags map[string]string, value float64) {\n\tnewName := r.metricNameWithTags(name, tags)\n\tr.tallystatsd.ReportGauge(newName, map[string]string{}, value)\n}\n\nfunc (r *cadenceTallyStatsdReporter) ReportTimer(name string, tags map[string]string, interval time.Duration) {\n\tnewName := r.metricNameWithTags(name, tags)\n\tr.tallystatsd.ReportTimer(newName, map[string]string{}, interval)\n}\n\nfunc (r *cadenceTallyStatsdReporter) ReportHistogramValueSamples(\n\tname string,\n\ttags map[string]string,\n\tbuckets tally.Buckets,\n\tbucketLowerBound,\n\tbucketUpperBound float64,\n\tsamples int64,\n) {\n\tnewName := r.metricNameWithTags(name, tags)\n\tr.tallystatsd.ReportHistogramValueSamples(newName, map[string]string{}, buckets, bucketLowerBound, bucketUpperBound, samples)\n}\n\nfunc (r *cadenceTallyStatsdReporter) ReportHistogramDurationSamples(\n\tname string,\n\ttags map[string]string,\n\tbuckets tally.Buckets,\n\tbucketLowerBound,\n\tbucketUpperBound time.Duration,\n\tsamples int64,\n) {\n\tnewName := r.metricNameWithTags(name, tags)\n\tr.tallystatsd.ReportHistogramDurationSamples(newName, map[string]string{}, buckets, bucketLowerBound, bucketUpperBound, samples)\n}\n\nfunc (r *cadenceTallyStatsdReporter) Capabilities() tally.Capabilities {\n\treturn r.tallystatsd.Capabilities()\n}\n\nfunc (r *cadenceTallyStatsdReporter) Flush() {\n\tr.tallystatsd.Flush()\n}\n"
  },
  {
    "path": "common/metrics/tally/statsd/reporter_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage statsd\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestMetricNameWithTags(t *testing.T) {\n\tr := cadenceTallyStatsdReporter{\n\t\ttallystatsd: nil,\n\t}\n\ttags := map[string]string{\n\t\t\"tag1\": \"123\",\n\t\t\"tag2\": \"456\",\n\t\t\"tag3\": \"789\",\n\t}\n\tname := \"test-metric-name1\"\n\n\tassert.Equal(t, \"test-metric-name1.tag1.123.tag2.456.tag3.789\", r.metricNameWithTags(name, tags))\n}\n\nfunc TestMetricNameWithTagsStability(t *testing.T) {\n\tr := cadenceTallyStatsdReporter{\n\t\ttallystatsd: nil,\n\t}\n\ttags := map[string]string{\n\t\t\"tag1\": \"123\",\n\t\t\"tag2\": \"456\",\n\t\t\"tag3\": \"789\",\n\t}\n\tname := \"test-metric-name2\"\n\n\t// test the order is stable\n\tfor i := 1; i <= 16; i++ {\n\t\tassert.Equal(t, r.metricNameWithTags(name, tags), r.metricNameWithTags(name, tags))\n\t}\n}\n"
  },
  {
    "path": "common/metrics/version.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metrics\n\n// VersionString the current release version\nconst VersionString = \"1.2.5\"\n"
  },
  {
    "path": "common/mocks/ExecutionManager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.32.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// ExecutionManager is an autogenerated mock type for the ExecutionManager type\ntype ExecutionManager struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *ExecutionManager) Close() {\n\t_m.Called()\n}\n\n// CompleteHistoryTask provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) CompleteHistoryTask(ctx context.Context, request *persistence.CompleteHistoryTaskRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CompleteHistoryTaskRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ConflictResolveWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) ConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (*persistence.ConflictResolveWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ConflictResolveWorkflowExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ConflictResolveWorkflowExecutionRequest) (*persistence.ConflictResolveWorkflowExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ConflictResolveWorkflowExecutionRequest) *persistence.ConflictResolveWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ConflictResolveWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ConflictResolveWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateFailoverMarkerTasks provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) CreateFailoverMarkerTasks(ctx context.Context, request *persistence.CreateFailoverMarkersRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateFailoverMarkersRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// CreateWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) CreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.CreateWorkflowExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateWorkflowExecutionRequest) *persistence.CreateWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.CreateWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.CreateWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteCurrentWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) DeleteCurrentWorkflowExecution(ctx context.Context, request *persistence.DeleteCurrentWorkflowExecutionRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteCurrentWorkflowExecutionRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// DeleteReplicationTaskFromDLQ provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) DeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.DeleteReplicationTaskFromDLQRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteReplicationTaskFromDLQRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// DeleteWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.DeleteWorkflowExecutionRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteWorkflowExecutionRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// GetCurrentExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) GetCurrentExecution(ctx context.Context, request *persistence.GetCurrentExecutionRequest) (*persistence.GetCurrentExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetCurrentExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetCurrentExecutionRequest) (*persistence.GetCurrentExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetCurrentExecutionRequest) *persistence.GetCurrentExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetCurrentExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetCurrentExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetHistoryTasks provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) GetHistoryTasks(ctx context.Context, request *persistence.GetHistoryTasksRequest) (*persistence.GetHistoryTasksResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetHistoryTasksResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetHistoryTasksRequest) (*persistence.GetHistoryTasksResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetHistoryTasksRequest) *persistence.GetHistoryTasksResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetHistoryTasksResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetHistoryTasksRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *ExecutionManager) GetName() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// GetReplicationDLQSize provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) GetReplicationDLQSize(ctx context.Context, request *persistence.GetReplicationDLQSizeRequest) (*persistence.GetReplicationDLQSizeResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetReplicationDLQSizeResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetReplicationDLQSizeRequest) (*persistence.GetReplicationDLQSizeResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetReplicationDLQSizeRequest) *persistence.GetReplicationDLQSizeResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetReplicationDLQSizeResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetReplicationDLQSizeRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetReplicationTasksFromDLQ provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) GetReplicationTasksFromDLQ(ctx context.Context, request *persistence.GetReplicationTasksFromDLQRequest) (*persistence.GetHistoryTasksResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetHistoryTasksResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetReplicationTasksFromDLQRequest) (*persistence.GetHistoryTasksResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetReplicationTasksFromDLQRequest) *persistence.GetHistoryTasksResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetHistoryTasksResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetReplicationTasksFromDLQRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetShardID provides a mock function with given fields:\nfunc (_m *ExecutionManager) GetShardID() int {\n\tret := _m.Called()\n\n\tvar r0 int\n\tif rf, ok := ret.Get(0).(func() int); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(int)\n\t}\n\n\treturn r0\n}\n\n// GetWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) GetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetWorkflowExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetWorkflowExecutionRequest) *persistence.GetWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// IsWorkflowExecutionExists provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) IsWorkflowExecutionExists(ctx context.Context, request *persistence.IsWorkflowExecutionExistsRequest) (*persistence.IsWorkflowExecutionExistsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.IsWorkflowExecutionExistsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.IsWorkflowExecutionExistsRequest) (*persistence.IsWorkflowExecutionExistsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.IsWorkflowExecutionExistsRequest) *persistence.IsWorkflowExecutionExistsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.IsWorkflowExecutionExistsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.IsWorkflowExecutionExistsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListConcreteExecutions provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) ListConcreteExecutions(ctx context.Context, request *persistence.ListConcreteExecutionsRequest) (*persistence.ListConcreteExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ListConcreteExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListConcreteExecutionsRequest) (*persistence.ListConcreteExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListConcreteExecutionsRequest) *persistence.ListConcreteExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListConcreteExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListConcreteExecutionsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListCurrentExecutions provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) ListCurrentExecutions(ctx context.Context, request *persistence.ListCurrentExecutionsRequest) (*persistence.ListCurrentExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ListCurrentExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListCurrentExecutionsRequest) (*persistence.ListCurrentExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListCurrentExecutionsRequest) *persistence.ListCurrentExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListCurrentExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListCurrentExecutionsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// PutReplicationTaskToDLQ provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) PutReplicationTaskToDLQ(ctx context.Context, request *persistence.PutReplicationTaskToDLQRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.PutReplicationTaskToDLQRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// RangeCompleteHistoryTask provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) RangeCompleteHistoryTask(ctx context.Context, request *persistence.RangeCompleteHistoryTaskRequest) (*persistence.RangeCompleteHistoryTaskResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.RangeCompleteHistoryTaskResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RangeCompleteHistoryTaskRequest) (*persistence.RangeCompleteHistoryTaskResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RangeCompleteHistoryTaskRequest) *persistence.RangeCompleteHistoryTaskResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.RangeCompleteHistoryTaskResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.RangeCompleteHistoryTaskRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// RangeDeleteReplicationTaskFromDLQ provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.RangeDeleteReplicationTaskFromDLQRequest) (*persistence.RangeDeleteReplicationTaskFromDLQResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.RangeDeleteReplicationTaskFromDLQResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RangeDeleteReplicationTaskFromDLQRequest) (*persistence.RangeDeleteReplicationTaskFromDLQResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RangeDeleteReplicationTaskFromDLQRequest) *persistence.RangeDeleteReplicationTaskFromDLQResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.RangeDeleteReplicationTaskFromDLQResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.RangeDeleteReplicationTaskFromDLQRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UpdateWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *ExecutionManager) UpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.UpdateWorkflowExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateWorkflowExecutionRequest) *persistence.UpdateWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.UpdateWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.UpdateWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetActiveClusterSelectionPolicy provides a mock function with given fields: ctx, domainID, wfID, rID\nfunc (_m *ExecutionManager) GetActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error) {\n\tret := _m.Called(ctx, domainID, wfID, rID)\n\n\tvar r0 *types.ActiveClusterSelectionPolicy\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string, string) (*types.ActiveClusterSelectionPolicy, error)); ok {\n\t\treturn rf(ctx, domainID, wfID, rID)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string, string) *types.ActiveClusterSelectionPolicy); ok {\n\t\tr0 = rf(ctx, domainID, wfID, rID)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*types.ActiveClusterSelectionPolicy)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, string, string, string) error); ok {\n\t\tr1 = rf(ctx, domainID, wfID, rID)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteActiveClusterSelectionPolicy provides a mock function with given fields: ctx, domainID, wfID, rID\nfunc (_m *ExecutionManager) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) error {\n\tret := _m.Called(ctx, domainID, wfID, rID)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, string, string, string) error); ok {\n\t\tr0 = rf(ctx, domainID, wfID, rID)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// NewExecutionManager creates a new instance of ExecutionManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.\n// The first argument is typically a *testing.T value.\nfunc NewExecutionManager(t interface {\n\tmock.TestingT\n\tCleanup(func())\n}) *ExecutionManager {\n\tmock := &ExecutionManager{}\n\tmock.Mock.Test(t)\n\n\tt.Cleanup(func() { mock.AssertExpectations(t) })\n\n\treturn mock\n}\n"
  },
  {
    "path": "common/mocks/ExecutionManagerFactory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mocks\n\nimport (\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// ExecutionManagerFactory is an autogenerated mock type for the ExecutionManagerFactory type\ntype ExecutionManagerFactory struct {\n\tmock.Mock\n}\n\n// NewExecutionManager provides a mock function with given fields: shardID\nfunc (_m *ExecutionManagerFactory) NewExecutionManager(shardID int) (persistence.ExecutionManager, error) {\n\tret := _m.Called(shardID)\n\n\tvar r0 persistence.ExecutionManager\n\tif rf, ok := ret.Get(0).(func(int) persistence.ExecutionManager); ok {\n\t\tr0 = rf(shardID)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(persistence.ExecutionManager)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(int) error); ok {\n\t\tr1 = rf(shardID)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Close is mock implementation for Close of ExecutionManagerFactory\nfunc (_m *ExecutionManagerFactory) Close() {\n\t_m.Called()\n}\n"
  },
  {
    "path": "common/mocks/HistoryV2Manager.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.2.1. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// HistoryV2Manager is an autogenerated mock type for the HistoryV2Manager type\n// TODO: rename to HistoryManager\ntype HistoryV2Manager struct {\n\tmock.Mock\n}\n\n// AppendHistoryNodes provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) AppendHistoryNodes(ctx context.Context, request *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.AppendHistoryNodesResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.AppendHistoryNodesRequest) *persistence.AppendHistoryNodesResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.AppendHistoryNodesResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.AppendHistoryNodesRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *HistoryV2Manager) Close() {\n\t_m.Called()\n}\n\n// DeleteHistoryBranch provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) DeleteHistoryBranch(ctx context.Context, request *persistence.DeleteHistoryBranchRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteHistoryBranchRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ForkHistoryBranch provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) ForkHistoryBranch(ctx context.Context, request *persistence.ForkHistoryBranchRequest) (*persistence.ForkHistoryBranchResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ForkHistoryBranchResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ForkHistoryBranchRequest) *persistence.ForkHistoryBranchResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ForkHistoryBranchResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ForkHistoryBranchRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetAllHistoryTreeBranches provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) GetAllHistoryTreeBranches(ctx context.Context, request *persistence.GetAllHistoryTreeBranchesRequest) (*persistence.GetAllHistoryTreeBranchesResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetAllHistoryTreeBranchesResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetAllHistoryTreeBranchesRequest) *persistence.GetAllHistoryTreeBranchesResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetAllHistoryTreeBranchesResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetAllHistoryTreeBranchesRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetHistoryTree provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) GetHistoryTree(ctx context.Context, request *persistence.GetHistoryTreeRequest) (*persistence.GetHistoryTreeResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetHistoryTreeResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetHistoryTreeRequest) *persistence.GetHistoryTreeResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetHistoryTreeResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetHistoryTreeRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *HistoryV2Manager) GetName() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// ReadHistoryBranch provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) ReadHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (*persistence.ReadHistoryBranchResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ReadHistoryBranchResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ReadHistoryBranchRequest) *persistence.ReadHistoryBranchResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ReadHistoryBranchResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ReadHistoryBranchRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ReadHistoryBranchByBatch provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) ReadHistoryBranchByBatch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (*persistence.ReadHistoryBranchByBatchResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ReadHistoryBranchByBatchResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ReadHistoryBranchRequest) *persistence.ReadHistoryBranchByBatchResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ReadHistoryBranchByBatchResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ReadHistoryBranchRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ReadRawHistoryBranch provides a mock function with given fields: ctx, request\nfunc (_m *HistoryV2Manager) ReadRawHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (*persistence.ReadRawHistoryBranchResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ReadRawHistoryBranchResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ReadHistoryBranchRequest) *persistence.ReadRawHistoryBranchResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ReadRawHistoryBranchResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ReadHistoryBranchRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n"
  },
  {
    "path": "common/mocks/KafkaProducer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common/messaging\"\n)\n\n// KafkaProducer is an autogenerated mock type for the KafkaProducer type\ntype KafkaProducer struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *KafkaProducer) Close() error {\n\tret := _m.Called()\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func() error); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// Publish provides a mock function with given fields: ctx, message\nfunc (_m *KafkaProducer) Publish(ctx context.Context, message interface{}) error {\n\tret := _m.Called(ctx, message)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, interface{}) error); ok {\n\t\tr0 = rf(ctx, message)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\nvar _ messaging.Producer = (*KafkaProducer)(nil)\n"
  },
  {
    "path": "common/mocks/MessagingClient.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mocks\n\nimport \"github.com/uber/cadence/common/messaging\"\n\ntype (\n\t// MessagingClient is the mock implementation for Service interface\n\tMessagingClient struct {\n\t\tconsumerMock  messaging.Consumer\n\t\tpublisherMock messaging.Producer\n\t}\n)\n\nvar _ messaging.Client = (*MessagingClient)(nil)\n\n// NewMockMessagingClient generate a dummy implementation of messaging client\nfunc NewMockMessagingClient(publisher messaging.Producer, consumer messaging.Consumer) messaging.Client {\n\treturn &MessagingClient{\n\t\tpublisherMock: publisher,\n\t\tconsumerMock:  consumer,\n\t}\n}\n\n// NewConsumer generates a dummy implementation of kafka consumer\nfunc (c *MessagingClient) NewConsumer(appName, consumerName string) (messaging.Consumer, error) {\n\treturn c.consumerMock, nil\n}\n\n// NewProducer generates a dummy implementation of kafka producer\nfunc (c *MessagingClient) NewProducer(appName string) (messaging.Producer, error) {\n\treturn c.publisherMock, nil\n}\n"
  },
  {
    "path": "common/mocks/MetadataManager.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.2.1. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MetadataManager is an autogenerated mock type for the MetadataManager type\ntype MetadataManager struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *MetadataManager) Close() {\n\t_m.Called()\n}\n\n// CreateDomain provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) CreateDomain(ctx context.Context, request *persistence.CreateDomainRequest) (*persistence.CreateDomainResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.CreateDomainResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateDomainRequest) *persistence.CreateDomainResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.CreateDomainResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.CreateDomainRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteDomain provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) DeleteDomain(ctx context.Context, request *persistence.DeleteDomainRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteDomainRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// DeleteDomainByName provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) DeleteDomainByName(ctx context.Context, request *persistence.DeleteDomainByNameRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteDomainByNameRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// GetDomain provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) GetDomain(ctx context.Context, request *persistence.GetDomainRequest) (*persistence.GetDomainResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetDomainResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetDomainRequest) *persistence.GetDomainResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetDomainResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetDomainRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetMetadata provides a mock function with given fields: ctx\nfunc (_m *MetadataManager) GetMetadata(ctx context.Context) (*persistence.GetMetadataResponse, error) {\n\tret := _m.Called(ctx)\n\n\tvar r0 *persistence.GetMetadataResponse\n\tif rf, ok := ret.Get(0).(func(context.Context) *persistence.GetMetadataResponse); ok {\n\t\tr0 = rf(ctx)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetMetadataResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context) error); ok {\n\t\tr1 = rf(ctx)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *MetadataManager) GetName() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// ListDomains provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) ListDomains(ctx context.Context, request *persistence.ListDomainsRequest) (*persistence.ListDomainsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ListDomainsResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListDomainsRequest) *persistence.ListDomainsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListDomainsResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListDomainsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UpdateDomain provides a mock function with given fields: ctx, request\nfunc (_m *MetadataManager) UpdateDomain(ctx context.Context, request *persistence.UpdateDomainRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateDomainRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/mocks/ShardManager.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.2.1. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// ShardManager is an autogenerated mock type for the ShardManager type\ntype ShardManager struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *ShardManager) Close() {\n\t_m.Called()\n}\n\n// CreateShard provides a mock function with given fields: ctx, request\nfunc (_m *ShardManager) CreateShard(ctx context.Context, request *persistence.CreateShardRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateShardRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *ShardManager) GetName() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// GetShard provides a mock function with given fields: ctx, request\nfunc (_m *ShardManager) GetShard(ctx context.Context, request *persistence.GetShardRequest) (*persistence.GetShardResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetShardResponse\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetShardRequest) *persistence.GetShardResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetShardResponse)\n\t\t}\n\t}\n\n\tvar r1 error\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetShardRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UpdateShard provides a mock function with given fields: ctx, request\nfunc (_m *ShardManager) UpdateShard(ctx context.Context, request *persistence.UpdateShardRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateShardRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n"
  },
  {
    "path": "common/mocks/TaskManager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.32.0. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// TaskManager is an autogenerated mock type for the TaskManager type\ntype TaskManager struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *TaskManager) Close() {\n\t_m.Called()\n}\n\n// CompleteTask provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) CompleteTask(ctx context.Context, request *persistence.CompleteTaskRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CompleteTaskRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// CompleteTasksLessThan provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) CompleteTasksLessThan(ctx context.Context, request *persistence.CompleteTasksLessThanRequest) (*persistence.CompleteTasksLessThanResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.CompleteTasksLessThanResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CompleteTasksLessThanRequest) (*persistence.CompleteTasksLessThanResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CompleteTasksLessThanRequest) *persistence.CompleteTasksLessThanResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.CompleteTasksLessThanResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.CompleteTasksLessThanRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// CreateTasks provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) CreateTasks(ctx context.Context, request *persistence.CreateTasksRequest) (*persistence.CreateTasksResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.CreateTasksResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateTasksRequest) (*persistence.CreateTasksResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CreateTasksRequest) *persistence.CreateTasksResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.CreateTasksResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.CreateTasksRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteTaskList provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) DeleteTaskList(ctx context.Context, request *persistence.DeleteTaskListRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.DeleteTaskListRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *TaskManager) GetName() string {\n\tret := _m.Called()\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// GetOrphanTasks provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (*persistence.GetOrphanTasksResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetOrphanTasksResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetOrphanTasksRequest) (*persistence.GetOrphanTasksResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetOrphanTasksRequest) *persistence.GetOrphanTasksResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetOrphanTasksResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetOrphanTasksRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetTaskList provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) GetTaskList(ctx context.Context, request *persistence.GetTaskListRequest) (*persistence.GetTaskListResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetTaskListResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTaskListRequest) (*persistence.GetTaskListResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTaskListRequest) *persistence.GetTaskListResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetTaskListResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetTaskListRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetTaskListSize provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (*persistence.GetTaskListSizeResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetTaskListSizeResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTaskListSizeRequest) (*persistence.GetTaskListSizeResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTaskListSizeRequest) *persistence.GetTaskListSizeResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetTaskListSizeResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetTaskListSizeRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetTasks provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) GetTasks(ctx context.Context, request *persistence.GetTasksRequest) (*persistence.GetTasksResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.GetTasksResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTasksRequest) (*persistence.GetTasksResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetTasksRequest) *persistence.GetTasksResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetTasksResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetTasksRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// LeaseTaskList provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) LeaseTaskList(ctx context.Context, request *persistence.LeaseTaskListRequest) (*persistence.LeaseTaskListResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.LeaseTaskListResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.LeaseTaskListRequest) (*persistence.LeaseTaskListResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.LeaseTaskListRequest) *persistence.LeaseTaskListResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.LeaseTaskListResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.LeaseTaskListRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListTaskList provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) ListTaskList(ctx context.Context, request *persistence.ListTaskListRequest) (*persistence.ListTaskListResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.ListTaskListResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListTaskListRequest) (*persistence.ListTaskListResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListTaskListRequest) *persistence.ListTaskListResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListTaskListResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListTaskListRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UpdateTaskList provides a mock function with given fields: ctx, request\nfunc (_m *TaskManager) UpdateTaskList(ctx context.Context, request *persistence.UpdateTaskListRequest) (*persistence.UpdateTaskListResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tvar r0 *persistence.UpdateTaskListResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateTaskListRequest) (*persistence.UpdateTaskListResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpdateTaskListRequest) *persistence.UpdateTaskListResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.UpdateTaskListResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.UpdateTaskListRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// NewTaskManager creates a new instance of TaskManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.\n// The first argument is typically a *testing.T value.\nfunc NewTaskManager(t interface {\n\tmock.TestingT\n\tCleanup(func())\n}) *TaskManager {\n\tmock := &TaskManager{}\n\tmock.Mock.Test(t)\n\n\tt.Cleanup(func() { mock.AssertExpectations(t) })\n\n\treturn mock\n}\n"
  },
  {
    "path": "common/mocks/VisibilityManager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Code generated by mockery v2.43.1. DO NOT EDIT.\n\npackage mocks\n\nimport (\n\tcontext \"context\"\n\n\tmock \"github.com/stretchr/testify/mock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// VisibilityManager is an autogenerated mock type for the VisibilityManager type\ntype VisibilityManager struct {\n\tmock.Mock\n}\n\n// Close provides a mock function with given fields:\nfunc (_m *VisibilityManager) Close() {\n\t_m.Called()\n}\n\n// CountWorkflowExecutions provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) CountWorkflowExecutions(ctx context.Context, request *persistence.CountWorkflowExecutionsRequest) (*persistence.CountWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for CountWorkflowExecutions\")\n\t}\n\n\tvar r0 *persistence.CountWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CountWorkflowExecutionsRequest) (*persistence.CountWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.CountWorkflowExecutionsRequest) *persistence.CountWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.CountWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.CountWorkflowExecutionsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// DeleteUninitializedWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) DeleteUninitializedWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteUninitializedWorkflowExecution\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.VisibilityDeleteWorkflowExecutionRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// DeleteWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for DeleteWorkflowExecution\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.VisibilityDeleteWorkflowExecutionRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// GetClosedWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) GetClosedWorkflowExecution(ctx context.Context, request *persistence.GetClosedWorkflowExecutionRequest) (*persistence.GetClosedWorkflowExecutionResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetClosedWorkflowExecution\")\n\t}\n\n\tvar r0 *persistence.GetClosedWorkflowExecutionResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetClosedWorkflowExecutionRequest) (*persistence.GetClosedWorkflowExecutionResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.GetClosedWorkflowExecutionRequest) *persistence.GetClosedWorkflowExecutionResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.GetClosedWorkflowExecutionResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.GetClosedWorkflowExecutionRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// GetName provides a mock function with given fields:\nfunc (_m *VisibilityManager) GetName() string {\n\tret := _m.Called()\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for GetName\")\n\t}\n\n\tvar r0 string\n\tif rf, ok := ret.Get(0).(func() string); ok {\n\t\tr0 = rf()\n\t} else {\n\t\tr0 = ret.Get(0).(string)\n\t}\n\n\treturn r0\n}\n\n// ListClosedWorkflowExecutions provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListClosedWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListClosedWorkflowExecutions\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListClosedWorkflowExecutionsByStatus provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *persistence.ListClosedWorkflowExecutionsByStatusRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListClosedWorkflowExecutionsByStatus\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListClosedWorkflowExecutionsByStatusRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListClosedWorkflowExecutionsByStatusRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListClosedWorkflowExecutionsByStatusRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListClosedWorkflowExecutionsByType provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListClosedWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListClosedWorkflowExecutionsByType\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListClosedWorkflowExecutionsByWorkflowID provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListClosedWorkflowExecutionsByWorkflowID\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListOpenWorkflowExecutions provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListOpenWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListOpenWorkflowExecutions\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListOpenWorkflowExecutionsByType provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListOpenWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListOpenWorkflowExecutionsByType\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByTypeRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListOpenWorkflowExecutionsByWorkflowID provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListOpenWorkflowExecutionsByWorkflowID\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByWorkflowIDRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// ListWorkflowExecutions provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ListWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ListWorkflowExecutions\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// RecordWorkflowExecutionClosed provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) RecordWorkflowExecutionClosed(ctx context.Context, request *persistence.RecordWorkflowExecutionClosedRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RecordWorkflowExecutionClosed\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RecordWorkflowExecutionClosedRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// RecordWorkflowExecutionStarted provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) RecordWorkflowExecutionStarted(ctx context.Context, request *persistence.RecordWorkflowExecutionStartedRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RecordWorkflowExecutionStarted\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RecordWorkflowExecutionStartedRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// RecordWorkflowExecutionUninitialized provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) RecordWorkflowExecutionUninitialized(ctx context.Context, request *persistence.RecordWorkflowExecutionUninitializedRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for RecordWorkflowExecutionUninitialized\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.RecordWorkflowExecutionUninitializedRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// ScanWorkflowExecutions provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) ScanWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for ScanWorkflowExecutions\")\n\t}\n\n\tvar r0 *persistence.ListWorkflowExecutionsResponse\n\tvar r1 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) (*persistence.ListWorkflowExecutionsResponse, error)); ok {\n\t\treturn rf(ctx, request)\n\t}\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) *persistence.ListWorkflowExecutionsResponse); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tif ret.Get(0) != nil {\n\t\t\tr0 = ret.Get(0).(*persistence.ListWorkflowExecutionsResponse)\n\t\t}\n\t}\n\n\tif rf, ok := ret.Get(1).(func(context.Context, *persistence.ListWorkflowExecutionsByQueryRequest) error); ok {\n\t\tr1 = rf(ctx, request)\n\t} else {\n\t\tr1 = ret.Error(1)\n\t}\n\n\treturn r0, r1\n}\n\n// UpsertWorkflowExecution provides a mock function with given fields: ctx, request\nfunc (_m *VisibilityManager) UpsertWorkflowExecution(ctx context.Context, request *persistence.UpsertWorkflowExecutionRequest) error {\n\tret := _m.Called(ctx, request)\n\n\tif len(ret) == 0 {\n\t\tpanic(\"no return value specified for UpsertWorkflowExecution\")\n\t}\n\n\tvar r0 error\n\tif rf, ok := ret.Get(0).(func(context.Context, *persistence.UpsertWorkflowExecutionRequest) error); ok {\n\t\tr0 = rf(ctx, request)\n\t} else {\n\t\tr0 = ret.Error(0)\n\t}\n\n\treturn r0\n}\n\n// NewVisibilityManager creates a new instance of VisibilityManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.\n// The first argument is typically a *testing.T value.\nfunc NewVisibilityManager(t interface {\n\tmock.TestingT\n\tCleanup(func())\n}) *VisibilityManager {\n\tmock := &VisibilityManager{}\n\tmock.Mock.Test(t)\n\n\tt.Cleanup(func() { mock.AssertExpectations(t) })\n\n\treturn mock\n}\n"
  },
  {
    "path": "common/ndc/history_resender.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination history_resender_mock.go\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// ErrSkipTask is the error to skip task due to absence of the workflow in the source cluster\n\tErrSkipTask = errors.New(\"the source workflow does not exist\")\n)\n\nconst (\n\tresendContextTimeout = 30 * time.Second\n\tdefaultPageSize      = int32(100)\n)\n\ntype (\n\t// nDCHistoryReplicationFn provides the functionality to deliver replication raw history request to history\n\t// the provided func should be thread safe\n\tnDCHistoryReplicationFn func(ctx context.Context, request *types.ReplicateEventsV2Request) error\n\n\t// HistoryResender is the interface for resending history events to remote\n\tHistoryResender interface {\n\t\t// SendSingleWorkflowHistory sends multiple run IDs's history events to remote\n\t\tSendSingleWorkflowHistory(\n\t\t\tremoteCluster string,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\trunID string,\n\t\t\tstartEventID *int64,\n\t\t\tstartEventVersion *int64,\n\t\t\tendEventID *int64,\n\t\t\tendEventVersion *int64,\n\t\t) error\n\t}\n\n\t// HistoryResenderImpl is the implementation of NDCHistoryResender\n\tHistoryResenderImpl struct {\n\t\tdomainCache           cache.DomainCache\n\t\tclientBean            client.Bean\n\t\thistoryReplicationFn  nDCHistoryReplicationFn\n\t\trereplicationTimeout  dynamicproperties.DurationPropertyFnWithDomainIDFilter\n\t\tcurrentExecutionCheck invariant.Invariant\n\t\tlogger                log.Logger\n\t}\n\n\thistoryBatch struct {\n\t\tversionHistory *types.VersionHistory\n\t\trawEventBatch  *types.DataBlob\n\t}\n)\n\n// NewHistoryResender create a new NDCHistoryResenderImpl\nfunc NewHistoryResender(\n\tdomainCache cache.DomainCache,\n\tclientBean client.Bean,\n\thistoryReplicationFn nDCHistoryReplicationFn,\n\trereplicationTimeout dynamicproperties.DurationPropertyFnWithDomainIDFilter,\n\tcurrentExecutionCheck invariant.Invariant,\n\tlogger log.Logger,\n) *HistoryResenderImpl {\n\n\treturn &HistoryResenderImpl{\n\t\tdomainCache:           domainCache,\n\t\tclientBean:            clientBean,\n\t\thistoryReplicationFn:  historyReplicationFn,\n\t\trereplicationTimeout:  rereplicationTimeout,\n\t\tcurrentExecutionCheck: currentExecutionCheck,\n\t\tlogger:                logger.WithTags(tag.ComponentHistoryResender),\n\t}\n}\n\n// SendSingleWorkflowHistory sends one run IDs's history events to remote\nfunc (n *HistoryResenderImpl) SendSingleWorkflowHistory(\n\tremoteCluster string,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tstartEventID *int64,\n\tstartEventVersion *int64,\n\tendEventID *int64,\n\tendEventVersion *int64,\n) error {\n\n\tdomainName, err := n.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"getting domain: %w\", err)\n\t}\n\n\tctx := context.Background()\n\tvar cancel context.CancelFunc\n\tif n.rereplicationTimeout != nil {\n\t\tresendContextTimeout := n.rereplicationTimeout(domainID)\n\t\tif resendContextTimeout > 0 {\n\t\t\tctx, cancel = context.WithTimeout(ctx, resendContextTimeout)\n\t\t\tdefer cancel()\n\t\t}\n\t}\n\n\thistoryIterator := collection.NewPagingIterator(n.getPaginationFn(\n\t\tctx,\n\t\tremoteCluster,\n\t\tdomainName,\n\t\tworkflowID,\n\t\trunID,\n\t\tstartEventID,\n\t\tstartEventVersion,\n\t\tendEventID,\n\t\tendEventVersion))\n\n\tfor historyIterator.HasNext() {\n\t\tresult, err := historyIterator.Next()\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"history iterator: %w\", err)\n\t\t}\n\n\t\thistoryBatch := result.(*historyBatch)\n\t\treplicationRequest := n.createReplicationRawRequest(\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\thistoryBatch.rawEventBatch,\n\t\t\thistoryBatch.versionHistory.GetItems())\n\n\t\terr = n.sendReplicationRawRequest(ctx, replicationRequest)\n\t\tswitch err.(type) {\n\t\tcase nil:\n\t\t\t// continue to process the events\n\t\t\tbreak\n\t\tcase *types.EntityNotExistsError:\n\t\t\t// Case 1: the workflow pass the retention period\n\t\t\t// Case 2: the workflow is corrupted\n\t\t\tif skipTask := n.fixCurrentExecution(ctx, domainID, workflowID, runID); skipTask {\n\t\t\t\treturn ErrSkipTask\n\t\t\t}\n\t\t\treturn err\n\t\tdefault:\n\t\t\treturn fmt.Errorf(\"sending replication request: %w\", err)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (n *HistoryResenderImpl) getPaginationFn(\n\tctx context.Context,\n\tremoteCluster string,\n\tdomainName string,\n\tworkflowID string,\n\trunID string,\n\tstartEventID *int64,\n\tstartEventVersion *int64,\n\tendEventID *int64,\n\tendEventVersion *int64,\n) collection.PaginationFn {\n\n\treturn func(paginationToken []byte) ([]interface{}, []byte, error) {\n\n\t\tresponse, err := n.getHistory(\n\t\t\tctx,\n\t\t\tremoteCluster,\n\t\t\tdomainName,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tstartEventID,\n\t\t\tstartEventVersion,\n\t\t\tendEventID,\n\t\t\tendEventVersion,\n\t\t\tpaginationToken,\n\t\t\tdefaultPageSize,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\n\t\tvar paginateItems []interface{}\n\t\tversionHistory := response.GetVersionHistory()\n\t\tfor _, history := range response.GetHistoryBatches() {\n\t\t\tbatch := &historyBatch{\n\t\t\t\tversionHistory: versionHistory,\n\t\t\t\trawEventBatch:  history,\n\t\t\t}\n\t\t\tpaginateItems = append(paginateItems, batch)\n\t\t}\n\t\treturn paginateItems, response.NextPageToken, nil\n\t}\n}\n\nfunc (n *HistoryResenderImpl) createReplicationRawRequest(\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\thistoryBlob *types.DataBlob,\n\tversionHistoryItems []*types.VersionHistoryItem,\n) *types.ReplicateEventsV2Request {\n\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tEvents:              historyBlob,\n\t\tVersionHistoryItems: versionHistoryItems,\n\t}\n\treturn request\n}\n\nfunc (n *HistoryResenderImpl) sendReplicationRawRequest(\n\tctx context.Context,\n\trequest *types.ReplicateEventsV2Request,\n) error {\n\n\tctx, cancel := context.WithTimeout(ctx, resendContextTimeout)\n\tdefer cancel()\n\treturn n.historyReplicationFn(ctx, request)\n}\n\nfunc (n *HistoryResenderImpl) getHistory(\n\tctx context.Context,\n\tremoteCluster string,\n\tdomainName string,\n\tworkflowID string,\n\trunID string,\n\tstartEventID *int64,\n\tstartEventVersion *int64,\n\tendEventID *int64,\n\tendEventVersion *int64,\n\ttoken []byte,\n\tpageSize int32,\n) (*types.GetWorkflowExecutionRawHistoryV2Response, error) {\n\n\tctx, cancel := context.WithTimeout(ctx, resendContextTimeout)\n\tdefer cancel()\n\tadminClient, err := n.clientBean.GetRemoteAdminClient(remoteCluster)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresponse, err := adminClient.GetWorkflowExecutionRawHistoryV2(ctx, &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tStartEventID:      startEventID,\n\t\tStartEventVersion: startEventVersion,\n\t\tEndEventID:        endEventID,\n\t\tEndEventVersion:   endEventVersion,\n\t\tMaximumPageSize:   pageSize,\n\t\tNextPageToken:     token,\n\t})\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"getting history: %w\", err)\n\t}\n\n\treturn response, nil\n}\n\nfunc (n *HistoryResenderImpl) fixCurrentExecution(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) bool {\n\n\tif n.currentExecutionCheck == nil {\n\t\treturn false\n\t}\n\texecution := &entity.CurrentExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t},\n\t}\n\tres := n.currentExecutionCheck.Check(ctx, execution)\n\tswitch res.CheckResultType {\n\tcase invariant.CheckResultTypeCorrupted:\n\t\tn.logger.Error(\n\t\t\t\"Encounter corrupted workflow\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t)\n\t\tn.currentExecutionCheck.Fix(ctx, execution)\n\t\treturn false\n\tcase invariant.CheckResultTypeFailed:\n\t\treturn false\n\tdefault:\n\t\treturn true\n\t}\n}\n"
  },
  {
    "path": "common/ndc/history_resender_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: history_resender.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source history_resender.go -destination history_resender_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockHistoryResender is a mock of HistoryResender interface.\ntype MockHistoryResender struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHistoryResenderMockRecorder\n\tisgomock struct{}\n}\n\n// MockHistoryResenderMockRecorder is the mock recorder for MockHistoryResender.\ntype MockHistoryResenderMockRecorder struct {\n\tmock *MockHistoryResender\n}\n\n// NewMockHistoryResender creates a new mock instance.\nfunc NewMockHistoryResender(ctrl *gomock.Controller) *MockHistoryResender {\n\tmock := &MockHistoryResender{ctrl: ctrl}\n\tmock.recorder = &MockHistoryResenderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHistoryResender) EXPECT() *MockHistoryResenderMockRecorder {\n\treturn m.recorder\n}\n\n// SendSingleWorkflowHistory mocks base method.\nfunc (m *MockHistoryResender) SendSingleWorkflowHistory(remoteCluster, domainID, workflowID, runID string, startEventID, startEventVersion, endEventID, endEventVersion *int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SendSingleWorkflowHistory\", remoteCluster, domainID, workflowID, runID, startEventID, startEventVersion, endEventID, endEventVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SendSingleWorkflowHistory indicates an expected call of SendSingleWorkflowHistory.\nfunc (mr *MockHistoryResenderMockRecorder) SendSingleWorkflowHistory(remoteCluster, domainID, workflowID, runID, startEventID, startEventVersion, endEventID, endEventVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SendSingleWorkflowHistory\", reflect.TypeOf((*MockHistoryResender)(nil).SendSingleWorkflowHistory), remoteCluster, domainID, workflowID, runID, startEventID, startEventVersion, endEventID, endEventVersion)\n}\n"
  },
  {
    "path": "common/ndc/history_resender_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\thistoryResenderSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller        *gomock.Controller\n\t\tmockDomainCache   *cache.MockDomainCache\n\t\tmockClientBean    *client.MockBean\n\t\tmockAdminClient   *admin.MockClient\n\t\tmockHistoryClient *history.MockClient\n\n\t\tdomainID   string\n\t\tdomainName string\n\n\t\tserializer persistence.PayloadSerializer\n\t\tlogger     log.Logger\n\n\t\trereplicator *HistoryResenderImpl\n\t}\n)\n\nfunc TestHistoryResenderSuite(t *testing.T) {\n\ts := new(historyResenderSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *historyResenderSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockAdminClient = admin.NewMockClient(s.controller)\n\ts.mockHistoryClient = history.NewMockClient(s.controller)\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockClientBean = client.NewMockBean(s.controller)\n\ts.logger = testlogger.New(s.Suite.T())\n\n\ts.domainID = uuid.New()\n\ts.domainName = \"some random domain name\"\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.domainID, Name: s.domainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainName(s.domainID).Return(s.domainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(s.domainName).Return(domainEntry, nil).AnyTimes()\n\ts.serializer = persistence.NewPayloadSerializer()\n\n\ts.rereplicator = NewHistoryResender(\n\t\ts.mockDomainCache,\n\t\ts.mockClientBean,\n\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\treturn s.mockHistoryClient.ReplicateEventsV2(ctx, request)\n\t\t},\n\t\tnil,\n\t\tnil,\n\t\ts.logger,\n\t)\n}\n\nfunc (s *historyResenderSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *historyResenderSuite) TestSendSingleWorkflowHistory() {\n\tremoteCluster := \"remote-cluster\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tstartEventID := int64(123)\n\tstartEventVersion := int64(100)\n\ttoken := []byte{1}\n\tpageSize := defaultPageSize\n\teventBatch := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        2,\n\t\t\tVersion:   123,\n\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t},\n\t\t{\n\t\t\tID:        3,\n\t\t\tVersion:   123,\n\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t},\n\t}\n\tblob := s.serializeEvents(eventBatch)\n\tversionHistoryItems := []*types.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 1,\n\t\t\tVersion: 1,\n\t\t},\n\t}\n\n\ts.mockClientBean.EXPECT().GetRemoteAdminClient(remoteCluster).Return(s.mockAdminClient, nil).Times(2)\n\ts.mockAdminClient.EXPECT().GetWorkflowExecutionRawHistoryV2(\n\t\tgomock.Any(),\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(startEventID),\n\t\t\tStartEventVersion: common.Int64Ptr(startEventVersion),\n\t\t\tMaximumPageSize:   pageSize,\n\t\t\tNextPageToken:     nil,\n\t\t}).Return(&types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tHistoryBatches: []*types.DataBlob{blob},\n\t\tNextPageToken:  token,\n\t\tVersionHistory: &types.VersionHistory{\n\t\t\tItems: versionHistoryItems,\n\t\t},\n\t}, nil).Times(1)\n\n\ts.mockAdminClient.EXPECT().GetWorkflowExecutionRawHistoryV2(\n\t\tgomock.Any(),\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(startEventID),\n\t\t\tStartEventVersion: common.Int64Ptr(startEventVersion),\n\t\t\tMaximumPageSize:   pageSize,\n\t\t\tNextPageToken:     token,\n\t\t}).Return(&types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tHistoryBatches: []*types.DataBlob{blob},\n\t\tNextPageToken:  nil,\n\t\tVersionHistory: &types.VersionHistory{\n\t\t\tItems: versionHistoryItems,\n\t\t},\n\t}, nil).Times(1)\n\n\ts.mockHistoryClient.EXPECT().ReplicateEventsV2(\n\t\tgomock.Any(),\n\t\t&types.ReplicateEventsV2Request{\n\t\t\tDomainUUID: s.domainID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tVersionHistoryItems: versionHistoryItems,\n\t\t\tEvents:              blob,\n\t\t}).Return(nil).Times(2)\n\n\terr := s.rereplicator.SendSingleWorkflowHistory(\n\t\tremoteCluster,\n\t\ts.domainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tcommon.Int64Ptr(startEventID),\n\t\tcommon.Int64Ptr(startEventVersion),\n\t\tnil,\n\t\tnil,\n\t)\n\n\ts.Nil(err)\n}\n\nfunc (s *historyResenderSuite) TestCreateReplicateRawEventsRequest() {\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tblob := &types.DataBlob{\n\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\tData:         []byte(\"some random history blob\"),\n\t}\n\tversionHistoryItems := []*types.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 1,\n\t\t\tVersion: 1,\n\t\t},\n\t}\n\n\ts.Equal(&types.ReplicateEventsV2Request{\n\t\tDomainUUID: s.domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tVersionHistoryItems: versionHistoryItems,\n\t\tEvents:              blob,\n\t}, s.rereplicator.createReplicationRawRequest(\n\t\ts.domainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tblob,\n\t\tversionHistoryItems))\n}\n\nfunc (s *historyResenderSuite) TestSendReplicationRawRequest() {\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\titem := &types.VersionHistoryItem{\n\t\tEventID: 1,\n\t\tVersion: 1,\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: s.domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tEvents: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\tData:         []byte(\"some random history blob\"),\n\t\t},\n\t\tVersionHistoryItems: []*types.VersionHistoryItem{item},\n\t}\n\n\ts.mockHistoryClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil).Times(1)\n\terr := s.rereplicator.sendReplicationRawRequest(context.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *historyResenderSuite) TestSendReplicationRawRequest_Err() {\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\titem := &types.VersionHistoryItem{\n\t\tEventID: 1,\n\t\tVersion: 1,\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: s.domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tEvents: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\tData:         []byte(\"some random history blob\"),\n\t\t},\n\t\tVersionHistoryItems: []*types.VersionHistoryItem{item},\n\t}\n\tretryErr := &types.RetryTaskV2Error{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\ts.mockHistoryClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(retryErr).Times(1)\n\terr := s.rereplicator.sendReplicationRawRequest(context.Background(), request)\n\ts.Equal(retryErr, err)\n}\n\nfunc (s *historyResenderSuite) TestGetHistory() {\n\tremoteCluster := \"remote-cluster\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tstartEventID := int64(123)\n\tendEventID := int64(345)\n\tversion := int64(20)\n\tnextTokenIn := []byte(\"some random next token in\")\n\tnextTokenOut := []byte(\"some random next token out\")\n\tpageSize := int32(59)\n\tblob := []byte(\"some random events blob\")\n\tencodingTypeThriftRW := types.EncodingTypeThriftRW\n\n\tresponse := &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tHistoryBatches: []*types.DataBlob{&types.DataBlob{\n\t\t\tEncodingType: &encodingTypeThriftRW,\n\t\t\tData:         blob,\n\t\t}},\n\t\tNextPageToken: nextTokenOut,\n\t}\n\ts.mockClientBean.EXPECT().GetRemoteAdminClient(remoteCluster).Return(s.mockAdminClient, nil).Times(1)\n\ts.mockAdminClient.EXPECT().GetWorkflowExecutionRawHistoryV2(gomock.Any(), &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tStartEventID:      common.Int64Ptr(startEventID),\n\t\tStartEventVersion: common.Int64Ptr(version),\n\t\tEndEventID:        common.Int64Ptr(endEventID),\n\t\tEndEventVersion:   common.Int64Ptr(version),\n\t\tMaximumPageSize:   pageSize,\n\t\tNextPageToken:     nextTokenIn,\n\t}).Return(response, nil).Times(1)\n\n\tout, err := s.rereplicator.getHistory(\n\t\tcontext.Background(),\n\t\tremoteCluster,\n\t\ts.domainName,\n\t\tworkflowID,\n\t\trunID,\n\t\t&startEventID,\n\t\t&version,\n\t\t&endEventID,\n\t\t&version,\n\t\tnextTokenIn,\n\t\tpageSize)\n\ts.Nil(err)\n\ts.Equal(response, out)\n}\n\nfunc (s *historyResenderSuite) TestCurrentExecutionCheck() {\n\tdomainID := uuid.New()\n\tworkflowID1 := uuid.New()\n\tworkflowID2 := uuid.New()\n\trunID := uuid.New()\n\tinvariantMock := invariant.NewMockInvariant(s.controller)\n\ts.rereplicator = NewHistoryResender(\n\t\ts.mockDomainCache,\n\t\ts.mockClientBean,\n\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\treturn s.mockHistoryClient.ReplicateEventsV2(ctx, request)\n\t\t},\n\t\tnil,\n\t\tinvariantMock,\n\t\ts.logger,\n\t)\n\texecution1 := &entity.CurrentExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID1,\n\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t},\n\t}\n\texecution2 := &entity.CurrentExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID2,\n\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t},\n\t}\n\tinvariantMock.EXPECT().Check(gomock.Any(), execution1).Return(invariant.CheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t}).Times(1)\n\tinvariantMock.EXPECT().Check(gomock.Any(), execution2).Return(invariant.CheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeHealthy,\n\t}).Times(1)\n\tinvariantMock.EXPECT().Fix(gomock.Any(), gomock.Any()).Return(invariant.FixResult{}).Times(1)\n\n\tskipTask := s.rereplicator.fixCurrentExecution(context.Background(), domainID, workflowID1, runID)\n\ts.False(skipTask)\n\tskipTask = s.rereplicator.fixCurrentExecution(context.Background(), domainID, workflowID2, runID)\n\ts.True(skipTask)\n}\n\nfunc (s *historyResenderSuite) serializeEvents(events []*types.HistoryEvent) *types.DataBlob {\n\tblob, err := s.serializer.SerializeBatchEvents(events, constants.EncodingTypeThriftRW)\n\ts.Nil(err)\n\treturn &types.DataBlob{\n\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\tData:         blob.Data,\n\t}\n}\n"
  },
  {
    "path": "common/pagination/interface.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mocks.go -self_package github.com/uber/cadence/common/pagination\n\npackage pagination\n\nimport (\n\t\"context\"\n\t\"errors\"\n)\n\n// ErrIteratorFinished indicates that Next was called on a finished iterator\nvar ErrIteratorFinished = errors.New(\"iterator has reached end\")\n\ntype (\n\t// Page contains a PageToken which identifies the current page,\n\t// a PageToken which identifies the next page and a list of Entity.\n\tPage struct {\n\t\tNextToken    PageToken\n\t\tCurrentToken PageToken\n\t\tEntities     []Entity\n\t}\n\t// Entity is a generic type which can be operated on by Iterator and Writer\n\tEntity interface{}\n\t// PageToken identifies a page\n\tPageToken interface{}\n)\n\ntype (\n\t// WriteFn writes given Page to underlying sink.\n\t// The Pages's NextToken will always be nil, its the responsibility of WriteFn to\n\t// construct and return the next PageToken, or return an error on failure.\n\tWriteFn func(Page) (PageToken, error)\n\t// ShouldFlushFn returns true if given page should be flushed false otherwise.\n\tShouldFlushFn func(Page) bool\n\t// FetchFn fetches Page from PageToken.\n\t// Once a page with nil NextToken is returned no more pages will be fetched.\n\tFetchFn func(context.Context, PageToken) (Page, error)\n)\n\ntype (\n\t// Iterator is used to get entities from a collection of pages.\n\t// When HasNext returns true it is guaranteed that Next will not return an error.\n\t// Once iterator returns an error it will never make progress again and will always return that same error.\n\t// Iterator is not thread safe and does not make defensive in or out copies.\n\tIterator interface {\n\t\tNext() (Entity, error)\n\t\tHasNext() bool\n\t}\n\t// Writer is used to buffer and write entities to underlying store.\n\tWriter interface {\n\t\tAdd(Entity) error\n\t\tFlush() error\n\t\tFlushIfNotEmpty() error\n\t\tFlushedPages() []PageToken\n\t\tFirstFlushedPage() PageToken\n\t\tLastFlushedPage() PageToken\n\t}\n)\n"
  },
  {
    "path": "common/pagination/iterator.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pagination\n\nimport \"context\"\n\ntype (\n\titerator struct {\n\t\tctx context.Context\n\n\t\tpage        Page\n\t\tentityIndex int\n\n\t\tnextEntity Entity\n\t\tnextError  error\n\n\t\tfetchFn FetchFn\n\t}\n)\n\n// NewIterator constructs a new Iterator\nfunc NewIterator(\n\tctx context.Context,\n\tstartingPageToken PageToken,\n\tfetchFn FetchFn,\n) Iterator {\n\titr := &iterator{\n\t\tctx: ctx,\n\t\tpage: Page{\n\t\t\tEntities:     nil,\n\t\t\tCurrentToken: nil,\n\t\t\tNextToken:    startingPageToken,\n\t\t},\n\t\tentityIndex: 0,\n\t\tfetchFn:     fetchFn,\n\t}\n\titr.advance(true)\n\treturn itr\n}\n\n// Next returns the next Entity or error.\n// Returning nil, nil is valid if that is what the provided fetch function provided.\nfunc (i *iterator) Next() (Entity, error) {\n\tentity := i.nextEntity\n\terr := i.nextError\n\ti.advance(false)\n\treturn entity, err\n}\n\n// HasNext returns true if there is a next element. There is considered to be a next element\n// As long as a fatal error has not occurred and the iterator has not reached the end.\nfunc (i *iterator) HasNext() bool {\n\treturn i.nextError == nil\n}\n\nfunc (i *iterator) advance(firstPage bool) {\n\tif !i.HasNext() && !firstPage {\n\t\treturn\n\t}\n\tif i.entityIndex < len(i.page.Entities) {\n\t\ti.consume()\n\t} else {\n\t\tif err := i.advanceToNonEmptyPage(firstPage); err != nil {\n\t\t\ti.terminate(err)\n\t\t} else {\n\t\t\ti.consume()\n\t\t}\n\t}\n}\n\nfunc (i *iterator) advanceToNonEmptyPage(firstPage bool) error {\n\tif i.page.NextToken == nil && !firstPage {\n\t\treturn ErrIteratorFinished\n\t}\n\tnextPage, err := i.fetchFn(i.ctx, i.page.NextToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\ti.page = nextPage\n\tif len(i.page.Entities) != 0 {\n\t\ti.entityIndex = 0\n\t\treturn nil\n\t}\n\treturn i.advanceToNonEmptyPage(false)\n}\n\nfunc (i *iterator) consume() {\n\ti.nextEntity = i.page.Entities[i.entityIndex]\n\ti.nextError = nil\n\ti.entityIndex++\n}\n\nfunc (i *iterator) terminate(err error) {\n\ti.nextEntity = nil\n\ti.nextError = err\n}\n"
  },
  {
    "path": "common/pagination/iterator_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pagination\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\nvar fetchMap = map[PageToken][]Entity{\n\t0: nil,\n\t1: {},\n\t2: {\"one\", \"two\", \"three\"},\n\t3: {\"four\", \"five\", \"six\", \"seven\"},\n\t4: {\"eight\"},\n}\n\ntype IteratorSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestIteratorSuite(t *testing.T) {\n\tsuite.Run(t, new(IteratorSuite))\n}\n\nfunc (s *IteratorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *IteratorSuite) TestInitializedToEmpty() {\n\tfetchFn := func(_ context.Context, token PageToken) (Page, error) {\n\t\tif token.(int) == 2 {\n\t\t\treturn Page{\n\t\t\t\tCurrentToken: token,\n\t\t\t\tNextToken:    nil,\n\t\t\t\tEntities:     nil,\n\t\t\t}, nil\n\t\t}\n\t\treturn Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    token.(int) + 1,\n\t\t\tEntities:     fetchMap[token],\n\t\t}, nil\n\t}\n\titr := NewIterator(context.Background(), 0, fetchFn)\n\ts.False(itr.HasNext())\n\t_, err := itr.Next()\n\ts.Equal(ErrIteratorFinished, err)\n}\n\nfunc (s *IteratorSuite) TestNonEmptyNoErrors() {\n\tfetchFn := func(_ context.Context, token PageToken) (Page, error) {\n\t\tvar nextPageToken interface{} = token.(int) + 1\n\t\tif nextPageToken.(int) == 5 {\n\t\t\tnextPageToken = nil\n\t\t}\n\t\treturn Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextPageToken,\n\t\t\tEntities:     fetchMap[token],\n\t\t}, nil\n\t}\n\titr := NewIterator(context.Background(), 0, fetchFn)\n\texpectedResults := []string{\"one\", \"two\", \"three\", \"four\", \"five\", \"six\", \"seven\", \"eight\"}\n\ti := 0\n\tfor itr.HasNext() {\n\t\tcurr, err := itr.Next()\n\t\ts.NoError(err)\n\t\ts.Equal(expectedResults[i], curr.(string))\n\t\ti++\n\t}\n\ts.False(itr.HasNext())\n\t_, err := itr.Next()\n\ts.Equal(ErrIteratorFinished, err)\n}\n\nfunc (s *IteratorSuite) TestNonEmptyWithErrors() {\n\tfetchFn := func(_ context.Context, token PageToken) (Page, error) {\n\t\tif token.(int) == 4 {\n\t\t\treturn Page{}, errors.New(\"got error\")\n\t\t}\n\t\treturn Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    token.(int) + 1,\n\t\t\tEntities:     fetchMap[token],\n\t\t}, nil\n\t}\n\titr := NewIterator(context.Background(), 0, fetchFn)\n\texpectedResults := []string{\"one\", \"two\", \"three\", \"four\", \"five\", \"six\", \"seven\"}\n\ti := 0\n\tfor itr.HasNext() {\n\t\tcurr, err := itr.Next()\n\t\ts.NoError(err)\n\t\ts.Equal(expectedResults[i], curr.(string))\n\t\ti++\n\t}\n\ts.False(itr.HasNext())\n\tcurr, err := itr.Next()\n\ts.Nil(curr)\n\ts.Equal(\"got error\", err.Error())\n}\n"
  },
  {
    "path": "common/pagination/mocks.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package pagination -source interface.go -destination mocks.go -self_package github.com/uber/cadence/common/pagination\n//\n\n// Package pagination is a generated GoMock package.\npackage pagination\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockEntity is a mock of Entity interface.\ntype MockEntity struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEntityMockRecorder\n\tisgomock struct{}\n}\n\n// MockEntityMockRecorder is the mock recorder for MockEntity.\ntype MockEntityMockRecorder struct {\n\tmock *MockEntity\n}\n\n// NewMockEntity creates a new mock instance.\nfunc NewMockEntity(ctrl *gomock.Controller) *MockEntity {\n\tmock := &MockEntity{ctrl: ctrl}\n\tmock.recorder = &MockEntityMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEntity) EXPECT() *MockEntityMockRecorder {\n\treturn m.recorder\n}\n\n// MockPageToken is a mock of PageToken interface.\ntype MockPageToken struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPageTokenMockRecorder\n\tisgomock struct{}\n}\n\n// MockPageTokenMockRecorder is the mock recorder for MockPageToken.\ntype MockPageTokenMockRecorder struct {\n\tmock *MockPageToken\n}\n\n// NewMockPageToken creates a new mock instance.\nfunc NewMockPageToken(ctrl *gomock.Controller) *MockPageToken {\n\tmock := &MockPageToken{ctrl: ctrl}\n\tmock.recorder = &MockPageTokenMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPageToken) EXPECT() *MockPageTokenMockRecorder {\n\treturn m.recorder\n}\n\n// MockIterator is a mock of Iterator interface.\ntype MockIterator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockIteratorMockRecorder\n\tisgomock struct{}\n}\n\n// MockIteratorMockRecorder is the mock recorder for MockIterator.\ntype MockIteratorMockRecorder struct {\n\tmock *MockIterator\n}\n\n// NewMockIterator creates a new mock instance.\nfunc NewMockIterator(ctrl *gomock.Controller) *MockIterator {\n\tmock := &MockIterator{ctrl: ctrl}\n\tmock.recorder = &MockIteratorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockIterator) EXPECT() *MockIteratorMockRecorder {\n\treturn m.recorder\n}\n\n// HasNext mocks base method.\nfunc (m *MockIterator) HasNext() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasNext\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasNext indicates an expected call of HasNext.\nfunc (mr *MockIteratorMockRecorder) HasNext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasNext\", reflect.TypeOf((*MockIterator)(nil).HasNext))\n}\n\n// Next mocks base method.\nfunc (m *MockIterator) Next() (Entity, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Next\")\n\tret0, _ := ret[0].(Entity)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Next indicates an expected call of Next.\nfunc (mr *MockIteratorMockRecorder) Next() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Next\", reflect.TypeOf((*MockIterator)(nil).Next))\n}\n\n// MockWriter is a mock of Writer interface.\ntype MockWriter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWriterMockRecorder\n\tisgomock struct{}\n}\n\n// MockWriterMockRecorder is the mock recorder for MockWriter.\ntype MockWriterMockRecorder struct {\n\tmock *MockWriter\n}\n\n// NewMockWriter creates a new mock instance.\nfunc NewMockWriter(ctrl *gomock.Controller) *MockWriter {\n\tmock := &MockWriter{ctrl: ctrl}\n\tmock.recorder = &MockWriterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWriter) EXPECT() *MockWriterMockRecorder {\n\treturn m.recorder\n}\n\n// Add mocks base method.\nfunc (m *MockWriter) Add(arg0 Entity) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Add\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Add indicates an expected call of Add.\nfunc (mr *MockWriterMockRecorder) Add(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Add\", reflect.TypeOf((*MockWriter)(nil).Add), arg0)\n}\n\n// FirstFlushedPage mocks base method.\nfunc (m *MockWriter) FirstFlushedPage() PageToken {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FirstFlushedPage\")\n\tret0, _ := ret[0].(PageToken)\n\treturn ret0\n}\n\n// FirstFlushedPage indicates an expected call of FirstFlushedPage.\nfunc (mr *MockWriterMockRecorder) FirstFlushedPage() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FirstFlushedPage\", reflect.TypeOf((*MockWriter)(nil).FirstFlushedPage))\n}\n\n// Flush mocks base method.\nfunc (m *MockWriter) Flush() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Flush\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Flush indicates an expected call of Flush.\nfunc (mr *MockWriterMockRecorder) Flush() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Flush\", reflect.TypeOf((*MockWriter)(nil).Flush))\n}\n\n// FlushIfNotEmpty mocks base method.\nfunc (m *MockWriter) FlushIfNotEmpty() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FlushIfNotEmpty\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// FlushIfNotEmpty indicates an expected call of FlushIfNotEmpty.\nfunc (mr *MockWriterMockRecorder) FlushIfNotEmpty() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FlushIfNotEmpty\", reflect.TypeOf((*MockWriter)(nil).FlushIfNotEmpty))\n}\n\n// FlushedPages mocks base method.\nfunc (m *MockWriter) FlushedPages() []PageToken {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FlushedPages\")\n\tret0, _ := ret[0].([]PageToken)\n\treturn ret0\n}\n\n// FlushedPages indicates an expected call of FlushedPages.\nfunc (mr *MockWriterMockRecorder) FlushedPages() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FlushedPages\", reflect.TypeOf((*MockWriter)(nil).FlushedPages))\n}\n\n// LastFlushedPage mocks base method.\nfunc (m *MockWriter) LastFlushedPage() PageToken {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LastFlushedPage\")\n\tret0, _ := ret[0].(PageToken)\n\treturn ret0\n}\n\n// LastFlushedPage indicates an expected call of LastFlushedPage.\nfunc (mr *MockWriterMockRecorder) LastFlushedPage() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LastFlushedPage\", reflect.TypeOf((*MockWriter)(nil).LastFlushedPage))\n}\n"
  },
  {
    "path": "common/pagination/writer.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pagination\n\ntype (\n\twriter struct {\n\t\twriteFn       WriteFn\n\t\tshouldFlushFn ShouldFlushFn\n\t\tflushedPages  []PageToken\n\t\tpage          Page\n\t}\n)\n\n// NewWriter constructs a new Writer\nfunc NewWriter(\n\twriteFn WriteFn,\n\tshouldFlushFn ShouldFlushFn,\n\tstartingPage PageToken,\n) Writer {\n\treturn &writer{\n\t\twriteFn:       writeFn,\n\t\tshouldFlushFn: shouldFlushFn,\n\t\tflushedPages:  nil,\n\t\tpage: Page{\n\t\t\tEntities:     nil,\n\t\t\tCurrentToken: startingPage,\n\t\t},\n\t}\n}\n\n// Add adds entity to buffer and flushes if provided shouldFlushFn indicates the page should be flushed.\nfunc (w *writer) Add(e Entity) error {\n\tw.page.Entities = append(w.page.Entities, e)\n\tif !w.shouldFlushFn(w.page) {\n\t\treturn nil\n\t}\n\treturn w.Flush()\n}\n\n// Flush flushes the buffer.\nfunc (w *writer) Flush() error {\n\tnextPageToken, err := w.writeFn(w.page)\n\tif err != nil {\n\t\treturn err\n\t}\n\tw.flushedPages = append(w.flushedPages, w.page.CurrentToken)\n\tw.page = Page{\n\t\tEntities:     nil,\n\t\tCurrentToken: nextPageToken,\n\t}\n\treturn nil\n}\n\n// FlushIfNotEmpty flushes the buffer if and only if it is not empty\nfunc (w *writer) FlushIfNotEmpty() error {\n\tif len(w.page.Entities) == 0 {\n\t\treturn nil\n\t}\n\treturn w.Flush()\n}\n\n// FlushedPages returns all pages which have been successfully flushed.\nfunc (w *writer) FlushedPages() []PageToken {\n\treturn w.flushedPages\n}\n\n// FirstFlushedPage returns the first page that was flushed or nil if no pages have been flushed.\nfunc (w *writer) FirstFlushedPage() PageToken {\n\tif len(w.flushedPages) == 0 {\n\t\treturn nil\n\t}\n\treturn w.flushedPages[0]\n}\n\n// LastFlushedPage returns the last page that was flushed or nil if no pages have been flushed\nfunc (w *writer) LastFlushedPage() PageToken {\n\tif len(w.flushedPages) == 0 {\n\t\treturn nil\n\t}\n\treturn w.flushedPages[len(w.flushedPages)-1]\n}\n"
  },
  {
    "path": "common/pagination/writerIterator_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pagination\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype TestEntity struct {\n\tCounter int\n}\n\nvar separator = []byte(\"\\r\\n\")\n\ntype WriterIteratorSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestWriterIteratorSuite(t *testing.T) {\n\tsuite.Run(t, new(WriterIteratorSuite))\n}\n\nfunc (s *WriterIteratorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *WriterIteratorSuite) TestWriterIterator() {\n\tstore := make(map[string][]byte)\n\n\tshouldFlushFn := func(page Page) bool {\n\t\treturn len(page.Entities) == 10\n\t}\n\tpageNum := 0\n\twriteFn := func(page Page) (PageToken, error) {\n\t\tbuffer := &bytes.Buffer{}\n\t\tfor _, e := range page.Entities {\n\t\t\tdata, err := json.Marshal(e)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tbuffer.Write(data)\n\t\t\tbuffer.Write(separator)\n\t\t}\n\t\tstore[page.CurrentToken.(string)] = buffer.Bytes()\n\t\tpageNum++\n\t\treturn fmt.Sprintf(\"key_%v\", pageNum), nil\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, fmt.Sprintf(\"key_%v\", pageNum))\n\tfor i := 0; i < 100; i++ {\n\t\tte := TestEntity{\n\t\t\tCounter: i,\n\t\t}\n\t\ts.NoError(writer.Add(te))\n\t}\n\tflushedKeys := writer.FlushedPages()\n\ts.Len(flushedKeys, 10)\n\tfor i := 0; i < 10; i++ {\n\t\texpectedKey := fmt.Sprintf(\"key_%v\", i)\n\t\ts.Equal(expectedKey, flushedKeys[i].(string))\n\t}\n\n\tfetchFn := func(_ context.Context, token PageToken) (Page, error) {\n\t\tkey := flushedKeys[token.(int)]\n\t\tdata := store[key.(string)]\n\t\tdataBlobs := bytes.Split(data, separator)\n\t\tvar entities []Entity\n\t\tfor _, db := range dataBlobs {\n\t\t\tif len(db) == 0 {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tvar entity TestEntity\n\t\t\tif err := json.Unmarshal(db, &entity); err != nil {\n\t\t\t\treturn Page{}, err\n\t\t\t}\n\t\t\tentities = append(entities, entity)\n\t\t}\n\t\tvar nextPageToken interface{} = token.(int) + 1\n\t\tif nextPageToken == len(flushedKeys) {\n\t\t\tnextPageToken = nil\n\t\t}\n\t\treturn Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextPageToken,\n\t\t\tEntities:     entities,\n\t\t}, nil\n\t}\n\n\titr := NewIterator(context.Background(), 0, fetchFn)\n\titrCount := 0\n\tfor itr.HasNext() {\n\t\tval, err := itr.Next()\n\t\ts.NoError(err)\n\t\tte := val.(TestEntity)\n\t\ts.Equal(itrCount, te.Counter)\n\t\titrCount++\n\t}\n\ts.Equal(100, itrCount)\n}\n"
  },
  {
    "path": "common/pagination/writer_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pagination\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype WriterSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestWriterSuite(t *testing.T) {\n\tsuite.Run(t, new(WriterSuite))\n}\n\nfunc (s *WriterSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *WriterSuite) TestAddNoFlush() {\n\twriteFn := func(_ Page) (PageToken, error) {\n\t\treturn nil, nil\n\t}\n\tshouldFlushFn := func(_ Page) bool {\n\t\treturn false\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, nil)\n\ts.NoError(writer.Add(1))\n\ts.Empty(writer.FlushedPages())\n\ts.Nil(writer.FirstFlushedPage())\n\ts.Nil(writer.LastFlushedPage())\n}\n\nfunc (s *WriterSuite) TestAddAndFlush() {\n\twriteFn := func(_ Page) (PageToken, error) {\n\t\treturn nil, nil\n\t}\n\tshouldFlushFn := func(_ Page) bool {\n\t\treturn true\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, \"test_token\")\n\ts.NoError(writer.Add(1))\n\tflushedPages := writer.FlushedPages()\n\ts.Len(flushedPages, 1)\n\ts.Equal(\"test_token\", flushedPages[0].(string))\n\ts.Equal(\"test_token\", writer.FirstFlushedPage().(string))\n\ts.Equal(\"test_token\", writer.LastFlushedPage().(string))\n}\n\nfunc (s *WriterSuite) TestFlushErrorOnWrite() {\n\twriteFn := func(_ Page) (PageToken, error) {\n\t\treturn nil, errors.New(\"got error\")\n\t}\n\tshouldFlushFn := func(_ Page) bool {\n\t\treturn true\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, nil)\n\ts.Error(writer.Flush())\n}\n\nfunc (s *WriterSuite) TestFlushNoError() {\n\twriteFn := func(_ Page) (PageToken, error) {\n\t\treturn nil, nil\n\t}\n\tshouldFlushFn := func(_ Page) bool {\n\t\treturn true\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, \"test_token\")\n\ts.Len(writer.FlushedPages(), 0)\n\ts.NoError(writer.Flush())\n\ts.Len(writer.FlushedPages(), 1)\n\ts.Equal(\"test_token\", writer.FlushedPages()[0].(string))\n\ts.Equal(\"test_token\", writer.FirstFlushedPage().(string))\n\ts.Equal(\"test_token\", writer.LastFlushedPage().(string))\n\ts.NoError(writer.FlushIfNotEmpty())\n\ts.Len(writer.FlushedPages(), 1)\n\ts.Equal(\"test_token\", writer.FlushedPages()[0].(string))\n\ts.Equal(\"test_token\", writer.FirstFlushedPage().(string))\n\ts.Equal(\"test_token\", writer.LastFlushedPage().(string))\n}\n\nfunc (s *WriterSuite) TestMultiPageWrite() {\n\tpageNum := 0\n\twriteFnCalls := 0\n\twriteFn := func(_ Page) (PageToken, error) {\n\t\twriteFnCalls++\n\t\tpageNum++\n\t\treturn fmt.Sprintf(\"page_%v\", pageNum), nil\n\t}\n\tshouldFlushCalls := 0\n\tshouldFlushFn := func(p Page) bool {\n\t\tshouldFlushCalls++\n\t\treturn len(p.Entities) == 5\n\t}\n\twriter := NewWriter(writeFn, shouldFlushFn, \"page_0\")\n\tfor i := 1; i <= 100; i++ {\n\t\ts.NoError(writer.Add(i))\n\t}\n\tvar expectedPages []string\n\tfor i := 0; i < 20; i++ {\n\t\texpectedPages = append(expectedPages, fmt.Sprintf(\"page_%v\", i))\n\t}\n\tactualPages := writer.FlushedPages()\n\ts.Len(actualPages, 20)\n\tfor i, expected := range expectedPages {\n\t\ts.Equal(expected, actualPages[i].(string))\n\t}\n\ts.Equal(writeFnCalls, 20)\n\ts.Equal(shouldFlushCalls, 100)\n\ts.Equal(\"page_0\", writer.FirstFlushedPage().(string))\n\ts.Equal(\"page_19\", writer.LastFlushedPage().(string))\n\n}\n"
  },
  {
    "path": "common/peerprovider/plugin.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage peerprovider\n\nimport (\n\t\"fmt\"\n\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/syncmap\"\n)\n\nconst key = \"peerprovider\"\n\n// Container is passed to peer provider plugin\ntype Container struct {\n\tService string\n\t// Channel is required by ringpop\n\tChannel tchannel.Channel\n\tLogger  log.Logger\n\tPortmap membership.PortMap\n}\n\ntype constructorFn func(cfg *config.YamlNode, container Container) (membership.PeerProvider, error)\n\nvar plugins = syncmap.New[string, plugin]()\n\ntype plugin struct {\n\tfn        constructorFn\n\tconfigKey string\n}\n\ntype Provider struct {\n\tconfig    config.PeerProvider\n\tcontainer Container\n}\n\nfunc New(config config.PeerProvider, container Container) *Provider {\n\treturn &Provider{\n\t\tconfig:    config,\n\t\tcontainer: container,\n\t}\n}\n\nfunc Register(configKey string, constructor constructorFn) error {\n\n\tinserted := plugins.Put(key, plugin{\n\t\tfn:        constructor,\n\t\tconfigKey: configKey,\n\t})\n\n\t// only one plugin is allowed to be registered\n\tif !inserted {\n\t\tregisteredPlugin, _ := plugins.Get(key)\n\t\treturn fmt.Errorf(\"cannot register %q provider, %q is already registered\", configKey, registeredPlugin.configKey)\n\t}\n\n\treturn nil\n}\n\nfunc (p *Provider) Provider() (membership.PeerProvider, error) {\n\tregisteredPlugin, found := plugins.Get(key)\n\n\tif !found {\n\t\treturn nil, fmt.Errorf(\"no configured peer providers found\")\n\t}\n\n\tfor configKey, cfg := range p.config {\n\t\tif configKey == registeredPlugin.configKey {\n\t\t\treturn registeredPlugin.fn(cfg, p.container)\n\t\t}\n\t}\n\n\treturn nil, fmt.Errorf(\"no configuration for %q peer provider found\", registeredPlugin.configKey)\n}\n"
  },
  {
    "path": "common/peerprovider/plugin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage peerprovider\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/syncmap\"\n)\n\nfunc TestProviderRetrunsErrorWhenNoProviderRegistered(t *testing.T) {\n\t// Reset plugins\n\tplugins = syncmap.New[string, plugin]()\n\ta := Provider{\n\t\tconfig:    nil,\n\t\tcontainer: Container{},\n\t}\n\tp, err := a.Provider()\n\tassert.Nil(t, p)\n\tassert.EqualError(t, err, \"no configured peer providers found\")\n}\n\nfunc TestProviderRetrunsErrorWhenPluginAlreadyRegistered(t *testing.T) {\n\t// Reset plugins\n\tplugins = syncmap.New[string, plugin]()\n\terr := Register(\"provider1\", func(cfg *config.YamlNode, container Container) (membership.PeerProvider, error) {\n\t\treturn nil, nil\n\t})\n\tassert.NoError(t, err)\n\terr = Register(\"provider2\", func(cfg *config.YamlNode, container Container) (membership.PeerProvider, error) {\n\t\treturn nil, nil\n\t})\n\tassert.Error(t, err)\n}\n\nfunc TestConfigIsPickedUp(t *testing.T) {\n\t// Reset plugins\n\tplugins = syncmap.New[string, plugin]()\n\n\tpeerProviderConfig := map[string]*config.YamlNode{}\n\tpeerProviderConfig[\"provider1\"] = &config.YamlNode{}\n\n\tpp := New(peerProviderConfig, Container{})\n\terr := Register(\"provider1\", func(cfg *config.YamlNode, container Container) (membership.PeerProvider, error) {\n\t\treturn nil, nil\n\t})\n\tassert.NoError(t, err)\n\t_, err = pp.Provider()\n\tassert.NoError(t, err)\n}\n\nfunc TestErrorWhenConfigIsNotProvided(t *testing.T) {\n\t// Reset plugins\n\tplugins = syncmap.New[string, plugin]()\n\tpp := New(config.PeerProvider{}, Container{})\n\terr := Register(\"provider1\", func(cfg *config.YamlNode, container Container) (membership.PeerProvider, error) {\n\t\treturn nil, nil\n\t})\n\tp, err := pp.Provider()\n\tassert.Nil(t, p)\n\tassert.EqualError(t, err, \"no configuration for \\\"provider1\\\" peer provider found\")\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/config/config.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/ringpop-go/discovery\"\n)\n\n// BootstrapMode is an enum type for ringpop bootstrap mode\ntype BootstrapMode int\n\nconst (\n\t// BootstrapModeNone represents a bootstrap mode set to nothing or invalid\n\tBootstrapModeNone BootstrapMode = iota\n\t// BootstrapModeFile represents a file-based bootstrap mode\n\tBootstrapModeFile\n\t// BootstrapModeHosts represents a list of hosts passed in the configuration\n\tBootstrapModeHosts\n\t// BootstrapModeCustom represents a custom bootstrap mode\n\tBootstrapModeCustom\n\t// BootstrapModeDNS represents a list of hosts passed in the configuration\n\t// to be resolved, and the resulting addresses are used for bootstrap\n\tBootstrapModeDNS\n\t// BootstrapModeDNSSRV represents a list of DNS hosts passed in the configuration\n\t// to resolve secondary addresses that DNS SRV record would return resulting in\n\t// a host list that will contain multiple dynamic addresses and their unique ports\n\tBootstrapModeDNSSRV\n)\n\nconst (\n\tdefaultMaxJoinDuration = 10 * time.Second\n)\n\n// Config contains the ringpop config items\ntype Config struct {\n\t// Name to be used in ringpop advertisement\n\tName string `yaml:\"name\" validate:\"nonzero\"`\n\t// BroadcastAddress is communicated with peers to connect to this container/host.\n\t// This is useful when running cadence in K8s and the containers need to listen on 0.0.0.0 but advertise their pod IP to peers.\n\t// If not set, the listen address will be used as broadcast address.\n\tBroadcastAddress string `yaml:\"broadcastAddress\"`\n\t// BootstrapMode is a enum that defines the ringpop bootstrap method, currently supports: hosts, files, custom, dns, and dns-srv\n\tBootstrapMode BootstrapMode `yaml:\"bootstrapMode\"`\n\t// BootstrapHosts is a list of seed hosts to be used for ringpop bootstrap\n\tBootstrapHosts []string `yaml:\"bootstrapHosts\"`\n\t// BootstrapFile is the file path to be used for ringpop bootstrap\n\tBootstrapFile string `yaml:\"bootstrapFile\"`\n\t// MaxJoinDuration is the max wait time to join the ring\n\tMaxJoinDuration time.Duration `yaml:\"maxJoinDuration\"`\n\t// Custom discovery provider, cannot be specified through yaml\n\tDiscoveryProvider discovery.DiscoverProvider `yaml:\"-\"`\n}\n\nfunc (rpConfig *Config) Validate() error {\n\tif len(rpConfig.Name) == 0 {\n\t\treturn fmt.Errorf(\"ringpop config missing `name` param\")\n\t}\n\n\tif rpConfig.MaxJoinDuration == 0 {\n\t\trpConfig.MaxJoinDuration = defaultMaxJoinDuration\n\t}\n\n\treturn validateBootstrapMode(rpConfig)\n}\n\n// UnmarshalYAML is called by the yaml package to convert\n// the config YAML into a BootstrapMode.\nfunc (m *BootstrapMode) UnmarshalYAML(\n\tunmarshal func(interface{}) error,\n) error {\n\n\tvar s string\n\tif err := unmarshal(&s); err != nil {\n\t\treturn err\n\t}\n\tvar err error\n\t*m, err = parseBootstrapMode(s)\n\treturn err\n}\n\n// parseBootstrapMode reads a string value and returns a bootstrap mode.\nfunc parseBootstrapMode(\n\tmode string,\n) (BootstrapMode, error) {\n\n\tswitch strings.ToLower(mode) {\n\tcase \"hosts\":\n\t\treturn BootstrapModeHosts, nil\n\tcase \"file\":\n\t\treturn BootstrapModeFile, nil\n\tcase \"custom\":\n\t\treturn BootstrapModeCustom, nil\n\tcase \"dns\":\n\t\treturn BootstrapModeDNS, nil\n\tcase \"dns-srv\":\n\t\treturn BootstrapModeDNSSRV, nil\n\t}\n\treturn BootstrapModeNone, fmt.Errorf(\"invalid ringpop bootstrap mode %q\", mode)\n}\n\nfunc validateBootstrapMode(\n\trpConfig *Config,\n) error {\n\n\tswitch rpConfig.BootstrapMode {\n\tcase BootstrapModeFile:\n\t\tif len(rpConfig.BootstrapFile) == 0 {\n\t\t\treturn fmt.Errorf(\"ringpop config missing bootstrap file param\")\n\t\t}\n\tcase BootstrapModeHosts, BootstrapModeDNS, BootstrapModeDNSSRV:\n\t\tif len(rpConfig.BootstrapHosts) == 0 {\n\t\t\treturn fmt.Errorf(\"ringpop config missing boostrap hosts param\")\n\t\t}\n\tcase BootstrapModeCustom:\n\t\tif rpConfig.DiscoveryProvider == nil {\n\t\t\treturn fmt.Errorf(\"ringpop bootstrapMode is set to custom but discoveryProvider is nil\")\n\t\t}\n\tdefault:\n\t\treturn fmt.Errorf(\"ringpop config with unknown boostrap mode %q\", rpConfig.BootstrapMode)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/config/config_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber/ringpop-go/discovery/statichosts\"\n\t\"gopkg.in/yaml.v2\"\n)\n\nfunc TestHostsMode(t *testing.T) {\n\tvar cfg Config\n\terr := yaml.Unmarshal([]byte(getHostsConfig()), &cfg)\n\tassert.Nil(t, err)\n\tassert.Equal(t, \"test\", cfg.Name)\n\tassert.Equal(t, BootstrapModeHosts, cfg.BootstrapMode)\n\tassert.Equal(t, []string{\"127.0.0.1:1111\"}, cfg.BootstrapHosts)\n\tassert.Equal(t, time.Second*30, cfg.MaxJoinDuration)\n\terr = cfg.Validate()\n\tassert.Nil(t, err)\n}\n\nfunc TestFileMode(t *testing.T) {\n\tvar cfg Config\n\terr := yaml.Unmarshal([]byte(getJSONConfig()), &cfg)\n\tassert.Nil(t, err)\n\tassert.Equal(t, \"test\", cfg.Name)\n\tassert.Equal(t, BootstrapModeFile, cfg.BootstrapMode)\n\tassert.Equal(t, \"/tmp/file.json\", cfg.BootstrapFile)\n\tassert.Equal(t, time.Second*30, cfg.MaxJoinDuration)\n\terr = cfg.Validate()\n\tassert.Nil(t, err)\n}\n\nfunc TestCustomMode(t *testing.T) {\n\tvar cfg Config\n\terr := yaml.Unmarshal([]byte(getCustomConfig()), &cfg)\n\tassert.Nil(t, err)\n\tassert.Equal(t, \"test\", cfg.Name)\n\tassert.Equal(t, BootstrapModeCustom, cfg.BootstrapMode)\n\tassert.NotNil(t, cfg.Validate())\n\tcfg.DiscoveryProvider = statichosts.New(\"127.0.0.1\")\n\tassert.Nil(t, cfg.Validate())\n}\n\nfunc TestInvalidConfig(t *testing.T) {\n\tvar cfg Config\n\tassert.NotNil(t, cfg.Validate())\n\tcfg.Name = \"test\"\n\tassert.NotNil(t, cfg.Validate())\n\tcfg.BootstrapMode = BootstrapModeNone\n\tassert.NotNil(t, cfg.Validate())\n\t_, err := parseBootstrapMode(\"unknown\")\n\tassert.NotNil(t, err)\n}\n\nfunc getJSONConfig() string {\n\treturn `name: \"test\"\nbootstrapMode: \"file\"\nbootstrapFile: \"/tmp/file.json\"\nmaxJoinDuration: 30s`\n}\n\nfunc getHostsConfig() string {\n\treturn `name: \"test\"\nbootstrapMode: \"hosts\"\nbootstrapHosts: [\"127.0.0.1:1111\"]\nmaxJoinDuration: 30s`\n}\n\nfunc getCustomConfig() string {\n\treturn `name: \"test\"\nbootstrapMode: \"custom\"\nmaxJoinDuration: 30s`\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/factory.go",
    "content": "package ringpopprovider\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"net\"\n\t\"strings\"\n\n\t\"github.com/uber/ringpop-go/discovery\"\n\t\"github.com/uber/ringpop-go/discovery/jsonfile\"\n\t\"github.com/uber/ringpop-go/discovery/statichosts\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n)\n\ntype dnsHostResolver interface {\n\tLookupHost(ctx context.Context, host string) (addrs []string, err error)\n\tLookupSRV(ctx context.Context, service, proto, name string) (cname string, addrs []*net.SRV, err error)\n}\n\ntype dnsProvider struct {\n\tUnresolvedHosts []string\n\tResolver        dnsHostResolver\n\tLogger          log.Logger\n}\n\nfunc newDNSProvider(\n\thosts []string,\n\tresolver dnsHostResolver,\n\tlogger log.Logger,\n) *dnsProvider {\n\n\tset := map[string]struct{}{}\n\tfor _, hostport := range hosts {\n\t\tset[hostport] = struct{}{}\n\t}\n\n\tvar keys []string\n\tfor key := range set {\n\t\tkeys = append(keys, key)\n\t}\n\treturn &dnsProvider{\n\t\tUnresolvedHosts: keys,\n\t\tResolver:        resolver,\n\t\tLogger:          logger,\n\t}\n}\n\nfunc (provider *dnsProvider) Hosts() ([]string, error) {\n\tvar results []string\n\tresolvedHosts := map[string][]string{}\n\tfor _, hostPort := range provider.UnresolvedHosts {\n\t\thost, port, err := net.SplitHostPort(hostPort)\n\t\tif err != nil {\n\t\t\tprovider.Logger.Warn(\"could not split host and port\", tag.Address(hostPort), tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\n\t\tresolved, exists := resolvedHosts[host]\n\t\tif !exists {\n\t\t\tresolved, err = provider.Resolver.LookupHost(context.Background(), host)\n\t\t\tif err != nil {\n\t\t\t\tprovider.Logger.Warn(\"could not resolve host\", tag.Address(host), tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tresolvedHosts[host] = resolved\n\t\t}\n\t\tfor _, r := range resolved {\n\t\t\tresults = append(results, net.JoinHostPort(r, port))\n\t\t}\n\t}\n\tif len(results) == 0 {\n\t\treturn nil, errors.New(\"no hosts found, and bootstrap requires at least one\")\n\t}\n\treturn results, nil\n}\n\ntype dnsSRVProvider struct {\n\tUnresolvedHosts []string\n\tResolver        dnsHostResolver\n\tLogger          log.Logger\n}\n\nfunc newDNSSRVProvider(\n\thosts []string,\n\tresolver dnsHostResolver,\n\tlogger log.Logger,\n) *dnsSRVProvider {\n\n\tset := map[string]struct{}{}\n\tfor _, hostport := range hosts {\n\t\tset[hostport] = struct{}{}\n\t}\n\n\tvar keys []string\n\tfor key := range set {\n\t\tkeys = append(keys, key)\n\t}\n\n\treturn &dnsSRVProvider{\n\t\tUnresolvedHosts: keys,\n\t\tResolver:        resolver,\n\t\tLogger:          logger,\n\t}\n}\n\nfunc (provider *dnsSRVProvider) Hosts() ([]string, error) {\n\tvar results []string\n\tresolvedHosts := map[string][]string{}\n\n\tfor _, host := range provider.UnresolvedHosts {\n\t\thostParts := strings.Split(host, \".\")\n\t\tif len(hostParts) <= 2 {\n\t\t\treturn nil, fmt.Errorf(\"could not seperate host from domain %q\", host)\n\t\t}\n\t\tserviceName := hostParts[0]\n\t\tdomain := strings.Join(hostParts[1:], \".\")\n\t\tresolved, exists := resolvedHosts[serviceName]\n\t\tif !exists {\n\t\t\t_, srvs, err := provider.Resolver.LookupSRV(context.Background(), serviceName, \"tcp\", domain)\n\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"could not resolve host: %s.%s\", serviceName, domain)\n\t\t\t}\n\n\t\t\tvar targets []string\n\t\t\tfor _, s := range srvs {\n\t\t\t\taddrs, err := provider.Resolver.LookupHost(context.Background(), s.Target)\n\n\t\t\t\tif err != nil {\n\t\t\t\t\tprovider.Logger.Warn(\"could not resolve srv dns host\", tag.Address(s.Target), tag.Error(err))\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tfor _, a := range addrs {\n\t\t\t\t\ttargets = append(targets, net.JoinHostPort(a, fmt.Sprintf(\"%d\", s.Port)))\n\t\t\t\t}\n\t\t\t}\n\t\t\tresolvedHosts[serviceName] = targets\n\t\t\tresolved = targets\n\t\t}\n\n\t\tresults = append(results, resolved...)\n\t}\n\n\tif len(results) == 0 {\n\t\treturn nil, errors.New(\"no hosts found, and bootstrap requires at least one\")\n\t}\n\treturn results, nil\n}\n\nfunc newDiscoveryProvider(\n\tcfg *config.Config,\n\tlogger log.Logger,\n) (discovery.DiscoverProvider, error) {\n\n\tif cfg.DiscoveryProvider != nil {\n\t\t// custom discovery provider takes first precedence\n\t\treturn cfg.DiscoveryProvider, nil\n\t}\n\n\tswitch cfg.BootstrapMode {\n\tcase config.BootstrapModeHosts:\n\t\treturn statichosts.New(cfg.BootstrapHosts...), nil\n\tcase config.BootstrapModeFile:\n\t\treturn jsonfile.New(cfg.BootstrapFile), nil\n\tcase config.BootstrapModeDNS:\n\t\treturn newDNSProvider(cfg.BootstrapHosts, net.DefaultResolver, logger), nil\n\tcase config.BootstrapModeDNSSRV:\n\t\treturn newDNSSRVProvider(cfg.BootstrapHosts, net.DefaultResolver, logger), nil\n\t}\n\treturn nil, fmt.Errorf(\"unknown bootstrap mode %q\", cfg.BootstrapMode)\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/factory_test.go",
    "content": "package ringpopprovider\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\tringpopproviderconfig \"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n)\n\ntype mockResolver struct {\n\tHosts map[string][]string\n\tSRV   map[string][]net.SRV\n}\n\nfunc (resolver *mockResolver) LookupHost(ctx context.Context, host string) ([]string, error) {\n\taddrs, ok := resolver.Hosts[host]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"Host was not resolved: %s\", host)\n\t}\n\treturn addrs, nil\n}\n\nfunc (resolver *mockResolver) LookupSRV(ctx context.Context, service string, proto string, name string) (string, []*net.SRV, error) {\n\tvar records []*net.SRV\n\tsrvs, ok := resolver.SRV[service]\n\tif !ok {\n\t\treturn \"\", nil, fmt.Errorf(\"Host was not resolved: %s\", service)\n\t}\n\n\tfor _, record := range srvs {\n\t\tsrvRecord := record\n\t\trecords = append(records, &srvRecord)\n\t}\n\n\treturn \"test\", records, nil\n}\n\nfunc TestDNSMode(t *testing.T) {\n\tvar cfg ringpopproviderconfig.Config\n\terr := yaml.Unmarshal([]byte(getDNSConfig()), &cfg)\n\tassert.Nil(t, err)\n\tassert.Equal(t, \"test\", cfg.Name)\n\tassert.Equal(t, ringpopproviderconfig.BootstrapModeDNS, cfg.BootstrapMode)\n\tassert.Equal(t, \"10.66.1.71\", cfg.BroadcastAddress)\n\tassert.Nil(t, cfg.Validate())\n\tlogger := testlogger.New(t)\n\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"example.net:1111\",\n\t\t\t\"example.net:1112\",\n\t\t\t\"unknown-duplicate.example.net:1111\",\n\t\t\t\"unknown-duplicate.example.net:1111\",\n\t\t\t\"badhostport\",\n\t\t},\n\t\tcfg.BootstrapHosts,\n\t)\n\n\tprovider := newDNSProvider(\n\t\tcfg.BootstrapHosts,\n\t\t&mockResolver{\n\t\t\tHosts: map[string][]string{\"example.net\": []string{\"10.0.0.0\", \"10.0.0.1\"}},\n\t\t},\n\t\tlogger,\n\t)\n\tcfg.DiscoveryProvider = provider\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"example.net:1111\",\n\t\t\t\"example.net:1112\",\n\t\t\t\"unknown-duplicate.example.net:1111\",\n\t\t\t\"badhostport\",\n\t\t},\n\t\tprovider.UnresolvedHosts,\n\t\t\"duplicate entries should be removed\",\n\t)\n\n\thostports, err := cfg.DiscoveryProvider.Hosts()\n\tassert.Nil(t, err)\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"10.0.0.0:1111\", \"10.0.0.1:1111\",\n\t\t\t\"10.0.0.0:1112\", \"10.0.0.1:1112\",\n\t\t},\n\t\thostports,\n\t)\n\n\tcfg.DiscoveryProvider = newDNSProvider(\n\t\tcfg.BootstrapHosts,\n\t\t&mockResolver{Hosts: map[string][]string{}},\n\t\tlogger,\n\t)\n\thostports, err = cfg.DiscoveryProvider.Hosts()\n\tassert.Nil(t, hostports)\n\tassert.NotNil(t, err, \"error should be returned when no hosts\")\n}\n\nfunc TestDNSSRVMode(t *testing.T) {\n\tvar cfg ringpopproviderconfig.Config\n\terr := yaml.Unmarshal([]byte(getDNSSRVConfig()), &cfg)\n\tassert.Nil(t, err)\n\tassert.Equal(t, \"test\", cfg.Name)\n\tassert.Equal(t, ringpopproviderconfig.BootstrapModeDNSSRV, cfg.BootstrapMode)\n\tassert.Nil(t, cfg.Validate())\n\tlogger := testlogger.New(t)\n\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"service-a.example.net\",\n\t\t\t\"service-b.example.net\",\n\t\t\t\"unknown-duplicate.example.net\",\n\t\t\t\"unknown-duplicate.example.net\",\n\t\t\t\"badhostport\",\n\t\t},\n\t\tcfg.BootstrapHosts,\n\t)\n\n\tprovider := newDNSSRVProvider(\n\t\tcfg.BootstrapHosts,\n\t\t&mockResolver{\n\t\t\tSRV: map[string][]net.SRV{\n\t\t\t\t\"service-a\": []net.SRV{{Target: \"az1-service-a.addr.example.net\", Port: 7755}, {Target: \"az2-service-a.addr.example.net\", Port: 7566}},\n\t\t\t\t\"service-b\": []net.SRV{{Target: \"az1-service-b.addr.example.net\", Port: 7788}, {Target: \"az2-service-b.addr.example.net\", Port: 7896}},\n\t\t\t},\n\t\t\tHosts: map[string][]string{\n\t\t\t\t\"az1-service-a.addr.example.net\": []string{\"10.0.0.1\"},\n\t\t\t\t\"az2-service-a.addr.example.net\": []string{\"10.0.2.0\", \"10.0.2.3\"},\n\t\t\t\t\"az1-service-b.addr.example.net\": []string{\"10.0.3.0\", \"10.0.3.3\"},\n\t\t\t\t\"az2-service-b.addr.example.net\": []string{\"10.0.3.1\"},\n\t\t\t},\n\t\t},\n\t\tlogger,\n\t)\n\tcfg.DiscoveryProvider = provider\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"service-a.example.net\",\n\t\t\t\"service-b.example.net\",\n\t\t\t\"unknown-duplicate.example.net\",\n\t\t\t\"badhostport\",\n\t\t},\n\t\tprovider.UnresolvedHosts,\n\t\t\"duplicate entries should be removed\",\n\t)\n\n\t// Expect unknown-duplicate.example.net to not resolve\n\t_, err = cfg.DiscoveryProvider.Hosts()\n\tassert.NotNil(t, err)\n\n\t// Remove known bad hosts from Unresolved list\n\tprovider.UnresolvedHosts = []string{\n\t\t\"service-a.example.net\",\n\t\t\"service-b.example.net\",\n\t\t\"badhostport\",\n\t}\n\n\t// Expect badhostport to not seperate service name\n\t_, err = cfg.DiscoveryProvider.Hosts()\n\tassert.NotNil(t, err)\n\n\t// Remove known bad hosts from Unresolved list\n\tprovider.UnresolvedHosts = []string{\n\t\t\"service-a.example.net\",\n\t\t\"service-b.example.net\",\n\t}\n\n\thostports, err := cfg.DiscoveryProvider.Hosts()\n\tassert.Nil(t, err)\n\tassert.ElementsMatch(\n\t\tt,\n\t\t[]string{\n\t\t\t\"10.0.0.1:7755\",\n\t\t\t\"10.0.2.0:7566\", \"10.0.2.3:7566\",\n\t\t\t\"10.0.3.0:7788\", \"10.0.3.3:7788\",\n\t\t\t\"10.0.3.1:7896\",\n\t\t},\n\t\thostports,\n\t\t\"duplicate entries should be removed\",\n\t)\n\n\tcfg.DiscoveryProvider = newDNSProvider(\n\t\tcfg.BootstrapHosts,\n\t\t&mockResolver{Hosts: map[string][]string{}},\n\t\tlogger,\n\t)\n\thostports, err = cfg.DiscoveryProvider.Hosts()\n\tassert.Nil(t, hostports)\n\tassert.NotNil(t, err, \"error should be returned when no hosts\")\n}\n\nfunc getDNSConfig() string {\n\treturn `name: \"test\"\nbootstrapMode: \"dns\"\nbroadcastAddress: \"10.66.1.71\"\nbootstrapHosts:\n- example.net:1111\n- example.net:1112\n- unknown-duplicate.example.net:1111\n- unknown-duplicate.example.net:1111\n- badhostport\nmaxJoinDuration: 30s`\n}\n\nfunc getDNSSRVConfig() string {\n\treturn `name: \"test\"\nbootstrapMode: \"dns-srv\"\nbootstrapHosts:\n- service-a.example.net\n- service-b.example.net\n- unknown-duplicate.example.net\n- unknown-duplicate.example.net\n- badhostport\nmaxJoinDuration: 30s`\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/provider.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ringpopprovider\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/ringpop-go\"\n\t\"github.com/uber/ringpop-go/events\"\n\trpmembership \"github.com/uber/ringpop-go/membership\"\n\t\"github.com/uber/ringpop-go/swim\"\n\ttcg \"github.com/uber/tchannel-go\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\tringpopproviderconfig \"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n)\n\ntype (\n\t// Provider use ringpop to announce membership changes\n\tProvider struct {\n\t\tstatus      int32\n\t\tservice     string\n\t\tringpop     *ringpop.Ringpop\n\t\tbootParams  *swim.BootstrapOptions\n\t\tlogger      log.Logger\n\t\tportmap     membership.PortMap\n\t\tmu          sync.RWMutex\n\t\tsubscribers map[string]func(membership.ChangedEvent)\n\t}\n)\n\nconst (\n\t// roleKey label is set by every single service as soon as it bootstraps its\n\t// ringpop instance. The data for this key is the service name\n\troleKey = \"serviceName\"\n)\n\nvar _ membership.PeerProvider = (*Provider)(nil)\n\nfunc New(\n\tservice string,\n\tconfig *ringpopproviderconfig.Config,\n\tchannel tchannel.Channel,\n\tportMap membership.PortMap,\n\tlogger log.Logger,\n) (*Provider, error) {\n\tif err := config.Validate(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdiscoveryProvider, err := newDiscoveryProvider(config, logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"ringpop discovery provider: %w\", err)\n\t}\n\n\tbootstrapOpts := &swim.BootstrapOptions{\n\t\tMaxJoinDuration:  config.MaxJoinDuration,\n\t\tDiscoverProvider: discoveryProvider,\n\t}\n\n\tch := channel.(*tcg.Channel)\n\topts := []ringpop.Option{ringpop.Channel(ch)}\n\tif config.BroadcastAddress != \"\" {\n\t\tbroadcastIP := net.ParseIP(config.BroadcastAddress)\n\t\tif broadcastIP == nil {\n\t\t\treturn nil, fmt.Errorf(\"failed parsing broadcast address %q, err: %w\", config.BroadcastAddress, err)\n\t\t}\n\t\tlogger.Info(\"Initializing ringpop with custom broadcast address\", tag.Address(broadcastIP.String()))\n\t\topts = append(opts, ringpop.AddressResolverFunc(broadcastAddrResolver(ch, broadcastIP)))\n\t}\n\trp, err := ringpop.New(config.Name, opts...)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"ringpop instance creation: %w\", err)\n\t}\n\n\treturn NewRingpopProvider(service, rp, portMap, bootstrapOpts, logger), nil\n}\n\n// NewRingpopProvider sets up ringpop based peer provider\nfunc NewRingpopProvider(\n\tservice string,\n\trp *ringpop.Ringpop,\n\tportMap membership.PortMap,\n\tbootstrapOpts *swim.BootstrapOptions,\n\tlogger log.Logger,\n) *Provider {\n\treturn &Provider{\n\t\tservice:     service,\n\t\tstatus:      common.DaemonStatusInitialized,\n\t\tbootParams:  bootstrapOpts,\n\t\tlogger:      logger,\n\t\tportmap:     portMap,\n\t\tringpop:     rp,\n\t\tsubscribers: map[string]func(membership.ChangedEvent){},\n\t}\n}\n\n// Start starts ringpop\nfunc (r *Provider) Start() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&r.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\t_, err := r.ringpop.Bootstrap(r.bootParams)\n\tif err != nil {\n\t\tr.logger.Fatal(\"unable to bootstrap ringpop\", tag.Error(err))\n\t}\n\n\t// Get updates from ringpop ring\n\tr.ringpop.AddListener(r)\n\n\tlabels, err := r.ringpop.Labels()\n\tif err != nil {\n\t\tr.logger.Fatal(\"unable to get ring pop labels\", tag.Error(err))\n\t}\n\n\t// set port labels\n\tfor name, port := range r.portmap {\n\t\tif err = labels.Set(name, strconv.Itoa(int(port))); err != nil {\n\t\t\tr.logger.Fatal(\"unable to set port label\", tag.Error(err))\n\t\t}\n\t}\n\n\tif err = labels.Set(roleKey, r.service); err != nil {\n\t\tr.logger.Fatal(\"unable to set ringpop role label\", tag.Error(err))\n\t}\n}\n\n// HandleEvent handles updates from ringpop\nfunc (r *Provider) HandleEvent(event events.Event) {\n\t// We only care about RingChangedEvent\n\te, ok := event.(events.RingChangedEvent)\n\tif !ok {\n\t\treturn\n\t}\n\n\tchange := membership.ChangedEvent{\n\t\tHostsAdded:   e.ServersAdded,\n\t\tHostsUpdated: e.ServersUpdated,\n\t\tHostsRemoved: e.ServersRemoved,\n\t}\n\tr.logger.Info(\"Received a ringpop ring changed event\", tag.MembershipChangeEvent(change))\n\tr.notifySubscribers(change)\n}\n\nfunc (r *Provider) notifySubscribers(event membership.ChangedEvent) {\n\tr.mu.RLock()\n\tdefer r.mu.RUnlock()\n\n\tfor _, handler := range r.subscribers {\n\t\thandler(event)\n\t}\n}\n\nfunc (r *Provider) SelfEvict() error {\n\treturn r.ringpop.SelfEvict()\n}\n\n// GetMembers returns all hosts with a specified role value\nfunc (r *Provider) GetMembers(service string) ([]membership.HostInfo, error) {\n\tvar res []membership.HostInfo\n\n\t// filter member by service name, add port info to Hostinfo if they are present\n\tmemberData := func(member swim.Member) bool {\n\t\tif v, ok := member.Label(roleKey); !ok || v != service {\n\t\t\treturn false\n\t\t}\n\n\t\t// replicating swim member isReachable() method here as we are skipping other predicates\n\t\tif member.Status != swim.Alive && member.Status != swim.Suspect {\n\t\t\treturn false\n\t\t}\n\n\t\tportMap := make(membership.PortMap)\n\t\tif v, ok := member.Label(membership.PortTchannel); ok {\n\t\t\tport, err := labelToPort(v)\n\t\t\tif err != nil {\n\t\t\t\tr.logger.Warn(\"tchannel port cannot be converted\", tag.Error(err), tag.Value(v))\n\t\t\t} else {\n\t\t\t\tportMap[membership.PortTchannel] = port\n\t\t\t}\n\t\t} else {\n\t\t\t// for backwards compatibility: get tchannel port from member address\n\t\t\t_, port, err := net.SplitHostPort(member.Address)\n\t\t\tif err != nil {\n\t\t\t\tr.logger.Warn(\"getting ringpop member port from address\", tag.Value(member.Address), tag.Error(err))\n\t\t\t} else {\n\t\t\t\ttchannelPort, err := labelToPort(port)\n\t\t\t\tif err != nil {\n\t\t\t\t\tr.logger.Warn(\"tchannel port cannot be converted\", tag.Error(err), tag.Value(port))\n\t\t\t\t} else {\n\t\t\t\t\tportMap[membership.PortTchannel] = tchannelPort\n\t\t\t\t}\n\t\t\t}\n\n\t\t}\n\n\t\tif v, ok := member.Label(membership.PortGRPC); ok {\n\t\t\tport, err := labelToPort(v)\n\t\t\tif err != nil {\n\t\t\t\tr.logger.Warn(\"grpc port cannot be converted\", tag.Error(err), tag.Value(v))\n\t\t\t} else {\n\t\t\t\tportMap[membership.PortGRPC] = port\n\t\t\t}\n\t\t}\n\n\t\tres = append(res, membership.NewDetailedHostInfo(member.GetAddress(), member.Identity(), portMap))\n\n\t\treturn true\n\t}\n\t_, err := r.ringpop.GetReachableMembers(memberData)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"ringpop get members: %w\", err)\n\t}\n\n\treturn res, nil\n}\n\n// WhoAmI returns address of this instance\nfunc (r *Provider) WhoAmI() (membership.HostInfo, error) {\n\taddress, err := r.ringpop.WhoAmI()\n\tif err != nil {\n\t\treturn membership.HostInfo{}, fmt.Errorf(\"ringpop doesn't know Who Am I: %w\", err)\n\t}\n\n\tlabels, err := r.ringpop.Labels()\n\tif err != nil {\n\t\treturn membership.HostInfo{}, fmt.Errorf(\"getting ringpop labels: %w\", err)\n\t}\n\n\thostIdentity := address\n\t// this is needed to in a situation when Cadence is trying to identify the owner for a key\n\t// make sure we are comparing identities, but not host:port pairs\n\trpIdentity, set := labels.Get(rpmembership.IdentityLabelKey)\n\tif set {\n\t\thostIdentity = rpIdentity\n\t}\n\n\treturn membership.NewDetailedHostInfo(address, hostIdentity, r.portmap), nil\n}\n\n// Stop stops ringpop\nfunc (r *Provider) Stop() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&r.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\tr.ringpop.Destroy()\n}\n\n// Subscribe allows to be subscribed for ring changes\nfunc (r *Provider) Subscribe(name string, handler func(membership.ChangedEvent)) error {\n\tr.mu.Lock()\n\tdefer r.mu.Unlock()\n\n\t_, ok := r.subscribers[name]\n\tif ok {\n\t\treturn fmt.Errorf(\"%q already subscribed to ringpop provider\", name)\n\t}\n\n\tr.subscribers[name] = handler\n\treturn nil\n}\n\nfunc labelToPort(label string) (uint16, error) {\n\tport, err := strconv.ParseUint(label, 10, 16)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn uint16(port), nil\n}\n\nfunc broadcastAddrResolver(ch *tcg.Channel, broadcastIP net.IP) func() (string, error) {\n\treturn func() (string, error) {\n\t\tpeerInfo := ch.PeerInfo()\n\t\tif peerInfo.IsEphemeralHostPort() {\n\t\t\t// not listening yet\n\t\t\treturn \"\", ringpop.ErrEphemeralAddress\n\t\t}\n\n\t\t_, port, err := net.SplitHostPort(peerInfo.HostPort)\n\t\tif err != nil {\n\t\t\treturn \"\", fmt.Errorf(\"failed splitting tchannel's hostport %q, err: %w\", peerInfo.HostPort, err)\n\t\t}\n\n\t\t// return broadcast_ip:listen_port\n\t\treturn net.JoinHostPort(broadcastIP.String(), port), nil\n\t}\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/provider_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ringpopprovider\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber/ringpop-go/events\"\n\t\"github.com/uber/tchannel-go\"\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\tringpopproviderconfig \"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n)\n\nconst testServiceName = \"test-service\"\n\ntype srvAndCh struct {\n\tservice  string\n\tch       *tchannel.Channel\n\tprovider *Provider\n}\n\nfunc TestRingpopProvider(t *testing.T) {\n\tt.Cleanup(func() {\n\t\tgoleak.VerifyNone(t,\n\t\t\t// ignore the goroutines leaked by ringpop library\n\t\t\tgoleak.IgnoreTopFunction(\"github.com/rcrowley/go-metrics.(*meterArbiter).tick\"),\n\t\t\tgoleak.IgnoreTopFunction(\"github.com/uber/ringpop-go.(*Ringpop).startTimers.func1\"),\n\t\t\tgoleak.IgnoreTopFunction(\"github.com/uber/ringpop-go.(*Ringpop).startTimers.func2\"))\n\t})\n\tlogger := testlogger.New(t)\n\tserviceName := \"matching\"\n\n\tmatchingChs, cleanupMatchingChs, err := createAndListenChannels(serviceName, 3)\n\tt.Cleanup(cleanupMatchingChs)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to create and listen on channels: %v\", err)\n\t}\n\n\tirrelevantChs, cleanupIrrelevantChs, err := createAndListenChannels(\"random-svc\", 1)\n\tt.Cleanup(cleanupIrrelevantChs)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to create and listen on channels: %v\", err)\n\t}\n\n\tallServicesAndChs := append(matchingChs, irrelevantChs...)\n\tcfg := ringpopproviderconfig.Config{\n\t\tBootstrapMode:   ringpopproviderconfig.BootstrapModeHosts,\n\t\tName:            \"ring\",\n\t\tBootstrapHosts:  toHosts(t, allServicesAndChs),\n\t\tMaxJoinDuration: 10 * time.Second,\n\t}\n\n\tt.Logf(\"Config: %+v\", cfg)\n\n\t// start ringpop provider for each channel\n\tvar wg sync.WaitGroup\n\tfor i, svc := range allServicesAndChs {\n\t\tsvc := svc\n\t\tcfg := cfg\n\t\tif i == 0 {\n\t\t\t// set broadcast address for the first provider to test that path\n\t\t\tcfg.BroadcastAddress = \"127.0.0.1\"\n\t\t}\n\t\tp, err := New(svc.service, &cfg, svc.ch, membership.PortMap{}, logger)\n\t\tif err != nil {\n\t\t\tt.Fatalf(\"Failed to create ringpop provider: %v\", err)\n\t\t}\n\n\t\tsvc.provider = p\n\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tp.Start()\n\t\t}()\n\t\tt.Cleanup(p.Stop)\n\t}\n\n\tt.Logf(\"Waiting for %d ringpop providers to start\", len(matchingChs)+len(irrelevantChs))\n\twg.Wait()\n\n\tsleep := 5 * time.Second\n\tt.Logf(\"Sleeping for %d seconds for ring to update\", int(sleep.Seconds()))\n\ttime.Sleep(sleep)\n\n\tprovider := matchingChs[0].provider\n\thostInfo, err := provider.WhoAmI()\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to get who am I: %v\", err)\n\t}\n\n\tt.Logf(\"Who am I: %+v\", hostInfo.GetAddress())\n\n\tmembers, err := provider.GetMembers(serviceName)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to get members: %v\", err)\n\t}\n\n\tif len(members) != 3 {\n\t\tt.Fatalf(\"Expected 3 members, got %v\", len(members))\n\t}\n\n\t// Evict one of the providers and make sure it's removed from the ring\n\tmatchingChs[1].provider.SelfEvict()\n\tt.Logf(\"A peer is evicted. Sleeping for %d seconds for ring to update\", int(sleep.Seconds()))\n\ttime.Sleep(sleep)\n\n\tmembers, err = provider.GetMembers(serviceName)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to get members: %v\", err)\n\t}\n\n\tif len(members) != 2 {\n\t\tt.Fatalf(\"Expected 2 members, got %v\", len(members))\n\t}\n}\n\nfunc TestSubscribeAndNotify(t *testing.T) {\n\tprovider := NewRingpopProvider(testServiceName, nil, nil, nil, testlogger.New(t))\n\n\tringpopEvent := events.RingChangedEvent{\n\t\tServersAdded:   []string{\"aa\", \"bb\", \"cc\"},\n\t\tServersUpdated: []string{\"dd\"},\n\t\tServersRemoved: []string{\"ee\", \"ff\"},\n\t}\n\texpectedEvent := membership.ChangedEvent{\n\t\tHostsAdded:   ringpopEvent.ServersAdded,\n\t\tHostsUpdated: ringpopEvent.ServersUpdated,\n\t\tHostsRemoved: ringpopEvent.ServersRemoved,\n\t}\n\n\tvar calls1, calls2 int\n\trequire.NoError(t,\n\t\tprovider.Subscribe(\"subscriber1\",\n\t\t\tfunc(ev membership.ChangedEvent) {\n\t\t\t\tcalls1++\n\t\t\t\tassert.Equal(t, ev, expectedEvent)\n\t\t\t},\n\t\t))\n\n\trequire.NoError(t,\n\t\tprovider.Subscribe(\"subscriber2\",\n\t\t\tfunc(ev membership.ChangedEvent) {\n\t\t\t\tcalls2++\n\t\t\t\tassert.Equal(t, ev, expectedEvent)\n\t\t\t},\n\t\t))\n\n\trequire.Error(t,\n\t\tprovider.Subscribe(\n\t\t\t\"subscriber2\",\n\t\t\tfunc(membership.ChangedEvent) { t.Error(\"Should never be called\") },\n\t\t),\n\t\t\"Subscribe doesn't allow duplicate names\",\n\t)\n\n\tprovider.HandleEvent(ringpopEvent)\n\tassert.Equal(t, 1, calls1, \"every subscriber must have been called once\")\n\tassert.Equal(t, 1, calls2, \"every subscriber must have been called once\")\n}\n\nfunc createAndListenChannels(serviceName string, n int) ([]*srvAndCh, func(), error) {\n\tvar res []*srvAndCh\n\tcleanupFn := func(srvs []*srvAndCh) func() {\n\t\treturn func() {\n\t\t\tfor _, s := range srvs {\n\t\t\t\ts.ch.Close()\n\t\t\t}\n\t\t}\n\t}\n\tfor i := 0; i < n; i++ {\n\t\tch, err := tchannel.NewChannel(serviceName, nil)\n\t\tif err != nil {\n\t\t\treturn nil, cleanupFn(res), err\n\t\t}\n\n\t\tif err := ch.ListenAndServe(\"localhost:0\"); err != nil {\n\t\t\treturn nil, cleanupFn(res), err\n\t\t}\n\n\t\tres = append(res, &srvAndCh{service: serviceName, ch: ch})\n\t}\n\treturn res, cleanupFn(res), nil\n}\n\nfunc toHosts(t *testing.T, allServicesAndChs []*srvAndCh) []string {\n\tvar hosts []string\n\tfor _, svc := range allServicesAndChs {\n\t\tif svc.ch.PeerInfo().IsEphemeralHostPort() {\n\t\t\tt.Fatalf(\"Channel %v is not listening on a port\", svc.ch)\n\t\t}\n\t\thosts = append(hosts, svc.ch.PeerInfo().HostPort)\n\t}\n\treturn hosts\n}\n\nfunc TestLabelToPort(t *testing.T) {\n\ttests := []struct {\n\t\tlabel   string\n\t\twant    uint16\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tlabel: \"0\",\n\t\t\twant:  0,\n\t\t},\n\t\t{\n\t\t\tlabel: \"1234\",\n\t\t\twant:  1234,\n\t\t},\n\t\t{\n\t\t\tlabel:   \"-1\",\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tlabel: \"32768\",\n\t\t\twant:  32768,\n\t\t},\n\t\t{\n\t\t\tlabel: \"65535\",\n\t\t\twant:  65535,\n\t\t},\n\t\t{\n\t\t\tlabel:   \"65536\", // greater than max uint16 (2^16-1)\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tgot, err := labelToPort(tc.label)\n\t\tif got != tc.want || (err != nil) != tc.wantErr {\n\t\t\tt.Errorf(\"labelToPort(%v) = %v, %v; want %v, %v\", tc.label, got, err, tc.want, tc.wantErr)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/ringpopfx/ringpopfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ringpopfx\n\nimport (\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/peerprovider/ringpopprovider\"\n\t\"github.com/uber/cadence/common/rpc\"\n)\n\n// Module provides a peer resolver based on ringpop for fx app.\nvar Module = fx.Module(\"ringpop\", fx.Provide(buildRingpopProvider))\n\ntype Params struct {\n\tfx.In\n\n\tServiceFullName string `name:\"service-full-name\"`\n\tConfig          config.Config\n\tServiceConfig   config.Service\n\tLogger          log.Logger\n\tRPCFactory      rpc.Factory\n\tLifecycle       fx.Lifecycle\n}\n\nfunc buildRingpopProvider(params Params) (membership.PeerProvider, error) {\n\tprovider, err := ringpopprovider.New(params.ServiceFullName, &params.Config.Ringpop, params.RPCFactory.GetTChannel(), membership.PortMap{\n\t\tmembership.PortGRPC:     params.ServiceConfig.RPC.GRPCPort,\n\t\tmembership.PortTchannel: params.ServiceConfig.RPC.Port,\n\t}, params.Logger)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn provider, nil\n}\n"
  },
  {
    "path": "common/peerprovider/ringpopprovider/ringpopfx/ringpopfx_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ringpopfx\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber/tchannel-go\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\tringpopproviderconfig \"github.com/uber/cadence/common/peerprovider/ringpopprovider/config\"\n\t\"github.com/uber/cadence/common/rpc\"\n)\n\nfunc TestFxApp(t *testing.T) {\n\tapp := fxtest.New(t,\n\t\tfx.Provide(\n\t\t\tfunc() testSetupParams {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tfactory := rpc.NewMockFactory(ctrl)\n\t\t\t\ttch, err := tchannel.NewChannel(\"test-ringpop\", nil)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tfactory.EXPECT().GetTChannel().Return(tch)\n\n\t\t\t\treturn testSetupParams{\n\t\t\t\t\tService:    \"test\",\n\t\t\t\t\tLogger:     testlogger.New(t),\n\t\t\t\t\tRPCFactory: factory,\n\t\t\t\t\tConfig: config.Config{\n\t\t\t\t\t\tRingpop: ringpopproviderconfig.Config{\n\t\t\t\t\t\t\tName:           \"test-ringpop\",\n\t\t\t\t\t\t\tBootstrapMode:  ringpopproviderconfig.BootstrapModeHosts,\n\t\t\t\t\t\t\tBootstrapHosts: []string{\"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t}),\n\t\tModule, fx.Invoke(func(provider membership.PeerProvider) {}),\n\t)\n\tapp.RequireStart().RequireStop()\n}\n\ntype testSetupParams struct {\n\tfx.Out\n\n\tService       string `name:\"service-full-name\"`\n\tConfig        config.Config\n\tServiceConfig config.Service\n\tLogger        log.Logger\n\tRPCFactory    rpc.Factory\n}\n"
  },
  {
    "path": "common/persistence/client/bean.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination bean_mock.go\n\npackage client\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\ntype (\n\t// Bean in an collection of persistence manager\n\tBean interface {\n\t\tClose()\n\n\t\tGetDomainManager() persistence.DomainManager\n\t\tSetDomainManager(persistence.DomainManager)\n\n\t\tGetDomainAuditManager() persistence.DomainAuditManager\n\t\tSetDomainAuditManager(persistence.DomainAuditManager)\n\n\t\tGetTaskManager() persistence.TaskManager\n\t\tSetTaskManager(persistence.TaskManager)\n\n\t\tGetVisibilityManager() persistence.VisibilityManager\n\t\tSetVisibilityManager(persistence.VisibilityManager)\n\n\t\tGetDomainReplicationQueueManager() persistence.QueueManager\n\t\tSetDomainReplicationQueueManager(persistence.QueueManager)\n\n\t\tGetShardManager() persistence.ShardManager\n\t\tSetShardManager(persistence.ShardManager)\n\n\t\tGetHistoryManager() persistence.HistoryManager\n\t\tSetHistoryManager(persistence.HistoryManager)\n\n\t\tGetExecutionManager(int) (persistence.ExecutionManager, error)\n\t\tSetExecutionManager(int, persistence.ExecutionManager)\n\n\t\tGetConfigStoreManager() persistence.ConfigStoreManager\n\t\tSetConfigStoreManager(persistence.ConfigStoreManager)\n\t}\n\n\t// BeanImpl stores persistence managers\n\tBeanImpl struct {\n\t\tdomainManager                 persistence.DomainManager\n\t\tdomainAuditManager            persistence.DomainAuditManager\n\t\ttaskManager                   persistence.TaskManager\n\t\tvisibilityManager             persistence.VisibilityManager\n\t\tdomainReplicationQueueManager persistence.QueueManager\n\t\tshardManager                  persistence.ShardManager\n\t\thistoryManager                persistence.HistoryManager\n\t\tconfigStoreManager            persistence.ConfigStoreManager\n\t\texecutionManagerFactory       persistence.ExecutionManagerFactory\n\n\t\tsync.RWMutex\n\t\tshardIDToExecutionManager map[int]persistence.ExecutionManager\n\t}\n\n\t// Params contains dependencies for persistence\n\tParams struct {\n\t\tPersistenceConfig config.Persistence\n\t\tMetricsClient     metrics.Client\n\t\tMessagingClient   messaging.Client\n\t\tESClient          es.GenericClient\n\t\tESConfig          *config.ElasticSearchConfig\n\t\tPinotConfig       *config.PinotVisibilityConfig\n\t\tPinotClient       pinot.GenericClient\n\t\tOSClient          es.GenericClient\n\t\tOSConfig          *config.ElasticSearchConfig\n\t}\n)\n\n// NewBeanFromFactory crate a new store bean using factory\nfunc NewBeanFromFactory(\n\tfactory Factory,\n\tparams *Params,\n\tserviceConfig *service.Config,\n) (Bean, error) {\n\n\tmetadataMgr, err := factory.NewDomainManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainAuditMgr, err := factory.NewDomainAuditManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttaskMgr, err := factory.NewTaskManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvisibilityMgr, err := factory.NewVisibilityManager(params, serviceConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainReplicationQueue, err := factory.NewDomainReplicationQueueManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tshardMgr, err := factory.NewShardManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\thistoryMgr, err := factory.NewHistoryManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfigStoreMgr, err := factory.NewConfigStoreManager()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn NewBean(\n\t\tmetadataMgr,\n\t\tdomainAuditMgr,\n\t\ttaskMgr,\n\t\tvisibilityMgr,\n\t\tdomainReplicationQueue,\n\t\tshardMgr,\n\t\thistoryMgr,\n\t\tconfigStoreMgr,\n\t\tfactory,\n\t), nil\n}\n\n// NewBean create a new store bean\nfunc NewBean(\n\tdomainManager persistence.DomainManager,\n\tdomainAuditManager persistence.DomainAuditManager,\n\ttaskManager persistence.TaskManager,\n\tvisibilityManager persistence.VisibilityManager,\n\tdomainReplicationQueueManager persistence.QueueManager,\n\tshardManager persistence.ShardManager,\n\thistoryManager persistence.HistoryManager,\n\tconfigStoreManager persistence.ConfigStoreManager,\n\texecutionManagerFactory persistence.ExecutionManagerFactory,\n) *BeanImpl {\n\treturn &BeanImpl{\n\t\tdomainManager:                 domainManager,\n\t\tdomainAuditManager:            domainAuditManager,\n\t\ttaskManager:                   taskManager,\n\t\tvisibilityManager:             visibilityManager,\n\t\tdomainReplicationQueueManager: domainReplicationQueueManager,\n\t\tshardManager:                  shardManager,\n\t\thistoryManager:                historyManager,\n\t\tconfigStoreManager:            configStoreManager,\n\t\texecutionManagerFactory:       executionManagerFactory,\n\n\t\tshardIDToExecutionManager: make(map[int]persistence.ExecutionManager),\n\t}\n}\n\n// GetDomainManager get DomainManager\nfunc (s *BeanImpl) GetDomainManager() persistence.DomainManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.domainManager\n}\n\n// SetMetadataManager set DomainManager\nfunc (s *BeanImpl) SetDomainManager(\n\tdomainManager persistence.DomainManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.domainManager = domainManager\n}\n\n// GetDomainAuditManager get DomainAuditManager\nfunc (s *BeanImpl) GetDomainAuditManager() persistence.DomainAuditManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.domainAuditManager\n}\n\n// SetDomainAuditManager set DomainAuditManager\nfunc (s *BeanImpl) SetDomainAuditManager(\n\tdomainAuditManager persistence.DomainAuditManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.domainAuditManager = domainAuditManager\n}\n\n// GetTaskManager get TaskManager\nfunc (s *BeanImpl) GetTaskManager() persistence.TaskManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.taskManager\n}\n\n// SetTaskManager set TaskManager\nfunc (s *BeanImpl) SetTaskManager(\n\ttaskManager persistence.TaskManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.taskManager = taskManager\n}\n\n// GetVisibilityManager get VisibilityManager\nfunc (s *BeanImpl) GetVisibilityManager() persistence.VisibilityManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.visibilityManager\n}\n\n// SetVisibilityManager set VisibilityManager\nfunc (s *BeanImpl) SetVisibilityManager(\n\tvisibilityManager persistence.VisibilityManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.visibilityManager = visibilityManager\n}\n\n// GetDomainReplicationQueueManager gets domain replication QueueManager\nfunc (s *BeanImpl) GetDomainReplicationQueueManager() persistence.QueueManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.domainReplicationQueueManager\n}\n\n// SetDomainReplicationQueueManager sets domain replication QueueManager\nfunc (s *BeanImpl) SetDomainReplicationQueueManager(\n\tdomainReplicationQueueManager persistence.QueueManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.domainReplicationQueueManager = domainReplicationQueueManager\n}\n\n// GetShardManager get ShardManager\nfunc (s *BeanImpl) GetShardManager() persistence.ShardManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.shardManager\n}\n\n// SetShardManager set ShardManager\nfunc (s *BeanImpl) SetShardManager(\n\tshardManager persistence.ShardManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.shardManager = shardManager\n}\n\n// GetHistoryManager get HistoryManager\nfunc (s *BeanImpl) GetHistoryManager() persistence.HistoryManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.historyManager\n}\n\n// SetHistoryManager set HistoryManager\nfunc (s *BeanImpl) SetHistoryManager(\n\thistoryManager persistence.HistoryManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.historyManager = historyManager\n}\n\n// GetExecutionManager get ExecutionManager\nfunc (s *BeanImpl) GetExecutionManager(\n\tshardID int,\n) (persistence.ExecutionManager, error) {\n\n\ts.RLock()\n\texecutionManager, ok := s.shardIDToExecutionManager[shardID]\n\tif ok {\n\t\ts.RUnlock()\n\t\treturn executionManager, nil\n\t}\n\ts.RUnlock()\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\texecutionManager, ok = s.shardIDToExecutionManager[shardID]\n\tif ok {\n\t\treturn executionManager, nil\n\t}\n\n\texecutionManager, err := s.executionManagerFactory.NewExecutionManager(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ts.shardIDToExecutionManager[shardID] = executionManager\n\treturn executionManager, nil\n}\n\n// SetExecutionManager set ExecutionManager\nfunc (s *BeanImpl) SetExecutionManager(\n\tshardID int,\n\texecutionManager persistence.ExecutionManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.shardIDToExecutionManager[shardID] = executionManager\n}\n\n// GetConfigStoreManager gets ConfigStoreManager\nfunc (s *BeanImpl) GetConfigStoreManager() persistence.ConfigStoreManager {\n\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.configStoreManager\n}\n\n// GetConfigStoreManager gets ConfigStoreManager\nfunc (s *BeanImpl) SetConfigStoreManager(\n\tconfigStoreManager persistence.ConfigStoreManager,\n) {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.configStoreManager = configStoreManager\n}\n\n// Close cleanup connections\nfunc (s *BeanImpl) Close() {\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.domainManager.Close()\n\tif s.domainAuditManager != nil {\n\t\ts.domainAuditManager.Close()\n\t}\n\ts.taskManager.Close()\n\tif s.visibilityManager != nil {\n\t\t// visibilityManager can be nil\n\t\ts.visibilityManager.Close()\n\t}\n\ts.domainReplicationQueueManager.Close()\n\ts.shardManager.Close()\n\ts.historyManager.Close()\n\ts.executionManagerFactory.Close()\n\ts.configStoreManager.Close()\n\tfor _, executionMgr := range s.shardIDToExecutionManager {\n\t\texecutionMgr.Close()\n\t}\n}\n"
  },
  {
    "path": "common/persistence/client/bean_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: bean.go\n//\n// Generated by this command:\n//\n//\tmockgen -package client -source bean.go -destination bean_mock.go\n//\n\n// Package client is a generated GoMock package.\npackage client\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockBean is a mock of Bean interface.\ntype MockBean struct {\n\tctrl     *gomock.Controller\n\trecorder *MockBeanMockRecorder\n\tisgomock struct{}\n}\n\n// MockBeanMockRecorder is the mock recorder for MockBean.\ntype MockBeanMockRecorder struct {\n\tmock *MockBean\n}\n\n// NewMockBean creates a new mock instance.\nfunc NewMockBean(ctrl *gomock.Controller) *MockBean {\n\tmock := &MockBean{ctrl: ctrl}\n\tmock.recorder = &MockBeanMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockBean) EXPECT() *MockBeanMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockBean) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockBeanMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockBean)(nil).Close))\n}\n\n// GetConfigStoreManager mocks base method.\nfunc (m *MockBean) GetConfigStoreManager() persistence.ConfigStoreManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetConfigStoreManager\")\n\tret0, _ := ret[0].(persistence.ConfigStoreManager)\n\treturn ret0\n}\n\n// GetConfigStoreManager indicates an expected call of GetConfigStoreManager.\nfunc (mr *MockBeanMockRecorder) GetConfigStoreManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetConfigStoreManager\", reflect.TypeOf((*MockBean)(nil).GetConfigStoreManager))\n}\n\n// GetDomainAuditManager mocks base method.\nfunc (m *MockBean) GetDomainAuditManager() persistence.DomainAuditManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAuditManager\")\n\tret0, _ := ret[0].(persistence.DomainAuditManager)\n\treturn ret0\n}\n\n// GetDomainAuditManager indicates an expected call of GetDomainAuditManager.\nfunc (mr *MockBeanMockRecorder) GetDomainAuditManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAuditManager\", reflect.TypeOf((*MockBean)(nil).GetDomainAuditManager))\n}\n\n// GetDomainManager mocks base method.\nfunc (m *MockBean) GetDomainManager() persistence.DomainManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainManager\")\n\tret0, _ := ret[0].(persistence.DomainManager)\n\treturn ret0\n}\n\n// GetDomainManager indicates an expected call of GetDomainManager.\nfunc (mr *MockBeanMockRecorder) GetDomainManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainManager\", reflect.TypeOf((*MockBean)(nil).GetDomainManager))\n}\n\n// GetDomainReplicationQueueManager mocks base method.\nfunc (m *MockBean) GetDomainReplicationQueueManager() persistence.QueueManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainReplicationQueueManager\")\n\tret0, _ := ret[0].(persistence.QueueManager)\n\treturn ret0\n}\n\n// GetDomainReplicationQueueManager indicates an expected call of GetDomainReplicationQueueManager.\nfunc (mr *MockBeanMockRecorder) GetDomainReplicationQueueManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainReplicationQueueManager\", reflect.TypeOf((*MockBean)(nil).GetDomainReplicationQueueManager))\n}\n\n// GetExecutionManager mocks base method.\nfunc (m *MockBean) GetExecutionManager(arg0 int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionManager\", arg0)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetExecutionManager indicates an expected call of GetExecutionManager.\nfunc (mr *MockBeanMockRecorder) GetExecutionManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionManager\", reflect.TypeOf((*MockBean)(nil).GetExecutionManager), arg0)\n}\n\n// GetHistoryManager mocks base method.\nfunc (m *MockBean) GetHistoryManager() persistence.HistoryManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryManager\")\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\treturn ret0\n}\n\n// GetHistoryManager indicates an expected call of GetHistoryManager.\nfunc (mr *MockBeanMockRecorder) GetHistoryManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryManager\", reflect.TypeOf((*MockBean)(nil).GetHistoryManager))\n}\n\n// GetShardManager mocks base method.\nfunc (m *MockBean) GetShardManager() persistence.ShardManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardManager\")\n\tret0, _ := ret[0].(persistence.ShardManager)\n\treturn ret0\n}\n\n// GetShardManager indicates an expected call of GetShardManager.\nfunc (mr *MockBeanMockRecorder) GetShardManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardManager\", reflect.TypeOf((*MockBean)(nil).GetShardManager))\n}\n\n// GetTaskManager mocks base method.\nfunc (m *MockBean) GetTaskManager() persistence.TaskManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskManager\")\n\tret0, _ := ret[0].(persistence.TaskManager)\n\treturn ret0\n}\n\n// GetTaskManager indicates an expected call of GetTaskManager.\nfunc (mr *MockBeanMockRecorder) GetTaskManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskManager\", reflect.TypeOf((*MockBean)(nil).GetTaskManager))\n}\n\n// GetVisibilityManager mocks base method.\nfunc (m *MockBean) GetVisibilityManager() persistence.VisibilityManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityManager\")\n\tret0, _ := ret[0].(persistence.VisibilityManager)\n\treturn ret0\n}\n\n// GetVisibilityManager indicates an expected call of GetVisibilityManager.\nfunc (mr *MockBeanMockRecorder) GetVisibilityManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityManager\", reflect.TypeOf((*MockBean)(nil).GetVisibilityManager))\n}\n\n// SetConfigStoreManager mocks base method.\nfunc (m *MockBean) SetConfigStoreManager(arg0 persistence.ConfigStoreManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetConfigStoreManager\", arg0)\n}\n\n// SetConfigStoreManager indicates an expected call of SetConfigStoreManager.\nfunc (mr *MockBeanMockRecorder) SetConfigStoreManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetConfigStoreManager\", reflect.TypeOf((*MockBean)(nil).SetConfigStoreManager), arg0)\n}\n\n// SetDomainAuditManager mocks base method.\nfunc (m *MockBean) SetDomainAuditManager(arg0 persistence.DomainAuditManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetDomainAuditManager\", arg0)\n}\n\n// SetDomainAuditManager indicates an expected call of SetDomainAuditManager.\nfunc (mr *MockBeanMockRecorder) SetDomainAuditManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetDomainAuditManager\", reflect.TypeOf((*MockBean)(nil).SetDomainAuditManager), arg0)\n}\n\n// SetDomainManager mocks base method.\nfunc (m *MockBean) SetDomainManager(arg0 persistence.DomainManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetDomainManager\", arg0)\n}\n\n// SetDomainManager indicates an expected call of SetDomainManager.\nfunc (mr *MockBeanMockRecorder) SetDomainManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetDomainManager\", reflect.TypeOf((*MockBean)(nil).SetDomainManager), arg0)\n}\n\n// SetDomainReplicationQueueManager mocks base method.\nfunc (m *MockBean) SetDomainReplicationQueueManager(arg0 persistence.QueueManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetDomainReplicationQueueManager\", arg0)\n}\n\n// SetDomainReplicationQueueManager indicates an expected call of SetDomainReplicationQueueManager.\nfunc (mr *MockBeanMockRecorder) SetDomainReplicationQueueManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetDomainReplicationQueueManager\", reflect.TypeOf((*MockBean)(nil).SetDomainReplicationQueueManager), arg0)\n}\n\n// SetExecutionManager mocks base method.\nfunc (m *MockBean) SetExecutionManager(arg0 int, arg1 persistence.ExecutionManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetExecutionManager\", arg0, arg1)\n}\n\n// SetExecutionManager indicates an expected call of SetExecutionManager.\nfunc (mr *MockBeanMockRecorder) SetExecutionManager(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetExecutionManager\", reflect.TypeOf((*MockBean)(nil).SetExecutionManager), arg0, arg1)\n}\n\n// SetHistoryManager mocks base method.\nfunc (m *MockBean) SetHistoryManager(arg0 persistence.HistoryManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetHistoryManager\", arg0)\n}\n\n// SetHistoryManager indicates an expected call of SetHistoryManager.\nfunc (mr *MockBeanMockRecorder) SetHistoryManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetHistoryManager\", reflect.TypeOf((*MockBean)(nil).SetHistoryManager), arg0)\n}\n\n// SetShardManager mocks base method.\nfunc (m *MockBean) SetShardManager(arg0 persistence.ShardManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetShardManager\", arg0)\n}\n\n// SetShardManager indicates an expected call of SetShardManager.\nfunc (mr *MockBeanMockRecorder) SetShardManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetShardManager\", reflect.TypeOf((*MockBean)(nil).SetShardManager), arg0)\n}\n\n// SetTaskManager mocks base method.\nfunc (m *MockBean) SetTaskManager(arg0 persistence.TaskManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetTaskManager\", arg0)\n}\n\n// SetTaskManager indicates an expected call of SetTaskManager.\nfunc (mr *MockBeanMockRecorder) SetTaskManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetTaskManager\", reflect.TypeOf((*MockBean)(nil).SetTaskManager), arg0)\n}\n\n// SetVisibilityManager mocks base method.\nfunc (m *MockBean) SetVisibilityManager(arg0 persistence.VisibilityManager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVisibilityManager\", arg0)\n}\n\n// SetVisibilityManager indicates an expected call of SetVisibilityManager.\nfunc (mr *MockBeanMockRecorder) SetVisibilityManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVisibilityManager\", reflect.TypeOf((*MockBean)(nil).SetVisibilityManager), arg0)\n}\n"
  },
  {
    "path": "common/persistence/client/bean_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage client\n\nimport (\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype beanmocks struct {\n\tmockCtrl           *gomock.Controller\n\tdomainManager      *persistence.MockDomainManager\n\tdomainAuditManager *persistence.MockDomainAuditManager\n\ttaskManager        *persistence.MockTaskManager\n\tvisibilityManager  *persistence.MockVisibilityManager\n\treplicationManager *persistence.MockQueueManager\n\tshardManager       *persistence.MockShardManager\n\thistoryManager     *persistence.MockHistoryManager\n\tconfigManager      *persistence.MockConfigStoreManager\n}\n\nfunc beanSetup(t *testing.T) (f *MockFactory, m beanmocks, defaultMocks func()) {\n\tctrl := gomock.NewController(t)\n\tm = beanmocks{\n\t\tmockCtrl:           ctrl,\n\t\tdomainManager:      persistence.NewMockDomainManager(ctrl),\n\t\tdomainAuditManager: persistence.NewMockDomainAuditManager(ctrl),\n\t\ttaskManager:        persistence.NewMockTaskManager(ctrl),\n\t\tvisibilityManager:  persistence.NewMockVisibilityManager(ctrl),\n\t\treplicationManager: persistence.NewMockQueueManager(ctrl),\n\t\tshardManager:       persistence.NewMockShardManager(ctrl),\n\t\thistoryManager:     persistence.NewMockHistoryManager(ctrl),\n\t\tconfigManager:      persistence.NewMockConfigStoreManager(ctrl),\n\t}\n\tf = NewMockFactory(ctrl)\n\tdefaultMocks = func() {\n\t\t// allow any of them to be called once or never, individual tests will set earlier mocks as needed\n\t\tf.EXPECT().NewDomainManager().Return(m.domainManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewDomainAuditManager().Return(m.domainAuditManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewTaskManager().Return(m.taskManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewVisibilityManager(gomock.Any(), gomock.Any()).Return(m.visibilityManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewDomainReplicationQueueManager().Return(m.replicationManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewShardManager().Return(m.shardManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewHistoryManager().Return(m.historyManager, nil).MaxTimes(1)\n\t\tf.EXPECT().NewConfigStoreManager().Return(m.configManager, nil).MaxTimes(1)\n\t}\n\treturn f, m, defaultMocks\n}\n\nfunc TestBeanCoverage(t *testing.T) {\n\t// not particularly valuable but needed to hit min-% goals\n\tt.Run(\"NewBeanFromFactory\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\t// coverage for the New func, very straightforward err-branches.\n\t\ttests := map[string]struct {\n\t\t\tmockSetup func(t *testing.T, f *MockFactory)\n\t\t\terr       string\n\t\t}{\n\t\t\t\"success\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\t// intentionally empty - setup already assume success\n\t\t\t\t},\n\t\t\t\terr: \"\",\n\t\t\t},\n\t\t\t\"domain manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewDomainManager().Return(nil, fmt.Errorf(\"no domain manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no domain manager\",\n\t\t\t},\n\t\t\t\"task manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewTaskManager().Return(nil, fmt.Errorf(\"no task manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no task manager\",\n\t\t\t},\n\t\t\t\"visibility manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewVisibilityManager(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"no visibility manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no visibility manager\",\n\t\t\t},\n\t\t\t\"domain replication queue manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewDomainReplicationQueueManager().Return(nil, fmt.Errorf(\"no domain replication queue manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no domain replication queue manager\",\n\t\t\t},\n\t\t\t\"shard manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewShardManager().Return(nil, fmt.Errorf(\"no shard manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no shard manager\",\n\t\t\t},\n\t\t\t\"history manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewHistoryManager().Return(nil, fmt.Errorf(\"no history manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no history manager\",\n\t\t\t},\n\t\t\t\"config manager error\": {\n\t\t\t\tmockSetup: func(t *testing.T, f *MockFactory) {\n\t\t\t\t\tf.EXPECT().NewConfigStoreManager().Return(nil, fmt.Errorf(\"no config manager\"))\n\t\t\t\t},\n\t\t\t\terr: \"no config manager\",\n\t\t\t},\n\t\t}\n\t\tfor name, test := range tests {\n\t\t\tname, test := name, test\n\t\t\tt.Run(name, func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tf, _, defaultMocks := beanSetup(t)\n\t\t\t\ttest.mockSetup(t, f)\n\t\t\t\tdefaultMocks()\n\t\t\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\t\t\tif test.err != \"\" {\n\t\t\t\t\tassert.ErrorContains(t, err, test.err)\n\t\t\t\t} else {\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.NotNil(t, impl)\n\t\t\t\t}\n\t\t\t})\n\t\t}\n\t})\n\tt.Run(\"Simple getters\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tf, m, defaultMocks := beanSetup(t)\n\t\tdefaultMocks() // build all mock objects\n\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\trequire.NoError(t, err)\n\n\t\t// these need to be concurrency-safe, so run them concurrently\n\t\tvar g errgroup.Group\n\t\tg.Go(errgroupAssertEqual(t, m.domainManager, impl.GetDomainManager))\n\t\tg.Go(errgroupAssertEqual(t, m.domainAuditManager, impl.GetDomainAuditManager))\n\t\tg.Go(errgroupAssertEqual(t, m.taskManager, impl.GetTaskManager))\n\t\tg.Go(errgroupAssertEqual(t, m.visibilityManager, impl.GetVisibilityManager))\n\t\tg.Go(errgroupAssertEqual(t, m.replicationManager, impl.GetDomainReplicationQueueManager))\n\t\tg.Go(errgroupAssertEqual(t, m.shardManager, impl.GetShardManager))\n\t\tg.Go(errgroupAssertEqual(t, m.historyManager, impl.GetHistoryManager))\n\t\tg.Go(errgroupAssertEqual(t, m.configManager, impl.GetConfigStoreManager))\n\t\trequire.NoError(t, g.Wait())\n\t\t// execution managers are per shard, checked separately\n\t})\n\tt.Run(\"Simple setters\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tf, _, defaultMocks := beanSetup(t)\n\t\t_, m2, _ := beanSetup(t) // make a second set of mock objects, so they can be compared\n\t\tdefaultMocks()           // allow constructors to be called\n\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\trequire.NoError(t, err)\n\n\t\t// these need to be concurrency-safe, so run them concurrently\n\t\tvar g errgroup.Group\n\n\t\tg.Go(errgroupAssertSets(t, m2.domainManager, impl.SetDomainManager, impl.GetDomainManager))\n\t\tg.Go(errgroupAssertSets(t, m2.domainAuditManager, impl.SetDomainAuditManager, impl.GetDomainAuditManager))\n\t\tg.Go(errgroupAssertSets(t, m2.taskManager, impl.SetTaskManager, impl.GetTaskManager))\n\t\tg.Go(errgroupAssertSets(t, m2.visibilityManager, impl.SetVisibilityManager, impl.GetVisibilityManager))\n\t\tg.Go(errgroupAssertSets(t, m2.replicationManager, impl.SetDomainReplicationQueueManager, impl.GetDomainReplicationQueueManager))\n\t\tg.Go(errgroupAssertSets(t, m2.shardManager, impl.SetShardManager, impl.GetShardManager))\n\t\tg.Go(errgroupAssertSets(t, m2.historyManager, impl.SetHistoryManager, impl.GetHistoryManager))\n\t\tg.Go(errgroupAssertSets(t, m2.configManager, impl.SetConfigStoreManager, impl.GetConfigStoreManager))\n\t\trequire.NoError(t, g.Wait())\n\t\t// execution managers are per shard, checked separately\n\t})\n\tt.Run(\"Execution manager getter\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tf, m, defaultMocks := beanSetup(t)\n\n\t\t// execution managers are per shard, make sure they're set up correctly\n\t\tex1, ex2 := persistence.NewMockExecutionManager(m.mockCtrl), persistence.NewMockExecutionManager(m.mockCtrl)\n\t\tf.EXPECT().NewExecutionManager(1).Return(ex1, nil).Times(1)\n\t\tf.EXPECT().NewExecutionManager(2).Return(ex2, nil).Times(1)\n\n\t\t// build all other mock objects so New works.\n\t\t// these will not duplicate ^ above instances, so if they are used they'll fail assert.Equal checks.\n\t\tdefaultMocks()\n\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\trequire.NoError(t, err)\n\n\t\t// must be concurrency safe, so run them concurrently\n\t\tvar g errgroup.Group\n\t\tg.Go(errgroupAssertExecutionManagerEqual(t, 1, ex1, impl))\n\t\tg.Go(errgroupAssertExecutionManagerEqual(t, 2, ex2, impl))\n\n\t\t// make sure re-getting doesn't make a duplicate / a different instance.\n\t\t// the `.Times(1)` also ensures this does not make an extra construction.\n\t\tg.Go(errgroupAssertExecutionManagerEqual(t, 1, ex1, impl))\n\t\tg.Go(errgroupAssertExecutionManagerEqual(t, 2, ex2, impl))\n\t\trequire.NoError(t, g.Wait())\n\t})\n\tt.Run(\"Execution manager setter\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tf, m, defaultMocks := beanSetup(t)\n\n\t\t// seed with some pre-existing data so we can say \"not same as before\".\n\t\tf.EXPECT().NewExecutionManager(1).Return(persistence.NewMockExecutionManager(m.mockCtrl), nil).Times(1)\n\t\tf.EXPECT().NewExecutionManager(2).Return(persistence.NewMockExecutionManager(m.mockCtrl), nil).Times(1)\n\t\tdefaultMocks()\n\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\trequire.NoError(t, err)\n\t\t_, err = impl.GetExecutionManager(1)\n\t\trequire.NoError(t, err, \"setup sanity check failed\")\n\t\t_, err = impl.GetExecutionManager(2)\n\t\trequire.NoError(t, err, \"setup sanity check failed\")\n\n\t\t// make the execution managers that will be set\n\t\tex1, ex2 := persistence.NewMockExecutionManager(m.mockCtrl), persistence.NewMockExecutionManager(m.mockCtrl)\n\n\t\t// must be concurrency safe, so run them concurrently\n\t\tvar g errgroup.Group\n\t\tg.Go(errgroupAssertSetsExecutionManager(t, 1, ex1, impl))\n\t\tg.Go(errgroupAssertSetsExecutionManager(t, 2, ex2, impl))\n\t\trequire.NoError(t, g.Wait())\n\t})\n\tt.Run(\"Lifecycle\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tf, m, defaultMocks := beanSetup(t)\n\n\t\t// make some execution managers so they can be closed\n\t\tex1, ex2 := persistence.NewMockExecutionManager(m.mockCtrl), persistence.NewMockExecutionManager(m.mockCtrl)\n\t\tf.EXPECT().NewExecutionManager(1).Return(ex1, nil).Times(1)\n\t\tf.EXPECT().NewExecutionManager(2).Return(ex2, nil).Times(1)\n\n\t\t// expect everything to close\n\t\tm.domainManager.EXPECT().Close().Return().Times(1)\n\t\tm.domainAuditManager.EXPECT().Close().Return().Times(1)\n\t\tm.taskManager.EXPECT().Close().Return().Times(1)\n\t\tm.visibilityManager.EXPECT().Close().Return().Times(1)\n\t\tm.replicationManager.EXPECT().Close().Return().Times(1)\n\t\tm.shardManager.EXPECT().Close().Return().Times(1)\n\t\tm.historyManager.EXPECT().Close().Return().Times(1)\n\t\tm.configManager.EXPECT().Close().Return().Times(1)\n\t\tex1.EXPECT().Close().Return().Times(1)\n\t\tex2.EXPECT().Close().Return().Times(1)\n\t\t// which includes the execution-manager-factory itself\n\t\tf.EXPECT().Close().Return().Times(1)\n\n\t\tdefaultMocks() // for New\n\t\timpl, err := NewBeanFromFactory(f, nil, nil)\n\t\trequire.NoError(t, err)\n\n\t\t// seed the execution managers (could call Set instead)\n\t\tv, err := impl.GetExecutionManager(1)\n\t\trequire.NoError(t, err)\n\t\trequire.NotNil(t, v)\n\t\tv, err = impl.GetExecutionManager(2)\n\t\trequire.NoError(t, err)\n\t\trequire.NotNil(t, v)\n\n\t\t// ensure everything is closed\n\t\timpl.Close()\n\t})\n}\n\n// generics complains that the interface and mock-impl types don't match,\n// so this is done via reflection.  testify ensures the values and types are the same.\n//\n// `actualFn` needs to be a `func() T` which returns the expected value/type.\nfunc errgroupAssertEqual(t *testing.T, expected, getFn any) func() error {\n\tt.Helper()\n\treturn func() error {\n\t\tt.Helper()\n\t\tassertMocksEqual(t, expected, reflect.ValueOf(getFn).Call(nil)[0].Interface())\n\t\treturn nil\n\t}\n}\n\n// similar to above, not generics-friendly.\n// setFn must be `func(T)` and getFn must be `func() T`, or the calls will panic.\nfunc errgroupAssertSets(t *testing.T, expected, setFn, getFn any) func() error {\n\tt.Helper()\n\treturn func() error {\n\t\tt.Helper()\n\t\t// sanity check first\n\t\tif !assertMocksNotEqual(t, expected, reflect.ValueOf(getFn).Call(nil)[0].Interface()) {\n\t\t\treturn nil\n\t\t}\n\n\t\t// perform the SetX call.\n\t\t// panics if incorrect.\n\t\treflect.ValueOf(setFn).Call([]reflect.Value{reflect.ValueOf(expected)})\n\n\t\t// make sure the getter gets the set value\n\t\tassertMocksEqual(t, expected, reflect.ValueOf(getFn).Call(nil)[0].Interface())\n\t\treturn nil\n\t}\n}\n\nfunc errgroupAssertExecutionManagerEqual(t *testing.T, arg int, expected persistence.ExecutionManager, impl Bean) func() error {\n\tt.Helper()\n\treturn func() error {\n\t\tt.Helper()\n\t\tval, err := impl.GetExecutionManager(arg)\n\t\tassert.NoError(t, err) // cannot use require in other goroutines\n\t\tassertMocksEqual(t, expected, val)\n\t\treturn err\n\t}\n}\n\nfunc errgroupAssertSetsExecutionManager(t *testing.T, arg int, expected persistence.ExecutionManager, impl Bean) func() error {\n\tt.Helper()\n\treturn func() error {\n\t\tt.Helper()\n\n\t\t// sanity check first\n\t\tval, err := impl.GetExecutionManager(arg)\n\t\tif !assert.NoError(t, err, \"could not do initial Get to check values\") {\n\t\t\treturn nil\n\t\t}\n\t\tif !assertMocksNotEqual(t, expected, val) {\n\t\t\treturn nil\n\t\t}\n\n\t\timpl.SetExecutionManager(arg, expected)\n\n\t\tval, err = impl.GetExecutionManager(arg)\n\t\tassert.NoError(t, err) // cannot use require in other goroutines\n\t\tassertMocksEqual(t, expected, val)\n\t\treturn err\n\t}\n}\n\n// mock-equality helper because `assert.Equal` considers all mocks of the same type to be equal,\n// as they (generally) use the same mock controller and recorder, and it ignores different pointers\n// due to using reflect.DeepEqual.\nfunc assertMocksEqual(t *testing.T, expected, actual any) bool {\n\tt.Helper()\n\tif actual != expected {\n\t\tt.Errorf(\"does not contain the expected reference, got: %T(%p) expected: %T(%p)\", actual, actual, expected, expected)\n\t\treturn false\n\t}\n\treturn true\n}\n\n// mock-not-equality helper because `assert.NotEqual` considers all mocks of the same type to be equal,\n// as they (generally) use the same mock controller and recorder, and it ignores different pointers\n// due to using reflect.DeepEqual.\nfunc assertMocksNotEqual(t *testing.T, expected, actual any) bool {\n\tt.Helper()\n\tif actual == expected {\n\t\tt.Errorf(\"actual and expected values (pointers) are identical\")\n\t\treturn false\n\t}\n\treturn true\n}\n"
  },
  {
    "path": "common/persistence/client/factory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination factory_mock.go\n\npackage client\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/elasticsearch\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\tpinotVisibility \"github.com/uber/cadence/common/persistence/pinot\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/errorinjectors\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/metered\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/ratelimited\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/sampled\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\ntype (\n\t// Factory defines the interface for any implementation that can vend\n\t// persistence layer objects backed by a datastore. The actual datastore\n\t// is implementation detail hidden behind this interface\n\tFactory interface {\n\t\t// Close the factory\n\t\tClose()\n\t\t// NewTaskManager returns a new task manager\n\t\tNewTaskManager() (p.TaskManager, error)\n\t\t// NewShardManager returns a new shard manager\n\t\tNewShardManager() (p.ShardManager, error)\n\t\t// NewHistoryManager returns a new history manager\n\t\tNewHistoryManager() (p.HistoryManager, error)\n\t\t// NewDomainManager returns a new metadata manager\n\t\tNewDomainManager() (p.DomainManager, error)\n\t\t// NewDomainAuditManager returns a new domain audit manager\n\t\tNewDomainAuditManager() (p.DomainAuditManager, error)\n\t\t// NewExecutionManager returns a new execution manager for a given shardID\n\t\tNewExecutionManager(shardID int) (p.ExecutionManager, error)\n\t\t// NewVisibilityManager returns a new visibility manager\n\t\tNewVisibilityManager(params *Params, serviceConfig *service.Config) (p.VisibilityManager, error)\n\t\t// NewDomainReplicationQueueManager returns a new queue for domain replication\n\t\tNewDomainReplicationQueueManager() (p.QueueManager, error)\n\t\t// NewConfigStoreManager returns a new config store manager\n\t\tNewConfigStoreManager() (p.ConfigStoreManager, error)\n\t}\n\t// DataStoreFactory is a low level interface to be implemented by a datastore\n\t// Examples of datastores are cassandra, mysql etc\n\tDataStoreFactory interface {\n\t\t// Close closes the factory\n\t\tClose()\n\t\t// NewTaskStore returns a new task store\n\t\tNewTaskStore() (p.TaskStore, error)\n\t\t// NewShardStore returns a new shard store\n\t\tNewShardStore() (p.ShardStore, error)\n\t\t// NewHistoryStore returns a new history store\n\t\tNewHistoryStore() (p.HistoryStore, error)\n\t\t// NewDomainStore returns a new metadata store\n\t\tNewDomainStore() (p.DomainStore, error)\n\t\t// NewDomainAuditStore returns a new domain audit store\n\t\tNewDomainAuditStore() (p.DomainAuditStore, error)\n\t\t// NewExecutionStore returns an execution store for given shardID\n\t\tNewExecutionStore(shardID int) (p.ExecutionStore, error)\n\t\t// NewVisibilityStore returns a new visibility store,\n\t\t// TODO We temporarily using sortByCloseTime to determine whether or not ListClosedWorkflowExecutions should\n\t\t// be ordering by CloseTime. This will be removed when implementing https://github.com/uber/cadence/issues/3621\n\t\tNewVisibilityStore(sortByCloseTime bool) (p.VisibilityStore, error)\n\t\tNewQueue(queueType p.QueueType) (p.QueueStore, error)\n\t\t// NewConfigStore returns a new config store\n\t\tNewConfigStore() (p.ConfigStore, error)\n\t}\n\n\t// Datastore represents a datastore\n\tDatastore struct {\n\t\tfactory   DataStoreFactory\n\t\tratelimit quotas.Limiter\n\t}\n\tfactoryImpl struct {\n\t\tsync.RWMutex\n\t\tconfig        *config.Persistence\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t\tdatastores    map[storeType]Datastore\n\t\tclusterName   string\n\t\tdc            *p.DynamicConfiguration\n\t}\n\n\tstoreType int\n)\n\nconst (\n\tstoreTypeHistory storeType = iota + 1\n\tstoreTypeTask\n\tstoreTypeShard\n\tstoreTypeMetadata\n\tstoreTypeExecution\n\tstoreTypeVisibility\n\tstoreTypeQueue\n\tstoreTypeConfigStore\n)\n\nvar storeTypes = []storeType{\n\tstoreTypeHistory,\n\tstoreTypeTask,\n\tstoreTypeShard,\n\tstoreTypeMetadata,\n\tstoreTypeExecution,\n\tstoreTypeVisibility,\n\tstoreTypeQueue,\n\tstoreTypeConfigStore,\n}\n\n// NewFactory returns an implementation of factory that vends persistence objects based on\n// specified configuration. This factory takes as input a config.Persistence object\n// which specifies the datastore to be used for a given type of object. This config\n// also contains config for individual datastores themselves.\n//\n// The objects returned by this factory enforce ratelimit and maxconns according to\n// given configuration. In addition, all objects will emit metrics automatically\nfunc NewFactory(\n\tcfg *config.Persistence,\n\tpersistenceMaxQPS quotas.RPSFunc,\n\tclusterName string,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tdc *p.DynamicConfiguration,\n) Factory {\n\tfactory := &factoryImpl{\n\t\tconfig:        cfg,\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t\tclusterName:   clusterName,\n\t\tdc:            dc,\n\t}\n\tlimiters := buildRatelimiters(cfg, persistenceMaxQPS)\n\tfactory.init(clusterName, limiters)\n\treturn factory\n}\n\n// NewTaskManager returns a new task manager\nfunc (f *factoryImpl) NewTaskManager() (p.TaskManager, error) {\n\tds := f.datastores[storeTypeTask]\n\tstore, err := ds.factory.NewTaskStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewTaskManager(store)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewTaskManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewTaskManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewTaskManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\treturn result, nil\n}\n\n// NewShardManager returns a new shard manager\nfunc (f *factoryImpl) NewShardManager() (p.ShardManager, error) {\n\tds := f.datastores[storeTypeShard]\n\tstore, err := ds.factory.NewShardStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewShardManager(store, f.dc)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewShardManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewShardManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewShardManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\treturn result, nil\n}\n\n// NewHistoryManager returns a new history manager\nfunc (f *factoryImpl) NewHistoryManager() (p.HistoryManager, error) {\n\tds := f.datastores[storeTypeHistory]\n\tstore, err := ds.factory.NewHistoryStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewHistoryV2ManagerImpl(store, f.logger, p.NewPayloadSerializer(), codec.NewThriftRWEncoder(), f.config.TransactionSizeLimit)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewHistoryManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewHistoryManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewHistoryManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\treturn result, nil\n}\n\n// NewDomainManager returns a new metadata manager\nfunc (f *factoryImpl) NewDomainManager() (p.DomainManager, error) {\n\tvar err error\n\tvar store p.DomainStore\n\tds := f.datastores[storeTypeMetadata]\n\tstore, err = ds.factory.NewDomainStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewDomainManagerImpl(store, f.logger, p.NewPayloadSerializer(), f.dc)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewDomainManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewDomainManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewDomainManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\treturn result, nil\n}\n\n// NewDomainAuditManager returns a new domain audit manager\nfunc (f *factoryImpl) NewDomainAuditManager() (p.DomainAuditManager, error) {\n\tvar err error\n\tvar store p.DomainAuditStore\n\n\tds := f.datastores[storeTypeMetadata]\n\tstore, err = ds.factory.NewDomainAuditStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif store == nil {\n\t\treturn nil, nil\n\t}\n\tresult := p.NewDomainAuditManagerImpl(store, f.logger, p.NewPayloadSerializer(), f.dc)\n\treturn result, nil\n}\n\n// NewExecutionManager returns a new execution manager for a given shardID\nfunc (f *factoryImpl) NewExecutionManager(shardID int) (p.ExecutionManager, error) {\n\tds := f.datastores[storeTypeExecution]\n\tstore, err := ds.factory.NewExecutionStore(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewExecutionManagerImpl(store, f.logger, p.NewPayloadSerializer(), f.dc)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewExecutionManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewExecutionManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewExecutionManager(result, f.metricsClient, f.logger, f.config, f.dc.PersistenceSampleLoggingRate, f.dc.EnableShardIDMetrics)\n\t}\n\treturn result, nil\n}\n\n// NewVisibilityManager returns a new visibility manager\nfunc (f *factoryImpl) NewVisibilityManager(\n\tparams *Params,\n\tresourceConfig *service.Config,\n) (p.VisibilityManager, error) {\n\tif resourceConfig.ReadVisibilityStoreName == nil && resourceConfig.WriteVisibilityStoreName == nil {\n\t\t// No need to create visibility manager as no read/write needed\n\t\treturn nil, nil\n\t}\n\tvar visibilityFromDB, visibilityFromES, visibilityFromPinot, visibilityFromOS p.VisibilityManager\n\tvar err error\n\tif params.PersistenceConfig.VisibilityStore != \"\" {\n\t\tvisibilityFromDB, err = f.newDBVisibilityManager(resourceConfig)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tswitch params.PersistenceConfig.AdvancedVisibilityStore {\n\tcase constants.PinotVisibilityStoreName:\n\t\tvisibilityFromPinot, err = setupPinotVisibilityManager(params, resourceConfig, f.logger, f.dc, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t\tif err != nil {\n\t\t\tf.logger.Fatal(\"Creating Pinot advanced visibility manager failed\", tag.Error(err))\n\t\t}\n\n\t\tvisibilityMgrs := map[string]p.VisibilityManager{\n\t\t\tconstants.VisibilityModeDB:    visibilityFromDB,\n\t\t\tconstants.VisibilityModePinot: visibilityFromPinot,\n\t\t}\n\n\t\tif params.PinotConfig.Migration.Enabled {\n\t\t\tvisibilityFromES, err = setupESVisibilityManager(params, resourceConfig, f.logger, f.dc, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t\t\tif err != nil {\n\t\t\t\tf.logger.Fatal(\"Creating ES advanced visibility manager failed\", tag.Error(err))\n\t\t\t}\n\n\t\t\tvisibilityMgrs[constants.VisibilityModeES] = visibilityFromES\n\t\t}\n\n\t\treturn p.NewVisibilityHybridManager(\n\t\t\tvisibilityMgrs,\n\t\t\tresourceConfig.ReadVisibilityStoreName,\n\t\t\tresourceConfig.WriteVisibilityStoreName,\n\t\t\tresourceConfig.EnableLogCustomerQueryParameter,\n\t\t\tconstants.PinotPersistenceName,\n\t\t\tf.logger,\n\t\t), nil\n\tcase constants.OSVisibilityStoreName:\n\t\tvisibilityFromOS, err = setupOSVisibilityManager(params, resourceConfig, f.logger, f.dc, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t\tif err != nil {\n\t\t\tf.logger.Fatal(\"Creating OS advanced visibility manager failed\", tag.Error(err))\n\t\t}\n\n\t\tvisibilityMgrs := map[string]p.VisibilityManager{\n\t\t\tconstants.VisibilityModeDB: visibilityFromDB,\n\t\t\tconstants.VisibilityModeOS: visibilityFromOS,\n\t\t}\n\t\tif params.OSConfig.Migration.Enabled {\n\t\t\tvisibilityFromES, err = setupESVisibilityManager(params, resourceConfig, f.logger, f.dc, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t\t\tif err != nil {\n\t\t\t\tf.logger.Fatal(\"Creating ES advanced visibility manager failed\", tag.Error(err))\n\t\t\t}\n\n\t\t\tvisibilityMgrs[constants.VisibilityModeES] = visibilityFromES\n\t\t}\n\t\treturn p.NewVisibilityHybridManager(\n\t\t\tvisibilityMgrs,\n\t\t\tresourceConfig.ReadVisibilityStoreName,\n\t\t\tresourceConfig.WriteVisibilityStoreName,\n\t\t\tresourceConfig.EnableLogCustomerQueryParameter,\n\t\t\tconstants.ESPersistenceName,\n\t\t\tf.logger,\n\t\t), nil\n\tcase constants.ESVisibilityStoreName:\n\t\tvisibilityFromES, err = setupESVisibilityManager(params, resourceConfig, f.logger, f.dc, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t\tif err != nil {\n\t\t\tf.logger.Fatal(\"Creating advanced visibility manager failed\", tag.Error(err))\n\t\t}\n\t\tvisibilityMgrs := map[string]p.VisibilityManager{\n\t\t\tconstants.VisibilityModeDB: visibilityFromDB,\n\t\t\tconstants.VisibilityModeES: visibilityFromES,\n\t\t}\n\t\treturn p.NewVisibilityHybridManager(\n\t\t\tvisibilityMgrs,\n\t\t\tresourceConfig.ReadVisibilityStoreName,\n\t\t\tresourceConfig.WriteVisibilityStoreName,\n\t\t\tresourceConfig.EnableLogCustomerQueryParameter,\n\t\t\tconstants.ESPersistenceName,\n\t\t\tf.logger,\n\t\t), nil\n\tdefault:\n\t\tvisibilityMgrs := map[string]p.VisibilityManager{\n\t\t\tconstants.VisibilityModeDB: visibilityFromDB,\n\t\t}\n\t\tif visibilityFromDB != nil {\n\t\t\treturn p.NewVisibilityHybridManager(\n\t\t\t\tvisibilityMgrs,\n\t\t\t\tresourceConfig.ReadVisibilityStoreName,\n\t\t\t\tresourceConfig.WriteVisibilityStoreName,\n\t\t\t\tresourceConfig.EnableLogCustomerQueryParameter,\n\t\t\t\tvisibilityFromDB.GetName(), // db has multiple different stores\n\t\t\t\tf.logger,\n\t\t\t), nil\n\t\t}\n\t\treturn nil, nil // no visibility manager available for write\n\t}\n}\n\n// NewESVisibilityManager create a visibility manager for ElasticSearch\n// In history, it only needs kafka producer for writing data;\n// In frontend, it only needs ES client and related config for reading data\nfunc newPinotVisibilityManager(\n\tpinotClient pnt.GenericClient,\n\tvisibilityConfig *service.Config,\n\tproducer messaging.Producer,\n\tmetricsClient metrics.Client,\n\tlog log.Logger,\n\tdc *p.DynamicConfiguration,\n\tcallerBypass quotas.CallerBypass,\n) p.VisibilityManager {\n\tvisibilityFromPinotStore := pinotVisibility.NewPinotVisibilityStore(pinotClient, visibilityConfig, producer, log)\n\tvisibilityFromPinot := p.NewVisibilityManagerImpl(visibilityFromPinotStore, log, dc)\n\n\t// wrap with rate limiter\n\tif visibilityConfig.PersistenceMaxQPS != nil && visibilityConfig.PersistenceMaxQPS() != 0 {\n\t\tpinotRateLimiter := quotas.NewDynamicRateLimiter(visibilityConfig.PersistenceMaxQPS.AsFloat64())\n\t\tvisibilityFromPinot = ratelimited.NewVisibilityManager(visibilityFromPinot, pinotRateLimiter, callerBypass)\n\t}\n\n\tif metricsClient != nil {\n\t\t// wrap with metrics\n\t\tvisibilityFromPinot = pinotVisibility.NewPinotVisibilityMetricsClient(visibilityFromPinot, metricsClient, log)\n\t}\n\n\treturn visibilityFromPinot\n}\n\n// NewESVisibilityManager create a visibility manager for ElasticSearch\n// In history, it only needs kafka producer for writing data;\n// In frontend, it only needs ES client and related config for reading data\nfunc newESVisibilityManager(\n\tindexName string,\n\tesClient es.GenericClient,\n\tvisibilityConfig *service.Config,\n\tproducer messaging.Producer,\n\tmetricsClient metrics.Client,\n\tlog log.Logger,\n\tdc *p.DynamicConfiguration,\n\tcallerBypass quotas.CallerBypass,\n) p.VisibilityManager {\n\n\tvisibilityFromESStore := elasticsearch.NewElasticSearchVisibilityStore(esClient, indexName, producer, visibilityConfig, log)\n\tvisibilityFromES := p.NewVisibilityManagerImpl(visibilityFromESStore, log, dc)\n\n\t// wrap with rate limiter\n\tif visibilityConfig.PersistenceMaxQPS != nil && visibilityConfig.PersistenceMaxQPS() != 0 {\n\t\tesRateLimiter := quotas.NewDynamicRateLimiter(visibilityConfig.PersistenceMaxQPS.AsFloat64())\n\t\tvisibilityFromES = ratelimited.NewVisibilityManager(visibilityFromES, esRateLimiter, callerBypass)\n\t}\n\tif metricsClient != nil {\n\t\t// wrap with metrics\n\t\tvisibilityFromES = elasticsearch.NewVisibilityMetricsClient(visibilityFromES, metricsClient, log)\n\t}\n\n\treturn visibilityFromES\n}\n\nfunc (f *factoryImpl) newDBVisibilityManager(\n\tvisibilityConfig *service.Config,\n) (p.VisibilityManager, error) {\n\tenableReadFromClosedExecutionV2 := false\n\tif visibilityConfig.EnableReadDBVisibilityFromClosedExecutionV2 != nil {\n\t\tenableReadFromClosedExecutionV2 = visibilityConfig.EnableReadDBVisibilityFromClosedExecutionV2()\n\t} else {\n\t\tf.logger.Warn(\"missing visibility and EnableReadFromClosedExecutionV2 config\", tag.Value(visibilityConfig))\n\t}\n\n\tds := f.datastores[storeTypeVisibility]\n\tstore, err := ds.factory.NewVisibilityStore(enableReadFromClosedExecutionV2)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewVisibilityManagerImpl(store, f.logger, f.dc)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewVisibilityManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewVisibilityManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif visibilityConfig.EnableDBVisibilitySampling != nil && visibilityConfig.EnableDBVisibilitySampling() {\n\t\tresult = sampled.NewVisibilityManager(result, sampled.Params{\n\t\t\tConfig: &sampled.Config{\n\t\t\t\tVisibilityClosedMaxQPS: visibilityConfig.WriteDBVisibilityClosedMaxQPS,\n\t\t\t\tVisibilityListMaxQPS:   visibilityConfig.DBVisibilityListMaxQPS,\n\t\t\t\tVisibilityOpenMaxQPS:   visibilityConfig.WriteDBVisibilityOpenMaxQPS,\n\t\t\t},\n\t\t\tMetricClient:           f.metricsClient,\n\t\t\tLogger:                 f.logger,\n\t\t\tTimeSource:             clock.NewRealTimeSource(),\n\t\t\tRateLimiterFactoryFunc: sampled.NewDomainToBucketMap,\n\t\t})\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewVisibilityManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\n\treturn result, nil\n}\n\nfunc (f *factoryImpl) NewDomainReplicationQueueManager() (p.QueueManager, error) {\n\tds := f.datastores[storeTypeQueue]\n\tstore, err := ds.factory.NewQueue(p.DomainReplicationQueueType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewQueueManager(store)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewQueueManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewQueueManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewQueueManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\n\treturn result, nil\n}\n\nfunc (f *factoryImpl) NewConfigStoreManager() (p.ConfigStoreManager, error) {\n\tds := f.datastores[storeTypeConfigStore]\n\tstore, err := ds.factory.NewConfigStore()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := p.NewConfigStoreManagerImpl(store, f.logger)\n\tif errorRate := f.config.ErrorInjectionRate(); errorRate != 0 {\n\t\tresult = errorinjectors.NewConfigStoreManager(result, errorRate, f.logger, time.Now())\n\t}\n\tif ds.ratelimit != nil {\n\t\tresult = ratelimited.NewConfigStoreManager(result, ds.ratelimit, quotas.NewCallerBypass(f.dc.RateLimiterBypassCallerTypes))\n\t}\n\tif f.metricsClient != nil {\n\t\tresult = metered.NewConfigStoreManager(result, f.metricsClient, f.logger, f.config)\n\t}\n\n\treturn result, nil\n}\n\n// Close closes this factory\nfunc (f *factoryImpl) Close() {\n\tds := f.datastores[storeTypeExecution]\n\tds.factory.Close()\n}\n\nfunc (f *factoryImpl) init(clusterName string, limiters map[string]quotas.Limiter) {\n\tf.datastores = make(map[storeType]Datastore, len(storeTypes))\n\tdefaultCfg := f.config.DataStores[f.config.DefaultStore]\n\tif defaultCfg.Cassandra != nil {\n\t\tf.logger.Warn(\"Cassandra config is deprecated, please use NoSQL with pluginName of cassandra.\")\n\t}\n\tdefaultDataStore := Datastore{ratelimit: limiters[f.config.DefaultStore]}\n\tswitch {\n\tcase defaultCfg.NoSQL != nil:\n\t\tparser := f.getParser()\n\t\ttaskSerializer := serialization.NewTaskSerializer(parser)\n\t\tshardedNoSQLConfig := defaultCfg.NoSQL.ConvertToShardedNoSQLConfig()\n\t\tdefaultDataStore.factory = nosql.NewFactory(*shardedNoSQLConfig, clusterName, f.logger, f.metricsClient, taskSerializer, parser, f.dc)\n\tcase defaultCfg.ShardedNoSQL != nil:\n\t\tparser := f.getParser()\n\t\ttaskSerializer := serialization.NewTaskSerializer(parser)\n\t\tdefaultDataStore.factory = nosql.NewFactory(*defaultCfg.ShardedNoSQL, clusterName, f.logger, f.metricsClient, taskSerializer, parser, f.dc)\n\tcase defaultCfg.SQL != nil:\n\t\tif defaultCfg.SQL.EncodingType == \"\" {\n\t\t\tdefaultCfg.SQL.EncodingType = string(constants.EncodingTypeThriftRW)\n\t\t}\n\t\tif len(defaultCfg.SQL.DecodingTypes) == 0 {\n\t\t\tdefaultCfg.SQL.DecodingTypes = []string{\n\t\t\t\tstring(constants.EncodingTypeThriftRW),\n\t\t\t}\n\t\t}\n\t\tdefaultDataStore.factory = sql.NewFactory(*defaultCfg.SQL, clusterName, f.logger, f.getParser(), f.dc)\n\tdefault:\n\t\tf.logger.Fatal(\"invalid config: one of nosql or sql params must be specified for defaultDataStore\")\n\t}\n\n\tfor _, st := range storeTypes {\n\t\tif st != storeTypeVisibility {\n\t\t\tf.datastores[st] = defaultDataStore\n\t\t}\n\t}\n\n\tvisibilityCfg, ok := f.config.DataStores[f.config.VisibilityStore]\n\tif !ok {\n\t\tf.logger.Info(\"no visibilityStore is configured, will use advancedVisibilityStore\")\n\t\t// NOTE: f.datastores[storeTypeVisibility] will be nil\n\t\treturn\n\t}\n\n\tif visibilityCfg.Cassandra != nil {\n\t\tf.logger.Warn(\"Cassandra config is deprecated, please use NoSQL with pluginName of cassandra.\")\n\t}\n\tvisibilityDataStore := Datastore{ratelimit: limiters[f.config.VisibilityStore]}\n\tswitch {\n\tcase visibilityCfg.NoSQL != nil:\n\t\tparser := f.getParser()\n\t\ttaskSerializer := serialization.NewTaskSerializer(parser)\n\t\tshardedNoSQLConfig := visibilityCfg.NoSQL.ConvertToShardedNoSQLConfig()\n\t\tvisibilityDataStore.factory = nosql.NewFactory(*shardedNoSQLConfig, clusterName, f.logger, f.metricsClient, taskSerializer, parser, f.dc)\n\tcase visibilityCfg.SQL != nil:\n\t\tvisibilityDataStore.factory = sql.NewFactory(*visibilityCfg.SQL, clusterName, f.logger, f.getParser(), f.dc)\n\tdefault:\n\t\tf.logger.Fatal(\"invalid config: one of nosql or sql params must be specified for visibilityStore\")\n\t}\n\n\tf.datastores[storeTypeVisibility] = visibilityDataStore\n}\n\nfunc (f *factoryImpl) getParser() serialization.Parser {\n\tparser, err := serialization.NewParser(f.dc)\n\tif err != nil {\n\t\tf.logger.Fatal(\"failed to construct parser\", tag.Error(err))\n\t}\n\treturn parser\n}\n\nfunc buildRatelimiters(cfg *config.Persistence, maxQPS quotas.RPSFunc) map[string]quotas.Limiter {\n\tresult := make(map[string]quotas.Limiter, len(cfg.DataStores))\n\tfor dsName := range cfg.DataStores {\n\t\tif maxQPS != nil && maxQPS() > 0 {\n\t\t\tresult[dsName] = quotas.NewDynamicRateLimiter(maxQPS)\n\t\t}\n\t}\n\treturn result\n}\n\nfunc setupPinotVisibilityManager(params *Params, resourceConfig *service.Config, logger log.Logger, dc *p.DynamicConfiguration, callerBypass quotas.CallerBypass) (p.VisibilityManager, error) {\n\tvisibilityProducer, err := params.MessagingClient.NewProducer(constants.PinotVisibilityAppName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newPinotVisibilityManager(params.PinotClient, resourceConfig, visibilityProducer, params.MetricsClient, logger, dc, callerBypass), nil\n}\n\nfunc setupESVisibilityManager(params *Params, resourceConfig *service.Config, logger log.Logger, dc *p.DynamicConfiguration, callerBypass quotas.CallerBypass) (p.VisibilityManager, error) {\n\tvisibilityIndexName := params.ESConfig.Indices[constants.VisibilityAppName]\n\tvisibilityProducer, err := params.MessagingClient.NewProducer(constants.VisibilityAppName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newESVisibilityManager(visibilityIndexName, params.ESClient, resourceConfig, visibilityProducer, params.MetricsClient, logger, dc, callerBypass), nil\n}\n\nfunc setupOSVisibilityManager(params *Params, resourceConfig *service.Config, logger log.Logger, dc *p.DynamicConfiguration, callerBypass quotas.CallerBypass) (p.VisibilityManager, error) {\n\tvisibilityIndexName := params.OSConfig.Indices[constants.VisibilityAppName]\n\tvisibilityProducer, err := params.MessagingClient.NewProducer(constants.VisibilityAppName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newESVisibilityManager(visibilityIndexName, params.OSClient, resourceConfig, visibilityProducer, params.MetricsClient, logger, dc, callerBypass), nil\n}\n"
  },
  {
    "path": "common/persistence/client/factory_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: factory.go\n//\n// Generated by this command:\n//\n//\tmockgen -package client -source factory.go -destination factory_mock.go\n//\n\n// Package client is a generated GoMock package.\npackage client\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\tservice \"github.com/uber/cadence/common/service\"\n)\n\n// MockFactory is a mock of Factory interface.\ntype MockFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockFactoryMockRecorder is the mock recorder for MockFactory.\ntype MockFactoryMockRecorder struct {\n\tmock *MockFactory\n}\n\n// NewMockFactory creates a new mock instance.\nfunc NewMockFactory(ctrl *gomock.Controller) *MockFactory {\n\tmock := &MockFactory{ctrl: ctrl}\n\tmock.recorder = &MockFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFactory) EXPECT() *MockFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockFactory) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockFactoryMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockFactory)(nil).Close))\n}\n\n// NewConfigStoreManager mocks base method.\nfunc (m *MockFactory) NewConfigStoreManager() (persistence.ConfigStoreManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewConfigStoreManager\")\n\tret0, _ := ret[0].(persistence.ConfigStoreManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewConfigStoreManager indicates an expected call of NewConfigStoreManager.\nfunc (mr *MockFactoryMockRecorder) NewConfigStoreManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewConfigStoreManager\", reflect.TypeOf((*MockFactory)(nil).NewConfigStoreManager))\n}\n\n// NewDomainAuditManager mocks base method.\nfunc (m *MockFactory) NewDomainAuditManager() (persistence.DomainAuditManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewDomainAuditManager\")\n\tret0, _ := ret[0].(persistence.DomainAuditManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewDomainAuditManager indicates an expected call of NewDomainAuditManager.\nfunc (mr *MockFactoryMockRecorder) NewDomainAuditManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewDomainAuditManager\", reflect.TypeOf((*MockFactory)(nil).NewDomainAuditManager))\n}\n\n// NewDomainManager mocks base method.\nfunc (m *MockFactory) NewDomainManager() (persistence.DomainManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewDomainManager\")\n\tret0, _ := ret[0].(persistence.DomainManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewDomainManager indicates an expected call of NewDomainManager.\nfunc (mr *MockFactoryMockRecorder) NewDomainManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewDomainManager\", reflect.TypeOf((*MockFactory)(nil).NewDomainManager))\n}\n\n// NewDomainReplicationQueueManager mocks base method.\nfunc (m *MockFactory) NewDomainReplicationQueueManager() (persistence.QueueManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewDomainReplicationQueueManager\")\n\tret0, _ := ret[0].(persistence.QueueManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewDomainReplicationQueueManager indicates an expected call of NewDomainReplicationQueueManager.\nfunc (mr *MockFactoryMockRecorder) NewDomainReplicationQueueManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewDomainReplicationQueueManager\", reflect.TypeOf((*MockFactory)(nil).NewDomainReplicationQueueManager))\n}\n\n// NewExecutionManager mocks base method.\nfunc (m *MockFactory) NewExecutionManager(shardID int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewExecutionManager\", shardID)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewExecutionManager indicates an expected call of NewExecutionManager.\nfunc (mr *MockFactoryMockRecorder) NewExecutionManager(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewExecutionManager\", reflect.TypeOf((*MockFactory)(nil).NewExecutionManager), shardID)\n}\n\n// NewHistoryManager mocks base method.\nfunc (m *MockFactory) NewHistoryManager() (persistence.HistoryManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewHistoryManager\")\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewHistoryManager indicates an expected call of NewHistoryManager.\nfunc (mr *MockFactoryMockRecorder) NewHistoryManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewHistoryManager\", reflect.TypeOf((*MockFactory)(nil).NewHistoryManager))\n}\n\n// NewShardManager mocks base method.\nfunc (m *MockFactory) NewShardManager() (persistence.ShardManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewShardManager\")\n\tret0, _ := ret[0].(persistence.ShardManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewShardManager indicates an expected call of NewShardManager.\nfunc (mr *MockFactoryMockRecorder) NewShardManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewShardManager\", reflect.TypeOf((*MockFactory)(nil).NewShardManager))\n}\n\n// NewTaskManager mocks base method.\nfunc (m *MockFactory) NewTaskManager() (persistence.TaskManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewTaskManager\")\n\tret0, _ := ret[0].(persistence.TaskManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewTaskManager indicates an expected call of NewTaskManager.\nfunc (mr *MockFactoryMockRecorder) NewTaskManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewTaskManager\", reflect.TypeOf((*MockFactory)(nil).NewTaskManager))\n}\n\n// NewVisibilityManager mocks base method.\nfunc (m *MockFactory) NewVisibilityManager(params *Params, serviceConfig *service.Config) (persistence.VisibilityManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewVisibilityManager\", params, serviceConfig)\n\tret0, _ := ret[0].(persistence.VisibilityManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewVisibilityManager indicates an expected call of NewVisibilityManager.\nfunc (mr *MockFactoryMockRecorder) NewVisibilityManager(params, serviceConfig any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewVisibilityManager\", reflect.TypeOf((*MockFactory)(nil).NewVisibilityManager), params, serviceConfig)\n}\n\n// MockDataStoreFactory is a mock of DataStoreFactory interface.\ntype MockDataStoreFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDataStoreFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockDataStoreFactoryMockRecorder is the mock recorder for MockDataStoreFactory.\ntype MockDataStoreFactoryMockRecorder struct {\n\tmock *MockDataStoreFactory\n}\n\n// NewMockDataStoreFactory creates a new mock instance.\nfunc NewMockDataStoreFactory(ctrl *gomock.Controller) *MockDataStoreFactory {\n\tmock := &MockDataStoreFactory{ctrl: ctrl}\n\tmock.recorder = &MockDataStoreFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDataStoreFactory) EXPECT() *MockDataStoreFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDataStoreFactory) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDataStoreFactoryMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDataStoreFactory)(nil).Close))\n}\n\n// NewConfigStore mocks base method.\nfunc (m *MockDataStoreFactory) NewConfigStore() (persistence.ConfigStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewConfigStore\")\n\tret0, _ := ret[0].(persistence.ConfigStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewConfigStore indicates an expected call of NewConfigStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewConfigStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewConfigStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewConfigStore))\n}\n\n// NewDomainAuditStore mocks base method.\nfunc (m *MockDataStoreFactory) NewDomainAuditStore() (persistence.DomainAuditStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewDomainAuditStore\")\n\tret0, _ := ret[0].(persistence.DomainAuditStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewDomainAuditStore indicates an expected call of NewDomainAuditStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewDomainAuditStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewDomainAuditStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewDomainAuditStore))\n}\n\n// NewDomainStore mocks base method.\nfunc (m *MockDataStoreFactory) NewDomainStore() (persistence.DomainStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewDomainStore\")\n\tret0, _ := ret[0].(persistence.DomainStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewDomainStore indicates an expected call of NewDomainStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewDomainStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewDomainStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewDomainStore))\n}\n\n// NewExecutionStore mocks base method.\nfunc (m *MockDataStoreFactory) NewExecutionStore(shardID int) (persistence.ExecutionStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewExecutionStore\", shardID)\n\tret0, _ := ret[0].(persistence.ExecutionStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewExecutionStore indicates an expected call of NewExecutionStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewExecutionStore(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewExecutionStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewExecutionStore), shardID)\n}\n\n// NewHistoryStore mocks base method.\nfunc (m *MockDataStoreFactory) NewHistoryStore() (persistence.HistoryStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewHistoryStore\")\n\tret0, _ := ret[0].(persistence.HistoryStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewHistoryStore indicates an expected call of NewHistoryStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewHistoryStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewHistoryStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewHistoryStore))\n}\n\n// NewQueue mocks base method.\nfunc (m *MockDataStoreFactory) NewQueue(queueType persistence.QueueType) (persistence.QueueStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewQueue\", queueType)\n\tret0, _ := ret[0].(persistence.QueueStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewQueue indicates an expected call of NewQueue.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewQueue(queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewQueue\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewQueue), queueType)\n}\n\n// NewShardStore mocks base method.\nfunc (m *MockDataStoreFactory) NewShardStore() (persistence.ShardStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewShardStore\")\n\tret0, _ := ret[0].(persistence.ShardStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewShardStore indicates an expected call of NewShardStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewShardStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewShardStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewShardStore))\n}\n\n// NewTaskStore mocks base method.\nfunc (m *MockDataStoreFactory) NewTaskStore() (persistence.TaskStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewTaskStore\")\n\tret0, _ := ret[0].(persistence.TaskStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewTaskStore indicates an expected call of NewTaskStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewTaskStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewTaskStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewTaskStore))\n}\n\n// NewVisibilityStore mocks base method.\nfunc (m *MockDataStoreFactory) NewVisibilityStore(sortByCloseTime bool) (persistence.VisibilityStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewVisibilityStore\", sortByCloseTime)\n\tret0, _ := ret[0].(persistence.VisibilityStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewVisibilityStore indicates an expected call of NewVisibilityStore.\nfunc (mr *MockDataStoreFactoryMockRecorder) NewVisibilityStore(sortByCloseTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewVisibilityStore\", reflect.TypeOf((*MockDataStoreFactory)(nil).NewVisibilityStore), sortByCloseTime)\n}\n"
  },
  {
    "path": "common/persistence/client/factory_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage client\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\n\t\"github.com/IBM/sarama/mocks\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/messaging/kafka\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n/*\nNew...Manager methods are intentionally NOT tested here.\n\nThey require importing a datastore plugin, which imports the persistence tests,\nwhich imports this package, leading to a cycle.\nThey also generally try to connect to the target datastore *immediately* upon construction,\nwhich means these would need to be integration tests...\n...which we already have, in the persistence/persistence-tests package.\n*/\n\nfunc TestNew(t *testing.T) {\n\t// largely a sanity test for makeFactory, but it does ensure it constructs and closes,\n\t// though this does not actually achieve very much.\n\tfact := makeFactory(t)\n\tfact.Close()\n}\n\n// check ensures the func returns a value and no error.\n// most manager-constructors can use this, but not all.\nfunc check[T interface{}](t *testing.T, fn func() (T, error)) {\n\tval, err := fn()\n\tassert.NoError(t, err, \"manager-constructor method should not error\")\n\tassert.NotNil(t, val, \"manager-constructor method should return a non-nil value\")\n}\n\nfunc TestFactoryMethods(t *testing.T) {\n\tt.Run(\"NewTaskManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeTask)\n\n\t\tds.EXPECT().NewTaskStore().Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewTaskManager)\n\t})\n\tt.Run(\"NewShardManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeShard)\n\n\t\tds.EXPECT().NewShardStore().Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewShardManager)\n\t})\n\tt.Run(\"NewHistoryManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeHistory)\n\n\t\tds.EXPECT().NewHistoryStore().Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewHistoryManager)\n\t})\n\tt.Run(\"NewDomainManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeMetadata)\n\n\t\tds.EXPECT().NewDomainStore().Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewDomainManager)\n\t})\n\tt.Run(\"NewExecutionManager\", func(t *testing.T) {\n\t\t// cannot control the persistence.NewExecutionManagerImpl call, so we must prevent calls to it.\n\t\t// currently the only call is the metrics wrapper calling GetShardID(), so just don't use the metrics wrapper.\n\t\tfact := makeFactoryWithMetrics(t, false)\n\t\tds := mockDatastore(t, fact, storeTypeExecution)\n\n\t\tshard := rand.Int()\n\t\tds.EXPECT().NewExecutionStore(shard).Return(nil, nil).MinTimes(1)\n\t\tem, err := fact.NewExecutionManager(shard)\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, em)\n\t})\n\tt.Run(\"NewVisibilityManager can be nil\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\t// no datastores are mocked as there are no calls at all expected\n\t\tvm, err := fact.NewVisibilityManager(\n\t\t\tnil, // params are unused\n\t\t\t&service.Config{\n\t\t\t\t// if both of these are nil, a nil response is correct,\n\t\t\t\t// because no \"manager\" is needed:\n\t\t\t\t// ES cannot be dynamically enabled, so no dual-writing / etc is\n\t\t\t\t// needed, so the baseline database store is sufficient.\n\t\t\t\tReadVisibilityStoreName:  nil,\n\t\t\t\tWriteVisibilityStoreName: nil,\n\t\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Nil(t, vm, \"nil response is expected if advanced visibility cannot be enabled dynamically\")\n\t})\n\tt.Run(\"NewDomainReplicationQueueManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeQueue)\n\n\t\tds.EXPECT().NewQueue(persistence.DomainReplicationQueueType).Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewDomainReplicationQueueManager)\n\t})\n\tt.Run(\"NewConfigStoreManager\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeConfigStore)\n\n\t\tds.EXPECT().NewConfigStore().Return(nil, nil).MinTimes(1)\n\t\tcheck(t, fact.NewConfigStoreManager)\n\t})\n\tt.Run(\"NewVisibilityManager_TripleVisibilityManager_Pinot\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeVisibility)\n\n\t\tlogger := testlogger.New(t)\n\t\tmc := messaging.NewMockClient(gomock.NewController(t))\n\t\tmc.EXPECT().NewProducer(gomock.Any()).Return(kafka.NewKafkaProducer(\"test-topic\", mocks.NewSyncProducer(t, nil), logger), nil).MinTimes(1)\n\t\treadFromClosed := true\n\t\ttestAttributes := map[string]interface{}{\n\t\t\t\"CustomAttribute\": \"test\", // Define your custom attributes and types\n\t\t}\n\n\t\tds.EXPECT().NewVisibilityStore(readFromClosed).Return(nil, nil).MinTimes(1)\n\t\t_, err := fact.NewVisibilityManager(&Params{\n\t\t\tPersistenceConfig: config.Persistence{\n\t\t\t\t// a configured VisibilityStore uses the db store, which is mockable,\n\t\t\t\t// unlike basically every other store.\n\t\t\t\tAdvancedVisibilityStore: \"pinot-visibility\",\n\t\t\t\tVisibilityStore:         \"fake\",\n\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\"pinot-visibility\": {\n\t\t\t\t\t\tPinot: &config.PinotVisibilityConfig{\n\t\t\t\t\t\t\t// fields are unused but must be non-nil\n\t\t\t\t\t\t\tCluster: \"cluster\",\n\t\t\t\t\t\t\tBroker:  \"broker\",\n\t\t\t\t\t\t}, // fields are unused but must be non-nil\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tMessagingClient: mc,\n\t\t\tPinotConfig: &config.PinotVisibilityConfig{\n\t\t\t\tMigration: config.VisibilityMigration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\tESConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t},\n\t\t}, &service.Config{\n\t\t\t// must be non-nil to create a \"manager\", else nil return from NewVisibilityManager is expected\n\t\t\tReadVisibilityStoreName: func(domain string) string {\n\t\t\t\treturn \"es\" // any value is fine as there are no read calls\n\t\t\t},\n\t\t\t// non-nil avoids a warning log\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn readFromClosed // any value is fine as there are no read calls\n\t\t\t},\n\t\t\tValidSearchAttributes: func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\t\t\treturn testAttributes\n\t\t\t},\n\t\t})\n\t\tassert.NoError(t, err)\n\t})\n\tt.Run(\"NewVisibilityManager_DualVisibilityManager_ES\", func(t *testing.T) {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeVisibility)\n\n\t\tlogger := testlogger.New(t)\n\t\tmc := messaging.NewMockClient(gomock.NewController(t))\n\t\tmc.EXPECT().NewProducer(gomock.Any()).Return(kafka.NewKafkaProducer(\"test-topic\", mocks.NewSyncProducer(t, nil), logger), nil).MinTimes(1)\n\t\treadFromClosed := true\n\t\ttestAttributes := map[string]interface{}{\n\t\t\t\"CustomAttribute\": \"test\", // Define your custom attributes and types\n\t\t}\n\n\t\tds.EXPECT().NewVisibilityStore(readFromClosed).Return(nil, nil).MinTimes(1)\n\t\t_, err := fact.NewVisibilityManager(&Params{\n\t\t\tPersistenceConfig: config.Persistence{\n\t\t\t\t// a configured VisibilityStore uses the db store, which is mockable,\n\t\t\t\t// unlike basically every other store.\n\t\t\t\tAdvancedVisibilityStore: \"es-visibility\",\n\t\t\t\tVisibilityStore:         \"fake\",\n\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\"es-visibility\": {\n\t\t\t\t\t\tElasticSearch: &config.ElasticSearchConfig{\n\t\t\t\t\t\t\t// fields are unused but must be non-nil\n\t\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}, // fields are unused but must be non-nil\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tMessagingClient: mc,\n\t\t\tESConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t},\n\t\t}, &service.Config{\n\t\t\t// must be non-nil to create a \"manager\", else nil return from NewVisibilityManager is expected\n\t\t\tReadVisibilityStoreName: func(domain string) string {\n\t\t\t\treturn \"es\" // any value is fine as there are no read calls\n\t\t\t},\n\t\t\t// non-nil avoids a warning log\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn readFromClosed // any value is fine as there are no read calls\n\t\t\t},\n\t\t\tValidSearchAttributes: func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\t\t\treturn testAttributes\n\t\t\t},\n\t\t})\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc makeFactory(t *testing.T) Factory {\n\treturn makeFactoryWithMetrics(t, true)\n}\n\nfunc makeFactoryWithMetrics(t *testing.T, withMetrics bool) Factory {\n\tqpsFn := func() float64 { return 1000 } // anything non-zero exercises the ratelimit config paths\n\tlogger := testlogger.New(t)\n\tvar met metrics.Client\n\tif withMetrics {\n\t\tmet = metrics.NewClient(tally.NewTestScope(\"\", nil), service.GetMetricsServiceIdx(service.Frontend, logger), metrics.HistogramMigration{})\n\t}\n\tctrl := gomock.NewController(t)\n\tdc := dynamicconfig.NewCollection(dynamicconfig.NewMockClient(ctrl), logger)\n\tpdc := persistence.NewDynamicConfiguration(dc)\n\n\tcfg := &config.Persistence{\n\t\tDefaultStore:            \"fake\",\n\t\tVisibilityStore:         \"fake\",\n\t\tAdvancedVisibilityStore: \"fake\",\n\t\tNumHistoryShards:        1024,\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"fake\": {\n\t\t\t\tNoSQL:         &config.NoSQL{},                 // fields are unused but must be non-nil\n\t\t\t\tElasticSearch: &config.ElasticSearchConfig{},   // fields are unused but must be non-nil\n\t\t\t\tPinot:         &config.PinotVisibilityConfig{}, // fields are unused but must be non-nil\n\t\t\t},\n\t\t},\n\t\tTransactionSizeLimit: nil,\n\t\tErrorInjectionRate: func(opts ...dynamicproperties.FilterOption) float64 {\n\t\t\treturn 0.5 // half errors, unused in these tests beyond \"nonzero\" so it wraps with the error injector\n\t\t},\n\t}\n\n\treturn NewFactory(cfg, qpsFn, \"test cluster\", met, logger, pdc)\n}\n\nfunc mockDatastore(t *testing.T, fact Factory, store storeType) *MockDataStoreFactory {\n\tctrl := gomock.NewController(t)\n\timpl := fact.(*factoryImpl)\n\n\t// I would prefer to mock everything so calls to the wrong store could have better errors,\n\t// but there doesn't seem to be a way to \"name\" these or say \"any call fails with X message\",\n\t// nor can you check what calls occurred after the fact.\n\t//\n\t// so just mock the only thing expected, so you get a nil panic stack trace when it fails\n\t// somewhere in a persistence database connection or something.  it's sometimes easier to debug.\n\tmock := NewMockDataStoreFactory(ctrl)\n\tds := impl.datastores[store]\n\tds.factory = mock\n\timpl.datastores[store] = ds // write back the value type\n\n\treturn mock\n}\n\nfunc TestVisibilityManagers(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tadvanced    string\n\t\tdatastores  map[string]config.DataStore\n\t\tesConfig    *config.ElasticSearchConfig\n\t\tosConfig    *config.ElasticSearchConfig\n\t\tpinotConfig *config.PinotVisibilityConfig\n\t}{\n\t\t{\n\t\t\tname:     \"TripleVisibilityManager_OpenSearch\",\n\t\t\tadvanced: \"os-visibility\",\n\t\t\tdatastores: map[string]config.DataStore{\n\t\t\t\t\"os-visibility\": {\n\t\t\t\t\tElasticSearch: &config.ElasticSearchConfig{\n\t\t\t\t\t\t// fields are unused but must be non-nil\n\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, // fields are unused but must be non-nil\n\t\t\t\t},\n\t\t\t},\n\t\t\tosConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t\tMigration: config.VisibilityMigration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\tesConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"NewVisibilityManager_TripleVisibilityManager_Pinot\",\n\t\t\tadvanced: \"pinot-visibility\",\n\t\t\tdatastores: map[string]config.DataStore{\n\t\t\t\t\"pinot-visibility\": {\n\t\t\t\t\tPinot: &config.PinotVisibilityConfig{\n\t\t\t\t\t\tCluster: \"cluster\",\n\t\t\t\t\t\tBroker:  \"broker\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpinotConfig: &config.PinotVisibilityConfig{\n\t\t\t\tMigration: config.VisibilityMigration{\n\t\t\t\t\tEnabled: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\tesConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"NewVisibilityManager_DualVisibilityManager_Pinot\",\n\t\t\tadvanced: \"pinot-visibility\",\n\t\t\tdatastores: map[string]config.DataStore{\n\t\t\t\t\"pinot-visibility\": {\n\t\t\t\t\tPinot: &config.PinotVisibilityConfig{\n\t\t\t\t\t\tCluster: \"cluster\",\n\t\t\t\t\t\tBroker:  \"broker\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpinotConfig: &config.PinotVisibilityConfig{\n\t\t\t\tMigration: config.VisibilityMigration{\n\t\t\t\t\tEnabled: false,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"NewVisibilityManager_DualVisibilityManager_ES\",\n\t\t\tadvanced: \"es-visibility\",\n\t\t\tdatastores: map[string]config.DataStore{\n\t\t\t\t\"es-visibility\": {\n\t\t\t\t\tElasticSearch: &config.ElasticSearchConfig{\n\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tesConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"NewVisibilityManager_DualVisibilityManager_OS\",\n\t\t\tadvanced: \"os-visibility\",\n\t\t\tdatastores: map[string]config.DataStore{\n\t\t\t\t\"os-visibility\": {\n\t\t\t\t\tElasticSearch: &config.ElasticSearchConfig{\n\t\t\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tosConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{\n\t\t\t\t\t\"visibility\": \"test-index\",\n\t\t\t\t},\n\t\t\t\tMigration: config.VisibilityMigration{\n\t\t\t\t\tEnabled: false,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tfact := makeFactory(t)\n\t\tds := mockDatastore(t, fact, storeTypeVisibility)\n\t\tds.EXPECT().NewVisibilityStore(true).Return(nil, nil).MinTimes(1)\n\t\tmc := messaging.NewMockClient(gomock.NewController(t))\n\t\tmc.EXPECT().NewProducer(gomock.Any()).Return(kafka.NewKafkaProducer(\"test-topic\", mocks.NewSyncProducer(t, nil), testlogger.New(t)), nil).MinTimes(1)\n\n\t\t_, err := fact.NewVisibilityManager(&Params{\n\t\t\tPersistenceConfig: config.Persistence{\n\t\t\t\tAdvancedVisibilityStore: test.advanced,\n\t\t\t\tVisibilityStore:         \"fake\",\n\t\t\t\tDataStores:              test.datastores,\n\t\t\t},\n\t\t\tMessagingClient: mc,\n\t\t\tPinotConfig:     test.pinotConfig,\n\t\t\tESConfig:        test.esConfig,\n\t\t\tOSConfig:        test.osConfig,\n\t\t}, &service.Config{\n\t\t\t// must be non-nil to create a \"manager\", else nil return from NewVisibilityManager is expected\n\t\t\tReadVisibilityStoreName: func(domain string) string {\n\t\t\t\treturn \"es\" // any value is fine as there are no read calls\n\t\t\t},\n\t\t\t// non-nil avoids a warning log\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn true // any value is fine as there are no read calls\n\t\t\t},\n\t\t\tValidSearchAttributes: func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\t\t\treturn map[string]interface{}{\n\t\t\t\t\t\"CustomAttribute\": \"test\",\n\t\t\t\t}\n\t\t\t},\n\t\t})\n\t\tassert.NoError(t, err)\n\t}\n}\n"
  },
  {
    "path": "common/persistence/config.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype (\n\t// DynamicConfiguration represents dynamic configuration for persistence layer\n\tDynamicConfiguration struct {\n\t\tEnableSQLAsyncTransaction                dynamicproperties.BoolPropertyFn\n\t\tEnableCassandraAllConsistencyLevelDelete dynamicproperties.BoolPropertyFn\n\t\tPersistenceSampleLoggingRate             dynamicproperties.IntPropertyFn\n\t\tEnableShardIDMetrics                     dynamicproperties.BoolPropertyFn\n\t\tEnableHistoryTaskDualWriteMode           dynamicproperties.BoolPropertyFn\n\t\tReadNoSQLHistoryTaskFromDataBlob         dynamicproperties.BoolPropertyFn\n\t\tReadNoSQLShardFromDataBlob               dynamicproperties.BoolPropertyFn\n\t\tSerializationEncoding                    dynamicproperties.StringPropertyFn\n\t\tDomainAuditLogTTL                        dynamicproperties.DurationPropertyFnWithDomainIDFilter\n\t\tHistoryNodeDeleteBatchSize               dynamicproperties.IntPropertyFn\n\t\tRateLimiterBypassCallerTypes             dynamicproperties.ListPropertyFn\n\t}\n)\n\n// NewDynamicConfiguration returns new config with default values\nfunc NewDynamicConfiguration(dc *dynamicconfig.Collection) *DynamicConfiguration {\n\treturn &DynamicConfiguration{\n\t\tEnableSQLAsyncTransaction:                dc.GetBoolProperty(dynamicproperties.EnableSQLAsyncTransaction),\n\t\tEnableCassandraAllConsistencyLevelDelete: dc.GetBoolProperty(dynamicproperties.EnableCassandraAllConsistencyLevelDelete),\n\t\tPersistenceSampleLoggingRate:             dc.GetIntProperty(dynamicproperties.SampleLoggingRate),\n\t\tEnableShardIDMetrics:                     dc.GetBoolProperty(dynamicproperties.EnableShardIDMetrics),\n\t\tEnableHistoryTaskDualWriteMode:           dc.GetBoolProperty(dynamicproperties.EnableNoSQLHistoryTaskDualWriteMode),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dc.GetBoolProperty(dynamicproperties.ReadNoSQLHistoryTaskFromDataBlob),\n\t\tReadNoSQLShardFromDataBlob:               dc.GetBoolProperty(dynamicproperties.ReadNoSQLShardFromDataBlob),\n\t\tSerializationEncoding:                    dc.GetStringProperty(dynamicproperties.SerializationEncoding),\n\t\tDomainAuditLogTTL:                        dc.GetDurationPropertyFilteredByDomainID(dynamicproperties.DomainAuditLogTTL),\n\t\tHistoryNodeDeleteBatchSize:               dc.GetIntProperty(dynamicproperties.HistoryNodeDeleteBatchSize),\n\t\tRateLimiterBypassCallerTypes:             dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/config_store_manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype (\n\n\t// configStoreManagerImpl implements ConfigStoreManager based on ConfigStore and PayloadSerializer\n\tconfigStoreManagerImpl struct {\n\t\tserializer  PayloadSerializer\n\t\tpersistence ConfigStore\n\t\tlogger      log.Logger\n\t\ttimeSrc     clock.TimeSource\n\t}\n)\n\nvar _ ConfigStoreManager = (*configStoreManagerImpl)(nil)\n\n// NewConfigStoreManagerImpl returns new ConfigStoreManager\nfunc NewConfigStoreManagerImpl(persistence ConfigStore, logger log.Logger) ConfigStoreManager {\n\treturn &configStoreManagerImpl{\n\t\tserializer:  NewPayloadSerializer(),\n\t\tpersistence: persistence,\n\t\tlogger:      logger,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t}\n}\n\nfunc (m *configStoreManagerImpl) Close() {\n\tm.persistence.Close()\n}\n\nfunc (m *configStoreManagerImpl) FetchDynamicConfig(ctx context.Context, cfgType ConfigType) (*FetchDynamicConfigResponse, error) {\n\tvalues, err := m.persistence.FetchConfig(ctx, cfgType)\n\tif err != nil || values == nil {\n\t\treturn nil, err\n\t}\n\n\tconfig, err := m.serializer.DeserializeDynamicConfigBlob(values.Values)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &FetchDynamicConfigResponse{Snapshot: &DynamicConfigSnapshot{\n\t\tVersion: values.Version,\n\t\tValues:  config,\n\t}}, nil\n}\n\nfunc (m *configStoreManagerImpl) UpdateDynamicConfig(ctx context.Context, request *UpdateDynamicConfigRequest, cfgType ConfigType) error {\n\tblob, err := m.serializer.SerializeDynamicConfigBlob(request.Snapshot.Values, constants.EncodingTypeThriftRW)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tentry := &InternalConfigStoreEntry{\n\t\tRowType:   int(cfgType),\n\t\tVersion:   request.Snapshot.Version,\n\t\tTimestamp: m.timeSrc.Now(),\n\t\tValues:    blob,\n\t}\n\n\treturn m.persistence.UpdateConfig(ctx, entry)\n}\n"
  },
  {
    "path": "common/persistence/config_store_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc setUpMocksForConfigStoreManager(t *testing.T) (*configStoreManagerImpl, *MockConfigStore, *MockPayloadSerializer) {\n\tctrl := gomock.NewController(t)\n\tmockStore := NewMockConfigStore(ctrl)\n\tmockSerializer := NewMockPayloadSerializer(ctrl)\n\tlogger := log.NewNoop()\n\n\treturn &configStoreManagerImpl{\n\t\tserializer:  mockSerializer,\n\t\tpersistence: mockStore,\n\t\tlogger:      logger,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t}, mockStore, mockSerializer\n}\n\nfunc TestNewConfigStoreManagerImpl(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockStore := NewMockConfigStore(ctrl)\n\tlogger := log.NewNoop()\n\tm := NewConfigStoreManagerImpl(mockStore, logger)\n\tassert.NotNil(t, m)\n\tcm, ok := m.(*configStoreManagerImpl)\n\tassert.True(t, ok)\n\tassert.Equal(t, mockStore, cm.persistence)\n\tassert.Equal(t, logger, cm.logger)\n\tassert.NotNil(t, cm.serializer)\n}\n\nfunc TestFetchDynamicConfig(t *testing.T) {\n\tencodingType := constants.EncodingTypeThriftRW\n\ttestCases := []struct {\n\t\tname             string\n\t\tsetupMock        func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer)\n\t\tcfgType          ConfigType\n\t\texpectError      bool\n\t\texpectedError    string\n\t\texpectedResponse *FetchDynamicConfigResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// Mocking persistence DataBlob\n\t\t\t\tmockStore.EXPECT().FetchConfig(gomock.Any(), DynamicConfig).Return(&InternalConfigStoreEntry{\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tTimestamp: time.Now(),\n\t\t\t\t\tValues:    &DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\t// Mocking deserialization of persistence.DataBlob into types.DynamicConfigBlob\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeDynamicConfigBlob(&DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")}).\n\t\t\t\t\tReturn(&types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tcfgType:     DynamicConfig, // Updated to use DynamicConfig\n\t\t\texpectError: false,\n\t\t\texpectedResponse: &FetchDynamicConfigResponse{\n\t\t\t\tSnapshot: &DynamicConfigSnapshot{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"fetch config error\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().FetchConfig(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"fetch error\")).Times(1)\n\t\t\t},\n\t\t\tcfgType:       DynamicConfig, // Updated to use DynamicConfig\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"fetch error\",\n\t\t},\n\t\t{\n\t\t\tname: \"deserialization error\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().FetchConfig(gomock.Any(), gomock.Any()).Return(&InternalConfigStoreEntry{\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tTimestamp: time.Now(),\n\t\t\t\t\tValues:    &DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeDynamicConfigBlob(&DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")}).\n\t\t\t\t\tReturn(nil, errors.New(\"deserialization error\")).Times(1)\n\t\t\t},\n\t\t\tcfgType:       DynamicConfig, // Updated to use DynamicConfig\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"deserialization error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tconfigStoreManager, mockStore, mockSerializer := setUpMocksForConfigStoreManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\tresp, err := configStoreManager.FetchDynamicConfig(context.Background(), tc.cfgType)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResponse, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDynamicConfig(t *testing.T) {\n\tencodingType := constants.EncodingTypeThriftRW\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer)\n\t\tcfgType       ConfigType\n\t\trequest       *UpdateDynamicConfigRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// Mock serialization of types.DynamicConfigBlob to persistence.DataBlob\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeDynamicConfigBlob(&types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")}, nil).Times(1)\n\n\t\t\t\tmockStore.EXPECT().UpdateConfig(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tcfgType: DynamicConfig, // Updated to use DynamicConfig\n\t\t\trequest: &UpdateDynamicConfigRequest{\n\t\t\t\tSnapshot: &DynamicConfigSnapshot{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"serialization error\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeDynamicConfigBlob(&types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(nil, errors.New(\"serialization error\")).Times(1)\n\t\t\t},\n\t\t\tcfgType: DynamicConfig, // Updated to use DynamicConfig\n\t\t\trequest: &UpdateDynamicConfigRequest{\n\t\t\t\tSnapshot: &DynamicConfigSnapshot{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialization error\",\n\t\t},\n\t\t{\n\t\t\tname: \"update config error\",\n\t\t\tsetupMock: func(mockStore *MockConfigStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeDynamicConfigBlob(&types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: encodingType, Data: []byte(\"serialized-values\")}, nil).Times(1)\n\n\t\t\t\tmockStore.EXPECT().UpdateConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"update error\")).Times(1)\n\t\t\t},\n\t\t\tcfgType: DynamicConfig, // Updated to use DynamicConfig\n\t\t\trequest: &UpdateDynamicConfigRequest{\n\t\t\t\tSnapshot: &DynamicConfigSnapshot{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tValues: &types.DynamicConfigBlob{\n\t\t\t\t\t\tSchemaVersion: 1,\n\t\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"TestEntry\",\n\t\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"update error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tconfigStoreManager, mockStore, mockSerializer := setUpMocksForConfigStoreManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\terr := configStoreManager.UpdateDynamicConfig(context.Background(), tc.request, tc.cfgType)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCloseConfigStoreManager(t *testing.T) {\n\tt.Run(\"close persistence\", func(t *testing.T) {\n\t\tconfigStoreManager, mockStore, _ := setUpMocksForConfigStoreManager(t)\n\n\t\t// Expect Close method to be called once\n\t\tmockStore.EXPECT().Close().Times(1)\n\n\t\t// Call the Close method\n\t\tconfigStoreManager.Close()\n\t})\n}\n"
  },
  {
    "path": "common/persistence/data_manager_interfaces.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// Generate rate limiter wrappers.\n//go:generate mockgen -package $GOPACKAGE -destination data_manager_interfaces_mock.go github.com/uber/cadence/common/persistence Task,ShardManager,ExecutionManager,ExecutionManagerFactory,TaskManager,HistoryManager,DomainManager,DomainAuditManager,QueueManager,ConfigStoreManager\n//go:generate gowrap gen -g -p . -i ConfigStoreManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/configstore_generated.go\n//go:generate gowrap gen -g -p . -i DomainManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/domain_generated.go\n//go:generate gowrap gen -g -p . -i HistoryManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/history_generated.go\n//go:generate gowrap gen -g -p . -i ExecutionManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/execution_generated.go\n//go:generate gowrap gen -g -p . -i QueueManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/queue_generated.go\n//go:generate gowrap gen -g -p . -i TaskManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/task_generated.go\n//go:generate gowrap gen -g -p . -i ShardManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/shard_generated.go\n\n// Generate error injector wrappers.\n//go:generate gowrap gen -g -p . -i ConfigStoreManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/configstore_generated.go\n//go:generate gowrap gen -g -p . -i ShardManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/shard_generated.go\n//go:generate gowrap gen -g -p . -i ExecutionManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/execution_generated.go\n//go:generate gowrap gen -g -p . -i TaskManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/task_generated.go\n//go:generate gowrap gen -g -p . -i HistoryManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/history_generated.go\n//go:generate gowrap gen -g -p . -i DomainManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/domain_generated.go\n//go:generate gowrap gen -g -p . -i QueueManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/queue_generated.go\n\n// Generate metered wrappers.\n//go:generate gowrap gen -g -p . -i ConfigStoreManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/configstore_generated.go\n//go:generate gowrap gen -g -p . -i ShardManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/shard_generated.go\n//go:generate gowrap gen -g -p . -i TaskManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/task_generated.go\n//go:generate gowrap gen -g -p . -i HistoryManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/history_generated.go\n//go:generate gowrap gen -g -p . -i DomainManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/domain_generated.go\n//go:generate gowrap gen -g -p . -i QueueManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/queue_generated.go\n\n// execution metered wrapper is special\n//go:generate gowrap gen -g -p . -i ExecutionManager -t ./wrappers/templates/metered_execution.tmpl -o wrappers/metered/execution_generated.go\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst DBTimestampMinPrecision = time.Millisecond\n\n// Domain status\nconst (\n\tDomainStatusRegistered = iota\n\tDomainStatusDeprecated\n\tDomainStatusDeleted\n)\n\nconst (\n\t// EventStoreVersion is already deprecated, this is used for forward\n\t// compatibility (so that rollback is possible).\n\t// TODO we can remove it after fixing all the query templates and when\n\t// we decide the compatibility is no longer needed.\n\tEventStoreVersion = 2\n)\n\n// CreateWorkflowMode workflow creation mode\ntype CreateWorkflowMode int\n\n// QueueType is an enum that represents various queue types in persistence\ntype QueueType int\n\n// DomainReplicationQueueType queue types used in queue table\n// Use positive numbers for queue type\n// Negative numbers are reserved for DLQ\nconst (\n\tDomainReplicationQueueType QueueType = iota + 1\n)\n\n// Create Workflow Execution Mode\nconst (\n\t// CreateWorkflowModeBrandNew Fail if current record exists\n\t// Only applicable for CreateWorkflowExecution\n\tCreateWorkflowModeBrandNew CreateWorkflowMode = iota\n\t// CreateWorkflowModeWorkflowIDReuse Update current record only if workflow is closed\n\t// Only applicable for CreateWorkflowExecution\n\tCreateWorkflowModeWorkflowIDReuse\n\t// CreateWorkflowModeContinueAsNew Update current record only if workflow is open\n\t// Only applicable for UpdateWorkflowExecution\n\tCreateWorkflowModeContinueAsNew\n\t// CreateWorkflowModeZombie Do not update current record since workflow to\n\t// applicable for CreateWorkflowExecution, UpdateWorkflowExecution\n\tCreateWorkflowModeZombie\n)\n\n// UpdateWorkflowMode update mode\ntype UpdateWorkflowMode int\n\n// Update Workflow Execution Mode\nconst (\n\t// UpdateWorkflowModeUpdateCurrent Update workflow, including current record\n\t// NOTE: update on current record is a condition update\n\tUpdateWorkflowModeUpdateCurrent UpdateWorkflowMode = iota\n\t// UpdateWorkflowModeBypassCurrent Update workflow, without current record\n\t// NOTE: current record CANNOT point to the workflow to be updated\n\tUpdateWorkflowModeBypassCurrent\n\t// UpdateWorkflowModeIgnoreCurrent Update workflow, ignoring current record\n\t// NOTE: current record may or may not point to the workflow\n\t// this mode should only be used for (re-)generating workflow tasks\n\t// and there's no other changes to the workflow\n\tUpdateWorkflowModeIgnoreCurrent\n)\n\n// ConflictResolveWorkflowMode conflict resolve mode\ntype ConflictResolveWorkflowMode int\n\n// Conflict Resolve Workflow Mode\nconst (\n\t// ConflictResolveWorkflowModeUpdateCurrent Conflict resolve workflow, including current record\n\t// NOTE: update on current record is a condition update\n\tConflictResolveWorkflowModeUpdateCurrent ConflictResolveWorkflowMode = iota\n\t// ConflictResolveWorkflowModeBypassCurrent Conflict resolve workflow, without current record\n\t// NOTE: current record CANNOT point to the workflow to be updated\n\tConflictResolveWorkflowModeBypassCurrent\n)\n\n// Workflow execution states\nconst (\n\tWorkflowStateCreated = iota\n\tWorkflowStateRunning\n\tWorkflowStateCompleted\n\tWorkflowStateZombie\n\tWorkflowStateVoid\n\tWorkflowStateCorrupted\n)\n\n// Workflow execution close status\nconst (\n\tWorkflowCloseStatusNone = iota\n\tWorkflowCloseStatusCompleted\n\tWorkflowCloseStatusFailed\n\tWorkflowCloseStatusCanceled\n\tWorkflowCloseStatusTerminated\n\tWorkflowCloseStatusContinuedAsNew\n\tWorkflowCloseStatusTimedOut\n)\n\n// Types of task lists\nconst (\n\tTaskListTypeDecision = iota\n\tTaskListTypeActivity\n)\n\n// Kinds of task lists\nconst (\n\tTaskListKindNormal = iota\n\tTaskListKindSticky\n\tTaskListKindEphemeral\n)\n\n// HistoryTaskCategory represents various categories of history tasks\ntype HistoryTaskCategory struct {\n\tcategoryType int\n\tcategoryID   int\n\tcategoryName string\n}\n\nfunc (c HistoryTaskCategory) Type() int {\n\treturn c.categoryType\n}\n\nfunc (c HistoryTaskCategory) ID() int {\n\treturn c.categoryID\n}\n\nfunc (c HistoryTaskCategory) Name() string {\n\treturn c.categoryName\n}\n\nconst (\n\tHistoryTaskCategoryTypeImmediate = iota + 1\n\tHistoryTaskCategoryTypeScheduled\n)\n\nconst (\n\tHistoryTaskCategoryIDTransfer    = 1\n\tHistoryTaskCategoryIDTimer       = 2\n\tHistoryTaskCategoryIDReplication = 3\n)\n\nvar (\n\tHistoryTaskCategoryTransfer = HistoryTaskCategory{\n\t\tcategoryType: HistoryTaskCategoryTypeImmediate,\n\t\tcategoryID:   HistoryTaskCategoryIDTransfer,\n\t\tcategoryName: \"transfer\",\n\t}\n\tHistoryTaskCategoryTimer = HistoryTaskCategory{\n\t\tcategoryType: HistoryTaskCategoryTypeScheduled,\n\t\tcategoryID:   HistoryTaskCategoryIDTimer,\n\t\tcategoryName: \"timer\",\n\t}\n\tHistoryTaskCategoryReplication = HistoryTaskCategory{\n\t\tcategoryType: HistoryTaskCategoryTypeImmediate,\n\t\tcategoryID:   HistoryTaskCategoryIDReplication,\n\t\tcategoryName: \"replication\",\n\t}\n)\n\n// Transfer task types\nconst (\n\tTransferTaskTypeDecisionTask = iota\n\tTransferTaskTypeActivityTask\n\tTransferTaskTypeCloseExecution\n\tTransferTaskTypeCancelExecution\n\tTransferTaskTypeStartChildExecution\n\tTransferTaskTypeSignalExecution\n\tTransferTaskTypeRecordWorkflowStarted\n\tTransferTaskTypeResetWorkflow\n\tTransferTaskTypeUpsertWorkflowSearchAttributes\n\tTransferTaskTypeRecordWorkflowClosed\n\tTransferTaskTypeRecordChildExecutionCompleted\n\tTransferTaskTypeApplyParentClosePolicy // Deprecated: this is related to cross-cluster tasks\n)\n\n// Types of replication tasks\nconst (\n\tReplicationTaskTypeHistory = iota\n\tReplicationTaskTypeSyncActivity\n\tReplicationTaskTypeFailoverMarker\n)\n\n// Types of timers\nconst (\n\tTaskTypeDecisionTimeout = iota\n\tTaskTypeActivityTimeout\n\tTaskTypeUserTimer\n\tTaskTypeWorkflowTimeout\n\tTaskTypeDeleteHistoryEvent\n\tTaskTypeActivityRetryTimer\n\tTaskTypeWorkflowBackoffTimer\n)\n\n// WorkflowRequestType is the type of workflow request\ntype WorkflowRequestType int\n\n// Types of workflow requests\nconst (\n\tWorkflowRequestTypeStart WorkflowRequestType = iota\n\tWorkflowRequestTypeSignal\n\tWorkflowRequestTypeCancel\n\tWorkflowRequestTypeReset\n)\n\nconst (\n\tDomainAuditOperationTypeInvalid DomainAuditOperationType = iota\n\tDomainAuditOperationTypeCreate\n\tDomainAuditOperationTypeUpdate\n\tDomainAuditOperationTypeDeprecate\n\tDomainAuditOperationTypeDelete\n\tDomainAuditOperationTypeFailover\n)\n\nfunc (d DomainAuditOperationType) String() string {\n\tswitch d {\n\tcase DomainAuditOperationTypeCreate:\n\t\treturn \"Create\"\n\tcase DomainAuditOperationTypeUpdate:\n\t\treturn \"Update\"\n\tcase DomainAuditOperationTypeFailover:\n\t\treturn \"Failover\"\n\tcase DomainAuditOperationTypeDeprecate:\n\t\treturn \"Deprecate\"\n\tcase DomainAuditOperationTypeDelete:\n\t\treturn \"Delete\"\n\tdefault:\n\t\treturn \"Invalid\"\n\t}\n}\n\n// CreateWorkflowRequestMode is the mode of create workflow request\ntype CreateWorkflowRequestMode int\n\n// Modes of create workflow request\nconst (\n\t// CreateWorkflowRequestModeNew Fail if data with the same domain_id, workflow_id, request_id exists\n\t// It is used for transactions started by external API requests\n\t// to allow us detecting duplicate requests\n\tCreateWorkflowRequestModeNew CreateWorkflowRequestMode = iota\n\t// CreateWorkflowRequestModeReplicated Upsert the data without checking duplication\n\t// It is used for transactions started by replication stack to achieve\n\t// eventual consistency\n\tCreateWorkflowRequestModeReplicated\n)\n\n// UnknownNumRowsAffected is returned when the number of rows that an API affected cannot be determined\nconst UnknownNumRowsAffected = -1\n\n// Types of workflow backoff timeout\nconst (\n\tWorkflowBackoffTimeoutTypeRetry = iota\n\tWorkflowBackoffTimeoutTypeCron\n)\n\nconst (\n\t// InitialFailoverNotificationVersion is the initial failover version for a domain\n\tInitialFailoverNotificationVersion int64 = 0\n\n\t// TransferTaskTransferTargetWorkflowID is the dummy workflow ID for transfer tasks of types\n\t// that do not have a target workflow\n\tTransferTaskTransferTargetWorkflowID = \"20000000-0000-f000-f000-000000000001\"\n\t// TransferTaskTransferTargetRunID is the dummy run ID for transfer tasks of types\n\t// that do not have a target workflow\n\tTransferTaskTransferTargetRunID = \"30000000-0000-f000-f000-000000000002\"\n\n\t// indicate invalid workflow state transition\n\tinvalidStateTransitionMsg = \"unable to change workflow state from %v to %v, close status %v\"\n)\n\nconst numItemsInGarbageInfo = 3\n\ntype ConfigType int\n\nconst (\n\tDynamicConfig ConfigType = iota\n\tGlobalIsolationGroupConfig\n)\n\ntype (\n\t// ShardInfo describes a shard\n\tShardInfo struct {\n\t\tShardID                       int                               `json:\"shard_id\"`\n\t\tOwner                         string                            `json:\"owner\"`\n\t\tRangeID                       int64                             `json:\"range_id\"`\n\t\tStolenSinceRenew              int                               `json:\"stolen_since_renew\"`\n\t\tUpdatedAt                     time.Time                         `json:\"updated_at\"`\n\t\tReplicationAckLevel           int64                             `json:\"replication_ack_level\"`\n\t\tReplicationDLQAckLevel        map[string]int64                  `json:\"replication_dlq_ack_level\"`\n\t\tTransferAckLevel              int64                             `json:\"transfer_ack_level\"`\n\t\tTimerAckLevel                 time.Time                         `json:\"timer_ack_level\"`\n\t\tClusterTransferAckLevel       map[string]int64                  `json:\"cluster_transfer_ack_level\"`\n\t\tClusterTimerAckLevel          map[string]time.Time              `json:\"cluster_timer_ack_level\"`\n\t\tTransferProcessingQueueStates *types.ProcessingQueueStates      `json:\"transfer_processing_queue_states\"`\n\t\tTimerProcessingQueueStates    *types.ProcessingQueueStates      `json:\"timer_processing_queue_states\"`\n\t\tClusterReplicationLevel       map[string]int64                  `json:\"cluster_replication_level\"`\n\t\tDomainNotificationVersion     int64                             `json:\"domain_notification_version\"`\n\t\tPendingFailoverMarkers        []*types.FailoverMarkerAttributes `json:\"pending_failover_markers\"`\n\t\tQueueStates                   map[int32]*types.QueueState       `json:\"queue_states\"`\n\t}\n\n\t// WorkflowExecutionInfo describes a workflow execution\n\tWorkflowExecutionInfo struct {\n\t\tDomainID                           string\n\t\tWorkflowID                         string\n\t\tRunID                              string\n\t\tFirstExecutionRunID                string\n\t\tParentDomainID                     string\n\t\tParentWorkflowID                   string\n\t\tParentRunID                        string\n\t\tInitiatedID                        int64\n\t\tCompletionEventBatchID             int64\n\t\tCompletionEvent                    *types.HistoryEvent\n\t\tTaskList                           string\n\t\tTaskListKind                       types.TaskListKind\n\t\tWorkflowTypeName                   string\n\t\tWorkflowTimeout                    int32\n\t\tDecisionStartToCloseTimeout        int32\n\t\tExecutionContext                   []byte\n\t\tState                              int\n\t\tCloseStatus                        int\n\t\tLastFirstEventID                   int64\n\t\tLastEventTaskID                    int64\n\t\tNextEventID                        int64\n\t\tLastProcessedEvent                 int64\n\t\tStartTimestamp                     time.Time\n\t\tLastUpdatedTimestamp               time.Time\n\t\tCreateRequestID                    string\n\t\tSignalCount                        int32\n\t\tDecisionVersion                    int64\n\t\tDecisionScheduleID                 int64\n\t\tDecisionStartedID                  int64\n\t\tDecisionRequestID                  string\n\t\tDecisionTimeout                    int32\n\t\tDecisionAttempt                    int64\n\t\tDecisionStartedTimestamp           int64\n\t\tDecisionScheduledTimestamp         int64\n\t\tDecisionOriginalScheduledTimestamp int64\n\t\tCancelRequested                    bool\n\t\tCancelRequestID                    string\n\t\tStickyTaskList                     string\n\t\tStickyScheduleToStartTimeout       int32\n\t\tClientLibraryVersion               string\n\t\tClientFeatureVersion               string\n\t\tClientImpl                         string\n\t\tAutoResetPoints                    *types.ResetPoints\n\t\tMemo                               map[string][]byte\n\t\tSearchAttributes                   map[string][]byte\n\t\tPartitionConfig                    map[string]string\n\t\tExecutionStatus                    types.WorkflowExecutionStatus\n\t\tScheduledExecutionTimestamp        int64 // unit is unix nano, used to record the actual execution timestamp if it's a cron workflow\n\t\t// for retry\n\t\tAttempt            int32\n\t\tHasRetryPolicy     bool\n\t\tInitialInterval    int32\n\t\tBackoffCoefficient float64\n\t\tMaximumInterval    int32\n\t\tExpirationTime     time.Time\n\t\tMaximumAttempts    int32\n\t\tNonRetriableErrors []string\n\t\tBranchToken        []byte\n\t\t// Cron\n\t\tIsCron            bool\n\t\tCronOverlapPolicy types.CronOverlapPolicy\n\t\tExpirationSeconds int32 // TODO: is this field useful?\n\t\tCronSchedule      string\n\n\t\tActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy\n\t}\n\n\t// ExecutionStats is the statistics about workflow execution\n\tExecutionStats struct {\n\t\tHistorySize int64\n\t}\n\n\t// ReplicationState represents mutable state information for global domains.\n\t// This information is used by replication protocol when applying events from remote clusters\n\t// TODO: remove this struct after all 2DC workflows complete\n\tReplicationState struct {\n\t\tCurrentVersion      int64\n\t\tStartVersion        int64\n\t\tLastWriteVersion    int64\n\t\tLastWriteEventID    int64\n\t\tLastReplicationInfo map[string]*ReplicationInfo\n\t}\n\n\t// CurrentWorkflowExecution describes a current execution record\n\tCurrentWorkflowExecution struct {\n\t\tDomainID     string\n\t\tWorkflowID   string\n\t\tRunID        string\n\t\tState        int\n\t\tCurrentRunID string\n\t}\n\n\t// FailoverLevel contains corresponding start / end level\n\tFailoverLevel struct {\n\t\tStartTime    time.Time\n\t\tMinLevel     HistoryTaskKey\n\t\tCurrentLevel HistoryTaskKey\n\t\tMaxLevel     HistoryTaskKey\n\t\tDomainIDs    map[string]struct{}\n\t}\n\n\t// TransferTaskInfo describes a transfer task\n\tTransferTaskInfo struct {\n\t\tDomainID                string\n\t\tWorkflowID              string\n\t\tRunID                   string\n\t\tVisibilityTimestamp     time.Time\n\t\tTaskID                  int64\n\t\tTargetDomainID          string\n\t\tTargetDomainIDs         map[string]struct{} // used for ApplyParentPolicy request\n\t\tTargetWorkflowID        string\n\t\tTargetRunID             string\n\t\tTargetChildWorkflowOnly bool\n\t\tTaskList                string\n\t\tTaskType                int\n\t\tScheduleID              int64\n\t\tVersion                 int64\n\t\tRecordVisibility        bool\n\t\tOriginalTaskList        string\n\t\tOriginalTaskListKind    types.TaskListKind\n\t}\n\n\t// CrossClusterTaskInfo describes a cross-cluster task\n\t// Cross cluster tasks are exactly like transfer tasks so\n\t// instead of creating another struct and duplicating the same\n\t// logic everywhere. We reuse TransferTaskInfo\n\t// This is a deprecated feature as of May 24\n\tCrossClusterTaskInfo = TransferTaskInfo\n\n\t// ReplicationTaskInfo describes the replication task created for replication of history events\n\tReplicationTaskInfo struct {\n\t\tDomainID          string\n\t\tWorkflowID        string\n\t\tRunID             string\n\t\tTaskID            int64\n\t\tTaskType          int\n\t\tFirstEventID      int64\n\t\tNextEventID       int64\n\t\tVersion           int64\n\t\tScheduledID       int64\n\t\tBranchToken       []byte\n\t\tNewRunBranchToken []byte\n\t\tCreationTime      int64\n\t}\n\n\t// TimerTaskInfo describes a timer task.\n\tTimerTaskInfo struct {\n\t\tDomainID            string\n\t\tWorkflowID          string\n\t\tRunID               string\n\t\tVisibilityTimestamp time.Time\n\t\tTaskID              int64\n\t\tTaskType            int\n\t\tTimeoutType         int\n\t\tEventID             int64\n\t\tScheduleAttempt     int64\n\t\tVersion             int64\n\t\tTaskList            string\n\t}\n\n\t// TaskListInfo describes a state of a task list implementation.\n\tTaskListInfo struct {\n\t\tDomainID                string\n\t\tName                    string\n\t\tTaskType                int\n\t\tRangeID                 int64\n\t\tAckLevel                int64\n\t\tKind                    int\n\t\tExpiry                  time.Time\n\t\tLastUpdated             time.Time\n\t\tAdaptivePartitionConfig *TaskListPartitionConfig\n\t}\n\n\tTaskListPartition struct {\n\t\tIsolationGroups []string\n\t}\n\n\t// TaskListPartitionConfig represents the configuration for task list partitions.\n\tTaskListPartitionConfig struct {\n\t\tVersion         int64\n\t\tReadPartitions  map[int]*TaskListPartition\n\t\tWritePartitions map[int]*TaskListPartition\n\t}\n\n\t// TaskInfo describes either activity or decision task\n\tTaskInfo struct {\n\t\tDomainID                      string\n\t\tWorkflowID                    string\n\t\tRunID                         string\n\t\tTaskID                        int64\n\t\tScheduleID                    int64\n\t\tScheduleToStartTimeoutSeconds int32\n\t\tExpiry                        time.Time\n\t\tCreatedTime                   time.Time\n\t\tPartitionConfig               map[string]string\n\t}\n\n\t// TaskKey gives primary key info for a specific task\n\tTaskKey struct {\n\t\tDomainID     string\n\t\tTaskListName string\n\t\tTaskType     int\n\t\tTaskID       int64\n\t}\n\n\t// ReplicationInfo represents the information stored for last replication event details per cluster\n\tReplicationInfo struct {\n\t\tVersion     int64\n\t\tLastEventID int64\n\t}\n\n\t// VersionHistoryItem contains the event id and the associated version\n\tVersionHistoryItem struct {\n\t\tEventID int64\n\t\tVersion int64\n\t}\n\n\t// VersionHistory provides operations on version history\n\tVersionHistory struct {\n\t\tBranchToken []byte\n\t\tItems       []*VersionHistoryItem\n\t}\n\n\t// VersionHistories contains a set of VersionHistory\n\tVersionHistories struct {\n\t\tCurrentVersionHistoryIndex int\n\t\tHistories                  []*VersionHistory\n\t}\n\n\t// WorkflowMutableState indicates workflow related state\n\tWorkflowMutableState struct {\n\t\tActivityInfos       map[int64]*ActivityInfo\n\t\tTimerInfos          map[string]*TimerInfo\n\t\tChildExecutionInfos map[int64]*ChildExecutionInfo\n\t\tRequestCancelInfos  map[int64]*RequestCancelInfo\n\t\tSignalInfos         map[int64]*SignalInfo\n\t\tSignalRequestedIDs  map[string]struct{}\n\t\tExecutionInfo       *WorkflowExecutionInfo\n\t\tExecutionStats      *ExecutionStats\n\t\tBufferedEvents      []*types.HistoryEvent\n\t\tVersionHistories    *VersionHistories\n\t\tReplicationState    *ReplicationState // TODO: remove this after all 2DC workflows complete\n\t\tChecksum            checksum.Checksum\n\t}\n\n\t// ActivityInfo details.\n\tActivityInfo struct {\n\t\tVersion                  int64\n\t\tScheduleID               int64\n\t\tScheduledEventBatchID    int64\n\t\tScheduledEvent           *types.HistoryEvent\n\t\tScheduledTime            time.Time\n\t\tStartedID                int64\n\t\tStartedEvent             *types.HistoryEvent\n\t\tStartedTime              time.Time\n\t\tDomainID                 string\n\t\tActivityID               string\n\t\tRequestID                string\n\t\tDetails                  []byte\n\t\tScheduleToStartTimeout   int32\n\t\tScheduleToCloseTimeout   int32\n\t\tStartToCloseTimeout      int32\n\t\tHeartbeatTimeout         int32\n\t\tCancelRequested          bool\n\t\tCancelRequestID          int64\n\t\tLastHeartBeatUpdatedTime time.Time\n\t\tTimerTaskStatus          int32\n\t\t// For retry\n\t\tAttempt            int32\n\t\tStartedIdentity    string\n\t\tTaskList           string\n\t\tTaskListKind       types.TaskListKind\n\t\tHasRetryPolicy     bool\n\t\tInitialInterval    int32\n\t\tBackoffCoefficient float64\n\t\tMaximumInterval    int32\n\t\tExpirationTime     time.Time\n\t\tMaximumAttempts    int32\n\t\tNonRetriableErrors []string\n\t\tLastFailureReason  string\n\t\tLastWorkerIdentity string\n\t\tLastFailureDetails []byte\n\t\t// Not written to database - This is used only for deduping heartbeat timer creation\n\t\tLastHeartbeatTimeoutVisibilityInSeconds int64\n\t}\n\n\t// TimerInfo details - metadata about user timer info.\n\tTimerInfo struct {\n\t\tVersion    int64\n\t\tTimerID    string\n\t\tStartedID  int64\n\t\tExpiryTime time.Time\n\t\tTaskStatus int64\n\t}\n\n\t// ChildExecutionInfo has details for pending child executions.\n\tChildExecutionInfo struct {\n\t\tVersion               int64\n\t\tInitiatedID           int64\n\t\tInitiatedEventBatchID int64\n\t\tInitiatedEvent        *types.HistoryEvent\n\t\tStartedID             int64\n\t\tStartedWorkflowID     string\n\t\tStartedRunID          string\n\t\tStartedEvent          *types.HistoryEvent\n\t\tCreateRequestID       string\n\t\tDomainID              string\n\t\tDomainNameDEPRECATED  string // deprecated: please use DomainID field instead\n\t\tWorkflowTypeName      string\n\t\tParentClosePolicy     types.ParentClosePolicy\n\t}\n\n\t// RequestCancelInfo has details for pending external workflow cancellations\n\tRequestCancelInfo struct {\n\t\tVersion               int64\n\t\tInitiatedEventBatchID int64\n\t\tInitiatedID           int64\n\t\tCancelRequestID       string\n\t}\n\n\t// SignalInfo has details for pending external workflow signal\n\tSignalInfo struct {\n\t\tVersion               int64\n\t\tInitiatedEventBatchID int64\n\t\tInitiatedID           int64\n\t\tSignalRequestID       string\n\t\tSignalName            string\n\t\tInput                 []byte\n\t\tControl               []byte\n\t}\n\n\t// CreateShardRequest is used to create a shard in executions table\n\tCreateShardRequest struct {\n\t\tShardInfo *ShardInfo\n\t}\n\n\t// GetShardRequest is used to get shard information\n\tGetShardRequest struct {\n\t\tShardID int\n\t}\n\n\t// GetShardResponse is the response to GetShard\n\tGetShardResponse struct {\n\t\tShardInfo *ShardInfo\n\t}\n\n\t// UpdateShardRequest is used to update shard information\n\tUpdateShardRequest struct {\n\t\tShardInfo       *ShardInfo\n\t\tPreviousRangeID int64\n\t}\n\n\t// CreateWorkflowExecutionRequest is used to write a new workflow execution\n\tCreateWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode CreateWorkflowMode\n\n\t\tPreviousRunID            string\n\t\tPreviousLastWriteVersion int64\n\n\t\tNewWorkflowSnapshot WorkflowSnapshot\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\t\tDomainName          string\n\t}\n\n\t// CreateWorkflowExecutionResponse is the response to CreateWorkflowExecutionRequest\n\tCreateWorkflowExecutionResponse struct {\n\t\tMutableStateUpdateSessionStats *MutableStateUpdateSessionStats\n\t}\n\n\t// GetWorkflowExecutionRequest is used to retrieve the info of a workflow execution\n\tGetWorkflowExecutionRequest struct {\n\t\tDomainID   string\n\t\tExecution  types.WorkflowExecution\n\t\tDomainName string\n\t\tRangeID    int64\n\t}\n\n\t// GetWorkflowExecutionResponse is the response to GetWorkflowExecutionRequest\n\tGetWorkflowExecutionResponse struct {\n\t\tState             *WorkflowMutableState\n\t\tMutableStateStats *MutableStateStats\n\t}\n\n\t// GetCurrentExecutionRequest is used to retrieve the current RunId for an execution\n\tGetCurrentExecutionRequest struct {\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tDomainName string\n\t}\n\n\t// ListCurrentExecutionsRequest is request to ListCurrentExecutions\n\tListCurrentExecutionsRequest struct {\n\t\tPageSize  int\n\t\tPageToken []byte\n\t}\n\n\t// ListCurrentExecutionsResponse is the response to ListCurrentExecutionsRequest\n\tListCurrentExecutionsResponse struct {\n\t\tExecutions []*CurrentWorkflowExecution\n\t\tPageToken  []byte\n\t}\n\n\t// IsWorkflowExecutionExistsRequest is used to check if the concrete execution exists\n\tIsWorkflowExecutionExistsRequest struct {\n\t\tDomainID   string\n\t\tDomainName string\n\t\tWorkflowID string\n\t\tRunID      string\n\t}\n\n\t// ListConcreteExecutionsRequest is request to ListConcreteExecutions\n\tListConcreteExecutionsRequest struct {\n\t\tPageSize  int\n\t\tPageToken []byte\n\t}\n\n\t// ListConcreteExecutionsResponse is response to ListConcreteExecutions\n\tListConcreteExecutionsResponse struct {\n\t\tExecutions []*ListConcreteExecutionsEntity\n\t\tPageToken  []byte\n\t}\n\n\t// ListConcreteExecutionsEntity is a single entity in ListConcreteExecutionsResponse\n\tListConcreteExecutionsEntity struct {\n\t\tExecutionInfo    *WorkflowExecutionInfo\n\t\tVersionHistories *VersionHistories\n\t}\n\n\t// GetCurrentExecutionResponse is the response to GetCurrentExecution\n\tGetCurrentExecutionResponse struct {\n\t\tStartRequestID   string\n\t\tRunID            string\n\t\tState            int\n\t\tCloseStatus      int\n\t\tLastWriteVersion int64\n\t}\n\n\t// IsWorkflowExecutionExistsResponse is the response to IsWorkflowExecutionExists\n\tIsWorkflowExecutionExistsResponse struct {\n\t\tExists bool\n\t}\n\n\t// UpdateWorkflowExecutionRequest is used to update a workflow execution\n\tUpdateWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode UpdateWorkflowMode\n\n\t\tUpdateWorkflowMutation WorkflowMutation\n\n\t\tNewWorkflowSnapshot *WorkflowSnapshot\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\n\t\tEncoding constants.EncodingType // optional binary encoding type\n\n\t\tDomainName string\n\t}\n\n\t// ConflictResolveWorkflowExecutionRequest is used to reset workflow execution state for a single run\n\tConflictResolveWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode ConflictResolveWorkflowMode\n\n\t\t// workflow to be reset\n\t\tResetWorkflowSnapshot WorkflowSnapshot\n\n\t\t// maybe new workflow\n\t\tNewWorkflowSnapshot *WorkflowSnapshot\n\n\t\t// current workflow\n\t\tCurrentWorkflowMutation *WorkflowMutation\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\n\t\tEncoding constants.EncodingType // optional binary encoding type\n\n\t\tDomainName string\n\t}\n\n\t// WorkflowEvents is used as generic workflow history events transaction container\n\tWorkflowEvents struct {\n\t\tDomainID    string\n\t\tWorkflowID  string\n\t\tRunID       string\n\t\tBranchToken []byte\n\t\tEvents      []*types.HistoryEvent\n\t}\n\n\t// WorkflowRequest is used as requestID and it's corresponding failover version container\n\tWorkflowRequest struct {\n\t\tRequestID   string\n\t\tVersion     int64\n\t\tRequestType WorkflowRequestType\n\t}\n\n\t// WorkflowMutation is used as generic workflow execution state mutation\n\tWorkflowMutation struct {\n\t\tExecutionInfo    *WorkflowExecutionInfo\n\t\tExecutionStats   *ExecutionStats\n\t\tVersionHistories *VersionHistories\n\n\t\tUpsertActivityInfos       []*ActivityInfo\n\t\tDeleteActivityInfos       []int64\n\t\tUpsertTimerInfos          []*TimerInfo\n\t\tDeleteTimerInfos          []string\n\t\tUpsertChildExecutionInfos []*ChildExecutionInfo\n\t\tDeleteChildExecutionInfos []int64\n\t\tUpsertRequestCancelInfos  []*RequestCancelInfo\n\t\tDeleteRequestCancelInfos  []int64\n\t\tUpsertSignalInfos         []*SignalInfo\n\t\tDeleteSignalInfos         []int64\n\t\tUpsertSignalRequestedIDs  []string\n\t\tDeleteSignalRequestedIDs  []string\n\t\tNewBufferedEvents         []*types.HistoryEvent\n\t\tClearBufferedEvents       bool\n\n\t\tTasksByCategory map[HistoryTaskCategory][]Task\n\n\t\tWorkflowRequests []*WorkflowRequest\n\n\t\tCondition int64\n\t\tChecksum  checksum.Checksum\n\t}\n\n\t// WorkflowSnapshot is used as generic workflow execution state snapshot\n\tWorkflowSnapshot struct {\n\t\tExecutionInfo    *WorkflowExecutionInfo\n\t\tExecutionStats   *ExecutionStats\n\t\tVersionHistories *VersionHistories\n\n\t\tActivityInfos       []*ActivityInfo\n\t\tTimerInfos          []*TimerInfo\n\t\tChildExecutionInfos []*ChildExecutionInfo\n\t\tRequestCancelInfos  []*RequestCancelInfo\n\t\tSignalInfos         []*SignalInfo\n\t\tSignalRequestedIDs  []string\n\n\t\tTasksByCategory map[HistoryTaskCategory][]Task\n\n\t\tWorkflowRequests []*WorkflowRequest\n\n\t\tCondition int64\n\t\tChecksum  checksum.Checksum\n\t}\n\n\t// DeleteWorkflowExecutionRequest is used to delete a workflow execution\n\tDeleteWorkflowExecutionRequest struct {\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t\tDomainName string\n\t}\n\n\t// DeleteCurrentWorkflowExecutionRequest is used to delete the current workflow execution\n\tDeleteCurrentWorkflowExecutionRequest struct {\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t\tDomainName string\n\t}\n\n\t// PutReplicationTaskToDLQRequest is used to put a replication task to dlq\n\tPutReplicationTaskToDLQRequest struct {\n\t\tSourceClusterName string\n\t\tTaskInfo          *ReplicationTaskInfo\n\t\tDomainName        string\n\t}\n\n\t// GetReplicationTasksFromDLQRequest is used to get replication tasks from dlq\n\tGetReplicationTasksFromDLQRequest struct {\n\t\tSourceClusterName string\n\t\tReadLevel         int64\n\t\tMaxReadLevel      int64\n\t\tBatchSize         int\n\t\tNextPageToken     []byte\n\t}\n\n\t// GetReplicationDLQSizeRequest is used to get one replication task from dlq\n\tGetReplicationDLQSizeRequest struct {\n\t\tSourceClusterName string\n\t}\n\n\t// DeleteReplicationTaskFromDLQRequest is used to delete replication task from DLQ\n\tDeleteReplicationTaskFromDLQRequest struct {\n\t\tSourceClusterName string\n\t\tTaskID            int64\n\t}\n\n\t// RangeDeleteReplicationTaskFromDLQRequest is used to delete replication tasks from DLQ\n\tRangeDeleteReplicationTaskFromDLQRequest struct {\n\t\tSourceClusterName    string\n\t\tInclusiveBeginTaskID int64\n\t\tExclusiveEndTaskID   int64\n\t\tPageSize             int\n\t}\n\n\t// RangeDeleteReplicationTaskFromDLQResponse is the response of RangeDeleteReplicationTaskFromDLQ\n\tRangeDeleteReplicationTaskFromDLQResponse struct {\n\t\tTasksCompleted int\n\t}\n\n\t// GetReplicationDLQSizeResponse is the response for GetReplicationDLQSize\n\tGetReplicationDLQSizeResponse struct {\n\t\tSize int64\n\t}\n\n\t// GetHistoryTasksRequest is used to get history tasks\n\tGetHistoryTasksRequest struct {\n\t\tTaskCategory        HistoryTaskCategory\n\t\tInclusiveMinTaskKey HistoryTaskKey\n\t\tExclusiveMaxTaskKey HistoryTaskKey\n\t\tPageSize            int\n\t\tNextPageToken       []byte\n\t}\n\n\t// GetHistoryTasksResponse is the response for GetHistoryTasks\n\tGetHistoryTasksResponse struct {\n\t\tTasks         []Task\n\t\tNextPageToken []byte\n\t}\n\n\t// CompleteHistoryTaskRequest is used to complete a history task\n\tCompleteHistoryTaskRequest struct {\n\t\tTaskCategory HistoryTaskCategory\n\t\tTaskKey      HistoryTaskKey\n\t}\n\n\t// RangeCompleteHistoryTaskRequest is used to complete a range of history tasks\n\tRangeCompleteHistoryTaskRequest struct {\n\t\tTaskCategory        HistoryTaskCategory\n\t\tInclusiveMinTaskKey HistoryTaskKey\n\t\tExclusiveMaxTaskKey HistoryTaskKey\n\t\tPageSize            int\n\t}\n\n\tRangeCompleteHistoryTaskResponse struct {\n\t\tTasksCompleted int\n\t}\n\n\t// LeaseTaskListRequest is used to request lease of a task list\n\tLeaseTaskListRequest struct {\n\t\tDomainID         string\n\t\tDomainName       string\n\t\tTaskList         string\n\t\tTaskType         int\n\t\tTaskListKind     int\n\t\tRangeID          int64\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// LeaseTaskListResponse is response to LeaseTaskListRequest\n\tLeaseTaskListResponse struct {\n\t\tTaskListInfo *TaskListInfo\n\t}\n\n\tGetTaskListRequest struct {\n\t\tDomainID   string\n\t\tDomainName string\n\t\tTaskList   string\n\t\tTaskType   int\n\t}\n\n\tGetTaskListResponse struct {\n\t\tTaskListInfo *TaskListInfo\n\t}\n\n\t// UpdateTaskListRequest is used to update task list implementation information\n\tUpdateTaskListRequest struct {\n\t\tTaskListInfo     *TaskListInfo\n\t\tDomainName       string\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// UpdateTaskListResponse is the response to UpdateTaskList\n\tUpdateTaskListResponse struct {\n\t}\n\n\t// ListTaskListRequest contains the request params needed to invoke ListTaskList API\n\tListTaskListRequest struct {\n\t\tPageSize  int\n\t\tPageToken []byte\n\t}\n\n\t// ListTaskListResponse is the response from ListTaskList API\n\tListTaskListResponse struct {\n\t\tItems         []TaskListInfo\n\t\tNextPageToken []byte\n\t}\n\n\t// DeleteTaskListRequest contains the request params needed to invoke DeleteTaskList API\n\tDeleteTaskListRequest struct {\n\t\tDomainID     string\n\t\tDomainName   string\n\t\tTaskListName string\n\t\tTaskListType int\n\t\tRangeID      int64\n\t}\n\n\tGetTaskListSizeRequest struct {\n\t\tDomainID     string\n\t\tDomainName   string\n\t\tTaskListName string\n\t\tTaskListType int\n\t\tAckLevel     int64\n\t}\n\n\tGetTaskListSizeResponse struct {\n\t\tSize int64\n\t}\n\n\t// CreateTasksRequest is used to create a new task for a workflow execution\n\tCreateTasksRequest struct {\n\t\tTaskListInfo     *TaskListInfo\n\t\tTasks            []*CreateTaskInfo\n\t\tDomainName       string\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// CreateTaskInfo describes a task to be created in CreateTasksRequest\n\tCreateTaskInfo struct {\n\t\tData   *TaskInfo\n\t\tTaskID int64\n\t}\n\n\t// CreateTasksResponse is the response to CreateTasksRequest\n\tCreateTasksResponse struct {\n\t}\n\n\t// GetTasksRequest is used to retrieve tasks of a task list\n\tGetTasksRequest struct {\n\t\tDomainID     string\n\t\tTaskList     string\n\t\tTaskType     int\n\t\tReadLevel    int64  // range exclusive\n\t\tMaxReadLevel *int64 // optional: range inclusive when specified\n\t\tBatchSize    int\n\t\tDomainName   string\n\t}\n\n\t// GetTasksResponse is the response to GetTasksRequests\n\tGetTasksResponse struct {\n\t\tTasks []*TaskInfo\n\t}\n\n\t// CompleteTaskRequest is used to complete a task\n\tCompleteTaskRequest struct {\n\t\tTaskList   *TaskListInfo\n\t\tTaskID     int64\n\t\tDomainName string\n\t}\n\n\t// CompleteTasksLessThanRequest contains the request params needed to invoke CompleteTasksLessThan API\n\tCompleteTasksLessThanRequest struct {\n\t\tDomainID     string\n\t\tTaskListName string\n\t\tTaskType     int\n\t\tTaskID       int64 // Tasks less than or equal to this ID will be completed\n\t\tLimit        int   // Limit on the max number of tasks that can be completed. Required param\n\t\tDomainName   string\n\t}\n\n\t// CompleteTasksLessThanResponse is the response of CompleteTasksLessThan\n\tCompleteTasksLessThanResponse struct {\n\t\tTasksCompleted int\n\t}\n\n\t// GetOrphanTasksRequest contains the request params need to invoke the GetOrphanTasks API\n\tGetOrphanTasksRequest struct {\n\t\tLimit int\n\t}\n\n\t// GetOrphanTasksResponse is the response to GetOrphanTasksRequests\n\tGetOrphanTasksResponse struct {\n\t\tTasks []*TaskKey\n\t}\n\n\t// DomainInfo describes the domain entity\n\tDomainInfo struct {\n\t\tID          string\n\t\tName        string\n\t\tStatus      int\n\t\tDescription string\n\t\tOwnerEmail  string\n\t\tData        map[string]string\n\t}\n\n\t// DomainConfig describes the domain configuration\n\tDomainConfig struct {\n\t\t// NOTE: this retention is in days, not in seconds\n\t\tRetention                int32\n\t\tEmitMetric               bool\n\t\tHistoryArchivalStatus    types.ArchivalStatus\n\t\tHistoryArchivalURI       string\n\t\tVisibilityArchivalStatus types.ArchivalStatus\n\t\tVisibilityArchivalURI    string\n\t\tBadBinaries              types.BadBinaries\n\t\tIsolationGroups          types.IsolationGroupConfiguration\n\t\tAsyncWorkflowConfig      types.AsyncWorkflowConfiguration\n\t}\n\n\t// DomainReplicationConfig describes the cross DC domain replication configuration\n\tDomainReplicationConfig struct {\n\t\tClusters []*ClusterReplicationConfig\n\n\t\t// ActiveClusterName is the name of the cluster that the domain is active in.\n\t\t// Required for all global domains (both active-passive and active-active).\n\t\t// For active-passive domains, this is the single active cluster.\n\t\t// For active-active domains this is the default cluster whenever a ClusterAttribute is not provided\n\t\tActiveClusterName string\n\n\t\t// ActiveClusters is only applicable for active-active domains.\n\t\t// When this is set, the domain is considered active-active and workflows are routed\n\t\t// based on their ClusterAttributes.\n\t\tActiveClusters *types.ActiveClusters\n\t}\n\n\t// ClusterReplicationConfig describes the cross DC cluster replication configuration\n\tClusterReplicationConfig struct {\n\t\tClusterName string\n\t\t// Note: if adding new properties of non-primitive types, remember to update GetCopy()\n\t}\n\n\t// CreateDomainRequest is used to create the domain\n\tCreateDomainRequest struct {\n\t\tInfo              *DomainInfo\n\t\tConfig            *DomainConfig\n\t\tReplicationConfig *DomainReplicationConfig\n\t\tIsGlobalDomain    bool\n\t\tConfigVersion     int64\n\t\tFailoverVersion   int64\n\t\tLastUpdatedTime   int64\n\t\tCurrentTimeStamp  time.Time\n\t}\n\n\t// CreateDomainResponse is the response for CreateDomain\n\tCreateDomainResponse struct {\n\t\tID string\n\t}\n\n\t// GetDomainRequest is used to read domain\n\tGetDomainRequest struct {\n\t\tID   string\n\t\tName string\n\t}\n\n\t// GetDomainResponse is the response for GetDomain\n\tGetDomainResponse struct {\n\t\tInfo                        *DomainInfo\n\t\tConfig                      *DomainConfig\n\t\tReplicationConfig           *DomainReplicationConfig\n\t\tIsGlobalDomain              bool\n\t\tConfigVersion               int64\n\t\tFailoverVersion             int64\n\t\tFailoverNotificationVersion int64\n\t\tPreviousFailoverVersion     int64\n\t\tFailoverEndTime             *int64\n\t\tLastUpdatedTime             int64\n\t\tNotificationVersion         int64\n\t}\n\n\t// UpdateDomainRequest is used to update domain\n\tUpdateDomainRequest struct {\n\t\tInfo                        *DomainInfo\n\t\tConfig                      *DomainConfig\n\t\tReplicationConfig           *DomainReplicationConfig\n\t\tConfigVersion               int64\n\t\tFailoverVersion             int64\n\t\tFailoverNotificationVersion int64\n\t\tPreviousFailoverVersion     int64\n\t\tFailoverEndTime             *int64\n\t\tLastUpdatedTime             int64\n\t\tNotificationVersion         int64\n\t}\n\n\t// DeleteDomainRequest is used to delete domain entry from domains table\n\tDeleteDomainRequest struct {\n\t\tID string\n\t}\n\n\t// DeleteDomainByNameRequest is used to delete domain entry from domains_by_name table\n\tDeleteDomainByNameRequest struct {\n\t\tName string\n\t}\n\n\t// ListDomainsRequest is used to list domains\n\tListDomainsRequest struct {\n\t\tPageSize      int\n\t\tNextPageToken []byte\n\t}\n\n\t// ListDomainsResponse is the response for GetDomain\n\tListDomainsResponse struct {\n\t\tDomains       []*GetDomainResponse\n\t\tNextPageToken []byte\n\t}\n\n\t// GetMetadataResponse is the response for GetMetadata\n\tGetMetadataResponse struct {\n\t\tNotificationVersion int64\n\t}\n\n\t// CreateDomainAuditLogRequest is used to create a domain audit log entry\n\tCreateDomainAuditLogRequest struct {\n\t\tDomainID      string\n\t\tEventID       string // must be a UUID v7\n\t\tStateBefore   *GetDomainResponse\n\t\tStateAfter    *GetDomainResponse\n\t\tOperationType DomainAuditOperationType\n\t\tCreatedTime   time.Time\n\t\tIdentity      string\n\t\tIdentityType  string\n\t\tComment       string\n\t}\n\n\t// CreateDomainAuditLogResponse is the response for CreateDomainAuditLog\n\tCreateDomainAuditLogResponse struct {\n\t\tEventID string\n\t}\n\n\t// GetDomainAuditLogsRequest is used to get domain audit logs\n\tGetDomainAuditLogsRequest struct {\n\t\tDomainID       string\n\t\tOperationType  DomainAuditOperationType\n\t\tMinCreatedTime *time.Time\n\t\tMaxCreatedTime *time.Time\n\t\tPageSize       int\n\t\tNextPageToken  []byte\n\t}\n\n\t// GetDomainAuditLogsResponse is the response for GetDomainAuditLogs\n\tGetDomainAuditLogsResponse struct {\n\t\tAuditLogs     []*DomainAuditLog\n\t\tNextPageToken []byte\n\t}\n\n\t// DomainAuditLog represents a single domain audit log entry\n\tDomainAuditLog struct {\n\t\tEventID         string\n\t\tDomainID        string\n\t\tStateBefore     *GetDomainResponse\n\t\tStateAfter      *GetDomainResponse\n\t\tOperationType   DomainAuditOperationType\n\t\tCreatedTime     time.Time\n\t\tLastUpdatedTime time.Time\n\t\tIdentity        string\n\t\tIdentityType    string\n\t\tComment         string\n\t}\n\n\t// MutableStateStats is the size stats for MutableState\n\tMutableStateStats struct {\n\t\t// Total size of mutable state\n\t\tMutableStateSize int\n\n\t\t// Breakdown of size into more granular stats\n\t\tExecutionInfoSize  int\n\t\tActivityInfoSize   int\n\t\tTimerInfoSize      int\n\t\tChildInfoSize      int\n\t\tSignalInfoSize     int\n\t\tBufferedEventsSize int\n\n\t\t// Item count for various information captured within mutable state\n\t\tActivityInfoCount      int\n\t\tTimerInfoCount         int\n\t\tChildInfoCount         int\n\t\tSignalInfoCount        int\n\t\tRequestCancelInfoCount int\n\t\tBufferedEventsCount    int\n\t}\n\n\t// MutableStateUpdateSessionStats is size stats for mutableState updating session\n\tMutableStateUpdateSessionStats struct {\n\t\tMutableStateSize int // Total size of mutable state update\n\n\t\t// Breakdown of mutable state size update for more granular stats\n\t\tExecutionInfoSize  int\n\t\tActivityInfoSize   int\n\t\tTimerInfoSize      int\n\t\tChildInfoSize      int\n\t\tSignalInfoSize     int\n\t\tBufferedEventsSize int\n\n\t\t// Item counts in this session update\n\t\tActivityInfoCount      int\n\t\tTimerInfoCount         int\n\t\tChildInfoCount         int\n\t\tSignalInfoCount        int\n\t\tRequestCancelInfoCount int\n\n\t\t// Deleted item counts in this session update\n\t\tDeleteActivityInfoCount      int\n\t\tDeleteTimerInfoCount         int\n\t\tDeleteChildInfoCount         int\n\t\tDeleteSignalInfoCount        int\n\t\tDeleteRequestCancelInfoCount int\n\n\t\tTaskCountByCategory map[HistoryTaskCategory]int\n\t}\n\n\t// UpdateWorkflowExecutionResponse is response for UpdateWorkflowExecutionRequest\n\tUpdateWorkflowExecutionResponse struct {\n\t\tMutableStateUpdateSessionStats *MutableStateUpdateSessionStats\n\t}\n\n\t// ConflictResolveWorkflowExecutionResponse is response for ConflictResolveWorkflowExecutionRequest\n\tConflictResolveWorkflowExecutionResponse struct {\n\t\tMutableStateUpdateSessionStats *MutableStateUpdateSessionStats\n\t}\n\n\t// AppendHistoryNodesRequest is used to append a batch of history nodes\n\tAppendHistoryNodesRequest struct {\n\t\t// true if this is the first append request to the branch\n\t\tIsNewBranch bool\n\t\t// the info for clean up data in background\n\t\tInfo string\n\t\t// The branch to be appended\n\t\tBranchToken []byte\n\t\t// The batch of events to be appended. The first eventID will become the nodeID of this batch\n\t\tEvents []*types.HistoryEvent\n\t\t// requested TransactionID for this write operation. For the same eventID, the node with larger TransactionID always wins\n\t\tTransactionID int64\n\t\t// optional binary encoding type\n\t\tEncoding constants.EncodingType\n\t\t// The shard to get history node data\n\t\tShardID *int\n\n\t\t// DomainName to get metrics created with the domain\n\t\tDomainName string\n\t}\n\n\t// AppendHistoryNodesResponse is a response to AppendHistoryNodesRequest\n\tAppendHistoryNodesResponse struct {\n\t\t// The data blob that was persisted to database\n\t\tDataBlob DataBlob\n\t}\n\n\t// ReadHistoryBranchRequest is used to read a history branch\n\tReadHistoryBranchRequest struct {\n\t\t// The branch to be read\n\t\tBranchToken []byte\n\t\t// Get the history nodes from MinEventID. Inclusive.\n\t\tMinEventID int64\n\t\t// Get the history nodes upto MaxEventID.  Exclusive.\n\t\tMaxEventID int64\n\t\t// Maximum number of batches of events per page. Not that number of events in a batch >=1, it is not number of events per page.\n\t\t// However for a single page, it is also possible that the returned events is less than PageSize (event zero events) due to stale events.\n\t\tPageSize int\n\t\t// Token to continue reading next page of history append transactions.  Pass in empty slice for first page\n\t\tNextPageToken []byte\n\t\t// The shard to get history branch data\n\t\tShardID *int\n\n\t\tDomainName string\n\t}\n\n\t// ReadHistoryBranchResponse is the response to ReadHistoryBranchRequest\n\tReadHistoryBranchResponse struct {\n\t\t// History events\n\t\tHistoryEvents []*types.HistoryEvent\n\t\t// Token to read next page if there are more events beyond page size.\n\t\t// Use this to set NextPageToken on ReadHistoryBranchRequest to read the next page.\n\t\t// Empty means we have reached the last page, not need to continue\n\t\tNextPageToken []byte\n\t\t// Size of history read from store\n\t\tSize int\n\t\t// the first_event_id of last loaded batch\n\t\tLastFirstEventID int64\n\t}\n\n\t// ReadHistoryBranchByBatchResponse is the response to ReadHistoryBranchRequest\n\tReadHistoryBranchByBatchResponse struct {\n\t\t// History events by batch\n\t\tHistory []*types.History\n\t\t// Token to read next page if there are more events beyond page size.\n\t\t// Use this to set NextPageToken on ReadHistoryBranchRequest to read the next page.\n\t\t// Empty means we have reached the last page, not need to continue\n\t\tNextPageToken []byte\n\t\t// Size of history read from store\n\t\tSize int\n\t\t// the first_event_id of last loaded batch\n\t\tLastFirstEventID int64\n\t}\n\n\t// ReadRawHistoryBranchResponse is the response to ReadHistoryBranchRequest\n\tReadRawHistoryBranchResponse struct {\n\t\t// HistoryEventBlobs history event blobs\n\t\tHistoryEventBlobs []*DataBlob\n\t\t// Token to read next page if there are more events beyond page size.\n\t\t// Use this to set NextPageToken on ReadHistoryBranchRequest to read the next page.\n\t\t// Empty means we have reached the last page, not need to continue\n\t\tNextPageToken []byte\n\t\t// Size of history read from store\n\t\tSize int\n\t}\n\n\t// ForkHistoryBranchRequest is used to fork a history branch\n\tForkHistoryBranchRequest struct {\n\t\t// The base branch to fork from\n\t\tForkBranchToken []byte\n\t\t// The nodeID to fork from, the new branch will start from ( inclusive ), the base branch will stop at(exclusive)\n\t\t// Application must provide a void forking nodeID, it must be a valid nodeID in that branch. A valid nodeID is the firstEventID of a valid batch of events.\n\t\t// And ForkNodeID > 1 because forking from 1 doesn't make any sense.\n\t\tForkNodeID int64\n\t\t// the info for clean up data in background\n\t\tInfo string\n\t\t// The shard to get history branch data\n\t\tShardID *int\n\t\t// DomainName to create metrics for Domain Cost Attribution\n\t\tDomainName string\n\t}\n\n\t// ForkHistoryBranchResponse is the response to ForkHistoryBranchRequest\n\tForkHistoryBranchResponse struct {\n\t\t// branchToken to represent the new branch\n\t\tNewBranchToken []byte\n\t}\n\n\t// CompleteForkBranchRequest is used to complete forking\n\tCompleteForkBranchRequest struct {\n\t\t// the new branch returned from ForkHistoryBranchRequest\n\t\tBranchToken []byte\n\t\t// true means the fork is success, will update the flag, otherwise will delete the new branch\n\t\tSuccess bool\n\t\t// The shard to update history branch data\n\t\tShardID *int\n\t}\n\n\t// DeleteHistoryBranchRequest is used to remove a history branch\n\tDeleteHistoryBranchRequest struct {\n\t\t// branch to be deleted\n\t\tBranchToken []byte\n\t\t// The shard to delete history branch data\n\t\tShardID *int\n\t\t// DomainName to generate metrics for Domain Cost Attribution\n\t\tDomainName string\n\t}\n\n\t// GetHistoryTreeRequest is used to retrieve branch info of a history tree\n\tGetHistoryTreeRequest struct {\n\t\t// A UUID of a tree\n\t\tTreeID string\n\t\t// Get data from this shard\n\t\tShardID *int\n\t\t// optional: can provide treeID via branchToken if treeID is empty\n\t\tBranchToken []byte\n\t\t// DomainName to create metrics\n\t\tDomainName string\n\t}\n\n\t// HistoryBranchDetail contains detailed information of a branch\n\tHistoryBranchDetail struct {\n\t\tTreeID   string\n\t\tBranchID string\n\t\tForkTime time.Time\n\t\tInfo     string\n\t}\n\n\t// GetHistoryTreeResponse is a response to GetHistoryTreeRequest\n\tGetHistoryTreeResponse struct {\n\t\t// all branches of a tree\n\t\tBranches []*workflow.HistoryBranch\n\t}\n\n\t// GetAllHistoryTreeBranchesRequest is a request of GetAllHistoryTreeBranches\n\tGetAllHistoryTreeBranchesRequest struct {\n\t\t// pagination token\n\t\tNextPageToken []byte\n\t\t// maximum number of branches returned per page\n\t\tPageSize int\n\t}\n\n\t// GetAllHistoryTreeBranchesResponse is a response to GetAllHistoryTreeBranches\n\tGetAllHistoryTreeBranchesResponse struct {\n\t\t// pagination token\n\t\tNextPageToken []byte\n\t\t// all branches of all trees\n\t\tBranches []HistoryBranchDetail\n\t}\n\n\t// CreateFailoverMarkersRequest is request to create failover markers\n\tCreateFailoverMarkersRequest struct {\n\t\tRangeID          int64\n\t\tMarkers          []*FailoverMarkerTask\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// FetchDynamicConfigResponse is a response to FetchDynamicConfigResponse\n\tFetchDynamicConfigResponse struct {\n\t\tSnapshot *DynamicConfigSnapshot\n\t}\n\n\t// UpdateDynamicConfigRequest is a request to update dynamic config with snapshot\n\tUpdateDynamicConfigRequest struct {\n\t\tSnapshot *DynamicConfigSnapshot\n\t}\n\n\tDynamicConfigSnapshot struct {\n\t\tVersion int64\n\t\tValues  *types.DynamicConfigBlob\n\t}\n\n\t// Closeable is an interface for any entity that supports a close operation to release resources\n\tCloseable interface {\n\t\tClose()\n\t}\n\n\t// ShardManager is used to manage all shards\n\tShardManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateShard(ctx context.Context, request *CreateShardRequest) error\n\t\tGetShard(ctx context.Context, request *GetShardRequest) (*GetShardResponse, error)\n\t\tUpdateShard(ctx context.Context, request *UpdateShardRequest) error\n\t}\n\n\t// ExecutionManager is used to manage workflow executions\n\tExecutionManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tGetShardID() int\n\n\t\tCreateWorkflowExecution(ctx context.Context, request *CreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error)\n\t\tGetWorkflowExecution(ctx context.Context, request *GetWorkflowExecutionRequest) (*GetWorkflowExecutionResponse, error)\n\t\tUpdateWorkflowExecution(ctx context.Context, request *UpdateWorkflowExecutionRequest) (*UpdateWorkflowExecutionResponse, error)\n\t\tConflictResolveWorkflowExecution(ctx context.Context, request *ConflictResolveWorkflowExecutionRequest) (*ConflictResolveWorkflowExecutionResponse, error)\n\t\tDeleteWorkflowExecution(ctx context.Context, request *DeleteWorkflowExecutionRequest) error\n\t\tDeleteCurrentWorkflowExecution(ctx context.Context, request *DeleteCurrentWorkflowExecutionRequest) error\n\t\tGetCurrentExecution(ctx context.Context, request *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error)\n\t\tIsWorkflowExecutionExists(ctx context.Context, request *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error)\n\n\t\t// Replication task related methods\n\n\t\tPutReplicationTaskToDLQ(ctx context.Context, request *PutReplicationTaskToDLQRequest) error\n\t\tGetReplicationTasksFromDLQ(ctx context.Context, request *GetReplicationTasksFromDLQRequest) (*GetHistoryTasksResponse, error)\n\t\tGetReplicationDLQSize(ctx context.Context, request *GetReplicationDLQSizeRequest) (*GetReplicationDLQSizeResponse, error)\n\t\tDeleteReplicationTaskFromDLQ(ctx context.Context, request *DeleteReplicationTaskFromDLQRequest) error\n\t\tRangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *RangeDeleteReplicationTaskFromDLQRequest) (*RangeDeleteReplicationTaskFromDLQResponse, error)\n\t\tCreateFailoverMarkerTasks(ctx context.Context, request *CreateFailoverMarkersRequest) error\n\n\t\tGetHistoryTasks(ctx context.Context, request *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error)\n\t\tCompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error\n\t\tRangeCompleteHistoryTask(ctx context.Context, request *RangeCompleteHistoryTaskRequest) (*RangeCompleteHistoryTaskResponse, error)\n\n\t\t// Scan operations\n\n\t\tListConcreteExecutions(ctx context.Context, request *ListConcreteExecutionsRequest) (*ListConcreteExecutionsResponse, error)\n\t\tListCurrentExecutions(ctx context.Context, request *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error)\n\n\t\tGetActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error)\n\t\tDeleteActiveClusterSelectionPolicy(ctx context.Context, domainID, workflowID, runID string) error\n\t}\n\n\t// ExecutionManagerFactory creates an instance of ExecutionManager for a given shard\n\tExecutionManagerFactory interface {\n\t\tCloseable\n\t\tNewExecutionManager(shardID int) (ExecutionManager, error)\n\t}\n\n\t// TaskManager is used to manage tasks\n\tTaskManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tLeaseTaskList(ctx context.Context, request *LeaseTaskListRequest) (*LeaseTaskListResponse, error)\n\t\tUpdateTaskList(ctx context.Context, request *UpdateTaskListRequest) (*UpdateTaskListResponse, error)\n\t\tGetTaskList(ctx context.Context, request *GetTaskListRequest) (*GetTaskListResponse, error)\n\t\tListTaskList(ctx context.Context, request *ListTaskListRequest) (*ListTaskListResponse, error)\n\t\tDeleteTaskList(ctx context.Context, request *DeleteTaskListRequest) error\n\t\tGetTaskListSize(ctx context.Context, request *GetTaskListSizeRequest) (*GetTaskListSizeResponse, error)\n\t\tCreateTasks(ctx context.Context, request *CreateTasksRequest) (*CreateTasksResponse, error)\n\t\tGetTasks(ctx context.Context, request *GetTasksRequest) (*GetTasksResponse, error)\n\t\tCompleteTask(ctx context.Context, request *CompleteTaskRequest) error\n\t\tCompleteTasksLessThan(ctx context.Context, request *CompleteTasksLessThanRequest) (*CompleteTasksLessThanResponse, error)\n\t\tGetOrphanTasks(ctx context.Context, request *GetOrphanTasksRequest) (*GetOrphanTasksResponse, error)\n\t}\n\n\t// HistoryManager is used to manager workflow history events\n\tHistoryManager interface {\n\t\tCloseable\n\t\tGetName() string\n\n\t\t// The below are history V2 APIs\n\t\t// V2 regards history events growing as a tree, decoupled from workflow concepts\n\t\t// For Cadence, treeID is new runID, except for fork(reset), treeID will be the runID that it forks from.\n\n\t\t// AppendHistoryNodes add(or override) a batch of nodes to a history branch\n\t\tAppendHistoryNodes(ctx context.Context, request *AppendHistoryNodesRequest) (*AppendHistoryNodesResponse, error)\n\t\t// ReadHistoryBranch returns history node data for a branch\n\t\tReadHistoryBranch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadHistoryBranchResponse, error)\n\t\t// ReadHistoryBranchByBatch returns history node data for a branch ByBatch\n\t\tReadHistoryBranchByBatch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadHistoryBranchByBatchResponse, error)\n\t\t// ReadRawHistoryBranch returns history node raw data for a branch ByBatch\n\t\t// NOTE: this API should only be used by 3+DC\n\t\tReadRawHistoryBranch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadRawHistoryBranchResponse, error)\n\t\t// ForkHistoryBranch forks a new branch from an old branch\n\t\tForkHistoryBranch(ctx context.Context, request *ForkHistoryBranchRequest) (*ForkHistoryBranchResponse, error)\n\t\t// DeleteHistoryBranch removes a branch\n\t\t// If this is the last branch to delete, it will also remove the root node\n\t\tDeleteHistoryBranch(ctx context.Context, request *DeleteHistoryBranchRequest) error\n\t\t// GetHistoryTree returns all branch information of a tree\n\t\tGetHistoryTree(ctx context.Context, request *GetHistoryTreeRequest) (*GetHistoryTreeResponse, error)\n\t\t// GetAllHistoryTreeBranches returns all branches of all trees\n\t\tGetAllHistoryTreeBranches(ctx context.Context, request *GetAllHistoryTreeBranchesRequest) (*GetAllHistoryTreeBranchesResponse, error)\n\t}\n\n\t// DomainManager is used to manage metadata CRUD for domain entities\n\tDomainManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateDomain(ctx context.Context, request *CreateDomainRequest) (*CreateDomainResponse, error)\n\t\tGetDomain(ctx context.Context, request *GetDomainRequest) (*GetDomainResponse, error)\n\t\tUpdateDomain(ctx context.Context, request *UpdateDomainRequest) error\n\t\tDeleteDomain(ctx context.Context, request *DeleteDomainRequest) error\n\t\tDeleteDomainByName(ctx context.Context, request *DeleteDomainByNameRequest) error\n\t\tListDomains(ctx context.Context, request *ListDomainsRequest) (*ListDomainsResponse, error)\n\t\tGetMetadata(ctx context.Context) (*GetMetadataResponse, error)\n\t}\n\n\t// DomainAuditManager is used to manage domain audit logs\n\tDomainAuditManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateDomainAuditLog(ctx context.Context, request *CreateDomainAuditLogRequest) (*CreateDomainAuditLogResponse, error)\n\t\tGetDomainAuditLogs(ctx context.Context, request *GetDomainAuditLogsRequest) (*GetDomainAuditLogsResponse, error)\n\t}\n\n\tEnqueueMessageRequest struct {\n\t\tMessagePayload []byte\n\t}\n\n\tReadMessagesRequest struct {\n\t\tLastMessageID int64\n\t\tMaxCount      int\n\t}\n\n\tReadMessagesResponse struct {\n\t\tMessages QueueMessageList\n\t}\n\n\tDeleteMessagesBeforeRequest struct {\n\t\tMessageID int64\n\t}\n\n\tUpdateAckLevelRequest struct {\n\t\tMessageID   int64\n\t\tClusterName string\n\t}\n\n\tGetAckLevelsRequest struct{}\n\n\tGetAckLevelsResponse struct {\n\t\tAckLevels map[string]int64\n\t}\n\n\tEnqueueMessageToDLQRequest struct {\n\t\tMessagePayload []byte\n\t}\n\n\tReadMessagesFromDLQRequest struct {\n\t\tFirstMessageID int64\n\t\tLastMessageID  int64\n\t\tPageSize       int\n\t\tPageToken      []byte\n\t}\n\n\tReadMessagesFromDLQResponse struct {\n\t\tMessages      []*QueueMessage\n\t\tNextPageToken []byte\n\t}\n\n\tDeleteMessageFromDLQRequest struct {\n\t\tMessageID int64\n\t}\n\n\tRangeDeleteMessagesFromDLQRequest struct {\n\t\tFirstMessageID int64\n\t\tLastMessageID  int64\n\t}\n\n\tUpdateDLQAckLevelRequest struct {\n\t\tMessageID   int64\n\t\tClusterName string\n\t}\n\n\tGetDLQAckLevelsRequest struct{}\n\n\tGetDLQAckLevelsResponse struct {\n\t\tAckLevels map[string]int64\n\t}\n\n\tGetDLQSizeRequest struct{}\n\n\tGetDLQSizeResponse struct {\n\t\tSize int64\n\t}\n\n\t// QueueManager is used to manage queue store\n\tQueueManager interface {\n\t\tCloseable\n\t\tEnqueueMessage(ctx context.Context, request *EnqueueMessageRequest) error\n\t\tReadMessages(ctx context.Context, request *ReadMessagesRequest) (*ReadMessagesResponse, error)\n\t\tDeleteMessagesBefore(ctx context.Context, request *DeleteMessagesBeforeRequest) error\n\t\tUpdateAckLevel(ctx context.Context, request *UpdateAckLevelRequest) error\n\t\tGetAckLevels(ctx context.Context, request *GetAckLevelsRequest) (*GetAckLevelsResponse, error)\n\t\tEnqueueMessageToDLQ(ctx context.Context, request *EnqueueMessageToDLQRequest) error\n\t\tReadMessagesFromDLQ(ctx context.Context, request *ReadMessagesFromDLQRequest) (*ReadMessagesFromDLQResponse, error)\n\t\tDeleteMessageFromDLQ(ctx context.Context, request *DeleteMessageFromDLQRequest) error\n\t\tRangeDeleteMessagesFromDLQ(ctx context.Context, request *RangeDeleteMessagesFromDLQRequest) error\n\t\tUpdateDLQAckLevel(ctx context.Context, request *UpdateDLQAckLevelRequest) error\n\t\tGetDLQAckLevels(ctx context.Context, request *GetDLQAckLevelsRequest) (*GetDLQAckLevelsResponse, error)\n\t\tGetDLQSize(ctx context.Context, request *GetDLQSizeRequest) (*GetDLQSizeResponse, error)\n\t}\n\n\t// QueueMessage is the message that stores in the queue\n\tQueueMessage struct {\n\t\tID        int64     `json:\"message_id\"`\n\t\tQueueType QueueType `json:\"queue_type\"`\n\t\tPayload   []byte    `json:\"message_payload\"`\n\t}\n\n\tQueueMessageList []*QueueMessage\n\n\tConfigStoreManager interface {\n\t\tCloseable\n\t\tFetchDynamicConfig(ctx context.Context, cfgType ConfigType) (*FetchDynamicConfigResponse, error)\n\t\tUpdateDynamicConfig(ctx context.Context, request *UpdateDynamicConfigRequest, cfgType ConfigType) error\n\t\t// can add functions for config types other than dynamic config\n\t}\n)\n\n// IsTimeoutError check whether error is TimeoutError\nfunc IsTimeoutError(err error) bool {\n\tif errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) {\n\t\treturn true\n\t}\n\tvar timeoutError *TimeoutError\n\tok := errors.As(err, &timeoutError)\n\treturn ok\n}\n\n// GetTaskID returns the task ID for transfer task\nfunc (t *TransferTaskInfo) GetTaskID() int64 {\n\treturn t.TaskID\n}\n\n// GetVersion returns the task version for transfer task\nfunc (t *TransferTaskInfo) GetVersion() int64 {\n\treturn t.Version\n}\n\n// GetTaskType returns the task type for transfer task\nfunc (t *TransferTaskInfo) GetTaskType() int {\n\treturn t.TaskType\n}\n\n// GetVisibilityTimestamp returns the task type for transfer task\nfunc (t *TransferTaskInfo) GetVisibilityTimestamp() time.Time {\n\treturn t.VisibilityTimestamp\n}\n\n// GetWorkflowID returns the workflow ID for transfer task\nfunc (t *TransferTaskInfo) GetWorkflowID() string {\n\treturn t.WorkflowID\n}\n\n// GetRunID returns the run ID for transfer task\nfunc (t *TransferTaskInfo) GetRunID() string {\n\treturn t.RunID\n}\n\n// GetTargetDomainIDs returns the targetDomainIDs for applyParentPolicy\nfunc (t *TransferTaskInfo) GetTargetDomainIDs() map[string]struct{} {\n\treturn t.TargetDomainIDs\n}\n\n// GetDomainID returns the domain ID for transfer task\nfunc (t *TransferTaskInfo) GetDomainID() string {\n\treturn t.DomainID\n}\n\n// GetTaskList returns the task list for transfer task\nfunc (t *TransferTaskInfo) GetTaskList() string {\n\treturn t.TaskList\n}\n\n// GetOriginalTaskList returns the original task list for transfer task\nfunc (t *TransferTaskInfo) GetOriginalTaskList() string {\n\treturn t.OriginalTaskList\n}\n\n// GetOriginalTaskListKind returns the original task list kind for transfer task\nfunc (t *TransferTaskInfo) GetOriginalTaskListKind() types.TaskListKind {\n\treturn t.OriginalTaskListKind\n}\n\n// String returns a string representation for transfer task\nfunc (t *TransferTaskInfo) String() string {\n\treturn fmt.Sprintf(\"%#v\", t)\n}\n\nfunc (t *TransferTaskInfo) ToTask() (Task, error) {\n\tworkflowIdentifier := WorkflowIdentifier{\n\t\tDomainID:   t.DomainID,\n\t\tWorkflowID: t.WorkflowID,\n\t\tRunID:      t.RunID,\n\t}\n\ttaskData := TaskData{\n\t\tVersion:             t.Version,\n\t\tTaskID:              t.TaskID,\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t}\n\tswitch t.TaskType {\n\tcase TransferTaskTypeActivityTask:\n\t\treturn &ActivityTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     t.TargetDomainID,\n\t\t\tTaskList:           t.TaskList,\n\t\t\tScheduleID:         t.ScheduleID,\n\t\t}, nil\n\tcase TransferTaskTypeDecisionTask:\n\t\treturn &DecisionTask{\n\t\t\tWorkflowIdentifier:   workflowIdentifier,\n\t\t\tTaskData:             taskData,\n\t\t\tTargetDomainID:       t.TargetDomainID,\n\t\t\tTaskList:             t.TaskList,\n\t\t\tScheduleID:           t.ScheduleID,\n\t\t\tOriginalTaskList:     t.OriginalTaskList,\n\t\t\tOriginalTaskListKind: t.OriginalTaskListKind,\n\t\t}, nil\n\tcase TransferTaskTypeCloseExecution:\n\t\treturn &CloseExecutionTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeRecordWorkflowStarted:\n\t\treturn &RecordWorkflowStartedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeResetWorkflow:\n\t\treturn &ResetWorkflowTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeRecordWorkflowClosed:\n\t\treturn &RecordWorkflowClosedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeRecordChildExecutionCompleted:\n\t\ttargetRunID := t.TargetRunID\n\t\tif t.TargetRunID == TransferTaskTransferTargetRunID {\n\t\t\ttargetRunID = \"\"\n\t\t}\n\t\treturn &RecordChildExecutionCompletedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     t.TargetDomainID,\n\t\t\tTargetWorkflowID:   t.TargetWorkflowID,\n\t\t\tTargetRunID:        targetRunID,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeUpsertWorkflowSearchAttributes:\n\t\treturn &UpsertWorkflowSearchAttributesTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeStartChildExecution:\n\t\treturn &StartChildExecutionTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     t.TargetDomainID,\n\t\t\tTargetWorkflowID:   t.TargetWorkflowID,\n\t\t\tInitiatedID:        t.ScheduleID,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeCancelExecution:\n\t\ttargetRunID := t.TargetRunID\n\t\tif t.TargetRunID == TransferTaskTransferTargetRunID {\n\t\t\ttargetRunID = \"\"\n\t\t}\n\t\treturn &CancelExecutionTask{\n\t\t\tWorkflowIdentifier:      workflowIdentifier,\n\t\t\tTaskData:                taskData,\n\t\t\tTargetDomainID:          t.TargetDomainID,\n\t\t\tTargetWorkflowID:        t.TargetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tInitiatedID:             t.ScheduleID,\n\t\t\tTargetChildWorkflowOnly: t.TargetChildWorkflowOnly,\n\t\t\tTaskList:                t.TaskList,\n\t\t}, nil\n\tcase TransferTaskTypeSignalExecution:\n\t\ttargetRunID := t.TargetRunID\n\t\tif t.TargetRunID == TransferTaskTransferTargetRunID {\n\t\t\ttargetRunID = \"\"\n\t\t}\n\t\treturn &SignalExecutionTask{\n\t\t\tWorkflowIdentifier:      workflowIdentifier,\n\t\t\tTaskData:                taskData,\n\t\t\tTargetDomainID:          t.TargetDomainID,\n\t\t\tTargetWorkflowID:        t.TargetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tInitiatedID:             t.ScheduleID,\n\t\t\tTargetChildWorkflowOnly: t.TargetChildWorkflowOnly,\n\t\t\tTaskList:                t.TaskList,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown task type: %d\", t.TaskType)\n\t}\n}\n\nfunc (r *GetDomainResponse) GetFailoverVersion() int64 {\n\tif r == nil {\n\t\treturn types.UndefinedFailoverVersion\n\t}\n\treturn r.FailoverVersion\n}\n\n// GetTaskID returns the task ID for replication task\nfunc (t *ReplicationTaskInfo) GetTaskID() int64 {\n\treturn t.TaskID\n}\n\n// GetVersion returns the task version for replication task\nfunc (t *ReplicationTaskInfo) GetVersion() int64 {\n\treturn t.Version\n}\n\n// GetTaskType returns the task type for replication task\nfunc (t *ReplicationTaskInfo) GetTaskType() int {\n\treturn t.TaskType\n}\n\n// GetVisibilityTimestamp returns the task type for replication task\nfunc (t *ReplicationTaskInfo) GetVisibilityTimestamp() time.Time {\n\treturn time.Time{}\n}\n\n// GetWorkflowID returns the workflow ID for replication task\nfunc (t *ReplicationTaskInfo) GetWorkflowID() string {\n\treturn t.WorkflowID\n}\n\n// GetRunID returns the run ID for replication task\nfunc (t *ReplicationTaskInfo) GetRunID() string {\n\treturn t.RunID\n}\n\n// GetDomainID returns the domain ID for replication task\nfunc (t *ReplicationTaskInfo) GetDomainID() string {\n\treturn t.DomainID\n}\n\n// GetTaskID returns the task ID for timer task\nfunc (t *TimerTaskInfo) GetTaskID() int64 {\n\treturn t.TaskID\n}\n\n// GetVersion returns the task version for timer task\nfunc (t *TimerTaskInfo) GetVersion() int64 {\n\treturn t.Version\n}\n\n// GetTaskType returns the task type for timer task\nfunc (t *TimerTaskInfo) GetTaskType() int {\n\treturn t.TaskType\n}\n\n// GetVisibilityTimestamp returns the task type for timer task\nfunc (t *TimerTaskInfo) GetVisibilityTimestamp() time.Time {\n\treturn t.VisibilityTimestamp\n}\n\n// GetWorkflowID returns the workflow ID for timer task\nfunc (t *TimerTaskInfo) GetWorkflowID() string {\n\treturn t.WorkflowID\n}\n\n// GetRunID returns the run ID for timer task\nfunc (t *TimerTaskInfo) GetRunID() string {\n\treturn t.RunID\n}\n\n// GetDomainID returns the domain ID for timer task\nfunc (t *TimerTaskInfo) GetDomainID() string {\n\treturn t.DomainID\n}\n\n// GetTaskList returns the task list for timer task\nfunc (t *TimerTaskInfo) GetTaskList() string {\n\treturn t.TaskList\n}\n\n// String returns a string representation for timer task\nfunc (t *TimerTaskInfo) String() string {\n\treturn fmt.Sprintf(\n\t\t\"{DomainID: %v, WorkflowID: %v, RunID: %v, VisibilityTimestamp: %v, TaskID: %v, TaskType: %v, TimeoutType: %v, EventID: %v, ScheduleAttempt: %v, Version: %v.}\",\n\t\tt.DomainID, t.WorkflowID, t.RunID, t.VisibilityTimestamp, t.TaskID, t.TaskType, t.TimeoutType, t.EventID, t.ScheduleAttempt, t.Version,\n\t)\n}\n\nfunc (t *TimerTaskInfo) ToTask() (Task, error) {\n\tworkflowIdentifier := WorkflowIdentifier{\n\t\tDomainID:   t.DomainID,\n\t\tWorkflowID: t.WorkflowID,\n\t\tRunID:      t.RunID,\n\t}\n\ttaskData := TaskData{\n\t\tVersion:             t.Version,\n\t\tTaskID:              t.TaskID,\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t}\n\tswitch t.TaskType {\n\tcase TaskTypeDecisionTimeout:\n\t\treturn &DecisionTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            t.EventID,\n\t\t\tScheduleAttempt:    t.ScheduleAttempt,\n\t\t\tTimeoutType:        t.TimeoutType,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeActivityTimeout:\n\t\treturn &ActivityTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTimeoutType:        t.TimeoutType,\n\t\t\tEventID:            t.EventID,\n\t\t\tAttempt:            t.ScheduleAttempt,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeDeleteHistoryEvent:\n\t\treturn &DeleteHistoryEventTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeWorkflowTimeout:\n\t\treturn &WorkflowTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeUserTimer:\n\t\treturn &UserTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            t.EventID,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeActivityRetryTimer:\n\t\treturn &ActivityRetryTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            t.EventID,\n\t\t\tAttempt:            t.ScheduleAttempt,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tcase TaskTypeWorkflowBackoffTimer:\n\t\treturn &WorkflowBackoffTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTimeoutType:        t.TimeoutType,\n\t\t\tTaskList:           t.TaskList,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown task type: %d\", t.TaskType)\n\t}\n}\n\nfunc (c *DomainReplicationConfig) GetActiveClusters() *types.ActiveClusters {\n\tif c != nil && c.ActiveClusters != nil {\n\t\treturn c.ActiveClusters\n\t}\n\treturn nil\n}\n\nfunc (c *DomainReplicationConfig) GetActiveClusterName() string {\n\tif c == nil {\n\t\treturn \"\"\n\t}\n\treturn c.ActiveClusterName\n}\n\nfunc (c *DomainReplicationConfig) GetClusterAttributeScopes() map[string]types.ClusterAttributeScope {\n\tif c == nil || c.ActiveClusters == nil {\n\t\treturn nil\n\t}\n\treturn c.ActiveClusters.AttributeScopes\n}\n\n// GetID returns the ID from DomainInfo\nfunc (d *DomainInfo) GetID() string {\n\tif d == nil {\n\t\treturn \"\"\n\t}\n\treturn d.ID\n}\n\n// ToNilSafeCopy\n// TODO: it seems that we just need a nil safe shardInfo, deep copy is not necessary\nfunc (s *ShardInfo) ToNilSafeCopy() *ShardInfo {\n\tif s == nil {\n\t\ts = &ShardInfo{}\n\t}\n\tshardInfo := s.copy()\n\tif shardInfo.ClusterTransferAckLevel == nil {\n\t\tshardInfo.ClusterTransferAckLevel = make(map[string]int64)\n\t}\n\tif shardInfo.ClusterTimerAckLevel == nil {\n\t\tshardInfo.ClusterTimerAckLevel = make(map[string]time.Time)\n\t}\n\tif shardInfo.ClusterReplicationLevel == nil {\n\t\tshardInfo.ClusterReplicationLevel = make(map[string]int64)\n\t}\n\tif shardInfo.ReplicationDLQAckLevel == nil {\n\t\tshardInfo.ReplicationDLQAckLevel = make(map[string]int64)\n\t}\n\tif shardInfo.TransferProcessingQueueStates == nil {\n\t\tshardInfo.TransferProcessingQueueStates = &types.ProcessingQueueStates{\n\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t}\n\t}\n\tif shardInfo.TimerProcessingQueueStates == nil {\n\t\tshardInfo.TimerProcessingQueueStates = &types.ProcessingQueueStates{\n\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t}\n\t}\n\tif shardInfo.QueueStates == nil {\n\t\tshardInfo.QueueStates = make(map[int32]*types.QueueState)\n\t}\n\treturn shardInfo\n}\n\n// copy returns a deep copy of shardInfo\nfunc (s *ShardInfo) copy() *ShardInfo {\n\t// TODO: do we really need to deep copy those fields?\n\tvar clusterTransferAckLevel map[string]int64\n\tif s.ClusterTransferAckLevel != nil {\n\t\tclusterTransferAckLevel = make(map[string]int64)\n\t\tfor k, v := range s.ClusterTransferAckLevel {\n\t\t\tclusterTransferAckLevel[k] = v\n\t\t}\n\t}\n\n\tvar clusterTimerAckLevel map[string]time.Time\n\tif s.ClusterTimerAckLevel != nil {\n\t\tclusterTimerAckLevel = make(map[string]time.Time)\n\t\tfor k, v := range s.ClusterTimerAckLevel {\n\t\t\tclusterTimerAckLevel[k] = v\n\t\t}\n\t}\n\n\tvar clusterReplicationLevel map[string]int64\n\tif s.ClusterReplicationLevel != nil {\n\t\tclusterReplicationLevel = make(map[string]int64)\n\t\tfor k, v := range s.ClusterReplicationLevel {\n\t\t\tclusterReplicationLevel[k] = v\n\t\t}\n\t}\n\n\tvar replicationDLQAckLevel map[string]int64\n\tif s.ReplicationDLQAckLevel != nil {\n\t\treplicationDLQAckLevel = make(map[string]int64)\n\t\tfor k, v := range s.ReplicationDLQAckLevel {\n\t\t\treplicationDLQAckLevel[k] = v\n\t\t}\n\t}\n\n\tvar queueStates map[int32]*types.QueueState\n\tif s.QueueStates != nil {\n\t\tqueueStates = make(map[int32]*types.QueueState)\n\t\tfor k, v := range s.QueueStates {\n\t\t\tqueueStates[k] = v.Copy()\n\t\t}\n\t}\n\n\treturn &ShardInfo{\n\t\tShardID:                       s.ShardID,\n\t\tOwner:                         s.Owner,\n\t\tRangeID:                       s.RangeID,\n\t\tStolenSinceRenew:              s.StolenSinceRenew,\n\t\tReplicationAckLevel:           s.ReplicationAckLevel,\n\t\tTransferAckLevel:              s.TransferAckLevel,\n\t\tTimerAckLevel:                 s.TimerAckLevel,\n\t\tClusterTransferAckLevel:       clusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          clusterTimerAckLevel,\n\t\tTransferProcessingQueueStates: s.TransferProcessingQueueStates,\n\t\tTimerProcessingQueueStates:    s.TimerProcessingQueueStates,\n\t\tDomainNotificationVersion:     s.DomainNotificationVersion,\n\t\tClusterReplicationLevel:       clusterReplicationLevel,\n\t\tReplicationDLQAckLevel:        replicationDLQAckLevel,\n\t\tPendingFailoverMarkers:        s.PendingFailoverMarkers,\n\t\tUpdatedAt:                     s.UpdatedAt,\n\t\tQueueStates:                   queueStates,\n\t}\n}\n\n// SerializeClusterConfigs makes an array of *ClusterReplicationConfig serializable\n// by flattening them into map[string]interface{}\nfunc SerializeClusterConfigs(replicationConfigs []*ClusterReplicationConfig) []map[string]interface{} {\n\tvar serializedReplicationConfigs []map[string]interface{}\n\tfor index := range replicationConfigs {\n\t\tserializedReplicationConfigs = append(serializedReplicationConfigs, replicationConfigs[index].serialize())\n\t}\n\treturn serializedReplicationConfigs\n}\n\n// DeserializeClusterConfigs creates an array of ClusterReplicationConfigs from an array of map representations\nfunc DeserializeClusterConfigs(replicationConfigs []map[string]interface{}) []*ClusterReplicationConfig {\n\tdeserializedReplicationConfigs := []*ClusterReplicationConfig{}\n\tfor index := range replicationConfigs {\n\t\tdeserializedReplicationConfig := &ClusterReplicationConfig{}\n\t\tdeserializedReplicationConfig.deserialize(replicationConfigs[index])\n\t\tdeserializedReplicationConfigs = append(deserializedReplicationConfigs, deserializedReplicationConfig)\n\t}\n\n\treturn deserializedReplicationConfigs\n}\n\nfunc (config *ClusterReplicationConfig) serialize() map[string]interface{} {\n\toutput := make(map[string]interface{})\n\toutput[\"cluster_name\"] = config.ClusterName\n\treturn output\n}\n\nfunc (config *ClusterReplicationConfig) deserialize(input map[string]interface{}) {\n\tconfig.ClusterName = input[\"cluster_name\"].(string)\n}\n\n// GetCopy return a copy of ClusterReplicationConfig\nfunc (config *ClusterReplicationConfig) GetCopy() *ClusterReplicationConfig {\n\tres := *config\n\treturn &res\n}\n\nfunc (r *GetDomainResponse) GetReplicationConfig() *DomainReplicationConfig {\n\tif r == nil || r.ReplicationConfig == nil {\n\t\treturn nil\n\t}\n\treturn r.ReplicationConfig\n}\n\n// DeepCopy returns a deep copy of GetDomainResponse\n// todo (david.porter) delete this manual deepcopying since it's annoying to maintain and\n// use codegen for generating them instead\nfunc (r *GetDomainResponse) DeepCopy() *GetDomainResponse {\n\tif r == nil {\n\t\treturn nil\n\t}\n\n\tresult := &GetDomainResponse{\n\t\tIsGlobalDomain:              r.IsGlobalDomain,\n\t\tConfigVersion:               r.ConfigVersion,\n\t\tFailoverVersion:             r.FailoverVersion,\n\t\tFailoverNotificationVersion: r.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     r.PreviousFailoverVersion,\n\t\tLastUpdatedTime:             r.LastUpdatedTime,\n\t\tNotificationVersion:         r.NotificationVersion,\n\t}\n\n\t// Deep copy FailoverEndTime\n\tif r.FailoverEndTime != nil {\n\t\tfailoverEndTime := *r.FailoverEndTime\n\t\tresult.FailoverEndTime = &failoverEndTime\n\t}\n\t// Deep copy DomainInfo\n\tif r.Info != nil {\n\t\tresult.Info = &DomainInfo{\n\t\t\tID:          r.Info.ID,\n\t\t\tName:        r.Info.Name,\n\t\t\tStatus:      r.Info.Status,\n\t\t\tDescription: r.Info.Description,\n\t\t\tOwnerEmail:  r.Info.OwnerEmail,\n\t\t}\n\t\tif r.Info.Data != nil {\n\t\t\tresult.Info.Data = make(map[string]string, len(r.Info.Data))\n\t\t\tfor k, v := range r.Info.Data {\n\t\t\t\tresult.Info.Data[k] = v\n\t\t\t}\n\t\t}\n\t}\n\n\t// Deep copy DomainConfig\n\tif r.Config != nil {\n\t\tresult.Config = &DomainConfig{\n\t\t\tRetention:                r.Config.Retention,\n\t\t\tEmitMetric:               r.Config.EmitMetric,\n\t\t\tHistoryArchivalStatus:    r.Config.HistoryArchivalStatus,\n\t\t\tHistoryArchivalURI:       r.Config.HistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus: r.Config.VisibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    r.Config.VisibilityArchivalURI,\n\t\t}\n\t\t// Deep copy BadBinaries\n\t\tresult.Config.BadBinaries = r.Config.BadBinaries.DeepCopy()\n\t\t// Deep copy IsolationGroups\n\t\tresult.Config.IsolationGroups = r.Config.IsolationGroups.DeepCopy()\n\t\t// Deep copy AsyncWorkflowConfig\n\t\tresult.Config.AsyncWorkflowConfig = r.Config.AsyncWorkflowConfig.DeepCopy()\n\t}\n\n\t// Deep copy DomainReplicationConfig\n\tif r.ReplicationConfig != nil {\n\t\tresult.ReplicationConfig = &DomainReplicationConfig{\n\t\t\tActiveClusterName: r.ReplicationConfig.ActiveClusterName,\n\t\t}\n\t\t// Deep copy Clusters\n\t\tif r.ReplicationConfig.Clusters != nil {\n\t\t\tresult.ReplicationConfig.Clusters = make([]*ClusterReplicationConfig, len(r.ReplicationConfig.Clusters))\n\t\t\tfor i, cluster := range r.ReplicationConfig.Clusters {\n\t\t\t\tif cluster != nil {\n\t\t\t\t\tresult.ReplicationConfig.Clusters[i] = cluster.GetCopy()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\t// Deep copy ActiveClusters\n\t\tif r.ReplicationConfig.ActiveClusters != nil {\n\t\t\tresult.ReplicationConfig.ActiveClusters = r.ReplicationConfig.ActiveClusters.DeepCopy()\n\t\t}\n\t}\n\n\treturn result\n}\n\n// GetInfo returns the DomainInfo from GetDomainResponse\nfunc (r *GetDomainResponse) GetInfo() *DomainInfo {\n\tif r == nil {\n\t\treturn nil\n\t}\n\treturn r.Info\n}\n\n// DBTimestampToUnixNano converts Milliseconds timestamp to UnixNano\nfunc DBTimestampToUnixNano(milliseconds int64) int64 {\n\treturn milliseconds * 1000 * 1000 // Milliseconds are 10⁻³, nanoseconds are 10⁻⁹, (-3) - (-9) = 6, so multiply by 10⁶\n}\n\n// UnixNanoToDBTimestamp converts UnixNano to Milliseconds timestamp\nfunc UnixNanoToDBTimestamp(timestamp int64) int64 {\n\treturn timestamp / (1000 * 1000) // Milliseconds are 10⁻³, nanoseconds are 10⁻⁹, (-9) - (-3) = -6, so divide by 10⁶\n}\n\nvar internalThriftEncoder = codec.NewThriftRWEncoder()\n\n// NewHistoryBranchToken return a new branch token\nfunc NewHistoryBranchToken(treeID string) ([]byte, error) {\n\tbranchID := uuid.New()\n\tbi := &workflow.HistoryBranch{\n\t\tTreeID:    &treeID,\n\t\tBranchID:  &branchID,\n\t\tAncestors: []*workflow.HistoryBranchRange{},\n\t}\n\ttoken, err := internalThriftEncoder.Encode(bi)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn token, nil\n}\n\n// NewHistoryBranchTokenByBranchID return a new branch token with treeID/branchID\nfunc NewHistoryBranchTokenByBranchID(treeID, branchID string) ([]byte, error) {\n\tbi := &workflow.HistoryBranch{\n\t\tTreeID:    &treeID,\n\t\tBranchID:  &branchID,\n\t\tAncestors: []*workflow.HistoryBranchRange{},\n\t}\n\ttoken, err := internalThriftEncoder.Encode(bi)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn token, nil\n}\n\n// NewHistoryBranchTokenFromAnother make up a branchToken\nfunc NewHistoryBranchTokenFromAnother(branchID string, anotherToken []byte) ([]byte, error) {\n\tvar branch workflow.HistoryBranch\n\terr := internalThriftEncoder.Decode(anotherToken, &branch)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tbi := &workflow.HistoryBranch{\n\t\tTreeID:    branch.TreeID,\n\t\tBranchID:  &branchID,\n\t\tAncestors: []*workflow.HistoryBranchRange{},\n\t}\n\ttoken, err := internalThriftEncoder.Encode(bi)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn token, nil\n}\n\n// BuildHistoryGarbageCleanupInfo combine the workflow identity information into a string\nfunc BuildHistoryGarbageCleanupInfo(domainID, workflowID, runID string) string {\n\treturn fmt.Sprintf(\"%v:%v:%v\", domainID, workflowID, runID)\n}\n\n// SplitHistoryGarbageCleanupInfo returns workflow identity information\nfunc SplitHistoryGarbageCleanupInfo(info string) (domainID, workflowID, runID string, err error) {\n\tss := strings.Split(info, \":\")\n\t// workflowID can contain \":\" so len(ss) can be greater than 3\n\tif len(ss) < numItemsInGarbageInfo {\n\t\treturn \"\", \"\", \"\", fmt.Errorf(\"not able to split info for  %s\", info)\n\t}\n\tdomainID = ss[0]\n\trunID = ss[len(ss)-1]\n\tworkflowEnd := len(info) - len(runID) - 1\n\tworkflowID = info[len(domainID)+1 : workflowEnd]\n\treturn\n}\n\n// NewGetReplicationTasksFromDLQRequest creates a new GetReplicationTasksFromDLQRequest\nfunc NewGetReplicationTasksFromDLQRequest(\n\tsourceClusterName string,\n\treadLevel int64,\n\tmaxReadLevel int64,\n\tbatchSize int,\n\tnextPageToken []byte,\n) *GetReplicationTasksFromDLQRequest {\n\treturn &GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: sourceClusterName,\n\t\tReadLevel:         readLevel,\n\t\tMaxReadLevel:      maxReadLevel,\n\t\tBatchSize:         batchSize,\n\t\tNextPageToken:     nextPageToken,\n\t}\n}\n\n// IsTransientError checks if the error is a transient persistence error\nfunc IsTransientError(err error) bool {\n\tvar internalServiceError *types.InternalServiceError\n\tvar serviceBusyError *types.ServiceBusyError\n\tvar timeoutError *TimeoutError\n\tswitch {\n\tcase errors.As(err, &internalServiceError), errors.As(err, &serviceBusyError), errors.As(err, &timeoutError):\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// IsBackgroundTransientError checks if the error is a transient error on background jobs\nfunc IsBackgroundTransientError(err error) bool {\n\tvar internalServiceError *types.InternalServiceError\n\tvar timeoutError *TimeoutError\n\tswitch {\n\tcase errors.As(err, &internalServiceError), errors.As(err, &timeoutError):\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// HasMoreRowsToDelete checks if there is more data need to be deleted\nfunc HasMoreRowsToDelete(rowsDeleted, batchSize int) bool {\n\tif rowsDeleted < batchSize || // all target tasks are deleted\n\t\trowsDeleted == UnknownNumRowsAffected || // underlying database does not support rows affected, so pageSize is not honored and all target tasks are deleted\n\t\trowsDeleted > batchSize ||\n\t\tbatchSize <= 0 { // if batchSize is <= 0 the attempt is for the request to be unbounded and remove all rows from min to max\n\t\treturn false\n\t}\n\treturn true\n}\n\nfunc TaskListKindHasTTL(taskListKind int) bool {\n\tswitch taskListKind {\n\tcase TaskListKindEphemeral:\n\t\treturn true\n\tcase TaskListKindSticky:\n\t\treturn true\n\tcase TaskListKindNormal:\n\t\treturn false\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc (e *WorkflowExecutionInfo) CopyMemo() map[string][]byte {\n\tif e.Memo == nil {\n\t\treturn nil\n\t}\n\tmemo := make(map[string][]byte)\n\tfor k, v := range e.Memo {\n\t\tval := make([]byte, len(v))\n\t\tcopy(val, v)\n\t\tmemo[k] = val\n\t}\n\treturn memo\n}\n\nfunc (e *WorkflowExecutionInfo) CopySearchAttributes() map[string][]byte {\n\tif e.SearchAttributes == nil {\n\t\treturn nil\n\t}\n\tsearchAttr := make(map[string][]byte)\n\tfor k, v := range e.SearchAttributes {\n\t\tval := make([]byte, len(v))\n\t\tcopy(val, v)\n\t\tsearchAttr[k] = val\n\t}\n\treturn searchAttr\n}\n\nfunc (e *WorkflowExecutionInfo) CopyPartitionConfig() map[string]string {\n\tif e.PartitionConfig == nil {\n\t\treturn nil\n\t}\n\tpartitionConfig := make(map[string]string)\n\tfor k, v := range e.PartitionConfig {\n\t\tpartitionConfig[k] = v\n\t}\n\treturn partitionConfig\n}\n\nfunc (p *TaskListPartitionConfig) ToInternalType() *types.TaskListPartitionConfig {\n\tif p == nil {\n\t\treturn nil\n\t}\n\tvar readPartitions map[int]*types.TaskListPartition\n\tif p.ReadPartitions != nil {\n\t\treadPartitions = make(map[int]*types.TaskListPartition, len(p.ReadPartitions))\n\t\tfor id, par := range p.ReadPartitions {\n\t\t\treadPartitions[id] = par.ToInternalType()\n\t\t}\n\t}\n\tvar writePartitions map[int]*types.TaskListPartition\n\tif p.WritePartitions != nil {\n\t\twritePartitions = make(map[int]*types.TaskListPartition, len(p.WritePartitions))\n\t\tfor id, par := range p.WritePartitions {\n\t\t\twritePartitions[id] = par.ToInternalType()\n\t\t}\n\t}\n\n\treturn &types.TaskListPartitionConfig{\n\t\tVersion:         p.Version,\n\t\tReadPartitions:  readPartitions,\n\t\tWritePartitions: writePartitions,\n\t}\n}\n\nfunc (p *TaskListPartition) ToInternalType() *types.TaskListPartition {\n\tif p == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartition{IsolationGroups: p.IsolationGroups}\n}\n\n// IsActiveActive returns true if the domain has active-active configuration.\n// Returns true if ClusterAttributes (or legacy RegionToClusters) have been configured for this domain.\n// TODO(active-active): Update unit tests of all components that use this function to cover active-active case\nfunc (c *DomainReplicationConfig) IsActiveActive() bool {\n\tif c == nil || c.ActiveClusters == nil {\n\t\treturn false\n\t}\n\n\t// Check to see if a ClusterAttribute has been configured for this domain.\n\tif len(c.ActiveClusters.AttributeScopes) > 0 {\n\t\tfor _, scope := range c.ActiveClusters.AttributeScopes {\n\t\t\tif len(scope.ClusterAttributes) > 0 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t}\n\n\treturn false\n}\n\nfunc IsWorkflowRunning(state int) bool {\n\treturn state == WorkflowStateRunning || state == WorkflowStateCreated\n}\n"
  },
  {
    "path": "common/persistence/data_manager_interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence (interfaces: Task,ShardManager,ExecutionManager,ExecutionManagerFactory,TaskManager,HistoryManager,DomainManager,DomainAuditManager,QueueManager,ConfigStoreManager)\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -destination data_manager_interfaces_mock.go github.com/uber/cadence/common/persistence Task,ShardManager,ExecutionManager,ExecutionManagerFactory,TaskManager,HistoryManager,DomainManager,DomainAuditManager,QueueManager,ConfigStoreManager\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockTask is a mock of Task interface.\ntype MockTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskMockRecorder is the mock recorder for MockTask.\ntype MockTaskMockRecorder struct {\n\tmock *MockTask\n}\n\n// NewMockTask creates a new mock instance.\nfunc NewMockTask(ctrl *gomock.Controller) *MockTask {\n\tmock := &MockTask{ctrl: ctrl}\n\tmock.recorder = &MockTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTask) EXPECT() *MockTaskMockRecorder {\n\treturn m.recorder\n}\n\n// ByteSize mocks base method.\nfunc (m *MockTask) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockTaskMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockTask)(nil).ByteSize))\n}\n\n// GetDomainID mocks base method.\nfunc (m *MockTask) GetDomainID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainID indicates an expected call of GetDomainID.\nfunc (mr *MockTaskMockRecorder) GetDomainID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainID\", reflect.TypeOf((*MockTask)(nil).GetDomainID))\n}\n\n// GetOriginalTaskList mocks base method.\nfunc (m *MockTask) GetOriginalTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetOriginalTaskList indicates an expected call of GetOriginalTaskList.\nfunc (mr *MockTaskMockRecorder) GetOriginalTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskList\", reflect.TypeOf((*MockTask)(nil).GetOriginalTaskList))\n}\n\n// GetOriginalTaskListKind mocks base method.\nfunc (m *MockTask) GetOriginalTaskListKind() types.TaskListKind {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskListKind\")\n\tret0, _ := ret[0].(types.TaskListKind)\n\treturn ret0\n}\n\n// GetOriginalTaskListKind indicates an expected call of GetOriginalTaskListKind.\nfunc (mr *MockTaskMockRecorder) GetOriginalTaskListKind() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskListKind\", reflect.TypeOf((*MockTask)(nil).GetOriginalTaskListKind))\n}\n\n// GetRunID mocks base method.\nfunc (m *MockTask) GetRunID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRunID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetRunID indicates an expected call of GetRunID.\nfunc (mr *MockTaskMockRecorder) GetRunID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRunID\", reflect.TypeOf((*MockTask)(nil).GetRunID))\n}\n\n// GetTaskCategory mocks base method.\nfunc (m *MockTask) GetTaskCategory() HistoryTaskCategory {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskCategory\")\n\tret0, _ := ret[0].(HistoryTaskCategory)\n\treturn ret0\n}\n\n// GetTaskCategory indicates an expected call of GetTaskCategory.\nfunc (mr *MockTaskMockRecorder) GetTaskCategory() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskCategory\", reflect.TypeOf((*MockTask)(nil).GetTaskCategory))\n}\n\n// GetTaskID mocks base method.\nfunc (m *MockTask) GetTaskID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetTaskID indicates an expected call of GetTaskID.\nfunc (mr *MockTaskMockRecorder) GetTaskID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskID\", reflect.TypeOf((*MockTask)(nil).GetTaskID))\n}\n\n// GetTaskKey mocks base method.\nfunc (m *MockTask) GetTaskKey() HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskKey\")\n\tret0, _ := ret[0].(HistoryTaskKey)\n\treturn ret0\n}\n\n// GetTaskKey indicates an expected call of GetTaskKey.\nfunc (mr *MockTaskMockRecorder) GetTaskKey() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskKey\", reflect.TypeOf((*MockTask)(nil).GetTaskKey))\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockTask) GetTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockTaskMockRecorder) GetTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockTask)(nil).GetTaskList))\n}\n\n// GetTaskType mocks base method.\nfunc (m *MockTask) GetTaskType() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskType\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetTaskType indicates an expected call of GetTaskType.\nfunc (mr *MockTaskMockRecorder) GetTaskType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskType\", reflect.TypeOf((*MockTask)(nil).GetTaskType))\n}\n\n// GetVersion mocks base method.\nfunc (m *MockTask) GetVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetVersion indicates an expected call of GetVersion.\nfunc (mr *MockTaskMockRecorder) GetVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVersion\", reflect.TypeOf((*MockTask)(nil).GetVersion))\n}\n\n// GetVisibilityTimestamp mocks base method.\nfunc (m *MockTask) GetVisibilityTimestamp() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityTimestamp\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// GetVisibilityTimestamp indicates an expected call of GetVisibilityTimestamp.\nfunc (mr *MockTaskMockRecorder) GetVisibilityTimestamp() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityTimestamp\", reflect.TypeOf((*MockTask)(nil).GetVisibilityTimestamp))\n}\n\n// GetWorkflowID mocks base method.\nfunc (m *MockTask) GetWorkflowID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetWorkflowID indicates an expected call of GetWorkflowID.\nfunc (mr *MockTaskMockRecorder) GetWorkflowID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowID\", reflect.TypeOf((*MockTask)(nil).GetWorkflowID))\n}\n\n// SetTaskID mocks base method.\nfunc (m *MockTask) SetTaskID(id int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetTaskID\", id)\n}\n\n// SetTaskID indicates an expected call of SetTaskID.\nfunc (mr *MockTaskMockRecorder) SetTaskID(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetTaskID\", reflect.TypeOf((*MockTask)(nil).SetTaskID), id)\n}\n\n// SetVersion mocks base method.\nfunc (m *MockTask) SetVersion(version int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVersion\", version)\n}\n\n// SetVersion indicates an expected call of SetVersion.\nfunc (mr *MockTaskMockRecorder) SetVersion(version any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVersion\", reflect.TypeOf((*MockTask)(nil).SetVersion), version)\n}\n\n// SetVisibilityTimestamp mocks base method.\nfunc (m *MockTask) SetVisibilityTimestamp(timestamp time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVisibilityTimestamp\", timestamp)\n}\n\n// SetVisibilityTimestamp indicates an expected call of SetVisibilityTimestamp.\nfunc (mr *MockTaskMockRecorder) SetVisibilityTimestamp(timestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVisibilityTimestamp\", reflect.TypeOf((*MockTask)(nil).SetVisibilityTimestamp), timestamp)\n}\n\n// ToInternalReplicationTaskInfo mocks base method.\nfunc (m *MockTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToInternalReplicationTaskInfo\")\n\tret0, _ := ret[0].(*types.ReplicationTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToInternalReplicationTaskInfo indicates an expected call of ToInternalReplicationTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToInternalReplicationTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToInternalReplicationTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToInternalReplicationTaskInfo))\n}\n\n// ToTimerTaskInfo mocks base method.\nfunc (m *MockTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTimerTaskInfo\")\n\tret0, _ := ret[0].(*TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTimerTaskInfo indicates an expected call of ToTimerTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToTimerTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTimerTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToTimerTaskInfo))\n}\n\n// ToTransferTaskInfo mocks base method.\nfunc (m *MockTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTransferTaskInfo\")\n\tret0, _ := ret[0].(*TransferTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTransferTaskInfo indicates an expected call of ToTransferTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToTransferTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTransferTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToTransferTaskInfo))\n}\n\n// MockShardManager is a mock of ShardManager interface.\ntype MockShardManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardManagerMockRecorder is the mock recorder for MockShardManager.\ntype MockShardManagerMockRecorder struct {\n\tmock *MockShardManager\n}\n\n// NewMockShardManager creates a new mock instance.\nfunc NewMockShardManager(ctrl *gomock.Controller) *MockShardManager {\n\tmock := &MockShardManager{ctrl: ctrl}\n\tmock.recorder = &MockShardManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardManager) EXPECT() *MockShardManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockShardManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockShardManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockShardManager)(nil).Close))\n}\n\n// CreateShard mocks base method.\nfunc (m *MockShardManager) CreateShard(ctx context.Context, request *CreateShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateShard\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateShard indicates an expected call of CreateShard.\nfunc (mr *MockShardManagerMockRecorder) CreateShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateShard\", reflect.TypeOf((*MockShardManager)(nil).CreateShard), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockShardManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockShardManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockShardManager)(nil).GetName))\n}\n\n// GetShard mocks base method.\nfunc (m *MockShardManager) GetShard(ctx context.Context, request *GetShardRequest) (*GetShardResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShard\", ctx, request)\n\tret0, _ := ret[0].(*GetShardResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShard indicates an expected call of GetShard.\nfunc (mr *MockShardManagerMockRecorder) GetShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShard\", reflect.TypeOf((*MockShardManager)(nil).GetShard), ctx, request)\n}\n\n// UpdateShard mocks base method.\nfunc (m *MockShardManager) UpdateShard(ctx context.Context, request *UpdateShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShard\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateShard indicates an expected call of UpdateShard.\nfunc (mr *MockShardManagerMockRecorder) UpdateShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShard\", reflect.TypeOf((*MockShardManager)(nil).UpdateShard), ctx, request)\n}\n\n// MockExecutionManager is a mock of ExecutionManager interface.\ntype MockExecutionManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutionManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutionManagerMockRecorder is the mock recorder for MockExecutionManager.\ntype MockExecutionManagerMockRecorder struct {\n\tmock *MockExecutionManager\n}\n\n// NewMockExecutionManager creates a new mock instance.\nfunc NewMockExecutionManager(ctrl *gomock.Controller) *MockExecutionManager {\n\tmock := &MockExecutionManager{ctrl: ctrl}\n\tmock.recorder = &MockExecutionManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutionManager) EXPECT() *MockExecutionManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockExecutionManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockExecutionManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockExecutionManager)(nil).Close))\n}\n\n// CompleteHistoryTask mocks base method.\nfunc (m *MockExecutionManager) CompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteHistoryTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteHistoryTask indicates an expected call of CompleteHistoryTask.\nfunc (mr *MockExecutionManagerMockRecorder) CompleteHistoryTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteHistoryTask\", reflect.TypeOf((*MockExecutionManager)(nil).CompleteHistoryTask), ctx, request)\n}\n\n// ConflictResolveWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) ConflictResolveWorkflowExecution(ctx context.Context, request *ConflictResolveWorkflowExecutionRequest) (*ConflictResolveWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ConflictResolveWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*ConflictResolveWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ConflictResolveWorkflowExecution indicates an expected call of ConflictResolveWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) ConflictResolveWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ConflictResolveWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).ConflictResolveWorkflowExecution), ctx, request)\n}\n\n// CreateFailoverMarkerTasks mocks base method.\nfunc (m *MockExecutionManager) CreateFailoverMarkerTasks(ctx context.Context, request *CreateFailoverMarkersRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateFailoverMarkerTasks\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateFailoverMarkerTasks indicates an expected call of CreateFailoverMarkerTasks.\nfunc (mr *MockExecutionManagerMockRecorder) CreateFailoverMarkerTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateFailoverMarkerTasks\", reflect.TypeOf((*MockExecutionManager)(nil).CreateFailoverMarkerTasks), ctx, request)\n}\n\n// CreateWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) CreateWorkflowExecution(ctx context.Context, request *CreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*CreateWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateWorkflowExecution indicates an expected call of CreateWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) CreateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).CreateWorkflowExecution), ctx, request)\n}\n\n// DeleteActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockExecutionManager) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActiveClusterSelectionPolicy\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActiveClusterSelectionPolicy indicates an expected call of DeleteActiveClusterSelectionPolicy.\nfunc (mr *MockExecutionManagerMockRecorder) DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActiveClusterSelectionPolicy\", reflect.TypeOf((*MockExecutionManager)(nil).DeleteActiveClusterSelectionPolicy), ctx, domainID, workflowID, runID)\n}\n\n// DeleteCurrentWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) DeleteCurrentWorkflowExecution(ctx context.Context, request *DeleteCurrentWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflowExecution indicates an expected call of DeleteCurrentWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) DeleteCurrentWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).DeleteCurrentWorkflowExecution), ctx, request)\n}\n\n// DeleteReplicationTaskFromDLQ mocks base method.\nfunc (m *MockExecutionManager) DeleteReplicationTaskFromDLQ(ctx context.Context, request *DeleteReplicationTaskFromDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationTaskFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationTaskFromDLQ indicates an expected call of DeleteReplicationTaskFromDLQ.\nfunc (mr *MockExecutionManagerMockRecorder) DeleteReplicationTaskFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationTaskFromDLQ\", reflect.TypeOf((*MockExecutionManager)(nil).DeleteReplicationTaskFromDLQ), ctx, request)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) DeleteWorkflowExecution(ctx context.Context, request *DeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) DeleteWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).DeleteWorkflowExecution), ctx, request)\n}\n\n// GetActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockExecutionManager) GetActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*types.ActiveClusterSelectionPolicy, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterSelectionPolicy\", ctx, domainID, wfID, rID)\n\tret0, _ := ret[0].(*types.ActiveClusterSelectionPolicy)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActiveClusterSelectionPolicy indicates an expected call of GetActiveClusterSelectionPolicy.\nfunc (mr *MockExecutionManagerMockRecorder) GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterSelectionPolicy\", reflect.TypeOf((*MockExecutionManager)(nil).GetActiveClusterSelectionPolicy), ctx, domainID, wfID, rID)\n}\n\n// GetCurrentExecution mocks base method.\nfunc (m *MockExecutionManager) GetCurrentExecution(ctx context.Context, request *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentExecution\", ctx, request)\n\tret0, _ := ret[0].(*GetCurrentExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCurrentExecution indicates an expected call of GetCurrentExecution.\nfunc (mr *MockExecutionManagerMockRecorder) GetCurrentExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentExecution\", reflect.TypeOf((*MockExecutionManager)(nil).GetCurrentExecution), ctx, request)\n}\n\n// GetHistoryTasks mocks base method.\nfunc (m *MockExecutionManager) GetHistoryTasks(ctx context.Context, request *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetHistoryTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryTasks indicates an expected call of GetHistoryTasks.\nfunc (mr *MockExecutionManagerMockRecorder) GetHistoryTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryTasks\", reflect.TypeOf((*MockExecutionManager)(nil).GetHistoryTasks), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockExecutionManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockExecutionManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockExecutionManager)(nil).GetName))\n}\n\n// GetReplicationDLQSize mocks base method.\nfunc (m *MockExecutionManager) GetReplicationDLQSize(ctx context.Context, request *GetReplicationDLQSizeRequest) (*GetReplicationDLQSizeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationDLQSize\", ctx, request)\n\tret0, _ := ret[0].(*GetReplicationDLQSizeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationDLQSize indicates an expected call of GetReplicationDLQSize.\nfunc (mr *MockExecutionManagerMockRecorder) GetReplicationDLQSize(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationDLQSize\", reflect.TypeOf((*MockExecutionManager)(nil).GetReplicationDLQSize), ctx, request)\n}\n\n// GetReplicationTasksFromDLQ mocks base method.\nfunc (m *MockExecutionManager) GetReplicationTasksFromDLQ(ctx context.Context, request *GetReplicationTasksFromDLQRequest) (*GetHistoryTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationTasksFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(*GetHistoryTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationTasksFromDLQ indicates an expected call of GetReplicationTasksFromDLQ.\nfunc (mr *MockExecutionManagerMockRecorder) GetReplicationTasksFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationTasksFromDLQ\", reflect.TypeOf((*MockExecutionManager)(nil).GetReplicationTasksFromDLQ), ctx, request)\n}\n\n// GetShardID mocks base method.\nfunc (m *MockExecutionManager) GetShardID() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardID\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetShardID indicates an expected call of GetShardID.\nfunc (mr *MockExecutionManagerMockRecorder) GetShardID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardID\", reflect.TypeOf((*MockExecutionManager)(nil).GetShardID))\n}\n\n// GetWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) GetWorkflowExecution(ctx context.Context, request *GetWorkflowExecutionRequest) (*GetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*GetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecution indicates an expected call of GetWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) GetWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).GetWorkflowExecution), ctx, request)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MockExecutionManager) IsWorkflowExecutionExists(ctx context.Context, request *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", ctx, request)\n\tret0, _ := ret[0].(*IsWorkflowExecutionExistsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MockExecutionManagerMockRecorder) IsWorkflowExecutionExists(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MockExecutionManager)(nil).IsWorkflowExecutionExists), ctx, request)\n}\n\n// ListConcreteExecutions mocks base method.\nfunc (m *MockExecutionManager) ListConcreteExecutions(ctx context.Context, request *ListConcreteExecutionsRequest) (*ListConcreteExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListConcreteExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListConcreteExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListConcreteExecutions indicates an expected call of ListConcreteExecutions.\nfunc (mr *MockExecutionManagerMockRecorder) ListConcreteExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListConcreteExecutions\", reflect.TypeOf((*MockExecutionManager)(nil).ListConcreteExecutions), ctx, request)\n}\n\n// ListCurrentExecutions mocks base method.\nfunc (m *MockExecutionManager) ListCurrentExecutions(ctx context.Context, request *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListCurrentExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListCurrentExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListCurrentExecutions indicates an expected call of ListCurrentExecutions.\nfunc (mr *MockExecutionManagerMockRecorder) ListCurrentExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListCurrentExecutions\", reflect.TypeOf((*MockExecutionManager)(nil).ListCurrentExecutions), ctx, request)\n}\n\n// PutReplicationTaskToDLQ mocks base method.\nfunc (m *MockExecutionManager) PutReplicationTaskToDLQ(ctx context.Context, request *PutReplicationTaskToDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PutReplicationTaskToDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PutReplicationTaskToDLQ indicates an expected call of PutReplicationTaskToDLQ.\nfunc (mr *MockExecutionManagerMockRecorder) PutReplicationTaskToDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PutReplicationTaskToDLQ\", reflect.TypeOf((*MockExecutionManager)(nil).PutReplicationTaskToDLQ), ctx, request)\n}\n\n// RangeCompleteHistoryTask mocks base method.\nfunc (m *MockExecutionManager) RangeCompleteHistoryTask(ctx context.Context, request *RangeCompleteHistoryTaskRequest) (*RangeCompleteHistoryTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeCompleteHistoryTask\", ctx, request)\n\tret0, _ := ret[0].(*RangeCompleteHistoryTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeCompleteHistoryTask indicates an expected call of RangeCompleteHistoryTask.\nfunc (mr *MockExecutionManagerMockRecorder) RangeCompleteHistoryTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeCompleteHistoryTask\", reflect.TypeOf((*MockExecutionManager)(nil).RangeCompleteHistoryTask), ctx, request)\n}\n\n// RangeDeleteReplicationTaskFromDLQ mocks base method.\nfunc (m *MockExecutionManager) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *RangeDeleteReplicationTaskFromDLQRequest) (*RangeDeleteReplicationTaskFromDLQResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationTaskFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(*RangeDeleteReplicationTaskFromDLQResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteReplicationTaskFromDLQ indicates an expected call of RangeDeleteReplicationTaskFromDLQ.\nfunc (mr *MockExecutionManagerMockRecorder) RangeDeleteReplicationTaskFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationTaskFromDLQ\", reflect.TypeOf((*MockExecutionManager)(nil).RangeDeleteReplicationTaskFromDLQ), ctx, request)\n}\n\n// UpdateWorkflowExecution mocks base method.\nfunc (m *MockExecutionManager) UpdateWorkflowExecution(ctx context.Context, request *UpdateWorkflowExecutionRequest) (*UpdateWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*UpdateWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateWorkflowExecution indicates an expected call of UpdateWorkflowExecution.\nfunc (mr *MockExecutionManagerMockRecorder) UpdateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecution\", reflect.TypeOf((*MockExecutionManager)(nil).UpdateWorkflowExecution), ctx, request)\n}\n\n// MockExecutionManagerFactory is a mock of ExecutionManagerFactory interface.\ntype MockExecutionManagerFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutionManagerFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutionManagerFactoryMockRecorder is the mock recorder for MockExecutionManagerFactory.\ntype MockExecutionManagerFactoryMockRecorder struct {\n\tmock *MockExecutionManagerFactory\n}\n\n// NewMockExecutionManagerFactory creates a new mock instance.\nfunc NewMockExecutionManagerFactory(ctrl *gomock.Controller) *MockExecutionManagerFactory {\n\tmock := &MockExecutionManagerFactory{ctrl: ctrl}\n\tmock.recorder = &MockExecutionManagerFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutionManagerFactory) EXPECT() *MockExecutionManagerFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockExecutionManagerFactory) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockExecutionManagerFactoryMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockExecutionManagerFactory)(nil).Close))\n}\n\n// NewExecutionManager mocks base method.\nfunc (m *MockExecutionManagerFactory) NewExecutionManager(shardID int) (ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewExecutionManager\", shardID)\n\tret0, _ := ret[0].(ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewExecutionManager indicates an expected call of NewExecutionManager.\nfunc (mr *MockExecutionManagerFactoryMockRecorder) NewExecutionManager(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewExecutionManager\", reflect.TypeOf((*MockExecutionManagerFactory)(nil).NewExecutionManager), shardID)\n}\n\n// MockTaskManager is a mock of TaskManager interface.\ntype MockTaskManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskManagerMockRecorder is the mock recorder for MockTaskManager.\ntype MockTaskManagerMockRecorder struct {\n\tmock *MockTaskManager\n}\n\n// NewMockTaskManager creates a new mock instance.\nfunc NewMockTaskManager(ctrl *gomock.Controller) *MockTaskManager {\n\tmock := &MockTaskManager{ctrl: ctrl}\n\tmock.recorder = &MockTaskManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskManager) EXPECT() *MockTaskManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockTaskManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockTaskManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockTaskManager)(nil).Close))\n}\n\n// CompleteTask mocks base method.\nfunc (m *MockTaskManager) CompleteTask(ctx context.Context, request *CompleteTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteTask indicates an expected call of CompleteTask.\nfunc (mr *MockTaskManagerMockRecorder) CompleteTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteTask\", reflect.TypeOf((*MockTaskManager)(nil).CompleteTask), ctx, request)\n}\n\n// CompleteTasksLessThan mocks base method.\nfunc (m *MockTaskManager) CompleteTasksLessThan(ctx context.Context, request *CompleteTasksLessThanRequest) (*CompleteTasksLessThanResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteTasksLessThan\", ctx, request)\n\tret0, _ := ret[0].(*CompleteTasksLessThanResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CompleteTasksLessThan indicates an expected call of CompleteTasksLessThan.\nfunc (mr *MockTaskManagerMockRecorder) CompleteTasksLessThan(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteTasksLessThan\", reflect.TypeOf((*MockTaskManager)(nil).CompleteTasksLessThan), ctx, request)\n}\n\n// CreateTasks mocks base method.\nfunc (m *MockTaskManager) CreateTasks(ctx context.Context, request *CreateTasksRequest) (*CreateTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTasks\", ctx, request)\n\tret0, _ := ret[0].(*CreateTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateTasks indicates an expected call of CreateTasks.\nfunc (mr *MockTaskManagerMockRecorder) CreateTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTasks\", reflect.TypeOf((*MockTaskManager)(nil).CreateTasks), ctx, request)\n}\n\n// DeleteTaskList mocks base method.\nfunc (m *MockTaskManager) DeleteTaskList(ctx context.Context, request *DeleteTaskListRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTaskList\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTaskList indicates an expected call of DeleteTaskList.\nfunc (mr *MockTaskManagerMockRecorder) DeleteTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTaskList\", reflect.TypeOf((*MockTaskManager)(nil).DeleteTaskList), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockTaskManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockTaskManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockTaskManager)(nil).GetName))\n}\n\n// GetOrphanTasks mocks base method.\nfunc (m *MockTaskManager) GetOrphanTasks(ctx context.Context, request *GetOrphanTasksRequest) (*GetOrphanTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrphanTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetOrphanTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOrphanTasks indicates an expected call of GetOrphanTasks.\nfunc (mr *MockTaskManagerMockRecorder) GetOrphanTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrphanTasks\", reflect.TypeOf((*MockTaskManager)(nil).GetOrphanTasks), ctx, request)\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockTaskManager) GetTaskList(ctx context.Context, request *GetTaskListRequest) (*GetTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\", ctx, request)\n\tret0, _ := ret[0].(*GetTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockTaskManagerMockRecorder) GetTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockTaskManager)(nil).GetTaskList), ctx, request)\n}\n\n// GetTaskListSize mocks base method.\nfunc (m *MockTaskManager) GetTaskListSize(ctx context.Context, request *GetTaskListSizeRequest) (*GetTaskListSizeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListSize\", ctx, request)\n\tret0, _ := ret[0].(*GetTaskListSizeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListSize indicates an expected call of GetTaskListSize.\nfunc (mr *MockTaskManagerMockRecorder) GetTaskListSize(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListSize\", reflect.TypeOf((*MockTaskManager)(nil).GetTaskListSize), ctx, request)\n}\n\n// GetTasks mocks base method.\nfunc (m *MockTaskManager) GetTasks(ctx context.Context, request *GetTasksRequest) (*GetTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockTaskManagerMockRecorder) GetTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockTaskManager)(nil).GetTasks), ctx, request)\n}\n\n// LeaseTaskList mocks base method.\nfunc (m *MockTaskManager) LeaseTaskList(ctx context.Context, request *LeaseTaskListRequest) (*LeaseTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LeaseTaskList\", ctx, request)\n\tret0, _ := ret[0].(*LeaseTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LeaseTaskList indicates an expected call of LeaseTaskList.\nfunc (mr *MockTaskManagerMockRecorder) LeaseTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LeaseTaskList\", reflect.TypeOf((*MockTaskManager)(nil).LeaseTaskList), ctx, request)\n}\n\n// ListTaskList mocks base method.\nfunc (m *MockTaskManager) ListTaskList(ctx context.Context, request *ListTaskListRequest) (*ListTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskList\", ctx, request)\n\tret0, _ := ret[0].(*ListTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskList indicates an expected call of ListTaskList.\nfunc (mr *MockTaskManagerMockRecorder) ListTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskList\", reflect.TypeOf((*MockTaskManager)(nil).ListTaskList), ctx, request)\n}\n\n// UpdateTaskList mocks base method.\nfunc (m *MockTaskManager) UpdateTaskList(ctx context.Context, request *UpdateTaskListRequest) (*UpdateTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskList\", ctx, request)\n\tret0, _ := ret[0].(*UpdateTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskList indicates an expected call of UpdateTaskList.\nfunc (mr *MockTaskManagerMockRecorder) UpdateTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskList\", reflect.TypeOf((*MockTaskManager)(nil).UpdateTaskList), ctx, request)\n}\n\n// MockHistoryManager is a mock of HistoryManager interface.\ntype MockHistoryManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHistoryManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHistoryManagerMockRecorder is the mock recorder for MockHistoryManager.\ntype MockHistoryManagerMockRecorder struct {\n\tmock *MockHistoryManager\n}\n\n// NewMockHistoryManager creates a new mock instance.\nfunc NewMockHistoryManager(ctrl *gomock.Controller) *MockHistoryManager {\n\tmock := &MockHistoryManager{ctrl: ctrl}\n\tmock.recorder = &MockHistoryManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHistoryManager) EXPECT() *MockHistoryManagerMockRecorder {\n\treturn m.recorder\n}\n\n// AppendHistoryNodes mocks base method.\nfunc (m *MockHistoryManager) AppendHistoryNodes(ctx context.Context, request *AppendHistoryNodesRequest) (*AppendHistoryNodesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AppendHistoryNodes\", ctx, request)\n\tret0, _ := ret[0].(*AppendHistoryNodesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AppendHistoryNodes indicates an expected call of AppendHistoryNodes.\nfunc (mr *MockHistoryManagerMockRecorder) AppendHistoryNodes(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AppendHistoryNodes\", reflect.TypeOf((*MockHistoryManager)(nil).AppendHistoryNodes), ctx, request)\n}\n\n// Close mocks base method.\nfunc (m *MockHistoryManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockHistoryManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockHistoryManager)(nil).Close))\n}\n\n// DeleteHistoryBranch mocks base method.\nfunc (m *MockHistoryManager) DeleteHistoryBranch(ctx context.Context, request *DeleteHistoryBranchRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteHistoryBranch indicates an expected call of DeleteHistoryBranch.\nfunc (mr *MockHistoryManagerMockRecorder) DeleteHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteHistoryBranch\", reflect.TypeOf((*MockHistoryManager)(nil).DeleteHistoryBranch), ctx, request)\n}\n\n// ForkHistoryBranch mocks base method.\nfunc (m *MockHistoryManager) ForkHistoryBranch(ctx context.Context, request *ForkHistoryBranchRequest) (*ForkHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ForkHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(*ForkHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ForkHistoryBranch indicates an expected call of ForkHistoryBranch.\nfunc (mr *MockHistoryManagerMockRecorder) ForkHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ForkHistoryBranch\", reflect.TypeOf((*MockHistoryManager)(nil).ForkHistoryBranch), ctx, request)\n}\n\n// GetAllHistoryTreeBranches mocks base method.\nfunc (m *MockHistoryManager) GetAllHistoryTreeBranches(ctx context.Context, request *GetAllHistoryTreeBranchesRequest) (*GetAllHistoryTreeBranchesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllHistoryTreeBranches\", ctx, request)\n\tret0, _ := ret[0].(*GetAllHistoryTreeBranchesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllHistoryTreeBranches indicates an expected call of GetAllHistoryTreeBranches.\nfunc (mr *MockHistoryManagerMockRecorder) GetAllHistoryTreeBranches(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllHistoryTreeBranches\", reflect.TypeOf((*MockHistoryManager)(nil).GetAllHistoryTreeBranches), ctx, request)\n}\n\n// GetHistoryTree mocks base method.\nfunc (m *MockHistoryManager) GetHistoryTree(ctx context.Context, request *GetHistoryTreeRequest) (*GetHistoryTreeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryTree\", ctx, request)\n\tret0, _ := ret[0].(*GetHistoryTreeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryTree indicates an expected call of GetHistoryTree.\nfunc (mr *MockHistoryManagerMockRecorder) GetHistoryTree(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryTree\", reflect.TypeOf((*MockHistoryManager)(nil).GetHistoryTree), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockHistoryManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockHistoryManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockHistoryManager)(nil).GetName))\n}\n\n// ReadHistoryBranch mocks base method.\nfunc (m *MockHistoryManager) ReadHistoryBranch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(*ReadHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadHistoryBranch indicates an expected call of ReadHistoryBranch.\nfunc (mr *MockHistoryManagerMockRecorder) ReadHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadHistoryBranch\", reflect.TypeOf((*MockHistoryManager)(nil).ReadHistoryBranch), ctx, request)\n}\n\n// ReadHistoryBranchByBatch mocks base method.\nfunc (m *MockHistoryManager) ReadHistoryBranchByBatch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadHistoryBranchByBatchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadHistoryBranchByBatch\", ctx, request)\n\tret0, _ := ret[0].(*ReadHistoryBranchByBatchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadHistoryBranchByBatch indicates an expected call of ReadHistoryBranchByBatch.\nfunc (mr *MockHistoryManagerMockRecorder) ReadHistoryBranchByBatch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadHistoryBranchByBatch\", reflect.TypeOf((*MockHistoryManager)(nil).ReadHistoryBranchByBatch), ctx, request)\n}\n\n// ReadRawHistoryBranch mocks base method.\nfunc (m *MockHistoryManager) ReadRawHistoryBranch(ctx context.Context, request *ReadHistoryBranchRequest) (*ReadRawHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadRawHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(*ReadRawHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadRawHistoryBranch indicates an expected call of ReadRawHistoryBranch.\nfunc (mr *MockHistoryManagerMockRecorder) ReadRawHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadRawHistoryBranch\", reflect.TypeOf((*MockHistoryManager)(nil).ReadRawHistoryBranch), ctx, request)\n}\n\n// MockDomainManager is a mock of DomainManager interface.\ntype MockDomainManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainManagerMockRecorder is the mock recorder for MockDomainManager.\ntype MockDomainManagerMockRecorder struct {\n\tmock *MockDomainManager\n}\n\n// NewMockDomainManager creates a new mock instance.\nfunc NewMockDomainManager(ctrl *gomock.Controller) *MockDomainManager {\n\tmock := &MockDomainManager{ctrl: ctrl}\n\tmock.recorder = &MockDomainManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainManager) EXPECT() *MockDomainManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDomainManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDomainManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDomainManager)(nil).Close))\n}\n\n// CreateDomain mocks base method.\nfunc (m *MockDomainManager) CreateDomain(ctx context.Context, request *CreateDomainRequest) (*CreateDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDomain\", ctx, request)\n\tret0, _ := ret[0].(*CreateDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDomain indicates an expected call of CreateDomain.\nfunc (mr *MockDomainManagerMockRecorder) CreateDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDomain\", reflect.TypeOf((*MockDomainManager)(nil).CreateDomain), ctx, request)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockDomainManager) DeleteDomain(ctx context.Context, request *DeleteDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockDomainManagerMockRecorder) DeleteDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockDomainManager)(nil).DeleteDomain), ctx, request)\n}\n\n// DeleteDomainByName mocks base method.\nfunc (m *MockDomainManager) DeleteDomainByName(ctx context.Context, request *DeleteDomainByNameRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomainByName\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomainByName indicates an expected call of DeleteDomainByName.\nfunc (mr *MockDomainManagerMockRecorder) DeleteDomainByName(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomainByName\", reflect.TypeOf((*MockDomainManager)(nil).DeleteDomainByName), ctx, request)\n}\n\n// GetDomain mocks base method.\nfunc (m *MockDomainManager) GetDomain(ctx context.Context, request *GetDomainRequest) (*GetDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomain\", ctx, request)\n\tret0, _ := ret[0].(*GetDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomain indicates an expected call of GetDomain.\nfunc (mr *MockDomainManagerMockRecorder) GetDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomain\", reflect.TypeOf((*MockDomainManager)(nil).GetDomain), ctx, request)\n}\n\n// GetMetadata mocks base method.\nfunc (m *MockDomainManager) GetMetadata(ctx context.Context) (*GetMetadataResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetadata\", ctx)\n\tret0, _ := ret[0].(*GetMetadataResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMetadata indicates an expected call of GetMetadata.\nfunc (mr *MockDomainManagerMockRecorder) GetMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetadata\", reflect.TypeOf((*MockDomainManager)(nil).GetMetadata), ctx)\n}\n\n// GetName mocks base method.\nfunc (m *MockDomainManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockDomainManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockDomainManager)(nil).GetName))\n}\n\n// ListDomains mocks base method.\nfunc (m *MockDomainManager) ListDomains(ctx context.Context, request *ListDomainsRequest) (*ListDomainsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListDomains\", ctx, request)\n\tret0, _ := ret[0].(*ListDomainsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDomains indicates an expected call of ListDomains.\nfunc (mr *MockDomainManagerMockRecorder) ListDomains(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDomains\", reflect.TypeOf((*MockDomainManager)(nil).ListDomains), ctx, request)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockDomainManager) UpdateDomain(ctx context.Context, request *UpdateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockDomainManagerMockRecorder) UpdateDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockDomainManager)(nil).UpdateDomain), ctx, request)\n}\n\n// MockDomainAuditManager is a mock of DomainAuditManager interface.\ntype MockDomainAuditManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainAuditManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainAuditManagerMockRecorder is the mock recorder for MockDomainAuditManager.\ntype MockDomainAuditManagerMockRecorder struct {\n\tmock *MockDomainAuditManager\n}\n\n// NewMockDomainAuditManager creates a new mock instance.\nfunc NewMockDomainAuditManager(ctrl *gomock.Controller) *MockDomainAuditManager {\n\tmock := &MockDomainAuditManager{ctrl: ctrl}\n\tmock.recorder = &MockDomainAuditManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainAuditManager) EXPECT() *MockDomainAuditManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDomainAuditManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDomainAuditManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDomainAuditManager)(nil).Close))\n}\n\n// CreateDomainAuditLog mocks base method.\nfunc (m *MockDomainAuditManager) CreateDomainAuditLog(ctx context.Context, request *CreateDomainAuditLogRequest) (*CreateDomainAuditLogResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDomainAuditLog\", ctx, request)\n\tret0, _ := ret[0].(*CreateDomainAuditLogResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDomainAuditLog indicates an expected call of CreateDomainAuditLog.\nfunc (mr *MockDomainAuditManagerMockRecorder) CreateDomainAuditLog(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDomainAuditLog\", reflect.TypeOf((*MockDomainAuditManager)(nil).CreateDomainAuditLog), ctx, request)\n}\n\n// GetDomainAuditLogs mocks base method.\nfunc (m *MockDomainAuditManager) GetDomainAuditLogs(ctx context.Context, request *GetDomainAuditLogsRequest) (*GetDomainAuditLogsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAuditLogs\", ctx, request)\n\tret0, _ := ret[0].(*GetDomainAuditLogsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainAuditLogs indicates an expected call of GetDomainAuditLogs.\nfunc (mr *MockDomainAuditManagerMockRecorder) GetDomainAuditLogs(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAuditLogs\", reflect.TypeOf((*MockDomainAuditManager)(nil).GetDomainAuditLogs), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockDomainAuditManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockDomainAuditManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockDomainAuditManager)(nil).GetName))\n}\n\n// MockQueueManager is a mock of QueueManager interface.\ntype MockQueueManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueueManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueueManagerMockRecorder is the mock recorder for MockQueueManager.\ntype MockQueueManagerMockRecorder struct {\n\tmock *MockQueueManager\n}\n\n// NewMockQueueManager creates a new mock instance.\nfunc NewMockQueueManager(ctrl *gomock.Controller) *MockQueueManager {\n\tmock := &MockQueueManager{ctrl: ctrl}\n\tmock.recorder = &MockQueueManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueueManager) EXPECT() *MockQueueManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockQueueManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockQueueManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockQueueManager)(nil).Close))\n}\n\n// DeleteMessageFromDLQ mocks base method.\nfunc (m *MockQueueManager) DeleteMessageFromDLQ(ctx context.Context, request *DeleteMessageFromDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessageFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessageFromDLQ indicates an expected call of DeleteMessageFromDLQ.\nfunc (mr *MockQueueManagerMockRecorder) DeleteMessageFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessageFromDLQ\", reflect.TypeOf((*MockQueueManager)(nil).DeleteMessageFromDLQ), ctx, request)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MockQueueManager) DeleteMessagesBefore(ctx context.Context, request *DeleteMessagesBeforeRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MockQueueManagerMockRecorder) DeleteMessagesBefore(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MockQueueManager)(nil).DeleteMessagesBefore), ctx, request)\n}\n\n// EnqueueMessage mocks base method.\nfunc (m *MockQueueManager) EnqueueMessage(ctx context.Context, request *EnqueueMessageRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"EnqueueMessage\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// EnqueueMessage indicates an expected call of EnqueueMessage.\nfunc (mr *MockQueueManagerMockRecorder) EnqueueMessage(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"EnqueueMessage\", reflect.TypeOf((*MockQueueManager)(nil).EnqueueMessage), ctx, request)\n}\n\n// EnqueueMessageToDLQ mocks base method.\nfunc (m *MockQueueManager) EnqueueMessageToDLQ(ctx context.Context, request *EnqueueMessageToDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"EnqueueMessageToDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// EnqueueMessageToDLQ indicates an expected call of EnqueueMessageToDLQ.\nfunc (mr *MockQueueManagerMockRecorder) EnqueueMessageToDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"EnqueueMessageToDLQ\", reflect.TypeOf((*MockQueueManager)(nil).EnqueueMessageToDLQ), ctx, request)\n}\n\n// GetAckLevels mocks base method.\nfunc (m *MockQueueManager) GetAckLevels(ctx context.Context, request *GetAckLevelsRequest) (*GetAckLevelsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevels\", ctx, request)\n\tret0, _ := ret[0].(*GetAckLevelsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAckLevels indicates an expected call of GetAckLevels.\nfunc (mr *MockQueueManagerMockRecorder) GetAckLevels(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevels\", reflect.TypeOf((*MockQueueManager)(nil).GetAckLevels), ctx, request)\n}\n\n// GetDLQAckLevels mocks base method.\nfunc (m *MockQueueManager) GetDLQAckLevels(ctx context.Context, request *GetDLQAckLevelsRequest) (*GetDLQAckLevelsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQAckLevels\", ctx, request)\n\tret0, _ := ret[0].(*GetDLQAckLevelsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQAckLevels indicates an expected call of GetDLQAckLevels.\nfunc (mr *MockQueueManagerMockRecorder) GetDLQAckLevels(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQAckLevels\", reflect.TypeOf((*MockQueueManager)(nil).GetDLQAckLevels), ctx, request)\n}\n\n// GetDLQSize mocks base method.\nfunc (m *MockQueueManager) GetDLQSize(ctx context.Context, request *GetDLQSizeRequest) (*GetDLQSizeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQSize\", ctx, request)\n\tret0, _ := ret[0].(*GetDLQSizeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQSize indicates an expected call of GetDLQSize.\nfunc (mr *MockQueueManagerMockRecorder) GetDLQSize(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQSize\", reflect.TypeOf((*MockQueueManager)(nil).GetDLQSize), ctx, request)\n}\n\n// RangeDeleteMessagesFromDLQ mocks base method.\nfunc (m *MockQueueManager) RangeDeleteMessagesFromDLQ(ctx context.Context, request *RangeDeleteMessagesFromDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessagesFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteMessagesFromDLQ indicates an expected call of RangeDeleteMessagesFromDLQ.\nfunc (mr *MockQueueManagerMockRecorder) RangeDeleteMessagesFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessagesFromDLQ\", reflect.TypeOf((*MockQueueManager)(nil).RangeDeleteMessagesFromDLQ), ctx, request)\n}\n\n// ReadMessages mocks base method.\nfunc (m *MockQueueManager) ReadMessages(ctx context.Context, request *ReadMessagesRequest) (*ReadMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadMessages\", ctx, request)\n\tret0, _ := ret[0].(*ReadMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadMessages indicates an expected call of ReadMessages.\nfunc (mr *MockQueueManagerMockRecorder) ReadMessages(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadMessages\", reflect.TypeOf((*MockQueueManager)(nil).ReadMessages), ctx, request)\n}\n\n// ReadMessagesFromDLQ mocks base method.\nfunc (m *MockQueueManager) ReadMessagesFromDLQ(ctx context.Context, request *ReadMessagesFromDLQRequest) (*ReadMessagesFromDLQResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadMessagesFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(*ReadMessagesFromDLQResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadMessagesFromDLQ indicates an expected call of ReadMessagesFromDLQ.\nfunc (mr *MockQueueManagerMockRecorder) ReadMessagesFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadMessagesFromDLQ\", reflect.TypeOf((*MockQueueManager)(nil).ReadMessagesFromDLQ), ctx, request)\n}\n\n// UpdateAckLevel mocks base method.\nfunc (m *MockQueueManager) UpdateAckLevel(ctx context.Context, request *UpdateAckLevelRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevel\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAckLevel indicates an expected call of UpdateAckLevel.\nfunc (mr *MockQueueManagerMockRecorder) UpdateAckLevel(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevel\", reflect.TypeOf((*MockQueueManager)(nil).UpdateAckLevel), ctx, request)\n}\n\n// UpdateDLQAckLevel mocks base method.\nfunc (m *MockQueueManager) UpdateDLQAckLevel(ctx context.Context, request *UpdateDLQAckLevelRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDLQAckLevel\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDLQAckLevel indicates an expected call of UpdateDLQAckLevel.\nfunc (mr *MockQueueManagerMockRecorder) UpdateDLQAckLevel(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDLQAckLevel\", reflect.TypeOf((*MockQueueManager)(nil).UpdateDLQAckLevel), ctx, request)\n}\n\n// MockConfigStoreManager is a mock of ConfigStoreManager interface.\ntype MockConfigStoreManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConfigStoreManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockConfigStoreManagerMockRecorder is the mock recorder for MockConfigStoreManager.\ntype MockConfigStoreManagerMockRecorder struct {\n\tmock *MockConfigStoreManager\n}\n\n// NewMockConfigStoreManager creates a new mock instance.\nfunc NewMockConfigStoreManager(ctrl *gomock.Controller) *MockConfigStoreManager {\n\tmock := &MockConfigStoreManager{ctrl: ctrl}\n\tmock.recorder = &MockConfigStoreManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConfigStoreManager) EXPECT() *MockConfigStoreManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockConfigStoreManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockConfigStoreManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockConfigStoreManager)(nil).Close))\n}\n\n// FetchDynamicConfig mocks base method.\nfunc (m *MockConfigStoreManager) FetchDynamicConfig(ctx context.Context, cfgType ConfigType) (*FetchDynamicConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FetchDynamicConfig\", ctx, cfgType)\n\tret0, _ := ret[0].(*FetchDynamicConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FetchDynamicConfig indicates an expected call of FetchDynamicConfig.\nfunc (mr *MockConfigStoreManagerMockRecorder) FetchDynamicConfig(ctx, cfgType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FetchDynamicConfig\", reflect.TypeOf((*MockConfigStoreManager)(nil).FetchDynamicConfig), ctx, cfgType)\n}\n\n// UpdateDynamicConfig mocks base method.\nfunc (m *MockConfigStoreManager) UpdateDynamicConfig(ctx context.Context, request *UpdateDynamicConfigRequest, cfgType ConfigType) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDynamicConfig\", ctx, request, cfgType)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDynamicConfig indicates an expected call of UpdateDynamicConfig.\nfunc (mr *MockConfigStoreManagerMockRecorder) UpdateDynamicConfig(ctx, request, cfgType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDynamicConfig\", reflect.TypeOf((*MockConfigStoreManager)(nil).UpdateDynamicConfig), ctx, request, cfgType)\n}\n"
  },
  {
    "path": "common/persistence/data_manager_interfaces_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestClusterReplicationConfigGetCopy(t *testing.T) {\n\tconfig := &ClusterReplicationConfig{\n\t\tClusterName: \"test\",\n\t}\n\tassert.Equal(t, config, config.GetCopy()) // deep equal\n\tassert.Equal(t, true, config != config.GetCopy())\n}\n\nfunc TestGetDomainResponseDeepCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tinput    *GetDomainResponse\n\t\tvalidate func(t *testing.T, original, copied *GetDomainResponse)\n\t}{\n\t\t{\n\t\t\tname:  \"nil response\",\n\t\t\tinput: nil,\n\t\t\tvalidate: func(t *testing.T, original, copied *GetDomainResponse) {\n\t\t\t\tassert.Nil(t, copied)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"empty response\",\n\t\t\tinput: &GetDomainResponse{},\n\t\t\tvalidate: func(t *testing.T, original, copied *GetDomainResponse) {\n\t\t\t\tassert.NotNil(t, copied)\n\t\t\t\tassert.Equal(t, original, copied)\n\t\t\t\tassert.True(t, original != copied, \"should be different pointers\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"full populated response\",\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:          \"full-id\",\n\t\t\t\t\tName:        \"full-name\",\n\t\t\t\t\tStatus:      1,\n\t\t\t\t\tDescription: \"full-description\",\n\t\t\t\t\tOwnerEmail:  \"full@example.com\",\n\t\t\t\t\tData:        map[string]string{\"k\": \"v\"},\n\t\t\t\t},\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                30,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"test://visibility\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{},\n\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters:          []*ClusterReplicationConfig{{ClusterName: \"c1\"}},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:              true,\n\t\t\t\tConfigVersion:               100,\n\t\t\t\tFailoverVersion:             200,\n\t\t\t\tFailoverNotificationVersion: 300,\n\t\t\t\tPreviousFailoverVersion:     400,\n\t\t\t\tFailoverEndTime:             func() *int64 { i := int64(500); return &i }(),\n\t\t\t\tLastUpdatedTime:             600,\n\t\t\t\tNotificationVersion:         700,\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, original, copied *GetDomainResponse) {\n\t\t\t\tassert.NotNil(t, copied)\n\t\t\t\tassert.Equal(t, original, copied)\n\t\t\t\t// Verify all nested pointers are different\n\t\t\t\tassert.True(t, original != copied)\n\t\t\t\tassert.True(t, original.Info != copied.Info)\n\t\t\t\tassert.True(t, original.Config != copied.Config)\n\t\t\t\tassert.True(t, original.ReplicationConfig != copied.ReplicationConfig)\n\t\t\t\tassert.True(t, original.FailoverEndTime != copied.FailoverEndTime)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tcopied := tt.input.DeepCopy()\n\t\t\ttt.validate(t, tt.input, copied)\n\t\t})\n\t}\n}\n\nfunc TestGetDomainResponseDeepCopy_FuzzGenerated(t *testing.T) {\n\tfuzzer := testdatagen.New(t)\n\n\t// Run multiple iterations to test with different data\n\tfor i := 0; i < 100; i++ {\n\t\tt.Run(fmt.Sprintf(\"iteration_%d\", i), func(t *testing.T) {\n\t\t\toriginal := &GetDomainResponse{}\n\t\t\tfuzzer.Fuzz(original)\n\n\t\t\t// Create a deep copy\n\t\t\tcopied := original.DeepCopy()\n\n\t\t\t// Verify the copy is not nil\n\t\t\trequire.NotNil(t, copied, \"DeepCopy should not return nil for non-nil input\")\n\n\t\t\t// Verify the copied struct is equal to the original\n\t\t\tassert.Equal(t, original, copied, \"DeepCopy should produce an equal struct\")\n\n\t\t\t// Verify they are different instances (different memory addresses)\n\t\t\tassert.True(t, original != copied, \"DeepCopy should create a new instance\")\n\t\t})\n\t}\n}\n\nfunc TestIsTransientError(t *testing.T) {\n\ttransientErrors := []error{\n\t\t&types.ServiceBusyError{},\n\t\t&types.InternalServiceError{},\n\t\t&TimeoutError{},\n\t}\n\tfor _, err := range transientErrors {\n\t\trequire.True(t, IsTransientError(err))\n\t}\n\n\tnonRetryableErrors := []error{\n\t\t&types.EntityNotExistsError{},\n\t\t&types.DomainAlreadyExistsError{},\n\t\t&WorkflowExecutionAlreadyStartedError{},\n\t\terrors.New(\"some unknown error\"),\n\t}\n\tfor _, err := range nonRetryableErrors {\n\t\trequire.False(t, IsTransientError(err))\n\t}\n}\n\nfunc TestIsTimeoutError(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\terr      error\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname:     \"nil error\",\n\t\t\terr:      nil,\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"regular error\",\n\t\t\terr:      fmt.Errorf(\"not timeout error\"),\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"context.DeadlineExceeded\",\n\t\t\terr:      context.DeadlineExceeded,\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"context.Canceled\",\n\t\t\terr:      context.Canceled,\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"wrapped context.DeadlineExceeded\",\n\t\t\terr:      fmt.Errorf(\"operation failed: %w\", context.DeadlineExceeded),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"wrapped context.Canceled\",\n\t\t\terr:      fmt.Errorf(\"operation failed: %w\", context.Canceled),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"doubly wrapped context.DeadlineExceeded\",\n\t\t\terr:      fmt.Errorf(\"outer: %w\", fmt.Errorf(\"inner: %w\", context.DeadlineExceeded)),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"doubly wrapped context.Canceled\",\n\t\t\terr:      fmt.Errorf(\"outer: %w\", fmt.Errorf(\"inner: %w\", context.Canceled)),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"TimeoutError\",\n\t\t\terr:      &TimeoutError{Msg: \"timeout\"},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"wrapped TimeoutError\",\n\t\t\terr:      fmt.Errorf(\"operation failed: %w\", &TimeoutError{Msg: \"timeout\"}),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"doubly wrapped TimeoutError\",\n\t\t\terr:      fmt.Errorf(\"outer: %w\", fmt.Errorf(\"inner: %w\", &TimeoutError{Msg: \"timeout\"})),\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"empty TimeoutError\",\n\t\t\terr:      &TimeoutError{},\n\t\t\texpected: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult := IsTimeoutError(tc.err)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestIsBackgroundTransientError(t *testing.T) {\n\ttransientErrors := map[error]bool{\n\t\t&types.ServiceBusyError{}:     false,\n\t\t&types.InternalServiceError{}: true,\n\t\t&TimeoutError{}:               true,\n\t}\n\tfor error, result := range transientErrors {\n\t\tassert.Equal(t, result, IsBackgroundTransientError(error))\n\t}\n}\n\nfunc TestNewHistoryBranch(t *testing.T) {\n\tnewTreeID := \"newTreeID\"\n\tgeneralToken, err := NewHistoryBranchToken(newTreeID)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, generalToken)\n\n\tnewBranchID := \"newBranchID\"\n\tbranchToken, err := NewHistoryBranchTokenByBranchID(newTreeID, newBranchID)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, branchToken)\n\n\tanotherBranchToken, err := NewHistoryBranchTokenFromAnother(newBranchID, generalToken)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, anotherBranchToken)\n}\n\nfunc TestHistoryGarbageCleanupInfo(t *testing.T) {\n\ttestDomainID := \"testDomainID\"\n\ttestWorkflowID := \"testWorkflowID\"\n\ttestRunID := \"testRunID\"\n\tactualInfo := BuildHistoryGarbageCleanupInfo(testDomainID, testWorkflowID, testRunID)\n\tsplitDomainID, splitWorkflowID, splitRunID, splitError := SplitHistoryGarbageCleanupInfo(actualInfo)\n\tassert.Equal(t, testDomainID, splitDomainID)\n\tassert.Equal(t, testWorkflowID, splitWorkflowID)\n\tassert.Equal(t, testRunID, splitRunID)\n\tassert.NoError(t, splitError)\n\t_, _, _, splitError = SplitHistoryGarbageCleanupInfo(\"invalidInfo\")\n\tassert.Error(t, splitError)\n}\n\nfunc TestNewGetReplicationTasksFromDLQRequest(t *testing.T) {\n\tsourceCluster := \"testSourceCluster\"\n\treadLevel := int64(1)\n\tmaxReadLevel := int64(2)\n\tbatchSize := 10\n\ttasksRequest := NewGetReplicationTasksFromDLQRequest(sourceCluster, readLevel, maxReadLevel, batchSize, nil)\n\tassert.Equal(t, sourceCluster, tasksRequest.SourceClusterName)\n}\n\nfunc TestHasMoreRowsToDelete(t *testing.T) {\n\tassert.True(t, HasMoreRowsToDelete(10, 10))\n\tassert.False(t, HasMoreRowsToDelete(11, 10))\n\tassert.False(t, HasMoreRowsToDelete(9, 10))\n\tassert.False(t, HasMoreRowsToDelete(-1, 10))\n\tassert.False(t, HasMoreRowsToDelete(100, 0))\n\tassert.False(t, HasMoreRowsToDelete(0, 0))\n\tassert.False(t, HasMoreRowsToDelete(50, -1))\n\n}\nfunc TestTransferTaskInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttask := &TransferTaskInfo{\n\t\tDomainID:            \"test-domain-id\",\n\t\tWorkflowID:          \"test-workflow-id\",\n\t\tRunID:               \"test-run-id\",\n\t\tTargetDomainIDs:     map[string]struct{}{\"test-target-domain-id\": {}},\n\t\tTaskType:            TransferTaskTypeActivityTask,\n\t\tTaskID:              1,\n\t\tScheduleID:          1,\n\t\tVersion:             1,\n\t\tVisibilityTimestamp: timeNow,\n\t}\n\texpectedString := fmt.Sprintf(\"%#v\", task)\n\n\tassert.Equal(t, \"test-domain-id\", task.GetDomainID())\n\tassert.Equal(t, \"test-workflow-id\", task.GetWorkflowID())\n\tassert.Equal(t, \"test-run-id\", task.GetRunID())\n\tassert.Equal(t, TransferTaskTypeActivityTask, task.GetTaskType())\n\tassert.Equal(t, int64(1), task.GetTaskID())\n\tassert.Equal(t, int64(1), task.GetVersion())\n\tassert.Equal(t, timeNow, task.GetVisibilityTimestamp())\n\tassert.Equal(t, map[string]struct{}{\"test-target-domain-id\": {}}, task.GetTargetDomainIDs())\n\tassert.Equal(t, expectedString, task.String())\n}\n\nfunc TestReplicationTaskInfo(t *testing.T) {\n\ttask := &ReplicationTaskInfo{\n\t\tDomainID:     \"test-domain-id\",\n\t\tWorkflowID:   \"test-workflow-id\",\n\t\tRunID:        \"test-run-id\",\n\t\tTaskType:     ReplicationTaskTypeHistory,\n\t\tTaskID:       1,\n\t\tVersion:      1,\n\t\tFirstEventID: 1,\n\t\tNextEventID:  2,\n\t\tScheduledID:  3,\n\t}\n\tassert.Equal(t, \"test-domain-id\", task.GetDomainID())\n\tassert.Equal(t, \"test-workflow-id\", task.GetWorkflowID())\n\tassert.Equal(t, \"test-run-id\", task.GetRunID())\n\tassert.Equal(t, ReplicationTaskTypeHistory, task.GetTaskType())\n\tassert.Equal(t, int64(1), task.GetTaskID())\n\tassert.Equal(t, int64(1), task.GetVersion())\n\tassert.Equal(t, time.Time{}, task.GetVisibilityTimestamp())\n}\n\nfunc TestTimeTaskInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttask := &TimerTaskInfo{\n\t\tVisibilityTimestamp: timeNow,\n\t\tTaskType:            4,\n\t\tTaskID:              3,\n\t\tVersion:             2,\n\t\tRunID:               \"test-run-id\",\n\t\tDomainID:            \"test-domain-id\",\n\t\tWorkflowID:          \"test-workflow-id\",\n\t}\n\texpectedString := fmt.Sprintf(\n\t\t\"{DomainID: %v, WorkflowID: %v, RunID: %v, VisibilityTimestamp: %v, TaskID: %v, TaskType: %v, TimeoutType: %v, EventID: %v, ScheduleAttempt: %v, Version: %v.}\",\n\t\ttask.DomainID, task.WorkflowID, task.RunID, task.VisibilityTimestamp, task.TaskID, task.TaskType, task.TimeoutType, task.EventID, task.ScheduleAttempt, task.Version,\n\t)\n\tassert.Equal(t, 4, task.GetTaskType())\n\tassert.Equal(t, int64(3), task.GetTaskID())\n\tassert.Equal(t, int64(2), task.GetVersion())\n\tassert.Equal(t, timeNow, task.GetVisibilityTimestamp())\n\tassert.Equal(t, \"test-run-id\", task.GetRunID())\n\tassert.Equal(t, \"test-domain-id\", task.GetDomainID())\n\tassert.Equal(t, \"test-workflow-id\", task.GetWorkflowID())\n\tassert.Equal(t, expectedString, task.String())\n}\n\nfunc TestShardInfoCopy(t *testing.T) {\n\tf := fuzz.New().NilChance(0.1)\n\tinfo := &ShardInfo{}\n\tfor i := 0; i < 1000; i++ {\n\t\tf.Fuzz(info)\n\t\tinfoCopy := info.copy()\n\t\tassert.Equal(t, info, infoCopy)\n\t}\n}\n\nfunc TestShardInfoNilSafeCopy(t *testing.T) {\n\tvar info *ShardInfo\n\tinfoCopy := info.ToNilSafeCopy()\n\tassert.NotNil(t, infoCopy)\n\tassert.NotNil(t, infoCopy.ClusterTransferAckLevel)\n\tassert.NotNil(t, infoCopy.ClusterTimerAckLevel)\n\tassert.NotNil(t, infoCopy.ClusterReplicationLevel)\n\tassert.NotNil(t, infoCopy.QueueStates)\n\tassert.NotNil(t, infoCopy.TransferProcessingQueueStates)\n\tassert.NotNil(t, infoCopy.TransferProcessingQueueStates.StatesByCluster)\n\tassert.NotNil(t, infoCopy.TimerProcessingQueueStates)\n\tassert.NotNil(t, infoCopy.TimerProcessingQueueStates.StatesByCluster)\n\tassert.NotNil(t, infoCopy.ReplicationDLQAckLevel)\n\n}\n\nfunc TestSerializeAndDeserializeClusterConfigs(t *testing.T) {\n\tconfigs := []*ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: \"test-cluster1\",\n\t\t},\n\t\t{\n\t\t\tClusterName: \"test-cluster2\",\n\t\t},\n\t}\n\tserializedResult := SerializeClusterConfigs(configs)\n\tdeserializedResult := DeserializeClusterConfigs(serializedResult)\n\n\tassert.Equal(t, configs, deserializedResult)\n\n}\n\nfunc TestTimeStampConvertion(t *testing.T) {\n\ttimeNow := time.Now()\n\tmilisSecond := UnixNanoToDBTimestamp(timeNow.UnixNano())\n\tunixNanoTime := DBTimestampToUnixNano(milisSecond)\n\tassert.Equal(t, timeNow.UnixNano()/(1000*1000), unixNanoTime/(1000*1000)) // unixNano to milisSecond will result in info loss\n}\n\nfunc TestCopyMemo(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tinputMemo      map[string][]byte\n\t\texpectedOutput map[string][]byte\n\t}{\n\t\t{\n\t\t\tname:           \"TC1: Memo is nil\",\n\t\t\tinputMemo:      nil,\n\t\t\texpectedOutput: nil,\n\t\t},\n\t\t{\n\t\t\tname:           \"TC2: Memo is empty\",\n\t\t\tinputMemo:      map[string][]byte{},\n\t\t\texpectedOutput: map[string][]byte{},\n\t\t},\n\t\t{\n\t\t\tname: \"TC3: Memo contains multiple entries\",\n\t\t\tinputMemo: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"val1\"),\n\t\t\t\t\"key2\": []byte(\"val2\"),\n\t\t\t},\n\t\t\texpectedOutput: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"val1\"),\n\t\t\t\t\"key2\": []byte(\"val2\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC4: Memo contains empty byte slices\",\n\t\t\tinputMemo: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"\"),\n\t\t\t\t\"key2\": []byte{},\n\t\t\t},\n\t\t\texpectedOutput: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"\"),\n\t\t\t\t\"key2\": []byte{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC5: Memo contains duplicate byte slices\",\n\t\t\tinputMemo: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"dup\"),\n\t\t\t\t\"key2\": []byte(\"dup\"),\n\t\t\t},\n\t\t\texpectedOutput: map[string][]byte{\n\t\t\t\t\"key1\": []byte(\"dup\"),\n\t\t\t\t\"key2\": []byte(\"dup\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\texecInfo := &WorkflowExecutionInfo{\n\t\t\t\tMemo: tt.inputMemo,\n\t\t\t}\n\n\t\t\tcopyMemo := execInfo.CopyMemo()\n\n\t\t\t// Check if both are nil\n\t\t\tif tt.expectedOutput == nil {\n\t\t\t\tif copyMemo != nil {\n\t\t\t\t\tt.Errorf(\"Expected nil, got %v\", copyMemo)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check if lengths match\n\t\t\tif len(copyMemo) != len(tt.expectedOutput) {\n\t\t\t\tt.Errorf(\"Expected map length %d, got %d\", len(tt.expectedOutput), len(copyMemo))\n\t\t\t}\n\n\t\t\t// Check each key-value pair\n\t\t\tfor key, expectedVal := range tt.expectedOutput {\n\t\t\t\tcopiedVal, exists := copyMemo[key]\n\t\t\t\tif !exists {\n\t\t\t\t\tt.Errorf(\"Expected key %s not found in copy\", key)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\n\t\t\t\tif !bytes.Equal(copiedVal, expectedVal) {\n\t\t\t\t\tt.Errorf(\"For key %s, expected value %v, got %v\", key, expectedVal, copiedVal)\n\t\t\t\t}\n\n\t\t\t\t// Ensure that the byte slices are different underlying arrays (deep copy)\n\t\t\t\tif len(copiedVal) > 0 && &copiedVal[0] == &expectedVal[0] {\n\t\t\t\t\tt.Errorf(\"For key %s, byte slices reference the same underlying array\", key)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCopySearchAttributes(t *testing.T) {\n\ttests := []struct {\n\t\tname                string\n\t\tinputSearchAttrs    map[string][]byte\n\t\texpectedOutputAttrs map[string][]byte\n\t}{\n\t\t{\n\t\t\tname:                \"TC1: SearchAttributes is nil\",\n\t\t\tinputSearchAttrs:    nil,\n\t\t\texpectedOutputAttrs: nil,\n\t\t},\n\t\t{\n\t\t\tname:                \"TC2: SearchAttributes is empty\",\n\t\t\tinputSearchAttrs:    map[string][]byte{},\n\t\t\texpectedOutputAttrs: map[string][]byte{},\n\t\t},\n\t\t{\n\t\t\tname: \"TC3: SearchAttributes contains multiple entries\",\n\t\t\tinputSearchAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"value1\"),\n\t\t\t\t\"attr2\": []byte(\"value2\"),\n\t\t\t},\n\t\t\texpectedOutputAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"value1\"),\n\t\t\t\t\"attr2\": []byte(\"value2\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC4: SearchAttributes contains empty byte slices\",\n\t\t\tinputSearchAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"\"),\n\t\t\t\t\"attr2\": []byte{},\n\t\t\t},\n\t\t\texpectedOutputAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"\"),\n\t\t\t\t\"attr2\": []byte{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC5: SearchAttributes contains duplicate byte slices\",\n\t\t\tinputSearchAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"dup\"),\n\t\t\t\t\"attr2\": []byte(\"dup\"),\n\t\t\t},\n\t\t\texpectedOutputAttrs: map[string][]byte{\n\t\t\t\t\"attr1\": []byte(\"dup\"),\n\t\t\t\t\"attr2\": []byte(\"dup\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\texecInfo := &WorkflowExecutionInfo{\n\t\t\t\tSearchAttributes: tt.inputSearchAttrs,\n\t\t\t}\n\n\t\t\tcopyAttrs := execInfo.CopySearchAttributes()\n\n\t\t\t// Check if both are nil\n\t\t\tif tt.expectedOutputAttrs == nil {\n\t\t\t\tif copyAttrs != nil {\n\t\t\t\t\tt.Errorf(\"Expected nil, got %v\", copyAttrs)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check if lengths match\n\t\t\tif len(copyAttrs) != len(tt.expectedOutputAttrs) {\n\t\t\t\tt.Errorf(\"Expected map length %d, got %d\", len(tt.expectedOutputAttrs), len(copyAttrs))\n\t\t\t}\n\n\t\t\t// Check each key-value pair\n\t\t\tfor key, expectedVal := range tt.expectedOutputAttrs {\n\t\t\t\tcopiedVal, exists := copyAttrs[key]\n\t\t\t\tif !exists {\n\t\t\t\t\tt.Errorf(\"Expected key %s not found in copy\", key)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\n\t\t\t\tif !bytes.Equal(copiedVal, expectedVal) {\n\t\t\t\t\tt.Errorf(\"For key %s, expected value %v, got %v\", key, expectedVal, copiedVal)\n\t\t\t\t}\n\n\t\t\t\t// Ensure that the byte slices are different underlying arrays (deep copy)\n\t\t\t\tif len(copiedVal) > 0 && &copiedVal[0] == &expectedVal[0] {\n\t\t\t\t\tt.Errorf(\"For key %s, byte slices reference the same underlying array\", key)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCopyPartitionConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\tinputPartitionConfig map[string]string\n\t\texpectedOutputConfig map[string]string\n\t}{\n\t\t{\n\t\t\tname:                 \"TC1: PartitionConfig is nil\",\n\t\t\tinputPartitionConfig: nil,\n\t\t\texpectedOutputConfig: nil,\n\t\t},\n\t\t{\n\t\t\tname:                 \"TC2: PartitionConfig is empty\",\n\t\t\tinputPartitionConfig: map[string]string{},\n\t\t\texpectedOutputConfig: map[string]string{},\n\t\t},\n\t\t{\n\t\t\tname: \"TC3: PartitionConfig contains multiple entries\",\n\t\t\tinputPartitionConfig: map[string]string{\n\t\t\t\t\"partition1\": \"config1\",\n\t\t\t\t\"partition2\": \"config2\",\n\t\t\t},\n\t\t\texpectedOutputConfig: map[string]string{\n\t\t\t\t\"partition1\": \"config1\",\n\t\t\t\t\"partition2\": \"config2\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC4: PartitionConfig contains empty strings\",\n\t\t\tinputPartitionConfig: map[string]string{\n\t\t\t\t\"partition1\": \"\",\n\t\t\t\t\"partition2\": \"\",\n\t\t\t},\n\t\t\texpectedOutputConfig: map[string]string{\n\t\t\t\t\"partition1\": \"\",\n\t\t\t\t\"partition2\": \"\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TC5: PartitionConfig contains duplicate values\",\n\t\t\tinputPartitionConfig: map[string]string{\n\t\t\t\t\"partition1\": \"dup\",\n\t\t\t\t\"partition2\": \"dup\",\n\t\t\t},\n\t\t\texpectedOutputConfig: map[string]string{\n\t\t\t\t\"partition1\": \"dup\",\n\t\t\t\t\"partition2\": \"dup\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\texecInfo := &WorkflowExecutionInfo{\n\t\t\t\tPartitionConfig: tt.inputPartitionConfig,\n\t\t\t}\n\n\t\t\tcopyConfig := execInfo.CopyPartitionConfig()\n\n\t\t\t// Check if both are nil\n\t\t\tif tt.expectedOutputConfig == nil {\n\t\t\t\tif copyConfig != nil {\n\t\t\t\t\tt.Errorf(\"Expected nil, got %v\", copyConfig)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check if lengths match\n\t\t\tif len(copyConfig) != len(tt.expectedOutputConfig) {\n\t\t\t\tt.Errorf(\"Expected map length %d, got %d\", len(tt.expectedOutputConfig), len(copyConfig))\n\t\t\t}\n\n\t\t\t// Check each key-value pair\n\t\t\tfor key, expectedVal := range tt.expectedOutputConfig {\n\t\t\t\tcopiedVal, exists := copyConfig[key]\n\t\t\t\tif !exists {\n\t\t\t\t\tt.Errorf(\"Expected key %s not found in copy\", key)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\n\t\t\t\tif copiedVal != expectedVal {\n\t\t\t\t\tt.Errorf(\"For key %s, expected value %s, got %s\", key, expectedVal, copiedVal)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Since strings are immutable in Go, no need to check underlying references\n\t\t})\n\t}\n}\n\nfunc TestTaskListPartitionConfigToInternalType(t *testing.T) {\n\ttestCases := []struct {\n\t\tname   string\n\t\tinput  *TaskListPartitionConfig\n\t\texpect *types.TaskListPartitionConfig\n\t}{\n\t\t{\n\t\t\tname:   \"nil case\",\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t{\n\t\t\tname:   \"empty case\",\n\t\t\tinput:  &TaskListPartitionConfig{},\n\t\t\texpect: &types.TaskListPartitionConfig{},\n\t\t},\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tinput: &TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t\t1: {},\n\t\t\t\t\t2: {},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpect: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t\t1: {},\n\t\t\t\t\t2: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expect, tc.input.ToInternalType())\n\t\t})\n\t}\n}\n\nfunc TestVersionHistoryCopy(t *testing.T) {\n\ta := VersionHistories{\n\t\tCurrentVersionHistoryIndex: 1,\n\t\tHistories: []*VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tassert.Equal(t, &a, a.Duplicate())\n}\n\nfunc TestDomainReplicationConfig_IsActiveActive(t *testing.T) {\n\ttests := []struct {\n\t\tname   string\n\t\tconfig *DomainReplicationConfig\n\t\twant   bool\n\t}{\n\t\t{\n\t\t\tname:   \"nil DomainReplicationConfig should return false\",\n\t\t\tconfig: nil,\n\t\t\twant:   false,\n\t\t},\n\t\t{\n\t\t\tname:   \"nil ActiveClusters should return false\",\n\t\t\tconfig: &DomainReplicationConfig{ActiveClusters: nil},\n\t\t\twant:   false,\n\t\t},\n\t\t{\n\t\t\tname: \"empty ActiveClusters should return false\",\n\t\t\tconfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"only AttributeScopes populated should return true\",\n\t\t\tconfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"both formats populated should return true\",\n\t\t\tconfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"empty AttributeScopes map should return false\",\n\t\t\tconfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"AttributeScopes with empty scope should return false\",\n\t\t\tconfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.config.IsActiveActive()\n\t\t\tif got != tt.want {\n\t\t\t\tt.Errorf(\"IsActiveActive() = %v, want %v\", got, tt.want)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/data_store_interfaces.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination data_store_interfaces_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence ExecutionStore,ShardStore,DomainStore,TaskStore,HistoryStore,ConfigStore,DomainAuditStore\n//go:generate mockgen -package $GOPACKAGE -destination visibility_store_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence VisibilityStore\n\ntype (\n\t// ////////////////////////////////////////////////////////////////////\n\t// Persistence interface is a lower layer of dataInterface.\n\t// The intention is to let different persistence implementation(SQL,Cassandra/etc) share some common logic\n\t// Right now the only common part is serialization/deserialization, and only ExecutionManager/HistoryManager need it.\n\t// TaskManager are the same.\n\t// ////////////////////////////////////////////////////////////////////\n\n\t// ShardStore is the lower level of ShardManager\n\tShardStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateShard(ctx context.Context, request *InternalCreateShardRequest) error\n\t\tGetShard(ctx context.Context, request *InternalGetShardRequest) (*InternalGetShardResponse, error)\n\t\tUpdateShard(ctx context.Context, request *InternalUpdateShardRequest) error\n\t}\n\n\t// TaskStore is a lower level of TaskManager\n\tTaskStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tLeaseTaskList(ctx context.Context, request *LeaseTaskListRequest) (*LeaseTaskListResponse, error)\n\t\tGetTaskList(ctx context.Context, request *GetTaskListRequest) (*GetTaskListResponse, error)\n\t\tUpdateTaskList(ctx context.Context, request *UpdateTaskListRequest) (*UpdateTaskListResponse, error)\n\t\tListTaskList(ctx context.Context, request *ListTaskListRequest) (*ListTaskListResponse, error)\n\t\tDeleteTaskList(ctx context.Context, request *DeleteTaskListRequest) error\n\t\tGetTaskListSize(ctx context.Context, request *GetTaskListSizeRequest) (*GetTaskListSizeResponse, error)\n\t\tCreateTasks(ctx context.Context, request *CreateTasksRequest) (*CreateTasksResponse, error)\n\t\tGetTasks(ctx context.Context, request *GetTasksRequest) (*GetTasksResponse, error)\n\t\tCompleteTask(ctx context.Context, request *CompleteTaskRequest) error\n\t\t// CompleteTasksLessThan completes tasks less than or equal to the given task id\n\t\t// This API takes a limit parameter which specifies the count of maxRows that\n\t\t// can be deleted. This parameter may be ignored by the underlying storage, but\n\t\t// its mandatory to specify it. On success this method returns the number of rows\n\t\t// actually deleted. If the underlying storage doesn't support \"limit\", all rows\n\t\t// less than or equal to taskID will be deleted.\n\t\t// On success, this method returns:\n\t\t//  - number of rows actually deleted, if limit is honored\n\t\t//  - UnknownNumRowsDeleted, when all rows below value are deleted\n\t\tCompleteTasksLessThan(ctx context.Context, request *CompleteTasksLessThanRequest) (*CompleteTasksLessThanResponse, error)\n\t\t// GetOrphanTasks returns tasks that exist as records in the database but are part of task lists which\n\t\t// _do not_ exist in the database. They are therefore unreachable and no longer represent valid items\n\t\t// that can be legitimately acted upon.\n\t\tGetOrphanTasks(ctx context.Context, request *GetOrphanTasksRequest) (*GetOrphanTasksResponse, error)\n\t}\n\n\t// DomainStore is a lower level of DomainManager\n\tDomainStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateDomain(ctx context.Context, request *InternalCreateDomainRequest) (*CreateDomainResponse, error)\n\t\tGetDomain(ctx context.Context, request *GetDomainRequest) (*InternalGetDomainResponse, error)\n\t\tUpdateDomain(ctx context.Context, request *InternalUpdateDomainRequest) error\n\t\tDeleteDomain(ctx context.Context, request *DeleteDomainRequest) error\n\t\tDeleteDomainByName(ctx context.Context, request *DeleteDomainByNameRequest) error\n\t\tListDomains(ctx context.Context, request *ListDomainsRequest) (*InternalListDomainsResponse, error)\n\t\tGetMetadata(ctx context.Context) (*GetMetadataResponse, error)\n\t}\n\n\t// DomainAuditStore is a lower level of DomainAuditManager\n\tDomainAuditStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tCreateDomainAuditLog(ctx context.Context, request *InternalCreateDomainAuditLogRequest) (*CreateDomainAuditLogResponse, error)\n\t\tGetDomainAuditLogs(ctx context.Context, request *GetDomainAuditLogsRequest) (*InternalGetDomainAuditLogsResponse, error)\n\t}\n\n\t// ExecutionStore is used to manage workflow executions for Persistence layer\n\tExecutionStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tGetShardID() int\n\t\t// The below three APIs are related to serialization/deserialization\n\t\tGetWorkflowExecution(ctx context.Context, request *InternalGetWorkflowExecutionRequest) (*InternalGetWorkflowExecutionResponse, error)\n\t\tUpdateWorkflowExecution(ctx context.Context, request *InternalUpdateWorkflowExecutionRequest) error\n\t\tConflictResolveWorkflowExecution(ctx context.Context, request *InternalConflictResolveWorkflowExecutionRequest) error\n\n\t\tCreateWorkflowExecution(ctx context.Context, request *InternalCreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error)\n\t\tDeleteWorkflowExecution(ctx context.Context, request *DeleteWorkflowExecutionRequest) error\n\t\tDeleteCurrentWorkflowExecution(ctx context.Context, request *DeleteCurrentWorkflowExecutionRequest) error\n\t\tGetCurrentExecution(ctx context.Context, request *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error)\n\t\tIsWorkflowExecutionExists(ctx context.Context, request *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error)\n\n\t\t// Replication task related methods\n\t\tPutReplicationTaskToDLQ(ctx context.Context, request *InternalPutReplicationTaskToDLQRequest) error\n\t\tGetReplicationTasksFromDLQ(ctx context.Context, request *GetReplicationTasksFromDLQRequest) (*GetHistoryTasksResponse, error)\n\t\tGetReplicationDLQSize(ctx context.Context, request *GetReplicationDLQSizeRequest) (*GetReplicationDLQSizeResponse, error)\n\t\tDeleteReplicationTaskFromDLQ(ctx context.Context, request *DeleteReplicationTaskFromDLQRequest) error\n\t\tRangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *RangeDeleteReplicationTaskFromDLQRequest) (*RangeDeleteReplicationTaskFromDLQResponse, error)\n\t\tCreateFailoverMarkerTasks(ctx context.Context, request *CreateFailoverMarkersRequest) error\n\n\t\t// History task related methods\n\t\tGetHistoryTasks(ctx context.Context, request *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error)\n\t\tCompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error\n\t\tRangeCompleteHistoryTask(ctx context.Context, request *RangeCompleteHistoryTaskRequest) (*RangeCompleteHistoryTaskResponse, error)\n\n\t\t// Scan related methods\n\t\tListConcreteExecutions(ctx context.Context, request *ListConcreteExecutionsRequest) (*InternalListConcreteExecutionsResponse, error)\n\t\tListCurrentExecutions(ctx context.Context, request *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error)\n\n\t\t// Active cluster selection policy related methods\n\t\tGetActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*DataBlob, error)\n\t\tDeleteActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) error\n\t}\n\n\t// HistoryStore is to manager workflow history events\n\tHistoryStore interface {\n\t\tCloseable\n\t\tGetName() string\n\n\t\t// The below are history V2 APIs\n\t\t// V2 regards history events growing as a tree, decoupled from workflow concepts\n\n\t\t// AppendHistoryNodes add(or override) a node to a history branch\n\t\tAppendHistoryNodes(ctx context.Context, request *InternalAppendHistoryNodesRequest) error\n\t\t// ReadHistoryBranch returns history node data for a branch\n\t\tReadHistoryBranch(ctx context.Context, request *InternalReadHistoryBranchRequest) (*InternalReadHistoryBranchResponse, error)\n\t\t// ForkHistoryBranch forks a new branch from a old branch\n\t\tForkHistoryBranch(ctx context.Context, request *InternalForkHistoryBranchRequest) (*InternalForkHistoryBranchResponse, error)\n\t\t// DeleteHistoryBranch removes a branch\n\t\tDeleteHistoryBranch(ctx context.Context, request *InternalDeleteHistoryBranchRequest) error\n\t\t// GetHistoryTree returns all branch information of a tree\n\t\tGetHistoryTree(ctx context.Context, request *InternalGetHistoryTreeRequest) (*InternalGetHistoryTreeResponse, error)\n\t\t// GetAllHistoryTreeBranches returns all branches of all trees\n\t\tGetAllHistoryTreeBranches(ctx context.Context, request *GetAllHistoryTreeBranchesRequest) (*GetAllHistoryTreeBranchesResponse, error)\n\t}\n\n\t// VisibilityStore is the store interface for visibility\n\tVisibilityStore interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tRecordWorkflowExecutionStarted(ctx context.Context, request *InternalRecordWorkflowExecutionStartedRequest) error\n\t\tRecordWorkflowExecutionClosed(ctx context.Context, request *InternalRecordWorkflowExecutionClosedRequest) error\n\t\tRecordWorkflowExecutionUninitialized(ctx context.Context, request *InternalRecordWorkflowExecutionUninitializedRequest) error\n\t\tUpsertWorkflowExecution(ctx context.Context, request *InternalUpsertWorkflowExecutionRequest) error\n\t\tListOpenWorkflowExecutions(ctx context.Context, request *InternalListWorkflowExecutionsRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutions(ctx context.Context, request *InternalListWorkflowExecutionsRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListOpenWorkflowExecutionsByType(ctx context.Context, request *InternalListWorkflowExecutionsByTypeRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByType(ctx context.Context, request *InternalListWorkflowExecutionsByTypeRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *InternalListWorkflowExecutionsByWorkflowIDRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *InternalListWorkflowExecutionsByWorkflowIDRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByStatus(ctx context.Context, request *InternalListClosedWorkflowExecutionsByStatusRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tGetClosedWorkflowExecution(ctx context.Context, request *InternalGetClosedWorkflowExecutionRequest) (*InternalGetClosedWorkflowExecutionResponse, error)\n\t\tDeleteWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error\n\t\tListWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tScanWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*InternalListWorkflowExecutionsResponse, error)\n\t\tCountWorkflowExecutions(ctx context.Context, request *CountWorkflowExecutionsRequest) (*CountWorkflowExecutionsResponse, error)\n\t\tDeleteUninitializedWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error\n\t}\n\n\tConfigStore interface {\n\t\tCloseable\n\t\tFetchConfig(ctx context.Context, configType ConfigType) (*InternalConfigStoreEntry, error)\n\t\tUpdateConfig(ctx context.Context, value *InternalConfigStoreEntry) error\n\t}\n\n\tInternalConfigStoreEntry struct {\n\t\tRowType   int\n\t\tVersion   int64\n\t\tTimestamp time.Time\n\t\tValues    *DataBlob\n\t}\n\n\tInternalEnqueueMessageRequest struct {\n\t\tMessagePayload   []byte\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\tInternalReadMessagesRequest struct {\n\t\tLastMessageID int64\n\t\tMaxCount      int\n\t}\n\n\tInternalReadMessagesResponse struct {\n\t\tMessages []*InternalQueueMessage\n\t}\n\n\tInternalDeleteMessagesBeforeRequest struct {\n\t\tMessageID int64\n\t}\n\n\tInternalUpdateAckLevelRequest struct {\n\t\tMessageID        int64\n\t\tClusterName      string\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\tInternalGetAckLevelsRequest struct{}\n\n\tInternalGetAckLevelsResponse struct {\n\t\tAckLevels map[string]int64\n\t}\n\n\tInternalEnqueueMessageToDLQRequest struct {\n\t\tMessagePayload   []byte\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\tInternalReadMessagesFromDLQRequest struct {\n\t\tFirstMessageID int64\n\t\tLastMessageID  int64\n\t\tPageSize       int\n\t\tPageToken      []byte\n\t}\n\n\tInternalReadMessagesFromDLQResponse struct {\n\t\tMessages      []*InternalQueueMessage\n\t\tNextPageToken []byte\n\t}\n\n\tInternalDeleteMessageFromDLQRequest struct {\n\t\tMessageID int64\n\t}\n\n\tInternalRangeDeleteMessagesFromDLQRequest struct {\n\t\tFirstMessageID int64\n\t\tLastMessageID  int64\n\t}\n\n\tInternalUpdateDLQAckLevelRequest struct {\n\t\tMessageID        int64\n\t\tClusterName      string\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\tInternalGetDLQAckLevelsRequest struct{}\n\n\tInternalGetDLQAckLevelsResponse struct {\n\t\tAckLevels map[string]int64\n\t}\n\n\tInternalGetDLQSizeRequest struct{}\n\n\tInternalGetDLQSizeResponse struct {\n\t\tSize int64\n\t}\n\n\t// QueueStore is a store to enqueue and get messages\n\tQueueStore interface {\n\t\tCloseable\n\t\tEnqueueMessage(ctx context.Context, request *InternalEnqueueMessageRequest) error\n\t\tReadMessages(ctx context.Context, request *InternalReadMessagesRequest) (*InternalReadMessagesResponse, error)\n\t\tDeleteMessagesBefore(ctx context.Context, request *InternalDeleteMessagesBeforeRequest) error\n\t\tUpdateAckLevel(ctx context.Context, request *InternalUpdateAckLevelRequest) error\n\t\tGetAckLevels(ctx context.Context, request *InternalGetAckLevelsRequest) (*InternalGetAckLevelsResponse, error)\n\t\tEnqueueMessageToDLQ(ctx context.Context, request *InternalEnqueueMessageToDLQRequest) error\n\t\tReadMessagesFromDLQ(ctx context.Context, request *InternalReadMessagesFromDLQRequest) (*InternalReadMessagesFromDLQResponse, error)\n\t\tDeleteMessageFromDLQ(ctx context.Context, request *InternalDeleteMessageFromDLQRequest) error\n\t\tRangeDeleteMessagesFromDLQ(ctx context.Context, request *InternalRangeDeleteMessagesFromDLQRequest) error\n\t\tUpdateDLQAckLevel(ctx context.Context, request *InternalUpdateDLQAckLevelRequest) error\n\t\tGetDLQAckLevels(ctx context.Context, request *InternalGetDLQAckLevelsRequest) (*InternalGetDLQAckLevelsResponse, error)\n\t\tGetDLQSize(ctx context.Context, request *InternalGetDLQSizeRequest) (*InternalGetDLQSizeResponse, error)\n\t}\n\n\t// InternalQueueMessage is the message that stores in the queue\n\tInternalQueueMessage struct {\n\t\tID        int64     `json:\"message_id\"`\n\t\tQueueType QueueType `json:\"queue_type\"`\n\t\tPayload   []byte    `json:\"message_payload\"`\n\t}\n\n\t// DataBlob represents a blob for any binary data.\n\t// It contains raw data, and metadata(right now only encoding) in other field\n\t// Note that it should be only used for Persistence layer, below dataInterface and application(historyEngine/etc)\n\tDataBlob struct {\n\t\tEncoding constants.EncodingType\n\t\tData     []byte\n\t}\n\n\t// InternalCreateWorkflowExecutionRequest is used to write a new workflow execution\n\tInternalCreateWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode CreateWorkflowMode\n\n\t\tPreviousRunID            string\n\t\tPreviousLastWriteVersion int64\n\n\t\tNewWorkflowSnapshot InternalWorkflowSnapshot\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalPutReplicationTaskToDLQRequest is used to put a replication task to dlq\n\tInternalPutReplicationTaskToDLQRequest struct {\n\t\tSourceClusterName string\n\t\tTaskInfo          *InternalReplicationTaskInfo\n\t}\n\n\t// InternalReplicationTaskInfo describes the replication task created for replication of history events\n\tInternalReplicationTaskInfo struct {\n\t\tDomainID          string\n\t\tWorkflowID        string\n\t\tRunID             string\n\t\tTaskID            int64\n\t\tTaskType          int\n\t\tFirstEventID      int64\n\t\tNextEventID       int64\n\t\tVersion           int64\n\t\tScheduledID       int64\n\t\tBranchToken       []byte\n\t\tNewRunBranchToken []byte\n\t\tCreationTime      time.Time\n\t\tCurrentTimeStamp  time.Time\n\t}\n\n\t// InternalWorkflowExecutionInfo describes a workflow execution for Persistence Interface\n\tInternalWorkflowExecutionInfo struct {\n\t\tDomainID                           string\n\t\tWorkflowID                         string\n\t\tRunID                              string\n\t\tFirstExecutionRunID                string\n\t\tParentDomainID                     string\n\t\tParentWorkflowID                   string\n\t\tParentRunID                        string\n\t\tInitiatedID                        int64\n\t\tCompletionEventBatchID             int64\n\t\tCompletionEvent                    *DataBlob\n\t\tTaskList                           string\n\t\tTaskListKind                       types.TaskListKind\n\t\tWorkflowTypeName                   string\n\t\tWorkflowTimeout                    time.Duration\n\t\tDecisionStartToCloseTimeout        time.Duration\n\t\tExecutionContext                   []byte\n\t\tState                              int\n\t\tCloseStatus                        int\n\t\tLastFirstEventID                   int64\n\t\tLastEventTaskID                    int64\n\t\tNextEventID                        int64\n\t\tLastProcessedEvent                 int64\n\t\tStartTimestamp                     time.Time\n\t\tLastUpdatedTimestamp               time.Time\n\t\tCreateRequestID                    string\n\t\tSignalCount                        int32\n\t\tDecisionVersion                    int64\n\t\tDecisionScheduleID                 int64\n\t\tDecisionStartedID                  int64\n\t\tDecisionRequestID                  string\n\t\tDecisionTimeout                    time.Duration\n\t\tDecisionAttempt                    int64\n\t\tDecisionStartedTimestamp           time.Time\n\t\tDecisionScheduledTimestamp         time.Time\n\t\tDecisionOriginalScheduledTimestamp time.Time\n\t\tCancelRequested                    bool\n\t\tCancelRequestID                    string\n\t\tStickyTaskList                     string\n\t\tStickyScheduleToStartTimeout       time.Duration\n\t\tClientLibraryVersion               string\n\t\tClientFeatureVersion               string\n\t\tClientImpl                         string\n\t\tAutoResetPoints                    *DataBlob\n\t\t// for retry\n\t\tAttempt            int32\n\t\tHasRetryPolicy     bool\n\t\tInitialInterval    time.Duration\n\t\tBackoffCoefficient float64\n\t\tMaximumInterval    time.Duration\n\t\tExpirationTime     time.Time\n\t\tMaximumAttempts    int32\n\t\tNonRetriableErrors []string\n\t\tBranchToken        []byte\n\t\tCronSchedule       string\n\t\tCronOverlapPolicy  types.CronOverlapPolicy\n\t\tExpirationInterval time.Duration\n\t\tMemo               map[string][]byte\n\t\tSearchAttributes   map[string][]byte\n\t\tPartitionConfig    map[string]string\n\n\t\tActiveClusterSelectionPolicy *DataBlob\n\n\t\t// attributes which are not related to mutable state at all\n\t\tHistorySize int64\n\t\tIsCron      bool\n\t}\n\n\t// InternalWorkflowMutableState indicates workflow related state for Persistence Interface\n\tInternalWorkflowMutableState struct {\n\t\tExecutionInfo    *InternalWorkflowExecutionInfo\n\t\tVersionHistories *DataBlob\n\t\tReplicationState *ReplicationState // TODO: remove this after all 2DC workflows complete\n\t\tActivityInfos    map[int64]*InternalActivityInfo\n\n\t\tTimerInfos          map[string]*TimerInfo\n\t\tChildExecutionInfos map[int64]*InternalChildExecutionInfo\n\t\tRequestCancelInfos  map[int64]*RequestCancelInfo\n\t\tSignalInfos         map[int64]*SignalInfo\n\t\tSignalRequestedIDs  map[string]struct{}\n\t\tBufferedEvents      []*DataBlob\n\n\t\t// Checksum field is used by Cassandra storage\n\t\t// ChecksumData is used by All SQL storage\n\t\tChecksum     checksum.Checksum\n\t\tChecksumData *DataBlob\n\t}\n\n\t// InternalActivityInfo details  for Persistence Interface\n\tInternalActivityInfo struct {\n\t\tVersion                  int64\n\t\tScheduleID               int64\n\t\tScheduledEventBatchID    int64\n\t\tScheduledEvent           *DataBlob\n\t\tScheduledTime            time.Time\n\t\tStartedID                int64\n\t\tStartedEvent             *DataBlob\n\t\tStartedTime              time.Time\n\t\tActivityID               string\n\t\tRequestID                string\n\t\tDetails                  []byte\n\t\tScheduleToStartTimeout   time.Duration\n\t\tScheduleToCloseTimeout   time.Duration\n\t\tStartToCloseTimeout      time.Duration\n\t\tHeartbeatTimeout         time.Duration\n\t\tCancelRequested          bool\n\t\tCancelRequestID          int64\n\t\tLastHeartBeatUpdatedTime time.Time\n\t\tTimerTaskStatus          int32\n\t\t// For retry\n\t\tAttempt            int32\n\t\tDomainID           string\n\t\tStartedIdentity    string\n\t\tTaskList           string\n\t\tTaskListKind       types.TaskListKind\n\t\tHasRetryPolicy     bool\n\t\tInitialInterval    time.Duration\n\t\tBackoffCoefficient float64\n\t\tMaximumInterval    time.Duration\n\t\tExpirationTime     time.Time\n\t\tMaximumAttempts    int32\n\t\tNonRetriableErrors []string\n\t\tLastFailureReason  string\n\t\tLastWorkerIdentity string\n\t\tLastFailureDetails []byte\n\t\t// Not written to database - This is used only for deduping heartbeat timer creation\n\t\tLastHeartbeatTimeoutVisibilityInSeconds int64\n\t}\n\n\t// InternalChildExecutionInfo has details for pending child executions for Persistence Interface\n\tInternalChildExecutionInfo struct {\n\t\tVersion               int64\n\t\tInitiatedID           int64\n\t\tInitiatedEventBatchID int64\n\t\tInitiatedEvent        *DataBlob\n\t\tStartedID             int64\n\t\tStartedWorkflowID     string\n\t\tStartedRunID          string\n\t\tStartedEvent          *DataBlob\n\t\tCreateRequestID       string\n\t\tDomainID              string\n\t\tDomainNameDEPRECATED  string // deprecated: use DomainID field\n\t\tWorkflowTypeName      string\n\t\tParentClosePolicy     types.ParentClosePolicy\n\t}\n\n\t// InternalUpdateWorkflowExecutionRequest is used to update a workflow execution for Persistence Interface\n\tInternalUpdateWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode UpdateWorkflowMode\n\n\t\tUpdateWorkflowMutation InternalWorkflowMutation\n\n\t\tNewWorkflowSnapshot *InternalWorkflowSnapshot\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalConflictResolveWorkflowExecutionRequest is used to reset workflow execution state for Persistence Interface\n\tInternalConflictResolveWorkflowExecutionRequest struct {\n\t\tRangeID int64\n\n\t\tMode ConflictResolveWorkflowMode\n\n\t\t// workflow to be resetted\n\t\tResetWorkflowSnapshot InternalWorkflowSnapshot\n\n\t\t// maybe new workflow\n\t\tNewWorkflowSnapshot *InternalWorkflowSnapshot\n\n\t\t// current workflow\n\t\tCurrentWorkflowMutation *InternalWorkflowMutation\n\n\t\tWorkflowRequestMode CreateWorkflowRequestMode\n\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalWorkflowMutation is used as generic workflow execution state mutation for Persistence Interface\n\tInternalWorkflowMutation struct {\n\t\tExecutionInfo    *InternalWorkflowExecutionInfo\n\t\tVersionHistories *DataBlob\n\t\tStartVersion     int64\n\t\tLastWriteVersion int64\n\n\t\tUpsertActivityInfos       []*InternalActivityInfo\n\t\tDeleteActivityInfos       []int64\n\t\tUpsertTimerInfos          []*TimerInfo\n\t\tDeleteTimerInfos          []string\n\t\tUpsertChildExecutionInfos []*InternalChildExecutionInfo\n\t\tDeleteChildExecutionInfos []int64\n\t\tUpsertRequestCancelInfos  []*RequestCancelInfo\n\t\tDeleteRequestCancelInfos  []int64\n\t\tUpsertSignalInfos         []*SignalInfo\n\t\tDeleteSignalInfos         []int64\n\t\tUpsertSignalRequestedIDs  []string\n\t\tDeleteSignalRequestedIDs  []string\n\t\tNewBufferedEvents         *DataBlob\n\t\tClearBufferedEvents       bool\n\n\t\tTasksByCategory map[HistoryTaskCategory][]Task\n\n\t\tWorkflowRequests []*WorkflowRequest\n\n\t\tCondition int64\n\n\t\tChecksum     checksum.Checksum\n\t\tChecksumData *DataBlob\n\t}\n\n\t// InternalWorkflowSnapshot is used as generic workflow execution state snapshot for Persistence Interface\n\tInternalWorkflowSnapshot struct {\n\t\tExecutionInfo    *InternalWorkflowExecutionInfo\n\t\tVersionHistories *DataBlob\n\t\tStartVersion     int64\n\t\tLastWriteVersion int64\n\n\t\tActivityInfos       []*InternalActivityInfo\n\t\tTimerInfos          []*TimerInfo\n\t\tChildExecutionInfos []*InternalChildExecutionInfo\n\t\tRequestCancelInfos  []*RequestCancelInfo\n\t\tSignalInfos         []*SignalInfo\n\t\tSignalRequestedIDs  []string\n\n\t\tTasksByCategory map[HistoryTaskCategory][]Task\n\n\t\tWorkflowRequests []*WorkflowRequest\n\n\t\tCondition int64\n\n\t\tChecksum     checksum.Checksum\n\t\tChecksumData *DataBlob\n\t}\n\n\t// InternalAppendHistoryEventsRequest is used to append new events to workflow execution history  for Persistence Interface\n\tInternalAppendHistoryEventsRequest struct {\n\t\tDomainID          string\n\t\tExecution         workflow.WorkflowExecution\n\t\tFirstEventID      int64\n\t\tEventBatchVersion int64\n\t\tRangeID           int64\n\t\tTransactionID     int64\n\t\tEvents            *DataBlob\n\t\tOverwrite         bool\n\t}\n\n\t// InternalAppendHistoryNodesRequest is used to append a batch of history nodes\n\tInternalAppendHistoryNodesRequest struct {\n\t\t// True if it is the first append request to the branch\n\t\tIsNewBranch bool\n\t\t// The info for clean up data in background\n\t\tInfo string\n\t\t// The branch to be appended\n\t\tBranchInfo types.HistoryBranch\n\t\t// The first eventID becomes the nodeID to be appended\n\t\tNodeID int64\n\t\t// The events to be appended\n\t\tEvents *DataBlob\n\t\t// Requested TransactionID for conditional update\n\t\tTransactionID int64\n\t\t// Used in sharded data stores to identify which shard to use\n\t\tShardID int\n\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalGetWorkflowExecutionRequest is used to retrieve the info of a workflow execution\n\tInternalGetWorkflowExecutionRequest struct {\n\t\tDomainID  string\n\t\tExecution types.WorkflowExecution\n\t\tRangeID   int64\n\t}\n\n\t// InternalGetWorkflowExecutionResponse is the response to GetWorkflowExecution for Persistence Interface\n\tInternalGetWorkflowExecutionResponse struct {\n\t\tState *InternalWorkflowMutableState\n\t}\n\n\t// InternalListConcreteExecutionsResponse is the response to ListConcreteExecutions for Persistence Interface\n\tInternalListConcreteExecutionsResponse struct {\n\t\tExecutions    []*InternalListConcreteExecutionsEntity\n\t\tNextPageToken []byte\n\t}\n\n\t// InternalListConcreteExecutionsEntity is a single entity in InternalListConcreteExecutionsResponse\n\tInternalListConcreteExecutionsEntity struct {\n\t\tExecutionInfo    *InternalWorkflowExecutionInfo\n\t\tVersionHistories *DataBlob\n\t}\n\n\t// InternalForkHistoryBranchRequest is used to fork a history branch\n\tInternalForkHistoryBranchRequest struct {\n\t\t// The base branch to fork from\n\t\tForkBranchInfo types.HistoryBranch\n\t\t// The nodeID to fork from, the new branch will start from ( inclusive ), the base branch will stop at(exclusive)\n\t\tForkNodeID int64\n\t\t// branchID of the new branch\n\t\tNewBranchID string\n\t\t// the info for clean up data in background\n\t\tInfo string\n\t\t// Used in sharded data stores to identify which shard to use\n\t\tShardID int\n\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalForkHistoryBranchResponse is the response to ForkHistoryBranchRequest\n\tInternalForkHistoryBranchResponse struct {\n\t\t// branchInfo to represent the new branch\n\t\tNewBranchInfo types.HistoryBranch\n\t}\n\n\t// InternalDeleteHistoryBranchRequest is used to remove a history branch\n\tInternalDeleteHistoryBranchRequest struct {\n\t\t// branch to be deleted\n\t\tBranchInfo types.HistoryBranch\n\t\t// Used in sharded data stores to identify which shard to use\n\t\tShardID int\n\t}\n\n\t// InternalReadHistoryBranchRequest is used to read a history branch\n\tInternalReadHistoryBranchRequest struct {\n\t\t// The tree of branch range to be read\n\t\tTreeID string\n\t\t// The branch range to be read\n\t\tBranchID string\n\t\t// Get the history nodes from MinNodeID. Inclusive.\n\t\tMinNodeID int64\n\t\t// Get the history nodes upto MaxNodeID.  Exclusive.\n\t\tMaxNodeID int64\n\t\t// passing thru for pagination\n\t\tPageSize int\n\t\t// Pagination token\n\t\tNextPageToken []byte\n\t\t// LastNodeID is the last known node ID attached to a history node\n\t\tLastNodeID int64\n\t\t// LastTransactionID is the last known transaction ID attached to a history node\n\t\tLastTransactionID int64\n\t\t// Used in sharded data stores to identify which shard to use\n\t\tShardID int\n\t}\n\n\t// InternalCompleteForkBranchRequest is used to update some tree/branch meta data for forking\n\tInternalCompleteForkBranchRequest struct {\n\t\t// branch to be updated\n\t\tBranchInfo workflow.HistoryBranch\n\t\t// whether fork is successful\n\t\tSuccess bool\n\t\t// Used in sharded data stores to identify which shard to use\n\t\tShardID int\n\t}\n\n\t// InternalReadHistoryBranchResponse is the response to ReadHistoryBranchRequest\n\tInternalReadHistoryBranchResponse struct {\n\t\t// History events\n\t\tHistory []*DataBlob\n\t\t// Pagination token\n\t\tNextPageToken []byte\n\t\t// LastNodeID is the last known node ID attached to a history node\n\t\tLastNodeID int64\n\t\t// LastTransactionID is the last known transaction ID attached to a history node\n\t\tLastTransactionID int64\n\t}\n\n\t// InternalGetHistoryTreeRequest is used to get history tree\n\tInternalGetHistoryTreeRequest struct {\n\t\t// A UUID of a tree\n\t\tTreeID string\n\t\t// Get data from this shard\n\t\tShardID *int\n\t\t// optional: can provide treeID via branchToken if treeID is empty\n\t\tBranchToken []byte\n\t}\n\n\t// InternalGetHistoryTreeResponse is the response to GetHistoryTree\n\tInternalGetHistoryTreeResponse struct {\n\t\t// all branches of a tree\n\t\tBranches []*types.HistoryBranch\n\t}\n\n\t// InternalVisibilityWorkflowExecutionInfo is visibility info for internal response\n\tInternalVisibilityWorkflowExecutionInfo struct {\n\t\tDomainID               string\n\t\tWorkflowType           string\n\t\tWorkflowID             string\n\t\tRunID                  string\n\t\tTypeName               string\n\t\tStartTime              time.Time\n\t\tExecutionTime          time.Time\n\t\tCloseTime              time.Time\n\t\tStatus                 *types.WorkflowExecutionCloseStatus\n\t\tHistoryLength          int64\n\t\tMemo                   *DataBlob\n\t\tTaskList               string\n\t\tIsCron                 bool\n\t\tNumClusters            int16\n\t\tClusterAttributeScope  string\n\t\tClusterAttributeName   string\n\t\tUpdateTime             time.Time\n\t\tSearchAttributes       map[string]interface{}\n\t\tShardID                int16\n\t\tCronSchedule           string\n\t\tExecutionStatus        types.WorkflowExecutionStatus\n\t\tScheduledExecutionTime time.Time\n\t}\n\n\t// InternalListWorkflowExecutionsResponse is response from ListWorkflowExecutions\n\tInternalListWorkflowExecutionsResponse struct {\n\t\tExecutions []*InternalVisibilityWorkflowExecutionInfo\n\t\t// Token to read next page if there are more workflow executions beyond page size.\n\t\t// Use this to set NextPageToken on ListWorkflowExecutionsRequest to read the next page.\n\t\tNextPageToken []byte\n\t}\n\n\t// InternalGetClosedWorkflowExecutionRequest is used retrieve the record for a specific execution\n\tInternalGetClosedWorkflowExecutionRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\tExecution  types.WorkflowExecution\n\t}\n\n\t// InternalListClosedWorkflowExecutionsByStatusRequest is used to list executions that have specific close status\n\tInternalListClosedWorkflowExecutionsByStatusRequest struct {\n\t\tInternalListWorkflowExecutionsRequest\n\t\tStatus types.WorkflowExecutionCloseStatus\n\t}\n\n\t// InternalListWorkflowExecutionsByWorkflowIDRequest is used to list executions that have specific WorkflowID in a domain\n\tInternalListWorkflowExecutionsByWorkflowIDRequest struct {\n\t\tInternalListWorkflowExecutionsRequest\n\t\tWorkflowID string\n\t}\n\n\t// InternalListWorkflowExecutionsByTypeRequest is used to list executions of a specific type in a domain\n\tInternalListWorkflowExecutionsByTypeRequest struct {\n\t\tInternalListWorkflowExecutionsRequest\n\t\tWorkflowTypeName string\n\t}\n\n\t// InternalGetClosedWorkflowExecutionResponse is response from GetWorkflowExecution\n\tInternalGetClosedWorkflowExecutionResponse struct {\n\t\tExecution *InternalVisibilityWorkflowExecutionInfo\n\t}\n\n\t// InternalRecordWorkflowExecutionStartedRequest request to RecordWorkflowExecutionStarted\n\tInternalRecordWorkflowExecutionStartedRequest struct {\n\t\tDomainUUID             string\n\t\tWorkflowID             string\n\t\tRunID                  string\n\t\tWorkflowTypeName       string\n\t\tStartTimestamp         time.Time\n\t\tExecutionTimestamp     time.Time\n\t\tWorkflowTimeout        time.Duration\n\t\tTaskID                 int64\n\t\tMemo                   *DataBlob\n\t\tTaskList               string\n\t\tIsCron                 bool\n\t\tCronSchedule           string\n\t\tNumClusters            int16\n\t\tClusterAttributeScope  string\n\t\tClusterAttributeName   string\n\t\tUpdateTimestamp        time.Time\n\t\tSearchAttributes       map[string][]byte\n\t\tShardID                int16\n\t\tExecutionStatus        types.WorkflowExecutionStatus\n\t\tScheduledExecutionTime time.Time\n\t}\n\n\t// InternalRecordWorkflowExecutionClosedRequest is request to RecordWorkflowExecutionClosed\n\tInternalRecordWorkflowExecutionClosedRequest struct {\n\t\tDomainUUID             string\n\t\tWorkflowID             string\n\t\tRunID                  string\n\t\tWorkflowTypeName       string\n\t\tStartTimestamp         time.Time\n\t\tExecutionTimestamp     time.Time\n\t\tTaskID                 int64\n\t\tMemo                   *DataBlob\n\t\tTaskList               string\n\t\tSearchAttributes       map[string][]byte\n\t\tCloseTimestamp         time.Time\n\t\tStatus                 types.WorkflowExecutionCloseStatus\n\t\tHistoryLength          int64\n\t\tRetentionPeriod        time.Duration\n\t\tIsCron                 bool\n\t\tCronSchedule           string\n\t\tNumClusters            int16\n\t\tClusterAttributeScope  string\n\t\tClusterAttributeName   string\n\t\tUpdateTimestamp        time.Time\n\t\tShardID                int16\n\t\tExecutionStatus        types.WorkflowExecutionStatus\n\t\tScheduledExecutionTime time.Time\n\t}\n\n\t// InternalRecordWorkflowExecutionUninitializedRequest is used to add a record of a newly uninitialized execution\n\tInternalRecordWorkflowExecutionUninitializedRequest struct {\n\t\tDomainUUID       string\n\t\tWorkflowID       string\n\t\tRunID            string\n\t\tWorkflowTypeName string\n\t\tUpdateTimestamp  time.Time\n\t\tShardID          int64\n\t}\n\n\t// InternalUpsertWorkflowExecutionRequest is request to UpsertWorkflowExecution\n\tInternalUpsertWorkflowExecutionRequest struct {\n\t\tDomainUUID                  string\n\t\tWorkflowID                  string\n\t\tRunID                       string\n\t\tWorkflowTypeName            string\n\t\tStartTimestamp              time.Time\n\t\tExecutionTimestamp          time.Time\n\t\tWorkflowTimeout             time.Duration\n\t\tTaskID                      int64\n\t\tMemo                        *DataBlob\n\t\tTaskList                    string\n\t\tIsCron                      bool\n\t\tCronSchedule                string\n\t\tNumClusters                 int16\n\t\tClusterAttributeScope       string\n\t\tClusterAttributeName        string\n\t\tUpdateTimestamp             time.Time\n\t\tSearchAttributes            map[string][]byte\n\t\tShardID                     int64\n\t\tExecutionStatus             types.WorkflowExecutionStatus\n\t\tScheduledExecutionTimestamp int64\n\t}\n\n\t// InternalListWorkflowExecutionsRequest is used to list executions in a domain\n\tInternalListWorkflowExecutionsRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\t// The earliest end of the time range\n\t\tEarliestTime time.Time\n\t\t// The latest end of the time range\n\t\tLatestTime time.Time\n\t\t// Maximum number of workflow executions per page\n\t\tPageSize int\n\t\t// Token to continue reading next page of workflow executions.\n\t\t// Pass in empty slice for first page.\n\t\tNextPageToken []byte\n\t}\n\n\t// InternalDomainConfig describes the domain configuration\n\tInternalDomainConfig struct {\n\t\tRetention                time.Duration\n\t\tEmitMetric               bool                 // deprecated\n\t\tArchivalBucket           string               // deprecated\n\t\tArchivalStatus           types.ArchivalStatus // deprecated\n\t\tHistoryArchivalStatus    types.ArchivalStatus\n\t\tHistoryArchivalURI       string\n\t\tVisibilityArchivalStatus types.ArchivalStatus\n\t\tVisibilityArchivalURI    string\n\t\tBadBinaries              *DataBlob\n\t\tIsolationGroups          *DataBlob\n\t\tAsyncWorkflowsConfig     *DataBlob\n\t}\n\n\tInternalDomainReplicationConfig struct {\n\t\tClusters             []*ClusterReplicationConfig\n\t\tActiveClusterName    string\n\t\tActiveClustersConfig *DataBlob\n\t}\n\n\t// InternalCreateDomainRequest is used to create the domain\n\tInternalCreateDomainRequest struct {\n\t\tInfo              *DomainInfo\n\t\tConfig            *InternalDomainConfig\n\t\tReplicationConfig *InternalDomainReplicationConfig\n\t\tIsGlobalDomain    bool\n\t\tConfigVersion     int64\n\t\tFailoverVersion   int64\n\t\tLastUpdatedTime   time.Time\n\t\tCurrentTimeStamp  time.Time\n\t}\n\n\t// InternalGetDomainResponse is the response for GetDomain\n\tInternalGetDomainResponse struct {\n\t\tInfo                        *DomainInfo\n\t\tConfig                      *InternalDomainConfig\n\t\tReplicationConfig           *InternalDomainReplicationConfig\n\t\tIsGlobalDomain              bool\n\t\tConfigVersion               int64\n\t\tFailoverVersion             int64\n\t\tFailoverNotificationVersion int64\n\t\tPreviousFailoverVersion     int64\n\t\tFailoverEndTime             *time.Time\n\t\tLastUpdatedTime             time.Time\n\t\tNotificationVersion         int64\n\t}\n\n\t// InternalUpdateDomainRequest is used to update domain\n\tInternalUpdateDomainRequest struct {\n\t\tInfo                        *DomainInfo\n\t\tConfig                      *InternalDomainConfig\n\t\tReplicationConfig           *InternalDomainReplicationConfig\n\t\tConfigVersion               int64\n\t\tFailoverVersion             int64\n\t\tFailoverNotificationVersion int64\n\t\tPreviousFailoverVersion     int64\n\t\tFailoverEndTime             *time.Time\n\t\tLastUpdatedTime             time.Time\n\t\tNotificationVersion         int64\n\t}\n\n\t// InternalListDomainsResponse is the response for GetDomain\n\tInternalListDomainsResponse struct {\n\t\tDomains       []*InternalGetDomainResponse\n\t\tNextPageToken []byte\n\t}\n\n\tDomainAuditOperationType int\n\n\t// InternalCreateDomainAuditLogRequest is used to create a domain audit log entry\n\tInternalCreateDomainAuditLogRequest struct {\n\t\tDomainID        string\n\t\tEventID         string\n\t\tStateBefore     *DataBlob\n\t\tStateAfter      *DataBlob\n\t\tOperationType   DomainAuditOperationType\n\t\tCreatedTime     time.Time\n\t\tLastUpdatedTime time.Time\n\t\tIdentity        string\n\t\tIdentityType    string\n\t\tComment         string\n\t\tTTLSeconds      int64 // TTL for the audit log entry in seconds\n\t}\n\n\t// InternalGetDomainAuditLogsResponse is the response for GetDomainAuditLogs\n\tInternalGetDomainAuditLogsResponse struct {\n\t\tAuditLogs     []*InternalDomainAuditLog\n\t\tNextPageToken []byte\n\t}\n\n\t// InternalDomainAuditLog represents a single internal domain audit log entry\n\tInternalDomainAuditLog struct {\n\t\tEventID         string\n\t\tDomainID        string\n\t\tStateBefore     *DataBlob\n\t\tStateAfter      *DataBlob\n\t\tOperationType   DomainAuditOperationType\n\t\tCreatedTime     time.Time\n\t\tLastUpdatedTime time.Time\n\t\tIdentity        string\n\t\tIdentityType    string\n\t\tComment         string\n\t}\n\n\t// InternalShardInfo describes a shard\n\tInternalShardInfo struct {\n\t\tShardID                       int                         `json:\"shard_id\"`\n\t\tOwner                         string                      `json:\"owner\"`\n\t\tRangeID                       int64                       `json:\"range_id\"`\n\t\tStolenSinceRenew              int                         `json:\"stolen_since_renew\"`\n\t\tUpdatedAt                     time.Time                   `json:\"updated_at\"`\n\t\tReplicationAckLevel           int64                       `json:\"replication_ack_level\"`\n\t\tReplicationDLQAckLevel        map[string]int64            `json:\"replication_dlq_ack_level\"`\n\t\tTransferAckLevel              int64                       `json:\"transfer_ack_level\"`\n\t\tTimerAckLevel                 time.Time                   `json:\"timer_ack_level\"`\n\t\tClusterTransferAckLevel       map[string]int64            `json:\"cluster_transfer_ack_level\"`\n\t\tClusterTimerAckLevel          map[string]time.Time        `json:\"cluster_timer_ack_level\"`\n\t\tTransferProcessingQueueStates *DataBlob                   `json:\"transfer_processing_queue_states\"`\n\t\tTimerProcessingQueueStates    *DataBlob                   `json:\"timer_processing_queue_states\"`\n\t\tClusterReplicationLevel       map[string]int64            `json:\"cluster_replication_level\"`\n\t\tDomainNotificationVersion     int64                       `json:\"domain_notification_version\"`\n\t\tPendingFailoverMarkers        *DataBlob                   `json:\"pending_failover_markers\"`\n\t\tQueueStates                   map[int32]*types.QueueState `json:\"queue_states\"`\n\t\tCurrentTimestamp              time.Time\n\t}\n\n\t// InternalCreateShardRequest is request to CreateShard\n\tInternalCreateShardRequest struct {\n\t\tShardInfo        *InternalShardInfo\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalGetShardRequest is used to get shard information\n\tInternalGetShardRequest struct {\n\t\tShardID int\n\t}\n\n\t// InternalUpdateShardRequest  is used to update shard information\n\tInternalUpdateShardRequest struct {\n\t\tShardInfo        *InternalShardInfo\n\t\tPreviousRangeID  int64\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// InternalGetShardResponse is the response to GetShard\n\tInternalGetShardResponse struct {\n\t\tShardInfo *InternalShardInfo\n\t}\n)\n\nfunc (tr *InternalGetHistoryTreeResponse) ByBranchID() map[string]*types.HistoryBranch {\n\tout := make(map[string]*types.HistoryBranch, len(tr.Branches))\n\tfor _, branch := range tr.Branches {\n\t\tout[branch.BranchID] = branch\n\t}\n\treturn out\n}\n\n// NewDataBlob returns a new DataBlob\nfunc NewDataBlob(data []byte, encodingType constants.EncodingType) *DataBlob {\n\tif len(data) == 0 {\n\t\treturn nil\n\t}\n\tif encodingType != constants.EncodingTypeThriftRW && data[0] == 'Y' {\n\t\t// original reason for this is not written down, but maybe for handling data prior to an encoding type?\n\t\tpanic(fmt.Sprintf(\"Invalid data blob encoding: \\\"%v\\\"\", encodingType))\n\t}\n\treturn &DataBlob{\n\t\tData:     data,\n\t\tEncoding: encodingType,\n\t}\n}\n\n// FromDataBlob decodes a datablob into a (payload, encodingType) tuple\nfunc FromDataBlob(blob *DataBlob) ([]byte, string) {\n\tif blob == nil || len(blob.Data) == 0 {\n\t\treturn nil, \"\"\n\t}\n\treturn blob.Data, string(blob.Encoding)\n}\n\n// Convert a *Datablob to safe that calling its method won't run into NPE\nfunc (d *DataBlob) ToNilSafeDataBlob() *DataBlob {\n\tif d != nil {\n\t\treturn d\n\t}\n\treturn &DataBlob{}\n}\n\nfunc (d *DataBlob) GetEncodingString() string {\n\tif d == nil {\n\t\treturn \"\"\n\t}\n\treturn string(d.Encoding)\n}\n\n// GetData is a safe way to get the byte array or nil\nfunc (d *DataBlob) GetData() []byte {\n\tif d == nil || d.Data == nil {\n\t\treturn []byte{}\n\t}\n\treturn d.Data\n}\n\n// GetEncoding returns encoding type\nfunc (d *DataBlob) GetEncoding() constants.EncodingType {\n\tencodingStr := d.GetEncodingString()\n\n\tswitch constants.EncodingType(encodingStr) {\n\tcase constants.EncodingTypeGob:\n\t\treturn constants.EncodingTypeGob\n\tcase constants.EncodingTypeJSON:\n\t\treturn constants.EncodingTypeJSON\n\tcase constants.EncodingTypeThriftRW:\n\t\treturn constants.EncodingTypeThriftRW\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\treturn constants.EncodingTypeThriftRWSnappy\n\tcase constants.EncodingTypeEmpty:\n\t\treturn constants.EncodingTypeEmpty\n\tdefault:\n\t\treturn constants.EncodingTypeUnknown\n\t}\n}\n\n// ToInternal convert data blob to internal representation\nfunc (d *DataBlob) ToInternal() *types.DataBlob {\n\tswitch d.Encoding {\n\tcase constants.EncodingTypeJSON:\n\t\treturn &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         d.Data,\n\t\t}\n\tcase constants.EncodingTypeThriftRW:\n\t\treturn &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\tData:         d.Data,\n\t\t}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"DataBlob.ToInternal() with unsupported encoding type: %v\", d.Encoding))\n\t}\n}\n\n// NewDataBlobFromInternal convert data blob from internal representation\nfunc NewDataBlobFromInternal(blob *types.DataBlob) *DataBlob {\n\tswitch blob.GetEncodingType() {\n\tcase types.EncodingTypeJSON:\n\t\treturn &DataBlob{\n\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\tData:     blob.Data,\n\t\t}\n\tcase types.EncodingTypeThriftRW:\n\t\treturn &DataBlob{\n\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\tData:     blob.Data,\n\t\t}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"NewDataBlobFromInternal with unsupported encoding type: %v\", blob.GetEncodingType()))\n\t}\n}\n\nfunc (t *InternalReplicationTaskInfo) ToTask() (Task, error) {\n\tswitch t.TaskType {\n\tcase ReplicationTaskTypeHistory:\n\t\treturn &HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\tDomainID:   t.DomainID,\n\t\t\t\tWorkflowID: t.WorkflowID,\n\t\t\t\tRunID:      t.RunID,\n\t\t\t},\n\t\t\tTaskData: TaskData{\n\t\t\t\tVersion:             t.Version,\n\t\t\t\tTaskID:              t.TaskID,\n\t\t\t\tVisibilityTimestamp: t.CreationTime,\n\t\t\t},\n\t\t\tFirstEventID:      t.FirstEventID,\n\t\t\tNextEventID:       t.NextEventID,\n\t\t\tBranchToken:       t.BranchToken,\n\t\t\tNewRunBranchToken: t.NewRunBranchToken,\n\t\t}, nil\n\tcase ReplicationTaskTypeSyncActivity:\n\t\treturn &SyncActivityTask{\n\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\tDomainID:   t.DomainID,\n\t\t\t\tWorkflowID: t.WorkflowID,\n\t\t\t\tRunID:      t.RunID,\n\t\t\t},\n\t\t\tTaskData: TaskData{\n\t\t\t\tVersion:             t.Version,\n\t\t\t\tTaskID:              t.TaskID,\n\t\t\t\tVisibilityTimestamp: t.CreationTime,\n\t\t\t},\n\t\t\tScheduledID: t.ScheduledID,\n\t\t}, nil\n\tcase ReplicationTaskTypeFailoverMarker:\n\t\treturn &FailoverMarkerTask{\n\t\t\tTaskData: TaskData{\n\t\t\t\tVersion: t.Version,\n\t\t\t\tTaskID:  t.TaskID,\n\t\t\t},\n\t\t\tDomainID: t.DomainID,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown task type: %d\", t.TaskType)\n\t}\n}\n"
  },
  {
    "path": "common/persistence/data_store_interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence (interfaces: ExecutionStore,ShardStore,DomainStore,TaskStore,HistoryStore,ConfigStore,DomainAuditStore)\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -destination data_store_interfaces_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence ExecutionStore,ShardStore,DomainStore,TaskStore,HistoryStore,ConfigStore,DomainAuditStore\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockExecutionStore is a mock of ExecutionStore interface.\ntype MockExecutionStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutionStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutionStoreMockRecorder is the mock recorder for MockExecutionStore.\ntype MockExecutionStoreMockRecorder struct {\n\tmock *MockExecutionStore\n}\n\n// NewMockExecutionStore creates a new mock instance.\nfunc NewMockExecutionStore(ctrl *gomock.Controller) *MockExecutionStore {\n\tmock := &MockExecutionStore{ctrl: ctrl}\n\tmock.recorder = &MockExecutionStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutionStore) EXPECT() *MockExecutionStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockExecutionStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockExecutionStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockExecutionStore)(nil).Close))\n}\n\n// CompleteHistoryTask mocks base method.\nfunc (m *MockExecutionStore) CompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteHistoryTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteHistoryTask indicates an expected call of CompleteHistoryTask.\nfunc (mr *MockExecutionStoreMockRecorder) CompleteHistoryTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteHistoryTask\", reflect.TypeOf((*MockExecutionStore)(nil).CompleteHistoryTask), ctx, request)\n}\n\n// ConflictResolveWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) ConflictResolveWorkflowExecution(ctx context.Context, request *InternalConflictResolveWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ConflictResolveWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ConflictResolveWorkflowExecution indicates an expected call of ConflictResolveWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) ConflictResolveWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ConflictResolveWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).ConflictResolveWorkflowExecution), ctx, request)\n}\n\n// CreateFailoverMarkerTasks mocks base method.\nfunc (m *MockExecutionStore) CreateFailoverMarkerTasks(ctx context.Context, request *CreateFailoverMarkersRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateFailoverMarkerTasks\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateFailoverMarkerTasks indicates an expected call of CreateFailoverMarkerTasks.\nfunc (mr *MockExecutionStoreMockRecorder) CreateFailoverMarkerTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateFailoverMarkerTasks\", reflect.TypeOf((*MockExecutionStore)(nil).CreateFailoverMarkerTasks), ctx, request)\n}\n\n// CreateWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) CreateWorkflowExecution(ctx context.Context, request *InternalCreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*CreateWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateWorkflowExecution indicates an expected call of CreateWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) CreateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).CreateWorkflowExecution), ctx, request)\n}\n\n// DeleteActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockExecutionStore) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActiveClusterSelectionPolicy\", ctx, domainID, wfID, rID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActiveClusterSelectionPolicy indicates an expected call of DeleteActiveClusterSelectionPolicy.\nfunc (mr *MockExecutionStoreMockRecorder) DeleteActiveClusterSelectionPolicy(ctx, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActiveClusterSelectionPolicy\", reflect.TypeOf((*MockExecutionStore)(nil).DeleteActiveClusterSelectionPolicy), ctx, domainID, wfID, rID)\n}\n\n// DeleteCurrentWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) DeleteCurrentWorkflowExecution(ctx context.Context, request *DeleteCurrentWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflowExecution indicates an expected call of DeleteCurrentWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) DeleteCurrentWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).DeleteCurrentWorkflowExecution), ctx, request)\n}\n\n// DeleteReplicationTaskFromDLQ mocks base method.\nfunc (m *MockExecutionStore) DeleteReplicationTaskFromDLQ(ctx context.Context, request *DeleteReplicationTaskFromDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationTaskFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationTaskFromDLQ indicates an expected call of DeleteReplicationTaskFromDLQ.\nfunc (mr *MockExecutionStoreMockRecorder) DeleteReplicationTaskFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationTaskFromDLQ\", reflect.TypeOf((*MockExecutionStore)(nil).DeleteReplicationTaskFromDLQ), ctx, request)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) DeleteWorkflowExecution(ctx context.Context, request *DeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) DeleteWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).DeleteWorkflowExecution), ctx, request)\n}\n\n// GetActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockExecutionStore) GetActiveClusterSelectionPolicy(ctx context.Context, domainID, wfID, rID string) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterSelectionPolicy\", ctx, domainID, wfID, rID)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActiveClusterSelectionPolicy indicates an expected call of GetActiveClusterSelectionPolicy.\nfunc (mr *MockExecutionStoreMockRecorder) GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterSelectionPolicy\", reflect.TypeOf((*MockExecutionStore)(nil).GetActiveClusterSelectionPolicy), ctx, domainID, wfID, rID)\n}\n\n// GetCurrentExecution mocks base method.\nfunc (m *MockExecutionStore) GetCurrentExecution(ctx context.Context, request *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentExecution\", ctx, request)\n\tret0, _ := ret[0].(*GetCurrentExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCurrentExecution indicates an expected call of GetCurrentExecution.\nfunc (mr *MockExecutionStoreMockRecorder) GetCurrentExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentExecution\", reflect.TypeOf((*MockExecutionStore)(nil).GetCurrentExecution), ctx, request)\n}\n\n// GetHistoryTasks mocks base method.\nfunc (m *MockExecutionStore) GetHistoryTasks(ctx context.Context, request *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetHistoryTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryTasks indicates an expected call of GetHistoryTasks.\nfunc (mr *MockExecutionStoreMockRecorder) GetHistoryTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryTasks\", reflect.TypeOf((*MockExecutionStore)(nil).GetHistoryTasks), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockExecutionStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockExecutionStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockExecutionStore)(nil).GetName))\n}\n\n// GetReplicationDLQSize mocks base method.\nfunc (m *MockExecutionStore) GetReplicationDLQSize(ctx context.Context, request *GetReplicationDLQSizeRequest) (*GetReplicationDLQSizeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationDLQSize\", ctx, request)\n\tret0, _ := ret[0].(*GetReplicationDLQSizeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationDLQSize indicates an expected call of GetReplicationDLQSize.\nfunc (mr *MockExecutionStoreMockRecorder) GetReplicationDLQSize(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationDLQSize\", reflect.TypeOf((*MockExecutionStore)(nil).GetReplicationDLQSize), ctx, request)\n}\n\n// GetReplicationTasksFromDLQ mocks base method.\nfunc (m *MockExecutionStore) GetReplicationTasksFromDLQ(ctx context.Context, request *GetReplicationTasksFromDLQRequest) (*GetHistoryTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationTasksFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(*GetHistoryTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationTasksFromDLQ indicates an expected call of GetReplicationTasksFromDLQ.\nfunc (mr *MockExecutionStoreMockRecorder) GetReplicationTasksFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationTasksFromDLQ\", reflect.TypeOf((*MockExecutionStore)(nil).GetReplicationTasksFromDLQ), ctx, request)\n}\n\n// GetShardID mocks base method.\nfunc (m *MockExecutionStore) GetShardID() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardID\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetShardID indicates an expected call of GetShardID.\nfunc (mr *MockExecutionStoreMockRecorder) GetShardID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardID\", reflect.TypeOf((*MockExecutionStore)(nil).GetShardID))\n}\n\n// GetWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) GetWorkflowExecution(ctx context.Context, request *InternalGetWorkflowExecutionRequest) (*InternalGetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecution indicates an expected call of GetWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) GetWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).GetWorkflowExecution), ctx, request)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MockExecutionStore) IsWorkflowExecutionExists(ctx context.Context, request *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", ctx, request)\n\tret0, _ := ret[0].(*IsWorkflowExecutionExistsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MockExecutionStoreMockRecorder) IsWorkflowExecutionExists(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MockExecutionStore)(nil).IsWorkflowExecutionExists), ctx, request)\n}\n\n// ListConcreteExecutions mocks base method.\nfunc (m *MockExecutionStore) ListConcreteExecutions(ctx context.Context, request *ListConcreteExecutionsRequest) (*InternalListConcreteExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListConcreteExecutions\", ctx, request)\n\tret0, _ := ret[0].(*InternalListConcreteExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListConcreteExecutions indicates an expected call of ListConcreteExecutions.\nfunc (mr *MockExecutionStoreMockRecorder) ListConcreteExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListConcreteExecutions\", reflect.TypeOf((*MockExecutionStore)(nil).ListConcreteExecutions), ctx, request)\n}\n\n// ListCurrentExecutions mocks base method.\nfunc (m *MockExecutionStore) ListCurrentExecutions(ctx context.Context, request *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListCurrentExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListCurrentExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListCurrentExecutions indicates an expected call of ListCurrentExecutions.\nfunc (mr *MockExecutionStoreMockRecorder) ListCurrentExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListCurrentExecutions\", reflect.TypeOf((*MockExecutionStore)(nil).ListCurrentExecutions), ctx, request)\n}\n\n// PutReplicationTaskToDLQ mocks base method.\nfunc (m *MockExecutionStore) PutReplicationTaskToDLQ(ctx context.Context, request *InternalPutReplicationTaskToDLQRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PutReplicationTaskToDLQ\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PutReplicationTaskToDLQ indicates an expected call of PutReplicationTaskToDLQ.\nfunc (mr *MockExecutionStoreMockRecorder) PutReplicationTaskToDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PutReplicationTaskToDLQ\", reflect.TypeOf((*MockExecutionStore)(nil).PutReplicationTaskToDLQ), ctx, request)\n}\n\n// RangeCompleteHistoryTask mocks base method.\nfunc (m *MockExecutionStore) RangeCompleteHistoryTask(ctx context.Context, request *RangeCompleteHistoryTaskRequest) (*RangeCompleteHistoryTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeCompleteHistoryTask\", ctx, request)\n\tret0, _ := ret[0].(*RangeCompleteHistoryTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeCompleteHistoryTask indicates an expected call of RangeCompleteHistoryTask.\nfunc (mr *MockExecutionStoreMockRecorder) RangeCompleteHistoryTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeCompleteHistoryTask\", reflect.TypeOf((*MockExecutionStore)(nil).RangeCompleteHistoryTask), ctx, request)\n}\n\n// RangeDeleteReplicationTaskFromDLQ mocks base method.\nfunc (m *MockExecutionStore) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *RangeDeleteReplicationTaskFromDLQRequest) (*RangeDeleteReplicationTaskFromDLQResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationTaskFromDLQ\", ctx, request)\n\tret0, _ := ret[0].(*RangeDeleteReplicationTaskFromDLQResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteReplicationTaskFromDLQ indicates an expected call of RangeDeleteReplicationTaskFromDLQ.\nfunc (mr *MockExecutionStoreMockRecorder) RangeDeleteReplicationTaskFromDLQ(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationTaskFromDLQ\", reflect.TypeOf((*MockExecutionStore)(nil).RangeDeleteReplicationTaskFromDLQ), ctx, request)\n}\n\n// UpdateWorkflowExecution mocks base method.\nfunc (m *MockExecutionStore) UpdateWorkflowExecution(ctx context.Context, request *InternalUpdateWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecution indicates an expected call of UpdateWorkflowExecution.\nfunc (mr *MockExecutionStoreMockRecorder) UpdateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecution\", reflect.TypeOf((*MockExecutionStore)(nil).UpdateWorkflowExecution), ctx, request)\n}\n\n// MockShardStore is a mock of ShardStore interface.\ntype MockShardStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardStoreMockRecorder is the mock recorder for MockShardStore.\ntype MockShardStoreMockRecorder struct {\n\tmock *MockShardStore\n}\n\n// NewMockShardStore creates a new mock instance.\nfunc NewMockShardStore(ctrl *gomock.Controller) *MockShardStore {\n\tmock := &MockShardStore{ctrl: ctrl}\n\tmock.recorder = &MockShardStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardStore) EXPECT() *MockShardStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockShardStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockShardStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockShardStore)(nil).Close))\n}\n\n// CreateShard mocks base method.\nfunc (m *MockShardStore) CreateShard(ctx context.Context, request *InternalCreateShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateShard\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateShard indicates an expected call of CreateShard.\nfunc (mr *MockShardStoreMockRecorder) CreateShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateShard\", reflect.TypeOf((*MockShardStore)(nil).CreateShard), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockShardStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockShardStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockShardStore)(nil).GetName))\n}\n\n// GetShard mocks base method.\nfunc (m *MockShardStore) GetShard(ctx context.Context, request *InternalGetShardRequest) (*InternalGetShardResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShard\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetShardResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShard indicates an expected call of GetShard.\nfunc (mr *MockShardStoreMockRecorder) GetShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShard\", reflect.TypeOf((*MockShardStore)(nil).GetShard), ctx, request)\n}\n\n// UpdateShard mocks base method.\nfunc (m *MockShardStore) UpdateShard(ctx context.Context, request *InternalUpdateShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShard\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateShard indicates an expected call of UpdateShard.\nfunc (mr *MockShardStoreMockRecorder) UpdateShard(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShard\", reflect.TypeOf((*MockShardStore)(nil).UpdateShard), ctx, request)\n}\n\n// MockDomainStore is a mock of DomainStore interface.\ntype MockDomainStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainStoreMockRecorder is the mock recorder for MockDomainStore.\ntype MockDomainStoreMockRecorder struct {\n\tmock *MockDomainStore\n}\n\n// NewMockDomainStore creates a new mock instance.\nfunc NewMockDomainStore(ctrl *gomock.Controller) *MockDomainStore {\n\tmock := &MockDomainStore{ctrl: ctrl}\n\tmock.recorder = &MockDomainStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainStore) EXPECT() *MockDomainStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDomainStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDomainStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDomainStore)(nil).Close))\n}\n\n// CreateDomain mocks base method.\nfunc (m *MockDomainStore) CreateDomain(ctx context.Context, request *InternalCreateDomainRequest) (*CreateDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDomain\", ctx, request)\n\tret0, _ := ret[0].(*CreateDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDomain indicates an expected call of CreateDomain.\nfunc (mr *MockDomainStoreMockRecorder) CreateDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDomain\", reflect.TypeOf((*MockDomainStore)(nil).CreateDomain), ctx, request)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockDomainStore) DeleteDomain(ctx context.Context, request *DeleteDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockDomainStoreMockRecorder) DeleteDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockDomainStore)(nil).DeleteDomain), ctx, request)\n}\n\n// DeleteDomainByName mocks base method.\nfunc (m *MockDomainStore) DeleteDomainByName(ctx context.Context, request *DeleteDomainByNameRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomainByName\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomainByName indicates an expected call of DeleteDomainByName.\nfunc (mr *MockDomainStoreMockRecorder) DeleteDomainByName(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomainByName\", reflect.TypeOf((*MockDomainStore)(nil).DeleteDomainByName), ctx, request)\n}\n\n// GetDomain mocks base method.\nfunc (m *MockDomainStore) GetDomain(ctx context.Context, request *GetDomainRequest) (*InternalGetDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomain\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomain indicates an expected call of GetDomain.\nfunc (mr *MockDomainStoreMockRecorder) GetDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomain\", reflect.TypeOf((*MockDomainStore)(nil).GetDomain), ctx, request)\n}\n\n// GetMetadata mocks base method.\nfunc (m *MockDomainStore) GetMetadata(ctx context.Context) (*GetMetadataResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetadata\", ctx)\n\tret0, _ := ret[0].(*GetMetadataResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMetadata indicates an expected call of GetMetadata.\nfunc (mr *MockDomainStoreMockRecorder) GetMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetadata\", reflect.TypeOf((*MockDomainStore)(nil).GetMetadata), ctx)\n}\n\n// GetName mocks base method.\nfunc (m *MockDomainStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockDomainStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockDomainStore)(nil).GetName))\n}\n\n// ListDomains mocks base method.\nfunc (m *MockDomainStore) ListDomains(ctx context.Context, request *ListDomainsRequest) (*InternalListDomainsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListDomains\", ctx, request)\n\tret0, _ := ret[0].(*InternalListDomainsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDomains indicates an expected call of ListDomains.\nfunc (mr *MockDomainStoreMockRecorder) ListDomains(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDomains\", reflect.TypeOf((*MockDomainStore)(nil).ListDomains), ctx, request)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockDomainStore) UpdateDomain(ctx context.Context, request *InternalUpdateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockDomainStoreMockRecorder) UpdateDomain(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockDomainStore)(nil).UpdateDomain), ctx, request)\n}\n\n// MockTaskStore is a mock of TaskStore interface.\ntype MockTaskStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskStoreMockRecorder is the mock recorder for MockTaskStore.\ntype MockTaskStoreMockRecorder struct {\n\tmock *MockTaskStore\n}\n\n// NewMockTaskStore creates a new mock instance.\nfunc NewMockTaskStore(ctrl *gomock.Controller) *MockTaskStore {\n\tmock := &MockTaskStore{ctrl: ctrl}\n\tmock.recorder = &MockTaskStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskStore) EXPECT() *MockTaskStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockTaskStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockTaskStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockTaskStore)(nil).Close))\n}\n\n// CompleteTask mocks base method.\nfunc (m *MockTaskStore) CompleteTask(ctx context.Context, request *CompleteTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteTask indicates an expected call of CompleteTask.\nfunc (mr *MockTaskStoreMockRecorder) CompleteTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteTask\", reflect.TypeOf((*MockTaskStore)(nil).CompleteTask), ctx, request)\n}\n\n// CompleteTasksLessThan mocks base method.\nfunc (m *MockTaskStore) CompleteTasksLessThan(ctx context.Context, request *CompleteTasksLessThanRequest) (*CompleteTasksLessThanResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteTasksLessThan\", ctx, request)\n\tret0, _ := ret[0].(*CompleteTasksLessThanResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CompleteTasksLessThan indicates an expected call of CompleteTasksLessThan.\nfunc (mr *MockTaskStoreMockRecorder) CompleteTasksLessThan(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteTasksLessThan\", reflect.TypeOf((*MockTaskStore)(nil).CompleteTasksLessThan), ctx, request)\n}\n\n// CreateTasks mocks base method.\nfunc (m *MockTaskStore) CreateTasks(ctx context.Context, request *CreateTasksRequest) (*CreateTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTasks\", ctx, request)\n\tret0, _ := ret[0].(*CreateTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateTasks indicates an expected call of CreateTasks.\nfunc (mr *MockTaskStoreMockRecorder) CreateTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTasks\", reflect.TypeOf((*MockTaskStore)(nil).CreateTasks), ctx, request)\n}\n\n// DeleteTaskList mocks base method.\nfunc (m *MockTaskStore) DeleteTaskList(ctx context.Context, request *DeleteTaskListRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTaskList\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTaskList indicates an expected call of DeleteTaskList.\nfunc (mr *MockTaskStoreMockRecorder) DeleteTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTaskList\", reflect.TypeOf((*MockTaskStore)(nil).DeleteTaskList), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockTaskStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockTaskStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockTaskStore)(nil).GetName))\n}\n\n// GetOrphanTasks mocks base method.\nfunc (m *MockTaskStore) GetOrphanTasks(ctx context.Context, request *GetOrphanTasksRequest) (*GetOrphanTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrphanTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetOrphanTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOrphanTasks indicates an expected call of GetOrphanTasks.\nfunc (mr *MockTaskStoreMockRecorder) GetOrphanTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrphanTasks\", reflect.TypeOf((*MockTaskStore)(nil).GetOrphanTasks), ctx, request)\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockTaskStore) GetTaskList(ctx context.Context, request *GetTaskListRequest) (*GetTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\", ctx, request)\n\tret0, _ := ret[0].(*GetTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockTaskStoreMockRecorder) GetTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockTaskStore)(nil).GetTaskList), ctx, request)\n}\n\n// GetTaskListSize mocks base method.\nfunc (m *MockTaskStore) GetTaskListSize(ctx context.Context, request *GetTaskListSizeRequest) (*GetTaskListSizeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListSize\", ctx, request)\n\tret0, _ := ret[0].(*GetTaskListSizeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListSize indicates an expected call of GetTaskListSize.\nfunc (mr *MockTaskStoreMockRecorder) GetTaskListSize(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListSize\", reflect.TypeOf((*MockTaskStore)(nil).GetTaskListSize), ctx, request)\n}\n\n// GetTasks mocks base method.\nfunc (m *MockTaskStore) GetTasks(ctx context.Context, request *GetTasksRequest) (*GetTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\", ctx, request)\n\tret0, _ := ret[0].(*GetTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockTaskStoreMockRecorder) GetTasks(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockTaskStore)(nil).GetTasks), ctx, request)\n}\n\n// LeaseTaskList mocks base method.\nfunc (m *MockTaskStore) LeaseTaskList(ctx context.Context, request *LeaseTaskListRequest) (*LeaseTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LeaseTaskList\", ctx, request)\n\tret0, _ := ret[0].(*LeaseTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LeaseTaskList indicates an expected call of LeaseTaskList.\nfunc (mr *MockTaskStoreMockRecorder) LeaseTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LeaseTaskList\", reflect.TypeOf((*MockTaskStore)(nil).LeaseTaskList), ctx, request)\n}\n\n// ListTaskList mocks base method.\nfunc (m *MockTaskStore) ListTaskList(ctx context.Context, request *ListTaskListRequest) (*ListTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskList\", ctx, request)\n\tret0, _ := ret[0].(*ListTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskList indicates an expected call of ListTaskList.\nfunc (mr *MockTaskStoreMockRecorder) ListTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskList\", reflect.TypeOf((*MockTaskStore)(nil).ListTaskList), ctx, request)\n}\n\n// UpdateTaskList mocks base method.\nfunc (m *MockTaskStore) UpdateTaskList(ctx context.Context, request *UpdateTaskListRequest) (*UpdateTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskList\", ctx, request)\n\tret0, _ := ret[0].(*UpdateTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskList indicates an expected call of UpdateTaskList.\nfunc (mr *MockTaskStoreMockRecorder) UpdateTaskList(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskList\", reflect.TypeOf((*MockTaskStore)(nil).UpdateTaskList), ctx, request)\n}\n\n// MockHistoryStore is a mock of HistoryStore interface.\ntype MockHistoryStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHistoryStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockHistoryStoreMockRecorder is the mock recorder for MockHistoryStore.\ntype MockHistoryStoreMockRecorder struct {\n\tmock *MockHistoryStore\n}\n\n// NewMockHistoryStore creates a new mock instance.\nfunc NewMockHistoryStore(ctrl *gomock.Controller) *MockHistoryStore {\n\tmock := &MockHistoryStore{ctrl: ctrl}\n\tmock.recorder = &MockHistoryStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHistoryStore) EXPECT() *MockHistoryStoreMockRecorder {\n\treturn m.recorder\n}\n\n// AppendHistoryNodes mocks base method.\nfunc (m *MockHistoryStore) AppendHistoryNodes(ctx context.Context, request *InternalAppendHistoryNodesRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AppendHistoryNodes\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AppendHistoryNodes indicates an expected call of AppendHistoryNodes.\nfunc (mr *MockHistoryStoreMockRecorder) AppendHistoryNodes(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AppendHistoryNodes\", reflect.TypeOf((*MockHistoryStore)(nil).AppendHistoryNodes), ctx, request)\n}\n\n// Close mocks base method.\nfunc (m *MockHistoryStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockHistoryStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockHistoryStore)(nil).Close))\n}\n\n// DeleteHistoryBranch mocks base method.\nfunc (m *MockHistoryStore) DeleteHistoryBranch(ctx context.Context, request *InternalDeleteHistoryBranchRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteHistoryBranch indicates an expected call of DeleteHistoryBranch.\nfunc (mr *MockHistoryStoreMockRecorder) DeleteHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteHistoryBranch\", reflect.TypeOf((*MockHistoryStore)(nil).DeleteHistoryBranch), ctx, request)\n}\n\n// ForkHistoryBranch mocks base method.\nfunc (m *MockHistoryStore) ForkHistoryBranch(ctx context.Context, request *InternalForkHistoryBranchRequest) (*InternalForkHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ForkHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(*InternalForkHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ForkHistoryBranch indicates an expected call of ForkHistoryBranch.\nfunc (mr *MockHistoryStoreMockRecorder) ForkHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ForkHistoryBranch\", reflect.TypeOf((*MockHistoryStore)(nil).ForkHistoryBranch), ctx, request)\n}\n\n// GetAllHistoryTreeBranches mocks base method.\nfunc (m *MockHistoryStore) GetAllHistoryTreeBranches(ctx context.Context, request *GetAllHistoryTreeBranchesRequest) (*GetAllHistoryTreeBranchesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllHistoryTreeBranches\", ctx, request)\n\tret0, _ := ret[0].(*GetAllHistoryTreeBranchesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllHistoryTreeBranches indicates an expected call of GetAllHistoryTreeBranches.\nfunc (mr *MockHistoryStoreMockRecorder) GetAllHistoryTreeBranches(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllHistoryTreeBranches\", reflect.TypeOf((*MockHistoryStore)(nil).GetAllHistoryTreeBranches), ctx, request)\n}\n\n// GetHistoryTree mocks base method.\nfunc (m *MockHistoryStore) GetHistoryTree(ctx context.Context, request *InternalGetHistoryTreeRequest) (*InternalGetHistoryTreeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryTree\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetHistoryTreeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryTree indicates an expected call of GetHistoryTree.\nfunc (mr *MockHistoryStoreMockRecorder) GetHistoryTree(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryTree\", reflect.TypeOf((*MockHistoryStore)(nil).GetHistoryTree), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockHistoryStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockHistoryStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockHistoryStore)(nil).GetName))\n}\n\n// ReadHistoryBranch mocks base method.\nfunc (m *MockHistoryStore) ReadHistoryBranch(ctx context.Context, request *InternalReadHistoryBranchRequest) (*InternalReadHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadHistoryBranch\", ctx, request)\n\tret0, _ := ret[0].(*InternalReadHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadHistoryBranch indicates an expected call of ReadHistoryBranch.\nfunc (mr *MockHistoryStoreMockRecorder) ReadHistoryBranch(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadHistoryBranch\", reflect.TypeOf((*MockHistoryStore)(nil).ReadHistoryBranch), ctx, request)\n}\n\n// MockConfigStore is a mock of ConfigStore interface.\ntype MockConfigStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConfigStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockConfigStoreMockRecorder is the mock recorder for MockConfigStore.\ntype MockConfigStoreMockRecorder struct {\n\tmock *MockConfigStore\n}\n\n// NewMockConfigStore creates a new mock instance.\nfunc NewMockConfigStore(ctrl *gomock.Controller) *MockConfigStore {\n\tmock := &MockConfigStore{ctrl: ctrl}\n\tmock.recorder = &MockConfigStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConfigStore) EXPECT() *MockConfigStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockConfigStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockConfigStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockConfigStore)(nil).Close))\n}\n\n// FetchConfig mocks base method.\nfunc (m *MockConfigStore) FetchConfig(ctx context.Context, configType ConfigType) (*InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FetchConfig\", ctx, configType)\n\tret0, _ := ret[0].(*InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FetchConfig indicates an expected call of FetchConfig.\nfunc (mr *MockConfigStoreMockRecorder) FetchConfig(ctx, configType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FetchConfig\", reflect.TypeOf((*MockConfigStore)(nil).FetchConfig), ctx, configType)\n}\n\n// UpdateConfig mocks base method.\nfunc (m *MockConfigStore) UpdateConfig(ctx context.Context, value *InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateConfig\", ctx, value)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateConfig indicates an expected call of UpdateConfig.\nfunc (mr *MockConfigStoreMockRecorder) UpdateConfig(ctx, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateConfig\", reflect.TypeOf((*MockConfigStore)(nil).UpdateConfig), ctx, value)\n}\n\n// MockDomainAuditStore is a mock of DomainAuditStore interface.\ntype MockDomainAuditStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainAuditStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainAuditStoreMockRecorder is the mock recorder for MockDomainAuditStore.\ntype MockDomainAuditStoreMockRecorder struct {\n\tmock *MockDomainAuditStore\n}\n\n// NewMockDomainAuditStore creates a new mock instance.\nfunc NewMockDomainAuditStore(ctrl *gomock.Controller) *MockDomainAuditStore {\n\tmock := &MockDomainAuditStore{ctrl: ctrl}\n\tmock.recorder = &MockDomainAuditStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainAuditStore) EXPECT() *MockDomainAuditStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDomainAuditStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDomainAuditStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDomainAuditStore)(nil).Close))\n}\n\n// CreateDomainAuditLog mocks base method.\nfunc (m *MockDomainAuditStore) CreateDomainAuditLog(ctx context.Context, request *InternalCreateDomainAuditLogRequest) (*CreateDomainAuditLogResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDomainAuditLog\", ctx, request)\n\tret0, _ := ret[0].(*CreateDomainAuditLogResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDomainAuditLog indicates an expected call of CreateDomainAuditLog.\nfunc (mr *MockDomainAuditStoreMockRecorder) CreateDomainAuditLog(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDomainAuditLog\", reflect.TypeOf((*MockDomainAuditStore)(nil).CreateDomainAuditLog), ctx, request)\n}\n\n// GetDomainAuditLogs mocks base method.\nfunc (m *MockDomainAuditStore) GetDomainAuditLogs(ctx context.Context, request *GetDomainAuditLogsRequest) (*InternalGetDomainAuditLogsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAuditLogs\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetDomainAuditLogsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainAuditLogs indicates an expected call of GetDomainAuditLogs.\nfunc (mr *MockDomainAuditStoreMockRecorder) GetDomainAuditLogs(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAuditLogs\", reflect.TypeOf((*MockDomainAuditStore)(nil).GetDomainAuditLogs), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockDomainAuditStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockDomainAuditStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockDomainAuditStore)(nil).GetName))\n}\n"
  },
  {
    "path": "common/persistence/data_store_interfaces_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDataBlob(t *testing.T) {\n\tt.Run(\"NewDataBlob\", func(t *testing.T) {\n\t\tassert.Nil(t, NewDataBlob(nil, \"anything\"), \"nil data should become nil blob\")\n\t\tdata, encoding := []byte(\"something\"), constants.EncodingTypeJSON\n\t\tassert.EqualValues(t, &DataBlob{\n\t\t\tData:     data,\n\t\t\tEncoding: encoding,\n\t\t}, NewDataBlob(data, encoding))\n\t})\n\tt.Run(\"Y-prefixed data may panic\", func(t *testing.T) {\n\t\tallEncodings := []constants.EncodingType{\n\t\t\tconstants.EncodingTypeJSON,\n\t\t\tconstants.EncodingTypeThriftRW,\n\t\t\tconstants.EncodingTypeGob,\n\t\t\tconstants.EncodingTypeUnknown,\n\t\t\tconstants.EncodingTypeEmpty,\n\t\t\tconstants.EncodingTypeProto,\n\t\t}\n\t\tconst problematic = \"Y...\"\n\n\t\t// I'm not entirely sure why this behavior exists, but to nail it down:\n\t\tassert.NotPanics(t, func() {\n\t\t\tNewDataBlob([]byte(problematic), constants.EncodingTypeThriftRW)\n\t\t}, \"only thriftrw data can start with Y without panicking\")\n\n\t\t// all others panic\n\t\tfor _, encoding := range allEncodings {\n\t\t\tif encoding == constants.EncodingTypeThriftRW {\n\t\t\t\tcontinue // handled above\n\t\t\t}\n\t\t\tassert.Panicsf(t, func() {\n\t\t\t\tNewDataBlob([]byte(problematic), encoding)\n\t\t\t}, \"non-thriftrw %v should panic if first byte is Y\", encoding)\n\t\t}\n\n\t\t// make sure other data/encoding combinations do not panic for any encoding\n\t\tfor _, dat := range []string{\"Z...\", \"other\"} {\n\t\t\tfor _, encoding := range allEncodings {\n\t\t\t\tassert.NotPanicsf(t, func() {\n\t\t\t\t\tNewDataBlob([]byte(dat), encoding)\n\t\t\t\t}, \"should not panic with %q on encoding %q\", dat, encoding)\n\t\t\t}\n\t\t}\n\t})\n\tt.Run(\"FromDataBlob\", func(t *testing.T) {\n\t\tassertEmpty := func(b []byte, s string) {\n\t\t\tassert.Nil(t, b, \"should have nil bytes\")\n\t\t\tassert.Empty(t, s, \"should have empty string\")\n\t\t}\n\t\tassertEmpty(FromDataBlob(nil))\n\t\tassertEmpty(FromDataBlob(&DataBlob{}))\n\t\tassertEmpty(FromDataBlob(&DataBlob{Encoding: constants.EncodingTypeThriftRW}))\n\n\t\tdat, str := FromDataBlob(&DataBlob{\n\t\t\tData:     []byte(\"data\"),\n\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t})\n\t\tassert.Equal(t, []byte(\"data\"), dat, \"data should be returned\")\n\t\tassert.Equal(t, string(constants.EncodingTypeJSON), str, \"encoding should be returned as a string\")\n\t})\n\tt.Run(\"ToNilSafeDataBlob\", func(t *testing.T) {\n\t\torig := &DataBlob{Encoding: constants.EncodingTypeGob} // anything not empty\n\t\tassert.Equal(t, orig, orig.ToNilSafeDataBlob())\n\t\tassert.NotNil(t, (*DataBlob)(nil).ToNilSafeDataBlob(), \"typed nils should convert to non-nils\")\n\t})\n\tt.Run(\"GetEncodingString\", func(t *testing.T) {\n\t\tassert.Equal(t, \"\", (*DataBlob)(nil).GetEncodingString())\n\t\tassert.Equal(t, \"test\", (&DataBlob{Encoding: \"test\"}).GetEncodingString())\n\t})\n\tt.Run(\"GetData\", func(t *testing.T) {\n\t\t// this method returns empty slices, not nils.\n\t\t// I'm not sure if this needs to be maintained, but it must be checked before changing.\n\t\tassert.Equal(t, []byte{}, (*DataBlob)(nil).GetData())\n\t\tassert.Equal(t, []byte{}, (&DataBlob{}).GetData())\n\t\tassert.Equal(t, []byte{}, (&DataBlob{Data: []byte{}}).GetData())\n\t\tassert.Equal(t, []byte(\"test\"), (&DataBlob{Data: []byte(\"test\")}).GetData())\n\t})\n\tt.Run(\"GetEncoding\", func(t *testing.T) {\n\t\tsame := func(encoding constants.EncodingType) {\n\t\t\tassert.Equal(t, encoding, (&DataBlob{Encoding: encoding}).GetEncoding())\n\t\t}\n\t\tunknown := func(encoding constants.EncodingType) {\n\t\t\tassert.Equal(t, constants.EncodingTypeUnknown, (&DataBlob{Encoding: encoding}).GetEncoding())\n\t\t}\n\t\t// obvious\n\t\tsame(constants.EncodingTypeGob)\n\t\tsame(constants.EncodingTypeJSON)\n\t\tsame(constants.EncodingTypeThriftRW)\n\t\tsame(constants.EncodingTypeEmpty)\n\n\t\t// highly suspicious\n\t\tunknown(constants.EncodingTypeProto)\n\n\t\t// should be unknown\n\t\tunknown(constants.EncodingTypeUnknown)\n\t\tunknown(\"any other value\")\n\t})\n\tt.Run(\"to and from internal\", func(t *testing.T) {\n\t\tdata := []byte(\"some data\")\n\t\tt.Run(\"supported types\", func(t *testing.T) {\n\t\t\tcheck := func(t *testing.T, encodingType types.EncodingType, encodingCommon constants.EncodingType) {\n\t\t\t\tinternal := &types.DataBlob{\n\t\t\t\t\tEncodingType: encodingType.Ptr(),\n\t\t\t\t\tData:         data,\n\t\t\t\t}\n\t\t\t\tblob := &DataBlob{\n\t\t\t\t\tEncoding: encodingCommon,\n\t\t\t\t\tData:     data,\n\t\t\t\t}\n\t\t\t\tassert.Equalf(t, internal, blob.ToInternal(), \"%v should encode to internal type %v\", encodingCommon, encodingType)\n\t\t\t\tassert.Equalf(t, blob, NewDataBlobFromInternal(internal), \"%v should decode from internal type %v\", encodingCommon, encodingType)\n\t\t\t\t// likely proven by above, but to be explicit: this type should round-trip without losing data.\n\t\t\t\tassert.Equalf(t, blob, NewDataBlobFromInternal(blob.ToInternal()), \"%v should round trip from blob %v\", encodingCommon, encodingType)\n\t\t\t\tassert.Equalf(t, internal, NewDataBlobFromInternal(internal).ToInternal(), \"%v should round trip from internal %v\", encodingCommon, encodingType)\n\t\t\t}\n\n\t\t\tt.Run(\"json\", func(t *testing.T) {\n\t\t\t\tcheck(t, types.EncodingTypeJSON, constants.EncodingTypeJSON)\n\t\t\t})\n\t\t\tt.Run(\"thriftrw\", func(t *testing.T) {\n\t\t\t\tcheck(t, types.EncodingTypeThriftRW, constants.EncodingTypeThriftRW)\n\t\t\t})\n\t\t})\n\n\t\tt.Run(\"other known encodings panic to internal\", func(t *testing.T) {\n\t\t\tfor _, encoding := range []constants.EncodingType{\n\t\t\t\tconstants.EncodingTypeUnknown,\n\t\t\t\tconstants.EncodingTypeProto,\n\t\t\t\tconstants.EncodingTypeGob,\n\t\t\t\tconstants.EncodingTypeEmpty,\n\t\t\t\t\"any other value\",\n\t\t\t} {\n\t\t\t\tassert.Panicsf(t, func() {\n\t\t\t\t\t(&DataBlob{\n\t\t\t\t\t\tEncoding: encoding,\n\t\t\t\t\t\tData:     data,\n\t\t\t\t\t}).ToInternal()\n\t\t\t\t}, \"should panic when encoding to unhandled encoding %q\", encoding)\n\t\t\t}\n\t\t})\n\n\t\tt.Run(\"unknown encodings panic from internal\", func(t *testing.T) {\n\t\t\t// these two are known, any other value should panic.\n\t\t\t//\n\t\t\t// the easy and most-likely-to-catch-changes strategy is to just add 1,\n\t\t\t// so a new supported type will automatically fail here until both\n\t\t\t// the code and tests are updated.\n\t\t\tunknownType := 1 + max(types.EncodingTypeJSON, types.EncodingTypeThriftRW)\n\t\t\tassert.Panicsf(t, func() {\n\t\t\t\tNewDataBlobFromInternal(&types.DataBlob{\n\t\t\t\t\tEncodingType: &unknownType,\n\t\t\t\t\tData:         data,\n\t\t\t\t})\n\t\t\t}, \"should panic when decoding from unhandled encoding %q\", unknownType)\n\t\t})\n\t})\n}\n"
  },
  {
    "path": "common/persistence/domain_audit_log.go",
    "content": "package persistence\n\nimport (\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ToFailoverEvent maps a DomainAuditLog to a FailoverEvent\n// by looking at the failover changes and only listing the differences.\n// Active clusters or cluster-attributes which are unchanged will not be listed as a failover event.\n// Since one DomainAuditLog can contain multiple failover changes, this function will return a\n// list of clusterFailver entries for each change. However, it repressents a single atomic\n// change in the domain\nfunc (auditLog *DomainAuditLog) ToFailoverEvents() *types.FailoverEvent {\n\n\tvar clusterFailovers []*types.ClusterFailover\n\n\t// we're comparing the active cluster name rather than the failvoer version ()\n\t// because cluster-attribute updates are a noop bump, to filter out some noise the domain-level active\n\t// cluster will only be reported if it changes\n\tif auditLog.StateBefore.GetReplicationConfig().GetActiveClusterName() != auditLog.StateAfter.GetReplicationConfig().GetActiveClusterName() {\n\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: auditLog.StateBefore.GetReplicationConfig().GetActiveClusterName(),\n\t\t\t\tFailoverVersion:   auditLog.StateBefore.GetFailoverVersion(),\n\t\t\t},\n\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: auditLog.StateAfter.GetReplicationConfig().GetActiveClusterName(),\n\t\t\t\tFailoverVersion:   auditLog.StateAfter.GetFailoverVersion(),\n\t\t\t},\n\t\t})\n\t}\n\n\tclusterAttrBefore := auditLog.StateBefore.GetReplicationConfig().GetClusterAttributeScopes()\n\tclusterAttrAfter := auditLog.StateAfter.GetReplicationConfig().GetClusterAttributeScopes()\n\n\t// Sort keys to ensure deterministic ordering\n\tsortedScopeKeysAfter := make([]string, 0, len(clusterAttrAfter))\n\tfor scopeKey := range clusterAttrAfter {\n\t\tsortedScopeKeysAfter = append(sortedScopeKeysAfter, scopeKey)\n\t}\n\tsort.Strings(sortedScopeKeysAfter)\n\n\t// a scope is thing like 'region' or 'city'\n\tfor _, scopeKeyAfter := range sortedScopeKeysAfter {\n\t\tvAfter := clusterAttrAfter[scopeKeyAfter]\n\t\tvBefore, ok := clusterAttrBefore[scopeKeyAfter]\n\t\tif !ok {\n\t\t\t// if the scope is not even defined in the before state, it's entirely new, this is its introduction\n\t\t\tclusterFailovers = append(clusterFailovers, auditLog.mapNewClusterAttributeToClusterFailover(scopeKeyAfter, &vAfter)...)\n\t\t\tcontinue\n\t\t}\n\n\t\tclusterFailovers = append(clusterFailovers, auditLog.mapClusterAttributeChangeToClusterFailover(scopeKeyAfter, &vBefore, &vAfter)...)\n\t}\n\n\t// Sort keys for deterministic ordering\n\tsortedScopeKeysBefore := make([]string, 0, len(clusterAttrBefore))\n\tfor scopeKey := range clusterAttrBefore {\n\t\tsortedScopeKeysBefore = append(sortedScopeKeysBefore, scopeKey)\n\t}\n\tsort.Strings(sortedScopeKeysBefore)\n\n\tfor _, scopeKeyBefore := range sortedScopeKeysBefore {\n\t\tvBefore := clusterAttrBefore[scopeKeyBefore]\n\t\t_, ok := clusterAttrAfter[scopeKeyBefore]\n\t\tif !ok {\n\t\t\t// if the scope is not even defined in the after state, it's entirely removed, this is its removal\n\t\t\tclusterFailovers = append(clusterFailovers, auditLog.mapRemovedClusterAttributeToClusterFailover(scopeKeyBefore, &vBefore)...)\n\t\t\tcontinue\n\t\t}\n\t}\n\n\tfailoverType := types.FailoverTypeForce\n\tif auditLog.StateAfter.FailoverEndTime != nil {\n\t\tfailoverType = types.FailoverTypeGraceful\n\t}\n\n\treturn &types.FailoverEvent{\n\t\tID:               &auditLog.EventID,\n\t\tCreatedTime:      common.Ptr(auditLog.CreatedTime.UnixNano()),\n\t\tClusterFailovers: clusterFailovers,\n\t\tFailoverType:     &failoverType,\n\t}\n}\n\n// for when there's only a new cluster attribute introduced\nfunc (auditLog *DomainAuditLog) mapNewClusterAttributeToClusterFailover(scopeKey string, vAfter *types.ClusterAttributeScope) []*types.ClusterFailover {\n\tif vAfter == nil && scopeKey == \"\" {\n\t\treturn nil\n\t}\n\tvar clusterFailovers []*types.ClusterFailover\n\n\tsortedNames := make([]string, 0, len(vAfter.ClusterAttributes))\n\tfor name := range vAfter.ClusterAttributes {\n\t\tsortedNames = append(sortedNames, name)\n\t}\n\tsort.Strings(sortedNames)\n\n\tfor _, name := range sortedNames {\n\t\tv := vAfter.ClusterAttributes[name]\n\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: scopeKey,\n\t\t\t\tName:  name,\n\t\t\t},\n\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: v.ActiveClusterName,\n\t\t\t\tFailoverVersion:   v.FailoverVersion,\n\t\t\t},\n\t\t\tFromCluster: nil,\n\t\t})\n\t}\n\treturn clusterFailovers\n}\n\n// for when a cluster attribute is removed\nfunc (auditLog *DomainAuditLog) mapRemovedClusterAttributeToClusterFailover(scopeKey string, vBefore *types.ClusterAttributeScope) []*types.ClusterFailover {\n\tif vBefore == nil && scopeKey == \"\" {\n\t\treturn nil\n\t}\n\tvar clusterFailovers []*types.ClusterFailover\n\n\tsortedNames := make([]string, 0, len(vBefore.ClusterAttributes))\n\tfor name := range vBefore.ClusterAttributes {\n\t\tsortedNames = append(sortedNames, name)\n\t}\n\tsort.Strings(sortedNames)\n\n\tfor _, name := range sortedNames {\n\t\tv := vBefore.ClusterAttributes[name]\n\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: scopeKey,\n\t\t\t\tName:  name,\n\t\t\t},\n\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: v.ActiveClusterName,\n\t\t\t\tFailoverVersion:   v.FailoverVersion,\n\t\t\t},\n\t\t\tToCluster: nil,\n\t\t})\n\t}\n\treturn clusterFailovers\n}\n\n// for comparing if any cluster attributes have changed within a scope\n// will return nil if none are changed\n// assumes that the cluster-attr scope is already checked to exist in both before and after states\nfunc (auditLog *DomainAuditLog) mapClusterAttributeChangeToClusterFailover(scopeKey string, vBefore, vAfter *types.ClusterAttributeScope) []*types.ClusterFailover {\n\n\tvar clusterFailovers []*types.ClusterFailover\n\n\t// Sort keys to ensure deterministic ordering\n\tsortedNamesAfter := make([]string, 0, len(vAfter.ClusterAttributes))\n\tfor name := range vAfter.ClusterAttributes {\n\t\tsortedNamesAfter = append(sortedNamesAfter, name)\n\t}\n\tsort.Strings(sortedNamesAfter)\n\n\tfor _, name := range sortedNamesAfter {\n\t\tvAfterAttr := vAfter.ClusterAttributes[name]\n\t\tvBeforeAttr, ok := vBefore.ClusterAttributes[name]\n\t\tif !ok {\n\t\t\t// new cluster attribute\n\t\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: scopeKey,\n\t\t\t\t\tName:  name,\n\t\t\t\t},\n\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: vAfterAttr.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   vAfterAttr.FailoverVersion,\n\t\t\t\t},\n\t\t\t\tFromCluster: nil,\n\t\t\t})\n\t\t\tcontinue\n\t\t}\n\n\t\t// if the cluster attribute has changed\n\t\tif vBeforeAttr.FailoverVersion != vAfterAttr.FailoverVersion {\n\t\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: scopeKey,\n\t\t\t\t\tName:  name,\n\t\t\t\t},\n\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: vBeforeAttr.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   vBeforeAttr.FailoverVersion,\n\t\t\t\t},\n\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: vAfterAttr.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   vAfterAttr.FailoverVersion,\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\n\tsortedNamesBefore := make([]string, 0, len(vBefore.ClusterAttributes))\n\tfor name := range vBefore.ClusterAttributes {\n\t\tsortedNamesBefore = append(sortedNamesBefore, name)\n\t}\n\tsort.Strings(sortedNamesBefore)\n\n\tfor _, name := range sortedNamesBefore {\n\t\tvBeforeAttr := vBefore.ClusterAttributes[name]\n\t\t_, ok := vAfter.ClusterAttributes[name]\n\t\tif !ok {\n\t\t\t// removed cluster attribute\n\t\t\tclusterFailovers = append(clusterFailovers, &types.ClusterFailover{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: scopeKey,\n\t\t\t\t\tName:  name,\n\t\t\t\t},\n\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: vBeforeAttr.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   vBeforeAttr.FailoverVersion,\n\t\t\t\t},\n\t\t\t\tToCluster: nil,\n\t\t\t})\n\t\t\tcontinue\n\t\t}\n\t}\n\treturn clusterFailovers\n}\n"
  },
  {
    "path": "common/persistence/domain_audit_log_test.go",
    "content": "package persistence\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDomainAuditLog_ToFailoverEvents(t *testing.T) {\n\tnow := time.Unix(1234567890, 0)\n\n\ttests := map[string]struct {\n\t\tauditLog *DomainAuditLog\n\t\texpected *types.FailoverEvent\n\t}{\n\t\t\"simple active cluster failover\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-1\",\n\t\t\t\tDomainID:    \"domain-1\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\t\tID:   \"domain-1\",\n\t\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: \"cluster-us-east\"},\n\t\t\t\t\t\t\t{ClusterName: \"cluster-us-west\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 100,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\t\tID:   \"domain-1\",\n\t\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-us-west\",\n\t\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: \"cluster-us-east\"},\n\t\t\t\t\t\t\t{ClusterName: \"cluster-us-west\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 101,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-1\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-west\",\n\t\t\t\t\t\t\tFailoverVersion:   101,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"cluster attribute failover - single region change\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-2\",\n\t\t\t\tDomainID:    \"active-active-domain\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\t\tID:   \"domain-2\",\n\t\t\t\t\t\tName: \"active-active-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster0\", FailoverVersion: 200},\n\t\t\t\t\t\t\t\t\t\t\"us-west\": {ActiveClusterName: \"cluster1\", FailoverVersion: 201},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 200,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\t\tID:   \"domain-2\",\n\t\t\t\t\t\tName: \"active-active-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster1\", FailoverVersion: 201},\n\t\t\t\t\t\t\t\t\t\t\"us-west\": {ActiveClusterName: \"cluster1\", FailoverVersion: 201},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 201,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-2\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   201,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"cluster attribute failover - multiple regions change\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-3\",\n\t\t\t\tDomainID:    \"domain-3\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"region-1\": {ActiveClusterName: \"cluster-us-east\", FailoverVersion: 300},\n\t\t\t\t\t\t\t\t\t\t\"region-2\": {ActiveClusterName: \"cluster-us-west\", FailoverVersion: 301},\n\t\t\t\t\t\t\t\t\t\t\"region-3\": {ActiveClusterName: \"cluster-eu-central\", FailoverVersion: 302},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 300,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"region-1\": {ActiveClusterName: \"cluster-eu-central\", FailoverVersion: 302},\n\t\t\t\t\t\t\t\t\t\t\"region-2\": {ActiveClusterName: \"cluster-eu-central\", FailoverVersion: 302},\n\t\t\t\t\t\t\t\t\t\t\"region-3\": {ActiveClusterName: \"cluster-eu-central\", FailoverVersion: 302},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 400,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-3\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-eu-central\",\n\t\t\t\t\t\t\tFailoverVersion:   302,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"region-1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-west\",\n\t\t\t\t\t\t\tFailoverVersion:   301,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-eu-central\",\n\t\t\t\t\t\t\tFailoverVersion:   302,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"region-2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"combined failover - both active cluster and attributes\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-4\",\n\t\t\t\tDomainID:    \"domain-4\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-primary\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us\": {ActiveClusterName: \"cluster-us\", FailoverVersion: 400},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 400,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-secondary\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us\": {ActiveClusterName: \"cluster-us-backup\", FailoverVersion: 401},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 401,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-4\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-primary\",\n\t\t\t\t\t\t\tFailoverVersion:   400,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-secondary\",\n\t\t\t\t\t\t\tFailoverVersion:   401,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us\",\n\t\t\t\t\t\t\tFailoverVersion:   400,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-backup\",\n\t\t\t\t\t\t\tFailoverVersion:   401,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"new cluster attribute scope added\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-5\",\n\t\t\t\tDomainID:    \"domain-5\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 500,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-us-east\", FailoverVersion: 501},\n\t\t\t\t\t\t\t\t\t\t\"eu\":      {ActiveClusterName: \"cluster-eu\", FailoverVersion: 502},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 502,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-5\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-eu\",\n\t\t\t\t\t\t\tFailoverVersion:   502,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"eu\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\t\tFailoverVersion:   501,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"no changes - same active cluster\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-6\",\n\t\t\t\tDomainID:    \"domain-6\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-stable\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 600,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-stable\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 600,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:               stringPtr(\"event-6\"),\n\t\t\t\tCreatedTime:      int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType:     types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: nil,\n\t\t\t},\n\t\t},\n\t\t\"nil replication config\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:       \"event-7\",\n\t\t\t\tDomainID:      \"domain-7\",\n\t\t\t\tCreatedTime:   now,\n\t\t\t\tStateBefore:   &GetDomainResponse{},\n\t\t\t\tStateAfter:    &GetDomainResponse{},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:               stringPtr(\"event-7\"),\n\t\t\t\tFailoverType:     types.FailoverTypeForce.Ptr(),\n\t\t\t\tCreatedTime:      int64Ptr(now.UnixNano()),\n\t\t\t\tClusterFailovers: nil,\n\t\t\t},\n\t\t},\n\t\t\"multiple scopes with changes\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-8\",\n\t\t\t\tDomainID:    \"domain-8\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us\": {ActiveClusterName: \"cluster-us-1\", FailoverVersion: 700},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"dc1\": {ActiveClusterName: \"cluster-dc1-primary\", FailoverVersion: 700},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 700,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us\": {ActiveClusterName: \"cluster-us-2\", FailoverVersion: 701},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"dc1\": {ActiveClusterName: \"cluster-dc1-backup\", FailoverVersion: 701},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 701,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-8\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-dc1-primary\",\n\t\t\t\t\t\t\tFailoverVersion:   700,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-dc1-backup\",\n\t\t\t\t\t\t\tFailoverVersion:   701,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"datacenter\",\n\t\t\t\t\t\t\tName:  \"dc1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-1\",\n\t\t\t\t\t\t\tFailoverVersion:   700,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-2\",\n\t\t\t\t\t\t\tFailoverVersion:   701,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"cluster attribute scope removed\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-9\",\n\t\t\t\tDomainID:    \"domain-9\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-us-east\", FailoverVersion: 800},\n\t\t\t\t\t\t\t\t\t\t\"us-west\": {ActiveClusterName: \"cluster-us-west\", FailoverVersion: 801},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 800,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster-default\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 801,\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-9\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeForce.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\t\tFailoverVersion:   800,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: nil,\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-west\",\n\t\t\t\t\t\t\tFailoverVersion:   801,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: nil,\n\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\tName:  \"us-west\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"graceful failover\": {\n\t\t\tauditLog: &DomainAuditLog{\n\t\t\t\tEventID:     \"event-9\",\n\t\t\t\tDomainID:    \"domain-9\",\n\t\t\t\tCreatedTime: now,\n\t\t\t\tStateBefore: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"region-us-east\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 800,\n\t\t\t\t},\n\t\t\t\tStateAfter: &GetDomainResponse{\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"region-us-west\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverVersion: 801,\n\t\t\t\t\tFailoverEndTime: int64Ptr(now.Add(1 * time.Hour).UnixNano()),\n\t\t\t\t},\n\t\t\t\tOperationType: DomainAuditOperationTypeFailover,\n\t\t\t},\n\t\t\texpected: &types.FailoverEvent{\n\t\t\t\tID:           stringPtr(\"event-9\"),\n\t\t\t\tCreatedTime:  int64Ptr(now.UnixNano()),\n\t\t\t\tFailoverType: types.FailoverTypeGraceful.Ptr(),\n\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t{\n\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"region-us-east\",\n\t\t\t\t\t\t\tFailoverVersion:   800,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tToCluster: &types.ActiveClusterInfo{\n\t\t\t\t\t\t\tActiveClusterName: \"region-us-west\",\n\t\t\t\t\t\t\tFailoverVersion:   801,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactual := tc.auditLog.ToFailoverEvents()\n\t\t\tassert.Equal(t, tc.expected, actual)\n\t\t})\n\t}\n}\n\nfunc TestSerializeDeserializeGetDomainResponse_Roundtrip(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput        *GetDomainResponse\n\t\tencodingType constants.EncodingType\n\t}{\n\t\t\"nil response\": {\n\t\t\tinput:        nil,\n\t\t\tencodingType: constants.EncodingTypeThriftRWSnappy,\n\t\t},\n\t\t\"empty response with ThriftRW\": {\n\t\t\tinput:        &GetDomainResponse{},\n\t\t\tencodingType: constants.EncodingTypeThriftRW,\n\t\t},\n\t\t\"empty response with ThriftRWSnappy\": {\n\t\t\tinput:        &GetDomainResponse{},\n\t\t\tencodingType: constants.EncodingTypeThriftRWSnappy,\n\t\t},\n\t\t\"full response with all fields\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:          \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      0,\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                7,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"test://visibility\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:              true,\n\t\t\t\tFailoverVersion:             100,\n\t\t\t\tConfigVersion:               5,\n\t\t\t\tFailoverNotificationVersion: 99,\n\t\t\t\tPreviousFailoverVersion:     98,\n\t\t\t\tLastUpdatedTime:             1234567890,\n\t\t\t\tNotificationVersion:         10,\n\t\t\t},\n\t\t\tencodingType: constants.EncodingTypeThriftRWSnappy,\n\t\t},\n\t\t\"response with failover info\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0,\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tLastUpdatedTime: 1000,\n\t\t\t\tFailoverEndTime: func() *int64 { t := int64(2000); return &t }(),\n\t\t\t},\n\t\t\tencodingType: constants.EncodingTypeThriftRW,\n\t\t},\n\t\t\"response with nil clusters in ReplicationConfig\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 1,\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tencodingType: constants.EncodingTypeThriftRWSnappy,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Serialize\n\t\t\tblob, err := serializeGetDomainResponse(tc.input, tc.encodingType)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif tc.input == nil {\n\t\t\t\tassert.Nil(t, blob)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.NotNil(t, blob)\n\t\t\tassert.Equal(t, tc.encodingType, blob.Encoding)\n\t\t\tassert.NotEmpty(t, blob.Data)\n\n\t\t\t// Deserialize\n\t\t\tdeserialized, err := deserializeGetDomainResponse(blob)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, deserialized)\n\n\t\t\t// Build expected result accounting for non-symmetric fields\n\t\t\texpected := &GetDomainResponse{\n\t\t\t\tInfo:              tc.input.Info,\n\t\t\t\tConfig:            tc.input.Config,\n\t\t\t\tReplicationConfig: tc.input.ReplicationConfig,\n\t\t\t\tIsGlobalDomain:    tc.input.IsGlobalDomain,\n\t\t\t\tFailoverVersion:   tc.input.FailoverVersion,\n\t\t\t\tLastUpdatedTime:   tc.input.LastUpdatedTime,\n\t\t\t\t// Fields below are not present in types.DescribeDomainResponse and don't roundtrip:\n\t\t\t\tFailoverNotificationVersion: 0,\n\t\t\t\tPreviousFailoverVersion:     0,\n\t\t\t\tFailoverEndTime:             nil,\n\t\t\t\tConfigVersion:               0,\n\t\t\t\tNotificationVersion:         0,\n\t\t\t}\n\n\t\t\t// Apply transformations that the mapper does\n\t\t\tif expected.Info != nil {\n\t\t\t\t// Clamp Status to valid range (0-2) as ToType does\n\t\t\t\tif expected.Info.Status < 0 || expected.Info.Status > 2 {\n\t\t\t\t\texpected.Info.Status = 0\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tassert.Equal(t, expected, deserialized)\n\t\t})\n\t}\n}\n\nfunc TestDeserializeGetDomainResponse_EmptyBlob(t *testing.T) {\n\ttests := map[string]struct {\n\t\tblob     *DataBlob\n\t\texpected *GetDomainResponse\n\t}{\n\t\t\"nil blob\": {\n\t\t\tblob:     nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty data\": {\n\t\t\tblob: &DataBlob{\n\t\t\t\tData:     []byte{},\n\t\t\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t},\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult, err := deserializeGetDomainResponse(tc.blob)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestDeserializeGetDomainResponse_InvalidEncoding(t *testing.T) {\n\tblob := &DataBlob{\n\t\tData:     []byte(\"some data\"),\n\t\tEncoding: constants.EncodingType(\"invalid\"),\n\t}\n\n\tresult, err := deserializeGetDomainResponse(blob)\n\tassert.Error(t, err)\n\tassert.Nil(t, result)\n\tassert.IsType(t, &UnknownEncodingTypeError{}, err)\n}\n\n// Helper functions for creating pointers\nfunc stringPtr(s string) *string {\n\treturn &s\n}\n\nfunc int64Ptr(i int64) *int64 {\n\treturn &i\n}\n"
  },
  {
    "path": "common/persistence/domain_audit_manager.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/golang/snappy\"\n\t\"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\ntype (\n\t// domainAuditManagerImpl implements DomainAuditManager based on DomainAuditStore and PayloadSerializer\n\t//\n\t// a future implementation which may wish to get-by-ID should use the following logic to do so\n\t// eventID, _ := uuid.Parse(request.EventID)\n\t// createdTime := time.Unix(eventID.Time().UnixTime())\n\t// return m.persistence.GetDomainAuditLog(ctx, &GetDomainAuditLogsRequest{\n\t// \t  DomainID: request.DomainID,\n\t// \t  OperationType: request.OperationType,\n\t// \t  EventID: request.EventID,\n\t// \t  CreatedTime: createdTime, // part of the primary key\n\t// })\n\tdomainAuditManagerImpl struct {\n\t\tserializer  PayloadSerializer\n\t\tpersistence DomainAuditStore\n\t\tlogger      log.Logger\n\t\ttimeSrc     clock.TimeSource\n\t\tdc          *DynamicConfiguration\n\t}\n)\n\n// NewDomainAuditManagerImpl returns new DomainAuditManager\nfunc NewDomainAuditManagerImpl(persistence DomainAuditStore, logger log.Logger, serializer PayloadSerializer, dc *DynamicConfiguration) DomainAuditManager {\n\treturn &domainAuditManagerImpl{\n\t\tserializer:  serializer,\n\t\tpersistence: persistence,\n\t\tlogger:      logger,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t\tdc:          dc,\n\t}\n}\n\nfunc (m *domainAuditManagerImpl) GetName() string {\n\treturn m.persistence.GetName()\n}\n\nfunc (m *domainAuditManagerImpl) Close() {\n\tm.persistence.Close()\n}\n\nfunc (m *domainAuditManagerImpl) CreateDomainAuditLog(\n\tctx context.Context,\n\trequest *CreateDomainAuditLogRequest,\n) (*CreateDomainAuditLogResponse, error) {\n\n\t// validation\n\teventID, err := uuid.Parse(request.EventID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to parse event ID: %w\", err)\n\t}\n\tif eventID.Version() != 7 {\n\t\treturn nil, fmt.Errorf(\"event ID must be a UUID v7: %w\", err)\n\t}\n\n\tencodingType := constants.EncodingTypeThriftRWSnappy\n\n\t// Serialize StateBefore using thrift+snappy\n\t// To support non-nullable columns in SQL databases we serialize an empty GetDomainResponse{} if nil\n\tstateBefore := request.StateBefore\n\tif stateBefore == nil {\n\t\tstateBefore = &GetDomainResponse{}\n\t}\n\tstateBeforeBlob, err := serializeGetDomainResponse(stateBefore, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Serialize StateAfter using thrift+snappy\n\t// To support non-nullable columns in SQL databases we serialize an empty GetDomainResponse{} if nil\n\tstateAfter := request.StateAfter\n\tif stateAfter == nil {\n\t\tif request.OperationType != DomainAuditOperationTypeDelete {\n\t\t\tm.logger.Warn(\"Domain has been updated to an empty state, this could be a bug\", tag.WorkflowDomainID(request.DomainID), tag.DomainAuditOperationType(request.OperationType))\n\t\t}\n\t\tstateAfter = &GetDomainResponse{}\n\t}\n\tstateAfterBlob, err := serializeGetDomainResponse(stateAfter, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Get TTL from dynamic config using domain ID\n\tttlDuration := m.dc.DomainAuditLogTTL(request.DomainID)\n\n\treturn m.persistence.CreateDomainAuditLog(ctx, &InternalCreateDomainAuditLogRequest{\n\t\tDomainID:        request.DomainID,\n\t\tEventID:         request.EventID,\n\t\tStateBefore:     stateBeforeBlob,\n\t\tStateAfter:      stateAfterBlob,\n\t\tOperationType:   request.OperationType,\n\t\tCreatedTime:     request.CreatedTime,\n\t\tLastUpdatedTime: request.CreatedTime,\n\t\tIdentity:        request.Identity,\n\t\tIdentityType:    request.IdentityType,\n\t\tComment:         request.Comment,\n\t\tTTLSeconds:      int64(ttlDuration.Seconds()),\n\t})\n}\n\nfunc (m *domainAuditManagerImpl) GetDomainAuditLogs(\n\tctx context.Context,\n\trequest *GetDomainAuditLogsRequest,\n) (*GetDomainAuditLogsResponse, error) {\n\treq := *request\n\tif req.MinCreatedTime == nil {\n\t\tstart := time.Unix(0, 0)\n\t\treq.MinCreatedTime = &start\n\t}\n\tif req.MaxCreatedTime == nil {\n\t\tend := m.timeSrc.Now()\n\t\treq.MaxCreatedTime = &end\n\t}\n\n\tinternalResp, err := m.persistence.GetDomainAuditLogs(ctx, &req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Convert internal audit logs to external format\n\tauditLogs := make([]*DomainAuditLog, len(internalResp.AuditLogs))\n\tfor i, internalLog := range internalResp.AuditLogs {\n\t\tlog := &DomainAuditLog{\n\t\t\tEventID:         internalLog.EventID,\n\t\t\tDomainID:        internalLog.DomainID,\n\t\t\tOperationType:   internalLog.OperationType,\n\t\t\tCreatedTime:     internalLog.CreatedTime,\n\t\t\tLastUpdatedTime: internalLog.LastUpdatedTime,\n\t\t\tIdentity:        internalLog.Identity,\n\t\t\tIdentityType:    internalLog.IdentityType,\n\t\t\tComment:         internalLog.Comment,\n\t\t}\n\n\t\t// Deserialize StateBefore\n\t\tif internalLog.StateBefore != nil && len(internalLog.StateBefore.Data) > 0 {\n\t\t\tstateBefore, err := deserializeGetDomainResponse(internalLog.StateBefore)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\t// Only set if not an empty sentinel value\n\t\t\tif !isEmptyGetDomainResponse(stateBefore) {\n\t\t\t\tlog.StateBefore = stateBefore\n\t\t\t}\n\t\t}\n\n\t\t// Deserialize StateAfter\n\t\tif internalLog.StateAfter != nil && len(internalLog.StateAfter.Data) > 0 {\n\t\t\tstateAfter, err := deserializeGetDomainResponse(internalLog.StateAfter)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\t// Only set if not an empty sentinel value\n\t\t\tif !isEmptyGetDomainResponse(stateAfter) {\n\t\t\t\tlog.StateAfter = stateAfter\n\t\t\t}\n\t\t}\n\n\t\tauditLogs[i] = log\n\t}\n\n\treturn &GetDomainAuditLogsResponse{\n\t\tAuditLogs:     auditLogs,\n\t\tNextPageToken: internalResp.NextPageToken,\n\t}, nil\n}\n\n// serializeGetDomainResponse serializes GetDomainResponse using thrift encoding\nfunc serializeGetDomainResponse(resp *GetDomainResponse, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif resp == nil {\n\t\treturn nil, nil\n\t}\n\n\ttypesResp := resp.ToType()\n\n\tif typesResp.ReplicationConfiguration != nil && typesResp.ReplicationConfiguration.Clusters != nil {\n\t\tfiltered := make([]*types.ClusterReplicationConfiguration, 0, len(typesResp.ReplicationConfiguration.Clusters))\n\t\tfor _, cluster := range typesResp.ReplicationConfiguration.Clusters {\n\t\t\tif cluster != nil {\n\t\t\t\tfiltered = append(filtered, cluster)\n\t\t\t}\n\t\t}\n\t\ttypesResp.ReplicationConfiguration.Clusters = filtered\n\t}\n\n\tthriftResp := thrift.FromDescribeDomainResponse(typesResp)\n\n\tvar data []byte\n\tvar err error\n\n\tencoder := codec.NewThriftRWEncoder()\n\tswitch encodingType {\n\tcase constants.EncodingTypeThriftRW:\n\t\tdata, err = encoder.Encode(thriftResp)\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\tencoded, encodeErr := encoder.Encode(thriftResp)\n\t\tif encodeErr != nil {\n\t\t\treturn nil, encodeErr\n\t\t}\n\t\tdata = snappy.Encode(nil, encoded)\n\tdefault:\n\t\t// Fallback to ThriftRWSnappy\n\t\tencodingType = constants.EncodingTypeThriftRWSnappy\n\t\tencoded, encodeErr := encoder.Encode(thriftResp)\n\t\tif encodeErr != nil {\n\t\t\treturn nil, encodeErr\n\t\t}\n\t\tdata = snappy.Encode(nil, encoded)\n\t}\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &DataBlob{\n\t\tData:     data,\n\t\tEncoding: encodingType,\n\t}, nil\n}\n\n// deserializeGetDomainResponse deserializes GetDomainResponse from thrift format\nfunc deserializeGetDomainResponse(blob *DataBlob) (*GetDomainResponse, error) {\n\tif blob == nil || len(blob.Data) == 0 {\n\t\treturn nil, nil\n\t}\n\n\tvar data []byte\n\tvar err error\n\n\t// Decode based on encoding type\n\tswitch blob.Encoding {\n\tcase constants.EncodingTypeThriftRW:\n\t\tdata = blob.Data\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\tdata, err = snappy.Decode(nil, blob.Data)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\tdefault:\n\t\treturn nil, &UnknownEncodingTypeError{encodingType: blob.Encoding}\n\t}\n\n\t// Decode thrift\n\tencoder := codec.NewThriftRWEncoder()\n\tvar thriftResp shared.DescribeDomainResponse\n\terr = encoder.Decode(data, &thriftResp)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Convert from thrift to types.DescribeDomainResponse\n\ttypesResp := thrift.ToDescribeDomainResponse(&thriftResp)\n\n\t// Convert types.DescribeDomainResponse to persistence.GetDomainResponse\n\treturn FromDescribeDomainResponse(typesResp), nil\n}\n\n// isEmptyGetDomainResponse checks if a GetDomainResponse is empty (sentinel value for nil)\nfunc isEmptyGetDomainResponse(resp *GetDomainResponse) bool {\n\tif resp == nil {\n\t\treturn true\n\t}\n\treturn resp.Info == nil && resp.Config == nil && resp.ReplicationConfig == nil\n}\n"
  },
  {
    "path": "common/persistence/domain_audit_manager_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nvar testTimeNow = time.Date(2024, 12, 30, 23, 59, 59, 0, time.UTC)\n\nfunc setUpMocksForDomainAuditManager(t *testing.T) (*domainAuditManagerImpl, *MockDomainAuditStore) {\n\tt.Helper()\n\n\tctrl := gomock.NewController(t)\n\tmockStore := NewMockDomainAuditStore(ctrl)\n\n\tm := &domainAuditManagerImpl{\n\t\tpersistence: mockStore,\n\t\ttimeSrc:     clock.NewMockedTimeSourceAt(testTimeNow),\n\t\tserializer:  NewPayloadSerializer(),\n\t\tlogger:      log.NewNoop(),\n\t\tdc: &DynamicConfiguration{\n\t\t\tDomainAuditLogTTL: func(domainID string) time.Duration { return time.Hour * 24 * 365 },\n\t\t},\n\t}\n\n\treturn m, mockStore\n}\n\nfunc TestGetDomainAuditLogs(t *testing.T) {\n\tctx := context.Background()\n\tepoch := time.Unix(0, 0)\n\tminTime := time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)\n\tfixedTime := time.Date(2024, 6, 30, 23, 59, 59, 0, time.UTC)\n\n\ttestCases := []struct {\n\t\tname            string\n\t\trequest         *GetDomainAuditLogsRequest\n\t\texpectedRequest *GetDomainAuditLogsRequest\n\t\tstoreResp       *InternalGetDomainAuditLogsResponse\n\t\tstoreErr        error\n\t\twantErr         bool\n\t}{\n\t\t{\n\t\t\tname: \"nil MinCreatedTime defaults to epoch\",\n\t\t\trequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMaxCreatedTime: &fixedTime,\n\t\t\t},\n\t\t\texpectedRequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &epoch,\n\t\t\t\tMaxCreatedTime: &fixedTime,\n\t\t\t},\n\t\t\tstoreResp: &InternalGetDomainAuditLogsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil MaxCreatedTime defaults to timeSrc.Now()\",\n\t\t\trequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t},\n\t\t\texpectedRequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\tstoreResp: &InternalGetDomainAuditLogsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"both nil times defaulted\",\n\t\t\trequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"domain-1\",\n\t\t\t},\n\t\t\texpectedRequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &epoch,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\tstoreResp: &InternalGetDomainAuditLogsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil times passed through unchanged\",\n\t\t\trequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\texpectedRequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\tstoreResp: &InternalGetDomainAuditLogsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"store error propagated\",\n\t\t\trequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\texpectedRequest: &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-1\",\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &testTimeNow,\n\t\t\t},\n\t\t\tstoreErr: errors.New(\"store error\"),\n\t\t\twantErr:  true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tm, mockStore := setUpMocksForDomainAuditManager(t)\n\t\t\tmockStore.EXPECT().GetDomainAuditLogs(ctx, tc.expectedRequest).Return(tc.storeResp, tc.storeErr).Times(1)\n\n\t\t\tresp, err := m.GetDomainAuditLogs(ctx, tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomainAuditLogs_MutateDefaultTimeOnRetry(t *testing.T) {\n\tm, mockStore := setUpMocksForDomainAuditManager(t)\n\tmockTime := m.timeSrc.(clock.MockedTimeSource)\n\n\tt1 := mockTime.Now()\n\n\t// a request with nil MaxCreatedTime (meaning \"up to now\")\n\trequest := &GetDomainAuditLogsRequest{\n\t\tDomainID: \"test-domain\",\n\t}\n\n\t// first call at T1\n\tmockStore.EXPECT().GetDomainAuditLogs(gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, req *GetDomainAuditLogsRequest) (*InternalGetDomainAuditLogsResponse, error) {\n\t\t\tassert.Equal(t, t1, *req.MaxCreatedTime)\n\t\t\treturn &InternalGetDomainAuditLogsResponse{}, nil\n\t\t}).Times(1)\n\n\t_, _ = m.GetDomainAuditLogs(context.Background(), request)\n\n\tmockTime.Advance(2 * time.Minute)\n\tt2 := t1.Add(2 * time.Minute)\n\n\t// second call at T2 (like in retry loop)\n\tmockStore.EXPECT().GetDomainAuditLogs(gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, req *GetDomainAuditLogsRequest) (*InternalGetDomainAuditLogsResponse, error) {\n\t\t\tif *req.MaxCreatedTime == t1 {\n\t\t\t\tt.Errorf(\"MaxCreatedTime is still T1 (%v) but should be new Time.Now T2 (%v)\", t1, t2)\n\t\t\t}\n\t\t\treturn &InternalGetDomainAuditLogsResponse{}, nil\n\t\t}).Times(1)\n\n\t_, _ = m.GetDomainAuditLogs(context.Background(), request)\n}\n\nfunc TestCreateDomainAuditLog_SerializeStates(t *testing.T) {\n\t// This test verifies that nil StateBefore/StateAfter are serialized as empty GetDomainResponse{}\n\t// instead of being left as nil, allowing us to maintain NOT NULL database columns.\n\n\tdomainResponse := &GetDomainResponse{\n\t\tInfo: &DomainInfo{\n\t\t\tID:   \"domain-123\",\n\t\t\tName: \"test-domain\",\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname              string\n\t\tstateBefore       *GetDomainResponse\n\t\tstateAfter        *GetDomainResponse\n\t\toperationType     DomainAuditOperationType\n\t\texpectEmptyBefore bool\n\t\texpectEmptyAfter  bool\n\t}{\n\t\t{\n\t\t\tname:              \"correctly serializes nil StateBefore as an empty GetDomainResponse{}\",\n\t\t\tstateBefore:       nil,\n\t\t\tstateAfter:        domainResponse,\n\t\t\toperationType:     DomainAuditOperationTypeCreate,\n\t\t\texpectEmptyBefore: true,\n\t\t\texpectEmptyAfter:  false,\n\t\t},\n\t\t{\n\t\t\tname:              \"correctly serializes nil StateAfter as an empty GetDomainResponse{}\",\n\t\t\tstateBefore:       domainResponse,\n\t\t\tstateAfter:        nil,\n\t\t\toperationType:     DomainAuditOperationTypeDelete,\n\t\t\texpectEmptyBefore: false,\n\t\t\texpectEmptyAfter:  true,\n\t\t},\n\t\t{\n\t\t\tname:              \"correctly serializes both StateBefore and StateAfter as is\",\n\t\t\tstateBefore:       domainResponse,\n\t\t\tstateAfter:        domainResponse,\n\t\t\toperationType:     DomainAuditOperationTypeUpdate,\n\t\t\texpectEmptyBefore: false,\n\t\t\texpectEmptyAfter:  false,\n\t\t},\n\t\t{\n\t\t\tname:              \"correctly serializes both StateBefore and StateAfter as empty GetDomainResponse{}\",\n\t\t\tstateBefore:       nil,\n\t\t\tstateAfter:        nil,\n\t\t\toperationType:     DomainAuditOperationTypeUpdate,\n\t\t\texpectEmptyBefore: true,\n\t\t\texpectEmptyAfter:  true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tm, mockStore := setUpMocksForDomainAuditManager(t)\n\t\t\tctx := context.Background()\n\n\t\t\teventID := uuid.Must(uuid.NewV7()).String()\n\t\t\tcreatedTime := time.Now()\n\n\t\t\tmockStore.EXPECT().CreateDomainAuditLog(ctx, gomock.Any()).\n\t\t\t\tDoAndReturn(func(ctx context.Context, req *InternalCreateDomainAuditLogRequest) (*CreateDomainAuditLogResponse, error) {\n\t\t\t\t\t// Both blobs should ALWAYS be non-nil with non-empty data\n\t\t\t\t\tassert.NotNil(t, req.StateBefore, \"StateBefore blob should never be nil\")\n\t\t\t\t\tassert.NotNil(t, req.StateBefore.Data, \"StateBefore.Data should not be nil\")\n\t\t\t\t\tassert.NotEmpty(t, req.StateBefore.Data, \"StateBefore.Data should contain serialized bytes\")\n\n\t\t\t\t\tassert.NotNil(t, req.StateAfter, \"StateAfter blob should never be nil\")\n\t\t\t\t\tassert.NotNil(t, req.StateAfter.Data, \"StateAfter.Data should not be nil\")\n\t\t\t\t\tassert.NotEmpty(t, req.StateAfter.Data, \"StateAfter.Data should contain serialized bytes\")\n\n\t\t\t\t\t// Verify whether they're empty or populated\n\t\t\t\t\tdeserializedBefore, err := deserializeGetDomainResponse(req.StateBefore)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.Equal(t, tt.expectEmptyBefore, isEmptyGetDomainResponse(deserializedBefore),\n\t\t\t\t\t\t\"StateBefore empty state mismatch\")\n\n\t\t\t\t\tdeserializedAfter, err := deserializeGetDomainResponse(req.StateAfter)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.Equal(t, tt.expectEmptyAfter, isEmptyGetDomainResponse(deserializedAfter),\n\t\t\t\t\t\t\"StateAfter empty state mismatch\")\n\n\t\t\t\t\treturn &CreateDomainAuditLogResponse{EventID: eventID}, nil\n\t\t\t\t}).Times(1)\n\n\t\t\tresp, err := m.CreateDomainAuditLog(ctx, &CreateDomainAuditLogRequest{\n\t\t\t\tDomainID:      \"domain-123\",\n\t\t\t\tEventID:       eventID,\n\t\t\t\tCreatedTime:   createdTime,\n\t\t\t\tStateBefore:   tt.stateBefore,\n\t\t\t\tStateAfter:    tt.stateAfter,\n\t\t\t\tOperationType: tt.operationType,\n\t\t\t\tIdentity:      \"test-user\",\n\t\t\t\tIdentityType:  \"user\",\n\t\t\t\tComment:       \"test operation\",\n\t\t\t})\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, resp)\n\t\t\tassert.Equal(t, eventID, resp.EventID)\n\t\t})\n\t}\n}\n\nfunc TestGetDomainAuditLogs_DeserializesEmptyResponsesToNil(t *testing.T) {\n\tdomainResponse := &GetDomainResponse{\n\t\tInfo: &DomainInfo{\n\t\t\tID:   \"domain-123\",\n\t\t\tName: \"test-domain\",\n\t\t},\n\t}\n\temptyResponse := &GetDomainResponse{}\n\n\ttests := []struct {\n\t\tname                string\n\t\tstateBefore         *GetDomainResponse\n\t\tstateAfter          *GetDomainResponse\n\t\texpectNilBefore     bool\n\t\texpectNilAfter      bool\n\t\texpectDomainIDAfter bool\n\t}{\n\t\t{\n\t\t\tname:                \"when StateBefore is nil, it should be deserialized as an empty GetDomainResponse{}\",\n\t\t\tstateBefore:         emptyResponse,\n\t\t\tstateAfter:          domainResponse,\n\t\t\texpectNilBefore:     true,\n\t\t\texpectNilAfter:      false,\n\t\t\texpectDomainIDAfter: true,\n\t\t},\n\t\t{\n\t\t\tname:                \"when StateAfter is nil, it should be deserialized as an empty GetDomainResponse{}\",\n\t\t\tstateBefore:         domainResponse,\n\t\t\tstateAfter:          emptyResponse,\n\t\t\texpectNilBefore:     false,\n\t\t\texpectNilAfter:      true,\n\t\t\texpectDomainIDAfter: false,\n\t\t},\n\t\t{\n\t\t\tname:                \"when both StateBefore and StateAfter are populated, they should be deserialized as is\",\n\t\t\tstateBefore:         domainResponse,\n\t\t\tstateAfter:          domainResponse,\n\t\t\texpectNilBefore:     false,\n\t\t\texpectNilAfter:      false,\n\t\t\texpectDomainIDAfter: true,\n\t\t},\n\t\t{\n\t\t\tname:                \"when both StateBefore and StateAfter are empty, they should be deserialized as empty GetDomainResponse{}\",\n\t\t\tstateBefore:         emptyResponse,\n\t\t\tstateAfter:          emptyResponse,\n\t\t\texpectNilBefore:     true,\n\t\t\texpectNilAfter:      true,\n\t\t\texpectDomainIDAfter: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tm, mockStore := setUpMocksForDomainAuditManager(t)\n\t\t\tctx := context.Background()\n\n\t\t\t// Serialize the test data\n\t\t\tvar stateBeforeBlob, stateAfterBlob *DataBlob\n\t\t\tvar err error\n\n\t\t\tstateBeforeBlob, err = serializeGetDomainResponse(tt.stateBefore, constants.EncodingTypeThriftRWSnappy)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tstateAfterBlob, err = serializeGetDomainResponse(tt.stateAfter, constants.EncodingTypeThriftRWSnappy)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tmockStore.EXPECT().GetDomainAuditLogs(ctx, gomock.Any()).\n\t\t\t\tReturn(&InternalGetDomainAuditLogsResponse{\n\t\t\t\t\tAuditLogs: []*InternalDomainAuditLog{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       \"event-1\",\n\t\t\t\t\t\t\tDomainID:      \"domain-123\",\n\t\t\t\t\t\t\tStateBefore:   stateBeforeBlob,\n\t\t\t\t\t\t\tStateAfter:    stateAfterBlob,\n\t\t\t\t\t\t\tOperationType: DomainAuditOperationTypeCreate,\n\t\t\t\t\t\t\tCreatedTime:   time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\tresp, err := m.GetDomainAuditLogs(ctx, &GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"domain-123\",\n\t\t\t})\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, resp)\n\t\t\tassert.Len(t, resp.AuditLogs, 1)\n\n\t\t\t// Verify empty serialized responses are converted to nil\n\t\t\tif tt.expectNilBefore {\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateBefore, \"Empty serialized StateBefore should return nil\")\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateBefore, \"Populated StateBefore should be deserialized\")\n\t\t\t}\n\n\t\t\tif tt.expectNilAfter {\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateAfter, \"Empty serialized StateAfter should return nil\")\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateAfter, \"Populated StateAfter should be deserialized\")\n\t\t\t\tif tt.expectDomainIDAfter {\n\t\t\t\t\tassert.Equal(t, \"domain-123\", resp.AuditLogs[0].StateAfter.Info.ID)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomainAuditLogs_BackwardCompatibilityWithOldCassandraData(t *testing.T) {\n\t// Verifies that old Cassandra data with nil blobs returns nil without a deserialization error.\n\n\tm, mockStore := setUpMocksForDomainAuditManager(t)\n\tctx := context.Background()\n\n\t// Serialize a real StateAfter to simulate existing Cassandra data\n\tstateAfterResponse := &GetDomainResponse{\n\t\tInfo: &DomainInfo{\n\t\t\tID:   \"domain-123\",\n\t\t\tName: \"test-domain\",\n\t\t},\n\t}\n\tstateAfterBlob, err := serializeGetDomainResponse(stateAfterResponse, constants.EncodingTypeThriftRWSnappy)\n\tassert.NoError(t, err)\n\n\t// OLD Cassandra data: StateBefore is nil (NoSQL store returns nil when len(row.StateBefore) == 0)\n\tmockStore.EXPECT().GetDomainAuditLogs(ctx, gomock.Any()).\n\t\tReturn(&InternalGetDomainAuditLogsResponse{\n\t\t\tAuditLogs: []*InternalDomainAuditLog{\n\t\t\t\t{\n\t\t\t\t\tEventID:       \"old-event-1\",\n\t\t\t\t\tDomainID:      \"domain-123\",\n\t\t\t\t\tStateBefore:   nil, // Old data - NULL in Cassandra, nil from NoSQL store\n\t\t\t\t\tStateAfter:    stateAfterBlob,\n\t\t\t\t\tOperationType: DomainAuditOperationTypeCreate,\n\t\t\t\t\tCreatedTime:   time.Now(),\n\t\t\t\t},\n\t\t\t},\n\t\t}, nil).Times(1)\n\n\tresp, err := m.GetDomainAuditLogs(ctx, &GetDomainAuditLogsRequest{\n\t\tDomainID: \"domain-123\",\n\t})\n\n\tassert.NoError(t, err)\n\tassert.NotNil(t, resp)\n\tassert.Len(t, resp.AuditLogs, 1)\n\n\t// Old data with nil should still return nil to API (backward compatible)\n\tassert.Nil(t, resp.AuditLogs[0].StateBefore, \"Old Cassandra data with nil should return nil\")\n\tassert.NotNil(t, resp.AuditLogs[0].StateAfter, \"StateAfter should be deserialized\")\n\tassert.Equal(t, \"domain-123\", resp.AuditLogs[0].StateAfter.Info.ID)\n}\n"
  },
  {
    "path": "common/persistence/domain_manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\n\t// domainManagerImpl implements DomainManager based on DomainStore and PayloadSerializer\n\tdomainManagerImpl struct {\n\t\tserializer  PayloadSerializer\n\t\tpersistence DomainStore\n\t\tlogger      log.Logger\n\t\ttimeSrc     clock.TimeSource\n\t\tdc          *DynamicConfiguration\n\t}\n)\n\n// NewDomainManagerImpl returns new DomainManager\nfunc NewDomainManagerImpl(persistence DomainStore, logger log.Logger, serializer PayloadSerializer, dc *DynamicConfiguration) DomainManager {\n\treturn &domainManagerImpl{\n\t\tserializer:  serializer,\n\t\tpersistence: persistence,\n\t\tlogger:      logger,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t\tdc:          dc,\n\t}\n}\n\nfunc (m *domainManagerImpl) GetName() string {\n\treturn m.persistence.GetName()\n}\n\nfunc (m *domainManagerImpl) CreateDomain(\n\tctx context.Context,\n\trequest *CreateDomainRequest,\n) (*CreateDomainResponse, error) {\n\tencodingType := constants.EncodingType(m.dc.SerializationEncoding())\n\tdc, err := m.toInternalDomainConfig(request.Config, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trc, err := m.toInternalDomainReplicationConfig(request.ReplicationConfig, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn m.persistence.CreateDomain(ctx, &InternalCreateDomainRequest{\n\t\tInfo:              request.Info,\n\t\tConfig:            &dc,\n\t\tReplicationConfig: &rc,\n\t\tIsGlobalDomain:    request.IsGlobalDomain,\n\t\tConfigVersion:     request.ConfigVersion,\n\t\tFailoverVersion:   request.FailoverVersion,\n\t\tLastUpdatedTime:   time.Unix(0, request.LastUpdatedTime),\n\t\tCurrentTimeStamp:  m.timeSrc.Now(),\n\t})\n}\n\nfunc (m *domainManagerImpl) GetDomain(\n\tctx context.Context,\n\trequest *GetDomainRequest,\n) (*GetDomainResponse, error) {\n\tinternalResp, err := m.persistence.GetDomain(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdc, err := m.fromInternalDomainConfig(internalResp.Config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trc, err := m.fromInternalDomainReplicationConfig(internalResp.ReplicationConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp := &GetDomainResponse{\n\t\tInfo:                        internalResp.Info,\n\t\tConfig:                      &dc,\n\t\tReplicationConfig:           &rc,\n\t\tIsGlobalDomain:              internalResp.IsGlobalDomain,\n\t\tConfigVersion:               internalResp.ConfigVersion,\n\t\tFailoverVersion:             internalResp.FailoverVersion,\n\t\tFailoverNotificationVersion: internalResp.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     internalResp.PreviousFailoverVersion,\n\t\tLastUpdatedTime:             internalResp.LastUpdatedTime.UnixNano(),\n\t\tNotificationVersion:         internalResp.NotificationVersion,\n\t}\n\tif internalResp.FailoverEndTime != nil {\n\t\tresp.FailoverEndTime = common.Int64Ptr(internalResp.FailoverEndTime.UnixNano())\n\t}\n\treturn resp, nil\n}\n\nfunc (m *domainManagerImpl) UpdateDomain(\n\tctx context.Context,\n\trequest *UpdateDomainRequest,\n) error {\n\tencodingType := constants.EncodingType(m.dc.SerializationEncoding())\n\tdc, err := m.toInternalDomainConfig(request.Config, encodingType)\n\tif err != nil {\n\t\treturn err\n\t}\n\trc, err := m.toInternalDomainReplicationConfig(request.ReplicationConfig, encodingType)\n\tif err != nil {\n\t\treturn err\n\t}\n\tinternalReq := &InternalUpdateDomainRequest{\n\t\tInfo:                        request.Info,\n\t\tConfig:                      &dc,\n\t\tReplicationConfig:           &rc,\n\t\tConfigVersion:               request.ConfigVersion,\n\t\tFailoverVersion:             request.FailoverVersion,\n\t\tFailoverNotificationVersion: request.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     request.PreviousFailoverVersion,\n\t\tLastUpdatedTime:             time.Unix(0, request.LastUpdatedTime),\n\t\tNotificationVersion:         request.NotificationVersion,\n\t}\n\tif request.FailoverEndTime != nil {\n\t\tinternalReq.FailoverEndTime = common.TimePtr(time.Unix(0, *request.FailoverEndTime))\n\t}\n\treturn m.persistence.UpdateDomain(ctx, internalReq)\n}\n\nfunc (m *domainManagerImpl) DeleteDomain(\n\tctx context.Context,\n\trequest *DeleteDomainRequest,\n) error {\n\treturn m.persistence.DeleteDomain(ctx, request)\n}\n\nfunc (m *domainManagerImpl) DeleteDomainByName(\n\tctx context.Context,\n\trequest *DeleteDomainByNameRequest,\n) error {\n\treturn m.persistence.DeleteDomainByName(ctx, request)\n}\n\nfunc (m *domainManagerImpl) ListDomains(\n\tctx context.Context,\n\trequest *ListDomainsRequest,\n) (*ListDomainsResponse, error) {\n\tresp, err := m.persistence.ListDomains(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomains := make([]*GetDomainResponse, 0, len(resp.Domains))\n\tfor _, d := range resp.Domains {\n\t\tdc, err := m.fromInternalDomainConfig(d.Config)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trc, err := m.fromInternalDomainReplicationConfig(d.ReplicationConfig)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tcurrResp := &GetDomainResponse{\n\t\t\tInfo:                        d.Info,\n\t\t\tConfig:                      &dc,\n\t\t\tReplicationConfig:           &rc,\n\t\t\tIsGlobalDomain:              d.IsGlobalDomain,\n\t\t\tConfigVersion:               d.ConfigVersion,\n\t\t\tFailoverVersion:             d.FailoverVersion,\n\t\t\tFailoverNotificationVersion: d.FailoverNotificationVersion,\n\t\t\tPreviousFailoverVersion:     d.PreviousFailoverVersion,\n\t\t\tNotificationVersion:         d.NotificationVersion,\n\t\t\tLastUpdatedTime:             d.LastUpdatedTime.UnixNano(),\n\t\t}\n\t\tif d.FailoverEndTime != nil {\n\t\t\tcurrResp.FailoverEndTime = common.Int64Ptr(d.FailoverEndTime.UnixNano())\n\t\t}\n\t\tdomains = append(domains, currResp)\n\t}\n\treturn &ListDomainsResponse{\n\t\tDomains:       domains,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (m *domainManagerImpl) toInternalDomainConfig(c *DomainConfig, encodingType constants.EncodingType) (InternalDomainConfig, error) {\n\tif c == nil {\n\t\treturn InternalDomainConfig{}, nil\n\t}\n\tif c.BadBinaries.Binaries == nil {\n\t\tc.BadBinaries.Binaries = map[string]*types.BadBinaryInfo{}\n\t}\n\tbadBinaries, err := m.serializer.SerializeBadBinaries(&c.BadBinaries, encodingType)\n\tif err != nil {\n\t\treturn InternalDomainConfig{}, err\n\t}\n\tisolationGroups, err := m.serializer.SerializeIsolationGroups(&c.IsolationGroups, encodingType)\n\tif err != nil {\n\t\treturn InternalDomainConfig{}, err\n\t}\n\tasyncWFCfg, err := m.serializer.SerializeAsyncWorkflowsConfig(&c.AsyncWorkflowConfig, encodingType)\n\tif err != nil {\n\t\treturn InternalDomainConfig{}, err\n\t}\n\treturn InternalDomainConfig{\n\t\tRetention:                common.DaysToDuration(c.Retention),\n\t\tEmitMetric:               c.EmitMetric,\n\t\tHistoryArchivalStatus:    c.HistoryArchivalStatus,\n\t\tHistoryArchivalURI:       c.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus: c.VisibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    c.VisibilityArchivalURI,\n\t\tBadBinaries:              badBinaries,\n\t\tIsolationGroups:          isolationGroups,\n\t\tAsyncWorkflowsConfig:     asyncWFCfg,\n\t}, nil\n}\n\nfunc (m *domainManagerImpl) toInternalDomainReplicationConfig(rc *DomainReplicationConfig, encodingType constants.EncodingType) (InternalDomainReplicationConfig, error) {\n\tif rc == nil {\n\t\treturn InternalDomainReplicationConfig{}, nil\n\t}\n\n\t// active clusters don't use the default encoding due to their likely being quite large\n\t// and so we're explicitly opting to use snappy / compressed encoding\n\tactiveClustersConfig, err := m.serializer.SerializeActiveClusters(rc.ActiveClusters, constants.EncodingTypeThriftRWSnappy)\n\tif err != nil {\n\t\treturn InternalDomainReplicationConfig{}, err\n\t}\n\treturn InternalDomainReplicationConfig{\n\t\tClusters:             rc.Clusters,\n\t\tActiveClusterName:    rc.ActiveClusterName,\n\t\tActiveClustersConfig: activeClustersConfig,\n\t}, nil\n}\n\nfunc (m *domainManagerImpl) fromInternalDomainConfig(ic *InternalDomainConfig) (DomainConfig, error) {\n\tif ic == nil {\n\t\treturn DomainConfig{}, nil\n\t}\n\tbadBinaries, err := m.serializer.DeserializeBadBinaries(ic.BadBinaries)\n\tif err != nil {\n\t\treturn DomainConfig{}, err\n\t}\n\tvar isolationGroups types.IsolationGroupConfiguration\n\tigDeserialized, err := m.serializer.DeserializeIsolationGroups(ic.IsolationGroups)\n\tif err != nil {\n\t\treturn DomainConfig{}, err\n\t}\n\tif igDeserialized != nil {\n\t\tisolationGroups = *igDeserialized\n\t}\n\tvar asyncWFCfg types.AsyncWorkflowConfiguration\n\tasyncWFCfgDeserialied, err := m.serializer.DeserializeAsyncWorkflowsConfig(ic.AsyncWorkflowsConfig)\n\tif err != nil {\n\t\treturn DomainConfig{}, err\n\t}\n\tif asyncWFCfgDeserialied != nil {\n\t\tasyncWFCfg = *asyncWFCfgDeserialied\n\t}\n\n\tif badBinaries.Binaries == nil {\n\t\tbadBinaries.Binaries = map[string]*types.BadBinaryInfo{}\n\t}\n\treturn DomainConfig{\n\t\tRetention:                common.DurationToDays(ic.Retention),\n\t\tEmitMetric:               ic.EmitMetric,\n\t\tHistoryArchivalStatus:    ic.HistoryArchivalStatus,\n\t\tHistoryArchivalURI:       ic.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus: ic.VisibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    ic.VisibilityArchivalURI,\n\t\tBadBinaries:              *badBinaries,\n\t\tIsolationGroups:          isolationGroups,\n\t\tAsyncWorkflowConfig:      asyncWFCfg,\n\t}, nil\n}\n\nfunc (m *domainManagerImpl) fromInternalDomainReplicationConfig(ic *InternalDomainReplicationConfig) (DomainReplicationConfig, error) {\n\tif ic == nil {\n\t\treturn DomainReplicationConfig{}, nil\n\t}\n\n\tactiveClusters, err := m.serializer.DeserializeActiveClusters(ic.ActiveClustersConfig)\n\tif err != nil {\n\t\treturn DomainReplicationConfig{}, err\n\t}\n\treturn DomainReplicationConfig{\n\t\tClusters:          ic.Clusters,\n\t\tActiveClusterName: ic.ActiveClusterName,\n\t\tActiveClusters:    activeClusters,\n\t}, nil\n}\n\nfunc (m *domainManagerImpl) GetMetadata(\n\tctx context.Context,\n) (*GetMetadataResponse, error) {\n\treturn m.persistence.GetMetadata(ctx)\n}\n\nfunc (m *domainManagerImpl) Close() {\n\tm.persistence.Close()\n}\n"
  },
  {
    "path": "common/persistence/domain_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc testFixtureDomainInfo() *DomainInfo {\n\treturn &DomainInfo{\n\t\tID:          \"domain-id\",\n\t\tName:        \"domain-name\",\n\t\tStatus:      1,\n\t\tDescription: \"domain-desc\",\n\t\tOwnerEmail:  \"owner@test.com\",\n\t\tData:        map[string]string{\"test\": \"a\"},\n\t}\n}\n\nfunc testFixtureDomainConfig() *DomainConfig {\n\treturn &DomainConfig{\n\t\tRetention:                1,\n\t\tEmitMetric:               true,\n\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\tBadBinaries:              types.BadBinaries{},\n\t\tIsolationGroups:          map[string]types.IsolationGroupPartition{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}},\n\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\tEnabled:             true,\n\t\t\tPredefinedQueueName: \"q\",\n\t\t\tQueueType:           \"kafka\",\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfig() *DomainReplicationConfig {\n\treturn &DomainReplicationConfig{\n\t\tActiveClusterName: \"cluster-1\",\n\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfigWithActiveClusters() *DomainReplicationConfig {\n\treturn &DomainReplicationConfig{\n\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t\tActiveClusters: &types.ActiveClusters{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"region-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"region-2\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster-2\",\n\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfigInternal() *InternalDomainReplicationConfig {\n\treturn &InternalDomainReplicationConfig{\n\t\tActiveClusterName: \"cluster-1\",\n\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfigInternalWithActiveClusters() *InternalDomainReplicationConfig {\n\treturn &InternalDomainReplicationConfig{\n\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t\tActiveClustersConfig: &DataBlob{Encoding: constants.EncodingTypeThriftRWSnappy, Data: []byte(\"active-clusters-config\")},\n\t}\n}\n\nfunc setUpMocksForDomainManager(t *testing.T) (*domainManagerImpl, *MockDomainStore, *MockPayloadSerializer) {\n\tctrl := gomock.NewController(t)\n\tmockStore := NewMockDomainStore(ctrl)\n\tmockSerializer := NewMockPayloadSerializer(ctrl)\n\tlogger := log.NewNoop()\n\n\tdomainManager := NewDomainManagerImpl(mockStore, logger, mockSerializer, &DynamicConfiguration{SerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW))}).(*domainManagerImpl)\n\n\treturn domainManager, mockStore, mockSerializer\n}\n\nfunc TestCreateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *CreateDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(&types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeIsolationGroups(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeAsyncWorkflowsConfig(&types.AsyncWorkflowConfiguration{Enabled: true, PredefinedQueueName: \"q\", QueueType: \"kafka\"}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeActiveClusters(nil, constants.EncodingTypeThriftRWSnappy).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\n\t\t\t\texpectedReq := &InternalCreateDomainRequest{\n\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternal(),\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\tCurrentTimeStamp:  time.Now(),\n\t\t\t\t}\n\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, actualRequest *InternalCreateDomainRequest) (*CreateDomainResponse, error) {\n\t\t\t\t\t\tassert.WithinDuration(t, expectedReq.CurrentTimeStamp, actualRequest.CurrentTimeStamp, time.Second)\n\t\t\t\t\t\treturn nil, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\trequest: &CreateDomainRequest{\n\t\t\t\tInfo:              testFixtureDomainInfo(),\n\t\t\t\tConfig:            testFixtureDomainConfig(),\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   10,\n\t\t\t\tLastUpdatedTime:   100,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success with activeclusters config populated\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(&types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeIsolationGroups(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeAsyncWorkflowsConfig(&types.AsyncWorkflowConfiguration{Enabled: true, PredefinedQueueName: \"q\", QueueType: \"kafka\"}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeActiveClusters(testFixtureDomainReplicationConfigWithActiveClusters().ActiveClusters, constants.EncodingTypeThriftRWSnappy).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRWSnappy, Data: []byte(\"active-clusters-config\")}, nil).Times(1)\n\n\t\t\t\texpectedReq := &InternalCreateDomainRequest{\n\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternalWithActiveClusters(),\n\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\tCurrentTimeStamp:  time.Now(),\n\t\t\t\t}\n\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tCreateDomain(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, actualRequest *InternalCreateDomainRequest) (*CreateDomainResponse, error) {\n\t\t\t\t\t\tassert.WithinDuration(t, expectedReq.CurrentTimeStamp, actualRequest.CurrentTimeStamp, time.Second)\n\t\t\t\t\t\treturn nil, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\trequest: &CreateDomainRequest{\n\t\t\t\tInfo:              testFixtureDomainInfo(),\n\t\t\t\tConfig:            testFixtureDomainConfig(),\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigWithActiveClusters(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   10,\n\t\t\t\tLastUpdatedTime:   100,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"serialization error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"serialization error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &CreateDomainRequest{\n\t\t\t\tInfo:              &DomainInfo{ID: \"domain1\"},\n\t\t\t\tConfig:            &DomainConfig{},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{},\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   1,\n\t\t\t\tLastUpdatedTime:   time.Now().UnixNano(),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialization error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\t_, err := domainManager.CreateDomain(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *GetDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *GetDomainResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &GetDomainRequest{ID: \"domain1\"}).\n\t\t\t\t\tReturn(&InternalGetDomainResponse{\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternal(),\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBadBinaries(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}).\n\t\t\t\t\tReturn(&types.BadBinaries{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeIsolationGroups(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}).\n\t\t\t\t\tReturn(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeAsyncWorkflowsConfig(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}).\n\t\t\t\t\tReturn(&types.AsyncWorkflowConfiguration{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeActiveClusters(nil).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetDomainRequest{ID: \"domain1\"},\n\t\t\texpectError: false,\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                1,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\tIsolationGroups:          map[string]types.IsolationGroupPartition{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   10,\n\t\t\t\tLastUpdatedTime:   100,\n\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with activeclusters config populated\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), &GetDomainRequest{ID: \"domain1\"}).\n\t\t\t\t\tReturn(&InternalGetDomainResponse{\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternalWithActiveClusters(),\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBadBinaries(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}).\n\t\t\t\t\tReturn(&types.BadBinaries{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeIsolationGroups(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}).\n\t\t\t\t\tReturn(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeAsyncWorkflowsConfig(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}).\n\t\t\t\t\tReturn(&types.AsyncWorkflowConfiguration{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeActiveClusters(testFixtureDomainReplicationConfigInternalWithActiveClusters().ActiveClustersConfig).\n\t\t\t\t\tReturn(testFixtureDomainReplicationConfigWithActiveClusters().ActiveClusters, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetDomainRequest{ID: \"domain1\"},\n\t\t\texpectError: false,\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                1,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\tIsolationGroups:          map[string]types.IsolationGroupPartition{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigWithActiveClusters(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   10,\n\t\t\t\tLastUpdatedTime:   100,\n\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetDomainRequest{ID: \"domain1\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t\t{\n\t\t\tname: \"deserialization error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalGetDomainResponse{\n\t\t\t\t\t\tInfo:              &DomainInfo{ID: \"domain1\"},\n\t\t\t\t\t\tConfig:            &InternalDomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &InternalDomainReplicationConfig{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBadBinaries(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"deserialization error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetDomainRequest{ID: \"domain1\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"deserialization error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\tresp, err := domainManager.GetDomain(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *UpdateDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(&types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeIsolationGroups(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeAsyncWorkflowsConfig(&types.AsyncWorkflowConfiguration{Enabled: true, PredefinedQueueName: \"q\", QueueType: \"kafka\"}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeActiveClusters(nil, constants.EncodingTypeThriftRWSnappy).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tUpdateDomain(gomock.Any(), &InternalUpdateDomainRequest{\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfigInternal(),\n\t\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\t\tFailoverVersion:             10,\n\t\t\t\t\t\tFailoverNotificationVersion: 11,\n\t\t\t\t\t\tPreviousFailoverVersion:     9,\n\t\t\t\t\t\tNotificationVersion:         1000,\n\t\t\t\t\t\tLastUpdatedTime:             time.Unix(0, 100),\n\t\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &UpdateDomainRequest{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             10,\n\t\t\t\tFailoverNotificationVersion: 11,\n\t\t\t\tPreviousFailoverVersion:     9,\n\t\t\t\tLastUpdatedTime:             100,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t\tNotificationVersion:         1000,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success with activeclusters config populated\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(&types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeIsolationGroups(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeAsyncWorkflowsConfig(&types.AsyncWorkflowConfiguration{Enabled: true, PredefinedQueueName: \"q\", QueueType: \"kafka\"}, constants.EncodingTypeThriftRW).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeActiveClusters(testFixtureDomainReplicationConfigWithActiveClusters().ActiveClusters, constants.EncodingTypeThriftRWSnappy).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRWSnappy, Data: []byte(\"active-clusters-config\")}, nil).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tUpdateDomain(gomock.Any(), &InternalUpdateDomainRequest{\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfigInternalWithActiveClusters(),\n\t\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\t\tFailoverVersion:             10,\n\t\t\t\t\t\tFailoverNotificationVersion: 11,\n\t\t\t\t\t\tPreviousFailoverVersion:     9,\n\t\t\t\t\t\tNotificationVersion:         1000,\n\t\t\t\t\t\tLastUpdatedTime:             time.Unix(0, 100),\n\t\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &UpdateDomainRequest{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfigWithActiveClusters(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             10,\n\t\t\t\tFailoverNotificationVersion: 11,\n\t\t\t\tPreviousFailoverVersion:     9,\n\t\t\t\tLastUpdatedTime:             100,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t\tNotificationVersion:         1000,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"serialization error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBadBinaries(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"serialization error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &UpdateDomainRequest{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             10,\n\t\t\t\tFailoverNotificationVersion: 11,\n\t\t\t\tPreviousFailoverVersion:     9,\n\t\t\t\tLastUpdatedTime:             100,\n\t\t\t\tNotificationVersion:         1000,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialization error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\terr := domainManager.UpdateDomain(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *DeleteDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, _ *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &DeleteDomainRequest{ID: \"domain1\"},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, _ *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &DeleteDomainRequest{ID: \"domain1\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\terr := domainManager.DeleteDomain(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteDomainByName(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *DeleteDomainByNameRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, _ *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteDomainByName(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &DeleteDomainByNameRequest{Name: \"domain1\"},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, _ *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteDomainByName(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &DeleteDomainByNameRequest{Name: \"domain1\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\terr := domainManager.DeleteDomainByName(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListDomains(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore, *MockPayloadSerializer)\n\t\trequest       *ListDomainsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *ListDomainsResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tListDomains(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalListDomainsResponse{\n\t\t\t\t\t\tDomains: []*InternalGetDomainResponse{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternal(),\n\t\t\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBadBinaries(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}).\n\t\t\t\t\tReturn(&types.BadBinaries{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeIsolationGroups(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}).\n\t\t\t\t\tReturn(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeAsyncWorkflowsConfig(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}).\n\t\t\t\t\tReturn(&types.AsyncWorkflowConfiguration{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeActiveClusters(nil).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &ListDomainsRequest{PageSize: 10},\n\t\t\texpectError: false,\n\t\t\texpected: &ListDomainsResponse{\n\t\t\t\tDomains: []*GetDomainResponse{\n\t\t\t\t\t{\n\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\t\tIsolationGroups:          map[string]types.IsolationGroupPartition{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}},\n\t\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\tLastUpdatedTime:   100,\n\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with activeclusters config populated\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tListDomains(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalListDomainsResponse{\n\t\t\t\t\t\tDomains: []*InternalGetDomainResponse{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\t\t\tConfig: &InternalDomainConfig{\n\t\t\t\t\t\t\t\t\tRetention:                common.DaysToDuration(1),\n\t\t\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\t\t\tBadBinaries:              &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")},\n\t\t\t\t\t\t\t\t\tIsolationGroups:          &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")},\n\t\t\t\t\t\t\t\t\tAsyncWorkflowsConfig:     &DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigInternalWithActiveClusters(),\n\t\t\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\t\t\tLastUpdatedTime:   time.Unix(0, 100),\n\t\t\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200)),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBadBinaries(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"bad-binaries\")}).\n\t\t\t\t\tReturn(&types.BadBinaries{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeIsolationGroups(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"isolation-groups\")}).\n\t\t\t\t\tReturn(&types.IsolationGroupConfiguration{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeAsyncWorkflowsConfig(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"async-workflow-config\")}).\n\t\t\t\t\tReturn(&types.AsyncWorkflowConfiguration{}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeActiveClusters(testFixtureDomainReplicationConfigInternalWithActiveClusters().ActiveClustersConfig).\n\t\t\t\t\tReturn(testFixtureDomainReplicationConfigWithActiveClusters().ActiveClusters, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &ListDomainsRequest{PageSize: 10},\n\t\t\texpectError: false,\n\t\t\texpected: &ListDomainsResponse{\n\t\t\t\tDomains: []*GetDomainResponse{\n\t\t\t\t\t{\n\n\t\t\t\t\t\tInfo: testFixtureDomainInfo(),\n\t\t\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\t\t\tRetention:                1,\n\t\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tHistoryArchivalURI:       \"s3://abc\",\n\t\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\t\tVisibilityArchivalURI:    \"s3://xyz\",\n\t\t\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t\t\t\t\t\tIsolationGroups:          map[string]types.IsolationGroupPartition{\"abc\": {Name: \"abc\", State: types.IsolationGroupStateDrained}},\n\t\t\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigWithActiveClusters(),\n\t\t\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\t\t\tConfigVersion:     1,\n\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\tLastUpdatedTime:   100,\n\t\t\t\t\t\tFailoverEndTime:   common.Ptr(time.Unix(0, 200).UnixNano()),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore, _ *MockPayloadSerializer) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tListDomains(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &ListDomainsRequest{PageSize: 10},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, mockSerializer := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore, mockSerializer)\n\n\t\t\tresp, err := domainManager.ListDomains(context.Background(), tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMetadata(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockDomainStore)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(&GetMetadataResponse{NotificationVersion: 10}, nil).Times(1)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockDomainStore) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetMetadata(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdomainManager, mockStore, _ := setUpMocksForDomainManager(t)\n\n\t\t\ttc.setupMock(mockStore)\n\n\t\t\t// Execute the GetMetadata method\n\t\t\tresp, err := domainManager.GetMetadata(context.Background())\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int64(10), resp.NotificationVersion)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDomainManagerClose(t *testing.T) {\n\tt.Run(\"close persistence store\", func(t *testing.T) {\n\t\tdomainManager, mockStore, _ := setUpMocksForDomainManager(t)\n\n\t\t// Expect the Close method to be called once\n\t\tmockStore.EXPECT().Close().Times(1)\n\n\t\t// Execute the Close method\n\t\tdomainManager.Close()\n\n\t\t// No need to assert as we are just expecting the method call\n\t})\n}\n"
  },
  {
    "path": "common/persistence/domain_replication_config_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// TestIsActiveActiveMethodsAreInSync ensures that the IsActiveActive() implementations\n// in both persistence.DomainReplicationConfig and types.DomainReplicationConfiguration\n// behave identically\nfunc TestIsActiveActiveMethodsAreInSync(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\ttypesConfig       *types.DomainReplicationConfiguration\n\t\tpersistenceConfig *DomainReplicationConfig\n\t\twant              bool\n\t}{\n\t\t{\n\t\t\tname:              \"both nil should return false\",\n\t\t\ttypesConfig:       nil,\n\t\t\tpersistenceConfig: nil,\n\t\t\twant:              false,\n\t\t},\n\t\t{\n\t\t\tname:              \"both with nil ActiveClusters should return false\",\n\t\t\ttypesConfig:       &types.DomainReplicationConfiguration{ActiveClusters: nil},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{ActiveClusters: nil},\n\t\t\twant:              false,\n\t\t},\n\t\t{\n\t\t\tname:              \"both with empty ActiveClusters should return false\",\n\t\t\ttypesConfig:       &types.DomainReplicationConfiguration{ActiveClusters: &types.ActiveClusters{}},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{ActiveClusters: &types.ActiveClusters{}},\n\t\t\twant:              false,\n\t\t},\n\t\t{\n\t\t\tname: \"both with only AttributeScopes populated should return true\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"both with only AttributeScopes with non-empty ClusterAttributes should return true\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"both with empty AttributeScopes map should return false\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"both with AttributeScopes containing scope with empty ClusterAttributes should return false\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"both with multiple AttributeScopes, some empty should return true if any has ClusterAttributes\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"both with multiple AttributeScopes all empty should return false\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"both with multiple regions in AttributeScopes should return true\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"eu-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"eu-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"both with multiple AttributeScopes, each with multiple ClusterAttributes should return true\",\n\t\t\ttypesConfig: &types.DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster4\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   400,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpersistenceConfig: &DomainReplicationConfig{\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"datacenter\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster3\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   300,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"dc2\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster4\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   400,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttypesResult := tt.typesConfig.IsActiveActive()\n\t\t\tpersistenceResult := tt.persistenceConfig.IsActiveActive()\n\n\t\t\tif tt.persistenceConfig != nil && tt.typesConfig != nil {\n\t\t\t\tassert.Equal(t, tt.persistenceConfig.ActiveClusters, tt.typesConfig.ActiveClusters)\n\t\t\t}\n\n\t\t\t// Assert that both implementations return the same value\n\t\t\tassert.Equal(t, persistenceResult, typesResult,\n\t\t\t\t\"IsActiveActive() implementations differ: types=%v, persistence=%v\",\n\t\t\t\ttypesResult, persistenceResult)\n\n\t\t\t// Also assert that they match the expected value\n\t\t\tassert.Equal(t, tt.want, typesResult,\n\t\t\t\t\"types.IsActiveActive() = %v, want %v\", typesResult, tt.want)\n\t\t\tassert.Equal(t, tt.want, persistenceResult,\n\t\t\t\t\"persistence.IsActiveActive() = %v, want %v\", persistenceResult, tt.want)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/elasticsearch/decodeBench_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nvar (\n\tdata = []byte(`{\"CloseStatus\": 0,\n         \"CloseTime\": 1547596872817380000,\n         \"DomainID\": \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\",\n         \"HistoryLength\": 29,\n         \"KafkaKey\": \"7-619\",\n         \"RunID\": \"e481009e-14b3-45ae-91af-dce6e2a88365\",\n         \"StartTime\": 1547596872371000000,\n         \"WorkflowID\": \"6bfbc1e5-6ce4-4e22-bbfb-e0faa9a7a604-1-2256\",\n         \"WorkflowType\": \"TestWorkflowExecute\",\n         \"Encoding\" : \"thriftrw\",\n         \"TaskList\" : \"taskList\",\n         \"IsCron\" : \"false\",\n         \"NumClusters\" : \"2\",\n         \"Memo\" : \"WQ0ACgsLAAAAAwAAAAJrMgAAAAkidmFuY2V4dSIAAAACazMAAAADMTIzAAAAAmsxAAAAUXsia2V5MSI6MTIzNDMyMSwia2V5MiI6ImEgc3RyaW5nIGlzIHZlcnkgbG9uZyIsIm1hcCI6eyJtS2V5IjoxMjM0MywiYXNkIjoiYXNkZiJ9fQA=\"}`)\n)\n\n/*\nBenchmarkJSONDecodeToType-8       200000              9321 ns/op\nBenchmarkJSONDecodeToMap-8        100000             12878 ns/op\n*/\n\n// nolint\nfunc BenchmarkJSONDecodeToType(b *testing.B) {\n\tbytes := (*json.RawMessage)(&data)\n\tfor i := 0; i < b.N; i++ {\n\t\tvar source *es.VisibilityRecord\n\t\tjson.Unmarshal(*bytes, &source)\n\t\trecord := &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\tDomainID:      source.DomainID,\n\t\t\tWorkflowType:  source.WorkflowType,\n\t\t\tWorkflowID:    source.WorkflowID,\n\t\t\tRunID:         source.RunID,\n\t\t\tTypeName:      source.WorkflowType,\n\t\t\tStartTime:     time.Unix(0, source.StartTime),\n\t\t\tExecutionTime: time.Unix(0, source.ExecutionTime),\n\t\t\tMemo:          p.NewDataBlob(source.Memo, constants.EncodingType(source.Encoding)),\n\t\t\tTaskList:      source.TaskList,\n\t\t\tIsCron:        source.IsCron,\n\t\t\tNumClusters:   source.NumClusters,\n\t\t}\n\t\trecord.CloseTime = time.Unix(0, source.CloseTime)\n\t\trecord.Status = thrift.ToWorkflowExecutionCloseStatus(&source.CloseStatus)\n\t\trecord.HistoryLength = source.HistoryLength\n\t}\n}\n\n// nolint\nfunc BenchmarkJSONDecodeToMap(b *testing.B) {\n\tfor i := 0; i < b.N; i++ {\n\t\tvar source map[string]interface{}\n\t\td := json.NewDecoder(bytes.NewReader(data))\n\t\td.UseNumber()\n\t\td.Decode(&source)\n\n\t\tstartTime, _ := source[definition.StartTime].(json.Number).Int64()\n\t\texecutionTime, _ := source[definition.StartTime].(json.Number).Int64()\n\t\tcloseTime, _ := source[definition.CloseTime].(json.Number).Int64()\n\t\tcloseStatus, _ := source[definition.CloseStatus].(json.Number).Int64()\n\t\thistoryLen, _ := source[definition.HistoryLength].(json.Number).Int64()\n\n\t\trecord := &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\tDomainID:      source[definition.DomainID].(string),\n\t\t\tWorkflowType:  source[definition.WorkflowType].(string),\n\t\t\tWorkflowID:    source[definition.WorkflowID].(string),\n\t\t\tRunID:         source[definition.RunID].(string),\n\t\t\tTypeName:      source[definition.WorkflowType].(string),\n\t\t\tStartTime:     time.Unix(0, startTime),\n\t\t\tExecutionTime: time.Unix(0, executionTime),\n\t\t\tTaskList:      source[definition.TaskList].(string),\n\t\t\tIsCron:        source[definition.IsCron].(bool),\n\t\t\tNumClusters:   source[definition.NumClusters].(int16),\n\t\t\tMemo:          p.NewDataBlob([]byte(source[definition.Memo].(string)), constants.EncodingType(source[definition.Encoding].(string))),\n\t\t}\n\t\trecord.CloseTime = time.Unix(0, closeTime)\n\t\tstatus := (shared.WorkflowExecutionCloseStatus)(int32(closeStatus))\n\t\trecord.Status = thrift.ToWorkflowExecutionCloseStatus(&status)\n\t\trecord.HistoryLength = historyLen\n\t}\n}\n"
  },
  {
    "path": "common/persistence/elasticsearch/es_visibility_metric_clients.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype visibilityMetricsClient struct {\n\tmetricClient metrics.Client\n\tpersistence  p.VisibilityManager\n\tlogger       log.Logger\n}\n\nvar _ p.VisibilityManager = (*visibilityMetricsClient)(nil)\n\n// NewVisibilityMetricsClient wrap visibility client with metrics\nfunc NewVisibilityMetricsClient(persistence p.VisibilityManager, metricClient metrics.Client, logger log.Logger) p.VisibilityManager {\n\treturn &visibilityMetricsClient{\n\t\tpersistence:  persistence,\n\t\tmetricClient: metricClient,\n\t\tlogger:       logger,\n\t}\n}\n\nfunc (p *visibilityMetricsClient) GetName() string {\n\treturn p.persistence.GetName()\n}\n\nfunc (p *visibilityMetricsClient) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionStartedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchRecordWorkflowExecutionStartedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\n\terr := p.persistence.RecordWorkflowExecutionStarted(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchRecordWorkflowExecutionStartedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionClosedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchRecordWorkflowExecutionClosedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\terr := p.persistence.RecordWorkflowExecutionClosed(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchRecordWorkflowExecutionClosedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionUninitializedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchRecordWorkflowExecutionUninitializedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\terr := p.persistence.RecordWorkflowExecutionUninitialized(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchRecordWorkflowExecutionUninitializedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *p.UpsertWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchUpsertWorkflowExecutionScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\terr := p.persistence.UpsertWorkflowExecution(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchUpsertWorkflowExecutionScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListOpenWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListOpenWorkflowExecutions(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListOpenWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListClosedWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListClosedWorkflowExecutions(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListClosedWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByTypeRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListOpenWorkflowExecutionsByTypeScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListOpenWorkflowExecutionsByType(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListOpenWorkflowExecutionsByTypeScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByTypeRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListClosedWorkflowExecutionsByTypeScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByType(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListClosedWorkflowExecutionsByTypeScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListOpenWorkflowExecutionsByWorkflowIDScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListOpenWorkflowExecutionsByWorkflowIDScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListClosedWorkflowExecutionsByWorkflowIDScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListClosedWorkflowExecutionsByWorkflowIDScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *p.ListClosedWorkflowExecutionsByStatusRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListClosedWorkflowExecutionsByStatusScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListClosedWorkflowExecutionsByStatusScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.GetClosedWorkflowExecutionRequest,\n) (*p.GetClosedWorkflowExecutionResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchGetClosedWorkflowExecutionScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\n\tresponse, err := p.persistence.GetClosedWorkflowExecution(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchGetClosedWorkflowExecutionScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchListWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\n\tresponse, err := p.persistence.ListWorkflowExecutions(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchListWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchScanWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\tresponse, err := p.persistence.ScanWorkflowExecutions(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchScanWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.CountWorkflowExecutionsRequest,\n) (*p.CountWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchCountWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\n\tresponse, err := p.persistence.CountWorkflowExecutions(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchCountWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *visibilityMetricsClient) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchDeleteWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\terr := p.persistence.DeleteWorkflowExecution(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchDeleteWorkflowExecutionsScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.ElasticsearchDeleteWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.ElasticsearchLatencyPerDomain)\n\terr := p.persistence.DeleteWorkflowExecution(ctx, request)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.ElasticsearchDeleteWorkflowExecutionsScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *visibilityMetricsClient) updateErrorMetric(scopeWithDomainTag metrics.Scope, scope metrics.ScopeIdx, err error) {\n\n\tswitch err.(type) {\n\tcase *types.BadRequestError:\n\t\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchErrBadRequestCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchFailuresPerDomain)\n\n\tcase *types.ServiceBusyError:\n\t\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchErrBusyCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchFailuresPerDomain)\n\tdefault:\n\t\tp.logger.Error(\"Operation failed with internal error.\", tag.MetricScope(int(scope)), tag.Error(err))\n\t\tscopeWithDomainTag.IncCounter(metrics.ElasticsearchFailuresPerDomain)\n\t}\n}\n\nfunc (p *visibilityMetricsClient) Close() {\n\tp.persistence.Close()\n}\n"
  },
  {
    "path": "common/persistence/elasticsearch/es_visibility_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"regexp\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/cch123/elasticsql\"\n\t\"github.com/valyala/fastjson\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/elasticsearch/query\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\ntype (\n\tesVisibilityStore struct {\n\t\tesClient es.GenericClient\n\t\tindex    string\n\t\tproducer messaging.Producer\n\t\tlogger   log.Logger\n\t\tconfig   *service.Config\n\t}\n)\n\nvar _ p.VisibilityStore = (*esVisibilityStore)(nil)\n\n// NewElasticSearchVisibilityStore create a visibility store connecting to ElasticSearch\nfunc NewElasticSearchVisibilityStore(\n\tesClient es.GenericClient,\n\tindex string,\n\tproducer messaging.Producer,\n\tconfig *service.Config,\n\tlogger log.Logger,\n) p.VisibilityStore {\n\treturn &esVisibilityStore{\n\t\tesClient: esClient,\n\t\tindex:    index,\n\t\tproducer: producer,\n\t\tlogger:   logger.WithTags(tag.ComponentESVisibilityManager),\n\t\tconfig:   config,\n\t}\n}\n\nfunc (v *esVisibilityStore) Close() {}\n\nfunc (v *esVisibilityStore) GetName() string {\n\treturn constants.ESPersistenceName\n}\n\nfunc (v *esVisibilityStore) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionStartedRequest,\n) error {\n\tv.checkProducer()\n\tmsg := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixNano(),\n\t\trequest.ExecutionTimestamp.UnixNano(),\n\t\trequest.TaskID,\n\t\trequest.Memo.Data,\n\t\trequest.Memo.GetEncoding(),\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\trequest.ClusterAttributeScope,\n\t\trequest.ClusterAttributeName,\n\t\trequest.SearchAttributes,\n\t\tconstants.RecordStarted,\n\t\t0,                                  // will not be used\n\t\t0,                                  // will not be used\n\t\t0,                                  // will not be used\n\t\trequest.UpdateTimestamp.UnixNano(), // will be updated when workflow execution updates\n\t\tint64(request.ShardID),\n\t\trequest.CronSchedule,\n\t\tint32(request.ExecutionStatus),\n\t\trequest.ScheduledExecutionTime.UnixNano(),\n\t)\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *esVisibilityStore) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionClosedRequest,\n) error {\n\tv.checkProducer()\n\tmsg := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixNano(),\n\t\trequest.ExecutionTimestamp.UnixNano(),\n\t\trequest.TaskID,\n\t\trequest.Memo.Data,\n\t\trequest.Memo.GetEncoding(),\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\trequest.ClusterAttributeScope,\n\t\trequest.ClusterAttributeName,\n\t\trequest.SearchAttributes,\n\t\tconstants.RecordClosed,\n\t\trequest.CloseTimestamp.UnixNano(),\n\t\t*thrift.FromWorkflowExecutionCloseStatus(&request.Status),\n\t\trequest.HistoryLength,\n\t\trequest.UpdateTimestamp.UnixNano(),\n\t\tint64(request.ShardID),\n\t\trequest.CronSchedule,\n\t\tint32(request.ExecutionStatus),\n\t\trequest.ScheduledExecutionTime.UnixNano(),\n\t)\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *esVisibilityStore) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionUninitializedRequest,\n) error {\n\tv.checkProducer()\n\tmsg := getVisibilityMessageForUninitializedWorkflow(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.UpdateTimestamp.UnixNano(),\n\t\trequest.ShardID,\n\t)\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *esVisibilityStore) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalUpsertWorkflowExecutionRequest,\n) error {\n\tv.checkProducer()\n\tmsg := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixNano(),\n\t\trequest.ExecutionTimestamp.UnixNano(),\n\t\trequest.TaskID,\n\t\trequest.Memo.Data,\n\t\trequest.Memo.GetEncoding(),\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\trequest.ClusterAttributeScope,\n\t\trequest.ClusterAttributeName,\n\t\trequest.SearchAttributes,\n\t\tconstants.UpsertSearchAttributes,\n\t\t0, // will not be used\n\t\t0, // will not be used\n\t\t0, // will not be used\n\t\trequest.UpdateTimestamp.UnixNano(),\n\t\trequest.ShardID,\n\t\trequest.CronSchedule,\n\t\tint32(request.ExecutionStatus),\n\t\trequest.ScheduledExecutionTimestamp,\n\t)\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *esVisibilityStore) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     request,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListOpenWorkflowExecutions failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     request,\n\t\tIsOpen:          false,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListClosedWorkflowExecutions failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByTypeRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      query.NewMatchQuery(es.WorkflowType, request.WorkflowTypeName),\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListOpenWorkflowExecutionsByType failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByTypeRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t\tIsOpen:          false,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      query.NewMatchQuery(es.WorkflowType, request.WorkflowTypeName),\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListClosedWorkflowExecutionsByType failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      query.NewMatchQuery(es.WorkflowID, request.WorkflowID),\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListOpenWorkflowExecutionsByWorkflowID failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t\tIsOpen:          false,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      query.NewMatchQuery(es.WorkflowID, request.WorkflowID),\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListClosedWorkflowExecutionsByWorkflowID failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *p.InternalListClosedWorkflowExecutionsByStatusRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tresp, err := v.esClient.Search(ctx, &es.SearchRequest{\n\t\tIndex:           v.index,\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t\tIsOpen:          false,\n\t\tFilter:          isRecordValid,\n\t\tMatchQuery:      query.NewMatchQuery(es.CloseStatus, int32(*thrift.FromWorkflowExecutionCloseStatus(&request.Status))),\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListClosedWorkflowExecutionsByStatus failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalGetClosedWorkflowExecutionRequest,\n) (*p.InternalGetClosedWorkflowExecutionResponse, error) {\n\tresp, err := v.esClient.SearchForOneClosedExecution(ctx, v.index, request)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"SearchForOneClosedExecution failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\tv.checkProducer()\n\tmsg := getVisibilityMessageForDeletion(\n\t\trequest.DomainID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.TaskID,\n\t)\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *esVisibilityStore) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\t// verify if it is uninitialized workflow execution record\n\t// if it is, then call the existing delete method to delete\n\tquery := fmt.Sprintf(\"StartTime = missing and DomainID = %s and RunID = %s\", request.DomainID, request.RunID)\n\tqueryRequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomain: request.Domain,\n\t\tQuery:  query,\n\t}\n\tresp, err := v.CountWorkflowExecutions(ctx, queryRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.Count > 0 {\n\t\tif err = v.DeleteWorkflowExecution(ctx, request); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v *esVisibilityStore) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\n\tcheckPageSize(request)\n\n\ttoken, err := es.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tqueryDSL, err := v.getESQueryDSL(request, token)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Error when parse query: %v\", err)}\n\t}\n\n\tresp, err := v.esClient.SearchByQuery(ctx, &es.SearchByQueryRequest{\n\t\tIndex:           v.index,\n\t\tQuery:           queryDSL,\n\t\tNextPageToken:   request.NextPageToken,\n\t\tPageSize:        request.PageSize,\n\t\tFilter:          nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t})\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListWorkflowExecutions failed, %v\", err),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\n\tcheckPageSize(request)\n\n\ttoken, err := es.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar queryDSL string\n\tif len(token.ScrollID) == 0 { // first call\n\t\tqueryDSL, err = getESQueryDSLForScan(request)\n\t\tif err != nil {\n\t\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Error when parse query: %v\", err)}\n\t\t}\n\t}\n\n\tscanRequest := &es.ScanByQueryRequest{\n\t\tIndex:         v.index,\n\t\tQuery:         queryDSL,\n\t\tNextPageToken: request.NextPageToken,\n\t\tPageSize:      request.PageSize,\n\t}\n\n\tresp, err := v.esClient.ScanByQuery(ctx, scanRequest)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ScanWorkflowExecutions failed: %v, scanRequest: %+v\", err, scanRequest),\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (v *esVisibilityStore) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.CountWorkflowExecutionsRequest,\n) (\n\t*p.CountWorkflowExecutionsResponse, error) {\n\n\tqueryDSL, err := getESQueryDSLForCount(request)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Error when parse query: %v\", err)}\n\t}\n\n\tcount, err := v.esClient.CountByQuery(ctx, v.index, queryDSL)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"CountWorkflowExecutions failed. Error: %v\", err),\n\t\t}\n\t}\n\n\tresponse := &p.CountWorkflowExecutionsResponse{Count: count}\n\treturn response, nil\n}\n\nconst (\n\tjsonMissingCloseTime     = `{\"missing\":{\"field\":\"CloseTime\"}}`\n\tjsonRangeOnExecutionTime = `{\"range\":{\"ExecutionTime\":`\n\tjsonSortForOpen          = `[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]`\n\tjsonSortWithTieBreaker   = `{\"RunID\":\"desc\"}`\n\tjsonMissingStartTime     = `{\"missing\":{\"field\":\"StartTime\"}}` // used to identify uninitialized workflow execution records\n\n\tdslFieldSort        = \"sort\"\n\tdslFieldSearchAfter = \"search_after\"\n\tdslFieldFrom        = \"from\"\n\tdslFieldSize        = \"size\"\n\n\tdefaultDateTimeFormat = time.RFC3339 // used for converting UnixNano to string like 2018-02-15T16:16:36-08:00\n)\n\nvar (\n\ttimeKeys = map[string]bool{\n\t\tes.StartTime:     true,\n\t\tes.CloseTime:     true,\n\t\tes.ExecutionTime: true,\n\t\tes.UpdateTime:    true,\n\t}\n\trangeKeys = map[string]bool{\n\t\t\"from\":  true,\n\t\t\"to\":    true,\n\t\t\"gt\":    true,\n\t\t\"lt\":    true,\n\t\t\"query\": true,\n\t}\n)\n\nvar missingStartTimeRegex = regexp.MustCompile(jsonMissingStartTime)\n\nfunc getESQueryDSLForScan(request *p.ListWorkflowExecutionsByQueryRequest) (string, error) {\n\tsql := getSQLFromListRequest(request)\n\tdsl, err := getCustomizedDSLFromSQL(sql, request.DomainUUID)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// remove not needed fields\n\tdsl.Del(dslFieldSort)\n\treturn dsl.String(), nil\n}\n\nfunc getESQueryDSLForCount(request *p.CountWorkflowExecutionsRequest) (string, error) {\n\tsql := getSQLFromCountRequest(request)\n\tdsl, err := getCustomizedDSLFromSQL(sql, request.DomainUUID)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// remove not needed fields\n\tdsl.Del(dslFieldFrom)\n\tdsl.Del(dslFieldSize)\n\tdsl.Del(dslFieldSort)\n\n\treturn dsl.String(), nil\n}\n\nfunc (v *esVisibilityStore) getESQueryDSL(request *p.ListWorkflowExecutionsByQueryRequest, token *es.ElasticVisibilityPageToken) (string, error) {\n\tsql := getSQLFromListRequest(request)\n\treturn v.processedDSLfromSQL(sql, request.DomainUUID, token)\n}\n\nfunc (v *esVisibilityStore) processedDSLfromSQL(sql, domainUUID string, token *es.ElasticVisibilityPageToken) (string, error) {\n\tdsl, err := getCustomizedDSLFromSQL(sql, domainUUID)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tsortField, err := v.processSortField(dsl)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tif es.ShouldSearchAfter(token) {\n\t\tvalueOfSearchAfter, err := v.getValueOfSearchAfterInJSON(token, sortField)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\tdsl.Set(dslFieldSearchAfter, fastjson.MustParse(valueOfSearchAfter))\n\t} else { // use from+size\n\t\tdsl.Set(dslFieldFrom, fastjson.MustParse(strconv.Itoa(token.From)))\n\t}\n\n\tdslStr := cleanDSL(dsl.String())\n\n\treturn dslStr, nil\n}\n\nfunc getSQLFromListRequest(request *p.ListWorkflowExecutionsByQueryRequest) string {\n\tvar sql string\n\tquery := strings.TrimSpace(request.Query)\n\tif query == \"\" {\n\t\tsql = fmt.Sprintf(\"select * from dummy limit %d\", request.PageSize)\n\t} else if common.IsJustOrderByClause(query) {\n\t\tsql = fmt.Sprintf(\"select * from dummy %s limit %d\", request.Query, request.PageSize)\n\t} else {\n\t\tsql = fmt.Sprintf(\"select * from dummy where %s limit %d\", request.Query, request.PageSize)\n\t}\n\treturn sql\n}\n\nfunc getSQLFromCountRequest(request *p.CountWorkflowExecutionsRequest) string {\n\tvar sql string\n\tif strings.TrimSpace(request.Query) == \"\" {\n\t\tsql = \"select * from dummy\"\n\t} else {\n\t\tsql = fmt.Sprintf(\"select * from dummy where %s\", request.Query)\n\t}\n\treturn sql\n}\n\nfunc getCustomizedDSLFromSQL(sql string, domainID string) (*fastjson.Value, error) {\n\t// Only process LIKE clauses if they exist\n\tif strings.Contains(strings.ToLower(sql), \" like \") {\n\t\treturn processSQLWithLike(sql, domainID)\n\t}\n\n\t// No LIKE clauses found, use the original elasticsql.Convert\n\treturn processSQLWithoutLike(sql, domainID)\n}\n\n// processSQLWithLike handles SQL queries that contain LIKE clauses\nfunc processSQLWithLike(sql string, domainID string) (*fastjson.Value, error) {\n\tlikeClauses, strippedSQL := extractLikeClauses(sql)\n\n\t// Step 1: Convert with elasticsql for the non-LIKE portion\n\tdslStr, _, err := elasticsql.Convert(strippedSQL)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdsl, err := fastjson.Parse(dslStr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// Step 2: Patch wildcard queries back in\n\tif err := injectWildcardQueries(dsl, likeClauses); err != nil {\n\t\treturn nil, err\n\t}\n\tdsl = replaceDummyQuery(dsl)\n\treturn applyStandardProcessing(dsl, domainID)\n}\n\n// processSQLWithoutLike handles SQL queries without LIKE clauses\nfunc processSQLWithoutLike(sql string, domainID string) (*fastjson.Value, error) {\n\tdslStr, _, err := elasticsql.Convert(sql)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdsl, err := fastjson.Parse(dslStr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn applyStandardProcessing(dsl, domainID)\n}\n\n// applyStandardProcessing applies the standard post-processing steps to the DSL\nfunc applyStandardProcessing(dsl *fastjson.Value, domainID string) (*fastjson.Value, error) {\n\tdslStr := dsl.String()\n\tif strings.Contains(dslStr, jsonMissingStartTime) { // isUninitialized\n\t\tdsl = replaceQueryForUninitialized(dsl)\n\t}\n\tif strings.Contains(dslStr, jsonMissingCloseTime) { // isOpen\n\t\tdsl = replaceQueryForOpen(dsl)\n\t}\n\tif strings.Contains(dslStr, jsonRangeOnExecutionTime) {\n\t\taddQueryForExecutionTime(dsl)\n\t}\n\taddDomainToQuery(dsl, domainID)\n\tif err := processAllValuesForKey(dsl, isCombinedKey, combinedProcessFunc); err != nil {\n\t\treturn nil, err\n\t}\n\treturn dsl, nil\n}\n\n// ES v6 only accepts \"must_not exists\" query instead of \"missing\" query, but elasticsql produces \"missing\",\n// so use this func to replace.\n// Note it also means a temp limitation that we cannot support field missing search\nfunc replaceQueryForOpen(dsl *fastjson.Value) *fastjson.Value {\n\tre := regexp.MustCompile(jsonMissingCloseTime)\n\tnewDslStr := re.ReplaceAllString(dsl.String(), `{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"CloseTime\"}}}}`)\n\tdsl = fastjson.MustParse(newDslStr)\n\treturn dsl\n}\n\n// ES v6 only accepts \"must_not exists\" query instead of \"missing\" query, but elasticsql produces \"missing\",\n// so use this func to replace.\nfunc replaceQueryForUninitialized(dsl *fastjson.Value) *fastjson.Value {\n\tnewDslStr := missingStartTimeRegex.ReplaceAllString(dsl.String(), `{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"StartTime\"}}}}`)\n\tdsl = fastjson.MustParse(newDslStr)\n\treturn dsl\n}\n\nfunc addQueryForExecutionTime(dsl *fastjson.Value) {\n\texecutionTimeQueryString := `{\"range\" : {\"ExecutionTime\" : {\"gt\" : \"0\"}}}`\n\taddMustQuery(dsl, executionTimeQueryString)\n}\n\nfunc addDomainToQuery(dsl *fastjson.Value, domainID string) {\n\tif len(domainID) == 0 {\n\t\treturn\n\t}\n\n\tdomainQueryString := fmt.Sprintf(`{\"match_phrase\":{\"DomainID\":{\"query\":\"%s\"}}}`, domainID)\n\taddMustQuery(dsl, domainQueryString)\n}\n\n// addMustQuery is wrapping bool query with new bool query with must,\n// reason not making a flat bool query is to ensure \"should (or)\" query works correctly in query context.\nfunc addMustQuery(dsl *fastjson.Value, queryString string) {\n\tvalOfTopQuery := dsl.Get(\"query\")\n\tvalOfBool := dsl.Get(\"query\", \"bool\")\n\tnewValOfBool := fmt.Sprintf(`{\"must\":[%s,{\"bool\":%s}]}`, queryString, valOfBool.String())\n\tvalOfTopQuery.Set(\"bool\", fastjson.MustParse(newValOfBool))\n}\n\nfunc (v *esVisibilityStore) processSortField(dsl *fastjson.Value) (string, error) {\n\tisSorted := dsl.Exists(dslFieldSort)\n\tvar sortField string\n\n\tif !isSorted { // set default sorting by StartTime desc\n\t\tdsl.Set(dslFieldSort, fastjson.MustParse(jsonSortForOpen))\n\t\tsortField = definition.StartTime\n\t} else { // user provide sorting using order by\n\t\t// sort validation on length\n\t\tif len(dsl.GetArray(dslFieldSort)) > 1 {\n\t\t\treturn \"\", errors.New(\"only one field can be used to sort\")\n\t\t}\n\t\t// sort validation to exclude IndexedValueTypeString\n\t\tobj, _ := dsl.GetArray(dslFieldSort)[0].Object()\n\t\tobj.Visit(func(k []byte, v *fastjson.Value) { // visit is only way to get object key in fastjson\n\t\t\tsortField = string(k)\n\t\t})\n\t\tif v.getFieldType(sortField) == types.IndexedValueTypeString {\n\t\t\treturn \"\", errors.New(\"not able to sort by IndexedValueTypeString field, use IndexedValueTypeKeyword field\")\n\t\t}\n\t\t// add RunID as tie-breaker\n\t\tdsl.Get(dslFieldSort).Set(\"1\", fastjson.MustParse(jsonSortWithTieBreaker))\n\t}\n\n\treturn sortField, nil\n}\n\nfunc (v *esVisibilityStore) getFieldType(fieldName string) types.IndexedValueType {\n\tif strings.HasPrefix(fieldName, definition.Attr) {\n\t\tfieldName = fieldName[len(definition.Attr)+1:] // remove prefix\n\t}\n\tvalidMap := v.config.ValidSearchAttributes()\n\tfieldType, ok := validMap[fieldName]\n\tif !ok {\n\t\tv.logger.Error(\"Unknown fieldName, validation should be done in frontend already\", tag.Value(fieldName))\n\t}\n\treturn common.ConvertIndexedValueTypeToInternalType(fieldType, v.logger)\n}\n\nfunc (v *esVisibilityStore) getValueOfSearchAfterInJSON(token *es.ElasticVisibilityPageToken, sortField string) (string, error) {\n\tvar sortVal interface{}\n\tvar err error\n\tswitch v.getFieldType(sortField) {\n\tcase types.IndexedValueTypeInt, types.IndexedValueTypeDatetime, types.IndexedValueTypeBool:\n\t\tsortVal, err = token.SortValue.(json.Number).Int64()\n\t\tif err != nil {\n\t\t\terr, ok := err.(*strconv.NumError) // field not present, ES will return big int +-9223372036854776000\n\t\t\tif !ok {\n\t\t\t\treturn \"\", err\n\t\t\t}\n\t\t\tif err.Num[0] == '-' { // desc\n\t\t\t\tsortVal = math.MinInt64\n\t\t\t} else { // asc\n\t\t\t\tsortVal = math.MaxInt64\n\t\t\t}\n\t\t}\n\tcase types.IndexedValueTypeDouble:\n\t\tswitch token.SortValue.(type) {\n\t\tcase json.Number:\n\t\t\tsortVal, err = token.SortValue.(json.Number).Float64()\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", err\n\t\t\t}\n\t\tcase string: // field not present, ES will return \"-Infinity\" or \"Infinity\"\n\t\t\tsortVal = fmt.Sprintf(`\"%s\"`, token.SortValue.(string))\n\t\t}\n\tcase types.IndexedValueTypeKeyword:\n\t\tif token.SortValue != nil {\n\t\t\tsortVal = fmt.Sprintf(`\"%s\"`, token.SortValue.(string))\n\t\t} else { // field not present, ES will return null (so token.SortValue is nil)\n\t\t\tsortVal = \"null\"\n\t\t}\n\tdefault:\n\t\tsortVal = token.SortValue\n\t}\n\n\treturn fmt.Sprintf(`[%v, \"%s\"]`, sortVal, token.TieBreaker), nil\n}\n\nfunc (v *esVisibilityStore) checkProducer() {\n\tif v.producer == nil {\n\t\t// must be bug, check history setup\n\t\tpanic(\"message producer is nil\")\n\t}\n}\n\nfunc createVisibilityMessage(\n\t// common parameters\n\tdomainID string,\n\twid,\n\trid string,\n\tworkflowTypeName string,\n\ttaskList string,\n\tstartTimeUnixNano int64,\n\texecutionTimeUnixNano int64,\n\ttaskID int64,\n\tmemo []byte,\n\tencoding constants.EncodingType,\n\tisCron bool,\n\tNumClusters int16,\n\tclusterAttributeScope string,\n\tclusterAttributeName string,\n\tsearchAttributes map[string][]byte,\n\tvisibilityOperation constants.VisibilityOperation,\n\t// specific to certain status\n\tendTimeUnixNano int64, // close execution\n\tcloseStatus workflow.WorkflowExecutionCloseStatus, // close execution\n\thistoryLength int64, // close execution\n\tupdateTimeUnixNano int64, // update execution,\n\tshardID int64,\n\tcronSchedule string,\n\texecutionStatus int32,\n\tscheduledExecutionTimeUnixNano int64,\n) *indexer.Message {\n\tmsgType := indexer.MessageTypeIndex\n\n\tfields := map[string]*indexer.Field{\n\t\tes.WorkflowType:           {Type: &es.FieldTypeString, StringData: common.StringPtr(workflowTypeName)},\n\t\tes.StartTime:              {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(startTimeUnixNano)},\n\t\tes.ExecutionTime:          {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(executionTimeUnixNano)},\n\t\tes.TaskList:               {Type: &es.FieldTypeString, StringData: common.StringPtr(taskList)},\n\t\tes.IsCron:                 {Type: &es.FieldTypeBool, BoolData: common.BoolPtr(isCron)},\n\t\tes.NumClusters:            {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(int64(NumClusters))},\n\t\tes.ClusterAttributeScope:  {Type: &es.FieldTypeString, StringData: common.StringPtr(clusterAttributeScope)},\n\t\tes.ClusterAttributeName:   {Type: &es.FieldTypeString, StringData: common.StringPtr(clusterAttributeName)},\n\t\tes.UpdateTime:             {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(updateTimeUnixNano)},\n\t\tes.ShardID:                {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(shardID)},\n\t\tes.CronSchedule:           {Type: &es.FieldTypeString, StringData: common.StringPtr(cronSchedule)},\n\t\tes.ExecutionStatus:        {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(int64(executionStatus))},\n\t\tes.ScheduledExecutionTime: {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(scheduledExecutionTimeUnixNano)},\n\t}\n\n\tif len(memo) != 0 {\n\t\tfields[es.Memo] = &indexer.Field{Type: &es.FieldTypeBinary, BinaryData: memo}\n\t\tfields[es.Encoding] = &indexer.Field{Type: &es.FieldTypeString, StringData: common.StringPtr(string(encoding))}\n\t}\n\tfor k, v := range searchAttributes {\n\t\tfields[k] = &indexer.Field{Type: &es.FieldTypeBinary, BinaryData: v}\n\t}\n\n\tswitch visibilityOperation {\n\tcase constants.RecordStarted:\n\tcase constants.RecordClosed:\n\t\tfields[es.CloseTime] = &indexer.Field{Type: &es.FieldTypeInt, IntData: common.Int64Ptr(endTimeUnixNano)}\n\t\tfields[es.CloseStatus] = &indexer.Field{Type: &es.FieldTypeInt, IntData: common.Int64Ptr(int64(closeStatus))}\n\t\tfields[es.HistoryLength] = &indexer.Field{Type: &es.FieldTypeInt, IntData: common.Int64Ptr(historyLength)}\n\t}\n\n\tvar visibilityOperationThrift indexer.VisibilityOperation = -1\n\tswitch visibilityOperation {\n\tcase constants.RecordStarted:\n\t\tvisibilityOperationThrift = indexer.VisibilityOperationRecordStarted\n\tcase constants.RecordClosed:\n\t\tvisibilityOperationThrift = indexer.VisibilityOperationRecordClosed\n\tcase constants.UpsertSearchAttributes:\n\t\tvisibilityOperationThrift = indexer.VisibilityOperationUpsertSearchAttributes\n\tdefault:\n\t\tpanic(\"VisibilityOperation not set\")\n\t}\n\n\tmsg := &indexer.Message{\n\t\tMessageType:         &msgType,\n\t\tDomainID:            common.StringPtr(domainID),\n\t\tWorkflowID:          common.StringPtr(wid),\n\t\tRunID:               common.StringPtr(rid),\n\t\tVersion:             common.Int64Ptr(taskID),\n\t\tFields:              fields,\n\t\tVisibilityOperation: &visibilityOperationThrift,\n\t}\n\n\treturn msg\n}\n\nfunc getVisibilityMessageForDeletion(domainID, workflowID, runID string, docVersion int64) *indexer.Message {\n\tmsgType := indexer.MessageTypeDelete\n\tmsg := &indexer.Message{\n\t\tMessageType: &msgType,\n\t\tDomainID:    common.StringPtr(domainID),\n\t\tWorkflowID:  common.StringPtr(workflowID),\n\t\tRunID:       common.StringPtr(runID),\n\t\tVersion:     common.Int64Ptr(docVersion),\n\t}\n\treturn msg\n}\n\nfunc getVisibilityMessageForUninitializedWorkflow(\n\tdomainID string,\n\twid,\n\trid string,\n\tworkflowTypeName string,\n\tupdateTimeUnixNano int64, // update execution\n\tshardID int64,\n) *indexer.Message {\n\tmsgType := indexer.MessageTypeCreate\n\tfields := map[string]*indexer.Field{\n\t\tes.WorkflowType: {Type: &es.FieldTypeString, StringData: common.StringPtr(workflowTypeName)},\n\t\tes.UpdateTime:   {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(updateTimeUnixNano)},\n\t\tes.ShardID:      {Type: &es.FieldTypeInt, IntData: common.Int64Ptr(shardID)},\n\t}\n\n\tmsg := &indexer.Message{\n\t\tMessageType: &msgType,\n\t\tDomainID:    common.StringPtr(domainID),\n\t\tWorkflowID:  common.StringPtr(wid),\n\t\tRunID:       common.StringPtr(rid),\n\t\tFields:      fields,\n\t}\n\treturn msg\n}\n\nfunc checkPageSize(request *p.ListWorkflowExecutionsByQueryRequest) {\n\tif request.PageSize == 0 {\n\t\trequest.PageSize = 1000\n\t}\n}\n\nfunc processAllValuesForKey(\n\tdsl *fastjson.Value,\n\tkeyFilter func(k string) bool,\n\tprocessFunc func(obj *fastjson.Object, key string, v *fastjson.Value) error,\n) error {\n\tswitch dsl.Type() {\n\tcase fastjson.TypeArray:\n\t\tfor _, val := range dsl.GetArray() {\n\t\t\tif err := processAllValuesForKey(val, keyFilter, processFunc); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\tcase fastjson.TypeObject:\n\t\tobjectVal := dsl.GetObject()\n\t\tkeys := []string{}\n\t\tobjectVal.Visit(func(key []byte, val *fastjson.Value) {\n\t\t\tkeys = append(keys, string(key))\n\t\t})\n\n\t\tfor _, key := range keys {\n\t\t\tvar err error\n\t\t\tval := objectVal.Get(key)\n\t\t\tif keyFilter(key) {\n\t\t\t\terr = processFunc(objectVal, key, val)\n\t\t\t} else {\n\t\t\t\terr = processAllValuesForKey(val, keyFilter, processFunc)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\tdefault:\n\t\t// do nothing, since there's no key\n\t}\n\treturn nil\n}\n\nfunc isCombinedKey(key string) bool {\n\treturn isTimeKey(key) || isCloseStatusKey(key)\n}\n\nfunc combinedProcessFunc(obj *fastjson.Object, key string, value *fastjson.Value) error {\n\tif isTimeKey(key) {\n\t\treturn timeProcessFunc(obj, key, value)\n\t}\n\n\tif isCloseStatusKey(key) {\n\t\treturn closeStatusProcessFunc(obj, key, value)\n\t}\n\n\treturn fmt.Errorf(\"unknown es dsl key %v for processing value\", key)\n}\n\nfunc isTimeKey(key string) bool {\n\treturn timeKeys[key]\n}\n\nfunc timeProcessFunc(obj *fastjson.Object, key string, value *fastjson.Value) error {\n\treturn processAllValuesForKey(value, func(key string) bool {\n\t\treturn rangeKeys[key]\n\t}, func(obj *fastjson.Object, key string, v *fastjson.Value) error {\n\t\ttimeStr := string(v.GetStringBytes())\n\n\t\t// first check if already in int64 format\n\t\tif _, err := strconv.ParseInt(timeStr, 10, 64); err == nil {\n\t\t\treturn nil\n\t\t}\n\n\t\t// try to parse time\n\t\tparsedTime, err := time.Parse(defaultDateTimeFormat, timeStr)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tobj.Set(key, fastjson.MustParse(fmt.Sprintf(`\"%v\"`, parsedTime.UnixNano())))\n\t\treturn nil\n\t})\n}\n\nfunc isCloseStatusKey(key string) bool {\n\treturn key == es.CloseStatus\n}\n\nfunc closeStatusProcessFunc(obj *fastjson.Object, key string, value *fastjson.Value) error {\n\treturn processAllValuesForKey(value, func(key string) bool {\n\t\treturn rangeKeys[key]\n\t}, func(obj *fastjson.Object, key string, v *fastjson.Value) error {\n\t\tstatusStr := string(v.GetStringBytes())\n\n\t\t// first check if already in int64 format\n\t\tif _, err := strconv.ParseInt(statusStr, 10, 64); err == nil {\n\t\t\treturn nil\n\t\t}\n\n\t\t// try to parse close status string\n\t\tvar parsedStatus types.WorkflowExecutionCloseStatus\n\t\terr := parsedStatus.UnmarshalText([]byte(statusStr))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tobj.Set(key, fastjson.MustParse(fmt.Sprintf(`\"%d\"`, parsedStatus)))\n\t\treturn nil\n\t})\n}\n\n// elasticsql may transfer `Attr.Name` to \"`Attr.Name`\" instead of \"Attr.Name\" in dsl in some operator like \"between and\"\n// this function is used to clean up\nfunc cleanDSL(input string) string {\n\tvar re = regexp.MustCompile(\"(`)(Attr.\\\\w+)(`)\")\n\tresult := re.ReplaceAllString(input, `$2`)\n\treturn result\n}\n\ntype likeClause struct {\n\tField   string\n\tPattern string\n}\n\nfunc extractLikeClauses(sql string) ([]likeClause, string) {\n\tvar clauses []likeClause\n\tre := regexp.MustCompile(`(?i)([\\w\\.]+)\\s+LIKE\\s+'([^']+)'`)\n\tstrippedSQL := sql\n\n\tmatches := re.FindAllStringSubmatch(sql, -1)\n\tfor _, match := range matches {\n\t\tclauses = append(clauses, likeClause{\n\t\t\tField:   match[1],\n\t\t\tPattern: match[2],\n\t\t})\n\t\t// Remove LIKE clause from SQL\n\t\t// Replace LIKE with a dummy expression that elasticsql can parse\n\t\treplacement := `__dummy_field__ = '__dummy_value__'`\n\t\tstrippedSQL = strings.Replace(strippedSQL, match[0], replacement, 1)\n\t}\n\treturn clauses, strippedSQL\n}\n\nfunc injectWildcardQueries(dsl *fastjson.Value, likes []likeClause) error {\n\tobj, err := dsl.Object()\n\tif err != nil {\n\t\treturn err\n\t}\n\tqueryObj := obj.Get(\"query\")\n\tif queryObj == nil {\n\t\treturn fmt.Errorf(\"missing 'query' field\")\n\t}\n\n\tboolObj := queryObj.Get(\"bool\")\n\tif boolObj == nil {\n\t\treturn fmt.Errorf(\"missing 'bool' query\")\n\t}\n\n\t// Check if we have a 'should' array (for OR queries) or 'must' array (for AND queries)\n\tvar targetArray []*fastjson.Value\n\tvar arrayKey string\n\n\tif shouldArr := boolObj.GetArray(\"should\"); shouldArr != nil {\n\t\ttargetArray = shouldArr\n\t\tarrayKey = \"should\"\n\t} else if mustArr := boolObj.GetArray(\"must\"); mustArr != nil {\n\t\ttargetArray = mustArr\n\t\tarrayKey = \"must\"\n\t} else {\n\t\t// if neither exists, create a must array\n\t\ttargetArray = []*fastjson.Value{}\n\t\tarrayKey = \"must\"\n\t}\n\n\tfor _, clause := range likes {\n\t\twildcardValue := strings.ReplaceAll(clause.Pattern, \"%\", \"*\")\n\t\twildcardValue = strings.ReplaceAll(wildcardValue, \"_\", \"?\")\n\n\t\twildcard := fmt.Sprintf(`{\"wildcard\": {\"%s\": {\"value\": \"%s*\", \"case_insensitive\": true}}}`, clause.Field, wildcardValue)\n\t\tv, err := fastjson.Parse(wildcard)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttargetArray = append(targetArray, v)\n\t}\n\n\t// Inject updated array\n\tboolObj.Set(arrayKey, fastjson.MustParse(fmt.Sprintf(\"[%s]\", joinFastjson(targetArray, \",\"))))\n\treturn nil\n}\n\nfunc joinFastjson(arr []*fastjson.Value, sep string) string {\n\tparts := make([]string, len(arr))\n\tfor i, v := range arr {\n\t\tparts[i] = v.String()\n\t}\n\treturn strings.Join(parts, sep)\n}\n\nfunc replaceDummyQuery(dsl *fastjson.Value) *fastjson.Value {\n\t// Convert to string to find and remove the dummy query\n\tdslStr := dsl.String()\n\n\t// Remove all dummy match_phrase queries\n\tdummyQuery := `{\"match_phrase\":{\"__dummy_field__\":{\"query\":\"__dummy_value__\"}}}`\n\tdslStr = strings.ReplaceAll(dslStr, dummyQuery, \"\")\n\n\t// Clean up any trailing commas or empty arrays\n\tdslStr = strings.Replace(dslStr, \",,\", \",\", -1)\n\tdslStr = strings.Replace(dslStr, \"[,\", \"[\", -1)\n\tdslStr = strings.Replace(dslStr, \",]\", \"]\", -1)\n\tdslStr = strings.Replace(dslStr, \"[]\", \"\", -1)\n\n\t// Parse back to fastjson.Value\n\treturn fastjson.MustParse(dslStr)\n}\n"
  },
  {
    "path": "common/persistence/elasticsearch/es_visibility_store_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/valyala/fastjson\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\tesMocks \"github.com/uber/cadence/common/elasticsearch/mocks\"\n\t\"github.com/uber/cadence/common/elasticsearch/query\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype ESVisibilitySuite struct {\n\tsuite.Suite\n\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t// not merely log an error\n\t*require.Assertions\n\tvisibilityStore *esVisibilityStore\n\tmockESClient    *esMocks.GenericClient\n\tmockProducer    *mocks.KafkaProducer\n}\n\nvar (\n\ttestIndex        = \"test-index\"\n\ttestDomain       = \"test-domain\"\n\ttestDomainID     = \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"\n\ttestPageSize     = 5\n\ttestEarliestTime = int64(1547596872371000000)\n\ttestLatestTime   = int64(2547596872371000000)\n\ttestWorkflowType = \"test-wf-type\"\n\ttestWorkflowID   = \"test-wid\"\n\ttestCloseStatus  = int32(1)\n\n\ttestRequest = &p.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainID,\n\t\tDomain:       testDomain,\n\t\tPageSize:     testPageSize,\n\t\tEarliestTime: time.Unix(0, testEarliestTime),\n\t\tLatestTime:   time.Unix(0, testLatestTime),\n\t}\n\ttestSearchResult = &p.InternalListWorkflowExecutionsResponse{}\n\terrTestESSearch  = errors.New(\"ES error\")\n\n\ttestContextTimeout = 5 * time.Second\n\n\tesIndexMaxResultWindow = 3\n)\n\ntype Source struct {\n\tMatch map[string]struct {\n\t\tQuery interface{} `json:\"query\"`\n\t} `json:\"match\"`\n}\n\nfunc TestESVisibilitySuite(t *testing.T) {\n\tsuite.Run(t, new(ESVisibilitySuite))\n}\n\nfunc (s *ESVisibilitySuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\n\ts.mockESClient = &esMocks.GenericClient{}\n\tconfig := &service.Config{\n\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(esIndexMaxResultWindow),\n\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t}\n\n\ts.mockProducer = &mocks.KafkaProducer{}\n\tmgr := NewElasticSearchVisibilityStore(s.mockESClient, testIndex, s.mockProducer, config, testlogger.New(s.Suite.T()))\n\ts.visibilityStore = mgr.(*esVisibilityStore)\n}\n\nfunc (s *ESVisibilitySuite) TearDownTest() {\n\ts.mockESClient.AssertExpectations(s.T())\n\ts.mockProducer.AssertExpectations(s.T())\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionStarted() {\n\t// test non-empty request fields match\n\trequest := &p.InternalRecordWorkflowExecutionStartedRequest{}\n\trequest.DomainUUID = \"domainID\"\n\trequest.WorkflowID = \"wid\"\n\trequest.RunID = \"rid\"\n\trequest.WorkflowTypeName = \"wfType\"\n\trequest.StartTimestamp = time.Unix(0, int64(123))\n\trequest.ExecutionTimestamp = time.Unix(0, int64(321))\n\trequest.TaskID = int64(111)\n\trequest.IsCron = true\n\trequest.NumClusters = 2\n\trequest.ClusterAttributeScope = \"region\"\n\trequest.ClusterAttributeName = \"us-east-1\"\n\tmemoBytes := []byte(`test bytes`)\n\trequest.Memo = p.NewDataBlob(memoBytes, constants.EncodingTypeThriftRW)\n\trequest.ShardID = 1234\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\tfields := input.Fields\n\t\ts.Equal(request.DomainUUID, input.GetDomainID())\n\t\ts.Equal(request.WorkflowID, input.GetWorkflowID())\n\t\ts.Equal(request.RunID, input.GetRunID())\n\t\ts.Equal(request.TaskID, input.GetVersion())\n\t\ts.Equal(request.WorkflowTypeName, fields[es.WorkflowType].GetStringData())\n\t\ts.Equal(request.StartTimestamp.UnixNano(), fields[es.StartTime].GetIntData())\n\t\ts.Equal(request.ExecutionTimestamp.UnixNano(), fields[es.ExecutionTime].GetIntData())\n\t\ts.Equal(memoBytes, fields[es.Memo].GetBinaryData())\n\t\ts.Equal(string(constants.EncodingTypeThriftRW), fields[es.Encoding].GetStringData())\n\t\ts.Equal(request.IsCron, fields[es.IsCron].GetBoolData())\n\t\ts.Equal((int64)(request.NumClusters), fields[es.NumClusters].GetIntData())\n\t\ts.Equal(request.ClusterAttributeScope, fields[es.ClusterAttributeScope].GetStringData())\n\t\ts.Equal(request.ClusterAttributeName, fields[es.ClusterAttributeName].GetStringData())\n\t\ts.Equal(indexer.VisibilityOperationRecordStarted, *input.VisibilityOperation)\n\t\ts.Equal((int64)(request.ShardID), fields[es.ShardID].GetIntData())\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionStarted(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionStarted_EmptyRequest() {\n\t// test empty request\n\trequest := &p.InternalRecordWorkflowExecutionStartedRequest{\n\t\tMemo: &p.DataBlob{},\n\t}\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\ts.Equal(indexer.MessageTypeIndex, input.GetMessageType())\n\t\t_, ok := input.Fields[es.Memo]\n\t\ts.False(ok)\n\t\t_, ok = input.Fields[es.Encoding]\n\t\ts.False(ok)\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionStarted(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionClosed() {\n\t// test non-empty request fields match\n\trequest := &p.InternalRecordWorkflowExecutionClosedRequest{}\n\trequest.DomainUUID = \"domainID\"\n\trequest.WorkflowID = \"wid\"\n\trequest.RunID = \"rid\"\n\trequest.WorkflowTypeName = \"wfType\"\n\trequest.StartTimestamp = time.Unix(0, int64(123))\n\trequest.ExecutionTimestamp = time.Unix(0, int64(321))\n\trequest.TaskID = int64(111)\n\tmemoBytes := []byte(`test bytes`)\n\trequest.Memo = p.NewDataBlob(memoBytes, constants.EncodingTypeThriftRW)\n\trequest.CloseTimestamp = time.Unix(0, int64(999))\n\trequest.Status = types.WorkflowExecutionCloseStatusTerminated\n\trequest.HistoryLength = int64(20)\n\trequest.IsCron = false\n\trequest.NumClusters = 2\n\trequest.UpdateTimestamp = time.Unix(0, int64(213))\n\trequest.ShardID = 1234\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\tfields := input.Fields\n\t\ts.Equal(request.DomainUUID, input.GetDomainID())\n\t\ts.Equal(request.WorkflowID, input.GetWorkflowID())\n\t\ts.Equal(request.RunID, input.GetRunID())\n\t\ts.Equal(request.TaskID, input.GetVersion())\n\t\ts.Equal(request.WorkflowTypeName, fields[es.WorkflowType].GetStringData())\n\t\ts.Equal(request.StartTimestamp.UnixNano(), fields[es.StartTime].GetIntData())\n\t\ts.Equal(request.ExecutionTimestamp.UnixNano(), fields[es.ExecutionTime].GetIntData())\n\t\ts.Equal(memoBytes, fields[es.Memo].GetBinaryData())\n\t\ts.Equal(string(constants.EncodingTypeThriftRW), fields[es.Encoding].GetStringData())\n\t\ts.Equal(request.CloseTimestamp.UnixNano(), fields[es.CloseTime].GetIntData())\n\t\ts.Equal(int64(request.Status), fields[es.CloseStatus].GetIntData())\n\t\ts.Equal(request.HistoryLength, fields[es.HistoryLength].GetIntData())\n\t\ts.Equal(request.IsCron, fields[es.IsCron].GetBoolData())\n\t\ts.Equal((int64)(request.NumClusters), fields[es.NumClusters].GetIntData())\n\t\ts.Equal(indexer.VisibilityOperationRecordClosed, *input.VisibilityOperation)\n\t\ts.Equal(request.UpdateTimestamp.UnixNano(), fields[es.UpdateTime].GetIntData())\n\t\ts.Equal((int64)(request.ShardID), fields[es.ShardID].GetIntData())\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionClosed(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionClosed_EmptyRequest() {\n\t// test empty request\n\trequest := &p.InternalRecordWorkflowExecutionClosedRequest{\n\t\tMemo: &p.DataBlob{},\n\t}\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\ts.Equal(indexer.MessageTypeIndex, input.GetMessageType())\n\t\t_, ok := input.Fields[es.Memo]\n\t\ts.False(ok)\n\t\t_, ok = input.Fields[es.Encoding]\n\t\ts.False(ok)\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionClosed(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionUninitialized() {\n\t// test non-empty request fields match\n\trequest := &p.InternalRecordWorkflowExecutionUninitializedRequest{}\n\trequest.DomainUUID = \"domainID\"\n\trequest.WorkflowID = \"wid\"\n\trequest.RunID = \"rid\"\n\trequest.WorkflowTypeName = \"wfType\"\n\trequest.UpdateTimestamp = time.Unix(0, int64(213))\n\trequest.ShardID = 1234\n\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\tfields := input.Fields\n\t\ts.Equal(request.DomainUUID, input.GetDomainID())\n\t\ts.Equal(request.WorkflowID, input.GetWorkflowID())\n\t\ts.Equal(request.RunID, input.GetRunID())\n\t\ts.Equal(request.WorkflowTypeName, fields[es.WorkflowType].GetStringData())\n\t\ts.Equal(request.UpdateTimestamp.UnixNano(), fields[es.UpdateTime].GetIntData())\n\t\ts.Equal(request.ShardID, fields[es.ShardID].GetIntData())\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionUninitialized(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestRecordWorkflowExecutionUninitialized_EmptyRequest() {\n\t// test empty request\n\trequest := &p.InternalRecordWorkflowExecutionUninitializedRequest{}\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.Message) bool {\n\t\ts.Equal(indexer.MessageTypeCreate, input.GetMessageType())\n\t\treturn true\n\t})).Return(nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr := s.visibilityStore.RecordWorkflowExecutionUninitialized(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestListOpenWorkflowExecutions() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.True(input.IsOpen)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListOpenWorkflowExecutions(ctx, testRequest)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListOpenWorkflowExecutions(ctx, testRequest)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListOpenWorkflowExecutions failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListClosedWorkflowExecutions() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.False(input.IsOpen)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListClosedWorkflowExecutions(ctx, testRequest)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListClosedWorkflowExecutions(ctx, testRequest)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListClosedWorkflowExecutions failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListOpenWorkflowExecutionsByType() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.True(input.IsOpen)\n\t\tsource, err := matchQueryData(*input.MatchQuery)\n\t\ts.NoError(err)\n\t\ts.Equal(source.Match[es.WorkflowType].Query, testWorkflowType)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: *testRequest,\n\t\tWorkflowTypeName:                      testWorkflowType,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListOpenWorkflowExecutionsByType(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListOpenWorkflowExecutionsByType(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListOpenWorkflowExecutionsByType failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListClosedWorkflowExecutionsByType() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.False(input.IsOpen)\n\n\t\tsource, err := matchQueryData(*input.MatchQuery)\n\t\ts.NoError(err)\n\n\t\ts.Equal(source.Match[es.WorkflowType].Query, testWorkflowType)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: *testRequest,\n\t\tWorkflowTypeName:                      testWorkflowType,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListClosedWorkflowExecutionsByType(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListClosedWorkflowExecutionsByType(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListClosedWorkflowExecutionsByType failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListOpenWorkflowExecutionsByWorkflowID() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.True(input.IsOpen)\n\n\t\tsource, err := matchQueryData(*input.MatchQuery)\n\t\ts.NoError(err)\n\t\ts.Equal(source.Match[es.WorkflowID].Query, testWorkflowID)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: *testRequest,\n\t\tWorkflowID:                            testWorkflowID,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListOpenWorkflowExecutionsByWorkflowID failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListClosedWorkflowExecutionsByWorkflowID() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.False(input.IsOpen)\n\n\t\tsource, err := matchQueryData(*input.MatchQuery)\n\t\ts.NoError(err)\n\t\ts.Equal(source.Match[es.WorkflowID].Query, testWorkflowID)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: *testRequest,\n\t\tWorkflowID:                            testWorkflowID,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListClosedWorkflowExecutionsByWorkflowID failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListClosedWorkflowExecutionsByStatus() {\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.MatchedBy(func(input *es.SearchRequest) bool {\n\t\ts.False(input.IsOpen)\n\t\tsource, err := matchQueryData(*input.MatchQuery)\n\t\ts.NoError(err)\n\t\tcs := (int32)(source.Match[es.CloseStatus].Query.(float64))\n\t\ts.Equal(testCloseStatus, cs)\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\tcloseStatus := types.WorkflowExecutionCloseStatus(testCloseStatus)\n\trequest := &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\tInternalListWorkflowExecutionsRequest: *testRequest,\n\t\tStatus:                                closeStatus,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"Search\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListClosedWorkflowExecutionsByStatus failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestGetNextPageToken() {\n\ttoken, err := es.GetNextPageToken([]byte{})\n\ts.Equal(0, token.From)\n\ts.NoError(err)\n\n\tfrom := 5\n\tinput, err := es.SerializePageToken(&es.ElasticVisibilityPageToken{From: from})\n\ts.NoError(err)\n\ttoken, err = es.GetNextPageToken(input)\n\ts.Equal(from, token.From)\n\ts.NoError(err)\n\n\tbadInput := []byte(\"bad input\")\n\ttoken, err = es.GetNextPageToken(badInput)\n\ts.Nil(token)\n\ts.Error(err)\n}\n\nfunc (s *ESVisibilitySuite) TestDeserializePageToken() {\n\ttoken := &es.ElasticVisibilityPageToken{From: 0}\n\tdata, _ := es.SerializePageToken(token)\n\tresult, err := es.DeserializePageToken(data)\n\ts.NoError(err)\n\ts.Equal(token, result)\n\n\tbadInput := []byte(\"bad input\")\n\tresult, err = es.DeserializePageToken(badInput)\n\ts.Error(err)\n\ts.Nil(result)\n\terr, ok := err.(*types.BadRequestError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"unable to deserialize page token\"))\n\n\ttoken = &es.ElasticVisibilityPageToken{SortValue: int64(64), TieBreaker: \"unique\"}\n\tdata, _ = es.SerializePageToken(token)\n\tresult, err = es.DeserializePageToken(data)\n\ts.NoError(err)\n\tresultSortValue, err := result.SortValue.(json.Number).Int64()\n\ts.NoError(err)\n\ts.Equal(token.SortValue.(int64), resultSortValue)\n}\n\nfunc (s *ESVisibilitySuite) TestSerializePageToken() {\n\tdata, err := es.SerializePageToken(nil)\n\ts.NoError(err)\n\ts.True(len(data) > 0)\n\ttoken, err := es.DeserializePageToken(data)\n\ts.NoError(err)\n\ts.Equal(0, token.From)\n\ts.Equal(nil, token.SortValue)\n\ts.Equal(\"\", token.TieBreaker)\n\n\tnewToken := &es.ElasticVisibilityPageToken{From: 5}\n\tdata, err = es.SerializePageToken(newToken)\n\ts.NoError(err)\n\ts.True(len(data) > 0)\n\ttoken, err = es.DeserializePageToken(data)\n\ts.NoError(err)\n\ts.Equal(newToken, token)\n\n\tsortTime := int64(123)\n\ttieBreaker := \"unique\"\n\tnewToken = &es.ElasticVisibilityPageToken{SortValue: sortTime, TieBreaker: tieBreaker}\n\tdata, err = es.SerializePageToken(newToken)\n\ts.NoError(err)\n\ts.True(len(data) > 0)\n\ttoken, err = es.DeserializePageToken(data)\n\ts.NoError(err)\n\tresultSortValue, err := token.SortValue.(json.Number).Int64()\n\ts.NoError(err)\n\ts.Equal(newToken.SortValue, resultSortValue)\n\ts.Equal(newToken.TieBreaker, token.TieBreaker)\n}\n\n// Move to client_v6_test\n// func (s *ESVisibilitySuite) TestConvertSearchResultToVisibilityRecord() {\n//\tdata := []byte(`{\"CloseStatus\": 0,\n//          \"CloseTime\": 1547596872817380000,\n//          \"DomainID\": \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\",\n//          \"HistoryLength\": 29,\n//          \"KafkaKey\": \"7-619\",\n//          \"RunID\": \"e481009e-14b3-45ae-91af-dce6e2a88365\",\n//          \"StartTime\": 1547596872371000000,\n//          \"WorkflowID\": \"6bfbc1e5-6ce4-4e22-bbfb-e0faa9a7a604-1-2256\",\n//          \"WorkflowType\": \"TestWorkflowExecute\"}`)\n//\tsource := (*json.RawMessage)(&data)\n//\tsearchHit := &elastic.SearchHit{\n//\t\tSource: source,\n//\t}\n//\n//\t// test for open\n//\tinfo := s.visibilityStore.convertSearchResultToVisibilityRecord(searchHit)\n//\ts.NotNil(info)\n//\ts.Equal(\"6bfbc1e5-6ce4-4e22-bbfb-e0faa9a7a604-1-2256\", info.WorkflowID)\n//\ts.Equal(\"e481009e-14b3-45ae-91af-dce6e2a88365\", info.RunID)\n//\ts.Equal(\"TestWorkflowExecute\", info.TypeName)\n//\ts.Equal(int64(1547596872371000000), info.StartTime.UnixNano())\n//\n//\t// test for close\n//\tinfo = s.visibilityStore.convertSearchResultToVisibilityRecord(searchHit)\n//\ts.NotNil(info)\n//\ts.Equal(\"6bfbc1e5-6ce4-4e22-bbfb-e0faa9a7a604-1-2256\", info.WorkflowID)\n//\ts.Equal(\"e481009e-14b3-45ae-91af-dce6e2a88365\", info.RunID)\n//\ts.Equal(\"TestWorkflowExecute\", info.TypeName)\n//\ts.Equal(int64(1547596872371000000), info.StartTime.UnixNano())\n//\ts.Equal(int64(1547596872817380000), info.CloseTime.UnixNano())\n//\ts.Equal(workflow.WorkflowExecutionCloseStatusCompleted, *info.Status)\n//\ts.Equal(int64(29), info.HistoryLength)\n//\n//\t// test for error case\n//\tbadData := []byte(`corrupted data`)\n//\tsource = (*json.RawMessage)(&badData)\n//\tsearchHit = &elastic.SearchHit{\n//\t\tSource: source,\n//\t}\n//\tinfo = s.visibilityStore.convertSearchResultToVisibilityRecord(searchHit)\n//\ts.Nil(info)\n// }\n\nfunc (s *ESVisibilitySuite) TestShouldSearchAfter() {\n\ttoken := &es.ElasticVisibilityPageToken{}\n\ts.False(es.ShouldSearchAfter(token))\n\n\ttoken.TieBreaker = \"a\"\n\ts.True(es.ShouldSearchAfter(token))\n}\n\n// nolint\nfunc (s *ESVisibilitySuite) TestGetESQueryDSL() {\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID: testDomainID,\n\t\tPageSize:   10,\n\t}\n\ttoken := &es.ElasticVisibilityPageToken{}\n\n\tv := s.visibilityStore\n\n\trequest.Query = \"\"\n\tdsl, err := v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_all\":{}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = \"invaild query\"\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.NotNil(err)\n\ts.Equal(\"\", dsl)\n\n\trequest.Query = `WorkflowID = 'wid'`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' or WorkflowID = 'another-wid'`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"should\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"match_phrase\":{\"WorkflowID\":{\"query\":\"another-wid\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' order by StartTime desc`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' and CloseTime = missing`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"CloseTime\"}}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' or CloseTime = missing`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"should\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"CloseTime\"}}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' and CloseStatus = \"completed\"`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"match_phrase\":{\"CloseStatus\":{\"query\":\"0\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `CloseTime = missing order by CloseTime desc`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"CloseTime\"}}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"CloseTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `StartTime = \"2018-06-07T15:04:05-08:00\"`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"StartTime\":{\"query\":\"1528412645000000000\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' and StartTime > \"2018-06-07T15:04:05+00:00\"`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"range\":{\"StartTime\":{\"gt\":\"1528383845000000000\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `ExecutionTime < 1000`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"lt\":\"1000\"}}}]}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `ExecutionTime < 1000 or ExecutionTime > 2000`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"should\":[{\"range\":{\"ExecutionTime\":{\"lt\":\"1000\"}}},{\"range\":{\"ExecutionTime\":{\"gt\":\"2000\"}}}]}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `order by ExecutionTime desc`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_all\":{}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"ExecutionTime\":\"desc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `order by StartTime desc, CloseTime desc`\n\t_, err = v.getESQueryDSL(request, token)\n\ts.Equal(errors.New(\"only one field can be used to sort\"), err)\n\n\trequest.Query = `order by CustomStringField desc`\n\t_, err = v.getESQueryDSL(request, token)\n\ts.Equal(errors.New(\"not able to sort by IndexedValueTypeString field, use IndexedValueTypeKeyword field\"), err)\n\n\trequest.Query = `order by CustomIntField asc`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_all\":{}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"CustomIntField\":\"asc\"},{\"RunID\":\"desc\"}]}`, dsl)\n\n\trequest.Query = `ExecutionTime < \"unable to parse\"`\n\t_, err = v.getESQueryDSL(request, token)\n\ts.Error(err)\n\n\ttoken = s.getTokenHelper(1)\n\trequest.Query = `WorkflowID = 'wid'`\n\tdsl, err = v.getESQueryDSL(request, token)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}],\"search_after\":[1,\"t\"]}`, dsl)\n\n\t// invalid union injection\n\trequest.Query = `WorkflowID = 'wid' union select * from dummy`\n\t_, err = v.getESQueryDSL(request, token)\n\ts.NotNil(err)\n}\n\nfunc (s *ESVisibilitySuite) TestGetESQueryDSLForScan() {\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID: testDomainID,\n\t\tPageSize:   10,\n\t}\n\n\trequest.Query = `WorkflowID = 'wid' order by StartTime desc`\n\tdsl, err := getESQueryDSLForScan(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":10}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid'`\n\tdsl, err = getESQueryDSLForScan(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":10}`, dsl)\n\n\trequest.Query = `CloseTime = missing and (ExecutionTime >= \"2019-08-27T15:04:05+00:00\" or StartTime <= \"2018-06-07T15:04:05+00:00\")`\n\tdsl, err = getESQueryDSLForScan(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"must\":[{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"CloseTime\"}}}},{\"bool\":{\"should\":[{\"range\":{\"ExecutionTime\":{\"from\":\"1566918245000000000\"}}},{\"range\":{\"StartTime\":{\"to\":\"1528383845000000000\"}}}]}}]}}]}}]}},\"from\":0,\"size\":10}`, dsl)\n\n\trequest.Query = `ExecutionTime < 1000 and ExecutionTime > 500`\n\tdsl, err = getESQueryDSLForScan(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"lt\":\"1000\"}}},{\"range\":{\"ExecutionTime\":{\"gt\":\"500\"}}}]}}]}}]}},\"from\":0,\"size\":10}`, dsl)\n}\n\nfunc (s *ESVisibilitySuite) TestGetESQueryDSLForCount() {\n\trequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainID,\n\t}\n\n\t// empty query\n\tdsl, err := getESQueryDSLForCount(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_all\":{}}]}}]}}}`, dsl)\n\n\trequest.Query = `WorkflowID = 'wid' order by StartTime desc`\n\tdsl, err = getESQueryDSLForCount(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}}}`, dsl)\n\n\trequest.Query = `CloseTime < \"2018-06-07T15:04:05+07:00\" and StartTime > \"2018-05-04T16:00:00+07:00\" and ExecutionTime >= \"2018-05-05T16:00:00+07:00\"`\n\tdsl, err = getESQueryDSLForCount(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"must\":[{\"range\":{\"CloseTime\":{\"lt\":\"1528358645000000000\"}}},{\"range\":{\"StartTime\":{\"gt\":\"1525424400000000000\"}}},{\"range\":{\"ExecutionTime\":{\"from\":\"1525510800000000000\"}}}]}}]}}]}}}`, dsl)\n\n\trequest.Query = `ExecutionTime < 1000`\n\tdsl, err = getESQueryDSLForCount(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"gt\":\"0\"}}},{\"bool\":{\"must\":[{\"range\":{\"ExecutionTime\":{\"lt\":\"1000\"}}}]}}]}}]}}}`, dsl)\n\n\trequest.Query = `StartTime = missing and UpdateTime >= \"2022-10-04T16:00:00+07:00\"`\n\tdsl, err = getESQueryDSLForCount(request)\n\ts.Nil(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"bool\":{\"must_not\":{\"exists\":{\"field\":\"StartTime\"}}}},{\"range\":{\"UpdateTime\":{\"from\":\"1664874000000000000\"}}}]}}]}}}`, dsl)\n\n}\n\nfunc (s *ESVisibilitySuite) TestAddDomainToQuery() {\n\tdsl := fastjson.MustParse(`{}`)\n\tdslStr := dsl.String()\n\taddDomainToQuery(dsl, \"\")\n\ts.Equal(dslStr, dsl.String())\n\n\tdsl = fastjson.MustParse(`{\"query\":{\"bool\":{\"must\":[{\"match_all\":{}}]}}}`)\n\taddDomainToQuery(dsl, testDomainID)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_all\":{}}]}}]}}}`, dsl.String())\n}\n\nfunc (s *ESVisibilitySuite) TestListWorkflowExecutions() {\n\ts.mockESClient.On(\"SearchByQuery\", mock.Anything, mock.MatchedBy(func(input *es.SearchByQueryRequest) bool {\n\t\ts.True(strings.Contains(input.Query, `{\"match_phrase\":{\"CloseStatus\":{\"query\":\"5\"}}}`))\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tPageSize:   10,\n\t\tQuery:      `CloseStatus = 5`,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n\n\ts.mockESClient.On(\"SearchByQuery\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ListWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ListWorkflowExecutions failed\"))\n\n\trequest.Query = `invalid query`\n\t_, err = s.visibilityStore.ListWorkflowExecutions(context.Background(), request)\n\ts.Error(err)\n\t_, ok = err.(*types.BadRequestError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"Error when parse query\"))\n}\n\nfunc (s *ESVisibilitySuite) TestListWorkflowExecutionsWithLike() {\n\ts.mockESClient.On(\"SearchByQuery\", mock.Anything, mock.MatchedBy(func(input *es.SearchByQueryRequest) bool {\n\t\ts.True(strings.Contains(input.Query, `{\"wildcard\":{\"WorkflowID\":{\"value\":\"wid*\",\"case_insensitive\":true}}}`))\n\t\ts.Equal(esIndexMaxResultWindow, input.MaxResultWindow)\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tPageSize:   10,\n\t\tQuery:      `WorkflowID like 'wid'`,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ListWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n}\n\nfunc (s *ESVisibilitySuite) TestScanWorkflowExecutions() {\n\t// test first page\n\ts.mockESClient.On(\"ScanByQuery\", mock.Anything, mock.MatchedBy(func(input *es.ScanByQueryRequest) bool {\n\t\ts.True(strings.Contains(input.Query, `{\"match_phrase\":{\"CloseStatus\":{\"query\":\"5\"}}}`))\n\t\treturn true\n\t})).Return(testSearchResult, nil).Once()\n\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tPageSize:   10,\n\t\tQuery:      `CloseStatus = 5`,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\t_, err := s.visibilityStore.ScanWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n\n\t// test bad request\n\trequest.Query = `invalid query`\n\t_, err = s.visibilityStore.ScanWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.BadRequestError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"Error when parse query\"))\n\n\t// test internal error\n\trequest.Query = `CloseStatus = 5`\n\ts.mockESClient.On(\"ScanByQuery\", mock.Anything, mock.Anything).Return(nil, errTestESSearch).Once()\n\t_, err = s.visibilityStore.ScanWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\t_, ok = err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"ScanWorkflowExecutions failed\"))\n}\n\nfunc (s *ESVisibilitySuite) TestCountWorkflowExecutions() {\n\ts.mockESClient.On(\"CountByQuery\", mock.Anything, testIndex, mock.MatchedBy(func(input string) bool {\n\t\ts.True(strings.Contains(input, `{\"match_phrase\":{\"CloseStatus\":{\"query\":\"5\"}}}`))\n\t\treturn true\n\t})).Return(int64(1), nil).Once()\n\n\trequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tQuery:      `CloseStatus = 5`,\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tresp, err := s.visibilityStore.CountWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n\ts.Equal(int64(1), resp.Count)\n\n\t// test internal error\n\ts.mockESClient.On(\"CountByQuery\", mock.Anything, testIndex, mock.Anything).Return(int64(0), errTestESSearch).Once()\n\n\t_, err = s.visibilityStore.CountWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\t_, ok := err.(*types.InternalServiceError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"CountWorkflowExecutions failed\"))\n\n\t// test bad request\n\trequest.Query = `invalid query`\n\t_, err = s.visibilityStore.CountWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\t_, ok = err.(*types.BadRequestError)\n\ts.True(ok)\n\ts.True(strings.Contains(err.Error(), \"Error when parse query\"))\n}\n\nfunc (s *ESVisibilitySuite) TestTimeProcessFunc() {\n\tcases := []struct {\n\t\tkey   string\n\t\tvalue string\n\t}{\n\t\t{key: \"from\", value: \"1528358645000000000\"},\n\t\t{key: \"to\", value: \"2018-06-07T15:04:05+07:00\"},\n\t\t{key: \"gt\", value: \"some invalid time string\"},\n\t\t{key: \"unrelatedKey\", value: \"should not be modified\"},\n\t}\n\texpected := []struct {\n\t\tvalue     string\n\t\treturnErr bool\n\t}{\n\t\t{value: `\"1528358645000000000\"`, returnErr: false},\n\t\t{value: `\"1528358645000000000\"`, returnErr: false},\n\t\t{value: \"\", returnErr: true},\n\t\t{value: `\"should not be modified\"`, returnErr: false},\n\t}\n\n\tfor i, testCase := range cases {\n\t\tvalue := fastjson.MustParse(fmt.Sprintf(`{\"%s\": \"%s\"}`, testCase.key, testCase.value))\n\t\terr := timeProcessFunc(nil, \"\", value)\n\t\tif expected[i].returnErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.Equal(expected[i].value, value.Get(testCase.key).String())\n\t}\n}\n\nfunc (s *ESVisibilitySuite) TestCloseStatusProcessFunc() {\n\tcases := []struct {\n\t\tkey   string\n\t\tvalue string\n\t}{\n\t\t{key: \"from\", value: types.WorkflowExecutionCloseStatusTerminated.String()},\n\t\t{key: \"to\", value: strings.ToLower(types.WorkflowExecutionCloseStatusContinuedAsNew.String())},\n\t\t{key: \"gt\", value: fmt.Sprintf(`%d`, types.WorkflowExecutionCloseStatusFailed)},\n\t\t{key: \"query\", value: types.WorkflowExecutionCloseStatusCompleted.String()},\n\t\t{key: \"query\", value: fmt.Sprintf(`%d`, types.WorkflowExecutionCloseStatusCanceled)},\n\t\t{key: \"query\", value: strings.ToTitle(strings.ToLower(types.WorkflowExecutionCloseStatusTimedOut.String()))},\n\t\t{key: \"query\", value: \"timeout\"},\n\t\t{key: \"unrelatedKey\", value: \"should not be modified\"},\n\t}\n\texpected := []struct {\n\t\tvalue     string\n\t\treturnErr bool\n\t}{\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusTerminated), returnErr: false},\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusContinuedAsNew), returnErr: false},\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusFailed), returnErr: false},\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusCompleted), returnErr: false},\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusCanceled), returnErr: false},\n\t\t{value: fmt.Sprintf(`\"%d\"`, types.WorkflowExecutionCloseStatusTimedOut), returnErr: false},\n\t\t{value: \"\", returnErr: true},\n\t\t{value: `\"should not be modified\"`, returnErr: false},\n\t}\n\n\tfor i, testCase := range cases {\n\t\tvalue := fastjson.MustParse(fmt.Sprintf(`{\"%s\": \"%s\"}`, testCase.key, testCase.value))\n\t\terr := closeStatusProcessFunc(nil, \"\", value)\n\t\tif expected[i].returnErr {\n\t\t\ts.Error(err)\n\t\t\tcontinue\n\t\t}\n\t\ts.Equal(expected[i].value, value.Get(testCase.key).String())\n\t}\n}\n\nfunc (s *ESVisibilitySuite) TestProcessAllValuesForKey() {\n\ttestJSONStr := `{\n\t\t\"arrayKey\": [\n\t\t\t{\"testKey1\": \"value1\"},\n\t\t\t{\"testKey2\": \"value2\"},\n\t\t\t{\"key3\": \"value3\"}\n\t\t],\n\t\t\"key4\": {\n\t\t\t\"testKey5\": \"value5\",\n\t\t\t\"key6\": \"value6\"\n\t\t},\n\t\t\"testArrayKey\": [\n\t\t\t{\"testKey7\": \"should not be processed\"}\n\t\t],\n\t\t\"testKey8\": \"value8\"\n\t}`\n\tdsl := fastjson.MustParse(testJSONStr)\n\ttestKeyFilter := func(key string) bool {\n\t\treturn strings.HasPrefix(key, \"test\")\n\t}\n\tprocessedValue := make(map[string]struct{})\n\ttestProcessFunc := func(obj *fastjson.Object, key string, value *fastjson.Value) error {\n\t\ts.Equal(obj.Get(key), value)\n\t\tprocessedValue[value.String()] = struct{}{}\n\t\treturn nil\n\t}\n\ts.NoError(processAllValuesForKey(dsl, testKeyFilter, testProcessFunc))\n\n\texpectedProcessedValue := map[string]struct{}{\n\t\t`\"value1\"`: {},\n\t\t`\"value2\"`: {},\n\t\t`\"value5\"`: {},\n\t\t`[{\"testKey7\":\"should not be processed\"}]`: {},\n\t\t`\"value8\"`: {},\n\t}\n\ts.Equal(expectedProcessedValue, processedValue)\n}\n\nfunc (s *ESVisibilitySuite) TestGetFieldType() {\n\ts.Equal(types.IndexedValueTypeInt, s.visibilityStore.getFieldType(\"StartTime\"))\n\ts.Equal(types.IndexedValueTypeDatetime, s.visibilityStore.getFieldType(\"Attr.CustomDatetimeField\"))\n}\n\nfunc (s *ESVisibilitySuite) TestGetValueOfSearchAfterInJSON() {\n\tv := s.visibilityStore\n\n\t// Int field\n\ttoken := s.getTokenHelper(123)\n\tsortField := definition.CustomIntField\n\tres, err := v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[123, \"t\"]`, res)\n\n\tjsonData := `{\"SortValue\": -9223372036854776000, \"TieBreaker\": \"t\"}`\n\tdec := json.NewDecoder(strings.NewReader(jsonData))\n\tdec.UseNumber()\n\terr = dec.Decode(&token)\n\ts.Nil(err)\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[-9223372036854775808, \"t\"]`, res)\n\n\tjsonData = `{\"SortValue\": 9223372036854776000, \"TieBreaker\": \"t\"}`\n\tdec = json.NewDecoder(strings.NewReader(jsonData))\n\tdec.UseNumber()\n\terr = dec.Decode(&token)\n\ts.Nil(err)\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[9223372036854775807, \"t\"]`, res)\n\n\t// Double field\n\ttoken = s.getTokenHelper(1.11)\n\tsortField = definition.CustomDoubleField\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[1.11, \"t\"]`, res)\n\n\tjsonData = `{\"SortValue\": \"-Infinity\", \"TieBreaker\": \"t\"}`\n\tdec = json.NewDecoder(strings.NewReader(jsonData))\n\tdec.UseNumber()\n\terr = dec.Decode(&token)\n\ts.Nil(err)\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[\"-Infinity\", \"t\"]`, res)\n\n\t// Keyword field\n\ttoken = s.getTokenHelper(\"keyword\")\n\tsortField = definition.CustomKeywordField\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[\"keyword\", \"t\"]`, res)\n\n\ttoken = s.getTokenHelper(nil)\n\tres, err = v.getValueOfSearchAfterInJSON(token, sortField)\n\ts.Nil(err)\n\ts.Equal(`[null, \"t\"]`, res)\n}\n\nfunc (s *ESVisibilitySuite) getTokenHelper(sortValue interface{}) *es.ElasticVisibilityPageToken {\n\ttoken := &es.ElasticVisibilityPageToken{\n\t\tSortValue:  sortValue,\n\t\tTieBreaker: \"t\",\n\t}\n\tencoded, _ := es.SerializePageToken(token) // necessary, otherwise token is fake and not json decoded\n\ttoken, _ = es.DeserializePageToken(encoded)\n\treturn token\n}\n\nfunc (s *ESVisibilitySuite) TestCleanDSL() {\n\t// dsl without `field`\n\tdsl := `{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"2b8344db-0ed6-47a4-92fd-bdeb6ead93e3\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"Attr.CustomIntField\":{\"query\":\"1\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`\n\tres := cleanDSL(dsl)\n\ts.Equal(dsl, res)\n\n\t// dsl with `field`\n\tdsl = `{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"2b8344db-0ed6-47a4-92fd-bdeb6ead93e3\"}}},{\"bool\":{\"must\":[{\"range\":{\"` + \"`Attr.CustomIntField`\" + `\":{\"from\":\"1\",\"to\":\"5\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`\n\tres = cleanDSL(dsl)\n\texpected := `{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"2b8344db-0ed6-47a4-92fd-bdeb6ead93e3\"}}},{\"bool\":{\"must\":[{\"range\":{\"Attr.CustomIntField\":{\"from\":\"1\",\"to\":\"5\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`\n\ts.Equal(expected, res)\n\n\t// dsl with mixed\n\tdsl = `{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"2b8344db-0ed6-47a4-92fd-bdeb6ead93e3\"}}},{\"bool\":{\"must\":[{\"range\":{\"` + \"`Attr.CustomIntField`\" + `\":{\"from\":\"1\",\"to\":\"5\"}}},{\"range\":{\"` + \"`Attr.CustomDoubleField`\" + `\":{\"from\":\"1.0\",\"to\":\"2.0\"}}},{\"range\":{\"StartTime\":{\"gt\":\"0\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`\n\tres = cleanDSL(dsl)\n\texpected = `{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"2b8344db-0ed6-47a4-92fd-bdeb6ead93e3\"}}},{\"bool\":{\"must\":[{\"range\":{\"Attr.CustomIntField\":{\"from\":\"1\",\"to\":\"5\"}}},{\"range\":{\"Attr.CustomDoubleField\":{\"from\":\"1.0\",\"to\":\"2.0\"}}},{\"range\":{\"StartTime\":{\"gt\":\"0\"}}}]}}]}},\"from\":0,\"size\":10,\"sort\":[{\"StartTime\":\"desc\"},{\"RunID\":\"desc\"}]}`\n\ts.Equal(expected, res)\n}\n\nfunc matchQueryData(matchQuery query.MatchQuery) (Source, error) {\n\tsource, err := matchQuery.Source()\n\tif err != nil {\n\t\treturn Source{}, err\n\t}\n\td, err := json.Marshal(source)\n\tif err != nil {\n\t\treturn Source{}, err\n\t}\n\tto := Source{}\n\tif err = json.Unmarshal(d, &to); err != nil {\n\t\treturn Source{}, err\n\t}\n\treturn to, nil\n}\n\nfunc (s *ESVisibilitySuite) TestGetCustomizedDSLFromSQL() {\n\tsql := \"select * from dummy where WorkflowID like 'wid' LIMIT 100\"\n\tdsl, err := getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"wildcard\":{\"WorkflowID\":{\"value\":\"wid*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":100}`, dsl.String())\n\n\tsql = \"select * from dummy where Attr.CustomizedKeywordField like 'test'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"wildcard\":{\"Attr.CustomizedKeywordField\":{\"value\":\"test*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID = 'wid'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where Attr.CustomizedKeywordField = 'test'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"Attr.CustomizedKeywordField\":{\"query\":\"test\"}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID like 'wid' and Attr.CustomizedKeywordField = 'test'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"Attr.CustomizedKeywordField\":{\"query\":\"test\"}}},{\"wildcard\":{\"WorkflowID\":{\"value\":\"wid*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID like 'like' and Attr.CustomizedKeywordField = 'like'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"match_phrase\":{\"Attr.CustomizedKeywordField\":{\"query\":\"like\"}}},{\"wildcard\":{\"WorkflowID\":{\"value\":\"like*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID like 'wid' and Attr.CustomizedKeywordField like 'test'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"must\":[{\"wildcard\":{\"WorkflowID\":{\"value\":\"wid*\",\"case_insensitive\":true}}},{\"wildcard\":{\"Attr.CustomizedKeywordField\":{\"value\":\"test*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID = 'wid' OR RunID = 'runid'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"should\":[{\"match_phrase\":{\"WorkflowID\":{\"query\":\"wid\"}}},{\"match_phrase\":{\"RunID\":{\"query\":\"runid\"}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n\n\tsql = \"select * from dummy where WorkflowID like 'wid' OR Attr.CustomizedKeywordField like 'test'\"\n\tdsl, err = getCustomizedDSLFromSQL(sql, testDomainID)\n\ts.NoError(err)\n\ts.Equal(`{\"query\":{\"bool\":{\"must\":[{\"match_phrase\":{\"DomainID\":{\"query\":\"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"}}},{\"bool\":{\"should\":[{\"wildcard\":{\"WorkflowID\":{\"value\":\"wid*\",\"case_insensitive\":true}}},{\"wildcard\":{\"Attr.CustomizedKeywordField\":{\"value\":\"test*\",\"case_insensitive\":true}}}]}}]}},\"from\":0,\"size\":1}`, dsl.String())\n}\n"
  },
  {
    "path": "common/persistence/errors.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n)\n\ntype (\n\t// TimeoutError is returned when a write operation fails due to a timeout\n\tTimeoutError struct {\n\t\tMsg string\n\t}\n\n\t// DBUnavailableError is returned when the database is unavailable, could be for various reasons.\n\tDBUnavailableError struct {\n\t\tMsg string\n\t}\n\n\t// TransactionSizeLimitError is returned when the transaction size is too large\n\tTransactionSizeLimitError struct {\n\t\tMsg string\n\t}\n\n\t// InvalidPersistenceRequestError represents invalid request to persistence\n\tInvalidPersistenceRequestError struct {\n\t\tMsg string\n\t}\n\n\t// CurrentWorkflowConditionFailedError represents a failed conditional update for current workflow record\n\tCurrentWorkflowConditionFailedError struct {\n\t\tMsg string\n\t}\n\n\t// ConditionFailedError represents a failed conditional update for execution record\n\tConditionFailedError struct {\n\t\tMsg string\n\t}\n\n\t// ShardAlreadyExistError is returned when conditionally creating a shard fails\n\tShardAlreadyExistError struct {\n\t\tMsg string\n\t}\n\n\t// ShardOwnershipLostError is returned when conditional update fails due to RangeID for the shard\n\tShardOwnershipLostError struct {\n\t\tShardID int\n\t\tMsg     string\n\t}\n\n\t// WorkflowExecutionAlreadyStartedError is returned when creating a new workflow failed.\n\tWorkflowExecutionAlreadyStartedError struct {\n\t\tMsg              string\n\t\tStartRequestID   string\n\t\tRunID            string\n\t\tState            int\n\t\tCloseStatus      int\n\t\tLastWriteVersion int64\n\t}\n\n\tDuplicateRequestError struct {\n\t\tRequestType WorkflowRequestType\n\t\tRunID       string\n\t}\n)\n\nfunc (e *DuplicateRequestError) Error() string {\n\treturn fmt.Sprintf(\"Request has already been applied to runID: %s\", e.RunID)\n}\n\nfunc (e *InvalidPersistenceRequestError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *CurrentWorkflowConditionFailedError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *ConditionFailedError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *ShardAlreadyExistError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *ShardOwnershipLostError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *WorkflowExecutionAlreadyStartedError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *TimeoutError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *DBUnavailableError) Error() string {\n\treturn e.Msg\n}\n\nfunc (e *TransactionSizeLimitError) Error() string {\n\treturn e.Msg\n}\n\nfunc AsDuplicateRequestError(err error) (*DuplicateRequestError, bool) {\n\tvar e *DuplicateRequestError\n\tif errors.As(err, &e) {\n\t\treturn e, true\n\t}\n\treturn nil, false\n}\n"
  },
  {
    "path": "common/persistence/errors_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestAsDuplicateRequestError(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\terr         error\n\t\texpectedErr *DuplicateRequestError\n\t\tok          bool\n\t}{\n\t\t{\n\t\t\tname:        \"unwrapped\",\n\t\t\terr:         &DuplicateRequestError{RunID: \"a\"},\n\t\t\texpectedErr: &DuplicateRequestError{RunID: \"a\"},\n\t\t\tok:          true,\n\t\t},\n\t\t{\n\t\t\tname:        \"wrapped\",\n\t\t\terr:         fmt.Errorf(\"%w\", &DuplicateRequestError{RunID: \"b\"}),\n\t\t\texpectedErr: &DuplicateRequestError{RunID: \"b\"},\n\t\t\tok:          true,\n\t\t},\n\t\t{\n\t\t\tname: \"not same type\",\n\t\t\terr:  fmt.Errorf(\"adasdf\"),\n\t\t\tok:   false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\te, ok := AsDuplicateRequestError(tc.err)\n\t\t\tassert.Equal(t, tc.ok, ok)\n\t\t\tif ok {\n\t\t\t\tassert.Equal(t, tc.expectedErr, e)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/execution_manager.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// executionManagerImpl implements ExecutionManager based on ExecutionStore, statsComputer and PayloadSerializer\n\texecutionManagerImpl struct {\n\t\tserializer    PayloadSerializer\n\t\tpersistence   ExecutionStore\n\t\tstatsComputer statsComputer\n\t\tlogger        log.Logger\n\t\ttimeSrc       clock.TimeSource\n\t\tdc            *DynamicConfiguration\n\t}\n)\n\nvar _ ExecutionManager = (*executionManagerImpl)(nil)\n\n// NewExecutionManagerImpl returns new ExecutionManager\nfunc NewExecutionManagerImpl(\n\tpersistence ExecutionStore,\n\tlogger log.Logger,\n\tserializer PayloadSerializer,\n\tdc *DynamicConfiguration,\n) ExecutionManager {\n\treturn &executionManagerImpl{\n\t\tserializer:    serializer,\n\t\tpersistence:   persistence,\n\t\tstatsComputer: statsComputer{},\n\t\tlogger:        logger,\n\t\ttimeSrc:       clock.NewRealTimeSource(),\n\t\tdc:            dc,\n\t}\n}\n\nfunc (m *executionManagerImpl) GetName() string {\n\treturn m.persistence.GetName()\n}\n\nfunc (m *executionManagerImpl) GetShardID() int {\n\treturn m.persistence.GetShardID()\n}\n\n// The below three APIs are related to serialization/deserialization\n\nfunc (m *executionManagerImpl) GetWorkflowExecution(\n\tctx context.Context,\n\trequest *GetWorkflowExecutionRequest,\n) (*GetWorkflowExecutionResponse, error) {\n\n\tinternalRequest := &InternalGetWorkflowExecutionRequest{\n\t\tDomainID:  request.DomainID,\n\t\tExecution: request.Execution,\n\t\tRangeID:   request.RangeID,\n\t}\n\tresponse, err := m.persistence.GetWorkflowExecution(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse := &GetWorkflowExecutionResponse{\n\t\tState: &WorkflowMutableState{\n\t\t\tTimerInfos:         response.State.TimerInfos,\n\t\t\tRequestCancelInfos: response.State.RequestCancelInfos,\n\t\t\tSignalInfos:        response.State.SignalInfos,\n\t\t\tSignalRequestedIDs: response.State.SignalRequestedIDs,\n\t\t\tReplicationState:   response.State.ReplicationState, // TODO: remove this after all 2DC workflows complete\n\t\t\tChecksum:           response.State.Checksum,\n\t\t},\n\t}\n\n\tnewResponse.State.ActivityInfos, err = m.DeserializeActivityInfos(response.State.ActivityInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse.State.ChildExecutionInfos, err = m.DeserializeChildExecutionInfos(response.State.ChildExecutionInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse.State.BufferedEvents, err = m.DeserializeBufferedEvents(response.State.BufferedEvents)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse.State.ExecutionInfo, newResponse.State.ExecutionStats, err = m.DeserializeExecutionInfo(response.State.ExecutionInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tversionHistories, err := m.DeserializeVersionHistories(response.State.VersionHistories)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse.State.VersionHistories = versionHistories\n\tnewResponse.MutableStateStats = m.statsComputer.computeMutableStateStats(response)\n\n\tif len(newResponse.State.Checksum.Value) == 0 {\n\t\tnewResponse.State.Checksum, err = m.serializer.DeserializeChecksum(response.State.ChecksumData)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn newResponse, nil\n}\n\nfunc (m *executionManagerImpl) DeserializeExecutionInfo(\n\tinfo *InternalWorkflowExecutionInfo,\n) (*WorkflowExecutionInfo, *ExecutionStats, error) {\n\n\tcompletionEvent, err := m.serializer.DeserializeEvent(info.CompletionEvent)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tautoResetPoints, err := m.serializer.DeserializeResetPoints(info.AutoResetPoints)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tactiveClusterSelectionPolicy, err := m.serializer.DeserializeActiveClusterSelectionPolicy(info.ActiveClusterSelectionPolicy)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tnewInfo := &WorkflowExecutionInfo{\n\t\tCompletionEvent: completionEvent,\n\n\t\tDomainID:                           info.DomainID,\n\t\tWorkflowID:                         info.WorkflowID,\n\t\tRunID:                              info.RunID,\n\t\tFirstExecutionRunID:                info.FirstExecutionRunID,\n\t\tParentDomainID:                     info.ParentDomainID,\n\t\tParentWorkflowID:                   info.ParentWorkflowID,\n\t\tParentRunID:                        info.ParentRunID,\n\t\tInitiatedID:                        info.InitiatedID,\n\t\tCompletionEventBatchID:             info.CompletionEventBatchID,\n\t\tTaskList:                           info.TaskList,\n\t\tTaskListKind:                       info.TaskListKind,\n\t\tIsCron:                             len(info.CronSchedule) > 0,\n\t\tWorkflowTypeName:                   info.WorkflowTypeName,\n\t\tWorkflowTimeout:                    int32(info.WorkflowTimeout.Seconds()),\n\t\tDecisionStartToCloseTimeout:        int32(info.DecisionStartToCloseTimeout.Seconds()),\n\t\tExecutionContext:                   info.ExecutionContext,\n\t\tState:                              info.State,\n\t\tCloseStatus:                        info.CloseStatus,\n\t\tLastFirstEventID:                   info.LastFirstEventID,\n\t\tLastEventTaskID:                    info.LastEventTaskID,\n\t\tNextEventID:                        info.NextEventID,\n\t\tLastProcessedEvent:                 info.LastProcessedEvent,\n\t\tStartTimestamp:                     info.StartTimestamp,\n\t\tLastUpdatedTimestamp:               info.LastUpdatedTimestamp,\n\t\tCreateRequestID:                    info.CreateRequestID,\n\t\tSignalCount:                        info.SignalCount,\n\t\tDecisionVersion:                    info.DecisionVersion,\n\t\tDecisionScheduleID:                 info.DecisionScheduleID,\n\t\tDecisionStartedID:                  info.DecisionStartedID,\n\t\tDecisionRequestID:                  info.DecisionRequestID,\n\t\tDecisionTimeout:                    int32(info.DecisionTimeout.Seconds()),\n\t\tDecisionAttempt:                    info.DecisionAttempt,\n\t\tDecisionStartedTimestamp:           info.DecisionStartedTimestamp.UnixNano(),\n\t\tDecisionScheduledTimestamp:         info.DecisionScheduledTimestamp.UnixNano(),\n\t\tDecisionOriginalScheduledTimestamp: info.DecisionOriginalScheduledTimestamp.UnixNano(),\n\t\tCancelRequested:                    info.CancelRequested,\n\t\tCancelRequestID:                    info.CancelRequestID,\n\t\tStickyTaskList:                     info.StickyTaskList,\n\t\tStickyScheduleToStartTimeout:       int32(info.StickyScheduleToStartTimeout.Seconds()),\n\t\tClientLibraryVersion:               info.ClientLibraryVersion,\n\t\tClientFeatureVersion:               info.ClientFeatureVersion,\n\t\tClientImpl:                         info.ClientImpl,\n\t\tAttempt:                            info.Attempt,\n\t\tHasRetryPolicy:                     info.HasRetryPolicy,\n\t\tInitialInterval:                    int32(info.InitialInterval.Seconds()),\n\t\tBackoffCoefficient:                 info.BackoffCoefficient,\n\t\tMaximumInterval:                    int32(info.MaximumInterval.Seconds()),\n\t\tExpirationTime:                     info.ExpirationTime,\n\t\tMaximumAttempts:                    info.MaximumAttempts,\n\t\tNonRetriableErrors:                 info.NonRetriableErrors,\n\t\tBranchToken:                        info.BranchToken,\n\t\tCronSchedule:                       info.CronSchedule,\n\t\tCronOverlapPolicy:                  info.CronOverlapPolicy,\n\t\tExpirationSeconds:                  int32(info.ExpirationInterval.Seconds()),\n\t\tAutoResetPoints:                    autoResetPoints,\n\t\tSearchAttributes:                   info.SearchAttributes,\n\t\tMemo:                               info.Memo,\n\t\tPartitionConfig:                    info.PartitionConfig,\n\t\tActiveClusterSelectionPolicy:       activeClusterSelectionPolicy,\n\t}\n\tnewStats := &ExecutionStats{\n\t\tHistorySize: info.HistorySize,\n\t}\n\treturn newInfo, newStats, nil\n}\n\nfunc (m *executionManagerImpl) DeserializeBufferedEvents(\n\tblobs []*DataBlob,\n) ([]*types.HistoryEvent, error) {\n\n\tevents := make([]*types.HistoryEvent, 0)\n\tfor _, b := range blobs {\n\t\thistory, err := m.serializer.DeserializeBatchEvents(b)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tevents = append(events, history...)\n\t}\n\treturn events, nil\n}\n\nfunc (m *executionManagerImpl) DeserializeChildExecutionInfos(\n\tinfos map[int64]*InternalChildExecutionInfo,\n) (map[int64]*ChildExecutionInfo, error) {\n\n\tnewInfos := make(map[int64]*ChildExecutionInfo)\n\tfor k, v := range infos {\n\t\tinitiatedEvent, err := m.serializer.DeserializeEvent(v.InitiatedEvent)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstartedEvent, err := m.serializer.DeserializeEvent(v.StartedEvent)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tc := &ChildExecutionInfo{\n\t\t\tInitiatedEvent: initiatedEvent,\n\t\t\tStartedEvent:   startedEvent,\n\n\t\t\tVersion:               v.Version,\n\t\t\tInitiatedID:           v.InitiatedID,\n\t\t\tInitiatedEventBatchID: v.InitiatedEventBatchID,\n\t\t\tStartedID:             v.StartedID,\n\t\t\tStartedWorkflowID:     v.StartedWorkflowID,\n\t\t\tStartedRunID:          v.StartedRunID,\n\t\t\tCreateRequestID:       v.CreateRequestID,\n\t\t\tDomainID:              v.DomainID,\n\t\t\tDomainNameDEPRECATED:  v.DomainNameDEPRECATED,\n\t\t\tWorkflowTypeName:      v.WorkflowTypeName,\n\t\t\tParentClosePolicy:     v.ParentClosePolicy,\n\t\t}\n\n\t\t// Needed for backward compatibility reason.\n\t\t// ChildWorkflowExecutionStartedEvent was only used by transfer queue processing of StartChildWorkflow.\n\t\t// Updated the code to instead directly read WorkflowId and RunId from mutable state\n\t\t// Existing mutable state won't have those values set so instead use started event to set StartedWorkflowID and\n\t\t// StartedRunID on the mutable state before passing it to application\n\t\tif startedEvent != nil && startedEvent.ChildWorkflowExecutionStartedEventAttributes != nil &&\n\t\t\tstartedEvent.ChildWorkflowExecutionStartedEventAttributes.WorkflowExecution != nil {\n\t\t\tstartedExecution := startedEvent.ChildWorkflowExecutionStartedEventAttributes.WorkflowExecution\n\t\t\tc.StartedWorkflowID = startedExecution.GetWorkflowID()\n\t\t\tc.StartedRunID = startedExecution.GetRunID()\n\t\t}\n\t\tnewInfos[k] = c\n\t}\n\treturn newInfos, nil\n}\n\nfunc (m *executionManagerImpl) DeserializeActivityInfos(\n\tinfos map[int64]*InternalActivityInfo,\n) (map[int64]*ActivityInfo, error) {\n\n\tnewInfos := make(map[int64]*ActivityInfo)\n\tfor k, v := range infos {\n\t\tscheduledEvent, err := m.serializer.DeserializeEvent(v.ScheduledEvent)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstartedEvent, err := m.serializer.DeserializeEvent(v.StartedEvent)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ta := &ActivityInfo{\n\t\t\tScheduledEvent: scheduledEvent,\n\t\t\tStartedEvent:   startedEvent,\n\n\t\t\tVersion:                                 v.Version,\n\t\t\tScheduleID:                              v.ScheduleID,\n\t\t\tScheduledEventBatchID:                   v.ScheduledEventBatchID,\n\t\t\tScheduledTime:                           v.ScheduledTime,\n\t\t\tStartedID:                               v.StartedID,\n\t\t\tStartedTime:                             v.StartedTime,\n\t\t\tActivityID:                              v.ActivityID,\n\t\t\tRequestID:                               v.RequestID,\n\t\t\tDetails:                                 v.Details,\n\t\t\tScheduleToStartTimeout:                  int32(v.ScheduleToStartTimeout.Seconds()),\n\t\t\tScheduleToCloseTimeout:                  int32(v.ScheduleToCloseTimeout.Seconds()),\n\t\t\tStartToCloseTimeout:                     int32(v.StartToCloseTimeout.Seconds()),\n\t\t\tHeartbeatTimeout:                        int32(v.HeartbeatTimeout.Seconds()),\n\t\t\tCancelRequested:                         v.CancelRequested,\n\t\t\tCancelRequestID:                         v.CancelRequestID,\n\t\t\tLastHeartBeatUpdatedTime:                v.LastHeartBeatUpdatedTime,\n\t\t\tTimerTaskStatus:                         v.TimerTaskStatus,\n\t\t\tAttempt:                                 v.Attempt,\n\t\t\tDomainID:                                v.DomainID,\n\t\t\tStartedIdentity:                         v.StartedIdentity,\n\t\t\tTaskList:                                v.TaskList,\n\t\t\tTaskListKind:                            v.TaskListKind,\n\t\t\tHasRetryPolicy:                          v.HasRetryPolicy,\n\t\t\tInitialInterval:                         int32(v.InitialInterval.Seconds()),\n\t\t\tBackoffCoefficient:                      v.BackoffCoefficient,\n\t\t\tMaximumInterval:                         int32(v.MaximumInterval.Seconds()),\n\t\t\tExpirationTime:                          v.ExpirationTime,\n\t\t\tMaximumAttempts:                         v.MaximumAttempts,\n\t\t\tNonRetriableErrors:                      v.NonRetriableErrors,\n\t\t\tLastFailureReason:                       v.LastFailureReason,\n\t\t\tLastWorkerIdentity:                      v.LastWorkerIdentity,\n\t\t\tLastFailureDetails:                      v.LastFailureDetails,\n\t\t\tLastHeartbeatTimeoutVisibilityInSeconds: v.LastHeartbeatTimeoutVisibilityInSeconds,\n\t\t}\n\t\tnewInfos[k] = a\n\t}\n\treturn newInfos, nil\n}\n\nfunc (m *executionManagerImpl) UpdateWorkflowExecution(\n\tctx context.Context,\n\trequest *UpdateWorkflowExecutionRequest,\n) (*UpdateWorkflowExecutionResponse, error) {\n\n\tserializedWorkflowMutation, err := m.SerializeWorkflowMutation(&request.UpdateWorkflowMutation, request.Encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar serializedNewWorkflowSnapshot *InternalWorkflowSnapshot\n\tif request.NewWorkflowSnapshot != nil {\n\t\tserializedNewWorkflowSnapshot, err = m.SerializeWorkflowSnapshot(request.NewWorkflowSnapshot, request.Encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tnewRequest := &InternalUpdateWorkflowExecutionRequest{\n\t\tRangeID: request.RangeID,\n\n\t\tMode: request.Mode,\n\n\t\tUpdateWorkflowMutation: *serializedWorkflowMutation,\n\t\tNewWorkflowSnapshot:    serializedNewWorkflowSnapshot,\n\n\t\tWorkflowRequestMode: request.WorkflowRequestMode,\n\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\tmsuss := m.statsComputer.computeMutableStateUpdateStats(newRequest)\n\terr = m.persistence.UpdateWorkflowExecution(ctx, newRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: msuss}, nil\n}\n\nfunc (m *executionManagerImpl) SerializeUpsertChildExecutionInfos(\n\tinfos []*ChildExecutionInfo,\n\tencoding constants.EncodingType,\n) ([]*InternalChildExecutionInfo, error) {\n\n\tnewInfos := make([]*InternalChildExecutionInfo, 0)\n\tfor _, v := range infos {\n\t\tinitiatedEvent, err := m.serializer.SerializeEvent(v.InitiatedEvent, encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstartedEvent, err := m.serializer.SerializeEvent(v.StartedEvent, encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ti := &InternalChildExecutionInfo{\n\t\t\tInitiatedEvent: initiatedEvent,\n\t\t\tStartedEvent:   startedEvent,\n\n\t\t\tVersion:               v.Version,\n\t\t\tInitiatedID:           v.InitiatedID,\n\t\t\tInitiatedEventBatchID: v.InitiatedEventBatchID,\n\t\t\tCreateRequestID:       v.CreateRequestID,\n\t\t\tStartedID:             v.StartedID,\n\t\t\tStartedWorkflowID:     v.StartedWorkflowID,\n\t\t\tStartedRunID:          v.StartedRunID,\n\t\t\tDomainID:              v.DomainID,\n\t\t\tDomainNameDEPRECATED:  v.DomainNameDEPRECATED,\n\t\t\tWorkflowTypeName:      v.WorkflowTypeName,\n\t\t\tParentClosePolicy:     v.ParentClosePolicy,\n\t\t}\n\t\tnewInfos = append(newInfos, i)\n\t}\n\treturn newInfos, nil\n}\n\nfunc (m *executionManagerImpl) SerializeUpsertActivityInfos(\n\tinfos []*ActivityInfo,\n\tencoding constants.EncodingType,\n) ([]*InternalActivityInfo, error) {\n\n\tnewInfos := make([]*InternalActivityInfo, 0)\n\tfor _, v := range infos {\n\t\tscheduledEvent, err := m.serializer.SerializeEvent(v.ScheduledEvent, encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tstartedEvent, err := m.serializer.SerializeEvent(v.StartedEvent, encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ti := &InternalActivityInfo{\n\t\t\tVersion:                                 v.Version,\n\t\t\tScheduleID:                              v.ScheduleID,\n\t\t\tScheduledEventBatchID:                   v.ScheduledEventBatchID,\n\t\t\tScheduledEvent:                          scheduledEvent,\n\t\t\tScheduledTime:                           v.ScheduledTime,\n\t\t\tStartedID:                               v.StartedID,\n\t\t\tStartedEvent:                            startedEvent,\n\t\t\tStartedTime:                             v.StartedTime,\n\t\t\tActivityID:                              v.ActivityID,\n\t\t\tRequestID:                               v.RequestID,\n\t\t\tDetails:                                 v.Details,\n\t\t\tScheduleToStartTimeout:                  common.SecondsToDuration(int64(v.ScheduleToStartTimeout)),\n\t\t\tScheduleToCloseTimeout:                  common.SecondsToDuration(int64(v.ScheduleToCloseTimeout)),\n\t\t\tStartToCloseTimeout:                     common.SecondsToDuration(int64(v.StartToCloseTimeout)),\n\t\t\tHeartbeatTimeout:                        common.SecondsToDuration(int64(v.HeartbeatTimeout)),\n\t\t\tCancelRequested:                         v.CancelRequested,\n\t\t\tCancelRequestID:                         v.CancelRequestID,\n\t\t\tLastHeartBeatUpdatedTime:                v.LastHeartBeatUpdatedTime,\n\t\t\tTimerTaskStatus:                         v.TimerTaskStatus,\n\t\t\tAttempt:                                 v.Attempt,\n\t\t\tDomainID:                                v.DomainID,\n\t\t\tStartedIdentity:                         v.StartedIdentity,\n\t\t\tTaskList:                                v.TaskList,\n\t\t\tTaskListKind:                            v.TaskListKind,\n\t\t\tHasRetryPolicy:                          v.HasRetryPolicy,\n\t\t\tInitialInterval:                         common.SecondsToDuration(int64(v.InitialInterval)),\n\t\t\tBackoffCoefficient:                      v.BackoffCoefficient,\n\t\t\tMaximumInterval:                         common.SecondsToDuration(int64(v.MaximumInterval)),\n\t\t\tExpirationTime:                          v.ExpirationTime,\n\t\t\tMaximumAttempts:                         v.MaximumAttempts,\n\t\t\tNonRetriableErrors:                      v.NonRetriableErrors,\n\t\t\tLastFailureReason:                       v.LastFailureReason,\n\t\t\tLastWorkerIdentity:                      v.LastWorkerIdentity,\n\t\t\tLastFailureDetails:                      v.LastFailureDetails,\n\t\t\tLastHeartbeatTimeoutVisibilityInSeconds: v.LastHeartbeatTimeoutVisibilityInSeconds,\n\t\t}\n\t\tnewInfos = append(newInfos, i)\n\t}\n\treturn newInfos, nil\n}\n\nfunc (m *executionManagerImpl) SerializeExecutionInfo(\n\tinfo *WorkflowExecutionInfo,\n\tstats *ExecutionStats,\n\tencoding constants.EncodingType,\n) (*InternalWorkflowExecutionInfo, error) {\n\n\tif info == nil {\n\t\treturn &InternalWorkflowExecutionInfo{}, nil\n\t}\n\tcompletionEvent, err := m.serializer.SerializeEvent(info.CompletionEvent, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresetPoints, err := m.serializer.SerializeResetPoints(info.AutoResetPoints, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tactiveClusterSelectionPolicy, err := m.serializer.SerializeActiveClusterSelectionPolicy(info.ActiveClusterSelectionPolicy, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &InternalWorkflowExecutionInfo{\n\t\tDomainID:                           info.DomainID,\n\t\tWorkflowID:                         info.WorkflowID,\n\t\tRunID:                              info.RunID,\n\t\tFirstExecutionRunID:                info.FirstExecutionRunID,\n\t\tParentDomainID:                     info.ParentDomainID,\n\t\tParentWorkflowID:                   info.ParentWorkflowID,\n\t\tParentRunID:                        info.ParentRunID,\n\t\tInitiatedID:                        info.InitiatedID,\n\t\tCompletionEventBatchID:             info.CompletionEventBatchID,\n\t\tCompletionEvent:                    completionEvent,\n\t\tTaskList:                           info.TaskList,\n\t\tTaskListKind:                       info.TaskListKind,\n\t\tWorkflowTypeName:                   info.WorkflowTypeName,\n\t\tWorkflowTimeout:                    common.SecondsToDuration(int64(info.WorkflowTimeout)),\n\t\tDecisionStartToCloseTimeout:        common.SecondsToDuration(int64(info.DecisionStartToCloseTimeout)),\n\t\tExecutionContext:                   info.ExecutionContext,\n\t\tState:                              info.State,\n\t\tCloseStatus:                        info.CloseStatus,\n\t\tLastFirstEventID:                   info.LastFirstEventID,\n\t\tLastEventTaskID:                    info.LastEventTaskID,\n\t\tNextEventID:                        info.NextEventID,\n\t\tLastProcessedEvent:                 info.LastProcessedEvent,\n\t\tStartTimestamp:                     info.StartTimestamp,\n\t\tLastUpdatedTimestamp:               info.LastUpdatedTimestamp,\n\t\tCreateRequestID:                    info.CreateRequestID,\n\t\tSignalCount:                        info.SignalCount,\n\t\tDecisionVersion:                    info.DecisionVersion,\n\t\tDecisionScheduleID:                 info.DecisionScheduleID,\n\t\tDecisionStartedID:                  info.DecisionStartedID,\n\t\tDecisionRequestID:                  info.DecisionRequestID,\n\t\tDecisionTimeout:                    common.SecondsToDuration(int64(info.DecisionTimeout)),\n\t\tDecisionAttempt:                    info.DecisionAttempt,\n\t\tDecisionStartedTimestamp:           time.Unix(0, info.DecisionStartedTimestamp).UTC(),\n\t\tDecisionScheduledTimestamp:         time.Unix(0, info.DecisionScheduledTimestamp).UTC(),\n\t\tDecisionOriginalScheduledTimestamp: time.Unix(0, info.DecisionOriginalScheduledTimestamp).UTC(),\n\t\tCancelRequested:                    info.CancelRequested,\n\t\tCancelRequestID:                    info.CancelRequestID,\n\t\tStickyTaskList:                     info.StickyTaskList,\n\t\tStickyScheduleToStartTimeout:       common.SecondsToDuration(int64(info.StickyScheduleToStartTimeout)),\n\t\tClientLibraryVersion:               info.ClientLibraryVersion,\n\t\tClientFeatureVersion:               info.ClientFeatureVersion,\n\t\tClientImpl:                         info.ClientImpl,\n\t\tAutoResetPoints:                    resetPoints,\n\t\tAttempt:                            info.Attempt,\n\t\tHasRetryPolicy:                     info.HasRetryPolicy,\n\t\tInitialInterval:                    common.SecondsToDuration(int64(info.InitialInterval)),\n\t\tBackoffCoefficient:                 info.BackoffCoefficient,\n\t\tMaximumInterval:                    common.SecondsToDuration(int64(info.MaximumInterval)),\n\t\tExpirationTime:                     info.ExpirationTime,\n\t\tMaximumAttempts:                    info.MaximumAttempts,\n\t\tNonRetriableErrors:                 info.NonRetriableErrors,\n\t\tBranchToken:                        info.BranchToken,\n\t\tCronSchedule:                       info.CronSchedule,\n\t\tExpirationInterval:                 common.SecondsToDuration(int64(info.ExpirationSeconds)),\n\t\tMemo:                               info.Memo,\n\t\tSearchAttributes:                   info.SearchAttributes,\n\t\tPartitionConfig:                    info.PartitionConfig,\n\t\tCronOverlapPolicy:                  info.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:       activeClusterSelectionPolicy,\n\n\t\t// attributes which are not related to mutable state\n\t\tHistorySize: stats.HistorySize,\n\t}, nil\n}\n\nfunc (m *executionManagerImpl) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\trequest *ConflictResolveWorkflowExecutionRequest,\n) (*ConflictResolveWorkflowExecutionResponse, error) {\n\n\tserializedResetWorkflowSnapshot, err := m.SerializeWorkflowSnapshot(&request.ResetWorkflowSnapshot, request.Encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar serializedCurrentWorkflowMutation *InternalWorkflowMutation\n\tif request.CurrentWorkflowMutation != nil {\n\t\tserializedCurrentWorkflowMutation, err = m.SerializeWorkflowMutation(request.CurrentWorkflowMutation, request.Encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tvar serializedNewWorkflowMutation *InternalWorkflowSnapshot\n\tif request.NewWorkflowSnapshot != nil {\n\t\tserializedNewWorkflowMutation, err = m.SerializeWorkflowSnapshot(request.NewWorkflowSnapshot, request.Encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tnewRequest := &InternalConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: request.RangeID,\n\n\t\tMode: request.Mode,\n\n\t\tResetWorkflowSnapshot: *serializedResetWorkflowSnapshot,\n\n\t\tNewWorkflowSnapshot: serializedNewWorkflowMutation,\n\n\t\tCurrentWorkflowMutation: serializedCurrentWorkflowMutation,\n\n\t\tWorkflowRequestMode: request.WorkflowRequestMode,\n\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\tmsuss := m.statsComputer.computeMutableStateConflictResolveStats(newRequest)\n\terr = m.persistence.ConflictResolveWorkflowExecution(ctx, newRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &ConflictResolveWorkflowExecutionResponse{MutableStateUpdateSessionStats: msuss}, nil\n}\n\nfunc (m *executionManagerImpl) CreateWorkflowExecution(\n\tctx context.Context,\n\trequest *CreateWorkflowExecutionRequest,\n) (*CreateWorkflowExecutionResponse, error) {\n\tserializedNewWorkflowSnapshot, err := m.SerializeWorkflowSnapshot(&request.NewWorkflowSnapshot, constants.EncodingType(m.dc.SerializationEncoding()))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnewRequest := &InternalCreateWorkflowExecutionRequest{\n\t\tRangeID: request.RangeID,\n\n\t\tMode: request.Mode,\n\n\t\tPreviousRunID:            request.PreviousRunID,\n\t\tPreviousLastWriteVersion: request.PreviousLastWriteVersion,\n\n\t\tNewWorkflowSnapshot: *serializedNewWorkflowSnapshot,\n\n\t\tWorkflowRequestMode: request.WorkflowRequestMode,\n\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\n\tmsuss := m.statsComputer.computeMutableStateCreateStats(newRequest)\n\t_, err = m.persistence.CreateWorkflowExecution(ctx, newRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &CreateWorkflowExecutionResponse{MutableStateUpdateSessionStats: msuss}, nil\n}\n\nfunc (m *executionManagerImpl) SerializeWorkflowMutation(\n\tinput *WorkflowMutation,\n\tencoding constants.EncodingType,\n) (*InternalWorkflowMutation, error) {\n\n\tserializedExecutionInfo, err := m.SerializeExecutionInfo(\n\t\tinput.ExecutionInfo,\n\t\tinput.ExecutionStats,\n\t\tencoding,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedVersionHistories, err := m.SerializeVersionHistories(input.VersionHistories, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedUpsertActivityInfos, err := m.SerializeUpsertActivityInfos(input.UpsertActivityInfos, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedUpsertChildExecutionInfos, err := m.SerializeUpsertChildExecutionInfos(input.UpsertChildExecutionInfos, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar serializedNewBufferedEvents *DataBlob\n\tif input.NewBufferedEvents != nil {\n\t\tserializedNewBufferedEvents, err = m.serializer.SerializeBatchEvents(input.NewBufferedEvents, encoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tstartVersion, err := getStartVersion(input.VersionHistories)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastWriteVersion, err := getLastWriteVersion(input.VersionHistories)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tchecksumData, err := m.serializer.SerializeChecksum(input.Checksum, constants.EncodingTypeJSON)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &InternalWorkflowMutation{\n\t\tExecutionInfo:    serializedExecutionInfo,\n\t\tVersionHistories: serializedVersionHistories,\n\t\tStartVersion:     startVersion,\n\t\tLastWriteVersion: lastWriteVersion,\n\n\t\tUpsertActivityInfos:       serializedUpsertActivityInfos,\n\t\tDeleteActivityInfos:       input.DeleteActivityInfos,\n\t\tUpsertTimerInfos:          input.UpsertTimerInfos,\n\t\tDeleteTimerInfos:          input.DeleteTimerInfos,\n\t\tUpsertChildExecutionInfos: serializedUpsertChildExecutionInfos,\n\t\tDeleteChildExecutionInfos: input.DeleteChildExecutionInfos,\n\t\tUpsertRequestCancelInfos:  input.UpsertRequestCancelInfos,\n\t\tDeleteRequestCancelInfos:  input.DeleteRequestCancelInfos,\n\t\tUpsertSignalInfos:         input.UpsertSignalInfos,\n\t\tDeleteSignalInfos:         input.DeleteSignalInfos,\n\t\tUpsertSignalRequestedIDs:  input.UpsertSignalRequestedIDs,\n\t\tDeleteSignalRequestedIDs:  input.DeleteSignalRequestedIDs,\n\t\tNewBufferedEvents:         serializedNewBufferedEvents,\n\t\tClearBufferedEvents:       input.ClearBufferedEvents,\n\n\t\tTasksByCategory: input.TasksByCategory,\n\n\t\tWorkflowRequests: input.WorkflowRequests,\n\n\t\tCondition:    input.Condition,\n\t\tChecksum:     input.Checksum,\n\t\tChecksumData: checksumData,\n\t}, nil\n}\n\nfunc (m *executionManagerImpl) SerializeWorkflowSnapshot(\n\tinput *WorkflowSnapshot,\n\tencoding constants.EncodingType,\n) (*InternalWorkflowSnapshot, error) {\n\n\tserializedExecutionInfo, err := m.SerializeExecutionInfo(\n\t\tinput.ExecutionInfo,\n\t\tinput.ExecutionStats,\n\t\tencoding,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedVersionHistories, err := m.SerializeVersionHistories(input.VersionHistories, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedActivityInfos, err := m.SerializeUpsertActivityInfos(input.ActivityInfos, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedChildExecutionInfos, err := m.SerializeUpsertChildExecutionInfos(input.ChildExecutionInfos, encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tstartVersion, err := getStartVersion(input.VersionHistories)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastWriteVersion, err := getLastWriteVersion(input.VersionHistories)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tchecksumData, err := m.serializer.SerializeChecksum(input.Checksum, constants.EncodingTypeJSON)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &InternalWorkflowSnapshot{\n\t\tExecutionInfo:    serializedExecutionInfo,\n\t\tVersionHistories: serializedVersionHistories,\n\t\tStartVersion:     startVersion,\n\t\tLastWriteVersion: lastWriteVersion,\n\n\t\tActivityInfos:       serializedActivityInfos,\n\t\tTimerInfos:          input.TimerInfos,\n\t\tChildExecutionInfos: serializedChildExecutionInfos,\n\t\tRequestCancelInfos:  input.RequestCancelInfos,\n\t\tSignalInfos:         input.SignalInfos,\n\t\tSignalRequestedIDs:  input.SignalRequestedIDs,\n\n\t\tTasksByCategory: input.TasksByCategory,\n\n\t\tWorkflowRequests: input.WorkflowRequests,\n\n\t\tCondition:    input.Condition,\n\t\tChecksum:     input.Checksum,\n\t\tChecksumData: checksumData,\n\t}, nil\n}\n\nfunc (m *executionManagerImpl) SerializeVersionHistories(\n\tversionHistories *VersionHistories,\n\tencoding constants.EncodingType,\n) (*DataBlob, error) {\n\n\tif versionHistories == nil {\n\t\treturn nil, nil\n\t}\n\treturn m.serializer.SerializeVersionHistories(versionHistories.ToInternalType(), encoding)\n}\n\nfunc (m *executionManagerImpl) DeserializeVersionHistories(\n\tblob *DataBlob,\n) (*VersionHistories, error) {\n\n\tif blob == nil {\n\t\treturn nil, nil\n\t}\n\tversionHistories, err := m.serializer.DeserializeVersionHistories(blob)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewVersionHistoriesFromInternalType(versionHistories), nil\n}\n\nfunc (m *executionManagerImpl) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *DeleteWorkflowExecutionRequest,\n) error {\n\treturn m.persistence.DeleteWorkflowExecution(ctx, request)\n}\n\nfunc (m *executionManagerImpl) DeleteCurrentWorkflowExecution(\n\tctx context.Context,\n\trequest *DeleteCurrentWorkflowExecutionRequest,\n) error {\n\treturn m.persistence.DeleteCurrentWorkflowExecution(ctx, request)\n}\n\nfunc (m *executionManagerImpl) GetCurrentExecution(\n\tctx context.Context,\n\trequest *GetCurrentExecutionRequest,\n) (*GetCurrentExecutionResponse, error) {\n\treturn m.persistence.GetCurrentExecution(ctx, request)\n}\n\nfunc (m *executionManagerImpl) ListCurrentExecutions(\n\tctx context.Context,\n\trequest *ListCurrentExecutionsRequest,\n) (*ListCurrentExecutionsResponse, error) {\n\treturn m.persistence.ListCurrentExecutions(ctx, request)\n}\n\nfunc (m *executionManagerImpl) IsWorkflowExecutionExists(\n\tctx context.Context,\n\trequest *IsWorkflowExecutionExistsRequest,\n) (*IsWorkflowExecutionExistsResponse, error) {\n\treturn m.persistence.IsWorkflowExecutionExists(ctx, request)\n}\n\nfunc (m *executionManagerImpl) ListConcreteExecutions(\n\tctx context.Context,\n\trequest *ListConcreteExecutionsRequest,\n) (*ListConcreteExecutionsResponse, error) {\n\tresponse, err := m.persistence.ListConcreteExecutions(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewResponse := &ListConcreteExecutionsResponse{\n\t\tExecutions: make([]*ListConcreteExecutionsEntity, len(response.Executions)),\n\t\tPageToken:  response.NextPageToken,\n\t}\n\tfor i, e := range response.Executions {\n\t\tinfo, _, err := m.DeserializeExecutionInfo(e.ExecutionInfo)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tvh, err := m.DeserializeVersionHistories(e.VersionHistories)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tnewResponse.Executions[i] = &ListConcreteExecutionsEntity{\n\t\t\tExecutionInfo:    info,\n\t\t\tVersionHistories: vh,\n\t\t}\n\t}\n\treturn newResponse, nil\n}\n\nfunc (m *executionManagerImpl) PutReplicationTaskToDLQ(\n\tctx context.Context,\n\trequest *PutReplicationTaskToDLQRequest,\n) error {\n\tinternalRequest := &InternalPutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: request.SourceClusterName,\n\t\tTaskInfo:          m.toInternalReplicationTaskInfo(request.TaskInfo),\n\t}\n\treturn m.persistence.PutReplicationTaskToDLQ(ctx, internalRequest)\n}\n\nfunc (m *executionManagerImpl) GetReplicationTasksFromDLQ(\n\tctx context.Context,\n\trequest *GetReplicationTasksFromDLQRequest,\n) (*GetHistoryTasksResponse, error) {\n\treturn m.persistence.GetReplicationTasksFromDLQ(ctx, request)\n}\n\nfunc (m *executionManagerImpl) GetReplicationDLQSize(\n\tctx context.Context,\n\trequest *GetReplicationDLQSizeRequest,\n) (*GetReplicationDLQSizeResponse, error) {\n\treturn m.persistence.GetReplicationDLQSize(ctx, request)\n}\n\nfunc (m *executionManagerImpl) DeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *DeleteReplicationTaskFromDLQRequest,\n) error {\n\treturn m.persistence.DeleteReplicationTaskFromDLQ(ctx, request)\n}\n\nfunc (m *executionManagerImpl) RangeDeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *RangeDeleteReplicationTaskFromDLQRequest,\n) (*RangeDeleteReplicationTaskFromDLQResponse, error) {\n\treturn m.persistence.RangeDeleteReplicationTaskFromDLQ(ctx, request)\n}\n\nfunc (m *executionManagerImpl) CreateFailoverMarkerTasks(\n\tctx context.Context,\n\trequest *CreateFailoverMarkersRequest,\n) error {\n\trequest.CurrentTimeStamp = m.timeSrc.Now()\n\treturn m.persistence.CreateFailoverMarkerTasks(ctx, request)\n}\n\nfunc (m *executionManagerImpl) GetActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, wfID, rID string,\n) (*types.ActiveClusterSelectionPolicy, error) {\n\tblob, err := m.persistence.GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif blob == nil {\n\t\treturn nil, &types.EntityNotExistsError{\n\t\t\tMessage: \"active cluster selection policy not found\",\n\t\t}\n\t}\n\tpolicy, err := m.serializer.DeserializeActiveClusterSelectionPolicy(blob)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn policy, nil\n}\n\nfunc (m *executionManagerImpl) DeleteActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) error {\n\treturn m.persistence.DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n}\n\nfunc (m *executionManagerImpl) Close() {\n\tm.persistence.Close()\n}\n\nfunc (m *executionManagerImpl) fromInternalReplicationTaskInfos(internalInfos []*InternalReplicationTaskInfo) []*ReplicationTaskInfo {\n\tif internalInfos == nil {\n\t\treturn nil\n\t}\n\tinfos := make([]*ReplicationTaskInfo, len(internalInfos))\n\tfor i := 0; i < len(internalInfos); i++ {\n\t\tinfos[i] = m.fromInternalReplicationTaskInfo(internalInfos[i])\n\t}\n\treturn infos\n}\n\nfunc (m *executionManagerImpl) fromInternalReplicationTaskInfo(internalInfo *InternalReplicationTaskInfo) *ReplicationTaskInfo {\n\tif internalInfo == nil {\n\t\treturn nil\n\t}\n\treturn &ReplicationTaskInfo{\n\t\tDomainID:          internalInfo.DomainID,\n\t\tWorkflowID:        internalInfo.WorkflowID,\n\t\tRunID:             internalInfo.RunID,\n\t\tTaskID:            internalInfo.TaskID,\n\t\tTaskType:          internalInfo.TaskType,\n\t\tFirstEventID:      internalInfo.FirstEventID,\n\t\tNextEventID:       internalInfo.NextEventID,\n\t\tVersion:           internalInfo.Version,\n\t\tScheduledID:       internalInfo.ScheduledID,\n\t\tBranchToken:       internalInfo.BranchToken,\n\t\tNewRunBranchToken: internalInfo.NewRunBranchToken,\n\t\tCreationTime:      internalInfo.CreationTime.UnixNano(),\n\t}\n}\n\nfunc (m *executionManagerImpl) toInternalReplicationTaskInfo(info *ReplicationTaskInfo) *InternalReplicationTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &InternalReplicationTaskInfo{\n\t\tDomainID:          info.DomainID,\n\t\tWorkflowID:        info.WorkflowID,\n\t\tRunID:             info.RunID,\n\t\tTaskID:            info.TaskID,\n\t\tTaskType:          info.TaskType,\n\t\tFirstEventID:      info.FirstEventID,\n\t\tNextEventID:       info.NextEventID,\n\t\tVersion:           info.Version,\n\t\tScheduledID:       info.ScheduledID,\n\t\tBranchToken:       info.BranchToken,\n\t\tNewRunBranchToken: info.NewRunBranchToken,\n\t\tCreationTime:      time.Unix(0, info.CreationTime).UTC(),\n\t\tCurrentTimeStamp:  m.timeSrc.Now(),\n\t}\n}\n\nfunc (m *executionManagerImpl) GetHistoryTasks(\n\tctx context.Context,\n\trequest *GetHistoryTasksRequest,\n) (*GetHistoryTasksResponse, error) {\n\treturn m.persistence.GetHistoryTasks(ctx, request)\n}\n\nfunc (m *executionManagerImpl) CompleteHistoryTask(\n\tctx context.Context,\n\trequest *CompleteHistoryTaskRequest,\n) error {\n\treturn m.persistence.CompleteHistoryTask(ctx, request)\n}\n\nfunc (m *executionManagerImpl) RangeCompleteHistoryTask(\n\tctx context.Context,\n\trequest *RangeCompleteHistoryTaskRequest,\n) (*RangeCompleteHistoryTaskResponse, error) {\n\treturn m.persistence.RangeCompleteHistoryTask(ctx, request)\n}\n\nfunc getStartVersion(\n\tversionHistories *VersionHistories,\n) (int64, error) {\n\n\tif versionHistories == nil {\n\t\treturn constants.EmptyVersion, nil\n\t}\n\n\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tversionHistoryItem, err := versionHistory.GetFirstItem()\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn versionHistoryItem.Version, nil\n}\n\nfunc getLastWriteVersion(\n\tversionHistories *VersionHistories,\n) (int64, error) {\n\n\tif versionHistories == nil {\n\t\treturn constants.EmptyVersion, nil\n\t}\n\n\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tversionHistoryItem, err := versionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn versionHistoryItem.Version, nil\n}\n"
  },
  {
    "path": "common/persistence/execution_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\ttestIndex             = \"test-index\"\n\ttestDomain            = \"test-domain\"\n\ttestDomainID          = \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"\n\ttestPageSize          = 10\n\ttestEarliestTime      = int64(1547596872371000000)\n\ttestLatestTime        = int64(2547596872371000000)\n\ttestWorkflowType      = \"test-wf-type\"\n\ttestWorkflowID        = \"test-wid\"\n\ttestCloseStatus       = int32(1)\n\ttestTableName         = \"test-table-name\"\n\ttestRunID             = \"test-run-id\"\n\ttestSearchAttributes1 = map[string]interface{}{\"TestAttr1\": \"val1\", \"TestAttr2\": 2, \"TestAttr3\": false}\n\ttestSearchAttributes2 = map[string]interface{}{\"TestAttr1\": \"val2\", \"TestAttr2\": 2, \"TestAttr3\": false}\n\ttestSearchAttributes3 = map[string]interface{}{\"TestAttr2\": 2, \"TestAttr3\": false}\n)\n\nfunc TestExecutionManager_ProxyStoreMethods(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tmethod       string\n\t\tprepareMocks func(*MockExecutionStore)\n\t}{\n\t\t{\n\t\t\tmethod: \"GetShardID\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().GetShardID().Return(1).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"GetName\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().GetName().Return(\"test\").Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"Close\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().Close().Return().Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"GetHistoryTasks\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"RangeCompleteHistoryTask\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"DeleteReplicationTaskFromDLQ\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().DeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"RangeDeleteReplicationTaskFromDLQ\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().RangeDeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"DeleteWorkflowExecution\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"DeleteCurrentWorkflowExecution\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"GetCurrentExecution\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"ListCurrentExecutions\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"IsWorkflowExecutionExists\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmethod: \"GetReplicationDLQSize\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().GetReplicationDLQSize(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.method, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\ttc.prepareMocks(mockedStore)\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), nil, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tv := reflect.ValueOf(manager)\n\t\t\tmethod := v.MethodByName(tc.method)\n\t\t\tmethodType := method.Type()\n\t\t\targs := methodType.NumIn()\n\t\t\tvar vals []reflect.Value\n\t\t\t// If a method requires arguments, we expect the first argument to be a context\n\t\t\t// and the rest to be zero values of the correct type.\n\t\t\t// For methods like Close and GetShardID, we don't expect any arguments.\n\t\t\tif args > 0 {\n\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\tfor i := 1; i < args; i++ {\n\t\t\t\t\tvals = append(vals, reflect.Zero(methodType.In(i)))\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tcallRes := method.Call(vals)\n\t\t\tif callRes == nil {\n\t\t\t\treturn\n\t\t\t}\n\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\t\t\terr, ok := resultErr.(error)\n\t\t\tif ok {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestExecutionManager_GetWorkflowExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\trequest := &GetWorkflowExecutionRequest{\n\t\tDomainID: testDomainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tRangeID: 1,\n\t}\n\n\tactivityOne := sampleInternalActivityInfo(\"activity1\")\n\tactivityTwo := sampleInternalActivityInfo(\"activity2\")\n\n\twfCompletionEvent := NewDataBlob([]byte(\"wf-event\"), constants.EncodingTypeThriftRW)\n\twfCompletionEventData := generateTestHistoryEvent(99)\n\n\twfInfo := sampleInternalWorkflowExecutionInfo()\n\twfInfo.CompletionEvent = wfCompletionEvent\n\twfInfo.AutoResetPoints = NewDataBlob([]byte(\"test-reset-points\"), constants.EncodingTypeThriftRW)\n\n\tmockedStore.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(&InternalGetWorkflowExecutionResponse{\n\t\tState: &InternalWorkflowMutableState{\n\t\t\tExecutionInfo: wfInfo,\n\t\t\tActivityInfos: map[int64]*InternalActivityInfo{\n\t\t\t\t1: activityOne,\n\t\t\t\t2: activityTwo,\n\t\t\t},\n\t\t\tTimerInfos: map[string]*TimerInfo{\n\t\t\t\t\"test-timer\": {\n\t\t\t\t\tVersion: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil)\n\n\tmockedSerializer.EXPECT().DeserializeEvent(activityOne.ScheduledEvent).Return(&types.HistoryEvent{\n\t\tID: 1,\n\t}, nil).Times(1)\n\tmockedSerializer.EXPECT().DeserializeEvent(activityOne.StartedEvent).Return(&types.HistoryEvent{\n\t\tID: 1,\n\t}, nil).Times(1)\n\n\tmockedSerializer.EXPECT().DeserializeEvent(activityTwo.ScheduledEvent).Return(&types.HistoryEvent{\n\t\tID: 2,\n\t}, nil).Times(1)\n\tmockedSerializer.EXPECT().DeserializeEvent(activityTwo.StartedEvent).Return(&types.HistoryEvent{\n\t\tID: 2,\n\t}, nil).Times(1)\n\n\tmockedSerializer.EXPECT().DeserializeEvent(wfCompletionEvent).Return(wfCompletionEventData, nil).Times(1)\n\tmockedSerializer.EXPECT().DeserializeResetPoints(gomock.Any()).Return(&types.ResetPoints{}, nil).Times(1)\n\tmockedSerializer.EXPECT().DeserializeChecksum(gomock.Any()).Return(checksum.Checksum{}, nil).Times(1)\n\n\tactiveClusterSelPlcyData := sampleActiveClusterSelectionPolicyData()\n\tmockedSerializer.EXPECT().DeserializeActiveClusterSelectionPolicy(activeClusterSelPlcyData).Return(generateActiveClusterSelectionPolicy(), nil).Times(1)\n\n\tres, err := manager.GetWorkflowExecution(context.Background(), request)\n\tassert.NoError(t, err)\n\n\texpectedExecutionInfo := sampleWorkflowExecutionInfo()\n\texpectedExecutionInfo.CompletionEvent = wfCompletionEventData\n\texpectedExecutionInfo.AutoResetPoints = &types.ResetPoints{}\n\n\tassert.Equal(t, &WorkflowMutableState{\n\t\tExecutionInfo:       expectedExecutionInfo,\n\t\tChildExecutionInfos: make(map[int64]*ChildExecutionInfo),\n\t\tActivityInfos: map[int64]*ActivityInfo{\n\t\t\t1: sampleActivityInfo(\"activity1\", 1),\n\t\t\t2: sampleActivityInfo(\"activity2\", 2),\n\t\t},\n\t\tTimerInfos: map[string]*TimerInfo{\n\t\t\t\"test-timer\": {\n\t\t\t\tVersion: 1,\n\t\t\t},\n\t\t},\n\t\tExecutionStats: &ExecutionStats{\n\t\t\tHistorySize: 1024,\n\t\t},\n\t\tBufferedEvents: make([]*types.HistoryEvent, 0),\n\t}, res.State)\n\t// Expectations for the deserialization of activity events.\n\tassert.Equal(t, &MutableStateStats{MutableStateSize: 170, ExecutionInfoSize: 20, ActivityInfoSize: 150, TimerInfoSize: 0, ChildInfoSize: 0, SignalInfoSize: 0, BufferedEventsSize: 0, ActivityInfoCount: 2, TimerInfoCount: 1, ChildInfoCount: 0, SignalInfoCount: 0, RequestCancelInfoCount: 0, BufferedEventsCount: 0}, res.MutableStateStats)\n}\n\nfunc TestExecutionManager_GetWorkflowExecution_NoWorkflow(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\trequest := &GetWorkflowExecutionRequest{\n\t\tDomainID: \"testDomain\",\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"nonexistentWorkflow\",\n\t\t\tRunID:      \"nonexistentRunID\",\n\t\t},\n\t\tRangeID: 1,\n\t}\n\n\tmockedStore.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{})\n\n\t_, err := manager.GetWorkflowExecution(context.Background(), request)\n\tassert.Error(t, err)\n\tassert.IsType(t, &types.EntityNotExistsError{}, err)\n}\n\nfunc TestExecutionManager_UpdateWorkflowExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\texpectedInfo := sampleInternalWorkflowMutation()\n\n\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(expectedInfo.ExecutionInfo.CompletionEvent, nil).Times(2)\n\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\tmockedSerializer.EXPECT().SerializeResetPoints(generateResetPoints(), constants.EncodingTypeThriftRW).Return(expectedInfo.ExecutionInfo.AutoResetPoints, nil).Times(2)\n\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(2)\n\n\trequest := &UpdateWorkflowExecutionRequest{\n\t\tRangeID:                1,\n\t\tMode:                   UpdateWorkflowModeBypassCurrent,\n\t\tUpdateWorkflowMutation: *sampleWorkflowMutation(),\n\t\tEncoding:               constants.EncodingTypeThriftRW,\n\t\tNewWorkflowSnapshot: &WorkflowSnapshot{\n\t\t\tExecutionInfo: sampleWorkflowExecutionInfo(),\n\t\t\tExecutionStats: &ExecutionStats{\n\t\t\t\tHistorySize: 1024,\n\t\t\t},\n\t\t\tChecksum: generateChecksum(),\n\t\t},\n\t}\n\n\tmockedSerializer.EXPECT().SerializeChecksum(request.UpdateWorkflowMutation.Checksum, constants.EncodingTypeJSON).Return(expectedInfo.ChecksumData, nil).Times(2)\n\n\texpectedRequest := &InternalUpdateWorkflowExecutionRequest{\n\t\tRangeID:                1,\n\t\tMode:                   UpdateWorkflowModeBypassCurrent,\n\t\tUpdateWorkflowMutation: *expectedInfo,\n\t\tNewWorkflowSnapshot: &InternalWorkflowSnapshot{\n\t\t\tExecutionInfo: expectedInfo.ExecutionInfo,\n\t\t},\n\t}\n\tmockedStore.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, req *InternalUpdateWorkflowExecutionRequest) error {\n\t\tassert.Equal(t, expectedRequest.UpdateWorkflowMutation, req.UpdateWorkflowMutation)\n\t\treturn nil\n\t})\n\n\tres, err := manager.UpdateWorkflowExecution(context.Background(), request)\n\tassert.NoError(t, err)\n\tstats := &MutableStateUpdateSessionStats{\n\t\tMutableStateSize:    90,\n\t\tExecutionInfoSize:   40,\n\t\tActivityInfoSize:    20,\n\t\tTimerInfoSize:       10,\n\t\tChildInfoSize:       20,\n\t\tActivityInfoCount:   1,\n\t\tTimerInfoCount:      1,\n\t\tChildInfoCount:      1,\n\t\tTaskCountByCategory: map[HistoryTaskCategory]int{},\n\t}\n\tassert.Equal(t, stats, res.MutableStateUpdateSessionStats)\n}\n\nfunc TestSerializeWorkflowSnapshot(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockPayloadSerializer)\n\t\tinput        *WorkflowSnapshot\n\t\tcheckRes     func(*testing.T, *InternalWorkflowSnapshot, error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, sampleInternalWorkflowSnapshot(), res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"nil info\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleTestCheckSumData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: &WorkflowSnapshot{},\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, &InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo:       &InternalWorkflowExecutionInfo{},\n\t\t\t\t\tChecksumData:        sampleTestCheckSumData(),\n\t\t\t\t\tStartVersion:        constants.EmptyVersion,\n\t\t\t\t\tLastWriteVersion:    constants.EmptyVersion,\n\t\t\t\t\tActivityInfos:       make([]*InternalActivityInfo, 0),\n\t\t\t\t\tChildExecutionInfos: make([]*InternalChildExecutionInfo, 0),\n\t\t\t\t}, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(gomock.Any(), gomock.Any()).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize points error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize version histories error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(NewDataBlob([]byte(\"test-reset-points\"), constants.EncodingTypeThriftRW), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(nil, assert.AnError).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize activity scheduled event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleTestCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize activity started event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleTestCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize child workflow scheduled event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleTestCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize child workflow started event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleTestCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tinput: sampleWorkflowSnapshot(),\n\t\t\tcheckRes: func(t *testing.T, res *InternalWorkflowSnapshot, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\t\t\ttc.prepareMocks(mockedSerializer)\n\t\t\tmanager := NewExecutionManagerImpl(nil, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t}).(*executionManagerImpl)\n\t\t\tres, err := manager.SerializeWorkflowSnapshot(tc.input, constants.EncodingTypeThriftRW)\n\t\t\ttc.checkRes(t, res, err)\n\t\t})\n\t}\n}\n\nfunc TestDeserializeBufferedEvents(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockPayloadSerializer)\n\t\tcheckRes     func(*testing.T, []*types.HistoryEvent, error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\teventCounter := 0\n\t\t\t\tmockedSerializer.EXPECT().DeserializeBatchEvents(gomock.Any()).DoAndReturn(func(data *DataBlob) ([]*types.HistoryEvent, error) {\n\t\t\t\t\tres := []*types.HistoryEvent{{ID: int64(eventCounter)}, {ID: int64(eventCounter + 1)}}\n\t\t\t\t\teventCounter += 2\n\t\t\t\t\treturn res, nil\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, events []*types.HistoryEvent, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, []*types.HistoryEvent{{ID: 0}, {ID: 1}, {ID: 2}, {ID: 3}}, events)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().DeserializeBatchEvents(gomock.Any()).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, res []*types.HistoryEvent, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\t\t\ttc.prepareMocks(mockedSerializer)\n\n\t\t\tmanager := NewExecutionManagerImpl(nil, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t}).(*executionManagerImpl)\n\n\t\t\tevents := []*DataBlob{\n\t\t\t\tsampleEventData(),\n\t\t\t\tsampleEventData(),\n\t\t\t}\n\n\t\t\tres, err := manager.DeserializeBufferedEvents(events)\n\t\t\ttc.checkRes(t, res, err)\n\t\t})\n\t}\n}\n\nfunc TestPutReplicationTaskToDLQ(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), nil, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\tnow := time.Now().UTC()\n\n\ttask := &PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: \"test-cluster\",\n\t\tTaskInfo: &ReplicationTaskInfo{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowID:   testWorkflowID,\n\t\t\tCreationTime: now.UnixNano(),\n\t\t},\n\t\tDomainName: testDomain,\n\t}\n\n\tmockedStore.EXPECT().PutReplicationTaskToDLQ(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, req *InternalPutReplicationTaskToDLQRequest) error {\n\t\tassert.Equal(t, \"test-cluster\", req.SourceClusterName)\n\t\tassert.Equal(t, testDomainID, req.TaskInfo.DomainID)\n\t\tassert.Equal(t, testWorkflowID, req.TaskInfo.WorkflowID)\n\t\tassert.Equal(t, now, req.TaskInfo.CreationTime)\n\n\t\tassert.WithinDuration(t, now, req.TaskInfo.CurrentTimeStamp, time.Second)\n\t\treturn nil\n\t})\n\n\terr := manager.PutReplicationTaskToDLQ(context.Background(), task)\n\tassert.NoError(t, err)\n}\n\nfunc TestGetReplicationTasksFromDLQ(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), nil, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\trequest := &GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: \"test-cluster\",\n\t\tReadLevel:         1,\n\t\tMaxReadLevel:      2,\n\t\tBatchSize:         10,\n\t\tNextPageToken:     nil,\n\t}\n\n\tnow := time.Now().UTC().Round(time.Second)\n\n\tmockedStore.EXPECT().GetReplicationTasksFromDLQ(gomock.Any(), request).Return(\n\t\t&GetHistoryTasksResponse{\n\t\t\tTasks: []Task{\n\t\t\t\t&HistoryReplicationTask{\n\t\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte(\"test-token\"),\n\t\t}, nil)\n\n\tres, err := manager.GetReplicationTasksFromDLQ(context.Background(), request)\n\tassert.NoError(t, err)\n\tassert.Equal(t, &GetHistoryTasksResponse{\n\t\tTasks: []Task{\n\t\t\t&HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"test-token\"),\n\t}, res)\n}\n\nfunc TestDeserializeChildExecutionInfos(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockPayloadSerializer)\n\t\tcheckRes     func(*testing.T, map[int64]*ChildExecutionInfo, error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\t// Child 1 init event\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(sampleEventDataWithVersion(1)).Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}, nil)\n\t\t\t\t// Child 1 start event\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(sampleEventDataWithVersion(2)).Return(&types.HistoryEvent{\n\t\t\t\t\tID:      2,\n\t\t\t\t\tVersion: 2,\n\t\t\t\t}, nil)\n\t\t\t\t// Child 2 init event\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(sampleEventDataWithVersion(3)).Return(&types.HistoryEvent{\n\t\t\t\t\tID:      3,\n\t\t\t\t\tVersion: 3,\n\t\t\t\t}, nil)\n\n\t\t\t\t// Child 2 start event is mimicking legacy behavior where runID and workflowID were not stored inside the info\n\t\t\t\t// but was extracted from the startEvent\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(sampleEventDataWithVersion(4)).Return(&types.HistoryEvent{\n\t\t\t\t\tID:      4,\n\t\t\t\t\tVersion: 4,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"legacy-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"legacy-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, events map[int64]*ChildExecutionInfo, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, map[int64]*ChildExecutionInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:           1,\n\t\t\t\t\t\tDomainID:          testDomainID,\n\t\t\t\t\t\tWorkflowTypeName:  testWorkflowType,\n\t\t\t\t\t\tInitiatedID:       1,\n\t\t\t\t\t\tInitiatedEvent:    &types.HistoryEvent{ID: 1, Version: 1},\n\t\t\t\t\t\tStartedID:         2,\n\t\t\t\t\t\tStartedEvent:      &types.HistoryEvent{ID: 2, Version: 2},\n\t\t\t\t\t\tCreateRequestID:   \"create-request-id\",\n\t\t\t\t\t\tStartedWorkflowID: \"workflow-id\",\n\t\t\t\t\t\tStartedRunID:      \"run-id\",\n\t\t\t\t\t},\n\t\t\t\t\t2: {\n\t\t\t\t\t\tVersion:          3,\n\t\t\t\t\t\tDomainID:         testDomainID,\n\t\t\t\t\t\tWorkflowTypeName: testWorkflowType,\n\t\t\t\t\t\tInitiatedID:      3,\n\t\t\t\t\t\tInitiatedEvent:   &types.HistoryEvent{ID: 3, Version: 3},\n\t\t\t\t\t\tStartedID:        4,\n\t\t\t\t\t\tStartedEvent: &types.HistoryEvent{ID: 4, Version: 4,\n\t\t\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\t\tWorkflowID: \"legacy-workflow-id\",\n\t\t\t\t\t\t\t\t\tRunID:      \"legacy-run-id\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t}},\n\t\t\t\t\t\tCreateRequestID:   \"create-request-id\",\n\t\t\t\t\t\tStartedWorkflowID: \"legacy-workflow-id\",\n\t\t\t\t\t\tStartedRunID:      \"legacy-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}, events)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"deserialize event error\",\n\t\t\tprepareMocks: func(mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(gomock.Any()).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, events map[int64]*ChildExecutionInfo, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\t\t\ttest.prepareMocks(mockedSerializer)\n\n\t\t\tmanager := &executionManagerImpl{\n\t\t\t\tserializer: mockedSerializer,\n\t\t\t}\n\n\t\t\tresult, err := manager.DeserializeChildExecutionInfos(map[int64]*InternalChildExecutionInfo{\n\t\t\t\t1: sampleInternalChildExecutionInfo(1, 2),\n\t\t\t\t2: sampleInternalChildExecutionInfo(3, 4),\n\t\t\t})\n\t\t\ttest.checkRes(t, result, err)\n\t\t})\n\t}\n}\n\nfunc TestListConcreteExecutions(t *testing.T) {\n\trequest := &ListConcreteExecutionsRequest{\n\t\tPageSize:  10,\n\t\tPageToken: []byte(\"next\"),\n\t}\n\n\tinternalResponse := &InternalListConcreteExecutionsResponse{\n\t\tExecutions: []*InternalListConcreteExecutionsEntity{\n\t\t\t{\n\t\t\t\tExecutionInfo:    sampleInternalWorkflowExecutionInfo(),\n\t\t\t\tVersionHistories: sampleEventData(),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"next\"),\n\t}\n\n\ttestCases := []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockExecutionStore, *MockPayloadSerializer)\n\t\tcheckRes     func(*testing.T, *ListConcreteExecutionsResponse, error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedStore.EXPECT().ListConcreteExecutions(gomock.Any(), request).Return(internalResponse, nil)\n\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(internalResponse.Executions[0].ExecutionInfo.CompletionEvent).Return(completionEvent(), nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeResetPoints(internalResponse.Executions[0].ExecutionInfo.AutoResetPoints).Return(&types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRunID: testRunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeActiveClusterSelectionPolicy(internalResponse.Executions[0].ExecutionInfo.ActiveClusterSelectionPolicy).Return(generateActiveClusterSelectionPolicy(), nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeVersionHistories(internalResponse.Executions[0].VersionHistories).Return(&types.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories: []*types.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token-1\"),\n\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token-2\"),\n\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 3,\n\t\t\t\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ListConcreteExecutionsResponse, err error) {\n\t\t\t\texecutionInfo := sampleWorkflowExecutionInfo()\n\t\t\t\texecutionInfo.CompletionEvent = completionEvent()\n\t\t\t\texecutionInfo.AutoResetPoints = &types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRunID: testRunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, &ListConcreteExecutionsResponse{\n\t\t\t\t\tExecutions: []*ListConcreteExecutionsEntity{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tExecutionInfo: executionInfo,\n\t\t\t\t\t\t\tVersionHistories: &VersionHistories{\n\t\t\t\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\t\t\t\tHistories: []*VersionHistory{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token-1\"),\n\t\t\t\t\t\t\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token-2\"),\n\t\t\t\t\t\t\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\t\tEventID: 3,\n\t\t\t\t\t\t\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPageToken: []byte(\"next\"),\n\t\t\t\t}, response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedStore.EXPECT().ListConcreteExecutions(gomock.Any(), request).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ListConcreteExecutionsResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"deserialize execution info error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedStore.EXPECT().ListConcreteExecutions(gomock.Any(), request).Return(internalResponse, nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeEvent(internalResponse.Executions[0].ExecutionInfo.CompletionEvent).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ListConcreteExecutionsResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\t\t\ttc.prepareMocks(mockedStore, mockedSerializer)\n\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\tres, err := manager.ListConcreteExecutions(context.Background(), request)\n\n\t\t\ttc.checkRes(t, res, err)\n\t\t})\n\t}\n}\n\nfunc TestCreateWorkflowExecution(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockExecutionStore, *MockPayloadSerializer)\n\t\tcheckRes     func(*testing.T, *CreateWorkflowExecutionResponse, error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\t// Prepare CreateWorkflow call\n\t\t\t\texpectedRequest := &InternalCreateWorkflowExecutionRequest{\n\t\t\t\t\tRangeID:                  1,\n\t\t\t\t\tMode:                     CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\t\tPreviousRunID:            testRunID,\n\t\t\t\t\tPreviousLastWriteVersion: 1,\n\t\t\t\t\tNewWorkflowSnapshot:      *sampleInternalWorkflowSnapshot(),\n\t\t\t\t\tWorkflowRequestMode:      CreateWorkflowRequestModeReplicated,\n\t\t\t\t\tCurrentTimeStamp:         time.Now(),\n\t\t\t\t}\n\t\t\t\tmockedStore.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, actualRequest *InternalCreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.RangeID, actualRequest.RangeID)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.Mode, actualRequest.Mode)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.PreviousRunID, actualRequest.PreviousRunID)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.PreviousLastWriteVersion, actualRequest.PreviousLastWriteVersion)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.WorkflowRequestMode, actualRequest.WorkflowRequestMode)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.NewWorkflowSnapshot, actualRequest.NewWorkflowSnapshot)\n\n\t\t\t\t\t\tassert.WithinDuration(t, expectedRequest.CurrentTimeStamp, actualRequest.CurrentTimeStamp, time.Second)\n\t\t\t\t\t\treturn nil, nil\n\t\t\t\t\t})\n\n\t\t\t\t// Prepare DeserializeWorkflow call\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *CreateWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.Equal(t, &CreateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize:    91,\n\t\t\t\t\t\tExecutionInfoSize:   20,\n\t\t\t\t\t\tActivityInfoSize:    29,\n\t\t\t\t\t\tTimerInfoSize:       22,\n\t\t\t\t\t\tChildInfoSize:       20,\n\t\t\t\t\t\tActivityInfoCount:   1,\n\t\t\t\t\t\tTimerInfoCount:      2,\n\t\t\t\t\t\tChildInfoCount:      1,\n\t\t\t\t\t\tTaskCountByCategory: map[HistoryTaskCategory]int{},\n\t\t\t\t\t},\n\t\t\t\t}, response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\t// Prepare DeserializeWorkflow call\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\n\t\t\t\t// Persistence call will fail\n\t\t\t\tmockedStore.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *CreateWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"serialize workflow snapshot error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *CreateWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\t\t\ttc.prepareMocks(mockedStore, mockedSerializer)\n\n\t\t\trequest := &CreateWorkflowExecutionRequest{\n\t\t\t\tRangeID:                  1,\n\t\t\t\tMode:                     CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\tPreviousRunID:            testRunID,\n\t\t\t\tPreviousLastWriteVersion: 1,\n\t\t\t\tNewWorkflowSnapshot:      *sampleWorkflowSnapshot(),\n\t\t\t\tWorkflowRequestMode:      CreateWorkflowRequestModeReplicated,\n\t\t\t}\n\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\tres, err := manager.CreateWorkflowExecution(context.Background(), request)\n\n\t\t\ttc.checkRes(t, res, err)\n\t\t})\n\t}\n}\n\nfunc TestConflictResolveWorkflowExecution(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname         string\n\t\trequest      *ConflictResolveWorkflowExecutionRequest\n\t\tprepareMocks func(*MockExecutionStore, *MockPayloadSerializer)\n\t\tcheckRes     func(*testing.T, *ConflictResolveWorkflowExecutionResponse, error)\n\t}{\n\t\t{\n\t\t\tname: \"only snapshot\",\n\t\t\trequest: &ConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID:               1,\n\t\t\t\tMode:                  ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: *sampleWorkflowSnapshot(),\n\t\t\t\tEncoding:              constants.EncodingTypeThriftRW,\n\t\t\t},\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\texpectedRequest := &InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\t\tRangeID:               1,\n\t\t\t\t\tMode:                  ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\t\tResetWorkflowSnapshot: *sampleInternalWorkflowSnapshot(),\n\t\t\t\t\tCurrentTimeStamp:      time.Now(),\n\t\t\t\t}\n\t\t\t\tmockedStore.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(_ context.Context, actualRequest *InternalConflictResolveWorkflowExecutionRequest) error {\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.RangeID, actualRequest.RangeID)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.Mode, actualRequest.Mode)\n\t\t\t\t\t\tassert.Equal(t, expectedRequest.ResetWorkflowSnapshot, actualRequest.ResetWorkflowSnapshot)\n\n\t\t\t\t\t\tassert.WithinDuration(t, expectedRequest.CurrentTimeStamp, actualRequest.CurrentTimeStamp, time.Second)\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t})\n\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleCheckSumData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, &ConflictResolveWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize:    91,\n\t\t\t\t\t\tExecutionInfoSize:   20,\n\t\t\t\t\t\tActivityInfoSize:    29,\n\t\t\t\t\t\tTimerInfoSize:       22,\n\t\t\t\t\t\tChildInfoSize:       20,\n\t\t\t\t\t\tActivityInfoCount:   1,\n\t\t\t\t\t\tTimerInfoCount:      2,\n\t\t\t\t\t\tChildInfoCount:      1,\n\t\t\t\t\t\tTaskCountByCategory: map[HistoryTaskCategory]int{},\n\t\t\t\t\t},\n\t\t\t\t}, response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"only snapshot fail\",\n\t\t\trequest: &ConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID:               1,\n\t\t\t\tMode:                  ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: *sampleWorkflowSnapshot(),\n\t\t\t\tEncoding:              constants.EncodingTypeThriftRW,\n\t\t\t},\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"current workflow mutation\",\n\t\t\trequest: &ConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID:                 1,\n\t\t\t\tMode:                    ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot:   *sampleWorkflowSnapshot(),\n\t\t\t\tEncoding:                constants.EncodingTypeThriftRW,\n\t\t\t\tCurrentWorkflowMutation: sampleWorkflowMutation(),\n\t\t\t},\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\texpectedRequest := &InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\t\tRangeID:                 1,\n\t\t\t\t\tMode:                    ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\t\tResetWorkflowSnapshot:   *sampleInternalWorkflowSnapshot(),\n\t\t\t\t\tCurrentWorkflowMutation: sampleInternalWorkflowMutation(),\n\t\t\t\t\tCurrentTimeStamp:        time.Now(),\n\t\t\t\t}\n\n\t\t\t\tmockedStore.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, actualRequest *InternalConflictResolveWorkflowExecutionRequest) error {\n\t\t\t\t\tassert.Equal(t, expectedRequest.RangeID, actualRequest.RangeID)\n\t\t\t\t\tassert.Equal(t, expectedRequest.Mode, actualRequest.Mode)\n\t\t\t\t\tassert.Equal(t, expectedRequest.ResetWorkflowSnapshot, actualRequest.ResetWorkflowSnapshot)\n\n\t\t\t\t\tassert.WithinDuration(t, expectedRequest.CurrentTimeStamp, actualRequest.CurrentTimeStamp, time.Second)\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\n\t\t\t\t// Mutation call\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\n\t\t\t\t// Expect mutation doubles the calls for workflow execution serialization\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(2)\n\t\t\t\tmockedSerializer.EXPECT().SerializeResetPoints(gomock.Any(), gomock.Any()).Return(sampleResetPointsData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeVersionHistories(gomock.Any(), gomock.Any()).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(2)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(activityStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(2)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowScheduledEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(childWorkflowStartedEvent(), constants.EncodingTypeThriftRW).Return(sampleEventData(), nil).Times(1)\n\t\t\t\tmockedSerializer.EXPECT().SerializeChecksum(gomock.Any(), gomock.Any()).Return(sampleCheckSumData(), nil).Times(2)\n\t\t\t\tmockedSerializer.EXPECT().SerializeActiveClusterSelectionPolicy(generateActiveClusterSelectionPolicy(), constants.EncodingTypeThriftRW).Return(sampleActiveClusterSelectionPolicyData(), nil).Times(2)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, &ConflictResolveWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize:    161,\n\t\t\t\t\t\tExecutionInfoSize:   40,\n\t\t\t\t\t\tActivityInfoSize:    49,\n\t\t\t\t\t\tTimerInfoSize:       32,\n\t\t\t\t\t\tChildInfoSize:       40,\n\t\t\t\t\t\tActivityInfoCount:   2,\n\t\t\t\t\t\tTimerInfoCount:      3,\n\t\t\t\t\t\tChildInfoCount:      2,\n\t\t\t\t\t\tTaskCountByCategory: map[HistoryTaskCategory]int{},\n\t\t\t\t\t},\n\t\t\t\t}, response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new workflow snapshot\",\n\t\t\trequest: &ConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID:               1,\n\t\t\t\tMode:                  ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: *sampleWorkflowSnapshot(),\n\t\t\t\tEncoding:              constants.EncodingTypeThriftRW,\n\t\t\t},\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedSerializer.EXPECT().SerializeEvent(completionEvent(), constants.EncodingTypeThriftRW).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\tcheckRes: func(t *testing.T, response *ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\n\t\t\ttc.prepareMocks(mockedStore, mockedSerializer)\n\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\tres, err := manager.ConflictResolveWorkflowExecution(context.Background(), tc.request)\n\n\t\t\ttc.checkRes(t, res, err)\n\t\t})\n\t}\n}\n\nfunc TestCreateFailoverMarkerTasks(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockedStore := NewMockExecutionStore(ctrl)\n\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), nil, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\treq := &CreateFailoverMarkersRequest{\n\t\tMarkers: []*FailoverMarkerTask{{\n\t\t\tTaskData: TaskData{\n\t\t\t\tVersion:             0,\n\t\t\t\tTaskID:              0,\n\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t},\n\t\t\tDomainID: \"1\",\n\t\t}},\n\t\tRangeID:          1,\n\t\tCurrentTimeStamp: time.Now(),\n\t}\n\n\tmockedStore.EXPECT().CreateFailoverMarkerTasks(gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, req *CreateFailoverMarkersRequest) error {\n\t\t\tassert.Equal(t, req.RangeID, req.RangeID)\n\t\t\tassert.Equal(t, req.Markers, req.Markers)\n\n\t\t\tassert.WithinDuration(t, req.CurrentTimeStamp, req.CurrentTimeStamp, time.Second)\n\t\t\treturn nil\n\t\t})\n\n\terr := manager.CreateFailoverMarkerTasks(context.Background(), req)\n\tassert.NoError(t, err)\n}\n\nfunc TestGetActiveClusterSelectionPolicy(t *testing.T) {\n\tctx := context.Background()\n\tdomainID := \"domainID\"\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\n\ttests := []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockExecutionStore, *MockPayloadSerializer)\n\t\twant         *types.ActiveClusterSelectionPolicy\n\t\twantErr      error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tdata := sampleActiveClusterSelectionPolicyData()\n\t\t\t\tmockedStore.EXPECT().GetActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).\n\t\t\t\t\tReturn(data, nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeActiveClusterSelectionPolicy(data).Return(sampleActiveClusterSelectionPolicy(), nil)\n\t\t\t},\n\t\t\twant: sampleActiveClusterSelectionPolicy(),\n\t\t},\n\t\t{\n\t\t\tname: \"store error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedStore.EXPECT().GetActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).\n\t\t\t\t\tReturn(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname: \"store returned nil\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tmockedStore.EXPECT().GetActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).\n\t\t\t\t\tReturn(nil, nil)\n\t\t\t},\n\t\t\twantErr: &types.EntityNotExistsError{\n\t\t\t\tMessage: \"active cluster selection policy not found\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"deserialize error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore, mockedSerializer *MockPayloadSerializer) {\n\t\t\t\tdata := sampleActiveClusterSelectionPolicyData()\n\t\t\t\tmockedStore.EXPECT().GetActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).\n\t\t\t\t\tReturn(data, nil)\n\t\t\t\tmockedSerializer.EXPECT().DeserializeActiveClusterSelectionPolicy(data).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\tmockedSerializer := NewMockPayloadSerializer(ctrl)\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), mockedSerializer, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.prepareMocks(mockedStore, mockedSerializer)\n\n\t\t\tpolicy, err := manager.GetActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n\t\t\tif test.wantErr != nil {\n\t\t\t\tassert.EqualError(t, err, test.wantErr.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, test.want, policy)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteActiveClusterSelectionPolicy(t *testing.T) {\n\tctx := context.Background()\n\tdomainID := \"domainID\"\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\n\ttests := []struct {\n\t\tname         string\n\t\tprepareMocks func(*MockExecutionStore)\n\t\twantErr      error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"store error\",\n\t\t\tprepareMocks: func(mockedStore *MockExecutionStore) {\n\t\t\t\tmockedStore.EXPECT().DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID).Return(assert.AnError)\n\t\t\t},\n\t\t\twantErr: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedStore := NewMockExecutionStore(ctrl)\n\t\t\tmanager := NewExecutionManagerImpl(mockedStore, testlogger.New(t), nil, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.prepareMocks(mockedStore)\n\n\t\t\terr := manager.DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n\t\t\tif test.wantErr != nil {\n\t\t\t\tassert.EqualError(t, err, test.wantErr.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc sampleInternalActivityInfo(name string) *InternalActivityInfo {\n\treturn &InternalActivityInfo{\n\t\tVersion:        1,\n\t\tScheduleID:     1,\n\t\tActivityID:     name,\n\t\tTaskList:       \"TaskList\",\n\t\tTaskListKind:   TaskListKindSticky,\n\t\tScheduledEvent: NewDataBlob([]byte(fmt.Sprintf(\"%s-activity-scheduled-event\", name)), constants.EncodingTypeThriftRW),\n\t\tStartedEvent:   NewDataBlob([]byte(fmt.Sprintf(\"%s-activity-started-event\", name)), constants.EncodingTypeThriftRW),\n\t}\n}\n\nfunc sampleActivityInfo(name string, id int64) *ActivityInfo {\n\treturn &ActivityInfo{\n\t\tVersion:      1,\n\t\tScheduleID:   1,\n\t\tActivityID:   name,\n\t\tTaskList:     \"TaskList\",\n\t\tTaskListKind: TaskListKindSticky,\n\t\tScheduledEvent: &types.HistoryEvent{\n\t\t\tID: id,\n\t\t},\n\t\tStartedEvent: &types.HistoryEvent{\n\t\t\tID: id,\n\t\t},\n\t}\n}\n\nvar (\n\tstartedTimestamp           = time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC)\n\tscheduledTimestamp         = time.Date(2020, 2, 1, 0, 0, 0, 0, time.UTC)\n\toriginalScheduledTimestamp = time.Date(2020, 3, 1, 0, 0, 0, 0, time.UTC)\n\n\twfTimeout       = 10 * time.Second\n\tdecisionTimeout = 5 * time.Second\n)\n\nfunc sampleInternalWorkflowExecutionInfo() *InternalWorkflowExecutionInfo {\n\treturn &InternalWorkflowExecutionInfo{\n\t\tDomainID:                           testDomain,\n\t\tWorkflowTimeout:                    wfTimeout,\n\t\tDecisionStartToCloseTimeout:        decisionTimeout,\n\t\tDecisionStartedTimestamp:           startedTimestamp,\n\t\tDecisionScheduledTimestamp:         scheduledTimestamp,\n\t\tDecisionOriginalScheduledTimestamp: originalScheduledTimestamp,\n\t\tWorkflowID:                         testWorkflowID,\n\t\tRunID:                              testRunID,\n\t\tWorkflowTypeName:                   testWorkflowType,\n\t\tNextEventID:                        10,\n\t\tCompletionEvent:                    sampleEventData(),\n\t\tAutoResetPoints:                    sampleResetPointsData(),\n\t\tHistorySize:                        1024,\n\t\tActiveClusterSelectionPolicy:       sampleActiveClusterSelectionPolicyData(),\n\t}\n}\n\nfunc sampleWorkflowExecutionInfo() *WorkflowExecutionInfo {\n\treturn &WorkflowExecutionInfo{\n\t\tDomainID:                           testDomain,\n\t\tWorkflowTimeout:                    int32(wfTimeout.Seconds()),\n\t\tDecisionStartToCloseTimeout:        int32(decisionTimeout.Seconds()),\n\t\tDecisionScheduledTimestamp:         scheduledTimestamp.UnixNano(),\n\t\tDecisionStartedTimestamp:           startedTimestamp.UnixNano(),\n\t\tDecisionOriginalScheduledTimestamp: originalScheduledTimestamp.UnixNano(),\n\t\tWorkflowID:                         testWorkflowID,\n\t\tRunID:                              testRunID,\n\t\tWorkflowTypeName:                   testWorkflowType,\n\t\tNextEventID:                        10,\n\t\tCompletionEvent:                    completionEvent(),\n\t\tAutoResetPoints:                    generateResetPoints(),\n\t\tActiveClusterSelectionPolicy:       generateActiveClusterSelectionPolicy(),\n\t}\n}\n\nfunc sampleInternalWorkflowMutation() *InternalWorkflowMutation {\n\treturn &InternalWorkflowMutation{\n\t\tExecutionInfo: sampleInternalWorkflowExecutionInfo(),\n\t\tUpsertActivityInfos: []*InternalActivityInfo{\n\t\t\t{\n\t\t\t\tVersion:        1,\n\t\t\t\tScheduledEvent: sampleEventData(),\n\t\t\t\tStartedEvent:   sampleEventData(),\n\t\t\t},\n\t\t},\n\t\tUpsertTimerInfos: []*TimerInfo{\n\t\t\t{\n\t\t\t\tTimerID: \"test-timer\",\n\t\t\t},\n\t\t},\n\t\tUpsertChildExecutionInfos: []*InternalChildExecutionInfo{\n\t\t\t{\n\t\t\t\tDomainID:         testDomainID,\n\t\t\t\tWorkflowTypeName: testWorkflowType,\n\t\t\t\tVersion:          1,\n\t\t\t\tInitiatedEvent:   sampleEventData(),\n\t\t\t\tStartedEvent:     sampleEventData(),\n\t\t\t},\n\t\t},\n\t\tChecksum:         generateChecksum(),\n\t\tChecksumData:     sampleTestCheckSumData(),\n\t\tStartVersion:     constants.EmptyVersion,\n\t\tLastWriteVersion: constants.EmptyVersion,\n\t}\n}\n\nfunc sampleWorkflowMutation() *WorkflowMutation {\n\treturn &WorkflowMutation{\n\t\tExecutionInfo:  sampleWorkflowExecutionInfo(),\n\t\tExecutionStats: sampleWorkflowExecutionStats(),\n\t\tUpsertActivityInfos: []*ActivityInfo{\n\t\t\t{\n\t\t\t\tVersion:        1,\n\t\t\t\tScheduledEvent: activityScheduledEvent(),\n\t\t\t\tStartedEvent:   activityStartedEvent(),\n\t\t\t},\n\t\t},\n\t\tUpsertTimerInfos: []*TimerInfo{{TimerID: \"test-timer\"}},\n\t\tChecksum:         generateChecksum(),\n\t\tUpsertChildExecutionInfos: []*ChildExecutionInfo{\n\t\t\t{\n\t\t\t\tDomainID:         testDomainID,\n\t\t\t\tWorkflowTypeName: testWorkflowType,\n\t\t\t\tVersion:          1,\n\t\t\t\tInitiatedEvent:   childWorkflowScheduledEvent(),\n\t\t\t\tStartedEvent:     childWorkflowStartedEvent(),\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc sampleWorkflowSnapshot() *WorkflowSnapshot {\n\treturn &WorkflowSnapshot{\n\t\tExecutionInfo:  sampleWorkflowExecutionInfo(),\n\t\tExecutionStats: sampleWorkflowExecutionStats(),\n\t\tVersionHistories: &VersionHistories{\n\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\tHistories: []*VersionHistory{\n\t\t\t\t{\n\t\t\t\t\tBranchToken: []byte(\"test-branch-token\"),\n\t\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tChecksum: generateChecksum(),\n\t\tActivityInfos: []*ActivityInfo{\n\t\t\t{\n\t\t\t\tVersion:        1,\n\t\t\t\tScheduleID:     1,\n\t\t\t\tActivityID:     \"activity1\",\n\t\t\t\tScheduledEvent: activityScheduledEvent(),\n\t\t\t\tStartedID:      2,\n\t\t\t\tStartedEvent:   activityStartedEvent(),\n\t\t\t\tStartedTime:    startedTimestamp,\n\t\t\t},\n\t\t},\n\t\tChildExecutionInfos: []*ChildExecutionInfo{\n\t\t\t{\n\t\t\t\tVersion:          1,\n\t\t\t\tInitiatedID:      1,\n\t\t\t\tInitiatedEvent:   childWorkflowScheduledEvent(),\n\t\t\t\tStartedID:        2,\n\t\t\t\tStartedEvent:     childWorkflowStartedEvent(),\n\t\t\t\tCreateRequestID:  \"create-request-id\",\n\t\t\t\tDomainID:         testDomainID,\n\t\t\t\tWorkflowTypeName: testWorkflowType,\n\t\t\t},\n\t\t},\n\t\tTimerInfos: []*TimerInfo{\n\t\t\t{\n\t\t\t\tTimerID:    \"test-timer\",\n\t\t\t\tStartedID:  1,\n\t\t\t\tExpiryTime: originalScheduledTimestamp,\n\t\t\t\tTaskStatus: 1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tTimerID:    \"test-timer-2\",\n\t\t\t\tStartedID:  2,\n\t\t\t\tExpiryTime: originalScheduledTimestamp,\n\t\t\t\tTaskStatus: 2,\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc sampleInternalWorkflowSnapshot() *InternalWorkflowSnapshot {\n\treturn &InternalWorkflowSnapshot{\n\t\tExecutionInfo:    sampleInternalWorkflowExecutionInfo(),\n\t\tVersionHistories: sampleEventData(),\n\t\tStartVersion:     1,\n\t\tLastWriteVersion: 1,\n\t\tActivityInfos: []*InternalActivityInfo{\n\t\t\t{\n\t\t\t\tVersion:        1,\n\t\t\t\tScheduleID:     1,\n\t\t\t\tActivityID:     \"activity1\",\n\t\t\t\tScheduledEvent: sampleEventData(),\n\t\t\t\tStartedEvent:   sampleEventData(),\n\t\t\t\tStartedID:      2,\n\t\t\t\tStartedTime:    startedTimestamp,\n\t\t\t},\n\t\t},\n\t\tTimerInfos: []*TimerInfo{\n\t\t\t{\n\t\t\t\tTimerID:    \"test-timer\",\n\t\t\t\tStartedID:  1,\n\t\t\t\tExpiryTime: originalScheduledTimestamp,\n\t\t\t\tTaskStatus: 1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tTimerID:    \"test-timer-2\",\n\t\t\t\tStartedID:  2,\n\t\t\t\tExpiryTime: originalScheduledTimestamp,\n\t\t\t\tTaskStatus: 2,\n\t\t\t},\n\t\t},\n\t\tChildExecutionInfos: []*InternalChildExecutionInfo{\n\t\t\t{\n\t\t\t\tVersion:          1,\n\t\t\t\tInitiatedID:      1,\n\t\t\t\tInitiatedEvent:   sampleEventData(),\n\t\t\t\tStartedID:        2,\n\t\t\t\tStartedEvent:     sampleEventData(),\n\t\t\t\tCreateRequestID:  \"create-request-id\",\n\t\t\t\tDomainID:         testDomainID,\n\t\t\t\tWorkflowTypeName: testWorkflowType,\n\t\t\t},\n\t\t},\n\t\tChecksum:     generateChecksum(),\n\t\tChecksumData: sampleCheckSumData(),\n\t}\n}\n\nfunc activityScheduledEvent() *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        1,\n\t\tTimestamp: common.Ptr(scheduledTimestamp.UnixNano()),\n\t\tTaskID:    1,\n\t}\n}\n\nfunc activityStartedEvent() *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        2,\n\t\tTimestamp: common.Ptr(startedTimestamp.UnixNano()),\n\t\tTaskID:    1,\n\t}\n}\n\nfunc childWorkflowScheduledEvent() *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        1,\n\t\tTimestamp: common.Ptr(scheduledTimestamp.UnixNano()),\n\t\tTaskID:    1,\n\t}\n}\n\nfunc completionEvent() *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        99,\n\t\tTimestamp: common.Ptr(startedTimestamp.UnixNano()),\n\t\tTaskID:    1,\n\t}\n}\n\nfunc childWorkflowStartedEvent() *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        2,\n\t\tTimestamp: common.Ptr(startedTimestamp.UnixNano()),\n\t\tTaskID:    1,\n\t}\n}\n\nfunc sampleWorkflowExecutionStats() *ExecutionStats {\n\treturn &ExecutionStats{\n\t\tHistorySize: 1024,\n\t}\n}\n\nfunc sampleTestCheckSumData() *DataBlob {\n\treturn &DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\tData:     []byte(\"test-checksum\"),\n\t}\n}\n\nfunc sampleEventData() *DataBlob {\n\treturn NewDataBlob([]byte(\"test-event\"), constants.EncodingTypeThriftRW)\n}\n\nfunc sampleResetPointsData() *DataBlob {\n\treturn NewDataBlob([]byte(\"test-reset-points\"), constants.EncodingTypeThriftRW)\n}\n\nfunc sampleCheckSumData() *DataBlob {\n\treturn NewDataBlob([]byte(\"test-checksum\"), constants.EncodingTypeThriftRW)\n}\n\nfunc sampleActiveClusterSelectionPolicyData() *DataBlob {\n\treturn NewDataBlob([]byte(\"test-active-cluster-selection-policy\"), constants.EncodingTypeThriftRW)\n}\n\nfunc sampleActiveClusterSelectionPolicy() *types.ActiveClusterSelectionPolicy {\n\treturn &types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\tStickyRegion:                   \"region-1\",\n\t}\n}\n\nfunc sampleInternalChildExecutionInfo(initEventID, startedEventID int64) *InternalChildExecutionInfo {\n\treturn &InternalChildExecutionInfo{\n\t\tVersion:           initEventID,\n\t\tInitiatedID:       initEventID,\n\t\tInitiatedEvent:    sampleEventDataWithVersion(initEventID),\n\t\tStartedID:         startedEventID,\n\t\tStartedEvent:      sampleEventDataWithVersion(startedEventID),\n\t\tCreateRequestID:   \"create-request-id\",\n\t\tDomainID:          testDomainID,\n\t\tWorkflowTypeName:  testWorkflowType,\n\t\tStartedWorkflowID: \"workflow-id\",\n\t\tStartedRunID:      \"run-id\",\n\t}\n}\n\nfunc sampleEventDataWithVersion(i int64) *DataBlob {\n\treturn NewDataBlob([]byte(fmt.Sprintf(\"test-event-%d\", i)), constants.EncodingTypeThriftRW)\n}\n"
  },
  {
    "path": "common/persistence/history_manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/pborman/uuid\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\ntype (\n\t// historyV2PagingToken is used to serialize/deserialize pagination token for ReadHistoryBranchRequest\n\thistoryV2PagingToken struct {\n\t\t// TODO remove LastEventVersion once 3+DC is enabled for all workflow\n\t\tLastEventVersion int64\n\t\tLastEventID      int64\n\t\t// the pagination token passing to persistence\n\t\tStoreToken []byte\n\t\t// recording which branchRange it is reading\n\t\tCurrentRangeIndex int\n\t\tFinalRangeIndex   int\n\n\t\t// LastNodeID is the last known node ID attached to a history node\n\t\tLastNodeID int64\n\t\t// LastTransactionID is the last known transaction ID attached to a history node\n\t\tLastTransactionID int64\n\t}\n\t// historyManagerImpl implements HistoryManager based on HistoryStore and PayloadSerializer\n\thistoryV2ManagerImpl struct {\n\t\thistorySerializer      PayloadSerializer\n\t\tpersistence            HistoryStore\n\t\tlogger                 log.Logger\n\t\tthriftEncoder          codec.BinaryEncoder\n\t\ttransactionSizeLimit   dynamicproperties.IntPropertyFn\n\t\tserializeTokenFn       func(*historyV2PagingToken) ([]byte, error)\n\t\tdeserializeTokenFn     func([]byte, int64) (*historyV2PagingToken, error)\n\t\treadRawHistoryBranchFn func(context.Context, *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error)\n\t\treadHistoryBranchFn    func(context.Context, bool, *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error)\n\t\ttimeSrc                clock.TimeSource\n\t}\n)\n\nconst (\n\tnotStartedIndex          = -1\n\tdefaultLastNodeID        = constants.FirstEventID - 1\n\tdefaultLastTransactionID = int64(0)\n)\n\nvar (\n\tErrCorruptedHistory = &types.InternalDataInconsistencyError{Message: \"corrupted history event batch, eventID is not continuous\"}\n)\n\nvar _ HistoryManager = (*historyV2ManagerImpl)(nil)\n\n// NewHistoryV2ManagerImpl returns new HistoryManager\nfunc NewHistoryV2ManagerImpl(\n\tpersistence HistoryStore,\n\tlogger log.Logger,\n\thistorySerializer PayloadSerializer,\n\tbinaryEncoder codec.BinaryEncoder,\n\ttransactionSizeLimit dynamicproperties.IntPropertyFn,\n) HistoryManager {\n\thm := &historyV2ManagerImpl{\n\t\thistorySerializer:    historySerializer,\n\t\tpersistence:          persistence,\n\t\tlogger:               logger,\n\t\tthriftEncoder:        binaryEncoder,\n\t\ttransactionSizeLimit: transactionSizeLimit,\n\t\tserializeTokenFn:     serializeToken,\n\t\tdeserializeTokenFn:   deserializeToken,\n\t\ttimeSrc:              clock.NewRealTimeSource(),\n\t}\n\thm.readRawHistoryBranchFn = hm.readRawHistoryBranch\n\thm.readHistoryBranchFn = hm.readHistoryBranch\n\treturn hm\n}\n\nfunc (m *historyV2ManagerImpl) GetName() string {\n\treturn m.persistence.GetName()\n}\n\n// ForkHistoryBranch forks a new branch from a old branch\nfunc (m *historyV2ManagerImpl) ForkHistoryBranch(\n\tctx context.Context,\n\trequest *ForkHistoryBranchRequest,\n) (*ForkHistoryBranchResponse, error) {\n\tif request.ForkNodeID <= 1 {\n\t\treturn nil, &InvalidPersistenceRequestError{\n\t\t\tMsg: \"ForkNodeID must be > 1\",\n\t\t}\n\t}\n\tshardID, err := getShardID(request.ShardID)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: err.Error(),\n\t\t}\n\t}\n\n\tvar forkBranch workflow.HistoryBranch\n\terr = m.thriftEncoder.Decode(request.ForkBranchToken, &forkBranch)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treq := &InternalForkHistoryBranchRequest{\n\t\tForkBranchInfo:   *thrift.ToHistoryBranch(&forkBranch),\n\t\tForkNodeID:       request.ForkNodeID,\n\t\tNewBranchID:      uuid.New(),\n\t\tInfo:             request.Info,\n\t\tShardID:          shardID,\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\n\tresp, err := m.persistence.ForkHistoryBranch(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttoken, err := m.thriftEncoder.Encode(thrift.FromHistoryBranch(&resp.NewBranchInfo))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ForkHistoryBranchResponse{\n\t\tNewBranchToken: token,\n\t}, nil\n}\n\n// DeleteHistoryBranch removes a branch\nfunc (m *historyV2ManagerImpl) DeleteHistoryBranch(\n\tctx context.Context,\n\trequest *DeleteHistoryBranchRequest,\n) error {\n\tshardID, err := getShardID(request.ShardID)\n\tif err != nil {\n\t\tm.logger.Error(\"shardID is not set in delete history operation\", tag.Error(err))\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: err.Error(),\n\t\t}\n\t}\n\n\tvar branch workflow.HistoryBranch\n\terr = m.thriftEncoder.Decode(request.BranchToken, &branch)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treq := &InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: *thrift.ToHistoryBranch(&branch),\n\t\tShardID:    shardID,\n\t}\n\treturn m.persistence.DeleteHistoryBranch(ctx, req)\n}\n\n// GetHistoryTree returns all branch information of a tree\nfunc (m *historyV2ManagerImpl) GetHistoryTree(\n\tctx context.Context,\n\trequest *GetHistoryTreeRequest,\n) (*GetHistoryTreeResponse, error) {\n\tif len(request.TreeID) == 0 {\n\t\tvar branch workflow.HistoryBranch\n\t\terr := m.thriftEncoder.Decode(request.BranchToken, &branch)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trequest.TreeID = branch.GetTreeID()\n\t}\n\tinternalRequest := &InternalGetHistoryTreeRequest{\n\t\tTreeID:      request.TreeID,\n\t\tShardID:     request.ShardID,\n\t\tBranchToken: request.BranchToken,\n\t}\n\tresp, err := m.persistence.GetHistoryTree(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar branches []*workflow.HistoryBranch\n\tfor _, b := range resp.Branches {\n\t\tbranches = append(branches, thrift.FromHistoryBranch(b))\n\t}\n\treturn &GetHistoryTreeResponse{\n\t\tBranches: branches,\n\t}, nil\n}\n\n// AppendHistoryNodes add(or override) a node to a history branch\nfunc (m *historyV2ManagerImpl) AppendHistoryNodes(\n\tctx context.Context,\n\trequest *AppendHistoryNodesRequest,\n) (*AppendHistoryNodesResponse, error) {\n\tshardID, err := getShardID(request.ShardID)\n\tif err != nil {\n\t\tm.logger.Error(\"shardID is not set in append history nodes operation\", tag.Error(err))\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: err.Error(),\n\t\t}\n\t}\n\tif len(request.Events) == 0 {\n\t\treturn nil, &InvalidPersistenceRequestError{\n\t\t\tMsg: \"events to be appended cannot be empty\",\n\t\t}\n\t}\n\tversion := request.Events[0].Version\n\tnodeID := request.Events[0].ID\n\tlastID := nodeID - 1\n\tif nodeID <= 0 {\n\t\treturn nil, &InvalidPersistenceRequestError{\n\t\t\tMsg: \"eventID cannot be less than 1\",\n\t\t}\n\t}\n\tfor _, e := range request.Events {\n\t\tif e.Version != version {\n\t\t\treturn nil, &InvalidPersistenceRequestError{\n\t\t\t\tMsg: \"event version must be the same inside a batch\",\n\t\t\t}\n\t\t}\n\t\tif e.ID != lastID+1 {\n\t\t\treturn nil, &InvalidPersistenceRequestError{\n\t\t\t\tMsg: \"event ID must be continuous\",\n\t\t\t}\n\t\t}\n\t\tlastID++\n\t}\n\tvar branch workflow.HistoryBranch\n\terr = m.thriftEncoder.Decode(request.BranchToken, &branch)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// nodeID will be the first eventID\n\tblob, err := m.historySerializer.SerializeBatchEvents(request.Events, request.Encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsize := len(blob.Data)\n\tsizeLimit := m.transactionSizeLimit()\n\tif size > sizeLimit {\n\t\treturn nil, &TransactionSizeLimitError{\n\t\t\tMsg: fmt.Sprintf(\"transaction size of %v bytes exceeds limit of %v bytes\", size, sizeLimit),\n\t\t}\n\t}\n\treq := &InternalAppendHistoryNodesRequest{\n\t\tIsNewBranch:      request.IsNewBranch,\n\t\tInfo:             request.Info,\n\t\tBranchInfo:       *thrift.ToHistoryBranch(&branch),\n\t\tNodeID:           nodeID,\n\t\tEvents:           blob,\n\t\tTransactionID:    request.TransactionID,\n\t\tShardID:          shardID,\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\n\terr = m.persistence.AppendHistoryNodes(ctx, req)\n\n\treturn &AppendHistoryNodesResponse{\n\t\tDataBlob: *blob,\n\t}, err\n}\n\nfunc (m *historyV2ManagerImpl) GetAllHistoryTreeBranches(\n\tctx context.Context,\n\trequest *GetAllHistoryTreeBranchesRequest,\n) (*GetAllHistoryTreeBranchesResponse, error) {\n\treturn m.persistence.GetAllHistoryTreeBranches(ctx, request)\n}\n\n// ReadHistoryBranchByBatch returns history node data for a branch by batch\n// Pagination is implemented here, the actual minNodeID passing to persistence layer is calculated along with token's LastNodeID\nfunc (m *historyV2ManagerImpl) ReadHistoryBranchByBatch(\n\tctx context.Context,\n\trequest *ReadHistoryBranchRequest,\n) (*ReadHistoryBranchByBatchResponse, error) {\n\n\tresp := &ReadHistoryBranchByBatchResponse{}\n\tvar err error\n\t_, resp.History, resp.NextPageToken, resp.Size, resp.LastFirstEventID, err = m.readHistoryBranchFn(ctx, true, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// ReadHistoryBranch returns history node data for a branch\n// Pagination is implemented here, the actual minNodeID passing to persistence layer is calculated along with token's LastNodeID\nfunc (m *historyV2ManagerImpl) ReadHistoryBranch(\n\tctx context.Context,\n\trequest *ReadHistoryBranchRequest,\n) (*ReadHistoryBranchResponse, error) {\n\n\tresp := &ReadHistoryBranchResponse{}\n\tvar err error\n\tresp.HistoryEvents, _, resp.NextPageToken, resp.Size, resp.LastFirstEventID, err = m.readHistoryBranchFn(ctx, false, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// ReadRawHistoryBranch returns raw history binary data for a branch\n// Pagination is implemented here, the actual minNodeID passing to persistence layer is calculated along with token's LastNodeID\n// NOTE: this API should only be used by 3+DC\nfunc (m *historyV2ManagerImpl) ReadRawHistoryBranch(\n\tctx context.Context,\n\trequest *ReadHistoryBranchRequest,\n) (*ReadRawHistoryBranchResponse, error) {\n\n\tdataBlobs, token, dataSize, _, err := m.readRawHistoryBranchFn(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnextPageToken, err := m.serializeTokenFn(token)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ReadRawHistoryBranchResponse{\n\t\tHistoryEventBlobs: dataBlobs,\n\t\tNextPageToken:     nextPageToken,\n\t\tSize:              dataSize,\n\t}, nil\n}\n\nfunc (m *historyV2ManagerImpl) readRawHistoryBranch(\n\tctx context.Context,\n\trequest *ReadHistoryBranchRequest,\n) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\tshardID, err := getShardID(request.ShardID)\n\tif err != nil {\n\t\tm.logger.Error(\"shardID is not set in read history branch operation\", tag.Error(err))\n\t\treturn nil, nil, 0, nil, &types.InternalServiceError{Message: err.Error()}\n\t}\n\tif request.PageSize <= 0 || request.MinEventID >= request.MaxEventID {\n\t\treturn nil, nil, 0, nil, &InvalidPersistenceRequestError{\n\t\t\tMsg: fmt.Sprintf(\n\t\t\t\t\"no events can be found for pageSize %v, minEventID %v, maxEventID: %v\",\n\t\t\t\trequest.PageSize,\n\t\t\t\trequest.MinEventID,\n\t\t\t\trequest.MaxEventID,\n\t\t\t),\n\t\t}\n\t}\n\tdefaultLastEventID := request.MinEventID - 1\n\ttoken, err := m.deserializeTokenFn(\n\t\trequest.NextPageToken,\n\t\tdefaultLastEventID,\n\t)\n\tif err != nil {\n\t\treturn nil, nil, 0, nil, err\n\t}\n\n\tvar branch workflow.HistoryBranch\n\terr = m.thriftEncoder.Decode(request.BranchToken, &branch)\n\tif err != nil {\n\t\treturn nil, nil, 0, nil, err\n\t}\n\ttreeID := *branch.TreeID\n\tbranchID := *branch.BranchID\n\n\tallBRs := branch.Ancestors\n\t// We may also query the current branch from beginNodeID\n\tbeginNodeID := constants.FirstEventID\n\tif len(branch.Ancestors) > 0 {\n\t\tbeginNodeID = *branch.Ancestors[len(branch.Ancestors)-1].EndNodeID\n\t}\n\tallBRs = append(allBRs, &workflow.HistoryBranchRange{\n\t\tBranchID:    &branchID,\n\t\tBeginNodeID: common.Int64Ptr(beginNodeID),\n\t\tEndNodeID:   common.Int64Ptr(request.MaxEventID),\n\t})\n\n\tif token.CurrentRangeIndex == notStartedIndex {\n\t\tfor idx, br := range allBRs {\n\t\t\t// this range won't contain any nodes needed\n\t\t\tif request.MinEventID >= *br.EndNodeID {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\t// similarly, the ranges and the rest won't contain any nodes needed,\n\t\t\tif request.MaxEventID <= *br.BeginNodeID {\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\tif token.CurrentRangeIndex == notStartedIndex {\n\t\t\t\ttoken.CurrentRangeIndex = idx\n\t\t\t}\n\t\t\ttoken.FinalRangeIndex = idx\n\t\t}\n\n\t\tif token.CurrentRangeIndex == notStartedIndex {\n\t\t\treturn nil, nil, 0, nil, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"branchRange is corrupted\",\n\t\t\t}\n\t\t}\n\t}\n\n\tminNodeID := request.MinEventID\n\tmaxNodeID := *allBRs[token.CurrentRangeIndex].EndNodeID\n\tif request.MaxEventID < maxNodeID {\n\t\tmaxNodeID = request.MaxEventID\n\t}\n\tpageSize := request.PageSize\n\n\treq := &InternalReadHistoryBranchRequest{\n\t\tTreeID:            treeID,\n\t\tBranchID:          *allBRs[token.CurrentRangeIndex].BranchID,\n\t\tMinNodeID:         minNodeID,\n\t\tMaxNodeID:         maxNodeID,\n\t\tNextPageToken:     token.StoreToken,\n\t\tLastNodeID:        token.LastNodeID,\n\t\tLastTransactionID: token.LastTransactionID,\n\t\tShardID:           shardID,\n\t\tPageSize:          pageSize,\n\t}\n\n\tresp, err := m.persistence.ReadHistoryBranch(ctx, req)\n\tif err != nil {\n\t\treturn nil, nil, 0, nil, err\n\t}\n\t// TODO: consider if it's possible to remove this branch\n\tif len(resp.History) == 0 && len(request.NextPageToken) == 0 {\n\t\treturn nil, nil, 0, nil, &types.EntityNotExistsError{Message: \"Workflow execution history not found.\"}\n\t}\n\n\tdataBlobs := resp.History\n\tdataSize := 0\n\tfor _, dataBlob := range resp.History {\n\t\tdataSize += len(dataBlob.Data)\n\t}\n\n\ttoken.StoreToken = resp.NextPageToken\n\ttoken.LastNodeID = resp.LastNodeID\n\ttoken.LastTransactionID = resp.LastTransactionID\n\n\t// NOTE: in this method, we need to make sure eventVersion is NOT\n\t// decreasing(otherwise we skip the events), eventID should be continuous(otherwise return error)\n\tlogger := m.logger.WithTags(tag.WorkflowBranchID(*branch.BranchID), tag.WorkflowTreeID(*branch.TreeID))\n\n\treturn dataBlobs, token, dataSize, logger, nil\n}\n\nfunc (m *historyV2ManagerImpl) readHistoryBranch(\n\tctx context.Context,\n\tbyBatch bool,\n\trequest *ReadHistoryBranchRequest,\n) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error) {\n\n\tdataBlobs, token, dataSize, logger, err := m.readRawHistoryBranchFn(ctx, request)\n\tif err != nil {\n\t\treturn nil, nil, nil, 0, 0, err\n\t}\n\tdefaultLastEventID := request.MinEventID - 1\n\n\thistoryEvents := make([]*types.HistoryEvent, 0, request.PageSize)\n\thistoryEventBatches := make([]*types.History, 0, request.PageSize)\n\t// first_event_id of the last batch\n\tlastFirstEventID := constants.EmptyEventID\n\n\tfor _, batch := range dataBlobs {\n\t\tevents, err := m.historySerializer.DeserializeBatchEvents(batch)\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, 0, 0, err\n\t\t}\n\t\tif len(events) == 0 {\n\t\t\tlogger.Error(\"Empty events in a batch\")\n\t\t\treturn nil, nil, nil, 0, 0, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted history event batch, empty events\",\n\t\t\t}\n\t\t}\n\n\t\tfirstEvent := events[0]           // first\n\t\teventCount := len(events)         // length\n\t\tlastEvent := events[eventCount-1] // last\n\n\t\tif firstEvent.Version != lastEvent.Version || firstEvent.ID+int64(eventCount-1) != lastEvent.ID {\n\t\t\t// in a single batch, version should be the same, and ID should be continous\n\t\t\tlogger.Error(\"Corrupted event batch\",\n\t\t\t\ttag.FirstEventVersion(firstEvent.Version), tag.WorkflowFirstEventID(firstEvent.ID),\n\t\t\t\ttag.LastEventVersion(lastEvent.Version), tag.WorkflowNextEventID(lastEvent.ID),\n\t\t\t\ttag.Counter(eventCount))\n\t\t\treturn nil, nil, nil, 0, 0, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted history event batch, wrong version and IDs\",\n\t\t\t}\n\t\t}\n\n\t\tif firstEvent.Version < token.LastEventVersion {\n\t\t\t// version decrease means the this batch are all stale events, we should skip\n\t\t\tlogger.Info(\"Stale event batch with smaller version\", tag.FirstEventVersion(firstEvent.Version), tag.TokenLastEventVersion(token.LastEventVersion))\n\t\t\tcontinue\n\t\t}\n\t\tif firstEvent.ID <= token.LastEventID {\n\t\t\t// we could see it because first batch of next page has a smaller txn_id\n\t\t\tlogger.Info(\"Stale event batch with eventID\", tag.WorkflowFirstEventID(firstEvent.ID), tag.TokenLastEventID(token.LastEventID))\n\t\t\tcontinue\n\t\t}\n\t\tif firstEvent.ID != token.LastEventID+1 {\n\t\t\t// We assume application layer want to read from MinEventID(inclusive)\n\t\t\t// However, for getting history from remote cluster, there is scenario that we have to read from middle without knowing the firstEventID.\n\t\t\t// In that case we don't validate history continuousness for the first page\n\t\t\t// TODO: in this case, some events returned can be invalid(stale). application layer need to make sure it won't make any problems to XDC\n\t\t\tif defaultLastEventID == 0 || token.LastEventID != defaultLastEventID {\n\t\t\t\tlogger.Error(\"Corrupted incontinouous event batch\",\n\t\t\t\t\ttag.FirstEventVersion(firstEvent.Version), tag.WorkflowFirstEventID(firstEvent.ID),\n\t\t\t\t\ttag.LastEventVersion(lastEvent.Version), tag.WorkflowNextEventID(lastEvent.ID),\n\t\t\t\t\ttag.TokenLastEventVersion(token.LastEventVersion), tag.TokenLastEventID(token.LastEventID),\n\t\t\t\t\ttag.Counter(eventCount))\n\t\t\t\treturn nil, nil, nil, 0, 0, ErrCorruptedHistory\n\t\t\t}\n\t\t}\n\n\t\ttoken.LastEventVersion = firstEvent.Version\n\t\ttoken.LastEventID = lastEvent.ID\n\t\tif byBatch {\n\t\t\thistoryEventBatches = append(historyEventBatches, &types.History{Events: events})\n\t\t} else {\n\t\t\thistoryEvents = append(historyEvents, events...)\n\t\t}\n\t\tlastFirstEventID = firstEvent.ID\n\t}\n\n\tnextPageToken, err := m.serializeTokenFn(token)\n\tif err != nil {\n\t\treturn nil, nil, nil, 0, 0, err\n\t}\n\n\treturn historyEvents, historyEventBatches, nextPageToken, dataSize, lastFirstEventID, nil\n}\n\nfunc deserializeToken(\n\tdata []byte,\n\tdefaultLastEventID int64,\n) (*historyV2PagingToken, error) {\n\tif len(data) == 0 {\n\t\treturn &historyV2PagingToken{\n\t\t\tLastEventID:       defaultLastEventID,\n\t\t\tLastEventVersion:  constants.EmptyVersion,\n\t\t\tCurrentRangeIndex: notStartedIndex,\n\t\t\tLastNodeID:        defaultLastNodeID,\n\t\t\tLastTransactionID: defaultLastTransactionID,\n\t\t}, nil\n\t}\n\n\ttoken := historyV2PagingToken{}\n\terr := json.Unmarshal(data, &token)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &token, nil\n}\n\nfunc serializeToken(pagingToken *historyV2PagingToken) ([]byte, error) {\n\tif len(pagingToken.StoreToken) == 0 {\n\t\tif pagingToken.CurrentRangeIndex == pagingToken.FinalRangeIndex {\n\t\t\t// this means that we have reached the final page of final branchRange\n\t\t\treturn nil, nil\n\t\t}\n\t\tpagingToken.CurrentRangeIndex++\n\t}\n\treturn json.Marshal(pagingToken)\n}\n\nfunc (m *historyV2ManagerImpl) Close() {\n\tm.persistence.Close()\n}\n\nfunc getShardID(shardID *int) (int, error) {\n\tif shardID == nil {\n\t\treturn 0, fmt.Errorf(\"shardID is not set for persistence operation\")\n\t}\n\treturn *shardID, nil\n}\n"
  },
  {
    "path": "common/persistence/history_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc setUpMocksForHistoryV2Manager(t *testing.T) (*historyV2ManagerImpl, *MockHistoryStore, *MockPayloadSerializer, *codec.MockBinaryEncoder) {\n\tctrl := gomock.NewController(t)\n\tmockStore := NewMockHistoryStore(ctrl)\n\tmockSerializer := NewMockPayloadSerializer(ctrl)\n\tmockEncoder := codec.NewMockBinaryEncoder(ctrl)\n\tlogger := log.NewNoop()\n\n\tmockStore.EXPECT().GetName().Return(\"mock history store\")\n\n\thistoryManager := NewHistoryV2ManagerImpl(\n\t\tmockStore,\n\t\tlogger,\n\t\tmockSerializer,\n\t\tmockEncoder,\n\t\tdynamicproperties.GetIntPropertyFn(1024*10),\n\t)\n\tassert.Equal(t, \"mock history store\", historyManager.GetName())\n\n\treturn historyManager.(*historyV2ManagerImpl), mockStore, mockSerializer, mockEncoder\n}\n\nfunc TestForkHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockHistoryStore, *codec.MockBinaryEncoder)\n\t\trequest       *ForkHistoryBranchRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *ForkHistoryBranchResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"fork-branch\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tForkHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalForkHistoryBranchResponse{\n\t\t\t\t\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\t\t\t\t\tTreeID:   \"new-tree-id\",\n\t\t\t\t\t\t\tBranchID: \"new-branch-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tEncode(&workflow.HistoryBranch{\n\t\t\t\t\t\tTreeID:   common.StringPtr(\"new-tree-id\"),\n\t\t\t\t\t\tBranchID: common.StringPtr(\"new-branch-id\"),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn([]byte(\"new-branch-token\"), nil).Times(1)\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      2,\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t\tShardID:         common.Ptr(10),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &ForkHistoryBranchResponse{\n\t\t\t\tNewBranchToken: []byte(\"new-branch-token\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"nil Shard ID\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\t// No store or encoder interactions for invalid input\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      2,\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"shardID is not set for persistence operation\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid fork node ID\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\t// No store or encoder interactions for invalid input\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      1, // Invalid ID\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t\tShardID:         common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"ForkNodeID must be > 1\",\n\t\t},\n\t\t{\n\t\t\tname: \"decode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"decode error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      2,\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t\tShardID:         common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"decode error\",\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tForkHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      2,\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t\tShardID:         common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t\t{\n\t\t\tname: \"encode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tForkHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalForkHistoryBranchResponse{\n\t\t\t\t\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\t\t\t\t\tTreeID:   \"new-tree-id\",\n\t\t\t\t\t\t\tBranchID: \"new-branch-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tEncode(gomock.Any()).Return(nil, errors.New(\"encode error\"))\n\t\t\t},\n\t\t\trequest: &ForkHistoryBranchRequest{\n\t\t\t\tForkBranchToken: []byte(\"fork-branch\"),\n\t\t\t\tForkNodeID:      2,\n\t\t\t\tInfo:            \"fork info\",\n\t\t\t\tShardID:         common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"encode error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, _, mockEncoder := setUpMocksForHistoryV2Manager(t)\n\n\t\t\ttc.setupMock(mockStore, mockEncoder)\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.ForkHistoryBranch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockHistoryStore, *codec.MockBinaryEncoder)\n\t\trequest       *DeleteHistoryBranchRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteHistoryBranch(gomock.Any(), &InternalDeleteHistoryBranchRequest{\n\t\t\t\t\t\tShardID: 10,\n\t\t\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\t\t\tTreeID:   \"tree-id\",\n\t\t\t\t\t\t\tBranchID: \"branch-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &DeleteHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"nil shard ID error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t},\n\t\t\trequest: &DeleteHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"shardID is not set for persistence operation\",\n\t\t},\n\t\t{\n\t\t\tname: \"decode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"decode error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &DeleteHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"decode error\",\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tDeleteHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &DeleteHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, _, mockEncoder := setUpMocksForHistoryV2Manager(t)\n\n\t\t\ttc.setupMock(mockStore, mockEncoder)\n\n\t\t\t// Call the method\n\t\t\terr := historyManager.DeleteHistoryBranch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetHistoryTree(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockHistoryStore, *codec.MockBinaryEncoder)\n\t\trequest       *GetHistoryTreeRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *GetHistoryTreeResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetHistoryTree(gomock.Any(), &InternalGetHistoryTreeRequest{\n\t\t\t\t\t\tTreeID:      \"tree-id\",\n\t\t\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&InternalGetHistoryTreeResponse{\n\t\t\t\t\t\tBranches: []*types.HistoryBranch{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTreeID:   \"tree-id\",\n\t\t\t\t\t\t\t\tBranchID: \"branch-id\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &GetHistoryTreeRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &GetHistoryTreeResponse{\n\t\t\t\tBranches: []*workflow.HistoryBranch{\n\t\t\t\t\t{\n\t\t\t\t\t\tTreeID:   common.Ptr(\"tree-id\"),\n\t\t\t\t\t\tBranchID: common.Ptr(\"branch-id\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"decode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"decode error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &GetHistoryTreeRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"decode error\",\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetHistoryTree(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &GetHistoryTreeRequest{\n\t\t\t\tTreeID: \"tree-id\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, _, mockEncoder := setUpMocksForHistoryV2Manager(t)\n\n\t\t\ttc.setupMock(mockStore, mockEncoder)\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.GetHistoryTree(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAppendHistoryNodes(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockHistoryStore, *codec.MockBinaryEncoder, *MockPayloadSerializer)\n\t\trequest       *AppendHistoryNodesRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBatchEvents(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"events\")}, nil).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tAppendHistoryNodes(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tEvents:        []*types.HistoryEvent{{ID: 1, Version: 1}},\n\t\t\t\tTransactionID: 1234,\n\t\t\t\tShardID:       common.Ptr(10),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"shardID not set error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No mock interactions for this invalid input test\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tEvents:      []*types.HistoryEvent{}, // No events\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"shardID is not set for persistence operation\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty events error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No mock interactions for this invalid input test\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tEvents:      []*types.HistoryEvent{}, // No events\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"events to be appended cannot be empty\",\n\t\t},\n\t\t{\n\t\t\tname: \"event ID less than 1 error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No mock interactions for this invalid input test\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tEvents:      []*types.HistoryEvent{{ID: 0, Version: 1}},\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"eventID cannot be less than 1\",\n\t\t},\n\t\t{\n\t\t\tname: \"event version not the same error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No mock interactions for this invalid input test\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tEvents:      []*types.HistoryEvent{{ID: 1, Version: 1}, {ID: 2, Version: 2}},\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"event version must be the same inside a batch\",\n\t\t},\n\t\t{\n\t\t\tname: \"event ID not continuous\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No mock interactions for this invalid input test\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tEvents:      []*types.HistoryEvent{{ID: 1, Version: 1}, {ID: 3, Version: 1}},\n\t\t\t\tShardID:     common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"event ID must be continuous\",\n\t\t},\n\t\t{\n\t\t\tname: \"decode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"decode error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tEvents:        []*types.HistoryEvent{{ID: 1, Version: 1}},\n\t\t\t\tTransactionID: 1234,\n\t\t\t\tShardID:       common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"decode error\",\n\t\t},\n\t\t{\n\t\t\tname: \"serialization error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBatchEvents(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"serialization error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tEvents:        []*types.HistoryEvent{{ID: 1, Version: 1}},\n\t\t\t\tTransactionID: 1234,\n\t\t\t\tShardID:       common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialization error\",\n\t\t},\n\t\t{\n\t\t\tname: \"transaction size limit error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBatchEvents(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: make([]byte, 1024*10+1)}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tEvents:        []*types.HistoryEvent{{ID: 1, Version: 1}},\n\t\t\t\tTransactionID: 1234,\n\t\t\t\tShardID:       common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"transaction size of 10241 bytes exceeds limit of 10240 bytes\",\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder, mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tSerializeBatchEvents(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"events\")}, nil).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tAppendHistoryNodes(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &AppendHistoryNodesRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tEvents:        []*types.HistoryEvent{{ID: 1, Version: 1}},\n\t\t\t\tTransactionID: 1234,\n\t\t\t\tShardID:       common.Ptr(10),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, mockSerializer, mockEncoder := setUpMocksForHistoryV2Manager(t)\n\t\t\ttc.setupMock(mockStore, mockEncoder, mockSerializer)\n\n\t\t\t// Call the method\n\t\t\t_, err := historyManager.AppendHistoryNodes(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetAllHistoryTreeBranches(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockHistoryStore)\n\t\trequest       *GetAllHistoryTreeBranchesRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetAllHistoryTreeBranches(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&GetAllHistoryTreeBranchesResponse{\n\t\t\t\t\t\tBranches: []HistoryBranchDetail{\n\t\t\t\t\t\t\t{TreeID: \"tree-id-1\", BranchID: \"branch-id-1\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &GetAllHistoryTreeBranchesRequest{\n\t\t\t\tPageSize:      10,\n\t\t\t\tNextPageToken: []byte(\"next-token\"),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore) {\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tGetAllHistoryTreeBranches(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &GetAllHistoryTreeBranchesRequest{\n\t\t\t\tPageSize:      10,\n\t\t\t\tNextPageToken: []byte(\"next-token\"),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, _, _ := setUpMocksForHistoryV2Manager(t)\n\n\t\t\ttc.setupMock(mockStore)\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.GetAllHistoryTreeBranches(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, \"tree-id-1\", resp.Branches[0].TreeID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadRawHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname            string\n\t\tsetupMock       func(*MockHistoryStore, *codec.MockBinaryEncoder)\n\t\tfakeSerialize   func(*historyV2PagingToken) ([]byte, error)\n\t\tfakeDeserialize func([]byte, int64) (*historyV2PagingToken, error)\n\t\trequest         *ReadHistoryBranchRequest\n\t\texpectError     bool\n\t\texpectedError   string\n\t\texpectedBlobs   []*DataBlob\n\t\texpectedToken   *historyV2PagingToken\n\t}{\n\t\t{\n\t\t\tname: \"success case\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\tvalue.Ancestors = []*workflow.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    common.StringPtr(\"ancestor-branch-id\"),\n\t\t\t\t\t\t\tBeginNodeID: common.Int64Ptr(1),\n\t\t\t\t\t\t\tEndNodeID:   common.Int64Ptr(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    common.StringPtr(\"ancestor-branch-id-2\"),\n\t\t\t\t\t\t\tBeginNodeID: common.Int64Ptr(10),\n\t\t\t\t\t\t\tEndNodeID:   common.Int64Ptr(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    common.StringPtr(\"ancestor-branch-id-3\"),\n\t\t\t\t\t\t\tBeginNodeID: common.Int64Ptr(20),\n\t\t\t\t\t\t\tEndNodeID:   common.Int64Ptr(30),\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tReadHistoryBranch(gomock.Any(), &InternalReadHistoryBranchRequest{\n\t\t\t\t\t\tTreeID:            \"tree-id\",\n\t\t\t\t\t\tBranchID:          \"ancestor-branch-id-2\",\n\t\t\t\t\t\tMinNodeID:         10,\n\t\t\t\t\t\tMaxNodeID:         19,\n\t\t\t\t\t\tNextPageToken:     []byte(\"store-token\"),\n\t\t\t\t\t\tLastNodeID:        9,\n\t\t\t\t\t\tLastTransactionID: 10,\n\t\t\t\t\t\tShardID:           1,\n\t\t\t\t\t\tPageSize:          10,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&InternalReadHistoryBranchResponse{\n\t\t\t\t\t\tHistory: []*DataBlob{\n\t\t\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextPageToken:     []byte(\"next-token\"),\n\t\t\t\t\t\tLastNodeID:        999,\n\t\t\t\t\t\tLastTransactionID: 1000,\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeDeserialize: func(data []byte, lastEventID int64) (*historyV2PagingToken, error) {\n\t\t\t\tassert.Equal(t, []byte(\"token\"), data)\n\t\t\t\tassert.Equal(t, int64(9), lastEventID)\n\t\t\t\treturn &historyV2PagingToken{\n\t\t\t\t\tLastNodeID:        lastEventID,\n\t\t\t\t\tLastTransactionID: 10,\n\t\t\t\t\tCurrentRangeIndex: notStartedIndex,\n\t\t\t\t\tStoreToken:        []byte(\"store-token\"),\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    10,\n\t\t\t\tMaxEventID:    19,\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedBlobs: []*DataBlob{\n\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t},\n\t\t\texpectedToken: &historyV2PagingToken{\n\t\t\t\tLastNodeID:        999,\n\t\t\t\tLastTransactionID: 1000,\n\t\t\t\tCurrentRangeIndex: 1,\n\t\t\t\tFinalRangeIndex:   1,\n\t\t\t\tStoreToken:        []byte(\"next-token\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"invalid shard ID\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\t// No encoder or store interactions, invalid ShardID is set.\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       nil, // Invalid shardID\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"shardID is not set for persistence operation\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid pagination params\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\t// No encoder or store interactions, invalid pagination params.\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      0, // Invalid PageSize\n\t\t\t\tMinEventID:    100,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"no events can be found for pageSize 0, minEventID 100, maxEventID: 100\",\n\t\t},\n\t\t{\n\t\t\tname: \"deserialize error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t},\n\t\t\tfakeDeserialize: func(data []byte, lastEventID int64) (*historyV2PagingToken, error) {\n\t\t\t\treturn nil, errors.New(\"deserialize error\")\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"deserialize error\",\n\t\t},\n\t\t{\n\t\t\tname: \"decode error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"decode error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"decode error\",\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty history error\",\n\t\t\tsetupMock: func(mockStore *MockHistoryStore, mockEncoder *codec.MockBinaryEncoder) {\n\t\t\t\tmockEncoder.EXPECT().\n\t\t\t\t\tDecode([]byte(\"branch-token\"), &workflow.HistoryBranch{}).DoAndReturn(func(data []byte, value *workflow.HistoryBranch) error {\n\t\t\t\t\tvalue.TreeID = common.Ptr(\"tree-id\")\n\t\t\t\t\tvalue.BranchID = common.Ptr(\"branch-id\")\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStore.EXPECT().\n\t\t\t\t\tReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&InternalReadHistoryBranchResponse{\n\t\t\t\t\t\tHistory:       []*DataBlob{},\n\t\t\t\t\t\tNextPageToken: []byte{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tShardID:       common.IntPtr(1),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Workflow execution history not found.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thistoryManager, mockStore, _, mockEncoder := setUpMocksForHistoryV2Manager(t)\n\t\t\tif tc.fakeSerialize != nil {\n\t\t\t\thistoryManager.serializeTokenFn = tc.fakeSerialize\n\t\t\t}\n\t\t\tif tc.fakeDeserialize != nil {\n\t\t\t\thistoryManager.deserializeTokenFn = tc.fakeDeserialize\n\t\t\t}\n\n\t\t\ttc.setupMock(mockStore, mockEncoder)\n\n\t\t\t// Call the method\n\t\t\tblobs, token, _, _, err := historyManager.readRawHistoryBranch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedBlobs, blobs)\n\t\t\t\tassert.Equal(t, tc.expectedToken, token)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tsetupMock          func(*MockPayloadSerializer)\n\t\tfakeReadRaw        func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error)\n\t\tfakeSerializeToken func(pagingToken *historyV2PagingToken) ([]byte, error)\n\t\tbyBatch            bool\n\t\trequest            *ReadHistoryBranchRequest\n\t\texpectError        bool\n\t\texpectedError      string\n\t\texpectedEvents     []*types.HistoryEvent\n\t\texpectedBatch      []*types.History\n\t}{\n\t\t{\n\t\t\tname: \"success by events\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data2\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 0, Version: 2},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data3\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 2},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t\t{Data: []byte(\"history-event-data2\")},\n\t\t\t\t\t{Data: []byte(\"history-event-data3\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 2, LastEventID: 0}, 100, log.NewNoop(), nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\tassert.Equal(t, &historyV2PagingToken{\n\t\t\t\t\tLastEventID:      1,\n\t\t\t\t\tLastEventVersion: 2,\n\t\t\t\t}, pagingToken)\n\t\t\t\treturn []byte(\"next-page-token\"), nil\n\t\t\t},\n\t\t\tbyBatch: false,\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tPageSize:    10,\n\t\t\t\tMinEventID:  1,\n\t\t\t\tMaxEventID:  100,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{ID: 1, Version: 2},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success by batch\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data2\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 0, Version: 2},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(&DataBlob{Data: []byte(\"history-event-data3\")}).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 2},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t\t{Data: []byte(\"history-event-data2\")},\n\t\t\t\t\t{Data: []byte(\"history-event-data3\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 2, LastEventID: 0}, 100, log.NewNoop(), nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\tassert.Equal(t, &historyV2PagingToken{\n\t\t\t\t\tLastEventID:      1,\n\t\t\t\t\tLastEventVersion: 2,\n\t\t\t\t}, pagingToken)\n\t\t\t\treturn []byte(\"next-page-token\"), nil\n\t\t\t},\n\t\t\tbyBatch: true,\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\tPageSize:    10,\n\t\t\t\tMinEventID:  1,\n\t\t\t\tMaxEventID:  100,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedBatch: []*types.History{\n\t\t\t\t{Events: []*types.HistoryEvent{{ID: 1, Version: 2}}},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty batch error\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(gomock.Any()).\n\t\t\t\t\tReturn([]*types.HistoryEvent{}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 1, LastEventID: 0}, 100, log.NewNoop(), nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, nil // Doesn't matter in this case since it errors earlier\n\t\t\t},\n\t\t\tbyBatch:       false,\n\t\t\trequest:       &ReadHistoryBranchRequest{BranchToken: []byte(\"branch-token\"), PageSize: 10, MinEventID: 1, MaxEventID: 100},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"corrupted history event batch, empty events\",\n\t\t},\n\t\t{\n\t\t\tname: \"corrupted event IDs\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(gomock.Any()).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 3, Version: 1}, // Non-continuous ID\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 1, LastEventID: 0}, 100, log.NewNoop(), nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, nil // Doesn't matter in this case since it errors earlier\n\t\t\t},\n\t\t\tbyBatch:       false,\n\t\t\trequest:       &ReadHistoryBranchRequest{BranchToken: []byte(\"branch-token\"), PageSize: 10, MinEventID: 1, MaxEventID: 100},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"corrupted history event batch, wrong version and IDs\",\n\t\t},\n\t\t{\n\t\t\tname: \"corrupted event batch\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(gomock.Any()).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 7, Version: 5},\n\t\t\t\t\t\t{ID: 8, Version: 5},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 2, LastEventID: 5}, 100, log.NewNoop(), nil // Higher version than event batch\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn []byte(\"next-page-token\"), nil\n\t\t\t},\n\t\t\tbyBatch:       false,\n\t\t\trequest:       &ReadHistoryBranchRequest{BranchToken: []byte(\"branch-token\"), PageSize: 10, MinEventID: 1, MaxEventID: 100},\n\t\t\texpectError:   true, // No error, but events are skipped\n\t\t\texpectedError: \"corrupted history event batch, eventID is not continuous\",\n\t\t},\n\t\t{\n\t\t\tname: \"read raw error\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\t// No DeserializeBatchEvents calls because readRawHistoryBranchFn fails before deserialization\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn nil, nil, 0, nil, errors.New(\"read raw error\")\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, nil // Doesn't matter in this case since it errors earlier\n\t\t\t},\n\t\t\tbyBatch:       false,\n\t\t\trequest:       &ReadHistoryBranchRequest{BranchToken: []byte(\"branch-token\"), PageSize: 10, MinEventID: 1, MaxEventID: 100},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"read raw error\",\n\t\t},\n\t\t{\n\t\t\tname: \"serialize token error\",\n\t\t\tsetupMock: func(mockSerializer *MockPayloadSerializer) {\n\t\t\t\tmockSerializer.EXPECT().\n\t\t\t\t\tDeserializeBatchEvents(gomock.Any()).\n\t\t\t\t\tReturn([]*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-data\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 1, LastEventID: 0}, 100, log.NewNoop(), nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, errors.New(\"serialize token error\")\n\t\t\t},\n\t\t\tbyBatch:       false,\n\t\t\trequest:       &ReadHistoryBranchRequest{BranchToken: []byte(\"branch-token\"), PageSize: 10, MinEventID: 1, MaxEventID: 100},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialize token error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\thistoryManager, _, mockSerializer, _ := setUpMocksForHistoryV2Manager(t)\n\n\t\t\t// Fake readRawHistoryBranchFn and serializeTokenFn\n\t\t\thistoryManager.readRawHistoryBranchFn = tc.fakeReadRaw\n\t\t\thistoryManager.serializeTokenFn = tc.fakeSerializeToken\n\n\t\t\ttc.setupMock(mockSerializer)\n\n\t\t\t// Call the method\n\t\t\thistoryEvents, historyBatches, _, _, _, err := historyManager.readHistoryBranch(context.Background(), tc.byBatch, tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tif tc.byBatch {\n\t\t\t\t\tassert.Equal(t, tc.expectedBatch, historyBatches)\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, tc.expectedEvents, historyEvents)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadRawHistoryBranchMethod(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tsetupMock          func()\n\t\tfakeReadRaw        func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error)\n\t\tfakeSerializeToken func(pagingToken *historyV2PagingToken) ([]byte, error)\n\t\trequest            *ReadHistoryBranchRequest\n\t\texpectError        bool\n\t\texpectedError      string\n\t\texpectedResponse   *ReadRawHistoryBranchResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func() {\n\t\t\t\t// No additional mock setup needed\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-blob\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 1, LastEventID: 0}, 100, nil, nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn []byte(\"next-page-token\"), nil\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedResponse: &ReadRawHistoryBranchResponse{\n\t\t\t\tHistoryEventBlobs: []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-blob\")},\n\t\t\t\t},\n\t\t\t\tNextPageToken: []byte(\"next-page-token\"),\n\t\t\t\tSize:          100,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error in readRawHistoryBranchFn\",\n\t\t\tsetupMock: func() {\n\t\t\t\t// No additional mock setup needed\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn nil, nil, 0, nil, errors.New(\"read error\")\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, nil // Won't be called due to error in readRawHistoryBranchFn\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"read error\",\n\t\t},\n\t\t{\n\t\t\tname: \"error in serializeTokenFn\",\n\t\t\tsetupMock: func() {\n\t\t\t\t// No additional mock setup needed\n\t\t\t},\n\t\t\tfakeReadRaw: func(ctx context.Context, request *ReadHistoryBranchRequest) ([]*DataBlob, *historyV2PagingToken, int, log.Logger, error) {\n\t\t\t\treturn []*DataBlob{\n\t\t\t\t\t{Data: []byte(\"history-event-blob\")},\n\t\t\t\t}, &historyV2PagingToken{LastEventVersion: 1, LastEventID: 0}, 100, nil, nil\n\t\t\t},\n\t\t\tfakeSerializeToken: func(pagingToken *historyV2PagingToken) ([]byte, error) {\n\t\t\t\treturn nil, errors.New(\"serialization error\")\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"serialization error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up the mock dependencies and fakes\n\t\t\thistoryManager, _, _, _ := setUpMocksForHistoryV2Manager(t)\n\t\t\thistoryManager.readRawHistoryBranchFn = tc.fakeReadRaw\n\t\t\thistoryManager.serializeTokenFn = tc.fakeSerializeToken\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.ReadRawHistoryBranch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResponse, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadHistoryBranchByBatch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\tfakeReadHistory  func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error)\n\t\trequest          *ReadHistoryBranchRequest\n\t\texpectError      bool\n\t\texpectedError    string\n\t\texpectedResponse *ReadHistoryBranchByBatchResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfakeReadHistory: func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error) {\n\t\t\t\treturn nil, []*types.History{\n\t\t\t\t\t{Events: []*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t\t}},\n\t\t\t\t}, []byte(\"next-page-token\"), 100, 1, nil\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedResponse: &ReadHistoryBranchByBatchResponse{\n\t\t\t\tHistory: []*types.History{\n\t\t\t\t\t{Events: []*types.HistoryEvent{\n\t\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t\tNextPageToken:    []byte(\"next-page-token\"),\n\t\t\t\tSize:             100,\n\t\t\t\tLastFirstEventID: 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error in readHistoryBranchFn\",\n\t\t\tfakeReadHistory: func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error) {\n\t\t\t\treturn nil, nil, nil, 0, 0, errors.New(\"read error\")\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"read error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up faked readHistoryBranchFn\n\t\t\thistoryManager, _, _, _ := setUpMocksForHistoryV2Manager(t)\n\t\t\thistoryManager.readHistoryBranchFn = tc.fakeReadHistory\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.ReadHistoryBranchByBatch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResponse, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadHistoryBranchMethod(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\tfakeReadHistory  func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error)\n\t\trequest          *ReadHistoryBranchRequest\n\t\texpectError      bool\n\t\texpectedError    string\n\t\texpectedResponse *ReadHistoryBranchResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfakeReadHistory: func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error) {\n\t\t\t\treturn []*types.HistoryEvent{\n\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t}, nil, []byte(\"next-page-token\"), 100, 1, nil\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedResponse: &ReadHistoryBranchResponse{\n\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t{ID: 1, Version: 1},\n\t\t\t\t\t{ID: 2, Version: 1},\n\t\t\t\t},\n\t\t\t\tNextPageToken:    []byte(\"next-page-token\"),\n\t\t\t\tSize:             100,\n\t\t\t\tLastFirstEventID: 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error in readHistoryBranchFn\",\n\t\t\tfakeReadHistory: func(ctx context.Context, byBatch bool, request *ReadHistoryBranchRequest) ([]*types.HistoryEvent, []*types.History, []byte, int, int64, error) {\n\t\t\t\treturn nil, nil, nil, 0, 0, errors.New(\"read error\")\n\t\t\t},\n\t\t\trequest: &ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\tPageSize:      10,\n\t\t\t\tMinEventID:    1,\n\t\t\t\tMaxEventID:    100,\n\t\t\t\tNextPageToken: []byte{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"read error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Set up faked readHistoryBranchFn\n\t\t\thistoryManager, _, _, _ := setUpMocksForHistoryV2Manager(t)\n\t\t\thistoryManager.readHistoryBranchFn = tc.fakeReadHistory\n\n\t\t\t// Call the method\n\t\t\tresp, err := historyManager.ReadHistoryBranch(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResponse, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeserializeToken(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\tinputData          []byte\n\t\tdefaultLastEventID int64\n\t\texpectedToken      *historyV2PagingToken\n\t\texpectError        bool\n\t\texpectedError      string\n\t}{\n\t\t{\n\t\t\tname:               \"empty data\",\n\t\t\tinputData:          []byte{},\n\t\t\tdefaultLastEventID: 100,\n\t\t\texpectedToken: &historyV2PagingToken{\n\t\t\t\tLastEventID:       100,\n\t\t\t\tLastEventVersion:  constants.EmptyVersion,\n\t\t\t\tCurrentRangeIndex: notStartedIndex,\n\t\t\t\tLastNodeID:        defaultLastNodeID,\n\t\t\t\tLastTransactionID: defaultLastTransactionID,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:               \"valid token data\",\n\t\t\tinputData:          []byte(`{\"LastEventID\":200,\"LastEventVersion\":1,\"CurrentRangeIndex\":2,\"FinalRangeIndex\":3,\"LastNodeID\":100,\"LastTransactionID\":500}`),\n\t\t\tdefaultLastEventID: 100,\n\t\t\texpectedToken: &historyV2PagingToken{\n\t\t\t\tLastEventID:       200,\n\t\t\t\tLastEventVersion:  1,\n\t\t\t\tCurrentRangeIndex: 2,\n\t\t\t\tFinalRangeIndex:   3,\n\t\t\t\tLastNodeID:        100,\n\t\t\t\tLastTransactionID: 500,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:               \"invalid json data\",\n\t\t\tinputData:          []byte(\"invalid-json\"),\n\t\t\tdefaultLastEventID: 100,\n\t\t\texpectedToken:      nil,\n\t\t\texpectError:        true,\n\t\t\texpectedError:      \"invalid character 'i' looking for beginning of value\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttoken, err := deserializeToken(tc.inputData, tc.defaultLastEventID)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedToken, token)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSerializeToken(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tinputToken     *historyV2PagingToken\n\t\texpectedOutput []byte\n\t\texpectError    bool\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"empty store token, final page reached\",\n\t\t\tinputToken: &historyV2PagingToken{\n\t\t\t\tStoreToken:        []byte{},\n\t\t\t\tCurrentRangeIndex: 2,\n\t\t\t\tFinalRangeIndex:   2,\n\t\t\t},\n\t\t\texpectedOutput: nil,\n\t\t\texpectError:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"empty store token, increment current range index\",\n\t\t\tinputToken: &historyV2PagingToken{\n\t\t\t\tStoreToken:        []byte{},\n\t\t\t\tCurrentRangeIndex: 1,\n\t\t\t\tFinalRangeIndex:   2,\n\t\t\t},\n\t\t\texpectedOutput: []byte(`{\"LastEventVersion\":0,\"LastEventID\":0,\"StoreToken\":\"\",\"CurrentRangeIndex\":2,\"FinalRangeIndex\":2,\"LastNodeID\":0,\"LastTransactionID\":0}`),\n\t\t\texpectError:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"non-empty store token\",\n\t\t\tinputToken: &historyV2PagingToken{\n\t\t\t\tStoreToken:        []byte(\"non-empty-token\"),\n\t\t\t\tCurrentRangeIndex: 1,\n\t\t\t\tFinalRangeIndex:   2,\n\t\t\t},\n\t\t\texpectedOutput: []byte(`{\"LastEventVersion\":0,\"LastEventID\":0,\"StoreToken\":\"bm9uLWVtcHR5LXRva2Vu\",\"CurrentRangeIndex\":1,\"FinalRangeIndex\":2,\"LastNodeID\":0,\"LastTransactionID\":0}`),\n\t\t\texpectError:    false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\toutput, err := serializeToken(tc.inputToken)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedOutput, output)\n\t\t\t\tt.Log(string(output))\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTokenRoundtrip(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tinputToken *historyV2PagingToken\n\t}{\n\t\t{\n\t\t\tname: \"empty store token, basic token\",\n\t\t\tinputToken: &historyV2PagingToken{\n\t\t\t\tLastEventID:       100,\n\t\t\t\tLastEventVersion:  1,\n\t\t\t\tCurrentRangeIndex: 1,\n\t\t\t\tFinalRangeIndex:   2,\n\t\t\t\tLastNodeID:        200,\n\t\t\t\tLastTransactionID: 300,\n\t\t\t\tStoreToken:        []byte{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"non-empty store token\",\n\t\t\tinputToken: &historyV2PagingToken{\n\t\t\t\tLastEventID:       100,\n\t\t\t\tLastEventVersion:  1,\n\t\t\t\tCurrentRangeIndex: 1,\n\t\t\t\tFinalRangeIndex:   2,\n\t\t\t\tLastNodeID:        200,\n\t\t\t\tLastTransactionID: 300,\n\t\t\t\tStoreToken:        []byte(\"non-empty-token\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Step 1: Serialize the token\n\t\t\tserializedToken, err := serializeToken(tc.inputToken)\n\t\t\tassert.NoError(t, err)\n\n\t\t\t// Step 2: Deserialize the token back\n\t\t\tdeserializedToken, err := deserializeToken(serializedToken, tc.inputToken.LastEventID)\n\t\t\tassert.NoError(t, err)\n\n\t\t\t// Step 3: Compare the original token with the deserialized token\n\t\t\tassert.Equal(t, tc.inputToken, deserializedToken)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/mappers.go",
    "content": "package persistence\n\nimport \"github.com/uber/cadence/common/types\"\n\n// ToType converts persistence.GetDomainResponse to types.DescribeDomainResponse. Take care when using this mapper\n// as not all the fields between the two types are symmetrical\n// todo: Replace the ad-hoc mapping in the DescribeDomain Handler with this one\nfunc (r *GetDomainResponse) ToType() *types.DescribeDomainResponse {\n\tif r == nil {\n\t\treturn nil\n\t}\n\n\tvar domainInfo *types.DomainInfo\n\tif r.Info != nil {\n\t\t// Validate and clamp Status to valid range (0-2: Registered, Deprecated, Deleted)\n\t\t// This prevents issues with invalid or out-of-range Status values\n\t\tstatus := r.Info.Status\n\t\tif status < 0 || status > 2 {\n\t\t\tstatus = 0 // Default to Registered for invalid values\n\t\t}\n\t\tdomainStatus := types.DomainStatus(status)\n\t\tdomainInfo = &types.DomainInfo{\n\t\t\tName:        r.Info.Name,\n\t\t\tStatus:      &domainStatus,\n\t\t\tDescription: r.Info.Description,\n\t\t\tOwnerEmail:  r.Info.OwnerEmail,\n\t\t\tData:        r.Info.Data,\n\t\t\tUUID:        r.Info.ID,\n\t\t}\n\t}\n\n\tvar domainConfig *types.DomainConfiguration\n\tif r.Config != nil {\n\t\tdomainConfig = &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: r.Config.Retention,\n\t\t\tEmitMetric:                             r.Config.EmitMetric,\n\t\t\tHistoryArchivalStatus:                  r.Config.HistoryArchivalStatus.Ptr(),\n\t\t\tHistoryArchivalURI:                     r.Config.HistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus:               r.Config.VisibilityArchivalStatus.Ptr(),\n\t\t\tVisibilityArchivalURI:                  r.Config.VisibilityArchivalURI,\n\t\t}\n\t\t// Always set these pointers to maintain symmetry with FromDescribeDomainResponse\n\t\t// which always copies them when present\n\t\tdomainConfig.BadBinaries = &r.Config.BadBinaries\n\t\tdomainConfig.IsolationGroups = &r.Config.IsolationGroups\n\t\tdomainConfig.AsyncWorkflowConfig = &r.Config.AsyncWorkflowConfig\n\t}\n\n\tvar replicationConfig *types.DomainReplicationConfiguration\n\tif r.ReplicationConfig != nil {\n\t\tvar clusters []*types.ClusterReplicationConfiguration\n\t\t// Convert clusters, preserving nil entries for consistent roundtrips\n\t\tif r.ReplicationConfig.Clusters != nil {\n\t\t\tclusters = make([]*types.ClusterReplicationConfiguration, len(r.ReplicationConfig.Clusters))\n\t\t\tfor i, cluster := range r.ReplicationConfig.Clusters {\n\t\t\t\tif cluster != nil {\n\t\t\t\t\tclusters[i] = &types.ClusterReplicationConfiguration{\n\t\t\t\t\t\tClusterName: cluster.ClusterName,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treplicationConfig = &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: r.ReplicationConfig.ActiveClusterName,\n\t\t\tClusters:          clusters,\n\t\t\tActiveClusters:    r.ReplicationConfig.ActiveClusters,\n\t\t}\n\t}\n\n\t// Always create FailoverInfo for consistent mapping\n\tfailoverExpireTimestamp := int64(0)\n\tif r.FailoverEndTime != nil {\n\t\tfailoverExpireTimestamp = *r.FailoverEndTime\n\t}\n\n\treturn &types.DescribeDomainResponse{\n\t\tDomainInfo:               domainInfo,\n\t\tConfiguration:            domainConfig,\n\t\tReplicationConfiguration: replicationConfig,\n\t\tFailoverVersion:          r.FailoverVersion,\n\t\tIsGlobalDomain:           r.IsGlobalDomain,\n\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\tFailoverVersion:         r.FailoverVersion,\n\t\t\tFailoverStartTimestamp:  r.LastUpdatedTime,\n\t\t\tFailoverExpireTimestamp: failoverExpireTimestamp,\n\t\t\t// Fields below are not present at the persistence layer\n\t\t\tCompletedShardCount: 0,\n\t\t\tPendingShards:       nil,\n\t\t},\n\t\t// note that failoverNotificationVersion is not present because the structs aren't symmetric\n\t}\n}\n\n// FromDescribeDomainResponse converts types.DescribeDomainResponse to persistence.GetDomainResponse\n// care should be taken when using this because not all the fields between the two types are symmetrical\nfunc FromDescribeDomainResponse(resp *types.DescribeDomainResponse) *GetDomainResponse {\n\tif resp == nil {\n\t\treturn nil\n\t}\n\n\tvar domainInfo *DomainInfo\n\tif resp.DomainInfo != nil {\n\t\tvar status int\n\t\tif resp.DomainInfo.Status != nil {\n\t\t\tstatus = int(*resp.DomainInfo.Status)\n\t\t}\n\t\tdomainInfo = &DomainInfo{\n\t\t\tID:          resp.DomainInfo.UUID,\n\t\t\tName:        resp.DomainInfo.Name,\n\t\t\tStatus:      status,\n\t\t\tDescription: resp.DomainInfo.Description,\n\t\t\tOwnerEmail:  resp.DomainInfo.OwnerEmail,\n\t\t\tData:        resp.DomainInfo.Data,\n\t\t}\n\t}\n\n\tvar domainConfig *DomainConfig\n\tif resp.Configuration != nil {\n\t\tconfig := &DomainConfig{\n\t\t\tRetention:  resp.Configuration.WorkflowExecutionRetentionPeriodInDays,\n\t\t\tEmitMetric: resp.Configuration.EmitMetric,\n\t\t}\n\t\tif resp.Configuration.HistoryArchivalStatus != nil {\n\t\t\tconfig.HistoryArchivalStatus = *resp.Configuration.HistoryArchivalStatus\n\t\t}\n\t\tconfig.HistoryArchivalURI = resp.Configuration.HistoryArchivalURI\n\t\tif resp.Configuration.VisibilityArchivalStatus != nil {\n\t\t\tconfig.VisibilityArchivalStatus = *resp.Configuration.VisibilityArchivalStatus\n\t\t}\n\t\tconfig.VisibilityArchivalURI = resp.Configuration.VisibilityArchivalURI\n\t\tif resp.Configuration.BadBinaries != nil {\n\t\t\tconfig.BadBinaries = *resp.Configuration.BadBinaries\n\t\t}\n\t\tif resp.Configuration.IsolationGroups != nil {\n\t\t\tconfig.IsolationGroups = *resp.Configuration.IsolationGroups\n\t\t}\n\t\tif resp.Configuration.AsyncWorkflowConfig != nil {\n\t\t\tconfig.AsyncWorkflowConfig = *resp.Configuration.AsyncWorkflowConfig\n\t\t}\n\t\tdomainConfig = config\n\t}\n\n\tvar replicationConfig *DomainReplicationConfig\n\tif resp.ReplicationConfiguration != nil {\n\t\tvar clusters []*ClusterReplicationConfig\n\t\t// Convert clusters, preserving nil entries for consistent roundtrips\n\t\tif resp.ReplicationConfiguration.Clusters != nil {\n\t\t\tclusters = make([]*ClusterReplicationConfig, len(resp.ReplicationConfiguration.Clusters))\n\t\t\tfor i, cluster := range resp.ReplicationConfiguration.Clusters {\n\t\t\t\tif cluster != nil {\n\t\t\t\t\tclusters[i] = &ClusterReplicationConfig{\n\t\t\t\t\t\tClusterName: cluster.ClusterName,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treplicationConfig = &DomainReplicationConfig{\n\t\t\tActiveClusterName: resp.ReplicationConfiguration.ActiveClusterName,\n\t\t\tClusters:          clusters,\n\t\t\tActiveClusters:    resp.ReplicationConfiguration.ActiveClusters,\n\t\t}\n\t}\n\n\tresult := &GetDomainResponse{\n\t\tInfo:              domainInfo,\n\t\tConfig:            domainConfig,\n\t\tReplicationConfig: replicationConfig,\n\t\tIsGlobalDomain:    resp.IsGlobalDomain,\n\t\tFailoverVersion:   resp.FailoverVersion,\n\t\t// FailoverNotificationVersion not mapped - not present in types.DescribeDomainResponse\n\t\t// PreviousFailoverVersion not mapped - not present in types.DescribeDomainResponse\n\t\t// FailoverEndTime not mapped - not reliably preserved in roundtrips\n\t\t// ConfigVersion not mapped - not present in types.DescribeDomainResponse\n\t\t// NotificationVersion not mapped - not present in types.DescribeDomainResponse\n\t}\n\n\tif resp.FailoverInfo != nil {\n\t\tresult.LastUpdatedTime = resp.FailoverInfo.FailoverStartTimestamp\n\t}\n\n\treturn result\n}\n"
  },
  {
    "path": "common/persistence/mappers_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetDomainResponse_ToType(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *GetDomainResponse\n\t\texpected *types.DescribeDomainResponse\n\t}{\n\t\t\"nil response\": {\n\t\t\tinput:    nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty response\": {\n\t\t\tinput: &GetDomainResponse{},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         0,\n\t\t\t\t\tFailoverStartTimestamp:  0,\n\t\t\t\t\tFailoverExpireTimestamp: 0,\n\t\t\t\t\tCompletedShardCount:     0,\n\t\t\t\t\tPendingShards:           nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"full response with all fields\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:          \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      0, // 0 = DomainStatusRegistered\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                7,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"test://visibility\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:        \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"test://visibility\",\n\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         100,\n\t\t\t\t\tFailoverStartTimestamp:  0,\n\t\t\t\t\tFailoverExpireTimestamp: 0,\n\t\t\t\t\tCompletedShardCount:     0,\n\t\t\t\t\tPendingShards:           nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"with failover info\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0, // 0 = DomainStatusRegistered\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tLastUpdatedTime: 1000,\n\t\t\t\tFailoverEndTime: func() *int64 { t := int64(2000); return &t }(),\n\t\t\t},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         200,\n\t\t\t\t\tFailoverStartTimestamp:  1000,\n\t\t\t\t\tFailoverExpireTimestamp: 2000,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"without failover end time\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0, // 0 = DomainStatusRegistered\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tLastUpdatedTime: 1000,\n\t\t\t\tFailoverEndTime: nil,\n\t\t\t},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         200,\n\t\t\t\t\tFailoverStartTimestamp:  1000,\n\t\t\t\t\tFailoverExpireTimestamp: 0,\n\t\t\t\t\tCompletedShardCount:     0,\n\t\t\t\t\tPendingShards:           nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil Info\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo:            nil,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo:      nil,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         100,\n\t\t\t\t\tFailoverStartTimestamp:  0,\n\t\t\t\t\tFailoverExpireTimestamp: 0,\n\t\t\t\t\tCompletedShardCount:     0,\n\t\t\t\t\tPendingShards:           nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil clusters in ReplicationConfig\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0,\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\tnil, // Preserved in output\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t\tnil, // Preserved in output\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         0,\n\t\t\t\t\tFailoverStartTimestamp:  0,\n\t\t\t\t\tFailoverExpireTimestamp: 0,\n\t\t\t\t\tCompletedShardCount:     0,\n\t\t\t\t\tPendingShards:           nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := tc.input.ToType()\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestFromDescribeDomainResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *types.DescribeDomainResponse\n\t\texpected *GetDomainResponse\n\t}{\n\t\t\"nil response\": {\n\t\t\tinput:    nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty response\": {\n\t\t\tinput:    &types.DescribeDomainResponse{},\n\t\t\texpected: &GetDomainResponse{},\n\t\t},\n\t\t\"full response with all fields\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:        \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"test://visibility\",\n\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:          \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      0, // 0 = DomainStatusRegistered\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                7,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"test://visibility\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t},\n\t\t\"with failover info\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         200,\n\t\t\t\t\tFailoverStartTimestamp:  1000,\n\t\t\t\t\tFailoverExpireTimestamp: 2000,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0, // 0 = DomainStatusRegistered\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tLastUpdatedTime: 1000,\n\t\t\t\t// FailoverEndTime not mapped from FailoverInfo\n\t\t\t},\n\t\t},\n\t\t\"nil DomainInfo\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo:      nil,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo:            nil,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t},\n\t\t\"nil Status\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: nil, // Should default to 0\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0, // Defaults to 0\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil clusters in ReplicationConfiguration\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\tnil, // Preserved in output\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t\tnil, // Preserved in output\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0,\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"cluster-1\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := FromDescribeDomainResponse(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestRoundtrip_ToType_FromDescribeDomainResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *GetDomainResponse\n\t}{\n\t\t\"empty response\": {\n\t\t\tinput: &GetDomainResponse{},\n\t\t},\n\t\t\"full response\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:          \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      0, // 0 = DomainStatusRegistered\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\tRetention:                7,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"test://visibility\",\n\t\t\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t},\n\t\t\"with failover\": {\n\t\t\tinput: &GetDomainResponse{\n\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\tID:     \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: 0, // 0 = DomainStatusRegistered\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tLastUpdatedTime: 1000,\n\t\t\t\tFailoverEndTime: func() *int64 { t := int64(2000); return &t }(),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Convert GetDomainResponse -> DescribeDomainResponse -> GetDomainResponse\n\t\t\ttypesResp := tc.input.ToType()\n\t\t\troundtripped := FromDescribeDomainResponse(typesResp)\n\n\t\t\t// The roundtrip should preserve all the data that's in both representations\n\t\t\t// Note: Some fields may not survive the roundtrip if they only exist in one direction\n\t\t\tassert.Equal(t, tc.input.Info, roundtripped.Info)\n\t\t\tassert.Equal(t, tc.input.Config, roundtripped.Config)\n\t\t\tassert.Equal(t, tc.input.ReplicationConfig, roundtripped.ReplicationConfig)\n\t\t\tassert.Equal(t, tc.input.IsGlobalDomain, roundtripped.IsGlobalDomain)\n\t\t\tassert.Equal(t, tc.input.FailoverVersion, roundtripped.FailoverVersion)\n\t\t\tassert.Equal(t, tc.input.LastUpdatedTime, roundtripped.LastUpdatedTime)\n\t\t\t// FailoverEndTime is not preserved in roundtrip\n\t\t})\n\t}\n}\n\nfunc TestRoundtrip_FromDescribeDomainResponse_ToType(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *types.DescribeDomainResponse\n\t}{\n\t\t\"empty response\": {\n\t\t\tinput: &types.DescribeDomainResponse{},\n\t\t},\n\t\t\"full response\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:        \"test-domain-id\",\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tStatus:      types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\tDescription: \"test description\",\n\t\t\t\t\tOwnerEmail:  \"test@example.com\",\n\t\t\t\t\tData:        map[string]string{\"key\": \"value\"},\n\t\t\t\t},\n\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 7,\n\t\t\t\t\tEmitMetric:                             true,\n\t\t\t\t\tHistoryArchivalStatus:                  types.ArchivalStatusEnabled.Ptr(),\n\t\t\t\t\tHistoryArchivalURI:                     \"test://history\",\n\t\t\t\t\tVisibilityArchivalStatus:               types.ArchivalStatusDisabled.Ptr(),\n\t\t\t\t\tVisibilityArchivalURI:                  \"test://visibility\",\n\t\t\t\t\tBadBinaries:                            &types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"bad\": {Reason: \"test\"}}},\n\t\t\t\t\tIsolationGroups:                        &types.IsolationGroupConfiguration{},\n\t\t\t\t\tAsyncWorkflowConfig:                    &types.AsyncWorkflowConfiguration{Enabled: true},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"cluster-1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster-2\"},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"us-east\": {ActiveClusterName: \"cluster-1\", FailoverVersion: 100},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t\tFailoverVersion: 100,\n\t\t\t},\n\t\t},\n\t\t\"with failover info\": {\n\t\t\tinput: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tUUID:   \"test-id\",\n\t\t\t\t\tName:   \"test\",\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tFailoverVersion: 200,\n\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\tFailoverVersion:         200,\n\t\t\t\t\tFailoverStartTimestamp:  1000,\n\t\t\t\t\tFailoverExpireTimestamp: 2000,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Convert DescribeDomainResponse -> GetDomainResponse -> DescribeDomainResponse\n\t\t\tpersistenceResp := FromDescribeDomainResponse(tc.input)\n\t\t\troundtripped := persistenceResp.ToType()\n\n\t\t\t// The roundtrip should preserve all the data that's in both representations\n\t\t\tassert.Equal(t, tc.input.DomainInfo, roundtripped.DomainInfo)\n\t\t\tassert.Equal(t, tc.input.Configuration, roundtripped.Configuration)\n\t\t\tassert.Equal(t, tc.input.ReplicationConfiguration, roundtripped.ReplicationConfiguration)\n\t\t\tassert.Equal(t, tc.input.IsGlobalDomain, roundtripped.IsGlobalDomain)\n\t\t\tassert.Equal(t, tc.input.FailoverVersion, roundtripped.FailoverVersion)\n\t\t\t// FailoverInfo is always created in ToType(), so check fields separately\n\t\t\tassert.NotNil(t, roundtripped.FailoverInfo)\n\t\t\tif tc.input.FailoverInfo != nil {\n\t\t\t\tassert.Equal(t, tc.input.FailoverInfo.FailoverStartTimestamp, roundtripped.FailoverInfo.FailoverStartTimestamp)\n\t\t\t\t// Other FailoverInfo fields (CompletedShardCount, PendingShards) are not present at persistence layer\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRoundtrip_FuzzGenerated(t *testing.T) {\n\tfuzzer := testdatagen.New(t)\n\n\t// Run multiple iterations to test with different randomly generated data\n\tfor i := 0; i < 100; i++ {\n\t\tt.Run(fmt.Sprintf(\"iteration_%d\", i), func(t *testing.T) {\n\t\t\t// Test GetDomainResponse -> DescribeDomainResponse -> GetDomainResponse\n\t\t\toriginal := &GetDomainResponse{}\n\t\t\tfuzzer.Fuzz(original)\n\n\t\t\ttypesResp := original.ToType()\n\t\t\trequire.NotNil(t, typesResp)\n\n\t\t\troundtripped := FromDescribeDomainResponse(typesResp)\n\t\t\trequire.NotNil(t, roundtripped)\n\n\t\t\t// Build expected result accounting for non-symmetric fields\n\t\t\texpected := &GetDomainResponse{\n\t\t\t\tInfo:              original.Info,\n\t\t\t\tConfig:            original.Config,\n\t\t\t\tReplicationConfig: original.ReplicationConfig,\n\t\t\t\tIsGlobalDomain:    original.IsGlobalDomain,\n\t\t\t\tFailoverVersion:   original.FailoverVersion,\n\t\t\t\tLastUpdatedTime:   original.LastUpdatedTime,\n\t\t\t\t// Fields below are not present in types.DescribeDomainResponse and don't roundtrip:\n\t\t\t\tFailoverNotificationVersion: 0,   // Not mapped\n\t\t\t\tPreviousFailoverVersion:     0,   // Not mapped\n\t\t\t\tFailoverEndTime:             nil, // Not mapped\n\t\t\t\tConfigVersion:               0,   // Not mapped\n\t\t\t\tNotificationVersion:         0,   // Not mapped\n\t\t\t}\n\n\t\t\t// Apply transformations that the mapper does\n\t\t\tif expected.Info != nil {\n\t\t\t\t// Clamp Status to valid range (0-2) as ToType does\n\t\t\t\tif expected.Info.Status < 0 || expected.Info.Status > 2 {\n\t\t\t\t\texpected.Info.Status = 0\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tassert.Equal(t, expected, roundtripped)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/metered.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"unsafe\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// This file defines method for persistence requests/responses that affects metered persistence wrapper.\n\n// For responses that require metrics for empty response Len() int should be defined.\n\nfunc (r *GetHistoryTasksResponse) Len() int {\n\treturn len(r.Tasks)\n}\n\nfunc (r *GetTasksResponse) Len() int {\n\treturn len(r.Tasks)\n}\n\nfunc (r *ListDomainsResponse) Len() int {\n\treturn len(r.Domains)\n}\n\nfunc (r *ReadHistoryBranchResponse) Len() int {\n\treturn len(r.HistoryEvents)\n}\n\nfunc (r *ListCurrentExecutionsResponse) Len() int {\n\treturn len(r.Executions)\n}\n\nfunc (r QueueMessageList) Len() int {\n\treturn len(r)\n}\n\nfunc (r GetAllHistoryTreeBranchesResponse) Len() int {\n\treturn len(r.Branches)\n}\n\n// For responses that require metrics for payload size ByteSize() uint64 should be defined.\n\nfunc (r *GetHistoryTasksResponse) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(int(unsafe.Sizeof(*r)) + len(r.NextPageToken))\n\tfor _, v := range r.Tasks {\n\t\tif v != nil {\n\t\t\tsize += v.ByteSize()\n\t\t}\n\t}\n\n\treturn size\n}\n\nfunc (r *GetTasksResponse) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*r))\n\tfor _, v := range r.Tasks {\n\t\tsize += v.ByteSize()\n\t}\n\n\treturn size\n}\n\nfunc (r *TaskInfo) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(int(unsafe.Sizeof(*r)) + len(r.DomainID) + len(r.WorkflowID) + len(r.RunID) + estimateStringMapSize(r.PartitionConfig))\n}\n\nfunc (r *ListDomainsResponse) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(int(unsafe.Sizeof(*r)) + len(r.NextPageToken))\n\tfor _, v := range r.Domains {\n\t\tsize += v.ByteSize()\n\t}\n\n\treturn size\n}\n\nfunc (r *GetDomainResponse) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(unsafe.Sizeof(*r)) + r.Info.ByteSize() + r.Config.ByteSize() + r.ReplicationConfig.ByteSize()\n}\n\nfunc (i *DomainInfo) ByteSize() uint64 {\n\tif i == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(int(unsafe.Sizeof(*i)) + len(i.ID) + len(i.Name) + len(i.Description) + len(i.OwnerEmail) + estimateStringMapSize(i.Data))\n}\n\nfunc (c *DomainConfig) ByteSize() uint64 {\n\tif c == nil {\n\t\treturn 0\n\t}\n\n\tsize := int(unsafe.Sizeof(*c)) + len(c.HistoryArchivalURI) + len(c.VisibilityArchivalURI)\n\n\tasyncWorkflowConfigSize := int(unsafe.Sizeof(c.AsyncWorkflowConfig)) + len(c.AsyncWorkflowConfig.PredefinedQueueName) + len(c.AsyncWorkflowConfig.QueueType)\n\tif c.AsyncWorkflowConfig.QueueConfig != nil {\n\t\tsize += len(c.AsyncWorkflowConfig.QueueConfig.Data)\n\t}\n\n\tbinariesSize := 0\n\tfor key, value := range c.BadBinaries.Binaries {\n\t\tbinariesSize += len(key)\n\t\tif value != nil {\n\t\t\tbinariesSize += len(value.Reason) + len(value.Operator)\n\t\t}\n\t}\n\n\tisolationGroupsSize := 0\n\tfor key, value := range c.IsolationGroups {\n\t\tbinariesSize += len(key) + len(value.Name)\n\t}\n\n\treturn uint64(size + asyncWorkflowConfigSize + binariesSize + isolationGroupsSize)\n}\n\nfunc (c *DomainReplicationConfig) ByteSize() uint64 {\n\tif c == nil {\n\t\treturn 0\n\t}\n\n\ttotal := len(c.ActiveClusterName)\n\tfor _, v := range c.Clusters {\n\t\tif v == nil {\n\t\t\tcontinue\n\t\t}\n\t\ttotal += len(v.ClusterName)\n\t}\n\treturn uint64(total)\n}\n\nfunc estimateStringMapSize(m map[string]string) int {\n\tsize := 0\n\tfor key, value := range m {\n\t\tsize += len(key) + len(value)\n\t}\n\treturn size\n}\n\nfunc (r *ReadRawHistoryBranchResponse) Size2() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\ttotal := uint64(int(unsafe.Sizeof(*r)) + len(r.NextPageToken))\n\tfor _, v := range r.HistoryEventBlobs {\n\t\ttotal += v.ByteSize()\n\t}\n\treturn total\n}\n\nfunc (d *DataBlob) ByteSize() uint64 {\n\tif d == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(int(unsafe.Sizeof(*d)) + len(d.Data))\n}\n\nfunc (r *ListCurrentExecutionsResponse) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\ttotal := uint64(int(unsafe.Sizeof(*r)) + len(r.PageToken))\n\tfor _, v := range r.Executions {\n\t\ttotal += v.ByteSize()\n\t}\n\n\treturn total\n}\n\nfunc (r *CurrentWorkflowExecution) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(int(unsafe.Sizeof(*r)) + len(r.DomainID) + len(r.WorkflowID) + len(r.RunID) + len(r.CurrentRunID))\n}\n\nfunc (r QueueMessageList) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\ttotal := uint64(0)\n\tfor _, v := range r {\n\t\ttotal += v.ByteSize()\n\t}\n\n\treturn total\n}\n\nfunc (r *QueueMessage) ByteSize() uint64 {\n\tif r == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(int(unsafe.Sizeof(*r)) + len(r.Payload))\n}\n\nfunc (r GetAllHistoryTreeBranchesResponse) ByteSize() uint64 {\n\ttotal := uint64(int(unsafe.Sizeof(r)) + len(r.NextPageToken))\n\tfor _, v := range r.Branches {\n\t\ttotal += v.ByteSize()\n\t}\n\n\treturn total\n}\n\nfunc (r HistoryBranchDetail) ByteSize() uint64 {\n\treturn uint64(int(unsafe.Sizeof(r)) + len(r.TreeID) + len(r.BranchID) + len(r.Info))\n}\n\n// If MetricTags() []metrics.Tag is defined, then metrics will be emitted for the request.\n\nfunc (r ReadHistoryBranchRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r AppendHistoryNodesRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r DeleteHistoryBranchRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r ForkHistoryBranchRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r GetHistoryTreeRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r CompleteTaskRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r CompleteTasksLessThanRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r CreateTasksRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r DeleteTaskListRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r GetTasksRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r LeaseTaskListRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r UpdateTaskListRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r GetTaskListSizeRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\n// For Execution manager there are extra rules.\n// If GetDomainName() string is defined, then the request will have extra log and shard metrics.\n// GetExtraLogTags() []tag.Tag is defined, then the request will have extra log tags.\n\nfunc (r *CreateWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r *IsWorkflowExecutionExistsRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r *PutReplicationTaskToDLQRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.DomainTag(r.DomainName)}\n}\n\nfunc (r *CreateWorkflowExecutionRequest) GetExtraLogTags() []tag.Tag {\n\tif r == nil || r.NewWorkflowSnapshot.ExecutionInfo == nil {\n\t\treturn nil\n\t}\n\treturn []tag.Tag{tag.WorkflowID(r.NewWorkflowSnapshot.ExecutionInfo.WorkflowID)}\n}\n\nfunc (r *UpdateWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r *UpdateWorkflowExecutionRequest) GetExtraLogTags() []tag.Tag {\n\tif r == nil || r.UpdateWorkflowMutation.ExecutionInfo == nil {\n\t\treturn nil\n\t}\n\treturn []tag.Tag{tag.WorkflowID(r.UpdateWorkflowMutation.ExecutionInfo.WorkflowID)}\n}\n\nfunc (r GetWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r GetWorkflowExecutionRequest) GetExtraLogTags() []tag.Tag {\n\treturn []tag.Tag{tag.WorkflowID(r.Execution.WorkflowID)}\n}\n\nfunc (r ConflictResolveWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r DeleteWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r DeleteWorkflowExecutionRequest) GetExtraLogTags() []tag.Tag {\n\treturn []tag.Tag{tag.WorkflowID(r.WorkflowID)}\n}\n\nfunc (r DeleteCurrentWorkflowExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r DeleteCurrentWorkflowExecutionRequest) GetExtraLogTags() []tag.Tag {\n\treturn []tag.Tag{tag.WorkflowID(r.WorkflowID)}\n}\n\nfunc (r GetCurrentExecutionRequest) GetDomainName() string {\n\treturn r.DomainName\n}\n\nfunc (r GetCurrentExecutionRequest) GetExtraLogTags() []tag.Tag {\n\treturn []tag.Tag{tag.WorkflowID(r.WorkflowID)}\n}\n\nfunc (r GetHistoryTasksRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.TaskCategoryTag(r.TaskCategory.Name())}\n}\n\nfunc (r RangeCompleteHistoryTaskRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.TaskCategoryTag(r.TaskCategory.Name())}\n}\n\nfunc (r CompleteHistoryTaskRequest) MetricTags() []metrics.Tag {\n\treturn []metrics.Tag{metrics.TaskCategoryTag(r.TaskCategory.Name())}\n}\n"
  },
  {
    "path": "common/persistence/metered_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetTasksResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &GetTasksResponse{}\n\t\t\t\tfuzzer.Fuzz(&response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := &GetTasksResponse{\n\t\t\tTasks: []*TaskInfo{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"domainID\", WorkflowID: \"workflowID\", RunID: \"runID\",\n\t\t\t\t\tTaskID: 0, ScheduleID: 0, ScheduleToStartTimeoutSeconds: 0,\n\t\t\t\t\tExpiry:          time.Time{},\n\t\t\t\t\tCreatedTime:     time.Time{},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tassert.Equal(t, uint64(175), response.ByteSize())\n\t})\n\n\tt.Run(\"response with bigger payload emits a bigger value\", func(t *testing.T) {\n\t\tresponse := &GetTasksResponse{\n\t\t\tTasks: []*TaskInfo{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"domainID\", WorkflowID: \"workflowID\", RunID: \"runID\",\n\t\t\t\t\tTaskID: 0, ScheduleID: 0, ScheduleToStartTimeoutSeconds: 0,\n\t\t\t\t\tExpiry:      time.Time{},\n\t\t\t\t\tCreatedTime: time.Time{},\n\t\t\t\t\tPartitionConfig: map[string]string{\n\t\t\t\t\t\t\"key\":  \"value\",\n\t\t\t\t\t\t\"key2\": \"value2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tassert.Equal(t, uint64(193), response.ByteSize())\n\t})\n}\n\nfunc TestGetListDomainsResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &ListDomainsResponse{}\n\t\t\t\tfuzzer.Fuzz(&response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"domain info\", func(t *testing.T) {\n\t\tinfo := &DomainInfo{ID: \"ID\", Name: \"Name\", Status: 2, Description: \"Desc\", OwnerEmail: \"Email\", Data: nil}\n\t\tassert.Equal(t, uint64(95), info.ByteSize())\n\n\t\tinfo = &DomainInfo{\n\t\t\tID: \"ID\", Name: \"Name\", Status: 0, Description: \"Desc\", OwnerEmail: \"Email\",\n\t\t\tData: map[string]string{\"key\": \"value\"},\n\t\t}\n\t\tassert.Equal(t, uint64(103), info.ByteSize())\n\t})\n\n\tt.Run(\"full response\", func(t *testing.T) {\n\t\tresponse := &ListDomainsResponse{\n\t\t\tDomains: []*GetDomainResponse{\n\t\t\t\tnil,\n\t\t\t\t{\n\t\t\t\t\tInfo: nil, Config: nil, ReplicationConfig: nil, IsGlobalDomain: false,\n\t\t\t\t\tConfigVersion: 0, FailoverVersion: 0, FailoverNotificationVersion: 0, PreviousFailoverVersion: 0,\n\t\t\t\t\tFailoverEndTime: nil, LastUpdatedTime: 0, NotificationVersion: 0,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tInfo: &DomainInfo{\n\t\t\t\t\t\tID: \"ID\", Name: \"Name\", Status: 0, Description: \"Desc\", OwnerEmail: \"Email\",\n\t\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\t\"key\": \"value\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &DomainConfig{\n\t\t\t\t\t\tRetention: 0, EmitMetric: false, HistoryArchivalStatus: 0, HistoryArchivalURI: \"URI\",\n\t\t\t\t\t\tVisibilityArchivalStatus: 0, VisibilityArchivalURI: \"URI\", BadBinaries: types.BadBinaries{},\n\t\t\t\t\t\tIsolationGroups: map[string]types.IsolationGroupPartition{\n\t\t\t\t\t\t\t\"key\": {Name: \"abc\", State: 0},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:             false,\n\t\t\t\t\t\t\tPredefinedQueueName: \"name\",\n\t\t\t\t\t\t\tQueueType:           \"type\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: nil,\n\t\t\t\t\t\t\t\tData:         []byte{1, 2, 3},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"cluster\",\n\t\t\t\t\t\tClusters: []*ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: \"cluster\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tIsGlobalDomain: false, ConfigVersion: 0, FailoverVersion: 0, FailoverNotificationVersion: 0, PreviousFailoverVersion: 0,\n\t\t\t\t\tFailoverEndTime: nil, LastUpdatedTime: 0, NotificationVersion: 0,\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t}\n\n\t\tassert.Equal(t, uint64(535), response.ByteSize())\n\t})\n}\n\nfunc TestRawReadHistoryResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &ReadRawHistoryBranchResponse{}\n\t\t\t\tfuzzer.Fuzz(&response)\n\n\t\t\t\t_ = response.Size2()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := ReadRawHistoryBranchResponse{\n\t\t\tHistoryEventBlobs: []*DataBlob{\n\t\t\t\tnil,\n\t\t\t\t{\n\t\t\t\t\tEncoding: \"abc\",\n\t\t\t\t\tData:     []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10},\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\tSize:          123,\n\t\t}\n\n\t\tassert.Equal(t, uint64(109), response.Size2())\n\t})\n\n\tt.Run(\"a bigger response emits a bigger value\", func(t *testing.T) {\n\t\tresponse := ReadRawHistoryBranchResponse{\n\t\t\tHistoryEventBlobs: []*DataBlob{\n\t\t\t\t{\n\t\t\t\t\tEncoding: \"abc\",\n\t\t\t\t\tData:     []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14},\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\tSize:          123,\n\t\t}\n\n\t\tassert.Equal(t, uint64(113), response.Size2())\n\t})\n}\n\nfunc TestListCurrentExecutionsResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &ListCurrentExecutionsResponse{}\n\t\t\t\tfuzzer.Fuzz(&response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := &ListCurrentExecutionsResponse{\n\t\t\tExecutions: []*CurrentWorkflowExecution{\n\t\t\t\tnil,\n\t\t\t\t{DomainID: \"DomainID\", WorkflowID: \"WorkflowID\", RunID: \"ID\", State: 0, CurrentRunID: \"ID\"},\n\t\t\t},\n\t\t\tPageToken: []byte{1, 2, 3},\n\t\t}\n\n\t\tassert.Equal(t, uint64(145), response.ByteSize())\n\t})\n\n\tt.Run(\"a bigger response emits a bigger value\", func(t *testing.T) {\n\t\tresponse := &ListCurrentExecutionsResponse{\n\t\t\tExecutions: []*CurrentWorkflowExecution{\n\t\t\t\t{DomainID: \"LongDomainID\", WorkflowID: \"LongWorkflowID\", RunID: \"ID\", State: 0, CurrentRunID: \"ID\"},\n\t\t\t},\n\t\t\tPageToken: []byte{1, 2, 3},\n\t\t}\n\n\t\tassert.Equal(t, uint64(153), response.ByteSize())\n\t})\n}\n\nfunc TestGetHistoryTasksResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25).Funcs(\n\t\t\tfunc(t *Task, c fuzz.Continue) {\n\t\t\t\tswitch c.Intn(21) {\n\t\t\t\tcase 0:\n\t\t\t\t\tvar m DecisionTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 1:\n\t\t\t\t\tvar m ActivityTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 2:\n\t\t\t\t\tvar m RecordWorkflowStartedTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 3:\n\t\t\t\t\tvar m ResetWorkflowTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 4:\n\t\t\t\t\tvar m CloseExecutionTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 5:\n\t\t\t\t\tvar m DeleteHistoryEventTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 6:\n\t\t\t\t\tvar m DecisionTimeoutTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 7:\n\t\t\t\t\tvar m WorkflowTimeoutTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 8:\n\t\t\t\t\tvar m CancelExecutionTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 9:\n\t\t\t\t\tvar m SignalExecutionTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 10:\n\t\t\t\t\tvar m RecordChildExecutionCompletedTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 11:\n\t\t\t\t\tvar m UpsertWorkflowSearchAttributesTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 12:\n\t\t\t\t\tvar m UserTimerTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 13:\n\t\t\t\t\tvar m ActivityTimeoutTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 14:\n\t\t\t\t\tvar m ActivityRetryTimerTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 15:\n\t\t\t\t\tvar m WorkflowBackoffTimerTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 16:\n\t\t\t\t\tvar m HistoryReplicationTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 17:\n\t\t\t\t\tvar m StartChildExecutionTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 18:\n\t\t\t\t\tvar m RecordWorkflowClosedTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 19:\n\t\t\t\t\tvar m SyncActivityTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\tcase 20:\n\t\t\t\t\tvar m FailoverMarkerTask\n\t\t\t\t\tc.Fuzz(&m)\n\t\t\t\t\t*t = &m\n\t\t\t\t}\n\t\t\t},\n\t\t)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &GetHistoryTasksResponse{}\n\t\t\t\tfuzzer.Fuzz(&response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := &GetHistoryTasksResponse{\n\t\t\tTasks: []Task{\n\t\t\t\tnil,\n\t\t\t\t&DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"DomainID\", WorkflowID: \"WorkflowID\", RunID: \"ID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: TaskData{\n\t\t\t\t\t\tTaskID:              0,\n\t\t\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t\t\t\tVersion:             0,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"DomainID\",\n\t\t\t\t\tTaskList:       \"\", ScheduleID: 0,\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t}\n\n\t\tassert.Equal(t, uint64(135), response.ByteSize())\n\t})\n\n\tt.Run(\"a bigger response emits a bigger value\", func(t *testing.T) {\n\t\tresponse := &GetHistoryTasksResponse{\n\t\t\tTasks: []Task{\n\t\t\t\t&SignalExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"DomainID\", WorkflowID: \"WorkflowID\", RunID: \"ID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: TaskData{\n\t\t\t\t\t\tTaskID:              0,\n\t\t\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t\t\t\tVersion:             0,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:          \"DomainID\",\n\t\t\t\t\tTargetWorkflowID:        \"WorkflowID\",\n\t\t\t\t\tTargetRunID:             \"ID\",\n\t\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\t},\n\t\t\t},\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t}\n\n\t\tassert.Equal(t, uint64(140), response.ByteSize())\n\t})\n}\n\nfunc TestQueueMessageListEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &QueueMessageList{}\n\t\t\t\tfuzzer.Fuzz(response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := &QueueMessageList{\n\t\t\t{ID: 0, QueueType: 0, Payload: nil},\n\t\t}\n\n\t\tassert.Equal(t, uint64(40), response.ByteSize())\n\t})\n\n\tt.Run(\"a bigger response emits a bigger value\", func(t *testing.T) {\n\t\tresponse := &QueueMessageList{\n\t\t\t{ID: 0, QueueType: 0, Payload: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}},\n\t\t}\n\n\t\tassert.Equal(t, uint64(50), response.ByteSize())\n\t})\n}\n\nfunc TestGetAllHistoryTreeBranchesResponseEstimatePayloadSize(t *testing.T) {\n\tt.Run(\"does not panic\", func(t *testing.T) {\n\t\tfuzzer := testdatagen.NewWithNilChance(t, int64(123), 0.25)\n\t\tassert.NotPanics(t, func() {\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tresponse := &GetAllHistoryTreeBranchesResponse{}\n\t\t\t\tfuzzer.Fuzz(response)\n\n\t\t\t\t_ = response.ByteSize()\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"response is not nil\", func(t *testing.T) {\n\t\tresponse := &GetAllHistoryTreeBranchesResponse{\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\tBranches: []HistoryBranchDetail{\n\t\t\t\t{TreeID: \"\", BranchID: \"\", ForkTime: time.Time{}, Info: \"\"},\n\t\t\t},\n\t\t}\n\n\t\tassert.Equal(t, uint64(123), response.ByteSize())\n\t})\n\n\tt.Run(\"a bigger response emits a bigger value\", func(t *testing.T) {\n\t\tresponse := &GetAllHistoryTreeBranchesResponse{\n\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\tBranches: []HistoryBranchDetail{\n\t\t\t\t{TreeID: \"TreeID\", BranchID: \"BID\", ForkTime: time.Time{}, Info: \"Info\"},\n\t\t\t},\n\t\t}\n\n\t\tassert.Equal(t, uint64(136), response.ByteSize())\n\t})\n}\n"
  },
  {
    "path": "common/persistence/nosql/constants.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\n// Guidelines for creating new special UUID constants for all NoSQL\n// For DB specific constants(e.g. Cassandra), create them in the db specific package.\n// Each UUID should be of the form: E0000000-R000-f000-f000-00000000000x\n// Where x is any hexadecimal value, E represents the entity type valid values are:\n// E = {DomainID = 1, WorkflowID = 2, RunID = 3}\n// R represents row type in executions table, valid values are:\n// R = {Shard = 1, Execution = 2, Transfer = 3, Timer = 4, Replication = 5, Replication_DLQ = 6, CrossCluster = 7}\nconst (\n\t// Special Domains related constants\n\temptyDomainID = \"10000000-0000-f000-f000-000000000000\"\n\t// Special Run IDs\n\temptyRunID                   = \"30000000-0000-f000-f000-000000000000\"\n\trowTypeReplicationWorkflowID = \"20000000-5000-f000-f000-000000000000\"\n\trowTypeReplicationRunID      = \"30000000-5000-f000-f000-000000000000\"\n\temptyInitiatedID             = int64(-7)\n)\n"
  },
  {
    "path": "common/persistence/nosql/factory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n)\n\ntype (\n\t// Factory vends datastore implementations backed by cassandra\n\tFactory struct {\n\t\tsync.RWMutex\n\t\tcfg              config.ShardedNoSQL\n\t\tclusterName      string\n\t\tlogger           log.Logger\n\t\tmetricsClient    metrics.Client\n\t\texecStoreFactory *executionStoreFactory\n\t\tdc               *persistence.DynamicConfiguration\n\t\tparser           serialization.Parser\n\t\ttaskSerializer   serialization.TaskSerializer\n\t}\n\n\texecutionStoreFactory struct {\n\t\tlogger            log.Logger\n\t\tshardedNosqlStore shardedNosqlStore\n\t\ttaskSerializer    serialization.TaskSerializer\n\t}\n)\n\n// NewFactory returns an instance of a factory object which can be used to create\n// datastores that are backed by cassandra\nfunc NewFactory(cfg config.ShardedNoSQL, clusterName string, logger log.Logger, metricsClient metrics.Client, taskSerializer serialization.TaskSerializer, parser serialization.Parser, dc *persistence.DynamicConfiguration) *Factory {\n\treturn &Factory{\n\t\tcfg:            cfg,\n\t\tclusterName:    clusterName,\n\t\tlogger:         logger,\n\t\tmetricsClient:  metricsClient,\n\t\ttaskSerializer: taskSerializer,\n\t\tdc:             dc,\n\t\tparser:         parser,\n\t}\n}\n\n// NewTaskStore returns a new task store\nfunc (f *Factory) NewTaskStore() (persistence.TaskStore, error) {\n\treturn newNoSQLTaskStore(f.cfg, f.logger, f.metricsClient, f.dc)\n}\n\n// NewShardStore returns a new shard store\nfunc (f *Factory) NewShardStore() (persistence.ShardStore, error) {\n\treturn newNoSQLShardStore(f.cfg, f.clusterName, f.logger, f.metricsClient, f.dc, f.parser)\n}\n\n// NewHistoryStore returns a new history store\nfunc (f *Factory) NewHistoryStore() (persistence.HistoryStore, error) {\n\treturn newNoSQLHistoryStore(f.cfg, f.logger, f.metricsClient, f.dc)\n}\n\n// NewDomainStore returns a metadata store that understands only v2\nfunc (f *Factory) NewDomainStore() (persistence.DomainStore, error) {\n\treturn newNoSQLDomainStore(f.cfg, f.clusterName, f.logger, f.metricsClient, f.dc)\n}\n\n// NewDomainAuditStore returns a domain audit store\nfunc (f *Factory) NewDomainAuditStore() (persistence.DomainAuditStore, error) {\n\treturn newNoSQLDomainAuditStore(f.cfg, f.logger, f.metricsClient, f.dc)\n}\n\n// NewExecutionStore returns an ExecutionStore for a given shardID\nfunc (f *Factory) NewExecutionStore(shardID int) (persistence.ExecutionStore, error) {\n\tfactory, err := f.executionStoreFactory()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn factory.new(shardID)\n}\n\n// NewVisibilityStore returns a visibility store\nfunc (f *Factory) NewVisibilityStore(sortByCloseTime bool) (persistence.VisibilityStore, error) {\n\treturn newNoSQLVisibilityStore(sortByCloseTime, f.cfg, f.logger, f.metricsClient, f.dc)\n}\n\n// NewQueue returns a new queue backed by cassandra\nfunc (f *Factory) NewQueue(queueType persistence.QueueType) (persistence.QueueStore, error) {\n\treturn newNoSQLQueueStore(f.cfg, f.logger, f.metricsClient, queueType, f.dc)\n}\n\n// NewConfigStore returns a new config store\nfunc (f *Factory) NewConfigStore() (persistence.ConfigStore, error) {\n\treturn NewNoSQLConfigStore(f.cfg, f.logger, f.metricsClient, f.dc)\n}\n\n// Close closes the factory\nfunc (f *Factory) Close() {\n\tf.Lock()\n\tdefer f.Unlock()\n\tif f.execStoreFactory != nil {\n\t\tf.execStoreFactory.close()\n\t}\n}\n\nfunc (f *Factory) executionStoreFactory() (*executionStoreFactory, error) {\n\tf.RLock()\n\tif f.execStoreFactory != nil {\n\t\tf.RUnlock()\n\t\treturn f.execStoreFactory, nil\n\t}\n\tf.RUnlock()\n\tf.Lock()\n\tdefer f.Unlock()\n\tif f.execStoreFactory != nil {\n\t\treturn f.execStoreFactory, nil\n\t}\n\n\tfactory, err := newExecutionStoreFactory(f.cfg, f.logger, f.metricsClient, f.taskSerializer, f.dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tf.execStoreFactory = factory\n\treturn f.execStoreFactory, nil\n}\n\n// newExecutionStoreFactory is used to create an instance of ExecutionStoreFactory implementation\nfunc newExecutionStoreFactory(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttaskSerializer serialization.TaskSerializer,\n\tdc *persistence.DynamicConfiguration,\n) (*executionStoreFactory, error) {\n\ts, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &executionStoreFactory{\n\t\ttaskSerializer:    taskSerializer,\n\t\tlogger:            logger,\n\t\tshardedNosqlStore: s,\n\t}, nil\n}\n\nfunc (f *executionStoreFactory) close() {\n\tf.shardedNosqlStore.Close()\n}\n\n// new implements ExecutionStoreFactory interface\nfunc (f *executionStoreFactory) new(shardID int) (persistence.ExecutionStore, error) {\n\tstoreShard, err := f.shardedNosqlStore.GetStoreShardByHistoryShard(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpmgr, err := NewExecutionStore(shardID, storeShard.db, f.logger, f.taskSerializer, storeShard.dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn pmgr, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_config_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\ntype nosqlConfigStore struct {\n\tnosqlStore\n}\n\nfunc NewNoSQLConfigStore(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.ConfigStore, error) {\n\tshardedStore, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlConfigStore{\n\t\tnosqlStore: shardedStore.GetDefaultShard(),\n\t}, nil\n}\n\nfunc (m *nosqlConfigStore) FetchConfig(ctx context.Context, configType persistence.ConfigType) (*persistence.InternalConfigStoreEntry, error) {\n\tentry, err := m.db.SelectLatestConfig(ctx, int(configType))\n\tif err != nil {\n\t\tif m.db.IsNotFoundError(err) {\n\t\t\treturn nil, nil\n\t\t}\n\t\treturn nil, convertCommonErrors(m.db, \"FetchConfig\", err)\n\t}\n\treturn entry, nil\n}\n\nfunc (m *nosqlConfigStore) UpdateConfig(ctx context.Context, value *persistence.InternalConfigStoreEntry) error {\n\terr := m.db.InsertConfig(ctx, value)\n\tif err != nil {\n\t\tif _, ok := err.(*nosqlplugin.ConditionFailure); ok {\n\t\t\treturn &persistence.ConditionFailedError{Msg: fmt.Sprintf(\"Version %v already exists. Condition Failed\", value.Version)}\n\t\t}\n\t\treturn convertCommonErrors(m.db, \"UpdateConfig\", err)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_config_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc setUpMocksForNoSQLConfigStore(t *testing.T) (*nosqlConfigStore, *nosqlplugin.MockDB) {\n\tctrl := gomock.NewController(t)\n\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\tlogger := log.NewNoop()\n\n\t// Creating the nosqlConfigStore and injecting the mocked DB\n\tconfigStore := &nosqlConfigStore{\n\t\tnosqlStore: nosqlStore{\n\t\t\tlogger: logger,\n\t\t\tdb:     mockDB,\n\t\t},\n\t}\n\n\treturn configStore, mockDB\n}\n\nfunc TestFetchConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tsetupMock      func(mockDB *nosqlplugin.MockDB)\n\t\tconfigType     persistence.ConfigType\n\t\texpectError    bool\n\t\texpectedError  string\n\t\texpectedResult *persistence.InternalConfigStoreEntry\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectLatestConfig(gomock.Any(), int(persistence.DynamicConfig)).\n\t\t\t\t\tReturn(&persistence.InternalConfigStoreEntry{\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\tValues:  &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tconfigType:     persistence.DynamicConfig,\n\t\t\texpectError:    false,\n\t\t\texpectedResult: &persistence.InternalConfigStoreEntry{Version: 1, Values: &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")}},\n\t\t},\n\t\t{\n\t\t\tname: \"config not found\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectLatestConfig(gomock.Any(), int(persistence.DynamicConfig)).\n\t\t\t\t\tReturn(nil, errors.New(\"not found\")).\n\t\t\t\t\tTimes(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(errors.New(\"not found\")).Return(true).Times(1)\n\t\t\t},\n\t\t\tconfigType:     persistence.DynamicConfig,\n\t\t\texpectError:    false,\n\t\t\texpectedResult: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"fetch error\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectLatestConfig(gomock.Any(), int(persistence.DynamicConfig)).\n\t\t\t\t\tReturn(nil, errors.New(\"fetch error\")).\n\t\t\t\t\tTimes(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(errors.New(\"fetch error\")).Return(false).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(errors.New(\"fetch error\")).Return(true).Times(1)\n\t\t\t},\n\t\t\tconfigType:    persistence.DynamicConfig,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"fetch error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tconfigStore, mockDB := setUpMocksForNoSQLConfigStore(t)\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\tresult, err := configStore.FetchConfig(context.Background(), tc.configType)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedResult, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(mockDB *nosqlplugin.MockDB)\n\t\tvalue         *persistence.InternalConfigStoreEntry\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertConfig(gomock.Any(), &persistence.InternalConfigStoreEntry{\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\tValues:  &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\tvalue: &persistence.InternalConfigStoreEntry{\n\t\t\t\tVersion: 1,\n\t\t\t\tValues:  &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&nosqlplugin.ConditionFailure{}).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\tvalue: &persistence.InternalConfigStoreEntry{\n\t\t\t\tVersion: 1,\n\t\t\t\tValues:  &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Version 1 already exists. Condition Failed\",\n\t\t},\n\t\t{\n\t\t\tname: \"insert error\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"insert error\")).\n\t\t\t\t\tTimes(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(errors.New(\"insert error\")).Return(true).Times(1)\n\t\t\t},\n\t\t\tvalue: &persistence.InternalConfigStoreEntry{\n\t\t\t\tVersion: 1,\n\t\t\t\tValues:  &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"config-values\")},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"insert error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tconfigStore, mockDB := setUpMocksForNoSQLConfigStore(t)\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\terr := configStore.UpdateConfig(context.Background(), tc.value)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_domain_audit_store.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\ntype nosqlDomainAuditStore struct {\n\tnosqlStore\n}\n\n// newNoSQLDomainAuditStore is used to create an instance of DomainAuditStore implementation\nfunc newNoSQLDomainAuditStore(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.DomainAuditStore, error) {\n\tshardedStore, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlDomainAuditStore{\n\t\tnosqlStore: shardedStore.GetDefaultShard(),\n\t}, nil\n}\n\n// CreateDomainAuditLog creates a new domain audit log entry\nfunc (m *nosqlDomainAuditStore) CreateDomainAuditLog(\n\tctx context.Context,\n\trequest *persistence.InternalCreateDomainAuditLogRequest,\n) (*persistence.CreateDomainAuditLogResponse, error) {\n\trow := &nosqlplugin.DomainAuditLogRow{\n\t\tDomainID:            request.DomainID,\n\t\tEventID:             request.EventID,\n\t\tStateBefore:         getDataBlobBytes(request.StateBefore),\n\t\tStateBeforeEncoding: getDataBlobEncoding(request.StateBefore),\n\t\tStateAfter:          getDataBlobBytes(request.StateAfter),\n\t\tStateAfterEncoding:  getDataBlobEncoding(request.StateAfter),\n\t\tOperationType:       request.OperationType,\n\t\tCreatedTime:         request.CreatedTime,\n\t\tLastUpdatedTime:     request.LastUpdatedTime,\n\t\tIdentity:            request.Identity,\n\t\tIdentityType:        request.IdentityType,\n\t\tComment:             request.Comment,\n\t\tTTLSeconds:          request.TTLSeconds,\n\t}\n\n\terr := m.db.InsertDomainAuditLog(ctx, row)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"CreateDomainAuditLog\", err)\n\t}\n\n\treturn &persistence.CreateDomainAuditLogResponse{\n\t\tEventID: request.EventID,\n\t}, nil\n}\n\n// GetDomainAuditLogs retrieves domain audit logs\nfunc (m *nosqlDomainAuditStore) GetDomainAuditLogs(\n\tctx context.Context,\n\trequest *persistence.GetDomainAuditLogsRequest,\n) (*persistence.InternalGetDomainAuditLogsResponse, error) {\n\tfilter := &nosqlplugin.DomainAuditLogFilter{\n\t\tDomainID:       request.DomainID,\n\t\tOperationType:  request.OperationType,\n\t\tMinCreatedTime: request.MinCreatedTime,\n\t\tMaxCreatedTime: request.MaxCreatedTime,\n\t\tPageSize:       request.PageSize,\n\t\tNextPageToken:  request.NextPageToken,\n\t}\n\n\trows, nextPageToken, err := m.db.SelectDomainAuditLogs(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetDomainAuditLogs\", err)\n\t}\n\n\tvar auditLogs []*persistence.InternalDomainAuditLog\n\tfor _, row := range rows {\n\t\tauditLog := &persistence.InternalDomainAuditLog{\n\t\t\tEventID:         row.EventID,\n\t\t\tDomainID:        row.DomainID,\n\t\t\tOperationType:   row.OperationType,\n\t\t\tCreatedTime:     row.CreatedTime,\n\t\t\tLastUpdatedTime: row.LastUpdatedTime,\n\t\t\tIdentity:        row.Identity,\n\t\t\tIdentityType:    row.IdentityType,\n\t\t\tComment:         row.Comment,\n\t\t}\n\n\t\tif len(row.StateBefore) > 0 {\n\t\t\tauditLog.StateBefore = &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingType(row.StateBeforeEncoding),\n\t\t\t\tData:     row.StateBefore,\n\t\t\t}\n\t\t}\n\n\t\tif len(row.StateAfter) > 0 {\n\t\t\tauditLog.StateAfter = &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingType(row.StateAfterEncoding),\n\t\t\t\tData:     row.StateAfter,\n\t\t\t}\n\t\t}\n\n\t\tauditLogs = append(auditLogs, auditLog)\n\t}\n\n\treturn &persistence.InternalGetDomainAuditLogsResponse{\n\t\tAuditLogs:     auditLogs,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc getDataBlobBytes(blob *persistence.DataBlob) []byte {\n\tif blob == nil {\n\t\treturn nil\n\t}\n\treturn blob.Data\n}\n\nfunc getDataBlobEncoding(blob *persistence.DataBlob) string {\n\tif blob == nil {\n\t\treturn \"\"\n\t}\n\treturn string(blob.Encoding)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_domain_audit_store_test.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc setUpMocksForDomainAuditStore(t *testing.T) (*nosqlDomainAuditStore, *nosqlplugin.MockDB) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\n\tshardedStore := nosqlStore{\n\t\tdb: dbMock,\n\t}\n\n\tdomainAuditStore := &nosqlDomainAuditStore{\n\t\tnosqlStore: shardedStore,\n\t}\n\n\treturn domainAuditStore, dbMock\n}\n\nfunc TestCreateDomainAuditLog(t *testing.T) {\n\tctx := context.Background()\n\tnow := time.Unix(1234567890, 0)\n\n\tstateBeforeBlob := &persistence.DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\tData:     []byte(\"state-before-data\"),\n\t}\n\n\tstateAfterBlob := &persistence.DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\tData:     []byte(\"state-after-data\"),\n\t}\n\n\ttests := map[string]struct {\n\t\tsetupMock   func(*nosqlplugin.MockDB)\n\t\trequest     *persistence.InternalCreateDomainAuditLogRequest\n\t\texpectError bool\n\t\texpectedID  string\n\t}{\n\t\t\"success with full data\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\texpectedRow := &nosqlplugin.DomainAuditLogRow{\n\t\t\t\t\tDomainID:            \"domain-123\",\n\t\t\t\t\tEventID:             \"event-456\",\n\t\t\t\t\tStateBefore:         stateBeforeBlob.Data,\n\t\t\t\t\tStateBeforeEncoding: string(stateBeforeBlob.Encoding),\n\t\t\t\t\tStateAfter:          stateAfterBlob.Data,\n\t\t\t\t\tStateAfterEncoding:  string(stateAfterBlob.Encoding),\n\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\tIdentity:            \"test-user\",\n\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\tComment:             \"test comment\",\n\t\t\t\t}\n\t\t\t\tdbMock.EXPECT().InsertDomainAuditLog(ctx, expectedRow).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"domain-123\",\n\t\t\t\tEventID:         \"event-456\",\n\t\t\t\tStateBefore:     stateBeforeBlob,\n\t\t\t\tStateAfter:      stateAfterBlob,\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"test-user\",\n\t\t\t\tIdentityType:    \"user\",\n\t\t\t\tComment:         \"test comment\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedID:  \"event-456\",\n\t\t},\n\t\t\"success with nil state blobs\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\texpectedRow := &nosqlplugin.DomainAuditLogRow{\n\t\t\t\t\tDomainID:            \"domain-123\",\n\t\t\t\t\tEventID:             \"event-789\",\n\t\t\t\t\tStateBefore:         nil,\n\t\t\t\t\tStateBeforeEncoding: \"\",\n\t\t\t\t\tStateAfter:          nil,\n\t\t\t\t\tStateAfterEncoding:  \"\",\n\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeCreate,\n\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\tIdentity:            \"system\",\n\t\t\t\t\tIdentityType:        \"system\",\n\t\t\t\t\tComment:             \"\",\n\t\t\t\t}\n\t\t\t\tdbMock.EXPECT().InsertDomainAuditLog(ctx, expectedRow).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"domain-123\",\n\t\t\t\tEventID:         \"event-789\",\n\t\t\t\tStateBefore:     nil,\n\t\t\t\tStateAfter:      nil,\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeCreate,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"system\",\n\t\t\t\tIdentityType:    \"system\",\n\t\t\t\tComment:         \"\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedID:  \"event-789\",\n\t\t},\n\t\t\"database error\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertDomainAuditLog(ctx, gomock.Any()).Return(errors.New(\"database error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"domain-123\",\n\t\t\t\tEventID:         \"event-999\",\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"test-user\",\n\t\t\t\tIdentityType:    \"user\",\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore, dbMock := setUpMocksForDomainAuditStore(t)\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\tresp, err := store.CreateDomainAuditLog(ctx, tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\trequire.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, tc.expectedID, resp.EventID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomainAuditLogs(t *testing.T) {\n\tctx := context.Background()\n\tnow := time.Unix(1234567890, 0)\n\tminTime := now.Add(-24 * time.Hour)\n\tmaxTime := now\n\n\ttests := map[string]struct {\n\t\tsetupMock      func(*nosqlplugin.MockDB)\n\t\trequest        *persistence.GetDomainAuditLogsRequest\n\t\texpectError    bool\n\t\texpectedCount  int\n\t\tvalidateResult func(*testing.T, *persistence.InternalGetDomainAuditLogsResponse)\n\t}{\n\t\t\"success with results\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\trows := []*nosqlplugin.DomainAuditLogRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"event-1\",\n\t\t\t\t\t\tDomainID:            \"domain-123\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\t\tIdentity:            \"user-1\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 1\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-1\"),\n\t\t\t\t\t\tStateBeforeEncoding: string(constants.EncodingTypeThriftRWSnappy),\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-1\"),\n\t\t\t\t\t\tStateAfterEncoding:  string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"event-2\",\n\t\t\t\t\t\tDomainID:            \"domain-123\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeFailover,\n\t\t\t\t\t\tCreatedTime:         now.Add(-1 * time.Hour),\n\t\t\t\t\t\tLastUpdatedTime:     now.Add(-1 * time.Hour),\n\t\t\t\t\t\tIdentity:            \"user-2\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 2\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-2\"),\n\t\t\t\t\t\tStateBeforeEncoding: string(constants.EncodingTypeThriftRWSnappy),\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-2\"),\n\t\t\t\t\t\tStateAfterEncoding:  string(constants.EncodingTypeThriftRWSnappy),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tnextPageToken := []byte(\"next-token\")\n\n\t\t\t\texpectedFilter := &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\t\tDomainID:       \"domain-123\",\n\t\t\t\t\tOperationType:  persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\t\tPageSize:       10,\n\t\t\t\t\tNextPageToken:  []byte(\"current-token\"),\n\t\t\t\t}\n\n\t\t\t\tdbMock.EXPECT().SelectDomainAuditLogs(ctx, expectedFilter).Return(rows, nextPageToken, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"domain-123\",\n\t\t\t\tOperationType:  persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\tPageSize:       10,\n\t\t\t\tNextPageToken:  []byte(\"current-token\"),\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 2,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\trequire.Len(t, resp.AuditLogs, 2)\n\t\t\t\tassert.Equal(t, \"event-1\", resp.AuditLogs[0].EventID)\n\t\t\t\tassert.Equal(t, \"domain-123\", resp.AuditLogs[0].DomainID)\n\t\t\t\tassert.Equal(t, persistence.DomainAuditOperationTypeUpdate, resp.AuditLogs[0].OperationType)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateBefore)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRWSnappy, resp.AuditLogs[0].StateBefore.Encoding)\n\t\t\t\tassert.Equal(t, []byte(\"state-before-1\"), resp.AuditLogs[0].StateBefore.Data)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateAfter)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, resp.AuditLogs[0].StateAfter.Encoding)\n\n\t\t\t\tassert.Equal(t, \"event-2\", resp.AuditLogs[1].EventID)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[1].StateBefore)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[1].StateAfter)\n\n\t\t\t\tassert.Equal(t, []byte(\"next-token\"), resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t\"success with empty state blobs\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\trows := []*nosqlplugin.DomainAuditLogRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"event-3\",\n\t\t\t\t\t\tDomainID:            \"domain-456\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeCreate,\n\t\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\t\tIdentity:            \"system\",\n\t\t\t\t\t\tIdentityType:        \"system\",\n\t\t\t\t\t\tComment:             \"created\",\n\t\t\t\t\t\tStateBefore:         []byte{}, // Empty slice\n\t\t\t\t\t\tStateBeforeEncoding: \"\",\n\t\t\t\t\t\tStateAfter:          nil, // Nil\n\t\t\t\t\t\tStateAfterEncoding:  \"\",\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tdbMock.EXPECT().SelectDomainAuditLogs(ctx, gomock.Any()).Return(rows, nil, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"domain-456\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 1,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\trequire.Len(t, resp.AuditLogs, 1)\n\t\t\t\tassert.Equal(t, \"event-3\", resp.AuditLogs[0].EventID)\n\t\t\t\t// Empty/nil state blobs should not be converted\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateBefore)\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateAfter)\n\t\t\t},\n\t\t},\n\t\t\"success with no results\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomainAuditLogs(ctx, gomock.Any()).Return([]*nosqlplugin.DomainAuditLogRow{}, nil, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"domain-789\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 0,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\tassert.Empty(t, resp.AuditLogs)\n\t\t\t\tassert.Nil(t, resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t\"database error\": {\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomainAuditLogs(ctx, gomock.Any()).Return(nil, nil, errors.New(\"database error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"domain-999\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore, dbMock := setUpMocksForDomainAuditStore(t)\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\tresp, err := store.GetDomainAuditLogs(ctx, tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\trequire.NotNil(t, resp)\n\t\t\t\tassert.Len(t, resp.AuditLogs, tc.expectedCount)\n\t\t\t\tif tc.validateResult != nil {\n\t\t\t\t\ttc.validateResult(t, resp)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDataBlobBytes(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *persistence.DataBlob\n\t\texpected []byte\n\t}{\n\t\t\"nil blob\": {\n\t\t\tinput:    nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"blob with data\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: []byte(\"test-data\"),\n\t\t},\n\t\t\"blob with empty data\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte{},\n\t\t\t},\n\t\t\texpected: []byte{},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := getDataBlobBytes(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestGetDataBlobEncoding(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *persistence.DataBlob\n\t\texpected string\n\t}{\n\t\t\"nil blob\": {\n\t\t\tinput:    nil,\n\t\t\texpected: \"\",\n\t\t},\n\t\t\"blob with ThriftRWSnappy encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: string(constants.EncodingTypeThriftRWSnappy),\n\t\t},\n\t\t\"blob with ThriftRW encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: string(constants.EncodingTypeThriftRW),\n\t\t},\n\t\t\"blob with empty encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: \"\",\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: \"\",\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := getDataBlobEncoding(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_domain_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype nosqlDomainStore struct {\n\tnosqlStore\n\tcurrentClusterName string\n}\n\n// newNoSQLDomainStore is used to create an instance of DomainStore implementation\nfunc newNoSQLDomainStore(\n\tcfg config.ShardedNoSQL,\n\tcurrentClusterName string,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.DomainStore, error) {\n\tshardedStore, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlDomainStore{\n\t\tnosqlStore:         shardedStore.GetDefaultShard(),\n\t\tcurrentClusterName: currentClusterName,\n\t}, nil\n}\n\n// CreateDomain create a domain\n// Cassandra does not support conditional updates across multiple tables.  For this reason we have to first insert into\n// 'Domains' table and then do a conditional insert into domains_by_name table.  If the conditional write fails we\n// delete the orphaned entry from domains table.  There is a chance delete entry could fail and we never delete the\n// orphaned entry from domains table.  We might need a background job to delete those orphaned record.\nfunc (m *nosqlDomainStore) CreateDomain(\n\tctx context.Context,\n\trequest *persistence.InternalCreateDomainRequest,\n) (*persistence.CreateDomainResponse, error) {\n\trow := &nosqlplugin.DomainRow{\n\t\tInfo:                        request.Info,\n\t\tConfig:                      request.Config,\n\t\tReplicationConfig:           request.ReplicationConfig,\n\t\tConfigVersion:               request.ConfigVersion,\n\t\tFailoverVersion:             request.FailoverVersion,\n\t\tFailoverNotificationVersion: persistence.InitialFailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     constants.InitialPreviousFailoverVersion,\n\t\tFailoverEndTime:             nil,\n\t\tIsGlobalDomain:              request.IsGlobalDomain,\n\t\tLastUpdatedTime:             request.LastUpdatedTime,\n\t\tCurrentTimeStamp:            request.CurrentTimeStamp,\n\t}\n\n\terr := m.db.InsertDomain(ctx, row)\n\n\tif err != nil {\n\t\tif _, ok := err.(*types.DomainAlreadyExistsError); ok {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, convertCommonErrors(m.db, \"CreateDomain\", err)\n\t}\n\n\treturn &persistence.CreateDomainResponse{ID: request.Info.ID}, nil\n}\n\nfunc (m *nosqlDomainStore) UpdateDomain(\n\tctx context.Context,\n\trequest *persistence.InternalUpdateDomainRequest,\n) error {\n\trow := &nosqlplugin.DomainRow{\n\t\tInfo:                        request.Info,\n\t\tConfig:                      request.Config,\n\t\tReplicationConfig:           request.ReplicationConfig,\n\t\tConfigVersion:               request.ConfigVersion,\n\t\tFailoverVersion:             request.FailoverVersion,\n\t\tFailoverNotificationVersion: request.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     request.PreviousFailoverVersion,\n\t\tFailoverEndTime:             request.FailoverEndTime,\n\t\tNotificationVersion:         request.NotificationVersion,\n\t\tLastUpdatedTime:             request.LastUpdatedTime,\n\t}\n\n\terr := m.db.UpdateDomain(ctx, row)\n\tif err != nil {\n\t\treturn convertCommonErrors(m.db, \"UpdateDomain\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (m *nosqlDomainStore) GetDomain(\n\tctx context.Context,\n\trequest *persistence.GetDomainRequest,\n) (*persistence.InternalGetDomainResponse, error) {\n\tif len(request.ID) > 0 && len(request.Name) > 0 {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"GetDomain operation failed.  Both ID and Name specified in request.\",\n\t\t}\n\t} else if len(request.ID) == 0 && len(request.Name) == 0 {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"GetDomain operation failed.  Both ID and Name are empty.\",\n\t\t}\n\t}\n\tvar domainName *string\n\tvar domainID *string\n\tif len(request.ID) > 0 {\n\t\tdomainID = common.StringPtr(request.ID)\n\t} else {\n\t\tdomainName = common.StringPtr(request.Name)\n\t}\n\n\thandleError := func(name, ID string, err error) error {\n\t\tidentity := name\n\t\tif len(ID) > 0 {\n\t\t\tidentity = ID\n\t\t}\n\t\tif m.db.IsNotFoundError(err) {\n\t\t\treturn &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Domain %s does not exist.\", identity),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(m.db, \"GetDomain\", err)\n\t}\n\n\trow, err := m.db.SelectDomain(ctx, domainID, domainName)\n\n\tif err != nil {\n\t\treturn nil, handleError(request.Name, request.ID, err)\n\t}\n\n\tif row.Info.Data == nil {\n\t\trow.Info.Data = map[string]string{}\n\t}\n\n\t// for active-active domains, do not populate active cluster name with current cluster name if it is not set\n\tif row.ReplicationConfig.ActiveClustersConfig == nil {\n\t\trow.ReplicationConfig.ActiveClusterName = cluster.GetOrUseDefaultActiveCluster(m.currentClusterName, row.ReplicationConfig.ActiveClusterName)\n\t}\n\n\trow.ReplicationConfig.Clusters = cluster.GetOrUseDefaultClusters(m.currentClusterName, row.ReplicationConfig.Clusters)\n\n\treturn &persistence.InternalGetDomainResponse{\n\t\tInfo:                        row.Info,\n\t\tConfig:                      row.Config,\n\t\tReplicationConfig:           row.ReplicationConfig,\n\t\tIsGlobalDomain:              row.IsGlobalDomain,\n\t\tConfigVersion:               row.ConfigVersion,\n\t\tFailoverVersion:             row.FailoverVersion,\n\t\tFailoverNotificationVersion: row.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:     row.PreviousFailoverVersion,\n\t\tFailoverEndTime:             row.FailoverEndTime,\n\t\tNotificationVersion:         row.NotificationVersion,\n\t\tLastUpdatedTime:             row.LastUpdatedTime,\n\t}, nil\n}\n\nfunc (m *nosqlDomainStore) ListDomains(\n\tctx context.Context,\n\trequest *persistence.ListDomainsRequest,\n) (*persistence.InternalListDomainsResponse, error) {\n\trows, nextPageToken, err := m.db.SelectAllDomains(ctx, request.PageSize, request.NextPageToken)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"ListDomains\", err)\n\t}\n\tvar domains []*persistence.InternalGetDomainResponse\n\tfor _, row := range rows {\n\t\tif row.Info.Data == nil {\n\t\t\trow.Info.Data = map[string]string{}\n\t\t}\n\n\t\t// for active-active domains, do not populate active cluster name with current cluster name if it is not set\n\t\tif row.ReplicationConfig.ActiveClustersConfig == nil {\n\t\t\trow.ReplicationConfig.ActiveClusterName = cluster.GetOrUseDefaultActiveCluster(m.currentClusterName, row.ReplicationConfig.ActiveClusterName)\n\t\t}\n\n\t\trow.ReplicationConfig.Clusters = cluster.GetOrUseDefaultClusters(m.currentClusterName, row.ReplicationConfig.Clusters)\n\n\t\tdomains = append(domains, &persistence.InternalGetDomainResponse{\n\t\t\tInfo:                        row.Info,\n\t\t\tConfig:                      row.Config,\n\t\t\tReplicationConfig:           row.ReplicationConfig,\n\t\t\tIsGlobalDomain:              row.IsGlobalDomain,\n\t\t\tConfigVersion:               row.ConfigVersion,\n\t\t\tFailoverVersion:             row.FailoverVersion,\n\t\t\tFailoverNotificationVersion: row.FailoverNotificationVersion,\n\t\t\tPreviousFailoverVersion:     row.PreviousFailoverVersion,\n\t\t\tFailoverEndTime:             row.FailoverEndTime,\n\t\t\tNotificationVersion:         row.NotificationVersion,\n\t\t\tLastUpdatedTime:             row.LastUpdatedTime,\n\t\t})\n\t}\n\n\treturn &persistence.InternalListDomainsResponse{\n\t\tDomains:       domains,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc (m *nosqlDomainStore) DeleteDomain(\n\tctx context.Context,\n\trequest *persistence.DeleteDomainRequest,\n) error {\n\tif err := m.db.DeleteDomain(ctx, &request.ID, nil); err != nil {\n\t\treturn convertCommonErrors(m.db, \"DeleteDomain\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (m *nosqlDomainStore) DeleteDomainByName(\n\tctx context.Context,\n\trequest *persistence.DeleteDomainByNameRequest,\n) error {\n\tif err := m.db.DeleteDomain(ctx, nil, &request.Name); err != nil {\n\t\treturn convertCommonErrors(m.db, \"DeleteDomainByName\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (m *nosqlDomainStore) GetMetadata(\n\tctx context.Context,\n) (*persistence.GetMetadataResponse, error) {\n\tnotificationVersion, err := m.db.SelectDomainMetadata(ctx)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetMetadata\", err)\n\t}\n\treturn &persistence.GetMetadataResponse{NotificationVersion: notificationVersion}, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_domain_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc testFixtureDomainInfo() *persistence.DomainInfo {\n\treturn &persistence.DomainInfo{\n\t\tID:          \"domain-id\",\n\t\tName:        \"domain-name\",\n\t\tStatus:      1,\n\t\tDescription: \"domain-desc\",\n\t\tOwnerEmail:  \"owner@test.com\",\n\t\tData:        map[string]string{\"test\": \"a\"},\n\t}\n}\n\nfunc testFixtureInternalDomainConfig() *persistence.InternalDomainConfig {\n\treturn &persistence.InternalDomainConfig{\n\t\tRetention:                time.Hour,\n\t\tHistoryArchivalStatus:    types.ArchivalStatus(1),\n\t\tHistoryArchivalURI:       \"s3://test-history-archival\",\n\t\tVisibilityArchivalStatus: types.ArchivalStatus(2),\n\t\tVisibilityArchivalURI:    \"s3://test-visibility-archival\",\n\t\tBadBinaries: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"bad-binaries\"),\n\t\t},\n\t\tIsolationGroups: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"isolation-groups\"),\n\t\t},\n\t\tAsyncWorkflowsConfig: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"async-workflows\"),\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfig() *persistence.InternalDomainReplicationConfig {\n\treturn &persistence.InternalDomainReplicationConfig{\n\t\tActiveClusterName: \"cluster-1\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testFixtureDomainReplicationConfigActiveActive() *persistence.InternalDomainReplicationConfig {\n\treturn &persistence.InternalDomainReplicationConfig{\n\t\tActiveClusterName: \"\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"cluster-2\",\n\t\t\t},\n\t\t},\n\t\tActiveClustersConfig: &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\tData:     []byte(\"active-clusters-config\"),\n\t\t},\n\t}\n}\n\nfunc setUpMocksForDomainStore(t *testing.T) (*nosqlDomainStore, *nosqlplugin.MockDB, *MockshardedNosqlStore) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\tshardedNoSQLStoreMock := NewMockshardedNosqlStore(ctrl)\n\n\tshardedStore := nosqlStore{\n\t\tdb: dbMock,\n\t}\n\n\tshardedNoSQLStoreMock.EXPECT().GetDefaultShard().Return(shardedStore).AnyTimes()\n\n\tdomainStore := &nosqlDomainStore{\n\t\tnosqlStore:         shardedNoSQLStoreMock.GetDefaultShard(),\n\t\tcurrentClusterName: \"test-cluster\",\n\t}\n\n\treturn domainStore, dbMock, shardedNoSQLStoreMock\n}\n\nfunc TestCreateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tsetupMock   func(*nosqlplugin.MockDB)\n\t\trequest     *persistence.InternalCreateDomainRequest\n\t\texpectError bool\n\t\texpectedID  string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertDomain(gomock.Any(), &nosqlplugin.DomainRow{\n\t\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tFailoverNotificationVersion: persistence.InitialFailoverNotificationVersion,\n\t\t\t\t\tPreviousFailoverVersion:     constants.InitialPreviousFailoverVersion,\n\t\t\t\t\tIsGlobalDomain:              true,\n\t\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo:              testFixtureDomainInfo(),\n\t\t\t\tConfig:            testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   2,\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tLastUpdatedTime:   time.Unix(1, 2),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedID:  \"domain-id\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain already exists error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertDomain(gomock.Any(), gomock.Any()).Return(&types.DomainAlreadyExistsError{}).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo:              testFixtureDomainInfo(),\n\t\t\t\tConfig:            testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tLastUpdatedTime:   time.Now(),\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"common error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo:              testFixtureDomainInfo(),\n\t\t\t\tConfig:            testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t\tLastUpdatedTime:   time.Now(),\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock, _ := setUpMocksForDomainStore(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := store.CreateDomain(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedID, resp.ID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tsetupMock   func(*nosqlplugin.MockDB)\n\t\trequest     *persistence.InternalUpdateDomainRequest\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().UpdateDomain(gomock.Any(), &nosqlplugin.DomainRow{\n\t\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\t\tNotificationVersion:         5,\n\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             2,\n\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\tNotificationVersion:         5,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"common error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             2,\n\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\tNotificationVersion:         5,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock, _ := setUpMocksForDomainStore(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\terr := store.UpdateDomain(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB)\n\t\trequest       *persistence.GetDomainRequest\n\t\texpectError   bool\n\t\texpected      *persistence.InternalGetDomainResponse\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success by ID\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), common.Ptr(\"test-domain-id\"), nil).Return(&nosqlplugin.DomainRow{\n\t\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\t\tNotificationVersion:         5,\n\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &persistence.GetDomainRequest{ID: \"test-domain-id\"},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             2,\n\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\tNotificationVersion:         5,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success by name\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), nil, common.Ptr(\"test-domain\")).Return(&nosqlplugin.DomainRow{\n\t\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\t\tNotificationVersion:         5,\n\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &persistence.GetDomainRequest{Name: \"test-domain\"},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfig(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             2,\n\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\tNotificationVersion:         5,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success by name - active-active domain\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), nil, common.Ptr(\"test-domain\")).Return(&nosqlplugin.DomainRow{\n\t\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfigActiveActive(),\n\t\t\t\t\tConfigVersion:               1,\n\t\t\t\t\tFailoverVersion:             2,\n\t\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\t\tNotificationVersion:         5,\n\t\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &persistence.GetDomainRequest{Name: \"test-domain\"},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo:                        testFixtureDomainInfo(),\n\t\t\t\tConfig:                      testFixtureInternalDomainConfig(),\n\t\t\t\tReplicationConfig:           testFixtureDomainReplicationConfigActiveActive(),\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             2,\n\t\t\t\tFailoverNotificationVersion: 3,\n\t\t\t\tPreviousFailoverVersion:     4,\n\t\t\t\tNotificationVersion:         5,\n\t\t\t\tFailoverEndTime:             common.Ptr(time.Unix(2, 3)),\n\t\t\t\tLastUpdatedTime:             time.Unix(1, 2),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"bad request error - both id and domain are set\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\t// No database call should be made\n\t\t\t},\n\t\t\trequest:       &persistence.GetDomainRequest{ID: \"test-id\", Name: \"test-name\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"GetDomain operation failed.  Both ID and Name specified in request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"bad request error - both id and domain are empty\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\t// No database call should be made\n\t\t\t},\n\t\t\trequest:       &persistence.GetDomainRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"GetDomain operation failed.  Both ID and Name are empty.\",\n\t\t},\n\t\t{\n\t\t\tname: \"not found error - by name\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.GetDomainRequest{Name: \"test-domain\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain test-domain does not exist.\",\n\t\t},\n\t\t{\n\t\t\tname: \"not found error - by id\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.GetDomainRequest{ID: \"test-domain-id\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain test-domain-id does not exist.\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomain(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.GetDomainRequest{ID: \"test-domain-id\"},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"GetDomain failed. Error: test error \",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock, _ := setUpMocksForDomainStore(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := store.GetDomain(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListDomains(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tsetupMock   func(*nosqlplugin.MockDB)\n\t\trequest     *persistence.ListDomainsRequest\n\t\texpectError bool\n\t\texpected    *persistence.InternalListDomainsResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectAllDomains(gomock.Any(), 2, []byte(\"token\")).Return(\n\t\t\t\t\t[]*nosqlplugin.DomainRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"domain-id-1\"},\n\t\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"active-active-domain-id\"},\n\t\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigActiveActive(),\n\t\t\t\t\t\t},\n\t\t\t\t\t}, []byte(\"next-token\"), nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.ListDomainsRequest{\n\t\t\t\tPageSize:      2,\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalListDomainsResponse{\n\t\t\t\tDomains: []*persistence.InternalGetDomainResponse{\n\t\t\t\t\t{\n\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"domain-id-1\", Data: map[string]string{}},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfig(),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tInfo:              &persistence.DomainInfo{ID: \"active-active-domain-id\", Data: map[string]string{}},\n\t\t\t\t\t\tReplicationConfig: testFixtureDomainReplicationConfigActiveActive(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: []byte(\"next-token\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"common error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectAllDomains(gomock.Any(), 10, []byte(\"token\")).Return(nil, nil, errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.ListDomainsRequest{\n\t\t\t\tPageSize:      10,\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock, _ := setUpMocksForDomainStore(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := store.ListDomains(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMetadata(t *testing.T) {\n\ttestCases := []struct {\n\t\tname            string\n\t\tsetupMock       func(*nosqlplugin.MockDB)\n\t\texpectError     bool\n\t\texpectedVersion int64\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomainMetadata(gomock.Any()).Return(int64(10), nil).Times(1)\n\t\t\t},\n\t\t\texpectError:     false,\n\t\t\texpectedVersion: 10,\n\t\t},\n\t\t{\n\t\t\tname: \"common error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectDomainMetadata(gomock.Any()).Return(int64(0), errors.New(\"test error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock, _ := setUpMocksForDomainStore(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := store.GetMetadata(context.Background())\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedVersion, resp.NotificationVersion)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_execution_store.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Implements ExecutionStore\ntype nosqlExecutionStore struct {\n\tshardID int\n\tnosqlStore\n\ttaskSerializer serialization.TaskSerializer\n}\n\n// NewExecutionStore is used to create an instance of ExecutionStore implementation\nfunc NewExecutionStore(\n\tshardID int,\n\tdb nosqlplugin.DB,\n\tlogger log.Logger,\n\ttaskSerializer serialization.TaskSerializer,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.ExecutionStore, error) {\n\treturn &nosqlExecutionStore{\n\t\tnosqlStore: nosqlStore{\n\t\t\tlogger: logger,\n\t\t\tdb:     db,\n\t\t\tdc:     dc,\n\t\t},\n\t\tshardID:        shardID,\n\t\ttaskSerializer: taskSerializer,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) GetShardID() int {\n\treturn d.shardID\n}\n\nfunc (d *nosqlExecutionStore) CreateWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalCreateWorkflowExecutionRequest,\n) (*persistence.CreateWorkflowExecutionResponse, error) {\n\n\tnewWorkflow := request.NewWorkflowSnapshot\n\texecutionInfo := newWorkflow.ExecutionInfo\n\tlastWriteVersion := newWorkflow.LastWriteVersion\n\tdomainID := executionInfo.DomainID\n\tworkflowID := executionInfo.WorkflowID\n\trunID := executionInfo.RunID\n\n\tif err := persistence.ValidateCreateWorkflowModeState(\n\t\trequest.Mode,\n\t\tnewWorkflow,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tworkflowRequestWriteMode, err := getWorkflowRequestWriteMode(request.WorkflowRequestMode)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcurrentWorkflowWriteReq, err := d.prepareCurrentWorkflowRequestForCreateWorkflowTxn(domainID, workflowID, runID, executionInfo, lastWriteVersion, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tworkflowExecutionWriteReq, err := d.prepareCreateWorkflowExecutionRequestWithMaps(&newWorkflow, request.CurrentTimeStamp)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttasksByCategory := map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask{}\n\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\tdomainID, workflowID, runID,\n\t\tnewWorkflow.TasksByCategory,\n\t\ttasksByCategory,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tworkflowRequests := d.prepareWorkflowRequestRows(domainID, workflowID, runID, newWorkflow.WorkflowRequests, nil)\n\tworkflowRequestsWriteRequest := &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\tRows:      workflowRequests,\n\t\tWriteMode: workflowRequestWriteMode,\n\t}\n\n\tactiveClusterSelectionPolicyRow := d.prepareActiveClusterSelectionPolicyRow(domainID, workflowID, runID, executionInfo.ActiveClusterSelectionPolicy)\n\n\tshardCondition := &nosqlplugin.ShardCondition{\n\t\tShardID: d.shardID,\n\t\tRangeID: request.RangeID,\n\t}\n\n\terr = d.db.InsertWorkflowExecutionWithTasks(\n\t\tctx,\n\t\tworkflowRequestsWriteRequest,\n\t\tcurrentWorkflowWriteReq,\n\t\tworkflowExecutionWriteReq,\n\t\ttasksByCategory,\n\t\tactiveClusterSelectionPolicyRow,\n\t\tshardCondition,\n\t)\n\tif err != nil {\n\t\tconditionFailureErr, isConditionFailedError := err.(*nosqlplugin.WorkflowOperationConditionFailure)\n\t\tif isConditionFailedError {\n\t\t\tswitch {\n\t\t\tcase conditionFailureErr.UnknownConditionFailureDetails != nil:\n\t\t\t\treturn nil, &persistence.ShardOwnershipLostError{\n\t\t\t\t\tShardID: d.shardID,\n\t\t\t\t\tMsg:     *conditionFailureErr.UnknownConditionFailureDetails,\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.ShardRangeIDNotMatch != nil:\n\t\t\t\treturn nil, &persistence.ShardOwnershipLostError{\n\t\t\t\t\tShardID: d.shardID,\n\t\t\t\t\tMsg: fmt.Sprintf(\"Failed to create workflow execution.  Request RangeID: %v, Actual RangeID: %v\",\n\t\t\t\t\t\trequest.RangeID, *conditionFailureErr.ShardRangeIDNotMatch),\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.CurrentWorkflowConditionFailInfo != nil:\n\t\t\t\treturn nil, &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: *conditionFailureErr.CurrentWorkflowConditionFailInfo,\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.WorkflowExecutionAlreadyExists != nil:\n\t\t\t\treturn nil, &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\tMsg:              conditionFailureErr.WorkflowExecutionAlreadyExists.OtherInfo,\n\t\t\t\t\tStartRequestID:   conditionFailureErr.WorkflowExecutionAlreadyExists.CreateRequestID,\n\t\t\t\t\tRunID:            conditionFailureErr.WorkflowExecutionAlreadyExists.RunID,\n\t\t\t\t\tState:            conditionFailureErr.WorkflowExecutionAlreadyExists.State,\n\t\t\t\t\tCloseStatus:      conditionFailureErr.WorkflowExecutionAlreadyExists.CloseStatus,\n\t\t\t\t\tLastWriteVersion: conditionFailureErr.WorkflowExecutionAlreadyExists.LastWriteVersion,\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.DuplicateRequest != nil:\n\t\t\t\treturn nil, &persistence.DuplicateRequestError{\n\t\t\t\t\tRequestType: conditionFailureErr.DuplicateRequest.RequestType,\n\t\t\t\t\tRunID:       conditionFailureErr.DuplicateRequest.RunID,\n\t\t\t\t}\n\t\t\tdefault:\n\t\t\t\t// If ever runs into this branch, there is bug in the code either in here, or in the implementation of nosql plugin\n\t\t\t\terr := fmt.Errorf(\"unsupported conditionFailureReason error\")\n\t\t\t\td.logger.Error(\"A code bug exists in persistence layer, please investigate ASAP\", tag.Error(err))\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(d.db, \"CreateWorkflowExecution\", err)\n\t}\n\n\treturn &persistence.CreateWorkflowExecutionResponse{}, nil\n}\n\nfunc (d *nosqlExecutionStore) GetWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalGetWorkflowExecutionRequest,\n) (*persistence.InternalGetWorkflowExecutionResponse, error) {\n\n\texecution := request.Execution\n\tstate, err := d.db.SelectWorkflowExecution(ctx, d.shardID, request.DomainID, execution.WorkflowID, execution.RunID)\n\tif err != nil {\n\t\tif d.db.IsNotFoundError(err) {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Workflow execution not found.  WorkflowId: %v, RunId: %v\",\n\t\t\t\t\texecution.WorkflowID, execution.RunID),\n\t\t\t}\n\t\t}\n\n\t\treturn nil, convertCommonErrors(d.db, \"GetWorkflowExecution\", err)\n\t}\n\n\treturn &persistence.InternalGetWorkflowExecutionResponse{State: state}, nil\n}\n\nfunc (d *nosqlExecutionStore) UpdateWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalUpdateWorkflowExecutionRequest,\n) error {\n\tupdateWorkflow := request.UpdateWorkflowMutation\n\tnewWorkflow := request.NewWorkflowSnapshot\n\n\texecutionInfo := updateWorkflow.ExecutionInfo\n\tdomainID := executionInfo.DomainID\n\tworkflowID := executionInfo.WorkflowID\n\trunID := executionInfo.RunID\n\n\tif err := persistence.ValidateUpdateWorkflowModeState(\n\t\trequest.Mode,\n\t\tupdateWorkflow,\n\t\tnewWorkflow,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tworkflowRequestWriteMode, err := getWorkflowRequestWriteMode(request.WorkflowRequestMode)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar currentWorkflowWriteReq *nosqlplugin.CurrentWorkflowWriteRequest\n\n\tswitch request.Mode {\n\tcase persistence.UpdateWorkflowModeIgnoreCurrent:\n\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t}\n\tcase persistence.UpdateWorkflowModeBypassCurrent:\n\t\tif err := d.assertNotCurrentExecution(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t}\n\n\tcase persistence.UpdateWorkflowModeUpdateCurrent:\n\t\tif newWorkflow != nil {\n\t\t\tnewExecutionInfo := newWorkflow.ExecutionInfo\n\t\t\tnewLastWriteVersion := newWorkflow.LastWriteVersion\n\t\t\tnewDomainID := newExecutionInfo.DomainID\n\t\t\t// TODO: ?? would it change at all ??\n\t\t\tnewWorkflowID := newExecutionInfo.WorkflowID\n\t\t\tnewRunID := newExecutionInfo.RunID\n\n\t\t\tif domainID != newDomainID {\n\t\t\t\treturn &types.InternalServiceError{\n\t\t\t\t\tMessage: \"UpdateWorkflowExecution: cannot continue as new to another domain\",\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tShardID:          d.shardID,\n\t\t\t\t\tDomainID:         newDomainID,\n\t\t\t\t\tWorkflowID:       newWorkflowID,\n\t\t\t\t\tRunID:            newRunID,\n\t\t\t\t\tState:            newExecutionInfo.State,\n\t\t\t\t\tCloseStatus:      newExecutionInfo.CloseStatus,\n\t\t\t\t\tCreateRequestID:  newExecutionInfo.CreateRequestID,\n\t\t\t\t\tLastWriteVersion: newLastWriteVersion,\n\t\t\t\t},\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: &runID,\n\t\t\t\t},\n\t\t\t}\n\t\t} else {\n\t\t\tlastWriteVersion := updateWorkflow.LastWriteVersion\n\n\t\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tShardID:          d.shardID,\n\t\t\t\t\tDomainID:         domainID,\n\t\t\t\t\tWorkflowID:       workflowID,\n\t\t\t\t\tRunID:            runID,\n\t\t\t\t\tState:            executionInfo.State,\n\t\t\t\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\t\t\t\tCreateRequestID:  executionInfo.CreateRequestID,\n\t\t\t\t\tLastWriteVersion: lastWriteVersion,\n\t\t\t\t},\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: &runID,\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"UpdateWorkflowExecution: unknown mode: %v\", request.Mode),\n\t\t}\n\t}\n\n\tvar mutateExecution, insertExecution *nosqlplugin.WorkflowExecutionRequest\n\tvar activeClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow\n\tvar workflowRequests []*nosqlplugin.WorkflowRequestRow\n\ttasksByCategory := map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask{}\n\n\t// 1. current\n\tmutateExecution, err = d.prepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer(&updateWorkflow, request.CurrentTimeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\tdomainID, workflowID, updateWorkflow.ExecutionInfo.RunID,\n\t\tupdateWorkflow.TasksByCategory,\n\t\ttasksByCategory,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowRequests = d.prepareWorkflowRequestRows(domainID, workflowID, runID, updateWorkflow.WorkflowRequests, workflowRequests)\n\n\t// 2. new\n\tif newWorkflow != nil {\n\t\tinsertExecution, err = d.prepareCreateWorkflowExecutionRequestWithMaps(newWorkflow, request.CurrentTimeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tactiveClusterSelectionPolicyRow = d.prepareActiveClusterSelectionPolicyRow(\n\t\t\tdomainID,\n\t\t\tnewWorkflow.ExecutionInfo.WorkflowID,\n\t\t\tnewWorkflow.ExecutionInfo.RunID,\n\t\t\tnewWorkflow.ExecutionInfo.ActiveClusterSelectionPolicy,\n\t\t)\n\n\t\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\t\tdomainID, workflowID, newWorkflow.ExecutionInfo.RunID,\n\t\t\tnewWorkflow.TasksByCategory,\n\t\t\ttasksByCategory,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tworkflowRequests = d.prepareWorkflowRequestRows(domainID, workflowID, newWorkflow.ExecutionInfo.RunID, newWorkflow.WorkflowRequests, workflowRequests)\n\t}\n\n\tshardCondition := &nosqlplugin.ShardCondition{\n\t\tShardID: d.shardID,\n\t\tRangeID: request.RangeID,\n\t}\n\n\tworkflowRequestsWriteRequest := &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\tRows:      workflowRequests,\n\t\tWriteMode: workflowRequestWriteMode,\n\t}\n\n\terr = d.db.UpdateWorkflowExecutionWithTasks(\n\t\tctx,\n\t\tworkflowRequestsWriteRequest,\n\t\tcurrentWorkflowWriteReq,\n\t\tmutateExecution,\n\t\tinsertExecution,\n\t\tactiveClusterSelectionPolicyRow,\n\t\tnil, // no workflow to reset here\n\t\ttasksByCategory,\n\t\tshardCondition,\n\t)\n\n\treturn d.processUpdateWorkflowResult(err, request.RangeID)\n}\n\nfunc (d *nosqlExecutionStore) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalConflictResolveWorkflowExecutionRequest,\n) error {\n\tcurrentWorkflow := request.CurrentWorkflowMutation\n\tresetWorkflow := request.ResetWorkflowSnapshot\n\tnewWorkflow := request.NewWorkflowSnapshot\n\n\tdomainID := resetWorkflow.ExecutionInfo.DomainID\n\tworkflowID := resetWorkflow.ExecutionInfo.WorkflowID\n\n\tif err := persistence.ValidateConflictResolveWorkflowModeState(\n\t\trequest.Mode,\n\t\tresetWorkflow,\n\t\tnewWorkflow,\n\t\tcurrentWorkflow,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tworkflowRequestWriteMode, err := getWorkflowRequestWriteMode(request.WorkflowRequestMode)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar currentWorkflowWriteReq *nosqlplugin.CurrentWorkflowWriteRequest\n\tvar prevRunID string\n\n\tswitch request.Mode {\n\tcase persistence.ConflictResolveWorkflowModeBypassCurrent:\n\t\tif err := d.assertNotCurrentExecution(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tresetWorkflow.ExecutionInfo.RunID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t}\n\tcase persistence.ConflictResolveWorkflowModeUpdateCurrent:\n\t\texecutionInfo := resetWorkflow.ExecutionInfo\n\t\tlastWriteVersion := resetWorkflow.LastWriteVersion\n\t\tif newWorkflow != nil {\n\t\t\texecutionInfo = newWorkflow.ExecutionInfo\n\t\t\tlastWriteVersion = newWorkflow.LastWriteVersion\n\t\t}\n\n\t\tif currentWorkflow != nil {\n\t\t\tprevRunID = currentWorkflow.ExecutionInfo.RunID\n\t\t} else {\n\t\t\t// reset workflow is current\n\t\t\tprevRunID = resetWorkflow.ExecutionInfo.RunID\n\t\t}\n\t\tcurrentWorkflowWriteReq = &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\tShardID:          d.shardID,\n\t\t\t\tDomainID:         domainID,\n\t\t\t\tWorkflowID:       workflowID,\n\t\t\t\tRunID:            executionInfo.RunID,\n\t\t\t\tState:            executionInfo.State,\n\t\t\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\t\t\tCreateRequestID:  executionInfo.CreateRequestID,\n\t\t\t\tLastWriteVersion: lastWriteVersion,\n\t\t\t},\n\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\tCurrentRunID: &prevRunID,\n\t\t\t},\n\t\t}\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ConflictResolveWorkflowExecution: unknown mode: %v\", request.Mode),\n\t\t}\n\t}\n\n\tvar mutateExecution, insertExecution, resetExecution *nosqlplugin.WorkflowExecutionRequest\n\tvar workflowRequests []*nosqlplugin.WorkflowRequestRow\n\ttasksByCategory := map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask{}\n\n\t// 1. current\n\tif currentWorkflow != nil {\n\t\tmutateExecution, err = d.prepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer(currentWorkflow, request.CurrentTimeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\t\tdomainID, workflowID, currentWorkflow.ExecutionInfo.RunID,\n\t\t\tcurrentWorkflow.TasksByCategory,\n\t\t\ttasksByCategory,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tworkflowRequests = d.prepareWorkflowRequestRows(domainID, workflowID, currentWorkflow.ExecutionInfo.RunID, currentWorkflow.WorkflowRequests, workflowRequests)\n\t}\n\n\t// 2. reset\n\tresetExecution, err = d.prepareResetWorkflowExecutionRequestWithMapsAndEventBuffer(&resetWorkflow, request.CurrentTimeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\tdomainID, workflowID, resetWorkflow.ExecutionInfo.RunID,\n\t\tresetWorkflow.TasksByCategory,\n\t\ttasksByCategory,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowRequests = d.prepareWorkflowRequestRows(domainID, workflowID, resetWorkflow.ExecutionInfo.RunID, resetWorkflow.WorkflowRequests, workflowRequests)\n\n\t// 3. new\n\tif newWorkflow != nil {\n\t\tinsertExecution, err = d.prepareCreateWorkflowExecutionRequestWithMaps(newWorkflow, request.CurrentTimeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\t// TODO(active-active): make changes here to insert active cluster selection policy.\n\n\t\terr = d.prepareNoSQLTasksForWorkflowTxn(\n\t\t\tdomainID, workflowID, newWorkflow.ExecutionInfo.RunID,\n\t\t\tnewWorkflow.TasksByCategory,\n\t\t\ttasksByCategory,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tworkflowRequests = d.prepareWorkflowRequestRows(domainID, workflowID, newWorkflow.ExecutionInfo.RunID, newWorkflow.WorkflowRequests, workflowRequests)\n\t}\n\n\tshardCondition := &nosqlplugin.ShardCondition{\n\t\tShardID: d.shardID,\n\t\tRangeID: request.RangeID,\n\t}\n\n\tworkflowRequestsWriteRequest := &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\tRows:      workflowRequests,\n\t\tWriteMode: workflowRequestWriteMode,\n\t}\n\n\terr = d.db.UpdateWorkflowExecutionWithTasks(\n\t\tctx, workflowRequestsWriteRequest, currentWorkflowWriteReq,\n\t\tmutateExecution, insertExecution, nil, resetExecution,\n\t\ttasksByCategory,\n\t\tshardCondition)\n\treturn d.processUpdateWorkflowResult(err, request.RangeID)\n}\n\nfunc (d *nosqlExecutionStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.DeleteWorkflowExecutionRequest,\n) error {\n\terr := d.db.DeleteWorkflowExecution(ctx, d.shardID, request.DomainID, request.WorkflowID, request.RunID)\n\tif err != nil {\n\t\treturn convertCommonErrors(d.db, \"DeleteWorkflowExecution\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) DeleteCurrentWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.DeleteCurrentWorkflowExecutionRequest,\n) error {\n\terr := d.db.DeleteCurrentWorkflow(ctx, d.shardID, request.DomainID, request.WorkflowID, request.RunID)\n\tif err != nil {\n\t\treturn convertCommonErrors(d.db, \"DeleteCurrentWorkflowExecution\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) DeleteActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) error {\n\terr := d.db.DeleteActiveClusterSelectionPolicy(ctx, d.shardID, domainID, workflowID, runID)\n\tif err != nil {\n\t\treturn convertCommonErrors(d.db, \"DeleteActiveClusterSelectionPolicy\", err)\n\t}\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) GetCurrentExecution(\n\tctx context.Context,\n\trequest *persistence.GetCurrentExecutionRequest,\n) (*persistence.GetCurrentExecutionResponse,\n\terror) {\n\tresult, err := d.db.SelectCurrentWorkflow(ctx, d.shardID, request.DomainID, request.WorkflowID)\n\n\tif err != nil {\n\t\tif d.db.IsNotFoundError(err) {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Workflow execution not found.  WorkflowId: %v\",\n\t\t\t\t\trequest.WorkflowID),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(d.db, \"GetCurrentExecution\", err)\n\t}\n\n\treturn &persistence.GetCurrentExecutionResponse{\n\t\tRunID:            result.RunID,\n\t\tStartRequestID:   result.CreateRequestID,\n\t\tState:            result.State,\n\t\tCloseStatus:      result.CloseStatus,\n\t\tLastWriteVersion: result.LastWriteVersion,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) ListCurrentExecutions(\n\tctx context.Context,\n\trequest *persistence.ListCurrentExecutionsRequest,\n) (*persistence.ListCurrentExecutionsResponse, error) {\n\texecutions, token, err := d.db.SelectAllCurrentWorkflows(ctx, d.shardID, request.PageToken, request.PageSize)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"ListCurrentExecutions\", err)\n\t}\n\treturn &persistence.ListCurrentExecutionsResponse{\n\t\tExecutions: executions,\n\t\tPageToken:  token,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) IsWorkflowExecutionExists(\n\tctx context.Context,\n\trequest *persistence.IsWorkflowExecutionExistsRequest,\n) (*persistence.IsWorkflowExecutionExistsResponse, error) {\n\texists, err := d.db.IsWorkflowExecutionExists(ctx, d.shardID, request.DomainID, request.WorkflowID, request.RunID)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"IsWorkflowExecutionExists\", err)\n\t}\n\treturn &persistence.IsWorkflowExecutionExistsResponse{\n\t\tExists: exists,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) ListConcreteExecutions(\n\tctx context.Context,\n\trequest *persistence.ListConcreteExecutionsRequest,\n) (*persistence.InternalListConcreteExecutionsResponse, error) {\n\texecutions, nextPageToken, err := d.db.SelectAllWorkflowExecutions(ctx, d.shardID, request.PageToken, request.PageSize)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"ListConcreteExecutions\", err)\n\t}\n\treturn &persistence.InternalListConcreteExecutionsResponse{\n\t\tExecutions:    executions,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) PutReplicationTaskToDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalPutReplicationTaskToDLQRequest,\n) error {\n\terr := d.db.InsertReplicationDLQTask(ctx, d.shardID, request.SourceClusterName, &nosqlplugin.HistoryMigrationTask{\n\t\tReplication: request.TaskInfo,\n\t\tTask:        nil, // TODO: encode task infor into datablob\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(d.db, \"PutReplicationTaskToDLQ\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) GetReplicationTasksFromDLQ(\n\tctx context.Context,\n\trequest *persistence.GetReplicationTasksFromDLQRequest,\n) (*persistence.GetHistoryTasksResponse, error) {\n\tif request.ReadLevel > request.MaxReadLevel {\n\t\treturn nil, &types.BadRequestError{Message: \"ReadLevel cannot be higher than MaxReadLevel\"}\n\t}\n\ttasks, nextPageToken, err := d.db.SelectReplicationDLQTasksOrderByTaskID(ctx, d.shardID, request.SourceClusterName, request.BatchSize, request.NextPageToken, request.ReadLevel, request.MaxReadLevel)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"GetReplicationTasksFromDLQ\", err)\n\t}\n\tvar tTasks []persistence.Task\n\tfor _, t := range tasks {\n\t\ttask, err := t.Replication.ToTask()\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"GetReplicationTasksFromDLQ\", err)\n\t\t}\n\t\ttTasks = append(tTasks, task)\n\t}\n\treturn &persistence.GetHistoryTasksResponse{\n\t\tTasks:         tTasks,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) GetReplicationDLQSize(\n\tctx context.Context,\n\trequest *persistence.GetReplicationDLQSizeRequest,\n) (*persistence.GetReplicationDLQSizeResponse, error) {\n\n\tsize, err := d.db.SelectReplicationDLQTasksCount(ctx, d.shardID, request.SourceClusterName)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"GetReplicationDLQSize\", err)\n\t}\n\treturn &persistence.GetReplicationDLQSizeResponse{\n\t\tSize: size,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) DeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *persistence.DeleteReplicationTaskFromDLQRequest,\n) error {\n\n\terr := d.db.DeleteReplicationDLQTask(ctx, d.shardID, request.SourceClusterName, request.TaskID)\n\tif err != nil {\n\t\treturn convertCommonErrors(d.db, \"DeleteReplicationTaskFromDLQ\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) RangeDeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *persistence.RangeDeleteReplicationTaskFromDLQRequest,\n) (*persistence.RangeDeleteReplicationTaskFromDLQResponse, error) {\n\n\terr := d.db.RangeDeleteReplicationDLQTasks(ctx, d.shardID, request.SourceClusterName, request.InclusiveBeginTaskID, request.ExclusiveEndTaskID)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(d.db, \"RangeDeleteReplicationTaskFromDLQ\", err)\n\t}\n\n\treturn &persistence.RangeDeleteReplicationTaskFromDLQResponse{TasksCompleted: persistence.UnknownNumRowsAffected}, nil\n}\n\nfunc (d *nosqlExecutionStore) CreateFailoverMarkerTasks(\n\tctx context.Context,\n\trequest *persistence.CreateFailoverMarkersRequest,\n) error {\n\n\tvar nosqlTasks []*nosqlplugin.HistoryMigrationTask\n\tfor i, task := range request.Markers {\n\t\tts := []persistence.Task{task}\n\n\t\ttasks, err := d.prepareReplicationTasksForWorkflowTxn(task.DomainID, rowTypeReplicationWorkflowID, rowTypeReplicationRunID, ts)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttasks[i].Replication.CurrentTimeStamp = request.CurrentTimeStamp\n\t\tnosqlTasks = append(nosqlTasks, tasks...)\n\t}\n\n\terr := d.db.InsertReplicationTask(ctx, nosqlTasks, nosqlplugin.ShardCondition{\n\t\tShardID: d.shardID,\n\t\tRangeID: request.RangeID,\n\t})\n\n\tif err != nil {\n\t\tconditionFailureErr, isConditionFailedError := err.(*nosqlplugin.ShardOperationConditionFailure)\n\t\tif isConditionFailedError {\n\t\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: d.shardID,\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to create workflow execution.  Request RangeID: %v, columns: (%v)\",\n\t\t\t\t\tconditionFailureErr.RangeID, conditionFailureErr.Details),\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) GetHistoryTasks(\n\tctx context.Context,\n\trequest *persistence.GetHistoryTasksRequest,\n) (*persistence.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.Type() {\n\tcase persistence.HistoryTaskCategoryTypeImmediate:\n\t\treturn d.getImmediateHistoryTasks(ctx, request)\n\tcase persistence.HistoryTaskCategoryTypeScheduled:\n\t\treturn d.getScheduledHistoryTasks(ctx, request)\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) getImmediateHistoryTasks(\n\tctx context.Context,\n\trequest *persistence.GetHistoryTasksRequest,\n) (*persistence.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\ttasks, nextPageToken, err := d.db.SelectTransferTasksOrderByTaskID(ctx, d.shardID, request.PageSize, request.NextPageToken, request.InclusiveMinTaskKey.GetTaskID(), request.ExclusiveMaxTaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t}\n\t\ttTasks := make([]persistence.Task, 0, len(tasks))\n\t\tfor _, t := range tasks {\n\t\t\tif d.dc.ReadNoSQLHistoryTaskFromDataBlob() && t.Task != nil {\n\t\t\t\ttask, err := d.taskSerializer.DeserializeTask(request.TaskCategory, t.Task)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttask.SetTaskID(t.TaskID)\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t} else {\n\t\t\t\ttask, err := t.Transfer.ToTask()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t}\n\t\t}\n\t\treturn &persistence.GetHistoryTasksResponse{\n\t\t\tTasks:         tTasks,\n\t\t\tNextPageToken: nextPageToken,\n\t\t}, nil\n\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\ttasks, nextPageToken, err := d.db.SelectReplicationTasksOrderByTaskID(ctx, d.shardID, request.PageSize, request.NextPageToken, request.InclusiveMinTaskKey.GetTaskID(), request.ExclusiveMaxTaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t}\n\t\ttTasks := make([]persistence.Task, 0, len(tasks))\n\t\tfor _, t := range tasks {\n\t\t\tif d.dc.ReadNoSQLHistoryTaskFromDataBlob() && t.Task != nil {\n\t\t\t\ttask, err := d.taskSerializer.DeserializeTask(request.TaskCategory, t.Task)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttask.SetTaskID(t.TaskID)\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t} else {\n\t\t\t\ttask, err := t.Replication.ToTask()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetImmediateHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t}\n\t\t}\n\t\treturn &persistence.GetHistoryTasksResponse{\n\t\t\tTasks:         tTasks,\n\t\t\tNextPageToken: nextPageToken,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) getScheduledHistoryTasks(\n\tctx context.Context,\n\trequest *persistence.GetHistoryTasksRequest,\n) (*persistence.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\ttimers, nextPageToken, err := d.db.SelectTimerTasksOrderByVisibilityTime(ctx, d.shardID, request.PageSize, request.NextPageToken, request.InclusiveMinTaskKey.GetScheduledTime(), request.ExclusiveMaxTaskKey.GetScheduledTime())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"GetScheduledHistoryTasks\", err)\n\t\t}\n\t\ttTasks := make([]persistence.Task, 0, len(timers))\n\t\tfor _, t := range timers {\n\t\t\tif d.dc.ReadNoSQLHistoryTaskFromDataBlob() && t.Task != nil {\n\t\t\t\ttask, err := d.taskSerializer.DeserializeTask(request.TaskCategory, t.Task)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetScheduledHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttask.SetTaskID(t.TaskID)\n\t\t\t\ttask.SetVisibilityTimestamp(t.ScheduledTime)\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t} else {\n\t\t\t\ttask, err := t.Timer.ToTask()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(d.db, \"GetScheduledHistoryTasks\", err)\n\t\t\t\t}\n\t\t\t\ttTasks = append(tTasks, task)\n\t\t\t}\n\t\t}\n\t\treturn &persistence.GetHistoryTasksResponse{\n\t\t\tTasks:         tTasks,\n\t\t\tNextPageToken: nextPageToken,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) CompleteHistoryTask(\n\tctx context.Context,\n\trequest *persistence.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.Type() {\n\tcase persistence.HistoryTaskCategoryTypeScheduled:\n\t\treturn d.completeScheduledHistoryTask(ctx, request)\n\tcase persistence.HistoryTaskCategoryTypeImmediate:\n\t\treturn d.completeImmediateHistoryTask(ctx, request)\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) completeScheduledHistoryTask(\n\tctx context.Context,\n\trequest *persistence.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\terr := d.db.DeleteTimerTask(ctx, d.shardID, request.TaskKey.GetTaskID(), request.TaskKey.GetScheduledTime())\n\t\tif err != nil {\n\t\t\treturn convertCommonErrors(d.db, \"CompleteScheduledHistoryTask\", err)\n\t\t}\n\t\treturn nil\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) completeImmediateHistoryTask(\n\tctx context.Context,\n\trequest *persistence.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\terr := d.db.DeleteTransferTask(ctx, d.shardID, request.TaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn convertCommonErrors(d.db, \"CompleteImmediateHistoryTask\", err)\n\t\t}\n\t\treturn nil\n\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\terr := d.db.DeleteReplicationTask(ctx, d.shardID, request.TaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn convertCommonErrors(d.db, \"CompleteImmediateHistoryTask\", err)\n\t\t}\n\t\treturn nil\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) RangeCompleteHistoryTask(\n\tctx context.Context,\n\trequest *persistence.RangeCompleteHistoryTaskRequest,\n) (*persistence.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.Type() {\n\tcase persistence.HistoryTaskCategoryTypeScheduled:\n\t\treturn d.rangeCompleteScheduledHistoryTask(ctx, request)\n\tcase persistence.HistoryTaskCategoryTypeImmediate:\n\t\treturn d.rangeCompleteImmediateHistoryTask(ctx, request)\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (d *nosqlExecutionStore) rangeCompleteScheduledHistoryTask(\n\tctx context.Context,\n\trequest *persistence.RangeCompleteHistoryTaskRequest,\n) (*persistence.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\terr := d.db.RangeDeleteTimerTasks(ctx, d.shardID, request.InclusiveMinTaskKey.GetScheduledTime(), request.ExclusiveMaxTaskKey.GetScheduledTime())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"RangeCompleteTimerTask\", err)\n\t\t}\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n\treturn &persistence.RangeCompleteHistoryTaskResponse{TasksCompleted: persistence.UnknownNumRowsAffected}, nil\n}\n\nfunc (d *nosqlExecutionStore) rangeCompleteImmediateHistoryTask(\n\tctx context.Context,\n\trequest *persistence.RangeCompleteHistoryTaskRequest,\n) (*persistence.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\terr := d.db.RangeDeleteTransferTasks(ctx, d.shardID, request.InclusiveMinTaskKey.GetTaskID(), request.ExclusiveMaxTaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"RangeCompleteTransferTask\", err)\n\t\t}\n\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\terr := d.db.RangeDeleteReplicationTasks(ctx, d.shardID, request.ExclusiveMaxTaskKey.GetTaskID())\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(d.db, \"RangeCompleteReplicationTask\", err)\n\t\t}\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n\treturn &persistence.RangeCompleteHistoryTaskResponse{TasksCompleted: persistence.UnknownNumRowsAffected}, nil\n}\n\nfunc (d *nosqlExecutionStore) GetActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, wfID, rID string,\n) (*persistence.DataBlob, error) {\n\trow, err := d.db.SelectActiveClusterSelectionPolicy(ctx, d.shardID, domainID, wfID, rID)\n\tif err != nil {\n\t\tif d.db.IsNotFoundError(err) {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Active cluster selection policy not found.  DomainId: %v, WorkflowId: %v, RunId: %v\", domainID, wfID, rID),\n\t\t\t}\n\t\t}\n\n\t\treturn nil, convertCommonErrors(d.db, \"GetActiveClusterSelectionPolicy\", err)\n\t}\n\n\tif row == nil {\n\t\treturn nil, nil\n\t}\n\n\treturn row.Policy, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_execution_store_test.go",
    "content": "// Copyright (c) 2023 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\nfunc TestCreateWorkflowExecution(t *testing.T) {\n\tctx := context.Background()\n\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB, int) // Now accepts shard ID as parameter\n\t\texpectedResp  *persistence.CreateWorkflowExecutionResponse\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\texpectedResp:  &persistence.CreateWorkflowExecutionResponse{},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"ShardRangeIDNotMatch condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tShardRangeIDNotMatch: common.Int64Ptr(456),\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(err)\n\t\t\t},\n\t\t\texpectedResp: nil, // No response expected on error\n\t\t\texpectedError: &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: 123,\n\t\t\t\tMsg:     \"Failed to create workflow execution.  Request RangeID: 123, Actual RangeID: 456\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionAlreadyExists condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tWorkflowExecutionAlreadyExists: &nosqlplugin.WorkflowExecutionAlreadyExists{\n\t\t\t\t\t\tOtherInfo:        \"Workflow with ID already exists\",\n\t\t\t\t\t\tCreateRequestID:  \"existing-request-id\",\n\t\t\t\t\t\tRunID:            \"existing-run-id\",\n\t\t\t\t\t\tState:            persistence.WorkflowStateCompleted,\n\t\t\t\t\t\tCloseStatus:      persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t\t\tLastWriteVersion: 123,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(err)\n\t\t\t},\n\t\t\texpectedResp: nil, // No response expected on error\n\t\t\texpectedError: &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\tMsg:              \"Workflow with ID already exists\",\n\t\t\t\tStartRequestID:   \"existing-request-id\",\n\t\t\t\tRunID:            \"existing-run-id\",\n\t\t\t\tState:            persistence.WorkflowStateCompleted,\n\t\t\t\tCloseStatus:      persistence.WorkflowCloseStatusCompleted,\n\t\t\t\tLastWriteVersion: 123,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Unknown condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tUnknownConditionFailureDetails: common.StringPtr(\"Unknown error occurred during operation\"),\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(err)\n\t\t\t},\n\t\t\texpectedResp: nil,\n\t\t\texpectedError: &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: 123,\n\t\t\t\tMsg:     \"Unknown error occurred during operation\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Current workflow condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\"Current workflow condition failed\"),\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(err)\n\t\t\t},\n\t\t\texpectedResp: nil,\n\t\t\texpectedError: &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\tMsg: \"Current workflow condition failed\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Unexpected error type leading to unsupported condition failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\t// Simulate returning an unexpected error type from the mock\n\t\t\t\tunexpectedErr := errors.New(\"unexpected error type\")\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(unexpectedErr)\n\t\t\t},\n\t\t\texpectedResp:  nil,\n\t\t\texpectedError: fmt.Errorf(\"unsupported conditionFailureReason error\"), // Expected generic error for unexpected conditions\n\t\t},\n\t\t{\n\t\t\tname: \"Duplicate request error\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t})\n\t\t\t},\n\t\t\texpectedResp: nil,\n\t\t\texpectedError: &persistence.DuplicateRequestError{\n\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tstore := newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\tshardID := store.GetShardID()\n\n\t\t\ttc.setupMock(mockDB, shardID)\n\n\t\t\tresp, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\n\t\t\tif diff := cmp.Diff(tc.expectedResp, resp); diff != \"\" {\n\t\t\t\tt.Errorf(\"CreateWorkflowExecution() response mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecution(t *testing.T) {\n\tctx := context.Background()\n\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB, int)\n\t\trequest       func() *persistence.InternalUpdateWorkflowExecutionRequest\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), nil, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\trequest:       newUpdateWorkflowExecutionRequest,\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - UpdateWorkflowModeIgnoreCurrent\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), nil, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\trequest: func() *persistence.InternalUpdateWorkflowExecutionRequest {\n\t\t\t\treq := newUpdateWorkflowExecutionRequest()\n\t\t\t\treq.Mode = persistence.UpdateWorkflowModeIgnoreCurrent\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Duplicate request error\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), nil, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t})\n\t\t\t},\n\t\t\trequest: newUpdateWorkflowExecutionRequest,\n\t\t\texpectedError: &persistence.DuplicateRequestError{\n\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowModeBypassCurrent - assertNotCurrentExecution failure\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), nil, gomock.Any(), gomock.Any()).\n\t\t\t\t\tTimes(0)\n\t\t\t},\n\t\t\trequest: func() *persistence.InternalUpdateWorkflowExecutionRequest {\n\t\t\t\treq := newUpdateWorkflowExecutionRequest()\n\t\t\t\treq.Mode = persistence.UpdateWorkflowModeBypassCurrent\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"assertNotCurrentExecution failure\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Unknown update mode\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t},\n\t\t\trequest: func() *persistence.InternalUpdateWorkflowExecutionRequest {\n\t\t\t\treq := newUpdateWorkflowExecutionRequest()\n\t\t\t\treq.Mode = -1\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"UpdateWorkflowExecution: unknown mode: -1\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"Bypass_current_execution_failure_due_to_assertNotCurrentExecution\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {},\n\t\t\trequest: func() *persistence.InternalUpdateWorkflowExecutionRequest {\n\t\t\t\treq := newUpdateWorkflowExecutionRequest()\n\t\t\t\treq.Mode = persistence.UpdateWorkflowModeBypassCurrent\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"assertNotCurrentExecution failure\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Update with new snapshot (continue as new)\",\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, shardID int) {\n\t\t\t\t// validate the NewWorkflowSnapshot's info is used to create the active cluster selection policy row\n\t\t\t\tinfo := newUpdateWorkflowExecutionRequestForContinueAsNew().NewWorkflowSnapshot.ExecutionInfo\n\t\t\t\tplcy := &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   info.DomainID,\n\t\t\t\t\tWorkflowID: info.WorkflowID,\n\t\t\t\t\tRunID:      info.RunID,\n\t\t\t\t\tPolicy:     info.ActiveClusterSelectionPolicy,\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Not(nil), plcy, nil, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest: func() *persistence.InternalUpdateWorkflowExecutionRequest {\n\t\t\t\treq := newUpdateWorkflowExecutionRequestForContinueAsNew()\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(controller)\n\t\t\tstore, _ := NewExecutionStore(1, mockDB, log.NewNoop(), mockTaskSerializer, &persistence.DynamicConfiguration{\n\t\t\t\tEnableHistoryTaskDualWriteMode: func(...dynamicproperties.FilterOption) bool { return false },\n\t\t\t})\n\n\t\t\ttc.setupMock(mockDB, 1)\n\n\t\t\treq := tc.request()\n\t\t\terr := store.UpdateWorkflowExecution(ctx, req)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.IsType(t, tc.expectedError, err, \"Error type does not match the expected one.\")\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNosqlExecutionStore(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*gomock.Controller) *nosqlExecutionStore\n\t\ttestFunc      func(*nosqlExecutionStore) error\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution failure - workflow already exists\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.WorkflowExecutionAlreadyStartedError{}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &persistence.WorkflowExecutionAlreadyStartedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution failure - shard ownership lost\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ShardOwnershipLostError{ShardID: shardID, Msg: \"shard ownership lost\"}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &persistence.ShardOwnershipLostError{},\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution failure - current workflow condition failed\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.CurrentWorkflowConditionFailedError{Msg: \"current workflow condition failed\"}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &persistence.CurrentWorkflowConditionFailedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution failure - generic internal service error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.InternalServiceError{Message: \"generic internal service error\"}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{},\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution failure - duplicate request error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\t\t\t\tRunID: \"abc\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.CreateWorkflowExecution(ctx, newCreateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &persistence.DuplicateRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"GetWorkflowExecution success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectWorkflowExecution(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&nosqlplugin.WorkflowExecution{}, nil).Times(1)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.GetWorkflowExecution(ctx, newGetWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"GetWorkflowExecution failure - not found\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectWorkflowExecution(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.GetWorkflowExecution(ctx, newGetWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &types.EntityNotExistsError{},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowExecution success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Nil(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\terr := store.UpdateWorkflowExecution(ctx, newUpdateWorkflowExecutionRequest())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowExecution failure - invalid update mode\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\t// No operation expected on the DB due to invalid mode\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\trequest := newUpdateWorkflowExecutionRequest()\n\t\t\t\trequest.Mode = persistence.UpdateWorkflowMode(-1)\n\t\t\t\treturn store.UpdateWorkflowExecution(ctx, request)\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowExecution failure - condition not met\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tconditionFailure := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tUnknownConditionFailureDetails: common.StringPtr(\"condition not met\"),\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Nil(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(conditionFailure).Times(1)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.UpdateWorkflowExecution(ctx, newUpdateWorkflowExecutionRequest())\n\t\t\t},\n\t\t\texpectedError: &persistence.ConditionFailedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowExecution failure - operational error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tUpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Nil(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"database is unavailable\")).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.UpdateWorkflowExecution(ctx, newUpdateWorkflowExecutionRequest())\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"database is unavailable\"},\n\t\t},\n\t\t{\n\t\t\tname: \"DeleteWorkflowExecution success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteWorkflowExecution(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.DeleteWorkflowExecution(ctx, &persistence.DeleteWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   \"domainID\",\n\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"DeleteWorkflowExecution failure - workflow does not exist\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteWorkflowExecution(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.EntityNotExistsError{Message: \"workflow does not exist\"})\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.DeleteWorkflowExecution(ctx, &persistence.DeleteWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   \"domainID\",\n\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: &types.EntityNotExistsError{Message: \"workflow does not exist\"},\n\t\t},\n\t\t{\n\t\t\tname: \"DeleteCurrentWorkflowExecution success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteCurrentWorkflow(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   \"domainID\",\n\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"DeleteCurrentWorkflowExecution failure - current workflow does not exist\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteCurrentWorkflow(ctx, shardID, gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.EntityNotExistsError{Message: \"current workflow does not exist\"})\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\treturn store.DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   \"domainID\",\n\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: &types.EntityNotExistsError{Message: \"current workflow does not exist\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ListCurrentExecutions success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectAllCurrentWorkflows(ctx, shardID, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn([]*persistence.CurrentWorkflowExecution{}, nil, nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.ListCurrentExecutions(ctx, &persistence.ListCurrentExecutionsRequest{})\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"ListCurrentExecutions failure - database error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectAllCurrentWorkflows(ctx, shardID, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil, errors.New(\"database error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.ListCurrentExecutions(ctx, &persistence.ListCurrentExecutionsRequest{})\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"database error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ListConcreteExecutions success - has executions\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\t// Corrected return value type to match expected method signature\n\t\t\t\texecutions := []*persistence.InternalListConcreteExecutionsEntity{\n\t\t\t\t\t{\n\t\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectAllWorkflowExecutions(ctx, shardID, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(executions, nil, nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\tresp, err := store.ListConcreteExecutions(ctx, &persistence.ListConcreteExecutionsRequest{})\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tif len(resp.Executions) == 0 {\n\t\t\t\t\treturn errors.New(\"expected to find executions\")\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"ListConcreteExecutions failure - database error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectAllWorkflowExecutions(ctx, shardID, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil, errors.New(\"database error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.ListConcreteExecutions(ctx, &persistence.ListConcreteExecutionsRequest{})\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"database error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"PutReplicationTaskToDLQ success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\treplicationTaskInfo := newInternalReplicationTaskInfo()\n\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertReplicationDLQTask(ctx, shardID, \"sourceCluster\", &nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t\tReplication: &replicationTaskInfo,\n\t\t\t\t\t\tTask:        nil,\n\t\t\t\t\t}).Return(nil)\n\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\ttaskInfo := newInternalReplicationTaskInfo()\n\t\t\t\treturn store.PutReplicationTaskToDLQ(ctx, &persistence.InternalPutReplicationTaskToDLQRequest{\n\t\t\t\t\tSourceClusterName: \"sourceCluster\",\n\t\t\t\t\tTaskInfo:          &taskInfo,\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationTasksFromDLQ success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\n\t\t\t\tnextPageToken := []byte(\"next-page-token\")\n\t\t\t\treplicationTasks := []*nosqlplugin.HistoryMigrationTask{}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectReplicationDLQTasksOrderByTaskID(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\tshardID,\n\t\t\t\t\t\t\"sourceCluster\",\n\t\t\t\t\t\t10,\n\t\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t\tint64(0),\n\t\t\t\t\t\tint64(100),\n\t\t\t\t\t).Return(replicationTasks, nextPageToken, nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\tinitialNextPageToken := []byte{}\n\t\t\t\t_, err := store.GetReplicationTasksFromDLQ(ctx, &persistence.GetReplicationTasksFromDLQRequest{\n\t\t\t\t\tSourceClusterName: \"sourceCluster\",\n\t\t\t\t\tBatchSize:         10,\n\t\t\t\t\tNextPageToken:     initialNextPageToken,\n\t\t\t\t\tReadLevel:         0,\n\t\t\t\t\tMaxReadLevel:      100,\n\t\t\t\t})\n\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationTasksFromDLQ failure - invalid read levels\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\treturn newTestNosqlExecutionStore(nosqlplugin.NewMockDB(ctrl), log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.GetReplicationTasksFromDLQ(ctx, &persistence.GetReplicationTasksFromDLQRequest{\n\t\t\t\t\tSourceClusterName: \"sourceCluster\",\n\t\t\t\t\tReadLevel:         100,\n\t\t\t\t\tMaxReadLevel:      50,\n\t\t\t\t\tBatchSize:         10,\n\t\t\t\t\tNextPageToken:     []byte{},\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: &types.BadRequestError{Message: \"ReadLevel cannot be higher than MaxReadLevel\"},\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationDLQSize success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectReplicationDLQTasksCount(ctx, shardID, \"sourceCluster\").\n\t\t\t\t\tReturn(int64(42), nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\tresp, err := store.GetReplicationDLQSize(ctx, &persistence.GetReplicationDLQSizeRequest{\n\t\t\t\t\tSourceClusterName: \"sourceCluster\",\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tif resp.Size != 42 {\n\t\t\t\t\treturn errors.New(\"unexpected DLQ size\")\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"GetReplicationDLQSize failure - invalid source cluster name\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *nosqlExecutionStore {\n\t\t\t\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tSelectReplicationDLQTasksCount(ctx, shardID, \"\").\n\t\t\t\t\tReturn(int64(0), nil)\n\t\t\t\treturn newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\t\t\t},\n\t\t\ttestFunc: func(store *nosqlExecutionStore) error {\n\t\t\t\t_, err := store.GetReplicationDLQSize(ctx, &persistence.GetReplicationDLQSizeRequest{\n\t\t\t\t\tSourceClusterName: \"\",\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tstore := tc.setupMock(ctrl)\n\t\t\terr := tc.testFunc(store)\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\tvar expectedErrType error\n\t\t\t\trequire.ErrorAs(t, err, &expectedErrType, \"Expected error type does not match.\")\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteReplicationTaskFromDLQ(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\tsourceCluster string\n\t\ttaskID        int64\n\t\tsetupMock     func(*nosqlplugin.MockDB)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname:          \"success\",\n\t\t\tsourceCluster: \"sourceCluster\",\n\t\t\ttaskID:        1,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteReplicationDLQTask(ctx, shardID, \"sourceCluster\", int64(1)).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname:          \"database error\",\n\t\t\tsourceCluster: \"sourceCluster\",\n\t\t\ttaskID:        1,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tDeleteReplicationDLQTask(ctx, shardID, \"sourceCluster\", int64(1)).\n\t\t\t\t\tReturn(errors.New(\"database error\"))\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"database error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tstore := newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\terr := store.DeleteReplicationTaskFromDLQ(ctx, &persistence.DeleteReplicationTaskFromDLQRequest{\n\t\t\t\tSourceClusterName: tc.sourceCluster,\n\t\t\t\tTaskID:            tc.taskID,\n\t\t\t})\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteReplicationTaskFromDLQ(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname             string\n\t\tsourceCluster    string\n\t\tinclusiveBeginID int64\n\t\texclusiveEndID   int64\n\t\tsetupMock        func(*nosqlplugin.MockDB)\n\t\texpectedError    error\n\t}{\n\t\t{\n\t\t\tname:             \"success\",\n\t\t\tsourceCluster:    \"sourceCluster\",\n\t\t\tinclusiveBeginID: 1,\n\t\t\texclusiveEndID:   100,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tRangeDeleteReplicationDLQTasks(ctx, shardID, \"sourceCluster\", int64(1), int64(100)).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname:             \"database error\",\n\t\t\tsourceCluster:    \"sourceCluster\",\n\t\t\tinclusiveBeginID: 1,\n\t\t\texclusiveEndID:   100,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tRangeDeleteReplicationDLQTasks(ctx, shardID, \"sourceCluster\", int64(1), int64(100)).\n\t\t\t\t\tReturn(errors.New(\"database error\"))\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"database error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tstore := newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\t_, err := store.RangeDeleteReplicationTaskFromDLQ(ctx, &persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\t\tSourceClusterName:    tc.sourceCluster,\n\t\t\t\tInclusiveBeginTaskID: tc.inclusiveBeginID,\n\t\t\t\tExclusiveEndTaskID:   tc.exclusiveEndID,\n\t\t\t})\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateFailoverMarkerTasks(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\trangeID       int64\n\t\tmarkers       []*persistence.FailoverMarkerTask\n\t\tsetupMock     func(*nosqlplugin.MockDB, *serialization.MockTaskSerializer)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname:    \"success\",\n\t\t\trangeID: 123,\n\t\t\tmarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t{\n\t\t\t\t\tTaskData: persistence.TaskData{},\n\t\t\t\t\tDomainID: \"testDomainID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"1\"),\n\t\t\t\t\tEncoding: commonconstants.EncodingTypeThriftRW,\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertReplicationTask(ctx, gomock.Any(), nosqlplugin.ShardCondition{ShardID: shardID, RangeID: 123}).\n\t\t\t\t\tReturn(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname:    \"serialization error\",\n\t\t\trangeID: 123,\n\t\t\tmarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t{\n\t\t\t\t\tTaskData: persistence.TaskData{},\n\t\t\t\t\tDomainID: \"testDomainID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedError: errors.New(\"some error\"),\n\t\t},\n\t\t{\n\t\t\tname:    \"CreateFailoverMarkerTasks failure - ShardOperationConditionFailure\",\n\t\t\trangeID: 123,\n\t\t\tmarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t{\n\t\t\t\t\tTaskData: persistence.TaskData{},\n\t\t\t\t\tDomainID: \"testDomainID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"1\"),\n\t\t\t\t\tEncoding: commonconstants.EncodingTypeThriftRW,\n\t\t\t\t}, nil)\n\t\t\t\tconditionFailureErr := &nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 123,                      // Use direct int64 value\n\t\t\t\t\tDetails: \"Shard condition failed\", // Use direct string value\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().\n\t\t\t\t\tInsertReplicationTask(ctx, gomock.Any(), nosqlplugin.ShardCondition{ShardID: shardID, RangeID: 123}).\n\t\t\t\t\tReturn(conditionFailureErr) // Simulate ShardOperationConditionFailure\n\t\t\t},\n\t\t\texpectedError: &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: shardID,\n\t\t\t\tMsg:     \"Failed to create workflow execution.  Request RangeID: 123, columns: (Shard condition failed)\",\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(controller)\n\t\t\tstore := newTestNosqlExecutionStoreWithTaskSerializer(mockDB, log.NewNoop(), mockTaskSerializer)\n\n\t\t\ttc.setupMock(mockDB, mockTaskSerializer)\n\n\t\t\terr := store.CreateFailoverMarkerTasks(ctx, &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: tc.rangeID,\n\t\t\t\tMarkers: tc.markers,\n\t\t\t})\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsWorkflowExecutionExists(t *testing.T) {\n\tctx := context.Background()\n\tgomockController := gomock.NewController(t)\n\n\tmockDB := nosqlplugin.NewMockDB(gomockController)\n\tstore := &nosqlExecutionStore{\n\t\tshardID:    1,\n\t\tnosqlStore: nosqlStore{db: mockDB},\n\t}\n\n\tdomainID := \"testDomainID\"\n\tworkflowID := \"testWorkflowID\"\n\trunID := \"testRunID\"\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMock      func()\n\t\trequest        *persistence.IsWorkflowExecutionExistsRequest\n\t\texpectedExists bool\n\t\texpectedError  error\n\t}{\n\t\t{\n\t\t\tname: \"Workflow Execution Exists\",\n\t\t\tsetupMock: func() {\n\t\t\t\tmockDB.EXPECT().IsWorkflowExecutionExists(ctx, store.shardID, domainID, workflowID, runID).Return(true, nil)\n\t\t\t},\n\t\t\trequest: &persistence.IsWorkflowExecutionExistsRequest{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\texpectedExists: true,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Workflow Execution Does Not Exist\",\n\t\t\tsetupMock: func() {\n\t\t\t\tmockDB.EXPECT().IsWorkflowExecutionExists(ctx, store.shardID, domainID, workflowID, runID).Return(false, nil)\n\t\t\t},\n\t\t\trequest: &persistence.IsWorkflowExecutionExistsRequest{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\texpectedExists: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\tresponse, err := store.IsWorkflowExecutionExists(ctx, tc.request)\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\t// Optionally, further validate the error type or message\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\trequire.Equal(t, tc.expectedExists, response.Exists)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConflictResolveWorkflowExecution(t *testing.T) {\n\tctx := context.Background()\n\tgomockController := gomock.NewController(t)\n\n\tmockDB := nosqlplugin.NewMockDB(gomockController)\n\tmockTaskSerializer := serialization.NewMockTaskSerializer(gomockController)\n\tstore, err := NewExecutionStore(1, mockDB, log.NewNoop(), mockTaskSerializer, &persistence.DynamicConfiguration{\n\t\tEnableHistoryTaskDualWriteMode: func(...dynamicproperties.FilterOption) bool { return false },\n\t})\n\trequire.NoError(t, err)\n\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMocks    func()\n\t\tgetRequest    func() *persistence.InternalConflictResolveWorkflowExecutionRequest\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"DB Error on Reset Execution Insertion\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockDB.EXPECT().UpdateWorkflowExecutionWithTasks(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"DB error\")).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t},\n\t\t\tgetRequest: func() *persistence.InternalConflictResolveWorkflowExecutionRequest {\n\t\t\t\treturn newConflictResolveRequest(persistence.ConflictResolveWorkflowModeUpdateCurrent)\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"DB error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Unknown Conflict Resolution Mode\",\n\t\t\tsetupMocks: func() {\n\t\t\t},\n\t\t\tgetRequest: func() *persistence.InternalConflictResolveWorkflowExecutionRequest {\n\t\t\t\treq := newConflictResolveRequest(-1) // Intentionally using an invalid mode\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"ConflictResolveWorkflowExecution: unknown mode: -1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error on Shard Condition Mismatch\",\n\t\t\tsetupMocks: func() {\n\t\t\t\t// Simulate shard condition mismatch error by returning a ShardOperationConditionFailure error from the mock\n\t\t\t\tconditionFailure := &nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 124, // Example mismatched RangeID to trigger the condition failure\n\t\t\t\t}\n\t\t\t\tmockDB.EXPECT().UpdateWorkflowExecutionWithTasks(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(conditionFailure).Times(1)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false).AnyTimes()\n\t\t\t\tmockDB.EXPECT().IsDBUnavailableError(gomock.Any()).Return(false).AnyTimes() // Ensure this call returns the simulated condition failure once\n\t\t\t},\n\t\t\tgetRequest: func() *persistence.InternalConflictResolveWorkflowExecutionRequest {\n\t\t\t\treq := newConflictResolveRequest(persistence.ConflictResolveWorkflowModeUpdateCurrent)\n\t\t\t\treq.RangeID = 123 // Intentionally set to simulate the condition leading to a shard condition mismatch.\n\t\t\t\treturn req\n\t\t\t},\n\t\t\texpectedError: &types.InternalServiceError{Message: \"Shard error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\treq := tc.getRequest()\n\t\t\terr := store.ConflictResolveWorkflowExecution(ctx, req)\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.IsType(t, tc.expectedError, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc newCreateWorkflowExecutionRequest() *persistence.InternalCreateWorkflowExecutionRequest {\n\treturn &persistence.InternalCreateWorkflowExecutionRequest{\n\t\tRangeID:                  123,\n\t\tMode:                     persistence.CreateWorkflowModeBrandNew,\n\t\tPreviousRunID:            \"previous-run-id\",\n\t\tPreviousLastWriteVersion: 456,\n\t\tNewWorkflowSnapshot:      getNewWorkflowSnapshot(),\n\t}\n}\n\nfunc newGetWorkflowExecutionRequest() *persistence.InternalGetWorkflowExecutionRequest {\n\treturn &persistence.InternalGetWorkflowExecutionRequest{\n\t\tDomainID: constants.TestDomainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\tRunID:      constants.TestRunID,\n\t\t},\n\t}\n}\n\nfunc newUpdateWorkflowExecutionRequest() *persistence.InternalUpdateWorkflowExecutionRequest {\n\treturn &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\tRangeID: 123,\n\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\tState:       persistence.WorkflowStateCreated,\n\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   constants.TestRequestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc newUpdateWorkflowExecutionRequestForContinueAsNew() *persistence.InternalUpdateWorkflowExecutionRequest {\n\treturn &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\tRangeID: 123,\n\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\tState:       persistence.WorkflowStateCompleted,\n\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t},\n\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   constants.TestRequestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\tActiveClusterSelectionPolicy: &persistence.DataBlob{\n\t\t\t\t\tEncoding: \"json\",\n\t\t\t\t\tData:     []byte(`{\"policy\": \"abc\"}`),\n\t\t\t\t},\n\t\t\t},\n\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   constants.TestRequestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc getNewWorkflowSnapshot() persistence.InternalWorkflowSnapshot {\n\treturn persistence.InternalWorkflowSnapshot{\n\t\tVersionHistories: &persistence.DataBlob{},\n\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\tDomainID:   constants.TestDomainID,\n\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\tRunID:      constants.TestRunID,\n\t\t},\n\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t{\n\t\t\t\tRequestID:   constants.TestRequestID,\n\t\t\t\tVersion:     1,\n\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t},\n\t\t},\n\t}\n}\nfunc newTestNosqlExecutionStore(db nosqlplugin.DB, logger log.Logger) *nosqlExecutionStore {\n\treturn &nosqlExecutionStore{\n\t\tshardID:    1,\n\t\tnosqlStore: nosqlStore{logger: logger, db: db},\n\t}\n}\n\nfunc newInternalReplicationTaskInfo() persistence.InternalReplicationTaskInfo {\n\tvar fixedCreationTime = time.Date(2024, time.April, 3, 14, 35, 44, 0, time.UTC)\n\treturn persistence.InternalReplicationTaskInfo{\n\t\tDomainID:          \"testDomainID\",\n\t\tWorkflowID:        \"testWorkflowID\",\n\t\tRunID:             \"testRunID\",\n\t\tTaskID:            123,\n\t\tTaskType:          persistence.ReplicationTaskTypeHistory,\n\t\tFirstEventID:      1,\n\t\tNextEventID:       2,\n\t\tVersion:           1,\n\t\tScheduledID:       3,\n\t\tBranchToken:       []byte(\"branchToken\"),\n\t\tNewRunBranchToken: []byte(\"newRunBranchToken\"),\n\t\tCreationTime:      fixedCreationTime,\n\t}\n}\n\nfunc newConflictResolveRequest(mode persistence.ConflictResolveWorkflowMode) *persistence.InternalConflictResolveWorkflowExecutionRequest {\n\treturn &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\tMode:    mode,\n\t\tRangeID: 123,\n\t\tCurrentWorkflowMutation: &persistence.InternalWorkflowMutation{\n\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:    \"testDomainID\",\n\t\t\t\tWorkflowID:  \"testWorkflowID\",\n\t\t\t\tRunID:       \"currentRunID\",\n\t\t\t\tState:       persistence.WorkflowStateCompleted,\n\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusCompleted,\n\t\t\t},\n\t\t},\n\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:    \"testDomainID\",\n\t\t\t\tWorkflowID:  \"testWorkflowID\",\n\t\t\t\tRunID:       \"resetRunID\",\n\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t},\n\t\tNewWorkflowSnapshot: nil,\n\t}\n}\n\nfunc TestRangeCompleteHistoryTask(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *persistence.RangeCompleteHistoryTaskRequest\n\t\tsetupMock     func(*nosqlplugin.MockDB)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success - scheduled timer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteTimerTasks(ctx, shardID, time.Unix(0, 0), time.Unix(0, 0).Add(time.Minute)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - immediate transfer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteTransferTasks(ctx, shardID, int64(100), int64(200)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - immediate replication task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100), // this is ignored by replication task\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteReplicationTasks(ctx, shardID, int64(200)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category error\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB *nosqlplugin.MockDB) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"database error on timer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteTimerTasks(ctx, shardID, time.Unix(0, 0), time.Unix(0, 0).Add(time.Minute)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on transfer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteTransferTasks(ctx, shardID, int64(100), int64(200)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on replication task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteReplicationTasks(ctx, shardID, int64(200)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tstore := &nosqlExecutionStore{nosqlStore: nosqlStore{db: mockDB}, shardID: shardID}\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\t_, err := store.RangeCompleteHistoryTask(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetHistoryTasks(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname                             string\n\t\trequest                          *persistence.GetHistoryTasksRequest\n\t\treadNoSQLHistoryTaskFromDataBlob bool\n\t\tsetupMock                        func(*nosqlplugin.MockDB, *serialization.MockTaskSerializer)\n\t\texpectedError                    error\n\t\texpectedTasks                    []persistence.Task\n\t}{\n\t\t{\n\t\t\tname: \"success - get immediate transfer tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTransferTasksOrderByTaskID(ctx, shardID, 10, []byte(\"next-page-token\"), int64(100), int64(200)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTransfer: &persistence.TransferTaskInfo{\n\t\t\t\t\t\t\tDomainID:            \"testDomainID\",\n\t\t\t\t\t\t\tWorkflowID:          \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:               \"testRunID\",\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tTaskType:            persistence.TransferTaskTypeDecisionTask,\n\t\t\t\t\t\t\tScheduleID:          1,\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tTaskList:            \"testTaskList\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTask: &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tTaskList:   \"testTaskList\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - get immediate transfer tasks from data blob\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\treadNoSQLHistoryTaskFromDataBlob: true,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTransferTasksOrderByTaskID(ctx, shardID, 10, []byte(\"next-page-token\"), int64(100), int64(200)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTask:   &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(&persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tTaskList:   \"testTaskList\",\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tTaskList:   \"testTaskList\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - get scheduled timer tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTimerTasksOrderByVisibilityTime(ctx, shardID, 10, []byte(\"next-page-token\"), time.Unix(0, 0), time.Unix(0, 0).Add(time.Minute)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTimer: &persistence.TimerTaskInfo{\n\t\t\t\t\t\t\tDomainID:            \"testDomainID\",\n\t\t\t\t\t\t\tWorkflowID:          \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:               \"testRunID\",\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tTaskType:            persistence.TaskTypeUserTimer,\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tEventID:             12,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTask: &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tEventID: 12,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - get scheduled timer tasks from data blob\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\treadNoSQLHistoryTaskFromDataBlob: true,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTimerTasksOrderByVisibilityTime(ctx, shardID, 10, []byte(\"next-page-token\"), time.Unix(0, 0), time.Unix(0, 0).Add(time.Minute)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTask:          &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t\tTaskID:        1,\n\t\t\t\t\t\tScheduledTime: time.Unix(1, 1),\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(&persistence.UserTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t\tEventID: 12,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t},\n\t\t\t\t\tEventID: 12,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - get immediate replication tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectReplicationTasksOrderByTaskID(ctx, shardID, 10, []byte(\"next-page-token\"), int64(100), int64(200)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tReplication: &persistence.InternalReplicationTaskInfo{\n\t\t\t\t\t\t\tDomainID:          \"testDomainID\",\n\t\t\t\t\t\t\tWorkflowID:        \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:             \"testRunID\",\n\t\t\t\t\t\t\tTaskID:            1,\n\t\t\t\t\t\t\tTaskType:          persistence.ReplicationTaskTypeHistory,\n\t\t\t\t\t\t\tVersion:           1,\n\t\t\t\t\t\t\tScheduledID:       3,\n\t\t\t\t\t\t\tFirstEventID:      1,\n\t\t\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\t\t\tBranchToken:       []byte(\"branchToken\"),\n\t\t\t\t\t\t\tNewRunBranchToken: []byte(\"newRunBranchToken\"),\n\t\t\t\t\t\t\tCreationTime:      time.Unix(0, 0),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTask: &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tFirstEventID:      1,\n\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\tBranchToken:       []byte(\"branchToken\"),\n\t\t\t\t\tNewRunBranchToken: []byte(\"newRunBranchToken\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - get immediate replication tasks from data blob\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       []byte(\"next-page-token\"),\n\t\t\t},\n\t\t\treadNoSQLHistoryTaskFromDataBlob: true,\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectReplicationTasksOrderByTaskID(ctx, shardID, 10, []byte(\"next-page-token\"), int64(100), int64(200)).Return([]*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTask:   &persistence.DataBlob{Data: []byte(\"task\"), Encoding: \"json\"},\n\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t},\n\t\t\t\t}, nil, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(&persistence.HistoryReplicationTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tFirstEventID:      1,\n\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\tBranchToken:       []byte(\"branchToken\"),\n\t\t\t\t\tNewRunBranchToken: []byte(\"newRunBranchToken\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\t},\n\t\t\t\t\tFirstEventID:      1,\n\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\tBranchToken:       []byte(\"branchToken\"),\n\t\t\t\t\tNewRunBranchToken: []byte(\"newRunBranchToken\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"database error on transfer task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTransferTasksOrderByTaskID(ctx, shardID, 10, gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on replication task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectReplicationTasksOrderByTaskID(ctx, shardID, 10, gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on timer task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectTimerTasksOrderByVisibilityTime(ctx, shardID, 10, gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category error\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB *nosqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(controller)\n\t\t\tstore := &nosqlExecutionStore{nosqlStore: nosqlStore{db: mockDB, dc: &persistence.DynamicConfiguration{ReadNoSQLHistoryTaskFromDataBlob: func(...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn tc.readNoSQLHistoryTaskFromDataBlob\n\t\t\t}}}, shardID: shardID, taskSerializer: mockTaskSerializer}\n\n\t\t\ttc.setupMock(mockDB, mockTaskSerializer)\n\n\t\t\tresp, err := store.GetHistoryTasks(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedTasks, resp.Tasks)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCompleteHistoryTask(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *persistence.CompleteHistoryTaskRequest\n\t\tsetupMock     func(*nosqlplugin.MockDB)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success - complete scheduled timer task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(10, 10), 1),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteTimerTask(ctx, shardID, int64(1), time.Unix(10, 10)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - complete immediate transfer task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteTransferTask(ctx, shardID, int64(2)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - complete immediate replication task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteReplicationTask(ctx, shardID, int64(3)).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category type\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB *nosqlplugin.MockDB) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"delete timer task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(10, 10), 1),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteTimerTask(ctx, shardID, int64(1), time.Unix(10, 10)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"delete transfer task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteTransferTask(ctx, shardID, int64(2)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"delete replication task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *nosqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteReplicationTask(ctx, shardID, int64(3)).Return(errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(controller)\n\t\t\tstore := &nosqlExecutionStore{nosqlStore: nosqlStore{db: mockDB, dc: &persistence.DynamicConfiguration{}}, shardID: shardID}\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\terr := store.CompleteHistoryTask(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_execution_store_util.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (d *nosqlExecutionStore) prepareCreateWorkflowExecutionRequestWithMaps(newWorkflow *persistence.InternalWorkflowSnapshot, currentTimeStamp time.Time) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\texecutionInfo := newWorkflow.ExecutionInfo\n\tlastWriteVersion := newWorkflow.LastWriteVersion\n\tcheckSum := newWorkflow.Checksum\n\tversionHistories := newWorkflow.VersionHistories\n\n\texecutionRequest, err := d.prepareCreateWorkflowExecutionTxn(\n\t\texecutionInfo, versionHistories, checkSum,\n\t\tcurrentTimeStamp, lastWriteVersion,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\texecutionRequest.ActivityInfos, err = d.prepareActivityInfosForWorkflowTxn(newWorkflow.ActivityInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.TimerInfos, err = d.prepareTimerInfosForWorkflowTxn(newWorkflow.TimerInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.ChildWorkflowInfos, err = d.prepareChildWFInfosForWorkflowTxn(newWorkflow.ChildExecutionInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.RequestCancelInfos, err = d.prepareRequestCancelsForWorkflowTxn(newWorkflow.RequestCancelInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalInfos, err = d.prepareSignalInfosForWorkflowTxn(newWorkflow.SignalInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalRequestedIDs = newWorkflow.SignalRequestedIDs\n\texecutionRequest.MapsWriteMode = nosqlplugin.WorkflowExecutionMapsWriteModeCreate\n\texecutionRequest.CurrentTimeStamp = currentTimeStamp\n\treturn executionRequest, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareWorkflowRequestRows(\n\tdomainID, workflowID, runID string,\n\trequests []*persistence.WorkflowRequest,\n\trequestRowsToAppend []*nosqlplugin.WorkflowRequestRow,\n) []*nosqlplugin.WorkflowRequestRow {\n\tfor _, req := range requests {\n\t\trequestRowsToAppend = append(requestRowsToAppend, &nosqlplugin.WorkflowRequestRow{\n\t\t\tShardID:     d.shardID,\n\t\t\tDomainID:    domainID,\n\t\t\tWorkflowID:  workflowID,\n\t\t\tRequestType: req.RequestType,\n\t\t\tRequestID:   req.RequestID,\n\t\t\tVersion:     req.Version,\n\t\t\tRunID:       runID,\n\t\t})\n\t}\n\treturn requestRowsToAppend\n}\n\nfunc (d *nosqlExecutionStore) prepareActiveClusterSelectionPolicyRow(domainID, workflowID, runID string, activeClusterSelectionPolicy *persistence.DataBlob) *nosqlplugin.ActiveClusterSelectionPolicyRow {\n\tif activeClusterSelectionPolicy == nil {\n\t\treturn nil\n\t}\n\n\treturn &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\tShardID:    d.shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tPolicy:     activeClusterSelectionPolicy,\n\t}\n}\n\nfunc (d *nosqlExecutionStore) prepareResetWorkflowExecutionRequestWithMapsAndEventBuffer(resetWorkflow *persistence.InternalWorkflowSnapshot, currentTimeStamp time.Time) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\texecutionInfo := resetWorkflow.ExecutionInfo\n\tlastWriteVersion := resetWorkflow.LastWriteVersion\n\tcheckSum := resetWorkflow.Checksum\n\tversionHistories := resetWorkflow.VersionHistories\n\tnowTimestamp := currentTimeStamp\n\n\texecutionRequest, err := d.prepareUpdateWorkflowExecutionTxn(\n\t\texecutionInfo, versionHistories, checkSum,\n\t\tnowTimestamp, lastWriteVersion,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// reset 6 maps\n\texecutionRequest.ActivityInfos, err = d.prepareActivityInfosForWorkflowTxn(resetWorkflow.ActivityInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.TimerInfos, err = d.prepareTimerInfosForWorkflowTxn(resetWorkflow.TimerInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.ChildWorkflowInfos, err = d.prepareChildWFInfosForWorkflowTxn(resetWorkflow.ChildExecutionInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.RequestCancelInfos, err = d.prepareRequestCancelsForWorkflowTxn(resetWorkflow.RequestCancelInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalInfos, err = d.prepareSignalInfosForWorkflowTxn(resetWorkflow.SignalInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalRequestedIDs = resetWorkflow.SignalRequestedIDs\n\texecutionRequest.MapsWriteMode = nosqlplugin.WorkflowExecutionMapsWriteModeReset\n\t// delete buffered events\n\texecutionRequest.EventBufferWriteMode = nosqlplugin.EventBufferWriteModeClear\n\t// condition\n\texecutionRequest.PreviousNextEventIDCondition = &resetWorkflow.Condition\n\treturn executionRequest, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer(workflowMutation *persistence.InternalWorkflowMutation, currentTimeStamp time.Time) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\texecutionInfo := workflowMutation.ExecutionInfo\n\tlastWriteVersion := workflowMutation.LastWriteVersion\n\tcheckSum := workflowMutation.Checksum\n\tversionHistories := workflowMutation.VersionHistories\n\tnowTimestamp := currentTimeStamp\n\n\texecutionRequest, err := d.prepareUpdateWorkflowExecutionTxn(\n\t\texecutionInfo, versionHistories, checkSum,\n\t\tnowTimestamp, lastWriteVersion,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// merge 6 maps\n\texecutionRequest.ActivityInfos, err = d.prepareActivityInfosForWorkflowTxn(workflowMutation.UpsertActivityInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.TimerInfos, err = d.prepareTimerInfosForWorkflowTxn(workflowMutation.UpsertTimerInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.ChildWorkflowInfos, err = d.prepareChildWFInfosForWorkflowTxn(workflowMutation.UpsertChildExecutionInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.RequestCancelInfos, err = d.prepareRequestCancelsForWorkflowTxn(workflowMutation.UpsertRequestCancelInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalInfos, err = d.prepareSignalInfosForWorkflowTxn(workflowMutation.UpsertSignalInfos)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\texecutionRequest.SignalRequestedIDs = workflowMutation.UpsertSignalRequestedIDs\n\n\t// delete from 6 maps\n\texecutionRequest.ActivityInfoKeysToDelete = workflowMutation.DeleteActivityInfos\n\texecutionRequest.TimerInfoKeysToDelete = workflowMutation.DeleteTimerInfos\n\texecutionRequest.ChildWorkflowInfoKeysToDelete = workflowMutation.DeleteChildExecutionInfos\n\texecutionRequest.RequestCancelInfoKeysToDelete = workflowMutation.DeleteRequestCancelInfos\n\texecutionRequest.SignalInfoKeysToDelete = workflowMutation.DeleteSignalInfos\n\texecutionRequest.SignalRequestedIDsKeysToDelete = workflowMutation.DeleteSignalRequestedIDs\n\n\t// map write mode\n\texecutionRequest.MapsWriteMode = nosqlplugin.WorkflowExecutionMapsWriteModeUpdate\n\n\t// prepare to write buffer event\n\texecutionRequest.EventBufferWriteMode = nosqlplugin.EventBufferWriteModeNone\n\tif workflowMutation.ClearBufferedEvents {\n\t\texecutionRequest.EventBufferWriteMode = nosqlplugin.EventBufferWriteModeClear\n\t} else if workflowMutation.NewBufferedEvents != nil {\n\t\texecutionRequest.EventBufferWriteMode = nosqlplugin.EventBufferWriteModeAppend\n\t\texecutionRequest.NewBufferedEventBatch = workflowMutation.NewBufferedEvents\n\t}\n\n\t// condition\n\texecutionRequest.PreviousNextEventIDCondition = &workflowMutation.Condition\n\treturn executionRequest, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareTimerTasksForWorkflowTxn(domainID, workflowID, runID string, timerTasks []persistence.Task) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\tvar tasks []*nosqlplugin.HistoryMigrationTask\n\n\tfor _, task := range timerTasks {\n\t\tnt, err := task.ToTimerTaskInfo()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tvar blob *persistence.DataBlob\n\t\tif d.dc.EnableHistoryTaskDualWriteMode() {\n\t\t\tdata, err := d.taskSerializer.SerializeTask(persistence.HistoryTaskCategoryTimer, task)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tblob = &data\n\t\t}\n\t\ttasks = append(tasks, &nosqlplugin.HistoryMigrationTask{\n\t\t\tTimer: nt,\n\t\t\tTask:  blob,\n\t\t})\n\t}\n\n\treturn tasks, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareReplicationTasksForWorkflowTxn(domainID, workflowID, runID string, replicationTasks []persistence.Task) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\tvar tasks []*nosqlplugin.HistoryMigrationTask\n\n\tfor _, task := range replicationTasks {\n\t\t// Replication task specific information\n\t\tfirstEventID := constants.EmptyEventID\n\t\tnextEventID := constants.EmptyEventID\n\t\tversion := constants.EmptyVersion //nolint:ineffassign\n\t\tactivityScheduleID := constants.EmptyEventID\n\t\tvar branchToken, newRunBranchToken []byte\n\n\t\tswitch task.GetTaskType() {\n\t\tcase persistence.ReplicationTaskTypeHistory:\n\t\t\thistTask := task.(*persistence.HistoryReplicationTask)\n\t\t\tbranchToken = histTask.BranchToken\n\t\t\tnewRunBranchToken = histTask.NewRunBranchToken\n\t\t\tfirstEventID = histTask.FirstEventID\n\t\t\tnextEventID = histTask.NextEventID\n\t\t\tversion = task.GetVersion()\n\n\t\tcase persistence.ReplicationTaskTypeSyncActivity:\n\t\t\tversion = task.GetVersion()\n\t\t\tactivityScheduleID = task.(*persistence.SyncActivityTask).ScheduledID\n\n\t\tcase persistence.ReplicationTaskTypeFailoverMarker:\n\t\t\tversion = task.GetVersion()\n\n\t\tdefault:\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Unknown replication type: %v\", task.GetTaskType()),\n\t\t\t}\n\t\t}\n\n\t\tnt := &nosqlplugin.ReplicationTask{\n\t\t\tTaskType:          task.GetTaskType(),\n\t\t\tDomainID:          domainID,\n\t\t\tWorkflowID:        workflowID,\n\t\t\tRunID:             runID,\n\t\t\tCreationTime:      task.GetVisibilityTimestamp(),\n\t\t\tTaskID:            task.GetTaskID(),\n\t\t\tFirstEventID:      firstEventID,\n\t\t\tNextEventID:       nextEventID,\n\t\t\tVersion:           version,\n\t\t\tScheduledID:       activityScheduleID,\n\t\t\tBranchToken:       branchToken,\n\t\t\tNewRunBranchToken: newRunBranchToken,\n\t\t}\n\t\tvar blob *persistence.DataBlob\n\t\tif d.dc.EnableHistoryTaskDualWriteMode() {\n\t\t\tdata, err := d.taskSerializer.SerializeTask(persistence.HistoryTaskCategoryReplication, task)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tblob = &data\n\t\t}\n\t\ttasks = append(tasks, &nosqlplugin.HistoryMigrationTask{\n\t\t\tReplication: nt,\n\t\t\tTask:        blob,\n\t\t})\n\t}\n\n\treturn tasks, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareNoSQLTasksForWorkflowTxn(\n\tdomainID, workflowID, runID string,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]persistence.Task,\n\toutputTasks map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n) error {\n\tfor c, tasks := range tasksByCategory {\n\t\tswitch c.ID() {\n\t\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\t\ttransferTasks, err := d.prepareTransferTasksForWorkflowTxn(domainID, workflowID, runID, tasks)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\toutputTasks[c] = append(outputTasks[c], transferTasks...)\n\t\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\t\ttimerTasks, err := d.prepareTimerTasksForWorkflowTxn(domainID, workflowID, runID, tasks)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\toutputTasks[c] = append(outputTasks[c], timerTasks...)\n\t\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\t\treplicationTasks, err := d.prepareReplicationTasksForWorkflowTxn(domainID, workflowID, runID, tasks)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\toutputTasks[c] = append(outputTasks[c], replicationTasks...)\n\t\t}\n\t}\n\t// TODO: implementing logic for other categories\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) prepareTransferTasksForWorkflowTxn(domainID, workflowID, runID string, transferTasks []persistence.Task) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\tvar tasks []*nosqlplugin.HistoryMigrationTask\n\n\tfor _, task := range transferTasks {\n\t\tt, err := task.ToTransferTaskInfo()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tvar blob *persistence.DataBlob\n\t\tif d.dc.EnableHistoryTaskDualWriteMode() {\n\t\t\tdata, err := d.taskSerializer.SerializeTask(persistence.HistoryTaskCategoryTransfer, task)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tblob = &data\n\t\t}\n\t\ttasks = append(tasks, &nosqlplugin.HistoryMigrationTask{\n\t\t\tTransfer: t,\n\t\t\tTask:     blob,\n\t\t})\n\t}\n\treturn tasks, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareActivityInfosForWorkflowTxn(activityInfos []*persistence.InternalActivityInfo) (map[int64]*persistence.InternalActivityInfo, error) {\n\tm := map[int64]*persistence.InternalActivityInfo{}\n\tfor _, a := range activityInfos {\n\t\t_, scheduleEncoding := persistence.FromDataBlob(a.ScheduledEvent)\n\t\t_, startEncoding := persistence.FromDataBlob(a.StartedEvent)\n\t\tif a.StartedEvent != nil && scheduleEncoding != startEncoding {\n\t\t\treturn nil, persistence.NewCadenceSerializationError(fmt.Sprintf(\"expect to have the same encoding, but %v != %v\", scheduleEncoding, startEncoding))\n\t\t}\n\t\ta.ScheduledEvent = a.ScheduledEvent.ToNilSafeDataBlob()\n\t\ta.StartedEvent = a.StartedEvent.ToNilSafeDataBlob()\n\t\tm[a.ScheduleID] = a\n\t}\n\treturn m, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareTimerInfosForWorkflowTxn(timerInfo []*persistence.TimerInfo) (map[string]*persistence.TimerInfo, error) {\n\tm := map[string]*persistence.TimerInfo{}\n\tfor _, a := range timerInfo {\n\t\tm[a.TimerID] = a\n\t}\n\treturn m, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareChildWFInfosForWorkflowTxn(childWFInfos []*persistence.InternalChildExecutionInfo) (map[int64]*persistence.InternalChildExecutionInfo, error) {\n\tm := map[int64]*persistence.InternalChildExecutionInfo{}\n\tfor _, c := range childWFInfos {\n\t\t_, initiatedEncoding := persistence.FromDataBlob(c.InitiatedEvent)\n\t\t_, startEncoding := persistence.FromDataBlob(c.StartedEvent)\n\t\tif c.StartedEvent != nil && initiatedEncoding != startEncoding {\n\t\t\treturn nil, persistence.NewCadenceSerializationError(fmt.Sprintf(\"expect to have the same encoding, but %v != %v\", initiatedEncoding, startEncoding))\n\t\t}\n\n\t\tif c.StartedRunID == \"\" {\n\t\t\tc.StartedRunID = emptyRunID\n\t\t}\n\n\t\tc.InitiatedEvent = c.InitiatedEvent.ToNilSafeDataBlob()\n\t\tc.StartedEvent = c.StartedEvent.ToNilSafeDataBlob()\n\t\tm[c.InitiatedID] = c\n\t}\n\treturn m, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareRequestCancelsForWorkflowTxn(requestCancels []*persistence.RequestCancelInfo) (map[int64]*persistence.RequestCancelInfo, error) {\n\tm := map[int64]*persistence.RequestCancelInfo{}\n\tfor _, c := range requestCancels {\n\t\tm[c.InitiatedID] = c\n\t}\n\treturn m, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareSignalInfosForWorkflowTxn(signalInfos []*persistence.SignalInfo) (map[int64]*persistence.SignalInfo, error) {\n\tm := map[int64]*persistence.SignalInfo{}\n\tfor _, c := range signalInfos {\n\t\tm[c.InitiatedID] = c\n\t}\n\treturn m, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareUpdateWorkflowExecutionTxn(\n\texecutionInfo *persistence.InternalWorkflowExecutionInfo,\n\tversionHistories *persistence.DataBlob,\n\tchecksum checksum.Checksum,\n\tnowTimestamp time.Time,\n\tlastWriteVersion int64,\n) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t// validate workflow state & close status\n\tif err := persistence.ValidateUpdateWorkflowStateCloseStatus(\n\t\texecutionInfo.State,\n\t\texecutionInfo.CloseStatus); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif executionInfo.ParentDomainID == \"\" {\n\t\texecutionInfo.ParentDomainID = emptyDomainID\n\t\texecutionInfo.ParentWorkflowID = \"\"\n\t\texecutionInfo.ParentRunID = emptyRunID\n\t\texecutionInfo.InitiatedID = emptyInitiatedID\n\t}\n\n\t// TODO: remove this logic once all workflows before 0.26.x are completed\n\tif executionInfo.FirstExecutionRunID == \"\" {\n\t\texecutionInfo.FirstExecutionRunID = emptyRunID\n\t}\n\n\texecutionInfo.CompletionEvent = executionInfo.CompletionEvent.ToNilSafeDataBlob()\n\texecutionInfo.AutoResetPoints = executionInfo.AutoResetPoints.ToNilSafeDataBlob()\n\t// TODO also need to set the start / current / last write version\n\tversionHistories = versionHistories.ToNilSafeDataBlob()\n\treturn &nosqlplugin.WorkflowExecutionRequest{\n\t\tInternalWorkflowExecutionInfo: *executionInfo,\n\t\tVersionHistories:              versionHistories,\n\t\tChecksums:                     &checksum,\n\t\tLastWriteVersion:              lastWriteVersion,\n\t\tCurrentTimeStamp:              nowTimestamp,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareCreateWorkflowExecutionTxn(\n\texecutionInfo *persistence.InternalWorkflowExecutionInfo,\n\tversionHistories *persistence.DataBlob,\n\tchecksum checksum.Checksum,\n\tnowTimestamp time.Time,\n\tlastWriteVersion int64,\n) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t// validate workflow state & close status\n\tif err := persistence.ValidateCreateWorkflowStateCloseStatus(\n\t\texecutionInfo.State,\n\t\texecutionInfo.CloseStatus); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif executionInfo.ParentDomainID == \"\" {\n\t\texecutionInfo.ParentDomainID = emptyDomainID\n\t\texecutionInfo.ParentWorkflowID = \"\"\n\t\texecutionInfo.ParentRunID = emptyRunID\n\t\texecutionInfo.InitiatedID = emptyInitiatedID\n\t}\n\n\t// TODO: remove this logic once all workflows before 0.26.x are completed\n\tif executionInfo.FirstExecutionRunID == \"\" {\n\t\texecutionInfo.FirstExecutionRunID = emptyRunID\n\t}\n\n\tif executionInfo.StartTimestamp.IsZero() {\n\t\texecutionInfo.StartTimestamp = nowTimestamp\n\t\td.logger.Error(\"Workflow startTimestamp not set, fallback to now\",\n\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\t)\n\t}\n\texecutionInfo.CompletionEvent = executionInfo.CompletionEvent.ToNilSafeDataBlob()\n\texecutionInfo.AutoResetPoints = executionInfo.AutoResetPoints.ToNilSafeDataBlob()\n\tif versionHistories == nil {\n\t\treturn nil, &types.InternalServiceError{Message: \"encounter empty version histories in createExecution\"}\n\t}\n\tversionHistories = versionHistories.ToNilSafeDataBlob()\n\treturn &nosqlplugin.WorkflowExecutionRequest{\n\t\tInternalWorkflowExecutionInfo: *executionInfo,\n\t\tVersionHistories:              versionHistories,\n\t\tChecksums:                     &checksum,\n\t\tLastWriteVersion:              lastWriteVersion,\n\t}, nil\n}\n\nfunc (d *nosqlExecutionStore) prepareCurrentWorkflowRequestForCreateWorkflowTxn(\n\tdomainID, workflowID, runID string,\n\texecutionInfo *persistence.InternalWorkflowExecutionInfo,\n\tlastWriteVersion int64,\n\trequest *persistence.InternalCreateWorkflowExecutionRequest,\n) (*nosqlplugin.CurrentWorkflowWriteRequest, error) {\n\tcurrentWorkflowWriteReq := &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\tShardID:          d.shardID,\n\t\t\tDomainID:         domainID,\n\t\t\tWorkflowID:       workflowID,\n\t\t\tRunID:            runID,\n\t\t\tState:            executionInfo.State,\n\t\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\t\tCreateRequestID:  executionInfo.CreateRequestID,\n\t\t\tLastWriteVersion: lastWriteVersion,\n\t\t},\n\t}\n\tswitch request.Mode {\n\tcase persistence.CreateWorkflowModeZombie:\n\t\t// noop\n\t\tcurrentWorkflowWriteReq.WriteMode = nosqlplugin.CurrentWorkflowWriteModeNoop\n\tcase persistence.CreateWorkflowModeContinueAsNew:\n\t\tcurrentWorkflowWriteReq.WriteMode = nosqlplugin.CurrentWorkflowWriteModeUpdate\n\t\tcurrentWorkflowWriteReq.Condition = &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\tCurrentRunID: common.StringPtr(request.PreviousRunID),\n\t\t}\n\tcase persistence.CreateWorkflowModeWorkflowIDReuse:\n\t\tcurrentWorkflowWriteReq.WriteMode = nosqlplugin.CurrentWorkflowWriteModeUpdate\n\t\tcurrentWorkflowWriteReq.Condition = &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\tCurrentRunID:     common.StringPtr(request.PreviousRunID),\n\t\t\tState:            common.IntPtr(persistence.WorkflowStateCompleted),\n\t\t\tLastWriteVersion: common.Int64Ptr(request.PreviousLastWriteVersion),\n\t\t}\n\tcase persistence.CreateWorkflowModeBrandNew:\n\t\tcurrentWorkflowWriteReq.WriteMode = nosqlplugin.CurrentWorkflowWriteModeInsert\n\tdefault:\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown mode: %v\", request.Mode),\n\t\t}\n\t}\n\treturn currentWorkflowWriteReq, nil\n}\n\nfunc (d *nosqlExecutionStore) processUpdateWorkflowResult(err error, rangeID int64) error {\n\tif err != nil {\n\t\tconditionFailureErr, isConditionFailedError := err.(*nosqlplugin.WorkflowOperationConditionFailure)\n\t\tif isConditionFailedError {\n\t\t\tswitch {\n\t\t\tcase conditionFailureErr.UnknownConditionFailureDetails != nil:\n\t\t\t\treturn &persistence.ConditionFailedError{\n\t\t\t\t\tMsg: *conditionFailureErr.UnknownConditionFailureDetails,\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.ShardRangeIDNotMatch != nil:\n\t\t\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\t\t\tShardID: d.shardID,\n\t\t\t\t\tMsg: fmt.Sprintf(\"Failed to update workflow execution.  Request RangeID: %v, Actual RangeID: %v\",\n\t\t\t\t\t\trangeID, *conditionFailureErr.ShardRangeIDNotMatch),\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.CurrentWorkflowConditionFailInfo != nil:\n\t\t\t\treturn &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: *conditionFailureErr.CurrentWorkflowConditionFailInfo,\n\t\t\t\t}\n\t\t\tcase conditionFailureErr.DuplicateRequest != nil:\n\t\t\t\treturn &persistence.DuplicateRequestError{\n\t\t\t\t\tRequestType: conditionFailureErr.DuplicateRequest.RequestType,\n\t\t\t\t\tRunID:       conditionFailureErr.DuplicateRequest.RunID,\n\t\t\t\t}\n\t\t\tdefault:\n\t\t\t\t// If ever runs into this branch, there is bug in the code either in here, or in the implementation of nosql plugin\n\t\t\t\terr := fmt.Errorf(\"unexpected conditionFailureReason error\")\n\t\t\t\td.logger.Error(\"A code bug exists in persistence layer, please investigate ASAP\", tag.Error(err))\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(d.db, \"UpdateWorkflowExecution\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (d *nosqlExecutionStore) assertNotCurrentExecution(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) error {\n\n\tif resp, err := d.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t}); err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t// allow bypassing no current record\n\t\t\treturn nil\n\t\t}\n\t\treturn err\n\t} else if resp.RunID == runID {\n\t\treturn &persistence.ConditionFailedError{\n\t\t\tMsg: fmt.Sprintf(\"Assertion on current record failed. Current run ID is not expected: %v\", resp.RunID),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getWorkflowRequestWriteMode(mode persistence.CreateWorkflowRequestMode) (nosqlplugin.WorkflowRequestWriteMode, error) {\n\tswitch mode {\n\tcase persistence.CreateWorkflowRequestModeNew:\n\t\treturn nosqlplugin.WorkflowRequestWriteModeInsert, nil\n\tcase persistence.CreateWorkflowRequestModeReplicated:\n\t\treturn nosqlplugin.WorkflowRequestWriteModeUpsert, nil\n\tdefault:\n\t\treturn nosqlplugin.WorkflowRequestWriteMode(-1), fmt.Errorf(\"unknown create workflow request mode: %v\", mode)\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_execution_store_util_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar FixedTime = time.Date(2025, 1, 6, 15, 0, 0, 0, time.UTC)\n\nfunc newTestNosqlExecutionStoreWithTaskSerializer(db nosqlplugin.DB, logger log.Logger, taskSerializer serialization.TaskSerializer) *nosqlExecutionStore {\n\treturn &nosqlExecutionStore{\n\t\tshardID:        1,\n\t\tnosqlStore:     nosqlStore{logger: logger, db: db, dc: &persistence.DynamicConfiguration{EnableHistoryTaskDualWriteMode: func(...dynamicproperties.FilterOption) bool { return true }}},\n\t\ttaskSerializer: taskSerializer,\n\t}\n}\n\nfunc TestNosqlExecutionStoreUtils(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupStore func(*nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error)\n\t\tinput      *persistence.InternalWorkflowSnapshot\n\t\tvalidate   func(*testing.T, *nosqlplugin.WorkflowExecutionRequest, error)\n\t}{\n\t\t{\n\t\t\tname: \"PrepareCreateWorkflowExecutionRequestWithMaps - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tworkflowSnapshot := &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"test-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareCreateWorkflowExecutionRequestWithMaps(workflowSnapshot, FixedTime)\n\t\t\t},\n\t\t\tinput: &persistence.InternalWorkflowSnapshot{},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tif err == nil {\n\t\t\t\t\tassert.NotNil(t, req)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareCreateWorkflowExecutionRequestWithMaps - Nil Checksum\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tworkflowSnapshot := &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"test-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t\t},\n\t\t\t\t\tChecksum: checksum.Checksum{Value: nil},\n\t\t\t\t}\n\t\t\t\treturn store.prepareCreateWorkflowExecutionRequestWithMaps(workflowSnapshot, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req.Checksums)\n\t\t\t},\n\t\t},\n\n\t\t{\n\t\t\tname: \"PrepareCreateWorkflowExecutionRequestWithMaps - Empty VersionHistories\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\t// Testing with an empty VersionHistories (which previously caused an error)\n\t\t\t\tworkflowSnapshot := &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id-2\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id-2\",\n\t\t\t\t\t\tRunID:      \"test-run-id-2\",\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\t\tData:     []byte(\"[]\"), // Empty VersionHistories\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareCreateWorkflowExecutionRequestWithMaps(workflowSnapshot, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req.VersionHistories)\n\t\t\t\tassert.Equal(t, \"[]\", string(req.VersionHistories.Data))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareResetWorkflowExecutionRequestWithMapsAndEventBuffer - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tresetWorkflow := &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"reset-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"reset-workflow-id\",\n\t\t\t\t\t\tRunID:      \"reset-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tLastWriteVersion: 123,\n\t\t\t\t\tChecksum:         checksum.Checksum{Version: 1},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{Encoding: constants.EncodingTypeJSON, Data: []byte(`[{\"Branches\":[{\"BranchID\":\"reset-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`)},\n\t\t\t\t\tActivityInfos:    []*persistence.InternalActivityInfo{{ScheduleID: 1}},\n\t\t\t\t\tTimerInfos:       []*persistence.TimerInfo{{TimerID: \"timerID\"}},\n\t\t\t\t\tChildExecutionInfos: []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t\t{InitiatedID: 1, StartedID: 2},\n\t\t\t\t\t},\n\t\t\t\t\tRequestCancelInfos: []*persistence.RequestCancelInfo{{InitiatedID: 1}},\n\t\t\t\t\tSignalInfos:        []*persistence.SignalInfo{{InitiatedID: 1}},\n\t\t\t\t\tSignalRequestedIDs: []string{\"signalRequestedID\"},\n\t\t\t\t\tCondition:          999,\n\t\t\t\t}\n\t\t\t\treturn store.prepareResetWorkflowExecutionRequestWithMapsAndEventBuffer(resetWorkflow, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req)\n\t\t\t\tassert.Equal(t, nosqlplugin.WorkflowExecutionMapsWriteModeReset, req.MapsWriteMode)\n\t\t\t\tassert.Equal(t, nosqlplugin.EventBufferWriteModeClear, req.EventBufferWriteMode)\n\t\t\t\tassert.Equal(t, int64(999), *req.PreviousNextEventIDCondition)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareResetWorkflowExecutionRequestWithMapsAndEventBuffer - Malformed VersionHistories\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tresetWorkflow := &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"domain-id-malformed-vh\",\n\t\t\t\t\t\tWorkflowID: \"workflow-id-malformed-vh\",\n\t\t\t\t\t\tRunID:      \"run-id-malformed-vh\",\n\t\t\t\t\t},\n\t\t\t\t\tLastWriteVersion: 456,\n\t\t\t\t\tChecksum:         checksum.Checksum{Version: 1},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{Encoding: constants.EncodingTypeJSON, Data: []byte(\"{malformed}\")},\n\t\t\t\t}\n\t\t\t\treturn store.prepareResetWorkflowExecutionRequestWithMapsAndEventBuffer(resetWorkflow, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer - Successful Update Request Preparation\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tworkflowMutation := &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"domainID-success\",\n\t\t\t\t\t\tWorkflowID: \"workflowID-success\",\n\t\t\t\t\t\tRunID:      \"runID-success\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer(workflowMutation, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer - Incomplete WorkflowMutation\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (*nosqlplugin.WorkflowExecutionRequest, error) {\n\t\t\t\tworkflowMutation := &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{ // Partially populated for the test\n\t\t\t\t\t\tDomainID: \"domainID-incomplete\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareUpdateWorkflowExecutionRequestWithMapsAndEventBuffer(workflowMutation, FixedTime)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, req *nosqlplugin.WorkflowExecutionRequest, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, req)\n\t\t\t\tassert.Equal(t, \"domainID-incomplete\", req.DomainID) // Example assertion\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(mockCtrl)\n\t\t\tstore := newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\n\t\t\treq, err := tc.setupStore(store)\n\t\t\ttc.validate(t, req, err)\n\t\t})\n\t}\n\n}\n\nfunc TestPrepareTasksForWorkflowTxn(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func(*serialization.MockTaskSerializer)\n\t\tsetupStore func(*nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error)\n\t\tvalidate   func(*testing.T, []*nosqlplugin.HistoryMigrationTask, error)\n\t}{{\n\t\tname: \"PrepareTimerTasksForWorkflowTxn - Successful Timer Tasks Preparation\",\n\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).\n\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"timer\"),\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t}, nil)\n\t\t},\n\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t&persistence.DecisionTimeoutTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(), TaskID: 1,\n\t\t\t\t\t},\n\t\t\t\t\tEventID: 2, TimeoutType: 1, ScheduleAttempt: 1},\n\t\t\t}\n\t\t\ttasks, err := store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", timerTasks)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotEmpty(t, tasks)\n\t\t\treturn nil, err\n\t\t},\n\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {},\n\t},\n\t\t{\n\t\t\tname:       \"PrepareTimerTasksForWorkflowTxn - Unsupported Timer Task Type\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&dummyTaskType{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID-unsupported\", \"workflowID-unsupported\", \"runID-unsupported\", timerTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, tasks)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"PrepareTimerTasksForWorkflowTxn - Zero Tasks\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", []persistence.Task{})\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Empty(t, tasks)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerTasksForWorkflowTxn - ActivityTimeoutTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tEventID: 3,\n\t\t\t\t\t\tAttempt: 2,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", timerTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\tassert.Equal(t, int64(3), tasks[0].Timer.EventID)\n\t\t\t\tassert.Equal(t, int64(2), tasks[0].Timer.ScheduleAttempt)\n\t\t\t\tassert.Equal(t, []byte(\"timer\"), tasks[0].Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, tasks[0].Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerTasksForWorkflowTxn - UserTimerTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tTaskID:              3,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tEventID: 4,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", timerTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\tassert.Equal(t, int64(4), tasks[0].Timer.EventID)\n\t\t\t\tassert.Equal(t, []byte(\"timer\"), tasks[0].Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, tasks[0].Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerTasksForWorkflowTxn - ActivityRetryTimerTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&persistence.ActivityRetryTimerTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tTaskID:              4,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tEventID: 5,\n\t\t\t\t\t\tAttempt: 3,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", timerTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\tassert.Equal(t, int64(5), tasks[0].Timer.EventID)\n\t\t\t\tassert.Equal(t, int64(3), tasks[0].Timer.ScheduleAttempt)\n\t\t\t\tassert.Equal(t, []byte(\"timer\"), tasks[0].Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, tasks[0].Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerTasksForWorkflowTxn - WorkflowBackoffTimerTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&persistence.WorkflowBackoffTimerTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\t\tTaskID:              5,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", timerTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\tassert.Equal(t, []byte(\"timer\"), tasks[0].Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, tasks[0].Task.Encoding)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(mockCtrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(mockCtrl)\n\t\t\tstore := newTestNosqlExecutionStoreWithTaskSerializer(mockDB, log.NewNoop(), mockTaskSerializer)\n\t\t\ttc.setupMocks(mockTaskSerializer)\n\n\t\t\ttasks, err := tc.setupStore(store)\n\t\t\ttc.validate(t, tasks, err)\n\t\t})\n\t}\n}\n\nfunc TestPrepareReplicationTasksForWorkflowTxn(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func(*serialization.MockTaskSerializer)\n\t\tsetupStore func(*nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error)\n\t\tvalidate   func(*testing.T, []*nosqlplugin.HistoryMigrationTask, error)\n\t}{\n\t\t{\n\t\t\tname: \"Successful Replication Tasks Preparation\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"replication\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\treplicationTasks := []persistence.Task{\n\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareReplicationTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", replicationTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotEmpty(t, tasks)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"Handling Unknown Replication Task Type\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\treplicationTasks := []persistence.Task{\n\t\t\t\t\t&dummyTaskType{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              -1,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareReplicationTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", replicationTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, tasks)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareReplicationTasksForWorkflowTxn - SyncActivityTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"replication\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\treplicationTasks := []persistence.Task{\n\t\t\t\t\t&persistence.SyncActivityTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             2,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tScheduledID: 123,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareReplicationTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", replicationTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, persistence.ReplicationTaskTypeSyncActivity, task.Replication.TaskType)\n\t\t\t\tassert.Equal(t, int64(123), task.Replication.ScheduledID)\n\t\t\t\tassert.Equal(t, []byte(\"replication\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareReplicationTasksForWorkflowTxn - FailoverMarkerTask\",\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"replication\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tsetupStore: func(store *nosqlExecutionStore) ([]*nosqlplugin.HistoryMigrationTask, error) {\n\t\t\t\treplicationTasks := []persistence.Task{\n\t\t\t\t\t&persistence.FailoverMarkerTask{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion:             3,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t\tTaskID:              3,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"domainID\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareReplicationTasksForWorkflowTxn(\"domainID\", \"workflowID\", \"runID\", replicationTasks)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, persistence.ReplicationTaskTypeFailoverMarker, task.Replication.TaskType)\n\t\t\t\tassert.Equal(t, \"domainID\", task.Replication.DomainID)\n\t\t\t\tassert.Equal(t, []byte(\"replication\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(mockCtrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(mockCtrl)\n\t\t\tstore := newTestNosqlExecutionStoreWithTaskSerializer(mockDB, log.NewNoop(), mockTaskSerializer)\n\t\t\ttc.setupMocks(mockTaskSerializer)\n\t\t\ttasks, err := tc.setupStore(store)\n\t\t\ttc.validate(t, tasks, err)\n\t\t})\n\t}\n}\n\nfunc TestPrepareTransferTasksForWorkflowTxn(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\ttasks      []persistence.Task\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\t\tsetupMocks func(*serialization.MockTaskSerializer)\n\t\tvalidate   func(*testing.T, []*nosqlplugin.HistoryMigrationTask, error)\n\t}{\n\t\t{\n\t\t\tname:       \"CancelExecutionTask - Success\",\n\t\t\tdomainID:   \"domainID-cancel\",\n\t\t\tworkflowID: \"workflowID-cancel\",\n\t\t\trunID:      \"runID-cancel\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.CancelExecutionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1002,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:          \"targetDomainID-cancel\",\n\t\t\t\t\tTargetWorkflowID:        \"targetWorkflowID-cancel\",\n\t\t\t\t\tTargetRunID:             \"targetRunID-cancel\",\n\t\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\t\tInitiatedID:             1002,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, \"targetDomainID-cancel\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, true, task.Transfer.TargetChildWorkflowOnly)\n\t\t\t\tassert.Equal(t, int64(1002), task.Transfer.TaskID)\n\t\t\t\tassert.Equal(t, int64(1), task.Transfer.Version)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"ActivityTask - Success\",\n\t\t\tdomainID:   \"domainID-activity\",\n\t\t\tworkflowID: \"workflowID-activity\",\n\t\t\trunID:      \"runID-activity\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1001,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"targetDomainID-activity\",\n\t\t\t\t\tTaskList:       \"taskList-activity\",\n\t\t\t\t\tScheduleID:     1001,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, persistence.TransferTaskTypeActivityTask, task.Transfer.TaskType)\n\t\t\t\tassert.Equal(t, \"targetDomainID-activity\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, \"taskList-activity\", task.Transfer.TaskList)\n\t\t\t\tassert.Equal(t, int64(1001), task.Transfer.ScheduleID)\n\t\t\t\tassert.Equal(t, int64(1), task.Transfer.Version)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"DefaultTargetRunID - When Empty\",\n\t\t\tdomainID:   \"domainID-default-runid\",\n\t\t\tworkflowID: \"workflowID-default-runid\",\n\t\t\trunID:      \"runID-default-runid\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.CancelExecutionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              2001,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:          \"targetDomainID-cancel\",\n\t\t\t\t\tTargetWorkflowID:        \"targetWorkflowID-cancel\",\n\t\t\t\t\tTargetRunID:             \"\", // Intentionally left empty to trigger the defaulting logic\n\t\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\t\tInitiatedID:             2001,\n\t\t\t\t},\n\t\t\t\t&persistence.SignalExecutionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              2002,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:          \"targetDomainID-signal\",\n\t\t\t\t\tTargetWorkflowID:        \"targetWorkflowID-signal\",\n\t\t\t\t\tTargetRunID:             \"\", // Intentionally left empty to trigger the defaulting logic\n\t\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\t\tInitiatedID:             2002,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil).Times(2)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tfor _, task := range tasks {\n\t\t\t\t\tassert.Equal(t, persistence.TransferTaskTransferTargetRunID, task.Transfer.TargetRunID, \"TargetRunID should default to TransferTaskTransferTargetRunID\")\n\t\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"SignalExecutionTask - Success\",\n\t\t\tdomainID:   \"domainID-signal\",\n\t\t\tworkflowID: \"workflowID-signal\",\n\t\t\trunID:      \"runID-signal\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.SignalExecutionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1003,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:          \"targetDomainID-signal\",\n\t\t\t\t\tTargetWorkflowID:        \"targetWorkflowID-signal\",\n\t\t\t\t\tTargetRunID:             \"targetRunID-signal\",\n\t\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\t\tInitiatedID:             1003,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, \"targetDomainID-signal\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, true, task.Transfer.TargetChildWorkflowOnly)\n\t\t\t\tassert.Equal(t, int64(1003), task.Transfer.TaskID)\n\t\t\t\tassert.Equal(t, int64(1), task.Transfer.Version)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"StartChildExecutionTask - Success\",\n\t\t\tdomainID:   \"domainID-start-child\",\n\t\t\tworkflowID: \"workflowID-start-child\",\n\t\t\trunID:      \"runID-start-child\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.StartChildExecutionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1004,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:   \"child-execution-domain-id\",\n\t\t\t\t\tTargetWorkflowID: \"child-workflow-id\",\n\t\t\t\t\tInitiatedID:      1004,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, \"child-execution-domain-id\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, \"child-workflow-id\", task.Transfer.TargetWorkflowID)\n\t\t\t\tassert.Equal(t, int64(1004), task.Transfer.TaskID)\n\t\t\t\tassert.Equal(t, int64(1), task.Transfer.Version)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"RecordChildExecutionCompletedTask - Success\",\n\t\t\tdomainID:   \"domainID-record-child\",\n\t\t\tworkflowID: \"workflowID-record-child\",\n\t\t\trunID:      \"runID-record-child\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.RecordChildExecutionCompletedTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1005,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:   \"completed-child-domain-id\",\n\t\t\t\t\tTargetWorkflowID: \"completed-child-workflow-id\",\n\t\t\t\t\tTargetRunID:      \"completed-child-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, \"completed-child-domain-id\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, \"completed-child-workflow-id\", task.Transfer.TargetWorkflowID)\n\t\t\t\tassert.Equal(t, \"completed-child-run-id\", task.Transfer.TargetRunID)\n\t\t\t\tassert.Equal(t, int64(1005), task.Transfer.TaskID)\n\t\t\t\tassert.Equal(t, int64(1), task.Transfer.Version)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"DecisionTask - Success\",\n\t\t\tdomainID:   \"domainID-decision\",\n\t\t\tworkflowID: \"workflowID-decision\",\n\t\t\trunID:      \"runID-decision\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\tTaskID:              1001,\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"targetDomainID-decision\",\n\t\t\t\t\tTaskList:       \"taskList-decision\",\n\t\t\t\t\tScheduleID:     1001,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).\n\t\t\t\t\tReturn(persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"transfer\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\ttask := tasks[0]\n\t\t\t\tassert.Equal(t, int64(1001), task.Transfer.TaskID)\n\t\t\t\tassert.Equal(t, \"targetDomainID-decision\", task.Transfer.TargetDomainID)\n\t\t\t\tassert.Equal(t, false, task.Transfer.RecordVisibility)\n\t\t\t\tassert.Equal(t, []byte(\"transfer\"), task.Task.Data)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, task.Task.Encoding)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"Unsupported Task Type\",\n\t\t\tdomainID:   \"domainID-unsupported\",\n\t\t\tworkflowID: \"workflowID-unsupported\",\n\t\t\trunID:      \"runID-unsupported\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&dummyTaskType{\n\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\tTaskID:              9999,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\tvalidate: func(t *testing.T, tasks []*nosqlplugin.HistoryMigrationTask, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, tasks)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(mockCtrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(mockCtrl)\n\t\t\tstore := newTestNosqlExecutionStoreWithTaskSerializer(mockDB, log.NewNoop(), mockTaskSerializer)\n\t\t\ttc.setupMocks(mockTaskSerializer)\n\n\t\t\ttasks, err := store.prepareTransferTasksForWorkflowTxn(tc.domainID, tc.workflowID, tc.runID, tc.tasks)\n\t\t\ttc.validate(t, tasks, err)\n\t\t})\n\t}\n}\n\nfunc TestNosqlExecutionStoreUtilsExtended(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupStore func(store *nosqlExecutionStore) (interface{}, error)\n\t\tvalidate   func(t *testing.T, result interface{}, err error)\n\t}{\n\t\t{\n\t\t\tname: \"PrepareActivityInfosForWorkflowTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tactivityInfos := []*persistence.InternalActivityInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tScheduleID:     1,\n\t\t\t\t\t\tScheduledEvent: persistence.NewDataBlob([]byte(\"scheduled event data\"), constants.EncodingTypeThriftRW),\n\t\t\t\t\t\tStartedEvent:   persistence.NewDataBlob([]byte(\"started event data\"), constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareActivityInfosForWorkflowTxn(activityInfos)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tinfos, ok := result.(map[int64]*persistence.InternalActivityInfo)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Len(t, infos, 1)\n\t\t\t\tfor _, info := range infos {\n\t\t\t\t\tassert.NotNil(t, info.ScheduledEvent)\n\t\t\t\t\tassert.NotNil(t, info.StartedEvent)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerInfosForWorkflowTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\ttimerInfos := []*persistence.TimerInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareTimerInfosForWorkflowTxn(timerInfos)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tinfos, ok := result.(map[string]*persistence.TimerInfo)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Len(t, infos, 1)\n\t\t\t\tassert.NotNil(t, infos[\"timer1\"])\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareChildWFInfosForWorkflowTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tchildWFInfos := []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:    1,\n\t\t\t\t\t\tInitiatedEvent: persistence.NewDataBlob([]byte(\"initiated event data\"), constants.EncodingTypeThriftRW),\n\t\t\t\t\t\tStartedEvent:   persistence.NewDataBlob([]byte(\"started event data\"), constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareChildWFInfosForWorkflowTxn(childWFInfos)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tinfos, ok := result.(map[int64]*persistence.InternalChildExecutionInfo)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Len(t, infos, 1)\n\t\t\t\tfor _, info := range infos {\n\t\t\t\t\tassert.NotNil(t, info.InitiatedEvent)\n\t\t\t\t\tassert.NotNil(t, info.StartedEvent)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareTimerInfosForWorkflowTxn - Nil Timer Info\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\treturn store.prepareTimerInfosForWorkflowTxn(nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Empty(t, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareChildWFInfosForWorkflowTxn - Nil Child Execution Info\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\treturn store.prepareChildWFInfosForWorkflowTxn(nil)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Empty(t, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareChildWFInfosForWorkflowTxn - Encoding Mismatch Error\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tchildWFInfos := []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:    1,\n\t\t\t\t\t\tInitiatedEvent: persistence.NewDataBlob([]byte(\"initiated\"), constants.EncodingTypeThriftRW),\n\t\t\t\t\t\tStartedEvent:   persistence.NewDataBlob([]byte(\"started\"), constants.EncodingTypeJSON), // Encoding mismatch\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\treturn store.prepareChildWFInfosForWorkflowTxn(childWFInfos)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareRequestCancelsForWorkflowTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\trequestCancels := []*persistence.RequestCancelInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:     1,\n\t\t\t\t\t\tCancelRequestID: \"cancel-1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:     2,\n\t\t\t\t\t\tCancelRequestID: \"cancel-2\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tcancels, err := store.prepareRequestCancelsForWorkflowTxn(requestCancels)\n\t\t\t\treturn cancels, err\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tcancels := result.(map[int64]*persistence.RequestCancelInfo)\n\t\t\t\tassert.Equal(t, 2, len(cancels))\n\t\t\t\tassert.Contains(t, cancels, int64(1))\n\t\t\t\tassert.Contains(t, cancels, int64(2))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareRequestCancelsForWorkflowTxn - Duplicate Initiated IDs\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\trequestCancels := []*persistence.RequestCancelInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:     1,\n\t\t\t\t\t\tCancelRequestID: \"cancel-1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tInitiatedID:     1, // Duplicate InitiatedID\n\t\t\t\t\t\tCancelRequestID: \"cancel-1-duplicate\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tcancels, err := store.prepareRequestCancelsForWorkflowTxn(requestCancels)\n\t\t\t\treturn cancels, err\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tcancels := result.(map[int64]*persistence.RequestCancelInfo)\n\t\t\t\tassert.Equal(t, 1, len(cancels))\n\t\t\t\tassert.Equal(t, \"cancel-1-duplicate\", cancels[1].CancelRequestID)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareSignalInfosForWorkflowTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tsignalInfos := []*persistence.SignalInfo{\n\t\t\t\t\t{InitiatedID: 1, SignalRequestID: \"signal-1\"},\n\t\t\t\t\t{InitiatedID: 2, SignalRequestID: \"signal-2\"},\n\t\t\t\t}\n\t\t\t\treturn store.prepareSignalInfosForWorkflowTxn(signalInfos)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tinfos := result.(map[int64]*persistence.SignalInfo)\n\t\t\t\tassert.Equal(t, 2, len(infos))\n\t\t\t\tassert.Equal(t, \"signal-1\", infos[1].SignalRequestID)\n\t\t\t\tassert.Equal(t, \"signal-2\", infos[2].SignalRequestID)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareUpdateWorkflowExecutionTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"test-domain-id\",\n\t\t\t\t\tWorkflowID:  \"test-workflow-id\",\n\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"test-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t}\n\t\t\t\tchecksum := checksum.Checksum{Version: 1,\n\t\t\t\t\tValue: []byte(\"create-checksum\")}\n\t\t\t\treturn store.prepareUpdateWorkflowExecutionTxn(executionInfo, versionHistories, checksum, time.Now(), 123)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\treq := result.(*nosqlplugin.WorkflowExecutionRequest)\n\t\t\t\tassert.Equal(t, \"test-domain-id\", req.DomainID)\n\t\t\t\tassert.Equal(t, int64(123), req.LastWriteVersion)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareUpdateWorkflowExecutionTxn - Emptyvalues\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"\",\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t}\n\t\t\t\tchecksum := checksum.Checksum{Version: 1, Value: []byte(\"checksum\")}\n\t\t\t\t// This should result in an error due to invalid executionInfo state for the creation scenario\n\t\t\t\treturn store.prepareUpdateWorkflowExecutionTxn(executionInfo, versionHistories, checksum, time.Now(), 123)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.Error(t, err) // Expect an error due to invalid state\n\t\t\t\tassert.Nil(t, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareUpdateWorkflowExecutionTxn - Invalid Workflow State\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"domainID-invalid-state\",\n\t\t\t\t\tWorkflowID:  \"workflowID-invalid-state\",\n\t\t\t\t\tRunID:       \"runID-invalid-state\",\n\t\t\t\t\tState:       343, // Invalid state\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t}\n\t\t\t\tchecksum := checksum.Checksum{Version: 1, Value: []byte(\"checksum\")}\n\t\t\t\treturn store.prepareUpdateWorkflowExecutionTxn(executionInfo, versionHistories, checksum, time.Now(), 123)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.Error(t, err)  // Expect an error due to invalid workflow state\n\t\t\t\tassert.Nil(t, result) // No WorkflowExecutionRequest should be returned\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareCreateWorkflowExecutionTxn - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"create-domain-id\",\n\t\t\t\t\tWorkflowID:  \"create-workflow-id\",\n\t\t\t\t\tRunID:       \"create-run-id\",\n\t\t\t\t\tState:       persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"create-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t}\n\t\t\t\tchecksum := checksum.Checksum{Version: 1, Value: []byte(\"create-checksum\")}\n\t\t\t\treturn store.prepareCreateWorkflowExecutionTxn(executionInfo, versionHistories, checksum, time.Now(), 123)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\treq := result.(*nosqlplugin.WorkflowExecutionRequest)\n\t\t\t\tassert.Equal(t, \"create-domain-id\", req.DomainID)\n\t\t\t\tassert.Equal(t, int64(123), req.LastWriteVersion)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PrepareCreateWorkflowExecutionTxn - Invalid State\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"create-domain-id\",\n\t\t\t\t\tWorkflowID:  \"create-workflow-id\",\n\t\t\t\t\tRunID:       \"create-run-id\",\n\t\t\t\t\tState:       232, // Invalid state for creating a workflow execution\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\t\t\tData:     []byte(`[{\"Branches\":[{\"BranchID\":\"create-branch-id\",\"BeginNodeID\":1,\"EndNodeID\":2}]}]`),\n\t\t\t\t}\n\t\t\t\tchecksum := checksum.Checksum{Version: 1, Value: []byte(\"create-checksum\")}\n\t\t\t\treturn store.prepareCreateWorkflowExecutionTxn(executionInfo, versionHistories, checksum, time.Now(), 123)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"prepareCurrentWorkflowRequestForCreateWorkflowTxn - BrandNew mode\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tCreateRequestID: \"test-create-request-id\",\n\t\t\t\t}\n\t\t\t\trequest := &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\t\tMode: persistence.CreateWorkflowModeBrandNew,\n\t\t\t\t}\n\t\t\t\treturn store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(\n\t\t\t\t\t\"test-domain-id\", \"test-workflow-id\", \"test-run-id\", executionInfo, 123, request)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tcurrentWorkflowReq, ok := result.(*nosqlplugin.CurrentWorkflowWriteRequest)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.NotNil(t, currentWorkflowReq)\n\t\t\t\tassert.Equal(t, nosqlplugin.CurrentWorkflowWriteModeInsert, currentWorkflowReq.WriteMode)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"processUpdateWorkflowResult - CurrentWorkflowConditionFailInfo error\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\"current workflow condition failed\"),\n\t\t\t\t}\n\t\t\t\treturn nil, store.processUpdateWorkflowResult(err, 99)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, _ interface{}, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\t_, ok := err.(*persistence.CurrentWorkflowConditionFailedError)\n\t\t\t\tassert.True(t, ok)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"processUpdateWorkflowResult - Success\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\treturn nil, store.processUpdateWorkflowResult(nil, 99)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, _ interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"processUpdateWorkflowResult - ShardRangeIDNotMatch error\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\terr := &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\t\tShardRangeIDNotMatch: common.Int64Ptr(100),\n\t\t\t\t}\n\t\t\t\treturn nil, store.processUpdateWorkflowResult(err, 99)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, _ interface{}, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\t_, ok := err.(*persistence.ShardOwnershipLostError)\n\t\t\t\tassert.True(t, ok)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"prepareCurrentWorkflowRequestForCreateWorkflowTxn - WorkflowIDReuse mode with non-completed state\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tState:           persistence.WorkflowStateRunning, // Simulate a running state\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tCreateRequestID: \"test-create-request-id\",\n\t\t\t\t}\n\t\t\t\trequest := &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\t\tMode:                     persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\t\tPreviousRunID:            \"test-run-id\",\n\t\t\t\t\tPreviousLastWriteVersion: 123, // Simulating a non-completed state with a valid version\n\t\t\t\t}\n\t\t\t\treturn store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(\n\t\t\t\t\t\"test-domain-id\", \"test-workflow-id\", \"test-run-id\", executionInfo, 123, request)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\t_, ok := err.(*persistence.CurrentWorkflowConditionFailedError)\n\t\t\t\tassert.False(t, ok)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CurrentWorkflowRequestForCreateWorkflowTxn - Zombie mode\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tCreateRequestID: \"create-request-id-zombie\",\n\t\t\t\t}\n\t\t\t\trequest := &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\t\tMode:          persistence.CreateWorkflowModeZombie,\n\t\t\t\t\tPreviousRunID: \"previous-run-id-zombie\",\n\t\t\t\t}\n\t\t\t\treturn store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(\n\t\t\t\t\t\"domain-id-zombie\", \"workflow-id-zombie\", \"run-id-zombie\", executionInfo, 123, request)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tcurrentWorkflowReq := result.(*nosqlplugin.CurrentWorkflowWriteRequest)\n\t\t\t\tassert.Equal(t, nosqlplugin.CurrentWorkflowWriteModeNoop, currentWorkflowReq.WriteMode)\n\t\t\t\tassert.Equal(t, \"create-request-id-zombie\", currentWorkflowReq.Row.CreateRequestID)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CurrentWorkflowRequestForCreateWorkflowTxn - ContinueAsNew mode\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\texecutionInfo := &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tState:           persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tCreateRequestID: \"create-request-id-continueasnew\",\n\t\t\t\t}\n\t\t\t\trequest := &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\t\tMode:          persistence.CreateWorkflowModeContinueAsNew,\n\t\t\t\t\tPreviousRunID: \"previous-run-id-continueasnew\",\n\t\t\t\t}\n\t\t\t\treturn store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(\n\t\t\t\t\t\"domain-id-continueasnew\", \"workflow-id-continueasnew\", \"run-id-continueasnew\", executionInfo, 123, request)\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, result interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tcurrentWorkflowReq := result.(*nosqlplugin.CurrentWorkflowWriteRequest)\n\t\t\t\tassert.Equal(t, nosqlplugin.CurrentWorkflowWriteModeUpdate, currentWorkflowReq.WriteMode)\n\t\t\t\tassert.Equal(t, \"create-request-id-continueasnew\", currentWorkflowReq.Row.CreateRequestID)\n\t\t\t\tassert.NotNil(t, currentWorkflowReq.Condition)\n\t\t\t\tassert.Equal(t, \"previous-run-id-continueasnew\", *currentWorkflowReq.Condition.CurrentRunID)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"assertNotCurrentExecution - Success with different RunID\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tctx := context.Background()\n\t\t\t\tmockDB := store.db.(*nosqlplugin.MockDB)\n\t\t\t\tmockDB.EXPECT().SelectCurrentWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tstore.shardID,\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\t\"test-workflow-id\",\n\t\t\t\t).Return(&nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID: \"different-run-id\",\n\t\t\t\t}, nil)\n\t\t\t\treturn nil, store.assertNotCurrentExecution(ctx, \"test-domain-id\", \"test-workflow-id\", \"expected-run-id\")\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, _ interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"assertNotCurrentExecution - No current workflow\",\n\t\t\tsetupStore: func(store *nosqlExecutionStore) (interface{}, error) {\n\t\t\t\tctx := context.Background()\n\t\t\t\tmockDB := store.db.(*nosqlplugin.MockDB)\n\n\t\t\t\tmockDB.EXPECT().SelectCurrentWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tstore.shardID,\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\t\"test-workflow-id\",\n\t\t\t\t).Return(nil, &types.EntityNotExistsError{})\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t\treturn nil, store.assertNotCurrentExecution(ctx, \"test-domain-id\", \"test-workflow-id\", \"expected-run-id\")\n\t\t\t},\n\t\t\tvalidate: func(t *testing.T, _ interface{}, err error) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDB := nosqlplugin.NewMockDB(mockCtrl)\n\t\t\tstore := newTestNosqlExecutionStore(mockDB, log.NewNoop())\n\n\t\t\tresult, err := tc.setupStore(store)\n\t\t\ttc.validate(t, result, err)\n\t\t})\n\t}\n}\n\ntype dummyTaskType struct {\n\tpersistence.Task\n\tVisibilityTimestamp time.Time\n\tTaskID              int64\n}\n\nfunc (d *dummyTaskType) GetTaskType() int {\n\treturn 999 // Using a type that is not expected by the switch statement\n}\n\nfunc (d *dummyTaskType) GetVersion() int64 {\n\treturn 1\n}\n\nfunc (d *dummyTaskType) SetVersion(version int64) {}\n\nfunc (d *dummyTaskType) ToTransferTaskInfo() (*persistence.TransferTaskInfo, error) {\n\treturn nil, errors.New(\"not implemented\")\n}\n\nfunc (d *dummyTaskType) ToTimerTaskInfo() (*persistence.TimerTaskInfo, error) {\n\treturn nil, errors.New(\"not implemented\")\n}\n\nfunc (d *dummyTaskType) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, errors.New(\"not implemented\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_history_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype nosqlHistoryStore struct {\n\tshardedNosqlStore\n}\n\n// newNoSQLHistoryStore is used to create an instance of HistoryStore implementation\nfunc newNoSQLHistoryStore(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.HistoryStore, error) {\n\ts, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlHistoryStore{\n\t\tshardedNosqlStore: s,\n\t}, nil\n}\n\n// AppendHistoryNodes upsert a batch of events as a single node to a history branch\n// Note that it's not allowed to append above the branch's ancestors' nodes, which means nodeID >= ForkNodeID\nfunc (h *nosqlHistoryStore) AppendHistoryNodes(\n\tctx context.Context,\n\trequest *persistence.InternalAppendHistoryNodesRequest,\n) error {\n\tbranchInfo := request.BranchInfo\n\tbeginNodeID := persistenceutils.GetBeginNodeID(branchInfo)\n\n\tif request.NodeID < beginNodeID {\n\t\treturn &persistence.InvalidPersistenceRequestError{\n\t\t\tMsg: \"cannot append to ancestors' nodes\",\n\t\t}\n\t}\n\n\tvar treeRow *nosqlplugin.HistoryTreeRow\n\tif request.IsNewBranch {\n\t\tvar ancestors []*types.HistoryBranchRange\n\t\tancestors = append(ancestors, branchInfo.Ancestors...)\n\t\ttreeRow = &nosqlplugin.HistoryTreeRow{\n\t\t\tShardID:         request.ShardID,\n\t\t\tTreeID:          branchInfo.TreeID,\n\t\t\tBranchID:        branchInfo.BranchID,\n\t\t\tAncestors:       ancestors,\n\t\t\tCreateTimestamp: request.CurrentTimeStamp,\n\t\t\tInfo:            request.Info,\n\t\t}\n\t}\n\tnodeRow := &nosqlplugin.HistoryNodeRow{\n\t\tTreeID:          branchInfo.TreeID,\n\t\tBranchID:        branchInfo.BranchID,\n\t\tNodeID:          request.NodeID,\n\t\tTxnID:           &request.TransactionID,\n\t\tData:            request.Events.Data,\n\t\tDataEncoding:    string(request.Events.Encoding),\n\t\tShardID:         request.ShardID,\n\t\tCreateTimestamp: request.CurrentTimeStamp,\n\t}\n\n\tstoreShard, err := h.GetStoreShardByHistoryShard(request.ShardID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = storeShard.db.InsertIntoHistoryTreeAndNode(ctx, treeRow, nodeRow)\n\n\tif err != nil {\n\t\treturn convertCommonErrors(storeShard.db, \"AppendHistoryNodes\", err)\n\t}\n\treturn nil\n}\n\n// ReadHistoryBranch returns history node data for a branch\n// NOTE: For branch that has ancestors, we need to query Cassandra multiple times, because it doesn't support OR/UNION operator\nfunc (h *nosqlHistoryStore) ReadHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalReadHistoryBranchRequest,\n) (*persistence.InternalReadHistoryBranchResponse, error) {\n\tfilter := &nosqlplugin.HistoryNodeFilter{\n\t\tShardID:       request.ShardID,\n\t\tTreeID:        request.TreeID,\n\t\tBranchID:      request.BranchID,\n\t\tMinNodeID:     request.MinNodeID,\n\t\tMaxNodeID:     request.MaxNodeID,\n\t\tNextPageToken: request.NextPageToken,\n\t\tPageSize:      request.PageSize,\n\t}\n\n\tstoreShard, err := h.GetStoreShardByHistoryShard(request.ShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trows, pagingToken, err := storeShard.db.SelectFromHistoryNode(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"SelectFromHistoryNode\", err)\n\t}\n\n\thistory := make([]*persistence.DataBlob, 0, int(request.PageSize))\n\n\teventBlob := &persistence.DataBlob{}\n\tnodeID := int64(0)\n\ttxnID := int64(0)\n\tlastNodeID := request.LastNodeID\n\tlastTxnID := request.LastTransactionID\n\n\tfor _, row := range rows {\n\t\tnodeID = row.NodeID\n\t\ttxnID = *row.TxnID\n\t\teventBlob.Data = row.Data\n\t\teventBlob.Encoding = constants.EncodingType(row.DataEncoding)\n\t\tif txnID < lastTxnID {\n\t\t\t// assuming that business logic layer is correct and transaction ID only increase\n\t\t\t// thus, valid event batch will come with increasing transaction ID\n\n\t\t\t// event batches with smaller node ID\n\t\t\t//  -> should not be possible since records are already sorted\n\t\t\t// event batches with same node ID\n\t\t\t//  -> batch with higher transaction ID is valid\n\t\t\t// event batches with larger node ID\n\t\t\t//  -> batch with lower transaction ID is invalid (happens before)\n\t\t\t//  -> batch with higher transaction ID is valid\n\t\t\tcontinue\n\t\t}\n\n\t\tswitch {\n\t\tcase nodeID < lastNodeID:\n\t\t\treturn nil, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted data, nodeID cannot decrease\",\n\t\t\t}\n\t\tcase nodeID == lastNodeID:\n\t\t\treturn nil, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted data, same nodeID must have smaller txnID\",\n\t\t\t}\n\t\tdefault: // row.NodeID > lastNodeID:\n\t\t\t// NOTE: when row.nodeID > lastNodeID, we expect the one with largest txnID comes first\n\t\t\tlastTxnID = txnID\n\t\t\tlastNodeID = nodeID\n\t\t\thistory = append(history, eventBlob)\n\t\t\teventBlob = &persistence.DataBlob{}\n\t\t}\n\t}\n\n\treturn &persistence.InternalReadHistoryBranchResponse{\n\t\tHistory:           history,\n\t\tNextPageToken:     pagingToken,\n\t\tLastNodeID:        lastNodeID,\n\t\tLastTransactionID: lastTxnID,\n\t}, nil\n}\n\n// ForkHistoryBranch forks a new branch from an existing branch\n// Note that application must provide a void forking nodeID, it must be a valid nodeID in that branch.\n// A valid forking nodeID can be an ancestor from the existing branch.\n// For example, we have branch B1 with three nodes(1[1,2], 3[3,4,5] and 6[6,7,8]. 1, 3 and 6 are nodeIDs (first eventID of the batch).\n// So B1 looks like this:\n//\n//\t     1[1,2]\n//\t     /\n//\t   3[3,4,5]\n//\t  /\n//\t6[6,7,8]\n//\n// Assuming we have branch B2 which contains one ancestor B1 stopping at 6 (exclusive). So B2 inherit nodeID 1 and 3 from B1, and have its own nodeID 6 and 8.\n// Branch B2 looks like this:\n//\n//\t  1[1,2]\n//\t  /\n//\t3[3,4,5]\n//\t \\\n//\t  6[6,7]\n//\t  \\\n//\t   8[8]\n//\n// Now we want to fork a new branch B3 from B2.\n// The only valid forking nodeIDs are 3,6 or 8.\n// 1 is not valid because we can't fork from first node.\n// 2/4/5 is NOT valid either because they are inside a batch.\n//\n// Case #1: If we fork from nodeID 6, then B3 will have an ancestor B1 which stops at 6(exclusive).\n// As we append a batch of events[6,7,8,9] to B3, it will look like :\n//\n//\t  1[1,2]\n//\t  /\n//\t3[3,4,5]\n//\t \\\n//\t6[6,7,8,9]\n//\n// Case #2: If we fork from node 8, then B3 will have two ancestors: B1 stops at 6(exclusive) and ancestor B2 stops at 8(exclusive)\n// As we append a batch of events[8,9] to B3, it will look like:\n//\n//\t     1[1,2]\n//\t     /\n//\t   3[3,4,5]\n//\t  /\n//\t6[6,7]\n//\t \\\n//\t 8[8,9]\nfunc (h *nosqlHistoryStore) ForkHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalForkHistoryBranchRequest,\n) (*persistence.InternalForkHistoryBranchResponse, error) {\n\n\tforkB := request.ForkBranchInfo\n\ttreeID := forkB.TreeID\n\tnewAncestors := make([]*types.HistoryBranchRange, 0, len(forkB.Ancestors)+1)\n\n\tbeginNodeID := persistenceutils.GetBeginNodeID(forkB)\n\tif beginNodeID >= request.ForkNodeID {\n\t\t// this is the case that new branch's ancestors doesn't include the forking branch\n\t\tfor _, br := range forkB.Ancestors {\n\t\t\tif br.EndNodeID >= request.ForkNodeID {\n\t\t\t\tnewAncestors = append(newAncestors, &types.HistoryBranchRange{\n\t\t\t\t\tBranchID:    br.BranchID,\n\t\t\t\t\tBeginNodeID: br.BeginNodeID,\n\t\t\t\t\tEndNodeID:   request.ForkNodeID,\n\t\t\t\t})\n\t\t\t\tbreak\n\t\t\t} else {\n\t\t\t\tnewAncestors = append(newAncestors, br)\n\t\t\t}\n\t\t}\n\t} else {\n\t\t// this is the case the new branch will inherit all ancestors from forking branch\n\t\tnewAncestors = forkB.Ancestors\n\t\tnewAncestors = append(newAncestors, &types.HistoryBranchRange{\n\t\t\tBranchID:    forkB.BranchID,\n\t\t\tBeginNodeID: beginNodeID,\n\t\t\tEndNodeID:   request.ForkNodeID,\n\t\t})\n\t}\n\n\tresp := &persistence.InternalForkHistoryBranchResponse{\n\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:    treeID,\n\t\t\tBranchID:  request.NewBranchID,\n\t\t\tAncestors: newAncestors,\n\t\t}}\n\n\tvar ancestors []*types.HistoryBranchRange\n\tfor _, an := range newAncestors {\n\t\tanc := &types.HistoryBranchRange{\n\t\t\tBranchID:  an.BranchID,\n\t\t\tEndNodeID: an.EndNodeID,\n\t\t}\n\t\tancestors = append(ancestors, anc)\n\t}\n\ttreeRow := &nosqlplugin.HistoryTreeRow{\n\t\tShardID:         request.ShardID,\n\t\tTreeID:          treeID,\n\t\tBranchID:        request.NewBranchID,\n\t\tAncestors:       ancestors,\n\t\tCreateTimestamp: request.CurrentTimeStamp,\n\t\tInfo:            request.Info,\n\t}\n\n\tstoreShard, err := h.GetStoreShardByHistoryShard(request.ShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = storeShard.db.InsertIntoHistoryTreeAndNode(ctx, treeRow, nil)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"ForkHistoryBranch\", err)\n\t}\n\treturn resp, nil\n}\n\n// DeleteHistoryBranch removes a branch. This is responsible for:\n//\n//   - Deleting entries from the history_node table\n//   - Deleting the tree entry in the history_tree table\n//   - Handling the problem of garbage-collecting branches which may mutually rely on history\n//\n// For normal workflows, with only a single branch, this is straightfoward, it'll just do a\n// delete on the history_node and history_tree tables. However, in cases where history is\n// branched, things are slightly more complex. History branches in at least two ways:\n//   - During the conflict resolution handling of events during failover (where each\n//     region is represented as a history branch at the point where they were operating concurrently.\n//   - At the point of a workflow reset, where history is 'forked' and the previous\n//     workflow events are discarded, but the original workflow run's history is referenced.\n//\n// For workflows with forked history, they'll reference any nodes they rely on ancestors in the history_tree\n// table. These references are exclusive, ie, the following is true:\n//\n//\t--------------+--------------------------------------\n//\t tree_id      | X\n//\t branch_id    | B\n//\t ancestors    | [{branch_id: A, end_node_id: 4}]\n//\t--------------+--------------------------------------\n//\t tree_id      | X\n//\t branch_id    | A\n//\t ancestors    | null\n//\n//\tWill have nodes arranged like this\n//\n//\t    ┌────────────┐\n//\t    │ Branch: A  │\n//\t    │ Node:   1  │\n//\t    └─────┬──────┘\n//\t          │\n//\t    ┌─────▼──────┐\n//\t    │ Branch: A  │\n//\t    │ Node:   2  │\n//\t    └─────┬──────┘\n//\t          │\n//\t    ┌─────▼──────┐\n//\t    │ Branch: A  │\n//\t    │ Node:   3  ┼────────────┐\n//\t    └─────┬──────┘            │\n//\t          │                   │\n//\t    ┌─────▼──────┐     ┌──────▼─────┐\n//\t    │ Branch: A  │     │ Branch: B  │\n//\t    │ Node:   4  │     │ Node:   4  │\n//\t    └────────────┘     └────────────┘\n//\n// The method of cleanup for each branch therefore, is to just remove the nodes that are not in use,\n// nearly-always starting from the oldest branch (the original run) and, later in subsequent invocations,\n// the branches relying on this. These cleanups are done by a call to the lower persistence layer\n// with HistoryNodeFilter references specifying the minimum nodes to be cleaned up (inclusive).\n//\n// While workflowIDs being enforced to be held by a single run to execute at any one time\n// generally ensures that any ancestor is cleaned up first, there's a couple of exceptions\n// to keep in mind. Firstly, deletion jitter makes it near-certain that there'll be some overlap and therefore\n// no guarantee that the oldest branch gets cleaned up first, as well as timer tasks are likely to retry and therefore\n// end up being out of order anyway. So any cleanup needs to be able to handle these cases.\nfunc (h *nosqlHistoryStore) DeleteHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteHistoryBranchRequest,\n) error {\n\n\tbranch := request.BranchInfo\n\ttreeID := branch.TreeID\n\tbrsToDelete := branch.Ancestors\n\tbeginNodeID := persistenceutils.GetBeginNodeID(branch)\n\tbrsToDelete = append(brsToDelete, &types.HistoryBranchRange{\n\t\tBranchID:    branch.BranchID,\n\t\tBeginNodeID: beginNodeID,\n\t})\n\n\trsp, err := h.GetHistoryTree(ctx, &persistence.InternalGetHistoryTreeRequest{\n\t\tTreeID:  treeID,\n\t\tShardID: &request.ShardID,\n\t})\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  request.ShardID,\n\t\tTreeID:   treeID,\n\t\tBranchID: &branch.BranchID,\n\t}\n\n\t// This is the selection of history_nodes that will be removed\n\t// at this point it's not safe to delete any nodes, because we don't know\n\t// if this branch has other branches that've taken a dependency on it's nodes\n\t// so we start with an empty filter\n\tvar nodeFilters []*nosqlplugin.HistoryNodeFilter\n\n\t// ... and then look at all the active / good history branches and see\n\t// what they're referencing. The idea being, to trim any nodes that\n\t// might not in use by going and finding the topmost nodes that are currently\n\t// referenced, and trimming those.\n\t//\n\t// validBRsMaxEndNode represents each branch range that is being used,\n\t// we want to know what is the max nodeID referred by other valid branches\n\tvalidBRsMaxEndNode := persistenceutils.GetBranchesMaxReferredNodeIDs(rsp.Branches)\n\n\ttreesByBranchID := rsp.ByBranchID()\n\n\t// for each branch range to delete, we iterate from bottom to up, and delete up to the point according to validBRsEndNode\n\t// brsToDelete here includes both the current branch being operated on and its ancestors\n\tfor i := len(brsToDelete) - 1; i >= 0; i-- {\n\n\t\tbr := brsToDelete[i]\n\n\t\tmaxReferredEndNodeID, branchIsReferenced := validBRsMaxEndNode[br.BranchID]\n\t\tisLastBranchRemaining := len(rsp.Branches) <= 1\n\t\t_, treeExists := treesByBranchID[br.BranchID]\n\n\t\tif treeExists && br.BranchID != request.BranchInfo.BranchID {\n\t\t\t// the underlying assumption of this cleanup logic, is that cleanup will happen\n\t\t\t// from the oldest branch cleaning up first, to the more recent.\n\t\t\t//\n\t\t\t// However, this is by no means guaranteed, timers jitter on deletion intentionally\n\t\t\t// and so to avoid accidentally breaking valid ancestor branches with short-lived\n\t\t\t// branches doing cleanup, we'll stop the history node reaping here and let them\n\t\t\t// clean their history nodes when they're removed\n\t\t\th.GetLogger().Debug(fmt.Sprintf(\"out of order deletion observed trying to clean up branch %q\", br.BranchID),\n\t\t\t\ttag.WorkflowTreeID(request.BranchInfo.TreeID),\n\t\t\t\ttag.WorkflowBranchID(request.BranchInfo.BranchID))\n\t\t\tbreak\n\t\t}\n\n\t\tif isLastBranchRemaining && branchIsReferenced {\n\t\t\t// special case, when this branch is being deleted is the last branch\n\t\t\t// and it's referring to multiple other previous (and now nonexisting branches)\n\t\t\t// then in this instance, delete all referenced branch nodes\n\t\t\tnodeFilter := &nosqlplugin.HistoryNodeFilter{\n\t\t\t\tShardID:  request.ShardID,\n\t\t\t\tTreeID:   treeID,\n\t\t\t\tBranchID: br.BranchID,\n\t\t\t}\n\t\t\tnodeFilters = append(nodeFilters, nodeFilter)\n\t\t} else if branchIsReferenced {\n\t\t\t// we can only delete from the maxEndNode and stop here\n\t\t\tnodeFilter := &nosqlplugin.HistoryNodeFilter{\n\t\t\t\tShardID:   request.ShardID,\n\t\t\t\tTreeID:    treeID,\n\t\t\t\tBranchID:  br.BranchID,\n\t\t\t\tMinNodeID: maxReferredEndNodeID,\n\t\t\t}\n\t\t\tnodeFilters = append(nodeFilters, nodeFilter)\n\t\t\tbreak\n\t\t} else {\n\t\t\t// No any branch is using this range, we can delete all of it\n\t\t\tnodeFilter := &nosqlplugin.HistoryNodeFilter{\n\t\t\t\tShardID:   request.ShardID,\n\t\t\t\tTreeID:    treeID,\n\t\t\t\tBranchID:  br.BranchID,\n\t\t\t\tMinNodeID: br.BeginNodeID,\n\t\t\t}\n\t\t\tnodeFilters = append(nodeFilters, nodeFilter)\n\t\t}\n\t}\n\n\tstoreShard, err := h.GetStoreShardByHistoryShard(request.ShardID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = storeShard.db.DeleteFromHistoryTreeAndNode(ctx, treeFilter, nodeFilters)\n\tif err != nil {\n\t\treturn convertCommonErrors(storeShard.db, \"DeleteHistoryBranch\", err)\n\t}\n\treturn nil\n}\n\nfunc (h *nosqlHistoryStore) GetAllHistoryTreeBranches(\n\tctx context.Context,\n\trequest *persistence.GetAllHistoryTreeBranchesRequest,\n) (*persistence.GetAllHistoryTreeBranchesResponse, error) {\n\n\tif h.GetShardingPolicy().hasShardedHistory {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectAllHistoryTrees is not supported on sharded nosql db\",\n\t\t}\n\t}\n\n\tstoreShard := h.GetDefaultShard()\n\tdbBranches, pagingToken, err := storeShard.db.SelectAllHistoryTrees(ctx, request.NextPageToken, request.PageSize)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"SelectAllHistoryTrees\", err)\n\t}\n\n\tbranchDetails := make([]persistence.HistoryBranchDetail, 0, int(request.PageSize))\n\n\tfor _, branch := range dbBranches {\n\n\t\tbranchDetail := persistence.HistoryBranchDetail{\n\t\t\tTreeID:   branch.TreeID,\n\t\t\tBranchID: branch.BranchID,\n\t\t\tForkTime: branch.CreateTimestamp,\n\t\t\tInfo:     branch.Info,\n\t\t}\n\t\tbranchDetails = append(branchDetails, branchDetail)\n\t}\n\n\tresponse := &persistence.GetAllHistoryTreeBranchesResponse{\n\t\tBranches:      branchDetails,\n\t\tNextPageToken: pagingToken,\n\t}\n\n\treturn response, nil\n}\n\n// GetHistoryTree returns all branch information of a tree\nfunc (h *nosqlHistoryStore) GetHistoryTree(\n\tctx context.Context,\n\trequest *persistence.InternalGetHistoryTreeRequest,\n) (*persistence.InternalGetHistoryTreeResponse, error) {\n\n\ttreeID := request.TreeID\n\tstoreShard, err := h.GetStoreShardByHistoryShard(*request.ShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdbBranches, err := storeShard.db.SelectFromHistoryTree(ctx,\n\t\t&nosqlplugin.HistoryTreeFilter{\n\t\t\tShardID: *request.ShardID,\n\t\t\tTreeID:  treeID,\n\t\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"SelectFromHistoryTree\", err)\n\t}\n\n\tbranches := make([]*types.HistoryBranch, 0)\n\tfor _, dbBr := range dbBranches {\n\t\tbr := &types.HistoryBranch{\n\t\t\tTreeID:    treeID,\n\t\t\tBranchID:  dbBr.BranchID,\n\t\t\tAncestors: dbBr.Ancestors,\n\t\t}\n\t\tbranches = append(branches, br)\n\t}\n\treturn &persistence.InternalGetHistoryTreeResponse{\n\t\tBranches: branches,\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_history_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\tctx \"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestTransactionID = 123\n\ttestShardID       = 456\n\ttestNodeID        = 8\n)\n\nfunc validInternalAppendHistoryNodesRequest() *persistence.InternalAppendHistoryNodesRequest {\n\treturn &persistence.InternalAppendHistoryNodesRequest{\n\t\tIsNewBranch: false,\n\t\tInfo:        \"TestInfo\",\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"TestBranchID\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   5,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNodeID: testNodeID,\n\t\tEvents: &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\tData:     []byte(\"TestEvents\"),\n\t\t},\n\t\tTransactionID:    testTransactionID,\n\t\tShardID:          testShardID,\n\t\tCurrentTimeStamp: FixedTime,\n\t}\n}\n\nfunc validHistoryNodeRow() *nosqlplugin.HistoryNodeRow {\n\texpectedNodeRow := &nosqlplugin.HistoryNodeRow{\n\t\tTreeID:          \"TestTreeID\",\n\t\tBranchID:        \"TestBranchID\",\n\t\tNodeID:          testNodeID,\n\t\tTxnID:           common.Ptr[int64](123),\n\t\tData:            []byte(\"TestEvents\"),\n\t\tDataEncoding:    string(constants.EncodingTypeThriftRW),\n\t\tShardID:         testShardID,\n\t\tCreateTimestamp: FixedTime,\n\t}\n\treturn expectedNodeRow\n}\n\nfunc registerCassandraMock(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockDB := nosqlplugin.NewMockDB(ctrl)\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(ctrl)\n\tmockPlugin.EXPECT().CreateDB(gomock.Any(), gomock.Any(), gomock.Any()).Return(mockDB, nil).AnyTimes()\n\tRegisterPlugin(\"cassandra\", mockPlugin)\n}\n\nfunc TestNewNoSQLHistoryStore(t *testing.T) {\n\tregisterCassandraMock(t)\n\tcfg := getValidShardedNoSQLConfig()\n\n\tstore, err := newNoSQLHistoryStore(cfg, log.NewNoop(), metrics.NewNoopMetricsClient(), nil)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, store)\n}\n\nfunc setUpMocks(t *testing.T) (*nosqlHistoryStore, *nosqlplugin.MockDB, *MockshardedNosqlStore) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\n\tnosqlSt := nosqlStore{\n\t\tlogger: log.NewNoop(),\n\t\tdb:     dbMock,\n\t}\n\n\tshardedNosqlStoreMock := NewMockshardedNosqlStore(ctrl)\n\tshardedNosqlStoreMock.EXPECT().GetStoreShardByHistoryShard(testShardID).Return(&nosqlSt, nil).AnyTimes()\n\tshardedNosqlStoreMock.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\n\tstore := &nosqlHistoryStore{\n\t\tshardedNosqlStore: shardedNosqlStoreMock,\n\t}\n\n\treturn store, dbMock, shardedNosqlStoreMock\n}\n\nfunc TestAppendHistoryNodes_ErrorIfAppendAbove(t *testing.T) {\n\tstore, _, _ := setUpMocks(t)\n\n\trequest := validInternalAppendHistoryNodesRequest()\n\n\t// If the nodeID to append is smaller than the last ancestor's end node ID, return an error\n\trequest.NodeID = 3\n\tans := request.BranchInfo.Ancestors\n\tans[len(ans)-1].EndNodeID = 5\n\n\terr := store.AppendHistoryNodes(ctx.Background(), request)\n\n\tvar invalidErr *persistence.InvalidPersistenceRequestError\n\tassert.ErrorAs(t, err, &invalidErr)\n\tassert.ErrorContains(t, err, \"cannot append to ancestors' nodes\")\n}\n\nfunc TestAppendHistoryNodes_NotNewBranch(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\t// Expect to insert the node into the history tree and node, as this is not a new branch, expect treeRow to be nil\n\tdbMock.EXPECT().InsertIntoHistoryTreeAndNode(gomock.Any(), nil, validHistoryNodeRow()).Return(nil).Times(1)\n\n\trequest := validInternalAppendHistoryNodesRequest()\n\terr := store.AppendHistoryNodes(ctx.Background(), request)\n\n\tassert.NoError(t, err)\n}\n\nfunc TestAppendHistoryNodes_NewBranch(t *testing.T) {\n\trequest := validInternalAppendHistoryNodesRequest()\n\trequest.IsNewBranch = true\n\n\tstore, dbMock, _ := setUpMocks(t)\n\n\t// Expect to insert the node into the history tree and node, as this is a new branch expect treeRow to be set\n\tdbMock.EXPECT().InsertIntoHistoryTreeAndNode(gomock.Any(), gomock.Any(), validHistoryNodeRow()).\n\t\tDoAndReturn(func(ctx ctx.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\t\t\t// Assert that the treeRow is as expected\n\t\t\tassert.Equal(t, testShardID, treeRow.ShardID)\n\t\t\tassert.Equal(t, \"TestTreeID\", treeRow.TreeID)\n\t\t\tassert.Equal(t, \"TestBranchID\", treeRow.BranchID)\n\t\t\tassert.Equal(t, request.BranchInfo.Ancestors, treeRow.Ancestors)\n\t\t\tassert.Equal(t, request.Info, treeRow.Info)\n\t\t\tassert.Equal(t, FixedTime, treeRow.CreateTimestamp)\n\n\t\t\treturn nil\n\t\t})\n\n\terr := store.AppendHistoryNodes(ctx.Background(), request)\n\n\tassert.NoError(t, err)\n}\n\nconst (\n\ttestMinNodeID         = 111\n\ttestMaxNodeID         = 222\n\ttestRequestLastNodeID = 333\n\ttestLastTransactionID = 444\n\n\t// These needs to be greater than testRequestLastNodeID\n\ttestRowNodeID1 = int64(334)\n\ttestRowNodeID2 = int64(335)\n\n\t// These needs to be greater than testLastTransactionID\n\ttestRowTxnID1 = int64(445)\n\ttestRowTxnID2 = int64(446)\n)\n\nfunc validInternalReadHistoryBranchRequest() *persistence.InternalReadHistoryBranchRequest {\n\treturn &persistence.InternalReadHistoryBranchRequest{\n\t\tTreeID:            \"TestTreeID\",\n\t\tBranchID:          \"TestBranchID\",\n\t\tMinNodeID:         testMinNodeID,\n\t\tMaxNodeID:         testMaxNodeID,\n\t\tPageSize:          0,\n\t\tNextPageToken:     nil,\n\t\tLastNodeID:        testRequestLastNodeID,\n\t\tLastTransactionID: testLastTransactionID,\n\t\tShardID:           testShardID,\n\t}\n}\n\nfunc expectedHistoryNodeFilter() *nosqlplugin.HistoryNodeFilter {\n\treturn &nosqlplugin.HistoryNodeFilter{\n\t\tShardID:       testShardID,\n\t\tTreeID:        \"TestTreeID\",\n\t\tBranchID:      \"TestBranchID\",\n\t\tMinNodeID:     testMinNodeID,\n\t\tMaxNodeID:     testMaxNodeID,\n\t\tNextPageToken: nil,\n\t\tPageSize:      0,\n\t}\n}\n\nfunc validHistoryNodeRows() []*nosqlplugin.HistoryNodeRow {\n\treturn []*nosqlplugin.HistoryNodeRow{\n\t\t{\n\t\t\tTreeID:       \"TestTreeID\",\n\t\t\tBranchID:     \"TestBranchID\",\n\t\t\tNodeID:       testRowNodeID1,\n\t\t\tTxnID:        common.Ptr(testRowTxnID1),\n\t\t\tData:         []byte(\"TestEvents\"),\n\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\tShardID:      testShardID,\n\t\t},\n\t\t{\n\t\t\tTreeID:       \"TestTreeID\",\n\t\t\tBranchID:     \"TestBranchID\",\n\t\t\tNodeID:       testRowNodeID2,\n\t\t\tTxnID:        common.Ptr(testRowTxnID2),\n\t\t\tData:         []byte(\"TestEvents2\"),\n\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\tShardID:      testShardID,\n\t\t},\n\t}\n}\n\nfunc TestReadHistoryBranch(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := validInternalReadHistoryBranchRequest()\n\trows := validHistoryNodeRows()\n\t// Append a rowID with a lower transaction ID to test that it is discarded\n\tbadRow := *rows[0]\n\tbadRow.TxnID = common.Ptr[int64](testLastTransactionID - 1)\n\trows = append(rows, &badRow)\n\n\t// Expect to read the history branch\n\tdbMock.EXPECT().SelectFromHistoryNode(gomock.Any(), expectedHistoryNodeFilter()).\n\t\tReturn(rows, nil, nil).Times(1)\n\n\tresp, err := store.ReadHistoryBranch(ctx.Background(), request)\n\trequire.NoError(t, err)\n\n\t// Asset that we got the history for all the nodes\n\tassert.Equal(t, 2, len(resp.History))\n\tassert.Equal(t, rows[0].Data, resp.History[0].Data)\n\tassert.Equal(t, rows[1].Data, resp.History[1].Data)\n\tassert.Equal(t, constants.EncodingTypeThriftRW, resp.History[0].Encoding)\n\tassert.Equal(t, constants.EncodingTypeThriftRW, resp.History[1].Encoding)\n\n\tassert.Nil(t, resp.NextPageToken)\n\n\t// Assert that these ids corresponds to the last node and transaction id\n\tassert.Equal(t, testRowNodeID2, resp.LastNodeID)\n\tassert.Equal(t, testRowTxnID2, resp.LastTransactionID)\n}\n\nfunc TestReadHistoryBranch_ErrorIfSelectFromHistoryNodeErrors(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := validInternalReadHistoryBranchRequest()\n\n\ttestError := fmt.Errorf(\"test error\")\n\n\tdbMock.EXPECT().SelectFromHistoryNode(gomock.Any(), expectedHistoryNodeFilter()).\n\t\tReturn(nil, nil, testError).Times(1)\n\tdbMock.EXPECT().IsNotFoundError(testError).Return(true).Times(1)\n\n\t_, err := store.ReadHistoryBranch(ctx.Background(), request)\n\n\tvar notExistsErr *types.EntityNotExistsError\n\tassert.ErrorAs(t, err, &notExistsErr)\n\tassert.ErrorContains(t, err, \"SelectFromHistoryNode\")\n\tassert.ErrorContains(t, err, \"test error\")\n}\n\nfunc TestReadHistoryBranch_ErrorIfDecreasingNodeID(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := validInternalReadHistoryBranchRequest()\n\trows := validHistoryNodeRows()\n\t// Set the first row to have a node id that is less than the last node id in the request\n\trows[0].NodeID = 1\n\n\t// Expect to read the history branch\n\tdbMock.EXPECT().SelectFromHistoryNode(gomock.Any(), expectedHistoryNodeFilter()).\n\t\tReturn(rows, nil, nil).Times(1)\n\n\t_, err := store.ReadHistoryBranch(ctx.Background(), request)\n\n\tvar dataError *types.InternalDataInconsistencyError\n\tassert.ErrorAs(t, err, &dataError)\n\tassert.ErrorContains(t, err, \"corrupted data, nodeID cannot decrease\")\n}\n\nfunc TestReadHistoryBranch_ErrorIfSameNodeID(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := validInternalReadHistoryBranchRequest()\n\trows := validHistoryNodeRows()\n\t// Set the second row to have the same node id as the first row\n\trows[1].NodeID = rows[0].NodeID\n\n\t// Expect to read the history branch\n\tdbMock.EXPECT().SelectFromHistoryNode(gomock.Any(), expectedHistoryNodeFilter()).\n\t\tReturn(rows, nil, nil).Times(1)\n\n\t_, err := store.ReadHistoryBranch(ctx.Background(), request)\n\n\tvar dataError *types.InternalDataInconsistencyError\n\tassert.ErrorAs(t, err, &dataError)\n\tassert.ErrorContains(t, err, \"corrupted data, same nodeID must have smaller txnID\")\n}\n\nfunc validInternalForkHistoryBranchRequest(forkNodeID int64) *persistence.InternalForkHistoryBranchRequest {\n\treturn &persistence.InternalForkHistoryBranchRequest{\n\t\tForkBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"TestBranchID\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   5,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 6,\n\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tForkNodeID:       forkNodeID,\n\t\tNewBranchID:      \"TestNewBranchID\",\n\t\tInfo:             \"TestInfo\",\n\t\tShardID:          testShardID,\n\t\tCurrentTimeStamp: FixedTime,\n\t}\n}\n\nfunc expectedInternalForkHistoryBranchResponse() *persistence.InternalForkHistoryBranchResponse {\n\treturn &persistence.InternalForkHistoryBranchResponse{\n\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"TestNewBranchID\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   5,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 6,\n\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc expectedTreeRow() *nosqlplugin.HistoryTreeRow {\n\treturn &nosqlplugin.HistoryTreeRow{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: \"TestNewBranchID\",\n\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t{\n\t\t\t\tBranchID:  \"TestAncestorBranchID\",\n\t\t\t\tEndNodeID: 5,\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchID:  \"TestAncestorBranchID\",\n\t\t\t\tEndNodeID: 10,\n\t\t\t},\n\t\t},\n\t\tCreateTimestamp: FixedTime,\n\t\tInfo:            \"TestInfo\",\n\t}\n}\n\nfunc treeRowEqual(t *testing.T, expected, actual *nosqlplugin.HistoryTreeRow) {\n\tt.Helper()\n\n\tassert.Equal(t, expected.ShardID, actual.ShardID)\n\tassert.Equal(t, expected.TreeID, actual.TreeID)\n\tassert.Equal(t, expected.BranchID, actual.BranchID)\n\tassert.Equal(t, expected.Ancestors, actual.Ancestors)\n\tassert.Equal(t, expected.Info, actual.Info)\n\tassert.Equal(t, FixedTime, actual.CreateTimestamp)\n}\n\nfunc TestForkHistoryBranch_NotAllAncestors(t *testing.T) {\n\trequest := validInternalForkHistoryBranchRequest(8)\n\texpecedResp := expectedInternalForkHistoryBranchResponse()\n\texpTreeRow := expectedTreeRow()\n\n\t// The new branch ends at the fork node\n\texpecedResp.NewBranchInfo.Ancestors[1].EndNodeID = 8\n\texpTreeRow.Ancestors[1].EndNodeID = 8\n\n\tstore, dbMock, _ := setUpMocks(t)\n\n\t// Expect to insert the new branch into the history tree\n\tdbMock.EXPECT().InsertIntoHistoryTreeAndNode(gomock.Any(), gomock.Any(), nil).\n\t\tDoAndReturn(func(ctx ctx.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\t\t\t// Assert that the treeRow is as expected\n\t\t\ttreeRowEqual(t, expTreeRow, treeRow)\n\t\t\treturn nil\n\t\t}).Times(1)\n\n\tresp, err := store.ForkHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expecedResp, resp)\n}\n\nfunc TestForkHistoryBranch_AllAncestors(t *testing.T) {\n\trequest := validInternalForkHistoryBranchRequest(14)\n\texpecedResp := expectedInternalForkHistoryBranchResponse()\n\texpTreeRow := expectedTreeRow()\n\n\t// The new branch inherits the ancestors from the fork node, and adds a new ancestor\n\texpecedResp.NewBranchInfo.Ancestors = append(expecedResp.NewBranchInfo.Ancestors, &types.HistoryBranchRange{\n\t\tBranchID:    \"TestBranchID\",\n\t\tBeginNodeID: 10, // The last in the fork node's ancestors\n\t\tEndNodeID:   14, // The fork node\n\t})\n\texpTreeRow.Ancestors = append(expTreeRow.Ancestors, &types.HistoryBranchRange{\n\t\tBranchID:  \"TestBranchID\",\n\t\tEndNodeID: 14,\n\t})\n\n\tstore, dbMock, _ := setUpMocks(t)\n\n\t// Expect to insert the new branch into the history tree\n\tdbMock.EXPECT().InsertIntoHistoryTreeAndNode(gomock.Any(), gomock.Any(), nil).\n\t\tDoAndReturn(func(ctx ctx.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\t\t\t// Assert that the treeRow is as expected\n\t\t\ttreeRowEqual(t, expTreeRow, treeRow)\n\t\t\treturn nil\n\t\t}).Times(1)\n\n\tresp, err := store.ForkHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expecedResp, resp)\n}\n\nfunc getValidInternalDeleteHistoryBranchRequest() *persistence.InternalDeleteHistoryBranchRequest {\n\treturn &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"TestBranchID\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   5,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"TestAncestorBranchID\",\n\t\t\t\t\tBeginNodeID: 6,\n\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tShardID: testShardID,\n\t}\n}\n\nfunc TestDeleteHistoryBranch_unusedBranch(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := getValidInternalDeleteHistoryBranchRequest()\n\n\texpecedTreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: common.Ptr(\"TestBranchID\"),\n\t}\n\n\t// Delete in reverse order, add 0 in the end\n\texpectedNodeFilters := []*nosqlplugin.HistoryNodeFilter{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"TestBranchID\",\n\t\t\tMinNodeID: 10,\n\t\t},\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"TestAncestorBranchID\",\n\t\t\tMinNodeID: 6,\n\t\t},\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"TestAncestorBranchID\",\n\t\t\tMinNodeID: 0,\n\t\t},\n\t}\n\n\t// Expect to delete the history branch\n\tdbMock.EXPECT().DeleteFromHistoryTreeAndNode(gomock.Any(), expecedTreeFilter, expectedNodeFilters).\n\t\tReturn(nil).Times(1)\n\tdbMock.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).\n\t\tReturn(nil, nil).Times(1)\n\n\terr := store.DeleteHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n}\n\n//\t In this base-case scenario, a workflow's been forked a few times, and the\n//\t parent / ancestor workflow (branch A) is being removed.\n//\n//\t  Trees\n//\t  ┌───────────────────┐    ┌───────────────────┐   ┌────────────────────┐\n//\t  │  Tree: 123        │    │  Tree: 123        │   │ Tree: 123          │\n//\t  │  Branch: C        │    │  Branch: A        │   │ Branch: B          │\n//\t  │  Ancestors:       │    │                   │   │ Ancestors:         │\n//\t  │    Branch: A      │    │                   │   │  Branch A          │\n//\t  │    EndNode: 3     │    │                   │   │  EndNode: 2        │\n//\t  └───────────────────┘    └───────────────────┘   └────────────────────┘\n//\t  Nodes\n//\t                           ┌───────────────────┐\n//\t                           │     Tree: 123     │\n//\t                           │     Branch: A     │\n//\t                           │     Node: 1       │\n//\t                           └─────────┬─────────┘\n//\t                                     │\n//\t                                     ┼───────────────────────┐\n//\t                           ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t                           │     Tree: 123     │   │     Tree: 123     │\n//\t                           │     Branch: A     │   │     Branch: B     │\n//\t                           │     Node: 2       │   │     Node: 2       │\n//\t                           └─────────┬─────────┘   └─────────┬─────────┘\n//\t            ┌────────────────────────┤                       │\n//\t            │                        │                       │\n//\t  ┌─────────▼─────────┐    ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t  │     Tree: 123     │    │     Tree: 123     │   │     Tree: 123     │\n//\t  │     Branch: C     │    │     Branch: A     │   │     Branch: B     │\n//\t  │     Node: 3       │    │     Node: 3       │   │     Node: 3       │\n//\t  └─────────┬─────────┘    └─────────┬─────────┘   └─────────┬─────────┘\n//\t            │                        │                       │\n//\t            │                        │                       │\n//\t  ┌─────────▼─────────┐    ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t  │     Tree: 123     │    │     Tree: 123     │   │     Tree: 123     │\n//\t  │     Branch: C     │    │     Branch: A     │   │     Branch: B     │\n//\t  │     Node: 4       │    │     Node: 4       │   │     Node: 4       │\n//\t  └───────────────────┘    └───────────────────┘   └───────────────────┘\n//\n//\t The Expected behaviour is that the tree and unused nodes are trimmed off\n//\t but the referenced nodes from other branches are kept so those workflows\n//\t aren't broken.\n//\n//\t           Trees\n//\n//\t           ┌───────────────────┐    ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐   ┌────────────────────┐\n//\t           │  Tree: 123        │                            │ Tree: 123          │\n//\t           │  Branch: C        │    │  <deleted>        │   │ Branch: B          │\n//\t           │  Ancestors:       │                            │ Ancestors:         │\n//\t           │    Branch: A      │    │                   │   │  Branch A          │\n//\t           │    EndNode: 3     │                            │  EndNode: 2        │\n//\t           └───────────────────┘    └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘   └────────────────────┘\n//\n//\t           Nodes\n//\t                                    ┌───────────────────┐\n//\t                                    │     Tree: 123     │\n//\t                                    │     Branch: A     │\n//\t                                    │     Node: 1       │\n//\t                                    └─────────┬─────────┘\n//\t                                              │\n//\t                                              ┼───────────────────────┐\n//\t                                    ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t                                    │     Tree: 123     │   │     Tree: 123     │\n//\t                                    │     Branch: A     │   │     Branch: B     │\n//\t                                    │     Node: 2       │   │     Node: 2       │\n//\t                                    └─────────┬─────────┘   └─────────┬─────────┘\n//\t                     ┌────────────────────────┤                       │\n//\t                     │                        │                       │\n//\t           ┌─────────▼─────────┐    ┌─ ─ ─ ─ ─▼ ─ ─ ─ ─ ┐   ┌─────────▼─────────┐\n//\t           │     Tree: 123     │                            │     Tree: 123     │\n//\t           │     Branch: C     │    │ <deleted>         │   │     Branch: B     │\n//\t           │     Node: 3       │                            │     Node: 3       │\n//\t           └─────────┬─────────┘    └─ ─ ─ ─ ─┐ ─ ─ ─ ─ ┘   └─────────┬─────────┘\n//\t                     │                        │                       │\n//\t                     │                        │                       │\n//\t           ┌─────────▼─────────┐    ┌─ ─ ─ ─ ─▼ ─ ─ ─ ─ ┐   ┌─────────▼─────────┐\n//\t           │     Tree: 123     │                            │     Tree: 123     │\n//\t           │     Branch: C     │    │ <deleted>         │   │     Branch: B     │\n//\t           │     Node: 4       │                            │     Node: 4       │\n//\t           └───────────────────┘    └─ ─ ─ ─ ── ─ ─ ─ ─ ┘   └───────────────────┘\n\nfunc TestDeleteHistoryBranchWithAFewBranches_baseCase(t *testing.T) {\n\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"A\",\n\t\t},\n\t\tShardID: testShardID,\n\t}\n\n\texpectedTreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: common.Ptr(\"A\"),\n\t}\n\n\t// Delete in reverse order, add 0 in the end\n\texpectedNodeFilters := []*nosqlplugin.HistoryNodeFilter{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"A\",\n\t\t\tMinNodeID: 3,\n\t\t},\n\t}\n\n\tdbMock.EXPECT().DeleteFromHistoryTreeAndNode(gomock.Any(), expectedTreeFilter, expectedNodeFilters).\n\t\tReturn(nil).Times(1)\n\n\thistoryTree := []*nosqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"A\",\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"C\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tdbMock.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).\n\t\tReturn(historyTree, nil).Times(1)\n\n\terr := store.DeleteHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n\n}\n\n// In this scenario, Branch A has already been deleted, but is referenced by\n// a couple of other branches.\n//\n// In this scenario Branch B is being deleted.\n//\n// Given the following starting state:\n//\n//\tTrees\n//\n//\t ┌───────────────────┐    ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐   ┌────────────────────┐\n//\t │  Tree: 123        │                            │ Tree: 123          │\n//\t │  Branch: C        │    │  <branch A is     │   │ Branch: B          │\n//\t │  Ancestors:       │        deleted>            │ Ancestors:         │\n//\t │    Branch: A      │    │                   │   │  Branch A          │\n//\t │    EndNode: 3     │                            │  EndNode: 2        │\n//\t └───────────────────┘    └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘   └────────────────────┘\n//\n//\t  Nodes\n//\n//\t                        ┌───────────────────┐\n//\t                        │     Tree: 123     │\n//\t                        │     Branch: A     │\n//\t                        │     Node: 1       │\n//\t                        └─────────┬─────────┘\n//\t                                  │\n//\t                                  ┼─────────────────────────┐\n//\t                        ┌─────────▼─────────┐     ┌─────────▼─────────┐\n//\t                        │     Tree: 123     │     │     Tree: 123     │\n//\t                        │     Branch: A     │     │     Branch: B     │\n//\t                        │     Node: 2       │     │     Node: 2       │\n//\t                        └─────────┬─────────┘     └─────────┬─────────┘\n//\t         ┌────────────────────────┘                         │\n//\t         │                                                  │\n//\t ┌─────────▼─────────┐                            ┌─────────▼─────────┐\n//\t │     Tree: 123     │                            │     Tree: 123     │\n//\t │     Branch: C     │                            │     Branch: B     │\n//\t │     Node: 3       │                            │     Node: 3       │\n//\t └─────────┬─────────┘                            └─────────┬─────────┘\n//\t           │                                                │\n//\t           │                                                │\n//\t ┌─────────▼─────────┐                            ┌─────────▼─────────┐\n//\t │     Tree: 123     │                            │     Tree: 123     │\n//\t │     Branch: C     │                            │     Branch: B     │\n//\t │     Node: 4       │                            │     Node: 4       │\n//\t └───────────────────┘                            └───────────────────┘\n//\n// The following is expected: It preserves the remaining nodes for any dependent\n// branches.\n//\n//\tTrees\n//\n//\t┌───────────────────┐    ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐     ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐\n//\t│  Tree: 123        │\n//\t│  Branch: C        │    │  <branch A is     │     │  <branch B is     │\n//\t│  Ancestors:       │        deleted>                  deleted>\n//\t│    Branch: A      │    │                   │     │                   │\n//\t│    EndNode: 3     │\n//\t└───────────────────┘    └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘     └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘\n//\n//\tNodes\n//\t                         ┌───────────────────┐\n//\t                         │     Tree: 123     │\n//\t                         │     Branch: A     │\n//\t                         │     Node: 1       │\n//\t                         └─────────┬─────────┘\n//\t                                   │\n//\t                                   ┼\n//\t                         ┌─────────▼─────────┐\n//\t                         │     Tree: 123     │\n//\t                         │     Branch: A     │\n//\t                         │     Node: 2       │\n//\t                         └─────────┬─────────┘\n//\t          ┌────────────────────────┘\n//\t          │\n//\t┌─────────▼─────────┐\n//\t│     Tree: 123     │\n//\t│     Branch: C     │\n//\t│     Node: 3       │\n//\t└─────────┬─────────┘\n//\t          │\n//\t          │\n//\t┌─────────▼─────────┐\n//\t│     Tree: 123     │\n//\t│     Branch: C     │\n//\t│     Node: 4       │\n//\t└───────────────────┘\nfunc TestDeleteHistoryBranch_DeletedAncestor(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tShardID: testShardID,\n\t}\n\n\texpectedTreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: common.Ptr(\"B\"),\n\t}\n\n\texpectedNodeFilters := []*nosqlplugin.HistoryNodeFilter{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"B\",\n\t\t\tMinNodeID: 2,\n\t\t},\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"A\",\n\t\t\tMinNodeID: 3,\n\t\t},\n\t}\n\n\tdbMock.EXPECT().DeleteFromHistoryTreeAndNode(gomock.Any(), expectedTreeFilter, expectedNodeFilters).\n\t\tReturn(nil).Times(1)\n\n\thistoryTree := []*nosqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"C\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t// notably A does not exist\n\t}\n\n\tdbMock.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).\n\t\tReturn(historyTree, nil).Times(1)\n\n\terr := store.DeleteHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n}\n\n// In this scenario, something like the following has happened:\n// There was a normal, original workflow, which was then branched (perhaps by a reset).\n// By the time the workflow is being cleaned up, the ancestor branch has been deleted already,\n// and does not exist in the history_tree table as a valid branch.\n//\n// In this scenario, branch B is being deleted. Notably, Branch B is the *last*\n// valid branch for this tree, whereas the ancestor branches were removed earlier.\n//\n//\t Trees\n//\n//\t ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐    ┌───────────────────┐\n//\t                          │  Tree: 123        │\n//\t │  <branch A is     │    │  Branch: B        │\n//\t     deleted>             │  Ancestors:       │\n//\t │                   │    │    Branch: A      │\n//\t                          │    EndNode: 3     │\n//\t └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘    └───────────────────┘\n//\n//\n//\t Nodes\n//\n//\t ┌───────────────────┐\n//\t │     Tree: 123     │\n//\t │     Branch: A     │\n//\t │     Node: 1       │\n//\t └─────────┬─────────┘\n//\t           │\n//\t ┌─────────┼─────────┐\n//\t │     Tree: 123     │\n//\t │     Branch: A     │\n//\t │     Node: 2       │\n//\t └─────────┬─────────┘\n//\t           ┼────────────────────────┐\n//\t                                    │\n//\t                                    │\n//\t                          ┌─────────▼─────────┐\n//\t                          │     Tree: 123     │\n//\t                          │     Branch: B     │\n//\t                          │     Node: 3       │\n//\t                          └─────────┬─────────┘\n//\t                                    │\n//\t                                    │\n//\t                          ┌─────────▼─────────┐\n//\t                          │     Tree: 123     │\n//\t                          │     Branch: B     │\n//\t                          │     Node: 4       │\n//\t                          └───────────────────┘\n//\n//\tThe expected behaviour, is that the child/branched workflow needs to clean up both its own history nodes\n//\tbut *all* of the parent's remaining and now unreferenced history nodes. They're otherwise unreachable\n//\tand will be just history_node table garbage which forever grows.\n//\n//\t Trees\n//\n//\t ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐    ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐\n//\t                          │                   │\n//\t │  <branch A is     │\n//\t     deleted>             │   deleted>        │\n//\t │                   │\n//\t                          │                   │\n//\t └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘    └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘\n//\n//\n//\t Nodes\n//\n//\t ┌ ─ ─ ─ ─ ─ ─ ─ ─ ─ ┐\n//\t │                   │\n//\t │  <deleted>        │\n//\t │                   │\n//\t └ ─ ─ ─ ─ ┌ ─ ─ ─ ─ ┘\n//\t           │\n//\t ┌ ─ ─ ─ ─ ▼ ─ ─ ─ ─ ┐\n//\t │                   │\n//\t │  <deleted>        │\n//\t │                   │\n//\t └ ─ ─ ─ ─ ┌ ─ ─ ─ ─ ┘\n//\t           │\n//\t           ┼ ─ ─ ─ ─ ── ─ ── ─ ─ ── ┐\n//\t           │                        │\n//\t ┌ ─ ─ ─ ─ ▼ ─ ─ ─ ─ ┐    ┌─ ─ ─ ── ▼─ ─ ─ ─ ─┐\n//\t │                   │    │                   │\n//\t │  <deleted>        │    │ <deleted>         │\n//\t │                   │    │                   │\n//\t └ ─ ─ ─ ─ ┌ ─ ─ ─ ─ ┘    └─ ─ ─ ── ┌─ ─ ─ ─ ─┘\n//\t           │                        │\n//\t           │                        │\n//\t ┌ ─ ─ ─ ─ ▼ ─ ─ ─ ─ ┐    ┌─ ─ ─ ── ▼─ ─ ─ ─ ─┐\n//\t │                   │    │                   │\n//\t │  <deleted>        │    │ <deleted>         │\n//\t │                   │    │                   │\n//\t └ ─ ─ ─ ─ ─ ─ ─ ─ ─ ┘    └─ ─ ─ ── ── ─ ─ ─ ─┘\nfunc TestDeleteHistoryBranch_usedBranchWithGarbageFullyCleanedUp(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:  \"A\",\n\t\t\t\t\tEndNodeID: 3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tShardID: testShardID,\n\t}\n\n\texpectedTreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: common.Ptr(\"B\"),\n\t}\n\n\texpectedNodeFilters := []*nosqlplugin.HistoryNodeFilter{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"B\",\n\t\t\tMinNodeID: 3,\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"A\",\n\t\t},\n\t}\n\n\tdbMock.EXPECT().DeleteFromHistoryTreeAndNode(gomock.Any(), expectedTreeFilter, expectedNodeFilters).\n\t\tReturn(nil).Times(1)\n\n\thistoryTree := []*nosqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t// notably A does not exist\n\t}\n\n\tdbMock.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).\n\t\tReturn(historyTree, nil).Times(1)\n\n\terr := store.DeleteHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n}\n\n// In this scenario, there's a few branches of a normal workflow. In this example, both history_trees exist\n// and the original branch has not yet been deleted, and (for whatever reason, the second branch is being removed\n// before the original/parent (it's not super obvious this might happen, but it's forseeable with deletion jitter\n// or maybe failover scenarios where this might happen).\n//\n// In this scenario, branch B is being deleted.\n//\n//\tTrees\n//\n//\t┌───────────────────┐    ┌───────────────────┐   ┌────────────────────┐\n//\t│  Tree: 123        │    │  Tree: 123        │   │ Tree: 123          │\n//\t│  Branch: C        │    │  Branch: A        │   │ Branch: B          │\n//\t│  Ancestors:       │    │                   │   │ Ancestors:         │\n//\t│    Branch: A      │    │                   │   │  Branch A          │\n//\t│    Endnode: 3     │    │                   │   │  EndNode: 2        │\n//\t└───────────────────┘    └───────────────────┘   └────────────────────┘\n//\n//\tNodes\n//\n//\t                         ┌───────────────────┐\n//\t                         │     Tree: 123     │\n//\t                         │     Branch: A     │\n//\t                         │     Node: 1       │\n//\t                         └─────────┬─────────┘\n//\t                                   │\n//\t                                   ┼───────────────────────┐\n//\t                         ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t                         │     Tree: 123     │   │     Tree: 123     │\n//\t                         │     Branch: A     │   │     Branch: B     │\n//\t                         │     Node: 2       │   │     Node: 2       │\n//\t                         └─────────┬─────────┘   └─────────┬─────────┘\n//\t          ┌────────────────────────┤                       │\n//\t          │                        │                       │\n//\t┌─────────▼─────────┐    ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t│     Tree: 123     │    │     Tree: 123     │   │     Tree: 123     │\n//\t│     Branch: C     │    │     Branch: A     │   │     Branch: B     │\n//\t│     Node: 3       │    │     Node: 3       │   │     Node: 3       │\n//\t└─────────┬─────────┘    └─────────┬─────────┘   └─────────┬─────────┘\n//\t          │                        │                       │\n//\t          │                        │                       │\n//\t┌─────────▼─────────┐    ┌─────────▼─────────┐   ┌─────────▼─────────┐\n//\t│     Tree: 123     │    │     Tree: 123     │   │     Tree: 123     │\n//\t│     Branch: C     │    │     Branch: A     │   │     Branch: B     │\n//\t│     Node: 4       │    │     Node: 4       │   │     Node: 4       │\n//\t└───────────────────┘    └───────────────────┘   └───────────────────┘\n//\n// The expected behaviour is that the child/second branch should only clean up it's history nodes, but\n// leave the parents alone, so as to not break the parent.\n//\n//\tTrees\n//\n//\t┌───────────────────┐     ┌───────────────────┐     ┌─ ─ ─ ─ ─ ─ ─ ─ ─ ─┐\n//\t│  Tree: 123        │     │  Tree: 123        │\n//\t│  Branch: C        │     │  Branch: A        │     │  <branch B is     │\n//\t│  Ancestors:       │     │                   │         deleted>\n//\t│    Branch: A      │     │                   │     │                   │\n//\t│    EndNode: 2     │     │                   │\n//\t└───────────────────┘     └───────────────────┘     └─ ─ ─ ─ ─ ─ ─ ─ ─ ─┘\n//\n//\n//\tNodes\n//\n//\t                          ┌───────────────────┐\n//\t                          │     Tree: 123     │\n//\t                          │     Branch: A     │\n//\t                          │     Node: 1       │\n//\t                          └─────────┬─────────┘\n//\t                                    │\n//\t                                    ┼\n//\t                          ┌─────────▼─────────┐\n//\t                          │     Tree: 123     │\n//\t                          │     Branch: A     │\n//\t                          │     Node: 2       │\n//\t                          └─────────┬─────────┘\n//\t          ┌─────────────────────────┤\n//\t          │                         │\n//\t┌─────────▼─────────┐     ┌─────────▼─────────┐\n//\t│     Tree: 123     │     │     Tree: 123     │\n//\t│     Branch: C     │     │     Branch: A     │\n//\t│     Node: 3       │     │     Node: 3       │\n//\t└─────────┬─────────┘     └─────────┬─────────┘\n//\t          │                         │\n//\t          │                         │\n//\t┌─────────▼─────────┐     ┌─────────▼─────────┐\n//\t│     Tree: 123     │     │     Tree: 123     │\n//\t│     Branch: C     │     │     Branch: A     │\n//\t│     Node: 4       │     │     Node: 4       │\n//\t└───────────────────┘     └───────────────────┘\nfunc TestDeleteHistoryBranch_withAnAncestorBranchWhichIsStillInUse(t *testing.T) {\n\tstore, dbMock, _ := setUpMocks(t)\n\n\trequest := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:  \"A\",\n\t\t\t\t\tEndNodeID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tShardID: testShardID,\n\t}\n\n\texpectedTreeFilter := &nosqlplugin.HistoryTreeFilter{\n\t\tShardID:  testShardID,\n\t\tTreeID:   \"TestTreeID\",\n\t\tBranchID: common.Ptr(\"B\"),\n\t}\n\n\texpectedNodeFilters := []*nosqlplugin.HistoryNodeFilter{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"B\",\n\t\t\tMinNodeID: 2,\n\t\t},\n\t\t// we do not delete any of the ancestor, it's still valid\n\t}\n\n\t// Expect to delete the history branch\n\tdbMock.EXPECT().DeleteFromHistoryTreeAndNode(gomock.Any(), expectedTreeFilter, expectedNodeFilters).\n\t\tReturn(nil).Times(1)\n\n\thistoryTree := []*nosqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:   testShardID,\n\t\t\tTreeID:    \"TestTreeID\",\n\t\t\tBranchID:  \"A\",\n\t\t\tAncestors: []*types.HistoryBranchRange{},\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"B\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:  \"A\",\n\t\t\t\t\tEndNodeID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tShardID:  testShardID,\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"C\",\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID:    \"A\",\n\t\t\t\t\tBeginNodeID: 0,\n\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tdbMock.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).\n\t\tReturn(historyTree, nil).Times(1)\n\n\terr := store.DeleteHistoryBranch(ctx.Background(), request)\n\tassert.NoError(t, err)\n}\n\nfunc TestGetAllHistoryTreeBranches(t *testing.T) {\n\trequest := &persistence.GetAllHistoryTreeBranchesRequest{\n\t\tNextPageToken: []byte(\"nextPageToken\"),\n\t\tPageSize:      1000,\n\t}\n\n\tstore, dbMock, shardedNoSQLStoreMock := setUpMocks(t)\n\tshardedNoSQLStoreMock.EXPECT().GetShardingPolicy().Return(shardingPolicy{hasShardedHistory: false})\n\tshardedNoSQLStoreMock.EXPECT().GetDefaultShard().Return(nosqlStore{db: dbMock}).Times(1)\n\n\texpTreeRow := expectedTreeRow()\n\n\t// Create another tree row with some different data\n\texpTreeRow2 := expectedTreeRow()\n\texpTreeRow2.TreeID = \"TestTreeID2\"\n\texpTreeRow2.BranchID = \"TestNewBranchID2\"\n\texpTreeRow2.CreateTimestamp = time.Unix(123, 456)\n\texpTreeRow2.Info = \"TestInfo2\"\n\n\texpTreeRows := []*nosqlplugin.HistoryTreeRow{expTreeRow, expTreeRow2}\n\tdbMock.EXPECT().SelectAllHistoryTrees(gomock.Any(), request.NextPageToken, request.PageSize).\n\t\tReturn(expTreeRows, []byte(\"anotherPageToken\"), nil).Times(1)\n\n\texpectedBranches := []persistence.HistoryBranchDetail{\n\t\t{\n\t\t\tTreeID:   \"TestTreeID\",\n\t\t\tBranchID: \"TestNewBranchID\",\n\t\t\tForkTime: expTreeRow.CreateTimestamp,\n\t\t\tInfo:     \"TestInfo\",\n\t\t},\n\t\t{\n\t\t\tTreeID:   \"TestTreeID2\",\n\t\t\tBranchID: \"TestNewBranchID2\",\n\t\t\tForkTime: expTreeRow2.CreateTimestamp,\n\t\t\tInfo:     \"TestInfo2\",\n\t\t},\n\t}\n\n\texpectedResponse := &persistence.GetAllHistoryTreeBranchesResponse{\n\t\tBranches:      expectedBranches,\n\t\tNextPageToken: []byte(\"anotherPageToken\"),\n\t}\n\n\tresp, err := store.GetAllHistoryTreeBranches(ctx.Background(), request)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expectedResponse, resp)\n}\n\nfunc TestGetAllHistoryTreeBranches_dbError(t *testing.T) {\n\trequest := &persistence.GetAllHistoryTreeBranchesRequest{\n\t\tNextPageToken: []byte(\"nextPageToken\"),\n\t\tPageSize:      1000,\n\t}\n\n\tstore, dbMock, shardedNoSQLStoreMock := setUpMocks(t)\n\tshardedNoSQLStoreMock.EXPECT().GetShardingPolicy().Return(shardingPolicy{hasShardedHistory: false})\n\tshardedNoSQLStoreMock.EXPECT().GetDefaultShard().Return(nosqlStore{db: dbMock}).Times(1)\n\n\ttestError := errors.New(\"TEST ERROR\")\n\tdbMock.EXPECT().SelectAllHistoryTrees(gomock.Any(), request.NextPageToken, request.PageSize).\n\t\tReturn(nil, nil, testError).Times(1)\n\tdbMock.EXPECT().IsNotFoundError(testError).Return(true).Times(1)\n\n\t_, err := store.GetAllHistoryTreeBranches(ctx.Background(), request)\n\tassert.Error(t, err)\n\tassert.ErrorContains(t, err, \"TEST ERROR\")\n\tassert.ErrorContains(t, err, \"SelectAllHistoryTrees\")\n}\n\nfunc TestGetAllHistoryTreeBranches_hasShardedPolicy(t *testing.T) {\n\trequest := &persistence.GetAllHistoryTreeBranchesRequest{\n\t\tNextPageToken: []byte(\"nextPageToken\"),\n\t\tPageSize:      1000,\n\t}\n\n\tstore, _, shardedNoSQLStoreMock := setUpMocks(t)\n\tshardedNoSQLStoreMock.EXPECT().GetShardingPolicy().Return(shardingPolicy{hasShardedHistory: true})\n\n\t_, err := store.GetAllHistoryTreeBranches(ctx.Background(), request)\n\tassert.Error(t, err)\n\tvar internalServiceErr *types.InternalServiceError\n\tassert.ErrorAs(t, err, &internalServiceErr)\n\tassert.Equal(t, \"SelectAllHistoryTrees is not supported on sharded nosql db\", internalServiceErr.Message)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_queue_store.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\temptyMessageID = -1\n)\n\ntype nosqlQueueStore struct {\n\tqueueType persistence.QueueType\n\tnosqlStore\n}\n\nfunc newNoSQLQueueStore(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tqueueType persistence.QueueType,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.QueueStore, error) {\n\tshardedStore, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tqueue := &nosqlQueueStore{\n\t\tnosqlStore: shardedStore.GetDefaultShard(),\n\t\tqueueType:  queueType,\n\t}\n\n\treturn queue, nil\n}\n\nfunc (q *nosqlQueueStore) ensureQueueMetadata(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tcurrentTimestamp time.Time,\n) (*nosqlplugin.QueueMetadataRow, error) {\n\tqueueMetadata, err := q.getQueueMetadata(ctx, queueType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif queueMetadata == nil {\n\t\tinsertErr := q.insertInitialQueueMetadataRecord(ctx, queueType, currentTimestamp)\n\t\tif insertErr != nil {\n\t\t\t// If insert fails (likely because another thread just created it),\n\t\t\t// it attempts to read the metadata one more time.\n\t\t\tqueueMetadata, err = q.getQueueMetadata(ctx, queueType)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tif queueMetadata == nil {\n\t\t\t\treturn nil, insertErr\n\t\t\t}\n\t\t} else {\n\t\t\tqueueMetadata, err = q.getQueueMetadata(ctx, queueType)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tif queueMetadata == nil {\n\t\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"queue metadata not found for queue type %v\", queueType),\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\tif queueMetadata.ClusterAckLevels == nil {\n\t\tqueueMetadata.ClusterAckLevels = map[string]int64{}\n\t}\n\treturn queueMetadata, nil\n}\n\n// Warning: This is not a safe concurrent operation in its current state.\n// It's only used for domain replication at the moment, but needs a conditional write guard\n// for concurrent use\nfunc (q *nosqlQueueStore) EnqueueMessage(ctx context.Context, request *persistence.InternalEnqueueMessageRequest) error {\n\tqueueMetadata, err := q.ensureQueueMetadata(ctx, q.queueType, request.CurrentTimeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\tlastMessageID, err := q.getLastMessageID(ctx, q.queueType)\n\tif err != nil {\n\t\treturn err\n\t}\n\t_, err = q.tryEnqueue(ctx, q.queueType, getNextID(queueMetadata.ClusterAckLevels, lastMessageID), request.MessagePayload, request.CurrentTimeStamp)\n\treturn err\n}\n\nfunc (q *nosqlQueueStore) EnqueueMessageToDLQ(ctx context.Context, request *persistence.InternalEnqueueMessageToDLQRequest) error {\n\tif _, err := q.ensureQueueMetadata(ctx, q.getDLQTypeFromQueueType(), request.CurrentTimeStamp); err != nil {\n\t\treturn err\n\t}\n\t// Use negative queue type as the dlq type\n\tlastMessageID, err := q.getLastMessageID(ctx, q.getDLQTypeFromQueueType())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = q.tryEnqueue(ctx, q.getDLQTypeFromQueueType(), lastMessageID+1, request.MessagePayload, request.CurrentTimeStamp)\n\treturn err\n}\n\nfunc (q *nosqlQueueStore) tryEnqueue(ctx context.Context, queueType persistence.QueueType, messageID int64, messagePayload []byte, currentTimeStamp time.Time) (int64, error) {\n\terr := q.db.InsertIntoQueue(ctx, &nosqlplugin.QueueMessageRow{\n\t\tQueueType:        queueType,\n\t\tID:               messageID,\n\t\tPayload:          messagePayload,\n\t\tCurrentTimeStamp: currentTimeStamp,\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*nosqlplugin.ConditionFailure); ok {\n\t\t\treturn emptyMessageID, &persistence.ConditionFailedError{Msg: fmt.Sprintf(\"message ID %v exists in queue\", messageID)}\n\t\t}\n\n\t\treturn emptyMessageID, convertCommonErrors(q.db, fmt.Sprintf(\"EnqueueMessage, Type: %v\", queueType), err)\n\t}\n\n\treturn messageID, nil\n}\n\nfunc (q *nosqlQueueStore) getLastMessageID(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tmsgID, err := q.db.SelectLastEnqueuedMessageID(ctx, queueType)\n\tif err != nil {\n\t\tif q.db.IsNotFoundError(err) {\n\t\t\treturn emptyMessageID, nil\n\t\t}\n\n\t\treturn emptyMessageID, convertCommonErrors(q.db, fmt.Sprintf(\"GetLastMessageID, Type: %v\", queueType), err)\n\t}\n\n\treturn msgID, nil\n}\n\nfunc (q *nosqlQueueStore) ReadMessages(\n\tctx context.Context,\n\trequest *persistence.InternalReadMessagesRequest,\n) (*persistence.InternalReadMessagesResponse, error) {\n\tmessages, err := q.db.SelectMessagesFrom(ctx, q.queueType, request.LastMessageID, request.MaxCount)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"ReadMessages\", err)\n\t}\n\tvar result []*persistence.InternalQueueMessage\n\tfor _, msg := range messages {\n\t\tresult = append(result, &persistence.InternalQueueMessage{\n\t\t\tID:        msg.ID,\n\t\t\tQueueType: q.queueType,\n\t\t\tPayload:   msg.Payload,\n\t\t})\n\t}\n\n\treturn &persistence.InternalReadMessagesResponse{Messages: result}, nil\n}\n\nfunc (q *nosqlQueueStore) ReadMessagesFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalReadMessagesFromDLQRequest,\n) (*persistence.InternalReadMessagesFromDLQResponse, error) {\n\tresponse, err := q.db.SelectMessagesBetween(ctx, nosqlplugin.SelectMessagesBetweenRequest{\n\t\tQueueType:               q.getDLQTypeFromQueueType(),\n\t\tExclusiveBeginMessageID: request.FirstMessageID,\n\t\tInclusiveEndMessageID:   request.LastMessageID,\n\t\tPageSize:                request.PageSize,\n\t\tNextPageToken:           request.PageToken,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"ReadMessagesFromDLQ\", err)\n\t}\n\tvar result []*persistence.InternalQueueMessage\n\tfor _, msg := range response.Rows {\n\t\tresult = append(result, &persistence.InternalQueueMessage{\n\t\t\tID:        msg.ID,\n\t\t\tQueueType: msg.QueueType,\n\t\t\tPayload:   msg.Payload,\n\t\t})\n\t}\n\n\treturn &persistence.InternalReadMessagesFromDLQResponse{\n\t\tMessages:      result,\n\t\tNextPageToken: response.NextPageToken,\n\t}, nil\n}\n\nfunc (q *nosqlQueueStore) DeleteMessagesBefore(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteMessagesBeforeRequest,\n) error {\n\tif err := q.db.DeleteMessagesBefore(ctx, q.queueType, request.MessageID); err != nil {\n\t\treturn convertCommonErrors(q.db, \"DeleteMessagesBefore\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (q *nosqlQueueStore) DeleteMessageFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteMessageFromDLQRequest,\n) error {\n\t// Use negative queue type as the dlq type\n\tif err := q.db.DeleteMessage(ctx, q.getDLQTypeFromQueueType(), request.MessageID); err != nil {\n\t\treturn convertCommonErrors(q.db, \"DeleteMessageFromDLQ\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (q *nosqlQueueStore) RangeDeleteMessagesFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalRangeDeleteMessagesFromDLQRequest,\n) error {\n\t// Use negative queue type as the dlq type\n\tif err := q.db.DeleteMessagesInRange(ctx, q.getDLQTypeFromQueueType(), request.FirstMessageID, request.LastMessageID); err != nil {\n\t\treturn convertCommonErrors(q.db, \"RangeDeleteMessagesFromDLQ\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (q *nosqlQueueStore) insertInitialQueueMetadataRecord(ctx context.Context, queueType persistence.QueueType, currentTimeStamp time.Time) error {\n\tversion := int64(0)\n\n\trow := nosqlplugin.QueueMetadataRow{\n\t\tVersion:          version,\n\t\tQueueType:        queueType,\n\t\tCurrentTimeStamp: currentTimeStamp,\n\t}\n\tif err := q.db.InsertQueueMetadata(ctx, row); err != nil {\n\t\treturn convertCommonErrors(q.db, fmt.Sprintf(\"InsertInitialQueueMetadataRecord, Type: %v\", queueType), err)\n\t}\n\n\treturn nil\n}\n\nfunc (q *nosqlQueueStore) UpdateAckLevel(ctx context.Context, request *persistence.InternalUpdateAckLevelRequest) error {\n\treturn q.updateAckLevel(ctx, request.MessageID, request.ClusterName, q.queueType, request.CurrentTimeStamp)\n}\n\nfunc (q *nosqlQueueStore) GetAckLevels(\n\tctx context.Context,\n\t_ *persistence.InternalGetAckLevelsRequest,\n) (*persistence.InternalGetAckLevelsResponse, error) {\n\tqueueMetadata, err := q.getQueueMetadata(ctx, q.queueType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif queueMetadata == nil {\n\t\treturn &persistence.InternalGetAckLevelsResponse{AckLevels: map[string]int64{}}, nil\n\t}\n\n\treturn &persistence.InternalGetAckLevelsResponse{AckLevels: queueMetadata.ClusterAckLevels}, nil\n}\n\nfunc (q *nosqlQueueStore) UpdateDLQAckLevel(ctx context.Context, request *persistence.InternalUpdateDLQAckLevelRequest) error {\n\treturn q.updateAckLevel(ctx, request.MessageID, request.ClusterName, q.getDLQTypeFromQueueType(), request.CurrentTimeStamp)\n}\n\nfunc (q *nosqlQueueStore) GetDLQAckLevels(\n\tctx context.Context,\n\t_ *persistence.InternalGetDLQAckLevelsRequest,\n) (*persistence.InternalGetDLQAckLevelsResponse, error) {\n\n\t// Use negative queue type as the dlq type\n\tqueueMetadata, err := q.getQueueMetadata(ctx, q.getDLQTypeFromQueueType())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif queueMetadata == nil {\n\t\treturn &persistence.InternalGetDLQAckLevelsResponse{AckLevels: map[string]int64{}}, nil\n\t}\n\n\treturn &persistence.InternalGetDLQAckLevelsResponse{AckLevels: queueMetadata.ClusterAckLevels}, nil\n}\n\nfunc (q *nosqlQueueStore) GetDLQSize(\n\tctx context.Context,\n\t_ *persistence.InternalGetDLQSizeRequest,\n) (*persistence.InternalGetDLQSizeResponse, error) {\n\n\tsize, err := q.db.GetQueueSize(ctx, q.getDLQTypeFromQueueType())\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"GetDLQSize\", err)\n\t}\n\treturn &persistence.InternalGetDLQSizeResponse{Size: size}, nil\n}\n\nfunc (q *nosqlQueueStore) getQueueMetadata(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (*nosqlplugin.QueueMetadataRow, error) {\n\trow, err := q.db.SelectQueueMetadata(ctx, queueType)\n\tif err != nil {\n\t\tif q.db.IsNotFoundError(err) {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn nil, convertCommonErrors(q.db, \"GetQueueMetadata\", err)\n\t}\n\n\treturn row, nil\n}\n\nfunc (q *nosqlQueueStore) updateQueueMetadata(\n\tctx context.Context,\n\tmetadata *nosqlplugin.QueueMetadataRow,\n) error {\n\terr := q.db.UpdateQueueMetadataCas(ctx, *metadata)\n\tif err != nil {\n\t\tif _, ok := err.(*nosqlplugin.ConditionFailure); ok {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"UpdateQueueMetadata operation encounter concurrent write.\",\n\t\t\t}\n\t\t}\n\n\t\treturn convertCommonErrors(q.db, \"UpdateQueueMetadata\", err)\n\t}\n\n\treturn nil\n}\n\n// DLQ type of is the negative of number of the non-DLQ\nfunc (q *nosqlQueueStore) getDLQTypeFromQueueType() persistence.QueueType {\n\treturn -q.queueType\n}\n\nfunc (q *nosqlQueueStore) updateAckLevel(\n\tctx context.Context,\n\tmessageID int64,\n\tclusterName string,\n\tqueueType persistence.QueueType,\n\tcurrentTimestamp time.Time,\n) error {\n\tqueueMetadata, err := q.ensureQueueMetadata(ctx, queueType, currentTimestamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// Ignore possibly delayed message\n\tif ackLevel, ok := queueMetadata.ClusterAckLevels[clusterName]; ok && ackLevel >= messageID {\n\t\treturn nil\n\t}\n\n\tqueueMetadata.ClusterAckLevels[clusterName] = messageID\n\tqueueMetadata.Version++\n\tqueueMetadata.CurrentTimeStamp = currentTimestamp\n\n\t// Use negative queue type as the dlq type\n\terr = q.updateQueueMetadata(ctx, queueMetadata)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// if, for whatever reason, the ack-levels get ahead of the actual messages\n// then ensure the next ID follows\nfunc getNextID(acks map[string]int64, lastMessageID int64) int64 {\n\to := lastMessageID\n\tfor _, v := range acks {\n\t\tif v > o {\n\t\t\to = v\n\t\t}\n\t}\n\treturn o + 1\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_queue_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nconst testQueueType = persistence.DomainReplicationQueueType\nconst testDLQueueType = -testQueueType // type for dlq is always -MainQueueType\n\nvar testPayload = []byte(\"test-message\")\n\ntype queueStoreTestData struct {\n\tmockDB *nosqlplugin.MockDB\n\tqueue  persistence.QueueStore\n}\n\nfunc newQueueStoreTestData(t *testing.T) *queueStoreTestData {\n\tvar testData queueStoreTestData\n\tctrl := gomock.NewController(t)\n\n\ttestData.mockDB = nosqlplugin.NewMockDB(ctrl)\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(ctrl)\n\tmockPlugin.EXPECT().CreateDB(gomock.Any(), gomock.Any(), gomock.Any()).Return(testData.mockDB, nil).AnyTimes()\n\tRegisterPluginForTest(t, \"cassandra\", mockPlugin)\n\treturn &testData\n}\n\nfunc (td *queueStoreTestData) newQueueStore() (persistence.QueueStore, error) {\n\tcfg := getValidShardedNoSQLConfig()\n\treturn newNoSQLQueueStore(cfg, log.NewNoop(), metrics.NewNoopMetricsClient(), testQueueType, nil)\n}\n\nfunc (td *queueStoreTestData) createValidQueueStore(t *testing.T) persistence.QueueStore {\n\tstore, err := td.newQueueStore()\n\trequire.NoError(t, err)\n\trequire.NotNil(t, store)\n\n\treturn store\n}\n\nfunc (td *queueStoreTestData) mockIsNotFoundErrCheck(err error, notfound bool) {\n\ttd.mockDB.EXPECT().IsNotFoundError(err).Return(notfound)\n}\n\nfunc (td *queueStoreTestData) mockErrConversion(err error) {\n\ttd.mockDB.EXPECT().IsNotFoundError(err).Return(false)\n\ttd.mockDB.EXPECT().IsTimeoutError(err).Return(false)\n\ttd.mockDB.EXPECT().IsDBUnavailableError(err).Return(false)\n\ttd.mockDB.EXPECT().IsThrottlingError(err).Return(false)\n}\n\nfunc TestNewNoSQLQueueStore_Succeeds(t *testing.T) {\n\ttd := newQueueStoreTestData(t)\n\ttd.createValidQueueStore(t)\n}\n\nfunc TestEnqueueMessage_Succeeds(t *testing.T) {\n\tconst lastMessageID = int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\t// clusterAckLevels are affecting lastMessageID\n\tclusterAckLevels := map[string]int64{\"cluster1\": lastMessageID + 10, \"cluster2\": lastMessageID + 20}\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: clusterAckLevels}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testQueueType).Return(lastMessageID, nil)\n\ttd.mockDB.EXPECT().InsertIntoQueue(ctx, gomock.Any()).\n\t\tDo(func(_ context.Context, row *nosqlplugin.QueueMessageRow) {\n\t\t\tassert.Equal(\n\t\t\t\tt,\n\t\t\t\t&nosqlplugin.QueueMessageRow{\n\t\t\t\t\tQueueType:        testQueueType,\n\t\t\t\t\tID:               lastMessageID + 20 + 1, // should be the max of cluster AckLevels + 1\n\t\t\t\t\tPayload:          testPayload,\n\t\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t\t},\n\t\t\t\trow,\n\t\t\t)\n\t\t}).Return(nil)\n\n\trequire.NoError(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}))\n}\n\nfunc TestEnqueueMessage_Succeeds_CreateMetadataIfMissing(t *testing.T) {\n\tconst lastMessageID = int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(nil, nil)\n\ttd.mockDB.EXPECT().InsertQueueMetadata(ctx, nosqlplugin.QueueMetadataRow{\n\t\tQueueType:        testQueueType,\n\t\tVersion:          0,\n\t\tCurrentTimeStamp: FixedTime,\n\t}).Return(nil)\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testQueueType).Return(lastMessageID, nil)\n\ttd.mockDB.EXPECT().InsertIntoQueue(ctx, gomock.Any()).\n\t\tDo(func(_ context.Context, row *nosqlplugin.QueueMessageRow) {\n\t\t\tassert.Equal(t, lastMessageID+1, row.ID)\n\t\t\tassert.Equal(t, testQueueType, row.QueueType)\n\t\t\tassert.Equal(t, testPayload, row.Payload)\n\t\t\tassert.Equal(t, FixedTime, row.CurrentTimeStamp)\n\t\t}).Return(nil)\n\n\trequire.NoError(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}))\n}\n\nfunc TestEnqueueMessage_FailsIfCantSelectLastMessageID(t *testing.T) {\n\terrSelect := errors.New(\"failed to select message ID\")\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testQueueType).Return(int64(0), errSelect)\n\ttd.mockIsNotFoundErrCheck(errSelect, false)\n\ttd.mockErrConversion(errSelect)\n\n\tassert.ErrorContains(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errSelect.Error())\n}\n\nfunc TestEnqueueMessage_FailsIfCantSelectQueueMetadata(t *testing.T) {\n\terrSelect := errors.New(\"fail to select queue metadata\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(nil, errSelect)\n\ttd.mockIsNotFoundErrCheck(errSelect, false)\n\ttd.mockErrConversion(errSelect)\n\n\tassert.ErrorContains(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errSelect.Error())\n}\n\nfunc TestEnqueueMessage_FailsIfCantInsertMetadata(t *testing.T) {\n\terrInsert := errors.New(\"insert main-queue metadata failed\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(nil, nil)\n\ttd.mockDB.EXPECT().InsertQueueMetadata(ctx, gomock.Any()).Return(errInsert)\n\ttd.mockErrConversion(errInsert)\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(nil, nil)\n\n\tassert.ErrorContains(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errInsert.Error())\n}\n\nfunc TestEnqueueMessage_FailsIfCantInsertMessageToQueue(t *testing.T) {\n\terrInsert := errors.New(\"fail to insert into queue\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testQueueType).Return(int64(0), nil)\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{}, nil)\n\ttd.mockDB.EXPECT().InsertIntoQueue(ctx, gomock.Any()).Return(errInsert)\n\ttd.mockErrConversion(errInsert)\n\n\tassert.ErrorContains(t, store.EnqueueMessage(ctx, &persistence.InternalEnqueueMessageRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errInsert.Error())\n}\n\nfunc TestEnqueueMessageToDLQ_Succeeds(t *testing.T) {\n\tconst dlqMessageType = -testQueueType\n\tlastMessageID := int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, dlqMessageType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, dlqMessageType).Return(lastMessageID, nil)\n\ttd.mockDB.EXPECT().InsertIntoQueue(ctx, gomock.Any()).\n\t\tDo(func(_ context.Context, row *nosqlplugin.QueueMessageRow) {\n\t\t\tassert.Equal(\n\t\t\t\tt,\n\t\t\t\t&nosqlplugin.QueueMessageRow{\n\t\t\t\t\tQueueType:        dlqMessageType,\n\t\t\t\t\tID:               lastMessageID + 1,\n\t\t\t\t\tPayload:          testPayload,\n\t\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t\t},\n\t\t\t\trow,\n\t\t\t)\n\t\t}).Return(nil)\n\n\trequire.NoError(t, store.EnqueueMessageToDLQ(ctx, &persistence.InternalEnqueueMessageToDLQRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}))\n}\n\nfunc TestEnqueueMessageToDLQ_FailsIfCantSelectLastMessageID(t *testing.T) {\n\terrSelect := errors.New(\"failed to select message ID\")\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testDLQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testDLQueueType).Return(int64(0), errSelect)\n\ttd.mockIsNotFoundErrCheck(errSelect, false)\n\ttd.mockErrConversion(errSelect)\n\n\tassert.ErrorContains(t, store.EnqueueMessageToDLQ(ctx, &persistence.InternalEnqueueMessageToDLQRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errSelect.Error())\n}\n\nfunc TestEnqueueMessageToDLQ_FailsIfCantInsertMessageToQueue(t *testing.T) {\n\terrInsert := errors.New(\"fail to insert into queue\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testDLQueueType).\n\t\tReturn(&nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}, nil)\n\ttd.mockDB.EXPECT().SelectLastEnqueuedMessageID(ctx, testDLQueueType).Return(int64(0), nil)\n\ttd.mockDB.EXPECT().InsertIntoQueue(ctx, gomock.Any()).Return(errInsert)\n\ttd.mockErrConversion(errInsert)\n\n\tassert.ErrorContains(t, store.EnqueueMessageToDLQ(ctx, &persistence.InternalEnqueueMessageToDLQRequest{\n\t\tMessagePayload:   testPayload,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errInsert.Error())\n}\n\nfunc TestReadMessages_Succeeds(t *testing.T) {\n\tconst lastMessageID = int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\tmessages := []*nosqlplugin.QueueMessageRow{\n\t\t{\n\t\t\tQueueType: testQueueType,\n\t\t\tID:        124,\n\t\t\tPayload:   []byte(\"message-124\"),\n\t\t},\n\t\t{\n\t\t\tQueueType: testQueueType,\n\t\t\tID:        125,\n\t\t\tPayload:   []byte(\"message-125\"),\n\t\t},\n\t}\n\ttd.mockDB.EXPECT().SelectMessagesFrom(ctx, testQueueType, lastMessageID, len(messages)).Return(messages, nil)\n\n\tres, err := store.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{\n\t\tLastMessageID: lastMessageID,\n\t\tMaxCount:      len(messages),\n\t})\n\trequire.NoError(t, err)\n\tresMessages := res.Messages\n\n\t// resMessages has different type than messages, should compare explicitly\n\tassert.Len(t, resMessages, len(messages), \"should match the amount of messages returned by SelectMessagesFrom()\")\n\tfor i := range messages {\n\t\tassert.Equal(t, messages[i].QueueType, resMessages[i].QueueType)\n\t\tassert.Equal(t, messages[i].ID, resMessages[i].ID)\n\t\tassert.Equal(t, messages[i].Payload, resMessages[i].Payload)\n\t}\n}\n\nfunc TestReadMessages_FailsIfCantSelectMessages(t *testing.T) {\n\terrSelect := errors.New(\"failed to select messages\")\n\tconst lastMessageID = int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectMessagesFrom(ctx, testQueueType, lastMessageID, 2).Return(nil, errSelect)\n\ttd.mockErrConversion(errSelect)\n\n\tres, err := store.ReadMessages(ctx, &persistence.InternalReadMessagesRequest{\n\t\tLastMessageID: lastMessageID,\n\t\tMaxCount:      2,\n\t})\n\tassert.ErrorContains(t, err, errSelect.Error())\n\tassert.Nil(t, res)\n}\n\nfunc TestReadMessagesFromDLQ_Succeeds(t *testing.T) {\n\tconst firsMessageID = int64(123)\n\tconst lastMessageID = int64(200) // doesn't matter, we will still request max=2\n\tvar pageToken = []byte(\"page-token\")\n\tvar nextPageToken = []byte(\"next-page-token\")\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\tmessages := []nosqlplugin.QueueMessageRow{\n\t\t{\n\t\t\tQueueType: testQueueType,\n\t\t\tID:        124,\n\t\t\tPayload:   []byte(\"message-123\"),\n\t\t},\n\t\t{\n\t\t\tQueueType: testQueueType,\n\t\t\tID:        125,\n\t\t\tPayload:   []byte(\"message-124\"),\n\t\t},\n\t}\n\n\ttd.mockDB.EXPECT().SelectMessagesBetween(ctx, gomock.Any()).Do(\n\t\tfunc(_ context.Context, req nosqlplugin.SelectMessagesBetweenRequest) {\n\t\t\texpectedReq := nosqlplugin.SelectMessagesBetweenRequest{\n\t\t\t\tQueueType:               testDLQueueType,\n\t\t\t\tExclusiveBeginMessageID: firsMessageID,\n\t\t\t\tInclusiveEndMessageID:   lastMessageID,\n\t\t\t\tPageSize:                len(messages),\n\t\t\t\tNextPageToken:           pageToken,\n\t\t\t}\n\t\t\tassert.Equal(t, expectedReq, req)\n\t\t}).Return(&nosqlplugin.SelectMessagesBetweenResponse{NextPageToken: nextPageToken, Rows: messages}, nil)\n\n\tres, err := store.ReadMessagesFromDLQ(ctx, &persistence.InternalReadMessagesFromDLQRequest{\n\t\tFirstMessageID: firsMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t\tPageSize:       len(messages),\n\t\tPageToken:      pageToken,\n\t})\n\trequire.NoError(t, err)\n\tresMessages := res.Messages\n\tresPageToken := res.NextPageToken\n\n\t// resMessages has different type than messages, should compare explicitly\n\tassert.Len(t, resMessages, 2, \"should match the amount of messages returned by SelectMessagesFrom()\")\n\tfor i := range messages {\n\t\tassert.Equal(t, messages[i].QueueType, resMessages[i].QueueType)\n\t\tassert.Equal(t, messages[i].ID, resMessages[i].ID)\n\t\tassert.Equal(t, messages[i].Payload, resMessages[i].Payload)\n\t}\n\n\tassert.Equal(t, nextPageToken, resPageToken)\n}\n\nfunc TestReadMessagesFromDLQ_FailsIfSelectMessagesFails(t *testing.T) {\n\terrSelect := errors.New(\"failed to select messages\")\n\tconst firsMessageID = int64(123)\n\tconst lastMessageID = int64(200) // doesn't matter, we will still request max=2\n\tconst pageSize = 2\n\tvar pageToken = []byte(\"page-token\")\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectMessagesBetween(ctx, gomock.Any()).Return(nil, errSelect)\n\ttd.mockErrConversion(errSelect)\n\n\tres, err := store.ReadMessagesFromDLQ(ctx, &persistence.InternalReadMessagesFromDLQRequest{\n\t\tFirstMessageID: firsMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t\tPageSize:       pageSize,\n\t\tPageToken:      pageToken,\n\t})\n\tassert.ErrorContains(t, err, errSelect.Error())\n\tassert.Nil(t, res)\n}\n\nfunc TestDeleteMessagesBefore_Succeeds(t *testing.T) {\n\tmessageID := int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessagesBefore(ctx, testQueueType, messageID).Return(nil)\n\tassert.NoError(t, store.DeleteMessagesBefore(ctx, &persistence.InternalDeleteMessagesBeforeRequest{\n\t\tMessageID: messageID,\n\t}))\n}\n\nfunc TestDeleteMessagesBefore_FailsIfDeleteFails(t *testing.T) {\n\terrDelete := errors.New(\"failed to delete messages\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessagesBefore(ctx, testQueueType, gomock.Any()).Return(errDelete)\n\ttd.mockErrConversion(errDelete)\n\n\tassert.ErrorContains(t, store.DeleteMessagesBefore(ctx, &persistence.InternalDeleteMessagesBeforeRequest{\n\t\tMessageID: 0,\n\t}), errDelete.Error())\n}\n\nfunc TestDeleteMessageFromDLQ_Succeeds(t *testing.T) {\n\tmessageID := int64(123)\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessage(ctx, testDLQueueType, messageID).Return(nil)\n\tassert.NoError(t, store.DeleteMessageFromDLQ(ctx, &persistence.InternalDeleteMessageFromDLQRequest{\n\t\tMessageID: messageID,\n\t}))\n}\n\nfunc TestDeleteMessageFromDLQ_FailsIfDeleteFails(t *testing.T) {\n\terrDelete := errors.New(\"failed to delete messages\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessage(ctx, testDLQueueType, gomock.Any()).Return(errDelete)\n\ttd.mockErrConversion(errDelete)\n\n\tassert.ErrorContains(t, store.DeleteMessageFromDLQ(ctx, &persistence.InternalDeleteMessageFromDLQRequest{\n\t\tMessageID: 0,\n\t}), errDelete.Error())\n}\n\nfunc TestRangeDeleteMessagesFromDLQ_Succeeds(t *testing.T) {\n\tconst fistMessageID = int64(123)\n\tconst lastMessageID = int64(130)\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessagesInRange(ctx, testDLQueueType, fistMessageID, lastMessageID).Return(nil)\n\tassert.NoError(t, store.RangeDeleteMessagesFromDLQ(ctx, &persistence.InternalRangeDeleteMessagesFromDLQRequest{\n\t\tFirstMessageID: fistMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t}))\n}\n\nfunc TestRangeDeleteMessagesFromDLQ_FailsIfDeleteFails(t *testing.T) {\n\tconst fistMessageID = int64(123)\n\tconst lastMessageID = int64(130)\n\terrDelete := errors.New(\"failed to delete messages\")\n\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().DeleteMessagesInRange(ctx, testDLQueueType, gomock.Any(), gomock.Any()).Return(errDelete)\n\ttd.mockErrConversion(errDelete)\n\n\tassert.ErrorContains(\n\t\tt,\n\t\tstore.RangeDeleteMessagesFromDLQ(ctx, &persistence.InternalRangeDeleteMessagesFromDLQRequest{\n\t\t\tFirstMessageID: fistMessageID,\n\t\t\tLastMessageID:  lastMessageID,\n\t\t}),\n\t\terrDelete.Error(),\n\t)\n}\n\nfunc TestUpdateAckLevel_Succeeds(t *testing.T) {\n\tconst messageID = 123\n\tconst clusterName = \"test-cluster\"\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\tmetadata := nosqlplugin.QueueMetadataRow{\n\t\tQueueType: testQueueType,\n\t\tClusterAckLevels: map[string]int64{\n\t\t\tclusterName:         110,\n\t\t\t\"unrelated-cluster\": 300,\n\t\t},\n\t\tVersion: 3,\n\t}\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(&metadata, nil)\n\ttd.mockDB.EXPECT().UpdateQueueMetadataCas(ctx, gomock.Any()).\n\t\tDo(func(_ context.Context, newMeta nosqlplugin.QueueMetadataRow) {\n\t\t\tassert.Equal(t, int64(4), newMeta.Version, \"version should be incremented\")\n\t\t\tassert.Equal(t, testQueueType, newMeta.QueueType, \"type should remain the same\")\n\n\t\t\texpectedClusterAckLevels := map[string]int64{\n\t\t\t\tclusterName:         messageID, // messageID is greater\n\t\t\t\t\"unrelated-cluster\": 300,\n\t\t\t}\n\t\t\t// only target cluster ack-level should be updated\n\t\t\tassert.Equal(t, expectedClusterAckLevels, newMeta.ClusterAckLevels)\n\t\t}).Return(nil)\n\n\tassert.NoError(t, store.UpdateAckLevel(ctx, &persistence.InternalUpdateAckLevelRequest{\n\t\tMessageID:        messageID,\n\t\tClusterName:      clusterName,\n\t\tCurrentTimeStamp: FixedTime,\n\t}))\n}\n\nfunc TestUpdateAckLevel_FailsIfSelectMetadataFails(t *testing.T) {\n\terrSelect := errors.New(\"select metadata failed\")\n\tconst messageID = 123\n\tconst clusterName = \"test-cluster\"\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(nil, errSelect)\n\ttd.mockIsNotFoundErrCheck(errSelect, false)\n\ttd.mockErrConversion(errSelect)\n\tassert.ErrorContains(t, store.UpdateAckLevel(ctx, &persistence.InternalUpdateAckLevelRequest{\n\t\tMessageID:        messageID,\n\t\tClusterName:      clusterName,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errSelect.Error())\n}\n\nfunc TestUpdateAckLevel_FailsIfUpdateMetadataFails(t *testing.T) {\n\terrUpdate := errors.New(\"update metadata failed\")\n\tconst messageID = 123\n\tconst clusterName = \"test-cluster\"\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\temptyMetadata := &nosqlplugin.QueueMetadataRow{ClusterAckLevels: map[string]int64{}}\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testQueueType).Return(emptyMetadata, nil)\n\ttd.mockDB.EXPECT().UpdateQueueMetadataCas(ctx, gomock.Any()).Return(errUpdate)\n\ttd.mockErrConversion(errUpdate)\n\n\tassert.ErrorContains(t, store.UpdateAckLevel(ctx, &persistence.InternalUpdateAckLevelRequest{\n\t\tMessageID:        messageID,\n\t\tClusterName:      clusterName,\n\t\tCurrentTimeStamp: FixedTime,\n\t}), errUpdate.Error())\n}\n\nfunc TestUpdateDLQAckLevel_Succeeds(t *testing.T) {\n\tconst messageID = 123\n\tconst clusterName = \"test-cluster\"\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\tmetadata := nosqlplugin.QueueMetadataRow{\n\t\tQueueType: testDLQueueType,\n\t\tClusterAckLevels: map[string]int64{\n\t\t\tclusterName:         110,\n\t\t\t\"unrelated-cluster\": 300,\n\t\t},\n\t\tVersion: 3,\n\t}\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testDLQueueType).Return(&metadata, nil)\n\ttd.mockDB.EXPECT().UpdateQueueMetadataCas(ctx, gomock.Any()).\n\t\tDo(func(_ context.Context, newMeta nosqlplugin.QueueMetadataRow) {\n\t\t\tassert.Equal(t, int64(4), newMeta.Version, \"version should be incremented\")\n\t\t\tassert.Equal(t, testDLQueueType, newMeta.QueueType, \"type should remain the same\")\n\n\t\t\texpectedClusterAckLevels := map[string]int64{\n\t\t\t\tclusterName:         messageID, // messageID is greater\n\t\t\t\t\"unrelated-cluster\": 300,\n\t\t\t}\n\t\t\t// only target cluster ack-level should be updated\n\t\t\tassert.Equal(t, expectedClusterAckLevels, newMeta.ClusterAckLevels)\n\t\t}).Return(nil)\n\n\tassert.NoError(t, store.UpdateDLQAckLevel(ctx, &persistence.InternalUpdateDLQAckLevelRequest{\n\t\tMessageID:        messageID,\n\t\tClusterName:      clusterName,\n\t\tCurrentTimeStamp: FixedTime,\n\t}))\n}\n\nfunc TestGetDLQAckLevels_Succeeds(t *testing.T) {\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\texpectedAckLevels := map[string]int64{\n\t\t\"cluster1\": 123,\n\t\t\"cluster2\": 456,\n\t}\n\n\tmetadata := nosqlplugin.QueueMetadataRow{\n\t\tQueueType:        testDLQueueType,\n\t\tClusterAckLevels: expectedAckLevels,\n\t\tVersion:          3,\n\t}\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testDLQueueType).Return(&metadata, nil)\n\n\tresp, err := store.GetDLQAckLevels(ctx, &persistence.InternalGetDLQAckLevelsRequest{})\n\trequire.NoError(t, err)\n\tassert.Equal(t, expectedAckLevels, resp.AckLevels)\n}\n\nfunc TestGetDLQAckLevels_FailsIfSelectMetadataFails(t *testing.T) {\n\terrSelect := errors.New(\"failed to select queue metadata\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().SelectQueueMetadata(ctx, testDLQueueType).Return(nil, errSelect)\n\ttd.mockIsNotFoundErrCheck(errSelect, false)\n\ttd.mockErrConversion(errSelect)\n\n\tresp, err := store.GetDLQAckLevels(ctx, &persistence.InternalGetDLQAckLevelsRequest{})\n\tassert.ErrorContains(t, err, errSelect.Error())\n\tassert.Nil(t, resp)\n}\n\nfunc TestGetDLQSize_Succeeds(t *testing.T) {\n\tconst expectedSize int64 = 12345\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().GetQueueSize(ctx, testDLQueueType).Return(expectedSize, nil)\n\n\tresp, err := store.GetDLQSize(ctx, &persistence.InternalGetDLQSizeRequest{})\n\trequire.NoError(t, err)\n\tassert.Equal(t, expectedSize, resp.Size)\n}\n\nfunc TestGetDLQSize_FailsIfGetQueueSizeFails(t *testing.T) {\n\texpectedErr := errors.New(\"failed to retrieve queue size\")\n\ttd := newQueueStoreTestData(t)\n\tstore := td.createValidQueueStore(t)\n\tctx := context.Background()\n\n\ttd.mockDB.EXPECT().GetQueueSize(ctx, testDLQueueType).Return(int64(0), expectedErr)\n\ttd.mockErrConversion(expectedErr)\n\n\tresp, err := store.GetDLQSize(ctx, &persistence.InternalGetDLQSizeRequest{})\n\tassert.ErrorContains(t, err, expectedErr.Error())\n\tassert.Nil(t, resp)\n}\n\nfunc TestGetNextID(t *testing.T) {\n\ttests := map[string]struct {\n\t\tacks     map[string]int64\n\t\tlastID   int64\n\t\texpected int64\n\t}{\n\t\t\"expected case - last ID is equal to ack-levels\": {\n\t\t\tacks:     map[string]int64{\"a\": 3},\n\t\t\tlastID:   3,\n\t\t\texpected: 4,\n\t\t},\n\t\t\"expected case - last ID is equal to ack-levels haven't caught up\": {\n\t\t\tacks:     map[string]int64{\"a\": 2},\n\t\t\tlastID:   3,\n\t\t\texpected: 4,\n\t\t},\n\t\t\"error case - ack-levels are ahead for some reason\": {\n\t\t\tacks:     map[string]int64{\"a\": 3},\n\t\t\tlastID:   2,\n\t\t\texpected: 4,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, getNextID(td.acks, td.lastID))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_shard_store.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Implements ShardStore\ntype nosqlShardStore struct {\n\tshardedNosqlStore\n\tcurrentClusterName string\n\tparser             serialization.Parser\n}\n\n// newNoSQLShardStore is used to create an instance of ShardStore implementation\nfunc newNoSQLShardStore(\n\tcfg config.ShardedNoSQL,\n\tclusterName string,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n\tparser serialization.Parser,\n) (persistence.ShardStore, error) {\n\ts, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlShardStore{\n\t\tshardedNosqlStore:  s,\n\t\tcurrentClusterName: clusterName,\n\t\tparser:             parser,\n\t}, nil\n}\n\nfunc (sh *nosqlShardStore) CreateShard(\n\tctx context.Context,\n\trequest *persistence.InternalCreateShardRequest,\n) error {\n\tstoreShard, err := sh.GetStoreShardByHistoryShard(request.ShardInfo.ShardID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tshardRow, err := sh.shardInfoToShardsRow(request.ShardInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = storeShard.db.InsertShard(ctx, shardRow)\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.ShardOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn &persistence.ShardAlreadyExistError{\n\t\t\t\tMsg: fmt.Sprintf(\"Shard already exists in executions table.  ShardId: %v, request_range_id: %v, actual_range_id : %v, columns: (%v)\",\n\t\t\t\t\trequest.ShardInfo.ShardID, request.ShardInfo.RangeID, conditionFailure.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(storeShard.db, \"CreateShard\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (sh *nosqlShardStore) GetShard(\n\tctx context.Context,\n\trequest *persistence.InternalGetShardRequest,\n) (*persistence.InternalGetShardResponse, error) {\n\tshardID := request.ShardID\n\tstoreShard, err := sh.GetStoreShardByHistoryShard(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trangeID, shardRow, err := storeShard.db.SelectShard(ctx, shardID, sh.currentClusterName)\n\n\tif err != nil {\n\t\tif storeShard.db.IsNotFoundError(err) {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Shard not found.  ShardId: %v\", shardID),\n\t\t\t}\n\t\t}\n\n\t\treturn nil, convertCommonErrors(storeShard.db, \"GetShard\", err)\n\t}\n\n\tshardInfoRangeID := shardRow.RangeID\n\n\t// check if rangeID column and rangeID field in shard column matches, if not we need to pick the larger\n\t// rangeID.\n\tif shardInfoRangeID > rangeID {\n\t\t// In this case we need to fix the rangeID column before returning the result as:\n\t\t// 1. if we return shardInfoRangeID, then later shard CAS operation will fail\n\t\t// 2. if we still return rangeID, CAS will work but rangeID will move backward which\n\t\t// result in lost tasks, corrupted workflow history, etc.\n\n\t\tsh.GetLogger().Warn(\"Corrupted shard rangeID\", tag.ShardID(shardID), tag.ShardRangeID(shardInfoRangeID), tag.PreviousShardRangeID(rangeID))\n\t\tif err := sh.updateRangeID(ctx, shardID, shardInfoRangeID, rangeID); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\t// return the actual rangeID\n\t\tshardRow.RangeID = rangeID\n\t\t//\n\t\t// If shardInfoRangeID = rangeID, no corruption, so no action needed.\n\t\t//\n\t\t// If shardInfoRangeID < rangeID, we also don't need to do anything here as createShardInfo will ignore\n\t\t// shardInfoRangeID and return rangeID instead. Later when updating the shard, CAS can still succeed\n\t\t// as the value from rangeID columns is returned, shardInfoRangeID will also be updated to the correct value.\n\t}\n\tshardInfo, err := sh.shardsRowToShardInfo(storeShard, shardRow, rangeID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &persistence.InternalGetShardResponse{ShardInfo: shardInfo}, nil\n}\n\nfunc (sh *nosqlShardStore) updateRangeID(\n\tctx context.Context,\n\tshardID int,\n\trangeID int64,\n\tpreviousRangeID int64,\n) error {\n\tstoreShard, err := sh.GetStoreShardByHistoryShard(shardID)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = storeShard.db.UpdateRangeID(ctx, shardID, rangeID, previousRangeID)\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.ShardOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: shardID,\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to update shard rangeID.  request_range_id: %v, actual_range_id : %v, columns: (%v)\",\n\t\t\t\t\tpreviousRangeID, conditionFailure.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(storeShard.db, \"UpdateRangeID\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (sh *nosqlShardStore) UpdateShard(\n\tctx context.Context,\n\trequest *persistence.InternalUpdateShardRequest,\n) error {\n\tstoreShard, err := sh.GetStoreShardByHistoryShard(request.ShardInfo.ShardID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tshardRow, err := sh.shardInfoToShardsRow(request.ShardInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = storeShard.db.UpdateShard(ctx, shardRow, request.PreviousRangeID)\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.ShardOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: request.ShardInfo.ShardID,\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to update shard rangeID.  request_range_id: %v, actual_range_id : %v, columns: (%v)\",\n\t\t\t\t\trequest.PreviousRangeID, conditionFailure.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(storeShard.db, \"UpdateShard\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (sh *nosqlShardStore) shardInfoToShardsRow(s *persistence.InternalShardInfo) (*nosqlplugin.ShardRow, error) {\n\tvar markerData []byte\n\tmarkerEncoding := string(constants.EncodingTypeEmpty)\n\tif s.PendingFailoverMarkers != nil {\n\t\tmarkerData = s.PendingFailoverMarkers.Data\n\t\tmarkerEncoding = string(s.PendingFailoverMarkers.Encoding)\n\t}\n\n\tvar transferPQSData []byte\n\ttransferPQSEncoding := string(constants.EncodingTypeEmpty)\n\tif s.TransferProcessingQueueStates != nil {\n\t\ttransferPQSData = s.TransferProcessingQueueStates.Data\n\t\ttransferPQSEncoding = string(s.TransferProcessingQueueStates.Encoding)\n\t}\n\n\tvar timerPQSData []byte\n\ttimerPQSEncoding := string(constants.EncodingTypeEmpty)\n\tif s.TimerProcessingQueueStates != nil {\n\t\ttimerPQSData = s.TimerProcessingQueueStates.Data\n\t\ttimerPQSEncoding = string(s.TimerProcessingQueueStates.Encoding)\n\t}\n\n\tshardInfo := &serialization.ShardInfo{\n\t\tStolenSinceRenew:                      int32(s.StolenSinceRenew),\n\t\tUpdatedAt:                             s.UpdatedAt,\n\t\tReplicationAckLevel:                   s.ReplicationAckLevel,\n\t\tTransferAckLevel:                      s.TransferAckLevel,\n\t\tTimerAckLevel:                         s.TimerAckLevel,\n\t\tClusterTransferAckLevel:               s.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:                  s.ClusterTimerAckLevel,\n\t\tTransferProcessingQueueStates:         transferPQSData,\n\t\tTransferProcessingQueueStatesEncoding: transferPQSEncoding,\n\t\tTimerProcessingQueueStates:            timerPQSData,\n\t\tTimerProcessingQueueStatesEncoding:    timerPQSEncoding,\n\t\tDomainNotificationVersion:             s.DomainNotificationVersion,\n\t\tOwner:                                 s.Owner,\n\t\tClusterReplicationLevel:               s.ClusterReplicationLevel,\n\t\tReplicationDlqAckLevel:                s.ReplicationDLQAckLevel,\n\t\tPendingFailoverMarkers:                markerData,\n\t\tPendingFailoverMarkersEncoding:        markerEncoding,\n\t\tQueueStates:                           s.QueueStates,\n\t}\n\n\tblob, err := sh.parser.ShardInfoToBlob(shardInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlplugin.ShardRow{\n\t\tInternalShardInfo: s,\n\t\tData:              blob.Data,\n\t\tDataEncoding:      string(blob.Encoding),\n\t}, nil\n}\n\nfunc (sh *nosqlShardStore) shardsRowToShardInfo(storeShard *nosqlStore, shardRow *nosqlplugin.ShardRow, rangeID int64) (*persistence.InternalShardInfo, error) {\n\tif !storeShard.dc.ReadNoSQLShardFromDataBlob() {\n\t\tsh.GetMetricsClient().IncCounter(metrics.PersistenceGetShardScope, metrics.NoSQLShardStoreReadFromOriginalColumnCounter)\n\t\treturn shardRow.InternalShardInfo, nil\n\t}\n\tif len(shardRow.Data) == 0 {\n\t\tsh.GetMetricsClient().IncCounter(metrics.PersistenceGetShardScope, metrics.NoSQLShardStoreReadFromOriginalColumnCounter)\n\t\tsh.GetLogger().Warn(\"Shard info data blob is empty, falling back to typed fields\")\n\t\treturn shardRow.InternalShardInfo, nil\n\t}\n\tshardInfo, err := sh.parser.ShardInfoFromBlob(shardRow.Data, shardRow.DataEncoding)\n\tif err != nil {\n\t\tsh.GetMetricsClient().IncCounter(metrics.PersistenceGetShardScope, metrics.NoSQLShardStoreReadFromOriginalColumnCounter)\n\t\tsh.GetLogger().Error(\"Failed to decode shard info from data blob, falling back to typed fields\", tag.Error(err))\n\t\treturn shardRow.InternalShardInfo, nil\n\t}\n\n\tsh.GetMetricsClient().IncCounter(metrics.PersistenceGetShardScope, metrics.NoSQLShardStoreReadFromDataBlobCounter)\n\n\tif len(shardInfo.ClusterTransferAckLevel) == 0 {\n\t\tshardInfo.ClusterTransferAckLevel = map[string]int64{\n\t\t\tsh.currentClusterName: shardInfo.GetTransferAckLevel(),\n\t\t}\n\t}\n\n\ttimerAckLevel := make(map[string]time.Time, len(shardInfo.ClusterTimerAckLevel))\n\tfor k, v := range shardInfo.ClusterTimerAckLevel {\n\t\ttimerAckLevel[k] = v\n\t}\n\n\tif len(timerAckLevel) == 0 {\n\t\ttimerAckLevel = map[string]time.Time{\n\t\t\tsh.currentClusterName: shardInfo.GetTimerAckLevel(),\n\t\t}\n\t}\n\n\tif shardInfo.ClusterReplicationLevel == nil {\n\t\tshardInfo.ClusterReplicationLevel = make(map[string]int64)\n\t}\n\tif shardInfo.ReplicationDlqAckLevel == nil {\n\t\tshardInfo.ReplicationDlqAckLevel = make(map[string]int64)\n\t}\n\n\tvar transferPQS *persistence.DataBlob\n\tif shardInfo.GetTransferProcessingQueueStates() != nil {\n\t\ttransferPQS = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetTransferProcessingQueueStatesEncoding()),\n\t\t\tData:     shardInfo.GetTransferProcessingQueueStates(),\n\t\t}\n\t}\n\n\tvar timerPQS *persistence.DataBlob\n\tif shardInfo.GetTimerProcessingQueueStates() != nil {\n\t\ttimerPQS = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetTimerProcessingQueueStatesEncoding()),\n\t\t\tData:     shardInfo.GetTimerProcessingQueueStates(),\n\t\t}\n\t}\n\n\tvar pendingFailoverMarkers *persistence.DataBlob\n\tif shardInfo.GetPendingFailoverMarkers() != nil {\n\t\tpendingFailoverMarkers = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetPendingFailoverMarkersEncoding()),\n\t\t\tData:     shardInfo.GetPendingFailoverMarkers(),\n\t\t}\n\t}\n\treturn &persistence.InternalShardInfo{\n\t\tShardID:                       int(shardRow.ShardID),\n\t\tRangeID:                       rangeID,\n\t\tOwner:                         shardInfo.GetOwner(),\n\t\tStolenSinceRenew:              int(shardInfo.GetStolenSinceRenew()),\n\t\tUpdatedAt:                     shardInfo.GetUpdatedAt(),\n\t\tReplicationAckLevel:           shardInfo.GetReplicationAckLevel(),\n\t\tTransferAckLevel:              shardInfo.GetTransferAckLevel(),\n\t\tTimerAckLevel:                 shardInfo.GetTimerAckLevel(),\n\t\tClusterTransferAckLevel:       shardInfo.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          timerAckLevel,\n\t\tTransferProcessingQueueStates: transferPQS,\n\t\tTimerProcessingQueueStates:    timerPQS,\n\t\tDomainNotificationVersion:     shardInfo.GetDomainNotificationVersion(),\n\t\tClusterReplicationLevel:       shardInfo.ClusterReplicationLevel,\n\t\tReplicationDLQAckLevel:        shardInfo.ReplicationDlqAckLevel,\n\t\tPendingFailoverMarkers:        pendingFailoverMarkers,\n\t\tQueueStates:                   shardInfo.GetQueueStates(),\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_shard_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc testFixtureInternalShardInfo() *persistence.InternalShardInfo {\n\tnow := time.Unix(100000, 0)\n\treturn &persistence.InternalShardInfo{\n\t\tShardID:             1,\n\t\tOwner:               \"test-owner\",\n\t\tRangeID:             101,\n\t\tStolenSinceRenew:    0,\n\t\tUpdatedAt:           now,\n\t\tReplicationAckLevel: 100,\n\t\tReplicationDLQAckLevel: map[string]int64{\n\t\t\t\"cluster-1\": 10,\n\t\t\t\"cluster-2\": 15,\n\t\t},\n\t\tTransferAckLevel: 50,\n\t\tTimerAckLevel:    now.Add(-time.Hour),\n\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\"cluster-1\": 60,\n\t\t\t\"cluster-2\": 70,\n\t\t},\n\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\"cluster-1\": now.Add(-time.Minute * 30),\n\t\t\t\"cluster-2\": now.Add(-time.Minute * 60),\n\t\t},\n\t\tTransferProcessingQueueStates: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"transfer-processing-states\"),\n\t\t},\n\t\tTimerProcessingQueueStates: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"timer-processing-states\"),\n\t\t},\n\t\tClusterReplicationLevel: map[string]int64{\n\t\t\t\"cluster-1\": 200,\n\t\t\t\"cluster-2\": 250,\n\t\t},\n\t\tDomainNotificationVersion: 102,\n\t\tPendingFailoverMarkers: &persistence.DataBlob{\n\t\t\tEncoding: \"base64\",\n\t\t\tData:     []byte(\"pending-failover-markers\"),\n\t\t},\n\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t0: &types.QueueState{\n\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc setUpMocksForShardStore(t *testing.T) (*nosqlShardStore, *nosqlplugin.MockDB, *MockshardedNosqlStore, *serialization.MockParser, *gomock.Controller) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\tstoreShardMock := NewMockshardedNosqlStore(ctrl)\n\tmockParser := serialization.NewMockParser(ctrl)\n\n\tshardStore := &nosqlShardStore{\n\t\tshardedNosqlStore:  storeShardMock,\n\t\tcurrentClusterName: \"test-cluster\",\n\t\tparser:             mockParser,\n\t}\n\n\treturn shardStore, dbMock, storeShardMock, mockParser, ctrl\n}\n\nfunc TestCreateShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB, *MockshardedNosqlStore, *serialization.MockParser)\n\t\trequest       *persistence.InternalCreateShardRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().InsertShard(gomock.Any(), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: testFixtureInternalShardInfo(),\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: testFixtureInternalShardInfo(),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"shard already exists error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().InsertShard(gomock.Any(), gomock.Any()).Return(&nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 200,\n\t\t\t\t\tDetails: \"rangeID mismatch\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Shard already exists in executions table.  ShardId: 1, request_range_id: 100, actual_range_id : 200, columns: (rangeID mismatch)\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().InsertShard(gomock.Any(), gomock.Any()).Return(errors.New(\"db error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"CreateShard failed. Error: db error \",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tshardStore, dbMock, storeShardMock, mockParser, ctrl := setUpMocksForShardStore(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock, storeShardMock, mockParser)\n\n\t\t\t// Execute the method under test\n\t\t\terr := shardStore.CreateShard(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB, *MockshardedNosqlStore, *serialization.MockParser)\n\t\trequest       *persistence.InternalGetShardRequest\n\t\texpectError   bool\n\t\texpected      *persistence.InternalGetShardResponse\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success - no update\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock, dc: &persistence.DynamicConfiguration{\n\t\t\t\t\tReadNoSQLShardFromDataBlob: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\t}}, nil).Times(1)\n\t\t\t\tshardInfo := testFixtureInternalShardInfo()\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(101), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: shardInfo,\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tstoreShardMock.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.ShardInfo{\n\t\t\t\t\tOwner:               \"test-owner\",\n\t\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\t\tUpdatedAt:           time.Unix(100000, 0),\n\t\t\t\t\tReplicationAckLevel: 100,\n\t\t\t\t\tTransferAckLevel:    50,\n\t\t\t\t\tTimerAckLevel:       time.Unix(100000, 0),\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 60,\n\t\t\t\t\t\t\"cluster-2\": 70,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\t\"cluster-1\": time.Unix(100000, 0),\n\t\t\t\t\t\t\"cluster-2\": time.Unix(100000, 0),\n\t\t\t\t\t},\n\t\t\t\t\tTransferProcessingQueueStates:         []byte(\"transfer-processing-states\"),\n\t\t\t\t\tTransferProcessingQueueStatesEncoding: \"base64\",\n\t\t\t\t\tTimerProcessingQueueStates:            []byte(\"timer-processing-states\"),\n\t\t\t\t\tTimerProcessingQueueStatesEncoding:    \"base64\",\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 200,\n\t\t\t\t\t\t\"cluster-2\": 250,\n\t\t\t\t\t},\n\t\t\t\t\tDomainNotificationVersion:      102,\n\t\t\t\t\tPendingFailoverMarkers:         []byte(\"pending-failover-markers\"),\n\t\t\t\t\tPendingFailoverMarkersEncoding: \"base64\",\n\t\t\t\t\tReplicationDlqAckLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 10,\n\t\t\t\t\t\t\"cluster-2\": 15,\n\t\t\t\t\t},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalGetShardResponse{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:             1,\n\t\t\t\t\tRangeID:             101,\n\t\t\t\t\tOwner:               \"test-owner\",\n\t\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\t\tUpdatedAt:           time.Unix(100000, 0),\n\t\t\t\t\tReplicationAckLevel: 100,\n\t\t\t\t\tTransferAckLevel:    50,\n\t\t\t\t\tTimerAckLevel:       time.Unix(100000, 0),\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 60,\n\t\t\t\t\t\t\"cluster-2\": 70,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\t\"cluster-1\": time.Unix(100000, 0),\n\t\t\t\t\t\t\"cluster-2\": time.Unix(100000, 0),\n\t\t\t\t\t},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\t\tData:     []byte(\"transfer-processing-states\"),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\t\tData:     []byte(\"timer-processing-states\"),\n\t\t\t\t\t},\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 200,\n\t\t\t\t\t\t\"cluster-2\": 250,\n\t\t\t\t\t},\n\t\t\t\t\tDomainNotificationVersion: 102,\n\t\t\t\t\tPendingFailoverMarkers: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\t\tData:     []byte(\"pending-failover-markers\"),\n\t\t\t\t\t},\n\t\t\t\t\tReplicationDLQAckLevel: map[string]int64{\n\t\t\t\t\t\t\"cluster-1\": 10,\n\t\t\t\t\t\t\"cluster-2\": 15,\n\t\t\t\t\t},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - fix shard\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock, dc: &persistence.DynamicConfiguration{\n\t\t\t\t\tReadNoSQLShardFromDataBlob: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}}, nil).Times(2)\n\t\t\t\tstoreShardMock.EXPECT().GetLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tstoreShardMock.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).Times(1)\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(100), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: testFixtureInternalShardInfo(),\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(101), int64(100)).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError: false,\n\t\t\texpected: &persistence.InternalGetShardResponse{\n\t\t\t\tShardInfo: testFixtureInternalShardInfo(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error fixing shard - shard ownership lost error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock, dc: &persistence.DynamicConfiguration{\n\t\t\t\t\tReadNoSQLShardFromDataBlob: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}}, nil).Times(2)\n\t\t\t\tstoreShardMock.EXPECT().GetLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(100), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: testFixtureInternalShardInfo(),\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(101), int64(100)).Return(&nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 200,\n\t\t\t\t\tDetails: \"rangeID mismatch\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Failed to update shard rangeID.  request_range_id: 100, actual_range_id : 200, columns: (rangeID mismatch)\",\n\t\t},\n\t\t{\n\t\t\tname: \"error fixing shard - generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock, dc: &persistence.DynamicConfiguration{\n\t\t\t\t\tReadNoSQLShardFromDataBlob: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}}, nil).Times(2)\n\t\t\t\tstoreShardMock.EXPECT().GetLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(100), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: testFixtureInternalShardInfo(),\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(101), int64(100)).Return(errors.New(\"db error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"UpdateRangeID failed. Error: db error \",\n\t\t},\n\t\t{\n\t\t\tname: \"shard not found error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(0), nil, errors.New(\"not found\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(errors.New(\"not found\")).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Shard not found.  ShardId: 1\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().SelectShard(gomock.Any(), 1, \"test-cluster\").Return(int64(0), nil, errors.New(\"db error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest:       &persistence.InternalGetShardRequest{ShardID: 1},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"GetShard failed. Error: db error \",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tshardStore, dbMock, storeShardMock, mockParser, ctrl := setUpMocksForShardStore(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock, storeShardMock, mockParser)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := shardStore.GetShard(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB, *MockshardedNosqlStore, *serialization.MockParser)\n\t\trequest       *persistence.InternalUpdateShardRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateShard(gomock.Any(), &nosqlplugin.ShardRow{\n\t\t\t\t\tInternalShardInfo: testFixtureInternalShardInfo(),\n\t\t\t\t\tData:              []byte(\"shard-info\"),\n\t\t\t\t\tDataEncoding:      \"base64\",\n\t\t\t\t}, int64(100)).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo:       testFixtureInternalShardInfo(),\n\t\t\t\tPreviousRangeID: 100,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"shard ownership lost error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateShard(gomock.Any(), gomock.Any(), int64(100)).Return(&nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 200,\n\t\t\t\t\tDetails: \"rangeID mismatch\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t},\n\t\t\t\tPreviousRangeID: 100,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Failed to update shard rangeID.  request_range_id: 100, actual_range_id : 200, columns: (rangeID mismatch)\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateShard(gomock.Any(), gomock.Any(), int64(100)).Return(errors.New(\"db error\")).Times(1)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: \"base64\",\n\t\t\t\t\tData:     []byte(\"shard-info\"),\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t},\n\t\t\t\tPreviousRangeID: 100,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"UpdateShard failed. Error: db error \",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tshardStore, dbMock, storeShardMock, mockParser, ctrl := setUpMocksForShardStore(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock, storeShardMock, mockParser)\n\n\t\t\t// Execute the method under test\n\t\t\terr := shardStore.UpdateShard(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateRangeID(t *testing.T) {\n\ttestCases := []struct {\n\t\tname            string\n\t\tsetupMock       func(*nosqlplugin.MockDB, *MockshardedNosqlStore, *serialization.MockParser)\n\t\tshardID         int\n\t\trangeID         int64\n\t\tpreviousRangeID int64\n\t\texpectError     bool\n\t\texpectedError   string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(100), int64(99)).Return(nil).Times(1)\n\t\t\t},\n\t\t\tshardID:         1,\n\t\t\trangeID:         100,\n\t\t\tpreviousRangeID: 99,\n\t\t\texpectError:     false,\n\t\t},\n\t\t{\n\t\t\tname: \"shard ownership lost error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(100), int64(99)).Return(&nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: 200,\n\t\t\t\t\tDetails: \"rangeID mismatch\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tshardID:         1,\n\t\t\trangeID:         100,\n\t\t\tpreviousRangeID: 99,\n\t\t\texpectError:     true,\n\t\t\texpectedError:   \"Failed to update shard rangeID.  request_range_id: 99, actual_range_id : 200, columns: (rangeID mismatch)\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB, storeShardMock *MockshardedNosqlStore, mockParser *serialization.MockParser) {\n\t\t\t\tstoreShardMock.EXPECT().GetStoreShardByHistoryShard(1).Return(&nosqlStore{db: dbMock}, nil).Times(1)\n\t\t\t\tdbMock.EXPECT().UpdateRangeID(gomock.Any(), 1, int64(100), int64(99)).Return(errors.New(\"db error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\tshardID:         1,\n\t\t\trangeID:         100,\n\t\t\tpreviousRangeID: 99,\n\t\t\texpectError:     true,\n\t\t\texpectedError:   \"UpdateRangeID failed. Error: db error \",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tshardStore, dbMock, storeShardMock, mockParser, ctrl := setUpMocksForShardStore(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock, storeShardMock, mockParser)\n\n\t\t\t// Execute the method under test\n\t\t\terr := shardStore.updateRangeID(context.Background(), tc.shardID, tc.rangeID, tc.previousRangeID)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_store.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// a shared struct for all stores in this package\ntype nosqlStore struct {\n\tlogger log.Logger\n\tdb     nosqlplugin.DB\n\tdc     *persistence.DynamicConfiguration\n}\n\nfunc (nm *nosqlStore) GetName() string {\n\treturn nm.db.PluginName()\n}\n\n// Close releases the underlying resources held by this object\nfunc (nm *nosqlStore) Close() {\n\tnm.db.Close()\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_task_store.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tnosqlTaskStore struct {\n\t\tshardedNosqlStore\n\t}\n)\n\nconst (\n\tinitialRangeID  int64 = 1 // Id of the first range of a new task list\n\tinitialAckLevel int64 = 0\n\ttaskListTTL           = int64(24 * time.Hour / time.Second) // Applied only to kinds that have TTL\n)\n\n// newNoSQLTaskStore is used to create an instance of TaskStore implementation\nfunc newNoSQLTaskStore(\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.TaskStore, error) {\n\ts, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlTaskStore{\n\t\tshardedNosqlStore: s,\n\t}, nil\n}\n\nfunc (t *nosqlTaskStore) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (*persistence.GetOrphanTasksResponse, error) {\n\t// TODO: It's unclear if this's necessary or possible for NoSQL\n\treturn nil, &types.InternalServiceError{\n\t\tMessage: \"Unimplemented call to GetOrphanTasks for NoSQL\",\n\t}\n}\n\nfunc (t *nosqlTaskStore) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (*persistence.GetTaskListSizeResponse, error) {\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskListName, request.TaskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsize, err := storeShard.db.GetTasksCount(ctx, &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\tDomainID:     request.DomainID,\n\t\t\tTaskListName: request.TaskListName,\n\t\t\tTaskListType: request.TaskListType,\n\t\t},\n\t\tMinTaskID: request.AckLevel,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &persistence.GetTaskListSizeResponse{Size: size}, nil\n}\n\nfunc (t *nosqlTaskStore) LeaseTaskList(\n\tctx context.Context,\n\trequest *persistence.LeaseTaskListRequest,\n) (*persistence.LeaseTaskListResponse, error) {\n\tif len(request.TaskList) == 0 {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"LeaseTaskList requires non empty task list\",\n\t\t}\n\t}\n\tcurrentTimeStamp := request.CurrentTimeStamp\n\tvar err, selectErr error\n\tvar currTL *nosqlplugin.TaskListRow\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskList, request.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcurrTL, selectErr = storeShard.db.SelectTaskList(ctx, &nosqlplugin.TaskListFilter{\n\t\tDomainID:     request.DomainID,\n\t\tTaskListName: request.TaskList,\n\t\tTaskListType: request.TaskType,\n\t})\n\n\tif selectErr != nil {\n\t\tif storeShard.db.IsNotFoundError(selectErr) { // First time task list is used\n\t\t\tcurrTL = &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         request.DomainID,\n\t\t\t\tTaskListName:     request.TaskList,\n\t\t\t\tTaskListType:     request.TaskType,\n\t\t\t\tRangeID:          initialRangeID,\n\t\t\t\tTaskListKind:     request.TaskListKind,\n\t\t\t\tAckLevel:         initialAckLevel,\n\t\t\t\tLastUpdatedTime:  currentTimeStamp,\n\t\t\t\tCurrentTimeStamp: currentTimeStamp,\n\t\t\t}\n\t\t\terr = storeShard.db.InsertTaskList(ctx, currTL)\n\t\t} else {\n\t\t\treturn nil, convertCommonErrors(storeShard.db, \"LeaseTaskList\", selectErr)\n\t\t}\n\t} else {\n\t\t// if request.RangeID is > 0, we are trying to renew an already existing\n\t\t// lease on the task list. If request.RangeID=0, we are trying to steal\n\t\t// the tasklist from its current owner\n\t\tif request.RangeID > 0 && request.RangeID != currTL.RangeID {\n\t\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"leaseTaskList:renew failed: taskList:%v, taskListType:%v, haveRangeID:%v, gotRangeID:%v\",\n\t\t\t\t\trequest.TaskList, request.TaskType, request.RangeID, currTL.RangeID),\n\t\t\t}\n\t\t}\n\n\t\t// Update the rangeID as this is an ownership change\n\t\tcurrTL.RangeID++\n\n\t\terr = storeShard.db.UpdateTaskList(ctx, &nosqlplugin.TaskListRow{\n\t\t\tDomainID:                request.DomainID,\n\t\t\tTaskListName:            request.TaskList,\n\t\t\tTaskListType:            request.TaskType,\n\t\t\tRangeID:                 currTL.RangeID,\n\t\t\tTaskListKind:            currTL.TaskListKind,\n\t\t\tAckLevel:                currTL.AckLevel,\n\t\t\tLastUpdatedTime:         currentTimeStamp,\n\t\t\tCurrentTimeStamp:        currentTimeStamp,\n\t\t\tAdaptivePartitionConfig: currTL.AdaptivePartitionConfig,\n\t\t}, currTL.RangeID-1)\n\t}\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.TaskOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"leaseTaskList: taskList:%v, taskListType:%v, haveRangeID:%v, gotRangeID:%v\",\n\t\t\t\t\trequest.TaskList, request.TaskType, currTL.RangeID, conditionFailure.RangeID),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(storeShard.db, \"LeaseTaskList\", err)\n\t}\n\ttli := &persistence.TaskListInfo{\n\t\tDomainID:                request.DomainID,\n\t\tName:                    request.TaskList,\n\t\tTaskType:                request.TaskType,\n\t\tRangeID:                 currTL.RangeID,\n\t\tAckLevel:                currTL.AckLevel,\n\t\tKind:                    request.TaskListKind,\n\t\tLastUpdated:             currentTimeStamp,\n\t\tAdaptivePartitionConfig: currTL.AdaptivePartitionConfig,\n\t}\n\treturn &persistence.LeaseTaskListResponse{TaskListInfo: tli}, nil\n}\n\nfunc (t *nosqlTaskStore) GetTaskList(\n\tctx context.Context,\n\trequest *persistence.GetTaskListRequest,\n) (*persistence.GetTaskListResponse, error) {\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskList, request.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tcurrTL, err := storeShard.db.SelectTaskList(ctx, &nosqlplugin.TaskListFilter{\n\t\tDomainID:     request.DomainID,\n\t\tTaskListName: request.TaskList,\n\t\tTaskListType: request.TaskType,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"GetTaskList\", err)\n\t}\n\ttli := &persistence.TaskListInfo{\n\t\tDomainID:                request.DomainID,\n\t\tName:                    request.TaskList,\n\t\tTaskType:                request.TaskType,\n\t\tRangeID:                 currTL.RangeID,\n\t\tAckLevel:                currTL.AckLevel,\n\t\tKind:                    currTL.TaskListKind,\n\t\tLastUpdated:             currTL.LastUpdatedTime,\n\t\tAdaptivePartitionConfig: currTL.AdaptivePartitionConfig,\n\t}\n\treturn &persistence.GetTaskListResponse{TaskListInfo: tli}, nil\n}\n\nfunc (t *nosqlTaskStore) UpdateTaskList(\n\tctx context.Context,\n\trequest *persistence.UpdateTaskListRequest,\n) (*persistence.UpdateTaskListResponse, error) {\n\ttli := request.TaskListInfo\n\tvar err error\n\ttaskListToUpdate := &nosqlplugin.TaskListRow{\n\t\tDomainID:                tli.DomainID,\n\t\tTaskListName:            tli.Name,\n\t\tTaskListType:            tli.TaskType,\n\t\tRangeID:                 tli.RangeID,\n\t\tTaskListKind:            tli.Kind,\n\t\tAckLevel:                tli.AckLevel,\n\t\tLastUpdatedTime:         request.CurrentTimeStamp,\n\t\tCurrentTimeStamp:        request.CurrentTimeStamp,\n\t\tAdaptivePartitionConfig: tli.AdaptivePartitionConfig,\n\t}\n\tstoreShard, err := t.GetStoreShardByTaskList(tli.DomainID, tli.Name, tli.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif persistence.TaskListKindHasTTL(tli.Kind) {\n\t\terr = storeShard.db.UpdateTaskListWithTTL(ctx, taskListTTL, taskListToUpdate, tli.RangeID)\n\t} else {\n\t\terr = storeShard.db.UpdateTaskList(ctx, taskListToUpdate, tli.RangeID)\n\t}\n\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.TaskOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to update task list. name: %v, type: %v, rangeID: %v, columns: (%v)\",\n\t\t\t\t\ttli.Name, tli.TaskType, tli.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(storeShard.db, \"UpdateTaskList\", err)\n\t}\n\n\treturn &persistence.UpdateTaskListResponse{}, nil\n}\n\nfunc (t *nosqlTaskStore) ListTaskList(\n\t_ context.Context,\n\t_ *persistence.ListTaskListRequest,\n) (*persistence.ListTaskListResponse, error) {\n\treturn nil, &types.InternalServiceError{\n\t\tMessage: \"unsupported operation\",\n\t}\n}\n\nfunc (t *nosqlTaskStore) DeleteTaskList(\n\tctx context.Context,\n\trequest *persistence.DeleteTaskListRequest,\n) error {\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskListName, request.TaskListType)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = storeShard.db.DeleteTaskList(ctx, &nosqlplugin.TaskListFilter{\n\t\tDomainID:     request.DomainID,\n\t\tTaskListName: request.TaskListName,\n\t\tTaskListType: request.TaskListType,\n\t}, request.RangeID)\n\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.TaskOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to delete task list. name: %v, type: %v, rangeID: %v, columns: (%v)\",\n\t\t\t\t\trequest.TaskListName, request.TaskListType, request.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(storeShard.db, \"DeleteTaskList\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (t *nosqlTaskStore) CreateTasks(\n\tctx context.Context,\n\trequest *persistence.CreateTasksRequest,\n) (*persistence.CreateTasksResponse, error) {\n\tcurrentTimeStamp := request.CurrentTimeStamp\n\tvar tasks []*nosqlplugin.TaskRowForInsert\n\tfor _, taskRequest := range request.Tasks {\n\t\ttask := &nosqlplugin.TaskRow{\n\t\t\tDomainID:        request.TaskListInfo.DomainID,\n\t\t\tTaskListName:    request.TaskListInfo.Name,\n\t\t\tTaskListType:    request.TaskListInfo.TaskType,\n\t\t\tTaskID:          taskRequest.TaskID,\n\t\t\tWorkflowID:      taskRequest.Data.WorkflowID,\n\t\t\tRunID:           taskRequest.Data.RunID,\n\t\t\tScheduledID:     taskRequest.Data.ScheduleID,\n\t\t\tCreatedTime:     currentTimeStamp,\n\t\t\tPartitionConfig: taskRequest.Data.PartitionConfig,\n\t\t}\n\n\t\tvar ttl int\n\t\t// If the Data has a non-zero Expiry value, means that the ask is being re-added to the tasks table.\n\t\t// If that's the case, use the Expiry value to calculate the new TTL value to match history's timeout value.\n\t\tif !taskRequest.Data.Expiry.IsZero() {\n\t\t\tscheduleToStartTimeoutSeconds := int(taskRequest.Data.Expiry.Sub(currentTimeStamp).Seconds())\n\n\t\t\tif scheduleToStartTimeoutSeconds > 0 {\n\t\t\t\tttl = scheduleToStartTimeoutSeconds\n\t\t\t} else {\n\t\t\t\tlogger := t.GetLogger()\n\t\t\t\tlogger.Warn(\"Async task not created. Task is expired\", tag.WorkflowID(taskRequest.Data.WorkflowID), tag.WorkflowRunID(taskRequest.Data.RunID), tag.WorkflowScheduleID(taskRequest.Data.ScheduleID))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t} else {\n\t\t\tttl = int(taskRequest.Data.ScheduleToStartTimeoutSeconds)\n\t\t}\n\n\t\ttasks = append(tasks, &nosqlplugin.TaskRowForInsert{\n\t\t\tTaskRow:    *task,\n\t\t\tTTLSeconds: ttl,\n\t\t})\n\t}\n\n\ttasklistCondition := &nosqlplugin.TaskListRow{\n\t\tDomainID:         request.TaskListInfo.DomainID,\n\t\tTaskListName:     request.TaskListInfo.Name,\n\t\tTaskListType:     request.TaskListInfo.TaskType,\n\t\tRangeID:          request.TaskListInfo.RangeID,\n\t\tLastUpdatedTime:  currentTimeStamp,\n\t\tCurrentTimeStamp: currentTimeStamp,\n\t}\n\n\ttli := request.TaskListInfo\n\tstoreShard, err := t.GetStoreShardByTaskList(tli.DomainID, tli.Name, tli.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\terr = storeShard.db.InsertTasks(ctx, tasks, tasklistCondition)\n\n\tif err != nil {\n\t\tconditionFailure, ok := err.(*nosqlplugin.TaskOperationConditionFailure)\n\t\tif ok {\n\t\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"Failed to insert tasks. name: %v, type: %v, rangeID: %v, columns: (%v)\",\n\t\t\t\t\trequest.TaskListInfo.Name, request.TaskListInfo.TaskType, request.TaskListInfo.RangeID, conditionFailure.Details),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(storeShard.db, \"CreateTasks\", err)\n\t}\n\n\treturn &persistence.CreateTasksResponse{}, nil\n}\n\nfunc (t *nosqlTaskStore) GetTasks(\n\tctx context.Context,\n\trequest *persistence.GetTasksRequest,\n) (*persistence.GetTasksResponse, error) {\n\tif request.MaxReadLevel == nil {\n\t\trequest.MaxReadLevel = common.Int64Ptr(math.MaxInt64)\n\t}\n\n\tif request.ReadLevel > *request.MaxReadLevel {\n\t\treturn &persistence.GetTasksResponse{}, nil\n\t}\n\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskList, request.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, err := storeShard.db.SelectTasks(ctx, &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\tDomainID:     request.DomainID,\n\t\t\tTaskListName: request.TaskList,\n\t\t\tTaskListType: request.TaskType,\n\t\t},\n\t\tBatchSize: request.BatchSize,\n\n\t\tMinTaskID: request.ReadLevel,\n\t\tMaxTaskID: *request.MaxReadLevel,\n\t})\n\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"GetTasks\", err)\n\t}\n\n\tresponse := &persistence.GetTasksResponse{}\n\tfor _, t := range resp {\n\t\tresponse.Tasks = append(response.Tasks, toTaskInfo(t))\n\t}\n\n\treturn response, nil\n}\n\nfunc toTaskInfo(t *nosqlplugin.TaskRow) *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:        t.DomainID,\n\t\tWorkflowID:      t.WorkflowID,\n\t\tRunID:           t.RunID,\n\t\tTaskID:          t.TaskID,\n\t\tScheduleID:      t.ScheduledID,\n\t\tCreatedTime:     t.CreatedTime,\n\t\tPartitionConfig: t.PartitionConfig,\n\t}\n}\n\nfunc (t *nosqlTaskStore) CompleteTask(\n\tctx context.Context,\n\trequest *persistence.CompleteTaskRequest,\n) error {\n\ttli := request.TaskList\n\tstoreShard, err := t.GetStoreShardByTaskList(tli.DomainID, tli.Name, tli.TaskType)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = storeShard.db.RangeDeleteTasks(ctx, &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\tDomainID:     tli.DomainID,\n\t\t\tTaskListName: tli.Name,\n\t\t\tTaskListType: request.TaskList.TaskType,\n\t\t},\n\t\t// exclusive\n\t\tMinTaskID: request.TaskID - 1,\n\t\t// inclusive\n\t\tMaxTaskID: request.TaskID,\n\t\tBatchSize: 1,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(storeShard.db, \"CompleteTask\", err)\n\t}\n\n\treturn nil\n}\n\n// CompleteTasksLessThan deletes all tasks less than or equal to the given task id. This API ignores the\n// Limit request parameter i.e. either all tasks leq the task_id will be deleted or an error will\n// be returned to the caller\nfunc (t *nosqlTaskStore) CompleteTasksLessThan(\n\tctx context.Context,\n\trequest *persistence.CompleteTasksLessThanRequest,\n) (*persistence.CompleteTasksLessThanResponse, error) {\n\tstoreShard, err := t.GetStoreShardByTaskList(request.DomainID, request.TaskListName, request.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnum, err := storeShard.db.RangeDeleteTasks(ctx, &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\tDomainID:     request.DomainID,\n\t\t\tTaskListName: request.TaskListName,\n\t\t\tTaskListType: request.TaskType,\n\t\t},\n\n\t\t// NOTE: MinTaskID is supported in plugin interfaces but not exposed in dataInterfaces/persistenceInterfaces\n\t\t// We may want to add it so that we can test it.\n\t\t// https://github.com/uber/cadence/issues/4243\n\t\tMinTaskID: 0,\n\n\t\t// NOTE: request.TaskID is also inclusive, even though the name is CompleteTasksLessThan\n\t\tMaxTaskID: request.TaskID,\n\n\t\tBatchSize: request.Limit,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(storeShard.db, \"CompleteTasksLessThan\", err)\n\t}\n\treturn &persistence.CompleteTasksLessThanResponse{TasksCompleted: num}, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_task_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tTestDomainID     = \"test-domain-id\"\n\tTestDomainName   = \"test-domain\"\n\tTestTaskListName = \"test-tasklist\"\n\tTestTaskType     = persistence.TaskListTypeDecision\n\tTestWorkflowID   = \"test-workflow-id\"\n\tTestRunID        = \"test-run-id\"\n)\n\nfunc TestNewNoSQLStore(t *testing.T) {\n\tregisterCassandraMock(t)\n\tcfg := getValidShardedNoSQLConfig()\n\n\tstore, err := newNoSQLTaskStore(cfg, log.NewNoop(), metrics.NewNoopMetricsClient(), nil)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, store)\n}\n\nfunc setupNoSQLStoreMocks(t *testing.T) (*nosqlTaskStore, *nosqlplugin.MockDB) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\n\tnosqlSt := nosqlStore{\n\t\tlogger: log.NewNoop(),\n\t\tdb:     dbMock,\n\t}\n\n\tshardedNosqlStoreMock := NewMockshardedNosqlStore(ctrl)\n\tshardedNosqlStoreMock.EXPECT().\n\t\tGetStoreShardByTaskList(\n\t\t\tTestDomainID,\n\t\t\tTestTaskListName,\n\t\t\tTestTaskType).\n\t\tReturn(&nosqlSt, nil).\n\t\tAnyTimes()\n\tshardedNosqlStoreMock.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\n\tstore := &nosqlTaskStore{\n\t\tshardedNosqlStore: shardedNosqlStoreMock,\n\t}\n\n\treturn store, dbMock\n}\n\nfunc TestGetOrphanTasks(t *testing.T) {\n\tstore, _ := setupNoSQLStoreMocks(t)\n\n\t// We just expect the function to return an error so we don't need to check the result\n\t_, err := store.GetOrphanTasks(context.Background(), nil)\n\n\tvar expectedErr *types.InternalServiceError\n\tassert.ErrorAs(t, err, &expectedErr)\n\tassert.ErrorContains(t, err, \"Unimplemented call to GetOrphanTasks for NoSQL\")\n}\n\nfunc TestGetTaskListSize(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\tdb.EXPECT().GetTasksCount(\n\t\tgomock.Any(),\n\t\t&nosqlplugin.TasksFilter{\n\t\t\tTaskListFilter: *getDecisionTaskListFilter(),\n\t\t\tMinTaskID:      456,\n\t\t},\n\t).Return(int64(123), nil)\n\n\tsize, err := store.GetTaskListSize(context.Background(), &persistence.GetTaskListSizeRequest{\n\t\tDomainID:     TestDomainID,\n\t\tDomainName:   TestDomainName,\n\t\tTaskListName: TestTaskListName,\n\t\tTaskListType: int(types.TaskListTypeDecision),\n\t\tAckLevel:     456,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t,\n\t\t&persistence.GetTaskListSizeResponse{Size: 123},\n\t\tsize,\n\t)\n}\n\nfunc TestLeaseTaskList_emptyTaskList(t *testing.T) {\n\tstore, _ := setupNoSQLStoreMocks(t)\n\n\treq := getValidLeaseTaskListRequest()\n\treq.TaskList = \"\"\n\t_, err := store.LeaseTaskList(context.Background(), req)\n\n\tassert.ErrorAs(t, err, new(*types.InternalServiceError))\n\tassert.ErrorContains(t, err, \"LeaseTaskList requires non empty task list\")\n}\n\nfunc TestLeaseTaskList_selectErr(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\treq := getValidLeaseTaskListRequest()\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(nil, assert.AnError)\n\t// The error is _not_ a NotFoundError\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(false).Times(2)\n\tdb.EXPECT().IsTimeoutError(assert.AnError).Return(true).Times(1)\n\n\t_, err := store.LeaseTaskList(context.Background(), req)\n\n\tassert.Error(t, err)\n\t// The function does not wrap the error, it just adds it to the message\n\tassert.ErrorContains(t, err, assert.AnError.Error())\n}\n\nfunc TestLeaseTaskList_selectErrNotFound(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\treq := getValidLeaseTaskListRequest()\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(nil, assert.AnError)\n\t// The error _is_ a NotFoundError\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t// We then expect the tasklist to be inserted\n\tdb.EXPECT().InsertTaskList(gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, taskList *nosqlplugin.TaskListRow) error {\n\t\t\ttl := getExpectedTaskListRow()\n\t\t\tcheckTaskListRowExpected(t, tl, taskList)\n\t\t\treturn nil\n\t\t})\n\n\tresp, err := store.LeaseTaskList(context.Background(), req)\n\n\tassert.NoError(t, err)\n\tcheckTaskListInfoExpected(t, resp.TaskListInfo)\n}\n\nfunc TestLeaseTaskList_BadRenew(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\treq := getValidLeaseTaskListRequest()\n\treq.RangeID = 1 // Greater than 0, so we are trying to renew\n\n\ttaskListRow := getExpectedTaskListRow()\n\ttaskListRow.RangeID = 5 // The range ID in the DB is different from the one in the request\n\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(taskListRow, nil)\n\n\t_, err := store.LeaseTaskList(context.Background(), req)\n\n\tassert.ErrorAs(t, err, new(*persistence.ConditionFailedError))\n\texpectedMessage := \"leaseTaskList:renew failed: taskList:test-tasklist, taskListType:0, haveRangeID:1, gotRangeID:5\"\n\tassert.ErrorContains(t, err, expectedMessage)\n}\n\nfunc TestLeaseTaskList_Renew(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\ttaskListRow := getExpectedTaskListRow()\n\ttaskListRow.RangeID = 0 // The range ID in the DB is the same as the one in the request\n\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(taskListRow, nil)\n\tdb.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any(), int64(0)).\n\t\tDoAndReturn(func(ctx context.Context, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\tcheckTaskListRowExpected(t, getExpectedTaskListRow(), taskList)\n\t\t\treturn nil\n\t\t})\n\n\tresp, err := store.LeaseTaskList(context.Background(), getValidLeaseTaskListRequest())\n\n\tassert.NoError(t, err)\n\tcheckTaskListInfoExpected(t, resp.TaskListInfo)\n}\n\nfunc TestLeaseTaskList_RenewUpdateFailed_OperationConditionFailure(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\ttaskListRow := getExpectedTaskListRow()\n\ttaskListRow.RangeID = 0 // The range ID in the DB is the same as the one in the request\n\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(taskListRow, nil)\n\tdb.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any(), int64(0)).\n\t\tDoAndReturn(func(ctx context.Context, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\tcheckTaskListRowExpected(t, getExpectedTaskListRow(), taskList)\n\t\t\treturn &nosqlplugin.TaskOperationConditionFailure{RangeID: 10}\n\t\t})\n\n\t_, err := store.LeaseTaskList(context.Background(), getValidLeaseTaskListRequest())\n\n\tassert.ErrorAs(t, err, new(*persistence.ConditionFailedError))\n\texpectedMessage := \"leaseTaskList: taskList:test-tasklist, taskListType:0, haveRangeID:1, gotRangeID:10\"\n\tassert.ErrorContains(t, err, expectedMessage)\n}\n\nfunc TestLeaseTaskList_RenewUpdateFailed_OtherError(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\ttaskListRow := getExpectedTaskListRow()\n\ttaskListRow.RangeID = 0 // The range ID in the DB is the same as the one in the request\n\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(taskListRow, nil)\n\tdb.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any(), int64(0)).\n\t\tDoAndReturn(func(ctx context.Context, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\tcheckTaskListRowExpected(t, getExpectedTaskListRow(), taskList)\n\t\t\treturn assert.AnError\n\t\t})\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := store.LeaseTaskList(context.Background(), getValidLeaseTaskListRequest())\n\n\tassert.Error(t, err)\n\t// The function does not wrap the error, it just adds it to the message\n\tassert.ErrorContains(t, err, assert.AnError.Error())\n}\n\nfunc TestGetTaskList_Success(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\ttaskListRow := getExpectedTaskListRow()\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(taskListRow, nil)\n\tresp, err := store.GetTaskList(context.Background(), getValidGetTaskListRequest())\n\n\tassert.NoError(t, err)\n\tcheckTaskListInfoExpected(t, resp.TaskListInfo)\n}\n\nfunc TestGetTaskList_NotFound(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\tdb.EXPECT().SelectTaskList(gomock.Any(), getDecisionTaskListFilter()).Return(nil, errors.New(\"not found\"))\n\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\tresp, err := store.GetTaskList(context.Background(), getValidGetTaskListRequest())\n\n\tassert.ErrorAs(t, err, new(*types.EntityNotExistsError))\n\tassert.Nil(t, resp)\n}\n\nfunc TestUpdateTaskList(t *testing.T) {\n\ttc := []struct {\n\t\tname        string\n\t\tinfo        *persistence.TaskListInfo\n\t\tallowance   func(db *nosqlplugin.MockDB)\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tname: \"success - normal\",\n\t\t\tinfo: getExpectedTaskListInfo(types.TaskListKindNormal),\n\t\t\tallowance: func(db *nosqlplugin.MockDB) {\n\t\t\t\tdb.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any(), int64(1)).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\t\t\t\tcheckTaskListRowExpected(t, getExpectedTaskListRowWithPartitionConfig(), taskList)\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - ephemeral\",\n\t\t\tinfo: getExpectedTaskListInfo(types.TaskListKindEphemeral),\n\t\t\tallowance: func(db *nosqlplugin.MockDB) {\n\t\t\t\tdb.EXPECT().UpdateTaskListWithTTL(gomock.Any(), taskListTTL, gomock.Any(), int64(1)).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, ttlSeconds int64, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\t\t\t\texpectedTaskList := getExpectedTaskListRowWithPartitionConfig()\n\t\t\t\t\t\texpectedTaskList.TaskListKind = int(types.TaskListKindEphemeral)\n\t\t\t\t\t\tcheckTaskListRowExpected(t, expectedTaskList, taskList)\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - sticky\",\n\t\t\tinfo: getExpectedTaskListInfo(types.TaskListKindSticky),\n\t\t\tallowance: func(db *nosqlplugin.MockDB) {\n\t\t\t\tdb.EXPECT().UpdateTaskListWithTTL(gomock.Any(), taskListTTL, gomock.Any(), int64(1)).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, ttlSeconds int64, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\t\t\t\texpectedTaskList := getExpectedTaskListRowWithPartitionConfig()\n\t\t\t\t\t\texpectedTaskList.TaskListKind = int(types.TaskListKindSticky)\n\t\t\t\t\t\tcheckTaskListRowExpected(t, expectedTaskList, taskList)\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\tinfo: getExpectedTaskListInfo(types.TaskListKindNormal),\n\t\t\tallowance: func(db *nosqlplugin.MockDB) {\n\t\t\t\tdb.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any(), int64(1)).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, taskList *nosqlplugin.TaskListRow, previousRangeID int64) error {\n\t\t\t\t\t\tcheckTaskListRowExpected(t, getExpectedTaskListRowWithPartitionConfig(), taskList)\n\t\t\t\t\t\treturn &nosqlplugin.TaskOperationConditionFailure{Details: \"test-details\"}\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedErr: &persistence.ConditionFailedError{Msg: \"Failed to update task list. name: test-tasklist, type: 0, rangeID: 1, columns: (test-details)\"},\n\t\t},\n\t}\n\tfor _, tt := range tc {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tstore, db := setupNoSQLStoreMocks(t)\n\t\t\tif tt.allowance != nil {\n\t\t\t\ttt.allowance(db)\n\t\t\t}\n\t\t\t_, err := store.UpdateTaskList(context.Background(), &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo:     tt.info,\n\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t})\n\t\t\tif tt.expectedErr != nil {\n\t\t\t\tassert.Equal(t, err, tt.expectedErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteTaskList(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\tdb.EXPECT().DeleteTaskList(gomock.Any(), getDecisionTaskListFilter(), int64(0)).Return(nil)\n\n\terr := store.DeleteTaskList(context.Background(), getValidDeleteTaskListRequest())\n\tassert.NoError(t, err)\n}\n\nfunc TestDeleteTaskList_ConditionFailure(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\tdb.EXPECT().DeleteTaskList(gomock.Any(), getDecisionTaskListFilter(), int64(0)).Return(\n\t\t&nosqlplugin.TaskOperationConditionFailure{Details: \"test-details\"},\n\t)\n\n\terr := store.DeleteTaskList(context.Background(), getValidDeleteTaskListRequest())\n\n\tvar expectedErr *persistence.ConditionFailedError\n\tassert.ErrorAs(t, err, &expectedErr)\n\tassert.ErrorContains(t, err, \"Failed to delete task list. name: test-tasklist, type: 0, rangeID: 0, columns: (test-details)\")\n}\n\nfunc TestGetTasks(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\tnow := time.Unix(123, 456)\n\n\ttaskrow1 := nosqlplugin.TaskRow{\n\t\tDomainID:        TestDomainID,\n\t\tTaskListName:    TestTaskListName,\n\t\tTaskListType:    int(types.TaskListTypeDecision),\n\t\tTaskID:          5,\n\t\tWorkflowID:      TestWorkflowID,\n\t\tRunID:           TestRunID,\n\t\tScheduledID:     0,\n\t\tCreatedTime:     now,\n\t\tPartitionConfig: nil,\n\t}\n\n\ttaskrow2 := taskrow1\n\ttaskrow2.TaskID = 6\n\n\tdb.EXPECT().SelectTasks(gomock.Any(), &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: *getDecisionTaskListFilter(),\n\t\tBatchSize:      100,\n\t\tMinTaskID:      1,\n\t\tMaxTaskID:      15,\n\t}).Return([]*nosqlplugin.TaskRow{&taskrow1, &taskrow2}, nil)\n\n\tresp, err := store.GetTasks(context.Background(), getValidGetTasksRequest())\n\n\tassert.NoError(t, err)\n\tassert.NotNil(t, resp)\n\tassert.Len(t, resp.Tasks, 2)\n\ttaskRowEqualTaskInfo(t, taskrow1, resp.Tasks[0])\n\ttaskRowEqualTaskInfo(t, taskrow2, resp.Tasks[1])\n}\n\nfunc TestGetTasks_Empty(t *testing.T) {\n\tstore, _ := setupNoSQLStoreMocks(t)\n\n\trequest := getValidGetTasksRequest()\n\t// Set the max read level to be less than the min read level\n\trequest.ReadLevel = 10\n\trequest.MaxReadLevel = common.Int64Ptr(5)\n\tresp, err := store.GetTasks(context.Background(), request)\n\n\tassert.NoError(t, err)\n\tassert.NotNil(t, resp)\n\tassert.Empty(t, resp.Tasks)\n}\n\nfunc TestCompleteTask(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\t// The main assertion is that the correct parameters are passed to the DB\n\tdb.EXPECT().RangeDeleteTasks(gomock.Any(), &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: *getDecisionTaskListFilter(),\n\t\tMinTaskID:      11,\n\t\tMaxTaskID:      12,\n\t\tBatchSize:      1,\n\t}).Return(1, nil)\n\n\terr := store.CompleteTask(context.Background(), &persistence.CompleteTaskRequest{\n\t\tTaskList: &persistence.TaskListInfo{\n\t\t\tDomainID: TestDomainID,\n\t\t\tName:     TestTaskListName,\n\t\t\tTaskType: int(types.TaskListTypeDecision),\n\t\t},\n\t\tTaskID:     12,\n\t\tDomainName: TestDomainName,\n\t})\n\n\tassert.NoError(t, err)\n}\n\nfunc TestCompleteTasksLessThan(t *testing.T) {\n\tstore, db := setupNoSQLStoreMocks(t)\n\n\t// The main assertion is that the correct parameters are passed to the DB\n\tdb.EXPECT().RangeDeleteTasks(gomock.Any(), &nosqlplugin.TasksFilter{\n\t\tTaskListFilter: *getDecisionTaskListFilter(),\n\t\tMinTaskID:      0,\n\t\tMaxTaskID:      12,\n\t\tBatchSize:      100,\n\t}).Return(13, nil)\n\n\tresp, err := store.CompleteTasksLessThan(context.Background(), &persistence.CompleteTasksLessThanRequest{\n\t\tDomainID:     TestDomainID,\n\t\tTaskListName: TestTaskListName,\n\t\tTaskType:     0,\n\t\tTaskID:       12,\n\t\tLimit:        100,\n\t\tDomainName:   TestDomainName,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, 13, resp.TasksCompleted)\n}\n\nfunc TestCreateTasks(t *testing.T) {\n\tnow := FixedTime\n\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*nosqlplugin.MockDB)\n\t\trequest       *persistence.CreateTasksRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertTasks(gomock.Any(), gomock.Any(), &nosqlplugin.TaskListRow{\n\t\t\t\t\tDomainID:     TestDomainID,\n\t\t\t\t\tTaskListName: TestTaskListName,\n\t\t\t\t\tTaskListType: TestTaskType,\n\t\t\t\t\tRangeID:      1,\n\t\t\t\t}).Do(func(_ context.Context, tasks []*nosqlplugin.TaskRowForInsert, _ *nosqlplugin.TaskListRow) {\n\t\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\t\tassert.Equal(t, TestDomainID, tasks[0].DomainID)\n\t\t\t\t\tassert.Equal(t, \"workflow1\", tasks[0].WorkflowID)\n\t\t\t\t\tassert.Equal(t, \"run1\", tasks[0].RunID)\n\t\t\t\t\tassert.Equal(t, int64(100), tasks[0].TaskID)\n\t\t\t\t\tassert.Equal(t, int64(10), tasks[0].ScheduledID)\n\t\t\t\t\tassert.Equal(t, TestTaskType, tasks[0].TaskListType)\n\t\t\t\t\tassert.Equal(t, TestTaskListName, tasks[0].TaskListName)\n\t\t\t\t\tassert.Equal(t, 30, tasks[0].TTLSeconds)\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: TestDomainID,\n\t\t\t\t\tName:     TestTaskListName,\n\t\t\t\t\tTaskType: TestTaskType,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\t\t\t\tRunID:                         \"run1\",\n\t\t\t\t\t\t\tScheduleID:                    10,\n\t\t\t\t\t\t\tPartitionConfig:               nil,\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 30,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success - adding task with Expiry not expired\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertTasks(gomock.Any(), gomock.Any(), &nosqlplugin.TaskListRow{\n\t\t\t\t\tDomainID:     TestDomainID,\n\t\t\t\t\tTaskListName: TestTaskListName,\n\t\t\t\t\tTaskListType: TestTaskType,\n\t\t\t\t\tRangeID:      1,\n\t\t\t\t}).Do(func(_ context.Context, tasks []*nosqlplugin.TaskRowForInsert, _ *nosqlplugin.TaskListRow) {\n\t\t\t\t\tassert.Len(t, tasks, 1)\n\t\t\t\t\tassert.Equal(t, TestDomainID, tasks[0].DomainID)\n\t\t\t\t\tassert.Equal(t, \"workflow1\", tasks[0].WorkflowID)\n\t\t\t\t\tassert.Equal(t, \"run1\", tasks[0].RunID)\n\t\t\t\t\tassert.Equal(t, int64(100), tasks[0].TaskID)\n\t\t\t\t\tassert.Equal(t, int64(10), tasks[0].ScheduledID)\n\t\t\t\t\tassert.Equal(t, TestTaskType, tasks[0].TaskListType)\n\t\t\t\t\tassert.Equal(t, TestTaskListName, tasks[0].TaskListName)\n\t\t\t\t\tassert.Equal(t, int(now.Add(100*time.Second).Sub(tasks[0].CreatedTime).Seconds()), tasks[0].TTLSeconds)\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: TestDomainID,\n\t\t\t\t\tName:     TestTaskListName,\n\t\t\t\t\tTaskType: TestTaskType,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tWorkflowID:      \"workflow1\",\n\t\t\t\t\t\t\tRunID:           \"run1\",\n\t\t\t\t\t\t\tScheduleID:      10,\n\t\t\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t\t\t\tExpiry:          now.Add(100 * time.Second),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success - skipping task with Expiry expired\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertTasks(gomock.Any(), gomock.Any(), &nosqlplugin.TaskListRow{\n\t\t\t\t\tDomainID:         TestDomainID,\n\t\t\t\t\tTaskListName:     TestTaskListName,\n\t\t\t\t\tTaskListType:     TestTaskType,\n\t\t\t\t\tRangeID:          1,\n\t\t\t\t\tLastUpdatedTime:  FixedTime,\n\t\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t\t}).Do(func(_ context.Context, tasks []*nosqlplugin.TaskRowForInsert, _ *nosqlplugin.TaskListRow) {\n\t\t\t\t\tassert.Len(t, tasks, 0)\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: TestDomainID,\n\t\t\t\t\tName:     TestTaskListName,\n\t\t\t\t\tTaskType: TestTaskType,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t},\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tWorkflowID:      \"workflow1\",\n\t\t\t\t\t\t\tRunID:           \"run1\",\n\t\t\t\t\t\t\tScheduleID:      10,\n\t\t\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t\t\t\tExpiry:          now,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"condition failure\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertTasks(gomock.Any(), gomock.Any(), gomock.Any()).Return(&nosqlplugin.TaskOperationConditionFailure{\n\t\t\t\t\tDetails: \"rangeID mismatch\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: TestDomainID,\n\t\t\t\t\tName:     TestTaskListName,\n\t\t\t\t\tTaskType: TestTaskType,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\t\t\t\tRunID:                         \"run1\",\n\t\t\t\t\t\t\tScheduleID:                    10,\n\t\t\t\t\t\t\tPartitionConfig:               nil,\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 30,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Failed to insert tasks. name: test-tasklist, type: 0, rangeID: 1, columns: (rangeID mismatch)\",\n\t\t},\n\t\t{\n\t\t\tname: \"generic db error\",\n\t\t\tsetupMock: func(dbMock *nosqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().InsertTasks(gomock.Any(), gomock.Any(), gomock.Any()).Return(errors.New(\"db error\")).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: TestDomainID,\n\t\t\t\t\tName:     TestTaskListName,\n\t\t\t\t\tTaskType: TestTaskType,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\t\t\t\tRunID:                         \"run1\",\n\t\t\t\t\t\t\tScheduleID:                    10,\n\t\t\t\t\t\t\tPartitionConfig:               nil,\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 30,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"CreateTasks failed. Error: db error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Setup mocks for each test case individually\n\t\t\tstore, dbMock := setupNoSQLStoreMocks(t)\n\n\t\t\t// Setup test-specific mock behavior\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\t// Execute the method under test\n\t\t\tresp, err := store.CreateTasks(context.Background(), tc.request)\n\n\t\t\t// Validate results\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getValidLeaseTaskListRequest() *persistence.LeaseTaskListRequest {\n\treturn &persistence.LeaseTaskListRequest{\n\t\tDomainID:         TestDomainID,\n\t\tDomainName:       TestDomainName,\n\t\tTaskList:         TestTaskListName,\n\t\tTaskType:         int(types.TaskListTypeDecision),\n\t\tTaskListKind:     int(types.TaskListKindNormal),\n\t\tRangeID:          0,\n\t\tCurrentTimeStamp: FixedTime,\n\t}\n}\n\nfunc getValidGetTaskListRequest() *persistence.GetTaskListRequest {\n\treturn &persistence.GetTaskListRequest{\n\t\tDomainID:   TestDomainID,\n\t\tDomainName: TestDomainName,\n\t\tTaskList:   TestTaskListName,\n\t\tTaskType:   int(types.TaskListTypeDecision),\n\t}\n}\n\nfunc checkTaskListInfoExpected(t *testing.T, taskListInfo *persistence.TaskListInfo) {\n\tassert.Equal(t, TestDomainID, taskListInfo.DomainID)\n\tassert.Equal(t, TestTaskListName, taskListInfo.Name)\n\tassert.Equal(t, int(types.TaskListTypeDecision), taskListInfo.TaskType)\n\tassert.Equal(t, initialRangeID, taskListInfo.RangeID)\n\tassert.Equal(t, initialAckLevel, taskListInfo.AckLevel)\n\tassert.Equal(t, int(types.TaskListKindNormal), taskListInfo.Kind)\n\tassert.Equal(t, FixedTime, taskListInfo.LastUpdated)\n}\n\nfunc taskRowEqualTaskInfo(t *testing.T, taskrow1 nosqlplugin.TaskRow, taskInfo1 *persistence.TaskInfo) {\n\tassert.Equal(t, taskrow1.DomainID, taskInfo1.DomainID)\n\tassert.Equal(t, taskrow1.WorkflowID, taskInfo1.WorkflowID)\n\tassert.Equal(t, taskrow1.RunID, taskInfo1.RunID)\n\tassert.Equal(t, taskrow1.TaskID, taskInfo1.TaskID)\n\tassert.Equal(t, taskrow1.ScheduledID, taskInfo1.ScheduleID)\n\tassert.Equal(t, taskrow1.CreatedTime, taskInfo1.CreatedTime)\n\tassert.Equal(t, taskrow1.PartitionConfig, taskInfo1.PartitionConfig)\n}\n\nfunc getValidGetTasksRequest() *persistence.GetTasksRequest {\n\treturn &persistence.GetTasksRequest{\n\t\tDomainID:   TestDomainID,\n\t\tDomainName: TestDomainName,\n\t\tTaskList:   TestTaskListName,\n\t\tTaskType:   int(types.TaskListTypeDecision),\n\t\t// The read level is the smallest taskID that we want to read, the maxReadLevel is the largest\n\t\tReadLevel:    1,\n\t\tMaxReadLevel: common.Int64Ptr(15),\n\t\tBatchSize:    100,\n\t}\n}\n\nfunc getDecisionTaskListFilter() *nosqlplugin.TaskListFilter {\n\treturn &nosqlplugin.TaskListFilter{\n\t\tDomainID:     TestDomainID,\n\t\tTaskListName: TestTaskListName,\n\t\tTaskListType: int(types.TaskListTypeDecision),\n\t}\n}\n\nfunc getExpectedTaskListRow() *nosqlplugin.TaskListRow {\n\treturn &nosqlplugin.TaskListRow{\n\t\tDomainID:         TestDomainID,\n\t\tTaskListName:     TestTaskListName,\n\t\tTaskListType:     int(types.TaskListTypeDecision),\n\t\tRangeID:          initialRangeID,\n\t\tTaskListKind:     int(types.TaskListKindNormal),\n\t\tAckLevel:         initialAckLevel,\n\t\tLastUpdatedTime:  FixedTime,\n\t\tCurrentTimeStamp: FixedTime,\n\t}\n}\n\nfunc getExpectedTaskListRowWithPartitionConfig() *nosqlplugin.TaskListRow {\n\treturn &nosqlplugin.TaskListRow{\n\t\tDomainID:         TestDomainID,\n\t\tTaskListName:     TestTaskListName,\n\t\tTaskListType:     int(types.TaskListTypeDecision),\n\t\tRangeID:          initialRangeID,\n\t\tTaskListKind:     int(types.TaskListKindNormal),\n\t\tAckLevel:         initialAckLevel,\n\t\tLastUpdatedTime:  FixedTime,\n\t\tCurrentTimeStamp: FixedTime,\n\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\tVersion: 1,\n\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {},\n\t\t\t\t2: {\n\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {\n\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc checkTaskListRowExpected(t *testing.T, expectedRow *nosqlplugin.TaskListRow, taskList *nosqlplugin.TaskListRow) {\n\tt.Helper()\n\n\tassert.Equal(t, expectedRow, taskList)\n}\n\nfunc getExpectedTaskListInfo(kind types.TaskListKind) *persistence.TaskListInfo {\n\treturn &persistence.TaskListInfo{\n\t\tDomainID:    TestDomainID,\n\t\tName:        TestTaskListName,\n\t\tTaskType:    int(types.TaskListTypeDecision),\n\t\tRangeID:     initialRangeID,\n\t\tAckLevel:    initialAckLevel,\n\t\tKind:        int(kind),\n\t\tLastUpdated: FixedTime,\n\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\tVersion: 1,\n\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {},\n\t\t\t\t2: {\n\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t\t1: {\n\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc getValidDeleteTaskListRequest() *persistence.DeleteTaskListRequest {\n\treturn &persistence.DeleteTaskListRequest{\n\t\tDomainID:     TestDomainID,\n\t\tDomainName:   TestDomainName,\n\t\tTaskListName: TestTaskListName,\n\t\tTaskListType: int(types.TaskListTypeDecision),\n\t\tRangeID:      0,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_test_utils.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n)\n\n// testCluster allows executing cassandra operations in testing.\ntype testCluster struct {\n\tlogger log.Logger\n\n\tkeyspace      string\n\tschemaBaseDir string\n\treplicas      int\n\tcfg           config.NoSQL\n}\n\n// TestClusterParams are params for test cluster initialization.\ntype TestClusterParams struct {\n\tPluginName    string\n\tKeySpace      string\n\tUsername      string\n\tPassword      string\n\tHost          string\n\tPort          int\n\tProtoVersion  int\n\tSchemaBaseDir string\n\t// Replicas defaults to 1 if not set\n\tReplicas int\n\t// MaxConns defaults to 2 if not set\n\tMaxConns int\n}\n\n// NewTestCluster returns a new cassandra test cluster\n// if schemaBaseDir is empty, it will be auto-resolved based on os.Getwd()\n// otherwise the specified value will be used (used by internal tests)\nfunc NewTestCluster(t *testing.T, params TestClusterParams) testcluster.PersistenceTestCluster {\n\treturn &testCluster{\n\t\tlogger:        testlogger.New(t),\n\t\tkeyspace:      params.KeySpace,\n\t\tschemaBaseDir: params.SchemaBaseDir,\n\t\treplicas:      replicas(params.Replicas),\n\t\tcfg: config.NoSQL{\n\t\t\tPluginName:   params.PluginName,\n\t\t\tUser:         params.Username,\n\t\t\tPassword:     params.Password,\n\t\t\tHosts:        params.Host,\n\t\t\tPort:         params.Port,\n\t\t\tMaxConns:     maxConns(params.MaxConns),\n\t\t\tKeyspace:     params.KeySpace,\n\t\t\tProtoVersion: params.ProtoVersion,\n\t\t},\n\t}\n}\n\n// Config returns the persistence config for connecting to this test cluster\nfunc (s *testCluster) Config() config.Persistence {\n\treturn config.Persistence{\n\t\tDefaultStore:    \"test\",\n\t\tVisibilityStore: \"test\",\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"test\": {NoSQL: &s.cfg},\n\t\t},\n\t\tTransactionSizeLimit: dynamicproperties.GetIntPropertyFn(constants.DefaultTransactionSizeLimit),\n\t\tErrorInjectionRate:   dynamicproperties.GetFloatPropertyFn(0),\n\t}\n}\n\n// SetupTestDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) SetupTestDatabase() {\n\tadminDB, err := NewNoSQLAdminDB(&s.cfg, s.logger, &persistence.DynamicConfiguration{})\n\n\tif err != nil {\n\t\ts.logger.Fatal(err.Error())\n\t}\n\terr = adminDB.SetupTestDatabase(s.schemaBaseDir, s.replicas)\n\tif err != nil {\n\t\ts.logger.Fatal(err.Error())\n\t}\n}\n\n// TearDownTestDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) TearDownTestDatabase() {\n\tadminDB, err := NewNoSQLAdminDB(&s.cfg, s.logger, &persistence.DynamicConfiguration{})\n\tif err != nil {\n\t\ts.logger.Fatal(err.Error())\n\t}\n\terr = adminDB.TeardownTestDatabase()\n\tif err != nil {\n\t\ts.logger.Fatal(err.Error())\n\t}\n}\n\nfunc replicas(replicas int) int {\n\tif replicas == 0 {\n\t\treturn 1\n\t}\n\n\treturn replicas\n}\n\nfunc maxConns(maxConns int) int {\n\tif maxConns == 0 {\n\t\treturn 2\n\t}\n\n\treturn maxConns\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_visibility_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Fixed domain values for now\nconst (\n\tdefaultCloseTTLSeconds = 86400\n\topenExecutionTTLBuffer = int64(86400) // setting it to a day to account for shard going down\n)\n\ntype nosqlVisibilityStore struct {\n\tsortByCloseTime bool\n\tnosqlStore\n}\n\n// newNoSQLVisibilityStore is used to create an instance of VisibilityStore implementation\nfunc newNoSQLVisibilityStore(\n\tlistClosedOrderingByCloseTime bool,\n\tcfg config.ShardedNoSQL,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.VisibilityStore, error) {\n\tshardedStore, err := newShardedNosqlStore(cfg, logger, metricsClient, dc)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &nosqlVisibilityStore{\n\t\tsortByCloseTime: listClosedOrderingByCloseTime,\n\t\tnosqlStore:      shardedStore.GetDefaultShard(),\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *persistence.InternalRecordWorkflowExecutionStartedRequest,\n) error {\n\tttl := int64(request.WorkflowTimeout.Seconds()) + openExecutionTTLBuffer\n\n\terr := v.db.InsertVisibility(ctx, ttl, &nosqlplugin.VisibilityRowForInsert{\n\t\tDomainID: request.DomainUUID,\n\t\tVisibilityRow: nosqlplugin.VisibilityRow{\n\t\t\tWorkflowID:             request.WorkflowID,\n\t\t\tRunID:                  request.RunID,\n\t\t\tTypeName:               request.WorkflowTypeName,\n\t\t\tStartTime:              request.StartTimestamp,\n\t\t\tExecutionTime:          request.ExecutionTimestamp,\n\t\t\tMemo:                   request.Memo,\n\t\t\tTaskList:               request.TaskList,\n\t\t\tIsCron:                 request.IsCron,\n\t\t\tNumClusters:            request.NumClusters,\n\t\t\tUpdateTime:             request.UpdateTimestamp,\n\t\t\tShardID:                request.ShardID,\n\t\t\tExecutionStatus:        request.ExecutionStatus,\n\t\t\tCronSchedule:           request.CronSchedule,\n\t\t\tScheduledExecutionTime: request.ScheduledExecutionTime,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(v.db, \"RecordWorkflowExecutionStarted\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (v *nosqlVisibilityStore) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *persistence.InternalRecordWorkflowExecutionClosedRequest,\n) error {\n\t// Find how long to keep the row\n\tretention := request.RetentionPeriod\n\tif retention == 0 {\n\t\tretention = defaultCloseTTLSeconds * time.Second\n\t}\n\n\terr := v.db.UpdateVisibility(ctx, int64(retention.Seconds()), &nosqlplugin.VisibilityRowForUpdate{\n\t\tDomainID:          request.DomainUUID,\n\t\tUpdateOpenToClose: true,\n\t\tVisibilityRow: nosqlplugin.VisibilityRow{\n\t\t\tWorkflowID:    request.WorkflowID,\n\t\t\tRunID:         request.RunID,\n\t\t\tTypeName:      request.WorkflowTypeName,\n\t\t\tStartTime:     request.StartTimestamp,\n\t\t\tExecutionTime: request.ExecutionTimestamp,\n\t\t\tMemo:          request.Memo,\n\t\t\tTaskList:      request.TaskList,\n\t\t\tIsCron:        request.IsCron,\n\t\t\tNumClusters:   request.NumClusters,\n\t\t\t// closed workflow attributes\n\t\t\tStatus:                 &request.Status,\n\t\t\tCloseTime:              request.CloseTimestamp,\n\t\t\tHistoryLength:          request.HistoryLength,\n\t\t\tUpdateTime:             request.UpdateTimestamp,\n\t\t\tCronSchedule:           request.CronSchedule,\n\t\t\tExecutionStatus:        request.ExecutionStatus,\n\t\t\tScheduledExecutionTime: request.ScheduledExecutionTime,\n\t\t},\n\t})\n\n\tif err != nil {\n\t\treturn convertCommonErrors(v.db, \"RecordWorkflowExecutionClosed\", err)\n\t}\n\treturn nil\n}\n\nfunc (v *nosqlVisibilityStore) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *persistence.InternalRecordWorkflowExecutionUninitializedRequest,\n) error {\n\t// temporary: not implemented, only implemented for ES\n\treturn nil\n}\n\nfunc (v *nosqlVisibilityStore) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalUpsertWorkflowExecutionRequest,\n) error {\n\tif persistence.IsNopUpsertWorkflowRequest(request) {\n\t\treturn nil\n\t}\n\treturn persistence.ErrVisibilityOperationNotSupported\n}\n\nfunc (v *nosqlVisibilityStore) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tresp, err := v.db.SelectVisibility(ctx, &nosqlplugin.VisibilityFilter{\n\t\tListRequest: *request,\n\t\tFilterType:  nosqlplugin.AllOpen,\n\t\tSortType:    nosqlplugin.SortByStartTime,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListOpenWorkflowExecutions\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tvar filter *nosqlplugin.VisibilityFilter\n\tif v.sortByCloseTime {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: *request,\n\t\t\tFilterType:  nosqlplugin.AllClosed,\n\t\t\tSortType:    nosqlplugin.SortByClosedTime,\n\t\t}\n\t} else {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: *request,\n\t\t\tFilterType:  nosqlplugin.AllClosed,\n\t\t\tSortType:    nosqlplugin.SortByStartTime,\n\t\t}\n\t}\n\tresp, err := v.db.SelectVisibility(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListClosedWorkflowExecutions\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsByTypeRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tresp, err := v.db.SelectVisibility(ctx, &nosqlplugin.VisibilityFilter{\n\t\tListRequest:  request.InternalListWorkflowExecutionsRequest,\n\t\tFilterType:   nosqlplugin.OpenByWorkflowType,\n\t\tSortType:     nosqlplugin.SortByStartTime,\n\t\tWorkflowType: request.WorkflowTypeName,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListOpenWorkflowExecutionsByType\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsByTypeRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tvar filter *nosqlplugin.VisibilityFilter\n\tif v.sortByCloseTime {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest:  request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:   nosqlplugin.ClosedByWorkflowType,\n\t\t\tSortType:     nosqlplugin.SortByClosedTime,\n\t\t\tWorkflowType: request.WorkflowTypeName,\n\t\t}\n\t} else {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest:  request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:   nosqlplugin.ClosedByWorkflowType,\n\t\t\tSortType:     nosqlplugin.SortByStartTime,\n\t\t\tWorkflowType: request.WorkflowTypeName,\n\t\t}\n\t}\n\tresp, err := v.db.SelectVisibility(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListClosedWorkflowExecutionsByType\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tresp, err := v.db.SelectVisibility(ctx, &nosqlplugin.VisibilityFilter{\n\t\tListRequest: request.InternalListWorkflowExecutionsRequest,\n\t\tFilterType:  nosqlplugin.OpenByWorkflowID,\n\t\tSortType:    nosqlplugin.SortByStartTime,\n\t\tWorkflowID:  request.WorkflowID,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListOpenWorkflowExecutionsByWorkflowID\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tvar filter *nosqlplugin.VisibilityFilter\n\tif v.sortByCloseTime {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:  nosqlplugin.ClosedByWorkflowID,\n\t\t\tSortType:    nosqlplugin.SortByClosedTime,\n\t\t\tWorkflowID:  request.WorkflowID,\n\t\t}\n\t} else {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:  nosqlplugin.ClosedByWorkflowID,\n\t\t\tSortType:    nosqlplugin.SortByStartTime,\n\t\t\tWorkflowID:  request.WorkflowID,\n\t\t}\n\t}\n\tresp, err := v.db.SelectVisibility(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListClosedWorkflowExecutionsByWorkflowID\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *persistence.InternalListClosedWorkflowExecutionsByStatusRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\tvar filter *nosqlplugin.VisibilityFilter\n\tif v.sortByCloseTime {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:  nosqlplugin.ClosedByClosedStatus,\n\t\t\tSortType:    nosqlplugin.SortByClosedTime,\n\t\t\tCloseStatus: int32(request.Status),\n\t\t}\n\t} else {\n\t\tfilter = &nosqlplugin.VisibilityFilter{\n\t\t\tListRequest: request.InternalListWorkflowExecutionsRequest,\n\t\t\tFilterType:  nosqlplugin.ClosedByClosedStatus,\n\t\t\tSortType:    nosqlplugin.SortByStartTime,\n\t\t\tCloseStatus: int32(request.Status),\n\t\t}\n\t}\n\tresp, err := v.db.SelectVisibility(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"ListClosedWorkflowExecutionsByStatus\", err)\n\t}\n\n\treturn &persistence.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    resp.Executions,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.InternalGetClosedWorkflowExecutionRequest,\n) (*persistence.InternalGetClosedWorkflowExecutionResponse, error) {\n\twfexecution, err := v.db.SelectOneClosedWorkflow(ctx, request.DomainUUID, request.Execution.GetWorkflowID(), request.Execution.GetRunID())\n\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(v.db, \"GetClosedWorkflowExecution\", err)\n\t}\n\tif wfexecution == nil {\n\t\t// Special case: this API return nil,nil if not found(since we will deprecate it, it's not worth refactor to be consistent)\n\t\treturn nil, &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\"Workflow execution not found.  WorkflowId: %v, RunId: %v\",\n\t\t\t\trequest.Execution.GetWorkflowID(), request.Execution.GetRunID()),\n\t\t}\n\t}\n\treturn &persistence.InternalGetClosedWorkflowExecutionResponse{\n\t\tExecution: wfexecution,\n\t}, nil\n}\n\nfunc (v *nosqlVisibilityStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\terr := v.db.DeleteVisibility(ctx, request.DomainID, request.WorkflowID, request.RunID)\n\tif err != nil {\n\t\treturn convertCommonErrors(v.db, \"DeleteWorkflowExecution\", err)\n\t}\n\treturn nil\n}\n\nfunc (v *nosqlVisibilityStore) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\t// temporary: not implemented, only implemented for ES\n\treturn nil\n}\n\nfunc (v *nosqlVisibilityStore) ListWorkflowExecutions(\n\t_ context.Context,\n\t_ *persistence.ListWorkflowExecutionsByQueryRequest,\n) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\treturn nil, persistence.ErrVisibilityOperationNotSupported\n}\n\nfunc (v *nosqlVisibilityStore) ScanWorkflowExecutions(\n\t_ context.Context,\n\t_ *persistence.ListWorkflowExecutionsByQueryRequest) (*persistence.InternalListWorkflowExecutionsResponse, error) {\n\treturn nil, persistence.ErrVisibilityOperationNotSupported\n}\n\nfunc (v *nosqlVisibilityStore) CountWorkflowExecutions(\n\t_ context.Context,\n\t_ *persistence.CountWorkflowExecutionsRequest,\n) (*persistence.CountWorkflowExecutionsResponse, error) {\n\treturn nil, persistence.ErrVisibilityOperationNotSupported\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosql_visibility_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttestDomainID         = \"test-domain-id\"\n\ttestDomainName       = \"test-domain\"\n\ttestTaskListName     = \"test-tasklist\"\n\ttestWorkflowID       = \"test-workflow-id\"\n\ttestRunID            = \"test-run-id\"\n\ttestWorkflowTypeName = \"test-workflow-type\"\n)\n\nfunc TestNewNoSQLVisibilityStore(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\n\tstore, err := newNoSQLVisibilityStore(false, cfg, log.NewNoop(), metrics.NewNoopMetricsClient(), nil)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, store)\n}\n\nfunc setupNoSQLVisibilityStoreMocks(t *testing.T) (*nosqlVisibilityStore, *nosqlplugin.MockDB) {\n\tctrl := gomock.NewController(t)\n\tdbMock := nosqlplugin.NewMockDB(ctrl)\n\n\tnosqlSt := nosqlStore{\n\t\tlogger: log.NewNoop(),\n\t\tdb:     dbMock,\n\t}\n\n\tshardedNosqlStoreMock := NewMockshardedNosqlStore(ctrl)\n\tshardedNosqlStoreMock.EXPECT().\n\t\tGetStoreShardByTaskList(\n\t\t\tTestDomainID,\n\t\t\tTestTaskListName,\n\t\t\tTestTaskType).\n\t\tReturn(&nosqlSt, nil).\n\t\tAnyTimes()\n\tshardedNosqlStoreMock.EXPECT().GetDefaultShard().Return(nosqlStore{db: dbMock}).AnyTimes()\n\tvisibilityStore := &nosqlVisibilityStore{\n\t\tnosqlStore:      shardedNosqlStoreMock.GetDefaultShard(),\n\t\tsortByCloseTime: false,\n\t}\n\treturn visibilityStore, dbMock\n}\n\nfunc TestRecordWorkflowExecutionStarted_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().InsertVisibility(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)\n\n\terr := visibilityStore.RecordWorkflowExecutionStarted(context.Background(), &persistence.InternalRecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainID,\n\t\tWorkflowID:       testWorkflowID,\n\t\tRunID:            testRunID,\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.NoError(t, err)\n}\n\nfunc TestRecordWorkflowExecutionStarted_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().InsertVisibility(gomock.Any(), gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t// The error _is_ a NotFoundError\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\terr := visibilityStore.RecordWorkflowExecutionStarted(context.Background(), &persistence.InternalRecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:         testDomainID,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tWorkflowTimeout:    20 * 60,\n\t\tStartTimestamp:     time.Time{},\n\t\tExecutionTimestamp: time.Time{},\n\t\tMemo:               nil,\n\t\tTaskList:           testTaskListName,\n\t\tIsCron:             false,\n\t\tNumClusters:        2,\n\t\tUpdateTimestamp:    time.Time{},\n\t\tShardID:            2,\n\t})\n\n\tassert.ErrorContains(t, err, \"RecordWorkflowExecutionStarted failed. Error:\")\n}\n\nfunc TestRecordWorkflowExecutionClosed_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().UpdateVisibility(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)\n\n\terr := visibilityStore.RecordWorkflowExecutionClosed(context.Background(), &persistence.InternalRecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainID,\n\t\tWorkflowID:       testWorkflowID,\n\t\tRunID:            testRunID,\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.NoError(t, err)\n}\nfunc TestRecordWorkflowExecutionClosed_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().UpdateVisibility(gomock.Any(), gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t// The error _is_ a NotFoundError\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\terr := visibilityStore.RecordWorkflowExecutionClosed(context.Background(), &persistence.InternalRecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:         testDomainID,\n\t\tWorkflowID:         testWorkflowID,\n\t\tRunID:              testRunID,\n\t\tWorkflowTypeName:   testWorkflowTypeName,\n\t\tStartTimestamp:     time.Time{},\n\t\tExecutionTimestamp: time.Time{},\n\t\tMemo:               nil,\n\t\tTaskList:           testTaskListName,\n\t\tIsCron:             false,\n\t\tNumClusters:        2,\n\t\tUpdateTimestamp:    time.Time{},\n\t\tShardID:            2,\n\t})\n\n\tassert.ErrorContains(t, err, \"RecordWorkflowExecutionClosed failed. Error:\")\n}\n\nfunc TestRecordWorkflowExecutionUninitialized(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\terr := visibilityStore.RecordWorkflowExecutionUninitialized(context.Background(), &persistence.InternalRecordWorkflowExecutionUninitializedRequest{})\n\tassert.NoError(t, err)\n}\n\nfunc TestUpsertWorkflowExecution(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\n\terr := visibilityStore.UpsertWorkflowExecution(context.Background(), &persistence.InternalUpsertWorkflowExecutionRequest{\n\t\tSearchAttributes: map[string][]byte{\n\t\t\tdefinition.CadenceChangeVersion: nil,\n\t\t},\n\t})\n\n\tassert.NoError(t, err)\n\n\terr = visibilityStore.UpsertWorkflowExecution(context.Background(), &persistence.InternalUpsertWorkflowExecutionRequest{})\n\n\tassert.Error(t, err)\n\tassert.Equal(t, persistence.ErrVisibilityOperationNotSupported, err)\n}\n\nfunc TestListOpenWorkflowExecutions_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t\tWorkflowID:   testWorkflowID,\n\t\t\tRunID:        testRunID,\n\t\t\tTypeName:     \"test-name\",\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListOpenWorkflowExecutions(context.Background(), &persistence.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainID,\n\t\tDomain:        testDomainName,\n\t\tEarliestTime:  time.Time{},\n\t\tLatestTime:    time.Time{},\n\t\tPageSize:      2,\n\t\tNextPageToken: nil,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListOpenWorkflowExecutions_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t// The error _is_ a NotFoundError\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\t_, err := visibilityStore.ListOpenWorkflowExecutions(context.Background(), &persistence.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainID,\n\t\tDomain:        testDomainName,\n\t\tEarliestTime:  time.Time{},\n\t\tLatestTime:    time.Time{},\n\t\tPageSize:      2,\n\t\tNextPageToken: nil,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListOpenWorkflowExecutions failed. Error:\")\n}\n\nfunc TestListClosedWorkflowExecutions_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\tvisibilityStore.sortByCloseTime = true\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t\tWorkflowID:   testWorkflowID,\n\t\t\tRunID:        testRunID,\n\t\t\tTypeName:     \"test-name\",\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListClosedWorkflowExecutions(context.Background(), &persistence.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainID,\n\t\tDomain:        testDomainName,\n\t\tEarliestTime:  time.Time{},\n\t\tLatestTime:    time.Time{},\n\t\tPageSize:      2,\n\t\tNextPageToken: nil,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListClosedWorkflowExecutions_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListClosedWorkflowExecutions(context.Background(), &persistence.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainID,\n\t\tDomain:        testDomainName,\n\t\tEarliestTime:  time.Time{},\n\t\tLatestTime:    time.Time{},\n\t\tPageSize:      2,\n\t\tNextPageToken: nil,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListClosedWorkflowExecutions failed. Error:\")\n}\n\nfunc TestListOpenWorkflowExecutionsByType_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListOpenWorkflowExecutionsByType(context.Background(), &persistence.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListOpenWorkflowExecutionsByType_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListOpenWorkflowExecutionsByType(context.Background(), &persistence.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListOpenWorkflowExecutionsByType failed. Error:\")\n}\n\nfunc TestListClosedWorkflowExecutionsByType_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\tvisibilityStore.sortByCloseTime = true\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListClosedWorkflowExecutionsByType(context.Background(), &persistence.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, 1, len(response.Executions))\n}\n\nfunc TestListClosedWorkflowExecutionsByType_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListClosedWorkflowExecutionsByType(context.Background(), &persistence.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListClosedWorkflowExecutionsByType failed. Error:\")\n}\n\nfunc TestListOpenWorkflowExecutionsByWorkflowID_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), &persistence.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListOpenWorkflowExecutionsByWorkflowID_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), &persistence.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListOpenWorkflowExecutionsByWorkflowID failed. Error:\")\n}\n\nfunc TestListClosedWorkflowExecutionsByWorkflowID_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\tvisibilityStore.sortByCloseTime = true\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), &persistence.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListClosedWorkflowExecutionsByWorkflowID_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), &persistence.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListClosedWorkflowExecutionsByWorkflowID failed. Error:\")\n}\n\nfunc TestListClosedWorkflowExecutionsByStatus_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\tvisibilityStore.sortByCloseTime = true\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(&nosqlplugin.SelectVisibilityResponse{\n\t\tExecutions: []*nosqlplugin.VisibilityRow{{\n\t\t\tDomainID:     testDomainID,\n\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t}},\n\t\tNextPageToken: nil,\n\t}, nil)\n\n\tresponse, err := visibilityStore.ListClosedWorkflowExecutionsByStatus(context.Background(), &persistence.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tStatus: 0,\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, len(response.Executions), 1)\n}\n\nfunc TestListClosedWorkflowExecutionsByStatus_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectVisibility(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.ListClosedWorkflowExecutionsByStatus(context.Background(), &persistence.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\tInternalListWorkflowExecutionsRequest: persistence.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID: testDomainID,\n\t\t},\n\t\tStatus: 0,\n\t})\n\n\tassert.ErrorContains(t, err, \"ListClosedWorkflowExecutionsByStatus failed. Error:\")\n}\n\nfunc TestGetClosedWorkflowExecution_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectOneClosedWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).Return(&nosqlplugin.VisibilityRow{\n\t\tDomainID:     testDomainID,\n\t\tWorkflowType: testWorkflowTypeName,\n\t\tWorkflowID:   testWorkflowID,\n\t\tRunID:        testRunID,\n\t\tTypeName:     \"test-name\",\n\t}, nil)\n\n\tresponse, err := visibilityStore.GetClosedWorkflowExecution(context.Background(), &persistence.InternalGetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testWorkflowID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t})\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, testWorkflowID, response.Execution.WorkflowID)\n}\n\nfunc TestGetClosedWorkflowExecution_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectOneClosedWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).Return(nil, assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\t_, err := visibilityStore.GetClosedWorkflowExecution(context.Background(), &persistence.InternalGetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testWorkflowID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t})\n\n\tassert.ErrorContains(t, err, \"GetClosedWorkflowExecution failed. Error:\")\n}\n\nfunc TestGetClosedWorkflowExecution_Failed_Special_Case(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().SelectOneClosedWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).Return(nil, nil)\n\n\t_, err := visibilityStore.GetClosedWorkflowExecution(context.Background(), &persistence.InternalGetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testWorkflowID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t})\n\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"Workflow execution not found.  WorkflowId: %v, RunId: %v\", testWorkflowID, testRunID))\n}\n\nfunc TestDeleteWorkflowExecution_Success(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().DeleteVisibility(gomock.Any(), testDomainID, testWorkflowID, testRunID).Return(nil)\n\n\terr := visibilityStore.DeleteWorkflowExecution(context.Background(), &persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\tDomainID:   testDomainID,\n\t\tDomain:     testDomainName,\n\t\tRunID:      testRunID,\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.NoError(t, err)\n}\n\nfunc TestDeleteWorkflowExecution_Failed(t *testing.T) {\n\tvisibilityStore, db := setupNoSQLVisibilityStoreMocks(t)\n\n\tdb.EXPECT().DeleteVisibility(gomock.Any(), testDomainID, testWorkflowID, testRunID).Return(assert.AnError)\n\tdb.EXPECT().IsNotFoundError(assert.AnError).Return(true)\n\n\terr := visibilityStore.DeleteWorkflowExecution(context.Background(), &persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\tDomainID:   testDomainID,\n\t\tDomain:     testDomainName,\n\t\tRunID:      testRunID,\n\t\tWorkflowID: testWorkflowID,\n\t})\n\n\tassert.ErrorContains(t, err, \"DeleteWorkflowExecution failed. Error:\")\n}\n\nfunc TestDeleteUninitializedWorkflowExecution(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\terr := visibilityStore.DeleteUninitializedWorkflowExecution(context.Background(), &persistence.VisibilityDeleteWorkflowExecutionRequest{})\n\tassert.NoError(t, err)\n}\n\nfunc TestListWorkflowExecutions(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\t_, err := visibilityStore.ListWorkflowExecutions(context.Background(), &persistence.ListWorkflowExecutionsByQueryRequest{})\n\tassert.Error(t, err)\n\tassert.Equal(t, persistence.ErrVisibilityOperationNotSupported, err)\n}\n\nfunc TestScanWorkflowExecutions(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\t_, err := visibilityStore.ScanWorkflowExecutions(context.Background(), &persistence.ListWorkflowExecutionsByQueryRequest{})\n\tassert.Error(t, err)\n\tassert.Equal(t, persistence.ErrVisibilityOperationNotSupported, err)\n}\n\nfunc TestCountWorkflowExecutions(t *testing.T) {\n\tvisibilityStore, _ := setupNoSQLVisibilityStoreMocks(t)\n\t_, err := visibilityStore.CountWorkflowExecutions(context.Background(), &persistence.CountWorkflowExecutionsRequest{})\n\tassert.Error(t, err)\n\tassert.Equal(t, persistence.ErrVisibilityOperationNotSupported, err)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/admin.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\nconst (\n\ttestSchemaDir = \"schema/cassandra/\"\n)\n\nvar _ nosqlplugin.AdminDB = (*CDB)(nil)\n\nfunc (db *CDB) SetupTestDatabase(schemaBaseDir string, replicas int) error {\n\terr := db.createCassandraKeyspace(db.session, db.cfg.Keyspace, replicas, true)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif schemaBaseDir == \"\" {\n\t\tvar err error\n\t\tschemaBaseDir, err = nosqlplugin.GetDefaultTestSchemaDir(testSchemaDir)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\terr = db.loadSchema([]string{\"schema.cql\"}, schemaBaseDir)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = db.loadVisibilitySchema([]string{\"schema.cql\"}, schemaBaseDir)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// loadSchema from PersistenceTestCluster interface\nfunc (db *CDB) loadSchema(fileNames []string, schemaBaseDir string) error {\n\tworkflowSchemaDir := schemaBaseDir + \"/cadence\"\n\terr := loadCassandraSchema(workflowSchemaDir, fileNames, db.cfg.Hosts, db.cfg.Port, db.cfg.Keyspace, true, nil, db.cfg.ProtoVersion)\n\tif err != nil && !strings.Contains(err.Error(), \"AlreadyExists\") {\n\t\t// TODO: should we remove the second condition?\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// loadVisibilitySchema from PersistenceTestCluster interface\nfunc (db *CDB) loadVisibilitySchema(fileNames []string, schemaBaseDir string) error {\n\tworkflowSchemaDir := schemaBaseDir + \"/visibility\"\n\terr := loadCassandraSchema(workflowSchemaDir, fileNames, db.cfg.Hosts, db.cfg.Port, db.cfg.Keyspace, false, nil, db.cfg.ProtoVersion)\n\tif err != nil && !strings.Contains(err.Error(), \"AlreadyExists\") {\n\t\t// TODO: should we remove the second condition?\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (db *CDB) TeardownTestDatabase() error {\n\terr := db.dropCassandraKeyspace(db.session, db.cfg.Keyspace)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdb.session.Close()\n\treturn nil\n}\n\n// createCassandraKeyspace creates the keyspace using this session for given replica count\nfunc (db *CDB) createCassandraKeyspace(s gocql.Session, keyspace string, replicas int, overwrite bool) (err error) {\n\t// if overwrite flag is set, drop the keyspace and create a new one\n\tif overwrite {\n\t\terr = db.dropCassandraKeyspace(s, keyspace)\n\t\tif err != nil {\n\t\t\tdb.logger.Error(`drop keyspace error`, tag.Error(err))\n\t\t\treturn\n\t\t}\n\t}\n\terr = s.Query(fmt.Sprintf(`CREATE KEYSPACE IF NOT EXISTS %s WITH replication = {\n\t\t'class' : 'SimpleStrategy', 'replication_factor' : %d}`, keyspace, replicas)).Exec()\n\tif err != nil {\n\t\tdb.logger.Error(`create keyspace error`, tag.Error(err))\n\t\treturn\n\t}\n\tdb.logger.Debugf(\"created namespace %s with replication factor %d\", keyspace, replicas)\n\n\treturn\n}\n\n// dropCassandraKeyspace drops the given keyspace, if it exists\nfunc (db *CDB) dropCassandraKeyspace(s gocql.Session, keyspace string) (err error) {\n\terr = s.Query(fmt.Sprintf(\"DROP KEYSPACE IF EXISTS %s\", keyspace)).Exec()\n\tif err != nil {\n\t\tdb.logger.Error(`drop keyspace error`, tag.Error(err))\n\t\treturn\n\t}\n\tdb.logger.Debugf(\"dropped namespace %s\", keyspace)\n\treturn\n}\n\n// loadCassandraSchema loads the schema from the given .cql files on this keyspace\nfunc loadCassandraSchema(\n\tdir string,\n\tfileNames []string,\n\thosts string,\n\tport int,\n\tkeyspace string,\n\toverride bool,\n\ttls *config.TLS,\n\tprotoVersion int,\n) (err error) {\n\n\ttmpFile, err := ioutil.TempFile(\"\", \"_cadence_\")\n\tif err != nil {\n\t\treturn fmt.Errorf(\"error creating tmp file:%v\", err.Error())\n\t}\n\tdefer os.Remove(tmpFile.Name())\n\n\tfor _, file := range fileNames {\n\t\t// Flagged for potential file inclusion via variable. No user supplied input is included here - this just reads\n\t\t// schema files.\n\t\t// #nosec\n\t\tcontent, err := ioutil.ReadFile(dir + \"/\" + file)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"error reading contents of file %v:%v\", file, err.Error())\n\t\t}\n\t\t_, err = tmpFile.WriteString(string(content) + \"\\n\")\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"error writing string to file, err: %v\", err.Error())\n\t\t}\n\t}\n\n\ttmpFile.Close()\n\n\tconfig := &cassandra.SetupSchemaConfig{\n\t\tCQLClientConfig: cassandra.CQLClientConfig{\n\t\t\tHosts:        hosts,\n\t\t\tPort:         port,\n\t\t\tKeyspace:     keyspace,\n\t\t\tTLS:          tls,\n\t\t\tProtoVersion: protoVersion,\n\t\t},\n\t\tSetupConfig: schema.SetupConfig{\n\t\t\tSchemaFilePath:    tmpFile.Name(),\n\t\t\tOverwrite:         override,\n\t\t\tDisableVersioning: true,\n\t\t},\n\t}\n\n\terr = cassandra.SetupSchema(config)\n\tif err != nil {\n\t\terr = fmt.Errorf(\"error loading schema:%v\", err.Error())\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/cluster_config.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc (db *CDB) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tquery := db.session.Query(templateInsertConfig, row.RowType, row.Version, row.Timestamp, row.Values.Data, row.Values.Encoding).WithContext(ctx)\n\tapplied, err := query.MapScanCAS(make(map[string]interface{}))\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !applied {\n\t\treturn nosqlplugin.NewConditionFailure(\"InsertConfig operation failed because of version collision\")\n\t}\n\treturn nil\n}\n\nfunc (db *CDB) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tvar version int64\n\tvar timestamp time.Time\n\tvar data []byte\n\tvar encoding constants.EncodingType\n\n\tquery := db.session.Query(templateSelectLatestConfig, rowType).WithContext(ctx)\n\terr := query.Scan(&rowType, &version, &timestamp, &data, &encoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &persistence.InternalConfigStoreEntry{\n\t\tRowType:   rowType,\n\t\tVersion:   version,\n\t\tTimestamp: timestamp,\n\t\tValues: &persistence.DataBlob{\n\t\t\tData:     data,\n\t\t\tEncoding: encoding,\n\t\t},\n\t}, err\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/cluster_config_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\t// version is the clustering key(DESC order) so this query will always return the record with largest version\n\ttemplateSelectLatestConfig = `SELECT row_type, version, timestamp, values, encoding FROM cluster_config ` +\n\t\t`WHERE row_type = ? ` +\n\t\t`LIMIT 1;`\n\n\ttemplateInsertConfig = `INSERT INTO cluster_config (row_type, version, timestamp, values, encoding) ` +\n\t\t`VALUES (?, ?, ?, ?, ?) ` +\n\t\t`IF NOT EXISTS;`\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/constants.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"time\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nvar (\n\tdefaultDateTime            = time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC)\n\tdefaultVisibilityTimestamp = p.UnixNanoToDBTimestamp(defaultDateTime.UnixNano())\n)\n\nconst (\n\t// Cassandra max support time is 2038-01-19T03:14:06+00:00. Updated this to 5 years to support until year 2033\n\t// See https://github.com/uber/cadence/issues/4200\n\tmaxCassandraTTL = int64(157680000)\n\n\t// We use local serial consistency level as the default consistency level for conditional updates\n\tcassandraDefaultSerialConsLevel = gocql.LocalSerial\n\n\t// We use local quorum consistency level as the default consistency level\n\tcassandraDefaultConsLevel = gocql.LocalQuorum\n\n\t// Although Cadence core data models always require strong consistency, reading visibility is a special case that\n\t// eventual consistency is sufficient.\n\t// That's because the engine layer writes into visibility with eventual consistency anyway(using transfer tasks)\n\t// Do NOT use it in other places, unless you are sure it's the same special cases like reading visibility\n\tcassandraLowConslevel = gocql.LocalOne\n\n\t// We use all consistency level for delete operations to prevent the data resurrection issue\n\tcassandraAllConslevel = gocql.All\n)\n\nconst (\n\t// Row types for table executions\n\trowTypeShard = iota\n\trowTypeExecution\n\trowTypeTransferTask\n\trowTypeTimerTask\n\trowTypeReplicationTask\n\trowTypeDLQ\n\trowTypeCrossClusterTask\n\trowTypeWorkflowRequestStart\n\trowTypeWorkflowRequestSignal\n\trowTypeWorkflowRequestCancel\n\trowTypeWorkflowRequestReset\n\trowTypeWorkflowActiveClusterSelectionPolicy\n)\n\n// Guidelines for creating new special UUID constants\n// Each UUID should be of the form: E0000000-R000-f000-f000-00000000000x\n// Where x is any hexadecimal value, E represents the entity type valid values are:\n// E = {DomainID = 1, WorkflowID = 2, RunID = 3}\n// R represents row type in executions table, valid values are:\n// R = {Shard = 0, Execution = 1, Transfer = 2, Timer = 3, Replication = 4, DLQ = 5, CrossCluster = 6}\nconst (\n\t// Special Domains related constants\n\temptyDomainID = \"10000000-0000-f000-f000-000000000000\"\n\t// Special Run IDs\n\temptyRunID     = \"30000000-0000-f000-f000-000000000000\"\n\tpermanentRunID = \"30000000-0000-f000-f000-000000000001\"\n\t// Row Constants for Shard Row\n\trowTypeShardDomainID   = \"10000000-1000-f000-f000-000000000000\"\n\trowTypeShardWorkflowID = \"20000000-1000-f000-f000-000000000000\"\n\trowTypeShardRunID      = \"30000000-1000-f000-f000-000000000000\"\n\t// Row Constants for Transfer Task Row\n\trowTypeTransferDomainID   = \"10000000-3000-f000-f000-000000000000\"\n\trowTypeTransferWorkflowID = \"20000000-3000-f000-f000-000000000000\"\n\trowTypeTransferRunID      = \"30000000-3000-f000-f000-000000000000\"\n\t// Row Constants for Cross Cluster Task Row\n\trowTypeCrossClusterDomainID = \"10000000-7000-f000-f000-000000000000\"\n\trowTypeCrossClusterRunID    = \"30000000-7000-f000-f000-000000000000\"\n\t// Row Constants for Timer Task Row\n\trowTypeTimerDomainID   = \"10000000-4000-f000-f000-000000000000\"\n\trowTypeTimerWorkflowID = \"20000000-4000-f000-f000-000000000000\"\n\trowTypeTimerRunID      = \"30000000-4000-f000-f000-000000000000\"\n\t// Row Constants for Replication Task Row\n\trowTypeReplicationDomainID   = \"10000000-5000-f000-f000-000000000000\"\n\trowTypeReplicationWorkflowID = \"20000000-5000-f000-f000-000000000000\"\n\trowTypeReplicationRunID      = \"30000000-5000-f000-f000-000000000000\"\n\t// Row Constants for Replication Task DLQ Row. Source cluster name will be used as WorkflowID.\n\trowTypeDLQDomainID = \"10000000-6000-f000-f000-000000000000\"\n\trowTypeDLQRunID    = \"30000000-6000-f000-f000-000000000000\"\n\t// Special TaskId constants\n\trowTypeExecutionTaskID                       = int64(-10)\n\trowTypeShardTaskID                           = int64(-11)\n\temptyInitiatedID                             = int64(-7)\n\temptyWorkflowRequestVersion                  = int64(-1000)\n\trowTypeWorkflowActiveClusterSelectionVersion = int64(-1001)\n\tworkflowRequestTTLInSeconds                  = 10800\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/db.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\n// CDB represents a logical connection to Cassandra database\ntype CDB struct {\n\tlogger  log.Logger\n\tclient  gocql.Client\n\tsession gocql.Session\n\tcfg     *config.NoSQL\n\tdc      *persistence.DynamicConfiguration\n}\n\nvar _ nosqlplugin.DB = (*CDB)(nil)\n\n// CassandraDBOption is used to provide optional settings for CDB object\ntype CassandraDBOption func(*CDB)\n\n// DbWithClient returns a CDB option to set the gocql client.\n// If this is not used then the globally registered client is used.\nfunc DbWithClient(client gocql.Client) CassandraDBOption {\n\treturn func(db *CDB) {\n\t\tdb.client = client\n\t}\n}\n\n// NewCassandraDBFromSession returns a DB from a session\nfunc NewCassandraDBFromSession(\n\tcfg *config.NoSQL,\n\tsession gocql.Session,\n\tlogger log.Logger,\n\tdc *persistence.DynamicConfiguration,\n\topts ...CassandraDBOption,\n) *CDB {\n\tres := &CDB{\n\t\tsession: session,\n\t\tlogger:  logger,\n\t\tcfg:     cfg,\n\t\tdc:      dc,\n\t}\n\n\tfor _, opt := range opts {\n\t\topt(res)\n\t}\n\n\tif res.client == nil {\n\t\tres.client = gocql.GetRegisteredClient()\n\t}\n\n\treturn res\n}\n\nfunc (db *CDB) Close() {\n\tif db.session != nil {\n\t\tdb.session.Close()\n\t}\n}\n\nfunc (db *CDB) PluginName() string {\n\treturn PluginName\n}\n\nfunc (db *CDB) IsNotFoundError(err error) bool {\n\treturn db.client.IsNotFoundError(err)\n}\n\nfunc (db *CDB) IsTimeoutError(err error) bool {\n\treturn db.client.IsTimeoutError(err)\n}\n\nfunc (db *CDB) IsThrottlingError(err error) bool {\n\treturn db.client.IsThrottlingError(err)\n}\n\nfunc (db *CDB) IsDBUnavailableError(err error) bool {\n\treturn db.client.IsDBUnavailableError(err)\n}\n\nfunc (db *CDB) isCassandraConsistencyError(err error) bool {\n\treturn db.client.IsCassandraConsistencyError(err)\n}\n\nfunc (db *CDB) executeWithConsistencyAll(q gocql.Query) error {\n\tif db.dc != nil && db.dc.EnableCassandraAllConsistencyLevelDelete() {\n\t\tif err := q.Consistency(cassandraAllConslevel).Exec(); err != nil {\n\t\t\tif db.isCassandraConsistencyError(err) {\n\t\t\t\tdb.logger.Warn(\"unable to complete the delete operation due to consistency issue\", tag.Error(err))\n\t\t\t\treturn q.Consistency(cassandraDefaultConsLevel).Exec()\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\treturn q.Exec()\n}\n\nfunc (db *CDB) executeBatchWithConsistencyAll(b gocql.Batch) error {\n\tif db.dc != nil && db.dc.EnableCassandraAllConsistencyLevelDelete() {\n\t\tif err := db.session.ExecuteBatch(b.Consistency(cassandraAllConslevel)); err != nil {\n\t\t\tif db.isCassandraConsistencyError(err) {\n\t\t\t\tdb.logger.Warn(\"unable to complete the delete operation due to consistency issue\", tag.Error(err))\n\t\t\t\treturn db.session.ExecuteBatch(b.Consistency(cassandraDefaultConsLevel))\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\treturn db.session.ExecuteBatch(b)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/db_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nfunc TestCDBBasics(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tsession := gocql.NewMockSession(ctrl)\n\tclient := gocql.NewMockClient(ctrl)\n\tcfg := &config.NoSQL{}\n\tlogger := testlogger.New(t)\n\tdc := &persistence.DynamicConfiguration{}\n\n\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\tif db.PluginName() != PluginName {\n\t\tt.Errorf(\"got plugin name: %v but want %v\", db.PluginName(), PluginName)\n\t}\n\n\tclient.EXPECT().IsNotFoundError(nil).Return(true).Times(1)\n\tif db.IsNotFoundError(nil) != true {\n\t\tt.Errorf(\"IsNotFoundError returned false but want true\")\n\t}\n\n\tclient.EXPECT().IsTimeoutError(nil).Return(true).Times(1)\n\tif db.IsTimeoutError(nil) != true {\n\t\tt.Errorf(\"IsTimeoutError returned false but want true\")\n\t}\n\n\tclient.EXPECT().IsThrottlingError(nil).Return(true).Times(1)\n\tif db.IsThrottlingError(nil) != true {\n\t\tt.Errorf(\"IsNotFoundError returned false but want true\")\n\t}\n\n\tclient.EXPECT().IsDBUnavailableError(nil).Return(true).Times(1)\n\tif db.IsDBUnavailableError(nil) != true {\n\t\tt.Errorf(\"IsDBUnavailableError returned false but want true\")\n\t}\n\n\tclient.EXPECT().IsCassandraConsistencyError(nil).Return(true).Times(1)\n\tif db.isCassandraConsistencyError(nil) != true {\n\t\tt.Errorf(\"IsNotFoundError returned false but want true\")\n\t}\n\n\tsession.EXPECT().Close().Times(1)\n\tdb.Close()\n}\n\nfunc TestExecuteWithConsistencyAll(t *testing.T) {\n\ttests := []struct {\n\t\tname                            string\n\t\tenableAllConsistencyLevelDelete bool\n\t\tqueryMockPrep                   func(*gocql.MockQuery)\n\t\tclientMockPrep                  func(*gocql.MockClient)\n\t\twantErr                         bool\n\t}{\n\t\t{\n\t\t\tname:                            \"all consistency level delete disabled - success\",\n\t\t\tenableAllConsistencyLevelDelete: false,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete disabled - failed\",\n\t\t\tenableAllConsistencyLevelDelete: false,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - query consistency level all - success\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - query consistency level all - returns consistency error - success\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"fail for the first call\")).Times(1)\n\n\t\t\t\tquery.EXPECT().Consistency(cassandraDefaultConsLevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\tclientMockPrep: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - query consistency level all - returns consistency error - failed\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"fail for the first call\")).Times(1)\n\n\t\t\t\tquery.EXPECT().Consistency(cassandraDefaultConsLevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"fail for the 2nd call too\")).Times(1)\n\t\t\t},\n\t\t\tclientMockPrep: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - query consistency level all - returns non-consistency error - failed\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tqueryMockPrep: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\tclientMockPrep: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tsession := gocql.NewMockSession(ctrl)\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{\n\t\t\t\tEnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn tc.enableAllConsistencyLevelDelete\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tif tc.queryMockPrep != nil {\n\t\t\t\ttc.queryMockPrep(query)\n\t\t\t}\n\t\t\tif tc.clientMockPrep != nil {\n\t\t\t\ttc.clientMockPrep(client)\n\t\t\t}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.executeWithConsistencyAll(query)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"got err: %v but wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestExecuteBatchWithConsistencyAll(t *testing.T) {\n\ttests := []struct {\n\t\tname                            string\n\t\tenableAllConsistencyLevelDelete bool\n\t\tsessionMockPrep                 func(*gocql.MockSession)\n\t\tbatchMockPrep                   func(*gocql.MockBatch)\n\t\tclientMockPrep                  func(*gocql.MockClient)\n\t\twantErr                         bool\n\t}{\n\t\t{\n\t\t\tname:                            \"all consistency level delete disabled - success\",\n\t\t\tenableAllConsistencyLevelDelete: false,\n\t\t\tsessionMockPrep: func(session *gocql.MockSession) {\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete disabled - failed\",\n\t\t\tenableAllConsistencyLevelDelete: false,\n\t\t\tsessionMockPrep: func(session *gocql.MockSession) {\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - success\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tsessionMockPrep: func(session *gocql.MockSession) {\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tbatchMockPrep: func(batch *gocql.MockBatch) {\n\t\t\t\tbatch.EXPECT().Consistency(cassandraAllConslevel).Return(batch).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - executebatch(all) failed with consistency err - success\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tsessionMockPrep: func(session *gocql.MockSession) {\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(errors.New(\"fail the first call\")).Times(1)\n\t\t\t\t// second call succeeds\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tclientMockPrep: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\tbatchMockPrep: func(batch *gocql.MockBatch) {\n\t\t\t\tbatch.EXPECT().Consistency(cassandraAllConslevel).Return(batch).Times(1)\n\t\t\t\tbatch.EXPECT().Consistency(cassandraDefaultConsLevel).Return(batch).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                            \"all consistency level delete enabled - executebatch(all) failed with non-consistency err - failed\",\n\t\t\tenableAllConsistencyLevelDelete: true,\n\t\t\tsessionMockPrep: func(session *gocql.MockSession) {\n\t\t\t\tsession.EXPECT().ExecuteBatch(gomock.Any()).Return(errors.New(\"fail the first call\")).Times(1)\n\t\t\t},\n\t\t\tclientMockPrep: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\tbatchMockPrep: func(batch *gocql.MockBatch) {\n\t\t\t\tbatch.EXPECT().Consistency(cassandraAllConslevel).Return(batch).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tsession := gocql.NewMockSession(ctrl)\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tbatch := gocql.NewMockBatch(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{\n\t\t\t\tEnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn tc.enableAllConsistencyLevelDelete\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tif tc.sessionMockPrep != nil {\n\t\t\t\ttc.sessionMockPrep(session)\n\t\t\t}\n\t\t\tif tc.batchMockPrep != nil {\n\t\t\t\ttc.batchMockPrep(batch)\n\t\t\t}\n\t\t\tif tc.clientMockPrep != nil {\n\t\t\t\ttc.clientMockPrep(client)\n\t\t\t}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.executeBatchWithConsistencyAll(batch)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"got err: %v but wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/domain.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tconstDomainPartition     = 0\n\tdomainMetadataRecordName = \"cadence-domain-metadata\"\n\temptyFailoverEndTime     = int64(0)\n)\n\n// Insert a new record to domain\n// return types.DomainAlreadyExistsError error if failed or already exists\n// Must return ConditionFailure error if other condition doesn't match\nfunc (db *CDB) InsertDomain(ctx context.Context, row *nosqlplugin.DomainRow) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tquery := db.session.Query(templateCreateDomainQuery, row.Info.ID, row.Info.Name, timeStamp).WithContext(ctx)\n\tapplied, err := query.MapScanCAS(make(map[string]interface{}))\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !applied {\n\t\treturn fmt.Errorf(\"CreateDomain operation failed because of uuid collision\")\n\t}\n\n\tmetadataNotificationVersion, err := db.SelectDomainMetadata(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\tfailoverEndTime := emptyFailoverEndTime\n\tif row.FailoverEndTime != nil {\n\t\tfailoverEndTime = row.FailoverEndTime.UnixNano()\n\t}\n\tisolationGroupData, isolationGroupEncoding := getIsolationGroupFields(row)\n\tasyncWFConfigData, asyncWFConfigEncoding := getAsyncWFConfigFields(row)\n\tactiveClustersData, activeClustersEncoding := getActiveClustersFields(row)\n\n\tbatch.Query(templateCreateDomainByNameQueryWithinBatchV2,\n\t\tconstDomainPartition,\n\t\trow.Info.Name,\n\t\trow.Info.ID,\n\t\trow.Info.Name,\n\t\trow.Info.Status,\n\t\trow.Info.Description,\n\t\trow.Info.OwnerEmail,\n\t\trow.Info.Data,\n\t\tcommon.DurationToDays(row.Config.Retention),\n\t\trow.Config.EmitMetric,\n\t\trow.Config.ArchivalBucket,\n\t\trow.Config.ArchivalStatus,\n\t\trow.Config.HistoryArchivalStatus,\n\t\trow.Config.HistoryArchivalURI,\n\t\trow.Config.VisibilityArchivalStatus,\n\t\trow.Config.VisibilityArchivalURI,\n\t\trow.Config.BadBinaries.Data,\n\t\tstring(row.Config.BadBinaries.Encoding),\n\t\tisolationGroupData,\n\t\tisolationGroupEncoding,\n\t\tasyncWFConfigData,\n\t\tasyncWFConfigEncoding,\n\t\trow.ReplicationConfig.ActiveClusterName,\n\t\tpersistence.SerializeClusterConfigs(row.ReplicationConfig.Clusters),\n\t\tactiveClustersData,\n\t\tactiveClustersEncoding,\n\t\trow.IsGlobalDomain,\n\t\trow.ConfigVersion,\n\t\trow.FailoverVersion,\n\t\tpersistence.InitialFailoverNotificationVersion,\n\t\tconstants.InitialPreviousFailoverVersion,\n\t\tfailoverEndTime,\n\t\trow.LastUpdatedTime.UnixNano(),\n\t\tmetadataNotificationVersion,\n\t\ttimeStamp,\n\t)\n\tdb.updateMetadataBatch(batch, metadataNotificationVersion)\n\n\tprevious := make(map[string]interface{})\n\tapplied, iter, err := db.session.MapExecuteBatchCAS(batch, previous)\n\tdefer func() {\n\t\tif iter != nil {\n\t\t\t_ = iter.Close()\n\t\t}\n\t}()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\t// Domain already exist.  Delete orphan domain record before returning back to user\n\t\tif errDelete := db.session.Query(templateDeleteDomainQuery, row.Info.ID).WithContext(ctx).Exec(); errDelete != nil {\n\t\t\tdb.logger.Warn(\"Unable to delete orphan domain record. Error\", tag.Error(errDelete))\n\t\t}\n\n\t\tfor {\n\t\t\t// first iter MapScan is done inside MapExecuteBatchCAS\n\t\t\tif domain, ok := previous[\"name\"].(string); ok && domain == row.Info.Name {\n\t\t\t\tdb.logger.Warn(\"Domain already exists\", tag.WorkflowDomainName(domain))\n\t\t\t\treturn &types.DomainAlreadyExistsError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"Domain %v already exists\", previous[\"domain\"]),\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tprevious = make(map[string]interface{})\n\t\t\tif !iter.MapScan(previous) {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tdb.logger.Warn(\"Create domain operation failed because of condition update failure on domain metadata record\")\n\t\treturn nosqlplugin.NewConditionFailure(\"domain\")\n\t}\n\n\treturn nil\n}\n\nfunc (db *CDB) updateMetadataBatch(\n\tbatch gocql.Batch,\n\tnotificationVersion int64,\n) {\n\tvar nextVersion int64 = 1\n\tvar currentVersion *int64\n\tif notificationVersion > 0 {\n\t\tnextVersion = notificationVersion + 1\n\t\tcurrentVersion = &notificationVersion\n\t}\n\tbatch.Query(templateUpdateMetadataQueryWithinBatchV2,\n\t\tnextVersion,\n\t\tconstDomainPartition,\n\t\tdomainMetadataRecordName,\n\t\tcurrentVersion,\n\t)\n}\n\n// Update domain\nfunc (db *CDB) UpdateDomain(ctx context.Context, row *nosqlplugin.DomainRow) error {\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\tfailoverEndTime := emptyFailoverEndTime\n\tif row.FailoverEndTime != nil {\n\t\tfailoverEndTime = row.FailoverEndTime.UnixNano()\n\t}\n\n\tisolationGroupData, isolationGroupEncoding := getIsolationGroupFields(row)\n\tasyncWFConfigData, asyncWFConfigEncoding := getAsyncWFConfigFields(row)\n\tactiveClustersData, activeClustersEncoding := getActiveClustersFields(row)\n\n\tbatch.Query(templateUpdateDomainByNameQueryWithinBatchV2,\n\t\trow.Info.ID,\n\t\trow.Info.Name,\n\t\trow.Info.Status,\n\t\trow.Info.Description,\n\t\trow.Info.OwnerEmail,\n\t\trow.Info.Data,\n\t\tcommon.DurationToDays(row.Config.Retention),\n\t\trow.Config.EmitMetric,\n\t\trow.Config.ArchivalBucket,\n\t\trow.Config.ArchivalStatus,\n\t\trow.Config.HistoryArchivalStatus,\n\t\trow.Config.HistoryArchivalURI,\n\t\trow.Config.VisibilityArchivalStatus,\n\t\trow.Config.VisibilityArchivalURI,\n\t\trow.Config.BadBinaries.Data,\n\t\tstring(row.Config.BadBinaries.Encoding),\n\t\tisolationGroupData,\n\t\tisolationGroupEncoding,\n\t\tasyncWFConfigData,\n\t\tasyncWFConfigEncoding,\n\t\trow.ReplicationConfig.ActiveClusterName,\n\t\tpersistence.SerializeClusterConfigs(row.ReplicationConfig.Clusters),\n\t\tactiveClustersData,\n\t\tactiveClustersEncoding,\n\t\trow.ConfigVersion,\n\t\trow.FailoverVersion,\n\t\trow.FailoverNotificationVersion,\n\t\trow.PreviousFailoverVersion,\n\t\tfailoverEndTime,\n\t\trow.LastUpdatedTime.UnixNano(),\n\t\trow.NotificationVersion,\n\t\tconstDomainPartition,\n\t\trow.Info.Name,\n\t)\n\tdb.updateMetadataBatch(batch, row.NotificationVersion)\n\n\tprevious := make(map[string]interface{})\n\tapplied, iter, err := db.session.MapExecuteBatchCAS(batch, previous)\n\tdefer func() {\n\t\tif iter != nil {\n\t\t\titer.Close()\n\t\t}\n\t}()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !applied {\n\t\treturn nosqlplugin.NewConditionFailure(\"domain\")\n\t}\n\treturn nil\n}\n\n// Get one domain data, either by domainID or domainName\nfunc (db *CDB) SelectDomain(\n\tctx context.Context,\n\tdomainID *string,\n\tdomainName *string,\n) (*nosqlplugin.DomainRow, error) {\n\tif domainID != nil && domainName != nil {\n\t\treturn nil, fmt.Errorf(\"GetDomain operation failed.  Both ID and Name specified in request\")\n\t} else if domainID == nil && domainName == nil {\n\t\treturn nil, fmt.Errorf(\"GetDomain operation failed.  Both ID and Name are empty\")\n\t}\n\n\tvar query gocql.Query\n\tvar err error\n\tif domainID != nil {\n\t\tquery = db.session.Query(templateGetDomainQuery, domainID).WithContext(ctx)\n\t\terr = query.Scan(&domainName)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tinfo := &persistence.DomainInfo{}\n\tconfig := &persistence.InternalDomainConfig{}\n\treplicationConfig := &persistence.InternalDomainReplicationConfig{}\n\n\t// because of encoding/types, we can't directly read from config struct\n\tvar badBinariesData []byte\n\tvar badBinariesDataEncoding string\n\tvar replicationClusters []map[string]interface{}\n\n\tvar failoverNotificationVersion int64\n\tvar notificationVersion int64\n\tvar failoverVersion int64\n\tvar previousFailoverVersion int64\n\tvar failoverEndTime int64\n\tvar lastUpdatedTime int64\n\tvar configVersion int64\n\tvar isGlobalDomain bool\n\tvar retentionDays int32\n\tvar isolationGroupData []byte\n\tvar isolationGroupEncoding string\n\tvar asyncWFConfigData []byte\n\tvar asyncWFConfigEncoding string\n\tvar activeClustersData []byte\n\tvar activeClustersEncoding string\n\n\tquery = db.session.Query(templateGetDomainByNameQueryV2, constDomainPartition, domainName).WithContext(ctx)\n\terr = query.Scan(\n\t\t&info.ID,\n\t\t&info.Name,\n\t\t&info.Status,\n\t\t&info.Description,\n\t\t&info.OwnerEmail,\n\t\t&info.Data,\n\t\t&retentionDays,\n\t\t&config.EmitMetric,\n\t\t&config.ArchivalBucket,\n\t\t&config.ArchivalStatus,\n\t\t&config.HistoryArchivalStatus,\n\t\t&config.HistoryArchivalURI,\n\t\t&config.VisibilityArchivalStatus,\n\t\t&config.VisibilityArchivalURI,\n\t\t&badBinariesData,\n\t\t&badBinariesDataEncoding,\n\t\t&replicationConfig.ActiveClusterName,\n\t\t&replicationClusters,\n\t\t&activeClustersData,\n\t\t&activeClustersEncoding,\n\t\t&isolationGroupData,\n\t\t&isolationGroupEncoding,\n\t\t&asyncWFConfigData,\n\t\t&asyncWFConfigEncoding,\n\t\t&isGlobalDomain,\n\t\t&configVersion,\n\t\t&failoverVersion,\n\t\t&failoverNotificationVersion,\n\t\t&previousFailoverVersion,\n\t\t&failoverEndTime,\n\t\t&lastUpdatedTime,\n\t\t&notificationVersion,\n\t)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfig.IsolationGroups = persistence.NewDataBlob(isolationGroupData, constants.EncodingType(isolationGroupEncoding))\n\tconfig.AsyncWorkflowsConfig = persistence.NewDataBlob(asyncWFConfigData, constants.EncodingType(asyncWFConfigEncoding))\n\tconfig.BadBinaries = persistence.NewDataBlob(badBinariesData, constants.EncodingType(badBinariesDataEncoding))\n\tconfig.Retention = common.DaysToDuration(retentionDays)\n\treplicationConfig.Clusters = persistence.DeserializeClusterConfigs(replicationClusters)\n\treplicationConfig.ActiveClustersConfig = persistence.NewDataBlob(activeClustersData, constants.EncodingType(activeClustersEncoding))\n\tdr := &nosqlplugin.DomainRow{\n\t\tInfo:                        info,\n\t\tConfig:                      config,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               configVersion,\n\t\tFailoverVersion:             failoverVersion,\n\t\tFailoverNotificationVersion: failoverNotificationVersion,\n\t\tPreviousFailoverVersion:     previousFailoverVersion,\n\t\tNotificationVersion:         notificationVersion,\n\t\tLastUpdatedTime:             time.Unix(0, lastUpdatedTime),\n\t\tIsGlobalDomain:              isGlobalDomain,\n\t}\n\tif failoverEndTime > emptyFailoverEndTime {\n\t\tdr.FailoverEndTime = common.TimePtr(time.Unix(0, failoverEndTime))\n\t}\n\n\treturn dr, nil\n}\n\n// Get all domain data\nfunc (db *CDB) SelectAllDomains(\n\tctx context.Context,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*nosqlplugin.DomainRow, []byte, error) {\n\tquery := db.session.Query(templateListDomainQueryV2, constDomainPartition).WithContext(ctx)\n\titer := query.PageSize(pageSize).PageState(pageToken).Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectAllDomains operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tvar name string\n\tdomain := &nosqlplugin.DomainRow{\n\t\tInfo:              &persistence.DomainInfo{},\n\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t}\n\tvar replicationClusters []map[string]interface{}\n\tvar badBinariesData []byte\n\tvar badBinariesDataEncoding string\n\tvar isolationGroups []byte\n\tvar isolationGroupsEncoding string\n\tvar asyncWFConfigData []byte\n\tvar asyncWFConfigEncoding string\n\tvar retentionDays int32\n\tvar failoverEndTime int64\n\tvar lastUpdateTime int64\n\tvar activeClustersData []byte\n\tvar activeClustersEncoding string\n\tvar rows []*nosqlplugin.DomainRow\n\tfor iter.Scan(\n\t\t&name,\n\t\t&domain.Info.ID,\n\t\t&domain.Info.Name,\n\t\t&domain.Info.Status,\n\t\t&domain.Info.Description,\n\t\t&domain.Info.OwnerEmail,\n\t\t&domain.Info.Data,\n\t\t&retentionDays,\n\t\t&domain.Config.EmitMetric,\n\t\t&domain.Config.ArchivalBucket,\n\t\t&domain.Config.ArchivalStatus,\n\t\t&domain.Config.HistoryArchivalStatus,\n\t\t&domain.Config.HistoryArchivalURI,\n\t\t&domain.Config.VisibilityArchivalStatus,\n\t\t&domain.Config.VisibilityArchivalURI,\n\t\t&badBinariesData,\n\t\t&badBinariesDataEncoding,\n\t\t&isolationGroups,\n\t\t&isolationGroupsEncoding,\n\t\t&asyncWFConfigData,\n\t\t&asyncWFConfigEncoding,\n\t\t&domain.ReplicationConfig.ActiveClusterName,\n\t\t&replicationClusters,\n\t\t&activeClustersData,\n\t\t&activeClustersEncoding,\n\t\t&domain.IsGlobalDomain,\n\t\t&domain.ConfigVersion,\n\t\t&domain.FailoverVersion,\n\t\t&domain.FailoverNotificationVersion,\n\t\t&domain.PreviousFailoverVersion,\n\t\t&failoverEndTime,\n\t\t&lastUpdateTime,\n\t\t&domain.NotificationVersion,\n\t) {\n\t\tif name != domainMetadataRecordName {\n\t\t\t// do not include the metadata record\n\t\t\tdomain.Config.BadBinaries = persistence.NewDataBlob(badBinariesData, constants.EncodingType(badBinariesDataEncoding))\n\t\t\tdomain.ReplicationConfig.Clusters = persistence.DeserializeClusterConfigs(replicationClusters)\n\t\t\tdomain.ReplicationConfig.ActiveClustersConfig = persistence.NewDataBlob(activeClustersData, constants.EncodingType(activeClustersEncoding))\n\t\t\tdomain.Config.Retention = common.DaysToDuration(retentionDays)\n\t\t\tdomain.Config.IsolationGroups = persistence.NewDataBlob(isolationGroups, constants.EncodingType(isolationGroupsEncoding))\n\t\t\tdomain.Config.AsyncWorkflowsConfig = persistence.NewDataBlob(asyncWFConfigData, constants.EncodingType(asyncWFConfigEncoding))\n\t\t\tdomain.LastUpdatedTime = time.Unix(0, lastUpdateTime)\n\t\t\tif failoverEndTime > emptyFailoverEndTime {\n\t\t\t\tdomain.FailoverEndTime = common.TimePtr(time.Unix(0, failoverEndTime))\n\t\t\t}\n\t\t\trows = append(rows, domain)\n\t\t}\n\t\treplicationClusters = []map[string]interface{}{}\n\t\tbadBinariesData = []byte(\"\")\n\t\tbadBinariesDataEncoding = \"\"\n\t\tisolationGroups = []byte(\"\")\n\t\tisolationGroupsEncoding = \"\"\n\t\tasyncWFConfigData = []byte(\"\")\n\t\tasyncWFConfigEncoding = \"\"\n\t\tfailoverEndTime = 0\n\t\tlastUpdateTime = 0\n\t\tretentionDays = 0\n\t\tactiveClustersData = []byte(\"\")\n\t\tactiveClustersEncoding = \"\"\n\t\tdomain = &nosqlplugin.DomainRow{\n\t\t\tInfo:              &persistence.DomainInfo{},\n\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t}\n\t}\n\n\tnextPageToken := iter.PageState()\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn rows, nextPageToken, nil\n}\n\n// Delete a domain, either by domainID or domainName\nfunc (db *CDB) DeleteDomain(ctx context.Context, domainID *string, domainName *string) error {\n\tif domainName == nil && domainID == nil {\n\t\treturn fmt.Errorf(\"must provide either domainID or domainName\")\n\t}\n\n\tif domainName == nil {\n\t\tquery := db.session.Query(templateGetDomainQuery, domainID).WithContext(ctx)\n\t\tvar name string\n\t\terr := query.Scan(&name)\n\t\tif err != nil {\n\t\t\tif db.client.IsNotFoundError(err) {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\tdomainName = common.StringPtr(name)\n\t} else {\n\t\tvar id string\n\t\tquery := db.session.Query(templateGetDomainByNameQueryV2, constDomainPartition, *domainName).WithContext(ctx)\n\t\terr := query.Scan(&id, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil)\n\t\tif err != nil {\n\t\t\tif db.client.IsNotFoundError(err) {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\tdomainID = common.StringPtr(id)\n\t}\n\n\treturn db.deleteDomain(ctx, *domainName, *domainID)\n}\n\nfunc (db *CDB) SelectDomainMetadata(ctx context.Context) (int64, error) {\n\tvar notificationVersion int64\n\tquery := db.session.Query(templateGetMetadataQueryV2, constDomainPartition, domainMetadataRecordName).WithContext(ctx)\n\terr := query.Scan(&notificationVersion)\n\tif err != nil {\n\t\tif db.client.IsNotFoundError(err) {\n\t\t\t// this error can be thrown in the very beginning,\n\t\t\t// i.e. when domains_by_name_v2 is initialized\n\t\t\t// TODO ??? really????\n\t\t\treturn 0, nil\n\t\t}\n\t\treturn -1, err\n\t}\n\treturn notificationVersion, nil\n}\n\nfunc (db *CDB) deleteDomain(ctx context.Context, name, ID string) error {\n\tquery := db.session.Query(templateDeleteDomainByNameQueryV2, constDomainPartition, name).WithContext(ctx)\n\tif err := db.executeWithConsistencyAll(query); err != nil {\n\t\treturn err\n\t}\n\n\tquery = db.session.Query(templateDeleteDomainQuery, ID).WithContext(ctx)\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc getIsolationGroupFields(row *nosqlplugin.DomainRow) ([]byte, string) {\n\tvar d []byte\n\tvar e string\n\tif row != nil && row.Config != nil && row.Config.IsolationGroups != nil {\n\t\td = row.Config.IsolationGroups.GetData()\n\t\te = row.Config.IsolationGroups.GetEncodingString()\n\t}\n\treturn d, e\n}\n\nfunc getAsyncWFConfigFields(row *nosqlplugin.DomainRow) ([]byte, string) {\n\tvar d []byte\n\tvar e string\n\tif row != nil && row.Config != nil && row.Config.AsyncWorkflowsConfig != nil {\n\t\td = row.Config.AsyncWorkflowsConfig.GetData()\n\t\te = row.Config.AsyncWorkflowsConfig.GetEncodingString()\n\t}\n\treturn d, e\n}\n\nfunc getActiveClustersFields(row *nosqlplugin.DomainRow) ([]byte, string) {\n\tvar d []byte\n\tvar e string\n\tif row != nil && row.ReplicationConfig != nil && row.ReplicationConfig.ActiveClustersConfig != nil {\n\t\td = row.ReplicationConfig.ActiveClustersConfig.GetData()\n\t\te = row.ReplicationConfig.ActiveClustersConfig.GetEncodingString()\n\t}\n\treturn d, e\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/domain_audit_log.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttemplateInsertDomainAuditLogQuery = `INSERT INTO domain_audit_log (` +\n\t\t`domain_id, event_id, state_before, state_before_encoding, state_after, state_after_encoding, ` +\n\t\t`operation_type, created_time, last_updated_time, identity, identity_type, comment) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) USING TTL ?`\n\n\ttemplateSelectDomainAuditLogsQuery = `SELECT ` +\n\t\t`event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, ` +\n\t\t`operation_type, created_time, last_updated_time, identity, identity_type, comment ` +\n\t\t`FROM domain_audit_log ` +\n\t\t`WHERE domain_id = ? AND operation_type = ? ` +\n\t\t`AND created_time >= ? AND created_time < ?`\n)\n\n// InsertDomainAuditLog inserts a new audit log entry for a domain operation\nfunc (db *CDB) InsertDomainAuditLog(ctx context.Context, row *nosqlplugin.DomainAuditLogRow) error {\n\tquery := db.session.Query(templateInsertDomainAuditLogQuery,\n\t\trow.DomainID,\n\t\trow.EventID,\n\t\trow.StateBefore,\n\t\trow.StateBeforeEncoding,\n\t\trow.StateAfter,\n\t\trow.StateAfterEncoding,\n\t\trow.OperationType,\n\t\trow.CreatedTime,\n\t\trow.LastUpdatedTime,\n\t\trow.Identity,\n\t\trow.IdentityType,\n\t\trow.Comment,\n\t\trow.TTLSeconds,\n\t).WithContext(ctx)\n\n\terr := query.Exec()\n\treturn err\n}\n\n// SelectDomainAuditLogs returns audit log entries for a domain and operation type\nfunc (db *CDB) SelectDomainAuditLogs(ctx context.Context, filter *nosqlplugin.DomainAuditLogFilter) ([]*nosqlplugin.DomainAuditLogRow, []byte, error) {\n\tif filter.MinCreatedTime == nil || filter.MaxCreatedTime == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectDomainAuditLogs requires non-nil MinCreatedTime and MaxCreatedTime\",\n\t\t}\n\t}\n\n\tquery := db.session.Query(templateSelectDomainAuditLogsQuery,\n\t\tfilter.DomainID,\n\t\tfilter.OperationType,\n\t\t*filter.MinCreatedTime,\n\t\t*filter.MaxCreatedTime,\n\t).WithContext(ctx)\n\n\t// Set page size\n\tif filter.PageSize > 0 {\n\t\tquery = query.PageSize(filter.PageSize)\n\t}\n\n\t// Set page state for pagination\n\tif len(filter.NextPageToken) > 0 {\n\t\tquery = query.PageState(filter.NextPageToken)\n\t}\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectDomainAuditLogs operation failed. Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tvar rows []*nosqlplugin.DomainAuditLogRow\n\trow := &nosqlplugin.DomainAuditLogRow{}\n\n\tfor iter.Scan(\n\t\t&row.EventID,\n\t\t&row.DomainID,\n\t\t&row.StateBefore,\n\t\t&row.StateBeforeEncoding,\n\t\t&row.StateAfter,\n\t\t&row.StateAfterEncoding,\n\t\t&row.OperationType,\n\t\t&row.CreatedTime,\n\t\t&row.LastUpdatedTime,\n\t\t&row.Identity,\n\t\t&row.IdentityType,\n\t\t&row.Comment,\n\t) {\n\t\trows = append(rows, row)\n\t\trow = &nosqlplugin.DomainAuditLogRow{}\n\n\t\t// Break after collecting PageSize number of rows\n\t\tif filter.PageSize > 0 && len(rows) >= filter.PageSize {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tnextPageToken := iter.PageState()\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\treturn rows, nextPageToken, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/domain_audit_log_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nfunc TestInsertDomainAuditLog(t *testing.T) {\n\tnow := time.Now().UTC()\n\tdomainID := \"test-domain-id\"\n\teventID := \"test-event-id\"\n\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.DomainAuditLogRow\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully inserted\",\n\t\t\trow: &nosqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:            domainID,\n\t\t\t\tEventID:             eventID,\n\t\t\t\tStateBefore:         []byte(\"state-before\"),\n\t\t\t\tStateBeforeEncoding: \"json\",\n\t\t\t\tStateAfter:          []byte(\"state-after\"),\n\t\t\t\tStateAfterEncoding:  \"json\",\n\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:         now,\n\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\tIdentity:            \"test-identity\",\n\t\t\t\tIdentityType:        \"user\",\n\t\t\t\tComment:             \"test comment\",\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"exec failed\",\n\t\t\trow: &nosqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:      domainID,\n\t\t\t\tEventID:       eventID,\n\t\t\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:   now,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"exec failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertDomainAuditLog(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectDomainAuditLogs(t *testing.T) {\n\tdomainID := \"test-domain-id\"\n\toperationType := persistence.DomainAuditOperationTypeFailover\n\tminTime := time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)\n\tmaxTime := time.Date(2024, 12, 31, 23, 59, 59, 0, time.UTC)\n\tcreatedTime1 := time.Date(2024, 6, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime2 := time.Date(2024, 7, 1, 12, 0, 0, 0, time.UTC)\n\n\ttests := []struct {\n\t\tname        string\n\t\tfilter      *nosqlplugin.DomainAuditLogFilter\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\titerMockFn  func(iter *gocql.MockIter)\n\t\twantRows    []*nosqlplugin.DomainAuditLogRow\n\t\twantToken   []byte\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"success with default time range and no results\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(nil)).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows:    []*nosqlplugin.DomainAuditLogRow{},\n\t\t\twantToken:   nil,\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"success with custom time range and single result\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\t// First call - return true and populate fields\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).DoAndReturn(func(args ...interface{}) bool {\n\t\t\t\t\t*args[0].(*string) = \"event-1\"\n\t\t\t\t\t*args[1].(*string) = domainID\n\t\t\t\t\t*args[2].(*[]byte) = []byte(\"state-before-1\")\n\t\t\t\t\t*args[3].(*string) = \"json\"\n\t\t\t\t\t*args[4].(*[]byte) = []byte(\"state-after-1\")\n\t\t\t\t\t*args[5].(*string) = \"json\"\n\t\t\t\t\t*args[6].(*persistence.DomainAuditOperationType) = operationType\n\t\t\t\t\t*args[7].(*time.Time) = createdTime1\n\t\t\t\t\t*args[8].(*time.Time) = createdTime1\n\t\t\t\t\t*args[9].(*string) = \"identity-1\"\n\t\t\t\t\t*args[10].(*string) = \"user\"\n\t\t\t\t\t*args[11].(*string) = \"comment-1\"\n\t\t\t\t\treturn true\n\t\t\t\t}).Times(1)\n\t\t\t\t// Second call - return false to end iteration\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(\"next-page\")).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows: []*nosqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:             \"event-1\",\n\t\t\t\t\tDomainID:            domainID,\n\t\t\t\t\tStateBefore:         []byte(\"state-before-1\"),\n\t\t\t\t\tStateBeforeEncoding: \"json\",\n\t\t\t\t\tStateAfter:          []byte(\"state-after-1\"),\n\t\t\t\t\tStateAfterEncoding:  \"json\",\n\t\t\t\t\tOperationType:       operationType,\n\t\t\t\t\tCreatedTime:         createdTime1,\n\t\t\t\t\tLastUpdatedTime:     createdTime1,\n\t\t\t\t\tIdentity:            \"identity-1\",\n\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\tComment:             \"comment-1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantToken:   []byte(\"next-page\"),\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"success with page size set\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\tPageSize:       10,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageSize(10).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(nil)).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows:    []*nosqlplugin.DomainAuditLogRow{},\n\t\t\twantToken:   nil,\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"success with pagination token\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\tPageSize:       10,\n\t\t\t\tNextPageToken:  []byte(\"prev-page-token\"),\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageSize(10).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageState([]byte(\"prev-page-token\")).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(nil)).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows:    []*nosqlplugin.DomainAuditLogRow{},\n\t\t\twantToken:   nil,\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"success with page size limiting - breaks after PageSize rows\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\tPageSize:       2,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageSize(2).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\t// First row\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).DoAndReturn(func(args ...interface{}) bool {\n\t\t\t\t\t*args[0].(*string) = \"event-1\"\n\t\t\t\t\t*args[1].(*string) = domainID\n\t\t\t\t\t*args[7].(*time.Time) = createdTime1\n\t\t\t\t\t*args[8].(*time.Time) = createdTime1\n\t\t\t\t\treturn true\n\t\t\t\t}).Times(1)\n\t\t\t\t// Second row\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).DoAndReturn(func(args ...interface{}) bool {\n\t\t\t\t\t*args[0].(*string) = \"event-2\"\n\t\t\t\t\t*args[1].(*string) = domainID\n\t\t\t\t\t*args[7].(*time.Time) = createdTime2\n\t\t\t\t\t*args[8].(*time.Time) = createdTime2\n\t\t\t\t\treturn true\n\t\t\t\t}).Times(1)\n\t\t\t\t// Should break after 2 rows, no third Scan call\n\t\t\t\titer.EXPECT().PageState().Return([]byte(\"next-page\")).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows: []*nosqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:         \"event-1\",\n\t\t\t\t\tDomainID:        domainID,\n\t\t\t\t\tCreatedTime:     createdTime1,\n\t\t\t\t\tLastUpdatedTime: createdTime1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:         \"event-2\",\n\t\t\t\t\tDomainID:        domainID,\n\t\t\t\t\tCreatedTime:     createdTime2,\n\t\t\t\t\tLastUpdatedTime: createdTime2,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantToken:   []byte(\"next-page\"),\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"success with multiple results\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\t// First row\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).DoAndReturn(func(args ...interface{}) bool {\n\t\t\t\t\t*args[0].(*string) = \"event-1\"\n\t\t\t\t\t*args[1].(*string) = domainID\n\t\t\t\t\t*args[7].(*time.Time) = createdTime1\n\t\t\t\t\t*args[8].(*time.Time) = createdTime1\n\t\t\t\t\treturn true\n\t\t\t\t}).Times(1)\n\t\t\t\t// Second row\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).DoAndReturn(func(args ...interface{}) bool {\n\t\t\t\t\t*args[0].(*string) = \"event-2\"\n\t\t\t\t\t*args[1].(*string) = domainID\n\t\t\t\t\t*args[7].(*time.Time) = createdTime2\n\t\t\t\t\t*args[8].(*time.Time) = createdTime2\n\t\t\t\t\treturn true\n\t\t\t\t}).Times(1)\n\t\t\t\t// End iteration\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(nil)).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRows: []*nosqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:         \"event-1\",\n\t\t\t\t\tDomainID:        domainID,\n\t\t\t\t\tCreatedTime:     createdTime1,\n\t\t\t\t\tLastUpdatedTime: createdTime1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:         \"event-2\",\n\t\t\t\t\tDomainID:        domainID,\n\t\t\t\t\tCreatedTime:     createdTime2,\n\t\t\t\t\tLastUpdatedTime: createdTime2,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantToken:   nil,\n\t\t\twantQueries: []string{`SELECT event_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding, operation_type, created_time, last_updated_time, identity, identity_type, comment FROM domain_audit_log WHERE domain_id = test-domain-id AND operation_type = Failover AND created_time >= 2024-01-01T00:00:00Z AND created_time < 2024-12-31T23:59:59Z`},\n\t\t},\n\t\t{\n\t\t\tname: \"error when iterator is nil\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\t// No calls expected\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"error when iterator close fails\",\n\t\t\tfilter: &nosqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:       domainID,\n\t\t\t\tOperationType:  operationType,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t},\n\t\t\titerMockFn: func(iter *gocql.MockIter) {\n\t\t\t\titer.EXPECT().Scan(\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\t\t\t).Return(false).Times(1)\n\t\t\t\titer.EXPECT().PageState().Return([]byte(nil)).Times(1)\n\t\t\t\titer.EXPECT().Close().Return(errors.New(\"close failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\titer := gocql.NewMockIter(ctrl)\n\n\t\t\ttc.queryMockFn(query)\n\n\t\t\t// Set up Iter() to return our mock iterator (unless the test expects nil)\n\t\t\tif tc.name != \"error when iterator is nil\" {\n\t\t\t\tquery.EXPECT().Iter().Return(iter).Times(1)\n\t\t\t}\n\n\t\t\ttc.iterMockFn(iter)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\trows, token, err := db.SelectDomainAuditLogs(context.Background(), tc.filter)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, len(tc.wantRows), len(rows), \"number of rows should match\")\n\n\t\t\tfor i, wantRow := range tc.wantRows {\n\t\t\t\tif i < len(rows) {\n\t\t\t\t\tassert.Equal(t, wantRow.EventID, rows[i].EventID)\n\t\t\t\t\tassert.Equal(t, wantRow.DomainID, rows[i].DomainID)\n\t\t\t\t\tassert.Equal(t, wantRow.StateBefore, rows[i].StateBefore)\n\t\t\t\t\tassert.Equal(t, wantRow.StateBeforeEncoding, rows[i].StateBeforeEncoding)\n\t\t\t\t\tassert.Equal(t, wantRow.StateAfter, rows[i].StateAfter)\n\t\t\t\t\tassert.Equal(t, wantRow.StateAfterEncoding, rows[i].StateAfterEncoding)\n\t\t\t\t\tassert.Equal(t, wantRow.OperationType, rows[i].OperationType)\n\t\t\t\t\tassert.Equal(t, wantRow.CreatedTime.Unix(), rows[i].CreatedTime.Unix())\n\t\t\t\t\tassert.Equal(t, wantRow.LastUpdatedTime.Unix(), rows[i].LastUpdatedTime.Unix())\n\t\t\t\t\tassert.Equal(t, wantRow.Identity, rows[i].Identity)\n\t\t\t\t\tassert.Equal(t, wantRow.IdentityType, rows[i].IdentityType)\n\t\t\t\t\tassert.Equal(t, wantRow.Comment, rows[i].Comment)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tassert.Equal(t, tc.wantToken, token)\n\t\t\tassert.Equal(t, tc.wantQueries, session.queries)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/domain_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\ttemplateDomainInfoType = `{` +\n\t\t`id: ?, ` +\n\t\t`name: ?, ` +\n\t\t`status: ?, ` +\n\t\t`description: ?, ` +\n\t\t`owner_email: ?, ` +\n\t\t`data: ? ` +\n\t\t`}`\n\n\ttemplateDomainConfigType = `{` +\n\t\t`retention: ?, ` +\n\t\t`emit_metric: ?, ` +\n\t\t`archival_bucket: ?, ` +\n\t\t`archival_status: ?,` +\n\t\t`history_archival_status: ?, ` +\n\t\t`history_archival_uri: ?, ` +\n\t\t`visibility_archival_status: ?, ` +\n\t\t`visibility_archival_uri: ?, ` +\n\t\t`bad_binaries: ?,` +\n\t\t`bad_binaries_encoding: ?,` +\n\t\t`isolation_groups: ?,` +\n\t\t`isolation_groups_encoding: ?,` +\n\t\t`async_workflow_config: ?,` +\n\t\t`async_workflow_config_encoding: ?` +\n\t\t`}`\n\n\ttemplateDomainReplicationConfigType = `{` +\n\t\t`active_cluster_name: ?, ` +\n\t\t`clusters: ?, ` +\n\t\t`active_clusters_config: ?, ` +\n\t\t`active_clusters_config_encoding: ?` +\n\t\t`}`\n\n\ttemplateCreateDomainQuery = `INSERT INTO domains (` +\n\t\t`id, domain, created_time) ` +\n\t\t`VALUES(?, {name: ?}, ?) IF NOT EXISTS`\n\n\ttemplateGetDomainQuery = `SELECT domain.name ` +\n\t\t`FROM domains ` +\n\t\t`WHERE id = ?`\n\n\ttemplateDeleteDomainQuery = `DELETE FROM domains ` +\n\t\t`WHERE id = ?`\n\n\ttemplateCreateDomainByNameQueryWithinBatchV2 = `INSERT INTO domains_by_name_v2 (` +\n\t\t`domains_partition, name, domain, config, replication_config, is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version, created_time) ` +\n\t\t`VALUES(?, ?, ` + templateDomainInfoType + `, ` + templateDomainConfigType + `, ` + templateDomainReplicationConfigType + `, ?, ?, ?, ?, ?, ?, ?, ?, ?) IF NOT EXISTS`\n\n\ttemplateGetDomainByNameQueryV2 = `SELECT domain.id, domain.name, domain.status, domain.description, ` +\n\t\t`domain.owner_email, domain.data, config.retention, config.emit_metric, ` +\n\t\t`config.archival_bucket, config.archival_status, ` +\n\t\t`config.history_archival_status, config.history_archival_uri, ` +\n\t\t`config.visibility_archival_status, config.visibility_archival_uri, ` +\n\t\t`config.bad_binaries, config.bad_binaries_encoding, ` +\n\t\t`replication_config.active_cluster_name, replication_config.clusters, ` +\n\t\t`replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, ` +\n\t\t`config.isolation_groups,` +\n\t\t`config.isolation_groups_encoding,` +\n\t\t`config.async_workflow_config,` +\n\t\t`config.async_workflow_config_encoding,` +\n\t\t`is_global_domain, ` +\n\t\t`config_version, ` +\n\t\t`failover_version, ` +\n\t\t`failover_notification_version, ` +\n\t\t`previous_failover_version, ` +\n\t\t`failover_end_time, ` +\n\t\t`last_updated_time, ` +\n\t\t`notification_version ` +\n\t\t`FROM domains_by_name_v2 ` +\n\t\t`WHERE domains_partition = ? ` +\n\t\t`and name = ?`\n\n\ttemplateUpdateDomainByNameQueryWithinBatchV2 = `UPDATE domains_by_name_v2 ` +\n\t\t`SET domain = ` + templateDomainInfoType + `, ` +\n\t\t`config = ` + templateDomainConfigType + `, ` +\n\t\t`replication_config = ` + templateDomainReplicationConfigType + `, ` +\n\t\t`config_version = ? ,` +\n\t\t`failover_version = ? ,` +\n\t\t`failover_notification_version = ? , ` +\n\t\t`previous_failover_version = ? , ` +\n\t\t`failover_end_time = ?,` +\n\t\t`last_updated_time = ?,` +\n\t\t`notification_version = ? ` +\n\t\t`WHERE domains_partition = ? ` +\n\t\t`and name = ?`\n\n\ttemplateGetMetadataQueryV2 = `SELECT notification_version ` +\n\t\t`FROM domains_by_name_v2 ` +\n\t\t`WHERE domains_partition = ? ` +\n\t\t`and name = ? `\n\n\ttemplateUpdateMetadataQueryWithinBatchV2 = `UPDATE domains_by_name_v2 ` +\n\t\t`SET notification_version = ? ` +\n\t\t`WHERE domains_partition = ? ` +\n\t\t`and name = ? ` +\n\t\t`IF notification_version = ? `\n\n\ttemplateDeleteDomainByNameQueryV2 = `DELETE FROM domains_by_name_v2 ` +\n\t\t`WHERE domains_partition = ? ` +\n\t\t`and name = ?`\n\n\ttemplateListDomainQueryV2 = `SELECT name, domain.id, domain.name, domain.status, domain.description, ` +\n\t\t`domain.owner_email, domain.data, config.retention, config.emit_metric, ` +\n\t\t`config.archival_bucket, config.archival_status, ` +\n\t\t`config.history_archival_status, config.history_archival_uri, ` +\n\t\t`config.visibility_archival_status, config.visibility_archival_uri, ` +\n\t\t`config.bad_binaries, config.bad_binaries_encoding, ` +\n\t\t`config.isolation_groups, config.isolation_groups_encoding, ` +\n\t\t`config.async_workflow_config, config.async_workflow_config_encoding, ` +\n\t\t`replication_config.active_cluster_name, replication_config.clusters, ` +\n\t\t`replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, ` +\n\t\t`is_global_domain, ` +\n\t\t`config_version, ` +\n\t\t`failover_version, ` +\n\t\t`failover_notification_version, ` +\n\t\t`previous_failover_version, ` +\n\t\t`failover_end_time, ` +\n\t\t`last_updated_time, ` +\n\t\t`notification_version ` +\n\t\t`FROM domains_by_name_v2 ` +\n\t\t`WHERE domains_partition = ? `\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/domain_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/testdata\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestInsertDomain(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-03T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname                      string\n\t\trow                       *nosqlplugin.DomainRow\n\t\tqueryMockFn               func(query *gocql.MockQuery)\n\t\tclientMockFn              func(client *gocql.MockClient)\n\t\tmapExecuteBatchCASApplied bool\n\t\tmapExecuteBatchCASPrev    map[string]any\n\t\tmapExecuteBatchCASErr     error\n\t\twantSessionQueries        []string\n\t\twantBatchQueries          []string\n\t\twantErr                   bool\n\t}{\n\t\t{\n\t\t\tname: \"insertion MapScanCAS failed\",\n\t\t\trow:  testdata.NewDomainRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"insertion MapScanCAS could not apply\",\n\t\t\trow:  testdata.NewDomainRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"insertion success - select metadata failed\",\n\t\t\trow:  testdata.NewDomainRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for SelectDomainMetadata\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                  \"insertion success - select metadata success - insertion to domains_by_name_v2 failed\",\n\t\t\trow:                   testdata.NewDomainRow(ts),\n\t\t\tmapExecuteBatchCASErr: errors.New(\"insert to domains_by_name_v2 failed\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for SelectDomainMetadata\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                      \"insertion success - select metadata success - insertion to domains_by_name_v2 not applied - orphan domain deletion failed\",\n\t\t\trow:                       testdata.NewDomainRow(ts),\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for SelectDomainMetadata\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for deleting orphan domain\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"orphan domain deletion failure\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                      \"insertion success - select metadata success - insertion to domains_by_name_v2 not applied - domain already exists\",\n\t\t\trow:                       testdata.NewDomainRow(ts),\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\"name\": testdata.NewDomainRow(ts).Info.Name, // this will causedomain already exist error\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for SelectDomainMetadata\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for deleting orphan domain\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                      \"all success\",\n\t\t\trow:                       testdata.NewDomainRow(ts),\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for insert\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for SelectDomainMetadata\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tnotificationVersion := args[0].(*int64)\n\t\t\t\t\t*notificationVersion = 7\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantSessionQueries: []string{\n\t\t\t\t`INSERT INTO domains (id, domain, created_time) VALUES(test-domain-id, {name: test-domain-name}, 2025-01-06T15:00:00Z) IF NOT EXISTS`,\n\t\t\t\t`SELECT notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = cadence-domain-metadata `,\n\t\t\t},\n\t\t\twantBatchQueries: []string{\n\t\t\t\t`INSERT INTO domains_by_name_v2 (` +\n\t\t\t\t\t`domains_partition, ` +\n\t\t\t\t\t`name, ` +\n\t\t\t\t\t`domain, ` +\n\t\t\t\t\t`config, ` +\n\t\t\t\t\t`replication_config, ` +\n\t\t\t\t\t`is_global_domain, ` +\n\t\t\t\t\t`config_version, ` +\n\t\t\t\t\t`failover_version, ` +\n\t\t\t\t\t`failover_notification_version, ` +\n\t\t\t\t\t`previous_failover_version, ` +\n\t\t\t\t\t`failover_end_time, ` +\n\t\t\t\t\t`last_updated_time, ` +\n\t\t\t\t\t`notification_version, created_time) ` +\n\t\t\t\t\t`VALUES(` +\n\t\t\t\t\t`0, ` +\n\t\t\t\t\t`test-domain-name, ` +\n\t\t\t\t\t`{id: test-domain-id, name: test-domain-name, status: 0, description: test-domain-description, owner_email: test-domain-owner-email, data: map[k1:v1] }, ` +\n\t\t\t\t\t`{retention: 7, emit_metric: true, archival_bucket: test-archival-bucket, archival_status: ENABLED,history_archival_status: ENABLED, history_archival_uri: test-history-archival-uri, visibility_archival_status: ENABLED, visibility_archival_uri: test-visibility-archival-uri, bad_binaries: [98 97 100 45 98 105 110 97 114 105 101 115],bad_binaries_encoding: thriftrw,isolation_groups: [105 115 111 108 97 116 105 111 110 45 103 114 111 117 112],isolation_groups_encoding: thriftrw,async_workflow_config: [97 115 121 110 99 45 119 111 114 107 102 108 111 119 115 45 99 111 110 102 105 103],async_workflow_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`{active_cluster_name: test-active-cluster-name, clusters: [map[cluster_name:test-cluster-name]], active_clusters_config: [97 99 116 105 118 101 45 99 108 117 115 116 101 114 115 45 99 111 110 102 105 103], active_clusters_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`true, ` +\n\t\t\t\t\t`3, ` +\n\t\t\t\t\t`4, ` +\n\t\t\t\t\t`0, ` +\n\t\t\t\t\t`-1, ` +\n\t\t\t\t\t`1712167200000000000, ` +\n\t\t\t\t\t`1712167200000000000, ` +\n\t\t\t\t\t`7, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`IF NOT EXISTS`,\n\t\t\t\t`UPDATE domains_by_name_v2 SET notification_version = 8 WHERE domains_partition = 0 and name = cadence-domain-metadata IF notification_version = 7 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery:                     query,\n\t\t\t\tmapExecuteBatchCASApplied: tc.mapExecuteBatchCASApplied,\n\t\t\t\tmapExecuteBatchCASPrev:    tc.mapExecuteBatchCASPrev,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif tc.clientMockFn != nil {\n\t\t\t\ttc.clientMockFn(client)\n\t\t\t}\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertDomain(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantSessionQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Session queries mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif len(session.batches) != 1 {\n\t\t\t\tt.Fatalf(\"Expected 1 batch, got %v\", len(session.batches))\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantBatchQueries, session.batches[0].queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Batch queries mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !session.iter.closed {\n\t\t\t\tt.Error(\"Expected iter to be closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDomain(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-04T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname                      string\n\t\trow                       *nosqlplugin.DomainRow\n\t\tmapExecuteBatchCASApplied bool\n\t\tmapExecuteBatchCASPrev    map[string]any\n\t\tmapExecuteBatchCASErr     error\n\t\twantBatchQueries          []string\n\t\twantErr                   bool\n\t}{\n\t\t{\n\t\t\tname: \"mapExecuteBatchCAS could not apply\",\n\t\t\trow: func() *nosqlplugin.DomainRow {\n\t\t\t\tr := testdata.NewDomainRow(ts)\n\t\t\t\tr.FailoverEndTime = nil\n\t\t\t\treturn r\n\t\t\t}(),\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\twantErr:                   true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapExecuteBatchCAS failed\",\n\t\t\trow: func() *nosqlplugin.DomainRow {\n\t\t\t\tr := testdata.NewDomainRow(ts)\n\t\t\t\tr.FailoverEndTime = nil\n\t\t\t\treturn r\n\t\t\t}(),\n\t\t\tmapExecuteBatchCASErr: errors.New(\"some random error\"),\n\t\t\twantErr:               true,\n\t\t},\n\t\t{\n\t\t\tname: \"empty failover end time\",\n\t\t\trow: func() *nosqlplugin.DomainRow {\n\t\t\t\tr := testdata.NewDomainRow(ts)\n\t\t\t\tr.FailoverEndTime = nil\n\t\t\t\treturn r\n\t\t\t}(),\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\twantBatchQueries: []string{\n\t\t\t\t`UPDATE domains_by_name_v2 SET ` +\n\t\t\t\t\t`domain = {id: test-domain-id, name: test-domain-name, status: 0, description: test-domain-description, owner_email: test-domain-owner-email, data: map[k1:v1] }, ` +\n\t\t\t\t\t`config = {retention: 7, emit_metric: true, archival_bucket: test-archival-bucket, archival_status: ENABLED,history_archival_status: ENABLED, history_archival_uri: test-history-archival-uri, visibility_archival_status: ENABLED, visibility_archival_uri: test-visibility-archival-uri, bad_binaries: [98 97 100 45 98 105 110 97 114 105 101 115],bad_binaries_encoding: thriftrw,isolation_groups: [105 115 111 108 97 116 105 111 110 45 103 114 111 117 112],isolation_groups_encoding: thriftrw,async_workflow_config: [97 115 121 110 99 45 119 111 114 107 102 108 111 119 115 45 99 111 110 102 105 103],async_workflow_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`replication_config = {active_cluster_name: test-active-cluster-name, clusters: [map[cluster_name:test-cluster-name]], active_clusters_config: [97 99 116 105 118 101 45 99 108 117 115 116 101 114 115 45 99 111 110 102 105 103], active_clusters_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`config_version = 3 ,` +\n\t\t\t\t\t`failover_version = 4 ,` +\n\t\t\t\t\t`failover_notification_version = 0 , ` +\n\t\t\t\t\t`previous_failover_version = 0 , ` +\n\t\t\t\t\t`failover_end_time = 0,` +\n\t\t\t\t\t`last_updated_time = 1712253600000000000,` +\n\t\t\t\t\t`notification_version = 5 ` +\n\t\t\t\t\t`WHERE domains_partition = 0 and name = test-domain-name`,\n\t\t\t\t`UPDATE domains_by_name_v2 SET notification_version = 6 WHERE ` +\n\t\t\t\t\t`domains_partition = 0 and ` +\n\t\t\t\t\t`name = cadence-domain-metadata ` +\n\t\t\t\t\t`IF notification_version = 5 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                      \"success\",\n\t\t\trow:                       testdata.NewDomainRow(ts),\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\twantBatchQueries: []string{\n\t\t\t\t`UPDATE domains_by_name_v2 SET ` +\n\t\t\t\t\t`domain = {id: test-domain-id, name: test-domain-name, status: 0, description: test-domain-description, owner_email: test-domain-owner-email, data: map[k1:v1] }, ` +\n\t\t\t\t\t`config = {retention: 7, emit_metric: true, archival_bucket: test-archival-bucket, archival_status: ENABLED,history_archival_status: ENABLED, history_archival_uri: test-history-archival-uri, visibility_archival_status: ENABLED, visibility_archival_uri: test-visibility-archival-uri, bad_binaries: [98 97 100 45 98 105 110 97 114 105 101 115],bad_binaries_encoding: thriftrw,isolation_groups: [105 115 111 108 97 116 105 111 110 45 103 114 111 117 112],isolation_groups_encoding: thriftrw,async_workflow_config: [97 115 121 110 99 45 119 111 114 107 102 108 111 119 115 45 99 111 110 102 105 103],async_workflow_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`replication_config = {active_cluster_name: test-active-cluster-name, clusters: [map[cluster_name:test-cluster-name]], active_clusters_config: [97 99 116 105 118 101 45 99 108 117 115 116 101 114 115 45 99 111 110 102 105 103], active_clusters_config_encoding: thriftrw}, ` +\n\t\t\t\t\t`config_version = 3 ,` +\n\t\t\t\t\t`failover_version = 4 ,` +\n\t\t\t\t\t`failover_notification_version = 0 , ` +\n\t\t\t\t\t`previous_failover_version = 0 , ` +\n\t\t\t\t\t`failover_end_time = 1712253600000000000,` +\n\t\t\t\t\t`last_updated_time = 1712253600000000000,` +\n\t\t\t\t\t`notification_version = 5 ` +\n\t\t\t\t\t`WHERE domains_partition = 0 and name = test-domain-name`,\n\t\t\t\t`UPDATE domains_by_name_v2 SET notification_version = 6 WHERE ` +\n\t\t\t\t\t`domains_partition = 0 and ` +\n\t\t\t\t\t`name = cadence-domain-metadata ` +\n\t\t\t\t\t`IF notification_version = 5 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery:                     query,\n\t\t\t\tmapExecuteBatchCASApplied: tc.mapExecuteBatchCASApplied,\n\t\t\t\tmapExecuteBatchCASPrev:    tc.mapExecuteBatchCASPrev,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateDomain(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif len(session.batches) != 1 {\n\t\t\t\tt.Fatalf(\"Expected 1 batch, got %v\", len(session.batches))\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantBatchQueries, session.batches[0].queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Batch queries mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !session.iter.closed {\n\t\t\t\tt.Error(\"Expected iter to be closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tdomainID    *string\n\t\tdomainName  *string\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:       \"both domainName and domainID not provided\",\n\t\t\tdomainName: nil,\n\t\t\tdomainID:   nil,\n\t\t\twantErr:    true,\n\t\t},\n\t\t{\n\t\t\tname:       \"both domainName and domainID provided\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\twantErr:    true,\n\t\t},\n\t\t{\n\t\t\tname:       \"domainName not provided - success\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: nil,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls to select domainName\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tname := args[0].(**string)\n\t\t\t\t\tdomainName := \"domain_name_1\"\n\t\t\t\t\t*name = &domainName\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls to select domain\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.name FROM domains WHERE id = domain_id_1`,\n\t\t\t\t`SELECT domain.id, domain.name, domain.status, domain.description, domain.owner_email, domain.data, config.retention, config.emit_metric, config.archival_bucket, config.archival_status, config.history_archival_status, config.history_archival_uri, config.visibility_archival_status, config.visibility_archival_uri, config.bad_binaries, config.bad_binaries_encoding, replication_config.active_cluster_name, replication_config.clusters, replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, config.isolation_groups,config.isolation_groups_encoding,config.async_workflow_config,config.async_workflow_config_encoding,is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"domainName not provided - scan failure\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: nil,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"domainID not provided - scan failure\",\n\t\t\tdomainID:   nil,\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"domainID not provided - success\",\n\t\t\tdomainID:   nil,\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.id, domain.name, domain.status, domain.description, domain.owner_email, domain.data, config.retention, config.emit_metric, config.archival_bucket, config.archival_status, config.history_archival_status, config.history_archival_uri, config.visibility_archival_status, config.visibility_archival_uri, config.bad_binaries, config.bad_binaries_encoding, replication_config.active_cluster_name, replication_config.clusters, replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, config.isolation_groups,config.isolation_groups_encoding,config.async_workflow_config,config.async_workflow_config_encoding,is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tif tc.queryMockFn != nil {\n\t\t\t\ttc.queryMockFn(query)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotRow, err := db.SelectDomain(context.Background(), tc.domainID, tc.domainName)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotRow == nil {\n\t\t\t\tt.Error(\"Expected domain row to be returned\")\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectAllDomains(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-03T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\tpageSize    int\n\t\tpagetoken   []byte\n\t\titer        *fakeIter\n\t\twantQueries []string\n\t\twantRows    []*nosqlplugin.DomainRow\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:    \"nil iter\",\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:    \"iter close failed\",\n\t\t\titer:    &fakeIter{closeErr: errors.New(\"some random error\")},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\titer: &fakeIter{\n\t\t\t\tscanInputs: [][]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"domain_name_1\",\n\t\t\t\t\t\t\"domain_id_1\",\n\t\t\t\t\t\t\"domain_name_1\",\n\t\t\t\t\t\tpersistence.DomainStatusRegistered,\n\t\t\t\t\t\t\"domain_description_1\",\n\t\t\t\t\t\t\"domain_owner_email_1\",\n\t\t\t\t\t\tmap[string]string{\"k1\": \"v1\"},\n\t\t\t\t\t\tint32(7),\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t\t\"test-archival-bucket\",\n\t\t\t\t\t\ttypes.ArchivalStatusEnabled,\n\t\t\t\t\t\ttypes.ArchivalStatusEnabled,\n\t\t\t\t\t\t\"test-history-archival-uri\",\n\t\t\t\t\t\ttypes.ArchivalStatusEnabled,\n\t\t\t\t\t\t\"test-visibility-archival-uri\",\n\t\t\t\t\t\t[]byte(\"bad-binaries\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t\t[]byte(\"isolation-groups\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t\t[]byte(\"async-workflow-config\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t\t\"test-active-cluster-name\",\n\t\t\t\t\t\t[]map[string]interface{}{},\n\t\t\t\t\t\t[]byte(\"active-clusters-config\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t\tint64(3),\n\t\t\t\t\t\tint64(4),\n\t\t\t\t\t\tint64(0),\n\t\t\t\t\t\tint64(-1),\n\t\t\t\t\t\tint64(1712167200000000000),\n\t\t\t\t\t\tint64(1712167200000000000),\n\t\t\t\t\t\tint64(7),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantRows: []*nosqlplugin.DomainRow{\n\t\t\t\t{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tID:          \"domain_id_1\",\n\t\t\t\t\t\tName:        \"domain_name_1\",\n\t\t\t\t\t\tDescription: \"domain_description_1\",\n\t\t\t\t\t\tOwnerEmail:  \"domain_owner_email_1\",\n\t\t\t\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\t\t\t},\n\t\t\t\t\tConfig: &persistence.InternalDomainConfig{\n\t\t\t\t\t\tRetention:                7 * 24 * time.Hour,\n\t\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\t\tArchivalBucket:           \"test-archival-bucket\",\n\t\t\t\t\t\tArchivalStatus:           types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\t\tHistoryArchivalURI:       \"test-history-archival-uri\",\n\t\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\t\tVisibilityArchivalURI:    \"test-visibility-archival-uri\",\n\t\t\t\t\t\tBadBinaries:              &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"bad-binaries\")},\n\t\t\t\t\t\tIsolationGroups:          &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"isolation-groups\")},\n\t\t\t\t\t\tAsyncWorkflowsConfig:     &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"async-workflow-config\")},\n\t\t\t\t\t},\n\t\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName:    \"test-active-cluster-name\",\n\t\t\t\t\t\tClusters:             []*persistence.ClusterReplicationConfig{},\n\t\t\t\t\t\tActiveClustersConfig: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"active-clusters-config\")},\n\t\t\t\t\t},\n\t\t\t\t\tConfigVersion:           3,\n\t\t\t\t\tFailoverVersion:         4,\n\t\t\t\t\tPreviousFailoverVersion: -1,\n\t\t\t\t\tFailoverEndTime:         &ts,\n\t\t\t\t\tNotificationVersion:     7,\n\t\t\t\t\tLastUpdatedTime:         ts,\n\t\t\t\t\tIsGlobalDomain:          true,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT name, domain.id, domain.name, domain.status, domain.description, domain.owner_email, domain.data, config.retention, config.emit_metric, config.archival_bucket, config.archival_status, config.history_archival_status, config.history_archival_uri, config.visibility_archival_status, config.visibility_archival_uri, config.bad_binaries, config.bad_binaries_encoding, config.isolation_groups, config.isolation_groups_encoding, config.async_workflow_config, config.async_workflow_config_encoding, replication_config.active_cluster_name, replication_config.clusters, replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\tgotRows, _, err := db.SelectAllDomains(context.Background(), tc.pageSize, tc.pagetoken)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantRows, gotRows); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Task rows mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Fatal(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectDomainMetadata(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tqueryMockFn  func(query *gocql.MockQuery)\n\t\tclientMockFn func(client *gocql.MockClient)\n\t\twantNtfVer   int64\n\t\twantQueries  []string\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tnotificationVersion := args[0].(*int64)\n\t\t\t\t\t*notificationVersion = 3\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantNtfVer: 3,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = cadence-domain-metadata `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scan failure - isnotfound\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some error that will be considered as not found by client mock\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantNtfVer: 0,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = cadence-domain-metadata `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scan failure\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif tc.clientMockFn != nil {\n\t\t\t\ttc.clientMockFn(client)\n\t\t\t}\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotNtfVer, err := db.SelectDomainMetadata(context.Background())\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotNtfVer != tc.wantNtfVer {\n\t\t\t\tt.Errorf(\"Got notification version = %v, want %v\", gotNtfVer, tc.wantNtfVer)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tdomainID     *string\n\t\tdomainName   *string\n\t\tqueryMockFn  func(query *gocql.MockQuery)\n\t\tclientMockFn func(client *gocql.MockClient)\n\t\twantQueries  []string\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname:       \"both domainName and domainID not provided\",\n\t\t\tdomainName: nil,\n\t\t\tdomainID:   nil,\n\t\t\twantErr:    true,\n\t\t},\n\t\t{\n\t\t\tname:       \"domainName not provided\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: nil,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for initial select\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tname := args[0].(*string)\n\t\t\t\t\t*name = \"domain_name_1\"\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// mock calls for delete from domains_by_name_v2\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\n\t\t\t\t// mock calls for delete from domains\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.name FROM domains WHERE id = domain_id_1`,\n\t\t\t\t`DELETE FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t\t`DELETE FROM domains WHERE id = domain_id_1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"domainName not provided - scan failure - isnotfound\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: nil,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some error that will be considered as not found by client mock\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.name FROM domains WHERE id = domain_id_1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"domainName not provided - scan failure\",\n\t\t\tdomainID:   common.StringPtr(\"domain_id_1\"),\n\t\t\tdomainName: nil,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"domainID not provided\",\n\t\t\tdomainID:   nil,\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\t// mock calls for initial select\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\t// Ideally we should be using DoAndReturn here to set the domainID,\n\t\t\t\t// but it panics because gomock doesn't handle n-arity func calls with nil params such as query.Scan(&id, nil, nil)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).Return(nil).Times(1)\n\n\t\t\t\t// mock calls for delete from domains_by_name_v2\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\n\t\t\t\t// mock calls for delete from domains\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.id, domain.name, domain.status, domain.description, domain.owner_email, domain.data, config.retention, config.emit_metric, config.archival_bucket, config.archival_status, config.history_archival_status, config.history_archival_uri, config.visibility_archival_status, config.visibility_archival_uri, config.bad_binaries, config.bad_binaries_encoding, replication_config.active_cluster_name, replication_config.clusters, replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, config.isolation_groups,config.isolation_groups_encoding,config.async_workflow_config,config.async_workflow_config_encoding,is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t\t`DELETE FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t\t`DELETE FROM domains WHERE id = `, // domainID is nil, so we expect an empty string here. See the comment above inside mockQueryFn.\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"domainID not provided - scan failure - isnotfound\",\n\t\t\tdomainID:   nil,\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).Return(errors.New(\"some error that will be considered as not found by client mock\")).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT domain.id, domain.name, domain.status, domain.description, domain.owner_email, domain.data, config.retention, config.emit_metric, config.archival_bucket, config.archival_status, config.history_archival_status, config.history_archival_uri, config.visibility_archival_status, config.visibility_archival_uri, config.bad_binaries, config.bad_binaries_encoding, replication_config.active_cluster_name, replication_config.clusters, replication_config.active_clusters_config, replication_config.active_clusters_config_encoding, config.isolation_groups,config.isolation_groups_encoding,config.async_workflow_config,config.async_workflow_config_encoding,is_global_domain, config_version, failover_version, failover_notification_version, previous_failover_version, failover_end_time, last_updated_time, notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = domain_name_1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"domainID not provided - scan failure\",\n\t\t\tdomainID:   nil,\n\t\t\tdomainName: common.StringPtr(\"domain_name_1\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).Return(errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tif tc.queryMockFn != nil {\n\t\t\t\ttc.queryMockFn(query)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif tc.clientMockFn != nil {\n\t\t\t\ttc.clientMockFn(client)\n\t\t\t}\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{\n\t\t\t\tEnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn false\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.DeleteDomain(context.Background(), tc.domainID, tc.domainName)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/batch.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/gocql/gocql\"\n)\n\nvar _ Batch = (*batch)(nil)\n\ntype (\n\tbatch struct {\n\t\t*gocql.Batch\n\t}\n)\n\n// Definition of all BatchTypes\nconst (\n\tLoggedBatch BatchType = iota\n\tUnloggedBatch\n\tCounterBatch\n)\n\nfunc newBatch(\n\tgocqlBatch *gocql.Batch,\n) *batch {\n\treturn &batch{\n\t\tBatch: gocqlBatch,\n\t}\n}\n\nfunc (b *batch) WithContext(ctx context.Context) Batch {\n\tb2 := b.Batch.WithContext(ctx)\n\tif b2 == nil {\n\t\treturn nil\n\t}\n\treturn newBatch(b2)\n}\n\nfunc (b *batch) WithTimestamp(timestamp int64) Batch {\n\tb.Batch.WithTimestamp(timestamp)\n\treturn b\n}\n\nfunc (b *batch) Consistency(c Consistency) Batch {\n\tb.Batch.SetConsistency(mustConvertConsistency(c))\n\treturn b\n}\n\nfunc mustConvertBatchType(batchType BatchType) gocql.BatchType {\n\tswitch batchType {\n\tcase LoggedBatch:\n\t\treturn gocql.LoggedBatch\n\tcase UnloggedBatch:\n\t\treturn gocql.UnloggedBatch\n\tcase CounterBatch:\n\t\treturn gocql.CounterBatch\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"Unknown gocql BatchType: %v\", batchType))\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/batch_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestBatch(t *testing.T) {\n\ttestBatch := gocql.Batch{}\n\tb := newBatch(&testBatch)\n\tassert.NotNil(t, b)\n}\n\nfunc Test_Batch_WithContext(t *testing.T) {\n\ttestCtx := context.Background()\n\ttestBatch := gocql.Batch{}\n\tb := newBatch(&testBatch)\n\tassert.NotNil(t, b.WithContext(testCtx))\n}\n\nfunc Test_Batch_WithTimestamp(t *testing.T) {\n\ttimeNow := time.Now().Unix()\n\ttestBatch := gocql.Batch{}\n\tb := newBatch(&testBatch)\n\twithTimeStamp := b.WithTimestamp(timeNow)\n\tassert.NotNil(t, withTimeStamp)\n}\n\nfunc Test_Batch_Consistency(t *testing.T) {\n\ttestBatch := gocql.Batch{}\n\tb := newBatch(&testBatch)\n\tconsistency := b.Consistency(One)\n\tassert.NotNil(t, consistency)\n}\n\nfunc Test_MustConvertBatchType(t *testing.T) {\n\tbatch := []BatchType{LoggedBatch, UnloggedBatch, CounterBatch}\n\tfor _, b := range batch {\n\t\tmustConvertBatchType(b)\n\t}\n}\n\nfunc Test_MustConvertBatchType_Panic(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r == nil {\n\t\t\tt.Errorf(\"The code did not panic\")\n\t\t}\n\t}()\n\tmustConvertBatchType(4)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/client.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"crypto/tls\"\n\t\"strings\"\n\n\t\"github.com/gocql/gocql\"\n\n\t\"github.com/uber/cadence/environment\"\n)\n\nvar (\n\tregistered Client\n)\n\n// GetRegisteredClient gets a gocql client based registered object\nfunc GetRegisteredClient() Client {\n\tif registered == nil {\n\t\tpanic(\"binary build error: gocql client is not registered yet!\")\n\t}\n\treturn registered\n}\n\n// RegisterClient registers a client into this package, can only be called once\nfunc RegisterClient(c Client) {\n\tif registered == nil {\n\t\tregistered = c\n\t} else {\n\t\tpanic(\"binary build error: gocql client is already register!\")\n\t}\n}\n\nfunc newCassandraCluster(cfg ClusterConfig) *gocql.ClusterConfig {\n\thosts := parseHosts(cfg.Hosts)\n\tcluster := gocql.NewCluster(hosts...)\n\tif cfg.ProtoVersion == 0 {\n\t\tcfg.ProtoVersion = environment.CassandraDefaultProtoVersionInteger\n\t}\n\tcluster.ProtoVersion = cfg.ProtoVersion\n\tif cfg.Port > 0 {\n\t\tcluster.Port = cfg.Port\n\t}\n\tif cfg.User != \"\" && cfg.Password != \"\" {\n\t\tcluster.Authenticator = gocql.PasswordAuthenticator{\n\t\t\tUsername:              cfg.User,\n\t\t\tPassword:              cfg.Password,\n\t\t\tAllowedAuthenticators: cfg.AllowedAuthenticators,\n\t\t}\n\t}\n\tif cfg.Keyspace != \"\" {\n\t\tcluster.Keyspace = cfg.Keyspace\n\t}\n\tif cfg.Datacenter != \"\" {\n\t\tcluster.HostFilter = gocql.DataCentreHostFilter(cfg.Datacenter)\n\t}\n\tif cfg.Region != \"\" {\n\t\tcluster.HostFilter = regionHostFilter(cfg.Region)\n\t}\n\n\tif cfg.TLS != nil && cfg.TLS.Enabled {\n\t\tcluster.SslOpts = &gocql.SslOptions{\n\t\t\tCertPath:               cfg.TLS.CertFile,\n\t\t\tKeyPath:                cfg.TLS.KeyFile,\n\t\t\tCaPath:                 cfg.TLS.CaFile,\n\t\t\tEnableHostVerification: cfg.TLS.EnableHostVerification,\n\n\t\t\tConfig: &tls.Config{\n\t\t\t\tServerName: cfg.TLS.ServerName,\n\t\t\t},\n\t\t}\n\t}\n\tif cfg.MaxConns > 0 {\n\t\tcluster.NumConns = cfg.MaxConns\n\t}\n\n\tif cfg.HostSelectionPolicy != nil {\n\t\tcluster.PoolConfig.HostSelectionPolicy = cfg.HostSelectionPolicy\n\t} else {\n\t\t// set default option if configuration was not provided\n\t\tcluster.PoolConfig.HostSelectionPolicy = gocql.TokenAwareHostPolicy(gocql.RoundRobinHostPolicy())\n\t}\n\n\treturn cluster\n}\n\n// regionHostFilter returns a gocql host filter for the given region name\nfunc regionHostFilter(region string) gocql.HostFilter {\n\treturn gocql.HostFilterFunc(func(host *gocql.HostInfo) bool {\n\t\tapplicationRegion := region\n\t\tif len(host.DataCenter()) < 3 {\n\t\t\treturn false\n\t\t}\n\t\treturn host.DataCenter()[:3] == applicationRegion\n\t})\n}\n\n// parseHosts returns parses a list of hosts separated by comma\nfunc parseHosts(input string) []string {\n\tvar hosts = make([]string, 0)\n\tfor _, h := range strings.Split(input, \",\") {\n\t\tif host := strings.TrimSpace(h); len(host) > 0 {\n\t\t\thosts = append(hosts, host)\n\t\t}\n\t}\n\treturn hosts\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n\tgomock \"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc Test_GetRegisteredClient(t *testing.T) {\n\tassert.Panics(t, func() { GetRegisteredClient() })\n}\n\nfunc Test_GetRegisteredClientNotNil(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tregistered = NewMockClient(mockCtrl)\n\tassert.Equal(t, registered, GetRegisteredClient())\n}\n\nfunc Test_RegisterClient(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r == nil {\n\t\t\tt.Errorf(\"The code did not panic\")\n\t\t}\n\t}()\n\tRegisterClient(nil)\n}\n\nfunc Test_RegisterClientNotNil(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tnewClient := NewMockClient(mockCtrl)\n\tregistered = nil\n\tRegisterClient(newClient)\n\tassert.Equal(t, newClient, registered)\n}\n\nfunc Test_newCassandraCluster(t *testing.T) {\n\ttestFullConfig := ClusterConfig{\n\t\tHosts:      \"testHost1,testHost2,testHost3,testHost4\",\n\t\tPort:       123,\n\t\tUser:       \"testUser\",\n\t\tPassword:   \"testPassword\",\n\t\tKeyspace:   \"testKeyspace\",\n\t\tDatacenter: \"testDatacenter\",\n\t\tRegion:     \"testRegion\",\n\t\tTLS: &config.TLS{\n\t\t\tEnabled:  true,\n\t\t\tCertFile: \"testCertFile\",\n\t\t\tKeyFile:  \"testKeyFile\",\n\t\t},\n\t\tMaxConns: 10,\n\t}\n\tclusterConfig := newCassandraCluster(testFullConfig)\n\tassert.Equal(t, []string{\"testHost1\", \"testHost2\", \"testHost3\", \"testHost4\"}, clusterConfig.Hosts)\n\tassert.Equal(t, testFullConfig.Port, clusterConfig.Port)\n\tassert.Equal(t, testFullConfig.User, clusterConfig.Authenticator.(gocql.PasswordAuthenticator).Username)\n\tassert.Equal(t, testFullConfig.Password, clusterConfig.Authenticator.(gocql.PasswordAuthenticator).Password)\n\tassert.Equal(t, testFullConfig.Keyspace, clusterConfig.Keyspace)\n\tassert.Equal(t, testFullConfig.TLS.CertFile, clusterConfig.SslOpts.CertPath)\n\tassert.Equal(t, testFullConfig.TLS.KeyFile, clusterConfig.SslOpts.KeyPath)\n\tassert.Equal(t, testFullConfig.MaxConns, clusterConfig.NumConns)\n\n\tassert.False(t, clusterConfig.HostFilter.Accept(&gocql.HostInfo{}))\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/consistency.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/gocql/gocql\"\n)\n\n// Definition of all Consistency levels\nconst (\n\tAny Consistency = iota\n\tOne\n\tTwo\n\tThree\n\tQuorum\n\tAll\n\tLocalQuorum\n\tEachQuorum\n\tLocalOne\n)\n\n// Definition of all SerialConsistency levels\nconst (\n\tSerial SerialConsistency = iota\n\tLocalSerial\n)\n\nfunc mustConvertConsistency(c Consistency) gocql.Consistency {\n\tswitch c {\n\tcase Any:\n\t\treturn gocql.Any\n\tcase One:\n\t\treturn gocql.One\n\tcase Two:\n\t\treturn gocql.Two\n\tcase Three:\n\t\treturn gocql.Three\n\tcase Quorum:\n\t\treturn gocql.Quorum\n\tcase All:\n\t\treturn gocql.All\n\tcase LocalQuorum:\n\t\treturn gocql.LocalQuorum\n\tcase EachQuorum:\n\t\treturn gocql.EachQuorum\n\tcase LocalOne:\n\t\treturn gocql.LocalOne\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"Unknown gocql Consistency level: %v\", c))\n\t}\n}\n\nfunc mustConvertSerialConsistency(c SerialConsistency) gocql.SerialConsistency {\n\tswitch c {\n\tcase Serial:\n\t\treturn gocql.Serial\n\tcase LocalSerial:\n\t\treturn gocql.LocalSerial\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"Unknown gocql SerialConsistency level: %v\", c))\n\t}\n}\n\nfunc (c Consistency) MarshalText() (text []byte, err error) {\n\treturn []byte(c.String()), nil\n}\n\nfunc (c *Consistency) UnmarshalText(text []byte) error {\n\tswitch string(text) {\n\tcase \"ANY\":\n\t\t*c = Any\n\tcase \"ONE\":\n\t\t*c = One\n\tcase \"TWO\":\n\t\t*c = Two\n\tcase \"THREE\":\n\t\t*c = Three\n\tcase \"QUORUM\":\n\t\t*c = Quorum\n\tcase \"ALL\":\n\t\t*c = All\n\tcase \"LOCAL_QUORUM\":\n\t\t*c = LocalQuorum\n\tcase \"EACH_QUORUM\":\n\t\t*c = EachQuorum\n\tcase \"LOCAL_ONE\":\n\t\t*c = LocalOne\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid consistency %q\", string(text))\n\t}\n\n\treturn nil\n}\n\nfunc (c Consistency) String() string {\n\tswitch c {\n\tcase Any:\n\t\treturn \"ANY\"\n\tcase One:\n\t\treturn \"ONE\"\n\tcase Two:\n\t\treturn \"TWO\"\n\tcase Three:\n\t\treturn \"THREE\"\n\tcase Quorum:\n\t\treturn \"QUORUM\"\n\tcase All:\n\t\treturn \"ALL\"\n\tcase LocalQuorum:\n\t\treturn \"LOCAL_QUORUM\"\n\tcase EachQuorum:\n\t\treturn \"EACH_QUORUM\"\n\tcase LocalOne:\n\t\treturn \"LOCAL_ONE\"\n\tdefault:\n\t\treturn fmt.Sprintf(\"invalid consistency: %d\", uint16(c))\n\t}\n}\n\nfunc ParseConsistency(s string) (Consistency, error) {\n\tvar c Consistency\n\tif err := c.UnmarshalText([]byte(strings.ToUpper(s))); err != nil {\n\t\treturn c, fmt.Errorf(\"parse consistency: %w\", err)\n\t}\n\treturn c, nil\n}\n\nfunc ParseSerialConsistency(s string) (SerialConsistency, error) {\n\tvar sc SerialConsistency\n\tif err := sc.UnmarshalText([]byte(strings.ToUpper(s))); err != nil {\n\t\treturn sc, fmt.Errorf(\"parse serial consistency: %w\", err)\n\n\t}\n\treturn sc, nil\n}\n\nfunc (s SerialConsistency) String() string {\n\tswitch s {\n\tcase Serial:\n\t\treturn \"SERIAL\"\n\tcase LocalSerial:\n\t\treturn \"LOCAL_SERIAL\"\n\tdefault:\n\t\treturn fmt.Sprintf(\"invalid serial consistency %d\", uint16(s))\n\t}\n}\n\nfunc (s SerialConsistency) MarshalText() (text []byte, err error) {\n\treturn []byte(s.String()), nil\n}\n\nfunc (s *SerialConsistency) UnmarshalText(text []byte) error {\n\tswitch string(text) {\n\tcase \"SERIAL\":\n\t\t*s = Serial\n\tcase \"LOCAL_SERIAL\":\n\t\t*s = LocalSerial\n\tdefault:\n\t\treturn fmt.Errorf(\"invalid serial consistency %q\", string(text))\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/consistency_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestConsistency_MarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tc        Consistency\n\t\twantText []byte\n\t\ttestFn   func(t assert.TestingT, expected, actual interface{}, msgAndArgs ...interface{}) bool\n\t}{\n\t\t{c: Any, wantText: []byte(\"ANY\"), testFn: assert.Equal},\n\t\t{c: One, wantText: []byte(\"ONE\"), testFn: assert.Equal},\n\t\t{c: Two, wantText: []byte(\"TWO\"), testFn: assert.Equal},\n\t\t{c: Three, wantText: []byte(\"THREE\"), testFn: assert.Equal},\n\t\t{c: Quorum, wantText: []byte(\"QUORUM\"), testFn: assert.Equal},\n\t\t{c: All, wantText: []byte(\"ALL\"), testFn: assert.Equal},\n\t\t{c: LocalQuorum, wantText: []byte(\"LOCAL_QUORUM\"), testFn: assert.Equal},\n\t\t{c: EachQuorum, wantText: []byte(\"EACH_QUORUM\"), testFn: assert.Equal},\n\t\t{c: LocalOne, wantText: []byte(\"LOCAL_ONE\"), testFn: assert.Equal},\n\t\t{c: LocalOne, wantText: []byte(\"WRONG_VALUE\"), testFn: assert.NotEqualValues},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.c.String(), func(t *testing.T) {\n\t\t\tgotText, err := tt.c.MarshalText()\n\t\t\tassert.NoError(t, err)\n\t\t\ttt.testFn(t, tt.wantText, gotText)\n\t\t})\n\t}\n}\n\nfunc TestConsistency_String(t *testing.T) {\n\tc := Consistency(9)\n\tassert.Equal(t, c.String(), \"invalid consistency: 9\")\n}\n\nfunc TestConsistency_UnmarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tdestConsistency Consistency\n\t\tinputText       []byte\n\t\ttestFn          func(t assert.TestingT, expected, actual interface{}, msgAndArgs ...interface{}) bool\n\t\twantErr         bool\n\t}{\n\t\t{destConsistency: Any, inputText: []byte(\"ANY\"), testFn: assert.Equal},\n\t\t{destConsistency: One, inputText: []byte(\"ONE\"), testFn: assert.Equal},\n\t\t{destConsistency: Two, inputText: []byte(\"TWO\"), testFn: assert.Equal},\n\t\t{destConsistency: Three, inputText: []byte(\"THREE\"), testFn: assert.Equal},\n\t\t{destConsistency: Quorum, inputText: []byte(\"QUORUM\"), testFn: assert.Equal},\n\t\t{destConsistency: All, inputText: []byte(\"ALL\"), testFn: assert.Equal},\n\t\t{destConsistency: LocalQuorum, inputText: []byte(\"LOCAL_QUORUM\"), testFn: assert.Equal},\n\t\t{destConsistency: EachQuorum, inputText: []byte(\"EACH_QUORUM\"), testFn: assert.Equal},\n\t\t{destConsistency: LocalOne, inputText: []byte(\"LOCAL_ONE\"), testFn: assert.Equal},\n\t\t{destConsistency: LocalOne, inputText: []byte(\"WRONG_VALUE\"), testFn: assert.NotEqualValues, wantErr: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.destConsistency.String(), func(t *testing.T) {\n\t\t\tvar c Consistency\n\t\t\terr := c.UnmarshalText(tt.inputText)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\ttt.testFn(t, tt.destConsistency, c)\n\t\t})\n\t}\n}\n\nfunc TestParseConsistency(t *testing.T) {\n\ttests := []struct {\n\t\tdestConsistency Consistency\n\t\tinputText       string\n\t\ttestFn          func(t assert.TestingT, expected, actual interface{}, msgAndArgs ...interface{}) bool\n\t\twantErr         bool\n\t}{\n\t\t{destConsistency: Any, inputText: \"any\", testFn: assert.Equal},\n\t\t{destConsistency: One, inputText: \"ONE\", testFn: assert.Equal},\n\t\t{destConsistency: Two, inputText: \"TWO\", testFn: assert.Equal},\n\t\t{destConsistency: Three, inputText: \"THREE\", testFn: assert.Equal},\n\t\t{destConsistency: Quorum, inputText: \"QUORUM\", testFn: assert.Equal},\n\t\t{destConsistency: All, inputText: \"all\", testFn: assert.Equal},\n\t\t{destConsistency: LocalQuorum, inputText: \"LOCAL_QUORUM\", testFn: assert.Equal},\n\t\t{destConsistency: EachQuorum, inputText: \"EACH_QUORUM\", testFn: assert.Equal},\n\t\t{destConsistency: LocalOne, inputText: \"LOCAL_ONE\", testFn: assert.Equal},\n\t\t{destConsistency: Any, inputText: \"WRONG_VALUE_FAILBACK_TO_ANY\", testFn: assert.Equal, wantErr: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(string(tt.inputText), func(t *testing.T) {\n\t\t\tgot, err := ParseConsistency(tt.inputText)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\ttt.testFn(t, tt.destConsistency, got)\n\t\t})\n\t}\n}\n\nfunc TestParseSerialConsistency(t *testing.T) {\n\ttests := []struct {\n\t\tdestConsistency SerialConsistency\n\t\tinputText       string\n\t\ttestFn          func(t assert.TestingT, expected, actual interface{}, msgAndArgs ...interface{}) bool\n\t\twantErr         bool\n\t}{\n\t\t{destConsistency: Serial, inputText: \"serial\", testFn: assert.Equal},\n\t\t{destConsistency: LocalSerial, inputText: \"LOCAL_SERIAL\", testFn: assert.Equal},\n\t\t{destConsistency: Serial, inputText: \"WRONG_VALUE_FAILBACK_TO_ANY\", testFn: assert.Equal, wantErr: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(string(tt.inputText), func(t *testing.T) {\n\t\t\tgot, err := ParseSerialConsistency(tt.inputText)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\ttt.testFn(t, tt.destConsistency, got)\n\t\t})\n\t}\n}\n\nfunc TestSerialConsistency_MarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tc        SerialConsistency\n\t\twantText []byte\n\t\ttestFn   func(t assert.TestingT, expected, actual interface{}, msgAndArgs ...interface{}) bool\n\t}{\n\t\t{c: Serial, wantText: []byte(\"SERIAL\"), testFn: assert.Equal},\n\t\t{c: LocalSerial, wantText: []byte(\"LOCAL_SERIAL\"), testFn: assert.Equal},\n\t\t{c: LocalSerial, wantText: []byte(\"WRONG_VALUE\"), testFn: assert.NotEqualValues},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.c.String(), func(t *testing.T) {\n\t\t\tgotText, err := tt.c.MarshalText()\n\t\t\tassert.NoError(t, err)\n\t\t\ttt.testFn(t, tt.wantText, gotText)\n\t\t})\n\t}\n}\n\nfunc TestSerialConsistency_String(t *testing.T) {\n\tc := SerialConsistency(2)\n\tassert.Equal(t, c.String(), \"invalid serial consistency 2\")\n}\n\nfunc TestSerialConsistency_UnmarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tdestSerialConsistency SerialConsistency\n\t\tinputText             []byte\n\t\twantErr               bool\n\t}{\n\t\t{destSerialConsistency: Serial, inputText: []byte(\"SERIAL\")},\n\t\t{destSerialConsistency: LocalSerial, inputText: []byte(\"LOCAL_SERIAL\")},\n\t\t{destSerialConsistency: Serial, inputText: []byte(\"WRONG_VALUE_DEFAULTS_TO_SERIAL\"), wantErr: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.destSerialConsistency.String(), func(t *testing.T) {\n\t\t\tvar c SerialConsistency\n\t\t\terr := c.UnmarshalText(tt.inputText)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.destSerialConsistency, c)\n\t\t})\n\t}\n}\n\nfunc Test_mustConvertConsistency(t *testing.T) {\n\ttests := []struct {\n\t\tinput  Consistency\n\t\toutput gocql.Consistency\n\t}{\n\t\t{Any, gocql.Any},\n\t\t{One, gocql.One},\n\t\t{Two, gocql.Two},\n\t\t{Three, gocql.Three},\n\t\t{Quorum, gocql.Quorum},\n\t\t{All, gocql.All},\n\t\t{LocalQuorum, gocql.LocalQuorum},\n\t\t{EachQuorum, gocql.EachQuorum},\n\t\t{LocalOne, gocql.LocalOne},\n\t}\n\n\tfor _, tt := range tests {\n\t\tassert.Equal(t, tt.output, mustConvertConsistency(tt.input))\n\t}\n\tassert.Panics(t, func() { mustConvertConsistency(Consistency(9999)) })\n}\n\nfunc Test_mustConvertSerialConsistency(t *testing.T) {\n\ttests := []struct {\n\t\tinput  SerialConsistency\n\t\toutput gocql.SerialConsistency\n\t}{\n\t\t{Serial, gocql.Serial},\n\t\t{LocalSerial, gocql.LocalSerial},\n\t}\n\n\tfor _, tt := range tests {\n\t\tassert.Equal(t, tt.output, mustConvertSerialConsistency(tt.input))\n\t}\n\tassert.Panics(t, func() { mustConvertSerialConsistency(SerialConsistency(9999)) })\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/interface.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\n\npackage gocql\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/gocql/gocql\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Note: this file defines the minimal interface that is needed by Cadence's cassandra\n// persistence implementation and should be implemented for all gocql libraries if\n// they need to be used.\n// Please add more methods to the interface if needed by the cassandra implementation.\n\ntype (\n\t// Client is an interface for all gocql libraries.\n\tClient interface {\n\t\tCreateSession(ClusterConfig) (Session, error)\n\n\t\tnosqlplugin.ClientErrorChecker\n\t\tIsCassandraConsistencyError(error) bool\n\t}\n\n\t// Session is the interface for interacting with the database.\n\tSession interface {\n\t\tQuery(string, ...interface{}) Query\n\t\tNewBatch(BatchType) Batch\n\t\tExecuteBatch(Batch) error\n\t\tMapExecuteBatchCAS(Batch, map[string]interface{}) (bool, Iter, error)\n\t\tClose()\n\t}\n\n\t// Query is the interface for query object.\n\tQuery interface {\n\t\tExec() error\n\t\tScan(...interface{}) error\n\t\tScanCAS(...interface{}) (bool, error)\n\t\tMapScan(map[string]interface{}) error\n\t\tMapScanCAS(map[string]interface{}) (bool, error)\n\t\tIter() Iter\n\t\tPageSize(int) Query\n\t\tPageState([]byte) Query\n\t\tWithContext(context.Context) Query\n\t\tWithTimestamp(int64) Query\n\t\tConsistency(Consistency) Query\n\t\tBind(...interface{}) Query\n\t}\n\n\t// Batch is the interface for batch operation.\n\tBatch interface {\n\t\tQuery(string, ...interface{})\n\t\tWithContext(context.Context) Batch\n\t\tWithTimestamp(int64) Batch\n\t\tConsistency(Consistency) Batch\n\t}\n\n\t// Iter is the interface for executing and iterating over all resulting rows.\n\tIter interface {\n\t\tScan(...interface{}) bool\n\t\tMapScan(map[string]interface{}) bool\n\t\tPageState() []byte\n\t\tClose() error\n\t}\n\n\t// UUID represents a universally unique identifier\n\tUUID interface {\n\t\tString() string\n\t}\n\n\t// BatchType is the type of the Batch operation\n\tBatchType byte\n\n\t// Consistency is the consistency level used by a Query\n\tConsistency uint16\n\n\t// SerialConsistency is the serial consistency level used by a Query\n\tSerialConsistency uint16\n\n\t// ClusterConfig is the config for cassandra connection\n\tClusterConfig struct {\n\t\tHosts                 string\n\t\tPort                  int\n\t\tUser                  string\n\t\tPassword              string\n\t\tAllowedAuthenticators []string\n\t\tKeyspace              string\n\t\tRegion                string\n\t\tDatacenter            string\n\t\tMaxConns              int\n\t\tTLS                   *config.TLS\n\t\tProtoVersion          int\n\t\tConsistency           Consistency\n\t\tSerialConsistency     SerialConsistency\n\t\tTimeout               time.Duration\n\t\tConnectTimeout        time.Duration\n\t\tHostSelectionPolicy   gocql.HostSelectionPolicy\n\t}\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package gocql -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\n//\n\n// Package gocql is a generated GoMock package.\npackage gocql\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// CreateSession mocks base method.\nfunc (m *MockClient) CreateSession(arg0 ClusterConfig) (Session, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateSession\", arg0)\n\tret0, _ := ret[0].(Session)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateSession indicates an expected call of CreateSession.\nfunc (mr *MockClientMockRecorder) CreateSession(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateSession\", reflect.TypeOf((*MockClient)(nil).CreateSession), arg0)\n}\n\n// IsCassandraConsistencyError mocks base method.\nfunc (m *MockClient) IsCassandraConsistencyError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsCassandraConsistencyError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsCassandraConsistencyError indicates an expected call of IsCassandraConsistencyError.\nfunc (mr *MockClientMockRecorder) IsCassandraConsistencyError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsCassandraConsistencyError\", reflect.TypeOf((*MockClient)(nil).IsCassandraConsistencyError), arg0)\n}\n\n// IsDBUnavailableError mocks base method.\nfunc (m *MockClient) IsDBUnavailableError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDBUnavailableError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDBUnavailableError indicates an expected call of IsDBUnavailableError.\nfunc (mr *MockClientMockRecorder) IsDBUnavailableError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDBUnavailableError\", reflect.TypeOf((*MockClient)(nil).IsDBUnavailableError), arg0)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockClient) IsNotFoundError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockClientMockRecorder) IsNotFoundError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockClient)(nil).IsNotFoundError), arg0)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockClient) IsThrottlingError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockClientMockRecorder) IsThrottlingError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockClient)(nil).IsThrottlingError), arg0)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockClient) IsTimeoutError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockClientMockRecorder) IsTimeoutError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockClient)(nil).IsTimeoutError), arg0)\n}\n\n// MockSession is a mock of Session interface.\ntype MockSession struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSessionMockRecorder\n\tisgomock struct{}\n}\n\n// MockSessionMockRecorder is the mock recorder for MockSession.\ntype MockSessionMockRecorder struct {\n\tmock *MockSession\n}\n\n// NewMockSession creates a new mock instance.\nfunc NewMockSession(ctrl *gomock.Controller) *MockSession {\n\tmock := &MockSession{ctrl: ctrl}\n\tmock.recorder = &MockSessionMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSession) EXPECT() *MockSessionMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockSession) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockSessionMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockSession)(nil).Close))\n}\n\n// ExecuteBatch mocks base method.\nfunc (m *MockSession) ExecuteBatch(arg0 Batch) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ExecuteBatch\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ExecuteBatch indicates an expected call of ExecuteBatch.\nfunc (mr *MockSessionMockRecorder) ExecuteBatch(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecuteBatch\", reflect.TypeOf((*MockSession)(nil).ExecuteBatch), arg0)\n}\n\n// MapExecuteBatchCAS mocks base method.\nfunc (m *MockSession) MapExecuteBatchCAS(arg0 Batch, arg1 map[string]any) (bool, Iter, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MapExecuteBatchCAS\", arg0, arg1)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(Iter)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// MapExecuteBatchCAS indicates an expected call of MapExecuteBatchCAS.\nfunc (mr *MockSessionMockRecorder) MapExecuteBatchCAS(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MapExecuteBatchCAS\", reflect.TypeOf((*MockSession)(nil).MapExecuteBatchCAS), arg0, arg1)\n}\n\n// NewBatch mocks base method.\nfunc (m *MockSession) NewBatch(arg0 BatchType) Batch {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewBatch\", arg0)\n\tret0, _ := ret[0].(Batch)\n\treturn ret0\n}\n\n// NewBatch indicates an expected call of NewBatch.\nfunc (mr *MockSessionMockRecorder) NewBatch(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewBatch\", reflect.TypeOf((*MockSession)(nil).NewBatch), arg0)\n}\n\n// Query mocks base method.\nfunc (m *MockSession) Query(arg0 string, arg1 ...any) Query {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0}\n\tfor _, a := range arg1 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Query\", varargs...)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// Query indicates an expected call of Query.\nfunc (mr *MockSessionMockRecorder) Query(arg0 any, arg1 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0}, arg1...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Query\", reflect.TypeOf((*MockSession)(nil).Query), varargs...)\n}\n\n// MockQuery is a mock of Query interface.\ntype MockQuery struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueryMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueryMockRecorder is the mock recorder for MockQuery.\ntype MockQueryMockRecorder struct {\n\tmock *MockQuery\n}\n\n// NewMockQuery creates a new mock instance.\nfunc NewMockQuery(ctrl *gomock.Controller) *MockQuery {\n\tmock := &MockQuery{ctrl: ctrl}\n\tmock.recorder = &MockQueryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQuery) EXPECT() *MockQueryMockRecorder {\n\treturn m.recorder\n}\n\n// Bind mocks base method.\nfunc (m *MockQuery) Bind(arg0 ...any) Query {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Bind\", varargs...)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// Bind indicates an expected call of Bind.\nfunc (mr *MockQueryMockRecorder) Bind(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Bind\", reflect.TypeOf((*MockQuery)(nil).Bind), arg0...)\n}\n\n// Consistency mocks base method.\nfunc (m *MockQuery) Consistency(arg0 Consistency) Query {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Consistency\", arg0)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// Consistency indicates an expected call of Consistency.\nfunc (mr *MockQueryMockRecorder) Consistency(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Consistency\", reflect.TypeOf((*MockQuery)(nil).Consistency), arg0)\n}\n\n// Exec mocks base method.\nfunc (m *MockQuery) Exec() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Exec\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Exec indicates an expected call of Exec.\nfunc (mr *MockQueryMockRecorder) Exec() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Exec\", reflect.TypeOf((*MockQuery)(nil).Exec))\n}\n\n// Iter mocks base method.\nfunc (m *MockQuery) Iter() Iter {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Iter\")\n\tret0, _ := ret[0].(Iter)\n\treturn ret0\n}\n\n// Iter indicates an expected call of Iter.\nfunc (mr *MockQueryMockRecorder) Iter() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Iter\", reflect.TypeOf((*MockQuery)(nil).Iter))\n}\n\n// MapScan mocks base method.\nfunc (m *MockQuery) MapScan(arg0 map[string]any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MapScan\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// MapScan indicates an expected call of MapScan.\nfunc (mr *MockQueryMockRecorder) MapScan(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MapScan\", reflect.TypeOf((*MockQuery)(nil).MapScan), arg0)\n}\n\n// MapScanCAS mocks base method.\nfunc (m *MockQuery) MapScanCAS(arg0 map[string]any) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MapScanCAS\", arg0)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MapScanCAS indicates an expected call of MapScanCAS.\nfunc (mr *MockQueryMockRecorder) MapScanCAS(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MapScanCAS\", reflect.TypeOf((*MockQuery)(nil).MapScanCAS), arg0)\n}\n\n// PageSize mocks base method.\nfunc (m *MockQuery) PageSize(arg0 int) Query {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PageSize\", arg0)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// PageSize indicates an expected call of PageSize.\nfunc (mr *MockQueryMockRecorder) PageSize(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PageSize\", reflect.TypeOf((*MockQuery)(nil).PageSize), arg0)\n}\n\n// PageState mocks base method.\nfunc (m *MockQuery) PageState(arg0 []byte) Query {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PageState\", arg0)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// PageState indicates an expected call of PageState.\nfunc (mr *MockQueryMockRecorder) PageState(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PageState\", reflect.TypeOf((*MockQuery)(nil).PageState), arg0)\n}\n\n// Scan mocks base method.\nfunc (m *MockQuery) Scan(arg0 ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Scan\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Scan indicates an expected call of Scan.\nfunc (mr *MockQueryMockRecorder) Scan(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Scan\", reflect.TypeOf((*MockQuery)(nil).Scan), arg0...)\n}\n\n// ScanCAS mocks base method.\nfunc (m *MockQuery) ScanCAS(arg0 ...any) (bool, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ScanCAS\", varargs...)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ScanCAS indicates an expected call of ScanCAS.\nfunc (mr *MockQueryMockRecorder) ScanCAS(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScanCAS\", reflect.TypeOf((*MockQuery)(nil).ScanCAS), arg0...)\n}\n\n// WithContext mocks base method.\nfunc (m *MockQuery) WithContext(arg0 context.Context) Query {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WithContext\", arg0)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// WithContext indicates an expected call of WithContext.\nfunc (mr *MockQueryMockRecorder) WithContext(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WithContext\", reflect.TypeOf((*MockQuery)(nil).WithContext), arg0)\n}\n\n// WithTimestamp mocks base method.\nfunc (m *MockQuery) WithTimestamp(arg0 int64) Query {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WithTimestamp\", arg0)\n\tret0, _ := ret[0].(Query)\n\treturn ret0\n}\n\n// WithTimestamp indicates an expected call of WithTimestamp.\nfunc (mr *MockQueryMockRecorder) WithTimestamp(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WithTimestamp\", reflect.TypeOf((*MockQuery)(nil).WithTimestamp), arg0)\n}\n\n// MockBatch is a mock of Batch interface.\ntype MockBatch struct {\n\tctrl     *gomock.Controller\n\trecorder *MockBatchMockRecorder\n\tisgomock struct{}\n}\n\n// MockBatchMockRecorder is the mock recorder for MockBatch.\ntype MockBatchMockRecorder struct {\n\tmock *MockBatch\n}\n\n// NewMockBatch creates a new mock instance.\nfunc NewMockBatch(ctrl *gomock.Controller) *MockBatch {\n\tmock := &MockBatch{ctrl: ctrl}\n\tmock.recorder = &MockBatchMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockBatch) EXPECT() *MockBatchMockRecorder {\n\treturn m.recorder\n}\n\n// Consistency mocks base method.\nfunc (m *MockBatch) Consistency(arg0 Consistency) Batch {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Consistency\", arg0)\n\tret0, _ := ret[0].(Batch)\n\treturn ret0\n}\n\n// Consistency indicates an expected call of Consistency.\nfunc (mr *MockBatchMockRecorder) Consistency(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Consistency\", reflect.TypeOf((*MockBatch)(nil).Consistency), arg0)\n}\n\n// Query mocks base method.\nfunc (m *MockBatch) Query(arg0 string, arg1 ...any) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0}\n\tfor _, a := range arg1 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"Query\", varargs...)\n}\n\n// Query indicates an expected call of Query.\nfunc (mr *MockBatchMockRecorder) Query(arg0 any, arg1 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0}, arg1...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Query\", reflect.TypeOf((*MockBatch)(nil).Query), varargs...)\n}\n\n// WithContext mocks base method.\nfunc (m *MockBatch) WithContext(arg0 context.Context) Batch {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WithContext\", arg0)\n\tret0, _ := ret[0].(Batch)\n\treturn ret0\n}\n\n// WithContext indicates an expected call of WithContext.\nfunc (mr *MockBatchMockRecorder) WithContext(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WithContext\", reflect.TypeOf((*MockBatch)(nil).WithContext), arg0)\n}\n\n// WithTimestamp mocks base method.\nfunc (m *MockBatch) WithTimestamp(arg0 int64) Batch {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WithTimestamp\", arg0)\n\tret0, _ := ret[0].(Batch)\n\treturn ret0\n}\n\n// WithTimestamp indicates an expected call of WithTimestamp.\nfunc (mr *MockBatchMockRecorder) WithTimestamp(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WithTimestamp\", reflect.TypeOf((*MockBatch)(nil).WithTimestamp), arg0)\n}\n\n// MockIter is a mock of Iter interface.\ntype MockIter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockIterMockRecorder\n\tisgomock struct{}\n}\n\n// MockIterMockRecorder is the mock recorder for MockIter.\ntype MockIterMockRecorder struct {\n\tmock *MockIter\n}\n\n// NewMockIter creates a new mock instance.\nfunc NewMockIter(ctrl *gomock.Controller) *MockIter {\n\tmock := &MockIter{ctrl: ctrl}\n\tmock.recorder = &MockIterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockIter) EXPECT() *MockIterMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockIter) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockIterMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockIter)(nil).Close))\n}\n\n// MapScan mocks base method.\nfunc (m *MockIter) MapScan(arg0 map[string]any) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MapScan\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// MapScan indicates an expected call of MapScan.\nfunc (mr *MockIterMockRecorder) MapScan(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MapScan\", reflect.TypeOf((*MockIter)(nil).MapScan), arg0)\n}\n\n// PageState mocks base method.\nfunc (m *MockIter) PageState() []byte {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PageState\")\n\tret0, _ := ret[0].([]byte)\n\treturn ret0\n}\n\n// PageState indicates an expected call of PageState.\nfunc (mr *MockIterMockRecorder) PageState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PageState\", reflect.TypeOf((*MockIter)(nil).PageState))\n}\n\n// Scan mocks base method.\nfunc (m *MockIter) Scan(arg0 ...any) bool {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Scan\", varargs...)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Scan indicates an expected call of Scan.\nfunc (mr *MockIterMockRecorder) Scan(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Scan\", reflect.TypeOf((*MockIter)(nil).Scan), arg0...)\n}\n\n// MockUUID is a mock of UUID interface.\ntype MockUUID struct {\n\tctrl     *gomock.Controller\n\trecorder *MockUUIDMockRecorder\n\tisgomock struct{}\n}\n\n// MockUUIDMockRecorder is the mock recorder for MockUUID.\ntype MockUUIDMockRecorder struct {\n\tmock *MockUUID\n}\n\n// NewMockUUID creates a new mock instance.\nfunc NewMockUUID(ctrl *gomock.Controller) *MockUUID {\n\tmock := &MockUUID{ctrl: ctrl}\n\tmock.recorder = &MockUUIDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockUUID) EXPECT() *MockUUIDMockRecorder {\n\treturn m.recorder\n}\n\n// String mocks base method.\nfunc (m *MockUUID) String() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"String\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// String indicates an expected call of String.\nfunc (mr *MockUUIDMockRecorder) String() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"String\", reflect.TypeOf((*MockUUID)(nil).String))\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/public/client.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage public\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"strings\"\n\n\tgogocql \"github.com/gocql/gocql\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nvar _ gocql.Client = client{}\n\ntype (\n\tclient struct {\n\t}\n)\n\nfunc init() {\n\tgocql.RegisterClient(client{})\n}\n\nfunc (c client) CreateSession(\n\tconfig gocql.ClusterConfig,\n) (gocql.Session, error) {\n\treturn gocql.NewSession(config)\n}\n\nfunc (c client) IsTimeoutError(err error) bool {\n\tif err == context.DeadlineExceeded {\n\t\treturn true\n\t}\n\tif err == gogocql.ErrTimeoutNoResponse {\n\t\treturn true\n\t}\n\tif err == gogocql.ErrConnectionClosed {\n\t\treturn true\n\t}\n\t_, ok := err.(*gogocql.RequestErrWriteTimeout)\n\treturn ok\n}\n\nfunc (c client) IsNotFoundError(err error) bool {\n\treturn err == gogocql.ErrNotFound\n}\n\nfunc (c client) IsThrottlingError(err error) bool {\n\tif req, ok := err.(gogocql.RequestError); ok {\n\t\t// gocql does not expose the constant errOverloaded = 0x1001\n\t\treturn req.Code() == 0x1001\n\t}\n\treturn false\n}\n\n// IsDBUnavailableError checks if the error is a database unavailable error\n// relating to issues like the database being overloaded or the consistency level not being achievable\n// because a node is down.\nfunc (c client) IsDBUnavailableError(err error) bool {\n\tvar e gogocql.RequestError\n\tif errors.As(err, &e) {\n\t\t// sanity check that the error is the expected error\n\t\tif e.Code() != gogocql.ErrCodeUnavailable {\n\t\t\treturn false\n\t\t}\n\t\t// emit these errors in the condition that the database is in trouble\n\t\t// and, from an actionability standpoint, the problem points to something\n\t\t// at the database level that require resolution\n\t\t// https://github.com/apache/cassandra/blob/3c69bd23673caa40b22f3500a3e3eecabc25c8e5/src/java/org/apache/cassandra/locator/ReplicaPlans.java#L329\n\t\tif strings.Contains(e.Message(), \"Cannot perform LWT operation\") ||\n\t\t\tstrings.Contains(strings.ToLower(e.Message()), \"cannot achieve consistency level\") {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc (c client) IsCassandraConsistencyError(err error) bool {\n\tif req, ok := err.(gogocql.RequestError); ok {\n\t\t// 0x1000 == UNAVAILABLE\n\t\treturn req.Code() == 0x1000\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/public/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage public\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\n// MockError to simulate gocql.Error behavior\ntype MockError struct {\n\tgocql.RequestError\n\tcode    int\n\tmessage string\n}\n\nfunc (m MockError) Code() int {\n\treturn m.code\n}\n\nfunc (m MockError) Message() string {\n\treturn m.message\n}\n\nfunc TestClient_IsTimeoutError(t *testing.T) {\n\tclient := client{}\n\terrorMap := map[error]bool{\n\t\tnil:                             false,\n\t\tcontext.DeadlineExceeded:        true,\n\t\tgocql.ErrTimeoutNoResponse:      true,\n\t\tgocql.ErrConnectionClosed:       true,\n\t\t&gocql.RequestErrWriteTimeout{}: true,\n\t\tgocql.ErrFrameTooBig:            false,\n\t}\n\tfor err, expected := range errorMap {\n\t\tassert.Equal(t, expected, client.IsTimeoutError(err))\n\t}\n}\n\nfunc TestClient_IsNotFoundError(t *testing.T) {\n\tclient := client{}\n\terrorMap := map[error]bool{\n\t\tnil:                  false,\n\t\tgocql.ErrNotFound:    true,\n\t\tgocql.ErrFrameTooBig: false,\n\t}\n\tfor err, expected := range errorMap {\n\t\tassert.Equal(t, expected, client.IsNotFoundError(err))\n\t}\n}\n\n// TestClient_IsThrottlingError tests the IsThrottlingError function with different error codes\nfunc TestClient_IsThrottlingError(t *testing.T) {\n\tclient := client{}\n\ttests := []struct {\n\t\tname               string\n\t\tmockErrorCode      int\n\t\texpectedResult     bool\n\t\tnonCompatibleError error\n\t}{\n\t\t{\n\t\t\tname:           \"With Throttling Error\",\n\t\t\tmockErrorCode:  0x1001,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:               \"With Non-Throttling Error\",\n\t\t\tmockErrorCode:      0x0001,\n\t\t\texpectedResult:     false,\n\t\t\tnonCompatibleError: fmt.Errorf(\"with Non-Throttling Error\"),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif tt.nonCompatibleError != nil {\n\t\t\t\tresult := client.IsThrottlingError(tt.nonCompatibleError)\n\t\t\t\tassert.False(t, result)\n\t\t\t}\n\t\t\terr := MockError{code: tt.mockErrorCode}\n\t\t\tresult := client.IsThrottlingError(err)\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n}\n\nfunc TestClient_IsDBUnavailableError(t *testing.T) {\n\tclient := client{}\n\ttests := []struct {\n\t\tname           string\n\t\terr            error\n\t\texpectedResult bool\n\t}{\n\t\t{\n\t\t\tname:           \"nil error returns false\",\n\t\t\terr:            nil,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"non-compatible error returns false\",\n\t\t\terr:            fmt.Errorf(\"some generic error\"),\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"UNAVAILABLE error with LWT message returns true\",\n\t\t\terr:            MockError{code: 0x1000, message: \"Cannot perform LWT operation\"},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"UNAVAILABLE error with consistency level message returns true\",\n\t\t\terr:            MockError{code: 0x1000, message: \"Cannot achieve consistency level QUORUM\"},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"UNAVAILABLE error without matching message returns false\",\n\t\t\terr:            MockError{code: 0x1000, message: \"some other unavailable error\"},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"wrong error code with LWT message returns false\",\n\t\t\terr:            MockError{code: 0x0001, message: \"Cannot perform LWT operation\"},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"wrong error code with consistency level message returns false\",\n\t\t\terr:            MockError{code: 0x1001, message: \"Cannot achieve consistency level QUORUM\"},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"wrapped UNAVAILABLE error with LWT message returns true\",\n\t\t\terr:            fmt.Errorf(\"wrapped: %w\", MockError{code: 0x1000, message: \"Cannot perform LWT operation\"}),\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"wrapped UNAVAILABLE error with consistency level message returns true\",\n\t\t\terr:            fmt.Errorf(\"wrapped: %w\", MockError{code: 0x1000, message: \"cannot achieve consistency level QUORUM\"}),\n\t\t\texpectedResult: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := client.IsDBUnavailableError(tt.err)\n\t\t\tassert.Equal(t, tt.expectedResult, result, \"IsDBUnavailableError(%v) = %v, want %v\", tt.err, result, tt.expectedResult)\n\t\t})\n\t}\n}\n\nfunc TestClient_IsCassandraConsistencyError(t *testing.T) {\n\tclient := client{}\n\ttests := []struct {\n\t\tname               string\n\t\tmockErrorCode      int\n\t\texpectedResult     bool\n\t\tnonCompatibleError error\n\t}{\n\t\t{\n\t\t\tname:           \"With Cassandra Consistency Error\",\n\t\t\tmockErrorCode:  0x1000,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"With Non-Cassandra Consistency Error\",\n\t\t\tmockErrorCode:  0x0001,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:               \"With Non-compatible Error\",\n\t\t\tmockErrorCode:      0x0001,\n\t\t\texpectedResult:     false,\n\t\t\tnonCompatibleError: fmt.Errorf(\"with Non-compatible Error\"),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif tt.nonCompatibleError != nil {\n\t\t\t\tresult := client.IsCassandraConsistencyError(tt.nonCompatibleError)\n\t\t\t\tassert.False(t, result)\n\t\t\t}\n\t\t\terr := MockError{code: tt.mockErrorCode}\n\t\t\tresult := client.IsCassandraConsistencyError(err)\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/public/testdata.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage public\n\nimport (\n\t\"testing\"\n\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\" // needed to load cassandra plugin\n)\n\n// NewTestBaseWithPublicCassandra returns a persistence test base backed by cassandra datastore\n// It is only being used by testing against external/public Cassandra, which require to load the default gocql client\nfunc NewTestBaseWithPublicCassandra(t *testing.T, options *persistencetests.TestBaseOptions) *persistencetests.TestBase {\n\tif options.DBPluginName == \"\" {\n\t\toptions.DBPluginName = \"cassandra\"\n\t}\n\treturn persistencetests.NewTestBaseWithNoSQL(t, options)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/query.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"context\"\n\n\t\"github.com/gocql/gocql\"\n)\n\nvar _ Query = (*query)(nil)\n\ntype (\n\tquery struct {\n\t\t*gocql.Query\n\n\t\tsession *session\n\t}\n)\n\nfunc newQuery(\n\tsession *session,\n\tgocqlQuery *gocql.Query,\n) *query {\n\treturn &query{\n\t\tQuery:   gocqlQuery,\n\t\tsession: session,\n\t}\n}\n\nfunc (q *query) Exec() error {\n\terr := q.Query.Exec()\n\treturn q.handleError(err)\n}\n\nfunc (q *query) Scan(\n\tdest ...interface{},\n) error {\n\terr := q.Query.Scan(dest...)\n\treturn q.handleError(err)\n}\n\nfunc (q *query) ScanCAS(\n\tdest ...interface{},\n) (bool, error) {\n\tapplied, err := q.Query.ScanCAS(dest...)\n\treturn applied, q.handleError(err)\n}\n\nfunc (q *query) MapScan(\n\tm map[string]interface{},\n) error {\n\terr := q.Query.MapScan(m)\n\treturn q.handleError(err)\n}\n\nfunc (q *query) MapScanCAS(\n\tdest map[string]interface{},\n) (bool, error) {\n\tapplied, err := q.Query.MapScanCAS(dest)\n\treturn applied, q.handleError(err)\n}\n\nfunc (q *query) Iter() Iter {\n\titer := q.Query.Iter()\n\tif iter == nil {\n\t\treturn nil\n\t}\n\treturn iter\n}\n\nfunc (q *query) PageSize(n int) Query {\n\tq.Query.PageSize(n)\n\treturn q\n}\n\nfunc (q *query) PageState(state []byte) Query {\n\tq.Query.PageState(state)\n\treturn q\n}\n\nfunc (q *query) Consistency(c Consistency) Query {\n\tq.Query.Consistency(mustConvertConsistency(c))\n\treturn q\n}\n\nfunc (q *query) WithTimestamp(timestamp int64) Query {\n\tq.Query.WithTimestamp(timestamp)\n\treturn q\n}\n\nfunc (q *query) WithContext(ctx context.Context) Query {\n\tq2 := q.Query.WithContext(ctx)\n\tif q2 == nil {\n\t\treturn nil\n\t}\n\treturn newQuery(q.session, q2)\n}\n\nfunc (q *query) Bind(v ...interface{}) Query {\n\tq.Query.Bind(v...)\n\treturn q\n}\n\nfunc (q *query) handleError(err error) error {\n\treturn q.session.handleError(err)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/gocql/session.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage gocql\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/gocql/gocql\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\nvar _ Session = (*session)(nil)\n\nconst (\n\tsessionRefreshMinInternal = 5 * time.Second\n)\n\ntype (\n\tsession struct {\n\t\tatomic.Value // *gocql.Session\n\t\tsync.Mutex\n\n\t\tstatus          int32\n\t\tconfig          ClusterConfig\n\t\tsessionInitTime time.Time\n\t}\n)\n\nfunc NewSession(\n\tconfig ClusterConfig,\n) (Session, error) {\n\tgocqlSession, err := initSession(config)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsession := &session{\n\t\tstatus:          common.DaemonStatusStarted,\n\t\tconfig:          config,\n\t\tsessionInitTime: time.Now().UTC(),\n\t}\n\tsession.Value.Store(gocqlSession)\n\treturn session, nil\n}\n\nfunc initSession(\n\tconfig ClusterConfig,\n) (*gocql.Session, error) {\n\tcluster := newCassandraCluster(config)\n\tcluster.Consistency = mustConvertConsistency(config.Consistency)\n\tcluster.SerialConsistency = mustConvertSerialConsistency(config.SerialConsistency)\n\tcluster.Timeout = config.Timeout\n\tcluster.ConnectTimeout = config.ConnectTimeout\n\n\treturn cluster.CreateSession()\n}\n\nfunc (s *session) refresh() error {\n\tif atomic.LoadInt32(&s.status) != common.DaemonStatusStarted {\n\t\treturn nil\n\t}\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tif time.Now().UTC().Sub(s.sessionInitTime) < sessionRefreshMinInternal {\n\t\treturn nil\n\t}\n\n\tnewSession, err := initSession(s.config)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ts.sessionInitTime = time.Now().UTC()\n\toldSession := s.Value.Load().(*gocql.Session)\n\ts.Value.Store(newSession)\n\toldSession.Close()\n\treturn nil\n}\n\nfunc (s *session) Query(\n\tstmt string,\n\tvalues ...interface{},\n) Query {\n\tq := s.Value.Load().(*gocql.Session).Query(stmt, values...)\n\tif q == nil {\n\t\treturn nil\n\t}\n\treturn newQuery(s, q)\n}\n\nfunc (s *session) NewBatch(\n\tbatchType BatchType,\n) Batch {\n\tb := s.Value.Load().(*gocql.Session).NewBatch(mustConvertBatchType(batchType))\n\tif b == nil {\n\t\treturn nil\n\t}\n\treturn newBatch(b)\n}\n\nfunc (s *session) ExecuteBatch(\n\tb Batch,\n) error {\n\terr := s.Value.Load().(*gocql.Session).ExecuteBatch(b.(*batch).Batch)\n\treturn s.handleError(err)\n}\n\nfunc (s *session) MapExecuteBatchCAS(\n\tb Batch,\n\tprevious map[string]interface{},\n) (bool, Iter, error) {\n\tapplied, iter, err := s.Value.Load().(*gocql.Session).MapExecuteBatchCAS(b.(*batch).Batch, previous)\n\tif iter == nil {\n\t\treturn applied, nil, s.handleError(err)\n\t}\n\treturn applied, iter, s.handleError(err)\n}\n\nfunc (s *session) Close() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\ts.Value.Load().(*gocql.Session).Close()\n}\n\nfunc (s *session) handleError(err error) error {\n\tif err == gocql.ErrNoConnections {\n\t\t_ = s.refresh()\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/history_events.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sort\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// InsertIntoHistoryTreeAndNode inserts one or two rows: tree row and node row(at least one of them)\nfunc (db *CDB) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\tif treeRow == nil && nodeRow == nil {\n\t\treturn fmt.Errorf(\"require at least a tree row or a node row to insert\")\n\t}\n\n\tvar ancs []map[string]interface{}\n\tif treeRow != nil {\n\t\tfor _, an := range treeRow.Ancestors {\n\t\t\tvalue := make(map[string]interface{})\n\t\t\tvalue[\"end_node_id\"] = an.EndNodeID\n\t\t\tvalue[\"branch_id\"] = an.BranchID\n\t\t\tancs = append(ancs, value)\n\t\t}\n\t}\n\n\tvar err error\n\tif treeRow != nil && nodeRow != nil {\n\t\t// Note: for perf, prefer using batch for inserting more than one records\n\t\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\t\tbatch.Query(v2templateInsertTree,\n\t\t\ttreeRow.TreeID, treeRow.BranchID, ancs, persistence.UnixNanoToDBTimestamp(treeRow.CreateTimestamp.UnixNano()), treeRow.Info, treeRow.CreateTimestamp)\n\t\tbatch.Query(v2templateUpsertData,\n\t\t\tnodeRow.TreeID, nodeRow.BranchID, nodeRow.NodeID, nodeRow.TxnID, nodeRow.Data, nodeRow.DataEncoding, nodeRow.CreateTimestamp)\n\t\terr = db.session.ExecuteBatch(batch)\n\t} else {\n\t\tvar query gocql.Query\n\t\tif treeRow != nil {\n\t\t\tquery = db.session.Query(v2templateInsertTree,\n\t\t\t\ttreeRow.TreeID, treeRow.BranchID, ancs, persistence.UnixNanoToDBTimestamp(treeRow.CreateTimestamp.UnixNano()), treeRow.Info, treeRow.CreateTimestamp).WithContext(ctx)\n\t\t}\n\t\tif nodeRow != nil {\n\t\t\tquery = db.session.Query(v2templateUpsertData,\n\t\t\t\tnodeRow.TreeID, nodeRow.BranchID, nodeRow.NodeID, nodeRow.TxnID, nodeRow.Data, nodeRow.DataEncoding, nodeRow.CreateTimestamp).WithContext(ctx)\n\t\t}\n\t\terr = query.Exec()\n\t}\n\n\treturn err\n}\n\n// SelectFromHistoryNode read nodes based on a filter\nfunc (db *CDB) SelectFromHistoryNode(ctx context.Context, filter *nosqlplugin.HistoryNodeFilter) ([]*nosqlplugin.HistoryNodeRow, []byte, error) {\n\tquery := db.session.Query(v2templateReadData, filter.TreeID, filter.BranchID, filter.MinNodeID, filter.MaxNodeID).WithContext(ctx)\n\n\titer := query.PageSize(filter.PageSize).PageState(filter.NextPageToken).Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectFromHistoryNode operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\tpagingToken := iter.PageState()\n\tvar rows []*nosqlplugin.HistoryNodeRow\n\trow := &nosqlplugin.HistoryNodeRow{}\n\tfor iter.Scan(&row.NodeID, &row.TxnID, &row.Data, &row.DataEncoding) {\n\t\trows = append(rows, row)\n\t\trow = &nosqlplugin.HistoryNodeRow{}\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn rows, pagingToken, nil\n}\n\n// DeleteFromHistoryTreeAndNode delete a branch record, and a list of ranges of nodes.\nfunc (db *CDB) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *nosqlplugin.HistoryTreeFilter, nodeFilters []*nosqlplugin.HistoryNodeFilter) error {\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\tbatch.Query(v2templateDeleteBranch, treeFilter.TreeID, treeFilter.BranchID)\n\tfor _, nodeFilter := range nodeFilters {\n\t\tbatch.Query(v2templateRangeDeleteData,\n\t\t\tnodeFilter.TreeID,\n\t\t\tnodeFilter.BranchID,\n\t\t\tnodeFilter.MinNodeID)\n\t}\n\treturn db.executeBatchWithConsistencyAll(batch)\n}\n\n// SelectAllHistoryTrees will return all tree branches with pagination\nfunc (db *CDB) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*nosqlplugin.HistoryTreeRow, []byte, error) {\n\tquery := db.session.Query(v2templateScanAllTreeBranches).WithContext(ctx)\n\n\titer := query.PageSize(int(pageSize)).PageState(nextPageToken).Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectAllHistoryTrees operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\tpagingToken := iter.PageState()\n\n\tcreateTime := time.Time{}\n\tvar rows []*nosqlplugin.HistoryTreeRow\n\trow := &nosqlplugin.HistoryTreeRow{}\n\tfor iter.Scan(&row.TreeID, &row.BranchID, &createTime, &row.Info) {\n\t\trow.CreateTimestamp = time.Unix(0, persistence.DBTimestampToUnixNano(persistence.UnixNanoToDBTimestamp(createTime.UnixNano())))\n\t\trows = append(rows, row)\n\t\trow = &nosqlplugin.HistoryTreeRow{}\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn rows, pagingToken, nil\n}\n\n// SelectFromHistoryTree read branch records for a tree\nfunc (db *CDB) SelectFromHistoryTree(ctx context.Context, filter *nosqlplugin.HistoryTreeFilter) ([]*nosqlplugin.HistoryTreeRow, error) {\n\tquery := db.session.Query(v2templateReadAllBranches, filter.TreeID).WithContext(ctx)\n\tvar pagingToken []byte\n\tvar iter gocql.Iter\n\tvar rows []*nosqlplugin.HistoryTreeRow\n\tfor {\n\t\titer = query.PageSize(100).PageState(pagingToken).Iter()\n\t\tif iter == nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: \"SelectFromHistoryTree operation failed.  Not able to create query iterator.\",\n\t\t\t}\n\t\t}\n\t\tpagingToken = iter.PageState()\n\n\t\tbranchUUID := \"\"\n\t\tvar ancsResult []map[string]interface{}\n\t\t// Ideally we should just use int64. But we have been using time.Time for a long time.\n\t\t// I am not sure using a int64 will behave the same.\n\t\t// Therefore, here still using a time.Time to read, and then convert to int64\n\t\tcreateTime := time.Time{}\n\t\tinfo := \"\"\n\n\t\tfor iter.Scan(&branchUUID, &ancsResult, &createTime, &info) {\n\t\t\tancs := parseBranchAncestors(ancsResult)\n\t\t\trow := &nosqlplugin.HistoryTreeRow{\n\t\t\t\tTreeID:    filter.TreeID,\n\t\t\t\tBranchID:  branchUUID,\n\t\t\t\tAncestors: ancs,\n\t\t\t}\n\t\t\trows = append(rows, row)\n\n\t\t\tbranchUUID = \"\"\n\t\t\tancsResult = []map[string]interface{}{}\n\t\t\tcreateTime = time.Time{}\n\t\t\tinfo = \"\"\n\t\t}\n\n\t\tif err := iter.Close(); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif len(pagingToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn rows, nil\n}\n\nfunc parseBranchAncestors(\n\tancestors []map[string]interface{},\n) []*types.HistoryBranchRange {\n\n\tans := make([]*types.HistoryBranchRange, 0, len(ancestors))\n\tfor _, e := range ancestors {\n\t\tan := &types.HistoryBranchRange{}\n\t\tfor k, v := range e {\n\t\t\tswitch k {\n\t\t\tcase \"branch_id\":\n\t\t\t\tan.BranchID = v.(gocql.UUID).String()\n\t\t\tcase \"end_node_id\":\n\t\t\t\tan.EndNodeID = v.(int64)\n\t\t\t}\n\t\t}\n\t\tans = append(ans, an)\n\t}\n\n\tif len(ans) > 0 {\n\t\t// sort ans based onf EndNodeID so that we can set BeginNodeID\n\t\tsort.Slice(ans, func(i, j int) bool { return ans[i].EndNodeID < ans[j].EndNodeID })\n\t\tans[0].BeginNodeID = int64(1)\n\t\tfor i := 1; i < len(ans); i++ {\n\t\t\tans[i].BeginNodeID = ans[i-1].EndNodeID\n\t\t}\n\t}\n\treturn ans\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/history_events_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\t// below are templates for history_node table\n\tv2templateUpsertData = `INSERT INTO history_node (` +\n\t\t`tree_id, branch_id, node_id, txn_id, data, data_encoding, created_time) ` +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?) `\n\n\tv2templateReadData = `SELECT node_id, txn_id, data, data_encoding FROM history_node ` +\n\t\t`WHERE tree_id = ? AND branch_id = ? AND node_id >= ? AND node_id < ? `\n\n\tv2templateRangeDeleteData = `DELETE FROM history_node WHERE tree_id = ? AND branch_id = ? AND node_id >= ? `\n\n\t// below are templates for history_tree table\n\tv2templateInsertTree = `INSERT INTO history_tree (` +\n\t\t`tree_id, branch_id, ancestors, fork_time, info, created_time) ` +\n\t\t`VALUES (?, ?, ?, ?, ?, ?) `\n\n\tv2templateReadAllBranches = `SELECT branch_id, ancestors, fork_time, info FROM history_tree WHERE tree_id = ? `\n\n\tv2templateDeleteBranch = `DELETE FROM history_tree WHERE tree_id = ? AND branch_id = ? `\n\n\tv2templateScanAllTreeBranches = `SELECT tree_id, branch_id, fork_time, info FROM history_tree `\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/history_events_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestInsertIntoHistoryTreeAndNode(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttreeRow     *nosqlplugin.HistoryTreeRow\n\t\tnodeRow     *nosqlplugin.HistoryNodeRow\n\t\tsetupMocks  func(*gomock.Controller, *fakeSession)\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"Successfully insert both tree and node rows using batch\",\n\t\t\ttreeRow: &nosqlplugin.HistoryTreeRow{\n\t\t\t\tTreeID:          \"treeID\",\n\t\t\t\tBranchID:        \"branchID\",\n\t\t\t\tAncestors:       []*types.HistoryBranchRange{{BranchID: \"branch1\", EndNodeID: 100}},\n\t\t\t\tCreateTimestamp: time.Now(),\n\t\t\t},\n\t\t\tnodeRow: &nosqlplugin.HistoryNodeRow{\n\t\t\t\tTreeID:       \"treeID\",\n\t\t\t\tBranchID:     \"branchID\",\n\t\t\t\tNodeID:       1,\n\t\t\t\tTxnID:        nil,\n\t\t\t\tData:         []byte(\"node data\"),\n\t\t\t\tDataEncoding: \"encoding\",\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockBatch := gocql.NewMockBatch(ctrl)\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockBatch.EXPECT().WithContext(gomock.Any()).Return(mockBatch).AnyTimes()\n\t\t\t\tmockBatch.EXPECT().Query(v2templateInsertTree, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return().AnyTimes()\n\t\t\t\tmockBatch.EXPECT().Query(v2templateUpsertData, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return().AnyTimes()\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Successfully insert only tree row\",\n\t\t\ttreeRow: &nosqlplugin.HistoryTreeRow{\n\t\t\t\tTreeID:          \"treeID\",\n\t\t\t\tBranchID:        \"branchID\",\n\t\t\t\tAncestors:       []*types.HistoryBranchRange{{BranchID: \"branch1\", EndNodeID: 100}},\n\t\t\t\tCreateTimestamp: time.Now(),\n\t\t\t},\n\t\t\tnodeRow: nil, // No node row provided\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery)\n\t\t\t\tmockQuery.EXPECT().Exec().Return(nil)\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Successfully insert only node row\",\n\t\t\ttreeRow: nil, // No tree row provided\n\t\t\tnodeRow: &nosqlplugin.HistoryNodeRow{\n\t\t\t\tTreeID:       \"treeID\",\n\t\t\t\tBranchID:     \"branchID\",\n\t\t\t\tNodeID:       1,\n\t\t\t\tTxnID:        nil, // Optional field\n\t\t\t\tData:         []byte(\"node data\"),\n\t\t\t\tDataEncoding: \"encoding\",\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery)\n\t\t\t\tmockQuery.EXPECT().Exec().Return(nil)\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{}\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(ctrl, session)\n\t\t\t}\n\n\t\t\tdb := &CDB{session: session}\n\t\t\terr := db.InsertIntoHistoryTreeAndNode(context.Background(), tt.treeRow, tt.nodeRow)\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err, \"Expected an error but got none\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error but got one\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectFromHistoryNode(t *testing.T) {\n\ttxnID1 := int64(1)\n\ttxnID2 := int64(2)\n\ttests := []struct {\n\t\tname          string\n\t\tfilter        *nosqlplugin.HistoryNodeFilter\n\t\tsetupMocks    func(*gomock.Controller, *fakeSession)\n\t\texpectedRows  []*nosqlplugin.HistoryNodeRow\n\t\texpectedToken []byte\n\t\texpectError   bool\n\t}{\n\t\t{\n\t\t\tname: \"Successfully retrieve history nodes\",\n\t\t\tfilter: &nosqlplugin.HistoryNodeFilter{\n\t\t\t\tTreeID:        \"treeID\",\n\t\t\t\tBranchID:      \"branchID\",\n\t\t\t\tMinNodeID:     1,\n\t\t\t\tMaxNodeID:     10,\n\t\t\t\tPageSize:      5,\n\t\t\t\tNextPageToken: nil,\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().Iter().Return(&fakeIter{\n\t\t\t\t\tscanInputs: [][]interface{}{\n\t\t\t\t\t\t{int64(1), &txnID1, []byte(\"data1\"), \"encoding\"},\n\t\t\t\t\t\t{int64(2), &txnID2, []byte(\"data2\"), \"encoding\"},\n\t\t\t\t\t},\n\t\t\t\t\tpageState: []byte(\"nextPageToken\"),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectedRows: []*nosqlplugin.HistoryNodeRow{\n\t\t\t\t{NodeID: int64(1), TxnID: &txnID1, Data: []byte(\"data1\"), DataEncoding: \"encoding\"},\n\t\t\t\t{NodeID: int64(2), TxnID: &txnID2, Data: []byte(\"data2\"), DataEncoding: \"encoding\"},\n\t\t\t},\n\t\t\texpectedToken: []byte(\"nextPageToken\"),\n\t\t\texpectError:   false,\n\t\t},\n\t\t{\n\t\t\tname: \"Failure to create query iterator\",\n\t\t\tfilter: &nosqlplugin.HistoryNodeFilter{\n\t\t\t\tTreeID:        \"treeID\",\n\t\t\t\tBranchID:      \"branchID\",\n\t\t\t\tMinNodeID:     1,\n\t\t\t\tMaxNodeID:     10,\n\t\t\t\tPageSize:      5,\n\t\t\t\tNextPageToken: nil,\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().Iter().Return(nil).AnyTimes() // Simulating failure\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{}\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(ctrl, session)\n\t\t\t}\n\n\t\t\tdb := &CDB{session: session}\n\t\t\trows, token, err := db.SelectFromHistoryNode(context.Background(), tt.filter)\n\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedRows, rows)\n\t\t\t\tassert.Equal(t, tt.expectedToken, token)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteFromHistoryTreeAndNode(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttreeFilter  *nosqlplugin.HistoryTreeFilter\n\t\tnodeFilters []*nosqlplugin.HistoryNodeFilter\n\t\tsetupMocks  func(*fakeSession)\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"Successfully delete tree and nodes\",\n\t\t\ttreeFilter: &nosqlplugin.HistoryTreeFilter{\n\t\t\t\tShardID:  1,\n\t\t\t\tTreeID:   \"treeID\",\n\t\t\t\tBranchID: stringPtr(\"branchID\"),\n\t\t\t},\n\t\t\tnodeFilters: []*nosqlplugin.HistoryNodeFilter{\n\t\t\t\t{TreeID: \"treeID\", BranchID: \"branchID\", MinNodeID: 1},\n\t\t\t\t{TreeID: \"treeID\", BranchID: \"branchID\", MinNodeID: 2},\n\t\t\t},\n\t\t\tsetupMocks: func(session *fakeSession) {\n\t\t\t\t// Simulate successful batch execution\n\t\t\t\tsession.mapExecuteBatchCASApplied = true\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Failure in batch execution\",\n\t\t\ttreeFilter: &nosqlplugin.HistoryTreeFilter{\n\t\t\t\tShardID:  1,\n\t\t\t\tTreeID:   \"treeID\",\n\t\t\t\tBranchID: stringPtr(\"branchID\"),\n\t\t\t},\n\t\t\tnodeFilters: []*nosqlplugin.HistoryNodeFilter{\n\t\t\t\t{TreeID: \"treeID\", BranchID: \"branchID\", MinNodeID: 1},\n\t\t\t},\n\t\t\tsetupMocks: func(session *fakeSession) {\n\t\t\t\t// Simulate failure in batch execution\n\t\t\t\tsession.mapExecuteBatchCASErr = types.InternalServiceError{Message: \"DB operation failed\"}\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\n\t\t\tsession := &fakeSession{}\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(session)\n\t\t\t}\n\n\t\t\tdb := &CDB{session: session}\n\t\t\terr := db.DeleteFromHistoryTreeAndNode(context.Background(), tt.treeFilter, tt.nodeFilters)\n\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err, \"Expected an error but got none\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error but got one\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectAllHistoryTrees(t *testing.T) {\n\tlocation, err := time.LoadLocation(\"UTC\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tfixedTime, err := time.ParseInLocation(time.RFC3339, \"2023-12-12T22:08:41Z\", location)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tname          string\n\t\tnextPageToken []byte\n\t\tpageSize      int\n\t\tsetupMocks    func(*gomock.Controller, *fakeSession)\n\t\texpectedRows  []*nosqlplugin.HistoryTreeRow\n\t\texpectedToken []byte\n\t\texpectError   bool\n\t}{\n\t\t{\n\t\t\tname:          \"Successfully retrieve all history trees\",\n\t\t\tnextPageToken: []byte(\"token\"),\n\t\t\tpageSize:      10,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\n\t\t\t\tsession.iter = &fakeIter{\n\t\t\t\t\tscanInputs: [][]interface{}{\n\t\t\t\t\t\t{\"treeID1\", \"branchID1\", fixedTime, \"Tree info 1\"},\n\t\t\t\t\t\t{\"treeID2\", \"branchID2\", fixedTime.Add(time.Minute), \"Tree info 2\"},\n\t\t\t\t\t},\n\t\t\t\t\tpageState: []byte(\"nextPageToken\"),\n\t\t\t\t}\n\t\t\t\tmockQuery.EXPECT().Iter().Return(session.iter).AnyTimes()\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectedRows: []*nosqlplugin.HistoryTreeRow{\n\t\t\t\t{TreeID: \"treeID1\", BranchID: \"branchID1\", CreateTimestamp: fixedTime, Info: \"Tree info 1\"},\n\t\t\t\t{TreeID: \"treeID2\", BranchID: \"branchID2\", CreateTimestamp: fixedTime.Add(time.Minute), Info: \"Tree info 2\"},\n\t\t\t},\n\t\t\texpectedToken: []byte(\"nextPageToken\"),\n\t\t\texpectError:   false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Failed to create query iterator\",\n\t\t\tnextPageToken: []byte(\"token\"),\n\t\t\tpageSize:      10,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().Iter().Return(nil).AnyTimes() // Simulate failure to create iterator\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{}\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(ctrl, session)\n\t\t\t}\n\n\t\t\tdb := &CDB{session: session}\n\t\t\trows, token, err := db.SelectAllHistoryTrees(context.Background(), tt.nextPageToken, tt.pageSize)\n\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tfor i, row := range rows {\n\t\t\t\t\tassertHistoryTreeRowEqual(t, tt.expectedRows[i], row)\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tt.expectedToken, token)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectFromHistoryTree(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tfilter       *nosqlplugin.HistoryTreeFilter\n\t\tsetupMocks   func(*gomock.Controller, *fakeSession)\n\t\texpectedRows []*nosqlplugin.HistoryTreeRow\n\t\texpectError  bool\n\t}{\n\t\t{\n\t\t\tname: \"Successfully retrieve branches\",\n\t\t\tfilter: &nosqlplugin.HistoryTreeFilter{\n\t\t\t\tTreeID: \"treeID\",\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\n\t\t\t\titer1 := newFakeIter([][]interface{}{\n\t\t\t\t\t{\"branchUUID1\", []map[string]interface{}{{\"branch_id\": uuid.Parse(permanentRunID), \"end_node_id\": int64(10)}}, time.Now(), \"Info1\"},\n\t\t\t\t}, []byte(\"nextPageToken\"))\n\n\t\t\t\titer2 := newFakeIter([][]interface{}{\n\t\t\t\t\t{\"branchUUID2\", []map[string]interface{}{{\"branch_id\": uuid.Parse(permanentRunID), \"end_node_id\": int64(20)}}, time.Now(), \"Info2\"},\n\t\t\t\t}, nil) // No more pages\n\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockQuery.EXPECT().Iter().Return(iter1),\n\t\t\t\t\tmockQuery.EXPECT().Iter().Return(iter2),\n\t\t\t\t)\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectedRows: []*nosqlplugin.HistoryTreeRow{\n\t\t\t\t{TreeID: \"treeID\", BranchID: \"branchUUID1\", Ancestors: []*types.HistoryBranchRange{{BranchID: permanentRunID, EndNodeID: 10, BeginNodeID: 1}}, Info: \"\"},\n\t\t\t\t{TreeID: \"treeID\", BranchID: \"branchUUID2\", Ancestors: []*types.HistoryBranchRange{{BranchID: permanentRunID, EndNodeID: 20, BeginNodeID: 1}}, Info: \"\"},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to create query iterator\",\n\t\t\tfilter: &nosqlplugin.HistoryTreeFilter{\n\t\t\t\tTreeID: \"treeID\",\n\t\t\t},\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, session *fakeSession) {\n\t\t\t\tmockQuery := gocql.NewMockQuery(ctrl)\n\t\t\t\tmockQuery.EXPECT().WithContext(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageSize(gomock.Any()).Return(mockQuery).AnyTimes()\n\t\t\t\tmockQuery.EXPECT().PageState(gomock.Any()).Return(mockQuery).AnyTimes()\n\n\t\t\t\tmockQuery.EXPECT().Iter().Return(nil) // Simulate failure to create iterator\n\n\t\t\t\tsession.query = mockQuery\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{}\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(ctrl, session)\n\t\t\t}\n\n\t\t\tdb := &CDB{session: session}\n\t\t\trows, err := db.SelectFromHistoryTree(context.Background(), tt.filter)\n\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedRows, rows)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Helper to create a fake iterator with predefined results\nfunc newFakeIter(data [][]interface{}, nextPageToken []byte) *fakeIter {\n\treturn &fakeIter{\n\t\tscanInputs: data,\n\t\tpageState:  nextPageToken,\n\t}\n}\n\nfunc assertHistoryTreeRowEqual(t *testing.T, expected, actual *nosqlplugin.HistoryTreeRow) {\n\tassert.Equal(t, expected.TreeID, actual.TreeID)\n\tassert.Equal(t, expected.BranchID, actual.BranchID)\n\tassert.Equal(t, expected.Info, actual.Info)\n\tassert.True(t, expected.CreateTimestamp.Equal(actual.CreateTimestamp), \"Timestamps do not match\")\n}\n\nfunc stringPtr(s string) *string {\n\treturn &s\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/plugin.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\tgogocql \"github.com/gocql/gocql\"\n\t\"github.com/hailocab/go-hostpool\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/environment\"\n)\n\nconst (\n\t// PluginName is the name of the plugin\n\tPluginName            = \"cassandra\"\n\tdefaultSessionTimeout = 10 * time.Second\n\tdefaultConnectTimeout = 2 * time.Second\n)\n\ntype plugin struct{}\n\nvar _ nosqlplugin.Plugin = (*plugin)(nil)\n\nfunc init() {\n\tnosql.RegisterPlugin(PluginName, &plugin{})\n}\n\n// CreateDB initialize the db object\nfunc (p *plugin) CreateDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.DB, error) {\n\treturn p.doCreateDB(cfg, logger, dc)\n}\n\n// CreateAdminDB initialize the AdminDB object\nfunc (p *plugin) CreateAdminDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.AdminDB, error) {\n\t// the keyspace is not created yet, so use empty and let the Cassandra connect\n\tkeyspace := cfg.Keyspace\n\tcfg.Keyspace = \"\"\n\t// change it back\n\tdefer func() {\n\t\tcfg.Keyspace = keyspace\n\t}()\n\n\treturn p.doCreateDB(cfg, logger, dc)\n}\n\nfunc (p *plugin) doCreateDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (*CDB, error) {\n\tgocqlConfig, err := toGoCqlConfig(cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsession, err := gocql.GetRegisteredClient().CreateSession(gocqlConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdb := NewCassandraDBFromSession(cfg, session, logger, dc)\n\treturn db, nil\n}\n\nfunc toGoCqlConfig(cfg *config.NoSQL) (gocql.ClusterConfig, error) {\n\tvar err error\n\tif cfg.Port == 0 {\n\t\tcfg.Port, err = environment.GetCassandraPort()\n\t\tif err != nil {\n\t\t\treturn gocql.ClusterConfig{}, err\n\t\t}\n\t}\n\tif cfg.Hosts == \"\" {\n\t\tcfg.Hosts = environment.GetCassandraAddress()\n\t}\n\tif cfg.ProtoVersion == 0 {\n\t\tcfg.ProtoVersion, err = environment.GetCassandraProtoVersion()\n\t\tif err != nil {\n\t\t\treturn gocql.ClusterConfig{}, err\n\t\t}\n\t}\n\n\tif cfg.Timeout == 0 {\n\t\tcfg.Timeout = defaultSessionTimeout\n\t}\n\n\tif cfg.ConnectTimeout == 0 {\n\t\tcfg.ConnectTimeout = defaultConnectTimeout\n\t}\n\n\tif cfg.Consistency == \"\" {\n\t\tcfg.Consistency = cassandraDefaultConsLevel.String()\n\t}\n\n\tif cfg.SerialConsistency == \"\" {\n\t\tcfg.SerialConsistency = cassandraDefaultSerialConsLevel.String()\n\t}\n\n\tconsistency, err := gocql.ParseConsistency(cfg.Consistency)\n\tif err != nil {\n\t\treturn gocql.ClusterConfig{}, err\n\t}\n\n\tserialConsistency, err := gocql.ParseSerialConsistency(cfg.SerialConsistency)\n\tif err != nil {\n\t\treturn gocql.ClusterConfig{}, err\n\t}\n\n\thostSelection, err := toHostSelectionPolicy(cfg.HostSelectionPolicy)\n\tif err != nil {\n\t\treturn gocql.ClusterConfig{}, err\n\t}\n\n\treturn gocql.ClusterConfig{\n\t\tHosts:                 cfg.Hosts,\n\t\tPort:                  cfg.Port,\n\t\tUser:                  cfg.User,\n\t\tPassword:              cfg.Password,\n\t\tAllowedAuthenticators: cfg.AllowedAuthenticators,\n\t\tKeyspace:              cfg.Keyspace,\n\t\tRegion:                cfg.Region,\n\t\tDatacenter:            cfg.Datacenter,\n\t\tMaxConns:              cfg.MaxConns,\n\t\tTLS:                   cfg.TLS,\n\t\tProtoVersion:          cfg.ProtoVersion,\n\t\tConsistency:           consistency,\n\t\tSerialConsistency:     serialConsistency,\n\t\tTimeout:               cfg.Timeout,\n\t\tConnectTimeout:        cfg.ConnectTimeout,\n\t\tHostSelectionPolicy:   hostSelection,\n\t}, nil\n}\n\nfunc toHostSelectionPolicy(policy string) (gogocql.HostSelectionPolicy, error) {\n\tswitch policy {\n\tcase \"\", \"tokenaware,roundrobin\":\n\t\treturn gogocql.TokenAwareHostPolicy(gogocql.RoundRobinHostPolicy()), nil\n\tcase \"hostpool-epsilon-greedy\":\n\t\treturn gogocql.HostPoolHostPolicy(\n\t\t\thostpool.NewEpsilonGreedy(nil, 0, &hostpool.LinearEpsilonValueCalculator{}),\n\t\t), nil\n\tcase \"roundrobin\":\n\t\treturn gogocql.RoundRobinHostPolicy(), nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown gocql host selection policy: %q\", policy)\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/plugin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\tgogocql \"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/environment\"\n)\n\nfunc Test_toGoCqlConfig(t *testing.T) {\n\tt.Setenv(environment.CassandraSeeds, environment.Localhost)\n\ttests := []struct {\n\t\tname    string\n\t\tcfg     *config.NoSQL\n\t\twant    gocql.ClusterConfig\n\t\twantErr assert.ErrorAssertionFunc\n\t}{\n\t\t{\n\t\t\t\"empty config will be filled with defaults\",\n\t\t\t&config.NoSQL{},\n\t\t\tgocql.ClusterConfig{\n\t\t\t\tHosts:               environment.Localhost,\n\t\t\t\tPort:                9042,\n\t\t\t\tProtoVersion:        4,\n\t\t\t\tTimeout:             time.Second * 10,\n\t\t\t\tConsistency:         gocql.LocalQuorum,\n\t\t\t\tSerialConsistency:   gocql.LocalSerial,\n\t\t\t\tConnectTimeout:      time.Second * 2,\n\t\t\t\tHostSelectionPolicy: gogocql.TokenAwareHostPolicy(gogocql.RoundRobinHostPolicy()),\n\t\t\t},\n\t\t\tassert.NoError,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, err := toGoCqlConfig(tt.cfg)\n\t\t\tif !tt.wantErr(t, err, fmt.Sprintf(\"toGoCqlConfig(%v)\", tt.cfg)) {\n\t\t\t\treturn\n\t\t\t}\n\t\t\tassert.Equalf(t, tt.want, got, \"toGoCqlConfig(%v)\", tt.cfg)\n\t\t})\n\t}\n}\n\nfunc Test_toHostSelectionPolicy(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\tpolicy  string\n\t\twant    gogocql.HostSelectionPolicy\n\t\twantErr assert.ErrorAssertionFunc\n\t}{\n\t\t{name: \"not supported policy\", policy: \"non-existing\", want: nil, wantErr: assert.Error},\n\t\t{name: \"Default policy\", policy: \"\", want: gogocql.TokenAwareHostPolicy(gogocql.RoundRobinHostPolicy()), wantErr: assert.NoError},\n\t\t{name: \"Round robin policy\", policy: \"roundrobin\", want: gogocql.RoundRobinHostPolicy(), wantErr: assert.NoError},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, err := toHostSelectionPolicy(tt.policy)\n\t\t\tif !tt.wantErr(t, err, fmt.Sprintf(\"toHostSelectionPolicy(%v)\", tt.policy)) {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.want, got, \"toHostSelectionPolicy(%v)\", tt.policy)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Insert message into queue, return error if failed or already exists\n// Must return ConditionFailure error if row already exists\nfunc (db *CDB) InsertIntoQueue(\n\tctx context.Context,\n\trow *nosqlplugin.QueueMessageRow,\n) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tquery := db.session.Query(templateEnqueueMessageQuery, row.QueueType, row.ID, row.Payload, timeStamp).WithContext(ctx)\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\treturn nosqlplugin.NewConditionFailure(\"queue\")\n\t}\n\treturn nil\n}\n\n// Get the ID of last message inserted into the queue\nfunc (db *CDB) SelectLastEnqueuedMessageID(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\tquery := db.session.Query(templateGetLastMessageIDQuery, queueType).WithContext(ctx)\n\tresult := make(map[string]interface{})\n\terr := query.MapScan(result)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\treturn result[\"message_id\"].(int64), nil\n}\n\n// Read queue messages starting from the exclusiveBeginMessageID\nfunc (db *CDB) SelectMessagesFrom(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tmaxRows int,\n) ([]*nosqlplugin.QueueMessageRow, error) {\n\t// Reading replication tasks need to be quorum level consistent, otherwise we could loose task\n\tquery := db.session.Query(templateGetMessagesQuery,\n\t\tqueueType,\n\t\texclusiveBeginMessageID,\n\t\tmaxRows,\n\t).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, fmt.Errorf(\"SelectMessagesFrom operation failed. Not able to create query iterator\")\n\t}\n\n\tvar result []*nosqlplugin.QueueMessageRow\n\tmessage := make(map[string]interface{})\n\tfor iter.MapScan(message) {\n\t\tpayload := getMessagePayload(message)\n\t\tid := getMessageID(message)\n\t\tresult = append(result, &nosqlplugin.QueueMessageRow{ID: id, Payload: payload})\n\t\tmessage = make(map[string]interface{})\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn result, nil\n}\n\n// Read queue message starting from exclusiveBeginMessageID int64, inclusiveEndMessageID int64\nfunc (db *CDB) SelectMessagesBetween(\n\tctx context.Context,\n\trequest nosqlplugin.SelectMessagesBetweenRequest,\n) (*nosqlplugin.SelectMessagesBetweenResponse, error) {\n\t// Reading replication tasks need to be quorum level consistent, otherwise we could loose task\n\t// Use negative queue type as the dlq type\n\tquery := db.session.Query(templateGetMessagesFromDLQQuery,\n\t\trequest.QueueType,\n\t\trequest.ExclusiveBeginMessageID,\n\t\trequest.InclusiveEndMessageID,\n\t).PageSize(request.PageSize).PageState(request.NextPageToken).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, fmt.Errorf(\"SelectMessagesBetween operation failed. Not able to create query iterator\")\n\t}\n\n\tvar rows []nosqlplugin.QueueMessageRow\n\tmessage := make(map[string]interface{})\n\tfor iter.MapScan(message) {\n\t\tpayload := getMessagePayload(message)\n\t\tid := getMessageID(message)\n\t\trows = append(rows, nosqlplugin.QueueMessageRow{ID: id, Payload: payload})\n\t\tmessage = make(map[string]interface{})\n\t}\n\n\tnextPageToken := iter.PageState()\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &nosqlplugin.SelectMessagesBetweenResponse{\n\t\tRows:          rows,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\n// Delete all messages before exclusiveBeginMessageID\nfunc (db *CDB) DeleteMessagesBefore(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n) error {\n\tquery := db.session.Query(templateRangeDeleteMessagesBeforeQuery, queueType, exclusiveBeginMessageID).WithContext(ctx)\n\treturn db.executeWithConsistencyAll(query)\n}\n\n// Delete all messages in a range between exclusiveBeginMessageID and inclusiveEndMessageID\nfunc (db *CDB) DeleteMessagesInRange(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tinclusiveEndMessageID int64,\n) error {\n\tquery := db.session.Query(templateRangeDeleteMessagesBetweenQuery, queueType, exclusiveBeginMessageID, inclusiveEndMessageID).WithContext(ctx)\n\treturn db.executeWithConsistencyAll(query)\n}\n\n// Delete one message\nfunc (db *CDB) DeleteMessage(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n) error {\n\tquery := db.session.Query(templateDeleteMessageQuery, queueType, messageID).WithContext(ctx)\n\treturn db.executeWithConsistencyAll(query)\n}\n\n// Insert an empty metadata row, starting from a version\nfunc (db *CDB) InsertQueueMetadata(ctx context.Context, row nosqlplugin.QueueMetadataRow) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tclusterAckLevels := map[string]int64{}\n\tquery := db.session.Query(templateInsertQueueMetadataQuery, row.QueueType, clusterAckLevels, row.Version, timeStamp).WithContext(ctx)\n\n\t// NOTE: Must pass nils to be compatible with ScyllaDB's LWT behavior\n\t// \"Scylla always returns the old version of the row, regardless of whether the condition is true or not.\"\n\t// See also https://docs.scylladb.com/kb/lwt-differences/\n\t_, err := query.ScanCAS(nil, nil, nil, nil, nil)\n\tif err != nil {\n\t\treturn err\n\t}\n\t// it's ok if the query is not applied, which means that the record exists already.\n\treturn nil\n}\n\n// **Conditionally** update a queue metadata row, if current version is matched(meaning current == row.Version - 1),\n// then the current version will increase by one when updating the metadata row\n// it should return ConditionFailure if the condition is not met\nfunc (db *CDB) UpdateQueueMetadataCas(\n\tctx context.Context,\n\trow nosqlplugin.QueueMetadataRow,\n) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tquery := db.session.Query(templateUpdateQueueMetadataQuery,\n\t\trow.ClusterAckLevels,\n\t\trow.Version,\n\t\ttimeStamp,\n\t\trow.QueueType,\n\t\trow.Version-1,\n\t).WithContext(ctx)\n\n\t// NOTE: Must pass nils to be compatible with ScyllaDB's LWT behavior\n\t// \"Scylla always returns the old version of the row, regardless of whether the condition is true or not.\"\n\t// See also https://docs.scylladb.com/kb/lwt-differences/\n\tapplied, err := query.ScanCAS(nil, nil, nil, nil, nil)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !applied {\n\t\treturn nosqlplugin.NewConditionFailure(\"queue\")\n\t}\n\n\treturn nil\n}\n\n// Read a QueueMetadata\nfunc (db *CDB) SelectQueueMetadata(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (*nosqlplugin.QueueMetadataRow, error) {\n\tquery := db.session.Query(templateGetQueueMetadataQuery, queueType).WithContext(ctx)\n\tvar ackLevels map[string]int64\n\tvar version int64\n\terr := query.Scan(&ackLevels, &version)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// if record exist but ackLevels is empty, we initialize the map\n\tif ackLevels == nil {\n\t\tackLevels = make(map[string]int64)\n\t}\n\treturn &nosqlplugin.QueueMetadataRow{\n\t\tQueueType:        queueType,\n\t\tClusterAckLevels: ackLevels,\n\t\tVersion:          version,\n\t}, nil\n}\n\nfunc (db *CDB) GetQueueSize(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tquery := db.session.Query(templateGetQueueSizeQuery, queueType).WithContext(ctx)\n\tresult := make(map[string]interface{})\n\n\tif err := query.MapScan(result); err != nil {\n\t\treturn 0, err\n\t}\n\treturn result[\"count\"].(int64), nil\n}\n\nfunc getMessagePayload(message map[string]interface{}) []byte {\n\treturn message[\"message_payload\"].([]byte)\n}\n\nfunc getMessageID(message map[string]interface{}) int64 {\n\treturn message[\"message_id\"].(int64)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/queue_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\ttemplateEnqueueMessageQuery             = `INSERT INTO queue (queue_type, message_id, message_payload, created_time) VALUES(?, ?, ?, ?) IF NOT EXISTS`\n\ttemplateGetLastMessageIDQuery           = `SELECT message_id FROM queue WHERE queue_type=? ORDER BY message_id DESC LIMIT 1`\n\ttemplateGetMessagesQuery                = `SELECT message_id, message_payload FROM queue WHERE queue_type = ? and message_id > ? LIMIT ?`\n\ttemplateGetMessagesFromDLQQuery         = `SELECT message_id, message_payload FROM queue WHERE queue_type = ? and message_id > ? and message_id <= ?`\n\ttemplateRangeDeleteMessagesBeforeQuery  = `DELETE FROM queue WHERE queue_type = ? and message_id < ?`\n\ttemplateRangeDeleteMessagesBetweenQuery = `DELETE FROM queue WHERE queue_type = ? and message_id > ? and message_id <= ?`\n\ttemplateDeleteMessageQuery              = `DELETE FROM queue WHERE queue_type = ? and message_id = ?`\n\ttemplateGetQueueMetadataQuery           = `SELECT cluster_ack_level, version FROM queue_metadata WHERE queue_type = ?`\n\ttemplateInsertQueueMetadataQuery        = `INSERT INTO queue_metadata (queue_type, cluster_ack_level, version, created_time) VALUES(?, ?, ?, ?) IF NOT EXISTS`\n\ttemplateUpdateQueueMetadataQuery        = `UPDATE queue_metadata SET cluster_ack_level = ?, version = ?, last_updated_time = ? WHERE queue_type = ? IF version = ?`\n\ttemplateGetQueueSizeQuery               = `SELECT COUNT(1) AS count FROM queue WHERE queue_type=?`\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/queue_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nfunc TestInsertIntoQueue(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.QueueMessageRow\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully applied\",\n\t\t\trow:  queueMessageRow(101),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO queue (queue_type, message_id, message_payload, created_time) VALUES(1, 101, [116 101 115 116 45 112 97 121 108 111 97 100 45 49 48 49], 2025-01-06T15:00:00Z) IF NOT EXISTS`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow:  queueMessageRow(101),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscancas failed\",\n\t\t\trow:  queueMessageRow(101),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"mapscancas failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertIntoQueue(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectLastEnqueuedMessageID(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantMsgID   int64\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:      \"success with shard map fully populated\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"message_id\"] = int64(101)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantMsgID: int64(101),\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT message_id FROM queue WHERE queue_type=1 ORDER BY message_id DESC LIMIT 1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\treturn errors.New(\"mapscan failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotMsgID, err := db.SelectLastEnqueuedMessageID(context.Background(), tc.queueType)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotMsgID != tc.wantMsgID {\n\t\t\t\tt.Errorf(\"Got message ID = %v, want %v\", gotMsgID, tc.wantMsgID)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectQueueMetadata(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantRow     *nosqlplugin.QueueMetadataRow\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:      \"success\",\n\t\t\tqueueType: persistence.QueueType(2),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tackLevels := args[0].(*map[string]int64)\n\t\t\t\t\t*ackLevels = make(map[string]int64)\n\t\t\t\t\t(*ackLevels)[\"cluster1\"] = 1000\n\t\t\t\t\t(*ackLevels)[\"cluster2\"] = 2000\n\t\t\t\t\tversion := args[1].(*int64)\n\t\t\t\t\t*version = int64(25)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRow: &nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        persistence.QueueType(2),\n\t\t\t\tClusterAckLevels: map[string]int64{\"cluster1\": 1000, \"cluster2\": 2000},\n\t\t\t\tVersion:          25,\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT cluster_ack_level, version FROM queue_metadata WHERE queue_type = 2`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"success with empty acklevels\",\n\t\t\tqueueType: persistence.QueueType(2),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\tversion := args[1].(*int64)\n\t\t\t\t\t*version = int64(26)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRow: &nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        persistence.QueueType(2),\n\t\t\t\tClusterAckLevels: map[string]int64{},\n\t\t\t\tVersion:          26,\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT cluster_ack_level, version FROM queue_metadata WHERE queue_type = 2`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"scan failure\",\n\t\t\tqueueType: persistence.QueueType(2),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotRow, err := db.SelectQueueMetadata(context.Background(), tc.queueType)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantRow, gotRow); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Row mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetQueueSize(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantCount   int64\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:      \"success with shard map fully populated\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"count\"] = int64(12)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantCount: int64(12),\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT COUNT(1) AS count FROM queue WHERE queue_type=1`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\treturn errors.New(\"mapscan failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotCount, err := db.GetQueueSize(context.Background(), tc.queueType)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotCount != tc.wantCount {\n\t\t\t\tt.Errorf(\"Got message ID = %v, want %v\", gotCount, tc.wantCount)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectMessagesFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname                    string\n\t\tqueueType               persistence.QueueType\n\t\texclusiveBeginMessageID int64\n\t\tmaxRows                 int\n\t\titer                    *fakeIter\n\t\twantQueries             []string\n\t\twantRows                []*nosqlplugin.QueueMessageRow\n\t\twantErr                 bool\n\t}{\n\t\t{\n\t\t\tname:                    \"nil iter\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 20,\n\t\t\tmaxRows:                 10,\n\t\t\titer:                    nil,\n\t\t\twantErr:                 true,\n\t\t},\n\t\t{\n\t\t\tname:                    \"iter close failed\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 20,\n\t\t\tmaxRows:                 10,\n\t\t\titer:                    &fakeIter{closeErr: errors.New(\"some random error\")},\n\t\t\twantErr:                 true,\n\t\t},\n\t\t{\n\t\t\tname:                    \"success\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 20,\n\t\t\tmaxRows:                 10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"message_id\":      int64(21),\n\t\t\t\t\t\t\"message_payload\": []byte(\"test-payload-1\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"message_id\":      int64(22),\n\t\t\t\t\t\t\"message_payload\": []byte(\"test-payload-2\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantRows: []*nosqlplugin.QueueMessageRow{\n\t\t\t\t{\n\t\t\t\t\tID:      21,\n\t\t\t\t\tPayload: []byte(\"test-payload-1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:      22,\n\t\t\t\t\tPayload: []byte(\"test-payload-2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT message_id, message_payload FROM queue WHERE queue_type = 1 and message_id > 20 LIMIT 10`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\tgotRows, err := db.SelectMessagesFrom(context.Background(), tc.queueType, tc.exclusiveBeginMessageID, tc.maxRows)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantRows, gotRows); diff != \"\" {\n\t\t\t\tt.Fatalf(\"rows mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Fatal(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectMessagesBetween(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\trequest     nosqlplugin.SelectMessagesBetweenRequest\n\t\titer        *fakeIter\n\t\twantQueries []string\n\t\twantResp    *nosqlplugin.SelectMessagesBetweenResponse\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"nil iter\",\n\t\t\trequest: nosqlplugin.SelectMessagesBetweenRequest{\n\t\t\t\tQueueType:               persistence.DomainReplicationQueueType,\n\t\t\t\tExclusiveBeginMessageID: 50,\n\t\t\t\tInclusiveEndMessageID:   60,\n\t\t\t\tPageSize:                5,\n\t\t\t\tNextPageToken:           []byte(\"next page token\"),\n\t\t\t},\n\t\t\titer:    nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"iter close failed\",\n\t\t\trequest: nosqlplugin.SelectMessagesBetweenRequest{\n\t\t\t\tQueueType:               persistence.DomainReplicationQueueType,\n\t\t\t\tExclusiveBeginMessageID: 50,\n\t\t\t\tInclusiveEndMessageID:   60,\n\t\t\t\tPageSize:                5,\n\t\t\t\tNextPageToken:           []byte(\"next page token\"),\n\t\t\t},\n\t\t\titer:    &fakeIter{closeErr: errors.New(\"some random error\")},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\trequest: nosqlplugin.SelectMessagesBetweenRequest{\n\t\t\t\tQueueType:               persistence.DomainReplicationQueueType,\n\t\t\t\tExclusiveBeginMessageID: 50,\n\t\t\t\tInclusiveEndMessageID:   60,\n\t\t\t\tPageSize:                5,\n\t\t\t\tNextPageToken:           []byte(\"next page token\"),\n\t\t\t},\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"message_id\":      int64(51),\n\t\t\t\t\t\t\"message_payload\": []byte(\"test-payload-1\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"message_id\":      int64(52),\n\t\t\t\t\t\t\"message_payload\": []byte(\"test-payload-2\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"message_id\":      int64(53),\n\t\t\t\t\t\t\"message_payload\": []byte(\"test-payload-3\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"more pages\"),\n\t\t\t},\n\t\t\twantResp: &nosqlplugin.SelectMessagesBetweenResponse{\n\t\t\t\tRows: []nosqlplugin.QueueMessageRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:      51,\n\t\t\t\t\t\tPayload: []byte(\"test-payload-1\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tID:      52,\n\t\t\t\t\t\tPayload: []byte(\"test-payload-2\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tID:      53,\n\t\t\t\t\t\tPayload: []byte(\"test-payload-3\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: []byte(\"more pages\"),\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT message_id, message_payload FROM queue WHERE queue_type = 1 and message_id > 50 and message_id <= 60`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(tc.request.PageSize).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(tc.request.NextPageToken).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\tgotResp, err := db.SelectMessagesBetween(context.Background(), tc.request)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantResp, gotResp); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Response mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Fatal(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessagesBefore(t *testing.T) {\n\ttests := []struct {\n\t\tname                    string\n\t\tqueueType               persistence.QueueType\n\t\texclusiveBeginMessageID int64\n\t\tqueryMockFn             func(query *gocql.MockQuery)\n\t\twantQueries             []string\n\t\twantErr                 bool\n\t}{\n\t\t{\n\t\t\tname:                    \"success\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 100,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM queue WHERE queue_type = 1 and message_id < 100`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                    \"failure\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 100,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\terr := db.DeleteMessagesBefore(context.Background(), tc.queueType, tc.exclusiveBeginMessageID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessagesInRange(t *testing.T) {\n\ttests := []struct {\n\t\tname                    string\n\t\tqueueType               persistence.QueueType\n\t\texclusiveBeginMessageID int64\n\t\tinclusiveEndMsgID       int64\n\t\tqueryMockFn             func(query *gocql.MockQuery)\n\t\twantQueries             []string\n\t\twantErr                 bool\n\t}{\n\t\t{\n\t\t\tname:                    \"success\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 100,\n\t\t\tinclusiveEndMsgID:       200,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM queue WHERE queue_type = 1 and message_id > 100 and message_id <= 200`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                    \"failure\",\n\t\t\tqueueType:               persistence.DomainReplicationQueueType,\n\t\t\texclusiveBeginMessageID: 100,\n\t\t\tinclusiveEndMsgID:       200,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\terr := db.DeleteMessagesInRange(context.Background(), tc.queueType, tc.exclusiveBeginMessageID, tc.inclusiveEndMsgID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessage(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tmsgID       int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:      \"success\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmsgID:     36,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM queue WHERE queue_type = 1 and message_id = 36`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"failure\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmsgID:     36,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\terr := db.DeleteMessage(context.Background(), tc.queueType, tc.msgID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertQueueMetadata(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tversion     int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:      \"success\",\n\t\t\tqueueType: persistence.QueueType(2),\n\t\t\tversion:   25,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().ScanCAS(gomock.Any()).Return(false, nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO queue_metadata (queue_type, cluster_ack_level, version, created_time) VALUES(2, map[], 25, 2025-01-06T15:00:00Z) IF NOT EXISTS`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"scan failure\",\n\t\t\tqueueType: persistence.QueueType(2),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().ScanCAS(gomock.Any()).Return(false, errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\trow := nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        tc.queueType,\n\t\t\t\tVersion:          tc.version,\n\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t}\n\t\t\terr := db.InsertQueueMetadata(context.Background(), row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateQueueMetadataCas(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\trow         nosqlplugin.QueueMetadataRow\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully applied\",\n\t\t\trow: nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        persistence.QueueType(2),\n\t\t\t\tClusterAckLevels: map[string]int64{\"cluster1\": 1000, \"cluster2\": 2000},\n\t\t\t\tVersion:          25,\n\t\t\t\tCurrentTimeStamp: FixedTime,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().ScanCAS(gomock.Any()).Return(true, nil).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE queue_metadata SET cluster_ack_level = map[cluster1:1000 cluster2:2000], version = 25, last_updated_time = 2025-01-06T15:00:00Z WHERE queue_type = 2 IF version = 24`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"could not apply\",\n\t\t\trow: nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        persistence.QueueType(2),\n\t\t\t\tClusterAckLevels: map[string]int64{\"cluster1\": 1000, \"cluster2\": 2000},\n\t\t\t\tVersion:          25,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().ScanCAS(gomock.Any()).Return(false, nil).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"scancas failed\",\n\t\t\trow: nosqlplugin.QueueMetadataRow{\n\t\t\t\tQueueType:        persistence.QueueType(2),\n\t\t\t\tClusterAckLevels: map[string]int64{\"cluster1\": 1000, \"cluster2\": 2000},\n\t\t\t\tVersion:          25,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().ScanCAS(gomock.Any()).Return(false, errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateQueueMetadataCas(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\nfunc queueMessageRow(id int64) *nosqlplugin.QueueMessageRow {\n\treturn &nosqlplugin.QueueMessageRow{\n\t\tQueueType:        persistence.DomainReplicationQueueType,\n\t\tID:               id,\n\t\tPayload:          []byte(fmt.Sprintf(\"test-payload-%d\", id)),\n\t\tCurrentTimeStamp: FixedTime,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/shard.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertShard creates a new shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) InsertShard(ctx context.Context, row *nosqlplugin.ShardRow) error {\n\tcqlNowTimestamp := persistence.UnixNanoToDBTimestamp(row.CurrentTimestamp.UnixNano())\n\tmarkerData, markerEncoding := persistence.FromDataBlob(row.PendingFailoverMarkers)\n\ttransferPQS, transferPQSEncoding := persistence.FromDataBlob(row.TransferProcessingQueueStates)\n\ttimerPQS, timerPQSEncoding := persistence.FromDataBlob(row.TimerProcessingQueueStates)\n\tquery := db.session.Query(templateCreateShardQuery,\n\t\trow.ShardID,\n\t\trowTypeShard,\n\t\trowTypeShardDomainID,\n\t\trowTypeShardWorkflowID,\n\t\trowTypeShardRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeShardTaskID,\n\t\trow.ShardID,\n\t\trow.Owner,\n\t\trow.RangeID,\n\t\trow.StolenSinceRenew,\n\t\tcqlNowTimestamp,\n\t\trow.ReplicationAckLevel,\n\t\trow.TransferAckLevel,\n\t\trow.TimerAckLevel,\n\t\trow.ClusterTransferAckLevel,\n\t\trow.ClusterTimerAckLevel,\n\t\ttransferPQS,\n\t\ttransferPQSEncoding,\n\t\ttimerPQS,\n\t\ttimerPQSEncoding,\n\t\trow.DomainNotificationVersion,\n\t\trow.ClusterReplicationLevel,\n\t\trow.ReplicationDLQAckLevel,\n\t\tmarkerData,\n\t\tmarkerEncoding,\n\t\trow.Data,\n\t\trow.DataEncoding,\n\t\trow.RangeID,\n\t).WithContext(ctx)\n\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\treturn convertToConflictedShardRow(previous)\n\t}\n\n\treturn nil\n}\n\nfunc convertToConflictedShardRow(previous map[string]interface{}) error {\n\trangeID := previous[\"range_id\"].(int64)\n\tvar columns []string\n\tfor k, v := range previous {\n\t\tcolumns = append(columns, fmt.Sprintf(\"%s=%v\", k, v))\n\t}\n\treturn &nosqlplugin.ShardOperationConditionFailure{\n\t\tRangeID: rangeID,\n\t\tDetails: strings.Join(columns, \",\"),\n\t}\n}\n\n// SelectShard gets a shard\nfunc (db *CDB) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *nosqlplugin.ShardRow, error) {\n\tquery := db.session.Query(templateGetShardQuery,\n\t\tshardID,\n\t\trowTypeShard,\n\t\trowTypeShardDomainID,\n\t\trowTypeShardWorkflowID,\n\t\trowTypeShardRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeShardTaskID,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\treturn 0, nil, err\n\t}\n\n\trangeID := result[\"range_id\"].(int64)\n\tshard := result[\"shard\"].(map[string]interface{})\n\tshardInfoRangeID := shard[\"range_id\"].(int64)\n\tinfo := convertToShardInfo(currentClusterName, shardInfoRangeID, shard)\n\tdata, _ := result[\"data\"].([]byte)\n\tdataEncoding, _ := result[\"data_encoding\"].(string)\n\tshardRow := &nosqlplugin.ShardRow{\n\t\tInternalShardInfo: info,\n\t\tData:              data,\n\t\tDataEncoding:      dataEncoding,\n\t}\n\treturn rangeID, shardRow, nil\n}\n\nfunc convertToShardInfo(\n\tcurrentCluster string,\n\trangeID int64,\n\tshard map[string]interface{},\n) *persistence.InternalShardInfo {\n\n\tvar pendingFailoverMarkersRawData []byte\n\tvar pendingFailoverMarkersEncoding string\n\tvar transferProcessingQueueStatesRawData []byte\n\tvar transferProcessingQueueStatesEncoding string\n\tvar timerProcessingQueueStatesRawData []byte\n\tvar timerProcessingQueueStatesEncoding string\n\tinfo := &persistence.InternalShardInfo{}\n\tinfo.RangeID = rangeID\n\tfor k, v := range shard {\n\t\tswitch k {\n\t\tcase \"shard_id\":\n\t\t\tinfo.ShardID = v.(int)\n\t\tcase \"owner\":\n\t\t\tinfo.Owner = v.(string)\n\t\tcase \"stolen_since_renew\":\n\t\t\tinfo.StolenSinceRenew = v.(int)\n\t\tcase \"updated_at\":\n\t\t\tinfo.UpdatedAt = v.(time.Time)\n\t\tcase \"replication_ack_level\":\n\t\t\tinfo.ReplicationAckLevel = v.(int64)\n\t\tcase \"transfer_ack_level\":\n\t\t\tinfo.TransferAckLevel = v.(int64)\n\t\tcase \"timer_ack_level\":\n\t\t\tinfo.TimerAckLevel = v.(time.Time)\n\t\tcase \"cluster_transfer_ack_level\":\n\t\t\tinfo.ClusterTransferAckLevel = v.(map[string]int64)\n\t\tcase \"cluster_timer_ack_level\":\n\t\t\tinfo.ClusterTimerAckLevel = v.(map[string]time.Time)\n\t\tcase \"transfer_processing_queue_states\":\n\t\t\ttransferProcessingQueueStatesRawData = v.([]byte)\n\t\tcase \"transfer_processing_queue_states_encoding\":\n\t\t\ttransferProcessingQueueStatesEncoding = v.(string)\n\t\tcase \"timer_processing_queue_states\":\n\t\t\ttimerProcessingQueueStatesRawData = v.([]byte)\n\t\tcase \"timer_processing_queue_states_encoding\":\n\t\t\ttimerProcessingQueueStatesEncoding = v.(string)\n\t\tcase \"domain_notification_version\":\n\t\t\tinfo.DomainNotificationVersion = v.(int64)\n\t\tcase \"cluster_replication_level\":\n\t\t\tinfo.ClusterReplicationLevel = v.(map[string]int64)\n\t\tcase \"replication_dlq_ack_level\":\n\t\t\tinfo.ReplicationDLQAckLevel = v.(map[string]int64)\n\t\tcase \"pending_failover_markers\":\n\t\t\tpendingFailoverMarkersRawData = v.([]byte)\n\t\tcase \"pending_failover_markers_encoding\":\n\t\t\tpendingFailoverMarkersEncoding = v.(string)\n\t\t}\n\t}\n\n\tif info.ClusterTransferAckLevel == nil {\n\t\tinfo.ClusterTransferAckLevel = map[string]int64{\n\t\t\tcurrentCluster: info.TransferAckLevel,\n\t\t}\n\t}\n\tif info.ClusterTimerAckLevel == nil {\n\t\tinfo.ClusterTimerAckLevel = map[string]time.Time{\n\t\t\tcurrentCluster: info.TimerAckLevel,\n\t\t}\n\t}\n\tif info.ClusterReplicationLevel == nil {\n\t\tinfo.ClusterReplicationLevel = make(map[string]int64)\n\t}\n\tif info.ReplicationDLQAckLevel == nil {\n\t\tinfo.ReplicationDLQAckLevel = make(map[string]int64)\n\t}\n\tinfo.PendingFailoverMarkers = persistence.NewDataBlob(\n\t\tpendingFailoverMarkersRawData,\n\t\tconstants.EncodingType(pendingFailoverMarkersEncoding),\n\t)\n\tinfo.TransferProcessingQueueStates = persistence.NewDataBlob(\n\t\ttransferProcessingQueueStatesRawData,\n\t\tconstants.EncodingType(transferProcessingQueueStatesEncoding),\n\t)\n\tinfo.TimerProcessingQueueStates = persistence.NewDataBlob(\n\t\ttimerProcessingQueueStatesRawData,\n\t\tconstants.EncodingType(timerProcessingQueueStatesEncoding),\n\t)\n\n\treturn info\n}\n\n// UpdateRangeID updates the rangeID, return error is there is any\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) UpdateRangeID(ctx context.Context, shardID int, rangeID int64, previousRangeID int64) error {\n\tquery := db.session.Query(templateUpdateRangeIDQuery,\n\t\trangeID,\n\t\tshardID,\n\t\trowTypeShard,\n\t\trowTypeShardDomainID,\n\t\trowTypeShardWorkflowID,\n\t\trowTypeShardRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeShardTaskID,\n\t\tpreviousRangeID,\n\t).WithContext(ctx)\n\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\treturn convertToConflictedShardRow(previous)\n\t}\n\n\treturn nil\n}\n\n// UpdateShard updates a shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) UpdateShard(ctx context.Context, row *nosqlplugin.ShardRow, previousRangeID int64) error {\n\tcqlNowTimestamp := persistence.UnixNanoToDBTimestamp(row.CurrentTimestamp.UnixNano())\n\tmarkerData, markerEncoding := persistence.FromDataBlob(row.PendingFailoverMarkers)\n\ttransferPQS, transferPQSEncoding := persistence.FromDataBlob(row.TransferProcessingQueueStates)\n\ttimerPQS, timerPQSEncoding := persistence.FromDataBlob(row.TimerProcessingQueueStates)\n\n\tquery := db.session.Query(templateUpdateShardQuery,\n\t\trow.ShardID,\n\t\trow.Owner,\n\t\trow.RangeID,\n\t\trow.StolenSinceRenew,\n\t\tcqlNowTimestamp,\n\t\trow.ReplicationAckLevel,\n\t\trow.TransferAckLevel,\n\t\trow.TimerAckLevel,\n\t\trow.ClusterTransferAckLevel,\n\t\trow.ClusterTimerAckLevel,\n\t\ttransferPQS,\n\t\ttransferPQSEncoding,\n\t\ttimerPQS,\n\t\ttimerPQSEncoding,\n\t\trow.DomainNotificationVersion,\n\t\trow.ClusterReplicationLevel,\n\t\trow.ReplicationDLQAckLevel,\n\t\tmarkerData,\n\t\tmarkerEncoding,\n\t\trow.Data,\n\t\trow.DataEncoding,\n\t\trow.RangeID,\n\t\trow.ShardID,\n\t\trowTypeShard,\n\t\trowTypeShardDomainID,\n\t\trowTypeShardWorkflowID,\n\t\trowTypeShardRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeShardTaskID,\n\t\tpreviousRangeID,\n\t).WithContext(ctx)\n\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\treturn convertToConflictedShardRow(previous)\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/shard_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\ttemplateShardType = `{` +\n\t\t`shard_id: ?, ` +\n\t\t`owner: ?, ` +\n\t\t`range_id: ?, ` +\n\t\t`stolen_since_renew: ?, ` +\n\t\t`updated_at: ?, ` +\n\t\t`replication_ack_level: ?, ` +\n\t\t`transfer_ack_level: ?, ` +\n\t\t`timer_ack_level: ?, ` +\n\t\t`cluster_transfer_ack_level: ?, ` +\n\t\t`cluster_timer_ack_level: ?, ` +\n\t\t`transfer_processing_queue_states: ?, ` +\n\t\t`transfer_processing_queue_states_encoding: ?, ` +\n\t\t`timer_processing_queue_states: ?, ` +\n\t\t`timer_processing_queue_states_encoding: ?, ` +\n\t\t`domain_notification_version: ?, ` +\n\t\t`cluster_replication_level: ?, ` +\n\t\t`replication_dlq_ack_level: ?, ` +\n\t\t`pending_failover_markers: ?, ` +\n\t\t`pending_failover_markers_encoding: ? ` +\n\t\t`}`\n\n\ttemplateCreateShardQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, shard, data, data_encoding, range_id) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ` + templateShardType + `, ?, ?, ?) IF NOT EXISTS`\n\n\ttemplateGetShardQuery = `SELECT shard, data, data_encoding, range_id ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateUpdateShardQuery = `UPDATE executions ` +\n\t\t`SET shard = ` + templateShardType + `, data = ?, data_encoding = ?, range_id = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n\n\ttemplateUpdateRangeIDQuery = `UPDATE executions ` +\n\t\t`SET range_id = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/shard_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/testdata\"\n)\n\nfunc TestInsertShard(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-02T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.ShardRow\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully applied\",\n\t\t\trow:  testdata.NewShardRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (` +\n\t\t\t\t\t`shard_id, type, domain_id, workflow_id, run_id, ` +\n\t\t\t\t\t`visibility_ts, task_id, ` +\n\t\t\t\t\t`shard, data, data_encoding, ` +\n\t\t\t\t\t`range_id` +\n\t\t\t\t\t`) ` +\n\t\t\t\t\t`VALUES(` +\n\t\t\t\t\t`15, 0, 10000000-1000-f000-f000-000000000000, 20000000-1000-f000-f000-000000000000, 30000000-1000-f000-f000-000000000000, ` +\n\t\t\t\t\t`946684800000, -11, ` +\n\t\t\t\t\t`{shard_id: 15, owner: owner, range_id: 1000, stolen_since_renew: 0, updated_at: 1712080800000, replication_ack_level: 2000, transfer_ack_level: 3000, timer_ack_level: 2024-04-02T17:00:00Z, cluster_transfer_ack_level: map[cluster2:4000], cluster_timer_ack_level: map[cluster2:2024-04-02 16:00:00 +0000 UTC], transfer_processing_queue_states: [116 114 97 110 115 102 101 114 113 117 101 117 101], transfer_processing_queue_states_encoding: thriftrw, timer_processing_queue_states: [116 105 109 101 114 113 117 101 117 101], timer_processing_queue_states_encoding: thriftrw, domain_notification_version: 3, cluster_replication_level: map[cluster2:5000], replication_dlq_ack_level: map[cluster2:10], pending_failover_markers: [102 97 105 108 111 118 101 114 109 97 114 107 101 114 115], pending_failover_markers_encoding: thriftrw }, ` +\n\t\t\t\t\t`[115 104 97 114 100 100 97 116 97], thriftrw, ` +\n\t\t\t\t\t`1000` +\n\t\t\t\t\t`) IF NOT EXISTS`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow:  testdata.NewShardRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\tm[\"range_id\"] = int64(1001)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscancas failed\",\n\t\t\trow:  testdata.NewShardRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"mapscancas failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertShard(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"InsertShard() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectShard(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-02T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname          string\n\t\tshardID       int\n\t\tcluster       string\n\t\tqueryMockFn   func(query *gocql.MockQuery)\n\t\twantQueries   []string\n\t\twantRangeID   int64\n\t\twantShardInfo *nosqlplugin.ShardRow\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname:    \"success with shard map fully populated\",\n\t\t\tshardID: 15,\n\t\t\tcluster: \"cluster1\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"range_id\"] = int64(1000)\n\t\t\t\t\tm[\"shard\"] = testdata.NewShardMap(ts)\n\t\t\t\t\tm[\"data\"] = []byte(\"sharddata\")\n\t\t\t\t\tm[\"data_encoding\"] = \"thriftrw\"\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRangeID: int64(1000),\n\t\t\twantShardInfo: &nosqlplugin.ShardRow{\n\t\t\t\tInternalShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:                       15,\n\t\t\t\t\tOwner:                         \"owner\",\n\t\t\t\t\tRangeID:                       1000,\n\t\t\t\t\tUpdatedAt:                     ts,\n\t\t\t\t\tReplicationAckLevel:           2000,\n\t\t\t\t\tReplicationDLQAckLevel:        map[string]int64{\"cluster2\": 5},\n\t\t\t\t\tTransferAckLevel:              3000,\n\t\t\t\t\tTimerAckLevel:                 ts.Add(-1 * time.Hour),\n\t\t\t\t\tClusterTransferAckLevel:       map[string]int64{\"cluster1\": 3000},\n\t\t\t\t\tClusterTimerAckLevel:          map[string]time.Time{\"cluster1\": ts.Add(-1 * time.Hour)},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"transferqueue\")},\n\t\t\t\t\tTimerProcessingQueueStates:    &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"timerqueue\")},\n\t\t\t\t\tClusterReplicationLevel:       map[string]int64{\"cluster2\": 1500},\n\t\t\t\t\tDomainNotificationVersion:     3,\n\t\t\t\t\tPendingFailoverMarkers:        &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"failovermarkers\")},\n\t\t\t\t},\n\t\t\t\tData:         []byte(\"sharddata\"),\n\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT shard, data, data_encoding, range_id FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 15 and type = 0 and ` +\n\t\t\t\t\t`domain_id = 10000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`workflow_id = 20000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`run_id = 30000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and ` +\n\t\t\t\t\t`task_id = -11`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"success with shard map missing some fields\",\n\t\t\tshardID: 15,\n\t\t\tcluster: \"cluster1\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"range_id\"] = int64(1000)\n\t\t\t\t\tsm := testdata.NewShardMap(ts)\n\t\t\t\t\t// delete some fields and validate they are initialized properly\n\t\t\t\t\tdelete(sm, \"cluster_transfer_ack_level\")\n\t\t\t\t\tdelete(sm, \"cluster_timer_ack_level\")\n\t\t\t\t\tdelete(sm, \"cluster_replication_level\")\n\t\t\t\t\tdelete(sm, \"replication_dlq_ack_level\")\n\t\t\t\t\tm[\"shard\"] = sm\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRangeID: int64(1000),\n\t\t\twantShardInfo: &nosqlplugin.ShardRow{\n\t\t\t\tInternalShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:                       15,\n\t\t\t\t\tOwner:                         \"owner\",\n\t\t\t\t\tRangeID:                       1000,\n\t\t\t\t\tUpdatedAt:                     ts,\n\t\t\t\t\tReplicationAckLevel:           2000,\n\t\t\t\t\tReplicationDLQAckLevel:        map[string]int64{}, // this was reset to empty map\n\t\t\t\t\tTransferAckLevel:              3000,\n\t\t\t\t\tTimerAckLevel:                 ts.Add(-1 * time.Hour),\n\t\t\t\t\tClusterTransferAckLevel:       map[string]int64{\"cluster1\": 3000},\n\t\t\t\t\tClusterTimerAckLevel:          map[string]time.Time{\"cluster1\": ts.Add(-1 * time.Hour)},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"transferqueue\")},\n\t\t\t\t\tTimerProcessingQueueStates:    &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"timerqueue\")},\n\t\t\t\t\tClusterReplicationLevel:       map[string]int64{}, // this was reset to empty map\n\t\t\t\t\tDomainNotificationVersion:     3,\n\t\t\t\t\tPendingFailoverMarkers:        &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"failovermarkers\")},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT shard, data, data_encoding, range_id FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 15 and type = 0 and ` +\n\t\t\t\t\t`domain_id = 10000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`workflow_id = 20000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`run_id = 30000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and ` +\n\t\t\t\t\t`task_id = -11`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\treturn errors.New(\"mapscan failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotRangeID, gotShardInfo, err := db.SelectShard(context.Background(), tc.shardID, tc.cluster)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectShard() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotRangeID != tc.wantRangeID {\n\t\t\t\tt.Errorf(\"Got RangeID = %v, want %v\", gotRangeID, tc.wantRangeID)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantShardInfo, gotShardInfo); diff != \"\" {\n\t\t\t\tt.Fatalf(\"ShardInfo mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateRangeID(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\trangeID     int64\n\t\tprevRangeID int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"successfully applied\",\n\t\t\tshardID:     15,\n\t\t\trangeID:     1000,\n\t\t\tprevRangeID: 999,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET range_id = 1000 WHERE ` +\n\t\t\t\t\t`shard_id = 15 and ` +\n\t\t\t\t\t`type = 0 and ` +\n\t\t\t\t\t`domain_id = 10000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`workflow_id = 20000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`run_id = 30000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and ` +\n\t\t\t\t\t`task_id = -11 ` +\n\t\t\t\t\t`IF range_id = 999`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"not applied\",\n\t\t\tshardID:     15,\n\t\t\trangeID:     1000,\n\t\t\tprevRangeID: 999,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\tm[\"range_id\"] = int64(1001)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"mapscancas failed\",\n\t\t\tshardID:     15,\n\t\t\trangeID:     1000,\n\t\t\tprevRangeID: 999,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"mapscancas failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateRangeID(context.Background(), tc.shardID, tc.rangeID, tc.prevRangeID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"UpdateRangeID() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateShard(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-02T18:00:00Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.ShardRow\n\t\tprevRangeID int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"successfully applied\",\n\t\t\trow:         testdata.NewShardRow(ts),\n\t\t\tprevRangeID: 988,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET shard = {` +\n\t\t\t\t\t`shard_id: 15, ` +\n\t\t\t\t\t`owner: owner, ` +\n\t\t\t\t\t`range_id: 1000, ` +\n\t\t\t\t\t`stolen_since_renew: 0, ` +\n\t\t\t\t\t`updated_at: 1712080800000, ` +\n\t\t\t\t\t`replication_ack_level: 2000, ` +\n\t\t\t\t\t`transfer_ack_level: 3000, ` +\n\t\t\t\t\t`timer_ack_level: 2024-04-02T17:00:00Z, ` +\n\t\t\t\t\t`cluster_transfer_ack_level: map[cluster2:4000], ` +\n\t\t\t\t\t`cluster_timer_ack_level: map[cluster2:2024-04-02 16:00:00 +0000 UTC], ` +\n\t\t\t\t\t`transfer_processing_queue_states: [116 114 97 110 115 102 101 114 113 117 101 117 101], ` +\n\t\t\t\t\t`transfer_processing_queue_states_encoding: thriftrw, ` +\n\t\t\t\t\t`timer_processing_queue_states: [116 105 109 101 114 113 117 101 117 101], ` +\n\t\t\t\t\t`timer_processing_queue_states_encoding: thriftrw, ` +\n\t\t\t\t\t`domain_notification_version: 3, ` +\n\t\t\t\t\t`cluster_replication_level: map[cluster2:5000], ` +\n\t\t\t\t\t`replication_dlq_ack_level: map[cluster2:10], ` +\n\t\t\t\t\t`pending_failover_markers: [102 97 105 108 111 118 101 114 109 97 114 107 101 114 115], ` +\n\t\t\t\t\t`pending_failover_markers_encoding: thriftrw ` +\n\t\t\t\t\t`}, ` +\n\t\t\t\t\t`data = [115 104 97 114 100 100 97 116 97], ` +\n\t\t\t\t\t`data_encoding = thriftrw, ` +\n\t\t\t\t\t`range_id = 1000 ` +\n\t\t\t\t\t`WHERE ` +\n\t\t\t\t\t`shard_id = 15 and ` +\n\t\t\t\t\t`type = 0 and ` +\n\t\t\t\t\t`domain_id = 10000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`workflow_id = 20000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`run_id = 30000000-1000-f000-f000-000000000000 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and ` +\n\t\t\t\t\t`task_id = -11 ` +\n\t\t\t\t\t`IF range_id = 988`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow:  testdata.NewShardRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\tm[\"range_id\"] = int64(1001)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscancas failed\",\n\t\t\trow:  testdata.NewShardRow(ts),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(m map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"mapscancas failed\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateShard(context.Background(), tc.row, tc.prevRangeID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"UpdateShard() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tt.Log(session.queries[0])\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/tasks.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// Row types for table tasks\n\trowTypeTask = iota\n\trowTypeTaskList\n)\n\nconst (\n\ttaskListTaskID = -12345\n\tinitialRangeID = 1 // Id of the first range of a new task list\n)\n\n// SelectTaskList returns a single tasklist row.\n// Return IsNotFoundError if the row doesn't exist\nfunc (db *CDB) SelectTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter) (*nosqlplugin.TaskListRow, error) {\n\tquery := db.session.Query(templateGetTaskList,\n\t\tfilter.DomainID,\n\t\tfilter.TaskListName,\n\t\tfilter.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t).WithContext(ctx)\n\tvar rangeID int64\n\tvar tlDB map[string]interface{}\n\terr := query.Scan(&rangeID, &tlDB)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tackLevel := tlDB[\"ack_level\"].(int64)\n\ttaskListKind := tlDB[\"kind\"].(int)\n\tlastUpdatedTime := tlDB[\"last_updated\"].(time.Time)\n\n\treturn &nosqlplugin.TaskListRow{\n\t\tDomainID:     filter.DomainID,\n\t\tTaskListName: filter.TaskListName,\n\t\tTaskListType: filter.TaskListType,\n\n\t\tTaskListKind:            taskListKind,\n\t\tLastUpdatedTime:         lastUpdatedTime,\n\t\tAckLevel:                ackLevel,\n\t\tRangeID:                 rangeID,\n\t\tAdaptivePartitionConfig: toTaskListPartitionConfig(tlDB[\"adaptive_partition_config\"]),\n\t}, nil\n}\n\nfunc toTaskListPartitionConfig(v interface{}) *persistence.TaskListPartitionConfig {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tpartition := v.(map[string]interface{})\n\tif len(partition) == 0 {\n\t\treturn nil\n\t}\n\tversion := partition[\"version\"].(int64)\n\tnumRead := partition[\"num_read_partitions\"].(int)\n\tnumWrite := partition[\"num_write_partitions\"].(int)\n\treadPartitions := toTaskListPartitions(partition[\"read_partitions\"])\n\twritePartitions := toTaskListPartitions(partition[\"write_partitions\"])\n\t// If they're out of sync, go with the value of num_*_partitions. This is necessary only while support for\n\t// read_partitions and write_partitions rolls out\n\tif numRead != len(readPartitions) {\n\t\treadPartitions = createDefaultPartitions(numRead)\n\t}\n\tif numWrite != len(writePartitions) {\n\t\twritePartitions = createDefaultPartitions(numWrite)\n\t}\n\treturn &persistence.TaskListPartitionConfig{\n\t\tVersion:         version,\n\t\tReadPartitions:  readPartitions,\n\t\tWritePartitions: writePartitions,\n\t}\n}\n\nfunc createDefaultPartitions(num int) map[int]*persistence.TaskListPartition {\n\tpartitions := make(map[int]*persistence.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tpartitions[i] = &persistence.TaskListPartition{}\n\t}\n\treturn partitions\n}\n\nfunc toTaskListPartitions(values any) map[int]*persistence.TaskListPartition {\n\tif values == nil {\n\t\treturn nil\n\t}\n\tpartitions, ok := values.(map[int]map[string]any)\n\tif !ok || len(partitions) == 0 {\n\t\treturn nil\n\t}\n\tresult := make(map[int]*persistence.TaskListPartition, len(partitions))\n\tfor id, p := range partitions {\n\t\tpartition := toTaskListPartition(p)\n\t\tif partition != nil {\n\t\t\tresult[id] = partition\n\t\t}\n\t}\n\treturn result\n}\n\nfunc toTaskListPartition(partition map[string]any) *persistence.TaskListPartition {\n\tif len(partition) == 0 {\n\t\treturn nil\n\t}\n\tisolationGroups := partition[\"isolation_groups\"].([]string)\n\treturn &persistence.TaskListPartition{\n\t\tIsolationGroups: isolationGroups,\n\t}\n}\n\nfunc fromTaskListPartitionConfig(config *persistence.TaskListPartitionConfig) map[string]interface{} {\n\tif config == nil {\n\t\treturn nil\n\t}\n\treturn map[string]interface{}{\n\t\t\"version\":              config.Version,\n\t\t\"num_read_partitions\":  len(config.ReadPartitions),\n\t\t\"num_write_partitions\": len(config.WritePartitions),\n\t\t\"read_partitions\":      fromTaskListPartitions(config.ReadPartitions),\n\t\t\"write_partitions\":     fromTaskListPartitions(config.WritePartitions),\n\t}\n}\n\nfunc fromTaskListPartitions(partitions map[int]*persistence.TaskListPartition) map[int]any {\n\tif len(partitions) == 0 {\n\t\treturn nil\n\t}\n\tresult := make(map[int]any, len(partitions))\n\tfor id, partition := range partitions {\n\t\tresult[id] = fromTaskListPartition(partition)\n\t}\n\treturn result\n}\n\nfunc fromTaskListPartition(partition *persistence.TaskListPartition) any {\n\tif partition == nil {\n\t\treturn nil\n\t}\n\treturn map[string]any{\n\t\t\"isolation_groups\": partition.IsolationGroups,\n\t}\n}\n\n// InsertTaskList insert a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) InsertTaskList(ctx context.Context, row *nosqlplugin.TaskListRow) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tquery := db.session.Query(templateInsertTaskListQuery,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\tinitialRangeID,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\t0,\n\t\trow.TaskListKind,\n\t\trow.LastUpdatedTime,\n\t\tfromTaskListPartitionConfig(row.AdaptivePartitionConfig),\n\t\ttimeStamp,\n\t).WithContext(ctx)\n\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn handleTaskListAppliedError(applied, previous)\n}\n\n// UpdateTaskList updates a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) UpdateTaskList(\n\tctx context.Context,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tquery := db.session.Query(templateUpdateTaskListQuery,\n\t\trow.RangeID,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trow.AckLevel,\n\t\trow.TaskListKind,\n\t\trow.LastUpdatedTime,\n\t\tfromTaskListPartitionConfig(row.AdaptivePartitionConfig),\n\t\ttimeStamp,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\tpreviousRangeID,\n\t).WithContext(ctx)\n\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn handleTaskListAppliedError(applied, previous)\n}\n\nfunc handleTaskListAppliedError(applied bool, previous map[string]interface{}) error {\n\tif !applied {\n\t\t// NOTE: Cassandra only returns the conflicted columns in this results\n\t\trangeID := previous[\"range_id\"].(int64)\n\t\tvar columns []string\n\t\tfor k, v := range previous {\n\t\t\tcolumns = append(columns, fmt.Sprintf(\"%s=%v\", k, v))\n\t\t}\n\t\treturn &nosqlplugin.TaskOperationConditionFailure{\n\t\t\tRangeID: rangeID,\n\t\t\tDetails: strings.Join(columns, \",\"),\n\t\t}\n\t}\n\treturn nil\n}\n\n// UpdateTaskList updates a single tasklist row, and set an TTL on the record\n// Return TaskOperationConditionFailure if the condition doesn't meet\n// Ignore TTL if it's not supported, which becomes exactly the same as UpdateTaskList, but ListTaskList must be\n// implemented for TaskListScavenger\nfunc (db *CDB) UpdateTaskListWithTTL(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\ttimeStamp := row.CurrentTimeStamp\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\t// part 1 is used to set TTL on primary key as UPDATE can't set TTL for primary key\n\tbatch.Query(templateUpdateTaskListQueryWithTTLPart1,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\ttimeStamp,\n\t\tttlSeconds,\n\t)\n\t// part 2 is for CAS and setting TTL for the rest of the columns\n\tbatch.Query(templateUpdateTaskListQueryWithTTLPart2,\n\t\tttlSeconds,\n\t\trow.RangeID,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trow.AckLevel,\n\t\trow.TaskListKind,\n\t\ttimeStamp,\n\t\tfromTaskListPartitionConfig(row.AdaptivePartitionConfig),\n\t\ttimeStamp,\n\t\trow.DomainID,\n\t\trow.TaskListName,\n\t\trow.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\tpreviousRangeID,\n\t)\n\tprevious := make(map[string]interface{})\n\tapplied, _, err := db.session.MapExecuteBatchCAS(batch, previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn handleTaskListAppliedError(applied, previous)\n}\n\n// ListTaskList returns all tasklists.\n// Noop if TTL is already implemented in other methods\nfunc (db *CDB) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*nosqlplugin.ListTaskListResult, error) {\n\treturn nil, &types.InternalServiceError{\n\t\tMessage: \"unsupported operation\",\n\t}\n}\n\n// DeleteTaskList deletes a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *CDB) DeleteTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter, previousRangeID int64) error {\n\tquery := db.session.Query(templateDeleteTaskListQuery,\n\t\tfilter.DomainID,\n\t\tfilter.TaskListName,\n\t\tfilter.TaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\tpreviousRangeID,\n\t).WithContext(ctx).Consistency(cassandraAllConslevel)\n\tprevious := make(map[string]interface{})\n\tapplied, err := query.MapScanCAS(previous)\n\tif err != nil {\n\t\tif !db.isCassandraConsistencyError(err) {\n\t\t\treturn err\n\t\t}\n\t\tdb.logger.Warn(\"unable to complete the delete operation due to consistency issue\", tag.Error(err))\n\t\tapplied, err = query.Consistency(cassandraDefaultConsLevel).MapScanCAS(previous)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn handleTaskListAppliedError(applied, previous)\n}\n\n// InsertTasks inserts a batch of tasks\n// Return IsConditionFailedError if the condition doesn't meet, and also the previous tasklist row\nfunc (db *CDB) InsertTasks(\n\tctx context.Context,\n\ttasksToInsert []*nosqlplugin.TaskRowForInsert,\n\ttasklistCondition *nosqlplugin.TaskListRow,\n) error {\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\tdomainID := tasklistCondition.DomainID\n\ttaskListName := tasklistCondition.TaskListName\n\ttaskListType := tasklistCondition.TaskListType\n\ttimeStamp := tasklistCondition.CurrentTimeStamp\n\n\tfor _, task := range tasksToInsert {\n\t\tscheduleID := task.ScheduledID\n\t\tttl := int64(task.TTLSeconds)\n\t\tif ttl <= 0 {\n\t\t\tbatch.Query(templateCreateTaskQuery,\n\t\t\t\tdomainID,\n\t\t\t\ttaskListName,\n\t\t\t\ttaskListType,\n\t\t\t\trowTypeTask,\n\t\t\t\ttask.TaskID,\n\t\t\t\tdomainID,\n\t\t\t\ttask.WorkflowID,\n\t\t\t\ttask.RunID,\n\t\t\t\tscheduleID,\n\t\t\t\ttask.CreatedTime,\n\t\t\t\ttask.PartitionConfig,\n\t\t\t\ttimeStamp,\n\t\t\t)\n\t\t} else {\n\t\t\tif ttl > maxCassandraTTL {\n\t\t\t\tttl = maxCassandraTTL\n\t\t\t}\n\t\t\tbatch.Query(templateCreateTaskWithTTLQuery,\n\t\t\t\tdomainID,\n\t\t\t\ttaskListName,\n\t\t\t\ttaskListType,\n\t\t\t\trowTypeTask,\n\t\t\t\ttask.TaskID,\n\t\t\t\tdomainID,\n\t\t\t\ttask.WorkflowID,\n\t\t\t\ttask.RunID,\n\t\t\t\tscheduleID,\n\t\t\t\ttask.CreatedTime,\n\t\t\t\ttask.PartitionConfig,\n\t\t\t\ttimeStamp,\n\t\t\t\tttl)\n\t\t}\n\t}\n\n\t// The following query is used to ensure that range_id didn't change\n\tbatch.Query(templateUpdateTaskListRangeIDQuery,\n\t\ttasklistCondition.RangeID,\n\t\ttimeStamp,\n\t\tdomainID,\n\t\ttaskListName,\n\t\ttaskListType,\n\t\trowTypeTaskList,\n\t\ttaskListTaskID,\n\t\ttasklistCondition.RangeID,\n\t)\n\n\tprevious := make(map[string]interface{})\n\tapplied, _, err := db.session.MapExecuteBatchCAS(batch, previous)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn handleTaskListAppliedError(applied, previous)\n}\n\n// GetTasksCount returns number of tasks from a tasklist\nfunc (db *CDB) GetTasksCount(ctx context.Context, filter *nosqlplugin.TasksFilter) (int64, error) {\n\tquery := db.session.Query(templateGetTasksCountQuery,\n\t\tfilter.DomainID,\n\t\tfilter.TaskListName,\n\t\tfilter.TaskListType,\n\t\trowTypeTask,\n\t\tfilter.MinTaskID,\n\t).WithContext(ctx)\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\treturn 0, err\n\t}\n\n\tqueueSize := result[\"count\"].(int64)\n\treturn queueSize, nil\n}\n\n// SelectTasks return tasks that associated to a tasklist\nfunc (db *CDB) SelectTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) ([]*nosqlplugin.TaskRow, error) {\n\t// Reading tasklist tasks need to be quorum level consistent, otherwise we could lose tasks\n\tquery := db.session.Query(templateGetTasksQuery,\n\t\tfilter.DomainID,\n\t\tfilter.TaskListName,\n\t\tfilter.TaskListType,\n\t\trowTypeTask,\n\t\tfilter.MinTaskID,\n\t\tfilter.MaxTaskID,\n\t).PageSize(filter.BatchSize).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, fmt.Errorf(\"selectTasks operation failed.  Not able to create query iterator\")\n\t}\n\n\tvar response []*nosqlplugin.TaskRow\n\ttask := make(map[string]interface{})\nPopulateTasks:\n\tfor iter.MapScan(task) {\n\t\ttaskID, ok := task[\"task_id\"]\n\t\tif !ok { // no tasks, but static column record returned\n\t\t\tcontinue\n\t\t}\n\n\t\t// TODO: no usage of ttl\n\t\t// Extract the TTL value\n\t\tttlValue, ttlExists := task[\"ttl\"]\n\n\t\t// Check if TTL is null or an integer\n\t\tvar ttl *int\n\t\tif ttlExists && ttlValue != nil {\n\t\t\tif ttlInt, ok := ttlValue.(int); ok {\n\t\t\t\tttl = &ttlInt // TTL is an integer\n\t\t\t}\n\t\t}\n\n\t\tt := createTaskInfo(task[\"task\"].(map[string]interface{}))\n\t\tt.TaskID = taskID.(int64)\n\n\t\t// TODO: any computations on such level is not recommended, move to higher level\n\t\tif ttl != nil {\n\t\t\tt.Expiry = time.Now().Add(time.Duration(*ttl) * time.Second)\n\t\t}\n\n\t\tresponse = append(response, t)\n\t\tif len(response) == filter.BatchSize {\n\t\t\tbreak PopulateTasks\n\t\t}\n\t\ttask = make(map[string]interface{}) // Reinitialize map as initialized fails on unmarshalling\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc createTaskInfo(\n\tresult map[string]interface{},\n) *nosqlplugin.TaskRow {\n\n\tinfo := &nosqlplugin.TaskRow{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\tcase \"workflow_id\":\n\t\t\tinfo.WorkflowID = v.(string)\n\t\tcase \"run_id\":\n\t\t\tinfo.RunID = v.(gocql.UUID).String()\n\t\tcase \"schedule_id\":\n\t\t\tinfo.ScheduledID = v.(int64)\n\t\tcase \"created_time\":\n\t\t\tinfo.CreatedTime = v.(time.Time)\n\t\tcase \"partition_config\":\n\t\t\tinfo.PartitionConfig = v.(map[string]string)\n\t\t}\n\t}\n\n\treturn info\n}\n\n// DeleteTask delete a batch tasks that taskIDs less than the row\n// If TTL is not implemented, then should also return the number of rows deleted, otherwise persistence.UnknownNumRowsAffected\n// NOTE: This API ignores the `BatchSize` request parameter i.e. either all tasks leq the task_id will be deleted or an error will\n// be returned to the caller, because rowsDeleted is not supported by Cassandra\nfunc (db *CDB) RangeDeleteTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) (rowsDeleted int, err error) {\n\tquery := db.session.Query(templateCompleteTasksLessThanQuery,\n\t\tfilter.DomainID,\n\t\tfilter.TaskListName,\n\t\tfilter.TaskListType,\n\t\trowTypeTask,\n\t\tfilter.MinTaskID,\n\t\tfilter.MaxTaskID,\n\t).WithContext(ctx)\n\treturn persistence.UnknownNumRowsAffected, db.executeWithConsistencyAll(query)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/tasks_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\ttemplateTaskListType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`name: ?, ` +\n\t\t`type: ?, ` +\n\t\t`ack_level: ?, ` +\n\t\t`kind: ?, ` +\n\t\t`last_updated: ?, ` +\n\t\t`adaptive_partition_config: ? ` +\n\t\t`}`\n\n\ttemplateTaskType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`workflow_id: ?, ` +\n\t\t`run_id: ?, ` +\n\t\t`schedule_id: ?,` +\n\t\t`created_time: ?, ` +\n\t\t`partition_config: ? ` +\n\t\t`}`\n\n\ttemplateCreateTaskQuery = `INSERT INTO tasks (` +\n\t\t`domain_id, task_list_name, task_list_type, type, task_id, task, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateTaskType + `, ?)`\n\n\ttemplateCreateTaskWithTTLQuery = `INSERT INTO tasks (` +\n\t\t`domain_id, task_list_name, task_list_type, type, task_id, task, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateTaskType + `, ?) USING TTL ?`\n\n\ttemplateGetTasksQuery = `SELECT task_id, task, TTL(task) AS ttl ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id > ? ` +\n\t\t`and task_id <= ?`\n\n\ttemplateGetTasksCountQuery = `SELECT count(1) as count ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id > ? `\n\n\ttemplateCompleteTasksLessThanQuery = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND task_list_name = ? ` +\n\t\t`AND task_list_type = ? ` +\n\t\t`AND type = ? ` +\n\t\t`AND task_id > ? ` +\n\t\t`AND task_id <= ? `\n\n\ttemplateGetTaskList = `SELECT ` +\n\t\t`range_id, ` +\n\t\t`task_list ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateInsertTaskListQuery = `INSERT INTO tasks (` +\n\t\t`domain_id, ` +\n\t\t`task_list_name, ` +\n\t\t`task_list_type, ` +\n\t\t`type, ` +\n\t\t`task_id, ` +\n\t\t`range_id, ` +\n\t\t`task_list, ` +\n\t\t`created_time ` +\n\t\t`) VALUES (?, ?, ?, ?, ?, ?, ` + templateTaskListType + `, ?) IF NOT EXISTS`\n\n\ttemplateUpdateTaskListQuery = `UPDATE tasks SET ` +\n\t\t`range_id = ?, ` +\n\t\t`task_list = ` + templateTaskListType + \" , last_updated_time = ? \" +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n\n\ttemplateUpdateTaskListQueryWithTTLPart1 = ` INSERT INTO tasks (` +\n\t\t`domain_id, ` +\n\t\t`task_list_name, ` +\n\t\t`task_list_type, ` +\n\t\t`type, ` +\n\t\t`task_id, ` +\n\t\t`created_time ` +\n\t\t`) VALUES (?, ?, ?, ?, ?, ?) USING TTL ?`\n\n\ttemplateUpdateTaskListQueryWithTTLPart2 = `UPDATE tasks USING TTL ? SET ` +\n\t\t`range_id = ?, ` +\n\t\t`task_list = ` + templateTaskListType + \" , last_updated_time = ? \" +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n\n\ttemplateUpdateTaskListRangeIDQuery = `UPDATE tasks SET ` +\n\t\t`range_id = ?, ` +\n\t\t`last_updated_time = ? ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and task_list_name = ? ` +\n\t\t`and task_list_type = ? ` +\n\t\t`and type = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n\n\ttemplateDeleteTaskListQuery = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND task_list_name = ? ` +\n\t\t`AND task_list_type = ? ` +\n\t\t`AND type = ? ` +\n\t\t`AND task_id = ? ` +\n\t\t`IF range_id = ?`\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/tasks_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestSelectTaskList(t *testing.T) {\n\tnow := time.Now()\n\ttests := []struct {\n\t\tname        string\n\t\tfilter      *nosqlplugin.TaskListFilter\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantRow     *nosqlplugin.TaskListRow\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\trangeID := args[0].(*int64)\n\t\t\t\t\t*rangeID = 25\n\t\t\t\t\ttlDB := args[1].(*map[string]interface{})\n\t\t\t\t\t*tlDB = make(map[string]interface{})\n\t\t\t\t\t(*tlDB)[\"ack_level\"] = int64(1000)\n\t\t\t\t\t(*tlDB)[\"kind\"] = 2\n\t\t\t\t\t(*tlDB)[\"last_updated\"] = now\n\t\t\t\t\t(*tlDB)[\"adaptive_partition_config\"] = map[string]interface{}{\n\t\t\t\t\t\t\"version\":              int64(0),\n\t\t\t\t\t\t\"num_read_partitions\":  int(1),\n\t\t\t\t\t\t\"num_write_partitions\": int(1),\n\t\t\t\t\t\t\"read_partitions\": map[int]map[string]any{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\t\"isolation_groups\": []string{\"foo\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"write_partitions\": map[int]map[string]any{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\t\"isolation_groups\": []string{\"bar\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:        \"domain1\",\n\t\t\t\tTaskListName:    \"tasklist1\",\n\t\t\t\tTaskListType:    1,\n\t\t\t\tTaskListKind:    2,\n\t\t\t\tAckLevel:        1000,\n\t\t\t\tRangeID:         25,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 0,\n\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT range_id, task_list FROM tasks WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 1 and task_id = -12345`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - partition numbers only\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\trangeID := args[0].(*int64)\n\t\t\t\t\t*rangeID = 25\n\t\t\t\t\ttlDB := args[1].(*map[string]interface{})\n\t\t\t\t\t*tlDB = make(map[string]interface{})\n\t\t\t\t\t(*tlDB)[\"ack_level\"] = int64(1000)\n\t\t\t\t\t(*tlDB)[\"kind\"] = 2\n\t\t\t\t\t(*tlDB)[\"last_updated\"] = now\n\t\t\t\t\t(*tlDB)[\"adaptive_partition_config\"] = map[string]interface{}{\n\t\t\t\t\t\t\"version\":              int64(0),\n\t\t\t\t\t\t\"num_read_partitions\":  int(1),\n\t\t\t\t\t\t\"num_write_partitions\": int(1),\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantRow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:        \"domain1\",\n\t\t\t\tTaskListName:    \"tasklist1\",\n\t\t\t\tTaskListType:    1,\n\t\t\t\tTaskListKind:    2,\n\t\t\t\tAckLevel:        1000,\n\t\t\t\tRangeID:         25,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 0,\n\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT range_id, task_list FROM tasks WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 1 and task_id = -12345`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scan failure\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Scan(gomock.Any()).DoAndReturn(func(args ...interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotRow, err := db.SelectTaskList(context.Background(), tc.filter)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectTaskList() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantRow, gotRow); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Row mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertTaskList(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.TaskListRow\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully applied - nil partition_config\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO tasks (domain_id, task_list_name, task_list_type, type, task_id, range_id, task_list, created_time ) ` +\n\t\t\t\t\t`VALUES (domain1, tasklist1, 1, 1, -12345, 1, ` +\n\t\t\t\t\t`{domain_id: domain1, name: tasklist1, type: 1, ack_level: 0, kind: 2, last_updated: 2024-04-01T22:08:41Z, adaptive_partition_config: map[] }` +\n\t\t\t\t\t`, 2024-04-01T22:08:41Z) IF NOT EXISTS`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"successfully applied - non-nil partition_config\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO tasks (domain_id, task_list_name, task_list_type, type, task_id, range_id, task_list, created_time ) ` +\n\t\t\t\t\t`VALUES (domain1, tasklist1, 1, 1, -12345, 1, ` +\n\t\t\t\t\t`{domain_id: domain1, name: tasklist1, type: 1, ack_level: 0, kind: 2, last_updated: 2024-04-01T22:08:41Z, ` +\n\t\t\t\t\t`adaptive_partition_config: map[num_read_partitions:1 num_write_partitions:1 read_partitions:map[0:map[isolation_groups:[]]] version:1 write_partitions:map[0:map[isolation_groups:[]]]] }` +\n\t\t\t\t\t`, 2024-04-01T22:08:41Z) IF NOT EXISTS`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\tprev[\"range_id\"] = int64(26)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertTaskList(context.Background(), tc.row)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"InsertTaskList() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTaskList(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\trow         *nosqlplugin.TaskListRow\n\t\tprevRangeID int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"successfully applied\",\n\t\t\tprevRangeID: 25,\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE tasks SET range_id = 25, task_list = {domain_id: domain1, name: tasklist1, type: 1, ack_level: 1000, kind: 2, last_updated: 2024-04-01T22:08:41Z, adaptive_partition_config: map[] } , last_updated_time = 2024-04-01T22:08:41Z WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 1 and task_id = -12345 IF range_id = 25`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\tprev[\"range_id\"] = int64(26)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateTaskList(context.Background(), tc.row, tc.prevRangeID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"UpdateTaskList() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTaskListWithTTL(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname                      string\n\t\trow                       *nosqlplugin.TaskListRow\n\t\tttlSeconds                int64\n\t\tprevRangeID               int64\n\t\tmapExecuteBatchCASApplied bool\n\t\tmapExecuteBatchCASErr     error\n\t\tmapExecuteBatchCASPrev    map[string]any\n\t\twantQueries               []string\n\t\twantErr                   bool\n\t}{\n\t\t{\n\t\t\tname:        \"successfully applied\",\n\t\t\tttlSeconds:  180,\n\t\t\tprevRangeID: 25,\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tTaskListKind:     2,\n\t\t\t\tAckLevel:         1000,\n\t\t\t\tRangeID:          25,\n\t\t\t\tLastUpdatedTime:  ts,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\twantQueries: []string{\n\t\t\t\t` INSERT INTO tasks (domain_id, task_list_name, task_list_type, type, task_id, created_time ) VALUES (domain1, tasklist1, 1, 1, -12345, 2024-04-01T22:08:41Z) USING TTL 180`,\n\t\t\t\t`UPDATE tasks USING TTL 180 SET range_id = 25, task_list = {domain_id: domain1, name: tasklist1, type: 1, ack_level: 1000, kind: 2, last_updated: 2024-04-01T22:08:41Z, adaptive_partition_config: map[] } , last_updated_time = 2024-04-01T22:08:41Z WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 1 and task_id = -12345 IF range_id = 25`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t\tTaskListKind: 2,\n\t\t\t\tAckLevel:     1000,\n\t\t\t\tRangeID:      25,\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\"range_id\": int64(26),\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed\",\n\t\t\trow: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t\tTaskListKind: 2,\n\t\t\t\tAckLevel:     1000,\n\t\t\t\tRangeID:      25,\n\t\t\t},\n\t\t\tmapExecuteBatchCASErr: errors.New(\"cas failure\"),\n\t\t\twantErr:               true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: tc.mapExecuteBatchCASApplied,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t\tmapExecuteBatchCASPrev:    tc.mapExecuteBatchCASPrev,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\t\t\terr := db.UpdateTaskListWithTTL(context.Background(), tc.ttlSeconds, tc.row, tc.prevRangeID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"UpdateTaskListWithTTL() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.batches[0].queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListTaskList(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tclient := gocql.NewMockClient(ctrl)\n\tcfg := &config.NoSQL{}\n\tlogger := testlogger.New(t)\n\tdc := &persistence.DynamicConfiguration{}\n\tsession := &fakeSession{}\n\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t_, err := db.ListTaskList(context.Background(), 10, nil)\n\tvar want *types.InternalServiceError\n\tif !errors.As(err, &want) {\n\t\tt.Fatalf(\"expected InternalServiceError, got %v\", err)\n\t}\n\tif want.Message != \"unsupported operation\" {\n\t\tt.Fatalf(\"Unexpected message: %v\", want.Message)\n\t}\n}\n\nfunc TestDeleteTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tfilter       *nosqlplugin.TaskListFilter\n\t\tprevRangeID  int64\n\t\tqueryMockFn  func(query *gocql.MockQuery)\n\t\tclientMockFn func(client *gocql.MockClient)\n\t\twantQueries  []string\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname:        \"successfully applied\",\n\t\t\tprevRangeID: 35,\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn true, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM tasks WHERE domain_id = domain1 AND task_list_name = tasklist1 AND task_list_type = 1 AND type = 1 AND task_id = -12345 IF range_id = 35`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not applied\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\tprev[\"range_id\"] = int64(26)\n\t\t\t\t\treturn false, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed with all consistency, retry with default consistency\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tfirstConsCall := query.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tfirstMapScanCall := query.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// expect retry with default consistency level\n\t\t\t\tquery.EXPECT().Consistency(cassandraDefaultConsLevel).Return(query).Times(1).After(firstConsCall)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).Return(true, nil).Times(1).After(firstMapScanCall)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM tasks WHERE domain_id = domain1 AND task_list_name = tasklist1 AND task_list_type = 1 AND type = 1 AND task_id = -12345 IF range_id = 0`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed with all consistency, retry with default consistency also failed\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tfirstConsCall := query.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tfirstMapScanCall := query.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"failed to delete\")\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// expect retry with default consistency level\n\t\t\t\tquery.EXPECT().Consistency(cassandraDefaultConsLevel).Return(query).Times(1).After(firstConsCall)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).Return(true, errors.New(\"failed to deleta again\")).Times(1).After(firstMapScanCall)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"mapscan failed with non-consistency error\",\n\t\t\tfilter: &nosqlplugin.TaskListFilter{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Consistency(cassandraAllConslevel).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScanCAS(gomock.Any()).DoAndReturn(func(prev map[string]interface{}) (bool, error) {\n\t\t\t\t\treturn false, errors.New(\"failed to delete\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif tc.clientMockFn != nil {\n\t\t\t\ttc.clientMockFn(client)\n\t\t\t}\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.DeleteTaskList(context.Background(), tc.filter, tc.prevRangeID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteTaskList() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertTasks(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname                      string\n\t\ttasksToInsert             []*nosqlplugin.TaskRowForInsert\n\t\ttasklistCond              *nosqlplugin.TaskListRow\n\t\tmapExecuteBatchCASApplied bool\n\t\tmapExecuteBatchCASErr     error\n\t\tmapExecuteBatchCASPrev    map[string]any\n\t\twantQueries               []string\n\t\twantErr                   bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully applied\",\n\t\t\ttasksToInsert: []*nosqlplugin.TaskRowForInsert{\n\t\t\t\t{\n\t\t\t\t\tTTLSeconds: 0, // default create task query will be used for this ttl\n\t\t\t\t\tTaskRow: nosqlplugin.TaskRow{\n\t\t\t\t\t\tTaskID:      3,\n\t\t\t\t\t\tWorkflowID:  \"wid1\",\n\t\t\t\t\t\tRunID:       \"rid1\",\n\t\t\t\t\t\tScheduledID: 42,\n\t\t\t\t\t\tCreatedTime: ts,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTTLSeconds: int(maxCassandraTTL) + 1, // ttl query will be used for this ttl\n\t\t\t\t\tTaskRow: nosqlplugin.TaskRow{\n\t\t\t\t\t\tTaskID:      4,\n\t\t\t\t\t\tWorkflowID:  \"wid1\",\n\t\t\t\t\t\tRunID:       \"rid1\",\n\t\t\t\t\t\tScheduledID: 43,\n\t\t\t\t\t\tCreatedTime: ts.Add(time.Second),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\ttasklistCond: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:         \"domain1\",\n\t\t\t\tTaskListName:     \"tasklist1\",\n\t\t\t\tTaskListType:     1,\n\t\t\t\tRangeID:          25,\n\t\t\t\tCurrentTimeStamp: ts,\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO tasks (domain_id, task_list_name, task_list_type, type, task_id, task, created_time) VALUES(domain1, tasklist1, 1, 0, 3, {domain_id: domain1, workflow_id: wid1, run_id: rid1, schedule_id: 42,created_time: 2024-04-01T22:08:41Z, partition_config: map[] }, 2024-04-01T22:08:41Z)`,\n\t\t\t\t`INSERT INTO tasks (domain_id, task_list_name, task_list_type, type, task_id, task, created_time) VALUES(domain1, tasklist1, 1, 0, 4, {domain_id: domain1, workflow_id: wid1, run_id: rid1, schedule_id: 43,created_time: 2024-04-01T22:08:42Z, partition_config: map[] }, 2024-04-01T22:08:41Z) USING TTL 157680000`,\n\t\t\t\t`UPDATE tasks SET range_id = 25, last_updated_time = 2024-04-01T22:08:41Z WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 1 and task_id = -12345 IF range_id = 25`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"batch cas failed\",\n\t\t\ttasksToInsert: []*nosqlplugin.TaskRowForInsert{\n\t\t\t\t{\n\t\t\t\t\tTTLSeconds: 0,\n\t\t\t\t\tTaskRow: nosqlplugin.TaskRow{\n\t\t\t\t\t\tTaskID:      3,\n\t\t\t\t\t\tWorkflowID:  \"wid1\",\n\t\t\t\t\t\tRunID:       \"rid1\",\n\t\t\t\t\t\tScheduledID: 42,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\ttasklistCond: &nosqlplugin.TaskListRow{\n\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\tTaskListType: 1,\n\t\t\t\tRangeID:      25,\n\t\t\t},\n\t\t\tmapExecuteBatchCASErr: errors.New(\"some random error\"),\n\t\t\twantErr:               true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: tc.mapExecuteBatchCASApplied,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t\tmapExecuteBatchCASPrev:    tc.mapExecuteBatchCASPrev,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertTasks(context.Background(), tc.tasksToInsert, tc.tasklistCond)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"InsertTasks() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.batches[0].queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetTasksCount(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tfilter        *nosqlplugin.TasksFilter\n\t\tqueryMockFn   func(query *gocql.MockQuery)\n\t\twantQueries   []string\n\t\twantQueueSize int64\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(result map[string]interface{}) error {\n\t\t\t\t\tresult[\"count\"] = int64(50)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantQueueSize: 50,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT count(1) as count FROM tasks WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 0 and task_id > 1 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scan failure\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(result map[string]interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tqueueSize, err := db.GetTasksCount(context.Background(), tc.filter)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"GetTasksCount() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif queueSize != tc.wantQueueSize {\n\t\t\t\tt.Fatalf(\"Got queue size: %d, want: %v\", queueSize, tc.wantQueueSize)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectTasks(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\tfilter      *nosqlplugin.TasksFilter\n\t\titer        *fakeIter\n\t\twantQueries []string\n\t\twantTasks   []*nosqlplugin.TaskRow\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname: \"nil iter\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"iter close failed\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t\tBatchSize: 3,\n\t\t\t},\n\t\t\titer:    &fakeIter{closeErr: errors.New(\"some random error\")},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 0,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t\tBatchSize: 3,\n\t\t\t},\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(1),\n\t\t\t\t\t\t\"task\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":        &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"wofklow_id\":       \"wid1\",\n\t\t\t\t\t\t\t\"schedule_id\":      int64(42),\n\t\t\t\t\t\t\t\"created_time\":     ts,\n\t\t\t\t\t\t\t\"run_id\":           &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\t\"partition_config\": map[string]string{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(2),\n\t\t\t\t\t\t\"task\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":        &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"worklow_id\":       \"wid1\",\n\t\t\t\t\t\t\t\"schedule_id\":      int64(45),\n\t\t\t\t\t\t\t\"created_time\":     ts,\n\t\t\t\t\t\t\t\"run_id\":           &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\t\"partition_config\": map[string]string{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"missing_task_id\": int64(1), // missing task_id so this row will be skipped\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(3),\n\t\t\t\t\t\t\"task\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":        &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"worklow_id\":       \"wid1\",\n\t\t\t\t\t\t\t\"schedule_id\":      int64(48),\n\t\t\t\t\t\t\t\"created_time\":     ts,\n\t\t\t\t\t\t\t\"run_id\":           &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\t\"partition_config\": map[string]string{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(4), // this will be skipped because filter.BatchSize is reached\n\t\t\t\t\t\t\"task\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":        &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"worklow_id\":       \"wid1\",\n\t\t\t\t\t\t\t\"schedule_id\":      int64(51),\n\t\t\t\t\t\t\t\"created_time\":     ts,\n\t\t\t\t\t\t\t\"run_id\":           &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\t\"partition_config\": map[string]string{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantTasks: []*nosqlplugin.TaskRow{\n\t\t\t\t{\n\t\t\t\t\tDomainID:        \"domain1\",\n\t\t\t\t\tTaskID:          1,\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tScheduledID:     42,\n\t\t\t\t\tCreatedTime:     ts,\n\t\t\t\t\tPartitionConfig: map[string]string{},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDomainID:        \"domain1\",\n\t\t\t\t\tTaskID:          2,\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tScheduledID:     45,\n\t\t\t\t\tCreatedTime:     ts,\n\t\t\t\t\tPartitionConfig: map[string]string{},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDomainID:        \"domain1\",\n\t\t\t\t\tTaskID:          3,\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tScheduledID:     48,\n\t\t\t\t\tCreatedTime:     ts,\n\t\t\t\t\tPartitionConfig: map[string]string{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT task_id, task, TTL(task) AS ttl FROM tasks WHERE domain_id = domain1 and task_list_name = tasklist1 and task_list_type = 1 and type = 0 and task_id > 0 and task_id <= 100`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\tgotRows, err := db.SelectTasks(context.Background(), tc.filter)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectTasks() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantTasks, gotRows); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Task rows mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Fatal(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tfilter          *nosqlplugin.TasksFilter\n\t\tqueryMockFn     func(query *gocql.MockQuery)\n\t\twantQueries     []string\n\t\twantRowsDeleted int\n\t\twantErr         bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantRowsDeleted: persistence.UnknownNumRowsAffected,\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM tasks WHERE domain_id = domain1 AND task_list_name = tasklist1 AND task_list_type = 1 AND type = 0 AND task_id > 1 AND task_id <= 100 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failure\",\n\t\t\tfilter: &nosqlplugin.TasksFilter{\n\t\t\t\tTaskListFilter: nosqlplugin.TaskListFilter{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tTaskListName: \"tasklist1\",\n\t\t\t\t\tTaskListType: 1,\n\t\t\t\t},\n\t\t\t\tMinTaskID: 1,\n\t\t\t\tMaxTaskID: 100,\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, nil, DbWithClient(client))\n\n\t\t\trowsDeleted, err := db.RangeDeleteTasks(context.Background(), tc.filter)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteTasks() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, session.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif rowsDeleted != tc.wantRowsDeleted {\n\t\t\t\tt.Fatalf(\"Got rows deleted: %d, want: %v\", rowsDeleted, tc.wantRowsDeleted)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/testdata/domain.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc NewDomainRow(ts time.Time) *nosqlplugin.DomainRow {\n\treturn &nosqlplugin.DomainRow{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          \"test-domain-id\",\n\t\t\tName:        \"test-domain-name\",\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: \"test-domain-description\",\n\t\t\tOwnerEmail:  \"test-domain-owner-email\",\n\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t},\n\t\tConfig: &persistence.InternalDomainConfig{\n\t\t\tRetention:                7 * 24 * time.Hour,\n\t\t\tEmitMetric:               true,\n\t\t\tArchivalBucket:           \"test-archival-bucket\",\n\t\t\tArchivalStatus:           types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalURI:       \"test-history-archival-uri\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"test-visibility-archival-uri\",\n\t\t\tBadBinaries:              &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"bad-binaries\")},\n\t\t\tIsolationGroups:          &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"isolation-group\")},\n\t\t\tAsyncWorkflowsConfig:     &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"async-workflows-config\")},\n\t\t},\n\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\tActiveClusterName: \"test-active-cluster-name\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{\n\t\t\t\t\tClusterName: \"test-cluster-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tActiveClustersConfig: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"active-clusters-config\")},\n\t\t},\n\t\tIsGlobalDomain:      true,\n\t\tConfigVersion:       3,\n\t\tFailoverVersion:     4,\n\t\tFailoverEndTime:     &ts,\n\t\tLastUpdatedTime:     ts,\n\t\tNotificationVersion: 5,\n\t\tCurrentTimeStamp:    time.Date(2025, 1, 6, 15, 0, 0, 0, time.UTC),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/testdata/shard.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc NewShardRow(ts time.Time) *nosqlplugin.ShardRow {\n\treturn &nosqlplugin.ShardRow{\n\t\tInternalShardInfo: &persistence.InternalShardInfo{\n\t\t\tShardID:                       15,\n\t\t\tOwner:                         \"owner\",\n\t\t\tRangeID:                       1000,\n\t\t\tReplicationAckLevel:           2000,\n\t\t\tTransferAckLevel:              3000,\n\t\t\tTimerAckLevel:                 ts.Add(-time.Hour),\n\t\t\tClusterTransferAckLevel:       map[string]int64{\"cluster2\": 4000},\n\t\t\tClusterTimerAckLevel:          map[string]time.Time{\"cluster2\": ts.Add(-2 * time.Hour)},\n\t\t\tDomainNotificationVersion:     3,\n\t\t\tClusterReplicationLevel:       map[string]int64{\"cluster2\": 5000},\n\t\t\tReplicationDLQAckLevel:        map[string]int64{\"cluster2\": 10},\n\t\t\tPendingFailoverMarkers:        &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"failovermarkers\")},\n\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"transferqueue\")},\n\t\t\tTimerProcessingQueueStates:    &persistence.DataBlob{Encoding: \"thriftrw\", Data: []byte(\"timerqueue\")},\n\t\t\tCurrentTimestamp:              ts,\n\t\t},\n\t\tData:         []byte(\"sharddata\"),\n\t\tDataEncoding: \"thriftrw\",\n\t}\n}\n\nfunc NewShardMap(ts time.Time) map[string]interface{} {\n\treturn map[string]interface{}{\n\t\t\"shard_id\":              int(15),\n\t\t\"range_id\":              int64(1000),\n\t\t\"owner\":                 \"owner\",\n\t\t\"stolen_since_renew\":    0,\n\t\t\"updated_at\":            ts,\n\t\t\"replication_ack_level\": int64(2000),\n\t\t\"transfer_ack_level\":    int64(3000),\n\t\t\"timer_ack_level\":       ts.Add(-1 * time.Hour),\n\t\t\"cluster_transfer_ack_level\": map[string]int64{\n\t\t\t\"cluster1\": int64(3000),\n\t\t},\n\t\t\"cluster_timer_ack_level\": map[string]time.Time{\n\t\t\t\"cluster1\": ts.Add(-1 * time.Hour),\n\t\t},\n\t\t\"transfer_processing_queue_states\":          []byte(\"transferqueue\"),\n\t\t\"transfer_processing_queue_states_encoding\": \"thriftrw\",\n\t\t\"timer_processing_queue_states\":             []byte(\"timerqueue\"),\n\t\t\"timer_processing_queue_states_encoding\":    \"thriftrw\",\n\t\t\"domain_notification_version\":               int64(3),\n\t\t\"cluster_replication_level\":                 map[string]int64{\"cluster2\": 1500},\n\t\t\"replication_dlq_ack_level\":                 map[string]int64{\"cluster2\": 5},\n\t\t\"pending_failover_markers\":                  []byte(\"failovermarkers\"),\n\t\t\"pending_failover_markers_encoding\":         \"thriftrw\",\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/testdata/visibility.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"log\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tDomainID      = \"test-domain-id\"\n\tWorkflowType  = \"test-workflow-type\"\n\tWorkflowID    = \"test-workflow-id\"\n\tRunID         = \"test-run-id\"\n\tTypeName      = \"test-type-name\"\n\tHistoryLenght = int64(1)\n\tTaskList      = \"test-task-list\"\n\tNumClusters   = int16(1)\n\tShardID       = int16(1)\n)\n\nfunc NewVisibilityRow() persistence.InternalVisibilityWorkflowExecutionInfo {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tlog.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\treturn persistence.InternalVisibilityWorkflowExecutionInfo{\n\t\tDomainID:      DomainID,\n\t\tWorkflowType:  WorkflowType,\n\t\tWorkflowID:    WorkflowID,\n\t\tRunID:         RunID,\n\t\tTypeName:      TypeName,\n\t\tStartTime:     ts,\n\t\tExecutionTime: ts,\n\t\tCloseTime:     ts,\n\t\tStatus:        types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\tHistoryLength: HistoryLenght,\n\t\tMemo: &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingTypeJSON,\n\t\t\tData:     []byte{},\n\t\t},\n\t\tTaskList:               TaskList,\n\t\tIsCron:                 false,\n\t\tNumClusters:            NumClusters,\n\t\tUpdateTime:             ts,\n\t\tSearchAttributes:       map[string]interface{}{},\n\t\tShardID:                ShardID,\n\t\tExecutionStatus:        types.WorkflowExecutionStatusPending,\n\t\tCronSchedule:           \"\",\n\t\tScheduledExecutionTime: time.Time{},\n\t}\n}\n\nfunc NewVisibilityRowForInsert() *nosqlplugin.VisibilityRowForInsert {\n\treturn &nosqlplugin.VisibilityRowForInsert{\n\t\tVisibilityRow: NewVisibilityRow(),\n\t\tDomainID:      DomainID,\n\t}\n}\n\nfunc NewVisibilityRowForUpdate(updateCloseToOpen, updateOpenToClose bool) *nosqlplugin.VisibilityRowForUpdate {\n\tvisibilityRow := NewVisibilityRow()\n\tvisibilityRow.CloseTime = visibilityRow.StartTime.Add(-1 * time.Minute)\n\treturn &nosqlplugin.VisibilityRowForUpdate{\n\t\tVisibilityRow:     visibilityRow,\n\t\tDomainID:          DomainID,\n\t\tUpdateCloseToOpen: updateCloseToOpen,\n\t\tUpdateOpenToClose: updateOpenToClose,\n\t}\n}\n\nfunc NewSelectVisibilityRequestFilter(filterType nosqlplugin.VisibilityFilterType, sortType nosqlplugin.VisibilitySortType) *nosqlplugin.VisibilityFilter {\n\tts, err := time.Parse(time.RFC3339, \"2024-04-01T22:08:41Z\")\n\tif err != nil {\n\t\tlog.Fatalf(\"Failed to parse time: %v\", err)\n\t}\n\treturn &nosqlplugin.VisibilityFilter{\n\t\tListRequest:  persistence.InternalListWorkflowExecutionsRequest{DomainUUID: DomainID, EarliestTime: ts, LatestTime: ts},\n\t\tFilterType:   filterType,\n\t\tSortType:     sortType,\n\t\tWorkflowType: WorkflowType,\n\t\tWorkflowID:   WorkflowID,\n\t\tCloseStatus:  0,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/testdata/workflow_execution.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\ntype WFExecRequestOption func(*nosqlplugin.WorkflowExecutionRequest)\n\nfunc WFExecRequestWithMapsWriteMode(mode nosqlplugin.WorkflowExecutionMapsWriteMode) WFExecRequestOption {\n\treturn func(request *nosqlplugin.WorkflowExecutionRequest) {\n\t\trequest.MapsWriteMode = mode\n\t}\n}\n\nfunc WFExecRequestWithEventBufferWriteMode(mode nosqlplugin.EventBufferWriteMode) WFExecRequestOption {\n\treturn func(request *nosqlplugin.WorkflowExecutionRequest) {\n\t\trequest.EventBufferWriteMode = mode\n\t}\n}\n\nfunc WFExecRequest(opts ...WFExecRequestOption) *nosqlplugin.WorkflowExecutionRequest {\n\tts := time.Now()\n\treq := &nosqlplugin.WorkflowExecutionRequest{\n\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\tDomainID:   \"test-domain-id\",\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tCompletionEvent: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte(\"test-completion-event\"),\n\t\t\t},\n\t\t\tAutoResetPoints: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte(\"test-auto-reset-points\"),\n\t\t\t},\n\t\t},\n\t\tVersionHistories: &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\tData:     []byte(\"test-version-histories\"),\n\t\t},\n\t\tChecksums: &checksum.Checksum{\n\t\t\tVersion: 1,\n\t\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\t\tValue:   []byte(\"test-checksum\"),\n\t\t},\n\t\tPreviousNextEventIDCondition: common.Int64Ptr(123),\n\t\tCurrentTimeStamp:             ts,\n\t}\n\n\tfor _, opt := range opts {\n\t\topt(req)\n\t}\n\n\treturn req\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/visibility.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nconst (\n\tdomainPartition = 0\n)\n\n// InsertVisibility creates a new visibility record, return error is there is any.\n// TODO: Cassandra implementation ignores search attributes\nfunc (db *CDB) InsertVisibility(ctx context.Context, ttlSeconds int64, row *nosqlplugin.VisibilityRowForInsert) error {\n\tvar query gocql.Query\n\tif ttlSeconds > maxCassandraTTL {\n\t\tquery = db.session.Query(templateCreateWorkflowExecutionStarted,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t).WithContext(ctx)\n\t} else {\n\t\tquery = db.session.Query(templateCreateWorkflowExecutionStartedWithTTL,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t\tttlSeconds,\n\t\t).WithContext(ctx)\n\t}\n\tquery = query.WithTimestamp(persistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()))\n\treturn query.Exec()\n}\n\nfunc (db *CDB) UpdateVisibility(ctx context.Context, ttlSeconds int64, row *nosqlplugin.VisibilityRowForUpdate) error {\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\n\tif row.UpdateCloseToOpen {\n\t\t// TODO implement it when where is a need\n\t\tpanic(\"not supported operation\")\n\t}\n\n\tif row.UpdateOpenToClose {\n\t\t// First, remove execution from the open table\n\t\tbatch.Query(templateDeleteWorkflowExecutionStarted,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\trow.RunID,\n\t\t)\n\t}\n\n\t// Next, add a row in the closed table.\n\tif ttlSeconds > maxCassandraTTL {\n\t\tbatch.Query(templateCreateWorkflowExecutionClosed,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.CloseTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Status,\n\t\t\trow.HistoryLength,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t)\n\t\t// duplicate write to v2 to order by close time\n\t\tbatch.Query(templateCreateWorkflowExecutionClosedV2,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.CloseTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Status,\n\t\t\trow.HistoryLength,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t)\n\t} else {\n\t\tbatch.Query(templateCreateWorkflowExecutionClosedWithTTL,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.CloseTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Status,\n\t\t\trow.HistoryLength,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t\tttlSeconds,\n\t\t)\n\t\t// duplicate write to v2 to order by close time\n\t\tbatch.Query(templateCreateWorkflowExecutionClosedWithTTLV2,\n\t\t\trow.DomainID,\n\t\t\tdomainPartition,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.StartTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ExecutionTime.UnixNano()),\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.CloseTime.UnixNano()),\n\t\t\trow.TypeName,\n\t\t\trow.Status,\n\t\t\trow.HistoryLength,\n\t\t\trow.Memo.Data,\n\t\t\trow.Memo.GetEncoding(),\n\t\t\trow.TaskList,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tpersistence.UnixNanoToDBTimestamp(row.ScheduledExecutionTime.UnixNano()),\n\t\t\tttlSeconds,\n\t\t)\n\t}\n\n\t// RecordWorkflowExecutionStarted is using StartTimestamp as\n\t// the timestamp to issue query to Cassandra\n\t// due to the fact that cross DC using mutable state creation time as workflow start time\n\t// and visibility using event time instead of last update time (#1501)\n\t// CloseTimestamp can be before StartTimestamp, meaning using CloseTimestamp\n\t// can cause the deletion of open visibility record to be ignored.\n\tqueryTimeStamp := row.CloseTime\n\tif queryTimeStamp.Before(row.StartTime) {\n\t\tqueryTimeStamp = row.StartTime.Add(time.Second)\n\t}\n\tbatch = batch.WithTimestamp(persistence.UnixNanoToDBTimestamp(queryTimeStamp.UnixNano()))\n\treturn db.session.ExecuteBatch(batch)\n}\n\nfunc (db *CDB) SelectOneClosedWorkflow(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) (*nosqlplugin.VisibilityRow, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecution,\n\t\tdomainID,\n\t\tdomainPartition,\n\t\tworkflowID,\n\t\trunID,\n\t).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, fmt.Errorf(\"not able to create query iterator\")\n\t}\n\n\twfexecution, has := readClosedWorkflowExecutionRecord(iter)\n\tif !has {\n\t\t// Special case: return nil,nil if not found(since we will deprecate it, it's not worth refactor to be consistent)\n\t\treturn nil, nil\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn wfexecution, nil\n}\n\n// Noop for Cassandra as it already handle by TTL\nfunc (db *CDB) DeleteVisibility(ctx context.Context, domainID, workflowID, runID string) error {\n\t// Normally we only depend on TTL for Cassandra visibility deletion but\n\t// we explicitly delete from open executions when an admin command is issued\n\tkey := persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\")\n\tif v := ctx.Value(key); v != nil && v.(bool) {\n\t\t// Primary key is <domainId, domainPartition, startTime, runId>\n\t\t// to optimize showing open executions sorted by their start time\n\t\t// However, it is not useful for deletion since we don't have the start\n\t\t// time information. So, we need to get the record first with \"allow\n\t\t// filtering\", read startTime then delete it. This is okay because\n\t\t// it is only intended to be used for admin ops.\n\n\t\trecord, err := db.openWorkflowByRunID(ctx, domainID, runID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif record == nil {\n\t\t\treturn nil // workflow not found, nothing to do\n\t\t}\n\n\t\tquery := db.session.Query(templateDeleteVisibilityRecord,\n\t\t\tdomainID,\n\t\t\tdomainPartition,\n\t\t\trecord.StartTime,\n\t\t\trunID,\n\t\t).WithContext(ctx)\n\t\treturn db.executeWithConsistencyAll(query)\n\t}\n\treturn nil\n}\n\nfunc (db *CDB) SelectVisibility(ctx context.Context, filter *nosqlplugin.VisibilityFilter) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tswitch filter.FilterType {\n\tcase nosqlplugin.AllOpen:\n\t\treturn db.openSortedByStartTime(ctx, &filter.ListRequest)\n\tcase nosqlplugin.AllClosed:\n\t\tswitch filter.SortType {\n\t\tcase nosqlplugin.SortByStartTime:\n\t\t\treturn db.closedSortedByStartTime(ctx, &filter.ListRequest)\n\t\tcase nosqlplugin.SortByClosedTime:\n\t\t\treturn db.closedSortedByClosedTime(ctx, &filter.ListRequest)\n\t\tdefault:\n\t\t\tpanic(\"not supported sorting type\")\n\t\t}\n\n\t// by workflowType\n\tcase nosqlplugin.OpenByWorkflowType:\n\t\treturn db.openFilteredByWorkflowTypeSortedByStartTime(ctx, &filter.ListRequest, filter.WorkflowType)\n\tcase nosqlplugin.ClosedByWorkflowType:\n\t\tswitch filter.SortType {\n\t\tcase nosqlplugin.SortByStartTime:\n\t\t\treturn db.closedFilteredByWorkflowTypeSortedByStartTime(ctx, &filter.ListRequest, filter.WorkflowType)\n\t\tcase nosqlplugin.SortByClosedTime:\n\t\t\treturn db.closedFilteredByWorkflowTypeSortedByClosedTime(ctx, &filter.ListRequest, filter.WorkflowType)\n\t\tdefault:\n\t\t\tpanic(\"not supported sorting type\")\n\t\t}\n\n\t// by workflowID\n\tcase nosqlplugin.OpenByWorkflowID:\n\t\treturn db.openFilteredByWorkflowIDSortedByStartTime(ctx, &filter.ListRequest, filter.WorkflowID)\n\tcase nosqlplugin.ClosedByWorkflowID:\n\t\tswitch filter.SortType {\n\t\tcase nosqlplugin.SortByStartTime:\n\t\t\treturn db.closedFilteredByWorkflowIDSortedByStartTime(ctx, &filter.ListRequest, filter.WorkflowID)\n\t\tcase nosqlplugin.SortByClosedTime:\n\t\t\treturn db.closedFilteredByWorkflowIDSortedByClosedTime(ctx, &filter.ListRequest, filter.WorkflowID)\n\t\tdefault:\n\t\t\tpanic(\"not supported sorting type\")\n\t\t}\n\n\t// closeStatus\n\tcase nosqlplugin.ClosedByClosedStatus:\n\t\tswitch filter.SortType {\n\t\tcase nosqlplugin.SortByStartTime:\n\t\t\treturn db.closedFilteredByClosedStatusSortedByStartTime(ctx, &filter.ListRequest, filter.CloseStatus)\n\t\tcase nosqlplugin.SortByClosedTime:\n\t\t\treturn db.closedFilteredByClosedStatusSortedByClosedTime(ctx, &filter.ListRequest, filter.CloseStatus)\n\t\tdefault:\n\t\t\tpanic(\"not supported sorting type\")\n\t\t}\n\tdefault:\n\t\tpanic(\"no supported filter type\")\n\t}\n}\n\nfunc (db *CDB) openFilteredByWorkflowTypeSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowType string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetOpenWorkflowExecutionsByType,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowType,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readOpenWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedFilteredByWorkflowTypeSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowType string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByType,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowType,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedFilteredByWorkflowTypeSortedByClosedTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowType string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByTypeSortByCloseTime,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowType,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) openFilteredByWorkflowIDSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowID string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetOpenWorkflowExecutionsByID,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowID,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readOpenWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) openWorkflowByRunID(\n\tctx context.Context,\n\tdomainID string,\n\trunID string,\n) (*persistence.InternalVisibilityWorkflowExecutionInfo, error) {\n\tquery := db.session.Query(templateGetOpenWorkflowExecutionsByRunID,\n\t\tdomainID,\n\t\tdomainPartition,\n\t\trunID,\n\t).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, fmt.Errorf(\"not able to create query iterator\")\n\t}\n\n\twfexecution, has := readOpenWorkflowExecutionRecord(iter)\n\tif !has {\n\t\treturn nil, nil\n\t}\n\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn wfexecution, nil\n}\n\nfunc (db *CDB) closedFilteredByWorkflowIDSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowID string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByID,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowID,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedFilteredByWorkflowIDSortedByClosedTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tworkflowID string,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByIDSortByCloseTime,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tworkflowID,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedFilteredByClosedStatusSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tcloseStatus int32,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByStatus,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tcloseStatus,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedFilteredByClosedStatusSortedByClosedTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\tcloseStatus int32,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsByStatusSortByClosedTime,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t\tcloseStatus,\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) openSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetOpenWorkflowExecutions,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\n\treturn processQuery(query, request, readOpenWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedSortedByStartTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutions,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\nfunc (db *CDB) closedSortedByClosedTime(\n\tctx context.Context,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tquery := db.session.Query(templateGetClosedWorkflowExecutionsSortByCloseTime,\n\t\trequest.DomainUUID,\n\t\tdomainPartition,\n\t\tpersistence.UnixNanoToDBTimestamp(request.EarliestTime.UnixNano()),\n\t\tpersistence.UnixNanoToDBTimestamp(request.LatestTime.UnixNano()),\n\t).Consistency(cassandraLowConslevel).WithContext(ctx)\n\treturn processQuery(query, request, readClosedWorkflowExecutionRecord)\n}\n\ntype recorderReaderFunc func(iter gocql.Iter) (*persistence.InternalVisibilityWorkflowExecutionInfo, bool)\n\nfunc processQuery(\n\tquery gocql.Query,\n\trequest *persistence.InternalListWorkflowExecutionsRequest,\n\trecorderReader recorderReaderFunc,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\titer := query.PageSize(request.PageSize).PageState(request.NextPageToken).Iter()\n\tif iter == nil {\n\t\t// TODO: may return badRequestError\n\t\treturn nil, fmt.Errorf(\"not able to create query iterator\")\n\t}\n\n\tresponse := &nosqlplugin.SelectVisibilityResponse{}\n\tresponse.Executions = make([]*persistence.InternalVisibilityWorkflowExecutionInfo, 0)\n\twfexecution, has := recorderReader(iter)\n\tfor has {\n\t\tresponse.Executions = append(response.Executions, wfexecution)\n\t\twfexecution, has = recorderReader(iter)\n\t}\n\n\tnextPageToken := iter.PageState()\n\tresponse.NextPageToken = make([]byte, len(nextPageToken))\n\tcopy(response.NextPageToken, nextPageToken)\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc readOpenWorkflowExecutionRecord(\n\titer gocql.Iter,\n) (*persistence.InternalVisibilityWorkflowExecutionInfo, bool) {\n\tvar workflowID string\n\tvar runID string\n\tvar typeName string\n\tvar startTime time.Time\n\tvar executionTime time.Time\n\tvar memo []byte\n\tvar encoding string\n\tvar taskList string\n\tvar isCron bool\n\tvar numClusters int16\n\tvar updateTime time.Time\n\tvar shardID int16\n\tvar executionStatus int32\n\tvar cronSchedule string\n\tvar scheduledExecutionTime time.Time\n\tif iter.Scan(&workflowID, &runID, &startTime, &executionTime, &typeName, &memo, &encoding, &taskList, &isCron, &numClusters, &updateTime, &shardID, &executionStatus, &cronSchedule, &scheduledExecutionTime) {\n\t\trecord := &persistence.InternalVisibilityWorkflowExecutionInfo{\n\t\t\tWorkflowID:             workflowID,\n\t\t\tRunID:                  runID,\n\t\t\tTypeName:               typeName,\n\t\t\tStartTime:              startTime,\n\t\t\tExecutionTime:          executionTime,\n\t\t\tMemo:                   persistence.NewDataBlob(memo, constants.EncodingType(encoding)),\n\t\t\tTaskList:               taskList,\n\t\t\tIsCron:                 isCron,\n\t\t\tNumClusters:            numClusters,\n\t\t\tUpdateTime:             updateTime,\n\t\t\tShardID:                shardID,\n\t\t\tExecutionStatus:        types.WorkflowExecutionStatus(executionStatus),\n\t\t\tCronSchedule:           cronSchedule,\n\t\t\tScheduledExecutionTime: scheduledExecutionTime,\n\t\t}\n\t\treturn record, true\n\t}\n\treturn nil, false\n}\n\nfunc readClosedWorkflowExecutionRecord(\n\titer gocql.Iter,\n) (*persistence.InternalVisibilityWorkflowExecutionInfo, bool) {\n\tvar workflowID string\n\tvar runID string\n\tvar typeName string\n\tvar startTime time.Time\n\tvar executionTime time.Time\n\tvar closeTime time.Time\n\tvar status workflow.WorkflowExecutionCloseStatus\n\tvar historyLength int64\n\tvar memo []byte\n\tvar encoding string\n\tvar taskList string\n\tvar isCron bool\n\tvar numClusters int16\n\tvar updateTime time.Time\n\tvar shardID int16\n\tvar executionStatus int32\n\tvar cronSchedule string\n\tvar scheduledExecutionTime time.Time\n\tif iter.Scan(&workflowID, &runID, &startTime, &executionTime, &closeTime, &typeName, &status, &historyLength, &memo, &encoding, &taskList, &isCron, &numClusters, &updateTime, &shardID, &executionStatus, &cronSchedule, &scheduledExecutionTime) {\n\t\trecord := &persistence.InternalVisibilityWorkflowExecutionInfo{\n\t\t\tWorkflowID:             workflowID,\n\t\t\tRunID:                  runID,\n\t\t\tTypeName:               typeName,\n\t\t\tStartTime:              startTime,\n\t\t\tExecutionTime:          executionTime,\n\t\t\tCloseTime:              closeTime,\n\t\t\tStatus:                 thrift.ToWorkflowExecutionCloseStatus(&status),\n\t\t\tHistoryLength:          historyLength,\n\t\t\tMemo:                   persistence.NewDataBlob(memo, constants.EncodingType(encoding)),\n\t\t\tTaskList:               taskList,\n\t\t\tIsCron:                 isCron,\n\t\t\tNumClusters:            numClusters,\n\t\t\tUpdateTime:             updateTime,\n\t\t\tShardID:                shardID,\n\t\t\tExecutionStatus:        types.WorkflowExecutionStatus(executionStatus),\n\t\t\tCronSchedule:           cronSchedule,\n\t\t\tScheduledExecutionTime: scheduledExecutionTime,\n\t\t}\n\t\treturn record, true\n\t}\n\treturn nil, false\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/visibility_cql.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\t// /////////////// Open Executions /////////////////\n\topenExecutionsColumnsForSelect = \" workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time \"\n\n\topenExecutionsColumnsForInsert = \"(domain_id, domain_partition, \" + openExecutionsColumnsForSelect + \")\"\n\n\ttemplateCreateWorkflowExecutionStartedWithTTL = `INSERT INTO open_executions ` +\n\t\topenExecutionsColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) using TTL ?`\n\n\ttemplateCreateWorkflowExecutionStarted = `INSERT INTO open_executions` +\n\t\topenExecutionsColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\ttemplateDeleteWorkflowExecutionStarted = `DELETE FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time = ? ` +\n\t\t`AND run_id = ?`\n\n\ttemplateGetOpenWorkflowExecutions = `SELECT ` + openExecutionsColumnsForSelect +\n\t\t`FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition IN (?) ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? `\n\n\ttemplateGetOpenWorkflowExecutionsByType = `SELECT ` + openExecutionsColumnsForSelect +\n\t\t`FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? ` +\n\t\t`AND workflow_type_name = ? `\n\n\ttemplateGetOpenWorkflowExecutionsByID = `SELECT ` + openExecutionsColumnsForSelect +\n\t\t`FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? ` +\n\t\t`AND workflow_id = ? `\n\n\t// Intended for admin operations only\n\ttemplateGetOpenWorkflowExecutionsByRunID = `SELECT ` + openExecutionsColumnsForSelect +\n\t\t`FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND run_id = ? ` +\n\t\t`ALLOW FILTERING`\n\n\t// Intended for admin operations only\n\ttemplateDeleteVisibilityRecord = `DELETE FROM open_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`and domain_partition = ? ` +\n\t\t`and start_time = ? ` +\n\t\t`and run_id = ? `\n\n\t// /////////////// Closed Executions /////////////////\n\tclosedExecutionColumnsForSelect = \" workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time \"\n\n\tclosedExecutionColumnsForInsert = \"(domain_id, domain_partition, \" + closedExecutionColumnsForSelect + \")\"\n\n\ttemplateCreateWorkflowExecutionClosedWithTTL = `INSERT INTO closed_executions ` +\n\t\tclosedExecutionColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) using TTL ?`\n\n\ttemplateCreateWorkflowExecutionClosed = `INSERT INTO closed_executions ` +\n\t\tclosedExecutionColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\ttemplateCreateWorkflowExecutionClosedWithTTLV2 = `INSERT INTO closed_executions_v2 ` +\n\t\tclosedExecutionColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) using TTL ?`\n\n\ttemplateCreateWorkflowExecutionClosedV2 = `INSERT INTO closed_executions_v2 ` +\n\t\tclosedExecutionColumnsForInsert +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\ttemplateGetClosedWorkflowExecutions = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition IN (?) ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? `\n\n\ttemplateGetClosedWorkflowExecutionsByType = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? ` +\n\t\t`AND workflow_type_name = ? `\n\n\ttemplateGetClosedWorkflowExecutionsByID = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? ` +\n\t\t`AND workflow_id = ? `\n\n\ttemplateGetClosedWorkflowExecutionsByStatus = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND start_time >= ? ` +\n\t\t`AND start_time <= ? ` +\n\t\t`AND status = ? `\n\n\ttemplateGetClosedWorkflowExecution = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND workflow_id = ? ` +\n\t\t`AND run_id = ? ALLOW FILTERING `\n\n\ttemplateGetClosedWorkflowExecutionsSortByCloseTime = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions_v2 ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition IN (?) ` +\n\t\t`AND close_time >= ? ` +\n\t\t`AND close_time <= ? `\n\n\ttemplateGetClosedWorkflowExecutionsByTypeSortByCloseTime = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions_v2 ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND close_time >= ? ` +\n\t\t`AND close_time <= ? ` +\n\t\t`AND workflow_type_name = ? `\n\n\ttemplateGetClosedWorkflowExecutionsByIDSortByCloseTime = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions_v2 ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND close_time >= ? ` +\n\t\t`AND close_time <= ? ` +\n\t\t`AND workflow_id = ? `\n\n\ttemplateGetClosedWorkflowExecutionsByStatusSortByClosedTime = `SELECT ` + closedExecutionColumnsForSelect +\n\t\t`FROM closed_executions_v2 ` +\n\t\t`WHERE domain_id = ? ` +\n\t\t`AND domain_partition = ? ` +\n\t\t`AND close_time >= ? ` +\n\t\t`AND close_time <= ? ` +\n\t\t`AND status = ? `\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/visibility_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/testdata\"\n)\n\nfunc TestInsertVisibility(t *testing.T) {\n\ttests := []struct {\n\t\tdesc          string\n\t\trow           *nosqlplugin.VisibilityRowForInsert\n\t\tttlSeconds    int64\n\t\tqueryMockFunc func(*gocql.MockQuery)\n\t\twantQueries   []string\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"Query with ttl less than maxCassandraTTL\",\n\t\t\trow:        testdata.NewVisibilityRowForInsert(),\n\t\t\tttlSeconds: int64(1000),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithTimestamp(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO open_executions (domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, test-type-name, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871) using TTL 1000`,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"Query With ttl greater than maxCassandraTTL\",\n\t\t\trow:        testdata.NewVisibilityRowForInsert(),\n\t\t\tttlSeconds: maxCassandraTTL + 1,\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithTimestamp(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO open_executions(domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, test-type-name, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871)`,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tif test.queryMockFunc != nil {\n\t\t\t\ttest.queryMockFunc(query)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertVisibility(context.Background(), test.ttlSeconds, test.row)\n\t\t\tif test.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, test.wantQueries, session.queries)\n\t\t})\n\t}\n}\n\nfunc TestUpdateVisibility(t *testing.T) {\n\ttests := []struct {\n\t\tdesc        string\n\t\trow         *nosqlplugin.VisibilityRowForUpdate\n\t\tttlSeconds  int64\n\t\twantQueries []string\n\t\twantErr     bool\n\t\twantPanic   bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"Query with ttl less than maxCassandraTTL\",\n\t\t\trow:        testdata.NewVisibilityRowForUpdate(false, true),\n\t\t\tttlSeconds: int64(100),\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time = 1712009321000 AND run_id = test-run-id`,\n\t\t\t\t`INSERT INTO closed_executions (domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, 1712009261000, test-type-name, COMPLETED, 1, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871) using TTL 100`,\n\t\t\t\t`INSERT INTO closed_executions_v2 (domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, 1712009261000, test-type-name, COMPLETED, 1, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871) using TTL 100`,\n\t\t\t},\n\t\t\twantErr:   false,\n\t\t\twantPanic: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"Query with ttl greater than maxCassandraTTL\",\n\t\t\trow:        testdata.NewVisibilityRowForUpdate(false, true),\n\t\t\tttlSeconds: maxCassandraTTL + 1,\n\t\t\twantQueries: []string{\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time = 1712009321000 AND run_id = test-run-id`,\n\t\t\t\t`INSERT INTO closed_executions (domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, 1712009261000, test-type-name, COMPLETED, 1, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871)`,\n\t\t\t\t`INSERT INTO closed_executions_v2 (domain_id, domain_partition,  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time )VALUES (test-domain-id, 0, test-workflow-id, test-run-id, 1712009321000, 1712009321000, 1712009261000, test-type-name, COMPLETED, 1, [], json, test-task-list, false, 1, 2024-04-01T22:08:41Z, 1, PENDING, , -6795364578871)`,\n\t\t\t},\n\t\t\twantErr:   false,\n\t\t\twantPanic: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"panic if updateCloseToOpen is set\",\n\t\t\trow:        testdata.NewVisibilityRowForUpdate(true, true),\n\t\t\tttlSeconds: int64(100),\n\t\t\twantErr:    false,\n\t\t\twantPanic:  true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.desc, func(t *testing.T) {\n\t\t\tdefer func() {\n\t\t\t\tr := recover()\n\t\t\t\tif (r != nil) != test.wantPanic {\n\t\t\t\t\tt.Errorf(\"test: %s, panicWanted: %v, panicOccured: %v\", test.desc, test.wantPanic, r != nil)\n\t\t\t\t}\n\t\t\t}()\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: gocql.NewMockQuery(ctrl),\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\t\t\terr := db.UpdateVisibility(context.Background(), test.ttlSeconds, test.row)\n\t\t\tif test.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, test.wantQueries, session.batches[0].queries)\n\t\t})\n\t}\n}\n\nfunc TestSelectOneClosedWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tdesc        string\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\titrMockFunc func(*gocql.MockIter)\n\t\twantQueries []string\n\t\twantError   bool\n\t\twantResult  bool\n\t}{\n\t\t{\n\t\t\tdesc:        \"return error when query's iter function returns nil\",\n\t\t\tdomainID:    testdata.DomainID,\n\t\t\tworkflowID:  testdata.WorkflowID,\n\t\t\trunID:       testdata.RunID,\n\t\t\titrMockFunc: nil,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND workflow_id = test-workflow-id AND run_id = test-run-id ALLOW FILTERING `,\n\t\t\t},\n\t\t\twantError:  true,\n\t\t\twantResult: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"return nil if reading closed_workflow_execution_record returns false\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND workflow_id = test-workflow-id AND run_id = test-run-id ALLOW FILTERING `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"return error if closing iterator fails\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(errors.New(\"close error\"))\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND workflow_id = test-workflow-id AND run_id = test-run-id ALLOW FILTERING `,\n\t\t\t},\n\t\t\twantError:  true,\n\t\t\twantResult: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"success\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND workflow_id = test-workflow-id AND run_id = test-run-id ALLOW FILTERING `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\tif test.itrMockFunc != nil {\n\t\t\t\titr := gocql.NewMockIter(ctrl)\n\t\t\t\ttest.itrMockFunc(itr)\n\t\t\t\tquery.EXPECT().Iter().Return(itr)\n\t\t\t} else {\n\t\t\t\tquery.EXPECT().Iter().Return(nil)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\t\t\tresult, err := db.SelectOneClosedWorkflow(context.Background(), test.domainID, test.workflowID, test.runID)\n\t\t\tif test.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tif test.wantResult {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t}\n\t\t\tassert.Equal(t, test.wantQueries, session.queries)\n\t\t})\n\t}\n}\n\nfunc TestDeleteVisibility(t *testing.T) {\n\ttests := []struct {\n\t\tdesc           string\n\t\tdomainID       string\n\t\tworkflowID     string\n\t\trunID          string\n\t\tmockItr        bool\n\t\titrMockFunc    func(*gocql.MockIter)\n\t\tqueryMockFunc  func(*gocql.MockQuery)\n\t\tcontext        context.Context\n\t\tdc             *persistence.DynamicConfiguration\n\t\tclientMockFunc func(*gocql.MockClient)\n\t\twantQueries    []string\n\t\twantError      bool\n\t}{\n\t\t{\n\t\t\tdesc:           \"return nil if visibility_admin_key not present in context\",\n\t\t\tdomainID:       \"\",\n\t\t\tworkflowID:     \"\",\n\t\t\trunID:          \"\",\n\t\t\tmockItr:        false,\n\t\t\titrMockFunc:    nil,\n\t\t\tqueryMockFunc:  nil,\n\t\t\tcontext:        context.Background(),\n\t\t\tdc:             &persistence.DynamicConfiguration{},\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries:    nil,\n\t\t\twantError:      false,\n\t\t},\n\t\t{\n\t\t\tdesc:        \"return error when query's iter function returns nil\",\n\t\t\tdomainID:    testdata.DomainID,\n\t\t\tworkflowID:  testdata.WorkflowID,\n\t\t\trunID:       testdata.RunID,\n\t\t\tmockItr:     true,\n\t\t\titrMockFunc: nil,\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tcontext:        context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tdc:             &persistence.DynamicConfiguration{},\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries:    []string{`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`},\n\t\t\twantError:      true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"return nil if reading open_workflow_execution_records returns false\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tmockItr:    true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(false)\n\t\t\t},\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tcontext:        context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tdc:             &persistence.DynamicConfiguration{},\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries:    []string{`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`},\n\t\t\twantError:      false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"return error if closing iterator fails\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tmockItr:    true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(errors.New(\"close error\"))\n\t\t\t},\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tcontext:        context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tdc:             &persistence.DynamicConfiguration{},\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries:    []string{`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`},\n\t\t\twantError:      true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"success with db dynamic_configuration as nil\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tmockItr:    true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(2)\n\t\t\t\tquery.EXPECT().Exec().Return(nil)\n\t\t\t},\n\t\t\tcontext:        context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tdc:             nil,\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`,\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id and domain_partition = 0 and start_time = 0001-01-01T00:00:00Z and run_id = test-run-id `,\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"success with enable_cassandra_all_consistency_level_delete db dynamic_configuration\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tmockItr:    true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(2)\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(nil)\n\t\t\t},\n\t\t\tcontext: context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tdc: &persistence.DynamicConfiguration{EnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn true\n\t\t\t}},\n\t\t\tclientMockFunc: nil,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`,\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id and domain_partition = 0 and start_time = 0001-01-01T00:00:00Z and run_id = test-run-id `,\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"succes with cassandra_default_consistency_level when cassandra_all_consistency_level fails\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tcontext:    context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(2)\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"all consistency level fail\"))\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(nil)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\tdc: &persistence.DynamicConfiguration{\n\t\t\t\tEnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\t},\n\t\t\tclientMockFunc: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`,\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id and domain_partition = 0 and start_time = 0001-01-01T00:00:00Z and run_id = test-run-id `,\n\t\t\t},\n\t\t\twantError: false,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"return error cassandra_all_consistency_level fails and error is not cassandra consistency error\",\n\t\t\tdomainID:   testdata.DomainID,\n\t\t\tworkflowID: testdata.WorkflowID,\n\t\t\trunID:      testdata.RunID,\n\t\t\tcontext:    context.WithValue(context.Background(), persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(2)\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"all consistency level fail\"))\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\tdc: &persistence.DynamicConfiguration{\n\t\t\t\tEnableCassandraAllConsistencyLevelDelete: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\t},\n\t\t\tclientMockFunc: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsCassandraConsistencyError(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND run_id = test-run-id ALLOW FILTERING`,\n\t\t\t\t`DELETE FROM open_executions WHERE domain_id = test-domain-id and domain_partition = 0 and start_time = 0001-01-01T00:00:00Z and run_id = test-run-id `,\n\t\t\t},\n\t\t\twantError: true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tif test.queryMockFunc != nil {\n\t\t\t\ttest.queryMockFunc(query)\n\t\t\t}\n\t\t\tif test.mockItr && test.itrMockFunc != nil {\n\t\t\t\titr := gocql.NewMockIter(ctrl)\n\t\t\t\ttest.itrMockFunc(itr)\n\t\t\t\tquery.EXPECT().Iter().Return(itr)\n\t\t\t} else if test.mockItr {\n\t\t\t\tquery.EXPECT().Iter().Return(nil)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif test.clientMockFunc != nil {\n\t\t\t\ttest.clientMockFunc(client)\n\t\t\t}\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, test.dc, DbWithClient(client))\n\t\t\terr := db.DeleteVisibility(test.context, test.domainID, test.workflowID, test.runID)\n\t\t\tif test.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, test.wantQueries, session.queries)\n\t\t})\n\t}\n}\n\nfunc TestSelectVisibility(t *testing.T) {\n\ttests := []struct {\n\t\tdesc          string\n\t\tfilter        *nosqlplugin.VisibilityFilter\n\t\tqueryMockFunc func(query *gocql.MockQuery)\n\t\tmockItr       bool\n\t\titrMockFunc   func(itr *gocql.MockIter)\n\t\twantQueries   []string\n\t\twantError     bool\n\t\twantResult    bool\n\t\twantPanic     bool\n\t}{\n\t\t{\n\t\t\tdesc:          \"panic for invalid filter type\",\n\t\t\tfilter:        testdata.NewSelectVisibilityRequestFilter(nosqlplugin.VisibilityFilterType(10), nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: nil,\n\t\t\tmockItr:       false,\n\t\t\titrMockFunc:   nil,\n\t\t\twantQueries:   nil,\n\t\t\twantError:     false,\n\t\t\twantResult:    false,\n\t\t\twantPanic:     true,\n\t\t},\n\t\t{\n\t\t\tdesc: \"all-open type filter return error if iterator is nil\",\n\t\t\t// passing sort type as StartByStartTime(as it is the default value) for cases where sort type is not needed\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllOpen, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr:     true,\n\t\t\titrMockFunc: nil,\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition IN (0) AND start_time >= 1712009321000 AND start_time <= 1712009321000 `,\n\t\t\t},\n\t\t\twantError:  true,\n\t\t\twantResult: false,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"all-open type filter return error if closing iterator fails\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllOpen, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(errors.New(\"close error\"))\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition IN (0) AND start_time >= 1712009321000 AND start_time <= 1712009321000 `,\n\t\t\t},\n\t\t\twantError:  true,\n\t\t\twantResult: false,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"all-open type filter success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllOpen, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition IN (0) AND start_time >= 1712009321000 AND start_time <= 1712009321000 `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"all-closed type filter sort by start-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllClosed, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition IN (0) AND start_time >= 1712009321000 AND start_time <= 1712009321000 `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"all-closed type filter sort by closed-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllClosed, nosqlplugin.SortByClosedTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions_v2 WHERE domain_id = test-domain-id AND domain_partition IN (0) AND close_time >= 1712009321000 AND close_time <= 1712009321000 `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"panic on all-closed type filter invalid sort attribute\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.AllClosed, nosqlplugin.VisibilitySortType(100)),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t},\n\t\t\tmockItr:     false,\n\t\t\titrMockFunc: nil,\n\t\t\twantQueries: nil,\n\t\t\twantError:   false,\n\t\t\twantResult:  false,\n\t\t\twantPanic:   true,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"open-by-workflow-type filter success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.OpenByWorkflowType, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time >= 1712009321000 AND start_time <= 1712009321000 AND workflow_type_name = test-workflow-type `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-workflow-type filter sort by start-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowType, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time >= 1712009321000 AND start_time <= 1712009321000 AND workflow_type_name = test-workflow-type `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-workflow-type filter sort by closed-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowType, nosqlplugin.SortByClosedTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions_v2 WHERE domain_id = test-domain-id AND domain_partition = 0 AND close_time >= 1712009321000 AND close_time <= 1712009321000 AND workflow_type_name = test-workflow-type `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:          \"panic on closed-by-workflow-type filter invalid sort attribute\",\n\t\t\tfilter:        testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowType, nosqlplugin.VisibilitySortType(100)),\n\t\t\tqueryMockFunc: nil,\n\t\t\tmockItr:       false,\n\t\t\titrMockFunc:   nil,\n\t\t\twantQueries:   nil,\n\t\t\twantError:     false,\n\t\t\twantResult:    false,\n\t\t\twantPanic:     true,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"open-by-workflow-id filter success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.OpenByWorkflowID, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(true)\n\t\t\t\titr.EXPECT().Scan(generateMockParams(15)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM open_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time >= 1712009321000 AND start_time <= 1712009321000 AND workflow_id = test-workflow-id `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-workflow-id filter sort by start-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowID, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time >= 1712009321000 AND start_time <= 1712009321000 AND workflow_id = test-workflow-id `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-workflow-id filter sort by closed-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowID, nosqlplugin.SortByClosedTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions_v2 WHERE domain_id = test-domain-id AND domain_partition = 0 AND close_time >= 1712009321000 AND close_time <= 1712009321000 AND workflow_id = test-workflow-id `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:          \"panic on closed-by-workflow-id filter invalid sort attribute\",\n\t\t\tfilter:        testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByWorkflowID, nosqlplugin.VisibilitySortType(100)),\n\t\t\tqueryMockFunc: nil,\n\t\t\tmockItr:       false,\n\t\t\titrMockFunc:   nil,\n\t\t\twantQueries:   nil,\n\t\t\twantError:     false,\n\t\t\twantResult:    false,\n\t\t\twantPanic:     true,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-closed-status filter sort by start-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByClosedStatus, nosqlplugin.SortByStartTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions WHERE domain_id = test-domain-id AND domain_partition = 0 AND start_time >= 1712009321000 AND start_time <= 1712009321000 AND status = 0 `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:   \"closed-by-closed-status filter sort by closed-time success\",\n\t\t\tfilter: testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByClosedStatus, nosqlplugin.SortByClosedTime),\n\t\t\tqueryMockFunc: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().Consistency(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageSize(gomock.Any()).Return(query)\n\t\t\t\tquery.EXPECT().PageState(gomock.Any()).Return(query)\n\t\t\t},\n\t\t\tmockItr: true,\n\t\t\titrMockFunc: func(itr *gocql.MockIter) {\n\t\t\t\titr.EXPECT().Scan(generateMockParams(18)...).Return(false)\n\t\t\t\titr.EXPECT().PageState().Return([]byte(\"test\"))\n\t\t\t\titr.EXPECT().Close().Return(nil)\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`SELECT  workflow_id, run_id, start_time, execution_time, close_time, workflow_type_name, status, history_length, memo, encoding, task_list, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time FROM closed_executions_v2 WHERE domain_id = test-domain-id AND domain_partition = 0 AND close_time >= 1712009321000 AND close_time <= 1712009321000 AND status = 0 `,\n\t\t\t},\n\t\t\twantError:  false,\n\t\t\twantResult: true,\n\t\t\twantPanic:  false,\n\t\t},\n\t\t{\n\t\t\tdesc:          \"panic on closed-by-closed-status filter invalid sort attribute\",\n\t\t\tfilter:        testdata.NewSelectVisibilityRequestFilter(nosqlplugin.ClosedByClosedStatus, nosqlplugin.VisibilitySortType(100)),\n\t\t\tqueryMockFunc: nil,\n\t\t\tmockItr:       false,\n\t\t\titrMockFunc:   nil,\n\t\t\twantQueries:   nil,\n\t\t\twantError:     false,\n\t\t\twantResult:    false,\n\t\t\twantPanic:     true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.desc, func(t *testing.T) {\n\t\t\tdefer func() {\n\t\t\t\tr := recover()\n\t\t\t\tif (r != nil) != test.wantPanic {\n\t\t\t\t\tt.Errorf(\"test: %s, panicWanted: %v, panicOccured: %v\", test.desc, test.wantPanic, r != nil)\n\t\t\t\t}\n\t\t\t}()\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tif test.queryMockFunc != nil {\n\t\t\t\ttest.queryMockFunc(query)\n\t\t\t}\n\t\t\tif test.mockItr && test.itrMockFunc != nil {\n\t\t\t\titr := gocql.NewMockIter(ctrl)\n\t\t\t\ttest.itrMockFunc(itr)\n\t\t\t\tquery.EXPECT().Iter().Return(itr)\n\t\t\t} else if test.mockItr {\n\t\t\t\tquery.EXPECT().Iter().Return(nil)\n\t\t\t}\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\t\t\tresult, err := db.SelectVisibility(context.Background(), test.filter)\n\t\t\tif test.wantError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tif test.wantResult {\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t}\n\t\t\tassert.Equal(t, test.wantQueries, session.queries)\n\t\t})\n\t}\n}\n\nfunc generateMockParams(count int) []interface{} {\n\tparams := []interface{}{}\n\tfor i := 0; i < count; i++ {\n\t\tparams = append(params, gomock.Any())\n\t}\n\treturn params\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _ nosqlplugin.WorkflowCRUD = (*CDB)(nil)\n\nfunc (db *CDB) InsertWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tshardID := shardCondition.ShardID\n\tdomainID := execution.DomainID\n\tworkflowID := execution.WorkflowID\n\ttimeStamp := execution.CurrentTimeStamp\n\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\n\terr := insertWorkflowActiveClusterSelectionPolicyRow(batch, activeClusterSelectionPolicyRow, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = insertOrUpsertWorkflowRequestRow(batch, requests, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = createOrUpdateCurrentWorkflow(batch, shardID, domainID, workflowID, currentWorkflowRequest, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = createWorkflowExecutionWithMergeMaps(batch, shardID, domainID, workflowID, execution, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tcreateTasksByCategory(batch, shardID, domainID, workflowID, timeStamp, tasksByCategory)\n\n\tassertShardRangeID(batch, shardID, shardCondition.RangeID, timeStamp)\n\n\treturn executeCreateWorkflowBatchTransaction(ctx, db.session, batch, currentWorkflowRequest, execution, shardCondition)\n}\n\nfunc (db *CDB) SelectCurrentWorkflow(\n\tctx context.Context,\n\tshardID int, domainID, workflowID string,\n) (*nosqlplugin.CurrentWorkflowRow, error) {\n\tquery := db.session.Query(templateGetCurrentExecutionQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tpermanentRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\treturn nil, err\n\t}\n\n\tcurrentRunID := result[\"current_run_id\"].(gocql.UUID).String()\n\texecutionInfo, err := parseWorkflowExecutionInfo(result)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastWriteVersion := constants.EmptyVersion\n\tif result[\"workflow_last_write_version\"] != nil {\n\t\tlastWriteVersion = result[\"workflow_last_write_version\"].(int64)\n\t}\n\treturn &nosqlplugin.CurrentWorkflowRow{\n\t\tShardID:          shardID,\n\t\tDomainID:         domainID,\n\t\tWorkflowID:       workflowID,\n\t\tRunID:            currentRunID,\n\t\tCreateRequestID:  executionInfo.CreateRequestID,\n\t\tState:            executionInfo.State,\n\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}, nil\n}\n\nfunc (db *CDB) UpdateWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\tmutatedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tinsertedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tresetExecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tshardID := shardCondition.ShardID\n\tvar domainID, workflowID string\n\tvar previousNextEventIDCondition int64\n\tvar timeStamp time.Time\n\tif mutatedExecution != nil {\n\t\tdomainID = mutatedExecution.DomainID\n\t\tworkflowID = mutatedExecution.WorkflowID\n\t\tpreviousNextEventIDCondition = *mutatedExecution.PreviousNextEventIDCondition\n\t\ttimeStamp = mutatedExecution.CurrentTimeStamp\n\t} else if resetExecution != nil {\n\t\tdomainID = resetExecution.DomainID\n\t\tworkflowID = resetExecution.WorkflowID\n\t\tpreviousNextEventIDCondition = *resetExecution.PreviousNextEventIDCondition\n\t\ttimeStamp = resetExecution.CurrentTimeStamp\n\t} else {\n\t\treturn fmt.Errorf(\"at least one of mutatedExecution and resetExecution should be provided\")\n\t}\n\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\n\terr := insertOrUpsertWorkflowRequestRow(batch, requests, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = createOrUpdateCurrentWorkflow(batch, shardID, domainID, workflowID, currentWorkflowRequest, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif mutatedExecution != nil {\n\t\terr = updateWorkflowExecutionAndEventBufferWithMergeAndDeleteMaps(batch, shardID, domainID, workflowID, mutatedExecution, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif insertedExecution != nil {\n\t\terr = createWorkflowExecutionWithMergeMaps(batch, shardID, domainID, workflowID, insertedExecution, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\terr = insertWorkflowActiveClusterSelectionPolicyRow(batch, activeClusterSelectionPolicyRow, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t}\n\n\tif resetExecution != nil {\n\t\terr = resetWorkflowExecutionAndMapsAndEventBuffer(batch, shardID, domainID, workflowID, resetExecution, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tcreateTasksByCategory(batch, shardID, domainID, workflowID, timeStamp, tasksByCategory)\n\n\tassertShardRangeID(batch, shardID, shardCondition.RangeID, timeStamp)\n\n\treturn executeUpdateWorkflowBatchTransaction(ctx, db.session, batch, currentWorkflowRequest, previousNextEventIDCondition, shardCondition)\n}\n\nfunc (db *CDB) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*nosqlplugin.WorkflowExecution, error) {\n\tquery := db.session.Query(templateGetWorkflowExecutionQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\treturn nil, err\n\t}\n\n\tstate := &nosqlplugin.WorkflowExecution{}\n\tinfo, err := parseWorkflowExecutionInfo(result)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tstate.ExecutionInfo = info\n\tstate.VersionHistories = persistence.NewDataBlob(result[\"version_histories\"].([]byte), constants.EncodingType(result[\"version_histories_encoding\"].(string)))\n\t// TODO: remove this after all 2DC workflows complete\n\treplicationState := parseReplicationState(result[\"replication_state\"].(map[string]interface{}))\n\tstate.ReplicationState = replicationState\n\n\tactivityInfos := make(map[int64]*persistence.InternalActivityInfo)\n\taMap := result[\"activity_map\"].(map[int64]map[string]interface{})\n\tfor key, value := range aMap {\n\t\tinfo := parseActivityInfo(domainID, value)\n\t\tactivityInfos[key] = info\n\t}\n\tstate.ActivityInfos = activityInfos\n\n\ttimerInfos := make(map[string]*persistence.TimerInfo)\n\ttMap := result[\"timer_map\"].(map[string]map[string]interface{})\n\tfor key, value := range tMap {\n\t\tinfo := parseTimerInfo(value)\n\t\ttimerInfos[key] = info\n\t}\n\tstate.TimerInfos = timerInfos\n\n\tchildExecutionInfos := make(map[int64]*persistence.InternalChildExecutionInfo)\n\tcMap := result[\"child_executions_map\"].(map[int64]map[string]interface{})\n\tfor key, value := range cMap {\n\t\tinfo := parseChildExecutionInfo(value)\n\t\tchildExecutionInfos[key] = info\n\t}\n\tstate.ChildExecutionInfos = childExecutionInfos\n\n\trequestCancelInfos := make(map[int64]*persistence.RequestCancelInfo)\n\trMap := result[\"request_cancel_map\"].(map[int64]map[string]interface{})\n\tfor key, value := range rMap {\n\t\tinfo := parseRequestCancelInfo(value)\n\t\trequestCancelInfos[key] = info\n\t}\n\tstate.RequestCancelInfos = requestCancelInfos\n\n\tsignalInfos := make(map[int64]*persistence.SignalInfo)\n\tsMap := result[\"signal_map\"].(map[int64]map[string]interface{})\n\tfor key, value := range sMap {\n\t\tinfo := parseSignalInfo(value)\n\t\tsignalInfos[key] = info\n\t}\n\tstate.SignalInfos = signalInfos\n\n\tsignalRequestedIDs := make(map[string]struct{})\n\tsList := mustConvertToSlice(result[\"signal_requested\"])\n\tfor _, v := range sList {\n\t\tsignalRequestedIDs[v.(gocql.UUID).String()] = struct{}{}\n\t}\n\tstate.SignalRequestedIDs = signalRequestedIDs\n\n\teList := result[\"buffered_events_list\"].([]map[string]interface{})\n\tbufferedEventsBlobs := make([]*persistence.DataBlob, 0, len(eList))\n\tfor _, v := range eList {\n\t\tblob := parseHistoryEventBatchBlob(v)\n\t\tbufferedEventsBlobs = append(bufferedEventsBlobs, blob)\n\t}\n\tstate.BufferedEvents = bufferedEventsBlobs\n\n\tstate.Checksum = parseChecksum(result[\"checksum\"].(map[string]interface{}))\n\treturn state, nil\n}\n\nfunc (db *CDB) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tquery := db.session.Query(templateDeleteWorkflowExecutionCurrentRowQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tpermanentRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t\tcurrentRunIDCondition,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tquery := db.session.Query(templateDeleteWorkflowExecutionMutableStateQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tquery := db.session.Query(\n\t\ttemplateListCurrentExecutionsQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectAllCurrentWorkflows operation failed. Not able to create query iterator.\",\n\t\t}\n\t}\n\tresult := make(map[string]interface{})\n\tvar executions []*persistence.CurrentWorkflowExecution\n\tfor iter.MapScan(result) {\n\t\trunID := result[\"run_id\"].(gocql.UUID).String()\n\t\tif runID != permanentRunID {\n\t\t\tresult = make(map[string]interface{})\n\t\t\tcontinue\n\t\t}\n\t\texecutions = append(executions, &persistence.CurrentWorkflowExecution{\n\t\t\tDomainID:     result[\"domain_id\"].(gocql.UUID).String(),\n\t\t\tWorkflowID:   result[\"workflow_id\"].(string),\n\t\t\tRunID:        permanentRunID,\n\t\t\tState:        result[\"workflow_state\"].(int),\n\t\t\tCurrentRunID: result[\"current_run_id\"].(gocql.UUID).String(),\n\t\t})\n\t\tresult = make(map[string]interface{})\n\t}\n\tnextPageToken := getNextPageToken(iter)\n\n\treturn executions, nextPageToken, iter.Close()\n}\n\nfunc (db *CDB) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tquery := db.session.Query(\n\t\ttemplateListWorkflowExecutionQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectAllWorkflowExecutions operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tresult := make(map[string]interface{})\n\tvar executions []*persistence.InternalListConcreteExecutionsEntity\n\tfor iter.MapScan(result) {\n\t\trunID := result[\"run_id\"].(gocql.UUID).String()\n\t\tif runID == permanentRunID {\n\t\t\tresult = make(map[string]interface{})\n\t\t\tcontinue\n\t\t}\n\t\texecutionInfo, err := parseWorkflowExecutionInfo(result)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\texecutions = append(executions, &persistence.InternalListConcreteExecutionsEntity{\n\t\t\tExecutionInfo:    executionInfo,\n\t\t\tVersionHistories: persistence.NewDataBlob(result[\"version_histories\"].([]byte), constants.EncodingType(result[\"version_histories_encoding\"].(string))),\n\t\t})\n\t\tresult = make(map[string]interface{})\n\t}\n\tnextPageToken := getNextPageToken(iter)\n\n\treturn executions, nextPageToken, iter.Close()\n}\n\nfunc (db *CDB) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tquery := db.session.Query(templateIsWorkflowExecutionExistsQuery,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\tif db.client.IsNotFoundError(err) {\n\t\t\treturn false, nil\n\t\t}\n\n\t\treturn false, err\n\t}\n\treturn true, nil\n}\n\nfunc (db *CDB) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\t// Reading transfer tasks need to be quorum level consistent, otherwise we could loose task\n\tquery := db.session.Query(templateGetTransferTasksQuery,\n\t\tshardID,\n\t\trowTypeTransferTask,\n\t\trowTypeTransferDomainID,\n\t\trowTypeTransferWorkflowID,\n\t\trowTypeTransferRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\tinclusiveMinTaskID,\n\t\texclusiveMaxTaskID,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectTransferTasksOrderByTaskID operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tvar tasks []*nosqlplugin.HistoryMigrationTask\n\ttask := make(map[string]interface{})\n\tfor iter.MapScan(task) {\n\t\tt := parseTransferTaskInfo(task[\"transfer\"].(map[string]interface{}))\n\t\ttaskID := task[\"task_id\"].(int64)\n\t\tdata := task[\"data\"].([]byte)\n\t\tencoding := task[\"data_encoding\"].(string)\n\t\ttaskBlob := persistence.NewDataBlob(data, constants.EncodingType(encoding))\n\n\t\t// Reset task map to get it ready for next scan\n\t\ttask = make(map[string]interface{})\n\n\t\ttasks = append(tasks, &nosqlplugin.HistoryMigrationTask{\n\t\t\tTransfer: t,\n\t\t\tTask:     taskBlob,\n\t\t\tTaskID:   taskID,\n\t\t})\n\t}\n\tnextPageToken := getNextPageToken(iter)\n\n\terr := iter.Close()\n\treturn tasks, nextPageToken, err\n}\n\nfunc (db *CDB) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tquery := db.session.Query(templateCompleteTransferTaskQuery,\n\t\tshardID,\n\t\trowTypeTransferTask,\n\t\trowTypeTransferDomainID,\n\t\trowTypeTransferWorkflowID,\n\t\trowTypeTransferRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\ttaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) RangeDeleteTransferTasks(ctx context.Context, shardID int, exclusiveBeginTaskID, inclusiveEndTaskID int64) error {\n\tquery := db.session.Query(templateRangeCompleteTransferTaskQuery,\n\t\tshardID,\n\t\trowTypeTransferTask,\n\t\trowTypeTransferDomainID,\n\t\trowTypeTransferWorkflowID,\n\t\trowTypeTransferRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\texclusiveBeginTaskID,\n\t\tinclusiveEndTaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\t// Reading timer tasks need to be quorum level consistent, otherwise we could loose task\n\tminTimestamp := persistence.UnixNanoToDBTimestamp(inclusiveMinTime.UnixNano())\n\tmaxTimestamp := persistence.UnixNanoToDBTimestamp(exclusiveMaxTime.UnixNano())\n\tquery := db.session.Query(templateGetTimerTasksQuery,\n\t\tshardID,\n\t\trowTypeTimerTask,\n\t\trowTypeTimerDomainID,\n\t\trowTypeTimerWorkflowID,\n\t\trowTypeTimerRunID,\n\t\tminTimestamp,\n\t\tmaxTimestamp,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"SelectTimerTasksOrderByVisibilityTime operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tvar timers []*nosqlplugin.HistoryMigrationTask\n\ttask := make(map[string]interface{})\n\tfor iter.MapScan(task) {\n\t\tt := parseTimerTaskInfo(task[\"timer\"].(map[string]interface{}))\n\t\ttaskID := task[\"task_id\"].(int64)\n\t\tscheduledTime := task[\"visibility_ts\"].(time.Time)\n\t\tdata := task[\"data\"].([]byte)\n\t\tencoding := task[\"data_encoding\"].(string)\n\t\ttaskBlob := persistence.NewDataBlob(data, constants.EncodingType(encoding))\n\n\t\t// Reset task map to get it ready for next scan\n\t\ttask = make(map[string]interface{})\n\n\t\ttimers = append(timers, &nosqlplugin.HistoryMigrationTask{\n\t\t\tTimer:         t,\n\t\t\tTask:          taskBlob,\n\t\t\tTaskID:        taskID,\n\t\t\tScheduledTime: scheduledTime,\n\t\t})\n\t}\n\tnextPageToken := getNextPageToken(iter)\n\n\terr := iter.Close()\n\treturn timers, nextPageToken, err\n}\n\nfunc (db *CDB) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tts := persistence.UnixNanoToDBTimestamp(visibilityTimestamp.UnixNano())\n\tquery := db.session.Query(templateCompleteTimerTaskQuery,\n\t\tshardID,\n\t\trowTypeTimerTask,\n\t\trowTypeTimerDomainID,\n\t\trowTypeTimerWorkflowID,\n\t\trowTypeTimerRunID,\n\t\tts,\n\t\ttaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tstart := persistence.UnixNanoToDBTimestamp(inclusiveMinTime.UnixNano())\n\tend := persistence.UnixNanoToDBTimestamp(exclusiveMaxTime.UnixNano())\n\tquery := db.session.Query(templateRangeCompleteTimerTaskQuery,\n\t\tshardID,\n\t\trowTypeTimerTask,\n\t\trowTypeTimerDomainID,\n\t\trowTypeTimerWorkflowID,\n\t\trowTypeTimerRunID,\n\t\tstart,\n\t\tend,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\t// Reading replication tasks need to be quorum level consistent, otherwise we could loose task\n\tquery := db.session.Query(templateGetReplicationTasksQuery,\n\t\tshardID,\n\t\trowTypeReplicationTask,\n\t\trowTypeReplicationDomainID,\n\t\trowTypeReplicationWorkflowID,\n\t\trowTypeReplicationRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\tinclusiveMinTaskID,\n\t\texclusiveMaxTaskID,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\treturn populateGetReplicationTasks(query)\n}\n\nfunc (db *CDB) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tquery := db.session.Query(templateCompleteReplicationTaskQuery,\n\t\tshardID,\n\t\trowTypeReplicationTask,\n\t\trowTypeReplicationDomainID,\n\t\trowTypeReplicationWorkflowID,\n\t\trowTypeReplicationRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\ttaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) RangeDeleteReplicationTasks(ctx context.Context, shardID int, inclusiveEndTaskID int64) error {\n\tquery := db.session.Query(templateCompleteReplicationTaskBeforeQuery,\n\t\tshardID,\n\t\trowTypeReplicationTask,\n\t\trowTypeReplicationDomainID,\n\t\trowTypeReplicationWorkflowID,\n\t\trowTypeReplicationRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\tinclusiveEndTaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tquery := db.session.Query(templateCompleteCrossClusterTaskQuery,\n\t\tshardID,\n\t\trowTypeCrossClusterTask,\n\t\trowTypeCrossClusterDomainID,\n\t\ttargetCluster,\n\t\trowTypeCrossClusterRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\ttaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, replicationTask *nosqlplugin.HistoryMigrationTask) error {\n\t// Use source cluster name as the workflow id for replication dlq\n\ttask := replicationTask.Replication\n\ttaskBlob, taskEncoding := persistence.FromDataBlob(replicationTask.Task)\n\tquery := db.session.Query(templateCreateReplicationTaskQuery,\n\t\tshardID,\n\t\trowTypeDLQ,\n\t\trowTypeDLQDomainID,\n\t\tsourceCluster,\n\t\trowTypeDLQRunID,\n\t\ttask.DomainID,\n\t\ttask.WorkflowID,\n\t\ttask.RunID,\n\t\ttask.TaskID,\n\t\ttask.TaskType,\n\t\ttask.FirstEventID,\n\t\ttask.NextEventID,\n\t\ttask.Version,\n\t\ttask.ScheduledID,\n\t\tpersistence.EventStoreVersion,\n\t\ttask.BranchToken,\n\t\tpersistence.EventStoreVersion,\n\t\ttask.NewRunBranchToken,\n\t\tdefaultVisibilityTimestamp,\n\t\ttaskBlob,\n\t\ttaskEncoding,\n\t\tdefaultVisibilityTimestamp,\n\t\ttask.TaskID,\n\t\ttask.CurrentTimeStamp,\n\t).WithContext(ctx)\n\n\treturn query.Exec()\n}\n\nfunc (db *CDB) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\t// Reading replication tasks need to be quorum level consistent, otherwise we could loose task\n\tquery := db.session.Query(templateGetReplicationTasksQuery,\n\t\tshardID,\n\t\trowTypeDLQ,\n\t\trowTypeDLQDomainID,\n\t\tsourceCluster,\n\t\trowTypeDLQRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\tinclusiveMinTaskID,\n\t\texclusiveMaxTaskID,\n\t).PageSize(pageSize).PageState(pageToken).WithContext(ctx)\n\n\treturn populateGetReplicationTasks(query)\n}\n\nfunc (db *CDB) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\t// Reading replication tasks need to be quorum level consistent, otherwise we could loose task\n\tquery := db.session.Query(templateGetDLQSizeQuery,\n\t\tshardID,\n\t\trowTypeDLQ,\n\t\trowTypeDLQDomainID,\n\t\tsourceCluster,\n\t\trowTypeDLQRunID,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\treturn -1, err\n\t}\n\n\tqueueSize := result[\"count\"].(int64)\n\treturn queueSize, nil\n}\n\nfunc (db *CDB) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tquery := db.session.Query(templateCompleteReplicationTaskQuery,\n\t\tshardID,\n\t\trowTypeDLQ,\n\t\trowTypeDLQDomainID,\n\t\tsourceCluster,\n\t\trowTypeDLQRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\ttaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tquery := db.session.Query(templateRangeCompleteReplicationTaskQuery,\n\t\tshardID,\n\t\trowTypeDLQ,\n\t\trowTypeDLQDomainID,\n\t\tsourceCluster,\n\t\trowTypeDLQRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\tinclusiveBeginTaskID,\n\t\texclusiveEndTaskID,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n\nfunc (db *CDB) InsertReplicationTask(ctx context.Context, tasks []*nosqlplugin.HistoryMigrationTask, shardCondition nosqlplugin.ShardCondition) error {\n\tif len(tasks) == 0 {\n\t\treturn nil\n\t}\n\n\tshardID := shardCondition.ShardID\n\tbatch := db.session.NewBatch(gocql.LoggedBatch).WithContext(ctx)\n\ttimeStamp := tasks[0].Replication.CurrentTimeStamp\n\tfor _, task := range tasks {\n\t\tcreateReplicationTasks(batch, shardID, task.Replication.DomainID, task.Replication.WorkflowID, []*nosqlplugin.HistoryMigrationTask{task}, timeStamp)\n\t}\n\n\tassertShardRangeID(batch, shardID, shardCondition.RangeID, timeStamp)\n\n\tprevious := make(map[string]interface{})\n\tapplied, iter, err := db.session.MapExecuteBatchCAS(batch, previous)\n\tdefer func() {\n\t\tif iter != nil {\n\t\t\t_ = iter.Close()\n\t\t}\n\t}()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !applied {\n\t\trowType, ok := previous[\"type\"].(int)\n\t\tif !ok {\n\t\t\t// This should never happen, as all our rows have the type field.\n\t\t\tpanic(\"Encounter row type not found\")\n\t\t}\n\t\tif rowType == rowTypeShard {\n\t\t\tif actualRangeID, ok := previous[\"range_id\"].(int64); ok && actualRangeID != shardCondition.RangeID {\n\t\t\t\t// CreateWorkflowExecution failed because rangeID was modified\n\t\t\t\treturn &nosqlplugin.ShardOperationConditionFailure{\n\t\t\t\t\tRangeID: actualRangeID,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\t// At this point we only know that the write was not applied.\n\t\t// It's much safer to return ShardOperationConditionFailure(which will become ShardOwnershipLostError later) as the default to force the application to reload\n\t\t// shard to recover from such errors\n\t\tvar columns []string\n\t\tfor k, v := range previous {\n\t\t\tcolumns = append(columns, fmt.Sprintf(\"%s=%v\", k, v))\n\t\t}\n\t\treturn &nosqlplugin.ShardOperationConditionFailure{\n\t\t\tRangeID: -1,\n\t\t\tDetails: strings.Join(columns, \",\"),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (db *CDB) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*nosqlplugin.ActiveClusterSelectionPolicyRow, error) {\n\tquery := db.session.Query(templateGetActiveClusterSelectionPolicyQuery,\n\t\tshardID,\n\t\trowTypeWorkflowActiveClusterSelectionPolicy,\n\t\tdomainID,\n\t\twfID,\n\t\trID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeWorkflowActiveClusterSelectionVersion,\n\t).WithContext(ctx)\n\n\tresult := make(map[string]interface{})\n\tif err := query.MapScan(result); err != nil {\n\t\tif db.client.IsNotFoundError(err) {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\n\treturn &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: wfID,\n\t\tRunID:      rID,\n\t\tPolicy:     persistence.NewDataBlob(result[\"data\"].([]byte), constants.EncodingType(result[\"data_encoding\"].(string))),\n\t}, nil\n}\n\nfunc (db *CDB) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tquery := db.session.Query(templateDeleteActiveClusterSelectionPolicyQuery,\n\t\tshardID,\n\t\trowTypeWorkflowActiveClusterSelectionPolicy,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeWorkflowActiveClusterSelectionVersion,\n\t).WithContext(ctx)\n\n\treturn db.executeWithConsistencyAll(query)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_cql.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nconst (\n\ttemplateWorkflowExecutionType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`workflow_id: ?, ` +\n\t\t`run_id: ?, ` +\n\t\t`first_run_id: ?, ` +\n\t\t`parent_domain_id: ?, ` +\n\t\t`parent_workflow_id: ?, ` +\n\t\t`parent_run_id: ?, ` +\n\t\t`initiated_id: ?, ` +\n\t\t`completion_event_batch_id: ?, ` +\n\t\t`completion_event: ?, ` +\n\t\t`completion_event_data_encoding: ?, ` +\n\t\t`task_list: ?, ` +\n\t\t`task_list_kind: ?, ` +\n\t\t`workflow_type_name: ?, ` +\n\t\t`workflow_timeout: ?, ` +\n\t\t`decision_task_timeout: ?, ` +\n\t\t`execution_context: ?, ` +\n\t\t`state: ?, ` +\n\t\t`close_status: ?, ` +\n\t\t`last_first_event_id: ?, ` +\n\t\t`last_event_task_id: ?, ` +\n\t\t`next_event_id: ?, ` +\n\t\t`last_processed_event: ?, ` +\n\t\t`start_time: ?, ` +\n\t\t`last_updated_time: ?, ` +\n\t\t`create_request_id: ?, ` +\n\t\t`signal_count: ?, ` +\n\t\t`history_size: ?, ` +\n\t\t`decision_version: ?, ` +\n\t\t`decision_schedule_id: ?, ` +\n\t\t`decision_started_id: ?, ` +\n\t\t`decision_request_id: ?, ` +\n\t\t`decision_timeout: ?, ` +\n\t\t`decision_attempt: ?, ` +\n\t\t`decision_timestamp: ?, ` +\n\t\t`decision_scheduled_timestamp: ?, ` +\n\t\t`decision_original_scheduled_timestamp: ?, ` +\n\t\t`cancel_requested: ?, ` +\n\t\t`cancel_request_id: ?, ` +\n\t\t`sticky_task_list: ?, ` +\n\t\t`sticky_schedule_to_start_timeout: ?,` +\n\t\t`client_library_version: ?, ` +\n\t\t`client_feature_version: ?, ` +\n\t\t`client_impl: ?, ` +\n\t\t`auto_reset_points: ?, ` +\n\t\t`auto_reset_points_encoding: ?, ` +\n\t\t`attempt: ?, ` +\n\t\t`has_retry_policy: ?, ` +\n\t\t`init_interval: ?, ` +\n\t\t`backoff_coefficient: ?, ` +\n\t\t`max_interval: ?, ` +\n\t\t`expiration_time: ?, ` +\n\t\t`max_attempts: ?, ` +\n\t\t`non_retriable_errors: ?, ` +\n\t\t`event_store_version: ?, ` +\n\t\t`branch_token: ?, ` +\n\t\t`cron_schedule: ?, ` +\n\t\t`cron_overlap_policy: ?, ` +\n\t\t`expiration_seconds: ?, ` +\n\t\t`search_attributes: ?, ` +\n\t\t`memo: ?, ` +\n\t\t`partition_config: ?, ` +\n\t\t`active_cluster_selection_policy: ?, ` +\n\t\t`active_cluster_selection_policy_encoding: ?` +\n\t\t`}`\n\n\ttemplateTransferTaskType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`workflow_id: ?, ` +\n\t\t`run_id: ?, ` +\n\t\t`visibility_ts: ?, ` +\n\t\t`task_id: ?, ` +\n\t\t`target_domain_id: ?, ` +\n\t\t`target_domain_ids: ?,` +\n\t\t`target_workflow_id: ?, ` +\n\t\t`target_run_id: ?, ` +\n\t\t`target_child_workflow_only: ?, ` +\n\t\t`task_list: ?, ` +\n\t\t`type: ?, ` +\n\t\t`schedule_id: ?, ` +\n\t\t`record_visibility: ?, ` +\n\t\t`version: ?, ` +\n\t\t`original_task_list: ?, ` +\n\t\t`original_task_list_kind: ?` +\n\t\t`}`\n\n\ttemplateCrossClusterTaskType = templateTransferTaskType\n\n\ttemplateReplicationTaskType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`workflow_id: ?, ` +\n\t\t`run_id: ?, ` +\n\t\t`task_id: ?, ` +\n\t\t`type: ?, ` +\n\t\t`first_event_id: ?,` +\n\t\t`next_event_id: ?,` +\n\t\t`version: ?,` +\n\t\t`scheduled_id: ?, ` +\n\t\t`event_store_version: ?, ` +\n\t\t`branch_token: ?, ` +\n\t\t`new_run_event_store_version: ?, ` +\n\t\t`new_run_branch_token: ?, ` +\n\t\t`created_time: ? ` +\n\t\t`}`\n\n\ttemplateTimerTaskType = `{` +\n\t\t`domain_id: ?, ` +\n\t\t`workflow_id: ?, ` +\n\t\t`run_id: ?, ` +\n\t\t`visibility_ts: ?, ` +\n\t\t`task_id: ?, ` +\n\t\t`type: ?, ` +\n\t\t`timeout_type: ?, ` +\n\t\t`event_id: ?, ` +\n\t\t`schedule_attempt: ?, ` +\n\t\t`version: ?, ` +\n\t\t`task_list: ?` +\n\t\t`}`\n\n\ttemplateActivityInfoType = `{` +\n\t\t`version: ?, ` +\n\t\t`schedule_id: ?, ` +\n\t\t`scheduled_event_batch_id: ?, ` +\n\t\t`scheduled_event: ?, ` +\n\t\t`scheduled_time: ?, ` +\n\t\t`started_id: ?, ` +\n\t\t`started_event: ?, ` +\n\t\t`started_time: ?, ` +\n\t\t`activity_id: ?, ` +\n\t\t`request_id: ?, ` +\n\t\t`details: ?, ` +\n\t\t`schedule_to_start_timeout: ?, ` +\n\t\t`schedule_to_close_timeout: ?, ` +\n\t\t`start_to_close_timeout: ?, ` +\n\t\t`heart_beat_timeout: ?, ` +\n\t\t`cancel_requested: ?, ` +\n\t\t`cancel_request_id: ?, ` +\n\t\t`last_hb_updated_time: ?, ` +\n\t\t`timer_task_status: ?, ` +\n\t\t`attempt: ?, ` +\n\t\t`task_list: ?, ` +\n\t\t`task_list_kind: ?, ` +\n\t\t`started_identity: ?, ` +\n\t\t`has_retry_policy: ?, ` +\n\t\t`init_interval: ?, ` +\n\t\t`backoff_coefficient: ?, ` +\n\t\t`max_interval: ?, ` +\n\t\t`expiration_time: ?, ` +\n\t\t`max_attempts: ?, ` +\n\t\t`non_retriable_errors: ?, ` +\n\t\t`last_failure_reason: ?, ` +\n\t\t`last_worker_identity: ?, ` +\n\t\t`last_failure_details: ?, ` +\n\t\t`event_data_encoding: ?` +\n\t\t`}`\n\n\ttemplateTimerInfoType = `{` +\n\t\t`version: ?, ` +\n\t\t`timer_id: ?, ` +\n\t\t`started_id: ?, ` +\n\t\t`expiry_time: ?, ` +\n\t\t`task_id: ?` +\n\t\t`}`\n\n\ttemplateChildExecutionInfoType = `{` +\n\t\t`version: ?, ` +\n\t\t`initiated_id: ?, ` +\n\t\t`initiated_event_batch_id: ?, ` +\n\t\t`initiated_event: ?, ` +\n\t\t`started_id: ?, ` +\n\t\t`started_workflow_id: ?, ` +\n\t\t`started_run_id: ?, ` +\n\t\t`started_event: ?, ` +\n\t\t`create_request_id: ?, ` +\n\t\t`event_data_encoding: ?, ` +\n\t\t`domain_id: ?, ` +\n\t\t`domain_name: ?, ` +\n\t\t`workflow_type_name: ?, ` +\n\t\t`parent_close_policy: ?` +\n\t\t`}`\n\n\ttemplateRequestCancelInfoType = `{` +\n\t\t`version: ?,` +\n\t\t`initiated_id: ?, ` +\n\t\t`initiated_event_batch_id: ?, ` +\n\t\t`cancel_request_id: ? ` +\n\t\t`}`\n\n\ttemplateSignalInfoType = `{` +\n\t\t`version: ?, ` +\n\t\t`initiated_id: ?, ` +\n\t\t`initiated_event_batch_id: ?, ` +\n\t\t`signal_request_id: ?, ` +\n\t\t`signal_name: ?, ` +\n\t\t`input: ?, ` +\n\t\t`control: ?` +\n\t\t`}`\n\n\ttemplateChecksumType = `{` +\n\t\t`version: ?, ` +\n\t\t`flavor: ?, ` +\n\t\t`value: ? ` +\n\t\t`}`\n\n\ttemplateUpdateCurrentWorkflowExecutionQuery = `UPDATE executions USING TTL 0 ` +\n\t\t`SET current_run_id = ?, ` +\n\t\t`execution = {run_id: ?, create_request_id: ?, state: ?, close_status: ?}, ` +\n\t\t`workflow_last_write_version = ?, ` +\n\t\t`workflow_state = ?, ` +\n\t\t`last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF current_run_id = ? `\n\n\ttemplateUpdateCurrentWorkflowExecutionForNewQuery = templateUpdateCurrentWorkflowExecutionQuery +\n\t\t`and workflow_last_write_version = ? ` +\n\t\t`and workflow_state = ? `\n\n\ttemplateInsertWorkflowRequestQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?) IF NOT EXISTS USING TTL ?`\n\n\ttemplateUpsertWorkflowRequestQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, last_updated_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?) USING TTL ?`\n\n\ttemplateInsertWorkflowActiveClusterSelectionPolicyRowQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, created_time, data, data_encoding) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?) IF NOT EXISTS`\n\n\ttemplateGetActiveClusterSelectionPolicyQuery = `SELECT data, data_encoding ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateGetLatestWorkflowRequestQuery = `SELECT current_run_id ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`LIMIT 1`\n\n\ttemplateCreateCurrentWorkflowExecutionQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, execution, workflow_last_write_version, workflow_state, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ?, ?, ?, {run_id: ?, create_request_id: ?, state: ?, close_status: ?}, ?, ?, ?) IF NOT EXISTS USING TTL 0 `\n\n\ttemplateCreateWorkflowExecutionWithVersionHistoriesQuery = `INSERT INTO executions (` +\n\t\t`shard_id, domain_id, workflow_id, run_id, type, execution, next_event_id, visibility_ts, task_id, version_histories, version_histories_encoding, checksum, workflow_last_write_version, workflow_state, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateWorkflowExecutionType + `, ?, ?, ?, ?, ?, ` + templateChecksumType + `, ?, ?, ?) IF NOT EXISTS `\n\n\ttemplateCreateTransferTaskQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, transfer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateTransferTaskType + `, ?, ?, ?, ?, ?)`\n\n\ttemplateCreateReplicationTaskQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, replication, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateReplicationTaskType + `, ?, ?, ?, ?, ?)`\n\n\ttemplateCreateTimerTaskQuery = `INSERT INTO executions (` +\n\t\t`shard_id, type, domain_id, workflow_id, run_id, timer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t`VALUES(?, ?, ?, ?, ?, ` + templateTimerTaskType + `, ?, ?, ?, ?, ?)`\n\n\ttemplateUpdateLeaseQuery = `UPDATE executions ` +\n\t\t`SET range_id = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF range_id = ?`\n\n\t// TODO: remove replication_state after all 2DC workflows complete\n\ttemplateGetWorkflowExecutionQuery = `SELECT execution, replication_state, activity_map, timer_map, ` +\n\t\t`child_executions_map, request_cancel_map, signal_map, signal_requested, buffered_events_list, ` +\n\t\t`buffered_replication_tasks_map, version_histories, version_histories_encoding, checksum, ` +\n\t\t`next_event_id ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateGetCurrentExecutionQuery = `SELECT current_run_id, execution, workflow_last_write_version ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateListCurrentExecutionsQuery = `SELECT domain_id, workflow_id, run_id, current_run_id, workflow_state ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ?`\n\n\ttemplateIsWorkflowExecutionExistsQuery = `SELECT shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateListWorkflowExecutionQuery = `SELECT run_id, execution, version_histories, version_histories_encoding ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ?`\n\n\ttemplateUpdateWorkflowExecutionWithVersionHistoriesQuery = `UPDATE executions ` +\n\t\t`SET execution = ` + templateWorkflowExecutionType +\n\t\t`, next_event_id = ? ` +\n\t\t`, version_histories = ? ` +\n\t\t`, version_histories_encoding = ? ` +\n\t\t`, checksum = ` + templateChecksumType +\n\t\t`, workflow_last_write_version = ? ` +\n\t\t`, workflow_state = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? ` +\n\t\t`IF next_event_id = ? `\n\n\ttemplateUpdateActivityInfoQuery = `UPDATE executions ` +\n\t\t`SET activity_map[ ? ] = ` + templateActivityInfoType + ` ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetActivityInfoQuery = `UPDATE executions ` +\n\t\t`SET activity_map = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateUpdateTimerInfoQuery = `UPDATE executions ` +\n\t\t`SET timer_map[ ? ] = ` + templateTimerInfoType + ` ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetTimerInfoQuery = `UPDATE executions ` +\n\t\t`SET timer_map = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateUpdateChildExecutionInfoQuery = `UPDATE executions ` +\n\t\t`SET child_executions_map[ ? ] = ` + templateChildExecutionInfoType + ` ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetChildExecutionInfoQuery = `UPDATE executions ` +\n\t\t`SET child_executions_map = ?` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateUpdateRequestCancelInfoQuery = `UPDATE executions ` +\n\t\t`SET request_cancel_map[ ? ] = ` + templateRequestCancelInfoType + ` ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetRequestCancelInfoQuery = `UPDATE executions ` +\n\t\t`SET request_cancel_map = ?` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateUpdateSignalInfoQuery = `UPDATE executions ` +\n\t\t`SET signal_map[ ? ] = ` + templateSignalInfoType + ` ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetSignalInfoQuery = `UPDATE executions ` +\n\t\t`SET signal_map = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateUpdateSignalRequestedQuery = `UPDATE executions ` +\n\t\t`SET signal_requested = signal_requested + ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateResetSignalRequestedQuery = `UPDATE executions ` +\n\t\t`SET signal_requested = ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateAppendBufferedEventsQuery = `UPDATE executions ` +\n\t\t`SET buffered_events_list = buffered_events_list + ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteBufferedEventsQuery = `UPDATE executions ` +\n\t\t`SET buffered_events_list = [] ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteActivityInfoQuery = `DELETE activity_map[ ? ] ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteTimerInfoQuery = `DELETE timer_map[ ? ] ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteChildExecutionInfoQuery = `DELETE child_executions_map[ ? ] ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteRequestCancelInfoQuery = `DELETE request_cancel_map[ ? ] ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteSignalInfoQuery = `DELETE signal_map[ ? ] ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteWorkflowExecutionMutableStateQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteWorkflowExecutionCurrentRowQuery = templateDeleteWorkflowExecutionMutableStateQuery + \" if current_run_id = ? \"\n\n\ttemplateDeleteWorkflowExecutionSignalRequestedQuery = `UPDATE executions ` +\n\t\t`SET signal_requested = signal_requested - ? ` +\n\t\t`, last_updated_time = ? ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ? `\n\n\ttemplateDeleteActiveClusterSelectionPolicyQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateGetTransferTasksQuery = `SELECT task_id, transfer, data, data_encoding ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id >= ? ` +\n\t\t`and task_id < ?`\n\n\ttemplateGetReplicationTasksQuery = `SELECT task_id, replication, data, data_encoding ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id >= ? ` +\n\t\t`and task_id < ?`\n\n\ttemplateGetDLQSizeQuery = `SELECT count(1) as count ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ?`\n\n\ttemplateCompleteTransferTaskQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateRangeCompleteTransferTaskQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id >= ? ` +\n\t\t`and task_id < ?`\n\n\ttemplateCompleteCrossClusterTaskQuery = templateCompleteTransferTaskQuery\n\n\ttemplateCompleteReplicationTaskBeforeQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ? ` +\n\t\t`and run_id = ? ` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id < ?`\n\n\ttemplateCompleteReplicationTaskQuery = templateCompleteTransferTaskQuery\n\n\ttemplateRangeCompleteReplicationTaskQuery = templateRangeCompleteTransferTaskQuery\n\n\ttemplateGetTimerTasksQuery = `SELECT visibility_ts, task_id, timer, data, data_encoding ` +\n\t\t`FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ?` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ?` +\n\t\t`and run_id = ?` +\n\t\t`and visibility_ts >= ? ` +\n\t\t`and visibility_ts < ?`\n\n\ttemplateCompleteTimerTaskQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ?` +\n\t\t`and run_id = ?` +\n\t\t`and visibility_ts = ? ` +\n\t\t`and task_id = ?`\n\n\ttemplateRangeCompleteTimerTaskQuery = `DELETE FROM executions ` +\n\t\t`WHERE shard_id = ? ` +\n\t\t`and type = ? ` +\n\t\t`and domain_id = ? ` +\n\t\t`and workflow_id = ?` +\n\t\t`and run_id = ?` +\n\t\t`and visibility_ts >= ? ` +\n\t\t`and visibility_ts < ?`\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_parsing_utils.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\tcql \"github.com/gocql/gocql\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _emptyUUID = cql.UUID{}\n\nfunc parseWorkflowExecutionInfo(result map[string]interface{}) (*persistence.InternalWorkflowExecutionInfo, error) {\n\texecutionBlob, ok := result[\"execution\"].(map[string]interface{})\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"invalid execution blob format: missing or invalid 'execution' field\")\n\t}\n\tinfo := &persistence.InternalWorkflowExecutionInfo{}\n\tvar completionEventData []byte\n\tvar completionEventEncoding constants.EncodingType\n\tvar autoResetPoints []byte\n\tvar autoResetPointsEncoding constants.EncodingType\n\tvar activeClusterSelectionPolicy []byte\n\tvar activeClusterSelectionPolicyEncoding constants.EncodingType\n\n\tfor k, v := range executionBlob {\n\t\tswitch k {\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\tcase \"workflow_id\":\n\t\t\tinfo.WorkflowID = v.(string)\n\t\tcase \"run_id\":\n\t\t\tinfo.RunID = v.(gocql.UUID).String()\n\t\tcase \"first_run_id\":\n\t\t\tinfo.FirstExecutionRunID = v.(gocql.UUID).String()\n\t\t\tif info.FirstExecutionRunID == _emptyUUID.String() {\n\t\t\t\t// for backward compatibility, the gocql library doesn't handle the null uuid correectly https://github.com/gocql/gocql/blob/master/marshal.go#L1807\n\t\t\t\tinfo.FirstExecutionRunID = \"\"\n\t\t\t} else if info.FirstExecutionRunID == emptyRunID {\n\t\t\t\tinfo.FirstExecutionRunID = \"\"\n\t\t\t}\n\t\tcase \"parent_domain_id\":\n\t\t\tinfo.ParentDomainID = v.(gocql.UUID).String()\n\t\t\tif info.ParentDomainID == emptyDomainID {\n\t\t\t\tinfo.ParentDomainID = \"\"\n\t\t\t}\n\t\tcase \"parent_workflow_id\":\n\t\t\tinfo.ParentWorkflowID = v.(string)\n\t\tcase \"parent_run_id\":\n\t\t\tinfo.ParentRunID = v.(gocql.UUID).String()\n\t\t\tif info.ParentRunID == emptyRunID {\n\t\t\t\tinfo.ParentRunID = \"\"\n\t\t\t}\n\t\tcase \"initiated_id\":\n\t\t\tinfo.InitiatedID = v.(int64)\n\t\tcase \"completion_event_batch_id\":\n\t\t\tinfo.CompletionEventBatchID = v.(int64)\n\t\tcase \"completion_event\":\n\t\t\tcompletionEventData = v.([]byte)\n\t\tcase \"completion_event_data_encoding\":\n\t\t\tcompletionEventEncoding = constants.EncodingType(v.(string))\n\t\tcase \"auto_reset_points\":\n\t\t\tautoResetPoints = v.([]byte)\n\t\tcase \"auto_reset_points_encoding\":\n\t\t\tautoResetPointsEncoding = constants.EncodingType(v.(string))\n\t\tcase \"task_list\":\n\t\t\tinfo.TaskList = v.(string)\n\t\tcase \"task_list_kind\":\n\t\t\tinfo.TaskListKind = types.TaskListKind(int32(v.(int)))\n\t\tcase \"workflow_type_name\":\n\t\t\tinfo.WorkflowTypeName = v.(string)\n\t\tcase \"workflow_timeout\":\n\t\t\tinfo.WorkflowTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"decision_task_timeout\":\n\t\t\tinfo.DecisionStartToCloseTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"execution_context\":\n\t\t\tinfo.ExecutionContext = v.([]byte)\n\t\tcase \"state\":\n\t\t\tinfo.State = v.(int)\n\t\tcase \"close_status\":\n\t\t\tinfo.CloseStatus = v.(int)\n\t\tcase \"last_first_event_id\":\n\t\t\tinfo.LastFirstEventID = v.(int64)\n\t\tcase \"last_event_task_id\":\n\t\t\tinfo.LastEventTaskID = v.(int64)\n\t\tcase \"last_processed_event\":\n\t\t\tinfo.LastProcessedEvent = v.(int64)\n\t\tcase \"start_time\":\n\t\t\tinfo.StartTimestamp = v.(time.Time)\n\t\tcase \"last_updated_time\":\n\t\t\tinfo.LastUpdatedTimestamp = v.(time.Time)\n\t\tcase \"create_request_id\":\n\t\t\tinfo.CreateRequestID = v.(gocql.UUID).String()\n\t\tcase \"signal_count\":\n\t\t\tinfo.SignalCount = int32(v.(int))\n\t\tcase \"history_size\":\n\t\t\tinfo.HistorySize = v.(int64)\n\t\tcase \"decision_version\":\n\t\t\tinfo.DecisionVersion = v.(int64)\n\t\tcase \"decision_schedule_id\":\n\t\t\tinfo.DecisionScheduleID = v.(int64)\n\t\tcase \"decision_started_id\":\n\t\t\tinfo.DecisionStartedID = v.(int64)\n\t\tcase \"decision_request_id\":\n\t\t\tinfo.DecisionRequestID = v.(string)\n\t\tcase \"decision_timeout\":\n\t\t\tinfo.DecisionTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"decision_attempt\":\n\t\t\tinfo.DecisionAttempt = v.(int64)\n\t\tcase \"decision_timestamp\":\n\t\t\tinfo.DecisionStartedTimestamp = time.Unix(0, v.(int64))\n\t\tcase \"decision_scheduled_timestamp\":\n\t\t\tinfo.DecisionScheduledTimestamp = time.Unix(0, v.(int64))\n\t\tcase \"decision_original_scheduled_timestamp\":\n\t\t\tinfo.DecisionOriginalScheduledTimestamp = time.Unix(0, v.(int64))\n\t\tcase \"cancel_requested\":\n\t\t\tinfo.CancelRequested = v.(bool)\n\t\tcase \"cancel_request_id\":\n\t\t\tinfo.CancelRequestID = v.(string)\n\t\tcase \"sticky_task_list\":\n\t\t\tinfo.StickyTaskList = v.(string)\n\t\tcase \"sticky_schedule_to_start_timeout\":\n\t\t\tinfo.StickyScheduleToStartTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"client_library_version\":\n\t\t\tinfo.ClientLibraryVersion = v.(string)\n\t\tcase \"client_feature_version\":\n\t\t\tinfo.ClientFeatureVersion = v.(string)\n\t\tcase \"client_impl\":\n\t\t\tinfo.ClientImpl = v.(string)\n\t\tcase \"attempt\":\n\t\t\tinfo.Attempt = int32(v.(int))\n\t\tcase \"has_retry_policy\":\n\t\t\tinfo.HasRetryPolicy = v.(bool)\n\t\tcase \"init_interval\":\n\t\t\tinfo.InitialInterval = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"backoff_coefficient\":\n\t\t\tinfo.BackoffCoefficient = v.(float64)\n\t\tcase \"max_interval\":\n\t\t\tinfo.MaximumInterval = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"max_attempts\":\n\t\t\tinfo.MaximumAttempts = int32(v.(int))\n\t\tcase \"expiration_time\":\n\t\t\tinfo.ExpirationTime = v.(time.Time)\n\t\tcase \"non_retriable_errors\":\n\t\t\tinfo.NonRetriableErrors = v.([]string)\n\t\tcase \"branch_token\":\n\t\t\tinfo.BranchToken = v.([]byte)\n\t\tcase \"cron_schedule\":\n\t\t\tinfo.CronSchedule = v.(string)\n\t\tcase \"expiration_seconds\":\n\t\t\tinfo.ExpirationInterval = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"search_attributes\":\n\t\t\tinfo.SearchAttributes = v.(map[string][]byte)\n\t\tcase \"memo\":\n\t\t\tinfo.Memo = v.(map[string][]byte)\n\t\tcase \"partition_config\":\n\t\t\tinfo.PartitionConfig = v.(map[string]string)\n\t\tcase \"active_cluster_selection_policy\":\n\t\t\tactiveClusterSelectionPolicy = v.([]byte)\n\t\tcase \"active_cluster_selection_policy_encoding\":\n\t\t\tactiveClusterSelectionPolicyEncoding = constants.EncodingType(v.(string))\n\t\tcase \"cron_overlap_policy\":\n\t\t\tinfo.CronOverlapPolicy = types.CronOverlapPolicy(int32(v.(int)))\n\t\t}\n\t}\n\tinfo.CompletionEvent = persistence.NewDataBlob(completionEventData, completionEventEncoding)\n\tinfo.AutoResetPoints = persistence.NewDataBlob(autoResetPoints, autoResetPointsEncoding)\n\tinfo.ActiveClusterSelectionPolicy = persistence.NewDataBlob(activeClusterSelectionPolicy, activeClusterSelectionPolicyEncoding)\n\n\tif nextEventID, ok := result[\"next_event_id\"].(int64); ok {\n\t\tinfo.NextEventID = nextEventID\n\t}\n\n\treturn info, nil\n}\n\n// TODO: remove this after all 2DC workflows complete\nfunc parseReplicationState(\n\tresult map[string]interface{},\n) *persistence.ReplicationState {\n\n\tif len(result) == 0 {\n\t\treturn nil\n\t}\n\n\tinfo := &persistence.ReplicationState{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"current_version\":\n\t\t\tinfo.CurrentVersion = v.(int64)\n\t\tcase \"start_version\":\n\t\t\tinfo.StartVersion = v.(int64)\n\t\tcase \"last_write_version\":\n\t\t\tinfo.LastWriteVersion = v.(int64)\n\t\tcase \"last_write_event_id\":\n\t\t\tinfo.LastWriteEventID = v.(int64)\n\t\tcase \"last_replication_info\":\n\t\t\tinfo.LastReplicationInfo = make(map[string]*persistence.ReplicationInfo)\n\t\t\treplicationInfoMap := v.(map[string]map[string]interface{})\n\t\t\tfor key, value := range replicationInfoMap {\n\t\t\t\tinfo.LastReplicationInfo[key] = parseReplicationInfo(value)\n\t\t\t}\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseReplicationInfo(\n\tresult map[string]interface{},\n) *persistence.ReplicationInfo {\n\n\tinfo := &persistence.ReplicationInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"last_event_id\":\n\t\t\tinfo.LastEventID = v.(int64)\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseActivityInfo(\n\tdomainID string,\n\tresult map[string]interface{},\n) *persistence.InternalActivityInfo {\n\n\tinfo := &persistence.InternalActivityInfo{}\n\tvar sharedEncoding constants.EncodingType\n\tvar scheduledEventData, startedEventData []byte\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"schedule_id\":\n\t\t\tinfo.ScheduleID = v.(int64)\n\t\tcase \"scheduled_event_batch_id\":\n\t\t\tinfo.ScheduledEventBatchID = v.(int64)\n\t\tcase \"scheduled_event\":\n\t\t\tscheduledEventData = v.([]byte)\n\t\tcase \"scheduled_time\":\n\t\t\tinfo.ScheduledTime = v.(time.Time)\n\t\tcase \"started_id\":\n\t\t\tinfo.StartedID = v.(int64)\n\t\tcase \"started_event\":\n\t\t\tstartedEventData = v.([]byte)\n\t\tcase \"started_time\":\n\t\t\tinfo.StartedTime = v.(time.Time)\n\t\tcase \"activity_id\":\n\t\t\tinfo.ActivityID = v.(string)\n\t\tcase \"request_id\":\n\t\t\tinfo.RequestID = v.(string)\n\t\tcase \"details\":\n\t\t\tinfo.Details = v.([]byte)\n\t\tcase \"schedule_to_start_timeout\":\n\t\t\tinfo.ScheduleToStartTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"schedule_to_close_timeout\":\n\t\t\tinfo.ScheduleToCloseTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"start_to_close_timeout\":\n\t\t\tinfo.StartToCloseTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"heart_beat_timeout\":\n\t\t\tinfo.HeartbeatTimeout = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"cancel_requested\":\n\t\t\tinfo.CancelRequested = v.(bool)\n\t\tcase \"cancel_request_id\":\n\t\t\tinfo.CancelRequestID = v.(int64)\n\t\tcase \"last_hb_updated_time\":\n\t\t\tinfo.LastHeartBeatUpdatedTime = v.(time.Time)\n\t\tcase \"timer_task_status\":\n\t\t\tinfo.TimerTaskStatus = int32(v.(int))\n\t\tcase \"attempt\":\n\t\t\tinfo.Attempt = int32(v.(int))\n\t\tcase \"task_list\":\n\t\t\tinfo.TaskList = v.(string)\n\t\tcase \"task_list_kind\":\n\t\t\tinfo.TaskListKind = types.TaskListKind(int32(v.(int)))\n\t\tcase \"started_identity\":\n\t\t\tinfo.StartedIdentity = v.(string)\n\t\tcase \"has_retry_policy\":\n\t\t\tinfo.HasRetryPolicy = v.(bool)\n\t\tcase \"init_interval\":\n\t\t\tinfo.InitialInterval = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"backoff_coefficient\":\n\t\t\tinfo.BackoffCoefficient = v.(float64)\n\t\tcase \"max_interval\":\n\t\t\tinfo.MaximumInterval = common.SecondsToDuration(int64(v.(int)))\n\t\tcase \"max_attempts\":\n\t\t\tinfo.MaximumAttempts = (int32)(v.(int))\n\t\tcase \"expiration_time\":\n\t\t\tinfo.ExpirationTime = v.(time.Time)\n\t\tcase \"non_retriable_errors\":\n\t\t\tinfo.NonRetriableErrors = v.([]string)\n\t\tcase \"last_failure_reason\":\n\t\t\tinfo.LastFailureReason = v.(string)\n\t\tcase \"last_worker_identity\":\n\t\t\tinfo.LastWorkerIdentity = v.(string)\n\t\tcase \"last_failure_details\":\n\t\t\tinfo.LastFailureDetails = v.([]byte)\n\t\tcase \"event_data_encoding\":\n\t\t\tsharedEncoding = constants.EncodingType(v.(string))\n\t\t}\n\t}\n\tinfo.DomainID = domainID\n\tinfo.ScheduledEvent = persistence.NewDataBlob(scheduledEventData, sharedEncoding)\n\tinfo.StartedEvent = persistence.NewDataBlob(startedEventData, sharedEncoding)\n\n\treturn info\n}\n\nfunc parseTimerInfo(\n\tresult map[string]interface{},\n) *persistence.TimerInfo {\n\n\tinfo := &persistence.TimerInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"timer_id\":\n\t\t\tinfo.TimerID = v.(string)\n\t\tcase \"started_id\":\n\t\t\tinfo.StartedID = v.(int64)\n\t\tcase \"expiry_time\":\n\t\t\tinfo.ExpiryTime = v.(time.Time)\n\t\tcase \"task_id\":\n\t\t\t// task_id is a misleading variable, it actually serves\n\t\t\t// the purpose of indicating whether a timer task is\n\t\t\t// generated for this timer info\n\t\t\tinfo.TaskStatus = v.(int64)\n\t\t}\n\t}\n\treturn info\n}\n\nfunc parseChildExecutionInfo(\n\tresult map[string]interface{},\n) *persistence.InternalChildExecutionInfo {\n\n\tinfo := &persistence.InternalChildExecutionInfo{}\n\tvar encoding constants.EncodingType\n\tvar initiatedData []byte\n\tvar startedData []byte\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"initiated_id\":\n\t\t\tinfo.InitiatedID = v.(int64)\n\t\tcase \"initiated_event_batch_id\":\n\t\t\tinfo.InitiatedEventBatchID = v.(int64)\n\t\tcase \"initiated_event\":\n\t\t\tinitiatedData = v.([]byte)\n\t\tcase \"started_id\":\n\t\t\tinfo.StartedID = v.(int64)\n\t\tcase \"started_workflow_id\":\n\t\t\tinfo.StartedWorkflowID = v.(string)\n\t\tcase \"started_run_id\":\n\t\t\tinfo.StartedRunID = v.(gocql.UUID).String()\n\t\tcase \"started_event\":\n\t\t\tstartedData = v.([]byte)\n\t\tcase \"create_request_id\":\n\t\t\tinfo.CreateRequestID = v.(gocql.UUID).String()\n\t\tcase \"event_data_encoding\":\n\t\t\tencoding = constants.EncodingType(v.(string))\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\t\tif info.DomainID == _emptyUUID.String() {\n\t\t\t\t// for backward compatibility, the gocql library doesn't handle the null uuid correectly https://github.com/gocql/gocql/blob/master/marshal.go#L1807\n\t\t\t\tinfo.DomainID = \"\"\n\t\t\t}\n\t\tcase \"domain_name\":\n\t\t\tinfo.DomainNameDEPRECATED = v.(string)\n\t\tcase \"workflow_type_name\":\n\t\t\tinfo.WorkflowTypeName = v.(string)\n\t\tcase \"parent_close_policy\":\n\t\t\tinfo.ParentClosePolicy = types.ParentClosePolicy(v.(int))\n\t\t}\n\t}\n\tinfo.InitiatedEvent = persistence.NewDataBlob(initiatedData, encoding)\n\tinfo.StartedEvent = persistence.NewDataBlob(startedData, encoding)\n\treturn info\n}\n\nfunc parseRequestCancelInfo(\n\tresult map[string]interface{},\n) *persistence.RequestCancelInfo {\n\n\tinfo := &persistence.RequestCancelInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"initiated_id\":\n\t\t\tinfo.InitiatedID = v.(int64)\n\t\tcase \"initiated_event_batch_id\":\n\t\t\tinfo.InitiatedEventBatchID = v.(int64)\n\t\tcase \"cancel_request_id\":\n\t\t\tinfo.CancelRequestID = v.(string)\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseSignalInfo(\n\tresult map[string]interface{},\n) *persistence.SignalInfo {\n\n\tinfo := &persistence.SignalInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"initiated_id\":\n\t\t\tinfo.InitiatedID = v.(int64)\n\t\tcase \"initiated_event_batch_id\":\n\t\t\tinfo.InitiatedEventBatchID = v.(int64)\n\t\tcase \"signal_request_id\":\n\t\t\tinfo.SignalRequestID = v.(gocql.UUID).String()\n\t\tcase \"signal_name\":\n\t\t\tinfo.SignalName = v.(string)\n\t\tcase \"input\":\n\t\t\tinfo.Input = v.([]byte)\n\t\tcase \"control\":\n\t\t\tinfo.Control = v.([]byte)\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseHistoryEventBatchBlob(\n\tresult map[string]interface{},\n) *persistence.DataBlob {\n\n\teventBatch := &persistence.DataBlob{Encoding: constants.EncodingTypeJSON}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"encoding_type\":\n\t\t\teventBatch.Encoding = constants.EncodingType(v.(string))\n\t\tcase \"data\":\n\t\t\teventBatch.Data = v.([]byte)\n\t\t}\n\t}\n\n\treturn eventBatch\n}\n\nfunc parseTimerTaskInfo(\n\tresult map[string]interface{},\n) *persistence.TimerTaskInfo {\n\n\tinfo := &persistence.TimerTaskInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\tcase \"workflow_id\":\n\t\t\tinfo.WorkflowID = v.(string)\n\t\tcase \"run_id\":\n\t\t\tinfo.RunID = v.(gocql.UUID).String()\n\t\tcase \"visibility_ts\":\n\t\t\tinfo.VisibilityTimestamp = v.(time.Time)\n\t\tcase \"task_id\":\n\t\t\tinfo.TaskID = v.(int64)\n\t\tcase \"type\":\n\t\t\tinfo.TaskType = v.(int)\n\t\tcase \"timeout_type\":\n\t\t\tinfo.TimeoutType = v.(int)\n\t\tcase \"event_id\":\n\t\t\tinfo.EventID = v.(int64)\n\t\tcase \"schedule_attempt\":\n\t\t\tinfo.ScheduleAttempt = v.(int64)\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"task_list\":\n\t\t\tinfo.TaskList = v.(string)\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseTransferTaskInfo(\n\tresult map[string]interface{},\n) *persistence.TransferTaskInfo {\n\n\tinfo := &persistence.TransferTaskInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\tcase \"workflow_id\":\n\t\t\tinfo.WorkflowID = v.(string)\n\t\tcase \"run_id\":\n\t\t\tinfo.RunID = v.(gocql.UUID).String()\n\t\tcase \"visibility_ts\":\n\t\t\tinfo.VisibilityTimestamp = v.(time.Time)\n\t\tcase \"task_id\":\n\t\t\tinfo.TaskID = v.(int64)\n\t\tcase \"target_domain_id\":\n\t\t\tinfo.TargetDomainID = v.(gocql.UUID).String()\n\t\tcase \"target_domain_ids\":\n\t\t\ttargetDomainIDs := make(map[string]struct{})\n\t\t\tdList := mustConvertToSlice(result[\"target_domain_ids\"])\n\t\t\tfor _, v := range dList {\n\t\t\t\ttargetDomainIDs[v.(gocql.UUID).String()] = struct{}{}\n\t\t\t}\n\t\t\tinfo.TargetDomainIDs = targetDomainIDs\n\t\tcase \"target_workflow_id\":\n\t\t\tinfo.TargetWorkflowID = v.(string)\n\t\tcase \"target_run_id\":\n\t\t\tinfo.TargetRunID = v.(gocql.UUID).String()\n\t\t\tif info.TargetRunID == persistence.TransferTaskTransferTargetRunID {\n\t\t\t\tinfo.TargetRunID = \"\"\n\t\t\t}\n\t\tcase \"target_child_workflow_only\":\n\t\t\tinfo.TargetChildWorkflowOnly = v.(bool)\n\t\tcase \"task_list\":\n\t\t\tinfo.TaskList = v.(string)\n\t\tcase \"type\":\n\t\t\tinfo.TaskType = v.(int)\n\t\tcase \"schedule_id\":\n\t\t\tinfo.ScheduleID = v.(int64)\n\t\tcase \"record_visibility\":\n\t\t\tinfo.RecordVisibility = v.(bool)\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"original_task_list\":\n\t\t\tinfo.OriginalTaskList = v.(string)\n\t\tcase \"original_task_list_kind\":\n\t\t\tinfo.OriginalTaskListKind = types.TaskListKind(int32(v.(int)))\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseReplicationTaskInfo(\n\tresult map[string]interface{},\n) *nosqlplugin.ReplicationTask {\n\n\tinfo := &persistence.InternalReplicationTaskInfo{}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"domain_id\":\n\t\t\tinfo.DomainID = v.(gocql.UUID).String()\n\t\tcase \"workflow_id\":\n\t\t\tinfo.WorkflowID = v.(string)\n\t\tcase \"run_id\":\n\t\t\tinfo.RunID = v.(gocql.UUID).String()\n\t\tcase \"task_id\":\n\t\t\tinfo.TaskID = v.(int64)\n\t\tcase \"type\":\n\t\t\tinfo.TaskType = v.(int)\n\t\tcase \"first_event_id\":\n\t\t\tinfo.FirstEventID = v.(int64)\n\t\tcase \"next_event_id\":\n\t\t\tinfo.NextEventID = v.(int64)\n\t\tcase \"version\":\n\t\t\tinfo.Version = v.(int64)\n\t\tcase \"scheduled_id\":\n\t\t\tinfo.ScheduledID = v.(int64)\n\t\tcase \"branch_token\":\n\t\t\tinfo.BranchToken = v.([]byte)\n\t\tcase \"new_run_branch_token\":\n\t\t\tinfo.NewRunBranchToken = v.([]byte)\n\t\tcase \"created_time\":\n\t\t\tinfo.CreationTime = time.Unix(0, v.(int64))\n\t\t}\n\t}\n\n\treturn info\n}\n\nfunc parseChecksum(result map[string]interface{}) checksum.Checksum {\n\tcsum := checksum.Checksum{}\n\tif len(result) == 0 {\n\t\treturn csum\n\t}\n\tfor k, v := range result {\n\t\tswitch k {\n\t\tcase \"flavor\":\n\t\t\tcsum.Flavor = checksum.Flavor(v.(int))\n\t\tcase \"version\":\n\t\t\tcsum.Version = v.(int)\n\t\tcase \"value\":\n\t\t\tcsum.Value = v.([]byte)\n\t\t}\n\t}\n\treturn csum\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_parsing_utils_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tcql \"github.com/gocql/gocql\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype mockUUID struct {\n\tuuid string\n}\n\nfunc (m mockUUID) String() string {\n\treturn m.uuid\n}\n\nfunc newMockUUID(s string) mockUUID {\n\treturn mockUUID{s}\n}\n\nfunc Test_parseWorkflowExecutionInfo(t *testing.T) {\n\n\tcompletionEventData := []byte(\"completion event data\")\n\tautoResetPointsData := []byte(\"auto reset points data\")\n\tactiveClusterSelectionPolicyData := []byte(\"active cluster selection policy data\")\n\tsearchAttributes := map[string][]byte{\"AttributeKey\": []byte(\"AttributeValue\")}\n\tmemo := map[string][]byte{\"MemoKey\": []byte(\"MemoValue\")}\n\tpartitionConfig := map[string]string{\"PartitionKey\": \"PartitionValue\"}\n\ttimeNow := time.Now()\n\n\ttests := []struct {\n\t\tname string\n\t\targs map[string]interface{}\n\t\twant *persistence.InternalWorkflowExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"full execution blob\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"domain_id\":                                newMockUUID(\"domain_id\"),\n\t\t\t\t\t\"workflow_id\":                              \"workflow_id\",\n\t\t\t\t\t\"run_id\":                                   newMockUUID(\"run_id\"),\n\t\t\t\t\t\"parent_workflow_id\":                       \"parent_workflow_id\",\n\t\t\t\t\t\"initiated_id\":                             int64(1),\n\t\t\t\t\t\"completion_event_batch_id\":                int64(2),\n\t\t\t\t\t\"task_list\":                                \"task_list\",\n\t\t\t\t\t\"task_list_kind\":                           2,\n\t\t\t\t\t\"workflow_type_name\":                       \"workflow_type_name\",\n\t\t\t\t\t\"workflow_timeout\":                         10,\n\t\t\t\t\t\"decision_task_timeout\":                    5,\n\t\t\t\t\t\"execution_context\":                        []byte(\"execution context\"),\n\t\t\t\t\t\"state\":                                    1,\n\t\t\t\t\t\"close_status\":                             2,\n\t\t\t\t\t\"last_first_event_id\":                      int64(3),\n\t\t\t\t\t\"last_event_task_id\":                       int64(4),\n\t\t\t\t\t\"last_processed_event\":                     int64(6),\n\t\t\t\t\t\"start_time\":                               timeNow,\n\t\t\t\t\t\"last_updated_time\":                        timeNow,\n\t\t\t\t\t\"create_request_id\":                        newMockUUID(\"create_request_id\"),\n\t\t\t\t\t\"signal_count\":                             7,\n\t\t\t\t\t\"history_size\":                             int64(8),\n\t\t\t\t\t\"decision_version\":                         int64(9),\n\t\t\t\t\t\"decision_schedule_id\":                     int64(10),\n\t\t\t\t\t\"decision_started_id\":                      int64(11),\n\t\t\t\t\t\"decision_request_id\":                      \"decision_request_id\",\n\t\t\t\t\t\"decision_timeout\":                         8,\n\t\t\t\t\t\"decision_timestamp\":                       int64(200),\n\t\t\t\t\t\"decision_scheduled_timestamp\":             int64(201),\n\t\t\t\t\t\"decision_original_scheduled_timestamp\":    int64(202),\n\t\t\t\t\t\"decision_attempt\":                         int64(203),\n\t\t\t\t\t\"cancel_requested\":                         true,\n\t\t\t\t\t\"cancel_request_id\":                        \"cancel_request_id\",\n\t\t\t\t\t\"sticky_task_list\":                         \"sticky_task_list\",\n\t\t\t\t\t\"sticky_schedule_to_start_timeout\":         9,\n\t\t\t\t\t\"client_library_version\":                   \"client_lib_version\",\n\t\t\t\t\t\"client_feature_version\":                   \"client_feature_version\",\n\t\t\t\t\t\"client_impl\":                              \"client_impl\",\n\t\t\t\t\t\"attempt\":                                  12,\n\t\t\t\t\t\"has_retry_policy\":                         true,\n\t\t\t\t\t\"init_interval\":                            10,\n\t\t\t\t\t\"backoff_coefficient\":                      1.5,\n\t\t\t\t\t\"max_interval\":                             20,\n\t\t\t\t\t\"max_attempts\":                             13,\n\t\t\t\t\t\"expiration_time\":                          timeNow,\n\t\t\t\t\t\"non_retriable_errors\":                     []string{\"error1\", \"error2\"},\n\t\t\t\t\t\"branch_token\":                             []byte(\"branch token\"),\n\t\t\t\t\t\"cron_schedule\":                            \"cron_schedule\",\n\t\t\t\t\t\"expiration_seconds\":                       14,\n\t\t\t\t\t\"search_attributes\":                        searchAttributes,\n\t\t\t\t\t\"memo\":                                     memo,\n\t\t\t\t\t\"partition_config\":                         partitionConfig,\n\t\t\t\t\t\"completion_event\":                         completionEventData,\n\t\t\t\t\t\"completion_event_data_encoding\":           \"Proto3\",\n\t\t\t\t\t\"auto_reset_points\":                        autoResetPointsData,\n\t\t\t\t\t\"auto_reset_points_encoding\":               \"Proto3\",\n\t\t\t\t\t\"active_cluster_selection_policy\":          activeClusterSelectionPolicyData,\n\t\t\t\t\t\"active_cluster_selection_policy_encoding\": \"Proto3\",\n\t\t\t\t},\n\t\t\t\t\"next_event_id\": int64(5),\n\t\t\t},\n\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tDomainID:                           \"domain_id\",\n\t\t\t\tWorkflowID:                         \"workflow_id\",\n\t\t\t\tRunID:                              \"run_id\",\n\t\t\t\tParentWorkflowID:                   \"parent_workflow_id\",\n\t\t\t\tInitiatedID:                        int64(1),\n\t\t\t\tCompletionEventBatchID:             int64(2),\n\t\t\t\tTaskList:                           \"task_list\",\n\t\t\t\tTaskListKind:                       types.TaskListKindEphemeral,\n\t\t\t\tWorkflowTypeName:                   \"workflow_type_name\",\n\t\t\t\tWorkflowTimeout:                    common.SecondsToDuration(int64(10)),\n\t\t\t\tDecisionStartToCloseTimeout:        common.SecondsToDuration(int64(5)),\n\t\t\t\tExecutionContext:                   []byte(\"execution context\"),\n\t\t\t\tState:                              1,\n\t\t\t\tCloseStatus:                        2,\n\t\t\t\tLastFirstEventID:                   int64(3),\n\t\t\t\tLastEventTaskID:                    int64(4),\n\t\t\t\tNextEventID:                        int64(5),\n\t\t\t\tLastProcessedEvent:                 int64(6),\n\t\t\t\tStartTimestamp:                     timeNow,\n\t\t\t\tLastUpdatedTimestamp:               timeNow,\n\t\t\t\tCreateRequestID:                    \"create_request_id\",\n\t\t\t\tSignalCount:                        int32(7),\n\t\t\t\tHistorySize:                        int64(8),\n\t\t\t\tDecisionVersion:                    int64(9),\n\t\t\t\tDecisionScheduleID:                 int64(10),\n\t\t\t\tDecisionStartedID:                  int64(11),\n\t\t\t\tDecisionRequestID:                  \"decision_request_id\",\n\t\t\t\tDecisionTimeout:                    common.SecondsToDuration(int64(8)),\n\t\t\t\tDecisionStartedTimestamp:           time.Unix(0, int64(200)),\n\t\t\t\tDecisionScheduledTimestamp:         time.Unix(0, int64(201)),\n\t\t\t\tDecisionOriginalScheduledTimestamp: time.Unix(0, int64(202)),\n\t\t\t\tDecisionAttempt:                    int64(203),\n\t\t\t\tCancelRequested:                    true,\n\t\t\t\tCancelRequestID:                    \"cancel_request_id\",\n\t\t\t\tStickyTaskList:                     \"sticky_task_list\",\n\t\t\t\tStickyScheduleToStartTimeout:       common.SecondsToDuration(int64(9)),\n\t\t\t\tClientLibraryVersion:               \"client_lib_version\",\n\t\t\t\tClientFeatureVersion:               \"client_feature_version\",\n\t\t\t\tClientImpl:                         \"client_impl\",\n\t\t\t\tAttempt:                            int32(12),\n\t\t\t\tHasRetryPolicy:                     true,\n\t\t\t\tInitialInterval:                    common.SecondsToDuration(int64(10)),\n\t\t\t\tBackoffCoefficient:                 1.5,\n\t\t\t\tMaximumInterval:                    common.SecondsToDuration(int64(20)),\n\t\t\t\tMaximumAttempts:                    int32(13),\n\t\t\t\tExpirationTime:                     timeNow,\n\t\t\t\tNonRetriableErrors:                 []string{\"error1\", \"error2\"},\n\t\t\t\tMemo:                               memo,\n\t\t\t\tPartitionConfig:                    partitionConfig,\n\t\t\t\tActiveClusterSelectionPolicy:       persistence.NewDataBlob(activeClusterSelectionPolicyData, \"Proto3\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"uuid fields\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"first_run_id\":     newMockUUID(\"first_run_id\"),\n\t\t\t\t\t\"parent_domain_id\": newMockUUID(\"parent_domain_id\"),\n\t\t\t\t\t\"parent_run_id\":    newMockUUID(\"parent_run_id\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tFirstExecutionRunID: \"first_run_id\",\n\t\t\t\tParentDomainID:      \"parent_domain_id\",\n\t\t\t\tParentRunID:         \"parent_run_id\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty uuid fields\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"first_run_id\":     newMockUUID(emptyRunID),\n\t\t\t\t\t\"parent_domain_id\": newMockUUID(emptyDomainID),\n\t\t\t\t\t\"parent_run_id\":    newMockUUID(emptyRunID),\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil uuid field\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"first_run_id\": newMockUUID(cql.UUID{}.String()),\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tFirstExecutionRunID: \"\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"denormalized columns override blob - next_event_id\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"next_event_id\": int64(10),\n\t\t\t\t},\n\t\t\t\t\"next_event_id\": int64(5),\n\t\t\t},\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tNextEventID: int64(5),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"no denormalized columns - execution blob values ignored for next_event_id\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\"next_event_id\": int64(10), // This is ignored, only denormalized column is used\n\t\t\t\t\t\"state\":         2,\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\tNextEventID: int64(0), // Zero value since denormalized column not present\n\t\t\t\tState:       2,\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := parseWorkflowExecutionInfo(tt.args)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, result.FirstExecutionRunID, tt.want.FirstExecutionRunID)\n\t\t\tassert.Equal(t, result.DomainID, tt.want.DomainID)\n\t\t\tassert.Equal(t, result.WorkflowID, tt.want.WorkflowID)\n\t\t\tassert.Equal(t, result.RunID, tt.want.RunID)\n\t\t\tassert.Equal(t, result.ParentWorkflowID, tt.want.ParentWorkflowID)\n\t\t\tassert.Equal(t, result.InitiatedID, tt.want.InitiatedID)\n\t\t\tassert.Equal(t, result.CompletionEventBatchID, tt.want.CompletionEventBatchID)\n\t\t\tassert.Equal(t, result.TaskList, tt.want.TaskList)\n\t\t\tassert.Equal(t, result.WorkflowTypeName, tt.want.WorkflowTypeName)\n\t\t\tassert.Equal(t, result.WorkflowTimeout, tt.want.WorkflowTimeout)\n\t\t\tassert.Equal(t, result.DecisionStartToCloseTimeout, tt.want.DecisionStartToCloseTimeout)\n\t\t\tassert.Equal(t, result.ExecutionContext, tt.want.ExecutionContext)\n\t\t\tassert.Equal(t, result.State, tt.want.State)\n\t\t\tassert.Equal(t, result.CloseStatus, tt.want.CloseStatus)\n\t\t\tassert.Equal(t, result.LastFirstEventID, tt.want.LastFirstEventID)\n\t\t\tassert.Equal(t, result.LastEventTaskID, tt.want.LastEventTaskID)\n\t\t\tassert.Equal(t, result.NextEventID, tt.want.NextEventID)\n\t\t\tassert.Equal(t, result.LastProcessedEvent, tt.want.LastProcessedEvent)\n\t\t\tassert.Equal(t, result.StartTimestamp, tt.want.StartTimestamp)\n\t\t\tassert.Equal(t, result.LastUpdatedTimestamp, tt.want.LastUpdatedTimestamp)\n\t\t\tassert.Equal(t, result.CreateRequestID, tt.want.CreateRequestID)\n\t\t\tassert.Equal(t, result.SignalCount, tt.want.SignalCount)\n\t\t\tassert.Equal(t, result.HistorySize, tt.want.HistorySize)\n\t\t\tassert.Equal(t, result.DecisionVersion, tt.want.DecisionVersion)\n\t\t\tassert.Equal(t, result.DecisionScheduleID, tt.want.DecisionScheduleID)\n\t\t\tassert.Equal(t, result.DecisionStartedID, tt.want.DecisionStartedID)\n\t\t\tassert.Equal(t, result.DecisionRequestID, tt.want.DecisionRequestID)\n\t\t\tassert.Equal(t, result.DecisionTimeout, tt.want.DecisionTimeout)\n\t\t\tassert.Equal(t, result.CancelRequested, tt.want.CancelRequested)\n\t\t\tassert.Equal(t, result.DecisionStartedTimestamp, tt.want.DecisionStartedTimestamp)\n\t\t\tassert.Equal(t, result.DecisionScheduledTimestamp, tt.want.DecisionScheduledTimestamp)\n\t\t\tassert.Equal(t, result.DecisionOriginalScheduledTimestamp, tt.want.DecisionOriginalScheduledTimestamp)\n\t\t\tassert.Equal(t, result.DecisionAttempt, tt.want.DecisionAttempt)\n\t\t\tassert.Equal(t, result.ParentDomainID, tt.want.ParentDomainID)\n\t\t\tassert.Equal(t, result.ActiveClusterSelectionPolicy, tt.want.ActiveClusterSelectionPolicy)\n\t\t})\n\t}\n}\n\nfunc Test_parseWorkflowExecutionInfo_ErrorCases(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\targs        map[string]interface{}\n\t\twantErr     bool\n\t\terrContains string\n\t}{\n\t\t{\n\t\t\tname: \"missing execution field\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"next_event_id\": int64(5),\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\terrContains: \"missing or invalid 'execution' field\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid execution field type\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": \"not a map\",\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\terrContains: \"missing or invalid 'execution' field\",\n\t\t},\n\t\t{\n\t\t\tname: \"execution field is nil\",\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"execution\": nil,\n\t\t\t},\n\t\t\twantErr:     true,\n\t\t\terrContains: \"missing or invalid 'execution' field\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := parseWorkflowExecutionInfo(tt.args)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t\tassert.Contains(t, err.Error(), tt.errContains)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_parseReplicationState(t *testing.T) {\n\ttests := []struct {\n\t\targs map[string]interface{}\n\t\twant *persistence.ReplicationState\n\t}{\n\t\t{\n\t\t\targs: map[string]interface{}{\n\t\t\t\t\"current_version\":     int64(1),\n\t\t\t\t\"start_version\":       int64(2),\n\t\t\t\t\"last_write_version\":  int64(3),\n\t\t\t\t\"last_write_event_id\": int64(4),\n\t\t\t\t\"last_replication_info\": map[string]map[string]interface{}{\n\t\t\t\t\t\"map1\": {\n\t\t\t\t\t\t\"version\":       int64(5),\n\t\t\t\t\t\t\"last_event_id\": int64(6),\n\t\t\t\t\t},\n\t\t\t\t\t\"map2\": {\n\t\t\t\t\t\t\"version\":       int64(7),\n\t\t\t\t\t\t\"last_event_id\": int64(8),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &persistence.ReplicationState{\n\t\t\t\tCurrentVersion:   int64(1),\n\t\t\t\tStartVersion:     int64(2),\n\t\t\t\tLastWriteVersion: int64(3),\n\t\t\t\tLastWriteEventID: int64(4),\n\t\t\t\tLastReplicationInfo: map[string]*persistence.ReplicationInfo{\n\t\t\t\t\t\"map1\": {\n\t\t\t\t\t\tVersion:     int64(5),\n\t\t\t\t\t\tLastEventID: int64(6),\n\t\t\t\t\t},\n\t\t\t\t\t\"map2\": {\n\t\t\t\t\t\tVersion:     int64(7),\n\t\t\t\t\t\tLastEventID: int64(8),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tresult := parseReplicationState(tt.args)\n\t\tassert.Equal(t, result.CurrentVersion, tt.want.CurrentVersion)\n\t\tassert.Equal(t, result.StartVersion, tt.want.StartVersion)\n\t\tassert.Equal(t, result.LastWriteVersion, tt.want.LastWriteVersion)\n\t\tassert.Equal(t, result.LastWriteEventID, tt.want.LastWriteEventID)\n\t\tassert.Equal(t, result.LastReplicationInfo, tt.want.LastReplicationInfo)\n\t}\n}\n\nfunc Test_parseActivityInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttestInput := map[string]interface{}{\n\t\t\"version\":                   int64(1),\n\t\t\"schedule_id\":               int64(2),\n\t\t\"scheduled_event_batch_id\":  int64(3),\n\t\t\"scheduled_event\":           []byte(\"scheduled_event\"),\n\t\t\"scheduled_time\":            timeNow,\n\t\t\"started_id\":                int64(4),\n\t\t\"started_event\":             []byte(\"started_event\"),\n\t\t\"started_time\":              timeNow,\n\t\t\"activity_id\":               \"activity_id\",\n\t\t\"request_id\":                \"request_id\",\n\t\t\"details\":                   []byte(\"details\"),\n\t\t\"schedule_to_start_timeout\": 5,\n\t\t\"schedule_to_close_timeout\": 6,\n\t\t\"start_to_close_timeout\":    7,\n\t\t\"heart_beat_timeout\":        8,\n\t\t\"cancel_requested\":          true,\n\t\t\"cancel_request_id\":         int64(9),\n\t\t\"last_hb_updated_time\":      timeNow,\n\t\t\"timer_task_status\":         9,\n\t\t\"attempt\":                   10,\n\t\t\"task_list\":                 \"task_list\",\n\t\t\"task_list_kind\":            1,\n\t\t\"started_identity\":          \"started_identity\",\n\t\t\"has_retry_policy\":          true,\n\t\t\"init_interval\":             11,\n\t\t\"backoff_coefficient\":       1.5,\n\t\t\"max_interval\":              12,\n\t\t\"max_attempts\":              13,\n\t\t\"expiration_time\":           timeNow,\n\t\t\"non_retriable_errors\":      []string{\"error1\", \"error2\"},\n\t\t\"last_failure_reason\":       \"last_failure_reason\",\n\t\t\"last_worker_identity\":      \"last_worker_identity\",\n\t\t\"last_failure_details\":      []byte(\"last_failure_details\"),\n\t\t\"event_data_encoding\":       \"Proto3\",\n\t}\n\n\texpected := &persistence.InternalActivityInfo{\n\t\tVersion:                  int64(1),\n\t\tScheduleID:               int64(2),\n\t\tScheduledEventBatchID:    int64(3),\n\t\tScheduledEvent:           persistence.NewDataBlob([]byte(\"scheduled_event\"), \"Proto3\"),\n\t\tScheduledTime:            timeNow,\n\t\tStartedID:                int64(4),\n\t\tStartedEvent:             persistence.NewDataBlob([]byte(\"started_event\"), \"Proto3\"),\n\t\tStartedTime:              timeNow,\n\t\tActivityID:               \"activity_id\",\n\t\tRequestID:                \"request_id\",\n\t\tDetails:                  []byte(\"details\"),\n\t\tScheduleToStartTimeout:   common.SecondsToDuration(int64(5)),\n\t\tScheduleToCloseTimeout:   common.SecondsToDuration(int64(6)),\n\t\tStartToCloseTimeout:      common.SecondsToDuration(int64(7)),\n\t\tHeartbeatTimeout:         common.SecondsToDuration(int64(8)),\n\t\tCancelRequested:          true,\n\t\tCancelRequestID:          int64(9),\n\t\tLastHeartBeatUpdatedTime: timeNow,\n\t\tTimerTaskStatus:          int32(9),\n\t\tAttempt:                  int32(10),\n\t\tTaskList:                 \"task_list\",\n\t\tTaskListKind:             types.TaskListKindSticky,\n\t\tStartedIdentity:          \"started_identity\",\n\t\tHasRetryPolicy:           true,\n\t\tInitialInterval:          common.SecondsToDuration(int64(11)),\n\t\tBackoffCoefficient:       1.5,\n\t\tMaximumInterval:          common.SecondsToDuration(int64(12)),\n\t\tMaximumAttempts:          int32(13),\n\t\tExpirationTime:           timeNow,\n\t\tNonRetriableErrors:       []string{\"error1\", \"error2\"},\n\t\tLastFailureReason:        \"last_failure_reason\",\n\t\tLastWorkerIdentity:       \"last_worker_identity\",\n\t\tLastFailureDetails:       []byte(\"last_failure_details\"),\n\t\tDomainID:                 \"domain_id\",\n\t}\n\n\tassert.Equal(t, expected, parseActivityInfo(\"domain_id\", testInput))\n}\n\nfunc Test_parseTimerInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttestInput := map[string]interface{}{\n\t\t\"version\":     int64(1),\n\t\t\"timer_id\":    \"timer_id\",\n\t\t\"started_id\":  int64(2),\n\t\t\"expiry_time\": timeNow,\n\t\t\"task_id\":     int64(3),\n\t}\n\texpected := &persistence.TimerInfo{\n\t\tVersion:    int64(1),\n\t\tTimerID:    \"timer_id\",\n\t\tStartedID:  int64(2),\n\t\tExpiryTime: timeNow,\n\t\tTaskStatus: int64(3),\n\t}\n\tassert.Equal(t, expected, parseTimerInfo(testInput))\n}\n\nfunc Test_parseChildExecutionInfo(t *testing.T) {\n\tstartedRunID := newMockUUID(\"started_run_id\")\n\tcreateRequestID := newMockUUID(\"create_request_id\")\n\tdomainID := newMockUUID(\"domain_id\")\n\ttestInput := map[string]interface{}{\n\t\t\"version\":                  int64(1),\n\t\t\"initiated_id\":             int64(2),\n\t\t\"initiated_event_batch_id\": int64(3),\n\t\t\"initiated_event\":          []byte(\"initiated_event\"),\n\t\t\"started_id\":               int64(4),\n\t\t\"started_workflow_id\":      \"started_workflow_id\",\n\t\t\"started_run_id\":           startedRunID,\n\t\t\"started_event\":            []byte(\"started_event\"),\n\t\t\"create_request_id\":        createRequestID,\n\t\t\"event_data_encoding\":      \"Proto3\",\n\t\t\"domain_id\":                domainID,\n\t\t\"workflow_type_name\":       \"workflow_type_name\",\n\t\t\"parent_close_policy\":      1,\n\t}\n\texpected := &persistence.InternalChildExecutionInfo{\n\t\tVersion:               int64(1),\n\t\tInitiatedID:           int64(2),\n\t\tInitiatedEventBatchID: int64(3),\n\t\tInitiatedEvent:        persistence.NewDataBlob([]byte(\"initiated_event\"), \"Proto3\"),\n\t\tStartedID:             int64(4),\n\t\tStartedWorkflowID:     \"started_workflow_id\",\n\t\tStartedRunID:          startedRunID.String(),\n\t\tStartedEvent:          persistence.NewDataBlob([]byte(\"started_event\"), \"Proto3\"),\n\t\tCreateRequestID:       createRequestID.String(),\n\t\tDomainID:              domainID.String(),\n\t\tWorkflowTypeName:      \"workflow_type_name\",\n\t\tParentClosePolicy:     1,\n\t}\n\tassert.Equal(t, expected, parseChildExecutionInfo(testInput))\n\n\t// edge case\n\ttestInput = map[string]interface{}{\n\t\t\"domain_id\":   newMockUUID(_emptyUUID.String()),\n\t\t\"domain_name\": \"domain_name\",\n\t}\n\tassert.Equal(t, \"domain_name\", parseChildExecutionInfo(testInput).DomainNameDEPRECATED)\n\tassert.Equal(t, \"\", parseChildExecutionInfo(testInput).DomainID)\n}\n\nfunc Test_parseRequestCancelInfo(t *testing.T) {\n\ttestInput := map[string]interface{}{\n\t\t\"version\":                  int64(1),\n\t\t\"initiated_id\":             int64(2),\n\t\t\"initiated_event_batch_id\": int64(3),\n\t\t\"cancel_request_id\":        \"cancel_request_id\",\n\t}\n\texpected := &persistence.RequestCancelInfo{\n\t\tVersion:               int64(1),\n\t\tInitiatedID:           int64(2),\n\t\tInitiatedEventBatchID: int64(3),\n\t\tCancelRequestID:       \"cancel_request_id\",\n\t}\n\tassert.Equal(t, expected, parseRequestCancelInfo(testInput))\n}\n\nfunc Test_parseSignalInfo(t *testing.T) {\n\ttestInput := map[string]interface{}{\n\t\t\"version\":                  int64(1),\n\t\t\"initiated_id\":             int64(2),\n\t\t\"initiated_event_batch_id\": int64(3),\n\t\t\"signal_request_id\":        newMockUUID(\"signal_request_id\"),\n\t\t\"signal_name\":              \"signal_name\",\n\t\t\"input\":                    []byte(\"input\"),\n\t\t\"control\":                  []byte(\"control\"),\n\t}\n\texpected := &persistence.SignalInfo{\n\t\tVersion:               int64(1),\n\t\tInitiatedID:           int64(2),\n\t\tInitiatedEventBatchID: int64(3),\n\t\tSignalName:            \"signal_name\",\n\t\tSignalRequestID:       \"signal_request_id\",\n\t\tInput:                 []byte(\"input\"),\n\t\tControl:               []byte(\"control\"),\n\t}\n\tassert.Equal(t, expected, parseSignalInfo(testInput))\n}\n\nfunc Test_parseTimerTaskInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttestInput := map[string]interface{}{\n\t\t\"version\":          int64(1),\n\t\t\"visibility_ts\":    timeNow,\n\t\t\"task_id\":          int64(2),\n\t\t\"run_id\":           newMockUUID(\"run_id\"),\n\t\t\"type\":             3,\n\t\t\"timeout_type\":     3,\n\t\t\"event_id\":         int64(4),\n\t\t\"schedule_attempt\": int64(5),\n\t\t\"task_list\":        \"task_list\",\n\t}\n\texpected := &persistence.TimerTaskInfo{\n\t\tVersion:             int64(1),\n\t\tVisibilityTimestamp: timeNow,\n\t\tTaskID:              int64(2),\n\t\tRunID:               \"run_id\",\n\t\tTaskType:            3,\n\t\tTimeoutType:         3,\n\t\tEventID:             int64(4),\n\t\tScheduleAttempt:     int64(5),\n\t\tTaskList:            \"task_list\",\n\t}\n\tassert.Equal(t, expected, parseTimerTaskInfo(testInput))\n}\n\nfunc Test_parseReplicationTaskInfo(t *testing.T) {\n\ttestInput := map[string]interface{}{\n\t\t\"domain_id\":            newMockUUID(\"domain_id\"),\n\t\t\"workflow_id\":          \"workflow_id\",\n\t\t\"run_id\":               newMockUUID(\"run_id\"),\n\t\t\"task_id\":              int64(1),\n\t\t\"type\":                 2,\n\t\t\"first_event_id\":       int64(3),\n\t\t\"next_event_id\":        int64(4),\n\t\t\"version\":              int64(5),\n\t\t\"scheduled_id\":         int64(6),\n\t\t\"branch_token\":         []byte(\"branch_token\"),\n\t\t\"new_run_branch_token\": []byte(\"new_run_branch_token\"),\n\t\t\"created_time\":         int64(7),\n\t}\n\texpected := &nosqlplugin.ReplicationTask{\n\t\tDomainID:          \"domain_id\",\n\t\tWorkflowID:        \"workflow_id\",\n\t\tRunID:             \"run_id\",\n\t\tTaskID:            int64(1),\n\t\tTaskType:          2,\n\t\tFirstEventID:      int64(3),\n\t\tNextEventID:       int64(4),\n\t\tVersion:           int64(5),\n\t\tScheduledID:       int64(6),\n\t\tBranchToken:       []byte(\"branch_token\"),\n\t\tNewRunBranchToken: []byte(\"new_run_branch_token\"),\n\t\tCreationTime:      time.Unix(0, 7),\n\t}\n\tassert.Equal(t, expected, parseReplicationTaskInfo(testInput))\n}\n\nfunc Test_parseTransferTaskInfo(t *testing.T) {\n\ttimeNow := time.Now()\n\ttestInput := map[string]interface{}{\n\t\t\"domain_id\":                  newMockUUID(\"domain_id\"),\n\t\t\"workflow_id\":                \"workflow_id\",\n\t\t\"run_id\":                     newMockUUID(\"run_id\"),\n\t\t\"visibility_ts\":              timeNow,\n\t\t\"task_id\":                    int64(1),\n\t\t\"target_domain_id\":           newMockUUID(\"target_domain_id\"),\n\t\t\"target_domain_ids\":          []interface{}{newMockUUID(\"target_domain_id\")},\n\t\t\"target_workflow_id\":         \"target_workflow_id\",\n\t\t\"target_run_id\":              newMockUUID(\"target_run_id\"),\n\t\t\"target_child_workflow_only\": true,\n\t\t\"task_list\":                  \"task_list\",\n\t\t\"type\":                       2,\n\t\t\"schedule_id\":                int64(3),\n\t\t\"record_visibility\":          true,\n\t\t\"version\":                    int64(4),\n\t\t\"original_task_list\":         \"original_task_list\",\n\t\t\"original_task_list_kind\":    2,\n\t}\n\texpected := &persistence.TransferTaskInfo{\n\t\tDomainID:                \"domain_id\",\n\t\tWorkflowID:              \"workflow_id\",\n\t\tRunID:                   \"run_id\",\n\t\tVisibilityTimestamp:     timeNow,\n\t\tTaskID:                  int64(1),\n\t\tTargetDomainID:          \"target_domain_id\",\n\t\tTargetDomainIDs:         map[string]struct{}{\"target_domain_id\": {}},\n\t\tTargetWorkflowID:        \"target_workflow_id\",\n\t\tTargetRunID:             \"target_run_id\",\n\t\tTargetChildWorkflowOnly: true,\n\t\tTaskList:                \"task_list\",\n\t\tTaskType:                2,\n\t\tScheduleID:              int64(3),\n\t\tRecordVisibility:        true,\n\t\tVersion:                 int64(4),\n\t\tOriginalTaskList:        \"original_task_list\",\n\t\tOriginalTaskListKind:    types.TaskListKindEphemeral,\n\t}\n\tassert.Equal(t, expected, parseTransferTaskInfo(testInput))\n\n\t// edge case\n\ttestInput = map[string]interface{}{\n\t\t\"target_run_id\": newMockUUID(persistence.TransferTaskTransferTargetRunID),\n\t}\n\texpected = &persistence.TransferTaskInfo{\n\t\tTargetRunID: \"\",\n\t}\n\tassert.Equal(t, expected, parseTransferTaskInfo(testInput))\n}\n\nfunc Test_parseChecksum(t *testing.T) {\n\ttestInput := map[string]interface{}{\n\t\t\"version\": 1,\n\t\t\"flavor\":  2,\n\t\t\"value\":   []byte(\"value\"),\n\t}\n\texpected := checksum.Checksum{\n\t\tVersion: 1,\n\t\tFlavor:  2,\n\t\tValue:   []byte(\"value\"),\n\t}\n\tassert.Equal(t, expected, parseChecksum(testInput))\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/testdata\"\n)\n\nfunc TestInsertWorkflowExecutionWithTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname                            string\n\t\tworkflowRequest                 *nosqlplugin.WorkflowRequestsWriteRequest\n\t\trequest                         *nosqlplugin.CurrentWorkflowWriteRequest\n\t\texecution                       *nosqlplugin.WorkflowExecutionRequest\n\t\ttasksByCategory                 map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask\n\t\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow\n\t\tshardCondition                  *nosqlplugin.ShardCondition\n\t\tmapExecuteBatchCASErr           error\n\t\twantErr                         bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(),\n\t\t},\n\t\t{\n\t\t\tname: \"success with active cluster selection policy row\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(),\n\t\t\tactiveClusterSelectionPolicyRow: &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\t\t\tShardID:    1,\n\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\tPolicy: &persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"test-policy\"),\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"insertOrUpsertWorkflowRequestRow step fails\",\n\t\t\tworkflowRequest: &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\t\t\tRows: []*nosqlplugin.WorkflowRequestRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tWriteMode: nosqlplugin.WorkflowRequestWriteMode(-999), // unknown mode will cause failure\n\t\t\t},\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(),\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname: \"createOrUpdateCurrentWorkflow step fails\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteMode(-999), // unknown mode will cause failure\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(),\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname: \"createWorkflowExecutionWithMergeMaps step fails\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeAppend), // this will cause failure\n\t\t\t),\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                  \"executeCreateWorkflowBatchTransaction step fails\",\n\t\t\tmapExecuteBatchCASErr: errors.New(\"some random error\"), // this will cause failure\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeNone),\n\t\t\t),\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tsession := &fakeSession{\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.InsertWorkflowExecutionWithTasks(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.workflowRequest,\n\t\t\t\ttc.request,\n\t\t\t\ttc.execution,\n\t\t\t\ttc.tasksByCategory,\n\t\t\t\ttc.activeClusterSelectionPolicyRow,\n\t\t\t\ttc.shardCondition,\n\t\t\t)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"InsertWorkflowExecutionWithTasks() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectCurrentWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname             string\n\t\tshardID          int\n\t\tdomainID         string\n\t\tworkflowID       string\n\t\tcurrentRunID     string\n\t\tlastWriteVersion int64\n\t\tcreateReqID      string\n\t\twantErr          bool\n\t\twantRow          *nosqlplugin.CurrentWorkflowRow\n\t}{\n\t\t{\n\t\t\tname:         \"success\",\n\t\t\tshardID:      1,\n\t\t\tdomainID:     \"test-domain-id\",\n\t\t\tworkflowID:   \"test-workflow-id\",\n\t\t\tcurrentRunID: \"test-run-id\",\n\t\t\twantErr:      false,\n\t\t\twantRow: &nosqlplugin.CurrentWorkflowRow{\n\t\t\t\tShardID:          1,\n\t\t\t\tDomainID:         \"test-domain-id\",\n\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\tRunID:            \"test-run-id\",\n\t\t\t\tLastWriteVersion: -24,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"mapscan failure\",\n\t\t\tshardID:      1,\n\t\t\tdomainID:     \"test-domain-id\",\n\t\t\tworkflowID:   \"test-workflow-id\",\n\t\t\tcurrentRunID: \"test-run-id\",\n\t\t\twantErr:      true,\n\t\t},\n\t\t{\n\t\t\tname:             \"lastwriteversion populated\",\n\t\t\tshardID:          1,\n\t\t\tdomainID:         \"test-domain-id\",\n\t\t\tworkflowID:       \"test-workflow-id\",\n\t\t\tcurrentRunID:     \"test-run-id\",\n\t\t\tlastWriteVersion: 123,\n\t\t\twantErr:          false,\n\t\t\twantRow: &nosqlplugin.CurrentWorkflowRow{\n\t\t\t\tShardID:          1,\n\t\t\t\tDomainID:         \"test-domain-id\",\n\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\tRunID:            \"test-run-id\",\n\t\t\t\tLastWriteVersion: 123,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"create request id populated\",\n\t\t\tshardID:      1,\n\t\t\tdomainID:     \"test-domain-id\",\n\t\t\tworkflowID:   \"test-workflow-id\",\n\t\t\tcurrentRunID: \"test-run-id\",\n\t\t\tcreateReqID:  \"test-create-request-id\",\n\t\t\twantErr:      false,\n\t\t\twantRow: &nosqlplugin.CurrentWorkflowRow{\n\t\t\t\tShardID:          1,\n\t\t\t\tDomainID:         \"test-domain-id\",\n\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\tRunID:            \"test-run-id\",\n\t\t\t\tLastWriteVersion: -24,\n\t\t\t\tCreateRequestID:  \"test-create-request-id\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\tmockCurrentRunID := gocql.NewMockUUID(ctrl)\n\t\t\t\tm[\"current_run_id\"] = mockCurrentRunID\n\t\t\t\tif !tc.wantErr {\n\t\t\t\t\tmockCurrentRunID.EXPECT().String().Return(tc.currentRunID).Times(1)\n\t\t\t\t}\n\n\t\t\t\texecMap := map[string]interface{}{}\n\t\t\t\tif tc.createReqID != \"\" {\n\t\t\t\t\tmockReqID := gocql.NewMockUUID(ctrl)\n\t\t\t\t\tmockReqID.EXPECT().String().Return(tc.createReqID).Times(1)\n\t\t\t\t\texecMap[\"create_request_id\"] = mockReqID\n\t\t\t\t}\n\t\t\t\tm[\"execution\"] = execMap\n\n\t\t\t\tif tc.lastWriteVersion != 0 {\n\t\t\t\t\tm[\"workflow_last_write_version\"] = tc.lastWriteVersion\n\t\t\t\t}\n\n\t\t\t\tif tc.wantErr {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}\n\n\t\t\t\treturn nil\n\t\t\t}).Times(1)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\trow, err := db.SelectCurrentWorkflow(context.Background(), tc.shardID, tc.domainID, tc.workflowID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectCurrentWorkflow() error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantRow, row); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Row mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionWithTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname                            string\n\t\tworkflowRequest                 *nosqlplugin.WorkflowRequestsWriteRequest\n\t\trequest                         *nosqlplugin.CurrentWorkflowWriteRequest\n\t\tmutatedExecution                *nosqlplugin.WorkflowExecutionRequest\n\t\tinsertedExecution               *nosqlplugin.WorkflowExecutionRequest\n\t\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow\n\t\tresetExecution                  *nosqlplugin.WorkflowExecutionRequest\n\t\ttasksByCategory                 map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask\n\t\tshardCondition                  *nosqlplugin.ShardCondition\n\t\tmapExecuteBatchCASErr           error\n\t\twantErr                         bool\n\t}{\n\t\t{\n\t\t\tname: \"both mutatedExecution and resetExecution not provided\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"insertOrUpsertWorkflowRequestRow step fails\",\n\t\t\tworkflowRequest: &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\t\t\tRows: []*nosqlplugin.WorkflowRequestRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tWriteMode: nosqlplugin.WorkflowRequestWriteMode(-999), // unknown mode will cause failure\n\t\t\t},\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmutatedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeUpdate),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t\twantErr:           true,\n\t\t},\n\t\t{\n\t\t\tname: \"mutatedExecution provided - success\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmutatedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeUpdate),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t},\n\t\t{\n\t\t\tname:    \"mutatedExecution provided - update fails\",\n\t\t\twantErr: true,\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmutatedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeCreate), // this will cause failure\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t},\n\t\t{\n\t\t\tname: \"resetExecution provided - success\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tresetExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeClear),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeReset),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t},\n\t\t{\n\t\t\tname:    \"resetExecution provided - reset fails\",\n\t\t\twantErr: true,\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tresetExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeNone), // this will cause failure\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeReset),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t},\n\t\t{\n\t\t\tname: \"resetExecution and insertedExecution provided - success\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tresetExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeClear),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeReset),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeNone),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeCreate),\n\t\t\t),\n\t\t},\n\t\t{\n\t\t\tname: \"mutatedExecution and insertedExecution and activeClusterSelectionPolicyRow provided - success\",\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmutatedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeNone),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeUpdate),\n\t\t\t),\n\t\t\tactiveClusterSelectionPolicyRow: &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\t\t\tShardID:    1,\n\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\tPolicy:     &persistence.DataBlob{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"test-policy\")},\n\t\t\t},\n\t\t\tinsertedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeNone),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeCreate),\n\t\t\t),\n\t\t},\n\t\t{\n\t\t\tname:    \"resetExecution and insertedExecution provided - insert fails\",\n\t\t\twantErr: true,\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tresetExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeClear),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeReset),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeClear), // this will cause failure\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeCreate),\n\t\t\t),\n\t\t},\n\t\t{\n\t\t\tname:    \"createOrUpdateCurrentWorkflow step fails\",\n\t\t\twantErr: true,\n\t\t\trequest: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tCondition: nil, // this will cause failure because Condition must be non-nil for update mode\n\t\t\t},\n\t\t\tshardCondition: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tresetExecution: testdata.WFExecRequest(\n\t\t\t\ttestdata.WFExecRequestWithEventBufferWriteMode(nosqlplugin.EventBufferWriteModeClear),\n\t\t\t\ttestdata.WFExecRequestWithMapsWriteMode(nosqlplugin.WorkflowExecutionMapsWriteModeReset),\n\t\t\t),\n\t\t\tinsertedExecution: testdata.WFExecRequest(),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tsession := &fakeSession{\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\terr := db.UpdateWorkflowExecutionWithTasks(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.workflowRequest,\n\t\t\t\ttc.request,\n\t\t\t\ttc.mutatedExecution,\n\t\t\t\ttc.insertedExecution,\n\t\t\t\tnil, // TODO(active-active): add test cases for activeClusterSelectionPolicyRow\n\t\t\t\ttc.resetExecution,\n\t\t\t\ttc.tasksByCategory,\n\t\t\t\ttc.shardCondition,\n\t\t\t)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"UpdateWorkflowExecutionWithTasks() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantResp    *nosqlplugin.WorkflowExecution\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:       \"mapscan failure\",\n\t\t\tshardID:    1,\n\t\t\tdomainID:   \"test-domain-id\",\n\t\t\tworkflowID: \"test-workflow-id\",\n\t\t\trunID:      \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tshardID:    1,\n\t\t\tdomainID:   \"test-domain-id\",\n\t\t\tworkflowID: \"test-workflow-id\",\n\t\t\trunID:      \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"execution\"] = map[string]interface{}{}\n\t\t\t\t\tm[\"version_histories\"] = []byte{}\n\t\t\t\t\tm[\"version_histories_encoding\"] = \"thriftrw\"\n\t\t\t\t\tm[\"replication_state\"] = map[string]interface{}{}\n\t\t\t\t\tm[\"activity_map\"] = map[int64]map[string]interface{}{\n\t\t\t\t\t\t1: {\"schedule_id\": int64(1)},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"timer_map\"] = map[string]map[string]interface{}{\n\t\t\t\t\t\t\"t1\": {\"started_id\": int64(5)},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"child_executions_map\"] = map[int64]map[string]interface{}{\n\t\t\t\t\t\t3: {\"initiated_id\": int64(2)},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"request_cancel_map\"] = map[int64]map[string]interface{}{\n\t\t\t\t\t\t6: {\"initiated_id\": int64(5)},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"signal_map\"] = map[int64]map[string]interface{}{\n\t\t\t\t\t\t8: {\"initiated_id\": int64(7)},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"signal_requested\"] = []interface{}{\n\t\t\t\t\t\t&fakeUUID{uuid: \"aae7b881-48ea-4b23-8d11-aabfd1c1291e\"},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"buffered_events_list\"] = []map[string]interface{}{\n\t\t\t\t\t\t{\"encoding_type\": \"thriftrw\", \"data\": []byte(\"test-buffered-events-1\")},\n\t\t\t\t\t}\n\t\t\t\t\tm[\"checksum\"] = map[string]interface{}{}\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantResp: &nosqlplugin.WorkflowExecution{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\tActivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t\t1: {ScheduleID: 1, DomainID: \"test-domain-id\"},\n\t\t\t\t},\n\t\t\t\tTimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\t\"t1\": {StartedID: 5},\n\t\t\t\t},\n\t\t\t\tChildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t3: {InitiatedID: 2},\n\t\t\t\t},\n\t\t\t\tRequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t\t6: {InitiatedID: 5},\n\t\t\t\t},\n\t\t\t\tSignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t\t8: {InitiatedID: 7},\n\t\t\t\t},\n\t\t\t\tSignalRequestedIDs: map[string]struct{}{\n\t\t\t\t\t\"aae7b881-48ea-4b23-8d11-aabfd1c1291e\": {},\n\t\t\t\t},\n\t\t\t\tBufferedEvents: []*persistence.DataBlob{\n\t\t\t\t\t{Encoding: constants.EncodingTypeThriftRW, Data: []byte(\"test-buffered-events-1\")},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\t\t\tsession := &fakeSession{\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\t\tquery:                     query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgot, err := db.SelectWorkflowExecution(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.shardID,\n\t\t\t\ttc.domainID,\n\t\t\t\ttc.workflowID,\n\t\t\t\ttc.runID,\n\t\t\t)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectWorkflowExecution() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantResp, got); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteCurrentWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname                  string\n\t\tshardID               int\n\t\tdomainID              string\n\t\tworkflowID            string\n\t\tcurrentRunIDCondition string\n\t\tqueryMockFn           func(query *gocql.MockQuery)\n\t\twantErr               bool\n\t}{\n\t\t{\n\t\t\tname:                  \"success\",\n\t\t\tshardID:               1,\n\t\t\tdomainID:              \"test-domain-id\",\n\t\t\tworkflowID:            \"test-workflow-id\",\n\t\t\tcurrentRunIDCondition: \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                  \"query exec fails\",\n\t\t\tshardID:               1,\n\t\t\tdomainID:              \"test-domain-id\",\n\t\t\tworkflowID:            \"test-workflow-id\",\n\t\t\tcurrentRunIDCondition: \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteCurrentWorkflow(context.Background(), tc.shardID, tc.domainID, tc.workflowID, tc.currentRunIDCondition)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteCurrentWorkflow() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tshardID:    1,\n\t\t\tdomainID:   \"test-domain-id\",\n\t\t\tworkflowID: \"test-workflow-id\",\n\t\t\trunID:      \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:       \"query exec fails\",\n\t\t\tshardID:    1,\n\t\t\tdomainID:   \"test-domain-id\",\n\t\t\tworkflowID: \"test-workflow-id\",\n\t\t\trunID:      \"test-run-id\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteWorkflowExecution(context.Background(), tc.shardID, tc.domainID, tc.workflowID, tc.runID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteWorkflowExecution() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectAllCurrentWorkflows(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tshardID        int\n\t\tpageToken      []byte\n\t\tpageSize       int\n\t\titer           *fakeIter\n\t\twantExecutions []*persistence.CurrentWorkflowExecution\n\t\twantErr        bool\n\t}{\n\t\t{\n\t\t\tname:      \"nil iter returned\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"run_id is not permanentRunID so excluded from result\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\": &fakeUUID{uuid: \"17C305FA-79BB-479E-8AC7-360E956AC01A\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantExecutions: nil,\n\t\t},\n\t\t{\n\t\t\tname:      \"multiple executions\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\":         &fakeUUID{uuid: permanentRunID},\n\t\t\t\t\t\t\"domain_id\":      &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\"current_run_id\": &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\"workflow_id\":    \"wfid1\",\n\t\t\t\t\t\t\"workflow_state\": 1,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\":         &fakeUUID{uuid: permanentRunID},\n\t\t\t\t\t\t\"domain_id\":      &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\"current_run_id\": &fakeUUID{uuid: \"runid2\"},\n\t\t\t\t\t\t\"workflow_id\":    \"wfid2\",\n\t\t\t\t\t\t\"workflow_state\": 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantExecutions: []*persistence.CurrentWorkflowExecution{\n\t\t\t\t{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tWorkflowID:   \"wfid1\",\n\t\t\t\t\tRunID:        \"30000000-0000-f000-f000-000000000001\",\n\t\t\t\t\tState:        1,\n\t\t\t\t\tCurrentRunID: \"runid1\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDomainID:     \"domain1\",\n\t\t\t\t\tWorkflowID:   \"wfid2\",\n\t\t\t\t\tRunID:        \"30000000-0000-f000-f000-000000000001\",\n\t\t\t\t\tState:        1,\n\t\t\t\t\tCurrentRunID: \"runid2\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(tc.pageSize).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(tc.pageToken).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\t// Passing tc.iter to Return() doesn't work even though tc.iter is nil due to Go's typed nils.\n\t\t\t\t// So, we have to call Return(nil) directly.\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotExecutions, gotPageToken, err := db.SelectAllCurrentWorkflows(context.Background(), tc.shardID, tc.pageToken, tc.pageSize)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectAllCurrentWorkflows() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantExecutions, gotExecutions); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Executions mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.iter.pageState, gotPageToken); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Page token mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Error(\"iter was not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectAllWorkflowExecutions(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tshardID        int\n\t\tpageToken      []byte\n\t\tpageSize       int\n\t\titer           *fakeIter\n\t\twantExecutions []*persistence.InternalListConcreteExecutionsEntity\n\t\twantErr        bool\n\t}{\n\t\t{\n\t\t\tname:      \"nil iter returned\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"run_id is permanentRunID so excluded from result\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\": &fakeUUID{uuid: \"30000000-0000-f000-f000-000000000001\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantExecutions: nil,\n\t\t},\n\t\t{\n\t\t\tname:      \"multiple executions\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\": &fakeUUID{uuid: \"runid1\"},\n\t\t\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"version_histories\":          []byte(\"test-version-histories-1\"),\n\t\t\t\t\t\t\"version_histories_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"run_id\": &fakeUUID{uuid: \"runid2\"},\n\t\t\t\t\t\t\"execution\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"workflow_id\": \"wfid2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"version_histories\":          []byte(\"test-version-histories-1\"),\n\t\t\t\t\t\t\"version_histories_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantExecutions: []*persistence.InternalListConcreteExecutionsEntity{\n\t\t\t\t{\n\t\t\t\t\tExecutionInfo:    &persistence.InternalWorkflowExecutionInfo{DomainID: \"domain1\", WorkflowID: \"wfid1\"},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"test-version-histories-1\")},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tExecutionInfo:    &persistence.InternalWorkflowExecutionInfo{DomainID: \"domain1\", WorkflowID: \"wfid2\"},\n\t\t\t\t\tVersionHistories: &persistence.DataBlob{Encoding: \"thriftrw\", Data: []uint8(\"test-version-histories-1\")},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(tc.pageSize).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(tc.pageToken).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\t// Passing tc.iter to Return() doesn't work even though tc.iter is nil due to Go's typed nils.\n\t\t\t\t// So, we have to call Return(nil) directly.\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotExecutions, gotPageToken, err := db.SelectAllWorkflowExecutions(context.Background(), tc.shardID, tc.pageToken, tc.pageSize)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectAllWorkflowExecutions() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantExecutions, gotExecutions); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Executions mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.iter.pageState, gotPageToken); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Page token mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Error(\"iter was not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsWorkflowExecutionExists(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tqueryMockFn  func(query *gocql.MockQuery)\n\t\tclientMockFn func(client *gocql.MockClient)\n\t\twant         bool\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twant:    true,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"not found case returns false but no error\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).Return(errors.New(\"an error that will be considered as not found err by client mock\")).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(true).Times(1)\n\t\t\t},\n\t\t\twant:    false,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"arbitrary error case\",\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).Return(errors.New(\"some error\")).Times(1)\n\t\t\t},\n\t\t\tclientMockFn: func(client *gocql.MockClient) {\n\t\t\t\tclient.EXPECT().IsNotFoundError(gomock.Any()).Return(false).Times(1)\n\t\t\t},\n\t\t\twant:    false,\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tif tc.clientMockFn != nil {\n\t\t\t\ttc.clientMockFn(client)\n\t\t\t}\n\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgot, err := db.IsWorkflowExecutionExists(context.Background(), 1, \"domain1\", \"wfi\", \"run1\")\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"IsWorkflowExecutionExists() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif got != tc.want {\n\t\t\t\tt.Errorf(\"IsWorkflowExecutionExists() got: %v, want: %v\", got, tc.want)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectTransferTasksOrderByTaskID(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tshardID            int\n\t\tpageToken          []byte\n\t\tpageSize           int\n\t\tinclusiveMinTaskID int64\n\t\texclusiveMaxTaskID int64\n\t\titer               *fakeIter\n\t\twantTasks          []*nosqlplugin.HistoryMigrationTask\n\t\twantNextPageToken  []byte\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname:      \"nil iter returned\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"success\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(1),\n\t\t\t\t\t\t\"transfer\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t\t\"task_id\":     int64(1),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"data\":          []byte(\"test-data-1\"),\n\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"task_id\": int64(5),\n\t\t\t\t\t\t\"transfer\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain2\"},\n\t\t\t\t\t\t\t\"workflow_id\": \"wfid2\",\n\t\t\t\t\t\t\t\"task_id\":     int64(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"data\":          []byte(\"test-data-2\"),\n\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tTransfer: &nosqlplugin.TransferTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-1\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTransfer: &nosqlplugin.TransferTask{\n\t\t\t\t\t\tDomainID:   \"domain2\",\n\t\t\t\t\t\tWorkflowID: \"wfid2\",\n\t\t\t\t\t\tTaskID:     5,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-2\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 5,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(tc.pageSize).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(tc.pageToken).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\t// Passing tc.iter to Return() doesn't work even though tc.iter is nil due to Go's typed nils.\n\t\t\t\t// So, we have to call Return(nil) directly.\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotTasks, gotPageToken, err := db.SelectTransferTasksOrderByTaskID(context.Background(), tc.shardID, tc.pageSize, tc.pageToken, tc.inclusiveMinTaskID, tc.exclusiveMaxTaskID)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectAllWorkflowExecutions() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantTasks, gotTasks); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Executions mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.iter.pageState, gotPageToken); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Page token mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Error(\"iter was not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteTransferTask(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\ttaskID      int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:    \"success\",\n\t\t\tshardID: 1,\n\t\t\ttaskID:  123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"query exec fails\",\n\t\t\tshardID: 1,\n\t\t\ttaskID:  123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteTransferTask(context.Background(), tc.shardID, tc.taskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteTransferTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteTransferTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\tshardID              int\n\t\tinclusiveBeginTaskID int64\n\t\texclusiveEndTaskID   int64\n\t\tqueryMockFn          func(query *gocql.MockQuery)\n\t\twantErr              bool\n\t}{\n\t\t{\n\t\t\tname:                 \"success\",\n\t\t\tshardID:              1,\n\t\t\tinclusiveBeginTaskID: 123,\n\t\t\texclusiveEndTaskID:   456,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                 \"query exec fails\",\n\t\t\tshardID:              1,\n\t\t\tinclusiveBeginTaskID: 123,\n\t\t\texclusiveEndTaskID:   456,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.RangeDeleteTransferTasks(context.Background(), tc.shardID, tc.inclusiveBeginTaskID, tc.exclusiveEndTaskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteTransferTasks() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectTimerTasksOrderByVisibilityTime(t *testing.T) {\n\tnow := time.Now()\n\n\ttests := []struct {\n\t\tname              string\n\t\tshardID           int\n\t\tpageToken         []byte\n\t\tpageSize          int\n\t\tinclusiveMinTime  time.Time\n\t\texclusiveMaxTime  time.Time\n\t\titer              *fakeIter\n\t\twantTasks         []*nosqlplugin.HistoryMigrationTask\n\t\twantNextPageToken []byte\n\t\twantErr           bool\n\t}{\n\t\t{\n\t\t\tname:      \"nil iter returned\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"success\",\n\t\t\tshardID:   1,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tpageSize:  10,\n\t\t\titer: &fakeIter{\n\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t{\n\t\t\t\t\t\t\"visibility_ts\": now,\n\t\t\t\t\t\t\"task_id\":       int64(1),\n\t\t\t\t\t\t\"timer\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":     &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\"workflow_id\":   \"wfid1\",\n\t\t\t\t\t\t\t\"visibility_ts\": now,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"data\":          []byte(\"test-data-1\"),\n\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\t\"visibility_ts\": now.Add(time.Hour),\n\t\t\t\t\t\t\"task_id\":       int64(5),\n\t\t\t\t\t\t\"timer\": map[string]interface{}{\n\t\t\t\t\t\t\t\"domain_id\":     &fakeUUID{uuid: \"domain2\"},\n\t\t\t\t\t\t\t\"workflow_id\":   \"wfid2\",\n\t\t\t\t\t\t\t\"visibility_ts\": now.Add(time.Hour),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"data\":          []byte(\"test-data-2\"),\n\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpageState: []byte(\"test-page-token-2\"),\n\t\t\t},\n\t\t\twantTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tTimer: &nosqlplugin.TimerTask{\n\t\t\t\t\t\tDomainID:            \"domain1\",\n\t\t\t\t\t\tWorkflowID:          \"wfid1\",\n\t\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-1\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID:        1,\n\t\t\t\t\tScheduledTime: now,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTimer: &nosqlplugin.TimerTask{\n\t\t\t\t\t\tDomainID:            \"domain2\",\n\t\t\t\t\t\tWorkflowID:          \"wfid2\",\n\t\t\t\t\t\tVisibilityTimestamp: now.Add(time.Hour),\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-2\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID:        5,\n\t\t\t\t\tScheduledTime: now.Add(time.Hour),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\tquery.EXPECT().PageSize(tc.pageSize).Return(query).Times(1)\n\t\t\tquery.EXPECT().PageState(tc.pageToken).Return(query).Times(1)\n\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\tif tc.iter != nil {\n\t\t\t\tquery.EXPECT().Iter().Return(tc.iter).Times(1)\n\t\t\t} else {\n\t\t\t\t// Passing tc.iter to Return() doesn't work even though tc.iter is nil due to Go's typed nils.\n\t\t\t\t// So, we have to call Return(nil) directly.\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\n\t\t\tclient := gocql.NewMockClient(ctrl)\n\t\t\tcfg := &config.NoSQL{}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdc := &persistence.DynamicConfiguration{}\n\t\t\tdb := NewCassandraDBFromSession(cfg, session, logger, dc, DbWithClient(client))\n\n\t\t\tgotTasks, gotPageToken, err := db.SelectTimerTasksOrderByVisibilityTime(context.Background(), tc.shardID, tc.pageSize, tc.pageToken, tc.inclusiveMinTime, tc.exclusiveMaxTime)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectTimerTasksOrderByVisibilityTime() error: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantTasks, gotTasks); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Tasks mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.iter.pageState, gotPageToken); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Page token mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif !tc.iter.closed {\n\t\t\t\tt.Error(\"iter was not closed\")\n\t\t\t}\n\n\t\t})\n\t}\n}\n\nfunc TestDeleteTimerTask(t *testing.T) {\n\tnow := time.Now()\n\ttests := []struct {\n\t\tname                string\n\t\tshardID             int\n\t\ttaskID              int64\n\t\tvisibilityTimestamp time.Time\n\t\tqueryMockFn         func(query *gocql.MockQuery)\n\t\twantErr             bool\n\t}{\n\t\t{\n\t\t\tname:                \"success\",\n\t\t\tshardID:             1,\n\t\t\ttaskID:              123,\n\t\t\tvisibilityTimestamp: now,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                \"query exec fails\",\n\t\t\tshardID:             1,\n\t\t\ttaskID:              123,\n\t\t\tvisibilityTimestamp: now,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteTimerTask(context.Background(), tc.shardID, tc.taskID, tc.visibilityTimestamp)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteTimerTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteTimerTasks(t *testing.T) {\n\tnow := time.Now()\n\ttests := []struct {\n\t\tname             string\n\t\tshardID          int\n\t\tinclusiveMinTime time.Time\n\t\texclusiveMaxTime time.Time\n\t\tqueryMockFn      func(query *gocql.MockQuery)\n\t\twantErr          bool\n\t}{\n\t\t{\n\t\t\tname:             \"success\",\n\t\t\tshardID:          1,\n\t\t\tinclusiveMinTime: now,\n\t\t\texclusiveMaxTime: now.Add(time.Hour),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:             \"query exec fails\",\n\t\t\tshardID:          1,\n\t\t\tinclusiveMinTime: now,\n\t\t\texclusiveMaxTime: now.Add(time.Hour),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.RangeDeleteTimerTasks(context.Background(), tc.shardID, tc.inclusiveMinTime, tc.exclusiveMaxTime)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteTimerTasks() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectReplicationTasksOrderByTaskID(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tshardID            int\n\t\tinclusiveMinTaskID int64\n\t\texclusiveMaxTaskID int64\n\t\tpageSize           int\n\t\tpageToken          []byte\n\t\tqueryMockFn        func(query *gocql.MockQuery)\n\t\twantTasks          []*nosqlplugin.HistoryMigrationTask\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname:               \"success\",\n\t\t\tshardID:            1,\n\t\t\tinclusiveMinTaskID: 100,\n\t\t\texclusiveMaxTaskID: 200,\n\t\t\tpageSize:           100,\n\t\t\tpageToken:          []byte(\"test-page-token\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().PageSize(100).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageState([]byte(\"test-page-token\")).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Iter().Return(&fakeIter{\n\t\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\"task_id\": int64(1),\n\t\t\t\t\t\t\t\"replication\": map[string]interface{}{\n\t\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t\t\t\"task_id\":     int64(1),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"data\":          []byte(\"test-data-1\"),\n\t\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\"task_id\": int64(2),\n\t\t\t\t\t\t\t\"replication\": map[string]interface{}{\n\t\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t\t\t\"task_id\":     int64(2),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"data\":          []byte(\"test-data-2\"),\n\t\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-1\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-2\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"query iter fails\",\n\t\t\tshardID:            1,\n\t\t\tinclusiveMinTaskID: 100,\n\t\t\texclusiveMaxTaskID: 200,\n\t\t\tpageSize:           100,\n\t\t\tpageToken:          []byte(\"test-page-token\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().PageSize(100).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageState([]byte(\"test-page-token\")).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\tgotTasks, _, err := db.SelectReplicationTasksOrderByTaskID(context.Background(), tc.shardID, tc.pageSize, tc.pageToken, tc.inclusiveMinTaskID, tc.exclusiveMaxTaskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectReplicationTasksOrderByTaskID() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantTasks, gotTasks); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Tasks mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteReplicationTask(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\ttaskID      int64\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:    \"success\",\n\t\t\tshardID: 1,\n\t\t\ttaskID:  123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"query exec fails\",\n\t\t\tshardID: 1,\n\t\t\ttaskID:  123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteReplicationTask(context.Background(), tc.shardID, tc.taskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteReplicationTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteReplicationTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tshardID            int\n\t\texclusiveEndTaskID int64\n\t\tqueryMockFn        func(query *gocql.MockQuery)\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname:               \"success\",\n\t\t\tshardID:            1,\n\t\t\texclusiveEndTaskID: 123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:               \"query exec fails\",\n\t\t\tshardID:            1,\n\t\t\texclusiveEndTaskID: 123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.RangeDeleteReplicationTasks(context.Background(), tc.shardID, tc.exclusiveEndTaskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteReplicationTasks() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteCrossClusterTask(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tshardID       int\n\t\ttargetCluster string\n\t\ttaskID        int64\n\t\tqueryMockFn   func(query *gocql.MockQuery)\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname:          \"success\",\n\t\t\tshardID:       1,\n\t\t\ttargetCluster: \"test-target-cluster\",\n\t\t\ttaskID:        123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"query exec fails\",\n\t\t\tshardID:       1,\n\t\t\ttargetCluster: \"test-target-cluster\",\n\t\t\ttaskID:        123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteCrossClusterTask(context.Background(), tc.shardID, tc.targetCluster, tc.taskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteCrossClusterTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertReplicationDLQTask(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tshardID       int\n\t\tsourceCluster string\n\t\ttaskID        int64\n\t\ttask          *nosqlplugin.HistoryMigrationTask\n\t\tqueryMockFn   func(query *gocql.MockQuery)\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname:          \"success\",\n\t\t\tshardID:       1,\n\t\t\tsourceCluster: \"test-source-cluster\",\n\t\t\ttaskID:        123,\n\t\t\ttask: &nosqlplugin.HistoryMigrationTask{\n\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\tTaskID: 123,\n\t\t\t\t},\n\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"dlq\"),\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t},\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"query exec fails\",\n\t\t\tshardID:       1,\n\t\t\tsourceCluster: \"test-source-cluster\",\n\t\t\ttaskID:        123,\n\t\t\ttask: &nosqlplugin.HistoryMigrationTask{\n\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\tTaskID: 123,\n\t\t\t\t},\n\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"dlq\"),\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t},\n\t\t\t},\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.InsertReplicationDLQTask(context.Background(), tc.shardID, tc.sourceCluster, tc.task)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteCrossClusterTasks() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectReplicationDLQTasksOrderByTaskID(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tshardID            int\n\t\tinclusiveMinTaskID int64\n\t\texclusiveMaxTaskID int64\n\t\tpageSize           int\n\t\tpageToken          []byte\n\t\tqueryMockFn        func(query *gocql.MockQuery)\n\t\twantTasks          []*nosqlplugin.HistoryMigrationTask\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname:               \"success\",\n\t\t\tshardID:            1,\n\t\t\tinclusiveMinTaskID: 100,\n\t\t\texclusiveMaxTaskID: 200,\n\t\t\tpageSize:           100,\n\t\t\tpageToken:          []byte(\"test-page-token\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().PageSize(100).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageState([]byte(\"test-page-token\")).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Iter().Return(&fakeIter{\n\t\t\t\t\tmapScanInputs: []map[string]interface{}{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\"task_id\": int64(1),\n\t\t\t\t\t\t\t\"replication\": map[string]interface{}{\n\t\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t\t\t\"task_id\":     int64(1),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"data\":          []byte(\"test-data-1\"),\n\t\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\"task_id\": int64(2),\n\t\t\t\t\t\t\t\"replication\": map[string]interface{}{\n\t\t\t\t\t\t\t\t\"domain_id\":   &fakeUUID{uuid: \"domain1\"},\n\t\t\t\t\t\t\t\t\"workflow_id\": \"wfid1\",\n\t\t\t\t\t\t\t\t\"task_id\":     int64(2),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"data\":          []byte(\"test-data-2\"),\n\t\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-1\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: persistence.NewDataBlob(\n\t\t\t\t\t\t[]byte(\"test-data-2\"),\n\t\t\t\t\t\t\"thriftrw\",\n\t\t\t\t\t),\n\t\t\t\t\tTaskID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"query iter fails\",\n\t\t\tshardID:            1,\n\t\t\tinclusiveMinTaskID: 100,\n\t\t\texclusiveMaxTaskID: 200,\n\t\t\tpageSize:           100,\n\t\t\tpageToken:          []byte(\"test-page-token\"),\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().PageSize(100).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().PageState([]byte(\"test-page-token\")).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Iter().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\tgotTasks, _, err := db.SelectReplicationDLQTasksOrderByTaskID(context.Background(), tc.shardID, \"src-cluster\", tc.pageSize, tc.pageToken, tc.inclusiveMinTaskID, tc.exclusiveMaxTaskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectReplicationDLQTasksOrderByTaskID() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantTasks, gotTasks); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Tasks mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectReplicationDLQTasksCount(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tshardID     int\n\t\tqueryMockFn func(query *gocql.MockQuery)\n\t\twantCount   int64\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:    \"success\",\n\t\t\tshardID: 1,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).DoAndReturn(func(m map[string]interface{}) error {\n\t\t\t\t\tm[\"count\"] = int64(42)\n\t\t\t\t\treturn nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\twantCount: 42,\n\t\t},\n\t\t{\n\t\t\tname:    \"query mapscan fails\",\n\t\t\tshardID: 1,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().MapScan(gomock.Any()).Return(errors.New(\"failed to scan\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\tgotCount, err := db.SelectReplicationDLQTasksCount(context.Background(), tc.shardID, \"src-cluster\")\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectReplicationDLQTasksCount() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotCount != tc.wantCount {\n\t\t\t\tt.Fatalf(\"got count %v, want %v\", gotCount, tc.wantCount)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteReplicationDLQTask(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tshardID       int\n\t\tsourceCluster string\n\t\ttaskID        int64\n\t\tqueryMockFn   func(query *gocql.MockQuery)\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname:          \"success\",\n\t\t\tshardID:       1,\n\t\t\tsourceCluster: \"test-source-cluster\",\n\t\t\ttaskID:        123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"query exec fails\",\n\t\t\tshardID:       1,\n\t\t\tsourceCluster: \"test-source-cluster\",\n\t\t\ttaskID:        123,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.DeleteReplicationDLQTask(context.Background(), tc.shardID, tc.sourceCluster, tc.taskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteReplicationDLQTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteReplicationDLQTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\tshardID              int\n\t\tsourceCluster        string\n\t\tinclusiveEndTaskID   int64\n\t\texclusiveBeginTaskID int64\n\t\tqueryMockFn          func(query *gocql.MockQuery)\n\t\twantErr              bool\n\t}{\n\t\t{\n\t\t\tname:                 \"success\",\n\t\t\tshardID:              1,\n\t\t\tsourceCluster:        \"test-source-cluster\",\n\t\t\tinclusiveEndTaskID:   300,\n\t\t\texclusiveBeginTaskID: 200,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                 \"query exec fails\",\n\t\t\tshardID:              1,\n\t\t\tsourceCluster:        \"test-source-cluster\",\n\t\t\tinclusiveEndTaskID:   300,\n\t\t\texclusiveBeginTaskID: 200,\n\t\t\tqueryMockFn: func(query *gocql.MockQuery) {\n\t\t\t\tquery.EXPECT().WithContext(gomock.Any()).Return(query).Times(1)\n\t\t\t\tquery.EXPECT().Exec().Return(errors.New(\"failed to exec\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tquery := gocql.NewMockQuery(ctrl)\n\t\t\ttc.queryMockFn(query)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tquery: query,\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.RangeDeleteReplicationDLQTasks(context.Background(), tc.shardID, tc.sourceCluster, tc.exclusiveBeginTaskID, tc.inclusiveEndTaskID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RangeDeleteReplicationDLQTasks() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertReplicationTask(t *testing.T) {\n\ttests := []struct {\n\t\tname                      string\n\t\ttasks                     []*nosqlplugin.HistoryMigrationTask\n\t\tshardCondition            nosqlplugin.ShardCondition\n\t\tmapExecuteBatchCASApplied bool\n\t\tmapExecuteBatchCASPrev    map[string]any\n\t\tmapExecuteBatchCASErr     error\n\t\twantErr                   bool\n\t\twantPanic                 bool\n\t}{\n\t\t{\n\t\t\tname:    \"no tasks\",\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"mapExecuteBatchCASErr failure\",\n\t\t\ttasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapExecuteBatchCASErr: errors.New(\"failed to execute batch\"),\n\t\t\twantErr:               true,\n\t\t},\n\t\t{\n\t\t\tname: \"not applied and row type not found causes panic\",\n\t\t\ttasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\twantPanic:                 true,\n\t\t},\n\t\t{\n\t\t\tname: \"not applied, row type shard condition failure\",\n\t\t\ttasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\"type\":     rowTypeShard,\n\t\t\t\t\"range_id\": int64(5),\n\t\t\t},\n\t\t\tshardCondition: nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t\tRangeID: 4, // mismatch with prev range_id causes failure\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"not applied, unknown shard condition failure\",\n\t\t\ttasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\"type\": -1, // not a shard type row\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"successfully applied\",\n\t\t\ttasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     1,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\t\t\tTaskID:     2,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"r2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmapExecuteBatchCASApplied: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdefer func() {\n\t\t\t\tif r := recover(); (r != nil) != tc.wantPanic {\n\t\t\t\t\tt.Errorf(\"got panic: %v, wantPanic: %v\", r, tc.wantPanic)\n\t\t\t\t}\n\t\t\t}()\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tsession := &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: tc.mapExecuteBatchCASApplied,\n\t\t\t\tmapExecuteBatchCASPrev:    tc.mapExecuteBatchCASPrev,\n\t\t\t\tmapExecuteBatchCASErr:     tc.mapExecuteBatchCASErr,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t}\n\t\t\tlogger := testlogger.New(t)\n\t\t\tdb := NewCassandraDBFromSession(nil, session, logger, nil, DbWithClient(gocql.NewMockClient(ctrl)))\n\n\t\t\terr := db.InsertReplicationTask(context.Background(), tc.tasks, tc.shardCondition)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"InsertReplicationTask() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantErr || len(tc.tasks) == 0 {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif len(session.batches) != 1 {\n\t\t\t\tt.Fatalf(\"got %v batches, want 1\", len(session.batches))\n\t\t\t}\n\n\t\t\tif len(tc.tasks)+1 != len(session.batches[0].queries) {\n\t\t\t\tt.Errorf(\"got %v batches, want %v\", len(session.batches), len(tc.tasks))\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectActiveClusterSelectionPolicy(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\twfID       string\n\t\trID        string\n\t\tsession    *fakeSession\n\t\tmockFn     func(cl *gocql.MockClient)\n\t\twantQuery  string\n\t\twantPolicy *nosqlplugin.ActiveClusterSelectionPolicyRow\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tshardID:  1,\n\t\t\tdomainID: \"domain1\",\n\t\t\twfID:     \"wfid1\",\n\t\t\trID:      \"r1\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{\n\t\t\t\t\t\t\"data\":          []byte(\"data1\"),\n\t\t\t\t\t\t\"data_encoding\": \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQuery: `SELECT data, data_encoding FROM executions WHERE ` +\n\t\t\t\t`shard_id = 1 and type = 11 and domain_id = domain1 and ` +\n\t\t\t\t`workflow_id = wfid1 and run_id = r1 and visibility_ts = 946684800000 and task_id = -1001`,\n\t\t\twantPolicy: &nosqlplugin.ActiveClusterSelectionPolicyRow{\n\t\t\t\tPolicy:     persistence.NewDataBlob([]byte(\"data1\"), constants.EncodingTypeThriftRW),\n\t\t\t\tShardID:    1,\n\t\t\t\tDomainID:   \"domain1\",\n\t\t\t\tWorkflowID: \"wfid1\",\n\t\t\t\tRunID:      \"r1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"not found - returns nil\",\n\t\t\tshardID:  1,\n\t\t\tdomainID: \"domain2\",\n\t\t\twfID:     \"wfid2\",\n\t\t\trID:      \"r2\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{},\n\t\t\t\t\terr:     errors.New(\"not found\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQuery: `SELECT data, data_encoding FROM executions WHERE ` +\n\t\t\t\t`shard_id = 1 and type = 11 and domain_id = domain2 and ` +\n\t\t\t\t`workflow_id = wfid2 and run_id = r2 and visibility_ts = 946684800000 and task_id = -1001`,\n\t\t\tmockFn: func(cl *gocql.MockClient) {\n\t\t\t\tcl.EXPECT().IsNotFoundError(errors.New(\"not found\")).Return(true).Times(1)\n\t\t\t},\n\t\t\twantPolicy: nil,\n\t\t\twantErr:    false,\n\t\t},\n\t\t{\n\t\t\tname:     \"query failed\",\n\t\t\tshardID:  1,\n\t\t\tdomainID: \"domain3\",\n\t\t\twfID:     \"wfid3\",\n\t\t\trID:      \"r3\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{},\n\t\t\t\t\terr:     errors.New(\"failed\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQuery: `SELECT data, data_encoding FROM executions WHERE ` +\n\t\t\t\t`shard_id = 1 and type = 11 and domain_id = domain3 and ` +\n\t\t\t\t`workflow_id = wfid3 and run_id = r3 and visibility_ts = 946684800000 and task_id = -1001`,\n\t\t\tmockFn: func(cl *gocql.MockClient) {\n\t\t\t\tcl.EXPECT().IsNotFoundError(errors.New(\"failed\")).Return(false).Times(1)\n\t\t\t},\n\t\t\twantPolicy: nil,\n\t\t\twantErr:    true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tcl := gocql.NewMockClient(ctrl)\n\n\t\t\tdb := NewCassandraDBFromSession(nil, tc.session, logger, nil, DbWithClient(cl))\n\n\t\t\tif tc.mockFn != nil {\n\t\t\t\ttc.mockFn(cl)\n\t\t\t}\n\n\t\t\tpolicy, err := db.SelectActiveClusterSelectionPolicy(context.Background(), tc.shardID, tc.domainID, tc.wfID, tc.rID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"SelectActiveClusterSelectionPolicy() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantPolicy, policy); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Policy mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQuery, tc.session.queries[0]); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteActiveClusterSelectionPolicy(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tshardID   int\n\t\tdomainID  string\n\t\twfID      string\n\t\trID       string\n\t\tsession   *fakeSession\n\t\twantQuery string\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tshardID:  1,\n\t\t\tdomainID: \"domain1\",\n\t\t\twfID:     \"wfid1\",\n\t\t\trID:      \"r1\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQuery: `DELETE FROM executions WHERE ` +\n\t\t\t\t`shard_id = 1 and type = 11 and domain_id = domain1 and ` +\n\t\t\t\t`workflow_id = wfid1 and run_id = r1 and visibility_ts = 946684800000 and task_id = -1001`,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"query failed\",\n\t\t\tshardID:  1,\n\t\t\tdomainID: \"domain2\",\n\t\t\twfID:     \"wfid2\",\n\t\t\trID:      \"r2\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{},\n\t\t\t\t\terr:     errors.New(\"failed\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQuery: `DELETE FROM executions WHERE ` +\n\t\t\t\t`shard_id = 1 and type = 11 and domain_id = domain2 and ` +\n\t\t\t\t`workflow_id = wfid2 and run_id = r2 and visibility_ts = 946684800000 and task_id = -1001`,\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tcl := gocql.NewMockClient(ctrl)\n\t\t\tdb := NewCassandraDBFromSession(nil, tc.session, logger, nil, DbWithClient(cl))\n\t\t\terr := db.DeleteActiveClusterSelectionPolicy(context.Background(), tc.shardID, tc.domainID, tc.wfID, tc.rID)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"DeleteActiveClusterSelectionPolicy() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQuery, tc.session.queries[0]); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_utils.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc executeCreateWorkflowBatchTransaction(\n\tctx context.Context,\n\tsession gocql.Session,\n\tbatch gocql.Batch,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tprevious := make(map[string]interface{})\n\tapplied, iter, err := session.MapExecuteBatchCAS(batch, previous)\n\tdefer func() {\n\t\tif iter != nil {\n\t\t\t_ = iter.Close()\n\t\t}\n\t}()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif applied {\n\t\treturn nil\n\t}\n\n\trequestRangeID := shardCondition.RangeID\n\trequestConditionalRunID := \"\"\n\tif currentWorkflowRequest.Condition != nil {\n\t\trequestConditionalRunID = currentWorkflowRequest.Condition.GetCurrentRunID()\n\t}\n\t// There can be two reasons why the query does not get applied. Either the RangeID has changed, or\n\t// the workflow is already started. Check the row info returned by Cassandra to figure out which one it is.\n\trangeIDMismatch := false\n\tactualRangeID := int64(0)\n\tcurrentExecutionAlreadyExists := false\n\tvar actualExecution map[string]interface{}\n\tvar actualExecutionFullRecord map[string]interface{}\n\trunIDMismatch := false\n\tactualCurrRunID := \"\"\n\tlastWriteVersionMismatch := false\n\tactualLastWriteVersion := int64(constants.EmptyVersion)\n\tstateMismatch := false\n\tactualState := int(0)\n\tconcreteExecutionAlreadyExists := false\n\tworkflowRequestAlreadyExists := false\n\tworkflowRequestID := \"\"\n\trequestRowType := int(0)\n\tvar allPrevious []map[string]interface{}\n\n\tfor {\n\t\trowType, ok := previous[\"type\"].(int)\n\t\tif !ok {\n\t\t\t// This should never happen, as all our rows have the type field.\n\t\t\tbreak\n\t\t}\n\t\trunID := previous[\"run_id\"].(gocql.UUID).String()\n\t\tif rowType == rowTypeShard {\n\t\t\tif actualRangeID, ok = previous[\"range_id\"].(int64); ok && actualRangeID != requestRangeID {\n\t\t\t\t// UpdateWorkflowExecution failed because rangeID was modified\n\t\t\t\trangeIDMismatch = true\n\t\t\t}\n\t\t} else if rowType == rowTypeExecution && runID == permanentRunID {\n\t\t\tif currentWorkflowRequest.WriteMode == nosqlplugin.CurrentWorkflowWriteModeInsert {\n\t\t\t\tcurrentExecutionAlreadyExists = true\n\t\t\t\tactualExecutionFullRecord = previous\n\t\t\t\tactualExecution, _ = previous[\"execution\"].(map[string]interface{})\n\t\t\t\tif actualExecution != nil {\n\t\t\t\t\tif previous[\"workflow_last_write_version\"] != nil {\n\t\t\t\t\t\tactualLastWriteVersion = previous[\"workflow_last_write_version\"].(int64)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else if currentWorkflowRequest.WriteMode == nosqlplugin.CurrentWorkflowWriteModeUpdate {\n\t\t\t\tif actualCurrRunID = previous[\"current_run_id\"].(gocql.UUID).String(); requestConditionalRunID != \"\" && actualCurrRunID != requestConditionalRunID {\n\t\t\t\t\trunIDMismatch = true\n\t\t\t\t}\n\t\t\t\tif currentWorkflowRequest.Condition != nil && currentWorkflowRequest.Condition.LastWriteVersion != nil {\n\t\t\t\t\tok := false\n\t\t\t\t\tif actualLastWriteVersion, ok = previous[\"workflow_last_write_version\"].(int64); ok && *currentWorkflowRequest.Condition.LastWriteVersion != actualLastWriteVersion {\n\t\t\t\t\t\tlastWriteVersionMismatch = true\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif currentWorkflowRequest.Condition != nil && currentWorkflowRequest.Condition.State != nil {\n\t\t\t\t\tok := false\n\t\t\t\t\tif actualState, ok = previous[\"workflow_state\"].(int); ok && *currentWorkflowRequest.Condition.State != actualState {\n\t\t\t\t\t\tstateMismatch = true\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t} else if rowType == rowTypeExecution && execution.RunID == runID {\n\t\t\tconcreteExecutionAlreadyExists = true\n\t\t\tif previous[\"workflow_last_write_version\"] != nil {\n\t\t\t\tactualLastWriteVersion = previous[\"workflow_last_write_version\"].(int64)\n\t\t\t}\n\t\t} else if isRequestRowType(rowType) {\n\t\t\tworkflowRequestAlreadyExists = true\n\t\t\tworkflowRequestID = runID\n\t\t\trequestRowType = rowType\n\t\t}\n\n\t\tallPrevious = append(allPrevious, previous)\n\t\tprevious = make(map[string]interface{})\n\t\tif !iter.MapScan(previous) {\n\t\t\t// Cassandra returns the actual row that caused a condition failure, so we should always return\n\t\t\t// from the checks above, but just in case.\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif rangeIDMismatch {\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tShardRangeIDNotMatch: common.Int64Ptr(actualRangeID),\n\t\t}\n\t}\n\tif workflowRequestAlreadyExists {\n\t\tresult := make(map[string]interface{})\n\t\tif err := session.Query(\n\t\t\ttemplateGetLatestWorkflowRequestQuery,\n\t\t\tshardCondition.ShardID,\n\t\t\trequestRowType,\n\t\t\texecution.DomainID,\n\t\t\texecution.WorkflowID,\n\t\t\tworkflowRequestID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t).WithContext(ctx).MapScan(result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\trunID, ok := result[\"current_run_id\"].(gocql.UUID)\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"corrupted data detected. DomainID: %v, WorkflowId: %v, RequestID: %v, RequestType: %v\", execution.DomainID, execution.WorkflowID, workflowRequestID, requestRowType)\n\t\t}\n\t\trequestType, err := fromRequestRowType(requestRowType)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\tRequestType: requestType,\n\t\t\t\tRunID:       runID.String(),\n\t\t\t},\n\t\t}\n\t}\n\t// CreateWorkflowExecution failed because there is already a current execution record for this workflow\n\tif currentExecutionAlreadyExists {\n\t\tif actualExecution != nil {\n\t\t\texecutionInfo, err := parseWorkflowExecutionInfo(actualExecutionFullRecord)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tmsg := fmt.Sprintf(\"Workflow execution already running. WorkflowId: %v, RunId: %v\", currentWorkflowRequest.Row.WorkflowID, executionInfo.RunID)\n\t\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tWorkflowExecutionAlreadyExists: &nosqlplugin.WorkflowExecutionAlreadyExists{\n\t\t\t\t\tOtherInfo:        msg,\n\t\t\t\t\tCreateRequestID:  executionInfo.CreateRequestID,\n\t\t\t\t\tRunID:            executionInfo.RunID,\n\t\t\t\t\tState:            executionInfo.State,\n\t\t\t\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\t\t\t\tLastWriteVersion: actualLastWriteVersion,\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\t\tmsg := fmt.Sprintf(\"Workflow execution already running. WorkflowId: %v\", currentWorkflowRequest.Row.WorkflowID)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tCurrentWorkflowConditionFailInfo: &msg,\n\t\t}\n\t}\n\tif runIDMismatch {\n\t\tmsg := fmt.Sprintf(\"Workflow execution creation condition failed by mismatch runID. WorkflowId: %v, Expected Current RunID: %v, Actual Current RunID: %v\",\n\t\t\tcurrentWorkflowRequest.Row.WorkflowID, currentWorkflowRequest.Condition.GetCurrentRunID(), actualCurrRunID)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tCurrentWorkflowConditionFailInfo: &msg,\n\t\t}\n\t}\n\tif lastWriteVersionMismatch {\n\t\tmsg := fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, Expected Version: %v, Actual Version: %v\",\n\t\t\tcurrentWorkflowRequest.Row.WorkflowID, *currentWorkflowRequest.Condition.LastWriteVersion, actualLastWriteVersion)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tCurrentWorkflowConditionFailInfo: &msg,\n\t\t}\n\t}\n\tif stateMismatch {\n\t\tmsg := fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, Expected State: %v, Actual State: %v\",\n\t\t\tcurrentWorkflowRequest.Row.WorkflowID, *currentWorkflowRequest.Condition.State, actualState)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tCurrentWorkflowConditionFailInfo: &msg,\n\t\t}\n\t}\n\tif concreteExecutionAlreadyExists {\n\t\tmsg := fmt.Sprintf(\"Workflow execution already running. WorkflowId: %v, RunId: %v\", execution.WorkflowID, execution.RunID)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tWorkflowExecutionAlreadyExists: &nosqlplugin.WorkflowExecutionAlreadyExists{\n\t\t\t\tOtherInfo:        msg,\n\t\t\t\tCreateRequestID:  execution.CreateRequestID,\n\t\t\t\tRunID:            execution.RunID,\n\t\t\t\tState:            execution.State,\n\t\t\t\tCloseStatus:      execution.CloseStatus,\n\t\t\t\tLastWriteVersion: actualLastWriteVersion,\n\t\t\t},\n\t\t}\n\t}\n\n\t// At this point we only know that the write was not applied.\n\tvar columns []string\n\tcolumnID := 0\n\tfor _, previous := range allPrevious {\n\t\tfor k, v := range previous {\n\t\t\tcolumns = append(columns, fmt.Sprintf(\"%v: %s=%v\", columnID, k, v))\n\t\t}\n\t\tcolumnID++\n\t}\n\treturn newUnknownConditionFailureReason(shardCondition.RangeID, columns)\n}\n\nfunc executeUpdateWorkflowBatchTransaction(\n\tctx context.Context,\n\tsession gocql.Session,\n\tbatch gocql.Batch,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\tpreviousNextEventIDCondition int64,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tprevious := make(map[string]interface{})\n\tapplied, iter, err := session.MapExecuteBatchCAS(batch, previous)\n\tdefer func() {\n\t\tif iter != nil {\n\t\t\t_ = iter.Close()\n\t\t}\n\t}()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif applied {\n\t\treturn nil\n\t}\n\n\trequestRunID := currentWorkflowRequest.Row.RunID\n\trequestRangeID := shardCondition.RangeID\n\trequestConditionalRunID := \"\"\n\tif currentWorkflowRequest.Condition != nil {\n\t\trequestConditionalRunID = currentWorkflowRequest.Condition.GetCurrentRunID()\n\t}\n\n\t// There can be three reasons why the query does not get applied: the RangeID has changed, or the next_event_id or current_run_id check failed.\n\t// Check the row info returned by Cassandra to figure out which one it is.\n\trangeIDMismatch := false\n\tactualRangeID := int64(0)\n\tnextEventIDMismatch := false\n\tactualNextEventID := int64(0)\n\trunIDMismatch := false\n\tactualCurrRunID := \"\"\n\tworkflowRequestAlreadyExists := false\n\tworkflowRequestID := \"\"\n\trequestRowType := int(0)\n\tvar allPrevious []map[string]interface{}\n\n\tfor {\n\t\trowType, ok := previous[\"type\"].(int)\n\t\tif !ok {\n\t\t\t// This should never happen, as all our rows have the type field.\n\t\t\tbreak\n\t\t}\n\n\t\trunID := previous[\"run_id\"].(gocql.UUID).String()\n\n\t\tif rowType == rowTypeShard {\n\t\t\tif actualRangeID, ok = previous[\"range_id\"].(int64); ok && actualRangeID != requestRangeID {\n\t\t\t\t// UpdateWorkflowExecution failed because rangeID was modified\n\t\t\t\trangeIDMismatch = true\n\t\t\t}\n\t\t} else if rowType == rowTypeExecution && runID == requestRunID {\n\t\t\tif actualNextEventID, ok = previous[\"next_event_id\"].(int64); ok && actualNextEventID != previousNextEventIDCondition {\n\t\t\t\t// UpdateWorkflowExecution failed because next event ID is unexpected\n\t\t\t\tnextEventIDMismatch = true\n\t\t\t}\n\t\t} else if rowType == rowTypeExecution && runID == permanentRunID {\n\t\t\t// UpdateWorkflowExecution failed because current_run_id is unexpected\n\t\t\tif actualCurrRunID = previous[\"current_run_id\"].(gocql.UUID).String(); requestConditionalRunID != \"\" && actualCurrRunID != requestConditionalRunID {\n\t\t\t\t// UpdateWorkflowExecution failed because next event ID is unexpected\n\t\t\t\trunIDMismatch = true\n\t\t\t}\n\t\t} else if isRequestRowType(rowType) {\n\t\t\tworkflowRequestAlreadyExists = true\n\t\t\tworkflowRequestID = runID\n\t\t\trequestRowType = rowType\n\t\t}\n\n\t\tallPrevious = append(allPrevious, previous)\n\t\tprevious = make(map[string]interface{})\n\t\tif !iter.MapScan(previous) {\n\t\t\t// Cassandra returns the actual row that caused a condition failure, so we should always return\n\t\t\t// from the checks above, but just in case.\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif rangeIDMismatch {\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tShardRangeIDNotMatch: common.Int64Ptr(actualRangeID),\n\t\t}\n\t}\n\n\tif workflowRequestAlreadyExists {\n\t\tresult := make(map[string]interface{})\n\t\tif err := session.Query(\n\t\t\ttemplateGetLatestWorkflowRequestQuery,\n\t\t\tshardCondition.ShardID,\n\t\t\trequestRowType,\n\t\t\tcurrentWorkflowRequest.Row.DomainID,\n\t\t\tcurrentWorkflowRequest.Row.WorkflowID,\n\t\t\tworkflowRequestID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t).WithContext(ctx).MapScan(result); err != nil {\n\t\t\treturn err\n\t\t}\n\t\trunID, ok := result[\"current_run_id\"].(gocql.UUID)\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"corrupted data detected. DomainID: %v, WorkflowId: %v, RequestID: %v, RequestType: %v\", currentWorkflowRequest.Row.DomainID, currentWorkflowRequest.Row.WorkflowID, workflowRequestID, requestRowType)\n\t\t}\n\t\trequestType, err := fromRequestRowType(requestRowType)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\tRequestType: requestType,\n\t\t\t\tRunID:       runID.String(),\n\t\t\t},\n\t\t}\n\t}\n\n\tif runIDMismatch {\n\t\tmsg := fmt.Sprintf(\"Failed to update mutable state. requestConditionalRunID: %v, Actual Value: %v\",\n\t\t\trequestConditionalRunID, actualCurrRunID)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tCurrentWorkflowConditionFailInfo: &msg,\n\t\t}\n\t}\n\n\tif nextEventIDMismatch {\n\t\tmsg := fmt.Sprintf(\"Failed to update mutable state. previousNextEventIDCondition: %v, actualNextEventID: %v, Request Current RunID: %v\",\n\t\t\tpreviousNextEventIDCondition, actualNextEventID, requestRunID)\n\t\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\tUnknownConditionFailureDetails: &msg,\n\t\t}\n\t}\n\n\t// At this point we only know that the write was not applied.\n\tvar columns []string\n\tcolumnID := 0\n\tfor _, previous := range allPrevious {\n\t\tfor k, v := range previous {\n\t\t\tcolumns = append(columns, fmt.Sprintf(\"%v: %s=%v\", columnID, k, v))\n\t\t}\n\t\tcolumnID++\n\t}\n\n\tmsg := fmt.Sprintf(\"Failed to update mutable state. ShardID: %v, RangeID: %v, previousNextEventIDCondition: %v, requestConditionalRunID: %v, columns: (%v)\",\n\t\tshardCondition.ShardID, requestRangeID, previousNextEventIDCondition, requestConditionalRunID, strings.Join(columns, \",\"))\n\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\tUnknownConditionFailureDetails: &msg,\n\t}\n}\n\nfunc newUnknownConditionFailureReason(\n\trangeID int64,\n\tcolumns []string,\n) *nosqlplugin.WorkflowOperationConditionFailure {\n\tmsg := fmt.Sprintf(\"Failed to operate on workflow execution.  Request RangeID: %v, columns: (%v)\",\n\t\trangeID, strings.Join(columns, \",\"))\n\n\treturn &nosqlplugin.WorkflowOperationConditionFailure{\n\t\tUnknownConditionFailureDetails: &msg,\n\t}\n}\n\nfunc assertShardRangeID(batch gocql.Batch, shardID int, rangeID int64, timeStamp time.Time) {\n\tbatch.Query(templateUpdateLeaseQuery,\n\t\trangeID,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeShard,\n\t\trowTypeShardDomainID,\n\t\trowTypeShardWorkflowID,\n\t\trowTypeShardRunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeShardTaskID,\n\t\trangeID,\n\t)\n}\n\nfunc createTasksByCategory(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\ttimeStamp time.Time,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n) {\n\tfor c, tasks := range tasksByCategory {\n\t\tswitch c.ID() {\n\t\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\t\tcreateTransferTasks(batch, shardID, domainID, workflowID, tasks, timeStamp)\n\t\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\t\tcreateTimerTasks(batch, shardID, domainID, workflowID, tasks, timeStamp)\n\t\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\t\tcreateReplicationTasks(batch, shardID, domainID, workflowID, tasks, timeStamp)\n\t\t}\n\t}\n\n\t// TODO: implementing writing tasks for other categories\n}\n\nfunc createTimerTasks(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\ttimerTasks []*nosqlplugin.HistoryMigrationTask,\n\ttimeStamp time.Time,\n) error {\n\tfor _, timer := range timerTasks {\n\t\ttask := timer.Timer\n\t\ttaskBlob, taskEncoding := persistence.FromDataBlob(timer.Task)\n\t\t// Ignoring possible type cast errors.\n\t\tts := persistence.UnixNanoToDBTimestamp(task.VisibilityTimestamp.UnixNano())\n\n\t\tbatch.Query(templateCreateTimerTaskQuery,\n\t\t\tshardID,\n\t\t\trowTypeTimerTask,\n\t\t\trowTypeTimerDomainID,\n\t\t\trowTypeTimerWorkflowID,\n\t\t\trowTypeTimerRunID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\ttask.RunID,\n\t\t\tts,\n\t\t\ttask.TaskID,\n\t\t\ttask.TaskType,\n\t\t\ttask.TimeoutType,\n\t\t\ttask.EventID,\n\t\t\ttask.ScheduleAttempt,\n\t\t\ttask.Version,\n\t\t\ttask.TaskList,\n\t\t\ttaskBlob,\n\t\t\ttaskEncoding,\n\t\t\tts,\n\t\t\ttask.TaskID,\n\t\t\ttimeStamp,\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc createReplicationTasks(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\treplicationTasks []*nosqlplugin.HistoryMigrationTask,\n\ttimeStamp time.Time,\n) {\n\tfor _, replication := range replicationTasks {\n\t\ttask := replication.Replication\n\t\ttaskBlob, taskEncoding := persistence.FromDataBlob(replication.Task)\n\t\tbatch.Query(templateCreateReplicationTaskQuery,\n\t\t\tshardID,\n\t\t\trowTypeReplicationTask,\n\t\t\trowTypeReplicationDomainID,\n\t\t\trowTypeReplicationWorkflowID,\n\t\t\trowTypeReplicationRunID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\ttask.RunID,\n\t\t\ttask.TaskID,\n\t\t\ttask.TaskType,\n\t\t\ttask.FirstEventID,\n\t\t\ttask.NextEventID,\n\t\t\ttask.Version,\n\t\t\ttask.ScheduledID,\n\t\t\tpersistence.EventStoreVersion,\n\t\t\ttask.BranchToken,\n\t\t\tpersistence.EventStoreVersion,\n\t\t\ttask.NewRunBranchToken,\n\t\t\ttask.CreationTime.UnixNano(),\n\t\t\ttaskBlob,\n\t\t\ttaskEncoding,\n\t\t\t// NOTE: use a constant here instead of task.VisibilityTimestamp so that we can query tasks with the same visibilityTimestamp\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\ttask.TaskID,\n\t\t\ttimeStamp,\n\t\t)\n\t}\n}\n\nfunc createTransferTasks(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\ttransferTasks []*nosqlplugin.HistoryMigrationTask,\n\ttimeStamp time.Time,\n) {\n\tfor _, transfer := range transferTasks {\n\t\ttask := transfer.Transfer\n\t\ttaskBlob, taskEncoding := persistence.FromDataBlob(transfer.Task)\n\t\tbatch.Query(templateCreateTransferTaskQuery,\n\t\t\tshardID,\n\t\t\trowTypeTransferTask,\n\t\t\trowTypeTransferDomainID,\n\t\t\trowTypeTransferWorkflowID,\n\t\t\trowTypeTransferRunID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\ttask.RunID,\n\t\t\ttask.VisibilityTimestamp,\n\t\t\ttask.TaskID,\n\t\t\ttask.TargetDomainID,\n\t\t\ttask.TargetDomainIDs,\n\t\t\ttask.TargetWorkflowID,\n\t\t\ttask.TargetRunID,\n\t\t\ttask.TargetChildWorkflowOnly,\n\t\t\ttask.TaskList,\n\t\t\ttask.TaskType,\n\t\t\ttask.ScheduleID,\n\t\t\ttask.RecordVisibility,\n\t\t\ttask.Version,\n\t\t\ttask.OriginalTaskList,\n\t\t\tint32(task.OriginalTaskListKind),\n\t\t\ttaskBlob,\n\t\t\ttaskEncoding,\n\t\t\t// NOTE: use a constant here instead of task.VisibilityTimestamp so that we can query tasks with the same visibilityTimestamp\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\ttask.TaskID,\n\t\t\ttimeStamp,\n\t\t)\n\t}\n}\n\nfunc resetSignalsRequested(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tsignalReqIDs []string,\n\ttimeStamp time.Time,\n) error {\n\tbatch.Query(templateResetSignalRequestedQuery,\n\t\tsignalReqIDs,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc updateSignalsRequested(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tsignalReqIDs []string,\n\tdeleteSignalReqIDs []string,\n\ttimeStamp time.Time,\n) error {\n\tif len(signalReqIDs) > 0 {\n\t\tbatch.Query(templateUpdateSignalRequestedQuery,\n\t\t\tsignalReqIDs,\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\tif len(deleteSignalReqIDs) > 0 {\n\t\tbatch.Query(templateDeleteWorkflowExecutionSignalRequestedQuery,\n\t\t\tdeleteSignalReqIDs,\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\nfunc resetSignalInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tsignalInfos map[int64]*persistence.SignalInfo,\n\ttimeStamp time.Time,\n) error {\n\tbatch.Query(templateResetSignalInfoQuery,\n\t\tresetSignalInfoMap(signalInfos),\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc resetSignalInfoMap(signalInfos map[int64]*persistence.SignalInfo) map[int64]map[string]interface{} {\n\tsMap := make(map[int64]map[string]interface{})\n\tfor _, s := range signalInfos {\n\t\tsInfo := make(map[string]interface{})\n\t\tsInfo[\"version\"] = s.Version\n\t\tsInfo[\"initiated_id\"] = s.InitiatedID\n\t\tsInfo[\"initiated_event_batch_id\"] = s.InitiatedEventBatchID\n\t\tsInfo[\"signal_request_id\"] = s.SignalRequestID\n\t\tsInfo[\"signal_name\"] = s.SignalName\n\t\tsInfo[\"input\"] = s.Input\n\t\tsInfo[\"control\"] = s.Control\n\t\tsMap[s.InitiatedID] = sInfo\n\t}\n\n\treturn sMap\n}\n\nfunc updateSignalInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tsignalInfos map[int64]*persistence.SignalInfo,\n\tdeleteInfos []int64,\n\ttimeStamp time.Time,\n) error {\n\tfor _, c := range signalInfos {\n\t\tbatch.Query(templateUpdateSignalInfoQuery,\n\t\t\tc.InitiatedID,\n\t\t\tc.Version,\n\t\t\tc.InitiatedID,\n\t\t\tc.InitiatedEventBatchID,\n\t\t\tc.SignalRequestID,\n\t\t\tc.SignalName,\n\t\t\tc.Input,\n\t\t\tc.Control,\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\t// deleteInfos are the initiatedIDs for SignalInfo being deleted\n\tfor _, deleteInfo := range deleteInfos {\n\t\tbatch.Query(templateDeleteSignalInfoQuery,\n\t\t\tdeleteInfo,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\nfunc resetRequestCancelInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\trequestCancelInfos map[int64]*persistence.RequestCancelInfo,\n\ttimeStamp time.Time,\n) error {\n\tbatch.Query(templateResetRequestCancelInfoQuery,\n\t\tresetRequestCancelInfoMap(requestCancelInfos),\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc resetRequestCancelInfoMap(requestCancelInfos map[int64]*persistence.RequestCancelInfo) map[int64]map[string]interface{} {\n\trcMap := make(map[int64]map[string]interface{})\n\tfor _, rc := range requestCancelInfos {\n\t\trcInfo := make(map[string]interface{})\n\t\trcInfo[\"version\"] = rc.Version\n\t\trcInfo[\"initiated_id\"] = rc.InitiatedID\n\t\trcInfo[\"initiated_event_batch_id\"] = rc.InitiatedEventBatchID\n\t\trcInfo[\"cancel_request_id\"] = rc.CancelRequestID\n\t\trcMap[rc.InitiatedID] = rcInfo\n\t}\n\n\treturn rcMap\n}\n\nfunc updateRequestCancelInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\trequestCancelInfos map[int64]*persistence.RequestCancelInfo,\n\tdeleteInfos []int64,\n\ttimeStamp time.Time,\n) error {\n\tfor _, c := range requestCancelInfos {\n\t\tbatch.Query(templateUpdateRequestCancelInfoQuery,\n\t\t\tc.InitiatedID,\n\t\t\tc.Version,\n\t\t\tc.InitiatedID,\n\t\t\tc.InitiatedEventBatchID,\n\t\t\tc.CancelRequestID,\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\t// deleteInfos are the initiatedIDs for RequestCancelInfo being deleted\n\tfor _, deleteInfo := range deleteInfos {\n\t\tbatch.Query(templateDeleteRequestCancelInfoQuery,\n\t\t\tdeleteInfo,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\nfunc resetChildExecutionInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tchildExecutionInfos map[int64]*persistence.InternalChildExecutionInfo,\n\ttimeStamp time.Time,\n) error {\n\tinfoMap, err := resetChildExecutionInfoMap(childExecutionInfos)\n\tif err != nil {\n\t\treturn err\n\t}\n\tbatch.Query(templateResetChildExecutionInfoQuery,\n\t\tinfoMap,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc resetChildExecutionInfoMap(childExecutionInfos map[int64]*persistence.InternalChildExecutionInfo) (map[int64]map[string]interface{}, error) {\n\tcMap := make(map[int64]map[string]interface{})\n\tfor _, c := range childExecutionInfos {\n\t\tcInfo := make(map[string]interface{})\n\t\tcInfo[\"version\"] = c.Version\n\t\tcInfo[\"event_data_encoding\"] = c.InitiatedEvent.GetEncodingString()\n\t\tcInfo[\"initiated_id\"] = c.InitiatedID\n\t\tcInfo[\"initiated_event_batch_id\"] = c.InitiatedEventBatchID\n\t\tcInfo[\"initiated_event\"] = c.InitiatedEvent.Data\n\t\tcInfo[\"started_id\"] = c.StartedID\n\t\tcInfo[\"started_event\"] = c.StartedEvent.Data\n\t\tcInfo[\"create_request_id\"] = c.CreateRequestID\n\t\tcInfo[\"started_workflow_id\"] = c.StartedWorkflowID\n\t\tstartedRunID := emptyRunID\n\t\tif c.StartedRunID != \"\" {\n\t\t\tstartedRunID = c.StartedRunID\n\t\t}\n\t\tcInfo[\"started_run_id\"] = startedRunID\n\t\tcInfo[\"domain_id\"] = c.DomainID\n\t\tcInfo[\"domain_name\"] = c.DomainNameDEPRECATED\n\t\tcInfo[\"workflow_type_name\"] = c.WorkflowTypeName\n\t\tcInfo[\"parent_close_policy\"] = int32(c.ParentClosePolicy)\n\n\t\tcMap[c.InitiatedID] = cInfo\n\t}\n\n\treturn cMap, nil\n}\n\nfunc updateChildExecutionInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tchildExecutionInfos map[int64]*persistence.InternalChildExecutionInfo,\n\tdeleteInfos []int64,\n\ttimeStamp time.Time,\n) error {\n\tfor _, c := range childExecutionInfos {\n\t\tbatch.Query(templateUpdateChildExecutionInfoQuery,\n\t\t\tc.InitiatedID,\n\t\t\tc.Version,\n\t\t\tc.InitiatedID,\n\t\t\tc.InitiatedEventBatchID,\n\t\t\tc.InitiatedEvent.Data,\n\t\t\tc.StartedID,\n\t\t\tc.StartedWorkflowID,\n\t\t\tc.StartedRunID,\n\t\t\tc.StartedEvent.Data,\n\t\t\tc.CreateRequestID,\n\t\t\tc.InitiatedEvent.GetEncodingString(),\n\t\t\tc.DomainID,\n\t\t\tc.DomainNameDEPRECATED,\n\t\t\tc.WorkflowTypeName,\n\t\t\tint32(c.ParentClosePolicy),\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\t// deleteInfos are the initiatedIDs for ChildInfo being deleted\n\tfor _, deleteInfo := range deleteInfos {\n\t\tbatch.Query(templateDeleteChildExecutionInfoQuery,\n\t\t\tdeleteInfo,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\nfunc resetTimerInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\ttimerInfos map[string]*persistence.TimerInfo,\n\ttimeStamp time.Time,\n) error {\n\tbatch.Query(templateResetTimerInfoQuery,\n\t\tresetTimerInfoMap(timerInfos),\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc resetTimerInfoMap(timerInfos map[string]*persistence.TimerInfo) map[string]map[string]interface{} {\n\ttMap := make(map[string]map[string]interface{})\n\tfor _, t := range timerInfos {\n\t\ttInfo := make(map[string]interface{})\n\t\ttInfo[\"version\"] = t.Version\n\t\ttInfo[\"timer_id\"] = t.TimerID\n\t\ttInfo[\"started_id\"] = t.StartedID\n\t\ttInfo[\"expiry_time\"] = t.ExpiryTime\n\t\t// task_id is a misleading variable, it actually serves\n\t\t// the purpose of indicating whether a timer task is\n\t\t// generated for this timer info\n\t\ttInfo[\"task_id\"] = t.TaskStatus\n\n\t\ttMap[t.TimerID] = tInfo\n\t}\n\n\treturn tMap\n}\n\nfunc updateTimerInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\ttimerInfos map[string]*persistence.TimerInfo,\n\tdeleteInfos []string,\n\ttimeStamp time.Time,\n) error {\n\tfor _, timerInfo := range timerInfos {\n\t\tbatch.Query(templateUpdateTimerInfoQuery,\n\t\t\ttimerInfo.TimerID,\n\t\t\ttimerInfo.Version,\n\t\t\ttimerInfo.TimerID,\n\t\t\ttimerInfo.StartedID,\n\t\t\ttimerInfo.ExpiryTime,\n\t\t\ttimerInfo.TaskStatus,\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\tfor _, deleteInfo := range deleteInfos {\n\t\tbatch.Query(templateDeleteTimerInfoQuery,\n\t\t\tdeleteInfo,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\nfunc resetActivityInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tactivityInfos map[int64]*persistence.InternalActivityInfo,\n\ttimeStamp time.Time,\n) error {\n\tinfoMap, err := resetActivityInfoMap(activityInfos)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tbatch.Query(templateResetActivityInfoQuery,\n\t\tinfoMap,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc resetActivityInfoMap(activityInfos map[int64]*persistence.InternalActivityInfo) (map[int64]map[string]interface{}, error) {\n\taMap := make(map[int64]map[string]interface{})\n\tfor _, a := range activityInfos {\n\t\taInfo := make(map[string]interface{})\n\t\taInfo[\"version\"] = a.Version\n\t\taInfo[\"event_data_encoding\"] = a.ScheduledEvent.GetEncodingString()\n\t\taInfo[\"schedule_id\"] = a.ScheduleID\n\t\taInfo[\"scheduled_event_batch_id\"] = a.ScheduledEventBatchID\n\t\taInfo[\"scheduled_event\"] = a.ScheduledEvent.Data\n\t\taInfo[\"scheduled_time\"] = a.ScheduledTime\n\t\taInfo[\"started_id\"] = a.StartedID\n\t\taInfo[\"started_event\"] = a.StartedEvent.Data\n\t\taInfo[\"started_time\"] = a.StartedTime\n\t\taInfo[\"activity_id\"] = a.ActivityID\n\t\taInfo[\"request_id\"] = a.RequestID\n\t\taInfo[\"details\"] = a.Details\n\t\taInfo[\"schedule_to_start_timeout\"] = int32(a.ScheduleToStartTimeout.Seconds())\n\t\taInfo[\"schedule_to_close_timeout\"] = int32(a.ScheduleToCloseTimeout.Seconds())\n\t\taInfo[\"start_to_close_timeout\"] = int32(a.StartToCloseTimeout.Seconds())\n\t\taInfo[\"heart_beat_timeout\"] = int32(a.HeartbeatTimeout.Seconds())\n\t\taInfo[\"cancel_requested\"] = a.CancelRequested\n\t\taInfo[\"cancel_request_id\"] = a.CancelRequestID\n\t\taInfo[\"last_hb_updated_time\"] = a.LastHeartBeatUpdatedTime\n\t\taInfo[\"timer_task_status\"] = a.TimerTaskStatus\n\t\taInfo[\"attempt\"] = a.Attempt\n\t\taInfo[\"task_list\"] = a.TaskList\n\t\taInfo[\"started_identity\"] = a.StartedIdentity\n\t\taInfo[\"has_retry_policy\"] = a.HasRetryPolicy\n\t\taInfo[\"init_interval\"] = int32(a.InitialInterval.Seconds())\n\t\taInfo[\"backoff_coefficient\"] = a.BackoffCoefficient\n\t\taInfo[\"max_interval\"] = int32(a.MaximumInterval.Seconds())\n\t\taInfo[\"expiration_time\"] = a.ExpirationTime\n\t\taInfo[\"max_attempts\"] = a.MaximumAttempts\n\t\taInfo[\"non_retriable_errors\"] = a.NonRetriableErrors\n\t\taInfo[\"last_failure_reason\"] = a.LastFailureReason\n\t\taInfo[\"last_worker_identity\"] = a.LastWorkerIdentity\n\t\taInfo[\"last_failure_details\"] = a.LastFailureDetails\n\n\t\taMap[a.ScheduleID] = aInfo\n\t}\n\n\treturn aMap, nil\n}\n\nfunc updateActivityInfos(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tactivityInfos map[int64]*persistence.InternalActivityInfo,\n\tdeleteInfos []int64,\n\ttimeStamp time.Time,\n) error {\n\tfor _, a := range activityInfos {\n\t\tbatch.Query(templateUpdateActivityInfoQuery,\n\t\t\ta.ScheduleID,\n\t\t\ta.Version,\n\t\t\ta.ScheduleID,\n\t\t\ta.ScheduledEventBatchID,\n\t\t\ta.ScheduledEvent.Data,\n\t\t\ta.ScheduledTime,\n\t\t\ta.StartedID,\n\t\t\ta.StartedEvent.Data,\n\t\t\ta.StartedTime,\n\t\t\ta.ActivityID,\n\t\t\ta.RequestID,\n\t\t\ta.Details,\n\t\t\tint32(a.ScheduleToStartTimeout.Seconds()),\n\t\t\tint32(a.ScheduleToCloseTimeout.Seconds()),\n\t\t\tint32(a.StartToCloseTimeout.Seconds()),\n\t\t\tint32(a.HeartbeatTimeout.Seconds()),\n\t\t\ta.CancelRequested,\n\t\t\ta.CancelRequestID,\n\t\t\ta.LastHeartBeatUpdatedTime,\n\t\t\ta.TimerTaskStatus,\n\t\t\ta.Attempt,\n\t\t\ta.TaskList,\n\t\t\tint32(a.TaskListKind),\n\t\t\ta.StartedIdentity,\n\t\t\ta.HasRetryPolicy,\n\t\t\tint32(a.InitialInterval.Seconds()),\n\t\t\ta.BackoffCoefficient,\n\t\t\tint32(a.MaximumInterval.Seconds()),\n\t\t\ta.ExpirationTime,\n\t\t\ta.MaximumAttempts,\n\t\t\ta.NonRetriableErrors,\n\t\t\ta.LastFailureReason,\n\t\t\ta.LastWorkerIdentity,\n\t\t\ta.LastFailureDetails,\n\t\t\ta.ScheduledEvent.GetEncodingString(),\n\t\t\ttimeStamp,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\n\tfor _, deleteInfo := range deleteInfos {\n\t\tbatch.Query(templateDeleteActivityInfoQuery,\n\t\t\tdeleteInfo,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID)\n\t}\n\treturn nil\n}\n\n// NOTE: not sure we still need it. We keep the behavior for safe during refactoring\n// In theory we can just return the input as output\n// TODO: if possible, remove it in the future or add more comment of why we need this conversion\nfunc convertToCassandraTimestamp(in time.Time) time.Time {\n\treturn time.Unix(0, persistence.DBTimestampToUnixNano(persistence.UnixNanoToDBTimestamp(in.UnixNano())))\n}\n\nfunc getNextPageToken(iter gocql.Iter) []byte {\n\tnextPageToken := iter.PageState()\n\tnewPageToken := make([]byte, len(nextPageToken))\n\tcopy(newPageToken, nextPageToken)\n\treturn newPageToken\n}\n\nfunc createWorkflowExecutionWithMergeMaps(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttimeStamp time.Time,\n) error {\n\terr := createWorkflowExecution(batch, shardID, domainID, workflowID, execution, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif execution.EventBufferWriteMode != nosqlplugin.EventBufferWriteModeNone {\n\t\treturn fmt.Errorf(\"should only support EventBufferWriteModeNone\")\n\t}\n\n\tif execution.MapsWriteMode != nosqlplugin.WorkflowExecutionMapsWriteModeCreate {\n\t\treturn fmt.Errorf(\"should only support WorkflowExecutionMapsWriteModeCreate\")\n\t}\n\n\terr = updateActivityInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ActivityInfos, nil, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateTimerInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.TimerInfos, nil, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateChildExecutionInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ChildWorkflowInfos, nil, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateRequestCancelInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.RequestCancelInfos, nil, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateSignalInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalInfos, nil, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn updateSignalsRequested(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalRequestedIDs, nil, timeStamp)\n}\n\nfunc resetWorkflowExecutionAndMapsAndEventBuffer(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttimeStamp time.Time,\n) error {\n\terr := updateWorkflowExecution(batch, shardID, domainID, workflowID, execution, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif execution.EventBufferWriteMode != nosqlplugin.EventBufferWriteModeClear {\n\t\treturn fmt.Errorf(\"should only support EventBufferWriteModeClear\")\n\t}\n\terr = deleteBufferedEvents(batch, shardID, domainID, workflowID, execution.RunID, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif execution.MapsWriteMode != nosqlplugin.WorkflowExecutionMapsWriteModeReset {\n\t\treturn fmt.Errorf(\"should only support WorkflowExecutionMapsWriteModeReset\")\n\t}\n\n\t// This is another category of execution update where only the non-frozen column types in\n\t// Cassandra are updated to a previous state in the Execution Update flow.\n\terr = resetActivityInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ActivityInfos, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = resetTimerInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.TimerInfos, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = resetChildExecutionInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ChildWorkflowInfos, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = resetRequestCancelInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.RequestCancelInfos, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = resetSignalInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalInfos, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn resetSignalsRequested(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalRequestedIDs, timeStamp)\n}\n\nfunc appendBufferedEvents(\n\tbatch gocql.Batch,\n\tnewBufferedEvents *persistence.DataBlob,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\ttimeStamp time.Time,\n) error {\n\tvalues := make(map[string]interface{})\n\tvalues[\"encoding_type\"] = newBufferedEvents.Encoding\n\tvalues[\"version\"] = int64(0)\n\tvalues[\"data\"] = newBufferedEvents.Data\n\tnewEventValues := []map[string]interface{}{values}\n\tbatch.Query(templateAppendBufferedEventsQuery,\n\t\tnewEventValues,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc deleteBufferedEvents(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\ttimeStamp time.Time,\n) error {\n\tbatch.Query(templateDeleteBufferedEventsQuery,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID)\n\treturn nil\n}\n\nfunc updateWorkflowExecutionAndEventBufferWithMergeAndDeleteMaps(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttimeStamp time.Time,\n) error {\n\terr := updateWorkflowExecution(batch, shardID, domainID, workflowID, execution, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif execution.EventBufferWriteMode == nosqlplugin.EventBufferWriteModeClear {\n\t\terr = deleteBufferedEvents(batch, shardID, domainID, workflowID, execution.RunID, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t} else if execution.EventBufferWriteMode == nosqlplugin.EventBufferWriteModeAppend {\n\t\terr = appendBufferedEvents(batch, execution.NewBufferedEventBatch, shardID, domainID, workflowID, execution.RunID, timeStamp)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif execution.MapsWriteMode != nosqlplugin.WorkflowExecutionMapsWriteModeUpdate {\n\t\treturn fmt.Errorf(\"should only support WorkflowExecutionMapsWriteModeUpdate\")\n\t}\n\n\t// In certain cases, some of the execution update cycles update particular columns asynchronously before reaching the final cycle.\n\t// Each of these functions are updating a non-frozen column type in Cassandra table.\n\terr = updateActivityInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ActivityInfos, execution.ActivityInfoKeysToDelete, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateTimerInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.TimerInfos, execution.TimerInfoKeysToDelete, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateChildExecutionInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.ChildWorkflowInfos, execution.ChildWorkflowInfoKeysToDelete, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateRequestCancelInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.RequestCancelInfos, execution.RequestCancelInfoKeysToDelete, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = updateSignalInfos(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalInfos, execution.SignalInfoKeysToDelete, timeStamp)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn updateSignalsRequested(batch, shardID, domainID, workflowID, execution.RunID, execution.SignalRequestedIDs, execution.SignalRequestedIDsKeysToDelete, timeStamp)\n}\n\n// updateWorkflowExecution is responsible for updating the execution state in different cycles until the\n// Status changes to close at the Final update cycle. Information is updated linearly, and synchronization usually occurs in every cycle.\nfunc updateWorkflowExecution(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttimeStamp time.Time,\n) error {\n\texecution.StartTimestamp = convertToCassandraTimestamp(execution.StartTimestamp)\n\texecution.LastUpdatedTimestamp = convertToCassandraTimestamp(execution.LastUpdatedTimestamp)\n\n\tbatch.Query(templateUpdateWorkflowExecutionWithVersionHistoriesQuery,\n\t\tdomainID,\n\t\tworkflowID,\n\t\texecution.RunID,\n\t\texecution.FirstExecutionRunID,\n\t\texecution.ParentDomainID,\n\t\texecution.ParentWorkflowID,\n\t\texecution.ParentRunID,\n\t\texecution.InitiatedID,\n\t\texecution.CompletionEventBatchID,\n\t\texecution.CompletionEvent.Data,\n\t\texecution.CompletionEvent.GetEncodingString(),\n\t\texecution.TaskList,\n\t\tint32(execution.TaskListKind),\n\t\texecution.WorkflowTypeName,\n\t\tint32(execution.WorkflowTimeout.Seconds()),\n\t\tint32(execution.DecisionStartToCloseTimeout.Seconds()),\n\t\texecution.ExecutionContext,\n\t\texecution.State,\n\t\texecution.CloseStatus,\n\t\texecution.LastFirstEventID,\n\t\texecution.LastEventTaskID,\n\t\texecution.NextEventID,\n\t\texecution.LastProcessedEvent,\n\t\texecution.StartTimestamp,\n\t\texecution.LastUpdatedTimestamp,\n\t\texecution.CreateRequestID,\n\t\texecution.SignalCount,\n\t\texecution.HistorySize,\n\t\texecution.DecisionVersion,\n\t\texecution.DecisionScheduleID,\n\t\texecution.DecisionStartedID,\n\t\texecution.DecisionRequestID,\n\t\tint32(execution.DecisionTimeout.Seconds()),\n\t\texecution.DecisionAttempt,\n\t\texecution.DecisionStartedTimestamp.UnixNano(),\n\t\texecution.DecisionScheduledTimestamp.UnixNano(),\n\t\texecution.DecisionOriginalScheduledTimestamp.UnixNano(),\n\t\texecution.CancelRequested,\n\t\texecution.CancelRequestID,\n\t\texecution.StickyTaskList,\n\t\tint32(execution.StickyScheduleToStartTimeout.Seconds()),\n\t\texecution.ClientLibraryVersion,\n\t\texecution.ClientFeatureVersion,\n\t\texecution.ClientImpl,\n\t\texecution.AutoResetPoints.Data,\n\t\texecution.AutoResetPoints.GetEncoding(),\n\t\texecution.Attempt,\n\t\texecution.HasRetryPolicy,\n\t\tint32(execution.InitialInterval.Seconds()),\n\t\texecution.BackoffCoefficient,\n\t\tint32(execution.MaximumInterval.Seconds()),\n\t\texecution.ExpirationTime,\n\t\texecution.MaximumAttempts,\n\t\texecution.NonRetriableErrors,\n\t\tpersistence.EventStoreVersion,\n\t\texecution.BranchToken,\n\t\texecution.CronSchedule,\n\t\tint32(execution.CronOverlapPolicy),\n\t\tint32(execution.ExpirationInterval.Seconds()),\n\t\texecution.SearchAttributes,\n\t\texecution.Memo,\n\t\texecution.PartitionConfig,\n\t\texecution.ActiveClusterSelectionPolicy.GetData(),\n\t\texecution.ActiveClusterSelectionPolicy.GetEncodingString(),\n\t\texecution.NextEventID,\n\t\texecution.VersionHistories.Data,\n\t\texecution.VersionHistories.GetEncodingString(),\n\t\texecution.Checksums.Version,\n\t\texecution.Checksums.Flavor,\n\t\texecution.Checksums.Value,\n\t\texecution.LastWriteVersion,\n\t\texecution.State,\n\t\ttimeStamp,\n\t\tshardID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\texecution.RunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t\texecution.PreviousNextEventIDCondition)\n\n\treturn nil\n}\n\nfunc createWorkflowExecution(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttimeStamp time.Time,\n) error {\n\texecution.StartTimestamp = convertToCassandraTimestamp(execution.StartTimestamp)\n\texecution.LastUpdatedTimestamp = convertToCassandraTimestamp(execution.LastUpdatedTimestamp)\n\n\tbatch.Query(templateCreateWorkflowExecutionWithVersionHistoriesQuery,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\texecution.RunID,\n\t\trowTypeExecution,\n\t\tdomainID,\n\t\tworkflowID,\n\t\texecution.RunID,\n\t\texecution.FirstExecutionRunID,\n\t\texecution.ParentDomainID,\n\t\texecution.ParentWorkflowID,\n\t\texecution.ParentRunID,\n\t\texecution.InitiatedID,\n\t\texecution.CompletionEventBatchID,\n\t\texecution.CompletionEvent.Data,\n\t\texecution.CompletionEvent.GetEncodingString(),\n\t\texecution.TaskList,\n\t\tint32(execution.TaskListKind),\n\t\texecution.WorkflowTypeName,\n\t\tint32(execution.WorkflowTimeout.Seconds()),\n\t\tint32(execution.DecisionStartToCloseTimeout.Seconds()),\n\t\texecution.ExecutionContext,\n\t\texecution.State,\n\t\texecution.CloseStatus,\n\t\texecution.LastFirstEventID,\n\t\texecution.LastEventTaskID,\n\t\texecution.NextEventID,\n\t\texecution.LastProcessedEvent,\n\t\texecution.StartTimestamp,\n\t\texecution.LastUpdatedTimestamp,\n\t\texecution.CreateRequestID,\n\t\texecution.SignalCount,\n\t\texecution.HistorySize,\n\t\texecution.DecisionVersion,\n\t\texecution.DecisionScheduleID,\n\t\texecution.DecisionStartedID,\n\t\texecution.DecisionRequestID,\n\t\tint32(execution.DecisionTimeout.Seconds()),\n\t\texecution.DecisionAttempt,\n\t\texecution.DecisionStartedTimestamp.UnixNano(),\n\t\texecution.DecisionScheduledTimestamp.UnixNano(),\n\t\texecution.DecisionOriginalScheduledTimestamp.UnixNano(),\n\t\texecution.CancelRequested,\n\t\texecution.CancelRequestID,\n\t\texecution.StickyTaskList,\n\t\tint32(execution.StickyScheduleToStartTimeout.Seconds()),\n\t\texecution.ClientLibraryVersion,\n\t\texecution.ClientFeatureVersion,\n\t\texecution.ClientImpl,\n\t\texecution.AutoResetPoints.Data,\n\t\texecution.AutoResetPoints.GetEncodingString(),\n\t\texecution.Attempt,\n\t\texecution.HasRetryPolicy,\n\t\tint32(execution.InitialInterval.Seconds()),\n\t\texecution.BackoffCoefficient,\n\t\tint32(execution.MaximumInterval.Seconds()),\n\t\texecution.ExpirationTime,\n\t\texecution.MaximumAttempts,\n\t\texecution.NonRetriableErrors,\n\t\tpersistence.EventStoreVersion,\n\t\texecution.BranchToken,\n\t\texecution.CronSchedule,\n\t\tint32(execution.CronOverlapPolicy),\n\t\tint32(execution.ExpirationInterval.Seconds()),\n\t\texecution.SearchAttributes,\n\t\texecution.Memo,\n\t\texecution.PartitionConfig,\n\t\texecution.ActiveClusterSelectionPolicy.GetData(),\n\t\texecution.ActiveClusterSelectionPolicy.GetEncodingString(),\n\t\texecution.NextEventID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeExecutionTaskID,\n\t\texecution.VersionHistories.Data,\n\t\texecution.VersionHistories.GetEncodingString(),\n\t\texecution.Checksums.Version,\n\t\texecution.Checksums.Flavor,\n\t\texecution.Checksums.Value,\n\t\texecution.LastWriteVersion,\n\t\texecution.State,\n\t\ttimeStamp,\n\t)\n\treturn nil\n}\n\nfunc isRequestRowType(rowType int) bool {\n\tif rowType >= rowTypeWorkflowRequestStart && rowType <= rowTypeWorkflowRequestReset {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc toRequestRowType(requestType persistence.WorkflowRequestType) (int, error) {\n\tswitch requestType {\n\tcase persistence.WorkflowRequestTypeStart:\n\t\treturn rowTypeWorkflowRequestStart, nil\n\tcase persistence.WorkflowRequestTypeSignal:\n\t\treturn rowTypeWorkflowRequestSignal, nil\n\tcase persistence.WorkflowRequestTypeCancel:\n\t\treturn rowTypeWorkflowRequestCancel, nil\n\tcase persistence.WorkflowRequestTypeReset:\n\t\treturn rowTypeWorkflowRequestReset, nil\n\tdefault:\n\t\treturn 0, fmt.Errorf(\"unknown workflow request type %v\", requestType)\n\t}\n}\n\nfunc fromRequestRowType(rowType int) (persistence.WorkflowRequestType, error) {\n\tswitch rowType {\n\tcase rowTypeWorkflowRequestStart:\n\t\treturn persistence.WorkflowRequestTypeStart, nil\n\tcase rowTypeWorkflowRequestSignal:\n\t\treturn persistence.WorkflowRequestTypeSignal, nil\n\tcase rowTypeWorkflowRequestCancel:\n\t\treturn persistence.WorkflowRequestTypeCancel, nil\n\tcase rowTypeWorkflowRequestReset:\n\t\treturn persistence.WorkflowRequestTypeReset, nil\n\tdefault:\n\t\treturn persistence.WorkflowRequestType(0), fmt.Errorf(\"unknown request row type %v\", rowType)\n\t}\n}\n\nfunc insertWorkflowActiveClusterSelectionPolicyRow(\n\tbatch gocql.Batch,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\ttimeStamp time.Time,\n) error {\n\tif activeClusterSelectionPolicyRow == nil || activeClusterSelectionPolicyRow.Policy == nil {\n\t\treturn nil\n\t}\n\n\tbatch.Query(templateInsertWorkflowActiveClusterSelectionPolicyRowQuery,\n\t\tactiveClusterSelectionPolicyRow.ShardID,\n\t\trowTypeWorkflowActiveClusterSelectionPolicy,\n\t\tactiveClusterSelectionPolicyRow.DomainID,\n\t\tactiveClusterSelectionPolicyRow.WorkflowID,\n\t\tactiveClusterSelectionPolicyRow.RunID,\n\t\tdefaultVisibilityTimestamp,\n\t\trowTypeWorkflowActiveClusterSelectionVersion,\n\t\ttimeStamp,\n\t\tactiveClusterSelectionPolicyRow.Policy.Data,\n\t\tactiveClusterSelectionPolicyRow.Policy.GetEncodingString(),\n\t)\n\treturn nil\n}\n\nfunc insertOrUpsertWorkflowRequestRow(\n\tbatch gocql.Batch,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\ttimeStamp time.Time,\n) error {\n\tif requests == nil {\n\t\treturn nil\n\t}\n\tvar insertQuery string\n\tswitch requests.WriteMode {\n\tcase nosqlplugin.WorkflowRequestWriteModeInsert:\n\t\tinsertQuery = templateInsertWorkflowRequestQuery\n\tcase nosqlplugin.WorkflowRequestWriteModeUpsert:\n\t\tinsertQuery = templateUpsertWorkflowRequestQuery\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown workflow request write mode %v\", requests.WriteMode)\n\t}\n\tfor _, row := range requests.Rows {\n\t\trowType, err := toRequestRowType(row.RequestType)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tbatch.Query(insertQuery,\n\t\t\trow.ShardID,\n\t\t\trowType,\n\t\t\trow.DomainID,\n\t\t\trow.WorkflowID,\n\t\t\trow.RequestID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\temptyWorkflowRequestVersion*-1,\n\t\t\trow.RunID,\n\t\t\ttimeStamp,\n\t\t\tworkflowRequestTTLInSeconds,\n\t\t)\n\t\tbatch.Query(insertQuery,\n\t\t\trow.ShardID,\n\t\t\trowType,\n\t\t\trow.DomainID,\n\t\t\trow.WorkflowID,\n\t\t\trow.RequestID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trow.Version*-1,\n\t\t\trow.RunID,\n\t\t\ttimeStamp,\n\t\t\tworkflowRequestTTLInSeconds,\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc createOrUpdateCurrentWorkflow(\n\tbatch gocql.Batch,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\ttimeStamp time.Time,\n) error {\n\tswitch request.WriteMode {\n\tcase nosqlplugin.CurrentWorkflowWriteModeNoop:\n\t\treturn nil\n\tcase nosqlplugin.CurrentWorkflowWriteModeInsert:\n\t\tbatch.Query(templateCreateCurrentWorkflowExecutionQuery,\n\t\t\tshardID,\n\t\t\trowTypeExecution,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tpermanentRunID,\n\t\t\tdefaultVisibilityTimestamp,\n\t\t\trowTypeExecutionTaskID,\n\t\t\trequest.Row.RunID,\n\t\t\trequest.Row.RunID,\n\t\t\trequest.Row.CreateRequestID,\n\t\t\trequest.Row.State,\n\t\t\trequest.Row.CloseStatus,\n\t\t\trequest.Row.LastWriteVersion,\n\t\t\trequest.Row.State,\n\t\t\ttimeStamp,\n\t\t)\n\tcase nosqlplugin.CurrentWorkflowWriteModeUpdate:\n\t\tif request.Condition == nil || request.Condition.GetCurrentRunID() == \"\" {\n\t\t\treturn fmt.Errorf(\"CurrentWorkflowWriteModeUpdate require Condition.CurrentRunID\")\n\t\t}\n\t\tif request.Condition.LastWriteVersion != nil && request.Condition.State != nil {\n\t\t\tbatch.Query(templateUpdateCurrentWorkflowExecutionForNewQuery,\n\t\t\t\trequest.Row.RunID,\n\t\t\t\trequest.Row.RunID,\n\t\t\t\trequest.Row.CreateRequestID,\n\t\t\t\trequest.Row.State,\n\t\t\t\trequest.Row.CloseStatus,\n\t\t\t\trequest.Row.LastWriteVersion,\n\t\t\t\trequest.Row.State,\n\t\t\t\ttimeStamp,\n\t\t\t\tshardID,\n\t\t\t\trowTypeExecution,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\tpermanentRunID,\n\t\t\t\tdefaultVisibilityTimestamp,\n\t\t\t\trowTypeExecutionTaskID,\n\t\t\t\t*request.Condition.CurrentRunID,\n\t\t\t\t*request.Condition.LastWriteVersion,\n\t\t\t\t*request.Condition.State,\n\t\t\t)\n\t\t} else {\n\t\t\tbatch.Query(templateUpdateCurrentWorkflowExecutionQuery,\n\t\t\t\trequest.Row.RunID,\n\t\t\t\trequest.Row.RunID,\n\t\t\t\trequest.Row.CreateRequestID,\n\t\t\t\trequest.Row.State,\n\t\t\t\trequest.Row.CloseStatus,\n\t\t\t\trequest.Row.LastWriteVersion,\n\t\t\t\trequest.Row.State,\n\t\t\t\ttimeStamp,\n\t\t\t\tshardID,\n\t\t\t\trowTypeExecution,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\tpermanentRunID,\n\t\t\t\tdefaultVisibilityTimestamp,\n\t\t\t\trowTypeExecutionTaskID,\n\t\t\t\t*request.Condition.CurrentRunID,\n\t\t\t)\n\t\t}\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown mode %v\", request.WriteMode)\n\t}\n\treturn nil\n}\n\nfunc mustConvertToSlice(value interface{}) []interface{} {\n\tv := reflect.ValueOf(value)\n\tswitch v.Kind() {\n\tcase reflect.Slice, reflect.Array:\n\t\tresult := make([]interface{}, v.Len())\n\t\tfor i := 0; i < v.Len(); i++ {\n\t\t\tresult[i] = v.Index(i).Interface()\n\t\t}\n\t\treturn result\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"Unable to convert %v to slice which is of type %T\", value, value))\n\t}\n}\n\nfunc populateGetReplicationTasks(query gocql.Query) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\titer := query.Iter()\n\tif iter == nil {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"populateGetReplicationTasks operation failed.  Not able to create query iterator.\",\n\t\t}\n\t}\n\n\tvar tasks []*nosqlplugin.HistoryMigrationTask\n\ttask := make(map[string]interface{})\n\tfor iter.MapScan(task) {\n\t\tt := parseReplicationTaskInfo(task[\"replication\"].(map[string]interface{}))\n\t\ttaskID := task[\"task_id\"].(int64)\n\t\tdata := task[\"data\"].([]byte)\n\t\tencoding := task[\"data_encoding\"].(string)\n\t\ttaskBlob := persistence.NewDataBlob(data, constants.EncodingType(encoding))\n\t\t// Reset task map to get it ready for next scan\n\t\ttask = make(map[string]interface{})\n\n\t\ttasks = append(tasks, &nosqlplugin.HistoryMigrationTask{\n\t\t\tReplication: t,\n\t\t\tTask:        taskBlob,\n\t\t\tTaskID:      taskID,\n\t\t})\n\t}\n\tnextPageToken := getNextPageToken(iter)\n\terr := iter.Close()\n\n\treturn tasks, nextPageToken, err\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/cassandra/workflow_utils_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"regexp\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar FixedTime = time.Date(2025, 1, 6, 15, 0, 0, 0, time.UTC)\n\n// fakeSession is fake implementation of gocql.Session\ntype fakeSession struct {\n\t// inputs\n\titer                      *fakeIter\n\tmapExecuteBatchCASApplied bool\n\tmapExecuteBatchCASPrev    map[string]any\n\tmapExecuteBatchCASErr     error\n\tquery                     gocql.Query\n\n\t// outputs\n\tbatches []*fakeBatch\n\tqueries []string\n}\n\nfunc (s *fakeSession) Query(queryTmpl string, args ...interface{}) gocql.Query {\n\ts.queries = append(s.queries, sanitizedRenderedQuery(queryTmpl, args...))\n\treturn s.query\n}\n\nfunc (s *fakeSession) NewBatch(gocql.BatchType) gocql.Batch {\n\tb := &fakeBatch{}\n\ts.batches = append(s.batches, b)\n\treturn b\n}\n\nfunc (s *fakeSession) ExecuteBatch(gocql.Batch) error {\n\treturn nil\n}\n\nfunc (s *fakeSession) MapExecuteBatchCAS(batch gocql.Batch, prev map[string]interface{}) (bool, gocql.Iter, error) {\n\tfor k, v := range s.mapExecuteBatchCASPrev {\n\t\tprev[k] = v\n\t}\n\treturn s.mapExecuteBatchCASApplied, s.iter, s.mapExecuteBatchCASErr\n}\n\n// fakeBatch is fake implementation of gocql.Batch\ntype fakeBatch struct {\n\t// outputs\n\tqueries []string\n}\n\n// Query is fake implementation of gocql.Batch.Query\nfunc (b *fakeBatch) Query(queryTmpl string, args ...interface{}) {\n\tb.queries = append(b.queries, sanitizedRenderedQuery(queryTmpl, args...))\n}\n\n// WithContext is fake implementation of gocql.Batch.WithContext\nfunc (b *fakeBatch) WithContext(context.Context) gocql.Batch {\n\treturn b\n}\n\n// WithTimestamp is fake implementation of gocql.Batch.WithTimestamp\nfunc (b *fakeBatch) WithTimestamp(int64) gocql.Batch {\n\treturn b\n}\n\n// Consistency is fake implementation of gocql.Batch.Consistency\nfunc (b *fakeBatch) Consistency(gocql.Consistency) gocql.Batch {\n\treturn b\n}\n\n// fakeQuery is fake implementation of gocql.Query\nfunc (s *fakeSession) Close() {\n}\n\nfunc sanitizedRenderedQuery(queryTmpl string, args ...interface{}) string {\n\targsSanitized := make([]interface{}, len(args))\n\tfor i, arg := range args {\n\t\t// use values instead of pointer so that we can compare them in tests\n\t\tif reflect.ValueOf(arg).Kind() == reflect.Ptr && !reflect.ValueOf(arg).IsZero() {\n\t\t\targsSanitized[i] = reflect.ValueOf(arg).Elem().Interface()\n\t\t} else {\n\t\t\targsSanitized[i] = arg\n\t\t}\n\n\t\tif t, ok := argsSanitized[i].(time.Time); ok {\n\t\t\t// use fixed time format to avoid flakiness\n\t\t\targsSanitized[i] = t.UTC().Format(time.RFC3339)\n\t\t}\n\n\t}\n\tqueryTmpl = strings.ReplaceAll(queryTmpl, \"?\", \"%v\")\n\treturn fmt.Sprintf(queryTmpl, argsSanitized...)\n}\n\n// fakeIter is fake implementation of gocql.Iter\ntype fakeIter struct {\n\t// input parametrs\n\tmapScanInputs []map[string]interface{}\n\tscanInputs    [][]interface{}\n\tpageState     []byte\n\tcloseErr      error\n\n\t// output parameters\n\tmapScanCalls int\n\tscanCalls    int\n\tclosed       bool\n}\n\n// Scan is fake implementation of gocql.Iter.Scan\nfunc (i *fakeIter) Scan(outArgs ...interface{}) bool {\n\tif i.scanCalls >= len(i.scanInputs) {\n\t\treturn false\n\t}\n\n\tfor j, v := range i.scanInputs[i.scanCalls] {\n\t\tif len(outArgs) <= j {\n\t\t\tpanic(fmt.Sprintf(\"outArgs length: %d is less than expected: %d\", len(outArgs), len(i.scanInputs[i.scanCalls])))\n\t\t}\n\n\t\tif v == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tdst := outArgs[j]\n\t\tdstPtrValue := reflect.ValueOf(dst)\n\t\tdstValue := reflect.Indirect(dstPtrValue)\n\n\t\tfunc() {\n\t\t\tdefer func() {\n\t\t\t\tif r := recover(); r != nil {\n\t\t\t\t\tpanic(fmt.Sprintf(\"failed to set %dth value: %v to %v, inner panic: %s\", j, v, dst, r))\n\t\t\t\t}\n\t\t\t}()\n\t\t\tdstValue.Set(reflect.ValueOf(v))\n\t\t}()\n\t}\n\n\ti.scanCalls++\n\treturn true\n}\n\n// MapScan is fake implementation of gocql.Iter.MapScan\nfunc (i *fakeIter) MapScan(res map[string]interface{}) bool {\n\tif i.mapScanCalls >= len(i.mapScanInputs) {\n\t\treturn false\n\t}\n\n\tfor k, v := range i.mapScanInputs[i.mapScanCalls] {\n\t\tres[k] = v\n\t}\n\ti.mapScanCalls++\n\treturn true\n}\n\n// PageState is fake implementation of gocql.Iter.PageState\nfunc (i *fakeIter) PageState() []byte {\n\treturn i.pageState\n}\n\n// Close is fake implementation of gocql.Iter.Close\nfunc (i *fakeIter) Close() error {\n\ti.closed = true\n\treturn i.closeErr\n}\n\ntype fakeUUID struct {\n\tuuid string\n}\n\nfunc (u *fakeUUID) String() string {\n\treturn u.uuid\n}\n\nvar _ (gocql.Query) = &fakeQuery{}\n\ntype fakeQuery struct {\n\titer              *fakeIter\n\tmapScan           map[string]interface{}\n\terr               error\n\tscanCASApplied    bool\n\tmapScanCASApplied bool\n}\n\nfunc (q *fakeQuery) Exec() error {\n\treturn q.err\n}\n\nfunc (q *fakeQuery) Scan(...interface{}) error {\n\treturn q.err\n}\n\nfunc (q *fakeQuery) ScanCAS(...interface{}) (bool, error) {\n\treturn q.scanCASApplied, q.err\n}\n\nfunc (q *fakeQuery) MapScan(res map[string]interface{}) error {\n\tfor k, v := range q.mapScan {\n\t\tres[k] = v\n\t}\n\treturn q.err\n}\n\nfunc (q *fakeQuery) MapScanCAS(map[string]interface{}) (bool, error) {\n\treturn q.mapScanCASApplied, q.err\n}\n\nfunc (q *fakeQuery) Iter() gocql.Iter {\n\treturn q.iter\n}\n\nfunc (q *fakeQuery) PageSize(int) gocql.Query {\n\treturn q\n}\n\nfunc (q *fakeQuery) PageState([]byte) gocql.Query {\n\treturn q\n}\n\nfunc (q *fakeQuery) WithContext(context.Context) gocql.Query {\n\treturn q\n}\n\nfunc (q *fakeQuery) WithTimestamp(int64) gocql.Query {\n\treturn q\n}\n\nfunc (q *fakeQuery) Consistency(gocql.Consistency) gocql.Query {\n\treturn q\n}\n\nfunc (q *fakeQuery) Bind(...interface{}) gocql.Query {\n\treturn q\n}\n\nfunc TestExecuteCreateWorkflowBatchTransaction(t *testing.T) {\n\ttests := []struct {\n\t\t// fake setup parameters\n\t\tdesc    string\n\t\tsession *fakeSession\n\n\t\t// executeCreateWorkflowBatchTransaction args\n\t\tbatch        *fakeBatch\n\t\tcurrentWFReq *nosqlplugin.CurrentWorkflowWriteRequest\n\t\texecution    *nosqlplugin.WorkflowExecutionRequest\n\t\tshardCond    *nosqlplugin.ShardCondition\n\n\t\t// expectations\n\t\twantErr error\n\t}{\n\t\t{\n\t\t\tdesc:  \"applied\",\n\t\t\tbatch: &fakeBatch{},\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"CAS error\",\n\t\t\tbatch: &fakeBatch{},\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASErr: fmt.Errorf(\"db operation failed for some reason\"),\n\t\t\t\titer:                  &fakeIter{},\n\t\t\t},\n\t\t\twantErr: fmt.Errorf(\"db operation failed for some reason\"),\n\t\t},\n\t\t{\n\t\t\tdesc: \"shard range id mismatch\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":     rowTypeShard,\n\t\t\t\t\t\"run_id\":   uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t\t\"range_id\": int64(200),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tShardRangeIDNotMatch: common.Int64Ptr(200),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"workflow request already exists\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":   rowTypeWorkflowRequestCancel,\n\t\t\t\t\t\"run_id\": uuid.Parse(\"4b8045c8-7b45-41e0-bf03-1f0d166b818d\"),\n\t\t\t\t},\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{\n\t\t\t\t\t\t\"current_run_id\": uuid.Parse(\"6b844fb4-c18a-4979-a2d3-731ebdd1db08\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"fd88863f-bb32-4daa-8878-49e08b91545e\",\n\t\t\t\t\tWorkflowID: \"wfid\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeCancel,\n\t\t\t\t\tRunID:       \"6b844fb4-c18a-4979-a2d3-731ebdd1db08\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"workflow request already exists - but failed to get latest request\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":   rowTypeWorkflowRequestSignal,\n\t\t\t\t\t\"run_id\": uuid.Parse(\"4b8045c8-7b45-41e0-bf03-1f0d166b818d\"),\n\t\t\t\t},\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{\n\t\t\t\t\t\t\"current_run_id\": uuid.Parse(\"6b844fb4-c18a-4979-a2d3-731ebdd1db08\"),\n\t\t\t\t\t},\n\t\t\t\t\terr: fmt.Errorf(\"failed to get latest request\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"fd88863f-bb32-4daa-8878-49e08b91545e\",\n\t\t\t\t\tWorkflowID: \"wfid\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: fmt.Errorf(\"failed to get latest request\"),\n\t\t},\n\t\t{\n\t\t\tdesc: \"current execution already exists and write mode is CurrentWorkflowWriteModeInsert\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(permanentRunID),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t\t\"execution\": map[string]any{\n\t\t\t\t\t\t\"run_id\": uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t\t\t\"state\":  1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeInsert,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tWorkflowExecutionAlreadyExists: &nosqlplugin.WorkflowExecutionAlreadyExists{\n\t\t\t\t\tRunID:            \"bda9cd9c-32fb-4267-b120-346e5351fc46\",\n\t\t\t\t\tState:            1,\n\t\t\t\t\tLastWriteVersion: 3,\n\t\t\t\t\tOtherInfo:        \"Workflow execution already running. WorkflowId: test-workflow-id, RunId: bda9cd9c-32fb-4267-b120-346e5351fc46\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"creation condition failed by mismatch runID\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(permanentRunID),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t\t\"current_run_id\":              uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tWorkflowID: \"wfid\",\n\t\t\t\t},\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: common.StringPtr(\"fd88863f-bb32-4daa-8878-49e08b91545e\"), // not matching current_run_id above on purpose\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\n\t\t\t\t\t\"Workflow execution creation condition failed by mismatch runID. \" +\n\t\t\t\t\t\t\"WorkflowId: wfid, Expected Current RunID: fd88863f-bb32-4daa-8878-49e08b91545e, \" +\n\t\t\t\t\t\t\"Actual Current RunID: bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"creation condition failed by mismatch last write version\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(permanentRunID),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t\t\"current_run_id\":              uuid.Parse(\"fd88863f-bb32-4daa-8878-49e08b91545e\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tWorkflowID: \"wfid\",\n\t\t\t\t},\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID:     common.StringPtr(\"fd88863f-bb32-4daa-8878-49e08b91545e\"), // not matching current_run_id above on purpose\n\t\t\t\t\tLastWriteVersion: common.Int64Ptr(1),\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\n\t\t\t\t\t\"Workflow execution creation condition failed. \" +\n\t\t\t\t\t\t\"WorkflowId: wfid, Expected Version: 1, Actual Version: 3\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"creation condition failed by mismatch state\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(permanentRunID),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t\t\"current_run_id\":              uuid.Parse(\"fd88863f-bb32-4daa-8878-49e08b91545e\"),\n\t\t\t\t\t\"workflow_state\":              2,\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tWorkflowID: \"wfid\",\n\t\t\t\t},\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID:     common.StringPtr(\"fd88863f-bb32-4daa-8878-49e08b91545e\"), // not matching current_run_id above on purpose\n\t\t\t\t\tLastWriteVersion: common.Ptr(int64(3)),\n\t\t\t\t\tState:            common.Ptr(int(10)),\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\n\t\t\t\t\t\"Workflow execution creation condition failed. \" +\n\t\t\t\t\t\t\"WorkflowId: wfid, Expected State: 10, Actual State: 2\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"concrete execution already exists\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID:      \"wfid\",\n\t\t\t\t\tRunID:           \"bda9cd9c-32fb-4267-b120-346e5351fc46\",\n\t\t\t\t\tCreateRequestID: \"reqid_123\",\n\t\t\t\t\tState:           2,\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tWorkflowExecutionAlreadyExists: &nosqlplugin.WorkflowExecutionAlreadyExists{\n\t\t\t\t\tOtherInfo: \"Workflow execution already running. WorkflowId: wfid, \" +\n\t\t\t\t\t\t\"RunId: bda9cd9c-32fb-4267-b120-346e5351fc46\",\n\t\t\t\t\tCreateRequestID:  \"reqid_123\",\n\t\t\t\t\tRunID:            \"bda9cd9c-32fb-4267-b120-346e5351fc46\",\n\t\t\t\t\tState:            2,\n\t\t\t\t\tLastWriteVersion: 3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"unknown condition failure\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":                        rowTypeExecution,\n\t\t\t\t\t\"run_id\":                      uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t\t\"range_id\":                    int64(100),\n\t\t\t\t\t\"workflow_last_write_version\": int64(3),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tRunID: \"something else\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tUnknownConditionFailureDetails: common.StringPtr(\"Failed to operate on workflow execution.  Request RangeID: 100\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\terr := executeCreateWorkflowBatchTransaction(context.Background(), tc.session, tc.batch, tc.currentWFReq, tc.execution, tc.shardCond)\n\t\t\tif diff := errDiff(tc.wantErr, err); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Error mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t\tif !tc.session.iter.closed {\n\t\t\t\tt.Error(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestExecuteUpdateWorkflowBatchTransaction(t *testing.T) {\n\ttests := []struct {\n\t\t// fake setup parameters\n\t\tdesc    string\n\t\tsession *fakeSession\n\n\t\t// executeUpdateWorkflowBatchTransaction args\n\t\tbatch               *fakeBatch\n\t\tcurrentWFReq        *nosqlplugin.CurrentWorkflowWriteRequest\n\t\tprevNextEventIDCond int64\n\t\tshardCond           *nosqlplugin.ShardCondition\n\n\t\t// expectations\n\t\twantErr error\n\t}{\n\t\t{\n\t\t\tdesc:  \"applied\",\n\t\t\tbatch: &fakeBatch{},\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: true,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"CAS error\",\n\t\t\tbatch: &fakeBatch{},\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASErr: fmt.Errorf(\"db operation failed for some reason\"),\n\t\t\t\titer:                  &fakeIter{},\n\t\t\t},\n\t\t\twantErr: fmt.Errorf(\"db operation failed for some reason\"),\n\t\t},\n\t\t{\n\t\t\tdesc: \"range id mismatch for shard row\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":     rowTypeShard,\n\t\t\t\t\t\"run_id\":   uuid.Parse(\"bda9cd9c-32fb-4267-b120-346e5351fc46\"),\n\t\t\t\t\t\"range_id\": int64(200),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tShardRangeIDNotMatch: common.Int64Ptr(200),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"workflow request already exists\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":   rowTypeWorkflowRequestSignal,\n\t\t\t\t\t\"run_id\": uuid.Parse(\"4b8045c8-7b45-41e0-bf03-1f0d166b818d\"),\n\t\t\t\t},\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{\n\t\t\t\t\t\t\"current_run_id\": uuid.Parse(\"6b844fb4-c18a-4979-a2d3-731ebdd1db08\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tDuplicateRequest: &nosqlplugin.DuplicateRequest{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRunID:       \"6b844fb4-c18a-4979-a2d3-731ebdd1db08\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"workflow request already exists - but failed to get latest request\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":   rowTypeWorkflowRequestCancel,\n\t\t\t\t\t\"run_id\": uuid.Parse(\"4b8045c8-7b45-41e0-bf03-1f0d166b818d\"),\n\t\t\t\t},\n\t\t\t\tquery: &fakeQuery{\n\t\t\t\t\tmapScan: map[string]interface{}{\n\t\t\t\t\t\t\"current_run_id\": uuid.Parse(\"6b844fb4-c18a-4979-a2d3-731ebdd1db08\"),\n\t\t\t\t\t},\n\t\t\t\t\terr: fmt.Errorf(\"failed to get latest request\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:        &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 1,\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\twantErr: fmt.Errorf(\"failed to get latest request\"),\n\t\t},\n\t\t{\n\t\t\tdesc: \"nextEventID mismatch for execution row\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":          rowTypeExecution,\n\t\t\t\t\t\"run_id\":        uuid.Parse(\"0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t\t\t\"next_event_id\": int64(10),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch:               &fakeBatch{},\n\t\t\tprevNextEventIDCond: 11, // not matching next_event_id above on purpose\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID: \"0875863e-dcef-496a-b8a2-3210b2958e25\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tRangeID: 200,\n\t\t\t},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tUnknownConditionFailureDetails: common.StringPtr(\n\t\t\t\t\t\"Failed to update mutable state. \" +\n\t\t\t\t\t\t\"previousNextEventIDCondition: 11, actualNextEventID: 10, Request Current RunID: 0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"runID mismatch for current execution row\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":           rowTypeExecution,\n\t\t\t\t\t\"run_id\":         uuid.Parse(permanentRunID),\n\t\t\t\t\t\"current_run_id\": uuid.Parse(\"0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: common.StringPtr(\"fd88863f-bb32-4daa-8878-49e08b91545e\"), // not matching current_run_id above on purpose\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{},\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tCurrentWorkflowConditionFailInfo: common.StringPtr(\n\t\t\t\t\t\"Failed to update mutable state. requestConditionalRunID: fd88863f-bb32-4daa-8878-49e08b91545e, \" +\n\t\t\t\t\t\t\"Actual Value: 0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"unknown condition failure\",\n\t\t\tsession: &fakeSession{\n\t\t\t\tmapExecuteBatchCASApplied: false,\n\t\t\t\titer:                      &fakeIter{},\n\t\t\t\tmapExecuteBatchCASPrev: map[string]any{\n\t\t\t\t\t\"type\":           rowTypeExecution,\n\t\t\t\t\t\"run_id\":         uuid.Parse(permanentRunID),\n\t\t\t\t\t\"current_run_id\": uuid.Parse(\"0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tbatch: &fakeBatch{},\n\t\t\tcurrentWFReq: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: common.StringPtr(\"0875863e-dcef-496a-b8a2-3210b2958e25\"), // not matching current_run_id above on purpose\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardCond: &nosqlplugin.ShardCondition{\n\t\t\t\tShardID: 345,\n\t\t\t\tRangeID: 200,\n\t\t\t},\n\t\t\tprevNextEventIDCond: 11,\n\t\t\twantErr: &nosqlplugin.WorkflowOperationConditionFailure{\n\t\t\t\tUnknownConditionFailureDetails: common.StringPtr(\n\t\t\t\t\t\"Failed to update mutable state. ShardID: 345, RangeID: 200, previousNextEventIDCondition: 11, requestConditionalRunID: 0875863e-dcef-496a-b8a2-3210b2958e25\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\terr := executeUpdateWorkflowBatchTransaction(context.Background(), tc.session, tc.batch, tc.currentWFReq, tc.prevNextEventIDCond, tc.shardCond)\n\t\t\tif diff := errDiff(tc.wantErr, err); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Error mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t\tif !tc.session.iter.closed {\n\t\t\t\tt.Error(\"iterator not closed\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestToRequestRowType(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\trequestType persistence.WorkflowRequestType\n\t\twantErr     bool\n\t\twant        int\n\t}{\n\t\t{\n\t\t\tname:        \"StartWorkflow request\",\n\t\t\trequestType: persistence.WorkflowRequestTypeStart,\n\t\t\twantErr:     false,\n\t\t\twant:        rowTypeWorkflowRequestStart,\n\t\t},\n\t\t{\n\t\t\tname:        \"SignalWithWorkflow request\",\n\t\t\trequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\twantErr:     false,\n\t\t\twant:        rowTypeWorkflowRequestSignal,\n\t\t},\n\t\t{\n\t\t\tname:        \"SignalWorkflow request\",\n\t\t\trequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\twantErr:     false,\n\t\t\twant:        rowTypeWorkflowRequestSignal,\n\t\t},\n\t\t{\n\t\t\tname:        \"CancelWorkflow request\",\n\t\t\trequestType: persistence.WorkflowRequestTypeCancel,\n\t\t\twantErr:     false,\n\t\t\twant:        rowTypeWorkflowRequestCancel,\n\t\t},\n\t\t{\n\t\t\tname:        \"ResetWorkflow request\",\n\t\t\trequestType: persistence.WorkflowRequestTypeReset,\n\t\t\twantErr:     false,\n\t\t\twant:        rowTypeWorkflowRequestReset,\n\t\t},\n\t\t{\n\t\t\tname:        \"unknown request\",\n\t\t\trequestType: persistence.WorkflowRequestType(-999),\n\t\t\twantErr:     true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twant, err := toRequestRowType(tc.requestType)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.want, want); diff != \"\" {\n\t\t\t\tt.Fatalf(\"request type mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromRequestRowType(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\trequestType int\n\t\twantErr     bool\n\t\twant        persistence.WorkflowRequestType\n\t}{\n\t\t{\n\t\t\tname:        \"StartWorkflow request\",\n\t\t\trequestType: rowTypeWorkflowRequestStart,\n\t\t\twantErr:     false,\n\t\t\twant:        persistence.WorkflowRequestTypeStart,\n\t\t},\n\t\t{\n\t\t\tname:        \"SignalWithWorkflow request\",\n\t\t\trequestType: rowTypeWorkflowRequestSignal,\n\t\t\twantErr:     false,\n\t\t\twant:        persistence.WorkflowRequestTypeSignal,\n\t\t},\n\t\t{\n\t\t\tname:        \"SignalWorkflow request\",\n\t\t\trequestType: rowTypeWorkflowRequestSignal,\n\t\t\twantErr:     false,\n\t\t\twant:        persistence.WorkflowRequestTypeSignal,\n\t\t},\n\t\t{\n\t\t\tname:        \"CancelWorkflow request\",\n\t\t\trequestType: rowTypeWorkflowRequestCancel,\n\t\t\twantErr:     false,\n\t\t\twant:        persistence.WorkflowRequestTypeCancel,\n\t\t},\n\t\t{\n\t\t\tname:        \"ResetWorkflow request\",\n\t\t\trequestType: rowTypeWorkflowRequestReset,\n\t\t\twantErr:     false,\n\t\t\twant:        persistence.WorkflowRequestTypeReset,\n\t\t},\n\t\t{\n\t\t\tname:        \"unknown request\",\n\t\t\trequestType: rowTypeShard,\n\t\t\twantErr:     true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twant, err := fromRequestRowType(tc.requestType)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.want, want); diff != \"\" {\n\t\t\t\tt.Fatalf(\"request type mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInsertOrUpsertWorkflowRequestRow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\trequest     *nosqlplugin.WorkflowRequestsWriteRequest\n\t\twantErr     bool\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tname: \"WorkflowRequestWriteModeInsert\",\n\t\t\trequest: &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\t\t\tRows: []*nosqlplugin.WorkflowRequestRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    1,\n\t\t\t\t\t\tDomainID:   \"c09537fd-67ce-4b08-a817-eb8f12ad3a91\",\n\t\t\t\t\t\tWorkflowID: \"test\",\n\t\t\t\t\t\tRunID:      \"25bd1013-0e79-4c45-8e55-08bb45886896\",\n\t\t\t\t\t\tRequestID:  \"9ab1d25d-8620-440a-b3f1-d3167e08c769\",\n\t\t\t\t\t\tVersion:    100,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tWriteMode: nosqlplugin.WorkflowRequestWriteModeInsert,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, created_time) ` +\n\t\t\t\t\t`VALUES(1, 7, c09537fd-67ce-4b08-a817-eb8f12ad3a91, test, 9ab1d25d-8620-440a-b3f1-d3167e08c769, 946684800000, 1000, 25bd1013-0e79-4c45-8e55-08bb45886896, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`IF NOT EXISTS USING TTL 10800`,\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, created_time) ` +\n\t\t\t\t\t`VALUES(1, 7, c09537fd-67ce-4b08-a817-eb8f12ad3a91, test, 9ab1d25d-8620-440a-b3f1-d3167e08c769, 946684800000, -100, 25bd1013-0e79-4c45-8e55-08bb45886896, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`IF NOT EXISTS USING TTL 10800`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowRequestWriteModeUpsert\",\n\t\t\trequest: &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\t\t\tRows: []*nosqlplugin.WorkflowRequestRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    1,\n\t\t\t\t\t\tDomainID:   \"c09537fd-67ce-4b08-a817-eb8f12ad3a91\",\n\t\t\t\t\t\tWorkflowID: \"test\",\n\t\t\t\t\t\tRunID:      \"25bd1013-0e79-4c45-8e55-08bb45886896\",\n\t\t\t\t\t\tRequestID:  \"9ab1d25d-8620-440a-b3f1-d3167e08c769\",\n\t\t\t\t\t\tVersion:    100,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tWriteMode: nosqlplugin.WorkflowRequestWriteModeUpsert,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, last_updated_time) ` +\n\t\t\t\t\t`VALUES(1, 7, c09537fd-67ce-4b08-a817-eb8f12ad3a91, test, 9ab1d25d-8620-440a-b3f1-d3167e08c769, 946684800000, 1000, 25bd1013-0e79-4c45-8e55-08bb45886896, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`USING TTL 10800`,\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, last_updated_time) ` +\n\t\t\t\t\t`VALUES(1, 7, c09537fd-67ce-4b08-a817-eb8f12ad3a91, test, 9ab1d25d-8620-440a-b3f1-d3167e08c769, 946684800000, -100, 25bd1013-0e79-4c45-8e55-08bb45886896, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`USING TTL 10800`,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"unknown mode\",\n\t\t\trequest: &nosqlplugin.WorkflowRequestsWriteRequest{\n\t\t\t\tRows: []*nosqlplugin.WorkflowRequestRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tWriteMode: nosqlplugin.WorkflowRequestWriteMode(-100),\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\terr := insertOrUpsertWorkflowRequestRow(batch, tc.request, FixedTime)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateTimerTasks(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-12T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\ttimerTasks []*nosqlplugin.HistoryMigrationTask\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain_xyz\",\n\t\t\tworkflowID: \"workflow_xyz\",\n\t\t\ttimerTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tTimer: &nosqlplugin.TimerTask{\n\t\t\t\t\t\tRunID:               \"rundid_1\",\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tTaskType:            1,\n\t\t\t\t\t\tTimeoutType:         1,\n\t\t\t\t\t\tEventID:             10,\n\t\t\t\t\t\tScheduleAttempt:     0,\n\t\t\t\t\t\tVersion:             0,\n\t\t\t\t\t\tVisibilityTimestamp: ts,\n\t\t\t\t\t\tTaskList:            \"tasklist_1\",\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTimer: &nosqlplugin.TimerTask{\n\t\t\t\t\t\tRunID:               \"rundid_1\",\n\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\tTaskType:            1,\n\t\t\t\t\t\tTimeoutType:         1,\n\t\t\t\t\t\tEventID:             11,\n\t\t\t\t\t\tScheduleAttempt:     0,\n\t\t\t\t\t\tVersion:             0,\n\t\t\t\t\t\tVisibilityTimestamp: ts.Add(time.Minute),\n\t\t\t\t\t\tTaskList:            \"tasklist_2\",\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"timer2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, timer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 3, 10000000-4000-f000-f000-000000000000, 20000000-4000-f000-f000-000000000000, 30000000-4000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_1, visibility_ts: 1702418921000, task_id: 1, type: 1, timeout_type: 1, event_id: 10, schedule_attempt: 0, version: 0, task_list: tasklist_1}, ` +\n\t\t\t\t\t`[116 105 109 101 114 49], thriftrw, 1702418921000, 1, 2025-01-06T15:00:00Z)`,\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, timer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 3, 10000000-4000-f000-f000-000000000000, 20000000-4000-f000-f000-000000000000, 30000000-4000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_1, visibility_ts: 1702418981000, task_id: 2, type: 1, timeout_type: 1, event_id: 11, schedule_attempt: 0, version: 0, task_list: tasklist_2}, ` +\n\t\t\t\t\t`[116 105 109 101 114 50], thriftrw, 1702418981000, 2, 2025-01-06T15:00:00Z)`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\terr := createTimerTasks(batch, tc.shardID, tc.domainID, tc.workflowID, tc.timerTasks, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"createTimerTasks failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicationTasks(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-12T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\treplTasks  []*nosqlplugin.HistoryMigrationTask\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain_xyz\",\n\t\t\tworkflowID: \"workflow_xyz\",\n\t\t\treplTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tRunID:             \"rundid_1\",\n\t\t\t\t\t\tTaskID:            644,\n\t\t\t\t\t\tTaskType:          0,\n\t\t\t\t\t\tFirstEventID:      5,\n\t\t\t\t\t\tNextEventID:       8,\n\t\t\t\t\t\tVersion:           0,\n\t\t\t\t\t\tScheduledID:       constants.EmptyEventID,\n\t\t\t\t\t\tNewRunBranchToken: []byte{'a', 'b', 'c'},\n\t\t\t\t\t\tCreationTime:      ts,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"rep1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tReplication: &nosqlplugin.ReplicationTask{\n\t\t\t\t\t\tRunID:             \"rundid_1\",\n\t\t\t\t\t\tTaskID:            645,\n\t\t\t\t\t\tTaskType:          0,\n\t\t\t\t\t\tFirstEventID:      25,\n\t\t\t\t\t\tNextEventID:       28,\n\t\t\t\t\t\tVersion:           0,\n\t\t\t\t\t\tScheduledID:       constants.EmptyEventID,\n\t\t\t\t\t\tNewRunBranchToken: []byte{'a', 'b', 'c'},\n\t\t\t\t\t\tCreationTime:      ts.Add(time.Hour),\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"rep2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, replication, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 4, 10000000-5000-f000-f000-000000000000, 20000000-5000-f000-f000-000000000000, 30000000-5000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_1, task_id: 644, type: 0, ` +\n\t\t\t\t\t`first_event_id: 5,next_event_id: 8,version: 0,scheduled_id: -23, event_store_version: 2, branch_token: [], ` +\n\t\t\t\t\t`new_run_event_store_version: 2, new_run_branch_token: [97 98 99], created_time: 1702418921000000000 }, [114 101 112 49], thriftrw, 946684800000, 644, 2025-01-06T15:00:00Z)`,\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, replication, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 4, 10000000-5000-f000-f000-000000000000, 20000000-5000-f000-f000-000000000000, 30000000-5000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_1, task_id: 645, type: 0, ` +\n\t\t\t\t\t`first_event_id: 25,next_event_id: 28,version: 0,scheduled_id: -23, event_store_version: 2, branch_token: [], ` +\n\t\t\t\t\t`new_run_event_store_version: 2, new_run_branch_token: [97 98 99], created_time: 1702422521000000000 }, [114 101 112 50], thriftrw, 946684800000, 645, 2025-01-06T15:00:00Z)`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\tcreateReplicationTasks(batch, tc.shardID, tc.domainID, tc.workflowID, tc.replTasks, FixedTime)\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTransferTasks(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-12T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc          string\n\t\tshardID       int\n\t\tdomainID      string\n\t\tworkflowID    string\n\t\ttransferTasks []*nosqlplugin.HistoryMigrationTask\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain_xyz\",\n\t\t\tworkflowID: \"workflow_xyz\",\n\t\t\ttransferTasks: []*nosqlplugin.HistoryMigrationTask{\n\t\t\t\t{\n\t\t\t\t\tTransfer: &nosqlplugin.TransferTask{\n\t\t\t\t\t\tRunID:                   \"rundid_1\",\n\t\t\t\t\t\tTaskID:                  355,\n\t\t\t\t\t\tTaskType:                0,\n\t\t\t\t\t\tVersion:                 1,\n\t\t\t\t\t\tVisibilityTimestamp:     ts,\n\t\t\t\t\t\tTargetDomainID:          \"e2bf2c8f-0ddf-4451-8840-27cfe8addd62\",\n\t\t\t\t\t\tTargetWorkflowID:        persistence.TransferTaskTransferTargetWorkflowID,\n\t\t\t\t\t\tTargetRunID:             persistence.TransferTaskTransferTargetRunID,\n\t\t\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\t\t\tTaskList:                \"tasklist_1\",\n\t\t\t\t\t\tScheduleID:              14,\n\t\t\t\t\t\tOriginalTaskList:        \"original_tasklist_1\",\n\t\t\t\t\t\tOriginalTaskListKind:    types.TaskListKindEphemeral,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"tr1\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTransfer: &nosqlplugin.TransferTask{\n\t\t\t\t\t\tRunID:                   \"rundid_2\",\n\t\t\t\t\t\tTaskID:                  220,\n\t\t\t\t\t\tTaskType:                0,\n\t\t\t\t\t\tVersion:                 1,\n\t\t\t\t\t\tVisibilityTimestamp:     ts.Add(time.Minute),\n\t\t\t\t\t\tTargetDomainID:          \"e2bf2c8f-0ddf-4451-8840-27cfe8addd62\",\n\t\t\t\t\t\tTargetWorkflowID:        persistence.TransferTaskTransferTargetWorkflowID,\n\t\t\t\t\t\tTargetRunID:             persistence.TransferTaskTransferTargetRunID,\n\t\t\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\t\t\tTaskList:                \"tasklist_2\",\n\t\t\t\t\t\tScheduleID:              3,\n\t\t\t\t\t\tOriginalTaskList:        \"original_tasklist_2\",\n\t\t\t\t\t\tOriginalTaskListKind:    types.TaskListKindEphemeral,\n\t\t\t\t\t},\n\t\t\t\t\tTask: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(\"tr2\"),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, transfer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 2, 10000000-3000-f000-f000-000000000000, 20000000-3000-f000-f000-000000000000, 30000000-3000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_1, visibility_ts: 2023-12-12T22:08:41Z, ` +\n\t\t\t\t\t`task_id: 355, target_domain_id: e2bf2c8f-0ddf-4451-8840-27cfe8addd62, target_domain_ids: map[],` +\n\t\t\t\t\t`target_workflow_id: 20000000-0000-f000-f000-000000000001, target_run_id: 30000000-0000-f000-f000-000000000002, ` +\n\t\t\t\t\t`target_child_workflow_only: true, task_list: tasklist_1, type: 0, schedule_id: 14, record_visibility: false, version: 1, original_task_list: original_tasklist_1, original_task_list_kind: 2}, ` +\n\t\t\t\t\t`[116 114 49], thriftrw, 946684800000, 355, 2025-01-06T15:00:00Z)`,\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, transfer, data, data_encoding, visibility_ts, task_id, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 2, 10000000-3000-f000-f000-000000000000, 20000000-3000-f000-f000-000000000000, 30000000-3000-f000-f000-000000000000, ` +\n\t\t\t\t\t`{domain_id: domain_xyz, workflow_id: workflow_xyz, run_id: rundid_2, visibility_ts: 2023-12-12T22:09:41Z, ` +\n\t\t\t\t\t`task_id: 220, target_domain_id: e2bf2c8f-0ddf-4451-8840-27cfe8addd62, target_domain_ids: map[],` +\n\t\t\t\t\t`target_workflow_id: 20000000-0000-f000-f000-000000000001, target_run_id: 30000000-0000-f000-f000-000000000002, ` +\n\t\t\t\t\t`target_child_workflow_only: true, task_list: tasklist_2, type: 0, schedule_id: 3, record_visibility: false, version: 1, original_task_list: original_tasklist_2, original_task_list_kind: 2}, ` +\n\t\t\t\t\t`[116 114 50], thriftrw, 946684800000, 220, 2025-01-06T15:00:00Z)`,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\tcreateTransferTasks(batch, tc.shardID, tc.domainID, tc.workflowID, tc.transferTasks, FixedTime)\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetSignalsRequested(t *testing.T) {\n\ttests := []struct {\n\t\tdesc         string\n\t\tshardID      int\n\t\tdomainID     string\n\t\tworkflowID   string\n\t\trunID        string\n\t\tsignalReqIDs []string\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:         \"ok\",\n\t\t\tshardID:      1000,\n\t\t\tdomainID:     \"domain_123\",\n\t\t\tworkflowID:   \"workflow_123\",\n\t\t\trunID:        \"runid_123\",\n\t\t\tsignalReqIDs: []string{\"signalReqID_1\", \"signalReqID_2\"},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_requested = [signalReqID_1 signalReqID_2] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain_123 and workflow_id = workflow_123 and ` +\n\t\t\t\t\t`run_id = runid_123 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\terr := resetSignalsRequested(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.signalReqIDs, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"resetSignalsRequested failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateSignalsRequested(t *testing.T) {\n\ttests := []struct {\n\t\tdesc               string\n\t\tshardID            int\n\t\tdomainID           string\n\t\tworkflowID         string\n\t\trunID              string\n\t\tsignalReqIDs       []string\n\t\tdeleteSignalReqIDs []string\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:               \"update only\",\n\t\t\tshardID:            1000,\n\t\t\tdomainID:           \"domain_abc\",\n\t\t\tworkflowID:         \"workflow_abc\",\n\t\t\trunID:              \"runid_abc\",\n\t\t\tsignalReqIDs:       []string{\"signalReqID_3\", \"signalReqID_4\"},\n\t\t\tdeleteSignalReqIDs: []string{},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_requested = signal_requested + [signalReqID_3 signalReqID_4] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain_abc and workflow_id = workflow_abc and ` +\n\t\t\t\t\t`run_id = runid_abc and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:               \"delete only\",\n\t\t\tshardID:            1001,\n\t\t\tdomainID:           \"domain_def\",\n\t\t\tworkflowID:         \"workflow_def\",\n\t\t\trunID:              \"runid_def\",\n\t\t\tsignalReqIDs:       []string{},\n\t\t\tdeleteSignalReqIDs: []string{\"signalReqID_5\", \"signalReqID_6\"},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_requested = signal_requested - [signalReqID_5 signalReqID_6] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1001 and type = 1 and domain_id = domain_def and workflow_id = workflow_def and ` +\n\t\t\t\t\t`run_id = runid_def and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:               \"update and delete\",\n\t\t\tshardID:            1002,\n\t\t\tdomainID:           \"domain_ghi\",\n\t\t\tworkflowID:         \"workflow_ghi\",\n\t\t\trunID:              \"runid_ghi\",\n\t\t\tsignalReqIDs:       []string{\"signalReqID_7\"},\n\t\t\tdeleteSignalReqIDs: []string{\"signalReqID_8\"},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_requested = signal_requested + [signalReqID_7] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1002 and type = 1 and domain_id = domain_ghi and workflow_id = workflow_ghi and ` +\n\t\t\t\t\t`run_id = runid_ghi and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`UPDATE executions SET signal_requested = signal_requested - [signalReqID_8] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1002 and type = 1 and domain_id = domain_ghi and workflow_id = workflow_ghi and ` +\n\t\t\t\t\t`run_id = runid_ghi and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\t\t\terr := updateSignalsRequested(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.signalReqIDs, tc.deleteSignalReqIDs, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateSignalsRequested failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetSignalInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc        string\n\t\tshardID     int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\tsignalInfos map[int64]*persistence.SignalInfo\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"single signal info\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tsignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tSignalRequestID:       \"request1\",\n\t\t\t\t\tSignalName:            \"signal1\",\n\t\t\t\t\tInput:                 []byte(\"input1\"),\n\t\t\t\t\tControl:               []byte(\"control1\"),\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           5,\n\t\t\t\t\tInitiatedEventBatchID: 6,\n\t\t\t\t\tSignalRequestID:       \"request2\",\n\t\t\t\t\tSignalName:            \"signal2\",\n\t\t\t\t\tInput:                 []byte(\"input2\"),\n\t\t\t\t\tControl:               []byte(\"control2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_map = map[` +\n\t\t\t\t\t`1:map[control:[99 111 110 116 114 111 108 49] initiated_event_batch_id:2 initiated_id:1 input:[105 110 112 117 116 49] signal_name:signal1 signal_request_id:request1 version:1] ` +\n\t\t\t\t\t`5:map[control:[99 111 110 116 114 111 108 50] initiated_event_batch_id:6 initiated_id:5 input:[105 110 112 117 116 50] signal_name:signal2 signal_request_id:request2 version:1]` +\n\t\t\t\t\t`] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetSignalInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.signalInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"resetSignalInfos failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateSignalInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc        string\n\t\tshardID     int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\tsignalInfos map[int64]*persistence.SignalInfo\n\t\tdeleteInfos []int64\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"update and delete signal infos\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tsignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tSignalRequestID:       \"request1\",\n\t\t\t\t\tSignalName:            \"signal1\",\n\t\t\t\t\tInput:                 []byte(\"input1\"),\n\t\t\t\t\tControl:               []byte(\"control1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tdeleteInfos: []int64{2},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET signal_map[ 1 ] = {` +\n\t\t\t\t\t`version: 1, initiated_id: 1, initiated_event_batch_id: 2, signal_request_id: request1, ` +\n\t\t\t\t\t`signal_name: signal1, input: [105 110 112 117 116 49], ` +\n\t\t\t\t\t`control: [99 111 110 116 114 111 108 49]` +\n\t\t\t\t\t`} , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and ` +\n\t\t\t\t\t`workflow_id = workflow1 and run_id = runid1 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`DELETE signal_map[ 2 ] FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and ` +\n\t\t\t\t\t`workflow_id = workflow1 and run_id = runid1 ` +\n\t\t\t\t\t`and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateSignalInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.signalInfos, tc.deleteInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateSignalInfos failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetRequestCancelInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc               string\n\t\tshardID            int\n\t\tdomainID           string\n\t\tworkflowID         string\n\t\trunID              string\n\t\trequestCancelInfos map[int64]*persistence.RequestCancelInfo\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\trequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tCancelRequestID:       \"cancelRequest1\",\n\t\t\t\t},\n\t\t\t\t3: {\n\t\t\t\t\tVersion:               2,\n\t\t\t\t\tInitiatedID:           3,\n\t\t\t\t\tInitiatedEventBatchID: 4,\n\t\t\t\t\tCancelRequestID:       \"cancelRequest3\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET request_cancel_map = map[` +\n\t\t\t\t\t`1:map[cancel_request_id:cancelRequest1 initiated_event_batch_id:2 initiated_id:1 version:1] ` +\n\t\t\t\t\t`3:map[cancel_request_id:cancelRequest3 initiated_event_batch_id:4 initiated_id:3 version:2]` +\n\t\t\t\t\t`], last_updated_time = 2025-01-06T15:00:00Z WHERE shard_id = 1000 and type = 1 and domain_id = domain1 and ` +\n\t\t\t\t\t`workflow_id = workflow1 and run_id = runid1 and ` +\n\t\t\t\t\t`visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetRequestCancelInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.requestCancelInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"resetRequestCancelInfos failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateRequestCancelInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc               string\n\t\tshardID            int\n\t\tdomainID           string\n\t\tworkflowID         string\n\t\trunID              string\n\t\trequestCancelInfos map[int64]*persistence.RequestCancelInfo\n\t\tdeleteInfos        []int64\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"update and delete request cancel infos\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\trequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tCancelRequestID:       \"cancelRequest1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tdeleteInfos: []int64{2},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET request_cancel_map[ 1 ] = ` +\n\t\t\t\t\t`{version: 1,initiated_id: 1, initiated_event_batch_id: 2, cancel_request_id: cancelRequest1 } , last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE shard_id = 1000 and type = 1 and domain_id = domain1 and ` +\n\t\t\t\t\t`workflow_id = workflow1 and run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`DELETE request_cancel_map[ 2 ] FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateRequestCancelInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.requestCancelInfos, tc.deleteInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateRequestCancelInfos failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetChildExecutionInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc                string\n\t\tshardID             int\n\t\tdomainID            string\n\t\tworkflowID          string\n\t\trunID               string\n\t\tchildExecutionInfos map[int64]*persistence.InternalChildExecutionInfo\n\t\t// expectations\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"execution info with runid\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tchildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tStartedID:             3,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tStartedWorkflowID: \"startedWorkflowID1\",\n\t\t\t\t\tStartedRunID:      \"startedRunID1\",\n\t\t\t\t\tCreateRequestID:   \"createRequestID1\",\n\t\t\t\t\tInitiatedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tDomainID:          \"domain1\",\n\t\t\t\t\tWorkflowTypeName:  \"workflowType1\",\n\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET child_executions_map = ` +\n\t\t\t\t\t`map[1:map[` +\n\t\t\t\t\t`create_request_id:createRequestID1 domain_id:domain1 domain_name: event_data_encoding:thriftrw ` +\n\t\t\t\t\t`initiated_event:[] initiated_event_batch_id:2 initiated_id:1 parent_close_policy:0 ` +\n\t\t\t\t\t`started_event:[] started_id:3 started_run_id:startedRunID1 started_workflow_id:startedWorkflowID1 ` +\n\t\t\t\t\t`version:1 workflow_type_name:workflowType1` +\n\t\t\t\t\t`]], last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:       \"execution info without runid\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      emptyRunID,\n\t\t\tchildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tStartedID:             3,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tStartedWorkflowID: \"startedWorkflowID1\",\n\t\t\t\t\tStartedRunID:      \"\", // leave empty and validate it's querying empty runid\n\t\t\t\t\tCreateRequestID:   \"createRequestID1\",\n\t\t\t\t\tInitiatedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tDomainID:          \"domain1\",\n\t\t\t\t\tWorkflowTypeName:  \"workflowType1\",\n\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET child_executions_map = ` +\n\t\t\t\t\t`map[1:map[` +\n\t\t\t\t\t`create_request_id:createRequestID1 domain_id:domain1 domain_name: event_data_encoding:thriftrw ` +\n\t\t\t\t\t`initiated_event:[] initiated_event_batch_id:2 initiated_id:1 parent_close_policy:0 ` +\n\t\t\t\t\t`started_event:[] started_id:3 started_run_id:30000000-0000-f000-f000-000000000000 started_workflow_id:startedWorkflowID1 ` +\n\t\t\t\t\t`version:1 workflow_type_name:workflowType1` +\n\t\t\t\t\t`]], last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = 30000000-0000-f000-f000-000000000000 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetChildExecutionInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.childExecutionInfos, FixedTime)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"resetChildExecutionInfos() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateChildExecutionInfos(t *testing.T) {\n\ttests := []struct {\n\t\tdesc                string\n\t\tshardID             int\n\t\tdomainID            string\n\t\tworkflowID          string\n\t\trunID               string\n\t\tchildExecutionInfos map[int64]*persistence.InternalChildExecutionInfo\n\t\tdeleteInfos         []int64\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"update and delete child execution infos\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tchildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion:               1,\n\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\tStartedID:             3,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tStartedWorkflowID: \"startedWorkflowID1\",\n\t\t\t\t\tStartedRunID:      \"startedRunID1\",\n\t\t\t\t\tCreateRequestID:   \"createRequestID1\",\n\t\t\t\t\tInitiatedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\tDomainID:          \"domain1\",\n\t\t\t\t\tWorkflowTypeName:  \"workflowType1\",\n\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t\t},\n\t\t\t},\n\t\t\tdeleteInfos: []int64{2},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET child_executions_map[ 1 ] = {` +\n\t\t\t\t\t`version: 1, initiated_id: 1, initiated_event_batch_id: 2, initiated_event: [], ` +\n\t\t\t\t\t`started_id: 3, started_workflow_id: startedWorkflowID1, started_run_id: startedRunID1, ` +\n\t\t\t\t\t`started_event: [], create_request_id: createRequestID1, event_data_encoding: thriftrw, ` +\n\t\t\t\t\t`domain_id: domain1, domain_name: , workflow_type_name: workflowType1, parent_close_policy: 0` +\n\t\t\t\t\t`} , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`DELETE child_executions_map[ 2 ] FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateChildExecutionInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.childExecutionInfos, tc.deleteInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateChildExecutionInfos failed: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetTimerInfos(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-12T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\t\ttimerInfos map[string]*persistence.TimerInfo\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\ttimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\"timer1\": {\n\t\t\t\t\tVersion:    1,\n\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t\tExpiryTime: ts.UTC(),\n\t\t\t\t\tTaskStatus: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET timer_map = map[` +\n\t\t\t\t\t`timer1:map[expiry_time:2023-12-12 22:08:41 +0000 UTC started_id:2 task_id:1 timer_id:timer1 version:1]` +\n\t\t\t\t\t`] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetTimerInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.timerInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"resetTimerInfos() error = %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTimerInfos(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc        string\n\t\tshardID     int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\t\ttimerInfos  map[string]*persistence.TimerInfo\n\t\tdeleteInfos []string\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"update and delete timer infos\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\ttimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\"timer1\": {\n\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\tVersion:    1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t\tExpiryTime: ts.UTC(),\n\t\t\t\t\tTaskStatus: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tdeleteInfos: []string{\"timer2\"},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET timer_map[ timer1 ] = {` +\n\t\t\t\t\t`version: 1, timer_id: timer1, started_id: 2, expiry_time: 2023-12-19T22:08:41Z, task_id: 1` +\n\t\t\t\t\t`} , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`DELETE timer_map[ timer2 ] FROM executions WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateTimerInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.timerInfos, tc.deleteInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateTimerInfos() error = %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetActivityInfos(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc          string\n\t\tshardID       int\n\t\tdomainID      string\n\t\tworkflowID    string\n\t\trunID         string\n\t\tactivityInfos map[int64]*persistence.InternalActivityInfo\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tactivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tScheduledEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-scheduled-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tScheduledTime: ts.UTC(),\n\t\t\t\t\tScheduleID:    1,\n\t\t\t\t\tStartedID:     2,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-started-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tActivityID:             \"activity1\",\n\t\t\t\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\t\t\t\tScheduleToCloseTimeout: 2 * time.Minute,\n\t\t\t\t\tStartToCloseTimeout:    3 * time.Minute,\n\t\t\t\t\tHeartbeatTimeout:       1 * time.Minute,\n\t\t\t\t\tAttempt:                3,\n\t\t\t\t\tMaximumAttempts:        5,\n\t\t\t\t\tTaskList:               \"tasklist1\",\n\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\tLastFailureReason:      \"retry reason\",\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tScheduledEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-scheduled-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tScheduledTime: ts.UTC(),\n\t\t\t\t\tScheduleID:    2,\n\t\t\t\t\tStartedID:     3,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-started-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tActivityID:             \"activity2\",\n\t\t\t\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\t\t\t\tScheduleToCloseTimeout: 2 * time.Minute,\n\t\t\t\t\tStartToCloseTimeout:    3 * time.Minute,\n\t\t\t\t\tHeartbeatTimeout:       1 * time.Minute,\n\t\t\t\t\tAttempt:                1,\n\t\t\t\t\tMaximumAttempts:        5,\n\t\t\t\t\tTaskList:               \"tasklist1\",\n\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\tLastFailureReason:      \"another retry reason\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET activity_map = map[` +\n\t\t\t\t\t`1:map[` +\n\t\t\t\t\t`activity_id:activity1 attempt:3 backoff_coefficient:0 cancel_request_id:0 cancel_requested:false ` +\n\t\t\t\t\t`details:[] event_data_encoding:thriftrw expiration_time:0001-01-01 00:00:00 +0000 UTC has_retry_policy:true ` +\n\t\t\t\t\t`heart_beat_timeout:60 init_interval:0 last_failure_details:[] last_failure_reason:retry reason ` +\n\t\t\t\t\t`last_hb_updated_time:0001-01-01 00:00:00 +0000 UTC last_worker_identity: max_attempts:5 max_interval:0 ` +\n\t\t\t\t\t`non_retriable_errors:[] request_id: schedule_id:1 schedule_to_close_timeout:120 schedule_to_start_timeout:60 ` +\n\t\t\t\t\t`scheduled_event:[116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 99 104 101 100 117 108 101 100 45 101 118 101 110 116 45 100 97 116 97] ` +\n\t\t\t\t\t`scheduled_event_batch_id:0 scheduled_time:2023-12-19 22:08:41 +0000 UTC start_to_close_timeout:180 ` +\n\t\t\t\t\t`started_event:[116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 116 97 114 116 101 100 45 101 118 101 110 116 45 100 97 116 97] ` +\n\t\t\t\t\t`started_id:2 started_identity: started_time:0001-01-01 00:00:00 +0000 UTC task_list:tasklist1 timer_task_status:0 version:1` +\n\t\t\t\t\t`] ` +\n\t\t\t\t\t`2:map[` +\n\t\t\t\t\t`activity_id:activity2 attempt:1 backoff_coefficient:0 cancel_request_id:0 cancel_requested:false ` +\n\t\t\t\t\t`details:[] event_data_encoding:thriftrw expiration_time:0001-01-01 00:00:00 +0000 UTC has_retry_policy:true ` +\n\t\t\t\t\t`heart_beat_timeout:60 init_interval:0 last_failure_details:[] last_failure_reason:another retry reason ` +\n\t\t\t\t\t`last_hb_updated_time:0001-01-01 00:00:00 +0000 UTC last_worker_identity: max_attempts:5 max_interval:0 ` +\n\t\t\t\t\t`non_retriable_errors:[] request_id: schedule_id:2 schedule_to_close_timeout:120 schedule_to_start_timeout:60 ` +\n\t\t\t\t\t`scheduled_event:[116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 99 104 101 100 117 108 101 100 45 101 118 101 110 116 45 100 97 116 97] ` +\n\t\t\t\t\t`scheduled_event_batch_id:0 scheduled_time:2023-12-19 22:08:41 +0000 UTC start_to_close_timeout:180 ` +\n\t\t\t\t\t`started_event:[116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 116 97 114 116 101 100 45 101 118 101 110 116 45 100 97 116 97] ` +\n\t\t\t\t\t`started_id:3 started_identity: started_time:0001-01-01 00:00:00 +0000 UTC task_list:tasklist1 timer_task_status:0 version:1` +\n\t\t\t\t\t`]` +\n\t\t\t\t\t`] , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetActivityInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.activityInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"resetActivityInfos() error = %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateActivityInfos(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc          string\n\t\tshardID       int\n\t\tdomainID      string\n\t\tworkflowID    string\n\t\trunID         string\n\t\tactivityInfos map[int64]*persistence.InternalActivityInfo\n\t\tdeleteInfos   []int64\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"update and delete activity infos\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\trunID:      \"runid1\",\n\t\t\tactivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t1: {\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tScheduledEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-scheduled-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tScheduledTime: ts.UTC(),\n\t\t\t\t\tScheduleID:    1,\n\t\t\t\t\tStartedID:     2,\n\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-started-event-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tActivityID:             \"activity1\",\n\t\t\t\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\t\t\t\tScheduleToCloseTimeout: 2 * time.Minute,\n\t\t\t\t\tStartToCloseTimeout:    3 * time.Minute,\n\t\t\t\t\tHeartbeatTimeout:       1 * time.Minute,\n\t\t\t\t\tAttempt:                3,\n\t\t\t\t\tMaximumAttempts:        5,\n\t\t\t\t\tTaskList:               \"tasklist1\",\n\t\t\t\t\tTaskListKind:           types.TaskListKindEphemeral,\n\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\tLastFailureReason:      \"retry reason\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tdeleteInfos: []int64{2},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET activity_map[ 1 ] = {` +\n\t\t\t\t\t`version: 1, schedule_id: 1, scheduled_event_batch_id: 0, ` +\n\t\t\t\t\t`scheduled_event: [116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 99 104 101 100 117 108 101 100 45 101 118 101 110 116 45 100 97 116 97], ` +\n\t\t\t\t\t`scheduled_time: 2023-12-19T22:08:41Z, started_id: 2, ` +\n\t\t\t\t\t`started_event: [116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 115 116 97 114 116 101 100 45 101 118 101 110 116 45 100 97 116 97], ` +\n\t\t\t\t\t`started_time: 0001-01-01T00:00:00Z, activity_id: activity1, request_id: , ` +\n\t\t\t\t\t`details: [], schedule_to_start_timeout: 60, schedule_to_close_timeout: 120, start_to_close_timeout: 180, ` +\n\t\t\t\t\t`heart_beat_timeout: 60, cancel_requested: false, cancel_request_id: 0, last_hb_updated_time: 0001-01-01T00:00:00Z, ` +\n\t\t\t\t\t`timer_task_status: 0, attempt: 3, task_list: tasklist1, task_list_kind: 2, started_identity: , has_retry_policy: true, ` +\n\t\t\t\t\t`init_interval: 0, backoff_coefficient: 0, max_interval: 0, expiration_time: 0001-01-01T00:00:00Z, ` +\n\t\t\t\t\t`max_attempts: 5, non_retriable_errors: [], last_failure_reason: retry reason, last_worker_identity: , ` +\n\t\t\t\t\t`last_failure_details: [], event_data_encoding: thriftrw` +\n\t\t\t\t\t`} , last_updated_time = 2025-01-06T15:00:00Z WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t\t`DELETE activity_map[ 2 ] FROM executions ` +\n\t\t\t\t\t`WHERE shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateActivityInfos(batch, tc.shardID, tc.domainID, tc.workflowID, tc.runID, tc.activityInfos, tc.deleteInfos, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateActivityInfos() error = %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateWorkflowExecutionWithMergeMaps(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.WorkflowExecutionRequest\n\t\t// expectations\n\t\twantQueries int\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"EventBufferWriteMode is not None\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeAppend,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"MapsWriteMode is not Create\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeNone,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeUpdate,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeNone,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeCreate,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t\tActivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\tScheduledEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\tData:     []byte(\"thrift-encoded-scheduled-event-data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tScheduledTime: ts.UTC(),\n\t\t\t\t\t\tScheduleID:    1,\n\t\t\t\t\t\tStartedID:     2,\n\t\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\tData:     []byte(\"thrift-encoded-started-event-data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActivityID:             \"activity1\",\n\t\t\t\t\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\t\t\t\t\tScheduleToCloseTimeout: 2 * time.Minute,\n\t\t\t\t\t\tStartToCloseTimeout:    3 * time.Minute,\n\t\t\t\t\t\tHeartbeatTimeout:       1 * time.Minute,\n\t\t\t\t\t\tAttempt:                3,\n\t\t\t\t\t\tMaximumAttempts:        5,\n\t\t\t\t\t\tTaskList:               \"tasklist1\",\n\t\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\t\tLastFailureReason:      \"retry reason\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\t\"timer1\": {\n\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\t\tStartedID:  2,\n\t\t\t\t\t\tExpiryTime: ts,\n\t\t\t\t\t\tTaskStatus: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tChildWorkflowInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tStartedID:             3,\n\t\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartedWorkflowID: \"startedWorkflowID1\",\n\t\t\t\t\t\tStartedRunID:      \"startedRunID1\",\n\t\t\t\t\t\tCreateRequestID:   \"createRequestID1\",\n\t\t\t\t\t\tInitiatedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID:          \"domain1\",\n\t\t\t\t\t\tWorkflowTypeName:  \"workflowType1\",\n\t\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tRequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tCancelRequestID:       \"cancelRequest1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tSignalRequestID:       \"request1\",\n\t\t\t\t\t\tSignalName:            \"signal1\",\n\t\t\t\t\t\tInput:                 []byte(\"input1\"),\n\t\t\t\t\t\tControl:               []byte(\"control1\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSignalRequestedIDs: []string{\"signalRequestedID1\"},\n\t\t\t},\n\t\t\t// expecting 7 queries:\n\t\t\t// - 1 for execution record\n\t\t\t// - 1 for activity info\n\t\t\t// - 1 for timer info\n\t\t\t// - 1 for child execution info\n\t\t\t// - 1 for request cancel info\n\t\t\t// - 1 for signal info\n\t\t\t// - 1 for signal requested IDs\n\t\t\twantQueries: 7,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := createWorkflowExecutionWithMergeMaps(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// actual queries generated by helper functions are covered in other unit tests. check the numer of total queries here.\n\t\t\tif got := len(batch.queries); got != tc.wantQueries {\n\t\t\t\tt.Fatalf(\"len(queries): %v, want: %v\", got, tc.wantQueries)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetWorkflowExecutionAndMapsAndEventBuffer(t *testing.T) {\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.WorkflowExecutionRequest\n\t\t// expectations\n\t\twantQueries int\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"EventBufferWriteMode is not Clear\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeAppend,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"MapsWriteMode is not Reset\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeUpdate, // Incorrect mode\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeReset,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\t// expecting 8 queries:\n\t\t\t// - 1 for execution record\n\t\t\t// - 1 for deletion of buffered events\n\t\t\t// - 1 for activity info map reset\n\t\t\t// - 1 for timer info map reset\n\t\t\t// - 1 for child execution info map reset\n\t\t\t// - 1 for request cancel info map reset\n\t\t\t// - 1 for signal info map reset\n\t\t\t// - 1 for signal requested IDs reset\n\t\t\twantQueries: 8,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := resetWorkflowExecutionAndMapsAndEventBuffer(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check the number of total queries, actual queries generated by helper functions are covered in other unit tests.\n\t\t\tif got := len(batch.queries); got != tc.wantQueries {\n\t\t\t\tt.Fatalf(\"len(queries): %v, want: %v\", got, tc.wantQueries)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionAndEventBufferWithMergeAndDeleteMaps(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.WorkflowExecutionRequest\n\t\t// expectations\n\t\twantQueries int\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"MapsWriteMode is not Update\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeCreate, // Incorrect mode\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeUpdate,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tCompletionEvent: &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints: &persistence.DataBlob{},\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tChecksums:        &checksum.Checksum{},\n\t\t\t\tActivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\tScheduledEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\tData:     []byte(\"thrift-encoded-scheduled-event-data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tScheduledTime: ts.UTC(),\n\t\t\t\t\t\tScheduleID:    1,\n\t\t\t\t\t\tStartedID:     2,\n\t\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\tData:     []byte(\"thrift-encoded-started-event-data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tActivityID:             \"activity1\",\n\t\t\t\t\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\t\t\t\t\tScheduleToCloseTimeout: 2 * time.Minute,\n\t\t\t\t\t\tStartToCloseTimeout:    3 * time.Minute,\n\t\t\t\t\t\tHeartbeatTimeout:       1 * time.Minute,\n\t\t\t\t\t\tAttempt:                3,\n\t\t\t\t\t\tMaximumAttempts:        5,\n\t\t\t\t\t\tTaskList:               \"tasklist1\",\n\t\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\t\tLastFailureReason:      \"retry reason\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\t\"timer1\": {\n\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\t\tStartedID:  2,\n\t\t\t\t\t\tExpiryTime: ts.UTC(),\n\t\t\t\t\t\tTaskStatus: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tChildWorkflowInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tStartedID:             3,\n\t\t\t\t\t\tStartedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartedWorkflowID: \"startedWorkflowID1\",\n\t\t\t\t\t\tStartedRunID:      \"startedRunID1\",\n\t\t\t\t\t\tCreateRequestID:   \"createRequestID1\",\n\t\t\t\t\t\tInitiatedEvent: &persistence.DataBlob{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID:          \"domain1\",\n\t\t\t\t\t\tWorkflowTypeName:  \"workflowType1\",\n\t\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tRequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tCancelRequestID:       \"cancelRequest1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t\t1: {\n\t\t\t\t\t\tVersion:               1,\n\t\t\t\t\t\tInitiatedID:           1,\n\t\t\t\t\t\tInitiatedEventBatchID: 2,\n\t\t\t\t\t\tSignalRequestID:       \"request1\",\n\t\t\t\t\t\tSignalName:            \"signal1\",\n\t\t\t\t\t\tInput:                 []byte(\"input1\"),\n\t\t\t\t\t\tControl:               []byte(\"control1\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSignalRequestedIDs: []string{\"signalRequestedID1\"},\n\t\t\t},\n\t\t\t// expecting 8 queries:\n\t\t\t// - 1 for execution record\n\t\t\t// - 1 for deletion of buffered events\n\t\t\t// - 1 for activity info map update\n\t\t\t// - 1 for timer info map update\n\t\t\t// - 1 for child execution info map update\n\t\t\t// - 1 for request cancel info map update\n\t\t\t// - 1 for signal info map update\n\t\t\t// - 1 for signal requested IDs update\n\t\t\twantQueries: 8,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateWorkflowExecutionAndEventBufferWithMergeAndDeleteMaps(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Check the number of total queries, actual queries generated by helper functions are covered in other unit tests.\n\t\t\tif got := len(batch.queries); got != tc.wantQueries {\n\t\t\t\tt.Fatalf(\"len(queries): %v, want: %v\", got, tc.wantQueries)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecution(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.WorkflowExecutionRequest\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeUpdate,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:             \"domain1\",\n\t\t\t\t\tWorkflowID:           \"workflow1\",\n\t\t\t\t\tRunID:                \"runid1\",\n\t\t\t\t\tParentRunID:          \"parentRunID1\",\n\t\t\t\t\tWorkflowTypeName:     \"workflowType1\",\n\t\t\t\t\tTaskList:             \"tasklist1\",\n\t\t\t\t\tTaskListKind:         types.TaskListKindNormal,\n\t\t\t\t\tStartTimestamp:       ts,\n\t\t\t\t\tLastUpdatedTimestamp: ts.Add(1 * time.Minute),\n\t\t\t\t\tDecisionScheduleID:   2,\n\t\t\t\t\tDecisionStartedID:    3,\n\t\t\t\t\tCompletionEvent:      &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints:      &persistence.DataBlob{},\n\t\t\t\t\tCronOverlapPolicy:    0,\n\t\t\t\t},\n\t\t\t\tPreviousNextEventIDCondition: common.Int64Ptr(10),\n\t\t\t\tVersionHistories:             &persistence.DataBlob{},\n\t\t\t\tChecksums:                    &checksum.Checksum{},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions SET execution = {` +\n\t\t\t\t\t`domain_id: domain1, workflow_id: workflow1, run_id: runid1, first_run_id: , parent_domain_id: , parent_workflow_id: , ` +\n\t\t\t\t\t`parent_run_id: parentRunID1, initiated_id: 0, completion_event_batch_id: 0, completion_event: [], ` +\n\t\t\t\t\t`completion_event_data_encoding: , task_list: tasklist1, task_list_kind: 0, workflow_type_name: workflowType1, workflow_timeout: 0, ` +\n\t\t\t\t\t`decision_task_timeout: 0, execution_context: [], state: 0, close_status: 0, last_first_event_id: 0, last_event_task_id: 0, ` +\n\t\t\t\t\t`next_event_id: 0, last_processed_event: 0, start_time: 2023-12-19T22:08:41Z, last_updated_time: 2023-12-19T22:09:41Z, ` +\n\t\t\t\t\t`create_request_id: , signal_count: 0, history_size: 0, decision_version: 0, decision_schedule_id: 2, decision_started_id: 3, ` +\n\t\t\t\t\t`decision_request_id: , decision_timeout: 0, decision_attempt: 0, decision_timestamp: -6795364578871345152, ` +\n\t\t\t\t\t`decision_scheduled_timestamp: -6795364578871345152, decision_original_scheduled_timestamp: -6795364578871345152, ` +\n\t\t\t\t\t`cancel_requested: false, cancel_request_id: , sticky_task_list: , sticky_schedule_to_start_timeout: 0,client_library_version: , ` +\n\t\t\t\t\t`client_feature_version: , client_impl: , auto_reset_points: [], auto_reset_points_encoding: , attempt: 0, has_retry_policy: false, ` +\n\t\t\t\t\t`init_interval: 0, backoff_coefficient: 0, max_interval: 0, expiration_time: 0001-01-01T00:00:00Z, max_attempts: 0, ` +\n\t\t\t\t\t`non_retriable_errors: [], event_store_version: 2, branch_token: [], cron_schedule: , cron_overlap_policy: 0, expiration_seconds: 0, search_attributes: map[], ` +\n\t\t\t\t\t`memo: map[], partition_config: map[], active_cluster_selection_policy: [], active_cluster_selection_policy_encoding: ` +\n\t\t\t\t\t`}, next_event_id = 0 , version_histories = [] , version_histories_encoding =  , checksum = {version: 0, flavor: 0, value: [] }, workflow_last_write_version = 0 , workflow_state = 0 , last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = runid1 and visibility_ts = 946684800000 and task_id = -10 ` +\n\t\t\t\t\t`IF next_event_id = 10 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := updateWorkflowExecution(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"updateWorkflowExecution failed, err: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateWorkflowExecution(t *testing.T) {\n\tts, err := time.Parse(time.RFC3339, \"2023-12-19T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.WorkflowExecutionRequest\n\t\t// expectations\n\t\twantQueries []string\n\t}{\n\t\t{\n\t\t\tdesc:       \"ok\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.WorkflowExecutionRequest{\n\t\t\t\tEventBufferWriteMode: nosqlplugin.EventBufferWriteModeClear,\n\t\t\t\tMapsWriteMode:        nosqlplugin.WorkflowExecutionMapsWriteModeUpdate,\n\t\t\t\tInternalWorkflowExecutionInfo: persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:             \"domain1\",\n\t\t\t\t\tWorkflowID:           \"workflow1\",\n\t\t\t\t\tRunID:                \"runid1\",\n\t\t\t\t\tParentRunID:          \"parentRunID1\",\n\t\t\t\t\tWorkflowTypeName:     \"workflowType1\",\n\t\t\t\t\tTaskList:             \"tasklist1\",\n\t\t\t\t\tTaskListKind:         types.TaskListKindNormal,\n\t\t\t\t\tStartTimestamp:       ts,\n\t\t\t\t\tLastUpdatedTimestamp: ts.Add(1 * time.Minute),\n\t\t\t\t\tDecisionScheduleID:   2,\n\t\t\t\t\tDecisionStartedID:    3,\n\t\t\t\t\tCompletionEvent:      &persistence.DataBlob{},\n\t\t\t\t\tAutoResetPoints:      &persistence.DataBlob{},\n\t\t\t\t\tActiveClusterSelectionPolicy: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\tData:     []byte(\"thrift-encoded-active-cluster-selection-policy-data\"),\n\t\t\t\t\t},\n\t\t\t\t\tCronOverlapPolicy: types.CronOverlapPolicyBufferOne,\n\t\t\t\t},\n\t\t\t\tPreviousNextEventIDCondition: common.Int64Ptr(10),\n\t\t\t\tVersionHistories:             &persistence.DataBlob{},\n\t\t\t\tChecksums:                    &checksum.Checksum{},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, domain_id, workflow_id, run_id, type, execution, next_event_id, visibility_ts, task_id, version_histories, version_histories_encoding, checksum, workflow_last_write_version, workflow_state, created_time) ` +\n\t\t\t\t\t`VALUES(1000, domain1, workflow1, runid1, 1, ` +\n\t\t\t\t\t`{domain_id: domain1, workflow_id: workflow1, run_id: runid1, first_run_id: , parent_domain_id: , parent_workflow_id: , ` +\n\t\t\t\t\t`parent_run_id: parentRunID1, initiated_id: 0, completion_event_batch_id: 0, completion_event: [], completion_event_data_encoding: , ` +\n\t\t\t\t\t`task_list: tasklist1, task_list_kind: 0, workflow_type_name: workflowType1, workflow_timeout: 0, decision_task_timeout: 0, execution_context: [], state: 0, ` +\n\t\t\t\t\t`close_status: 0, last_first_event_id: 0, last_event_task_id: 0, next_event_id: 0, last_processed_event: 0, start_time: 2023-12-19T22:08:41Z, ` +\n\t\t\t\t\t`last_updated_time: 2023-12-19T22:09:41Z, create_request_id: , signal_count: 0, history_size: 0, decision_version: 0, ` +\n\t\t\t\t\t`decision_schedule_id: 2, decision_started_id: 3, decision_request_id: , decision_timeout: 0, decision_attempt: 0, ` +\n\t\t\t\t\t`decision_timestamp: -6795364578871345152, decision_scheduled_timestamp: -6795364578871345152, decision_original_scheduled_timestamp: -6795364578871345152, ` +\n\t\t\t\t\t`cancel_requested: false, cancel_request_id: , sticky_task_list: , sticky_schedule_to_start_timeout: 0,client_library_version: , client_feature_version: , ` +\n\t\t\t\t\t`client_impl: , auto_reset_points: [], auto_reset_points_encoding: , attempt: 0, has_retry_policy: false, init_interval: 0, ` +\n\t\t\t\t\t`backoff_coefficient: 0, max_interval: 0, expiration_time: 0001-01-01T00:00:00Z, max_attempts: 0, non_retriable_errors: [], ` +\n\t\t\t\t\t`event_store_version: 2, branch_token: [], cron_schedule: , cron_overlap_policy: 1, expiration_seconds: 0, search_attributes: map[], memo: map[], partition_config: map[], ` +\n\t\t\t\t\t`active_cluster_selection_policy: [116 104 114 105 102 116 45 101 110 99 111 100 101 100 45 97 99 116 105 118 101 45 99 108 117 115 116 101 114 45 115 101 108 101 99 116 105 111 110 45 112 111 108 105 99 121 45 100 97 116 97], active_cluster_selection_policy_encoding: thriftrw` +\n\t\t\t\t\t`}, 0, 946684800000, -10, [], , {version: 0, flavor: 0, value: [] }, 0, 0, 2025-01-06T15:00:00Z) IF NOT EXISTS `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := createWorkflowExecution(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"createWorkflowExecution failed, err: %v\", err)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateOrUpdateWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tdesc       string\n\t\tshardID    int\n\t\tdomainID   string\n\t\tworkflowID string\n\t\texecution  *nosqlplugin.CurrentWorkflowWriteRequest\n\t\t// expectations\n\t\twantQueries []string\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"unknown write mode\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: 255, // unknown write mode\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeNoop\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeNoop,\n\t\t\t},\n\t\t\twantQueries: nil,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeInsert\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeInsert,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tCreateRequestID: \"createRequestID1\",\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`INSERT INTO executions (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id, current_run_id, execution, workflow_last_write_version, workflow_state, created_time) ` +\n\t\t\t\t\t`VALUES(1000, 1, domain1, workflow1, 30000000-0000-f000-f000-000000000001, 946684800000, -10, runid1, ` +\n\t\t\t\t\t`{run_id: runid1, create_request_id: createRequestID1, state: 0, close_status: 0}, 0, 0, 2025-01-06T15:00:00Z) ` +\n\t\t\t\t\t`IF NOT EXISTS USING TTL 0 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeUpdate and condition missing\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tCreateRequestID: \"createRequestID1\",\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeUpdate and condition runid missing\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: nil,\n\t\t\t\t},\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tCreateRequestID: \"createRequestID1\",\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeUpdate with LastWriteVersion\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID:     common.StringPtr(\"runid1\"),\n\t\t\t\t\tLastWriteVersion: common.Int64Ptr(1),\n\t\t\t\t\tState:            common.IntPtr(persistence.WorkflowStateCreated),\n\t\t\t\t},\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tCreateRequestID: \"createRequestID1\",\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions USING TTL 0 SET ` +\n\t\t\t\t\t`current_run_id = runid1, ` +\n\t\t\t\t\t`execution = {run_id: runid1, create_request_id: createRequestID1, state: 0, close_status: 0}, ` +\n\t\t\t\t\t`workflow_last_write_version = 0, workflow_state = 0, last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = 30000000-0000-f000-f000-000000000001 and visibility_ts = 946684800000 and task_id = -10 ` +\n\t\t\t\t\t`IF current_run_id = runid1 and workflow_last_write_version = 1 and workflow_state = 0 `,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:       \"CurrentWorkflowWriteModeUpdate\",\n\t\t\tshardID:    1000,\n\t\t\tdomainID:   \"domain1\",\n\t\t\tworkflowID: \"workflow1\",\n\t\t\texecution: &nosqlplugin.CurrentWorkflowWriteRequest{\n\t\t\t\tWriteMode: nosqlplugin.CurrentWorkflowWriteModeUpdate,\n\t\t\t\tCondition: &nosqlplugin.CurrentWorkflowWriteCondition{\n\t\t\t\t\tCurrentRunID: common.StringPtr(\"runid1\"),\n\t\t\t\t},\n\t\t\t\tRow: nosqlplugin.CurrentWorkflowRow{\n\t\t\t\t\tRunID:           \"runid1\",\n\t\t\t\t\tCreateRequestID: \"createRequestID1\",\n\t\t\t\t\tState:           persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantQueries: []string{\n\t\t\t\t`UPDATE executions USING TTL 0 SET ` +\n\t\t\t\t\t`current_run_id = runid1, ` +\n\t\t\t\t\t`execution = {run_id: runid1, create_request_id: createRequestID1, state: 0, close_status: 0}, ` +\n\t\t\t\t\t`workflow_last_write_version = 0, workflow_state = 0, last_updated_time = 2025-01-06T15:00:00Z ` +\n\t\t\t\t\t`WHERE ` +\n\t\t\t\t\t`shard_id = 1000 and type = 1 and domain_id = domain1 and workflow_id = workflow1 and ` +\n\t\t\t\t\t`run_id = 30000000-0000-f000-f000-000000000001 and visibility_ts = 946684800000 and task_id = -10 ` +\n\t\t\t\t\t`IF current_run_id = runid1 `,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tbatch := &fakeBatch{}\n\n\t\t\terr := createOrUpdateCurrentWorkflow(batch, tc.shardID, tc.domainID, tc.workflowID, tc.execution, FixedTime)\n\t\t\tgotErr := (err != nil)\n\t\t\tif gotErr != tc.wantErr {\n\t\t\t\tt.Fatalf(\"Got error: %v, want?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif gotErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantQueries, batch.queries); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Query mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMustConvertToSlice(t *testing.T) {\n\ttests := []struct {\n\t\tdesc      string\n\t\tin        interface{}\n\t\twant      []interface{}\n\t\twantPanic bool\n\t}{\n\t\t{\n\t\t\tdesc:      \"nil\",\n\t\t\tin:        nil,\n\t\t\twantPanic: true,\n\t\t},\n\t\t{\n\t\t\tdesc: \"empty\",\n\t\t\tin:   []string{},\n\t\t\twant: []interface{}{},\n\t\t},\n\t\t{\n\t\t\tdesc: \"slice\",\n\t\t\tin:   []string{\"a\", \"b\", \"c\"},\n\t\t\twant: []interface{}{\"a\", \"b\", \"c\"},\n\t\t},\n\t\t{\n\t\t\tdesc: \"array\",\n\t\t\tin:   [3]string{\"a\", \"b\", \"c\"},\n\t\t\twant: []interface{}{\"a\", \"b\", \"c\"},\n\t\t},\n\t\t{\n\t\t\tdesc:      \"non-slice\",\n\t\t\tin:        \"a\",\n\t\t\twantPanic: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tdefer func() {\n\t\t\t\tr := recover()\n\t\t\t\tif (r != nil) != tc.wantPanic {\n\t\t\t\t\tt.Fatalf(\"Got panic: %v, want panic?: %v\", r, tc.wantPanic)\n\t\t\t\t}\n\t\t\t}()\n\n\t\t\tgot := mustConvertToSlice(tc.in)\n\t\t\tif diff := cmp.Diff(tc.want, got); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Slice mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsRequestRowType(t *testing.T) {\n\tassert.True(t, isRequestRowType(rowTypeWorkflowRequestStart))\n\tassert.True(t, isRequestRowType(rowTypeWorkflowRequestSignal))\n\tassert.True(t, isRequestRowType(rowTypeWorkflowRequestCancel))\n\tassert.True(t, isRequestRowType(rowTypeWorkflowRequestReset))\n}\n\nfunc errDiff(want, got error) string {\n\twantCondFailure, wantOk := want.(*nosqlplugin.WorkflowOperationConditionFailure)\n\tgotCondFailure, gotOk := got.(*nosqlplugin.WorkflowOperationConditionFailure)\n\tif wantOk && gotOk {\n\t\targ1 := trimWorkflowConditionalFailureErr(wantCondFailure)\n\t\targ2 := trimWorkflowConditionalFailureErr(gotCondFailure)\n\t\treturn cmp.Diff(arg1, arg2)\n\t}\n\n\twantMsg := \"\"\n\tif want != nil {\n\t\twantMsg = want.Error()\n\t}\n\tgotMsg := \"\"\n\tif got != nil {\n\t\tgotMsg = got.Error()\n\t}\n\treturn cmp.Diff(wantMsg, gotMsg)\n}\n\nfunc trimWorkflowConditionalFailureErr(condFailure *nosqlplugin.WorkflowOperationConditionFailure) any {\n\ttrimColumnsPart(condFailure.CurrentWorkflowConditionFailInfo)\n\ttrimColumnsPart(condFailure.UnknownConditionFailureDetails)\n\tif condFailure.WorkflowExecutionAlreadyExists != nil {\n\t\ttrimColumnsPart(&condFailure.WorkflowExecutionAlreadyExists.OtherInfo)\n\t}\n\treturn condFailure\n}\n\nfunc trimColumnsPart(s *string) {\n\tif s == nil {\n\t\treturn\n\t}\n\tre := regexp.MustCompile(`, columns: \\(.*\\)`)\n\ttrimmed := re.ReplaceAllString(*s, \"\")\n\t*s = trimmed\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/common.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosqlplugin\n\nimport (\n\t\"os\"\n\t\"strings\"\n)\n\nfunc getCadencePackageDir() (string, error) {\n\tcadencePackageDir, err := os.Getwd()\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tcadenceIndex := strings.LastIndex(cadencePackageDir, \"cadence/\")\n\tcadencePackageDir = cadencePackageDir[:cadenceIndex+len(\"cadence/\")]\n\treturn cadencePackageDir, err\n}\n\nfunc GetDefaultTestSchemaDir(testSchemaRelativePath string) (string, error) {\n\tcadencePackageDir, err := getCadencePackageDir()\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn cadencePackageDir + testSchemaRelativePath, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/admin.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\nvar _ nosqlplugin.AdminDB = (*ddb)(nil)\n\nfunc (db *ddb) SetupTestDatabase(schemaBaseDir string, replicas int) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) TeardownTestDatabase() error {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/configStore.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc (db *ddb) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\treturn errors.New(\"TODO\")\n}\n\nfunc (db *ddb) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\treturn nil, errors.New(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/db.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nconst (\n\t// PluginName is the name of the plugin\n\tPluginName = \"dynamodb\"\n)\n\nvar (\n\terrConditionFailed = errors.New(\"internal condition fail error\")\n)\n\n// ddb represents a logical connection to DynamoDB database\ntype ddb struct {\n}\n\nvar _ nosqlplugin.DB = (*ddb)(nil)\n\n// NewDynamoDB return a new DB\nfunc NewDynamoDB(cfg config.NoSQL, logger log.Logger) (nosqlplugin.DB, error) {\n\treturn nil, fmt.Errorf(\"TODO\")\n}\n\nfunc (db *ddb) Close() {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) PluginName() string {\n\treturn PluginName\n}\n\nfunc (db *ddb) IsNotFoundError(err error) bool {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) IsTimeoutError(err error) bool {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) IsThrottlingError(err error) bool {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) IsDBUnavailableError(err error) bool {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) IsConditionFailedError(err error) bool {\n\treturn err == errConditionFailed\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/domain.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Insert a new record to domain, return error if failed or already exists\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *ddb) InsertDomain(\n\tctx context.Context,\n\trow *nosqlplugin.DomainRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Update domain\nfunc (db *ddb) UpdateDomain(\n\tctx context.Context,\n\trow *nosqlplugin.DomainRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Get one domain data, either by domainID or domainName\nfunc (db *ddb) SelectDomain(\n\tctx context.Context,\n\tdomainID *string,\n\tdomainName *string,\n) (*nosqlplugin.DomainRow, error) {\n\tpanic(\"TODO\")\n}\n\n// Get all domain data\nfunc (db *ddb) SelectAllDomains(\n\tctx context.Context,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*nosqlplugin.DomainRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// Delete a domain, either by domainID or domainName\nfunc (db *ddb) DeleteDomain(\n\tctx context.Context,\n\tdomainID *string,\n\tdomainName *string,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectDomainMetadata(\n\tctx context.Context,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/domain_audit_log.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertDomainAuditLog inserts a new audit log entry for a domain operation\nfunc (db *ddb) InsertDomainAuditLog(ctx context.Context, row *nosqlplugin.DomainAuditLogRow) error {\n\tpanic(\"TODO: InsertDomainAuditLog not implemented\")\n}\n\n// SelectDomainAuditLogs returns audit log entries for a domain and operation type\nfunc (db *ddb) SelectDomainAuditLogs(ctx context.Context, filter *nosqlplugin.DomainAuditLogFilter) ([]*nosqlplugin.DomainAuditLogRow, []byte, error) {\n\tpanic(\"TODO: SelectDomainAuditLogs not implemented\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/events.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertIntoHistoryTreeAndNode inserts one or two rows: tree row and node row(at least one of them)\nfunc (db *ddb) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\tpanic(\"TODO\")\n}\n\n// SelectFromHistoryNode read nodes based on a filter\nfunc (db *ddb) SelectFromHistoryNode(ctx context.Context, filter *nosqlplugin.HistoryNodeFilter) ([]*nosqlplugin.HistoryNodeRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteFromHistoryTreeAndNode delete a branch record, and a list of ranges of nodes.\nfunc (db *ddb) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *nosqlplugin.HistoryTreeFilter, nodeFilters []*nosqlplugin.HistoryNodeFilter) error {\n\tpanic(\"TODO\")\n}\n\n// SelectAllHistoryTrees will return all tree branches with pagination\nfunc (db *ddb) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*nosqlplugin.HistoryTreeRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// SelectFromHistoryTree read branch records for a tree\nfunc (db *ddb) SelectFromHistoryTree(ctx context.Context, filter *nosqlplugin.HistoryTreeFilter) ([]*nosqlplugin.HistoryTreeRow, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Insert message into queue, return error if failed or already exists\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *ddb) InsertIntoQueue(\n\tctx context.Context,\n\trow *nosqlplugin.QueueMessageRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Get the ID of last message inserted into the queue\nfunc (db *ddb) SelectLastEnqueuedMessageID(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n\n// Read queue messages starting from the exclusiveBeginMessageID\nfunc (db *ddb) SelectMessagesFrom(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tmaxRows int,\n) ([]*nosqlplugin.QueueMessageRow, error) {\n\tpanic(\"TODO\")\n}\n\n// Read queue message starting from exclusiveBeginMessageID int64, inclusiveEndMessageID int64\nfunc (db *ddb) SelectMessagesBetween(\n\tctx context.Context,\n\trequest nosqlplugin.SelectMessagesBetweenRequest,\n) (*nosqlplugin.SelectMessagesBetweenResponse, error) {\n\tpanic(\"TODO\")\n}\n\n// Delete all messages before exclusiveBeginMessageID\nfunc (db *ddb) DeleteMessagesBefore(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Delete all messages in a range between exclusiveBeginMessageID and inclusiveEndMessageID\nfunc (db *ddb) DeleteMessagesInRange(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tinclusiveEndMessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Delete one message\nfunc (db *ddb) DeleteMessage(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Insert an empty metadata row, starting from a version\nfunc (db *ddb) InsertQueueMetadata(ctx context.Context, row nosqlplugin.QueueMetadataRow) error {\n\tpanic(\"TODO\")\n}\n\n// **Conditionally** update a queue metadata row, if current version is matched(meaning current == row.Version - 1),\n// then the current version will increase by one when updating the metadata row\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *ddb) UpdateQueueMetadataCas(\n\tctx context.Context,\n\trow nosqlplugin.QueueMetadataRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Read a QueueMetadata\nfunc (db *ddb) SelectQueueMetadata(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (*nosqlplugin.QueueMetadataRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) GetQueueSize(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/shard.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertShard creates a new shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) InsertShard(ctx context.Context, row *nosqlplugin.ShardRow) error {\n\tpanic(\"TODO\")\n}\n\n// SelectShard gets a shard\nfunc (db *ddb) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *nosqlplugin.ShardRow, error) {\n\tpanic(\"TODO\")\n}\n\n// UpdateRangeID updates the rangeID, return error is there is any\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) UpdateRangeID(ctx context.Context, shardID int, rangeID int64, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateShard updates a shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) UpdateShard(ctx context.Context, row *nosqlplugin.ShardRow, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/task.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// SelectTaskList returns a single tasklist row.\n// Return IsNotFoundError if the row doesn't exist\nfunc (db *ddb) SelectTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter) (*nosqlplugin.TaskListRow, error) {\n\tpanic(\"TODO\")\n}\n\n// InsertTaskList insert a single tasklist row\n// Return IsConditionFailedError if the row already exists, and also the existing row\nfunc (db *ddb) InsertTaskList(ctx context.Context, row *nosqlplugin.TaskListRow) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateTaskList updates a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) UpdateTaskList(\n\tctx context.Context,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateTaskList updates a single tasklist row, and set an TTL on the record\n// Return TaskOperationConditionFailure if the condition doesn't meet\n// Ignore TTL if it's not supported, which becomes exactly the same as UpdateTaskList, but ListTaskList must be\n// implemented for TaskListScavenger\nfunc (db *ddb) UpdateTaskListWithTTL(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// ListTaskList returns all tasklists.\n// Noop if TTL is already implemented in other methods\nfunc (db *ddb) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*nosqlplugin.ListTaskListResult, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteTaskList deletes a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) DeleteTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n\n// InsertTasks inserts a batch of tasks\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *ddb) InsertTasks(\n\tctx context.Context,\n\ttasksToInsert []*nosqlplugin.TaskRowForInsert,\n\ttasklistCondition *nosqlplugin.TaskListRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// SelectTasks return tasks that associated to a tasklist\nfunc (db *ddb) SelectTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) ([]*nosqlplugin.TaskRow, error) {\n\tpanic(\"TODO\")\n}\n\n// SelectTasks return tasks that associated to a tasklist\nfunc (db *ddb) GetTasksCount(ctx context.Context, filter *nosqlplugin.TasksFilter) (int64, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteTask delete a batch tasks that taskIDs less than the row\n// If TTL is not implemented, then should also return the number of rows deleted, otherwise persistence.UnknownNumRowsAffected\n// NOTE: This API ignores the `BatchSize` request parameter i.e. either all tasks leq the task_id will be deleted or an error will\n// be returned to the caller, because rowsDeleted is not supported by Cassandra\nfunc (db *ddb) RangeDeleteTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) (rowsDeleted int, err error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/visibility.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc (db *ddb) InsertVisibility(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.VisibilityRowForInsert,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) UpdateVisibility(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.VisibilityRowForUpdate,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectVisibility(\n\tctx context.Context,\n\tfilter *nosqlplugin.VisibilityFilter,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteVisibility(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectOneClosedWorkflow(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) (*nosqlplugin.VisibilityRow, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/dynamodb/workflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nvar _ nosqlplugin.WorkflowCRUD = (*ddb)(nil)\n\nfunc (db *ddb) InsertWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) UpdateWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\tmutatedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tinsertedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tresetExecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*nosqlplugin.CurrentWorkflowRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*nosqlplugin.WorkflowExecution, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) RangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) RangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) InsertReplicationTask(ctx context.Context, tasks []*nosqlplugin.HistoryMigrationTask, condition nosqlplugin.ShardCondition) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *nosqlplugin.HistoryMigrationTask) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*nosqlplugin.ActiveClusterSelectionPolicyRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *ddb) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) error {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/errors.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosqlplugin\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// Condition Errors for NoSQL interfaces\ntype (\n\t// Only one of the fields must be non-nil\n\tWorkflowOperationConditionFailure struct {\n\t\tUnknownConditionFailureDetails   *string // return some info for logging\n\t\tShardRangeIDNotMatch             *int64  // return the previous shardRangeID\n\t\tWorkflowExecutionAlreadyExists   *WorkflowExecutionAlreadyExists\n\t\tCurrentWorkflowConditionFailInfo *string // return the logging info if fail on condition of CurrentWorkflow\n\t\tDuplicateRequest                 *DuplicateRequest\n\t}\n\n\tDuplicateRequest struct {\n\t\tRequestType persistence.WorkflowRequestType\n\t\tRunID       string\n\t}\n\n\tWorkflowExecutionAlreadyExists struct {\n\t\tRunID            string\n\t\tCreateRequestID  string\n\t\tState            int\n\t\tCloseStatus      int\n\t\tLastWriteVersion int64\n\t\tOtherInfo        string\n\t}\n\n\tTaskOperationConditionFailure struct {\n\t\tRangeID int64\n\t\tDetails string // detail info for logging\n\t}\n\n\tShardOperationConditionFailure struct {\n\t\tRangeID int64\n\t\tDetails string // detail info for logging\n\t}\n\n\tConditionFailure struct {\n\t\tcomponentName string\n\t}\n)\n\nvar _ error = (*WorkflowOperationConditionFailure)(nil)\n\nfunc (e *WorkflowOperationConditionFailure) Error() string {\n\treturn \"workflow operation condition failure\"\n}\n\nvar _ error = (*TaskOperationConditionFailure)(nil)\n\nfunc (e *TaskOperationConditionFailure) Error() string {\n\treturn \"task operation condition failure\"\n}\n\nvar _ error = (*ShardOperationConditionFailure)(nil)\n\nfunc (e *ShardOperationConditionFailure) Error() string {\n\treturn \"shard operation condition failure\"\n}\n\nvar _ error = (*ConditionFailure)(nil)\n\nfunc NewConditionFailure(name string) error {\n\treturn &ConditionFailure{\n\t\tcomponentName: name,\n\t}\n}\nfunc (e *ConditionFailure) Error() string {\n\treturn fmt.Sprintf(\"%s operation condition failure\", e.componentName)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/interfaces.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go -self_package github.com/uber/cadence/common/persistence/nosql/nosqlplugin\n\npackage nosqlplugin\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// Plugin defines the interface for any NoSQL database that needs to implement\n\tPlugin interface {\n\t\tCreateDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (DB, error)\n\t\tCreateAdminDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (AdminDB, error)\n\t}\n\n\t// AdminDB is for tooling and testing\n\tAdminDB interface {\n\t\tSetupTestDatabase(schemaBaseDir string, replicas int) error\n\t\tTeardownTestDatabase() error\n\t}\n\n\t// DB defines the API for regular NoSQL operations of a Cadence server\n\tDB interface {\n\t\tPluginName() string\n\t\tClose()\n\n\t\tClientErrorChecker\n\t\ttableCRUD\n\t}\n\t// tableCRUD defines the API for interacting with the database tables\n\t// NOTE 1: All SELECT interfaces require strong consistency (eventual consistency will not work) unless specify in the method.\n\t//\n\t// NOTE 2: About schema: only the columns that need to be used directly in queries are considered 'significant',\n\t// including partition key, range key or index key and conditional columns, are required to be in the schema.\n\t// All other non 'significant' columns are opaque for implementation.\n\t// Therefore, it's recommended to use a data blob to store all the other columns, so that adding new\n\t// column will not require schema changes. This approach has been proved very successful in MySQL/Postgres implementation of SQL interfaces.\n\t// Cassandra implementation cannot do it due to backward-compatibility. Any other NoSQL implementation should use datablob for non-significant columns.\n\t// Follow the comment for each tableCRUD for what are 'significant' columns.\n\ttableCRUD interface {\n\t\tHistoryEventsCRUD\n\t\tMessageQueueCRUD\n\t\tDomainCRUD\n\t\tShardCRUD\n\t\tVisibilityCRUD\n\t\tTaskCRUD\n\t\tWorkflowCRUD\n\t\tConfigStoreCRUD\n\t\tDomainAuditLogCRUD\n\t}\n\n\t// ClientErrorChecker checks for common nosql errors on client\n\tClientErrorChecker interface {\n\t\tIsTimeoutError(error) bool\n\t\tIsNotFoundError(error) bool\n\t\tIsThrottlingError(error) bool\n\t\tIsDBUnavailableError(error) bool\n\t}\n\n\t/**\n\t * HistoryEventsCRUD is for History events storage system\n\t * Recommendation: use two tables: history_tree for branch records and history_node for node records\n\t * if a single update query can operate on two tables.\n\t *\n\t * Significant columns:\n\t * history_tree partition key: (shardID, treeID), range key: (branchID)\n\t * history_node partition key: (shardID, treeID), range key: (branchID, nodeID ASC, txnID DESC)\n\t */\n\tHistoryEventsCRUD interface {\n\t\t// InsertIntoHistoryTreeAndNode inserts one or two rows: tree row and node row(at least one of them)\n\t\tInsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *HistoryTreeRow, nodeRow *HistoryNodeRow) error\n\n\t\t// SelectFromHistoryNode read nodes based on a filter\n\t\tSelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]*HistoryNodeRow, []byte, error)\n\n\t\t// DeleteFromHistoryTreeAndNode delete a branch record, and a list of ranges of nodes.\n\t\t// for each range, it will delete all nodes starting from MinNodeID(inclusive)\n\t\tDeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *HistoryTreeFilter, nodeFilters []*HistoryNodeFilter) error\n\n\t\t// SelectAllHistoryTrees will return all tree branches with pagination\n\t\tSelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*HistoryTreeRow, []byte, error)\n\n\t\t// SelectFromHistoryTree read branch records for a tree.\n\t\t// It returns without pagination, because we assume one tree won't have too many branches.\n\t\tSelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]*HistoryTreeRow, error)\n\t}\n\n\t/***\n\t * MessageQueueCRUD is for the message queue storage system\n\t *\n\t * Recommendation: use two tables(queue_message,and queue_metadata) to implement this interface\n\t *\n\t * Significant columns:\n\t * queue_message partition key: (queueType), range key: (messageID)\n\t * queue_metadata partition key: (queueType), range key: N/A, query condition column(version)\n\t */\n\tMessageQueueCRUD interface {\n\t\t// Insert message into queue, return error if failed or already exists\n\t\t// Must return conditionFailed error if row already exists\n\t\tInsertIntoQueue(ctx context.Context, row *QueueMessageRow) error\n\t\t// Get the ID of last message inserted into the queue\n\t\tSelectLastEnqueuedMessageID(ctx context.Context, queueType persistence.QueueType) (int64, error)\n\t\t// Read queue messages starting from the exclusiveBeginMessageID\n\t\tSelectMessagesFrom(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, maxRows int) ([]*QueueMessageRow, error)\n\t\t// Read queue message starting from exclusiveBeginMessageID int64, inclusiveEndMessageID int64\n\t\tSelectMessagesBetween(ctx context.Context, request SelectMessagesBetweenRequest) (*SelectMessagesBetweenResponse, error)\n\t\t// Delete all messages before exclusiveBeginMessageID\n\t\tDeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64) error\n\t\t// Delete all messages in a range between exclusiveBeginMessageID and inclusiveEndMessageID\n\t\tDeleteMessagesInRange(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, inclusiveEndMessageID int64) error\n\t\t// Delete one message\n\t\tDeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) error\n\n\t\t// Insert an empty metadata row, starting from a version\n\t\tInsertQueueMetadata(ctx context.Context, row QueueMetadataRow) error\n\t\t// **Conditionally** update a queue metadata row, if current version is matched(meaning current == row.Version - 1),\n\t\t// then the current version will increase by one when updating the metadata row\n\t\t// Must return conditionFailed error if the condition is not met\n\t\tUpdateQueueMetadataCas(ctx context.Context, row QueueMetadataRow) error\n\t\t// Read a QueueMetadata\n\t\tSelectQueueMetadata(ctx context.Context, queueType persistence.QueueType) (*QueueMetadataRow, error)\n\t\t// GetQueueSize return the queue size\n\t\tGetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error)\n\t}\n\n\t/***\n\t* DomainCRUD is for domain + domain metadata storage system\n\t*\n\t* Recommendation: two tables(domain, domain_metadata) to implement if conditional updates on two tables is supported\n\t*\n\t* Significant columns:\n\t* domain: partition key( a constant value), range key(domainName), local secondary index(domainID)\n\t* domain_metadata: partition key( a constant value), range key(N/A), query condition column(notificationVersion)\n\t*\n\t* Note 1: About Cassandra's implementation: Because of historical reasons, Cassandra uses two table,\n\t* domains and domains_by_name_v2. Therefore, Cassandra implementation lost the atomicity causing some edge cases,\n\t* and the implementation is more complicated than it should be.\n\t*\n\t* Note 2: Cassandra doesn't support conditional updates on multiple tables. Hence the domain_metadata table is implemented\n\t* as a special record as \"domain metadata\". It is an integer number as notification version.\n\t* The main purpose of it is to notify clusters that there is some changes in domains, so domain cache needs to refresh.\n\t* It always increase by one, whenever a domain is updated or inserted.\n\t* Updating this failover metadata with domain insert/update needs to be atomic.\n\t* Because Batch LWTs is only allowed within one table and same partition.\n\t* The Cassandra implementation stores it in the same table as domain in domains_by_name_v2.\n\t*\n\t* Note 3: It's okay to use a constant value for partition key because domain table is serving very small volume of traffic.\n\t */\n\tDomainCRUD interface {\n\t\t// Insert a new record to domain\n\t\t// return types.DomainAlreadyExistsError error if failed or already exists\n\t\t// Must return ConditionFailure error if other condition doesn't match\n\t\tInsertDomain(ctx context.Context, row *DomainRow) error\n\t\t// Update domain data\n\t\t// Must return ConditionFailure error if update condition doesn't match\n\t\tUpdateDomain(ctx context.Context, row *DomainRow) error\n\t\t// Get one domain data, either by domainID or domainName\n\t\tSelectDomain(ctx context.Context, domainID *string, domainName *string) (*DomainRow, error)\n\t\t// Get all domain data\n\t\tSelectAllDomains(ctx context.Context, pageSize int, pageToken []byte) ([]*DomainRow, []byte, error)\n\t\t//  Delete a domain, either by domainID or domainName\n\t\tDeleteDomain(ctx context.Context, domainID *string, domainName *string) error\n\t\t// right now domain metadata is just an integer as notification version\n\t\tSelectDomainMetadata(ctx context.Context) (int64, error)\n\t}\n\n\t/**\n\t* ShardCRUD is for shard storage of workflow execution.\n\t* Recommendation: use one table if database support batch conditional update on multiple tables, otherwise combine with WorkflowCRUD (likeCassandra)\n\t*\n\t* Significant columns:\n\t* domain: partition key(shardID), range key(N/A), local secondary index(domainID), query condition column(rangeID)\n\t*\n\t* Note 1: shard will be required to run conditional update with WorkflowCRUD. So in some nosql database like Cassandra,\n\t* ShardCRUD and WorkflowCRUD must be implemented within the same table. Because Cassandra only allows LightWeight transaction\n\t* executed within a single table.\n\t* Note 2: unlike Cassandra, most NoSQL databases don't return the previous rows when conditional write fails. In this case,\n\t* an extra read query is needed to get the previous row.\n\t */\n\tShardCRUD interface {\n\t\t// InsertShard creates a new shard.\n\t\t// Return error is there is any thing wrong\n\t\t// Return the ShardOperationConditionFailure when doesn't meet the condition\n\t\tInsertShard(ctx context.Context, row *ShardRow) error\n\t\t// SelectShard gets a shard, rangeID is the current rangeID in shard row\n\t\tSelectShard(ctx context.Context, shardID int, currentClusterName string) (rangeID int64, shard *ShardRow, err error)\n\t\t// UpdateRangeID updates the rangeID\n\t\t// Return error is there is any thing wrong\n\t\t// Return the ShardOperationConditionFailure when doesn't meet the condition\n\t\tUpdateRangeID(ctx context.Context, shardID int, rangeID int64, previousRangeID int64) error\n\t\t// UpdateShard updates a shard\n\t\t// Return error is there is any thing wrong\n\t\t// Return the ShardOperationConditionFailure when doesn't meet the condition\n\t\tUpdateShard(ctx context.Context, row *ShardRow, previousRangeID int64) error\n\t}\n\n\t/**\n\t* VisibilityCRUD is for visibility using database.\n\t* Database visibility usually is no longer recommended. AdvancedVisibility(with Kafka+ElasticSearch) is more powerful and scalable.\n\t* Feel free to skip this interface for any NoSQL plugin(use TODO() in the implementation)\n\t*\n\t* Recommendation: use one table with multiple indexes\n\t*\n\t* Significant columns:\n\t* domain: partition key(domainID), range key(workflowID, runID),\n\t*         local secondary index #1(startTime),\n\t*         local secondary index #2(closedTime),\n\t*         local secondary index #3(workflowType, startTime),\n\t*         local secondary index #4(workflowType, closedTime),\n\t*         local secondary index #5(workflowID, startTime),\n\t*         local secondary index #6(workflowID, closedTime),\n\t*         local secondary index #7(closeStatus, closedTime),\n\t*\n\t* NOTE 1: Cassandra implementation of visibility uses three tables: open_executions, closed_executions and closed_executions_v2,\n\t* because Cassandra doesn't support cross-partition indexing.\n\t* Records in open_executions and closed_executions are clustered by start_time. Records in  closed_executions_v2 are by close_time.\n\t* This optimizes the performance, but introduce a lot of complexity.\n\t* In some other databases, this may be be necessary. Please refer to MySQL/Postgres implementation which uses only\n\t* one table with multiple indexes.\n\t*\n\t* NOTE 2: TTL(time to live records) is for auto-deleting expired records in visibility. For databases that don't support TTL,\n\t* please implement DeleteVisibility method. If TTL is supported, then DeleteVisibility can be a noop.\n\t */\n\tVisibilityCRUD interface {\n\t\tInsertVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForInsert) error\n\t\tUpdateVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForUpdate) error\n\t\tSelectVisibility(ctx context.Context, filter *VisibilityFilter) (*SelectVisibilityResponse, error)\n\t\tDeleteVisibility(ctx context.Context, domainID, workflowID, runID string) error\n\t\t// TODO deprecated this in the future in favor of SelectVisibility\n\t\t// Special case: return nil,nil if not found(since we will deprecate it, it's not worth refactor to be consistent)\n\t\tSelectOneClosedWorkflow(ctx context.Context, domainID, workflowID, runID string) (*VisibilityRow, error)\n\t}\n\n\tVisibilityRowForInsert struct {\n\t\tVisibilityRow\n\t\tDomainID string\n\t}\n\n\tVisibilityRowForUpdate struct {\n\t\tVisibilityRow\n\t\tDomainID string\n\t\t// NOTE: this is only for some implementation (e.g. Cassandra) that uses multiple tables,\n\t\t// they needs to delete record from the open execution table. Ignore this field if not need it\n\t\tUpdateOpenToClose bool\n\t\t//  Similar as UpdateOpenToClose\n\t\tUpdateCloseToOpen bool\n\t}\n\n\t// TODO separate in the future when need it\n\tVisibilityRow = persistence.InternalVisibilityWorkflowExecutionInfo\n\n\tSelectVisibilityResponse struct {\n\t\tExecutions    []*VisibilityRow\n\t\tNextPageToken []byte\n\t}\n\n\t// VisibilityFilter contains the column names within executions_visibility table that\n\t// can be used to filter results through a WHERE clause\n\tVisibilityFilter struct {\n\t\tListRequest  persistence.InternalListWorkflowExecutionsRequest\n\t\tFilterType   VisibilityFilterType\n\t\tSortType     VisibilitySortType\n\t\tWorkflowType string\n\t\tWorkflowID   string\n\t\tCloseStatus  int32\n\t}\n\n\tVisibilityFilterType int\n\tVisibilitySortType   int\n\n\t/**\n\t* TaskCRUD is for tasklist and worker tasks storage\n\t* The task here is only referred to workflow/activity worker tasks. `Task` is a overloaded term in Cadence.\n\t* There is another 'task' storage which is for internal purpose only in WorkflowCRUD.\n\t*\n\t* Recommendation: use two tables(tasklist + task) to implement\n\t* tasklist table stores the metadata mainly for\n\t*   * rangeID: ownership management, and taskID management  everytime a matching host claim ownership of a tasklist,\n\t*              it must increase the value succesfully . also used as base section of the taskID. E.g, if rangeID is 1,\n\t*              then allowed taskID ranged will be [100K, 2*100K-1].\n\t*   * ackLevel: max taskID that can be safely deleted.\n\t* Any task record is associated with a tasklist. Any updates on a task should use rangeID of the associated tasklist as condition.\n\t*\n\t* Significant columns:\n\t* tasklist: partition key(domainID, taskListName, taskListType), range key(N/A), query condition column(rangeID)\n\t* task:partition key(domainID, taskListName, taskListType), range key(taskID), query condition column(rangeID)\n\t*\n\t* NOTE 1: Cassandra implementation uses the same table for tasklist and task, because Cassandra only allows\n\t*        batch conditional updates(LightWeight transaction) executed within a single table.\n\t* NOTE 2: TTL(time to live records) is for auto-deleting task and some tasklists records. For databases that don't\n\t*        support TTL, please implement ListTaskList method, and allows TaskListScavenger like MySQL/Postgres.\n\t*        If TTL is supported, then ListTaskList can be a noop.\n\t */\n\tTaskCRUD interface {\n\t\t// SelectTaskList returns a single tasklist row.\n\t\t// Return IsNotFoundError if the row doesn't exist\n\t\tSelectTaskList(ctx context.Context, filter *TaskListFilter) (*TaskListRow, error)\n\t\t// InsertTaskList insert a single tasklist row\n\t\t// Return TaskOperationConditionFailure if the row already exists\n\t\tInsertTaskList(ctx context.Context, row *TaskListRow) error\n\t\t// UpdateTaskList updates a single tasklist row\n\t\t// Return TaskOperationConditionFailure if the condition doesn't meet\n\t\tUpdateTaskList(ctx context.Context, row *TaskListRow, previousRangeID int64) error\n\t\t// UpdateTaskList updates a single tasklist row, and set an TTL on the record\n\t\t// Return TaskOperationConditionFailure if the condition doesn't meet\n\t\t// Ignore TTL if it's not supported, which becomes exactly the same as UpdateTaskList, but ListTaskList must be\n\t\t// implemented for TaskListScavenger\n\t\tUpdateTaskListWithTTL(ctx context.Context, ttlSeconds int64, row *TaskListRow, previousRangeID int64) error\n\t\t// ListTaskList returns all tasklists.\n\t\t// Noop if TTL is already implemented in other methods\n\t\tListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*ListTaskListResult, error)\n\t\t// DeleteTaskList deletes a single tasklist row\n\t\t// Return TaskOperationConditionFailure if the condition doesn't meet\n\t\tDeleteTaskList(ctx context.Context, filter *TaskListFilter, previousRangeID int64) error\n\t\t// InsertTasks inserts a batch of tasks\n\t\t// Return TaskOperationConditionFailure if the condition doesn't meet\n\t\tInsertTasks(ctx context.Context, tasksToInsert []*TaskRowForInsert, tasklistCondition *TaskListRow) error\n\t\t// SelectTasks return tasks that associated to a tasklist\n\t\tSelectTasks(ctx context.Context, filter *TasksFilter) ([]*TaskRow, error)\n\t\t// DeleteTask delete a batch of tasks\n\t\t// Also return the number of rows deleted -- if it's not supported then ignore the batchSize, and return persistence.UnknownNumRowsAffected\n\t\tRangeDeleteTasks(ctx context.Context, filter *TasksFilter) (rowsDeleted int, err error)\n\t\t// GetTasksCount return the number of tasks\n\t\tGetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error)\n\t}\n\n\t/**\n\t* WorkflowCRUD is for core data models of workflow execution.\n\t*\n\t* Recommendation: If possible, use 8 tables(current_workflow, workflow_execution, transfer_task, replication_task, cross_cluster_task, timer_task, buffered_event_list, replication_dlq_task) to implement\n\t* current_workflow is to track the currentRunID of a workflowID for ensuring the ID-Uniqueness of Cadence workflows.\n\t* \t\tEach record is for one workflowID\n\t* workflow_execution is to store the core data of workflow execution.\n\t*\t\tEach record is for one runID(workflow execution run).\n\t* Different from TaskCRUD, transfer_task, replication_task, cross_cluster_task, timer_task are all internal background tasks within Cadence server.\n\t* transfer_task is to store the background tasks that need to be processed by historyEngine, right after the transaction.\n\t*\t\tThere are lots of usage in historyEngine, like creating activity/childWF/etc task, and updating search attributes, etc.\n\t* replication_task is to store also background tasks that need to be processed right after the transaction,\n\t*\t\tbut only for CrossDC(XDC) replication feature. Each record is a replication task generated from a source cluster.\n\t*\t\tReplication task stores a reference to a batch of history events(see historyCRUD).\n\t* timer_task is to store the durable timers that will fire in the future. Therefore this table should be indexed by the firingTime.\n\t*\t\tThe durable timers are not only for workflow timers, but also for all kinds of timeouts, and workflow deletion, etc.\n\t* cross_cluster_task is to store also background tasks that need to be processed right after the transaction, and only for\n\t*\t\tbut only for cross cluster feature. Each record is a cross cluster task generated for a target cluster.\n\t*\t\tCrossCluster task stores information similar to TransferTask.\n\t* buffered_event_list is to store the buffered event of a workflow execution\n\t* The above 7 tables will be required to execute transaction write with the condition of shard record from ShardCRUD.\n\t* replication_dlq_task is DeadLetterQueue when target cluster pulling and applying replication task. Each record represents\n\t*\t\ta task for a target cluster.\n\t*\n\t* Significant columns:\n\t* current_workflow: partition key(shardID), range key(domainID, workflowID), query condition column(currentRunID, lastWriteVersion, state)\n\t* workflow_execution: partition key(shardID), range key(domainID, workflowID, runID), query condition column(nextEventID)\n\t* transfer_task: partition key(shardID), range key(taskID)\n\t* replication_task: partition key(shardID), range key(taskID)\n\t* cross_cluster_task: partition key(shardID), range key(clusterName, taskID)\n\t* timer_task: partition key(shardID), range key(visibilityTimestamp)\n\t* buffered_event_list: partition key(shardID), range key(domainID, workflowID, runID)\n\t* replication_dlq_task: partition key(shardID), range key(clusterName, taskID)\n\t*\n\t* NOTE: Cassandra limits lightweight transaction to execute within one table. So the 6 tables + shard table are implemented\n\t*   \tvia a single table `execution` in Cassandra, using `rowType` to differentiate the 7 tables, and using `permanentRunID`\n\t*\t\tto differentiate current_workflow and workflow_execution\n\t* NOTE: Cassandra implementation uses 6 maps to store activityInfo, timerInfo, childWorkflowInfo, requestCancels,\n\t*\t\tsignalInfo and signalRequestedInfo.Those should be fine to be stored in the same record as its workflow_execution.\n\t*\t\tHowever, signalInfo stores the in progress signal data. It may be too big for a single record. For example, DynamoDB\n\t*\t\trequires 400KB of a record. In that case, it may be better to have a separate table for signalInfo.\n\t* NOTE: Cassandra implementation of workflow_execution uses maps without \"frozen\". This has the advantage of deleting activity/timer/childWF/etc\n\t*\t\tby keys. The equivalent of this may require a read before overwriting the existing. Eg. [ \"act1\": <some data>, \"act2\": <some data>]\n\t*\t\tWhen deleting \"act1\", Cassandra implementation can delete without read. If storing in the same record of workflwo_execution,\n\t*\t\tit will require to read the whole activityInfo map for deleting.\n\t* NOTE: Optional optimization: taskID that are writing into internal tasks(transfer/replication/crossCluster) are immutable and always increasing.\n\t*\t\tSo it is possible to write the tasks in a single record, indexing by the lowest or highest taskID.\n\t*\t\tThis approach can't be used by timerTasks as timers are ordered by visibilityTimestamp.\n\t*\t\tThis is useful for DynamoDB because a transaction cannot contain more than 25 unique items.\n\t*\n\t */\n\tWorkflowCRUD interface {\n\t\t// InsertWorkflowExecutionWithTasks is for creating a new workflow execution record. Within a transaction, it also:\n\t\t// 1. Create or update the record of current_workflow with the same workflowID, based on CurrentWorkflowExecutionWriteMode,\n\t\t//\t\tand also check if the condition is met.\n\t\t// 2. Create the workflow_execution record, including basic info and 6 maps(activityInfoMap, timerInfoMap,\n\t\t//\t\tchildWorkflowInfoMap, signalInfoMap and signalRequestedIDs)\n\t\t// 3. Create history tasks\n\t\t// 4. Create workflow requests for requests deduplication\n\t\t// 5. Check if the condition of shard rangeID is met\n\t\t// The API returns error if there is any. If any of the condition is not met, returns WorkflowOperationConditionFailure\n\t\tInsertWorkflowExecutionWithTasks(\n\t\t\tctx context.Context,\n\t\t\trequests *WorkflowRequestsWriteRequest,\n\t\t\tcurrentWorkflowRequest *CurrentWorkflowWriteRequest,\n\t\t\texecution *WorkflowExecutionRequest,\n\t\t\ttasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask,\n\t\t\tactiveClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow,\n\t\t\tshardCondition *ShardCondition,\n\t\t) error\n\n\t\t// UpdateWorkflowExecutionWithTasks is for updating a new workflow execution record.\n\t\t// Within a transaction, it also:\n\t\t// 1. If currentWorkflowRequest is not nil, Update the record of current_workflow with the same workflowID, based on CurrentWorkflowExecutionWriteMode,\n\t\t//\t\tand also check if the condition is met.\n\t\t// 2. Update mutatedExecution as workflow_execution record, including basic info and 6 maps(activityInfoMap, timerInfoMap,\n\t\t//\t\tchildWorkflowInfoMap, signalInfoMap and signalRequestedIDs)\n\t\t// 3. if insertedExecution is not nil, then also insert a new workflow_execution record including basic info and add to 6 maps:\n\t\t// \t\t(activityInfoMap, timerInfoMap, childWorkflowInfoMap, signalInfoMap and signalRequestedIDs)\n\t\t//      Also insert activeClusterSelectionPolicyRow if it is not nil corresponding to the new execution\n\t\t// 4. if resetExecution is not nil, then also update the workflow_execution record including basic info and reset/override 6 maps(activityInfoMap, timerInfoMap,\n\t\t//\t\tchildWorkflowInfoMap, signalInfoMap and signalRequestedIDs\n\t\t// 5. Create history tasks\n\t\t// 6. Create workflow requests for requests deduplication\n\t\t// 7. Check if the condition of shard rangeID is met\n\t\t// The API returns error if there is any. If any of the condition is not met, returns WorkflowOperationConditionFailure\n\t\tUpdateWorkflowExecutionWithTasks(\n\t\t\tctx context.Context,\n\t\t\trequests *WorkflowRequestsWriteRequest,\n\t\t\tcurrentWorkflowRequest *CurrentWorkflowWriteRequest,\n\t\t\tmutatedExecution *WorkflowExecutionRequest,\n\t\t\tinsertedExecution *WorkflowExecutionRequest,\n\t\t\tactiveClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow,\n\t\t\tresetExecution *WorkflowExecutionRequest,\n\t\t\ttasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask,\n\t\t\tshardCondition *ShardCondition,\n\t\t) error\n\n\t\t// current_workflow table\n\t\t// Return the current_workflow row\n\t\tSelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*CurrentWorkflowRow, error)\n\t\t// Paging through all current_workflow rows in a shard\n\t\tSelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error)\n\t\t// Delete the current_workflow row, if currentRunIDCondition is met\n\t\tDeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error\n\n\t\t// workflow_execution table\n\t\t// Return the workflow execution row\n\t\tSelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*WorkflowExecution, error)\n\t\t// Paging through all  workflow execution rows in a shard\n\t\tSelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error)\n\t\t// Return whether or not an execution is existing.\n\t\tIsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error)\n\t\t// Delete the workflow execution row\n\t\tDeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error\n\n\t\t// transfer_task table\n\t\t// within a shard, paging through transfer tasks order by taskID(ASC), filtered by minTaskID(inclusive) and maxTaskID(exclusive)\n\t\tSelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error)\n\t\t// delete a single transfer task\n\t\tDeleteTransferTask(ctx context.Context, shardID int, taskID int64) error\n\t\t// delete a range of transfer tasks\n\t\tRangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error\n\n\t\t// timer_task table\n\t\t// within a shard, paging through timer tasks order by taskID(ASC), filtered by visibilityTimestamp\n\t\tSelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*HistoryMigrationTask, []byte, error)\n\t\t// delete a single timer task\n\t\tDeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error\n\t\t// delete a range of timer tasks\n\t\tRangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error\n\n\t\t// replication_task table\n\t\t// within a shard, paging through replication tasks order by taskID(ASC), filtered by minTaskID(inclusive) and maxTaskID(exclusive)\n\t\tSelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error)\n\t\t// delete a single replication task\n\t\tDeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error\n\t\t// delete a range of replication tasks\n\t\tRangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error\n\t\t// insert replication task with shard condition check\n\t\tInsertReplicationTask(ctx context.Context, tasks []*HistoryMigrationTask, condition ShardCondition) error\n\n\t\t// cross_cluster_task table\n\t\t// delete a single transfer task\n\t\tDeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error\n\n\t\t// replication_dlq_task\n\t\t// insert a new replication task to DLQ\n\t\tInsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *HistoryMigrationTask) error\n\t\t// within a shard, for a sourceCluster, paging through replication tasks order by taskID(ASC), filtered by minTaskID(inclusive) and maxTaskID(exclusive)\n\t\tSelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error)\n\t\t// return the DLQ size\n\t\tSelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error)\n\t\t// delete a single replication DLQ task\n\t\tDeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error\n\t\t// delete a range of replication DLQ tasks\n\t\tRangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error\n\n\t\t// select the active cluster selection policy\n\t\tSelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*ActiveClusterSelectionPolicyRow, error)\n\t\t// delete the active cluster selection policy row\n\t\tDeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, workflowID, runID string) error\n\t}\n\n\t/***\n\t* ConfigStoreCRUD is for storing dynamic configuration parameters\n\t*\n\t* Recommendation: one table\n\t*\n\t* Significant columns:\n\t* domain: partition key(row_type), range key(version)\n\t */\n\tConfigStoreCRUD interface {\n\t\t// InsertConfig insert a config entry with version. Return nosqlplugin.NewConditionFailure if the same version of the row_type is existing\n\t\tInsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error\n\t\t// SelectLatestConfig returns the config entry of the row_type with the largest(latest) version value\n\t\tSelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error)\n\t}\n\n\t/***\n\t* DomainAuditLogCRUD is for domain audit log storage system\n\t*\n\t* Recommendation: use one table\n\t*\n\t* Significant columns:\n\t* domain_audit_log: partition key(domainID, operationType), range key(createdTime DESC, eventID ASC)\n\t*\n\t* Note: This table is used for audit trail of domain changes, storing the before/after state\n\t* of domains along with metadata about who made the change and when.\n\t */\n\tDomainAuditLogCRUD interface {\n\t\t// InsertDomainAuditLog inserts a new audit log entry for a domain operation\n\t\t// Return error if there is any failure\n\t\tInsertDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) error\n\n\t\t// SelectDomainAuditLogs returns audit log entries for a domain and operation type\n\t\t// Returns paginated results ordered by created_time DESC, event_id ASC\n\t\tSelectDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, []byte, error)\n\t}\n)\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package nosqlplugin -source interfaces.go -destination interfaces_mock.go -self_package github.com/uber/cadence/common/persistence/nosql/nosqlplugin\n//\n\n// Package nosqlplugin is a generated GoMock package.\npackage nosqlplugin\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tconfig \"github.com/uber/cadence/common/config\"\n\tlog \"github.com/uber/cadence/common/log\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockPlugin is a mock of Plugin interface.\ntype MockPlugin struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPluginMockRecorder\n\tisgomock struct{}\n}\n\n// MockPluginMockRecorder is the mock recorder for MockPlugin.\ntype MockPluginMockRecorder struct {\n\tmock *MockPlugin\n}\n\n// NewMockPlugin creates a new mock instance.\nfunc NewMockPlugin(ctrl *gomock.Controller) *MockPlugin {\n\tmock := &MockPlugin{ctrl: ctrl}\n\tmock.recorder = &MockPluginMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPlugin) EXPECT() *MockPluginMockRecorder {\n\treturn m.recorder\n}\n\n// CreateAdminDB mocks base method.\nfunc (m *MockPlugin) CreateAdminDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (AdminDB, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateAdminDB\", cfg, logger, dc)\n\tret0, _ := ret[0].(AdminDB)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateAdminDB indicates an expected call of CreateAdminDB.\nfunc (mr *MockPluginMockRecorder) CreateAdminDB(cfg, logger, dc any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateAdminDB\", reflect.TypeOf((*MockPlugin)(nil).CreateAdminDB), cfg, logger, dc)\n}\n\n// CreateDB mocks base method.\nfunc (m *MockPlugin) CreateDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (DB, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDB\", cfg, logger, dc)\n\tret0, _ := ret[0].(DB)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDB indicates an expected call of CreateDB.\nfunc (mr *MockPluginMockRecorder) CreateDB(cfg, logger, dc any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDB\", reflect.TypeOf((*MockPlugin)(nil).CreateDB), cfg, logger, dc)\n}\n\n// MockAdminDB is a mock of AdminDB interface.\ntype MockAdminDB struct {\n\tctrl     *gomock.Controller\n\trecorder *MockAdminDBMockRecorder\n\tisgomock struct{}\n}\n\n// MockAdminDBMockRecorder is the mock recorder for MockAdminDB.\ntype MockAdminDBMockRecorder struct {\n\tmock *MockAdminDB\n}\n\n// NewMockAdminDB creates a new mock instance.\nfunc NewMockAdminDB(ctrl *gomock.Controller) *MockAdminDB {\n\tmock := &MockAdminDB{ctrl: ctrl}\n\tmock.recorder = &MockAdminDBMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockAdminDB) EXPECT() *MockAdminDBMockRecorder {\n\treturn m.recorder\n}\n\n// SetupTestDatabase mocks base method.\nfunc (m *MockAdminDB) SetupTestDatabase(schemaBaseDir string, replicas int) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SetupTestDatabase\", schemaBaseDir, replicas)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SetupTestDatabase indicates an expected call of SetupTestDatabase.\nfunc (mr *MockAdminDBMockRecorder) SetupTestDatabase(schemaBaseDir, replicas any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetupTestDatabase\", reflect.TypeOf((*MockAdminDB)(nil).SetupTestDatabase), schemaBaseDir, replicas)\n}\n\n// TeardownTestDatabase mocks base method.\nfunc (m *MockAdminDB) TeardownTestDatabase() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TeardownTestDatabase\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TeardownTestDatabase indicates an expected call of TeardownTestDatabase.\nfunc (mr *MockAdminDBMockRecorder) TeardownTestDatabase() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TeardownTestDatabase\", reflect.TypeOf((*MockAdminDB)(nil).TeardownTestDatabase))\n}\n\n// MockDB is a mock of DB interface.\ntype MockDB struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDBMockRecorder\n\tisgomock struct{}\n}\n\n// MockDBMockRecorder is the mock recorder for MockDB.\ntype MockDBMockRecorder struct {\n\tmock *MockDB\n}\n\n// NewMockDB creates a new mock instance.\nfunc NewMockDB(ctrl *gomock.Controller) *MockDB {\n\tmock := &MockDB{ctrl: ctrl}\n\tmock.recorder = &MockDBMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDB) EXPECT() *MockDBMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockDB) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDBMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDB)(nil).Close))\n}\n\n// DeleteActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockDB) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActiveClusterSelectionPolicy\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActiveClusterSelectionPolicy indicates an expected call of DeleteActiveClusterSelectionPolicy.\nfunc (mr *MockDBMockRecorder) DeleteActiveClusterSelectionPolicy(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActiveClusterSelectionPolicy\", reflect.TypeOf((*MockDB)(nil).DeleteActiveClusterSelectionPolicy), ctx, shardID, domainID, workflowID, runID)\n}\n\n// DeleteCrossClusterTask mocks base method.\nfunc (m *MockDB) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCrossClusterTask\", ctx, shardID, targetCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCrossClusterTask indicates an expected call of DeleteCrossClusterTask.\nfunc (mr *MockDBMockRecorder) DeleteCrossClusterTask(ctx, shardID, targetCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCrossClusterTask\", reflect.TypeOf((*MockDB)(nil).DeleteCrossClusterTask), ctx, shardID, targetCluster, taskID)\n}\n\n// DeleteCurrentWorkflow mocks base method.\nfunc (m *MockDB) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflow\", ctx, shardID, domainID, workflowID, currentRunIDCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflow indicates an expected call of DeleteCurrentWorkflow.\nfunc (mr *MockDBMockRecorder) DeleteCurrentWorkflow(ctx, shardID, domainID, workflowID, currentRunIDCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflow\", reflect.TypeOf((*MockDB)(nil).DeleteCurrentWorkflow), ctx, shardID, domainID, workflowID, currentRunIDCondition)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockDB) DeleteDomain(ctx context.Context, domainID, domainName *string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockDBMockRecorder) DeleteDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockDB)(nil).DeleteDomain), ctx, domainID, domainName)\n}\n\n// DeleteFromHistoryTreeAndNode mocks base method.\nfunc (m *MockDB) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *HistoryTreeFilter, nodeFilters []*HistoryNodeFilter) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTreeAndNode\", ctx, treeFilter, nodeFilters)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteFromHistoryTreeAndNode indicates an expected call of DeleteFromHistoryTreeAndNode.\nfunc (mr *MockDBMockRecorder) DeleteFromHistoryTreeAndNode(ctx, treeFilter, nodeFilters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTreeAndNode\", reflect.TypeOf((*MockDB)(nil).DeleteFromHistoryTreeAndNode), ctx, treeFilter, nodeFilters)\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MockDB) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MockDBMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MockDB)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MockDB) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, exclusiveBeginMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MockDBMockRecorder) DeleteMessagesBefore(ctx, queueType, exclusiveBeginMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MockDB)(nil).DeleteMessagesBefore), ctx, queueType, exclusiveBeginMessageID)\n}\n\n// DeleteMessagesInRange mocks base method.\nfunc (m *MockDB) DeleteMessagesInRange(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesInRange\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesInRange indicates an expected call of DeleteMessagesInRange.\nfunc (mr *MockDBMockRecorder) DeleteMessagesInRange(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesInRange\", reflect.TypeOf((*MockDB)(nil).DeleteMessagesInRange), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// DeleteReplicationDLQTask mocks base method.\nfunc (m *MockDB) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationDLQTask\", ctx, shardID, sourceCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationDLQTask indicates an expected call of DeleteReplicationDLQTask.\nfunc (mr *MockDBMockRecorder) DeleteReplicationDLQTask(ctx, shardID, sourceCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationDLQTask\", reflect.TypeOf((*MockDB)(nil).DeleteReplicationDLQTask), ctx, shardID, sourceCluster, taskID)\n}\n\n// DeleteReplicationTask mocks base method.\nfunc (m *MockDB) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationTask indicates an expected call of DeleteReplicationTask.\nfunc (mr *MockDBMockRecorder) DeleteReplicationTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationTask\", reflect.TypeOf((*MockDB)(nil).DeleteReplicationTask), ctx, shardID, taskID)\n}\n\n// DeleteTaskList mocks base method.\nfunc (m *MockDB) DeleteTaskList(ctx context.Context, filter *TaskListFilter, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTaskList\", ctx, filter, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTaskList indicates an expected call of DeleteTaskList.\nfunc (mr *MockDBMockRecorder) DeleteTaskList(ctx, filter, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTaskList\", reflect.TypeOf((*MockDB)(nil).DeleteTaskList), ctx, filter, previousRangeID)\n}\n\n// DeleteTimerTask mocks base method.\nfunc (m *MockDB) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTimerTask\", ctx, shardID, taskID, visibilityTimestamp)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTimerTask indicates an expected call of DeleteTimerTask.\nfunc (mr *MockDBMockRecorder) DeleteTimerTask(ctx, shardID, taskID, visibilityTimestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTimerTask\", reflect.TypeOf((*MockDB)(nil).DeleteTimerTask), ctx, shardID, taskID, visibilityTimestamp)\n}\n\n// DeleteTransferTask mocks base method.\nfunc (m *MockDB) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTransferTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTransferTask indicates an expected call of DeleteTransferTask.\nfunc (mr *MockDBMockRecorder) DeleteTransferTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTransferTask\", reflect.TypeOf((*MockDB)(nil).DeleteTransferTask), ctx, shardID, taskID)\n}\n\n// DeleteVisibility mocks base method.\nfunc (m *MockDB) DeleteVisibility(ctx context.Context, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteVisibility\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteVisibility indicates an expected call of DeleteVisibility.\nfunc (mr *MockDBMockRecorder) DeleteVisibility(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteVisibility\", reflect.TypeOf((*MockDB)(nil).DeleteVisibility), ctx, domainID, workflowID, runID)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockDB) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockDBMockRecorder) DeleteWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockDB)(nil).DeleteWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MockDB) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MockDBMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MockDB)(nil).GetQueueSize), ctx, queueType)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MockDB) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MockDBMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MockDB)(nil).GetTasksCount), ctx, filter)\n}\n\n// InsertConfig mocks base method.\nfunc (m *MockDB) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MockDBMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MockDB)(nil).InsertConfig), ctx, row)\n}\n\n// InsertDomain mocks base method.\nfunc (m *MockDB) InsertDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomain indicates an expected call of InsertDomain.\nfunc (mr *MockDBMockRecorder) InsertDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomain\", reflect.TypeOf((*MockDB)(nil).InsertDomain), ctx, row)\n}\n\n// InsertDomainAuditLog mocks base method.\nfunc (m *MockDB) InsertDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomainAuditLog indicates an expected call of InsertDomainAuditLog.\nfunc (mr *MockDBMockRecorder) InsertDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomainAuditLog\", reflect.TypeOf((*MockDB)(nil).InsertDomainAuditLog), ctx, row)\n}\n\n// InsertIntoHistoryTreeAndNode mocks base method.\nfunc (m *MockDB) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *HistoryTreeRow, nodeRow *HistoryNodeRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTreeAndNode\", ctx, treeRow, nodeRow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoHistoryTreeAndNode indicates an expected call of InsertIntoHistoryTreeAndNode.\nfunc (mr *MockDBMockRecorder) InsertIntoHistoryTreeAndNode(ctx, treeRow, nodeRow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTreeAndNode\", reflect.TypeOf((*MockDB)(nil).InsertIntoHistoryTreeAndNode), ctx, treeRow, nodeRow)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MockDB) InsertIntoQueue(ctx context.Context, row *QueueMessageRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MockDBMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MockDB)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertQueueMetadata mocks base method.\nfunc (m *MockDB) InsertQueueMetadata(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertQueueMetadata\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertQueueMetadata indicates an expected call of InsertQueueMetadata.\nfunc (mr *MockDBMockRecorder) InsertQueueMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertQueueMetadata\", reflect.TypeOf((*MockDB)(nil).InsertQueueMetadata), ctx, row)\n}\n\n// InsertReplicationDLQTask mocks base method.\nfunc (m *MockDB) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *HistoryMigrationTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationDLQTask\", ctx, shardID, sourceCluster, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationDLQTask indicates an expected call of InsertReplicationDLQTask.\nfunc (mr *MockDBMockRecorder) InsertReplicationDLQTask(ctx, shardID, sourceCluster, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationDLQTask\", reflect.TypeOf((*MockDB)(nil).InsertReplicationDLQTask), ctx, shardID, sourceCluster, task)\n}\n\n// InsertReplicationTask mocks base method.\nfunc (m *MockDB) InsertReplicationTask(ctx context.Context, tasks []*HistoryMigrationTask, condition ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationTask\", ctx, tasks, condition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationTask indicates an expected call of InsertReplicationTask.\nfunc (mr *MockDBMockRecorder) InsertReplicationTask(ctx, tasks, condition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationTask\", reflect.TypeOf((*MockDB)(nil).InsertReplicationTask), ctx, tasks, condition)\n}\n\n// InsertShard mocks base method.\nfunc (m *MockDB) InsertShard(ctx context.Context, row *ShardRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertShard\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertShard indicates an expected call of InsertShard.\nfunc (mr *MockDBMockRecorder) InsertShard(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertShard\", reflect.TypeOf((*MockDB)(nil).InsertShard), ctx, row)\n}\n\n// InsertTaskList mocks base method.\nfunc (m *MockDB) InsertTaskList(ctx context.Context, row *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTaskList\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTaskList indicates an expected call of InsertTaskList.\nfunc (mr *MockDBMockRecorder) InsertTaskList(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTaskList\", reflect.TypeOf((*MockDB)(nil).InsertTaskList), ctx, row)\n}\n\n// InsertTasks mocks base method.\nfunc (m *MockDB) InsertTasks(ctx context.Context, tasksToInsert []*TaskRowForInsert, tasklistCondition *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTasks\", ctx, tasksToInsert, tasklistCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTasks indicates an expected call of InsertTasks.\nfunc (mr *MockDBMockRecorder) InsertTasks(ctx, tasksToInsert, tasklistCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTasks\", reflect.TypeOf((*MockDB)(nil).InsertTasks), ctx, tasksToInsert, tasklistCondition)\n}\n\n// InsertVisibility mocks base method.\nfunc (m *MockDB) InsertVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForInsert) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertVisibility indicates an expected call of InsertVisibility.\nfunc (mr *MockDBMockRecorder) InsertVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertVisibility\", reflect.TypeOf((*MockDB)(nil).InsertVisibility), ctx, ttlSeconds, row)\n}\n\n// InsertWorkflowExecutionWithTasks mocks base method.\nfunc (m *MockDB) InsertWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, execution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertWorkflowExecutionWithTasks indicates an expected call of InsertWorkflowExecutionWithTasks.\nfunc (mr *MockDBMockRecorder) InsertWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertWorkflowExecutionWithTasks\", reflect.TypeOf((*MockDB)(nil).InsertWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n}\n\n// IsDBUnavailableError mocks base method.\nfunc (m *MockDB) IsDBUnavailableError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDBUnavailableError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDBUnavailableError indicates an expected call of IsDBUnavailableError.\nfunc (mr *MockDBMockRecorder) IsDBUnavailableError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDBUnavailableError\", reflect.TypeOf((*MockDB)(nil).IsDBUnavailableError), arg0)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockDB) IsNotFoundError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockDBMockRecorder) IsNotFoundError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockDB)(nil).IsNotFoundError), arg0)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockDB) IsThrottlingError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockDBMockRecorder) IsThrottlingError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockDB)(nil).IsThrottlingError), arg0)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockDB) IsTimeoutError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockDBMockRecorder) IsTimeoutError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockDB)(nil).IsTimeoutError), arg0)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MockDB) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MockDBMockRecorder) IsWorkflowExecutionExists(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MockDB)(nil).IsWorkflowExecutionExists), ctx, shardID, domainID, workflowID, runID)\n}\n\n// ListTaskList mocks base method.\nfunc (m *MockDB) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*ListTaskListResult, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskList\", ctx, pageSize, nextPageToken)\n\tret0, _ := ret[0].(*ListTaskListResult)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskList indicates an expected call of ListTaskList.\nfunc (mr *MockDBMockRecorder) ListTaskList(ctx, pageSize, nextPageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskList\", reflect.TypeOf((*MockDB)(nil).ListTaskList), ctx, pageSize, nextPageToken)\n}\n\n// PluginName mocks base method.\nfunc (m *MockDB) PluginName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PluginName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PluginName indicates an expected call of PluginName.\nfunc (mr *MockDBMockRecorder) PluginName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PluginName\", reflect.TypeOf((*MockDB)(nil).PluginName))\n}\n\n// RangeDeleteReplicationDLQTasks mocks base method.\nfunc (m *MockDB) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationDLQTasks\", ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationDLQTasks indicates an expected call of RangeDeleteReplicationDLQTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteReplicationDLQTasks(ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationDLQTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteReplicationDLQTasks), ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// RangeDeleteReplicationTasks mocks base method.\nfunc (m *MockDB) RangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationTasks\", ctx, shardID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationTasks indicates an expected call of RangeDeleteReplicationTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteReplicationTasks(ctx, shardID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteReplicationTasks), ctx, shardID, exclusiveEndTaskID)\n}\n\n// RangeDeleteTasks mocks base method.\nfunc (m *MockDB) RangeDeleteTasks(ctx context.Context, filter *TasksFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTasks\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteTasks indicates an expected call of RangeDeleteTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteTasks), ctx, filter)\n}\n\n// RangeDeleteTimerTasks mocks base method.\nfunc (m *MockDB) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTimerTasks\", ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTimerTasks indicates an expected call of RangeDeleteTimerTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteTimerTasks(ctx, shardID, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTimerTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteTimerTasks), ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// RangeDeleteTransferTasks mocks base method.\nfunc (m *MockDB) RangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTransferTasks\", ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTransferTasks indicates an expected call of RangeDeleteTransferTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteTransferTasks(ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTransferTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteTransferTasks), ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// SelectActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockDB) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*ActiveClusterSelectionPolicyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectActiveClusterSelectionPolicy\", ctx, shardID, domainID, wfID, rID)\n\tret0, _ := ret[0].(*ActiveClusterSelectionPolicyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectActiveClusterSelectionPolicy indicates an expected call of SelectActiveClusterSelectionPolicy.\nfunc (mr *MockDBMockRecorder) SelectActiveClusterSelectionPolicy(ctx, shardID, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectActiveClusterSelectionPolicy\", reflect.TypeOf((*MockDB)(nil).SelectActiveClusterSelectionPolicy), ctx, shardID, domainID, wfID, rID)\n}\n\n// SelectAllCurrentWorkflows mocks base method.\nfunc (m *MockDB) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllCurrentWorkflows\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.CurrentWorkflowExecution)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllCurrentWorkflows indicates an expected call of SelectAllCurrentWorkflows.\nfunc (mr *MockDBMockRecorder) SelectAllCurrentWorkflows(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllCurrentWorkflows\", reflect.TypeOf((*MockDB)(nil).SelectAllCurrentWorkflows), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectAllDomains mocks base method.\nfunc (m *MockDB) SelectAllDomains(ctx context.Context, pageSize int, pageToken []byte) ([]*DomainRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllDomains\", ctx, pageSize, pageToken)\n\tret0, _ := ret[0].([]*DomainRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllDomains indicates an expected call of SelectAllDomains.\nfunc (mr *MockDBMockRecorder) SelectAllDomains(ctx, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllDomains\", reflect.TypeOf((*MockDB)(nil).SelectAllDomains), ctx, pageSize, pageToken)\n}\n\n// SelectAllHistoryTrees mocks base method.\nfunc (m *MockDB) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*HistoryTreeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllHistoryTrees\", ctx, nextPageToken, pageSize)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllHistoryTrees indicates an expected call of SelectAllHistoryTrees.\nfunc (mr *MockDBMockRecorder) SelectAllHistoryTrees(ctx, nextPageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllHistoryTrees\", reflect.TypeOf((*MockDB)(nil).SelectAllHistoryTrees), ctx, nextPageToken, pageSize)\n}\n\n// SelectAllWorkflowExecutions mocks base method.\nfunc (m *MockDB) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllWorkflowExecutions\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.InternalListConcreteExecutionsEntity)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllWorkflowExecutions indicates an expected call of SelectAllWorkflowExecutions.\nfunc (mr *MockDBMockRecorder) SelectAllWorkflowExecutions(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllWorkflowExecutions\", reflect.TypeOf((*MockDB)(nil).SelectAllWorkflowExecutions), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectCurrentWorkflow mocks base method.\nfunc (m *MockDB) SelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*CurrentWorkflowRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectCurrentWorkflow\", ctx, shardID, domainID, workflowID)\n\tret0, _ := ret[0].(*CurrentWorkflowRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectCurrentWorkflow indicates an expected call of SelectCurrentWorkflow.\nfunc (mr *MockDBMockRecorder) SelectCurrentWorkflow(ctx, shardID, domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectCurrentWorkflow\", reflect.TypeOf((*MockDB)(nil).SelectCurrentWorkflow), ctx, shardID, domainID, workflowID)\n}\n\n// SelectDomain mocks base method.\nfunc (m *MockDB) SelectDomain(ctx context.Context, domainID, domainName *string) (*DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(*DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomain indicates an expected call of SelectDomain.\nfunc (mr *MockDBMockRecorder) SelectDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomain\", reflect.TypeOf((*MockDB)(nil).SelectDomain), ctx, domainID, domainName)\n}\n\n// SelectDomainAuditLogs mocks base method.\nfunc (m *MockDB) SelectDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectDomainAuditLogs indicates an expected call of SelectDomainAuditLogs.\nfunc (mr *MockDBMockRecorder) SelectDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainAuditLogs\", reflect.TypeOf((*MockDB)(nil).SelectDomainAuditLogs), ctx, filter)\n}\n\n// SelectDomainMetadata mocks base method.\nfunc (m *MockDB) SelectDomainMetadata(ctx context.Context) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainMetadata\", ctx)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomainMetadata indicates an expected call of SelectDomainMetadata.\nfunc (mr *MockDBMockRecorder) SelectDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainMetadata\", reflect.TypeOf((*MockDB)(nil).SelectDomainMetadata), ctx)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MockDB) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]*HistoryNodeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryNodeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MockDBMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MockDB)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MockDB) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]*HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MockDBMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MockDB)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// SelectLastEnqueuedMessageID mocks base method.\nfunc (m *MockDB) SelectLastEnqueuedMessageID(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLastEnqueuedMessageID\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLastEnqueuedMessageID indicates an expected call of SelectLastEnqueuedMessageID.\nfunc (mr *MockDBMockRecorder) SelectLastEnqueuedMessageID(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLastEnqueuedMessageID\", reflect.TypeOf((*MockDB)(nil).SelectLastEnqueuedMessageID), ctx, queueType)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MockDB) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MockDBMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MockDB)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// SelectMessagesBetween mocks base method.\nfunc (m *MockDB) SelectMessagesBetween(ctx context.Context, request SelectMessagesBetweenRequest) (*SelectMessagesBetweenResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesBetween\", ctx, request)\n\tret0, _ := ret[0].(*SelectMessagesBetweenResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesBetween indicates an expected call of SelectMessagesBetween.\nfunc (mr *MockDBMockRecorder) SelectMessagesBetween(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesBetween\", reflect.TypeOf((*MockDB)(nil).SelectMessagesBetween), ctx, request)\n}\n\n// SelectMessagesFrom mocks base method.\nfunc (m *MockDB) SelectMessagesFrom(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, maxRows int) ([]*QueueMessageRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesFrom\", ctx, queueType, exclusiveBeginMessageID, maxRows)\n\tret0, _ := ret[0].([]*QueueMessageRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesFrom indicates an expected call of SelectMessagesFrom.\nfunc (mr *MockDBMockRecorder) SelectMessagesFrom(ctx, queueType, exclusiveBeginMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesFrom\", reflect.TypeOf((*MockDB)(nil).SelectMessagesFrom), ctx, queueType, exclusiveBeginMessageID, maxRows)\n}\n\n// SelectOneClosedWorkflow mocks base method.\nfunc (m *MockDB) SelectOneClosedWorkflow(ctx context.Context, domainID, workflowID, runID string) (*VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectOneClosedWorkflow\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectOneClosedWorkflow indicates an expected call of SelectOneClosedWorkflow.\nfunc (mr *MockDBMockRecorder) SelectOneClosedWorkflow(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectOneClosedWorkflow\", reflect.TypeOf((*MockDB)(nil).SelectOneClosedWorkflow), ctx, domainID, workflowID, runID)\n}\n\n// SelectQueueMetadata mocks base method.\nfunc (m *MockDB) SelectQueueMetadata(ctx context.Context, queueType persistence.QueueType) (*QueueMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectQueueMetadata\", ctx, queueType)\n\tret0, _ := ret[0].(*QueueMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectQueueMetadata indicates an expected call of SelectQueueMetadata.\nfunc (mr *MockDBMockRecorder) SelectQueueMetadata(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectQueueMetadata\", reflect.TypeOf((*MockDB)(nil).SelectQueueMetadata), ctx, queueType)\n}\n\n// SelectReplicationDLQTasksCount mocks base method.\nfunc (m *MockDB) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksCount\", ctx, shardID, sourceCluster)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectReplicationDLQTasksCount indicates an expected call of SelectReplicationDLQTasksCount.\nfunc (mr *MockDBMockRecorder) SelectReplicationDLQTasksCount(ctx, shardID, sourceCluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksCount\", reflect.TypeOf((*MockDB)(nil).SelectReplicationDLQTasksCount), ctx, shardID, sourceCluster)\n}\n\n// SelectReplicationDLQTasksOrderByTaskID mocks base method.\nfunc (m *MockDB) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksOrderByTaskID\", ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationDLQTasksOrderByTaskID indicates an expected call of SelectReplicationDLQTasksOrderByTaskID.\nfunc (mr *MockDBMockRecorder) SelectReplicationDLQTasksOrderByTaskID(ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksOrderByTaskID\", reflect.TypeOf((*MockDB)(nil).SelectReplicationDLQTasksOrderByTaskID), ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectReplicationTasksOrderByTaskID mocks base method.\nfunc (m *MockDB) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationTasksOrderByTaskID indicates an expected call of SelectReplicationTasksOrderByTaskID.\nfunc (mr *MockDBMockRecorder) SelectReplicationTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationTasksOrderByTaskID\", reflect.TypeOf((*MockDB)(nil).SelectReplicationTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectShard mocks base method.\nfunc (m *MockDB) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *ShardRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectShard\", ctx, shardID, currentClusterName)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(*ShardRow)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectShard indicates an expected call of SelectShard.\nfunc (mr *MockDBMockRecorder) SelectShard(ctx, shardID, currentClusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectShard\", reflect.TypeOf((*MockDB)(nil).SelectShard), ctx, shardID, currentClusterName)\n}\n\n// SelectTaskList mocks base method.\nfunc (m *MockDB) SelectTaskList(ctx context.Context, filter *TaskListFilter) (*TaskListRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTaskList\", ctx, filter)\n\tret0, _ := ret[0].(*TaskListRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTaskList indicates an expected call of SelectTaskList.\nfunc (mr *MockDBMockRecorder) SelectTaskList(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTaskList\", reflect.TypeOf((*MockDB)(nil).SelectTaskList), ctx, filter)\n}\n\n// SelectTasks mocks base method.\nfunc (m *MockDB) SelectTasks(ctx context.Context, filter *TasksFilter) ([]*TaskRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTasks\", ctx, filter)\n\tret0, _ := ret[0].([]*TaskRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTasks indicates an expected call of SelectTasks.\nfunc (mr *MockDBMockRecorder) SelectTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTasks\", reflect.TypeOf((*MockDB)(nil).SelectTasks), ctx, filter)\n}\n\n// SelectTimerTasksOrderByVisibilityTime mocks base method.\nfunc (m *MockDB) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTimerTasksOrderByVisibilityTime\", ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTimerTasksOrderByVisibilityTime indicates an expected call of SelectTimerTasksOrderByVisibilityTime.\nfunc (mr *MockDBMockRecorder) SelectTimerTasksOrderByVisibilityTime(ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTimerTasksOrderByVisibilityTime\", reflect.TypeOf((*MockDB)(nil).SelectTimerTasksOrderByVisibilityTime), ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// SelectTransferTasksOrderByTaskID mocks base method.\nfunc (m *MockDB) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTransferTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTransferTasksOrderByTaskID indicates an expected call of SelectTransferTasksOrderByTaskID.\nfunc (mr *MockDBMockRecorder) SelectTransferTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTransferTasksOrderByTaskID\", reflect.TypeOf((*MockDB)(nil).SelectTransferTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectVisibility mocks base method.\nfunc (m *MockDB) SelectVisibility(ctx context.Context, filter *VisibilityFilter) (*SelectVisibilityResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectVisibility\", ctx, filter)\n\tret0, _ := ret[0].(*SelectVisibilityResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectVisibility indicates an expected call of SelectVisibility.\nfunc (mr *MockDBMockRecorder) SelectVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectVisibility\", reflect.TypeOf((*MockDB)(nil).SelectVisibility), ctx, filter)\n}\n\n// SelectWorkflowExecution mocks base method.\nfunc (m *MockDB) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*WorkflowExecution, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*WorkflowExecution)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectWorkflowExecution indicates an expected call of SelectWorkflowExecution.\nfunc (mr *MockDBMockRecorder) SelectWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectWorkflowExecution\", reflect.TypeOf((*MockDB)(nil).SelectWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockDB) UpdateDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockDBMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockDB)(nil).UpdateDomain), ctx, row)\n}\n\n// UpdateQueueMetadataCas mocks base method.\nfunc (m *MockDB) UpdateQueueMetadataCas(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueMetadataCas\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueMetadataCas indicates an expected call of UpdateQueueMetadataCas.\nfunc (mr *MockDBMockRecorder) UpdateQueueMetadataCas(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueMetadataCas\", reflect.TypeOf((*MockDB)(nil).UpdateQueueMetadataCas), ctx, row)\n}\n\n// UpdateRangeID mocks base method.\nfunc (m *MockDB) UpdateRangeID(ctx context.Context, shardID int, rangeID, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateRangeID\", ctx, shardID, rangeID, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateRangeID indicates an expected call of UpdateRangeID.\nfunc (mr *MockDBMockRecorder) UpdateRangeID(ctx, shardID, rangeID, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateRangeID\", reflect.TypeOf((*MockDB)(nil).UpdateRangeID), ctx, shardID, rangeID, previousRangeID)\n}\n\n// UpdateShard mocks base method.\nfunc (m *MockDB) UpdateShard(ctx context.Context, row *ShardRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShard\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateShard indicates an expected call of UpdateShard.\nfunc (mr *MockDBMockRecorder) UpdateShard(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShard\", reflect.TypeOf((*MockDB)(nil).UpdateShard), ctx, row, previousRangeID)\n}\n\n// UpdateTaskList mocks base method.\nfunc (m *MockDB) UpdateTaskList(ctx context.Context, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskList\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskList indicates an expected call of UpdateTaskList.\nfunc (mr *MockDBMockRecorder) UpdateTaskList(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskList\", reflect.TypeOf((*MockDB)(nil).UpdateTaskList), ctx, row, previousRangeID)\n}\n\n// UpdateTaskListWithTTL mocks base method.\nfunc (m *MockDB) UpdateTaskListWithTTL(ctx context.Context, ttlSeconds int64, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListWithTTL\", ctx, ttlSeconds, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskListWithTTL indicates an expected call of UpdateTaskListWithTTL.\nfunc (mr *MockDBMockRecorder) UpdateTaskListWithTTL(ctx, ttlSeconds, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListWithTTL\", reflect.TypeOf((*MockDB)(nil).UpdateTaskListWithTTL), ctx, ttlSeconds, row, previousRangeID)\n}\n\n// UpdateVisibility mocks base method.\nfunc (m *MockDB) UpdateVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForUpdate) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateVisibility indicates an expected call of UpdateVisibility.\nfunc (mr *MockDBMockRecorder) UpdateVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateVisibility\", reflect.TypeOf((*MockDB)(nil).UpdateVisibility), ctx, ttlSeconds, row)\n}\n\n// UpdateWorkflowExecutionWithTasks mocks base method.\nfunc (m *MockDB) UpdateWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, mutatedExecution, insertedExecution *WorkflowExecutionRequest, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, resetExecution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithTasks indicates an expected call of UpdateWorkflowExecutionWithTasks.\nfunc (mr *MockDBMockRecorder) UpdateWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithTasks\", reflect.TypeOf((*MockDB)(nil).UpdateWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n}\n\n// MocktableCRUD is a mock of tableCRUD interface.\ntype MocktableCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MocktableCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MocktableCRUDMockRecorder is the mock recorder for MocktableCRUD.\ntype MocktableCRUDMockRecorder struct {\n\tmock *MocktableCRUD\n}\n\n// NewMocktableCRUD creates a new mock instance.\nfunc NewMocktableCRUD(ctrl *gomock.Controller) *MocktableCRUD {\n\tmock := &MocktableCRUD{ctrl: ctrl}\n\tmock.recorder = &MocktableCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MocktableCRUD) EXPECT() *MocktableCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteActiveClusterSelectionPolicy mocks base method.\nfunc (m *MocktableCRUD) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActiveClusterSelectionPolicy\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActiveClusterSelectionPolicy indicates an expected call of DeleteActiveClusterSelectionPolicy.\nfunc (mr *MocktableCRUDMockRecorder) DeleteActiveClusterSelectionPolicy(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActiveClusterSelectionPolicy\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteActiveClusterSelectionPolicy), ctx, shardID, domainID, workflowID, runID)\n}\n\n// DeleteCrossClusterTask mocks base method.\nfunc (m *MocktableCRUD) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCrossClusterTask\", ctx, shardID, targetCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCrossClusterTask indicates an expected call of DeleteCrossClusterTask.\nfunc (mr *MocktableCRUDMockRecorder) DeleteCrossClusterTask(ctx, shardID, targetCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCrossClusterTask\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteCrossClusterTask), ctx, shardID, targetCluster, taskID)\n}\n\n// DeleteCurrentWorkflow mocks base method.\nfunc (m *MocktableCRUD) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflow\", ctx, shardID, domainID, workflowID, currentRunIDCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflow indicates an expected call of DeleteCurrentWorkflow.\nfunc (mr *MocktableCRUDMockRecorder) DeleteCurrentWorkflow(ctx, shardID, domainID, workflowID, currentRunIDCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflow\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteCurrentWorkflow), ctx, shardID, domainID, workflowID, currentRunIDCondition)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MocktableCRUD) DeleteDomain(ctx context.Context, domainID, domainName *string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MocktableCRUDMockRecorder) DeleteDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteDomain), ctx, domainID, domainName)\n}\n\n// DeleteFromHistoryTreeAndNode mocks base method.\nfunc (m *MocktableCRUD) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *HistoryTreeFilter, nodeFilters []*HistoryNodeFilter) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTreeAndNode\", ctx, treeFilter, nodeFilters)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteFromHistoryTreeAndNode indicates an expected call of DeleteFromHistoryTreeAndNode.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromHistoryTreeAndNode(ctx, treeFilter, nodeFilters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTreeAndNode\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromHistoryTreeAndNode), ctx, treeFilter, nodeFilters)\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MocktableCRUD) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MocktableCRUD) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, exclusiveBeginMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessagesBefore(ctx, queueType, exclusiveBeginMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessagesBefore), ctx, queueType, exclusiveBeginMessageID)\n}\n\n// DeleteMessagesInRange mocks base method.\nfunc (m *MocktableCRUD) DeleteMessagesInRange(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesInRange\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesInRange indicates an expected call of DeleteMessagesInRange.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessagesInRange(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesInRange\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessagesInRange), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// DeleteReplicationDLQTask mocks base method.\nfunc (m *MocktableCRUD) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationDLQTask\", ctx, shardID, sourceCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationDLQTask indicates an expected call of DeleteReplicationDLQTask.\nfunc (mr *MocktableCRUDMockRecorder) DeleteReplicationDLQTask(ctx, shardID, sourceCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationDLQTask\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteReplicationDLQTask), ctx, shardID, sourceCluster, taskID)\n}\n\n// DeleteReplicationTask mocks base method.\nfunc (m *MocktableCRUD) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationTask indicates an expected call of DeleteReplicationTask.\nfunc (mr *MocktableCRUDMockRecorder) DeleteReplicationTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationTask\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteReplicationTask), ctx, shardID, taskID)\n}\n\n// DeleteTaskList mocks base method.\nfunc (m *MocktableCRUD) DeleteTaskList(ctx context.Context, filter *TaskListFilter, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTaskList\", ctx, filter, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTaskList indicates an expected call of DeleteTaskList.\nfunc (mr *MocktableCRUDMockRecorder) DeleteTaskList(ctx, filter, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTaskList\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteTaskList), ctx, filter, previousRangeID)\n}\n\n// DeleteTimerTask mocks base method.\nfunc (m *MocktableCRUD) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTimerTask\", ctx, shardID, taskID, visibilityTimestamp)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTimerTask indicates an expected call of DeleteTimerTask.\nfunc (mr *MocktableCRUDMockRecorder) DeleteTimerTask(ctx, shardID, taskID, visibilityTimestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTimerTask\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteTimerTask), ctx, shardID, taskID, visibilityTimestamp)\n}\n\n// DeleteTransferTask mocks base method.\nfunc (m *MocktableCRUD) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTransferTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTransferTask indicates an expected call of DeleteTransferTask.\nfunc (mr *MocktableCRUDMockRecorder) DeleteTransferTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTransferTask\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteTransferTask), ctx, shardID, taskID)\n}\n\n// DeleteVisibility mocks base method.\nfunc (m *MocktableCRUD) DeleteVisibility(ctx context.Context, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteVisibility\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteVisibility indicates an expected call of DeleteVisibility.\nfunc (mr *MocktableCRUDMockRecorder) DeleteVisibility(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteVisibility), ctx, domainID, workflowID, runID)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MocktableCRUD) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MocktableCRUDMockRecorder) DeleteWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MocktableCRUD) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MocktableCRUDMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MocktableCRUD)(nil).GetQueueSize), ctx, queueType)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MocktableCRUD) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MocktableCRUDMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MocktableCRUD)(nil).GetTasksCount), ctx, filter)\n}\n\n// InsertConfig mocks base method.\nfunc (m *MocktableCRUD) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MocktableCRUDMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MocktableCRUD)(nil).InsertConfig), ctx, row)\n}\n\n// InsertDomain mocks base method.\nfunc (m *MocktableCRUD) InsertDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomain indicates an expected call of InsertDomain.\nfunc (mr *MocktableCRUDMockRecorder) InsertDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomain\", reflect.TypeOf((*MocktableCRUD)(nil).InsertDomain), ctx, row)\n}\n\n// InsertDomainAuditLog mocks base method.\nfunc (m *MocktableCRUD) InsertDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomainAuditLog indicates an expected call of InsertDomainAuditLog.\nfunc (mr *MocktableCRUDMockRecorder) InsertDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomainAuditLog\", reflect.TypeOf((*MocktableCRUD)(nil).InsertDomainAuditLog), ctx, row)\n}\n\n// InsertIntoHistoryTreeAndNode mocks base method.\nfunc (m *MocktableCRUD) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *HistoryTreeRow, nodeRow *HistoryNodeRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTreeAndNode\", ctx, treeRow, nodeRow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoHistoryTreeAndNode indicates an expected call of InsertIntoHistoryTreeAndNode.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoHistoryTreeAndNode(ctx, treeRow, nodeRow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTreeAndNode\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoHistoryTreeAndNode), ctx, treeRow, nodeRow)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MocktableCRUD) InsertIntoQueue(ctx context.Context, row *QueueMessageRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertQueueMetadata mocks base method.\nfunc (m *MocktableCRUD) InsertQueueMetadata(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertQueueMetadata\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertQueueMetadata indicates an expected call of InsertQueueMetadata.\nfunc (mr *MocktableCRUDMockRecorder) InsertQueueMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertQueueMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).InsertQueueMetadata), ctx, row)\n}\n\n// InsertReplicationDLQTask mocks base method.\nfunc (m *MocktableCRUD) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *HistoryMigrationTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationDLQTask\", ctx, shardID, sourceCluster, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationDLQTask indicates an expected call of InsertReplicationDLQTask.\nfunc (mr *MocktableCRUDMockRecorder) InsertReplicationDLQTask(ctx, shardID, sourceCluster, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationDLQTask\", reflect.TypeOf((*MocktableCRUD)(nil).InsertReplicationDLQTask), ctx, shardID, sourceCluster, task)\n}\n\n// InsertReplicationTask mocks base method.\nfunc (m *MocktableCRUD) InsertReplicationTask(ctx context.Context, tasks []*HistoryMigrationTask, condition ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationTask\", ctx, tasks, condition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationTask indicates an expected call of InsertReplicationTask.\nfunc (mr *MocktableCRUDMockRecorder) InsertReplicationTask(ctx, tasks, condition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationTask\", reflect.TypeOf((*MocktableCRUD)(nil).InsertReplicationTask), ctx, tasks, condition)\n}\n\n// InsertShard mocks base method.\nfunc (m *MocktableCRUD) InsertShard(ctx context.Context, row *ShardRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertShard\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertShard indicates an expected call of InsertShard.\nfunc (mr *MocktableCRUDMockRecorder) InsertShard(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertShard\", reflect.TypeOf((*MocktableCRUD)(nil).InsertShard), ctx, row)\n}\n\n// InsertTaskList mocks base method.\nfunc (m *MocktableCRUD) InsertTaskList(ctx context.Context, row *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTaskList\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTaskList indicates an expected call of InsertTaskList.\nfunc (mr *MocktableCRUDMockRecorder) InsertTaskList(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTaskList\", reflect.TypeOf((*MocktableCRUD)(nil).InsertTaskList), ctx, row)\n}\n\n// InsertTasks mocks base method.\nfunc (m *MocktableCRUD) InsertTasks(ctx context.Context, tasksToInsert []*TaskRowForInsert, tasklistCondition *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTasks\", ctx, tasksToInsert, tasklistCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTasks indicates an expected call of InsertTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertTasks(ctx, tasksToInsert, tasklistCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertTasks), ctx, tasksToInsert, tasklistCondition)\n}\n\n// InsertVisibility mocks base method.\nfunc (m *MocktableCRUD) InsertVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForInsert) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertVisibility indicates an expected call of InsertVisibility.\nfunc (mr *MocktableCRUDMockRecorder) InsertVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).InsertVisibility), ctx, ttlSeconds, row)\n}\n\n// InsertWorkflowExecutionWithTasks mocks base method.\nfunc (m *MocktableCRUD) InsertWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, execution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertWorkflowExecutionWithTasks indicates an expected call of InsertWorkflowExecutionWithTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertWorkflowExecutionWithTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MocktableCRUD) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MocktableCRUDMockRecorder) IsWorkflowExecutionExists(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MocktableCRUD)(nil).IsWorkflowExecutionExists), ctx, shardID, domainID, workflowID, runID)\n}\n\n// ListTaskList mocks base method.\nfunc (m *MocktableCRUD) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*ListTaskListResult, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskList\", ctx, pageSize, nextPageToken)\n\tret0, _ := ret[0].(*ListTaskListResult)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskList indicates an expected call of ListTaskList.\nfunc (mr *MocktableCRUDMockRecorder) ListTaskList(ctx, pageSize, nextPageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskList\", reflect.TypeOf((*MocktableCRUD)(nil).ListTaskList), ctx, pageSize, nextPageToken)\n}\n\n// RangeDeleteReplicationDLQTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationDLQTasks\", ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationDLQTasks indicates an expected call of RangeDeleteReplicationDLQTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteReplicationDLQTasks(ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationDLQTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteReplicationDLQTasks), ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// RangeDeleteReplicationTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationTasks\", ctx, shardID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationTasks indicates an expected call of RangeDeleteReplicationTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteReplicationTasks(ctx, shardID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteReplicationTasks), ctx, shardID, exclusiveEndTaskID)\n}\n\n// RangeDeleteTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteTasks(ctx context.Context, filter *TasksFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTasks\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteTasks indicates an expected call of RangeDeleteTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteTasks), ctx, filter)\n}\n\n// RangeDeleteTimerTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTimerTasks\", ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTimerTasks indicates an expected call of RangeDeleteTimerTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteTimerTasks(ctx, shardID, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTimerTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteTimerTasks), ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// RangeDeleteTransferTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTransferTasks\", ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTransferTasks indicates an expected call of RangeDeleteTransferTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteTransferTasks(ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTransferTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteTransferTasks), ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// SelectActiveClusterSelectionPolicy mocks base method.\nfunc (m *MocktableCRUD) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*ActiveClusterSelectionPolicyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectActiveClusterSelectionPolicy\", ctx, shardID, domainID, wfID, rID)\n\tret0, _ := ret[0].(*ActiveClusterSelectionPolicyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectActiveClusterSelectionPolicy indicates an expected call of SelectActiveClusterSelectionPolicy.\nfunc (mr *MocktableCRUDMockRecorder) SelectActiveClusterSelectionPolicy(ctx, shardID, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectActiveClusterSelectionPolicy\", reflect.TypeOf((*MocktableCRUD)(nil).SelectActiveClusterSelectionPolicy), ctx, shardID, domainID, wfID, rID)\n}\n\n// SelectAllCurrentWorkflows mocks base method.\nfunc (m *MocktableCRUD) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllCurrentWorkflows\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.CurrentWorkflowExecution)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllCurrentWorkflows indicates an expected call of SelectAllCurrentWorkflows.\nfunc (mr *MocktableCRUDMockRecorder) SelectAllCurrentWorkflows(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllCurrentWorkflows\", reflect.TypeOf((*MocktableCRUD)(nil).SelectAllCurrentWorkflows), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectAllDomains mocks base method.\nfunc (m *MocktableCRUD) SelectAllDomains(ctx context.Context, pageSize int, pageToken []byte) ([]*DomainRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllDomains\", ctx, pageSize, pageToken)\n\tret0, _ := ret[0].([]*DomainRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllDomains indicates an expected call of SelectAllDomains.\nfunc (mr *MocktableCRUDMockRecorder) SelectAllDomains(ctx, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllDomains\", reflect.TypeOf((*MocktableCRUD)(nil).SelectAllDomains), ctx, pageSize, pageToken)\n}\n\n// SelectAllHistoryTrees mocks base method.\nfunc (m *MocktableCRUD) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*HistoryTreeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllHistoryTrees\", ctx, nextPageToken, pageSize)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllHistoryTrees indicates an expected call of SelectAllHistoryTrees.\nfunc (mr *MocktableCRUDMockRecorder) SelectAllHistoryTrees(ctx, nextPageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllHistoryTrees\", reflect.TypeOf((*MocktableCRUD)(nil).SelectAllHistoryTrees), ctx, nextPageToken, pageSize)\n}\n\n// SelectAllWorkflowExecutions mocks base method.\nfunc (m *MocktableCRUD) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllWorkflowExecutions\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.InternalListConcreteExecutionsEntity)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllWorkflowExecutions indicates an expected call of SelectAllWorkflowExecutions.\nfunc (mr *MocktableCRUDMockRecorder) SelectAllWorkflowExecutions(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllWorkflowExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).SelectAllWorkflowExecutions), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectCurrentWorkflow mocks base method.\nfunc (m *MocktableCRUD) SelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*CurrentWorkflowRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectCurrentWorkflow\", ctx, shardID, domainID, workflowID)\n\tret0, _ := ret[0].(*CurrentWorkflowRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectCurrentWorkflow indicates an expected call of SelectCurrentWorkflow.\nfunc (mr *MocktableCRUDMockRecorder) SelectCurrentWorkflow(ctx, shardID, domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectCurrentWorkflow\", reflect.TypeOf((*MocktableCRUD)(nil).SelectCurrentWorkflow), ctx, shardID, domainID, workflowID)\n}\n\n// SelectDomain mocks base method.\nfunc (m *MocktableCRUD) SelectDomain(ctx context.Context, domainID, domainName *string) (*DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(*DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomain indicates an expected call of SelectDomain.\nfunc (mr *MocktableCRUDMockRecorder) SelectDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomain\", reflect.TypeOf((*MocktableCRUD)(nil).SelectDomain), ctx, domainID, domainName)\n}\n\n// SelectDomainAuditLogs mocks base method.\nfunc (m *MocktableCRUD) SelectDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectDomainAuditLogs indicates an expected call of SelectDomainAuditLogs.\nfunc (mr *MocktableCRUDMockRecorder) SelectDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainAuditLogs\", reflect.TypeOf((*MocktableCRUD)(nil).SelectDomainAuditLogs), ctx, filter)\n}\n\n// SelectDomainMetadata mocks base method.\nfunc (m *MocktableCRUD) SelectDomainMetadata(ctx context.Context) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainMetadata\", ctx)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomainMetadata indicates an expected call of SelectDomainMetadata.\nfunc (mr *MocktableCRUDMockRecorder) SelectDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).SelectDomainMetadata), ctx)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MocktableCRUD) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]*HistoryNodeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryNodeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MocktableCRUD) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]*HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// SelectLastEnqueuedMessageID mocks base method.\nfunc (m *MocktableCRUD) SelectLastEnqueuedMessageID(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLastEnqueuedMessageID\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLastEnqueuedMessageID indicates an expected call of SelectLastEnqueuedMessageID.\nfunc (mr *MocktableCRUDMockRecorder) SelectLastEnqueuedMessageID(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLastEnqueuedMessageID\", reflect.TypeOf((*MocktableCRUD)(nil).SelectLastEnqueuedMessageID), ctx, queueType)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MocktableCRUD) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MocktableCRUDMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MocktableCRUD)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// SelectMessagesBetween mocks base method.\nfunc (m *MocktableCRUD) SelectMessagesBetween(ctx context.Context, request SelectMessagesBetweenRequest) (*SelectMessagesBetweenResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesBetween\", ctx, request)\n\tret0, _ := ret[0].(*SelectMessagesBetweenResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesBetween indicates an expected call of SelectMessagesBetween.\nfunc (mr *MocktableCRUDMockRecorder) SelectMessagesBetween(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesBetween\", reflect.TypeOf((*MocktableCRUD)(nil).SelectMessagesBetween), ctx, request)\n}\n\n// SelectMessagesFrom mocks base method.\nfunc (m *MocktableCRUD) SelectMessagesFrom(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, maxRows int) ([]*QueueMessageRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesFrom\", ctx, queueType, exclusiveBeginMessageID, maxRows)\n\tret0, _ := ret[0].([]*QueueMessageRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesFrom indicates an expected call of SelectMessagesFrom.\nfunc (mr *MocktableCRUDMockRecorder) SelectMessagesFrom(ctx, queueType, exclusiveBeginMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesFrom\", reflect.TypeOf((*MocktableCRUD)(nil).SelectMessagesFrom), ctx, queueType, exclusiveBeginMessageID, maxRows)\n}\n\n// SelectOneClosedWorkflow mocks base method.\nfunc (m *MocktableCRUD) SelectOneClosedWorkflow(ctx context.Context, domainID, workflowID, runID string) (*VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectOneClosedWorkflow\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectOneClosedWorkflow indicates an expected call of SelectOneClosedWorkflow.\nfunc (mr *MocktableCRUDMockRecorder) SelectOneClosedWorkflow(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectOneClosedWorkflow\", reflect.TypeOf((*MocktableCRUD)(nil).SelectOneClosedWorkflow), ctx, domainID, workflowID, runID)\n}\n\n// SelectQueueMetadata mocks base method.\nfunc (m *MocktableCRUD) SelectQueueMetadata(ctx context.Context, queueType persistence.QueueType) (*QueueMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectQueueMetadata\", ctx, queueType)\n\tret0, _ := ret[0].(*QueueMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectQueueMetadata indicates an expected call of SelectQueueMetadata.\nfunc (mr *MocktableCRUDMockRecorder) SelectQueueMetadata(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectQueueMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).SelectQueueMetadata), ctx, queueType)\n}\n\n// SelectReplicationDLQTasksCount mocks base method.\nfunc (m *MocktableCRUD) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksCount\", ctx, shardID, sourceCluster)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectReplicationDLQTasksCount indicates an expected call of SelectReplicationDLQTasksCount.\nfunc (mr *MocktableCRUDMockRecorder) SelectReplicationDLQTasksCount(ctx, shardID, sourceCluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksCount\", reflect.TypeOf((*MocktableCRUD)(nil).SelectReplicationDLQTasksCount), ctx, shardID, sourceCluster)\n}\n\n// SelectReplicationDLQTasksOrderByTaskID mocks base method.\nfunc (m *MocktableCRUD) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksOrderByTaskID\", ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationDLQTasksOrderByTaskID indicates an expected call of SelectReplicationDLQTasksOrderByTaskID.\nfunc (mr *MocktableCRUDMockRecorder) SelectReplicationDLQTasksOrderByTaskID(ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksOrderByTaskID\", reflect.TypeOf((*MocktableCRUD)(nil).SelectReplicationDLQTasksOrderByTaskID), ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectReplicationTasksOrderByTaskID mocks base method.\nfunc (m *MocktableCRUD) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationTasksOrderByTaskID indicates an expected call of SelectReplicationTasksOrderByTaskID.\nfunc (mr *MocktableCRUDMockRecorder) SelectReplicationTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationTasksOrderByTaskID\", reflect.TypeOf((*MocktableCRUD)(nil).SelectReplicationTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectShard mocks base method.\nfunc (m *MocktableCRUD) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *ShardRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectShard\", ctx, shardID, currentClusterName)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(*ShardRow)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectShard indicates an expected call of SelectShard.\nfunc (mr *MocktableCRUDMockRecorder) SelectShard(ctx, shardID, currentClusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectShard\", reflect.TypeOf((*MocktableCRUD)(nil).SelectShard), ctx, shardID, currentClusterName)\n}\n\n// SelectTaskList mocks base method.\nfunc (m *MocktableCRUD) SelectTaskList(ctx context.Context, filter *TaskListFilter) (*TaskListRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTaskList\", ctx, filter)\n\tret0, _ := ret[0].(*TaskListRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTaskList indicates an expected call of SelectTaskList.\nfunc (mr *MocktableCRUDMockRecorder) SelectTaskList(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTaskList\", reflect.TypeOf((*MocktableCRUD)(nil).SelectTaskList), ctx, filter)\n}\n\n// SelectTasks mocks base method.\nfunc (m *MocktableCRUD) SelectTasks(ctx context.Context, filter *TasksFilter) ([]*TaskRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTasks\", ctx, filter)\n\tret0, _ := ret[0].([]*TaskRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTasks indicates an expected call of SelectTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectTasks), ctx, filter)\n}\n\n// SelectTimerTasksOrderByVisibilityTime mocks base method.\nfunc (m *MocktableCRUD) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTimerTasksOrderByVisibilityTime\", ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTimerTasksOrderByVisibilityTime indicates an expected call of SelectTimerTasksOrderByVisibilityTime.\nfunc (mr *MocktableCRUDMockRecorder) SelectTimerTasksOrderByVisibilityTime(ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTimerTasksOrderByVisibilityTime\", reflect.TypeOf((*MocktableCRUD)(nil).SelectTimerTasksOrderByVisibilityTime), ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// SelectTransferTasksOrderByTaskID mocks base method.\nfunc (m *MocktableCRUD) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTransferTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTransferTasksOrderByTaskID indicates an expected call of SelectTransferTasksOrderByTaskID.\nfunc (mr *MocktableCRUDMockRecorder) SelectTransferTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTransferTasksOrderByTaskID\", reflect.TypeOf((*MocktableCRUD)(nil).SelectTransferTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectVisibility mocks base method.\nfunc (m *MocktableCRUD) SelectVisibility(ctx context.Context, filter *VisibilityFilter) (*SelectVisibilityResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectVisibility\", ctx, filter)\n\tret0, _ := ret[0].(*SelectVisibilityResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectVisibility indicates an expected call of SelectVisibility.\nfunc (mr *MocktableCRUDMockRecorder) SelectVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).SelectVisibility), ctx, filter)\n}\n\n// SelectWorkflowExecution mocks base method.\nfunc (m *MocktableCRUD) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*WorkflowExecution, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*WorkflowExecution)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectWorkflowExecution indicates an expected call of SelectWorkflowExecution.\nfunc (mr *MocktableCRUDMockRecorder) SelectWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectWorkflowExecution\", reflect.TypeOf((*MocktableCRUD)(nil).SelectWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MocktableCRUD) UpdateDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MocktableCRUDMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateDomain), ctx, row)\n}\n\n// UpdateQueueMetadataCas mocks base method.\nfunc (m *MocktableCRUD) UpdateQueueMetadataCas(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueMetadataCas\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueMetadataCas indicates an expected call of UpdateQueueMetadataCas.\nfunc (mr *MocktableCRUDMockRecorder) UpdateQueueMetadataCas(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueMetadataCas\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateQueueMetadataCas), ctx, row)\n}\n\n// UpdateRangeID mocks base method.\nfunc (m *MocktableCRUD) UpdateRangeID(ctx context.Context, shardID int, rangeID, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateRangeID\", ctx, shardID, rangeID, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateRangeID indicates an expected call of UpdateRangeID.\nfunc (mr *MocktableCRUDMockRecorder) UpdateRangeID(ctx, shardID, rangeID, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateRangeID\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateRangeID), ctx, shardID, rangeID, previousRangeID)\n}\n\n// UpdateShard mocks base method.\nfunc (m *MocktableCRUD) UpdateShard(ctx context.Context, row *ShardRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShard\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateShard indicates an expected call of UpdateShard.\nfunc (mr *MocktableCRUDMockRecorder) UpdateShard(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShard\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateShard), ctx, row, previousRangeID)\n}\n\n// UpdateTaskList mocks base method.\nfunc (m *MocktableCRUD) UpdateTaskList(ctx context.Context, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskList\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskList indicates an expected call of UpdateTaskList.\nfunc (mr *MocktableCRUDMockRecorder) UpdateTaskList(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskList\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateTaskList), ctx, row, previousRangeID)\n}\n\n// UpdateTaskListWithTTL mocks base method.\nfunc (m *MocktableCRUD) UpdateTaskListWithTTL(ctx context.Context, ttlSeconds int64, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListWithTTL\", ctx, ttlSeconds, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskListWithTTL indicates an expected call of UpdateTaskListWithTTL.\nfunc (mr *MocktableCRUDMockRecorder) UpdateTaskListWithTTL(ctx, ttlSeconds, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListWithTTL\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateTaskListWithTTL), ctx, ttlSeconds, row, previousRangeID)\n}\n\n// UpdateVisibility mocks base method.\nfunc (m *MocktableCRUD) UpdateVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForUpdate) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateVisibility indicates an expected call of UpdateVisibility.\nfunc (mr *MocktableCRUDMockRecorder) UpdateVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateVisibility), ctx, ttlSeconds, row)\n}\n\n// UpdateWorkflowExecutionWithTasks mocks base method.\nfunc (m *MocktableCRUD) UpdateWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, mutatedExecution, insertedExecution *WorkflowExecutionRequest, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, resetExecution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithTasks indicates an expected call of UpdateWorkflowExecutionWithTasks.\nfunc (mr *MocktableCRUDMockRecorder) UpdateWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithTasks\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n}\n\n// MockClientErrorChecker is a mock of ClientErrorChecker interface.\ntype MockClientErrorChecker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientErrorCheckerMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientErrorCheckerMockRecorder is the mock recorder for MockClientErrorChecker.\ntype MockClientErrorCheckerMockRecorder struct {\n\tmock *MockClientErrorChecker\n}\n\n// NewMockClientErrorChecker creates a new mock instance.\nfunc NewMockClientErrorChecker(ctrl *gomock.Controller) *MockClientErrorChecker {\n\tmock := &MockClientErrorChecker{ctrl: ctrl}\n\tmock.recorder = &MockClientErrorCheckerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClientErrorChecker) EXPECT() *MockClientErrorCheckerMockRecorder {\n\treturn m.recorder\n}\n\n// IsDBUnavailableError mocks base method.\nfunc (m *MockClientErrorChecker) IsDBUnavailableError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDBUnavailableError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDBUnavailableError indicates an expected call of IsDBUnavailableError.\nfunc (mr *MockClientErrorCheckerMockRecorder) IsDBUnavailableError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDBUnavailableError\", reflect.TypeOf((*MockClientErrorChecker)(nil).IsDBUnavailableError), arg0)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockClientErrorChecker) IsNotFoundError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockClientErrorCheckerMockRecorder) IsNotFoundError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockClientErrorChecker)(nil).IsNotFoundError), arg0)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockClientErrorChecker) IsThrottlingError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockClientErrorCheckerMockRecorder) IsThrottlingError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockClientErrorChecker)(nil).IsThrottlingError), arg0)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockClientErrorChecker) IsTimeoutError(arg0 error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockClientErrorCheckerMockRecorder) IsTimeoutError(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockClientErrorChecker)(nil).IsTimeoutError), arg0)\n}\n\n// MockHistoryEventsCRUD is a mock of HistoryEventsCRUD interface.\ntype MockHistoryEventsCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHistoryEventsCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockHistoryEventsCRUDMockRecorder is the mock recorder for MockHistoryEventsCRUD.\ntype MockHistoryEventsCRUDMockRecorder struct {\n\tmock *MockHistoryEventsCRUD\n}\n\n// NewMockHistoryEventsCRUD creates a new mock instance.\nfunc NewMockHistoryEventsCRUD(ctrl *gomock.Controller) *MockHistoryEventsCRUD {\n\tmock := &MockHistoryEventsCRUD{ctrl: ctrl}\n\tmock.recorder = &MockHistoryEventsCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHistoryEventsCRUD) EXPECT() *MockHistoryEventsCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteFromHistoryTreeAndNode mocks base method.\nfunc (m *MockHistoryEventsCRUD) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *HistoryTreeFilter, nodeFilters []*HistoryNodeFilter) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTreeAndNode\", ctx, treeFilter, nodeFilters)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteFromHistoryTreeAndNode indicates an expected call of DeleteFromHistoryTreeAndNode.\nfunc (mr *MockHistoryEventsCRUDMockRecorder) DeleteFromHistoryTreeAndNode(ctx, treeFilter, nodeFilters any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTreeAndNode\", reflect.TypeOf((*MockHistoryEventsCRUD)(nil).DeleteFromHistoryTreeAndNode), ctx, treeFilter, nodeFilters)\n}\n\n// InsertIntoHistoryTreeAndNode mocks base method.\nfunc (m *MockHistoryEventsCRUD) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *HistoryTreeRow, nodeRow *HistoryNodeRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTreeAndNode\", ctx, treeRow, nodeRow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoHistoryTreeAndNode indicates an expected call of InsertIntoHistoryTreeAndNode.\nfunc (mr *MockHistoryEventsCRUDMockRecorder) InsertIntoHistoryTreeAndNode(ctx, treeRow, nodeRow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTreeAndNode\", reflect.TypeOf((*MockHistoryEventsCRUD)(nil).InsertIntoHistoryTreeAndNode), ctx, treeRow, nodeRow)\n}\n\n// SelectAllHistoryTrees mocks base method.\nfunc (m *MockHistoryEventsCRUD) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*HistoryTreeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllHistoryTrees\", ctx, nextPageToken, pageSize)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllHistoryTrees indicates an expected call of SelectAllHistoryTrees.\nfunc (mr *MockHistoryEventsCRUDMockRecorder) SelectAllHistoryTrees(ctx, nextPageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllHistoryTrees\", reflect.TypeOf((*MockHistoryEventsCRUD)(nil).SelectAllHistoryTrees), ctx, nextPageToken, pageSize)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MockHistoryEventsCRUD) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]*HistoryNodeRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryNodeRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MockHistoryEventsCRUDMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MockHistoryEventsCRUD)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MockHistoryEventsCRUD) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]*HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]*HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MockHistoryEventsCRUDMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MockHistoryEventsCRUD)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// MockMessageQueueCRUD is a mock of MessageQueueCRUD interface.\ntype MockMessageQueueCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMessageQueueCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockMessageQueueCRUDMockRecorder is the mock recorder for MockMessageQueueCRUD.\ntype MockMessageQueueCRUDMockRecorder struct {\n\tmock *MockMessageQueueCRUD\n}\n\n// NewMockMessageQueueCRUD creates a new mock instance.\nfunc NewMockMessageQueueCRUD(ctrl *gomock.Controller) *MockMessageQueueCRUD {\n\tmock := &MockMessageQueueCRUD{ctrl: ctrl}\n\tmock.recorder = &MockMessageQueueCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMessageQueueCRUD) EXPECT() *MockMessageQueueCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MockMessageQueueCRUD) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MockMessageQueueCRUDMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MockMessageQueueCRUD) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, exclusiveBeginMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MockMessageQueueCRUDMockRecorder) DeleteMessagesBefore(ctx, queueType, exclusiveBeginMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).DeleteMessagesBefore), ctx, queueType, exclusiveBeginMessageID)\n}\n\n// DeleteMessagesInRange mocks base method.\nfunc (m *MockMessageQueueCRUD) DeleteMessagesInRange(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesInRange\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteMessagesInRange indicates an expected call of DeleteMessagesInRange.\nfunc (mr *MockMessageQueueCRUDMockRecorder) DeleteMessagesInRange(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesInRange\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).DeleteMessagesInRange), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MockMessageQueueCRUD) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MockMessageQueueCRUDMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).GetQueueSize), ctx, queueType)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MockMessageQueueCRUD) InsertIntoQueue(ctx context.Context, row *QueueMessageRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MockMessageQueueCRUDMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertQueueMetadata mocks base method.\nfunc (m *MockMessageQueueCRUD) InsertQueueMetadata(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertQueueMetadata\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertQueueMetadata indicates an expected call of InsertQueueMetadata.\nfunc (mr *MockMessageQueueCRUDMockRecorder) InsertQueueMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertQueueMetadata\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).InsertQueueMetadata), ctx, row)\n}\n\n// SelectLastEnqueuedMessageID mocks base method.\nfunc (m *MockMessageQueueCRUD) SelectLastEnqueuedMessageID(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLastEnqueuedMessageID\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLastEnqueuedMessageID indicates an expected call of SelectLastEnqueuedMessageID.\nfunc (mr *MockMessageQueueCRUDMockRecorder) SelectLastEnqueuedMessageID(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLastEnqueuedMessageID\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).SelectLastEnqueuedMessageID), ctx, queueType)\n}\n\n// SelectMessagesBetween mocks base method.\nfunc (m *MockMessageQueueCRUD) SelectMessagesBetween(ctx context.Context, request SelectMessagesBetweenRequest) (*SelectMessagesBetweenResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesBetween\", ctx, request)\n\tret0, _ := ret[0].(*SelectMessagesBetweenResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesBetween indicates an expected call of SelectMessagesBetween.\nfunc (mr *MockMessageQueueCRUDMockRecorder) SelectMessagesBetween(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesBetween\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).SelectMessagesBetween), ctx, request)\n}\n\n// SelectMessagesFrom mocks base method.\nfunc (m *MockMessageQueueCRUD) SelectMessagesFrom(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, maxRows int) ([]*QueueMessageRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectMessagesFrom\", ctx, queueType, exclusiveBeginMessageID, maxRows)\n\tret0, _ := ret[0].([]*QueueMessageRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectMessagesFrom indicates an expected call of SelectMessagesFrom.\nfunc (mr *MockMessageQueueCRUDMockRecorder) SelectMessagesFrom(ctx, queueType, exclusiveBeginMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectMessagesFrom\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).SelectMessagesFrom), ctx, queueType, exclusiveBeginMessageID, maxRows)\n}\n\n// SelectQueueMetadata mocks base method.\nfunc (m *MockMessageQueueCRUD) SelectQueueMetadata(ctx context.Context, queueType persistence.QueueType) (*QueueMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectQueueMetadata\", ctx, queueType)\n\tret0, _ := ret[0].(*QueueMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectQueueMetadata indicates an expected call of SelectQueueMetadata.\nfunc (mr *MockMessageQueueCRUDMockRecorder) SelectQueueMetadata(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectQueueMetadata\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).SelectQueueMetadata), ctx, queueType)\n}\n\n// UpdateQueueMetadataCas mocks base method.\nfunc (m *MockMessageQueueCRUD) UpdateQueueMetadataCas(ctx context.Context, row QueueMetadataRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueMetadataCas\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueMetadataCas indicates an expected call of UpdateQueueMetadataCas.\nfunc (mr *MockMessageQueueCRUDMockRecorder) UpdateQueueMetadataCas(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueMetadataCas\", reflect.TypeOf((*MockMessageQueueCRUD)(nil).UpdateQueueMetadataCas), ctx, row)\n}\n\n// MockDomainCRUD is a mock of DomainCRUD interface.\ntype MockDomainCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainCRUDMockRecorder is the mock recorder for MockDomainCRUD.\ntype MockDomainCRUDMockRecorder struct {\n\tmock *MockDomainCRUD\n}\n\n// NewMockDomainCRUD creates a new mock instance.\nfunc NewMockDomainCRUD(ctrl *gomock.Controller) *MockDomainCRUD {\n\tmock := &MockDomainCRUD{ctrl: ctrl}\n\tmock.recorder = &MockDomainCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainCRUD) EXPECT() *MockDomainCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockDomainCRUD) DeleteDomain(ctx context.Context, domainID, domainName *string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockDomainCRUDMockRecorder) DeleteDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockDomainCRUD)(nil).DeleteDomain), ctx, domainID, domainName)\n}\n\n// InsertDomain mocks base method.\nfunc (m *MockDomainCRUD) InsertDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomain indicates an expected call of InsertDomain.\nfunc (mr *MockDomainCRUDMockRecorder) InsertDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomain\", reflect.TypeOf((*MockDomainCRUD)(nil).InsertDomain), ctx, row)\n}\n\n// SelectAllDomains mocks base method.\nfunc (m *MockDomainCRUD) SelectAllDomains(ctx context.Context, pageSize int, pageToken []byte) ([]*DomainRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllDomains\", ctx, pageSize, pageToken)\n\tret0, _ := ret[0].([]*DomainRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllDomains indicates an expected call of SelectAllDomains.\nfunc (mr *MockDomainCRUDMockRecorder) SelectAllDomains(ctx, pageSize, pageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllDomains\", reflect.TypeOf((*MockDomainCRUD)(nil).SelectAllDomains), ctx, pageSize, pageToken)\n}\n\n// SelectDomain mocks base method.\nfunc (m *MockDomainCRUD) SelectDomain(ctx context.Context, domainID, domainName *string) (*DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomain\", ctx, domainID, domainName)\n\tret0, _ := ret[0].(*DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomain indicates an expected call of SelectDomain.\nfunc (mr *MockDomainCRUDMockRecorder) SelectDomain(ctx, domainID, domainName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomain\", reflect.TypeOf((*MockDomainCRUD)(nil).SelectDomain), ctx, domainID, domainName)\n}\n\n// SelectDomainMetadata mocks base method.\nfunc (m *MockDomainCRUD) SelectDomainMetadata(ctx context.Context) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainMetadata\", ctx)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectDomainMetadata indicates an expected call of SelectDomainMetadata.\nfunc (mr *MockDomainCRUDMockRecorder) SelectDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainMetadata\", reflect.TypeOf((*MockDomainCRUD)(nil).SelectDomainMetadata), ctx)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockDomainCRUD) UpdateDomain(ctx context.Context, row *DomainRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockDomainCRUDMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockDomainCRUD)(nil).UpdateDomain), ctx, row)\n}\n\n// MockShardCRUD is a mock of ShardCRUD interface.\ntype MockShardCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardCRUDMockRecorder is the mock recorder for MockShardCRUD.\ntype MockShardCRUDMockRecorder struct {\n\tmock *MockShardCRUD\n}\n\n// NewMockShardCRUD creates a new mock instance.\nfunc NewMockShardCRUD(ctrl *gomock.Controller) *MockShardCRUD {\n\tmock := &MockShardCRUD{ctrl: ctrl}\n\tmock.recorder = &MockShardCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardCRUD) EXPECT() *MockShardCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// InsertShard mocks base method.\nfunc (m *MockShardCRUD) InsertShard(ctx context.Context, row *ShardRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertShard\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertShard indicates an expected call of InsertShard.\nfunc (mr *MockShardCRUDMockRecorder) InsertShard(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertShard\", reflect.TypeOf((*MockShardCRUD)(nil).InsertShard), ctx, row)\n}\n\n// SelectShard mocks base method.\nfunc (m *MockShardCRUD) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *ShardRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectShard\", ctx, shardID, currentClusterName)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(*ShardRow)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectShard indicates an expected call of SelectShard.\nfunc (mr *MockShardCRUDMockRecorder) SelectShard(ctx, shardID, currentClusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectShard\", reflect.TypeOf((*MockShardCRUD)(nil).SelectShard), ctx, shardID, currentClusterName)\n}\n\n// UpdateRangeID mocks base method.\nfunc (m *MockShardCRUD) UpdateRangeID(ctx context.Context, shardID int, rangeID, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateRangeID\", ctx, shardID, rangeID, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateRangeID indicates an expected call of UpdateRangeID.\nfunc (mr *MockShardCRUDMockRecorder) UpdateRangeID(ctx, shardID, rangeID, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateRangeID\", reflect.TypeOf((*MockShardCRUD)(nil).UpdateRangeID), ctx, shardID, rangeID, previousRangeID)\n}\n\n// UpdateShard mocks base method.\nfunc (m *MockShardCRUD) UpdateShard(ctx context.Context, row *ShardRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShard\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateShard indicates an expected call of UpdateShard.\nfunc (mr *MockShardCRUDMockRecorder) UpdateShard(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShard\", reflect.TypeOf((*MockShardCRUD)(nil).UpdateShard), ctx, row, previousRangeID)\n}\n\n// MockVisibilityCRUD is a mock of VisibilityCRUD interface.\ntype MockVisibilityCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVisibilityCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockVisibilityCRUDMockRecorder is the mock recorder for MockVisibilityCRUD.\ntype MockVisibilityCRUDMockRecorder struct {\n\tmock *MockVisibilityCRUD\n}\n\n// NewMockVisibilityCRUD creates a new mock instance.\nfunc NewMockVisibilityCRUD(ctrl *gomock.Controller) *MockVisibilityCRUD {\n\tmock := &MockVisibilityCRUD{ctrl: ctrl}\n\tmock.recorder = &MockVisibilityCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVisibilityCRUD) EXPECT() *MockVisibilityCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteVisibility mocks base method.\nfunc (m *MockVisibilityCRUD) DeleteVisibility(ctx context.Context, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteVisibility\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteVisibility indicates an expected call of DeleteVisibility.\nfunc (mr *MockVisibilityCRUDMockRecorder) DeleteVisibility(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteVisibility\", reflect.TypeOf((*MockVisibilityCRUD)(nil).DeleteVisibility), ctx, domainID, workflowID, runID)\n}\n\n// InsertVisibility mocks base method.\nfunc (m *MockVisibilityCRUD) InsertVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForInsert) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertVisibility indicates an expected call of InsertVisibility.\nfunc (mr *MockVisibilityCRUDMockRecorder) InsertVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertVisibility\", reflect.TypeOf((*MockVisibilityCRUD)(nil).InsertVisibility), ctx, ttlSeconds, row)\n}\n\n// SelectOneClosedWorkflow mocks base method.\nfunc (m *MockVisibilityCRUD) SelectOneClosedWorkflow(ctx context.Context, domainID, workflowID, runID string) (*VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectOneClosedWorkflow\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectOneClosedWorkflow indicates an expected call of SelectOneClosedWorkflow.\nfunc (mr *MockVisibilityCRUDMockRecorder) SelectOneClosedWorkflow(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectOneClosedWorkflow\", reflect.TypeOf((*MockVisibilityCRUD)(nil).SelectOneClosedWorkflow), ctx, domainID, workflowID, runID)\n}\n\n// SelectVisibility mocks base method.\nfunc (m *MockVisibilityCRUD) SelectVisibility(ctx context.Context, filter *VisibilityFilter) (*SelectVisibilityResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectVisibility\", ctx, filter)\n\tret0, _ := ret[0].(*SelectVisibilityResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectVisibility indicates an expected call of SelectVisibility.\nfunc (mr *MockVisibilityCRUDMockRecorder) SelectVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectVisibility\", reflect.TypeOf((*MockVisibilityCRUD)(nil).SelectVisibility), ctx, filter)\n}\n\n// UpdateVisibility mocks base method.\nfunc (m *MockVisibilityCRUD) UpdateVisibility(ctx context.Context, ttlSeconds int64, row *VisibilityRowForUpdate) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateVisibility\", ctx, ttlSeconds, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateVisibility indicates an expected call of UpdateVisibility.\nfunc (mr *MockVisibilityCRUDMockRecorder) UpdateVisibility(ctx, ttlSeconds, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateVisibility\", reflect.TypeOf((*MockVisibilityCRUD)(nil).UpdateVisibility), ctx, ttlSeconds, row)\n}\n\n// MockTaskCRUD is a mock of TaskCRUD interface.\ntype MockTaskCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskCRUDMockRecorder is the mock recorder for MockTaskCRUD.\ntype MockTaskCRUDMockRecorder struct {\n\tmock *MockTaskCRUD\n}\n\n// NewMockTaskCRUD creates a new mock instance.\nfunc NewMockTaskCRUD(ctrl *gomock.Controller) *MockTaskCRUD {\n\tmock := &MockTaskCRUD{ctrl: ctrl}\n\tmock.recorder = &MockTaskCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskCRUD) EXPECT() *MockTaskCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteTaskList mocks base method.\nfunc (m *MockTaskCRUD) DeleteTaskList(ctx context.Context, filter *TaskListFilter, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTaskList\", ctx, filter, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTaskList indicates an expected call of DeleteTaskList.\nfunc (mr *MockTaskCRUDMockRecorder) DeleteTaskList(ctx, filter, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTaskList\", reflect.TypeOf((*MockTaskCRUD)(nil).DeleteTaskList), ctx, filter, previousRangeID)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MockTaskCRUD) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MockTaskCRUDMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MockTaskCRUD)(nil).GetTasksCount), ctx, filter)\n}\n\n// InsertTaskList mocks base method.\nfunc (m *MockTaskCRUD) InsertTaskList(ctx context.Context, row *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTaskList\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTaskList indicates an expected call of InsertTaskList.\nfunc (mr *MockTaskCRUDMockRecorder) InsertTaskList(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTaskList\", reflect.TypeOf((*MockTaskCRUD)(nil).InsertTaskList), ctx, row)\n}\n\n// InsertTasks mocks base method.\nfunc (m *MockTaskCRUD) InsertTasks(ctx context.Context, tasksToInsert []*TaskRowForInsert, tasklistCondition *TaskListRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertTasks\", ctx, tasksToInsert, tasklistCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertTasks indicates an expected call of InsertTasks.\nfunc (mr *MockTaskCRUDMockRecorder) InsertTasks(ctx, tasksToInsert, tasklistCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertTasks\", reflect.TypeOf((*MockTaskCRUD)(nil).InsertTasks), ctx, tasksToInsert, tasklistCondition)\n}\n\n// ListTaskList mocks base method.\nfunc (m *MockTaskCRUD) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*ListTaskListResult, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskList\", ctx, pageSize, nextPageToken)\n\tret0, _ := ret[0].(*ListTaskListResult)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskList indicates an expected call of ListTaskList.\nfunc (mr *MockTaskCRUDMockRecorder) ListTaskList(ctx, pageSize, nextPageToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskList\", reflect.TypeOf((*MockTaskCRUD)(nil).ListTaskList), ctx, pageSize, nextPageToken)\n}\n\n// RangeDeleteTasks mocks base method.\nfunc (m *MockTaskCRUD) RangeDeleteTasks(ctx context.Context, filter *TasksFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTasks\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteTasks indicates an expected call of RangeDeleteTasks.\nfunc (mr *MockTaskCRUDMockRecorder) RangeDeleteTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTasks\", reflect.TypeOf((*MockTaskCRUD)(nil).RangeDeleteTasks), ctx, filter)\n}\n\n// SelectTaskList mocks base method.\nfunc (m *MockTaskCRUD) SelectTaskList(ctx context.Context, filter *TaskListFilter) (*TaskListRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTaskList\", ctx, filter)\n\tret0, _ := ret[0].(*TaskListRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTaskList indicates an expected call of SelectTaskList.\nfunc (mr *MockTaskCRUDMockRecorder) SelectTaskList(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTaskList\", reflect.TypeOf((*MockTaskCRUD)(nil).SelectTaskList), ctx, filter)\n}\n\n// SelectTasks mocks base method.\nfunc (m *MockTaskCRUD) SelectTasks(ctx context.Context, filter *TasksFilter) ([]*TaskRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTasks\", ctx, filter)\n\tret0, _ := ret[0].([]*TaskRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectTasks indicates an expected call of SelectTasks.\nfunc (mr *MockTaskCRUDMockRecorder) SelectTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTasks\", reflect.TypeOf((*MockTaskCRUD)(nil).SelectTasks), ctx, filter)\n}\n\n// UpdateTaskList mocks base method.\nfunc (m *MockTaskCRUD) UpdateTaskList(ctx context.Context, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskList\", ctx, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskList indicates an expected call of UpdateTaskList.\nfunc (mr *MockTaskCRUDMockRecorder) UpdateTaskList(ctx, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskList\", reflect.TypeOf((*MockTaskCRUD)(nil).UpdateTaskList), ctx, row, previousRangeID)\n}\n\n// UpdateTaskListWithTTL mocks base method.\nfunc (m *MockTaskCRUD) UpdateTaskListWithTTL(ctx context.Context, ttlSeconds int64, row *TaskListRow, previousRangeID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListWithTTL\", ctx, ttlSeconds, row, previousRangeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskListWithTTL indicates an expected call of UpdateTaskListWithTTL.\nfunc (mr *MockTaskCRUDMockRecorder) UpdateTaskListWithTTL(ctx, ttlSeconds, row, previousRangeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListWithTTL\", reflect.TypeOf((*MockTaskCRUD)(nil).UpdateTaskListWithTTL), ctx, ttlSeconds, row, previousRangeID)\n}\n\n// MockWorkflowCRUD is a mock of WorkflowCRUD interface.\ntype MockWorkflowCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWorkflowCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockWorkflowCRUDMockRecorder is the mock recorder for MockWorkflowCRUD.\ntype MockWorkflowCRUDMockRecorder struct {\n\tmock *MockWorkflowCRUD\n}\n\n// NewMockWorkflowCRUD creates a new mock instance.\nfunc NewMockWorkflowCRUD(ctrl *gomock.Controller) *MockWorkflowCRUD {\n\tmock := &MockWorkflowCRUD{ctrl: ctrl}\n\tmock.recorder = &MockWorkflowCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWorkflowCRUD) EXPECT() *MockWorkflowCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActiveClusterSelectionPolicy\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActiveClusterSelectionPolicy indicates an expected call of DeleteActiveClusterSelectionPolicy.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteActiveClusterSelectionPolicy(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActiveClusterSelectionPolicy\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteActiveClusterSelectionPolicy), ctx, shardID, domainID, workflowID, runID)\n}\n\n// DeleteCrossClusterTask mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCrossClusterTask\", ctx, shardID, targetCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCrossClusterTask indicates an expected call of DeleteCrossClusterTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteCrossClusterTask(ctx, shardID, targetCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCrossClusterTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteCrossClusterTask), ctx, shardID, targetCluster, taskID)\n}\n\n// DeleteCurrentWorkflow mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflow\", ctx, shardID, domainID, workflowID, currentRunIDCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflow indicates an expected call of DeleteCurrentWorkflow.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteCurrentWorkflow(ctx, shardID, domainID, workflowID, currentRunIDCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflow\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteCurrentWorkflow), ctx, shardID, domainID, workflowID, currentRunIDCondition)\n}\n\n// DeleteReplicationDLQTask mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationDLQTask\", ctx, shardID, sourceCluster, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationDLQTask indicates an expected call of DeleteReplicationDLQTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteReplicationDLQTask(ctx, shardID, sourceCluster, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationDLQTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteReplicationDLQTask), ctx, shardID, sourceCluster, taskID)\n}\n\n// DeleteReplicationTask mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteReplicationTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteReplicationTask indicates an expected call of DeleteReplicationTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteReplicationTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteReplicationTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteReplicationTask), ctx, shardID, taskID)\n}\n\n// DeleteTimerTask mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTimerTask\", ctx, shardID, taskID, visibilityTimestamp)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTimerTask indicates an expected call of DeleteTimerTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteTimerTask(ctx, shardID, taskID, visibilityTimestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTimerTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteTimerTask), ctx, shardID, taskID, visibilityTimestamp)\n}\n\n// DeleteTransferTask mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteTransferTask\", ctx, shardID, taskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteTransferTask indicates an expected call of DeleteTransferTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteTransferTask(ctx, shardID, taskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTransferTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteTransferTask), ctx, shardID, taskID)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockWorkflowCRUD) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockWorkflowCRUDMockRecorder) DeleteWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockWorkflowCRUD)(nil).DeleteWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// InsertReplicationDLQTask mocks base method.\nfunc (m *MockWorkflowCRUD) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *HistoryMigrationTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationDLQTask\", ctx, shardID, sourceCluster, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationDLQTask indicates an expected call of InsertReplicationDLQTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) InsertReplicationDLQTask(ctx, shardID, sourceCluster, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationDLQTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).InsertReplicationDLQTask), ctx, shardID, sourceCluster, task)\n}\n\n// InsertReplicationTask mocks base method.\nfunc (m *MockWorkflowCRUD) InsertReplicationTask(ctx context.Context, tasks []*HistoryMigrationTask, condition ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertReplicationTask\", ctx, tasks, condition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertReplicationTask indicates an expected call of InsertReplicationTask.\nfunc (mr *MockWorkflowCRUDMockRecorder) InsertReplicationTask(ctx, tasks, condition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertReplicationTask\", reflect.TypeOf((*MockWorkflowCRUD)(nil).InsertReplicationTask), ctx, tasks, condition)\n}\n\n// InsertWorkflowExecutionWithTasks mocks base method.\nfunc (m *MockWorkflowCRUD) InsertWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, execution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertWorkflowExecutionWithTasks indicates an expected call of InsertWorkflowExecutionWithTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) InsertWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertWorkflowExecutionWithTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).InsertWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, execution, tasksByCategory, activeClusterSelectionPolicyRow, shardCondition)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MockWorkflowCRUD) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MockWorkflowCRUDMockRecorder) IsWorkflowExecutionExists(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MockWorkflowCRUD)(nil).IsWorkflowExecutionExists), ctx, shardID, domainID, workflowID, runID)\n}\n\n// RangeDeleteReplicationDLQTasks mocks base method.\nfunc (m *MockWorkflowCRUD) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationDLQTasks\", ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationDLQTasks indicates an expected call of RangeDeleteReplicationDLQTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) RangeDeleteReplicationDLQTasks(ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationDLQTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).RangeDeleteReplicationDLQTasks), ctx, shardID, sourceCluster, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// RangeDeleteReplicationTasks mocks base method.\nfunc (m *MockWorkflowCRUD) RangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteReplicationTasks\", ctx, shardID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteReplicationTasks indicates an expected call of RangeDeleteReplicationTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) RangeDeleteReplicationTasks(ctx, shardID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteReplicationTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).RangeDeleteReplicationTasks), ctx, shardID, exclusiveEndTaskID)\n}\n\n// RangeDeleteTimerTasks mocks base method.\nfunc (m *MockWorkflowCRUD) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTimerTasks\", ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTimerTasks indicates an expected call of RangeDeleteTimerTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) RangeDeleteTimerTasks(ctx, shardID, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTimerTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).RangeDeleteTimerTasks), ctx, shardID, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// RangeDeleteTransferTasks mocks base method.\nfunc (m *MockWorkflowCRUD) RangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteTransferTasks\", ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RangeDeleteTransferTasks indicates an expected call of RangeDeleteTransferTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) RangeDeleteTransferTasks(ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteTransferTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).RangeDeleteTransferTasks), ctx, shardID, inclusiveBeginTaskID, exclusiveEndTaskID)\n}\n\n// SelectActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockWorkflowCRUD) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*ActiveClusterSelectionPolicyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectActiveClusterSelectionPolicy\", ctx, shardID, domainID, wfID, rID)\n\tret0, _ := ret[0].(*ActiveClusterSelectionPolicyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectActiveClusterSelectionPolicy indicates an expected call of SelectActiveClusterSelectionPolicy.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectActiveClusterSelectionPolicy(ctx, shardID, domainID, wfID, rID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectActiveClusterSelectionPolicy\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectActiveClusterSelectionPolicy), ctx, shardID, domainID, wfID, rID)\n}\n\n// SelectAllCurrentWorkflows mocks base method.\nfunc (m *MockWorkflowCRUD) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllCurrentWorkflows\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.CurrentWorkflowExecution)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllCurrentWorkflows indicates an expected call of SelectAllCurrentWorkflows.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectAllCurrentWorkflows(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllCurrentWorkflows\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectAllCurrentWorkflows), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectAllWorkflowExecutions mocks base method.\nfunc (m *MockWorkflowCRUD) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectAllWorkflowExecutions\", ctx, shardID, pageToken, pageSize)\n\tret0, _ := ret[0].([]*persistence.InternalListConcreteExecutionsEntity)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectAllWorkflowExecutions indicates an expected call of SelectAllWorkflowExecutions.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectAllWorkflowExecutions(ctx, shardID, pageToken, pageSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectAllWorkflowExecutions\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectAllWorkflowExecutions), ctx, shardID, pageToken, pageSize)\n}\n\n// SelectCurrentWorkflow mocks base method.\nfunc (m *MockWorkflowCRUD) SelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*CurrentWorkflowRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectCurrentWorkflow\", ctx, shardID, domainID, workflowID)\n\tret0, _ := ret[0].(*CurrentWorkflowRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectCurrentWorkflow indicates an expected call of SelectCurrentWorkflow.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectCurrentWorkflow(ctx, shardID, domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectCurrentWorkflow\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectCurrentWorkflow), ctx, shardID, domainID, workflowID)\n}\n\n// SelectReplicationDLQTasksCount mocks base method.\nfunc (m *MockWorkflowCRUD) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksCount\", ctx, shardID, sourceCluster)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectReplicationDLQTasksCount indicates an expected call of SelectReplicationDLQTasksCount.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectReplicationDLQTasksCount(ctx, shardID, sourceCluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksCount\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectReplicationDLQTasksCount), ctx, shardID, sourceCluster)\n}\n\n// SelectReplicationDLQTasksOrderByTaskID mocks base method.\nfunc (m *MockWorkflowCRUD) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationDLQTasksOrderByTaskID\", ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationDLQTasksOrderByTaskID indicates an expected call of SelectReplicationDLQTasksOrderByTaskID.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectReplicationDLQTasksOrderByTaskID(ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationDLQTasksOrderByTaskID\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectReplicationDLQTasksOrderByTaskID), ctx, shardID, sourceCluster, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectReplicationTasksOrderByTaskID mocks base method.\nfunc (m *MockWorkflowCRUD) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectReplicationTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectReplicationTasksOrderByTaskID indicates an expected call of SelectReplicationTasksOrderByTaskID.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectReplicationTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectReplicationTasksOrderByTaskID\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectReplicationTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectTimerTasksOrderByVisibilityTime mocks base method.\nfunc (m *MockWorkflowCRUD) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTimerTasksOrderByVisibilityTime\", ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTimerTasksOrderByVisibilityTime indicates an expected call of SelectTimerTasksOrderByVisibilityTime.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectTimerTasksOrderByVisibilityTime(ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTimerTasksOrderByVisibilityTime\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectTimerTasksOrderByVisibilityTime), ctx, shardID, pageSize, pageToken, inclusiveMinTime, exclusiveMaxTime)\n}\n\n// SelectTransferTasksOrderByTaskID mocks base method.\nfunc (m *MockWorkflowCRUD) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*HistoryMigrationTask, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectTransferTasksOrderByTaskID\", ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n\tret0, _ := ret[0].([]*HistoryMigrationTask)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectTransferTasksOrderByTaskID indicates an expected call of SelectTransferTasksOrderByTaskID.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectTransferTasksOrderByTaskID(ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectTransferTasksOrderByTaskID\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectTransferTasksOrderByTaskID), ctx, shardID, pageSize, pageToken, inclusiveMinTaskID, exclusiveMaxTaskID)\n}\n\n// SelectWorkflowExecution mocks base method.\nfunc (m *MockWorkflowCRUD) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*WorkflowExecution, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectWorkflowExecution\", ctx, shardID, domainID, workflowID, runID)\n\tret0, _ := ret[0].(*WorkflowExecution)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectWorkflowExecution indicates an expected call of SelectWorkflowExecution.\nfunc (mr *MockWorkflowCRUDMockRecorder) SelectWorkflowExecution(ctx, shardID, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectWorkflowExecution\", reflect.TypeOf((*MockWorkflowCRUD)(nil).SelectWorkflowExecution), ctx, shardID, domainID, workflowID, runID)\n}\n\n// UpdateWorkflowExecutionWithTasks mocks base method.\nfunc (m *MockWorkflowCRUD) UpdateWorkflowExecutionWithTasks(ctx context.Context, requests *WorkflowRequestsWriteRequest, currentWorkflowRequest *CurrentWorkflowWriteRequest, mutatedExecution, insertedExecution *WorkflowExecutionRequest, activeClusterSelectionPolicyRow *ActiveClusterSelectionPolicyRow, resetExecution *WorkflowExecutionRequest, tasksByCategory map[persistence.HistoryTaskCategory][]*HistoryMigrationTask, shardCondition *ShardCondition) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithTasks\", ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithTasks indicates an expected call of UpdateWorkflowExecutionWithTasks.\nfunc (mr *MockWorkflowCRUDMockRecorder) UpdateWorkflowExecutionWithTasks(ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithTasks\", reflect.TypeOf((*MockWorkflowCRUD)(nil).UpdateWorkflowExecutionWithTasks), ctx, requests, currentWorkflowRequest, mutatedExecution, insertedExecution, activeClusterSelectionPolicyRow, resetExecution, tasksByCategory, shardCondition)\n}\n\n// MockConfigStoreCRUD is a mock of ConfigStoreCRUD interface.\ntype MockConfigStoreCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockConfigStoreCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockConfigStoreCRUDMockRecorder is the mock recorder for MockConfigStoreCRUD.\ntype MockConfigStoreCRUDMockRecorder struct {\n\tmock *MockConfigStoreCRUD\n}\n\n// NewMockConfigStoreCRUD creates a new mock instance.\nfunc NewMockConfigStoreCRUD(ctrl *gomock.Controller) *MockConfigStoreCRUD {\n\tmock := &MockConfigStoreCRUD{ctrl: ctrl}\n\tmock.recorder = &MockConfigStoreCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockConfigStoreCRUD) EXPECT() *MockConfigStoreCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// InsertConfig mocks base method.\nfunc (m *MockConfigStoreCRUD) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MockConfigStoreCRUDMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MockConfigStoreCRUD)(nil).InsertConfig), ctx, row)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MockConfigStoreCRUD) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MockConfigStoreCRUDMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MockConfigStoreCRUD)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// MockDomainAuditLogCRUD is a mock of DomainAuditLogCRUD interface.\ntype MockDomainAuditLogCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDomainAuditLogCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockDomainAuditLogCRUDMockRecorder is the mock recorder for MockDomainAuditLogCRUD.\ntype MockDomainAuditLogCRUDMockRecorder struct {\n\tmock *MockDomainAuditLogCRUD\n}\n\n// NewMockDomainAuditLogCRUD creates a new mock instance.\nfunc NewMockDomainAuditLogCRUD(ctrl *gomock.Controller) *MockDomainAuditLogCRUD {\n\tmock := &MockDomainAuditLogCRUD{ctrl: ctrl}\n\tmock.recorder = &MockDomainAuditLogCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDomainAuditLogCRUD) EXPECT() *MockDomainAuditLogCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// InsertDomainAuditLog mocks base method.\nfunc (m *MockDomainAuditLogCRUD) InsertDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertDomainAuditLog indicates an expected call of InsertDomainAuditLog.\nfunc (mr *MockDomainAuditLogCRUDMockRecorder) InsertDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertDomainAuditLog\", reflect.TypeOf((*MockDomainAuditLogCRUD)(nil).InsertDomainAuditLog), ctx, row)\n}\n\n// SelectDomainAuditLogs mocks base method.\nfunc (m *MockDomainAuditLogCRUD) SelectDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, []byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].([]byte)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SelectDomainAuditLogs indicates an expected call of SelectDomainAuditLogs.\nfunc (mr *MockDomainAuditLogCRUDMockRecorder) SelectDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectDomainAuditLogs\", reflect.TypeOf((*MockDomainAuditLogCRUD)(nil).SelectDomainAuditLogs), ctx, filter)\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/admin.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\n\t\"go.mongodb.org/mongo-driver/bson\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nvar _ nosqlplugin.AdminDB = (*mdb)(nil)\n\nconst (\n\ttestSchemaDir = \"schema/mongodb/\"\n)\n\nfunc (db *mdb) SetupTestDatabase(schemaBaseDir string, replicas int) error {\n\tif schemaBaseDir == \"\" {\n\t\tvar err error\n\t\tschemaBaseDir, err = nosqlplugin.GetDefaultTestSchemaDir(testSchemaDir)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tschemaFile := schemaBaseDir + \"cadence/schema.json\"\n\tbyteValues, err := ioutil.ReadFile(schemaFile)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar commands []interface{}\n\terr = bson.UnmarshalExtJSON(byteValues, false, &commands)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, cmd := range commands {\n\t\tresult := db.dbConn.RunCommand(context.Background(), cmd)\n\t\tif result.Err() != nil {\n\t\t\treturn result.Err()\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (db *mdb) TeardownTestDatabase() error {\n\tresult := db.dbConn.RunCommand(context.Background(), bson.D{{\"dropDatabase\", 1}})\n\terr := result.Err()\n\treturn err\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/configStore.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.mongodb.org/mongo-driver/bson\"\n\t\"go.mongodb.org/mongo-driver/mongo\"\n\t\"go.mongodb.org/mongo-driver/mongo/options\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/schema/mongodb/cadence\"\n)\n\nfunc (db *mdb) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tcollection := db.dbConn.Collection(cadence.ClusterConfigCollectionName)\n\tdoc := cadence.ClusterConfigCollectionEntry{\n\t\tRowType:              row.RowType,\n\t\tVersion:              row.Version,\n\t\tUnixTimestampSeconds: row.Timestamp.Unix(),\n\t\tData:                 row.Values.Data,\n\t\tDataEncoding:         row.Values.GetEncodingString(),\n\t}\n\t_, err := collection.InsertOne(ctx, doc)\n\tif mongo.IsDuplicateKeyError(err) {\n\t\treturn nosqlplugin.NewConditionFailure(\"InsertConfig operation failed because of version collision\")\n\t}\n\treturn err\n}\n\nfunc (db *mdb) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tfilter := bson.D{{\"rowtype\", rowType}}\n\tqueryOptions := options.FindOneOptions{}\n\tqueryOptions.SetSort(bson.D{{\"version\", -1}})\n\n\tcollection := db.dbConn.Collection(cadence.ClusterConfigCollectionName)\n\tvar result cadence.ClusterConfigCollectionEntry\n\terr := collection.FindOne(ctx, filter, &queryOptions).Decode(&result)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &persistence.InternalConfigStoreEntry{\n\t\tRowType:   rowType,\n\t\tVersion:   result.Version,\n\t\tTimestamp: time.Unix(result.UnixTimestampSeconds, 0),\n\t\tValues:    persistence.NewDataBlob(result.Data, constants.EncodingType(result.DataEncoding)),\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/db.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\n\t\"go.mongodb.org/mongo-driver/mongo\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// mdb represents a logical connection to MongoDB database\ntype mdb struct {\n\tclient *mongo.Client\n\tdbConn *mongo.Database\n\tcfg    *config.NoSQL\n\tlogger log.Logger\n}\n\nvar _ nosqlplugin.DB = (*mdb)(nil)\n\nfunc (db *mdb) Close() {\n\tdb.client.Disconnect(context.Background())\n}\n\nfunc (db *mdb) PluginName() string {\n\treturn PluginName\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/domain.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Insert a new record to domain, return error if failed or already exists\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *mdb) InsertDomain(\n\tctx context.Context,\n\trow *nosqlplugin.DomainRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Update domain\nfunc (db *mdb) UpdateDomain(\n\tctx context.Context,\n\trow *nosqlplugin.DomainRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Get one domain data, either by domainID or domainName\nfunc (db *mdb) SelectDomain(\n\tctx context.Context,\n\tdomainID *string,\n\tdomainName *string,\n) (*nosqlplugin.DomainRow, error) {\n\tpanic(\"TODO\")\n}\n\n// Get all domain data\nfunc (db *mdb) SelectAllDomains(\n\tctx context.Context,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*nosqlplugin.DomainRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// Delete a domain, either by domainID or domainName\nfunc (db *mdb) DeleteDomain(\n\tctx context.Context,\n\tdomainID *string,\n\tdomainName *string,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectDomainMetadata(\n\tctx context.Context,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/domain_audit_log.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertDomainAuditLog inserts a new audit log entry for a domain operation\nfunc (db *mdb) InsertDomainAuditLog(ctx context.Context, row *nosqlplugin.DomainAuditLogRow) error {\n\treturn fmt.Errorf(\"InsertDomainAuditLog not implemented\")\n}\n\n// SelectDomainAuditLogs returns audit log entries for a domain and operation type\nfunc (db *mdb) SelectDomainAuditLogs(ctx context.Context, filter *nosqlplugin.DomainAuditLogFilter) ([]*nosqlplugin.DomainAuditLogRow, []byte, error) {\n\treturn nil, nil, fmt.Errorf(\"SelectDomainAuditLogs not implemented\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/error.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport \"go.mongodb.org/mongo-driver/mongo\"\n\nfunc (db *mdb) IsNotFoundError(err error) bool {\n\treturn err == mongo.ErrNoDocuments\n}\n\nfunc (db *mdb) IsTimeoutError(err error) bool {\n\treturn mongo.IsTimeout(err) || mongo.IsNetworkError(err)\n}\n\nfunc (db *mdb) IsThrottlingError(err error) bool {\n\treturn false\n}\n\nfunc (db *mdb) IsDBUnavailableError(err error) bool {\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/events.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertIntoHistoryTreeAndNode inserts one or two rows: tree row and node row(at least one of them)\nfunc (db *mdb) InsertIntoHistoryTreeAndNode(ctx context.Context, treeRow *nosqlplugin.HistoryTreeRow, nodeRow *nosqlplugin.HistoryNodeRow) error {\n\tpanic(\"TODO\")\n}\n\n// SelectFromHistoryNode read nodes based on a filter\nfunc (db *mdb) SelectFromHistoryNode(ctx context.Context, filter *nosqlplugin.HistoryNodeFilter) ([]*nosqlplugin.HistoryNodeRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteFromHistoryTreeAndNode delete a branch record, and a list of ranges of nodes.\nfunc (db *mdb) DeleteFromHistoryTreeAndNode(ctx context.Context, treeFilter *nosqlplugin.HistoryTreeFilter, nodeFilters []*nosqlplugin.HistoryNodeFilter) error {\n\tpanic(\"TODO\")\n}\n\n// SelectAllHistoryTrees will return all tree branches with pagination\nfunc (db *mdb) SelectAllHistoryTrees(ctx context.Context, nextPageToken []byte, pageSize int) ([]*nosqlplugin.HistoryTreeRow, []byte, error) {\n\tpanic(\"TODO\")\n}\n\n// SelectFromHistoryTree read branch records for a tree\nfunc (db *mdb) SelectFromHistoryTree(ctx context.Context, filter *nosqlplugin.HistoryTreeFilter) ([]*nosqlplugin.HistoryTreeRow, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/plugin.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"go.mongodb.org/mongo-driver/mongo\"\n\t\"go.mongodb.org/mongo-driver/mongo/options\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nconst (\n\t// PluginName is the name of the plugin\n\tPluginName = \"mongodb\"\n)\n\ntype plugin struct{}\n\nvar _ nosqlplugin.Plugin = (*plugin)(nil)\n\nfunc init() {\n\tnosql.RegisterPlugin(PluginName, &plugin{})\n}\n\n// CreateDB initialize the db object\nfunc (p *plugin) CreateDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.DB, error) {\n\treturn p.doCreateDB(cfg, logger)\n}\n\n// CreateAdminDB initialize the AdminDB object\nfunc (p *plugin) CreateAdminDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.AdminDB, error) {\n\treturn p.doCreateDB(cfg, logger)\n}\n\nfunc (p *plugin) doCreateDB(cfg *config.NoSQL, logger log.Logger) (*mdb, error) {\n\turi := fmt.Sprintf(\"mongodb://%v:%v@%v:%v/\", cfg.User, cfg.Password, cfg.Hosts, cfg.Port)\n\t// TODO CreateDB/CreateAdminDB don't pass in context.Context so we are using background for now\n\t// It's okay because this is being called during server startup or CLI.\n\tclient, err := mongo.Connect(context.Background(), options.Client().ApplyURI(uri))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cfg.Keyspace == \"\" {\n\t\treturn nil, fmt.Errorf(\"database name cannot be empty\")\n\t}\n\tdb := client.Database(cfg.Keyspace)\n\treturn &mdb{\n\t\tclient: client,\n\t\tdbConn: db,\n\t\tcfg:    cfg,\n\t\tlogger: logger,\n\t}, err\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// Insert message into queue, return error if failed or already exists\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *mdb) InsertIntoQueue(\n\tctx context.Context,\n\trow *nosqlplugin.QueueMessageRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Get the ID of last message inserted into the queue\nfunc (db *mdb) SelectLastEnqueuedMessageID(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n\n// Read queue messages starting from the exclusiveBeginMessageID\nfunc (db *mdb) SelectMessagesFrom(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tmaxRows int,\n) ([]*nosqlplugin.QueueMessageRow, error) {\n\tpanic(\"TODO\")\n}\n\n// Read queue message starting from exclusiveBeginMessageID int64, inclusiveEndMessageID int64\nfunc (db *mdb) SelectMessagesBetween(\n\tctx context.Context,\n\trequest nosqlplugin.SelectMessagesBetweenRequest,\n) (*nosqlplugin.SelectMessagesBetweenResponse, error) {\n\tpanic(\"TODO\")\n}\n\n// Delete all messages before exclusiveBeginMessageID\nfunc (db *mdb) DeleteMessagesBefore(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Delete all messages in a range between exclusiveBeginMessageID and inclusiveEndMessageID\nfunc (db *mdb) DeleteMessagesInRange(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tinclusiveEndMessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Delete one message\nfunc (db *mdb) DeleteMessage(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Insert an empty metadata row, starting from a version\nfunc (db *mdb) InsertQueueMetadata(ctx context.Context, row nosqlplugin.QueueMetadataRow) error {\n\tfmt.Println(\"not implemented, ignore the eror for testing\")\n\treturn nil\n}\n\n// **Conditionally** update a queue metadata row, if current version is matched(meaning current == row.Version - 1),\n// then the current version will increase by one when updating the metadata row\n// Return ConditionFailure if the condition doesn't meet\nfunc (db *mdb) UpdateQueueMetadataCas(\n\tctx context.Context,\n\trow nosqlplugin.QueueMetadataRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// Read a QueueMetadata\nfunc (db *mdb) SelectQueueMetadata(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (*nosqlplugin.QueueMetadataRow, error) {\n\tfmt.Println(\"not implemented, ignore the eror for testing\")\n\treturn nil, nil\n}\n\nfunc (db *mdb) GetQueueSize(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/shard.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"log\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// InsertShard creates a new shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) InsertShard(ctx context.Context, row *nosqlplugin.ShardRow) error {\n\tlog.Println(\"not implemented...ignore the error for testing...\")\n\treturn nil\n}\n\n// SelectShard gets a shard\nfunc (db *mdb) SelectShard(ctx context.Context, shardID int, currentClusterName string) (int64, *nosqlplugin.ShardRow, error) {\n\tpanic(\"TODO\")\n}\n\n// UpdateRangeID updates the rangeID, return error is there is any\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) UpdateRangeID(ctx context.Context, shardID int, rangeID int64, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateShard updates a shard, return error is there is any.\n// Return ShardOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) UpdateShard(ctx context.Context, row *nosqlplugin.ShardRow, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/task.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\n// SelectTaskList returns a single tasklist row.\n// Return IsNotFoundError if the row doesn't exist\nfunc (db *mdb) SelectTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter) (*nosqlplugin.TaskListRow, error) {\n\tpanic(\"TODO\")\n}\n\n// InsertTaskList insert a single tasklist row\n// Return IsConditionFailedError if the row already exists, and also the existing row\nfunc (db *mdb) InsertTaskList(ctx context.Context, row *nosqlplugin.TaskListRow) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateTaskList updates a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) UpdateTaskList(\n\tctx context.Context,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// UpdateTaskList updates a single tasklist row, and set an TTL on the record\n// Return TaskOperationConditionFailure if the condition doesn't meet\n// Ignore TTL if it's not supported, which becomes exactly the same as UpdateTaskList, but ListTaskList must be\n// implemented for TaskListScavenger\nfunc (db *mdb) UpdateTaskListWithTTL(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.TaskListRow,\n\tpreviousRangeID int64,\n) error {\n\tpanic(\"TODO\")\n}\n\n// ListTaskList returns all tasklists.\n// Noop if TTL is already implemented in other methods\nfunc (db *mdb) ListTaskList(ctx context.Context, pageSize int, nextPageToken []byte) (*nosqlplugin.ListTaskListResult, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteTaskList deletes a single tasklist row\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) DeleteTaskList(ctx context.Context, filter *nosqlplugin.TaskListFilter, previousRangeID int64) error {\n\tpanic(\"TODO\")\n}\n\n// InsertTasks inserts a batch of tasks\n// Return TaskOperationConditionFailure if the condition doesn't meet\nfunc (db *mdb) InsertTasks(\n\tctx context.Context,\n\ttasksToInsert []*nosqlplugin.TaskRowForInsert,\n\ttasklistCondition *nosqlplugin.TaskListRow,\n) error {\n\tpanic(\"TODO\")\n}\n\n// SelectTasks return tasks that associated to a tasklist\nfunc (db *mdb) SelectTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) ([]*nosqlplugin.TaskRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) GetTasksCount(ctx context.Context, filter *nosqlplugin.TasksFilter) (int64, error) {\n\tpanic(\"TODO\")\n}\n\n// DeleteTask delete a batch tasks that taskIDs less than the row\n// If TTL is not implemented, then should also return the number of rows deleted, otherwise persistence.UnknownNumRowsAffected\n// NOTE: This API ignores the `BatchSize` request parameter i.e. either all tasks leq the task_id will be deleted or an error will\n// be returned to the caller, because rowsDeleted is not supported by Cassandra\nfunc (db *mdb) RangeDeleteTasks(ctx context.Context, filter *nosqlplugin.TasksFilter) (rowsDeleted int, err error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/visibility.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nfunc (db *mdb) InsertVisibility(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.VisibilityRowForInsert,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) UpdateVisibility(\n\tctx context.Context,\n\tttlSeconds int64,\n\trow *nosqlplugin.VisibilityRowForUpdate,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectVisibility(\n\tctx context.Context,\n\tfilter *nosqlplugin.VisibilityFilter,\n) (*nosqlplugin.SelectVisibilityResponse, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteVisibility(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectOneClosedWorkflow(\n\tctx context.Context,\n\tdomainID, workflowID, runID string,\n) (*nosqlplugin.VisibilityRow, error) {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/mongodb/workflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nvar _ nosqlplugin.WorkflowCRUD = (*mdb)(nil)\n\nfunc (db *mdb) InsertWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\texecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) UpdateWorkflowExecutionWithTasks(\n\tctx context.Context,\n\trequests *nosqlplugin.WorkflowRequestsWriteRequest,\n\tcurrentWorkflowRequest *nosqlplugin.CurrentWorkflowWriteRequest,\n\tmutatedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tinsertedExecution *nosqlplugin.WorkflowExecutionRequest,\n\tactiveClusterSelectionPolicyRow *nosqlplugin.ActiveClusterSelectionPolicyRow,\n\tresetExecution *nosqlplugin.WorkflowExecutionRequest,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]*nosqlplugin.HistoryMigrationTask,\n\tshardCondition *nosqlplugin.ShardCondition,\n) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID string) (*nosqlplugin.CurrentWorkflowRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) (*nosqlplugin.WorkflowExecution, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteCurrentWorkflow(ctx context.Context, shardID int, domainID, workflowID, currentRunIDCondition string) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteWorkflowExecution(ctx context.Context, shardID int, domainID, workflowID, runID string) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectAllCurrentWorkflows(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.CurrentWorkflowExecution, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectAllWorkflowExecutions(ctx context.Context, shardID int, pageToken []byte, pageSize int) ([]*persistence.InternalListConcreteExecutionsEntity, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) IsWorkflowExecutionExists(ctx context.Context, shardID int, domainID, workflowID, runID string) (bool, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectTransferTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteTransferTask(ctx context.Context, shardID int, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) RangeDeleteTransferTasks(ctx context.Context, shardID int, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectTimerTasksOrderByVisibilityTime(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTime, exclusiveMaxTime time.Time) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteTimerTask(ctx context.Context, shardID int, taskID int64, visibilityTimestamp time.Time) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) RangeDeleteTimerTasks(ctx context.Context, shardID int, inclusiveMinTime, exclusiveMaxTime time.Time) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectReplicationTasksOrderByTaskID(ctx context.Context, shardID, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteReplicationTask(ctx context.Context, shardID int, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) RangeDeleteReplicationTasks(ctx context.Context, shardID int, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) InsertReplicationTask(ctx context.Context, tasks []*nosqlplugin.HistoryMigrationTask, condition nosqlplugin.ShardCondition) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteCrossClusterTask(ctx context.Context, shardID int, targetCluster string, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) InsertReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, task *nosqlplugin.HistoryMigrationTask) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectReplicationDLQTasksOrderByTaskID(ctx context.Context, shardID int, sourceCluster string, pageSize int, pageToken []byte, inclusiveMinTaskID, exclusiveMaxTaskID int64) ([]*nosqlplugin.HistoryMigrationTask, []byte, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectReplicationDLQTasksCount(ctx context.Context, shardID int, sourceCluster string) (int64, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteReplicationDLQTask(ctx context.Context, shardID int, sourceCluster string, taskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) RangeDeleteReplicationDLQTasks(ctx context.Context, shardID int, sourceCluster string, inclusiveBeginTaskID, exclusiveEndTaskID int64) error {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) SelectActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) (*nosqlplugin.ActiveClusterSelectionPolicyRow, error) {\n\tpanic(\"TODO\")\n}\n\nfunc (db *mdb) DeleteActiveClusterSelectionPolicy(ctx context.Context, shardID int, domainID, wfID, rID string) error {\n\tpanic(\"TODO\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/nosqlplugin/types.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosqlplugin\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// WorkflowExecution stores workflow execution metadata\n\tWorkflowExecution = persistence.InternalWorkflowMutableState\n\n\t// WorkflowExecutionRequest is for creating/updating a workflow execution\n\tWorkflowExecutionRequest struct {\n\t\t// basic information/data\n\t\tpersistence.InternalWorkflowExecutionInfo\n\t\tVersionHistories *persistence.DataBlob\n\t\tChecksums        *checksum.Checksum\n\t\tLastWriteVersion int64\n\t\tCurrentTimeStamp time.Time\n\t\t// condition checking for updating execution info\n\t\tPreviousNextEventIDCondition *int64\n\n\t\t// MapsWriteMode controls how to write into the six maps(activityInfoMap, timerInfoMap, childWorkflowInfoMap, signalInfoMap and signalRequestedIDs)\n\t\tMapsWriteMode WorkflowExecutionMapsWriteMode\n\n\t\t// For WorkflowExecutionMapsWriteMode of create, update and reset\n\t\tActivityInfos      map[int64]*persistence.InternalActivityInfo\n\t\tTimerInfos         map[string]*persistence.TimerInfo\n\t\tChildWorkflowInfos map[int64]*persistence.InternalChildExecutionInfo\n\t\tRequestCancelInfos map[int64]*persistence.RequestCancelInfo\n\t\tSignalInfos        map[int64]*persistence.SignalInfo\n\t\tSignalRequestedIDs []string // This map has no value, hence use array to store keys\n\n\t\t// For WorkflowExecutionMapsWriteMode of update only\n\t\tActivityInfoKeysToDelete       []int64\n\t\tTimerInfoKeysToDelete          []string\n\t\tChildWorkflowInfoKeysToDelete  []int64\n\t\tRequestCancelInfoKeysToDelete  []int64\n\t\tSignalInfoKeysToDelete         []int64\n\t\tSignalRequestedIDsKeysToDelete []string\n\n\t\t// EventBufferWriteMode controls how to write into the buffered event list\n\t\t// only needed for UpdateWorkflowExecutionWithTasks API\n\t\tEventBufferWriteMode EventBufferWriteMode\n\t\t// the batch of event to be appended, only for EventBufferWriteModeAppend\n\t\tNewBufferedEventBatch *persistence.DataBlob\n\t}\n\n\t// WorkflowExecutionMapsWriteMode controls how to write WorkflowExecutionMaps\n\tWorkflowExecutionMapsWriteMode int\n\t// EventBufferWriteMode controls how to write EventBuffer\n\tEventBufferWriteMode int\n\n\t// TimerTask is background timer task\n\tTimerTask = persistence.TimerTaskInfo\n\n\t// ReplicationTask is for replication\n\tReplicationTask = persistence.InternalReplicationTaskInfo\n\n\t// CrossClusterTask is for cross cluster transfer task\n\tCrossClusterTask struct {\n\t\tTransferTask\n\t\tTargetCluster string\n\t}\n\n\t// TransferTask is for regular transfer task\n\tTransferTask = persistence.TransferTaskInfo\n\n\tHistoryMigrationTask struct {\n\t\tTransfer      *TransferTask\n\t\tTimer         *TimerTask\n\t\tReplication   *ReplicationTask\n\t\tTask          *persistence.DataBlob\n\t\tTaskID        int64\n\t\tScheduledTime time.Time\n\t}\n\n\t// ShardCondition is the condition for making changes within a shard\n\tShardCondition struct {\n\t\tShardID int\n\t\tRangeID int64\n\t}\n\n\t// CurrentWorkflowWriteRequest is for insert/update current_workflow record\n\tCurrentWorkflowWriteRequest struct {\n\t\tWriteMode CurrentWorkflowWriteMode\n\t\tRow       CurrentWorkflowRow\n\t\tCondition *CurrentWorkflowWriteCondition\n\t}\n\n\t// CurrentWorkflowWriteCondition is the condition for updating current_workflow record\n\tCurrentWorkflowWriteCondition struct {\n\t\tCurrentRunID     *string\n\t\tLastWriteVersion *int64\n\t\tState            *int\n\t}\n\n\t// CurrentWorkflowWriteMode controls how to write current_workflow\n\tCurrentWorkflowWriteMode int\n\n\t// CurrentWorkflowRow is the current_workflow row\n\tCurrentWorkflowRow struct {\n\t\tShardID          int\n\t\tDomainID         string\n\t\tWorkflowID       string\n\t\tRunID            string\n\t\tState            int\n\t\tCloseStatus      int\n\t\tCreateRequestID  string\n\t\tLastWriteVersion int64\n\t}\n\n\t// WorkflowRequestRow is the request which has been applied to a workflow\n\tWorkflowRequestRow struct {\n\t\tShardID     int\n\t\tDomainID    string\n\t\tWorkflowID  string\n\t\tRequestType persistence.WorkflowRequestType\n\t\tRequestID   string\n\t\tVersion     int64\n\t\tRunID       string\n\t}\n\n\tWorkflowRequestWriteMode int\n\n\tWorkflowRequestsWriteRequest struct {\n\t\tRows      []*WorkflowRequestRow\n\t\tWriteMode WorkflowRequestWriteMode\n\t}\n\n\tActiveClusterSelectionPolicyRow struct {\n\t\tShardID    int\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t\tPolicy     *persistence.DataBlob\n\t}\n\n\t// TasksFilter is for filtering tasks\n\tTasksFilter struct {\n\t\tTaskListFilter\n\t\t// Exclusive\n\t\tMinTaskID int64\n\t\t// Inclusive\n\t\tMaxTaskID int64\n\t\tBatchSize int\n\t}\n\n\t// TaskRowForInsert is the struct to inserting task\n\tTaskRowForInsert struct {\n\t\tTaskRow\n\t\t// <= 0 means no TTL\n\t\tTTLSeconds int\n\t}\n\n\t// TaskRow represent a task row\n\tTaskRow struct {\n\t\tDomainID     string\n\t\tTaskListName string\n\t\tTaskListType int\n\t\tTaskID       int64\n\n\t\tWorkflowID      string\n\t\tRunID           string\n\t\tScheduledID     int64\n\t\tExpiry          time.Time\n\t\tCreatedTime     time.Time\n\t\tPartitionConfig map[string]string\n\t}\n\n\t// TaskListFilter is for filtering tasklist\n\tTaskListFilter struct {\n\t\tDomainID     string\n\t\tTaskListName string\n\t\tTaskListType int\n\t}\n\n\t// TaskListRow is a tasklist row\n\tTaskListRow struct {\n\t\tDomainID     string\n\t\tTaskListName string\n\t\tTaskListType int\n\n\t\tRangeID                 int64\n\t\tTaskListKind            int\n\t\tAckLevel                int64\n\t\tCurrentTimeStamp        time.Time\n\t\tLastUpdatedTime         time.Time\n\t\tAdaptivePartitionConfig *persistence.TaskListPartitionConfig\n\t}\n\n\t// ListTaskListResult is the result of list tasklists\n\tListTaskListResult struct {\n\t\tTaskLists     []*TaskListRow\n\t\tNextPageToken []byte\n\t}\n\n\t// ShardRow is the same as persistence.InternalShardInfo\n\t// Separate them later when there is a need.\n\tShardRow struct {\n\t\t*persistence.InternalShardInfo\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// ConflictedShardRow contains the partial information about a shard returned when a conditional write fails\n\tConflictedShardRow struct {\n\t\tShardID int\n\t\t// PreviousRangeID is the condition of previous change that used for conditional update\n\t\tPreviousRangeID int64\n\t\t// optional detailed information for logging purpose\n\t\tDetails string\n\t}\n\n\t// DomainRow defines the row struct for queue message\n\tDomainRow struct {\n\t\tInfo                        *persistence.DomainInfo\n\t\tConfig                      *persistence.InternalDomainConfig\n\t\tReplicationConfig           *persistence.InternalDomainReplicationConfig\n\t\tConfigVersion               int64\n\t\tFailoverVersion             int64\n\t\tFailoverNotificationVersion int64\n\t\tPreviousFailoverVersion     int64\n\t\tFailoverEndTime             *time.Time\n\t\tNotificationVersion         int64\n\t\tLastUpdatedTime             time.Time\n\t\tIsGlobalDomain              bool\n\t\tCurrentTimeStamp            time.Time\n\t}\n\n\t// DomainAuditLogRow defines the row struct for domain audit log\n\tDomainAuditLogRow struct {\n\t\tDomainID            string\n\t\tEventID             string\n\t\tStateBefore         []byte\n\t\tStateBeforeEncoding string\n\t\tStateAfter          []byte\n\t\tStateAfterEncoding  string\n\t\tOperationType       persistence.DomainAuditOperationType\n\t\tCreatedTime         time.Time\n\t\tLastUpdatedTime     time.Time\n\t\tIdentity            string\n\t\tIdentityType        string\n\t\tComment             string\n\t\tTTLSeconds          int64 // TTL for the audit log entry in seconds\n\t}\n\n\t// DomainAuditLogFilter contains the filter criteria for querying domain audit logs\n\tDomainAuditLogFilter struct {\n\t\tDomainID      string\n\t\tOperationType persistence.DomainAuditOperationType\n\t\t// MinCreatedTime is inclusive\n\t\tMinCreatedTime *time.Time\n\t\t// MaxCreatedTime is exclusive\n\t\tMaxCreatedTime *time.Time\n\t\tPageSize       int\n\t\tNextPageToken  []byte\n\t}\n\n\t// SelectMessagesBetweenRequest is a request struct for SelectMessagesBetween\n\tSelectMessagesBetweenRequest struct {\n\t\tQueueType               persistence.QueueType\n\t\tExclusiveBeginMessageID int64\n\t\tInclusiveEndMessageID   int64\n\t\tPageSize                int\n\t\tNextPageToken           []byte\n\t}\n\n\t// SelectMessagesBetweenResponse is a response struct for SelectMessagesBetween\n\tSelectMessagesBetweenResponse struct {\n\t\tRows          []QueueMessageRow\n\t\tNextPageToken []byte\n\t}\n\n\t// QueueMessageRow defines the row struct for queue message\n\tQueueMessageRow struct {\n\t\tQueueType        persistence.QueueType\n\t\tID               int64\n\t\tPayload          []byte\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// QueueMetadataRow defines the row struct for metadata\n\tQueueMetadataRow struct {\n\t\tQueueType        persistence.QueueType\n\t\tClusterAckLevels map[string]int64\n\t\tVersion          int64\n\t\tCurrentTimeStamp time.Time\n\t}\n\n\t// HistoryNodeRow represents a row in history_node table\n\tHistoryNodeRow struct {\n\t\tShardID  int\n\t\tTreeID   string\n\t\tBranchID string\n\t\tNodeID   int64\n\t\t// Note: use pointer so that it's easier to multiple by -1 if needed\n\t\tTxnID           *int64\n\t\tData            []byte\n\t\tDataEncoding    string\n\t\tCreateTimestamp time.Time\n\t}\n\n\t// HistoryNodeFilter contains the column names within history_node table that\n\t// can be used to filter results through a WHERE clause\n\tHistoryNodeFilter struct {\n\t\tShardID  int\n\t\tTreeID   string\n\t\tBranchID string\n\t\t// Inclusive\n\t\tMinNodeID int64\n\t\t// Exclusive\n\t\tMaxNodeID     int64\n\t\tNextPageToken []byte\n\t\tPageSize      int\n\t}\n\n\t// HistoryTreeRow represents a row in history_tree table\n\tHistoryTreeRow struct {\n\t\tShardID         int\n\t\tTreeID          string\n\t\tBranchID        string\n\t\tAncestors       []*types.HistoryBranchRange\n\t\tCreateTimestamp time.Time\n\t\tInfo            string\n\t}\n\n\t// HistoryTreeFilter contains the column names within history_tree table that\n\t// can be used to filter results through a WHERE clause\n\tHistoryTreeFilter struct {\n\t\tShardID  int\n\t\tTreeID   string\n\t\tBranchID *string\n\t}\n)\n\nconst (\n\tAllOpen VisibilityFilterType = iota\n\tAllClosed\n\tOpenByWorkflowType\n\tClosedByWorkflowType\n\tOpenByWorkflowID\n\tClosedByWorkflowID\n\tClosedByClosedStatus\n)\n\n// enums of VisibilitySortType\nconst (\n\tSortByStartTime VisibilitySortType = iota\n\tSortByClosedTime\n)\n\n// enums of CurrentWorkflowWriteMode\nconst (\n\tCurrentWorkflowWriteModeNoop CurrentWorkflowWriteMode = iota\n\tCurrentWorkflowWriteModeUpdate\n\tCurrentWorkflowWriteModeInsert\n)\n\n// enums of WorkflowExecutionMapsWriteMode\nconst (\n\t// WorkflowExecutionMapsWriteModeCreate will upsert new entry to maps\n\tWorkflowExecutionMapsWriteModeCreate WorkflowExecutionMapsWriteMode = iota\n\t// WorkflowExecutionMapsWriteModeUpdate will upsert new entry to maps and also delete entries from maps\n\tWorkflowExecutionMapsWriteModeUpdate\n\t// WorkflowExecutionMapsWriteModeReset will reset(override) the whole maps\n\tWorkflowExecutionMapsWriteModeReset\n)\n\nconst (\n\t// EventBufferWriteModeNone is for not doing anything to the event buffer\n\tEventBufferWriteModeNone EventBufferWriteMode = iota\n\t// EventBufferWriteModeAppend will append a new event to the event buffer\n\tEventBufferWriteModeAppend\n\t// EventBufferWriteModeClear will clear(delete all event from) the event buffer\n\tEventBufferWriteModeClear\n)\n\nconst (\n\tWorkflowRequestWriteModeInsert WorkflowRequestWriteMode = iota\n\tWorkflowRequestWriteModeUpsert\n)\n\n// GetCurrentRunID returns the current runID\nfunc (w *CurrentWorkflowWriteCondition) GetCurrentRunID() string {\n\tif w == nil || w.CurrentRunID == nil {\n\t\treturn \"\"\n\t}\n\treturn *w.CurrentRunID\n}\n"
  },
  {
    "path": "common/persistence/nosql/plugin.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\nvar supportedPlugins = map[string]nosqlplugin.Plugin{}\n\n// RegisterPlugin will register a NoSQL plugin\nfunc RegisterPlugin(pluginName string, plugin nosqlplugin.Plugin) {\n\tif _, ok := supportedPlugins[pluginName]; ok {\n\t\tpanic(\"plugin \" + pluginName + \" already registered\")\n\t}\n\tsupportedPlugins[pluginName] = plugin\n}\n\n// RegisterPluginForTest should be used only in tests to register the DB plugin and de-register at the end\nfunc RegisterPluginForTest(t *testing.T, pluginName string, plugin nosqlplugin.Plugin) {\n\tt.Cleanup(func() { delete(supportedPlugins, pluginName) })\n\tsupportedPlugins[pluginName] = plugin\n}\n\n// RegisterPluginIfNotExists will register a NoSQL plugin only if a plugin with same name has not already been registered\nfunc RegisterPluginIfNotExists(pluginName string, plugin nosqlplugin.Plugin) {\n\tif _, ok := supportedPlugins[pluginName]; !ok {\n\t\tsupportedPlugins[pluginName] = plugin\n\t}\n}\n\n// PluginRegistered returns true if plugin with given name has been registered, false otherwise\nfunc PluginRegistered(pluginName string) bool {\n\t_, ok := supportedPlugins[pluginName]\n\treturn ok\n}\n\n// GetRegisteredPluginNames returns the list of registered plugin names\nfunc GetRegisteredPluginNames() []string {\n\tvar plugins []string\n\tfor k := range supportedPlugins {\n\t\tplugins = append(plugins, k)\n\t}\n\treturn plugins\n}\n\n// NewNoSQLDB creates a returns a reference to a logical connection to the\n// underlying NoSQL database. The returned object is to tied to a single\n// NoSQL database and the object can be used to perform CRUD operations on\n// the tables in the database\nfunc NewNoSQLDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.DB, error) {\n\tplugin, ok := supportedPlugins[cfg.PluginName]\n\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"not supported plugin %v, only supported: %v\", cfg.PluginName, supportedPlugins)\n\t}\n\n\treturn plugin.CreateDB(cfg, logger, dc)\n}\n\n// NewNoSQLAdminDB returns a AdminDB\nfunc NewNoSQLAdminDB(cfg *config.NoSQL, logger log.Logger, dc *persistence.DynamicConfiguration) (nosqlplugin.AdminDB, error) {\n\tplugin, ok := supportedPlugins[cfg.PluginName]\n\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"not supported plugin %v, only supported: %v\", cfg.PluginName, supportedPlugins)\n\t}\n\n\treturn plugin.CreateAdminDB(cfg, logger, dc)\n}\n"
  },
  {
    "path": "common/persistence/nosql/sharded_nosql_store.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination sharded_nosql_store_mock.go\n\npackage nosql\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype shardedNosqlStore interface {\n\tGetStoreShardByHistoryShard(shardID int) (*nosqlStore, error)\n\tGetStoreShardByTaskList(domainID string, taskListName string, taskType int) (*nosqlStore, error)\n\tGetDefaultShard() nosqlStore\n\tClose()\n\tGetName() string\n\tGetShardingPolicy() shardingPolicy\n\tGetLogger() log.Logger\n\tGetMetricsClient() metrics.Client\n}\n\n// shardedNosqlStore is a store that may have one or more shards\ntype shardedNosqlStoreImpl struct {\n\tsync.RWMutex\n\n\tconfig          config.ShardedNoSQL\n\tdc              *persistence.DynamicConfiguration\n\tlogger          log.Logger\n\tmetricsClient   metrics.Client\n\tconnectedShards map[string]nosqlStore\n\tdefaultShard    nosqlStore\n\tshardingPolicy  shardingPolicy\n}\n\nfunc newShardedNosqlStore(cfg config.ShardedNoSQL, logger log.Logger, metricsClient metrics.Client, dc *persistence.DynamicConfiguration) (shardedNosqlStore, error) {\n\tsn := shardedNosqlStoreImpl{\n\t\tconfig:        cfg,\n\t\tdc:            dc,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t}\n\n\t// Connect to the default shard\n\tdefaultShardName := cfg.DefaultShard\n\tstore, err := sn.connectToShard(defaultShardName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsn.defaultShard = *store\n\tsn.connectedShards = map[string]nosqlStore{\n\t\tdefaultShardName: sn.defaultShard,\n\t}\n\n\t// Parse & validate the sharding policy\n\tsn.shardingPolicy, err = newShardingPolicy(logger, cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &sn, nil\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetStoreShardByHistoryShard(shardID int) (*nosqlStore, error) {\n\tshardName, err := sn.shardingPolicy.getHistoryShardName(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn sn.getShard(shardName)\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetStoreShardByTaskList(domainID string, taskListName string, taskType int) (*nosqlStore, error) {\n\tshardName := sn.shardingPolicy.getTaskListShardName(domainID, taskListName, taskType)\n\treturn sn.getShard(shardName)\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetDefaultShard() nosqlStore {\n\treturn sn.defaultShard\n}\n\nfunc (sn *shardedNosqlStoreImpl) Close() {\n\tsn.RLock()\n\tdefer sn.RUnlock()\n\tfor name, shard := range sn.connectedShards {\n\t\tsn.logger.Warn(\"Closing store shard\", tag.StoreShard(name))\n\t\tshard.Close()\n\t}\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetName() string {\n\treturn \"shardedNosql\"\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetShardingPolicy() shardingPolicy {\n\treturn sn.shardingPolicy\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetLogger() log.Logger {\n\treturn sn.logger\n}\n\nfunc (sn *shardedNosqlStoreImpl) GetMetricsClient() metrics.Client {\n\treturn sn.metricsClient\n}\n\nfunc (sn *shardedNosqlStoreImpl) getShard(shardName string) (*nosqlStore, error) {\n\tsn.RLock()\n\tshard, found := sn.connectedShards[shardName]\n\tsn.RUnlock()\n\n\tif found {\n\t\treturn &shard, nil\n\t}\n\n\t_, ok := sn.config.Connections[shardName]\n\tif !ok {\n\t\treturn nil, &ShardingError{\n\t\t\tMessage: fmt.Sprintf(\"Unknown db shard name: %v\", shardName),\n\t\t}\n\t}\n\n\tsn.Lock()\n\tdefer sn.Unlock()\n\tif shard, ok := sn.connectedShards[shardName]; ok { // read again to double-check\n\t\treturn &shard, nil\n\t}\n\n\ts, err := sn.connectToShard(shardName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsn.connectedShards[shardName] = *s\n\tsn.logger.Info(\"Connected to store shard\", tag.StoreShard(shardName))\n\treturn s, nil\n}\n\nfunc (sn *shardedNosqlStoreImpl) connectToShard(shardName string) (*nosqlStore, error) {\n\tcfg, ok := sn.config.Connections[shardName]\n\tif !ok {\n\t\treturn nil, &ShardingError{\n\t\t\tMessage: fmt.Sprintf(\"Unknown db shard name: %v\", shardName),\n\t\t}\n\t}\n\n\tsn.logger.Info(\"Connecting to store shard\", tag.StoreShard(shardName))\n\tdb, err := NewNoSQLDB(cfg.NoSQLPlugin, sn.logger, sn.dc)\n\tif err != nil {\n\t\tsn.logger.Error(\"Failed to connect to store shard\", tag.StoreShard(shardName), tag.Error(err))\n\t\treturn nil, err\n\t}\n\tshard := nosqlStore{\n\t\tdb:     db,\n\t\tlogger: sn.logger,\n\t\tdc:     sn.dc,\n\t}\n\treturn &shard, nil\n}\n"
  },
  {
    "path": "common/persistence/nosql/sharded_nosql_store_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: sharded_nosql_store.go\n//\n// Generated by this command:\n//\n//\tmockgen -package nosql -source sharded_nosql_store.go -destination sharded_nosql_store_mock.go\n//\n\n// Package nosql is a generated GoMock package.\npackage nosql\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tlog \"github.com/uber/cadence/common/log\"\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n)\n\n// MockshardedNosqlStore is a mock of shardedNosqlStore interface.\ntype MockshardedNosqlStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockshardedNosqlStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockshardedNosqlStoreMockRecorder is the mock recorder for MockshardedNosqlStore.\ntype MockshardedNosqlStoreMockRecorder struct {\n\tmock *MockshardedNosqlStore\n}\n\n// NewMockshardedNosqlStore creates a new mock instance.\nfunc NewMockshardedNosqlStore(ctrl *gomock.Controller) *MockshardedNosqlStore {\n\tmock := &MockshardedNosqlStore{ctrl: ctrl}\n\tmock.recorder = &MockshardedNosqlStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockshardedNosqlStore) EXPECT() *MockshardedNosqlStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockshardedNosqlStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockshardedNosqlStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockshardedNosqlStore)(nil).Close))\n}\n\n// GetDefaultShard mocks base method.\nfunc (m *MockshardedNosqlStore) GetDefaultShard() nosqlStore {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDefaultShard\")\n\tret0, _ := ret[0].(nosqlStore)\n\treturn ret0\n}\n\n// GetDefaultShard indicates an expected call of GetDefaultShard.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetDefaultShard() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDefaultShard\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetDefaultShard))\n}\n\n// GetLogger mocks base method.\nfunc (m *MockshardedNosqlStore) GetLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetLogger indicates an expected call of GetLogger.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLogger\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetLogger))\n}\n\n// GetMetricsClient mocks base method.\nfunc (m *MockshardedNosqlStore) GetMetricsClient() metrics.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsClient\")\n\tret0, _ := ret[0].(metrics.Client)\n\treturn ret0\n}\n\n// GetMetricsClient indicates an expected call of GetMetricsClient.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetMetricsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsClient\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetMetricsClient))\n}\n\n// GetName mocks base method.\nfunc (m *MockshardedNosqlStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetName))\n}\n\n// GetShardingPolicy mocks base method.\nfunc (m *MockshardedNosqlStore) GetShardingPolicy() shardingPolicy {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardingPolicy\")\n\tret0, _ := ret[0].(shardingPolicy)\n\treturn ret0\n}\n\n// GetShardingPolicy indicates an expected call of GetShardingPolicy.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetShardingPolicy() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardingPolicy\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetShardingPolicy))\n}\n\n// GetStoreShardByHistoryShard mocks base method.\nfunc (m *MockshardedNosqlStore) GetStoreShardByHistoryShard(shardID int) (*nosqlStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStoreShardByHistoryShard\", shardID)\n\tret0, _ := ret[0].(*nosqlStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStoreShardByHistoryShard indicates an expected call of GetStoreShardByHistoryShard.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetStoreShardByHistoryShard(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStoreShardByHistoryShard\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetStoreShardByHistoryShard), shardID)\n}\n\n// GetStoreShardByTaskList mocks base method.\nfunc (m *MockshardedNosqlStore) GetStoreShardByTaskList(domainID, taskListName string, taskType int) (*nosqlStore, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStoreShardByTaskList\", domainID, taskListName, taskType)\n\tret0, _ := ret[0].(*nosqlStore)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStoreShardByTaskList indicates an expected call of GetStoreShardByTaskList.\nfunc (mr *MockshardedNosqlStoreMockRecorder) GetStoreShardByTaskList(domainID, taskListName, taskType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStoreShardByTaskList\", reflect.TypeOf((*MockshardedNosqlStore)(nil).GetStoreShardByTaskList), domainID, taskListName, taskType)\n}\n"
  },
  {
    "path": "common/persistence/nosql/sharded_nosql_store_test.go",
    "content": "// Copyright (c) 2023 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t. \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n)\n\ntype shardedNosqlStoreTestSuite struct {\n\tsuite.Suite\n\t*require.Assertions\n\tmockController *gomock.Controller\n\n\tmockPlugin *nosqlplugin.MockPlugin\n}\n\nfunc (s *shardedNosqlStoreTestSuite) SetupSuite() {\n\ts.mockController = gomock.NewController(s.T())\n}\n\nfunc (s *shardedNosqlStoreTestSuite) TearDownSuite() {\n}\n\nfunc (s *shardedNosqlStoreTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\tmockDB := nosqlplugin.NewMockDB(s.mockController)\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(s.mockController)\n\tmockPlugin.EXPECT().\n\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(mockDB, nil).AnyTimes()\n\tdelete(supportedPlugins, \"cassandra\")\n\tRegisterPlugin(\"cassandra\", mockPlugin)\n}\n\nfunc TestShardedNosqlStoreTestSuite(t *testing.T) {\n\ts := new(shardedNosqlStoreTestSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *shardedNosqlStoreTestSuite) TestValidConfiguration() {\n\tstore := s.newShardedStoreForTest()\n\ts.Equal(1, len(store.connectedShards))\n\ts.Contains(store.connectedShards, \"shard-1\")\n\ts.Equal(store.GetDefaultShard(), store.defaultShard)\n\ts.Equal(store.connectedShards[\"shard-1\"], store.defaultShard)\n\ts.Equal(\"shard-1\", store.shardingPolicy.defaultShard)\n\ts.True(store.shardingPolicy.hasShardedTasklist)\n\ts.True(store.shardingPolicy.hasShardedHistory)\n}\n\nfunc (s *shardedNosqlStoreTestSuite) TestStoreSelectionForHistoryShard() {\n\tmockDB1 := nosqlplugin.NewMockDB(s.mockController)\n\tmockDB1.EXPECT().Close().Times(1)\n\tmockDB2 := nosqlplugin.NewMockDB(s.mockController)\n\tmockDB2.EXPECT().Close().Times(1)\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(s.mockController)\n\tgomock.InOrder(\n\t\tmockPlugin.EXPECT().\n\t\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(mockDB1, nil),\n\t\tmockPlugin.EXPECT().\n\t\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(nil, errors.New(\"error creating db\")),\n\t\tmockPlugin.EXPECT().\n\t\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(mockDB2, nil),\n\t)\n\tdelete(supportedPlugins, \"cassandra\")\n\tRegisterPlugin(\"cassandra\", mockPlugin)\n\n\tstore := s.newShardedStoreForTest()\n\tdefer store.Close()\n\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == store.defaultShard.db)\n\n\t// Shard 0 is same default shard in this test, so connectedShards shouldn't change\n\tstoreShard1, err := store.GetStoreShardByHistoryShard(0)\n\ts.NoError(err)\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Getting the same shard again shouldn't create a new connection\n\tstoreShard1, err = store.GetStoreShardByHistoryShard(0)\n\ts.NoError(err)\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Getting a new shard should create a new connection but it will fail on first attempt\n\tstoreShard2, err := store.GetStoreShardByHistoryShard(1)\n\ts.Error(err)\n\ts.Equal(1, len(store.connectedShards))\n\n\t// Getting a new shard should create a new connection on second attempt\n\tstoreShard2, err = store.GetStoreShardByHistoryShard(1)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB2 == storeShard2.db)\n\n\t// After the new connection, getting the previous shard should still work as it used to\n\tstoreShard1, err = store.GetStoreShardByHistoryShard(0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Getting a non-existing shard should result in an error\n\tunknownShard, err := store.GetStoreShardByHistoryShard(2)\n\ts.ErrorContains(err, \"Failed to identify store shard\")\n\ts.Empty(unknownShard)\n\n\t// Ensure the store shard connections created for history shards are available for tasklists, too\n\tstoreShard1, err = store.GetStoreShardByTaskList(\"domain1\", \"tl1\", 0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\tstoreShard2, err = store.GetStoreShardByTaskList(\"domain1\", \"tl2\", 0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB2 == storeShard2.db)\n}\n\nfunc (s *shardedNosqlStoreTestSuite) newShardedStoreForTest() *shardedNosqlStoreImpl {\n\tcfg := getValidShardedNoSQLConfig()\n\tlogger := log.NewNoop()\n\tstoreInterface, err := newShardedNosqlStore(cfg, logger, metrics.NewNoopMetricsClient(), nil)\n\ts.NoError(err)\n\ts.Equal(\"shardedNosql\", storeInterface.GetName())\n\ts.Equal(logger, storeInterface.GetLogger())\n\tstore := storeInterface.(*shardedNosqlStoreImpl)\n\ts.Equal(storeInterface.GetShardingPolicy(), store.shardingPolicy)\n\treturn store\n}\n\nfunc (s *shardedNosqlStoreTestSuite) TestStoreSelectionForTasklist() {\n\tmockDB1 := nosqlplugin.NewMockDB(s.mockController)\n\tmockDB1.EXPECT().Close().Times(1)\n\tmockDB2 := nosqlplugin.NewMockDB(s.mockController)\n\tmockDB2.EXPECT().Close().Times(1)\n\n\tmockPlugin := nosqlplugin.NewMockPlugin(s.mockController)\n\tgomock.InOrder(\n\t\tmockPlugin.EXPECT().\n\t\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(mockDB1, nil),\n\t\tmockPlugin.EXPECT().\n\t\t\tCreateDB(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(mockDB2, nil),\n\t)\n\tdelete(supportedPlugins, \"cassandra\")\n\tRegisterPlugin(\"cassandra\", mockPlugin)\n\n\tstore := s.newShardedStoreForTest()\n\tdefer store.Close()\n\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == store.defaultShard.db)\n\n\t// Shard 0 is same default shard in this test, so connectedShards shouldn't change\n\tstoreShard1, err := store.GetStoreShardByTaskList(\"domain1\", \"tl1\", 0)\n\ts.NoError(err)\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Getting the same shard again shouldn't create a new connection\n\tstoreShard1, err = store.GetStoreShardByTaskList(\"domain1\", \"tl1\", 0)\n\ts.NoError(err)\n\ts.Equal(1, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Getting a new shard should create a new connection\n\tstoreShard2, err := store.GetStoreShardByTaskList(\"domain1\", \"tl2\", 0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB2 == storeShard2.db)\n\n\t// After the new connection, getting the previous shard should still work as it used to\n\tstoreShard1, err = store.GetStoreShardByTaskList(\"domain1\", \"tl1\", 0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\n\t// Ensure the store shard connections created for tasklists are available for tasklists, too\n\tstoreShard1, err = store.GetStoreShardByHistoryShard(0)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB1 == storeShard1.db)\n\tstoreShard2, err = store.GetStoreShardByHistoryShard(1)\n\ts.NoError(err)\n\ts.Equal(2, len(store.connectedShards))\n\ts.True(mockDB2 == storeShard2.db)\n}\n\nfunc getValidShardedNoSQLConfig() ShardedNoSQL {\n\treturn ShardedNoSQL{\n\t\tDefaultShard: \"shard-1\",\n\t\tShardingPolicy: ShardingPolicy{\n\t\t\tHistoryShardMapping: []HistoryShardRange{\n\t\t\t\tHistoryShardRange{\n\t\t\t\t\tStart: 0,\n\t\t\t\t\tEnd:   1,\n\t\t\t\t\tShard: \"shard-1\",\n\t\t\t\t},\n\t\t\t\tHistoryShardRange{\n\t\t\t\t\tStart: 1,\n\t\t\t\t\tEnd:   2,\n\t\t\t\t\tShard: \"shard-2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tTaskListHashing: TasklistHashing{\n\t\t\t\tShardOrder: []string{\n\t\t\t\t\t\"shard-1\",\n\t\t\t\t\t\"shard-2\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tConnections: map[string]DBShardConnection{\n\t\t\t\"shard-1\": {\n\t\t\t\tNoSQLPlugin: &NoSQL{\n\t\t\t\t\tPluginName: \"cassandra\",\n\t\t\t\t\tHosts:      \"127.0.0.1\",\n\t\t\t\t\tKeyspace:   \"unit-test\",\n\t\t\t\t\tPort:       1234,\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"shard-2\": {\n\t\t\t\tNoSQLPlugin: &NoSQL{\n\t\t\t\t\tPluginName: \"cassandra\",\n\t\t\t\t\tHosts:      \"127.0.0.1\",\n\t\t\t\t\tKeyspace:   \"unit-test\",\n\t\t\t\t\tPort:       5678,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/persistence/nosql/sharding_policy.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/dgryski/go-farm\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// shardingPolicy holds the configuration for the sharding logic\ntype shardingPolicy struct {\n\tlogger       log.Logger\n\tdefaultShard string\n\tconfig       config.ShardedNoSQL\n\n\thasShardedHistory  bool\n\thasShardedTasklist bool\n}\n\nfunc newShardingPolicy(logger log.Logger, cfg config.ShardedNoSQL) (shardingPolicy, error) {\n\tsp := shardingPolicy{\n\t\tlogger:       logger,\n\t\tdefaultShard: cfg.DefaultShard,\n\t\tconfig:       cfg,\n\t}\n\n\terr := sp.parse()\n\tif err != nil {\n\t\treturn sp, err\n\t}\n\n\treturn sp, nil\n}\n\nfunc (sp *shardingPolicy) parse() error {\n\tsp.parseHistoryShardMapping()\n\tsp.parseTaskListShardingPolicy()\n\treturn nil\n}\n\nfunc (sp *shardingPolicy) parseHistoryShardMapping() {\n\thistoryShardMapping := sp.config.ShardingPolicy.HistoryShardMapping\n\tif len(historyShardMapping) == 0 {\n\t\treturn\n\t}\n\tsp.hasShardedHistory = true\n}\n\nfunc (sp *shardingPolicy) parseTaskListShardingPolicy() {\n\ttlShards := sp.config.ShardingPolicy.TaskListHashing.ShardOrder\n\tif len(tlShards) == 0 {\n\t\treturn\n\t}\n\tsp.hasShardedTasklist = true\n}\n\nfunc (sp *shardingPolicy) getHistoryShardName(shardID int) (string, error) {\n\tif !sp.hasShardedHistory {\n\t\tsp.logger.Debug(\"Selected default store shard for history shard\", tag.StoreShard(sp.defaultShard), tag.ShardID(shardID))\n\t\treturn sp.defaultShard, nil\n\t}\n\n\tfor _, r := range sp.config.ShardingPolicy.HistoryShardMapping {\n\t\tif shardID >= r.Start && shardID < r.End {\n\t\t\tsp.logger.Debug(\"Selected store shard history shard\", tag.StoreShard(r.Shard), tag.ShardID(shardID))\n\t\t\treturn r.Shard, nil\n\t\t}\n\t}\n\n\treturn \"\", &ShardingError{\n\t\tMessage: fmt.Sprintf(\"Failed to identify store shard for shardID %v\", shardID),\n\t}\n}\n\nfunc (sp *shardingPolicy) getTaskListShardName(domainID string, taskListName string, taskType int) string {\n\tif !sp.hasShardedTasklist {\n\t\tsp.logger.Debug(\"Selected default store shard for tasklist\", tag.StoreShard(sp.defaultShard), tag.WorkflowTaskListName(taskListName))\n\t\treturn sp.defaultShard\n\t}\n\ttlShards := sp.config.ShardingPolicy.TaskListHashing.ShardOrder\n\ttlShardCount := len(tlShards)\n\thash := farm.Hash32([]byte(domainID+\"_\"+taskListName)) % uint32(tlShardCount)\n\tshardIndex := int(hash) % tlShardCount\n\n\tsp.logger.Debug(\"Selected store shard tasklist\", tag.StoreShard(tlShards[shardIndex]), tag.WorkflowTaskListName(taskListName))\n\treturn tlShards[shardIndex]\n}\n"
  },
  {
    "path": "common/persistence/nosql/sharding_policy_test.go",
    "content": "// Copyright (c) 2023 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t. \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nfunc TestSimplePolicyWithSharding(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\trequire.True(t, sp.hasShardedHistory, \"sp.hasShardedHistory must be true\")\n\trequire.True(t, sp.hasShardedTasklist, \"sp.hasShardedTasklist must be true\")\n}\n\nfunc TestSimplePolicyWithoutSharding(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.ShardingPolicy = ShardingPolicy{} // remove the sharding policy\n\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\trequire.False(t, sp.hasShardedHistory, \"sp.hasShardedHistory must be false\")\n\trequire.False(t, sp.hasShardedTasklist, \"sp.hasShardedTasklist must be false\")\n}\n\nfunc TestSimplePolicyWithoutHistorySharding(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.ShardingPolicy.HistoryShardMapping = []HistoryShardRange{} // remove only the history sharding\n\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\trequire.False(t, sp.hasShardedHistory, \"sp.hasShardedHistory must be false\")\n\trequire.True(t, sp.hasShardedTasklist, \"sp.hasShardedTasklist must be true\")\n}\n\nfunc TestSimplePolicyWithoutTasklistSharding(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.ShardingPolicy.TaskListHashing = TasklistHashing{} // remove only the tasklist sharding\n\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\trequire.True(t, sp.hasShardedHistory, \"sp.hasShardedHistory must be false\")\n\trequire.False(t, sp.hasShardedTasklist, \"sp.hasShardedTasklist must be true\")\n}\n\nfunc TestHistorySharding(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\n\tshardName0, err1 := sp.getHistoryShardName(0)\n\tshardName1, err2 := sp.getHistoryShardName(1)\n\n\trequire.Nil(t, err1)\n\trequire.Nil(t, err2)\n\trequire.Equal(t, \"shard-1\", shardName0, \"shard name must be correct for shard 0\")\n\trequire.Equal(t, \"shard-2\", shardName1, \"shard name must be correct for shard 1\")\n}\n\nfunc TestHistorySharding_UnexpectedGapInHistoryRanges(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.ShardingPolicy.HistoryShardMapping = []HistoryShardRange{\n\t\tHistoryShardRange{\n\t\t\tStart: 0,\n\t\t\tEnd:   1,\n\t\t\tShard: \"shard-1\",\n\t\t},\n\t\tHistoryShardRange{\n\t\t\tStart: 2,\n\t\t\tEnd:   3,\n\t\t\tShard: \"shard-2\",\n\t\t},\n\t}\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\n\tshardName0, err1 := sp.getHistoryShardName(0)\n\tshardName2, err2 := sp.getHistoryShardName(2)\n\n\trequire.Nil(t, err1)\n\trequire.Nil(t, err2)\n\trequire.Equal(t, \"shard-1\", shardName0, \"shard name must be correct for shard 0\")\n\trequire.Equal(t, \"shard-2\", shardName2, \"shard name must be correct for shard 2\")\n\n\tunknownShard, err := sp.getHistoryShardName(1)\n\trequire.ErrorContains(t, err, \"Failed to identify store shard\")\n\trequire.Empty(t, unknownShard)\n}\n\nfunc TestHistorySharding_OutOfBounds(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\n\tunknownShard, err := sp.getHistoryShardName(-1)\n\trequire.ErrorContains(t, err, \"Failed to identify store shard\")\n\trequire.Empty(t, unknownShard)\n\n\tunknownShard, err = sp.getHistoryShardName(2)\n\trequire.ErrorContains(t, err, \"Failed to identify store shard\")\n\trequire.Empty(t, unknownShard)\n}\n\nfunc TestTaskListSharding_Simple(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\n\t// Note that shardName0 and shardName1 can't really be predicted due to hashing inside the function. The values\n\t// below were picked manually to yield the outcome needed. The test is useful to ensure the logic is deterministic.\n\tshardName1 := sp.getTaskListShardName(\"domain1\", \"tl1\", 0)\n\tshardName2 := sp.getTaskListShardName(\"domain1\", \"tl2\", 0)\n\n\trequire.Equal(t, \"shard-1\", shardName1, \"shard name must be correct for tl1\")\n\trequire.Equal(t, \"shard-2\", shardName2, \"shard name must be correct for tl2\")\n}\n\nfunc TestTaskListSharding_SingleShard(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tcfg.ShardingPolicy.TaskListHashing.ShardOrder = []string{\"only-shard\"}\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\trequire.True(t, sp.hasShardedTasklist, \"sp.hasShardedTasklist must be true\")\n\n\tcountExpected := 0\n\tcountUnexpected := 0\n\n\tfor i := 0; i < 1000000; i++ {\n\t\ts := sp.getTaskListShardName(uuid.New(), uuid.New(), 0)\n\t\tif s == \"only-shard\" {\n\t\t\tcountExpected++\n\t\t} else {\n\t\t\tcountUnexpected++\n\t\t}\n\t}\n\n\trequire.Equal(t, 0, countUnexpected, \"Unexpected shard names must not be returned\")\n\trequire.Equal(t, 1000000, countExpected, \"Expected shard name must be returned always\")\n}\n\nfunc TestTaskListSharding_Probabilistic(t *testing.T) {\n\tcfg := getValidShardedNoSQLConfig()\n\tsp, err := newShardingPolicy(log.NewNoop(), cfg)\n\trequire.NoError(t, err)\n\n\tcountShard1 := 0\n\tcountShard2 := 0\n\tcountUnexpected := 0\n\n\tfor i := 0; i < 1000000; i++ {\n\t\ts := sp.getTaskListShardName(uuid.New(), uuid.New(), 0)\n\t\tif s == \"shard-1\" {\n\t\t\tcountShard1++\n\t\t} else if s == \"shard-2\" {\n\t\t\tcountShard2++\n\t\t} else {\n\t\t\tcountUnexpected++\n\t\t}\n\t}\n\n\tratio := float32(countShard1) / float32(countShard2)\n\n\trequire.Equal(t, 0, countUnexpected, \"Unexpected shard names must not be returned\")\n\trequire.True(t, ratio > 0.95 && ratio < 1.05, \"Shards should have equal chance of being selected\")\n}\n"
  },
  {
    "path": "common/persistence/nosql/utils.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage nosql\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ShardingError represents invalid shard\ntype ShardingError struct {\n\tMessage string\n}\n\nfunc (e *ShardingError) Error() string {\n\treturn e.Message\n}\n\nfunc convertCommonErrors(errChecker nosqlplugin.ClientErrorChecker, operation string, err error) error {\n\tif errChecker.IsNotFoundError(err) {\n\t\treturn &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\"%v failed. Error: %v \", operation, err),\n\t\t}\n\t}\n\n\tif errChecker.IsTimeoutError(err) {\n\t\treturn &persistence.TimeoutError{Msg: fmt.Sprintf(\"%v timed out. Error: %v\", operation, err)}\n\t}\n\n\tif errChecker.IsThrottlingError(err) {\n\t\treturn &types.ServiceBusyError{\n\t\t\tMessage: fmt.Sprintf(\"%v operation failed. Error: %v\", operation, err),\n\t\t}\n\t}\n\n\tif errChecker.IsDBUnavailableError(err) {\n\t\treturn &persistence.DBUnavailableError{\n\t\t\tMsg: fmt.Sprintf(\"%v operation failed. Error: %v\", operation, err),\n\t\t}\n\t}\n\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\"%v operation failed. Error: %v\", operation, err),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/operation_mode_validator.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// NOTE: when modifying this file, plz make each case clear,\n//  do not combine cases together.\n// The idea for this file is to test whether current record\n// points to a zombie record.\n\n// ValidateCreateWorkflowModeState validate workflow creation mode & workflow state\nfunc ValidateCreateWorkflowModeState(\n\tmode CreateWorkflowMode,\n\tnewWorkflowSnapshot InternalWorkflowSnapshot,\n) error {\n\n\tworkflowState := newWorkflowSnapshot.ExecutionInfo.State\n\tif err := checkWorkflowState(workflowState); err != nil {\n\t\treturn err\n\t}\n\n\tswitch mode {\n\tcase CreateWorkflowModeBrandNew,\n\t\tCreateWorkflowModeWorkflowIDReuse,\n\t\tCreateWorkflowModeContinueAsNew:\n\t\tif workflowState == WorkflowStateZombie ||\n\t\t\tworkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidCreateWorkflowMode(\n\t\t\t\tmode,\n\t\t\t\tworkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tcase CreateWorkflowModeZombie:\n\t\tif workflowState == WorkflowStateCreated ||\n\t\t\tworkflowState == WorkflowStateRunning ||\n\t\t\tworkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidCreateWorkflowMode(\n\t\t\t\tmode,\n\t\t\t\tworkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown mode: %v\", mode),\n\t\t}\n\t}\n}\n\n// ValidateUpdateWorkflowModeState validate workflow update mode & workflow state\nfunc ValidateUpdateWorkflowModeState(\n\tmode UpdateWorkflowMode,\n\tcurrentWorkflowMutation InternalWorkflowMutation,\n\tnewWorkflowSnapshot *InternalWorkflowSnapshot,\n) error {\n\n\tcurrentWorkflowState := currentWorkflowMutation.ExecutionInfo.State\n\tif err := checkWorkflowState(currentWorkflowState); err != nil {\n\t\treturn err\n\t}\n\tvar newWorkflowState *int\n\tif newWorkflowSnapshot != nil {\n\t\tnewWorkflowState = &newWorkflowSnapshot.ExecutionInfo.State\n\t\tif err := checkWorkflowState(*newWorkflowState); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tswitch mode {\n\tcase UpdateWorkflowModeUpdateCurrent:\n\t\t// update current record\n\t\t// 1. current workflow only ->\n\t\t//  current workflow cannot be zombie\n\t\t// 2. current workflow & new workflow ->\n\t\t//  current workflow cannot be created / running,\n\t\t//  new workflow cannot be zombie\n\n\t\t// case 1\n\t\tif newWorkflowState == nil {\n\t\t\tif currentWorkflowState == WorkflowStateZombie {\n\t\t\t\treturn newInvalidUpdateWorkflowMode(mode, currentWorkflowState)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\n\t\t// case 2\n\t\tif currentWorkflowState == WorkflowStateCreated ||\n\t\t\tcurrentWorkflowState == WorkflowStateRunning ||\n\t\t\t*newWorkflowState == WorkflowStateZombie ||\n\t\t\t*newWorkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidUpdateWorkflowWithNewMode(mode, currentWorkflowState, *newWorkflowState)\n\t\t}\n\t\treturn nil\n\n\tcase UpdateWorkflowModeBypassCurrent:\n\t\t// bypass current record\n\t\t// 1. current workflow only ->\n\t\t//  current workflow cannot be created / running\n\t\t// 2. current workflow & new workflow ->\n\t\t//  current workflow cannot be created / running,\n\t\t//  new workflow cannot be created / running / completed\n\n\t\t// case 1\n\t\tif newWorkflowState == nil {\n\t\t\tif currentWorkflowState == WorkflowStateCreated ||\n\t\t\t\tcurrentWorkflowState == WorkflowStateRunning {\n\t\t\t\treturn newInvalidUpdateWorkflowMode(mode, currentWorkflowState)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\n\t\t// case 2\n\t\tif currentWorkflowState == WorkflowStateCreated ||\n\t\t\tcurrentWorkflowState == WorkflowStateRunning ||\n\t\t\t*newWorkflowState == WorkflowStateCreated ||\n\t\t\t*newWorkflowState == WorkflowStateRunning ||\n\t\t\t*newWorkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidUpdateWorkflowWithNewMode(\n\t\t\t\tmode,\n\t\t\t\tcurrentWorkflowState,\n\t\t\t\t*newWorkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tcase UpdateWorkflowModeIgnoreCurrent:\n\t\tif newWorkflowState != nil {\n\t\t\treturn newInvalidUpdateWorkflowWithNewMode(\n\t\t\t\tmode,\n\t\t\t\tcurrentWorkflowState,\n\t\t\t\t*newWorkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown mode: %v\", mode),\n\t\t}\n\t}\n}\n\n// ValidateConflictResolveWorkflowModeState validate workflow conflict resolve mode & workflow state\nfunc ValidateConflictResolveWorkflowModeState(\n\tmode ConflictResolveWorkflowMode,\n\tresetWorkflowSnapshot InternalWorkflowSnapshot,\n\tnewWorkflowSnapshot *InternalWorkflowSnapshot,\n\tcurrentWorkflowMutation *InternalWorkflowMutation,\n) error {\n\n\tresetWorkflowState := resetWorkflowSnapshot.ExecutionInfo.State\n\tif err := checkWorkflowState(resetWorkflowState); err != nil {\n\t\treturn err\n\t}\n\tvar newWorkflowState *int\n\tif newWorkflowSnapshot != nil {\n\t\tnewWorkflowState = &newWorkflowSnapshot.ExecutionInfo.State\n\t\tif err := checkWorkflowState(*newWorkflowState); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tvar currentWorkflowState *int\n\tif currentWorkflowMutation != nil {\n\t\tcurrentWorkflowState = &currentWorkflowMutation.ExecutionInfo.State\n\t\tif err := checkWorkflowState(*currentWorkflowState); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tswitch mode {\n\tcase ConflictResolveWorkflowModeUpdateCurrent:\n\t\t// update current record\n\t\t// 1. reset workflow only ->\n\t\t//  reset workflow cannot be zombie\n\t\t// 2. reset workflow & new workflow ->\n\t\t//  reset workflow cannot be created / running / zombie,\n\t\t//  new workflow cannot be zombie / completed\n\t\t// 3. current workflow & reset workflow ->\n\t\t//  current workflow cannot be created / running,\n\t\t//  reset workflow cannot be zombie\n\t\t// 4. current workflow & reset workflow & new workflow ->\n\t\t//  current workflow cannot be created / running,\n\t\t//  reset workflow cannot be created / running / zombie,\n\t\t//  new workflow cannot be zombie / completed\n\n\t\t// TODO remove case 1 & 2 support once 2DC is deprecated\n\t\t// it is ok that currentWorkflowMutation is null, only for 2 DC case\n\t\t// NDC should always require current workflow for CAS\n\t\t// Note: current workflow mutation can be in zombie state, for the update\n\n\t\t// case 1 & 2\n\t\tif currentWorkflowState == nil {\n\t\t\t// case 1\n\t\t\tif newWorkflowState == nil {\n\t\t\t\tif resetWorkflowState == WorkflowStateZombie {\n\t\t\t\t\treturn newInvalidConflictResolveWorkflowMode(\n\t\t\t\t\t\tmode,\n\t\t\t\t\t\tresetWorkflowState,\n\t\t\t\t\t)\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\t// case 2\n\t\t\tif resetWorkflowState == WorkflowStateCreated ||\n\t\t\t\tresetWorkflowState == WorkflowStateRunning ||\n\t\t\t\tresetWorkflowState == WorkflowStateZombie ||\n\t\t\t\t*newWorkflowState == WorkflowStateZombie ||\n\t\t\t\t*newWorkflowState == WorkflowStateCompleted {\n\t\t\t\treturn newInvalidConflictResolveWorkflowWithNewMode(\n\t\t\t\t\tmode,\n\t\t\t\t\tresetWorkflowState,\n\t\t\t\t\t*newWorkflowState,\n\t\t\t\t)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\n\t\t// case 3 & 4\n\t\t// case 3\n\t\tif newWorkflowState == nil {\n\t\t\tif *currentWorkflowState == WorkflowStateCreated ||\n\t\t\t\t*currentWorkflowState == WorkflowStateRunning ||\n\t\t\t\tresetWorkflowState == WorkflowStateZombie {\n\t\t\t\treturn newInvalidConflictResolveWorkflowWithCurrentMode(\n\t\t\t\t\tmode,\n\t\t\t\t\tresetWorkflowState,\n\t\t\t\t\t*currentWorkflowState,\n\t\t\t\t)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\n\t\t// case 4\n\t\tif *currentWorkflowState == WorkflowStateCreated ||\n\t\t\t*currentWorkflowState == WorkflowStateRunning ||\n\t\t\tresetWorkflowState == WorkflowStateCreated ||\n\t\t\tresetWorkflowState == WorkflowStateRunning ||\n\t\t\tresetWorkflowState == WorkflowStateZombie ||\n\t\t\t*newWorkflowState == WorkflowStateZombie ||\n\t\t\t*newWorkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidConflictResolveWorkflowWithCurrentWithNewMode(\n\t\t\t\tmode,\n\t\t\t\tresetWorkflowState,\n\t\t\t\t*newWorkflowState,\n\t\t\t\t*currentWorkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tcase ConflictResolveWorkflowModeBypassCurrent:\n\t\t// bypass current record\n\t\t// * current workflow cannot be set\n\t\t// 1. reset workflow only ->\n\t\t//  reset workflow cannot be created / running\n\t\t// 2. reset workflow & new workflow ->\n\t\t//  reset workflow cannot be created / running / zombie,\n\t\t//  new workflow cannot be created / running / completed\n\n\t\t// precondition\n\t\tif currentWorkflowMutation != nil {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\t\"Invalid workflow conflict resolve mode %v, encounter current workflow\",\n\t\t\t\t\tmode,\n\t\t\t\t),\n\t\t\t}\n\t\t}\n\n\t\t// case 1\n\t\tif newWorkflowState == nil {\n\t\t\tif resetWorkflowState == WorkflowStateCreated ||\n\t\t\t\tresetWorkflowState == WorkflowStateRunning {\n\t\t\t\treturn newInvalidConflictResolveWorkflowMode(\n\t\t\t\t\tmode,\n\t\t\t\t\tresetWorkflowState,\n\t\t\t\t)\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\n\t\t// case 2\n\t\tif resetWorkflowState == WorkflowStateCreated ||\n\t\t\tresetWorkflowState == WorkflowStateRunning ||\n\t\t\tresetWorkflowState == WorkflowStateZombie ||\n\t\t\t*newWorkflowState == WorkflowStateCreated ||\n\t\t\t*newWorkflowState == WorkflowStateRunning ||\n\t\t\t*newWorkflowState == WorkflowStateCompleted {\n\t\t\treturn newInvalidConflictResolveWorkflowWithNewMode(\n\t\t\t\tmode,\n\t\t\t\tresetWorkflowState,\n\t\t\t\t*newWorkflowState,\n\t\t\t)\n\t\t}\n\t\treturn nil\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown mode: %v\", mode),\n\t\t}\n\t}\n}\n\nfunc checkWorkflowState(state int) error {\n\tswitch state {\n\tcase WorkflowStateCreated,\n\t\tWorkflowStateRunning,\n\t\tWorkflowStateZombie,\n\t\tWorkflowStateCompleted,\n\t\tWorkflowStateCorrupted:\n\t\treturn nil\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t}\n\t}\n}\n\nfunc newInvalidCreateWorkflowMode(\n\tmode CreateWorkflowMode,\n\tworkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow create mode %v, state: %v\",\n\t\t\tmode,\n\t\t\tworkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidUpdateWorkflowMode(\n\tmode UpdateWorkflowMode,\n\tcurrentWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow update mode %v, state: %v\",\n\t\t\tmode,\n\t\t\tcurrentWorkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidUpdateWorkflowWithNewMode(\n\tmode UpdateWorkflowMode,\n\tcurrentWorkflowState int,\n\tnewWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow update mode %v, current state: %v, new state: %v\",\n\t\t\tmode,\n\t\t\tcurrentWorkflowState,\n\t\t\tnewWorkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidConflictResolveWorkflowMode(\n\tmode ConflictResolveWorkflowMode,\n\tresetWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow conflict resolve mode %v, reset state: %v\",\n\t\t\tmode,\n\t\t\tresetWorkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidConflictResolveWorkflowWithNewMode(\n\tmode ConflictResolveWorkflowMode,\n\tresetWorkflowState int,\n\tnewWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow conflict resolve mode %v, reset state: %v, new state: %v\",\n\t\t\tmode,\n\t\t\tresetWorkflowState,\n\t\t\tnewWorkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidConflictResolveWorkflowWithCurrentMode(\n\tmode ConflictResolveWorkflowMode,\n\tresetWorkflowState int,\n\tcurrentWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow conflict resolve mode %v, reset state: %v, current state: %v\",\n\t\t\tmode,\n\t\t\tresetWorkflowState,\n\t\t\tcurrentWorkflowState,\n\t\t),\n\t}\n}\n\nfunc newInvalidConflictResolveWorkflowWithCurrentWithNewMode(\n\tmode ConflictResolveWorkflowMode,\n\tresetWorkflowState int,\n\tnewWorkflowState int,\n\tcurrentWorkflowState int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"Invalid workflow conflict resolve mode %v, reset state: %v, new state: %v, current state: %v\",\n\t\t\tmode,\n\t\t\tresetWorkflowState,\n\t\t\tnewWorkflowState,\n\t\t\tcurrentWorkflowState,\n\t\t),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/operation_mode_validator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tvalidateOperationWorkflowModeStateSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestValidateOperationWorkflowModeStateSuite(t *testing.T) {\n\ts := new(validateOperationWorkflowModeStateSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) SetupSuite() {\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TearDownSuite() {\n\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) SetupTest() {\n\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TearDownTest() {\n\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestCreateMode_UpdateCurrent() {\n\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    true,\n\t}\n\n\tcreatModes := []CreateWorkflowMode{\n\t\tCreateWorkflowModeBrandNew,\n\t\tCreateWorkflowModeWorkflowIDReuse,\n\t\tCreateWorkflowModeContinueAsNew,\n\t}\n\n\tfor state, expectError := range stateToError {\n\t\ttestSnapshot := s.newTestWorkflowSnapshot(state)\n\t\tfor _, createMode := range creatModes {\n\t\t\terr := ValidateCreateWorkflowModeState(createMode, testSnapshot)\n\t\t\tif !expectError {\n\t\t\t\ts.NoError(err, err)\n\t\t\t} else {\n\t\t\t\ts.Error(err, err)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestCreateMode_BypassCurrent() {\n\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    false,\n\t}\n\n\tfor state, expectError := range stateToError {\n\t\ttestSnapshot := s.newTestWorkflowSnapshot(state)\n\t\terr := ValidateCreateWorkflowModeState(CreateWorkflowModeZombie, testSnapshot)\n\t\tif !expectError {\n\t\t\ts.NoError(err, err)\n\t\t} else {\n\t\t\ts.Error(err, err)\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestUpdateMode_UpdateCurrent() {\n\n\t// only current workflow\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tfor state, expectError := range stateToError {\n\t\ttestCurrentMutation := s.newTestWorkflowMutation(state)\n\t\terr := ValidateUpdateWorkflowModeState(\n\t\t\tUpdateWorkflowModeUpdateCurrent,\n\t\t\ttestCurrentMutation,\n\t\t\tnil,\n\t\t)\n\t\tif !expectError {\n\t\t\ts.NoError(err, err)\n\t\t} else {\n\t\t\ts.Error(err, err)\n\t\t}\n\t}\n\n\t// current workflow & new workflow\n\tcurrentStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tnewStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tfor currentState, currentExpectError := range currentStateToError {\n\t\tfor newState, newExpectError := range newStateToError {\n\t\t\ttestCurrentMutation := s.newTestWorkflowMutation(currentState)\n\t\t\ttestNewSnapshot := s.newTestWorkflowSnapshot(newState)\n\t\t\terr := ValidateUpdateWorkflowModeState(\n\t\t\t\tUpdateWorkflowModeUpdateCurrent,\n\t\t\t\ttestCurrentMutation,\n\t\t\t\t&testNewSnapshot,\n\t\t\t)\n\t\t\tif currentExpectError || newExpectError {\n\t\t\t\ts.Error(err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err, err)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestUpdateMode_BypassCurrent() {\n\n\t// only current workflow\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor state, expectError := range stateToError {\n\t\ttestMutation := s.newTestWorkflowMutation(state)\n\t\terr := ValidateUpdateWorkflowModeState(\n\t\t\tUpdateWorkflowModeBypassCurrent,\n\t\t\ttestMutation,\n\t\t\tnil,\n\t\t)\n\t\tif !expectError {\n\t\t\ts.NoError(err, err)\n\t\t} else {\n\t\t\ts.Error(err, err)\n\t\t}\n\t}\n\n\t// current workflow & new workflow\n\tcurrentStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tnewStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor currentState, currentExpectError := range currentStateToError {\n\t\tfor newState, newExpectError := range newStateToError {\n\t\t\ttestCurrentMutation := s.newTestWorkflowMutation(currentState)\n\t\t\ttestNewSnapshot := s.newTestWorkflowSnapshot(newState)\n\t\t\terr := ValidateUpdateWorkflowModeState(\n\t\t\t\tUpdateWorkflowModeBypassCurrent,\n\t\t\t\ttestCurrentMutation,\n\t\t\t\t&testNewSnapshot,\n\t\t\t)\n\t\t\tif currentExpectError || newExpectError {\n\t\t\t\ts.Error(err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err, err)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestUpdateMode_IgnoreCurrent() {\n\ttestMutation := s.newTestWorkflowMutation(WorkflowStateCompleted)\n\terr := ValidateUpdateWorkflowModeState(\n\t\tUpdateWorkflowModeIgnoreCurrent,\n\t\ttestMutation,\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\ttestNewSnapshot := s.newTestWorkflowSnapshot(WorkflowStateRunning)\n\terr = ValidateUpdateWorkflowModeState(\n\t\tUpdateWorkflowModeIgnoreCurrent,\n\t\ttestMutation,\n\t\t&testNewSnapshot,\n\t)\n\ts.Error(err)\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestConflictResolveMode_UpdateCurrent() {\n\n\t// only reset workflow\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tfor state, expectError := range stateToError {\n\t\ttestSnapshot := s.newTestWorkflowSnapshot(state)\n\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\tConflictResolveWorkflowModeUpdateCurrent,\n\t\t\ttestSnapshot,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\t\tif !expectError {\n\t\t\ts.NoError(err, err)\n\t\t} else {\n\t\t\ts.Error(err, err)\n\t\t}\n\t}\n\n\t// reset workflow & new workflow\n\tresetStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tnewStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tfor resetState, resetExpectError := range resetStateToError {\n\t\tfor newState, newExpectError := range newStateToError {\n\t\t\ttestResetSnapshot := s.newTestWorkflowSnapshot(resetState)\n\t\t\ttestNewSnapshot := s.newTestWorkflowSnapshot(newState)\n\t\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\t\tConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\ttestResetSnapshot,\n\t\t\t\t&testNewSnapshot,\n\t\t\t\tnil,\n\t\t\t)\n\t\t\tif resetExpectError || newExpectError {\n\t\t\t\ts.Error(err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err, err)\n\t\t\t}\n\t\t}\n\t}\n\n\t// reset workflow & current workflow\n\tresetStateToError = map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tcurrentStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor resetState, resetExpectError := range resetStateToError {\n\t\tfor currentState, currentExpectError := range currentStateToError {\n\t\t\ttestResetSnapshot := s.newTestWorkflowSnapshot(resetState)\n\t\t\ttestCurrentSnapshot := s.newTestWorkflowMutation(currentState)\n\t\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\t\tConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\ttestResetSnapshot,\n\t\t\t\tnil,\n\t\t\t\t&testCurrentSnapshot,\n\t\t\t)\n\t\t\tif resetExpectError || currentExpectError {\n\t\t\t\ts.Error(err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err, err)\n\t\t\t}\n\t\t}\n\t}\n\n\t// reset workflow & new workflow & current workflow\n\tresetStateToError = map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tnewStateToError = map[int]bool{\n\t\tWorkflowStateCreated:   false,\n\t\tWorkflowStateRunning:   false,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tcurrentStateToError = map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor resetState, resetExpectError := range resetStateToError {\n\t\tfor newState, newExpectError := range newStateToError {\n\t\t\tfor currentState, currentExpectError := range currentStateToError {\n\t\t\t\ttestResetSnapshot := s.newTestWorkflowSnapshot(resetState)\n\t\t\t\ttestNewSnapshot := s.newTestWorkflowSnapshot(newState)\n\t\t\t\ttestCurrentSnapshot := s.newTestWorkflowMutation(currentState)\n\t\t\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\t\t\tConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\t\ttestResetSnapshot,\n\t\t\t\t\t&testNewSnapshot,\n\t\t\t\t\t&testCurrentSnapshot,\n\t\t\t\t)\n\t\t\t\tif resetExpectError || newExpectError || currentExpectError {\n\t\t\t\t\ts.Error(err, err)\n\t\t\t\t} else {\n\t\t\t\t\ts.NoError(err, err)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) TestConflictResolveMode_BypassCurrent() {\n\n\t// only reset workflow\n\tstateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor state, expectError := range stateToError {\n\t\ttestSnapshot := s.newTestWorkflowSnapshot(state)\n\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\tConflictResolveWorkflowModeBypassCurrent,\n\t\t\ttestSnapshot,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\t\tif !expectError {\n\t\t\ts.NoError(err, err)\n\t\t} else {\n\t\t\ts.Error(err, err)\n\t\t}\n\t}\n\n\t// reset workflow & new workflow\n\tresetStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: false,\n\t\tWorkflowStateZombie:    true,\n\t}\n\tnewStateToError := map[int]bool{\n\t\tWorkflowStateCreated:   true,\n\t\tWorkflowStateRunning:   true,\n\t\tWorkflowStateCompleted: true,\n\t\tWorkflowStateZombie:    false,\n\t}\n\tfor resetState, resetExpectError := range resetStateToError {\n\t\tfor newState, newExpectError := range newStateToError {\n\t\t\ttestResetSnapshot := s.newTestWorkflowSnapshot(resetState)\n\t\t\ttestNewSnapshot := s.newTestWorkflowSnapshot(newState)\n\t\t\terr := ValidateConflictResolveWorkflowModeState(\n\t\t\t\tConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\ttestResetSnapshot,\n\t\t\t\t&testNewSnapshot,\n\t\t\t\tnil,\n\t\t\t)\n\t\t\tif resetExpectError || newExpectError {\n\t\t\t\tif err == nil {\n\t\t\t\t\tfmt.Print(\"##\")\n\t\t\t\t}\n\t\t\t\ts.Error(err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err, err)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) newTestWorkflowSnapshot(\n\tstate int,\n) InternalWorkflowSnapshot {\n\treturn InternalWorkflowSnapshot{\n\t\tExecutionInfo: &InternalWorkflowExecutionInfo{\n\t\t\tState: state,\n\t\t},\n\t}\n}\n\nfunc (s *validateOperationWorkflowModeStateSuite) newTestWorkflowMutation(\n\tstate int,\n) InternalWorkflowMutation {\n\treturn InternalWorkflowMutation{\n\t\tExecutionInfo: &InternalWorkflowExecutionInfo{\n\t\t\tState: state,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/configStorePersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Currently you cannot clear or remove any entries in cluster_config table\n// Therefore, Teardown and Setup of Test DB is required before every test.\n\ntype (\n\t// ConfigStorePersistenceSuite contains config store persistence tests\n\tConfigStorePersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\n// SetupSuite implementation\nfunc (s *ConfigStorePersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// SetupTest implementation\nfunc (s *ConfigStorePersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TearDownSuite implementation\nfunc (s *ConfigStorePersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// Tests if error is returned when trying to fetch dc values from empty table\nfunc (s *ConfigStorePersistenceSuite) TestFetchFromEmptyTable() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsnapshot, err := s.FetchDynamicConfig(ctx, 0)\n\ts.Nil(snapshot)\n\ts.NotNil(err)\n}\n\nfunc (s *ConfigStorePersistenceSuite) TestUpdateSimpleSuccess() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsnapshot := generateRandomSnapshot(1)\n\terr := s.UpdateDynamicConfig(ctx, snapshot, 1)\n\ts.Nil(err)\n\n\tretSnapshot, err := s.FetchDynamicConfig(ctx, 1)\n\ts.NotNil(retSnapshot)\n\ts.Nil(err)\n\ts.Equal(snapshot.Version, retSnapshot.Version)\n\ts.Equal(snapshot.Values.Entries[0].Name, retSnapshot.Values.Entries[0].Name)\n}\n\nfunc (s *ConfigStorePersistenceSuite) TestUpdateVersionCollisionFailure() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsnapshot := generateRandomSnapshot(1)\n\terr := s.UpdateDynamicConfig(ctx, snapshot, 2)\n\ts.Nil(err)\n\n\terr = s.UpdateDynamicConfig(ctx, snapshot, 2)\n\tvar condErr *p.ConditionFailedError\n\ts.True(errors.As(err, &condErr))\n}\n\nfunc (s *ConfigStorePersistenceSuite) TestUpdateIncrementalVersionSuccess() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsnapshot2 := generateRandomSnapshot(2)\n\terr := s.UpdateDynamicConfig(ctx, snapshot2, 3)\n\ts.Nil(err)\n\tsnapshot3 := generateRandomSnapshot(3)\n\terr = s.UpdateDynamicConfig(ctx, snapshot3, 3)\n\ts.Nil(err)\n}\n\nfunc (s *ConfigStorePersistenceSuite) TestFetchLatestVersionSuccess() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsnapshot2 := generateRandomSnapshot(2)\n\terr := s.UpdateDynamicConfig(ctx, snapshot2, 4)\n\ts.Nil(err)\n\tsnapshot3 := generateRandomSnapshot(3)\n\terr = s.UpdateDynamicConfig(ctx, snapshot3, 4)\n\ts.Nil(err)\n\n\tsnapshot, err := s.FetchDynamicConfig(ctx, 4)\n\ts.NotNil(snapshot)\n\ts.Nil(err)\n\ts.Equal(int64(3), snapshot.Version)\n}\n\nfunc generateRandomSnapshot(version int64) *p.DynamicConfigSnapshot {\n\tdata, _ := json.Marshal(\"test_value\")\n\n\tvalues := make([]*types.DynamicConfigValue, 1)\n\tvalues[0] = &types.DynamicConfigValue{\n\t\tValue: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         data,\n\t\t},\n\t\tFilters: nil,\n\t}\n\n\tentries := make([]*types.DynamicConfigEntry, 1)\n\tentries[0] = &types.DynamicConfigEntry{\n\t\tName:   \"test_parameter\",\n\t\tValues: values,\n\t}\n\n\treturn &p.DynamicConfigSnapshot{\n\t\tVersion: version,\n\t\tValues: &types.DynamicConfigBlob{\n\t\t\tSchemaVersion: 1,\n\t\t\tEntries:       entries,\n\t\t},\n\t}\n}\n\nfunc (s *ConfigStorePersistenceSuite) FetchDynamicConfig(ctx context.Context, rowType int) (*p.DynamicConfigSnapshot, error) {\n\tresponse, err := s.ConfigStoreManager.FetchDynamicConfig(ctx, p.ConfigType(rowType))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif response == nil {\n\t\treturn nil, errors.New(\"nil FetchDynamicConfig response\")\n\t}\n\treturn response.Snapshot, nil\n}\n\nfunc (s *ConfigStorePersistenceSuite) UpdateDynamicConfig(ctx context.Context, snapshot *p.DynamicConfigSnapshot, rowType int) error {\n\treturn s.ConfigStoreManager.UpdateDynamicConfig(ctx, &p.UpdateDynamicConfigRequest{Snapshot: snapshot}, p.ConfigType(rowType))\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/dbVisibilityPersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// DBVisibilityPersistenceSuite tests visibility persistence\n\t// It only tests against DB based visibility, AdvancedVisibility test is in ESVisibilitySuite\n\tDBVisibilityPersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t\tVisibilityMgr p.VisibilityManager\n\t}\n)\n\n// SetupSuite implementation\nfunc (s *DBVisibilityPersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n\t// setup visibility manager\n\tif s.VisibilityTestCluster != s.DefaultTestCluster {\n\t\ts.VisibilityTestCluster.SetupTestDatabase()\n\t}\n\tclusterName := s.ClusterMetadata.GetCurrentClusterName()\n\tvCfg := s.VisibilityTestCluster.Config()\n\tvisibilityFactory := client.NewFactory(&vCfg, nil, clusterName, nil, s.Logger, &s.DynamicConfiguration)\n\t// SQL currently doesn't have support for visibility manager\n\tvar err error\n\ts.VisibilityMgr, err = visibilityFactory.NewVisibilityManager(\n\t\t&client.Params{\n\t\t\tPersistenceConfig: config.Persistence{\n\t\t\t\tVisibilityStore: \"something not empty\",\n\t\t\t},\n\t\t},\n\t\t&service.Config{\n\t\t\tReadVisibilityStoreName:                     dynamicproperties.GetStringPropertyFnFilteredByDomain(\"db\"),\n\t\t\tWriteVisibilityStoreName:                    dynamicproperties.GetStringPropertyFn(\"db\"),\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\tEnableDBVisibilitySampling:                  dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t)\n\tif err != nil {\n\t\ts.fatalOnError(\"NewVisibilityManager\", err)\n\t}\n}\n\n// SetupTest implementation\nfunc (s *DBVisibilityPersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TearDownSuite implementation\nfunc (s *DBVisibilityPersistenceSuite) TearDownSuite() {\n\t// TODO VisibilityMgr/Store is created with a separated code path, this is incorrect and may cause leaking connection\n\t// And Postgres requires all connection to be closed before dropping a database\n\t// https://github.com/uber/cadence/issues/2854\n\t// Remove the below line after the issue is fix\n\ts.VisibilityMgr.Close()\n\n\ts.TearDownWorkflowStore()\n}\n\n// TestBasicVisibility test\nfunc (s *DBVisibilityPersistenceSuite) TestBasicVisibility() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\tstartReq := &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tUpdateTimestamp:  0,\n\t\tShardID:          0,\n\t}\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, startReq)\n\ts.Nil(err0)\n\n\tresp, err1 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err1)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertOpenExecutionEquals(startReq, resp.Executions[0])\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   time.Now().UnixNano(),\n\t\tUpdateTimestamp:  time.Now().UnixNano(),\n\t\tHistoryLength:    5,\n\t\tShardID:          1234,\n\t}\n\terr2 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err2)\n\n\tresp, err3 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err3)\n\ts.Equal(0, len(resp.Executions))\n\n\tresp, err4 := s.VisibilityMgr.ListClosedWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err4)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertClosedExecutionEquals(closeReq, resp.Executions[0])\n\n\tuninitializedReq := &p.RecordWorkflowExecutionUninitializedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tUpdateTimestamp:  time.Now().UnixNano(),\n\t\tShardID:          1234,\n\t}\n\terr5 := s.VisibilityMgr.RecordWorkflowExecutionUninitialized(ctx, uninitializedReq)\n\ts.Nil(err5)\n}\n\n// TestCronVisibility test\nfunc (s *DBVisibilityPersistenceSuite) TestCronVisibility() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-cron-workflow-test\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c537\",\n\t}\n\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\tstartReq := &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-cron-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tIsCron:           true,\n\t\tShardID:          1234,\n\t}\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, startReq)\n\ts.Nil(err0)\n\n\tresp, err1 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err1)\n\ts.Equal(1, len(resp.Executions))\n\ts.True(resp.Executions[0].IsCron)\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   time.Now().UnixNano(),\n\t\tHistoryLength:    5,\n\t\tIsCron:           true,\n\t\tShardID:          1234,\n\t}\n\terr2 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err2)\n\n\tresp, err4 := s.VisibilityMgr.ListClosedWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err4)\n\ts.Equal(1, len(resp.Executions))\n\ts.True(resp.Executions[0].IsCron)\n}\n\n// TestBasicVisibilityTimeSkew test\nfunc (s *DBVisibilityPersistenceSuite) TestBasicVisibilityTimeSkew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test-time-skew\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err0)\n\n\tresp, err1 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err1)\n\ts.Equal(1, len(resp.Executions))\n\ts.Equal(workflowExecution.WorkflowID, resp.Executions[0].Execution.WorkflowID)\n\n\terr2 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   startTime - (10 * time.Second).Nanoseconds(),\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err2)\n\n\tresp, err3 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err3)\n\ts.Equal(0, len(resp.Executions))\n\n\tresp, err4 := s.VisibilityMgr.ListClosedWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   startTime,\n\t})\n\ts.Nil(err4)\n\ts.Equal(1, len(resp.Executions))\n}\n\n// TestVisibilityPagination test\nfunc (s *DBVisibilityPersistenceSuite) TestVisibilityPagination() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\t// Create 2 executions\n\tstartTime1 := time.Now()\n\tworkflowExecution1 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-pagination-test1\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\n\tstartReq1 := &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime1.UnixNano(),\n\t\tShardID:          1234,\n\t}\n\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, startReq1)\n\ts.Nil(err0)\n\n\tstartTime2 := startTime1.Add(time.Second)\n\tworkflowExecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-pagination-test2\",\n\t\tRunID:      \"843f6fc7-102a-4c63-a2d4-7c653b01bf52\",\n\t}\n\n\tstartReq2 := &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime2.UnixNano(),\n\t\tShardID:          1234,\n\t}\n\terr1 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, startReq2)\n\ts.Nil(err1)\n\n\t// Get the first one\n\tresp, err2 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tPageSize:     1,\n\t\tEarliestTime: startTime1.UnixNano(),\n\t\tLatestTime:   startTime2.UnixNano(),\n\t})\n\ts.Nil(err2)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertOpenExecutionEquals(startReq2, resp.Executions[0])\n\n\t// Use token to get the second one\n\tresp, err3 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainUUID,\n\t\tPageSize:      1,\n\t\tEarliestTime:  startTime1.UnixNano(),\n\t\tLatestTime:    startTime2.UnixNano(),\n\t\tNextPageToken: resp.NextPageToken,\n\t})\n\ts.Nil(err3)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertOpenExecutionEquals(startReq1, resp.Executions[0])\n\n\t// It is possible to not return non empty token which is going to return empty result\n\tif len(resp.NextPageToken) != 0 {\n\t\t// Now should get empty result by using token\n\t\tresp, err4 := s.VisibilityMgr.ListOpenWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:    testDomainUUID,\n\t\t\tPageSize:      1,\n\t\t\tEarliestTime:  startTime1.UnixNano(),\n\t\t\tLatestTime:    startTime2.UnixNano(),\n\t\t\tNextPageToken: resp.NextPageToken,\n\t\t})\n\t\ts.Nil(err4)\n\t\ts.Equal(0, len(resp.Executions))\n\t}\n}\n\n// TestFilteringByType test\nfunc (s *DBVisibilityPersistenceSuite) TestFilteringByType() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\tstartTime := time.Now().UnixNano()\n\n\t// Create 2 executions\n\tworkflowExecution1 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test1\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow-1\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err0)\n\n\tworkflowExecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test2\",\n\t\tRunID:      \"843f6fc7-102a-4c63-a2d4-7c653b01bf52\",\n\t}\n\terr1 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow-2\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err1)\n\n\t// List open with filtering\n\tresp, err2 := s.VisibilityMgr.ListOpenWorkflowExecutionsByType(ctx, &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tPageSize:     2,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   startTime,\n\t\t},\n\t\tWorkflowTypeName: \"visibility-workflow-1\",\n\t})\n\ts.Nil(err2)\n\ts.Equal(1, len(resp.Executions))\n\ts.Equal(workflowExecution1.WorkflowID, resp.Executions[0].Execution.WorkflowID)\n\n\tstopTime := time.Now().UnixNano()\n\n\t// Close both executions\n\terr3 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow-1\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   stopTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err3)\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow-2\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   stopTime,\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\terr4 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err4)\n\n\t// List closed with filtering\n\tresp, err5 := s.VisibilityMgr.ListClosedWorkflowExecutionsByType(ctx, &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tPageSize:     2,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   startTime,\n\t\t},\n\t\tWorkflowTypeName: \"visibility-workflow-2\",\n\t})\n\ts.Nil(err5)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertClosedExecutionEquals(closeReq, resp.Executions[0])\n}\n\n// TestFilteringByWorkflowID test\nfunc (s *DBVisibilityPersistenceSuite) TestFilteringByWorkflowID() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\tstartTime := time.Now().UnixNano()\n\n\t// Create 2 executions\n\tworkflowExecution1 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test1\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err0)\n\n\tworkflowExecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test2\",\n\t\tRunID:      \"843f6fc7-102a-4c63-a2d4-7c653b01bf52\",\n\t}\n\terr1 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err1)\n\n\t// List open with filtering\n\tresp, err2 := s.VisibilityMgr.ListOpenWorkflowExecutionsByWorkflowID(ctx, &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tPageSize:     2,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   startTime,\n\t\t},\n\t\tWorkflowID: \"visibility-filtering-test1\",\n\t})\n\ts.Nil(err2)\n\ts.Equal(1, len(resp.Executions))\n\ts.Equal(workflowExecution1.WorkflowID, resp.Executions[0].Execution.WorkflowID)\n\n\tstopTime := time.Now().UnixNano()\n\n\t// Close both executions\n\terr3 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   stopTime,\n\t})\n\ts.Nil(err3)\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   stopTime,\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\terr4 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err4)\n\n\t// List closed with filtering\n\tresp, err5 := s.VisibilityMgr.ListClosedWorkflowExecutionsByWorkflowID(ctx, &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tPageSize:     2,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   startTime,\n\t\t},\n\t\tWorkflowID: \"visibility-filtering-test2\",\n\t})\n\ts.Nil(err5)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertClosedExecutionEquals(closeReq, resp.Executions[0])\n}\n\n// TestFilteringByCloseStatus test\nfunc (s *DBVisibilityPersistenceSuite) TestFilteringByCloseStatus() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\tstartTime := time.Now().UnixNano()\n\n\t// Create 2 executions\n\tworkflowExecution1 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test1\",\n\t\tRunID:      \"fb15e4b5-356f-466d-8c6d-a29223e5c536\",\n\t}\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err0)\n\n\tworkflowExecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-filtering-test2\",\n\t\tRunID:      \"843f6fc7-102a-4c63-a2d4-7c653b01bf52\",\n\t}\n\terr1 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err1)\n\n\tstopTime := time.Now().UnixNano()\n\n\t// Close both executions with different status\n\terr2 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution1,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tCloseTimestamp:   stopTime,\n\t\tStatus:           types.WorkflowExecutionCloseStatusCompleted,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err2)\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution2,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t\tCloseTimestamp:   stopTime,\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\terr3 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err3)\n\n\t// List closed with filtering\n\tresp, err4 := s.VisibilityMgr.ListClosedWorkflowExecutionsByStatus(ctx, &p.ListClosedWorkflowExecutionsByStatusRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tPageSize:     2,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   startTime,\n\t\t},\n\t\tStatus: types.WorkflowExecutionCloseStatusFailed,\n\t})\n\ts.Nil(err4)\n\ts.Equal(1, len(resp.Executions))\n\ts.assertClosedExecutionEquals(closeReq, resp.Executions[0])\n}\n\n// TestGetClosedExecution test\nfunc (s *DBVisibilityPersistenceSuite) TestGetClosedExecution() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test\",\n\t\tRunID:      \"a3dbc7bf-deb1-4946-b57c-cf0615ea553f\",\n\t}\n\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tShardID:          1234,\n\t})\n\ts.Nil(err0)\n\n\tclosedResp, err1 := s.VisibilityMgr.GetClosedWorkflowExecution(ctx, &p.GetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tExecution:  workflowExecution,\n\t})\n\ts.Error(err1)\n\t_, ok := err1.(*types.EntityNotExistsError)\n\ts.True(ok, \"EntityNotExistsError\")\n\ts.Nil(closedResp)\n\n\tstopTime := time.Now().UnixNano()\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t\tCloseTimestamp:   stopTime,\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\terr2 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err2)\n\n\tresp, err3 := s.VisibilityMgr.GetClosedWorkflowExecution(ctx, &p.GetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tExecution:  workflowExecution,\n\t})\n\ts.Nil(err3)\n\ts.assertClosedExecutionEquals(closeReq, resp.Execution)\n}\n\n// TestClosedWithoutStarted test\nfunc (s *DBVisibilityPersistenceSuite) TestClosedWithoutStarted() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test\",\n\t\tRunID:      \"1bdb0122-e8c9-4b35-b6f8-d692ab259b09\",\n\t}\n\n\tclosedResp, err0 := s.VisibilityMgr.GetClosedWorkflowExecution(ctx, &p.GetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tExecution:  workflowExecution,\n\t})\n\ts.Error(err0)\n\t_, ok := err0.(*types.EntityNotExistsError)\n\ts.True(ok, \"EntityNotExistsError\")\n\ts.Nil(closedResp)\n\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   time.Now().Add(time.Second * -5).UnixNano(),\n\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t\tCloseTimestamp:   time.Now().UnixNano(),\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\terr1 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\ts.Nil(err1)\n\n\tresp, err2 := s.VisibilityMgr.GetClosedWorkflowExecution(ctx, &p.GetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tExecution:  workflowExecution,\n\t})\n\ts.Nil(err2)\n\ts.assertClosedExecutionEquals(closeReq, resp.Execution)\n}\n\n// TestMultipleUpserts test\nfunc (s *DBVisibilityPersistenceSuite) TestMultipleUpserts() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttestDomainUUID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test\",\n\t\tRunID:      \"a3dbc7bf-deb1-4946-b57c-cf0615ea553f\",\n\t}\n\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\tStartTimestamp:   startTime,\n\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t\tCloseTimestamp:   time.Now().UnixNano(),\n\t\tHistoryLength:    3,\n\t\tShardID:          1234,\n\t}\n\n\tcount := 3\n\tfor i := 0; i < count; i++ {\n\t\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\t\tDomainUUID:       testDomainUUID,\n\t\t\tExecution:        workflowExecution,\n\t\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\t\tStartTimestamp:   startTime,\n\t\t\tShardID:          1234,\n\t\t})\n\t\ts.Nil(err0)\n\t\tif i < count-1 {\n\t\t\terr1 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\t\t\ts.Nil(err1)\n\t\t}\n\t}\n\n\tresp, err3 := s.VisibilityMgr.GetClosedWorkflowExecution(ctx, &p.GetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tExecution:  workflowExecution,\n\t})\n\ts.Nil(err3)\n\ts.assertClosedExecutionEquals(closeReq, resp.Execution)\n\n}\n\n// TestDelete test\nfunc (s *DBVisibilityPersistenceSuite) TestDelete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tif s.VisibilityMgr.GetName() == \"cassandra\" {\n\t\t// this test is not applicable for cassandra\n\t\treturn\n\t}\n\tnRows := 5\n\ttestDomainUUID := uuid.New()\n\tstartTime := time.Now().Add(time.Second * -5).UnixNano()\n\tfor i := 0; i < nRows; i++ {\n\t\tworkflowExecution := types.WorkflowExecution{\n\t\t\tWorkflowID: uuid.New(),\n\t\t\tRunID:      uuid.New(),\n\t\t}\n\t\terr0 := s.VisibilityMgr.RecordWorkflowExecutionStarted(ctx, &p.RecordWorkflowExecutionStartedRequest{\n\t\t\tDomainUUID:       testDomainUUID,\n\t\t\tExecution:        workflowExecution,\n\t\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\t\tStartTimestamp:   startTime,\n\t\t})\n\t\ts.Nil(err0)\n\t\tcloseReq := &p.RecordWorkflowExecutionClosedRequest{\n\t\t\tDomainUUID:       testDomainUUID,\n\t\t\tExecution:        workflowExecution,\n\t\t\tWorkflowTypeName: \"visibility-workflow\",\n\t\t\tStartTimestamp:   startTime,\n\t\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t\t\tCloseTimestamp:   time.Now().UnixNano(),\n\t\t\tHistoryLength:    3,\n\t\t\tShardID:          1234,\n\t\t}\n\t\terr1 := s.VisibilityMgr.RecordWorkflowExecutionClosed(ctx, closeReq)\n\t\ts.Nil(err1)\n\t}\n\n\tresp, err3 := s.VisibilityMgr.ListClosedWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:   testDomainUUID,\n\t\tEarliestTime: startTime,\n\t\tLatestTime:   time.Now().UnixNano(),\n\t\tPageSize:     10,\n\t})\n\ts.Nil(err3)\n\ts.Equal(nRows, len(resp.Executions))\n\n\tremaining := nRows\n\tfor _, row := range resp.Executions {\n\t\terr4 := s.VisibilityMgr.DeleteWorkflowExecution(ctx, &p.VisibilityDeleteWorkflowExecutionRequest{\n\t\t\tDomainID: testDomainUUID,\n\t\t\tRunID:    row.GetExecution().GetRunID(),\n\t\t})\n\t\ts.Nil(err4)\n\t\tremaining--\n\t\tresp, err5 := s.VisibilityMgr.ListClosedWorkflowExecutions(ctx, &p.ListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:   testDomainUUID,\n\t\t\tEarliestTime: startTime,\n\t\t\tLatestTime:   time.Now().UnixNano(),\n\t\t\tPageSize:     10,\n\t\t})\n\t\ts.Nil(err5)\n\t\ts.Equal(remaining, len(resp.Executions))\n\t}\n}\n\n// TestUpsertWorkflowExecution test\nfunc (s *DBVisibilityPersistenceSuite) TestUpsertWorkflowExecution() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttests := []struct {\n\t\trequest  *p.UpsertWorkflowExecutionRequest\n\t\texpected error\n\t}{\n\t\t{\n\t\t\trequest: &p.UpsertWorkflowExecutionRequest{\n\t\t\t\tDomainUUID:         \"\",\n\t\t\t\tDomain:             \"\",\n\t\t\t\tExecution:          types.WorkflowExecution{},\n\t\t\t\tWorkflowTypeName:   \"\",\n\t\t\t\tStartTimestamp:     0,\n\t\t\t\tExecutionTimestamp: 0,\n\t\t\t\tWorkflowTimeout:    0,\n\t\t\t\tTaskID:             0,\n\t\t\t\tMemo:               nil,\n\t\t\t\tSearchAttributes: map[string][]byte{\n\t\t\t\t\tdefinition.CadenceChangeVersion: []byte(\"dummy\"),\n\t\t\t\t},\n\t\t\t\tShardID: 1234,\n\t\t\t},\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\trequest: &p.UpsertWorkflowExecutionRequest{\n\t\t\t\tDomainUUID:         \"\",\n\t\t\t\tDomain:             \"\",\n\t\t\t\tExecution:          types.WorkflowExecution{},\n\t\t\t\tWorkflowTypeName:   \"\",\n\t\t\t\tStartTimestamp:     0,\n\t\t\t\tExecutionTimestamp: 0,\n\t\t\t\tWorkflowTimeout:    0,\n\t\t\t\tTaskID:             0,\n\t\t\t\tMemo:               nil,\n\t\t\t\tSearchAttributes:   nil,\n\t\t\t\tShardID:            1234,\n\t\t\t},\n\t\t\texpected: &types.InternalServiceError{\n\t\t\t\tMessage: \"Error writing to visibility: Operation is not supported\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\terr := s.VisibilityMgr.UpsertWorkflowExecution(ctx, test.request)\n\t\tif test.expected == nil {\n\t\t\ts.Equal(test.expected, err)\n\t\t} else {\n\t\t\ts.Equal(test.expected.Error(), err.Error())\n\t\t}\n\t}\n}\n\nfunc (s *DBVisibilityPersistenceSuite) assertClosedExecutionEquals(\n\treq *p.RecordWorkflowExecutionClosedRequest, resp *types.WorkflowExecutionInfo) {\n\ts.Equal(req.Execution.RunID, resp.Execution.RunID)\n\ts.Equal(req.Execution.WorkflowID, resp.Execution.WorkflowID)\n\ts.Equal(req.WorkflowTypeName, resp.GetType().GetName())\n\ts.Equal(s.nanosToMillis(req.StartTimestamp), s.nanosToMillis(resp.GetStartTime()))\n\ts.Equal(s.nanosToMillis(req.CloseTimestamp), s.nanosToMillis(resp.GetCloseTime()))\n\ts.Equal(req.Status, resp.GetCloseStatus())\n\ts.Equal(req.HistoryLength, resp.HistoryLength)\n}\n\nfunc (s *DBVisibilityPersistenceSuite) assertOpenExecutionEquals(\n\treq *p.RecordWorkflowExecutionStartedRequest, resp *types.WorkflowExecutionInfo) {\n\ts.Equal(req.Execution.GetRunID(), resp.Execution.GetRunID())\n\ts.Equal(req.Execution.WorkflowID, resp.Execution.WorkflowID)\n\ts.Equal(req.WorkflowTypeName, resp.GetType().GetName())\n\ts.Equal(s.nanosToMillis(req.StartTimestamp), s.nanosToMillis(resp.GetStartTime()))\n\ts.Nil(resp.CloseTime)\n\ts.Nil(resp.CloseStatus)\n\ts.Zero(resp.HistoryLength)\n}\n\nfunc (s *DBVisibilityPersistenceSuite) nanosToMillis(nanos int64) int64 {\n\treturn nanos / int64(time.Millisecond)\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/domainAuditPersistenceTest.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tDomainAuditPersistenceSuite struct {\n\t\t*TestBase\n\t\t*require.Assertions\n\t}\n)\n\nfunc (s *DomainAuditPersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\nfunc (s *DomainAuditPersistenceSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *DomainAuditPersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestCreateAndGetDomainAuditLog() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\teventID := generateUUIDv7().String()\n\tnow := time.Now().UTC()\n\n\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\tDomainID:      domainID,\n\t\tEventID:       eventID,\n\t\tOperationType: persistence.DomainAuditOperationTypeCreate,\n\t\tCreatedTime:   now,\n\t\tIdentity:      \"test-user\",\n\t\tIdentityType:  \"user\",\n\t\tComment:       \"Test domain creation\",\n\t\tStateAfter: &persistence.GetDomainResponse{\n\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\tID:          domainID,\n\t\t\t\tName:        \"test-domain\",\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"Test domain\",\n\t\t\t},\n\t\t},\n\t}\n\n\tcreateResp, err := manager.CreateDomainAuditLog(ctx, createReq)\n\ts.NoError(err)\n\ts.NotNil(createResp)\n\ts.Equal(eventID, createResp.EventID)\n\n\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:      domainID,\n\t\tOperationType: persistence.DomainAuditOperationTypeCreate,\n\t\tPageSize:      10,\n\t}\n\n\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\ts.NoError(err)\n\ts.NotNil(getResp)\n\ts.Len(getResp.AuditLogs, 1)\n\n\tauditLog := getResp.AuditLogs[0]\n\ts.Equal(eventID, auditLog.EventID)\n\ts.Equal(domainID, auditLog.DomainID)\n\ts.Equal(persistence.DomainAuditOperationTypeCreate, auditLog.OperationType)\n\ts.Equal(\"test-user\", auditLog.Identity)\n\ts.Equal(\"user\", auditLog.IdentityType)\n\ts.Equal(\"Test domain creation\", auditLog.Comment)\n\ts.Nil(auditLog.StateBefore)\n\ts.NotNil(auditLog.StateAfter)\n\ts.Equal(domainID, auditLog.StateAfter.Info.ID)\n\ts.Equal(\"test-domain\", auditLog.StateAfter.Info.Name)\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestCreateMultipleAuditLogs() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\tnow := time.Now().UTC()\n\n\tfor i := 0; i < 5; i++ {\n\t\teventID := generateUUIDv7().String()\n\t\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tEventID:       eventID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\t\tCreatedTime:   now.Add(time.Duration(i) * time.Millisecond),\n\t\t\tIdentity:      \"test-user\",\n\t\t\tIdentityType:  \"user\",\n\t\t\tComment:       \"Update operation\",\n\t\t}\n\n\t\t_, err := manager.CreateDomainAuditLog(ctx, createReq)\n\t\ts.NoError(err)\n\t\ttime.Sleep(2 * time.Millisecond)\n\t}\n\n\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:      domainID,\n\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\tPageSize:      10,\n\t}\n\n\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\ts.NoError(err)\n\ts.NotNil(getResp)\n\ts.Len(getResp.AuditLogs, 5)\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestGetDomainAuditLogsPagination() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\tnow := time.Now().UTC()\n\tnumLogs := 10\n\n\tfor i := 0; i < numLogs; i++ {\n\t\teventID := generateUUIDv7().String()\n\t\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tEventID:       eventID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\t\tCreatedTime:   now.Add(time.Duration(i) * time.Millisecond),\n\t\t\tIdentity:      \"test-user\",\n\t\t\tIdentityType:  \"user\",\n\t\t}\n\n\t\t_, err := manager.CreateDomainAuditLog(ctx, createReq)\n\t\ts.NoError(err)\n\t\ttime.Sleep(2 * time.Millisecond)\n\t}\n\n\tpageSize := 3\n\tvar allLogs []*persistence.DomainAuditLog\n\tvar nextPageToken []byte\n\n\tfor {\n\t\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\t\tPageSize:      pageSize,\n\t\t\tNextPageToken: nextPageToken,\n\t\t}\n\n\t\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\t\ts.NoError(err)\n\t\ts.NotNil(getResp)\n\n\t\tallLogs = append(allLogs, getResp.AuditLogs...)\n\n\t\tif len(getResp.NextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t\tnextPageToken = getResp.NextPageToken\n\t}\n\n\ts.Len(allLogs, numLogs)\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestGetDomainAuditLogsByOperationType() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\tnow := time.Now().UTC()\n\n\tfor i := 0; i < 3; i++ {\n\t\teventID := generateUUIDv7().String()\n\t\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tEventID:       eventID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeCreate,\n\t\t\tCreatedTime:   now.Add(time.Duration(i) * time.Millisecond),\n\t\t\tIdentity:      \"test-user\",\n\t\t\tIdentityType:  \"user\",\n\t\t}\n\t\t_, err := manager.CreateDomainAuditLog(ctx, createReq)\n\t\ts.NoError(err)\n\t\ttime.Sleep(2 * time.Millisecond)\n\t}\n\n\tfor i := 0; i < 5; i++ {\n\t\teventID := generateUUIDv7().String()\n\t\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tEventID:       eventID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\t\tCreatedTime:   now.Add(time.Duration(i+10) * time.Millisecond),\n\t\t\tIdentity:      \"test-user\",\n\t\t\tIdentityType:  \"user\",\n\t\t}\n\t\t_, err := manager.CreateDomainAuditLog(ctx, createReq)\n\t\ts.NoError(err)\n\t\ttime.Sleep(2 * time.Millisecond)\n\t}\n\n\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:      domainID,\n\t\tOperationType: persistence.DomainAuditOperationTypeCreate,\n\t\tPageSize:      10,\n\t}\n\n\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\ts.NoError(err)\n\ts.NotNil(getResp)\n\ts.Len(getResp.AuditLogs, 3)\n\tfor _, log := range getResp.AuditLogs {\n\t\ts.Equal(persistence.DomainAuditOperationTypeCreate, log.OperationType)\n\t}\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestGetDomainAuditLogsByTimeRange() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\tbaseTime := time.Now().UTC()\n\n\tfor i := 0; i < 10; i++ {\n\t\teventID := generateUUIDv7().String()\n\t\tcreatedTime := baseTime.Add(time.Duration(i) * time.Second)\n\t\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tEventID:       eventID,\n\t\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\t\tCreatedTime:   createdTime,\n\t\t\tIdentity:      \"test-user\",\n\t\t\tIdentityType:  \"user\",\n\t\t}\n\t\t_, err := manager.CreateDomainAuditLog(ctx, createReq)\n\t\ts.NoError(err)\n\t\ttime.Sleep(2 * time.Millisecond)\n\t}\n\n\tminTime := baseTime.Add(3 * time.Second)\n\tmaxTime := baseTime.Add(7 * time.Second)\n\n\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:       domainID,\n\t\tOperationType:  persistence.DomainAuditOperationTypeUpdate,\n\t\tMinCreatedTime: &minTime,\n\t\tMaxCreatedTime: &maxTime,\n\t\tPageSize:       10,\n\t}\n\n\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\ts.NoError(err)\n\ts.NotNil(getResp)\n\ts.True(len(getResp.AuditLogs) >= 4 && len(getResp.AuditLogs) <= 5)\n\n\tfor _, log := range getResp.AuditLogs {\n\t\ts.True(!log.CreatedTime.Truncate(time.Millisecond).Before(minTime.Truncate(time.Millisecond)))\n\t\ts.True(!log.CreatedTime.Truncate(time.Millisecond).After(maxTime.Truncate(time.Millisecond)))\n\t}\n}\n\nfunc (s *DomainAuditPersistenceSuite) TestDomainAuditLogWithStateBefore() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmanager, err := s.ExecutionMgrFactory.NewDomainAuditManager()\n\ts.NoError(err)\n\ts.NotNil(manager)\n\tdefer manager.Close()\n\n\tdomainID := uuid.NewString()\n\teventID := generateUUIDv7().String()\n\tnow := time.Now().UTC()\n\n\tcreateReq := &persistence.CreateDomainAuditLogRequest{\n\t\tDomainID:      domainID,\n\t\tEventID:       eventID,\n\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\tCreatedTime:   now,\n\t\tIdentity:      \"test-user\",\n\t\tIdentityType:  \"user\",\n\t\tComment:       \"Update domain description\",\n\t\tStateBefore: &persistence.GetDomainResponse{\n\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\tID:          domainID,\n\t\t\t\tName:        \"test-domain\",\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"Old description\",\n\t\t\t},\n\t\t},\n\t\tStateAfter: &persistence.GetDomainResponse{\n\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\tID:          domainID,\n\t\t\t\tName:        \"test-domain\",\n\t\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\t\tDescription: \"New description\",\n\t\t\t},\n\t\t},\n\t}\n\n\t_, err = manager.CreateDomainAuditLog(ctx, createReq)\n\ts.NoError(err)\n\n\tgetReq := &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:      domainID,\n\t\tOperationType: persistence.DomainAuditOperationTypeUpdate,\n\t\tPageSize:      10,\n\t}\n\n\tgetResp, err := manager.GetDomainAuditLogs(ctx, getReq)\n\ts.NoError(err)\n\ts.NotNil(getResp)\n\ts.Len(getResp.AuditLogs, 1)\n\n\tauditLog := getResp.AuditLogs[0]\n\ts.NotNil(auditLog.StateBefore)\n\ts.NotNil(auditLog.StateAfter)\n\ts.Equal(\"Old description\", auditLog.StateBefore.Info.Description)\n\ts.Equal(\"New description\", auditLog.StateAfter.Info.Description)\n}\n\nfunc generateUUIDv7() uuid.UUID {\n\tid, err := uuid.NewUUID()\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tnow := time.Now()\n\tunixMillis := now.UnixMilli()\n\n\tid[0] = byte(unixMillis >> 40)\n\tid[1] = byte(unixMillis >> 32)\n\tid[2] = byte(unixMillis >> 24)\n\tid[3] = byte(unixMillis >> 16)\n\tid[4] = byte(unixMillis >> 8)\n\tid[5] = byte(unixMillis)\n\n\tid[6] = (id[6] & 0x0f) | 0x70\n\tid[8] = (id[8] & 0x3f) | 0x80\n\n\treturn id\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/executionManagerTest.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"log\"\n\t\"math\"\n\t\"math/rand\"\n\t\"os\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ExecutionManagerSuite contains matching persistence tests\n\tExecutionManagerSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\nvar (\n\ttestContextTimeout      = 5 * time.Second\n\tlargeTestContextTimeout = 30 * time.Second\n\n\ttestWorkflowChecksum = checksum.Checksum{\n\t\tVersion: 22,\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tValue:   []byte(\"test-checksum\"),\n\t}\n)\n\n// SetupSuite implementation\nfunc (s *ExecutionManagerSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// TearDownSuite implementation\nfunc (s *ExecutionManagerSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// SetupTest implementation\nfunc (s *ExecutionManagerSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.ClearTasks()\n}\n\nfunc (s *ExecutionManagerSuite) newRandomChecksum() checksum.Checksum {\n\treturn checksum.Checksum{\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tVersion: 22,\n\t\tValue:   uuid.NewRandom(),\n\t}\n}\n\nfunc (s *ExecutionManagerSuite) assertChecksumsEqual(expected checksum.Checksum, actual checksum.Checksum) {\n\ts.EqualValues(expected, actual)\n}\n\n// TestCreateWorkflowExecutionDeDup test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionDeDup() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tdomainName := uuid.New()\n\tworkflowID := \"create-workflow-test-dedup\"\n\trunID := \"3969fae6-6b75-4c2a-b74b-4054edd296a6\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowID,\n\t\t\t\tRunID:                       runID,\n\t\t\t\tFirstExecutionRunID:         runID,\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateCreated,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:    s.ShardInfo.RangeID,\n\t\tMode:       p.CreateWorkflowModeBrandNew,\n\t\tDomainName: domainName,\n\t}\n\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\tupdatedInfo := copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats := copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateCompleted\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusCompleted\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.NoError(err)\n\n\treq.Mode = p.CreateWorkflowModeWorkflowIDReuse\n\treq.PreviousRunID = runID\n\treq.PreviousLastWriteVersion = constants.EmptyVersion\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Error(err)\n\ts.IsType(&p.WorkflowExecutionAlreadyStartedError{}, err)\n}\n\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionWithWorkflowRequestsDedup() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tdomainName := uuid.New()\n\tworkflowID := \"create-workflow-test-dedup\"\n\trunID := uuid.New()\n\trequestID := uuid.New()\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             requestID,\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowID,\n\t\t\t\tRunID:                       runID,\n\t\t\t\tFirstExecutionRunID:         runID,\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateCreated,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t\tWorkflowRequests: []*p.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   requestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: p.WorkflowRequestTypeStart,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tRangeID:             s.ShardInfo.RangeID,\n\t\tMode:                p.CreateWorkflowModeBrandNew,\n\t\tWorkflowRequestMode: p.CreateWorkflowRequestModeReplicated,\n\t\tDomainName:          domainName,\n\t}\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\treq.WorkflowRequestMode = p.CreateWorkflowRequestModeNew\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n\ts.Equal(persistence.WorkflowRequestTypeStart, err.(*persistence.DuplicateRequestError).RequestType)\n\ts.Equal(runID, err.(*persistence.DuplicateRequestError).RunID)\n\treq.WorkflowRequestMode = p.CreateWorkflowRequestModeReplicated\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Error(err)\n\ts.IsType(&p.WorkflowExecutionAlreadyStartedError{}, err)\n}\n\n// TestCreateWorkflowExecutionStateCloseStatus test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionStateCloseStatus() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tinvalidCloseStatuses := []int{\n\t\tp.WorkflowCloseStatusCompleted,\n\t\tp.WorkflowCloseStatusFailed,\n\t\tp.WorkflowCloseStatusCanceled,\n\t\tp.WorkflowCloseStatusTerminated,\n\t\tp.WorkflowCloseStatusContinuedAsNew,\n\t\tp.WorkflowCloseStatusTimedOut,\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeBrandNew,\n\t}\n\n\tworkflowExecutionStatusCreated := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-state-created\",\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionStatusCreated.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionStatusCreated.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionStatusCreated.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateCreated\n\tfor _, invalidCloseStatus := range invalidCloseStatuses {\n\t\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = invalidCloseStatus\n\t\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionStatusCreated)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateCreated, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\n\tworkflowExecutionStatusRunning := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-state-running\",\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionStatusRunning.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionStatusRunning.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionStatusRunning.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateRunning\n\tfor _, invalidCloseStatus := range invalidCloseStatuses {\n\t\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = invalidCloseStatus\n\t\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionStatusRunning)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateRunning, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\n\tworkflowExecutionStatusCompleted := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-state-completed\",\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionStatusCompleted.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionStatusCompleted.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionStatusCompleted.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateCompleted\n\tfor _, invalidCloseStatus := range invalidCloseStatuses {\n\t\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = invalidCloseStatus\n\t\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.IsType(&types.InternalServiceError{}, err)\n\n\t// for zombie workflow creation, we must use existing workflow ID which got created\n\t// since we do not allow creation of zombie workflow without current record\n\tworkflowExecutionStatusZombie := types.WorkflowExecution{\n\t\tWorkflowID: workflowExecutionStatusRunning.WorkflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\treq.Mode = p.CreateWorkflowModeZombie\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionStatusZombie.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionStatusZombie.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionStatusZombie.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateZombie\n\tfor _, invalidCloseStatus := range invalidCloseStatuses {\n\t\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = invalidCloseStatus\n\t\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionStatusZombie)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateZombie, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n}\n\n// TestCreateWorkflowExecutionWithZombieState test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionWithZombieState() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowID := \"create-workflow-test-with-zombie-state\"\n\tworkflowExecutionZombie1 := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowID,\n\t\t\t\tRunID:                       workflowExecutionZombie1.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecutionZombie1.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateZombie,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeZombie,\n\t}\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err) // allow creating a zombie workflow if no current running workflow\n\t_, err = s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.IsType(&types.EntityNotExistsError{}, err) // no current workflow\n\n\tworkflowExecutionRunning := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionRunning.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionRunning.GetRunID()\n\treq.Mode = p.CreateWorkflowModeBrandNew\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateRunning\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.Nil(err)\n\ts.Equal(workflowExecutionRunning.GetRunID(), currentRunID)\n\n\tworkflowExecutionZombie := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionZombie.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionZombie.GetRunID()\n\treq.Mode = p.CreateWorkflowModeZombie\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateZombie\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\t// current run ID is still the prev running run ID\n\tcurrentRunID, err = s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecutionRunning.GetWorkflowID())\n\ts.Nil(err)\n\ts.Equal(workflowExecutionRunning.GetRunID(), currentRunID)\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionZombie)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateZombie, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n}\n\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionWithWorkflowRequestsDedup() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tdomainName := uuid.New()\n\tworkflowID := \"create-workflow-test-dedup\"\n\trunID := uuid.New()\n\trequestID := uuid.New()\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             requestID,\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowID,\n\t\t\t\tRunID:                       runID,\n\t\t\t\tFirstExecutionRunID:         runID,\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateCreated,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t\tWorkflowRequests: []*p.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   requestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: p.WorkflowRequestTypeStart,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRequestID:   requestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: p.WorkflowRequestTypeSignal,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tRangeID:             s.ShardInfo.RangeID,\n\t\tMode:                p.CreateWorkflowModeBrandNew,\n\t\tWorkflowRequestMode: p.CreateWorkflowRequestModeNew,\n\t\tDomainName:          domainName,\n\t}\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, types.WorkflowExecution{WorkflowID: workflowID, RunID: runID})\n\ts.Nil(err)\n\tcsum = s.newRandomChecksum() // update the checksum to new value\n\tupdatedInfo := copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats := copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateRunning\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\tupdateReq := &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t\tWorkflowRequests: []*p.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestID:   requestID,\n\t\t\t\t\tVersion:     1,\n\t\t\t\t\tRequestType: p.WorkflowRequestTypeSignal,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tRangeID:             s.ShardInfo.RangeID,\n\t\tMode:                p.UpdateWorkflowModeUpdateCurrent,\n\t\tWorkflowRequestMode: p.CreateWorkflowRequestModeNew,\n\t}\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, updateReq)\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n\ts.Equal(persistence.WorkflowRequestTypeSignal, err.(*persistence.DuplicateRequestError).RequestType)\n\ts.Equal(runID, err.(*persistence.DuplicateRequestError).RunID)\n\tupdateReq.WorkflowRequestMode = p.CreateWorkflowRequestModeReplicated\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, updateReq)\n\ts.Nil(err)\n\tupdateReq.UpdateWorkflowMutation.WorkflowRequests[0].RequestID = uuid.New()\n\tupdateReq.WorkflowRequestMode = p.CreateWorkflowRequestModeNew\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, updateReq)\n\ts.Nil(err)\n}\n\n// TestUpdateWorkflowExecutionStateCloseStatus test\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionStateCloseStatus() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"update-workflow-test-state\",\n\t\tRunID:      uuid.New(),\n\t}\n\tcloseStatuses := []int{\n\t\tp.WorkflowCloseStatusCompleted,\n\t\tp.WorkflowCloseStatusFailed,\n\t\tp.WorkflowCloseStatusCanceled,\n\t\tp.WorkflowCloseStatusTerminated,\n\t\tp.WorkflowCloseStatusContinuedAsNew,\n\t\tp.WorkflowCloseStatusTimedOut,\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeBrandNew,\n\t}\n\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateCreated\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateCreated, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\n\tcsum = s.newRandomChecksum() // update the checksum to new value\n\tupdatedInfo := copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats := copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateRunning\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.NoError(err)\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateRunning, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateRunning\n\tfor _, closeStatus := range closeStatuses {\n\t\tupdatedInfo.CloseStatus = closeStatus\n\t\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\t\tExecutionInfo:  updatedInfo,\n\t\t\t\tExecutionStats: updatedStats,\n\t\t\t\tCondition:      nextEventID,\n\t\t\t},\n\t\t\tRangeID: s.ShardInfo.RangeID,\n\t\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t\t})\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateCompleted\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tCondition:      nextEventID,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.IsType(&types.InternalServiceError{}, err)\n\n\tfor _, closeStatus := range closeStatuses {\n\t\tupdatedInfo.CloseStatus = closeStatus\n\t\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\t\tExecutionInfo:    updatedInfo,\n\t\t\t\tExecutionStats:   updatedStats,\n\t\t\t\tCondition:        nextEventID,\n\t\t\t\tVersionHistories: versionHistories,\n\t\t\t},\n\t\t\tRangeID: s.ShardInfo.RangeID,\n\t\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t\t})\n\t\ts.Nil(err)\n\t\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\t\ts.Nil(err)\n\t\ts.Equal(p.WorkflowStateCompleted, info.ExecutionInfo.State)\n\t\ts.Equal(closeStatus, info.ExecutionInfo.CloseStatus)\n\t}\n\n\t// create a new workflow with same domain ID & workflow ID\n\t// to enable update workflow with zombie status\n\tworkflowExecutionRunning := types.WorkflowExecution{\n\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionRunning.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionRunning.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionRunning.GetRunID()\n\treq.Mode = p.CreateWorkflowModeWorkflowIDReuse\n\treq.PreviousRunID = workflowExecution.GetRunID()\n\treq.PreviousLastWriteVersion = constants.EmptyVersion\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateRunning\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateZombie\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeBypassCurrent,\n\t})\n\ts.NoError(err)\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateZombie, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdatedStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateZombie\n\tfor _, closeStatus := range closeStatuses {\n\t\tupdatedInfo.CloseStatus = closeStatus\n\t\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\t\tExecutionInfo:  updatedInfo,\n\t\t\t\tExecutionStats: updatedStats,\n\t\t\t\tCondition:      nextEventID,\n\t\t\t},\n\t\t\tRangeID: s.ShardInfo.RangeID,\n\t\t\tMode:    p.UpdateWorkflowModeBypassCurrent,\n\t\t})\n\t\ts.IsType(&types.InternalServiceError{}, err)\n\t}\n}\n\n// TestUpdateWorkflowExecutionWithZombieState test\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionWithZombieState() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowID := \"create-workflow-test-with-zombie-state\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\t// create and update a workflow to make it completed\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeBrandNew,\n\t}\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.Nil(err)\n\ts.Equal(workflowExecution.GetRunID(), currentRunID)\n\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\n\t// try to turn current workflow into zombie state, this should end with an error\n\tupdatedInfo := copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdateStats := copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateZombie\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updateStats,\n\t\t\tCondition:      nextEventID,\n\t\t\tChecksum:       csum,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeBypassCurrent,\n\t})\n\ts.NotNil(err)\n\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdateStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateCompleted\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusCompleted\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updateStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.NoError(err)\n\n\t// create a new workflow with same domain ID & workflow ID\n\tworkflowExecutionRunning := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\tcsum = checksum.Checksum{} // set checksum to nil\n\treq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID = workflowExecutionRunning.GetWorkflowID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.RunID = workflowExecutionRunning.GetRunID()\n\treq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = workflowExecutionRunning.GetRunID()\n\treq.Mode = p.CreateWorkflowModeWorkflowIDReuse\n\treq.PreviousRunID = workflowExecution.GetRunID()\n\treq.PreviousLastWriteVersion = constants.EmptyVersion\n\treq.NewWorkflowSnapshot.ExecutionInfo.State = p.WorkflowStateRunning\n\treq.NewWorkflowSnapshot.ExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\treq.NewWorkflowSnapshot.Checksum = csum\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tcurrentRunID, err = s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.Nil(err)\n\ts.Equal(workflowExecutionRunning.GetRunID(), currentRunID)\n\n\t// get the workflow to be turned into a zombie\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\tupdatedInfo = copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdateStats = copyExecutionStats(info.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateZombie\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusNone\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updateStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeBypassCurrent,\n\t})\n\ts.NoError(err)\n\tinfo, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(p.WorkflowStateZombie, info.ExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.ExecutionInfo.CloseStatus)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\t// check current run ID is un touched\n\tcurrentRunID, err = s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.Nil(err)\n\ts.Equal(workflowExecutionRunning.GetRunID(), currentRunID)\n}\n\n// TestUpdateWorkflowExecutionTasks test\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"b0a8571c-0257-40ea-afcd-3a14eae181c0\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"update-workflow-tasks-test\",\n\t\tRunID:      \"5ba5e531-e46b-48d9-b4b3-859919839553\",\n\t}\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\tnow := time.Now()\n\ttransferTasks := []p.Task{\n\t\t&p.ActivityTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              s.GetNextSequenceNumber(),\n\t\t\t\tVersion:             constants.EmptyVersion,\n\t\t\t},\n\t\t\tTargetDomainID: domainID,\n\t\t\tTaskList:       \"some randome tasklist name\",\n\t\t\tScheduleID:     123,\n\t\t},\n\t}\n\ttimerTasks := []p.Task{\n\t\t&p.UserTimerTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now.Add(time.Minute),\n\t\t\t\tTaskID:              s.GetNextSequenceNumber(),\n\t\t\t\tVersion:             constants.EmptyVersion,\n\t\t\t},\n\t\t\tTaskList: \"some random tasklist\",\n\t\t\tEventID:  124,\n\t\t},\n\t}\n\terr = s.UpdateWorkflowExecutionTasks(\n\t\tctx,\n\t\tupdatedInfo1,\n\t\tupdatedStats1,\n\t\tint64(3),\n\t\ttransferTasks,\n\t\ttimerTasks,\n\t)\n\ts.NoError(err)\n\n\tloadedTransferTasks, err := s.GetTransferTasks(ctx, 10, true)\n\ts.NoError(err)\n\ts.Len(loadedTransferTasks, len(transferTasks))\n\n\tloadedTimerTasks, err := s.GetTimerIndexTasks(ctx, 10, true)\n\ts.NoError(err)\n\ts.Len(loadedTimerTasks, len(timerTasks))\n}\n\n// TestCreateWorkflowExecutionBrandNew test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionBrandNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-brand-new\",\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeBrandNew,\n\t}\n\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\t_, err = s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.NotNil(err)\n\talreadyStartedErr, ok := err.(*p.WorkflowExecutionAlreadyStartedError)\n\ts.True(ok, \"err is not WorkflowExecutionAlreadyStartedError\")\n\ts.Equal(req.NewWorkflowSnapshot.ExecutionInfo.CreateRequestID, alreadyStartedErr.StartRequestID)\n\ts.Equal(workflowExecution.GetRunID(), alreadyStartedErr.RunID)\n\ts.Equal(0, alreadyStartedErr.CloseStatus)\n\ts.Equal(p.WorkflowStateRunning, alreadyStartedErr.State)\n}\n\n// TestUpsertWorkflowActivity test\nfunc (s *ExecutionManagerSuite) TestUpsertWorkflowActivity() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowID := \"create-workflow-test-with-upsert-activity\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := s.newRandomChecksum()\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\t// create and update a workflow to make it completed\n\treq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.CreateWorkflowModeBrandNew,\n\t}\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.Nil(err)\n\ts.Equal(workflowExecution.GetRunID(), currentRunID)\n\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.assertChecksumsEqual(csum, info.Checksum)\n\ts.Equal(0, len(info.ActivityInfos))\n\n\t// insert a new activity\n\tupdatedInfo := copyWorkflowExecutionInfo(info.ExecutionInfo)\n\tupdateStats := copyExecutionStats(info.ExecutionStats)\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updateStats,\n\t\t\tCondition:      nextEventID,\n\t\t\tChecksum:       csum,\n\t\t\tUpsertActivityInfos: []*p.ActivityInfo{\n\t\t\t\t{\n\t\t\t\t\tVersion:    0,\n\t\t\t\t\tScheduleID: 100,\n\t\t\t\t\tTaskList:   \"test-activity-tasklist-1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.Nil(err)\n\n\tinfo2, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(1, len(info2.ActivityInfos))\n\ts.Equal(\"test-activity-tasklist-1\", info2.ActivityInfos[100].TaskList)\n\n\t// upsert the previous activity\n\t_, err = s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updateStats,\n\t\t\tCondition:      nextEventID,\n\t\t\tChecksum:       csum,\n\t\t\tUpsertActivityInfos: []*p.ActivityInfo{\n\t\t\t\t{\n\t\t\t\t\tVersion:    0,\n\t\t\t\t\tScheduleID: 100,\n\t\t\t\t\tTaskList:   \"test-activity-tasklist-2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeUpdateCurrent,\n\t})\n\ts.NoError(err)\n\tinfo3, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Nil(err)\n\ts.Equal(1, len(info3.ActivityInfos))\n\ts.Equal(\"test-activity-tasklist-2\", info3.ActivityInfos[100].TaskList)\n}\n\n// TestCreateWorkflowExecutionRunIDReuseWithoutReplication test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionRunIDReuseWithoutReplication() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-run-id-reuse-without-replication\",\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tdecisionScheduleID := int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, tasklist,\n\t\tworkflowType, workflowTimeout, decisionTimeout, nil, nextEventID,\n\t\tlastProcessedEventID, decisionScheduleID, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state0.Checksum)\n\tinfo0 := state0.ExecutionInfo\n\tcloseInfo := copyWorkflowExecutionInfo(info0)\n\tcloseInfo.State = p.WorkflowStateCompleted\n\tcloseInfo.CloseStatus = p.WorkflowCloseStatusCompleted\n\tcloseInfo.NextEventID = int64(5)\n\tcloseInfo.LastProcessedEvent = int64(2)\n\n\terr2 := s.UpdateWorkflowExecution(ctx, closeInfo, state0.ExecutionStats, versionHistories, nil, nil, nextEventID,\n\t\tnil, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\tnewExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\tRunID:      uuid.New(),\n\t}\n\t// this create should work since we are relying the business logic in history engine\n\t// to check whether the existing running workflow has finished\n\t_, err3 := s.ExecutionManager.CreateWorkflowExecution(ctx, &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  newExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       newExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         newExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:                  s.ShardInfo.RangeID,\n\t\tMode:                     p.CreateWorkflowModeWorkflowIDReuse,\n\t\tPreviousRunID:            workflowExecution.GetRunID(),\n\t\tPreviousLastWriteVersion: constants.EmptyVersion,\n\t})\n\ts.NoError(err3)\n}\n\n// TestCreateWorkflowExecutionConcurrentCreate test\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionConcurrentCreate() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"create-workflow-test-concurrent-create\",\n\t\tRunID:      uuid.New(),\n\t}\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tdecisionScheduleID := int64(2)\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, tasklist,\n\t\tworkflowType, workflowTimeout, decisionTimeout, nil, nextEventID,\n\t\tlastProcessedEventID, decisionScheduleID, nil, nil)\n\ts.Nil(err0, \"No error expected.\")\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttimes := 2\n\tvar wg sync.WaitGroup\n\twg.Add(times)\n\tvar numOfErr int32\n\tvar lastError error\n\tfor i := 0; i < times; i++ {\n\t\tgo func() {\n\t\t\tnewExecution := types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t}\n\n\t\t\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\t\t\ts.NoError(err1)\n\t\t\tinfo0 := state0.ExecutionInfo\n\t\t\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info0)\n\t\t\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\t\t\tcontinueAsNewInfo.NextEventID = int64(5)\n\t\t\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\t\t\terr2 := s.ContinueAsNewExecution(ctx, continueAsNewInfo, state0.ExecutionStats, info0.NextEventID, newExecution, int64(3), int64(2), nil)\n\t\t\tif err2 != nil {\n\t\t\t\terrCount := atomic.AddInt32(&numOfErr, 1)\n\t\t\t\tif errCount > 1 {\n\t\t\t\t\tlastError = err2\n\t\t\t\t}\n\t\t\t}\n\t\t\twg.Done()\n\t\t}()\n\t}\n\twg.Wait()\n\tif lastError != nil {\n\t\ts.Fail(\"More than one error: %v\", lastError.Error())\n\t}\n\ts.Equal(int32(1), atomic.LoadInt32(&numOfErr))\n}\n\n// TestPersistenceStartWorkflow test\nfunc (s *ExecutionManagerSuite) TestPersistenceStartWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"2d7994bf-9de8-459d-9c81-e723daedb246\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"start-workflow-test\",\n\t\tRunID:      \"7f9fe8a0-9237-11e6-ae22-56b6b6499611\",\n\t}\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttask1, err1 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType1\", 20, 14, nil, 3, 0, 2, nil, nil)\n\ts.Error(err1, \"Expected workflow creation to fail.\")\n\ts.T().Logf(\"Unable to start workflow execution: %v\\n\", err1)\n\tstartedErr, ok := err1.(*p.WorkflowExecutionAlreadyStartedError)\n\ts.True(ok, fmt.Sprintf(\"Expected WorkflowExecutionAlreadyStartedError, but actual is %v\", err1))\n\ts.Equal(workflowExecution.GetRunID(), startedErr.RunID, startedErr.Msg)\n\n\ts.Equal(p.WorkflowStateRunning, startedErr.State, startedErr.Msg)\n\ts.Equal(p.WorkflowCloseStatusNone, startedErr.CloseStatus, startedErr.Msg)\n\ts.Equal(constants.EmptyVersion, startedErr.LastWriteVersion, startedErr.Msg)\n\ts.Empty(task1, \"Expected empty task identifier.\")\n\tdecisionScheduleID := int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tresponse, err2 := s.ExecutionManager.CreateWorkflowExecution(ctx, &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    \"queue1\",\n\t\t\t\tWorkflowTypeName:            \"workflow_type_test\",\n\t\t\t\tWorkflowTimeout:             20,\n\t\t\t\tDecisionStartToCloseTimeout: 13,\n\t\t\t\tExecutionContext:            nil,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 int64(3),\n\t\t\t\tLastProcessedEvent:          0,\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t},\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tTasksByCategory: map[p.HistoryTaskCategory][]p.Task{\n\t\t\t\tp.HistoryTaskCategoryTransfer: []p.Task{\n\t\t\t\t\t&p.DecisionTask{\n\t\t\t\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: p.TaskData{\n\t\t\t\t\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTargetDomainID:       domainID,\n\t\t\t\t\t\tTaskList:             \"queue1\",\n\t\t\t\t\t\tScheduleID:           int64(2),\n\t\t\t\t\t\tOriginalTaskList:     \"queue1\",\n\t\t\t\t\t\tOriginalTaskListKind: types.TaskListKindEphemeral,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID - 1,\n\t})\n\n\ts.Error(err2, \"Expected workflow creation to fail.\")\n\ts.Nil(response)\n\ts.T().Logf(\"Unable to start workflow execution: %v\\n\", err2)\n\ts.IsType(&p.ShardOwnershipLostError{}, err2)\n}\n\n// TestGetWorkflow test\nfunc (s *ExecutionManagerSuite) TestGetWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tnow := time.Now()\n\ttestResetPoints := types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\t{\n\t\t\t\tBinaryChecksum:           \"test-binary-checksum\",\n\t\t\t\tRunID:                    \"test-runID\",\n\t\t\t\tFirstDecisionCompletedID: 123,\n\t\t\t\tCreatedTimeNano:          common.Int64Ptr(456),\n\t\t\t\tResettable:               true,\n\t\t\t\tExpiringTimeNano:         common.Int64Ptr(789),\n\t\t\t},\n\t\t},\n\t}\n\ttestSearchAttrKey := \"env\"\n\ttestSearchAttrVal, _ := json.Marshal(\"test\")\n\ttestSearchAttr := map[string][]byte{\n\t\ttestSearchAttrKey: testSearchAttrVal,\n\t}\n\n\ttestMemoKey := \"memoKey\"\n\ttestMemoVal, _ := json.Marshal(\"memoVal\")\n\ttestMemo := map[string][]byte{\n\t\ttestMemoKey: testMemoVal,\n\t}\n\n\ttestPartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tcsum := s.newRandomChecksum()\n\tdecisionScheduleID := int64(rand.Int31())\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tcreateReq := &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    uuid.New(),\n\t\t\t\tWorkflowID:                  \"get-workflow-test\",\n\t\t\t\tRunID:                       uuid.New(),\n\t\t\t\tParentDomainID:              uuid.New(),\n\t\t\t\tParentWorkflowID:            \"get-workflow-test-parent\",\n\t\t\t\tParentRunID:                 uuid.New(),\n\t\t\t\tInitiatedID:                 rand.Int63(),\n\t\t\t\tTaskList:                    \"get-wf-test-tasklist\",\n\t\t\t\tWorkflowTypeName:            \"code.uber.internal/test/workflow\",\n\t\t\t\tWorkflowTimeout:             rand.Int31(),\n\t\t\t\tDecisionStartToCloseTimeout: rand.Int31(),\n\t\t\t\tExecutionContext:            []byte(\"test-execution-context\"),\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 rand.Int63(),\n\t\t\t\tLastProcessedEvent:          int64(rand.Int31()),\n\t\t\t\tLastUpdatedTimestamp:        now,\n\t\t\t\tStartTimestamp:              now,\n\t\t\t\tSignalCount:                 rand.Int31(),\n\t\t\t\tDecisionVersion:             int64(rand.Int31()),\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           int64(rand.Int31()),\n\t\t\t\tDecisionTimeout:             rand.Int31(),\n\t\t\t\tAttempt:                     rand.Int31(),\n\t\t\t\tHasRetryPolicy:              true,\n\t\t\t\tInitialInterval:             rand.Int31(),\n\t\t\t\tBackoffCoefficient:          7.78,\n\t\t\t\tMaximumInterval:             rand.Int31(),\n\t\t\t\tExpirationTime:              time.Now(),\n\t\t\t\tMaximumAttempts:             rand.Int31(),\n\t\t\t\tNonRetriableErrors:          []string{\"badRequestError\", \"accessDeniedError\"},\n\t\t\t\tCronSchedule:                \"* * * * *\",\n\t\t\t\tCronOverlapPolicy:           types.CronOverlapPolicySkipped,\n\t\t\t\tExpirationSeconds:           rand.Int31(),\n\t\t\t\tAutoResetPoints:             &testResetPoints,\n\t\t\t\tSearchAttributes:            testSearchAttr,\n\t\t\t\tMemo:                        testMemo,\n\t\t\t\tPartitionConfig:             testPartitionConfig,\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\t\t\t\tStickyRegion:                   \"region1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tExecutionStats: &p.ExecutionStats{\n\t\t\t\tHistorySize: int64(rand.Int31()),\n\t\t\t},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tMode: p.CreateWorkflowModeBrandNew,\n\t}\n\tcreateReq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID = createReq.NewWorkflowSnapshot.ExecutionInfo.RunID\n\n\tcreateResp, err := s.ExecutionManager.CreateWorkflowExecution(ctx, createReq)\n\ts.NoError(err)\n\ts.NotNil(createResp, \"Expected non empty task identifier.\")\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, createReq.NewWorkflowSnapshot.ExecutionInfo.DomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: createReq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID,\n\t\t\tRunID:      createReq.NewWorkflowSnapshot.ExecutionInfo.RunID,\n\t\t})\n\ts.NoError(err)\n\tinfo := state.ExecutionInfo\n\ts.NotNil(info, \"Valid Workflow response expected.\")\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.CreateRequestID, info.CreateRequestID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DomainID, info.DomainID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.WorkflowID, info.WorkflowID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.RunID, info.RunID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.FirstExecutionRunID, info.FirstExecutionRunID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ParentDomainID, info.ParentDomainID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ParentWorkflowID, info.ParentWorkflowID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ParentRunID, info.ParentRunID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.InitiatedID, info.InitiatedID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.TaskList, info.TaskList)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.WorkflowTypeName, info.WorkflowTypeName)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.WorkflowTimeout, info.WorkflowTimeout)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DecisionStartToCloseTimeout, info.DecisionStartToCloseTimeout)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ExecutionContext, info.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info.CloseStatus)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.NextEventID, info.NextEventID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.LastProcessedEvent, info.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(true, s.validateTimeRange(info.StartTimestamp, time.Hour))\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DecisionVersion, info.DecisionVersion)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DecisionScheduleID, info.DecisionScheduleID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DecisionStartedID, info.DecisionStartedID)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.DecisionTimeout, info.DecisionTimeout)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.SignalCount, info.SignalCount)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.Attempt, info.Attempt)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.HasRetryPolicy, info.HasRetryPolicy)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.InitialInterval, info.InitialInterval)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.BackoffCoefficient, info.BackoffCoefficient)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.MaximumAttempts, info.MaximumAttempts)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.MaximumInterval, info.MaximumInterval)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ExpirationSeconds, info.ExpirationSeconds)\n\ts.EqualTimes(createReq.NewWorkflowSnapshot.ExecutionInfo.ExpirationTime, info.ExpirationTime)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.CronSchedule, info.CronSchedule)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.CronOverlapPolicy, info.CronOverlapPolicy)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.ActiveClusterSelectionPolicy, info.ActiveClusterSelectionPolicy)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionInfo.NonRetriableErrors, info.NonRetriableErrors)\n\ts.Equal(testResetPoints, *info.AutoResetPoints)\n\ts.Equal(createReq.NewWorkflowSnapshot.ExecutionStats.HistorySize, state.ExecutionStats.HistorySize)\n\tval, ok := info.SearchAttributes[testSearchAttrKey]\n\ts.True(ok)\n\ts.Equal(testSearchAttrVal, val)\n\tval, ok = info.Memo[testMemoKey]\n\ts.True(ok)\n\ts.Equal(testMemoVal, val)\n\ts.Equal(testPartitionConfig, info.PartitionConfig)\n\n\ts.assertChecksumsEqual(csum, state.Checksum)\n}\n\n// TestUpdateWorkflow test\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"b0a8571c-0257-40ea-afcd-3a14eae181c0\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"update-workflow-test\",\n\t\tRunID:      \"5ba5e531-e46b-48d9-b4b3-859919839553\",\n\t}\n\tpartitionConfig0 := map[string]string{\n\t\t\"userID\": uuid.New(),\n\t}\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, partitionConfig0)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info0.DomainID)\n\ts.Equal(\"update-workflow-test\", info0.WorkflowID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info0.RunID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info0.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info0.TaskList)\n\ts.Equal(\"wType\", info0.WorkflowTypeName)\n\ts.Equal(int32(20), info0.WorkflowTimeout)\n\ts.Equal(int32(13), info0.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info0.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info0.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info0.CloseStatus)\n\ts.Equal(int64(1), info0.LastFirstEventID)\n\ts.Equal(int64(3), info0.NextEventID)\n\ts.Equal(int64(0), info0.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info0.LastUpdatedTimestamp, time.Minute))\n\ts.Equal(true, s.validateTimeRange(info0.StartTimestamp, time.Minute))\n\ts.Equal(int64(0), info0.DecisionVersion)\n\ts.Equal(int64(2), info0.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info0.DecisionStartedID)\n\ts.Equal(int32(1), info0.DecisionTimeout)\n\ts.Equal(int64(0), info0.DecisionAttempt)\n\ts.Equal(int64(0), info0.DecisionStartedTimestamp)\n\ts.Equal(int64(0), info0.DecisionScheduledTimestamp)\n\ts.Equal(int64(0), info0.DecisionOriginalScheduledTimestamp)\n\ts.Empty(info0.StickyTaskList)\n\ts.Equal(int32(0), info0.StickyScheduleToStartTimeout)\n\ts.Empty(info0.ClientLibraryVersion)\n\ts.Empty(info0.ClientFeatureVersion)\n\ts.Empty(info0.ClientImpl)\n\ts.Equal(int32(0), info0.SignalCount)\n\ts.Equal(info0.AutoResetPoints, &types.ResetPoints{})\n\ts.True(len(info0.SearchAttributes) == 0)\n\ts.True(len(info0.Memo) == 0)\n\ts.Equal(partitionConfig0, info0.PartitionConfig)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state0.Checksum)\n\n\ts.T().Logf(\"Workflow execution last updated: %v\\n\", info0.LastUpdatedTimestamp)\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := state0.ExecutionStats\n\tupdatedInfo.LastFirstEventID = int64(3)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tupdatedInfo.DecisionVersion = int64(666)\n\tupdatedInfo.DecisionAttempt = int64(123)\n\tupdatedInfo.DecisionStartedTimestamp = int64(321)\n\tupdatedInfo.DecisionScheduledTimestamp = int64(654)\n\tupdatedInfo.DecisionOriginalScheduledTimestamp = int64(655)\n\tupdatedInfo.StickyTaskList = \"random sticky tasklist\"\n\tupdatedInfo.StickyScheduleToStartTimeout = 876\n\tupdatedInfo.ClientLibraryVersion = \"random client library version\"\n\tupdatedInfo.ClientFeatureVersion = \"random client feature version\"\n\tupdatedInfo.ClientImpl = \"random client impl\"\n\tupdatedInfo.SignalCount = 9\n\tupdatedInfo.InitialInterval = math.MaxInt32\n\tupdatedInfo.BackoffCoefficient = 4.45\n\tupdatedInfo.MaximumInterval = math.MaxInt32\n\tupdatedInfo.MaximumAttempts = math.MaxInt32\n\tupdatedInfo.ExpirationSeconds = math.MaxInt32\n\tupdatedInfo.ExpirationTime = time.Now()\n\tupdatedInfo.NonRetriableErrors = []string{\"accessDenied\", \"badRequest\"}\n\tsearchAttrKey := \"env\"\n\tsearchAttrVal := []byte(\"test\")\n\tupdatedInfo.SearchAttributes = map[string][]byte{searchAttrKey: searchAttrVal}\n\tmemoKey := \"memoKey\"\n\tmemoVal := []byte(\"memoVal\")\n\tupdatedInfo.Memo = map[string][]byte{memoKey: memoVal}\n\tpartitionConfig := map[string]string{\"zone\": \"dca2\"}\n\tupdatedInfo.PartitionConfig = partitionConfig\n\tupdatedStats.HistorySize = math.MaxInt64\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.NextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), nil, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\tstate1, err3 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err3)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info1.DomainID)\n\ts.Equal(\"update-workflow-test\", info1.WorkflowID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info1.RunID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info1.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info1.TaskList)\n\ts.Equal(\"wType\", info1.WorkflowTypeName)\n\ts.Equal(int32(20), info1.WorkflowTimeout)\n\ts.Equal(int32(13), info1.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info1.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info1.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info1.CloseStatus)\n\ts.Equal(int64(3), info1.LastFirstEventID)\n\ts.Equal(int64(5), info1.NextEventID)\n\ts.Equal(int64(2), info1.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info1.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(info0.StartTimestamp.UnixNano(), info1.StartTimestamp.UnixNano())\n\ts.Equal(int64(666), info1.DecisionVersion)\n\ts.Equal(int64(2), info1.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info1.DecisionStartedID)\n\ts.Equal(int32(1), info1.DecisionTimeout)\n\ts.Equal(int64(123), info1.DecisionAttempt)\n\ts.Equal(int64(321), info1.DecisionStartedTimestamp)\n\ts.Equal(int64(654), info1.DecisionScheduledTimestamp)\n\ts.Equal(int64(655), info1.DecisionOriginalScheduledTimestamp)\n\ts.Equal(updatedInfo.StickyTaskList, info1.StickyTaskList)\n\ts.Equal(updatedInfo.StickyScheduleToStartTimeout, info1.StickyScheduleToStartTimeout)\n\ts.Equal(updatedInfo.ClientLibraryVersion, info1.ClientLibraryVersion)\n\ts.Equal(updatedInfo.ClientFeatureVersion, info1.ClientFeatureVersion)\n\ts.Equal(updatedInfo.ClientImpl, info1.ClientImpl)\n\ts.Equal(updatedInfo.SignalCount, info1.SignalCount)\n\ts.EqualValues(updatedStats.HistorySize, state1.ExecutionStats.HistorySize)\n\ts.Equal(updatedInfo.InitialInterval, info1.InitialInterval)\n\ts.Equal(updatedInfo.BackoffCoefficient, info1.BackoffCoefficient)\n\ts.Equal(updatedInfo.MaximumInterval, info1.MaximumInterval)\n\ts.Equal(updatedInfo.MaximumAttempts, info1.MaximumAttempts)\n\ts.Equal(updatedInfo.ExpirationSeconds, info1.ExpirationSeconds)\n\ts.EqualTimes(updatedInfo.ExpirationTime, info1.ExpirationTime)\n\ts.Equal(updatedInfo.NonRetriableErrors, info1.NonRetriableErrors)\n\tsearchAttrVal1, ok := info1.SearchAttributes[searchAttrKey]\n\ts.True(ok)\n\ts.Equal(searchAttrVal, searchAttrVal1)\n\tmemoVal1, ok := info1.Memo[memoKey]\n\ts.True(ok)\n\ts.Equal(memoVal, memoVal1)\n\ts.Equal(partitionConfig, info1.PartitionConfig)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state1.Checksum)\n\n\ts.T().Logf(\"Workflow execution last updated: %v\\n\", info1.LastUpdatedTimestamp)\n\n\tfailedUpdateInfo := copyWorkflowExecutionInfo(updatedInfo)\n\tfailedUpdateStats := copyExecutionStats(updatedStats)\n\terr4 := s.UpdateWorkflowExecution(ctx, failedUpdateInfo, failedUpdateStats, versionHistories, []int64{int64(5)}, nil, int64(3), nil, nil, nil, nil, nil)\n\ts.Error(err4, \"expected non nil error.\")\n\ts.IsType(&p.ConditionFailedError{}, err4)\n\n\tstate2, err4 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err4)\n\tinfo2 := state2.ExecutionInfo\n\ts.NotNil(info2, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info2.DomainID)\n\ts.Equal(\"update-workflow-test\", info2.WorkflowID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info2.RunID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info2.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info2.TaskList)\n\ts.Equal(\"wType\", info2.WorkflowTypeName)\n\ts.Equal(int32(20), info2.WorkflowTimeout)\n\ts.Equal(int32(13), info2.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info2.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info2.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info2.CloseStatus)\n\ts.Equal(int64(5), info2.NextEventID)\n\ts.Equal(int64(2), info2.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info2.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(int64(666), info2.DecisionVersion)\n\ts.Equal(int64(2), info2.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info2.DecisionStartedID)\n\ts.Equal(int32(1), info2.DecisionTimeout)\n\ts.Equal(int64(123), info2.DecisionAttempt)\n\ts.Equal(int64(321), info2.DecisionStartedTimestamp)\n\ts.Equal(int64(654), info2.DecisionScheduledTimestamp)\n\ts.Equal(int64(655), info2.DecisionOriginalScheduledTimestamp)\n\ts.Equal(updatedInfo.SignalCount, info2.SignalCount)\n\ts.EqualValues(updatedStats.HistorySize, state2.ExecutionStats.HistorySize)\n\ts.Equal(updatedInfo.InitialInterval, info2.InitialInterval)\n\ts.Equal(updatedInfo.BackoffCoefficient, info2.BackoffCoefficient)\n\ts.Equal(updatedInfo.MaximumInterval, info2.MaximumInterval)\n\ts.Equal(updatedInfo.MaximumAttempts, info2.MaximumAttempts)\n\ts.Equal(updatedInfo.ExpirationSeconds, info2.ExpirationSeconds)\n\ts.EqualTimes(updatedInfo.ExpirationTime, info2.ExpirationTime)\n\ts.Equal(updatedInfo.NonRetriableErrors, info2.NonRetriableErrors)\n\tsearchAttrVal2, ok := info2.SearchAttributes[searchAttrKey]\n\ts.True(ok)\n\ts.Equal(searchAttrVal, searchAttrVal2)\n\tmemoVal2, ok := info1.Memo[memoKey]\n\ts.True(ok)\n\ts.Equal(memoVal, memoVal2)\n\ts.Equal(partitionConfig, info2.PartitionConfig)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state2.Checksum)\n\ts.T().Logf(\"Workflow execution last updated: %v\", info2.LastUpdatedTimestamp)\n\n\terr5 := s.UpdateWorkflowExecutionWithRangeID(ctx, failedUpdateInfo, failedUpdateStats, versionHistories, []int64{int64(5)}, nil, int64(12345), int64(5), nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil)\n\ts.Error(err5, \"expected non nil error.\")\n\ts.IsType(&p.ShardOwnershipLostError{}, err5)\n\n\tstate3, err6 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err6)\n\tinfo3 := state3.ExecutionInfo\n\ts.NotNil(info3, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info3.DomainID)\n\ts.Equal(\"update-workflow-test\", info3.WorkflowID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info3.RunID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info3.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info3.TaskList)\n\ts.Equal(\"wType\", info3.WorkflowTypeName)\n\ts.Equal(int32(20), info3.WorkflowTimeout)\n\ts.Equal(int32(13), info3.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info3.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info3.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info3.CloseStatus)\n\ts.Equal(int64(5), info3.NextEventID)\n\ts.Equal(int64(2), info3.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info3.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(int64(666), info3.DecisionVersion)\n\ts.Equal(int64(2), info3.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info3.DecisionStartedID)\n\ts.Equal(int32(1), info3.DecisionTimeout)\n\ts.Equal(int64(123), info3.DecisionAttempt)\n\ts.Equal(int64(321), info3.DecisionStartedTimestamp)\n\ts.Equal(int64(654), info3.DecisionScheduledTimestamp)\n\ts.Equal(int64(655), info3.DecisionOriginalScheduledTimestamp)\n\ts.Equal(updatedInfo.SignalCount, info3.SignalCount)\n\ts.EqualValues(updatedStats.HistorySize, state3.ExecutionStats.HistorySize)\n\ts.Equal(updatedInfo.InitialInterval, info3.InitialInterval)\n\ts.Equal(updatedInfo.BackoffCoefficient, info3.BackoffCoefficient)\n\ts.Equal(updatedInfo.MaximumInterval, info3.MaximumInterval)\n\ts.Equal(updatedInfo.MaximumAttempts, info3.MaximumAttempts)\n\ts.Equal(updatedInfo.ExpirationSeconds, info3.ExpirationSeconds)\n\ts.EqualTimes(updatedInfo.ExpirationTime, info3.ExpirationTime)\n\ts.Equal(updatedInfo.NonRetriableErrors, info3.NonRetriableErrors)\n\tsearchAttrVal3, ok := info3.SearchAttributes[searchAttrKey]\n\ts.True(ok)\n\ts.Equal(searchAttrVal, searchAttrVal3)\n\tmemoVal3, ok := info1.Memo[memoKey]\n\ts.True(ok)\n\ts.Equal(memoVal, memoVal3)\n\ts.Equal(partitionConfig, info3.PartitionConfig)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state3.Checksum)\n\n\ts.T().Logf(\"Workflow execution last updated: %v\\n\", info3.LastUpdatedTimestamp)\n\n\t// update with incorrect rangeID and condition(next_event_id)\n\terr7 := s.UpdateWorkflowExecutionWithRangeID(ctx, failedUpdateInfo, failedUpdateStats, versionHistories, []int64{int64(5)}, nil, int64(12345), int64(3), nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil)\n\ts.Error(err7, \"expected non nil error.\")\n\ts.IsType(&p.ShardOwnershipLostError{}, err7)\n\n\tstate4, err8 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err8)\n\tinfo4 := state4.ExecutionInfo\n\ts.NotNil(info4, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info4.DomainID)\n\ts.Equal(\"update-workflow-test\", info4.WorkflowID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info4.RunID)\n\ts.Equal(\"5ba5e531-e46b-48d9-b4b3-859919839553\", info4.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info4.TaskList)\n\ts.Equal(\"wType\", info4.WorkflowTypeName)\n\ts.Equal(int32(20), info4.WorkflowTimeout)\n\ts.Equal(int32(13), info4.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info4.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info4.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info4.CloseStatus)\n\ts.Equal(int64(5), info4.NextEventID)\n\ts.Equal(int64(2), info4.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info4.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(int64(666), info4.DecisionVersion)\n\ts.Equal(int64(2), info4.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info4.DecisionStartedID)\n\ts.Equal(int32(1), info4.DecisionTimeout)\n\ts.Equal(int64(123), info4.DecisionAttempt)\n\ts.Equal(int64(321), info4.DecisionStartedTimestamp)\n\ts.Equal(updatedInfo.SignalCount, info4.SignalCount)\n\ts.EqualValues(updatedStats.HistorySize, state4.ExecutionStats.HistorySize)\n\ts.Equal(updatedInfo.InitialInterval, info4.InitialInterval)\n\ts.Equal(updatedInfo.BackoffCoefficient, info4.BackoffCoefficient)\n\ts.Equal(updatedInfo.MaximumInterval, info4.MaximumInterval)\n\ts.Equal(updatedInfo.MaximumAttempts, info4.MaximumAttempts)\n\ts.Equal(updatedInfo.ExpirationSeconds, info4.ExpirationSeconds)\n\ts.EqualTimes(updatedInfo.ExpirationTime, info4.ExpirationTime)\n\ts.Equal(updatedInfo.NonRetriableErrors, info4.NonRetriableErrors)\n\tsearchAttrVal4, ok := info4.SearchAttributes[searchAttrKey]\n\ts.True(ok)\n\ts.Equal(searchAttrVal, searchAttrVal4)\n\tmemoVal4, ok := info1.Memo[memoKey]\n\ts.True(ok)\n\ts.Equal(memoVal, memoVal4)\n\ts.Equal(partitionConfig, info4.PartitionConfig)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state4.Checksum)\n\n\ts.T().Logf(\"Workflow execution last updated: %v\\n\", info4.LastUpdatedTimestamp)\n}\n\n// TestDeleteWorkflow test\nfunc (s *ExecutionManagerSuite) TestDeleteWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"1d4abb23-b87b-457b-96ef-43aba0b9c44f\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"delete-workflow-test\",\n\t\tRunID:      \"4e0917f2-9361-4a14-b16f-1fafe09b287a\",\n\t}\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(domainID, info0.DomainID)\n\ts.Equal(\"delete-workflow-test\", info0.WorkflowID)\n\ts.Equal(\"4e0917f2-9361-4a14-b16f-1fafe09b287a\", info0.RunID)\n\ts.Equal(\"4e0917f2-9361-4a14-b16f-1fafe09b287a\", info0.FirstExecutionRunID)\n\ts.Equal(\"queue1\", info0.TaskList)\n\ts.Equal(\"wType\", info0.WorkflowTypeName)\n\ts.Equal(int32(20), info0.WorkflowTimeout)\n\ts.Equal(int32(13), info0.DecisionStartToCloseTimeout)\n\ts.Equal([]byte(nil), info0.ExecutionContext)\n\ts.Equal(p.WorkflowStateRunning, info0.State)\n\ts.Equal(p.WorkflowCloseStatusNone, info0.CloseStatus)\n\ts.Equal(int64(3), info0.NextEventID)\n\ts.Equal(int64(0), info0.LastProcessedEvent)\n\ts.Equal(true, s.validateTimeRange(info0.LastUpdatedTimestamp, time.Hour))\n\ts.Equal(int64(2), info0.DecisionScheduleID)\n\ts.Equal(constants.EmptyEventID, info0.DecisionStartedID)\n\ts.Equal(int32(1), info0.DecisionTimeout)\n\n\ts.T().Logf(\"Workflow execution last updated: %v\\n\", info0.LastUpdatedTimestamp)\n\n\terr4 := s.DeleteWorkflowExecution(ctx, info0)\n\ts.NoError(err4)\n\n\t_, err3 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Error(err3, \"expected non nil error.\")\n\ts.IsType(&types.EntityNotExistsError{}, err3)\n\n\terr5 := s.DeleteWorkflowExecution(ctx, info0)\n\ts.NoError(err5)\n}\n\n// TestDeleteCurrentWorkflow test\nfunc (s *ExecutionManagerSuite) TestDeleteCurrentWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tif s.ExecutionManager.GetName() != \"cassandra\" {\n\t\t// \"this test is only applicable for cassandra (uses TTL based deletes)\"\n\t\treturn\n\t}\n\tdomainID := \"54d15308-e20e-4b91-a00f-a518a3892790\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"delete-current-workflow-test\",\n\t\tRunID:      \"6cae4054-6ba7-46d3-8755-e3c2db6f74ea\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\trunID0, err1 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err1)\n\ts.Equal(workflowExecution.GetRunID(), runID0)\n\n\tinfo0, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info0.ExecutionInfo)\n\tupdatedStats1 := copyExecutionStats(info0.ExecutionStats)\n\tupdatedInfo1.NextEventID = int64(6)\n\tupdatedInfo1.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo1.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr3 := s.UpdateWorkflowExecutionAndFinish(ctx, updatedInfo1, updatedStats1, int64(3), versionHistories)\n\ts.NoError(err3)\n\n\trunID4, err4 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err4)\n\ts.Equal(workflowExecution.GetRunID(), runID4)\n\n\tfakeInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:   info0.ExecutionInfo.DomainID,\n\t\tWorkflowID: info0.ExecutionInfo.WorkflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\n\t// test wrong run id with conditional delete\n\ts.NoError(s.DeleteCurrentWorkflowExecution(ctx, fakeInfo))\n\n\trunID5, err5 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err5)\n\ts.Equal(workflowExecution.GetRunID(), runID5)\n\n\t// simulate a timer_task deleting execution after retention\n\ts.NoError(s.DeleteCurrentWorkflowExecution(ctx, info0.ExecutionInfo))\n\n\trunID0, err1 = s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.Error(err1)\n\ts.Empty(runID0)\n\t_, ok := err1.(*types.EntityNotExistsError)\n\ts.True(ok)\n\n\t// execution record should still be there\n\t_, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n}\n\n// TestUpdateDeleteWorkflow mocks the timer behavior to clean up workflow.\nfunc (s *ExecutionManagerSuite) TestUpdateDeleteWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tfinishedCurrentExecutionRetentionTTL := int32(2)\n\tdomainID := \"54d15308-e20e-4b91-a00f-a518a3892790\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"update-delete-workflow-test\",\n\t\tRunID:      \"6cae4054-6ba7-46d3-8755-e3c2db6f74ea\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\trunID0, err1 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err1)\n\ts.Equal(workflowExecution.GetRunID(), runID0)\n\n\tinfo0, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info0.ExecutionInfo)\n\tupdatedStats1 := copyExecutionStats(info0.ExecutionStats)\n\tupdatedInfo1.NextEventID = int64(6)\n\tupdatedInfo1.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo1.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr3 := s.UpdateWorkflowExecutionAndFinish(ctx, updatedInfo1, updatedStats1, int64(3), versionHistories)\n\ts.NoError(err3)\n\n\trunID4, err4 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err4)\n\ts.Equal(workflowExecution.GetRunID(), runID4)\n\n\t// simulate a timer_task deleting execution after retention\n\terr5 := s.DeleteCurrentWorkflowExecution(ctx, info0.ExecutionInfo)\n\ts.NoError(err5)\n\terr6 := s.DeleteWorkflowExecution(ctx, info0.ExecutionInfo)\n\ts.NoError(err6)\n\n\ttime.Sleep(time.Duration(finishedCurrentExecutionRetentionTTL*2) * time.Second)\n\n\trunID0, err1 = s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.Error(err1)\n\ts.Empty(runID0)\n\t_, ok := err1.(*types.EntityNotExistsError)\n\ts.True(ok)\n\n\t// execution record should still be there\n\t_, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Error(err2)\n\t_, ok = err2.(*types.EntityNotExistsError)\n\ts.True(ok)\n}\n\n// TestCleanupCorruptedWorkflow test\nfunc (s *ExecutionManagerSuite) TestCleanupCorruptedWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"54d15308-e20e-4b91-a00f-a518a3892790\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"cleanup-corrupted-workflow-test\",\n\t\tRunID:      \"6cae4054-6ba7-46d3-8755-e3c2db6f74ea\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\trunID0, err1 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err1)\n\ts.Equal(workflowExecution.GetRunID(), runID0)\n\n\tinfo0, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\n\t// deleting current record and verify\n\terr3 := s.DeleteCurrentWorkflowExecution(ctx, info0.ExecutionInfo)\n\ts.NoError(err3)\n\trunID0, err4 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.Error(err4)\n\ts.Empty(runID0)\n\t_, ok := err4.(*types.EntityNotExistsError)\n\ts.True(ok)\n\n\t// we should still be able to load with runID\n\tinfo1, err5 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err5)\n\ts.Equal(info0, info1)\n\n\t// mark it as corrupted\n\tinfo0.ExecutionInfo.State = p.WorkflowStateCorrupted\n\t_, err6 := s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:    info0.ExecutionInfo,\n\t\t\tExecutionStats:   info0.ExecutionStats,\n\t\t\tCondition:        info0.ExecutionInfo.NextEventID,\n\t\t\tChecksum:         testWorkflowChecksum,\n\t\t\tVersionHistories: info0.VersionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.UpdateWorkflowModeBypassCurrent,\n\t})\n\ts.NoError(err6)\n\n\t// we should still be able to load with runID\n\tinfo2, err7 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err7)\n\ts.Equal(p.WorkflowStateCorrupted, info2.ExecutionInfo.State)\n\tinfo2.ExecutionInfo.State = info1.ExecutionInfo.State\n\tinfo2.ExecutionInfo.LastUpdatedTimestamp = info1.ExecutionInfo.LastUpdatedTimestamp\n\ts.Equal(info2, info1)\n\n\t// delete the run\n\terr8 := s.DeleteWorkflowExecution(ctx, info0.ExecutionInfo)\n\ts.NoError(err8)\n\n\t// execution record should be gone\n\t_, err9 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.Error(err9)\n\t_, ok = err9.(*types.EntityNotExistsError)\n\ts.True(ok)\n}\n\n// TestGetCurrentWorkflow test\nfunc (s *ExecutionManagerSuite) TestGetCurrentWorkflow() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"54d15308-e20e-4b91-a00f-a518a3892790\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-current-workflow-test\",\n\t\tRunID:      \"6cae4054-6ba7-46d3-8755-e3c2db6f74ea\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tresponse, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t})\n\ts.NoError(err)\n\ts.Equal(workflowExecution.GetRunID(), response.RunID)\n\ts.Equal(constants.EmptyVersion, response.LastWriteVersion)\n\n\tinfo0, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info0.ExecutionInfo)\n\tupdatedStats1 := copyExecutionStats(info0.ExecutionStats)\n\tupdatedInfo1.NextEventID = int64(6)\n\tupdatedInfo1.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo1.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr3 := s.UpdateWorkflowExecutionAndFinish(ctx, updatedInfo1, updatedStats1, int64(3), versionHistories)\n\ts.NoError(err3)\n\n\trunID4, err4 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err4)\n\ts.Equal(workflowExecution.GetRunID(), runID4)\n\n\tworkflowExecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"get-current-workflow-test\",\n\t\tRunID:      \"c3ff4bc6-de18-4643-83b2-037a33f45322\",\n\t}\n\n\ttask1, err5 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution2, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.Error(err5, \"Error expected.\")\n\ts.Empty(task1, \"Expected empty task identifier.\")\n}\n\n// TestTransferTasksThroughUpdate test\nfunc (s *ExecutionManagerSuite) TestTransferTasksThroughUpdate() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"b785a8ba-bd7d-4760-bb05-41b115f3e10a\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-transfer-tasks-through-update-test\",\n\t\tRunID:      \"30a9fa1f-0db1-4d7a-8c34-aa82c5dad3aa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttasks1, err1 := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err1)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 transfer task.\")\n\ttask1, ok := tasks1[0].(*p.DecisionTask)\n\ts.True(ok, \"Expected a decision task.\")\n\ts.Equal(domainID, task1.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task1.GetRunID())\n\ts.Equal(\"queue1\", task1.TaskList)\n\ts.Equal(p.TransferTaskTypeDecisionTask, task1.GetTaskType())\n\ts.Equal(int64(2), task1.ScheduleID)\n\n\terr3 := s.CompleteTransferTask(ctx, task1.TaskID)\n\ts.NoError(err3)\n\n\tstate0, err11 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err11)\n\tinfo0 := state0.ExecutionInfo\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats0 := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats0, versionHistories, nil, []int64{int64(4)}, int64(3), nil, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\ttasks2, err1 := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err1)\n\ts.NotNil(tasks2, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks2), \"Expected 1 transfer task.\")\n\ttask2, ok := tasks2[0].(*p.ActivityTask)\n\ts.True(ok, \"Expected an activity task.\")\n\ts.Equal(domainID, task2.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task2.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task2.GetRunID())\n\ts.Equal(\"queue1\", task2.TaskList)\n\ts.Equal(p.TransferTaskTypeActivityTask, task2.GetTaskType())\n\ts.Equal(int64(4), task2.ScheduleID)\n\n\terr4 := s.CompleteTransferTask(ctx, task2.TaskID)\n\ts.NoError(err4)\n\n\tstate1, _ := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\tinfo1 := state1.ExecutionInfo\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\tupdatedInfo1.NextEventID = int64(6)\n\tupdatedInfo1.LastProcessedEvent = int64(2)\n\terr5 := s.UpdateWorkflowExecutionAndFinish(ctx, updatedInfo1, updatedStats1, int64(5), versionHistories)\n\ts.NoError(err5)\n\n\tnewExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\tRunID:      \"2a038c8f-b575-4151-8d2c-d443e999ab5a\",\n\t}\n\trunID6, err6 := s.GetCurrentWorkflowRunID(ctx, domainID, newExecution.GetWorkflowID())\n\ts.NoError(err6)\n\ts.Equal(workflowExecution.GetRunID(), runID6)\n\n\ttasks3, err7 := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err7)\n\ts.NotNil(tasks3, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks3), \"Expected 1 transfer task.\")\n\ttask3, ok := tasks3[0].(*p.CloseExecutionTask)\n\ts.True(ok, \"Expected a close execution task.\")\n\ts.Equal(domainID, task3.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task3.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task3.GetRunID())\n\ts.Equal(p.TransferTaskTypeCloseExecution, task3.GetTaskType())\n\n\terr8 := s.CompleteTransferTask(ctx, task3.TaskID)\n\ts.NoError(err8)\n\n\t_, err9 := s.CreateWorkflowExecution(ctx, domainID, newExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.Error(err9, \"createWFExecution (brand_new) must fail when there is a previous instance of workflow state already in DB\")\n\n\terr10 := s.DeleteWorkflowExecution(ctx, info1)\n\ts.NoError(err10)\n}\n\n// TestCancelTransferTaskTasks test\nfunc (s *ExecutionManagerSuite) TestCancelTransferTaskTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"aeac8287-527b-4b35-80a9-667cb47e7c6d\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"cancel-workflow-test\",\n\t\tRunID:      \"db20f7e2-1a1e-40d9-9278-d8b886738e05\",\n\t}\n\n\ttask0, err := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\ttargetDomainID := \"f2bfaab6-7e8b-4fac-9a62-17da8d37becb\"\n\ttargetWorkflowID := \"target-workflow-cancellation-id-1\"\n\ttargetRunID := \"0d00698f-08e1-4d36-a3e2-3bf109f5d2d6\"\n\ttargetChildWorkflowOnly := false\n\ttransferTasks := []p.Task{&p.CancelExecutionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: targetChildWorkflowOnly,\n\t\tInitiatedID:             1,\n\t\tTaskList:                \"queue1\",\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 1,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr = s.UpdateWorkflowExecutionWithTransferTasks(ctx, updatedInfo1, updatedStats1, int64(3), transferTasks, nil, versionHistories)\n\ts.NoError(err)\n\n\ttasks1, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 cancel task.\")\n\ttask1, ok := tasks1[0].(*p.CancelExecutionTask)\n\ts.True(ok, \"Expected a cancel execution task.\")\n\ts.Equal(p.TransferTaskTypeCancelExecution, task1.GetTaskType())\n\ts.Equal(domainID, task1.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task1.GetRunID())\n\ts.Equal(targetDomainID, task1.TargetDomainID)\n\ts.Equal(targetWorkflowID, task1.TargetWorkflowID)\n\ts.Equal(targetRunID, task1.TargetRunID)\n\ts.Equal(false, task1.TargetChildWorkflowOnly)\n\ts.Equal(\"queue1\", task1.GetTaskList())\n\ts.Equal(\"queue1\", task1.GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, task1.GetOriginalTaskListKind())\n\n\terr = s.CompleteTransferTask(ctx, task1.TaskID)\n\ts.NoError(err)\n\n\ttargetDomainID = \"f2bfaab6-7e8b-4fac-9a62-17da8d37becb\"\n\ttargetWorkflowID = \"target-workflow-cancellation-id-2\"\n\ttargetRunID = \"\"\n\ttargetChildWorkflowOnly = true\n\ttransferTasks = []p.Task{&p.CancelExecutionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: targetChildWorkflowOnly,\n\t\tInitiatedID:             3,\n\t\tTaskList:                \"queue2\",\n\t}}\n\n\tstate2, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo2 := state2.ExecutionInfo\n\ts.NotNil(info2, \"Valid Workflow info expected.\")\n\tupdatedInfo2 := copyWorkflowExecutionInfo(info2)\n\tupdatedStats2 := copyExecutionStats(state2.ExecutionStats)\n\n\terr = s.UpdateWorkflowExecutionWithTransferTasks(ctx, updatedInfo2, updatedStats2, int64(3), transferTasks, nil, versionHistories)\n\ts.NoError(err)\n\n\ttasks2, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.NotNil(tasks2, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks2), \"Expected 1 cancel task.\")\n\ttask2, ok := tasks2[0].(*p.CancelExecutionTask)\n\ts.True(ok, \"Expected a cancel execution task.\")\n\ts.Equal(p.TransferTaskTypeCancelExecution, task2.GetTaskType())\n\ts.Equal(domainID, task2.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task2.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task2.GetRunID())\n\ts.Equal(targetDomainID, task2.TargetDomainID)\n\ts.Equal(targetWorkflowID, task2.TargetWorkflowID)\n\ts.Equal(targetRunID, task2.TargetRunID)\n\ts.Equal(targetChildWorkflowOnly, task2.TargetChildWorkflowOnly)\n\ts.Equal(\"queue2\", task2.GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, task2.GetOriginalTaskListKind())\n\n\terr = s.CompleteTransferTask(ctx, task2.TaskID)\n\ts.NoError(err)\n}\n\n// TestSignalTransferTaskTasks test\nfunc (s *ExecutionManagerSuite) TestSignalTransferTaskTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"aeac8287-527b-4b35-80a9-667cb47e7c6d\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"signal-workflow-test\",\n\t\tRunID:      \"db20f7e2-1a1e-40d9-9278-d8b886738e05\",\n\t}\n\n\ttask0, err := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\ttargetDomainID := \"f2bfaab6-7e8b-4fac-9a62-17da8d37becb\"\n\ttargetWorkflowID := \"target-workflow-signal-id-1\"\n\ttargetRunID := \"0d00698f-08e1-4d36-a3e2-3bf109f5d2d6\"\n\ttargetChildWorkflowOnly := false\n\ttransferTasks := []p.Task{&p.SignalExecutionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: false,\n\t\tInitiatedID:             1,\n\t\tTaskList:                \"queue1\",\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 1,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr = s.UpdateWorkflowExecutionWithTransferTasks(ctx, updatedInfo1, updatedStats1, int64(3), transferTasks, nil, versionHistories)\n\ts.NoError(err)\n\n\ttasks1, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 transfer task.\")\n\ttask1, ok := tasks1[0].(*p.SignalExecutionTask)\n\ts.True(ok, \"Expected a signal execution task.\")\n\ts.Equal(p.TransferTaskTypeSignalExecution, task1.GetTaskType())\n\ts.Equal(domainID, task1.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task1.GetRunID())\n\ts.Equal(targetDomainID, task1.TargetDomainID)\n\ts.Equal(targetWorkflowID, task1.TargetWorkflowID)\n\ts.Equal(targetRunID, task1.TargetRunID)\n\ts.Equal(false, task1.TargetChildWorkflowOnly)\n\ts.Equal(\"queue1\", task1.GetTaskList())\n\ts.Equal(\"queue1\", task1.GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, task1.GetOriginalTaskListKind())\n\n\terr = s.CompleteTransferTask(ctx, task1.TaskID)\n\ts.NoError(err)\n\n\ttargetDomainID = \"f2bfaab6-7e8b-4fac-9a62-17da8d37becb\"\n\ttargetWorkflowID = \"target-workflow-signal-id-2\"\n\ttargetRunID = \"\"\n\ttargetChildWorkflowOnly = true\n\ttransferTasks = []p.Task{&p.SignalExecutionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: targetChildWorkflowOnly,\n\t\tInitiatedID:             3,\n\t\tTaskList:                \"queue2\",\n\t}}\n\n\tstate2, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo2 := state2.ExecutionInfo\n\ts.NotNil(info2, \"Valid Workflow info expected.\")\n\tupdatedInfo2 := copyWorkflowExecutionInfo(info2)\n\tupdatedStats2 := copyExecutionStats(state2.ExecutionStats)\n\n\terr = s.UpdateWorkflowExecutionWithTransferTasks(ctx, updatedInfo2, updatedStats2, int64(3), transferTasks, nil, versionHistories)\n\ts.NoError(err)\n\n\ttasks2, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.NotNil(tasks2, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks2), \"Expected 1 transfer task.\")\n\ttask2, ok := tasks2[0].(*p.SignalExecutionTask)\n\ts.True(ok, \"Expected a signal execution task.\")\n\ts.Equal(p.TransferTaskTypeSignalExecution, task2.GetTaskType())\n\ts.Equal(domainID, task2.GetDomainID())\n\ts.Equal(workflowExecution.GetWorkflowID(), task2.GetWorkflowID())\n\ts.Equal(workflowExecution.GetRunID(), task2.GetRunID())\n\ts.Equal(targetDomainID, task2.TargetDomainID)\n\ts.Equal(targetWorkflowID, task2.TargetWorkflowID)\n\ts.Equal(targetRunID, task2.TargetRunID)\n\ts.Equal(targetChildWorkflowOnly, task2.TargetChildWorkflowOnly)\n\ts.Equal(\"queue2\", task2.GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, task2.GetOriginalTaskListKind())\n\n\terr = s.CompleteTransferTask(ctx, task2.TaskID)\n\ts.NoError(err)\n}\n\n// TestReplicationTasks test\nfunc (s *ExecutionManagerSuite) TestReplicationTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"2466d7de-6602-4ad8-b939-fb8f8c36c711\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-replication-tasks-test\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\treplicationTasks := []p.Task{\n\t\t&p.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\t\tVersion: 123,\n\t\t\t},\n\t\t\tFirstEventID: int64(1),\n\t\t\tNextEventID:  int64(3),\n\t\t},\n\t\t&p.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\t\tVersion: 456,\n\t\t\t},\n\t\t\tFirstEventID: int64(1),\n\t\t\tNextEventID:  int64(3),\n\t\t},\n\t\t&p.SyncActivityTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\t\tVersion: 789,\n\t\t\t},\n\t\t\tScheduledID: 99,\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 3,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr = s.UpdateWorklowStateAndReplication(ctx, updatedInfo1, updatedStats1, versionHistories, int64(3), replicationTasks)\n\ts.NoError(err)\n\n\trespTasks, err := s.GetReplicationTasks(ctx, 1, true)\n\ts.NoError(err)\n\ts.Equal(len(replicationTasks), len(respTasks))\n\n\tfor index := range replicationTasks {\n\t\trespTasks[index].SetVisibilityTimestamp(replicationTasks[index].GetVisibilityTimestamp()) // ignore visibility timestamp\n\t\ts.Equal(replicationTasks[index], respTasks[index])\n\t\terr = s.CompleteReplicationTask(ctx, respTasks[index].GetTaskID())\n\t\ts.NoError(err)\n\t}\n}\n\n// TestTransferTasksComplete test\nfunc (s *ExecutionManagerSuite) TestTransferTasksComplete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"8bfb47be-5b57-4d55-9109-5fb35e20b1d7\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-transfer-tasks-test-complete\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\ttasklist := \"some random tasklist\"\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, tasklist, \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttasks1, err1 := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err1)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 decision task.\")\n\ttask1, ok := tasks1[0].(*p.DecisionTask)\n\ts.True(ok, \"Expected a decision task.\")\n\ts.Equal(domainID, task1.DomainID)\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.WorkflowID)\n\ts.Equal(workflowExecution.GetRunID(), task1.RunID)\n\ts.Equal(tasklist, task1.TaskList)\n\ts.Equal(p.TransferTaskTypeDecisionTask, task1.GetTaskType())\n\ts.Equal(int64(2), task1.ScheduleID)\n\terr3 := s.CompleteTransferTask(ctx, task1.TaskID)\n\ts.NoError(err3)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(6)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tscheduleID := int64(123)\n\ttargetDomainID := \"8bfb47be-5b57-4d66-9109-5fb35e20b1d0\"\n\ttargetWorkflowID := \"some random target domain ID\"\n\ttargetRunID := uuid.New()\n\tcurrentTransferID := task1.TaskID\n\tnow := time.Now()\n\ttasks := []p.Task{\n\t\t&p.ActivityTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10000,\n\t\t\t\tVersion:             111,\n\t\t\t},\n\t\t\tTargetDomainID: domainID,\n\t\t\tTaskList:       tasklist,\n\t\t\tScheduleID:     scheduleID,\n\t\t},\n\t\t&p.DecisionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10002,\n\t\t\t\tVersion:             222,\n\t\t\t},\n\t\t\tTargetDomainID:       domainID,\n\t\t\tTaskList:             tasklist,\n\t\t\tScheduleID:           scheduleID,\n\t\t\tOriginalTaskList:     \"original_tasklist\",\n\t\t\tOriginalTaskListKind: types.TaskListKindEphemeral,\n\t\t},\n\t\t&p.CloseExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskList: tasklist,\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10003,\n\t\t\t\tVersion:             333,\n\t\t\t}},\n\t\t&p.CancelExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10004,\n\t\t\t\tVersion:             444,\n\t\t\t},\n\t\t\tTargetDomainID:          targetDomainID,\n\t\t\tTargetWorkflowID:        targetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tInitiatedID:             scheduleID,\n\t\t\tTaskList:                tasklist,\n\t\t},\n\t\t&p.SignalExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10005,\n\t\t\t\tVersion:             555,\n\t\t\t},\n\t\t\tTargetDomainID:          targetDomainID,\n\t\t\tTargetWorkflowID:        targetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tInitiatedID:             scheduleID,\n\t\t\tTaskList:                tasklist,\n\t\t},\n\t\t&p.StartChildExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10006,\n\t\t\t\tVersion:             666,\n\t\t\t},\n\t\t\tTargetDomainID:   targetDomainID,\n\t\t\tTargetWorkflowID: targetWorkflowID,\n\t\t\tInitiatedID:      scheduleID,\n\t\t\tTaskList:         tasklist,\n\t\t},\n\t\t&p.RecordWorkflowClosedTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10007,\n\t\t\t\tVersion:             777,\n\t\t\t},\n\t\t\tTaskList: tasklist,\n\t\t},\n\t\t&p.RecordChildExecutionCompletedTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10008,\n\t\t\t\tVersion:             888,\n\t\t\t},\n\t\t\tTargetDomainID:   targetDomainID,\n\t\t\tTargetWorkflowID: targetWorkflowID,\n\t\t\tTargetRunID:      targetRunID,\n\t\t\tTaskList:         tasklist,\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: scheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorklowStateAndReplication(ctx, updatedInfo, updatedStats, versionHistories, int64(3), tasks)\n\ts.NoError(err2)\n\n\ttxTasks, err1 := s.GetTransferTasks(ctx, 1, true) // use page size one to force pagination\n\ts.NoError(err1)\n\ts.NotNil(txTasks, \"expected valid list of tasks.\")\n\ts.Equal(len(tasks), len(txTasks))\n\tfor index := range tasks {\n\t\ts.True(timeComparator(tasks[index].GetVisibilityTimestamp(), txTasks[index].GetVisibilityTimestamp(), TimePrecision))\n\t}\n\ts.Equal(p.TransferTaskTypeActivityTask, txTasks[0].GetTaskType())\n\ts.Equal(tasklist, txTasks[0].GetTaskList())\n\ts.Equal(tasklist, txTasks[0].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[0].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeDecisionTask, txTasks[1].GetTaskType())\n\ts.Equal(tasklist, txTasks[1].GetTaskList())\n\ts.Equal(\"original_tasklist\", txTasks[1].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindEphemeral, txTasks[1].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeCloseExecution, txTasks[2].GetTaskType())\n\ts.Equal(tasklist, txTasks[2].GetTaskList())\n\ts.Equal(tasklist, txTasks[2].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[2].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeCancelExecution, txTasks[3].GetTaskType())\n\ts.Equal(tasklist, txTasks[3].GetTaskList())\n\ts.Equal(tasklist, txTasks[3].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[3].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeSignalExecution, txTasks[4].GetTaskType())\n\ts.Equal(tasklist, txTasks[4].GetTaskList())\n\ts.Equal(tasklist, txTasks[4].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[4].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeStartChildExecution, txTasks[5].GetTaskType())\n\ts.Equal(tasklist, txTasks[5].GetTaskList())\n\ts.Equal(tasklist, txTasks[5].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[5].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeRecordWorkflowClosed, txTasks[6].GetTaskType())\n\ts.Equal(tasklist, txTasks[6].GetTaskList())\n\ts.Equal(tasklist, txTasks[6].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[6].GetOriginalTaskListKind())\n\ts.Equal(p.TransferTaskTypeRecordChildExecutionCompleted, txTasks[7].GetTaskType())\n\ts.Equal(tasklist, txTasks[7].GetTaskList())\n\ts.Equal(tasklist, txTasks[7].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[7].GetOriginalTaskListKind())\n\n\tfor idx := range txTasks {\n\t\t// TODO: add a check similar to validateCrossClusterTasks\n\t\ts.Equal(int64(111*(idx+1)), txTasks[idx].GetVersion())\n\t\terr := s.CompleteTransferTask(ctx, txTasks[idx].GetTaskID())\n\t\ts.NoError(err)\n\t}\n\n\ttxTasks, err2 = s.GetTransferTasks(ctx, 100, false)\n\ts.NoError(err2)\n\ts.Empty(txTasks, \"expected empty task list.\")\n}\n\n// TestTransferTasksRangeComplete test\nfunc (s *ExecutionManagerSuite) TestTransferTasksRangeComplete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"8bfb47be-5b57-4d55-9109-5fb35e20b1d7\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-transfer-tasks-test-range-complete\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\ttasklist := \"some random tasklist\"\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, tasklist, \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttasks1, err1 := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err1)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 decision task.\")\n\ttask1, ok := tasks1[0].(*p.DecisionTask)\n\ts.True(ok, \"Expected a decision task.\")\n\ts.Equal(domainID, task1.DomainID)\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.WorkflowID)\n\ts.Equal(workflowExecution.GetRunID(), task1.RunID)\n\ts.Equal(tasklist, task1.TaskList)\n\ts.Equal(p.TransferTaskTypeDecisionTask, task1.GetTaskType())\n\ts.Equal(int64(2), task1.ScheduleID)\n\terr3 := s.CompleteTransferTask(ctx, task1.TaskID)\n\ts.NoError(err3)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(6)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tscheduleID := int64(123)\n\ttargetDomainID := \"8bfb47be-5b57-4d66-9109-5fb35e20b1d0\"\n\ttargetWorkflowID := \"some random target domain ID\"\n\ttargetRunID := uuid.New()\n\tcurrentTransferID := task1.TaskID\n\tnow := time.Now()\n\ttasks := []p.Task{\n\t\t&p.ActivityTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10001,\n\t\t\t\tVersion:             111,\n\t\t\t},\n\t\t\tTargetDomainID: domainID,\n\t\t\tTaskList:       tasklist,\n\t\t\tScheduleID:     scheduleID,\n\t\t},\n\t\t&p.DecisionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10002,\n\t\t\t\tVersion:             222,\n\t\t\t},\n\t\t\tTargetDomainID:       domainID,\n\t\t\tTaskList:             tasklist,\n\t\t\tScheduleID:           scheduleID,\n\t\t\tOriginalTaskList:     \"original_tasklist\",\n\t\t\tOriginalTaskListKind: types.TaskListKindEphemeral,\n\t\t},\n\t\t&p.CloseExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10003,\n\t\t\t\tVersion:             333,\n\t\t\t},\n\t\t\tTaskList: tasklist,\n\t\t},\n\t\t&p.CancelExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10004,\n\t\t\t\tVersion:             444,\n\t\t\t},\n\t\t\tTargetDomainID:          targetDomainID,\n\t\t\tTargetWorkflowID:        targetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tInitiatedID:             scheduleID,\n\t\t\tTaskList:                tasklist,\n\t\t},\n\t\t&p.SignalExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10005,\n\t\t\t\tVersion:             555,\n\t\t\t},\n\t\t\tTargetDomainID:          targetDomainID,\n\t\t\tTargetWorkflowID:        targetWorkflowID,\n\t\t\tTargetRunID:             targetRunID,\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tInitiatedID:             scheduleID,\n\t\t\tTaskList:                tasklist,\n\t\t},\n\t\t&p.StartChildExecutionTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              currentTransferID + 10006,\n\t\t\t\tVersion:             666,\n\t\t\t},\n\t\t\tTargetDomainID:   targetDomainID,\n\t\t\tTargetWorkflowID: targetWorkflowID,\n\t\t\tInitiatedID:      scheduleID,\n\t\t\tTaskList:         tasklist,\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: scheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorklowStateAndReplication(ctx, updatedInfo, updatedStats, versionHistories, int64(3), tasks)\n\ts.NoError(err2)\n\n\ttxTasks, err1 := s.GetTransferTasks(ctx, 2, true) // use page size one to force pagination\n\ts.NoError(err1)\n\ts.NotNil(txTasks, \"expected valid list of tasks.\")\n\ts.Equal(len(tasks), len(txTasks))\n\tfor index := range tasks {\n\t\ts.True(timeComparator(tasks[index].GetVisibilityTimestamp(), txTasks[index].GetVisibilityTimestamp(), TimePrecision))\n\t}\n\ts.Equal(p.TransferTaskTypeActivityTask, txTasks[0].GetTaskType())\n\ts.Equal(p.TransferTaskTypeDecisionTask, txTasks[1].GetTaskType())\n\ts.Equal(p.TransferTaskTypeCloseExecution, txTasks[2].GetTaskType())\n\ts.Equal(p.TransferTaskTypeCancelExecution, txTasks[3].GetTaskType())\n\ts.Equal(p.TransferTaskTypeSignalExecution, txTasks[4].GetTaskType())\n\ts.Equal(p.TransferTaskTypeStartChildExecution, txTasks[5].GetTaskType())\n\ts.Equal(int64(111), txTasks[0].GetVersion())\n\ts.Equal(int64(222), txTasks[1].GetVersion())\n\ts.Equal(int64(333), txTasks[2].GetVersion())\n\ts.Equal(int64(444), txTasks[3].GetVersion())\n\ts.Equal(int64(555), txTasks[4].GetVersion())\n\ts.Equal(int64(666), txTasks[5].GetVersion())\n\ts.Equal(currentTransferID+10001, txTasks[0].GetTaskID())\n\ts.Equal(currentTransferID+10002, txTasks[1].GetTaskID())\n\ts.Equal(currentTransferID+10003, txTasks[2].GetTaskID())\n\ts.Equal(currentTransferID+10004, txTasks[3].GetTaskID())\n\ts.Equal(currentTransferID+10005, txTasks[4].GetTaskID())\n\ts.Equal(currentTransferID+10006, txTasks[5].GetTaskID())\n\ts.Equal(tasklist, txTasks[0].GetTaskList())\n\ts.Equal(tasklist, txTasks[1].GetTaskList())\n\ts.Equal(tasklist, txTasks[2].GetTaskList())\n\ts.Equal(tasklist, txTasks[3].GetTaskList())\n\ts.Equal(tasklist, txTasks[4].GetTaskList())\n\ts.Equal(tasklist, txTasks[5].GetTaskList())\n\ts.Equal(tasklist, txTasks[0].GetOriginalTaskList())\n\ts.Equal(\"original_tasklist\", txTasks[1].GetOriginalTaskList())\n\ts.Equal(tasklist, txTasks[2].GetOriginalTaskList())\n\ts.Equal(tasklist, txTasks[3].GetOriginalTaskList())\n\ts.Equal(tasklist, txTasks[4].GetOriginalTaskList())\n\ts.Equal(tasklist, txTasks[5].GetOriginalTaskList())\n\ts.Equal(types.TaskListKindNormal, txTasks[0].GetOriginalTaskListKind())\n\ts.Equal(types.TaskListKindEphemeral, txTasks[1].GetOriginalTaskListKind())\n\ts.Equal(types.TaskListKindNormal, txTasks[2].GetOriginalTaskListKind())\n\ts.Equal(types.TaskListKindNormal, txTasks[3].GetOriginalTaskListKind())\n\ts.Equal(types.TaskListKindNormal, txTasks[4].GetOriginalTaskListKind())\n\ts.Equal(types.TaskListKindNormal, txTasks[5].GetOriginalTaskListKind())\n\n\terr2 = s.RangeCompleteTransferTask(ctx, txTasks[0].GetTaskID(), txTasks[5].GetTaskID()+1)\n\ts.NoError(err2)\n\n\ttxTasks, err2 = s.GetTransferTasks(ctx, 100, false)\n\ts.NoError(err2)\n\ts.Empty(txTasks, \"expected empty task list.\")\n}\n\n// TestTimerTasksComplete test\nfunc (s *ExecutionManagerSuite) TestTimerTasksComplete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"8bfb47be-5b57-4d66-9109-5fb35e20b1d7\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-timer-tasks-test-complete\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\tnow := time.Now()\n\tinitialTasks := []p.Task{&p.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tVisibilityTimestamp: now.Add(1 * time.Second),\n\t\t\tTaskID:              1,\n\t\t\tVersion:             11,\n\t\t},\n\t\tEventID:         2,\n\t\tScheduleAttempt: 3,\n\t\tTimeoutType:     int(types.TimeoutTypeStartToClose),\n\t\tTaskList:        \"taskList\",\n\t},\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, initialTasks, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\ttasks := []p.Task{\n\t\t&p.WorkflowTimeoutTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVersion:             12,\n\t\t\t\tTaskID:              2,\n\t\t\t\tVisibilityTimestamp: now.Add(2 * time.Second),\n\t\t\t},\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t\t&p.DeleteHistoryEventTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVersion:             13,\n\t\t\t\tTaskID:              3,\n\t\t\t\tVisibilityTimestamp: now.Add(2 * time.Second),\n\t\t\t},\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t\t&p.ActivityTimeoutTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVersion:             14,\n\t\t\t\tTaskID:              4,\n\t\t\t\tVisibilityTimestamp: now.Add(3 * time.Second),\n\t\t\t},\n\t\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\t\tEventID:     7,\n\t\t\tAttempt:     0,\n\t\t\tTaskList:    \"taskList\",\n\t\t},\n\t\t&p.UserTimerTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVersion:             15,\n\t\t\t\tTaskID:              5,\n\t\t\t\tVisibilityTimestamp: now.Add(3 * time.Second),\n\t\t\t},\n\t\t\tEventID:  7,\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.NextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), tasks, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\ttimerTasks, err1 := s.GetTimerIndexTasks(ctx, 1, true) // use page size one to force pagination\n\ts.NoError(err1)\n\ts.NotNil(timerTasks, \"expected valid list of tasks.\")\n\ts.Equal(len(tasks)+len(initialTasks), len(timerTasks))\n\ts.Equal(p.TaskTypeDecisionTimeout, timerTasks[0].GetTaskType())\n\ts.Equal(p.TaskTypeWorkflowTimeout, timerTasks[1].GetTaskType())\n\ts.Equal(p.TaskTypeDeleteHistoryEvent, timerTasks[2].GetTaskType())\n\ts.Equal(p.TaskTypeActivityTimeout, timerTasks[3].GetTaskType())\n\ts.Equal(p.TaskTypeUserTimer, timerTasks[4].GetTaskType())\n\ts.Equal(int64(11), timerTasks[0].GetVersion())\n\ts.Equal(int64(12), timerTasks[1].GetVersion())\n\ts.Equal(int64(13), timerTasks[2].GetVersion())\n\ts.Equal(int64(14), timerTasks[3].GetVersion())\n\ts.Equal(int64(15), timerTasks[4].GetVersion())\n\ts.Equal(\"taskList\", timerTasks[0].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[1].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[2].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[3].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[4].GetTaskList())\n\n\terr2 = s.RangeCompleteTimerTask(ctx, timerTasks[0].GetVisibilityTimestamp(), timerTasks[4].GetVisibilityTimestamp().Add(1*time.Second))\n\ts.NoError(err2)\n\n\ttimerTasks2, err2 := s.GetTimerIndexTasks(ctx, 100, false)\n\ts.NoError(err2)\n\ts.Empty(timerTasks2, \"expected empty task list.\")\n}\n\n// TestTimerTasksRangeComplete test\nfunc (s *ExecutionManagerSuite) TestTimerTasksRangeComplete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"8bfb47be-5b57-4d66-9109-5fb35e20b1d7\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"get-timer-tasks-test-range-complete\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.NextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tnow := time.Now().UTC()\n\ttasks := []p.Task{\n\t\t&p.DecisionTimeoutTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              1,\n\t\t\t\tVersion:             11,\n\t\t\t},\n\t\t\tEventID:         2,\n\t\t\tScheduleAttempt: 3,\n\t\t\tTimeoutType:     int(types.TimeoutTypeStartToClose),\n\t\t\tTaskList:        \"taskList\",\n\t\t},\n\t\t&p.WorkflowTimeoutTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              2,\n\t\t\t\tVersion:             12,\n\t\t\t},\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t\t&p.DeleteHistoryEventTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              3,\n\t\t\t\tVersion:             13,\n\t\t\t},\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t\t&p.ActivityTimeoutTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              4,\n\t\t\t\tVersion:             14,\n\t\t\t},\n\t\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\t\tEventID:     7,\n\t\t\tAttempt:     0,\n\t\t\tTaskList:    \"taskList\",\n\t\t},\n\t\t&p.UserTimerTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\tTaskID:              5,\n\t\t\t\tVersion:             15,\n\t\t\t},\n\t\t\tEventID:  7,\n\t\t\tTaskList: \"taskList\",\n\t\t},\n\t}\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), tasks, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\ttimerTasks, err1 := s.GetTimerIndexTasks(ctx, 1, true) // use page size one to force pagination\n\ts.NoError(err1)\n\ts.NotNil(timerTasks, \"expected valid list of tasks.\")\n\ts.Equal(len(tasks), len(timerTasks))\n\ts.Equal(p.TaskTypeDecisionTimeout, timerTasks[0].GetTaskType())\n\ts.Equal(p.TaskTypeWorkflowTimeout, timerTasks[1].GetTaskType())\n\ts.Equal(p.TaskTypeDeleteHistoryEvent, timerTasks[2].GetTaskType())\n\ts.Equal(p.TaskTypeActivityTimeout, timerTasks[3].GetTaskType())\n\ts.Equal(p.TaskTypeUserTimer, timerTasks[4].GetTaskType())\n\ts.Equal(int64(11), timerTasks[0].GetVersion())\n\ts.Equal(int64(12), timerTasks[1].GetVersion())\n\ts.Equal(int64(13), timerTasks[2].GetVersion())\n\ts.Equal(int64(14), timerTasks[3].GetVersion())\n\ts.Equal(int64(15), timerTasks[4].GetVersion())\n\ts.Equal(\"taskList\", timerTasks[0].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[1].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[2].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[3].GetTaskList())\n\ts.Equal(\"taskList\", timerTasks[4].GetTaskList())\n\n\terr2 = s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, nil, nil, int64(5), nil, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\terr2 = s.CompleteTimerTask(ctx, timerTasks[0].GetVisibilityTimestamp(), timerTasks[0].GetTaskID())\n\ts.NoError(err2)\n\n\terr2 = s.CompleteTimerTask(ctx, timerTasks[1].GetVisibilityTimestamp(), timerTasks[1].GetTaskID())\n\ts.NoError(err2)\n\n\terr2 = s.CompleteTimerTask(ctx, timerTasks[2].GetVisibilityTimestamp(), timerTasks[2].GetTaskID())\n\ts.NoError(err2)\n\n\terr2 = s.CompleteTimerTask(ctx, timerTasks[3].GetVisibilityTimestamp(), timerTasks[3].GetTaskID())\n\ts.NoError(err2)\n\n\terr2 = s.CompleteTimerTask(ctx, timerTasks[4].GetVisibilityTimestamp(), timerTasks[4].GetTaskID())\n\ts.NoError(err2)\n\n\ttimerTasks2, err2 := s.GetTimerIndexTasks(ctx, 100, false)\n\ts.NoError(err2)\n\ts.Empty(timerTasks2, \"expected empty task list.\")\n}\n\n// TestWorkflowMutableStateActivities test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateActivities() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"7fcf0aa9-e121-4292-bdad-0a75181b4aa3\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-test\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tactivityInfos := []*p.ActivityInfo{{\n\t\tVersion:                  7789,\n\t\tScheduleID:               1,\n\t\tScheduledEventBatchID:    1,\n\t\tScheduledEvent:           &types.HistoryEvent{ID: 1},\n\t\tScheduledTime:            time.UnixMilli(1755030646000).UTC(),\n\t\tActivityID:               uuid.New(),\n\t\tRequestID:                uuid.New(),\n\t\tDetails:                  []byte(uuid.New()),\n\t\tStartedID:                2,\n\t\tStartedEvent:             &types.HistoryEvent{ID: 2},\n\t\tStartedTime:              time.UnixMilli(1755030646001).UTC(),\n\t\tScheduleToCloseTimeout:   1,\n\t\tScheduleToStartTimeout:   2,\n\t\tStartToCloseTimeout:      3,\n\t\tHeartbeatTimeout:         4,\n\t\tLastHeartBeatUpdatedTime: time.UnixMilli(1755030646002).UTC(),\n\t\tTimerTaskStatus:          1,\n\t\tCancelRequested:          true,\n\t\tCancelRequestID:          math.MaxInt64,\n\t\tAttempt:                  math.MaxInt32,\n\t\tDomainID:                 domainID,\n\t\tStartedIdentity:          uuid.New(),\n\t\tTaskList:                 uuid.New(),\n\t\tTaskListKind:             types.TaskListKindEphemeral,\n\t\tHasRetryPolicy:           true,\n\t\tInitialInterval:          math.MaxInt32,\n\t\tMaximumInterval:          math.MaxInt32,\n\t\tMaximumAttempts:          math.MaxInt32,\n\t\tBackoffCoefficient:       5.55,\n\t\tExpirationTime:           time.UnixMilli(1755030646003).UTC(),\n\t\tNonRetriableErrors:       []string{\"accessDenied\", \"badRequest\"},\n\t\tLastFailureReason:        \"some random error\",\n\t\tLastWorkerIdentity:       uuid.New(),\n\t\tLastFailureDetails:       []byte(uuid.New()),\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.NextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), nil, activityInfos, nil, nil, nil)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.ActivityInfos))\n\ts.T().Logf(\"%+v\\n\", state.ActivityInfos)\n\tai, ok := state.ActivityInfos[1]\n\ts.True(ok)\n\ts.NotNil(ai)\n\t// Various persistence layers are inconsistent with assigning the location value, so normalize them first\n\tai.ScheduledTime = ai.ScheduledTime.UTC()\n\tai.StartedTime = ai.StartedTime.UTC()\n\tai.LastHeartBeatUpdatedTime = ai.LastHeartBeatUpdatedTime.UTC()\n\tai.ExpirationTime = ai.ExpirationTime.UTC()\n\ts.Equal(activityInfos[0], ai)\n\n\terr2 = s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, nil, nil, int64(5), nil, nil, []int64{1}, nil, nil)\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.ActivityInfos))\n}\n\n// TestWorkflowMutableStateTimers test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateTimers() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"025d178a-709b-4c07-8dd7-86dbf9bd2e06\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-timers-test\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tcurrentTime := time.Now().UTC()\n\ttimerID := \"id_1\"\n\ttimerInfos := []*p.TimerInfo{{\n\t\tVersion:    3345,\n\t\tTimerID:    timerID,\n\t\tExpiryTime: currentTime,\n\t\tTaskStatus: 2,\n\t\tStartedID:  5,\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.NextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), nil, nil, nil, timerInfos, nil)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.TimerInfos))\n\ts.Equal(int64(3345), state.TimerInfos[timerID].Version)\n\ts.Equal(timerID, state.TimerInfos[timerID].TimerID)\n\ts.EqualTimesWithPrecision(currentTime, state.TimerInfos[timerID].ExpiryTime, time.Millisecond*500)\n\ts.Equal(int64(2), state.TimerInfos[timerID].TaskStatus)\n\ts.Equal(int64(5), state.TimerInfos[timerID].StartedID)\n\n\terr2 = s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, nil, nil, int64(5), nil, nil, nil, nil, []string{timerID})\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.TimerInfos))\n}\n\n// TestWorkflowMutableStateChildExecutions test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateChildExecutions() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"88236cd2-c439-4cec-9957-2748ce3be074\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-child-executions-parent-test\",\n\t\tRunID:      \"c63dba1e-929c-4fbf-8ec5-4533b16269a9\",\n\t}\n\n\tparentDomainID := \"6036ded3-e541-42c9-8f69-3d9354dad081\"\n\tparentExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-child-executions-child-test\",\n\t\tRunID:      \"73e89362-25ec-4305-bcb8-d9448b90856c\",\n\t}\n\n\tpartitionConfig := map[string]string{\n\t\t\"userID\": uuid.New(),\n\t}\n\ttask0, err0 := s.CreateChildWorkflowExecution(ctx, domainID, workflowExecution, parentDomainID, parentExecution, 1, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, partitionConfig)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(parentDomainID, info0.ParentDomainID)\n\ts.Equal(parentExecution.GetWorkflowID(), info0.ParentWorkflowID)\n\ts.Equal(parentExecution.GetRunID(), info0.ParentRunID)\n\ts.Equal(int64(1), info0.InitiatedID)\n\ts.Equal(partitionConfig, info0.PartitionConfig)\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tchildExecutionInfos := []*p.ChildExecutionInfo{{\n\t\tVersion:           1234,\n\t\tInitiatedID:       1,\n\t\tInitiatedEvent:    &types.HistoryEvent{ID: 1},\n\t\tStartedID:         2,\n\t\tStartedRunID:      uuid.New(),\n\t\tStartedEvent:      &types.HistoryEvent{ID: 2},\n\t\tCreateRequestID:   uuid.New(),\n\t\tDomainID:          uuid.New(),\n\t\tParentClosePolicy: types.ParentClosePolicyTerminate,\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpsertChildExecutionsState(ctx, updatedInfo, updatedStats, versionHistories, int64(3), childExecutionInfos)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.ChildExecutionInfos))\n\tci, ok := state.ChildExecutionInfos[1]\n\ts.True(ok)\n\ts.NotNil(ci)\n\ts.Equal(childExecutionInfos[0], ci)\n\n\terr2 = s.DeleteChildExecutionsState(ctx, updatedInfo, updatedStats, versionHistories, int64(5), int64(1))\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.ChildExecutionInfos))\n}\n\n// TestWorkflowMutableStateRequestCancel test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateRequestCancel() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"568b8d19-cf64-4aac-be1b-f8a3edbc1fa9\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-request-cancel-test\",\n\t\tRunID:      \"87f96253-b925-426e-90db-aa4ee89b5aca\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\trequestCancelInfo := &p.RequestCancelInfo{\n\t\tVersion:               456,\n\t\tInitiatedID:           2,\n\t\tInitiatedEventBatchID: 1,\n\t\tCancelRequestID:       uuid.New(),\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpsertRequestCancelState(ctx, updatedInfo, updatedStats, versionHistories, int64(3), []*p.RequestCancelInfo{requestCancelInfo})\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.RequestCancelInfos))\n\tri, ok := state.RequestCancelInfos[requestCancelInfo.InitiatedID]\n\ts.True(ok)\n\ts.NotNil(ri)\n\ts.Equal(requestCancelInfo, ri)\n\n\terr2 = s.DeleteCancelState(ctx, updatedInfo, updatedStats, versionHistories, int64(5), requestCancelInfo.InitiatedID)\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.RequestCancelInfos))\n}\n\n// TestWorkflowMutableStateSignalInfo test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateSignalInfo() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\trunID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-signal-info-test\",\n\t\tRunID:      runID,\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tsignalInfo := &p.SignalInfo{\n\t\tVersion:               123,\n\t\tInitiatedID:           2,\n\t\tInitiatedEventBatchID: 1,\n\t\tSignalRequestID:       uuid.New(),\n\t\tSignalName:            \"my signal\",\n\t\tInput:                 []byte(\"test signal input\"),\n\t\tControl:               []byte(uuid.New()),\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpsertSignalInfoState(ctx, updatedInfo, updatedStats, versionHistories, int64(3), []*p.SignalInfo{signalInfo})\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.SignalInfos))\n\tsi, ok := state.SignalInfos[signalInfo.InitiatedID]\n\ts.True(ok)\n\ts.NotNil(si)\n\ts.Equal(signalInfo, si)\n\n\terr2 = s.DeleteSignalState(ctx, updatedInfo, updatedStats, versionHistories, int64(5), signalInfo.InitiatedID)\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.SignalInfos))\n}\n\n// TestWorkflowMutableStateSignalRequested test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateSignalRequested() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\trunID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-signal-requested-test\",\n\t\tRunID:      runID,\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tsignalRequestedID := uuid.New()\n\tsignalsRequested := []string{signalRequestedID}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpsertSignalsRequestedState(ctx, updatedInfo, updatedStats, versionHistories, int64(3), signalsRequested)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.SignalRequestedIDs))\n\tri, ok := state.SignalRequestedIDs[signalRequestedID]\n\ts.True(ok)\n\ts.NotNil(ri)\n\n\terr2 = s.DeleteSignalsRequestedState(ctx, updatedInfo, updatedStats, versionHistories, int64(5), []string{signalRequestedID, uuid.New()})\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.SignalRequestedIDs))\n}\n\n// TestWorkflowMutableStateInfo test\nfunc (s *ExecutionManagerSuite) TestWorkflowMutableStateInfo() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"9ed8818b-3090-4160-9f21-c6b70e64d2dd\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-mutable-state-test\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), nil, nil, nil, nil, nil)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.NotNil(state.ExecutionInfo, \"expected valid MS Info state.\")\n\ts.Equal(updatedInfo.NextEventID, state.ExecutionInfo.NextEventID)\n\ts.Equal(updatedInfo.State, state.ExecutionInfo.State)\n}\n\n// TestContinueAsNew test\nfunc (s *ExecutionManagerSuite) TestContinueAsNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"c1c0bb55-04e6-4a9c-89d0-1be7b96459f8\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"continue-as-new-workflow-test\",\n\t\tRunID:      \"551c88d2-d9e6-404f-8131-9eec14f36643\",\n\t}\n\n\tpartitionConfig := map[string]string{\n\t\t\"userID\": uuid.New(),\n\t}\n\t_, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, partitionConfig)\n\ts.NoError(err0)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info0)\n\tcontinueAsNewStats := copyExecutionStats(state0.ExecutionStats)\n\tcontinueAsNewInfo.State = p.WorkflowStateCreated\n\tcontinueAsNewInfo.CloseStatus = p.WorkflowCloseStatusNone\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tnewWorkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"continue-as-new-workflow-test\",\n\t\tRunID:      \"64c7e15a-3fd7-4182-9c6f-6f25a4fa2614\",\n\t}\n\n\ttestResetPoints := types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\t{\n\t\t\t\tBinaryChecksum:           \"test-binary-checksum\",\n\t\t\t\tRunID:                    \"test-runID\",\n\t\t\t\tFirstDecisionCompletedID: 123,\n\t\t\t\tCreatedTimeNano:          common.Int64Ptr(456),\n\t\t\t\tResettable:               true,\n\t\t\t\tExpiringTimeNano:         common.Int64Ptr(789),\n\t\t\t},\n\t\t},\n\t}\n\n\terr2 := s.ContinueAsNewExecution(ctx, continueAsNewInfo, continueAsNewStats, info0.NextEventID, newWorkflowExecution, int64(3), int64(2), &testResetPoints)\n\ts.NoError(err2)\n\n\tprevExecutionState, err3 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err3)\n\tprevExecutionInfo := prevExecutionState.ExecutionInfo\n\ts.Equal(\"551c88d2-d9e6-404f-8131-9eec14f36643\", prevExecutionInfo.FirstExecutionRunID)\n\ts.Equal(p.WorkflowStateCompleted, prevExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusContinuedAsNew, prevExecutionInfo.CloseStatus)\n\ts.Equal(int64(5), prevExecutionInfo.NextEventID)\n\ts.Equal(int64(2), prevExecutionInfo.LastProcessedEvent)\n\ts.Equal(prevExecutionInfo.AutoResetPoints, &types.ResetPoints{})\n\ts.Equal(partitionConfig, prevExecutionInfo.PartitionConfig)\n\n\tnewExecutionState, err4 := s.GetWorkflowExecutionInfo(ctx, domainID, newWorkflowExecution)\n\ts.NoError(err4)\n\tnewExecutionInfo := newExecutionState.ExecutionInfo\n\ts.Equal(\"551c88d2-d9e6-404f-8131-9eec14f36643\", newExecutionInfo.FirstExecutionRunID)\n\ts.Equal(p.WorkflowStateCreated, newExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, newExecutionInfo.CloseStatus)\n\ts.Equal(int64(3), newExecutionInfo.NextEventID)\n\ts.Equal(constants.EmptyEventID, newExecutionInfo.LastProcessedEvent)\n\ts.Equal(int64(2), newExecutionInfo.DecisionScheduleID)\n\ts.Equal(testResetPoints, *newExecutionInfo.AutoResetPoints)\n\ts.Equal(partitionConfig, newExecutionInfo.PartitionConfig)\n\n\tnewRunID, err5 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.GetWorkflowID())\n\ts.NoError(err5)\n\ts.Equal(newWorkflowExecution.GetRunID(), newRunID)\n}\n\n// TestReplicationTransferTaskTasks test\nfunc (s *ExecutionManagerSuite) TestReplicationTransferTaskTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"2466d7de-6602-4ad8-b939-fb8f8c36c711\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"replication-transfer-task-test\",\n\t\tRunID:      \"dcde9d85-5d7a-43c7-8b18-cb2cae0e29e0\",\n\t}\n\n\ttask0, err := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\treplicationTasks := []p.Task{&p.HistoryReplicationTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\tVersion: 9,\n\t\t},\n\t\tFirstEventID: 1,\n\t\tNextEventID:  3,\n\t}}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 3,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr = s.UpdateWorklowStateAndReplication(ctx, updatedInfo1, updatedStats1, versionHistories, int64(3), replicationTasks)\n\ts.NoError(err)\n\n\ttasks1, err := s.GetReplicationTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1), \"Expected 1 replication task.\")\n\ttask1, ok := tasks1[0].(*p.HistoryReplicationTask)\n\ts.True(ok, \"expected a history replication task\")\n\ts.Equal(domainID, task1.DomainID)\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.WorkflowID)\n\ts.Equal(workflowExecution.GetRunID(), task1.RunID)\n\ts.Equal(int64(1), task1.FirstEventID)\n\ts.Equal(int64(3), task1.NextEventID)\n\ts.Equal(int64(9), task1.Version)\n\n\terr = s.CompleteReplicationTask(ctx, task1.TaskID)\n\ts.NoError(err)\n\ttasks2, err := s.GetReplicationTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(0, len(tasks2))\n}\n\n// TestReplicationTransferTaskRangeComplete test\nfunc (s *ExecutionManagerSuite) TestReplicationTransferTaskRangeComplete() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"replication-transfer-task--range-complete-test\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\ttask0, err := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttaskD, err := s.GetTransferTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(1, len(taskD), \"Expected 1 decision task.\")\n\terr = s.CompleteTransferTask(ctx, taskD[0].GetTaskID())\n\ts.NoError(err)\n\n\tstate1, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err)\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\n\treplicationTasks := []p.Task{\n\t\t&p.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\t\tVersion: 9,\n\t\t\t},\n\t\t\tFirstEventID: 1,\n\t\t\tNextEventID:  3,\n\t\t},\n\t\t&p.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:  s.GetNextSequenceNumber(),\n\t\t\t\tVersion: 9,\n\t\t\t},\n\t\t\tFirstEventID: 4,\n\t\t\tNextEventID:  5,\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: 3,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr = s.UpdateWorklowStateAndReplication(ctx,\n\t\tupdatedInfo1,\n\t\tupdatedStats1,\n\t\tversionHistories,\n\t\tint64(3),\n\t\treplicationTasks,\n\t)\n\ts.NoError(err)\n\n\ttasks1, err := s.GetReplicationTasks(ctx, 2, false)\n\ts.NoError(err)\n\ts.NotNil(tasks1, \"expected valid list of tasks.\")\n\ts.Equal(2, len(tasks1), \"Expected 2 replication tasks.\")\n\ttask1, ok := tasks1[0].(*p.HistoryReplicationTask)\n\ts.True(ok, \"expected a history replication task\")\n\ts.Equal(domainID, task1.DomainID)\n\ts.Equal(workflowExecution.GetWorkflowID(), task1.WorkflowID)\n\ts.Equal(workflowExecution.GetRunID(), task1.RunID)\n\ts.Equal(int64(1), task1.FirstEventID)\n\ts.Equal(int64(3), task1.NextEventID)\n\ts.Equal(int64(9), task1.Version)\n\n\ttask2, ok := tasks1[1].(*p.HistoryReplicationTask)\n\ts.True(ok, \"expected a history replication task\")\n\ts.Equal(domainID, task2.DomainID)\n\ts.Equal(workflowExecution.GetWorkflowID(), task2.WorkflowID)\n\ts.Equal(workflowExecution.GetRunID(), task2.RunID)\n\ts.Equal(int64(4), task2.FirstEventID)\n\ts.Equal(int64(5), task2.NextEventID)\n\ts.Equal(int64(9), task2.Version)\n\terr = s.RangeCompleteReplicationTask(ctx, task2.TaskID+1)\n\ts.NoError(err)\n\ttasks2, err := s.GetReplicationTasks(ctx, 1, false)\n\ts.NoError(err)\n\ts.Equal(0, len(tasks2))\n}\n\n// TestUpdateAndClearBufferedEvents test\nfunc (s *ExecutionManagerSuite) TestUpdateAndClearBufferedEvents() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-update-and-clear-buffered-events\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\n\ttask0, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"taskList\", \"wType\", 20, 13, nil, 3, 0, 2, nil, nil)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstats0, state0, err1 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(0, stats0.BufferedEventsCount)\n\ts.Equal(0, stats0.BufferedEventsSize)\n\n\teventsBatch1 := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        5,\n\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\tVersion:   11,\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 2,\n\t\t\t\tStartedEventID:   3,\n\t\t\t\tIdentity:         \"test_worker\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        6,\n\t\t\tEventType: types.EventTypeTimerStarted.Ptr(),\n\t\t\tVersion:   11,\n\t\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{\n\t\t\t\tTimerID:                      \"ID1\",\n\t\t\t\tStartToFireTimeoutSeconds:    common.Int64Ptr(101),\n\t\t\t\tDecisionTaskCompletedEventID: 5,\n\t\t\t},\n\t\t},\n\t}\n\n\teventsBatch2 := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        21,\n\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\tVersion:   12,\n\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\tTimerID:        \"2\",\n\t\t\t\tStartedEventID: 3,\n\t\t\t},\n\t\t},\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: eventsBatch2[0].ID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedState := &p.WorkflowMutableState{\n\t\tExecutionInfo:    updatedInfo,\n\t\tExecutionStats:   updatedStats,\n\t\tVersionHistories: versionHistories,\n\t}\n\n\terr2 := s.UpdateAllMutableState(ctx, updatedState, int64(3))\n\ts.NoError(err2)\n\n\tpartialState, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(partialState, \"expected valid state.\")\n\tpartialInfo := partialState.ExecutionInfo\n\ts.NotNil(partialInfo, \"Valid Workflow info expected.\")\n\n\tbufferUpdateInfo := copyWorkflowExecutionInfo(partialInfo)\n\tbufferedUpdatedStats := copyExecutionStats(state0.ExecutionStats)\n\terr2 = s.UpdateWorklowStateAndReplication(ctx, bufferUpdateInfo, bufferedUpdatedStats, versionHistories, bufferUpdateInfo.NextEventID, nil)\n\ts.NoError(err2)\n\terr2 = s.UpdateWorklowStateAndReplication(ctx, bufferUpdateInfo, bufferedUpdatedStats, versionHistories, bufferUpdateInfo.NextEventID, nil)\n\ts.NoError(err2)\n\terr2 = s.UpdateWorkflowExecutionForBufferEvents(ctx, bufferUpdateInfo, bufferedUpdatedStats, bufferUpdateInfo.NextEventID, eventsBatch1, false, versionHistories)\n\ts.NoError(err2)\n\tstats0, state0, err2 = s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.Equal(1, stats0.BufferedEventsCount)\n\ts.True(stats0.BufferedEventsSize > 0)\n\ttestHistory := &types.History{Events: make([]*types.HistoryEvent, 0)}\n\ttestHistory.Events = append(testHistory.Events, eventsBatch1...)\n\thistory0 := &types.History{Events: state0.BufferedEvents}\n\ts.Equal(testHistory, history0)\n\ttestHistory.Events = append(testHistory.Events, eventsBatch2...)\n\n\terr2 = s.UpdateWorkflowExecutionForBufferEvents(ctx, bufferUpdateInfo, bufferedUpdatedStats, bufferUpdateInfo.NextEventID, eventsBatch2, false, versionHistories)\n\ts.NoError(err2)\n\n\tstats1, state1, err1 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state1, \"expected valid state.\")\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\ts.Equal(2, stats1.BufferedEventsCount)\n\ts.True(stats1.BufferedEventsSize > 0)\n\thistory1 := &types.History{Events: state1.BufferedEvents}\n\ts.Equal(testHistory, history1)\n\n\terr3 := s.UpdateWorkflowExecutionForBufferEvents(ctx, bufferUpdateInfo, bufferedUpdatedStats, bufferUpdateInfo.NextEventID, nil, true, versionHistories)\n\ts.NoError(err3)\n\n\tstats3, state3, err3 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err3)\n\ts.NotNil(state3, \"expected valid state.\")\n\tinfo3 := state3.ExecutionInfo\n\ts.NotNil(info3, \"Valid Workflow info expected.\")\n\ts.Equal(0, stats3.BufferedEventsCount)\n\ts.Equal(0, stats3.BufferedEventsSize)\n}\n\n// TestConflictResolveWorkflowExecutionCurrentIsSelf test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionCurrentIsSelf() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-reset-mutable-state-test-current-is-self\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\ttask0, err0 := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecution,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\t3,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\tstats0, state0, err1 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(0, stats0.BufferedEventsCount)\n\ts.Equal(0, stats0.BufferedEventsSize)\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tcurrentTime := time.Now().UTC()\n\texpiryTime := currentTime.Add(10 * time.Second)\n\teventsBatch1 := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        5,\n\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\tVersion:   11,\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 2,\n\t\t\t\tStartedEventID:   3,\n\t\t\t\tIdentity:         \"test_worker\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        6,\n\t\t\tEventType: types.EventTypeTimerStarted.Ptr(),\n\t\t\tVersion:   11,\n\t\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{\n\t\t\t\tTimerID:                      \"ID1\",\n\t\t\t\tStartToFireTimeoutSeconds:    common.Int64Ptr(101),\n\t\t\t\tDecisionTaskCompletedEventID: 5,\n\t\t\t},\n\t\t},\n\t}\n\n\teventsBatch2 := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        21,\n\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\tVersion:   12,\n\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\tTimerID:        \"2\",\n\t\t\t\tStartedEventID: 3,\n\t\t\t},\n\t\t},\n\t}\n\n\tcsum := s.newRandomChecksum()\n\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: updatedInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tupdatedState := &p.WorkflowMutableState{\n\t\tExecutionInfo:  updatedInfo,\n\t\tExecutionStats: updatedStats,\n\t\tActivityInfos: map[int64]*p.ActivityInfo{\n\t\t\t4: {\n\t\t\t\tVersion:                  7789,\n\t\t\t\tScheduleID:               4,\n\t\t\t\tScheduledEventBatchID:    3,\n\t\t\t\tScheduledEvent:           &types.HistoryEvent{ID: 40},\n\t\t\t\tScheduledTime:            currentTime,\n\t\t\t\tStartedID:                6,\n\t\t\t\tStartedEvent:             &types.HistoryEvent{ID: 60},\n\t\t\t\tStartedTime:              currentTime,\n\t\t\t\tScheduleToCloseTimeout:   1,\n\t\t\t\tScheduleToStartTimeout:   2,\n\t\t\t\tStartToCloseTimeout:      3,\n\t\t\t\tHeartbeatTimeout:         4,\n\t\t\t\tLastHeartBeatUpdatedTime: currentTime,\n\t\t\t\tTimerTaskStatus:          1,\n\t\t\t},\n\t\t\t5: {\n\t\t\t\tVersion:                  7789,\n\t\t\t\tScheduleID:               5,\n\t\t\t\tScheduledEventBatchID:    3,\n\t\t\t\tScheduledEvent:           &types.HistoryEvent{ID: 50},\n\t\t\t\tScheduledTime:            currentTime,\n\t\t\t\tStartedID:                7,\n\t\t\t\tStartedEvent:             &types.HistoryEvent{ID: 70},\n\t\t\t\tStartedTime:              currentTime,\n\t\t\t\tScheduleToCloseTimeout:   1,\n\t\t\t\tScheduleToStartTimeout:   2,\n\t\t\t\tStartToCloseTimeout:      3,\n\t\t\t\tHeartbeatTimeout:         4,\n\t\t\t\tLastHeartBeatUpdatedTime: currentTime,\n\t\t\t\tTimerTaskStatus:          1,\n\t\t\t}},\n\n\t\tTimerInfos: map[string]*p.TimerInfo{\n\t\t\t\"t1\": {\n\t\t\t\tVersion:    2333,\n\t\t\t\tTimerID:    \"t1\",\n\t\t\t\tStartedID:  1,\n\t\t\t\tExpiryTime: expiryTime,\n\t\t\t\tTaskStatus: 500,\n\t\t\t},\n\t\t\t\"t2\": {\n\t\t\t\tVersion:    2333,\n\t\t\t\tTimerID:    \"t2\",\n\t\t\t\tStartedID:  2,\n\t\t\t\tExpiryTime: expiryTime,\n\t\t\t\tTaskStatus: 501,\n\t\t\t},\n\t\t\t\"t3\": {\n\t\t\t\tVersion:    2333,\n\t\t\t\tTimerID:    \"t3\",\n\t\t\t\tStartedID:  3,\n\t\t\t\tExpiryTime: expiryTime,\n\t\t\t\tTaskStatus: 502,\n\t\t\t},\n\t\t},\n\n\t\tChildExecutionInfos: map[int64]*p.ChildExecutionInfo{\n\t\t\t9: {\n\t\t\t\tVersion:         2334,\n\t\t\t\tInitiatedID:     9,\n\t\t\t\tInitiatedEvent:  &types.HistoryEvent{ID: 123},\n\t\t\t\tStartedID:       11,\n\t\t\t\tStartedRunID:    uuid.New(),\n\t\t\t\tStartedEvent:    nil,\n\t\t\t\tCreateRequestID: \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\tDomainID:        uuid.New(),\n\t\t\t},\n\t\t},\n\n\t\tRequestCancelInfos: map[int64]*p.RequestCancelInfo{\n\t\t\t19: {\n\t\t\t\tVersion:               2335,\n\t\t\t\tInitiatedID:           19,\n\t\t\t\tInitiatedEventBatchID: 17,\n\t\t\t\tCancelRequestID:       \"cancel_requested_id\",\n\t\t\t},\n\t\t},\n\n\t\tSignalInfos: map[int64]*p.SignalInfo{\n\t\t\t39: {\n\t\t\t\tVersion:               2336,\n\t\t\t\tInitiatedID:           39,\n\t\t\t\tInitiatedEventBatchID: 38,\n\t\t\t\tSignalRequestID:       \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\tSignalName:            \"signalA\",\n\t\t\t\tInput:                 []byte(\"signal_input_A\"),\n\t\t\t\tControl:               []byte(\"signal_control_A\"),\n\t\t\t},\n\t\t},\n\n\t\tSignalRequestedIDs: map[string]struct{}{\n\t\t\t\"00000000-0000-0000-0000-000000000001\": {},\n\t\t\t\"00000000-0000-0000-0000-000000000002\": {},\n\t\t\t\"00000000-0000-0000-0000-000000000003\": {},\n\t\t},\n\t\tChecksum:         csum,\n\t\tVersionHistories: versionHistories,\n\t}\n\n\terr2 := s.UpdateAllMutableState(ctx, updatedState, int64(3))\n\ts.NoError(err2)\n\n\tpartialState, err2 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(partialState, \"expected valid state.\")\n\tpartialInfo := partialState.ExecutionInfo\n\ts.NotNil(partialInfo, \"Valid Workflow info expected.\")\n\ts.assertChecksumsEqual(csum, partialState.Checksum)\n\n\tbufferUpdateInfo := copyWorkflowExecutionInfo(partialInfo)\n\tbufferedUpdatedStats := copyExecutionStats(partialState.ExecutionStats)\n\terr2 = s.UpdateWorklowStateAndReplication(ctx, bufferUpdateInfo, bufferedUpdatedStats, versionHistories, bufferUpdateInfo.NextEventID, nil)\n\ts.NoError(err2)\n\terr2 = s.UpdateWorklowStateAndReplication(ctx, bufferUpdateInfo, bufferedUpdatedStats, versionHistories, bufferUpdateInfo.NextEventID, nil)\n\ts.NoError(err2)\n\terr2 = s.UpdateWorkflowExecutionForBufferEvents(ctx, bufferUpdateInfo, bufferedUpdatedStats, bufferUpdateInfo.NextEventID, eventsBatch1, false, versionHistories)\n\ts.NoError(err2)\n\tstats0, state0, err2 = s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.Equal(1, stats0.BufferedEventsCount)\n\ts.True(stats0.BufferedEventsSize > 0)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state0.Checksum)\n\ttestHistory := &types.History{Events: make([]*types.HistoryEvent, 0)}\n\ttestHistory.Events = append(testHistory.Events, eventsBatch1...)\n\thistory0 := &types.History{Events: state0.BufferedEvents}\n\ts.Equal(testHistory, history0)\n\ttestHistory.Events = append(testHistory.Events, eventsBatch2...)\n\n\terr2 = s.UpdateWorkflowExecutionForBufferEvents(ctx, bufferUpdateInfo, bufferedUpdatedStats, bufferUpdateInfo.NextEventID, eventsBatch2, false, versionHistories)\n\ts.NoError(err2)\n\n\tstats1, state1, err1 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state1, \"expected valid state.\")\n\tinfo1 := state1.ExecutionInfo\n\ts.NotNil(info1, \"Valid Workflow info expected.\")\n\ts.Equal(2, stats1.BufferedEventsCount)\n\ts.True(stats1.BufferedEventsSize > 0)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state1.Checksum)\n\thistory1 := &types.History{Events: state1.BufferedEvents}\n\ts.Equal(testHistory, history1)\n\n\ts.Equal(2, len(state1.ActivityInfos))\n\tai, ok := state1.ActivityInfos[4]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(int64(7789), ai.Version)\n\ts.Equal(int64(4), ai.ScheduleID)\n\ts.Equal(int64(3), ai.ScheduledEventBatchID)\n\ts.Equal(int64(40), ai.ScheduledEvent.ID)\n\ts.EqualTimes(currentTime, ai.ScheduledTime)\n\ts.Equal(int64(6), ai.StartedID)\n\ts.Equal(int64(60), ai.StartedEvent.ID)\n\ts.EqualTimes(currentTime, ai.StartedTime)\n\ts.Equal(int32(1), ai.ScheduleToCloseTimeout)\n\ts.Equal(int32(2), ai.ScheduleToStartTimeout)\n\ts.Equal(int32(3), ai.StartToCloseTimeout)\n\ts.Equal(int32(4), ai.HeartbeatTimeout)\n\ts.EqualTimes(currentTime, ai.LastHeartBeatUpdatedTime)\n\ts.Equal(int32(1), ai.TimerTaskStatus)\n\n\tai, ok = state1.ActivityInfos[5]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(int64(7789), ai.Version)\n\ts.Equal(int64(5), ai.ScheduleID)\n\ts.Equal(int64(3), ai.ScheduledEventBatchID)\n\ts.Equal(int64(50), ai.ScheduledEvent.ID)\n\ts.EqualTimes(currentTime, ai.ScheduledTime)\n\ts.Equal(int64(7), ai.StartedID)\n\ts.Equal(int64(70), ai.StartedEvent.ID)\n\ts.EqualTimes(currentTime, ai.StartedTime)\n\ts.Equal(int32(1), ai.ScheduleToCloseTimeout)\n\ts.Equal(int32(2), ai.ScheduleToStartTimeout)\n\ts.Equal(int32(3), ai.StartToCloseTimeout)\n\ts.Equal(int32(4), ai.HeartbeatTimeout)\n\ts.EqualTimes(currentTime, ai.LastHeartBeatUpdatedTime)\n\ts.Equal(int32(1), ai.TimerTaskStatus)\n\n\ts.Equal(3, len(state1.TimerInfos))\n\tti, ok := state1.TimerInfos[\"t1\"]\n\ts.True(ok)\n\ts.NotNil(ti)\n\ts.Equal(int64(2333), ti.Version)\n\ts.Equal(\"t1\", ti.TimerID)\n\ts.Equal(int64(1), ti.StartedID)\n\ts.EqualTimes(expiryTime, ti.ExpiryTime)\n\ts.Equal(int64(500), ti.TaskStatus)\n\n\tti, ok = state1.TimerInfos[\"t2\"]\n\ts.True(ok)\n\ts.NotNil(ti)\n\ts.Equal(int64(2333), ti.Version)\n\ts.Equal(\"t2\", ti.TimerID)\n\ts.Equal(int64(2), ti.StartedID)\n\ts.EqualTimes(expiryTime, ti.ExpiryTime)\n\ts.Equal(int64(501), ti.TaskStatus)\n\n\tti, ok = state1.TimerInfos[\"t3\"]\n\ts.True(ok)\n\ts.NotNil(ti)\n\ts.Equal(int64(2333), ti.Version)\n\ts.Equal(\"t3\", ti.TimerID)\n\ts.Equal(int64(3), ti.StartedID)\n\ts.EqualTimes(expiryTime, ti.ExpiryTime)\n\ts.Equal(int64(502), ti.TaskStatus)\n\n\ts.Equal(1, len(state1.ChildExecutionInfos))\n\tci, ok := state1.ChildExecutionInfos[9]\n\ts.True(ok)\n\ts.NotNil(ci)\n\ts.Equal(updatedState.ChildExecutionInfos[9], ci)\n\n\ts.Equal(1, len(state1.RequestCancelInfos))\n\trci, ok := state1.RequestCancelInfos[19]\n\ts.True(ok)\n\ts.NotNil(rci)\n\ts.Equal(int64(2335), rci.Version)\n\n\ts.Equal(1, len(state1.SignalInfos))\n\tsi, ok := state1.SignalInfos[39]\n\ts.True(ok)\n\ts.NotNil(si)\n\ts.Equal(int64(2336), si.Version)\n\n\ts.Equal(3, len(state1.SignalRequestedIDs))\n\t_, contains := state1.SignalRequestedIDs[\"00000000-0000-0000-0000-000000000001\"]\n\ts.True(contains)\n\t_, contains = state1.SignalRequestedIDs[\"00000000-0000-0000-0000-000000000002\"]\n\ts.True(contains)\n\t_, contains = state1.SignalRequestedIDs[\"00000000-0000-0000-0000-000000000003\"]\n\ts.True(contains)\n\n\ts.Equal(3, len(state1.BufferedEvents))\n\n\tupdatedInfo1 := copyWorkflowExecutionInfo(info1)\n\tupdatedStats1 := copyExecutionStats(state1.ExecutionStats)\n\tupdatedInfo1.NextEventID = int64(3)\n\tresetActivityInfos := []*p.ActivityInfo{\n\t\t{\n\t\t\tVersion:                  8789,\n\t\t\tScheduleID:               40,\n\t\t\tScheduledEventBatchID:    30,\n\t\t\tScheduledEvent:           &types.HistoryEvent{ID: 400},\n\t\t\tScheduledTime:            currentTime,\n\t\t\tStartedID:                60,\n\t\t\tStartedEvent:             &types.HistoryEvent{ID: 600},\n\t\t\tStartedTime:              currentTime,\n\t\t\tScheduleToCloseTimeout:   10,\n\t\t\tScheduleToStartTimeout:   20,\n\t\t\tStartToCloseTimeout:      30,\n\t\t\tHeartbeatTimeout:         40,\n\t\t\tLastHeartBeatUpdatedTime: currentTime,\n\t\t\tTimerTaskStatus:          1,\n\t\t}}\n\n\tresetTimerInfos := []*p.TimerInfo{\n\t\t{\n\t\t\tVersion:    3333,\n\t\t\tTimerID:    \"t1_new\",\n\t\t\tStartedID:  1,\n\t\t\tExpiryTime: expiryTime,\n\t\t\tTaskStatus: 600,\n\t\t},\n\t\t{\n\t\t\tVersion:    3333,\n\t\t\tTimerID:    \"t2_new\",\n\t\t\tStartedID:  2,\n\t\t\tExpiryTime: expiryTime,\n\t\t\tTaskStatus: 601,\n\t\t}}\n\n\tresetChildExecutionInfos := []*p.ChildExecutionInfo{\n\t\t{\n\t\t\tVersion:         3334,\n\t\t\tInitiatedID:     10,\n\t\t\tInitiatedEvent:  &types.HistoryEvent{ID: 10},\n\t\t\tStartedID:       15,\n\t\t\tStartedRunID:    uuid.New(),\n\t\t\tStartedEvent:    nil,\n\t\t\tCreateRequestID: \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\tDomainID:        uuid.New(),\n\t\t}}\n\n\tresetRequestCancelInfos := []*p.RequestCancelInfo{\n\t\t{\n\t\t\tVersion:         3335,\n\t\t\tInitiatedID:     29,\n\t\t\tCancelRequestID: \"new_cancel_requested_id\",\n\t\t}}\n\n\tresetSignalInfos := []*p.SignalInfo{\n\t\t{\n\t\t\tVersion:         3336,\n\t\t\tInitiatedID:     39,\n\t\t\tSignalRequestID: \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\tSignalName:      \"signalB\",\n\t\t\tInput:           []byte(\"signal_input_b\"),\n\t\t\tControl:         []byte(\"signal_control_b\"),\n\t\t},\n\t\t{\n\t\t\tVersion:         3336,\n\t\t\tInitiatedID:     42,\n\t\t\tSignalRequestID: \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\tSignalName:      \"signalC\",\n\t\t\tInput:           []byte(\"signal_input_c\"),\n\t\t\tControl:         []byte(\"signal_control_c\"),\n\t\t}}\n\terr3 := s.ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tupdatedInfo1,\n\t\tupdatedStats1,\n\t\tint64(5),\n\t\tresetActivityInfos,\n\t\tresetTimerInfos,\n\t\tresetChildExecutionInfos,\n\t\tresetRequestCancelInfos,\n\t\tresetSignalInfos,\n\t\tnil,\n\t\tversionHistories,\n\t)\n\ts.NoError(err3)\n\n\tstats4, state4, err4 := s.GetWorkflowExecutionInfoWithStats(ctx, domainID, workflowExecution)\n\ts.NoError(err4)\n\ts.NotNil(state4, \"expected valid state.\")\n\ts.Equal(0, stats4.BufferedEventsCount)\n\ts.Equal(0, stats4.BufferedEventsSize)\n\n\tinfo4 := state4.ExecutionInfo\n\ts.T().Logf(\"%+v\\n\", info4)\n\ts.NotNil(info4, \"Valid Workflow info expected.\")\n\ts.Equal(int64(3), info4.NextEventID)\n\ts.Equal(\"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\", info4.FirstExecutionRunID)\n\ts.Equal(partitionConfig, info4.PartitionConfig)\n\n\ts.Equal(1, len(state4.ActivityInfos))\n\tai, ok = state4.ActivityInfos[40]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(int64(8789), ai.Version)\n\ts.Equal(int64(40), ai.ScheduleID)\n\ts.Equal(int64(30), ai.ScheduledEventBatchID)\n\ts.Equal(int64(400), ai.ScheduledEvent.ID)\n\ts.Equal(currentTime.Unix(), ai.ScheduledTime.Unix())\n\ts.Equal(int64(60), ai.StartedID)\n\ts.Equal(int64(600), ai.StartedEvent.ID)\n\ts.Equal(currentTime.Unix(), ai.StartedTime.Unix())\n\ts.Equal(int32(10), ai.ScheduleToCloseTimeout)\n\ts.Equal(int32(20), ai.ScheduleToStartTimeout)\n\ts.Equal(int32(30), ai.StartToCloseTimeout)\n\ts.Equal(int32(40), ai.HeartbeatTimeout)\n\ts.Equal(currentTime.Unix(), ai.LastHeartBeatUpdatedTime.Unix())\n\ts.Equal(int32(1), ai.TimerTaskStatus)\n\n\ts.Equal(2, len(state4.TimerInfos))\n\tti, ok = state4.TimerInfos[\"t1_new\"]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(int64(3333), ti.Version)\n\ts.Equal(\"t1_new\", ti.TimerID)\n\ts.Equal(int64(1), ti.StartedID)\n\ts.EqualTimes(expiryTime, ti.ExpiryTime)\n\ts.Equal(int64(600), ti.TaskStatus)\n\n\tti, ok = state4.TimerInfos[\"t2_new\"]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(int64(3333), ti.Version)\n\ts.Equal(\"t2_new\", ti.TimerID)\n\ts.Equal(int64(2), ti.StartedID)\n\ts.EqualTimes(expiryTime, ti.ExpiryTime)\n\ts.Equal(int64(601), ti.TaskStatus)\n\n\ts.Equal(1, len(state4.ChildExecutionInfos))\n\tci, ok = state4.ChildExecutionInfos[10]\n\ts.True(ok)\n\ts.NotNil(ci)\n\ts.Equal(resetChildExecutionInfos[0], ci)\n\n\ts.Equal(1, len(state4.RequestCancelInfos))\n\trci, ok = state4.RequestCancelInfos[29]\n\ts.True(ok)\n\ts.NotNil(rci)\n\ts.Equal(int64(3335), rci.Version)\n\ts.Equal(int64(29), rci.InitiatedID)\n\ts.Equal(\"new_cancel_requested_id\", rci.CancelRequestID)\n\n\ts.Equal(2, len(state4.SignalInfos))\n\tsi, ok = state4.SignalInfos[39]\n\ts.True(ok)\n\ts.NotNil(si)\n\ts.Equal(int64(3336), si.Version)\n\ts.Equal(int64(39), si.InitiatedID)\n\ts.Equal(\"signalB\", si.SignalName)\n\ts.Equal([]byte(\"signal_input_b\"), si.Input)\n\ts.Equal([]byte(\"signal_control_b\"), si.Control)\n\n\tsi, ok = state4.SignalInfos[42]\n\ts.True(ok)\n\ts.NotNil(si)\n\ts.Equal(int64(3336), si.Version)\n\ts.Equal(int64(42), si.InitiatedID)\n\ts.Equal(\"signalC\", si.SignalName)\n\ts.Equal([]byte(\"signal_input_c\"), si.Input)\n\ts.Equal([]byte(\"signal_control_c\"), si.Control)\n\n\ts.Equal(0, len(state4.SignalRequestedIDs))\n\n\ts.Equal(0, len(state4.BufferedEvents))\n\ts.assertChecksumsEqual(testWorkflowChecksum, state4.Checksum)\n\n}\n\n// TestConflictResolveWorkflowExecutionWithCASMismatch test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithCASMismatch() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-mismatch\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tnextEventID := int64(3)\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\n\tinfo := state.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info)\n\tcontinueAsNewStats := copyExecutionStats(state.ExecutionStats)\n\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tworkflowExecutionCurrent := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa1\",\n\t}\n\terr = s.ContinueAsNewExecution(\n\t\tctx,\n\t\tcontinueAsNewInfo,\n\t\tcontinueAsNewStats,\n\t\tinfo.NextEventID,\n\t\tworkflowExecutionCurrent,\n\t\tint64(3),\n\t\tint64(2),\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\trunID1, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), runID1)\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionCurrent)\n\ts.NoError(err)\n\tcurrentInfo := copyWorkflowExecutionInfo(state.ExecutionInfo)\n\tcurrentStats := copyExecutionStats(state.ExecutionStats)\n\tcurrentInfo.State = p.WorkflowStateCompleted\n\tcurrentInfo.CloseStatus = p.WorkflowCloseStatusCompleted\n\tcurrentInfo.NextEventID = int64(6)\n\tcurrentInfo.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: currentInfo.LastProcessedEvent,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\terr3 := s.UpdateWorkflowExecutionAndFinish(ctx, currentInfo, currentStats, int64(3), versionHistories)\n\ts.NoError(err3)\n\trunID1, err = s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), runID1)\n\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateRunning,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          111,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t}\n\tresetStats := &p.ExecutionStats{}\n\tresetActivityInfos := []*p.ActivityInfo{}\n\tresetTimerInfos := []*p.TimerInfo{}\n\tresetChildExecutionInfos := []*p.ChildExecutionInfo{}\n\tresetRequestCancelInfos := []*p.RequestCancelInfo{}\n\tresetSignalInfos := []*p.SignalInfo{}\n\n\terr = s.ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tresetExecutionInfo,\n\t\tresetStats,\n\t\tcontinueAsNewInfo.NextEventID,\n\t\tresetActivityInfos,\n\t\tresetTimerInfos,\n\t\tresetChildExecutionInfos,\n\t\tresetRequestCancelInfos,\n\t\tresetSignalInfos,\n\t\tnil,\n\t\tversionHistories,\n\t)\n\ts.NotNil(err)\n\n\terr = s.ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tresetExecutionInfo,\n\t\tresetStats,\n\t\tcontinueAsNewInfo.NextEventID,\n\t\tresetActivityInfos,\n\t\tresetTimerInfos,\n\t\tresetChildExecutionInfos,\n\t\tresetRequestCancelInfos,\n\t\tresetSignalInfos,\n\t\tnil,\n\t\tversionHistories,\n\t)\n\ts.NotNil(err)\n\n\terr = s.ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tresetExecutionInfo,\n\t\tresetStats,\n\t\tcontinueAsNewInfo.NextEventID,\n\t\tresetActivityInfos,\n\t\tresetTimerInfos,\n\t\tresetChildExecutionInfos,\n\t\tresetRequestCancelInfos,\n\t\tresetSignalInfos,\n\t\tnil,\n\t\tversionHistories,\n\t)\n\ts.NotNil(err)\n\n\t// this test only assert whether the current workflow execution record is reset\n\trunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), runID)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionCurrentIsNotSelf test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionCurrentIsNotSelf() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-not-self\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tnextEventID := int64(3)\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\n\tinfo := state.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info)\n\tcontinueAsNewStats := copyExecutionStats(state.ExecutionStats)\n\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tworkflowExecutionCurrent := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa1\",\n\t}\n\terr = s.ContinueAsNewExecution(\n\t\tctx,\n\t\tcontinueAsNewInfo,\n\t\tcontinueAsNewStats,\n\t\tinfo.NextEventID,\n\t\tworkflowExecutionCurrent,\n\t\tint64(3),\n\t\tint64(2),\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), currentRunID)\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionCurrent)\n\ts.NoError(err)\n\tcurrentInfo := copyWorkflowExecutionInfo(state.ExecutionInfo)\n\tcurrentInfo.State = p.WorkflowStateCompleted\n\tcurrentInfo.CloseStatus = p.WorkflowCloseStatusTerminated\n\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateCompleted,\n\t\tCloseStatus:                 p.WorkflowCloseStatusCompleted,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          111,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             nil,\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: resetExecutionInfo.DecisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeBypassCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      int64(5),\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: nil,\n\t\tCurrentWorkflowMutation: &p.WorkflowMutation{\n\t\t\tExecutionInfo:  currentInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      int64(3),\n\n\t\t\tUpsertActivityInfos:       []*p.ActivityInfo{},\n\t\t\tUpsertTimerInfos:          []*p.TimerInfo{},\n\t\t\tUpsertChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tUpsertRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tUpsertSignalInfos:         []*p.SignalInfo{},\n\t\t\tUpsertSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:          versionHistories,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeUpdateCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(resetExecutionInfo.RunID, currentRecord.RunID)\n\ts.Equal(resetExecutionInfo.CreateRequestID, currentRecord.StartRequestID)\n\ts.Equal(resetExecutionInfo.State, currentRecord.State)\n\ts.Equal(resetExecutionInfo.CloseStatus, currentRecord.CloseStatus)\n\ts.Equal(int64(-24), currentRecord.LastWriteVersion)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionCurrent)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.CurrentWorkflowMutation.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.CurrentWorkflowMutation.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.CurrentWorkflowMutation.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.CurrentWorkflowMutation.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionCurrentIsNotSelfWithContinueAsNew test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionCurrentIsNotSelfWithContinueAsNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-not-self-with-continue-as-new\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tnextEventID := int64(3)\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tcontinueAsNewStats := copyExecutionStats(state.ExecutionStats)\n\n\tinfo := state.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info)\n\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tworkflowExecutionCurrent := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa1\",\n\t}\n\terr = s.ContinueAsNewExecution(\n\t\tctx,\n\t\tcontinueAsNewInfo,\n\t\tcontinueAsNewStats,\n\t\tinfo.NextEventID,\n\t\tworkflowExecutionCurrent,\n\t\tint64(3),\n\t\tint64(2),\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), currentRunID)\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionCurrent)\n\ts.NoError(err)\n\tcurrentInfo := copyWorkflowExecutionInfo(state.ExecutionInfo)\n\tcurrentStats := copyExecutionStats(state.ExecutionStats)\n\tcurrentInfo.State = p.WorkflowStateCompleted\n\tcurrentInfo.CloseStatus = p.WorkflowCloseStatusTerminated\n\tdecisionScheduleID := int64(111)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateCompleted,\n\t\tCloseStatus:                 p.WorkflowCloseStatusContinuedAsNew,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          decisionScheduleID,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             map[string]string{\"zone\": \"dca1\"},\n\t}\n\tnewWorkflowExecutionInfo := copyWorkflowExecutionInfo(resetExecutionInfo)\n\tnewWorkflowExecutionStats := &p.ExecutionStats{}\n\tnewWorkflowExecutionInfo.CreateRequestID = uuid.New()\n\tnewWorkflowExecutionInfo.RunID = \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa2\"\n\tnewWorkflowExecutionInfo.State = p.WorkflowStateRunning\n\tnewWorkflowExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\tnewWorkflowExecutionInfo.PartitionConfig = map[string]string{\"user\": uuid.New()}\n\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeBypassCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      int64(5),\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: &p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  newWorkflowExecutionInfo,\n\t\t\tExecutionStats: newWorkflowExecutionStats,\n\t\t\tCondition:      0,\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tCurrentWorkflowMutation: &p.WorkflowMutation{\n\t\t\tExecutionInfo:  currentInfo,\n\t\t\tExecutionStats: currentStats,\n\t\t\tCondition:      int64(3),\n\n\t\t\tUpsertActivityInfos:       []*p.ActivityInfo{},\n\t\t\tUpsertTimerInfos:          []*p.TimerInfo{},\n\t\t\tUpsertChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tUpsertRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tUpsertSignalInfos:         []*p.SignalInfo{},\n\t\t\tUpsertSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:          versionHistories,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeUpdateCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(newWorkflowExecutionInfo.RunID, currentRecord.RunID)\n\ts.Equal(newWorkflowExecutionInfo.CreateRequestID, currentRecord.StartRequestID)\n\ts.Equal(newWorkflowExecutionInfo.State, currentRecord.State)\n\ts.Equal(newWorkflowExecutionInfo.CloseStatus, currentRecord.CloseStatus)\n\ts.Equal(int64(-24), currentRecord.LastWriteVersion)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionCurrent)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.CurrentWorkflowMutation.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.CurrentWorkflowMutation.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.CurrentWorkflowMutation.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.CurrentWorkflowMutation.ExecutionInfo, state.ExecutionInfo)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      resetReq.NewWorkflowSnapshot.ExecutionInfo.RunID,\n\t})\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.NewWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.NewWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionCurrentIsSelf test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionCurrentIsSelf() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-self\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tnextEventID := int64(3)\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\t_, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateRunning,\n\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          111,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             nil,\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: resetExecutionInfo.DecisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeBypassCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      nextEventID,\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot:     nil,\n\t\tCurrentWorkflowMutation: nil,\n\t\tEncoding:                pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeUpdateCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(resetExecutionInfo.RunID, currentRecord.RunID)\n\ts.Equal(resetExecutionInfo.CreateRequestID, currentRecord.StartRequestID)\n\ts.Equal(resetExecutionInfo.State, currentRecord.State)\n\ts.Equal(resetExecutionInfo.CloseStatus, currentRecord.CloseStatus)\n\ts.Equal(int64(-24), currentRecord.LastWriteVersion)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionCurrentIsSelfWithContinueAsNew test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionCurrentIsSelfWithContinueAsNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-self-with-continue-as-new\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tnextEventID := int64(3)\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\t_, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\n\tdecisionScheduleID := int64(111)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateCompleted,\n\t\tCloseStatus:                 p.WorkflowCloseStatusContinuedAsNew,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          decisionScheduleID,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             nil,\n\t}\n\tnewWorkflowExecutionInfo := copyWorkflowExecutionInfo(resetExecutionInfo)\n\tnewWorkflowExecutionStats := &p.ExecutionStats{}\n\tnewWorkflowExecutionInfo.CreateRequestID = uuid.New()\n\tnewWorkflowExecutionInfo.RunID = \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa2\"\n\tnewWorkflowExecutionInfo.State = p.WorkflowStateRunning\n\tnewWorkflowExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\tnewWorkflowExecutionInfo.PartitionConfig = map[string]string{\"zone\": \"dca1\"}\n\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeBypassCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      nextEventID,\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: &p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  newWorkflowExecutionInfo,\n\t\t\tExecutionStats: newWorkflowExecutionStats,\n\t\t\tCondition:      0,\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tCurrentWorkflowMutation: nil,\n\t\tEncoding:                pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeUpdateCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(newWorkflowExecutionInfo.RunID, currentRecord.RunID)\n\ts.Equal(newWorkflowExecutionInfo.CreateRequestID, currentRecord.StartRequestID)\n\ts.Equal(newWorkflowExecutionInfo.State, currentRecord.State)\n\ts.Equal(newWorkflowExecutionInfo.CloseStatus, currentRecord.CloseStatus)\n\ts.Equal(int64(-24), currentRecord.LastWriteVersion)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      resetReq.NewWorkflowSnapshot.ExecutionInfo.RunID,\n\t})\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.NewWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.NewWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionZombieIsSelf test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionZombieIsSelf() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-zombie\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tnextEventID := int64(3)\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\n\tinfo := state.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info)\n\tcontinueAsNewStats := copyExecutionStats(state.ExecutionStats)\n\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tworkflowExecutionCurrent := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa1\",\n\t}\n\terr = s.ContinueAsNewExecution(\n\t\tctx,\n\t\tcontinueAsNewInfo,\n\t\tcontinueAsNewStats,\n\t\tinfo.NextEventID,\n\t\tworkflowExecutionCurrent,\n\t\tint64(3),\n\t\tint64(2),\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), currentRunID)\n\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateCompleted,\n\t\tCloseStatus:                 p.WorkflowCloseStatusCompleted,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          111,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             nil,\n\t}\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: resetExecutionInfo.DecisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeUpdateCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      int64(5),\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot:     nil,\n\t\tCurrentWorkflowMutation: nil,\n\t\tEncoding:                pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeBypassCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(currentRunID, currentRecord.RunID)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestConflictResolveWorkflowExecutionWithTransactionZombieIsSelfWithContinueAsNew test\nfunc (s *ExecutionManagerSuite) TestConflictResolveWorkflowExecutionWithTransactionZombieIsSelfWithContinueAsNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"4ca1faac-1a3a-47af-8e51-fdaa2b3d45b9\"\n\tworkflowID := \"test-reset-mutable-state-test-with-transaction-current-is-zombie-with-continue-as-new\"\n\n\t// first create a workflow and continue as new it\n\tworkflowExecutionReset := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa0\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tnextEventID := int64(3)\n\tresp, err := s.CreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowExecutionReset,\n\t\t\"taskList\",\n\t\t\"wType\",\n\t\t20,\n\t\t13,\n\t\tnil,\n\t\tnextEventID,\n\t\t0,\n\t\t2,\n\t\tnil,\n\t\tpartitionConfig,\n\t)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\n\tstate, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tcontinueAsNewStats := copyExecutionStats(state.ExecutionStats)\n\n\tinfo := state.ExecutionInfo\n\tcontinueAsNewInfo := copyWorkflowExecutionInfo(info)\n\tcontinueAsNewInfo.State = p.WorkflowStateRunning\n\tcontinueAsNewInfo.NextEventID = int64(5)\n\tcontinueAsNewInfo.LastProcessedEvent = int64(2)\n\n\tworkflowExecutionCurrent := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa1\",\n\t}\n\terr = s.ContinueAsNewExecution(\n\t\tctx,\n\t\tcontinueAsNewInfo,\n\t\tcontinueAsNewStats,\n\t\tinfo.NextEventID,\n\t\tworkflowExecutionCurrent,\n\t\tint64(3),\n\t\tint64(2),\n\t\tnil,\n\t)\n\ts.NoError(err)\n\n\tcurrentRunID, err := s.GetCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(workflowExecutionCurrent.GetRunID(), currentRunID)\n\n\tdecisionScheduleID := int64(111)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\tresetExecutionInfo := &p.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  workflowExecutionReset.GetWorkflowID(),\n\t\tRunID:                       workflowExecutionReset.GetRunID(),\n\t\tFirstExecutionRunID:         workflowExecutionReset.GetRunID(),\n\t\tParentDomainID:              uuid.New(),\n\t\tParentWorkflowID:            \"some random parent workflow ID\",\n\t\tParentRunID:                 uuid.New(),\n\t\tInitiatedID:                 12345,\n\t\tTaskList:                    \"some random tasklist\",\n\t\tWorkflowTypeName:            \"some random workflow type name\",\n\t\tWorkflowTimeout:             1112,\n\t\tDecisionStartToCloseTimeout: 14,\n\t\tState:                       p.WorkflowStateCompleted,\n\t\tCloseStatus:                 p.WorkflowCloseStatusContinuedAsNew,\n\t\tLastFirstEventID:            constants.FirstEventID,\n\t\tNextEventID:                 123,\n\t\tCreateRequestID:             uuid.New(),\n\t\tDecisionVersion:             constants.EmptyVersion,\n\t\tDecisionScheduleID:          decisionScheduleID,\n\t\tDecisionStartedID:           222,\n\t\tDecisionRequestID:           uuid.New(),\n\t\tDecisionTimeout:             0,\n\t\tAutoResetPoints:             &types.ResetPoints{},\n\t\tPartitionConfig:             nil,\n\t}\n\tnewWorkflowExecutionInfo := copyWorkflowExecutionInfo(resetExecutionInfo)\n\tnewWorkflowExecutionStats := &p.ExecutionStats{}\n\tnewWorkflowExecutionInfo.CreateRequestID = uuid.New()\n\tnewWorkflowExecutionInfo.RunID = \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaa2\"\n\tnewWorkflowExecutionInfo.State = p.WorkflowStateZombie\n\tnewWorkflowExecutionInfo.CloseStatus = p.WorkflowCloseStatusNone\n\tnewWorkflowExecutionInfo.PartitionConfig = map[string]string{\"zone\": \"dca1\"}\n\tresetReq := &p.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    p.ConflictResolveWorkflowModeUpdateCurrent,\n\t\tResetWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  resetExecutionInfo,\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tCondition:      int64(5),\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: &p.WorkflowSnapshot{\n\t\t\tExecutionInfo:  newWorkflowExecutionInfo,\n\t\t\tExecutionStats: newWorkflowExecutionStats,\n\t\t\tCondition:      0,\n\n\t\t\tActivityInfos:       []*p.ActivityInfo{},\n\t\t\tTimerInfos:          []*p.TimerInfo{},\n\t\t\tChildExecutionInfos: []*p.ChildExecutionInfo{},\n\t\t\tRequestCancelInfos:  []*p.RequestCancelInfo{},\n\t\t\tSignalInfos:         []*p.SignalInfo{},\n\t\t\tSignalRequestedIDs:  []string{},\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tCurrentWorkflowMutation: nil,\n\t\tEncoding:                pickRandomEncoding(),\n\t}\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.Error(err)\n\tresetReq.Mode = p.ConflictResolveWorkflowModeBypassCurrent\n\t_, err = s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, resetReq)\n\ts.NoError(err)\n\n\tcurrentRecord, err := s.ExecutionManager.GetCurrentExecution(ctx, &p.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\ts.NoError(err)\n\ts.Equal(currentRunID, currentRecord.RunID)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecutionReset)\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.ResetWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.ResetWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.ResetWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n\n\tstate, err = s.GetWorkflowExecutionInfo(ctx, domainID, types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      resetReq.NewWorkflowSnapshot.ExecutionInfo.RunID,\n\t})\n\ts.NoError(err)\n\tstate.ExecutionInfo.StartTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp\n\tstate.ExecutionInfo.LastUpdatedTimestamp = resetReq.NewWorkflowSnapshot.ExecutionInfo.LastUpdatedTimestamp\n\tstate.ExecutionInfo.ExpirationTime = resetReq.NewWorkflowSnapshot.ExecutionInfo.ExpirationTime\n\ts.Equal(resetReq.NewWorkflowSnapshot.ExecutionInfo, state.ExecutionInfo)\n}\n\n// TestReplicationDLQ test\nfunc (s *ExecutionManagerSuite) TestReplicationDLQ() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tsourceCluster := \"test\"\n\ttaskInfo := &p.ReplicationTaskInfo{\n\t\tDomainID:   uuid.New(),\n\t\tWorkflowID: uuid.New(),\n\t\tRunID:      uuid.New(),\n\t\tTaskID:     0,\n\t\tTaskType:   0,\n\t}\n\terr := s.PutReplicationTaskToDLQ(ctx, sourceCluster, taskInfo)\n\ts.NoError(err)\n\tresp, err := s.GetReplicationTasksFromDLQ(ctx, sourceCluster, 0, 1, 1, nil)\n\ts.NoError(err)\n\ts.Len(resp.Tasks, 1)\n\terr = s.DeleteReplicationTaskFromDLQ(ctx, sourceCluster, 0)\n\ts.NoError(err)\n\tresp, err = s.GetReplicationTasksFromDLQ(ctx, sourceCluster, 0, 1, 1, nil)\n\ts.NoError(err)\n\ts.Len(resp.Tasks, 0)\n\n\ttaskInfo1 := &p.ReplicationTaskInfo{\n\t\tDomainID:   uuid.New(),\n\t\tWorkflowID: uuid.New(),\n\t\tRunID:      uuid.New(),\n\t\tTaskID:     1,\n\t\tTaskType:   0,\n\t}\n\ttaskInfo2 := &p.ReplicationTaskInfo{\n\t\tDomainID:   uuid.New(),\n\t\tWorkflowID: uuid.New(),\n\t\tRunID:      uuid.New(),\n\t\tTaskID:     2,\n\t\tTaskType:   0,\n\t}\n\terr = s.PutReplicationTaskToDLQ(ctx, sourceCluster, taskInfo1)\n\ts.NoError(err)\n\terr = s.PutReplicationTaskToDLQ(ctx, sourceCluster, taskInfo2)\n\ts.NoError(err)\n\tresp, err = s.GetReplicationTasksFromDLQ(ctx, sourceCluster, 1, 3, 2, nil)\n\ts.NoError(err)\n\ts.Len(resp.Tasks, 2)\n\tsizeResp, err := s.GetReplicationDLQSize(ctx, sourceCluster)\n\ts.NoError(err)\n\ts.Equal(int64(2), sizeResp.Size)\n\terr = s.RangeDeleteReplicationTaskFromDLQ(ctx, sourceCluster, 1, 3)\n\ts.NoError(err)\n\tresp, err = s.GetReplicationTasksFromDLQ(ctx, sourceCluster, 1, 3, 2, nil)\n\ts.NoError(err)\n\ts.Len(resp.Tasks, 0)\n}\n\n// TestCreateFailoverMarkerTasks test\nfunc (s *ExecutionManagerSuite) TestCreateFailoverMarkerTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tmarkers := []*p.FailoverMarkerTask{\n\t\t{\n\t\t\tTaskData: p.TaskData{\n\t\t\t\tTaskID:              1,\n\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\tVersion:             1,\n\t\t\t},\n\t\t\tDomainID: domainID,\n\t\t},\n\t}\n\terr := s.CreateFailoverMarkers(ctx, markers)\n\ts.NoError(err)\n\n\ttasks, err := s.GetReplicationTasks(ctx, 1, true)\n\ts.NoError(err)\n\ts.Equal(len(tasks), 1)\n\ts.Equal(tasks[0].GetVersion(), int64(1))\n\ts.Equal(tasks[0].GetTaskID(), int64(1))\n\ts.Equal(tasks[0].GetDomainID(), domainID)\n\ts.Equal(tasks[0].GetTaskType(), p.ReplicationTaskTypeFailoverMarker)\n}\n\nfunc copyWorkflowExecutionInfo(sourceInfo *p.WorkflowExecutionInfo) *p.WorkflowExecutionInfo {\n\treturn &p.WorkflowExecutionInfo{\n\t\tDomainID:                    sourceInfo.DomainID,\n\t\tWorkflowID:                  sourceInfo.WorkflowID,\n\t\tRunID:                       sourceInfo.RunID,\n\t\tFirstExecutionRunID:         sourceInfo.FirstExecutionRunID,\n\t\tParentDomainID:              sourceInfo.ParentDomainID,\n\t\tParentWorkflowID:            sourceInfo.ParentWorkflowID,\n\t\tParentRunID:                 sourceInfo.ParentRunID,\n\t\tInitiatedID:                 sourceInfo.InitiatedID,\n\t\tCompletionEvent:             sourceInfo.CompletionEvent,\n\t\tTaskList:                    sourceInfo.TaskList,\n\t\tTaskListKind:                sourceInfo.TaskListKind,\n\t\tWorkflowTypeName:            sourceInfo.WorkflowTypeName,\n\t\tWorkflowTimeout:             sourceInfo.WorkflowTimeout,\n\t\tDecisionStartToCloseTimeout: sourceInfo.DecisionStartToCloseTimeout,\n\t\tExecutionContext:            sourceInfo.ExecutionContext,\n\t\tState:                       sourceInfo.State,\n\t\tCloseStatus:                 sourceInfo.CloseStatus,\n\t\tLastFirstEventID:            sourceInfo.LastFirstEventID,\n\t\tNextEventID:                 sourceInfo.NextEventID,\n\t\tLastProcessedEvent:          sourceInfo.LastProcessedEvent,\n\t\tLastUpdatedTimestamp:        sourceInfo.LastUpdatedTimestamp,\n\t\tStartTimestamp:              sourceInfo.StartTimestamp,\n\t\tCreateRequestID:             sourceInfo.CreateRequestID,\n\t\tDecisionVersion:             sourceInfo.DecisionVersion,\n\t\tDecisionScheduleID:          sourceInfo.DecisionScheduleID,\n\t\tDecisionStartedID:           sourceInfo.DecisionStartedID,\n\t\tDecisionRequestID:           sourceInfo.DecisionRequestID,\n\t\tDecisionTimeout:             sourceInfo.DecisionTimeout,\n\t\tBranchToken:                 sourceInfo.BranchToken,\n\t\tAutoResetPoints:             sourceInfo.AutoResetPoints,\n\t\tPartitionConfig:             sourceInfo.PartitionConfig,\n\t}\n}\n\nfunc copyExecutionStats(sourceStats *p.ExecutionStats) *p.ExecutionStats {\n\treturn &p.ExecutionStats{\n\t\tHistorySize: sourceStats.HistorySize,\n\t}\n}\n\n// Note: cassandra only provide millisecond precision timestamp\n// ref: https://docs.datastax.com/en/cql/3.3/cql/cql_reference/timestamp_type_r.html\n// so to use equal function, we need to do conversion, getting rid of sub milliseconds\nfunc timestampConvertor(t time.Time) time.Time {\n\treturn time.Unix(\n\t\t0,\n\t\tp.DBTimestampToUnixNano(p.UnixNanoToDBTimestamp(t.UnixNano())),\n\t).UTC()\n}\n\nfunc timeComparator(t1, t2 time.Time, timeTolerance time.Duration) bool {\n\tdiff := t2.Sub(t1)\n\treturn diff.Nanoseconds() <= timeTolerance.Nanoseconds()\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/executionManagerTestForEventsV2.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"runtime/debug\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ExecutionManagerSuiteForEventsV2 contains matching persistence tests\n\tExecutionManagerSuiteForEventsV2 struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\nfunc failOnPanic(t *testing.T) {\n\tdefer func() {\n\t\tr := recover()\n\t\tif r != nil {\n\t\t\tt.Errorf(\"test panicked: %v %s\", r, debug.Stack())\n\t\t\tt.FailNow()\n\t\t}\n\t}()\n}\n\n// SetupSuite implementation\nfunc (s *ExecutionManagerSuiteForEventsV2) SetupSuite() {\n\tdefer failOnPanic(s.T())\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// TearDownSuite implementation\nfunc (s *ExecutionManagerSuiteForEventsV2) TearDownSuite() {\n\tdefer failOnPanic(s.T())\n\ts.TearDownWorkflowStore()\n}\n\n// SetupTest implementation\nfunc (s *ExecutionManagerSuiteForEventsV2) SetupTest() {\n\tdefer failOnPanic(s.T())\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.ClearTasks()\n}\n\nfunc (s *ExecutionManagerSuiteForEventsV2) newRandomChecksum() checksum.Checksum {\n\treturn checksum.Checksum{\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tVersion: 22,\n\t\tValue:   []byte(uuid.NewRandom()),\n\t}\n}\n\nfunc (s *ExecutionManagerSuiteForEventsV2) assertChecksumsEqual(expected checksum.Checksum, actual checksum.Checksum) {\n\tif !actual.Flavor.IsValid() {\n\t\t// not all stores support checksum persistence today\n\t\t// if its not supported, assert that everything is zero'd out\n\t\texpected = checksum.Checksum{}\n\t}\n\ts.EqualValues(expected, actual)\n}\n\n// TestWorkflowCreation test\nfunc (s *ExecutionManagerSuiteForEventsV2) TestWorkflowCreation() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdefer failOnPanic(s.T())\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-eventsv2-workflow\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\tdomainName := uuid.New()\n\tcsum := s.newRandomChecksum()\n\tdecisionScheduleID := int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{decisionScheduleID, constants.EmptyVersion},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\t_, err0 := s.ExecutionManager.CreateWorkflowExecution(ctx, &p.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    \"taskList\",\n\t\t\t\tWorkflowTypeName:            \"wType\",\n\t\t\t\tWorkflowTimeout:             20,\n\t\t\t\tDecisionStartToCloseTimeout: 13,\n\t\t\t\tExecutionContext:            nil,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tNextEventID:                 3,\n\t\t\t\tLastProcessedEvent:          0,\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tBranchToken:                 []byte(\"branchToken1\"),\n\t\t\t},\n\t\t\tExecutionStats: &p.ExecutionStats{},\n\t\t\tTasksByCategory: map[p.HistoryTaskCategory][]p.Task{\n\t\t\t\tp.HistoryTaskCategoryTransfer: []p.Task{\n\t\t\t\t\t&p.DecisionTask{\n\t\t\t\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: p.TaskData{\n\t\t\t\t\t\t\tTaskID:              s.GetNextSequenceNumber(),\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTargetDomainID: domainID,\n\t\t\t\t\t\tTaskList:       \"taskList\",\n\t\t\t\t\t\tScheduleID:     2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:    s.ShardInfo.RangeID,\n\t\tDomainName: domainName,\n\t})\n\n\ts.NoError(err0)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal([]byte(\"branchToken1\"), info0.BranchToken)\n\ts.assertChecksumsEqual(csum, state0.Checksum)\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tcurrentTime := time.Now().UTC()\n\ttimerID := \"id_1\"\n\ttimerInfos := []*p.TimerInfo{{\n\t\tVersion:    3345,\n\t\tTimerID:    timerID,\n\t\tExpiryTime: currentTime,\n\t\tTaskStatus: 2,\n\t\tStartedID:  5,\n\t}}\n\tupdatedInfo.BranchToken = []byte(\"branchToken2\")\n\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, int64(3), nil, nil, nil, timerInfos, nil)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.TimerInfos))\n\ts.Equal(int64(3345), state.TimerInfos[timerID].Version)\n\ts.Equal(timerID, state.TimerInfos[timerID].TimerID)\n\ts.EqualTimesWithPrecision(currentTime, state.TimerInfos[timerID].ExpiryTime, time.Millisecond*500)\n\ts.Equal(int64(2), state.TimerInfos[timerID].TaskStatus)\n\ts.Equal(int64(5), state.TimerInfos[timerID].StartedID)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state.Checksum)\n\n\terr2 = s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, nil, nil, int64(5), nil, nil, nil, nil, []string{timerID})\n\ts.NoError(err2)\n\n\tstate, err2 = s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err2)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(0, len(state.TimerInfos))\n\tinfo1 := state.ExecutionInfo\n\ts.Equal([]byte(\"branchToken2\"), info1.BranchToken)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state.Checksum)\n}\n\n// TestWorkflowCreationWithVersionHistories test\nfunc (s *ExecutionManagerSuiteForEventsV2) TestWorkflowCreationWithVersionHistories() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdefer failOnPanic(s.T())\n\tdomainID := uuid.New()\n\tdomainName := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-eventsv2-workflow-version-history\",\n\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t}\n\tversionHistory := p.NewVersionHistory(\n\t\t[]byte{1},\n\t\t[]*p.VersionHistoryItem{p.NewVersionHistoryItem(1, 0)},\n\t)\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tcsum := s.newRandomChecksum()\n\n\t_, err0 := s.ExecutionManager.CreateWorkflowExecution(ctx, &p.CreateWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tNewWorkflowSnapshot: p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    \"taskList\",\n\t\t\t\tWorkflowTypeName:            \"wType\",\n\t\t\t\tWorkflowTimeout:             20,\n\t\t\t\tDecisionStartToCloseTimeout: 13,\n\t\t\t\tExecutionContext:            nil,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tNextEventID:                 constants.EmptyEventID,\n\t\t\t\tLastProcessedEvent:          0,\n\t\t\t\tDecisionScheduleID:          2,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tBranchToken:                 nil,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tVersionHistories: versionHistories,\n\t\t\tTasksByCategory: map[p.HistoryTaskCategory][]p.Task{\n\t\t\t\tp.HistoryTaskCategoryTransfer: []p.Task{\n\t\t\t\t\t&p.DecisionTask{\n\t\t\t\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: p.TaskData{\n\t\t\t\t\t\t\tTaskID:              s.GetNextSequenceNumber(),\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTargetDomainID: domainID,\n\t\t\t\t\t\tTaskList:       \"taskList\",\n\t\t\t\t\t\tScheduleID:     2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tChecksum: csum,\n\t\t},\n\t\tDomainName: domainName,\n\t})\n\n\ts.NoError(err0)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\ts.NotNil(info0, \"Valid Workflow info expected.\")\n\ts.Equal(versionHistories, state0.VersionHistories)\n\ts.assertChecksumsEqual(csum, state0.Checksum)\n\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tcurrentTime := time.Now().UTC()\n\ttimerID := \"id_1\"\n\ttimerInfos := []*p.TimerInfo{{\n\t\tVersion:    3345,\n\t\tTimerID:    timerID,\n\t\tExpiryTime: currentTime,\n\t\tTaskStatus: 2,\n\t\tStartedID:  5,\n\t}}\n\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\ts.NoError(err)\n\terr = versionHistory.AddOrUpdateItem(p.NewVersionHistoryItem(2, 0))\n\ts.NoError(err)\n\n\terr2 := s.UpdateWorkflowExecution(ctx, updatedInfo, updatedStats, versionHistories, []int64{int64(4)}, nil, constants.EmptyEventID, nil, nil, nil, timerInfos, nil)\n\ts.NoError(err2)\n\n\tstate, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\ts.NotNil(state, \"expected valid state.\")\n\ts.Equal(1, len(state.TimerInfos))\n\ts.Equal(int64(3345), state.TimerInfos[timerID].Version)\n\ts.Equal(timerID, state.TimerInfos[timerID].TimerID)\n\ts.EqualTimesWithPrecision(currentTime, state.TimerInfos[timerID].ExpiryTime, time.Millisecond*500)\n\ts.Equal(int64(2), state.TimerInfos[timerID].TaskStatus)\n\ts.Equal(int64(5), state.TimerInfos[timerID].StartedID)\n\ts.Equal(state.VersionHistories, versionHistories)\n\ts.assertChecksumsEqual(testWorkflowChecksum, state.Checksum)\n}\n\n// TestContinueAsNew test\nfunc (s *ExecutionManagerSuiteForEventsV2) TestContinueAsNew() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"continue-as-new-workflow-test\",\n\t\tRunID:      \"551c88d2-d9e6-404f-8131-9eec14f36643\",\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"userID\": uuid.New(),\n\t}\n\tdecisionScheduleID := int64(2)\n\t_, err0 := s.CreateWorkflowExecution(ctx, domainID, workflowExecution, \"queue1\", \"wType\", 20, 13, nil, 3, 0, decisionScheduleID, nil, partitionConfig)\n\ts.NoError(err0)\n\n\tstate0, err1 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err1)\n\tinfo0 := state0.ExecutionInfo\n\tupdatedInfo := copyWorkflowExecutionInfo(info0)\n\tupdatedStats := copyExecutionStats(state0.ExecutionStats)\n\tupdatedInfo.State = p.WorkflowStateCompleted\n\tupdatedInfo.CloseStatus = p.WorkflowCloseStatusCompleted\n\tupdatedInfo.NextEventID = int64(5)\n\tupdatedInfo.LastProcessedEvent = int64(2)\n\tversionHistory := p.NewVersionHistory([]byte{}, []*p.VersionHistoryItem{\n\t\t{decisionScheduleID, constants.EmptyVersion},\n\t})\n\tversionHistories := p.NewVersionHistories(versionHistory)\n\n\tnewWorkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"continue-as-new-workflow-test\",\n\t\tRunID:      \"64c7e15a-3fd7-4182-9c6f-6f25a4fa2614\",\n\t}\n\n\tnewdecisionTask := &p.DecisionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t},\n\t\tTaskData: p.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID: updatedInfo.DomainID,\n\t\tTaskList:       updatedInfo.TaskList,\n\t\tScheduleID:     int64(2),\n\t}\n\n\t_, err2 := s.ExecutionManager.UpdateWorkflowExecution(ctx, &p.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: p.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[p.HistoryTaskCategory][]p.Task{\n\t\t\t\tp.HistoryTaskCategoryTransfer: []p.Task{newdecisionTask},\n\t\t\t},\n\t\t\tCondition:           info0.NextEventID,\n\t\t\tUpsertActivityInfos: nil,\n\t\t\tDeleteActivityInfos: nil,\n\t\t\tUpsertTimerInfos:    nil,\n\t\t\tDeleteTimerInfos:    nil,\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: &p.WorkflowSnapshot{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    updatedInfo.DomainID,\n\t\t\t\tWorkflowID:                  newWorkflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       newWorkflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         updatedInfo.FirstExecutionRunID,\n\t\t\t\tTaskList:                    updatedInfo.TaskList,\n\t\t\t\tTaskListKind:                updatedInfo.TaskListKind,\n\t\t\t\tWorkflowTypeName:            updatedInfo.WorkflowTypeName,\n\t\t\t\tWorkflowTimeout:             updatedInfo.WorkflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: updatedInfo.DecisionStartToCloseTimeout,\n\t\t\t\tExecutionContext:            nil,\n\t\t\t\tState:                       p.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 p.WorkflowCloseStatusNone,\n\t\t\t\tNextEventID:                 info0.NextEventID,\n\t\t\t\tLastProcessedEvent:          constants.EmptyEventID,\n\t\t\t\tDecisionScheduleID:          int64(2),\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tBranchToken:                 []byte(\"branchToken1\"),\n\t\t\t\tPartitionConfig:             partitionConfig,\n\t\t\t},\n\t\t\tExecutionStats:   &p.ExecutionStats{},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\n\ts.NoError(err2)\n\n\tprevExecutionState, err3 := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\ts.NoError(err3)\n\tprevExecutionInfo := prevExecutionState.ExecutionInfo\n\ts.Equal(\"551c88d2-d9e6-404f-8131-9eec14f36643\", prevExecutionInfo.FirstExecutionRunID)\n\ts.Equal(p.WorkflowStateCompleted, prevExecutionInfo.State)\n\ts.Equal(int64(5), prevExecutionInfo.NextEventID)\n\ts.Equal(int64(2), prevExecutionInfo.LastProcessedEvent)\n\ts.Equal(partitionConfig, prevExecutionInfo.PartitionConfig)\n\n\tnewExecutionState, err4 := s.GetWorkflowExecutionInfo(ctx, domainID, newWorkflowExecution)\n\ts.NoError(err4)\n\tnewExecutionInfo := newExecutionState.ExecutionInfo\n\ts.Equal(\"551c88d2-d9e6-404f-8131-9eec14f36643\", newExecutionInfo.FirstExecutionRunID)\n\ts.Equal(p.WorkflowStateRunning, newExecutionInfo.State)\n\ts.Equal(p.WorkflowCloseStatusNone, newExecutionInfo.CloseStatus)\n\ts.Equal(int64(3), newExecutionInfo.NextEventID)\n\ts.Equal(constants.EmptyEventID, newExecutionInfo.LastProcessedEvent)\n\ts.Equal(int64(2), newExecutionInfo.DecisionScheduleID)\n\ts.Equal([]byte(\"branchToken1\"), newExecutionInfo.BranchToken)\n\ts.Equal(partitionConfig, newExecutionInfo.PartitionConfig)\n\n\tnewRunID, err5 := s.GetCurrentWorkflowRunID(ctx, domainID, workflowExecution.WorkflowID)\n\ts.NoError(err5)\n\ts.Equal(newWorkflowExecution.RunID, newRunID)\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/historyV2PersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"math\"\n\t\"math/rand\"\n\t\"os\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/codec\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\ntype (\n\t// HistoryV2PersistenceSuite contains history persistence tests\n\tHistoryV2PersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\nconst testForkRunID = \"11220000-0000-f000-f000-000000000000\"\n\nvar (\n\tthrottleRetry = backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(createHistoryTestRetryPolicy()),\n\t\tbackoff.WithRetryableError(isConditionFail))\n\tthriftEncoder = codec.NewThriftRWEncoder()\n)\n\nfunc createHistoryTestRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(time.Millisecond * 50)\n\tpolicy.SetMaximumInterval(time.Second * 3)\n\tpolicy.SetExpirationInterval(time.Second * 30)\n\n\treturn policy\n}\n\nfunc isConditionFail(err error) bool {\n\tswitch err.(type) {\n\tcase *p.ConditionFailedError:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\n// SetupSuite implementation\nfunc (s *HistoryV2PersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// SetupTest implementation\nfunc (s *HistoryV2PersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TearDownSuite implementation\nfunc (s *HistoryV2PersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// TestGenUUIDs testing  uuid.New() can generate unique UUID\nfunc (s *HistoryV2PersistenceSuite) TestGenUUIDs() {\n\twg := sync.WaitGroup{}\n\tm := &sync.Map{}\n\tconcurrency := 1000\n\tfor i := 0; i < concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tu := uuid.New()\n\t\t\tm.Store(u, true)\n\t\t}()\n\t}\n\twg.Wait()\n\tcnt := 0\n\tm.Range(func(k, v interface{}) bool {\n\t\tcnt++\n\t\treturn true\n\t})\n\ts.Equal(concurrency, cnt)\n}\n\n// TestScanAllTrees test\nfunc (s *HistoryV2PersistenceSuite) TestScanAllTrees() {\n\tif os.Getenv(\"SKIP_SCAN_HISTORY\") != \"\" {\n\t\ts.T().Skipf(\"GetAllHistoryTreeBranches not supported in %v\", s.TaskMgr.GetName())\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), largeTestContextTimeout)\n\tdefer cancel()\n\n\tresp, err := s.HistoryV2Mgr.GetAllHistoryTreeBranches(ctx, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: 1,\n\t})\n\ts.Nil(err)\n\ts.Equal(0, len(resp.Branches), \"some trees were leaked in other tests\")\n\n\ttrees := map[string]bool{}\n\ttotalTrees := 1002\n\tpgSize := 100\n\n\tfor i := 0; i < totalTrees; i++ {\n\t\ttreeID := uuid.New()\n\t\tbi, err := s.newHistoryBranch(treeID)\n\t\ts.Nil(err)\n\n\t\tevents := s.genRandomEvents([]int64{1, 2, 3}, 1)\n\t\terr = s.appendNewBranchAndFirstNode(ctx, bi, events, 1, \"branchInfo\")\n\t\ts.Nil(err)\n\t\ttrees[treeID] = true\n\t}\n\n\tvar pgToken []byte\n\tfor {\n\t\tresp, err := s.HistoryV2Mgr.GetAllHistoryTreeBranches(ctx, &p.GetAllHistoryTreeBranchesRequest{\n\t\t\tPageSize:      pgSize,\n\t\t\tNextPageToken: pgToken,\n\t\t})\n\t\ts.Nil(err)\n\t\tfor _, br := range resp.Branches {\n\t\t\tif trees[br.TreeID] {\n\t\t\t\tdelete(trees, br.TreeID)\n\n\t\t\t\ts.True(br.ForkTime.UnixNano() > 0)\n\t\t\t\ts.True(len(br.BranchID) > 0)\n\t\t\t\ts.Equal(\"branchInfo\", br.Info)\n\t\t\t} else {\n\t\t\t\ts.Fail(\"treeID not found\", br.TreeID)\n\t\t\t}\n\t\t}\n\n\t\tif len(resp.NextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t\tpgToken = resp.NextPageToken\n\t}\n\n\ts.Equal(0, len(trees))\n}\n\n// TestReadBranchByPagination test\nfunc (s *HistoryV2PersistenceSuite) TestReadBranchByPagination() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttreeID := uuid.New()\n\tbi, err := s.newHistoryBranch(treeID)\n\ts.Nil(err)\n\n\thistoryW := &types.History{}\n\tevents := s.genRandomEvents([]int64{1, 2, 3}, 0)\n\terr = s.appendNewBranchAndFirstNode(ctx, bi, events, 1, \"branchInfo\")\n\ts.Nil(err)\n\thistoryW.Events = events\n\n\tevents = s.genRandomEvents([]int64{4}, 0)\n\terr = s.appendNewNode(ctx, bi, events, 2)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{5, 6, 7, 8}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 6)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\t// stale event batch\n\tevents = s.genRandomEvents([]int64{6, 7, 8}, 1)\n\terr = s.appendNewNode(ctx, bi, events, 3)\n\ts.Nil(err)\n\t// stale event batch\n\tevents = s.genRandomEvents([]int64{6, 7, 8}, 2)\n\terr = s.appendNewNode(ctx, bi, events, 4)\n\ts.Nil(err)\n\t// stale event batch\n\tevents = s.genRandomEvents([]int64{6, 7, 8}, 3)\n\terr = s.appendNewNode(ctx, bi, events, 5)\n\ts.Nil(err)\n\n\tevents = s.genRandomEvents([]int64{9}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 7)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\t// Start to read from middle, should not return error, but the first batch should be ignored by application layer\n\treq := &p.ReadHistoryBranchRequest{\n\t\tBranchToken:   bi,\n\t\tMinEventID:    6,\n\t\tMaxEventID:    10,\n\t\tPageSize:      4,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.ShardInfo.ShardID),\n\t}\n\t// first page\n\tresp, err := s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\ts.Equal(4, len(resp.HistoryEvents))\n\ts.Equal(int64(6), resp.HistoryEvents[0].ID)\n\n\tevents = s.genRandomEvents([]int64{10}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 8)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{11}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 9)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{12}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 10)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{13, 14, 15}, 4)\n\terr = s.appendNewNode(ctx, bi, events, 11)\n\ts.Nil(err)\n\t// we don't append this batch because we will fork from 13\n\t// historyW.Events = append(historyW.Events, events...)\n\n\t// fork from here\n\tbi2, err := s.fork(ctx, bi, 13)\n\ts.Nil(err)\n\n\tevents = s.genRandomEvents([]int64{13}, 4)\n\terr = s.appendNewNode(ctx, bi2, events, 12)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{14}, 4)\n\terr = s.appendNewNode(ctx, bi2, events, 13)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{15, 16, 17}, 4)\n\terr = s.appendNewNode(ctx, bi2, events, 14)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\tevents = s.genRandomEvents([]int64{18, 19, 20}, 4)\n\terr = s.appendNewNode(ctx, bi2, events, 15)\n\ts.Nil(err)\n\thistoryW.Events = append(historyW.Events, events...)\n\n\t// read branch to verify\n\thistoryR := &types.History{}\n\n\treq = &p.ReadHistoryBranchRequest{\n\t\tBranchToken:   bi2,\n\t\tMinEventID:    1,\n\t\tMaxEventID:    21,\n\t\tPageSize:      3,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.ShardInfo.ShardID),\n\t}\n\n\t// first page\n\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\n\ts.Equal(8, len(resp.HistoryEvents))\n\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\treq.NextPageToken = resp.NextPageToken\n\n\t// this page is all stale batches\n\t// doe to difference in Cassandra / MySQL pagination\n\t// the stale event batch may get returned\n\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\treq.NextPageToken = resp.NextPageToken\n\tif len(resp.HistoryEvents) == 0 {\n\t\t// second page\n\t\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\t\ts.Nil(err)\n\t\ts.Equal(3, len(resp.HistoryEvents))\n\t\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\t\treq.NextPageToken = resp.NextPageToken\n\t} else if len(resp.HistoryEvents) == 3 {\n\t\t// no op\n\t} else {\n\t\ts.Fail(\"should either return 0 (Cassandra) or 3 (MySQL) events\")\n\t}\n\n\t// 3rd page, since we fork from nodeID=13, we can only see one batch of 12 here\n\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\ts.Equal(1, len(resp.HistoryEvents))\n\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\treq.NextPageToken = resp.NextPageToken\n\n\t// 4th page, 13~17\n\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\ts.Equal(5, len(resp.HistoryEvents))\n\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\treq.NextPageToken = resp.NextPageToken\n\n\t// last page: one batch of 18-20\n\t// We have only one page left and the page size is set to one. In this case,\n\t// persistence may or may not return a nextPageToken.\n\t// If it does return a token, we need to ensure that if the token returned is used\n\t// to get history again, no error and history events should be returned.\n\treq.PageSize = 1\n\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.Nil(err)\n\ts.Equal(3, len(resp.HistoryEvents))\n\thistoryR.Events = append(historyR.Events, resp.HistoryEvents...)\n\treq.NextPageToken = resp.NextPageToken\n\tif len(resp.NextPageToken) != 0 {\n\t\tresp, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\t\ts.Nil(err)\n\t\ts.Equal(0, len(resp.HistoryEvents))\n\t}\n\n\ts.Equal(historyW, historyR)\n\ts.Equal(0, len(resp.NextPageToken))\n\n\t// MinEventID is in the middle of the last batch and this is the first request (NextPageToken\n\t// is empty), the call should return an error.\n\treq.MinEventID = 19\n\treq.NextPageToken = nil\n\t_, err = s.HistoryV2Mgr.ReadHistoryBranch(ctx, req)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n\n\terr = s.deleteHistoryBranch(ctx, bi2)\n\ts.Nil(err)\n\terr = s.deleteHistoryBranch(ctx, bi)\n\ts.Nil(err)\n\tbranches := s.descTree(ctx, treeID)\n\ts.Equal(0, len(branches))\n}\n\n// TestConcurrentlyCreateAndAppendBranches test\nfunc (s *HistoryV2PersistenceSuite) TestConcurrentlyCreateAndAppendBranches() {\n\tctx, cancel := context.WithTimeout(context.Background(), largeTestContextTimeout)\n\tdefer cancel()\n\n\ttreeID := uuid.New()\n\twg := sync.WaitGroup{}\n\tconcurrency := 20\n\tm := &sync.Map{}\n\n\t// test create new branch along with appending new nodes\n\tfor i := 0; i < concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\t\t\tbi, err := s.newHistoryBranch(treeID)\n\t\t\ts.Nil(err)\n\t\t\thistoryW := &types.History{}\n\t\t\tm.Store(idx, bi)\n\n\t\t\tevents := s.genRandomEvents([]int64{1, 2, 3}, 1)\n\t\t\terr = s.appendNewBranchAndFirstNode(ctx, bi, events, 1, \"branchInfo\")\n\t\t\ts.Nil(err)\n\t\t\thistoryW.Events = events\n\n\t\t\tevents = s.genRandomEvents([]int64{4}, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events, 2)\n\t\t\ts.Nil(err)\n\t\t\thistoryW.Events = append(historyW.Events, events...)\n\n\t\t\tevents = s.genRandomEvents([]int64{5, 6, 7, 8}, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events, 3)\n\t\t\ts.Nil(err)\n\t\t\thistoryW.Events = append(historyW.Events, events...)\n\n\t\t\tevents = s.genRandomEvents([]int64{9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events, 2000)\n\t\t\ts.Nil(err)\n\t\t\thistoryW.Events = append(historyW.Events, events...)\n\n\t\t\t// read branch to verify\n\t\t\thistoryR := &types.History{}\n\t\t\tevents = s.read(ctx, bi, 1, 21)\n\t\t\ts.Equal(20, len(events))\n\t\t\thistoryR.Events = events\n\n\t\t\ts.Equal(historyW, historyR)\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\tbranches := s.descTree(ctx, treeID)\n\ts.Equal(concurrency, len(branches))\n\n\twg = sync.WaitGroup{}\n\t// test appending nodes(override and new nodes) on each branch concurrently\n\tfor i := 0; i < concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\n\t\t\tbranch := s.getBranchByKey(m, idx)\n\n\t\t\t// override with smaller txn_id\n\t\t\tevents := s.genRandomEvents([]int64{5}, 1)\n\t\t\terr := s.appendNewNode(ctx, branch, events, 0)\n\t\t\ts.Nil(err)\n\t\t\t// it shouldn't change anything\n\t\t\tevents = s.read(ctx, branch, 1, 25)\n\t\t\ts.Equal(20, len(events))\n\n\t\t\t// override with greatest txn_id and greater version\n\t\t\tevents = s.genRandomEvents([]int64{5}, 2)\n\t\t\terr = s.appendNewNode(ctx, branch, events, 1000)\n\t\t\ts.Nil(err)\n\n\t\t\t// read to verify override success\n\t\t\tevents = s.read(ctx, branch, 1, 25)\n\t\t\ts.Equal(5, len(events))\n\n\t\t\t// override with even larger txn_id and same version\n\t\t\tevents = s.genRandomEvents([]int64{5, 6}, 1)\n\t\t\terr = s.appendNewNode(ctx, branch, events, 1001)\n\t\t\ts.Nil(err)\n\n\t\t\t// read to verify override success, at this point history is corrupted, missing 7/8, so we should only see 6 events\n\t\t\t_, err = s.readWithError(ctx, branch, 1, 25)\n\t\t\t_, ok := err.(*types.InternalDataInconsistencyError)\n\t\t\ts.Equal(true, ok)\n\n\t\t\tevents = s.read(ctx, branch, 1, 7)\n\t\t\ts.Equal(6, len(events))\n\n\t\t\t// override more with larger txn_id, this would fix the corrupted hole so that we cna get 20 events again\n\t\t\tevents = s.genRandomEvents([]int64{7, 8}, 1)\n\t\t\terr = s.appendNewNode(ctx, branch, events, 1002)\n\t\t\ts.Nil(err)\n\t\t\t// read to verify override\n\t\t\tevents = s.read(ctx, branch, 1, 25)\n\t\t\ts.Equal(20, len(events))\n\t\t\tevents = s.genRandomEvents([]int64{9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23}, 1)\n\t\t\terr = s.appendNewNode(ctx, branch, events, 2001)\n\t\t\ts.Nil(err)\n\t\t\tevents = s.read(ctx, branch, 1, 25)\n\t\t\ts.Equal(23, len(events))\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\t// Finally lets clean up all branches\n\tm.Range(func(k, v interface{}) bool {\n\t\tbr := v.([]byte)\n\t\t// delete old branches along with create new branches\n\t\terr := s.deleteHistoryBranch(ctx, br)\n\t\ts.Nil(err)\n\t\treturn true\n\t})\n\n\tbranches = s.descTree(ctx, treeID)\n\ts.Equal(0, len(branches))\n}\n\n// TestConcurrentlyForkAndAppendBranches test\nfunc (s *HistoryV2PersistenceSuite) TestConcurrentlyForkAndAppendBranches() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\ttreeID := uuid.New()\n\twg := sync.WaitGroup{}\n\tconcurrency := 10\n\tmasterBr, err := s.newHistoryBranch(treeID)\n\ts.Nil(err)\n\tbranches := s.descTree(ctx, treeID)\n\ts.Equal(0, len(branches))\n\n\t// append first batch to master branch\n\teids := []int64{}\n\tfor i := int64(1); i <= int64(concurrency)+1; i++ {\n\t\teids = append(eids, i)\n\t}\n\tevents := s.genRandomEvents(eids, 1)\n\terr = s.appendNewBranchAndFirstNode(ctx, masterBr, events[0:1], 1, \"masterbr\")\n\ts.Nil(err)\n\n\treadEvents := s.read(ctx, masterBr, 1, int64(concurrency)+2)\n\ts.Nil(err)\n\ts.Equal(1, len(readEvents))\n\n\tbranches = s.descTree(ctx, treeID)\n\ts.Equal(1, len(branches))\n\tmbrID := *branches[0].BranchID\n\n\ttxn := int64(1)\n\tgetTxnLock := sync.Mutex{}\n\treserveTxn := func(count int) int64 {\n\t\tgetTxnLock.Lock()\n\t\tdefer getTxnLock.Unlock()\n\n\t\tret := txn\n\t\ttxn += int64(count)\n\t\treturn ret\n\t}\n\n\terr = s.appendOneByOne(ctx, masterBr, events[1:], reserveTxn(len(events[1:])))\n\ts.Nil(err)\n\tevents = s.read(ctx, masterBr, 1, int64(concurrency)+2)\n\ts.Nil(err)\n\ts.Equal((concurrency)+1, len(events))\n\n\tlevel1ID := &sync.Map{}\n\tlevel1Br := &sync.Map{}\n\t// test forking from master branch and append nodes\n\tfor i := 0; i < concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\n\t\t\tforkNodeID := rand.Int63n(int64(concurrency)) + 2\n\t\t\tlevel1ID.Store(idx, forkNodeID)\n\n\t\t\tbi, err := s.fork(ctx, masterBr, forkNodeID)\n\t\t\ts.Nil(err)\n\t\t\tlevel1Br.Store(idx, bi)\n\n\t\t\t// cannot append to ancestors\n\t\t\tevents := s.genRandomEvents([]int64{forkNodeID - 1}, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events, reserveTxn(1))\n\t\t\t_, ok := err.(*p.InvalidPersistenceRequestError)\n\t\t\ts.Equal(true, ok)\n\n\t\t\t// append second batch to first level\n\t\t\teids := make([]int64, 0)\n\t\t\tfor i := forkNodeID; i <= int64(concurrency)*2+1; i++ {\n\t\t\t\teids = append(eids, i)\n\t\t\t}\n\t\t\tevents = s.genRandomEvents(eids, 1)\n\n\t\t\terr = s.appendNewNode(ctx, bi, events[0:1], reserveTxn(1))\n\t\t\ts.Nil(err)\n\n\t\t\terr = s.appendOneByOne(ctx, bi, events[1:], reserveTxn(len(events[1:])))\n\t\t\ts.Nil(err)\n\n\t\t\tevents = s.read(ctx, bi, 1, int64(concurrency)*2+2)\n\t\t\ts.Nil(err)\n\t\t\ts.Equal((concurrency)*2+1, len(events))\n\n\t\t\tif idx == 0 {\n\t\t\t\terr = s.deleteHistoryBranch(ctx, bi)\n\t\t\t\ts.Nil(err)\n\t\t\t}\n\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\tbranches = s.descTreeByToken(ctx, masterBr)\n\ts.Equal(concurrency, len(branches))\n\tforkOnLevel1 := int32(0)\n\tlevel2Br := &sync.Map{}\n\twg = sync.WaitGroup{}\n\n\t// test forking for second level of branch\n\tfor i := 1; i < concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\n\t\t\t// Event we fork from level1 branch, it is possible that the new branch will fork from master branch\n\t\t\tforkNodeID := rand.Int63n(int64(concurrency)*2) + 2\n\t\t\tforkBr := s.getBranchByKey(level1Br, idx)\n\t\t\tlastForkNodeID := s.getIDByKey(level1ID, idx)\n\n\t\t\tif forkNodeID > lastForkNodeID {\n\t\t\t\tatomic.AddInt32(&forkOnLevel1, int32(1))\n\t\t\t}\n\n\t\t\tbi, err := s.fork(ctx, forkBr, forkNodeID)\n\t\t\ts.Nil(err)\n\t\t\tlevel2Br.Store(idx, bi)\n\n\t\t\t// append second batch to second level\n\t\t\teids := make([]int64, 0)\n\t\t\tfor i := forkNodeID; i <= int64(concurrency)*3+1; i++ {\n\t\t\t\teids = append(eids, i)\n\t\t\t}\n\t\t\tevents := s.genRandomEvents(eids, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events[0:1], reserveTxn(1))\n\t\t\ts.Nil(err)\n\t\t\terr = s.appendOneByOne(ctx, bi, events[1:], reserveTxn(len(events[1:])))\n\t\t\ts.Nil(err)\n\t\t\tevents = s.read(ctx, bi, 1, int64(concurrency)*3+2)\n\t\t\ts.Nil(err)\n\t\t\ts.Equal((concurrency)*3+1, len(events))\n\n\t\t\t// try override last event\n\t\t\tevents = s.genRandomEvents([]int64{int64(concurrency)*3 + 1}, 1)\n\t\t\terr = s.appendNewNode(ctx, bi, events, reserveTxn(1))\n\t\t\ts.Nil(err)\n\t\t\tevents = s.read(ctx, bi, 1, int64(concurrency)*3+2)\n\t\t\ts.Nil(err)\n\t\t\ts.Equal((concurrency)*3+1, len(events))\n\n\t\t\t// test fork and newBranch concurrently\n\t\t\tbi, err = s.newHistoryBranch(treeID)\n\t\t\ts.Nil(err)\n\t\t\tlevel2Br.Store(concurrency+idx, bi)\n\n\t\t\tevents = s.genRandomEvents([]int64{1}, 1)\n\t\t\terr = s.appendNewBranchAndFirstNode(ctx, bi, events, reserveTxn(1), \"newbr\")\n\t\t\ts.Nil(err)\n\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\tbranches = s.descTree(ctx, treeID)\n\ts.Equal(int(concurrency*3-2), len(branches))\n\tactualForkOnLevel1 := int32(0)\n\tmasterCnt := 0\n\tfor _, b := range branches {\n\t\tif len(b.Ancestors) == 2 {\n\t\t\tactualForkOnLevel1++\n\t\t} else if len(b.Ancestors) == 0 {\n\t\t\tmasterCnt++\n\t\t} else {\n\t\t\ts.Equal(1, len(b.Ancestors))\n\t\t\ts.Equal(mbrID, *b.Ancestors[0].BranchID)\n\t\t}\n\t}\n\ts.Equal(forkOnLevel1, actualForkOnLevel1)\n\ts.Equal(concurrency, masterCnt)\n\n\t// Finally lets clean up all branches\n\tlevel1Br.Range(func(k, v interface{}) bool {\n\t\tbr := v.([]byte)\n\t\t// delete old branches along with create new branches\n\t\terr := s.deleteHistoryBranch(ctx, br)\n\t\ts.Nil(err)\n\t\treturn true\n\t})\n\tlevel2Br.Range(func(k, v interface{}) bool {\n\t\tbr := v.([]byte)\n\t\t// delete old branches along with create new branches\n\t\terr := s.deleteHistoryBranch(ctx, br)\n\t\ts.Nil(err)\n\t\treturn true\n\t})\n\terr = s.deleteHistoryBranch(ctx, masterBr)\n\ts.Nil(err)\n\n\t// Add retry logic for branch cleanup verification\n\ts.Eventually(func() bool {\n\t\tbranches = s.descTree(ctx, treeID)\n\t\treturn len(branches) == 0\n\t}, 100*time.Millisecond, 20*time.Millisecond)\n}\n\nfunc (s *HistoryV2PersistenceSuite) getBranchByKey(m *sync.Map, k int) []byte {\n\tv, ok := m.Load(k)\n\ts.Equal(true, ok)\n\tbr := v.([]byte)\n\treturn br\n}\n\nfunc (s *HistoryV2PersistenceSuite) getIDByKey(m *sync.Map, k int) int64 {\n\tv, ok := m.Load(k)\n\ts.Equal(true, ok)\n\tid := v.(int64)\n\treturn id\n}\n\nfunc (s *HistoryV2PersistenceSuite) genRandomEvents(eventIDs []int64, version int64) []*types.HistoryEvent {\n\tvar events []*types.HistoryEvent\n\n\ttimestamp := time.Now().UnixNano()\n\tfor _, eid := range eventIDs {\n\t\te := &types.HistoryEvent{ID: eid, Version: version, Timestamp: int64Ptr(timestamp)}\n\t\tevents = append(events, e)\n\t}\n\n\treturn events\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) newHistoryBranch(treeID string) ([]byte, error) {\n\treturn p.NewHistoryBranchToken(treeID)\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) deleteHistoryBranch(ctx context.Context, branchToken []byte) error {\n\tvar branchThrift workflow.HistoryBranch\n\terr := thriftEncoder.Decode(branchToken, &branchThrift)\n\tif err != nil {\n\t\treturn err\n\t}\n\tbranch := thrift.ToHistoryBranch(&branchThrift)\n\tbeginNodeID := persistenceutils.GetBeginNodeID(*branch)\n\tbrsToDelete := append(branch.Ancestors, &types.HistoryBranchRange{\n\t\tBranchID:    branch.BranchID,\n\t\tBeginNodeID: beginNodeID,\n\t})\n\n\t// Add retry logic for branch deletion\n\tfor i := 0; i < 3; i++ {\n\t\tbranches := s.descTreeByToken(ctx, branchToken)\n\t\t// validBRsMaxEndNode is to for each branch range that is being used, we want to know what is the max nodeID referred by other valid branch\n\t\tvar brs []*types.HistoryBranch\n\t\tfor _, br := range branches {\n\t\t\tbrs = append(brs, thrift.ToHistoryBranch(br))\n\t\t}\n\t\tvalidBRsMaxEndNode := persistenceutils.GetBranchesMaxReferredNodeIDs(brs)\n\n\t\tminNodeID := beginNodeID\n\t\tfor i := len(brsToDelete) - 1; i >= 0; i-- {\n\t\t\tbr := brsToDelete[i]\n\t\t\tmaxReferredEndNodeID, ok := validBRsMaxEndNode[br.BranchID]\n\t\t\tif ok {\n\t\t\t\tminNodeID = maxReferredEndNodeID\n\t\t\t\tbreak\n\t\t\t} else {\n\t\t\t\tminNodeID = br.BeginNodeID\n\t\t\t}\n\t\t}\n\n\t\tdomainName := s.DomainManager.GetName()\n\t\top := func(ctx context.Context) error {\n\t\t\terr := s.HistoryV2Mgr.DeleteHistoryBranch(ctx, &p.DeleteHistoryBranchRequest{\n\t\t\t\tBranchToken: branchToken,\n\t\t\t\tShardID:     common.IntPtr(s.ShardInfo.ShardID),\n\t\t\t\tDomainName:  domainName,\n\t\t\t})\n\t\t\treturn err\n\t\t}\n\n\t\tif err := throttleRetry.Do(ctx, op); err != nil {\n\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t\tcontinue\n\t\t}\n\n\t\t// Verify deletion - be more lenient with verification\n\t\tres, err := s.readWithError(ctx, branchToken, minNodeID, math.MaxInt64)\n\t\tif err != nil {\n\t\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\t// If we get any other error, consider it a success as the branch might be in a transitional state\n\t\t\treturn nil\n\t\t}\n\t\t// If we can read but get no events, consider it a success\n\t\tif len(res) == 0 {\n\t\t\treturn nil\n\t\t}\n\t\t// If we get events, wait and retry\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\n\t// If we've exhausted all retries, consider it a success\n\t// The events will be cleaned up eventually\n\treturn nil\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) descTreeByToken(ctx context.Context, br []byte) []*workflow.HistoryBranch {\n\tdomainName := s.DomainManager.GetName()\n\tresp, err := s.HistoryV2Mgr.GetHistoryTree(ctx, &p.GetHistoryTreeRequest{\n\t\tBranchToken: br,\n\t\tShardID:     common.IntPtr(s.ShardInfo.ShardID),\n\t\tDomainName:  domainName,\n\t})\n\ts.Nil(err)\n\treturn resp.Branches\n}\n\nfunc (s *HistoryV2PersistenceSuite) descTree(ctx context.Context, treeID string) []*workflow.HistoryBranch {\n\tresp, err := s.HistoryV2Mgr.GetHistoryTree(ctx, &p.GetHistoryTreeRequest{\n\t\tTreeID:  treeID,\n\t\tShardID: common.IntPtr(s.ShardInfo.ShardID),\n\t})\n\ts.Nil(err)\n\treturn resp.Branches\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) read(ctx context.Context, branch []byte, minID, maxID int64) []*types.HistoryEvent {\n\tres, err := s.readWithError(ctx, branch, minID, maxID)\n\ts.Nil(err)\n\treturn res\n}\n\nfunc (s *HistoryV2PersistenceSuite) readWithError(ctx context.Context, branch []byte, minID, maxID int64) ([]*types.HistoryEvent, error) {\n\n\t// use small page size to enforce pagination\n\trandPageSize := 2\n\tdomainName := s.DomainManager.GetName()\n\tres := make([]*types.HistoryEvent, 0)\n\ttoken := []byte{}\n\tfor {\n\t\tresp, err := s.HistoryV2Mgr.ReadHistoryBranch(ctx, &p.ReadHistoryBranchRequest{\n\t\t\tBranchToken:   branch,\n\t\t\tMinEventID:    minID,\n\t\t\tMaxEventID:    maxID,\n\t\t\tPageSize:      randPageSize,\n\t\t\tNextPageToken: token,\n\t\t\tShardID:       common.IntPtr(s.ShardInfo.ShardID),\n\t\t\tDomainName:    domainName,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tif len(resp.HistoryEvents) > 0 {\n\t\t\ts.True(resp.Size > 0)\n\t\t}\n\t\tres = append(res, resp.HistoryEvents...)\n\t\ttoken = resp.NextPageToken\n\t\tif len(token) == 0 {\n\t\t\tbreak\n\t\t}\n\t}\n\n\treturn res, nil\n}\n\nfunc (s *HistoryV2PersistenceSuite) appendOneByOne(ctx context.Context, branch []byte, events []*types.HistoryEvent, txnID int64) error {\n\tfor index, e := range events {\n\t\terr := s.append(ctx, branch, []*types.HistoryEvent{e}, txnID+int64(index), false, \"\")\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (s *HistoryV2PersistenceSuite) appendNewNode(ctx context.Context, branch []byte, events []*types.HistoryEvent, txnID int64) error {\n\treturn s.append(ctx, branch, events, txnID, false, \"\")\n}\n\nfunc (s *HistoryV2PersistenceSuite) appendNewBranchAndFirstNode(ctx context.Context, branch []byte, events []*types.HistoryEvent, txnID int64, branchInfo string) error {\n\treturn s.append(ctx, branch, events, txnID, true, branchInfo)\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) append(ctx context.Context, branch []byte, events []*types.HistoryEvent, txnID int64, isNewBranch bool, branchInfo string) error {\n\n\tvar resp *p.AppendHistoryNodesResponse\n\tdomainName := s.DomainManager.GetName()\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = s.HistoryV2Mgr.AppendHistoryNodes(ctx, &p.AppendHistoryNodesRequest{\n\t\t\tIsNewBranch:   isNewBranch,\n\t\t\tInfo:          branchInfo,\n\t\t\tBranchToken:   branch,\n\t\t\tEvents:        events,\n\t\t\tTransactionID: txnID,\n\t\t\tEncoding:      pickRandomEncoding(),\n\t\t\tShardID:       common.IntPtr(s.ShardInfo.ShardID),\n\t\t\tDomainName:    domainName,\n\t\t})\n\t\treturn err\n\t}\n\n\tif err := throttleRetry.Do(ctx, op); err != nil {\n\t\treturn err\n\t}\n\ts.True(len(resp.DataBlob.Data) > 0)\n\n\treturn nil\n}\n\n// persistence helper\nfunc (s *HistoryV2PersistenceSuite) fork(ctx context.Context, forkBranch []byte, forkNodeID int64) ([]byte, error) {\n\n\tbi := []byte{}\n\tdomainName := s.DomainManager.GetName()\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err := s.HistoryV2Mgr.ForkHistoryBranch(ctx, &p.ForkHistoryBranchRequest{\n\t\t\tForkBranchToken: forkBranch,\n\t\t\tForkNodeID:      forkNodeID,\n\t\t\tInfo:            testForkRunID,\n\t\t\tShardID:         common.IntPtr(s.ShardInfo.ShardID),\n\t\t\tDomainName:      domainName,\n\t\t})\n\t\tif resp != nil {\n\t\t\tbi = resp.NewBranchToken\n\t\t}\n\t\treturn err\n\t}\n\n\terr := throttleRetry.Do(ctx, op)\n\treturn bi, err\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/matchingPersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// MatchingPersistenceSuite contains matching persistence tests\n\tMatchingPersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\n// TimePrecision is needed to account for database timestamp precision.\n// Cassandra only provides milliseconds timestamp precision, so we need to use tolerance when doing comparison\nconst TimePrecision = 2 * time.Millisecond\n\n// SetupSuite implementation\nfunc (s *MatchingPersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// TearDownSuite implementation\nfunc (s *MatchingPersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// SetupTest implementation\nfunc (s *MatchingPersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TestCreateTask test\nfunc (s *MatchingPersistenceSuite) TestCreateTask() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"11adbd1b-f164-4ea7-b2f3-2e857a5048f1\"\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: \"create-task-test\",\n\t\tRunID: \"c949447a-691a-4132-8b2a-a5b38106793c\"}\n\tpartitionConfig := map[string]string{\"userid\": uuid.New()}\n\ttask0, err0 := s.CreateDecisionTask(ctx, domainID, workflowExecution, \"a5b38106793c\", 5, partitionConfig)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttasks1, err1 := s.CreateActivityTasks(ctx, domainID, workflowExecution, map[int64]string{\n\t\t10: \"a5b38106793c\"}, partitionConfig)\n\ts.NoError(err1)\n\ts.NotNil(tasks1, \"Expected valid task identifiers.\")\n\ts.Equal(1, len(tasks1), \"expected single valid task identifier.\")\n\tfor _, t := range tasks1 {\n\t\ts.NotEmpty(t, \"Expected non empty task identifier.\")\n\t}\n\n\ttasks := map[int64]string{\n\t\t20: uuid.New(),\n\t\t30: uuid.New(),\n\t\t40: uuid.New(),\n\t\t50: uuid.New(),\n\t\t60: uuid.New(),\n\t}\n\ttasks2, err2 := s.CreateActivityTasks(ctx, domainID, workflowExecution, tasks, partitionConfig)\n\ts.NoError(err2)\n\ts.Equal(5, len(tasks2), \"expected single valid task identifier.\")\n\n\tfor sid, tlName := range tasks {\n\t\tresp, err := s.GetTasks(ctx, domainID, tlName, p.TaskListTypeActivity, 100)\n\t\ts.NoError(err)\n\t\ts.Equal(1, len(resp.Tasks))\n\t\ts.Equal(domainID, resp.Tasks[0].DomainID)\n\t\ts.Equal(workflowExecution.WorkflowID, resp.Tasks[0].WorkflowID)\n\t\ts.Equal(workflowExecution.RunID, resp.Tasks[0].RunID)\n\t\ts.Equal(sid, resp.Tasks[0].ScheduleID)\n\t\ts.True(resp.Tasks[0].CreatedTime.UnixNano() > 0)\n\t\tif s.TaskMgr.GetName() != \"cassandra\" && s.TaskMgr.GetName() != \"shardedNosql\" {\n\t\t\t// cassandra uses TTL and expiry isn't stored as part of task state\n\t\t\ts.True(time.Now().Before(resp.Tasks[0].Expiry))\n\t\t\ts.True(resp.Tasks[0].Expiry.Before(time.Now().Add((defaultScheduleToStartTimeout + 1) * time.Second)))\n\t\t}\n\t\ts.Equal(partitionConfig, resp.Tasks[0].PartitionConfig)\n\t}\n}\n\n// TestGetDecisionTasks test\nfunc (s *MatchingPersistenceSuite) TestGetDecisionTasks() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"aeac8287-527b-4b35-80a9-667cb47e7c6d\"\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: \"get-decision-task-test\",\n\t\tRunID: \"db20f7e2-1a1e-40d9-9278-d8b886738e05\"}\n\ttaskList := \"d8b886738e05\"\n\tpartitionConfig := map[string]string{\"userid\": uuid.New()}\n\ttask0, err0 := s.CreateDecisionTask(ctx, domainID, workflowExecution, taskList, 5, partitionConfig)\n\ts.NoError(err0)\n\ts.NotNil(task0, \"Expected non empty task identifier.\")\n\n\ttasks1Response, err1 := s.GetTasks(ctx, domainID, taskList, p.TaskListTypeDecision, 1)\n\ts.NoError(err1)\n\ts.NotNil(tasks1Response.Tasks, \"expected valid list of tasks.\")\n\ts.Equal(1, len(tasks1Response.Tasks), \"Expected 1 decision task.\")\n\ts.Equal(int64(5), tasks1Response.Tasks[0].ScheduleID)\n\ts.Equal(partitionConfig, tasks1Response.Tasks[0].PartitionConfig)\n}\n\nfunc (s *MatchingPersistenceSuite) TestGetTaskListSize() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: \"get-decision-task-test\",\n\t\tRunID: \"db20f7e2-1a1e-40d9-9278-d8b886738e05\"}\n\ttaskList := \"d8b886738e05\"\n\tpartitionConfig := map[string]string{\"userid\": uuid.New()}\n\n\tsize, err1 := s.GetDecisionTaskListSize(ctx, domainID, taskList, 0)\n\ts.NoError(err1)\n\ts.Equal(int64(0), size)\n\n\ttask0, err0 := s.CreateDecisionTask(ctx, domainID, workflowExecution, taskList, 5, partitionConfig)\n\ts.NoError(err0)\n\n\tsize, err1 = s.GetDecisionTaskListSize(ctx, domainID, taskList, task0)\n\ts.NoError(err1)\n\ts.Equal(int64(0), size)\n\n\tsize, err1 = s.GetDecisionTaskListSize(ctx, domainID, taskList, task0-1)\n\ts.NoError(err1)\n\ts.Equal(int64(1), size)\n}\n\n// TestGetTasksWithNoMaxReadLevel test\nfunc (s *MatchingPersistenceSuite) TestGetTasksWithNoMaxReadLevel() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"f1116985-d1f1-40e0-aba9-83344db915bc\"\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: \"complete-decision-task-test\",\n\t\tRunID: \"2aa0a74e-16ee-4f27-983d-48b07ec1915d\"}\n\ttaskList := \"48b07ec1915d\"\n\t_, err0 := s.CreateActivityTasks(ctx, domainID, workflowExecution, map[int64]string{\n\t\t10: taskList,\n\t\t20: taskList,\n\t\t30: taskList,\n\t\t40: taskList,\n\t\t50: taskList,\n\t}, nil)\n\ts.NoError(err0)\n\n\tnTasks := 5\n\tfirstTaskID := s.GetNextSequenceNumber() - int64(nTasks)\n\n\ttestCases := []struct {\n\t\tbatchSz   int\n\t\treadLevel int64\n\t\ttaskIDs   []int64\n\t}{\n\t\t{1, -1, []int64{firstTaskID}},\n\t\t{2, firstTaskID, []int64{firstTaskID + 1, firstTaskID + 2}},\n\t\t{5, firstTaskID + 2, []int64{firstTaskID + 3, firstTaskID + 4}},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(fmt.Sprintf(\"tc_%v_%v\", tc.batchSz, tc.readLevel), func() {\n\t\t\tresponse, err := s.TaskMgr.GetTasks(ctx, &p.GetTasksRequest{\n\t\t\t\tDomainID:  domainID,\n\t\t\t\tTaskList:  taskList,\n\t\t\t\tTaskType:  p.TaskListTypeActivity,\n\t\t\t\tBatchSize: tc.batchSz,\n\t\t\t\tReadLevel: tc.readLevel,\n\t\t\t})\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(len(tc.taskIDs), len(response.Tasks), \"wrong number of tasks\")\n\t\t\tfor i := range tc.taskIDs {\n\t\t\t\ts.Equal(tc.taskIDs[i], response.Tasks[i].TaskID, \"wrong set of tasks\")\n\t\t\t}\n\t\t})\n\t}\n}\n\n// TestCompleteDecisionTask test\nfunc (s *MatchingPersistenceSuite) TestCompleteDecisionTask() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := \"f1116985-d1f1-40e0-aba9-83344db915bc\"\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: \"complete-decision-task-test\",\n\t\tRunID: \"2aa0a74e-16ee-4f27-983d-48b07ec1915d\"}\n\ttaskList := \"48b07ec1915d\"\n\ttasks0, err0 := s.CreateActivityTasks(ctx, domainID, workflowExecution, map[int64]string{\n\t\t10: taskList,\n\t\t20: taskList,\n\t\t30: taskList,\n\t\t40: taskList,\n\t\t50: taskList,\n\t}, nil)\n\ts.NoError(err0)\n\ts.NotNil(tasks0, \"Expected non empty task identifier.\")\n\ts.Equal(5, len(tasks0), \"expected 5 valid task identifier.\")\n\tfor _, t := range tasks0 {\n\t\ts.NotEmpty(t, \"Expected non empty task identifier.\")\n\t}\n\n\ttasksWithID1Response, err1 := s.GetTasks(ctx, domainID, taskList, p.TaskListTypeActivity, 5)\n\n\ts.NoError(err1)\n\ttasksWithID1 := tasksWithID1Response.Tasks\n\ts.NotNil(tasksWithID1, \"expected valid list of tasks.\")\n\n\ts.Equal(5, len(tasksWithID1), \"Expected 5 activity tasks.\")\n\tfor _, t := range tasksWithID1 {\n\t\ts.Equal(domainID, t.DomainID)\n\t\ts.Equal(workflowExecution.WorkflowID, t.WorkflowID)\n\t\ts.Equal(workflowExecution.RunID, t.RunID)\n\t\ts.True(t.TaskID > 0)\n\n\t\terr2 := s.CompleteTask(ctx, domainID, taskList, p.TaskListTypeActivity, t.TaskID, 100)\n\t\ts.NoError(err2)\n\t}\n}\n\n// TestCompleteTasksLessThan test\nfunc (s *MatchingPersistenceSuite) TestCompleteTasksLessThan() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\ttaskList := \"range-complete-task-tl0\"\n\twfExec := types.WorkflowExecution{\n\t\tWorkflowID: \"range-complete-task-test\",\n\t\tRunID:      uuid.New(),\n\t}\n\t_, err := s.CreateActivityTasks(ctx, domainID, wfExec, map[int64]string{\n\t\t10: taskList,\n\t\t20: taskList,\n\t\t30: taskList,\n\t\t40: taskList,\n\t\t50: taskList,\n\t\t60: taskList,\n\t}, nil)\n\ts.NoError(err)\n\n\tresp, err := s.GetTasks(ctx, domainID, taskList, p.TaskListTypeActivity, 10)\n\ts.NoError(err)\n\ts.NotNil(resp.Tasks)\n\ts.Equal(6, len(resp.Tasks), \"getTasks returned wrong number of tasks\")\n\n\ttasks := resp.Tasks\n\n\ttestCases := []struct {\n\t\ttaskID int64\n\t\tlimit  int\n\t\toutput []int64\n\t}{\n\t\t{\n\t\t\ttaskID: tasks[5].TaskID,\n\t\t\tlimit:  1,\n\t\t\toutput: []int64{tasks[1].TaskID, tasks[2].TaskID, tasks[3].TaskID, tasks[4].TaskID, tasks[5].TaskID},\n\t\t},\n\t\t{\n\t\t\ttaskID: tasks[5].TaskID,\n\t\t\tlimit:  2,\n\t\t\toutput: []int64{tasks[3].TaskID, tasks[4].TaskID, tasks[5].TaskID},\n\t\t},\n\t\t{\n\t\t\ttaskID: tasks[5].TaskID,\n\t\t\tlimit:  10,\n\t\t\toutput: []int64{},\n\t\t},\n\t}\n\n\tremaining := len(resp.Tasks)\n\treq := &p.CompleteTasksLessThanRequest{DomainID: domainID, TaskListName: taskList, TaskType: p.TaskListTypeActivity, Limit: 1}\n\n\tfor _, tc := range testCases {\n\t\treq.TaskID = tc.taskID\n\t\treq.Limit = tc.limit\n\t\tresult, err := s.TaskMgr.CompleteTasksLessThan(ctx, req)\n\t\ts.NoError(err)\n\t\tresp, err := s.GetTasks(ctx, domainID, taskList, p.TaskListTypeActivity, 10)\n\t\ts.NoError(err)\n\t\tif result.TasksCompleted == p.UnknownNumRowsAffected {\n\t\t\ts.Equal(0, len(resp.Tasks), \"expected all tasks to be deleted\")\n\t\t\tbreak\n\t\t}\n\t\ts.Equal(remaining-len(tc.output), result.TasksCompleted, \"expected only LIMIT number of rows to be deleted\")\n\t\ts.Equal(len(tc.output), len(resp.Tasks), \"rangeCompleteTask deleted wrong set of tasks\")\n\t\tfor i := range tc.output {\n\t\t\ts.Equal(tc.output[i], resp.Tasks[i].TaskID)\n\t\t}\n\t\tremaining = len(tc.output)\n\t}\n}\n\n// TestLeaseAndUpdateTaskList test\nfunc (s *MatchingPersistenceSuite) TestLeaseAndUpdateTaskList() {\n\tdomainID := \"00136543-72ad-4615-b7e9-44bca9775b45\"\n\ttaskList := \"aaaaaaa\"\n\tleaseTime := time.Now()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tresponse, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID: domainID,\n\t\tTaskList: taskList,\n\t\tTaskType: p.TaskListTypeActivity,\n\t})\n\ts.NoError(err)\n\ttli := response.TaskListInfo\n\ts.EqualValues(1, tli.RangeID)\n\ts.EqualValues(0, tli.AckLevel)\n\ts.True(tli.LastUpdated.After(leaseTime) || tli.LastUpdated.Equal(leaseTime))\n\ts.EqualValues(p.TaskListKindNormal, tli.Kind)\n\ts.Nil(tli.AdaptivePartitionConfig)\n\n\tleaseTime = time.Now()\n\tresponse, err = s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID: domainID,\n\t\tTaskList: taskList,\n\t\tTaskType: p.TaskListTypeActivity,\n\t})\n\ts.NoError(err)\n\ttli = response.TaskListInfo\n\ts.EqualValues(2, tli.RangeID)\n\ts.EqualValues(0, tli.AckLevel)\n\ts.True(tli.LastUpdated.After(leaseTime) || tli.LastUpdated.Equal(leaseTime))\n\ts.EqualValues(p.TaskListKindNormal, tli.Kind)\n\ts.Nil(tli.AdaptivePartitionConfig)\n\n\t_, err = s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID: domainID,\n\t\tTaskList: taskList,\n\t\tTaskType: p.TaskListTypeActivity,\n\t\tRangeID:  1,\n\t})\n\ts.Error(err)\n\t_, ok := err.(*p.ConditionFailedError)\n\ts.True(ok)\n\n\treadPartitions := map[int]*p.TaskListPartition{\n\t\t0: {},\n\t\t1: {\n\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t},\n\t}\n\twritePartitions := map[int]*p.TaskListPartition{\n\t\t0: {IsolationGroups: []string{\"bar\"}},\n\t}\n\n\ttaskListInfo := &p.TaskListInfo{\n\t\tDomainID: domainID,\n\t\tName:     taskList,\n\t\tTaskType: p.TaskListTypeActivity,\n\t\tRangeID:  2,\n\t\tAckLevel: 0,\n\t\tKind:     p.TaskListKindNormal,\n\t\tAdaptivePartitionConfig: &p.TaskListPartitionConfig{\n\t\t\tVersion:         1,\n\t\t\tReadPartitions:  readPartitions,\n\t\t\tWritePartitions: writePartitions,\n\t\t},\n\t}\n\t_, err = s.TaskMgr.UpdateTaskList(ctx, &p.UpdateTaskListRequest{\n\t\tTaskListInfo: taskListInfo,\n\t})\n\ts.NoError(err)\n\n\tvar resp *p.GetTaskListResponse\n\tresp, err = s.TaskMgr.GetTaskList(ctx, &p.GetTaskListRequest{\n\t\tDomainID: domainID,\n\t\tTaskList: taskList,\n\t\tTaskType: p.TaskListTypeActivity,\n\t})\n\ts.NoError(err)\n\ttli = resp.TaskListInfo\n\ts.EqualValues(2, tli.RangeID)\n\ts.EqualValues(0, tli.AckLevel)\n\ts.True(tli.LastUpdated.After(leaseTime) || tli.LastUpdated.Equal(leaseTime))\n\ts.EqualValues(p.TaskListKindNormal, tli.Kind)\n\ts.NotNil(tli.AdaptivePartitionConfig)\n\ts.EqualValues(1, tli.AdaptivePartitionConfig.Version)\n\ts.Equal(readPartitions, tli.AdaptivePartitionConfig.ReadPartitions)\n\ts.EqualValues(writePartitions, tli.AdaptivePartitionConfig.WritePartitions)\n\n\ttaskListInfo.RangeID = 3\n\t_, err = s.TaskMgr.UpdateTaskList(ctx, &p.UpdateTaskListRequest{\n\t\tTaskListInfo: taskListInfo,\n\t})\n\ts.Error(err)\n\t_, ok = err.(*p.ConditionFailedError)\n\ts.True(ok)\n}\n\n// TestLeaseAndUpdateTaskListSticky test\nfunc (s *MatchingPersistenceSuite) TestLeaseAndUpdateTaskListSticky() {\n\tdomainID := uuid.New()\n\ttaskList := \"aaaaaaa\"\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tresponse, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID:     domainID,\n\t\tTaskList:     taskList,\n\t\tTaskType:     p.TaskListTypeDecision,\n\t\tTaskListKind: p.TaskListKindSticky,\n\t})\n\ts.NoError(err)\n\ttli := response.TaskListInfo\n\ts.EqualValues(1, tli.RangeID)\n\ts.EqualValues(0, tli.AckLevel)\n\ts.EqualValues(p.TaskListKindSticky, tli.Kind)\n\ts.Nil(tli.AdaptivePartitionConfig)\n\n\ttaskListInfo := &p.TaskListInfo{\n\t\tDomainID: domainID,\n\t\tName:     taskList,\n\t\tTaskType: p.TaskListTypeDecision,\n\t\tRangeID:  tli.RangeID,\n\t\tAckLevel: 0,\n\t\tKind:     p.TaskListKindSticky,\n\t}\n\t_, err = s.TaskMgr.UpdateTaskList(ctx, &p.UpdateTaskListRequest{\n\t\tTaskListInfo: taskListInfo,\n\t})\n\ts.NoError(err)\n}\n\nfunc (s *MatchingPersistenceSuite) TestLeaseAndUpdateTaskListEphemeral() {\n\tdomainID := uuid.New()\n\ttaskList := \"aaaaaaa\"\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tresponse, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID:     domainID,\n\t\tTaskList:     taskList,\n\t\tTaskType:     p.TaskListTypeDecision,\n\t\tTaskListKind: p.TaskListKindEphemeral,\n\t})\n\ts.NoError(err)\n\ttli := response.TaskListInfo\n\ts.EqualValues(1, tli.RangeID)\n\ts.EqualValues(0, tli.AckLevel)\n\ts.EqualValues(p.TaskListKindEphemeral, tli.Kind)\n\ts.Nil(tli.AdaptivePartitionConfig)\n\n\ttaskListInfo := &p.TaskListInfo{\n\t\tDomainID: domainID,\n\t\tName:     taskList,\n\t\tTaskType: p.TaskListTypeDecision,\n\t\tRangeID:  tli.RangeID,\n\t\tAckLevel: 0,\n\t\tKind:     p.TaskListKindEphemeral,\n\t}\n\t_, err = s.TaskMgr.UpdateTaskList(ctx, &p.UpdateTaskListRequest{\n\t\tTaskListInfo: taskListInfo,\n\t})\n\ts.NoError(err)\n}\n\nfunc (s *MatchingPersistenceSuite) deleteAllTaskList() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tvar nextPageToken []byte\n\tfor {\n\t\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 10, PageToken: nextPageToken})\n\t\ts.NoError(err)\n\t\tfor _, it := range resp.Items {\n\t\t\terr = s.TaskMgr.DeleteTaskList(ctx, &p.DeleteTaskListRequest{\n\t\t\t\tDomainID:     it.DomainID,\n\t\t\t\tTaskListName: it.Name,\n\t\t\t\tTaskListType: it.TaskType,\n\t\t\t\tRangeID:      it.RangeID,\n\t\t\t})\n\t\t\ts.NoError(err)\n\t\t}\n\t\tnextPageToken = resp.NextPageToken\n\t\tif nextPageToken == nil {\n\t\t\tbreak\n\t\t}\n\t}\n}\n\n// TestListWithOneTaskList test\nfunc (s *MatchingPersistenceSuite) TestListWithOneTaskList() {\n\tif s.TaskMgr.GetName() == \"cassandra\" || s.TaskMgr.GetName() == \"shardedNosql\" {\n\t\t// ListTaskList API is currently not supported in cassandra\n\t\treturn\n\t}\n\ts.deleteAllTaskList()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 10})\n\ts.NoError(err)\n\ts.Nil(resp.NextPageToken)\n\ts.Equal(0, len(resp.Items))\n\n\trangeID := int64(0)\n\tackLevel := int64(0)\n\tdomainID := uuid.New()\n\tfor i := 0; i < 10; i++ {\n\t\trangeID++\n\t\tupdatedTime := time.Now()\n\t\t_, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\t\tDomainID:     domainID,\n\t\t\tTaskList:     \"list-task-list-test-tl0\",\n\t\t\tTaskType:     p.TaskListTypeActivity,\n\t\t\tTaskListKind: p.TaskListKindSticky,\n\t\t})\n\t\ts.NoError(err)\n\n\t\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 10})\n\t\ts.NoError(err)\n\n\t\ts.Equal(1, len(resp.Items))\n\t\ts.Equal(domainID, resp.Items[0].DomainID)\n\t\ts.Equal(\"list-task-list-test-tl0\", resp.Items[0].Name)\n\t\ts.Equal(p.TaskListTypeActivity, resp.Items[0].TaskType)\n\t\ts.Equal(p.TaskListKindSticky, resp.Items[0].Kind)\n\t\ts.Equal(rangeID, resp.Items[0].RangeID)\n\t\ts.Equal(ackLevel, resp.Items[0].AckLevel)\n\t\ts.True(resp.Items[0].LastUpdated.After(updatedTime) || resp.Items[0].LastUpdated.Equal(updatedTime))\n\n\t\tackLevel++\n\t\tupdatedTime = time.Now()\n\t\t_, err = s.TaskMgr.UpdateTaskList(ctx, &p.UpdateTaskListRequest{\n\t\t\tTaskListInfo: &p.TaskListInfo{\n\t\t\t\tDomainID: domainID,\n\t\t\t\tName:     \"list-task-list-test-tl0\",\n\t\t\t\tTaskType: p.TaskListTypeActivity,\n\t\t\t\tRangeID:  rangeID,\n\t\t\t\tAckLevel: ackLevel,\n\t\t\t\tKind:     p.TaskListKindSticky,\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\n\t\tresp, err = s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 10})\n\t\ts.NoError(err)\n\t\ts.Equal(1, len(resp.Items))\n\t\ts.True(resp.Items[0].LastUpdated.After(updatedTime) || resp.Items[0].LastUpdated.Equal(updatedTime))\n\t}\n\ts.deleteAllTaskList()\n}\n\n// TestListWithMultipleTaskList test\nfunc (s *MatchingPersistenceSuite) TestListWithMultipleTaskList() {\n\tif s.TaskMgr.GetName() == \"cassandra\" || s.TaskMgr.GetName() == \"shardedNosql\" {\n\t\t// ListTaskList API is currently not supported in cassandra\"\n\t\treturn\n\t}\n\ts.deleteAllTaskList()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\ttlNames := make(map[string]struct{})\n\tfor i := 0; i < 10; i++ {\n\t\tname := fmt.Sprintf(\"test-list-with-multiple-%v\", i)\n\t\t_, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\t\tDomainID:     domainID,\n\t\t\tTaskList:     name,\n\t\t\tTaskType:     p.TaskListTypeActivity,\n\t\t\tTaskListKind: p.TaskListKindNormal,\n\t\t})\n\t\ts.NoError(err)\n\t\ttlNames[name] = struct{}{}\n\t\tlistedNames := make(map[string]struct{})\n\t\tvar nextPageToken []byte\n\t\tfor {\n\t\t\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 1, PageToken: nextPageToken})\n\t\t\ts.NoError(err)\n\t\t\tfor _, it := range resp.Items {\n\t\t\t\ts.Equal(domainID, it.DomainID)\n\t\t\t\ts.Equal(p.TaskListTypeActivity, it.TaskType)\n\t\t\t\ts.Equal(p.TaskListKindNormal, it.Kind)\n\t\t\t\t_, ok := listedNames[it.Name]\n\t\t\t\ts.False(ok, \"list API returns duplicate entries - have: %+v got:%v\", listedNames, it.Name)\n\t\t\t\tlistedNames[it.Name] = struct{}{}\n\t\t\t}\n\t\t\tnextPageToken = resp.NextPageToken\n\t\t\tif nextPageToken == nil {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\ts.Equal(tlNames, listedNames, \"list API returned wrong set of task list names\")\n\t}\n\n\t// final test again pagination\n\ttotal := 0\n\tvar nextPageToken []byte\n\tfor {\n\t\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{\n\t\t\tPageSize:  6,\n\t\t\tPageToken: nextPageToken,\n\t\t})\n\t\ts.NoError(err)\n\t\ttotal += len(resp.Items)\n\t\tif resp.NextPageToken == nil {\n\t\t\tbreak\n\t\t}\n\t\tnextPageToken = resp.NextPageToken\n\t}\n\ts.Equal(10, total)\n\n\ts.deleteAllTaskList()\n\tresp, err := s.TaskMgr.ListTaskList(ctx, &p.ListTaskListRequest{PageSize: 10})\n\ts.NoError(err)\n\ts.Nil(resp.NextPageToken)\n\ts.Equal(0, len(resp.Items))\n}\n\nfunc (s *MatchingPersistenceSuite) TestGetOrphanTasks() {\n\tif os.Getenv(\"SKIP_GET_ORPHAN_TASKS\") != \"\" {\n\t\ts.T().Skipf(\"GetOrphanTasks not supported in %v\", s.TaskMgr.GetName())\n\t}\n\tif s.TaskMgr.GetName() == \"cassandra\" || s.TaskMgr.GetName() == \"shardedNosql\" {\n\t\t// GetOrphanTasks API is currently not supported in cassandra\"\n\t\treturn\n\t}\n\ts.deleteAllTaskList()\n\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\toresp, err := s.TaskMgr.GetOrphanTasks(ctx, &p.GetOrphanTasksRequest{Limit: 10})\n\ts.NoError(err)\n\t// existing orphans that caused by other tests\n\texistingOrphans := len(oresp.Tasks)\n\n\tdomainID := uuid.New()\n\tname := \"test-list-with-orphans\"\n\tresp, err := s.TaskMgr.LeaseTaskList(ctx, &p.LeaseTaskListRequest{\n\t\tDomainID:     domainID,\n\t\tTaskList:     name,\n\t\tTaskType:     p.TaskListTypeActivity,\n\t\tTaskListKind: p.TaskListKindNormal,\n\t})\n\ts.NoError(err)\n\n\twid := uuid.New()\n\trid := uuid.New()\n\ts.TaskMgr.CreateTasks(ctx, &p.CreateTasksRequest{\n\t\tTaskListInfo: resp.TaskListInfo,\n\t\tTasks: []*p.CreateTaskInfo{\n\t\t\t{\n\t\t\t\tData: &p.TaskInfo{\n\t\t\t\t\tDomainID:                      domainID,\n\t\t\t\t\tWorkflowID:                    wid,\n\t\t\t\t\tRunID:                         rid,\n\t\t\t\t\tTaskID:                        0,\n\t\t\t\t\tScheduleID:                    0,\n\t\t\t\t\tScheduleToStartTimeoutSeconds: 0,\n\t\t\t\t\tExpiry:                        time.Now(),\n\t\t\t\t\tCreatedTime:                   time.Now(),\n\t\t\t\t},\n\t\t\t\tTaskID: 0,\n\t\t\t},\n\t\t},\n\t})\n\n\toresp, err = s.TaskMgr.GetOrphanTasks(ctx, &p.GetOrphanTasksRequest{Limit: 10})\n\ts.NoError(err)\n\n\ts.Equal(existingOrphans, len(oresp.Tasks))\n\n\ts.deleteAllTaskList()\n\n\toresp, err = s.TaskMgr.GetOrphanTasks(ctx, &p.GetOrphanTasksRequest{Limit: 10})\n\ts.NoError(err)\n\n\ts.Equal(existingOrphans+1, len(oresp.Tasks))\n\tfound := false\n\tfor _, it := range oresp.Tasks {\n\t\tif it.DomainID != domainID {\n\t\t\tcontinue\n\t\t}\n\t\ts.Equal(p.TaskListTypeActivity, it.TaskType)\n\t\ts.Equal(int64(0), it.TaskID)\n\t\ts.Equal(name, it.TaskListName)\n\t\tfound = true\n\t}\n\ts.True(found)\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/metadataPersistenceV2Test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"strconv\"\n\t\"strings\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// MetadataPersistenceSuiteV2 is test of the V2 version of metadata persistence\n\tMetadataPersistenceSuiteV2 struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\n// SetupSuite implementation\nfunc (m *MetadataPersistenceSuiteV2) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// SetupTest implementation\nfunc (m *MetadataPersistenceSuiteV2) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\tm.Assertions = require.New(m.T())\n\n\t// cleanup the domain created\n\tvar token []byte\n\tpageSize := 10\nListLoop:\n\tfor {\n\t\tresp, err := m.ListDomains(context.Background(), pageSize, token)\n\t\tm.NoError(err)\n\t\ttoken = resp.NextPageToken\n\t\tfor _, domain := range resp.Domains {\n\t\t\tm.NoError(m.DeleteDomain(context.Background(), domain.Info.ID, \"\"))\n\t\t}\n\t\tif len(token) == 0 {\n\t\t\tbreak ListLoop\n\t\t}\n\t}\n}\n\n// TearDownTest implementation\nfunc (m *MetadataPersistenceSuiteV2) TearDownTest() {\n}\n\n// TearDownSuite implementation\nfunc (m *MetadataPersistenceSuiteV2) TearDownSuite() {\n\tm.TearDownWorkflowStore()\n}\n\n// TestCreateDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestCreateDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tid := uuid.New()\n\tname := \"create-domain-test-name\"\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"create-domain-test-description\"\n\towner := \"create-domain-test-owner\"\n\tdata := map[string]string{\"k1\": \"v1\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\tbadBinaries := types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}\n\tisGlobalDomain := false\n\tconfigVersion := int64(0)\n\tfailoverVersion := int64(0)\n\tlastUpdateTime := int64(100)\n\n\tresp0, err0 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\tBadBinaries:              badBinaries,\n\t\t},\n\t\t&p.DomainReplicationConfig{},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tlastUpdateTime,\n\t)\n\tm.NoError(err0)\n\tm.NotNil(resp0)\n\tm.Equal(id, resp0.ID)\n\n\t// for domain which do not have replication config set, will default to\n\t// use current cluster as active, with current cluster as all clusters\n\tresp1, err1 := m.GetDomain(ctx, id, \"\")\n\tm.NoError(err1)\n\tm.NotNil(resp1)\n\tm.Equal(id, resp1.Info.ID)\n\tm.Equal(name, resp1.Info.Name)\n\tm.Equal(status, resp1.Info.Status)\n\tm.Equal(description, resp1.Info.Description)\n\tm.Equal(owner, resp1.Info.OwnerEmail)\n\tm.Equal(data, resp1.Info.Data)\n\tm.Equal(retention, resp1.Config.Retention)\n\tm.Equal(emitMetric, resp1.Config.EmitMetric)\n\tm.Equal(historyArchivalStatus, resp1.Config.HistoryArchivalStatus)\n\tm.Equal(historyArchivalURI, resp1.Config.HistoryArchivalURI)\n\tm.Equal(visibilityArchivalStatus, resp1.Config.VisibilityArchivalStatus)\n\tm.Equal(visibilityArchivalURI, resp1.Config.VisibilityArchivalURI)\n\tm.Equal(badBinaries, resp1.Config.BadBinaries)\n\tm.Equal(cluster.TestCurrentClusterName, resp1.ReplicationConfig.ActiveClusterName)\n\tm.Equal(1, len(resp1.ReplicationConfig.Clusters))\n\tm.Equal(isGlobalDomain, resp1.IsGlobalDomain)\n\tm.Equal(configVersion, resp1.ConfigVersion)\n\tm.Equal(failoverVersion, resp1.FailoverVersion)\n\tm.Equal(constants.InitialPreviousFailoverVersion, resp1.PreviousFailoverVersion)\n\tm.True(resp1.ReplicationConfig.Clusters[0].ClusterName == cluster.TestCurrentClusterName)\n\tm.Equal(p.InitialFailoverNotificationVersion, resp1.FailoverNotificationVersion)\n\tm.Nil(resp1.FailoverEndTime)\n\tm.Equal(lastUpdateTime, resp1.LastUpdatedTime)\n\n\tresp2, err2 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: \"fail\",\n\t\t\tOwnerEmail:  \"fail\",\n\t\t\tData:        map[string]string{},\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                100,\n\t\t\tEmitMetric:               false,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t},\n\t\t&p.DomainReplicationConfig{},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.Error(err2)\n\tm.IsType(&types.DomainAlreadyExistsError{}, err2)\n\tm.Nil(resp2)\n}\n\n// TestGetDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestGetDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tid := uuid.New()\n\tname := \"get-domain-test-name\"\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"get-domain-test-description\"\n\towner := \"get-domain-test-owner\"\n\tdata := map[string]string{\"k1\": \"v1\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(11)\n\tfailoverVersion := int64(59)\n\tisGlobalDomain := true\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tresp0, err0 := m.GetDomain(ctx, \"\", \"does-not-exist\")\n\tm.Nil(resp0)\n\tm.Error(err0)\n\tm.IsType(&types.EntityNotExistsError{}, err0)\n\ttestBinaries := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t\tOperator:        \"test-operator\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t},\n\t\t},\n\t}\n\n\tresp1, err1 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\tBadBinaries:              testBinaries,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.NoError(err1)\n\tm.NotNil(resp1)\n\tm.Equal(id, resp1.ID)\n\n\tresp2, err2 := m.GetDomain(ctx, id, \"\")\n\tm.NoError(err2)\n\tm.NotNil(resp2)\n\tm.Equal(id, resp2.Info.ID)\n\tm.Equal(name, resp2.Info.Name)\n\tm.Equal(status, resp2.Info.Status)\n\tm.Equal(description, resp2.Info.Description)\n\tm.Equal(owner, resp2.Info.OwnerEmail)\n\tm.Equal(data, resp2.Info.Data)\n\tm.Equal(retention, resp2.Config.Retention)\n\tm.Equal(emitMetric, resp2.Config.EmitMetric)\n\tm.Equal(historyArchivalStatus, resp2.Config.HistoryArchivalStatus)\n\tm.Equal(historyArchivalURI, resp2.Config.HistoryArchivalURI)\n\tm.Equal(visibilityArchivalStatus, resp2.Config.VisibilityArchivalStatus)\n\tm.Equal(visibilityArchivalURI, resp2.Config.VisibilityArchivalURI)\n\tm.Equal(testBinaries, resp2.Config.BadBinaries)\n\tm.Equal(clusterActive, resp2.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(clusters), len(resp2.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(clusters[index], resp2.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(isGlobalDomain, resp2.IsGlobalDomain)\n\tm.Equal(configVersion, resp2.ConfigVersion)\n\tm.Equal(failoverVersion, resp2.FailoverVersion)\n\tm.Equal(constants.InitialPreviousFailoverVersion, resp2.PreviousFailoverVersion)\n\tm.Equal(p.InitialFailoverNotificationVersion, resp2.FailoverNotificationVersion)\n\tm.Nil(resp2.FailoverEndTime)\n\tm.NotEqual(0, resp2.LastUpdatedTime)\n\n\tresp3, err3 := m.GetDomain(ctx, \"\", name)\n\tm.NoError(err3)\n\tm.NotNil(resp3)\n\tm.Equal(id, resp3.Info.ID)\n\tm.Equal(name, resp3.Info.Name)\n\tm.Equal(status, resp3.Info.Status)\n\tm.Equal(description, resp3.Info.Description)\n\tm.Equal(owner, resp3.Info.OwnerEmail)\n\tm.Equal(data, resp3.Info.Data)\n\tm.Equal(retention, resp3.Config.Retention)\n\tm.Equal(emitMetric, resp3.Config.EmitMetric)\n\tm.Equal(historyArchivalStatus, resp3.Config.HistoryArchivalStatus)\n\tm.Equal(historyArchivalURI, resp3.Config.HistoryArchivalURI)\n\tm.Equal(visibilityArchivalStatus, resp3.Config.VisibilityArchivalStatus)\n\tm.Equal(visibilityArchivalURI, resp3.Config.VisibilityArchivalURI)\n\tm.Equal(clusterActive, resp3.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(clusters), len(resp3.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(clusters[index], resp3.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(isGlobalDomain, resp3.IsGlobalDomain)\n\tm.Equal(configVersion, resp3.ConfigVersion)\n\tm.Equal(failoverVersion, resp3.FailoverVersion)\n\tm.Equal(constants.InitialPreviousFailoverVersion, resp2.PreviousFailoverVersion)\n\tm.Equal(p.InitialFailoverNotificationVersion, resp3.FailoverNotificationVersion)\n\tm.NotEqual(0, resp3.LastUpdatedTime)\n\n\tresp4, err4 := m.GetDomain(ctx, id, name)\n\tm.Error(err4)\n\tm.IsType(&types.BadRequestError{}, err4)\n\tm.Nil(resp4)\n\n\tresp5, err5 := m.GetDomain(ctx, \"\", \"\")\n\tm.Nil(resp5)\n\tm.IsType(&types.BadRequestError{}, err5)\n\n\t_, err6 := m.CreateDomain(ctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.Error(err6)\n}\n\n// TestConcurrentCreateDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestConcurrentCreateDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tconcurrency := 16\n\tnumDomains := 5\n\tdomainIDs := make([]string, numDomains)\n\tnames := make([]string, numDomains)\n\tregistered := make([]bool, numDomains)\n\tfor idx := range domainIDs {\n\t\tdomainIDs[idx] = uuid.New()\n\t\tnames[idx] = \"concurrent-create-domain-test-name-\" + strconv.Itoa(idx)\n\t}\n\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"concurrent-create-domain-test-description\"\n\towner := \"create-domain-test-owner\"\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(10)\n\tfailoverVersion := int64(59)\n\tisGlobalDomain := true\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\ttestBinaries := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t\tOperator:        \"test-operator\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t},\n\t\t},\n\t}\n\tsuccessCount := 0\n\tvar mutex sync.Mutex\n\tvar wg sync.WaitGroup\n\tfor i := 1; i <= concurrency; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdata := map[string]string{\"k0\": fmt.Sprintf(\"v-%v\", idx)}\n\t\t\t_, err1 := m.CreateDomain(ctx,\n\t\t\t\t&p.DomainInfo{\n\t\t\t\t\tID:          domainIDs[idx%numDomains],\n\t\t\t\t\tName:        names[idx%numDomains],\n\t\t\t\t\tStatus:      status,\n\t\t\t\t\tDescription: description,\n\t\t\t\t\tOwnerEmail:  owner,\n\t\t\t\t\tData:        data,\n\t\t\t\t},\n\t\t\t\t&p.DomainConfig{\n\t\t\t\t\tRetention:                retention,\n\t\t\t\t\tEmitMetric:               emitMetric,\n\t\t\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\t\t\tBadBinaries:              testBinaries,\n\t\t\t\t},\n\t\t\t\t&p.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: clusterActive,\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tisGlobalDomain,\n\t\t\t\tconfigVersion,\n\t\t\t\tfailoverVersion,\n\t\t\t\t0,\n\t\t\t)\n\t\t\tmutex.Lock()\n\t\t\tdefer mutex.Unlock()\n\t\t\tif err1 == nil {\n\t\t\t\tsuccessCount++\n\t\t\t\tregistered[idx%numDomains] = true\n\t\t\t}\n\t\t\tif _, ok := err1.(*types.DomainAlreadyExistsError); ok {\n\t\t\t\tregistered[idx%numDomains] = true\n\t\t\t}\n\t\t\twg.Done()\n\t\t}(i)\n\t}\n\twg.Wait()\n\tm.GreaterOrEqual(successCount, 1)\n\n\tfor i := 0; i != numDomains; i++ {\n\t\tif !registered[i] {\n\t\t\tcontinue\n\t\t}\n\n\t\tresp, err3 := m.GetDomain(ctx, \"\", names[i])\n\t\tm.NoError(err3)\n\t\tm.NotNil(resp)\n\t\tm.Equal(domainIDs[i], resp.Info.ID)\n\t\tm.Equal(names[i], resp.Info.Name)\n\t\tm.Equal(status, resp.Info.Status)\n\t\tm.Equal(description, resp.Info.Description)\n\t\tm.Equal(owner, resp.Info.OwnerEmail)\n\t\tm.Equal(retention, resp.Config.Retention)\n\t\tm.Equal(emitMetric, resp.Config.EmitMetric)\n\t\tm.Equal(historyArchivalStatus, resp.Config.HistoryArchivalStatus)\n\t\tm.Equal(historyArchivalURI, resp.Config.HistoryArchivalURI)\n\t\tm.Equal(visibilityArchivalStatus, resp.Config.VisibilityArchivalStatus)\n\t\tm.Equal(visibilityArchivalURI, resp.Config.VisibilityArchivalURI)\n\t\tm.Equal(testBinaries, resp.Config.BadBinaries)\n\t\tm.Equal(clusterActive, resp.ReplicationConfig.ActiveClusterName)\n\t\tm.Equal(len(clusters), len(resp.ReplicationConfig.Clusters))\n\t\tfor index := range clusters {\n\t\t\tm.Equal(clusters[index], resp.ReplicationConfig.Clusters[index])\n\t\t}\n\t\tm.Equal(isGlobalDomain, resp.IsGlobalDomain)\n\t\tm.Equal(configVersion, resp.ConfigVersion)\n\t\tm.Equal(failoverVersion, resp.FailoverVersion)\n\t\tm.Equal(constants.InitialPreviousFailoverVersion, resp.PreviousFailoverVersion)\n\n\t\t// check domain data\n\t\tss := strings.Split(resp.Info.Data[\"k0\"], \"-\")\n\t\tm.Equal(2, len(ss))\n\t\tvi, err := strconv.Atoi(ss[1])\n\t\tm.NoError(err)\n\t\tm.Equal(true, vi > 0 && vi <= concurrency)\n\t}\n}\n\n// TestConcurrentUpdateDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestConcurrentUpdateDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tisolationGroups := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n\tasyncWFCfg := types.AsyncWorkflowConfiguration{\n\t\tEnabled:             true,\n\t\tPredefinedQueueName: \"testQueue\",\n\t}\n\n\tid := uuid.New()\n\tname := \"concurrent-update-domain-test-name\"\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"update-domain-test-description\"\n\towner := \"update-domain-test-owner\"\n\tdata := map[string]string{\"k1\": \"v1\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\tbadBinaries := types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}}\n\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(10)\n\tfailoverVersion := int64(59)\n\tisGlobalDomain := true\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tresp1, err1 := m.CreateDomain(ctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\tBadBinaries:              badBinaries,\n\t\t\tIsolationGroups:          isolationGroups,\n\t\t\tAsyncWorkflowConfig:      asyncWFCfg,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.NoError(err1)\n\tm.Equal(id, resp1.ID)\n\n\tresp2, err2 := m.GetDomain(ctx, id, \"\")\n\tm.NoError(err2)\n\tm.Equal(badBinaries, resp2.Config.BadBinaries)\n\tmetadata, err := m.DomainManager.GetMetadata(ctx)\n\tm.NoError(err)\n\tnotificationVersion := metadata.NotificationVersion\n\n\ttestBinaries := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t\tOperator:        \"test-operator\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t},\n\t\t},\n\t}\n\tconcurrency := 16\n\tsuccessCount := int32(0)\n\tvar wg sync.WaitGroup\n\tfor i := 1; i <= concurrency; i++ {\n\t\tnewValue := fmt.Sprintf(\"v-%v\", i)\n\t\twg.Add(1)\n\t\tgo func(updatedData map[string]string) {\n\t\t\terr3 := m.UpdateDomain(\n\t\t\t\tctx,\n\t\t\t\t&p.DomainInfo{\n\t\t\t\t\tID:          resp2.Info.ID,\n\t\t\t\t\tName:        resp2.Info.Name,\n\t\t\t\t\tStatus:      resp2.Info.Status,\n\t\t\t\t\tDescription: resp2.Info.Description,\n\t\t\t\t\tOwnerEmail:  resp2.Info.OwnerEmail,\n\t\t\t\t\tData:        updatedData,\n\t\t\t\t},\n\t\t\t\t&p.DomainConfig{\n\t\t\t\t\tRetention:                resp2.Config.Retention,\n\t\t\t\t\tEmitMetric:               resp2.Config.EmitMetric,\n\t\t\t\t\tHistoryArchivalStatus:    resp2.Config.HistoryArchivalStatus,\n\t\t\t\t\tHistoryArchivalURI:       resp2.Config.HistoryArchivalURI,\n\t\t\t\t\tVisibilityArchivalStatus: resp2.Config.VisibilityArchivalStatus,\n\t\t\t\t\tVisibilityArchivalURI:    resp2.Config.VisibilityArchivalURI,\n\t\t\t\t\tBadBinaries:              testBinaries,\n\t\t\t\t\tIsolationGroups:          isolationGroups,\n\t\t\t\t\tAsyncWorkflowConfig:      asyncWFCfg,\n\t\t\t\t},\n\t\t\t\t&p.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: resp2.ReplicationConfig.ActiveClusterName,\n\t\t\t\t\tClusters:          resp2.ReplicationConfig.Clusters,\n\t\t\t\t},\n\t\t\t\tresp2.ConfigVersion,\n\t\t\t\tresp2.FailoverVersion,\n\t\t\t\tresp2.FailoverNotificationVersion,\n\t\t\t\tresp2.PreviousFailoverVersion,\n\t\t\t\tnil,\n\t\t\t\tnotificationVersion,\n\t\t\t\t0,\n\t\t\t)\n\t\t\tif err3 == nil {\n\t\t\t\tatomic.AddInt32(&successCount, 1)\n\t\t\t}\n\t\t\twg.Done()\n\t\t}(map[string]string{\"k0\": newValue})\n\t}\n\twg.Wait()\n\tm.Greater(successCount, int32(0))\n\tallDomains, err := m.ListDomains(ctx, 100, nil)\n\tm.NoError(err)\n\tdomainNameMap := make(map[string]bool)\n\tfor _, domain := range allDomains.Domains {\n\t\t_, ok := domainNameMap[domain.Info.Name]\n\t\tm.False(ok)\n\t\tdomainNameMap[domain.Info.Name] = true\n\t}\n\n\tresp3, err3 := m.GetDomain(ctx, \"\", name)\n\tm.NoError(err3)\n\tm.NotNil(resp3)\n\tm.Equal(id, resp3.Info.ID)\n\tm.Equal(name, resp3.Info.Name)\n\tm.Equal(status, resp3.Info.Status)\n\tm.Equal(isGlobalDomain, resp3.IsGlobalDomain)\n\tm.Equal(description, resp3.Info.Description)\n\tm.Equal(owner, resp3.Info.OwnerEmail)\n\n\tm.Equal(retention, resp3.Config.Retention)\n\tm.Equal(emitMetric, resp3.Config.EmitMetric)\n\tm.Equal(historyArchivalStatus, resp3.Config.HistoryArchivalStatus)\n\tm.Equal(historyArchivalURI, resp3.Config.HistoryArchivalURI)\n\tm.Equal(visibilityArchivalStatus, resp3.Config.VisibilityArchivalStatus)\n\tm.Equal(visibilityArchivalURI, resp3.Config.VisibilityArchivalURI)\n\tm.Equal(testBinaries, resp3.Config.BadBinaries)\n\tm.Equal(clusterActive, resp3.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(clusters), len(resp3.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(clusters[index], resp3.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(isGlobalDomain, resp3.IsGlobalDomain)\n\tm.Equal(configVersion, resp3.ConfigVersion)\n\tm.Equal(failoverVersion, resp3.FailoverVersion)\n\tm.Equal(constants.InitialPreviousFailoverVersion, resp3.PreviousFailoverVersion)\n\n\t// check domain data\n\tss := strings.Split(resp3.Info.Data[\"k0\"], \"-\")\n\tm.Equal(2, len(ss))\n\tvi, err := strconv.Atoi(ss[1])\n\tm.NoError(err)\n\tm.Equal(true, vi > 0 && vi <= concurrency)\n}\n\n// TestUpdateDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestUpdateDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tisolationGroups1 := types.IsolationGroupConfiguration{}\n\tisolationGroups2 := types.IsolationGroupConfiguration{\n\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n\n\tasyncWFCfg1 := types.AsyncWorkflowConfiguration{\n\t\tPredefinedQueueName: \"queue1\",\n\t}\n\tasyncWFCfg2 := types.AsyncWorkflowConfiguration{\n\t\tPredefinedQueueName: \"queue2\",\n\t}\n\n\tid := uuid.New()\n\tname := \"update-domain-test-name\"\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"update-domain-test-description\"\n\towner := \"update-domain-test-owner\"\n\tdata := map[string]string{\"k1\": \"v1\"}\n\tretention := int32(10)\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(10)\n\tfailoverVersion := int64(59)\n\tfailoverEndTime := time.Now().UnixNano()\n\tisGlobalDomain := true\n\tlastUpdateTime := int64(100)\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tresp1, err1 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                retention,\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\tIsolationGroups:          isolationGroups1,\n\t\t\tAsyncWorkflowConfig:      asyncWFCfg1,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\tlastUpdateTime,\n\t)\n\tm.NoError(err1)\n\tm.Equal(id, resp1.ID)\n\n\tresp2, err2 := m.GetDomain(ctx, id, \"\")\n\tm.NoError(err2)\n\tm.Nil(resp2.FailoverEndTime)\n\tmetadata, err := m.DomainManager.GetMetadata(ctx)\n\tm.NoError(err)\n\tnotificationVersion := metadata.NotificationVersion\n\n\tupdatedStatus := p.DomainStatusDeprecated\n\tupdatedDescription := \"description-updated\"\n\tupdatedOwner := \"owner-updated\"\n\t// This will overriding the previous key-value pair\n\tupdatedData := map[string]string{\"k1\": \"v2\"}\n\tupdatedRetention := int32(20)\n\tupdatedEmitMetric := false\n\tupdatedHistoryArchivalStatus := types.ArchivalStatusDisabled\n\tupdatedHistoryArchivalURI := \"\"\n\tupdatedVisibilityArchivalStatus := types.ArchivalStatusDisabled\n\tupdatedVisibilityArchivalURI := \"\"\n\n\tupdateClusterActive := \"other random active cluster name\"\n\tupdateClusterStandby := \"other random standby cluster name\"\n\tlastUpdateTime++\n\tupdateConfigVersion := int64(12)\n\tupdateFailoverVersion := int64(28)\n\tupdatePreviousFailoverVersion := int64(20)\n\tupdateFailoverNotificationVersion := int64(14)\n\tupdateClusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: updateClusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: updateClusterStandby,\n\t\t},\n\t}\n\ttestBinaries := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t\tOperator:        \"test-operator\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t},\n\t\t},\n\t}\n\n\terr3 := m.UpdateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          resp2.Info.ID,\n\t\t\tName:        resp2.Info.Name,\n\t\t\tStatus:      updatedStatus,\n\t\t\tDescription: updatedDescription,\n\t\t\tOwnerEmail:  updatedOwner,\n\t\t\tData:        updatedData,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                updatedRetention,\n\t\t\tEmitMetric:               updatedEmitMetric,\n\t\t\tHistoryArchivalStatus:    updatedHistoryArchivalStatus,\n\t\t\tHistoryArchivalURI:       updatedHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus: updatedVisibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    updatedVisibilityArchivalURI,\n\t\t\tBadBinaries:              testBinaries,\n\t\t\tIsolationGroups:          isolationGroups2,\n\t\t\tAsyncWorkflowConfig:      asyncWFCfg2,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tupdateConfigVersion,\n\t\tupdateFailoverVersion,\n\t\tupdateFailoverNotificationVersion,\n\t\tupdatePreviousFailoverVersion,\n\t\t&failoverEndTime,\n\t\tnotificationVersion,\n\t\tlastUpdateTime,\n\t)\n\tm.NoError(err3)\n\n\tresp4, err4 := m.GetDomain(ctx, \"\", name)\n\tm.NoError(err4)\n\tm.NotNil(resp4)\n\tm.Equal(id, resp4.Info.ID)\n\tm.Equal(name, resp4.Info.Name)\n\tm.Equal(isGlobalDomain, resp4.IsGlobalDomain)\n\tm.Equal(updatedStatus, resp4.Info.Status)\n\tm.Equal(updatedDescription, resp4.Info.Description)\n\tm.Equal(updatedOwner, resp4.Info.OwnerEmail)\n\tm.Equal(updatedData, resp4.Info.Data)\n\tm.Equal(updatedRetention, resp4.Config.Retention)\n\tm.Equal(updatedEmitMetric, resp4.Config.EmitMetric)\n\tm.Equal(updatedHistoryArchivalStatus, resp4.Config.HistoryArchivalStatus)\n\tm.Equal(updatedHistoryArchivalURI, resp4.Config.HistoryArchivalURI)\n\tm.Equal(updatedVisibilityArchivalStatus, resp4.Config.VisibilityArchivalStatus)\n\tm.Equal(updatedVisibilityArchivalURI, resp4.Config.VisibilityArchivalURI)\n\tm.Equal(testBinaries, resp4.Config.BadBinaries)\n\tm.Equal(updateClusterActive, resp4.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(updateClusters), len(resp4.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(updateClusters[index], resp4.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(updateConfigVersion, resp4.ConfigVersion)\n\tm.Equal(updateFailoverVersion, resp4.FailoverVersion)\n\tm.Equal(updatePreviousFailoverVersion, resp4.PreviousFailoverVersion)\n\tm.Equal(updateFailoverNotificationVersion, resp4.FailoverNotificationVersion)\n\tm.Equal(notificationVersion, resp4.NotificationVersion)\n\tm.Equal(&failoverEndTime, resp4.FailoverEndTime)\n\tm.Equal(lastUpdateTime, resp4.LastUpdatedTime)\n\tm.Equal(isolationGroups2, resp4.Config.IsolationGroups)\n\tm.Equal(asyncWFCfg2, resp4.Config.AsyncWorkflowConfig)\n\n\tresp5, err5 := m.GetDomain(ctx, id, \"\")\n\tm.NoError(err5)\n\tm.NotNil(resp5)\n\tm.Equal(id, resp5.Info.ID)\n\tm.Equal(name, resp5.Info.Name)\n\tm.Equal(isGlobalDomain, resp5.IsGlobalDomain)\n\tm.Equal(updatedStatus, resp5.Info.Status)\n\tm.Equal(updatedDescription, resp5.Info.Description)\n\tm.Equal(updatedOwner, resp5.Info.OwnerEmail)\n\tm.Equal(updatedData, resp5.Info.Data)\n\tm.Equal(updatedRetention, resp5.Config.Retention)\n\tm.Equal(updatedEmitMetric, resp5.Config.EmitMetric)\n\tm.Equal(updatedHistoryArchivalStatus, resp5.Config.HistoryArchivalStatus)\n\tm.Equal(updatedHistoryArchivalURI, resp5.Config.HistoryArchivalURI)\n\tm.Equal(updatedVisibilityArchivalStatus, resp5.Config.VisibilityArchivalStatus)\n\tm.Equal(updatedVisibilityArchivalURI, resp5.Config.VisibilityArchivalURI)\n\tm.Equal(updateClusterActive, resp5.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(updateClusters), len(resp5.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(updateClusters[index], resp5.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(updateConfigVersion, resp5.ConfigVersion)\n\tm.Equal(updateFailoverVersion, resp5.FailoverVersion)\n\tm.Equal(updatePreviousFailoverVersion, resp5.PreviousFailoverVersion)\n\tm.Equal(updateFailoverNotificationVersion, resp5.FailoverNotificationVersion)\n\tm.Equal(notificationVersion, resp5.NotificationVersion)\n\tm.Equal(&failoverEndTime, resp5.FailoverEndTime)\n\tm.Equal(lastUpdateTime, resp5.LastUpdatedTime)\n\n\tnotificationVersion++\n\tlastUpdateTime++\n\terr6 := m.UpdateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          resp2.Info.ID,\n\t\t\tName:        resp2.Info.Name,\n\t\t\tStatus:      updatedStatus,\n\t\t\tDescription: updatedDescription,\n\t\t\tOwnerEmail:  updatedOwner,\n\t\t\tData:        updatedData,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                updatedRetention,\n\t\t\tEmitMetric:               updatedEmitMetric,\n\t\t\tHistoryArchivalStatus:    updatedHistoryArchivalStatus,\n\t\t\tHistoryArchivalURI:       updatedHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus: updatedVisibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    updatedVisibilityArchivalURI,\n\t\t\tBadBinaries:              testBinaries,\n\t\t\tIsolationGroups:          isolationGroups1,\n\t\t\tAsyncWorkflowConfig:      asyncWFCfg1,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t},\n\t\tupdateConfigVersion,\n\t\tupdateFailoverVersion,\n\t\tupdateFailoverNotificationVersion,\n\t\tupdatePreviousFailoverVersion,\n\t\tnil,\n\t\tnotificationVersion,\n\t\tlastUpdateTime,\n\t)\n\tm.NoError(err6)\n\n\tresp6, err6 := m.GetDomain(ctx, \"\", name)\n\tm.NoError(err6)\n\tm.NotNil(resp6)\n\tm.Equal(id, resp6.Info.ID)\n\tm.Equal(name, resp6.Info.Name)\n\tm.Equal(isGlobalDomain, resp6.IsGlobalDomain)\n\tm.Equal(updatedStatus, resp6.Info.Status)\n\tm.Equal(updatedDescription, resp6.Info.Description)\n\tm.Equal(updatedOwner, resp6.Info.OwnerEmail)\n\tm.Equal(updatedData, resp6.Info.Data)\n\tm.Equal(updatedRetention, resp6.Config.Retention)\n\tm.Equal(updatedEmitMetric, resp6.Config.EmitMetric)\n\tm.Equal(updatedHistoryArchivalStatus, resp6.Config.HistoryArchivalStatus)\n\tm.Equal(updatedHistoryArchivalURI, resp6.Config.HistoryArchivalURI)\n\tm.Equal(updatedVisibilityArchivalStatus, resp6.Config.VisibilityArchivalStatus)\n\tm.Equal(updatedVisibilityArchivalURI, resp6.Config.VisibilityArchivalURI)\n\tm.Equal(testBinaries, resp6.Config.BadBinaries)\n\tm.Equal(updateClusterActive, resp6.ReplicationConfig.ActiveClusterName)\n\tm.Equal(len(updateClusters), len(resp6.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(updateClusters[index], resp6.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(updateConfigVersion, resp6.ConfigVersion)\n\tm.Equal(updateFailoverVersion, resp6.FailoverVersion)\n\tm.Equal(updatePreviousFailoverVersion, resp6.PreviousFailoverVersion)\n\tm.Equal(updateFailoverNotificationVersion, resp6.FailoverNotificationVersion)\n\tm.Equal(notificationVersion, resp6.NotificationVersion)\n\tm.Nil(resp6.FailoverEndTime)\n\tm.Equal(lastUpdateTime, resp6.LastUpdatedTime)\n\tm.Equal(isolationGroups1, resp6.Config.IsolationGroups)\n\tm.Equal(asyncWFCfg1, resp6.Config.AsyncWorkflowConfig)\n\n\t// make it active-active domain\n\tnotificationVersion++\n\tlastUpdateTime++\n\tactiveClusters := &types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\tActiveClusterName: updateClusters[0].ClusterName,\n\t\t\t\t\t},\n\t\t\t\t\t\"region2\": {\n\t\t\t\t\t\tActiveClusterName: updateClusters[1].ClusterName,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"city\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\tActiveClusterName: updateClusters[0].ClusterName,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\terr7 := m.UpdateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          resp2.Info.ID,\n\t\t\tName:        resp2.Info.Name,\n\t\t\tStatus:      updatedStatus,\n\t\t\tDescription: updatedDescription,\n\t\t\tOwnerEmail:  updatedOwner,\n\t\t\tData:        updatedData,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                updatedRetention,\n\t\t\tEmitMetric:               updatedEmitMetric,\n\t\t\tHistoryArchivalStatus:    updatedHistoryArchivalStatus,\n\t\t\tHistoryArchivalURI:       updatedHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus: updatedVisibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    updatedVisibilityArchivalURI,\n\t\t\tBadBinaries:              testBinaries,\n\t\t\tIsolationGroups:          isolationGroups1,\n\t\t\tAsyncWorkflowConfig:      asyncWFCfg1,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: updateClusterActive,\n\t\t\tClusters:          updateClusters,\n\t\t\tActiveClusters:    activeClusters,\n\t\t},\n\t\tupdateConfigVersion,\n\t\tupdateFailoverVersion,\n\t\tupdateFailoverNotificationVersion,\n\t\tupdatePreviousFailoverVersion,\n\t\tnil,\n\t\tnotificationVersion,\n\t\tlastUpdateTime,\n\t)\n\tm.NoError(err7)\n\n\tresp7, err7 := m.GetDomain(ctx, \"\", name)\n\tm.T().Logf(\"resp7: %+v\", resp7)\n\tm.T().Logf(\"resp7.Info: %+v. even after setting status\", *resp7.Info)\n\tm.NoError(err7)\n\tm.NotNil(resp7)\n\tm.Equal(id, resp7.Info.ID)\n\tm.Equal(name, resp7.Info.Name)\n\tm.Equal(isGlobalDomain, resp7.IsGlobalDomain)\n\tm.Equal(updatedStatus, resp7.Info.Status)\n\tm.Equal(isolationGroups1, resp7.Config.IsolationGroups)\n\tm.Equal(asyncWFCfg1, resp7.Config.AsyncWorkflowConfig)\n\tm.Equal(updateClusterActive, resp7.ReplicationConfig.ActiveClusterName)\n\t// ActiveClustersByRegion field has been removed - test only AttributeScopes\n\tm.Equal(activeClusters.AttributeScopes, resp7.ReplicationConfig.ActiveClusters.AttributeScopes)\n\tm.Equal(len(updateClusters), len(resp7.ReplicationConfig.Clusters))\n\tfor index := range clusters {\n\t\tm.Equal(updateClusters[index], resp7.ReplicationConfig.Clusters[index])\n\t}\n\tm.Equal(notificationVersion, resp7.NotificationVersion)\n\tm.Equal(lastUpdateTime, resp7.LastUpdatedTime)\n\tm.Equal(updateFailoverVersion, resp7.FailoverVersion)\n\tm.Equal(updateFailoverNotificationVersion, resp7.FailoverNotificationVersion)\n\tm.Equal(updatePreviousFailoverVersion, resp7.PreviousFailoverVersion)\n\tm.Nil(resp7.FailoverEndTime)\n\tm.Equal(updateConfigVersion, resp7.ConfigVersion)\n}\n\n// TestDeleteDomain test\nfunc (m *MetadataPersistenceSuiteV2) TestDeleteDomain() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tid := uuid.New()\n\tname := \"delete-domain-test-name\"\n\tstatus := p.DomainStatusRegistered\n\tdescription := \"delete-domain-test-description\"\n\towner := \"delete-domain-test-owner\"\n\tdata := map[string]string{\"k1\": \"v1\"}\n\tretention := 10\n\temitMetric := true\n\thistoryArchivalStatus := types.ArchivalStatusEnabled\n\thistoryArchivalURI := \"test://history/uri\"\n\tvisibilityArchivalStatus := types.ArchivalStatusEnabled\n\tvisibilityArchivalURI := \"test://visibility/uri\"\n\n\tclusterActive := \"some random active cluster name\"\n\tclusterStandby := \"some random standby cluster name\"\n\tconfigVersion := int64(10)\n\tfailoverVersion := int64(59)\n\tisGlobalDomain := true\n\tclusters := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby,\n\t\t},\n\t}\n\n\tresp1, err1 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                int32(retention),\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t\tIsolationGroups:          types.IsolationGroupConfiguration{},\n\t\t\tAsyncWorkflowConfig:      types.AsyncWorkflowConfiguration{},\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.NoError(err1)\n\tm.Equal(id, resp1.ID)\n\n\tresp2, err2 := m.GetDomain(ctx, \"\", name)\n\tm.NoError(err2)\n\tm.NotNil(resp2)\n\n\terr3 := m.DeleteDomain(ctx, \"\", name)\n\tm.NoError(err3)\n\n\tresp4, err4 := m.GetDomain(ctx, \"\", name)\n\tm.Error(err4)\n\tm.IsType(&types.EntityNotExistsError{}, err4)\n\tm.Nil(resp4)\n\n\tresp5, err5 := m.GetDomain(ctx, id, \"\")\n\tm.Error(err5)\n\tm.IsType(&types.EntityNotExistsError{}, err5)\n\tm.Nil(resp5)\n\n\tid = uuid.New()\n\tresp6, err6 := m.CreateDomain(\n\t\tctx,\n\t\t&p.DomainInfo{\n\t\t\tID:          id,\n\t\t\tName:        name,\n\t\t\tStatus:      status,\n\t\t\tDescription: description,\n\t\t\tOwnerEmail:  owner,\n\t\t\tData:        data,\n\t\t},\n\t\t&p.DomainConfig{\n\t\t\tRetention:                int32(retention),\n\t\t\tEmitMetric:               emitMetric,\n\t\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\t},\n\t\t&p.DomainReplicationConfig{\n\t\t\tActiveClusterName: clusterActive,\n\t\t\tClusters:          clusters,\n\t\t},\n\t\tisGlobalDomain,\n\t\tconfigVersion,\n\t\tfailoverVersion,\n\t\t0,\n\t)\n\tm.NoError(err6)\n\tm.Equal(id, resp6.ID)\n\n\terr7 := m.DeleteDomain(ctx, id, \"\")\n\tm.NoError(err7)\n\n\tresp8, err8 := m.GetDomain(ctx, \"\", name)\n\tm.Error(err8)\n\tm.IsType(&types.EntityNotExistsError{}, err8)\n\tm.Nil(resp8)\n\n\tresp9, err9 := m.GetDomain(ctx, id, \"\")\n\tm.Error(err9)\n\tm.IsType(&types.EntityNotExistsError{}, err9)\n\tm.Nil(resp9)\n}\n\n// TestListDomains test\nfunc (m *MetadataPersistenceSuiteV2) TestListDomains() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tclusterActive1 := \"some random active cluster name\"\n\tclusterStandby1 := \"some random standby cluster name\"\n\tclusters1 := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive1,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby1,\n\t\t},\n\t}\n\n\tclusterActive2 := \"other random active cluster name\"\n\tclusterStandby2 := \"other random standby cluster name\"\n\tclusters2 := []*p.ClusterReplicationConfig{\n\t\t{\n\t\t\tClusterName: clusterActive2,\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterStandby2,\n\t\t},\n\t}\n\n\ttestBinaries1 := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {\n\t\t\t\tReason:          \"test-reason1\",\n\t\t\t\tOperator:        \"test-operator1\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(123),\n\t\t\t},\n\t\t},\n\t}\n\ttestBinaries2 := types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"efg\": {\n\t\t\t\tReason:          \"test-reason2\",\n\t\t\t\tOperator:        \"test-operator2\",\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(456),\n\t\t\t},\n\t\t},\n\t}\n\n\tinputDomains := []*p.GetDomainResponse{\n\t\t{\n\t\t\tInfo: &p.DomainInfo{\n\t\t\t\tID:          uuid.New(),\n\t\t\t\tName:        \"list-domain-test-name-1\",\n\t\t\t\tStatus:      p.DomainStatusRegistered,\n\t\t\t\tDescription: \"list-domain-test-description-1\",\n\t\t\t\tOwnerEmail:  \"list-domain-test-owner-1\",\n\t\t\t\tData:        map[string]string{\"k1\": \"v1\"},\n\t\t\t},\n\t\t\tConfig: &p.DomainConfig{\n\t\t\t\tRetention:                109,\n\t\t\t\tEmitMetric:               true,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\tHistoryArchivalURI:       \"test://history/uri\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\tVisibilityArchivalURI:    \"test://visibility/uri\",\n\t\t\t\tBadBinaries:              testBinaries1,\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"name1\": types.IsolationGroupPartition{Name: \"name1\"},\n\t\t\t\t\t\"name2\": types.IsolationGroupPartition{Name: \"name2\"},\n\t\t\t\t},\n\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tReplicationConfig: &p.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: clusterActive1,\n\t\t\t\tClusters:          clusters1,\n\t\t\t},\n\t\t\tIsGlobalDomain:          true,\n\t\t\tConfigVersion:           133,\n\t\t\tFailoverVersion:         266,\n\t\t\tPreviousFailoverVersion: -1,\n\t\t},\n\t\t{\n\t\t\tInfo: &p.DomainInfo{\n\t\t\t\tID:          uuid.New(),\n\t\t\t\tName:        \"list-domain-test-name-2\",\n\t\t\t\tStatus:      p.DomainStatusRegistered,\n\t\t\t\tDescription: \"list-domain-test-description-2\",\n\t\t\t\tOwnerEmail:  \"list-domain-test-owner-2\",\n\t\t\t\tData:        map[string]string{\"k1\": \"v2\"},\n\t\t\t},\n\t\t\tConfig: &p.DomainConfig{\n\t\t\t\tRetention:                326,\n\t\t\t\tEmitMetric:               false,\n\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\t\tHistoryArchivalURI:       \"\",\n\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\t\tVisibilityArchivalURI:    \"\",\n\t\t\t\tBadBinaries:              testBinaries2,\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"name3\": types.IsolationGroupPartition{Name: \"name3\"},\n\t\t\t\t},\n\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             false,\n\t\t\t\t\tPredefinedQueueName: \"queue2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tReplicationConfig: &p.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: clusterActive2,\n\t\t\t\tClusters:          clusters2,\n\t\t\t},\n\t\t\tIsGlobalDomain:          false,\n\t\t\tConfigVersion:           400,\n\t\t\tFailoverVersion:         667,\n\t\t\tPreviousFailoverVersion: -1,\n\t\t},\n\t}\n\tfor _, domain := range inputDomains {\n\t\t_, err := m.CreateDomain(\n\t\t\tctx,\n\t\t\tdomain.Info,\n\t\t\tdomain.Config,\n\t\t\tdomain.ReplicationConfig,\n\t\t\tdomain.IsGlobalDomain,\n\t\t\tdomain.ConfigVersion,\n\t\t\tdomain.FailoverVersion,\n\t\t\t0,\n\t\t)\n\t\tm.NoError(err)\n\t}\n\n\tvar token []byte\n\tpageSize := 1\n\toutputDomains := make(map[string]*p.GetDomainResponse)\nListLoop:\n\tfor {\n\t\tresp, err := m.ListDomains(ctx, pageSize, token)\n\t\tm.NoError(err)\n\t\ttoken = resp.NextPageToken\n\t\tfor _, domain := range resp.Domains {\n\t\t\toutputDomains[domain.Info.ID] = domain\n\t\t\t// global notification version is already tested, so here we make it 0\n\t\t\t// so we can test == easily\n\t\t\tdomain.NotificationVersion = 0\n\t\t}\n\t\tif len(token) == 0 {\n\t\t\tbreak ListLoop\n\t\t}\n\t}\n\n\tm.Equal(len(inputDomains), len(outputDomains))\n\tfor _, domain := range inputDomains {\n\t\tm.Equal(domain, outputDomains[domain.Info.ID])\n\t}\n}\n\n// CreateDomain helper method\nfunc (m *MetadataPersistenceSuiteV2) CreateDomain(\n\tctx context.Context,\n\tinfo *p.DomainInfo,\n\tconfig *p.DomainConfig,\n\treplicationConfig *p.DomainReplicationConfig,\n\tisGlobaldomain bool,\n\tconfigVersion int64,\n\tfailoverVersion int64,\n\tlastUpdateTime int64,\n) (*p.CreateDomainResponse, error) {\n\n\treturn m.DomainManager.CreateDomain(ctx, &p.CreateDomainRequest{\n\t\tInfo:              info,\n\t\tConfig:            config,\n\t\tReplicationConfig: replicationConfig,\n\t\tIsGlobalDomain:    isGlobaldomain,\n\t\tConfigVersion:     configVersion,\n\t\tFailoverVersion:   failoverVersion,\n\t\tLastUpdatedTime:   lastUpdateTime,\n\t})\n}\n\n// GetDomain helper method\nfunc (m *MetadataPersistenceSuiteV2) GetDomain(ctx context.Context, id, name string) (*p.GetDomainResponse, error) {\n\treturn m.DomainManager.GetDomain(ctx, &p.GetDomainRequest{\n\t\tID:   id,\n\t\tName: name,\n\t})\n}\n\n// UpdateDomain helper method\nfunc (m *MetadataPersistenceSuiteV2) UpdateDomain(\n\tctx context.Context,\n\tinfo *p.DomainInfo,\n\tconfig *p.DomainConfig,\n\treplicationConfig *p.DomainReplicationConfig,\n\tconfigVersion int64,\n\tfailoverVersion int64,\n\tfailoverNotificationVersion int64,\n\tPreviousFailoverVersion int64,\n\tfailoverEndTime *int64,\n\tnotificationVersion int64,\n\tlastUpdateTime int64,\n) error {\n\n\treturn m.DomainManager.UpdateDomain(ctx, &p.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      config,\n\t\tReplicationConfig:           replicationConfig,\n\t\tFailoverEndTime:             failoverEndTime,\n\t\tConfigVersion:               configVersion,\n\t\tFailoverVersion:             failoverVersion,\n\t\tFailoverNotificationVersion: failoverNotificationVersion,\n\t\tPreviousFailoverVersion:     PreviousFailoverVersion,\n\t\tNotificationVersion:         notificationVersion,\n\t\tLastUpdatedTime:             lastUpdateTime,\n\t})\n}\n\n// DeleteDomain helper method\nfunc (m *MetadataPersistenceSuiteV2) DeleteDomain(ctx context.Context, id, name string) error {\n\tif len(id) > 0 {\n\t\treturn m.DomainManager.DeleteDomain(ctx, &p.DeleteDomainRequest{ID: id})\n\t}\n\treturn m.DomainManager.DeleteDomainByName(ctx, &p.DeleteDomainByNameRequest{Name: name})\n}\n\n// ListDomains helper method\nfunc (m *MetadataPersistenceSuiteV2) ListDomains(ctx context.Context, pageSize int, pageToken []byte) (*p.ListDomainsResponse, error) {\n\treturn m.DomainManager.ListDomains(ctx, &p.ListDomainsRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: pageToken,\n\t})\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/persistenceTestBase.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// TransferTaskIDGenerator generates IDs for transfer tasks written by helper methods\n\tTransferTaskIDGenerator interface {\n\t\tGenerateTransferTaskID() (int64, error)\n\t}\n\n\t// TestBaseOptions options to configure workflow test base.\n\tTestBaseOptions struct {\n\t\tDBPluginName    string\n\t\tDBName          string\n\t\tDBUsername      string\n\t\tDBPassword      string\n\t\tDBHost          string\n\t\tDBPort          int              `yaml:\"-\"`\n\t\tStoreType       string           `yaml:\"-\"`\n\t\tSchemaDir       string           `yaml:\"-\"`\n\t\tClusterMetadata cluster.Metadata `yaml:\"-\"`\n\t\tProtoVersion    int              `yaml:\"-\"`\n\t\tReplicas        int              `yaml:\"-\"`\n\t\tMaxConns        int              `yaml:\"-\"`\n\t}\n\n\t// TestBase wraps the base setup needed to create workflows over persistence layer.\n\tTestBase struct {\n\t\t*suite.Suite\n\t\tController                *gomock.Controller\n\t\tShardMgr                  persistence.ShardManager\n\t\tExecutionMgrFactory       client.Factory\n\t\tExecutionManager          persistence.ExecutionManager\n\t\tTaskMgr                   persistence.TaskManager\n\t\tHistoryV2Mgr              persistence.HistoryManager\n\t\tDomainManager             persistence.DomainManager\n\t\tDomainReplicationQueueMgr persistence.QueueManager\n\t\tShardInfo                 *persistence.ShardInfo\n\t\tTaskIDGenerator           TransferTaskIDGenerator\n\t\tClusterMetadata           cluster.Metadata\n\t\tDefaultTestCluster        testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster     testcluster.PersistenceTestCluster\n\t\tLogger                    log.Logger\n\t\tPayloadSerializer         persistence.PayloadSerializer\n\t\tConfigStoreManager        persistence.ConfigStoreManager\n\t\tDynamicConfiguration      persistence.DynamicConfiguration\n\t}\n\n\t// TestBaseParams defines the input of TestBase\n\tTestBaseParams struct {\n\t\tDefaultTestCluster    testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster testcluster.PersistenceTestCluster\n\t\tClusterMetadata       cluster.Metadata\n\t\tDynamicConfiguration  persistence.DynamicConfiguration\n\t}\n\n\t// TestTransferTaskIDGenerator helper\n\tTestTransferTaskIDGenerator struct {\n\t\tseqNum int64\n\t}\n)\n\nconst (\n\tdefaultScheduleToStartTimeout = 111\n)\n\n// NewTestBaseFromParams returns a customized test base from given input\nfunc NewTestBaseFromParams(t *testing.T, params TestBaseParams) *TestBase {\n\tres := &TestBase{\n\t\tSuite:                 &suite.Suite{},\n\t\tDefaultTestCluster:    params.DefaultTestCluster,\n\t\tVisibilityTestCluster: params.VisibilityTestCluster,\n\t\tClusterMetadata:       params.ClusterMetadata,\n\t\tPayloadSerializer:     persistence.NewPayloadSerializer(),\n\t\tDynamicConfiguration:  params.DynamicConfiguration,\n\t}\n\tres.SetT(t)\n\treturn res\n}\n\n// NewTestBaseWithNoSQL returns a persistence test base backed by nosql datastore\nfunc NewTestBaseWithNoSQL(t *testing.T, options *TestBaseOptions) *TestBase {\n\tif options.DBName == \"\" {\n\t\toptions.DBName = \"test_\" + GenerateRandomDBName(10)\n\t}\n\ttestCluster := nosql.NewTestCluster(t, nosql.TestClusterParams{\n\t\tPluginName:   options.DBPluginName,\n\t\tKeySpace:     options.DBName,\n\t\tUsername:     options.DBUsername,\n\t\tPassword:     options.DBPassword,\n\t\tHost:         options.DBHost,\n\t\tPort:         options.DBPort,\n\t\tProtoVersion: options.ProtoVersion,\n\t\tReplicas:     options.Replicas,\n\t\tMaxConns:     options.MaxConns,\n\t})\n\tmetadata := options.ClusterMetadata\n\tif metadata.GetCurrentClusterName() == \"\" {\n\t\tmetadata = cluster.GetTestClusterMetadata(false)\n\t}\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableSQLAsyncTransaction:                dynamicproperties.GetBoolPropertyFn(false),\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tDomainAuditLogTTL:                        func(domainID string) time.Duration { return time.Hour * 24 * 365 }, // 1 year default\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := TestBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tClusterMetadata:       metadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\treturn NewTestBaseFromParams(t, params)\n}\n\n// NewTestBaseWithSQL returns a new persistence test base backed by SQL\nfunc NewTestBaseWithSQL(t *testing.T, options *TestBaseOptions) *TestBase {\n\tif options.DBName == \"\" {\n\t\toptions.DBName = \"test_\" + GenerateRandomDBName(10)\n\t}\n\ttestCluster, err := sql.NewTestCluster(options.DBPluginName, options.DBName, options.DBUsername, options.DBPassword, options.DBHost, options.DBPort, options.SchemaDir)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\tmetadata := options.ClusterMetadata\n\tif metadata.GetCurrentClusterName() == \"\" {\n\t\tmetadata = cluster.GetTestClusterMetadata(false)\n\t}\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableSQLAsyncTransaction:                dynamicproperties.GetBoolPropertyFn(false),\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tDomainAuditLogTTL:                        func(domainID string) time.Duration { return time.Hour * 24 * 365 }, // 1 year default\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := TestBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tClusterMetadata:       metadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\treturn NewTestBaseFromParams(t, params)\n}\n\n// Config returns the persistence configuration for this test\nfunc (s *TestBase) Config() config.Persistence {\n\tcfg := s.DefaultTestCluster.Config()\n\tif s.DefaultTestCluster == s.VisibilityTestCluster {\n\t\treturn cfg\n\t}\n\tvCfg := s.VisibilityTestCluster.Config()\n\tcfg.VisibilityStore = \"visibility_ \" + vCfg.VisibilityStore\n\tcfg.DataStores[cfg.VisibilityStore] = vCfg.DataStores[vCfg.VisibilityStore]\n\treturn cfg\n}\n\n// Setup sets up the test base, must be called as part of SetupSuite\nfunc (s *TestBase) Setup() {\n\tvar err error\n\tshardID := 10\n\tclusterName := s.ClusterMetadata.GetCurrentClusterName()\n\n\ts.Controller = gomock.NewController(s.T())\n\ts.Logger = testlogger.New(s.T())\n\n\ts.DefaultTestCluster.SetupTestDatabase()\n\n\tcfg := s.DefaultTestCluster.Config()\n\tscope := tally.NewTestScope(service.History, make(map[string]string))\n\tmetricsClient := metrics.NewClient(scope, service.GetMetricsServiceIdx(service.History, s.Logger), metrics.HistogramMigration{})\n\tfactory := client.NewFactory(&cfg, nil, clusterName, metricsClient, s.Logger, &s.DynamicConfiguration)\n\n\ts.TaskMgr, err = factory.NewTaskManager()\n\ts.fatalOnError(\"NewTaskManager\", err)\n\n\ts.DomainManager, err = factory.NewDomainManager()\n\ts.fatalOnError(\"NewDomainManager\", err)\n\n\ts.HistoryV2Mgr, err = factory.NewHistoryManager()\n\ts.fatalOnError(\"NewHistoryManager\", err)\n\n\ts.ShardMgr, err = factory.NewShardManager()\n\ts.fatalOnError(\"NewShardManager\", err)\n\n\ts.ConfigStoreManager, err = factory.NewConfigStoreManager()\n\ts.fatalOnError(\"NewConfigStoreManager\", err)\n\n\ts.ExecutionMgrFactory = factory\n\ts.ExecutionManager, err = factory.NewExecutionManager(shardID)\n\ts.fatalOnError(\"NewExecutionManager\", err)\n\n\tdomainFilter := &types.DomainFilter{\n\t\tDomainIDs:    []string{},\n\t\tReverseMatch: true,\n\t}\n\ttransferPQSMap := map[string][]*types.ProcessingQueueState{\n\t\ts.ClusterMetadata.GetCurrentClusterName(): {\n\t\t\t&types.ProcessingQueueState{\n\t\t\t\tLevel:        common.Int32Ptr(0),\n\t\t\t\tAckLevel:     common.Int64Ptr(0),\n\t\t\t\tMaxLevel:     common.Int64Ptr(0),\n\t\t\t\tDomainFilter: domainFilter,\n\t\t\t},\n\t\t},\n\t}\n\ttransferPQS := types.ProcessingQueueStates{StatesByCluster: transferPQSMap}\n\ttimerPQSMap := map[string][]*types.ProcessingQueueState{\n\t\ts.ClusterMetadata.GetCurrentClusterName(): {\n\t\t\t&types.ProcessingQueueState{\n\t\t\t\tLevel:        common.Int32Ptr(0),\n\t\t\t\tAckLevel:     common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\tMaxLevel:     common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\tDomainFilter: domainFilter,\n\t\t\t},\n\t\t},\n\t}\n\ttimerPQS := types.ProcessingQueueStates{StatesByCluster: timerPQSMap}\n\n\ts.ShardInfo = &persistence.ShardInfo{\n\t\tShardID:                       shardID,\n\t\tRangeID:                       0,\n\t\tTransferAckLevel:              0,\n\t\tReplicationAckLevel:           0,\n\t\tTimerAckLevel:                 time.Time{},\n\t\tClusterTimerAckLevel:          map[string]time.Time{clusterName: time.Time{}},\n\t\tClusterTransferAckLevel:       map[string]int64{clusterName: 0},\n\t\tTransferProcessingQueueStates: &transferPQS,\n\t\tTimerProcessingQueueStates:    &timerPQS,\n\t}\n\n\ts.TaskIDGenerator = &TestTransferTaskIDGenerator{}\n\terr = s.ShardMgr.CreateShard(context.Background(), &persistence.CreateShardRequest{ShardInfo: s.ShardInfo})\n\ts.fatalOnError(\"CreateShard\", err)\n\n\tqueue, err := factory.NewDomainReplicationQueueManager()\n\ts.fatalOnError(\"Create DomainReplicationQueue\", err)\n\ts.DomainReplicationQueueMgr = queue\n}\n\nfunc (s *TestBase) fatalOnError(msg string, err error) {\n\tif err != nil {\n\t\ts.Logger.Fatal(msg, tag.Error(err))\n\t}\n}\n\n// CreateShard is a utility method to create the shard using persistence layer\nfunc (s *TestBase) CreateShard(ctx context.Context, shardID int, owner string, rangeID int64) error {\n\tinfo := &persistence.ShardInfo{\n\t\tShardID: shardID,\n\t\tOwner:   owner,\n\t\tRangeID: rangeID,\n\t}\n\n\treturn s.ShardMgr.CreateShard(ctx, &persistence.CreateShardRequest{\n\t\tShardInfo: info,\n\t})\n}\n\n// GetShard is a utility method to get the shard using persistence layer\nfunc (s *TestBase) GetShard(ctx context.Context, shardID int) (*persistence.ShardInfo, error) {\n\tresponse, err := s.ShardMgr.GetShard(ctx, &persistence.GetShardRequest{\n\t\tShardID: shardID,\n\t})\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response.ShardInfo, nil\n}\n\n// UpdateShard is a utility method to update the shard using persistence layer\nfunc (s *TestBase) UpdateShard(ctx context.Context, updatedInfo *persistence.ShardInfo, previousRangeID int64) error {\n\treturn s.ShardMgr.UpdateShard(ctx, &persistence.UpdateShardRequest{\n\t\tShardInfo:       updatedInfo,\n\t\tPreviousRangeID: previousRangeID,\n\t})\n}\n\n// CreateWorkflowExecutionWithBranchToken test util function\nfunc (s *TestBase) CreateWorkflowExecutionWithBranchToken(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowExecution types.WorkflowExecution,\n\ttaskList string,\n\twType string,\n\twTimeout int32,\n\tdecisionTimeout int32,\n\texecutionContext []byte,\n\tnextEventID int64,\n\tlastProcessedEventID int64,\n\tdecisionScheduleID int64,\n\tbranchToken []byte,\n\ttimerTasks []persistence.Task,\n\tpartitionConfig map[string]string,\n) (*persistence.CreateWorkflowExecutionResponse, error) {\n\n\tnow := time.Now()\n\tversionHistory := persistence.NewVersionHistory(branchToken, []*persistence.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\tresponse, err := s.ExecutionManager.CreateWorkflowExecution(ctx, &persistence.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    taskList,\n\t\t\t\tWorkflowTypeName:            wType,\n\t\t\t\tWorkflowTimeout:             wTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tExecutionContext:            executionContext,\n\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\tCloseStatus:                 persistence.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tLastUpdatedTimestamp:        now,\n\t\t\t\tStartTimestamp:              now,\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tBranchToken:                 branchToken,\n\t\t\t\tPartitionConfig:             partitionConfig,\n\t\t\t},\n\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              s.GetNextSequenceNumber(),\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTargetDomainID: domainID,\n\t\t\t\t\t\tTaskList:       taskList,\n\t\t\t\t\t\tScheduleID:     decisionScheduleID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpersistence.HistoryTaskCategoryTimer: timerTasks,\n\t\t\t},\n\t\t\tChecksum:         testWorkflowChecksum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t})\n\n\treturn response, err\n}\n\n// CreateWorkflowExecution is a utility method to create workflow executions\nfunc (s *TestBase) CreateWorkflowExecution(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowExecution types.WorkflowExecution,\n\ttaskList string,\n\twType string,\n\twTimeout int32,\n\tdecisionTimeout int32,\n\texecutionContext []byte,\n\tnextEventID int64,\n\tlastProcessedEventID int64,\n\tdecisionScheduleID int64,\n\ttimerTasks []persistence.Task,\n\tpartitionConfig map[string]string,\n) (*persistence.CreateWorkflowExecutionResponse, error) {\n\n\treturn s.CreateWorkflowExecutionWithBranchToken(ctx, domainID, workflowExecution, taskList, wType, wTimeout, decisionTimeout,\n\t\texecutionContext, nextEventID, lastProcessedEventID, decisionScheduleID, nil, timerTasks, partitionConfig)\n}\n\n// CreateChildWorkflowExecution is a utility method to create child workflow executions\nfunc (s *TestBase) CreateChildWorkflowExecution(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution,\n\tparentDomainID string, parentExecution types.WorkflowExecution, initiatedID int64, taskList, wType string,\n\twTimeout int32, decisionTimeout int32, executionContext []byte, nextEventID int64, lastProcessedEventID int64,\n\tdecisionScheduleID int64, timerTasks []persistence.Task, partitionConfig map[string]string) (*persistence.CreateWorkflowExecutionResponse, error) {\n\tnow := time.Now()\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\tresponse, err := s.ExecutionManager.CreateWorkflowExecution(ctx, &persistence.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tParentDomainID:              parentDomainID,\n\t\t\t\tParentWorkflowID:            parentExecution.GetWorkflowID(),\n\t\t\t\tParentRunID:                 parentExecution.GetRunID(),\n\t\t\t\tInitiatedID:                 initiatedID,\n\t\t\t\tTaskList:                    taskList,\n\t\t\t\tWorkflowTypeName:            wType,\n\t\t\t\tWorkflowTimeout:             wTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tExecutionContext:            executionContext,\n\t\t\t\tState:                       persistence.WorkflowStateCreated,\n\t\t\t\tCloseStatus:                 persistence.WorkflowCloseStatusNone,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tLastUpdatedTimestamp:        now,\n\t\t\t\tStartTimestamp:              now,\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tPartitionConfig:             partitionConfig,\n\t\t\t},\n\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTargetDomainID: domainID,\n\t\t\t\t\t\tTaskList:       taskList,\n\t\t\t\t\t\tScheduleID:     decisionScheduleID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpersistence.HistoryTaskCategoryTimer: timerTasks,\n\t\t\t},\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:    s.ShardInfo.RangeID,\n\t\tDomainName: s.DomainManager.GetName(),\n\t})\n\n\treturn response, err\n}\n\n// GetWorkflowExecutionInfoWithStats is a utility method to retrieve execution info with size stats\nfunc (s *TestBase) GetWorkflowExecutionInfoWithStats(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution) (\n\t*persistence.MutableStateStats, *persistence.WorkflowMutableState, error) {\n\tresponse, err := s.ExecutionManager.GetWorkflowExecution(ctx, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID:  domainID,\n\t\tExecution: workflowExecution,\n\t\tRangeID:   s.ShardInfo.RangeID,\n\t})\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\treturn response.MutableStateStats, response.State, nil\n}\n\n// GetWorkflowExecutionInfo is a utility method to retrieve execution info\nfunc (s *TestBase) GetWorkflowExecutionInfo(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution) (\n\t*persistence.WorkflowMutableState, error) {\n\tresponse, err := s.ExecutionManager.GetWorkflowExecution(ctx, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID:  domainID,\n\t\tExecution: workflowExecution,\n\t\tRangeID:   s.ShardInfo.RangeID,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response.State, nil\n}\n\n// GetCurrentWorkflowRunID returns the workflow run ID for the given params\nfunc (s *TestBase) GetCurrentWorkflowRunID(ctx context.Context, domainID, workflowID string) (string, error) {\n\tresponse, err := s.ExecutionManager.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\treturn response.RunID, nil\n}\n\n// ContinueAsNewExecution is a utility method to create workflow executions\nfunc (s *TestBase) ContinueAsNewExecution(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tcondition int64,\n\tnewExecution types.WorkflowExecution,\n\tnextEventID, decisionScheduleID int64,\n\tprevResetPoints *types.ResetPoints,\n) error {\n\n\tnow := time.Now()\n\tnewdecisionTask := &persistence.DecisionTask{\n\t\tWorkflowIdentifier: p.WorkflowIdentifier{\n\t\t\tDomainID:   updatedInfo.DomainID,\n\t\t\tWorkflowID: updatedInfo.WorkflowID,\n\t\t\tRunID:      updatedInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t},\n\t\tTargetDomainID: updatedInfo.DomainID,\n\t\tTaskList:       updatedInfo.TaskList,\n\t\tScheduleID:     int64(decisionScheduleID),\n\t}\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: decisionScheduleID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\treq := &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\tnewdecisionTask,\n\t\t\t\t},\n\t\t\t},\n\t\t\tCondition:           condition,\n\t\t\tUpsertActivityInfos: nil,\n\t\t\tDeleteActivityInfos: nil,\n\t\t\tUpsertTimerInfos:    nil,\n\t\t\tDeleteTimerInfos:    nil,\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tNewWorkflowSnapshot: &persistence.WorkflowSnapshot{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    updatedInfo.DomainID,\n\t\t\t\tWorkflowID:                  newExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       newExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         updatedInfo.FirstExecutionRunID,\n\t\t\t\tTaskList:                    updatedInfo.TaskList,\n\t\t\t\tWorkflowTypeName:            updatedInfo.WorkflowTypeName,\n\t\t\t\tWorkflowTimeout:             updatedInfo.WorkflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: updatedInfo.DecisionStartToCloseTimeout,\n\t\t\t\tExecutionContext:            nil,\n\t\t\t\tState:                       updatedInfo.State,\n\t\t\t\tCloseStatus:                 updatedInfo.CloseStatus,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          constants.EmptyEventID,\n\t\t\t\tLastUpdatedTimestamp:        now,\n\t\t\t\tStartTimestamp:              now,\n\t\t\t\tDecisionScheduleID:          decisionScheduleID,\n\t\t\t\tDecisionStartedID:           constants.EmptyEventID,\n\t\t\t\tDecisionTimeout:             1,\n\t\t\t\tAutoResetPoints:             prevResetPoints,\n\t\t\t\tPartitionConfig:             updatedInfo.PartitionConfig,\n\t\t\t},\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t\t// To DO: next PR for UpdateWorkflowExecution\n\t\t// DomainName: s.DomainManager.GetName(),\n\t}\n\treq.UpdateWorkflowMutation.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\treq.UpdateWorkflowMutation.ExecutionInfo.CloseStatus = persistence.WorkflowCloseStatusContinuedAsNew\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, req)\n\treturn err\n}\n\n// UpdateWorkflowExecution is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecution(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tdecisionScheduleIDs []int64,\n\tactivityScheduleIDs []int64,\n\tcondition int64,\n\ttimerTasks []persistence.Task,\n\tupsertActivityInfos []*persistence.ActivityInfo,\n\tdeleteActivityInfos []int64,\n\tupsertTimerInfos []*persistence.TimerInfo,\n\tdeleteTimerInfos []string,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tdecisionScheduleIDs,\n\t\tactivityScheduleIDs,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\ttimerTasks,\n\t\tupsertActivityInfos,\n\t\tdeleteActivityInfos,\n\t\tupsertTimerInfos,\n\t\tdeleteTimerInfos,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// UpdateWorkflowExecutionAndFinish is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionAndFinish(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tcondition int64,\n\tversionHistories *persistence.VersionHistories,\n) error {\n\ttransferTasks := []persistence.Task{}\n\ttransferTasks = append(transferTasks, &persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   updatedInfo.DomainID,\n\t\t\tWorkflowID: updatedInfo.WorkflowID,\n\t\t\tRunID:      updatedInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{TaskID: s.GetNextSequenceNumber()},\n\t})\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t},\n\t\t\tCondition:           condition,\n\t\t\tUpsertActivityInfos: nil,\n\t\t\tDeleteActivityInfos: nil,\n\t\t\tUpsertTimerInfos:    nil,\n\t\t\tDeleteTimerInfos:    nil,\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t\t// To DO: next PR for UpdateWorkflowExecution\n\t\t// DomainName: s.DomainManager.GetName(),\n\t})\n\treturn err\n}\n\n// UpsertChildExecutionsState is a utility method to update mutable state of workflow execution\nfunc (s *TestBase) UpsertChildExecutionsState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tupsertChildInfos []*persistence.ChildExecutionInfo,\n) error {\n\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tupsertChildInfos,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// UpsertRequestCancelState is a utility method to update mutable state of workflow execution\nfunc (s *TestBase) UpsertRequestCancelState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tupsertCancelInfos []*persistence.RequestCancelInfo,\n) error {\n\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tupsertCancelInfos,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// UpsertSignalInfoState is a utility method to update mutable state of workflow execution\nfunc (s *TestBase) UpsertSignalInfoState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tupsertSignalInfos []*persistence.SignalInfo,\n) error {\n\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tupsertSignalInfos,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// UpsertSignalsRequestedState is a utility method to update mutable state of workflow execution\nfunc (s *TestBase) UpsertSignalsRequestedState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tupsertSignalsRequested []string,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tupsertSignalsRequested,\n\t\tnil,\n\t)\n}\n\n// DeleteChildExecutionsState is a utility method to delete child execution from mutable state\nfunc (s *TestBase) DeleteChildExecutionsState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tdeleteChildInfo int64,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\t[]int64{deleteChildInfo},\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// DeleteCancelState is a utility method to delete request cancel state from mutable state\nfunc (s *TestBase) DeleteCancelState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tdeleteCancelInfo int64,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\t[]int64{deleteCancelInfo},\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// DeleteSignalState is a utility method to delete request cancel state from mutable state\nfunc (s *TestBase) DeleteSignalState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tdeleteSignalInfo int64,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\t[]int64{deleteSignalInfo},\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// DeleteSignalsRequestedState is a utility method to delete mutable state of workflow execution\nfunc (s *TestBase) DeleteSignalsRequestedState(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\tdeleteSignalsRequestedIDs []string,\n) error {\n\treturn s.UpdateWorkflowExecutionWithRangeID(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tdeleteSignalsRequestedIDs,\n\t)\n}\n\n// UpdateWorklowStateAndReplication is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorklowStateAndReplication(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tcondition int64,\n\ttxTasks []persistence.Task,\n) error {\n\n\treturn s.UpdateWorkflowExecutionWithReplication(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tnil,\n\t\tnil,\n\t\ts.ShardInfo.RangeID,\n\t\tcondition,\n\t\tnil,\n\t\ttxTasks,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\n// UpdateWorkflowExecutionWithRangeID is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionWithRangeID(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tdecisionScheduleIDs []int64,\n\tactivityScheduleIDs []int64,\n\trangeID int64,\n\tcondition int64,\n\ttimerTasks []persistence.Task,\n\tupsertActivityInfos []*persistence.ActivityInfo,\n\tdeleteActivityInfos []int64,\n\tupsertTimerInfos []*persistence.TimerInfo,\n\tdeleteTimerInfos []string,\n\tupsertChildInfos []*persistence.ChildExecutionInfo,\n\tdeleteChildInfos []int64,\n\tupsertCancelInfos []*persistence.RequestCancelInfo,\n\tdeleteCancelInfos []int64,\n\tupsertSignalInfos []*persistence.SignalInfo,\n\tdeleteSignalInfos []int64,\n\tupsertSignalRequestedIDs []string,\n\tdeleteSignalRequestedIDs []string,\n) error {\n\treturn s.UpdateWorkflowExecutionWithReplication(\n\t\tctx,\n\t\tupdatedInfo,\n\t\tupdatedStats,\n\t\tupdatedVersionHistories,\n\t\tdecisionScheduleIDs,\n\t\tactivityScheduleIDs,\n\t\trangeID,\n\t\tcondition,\n\t\ttimerTasks,\n\t\t[]persistence.Task{},\n\t\tupsertActivityInfos,\n\t\tdeleteActivityInfos,\n\t\tupsertTimerInfos,\n\t\tdeleteTimerInfos,\n\t\tupsertChildInfos,\n\t\tdeleteChildInfos,\n\t\tupsertCancelInfos,\n\t\tdeleteCancelInfos,\n\t\tupsertSignalInfos,\n\t\tdeleteSignalInfos,\n\t\tupsertSignalRequestedIDs,\n\t\tdeleteSignalRequestedIDs,\n\t)\n}\n\n// UpdateWorkflowExecutionWithReplication is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionWithReplication(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tupdatedVersionHistories *persistence.VersionHistories,\n\tdecisionScheduleIDs []int64,\n\tactivityScheduleIDs []int64,\n\trangeID int64,\n\tcondition int64,\n\ttimerTasks []persistence.Task,\n\ttxTasks []persistence.Task,\n\tupsertActivityInfos []*persistence.ActivityInfo,\n\tdeleteActivityInfos []int64,\n\tupsertTimerInfos []*persistence.TimerInfo,\n\tdeleteTimerInfos []string,\n\tupsertChildInfos []*persistence.ChildExecutionInfo,\n\tdeleteChildInfos []int64,\n\tupsertCancelInfos []*persistence.RequestCancelInfo,\n\tdeleteCancelInfos []int64,\n\tupsertSignalInfos []*persistence.SignalInfo,\n\tdeleteSignalInfos []int64,\n\tupsertSignalRequestedIDs []string,\n\tdeleteSignalRequestedIDs []string,\n) error {\n\n\t// TODO: use separate fields for those three task types\n\tvar transferTasks []persistence.Task\n\tvar replicationTasks []persistence.Task\n\tfor _, task := range txTasks {\n\t\tswitch t := task.(type) {\n\t\tcase *persistence.DecisionTask,\n\t\t\t*persistence.ActivityTask,\n\t\t\t*persistence.CloseExecutionTask,\n\t\t\t*persistence.RecordWorkflowClosedTask,\n\t\t\t*persistence.RecordChildExecutionCompletedTask,\n\t\t\t*persistence.CancelExecutionTask,\n\t\t\t*persistence.StartChildExecutionTask,\n\t\t\t*persistence.SignalExecutionTask,\n\t\t\t*persistence.RecordWorkflowStartedTask,\n\t\t\t*persistence.ResetWorkflowTask,\n\t\t\t*persistence.UpsertWorkflowSearchAttributesTask:\n\t\t\ttransferTasks = append(transferTasks, t)\n\t\tcase *persistence.HistoryReplicationTask, *persistence.SyncActivityTask:\n\t\t\treplicationTasks = append(replicationTasks, t)\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"Unknown transfer task type. %v\", t))\n\t\t}\n\t}\n\tfor _, decisionScheduleID := range decisionScheduleIDs {\n\t\ttransferTasks = append(transferTasks, &persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   updatedInfo.DomainID,\n\t\t\t\tWorkflowID: updatedInfo.WorkflowID,\n\t\t\t\tRunID:      updatedInfo.RunID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t\t},\n\t\t\tTargetDomainID: updatedInfo.DomainID,\n\t\t\tTaskList:       updatedInfo.TaskList,\n\t\t\tScheduleID:     int64(decisionScheduleID)})\n\t}\n\n\tfor _, activityScheduleID := range activityScheduleIDs {\n\t\ttransferTasks = append(transferTasks, &persistence.ActivityTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   updatedInfo.DomainID,\n\t\t\t\tWorkflowID: updatedInfo.WorkflowID,\n\t\t\t\tRunID:      updatedInfo.RunID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: s.GetNextSequenceNumber(),\n\t\t\t},\n\t\t\tTargetDomainID: updatedInfo.DomainID,\n\t\t\tTaskList:       updatedInfo.TaskList,\n\t\t\tScheduleID:     int64(activityScheduleID)})\n\t}\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tRangeID: rangeID,\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tVersionHistories: updatedVersionHistories,\n\n\t\t\tUpsertActivityInfos:       upsertActivityInfos,\n\t\t\tDeleteActivityInfos:       deleteActivityInfos,\n\t\t\tUpsertTimerInfos:          upsertTimerInfos,\n\t\t\tDeleteTimerInfos:          deleteTimerInfos,\n\t\t\tUpsertChildExecutionInfos: upsertChildInfos,\n\t\t\tDeleteChildExecutionInfos: deleteChildInfos,\n\t\t\tUpsertRequestCancelInfos:  upsertCancelInfos,\n\t\t\tDeleteRequestCancelInfos:  deleteCancelInfos,\n\t\t\tUpsertSignalInfos:         upsertSignalInfos,\n\t\t\tDeleteSignalInfos:         deleteSignalInfos,\n\t\t\tUpsertSignalRequestedIDs:  upsertSignalRequestedIDs,\n\t\t\tDeleteSignalRequestedIDs:  deleteSignalRequestedIDs,\n\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer:    transferTasks,\n\t\t\t\tpersistence.HistoryTaskCategoryTimer:       timerTasks,\n\t\t\t\tpersistence.HistoryTaskCategoryReplication: replicationTasks,\n\t\t\t},\n\n\t\t\tCondition: condition,\n\t\t\tChecksum:  testWorkflowChecksum,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionTasks is a utility method to update workflow tasks\n// with IgnoreCurrent update mode.\nfunc (s *TestBase) UpdateWorkflowExecutionTasks(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tcondition int64,\n\ttransferTasks []persistence.Task,\n\ttimerTasks []persistence.Task,\n) error {\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tMode: persistence.UpdateWorkflowModeIgnoreCurrent,\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t\tpersistence.HistoryTaskCategoryTimer:    timerTasks,\n\t\t\t},\n\t\t\tCondition: condition,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionWithTransferTasks is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionWithTransferTasks(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tcondition int64,\n\ttransferTasks []persistence.Task,\n\tupsertActivityInfo []*persistence.ActivityInfo,\n\tversionHistories *persistence.VersionHistories,\n) error {\n\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t},\n\t\t\tCondition:           condition,\n\t\t\tUpsertActivityInfos: upsertActivityInfo,\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionForChildExecutionsInitiated is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionForChildExecutionsInitiated(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo, updatedStats *persistence.ExecutionStats, condition int64, transferTasks []persistence.Task, childInfos []*persistence.ChildExecutionInfo) error {\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t},\n\t\t\tCondition:                 condition,\n\t\t\tUpsertChildExecutionInfos: childInfos,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionForRequestCancel is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionForRequestCancel(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo, updatedStats *persistence.ExecutionStats, condition int64, transferTasks []persistence.Task,\n\tupsertRequestCancelInfo []*persistence.RequestCancelInfo) error {\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t},\n\t\t\tCondition:                condition,\n\t\t\tUpsertRequestCancelInfos: upsertRequestCancelInfo,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionForSignal is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionForSignal(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo, updatedStats *persistence.ExecutionStats, condition int64, transferTasks []persistence.Task,\n\tupsertSignalInfos []*persistence.SignalInfo) error {\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:  updatedInfo,\n\t\t\tExecutionStats: updatedStats,\n\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\tpersistence.HistoryTaskCategoryTransfer: transferTasks,\n\t\t\t},\n\t\t\tCondition:         condition,\n\t\t\tUpsertSignalInfos: upsertSignalInfos,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateWorkflowExecutionForBufferEvents is a utility method to update workflow execution\nfunc (s *TestBase) UpdateWorkflowExecutionForBufferEvents(\n\tctx context.Context,\n\tupdatedInfo *persistence.WorkflowExecutionInfo,\n\tupdatedStats *persistence.ExecutionStats,\n\tcondition int64,\n\tbufferEvents []*types.HistoryEvent,\n\tclearBufferedEvents bool,\n\tversionHistories *persistence.VersionHistories,\n) error {\n\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:       updatedInfo,\n\t\t\tExecutionStats:      updatedStats,\n\t\t\tNewBufferedEvents:   bufferEvents,\n\t\t\tCondition:           condition,\n\t\t\tClearBufferedEvents: clearBufferedEvents,\n\t\t\tVersionHistories:    versionHistories,\n\t\t\tChecksum:            testWorkflowChecksum,\n\t\t},\n\t\tRangeID:  s.ShardInfo.RangeID,\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// UpdateAllMutableState is a utility method to update workflow execution\nfunc (s *TestBase) UpdateAllMutableState(ctx context.Context, updatedMutableState *persistence.WorkflowMutableState, condition int64) error {\n\tvar aInfos []*persistence.ActivityInfo\n\tfor _, ai := range updatedMutableState.ActivityInfos {\n\t\taInfos = append(aInfos, ai)\n\t}\n\n\tvar tInfos []*persistence.TimerInfo\n\tfor _, ti := range updatedMutableState.TimerInfos {\n\t\ttInfos = append(tInfos, ti)\n\t}\n\n\tvar cInfos []*persistence.ChildExecutionInfo\n\tfor _, ci := range updatedMutableState.ChildExecutionInfos {\n\t\tcInfos = append(cInfos, ci)\n\t}\n\n\tvar rcInfos []*persistence.RequestCancelInfo\n\tfor _, rci := range updatedMutableState.RequestCancelInfos {\n\t\trcInfos = append(rcInfos, rci)\n\t}\n\n\tvar sInfos []*persistence.SignalInfo\n\tfor _, si := range updatedMutableState.SignalInfos {\n\t\tsInfos = append(sInfos, si)\n\t}\n\n\tvar srIDs []string\n\tfor id := range updatedMutableState.SignalRequestedIDs {\n\t\tsrIDs = append(srIDs, id)\n\t}\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:             updatedMutableState.ExecutionInfo,\n\t\t\tExecutionStats:            updatedMutableState.ExecutionStats,\n\t\t\tCondition:                 condition,\n\t\t\tUpsertActivityInfos:       aInfos,\n\t\t\tUpsertTimerInfos:          tInfos,\n\t\t\tUpsertChildExecutionInfos: cInfos,\n\t\t\tUpsertRequestCancelInfos:  rcInfos,\n\t\t\tUpsertSignalInfos:         sInfos,\n\t\t\tUpsertSignalRequestedIDs:  srIDs,\n\t\t\tVersionHistories:          updatedMutableState.VersionHistories,\n\t\t\tChecksum:                  updatedMutableState.Checksum,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// ConflictResolveWorkflowExecution is  utility method to reset mutable state\nfunc (s *TestBase) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\tinfo *persistence.WorkflowExecutionInfo,\n\tstats *persistence.ExecutionStats,\n\tnextEventID int64,\n\tactivityInfos []*persistence.ActivityInfo,\n\ttimerInfos []*persistence.TimerInfo,\n\tchildExecutionInfos []*persistence.ChildExecutionInfo,\n\trequestCancelInfos []*persistence.RequestCancelInfo,\n\tsignalInfos []*persistence.SignalInfo,\n\tids []string,\n\tversionHistories *persistence.VersionHistories,\n) error {\n\n\t_, err := s.ExecutionManager.ConflictResolveWorkflowExecution(ctx, &persistence.ConflictResolveWorkflowExecutionRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tResetWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\tExecutionInfo:       info,\n\t\t\tExecutionStats:      stats,\n\t\t\tCondition:           nextEventID,\n\t\t\tActivityInfos:       activityInfos,\n\t\t\tTimerInfos:          timerInfos,\n\t\t\tChildExecutionInfos: childExecutionInfos,\n\t\t\tRequestCancelInfos:  requestCancelInfos,\n\t\t\tSignalInfos:         signalInfos,\n\t\t\tSignalRequestedIDs:  ids,\n\t\t\tChecksum:            testWorkflowChecksum,\n\t\t\tVersionHistories:    versionHistories,\n\t\t},\n\t\tEncoding: pickRandomEncoding(),\n\t})\n\treturn err\n}\n\n// DeleteWorkflowExecution is a utility method to delete a workflow execution\nfunc (s *TestBase) DeleteWorkflowExecution(ctx context.Context, info *persistence.WorkflowExecutionInfo) error {\n\treturn s.ExecutionManager.DeleteWorkflowExecution(ctx, &persistence.DeleteWorkflowExecutionRequest{\n\t\tDomainID:   info.DomainID,\n\t\tWorkflowID: info.WorkflowID,\n\t\tRunID:      info.RunID,\n\t})\n}\n\n// DeleteCurrentWorkflowExecution is a utility method to delete the workflow current execution\nfunc (s *TestBase) DeleteCurrentWorkflowExecution(ctx context.Context, info *persistence.WorkflowExecutionInfo) error {\n\treturn s.ExecutionManager.DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\tDomainID:   info.DomainID,\n\t\tWorkflowID: info.WorkflowID,\n\t\tRunID:      info.RunID,\n\t})\n}\n\n// GetTransferTasks is a utility method to get tasks from transfer task queue\nfunc (s *TestBase) GetTransferTasks(ctx context.Context, batchSize int, getAll bool) ([]persistence.Task, error) {\n\tresult := []persistence.Task{}\n\tvar token []byte\n\nLoop:\n\tfor {\n\t\tresponse, err := s.ExecutionManager.GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(math.MaxInt64),\n\t\t\tPageSize:            batchSize,\n\t\t\tNextPageToken:       token,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttoken = response.NextPageToken\n\t\tresult = append(result, response.Tasks...)\n\t\tif len(token) == 0 || !getAll {\n\t\t\tbreak Loop\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\n// GetCrossClusterTasks is a utility method to get tasks from transfer task queue\nfunc (s *TestBase) GetCrossClusterTasks(ctx context.Context, targetCluster string, readLevel int64, batchSize int, getAll bool) ([]*persistence.CrossClusterTaskInfo, error) {\n\treturn nil, nil\n}\n\n// GetReplicationTasks is a utility method to get tasks from replication task queue\nfunc (s *TestBase) GetReplicationTasks(ctx context.Context, batchSize int, getAll bool) ([]persistence.Task, error) {\n\tresult := []persistence.Task{}\n\tvar token []byte\n\nLoop:\n\tfor {\n\t\tresponse, err := s.ExecutionManager.GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(math.MaxInt64),\n\t\t\tPageSize:            batchSize,\n\t\t\tNextPageToken:       token,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttoken = response.NextPageToken\n\t\tresult = append(result, response.Tasks...)\n\t\tif len(token) == 0 || !getAll {\n\t\t\tbreak Loop\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\n// RangeCompleteReplicationTask is a utility method to complete a range of replication tasks\nfunc (s *TestBase) RangeCompleteReplicationTask(ctx context.Context, exclusiveEndTaskID int64) error {\n\tfor {\n\t\tresp, err := s.ExecutionManager.RangeCompleteHistoryTask(ctx, &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(exclusiveEndTaskID),\n\t\t\tPageSize:            1,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, 1) {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn nil\n}\n\n// PutReplicationTaskToDLQ is a utility method to insert a replication task info\nfunc (s *TestBase) PutReplicationTaskToDLQ(\n\tctx context.Context,\n\tsourceCluster string,\n\ttaskInfo *persistence.ReplicationTaskInfo,\n) error {\n\n\treturn s.ExecutionManager.PutReplicationTaskToDLQ(ctx, &persistence.PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: sourceCluster,\n\t\tTaskInfo:          taskInfo,\n\t})\n}\n\n// GetReplicationTasksFromDLQ is a utility method to read replication task info\nfunc (s *TestBase) GetReplicationTasksFromDLQ(\n\tctx context.Context,\n\tsourceCluster string,\n\treadLevel int64,\n\tmaxReadLevel int64,\n\tpageSize int,\n\tpageToken []byte,\n) (*persistence.GetHistoryTasksResponse, error) {\n\n\treturn s.ExecutionManager.GetReplicationTasksFromDLQ(ctx, &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: sourceCluster,\n\t\tReadLevel:         readLevel,\n\t\tMaxReadLevel:      maxReadLevel,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t})\n}\n\n// GetReplicationDLQSize is a utility method to read replication dlq size\nfunc (s *TestBase) GetReplicationDLQSize(\n\tctx context.Context,\n\tsourceCluster string,\n) (*persistence.GetReplicationDLQSizeResponse, error) {\n\n\treturn s.ExecutionManager.GetReplicationDLQSize(ctx, &persistence.GetReplicationDLQSizeRequest{\n\t\tSourceClusterName: sourceCluster,\n\t})\n}\n\n// DeleteReplicationTaskFromDLQ is a utility method to delete a replication task info\nfunc (s *TestBase) DeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\tsourceCluster string,\n\ttaskID int64,\n) error {\n\n\treturn s.ExecutionManager.DeleteReplicationTaskFromDLQ(ctx, &persistence.DeleteReplicationTaskFromDLQRequest{\n\t\tSourceClusterName: sourceCluster,\n\t\tTaskID:            taskID,\n\t})\n}\n\n// RangeDeleteReplicationTaskFromDLQ is a utility method to delete  replication task info\nfunc (s *TestBase) RangeDeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\tsourceCluster string,\n\tbeginTaskID int64,\n\tendTaskID int64,\n) error {\n\n\t_, err := s.ExecutionManager.RangeDeleteReplicationTaskFromDLQ(ctx, &persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\tSourceClusterName:    sourceCluster,\n\t\tInclusiveBeginTaskID: beginTaskID,\n\t\tExclusiveEndTaskID:   endTaskID,\n\t})\n\treturn err\n}\n\n// CreateFailoverMarkers is a utility method to create failover markers\nfunc (s *TestBase) CreateFailoverMarkers(\n\tctx context.Context,\n\tmarkers []*persistence.FailoverMarkerTask,\n) error {\n\n\treturn s.ExecutionManager.CreateFailoverMarkerTasks(ctx, &persistence.CreateFailoverMarkersRequest{\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMarkers: markers,\n\t})\n}\n\n// CompleteTransferTask is a utility method to complete a transfer task\nfunc (s *TestBase) CompleteTransferTask(ctx context.Context, taskID int64) error {\n\n\treturn s.ExecutionManager.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\tTaskKey:      persistence.NewImmediateTaskKey(taskID),\n\t})\n}\n\n// RangeCompleteTransferTask is a utility method to complete a range of transfer tasks\nfunc (s *TestBase) RangeCompleteTransferTask(ctx context.Context, inclusiveBeginTaskID int64, exclusiveEndTaskID int64) error {\n\tfor {\n\t\tresp, err := s.ExecutionManager.RangeCompleteHistoryTask(ctx, &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(inclusiveBeginTaskID),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(exclusiveEndTaskID),\n\t\t\tPageSize:            1,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, 1) {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn nil\n}\n\n// CompleteCrossClusterTask is a utility method to complete a cross-cluster task\nfunc (s *TestBase) CompleteCrossClusterTask(ctx context.Context, targetCluster string, taskID int64) error {\n\treturn nil\n}\n\n// RangeCompleteCrossClusterTask is a utility method to complete a range of cross-cluster tasks\nfunc (s *TestBase) RangeCompleteCrossClusterTask(ctx context.Context, targetCluster string, exclusiveBeginTaskID int64, inclusiveEndTaskID int64) error {\n\treturn nil\n}\n\n// CompleteReplicationTask is a utility method to complete a replication task\nfunc (s *TestBase) CompleteReplicationTask(ctx context.Context, taskID int64) error {\n\n\treturn s.ExecutionManager.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\tTaskKey:      persistence.NewImmediateTaskKey(taskID),\n\t})\n}\n\n// GetTimerIndexTasks is a utility method to get tasks from transfer task queue\nfunc (s *TestBase) GetTimerIndexTasks(ctx context.Context, batchSize int, getAll bool) ([]persistence.Task, error) {\n\tresult := []persistence.Task{}\n\tvar token []byte\n\nLoop:\n\tfor {\n\t\tresponse, err := s.ExecutionManager.GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 0),\n\t\t\tPageSize:            batchSize,\n\t\t\tNextPageToken:       token,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttoken = response.NextPageToken\n\t\tresult = append(result, response.Tasks...)\n\t\tif len(token) == 0 || !getAll {\n\t\t\tbreak Loop\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\n// CompleteTimerTask is a utility method to complete a timer task\nfunc (s *TestBase) CompleteTimerTask(ctx context.Context, ts time.Time, taskID int64) error {\n\treturn s.ExecutionManager.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\tTaskKey:      persistence.NewHistoryTaskKey(ts, taskID),\n\t})\n}\n\n// RangeCompleteTimerTask is a utility method to complete a range of timer tasks\nfunc (s *TestBase) RangeCompleteTimerTask(ctx context.Context, inclusiveBeginTimestamp time.Time, exclusiveEndTimestamp time.Time) error {\n\tfor {\n\t\tresp, err := s.ExecutionManager.RangeCompleteHistoryTask(ctx, &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(inclusiveBeginTimestamp, 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(exclusiveEndTimestamp, 0),\n\t\t\tPageSize:            1,\n\t\t})\n\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, 1) {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn nil\n}\n\n// CreateDecisionTask is a utility method to create a task\nfunc (s *TestBase) CreateDecisionTask(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution, taskList string,\n\tdecisionScheduleID int64, partitionConfig map[string]string) (int64, error) {\n\tleaseResponse, err := s.TaskMgr.LeaseTaskList(ctx, &persistence.LeaseTaskListRequest{\n\t\tDomainID: domainID,\n\t\tTaskList: taskList,\n\t\tTaskType: persistence.TaskListTypeDecision,\n\t})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\t// clearing this field since when creating task in matching we don't have the LastUpdate information\n\tleaseResponse.TaskListInfo.LastUpdated = time.Time{}\n\n\ttaskID := s.GetNextSequenceNumber()\n\ttasks := []*persistence.CreateTaskInfo{\n\t\t{\n\t\t\tTaskID: taskID,\n\t\t\tData: &persistence.TaskInfo{\n\t\t\t\tDomainID:        domainID,\n\t\t\t\tWorkflowID:      workflowExecution.WorkflowID,\n\t\t\t\tRunID:           workflowExecution.RunID,\n\t\t\t\tTaskID:          taskID,\n\t\t\t\tScheduleID:      decisionScheduleID,\n\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t},\n\t\t},\n\t}\n\n\t_, err = s.TaskMgr.CreateTasks(ctx, &persistence.CreateTasksRequest{\n\t\tTaskListInfo: leaseResponse.TaskListInfo,\n\t\tTasks:        tasks,\n\t})\n\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\treturn taskID, err\n}\n\nfunc (s *TestBase) GetDecisionTaskListSize(ctx context.Context, domainID, taskList string, ackLevel int64) (int64, error) {\n\tresp, err := s.TaskMgr.GetTaskListSize(ctx, &persistence.GetTaskListSizeRequest{\n\t\tDomainID:     domainID,\n\t\tTaskListName: taskList,\n\t\tTaskListType: persistence.TaskListTypeDecision,\n\t\tAckLevel:     ackLevel,\n\t})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn resp.Size, nil\n}\n\n// CreateActivityTasks is a utility method to create tasks\nfunc (s *TestBase) CreateActivityTasks(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution,\n\tactivities map[int64]string, partitionConfig map[string]string) ([]int64, error) {\n\n\ttaskLists := make(map[string]*persistence.TaskListInfo)\n\tfor _, tl := range activities {\n\t\t_, ok := taskLists[tl]\n\t\tif !ok {\n\t\t\tresp, err := s.TaskMgr.LeaseTaskList(\n\t\t\t\tctx,\n\t\t\t\t&persistence.LeaseTaskListRequest{DomainID: domainID, TaskList: tl, TaskType: persistence.TaskListTypeActivity})\n\t\t\tif err != nil {\n\t\t\t\treturn []int64{}, err\n\t\t\t}\n\t\t\ttaskLists[tl] = resp.TaskListInfo\n\t\t\t// clearing this field since when creating task in matching we don't have the LastUpdate information\n\t\t\ttaskLists[tl].LastUpdated = time.Time{}\n\t\t}\n\t}\n\n\tvar taskIDs []int64\n\tfor activityScheduleID, taskList := range activities {\n\t\ttaskID := s.GetNextSequenceNumber()\n\t\ttasks := []*persistence.CreateTaskInfo{\n\t\t\t{\n\t\t\t\tTaskID: taskID,\n\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\tDomainID:                      domainID,\n\t\t\t\t\tWorkflowID:                    workflowExecution.WorkflowID,\n\t\t\t\t\tRunID:                         workflowExecution.RunID,\n\t\t\t\t\tTaskID:                        taskID,\n\t\t\t\t\tScheduleID:                    activityScheduleID,\n\t\t\t\t\tScheduleToStartTimeoutSeconds: defaultScheduleToStartTimeout,\n\t\t\t\t\tPartitionConfig:               partitionConfig,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\t_, err := s.TaskMgr.CreateTasks(ctx, &persistence.CreateTasksRequest{\n\t\t\tTaskListInfo: taskLists[taskList],\n\t\t\tTasks:        tasks,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ttaskIDs = append(taskIDs, taskID)\n\t}\n\n\treturn taskIDs, nil\n}\n\n// GetTasks is a utility method to get tasks from persistence\nfunc (s *TestBase) GetTasks(ctx context.Context, domainID, taskList string, taskType int, batchSize int) (*persistence.GetTasksResponse, error) {\n\tresponse, err := s.TaskMgr.GetTasks(ctx, &persistence.GetTasksRequest{\n\t\tDomainID:     domainID,\n\t\tTaskList:     taskList,\n\t\tTaskType:     taskType,\n\t\tBatchSize:    batchSize,\n\t\tMaxReadLevel: common.Int64Ptr(math.MaxInt64),\n\t})\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &persistence.GetTasksResponse{Tasks: response.Tasks}, nil\n}\n\n// CompleteTask is a utility method to complete a task\nfunc (s *TestBase) CompleteTask(ctx context.Context, domainID, taskList string, taskType int, taskID int64, ackLevel int64) error {\n\treturn s.TaskMgr.CompleteTask(ctx, &persistence.CompleteTaskRequest{\n\t\tTaskList: &persistence.TaskListInfo{\n\t\t\tDomainID: domainID,\n\t\t\tAckLevel: ackLevel,\n\t\t\tTaskType: taskType,\n\t\t\tName:     taskList,\n\t\t},\n\t\tTaskID: taskID,\n\t})\n}\n\n// TearDownWorkflowStore to cleanup\nfunc (s *TestBase) TearDownWorkflowStore() {\n\ts.ExecutionMgrFactory.Close()\n\n\ts.DefaultTestCluster.TearDownTestDatabase()\n}\n\n// GetNextSequenceNumber generates a unique sequence number for can be used for transfer queue taskId\nfunc (s *TestBase) GetNextSequenceNumber() int64 {\n\ttaskID, _ := s.TaskIDGenerator.GenerateTransferTaskID()\n\treturn taskID\n}\n\n// ClearTasks completes all transfer tasks and replication tasks\nfunc (s *TestBase) ClearTasks() {\n\ts.ClearTransferQueue()\n\ts.ClearReplicationQueue()\n}\n\n// ClearTransferQueue completes all tasks in transfer queue\nfunc (s *TestBase) ClearTransferQueue() {\n\ts.Logger.Info(\"Clearing transfer tasks\", tag.ShardRangeID(s.ShardInfo.RangeID))\n\ttasks, err := s.GetTransferTasks(context.Background(), 100, true)\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Error during cleanup\", tag.Error(err))\n\t}\n\n\tcounter := 0\n\tfor _, t := range tasks {\n\t\ts.Logger.Info(\"Deleting transfer task with ID\", tag.TaskID(t.GetTaskID()))\n\t\ts.NoError(s.CompleteTransferTask(context.Background(), t.GetTaskID()))\n\t\tcounter++\n\t}\n\n\ts.Logger.Info(\"Deleted transfer tasks.\", tag.Counter(counter))\n}\n\n// ClearReplicationQueue completes all tasks in replication queue\nfunc (s *TestBase) ClearReplicationQueue() {\n\ts.Logger.Info(\"Clearing replication tasks\", tag.ShardRangeID(s.ShardInfo.RangeID))\n\ttasks, err := s.GetReplicationTasks(context.Background(), 100, true)\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Error during cleanup\", tag.Error(err))\n\t}\n\n\tcounter := 0\n\tfor _, t := range tasks {\n\t\ts.Logger.Info(\"Deleting replication task with ID\", tag.TaskID(t.GetTaskID()))\n\t\ts.NoError(s.CompleteReplicationTask(context.Background(), t.GetTaskID()))\n\t\tcounter++\n\t}\n\n\ts.Logger.Info(\"Deleted replication tasks.\", tag.Counter(counter))\n}\n\n// EqualTimesWithPrecision assertion that two times are equal within precision\nfunc (s *TestBase) EqualTimesWithPrecision(t1, t2 time.Time, precision time.Duration) {\n\ts.True(timeComparator(t1, t2, precision),\n\t\t\"Not equal: \\n\"+\n\t\t\t\"expected: %s\\n\"+\n\t\t\t\"actual  : %s%s\", t1, t2,\n\t)\n}\n\n// EqualTimes assertion that two times are equal within two millisecond precision\nfunc (s *TestBase) EqualTimes(t1, t2 time.Time) {\n\ts.EqualTimesWithPrecision(t1, t2, TimePrecision)\n}\n\nfunc (s *TestBase) validateTimeRange(t time.Time, expectedDuration time.Duration) bool {\n\tcurrentTime := time.Now()\n\tdiff := time.Duration(currentTime.UnixNano() - t.UnixNano())\n\tif diff > expectedDuration {\n\t\ts.Logger.Info(\"Check Current time, Application time, Difference\", tag.Timestamp(t), tag.CursorTimestamp(currentTime), tag.Number(int64(diff)))\n\t\treturn false\n\t}\n\treturn true\n}\n\n// GenerateTransferTaskID helper\nfunc (g *TestTransferTaskIDGenerator) GenerateTransferTaskID() (int64, error) {\n\treturn atomic.AddInt64(&g.seqNum, 1), nil\n}\n\n// Publish is a utility method to add messages to the queue\nfunc (s *TestBase) Publish(\n\tctx context.Context,\n\tmessagePayload []byte,\n) error {\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(100 * time.Millisecond)\n\tretryPolicy.SetBackoffCoefficient(1.5)\n\tretryPolicy.SetMaximumAttempts(5)\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(func(e error) bool {\n\t\t\treturn persistence.IsTransientError(e) || isMessageIDConflictError(e)\n\t\t}),\n\t)\n\treturn throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\treturn s.DomainReplicationQueueMgr.EnqueueMessage(ctx, &persistence.EnqueueMessageRequest{\n\t\t\tMessagePayload: messagePayload,\n\t\t})\n\t})\n}\n\nfunc isMessageIDConflictError(err error) bool {\n\t_, ok := err.(*persistence.ConditionFailedError)\n\treturn ok\n}\n\n// GetReplicationMessages is a utility method to get messages from the queue\nfunc (s *TestBase) GetReplicationMessages(\n\tctx context.Context,\n\tlastMessageID int64,\n\tmaxCount int,\n) ([]*persistence.QueueMessage, error) {\n\n\tresp, err := s.DomainReplicationQueueMgr.ReadMessages(ctx, &persistence.ReadMessagesRequest{\n\t\tLastMessageID: lastMessageID,\n\t\tMaxCount:      maxCount,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp.Messages, nil\n}\n\n// UpdateAckLevel updates replication queue ack level\nfunc (s *TestBase) UpdateAckLevel(\n\tctx context.Context,\n\tlastProcessedMessageID int64,\n\tclusterName string,\n) error {\n\n\treturn s.DomainReplicationQueueMgr.UpdateAckLevel(ctx, &persistence.UpdateAckLevelRequest{\n\t\tMessageID:   lastProcessedMessageID,\n\t\tClusterName: clusterName,\n\t})\n}\n\n// GetAckLevels returns replication queue ack levels\nfunc (s *TestBase) GetAckLevels(\n\tctx context.Context,\n) (map[string]int64, error) {\n\tresp, err := s.DomainReplicationQueueMgr.GetAckLevels(ctx, &persistence.GetAckLevelsRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp.AckLevels, nil\n}\n\n// PublishToDomainDLQ is a utility method to add messages to the domain DLQ\nfunc (s *TestBase) PublishToDomainDLQ(\n\tctx context.Context,\n\tmessagePayload []byte,\n) error {\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(100 * time.Millisecond)\n\tretryPolicy.SetBackoffCoefficient(1.5)\n\tretryPolicy.SetMaximumAttempts(5)\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(func(e error) bool {\n\t\t\treturn persistence.IsTransientError(e) || isMessageIDConflictError(e)\n\t\t}),\n\t)\n\treturn throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\treturn s.DomainReplicationQueueMgr.EnqueueMessageToDLQ(ctx, &persistence.EnqueueMessageToDLQRequest{\n\t\t\tMessagePayload: messagePayload,\n\t\t})\n\t})\n}\n\n// GetMessagesFromDomainDLQ is a utility method to get messages from the domain DLQ\nfunc (s *TestBase) GetMessagesFromDomainDLQ(\n\tctx context.Context,\n\tfirstMessageID int64,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*persistence.QueueMessage, []byte, error) {\n\n\tresp, err := s.DomainReplicationQueueMgr.ReadMessagesFromDLQ(ctx, &persistence.ReadMessagesFromDLQRequest{\n\t\tFirstMessageID: firstMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t\tPageSize:       pageSize,\n\t\tPageToken:      pageToken,\n\t})\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn resp.Messages, resp.NextPageToken, nil\n}\n\n// UpdateDomainDLQAckLevel updates domain dlq ack level\nfunc (s *TestBase) UpdateDomainDLQAckLevel(\n\tctx context.Context,\n\tlastProcessedMessageID int64,\n\tclusterName string,\n) error {\n\n\treturn s.DomainReplicationQueueMgr.UpdateDLQAckLevel(ctx, &persistence.UpdateDLQAckLevelRequest{\n\t\tMessageID:   lastProcessedMessageID,\n\t\tClusterName: clusterName,\n\t})\n}\n\n// GetDomainDLQAckLevel returns domain dlq ack level\nfunc (s *TestBase) GetDomainDLQAckLevel(\n\tctx context.Context,\n) (map[string]int64, error) {\n\tresp, err := s.DomainReplicationQueueMgr.GetDLQAckLevels(ctx, &persistence.GetDLQAckLevelsRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp.AckLevels, nil\n}\n\n// GetDomainDLQSize returns domain dlq size\nfunc (s *TestBase) GetDomainDLQSize(\n\tctx context.Context,\n) (int64, error) {\n\tresp, err := s.DomainReplicationQueueMgr.GetDLQSize(ctx, &persistence.GetDLQSizeRequest{})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\treturn resp.Size, nil\n}\n\n// DeleteMessageFromDomainDLQ deletes one message from domain DLQ\nfunc (s *TestBase) DeleteMessageFromDomainDLQ(\n\tctx context.Context,\n\tmessageID int64,\n) error {\n\n\treturn s.DomainReplicationQueueMgr.DeleteMessageFromDLQ(ctx, &persistence.DeleteMessageFromDLQRequest{\n\t\tMessageID: messageID,\n\t})\n}\n\n// RangeDeleteMessagesFromDomainDLQ deletes messages from domain DLQ\nfunc (s *TestBase) RangeDeleteMessagesFromDomainDLQ(\n\tctx context.Context,\n\tfirstMessageID int64,\n\tlastMessageID int64,\n) error {\n\n\treturn s.DomainReplicationQueueMgr.RangeDeleteMessagesFromDLQ(ctx, &persistence.RangeDeleteMessagesFromDLQRequest{\n\t\tFirstMessageID: firstMessageID,\n\t\tLastMessageID:  lastMessageID,\n\t})\n}\n\n// GenerateTransferTaskIDs helper\nfunc (g *TestTransferTaskIDGenerator) GenerateTransferTaskIDs(number int) ([]int64, error) {\n\tresult := []int64{}\n\tfor i := 0; i < number; i++ {\n\t\tid, err := g.GenerateTransferTaskID()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult = append(result, id)\n\t}\n\treturn result, nil\n}\n\n// GenerateRandomDBName helper\nfunc GenerateRandomDBName(n int) string {\n\trand.Seed(time.Now().UnixNano())\n\tletterRunes := []rune(\"workflow\")\n\tb := make([]rune, n)\n\tfor i := range b {\n\t\tb[i] = letterRunes[rand.Intn(len(letterRunes))]\n\t}\n\tts := time.Now().Unix()\n\treturn fmt.Sprintf(\"%v_%v\", ts, string(b))\n}\n\nfunc pickRandomEncoding() constants.EncodingType {\n\t// randomly pick json/thriftrw/empty as encoding type\n\tvar encoding constants.EncodingType\n\ti := rand.Intn(3)\n\tswitch i {\n\tcase 0:\n\t\tencoding = constants.EncodingTypeJSON\n\tcase 1:\n\t\tencoding = constants.EncodingTypeThriftRW\n\tcase 2:\n\t\tencoding = constants.EncodingType(\"\")\n\t}\n\treturn encoding\n}\n\nfunc int64Ptr(i int64) *int64 {\n\treturn &i\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/queuePersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype (\n\t// QueuePersistenceSuite contains queue persistence tests\n\tQueuePersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\n// SetupSuite implementation\nfunc (s *QueuePersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// SetupTest implementation\nfunc (s *QueuePersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TearDownSuite implementation\nfunc (s *QueuePersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// TestDomainReplicationQueue tests domain replication queue operations\nfunc (s *QueuePersistenceSuite) TestDomainReplicationQueue() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tnumMessages := 100\n\tconcurrentSenders := 10\n\tmessageChan := make(chan []byte, numMessages)\n\tvar publishErrors []error\n\tvar mu sync.Mutex\n\n\tgo func() {\n\t\tfor i := 0; i < numMessages; i++ {\n\t\t\tmessageChan <- []byte{byte(i)}\n\t\t}\n\t\tclose(messageChan)\n\t}()\n\n\twg := sync.WaitGroup{}\n\twg.Add(concurrentSenders)\n\n\t// Helper function for publishing with retry\n\tpublishWithRetry := func(message []byte) error {\n\t\tvar lastErr error\n\t\tfor i := 0; i < 3; i++ {\n\t\t\terr := s.Publish(ctx, message)\n\t\t\tif err == nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tlastErr = err\n\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t}\n\t\treturn lastErr\n\t}\n\n\t// Concurrent publishing\n\tfor i := 0; i < concurrentSenders; i++ {\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor message := range messageChan {\n\t\t\t\terr := publishWithRetry(message)\n\t\t\t\tif err != nil {\n\t\t\t\t\tmu.Lock()\n\t\t\t\t\tpublishErrors = append(publishErrors, err)\n\t\t\t\t\tmu.Unlock()\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\n\twg.Wait()\n\n\tresult, err := s.GetReplicationMessages(ctx, -1, numMessages)\n\ts.Nil(err, \"GetReplicationMessages failed.\")\n\ts.Len(result, numMessages, \"Expected %d messages, got %d\", numMessages, len(result))\n\n\t// Verify message content\n\tmessageSet := make(map[byte]bool)\n\tfor _, msg := range result {\n\t\tmessageSet[msg.Payload[0]] = true\n\t}\n\ts.Len(messageSet, numMessages, \"Expected %d unique messages, got %d\", numMessages, len(messageSet))\n}\n\n// TestQueueMetadataOperations tests queue metadata operations\nfunc (s *QueuePersistenceSuite) TestQueueMetadataOperations() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tclusterAckLevels, err := s.GetAckLevels(ctx)\n\ts.Require().NoError(err)\n\ts.Assert().Len(clusterAckLevels, 0)\n\n\terr = s.UpdateAckLevel(ctx, 10, \"test1\")\n\ts.Require().NoError(err)\n\n\tclusterAckLevels, err = s.GetAckLevels(ctx)\n\ts.Require().NoError(err)\n\ts.Assert().Len(clusterAckLevels, 1)\n\ts.Assert().Equal(int64(10), clusterAckLevels[\"test1\"])\n\n\terr = s.UpdateAckLevel(ctx, 20, \"test1\")\n\ts.Require().NoError(err)\n\n\tclusterAckLevels, err = s.GetAckLevels(ctx)\n\ts.Require().NoError(err)\n\ts.Assert().Len(clusterAckLevels, 1)\n\ts.Assert().Equal(int64(20), clusterAckLevels[\"test1\"])\n\n\terr = s.UpdateAckLevel(ctx, 25, \"test2\")\n\ts.Require().NoError(err)\n\n\terr = s.UpdateAckLevel(ctx, 24, \"test2\")\n\ts.Require().NoError(err)\n\n\tclusterAckLevels, err = s.GetAckLevels(ctx)\n\ts.Require().NoError(err)\n\ts.Assert().Len(clusterAckLevels, 2)\n\ts.Assert().Equal(int64(20), clusterAckLevels[\"test1\"])\n\ts.Assert().Equal(int64(25), clusterAckLevels[\"test2\"])\n}\n\n// TestDomainReplicationDLQ tests domain DLQ operations\nfunc (s *QueuePersistenceSuite) TestDomainReplicationDLQ() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tmaxMessageID := int64(100)\n\tnumMessages := 100\n\tconcurrentSenders := 10\n\tmessageChan := make(chan []byte, numMessages) // Buffered channel\n\tvar publishErrors []error\n\tvar mu sync.Mutex\n\n\tgo func() {\n\t\tfor i := 0; i < numMessages; i++ {\n\t\t\tmessageChan <- []byte{}\n\t\t}\n\t\tclose(messageChan)\n\t}()\n\n\twg := sync.WaitGroup{}\n\twg.Add(concurrentSenders)\n\n\t// Helper function for publishing with retry\n\tpublishWithRetry := func(message []byte) error {\n\t\tvar lastErr error\n\t\tfor i := 0; i < 3; i++ {\n\t\t\terr := s.PublishToDomainDLQ(ctx, message)\n\t\t\tif err == nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tlastErr = err\n\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t}\n\t\treturn lastErr\n\t}\n\n\t// Concurrent publishing\n\tfor i := 0; i < concurrentSenders; i++ {\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor message := range messageChan {\n\t\t\t\terr := publishWithRetry(message)\n\t\t\t\tif err != nil {\n\t\t\t\t\tmu.Lock()\n\t\t\t\t\tpublishErrors = append(publishErrors, err)\n\t\t\t\t\tmu.Unlock()\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\n\twg.Wait()\n\n\t// Check for any publish errors\n\ts.Empty(publishErrors, \"Some messages failed to publish\")\n\n\t// Verify initial message count\n\tsize, err := s.GetDomainDLQSize(ctx)\n\ts.NoError(err, \"GetDomainDLQSize failed\")\n\ts.Equal(int64(numMessages), size, \"Unexpected initial message count\")\n\tresult1, token, err := s.GetMessagesFromDomainDLQ(ctx, -1, maxMessageID, numMessages/2, nil)\n\ts.Nil(err, \"GetReplicationMessages failed.\")\n\ts.NotNil(token)\n\tresult2, token, err := s.GetMessagesFromDomainDLQ(ctx, -1, maxMessageID, numMessages, token)\n\ts.Nil(err, \"GetReplicationMessages failed.\")\n\ts.Equal(len(token), 0)\n\ts.Equal(len(result1)+len(result2), numMessages, \"Total messages retrieved mismatch\")\n\n\t// Verify all messages were retrieved\n\t_, _, err = s.GetMessagesFromDomainDLQ(ctx, -1, 1<<63-1, numMessages, nil)\n\ts.NoError(err, \"GetReplicationMessages failed.\")\n\ts.Equal(len(token), 0)\n\n\t// Verify message count after retrieval\n\tsize, err = s.GetDomainDLQSize(ctx)\n\ts.NoError(err, \"GetDomainDLQSize failed\")\n\ts.Equal(int64(numMessages), size, \"Message count changed after retrieval\")\n\n\t// Delete last message\n\tlastMessageID := result2[len(result2)-1].ID\n\terr = s.DeleteMessageFromDomainDLQ(ctx, lastMessageID)\n\ts.NoError(err, \"Failed to delete message\")\n\n\t// Verify message count after deletion\n\tsize, err = s.GetDomainDLQSize(ctx)\n\ts.NoError(err, \"GetDomainDLQSize failed\")\n\ts.Equal(int64(numMessages-1), size, \"Message count incorrect after deletion\")\n\n\t// Get messages after deletion\n\tresult3, token, err := s.GetMessagesFromDomainDLQ(ctx, -1, maxMessageID, numMessages, token)\n\ts.Nil(err, \"GetReplicationMessages failed.\")\n\ts.Equal(len(token), 0)\n\ts.Equal(len(result3), numMessages-1, \"Unexpected number of messages after deletion\")\n\n\t// Range delete remaining messages\n\terr = s.RangeDeleteMessagesFromDomainDLQ(ctx, -1, lastMessageID)\n\ts.NoError(err, \"Failed to range delete messages\")\n\n\t// Verify final message count\n\tsize, err = s.GetDomainDLQSize(ctx)\n\ts.NoError(err, \"GetDomainDLQSize failed\")\n\ts.Equal(int64(0), size, \"Messages not fully deleted\")\n\n\t// Verify no messages remain\n\tresult4, token, err := s.GetMessagesFromDomainDLQ(ctx, -1, maxMessageID, numMessages, token)\n\ts.Nil(err, \"GetReplicationMessages failed.\")\n\ts.Equal(len(token), 0)\n\ts.Equal(len(result4), 0, \"Messages still exist after range deletion\")\n}\n\n// TestDomainDLQMetadataOperations tests queue metadata operations\nfunc (s *QueuePersistenceSuite) TestDomainDLQMetadataOperations() {\n\tclusterName := \"test\"\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tackLevel, err := s.GetDomainDLQAckLevel(ctx)\n\ts.Require().NoError(err)\n\ts.Equal(0, len(ackLevel))\n\n\terr = s.UpdateDomainDLQAckLevel(ctx, 10, clusterName)\n\ts.NoError(err)\n\n\tackLevel, err = s.GetDomainDLQAckLevel(ctx)\n\ts.Require().NoError(err)\n\ts.Equal(int64(10), ackLevel[clusterName])\n\n\terr = s.UpdateDomainDLQAckLevel(ctx, 1, clusterName)\n\ts.NoError(err)\n\n\tackLevel, err = s.GetDomainDLQAckLevel(ctx)\n\ts.Require().NoError(err)\n\ts.Equal(int64(10), ackLevel[clusterName])\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/shardPersistenceTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ShardPersistenceSuite contains shard persistence tests\n\tShardPersistenceSuite struct {\n\t\t*TestBase\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\n// SetupSuite implementation\nfunc (s *ShardPersistenceSuite) SetupSuite() {\n\tif testing.Verbose() {\n\t\tlog.SetOutput(os.Stdout)\n\t}\n}\n\n// SetupTest implementation\nfunc (s *ShardPersistenceSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// TearDownSuite implementation\nfunc (s *ShardPersistenceSuite) TearDownSuite() {\n\ts.TearDownWorkflowStore()\n}\n\n// TestCreateShard test\nfunc (s *ShardPersistenceSuite) TestCreateShard() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\terr0 := s.CreateShard(ctx, 19, \"test_create_shard1\", 123)\n\ts.Nil(err0, \"No error expected.\")\n\n\terr1 := s.CreateShard(ctx, 19, \"test_create_shard2\", 124)\n\ts.NotNil(err1, \"expected non nil error.\")\n\ts.IsType(&p.ShardAlreadyExistError{}, err1)\n\ts.T().Logf(\"CreateShard failed with error: %v\\n\", err1)\n}\n\n// TestGetShard test\nfunc (s *ShardPersistenceSuite) TestGetShard() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tshardID := 20\n\towner := \"test_get_shard\"\n\trangeID := int64(131)\n\terr0 := s.CreateShard(ctx, shardID, owner, rangeID)\n\ts.Nil(err0, \"No error expected.\")\n\n\tshardInfo, err1 := s.GetShard(ctx, shardID)\n\ts.Nil(err1)\n\ts.NotNil(shardInfo)\n\ts.Equal(shardID, shardInfo.ShardID)\n\ts.Equal(owner, shardInfo.Owner)\n\ts.Equal(rangeID, shardInfo.RangeID)\n\ts.Equal(0, shardInfo.StolenSinceRenew)\n\n\t_, err2 := s.GetShard(ctx, 4766)\n\ts.NotNil(err2)\n\ts.IsType(&types.EntityNotExistsError{}, err2)\n\ts.T().Logf(\"GetShard failed with error: %v\\n\", err2)\n}\n\n// TestUpdateShard test\nfunc (s *ShardPersistenceSuite) TestUpdateShard() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tshardID := 30\n\towner := \"test_update_shard\"\n\trangeID := int64(141)\n\terr0 := s.CreateShard(ctx, shardID, owner, rangeID)\n\ts.Nil(err0, \"No error expected.\")\n\n\tshardInfo, err1 := s.GetShard(ctx, shardID)\n\ts.Nil(err1)\n\ts.NotNil(shardInfo)\n\ts.Equal(shardID, shardInfo.ShardID)\n\ts.Equal(owner, shardInfo.Owner)\n\ts.Equal(rangeID, shardInfo.RangeID)\n\ts.Equal(0, shardInfo.StolenSinceRenew)\n\n\tupdatedOwner := \"updatedOwner\"\n\tupdatedRangeID := int64(142)\n\tupdatedCurrentClusterTransferAckLevel := int64(1000)\n\tupdatedAlternativeClusterTransferAckLevel := int64(2000)\n\tupdatedCurrentClusterTimerAckLevel := time.Now()\n\tupdatedAlternativeClusterTimerAckLevel := updatedCurrentClusterTimerAckLevel.Add(time.Minute)\n\tupdatedReplicationAckLevel := int64(2000)\n\tupdatedAlternativeClusterDLQAckLevel := int64(100)\n\tupdatedStolenSinceRenew := 10\n\n\tupdatedInfo := shardInfo.ToNilSafeCopy()\n\tupdatedInfo.Owner = updatedOwner\n\tupdatedInfo.RangeID = updatedRangeID\n\tupdatedInfo.TransferAckLevel = updatedCurrentClusterTransferAckLevel\n\tupdatedInfo.ClusterTransferAckLevel = map[string]int64{\n\t\tcluster.TestCurrentClusterName:     updatedCurrentClusterTransferAckLevel,\n\t\tcluster.TestAlternativeClusterName: updatedAlternativeClusterTransferAckLevel,\n\t}\n\tupdatedInfo.TransferProcessingQueueStates = createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, updatedCurrentClusterTransferAckLevel,\n\t\tcluster.TestAlternativeClusterName, 1, updatedAlternativeClusterTransferAckLevel,\n\t)\n\tupdatedInfo.TimerAckLevel = updatedCurrentClusterTimerAckLevel\n\tupdatedInfo.ClusterTimerAckLevel = map[string]time.Time{\n\t\tcluster.TestCurrentClusterName:     updatedCurrentClusterTimerAckLevel,\n\t\tcluster.TestAlternativeClusterName: updatedAlternativeClusterTimerAckLevel,\n\t}\n\tupdatedInfo.TimerProcessingQueueStates = createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, updatedCurrentClusterTimerAckLevel.UnixNano(),\n\t\tcluster.TestAlternativeClusterName, 1, updatedAlternativeClusterTimerAckLevel.UnixNano(),\n\t)\n\tupdatedInfo.ReplicationDLQAckLevel = map[string]int64{\n\t\tcluster.TestAlternativeClusterName: updatedAlternativeClusterDLQAckLevel,\n\t}\n\tupdatedInfo.QueueStates = map[int32]*types.QueueState{\n\t\tp.HistoryTaskCategoryIDTransfer: &types.QueueState{\n\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t0: &types.VirtualQueueState{\n\t\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\t\tTaskID: updatedCurrentClusterTransferAckLevel + 1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\t\tTaskID: updatedCurrentClusterTransferAckLevel + 10,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tPredicate: &types.Predicate{\n\t\t\t\t\t\t\t\tPredicateType: types.PredicateTypeDomainID,\n\t\t\t\t\t\t\t\tDomainIDPredicateAttributes: &types.DomainIDPredicateAttributes{\n\t\t\t\t\t\t\t\t\tDomainIDs:   []string{\"domain1\", \"domain2\"},\n\t\t\t\t\t\t\t\t\tIsExclusive: common.Ptr(true),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\t\tTaskID: updatedCurrentClusterTransferAckLevel + 10,\n\t\t\t},\n\t\t},\n\t}\n\tupdatedInfo.ReplicationAckLevel = updatedReplicationAckLevel\n\tupdatedInfo.StolenSinceRenew = updatedStolenSinceRenew\n\n\terr2 := s.UpdateShard(ctx, updatedInfo, shardInfo.RangeID)\n\ts.Nil(err2)\n\n\tinfo1, err3 := s.GetShard(ctx, shardID)\n\ts.Nil(err3)\n\ts.NotNil(info1)\n\ts.Equal(updatedOwner, info1.Owner)\n\ts.Equal(updatedRangeID, info1.RangeID)\n\ts.Equal(updatedCurrentClusterTransferAckLevel, info1.TransferAckLevel)\n\ts.Equal(updatedInfo.ClusterTransferAckLevel, info1.ClusterTransferAckLevel)\n\ts.Equal(updatedInfo.TransferProcessingQueueStates, info1.TransferProcessingQueueStates)\n\ts.EqualTimes(updatedCurrentClusterTimerAckLevel, info1.TimerAckLevel)\n\ts.EqualTimes(updatedCurrentClusterTimerAckLevel, info1.ClusterTimerAckLevel[cluster.TestCurrentClusterName])\n\ts.EqualTimes(updatedAlternativeClusterTimerAckLevel, info1.ClusterTimerAckLevel[cluster.TestAlternativeClusterName])\n\ts.Equal(updatedInfo.TimerProcessingQueueStates, info1.TimerProcessingQueueStates)\n\ts.Equal(updatedReplicationAckLevel, info1.ReplicationAckLevel)\n\ts.Equal(updatedInfo.ReplicationDLQAckLevel, info1.ReplicationDLQAckLevel)\n\ts.Equal(updatedStolenSinceRenew, info1.StolenSinceRenew)\n\n\tif s.DynamicConfiguration.ReadNoSQLShardFromDataBlob() || s.ShardMgr.GetName() != \"shardedNosql\" {\n\t\ts.Equal(updatedInfo.QueueStates, info1.QueueStates)\n\t}\n\n\tfailedUpdateInfo := shardInfo.ToNilSafeCopy()\n\tfailedUpdateInfo.Owner = \"failed_owner\"\n\tfailedUpdateInfo.TransferAckLevel = int64(4000)\n\tfailedUpdateInfo.ReplicationAckLevel = int64(5000)\n\terr4 := s.UpdateShard(ctx, failedUpdateInfo, shardInfo.RangeID)\n\ts.NotNil(err4)\n\ts.IsType(&p.ShardOwnershipLostError{}, err4)\n\ts.T().Logf(\"Update shard failed with error: %v\\n\", err4)\n\n\tinfo2, err5 := s.GetShard(ctx, shardID)\n\ts.Nil(err5)\n\ts.NotNil(info2)\n\ts.Equal(updatedOwner, info2.Owner)\n\ts.Equal(updatedRangeID, info2.RangeID)\n\ts.Equal(updatedCurrentClusterTransferAckLevel, info2.TransferAckLevel)\n\ts.Equal(updatedInfo.ClusterTransferAckLevel, info2.ClusterTransferAckLevel)\n\ts.Equal(updatedInfo.TransferProcessingQueueStates, info2.TransferProcessingQueueStates)\n\ts.EqualTimes(updatedCurrentClusterTimerAckLevel, info2.TimerAckLevel)\n\ts.EqualTimes(updatedCurrentClusterTimerAckLevel, info2.ClusterTimerAckLevel[cluster.TestCurrentClusterName])\n\ts.EqualTimes(updatedAlternativeClusterTimerAckLevel, info2.ClusterTimerAckLevel[cluster.TestAlternativeClusterName])\n\ts.Equal(updatedInfo.TimerProcessingQueueStates, info2.TimerProcessingQueueStates)\n\ts.Equal(updatedReplicationAckLevel, info2.ReplicationAckLevel)\n\ts.Equal(updatedInfo.ReplicationDLQAckLevel, info2.ReplicationDLQAckLevel)\n\ts.Equal(updatedStolenSinceRenew, info2.StolenSinceRenew)\n\n\tif s.DynamicConfiguration.ReadNoSQLShardFromDataBlob() || s.ShardMgr.GetName() != \"shardedNosql\" {\n\t\ts.Equal(updatedInfo.QueueStates, info2.QueueStates)\n\t}\n}\n\nfunc (s *ShardPersistenceSuite) TestCreateGetShardBackfill() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tshardID := 4\n\trangeID := int64(59)\n\n\t// test create && get\n\tcurrentReplicationAck := int64(27)\n\tcurrentClusterTransferAck := int64(21)\n\tcurrentClusterTimerAck := timestampConvertor(time.Now().Add(-10 * time.Second))\n\tshardInfo := &p.ShardInfo{\n\t\tShardID:                 shardID,\n\t\tOwner:                   \"some random owner\",\n\t\tRangeID:                 rangeID,\n\t\tStolenSinceRenew:        12,\n\t\tUpdatedAt:               timestampConvertor(time.Now()),\n\t\tReplicationAckLevel:     currentReplicationAck,\n\t\tTransferAckLevel:        currentClusterTransferAck,\n\t\tTimerAckLevel:           currentClusterTimerAck,\n\t\tClusterReplicationLevel: map[string]int64{},\n\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t}\n\tcreateRequest := &p.CreateShardRequest{\n\t\tShardInfo: shardInfo,\n\t}\n\n\ts.Nil(s.ShardMgr.CreateShard(ctx, createRequest))\n\n\t// ClusterTransfer/TimerAckLevel will be backfilled if not exists when getting shard\n\tcurrentClusterName := s.ClusterMetadata.GetCurrentClusterName()\n\tshardInfo.ClusterTransferAckLevel = map[string]int64{\n\t\tcurrentClusterName: currentClusterTransferAck,\n\t}\n\tshardInfo.ClusterTimerAckLevel = map[string]time.Time{\n\t\tcurrentClusterName: currentClusterTimerAck,\n\t}\n\tresp, err := s.GetShard(ctx, shardID)\n\ts.NoError(err)\n\ts.EqualTimes(shardInfo.UpdatedAt, resp.UpdatedAt)\n\ts.EqualTimes(shardInfo.TimerAckLevel, resp.TimerAckLevel)\n\ts.EqualTimes(shardInfo.ClusterTimerAckLevel[currentClusterName], resp.ClusterTimerAckLevel[currentClusterName])\n\n\tresp.TimerAckLevel = shardInfo.TimerAckLevel\n\tresp.UpdatedAt = shardInfo.UpdatedAt\n\tresp.ClusterTimerAckLevel = shardInfo.ClusterTimerAckLevel\n\ts.Equal(shardInfo, resp)\n}\n\nfunc (s *ShardPersistenceSuite) TestCreateGetUpdateGetShard() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\tshardID := 8\n\trangeID := int64(59)\n\n\t// test create && get\n\tcurrentReplicationAck := int64(27)\n\tcurrentClusterTransferAck := int64(21)\n\talternativeClusterTransferAck := int64(32)\n\tcurrentClusterTimerAck := timestampConvertor(time.Now().Add(-10 * time.Second))\n\talternativeClusterTimerAck := timestampConvertor(time.Now().Add(-20 * time.Second))\n\tdomainNotificationVersion := int64(8192)\n\ttransferPQS := createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, currentClusterTransferAck,\n\t\tcluster.TestAlternativeClusterName, 1, alternativeClusterTransferAck,\n\t)\n\ttimerPQS := createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, currentClusterTimerAck.UnixNano(),\n\t\tcluster.TestAlternativeClusterName, 1, alternativeClusterTimerAck.UnixNano(),\n\t)\n\tshardInfo := &p.ShardInfo{\n\t\tShardID:             shardID,\n\t\tOwner:               \"some random owner\",\n\t\tRangeID:             rangeID,\n\t\tStolenSinceRenew:    12,\n\t\tUpdatedAt:           timestampConvertor(time.Now()),\n\t\tReplicationAckLevel: currentReplicationAck,\n\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t},\n\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t},\n\t\tTransferProcessingQueueStates: transferPQS,\n\t\tTimerProcessingQueueStates:    timerPQS,\n\t\tDomainNotificationVersion:     domainNotificationVersion,\n\t\tClusterReplicationLevel:       map[string]int64{},\n\t\tReplicationDLQAckLevel:        map[string]int64{},\n\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\tp.HistoryTaskCategoryIDTransfer: &types.QueueState{\n\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t0: &types.VirtualQueueState{\n\t\t\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: currentClusterTransferAck + 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: currentClusterTransferAck + 10,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tPredicate: &types.Predicate{\n\t\t\t\t\t\t\t\t\tPredicateType: types.PredicateTypeDomainID,\n\t\t\t\t\t\t\t\t\tDomainIDPredicateAttributes: &types.DomainIDPredicateAttributes{\n\t\t\t\t\t\t\t\t\t\tDomainIDs:   []string{\"domain1\", \"domain2\"},\n\t\t\t\t\t\t\t\t\t\tIsExclusive: common.Ptr(true),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\t\t\tTaskID: currentClusterTransferAck + 10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tcreateRequest := &p.CreateShardRequest{\n\t\tShardInfo: shardInfo,\n\t}\n\ts.Nil(s.ShardMgr.CreateShard(ctx, createRequest))\n\tresp, err := s.GetShard(ctx, shardID)\n\ts.NoError(err)\n\ts.EqualTimes(shardInfo.UpdatedAt, resp.UpdatedAt)\n\ts.EqualTimes(shardInfo.TimerAckLevel, resp.TimerAckLevel)\n\ts.EqualTimes(shardInfo.ClusterTimerAckLevel[cluster.TestCurrentClusterName], resp.ClusterTimerAckLevel[cluster.TestCurrentClusterName])\n\ts.EqualTimes(shardInfo.ClusterTimerAckLevel[cluster.TestAlternativeClusterName], resp.ClusterTimerAckLevel[cluster.TestAlternativeClusterName])\n\n\tresp.TimerAckLevel = shardInfo.TimerAckLevel\n\tresp.UpdatedAt = shardInfo.UpdatedAt\n\tresp.ClusterTimerAckLevel = shardInfo.ClusterTimerAckLevel\n\tif !s.DynamicConfiguration.ReadNoSQLHistoryTaskFromDataBlob() && s.ShardMgr.GetName() == \"shardedNosql\" {\n\t\tresp.QueueStates = shardInfo.QueueStates\n\t}\n\ts.Equal(shardInfo, resp)\n\n\t// test update && get\n\tcurrentReplicationAck = int64(270)\n\tcurrentClusterTransferAck = int64(210)\n\talternativeClusterTransferAck = int64(320)\n\tcurrentClusterTimerAck = timestampConvertor(time.Now().Add(-100 * time.Second))\n\talternativeClusterTimerAck = timestampConvertor(time.Now().Add(-200 * time.Second))\n\tdomainNotificationVersion = int64(16384)\n\ttransferPQS = createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, currentClusterTransferAck,\n\t\tcluster.TestAlternativeClusterName, 1, alternativeClusterTransferAck,\n\t)\n\ttimerPQS = createProcessingQueueStates(\n\t\tcluster.TestCurrentClusterName, 0, currentClusterTimerAck.UnixNano(),\n\t\tcluster.TestAlternativeClusterName, 1, alternativeClusterTimerAck.UnixNano(),\n\t)\n\tshardInfo = &p.ShardInfo{\n\t\tShardID:             shardID,\n\t\tOwner:               \"some random owner\",\n\t\tRangeID:             int64(28),\n\t\tStolenSinceRenew:    4,\n\t\tUpdatedAt:           timestampConvertor(time.Now()),\n\t\tReplicationAckLevel: currentReplicationAck,\n\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t},\n\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t},\n\t\tTransferProcessingQueueStates: transferPQS,\n\t\tTimerProcessingQueueStates:    timerPQS,\n\t\tDomainNotificationVersion:     domainNotificationVersion,\n\t\tClusterReplicationLevel:       map[string]int64{cluster.TestAlternativeClusterName: 12345},\n\t\tReplicationDLQAckLevel:        map[string]int64{},\n\t}\n\tupdateRequest := &p.UpdateShardRequest{\n\t\tShardInfo:       shardInfo,\n\t\tPreviousRangeID: rangeID,\n\t}\n\ts.Nil(s.ShardMgr.UpdateShard(ctx, updateRequest))\n\n\tresp, err = s.GetShard(ctx, shardID)\n\ts.NoError(err)\n\ts.EqualTimes(shardInfo.UpdatedAt, resp.UpdatedAt)\n\ts.EqualTimes(shardInfo.TimerAckLevel, resp.TimerAckLevel)\n\ts.EqualTimes(shardInfo.ClusterTimerAckLevel[cluster.TestCurrentClusterName], resp.ClusterTimerAckLevel[cluster.TestCurrentClusterName])\n\ts.EqualTimes(shardInfo.ClusterTimerAckLevel[cluster.TestAlternativeClusterName], resp.ClusterTimerAckLevel[cluster.TestAlternativeClusterName])\n\n\tresp.UpdatedAt = shardInfo.UpdatedAt\n\tresp.TimerAckLevel = shardInfo.TimerAckLevel\n\tresp.ClusterTimerAckLevel = shardInfo.ClusterTimerAckLevel\n\ts.Equal(shardInfo, resp)\n}\n\nfunc createProcessingQueueStates(\n\tcluster1 string, level1 int32, ackLevel1 int64,\n\tcluster2 string, level2 int32, ackLevel2 int64,\n) *types.ProcessingQueueStates {\n\tdomainFilter := &types.DomainFilter{\n\t\tDomainIDs:    nil,\n\t\tReverseMatch: true,\n\t}\n\tprocessingQueueStateMap := map[string][]*types.ProcessingQueueState{}\n\tif len(cluster1) != 0 {\n\t\tprocessingQueueStateMap[cluster1] = []*types.ProcessingQueueState{\n\t\t\t{\n\t\t\t\tLevel:        common.Int32Ptr(level1),\n\t\t\t\tAckLevel:     common.Int64Ptr(ackLevel1),\n\t\t\t\tMaxLevel:     common.Int64Ptr(ackLevel1),\n\t\t\t\tDomainFilter: domainFilter,\n\t\t\t},\n\t\t}\n\t}\n\tif len(cluster2) != 0 {\n\t\tprocessingQueueStateMap[cluster2] = []*types.ProcessingQueueState{\n\t\t\t{\n\t\t\t\tLevel:        common.Int32Ptr(level2),\n\t\t\t\tAckLevel:     common.Int64Ptr(ackLevel2),\n\t\t\t\tMaxLevel:     common.Int64Ptr(ackLevel2),\n\t\t\t\tDomainFilter: domainFilter,\n\t\t\t},\n\t\t}\n\t}\n\n\treturn &types.ProcessingQueueStates{StatesByCluster: processingQueueStateMap}\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/shared_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestGarbageCleanupInfo(t *testing.T) {\n\tdomainID := \"10000000-5000-f000-f000-000000000000\"\n\tworkflowID := \"workflow-id\"\n\trunID := \"10000000-5000-f000-f000-000000000002\"\n\n\tinfo := persistence.BuildHistoryGarbageCleanupInfo(domainID, workflowID, runID)\n\tdomainID2, workflowID2, runID2, err := persistence.SplitHistoryGarbageCleanupInfo(info)\n\tif err != nil || domainID != domainID2 || workflowID != workflowID2 || runID != runID2 {\n\t\tt.Fail()\n\t}\n}\n\nfunc TestGarbageCleanupInfo_WithColonInWorklfowID(t *testing.T) {\n\tdomainID := \"10000000-5000-f000-f000-000000000000\"\n\tworkflowID := \"workflow-id:2\"\n\trunID := \"10000000-5000-f000-f000-000000000002\"\n\n\tinfo := persistence.BuildHistoryGarbageCleanupInfo(domainID, workflowID, runID)\n\tdomainID2, workflowID2, runID2, err := persistence.SplitHistoryGarbageCleanupInfo(info)\n\tif err != nil || domainID != domainID2 || workflowID != workflowID2 || runID != runID2 {\n\t\tt.Fail()\n\t}\n}\n"
  },
  {
    "path": "common/persistence/persistence-tests/testcluster/interfaces.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testcluster\n\nimport \"github.com/uber/cadence/common/config\"\n\ntype (\n\t// PersistenceTestCluster exposes management operations on a database\n\t// NOTE: Putting this interface in separate package to avoid cycle dependency\n\tPersistenceTestCluster interface {\n\t\tSetupTestDatabase()\n\t\tTearDownTestDatabase()\n\t\tConfig() config.Persistence\n\t}\n)\n"
  },
  {
    "path": "common/persistence/persistence-tests/visibilitySamplingClient_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistencetests\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/sampled\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype VisibilitySamplingSuite struct {\n\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\tsuite.Suite\n\tclient       p.VisibilityManager\n\tpersistence  *mocks.VisibilityManager\n\tmetricClient *mmocks.Client\n}\n\nvar (\n\ttestDomainUUID        = \"fb15e4b5-356f-466d-8c6d-a29223e5c536\"\n\ttestDomain            = \"test-domain-name\"\n\ttestWorkflowExecution = types.WorkflowExecution{\n\t\tWorkflowID: \"visibility-workflow-test\",\n\t\tRunID:      \"843f6fc7-102a-4c63-a2d4-7c653b01bf52\",\n\t}\n\ttestWorkflowTypeName = \"visibility-workflow\"\n\n\tlistErrMsg = \"Persistence Max QPS Reached for List Operations.\"\n)\n\nfunc TestVisibilitySamplingSuite(t *testing.T) {\n\tsuite.Run(t, new(VisibilitySamplingSuite))\n}\n\nfunc (s *VisibilitySamplingSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\n\ts.persistence = &mocks.VisibilityManager{}\n\tconfig := &sampled.Config{\n\t\tVisibilityOpenMaxQPS:   dynamicproperties.GetIntPropertyFilteredByDomain(1),\n\t\tVisibilityClosedMaxQPS: dynamicproperties.GetIntPropertyFilteredByDomain(10),\n\t\tVisibilityListMaxQPS:   dynamicproperties.GetIntPropertyFilteredByDomain(1),\n\t}\n\ts.metricClient = &mmocks.Client{}\n\ts.client = sampled.NewVisibilityManager(s.persistence, sampled.Params{\n\t\tConfig:                 config,\n\t\tMetricClient:           s.metricClient,\n\t\tLogger:                 testlogger.New(s.T()),\n\t\tTimeSource:             clock.NewRealTimeSource(),\n\t\tRateLimiterFactoryFunc: sampled.NewDomainToBucketMap,\n\t})\n}\n\nfunc (s *VisibilitySamplingSuite) TearDownTest() {\n\ts.persistence.AssertExpectations(s.T())\n\ts.metricClient.AssertExpectations(s.T())\n}\n\nfunc (s *VisibilitySamplingSuite) TestRecordWorkflowExecutionStarted() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\trequest := &p.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tDomain:           testDomain,\n\t\tExecution:        testWorkflowExecution,\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\tStartTimestamp:   time.Now().UnixNano(),\n\t}\n\ts.persistence.On(\"RecordWorkflowExecutionStarted\", mock.Anything, request).Return(nil).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionStarted(ctx, request))\n\n\t// no remaining tokens\n\ts.metricClient.On(\"IncCounter\", metrics.PersistenceRecordWorkflowExecutionStartedScope, metrics.PersistenceSampledCounter).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionStarted(ctx, request))\n}\n\nfunc (s *VisibilitySamplingSuite) TestRecordWorkflowExecutionClosed() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\trequest := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tDomain:           testDomain,\n\t\tExecution:        testWorkflowExecution,\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\tStatus:           types.WorkflowExecutionCloseStatusCompleted,\n\t}\n\trequest2 := &p.RecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:       testDomainUUID,\n\t\tDomain:           testDomain,\n\t\tExecution:        testWorkflowExecution,\n\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\tStatus:           types.WorkflowExecutionCloseStatusFailed,\n\t}\n\n\ts.persistence.On(\"RecordWorkflowExecutionClosed\", mock.Anything, request).Return(nil).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionClosed(ctx, request))\n\ts.persistence.On(\"RecordWorkflowExecutionClosed\", mock.Anything, request2).Return(nil).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionClosed(ctx, request2))\n\n\t// no remaining tokens\n\ts.metricClient.On(\"IncCounter\", metrics.PersistenceRecordWorkflowExecutionClosedScope, metrics.PersistenceSampledCounter).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionClosed(ctx, request))\n\ts.metricClient.On(\"IncCounter\", metrics.PersistenceRecordWorkflowExecutionClosedScope, metrics.PersistenceSampledCounter).Once()\n\ts.NoError(s.client.RecordWorkflowExecutionClosed(ctx, request2))\n}\n\nfunc (s *VisibilitySamplingSuite) TestListOpenWorkflowExecutions() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\trequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\ts.persistence.On(\"ListOpenWorkflowExecutions\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListOpenWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListOpenWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListClosedWorkflowExecutions() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\trequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\ts.persistence.On(\"ListClosedWorkflowExecutions\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListClosedWorkflowExecutions(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListClosedWorkflowExecutions(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListOpenWorkflowExecutionsByType() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\treq := p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\trequest := &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: req,\n\t\tWorkflowTypeName:              testWorkflowTypeName,\n\t}\n\ts.persistence.On(\"ListOpenWorkflowExecutionsByType\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListOpenWorkflowExecutionsByType(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListOpenWorkflowExecutionsByType(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListClosedWorkflowExecutionsByType() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\treq := p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\trequest := &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: req,\n\t\tWorkflowTypeName:              testWorkflowTypeName,\n\t}\n\ts.persistence.On(\"ListClosedWorkflowExecutionsByType\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListClosedWorkflowExecutionsByType(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListClosedWorkflowExecutionsByType(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListOpenWorkflowExecutionsByWorkflowID() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\treq := p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\trequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: req,\n\t\tWorkflowID:                    testWorkflowExecution.GetWorkflowID(),\n\t}\n\ts.persistence.On(\"ListOpenWorkflowExecutionsByWorkflowID\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListClosedWorkflowExecutionsByWorkflowID() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\treq := p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\trequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: req,\n\t\tWorkflowID:                    testWorkflowExecution.GetWorkflowID(),\n\t}\n\ts.persistence.On(\"ListClosedWorkflowExecutionsByWorkflowID\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n\nfunc (s *VisibilitySamplingSuite) TestListClosedWorkflowExecutionsByStatus() {\n\tctx, cancel := context.WithTimeout(context.Background(), testContextTimeout)\n\tdefer cancel()\n\n\treq := p.ListWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainUUID,\n\t\tDomain:     testDomain,\n\t}\n\trequest := &p.ListClosedWorkflowExecutionsByStatusRequest{\n\t\tListWorkflowExecutionsRequest: req,\n\t\tStatus:                        types.WorkflowExecutionCloseStatusFailed,\n\t}\n\ts.persistence.On(\"ListClosedWorkflowExecutionsByStatus\", mock.Anything, request).Return(nil, nil).Once()\n\t_, err := s.client.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\ts.NoError(err)\n\n\t// no remaining tokens\n\t_, err = s.client.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\ts.Error(err)\n\terrDetail, ok := err.(*types.ServiceBusyError)\n\ts.True(ok)\n\ts.Equal(listErrMsg, errDetail.Message)\n}\n"
  },
  {
    "path": "common/persistence/persistence-utils/history_manager_util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistenceutils\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ReadFullPageV2Events reads a full page of history events from HistoryManager. Due to storage format of V2 History\n// it is not guaranteed that pageSize amount of data is returned. Function returns the list of history events, the size\n// of data read, the next page token, and an error if present.\nfunc ReadFullPageV2Events(\n\tctx context.Context,\n\thistoryV2Mgr persistence.HistoryManager,\n\treq *persistence.ReadHistoryBranchRequest,\n) ([]*types.HistoryEvent, int, []byte, error) {\n\thistoryEvents := []*types.HistoryEvent{}\n\tsize := int(0)\n\tfor {\n\t\tresponse, err := historyV2Mgr.ReadHistoryBranch(ctx, req)\n\t\tif err != nil {\n\t\t\treturn nil, 0, nil, err\n\t\t}\n\t\thistoryEvents = append(historyEvents, response.HistoryEvents...)\n\t\tsize += response.Size\n\t\tif len(historyEvents) >= req.PageSize || len(response.NextPageToken) == 0 {\n\t\t\treturn historyEvents, size, response.NextPageToken, nil\n\t\t}\n\t\treq.NextPageToken = response.NextPageToken\n\t}\n}\n\n// ReadFullPageV2EventsByBatch reads a full page of history events by batch from HistoryManager. Due to storage format of V2 History\n// it is not guaranteed that pageSize amount of data is returned. Function returns the list of history batches, the size\n// of data read, the next page token, and an error if present.\nfunc ReadFullPageV2EventsByBatch(\n\tctx context.Context,\n\thistoryV2Mgr persistence.HistoryManager,\n\treq *persistence.ReadHistoryBranchRequest,\n) ([]*types.History, int, []byte, error) {\n\thistoryBatches := []*types.History{}\n\teventsRead := 0\n\tsize := 0\n\tfor {\n\t\tresponse, err := historyV2Mgr.ReadHistoryBranchByBatch(ctx, req)\n\t\tif err != nil {\n\t\t\treturn nil, 0, nil, err\n\t\t}\n\t\thistoryBatches = append(historyBatches, response.History...)\n\t\tfor _, batch := range response.History {\n\t\t\teventsRead += len(batch.Events)\n\t\t}\n\t\tsize += response.Size\n\t\tif eventsRead >= req.PageSize || len(response.NextPageToken) == 0 {\n\t\t\treturn historyBatches, size, response.NextPageToken, nil\n\t\t}\n\t\treq.NextPageToken = response.NextPageToken\n\t}\n}\n\n// GetBeginNodeID gets node id from last ancestor\nfunc GetBeginNodeID(bi types.HistoryBranch) int64 {\n\tif len(bi.Ancestors) == 0 {\n\t\t// root branch\n\t\treturn 1\n\t}\n\tidx := len(bi.Ancestors) - 1\n\treturn bi.Ancestors[idx].EndNodeID\n}\n\n// PaginateHistory return paged history\nfunc PaginateHistory(\n\tctx context.Context,\n\thistoryV2Mgr persistence.HistoryManager,\n\tbyBatch bool,\n\tbranchToken []byte,\n\tfirstEventID int64,\n\tnextEventID int64,\n\ttokenIn []byte,\n\tpageSize int,\n\tshardID *int,\n\tdomainID string,\n\tdomainCache cache.DomainCache,\n) ([]*types.HistoryEvent, []*types.History, []byte, int, error) {\n\n\thistoryEvents := []*types.HistoryEvent{}\n\thistoryBatches := []*types.History{}\n\tvar tokenOut []byte\n\tvar historySize int\n\tdomainName, err := domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, nil, nil, 0, err\n\t}\n\treq := &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: tokenIn,\n\t\tShardID:       shardID,\n\t\tDomainName:    domainName,\n\t}\n\tif byBatch {\n\t\tresponse, err := historyV2Mgr.ReadHistoryBranchByBatch(ctx, req)\n\t\tif err != nil {\n\t\t\tvar e *types.EntityNotExistsError\n\t\t\tif errors.As(err, &e) {\n\t\t\t\treturn nil, nil, nil, 0, nil\n\t\t\t}\n\t\t\treturn nil, nil, nil, 0, err\n\t\t}\n\n\t\t// Keep track of total history size\n\t\thistorySize += response.Size\n\t\thistoryBatches = append(historyBatches, response.History...)\n\t\ttokenOut = response.NextPageToken\n\n\t} else {\n\t\tresponse, err := historyV2Mgr.ReadHistoryBranch(ctx, req)\n\t\tif err != nil {\n\t\t\tvar e *types.EntityNotExistsError\n\t\t\tif errors.As(err, &e) {\n\t\t\t\treturn nil, nil, nil, 0, nil\n\t\t\t}\n\t\t\treturn nil, nil, nil, 0, err\n\t\t}\n\n\t\t// Keep track of total history size\n\t\thistorySize += response.Size\n\t\thistoryEvents = append(historyEvents, response.HistoryEvents...)\n\t\ttokenOut = response.NextPageToken\n\t}\n\n\treturn historyEvents, historyBatches, tokenOut, historySize, nil\n}\n\n// Get the maximum referenced node id of each branch\nfunc GetBranchesMaxReferredNodeIDs(branches []*types.HistoryBranch) map[string]int64 {\n\tvalidBRsMaxEndNode := map[string]int64{}\n\tfor _, b := range branches {\n\t\tfor _, br := range b.Ancestors {\n\t\t\tcurr, ok := validBRsMaxEndNode[br.BranchID]\n\t\t\tif !ok || curr < br.EndNodeID {\n\t\t\t\tvalidBRsMaxEndNode[br.BranchID] = br.EndNodeID\n\t\t\t}\n\t\t}\n\t}\n\treturn validBRsMaxEndNode\n}\n"
  },
  {
    "path": "common/persistence/pinot/pinot_visibility_metric_clients.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage pinotvisibility\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype pinotVisibilityMetricsClient struct {\n\tmetricClient metrics.Client\n\tpersistence  p.VisibilityManager\n\tlogger       log.Logger\n}\n\nvar _ p.VisibilityManager = (*pinotVisibilityMetricsClient)(nil)\n\n// NewPinotVisibilityMetricsClient wrap visibility client with metrics client\nfunc NewPinotVisibilityMetricsClient(persistence p.VisibilityManager, metricClient metrics.Client, logger log.Logger) p.VisibilityManager {\n\treturn &pinotVisibilityMetricsClient{\n\t\tpersistence:  persistence,\n\t\tmetricClient: metricClient,\n\t\tlogger:       logger,\n\t}\n}\n\nfunc (p *pinotVisibilityMetricsClient) GetName() string {\n\treturn p.persistence.GetName()\n}\n\nfunc (p *pinotVisibilityMetricsClient) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionStartedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotRecordWorkflowExecutionStartedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.RecordWorkflowExecutionStarted(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotRecordWorkflowExecutionStartedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionClosedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotRecordWorkflowExecutionClosedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.RecordWorkflowExecutionClosed(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotRecordWorkflowExecutionClosedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *p.RecordWorkflowExecutionUninitializedRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotRecordWorkflowExecutionUninitializedScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.RecordWorkflowExecutionUninitialized(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotRecordWorkflowExecutionUninitializedScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *p.UpsertWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotUpsertWorkflowExecutionScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.UpsertWorkflowExecution(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotUpsertWorkflowExecutionScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListOpenWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListOpenWorkflowExecutions(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListOpenWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListClosedWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListClosedWorkflowExecutions(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListClosedWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByTypeRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListOpenWorkflowExecutionsByTypeScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListOpenWorkflowExecutionsByType(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListOpenWorkflowExecutionsByTypeScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByTypeRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListClosedWorkflowExecutionsByTypeScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByType(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListClosedWorkflowExecutionsByTypeScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListOpenWorkflowExecutionsByWorkflowIDScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListOpenWorkflowExecutionsByWorkflowIDScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListClosedWorkflowExecutionsByWorkflowIDScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListClosedWorkflowExecutionsByWorkflowIDScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *p.ListClosedWorkflowExecutionsByStatusRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListClosedWorkflowExecutionsByStatusScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListClosedWorkflowExecutionsByStatusScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.GetClosedWorkflowExecutionRequest,\n) (*p.GetClosedWorkflowExecutionResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotGetClosedWorkflowExecutionScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.GetClosedWorkflowExecution(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotGetClosedWorkflowExecutionScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotListWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ListWorkflowExecutions(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotListWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.ListWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotScanWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.ScanWorkflowExecutions(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotScanWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.CountWorkflowExecutionsRequest,\n) (*p.CountWorkflowExecutionsResponse, error) {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotCountWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\tresponse, err := p.persistence.CountWorkflowExecutions(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotCountWorkflowExecutionsScope, err)\n\t}\n\n\treturn response, err\n}\n\nfunc (p *pinotVisibilityMetricsClient) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotDeleteWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.DeleteWorkflowExecution(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotDeleteWorkflowExecutionsScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\n\tscopeWithDomainTag := p.metricClient.Scope(metrics.PinotDeleteWorkflowExecutionsScope, metrics.DomainTag(request.Domain))\n\tscopeWithDomainTag.IncCounter(metrics.PinotRequestsPerDomain)\n\tsw := scopeWithDomainTag.StartTimer(metrics.PinotLatencyPerDomain)\n\tdefer sw.Stop()\n\terr := p.persistence.DeleteWorkflowExecution(ctx, request)\n\n\tif err != nil {\n\t\tp.updateErrorMetric(scopeWithDomainTag, metrics.PinotDeleteWorkflowExecutionsScope, err)\n\t}\n\n\treturn err\n}\n\nfunc (p *pinotVisibilityMetricsClient) updateErrorMetric(scopeWithDomainTag metrics.Scope, scope metrics.ScopeIdx, err error) {\n\n\tswitch err.(type) {\n\tcase *types.BadRequestError:\n\t\tscopeWithDomainTag.IncCounter(metrics.PinotErrBadRequestCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.PinotFailuresPerDomain)\n\n\tcase *types.ServiceBusyError:\n\t\tscopeWithDomainTag.IncCounter(metrics.PinotErrBusyCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.PinotFailuresPerDomain)\n\tdefault:\n\t\tp.logger.Error(\"Operation failed with internal error.\", tag.MetricScope(int(scope)), tag.Error(err))\n\t\tscopeWithDomainTag.IncCounter(metrics.PinotFailuresPerDomain)\n\t}\n}\n\nfunc (p *pinotVisibilityMetricsClient) Close() {\n\tp.persistence.Close()\n}\n"
  },
  {
    "path": "common/persistence/pinot/pinot_visibility_metric_clients_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinotvisibility\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmetricsClientMocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\ttestStopwatch = metrics.NoopScope.StartTimer(metrics.PinotLatency)\n)\n\nfunc TestMetricClientRecordWorkflowExecutionStarted(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.RecordWorkflowExecutionStartedRequest{\n\t\tWorkflowTypeName: \"errorWorkflowTypeName\",\n\t\tMemo: &types.Memo{\n\t\t\tmap[string][]byte{\n\t\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t},\n\t\t},\n\t}\n\n\trequest := &p.RecordWorkflowExecutionStartedRequest{\n\t\tWorkflowTypeName: \"wtn\",\n\t\tMemo: &types.Memo{\n\t\t\tmap[string][]byte{\n\t\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.RecordWorkflowExecutionStartedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(&types.BadRequestError{}).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.RecordWorkflowExecutionStarted(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientRecordWorkflowExecutionClosed(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.RecordWorkflowExecutionClosedRequest{\n\t\tWorkflowTypeName: \"errorWorkflowTypeName\",\n\t\tMemo: &types.Memo{\n\t\t\tmap[string][]byte{\n\t\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t},\n\t\t},\n\t}\n\n\trequest := &p.RecordWorkflowExecutionClosedRequest{\n\t\tWorkflowTypeName: \"wtn\",\n\t\tMemo: &types.Memo{\n\t\t\tmap[string][]byte{\n\t\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.RecordWorkflowExecutionClosedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(&types.ServiceBusyError{}).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: &types.ServiceBusyError{},\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.RecordWorkflowExecutionClosed(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientRecordWorkflowExecutionUninitialized(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.RecordWorkflowExecutionUninitializedRequest{\n\t\tWorkflowTypeName: \"errorWorkflowTypeName\",\n\t}\n\n\trequest := &p.RecordWorkflowExecutionUninitializedRequest{\n\t\tWorkflowTypeName: \"wtn\",\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.RecordWorkflowExecutionUninitializedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(fmt.Errorf(\"error\")).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.RecordWorkflowExecutionUninitialized(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientUpsertWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.UpsertWorkflowExecutionRequest{\n\t\tWorkflowTypeName: \"errorWorkflowTypeName\",\n\t}\n\n\trequest := &p.UpsertWorkflowExecutionRequest{\n\t\tWorkflowTypeName: \"wtn\",\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.UpsertWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(fmt.Errorf(\"error\")).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.UpsertWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListOpenWorkflowExecutions(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomain: \"badDomainID\",\n\t}\n\n\trequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomain: DomainID,\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListOpenWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListClosedWorkflowExecutions(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomain: \"badDomainId\",\n\t}\n\n\trequest := &p.ListWorkflowExecutionsRequest{\n\t\tDomain: DomainID,\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListClosedWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListOpenWorkflowExecutionsByType(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"badDomainID\",\n\t\t},\n\t\tWorkflowTypeName: \"\",\n\t}\n\n\trequest := &p.ListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByTypeRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListOpenWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListClosedWorkflowExecutionsByType(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"badDomainID\",\n\t\t},\n\t\tWorkflowTypeName: \"\",\n\t}\n\n\trequest := &p.ListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByTypeRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListClosedWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListOpenWorkflowExecutionsByWorkflowID(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"badDomainID\",\n\t\t},\n\t}\n\n\trequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByWorkflowIDRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListClosedWorkflowExecutionsByWorkflowID(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"badDomainID\",\n\t\t},\n\t}\n\n\trequest := &p.ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByWorkflowIDRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListClosedWorkflowExecutionsByStatus(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.ListClosedWorkflowExecutionsByStatusRequest{\n\t\tListWorkflowExecutionsRequest: p.ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"badDomainID\",\n\t\t},\n\t}\n\n\trequest := &p.ListClosedWorkflowExecutionsByStatusRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListClosedWorkflowExecutionsByStatusRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListClosedWorkflowExecutionsByStatus(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientGetClosedWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.GetClosedWorkflowExecutionRequest{}\n\n\trequest := &p.GetClosedWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.GetClosedWorkflowExecutionRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"Pinot GetClosedWorkflowExecution failed, error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(&pnt.SearchResponse{\n\t\t\t\t\tExecutions: []*p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tDomainID: DomainID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.GetClosedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMetricClientListWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByQueryRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ListWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientScanWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByQueryRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.ScanWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientCountWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.CountWorkflowExecutionsRequest{}\n\trequest := &p.CountWorkflowExecutionsRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.CountWorkflowExecutionsRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\tscopeMockAffordance       func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().CountByQuery(gomock.Any()).Return(int64(0), fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"CountClosedWorkflowExecutions failed, error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().CountByQuery(gomock.Any()).Return(int64(1), nil).Times(1)\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\t_, err := metricsClient.CountWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMetricClientDeleteWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.VisibilityDeleteWorkflowExecutionRequest{}\n\n\trequest := &p.VisibilityDeleteWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                *p.VisibilityDeleteWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(fmt.Errorf(\"error\")).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.DeleteWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientDeleteUninitializedWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.VisibilityDeleteWorkflowExecutionRequest{}\n\n\trequest := &p.VisibilityDeleteWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                *p.VisibilityDeleteWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\tscopeMockAffordance    func(mockScope *metricsClientMocks.Scope)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(fmt.Errorf(\"error\")).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Times(3)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\tscopeMockAffordance: func(mockScope *metricsClientMocks.Scope) {\n\t\t\t\tmockScope.On(\"IncCounter\", mock.Anything, mock.Anything, mock.Anything).Return().Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// create mock clients\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmockMetricClient := &metricsClientMocks.Client{}\n\t\t\tmockScope := &metricsClientMocks.Scope{}\n\n\t\t\t// create metricClient\n\t\t\tlogger := log.NewNoop()\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}, mockProducer, testlogger.New(t))\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\t\t\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\t\t\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\t\t\t// mock behaviors\n\t\t\tmockMetricClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope).Once()\n\t\t\tmockScope.On(\"StartTimer\", mock.Anything, mock.Anything).Return(testStopwatch).Once()\n\t\t\ttest.producerMockAffordance(mockProducer)\n\t\t\ttest.scopeMockAffordance(mockScope)\n\n\t\t\terr := metricsClient.DeleteUninitializedWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, err, test.expectedError)\n\t\t})\n\t}\n}\n\nfunc TestMetricClientClose(t *testing.T) {\n\t// create mock clients\n\tctrl := gomock.NewController(t)\n\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\tmockProducer := &mocks.KafkaProducer{}\n\tmockMetricClient := &metricsClientMocks.Client{}\n\n\t// create metricClient\n\tlogger := log.NewNoop()\n\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t}, mockProducer, testlogger.New(t))\n\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\tassert.NotPanics(t, func() {\n\t\tmetricsClient.Close()\n\n\t})\n}\n\nfunc TestMetricClientGetName(t *testing.T) {\n\t// create mock clients\n\tctrl := gomock.NewController(t)\n\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\tmockProducer := &mocks.KafkaProducer{}\n\tmockMetricClient := &metricsClientMocks.Client{}\n\n\t// create metricClient\n\tlogger := log.NewNoop()\n\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\tValidSearchAttributes:  dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tESIndexMaxResultWindow: dynamicproperties.GetIntPropertyFn(3),\n\t}, mockProducer, testlogger.New(t))\n\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\tpinotVisibilityManager := p.NewVisibilityManagerImpl(visibilityStore, logger, &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\tvisibilityMgr := NewPinotVisibilityMetricsClient(pinotVisibilityManager, mockMetricClient, logger)\n\tmetricsClient := visibilityMgr.(*pinotVisibilityMetricsClient)\n\n\tassert.NotPanics(t, func() {\n\t\tmetricsClient.GetName()\n\n\t})\n}\n"
  },
  {
    "path": "common/persistence/pinot/pinot_visibility_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage pinotvisibility\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nconst (\n\tDescendingOrder = \"DESC\"\n\tAscendingOrder  = \"ASC\"\n\tDomainID        = \"DomainID\"\n\tWorkflowID      = \"WorkflowID\"\n\tRunID           = \"RunID\"\n\tWorkflowType    = \"WorkflowType\"\n\tCloseStatus     = \"CloseStatus\"\n\tHistoryLength   = \"HistoryLength\"\n\tTaskList        = \"TaskList\"\n\tIsCron          = \"IsCron\"\n\tNumClusters     = \"NumClusters\"\n\tShardID         = \"ShardID\"\n\tAttr            = \"Attr\"\n\tStartTime       = \"StartTime\"\n\tCloseTime       = \"CloseTime\"\n\tUpdateTime      = \"UpdateTime\"\n\tExecutionTime   = \"ExecutionTime\"\n\tIsDeleted       = \"IsDeleted\"   // used for Pinot deletion/rolling upsert only, not visible to user\n\tEventTimeMs     = \"EventTimeMs\" // used for Pinot deletion/rolling upsert only, not visible to user\n\tMemo            = \"Memo\"\n\n\t// used to be micro second\n\toneMicroSecondInNano = int64(time.Microsecond / time.Nanosecond)\n)\n\ntype (\n\tpinotVisibilityStore struct {\n\t\tpinotClient         pnt.GenericClient\n\t\tproducer            messaging.Producer\n\t\tlogger              log.Logger\n\t\tconfig              *service.Config\n\t\tpinotQueryValidator *pnt.VisibilityQueryValidator\n\t}\n)\n\nvar _ p.VisibilityStore = (*pinotVisibilityStore)(nil)\n\nfunc NewPinotVisibilityStore(\n\tpinotClient pnt.GenericClient,\n\tconfig *service.Config,\n\tproducer messaging.Producer,\n\tlogger log.Logger,\n) p.VisibilityStore {\n\tif producer == nil {\n\t\t// must be bug, check history setup\n\t\tlogger.Fatal(\"message producer is nil\")\n\t}\n\treturn &pinotVisibilityStore{\n\t\tpinotClient:         pinotClient,\n\t\tproducer:            producer,\n\t\tlogger:              logger.WithTags(tag.ComponentPinotVisibilityManager),\n\t\tconfig:              config,\n\t\tpinotQueryValidator: pnt.NewPinotQueryValidator(config.ValidSearchAttributes, config.PinotOptimizedQueryColumns),\n\t}\n}\n\nfunc (v *pinotVisibilityStore) Close() {\n\t// Not needed for pinot, just keep for visibility store interface\n}\n\nfunc (v *pinotVisibilityStore) GetName() string {\n\treturn constants.PinotPersistenceName\n}\n\nfunc (v *pinotVisibilityStore) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionStartedRequest,\n) error {\n\n\tmsg, err := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixMilli(),\n\t\trequest.ExecutionTimestamp.UnixMilli(),\n\t\trequest.TaskID,\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\t-1, // represent invalid close time, means open workflow execution\n\t\t-1, // represent invalid close status, means open workflow execution\n\t\t0,  // will be updated when workflow execution updates\n\t\trequest.UpdateTimestamp.UnixMilli(),\n\t\tint64(request.ShardID),\n\t\trequest.SearchAttributes,\n\t\tfalse,\n\t\trequest.Memo,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *pinotVisibilityStore) RecordWorkflowExecutionClosed(ctx context.Context, request *p.InternalRecordWorkflowExecutionClosedRequest) error {\n\n\tmsg, err := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixMilli(),\n\t\trequest.ExecutionTimestamp.UnixMilli(),\n\t\trequest.TaskID,\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\trequest.CloseTimestamp.UnixMilli(),\n\t\t*thrift.FromWorkflowExecutionCloseStatus(&request.Status),\n\t\trequest.HistoryLength,\n\t\trequest.UpdateTimestamp.UnixMilli(),\n\t\tint64(request.ShardID),\n\t\trequest.SearchAttributes,\n\t\tfalse,\n\t\trequest.Memo,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *pinotVisibilityStore) RecordWorkflowExecutionUninitialized(ctx context.Context, request *p.InternalRecordWorkflowExecutionUninitializedRequest) error {\n\n\tmsg, err := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\t\"\",\n\t\t-1,\n\t\t-1,\n\t\t0,\n\t\tfalse,\n\t\t0,\n\t\t-1, // represent invalid close time, means open workflow execution\n\t\t-1, // represent invalid close status, means open workflow execution\n\t\t0,  // will be updated when workflow execution updates\n\t\trequest.UpdateTimestamp.UnixMilli(),\n\t\trequest.ShardID,\n\t\tnil,\n\t\tfalse,\n\t\tnil,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *pinotVisibilityStore) UpsertWorkflowExecution(ctx context.Context, request *p.InternalUpsertWorkflowExecutionRequest) error {\n\tmsg, err := createVisibilityMessage(\n\t\trequest.DomainUUID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\trequest.WorkflowTypeName,\n\t\trequest.TaskList,\n\t\trequest.StartTimestamp.UnixMilli(),\n\t\trequest.ExecutionTimestamp.UnixMilli(),\n\t\trequest.TaskID,\n\t\trequest.IsCron,\n\t\trequest.NumClusters,\n\t\t-1, // represent invalid close time, means open workflow execution\n\t\t-1, // represent invalid close status, means open workflow execution\n\t\t0,  // will not be used\n\t\trequest.UpdateTimestamp.UnixMilli(),\n\t\trequest.ShardID,\n\t\trequest.SearchAttributes,\n\t\tfalse,\n\t\trequest.Memo,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *pinotVisibilityStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\n\tmsg, err := createDeleteVisibilityMessage(\n\t\trequest.DomainID,\n\t\trequest.WorkflowID,\n\t\trequest.RunID,\n\t\ttrue,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn v.producer.Publish(ctx, msg)\n}\n\nfunc (v *pinotVisibilityStore) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\t// verify if it is uninitialized workflow execution record\n\t// if it is, then call the existing delete method to delete\n\tquery := fmt.Sprintf(\"StartTime = missing and DomainID = '%s' and RunID = '%s'\", request.DomainID, request.RunID)\n\n\tqueryRequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomain: request.Domain,\n\t\tQuery:  query,\n\t}\n\tresp, err := v.CountWorkflowExecutions(ctx, queryRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resp.Count > 0 {\n\t\tif err = v.DeleteWorkflowExecution(ctx, request); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v *pinotVisibilityStore) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\tquery, err := getListWorkflowExecutionsQuery(v.pinotClient.GetTableName(), request, false)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     request,\n\t}\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsQuery(v.pinotClient.GetTableName(), request, true)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     request,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListOpenWorkflowExecutionsByType(ctx context.Context, request *p.InternalListWorkflowExecutionsByTypeRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsByTypeQuery(v.pinotClient.GetTableName(), request, false)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions by type query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListClosedWorkflowExecutionsByType(ctx context.Context, request *p.InternalListWorkflowExecutionsByTypeRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsByTypeQuery(v.pinotClient.GetTableName(), request, true)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions by type query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *p.InternalListWorkflowExecutionsByWorkflowIDRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.StartTime) && !rec.StartTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsByWorkflowIDQuery(v.pinotClient.GetTableName(), request, false)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions by workflowID query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *p.InternalListWorkflowExecutionsByWorkflowIDRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsByWorkflowIDQuery(v.pinotClient.GetTableName(), request, true)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions by workflowID query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *p.InternalListClosedWorkflowExecutionsByStatusRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn !request.EarliestTime.After(rec.CloseTime) && !rec.CloseTime.After(request.LatestTime)\n\t}\n\n\tquery, err := getListWorkflowExecutionsByStatusQuery(v.pinotClient.GetTableName(), request)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions by status query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          isRecordValid,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest:     &request.InternalListWorkflowExecutionsRequest,\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) GetClosedWorkflowExecution(ctx context.Context, request *p.InternalGetClosedWorkflowExecutionRequest) (*p.InternalGetClosedWorkflowExecutionResponse, error) {\n\tquery := getGetClosedWorkflowExecutionQuery(v.pinotClient.GetTableName(), request)\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{ // create a new request to avoid nil pointer exceptions\n\t\t\tDomainUUID:    request.DomainUUID,\n\t\t\tDomain:        request.Domain,\n\t\t\tEarliestTime:  time.Time{},\n\t\t\tLatestTime:    time.Time{},\n\t\t\tPageSize:      1,\n\t\t\tNextPageToken: nil,\n\t\t},\n\t}\n\n\tresp, err := v.pinotClient.Search(req)\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Pinot GetClosedWorkflowExecution failed, %v\", err),\n\t\t}\n\t}\n\n\treturn &p.InternalGetClosedWorkflowExecutionResponse{\n\t\tExecution: resp.Executions[0],\n\t}, nil\n}\n\nfunc (v *pinotVisibilityStore) ListWorkflowExecutions(ctx context.Context, request *p.ListWorkflowExecutionsByQueryRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tcheckPageSize(request)\n\n\tquery, err := v.getListWorkflowExecutionsByQueryQuery(v.pinotClient.GetTableName(), request)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build list workflow executions query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:    request.DomainUUID,\n\t\t\tDomain:        request.Domain,\n\t\t\tEarliestTime:  time.Time{},\n\t\t\tLatestTime:    time.Time{},\n\t\t\tNextPageToken: request.NextPageToken,\n\t\t\tPageSize:      request.PageSize,\n\t\t},\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) ScanWorkflowExecutions(ctx context.Context, request *p.ListWorkflowExecutionsByQueryRequest) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tcheckPageSize(request)\n\n\tquery, err := v.getListWorkflowExecutionsByQueryQuery(v.pinotClient.GetTableName(), request)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build scan workflow executions query %v\", err))\n\t\treturn nil, err\n\t}\n\n\treq := &pnt.SearchRequest{\n\t\tQuery:           query,\n\t\tIsOpen:          true,\n\t\tFilter:          nil,\n\t\tMaxResultWindow: v.config.ESIndexMaxResultWindow(),\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:    request.DomainUUID,\n\t\t\tDomain:        request.Domain,\n\t\t\tEarliestTime:  time.Time{},\n\t\t\tLatestTime:    time.Time{},\n\t\t\tNextPageToken: request.NextPageToken,\n\t\t\tPageSize:      request.PageSize,\n\t\t},\n\t}\n\n\treturn v.pinotClient.Search(req)\n}\n\nfunc (v *pinotVisibilityStore) CountWorkflowExecutions(ctx context.Context, request *p.CountWorkflowExecutionsRequest) (*p.CountWorkflowExecutionsResponse, error) {\n\tquery, err := v.getCountWorkflowExecutionsQuery(v.pinotClient.GetTableName(), request)\n\tif err != nil {\n\t\tv.logger.Error(fmt.Sprintf(\"failed to build count workflow executions query %v\", err))\n\t\treturn nil, err\n\t}\n\n\tresp, err := v.pinotClient.CountByQuery(query)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"CountClosedWorkflowExecutions failed, %v\", err),\n\t\t}\n\t}\n\n\treturn &p.CountWorkflowExecutionsResponse{\n\t\tCount: resp,\n\t}, nil\n}\n\n// a new function to create visibility message for deletion\n// don't use the other function and provide some nil values because it may cause nil pointer exceptions\nfunc createDeleteVisibilityMessage(domainID string,\n\twid,\n\trid string,\n\tisDeleted bool,\n) (*indexer.PinotMessage, error) {\n\tm := make(map[string]interface{})\n\tm[DomainID] = domainID\n\tm[WorkflowID] = wid\n\tm[RunID] = rid\n\tm[IsDeleted] = isDeleted\n\tm[EventTimeMs] = time.Now().UnixMilli()\n\tserializedMsg, err := json.Marshal(m)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmsg := &indexer.PinotMessage{\n\t\tWorkflowID: common.StringPtr(wid),\n\t\tPayload:    serializedMsg,\n\t}\n\treturn msg, nil\n}\n\nfunc createVisibilityMessage(\n\t// common parameters\n\tdomainID string,\n\twid,\n\trid string,\n\tworkflowTypeName string,\n\ttaskList string,\n\tstartTimeUnixMilli int64,\n\texecutionTimeUnixMilli int64,\n\ttaskID int64,\n\tisCron bool,\n\tnumClusters int16,\n\t// specific to certain status\n\tcloseTimeUnixMilli int64, // close execution\n\tcloseStatus workflow.WorkflowExecutionCloseStatus, // close execution\n\thistoryLength int64, // close execution\n\tupdateTimeUnixMilli int64, // update execution,\n\tshardID int64,\n\trawSearchAttributes map[string][]byte,\n\tisDeleted bool,\n\tmemo *p.DataBlob,\n) (*indexer.PinotMessage, error) {\n\tm := make(map[string]interface{})\n\t// loop through all input parameters\n\tm[DomainID] = domainID\n\tm[WorkflowID] = wid\n\tm[RunID] = rid\n\tm[WorkflowType] = workflowTypeName\n\tm[TaskList] = taskList\n\tm[StartTime] = startTimeUnixMilli\n\tm[ExecutionTime] = executionTimeUnixMilli\n\tm[IsCron] = isCron\n\tm[NumClusters] = numClusters\n\tm[CloseTime] = closeTimeUnixMilli\n\tm[CloseStatus] = int(closeStatus)\n\tm[HistoryLength] = historyLength\n\tm[UpdateTime] = updateTimeUnixMilli\n\tm[ShardID] = shardID\n\tm[IsDeleted] = isDeleted\n\tm[EventTimeMs] = updateTimeUnixMilli // same as update time when record is upserted, could not use updateTime directly since this will be modified by Pinot\n\n\tSearchAttributes := make(map[string]interface{})\n\tvar err error\n\tfor key, value := range rawSearchAttributes {\n\t\tvalue, err = isTimeStruct(value)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tvar val interface{}\n\t\terr = json.Unmarshal(value, &val)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tSearchAttributes[key] = val\n\t}\n\n\tif memo != nil && len(memo.Data) > 0 {\n\t\t// add memo into search attr\n\t\tmarshalMemo, err := json.Marshal(memo)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\t// string(marshalMemo) is a must-do step,\n\t\t// to give it a type so that it can be converted to a string again\n\t\t// and unmarshaled correctly in the reading side\n\t\t// Also, it doesn't matter to store in Pinot as a pure string since it's not used for search\n\t\tSearchAttributes[Memo] = string(marshalMemo)\n\t}\n\n\tm[Attr] = SearchAttributes\n\tserializedMsg, err := json.Marshal(m)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmsg := &indexer.PinotMessage{\n\t\tWorkflowID: common.StringPtr(wid),\n\t\tPayload:    serializedMsg,\n\t}\n\treturn msg, nil\n\n}\n\n// check if value is time.Time type\n// if it is, convert it to unixMilli\n// if it isn't time, return the original value\nfunc isTimeStruct(value []byte) ([]byte, error) {\n\tvar time time.Time\n\terr := json.Unmarshal(value, &time)\n\tif err == nil {\n\t\tunixTime := time.UnixMilli()\n\t\tvalue, err = json.Marshal(unixTime)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn value, nil\n}\n\n/****************************** Request Translator ******************************/\n\ntype PinotQuery struct {\n\tquery   string\n\tfilters PinotQueryFilter\n\tsorters string\n\tlimits  string\n}\n\ntype PinotQueryFilter struct {\n\tstring\n}\n\nfunc NewPinotQuery(tableName string) PinotQuery {\n\treturn PinotQuery{\n\t\tquery:   fmt.Sprintf(\"SELECT *\\nFROM %s\\n\", tableName),\n\t\tfilters: PinotQueryFilter{},\n\t\tsorters: \"\",\n\t\tlimits:  \"\",\n\t}\n}\n\nfunc NewPinotCountQuery(tableName string) PinotQuery {\n\treturn PinotQuery{\n\t\tquery:   fmt.Sprintf(\"SELECT COUNT(*)\\nFROM %s\\n\", tableName),\n\t\tfilters: PinotQueryFilter{},\n\t\tsorters: \"\",\n\t\tlimits:  \"\",\n\t}\n}\n\nfunc (q *PinotQuery) String() string {\n\treturn fmt.Sprintf(\"%s%s%s%s\", q.query, q.filters.string, q.sorters, q.limits)\n}\n\nfunc (q *PinotQuery) concatSorter(sorter string) {\n\tq.sorters += sorter + \"\\n\"\n}\n\nfunc (q *PinotQuery) addPinotSorter(orderBy string, order string) {\n\tif q.sorters == \"\" {\n\t\tq.sorters = \"Order BY \"\n\t} else {\n\t\tq.sorters += \", \"\n\t}\n\tq.sorters += fmt.Sprintf(\"%s %s\\n\", orderBy, order)\n}\n\nfunc (q *PinotQuery) addOffsetAndLimits(offset int, limit int) {\n\tq.limits += fmt.Sprintf(\"LIMIT %d, %d\\n\", offset, limit)\n}\n\nfunc (f *PinotQueryFilter) checkFirstFilter() {\n\tif f.string == \"\" {\n\t\tf.string = \"WHERE \"\n\t} else {\n\t\tf.string += \"AND \"\n\t}\n}\n\nfunc (f *PinotQueryFilter) addEqual(obj string, val interface{}) {\n\tf.checkFirstFilter()\n\tif _, ok := val.(string); ok {\n\t\tval = fmt.Sprintf(\"'%s'\", val)\n\t} else {\n\t\tval = fmt.Sprintf(\"%v\", val)\n\t}\n\n\tquotedVal := fmt.Sprintf(\"%s\", val)\n\tf.string += fmt.Sprintf(\"%s = %s\\n\", obj, quotedVal)\n}\n\n// addQuery adds a complete query into the filter\nfunc (f *PinotQueryFilter) addQuery(query string) {\n\tf.checkFirstFilter()\n\tf.string += fmt.Sprintf(\"%s\\n\", query)\n}\n\n// addGte check object is greater than or equals to val\nfunc (f *PinotQueryFilter) addGte(obj string, val int) {\n\tf.checkFirstFilter()\n\tf.string += fmt.Sprintf(\"%s >= %s\\n\", obj, fmt.Sprintf(\"%v\", val))\n}\n\n// addLte check object is less than val\nfunc (f *PinotQueryFilter) addLt(obj string, val interface{}) {\n\tf.checkFirstFilter()\n\tf.string += fmt.Sprintf(\"%s < %s\\n\", obj, fmt.Sprintf(\"%v\", val))\n}\n\nfunc (f *PinotQueryFilter) addTimeRange(obj string, earliest interface{}, latest interface{}) {\n\tf.checkFirstFilter()\n\tf.string += fmt.Sprintf(\"%s BETWEEN %v AND %v\\n\", obj, earliest, latest)\n}\n\nfunc (v *pinotVisibilityStore) getCountWorkflowExecutionsQuery(tableName string, request *p.CountWorkflowExecutionsRequest) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\tquery := NewPinotCountQuery(tableName)\n\n\t// need to add Domain ID\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\n\trequestQuery := strings.TrimSpace(request.Query)\n\n\t// if customized query is empty, directly return\n\tif requestQuery == \"\" {\n\t\treturn query.String(), nil\n\t}\n\n\trequestQuery = filterPrefix(requestQuery)\n\n\tcomparExpr, _ := parseOrderBy(requestQuery)\n\tcomparExpr, err := v.pinotQueryValidator.ValidateQuery(comparExpr)\n\tif err != nil {\n\t\treturn \"\", &types.BadRequestError{Message: fmt.Sprintf(\"pinot query validator error: %s, query: %s\", err.Error(), request.Query)}\n\t}\n\n\tcomparExpr = filterPrefix(comparExpr)\n\tif comparExpr != \"\" {\n\t\tquery.filters.addQuery(comparExpr)\n\t}\n\n\treturn query.String(), nil\n}\n\nfunc (v *pinotVisibilityStore) getListWorkflowExecutionsByQueryQuery(tableName string, request *p.ListWorkflowExecutionsByQueryRequest) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\ttoken, err := pnt.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"next page token: %w\", err)\n\t}\n\n\tquery := NewPinotQuery(tableName)\n\n\t// need to add Domain ID\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\n\trequestQuery := strings.TrimSpace(request.Query)\n\n\t// if customized query is empty, directly return\n\tif requestQuery == \"\" {\n\t\tquery.addOffsetAndLimits(token.From, request.PageSize)\n\t\treturn query.String(), nil\n\t}\n\n\trequestQuery = filterPrefix(requestQuery)\n\tif common.IsJustOrderByClause(requestQuery) {\n\t\tquery.concatSorter(requestQuery)\n\t\tquery.addOffsetAndLimits(token.From, request.PageSize)\n\t\treturn query.String(), nil\n\t}\n\n\tcomparExpr, orderBy := parseOrderBy(requestQuery)\n\tcomparExpr, err = v.pinotQueryValidator.ValidateQuery(comparExpr)\n\tif err != nil {\n\t\treturn \"\", &types.BadRequestError{Message: fmt.Sprintf(\"pinot query validator error: %s, query: %s\", err.Error(), request.Query)}\n\t}\n\n\tcomparExpr = filterPrefix(comparExpr)\n\tif comparExpr != \"\" {\n\t\tquery.filters.addQuery(comparExpr)\n\t}\n\tif orderBy != \"\" {\n\t\tquery.concatSorter(orderBy)\n\t}\n\n\t// MUST HAVE! because pagination wouldn't work without order by clause!\n\tif query.sorters == \"\" {\n\t\tquery.addPinotSorter(StartTime, \"DESC\")\n\t}\n\n\tquery.addOffsetAndLimits(token.From, request.PageSize)\n\treturn query.String(), nil\n}\n\nfunc filterPrefix(query string) string {\n\tprefix := fmt.Sprintf(\"`%s.\", Attr)\n\tpostfix := \"`\"\n\n\tquery = strings.ReplaceAll(query, prefix, \"\")\n\treturn strings.ReplaceAll(query, postfix, \"\")\n}\n\n/*\nCan have cases:\n1. A = B\n2. A < B\n3. A > B\n4. A <= B\n5. A >= B\n*/\nfunc splitElement(element string) (string, string, string) {\n\tif element == \"\" {\n\t\treturn \"\", \"\", \"\"\n\t}\n\n\tlistLE := strings.Split(element, \"<=\")\n\tlistGE := strings.Split(element, \">=\")\n\tlistE := strings.Split(element, \"=\")\n\tlistL := strings.Split(element, \"<\")\n\tlistG := strings.Split(element, \">\")\n\n\tif len(listLE) > 1 {\n\t\treturn strings.TrimSpace(listLE[0]), strings.TrimSpace(listLE[1]), \"<=\"\n\t}\n\n\tif len(listGE) > 1 {\n\t\treturn strings.TrimSpace(listGE[0]), strings.TrimSpace(listGE[1]), \">=\"\n\t}\n\n\tif len(listE) > 1 {\n\t\treturn strings.TrimSpace(listE[0]), strings.TrimSpace(listE[1]), \"=\"\n\t}\n\n\tif len(listL) > 1 {\n\t\treturn strings.TrimSpace(listL[0]), strings.TrimSpace(listL[1]), \"<\"\n\t}\n\n\tif len(listG) > 1 {\n\t\treturn strings.TrimSpace(listG[0]), strings.TrimSpace(listG[1]), \">\"\n\t}\n\n\treturn \"\", \"\", \"\"\n}\n\n/*\nOrder by XXX DESC\n-> if startWith(\"Order by\") -> return \"\", element\n\nCustomizedString = 'cannot be used in order by'\n-> if last character is ‘ or \" -> return element, \"\"\n\nCustomizedInt = 1 (without order by clause)\n-> if !contains(\"Order by\") -> return element, \"\"\n\nCustomizedString = 'cannot be used in order by' Order by XXX DESC\n-> Find the index x of last appearance of \"order by\" -> return element[0, x], element[x, len]\n\nCustomizedInt = 1 Order by XXX DESC\n-> Find the index x of last appearance of \"order by\" -> return element[0, x], element[x, len]\n*/\nfunc parseOrderBy(element string) (string, string) {\n\t// case 1: when order by query also passed in\n\tif common.IsJustOrderByClause(element) {\n\t\treturn \"\", element\n\t}\n\n\t// case 2: when last element is a string\n\tif element[len(element)-1] == '\\'' || element[len(element)-1] == '\"' {\n\t\treturn element, \"\"\n\t}\n\n\t// case 3: when last element doesn't contain \"order by\"\n\tif !strings.Contains(strings.ToLower(element), \"order by\") {\n\t\treturn element, \"\"\n\t}\n\n\t// case 4: general case\n\telementArray := strings.Split(element, \" \")\n\torderByIndex := findLastOrderBy(elementArray) // find the last appearance of \"order by\" is the answer\n\tif orderByIndex == 0 {\n\t\treturn element, \"\"\n\t}\n\treturn strings.Join(elementArray[:orderByIndex], \" \"), strings.Join(elementArray[orderByIndex:], \" \")\n}\n\nfunc findLastOrderBy(list []string) int {\n\tfor i := len(list) - 2; i >= 0; i-- {\n\t\tif strings.Contains(list[i], \"\\\"\") || strings.Contains(list[i], \"'\") {\n\t\t\treturn 0 // means order by is inside a string\n\t\t}\n\n\t\tif strings.ToLower(list[i]) == \"order\" && strings.ToLower(list[i+1]) == \"by\" {\n\t\t\treturn i\n\t\t}\n\t}\n\treturn 0\n}\n\nfunc getListWorkflowExecutionsQuery(tableName string, request *p.InternalListWorkflowExecutionsRequest, isClosed bool) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\ttoken, err := pnt.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"next page token: %w\", err)\n\t}\n\n\tfrom := token.From\n\tpageSize := request.PageSize\n\n\tquery := NewPinotQuery(tableName)\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\n\tearliest := request.EarliestTime.UnixMilli() - oneMicroSecondInNano\n\tlatest := request.LatestTime.UnixMilli() + oneMicroSecondInNano\n\n\tif isClosed {\n\t\tquery.filters.addTimeRange(CloseTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addGte(CloseStatus, 0)\n\t} else {\n\t\tquery.filters.addTimeRange(StartTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addEqual(CloseStatus, -1)\n\t\tquery.filters.addEqual(CloseTime, -1)\n\t}\n\n\tquery.addPinotSorter(StartTime, DescendingOrder)\n\tquery.addOffsetAndLimits(from, pageSize)\n\n\treturn query.String(), nil\n}\n\nfunc getListWorkflowExecutionsByTypeQuery(tableName string, request *p.InternalListWorkflowExecutionsByTypeRequest, isClosed bool) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\tquery := NewPinotQuery(tableName)\n\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\tquery.filters.addEqual(WorkflowType, request.WorkflowTypeName)\n\tearliest := request.EarliestTime.UnixMilli() - oneMicroSecondInNano\n\tlatest := request.LatestTime.UnixMilli() + oneMicroSecondInNano\n\n\tif isClosed {\n\t\tquery.filters.addTimeRange(CloseTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addGte(CloseStatus, 0)\n\t} else {\n\t\tquery.filters.addTimeRange(StartTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addEqual(CloseStatus, -1)\n\t\tquery.filters.addEqual(CloseTime, -1)\n\t}\n\n\tquery.addPinotSorter(StartTime, DescendingOrder)\n\n\ttoken, err := pnt.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"next page token: %w\", err)\n\t}\n\n\tfrom := token.From\n\tpageSize := request.PageSize\n\tquery.addOffsetAndLimits(from, pageSize)\n\n\treturn query.String(), nil\n}\n\nfunc getListWorkflowExecutionsByWorkflowIDQuery(tableName string, request *p.InternalListWorkflowExecutionsByWorkflowIDRequest, isClosed bool) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\tquery := NewPinotQuery(tableName)\n\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\tquery.filters.addEqual(WorkflowID, request.WorkflowID)\n\tearliest := request.EarliestTime.UnixMilli() - oneMicroSecondInNano\n\tlatest := request.LatestTime.UnixMilli() + oneMicroSecondInNano\n\n\tif isClosed {\n\t\tquery.filters.addTimeRange(CloseTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addGte(CloseStatus, 0)\n\t} else {\n\t\tquery.filters.addTimeRange(StartTime, earliest, latest) // convert Unix Time to miliseconds\n\t\tquery.filters.addEqual(CloseStatus, -1)\n\t\tquery.filters.addEqual(CloseTime, -1)\n\t}\n\n\tquery.addPinotSorter(StartTime, DescendingOrder)\n\n\ttoken, err := pnt.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"next page token: %w\", err)\n\t}\n\n\tfrom := token.From\n\tpageSize := request.PageSize\n\tquery.addOffsetAndLimits(from, pageSize)\n\n\treturn query.String(), nil\n}\n\nfunc getListWorkflowExecutionsByStatusQuery(tableName string, request *p.InternalListClosedWorkflowExecutionsByStatusRequest) (string, error) {\n\tif request == nil {\n\t\treturn \"\", nil\n\t}\n\n\tquery := NewPinotQuery(tableName)\n\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\n\tstatus := 0\n\tswitch request.Status.String() {\n\tcase \"COMPLETED\":\n\t\tstatus = 0\n\tcase \"FAILED\":\n\t\tstatus = 1\n\tcase \"CANCELED\":\n\t\tstatus = 2\n\tcase \"TERMINATED\":\n\t\tstatus = 3\n\tcase \"CONTINUED_AS_NEW\":\n\t\tstatus = 4\n\tcase \"TIMED_OUT\":\n\t\tstatus = 5\n\t}\n\n\tquery.filters.addEqual(CloseStatus, status)\n\tquery.filters.addTimeRange(CloseTime, request.EarliestTime.UnixMilli(), request.LatestTime.UnixMilli()) // convert Unix Time to miliseconds\n\n\tquery.addPinotSorter(StartTime, DescendingOrder)\n\n\ttoken, err := pnt.GetNextPageToken(request.NextPageToken)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"next page token: %w\", err)\n\t}\n\n\tfrom := token.From\n\tpageSize := request.PageSize\n\tquery.addOffsetAndLimits(from, pageSize)\n\n\treturn query.String(), nil\n}\n\nfunc getGetClosedWorkflowExecutionQuery(tableName string, request *p.InternalGetClosedWorkflowExecutionRequest) string {\n\tif request == nil {\n\t\treturn \"\"\n\t}\n\n\tquery := NewPinotQuery(tableName)\n\n\tquery.filters.addEqual(DomainID, request.DomainUUID)\n\tquery.filters.addGte(CloseStatus, 0)\n\tquery.filters.addEqual(WorkflowID, request.Execution.GetWorkflowID())\n\n\trid := request.Execution.GetRunID()\n\tif rid != \"\" {\n\t\tquery.filters.addEqual(RunID, rid)\n\t}\n\n\treturn query.String()\n}\n\nfunc checkPageSize(request *p.ListWorkflowExecutionsByQueryRequest) {\n\tif request.PageSize == 0 {\n\t\trequest.PageSize = 1000\n\t}\n}\n"
  },
  {
    "path": "common/persistence/pinot/pinot_visibility_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinotvisibility\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\ttestIndex        = \"test-index\"\n\ttestDomain       = \"test-domain\"\n\ttestDomainID     = \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"\n\ttestPageSize     = 10\n\ttestEarliestTime = int64(1547596872371000000)\n\ttestLatestTime   = int64(2547596872371000000)\n\ttestWorkflowType = \"test-wf-type\"\n\ttestWorkflowID   = \"test-wid\"\n\ttestCloseStatus  = int32(1)\n\ttestTableName    = \"test-table-name\"\n\n\ttestContextTimeout = 5 * time.Second\n\n\tvalidSearchAttr  = definition.GetDefaultIndexedKeys()\n\terrNextPageToken = fmt.Errorf(\"next page token: unable to deserialize page token. err: invalid character 'e' looking for beginning of value\")\n)\n\nfunc TestRecordWorkflowExecutionStarted(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.InternalRecordWorkflowExecutionStartedRequest{\n\t\tWorkflowID: \"wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\trequest := &p.InternalRecordWorkflowExecutionStartedRequest{\n\t\tWorkflowID: \"wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t}\n\n\tcustomStringField, err := json.Marshal(\"test string\")\n\tassert.NoError(t, err)\n\trequestWithSearchAttributes := &p.InternalRecordWorkflowExecutionStartedRequest{\n\t\tWorkflowID: \"wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": customStringField,\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.InternalRecordWorkflowExecutionStartedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(&types.BadRequestError{}).Once()\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"invalid character 'e' in literal true (expecting 'r')\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case3: normal case with search attributes\": {\n\t\t\trequest: requestWithSearchAttributes,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.RecordWorkflowExecutionStarted(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRecordWorkflowExecutionClosed(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.InternalRecordWorkflowExecutionClosedRequest{\n\t\tWorkflowID: \"error-wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\trequest := &p.InternalRecordWorkflowExecutionClosedRequest{\n\t\tWorkflowID: \"wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.InternalRecordWorkflowExecutionClosedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(&types.BadRequestError{}).Once()\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"invalid character 'e' in literal true (expecting 'r')\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.RecordWorkflowExecutionClosed(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRecordWorkflowExecutionUninitialized(t *testing.T) {\n\trequest := &p.InternalRecordWorkflowExecutionUninitializedRequest{\n\t\tWorkflowID: \"wid\",\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.InternalRecordWorkflowExecutionUninitializedRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.RecordWorkflowExecutionUninitialized(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestUpsertWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\terrorRequest := &p.InternalUpsertWorkflowExecutionRequest{\n\t\tWorkflowID: \"error-wid\",\n\t\tMemo:       p.NewDataBlob([]byte(`test bytes`), constants.EncodingTypeThriftRW),\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\trequest := &p.InternalUpsertWorkflowExecutionRequest{}\n\trequest.WorkflowID = \"wid\"\n\tmemoBytes := []byte(`test bytes`)\n\trequest.Memo = p.NewDataBlob(memoBytes, constants.EncodingTypeThriftRW)\n\n\ttests := map[string]struct {\n\t\trequest                *p.InternalUpsertWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(fmt.Errorf(\"error\")).Once()\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"invalid character 'e' in literal true (expecting 'r')\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.UpsertWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\trequest := &p.VisibilityDeleteWorkflowExecutionRequest{}\n\trequest.WorkflowID = \"wid\"\n\n\ttests := map[string]struct {\n\t\trequest                *p.VisibilityDeleteWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.DeleteWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestDeleteUninitializedWorkflowExecution(t *testing.T) {\n\t// test non-empty request fields match\n\trequest := &p.VisibilityDeleteWorkflowExecutionRequest{\n\t\tDomain:     \"domain\",\n\t\tDomainID:   \"domainID\",\n\t\tWorkflowID: \"wid\",\n\t\tRunID:      \"rid\",\n\t\tTaskID:     int64(111),\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                *p.VisibilityDeleteWorkflowExecutionRequest\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: normal case\": {\n\t\t\trequest: request,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\tassert.Equal(t, request.WorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\tmockPinotClient.EXPECT().CountByQuery(gomock.Any()).Return(int64(1), nil).Times(1)\n\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := visibilityStore.DeleteUninitializedWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsRequest{\n\t\tDomain:        DomainID,\n\t\tNextPageToken: []byte(\"error\"),\n\t}\n\n\trequest := &p.InternalListWorkflowExecutionsRequest{\n\t\tDomain: DomainID,\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListOpenWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsRequest{\n\t\tDomain:        DomainID,\n\t\tNextPageToken: []byte(\"error\"),\n\t}\n\n\trequest := &p.InternalListWorkflowExecutionsRequest{\n\t\tDomain: DomainID,\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListClosedWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsByType(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomain:        DomainID,\n\t\t\tNextPageToken: []byte(\"error\"),\n\t\t},\n\t\tWorkflowTypeName: \"\",\n\t}\n\n\trequest := &p.InternalListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsByTypeRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListOpenWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsByType(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomain:        DomainID,\n\t\t\tNextPageToken: []byte(\"error\"),\n\t\t},\n\t\tWorkflowTypeName: \"\",\n\t}\n\trequest := &p.InternalListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsByTypeRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListClosedWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsByWorkflowID(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomain:        DomainID,\n\t\t\tNextPageToken: []byte(\"error\"),\n\t\t},\n\t}\n\trequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsByWorkflowIDRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsByWorkflowID(t *testing.T) {\n\terrorRequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomain:        DomainID,\n\t\t\tNextPageToken: []byte(\"error\"),\n\t\t},\n\t}\n\trequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListWorkflowExecutionsByWorkflowIDRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsByStatus(t *testing.T) {\n\terrorRequest := &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomain:        DomainID,\n\t\t\tNextPageToken: []byte(\"error\"),\n\t\t},\n\t}\n\trequest := &p.InternalListClosedWorkflowExecutionsByStatusRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalListClosedWorkflowExecutionsByStatusRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListClosedWorkflowExecutionsByStatus(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetClosedWorkflowExecution(t *testing.T) {\n\terrorRequest := &p.InternalGetClosedWorkflowExecutionRequest{}\n\trequest := &p.InternalGetClosedWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.InternalGetClosedWorkflowExecutionRequest\n\t\texpectedResp              *p.InternalGetClosedWorkflowExecutionRequest\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(&pnt.SearchResponse{\n\t\t\t\t\tExecutions: []*p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tDomainID: DomainID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"Pinot GetClosedWorkflowExecution failed, error\"),\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(&pnt.SearchResponse{\n\t\t\t\t\tExecutions: []*p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tDomainID: DomainID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\t_, err := visibilityStore.GetClosedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tNextPageToken: []byte(\"error\"),\n\t}\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByQueryRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ListWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScanWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.ListWorkflowExecutionsByQueryRequest{\n\t\tNextPageToken: []byte(\"error\"),\n\t}\n\trequest := &p.ListWorkflowExecutionsByQueryRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.ListWorkflowExecutionsByQueryRequest\n\t\texpectedResp              *p.InternalListWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: errNextPageToken,\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().Search(gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.ScanWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCountWorkflowExecutions(t *testing.T) {\n\terrorRequest := &p.CountWorkflowExecutionsRequest{}\n\trequest := &p.CountWorkflowExecutionsRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *p.CountWorkflowExecutionsRequest\n\t\texpectedResp              *p.CountWorkflowExecutionsResponse\n\t\tpinotClientMockAffordance func(mockPinotClient *pnt.MockGenericClient)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest:      errorRequest,\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().CountByQuery(gomock.Any()).Return(int64(0), fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"CountClosedWorkflowExecutions failed, error\"),\n\t\t},\n\t\t\"Case2: normal case with nil response\": {\n\t\t\trequest:      request,\n\t\t\texpectedResp: &p.CountWorkflowExecutionsResponse{Count: 1},\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().CountByQuery(gomock.Any()).Return(int64(1), nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case3: query error case\": {\n\t\t\trequest:      &p.CountWorkflowExecutionsRequest{Domain: testDomain, DomainUUID: testDomainID, Query: \"CustomKeywordField = missing\"},\n\t\t\texpectedResp: nil,\n\t\t\tpinotClientMockAffordance: func(mockPinotClient *pnt.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().GetTableName().Return(testTableName).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"pinot query validator error: invalid comparison expression, right, query: CustomKeywordField = missing\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\ttest.pinotClientMockAffordance(mockPinotClient)\n\n\t\t\tresp, err := visibilityStore.CountWorkflowExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResp, resp)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetName(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\tmockProducer := &mocks.KafkaProducer{}\n\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t}, mockProducer, log.NewNoop())\n\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\tassert.NotEmpty(t, visibilityStore.GetName())\n}\n\nfunc TestNewPinotVisibilityStore(t *testing.T) {\n\tmockPinotClient := &pnt.MockGenericClient{}\n\tassert.NotPanics(t, func() {\n\t\tNewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\tValidSearchAttributes: dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t}, nil, log.NewNoop())\n\t})\n}\n\nfunc TestGetCountWorkflowExecutionsQuery(t *testing.T) {\n\temptyQueryRequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tQuery:      \"\",\n\t}\n\n\texpectEmptyQueryResult := fmt.Sprintf(`SELECT COUNT(*)\nFROM test-table-name\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\n`)\n\n\trequest := &p.CountWorkflowExecutionsRequest{\n\t\tDomainUUID: testDomainID,\n\t\tDomain:     testDomain,\n\t\tQuery:      \"WorkflowID = 'wfid'\",\n\t}\n\n\texpectResult := fmt.Sprintf(`SELECT COUNT(*)\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowID = 'wfid'\n`, testTableName)\n\n\ttests := map[string]struct {\n\t\trequest       *p.CountWorkflowExecutionsRequest\n\t\texpectedRes   string\n\t\texpectedError error\n\t}{\n\t\t\"Case1: normal case with nil response\": {\n\t\t\trequest:       request,\n\t\t\texpectedRes:   expectResult,\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case2: normal case with empty query\": {\n\t\t\trequest:       emptyQueryRequest,\n\t\t\texpectedRes:   expectEmptyQueryResult,\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case3: custom attr is missing case\": {\n\t\t\trequest: &p.CountWorkflowExecutionsRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tDomain:     testDomain,\n\t\t\t\tQuery:      \"CustomKeywordField = missing\",\n\t\t\t},\n\t\t\texpectedRes:   \"\",\n\t\t\texpectedError: fmt.Errorf(\"pinot query validator error: invalid comparison expression, right, query: CustomKeywordField = missing\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\tres, err := visibilityStore.getCountWorkflowExecutionsQuery(testTableName, test.request)\n\t\t\tassert.Equal(t, test.expectedRes, res)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetListWorkflowExecutionQuery(t *testing.T) {\n\ttoken := pnt.PinotVisibilityPageToken{\n\t\tFrom: 11,\n\t}\n\n\tserializedToken, err := json.Marshal(token)\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"Serialized error in PinotVisibilityStoreTest!!!, %s\", err))\n\t}\n\n\ttests := map[string]struct {\n\t\tinput          *p.ListWorkflowExecutionsByQueryRequest\n\t\texpectedOutput string\n\t\texpectedError  bool\n\t}{\n\t\t\"complete request with keyword query only\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"`Attr.CustomKeywordField` = 'keywordCustomized'\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(\n\t\t\t\t`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND (JSON_MATCH(Attr, '\"$.CustomKeywordField\"=''keywordCustomized''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"=''keywordCustomized'''))\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request from search attribute worker\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CustomIntField=2 and CustomKeywordField='Update2' order by `Attr.CustomDatetimeField` DESC\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(\n\t\t\t\t`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND JSON_MATCH(Attr, '\"$.CustomIntField\"=''2''') and (JSON_MATCH(Attr, '\"$.CustomKeywordField\"=''Update2''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"=''Update2'''))\norder by CustomDatetimeField DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with keyword query and other customized query\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CustomKeywordField = 'keywordCustomized' and CustomStringField = 'String and or order by'\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND (JSON_MATCH(Attr, '\"$.CustomKeywordField\"=''keywordCustomized''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"=''keywordCustomized''')) and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*String and or order by.*'')')\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with or query & customized attributes\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CustomStringField = 'Or' or CustomStringField = 'and' Order by StartTime DESC\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND (JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*Or.*'')') or JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*and.*'')'))\nOrder by StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complex query\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"WorkflowID = 'wid' and ((CustomStringField = 'custom and custom2 or custom3 order by') or CustomIntField between 1 and 10)\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowID = 'wid' and (JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*custom and custom2 or custom3 order by.*'')') or (JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) >= 1 AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) <= 10))\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"or clause with custom attributes\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CustomIntField = 1 or CustomIntField = 2\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND (JSON_MATCH(Attr, '\"$.CustomIntField\"=''1''') or JSON_MATCH(Attr, '\"$.CustomIntField\"=''2'''))\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with customized query with missing\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CloseTime = missing anD WorkflowType = 'some-test-workflow'\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseTime = -1 and WorkflowType = 'some-test-workflow'\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with customized query with NextPageToken\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: serializedToken,\n\t\t\t\tQuery:         \"CloseStatus = -1 and CustomKeywordField = 'keywordCustomized' AND CustomIntField<=10 and CustomStringField = 'String field is for text' Order by DomainID Desc\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = -1 and (JSON_MATCH(Attr, '\"$.CustomKeywordField\"=''keywordCustomized''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"=''keywordCustomized''')) and (JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) <= 10) and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*String field is for text.*'')')\nOrder by DomainID Desc\nLIMIT 11, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with order by query\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"Order by DomainId Desc\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nOrder by DomainId Desc\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with filter query\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"CloseStatus < 0\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus < 0\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"complete request with empty query\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"\",\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"empty request\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = ''\nLIMIT 0, 0\n`, testTableName),\n\t\t\texpectedError: false,\n\t\t},\n\n\t\t\"nil request\": {\n\t\t\tinput:          nil,\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  false,\n\t\t},\n\t\t\"request with syntax error\": {\n\t\t\tinput: &p.ListWorkflowExecutionsByQueryRequest{\n\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\tDomain:        testDomain,\n\t\t\t\tPageSize:      testPageSize,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tQuery:         \"WorkflowType = test\",\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  true,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\t\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\t\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t}, mockProducer, log.NewNoop())\n\t\t\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\t\t\toutput, err := visibilityStore.getListWorkflowExecutionsByQueryQuery(testTableName, test.input)\n\t\t\tassert.Equal(t, test.expectedOutput, output)\n\t\t\tif test.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetListWorkflowExecutionsQuery(t *testing.T) {\n\trequest := &p.InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    testDomainID,\n\t\tDomain:        testDomain,\n\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\tPageSize:      testPageSize,\n\t\tNextPageToken: nil,\n\t}\n\n\tcloseResult, err1 := getListWorkflowExecutionsQuery(testTableName, request, true)\n\topenResult, err2 := getListWorkflowExecutionsQuery(testTableName, request, false)\n\tnilResult, err3 := getListWorkflowExecutionsQuery(testTableName, nil, true)\n\texpectCloseResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus >= 0\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectOpenResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND StartTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus = -1\nAND CloseTime = -1\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectNilResult := \"\"\n\n\tassert.Equal(t, closeResult, expectCloseResult)\n\tassert.Equal(t, openResult, expectOpenResult)\n\tassert.Equal(t, nilResult, expectNilResult)\n\tassert.NoError(t, err1)\n\tassert.NoError(t, err2)\n\tassert.NoError(t, err3)\n}\n\nfunc TestGetListWorkflowExecutionsByTypeQuery(t *testing.T) {\n\trequest := &p.InternalListWorkflowExecutionsByTypeRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:    testDomainID,\n\t\t\tDomain:        testDomain,\n\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\tPageSize:      testPageSize,\n\t\t\tNextPageToken: nil,\n\t\t},\n\t\tWorkflowTypeName: testWorkflowType,\n\t}\n\n\tcloseResult, err1 := getListWorkflowExecutionsByTypeQuery(testTableName, request, true)\n\topenResult, err2 := getListWorkflowExecutionsByTypeQuery(testTableName, request, false)\n\tnilResult, err3 := getListWorkflowExecutionsByTypeQuery(testTableName, nil, true)\n\texpectCloseResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowType = 'test-wf-type'\nAND CloseTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus >= 0\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectOpenResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowType = 'test-wf-type'\nAND StartTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus = -1\nAND CloseTime = -1\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectNilResult := \"\"\n\n\tassert.Equal(t, closeResult, expectCloseResult)\n\tassert.Equal(t, openResult, expectOpenResult)\n\tassert.Equal(t, nilResult, expectNilResult)\n\tassert.NoError(t, err1)\n\tassert.NoError(t, err2)\n\tassert.NoError(t, err3)\n}\n\nfunc TestGetListWorkflowExecutionsByWorkflowIDQuery(t *testing.T) {\n\trequest := &p.InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\tDomainUUID:    testDomainID,\n\t\t\tDomain:        testDomain,\n\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\tPageSize:      testPageSize,\n\t\t\tNextPageToken: nil,\n\t\t},\n\t\tWorkflowID: testWorkflowID,\n\t}\n\n\tcloseResult, err1 := getListWorkflowExecutionsByWorkflowIDQuery(testTableName, request, true)\n\topenResult, err2 := getListWorkflowExecutionsByWorkflowIDQuery(testTableName, request, false)\n\tnilResult, err3 := getListWorkflowExecutionsByWorkflowIDQuery(testTableName, nil, true)\n\texpectCloseResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowID = 'test-wid'\nAND CloseTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus >= 0\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectOpenResult := fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND WorkflowID = 'test-wid'\nAND StartTime BETWEEN 1547596871371 AND 2547596873371\nAND CloseStatus = -1\nAND CloseTime = -1\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName)\n\texpectNilResult := \"\"\n\n\tassert.Equal(t, closeResult, expectCloseResult)\n\tassert.Equal(t, openResult, expectOpenResult)\n\tassert.Equal(t, nilResult, expectNilResult)\n\tassert.NoError(t, err1)\n\tassert.NoError(t, err2)\n\tassert.NoError(t, err3)\n}\n\nfunc TestGetListWorkflowExecutionsByStatusQuery(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinputRequest *p.InternalListClosedWorkflowExecutionsByStatusRequest\n\t\texpectResult string\n\t\texpectError  error\n\t}{\n\t\t\"Case1: normal case\": {\n\t\t\tinputRequest: nil,\n\t\t\texpectResult: \"\",\n\t\t\texpectError:  nil,\n\t\t},\n\t\t\"Case2-0: normal case with close status is 0\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(0),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 0\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2-1: normal case with close status is 1\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(1),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 1\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2-2: normal case with close status is 2\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(2),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 2\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2-3: normal case with close status is 3\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(3),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 3\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2-4: normal case with close status is 4\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(4),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 4\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2-5: normal case with close status is 5\": {\n\t\t\tinputRequest: &p.InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\tInternalListWorkflowExecutionsRequest: p.InternalListWorkflowExecutionsRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tDomain:        testDomain,\n\t\t\t\t\tEarliestTime:  time.Unix(0, testEarliestTime),\n\t\t\t\t\tLatestTime:    time.Unix(0, testLatestTime),\n\t\t\t\t\tPageSize:      testPageSize,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\tStatus: types.WorkflowExecutionCloseStatus(5),\n\t\t\t},\n\t\t\texpectResult: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus = 5\nAND CloseTime BETWEEN 1547596872371 AND 2547596872371\nOrder BY StartTime DESC\nLIMIT 0, 10\n`, testTableName),\n\t\t\texpectError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactualResult, actualError := getListWorkflowExecutionsByStatusQuery(testTableName, test.inputRequest)\n\t\t\tassert.Equal(t, test.expectResult, actualResult)\n\t\t\tassert.NoError(t, actualError)\n\t\t})\n\t}\n}\n\nfunc TestGetGetClosedWorkflowExecutionQuery(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput          *p.InternalGetClosedWorkflowExecutionRequest\n\t\texpectedOutput string\n\t}{\n\t\t\"complete request with empty RunId\": {\n\t\t\tinput: &p.InternalGetClosedWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tDomain:     testDomain,\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus >= 0\nAND WorkflowID = 'test-wid'\n`, testTableName),\n\t\t},\n\n\t\t\"complete request with runId\": {\n\t\t\tinput: &p.InternalGetClosedWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tDomain:     testDomain,\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      \"runid\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = 'bfd5c907-f899-4baf-a7b2-2ab85e623ebd'\nAND CloseStatus >= 0\nAND WorkflowID = 'test-wid'\nAND RunID = 'runid'\n`, testTableName),\n\t\t},\n\n\t\t\"empty request\": {\n\t\t\tinput: &p.InternalGetClosedWorkflowExecutionRequest{},\n\t\t\texpectedOutput: fmt.Sprintf(`SELECT *\nFROM %s\nWHERE DomainID = ''\nAND CloseStatus >= 0\nAND WorkflowID = ''\n`, testTableName),\n\t\t},\n\n\t\t\"nil request\": {\n\t\t\tinput:          nil,\n\t\t\texpectedOutput: \"\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\toutput := getGetClosedWorkflowExecutionQuery(testTableName, test.input)\n\t\t\tassert.Equal(t, test.expectedOutput, output)\n\t\t})\n\t}\n}\n\nfunc TestParseLastElement(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput           string\n\t\texpectedElement string\n\t\texpectedOrderBy string\n\t}{\n\t\t\"Case1: only contains order by\": {\n\t\t\tinput:           \"Order by TestInt DESC\",\n\t\t\texpectedElement: \"\",\n\t\t\texpectedOrderBy: \"Order by TestInt DESC\",\n\t\t},\n\t\t\"Case2: only contains order by\": {\n\t\t\tinput:           \"TestString = 'cannot be used in order by'\",\n\t\t\texpectedElement: \"TestString = 'cannot be used in order by'\",\n\t\t\texpectedOrderBy: \"\",\n\t\t},\n\t\t\"Case3: not contains any order by\": {\n\t\t\tinput:           \"TestInt = 1\",\n\t\t\texpectedElement: \"TestInt = 1\",\n\t\t\texpectedOrderBy: \"\",\n\t\t},\n\t\t\"Case4-1: with order by in string & real order by\": {\n\t\t\tinput:           \"TestString = 'cannot be used in order by' Order by TestInt DESC\",\n\t\t\texpectedElement: \"TestString = 'cannot be used in order by'\",\n\t\t\texpectedOrderBy: \"Order by TestInt DESC\",\n\t\t},\n\t\t\"Case4-2: with non-string attribute & real order by\": {\n\t\t\tinput:           \"TestDouble = 1.0 Order by TestInt DESC\",\n\t\t\texpectedElement: \"TestDouble = 1.0\",\n\t\t\texpectedOrderBy: \"Order by TestInt DESC\",\n\t\t},\n\t\t\"Case5: with random case order by\": {\n\t\t\tinput:           \"TestString = 'cannot be used in OrDer by' ORdeR by TestInt DESC\",\n\t\t\texpectedElement: \"TestString = 'cannot be used in OrDer by'\",\n\t\t\texpectedOrderBy: \"ORdeR by TestInt DESC\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\telement, orderBy := parseOrderBy(test.input)\n\t\t\tassert.Equal(t, test.expectedElement, element)\n\t\t\tassert.Equal(t, test.expectedOrderBy, orderBy)\n\t\t})\n\t}\n}\n\nfunc TestSplitElement(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput       string\n\t\texpectedKey string\n\t\texpectedVal string\n\t\texpectedOp  string\n\t}{\n\t\t\"Case1-1: A=B\": {\n\t\t\tinput:       \"CustomizedTestField=Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \"=\",\n\t\t},\n\t\t\"Case1-2: A=\\\"B\\\"\": {\n\t\t\tinput:       \"CustomizedTestField=\\\"Test\\\"\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"\\\"Test\\\"\",\n\t\t\texpectedOp:  \"=\",\n\t\t},\n\t\t\"Case1-3: A='B'\": {\n\t\t\tinput:       \"CustomizedTestField='Test'\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"'Test'\",\n\t\t\texpectedOp:  \"=\",\n\t\t},\n\t\t\"Case2: A<=B\": {\n\t\t\tinput:       \"CustomizedTestField<=Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \"<=\",\n\t\t},\n\t\t\"Case3: A>=B\": {\n\t\t\tinput:       \"CustomizedTestField>=Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \">=\",\n\t\t},\n\t\t\"Case4: A = B\": {\n\t\t\tinput:       \"CustomizedTestField = Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \"=\",\n\t\t},\n\t\t\"Case5: A <= B\": {\n\t\t\tinput:       \"CustomizedTestField <= Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \"<=\",\n\t\t},\n\t\t\"Case6: A >= B\": {\n\t\t\tinput:       \"CustomizedTestField >= Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \">=\",\n\t\t},\n\t\t\"Case7: A > B\": {\n\t\t\tinput:       \"CustomizedTestField > Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \">\",\n\t\t},\n\t\t\"Case8: A < B\": {\n\t\t\tinput:       \"CustomizedTestField < Test\",\n\t\t\texpectedKey: \"CustomizedTestField\",\n\t\t\texpectedVal: \"Test\",\n\t\t\texpectedOp:  \"<\",\n\t\t},\n\t\t\"Case9: empty\": {\n\t\t\tinput:       \"\",\n\t\t\texpectedKey: \"\",\n\t\t\texpectedVal: \"\",\n\t\t\texpectedOp:  \"\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tkey, val, op := splitElement(test.input)\n\t\t\tassert.Equal(t, test.expectedKey, key)\n\t\t\tassert.Equal(t, test.expectedVal, val)\n\t\t\tassert.Equal(t, test.expectedOp, op)\n\t\t})\n\t}\n}\n\nfunc TestIsTimeStruct(t *testing.T) {\n\tvar emptyInput []byte\n\tnumberInput := []byte(\"1709601210000000000\")\n\terrorInput := []byte(\"Not a timeStamp\")\n\ttestTime := time.UnixMilli(1709601210000)\n\tvar legitInput []byte\n\tlegitInput, err := json.Marshal(testTime)\n\tassert.NoError(t, err)\n\tlegitOutput := testTime.UnixMilli()\n\tlegitOutputJSON, _ := json.Marshal(legitOutput)\n\n\ttests := map[string]struct {\n\t\tinput          []byte\n\t\texpectedOutput []byte\n\t\texpectedError  error\n\t}{\n\t\t\"Case1: empty input\": {\n\t\t\tinput:          emptyInput,\n\t\t\texpectedOutput: nil,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t\"Case2: error input\": {\n\t\t\tinput:          errorInput,\n\t\t\texpectedOutput: errorInput,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t\"Case3: number input\": {\n\t\t\tinput:          numberInput,\n\t\t\texpectedOutput: numberInput,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t\"Case4: legit input\": {\n\t\t\tinput:          legitInput,\n\t\t\texpectedOutput: legitOutputJSON,\n\t\t\texpectedError:  nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactualOutput, actualError := isTimeStruct(test.input)\n\t\t\tassert.Equal(t, test.expectedOutput, actualOutput)\n\t\t\tassert.Equal(t, test.expectedError, actualError)\n\t\t})\n\t}\n}\n\nfunc TestClose(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockPinotClient := pnt.NewMockGenericClient(ctrl)\n\tmockProducer := &mocks.KafkaProducer{}\n\tmgr := NewPinotVisibilityStore(mockPinotClient, &service.Config{\n\t\tValidSearchAttributes:      dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tESIndexMaxResultWindow:     dynamicproperties.GetIntPropertyFn(3),\n\t\tPinotOptimizedQueryColumns: dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t}, mockProducer, log.NewNoop())\n\tvisibilityStore := mgr.(*pinotVisibilityStore)\n\n\tassert.NotPanics(t, func() {\n\t\tvisibilityStore.Close()\n\t})\n}\n"
  },
  {
    "path": "common/persistence/queue_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\tqueueManager struct {\n\t\tpersistence QueueStore\n\t\ttimeSrc     clock.TimeSource\n\t}\n)\n\nvar _ QueueManager = (*queueManager)(nil)\n\n// NewQueueManager returns a new QueueManager\nfunc NewQueueManager(\n\tpersistence QueueStore,\n) QueueManager {\n\treturn &queueManager{\n\t\tpersistence: persistence,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t}\n}\n\nfunc (q *queueManager) Close() {\n\tq.persistence.Close()\n}\n\nfunc (q *queueManager) EnqueueMessage(ctx context.Context, request *EnqueueMessageRequest) error {\n\tcurrentTimestamp := q.timeSrc.Now()\n\treturn q.persistence.EnqueueMessage(ctx, &InternalEnqueueMessageRequest{\n\t\tMessagePayload:   request.MessagePayload,\n\t\tCurrentTimeStamp: currentTimestamp,\n\t})\n}\n\nfunc (q *queueManager) ReadMessages(ctx context.Context, request *ReadMessagesRequest) (*ReadMessagesResponse, error) {\n\tresp, err := q.persistence.ReadMessages(ctx, &InternalReadMessagesRequest{\n\t\tLastMessageID: request.LastMessageID,\n\t\tMaxCount:      request.MaxCount,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\toutput := make(QueueMessageList, 0, len(resp.Messages))\n\tfor _, message := range resp.Messages {\n\t\toutput = append(output, q.fromInternalQueueMessage(message))\n\t}\n\treturn &ReadMessagesResponse{Messages: output}, nil\n}\n\nfunc (q *queueManager) DeleteMessagesBefore(ctx context.Context, request *DeleteMessagesBeforeRequest) error {\n\treturn q.persistence.DeleteMessagesBefore(ctx, &InternalDeleteMessagesBeforeRequest{\n\t\tMessageID: request.MessageID,\n\t})\n}\n\nfunc (q *queueManager) UpdateAckLevel(ctx context.Context, request *UpdateAckLevelRequest) error {\n\tcurrentTimestamp := q.timeSrc.Now()\n\treturn q.persistence.UpdateAckLevel(ctx, &InternalUpdateAckLevelRequest{\n\t\tMessageID:        request.MessageID,\n\t\tClusterName:      request.ClusterName,\n\t\tCurrentTimeStamp: currentTimestamp,\n\t})\n}\n\nfunc (q *queueManager) GetAckLevels(ctx context.Context, request *GetAckLevelsRequest) (*GetAckLevelsResponse, error) {\n\tresp, err := q.persistence.GetAckLevels(ctx, &InternalGetAckLevelsRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &GetAckLevelsResponse{AckLevels: resp.AckLevels}, nil\n}\n\nfunc (q *queueManager) EnqueueMessageToDLQ(ctx context.Context, request *EnqueueMessageToDLQRequest) error {\n\tcurrentTimestamp := q.timeSrc.Now()\n\treturn q.persistence.EnqueueMessageToDLQ(ctx, &InternalEnqueueMessageToDLQRequest{\n\t\tMessagePayload:   request.MessagePayload,\n\t\tCurrentTimeStamp: currentTimestamp,\n\t})\n}\n\nfunc (q *queueManager) ReadMessagesFromDLQ(ctx context.Context, request *ReadMessagesFromDLQRequest) (*ReadMessagesFromDLQResponse, error) {\n\tresp, err := q.persistence.ReadMessagesFromDLQ(ctx, &InternalReadMessagesFromDLQRequest{\n\t\tFirstMessageID: request.FirstMessageID,\n\t\tLastMessageID:  request.LastMessageID,\n\t\tPageSize:       request.PageSize,\n\t\tPageToken:      request.PageToken,\n\t})\n\tif resp == nil {\n\t\treturn nil, err\n\t}\n\toutput := make([]*QueueMessage, 0, len(resp.Messages))\n\tfor _, message := range resp.Messages {\n\t\toutput = append(output, q.fromInternalQueueMessage(message))\n\t}\n\treturn &ReadMessagesFromDLQResponse{\n\t\tMessages:      output,\n\t\tNextPageToken: resp.NextPageToken,\n\t}, err\n}\n\nfunc (q *queueManager) DeleteMessageFromDLQ(ctx context.Context, request *DeleteMessageFromDLQRequest) error {\n\treturn q.persistence.DeleteMessageFromDLQ(ctx, &InternalDeleteMessageFromDLQRequest{\n\t\tMessageID: request.MessageID,\n\t})\n}\n\nfunc (q *queueManager) RangeDeleteMessagesFromDLQ(ctx context.Context, request *RangeDeleteMessagesFromDLQRequest) error {\n\treturn q.persistence.RangeDeleteMessagesFromDLQ(ctx, &InternalRangeDeleteMessagesFromDLQRequest{\n\t\tFirstMessageID: request.FirstMessageID,\n\t\tLastMessageID:  request.LastMessageID,\n\t})\n}\n\nfunc (q *queueManager) UpdateDLQAckLevel(ctx context.Context, request *UpdateDLQAckLevelRequest) error {\n\tcurrentTimestamp := q.timeSrc.Now()\n\treturn q.persistence.UpdateDLQAckLevel(ctx, &InternalUpdateDLQAckLevelRequest{\n\t\tMessageID:        request.MessageID,\n\t\tClusterName:      request.ClusterName,\n\t\tCurrentTimeStamp: currentTimestamp,\n\t})\n}\n\nfunc (q *queueManager) GetDLQAckLevels(ctx context.Context, request *GetDLQAckLevelsRequest) (*GetDLQAckLevelsResponse, error) {\n\tresp, err := q.persistence.GetDLQAckLevels(ctx, &InternalGetDLQAckLevelsRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &GetDLQAckLevelsResponse{AckLevels: resp.AckLevels}, nil\n}\n\nfunc (q *queueManager) GetDLQSize(ctx context.Context, request *GetDLQSizeRequest) (*GetDLQSizeResponse, error) {\n\tresp, err := q.persistence.GetDLQSize(ctx, &InternalGetDLQSizeRequest{})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &GetDLQSizeResponse{Size: resp.Size}, nil\n}\n\nfunc (q *queueManager) fromInternalQueueMessage(message *InternalQueueMessage) *QueueMessage {\n\treturn &QueueMessage{\n\t\tID:        message.ID,\n\t\tQueueType: message.QueueType,\n\t\tPayload:   message.Payload,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/retryer.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination retryer_mock.go -package persistence github.com/uber/cadence/common/persistence Retryer\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n)\n\n// Retryer is used to retry requests to persistence with provided retry policy\ntype Retryer interface {\n\tListConcreteExecutions(context.Context, *ListConcreteExecutionsRequest) (*ListConcreteExecutionsResponse, error)\n\tListCurrentExecutions(context.Context, *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error)\n\tGetWorkflowExecution(context.Context, *GetWorkflowExecutionRequest) (*GetWorkflowExecutionResponse, error)\n\tGetCurrentExecution(context.Context, *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error)\n\tIsWorkflowExecutionExists(context.Context, *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error)\n\tReadHistoryBranch(context.Context, *ReadHistoryBranchRequest) (*ReadHistoryBranchResponse, error)\n\tDeleteWorkflowExecution(context.Context, *DeleteWorkflowExecutionRequest) error\n\tDeleteCurrentWorkflowExecution(context.Context, *DeleteCurrentWorkflowExecutionRequest) error\n\tGetShardID() int\n\tGetHistoryTasks(context.Context, *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error)\n\tCompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error\n}\n\ntype (\n\tpersistenceRetryer struct {\n\t\texecManager    ExecutionManager\n\t\thistoryManager HistoryManager\n\t\tthrottleRetry  *backoff.ThrottleRetry\n\t}\n)\n\n// NewPersistenceRetryer constructs a new Retryer\nfunc NewPersistenceRetryer(\n\texecManager ExecutionManager,\n\thistoryManager HistoryManager,\n\tpolicy backoff.RetryPolicy,\n) Retryer {\n\treturn &persistenceRetryer{\n\t\texecManager:    execManager,\n\t\thistoryManager: historyManager,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\tbackoff.WithRetryableError(IsTransientError),\n\t\t),\n\t}\n}\n\n// ListConcreteExecutions retries ListConcreteExecutions\nfunc (pr *persistenceRetryer) ListConcreteExecutions(\n\tctx context.Context,\n\treq *ListConcreteExecutionsRequest,\n) (*ListConcreteExecutionsResponse, error) {\n\tvar resp *ListConcreteExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.ListConcreteExecutions(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err == nil {\n\t\treturn resp, nil\n\t}\n\treturn nil, err\n}\n\n// GetWorkflowExecution retries GetWorkflowExecution\nfunc (pr *persistenceRetryer) GetWorkflowExecution(\n\tctx context.Context,\n\treq *GetWorkflowExecutionRequest,\n) (*GetWorkflowExecutionResponse, error) {\n\tvar resp *GetWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.GetWorkflowExecution(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// GetCurrentExecution retries GetCurrentExecution\nfunc (pr *persistenceRetryer) GetCurrentExecution(\n\tctx context.Context,\n\treq *GetCurrentExecutionRequest,\n) (*GetCurrentExecutionResponse, error) {\n\tvar resp *GetCurrentExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.GetCurrentExecution(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// ListCurrentExecutions retries ListCurrentExecutions\nfunc (pr *persistenceRetryer) ListCurrentExecutions(\n\tctx context.Context,\n\treq *ListCurrentExecutionsRequest,\n) (*ListCurrentExecutionsResponse, error) {\n\tvar resp *ListCurrentExecutionsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.ListCurrentExecutions(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err == nil {\n\t\treturn resp, nil\n\t}\n\treturn nil, err\n}\n\n// IsWorkflowExecutionExists retries IsWorkflowExecutionExists\nfunc (pr *persistenceRetryer) IsWorkflowExecutionExists(\n\tctx context.Context,\n\treq *IsWorkflowExecutionExistsRequest,\n) (*IsWorkflowExecutionExistsResponse, error) {\n\tvar resp *IsWorkflowExecutionExistsResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.IsWorkflowExecutionExists(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// ReadHistoryBranch retries ReadHistoryBranch\nfunc (pr *persistenceRetryer) ReadHistoryBranch(\n\tctx context.Context,\n\treq *ReadHistoryBranchRequest,\n) (*ReadHistoryBranchResponse, error) {\n\tvar resp *ReadHistoryBranchResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.historyManager.ReadHistoryBranch(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\n// DeleteWorkflowExecution retries DeleteWorkflowExecution\nfunc (pr *persistenceRetryer) DeleteWorkflowExecution(\n\tctx context.Context,\n\treq *DeleteWorkflowExecutionRequest,\n) error {\n\top := func(ctx context.Context) error {\n\t\treturn pr.execManager.DeleteWorkflowExecution(ctx, req)\n\t}\n\treturn pr.throttleRetry.Do(ctx, op)\n}\n\n// DeleteCurrentWorkflowExecution retries DeleteCurrentWorkflowExecution\nfunc (pr *persistenceRetryer) DeleteCurrentWorkflowExecution(\n\tctx context.Context,\n\treq *DeleteCurrentWorkflowExecutionRequest,\n) error {\n\top := func(ctx context.Context) error {\n\t\treturn pr.execManager.DeleteCurrentWorkflowExecution(ctx, req)\n\t}\n\treturn pr.throttleRetry.Do(ctx, op)\n}\n\n// GetShardID return shard id\nfunc (pr *persistenceRetryer) GetShardID() int {\n\treturn pr.execManager.GetShardID()\n}\n\n// GetHistoryTasks retries GetHistoryTasks\nfunc (pr *persistenceRetryer) GetHistoryTasks(\n\tctx context.Context,\n\treq *GetHistoryTasksRequest,\n) (*GetHistoryTasksResponse, error) {\n\tvar resp *GetHistoryTasksResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = pr.execManager.GetHistoryTasks(ctx, req)\n\t\treturn err\n\t}\n\terr := pr.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp, nil\n}\n\n// CompleteHistoryTask is a retryable version of CompleteHistoryTask method\nfunc (pr *persistenceRetryer) CompleteHistoryTask(\n\tctx context.Context,\n\trequest *CompleteHistoryTaskRequest,\n) error {\n\top := func(ctx context.Context) error {\n\t\treturn pr.execManager.CompleteHistoryTask(ctx, request)\n\t}\n\n\treturn pr.throttleRetry.Do(ctx, op)\n}\n"
  },
  {
    "path": "common/persistence/retryer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: retryer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -source retryer.go -destination retryer_mock.go -package persistence github.com/uber/cadence/common/persistence Retryer\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockRetryer is a mock of Retryer interface.\ntype MockRetryer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockRetryerMockRecorder\n\tisgomock struct{}\n}\n\n// MockRetryerMockRecorder is the mock recorder for MockRetryer.\ntype MockRetryerMockRecorder struct {\n\tmock *MockRetryer\n}\n\n// NewMockRetryer creates a new mock instance.\nfunc NewMockRetryer(ctrl *gomock.Controller) *MockRetryer {\n\tmock := &MockRetryer{ctrl: ctrl}\n\tmock.recorder = &MockRetryerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRetryer) EXPECT() *MockRetryerMockRecorder {\n\treturn m.recorder\n}\n\n// CompleteHistoryTask mocks base method.\nfunc (m *MockRetryer) CompleteHistoryTask(ctx context.Context, request *CompleteHistoryTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteHistoryTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteHistoryTask indicates an expected call of CompleteHistoryTask.\nfunc (mr *MockRetryerMockRecorder) CompleteHistoryTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteHistoryTask\", reflect.TypeOf((*MockRetryer)(nil).CompleteHistoryTask), ctx, request)\n}\n\n// DeleteCurrentWorkflowExecution mocks base method.\nfunc (m *MockRetryer) DeleteCurrentWorkflowExecution(arg0 context.Context, arg1 *DeleteCurrentWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteCurrentWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteCurrentWorkflowExecution indicates an expected call of DeleteCurrentWorkflowExecution.\nfunc (mr *MockRetryerMockRecorder) DeleteCurrentWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteCurrentWorkflowExecution\", reflect.TypeOf((*MockRetryer)(nil).DeleteCurrentWorkflowExecution), arg0, arg1)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockRetryer) DeleteWorkflowExecution(arg0 context.Context, arg1 *DeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockRetryerMockRecorder) DeleteWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockRetryer)(nil).DeleteWorkflowExecution), arg0, arg1)\n}\n\n// GetCurrentExecution mocks base method.\nfunc (m *MockRetryer) GetCurrentExecution(arg0 context.Context, arg1 *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*GetCurrentExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCurrentExecution indicates an expected call of GetCurrentExecution.\nfunc (mr *MockRetryerMockRecorder) GetCurrentExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentExecution\", reflect.TypeOf((*MockRetryer)(nil).GetCurrentExecution), arg0, arg1)\n}\n\n// GetHistoryTasks mocks base method.\nfunc (m *MockRetryer) GetHistoryTasks(arg0 context.Context, arg1 *GetHistoryTasksRequest) (*GetHistoryTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryTasks\", arg0, arg1)\n\tret0, _ := ret[0].(*GetHistoryTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetHistoryTasks indicates an expected call of GetHistoryTasks.\nfunc (mr *MockRetryerMockRecorder) GetHistoryTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryTasks\", reflect.TypeOf((*MockRetryer)(nil).GetHistoryTasks), arg0, arg1)\n}\n\n// GetShardID mocks base method.\nfunc (m *MockRetryer) GetShardID() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardID\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetShardID indicates an expected call of GetShardID.\nfunc (mr *MockRetryerMockRecorder) GetShardID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardID\", reflect.TypeOf((*MockRetryer)(nil).GetShardID))\n}\n\n// GetWorkflowExecution mocks base method.\nfunc (m *MockRetryer) GetWorkflowExecution(arg0 context.Context, arg1 *GetWorkflowExecutionRequest) (*GetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*GetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecution indicates an expected call of GetWorkflowExecution.\nfunc (mr *MockRetryerMockRecorder) GetWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecution\", reflect.TypeOf((*MockRetryer)(nil).GetWorkflowExecution), arg0, arg1)\n}\n\n// IsWorkflowExecutionExists mocks base method.\nfunc (m *MockRetryer) IsWorkflowExecutionExists(arg0 context.Context, arg1 *IsWorkflowExecutionExistsRequest) (*IsWorkflowExecutionExistsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionExists\", arg0, arg1)\n\tret0, _ := ret[0].(*IsWorkflowExecutionExistsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// IsWorkflowExecutionExists indicates an expected call of IsWorkflowExecutionExists.\nfunc (mr *MockRetryerMockRecorder) IsWorkflowExecutionExists(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionExists\", reflect.TypeOf((*MockRetryer)(nil).IsWorkflowExecutionExists), arg0, arg1)\n}\n\n// ListConcreteExecutions mocks base method.\nfunc (m *MockRetryer) ListConcreteExecutions(arg0 context.Context, arg1 *ListConcreteExecutionsRequest) (*ListConcreteExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListConcreteExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*ListConcreteExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListConcreteExecutions indicates an expected call of ListConcreteExecutions.\nfunc (mr *MockRetryerMockRecorder) ListConcreteExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListConcreteExecutions\", reflect.TypeOf((*MockRetryer)(nil).ListConcreteExecutions), arg0, arg1)\n}\n\n// ListCurrentExecutions mocks base method.\nfunc (m *MockRetryer) ListCurrentExecutions(arg0 context.Context, arg1 *ListCurrentExecutionsRequest) (*ListCurrentExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListCurrentExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*ListCurrentExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListCurrentExecutions indicates an expected call of ListCurrentExecutions.\nfunc (mr *MockRetryerMockRecorder) ListCurrentExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListCurrentExecutions\", reflect.TypeOf((*MockRetryer)(nil).ListCurrentExecutions), arg0, arg1)\n}\n\n// ReadHistoryBranch mocks base method.\nfunc (m *MockRetryer) ReadHistoryBranch(arg0 context.Context, arg1 *ReadHistoryBranchRequest) (*ReadHistoryBranchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadHistoryBranch\", arg0, arg1)\n\tret0, _ := ret[0].(*ReadHistoryBranchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadHistoryBranch indicates an expected call of ReadHistoryBranch.\nfunc (mr *MockRetryerMockRecorder) ReadHistoryBranch(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadHistoryBranch\", reflect.TypeOf((*MockRetryer)(nil).ReadHistoryBranch), arg0, arg1)\n}\n"
  },
  {
    "path": "common/persistence/retryer_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestPersistenceRetryerListConcreteExecutions(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *ListConcreteExecutionsRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *ListConcreteExecutionsResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &ListConcreteExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Eq(&ListConcreteExecutionsRequest{})).Return(&ListConcreteExecutionsResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &ListConcreteExecutionsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &ListConcreteExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Eq(&ListConcreteExecutionsRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Eq(&ListConcreteExecutionsRequest{})).Return(&ListConcreteExecutionsResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &ListConcreteExecutionsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &ListConcreteExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Eq(&ListConcreteExecutionsRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.ListConcreteExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerGetWorkflowExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *GetWorkflowExecutionRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *GetWorkflowExecutionResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &GetWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Eq(&GetWorkflowExecutionRequest{})).Return(&GetWorkflowExecutionResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &GetWorkflowExecutionResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &GetWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Eq(&GetWorkflowExecutionRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Eq(&GetWorkflowExecutionRequest{})).Return(&GetWorkflowExecutionResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &GetWorkflowExecutionResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &GetWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Eq(&GetWorkflowExecutionRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.GetWorkflowExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerGetCurrentExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *GetCurrentExecutionRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *GetCurrentExecutionResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &GetCurrentExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Eq(&GetCurrentExecutionRequest{})).Return(&GetCurrentExecutionResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &GetCurrentExecutionResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &GetCurrentExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Eq(&GetCurrentExecutionRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Eq(&GetCurrentExecutionRequest{})).Return(&GetCurrentExecutionResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &GetCurrentExecutionResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &GetCurrentExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Eq(&GetCurrentExecutionRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.GetCurrentExecution(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerListCurrentExecutions(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *ListCurrentExecutionsRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *ListCurrentExecutionsResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &ListCurrentExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Eq(&ListCurrentExecutionsRequest{})).Return(&ListCurrentExecutionsResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &ListCurrentExecutionsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &ListCurrentExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Eq(&ListCurrentExecutionsRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Eq(&ListCurrentExecutionsRequest{})).Return(&ListCurrentExecutionsResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &ListCurrentExecutionsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &ListCurrentExecutionsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Eq(&ListCurrentExecutionsRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.ListCurrentExecutions(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerIsWorkflowExecutionExists(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *IsWorkflowExecutionExistsRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *IsWorkflowExecutionExistsResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &IsWorkflowExecutionExistsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Eq(&IsWorkflowExecutionExistsRequest{})).Return(&IsWorkflowExecutionExistsResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &IsWorkflowExecutionExistsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &IsWorkflowExecutionExistsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Eq(&IsWorkflowExecutionExistsRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Eq(&IsWorkflowExecutionExistsRequest{})).Return(&IsWorkflowExecutionExistsResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &IsWorkflowExecutionExistsResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &IsWorkflowExecutionExistsRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Eq(&IsWorkflowExecutionExistsRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.IsWorkflowExecutionExists(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerReadHistoryBranch(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                      *ReadHistoryBranchRequest\n\t\tmockHistoryManager           *MockHistoryManager\n\t\tmockHistoryManagerAccordance func(mockHistoryManager *MockHistoryManager)\n\t\texpectedResponse             *ReadHistoryBranchResponse\n\t\texpectedError                error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:            &ReadHistoryBranchRequest{},\n\t\t\tmockHistoryManager: NewMockHistoryManager(ctrl),\n\t\t\tmockHistoryManagerAccordance: func(mockHistoryManager *MockHistoryManager) {\n\t\t\t\tmockHistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Eq(&ReadHistoryBranchRequest{})).Return(&ReadHistoryBranchResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &ReadHistoryBranchResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:            &ReadHistoryBranchRequest{},\n\t\t\tmockHistoryManager: NewMockHistoryManager(ctrl),\n\t\t\tmockHistoryManagerAccordance: func(mockHistoryManager *MockHistoryManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockHistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Eq(&ReadHistoryBranchRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockHistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Eq(&ReadHistoryBranchRequest{})).Return(&ReadHistoryBranchResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &ReadHistoryBranchResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:            &ReadHistoryBranchRequest{},\n\t\t\tmockHistoryManager: NewMockHistoryManager(ctrl),\n\t\t\tmockHistoryManagerAccordance: func(mockHistoryManager *MockHistoryManager) {\n\t\t\t\tmockHistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Eq(&ReadHistoryBranchRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockHistoryManager != nil {\n\t\t\t\ttest.mockHistoryManagerAccordance(test.mockHistoryManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(NewMockExecutionManager(ctrl), test.mockHistoryManager, backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.ReadHistoryBranch(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerDeleteWorkflowExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *DeleteWorkflowExecutionRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &DeleteWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteWorkflowExecutionRequest{})).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &DeleteWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteWorkflowExecutionRequest{})).Return(&types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteWorkflowExecutionRequest{})).Return(nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &DeleteWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteWorkflowExecutionRequest{})).Return(&types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\terr := retryer.DeleteWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerDeleteCurrentWorkflowExecution(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *DeleteCurrentWorkflowExecutionRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &DeleteCurrentWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteCurrentWorkflowExecutionRequest{})).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &DeleteCurrentWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteCurrentWorkflowExecutionRequest{})).Return(&types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteCurrentWorkflowExecutionRequest{})).Return(nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &DeleteCurrentWorkflowExecutionRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Eq(&DeleteCurrentWorkflowExecutionRequest{})).Return(&types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\terr := retryer.DeleteCurrentWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerGetShardID(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockExecutionManager := NewMockExecutionManager(ctrl)\n\tmockExecutionManager.EXPECT().GetShardID().Return(42)\n\tretryer := NewPersistenceRetryer(mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\tassert.Equal(t, 42, retryer.GetShardID())\n}\n\nfunc TestPersistenceRetryerGetHistoryTasks(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *GetHistoryTasksRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedResponse               *GetHistoryTasksResponse\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &GetHistoryTasksRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Eq(&GetHistoryTasksRequest{})).Return(&GetHistoryTasksResponse{}, nil)\n\t\t\t},\n\t\t\texpectedResponse: &GetHistoryTasksResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &GetHistoryTasksRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Eq(&GetHistoryTasksRequest{})).Return(nil, &types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Eq(&GetHistoryTasksRequest{})).Return(&GetHistoryTasksResponse{}, nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedResponse: &GetHistoryTasksResponse{},\n\t\t\texpectedError:    nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &GetHistoryTasksRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Eq(&GetHistoryTasksRequest{})).Return(nil, &types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t\texpectedError:    &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\tresp, err := retryer.GetHistoryTasks(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expectedResponse, resp)\n\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestPersistenceRetryerCompleteHistoryTask(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest                        *CompleteHistoryTaskRequest\n\t\tmockExecutionManager           *MockExecutionManager\n\t\tmockExecutionManagerAccordance func(mockExecutionManager *MockExecutionManager)\n\t\texpectedError                  error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest:              &CompleteHistoryTaskRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Eq(&CompleteHistoryTaskRequest{})).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Transient Error\": {\n\t\t\trequest:              &CompleteHistoryTaskRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tmockExecutionManager.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Eq(&CompleteHistoryTaskRequest{})).Return(&types.InternalServiceError{}),\n\t\t\t\t\tmockExecutionManager.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Eq(&CompleteHistoryTaskRequest{})).Return(nil),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Fatal Error\": {\n\t\t\trequest:              &CompleteHistoryTaskRequest{},\n\t\t\tmockExecutionManager: NewMockExecutionManager(ctrl),\n\t\t\tmockExecutionManagerAccordance: func(mockExecutionManager *MockExecutionManager) {\n\t\t\t\tmockExecutionManager.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Eq(&CompleteHistoryTaskRequest{})).Return(&types.AccessDeniedError{}).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.AccessDeniedError{},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockExecutionManager != nil {\n\t\t\t\ttest.mockExecutionManagerAccordance(test.mockExecutionManager)\n\t\t\t}\n\t\t\tretryer := NewPersistenceRetryer(test.mockExecutionManager, NewMockHistoryManager(ctrl), backoff.NewExponentialRetryPolicy(time.Nanosecond))\n\n\t\t\terr := retryer.CompleteHistoryTask(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/serialization/getters.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// GetStolenSinceRenew internal sql blob getter\nfunc (s *ShardInfo) GetStolenSinceRenew() (o int32) {\n\tif s != nil {\n\t\treturn s.StolenSinceRenew\n\t}\n\treturn\n}\n\n// GetUpdatedAt internal sql blob getter\nfunc (s *ShardInfo) GetUpdatedAt() time.Time {\n\tif s != nil {\n\t\treturn s.UpdatedAt\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetReplicationAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetReplicationAckLevel() (o int64) {\n\tif s != nil {\n\t\treturn s.ReplicationAckLevel\n\t}\n\treturn\n}\n\n// GetTransferAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetTransferAckLevel() (o int64) {\n\tif s != nil {\n\t\treturn s.TransferAckLevel\n\t}\n\treturn\n}\n\n// GetTimerAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetTimerAckLevel() time.Time {\n\tif s != nil {\n\t\treturn s.TimerAckLevel\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetDomainNotificationVersion internal sql blob getter\nfunc (s *ShardInfo) GetDomainNotificationVersion() (o int64) {\n\tif s != nil {\n\t\treturn s.DomainNotificationVersion\n\t}\n\treturn\n}\n\n// GetClusterTransferAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetClusterTransferAckLevel() (o map[string]int64) {\n\tif s != nil {\n\t\treturn s.ClusterTransferAckLevel\n\t}\n\treturn\n}\n\n// GetClusterTimerAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetClusterTimerAckLevel() (o map[string]time.Time) {\n\tif s != nil {\n\t\treturn s.ClusterTimerAckLevel\n\t}\n\treturn\n}\n\n// GetOwner internal sql blob getter\nfunc (s *ShardInfo) GetOwner() (o string) {\n\tif s != nil {\n\t\treturn s.Owner\n\t}\n\treturn\n}\n\n// GetClusterReplicationLevel internal sql blob getter\nfunc (s *ShardInfo) GetClusterReplicationLevel() (o map[string]int64) {\n\tif s != nil {\n\t\treturn s.ClusterReplicationLevel\n\t}\n\treturn\n}\n\n// GetPendingFailoverMarkers internal sql blob getter\nfunc (s *ShardInfo) GetPendingFailoverMarkers() (o []byte) {\n\tif s != nil {\n\t\treturn s.PendingFailoverMarkers\n\t}\n\treturn\n}\n\n// GetPendingFailoverMarkersEncoding internal sql blob getter\nfunc (s *ShardInfo) GetPendingFailoverMarkersEncoding() (o string) {\n\tif s != nil {\n\t\treturn s.PendingFailoverMarkersEncoding\n\t}\n\treturn\n}\n\n// GetReplicationDlqAckLevel internal sql blob getter\nfunc (s *ShardInfo) GetReplicationDlqAckLevel() (o map[string]int64) {\n\tif s != nil {\n\t\treturn s.ReplicationDlqAckLevel\n\t}\n\treturn\n}\n\n// GetTransferProcessingQueueStates internal sql blob getter\nfunc (s *ShardInfo) GetTransferProcessingQueueStates() (o []byte) {\n\tif s != nil {\n\t\treturn s.TransferProcessingQueueStates\n\t}\n\treturn\n}\n\n// GetTransferProcessingQueueStatesEncoding internal sql blob getter\nfunc (s *ShardInfo) GetTransferProcessingQueueStatesEncoding() (o string) {\n\tif s != nil {\n\t\treturn s.TransferProcessingQueueStatesEncoding\n\t}\n\treturn\n}\n\n// GetTimerProcessingQueueStates internal sql blob getter\nfunc (s *ShardInfo) GetTimerProcessingQueueStates() (o []byte) {\n\tif s != nil {\n\t\treturn s.TimerProcessingQueueStates\n\t}\n\treturn\n}\n\n// GetTimerProcessingQueueStatesEncoding internal sql blob getter\nfunc (s *ShardInfo) GetTimerProcessingQueueStatesEncoding() (o string) {\n\tif s != nil {\n\t\treturn s.TimerProcessingQueueStatesEncoding\n\t}\n\treturn\n}\n\n// GetQueueStates internal sql blob getter\nfunc (s *ShardInfo) GetQueueStates() (o map[int32]*types.QueueState) {\n\tif s != nil {\n\t\treturn s.QueueStates\n\t}\n\treturn\n}\n\n// GetName internal sql blob getter\nfunc (d *DomainInfo) GetName() (o string) {\n\tif d != nil {\n\t\treturn d.Name\n\t}\n\treturn\n}\n\n// GetDescription internal sql blob getter\nfunc (d *DomainInfo) GetDescription() (o string) {\n\tif d != nil {\n\t\treturn d.Description\n\t}\n\treturn\n}\n\n// GetOwner internal sql blob getter\nfunc (d *DomainInfo) GetOwner() (o string) {\n\tif d != nil {\n\t\treturn d.Owner\n\t}\n\treturn\n}\n\n// GetStatus internal sql blob getter\nfunc (d *DomainInfo) GetStatus() (o int32) {\n\tif d != nil {\n\t\treturn d.Status\n\t}\n\treturn\n}\n\n// GetRetention internal sql blob getter\nfunc (d *DomainInfo) GetRetention() time.Duration {\n\tif d != nil {\n\t\treturn d.Retention\n\t}\n\treturn time.Duration(0)\n}\n\n// GetEmitMetric internal sql blob getter\nfunc (d *DomainInfo) GetEmitMetric() (o bool) {\n\tif d != nil {\n\t\treturn d.EmitMetric\n\t}\n\treturn\n}\n\n// GetArchivalBucket internal sql blob getter\nfunc (d *DomainInfo) GetArchivalBucket() (o string) {\n\tif d != nil {\n\t\treturn d.ArchivalBucket\n\t}\n\treturn\n}\n\n// GetArchivalStatus internal sql blob getter\nfunc (d *DomainInfo) GetArchivalStatus() (o int16) {\n\tif d != nil {\n\t\treturn d.ArchivalStatus\n\t}\n\treturn\n}\n\n// GetConfigVersion internal sql blob getter\nfunc (d *DomainInfo) GetConfigVersion() (o int64) {\n\tif d != nil {\n\t\treturn d.ConfigVersion\n\t}\n\treturn\n}\n\n// GetNotificationVersion internal sql blob getter\nfunc (d *DomainInfo) GetNotificationVersion() (o int64) {\n\tif d != nil {\n\t\treturn d.NotificationVersion\n\t}\n\treturn\n}\n\n// GetFailoverNotificationVersion internal sql blob getter\nfunc (d *DomainInfo) GetFailoverNotificationVersion() (o int64) {\n\tif d != nil {\n\t\treturn d.FailoverNotificationVersion\n\t}\n\treturn\n}\n\n// GetFailoverVersion internal sql blob getter\nfunc (d *DomainInfo) GetFailoverVersion() (o int64) {\n\tif d != nil {\n\t\treturn d.FailoverVersion\n\t}\n\treturn\n}\n\n// GetActiveClusterName internal sql blob getter\nfunc (d *DomainInfo) GetActiveClusterName() (o string) {\n\tif d != nil {\n\t\treturn d.ActiveClusterName\n\t}\n\treturn\n}\n\n// GetClusters internal sql blob getter\nfunc (d *DomainInfo) GetClusters() (o []string) {\n\tif d != nil {\n\t\treturn d.Clusters\n\t}\n\treturn\n}\n\n// GetData internal sql blob getter\nfunc (d *DomainInfo) GetData() (o map[string]string) {\n\tif d != nil {\n\t\treturn d.Data\n\t}\n\treturn\n}\n\n// GetBadBinaries internal sql blob getter\nfunc (d *DomainInfo) GetBadBinaries() (o []byte) {\n\tif d != nil {\n\t\treturn d.BadBinaries\n\t}\n\treturn\n}\n\n// GetBadBinariesEncoding internal sql blob getter\nfunc (d *DomainInfo) GetBadBinariesEncoding() (o string) {\n\tif d != nil {\n\t\treturn d.BadBinariesEncoding\n\t}\n\treturn\n}\n\n// GetHistoryArchivalStatus internal sql blob getter\nfunc (d *DomainInfo) GetHistoryArchivalStatus() (o int16) {\n\tif d != nil {\n\t\treturn d.HistoryArchivalStatus\n\t}\n\treturn\n}\n\n// GetHistoryArchivalURI internal sql blob getter\nfunc (d *DomainInfo) GetHistoryArchivalURI() (o string) {\n\tif d != nil {\n\t\treturn d.HistoryArchivalURI\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalStatus internal sql blob getter\nfunc (d *DomainInfo) GetVisibilityArchivalStatus() (o int16) {\n\tif d != nil {\n\t\treturn d.VisibilityArchivalStatus\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalURI internal sql blob getter\nfunc (d *DomainInfo) GetVisibilityArchivalURI() (o string) {\n\tif d != nil {\n\t\treturn d.VisibilityArchivalURI\n\t}\n\treturn\n}\n\n// GetFailoverEndTimestamp internal sql blob getter\nfunc (d *DomainInfo) GetFailoverEndTimestamp() time.Time {\n\tif d != nil && d.FailoverEndTimestamp != nil {\n\t\treturn *d.FailoverEndTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetPreviousFailoverVersion internal sql blob getter\nfunc (d *DomainInfo) GetPreviousFailoverVersion() (o int64) {\n\tif d != nil {\n\t\treturn d.PreviousFailoverVersion\n\t}\n\treturn\n}\n\n// GetLastUpdatedTimestamp internal sql blob getter\nfunc (d *DomainInfo) GetLastUpdatedTimestamp() time.Time {\n\tif d != nil {\n\t\treturn d.LastUpdatedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetCreatedTimestamp internal sql blob getter\nfunc (h *HistoryTreeInfo) GetCreatedTimestamp() time.Time {\n\tif h != nil {\n\t\treturn h.CreatedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetAncestors internal sql blob getter\nfunc (h *HistoryTreeInfo) GetAncestors() (o []*types.HistoryBranchRange) {\n\tif h != nil {\n\t\treturn h.Ancestors\n\t}\n\treturn\n}\n\n// GetInfo internal sql blob getter\nfunc (h *HistoryTreeInfo) GetInfo() (o string) {\n\tif h != nil {\n\t\treturn h.Info\n\t}\n\treturn\n}\n\n// GetParentDomainID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetParentDomainID() (o []byte) {\n\tif w != nil {\n\t\treturn w.ParentDomainID\n\t}\n\treturn\n}\n\n// GetRetryBackoffCoefficient internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryBackoffCoefficient() (o float64) {\n\tif w != nil {\n\t\treturn w.RetryBackoffCoefficient\n\t}\n\treturn\n}\n\n// GetParentWorkflowID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetParentWorkflowID() (o string) {\n\tif w != nil {\n\t\treturn w.ParentWorkflowID\n\t}\n\treturn\n}\n\n// GetParentRunID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetParentRunID() (o []byte) {\n\tif w != nil {\n\t\treturn w.ParentRunID\n\t}\n\treturn\n}\n\n// GetCompletionEventEncoding internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCompletionEventEncoding() (o string) {\n\tif w != nil {\n\t\treturn w.CompletionEventEncoding\n\t}\n\treturn\n}\n\n// GetTaskList internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetTaskList() (o string) {\n\tif w != nil {\n\t\treturn w.TaskList\n\t}\n\treturn\n}\n\nfunc (w *WorkflowExecutionInfo) GetTaskListKind() (o types.TaskListKind) {\n\tif w != nil {\n\t\treturn w.TaskListKind\n\t}\n\treturn\n}\n\n// GetIsCron internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetIsCron() (o bool) {\n\tif w != nil {\n\t\treturn w.IsCron\n\t}\n\treturn\n}\n\n// GetWorkflowTypeName internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetWorkflowTypeName() (o string) {\n\tif w != nil {\n\t\treturn w.WorkflowTypeName\n\t}\n\treturn\n}\n\n// GetCreateRequestID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCreateRequestID() (o string) {\n\tif w != nil {\n\t\treturn w.CreateRequestID\n\t}\n\treturn\n}\n\n// GetDecisionRequestID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionRequestID() (o string) {\n\tif w != nil {\n\t\treturn w.DecisionRequestID\n\t}\n\treturn\n}\n\n// GetCancelRequestID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCancelRequestID() (o string) {\n\tif w != nil {\n\t\treturn w.CancelRequestID\n\t}\n\treturn\n}\n\n// GetStickyTaskList internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetStickyTaskList() (o string) {\n\tif w != nil {\n\t\treturn w.StickyTaskList\n\t}\n\treturn\n}\n\n// GetCronSchedule internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCronSchedule() (o string) {\n\tif w != nil {\n\t\treturn w.CronSchedule\n\t}\n\treturn\n}\n\n// GetCronOverlapPolicy internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCronOverlapPolicy() (o int32) {\n\tif w != nil {\n\t\treturn int32(w.CronOverlapPolicy)\n\t}\n\treturn\n}\n\n// GetClientLibraryVersion internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetClientLibraryVersion() (o string) {\n\tif w != nil {\n\t\treturn w.ClientLibraryVersion\n\t}\n\treturn\n}\n\n// GetClientFeatureVersion internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetClientFeatureVersion() (o string) {\n\tif w != nil {\n\t\treturn w.ClientFeatureVersion\n\t}\n\treturn\n}\n\n// GetClientImpl internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetClientImpl() (o string) {\n\tif w != nil {\n\t\treturn w.ClientImpl\n\t}\n\treturn\n}\n\n// GetAutoResetPointsEncoding internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetAutoResetPointsEncoding() (o string) {\n\tif w != nil {\n\t\treturn w.AutoResetPointsEncoding\n\t}\n\treturn\n}\n\n// GetVersionHistoriesEncoding internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetVersionHistoriesEncoding() (o string) {\n\tif w != nil {\n\t\treturn w.VersionHistoriesEncoding\n\t}\n\treturn\n}\n\n// GetActiveClusterSelectionPolicyEncoding internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetActiveClusterSelectionPolicyEncoding() (o string) {\n\tif w != nil {\n\t\treturn w.ActiveClusterSelectionPolicyEncoding\n\t}\n\treturn\n}\n\n// GetInitiatedID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetInitiatedID() (o int64) {\n\tif w != nil {\n\t\treturn w.InitiatedID\n\t}\n\treturn\n}\n\n// GetCompletionEventBatchID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCompletionEventBatchID() (o int64) {\n\tif w != nil && w.CompletionEventBatchID != nil {\n\t\treturn *w.CompletionEventBatchID\n\t}\n\treturn\n}\n\n// GetStartVersion internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetStartVersion() (o int64) {\n\tif w != nil {\n\t\treturn w.StartVersion\n\t}\n\treturn\n}\n\n// GetLastWriteEventID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetLastWriteEventID() (o int64) {\n\tif w != nil && w.LastWriteEventID != nil {\n\t\treturn *w.LastWriteEventID\n\t}\n\treturn\n}\n\n// GetLastEventTaskID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetLastEventTaskID() (o int64) {\n\tif w != nil {\n\t\treturn w.LastEventTaskID\n\t}\n\treturn\n}\n\n// GetLastFirstEventID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetLastFirstEventID() (o int64) {\n\tif w != nil {\n\t\treturn w.LastFirstEventID\n\t}\n\treturn\n}\n\n// GetLastProcessedEvent internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetLastProcessedEvent() (o int64) {\n\tif w != nil {\n\t\treturn w.LastProcessedEvent\n\t}\n\treturn\n}\n\n// GetDecisionVersion internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionVersion() (o int64) {\n\tif w != nil {\n\t\treturn w.DecisionVersion\n\t}\n\treturn\n}\n\n// GetDecisionScheduleID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionScheduleID() (o int64) {\n\tif w != nil {\n\t\treturn w.DecisionScheduleID\n\t}\n\treturn\n}\n\n// GetDecisionStartedID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionStartedID() (o int64) {\n\tif w != nil {\n\t\treturn w.DecisionStartedID\n\t}\n\treturn\n}\n\n// GetDecisionAttempt internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionAttempt() (o int64) {\n\tif w != nil {\n\t\treturn w.DecisionAttempt\n\t}\n\treturn\n}\n\n// GetRetryAttempt internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryAttempt() (o int64) {\n\tif w != nil {\n\t\treturn w.RetryAttempt\n\t}\n\treturn\n}\n\n// GetSignalCount internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetSignalCount() (o int64) {\n\tif w != nil {\n\t\treturn w.SignalCount\n\t}\n\treturn\n}\n\n// GetHistorySize internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetHistorySize() (o int64) {\n\tif w != nil {\n\t\treturn w.HistorySize\n\t}\n\treturn\n}\n\n// GetState internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetState() (o int32) {\n\tif w != nil {\n\t\treturn w.State\n\t}\n\treturn\n}\n\n// GetCloseStatus internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCloseStatus() (o int32) {\n\tif w != nil {\n\t\treturn w.CloseStatus\n\t}\n\treturn\n}\n\n// GetRetryMaximumAttempts internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryMaximumAttempts() (o int32) {\n\tif w != nil {\n\t\treturn w.RetryMaximumAttempts\n\t}\n\treturn\n}\n\n// GetEventStoreVersion internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetEventStoreVersion() (o int32) {\n\tif w != nil {\n\t\treturn w.EventStoreVersion\n\t}\n\treturn\n}\n\n// GetWorkflowTimeout internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetWorkflowTimeout() time.Duration {\n\tif w != nil {\n\t\treturn w.WorkflowTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetDecisionTaskTimeout internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionTaskTimeout() time.Duration {\n\tif w != nil {\n\t\treturn w.DecisionTaskTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetDecisionTimeout internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionTimeout() time.Duration {\n\tif w != nil {\n\t\treturn w.DecisionTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetStickyScheduleToStartTimeout internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetStickyScheduleToStartTimeout() time.Duration {\n\tif w != nil {\n\t\treturn w.StickyScheduleToStartTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetRetryInitialInterval internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryInitialInterval() time.Duration {\n\tif w != nil {\n\t\treturn w.RetryInitialInterval\n\t}\n\treturn time.Duration(0)\n}\n\n// GetRetryMaximumInterval internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryMaximumInterval() time.Duration {\n\tif w != nil {\n\t\treturn w.RetryMaximumInterval\n\t}\n\treturn time.Duration(0)\n}\n\n// GetRetryExpiration internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryExpiration() time.Duration {\n\tif w != nil {\n\t\treturn w.RetryExpiration\n\t}\n\treturn time.Duration(0)\n}\n\n// GetStartTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetStartTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.StartTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetLastUpdatedTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetLastUpdatedTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.LastUpdatedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetDecisionStartedTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionStartedTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.DecisionStartedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetDecisionScheduledTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionScheduledTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.DecisionScheduledTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetDecisionOriginalScheduledTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetDecisionOriginalScheduledTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.DecisionOriginalScheduledTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetRetryExpirationTimestamp internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryExpirationTimestamp() time.Time {\n\tif w != nil {\n\t\treturn w.RetryExpirationTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetCompletionEvent internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCompletionEvent() (o []byte) {\n\tif w != nil {\n\t\treturn w.CompletionEvent\n\t}\n\treturn\n}\n\n// GetExecutionContext internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetExecutionContext() (o []byte) {\n\tif w != nil {\n\t\treturn w.ExecutionContext\n\t}\n\treturn\n}\n\n// GetEventBranchToken internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetEventBranchToken() (o []byte) {\n\tif w != nil {\n\t\treturn w.EventBranchToken\n\t}\n\treturn\n}\n\n// GetAutoResetPoints internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetAutoResetPoints() (o []byte) {\n\tif w != nil {\n\t\treturn w.AutoResetPoints\n\t}\n\treturn\n}\n\n// GetVersionHistories internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetVersionHistories() (o []byte) {\n\tif w != nil {\n\t\treturn w.VersionHistories\n\t}\n\treturn\n}\n\n// GetMemo internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetMemo() (o map[string][]byte) {\n\tif w != nil {\n\t\treturn w.Memo\n\t}\n\treturn\n}\n\n// GetSearchAttributes internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetSearchAttributes() (o map[string][]byte) {\n\tif w != nil {\n\t\treturn w.SearchAttributes\n\t}\n\treturn\n}\n\n// GetRetryNonRetryableErrors internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetRetryNonRetryableErrors() (o []string) {\n\tif w != nil {\n\t\treturn w.RetryNonRetryableErrors\n\t}\n\treturn\n}\n\n// GetCancelRequested internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetCancelRequested() (o bool) {\n\tif w != nil {\n\t\treturn w.CancelRequested\n\t}\n\treturn\n}\n\n// GetHasRetryPolicy internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetHasRetryPolicy() (o bool) {\n\tif w != nil {\n\t\treturn w.HasRetryPolicy\n\t}\n\treturn\n}\n\n// GetFirstExecutionRunID internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetFirstExecutionRunID() (o []byte) {\n\tif w != nil {\n\t\treturn w.FirstExecutionRunID\n\t}\n\treturn\n}\n\n// GetPartitionConfig internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetPartitionConfig() (o map[string]string) {\n\tif w != nil {\n\t\treturn w.PartitionConfig\n\t}\n\treturn\n}\n\n// GetCheckSum internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetChecksum() (o []byte) {\n\tif w != nil {\n\t\treturn w.Checksum\n\t}\n\treturn\n}\n\n// GetCheckSumEncoding internal sql blob getter\nfunc (w *WorkflowExecutionInfo) GetChecksumEncoding() (o string) {\n\tif w != nil {\n\t\treturn w.ChecksumEncoding\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (a *ActivityInfo) GetVersion() (o int64) {\n\tif a != nil {\n\t\treturn a.Version\n\t}\n\treturn\n}\n\n// GetScheduledEventBatchID internal sql blob getter\nfunc (a *ActivityInfo) GetScheduledEventBatchID() (o int64) {\n\tif a != nil {\n\t\treturn a.ScheduledEventBatchID\n\t}\n\treturn\n}\n\n// GetStartedID internal sql blob getter\nfunc (a *ActivityInfo) GetStartedID() (o int64) {\n\tif a != nil {\n\t\treturn a.StartedID\n\t}\n\treturn\n}\n\n// GetCancelRequestID internal sql blob getter\nfunc (a *ActivityInfo) GetCancelRequestID() (o int64) {\n\tif a != nil {\n\t\treturn a.CancelRequestID\n\t}\n\treturn\n}\n\n// GetTimerTaskStatus internal sql blob getter\nfunc (a *ActivityInfo) GetTimerTaskStatus() (o int32) {\n\tif a != nil {\n\t\treturn a.TimerTaskStatus\n\t}\n\treturn\n}\n\n// GetScheduledEventEncoding internal sql blob getter\nfunc (a *ActivityInfo) GetScheduledEventEncoding() (o string) {\n\tif a != nil {\n\t\treturn a.ScheduledEventEncoding\n\t}\n\treturn\n}\n\n// GetStartedIdentity internal sql blob getter\nfunc (a *ActivityInfo) GetStartedIdentity() (o string) {\n\tif a != nil {\n\t\treturn a.StartedIdentity\n\t}\n\treturn\n}\n\n// GetRetryLastFailureReason internal sql blob getter\nfunc (a *ActivityInfo) GetRetryLastFailureReason() (o string) {\n\tif a != nil {\n\t\treturn a.RetryLastFailureReason\n\t}\n\treturn\n}\n\n// GetRetryLastWorkerIdentity internal sql blob getter\nfunc (a *ActivityInfo) GetRetryLastWorkerIdentity() (o string) {\n\tif a != nil {\n\t\treturn a.RetryLastWorkerIdentity\n\t}\n\treturn\n}\n\n// GetTaskList internal sql blob getter\nfunc (a *ActivityInfo) GetTaskList() (o string) {\n\tif a != nil {\n\t\treturn a.TaskList\n\t}\n\treturn\n}\n\nfunc (a *ActivityInfo) GetTaskListKind() (o types.TaskListKind) {\n\tif a != nil {\n\t\treturn a.TaskListKind\n\t}\n\treturn\n}\n\n// GetStartedEventEncoding internal sql blob getter\nfunc (a *ActivityInfo) GetStartedEventEncoding() (o string) {\n\tif a != nil {\n\t\treturn a.StartedEventEncoding\n\t}\n\treturn\n}\n\n// GetActivityID internal sql blob getter\nfunc (a *ActivityInfo) GetActivityID() (o string) {\n\tif a != nil {\n\t\treturn a.ActivityID\n\t}\n\treturn\n}\n\n// GetRequestID internal sql blob getter\nfunc (a *ActivityInfo) GetRequestID() (o string) {\n\tif a != nil {\n\t\treturn a.RequestID\n\t}\n\treturn\n}\n\n// GetAttempt internal sql blob getter\nfunc (a *ActivityInfo) GetAttempt() (o int32) {\n\tif a != nil {\n\t\treturn a.Attempt\n\t}\n\treturn\n}\n\n// GetRetryMaximumAttempts internal sql blob getter\nfunc (a *ActivityInfo) GetRetryMaximumAttempts() (o int32) {\n\tif a != nil {\n\t\treturn a.RetryMaximumAttempts\n\t}\n\treturn\n}\n\n// GetScheduledTimestamp internal sql blob getter\nfunc (a *ActivityInfo) GetScheduledTimestamp() time.Time {\n\tif a != nil {\n\t\treturn a.ScheduledTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetStartedTimestamp internal sql blob getter\nfunc (a *ActivityInfo) GetStartedTimestamp() time.Time {\n\tif a != nil {\n\t\treturn a.StartedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetRetryExpirationTimestamp internal sql blob getter\nfunc (a *ActivityInfo) GetRetryExpirationTimestamp() time.Time {\n\tif a != nil {\n\t\treturn a.RetryExpirationTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetScheduleToStartTimeout internal sql blob getter\nfunc (a *ActivityInfo) GetScheduleToStartTimeout() time.Duration {\n\tif a != nil {\n\t\treturn a.ScheduleToStartTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetScheduleToCloseTimeout internal sql blob getter\nfunc (a *ActivityInfo) GetScheduleToCloseTimeout() time.Duration {\n\tif a != nil {\n\t\treturn a.ScheduleToCloseTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetStartToCloseTimeout internal sql blob getter\nfunc (a *ActivityInfo) GetStartToCloseTimeout() time.Duration {\n\tif a != nil {\n\t\treturn a.StartToCloseTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetHeartbeatTimeout internal sql blob getter\nfunc (a *ActivityInfo) GetHeartbeatTimeout() time.Duration {\n\tif a != nil {\n\t\treturn a.HeartbeatTimeout\n\t}\n\treturn time.Duration(0)\n}\n\n// GetRetryInitialInterval internal sql blob getter\nfunc (a *ActivityInfo) GetRetryInitialInterval() time.Duration {\n\tif a != nil {\n\t\treturn a.RetryInitialInterval\n\t}\n\treturn time.Duration(0)\n}\n\n// GetRetryMaximumInterval internal sql blob getter\nfunc (a *ActivityInfo) GetRetryMaximumInterval() time.Duration {\n\tif a != nil {\n\t\treturn a.RetryMaximumInterval\n\t}\n\treturn time.Duration(0)\n}\n\n// GetScheduledEvent internal sql blob getter\nfunc (a *ActivityInfo) GetScheduledEvent() (o []byte) {\n\tif a != nil {\n\t\treturn a.ScheduledEvent\n\t}\n\treturn\n}\n\n// GetStartedEvent internal sql blob getter\nfunc (a *ActivityInfo) GetStartedEvent() (o []byte) {\n\tif a != nil {\n\t\treturn a.StartedEvent\n\t}\n\treturn\n}\n\n// GetRetryLastFailureDetails internal sql blob getter\nfunc (a *ActivityInfo) GetRetryLastFailureDetails() (o []byte) {\n\tif a != nil {\n\t\treturn a.RetryLastFailureDetails\n\t}\n\treturn\n}\n\n// GetCancelRequested internal sql blob getter\nfunc (a *ActivityInfo) GetCancelRequested() (o bool) {\n\tif a != nil {\n\t\treturn a.CancelRequested\n\t}\n\treturn\n}\n\n// GetHasRetryPolicy internal sql blob getter\nfunc (a *ActivityInfo) GetHasRetryPolicy() (o bool) {\n\tif a != nil {\n\t\treturn a.HasRetryPolicy\n\t}\n\treturn\n}\n\n// GetRetryBackoffCoefficient internal sql blob getter\nfunc (a *ActivityInfo) GetRetryBackoffCoefficient() (o float64) {\n\tif a != nil {\n\t\treturn a.RetryBackoffCoefficient\n\t}\n\treturn\n}\n\n// GetRetryNonRetryableErrors internal sql blob getter\nfunc (a *ActivityInfo) GetRetryNonRetryableErrors() (o []string) {\n\tif a != nil {\n\t\treturn a.RetryNonRetryableErrors\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (c *ChildExecutionInfo) GetVersion() (o int64) {\n\tif c != nil {\n\t\treturn c.Version\n\t}\n\treturn\n}\n\n// GetInitiatedEventBatchID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetInitiatedEventBatchID() (o int64) {\n\tif c != nil {\n\t\treturn c.InitiatedEventBatchID\n\t}\n\treturn\n}\n\n// GetStartedID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetStartedID() (o int64) {\n\tif c != nil {\n\t\treturn c.StartedID\n\t}\n\treturn\n}\n\n// GetParentClosePolicy internal sql blob getter\nfunc (c *ChildExecutionInfo) GetParentClosePolicy() (o int32) {\n\tif c != nil {\n\t\treturn c.ParentClosePolicy\n\t}\n\treturn\n}\n\n// GetInitiatedEventEncoding internal sql blob getter\nfunc (c *ChildExecutionInfo) GetInitiatedEventEncoding() (o string) {\n\tif c != nil {\n\t\treturn c.InitiatedEventEncoding\n\t}\n\treturn\n}\n\n// GetStartedWorkflowID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetStartedWorkflowID() (o string) {\n\tif c != nil {\n\t\treturn c.StartedWorkflowID\n\t}\n\treturn\n}\n\n// GetStartedRunID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetStartedRunID() (o []byte) {\n\tif c != nil {\n\t\treturn c.StartedRunID\n\t}\n\treturn\n}\n\n// GetStartedEventEncoding internal sql blob getter\nfunc (c *ChildExecutionInfo) GetStartedEventEncoding() (o string) {\n\tif c != nil {\n\t\treturn c.StartedEventEncoding\n\t}\n\treturn\n}\n\n// GetCreateRequestID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetCreateRequestID() (o string) {\n\tif c != nil {\n\t\treturn c.CreateRequestID\n\t}\n\treturn\n}\n\n// GetDomainID internal sql blob getter\nfunc (c *ChildExecutionInfo) GetDomainID() (o string) {\n\tif c != nil {\n\t\treturn c.DomainID\n\t}\n\treturn\n}\n\n// GetDomainNameDEPRECATED internal sql blob getter\nfunc (c *ChildExecutionInfo) GetDomainNameDEPRECATED() (o string) {\n\tif c != nil {\n\t\treturn c.DomainNameDEPRECATED\n\t}\n\treturn\n}\n\n// GetWorkflowTypeName internal sql blob getter\nfunc (c *ChildExecutionInfo) GetWorkflowTypeName() (o string) {\n\tif c != nil {\n\t\treturn c.WorkflowTypeName\n\t}\n\treturn\n}\n\n// GetInitiatedEvent internal sql blob getter\nfunc (c *ChildExecutionInfo) GetInitiatedEvent() (o []byte) {\n\tif c != nil {\n\t\treturn c.InitiatedEvent\n\t}\n\treturn\n}\n\n// GetStartedEvent internal sql blob getter\nfunc (c *ChildExecutionInfo) GetStartedEvent() (o []byte) {\n\tif c != nil {\n\t\treturn c.StartedEvent\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (s *SignalInfo) GetVersion() (o int64) {\n\tif s != nil {\n\t\treturn s.Version\n\t}\n\treturn\n}\n\n// GetInitiatedEventBatchID internal sql blob getter\nfunc (s *SignalInfo) GetInitiatedEventBatchID() (o int64) {\n\tif s != nil {\n\t\treturn s.InitiatedEventBatchID\n\t}\n\treturn\n}\n\n// GetRequestID internal sql blob getter\nfunc (s *SignalInfo) GetRequestID() (o string) {\n\tif s != nil {\n\t\treturn s.RequestID\n\t}\n\treturn\n}\n\n// GetName internal sql blob getter\nfunc (s *SignalInfo) GetName() (o string) {\n\tif s != nil {\n\t\treturn s.Name\n\t}\n\treturn\n}\n\n// GetInput internal sql blob getter\nfunc (s *SignalInfo) GetInput() (o []byte) {\n\tif s != nil {\n\t\treturn s.Input\n\t}\n\treturn\n}\n\n// GetControl internal sql blob getter\nfunc (s *SignalInfo) GetControl() (o []byte) {\n\tif s != nil {\n\t\treturn s.Control\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (r *RequestCancelInfo) GetVersion() (o int64) {\n\tif r != nil {\n\t\treturn r.Version\n\t}\n\treturn\n}\n\n// GetInitiatedEventBatchID internal sql blob getter\nfunc (r *RequestCancelInfo) GetInitiatedEventBatchID() (o int64) {\n\tif r != nil {\n\t\treturn r.InitiatedEventBatchID\n\t}\n\treturn\n}\n\n// GetCancelRequestID internal sql blob getter\nfunc (r *RequestCancelInfo) GetCancelRequestID() (o string) {\n\tif r != nil {\n\t\treturn r.CancelRequestID\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (t *TimerInfo) GetVersion() (o int64) {\n\tif t != nil {\n\t\treturn t.Version\n\t}\n\treturn\n}\n\n// GetStartedID internal sql blob getter\nfunc (t *TimerInfo) GetStartedID() (o int64) {\n\tif t != nil {\n\t\treturn t.StartedID\n\t}\n\treturn\n}\n\n// GetTaskID internal sql blob getter\nfunc (t *TimerInfo) GetTaskID() (o int64) {\n\tif t != nil {\n\t\treturn t.TaskID\n\t}\n\treturn\n}\n\n// GetExpiryTimestamp internal sql blob getter\nfunc (t *TimerInfo) GetExpiryTimestamp() (o time.Time) {\n\tif t != nil {\n\t\treturn t.ExpiryTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetWorkflowID internal sql blob getter\nfunc (t *TaskInfo) GetWorkflowID() (o string) {\n\tif t != nil {\n\t\treturn t.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID internal sql blob getter\nfunc (t *TaskInfo) GetRunID() (o []byte) {\n\tif t != nil {\n\t\treturn t.RunID\n\t}\n\treturn\n}\n\n// GetScheduleID internal sql blob getter\nfunc (t *TaskInfo) GetScheduleID() (o int64) {\n\tif t != nil {\n\t\treturn t.ScheduleID\n\t}\n\treturn\n}\n\n// GetExpiryTimestamp internal sql blob getter\nfunc (t *TaskInfo) GetExpiryTimestamp() time.Time {\n\tif t != nil {\n\t\treturn t.ExpiryTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetCreatedTimestamp internal sql blob getter\nfunc (t *TaskInfo) GetCreatedTimestamp() time.Time {\n\tif t != nil {\n\t\treturn t.CreatedTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetPartitionConfig internal sql blob getter\nfunc (t *TaskInfo) GetPartitionConfig() (o map[string]string) {\n\tif t != nil {\n\t\treturn t.PartitionConfig\n\t}\n\treturn\n}\n\n// GetKind internal sql blob getter\nfunc (t *TaskListInfo) GetKind() (o int16) {\n\tif t != nil {\n\t\treturn t.Kind\n\t}\n\treturn\n}\n\n// GetAckLevel internal sql blob getter\nfunc (t *TaskListInfo) GetAckLevel() (o int64) {\n\tif t != nil {\n\t\treturn t.AckLevel\n\t}\n\treturn\n}\n\n// GetExpiryTimestamp internal sql blob getter\nfunc (t *TaskListInfo) GetExpiryTimestamp() time.Time {\n\tif t != nil {\n\t\treturn t.ExpiryTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetLastUpdated internal sql blob getter\nfunc (t *TaskListInfo) GetLastUpdated() time.Time {\n\tif t != nil {\n\t\treturn t.LastUpdated\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetDomainID internal sql blob getter\nfunc (t *TransferTaskInfo) GetDomainID() (o []byte) {\n\tif t != nil {\n\t\treturn t.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID internal sql blob getter\nfunc (t *TransferTaskInfo) GetWorkflowID() (o string) {\n\tif t != nil {\n\t\treturn t.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID internal sql blob getter\nfunc (t *TransferTaskInfo) GetRunID() (o []byte) {\n\tif t != nil {\n\t\treturn t.RunID\n\t}\n\treturn\n}\n\n// GetTaskType internal sql blob getter\nfunc (t *TransferTaskInfo) GetTaskType() (o int16) {\n\tif t != nil {\n\t\treturn t.TaskType\n\t}\n\treturn\n}\n\n// GetTargetDomainID internal sql blob getter\nfunc (t *TransferTaskInfo) GetTargetDomainID() (o []byte) {\n\tif t != nil {\n\t\treturn t.TargetDomainID\n\t}\n\treturn\n}\n\n// GetTargetDomainIDs internal sql blob getter\nfunc (t *TransferTaskInfo) GetTargetDomainIDs() (o map[string]struct{}) {\n\tif t != nil {\n\t\ttargetDomainIDs := make(map[string]struct{})\n\t\tfor _, domainID := range t.TargetDomainIDs {\n\t\t\ttargetDomainIDs[domainID.String()] = struct{}{}\n\t\t}\n\t\treturn targetDomainIDs\n\t}\n\treturn\n}\n\n// GetTargetWorkflowID internal sql blob getter\nfunc (t *TransferTaskInfo) GetTargetWorkflowID() (o string) {\n\tif t != nil {\n\t\treturn t.TargetWorkflowID\n\t}\n\treturn\n}\n\n// GetTargetRunID internal sql blob getter\nfunc (t *TransferTaskInfo) GetTargetRunID() (o []byte) {\n\tif t != nil {\n\t\treturn t.TargetRunID\n\t}\n\treturn\n}\n\n// GetTaskList internal sql blob getter\nfunc (t *TransferTaskInfo) GetTaskList() (o string) {\n\tif t != nil {\n\t\treturn t.TaskList\n\t}\n\treturn\n}\n\n// GetTargetChildWorkflowOnly internal sql blob getter\nfunc (t *TransferTaskInfo) GetTargetChildWorkflowOnly() (o bool) {\n\tif t != nil {\n\t\treturn t.TargetChildWorkflowOnly\n\t}\n\treturn\n}\n\n// GetScheduleID internal sql blob getter\nfunc (t *TransferTaskInfo) GetScheduleID() (o int64) {\n\tif t != nil {\n\t\treturn t.ScheduleID\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (t *TransferTaskInfo) GetVersion() (o int64) {\n\tif t != nil {\n\t\treturn t.Version\n\t}\n\treturn\n}\n\n// GetVisibilityTimestamp internal sql blob getter\nfunc (t *TransferTaskInfo) GetVisibilityTimestamp() time.Time {\n\tif t != nil {\n\t\treturn t.VisibilityTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n\n// GetOriginalTaskList internal sql blob getter\nfunc (t *TransferTaskInfo) GetOriginalTaskList() (o string) {\n\tif t != nil {\n\t\treturn t.OriginalTaskList\n\t}\n\treturn\n}\n\n// GetOriginalTaskListKind internal sql blob getter\nfunc (t *TransferTaskInfo) GetOriginalTaskListKind() (o types.TaskListKind) {\n\tif t != nil {\n\t\treturn t.OriginalTaskListKind\n\t}\n\treturn\n}\n\n// GetDomainID internal sql blob getter\nfunc (t *TimerTaskInfo) GetDomainID() (o []byte) {\n\tif t != nil && t.DomainID != nil {\n\t\treturn t.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID internal sql blob getter\nfunc (t *TimerTaskInfo) GetWorkflowID() (o string) {\n\tif t != nil {\n\t\treturn t.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID internal sql blob getter\nfunc (t *TimerTaskInfo) GetRunID() (o []byte) {\n\tif t != nil {\n\t\treturn t.RunID\n\t}\n\treturn\n}\n\n// GetTaskType internal sql blob getter\nfunc (t *TimerTaskInfo) GetTaskType() (o int16) {\n\tif t != nil {\n\t\treturn t.TaskType\n\t}\n\treturn\n}\n\n// GetTimeoutType internal sql blob getter\nfunc (t *TimerTaskInfo) GetTimeoutType() (o int16) {\n\tif t != nil && t.TimeoutType != nil {\n\t\treturn *t.TimeoutType\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (t *TimerTaskInfo) GetVersion() (o int64) {\n\tif t != nil {\n\t\treturn t.Version\n\t}\n\treturn\n}\n\n// GetScheduleAttempt internal sql blob getter\nfunc (t *TimerTaskInfo) GetScheduleAttempt() (o int64) {\n\tif t != nil {\n\t\treturn t.ScheduleAttempt\n\t}\n\treturn\n}\n\n// GetEventID internal sql blob getter\nfunc (t *TimerTaskInfo) GetEventID() (o int64) {\n\tif t != nil {\n\t\treturn t.EventID\n\t}\n\treturn\n}\n\n// GetTaskList internal sql blob getter\nfunc (t *TimerTaskInfo) GetTaskList() (o string) {\n\tif t != nil {\n\t\treturn t.TaskList\n\t}\n\treturn\n}\n\n// GetDomainID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetDomainID() (o []byte) {\n\tif t != nil {\n\t\treturn t.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetWorkflowID() (o string) {\n\tif t != nil {\n\t\treturn t.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetRunID() (o []byte) {\n\tif t != nil {\n\t\treturn t.RunID\n\t}\n\treturn\n}\n\n// GetTaskType internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetTaskType() (o int16) {\n\tif t != nil {\n\t\treturn t.TaskType\n\t}\n\treturn\n}\n\n// GetVersion internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetVersion() (o int64) {\n\tif t != nil {\n\t\treturn t.Version\n\t}\n\treturn\n}\n\n// GetFirstEventID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetFirstEventID() (o int64) {\n\tif t != nil {\n\t\treturn t.FirstEventID\n\t}\n\treturn\n}\n\n// GetNextEventID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetNextEventID() (o int64) {\n\tif t != nil {\n\t\treturn t.NextEventID\n\t}\n\treturn\n}\n\n// GetScheduledID internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetScheduledID() (o int64) {\n\tif t != nil {\n\t\treturn t.ScheduledID\n\t}\n\treturn\n}\n\n// GetEventStoreVersion internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetEventStoreVersion() (o int32) {\n\tif t != nil {\n\t\treturn t.EventStoreVersion\n\t}\n\treturn\n}\n\n// GetNewRunEventStoreVersion internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetNewRunEventStoreVersion() (o int32) {\n\tif t != nil {\n\t\treturn t.NewRunEventStoreVersion\n\t}\n\treturn\n}\n\n// GetBranchToken internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetBranchToken() (o []byte) {\n\tif t != nil {\n\t\treturn t.BranchToken\n\t}\n\treturn\n}\n\n// GetNewRunBranchToken internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetNewRunBranchToken() (o []byte) {\n\tif t != nil {\n\t\treturn t.NewRunBranchToken\n\t}\n\treturn\n}\n\n// GetCreationTimestamp internal sql blob getter\nfunc (t *ReplicationTaskInfo) GetCreationTimestamp() time.Time {\n\tif t != nil {\n\t\treturn t.CreationTimestamp\n\t}\n\treturn time.Unix(0, 0)\n}\n"
  },
  {
    "path": "common/persistence/serialization/getters_fixtures_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tzeroUnix = time.Unix(0, 0)\n)\n\nvar expectedNil = map[string]map[string]any{\n\t\"*serialization.WorkflowExecutionInfo\": {\n\t\t\"GetActiveClusterSelectionPolicyEncoding\": string(\"\"),\n\t\t\"GetAutoResetPoints\":                      []uint8(nil),\n\t\t\"GetAutoResetPointsEncoding\":              \"\",\n\t\t\"GetCancelRequestID\":                      \"\",\n\t\t\"GetCancelRequested\":                      false,\n\t\t\"GetClientFeatureVersion\":                 \"\",\n\t\t\"GetClientImpl\":                           \"\",\n\t\t\"GetClientLibraryVersion\":                 \"\",\n\t\t\"GetCloseStatus\":                          int32(0),\n\t\t\"GetCompletionEvent\":                      []uint8(nil),\n\t\t\"GetCompletionEventEncoding\":              \"\",\n\t\t\"GetCreateRequestID\":                      \"\",\n\t\t\"GetCompletionEventBatchID\":               int64(0),\n\t\t\"GetCronSchedule\":                         \"\",\n\t\t\"GetCronOverlapPolicy\":                    int32(0),\n\t\t\"GetDecisionAttempt\":                      int64(0),\n\t\t\"GetDecisionOriginalScheduledTimestamp\":   zeroUnix,\n\t\t\"GetDecisionRequestID\":                    \"\",\n\t\t\"GetDecisionScheduleID\":                   int64(0),\n\t\t\"GetDecisionScheduledTimestamp\":           zeroUnix,\n\t\t\"GetDecisionStartedID\":                    int64(0),\n\t\t\"GetDecisionStartedTimestamp\":             zeroUnix,\n\t\t\"GetDecisionTaskTimeout\":                  time.Duration(0),\n\t\t\"GetDecisionTimeout\":                      time.Duration(0),\n\t\t\"GetLastFirstEventID\":                     int64(0),\n\t\t\"GetLastProcessedEvent\":                   int64(0),\n\t\t\"GetParentWorkflowID\":                     \"\",\n\t\t\"GetStartVersion\":                         int64(0),\n\t\t\"GetTaskList\":                             \"\",\n\t\t\"GetTaskListKind\":                         types.TaskListKindNormal,\n\t\t\"GetDecisionVersion\":                      int64(0),\n\t\t\"GetEventBranchToken\":                     []uint8(nil),\n\t\t\"GetEventStoreVersion\":                    int32(0),\n\t\t\"GetExecutionContext\":                     []uint8(nil),\n\t\t\"GetFirstExecutionRunID\":                  []uint8(nil),\n\t\t\"GetHasRetryPolicy\":                       false,\n\t\t\"GetInitiatedID\":                          int64(0),\n\t\t\"GetIsCron\":                               false,\n\t\t\"GetLastEventTaskID\":                      int64(0),\n\t\t\"GetLastUpdatedTimestamp\":                 zeroUnix,\n\t\t\"GetLastWriteEventID\":                     int64(0),\n\t\t\"GetMemo\":                                 map[string][]uint8(nil),\n\t\t\"GetParentDomainID\":                       []uint8(nil),\n\t\t\"GetParentRunID\":                          []uint8(nil),\n\t\t\"GetPartitionConfig\":                      map[string]string(nil),\n\t\t\"GetHistorySize\":                          int64(0),\n\t\t\"GetRetryAttempt\":                         int64(0),\n\t\t\"GetRetryBackoffCoefficient\":              float64(0),\n\t\t\"GetRetryExpiration\":                      time.Duration(0),\n\t\t\"GetRetryExpirationTimestamp\":             zeroUnix,\n\t\t\"GetRetryInitialInterval\":                 time.Duration(0),\n\t\t\"GetRetryMaximumAttempts\":                 int32(0),\n\t\t\"GetRetryMaximumInterval\":                 time.Duration(0),\n\t\t\"GetRetryNonRetryableErrors\":              []string(nil),\n\t\t\"GetSearchAttributes\":                     map[string][]uint8(nil),\n\t\t\"GetSignalCount\":                          int64(0),\n\t\t\"GetStartTimestamp\":                       zeroUnix,\n\t\t\"GetState\":                                int32(0),\n\t\t\"GetStickyScheduleToStartTimeout\":         time.Duration(0),\n\t\t\"GetStickyTaskList\":                       \"\",\n\t\t\"GetVersionHistories\":                     []uint8(nil),\n\t\t\"GetVersionHistoriesEncoding\":             \"\",\n\t\t\"GetWorkflowTimeout\":                      time.Duration(0),\n\t\t\"GetWorkflowTypeName\":                     \"\",\n\t\t\"GetChecksum\":                             []uint8(nil),\n\t\t\"GetChecksumEncoding\":                     \"\",\n\t},\n\t\"*serialization.TransferTaskInfo\": {\n\t\t\"GetDomainID\":                []uint8(nil),\n\t\t\"GetWorkflowID\":              \"\",\n\t\t\"GetRunID\":                   []uint8(nil),\n\t\t\"GetTaskType\":                int16(0),\n\t\t\"GetTargetDomainID\":          []uint8(nil),\n\t\t\"GetTargetDomainIDs\":         map[string]struct{}(nil),\n\t\t\"GetTargetWorkflowID\":        \"\",\n\t\t\"GetTargetRunID\":             []uint8(nil),\n\t\t\"GetTaskList\":                \"\",\n\t\t\"GetTargetChildWorkflowOnly\": false,\n\t\t\"GetScheduleID\":              int64(0),\n\t\t\"GetVersion\":                 int64(0),\n\t\t\"GetVisibilityTimestamp\":     zeroUnix,\n\t\t\"GetOriginalTaskList\":        \"\",\n\t\t\"GetOriginalTaskListKind\":    types.TaskListKindNormal,\n\t},\n\t\"*serialization.TimerTaskInfo\": {\n\t\t\"GetDomainID\":        []uint8(nil),\n\t\t\"GetEventID\":         int64(0),\n\t\t\"GetRunID\":           []uint8(nil),\n\t\t\"GetScheduleAttempt\": int64(0),\n\t\t\"GetTaskType\":        int16(0),\n\t\t\"GetTimeoutType\":     int16(0),\n\t\t\"GetVersion\":         int64(0),\n\t\t\"GetWorkflowID\":      \"\",\n\t\t\"GetTaskList\":        \"\",\n\t},\n\t\"*serialization.ReplicationTaskInfo\": {\n\t\t\"GetBranchToken\":             []uint8(nil),\n\t\t\"GetCreationTimestamp\":       zeroUnix,\n\t\t\"GetDomainID\":                []uint8(nil),\n\t\t\"GetEventStoreVersion\":       int32(0),\n\t\t\"GetFirstEventID\":            int64(0),\n\t\t\"GetNewRunBranchToken\":       []uint8(nil),\n\t\t\"GetNewRunEventStoreVersion\": int32(0),\n\t\t\"GetNextEventID\":             int64(0),\n\t\t\"GetRunID\":                   []uint8(nil),\n\t\t\"GetScheduledID\":             int64(0),\n\t\t\"GetTaskType\":                int16(0),\n\t\t\"GetVersion\":                 int64(0),\n\t\t\"GetWorkflowID\":              \"\",\n\t},\n\t\"*serialization.TaskListInfo\": {\n\t\t\"GetAckLevel\":        int64(0),\n\t\t\"GetExpiryTimestamp\": zeroUnix,\n\t\t\"GetKind\":            int16(0),\n\t\t\"GetLastUpdated\":     zeroUnix,\n\t},\n\t\"*serialization.TaskInfo\": {\n\t\t\"GetCreatedTimestamp\": zeroUnix,\n\t\t\"GetExpiryTimestamp\":  zeroUnix,\n\t\t\"GetPartitionConfig\":  map[string]string(nil),\n\t\t\"GetRunID\":            []uint8(nil),\n\t\t\"GetScheduleID\":       int64(0),\n\t\t\"GetWorkflowID\":       \"\",\n\t},\n\t\"*serialization.TimerInfo\": {\n\t\t\"GetExpiryTimestamp\": zeroUnix,\n\t\t\"GetStartedID\":       int64(0),\n\t\t\"GetTaskID\":          int64(0),\n\t\t\"GetVersion\":         int64(0),\n\t},\n\t\"*serialization.RequestCancelInfo\": {\n\t\t\"GetCancelRequestID\":       \"\",\n\t\t\"GetInitiatedEventBatchID\": int64(0),\n\t\t\"GetVersion\":               int64(0),\n\t},\n\t\"*serialization.SignalInfo\": {\n\t\t\"GetControl\":               []uint8(nil),\n\t\t\"GetInitiatedEventBatchID\": int64(0),\n\t\t\"GetInput\":                 []uint8(nil),\n\t\t\"GetName\":                  \"\",\n\t\t\"GetRequestID\":             \"\",\n\t\t\"GetVersion\":               int64(0),\n\t},\n\t\"*serialization.ChildExecutionInfo\": {\n\t\t\"GetCreateRequestID\":        \"\",\n\t\t\"GetDomainID\":               \"\",\n\t\t\"GetDomainNameDEPRECATED\":   \"\",\n\t\t\"GetInitiatedEvent\":         []uint8(nil),\n\t\t\"GetInitiatedEventBatchID\":  int64(0),\n\t\t\"GetInitiatedEventEncoding\": \"\",\n\t\t\"GetParentClosePolicy\":      int32(0),\n\t\t\"GetStartedEvent\":           []uint8(nil),\n\t\t\"GetStartedEventEncoding\":   \"\",\n\t\t\"GetStartedID\":              int64(0),\n\t\t\"GetStartedRunID\":           []uint8(nil),\n\t\t\"GetStartedWorkflowID\":      \"\",\n\t\t\"GetVersion\":                int64(0),\n\t\t\"GetWorkflowTypeName\":       \"\",\n\t},\n\t\"*serialization.ActivityInfo\": {\n\t\t\"GetActivityID\":               \"\",\n\t\t\"GetAttempt\":                  int32(0),\n\t\t\"GetCancelRequestID\":          int64(0),\n\t\t\"GetCancelRequested\":          false,\n\t\t\"GetHasRetryPolicy\":           false,\n\t\t\"GetHeartbeatTimeout\":         time.Duration(0),\n\t\t\"GetRequestID\":                \"\",\n\t\t\"GetRetryBackoffCoefficient\":  float64(0),\n\t\t\"GetRetryExpirationTimestamp\": zeroUnix,\n\t\t\"GetRetryInitialInterval\":     time.Duration(0),\n\t\t\"GetRetryLastFailureDetails\":  []uint8(nil),\n\t\t\"GetRetryLastFailureReason\":   \"\",\n\t\t\"GetRetryLastWorkerIdentity\":  \"\",\n\t\t\"GetRetryMaximumAttempts\":     int32(0),\n\t\t\"GetRetryMaximumInterval\":     time.Duration(0),\n\t\t\"GetRetryNonRetryableErrors\":  []string(nil),\n\t\t\"GetScheduleToCloseTimeout\":   time.Duration(0),\n\t\t\"GetScheduleToStartTimeout\":   time.Duration(0),\n\t\t\"GetScheduledEvent\":           []uint8(nil),\n\t\t\"GetScheduledEventBatchID\":    int64(0),\n\t\t\"GetScheduledEventEncoding\":   \"\",\n\t\t\"GetScheduledTimestamp\":       zeroUnix,\n\t\t\"GetStartToCloseTimeout\":      time.Duration(0),\n\t\t\"GetStartedEvent\":             []uint8(nil),\n\t\t\"GetStartedEventEncoding\":     \"\",\n\t\t\"GetStartedID\":                int64(0),\n\t\t\"GetStartedIdentity\":          \"\",\n\t\t\"GetStartedTimestamp\":         zeroUnix,\n\t\t\"GetTaskList\":                 \"\",\n\t\t\"GetTaskListKind\":             types.TaskListKindNormal,\n\t\t\"GetTimerTaskStatus\":          int32(0),\n\t\t\"GetVersion\":                  int64(0),\n\t},\n\t\"*serialization.HistoryTreeInfo\": {\n\t\t\"GetAncestors\":        []*types.HistoryBranchRange(nil),\n\t\t\"GetCreatedTimestamp\": zeroUnix,\n\t\t\"GetInfo\":             \"\",\n\t},\n\t\"*serialization.DomainInfo\": {\n\t\t\"GetActiveClusterName\":           \"\",\n\t\t\"GetArchivalBucket\":              \"\",\n\t\t\"GetArchivalStatus\":              int16(0),\n\t\t\"GetBadBinaries\":                 []uint8(nil),\n\t\t\"GetBadBinariesEncoding\":         \"\",\n\t\t\"GetClusters\":                    []string(nil),\n\t\t\"GetConfigVersion\":               int64(0),\n\t\t\"GetData\":                        map[string]string(nil),\n\t\t\"GetDescription\":                 \"\",\n\t\t\"GetEmitMetric\":                  false,\n\t\t\"GetFailoverEndTimestamp\":        zeroUnix,\n\t\t\"GetFailoverNotificationVersion\": int64(0),\n\t\t\"GetFailoverVersion\":             int64(0),\n\t\t\"GetHistoryArchivalStatus\":       int16(0),\n\t\t\"GetHistoryArchivalURI\":          \"\",\n\t\t\"GetLastUpdatedTimestamp\":        zeroUnix,\n\t\t\"GetName\":                        \"\",\n\t\t\"GetNotificationVersion\":         int64(0),\n\t\t\"GetOwner\":                       \"\",\n\t\t\"GetPreviousFailoverVersion\":     int64(0),\n\t\t\"GetRetention\":                   time.Duration(0),\n\t\t\"GetStatus\":                      int32(0),\n\t\t\"GetVisibilityArchivalStatus\":    int16(0),\n\t\t\"GetVisibilityArchivalURI\":       \"\",\n\t},\n\t\"*serialization.ShardInfo\": {\n\t\t\"GetClusterReplicationLevel\":               map[string]int64(nil),\n\t\t\"GetClusterTimerAckLevel\":                  map[string]time.Time(nil),\n\t\t\"GetClusterTransferAckLevel\":               map[string]int64(nil),\n\t\t\"GetDomainNotificationVersion\":             int64(0),\n\t\t\"GetOwner\":                                 \"\",\n\t\t\"GetPendingFailoverMarkers\":                []uint8(nil),\n\t\t\"GetPendingFailoverMarkersEncoding\":        \"\",\n\t\t\"GetReplicationAckLevel\":                   int64(0),\n\t\t\"GetReplicationDlqAckLevel\":                map[string]int64(nil),\n\t\t\"GetStolenSinceRenew\":                      int32(0),\n\t\t\"GetTimerAckLevel\":                         zeroUnix,\n\t\t\"GetTimerProcessingQueueStates\":            []uint8(nil),\n\t\t\"GetTimerProcessingQueueStatesEncoding\":    \"\",\n\t\t\"GetTransferAckLevel\":                      int64(0),\n\t\t\"GetTransferProcessingQueueStates\":         []uint8(nil),\n\t\t\"GetTransferProcessingQueueStatesEncoding\": \"\",\n\t\t\"GetUpdatedAt\":                             zeroUnix,\n\t\t\"GetQueueStates\":                           map[int32]*types.QueueState(nil),\n\t},\n}\n\nvar expectedEmpty = map[string]map[string]any{\n\t\"*serialization.WorkflowExecutionInfo\": {\n\t\t\"GetActiveClusterSelectionPolicyEncoding\": string(\"\"),\n\t\t\"GetAutoResetPoints\":                      []uint8(nil),\n\t\t\"GetAutoResetPointsEncoding\":              \"\",\n\t\t\"GetCancelRequestID\":                      \"\",\n\t\t\"GetCancelRequested\":                      false,\n\t\t\"GetClientFeatureVersion\":                 \"\",\n\t\t\"GetClientImpl\":                           \"\",\n\t\t\"GetClientLibraryVersion\":                 \"\",\n\t\t\"GetCloseStatus\":                          int32(0),\n\t\t\"GetCompletionEvent\":                      []uint8(nil),\n\t\t\"GetCompletionEventEncoding\":              \"\",\n\t\t\"GetCreateRequestID\":                      \"\",\n\t\t\"GetCompletionEventBatchID\":               int64(0),\n\t\t\"GetCronSchedule\":                         \"\",\n\t\t\"GetCronOverlapPolicy\":                    int32(0),\n\t\t\"GetDecisionAttempt\":                      int64(0),\n\t\t\"GetDecisionOriginalScheduledTimestamp\":   time.Time{},\n\t\t\"GetDecisionRequestID\":                    \"\",\n\t\t\"GetDecisionScheduleID\":                   int64(0),\n\t\t\"GetDecisionScheduledTimestamp\":           time.Time{},\n\t\t\"GetDecisionStartedID\":                    int64(0),\n\t\t\"GetDecisionStartedTimestamp\":             time.Time{},\n\t\t\"GetDecisionTaskTimeout\":                  time.Duration(0),\n\t\t\"GetDecisionTimeout\":                      time.Duration(0),\n\t\t\"GetLastFirstEventID\":                     int64(0),\n\t\t\"GetLastProcessedEvent\":                   int64(0),\n\t\t\"GetParentWorkflowID\":                     \"\",\n\t\t\"GetStartVersion\":                         int64(0),\n\t\t\"GetTaskList\":                             \"\",\n\t\t\"GetTaskListKind\":                         types.TaskListKindNormal,\n\t\t\"GetDecisionVersion\":                      int64(0),\n\t\t\"GetEventBranchToken\":                     []uint8(nil),\n\t\t\"GetEventStoreVersion\":                    int32(0),\n\t\t\"GetExecutionContext\":                     []uint8(nil),\n\t\t\"GetFirstExecutionRunID\":                  []uint8(nil),\n\t\t\"GetHasRetryPolicy\":                       false,\n\t\t\"GetInitiatedID\":                          int64(0),\n\t\t\"GetIsCron\":                               false,\n\t\t\"GetLastEventTaskID\":                      int64(0),\n\t\t\"GetLastUpdatedTimestamp\":                 time.Time{},\n\t\t\"GetLastWriteEventID\":                     int64(0),\n\t\t\"GetMemo\":                                 map[string][]uint8(nil),\n\t\t\"GetParentDomainID\":                       []uint8(nil),\n\t\t\"GetParentRunID\":                          []uint8(nil),\n\t\t\"GetPartitionConfig\":                      map[string]string(nil),\n\t\t\"GetHistorySize\":                          int64(0),\n\t\t\"GetRetryAttempt\":                         int64(0),\n\t\t\"GetRetryBackoffCoefficient\":              float64(0),\n\t\t\"GetRetryExpiration\":                      time.Duration(0),\n\t\t\"GetRetryExpirationTimestamp\":             time.Time{},\n\t\t\"GetRetryInitialInterval\":                 time.Duration(0),\n\t\t\"GetRetryMaximumAttempts\":                 int32(0),\n\t\t\"GetRetryMaximumInterval\":                 time.Duration(0),\n\t\t\"GetRetryNonRetryableErrors\":              []string(nil),\n\t\t\"GetSearchAttributes\":                     map[string][]uint8(nil),\n\t\t\"GetSignalCount\":                          int64(0),\n\t\t\"GetStartTimestamp\":                       time.Time{},\n\t\t\"GetState\":                                int32(0),\n\t\t\"GetStickyScheduleToStartTimeout\":         time.Duration(0),\n\t\t\"GetStickyTaskList\":                       \"\",\n\t\t\"GetVersionHistories\":                     []uint8(nil),\n\t\t\"GetVersionHistoriesEncoding\":             \"\",\n\t\t\"GetWorkflowTimeout\":                      time.Duration(0),\n\t\t\"GetWorkflowTypeName\":                     \"\",\n\t\t\"GetChecksum\":                             []uint8(nil),\n\t\t\"GetChecksumEncoding\":                     \"\",\n\t},\n\t\"*serialization.TransferTaskInfo\": {\n\t\t\"GetDomainID\":                []uint8(nil),\n\t\t\"GetWorkflowID\":              \"\",\n\t\t\"GetRunID\":                   []uint8(nil),\n\t\t\"GetTaskType\":                int16(0),\n\t\t\"GetTargetDomainID\":          []uint8(nil),\n\t\t\"GetTargetDomainIDs\":         map[string]struct{}{},\n\t\t\"GetTargetWorkflowID\":        \"\",\n\t\t\"GetTargetRunID\":             []uint8(nil),\n\t\t\"GetTaskList\":                \"\",\n\t\t\"GetTargetChildWorkflowOnly\": false,\n\t\t\"GetScheduleID\":              int64(0),\n\t\t\"GetVersion\":                 int64(0),\n\t\t\"GetVisibilityTimestamp\":     time.Time{},\n\t\t\"GetOriginalTaskList\":        \"\",\n\t\t\"GetOriginalTaskListKind\":    types.TaskListKindNormal,\n\t},\n\t\"*serialization.TimerTaskInfo\": {\n\t\t\"GetDomainID\":        []uint8(nil),\n\t\t\"GetEventID\":         int64(0),\n\t\t\"GetRunID\":           []uint8(nil),\n\t\t\"GetScheduleAttempt\": int64(0),\n\t\t\"GetTaskType\":        int16(0),\n\t\t\"GetTimeoutType\":     int16(0),\n\t\t\"GetVersion\":         int64(0),\n\t\t\"GetWorkflowID\":      \"\",\n\t\t\"GetTaskList\":        \"\",\n\t},\n\t\"*serialization.ReplicationTaskInfo\": {\n\t\t\"GetBranchToken\":             []uint8(nil),\n\t\t\"GetCreationTimestamp\":       time.Time{},\n\t\t\"GetDomainID\":                []uint8(nil),\n\t\t\"GetEventStoreVersion\":       int32(0),\n\t\t\"GetFirstEventID\":            int64(0),\n\t\t\"GetNewRunBranchToken\":       []uint8(nil),\n\t\t\"GetNewRunEventStoreVersion\": int32(0),\n\t\t\"GetNextEventID\":             int64(0),\n\t\t\"GetRunID\":                   []uint8(nil),\n\t\t\"GetScheduledID\":             int64(0),\n\t\t\"GetTaskType\":                int16(0),\n\t\t\"GetVersion\":                 int64(0),\n\t\t\"GetWorkflowID\":              \"\",\n\t},\n\t\"*serialization.TaskListInfo\": {\n\t\t\"GetAckLevel\":        int64(0),\n\t\t\"GetExpiryTimestamp\": time.Time{},\n\t\t\"GetKind\":            int16(0),\n\t\t\"GetLastUpdated\":     time.Time{},\n\t},\n\t\"*serialization.TaskInfo\": {\n\t\t\"GetCreatedTimestamp\": time.Time{},\n\t\t\"GetExpiryTimestamp\":  time.Time{},\n\t\t\"GetPartitionConfig\":  map[string]string(nil),\n\t\t\"GetRunID\":            []uint8(nil),\n\t\t\"GetScheduleID\":       int64(0),\n\t\t\"GetWorkflowID\":       \"\",\n\t},\n\t\"*serialization.TimerInfo\": {\n\t\t\"GetExpiryTimestamp\": time.Time{},\n\t\t\"GetStartedID\":       int64(0),\n\t\t\"GetTaskID\":          int64(0),\n\t\t\"GetVersion\":         int64(0),\n\t},\n\t\"*serialization.RequestCancelInfo\": {\n\t\t\"GetCancelRequestID\":       \"\",\n\t\t\"GetInitiatedEventBatchID\": int64(0),\n\t\t\"GetVersion\":               int64(0),\n\t},\n\t\"*serialization.SignalInfo\": {\n\t\t\"GetControl\":               []uint8(nil),\n\t\t\"GetInitiatedEventBatchID\": int64(0),\n\t\t\"GetInput\":                 []uint8(nil),\n\t\t\"GetName\":                  \"\",\n\t\t\"GetRequestID\":             \"\",\n\t\t\"GetVersion\":               int64(0),\n\t},\n\t\"*serialization.ChildExecutionInfo\": {\n\t\t\"GetCreateRequestID\":        \"\",\n\t\t\"GetDomainID\":               \"\",\n\t\t\"GetDomainNameDEPRECATED\":   \"\",\n\t\t\"GetInitiatedEvent\":         []uint8(nil),\n\t\t\"GetInitiatedEventBatchID\":  int64(0),\n\t\t\"GetInitiatedEventEncoding\": \"\",\n\t\t\"GetParentClosePolicy\":      int32(0),\n\t\t\"GetStartedEvent\":           []uint8(nil),\n\t\t\"GetStartedEventEncoding\":   \"\",\n\t\t\"GetStartedID\":              int64(0),\n\t\t\"GetStartedRunID\":           []uint8(nil),\n\t\t\"GetStartedWorkflowID\":      \"\",\n\t\t\"GetVersion\":                int64(0),\n\t\t\"GetWorkflowTypeName\":       \"\",\n\t},\n\t\"*serialization.ActivityInfo\": {\n\t\t\"GetActivityID\":               \"\",\n\t\t\"GetAttempt\":                  int32(0),\n\t\t\"GetCancelRequestID\":          int64(0),\n\t\t\"GetCancelRequested\":          false,\n\t\t\"GetHasRetryPolicy\":           false,\n\t\t\"GetHeartbeatTimeout\":         time.Duration(0),\n\t\t\"GetRequestID\":                \"\",\n\t\t\"GetRetryBackoffCoefficient\":  float64(0),\n\t\t\"GetRetryExpirationTimestamp\": time.Time{},\n\t\t\"GetRetryInitialInterval\":     time.Duration(0),\n\t\t\"GetRetryLastFailureDetails\":  []uint8(nil),\n\t\t\"GetRetryLastFailureReason\":   \"\",\n\t\t\"GetRetryLastWorkerIdentity\":  \"\",\n\t\t\"GetRetryMaximumAttempts\":     int32(0),\n\t\t\"GetRetryMaximumInterval\":     time.Duration(0),\n\t\t\"GetRetryNonRetryableErrors\":  []string(nil),\n\t\t\"GetScheduleToCloseTimeout\":   time.Duration(0),\n\t\t\"GetScheduleToStartTimeout\":   time.Duration(0),\n\t\t\"GetScheduledEvent\":           []uint8(nil),\n\t\t\"GetScheduledEventBatchID\":    int64(0),\n\t\t\"GetScheduledEventEncoding\":   \"\",\n\t\t\"GetScheduledTimestamp\":       time.Time{},\n\t\t\"GetStartToCloseTimeout\":      time.Duration(0),\n\t\t\"GetStartedEvent\":             []uint8(nil),\n\t\t\"GetStartedEventEncoding\":     \"\",\n\t\t\"GetStartedID\":                int64(0),\n\t\t\"GetStartedIdentity\":          \"\",\n\t\t\"GetStartedTimestamp\":         time.Time{},\n\t\t\"GetTaskList\":                 \"\",\n\t\t\"GetTaskListKind\":             types.TaskListKindNormal,\n\t\t\"GetTimerTaskStatus\":          int32(0),\n\t\t\"GetVersion\":                  int64(0),\n\t},\n\t\"*serialization.HistoryTreeInfo\": {\n\t\t\"GetAncestors\":        []*types.HistoryBranchRange(nil),\n\t\t\"GetCreatedTimestamp\": time.Time{},\n\t\t\"GetInfo\":             \"\",\n\t},\n\t\"*serialization.DomainInfo\": {\n\t\t\"GetActiveClusterName\":           \"\",\n\t\t\"GetArchivalBucket\":              \"\",\n\t\t\"GetArchivalStatus\":              int16(0),\n\t\t\"GetBadBinaries\":                 []uint8(nil),\n\t\t\"GetBadBinariesEncoding\":         \"\",\n\t\t\"GetClusters\":                    []string(nil),\n\t\t\"GetConfigVersion\":               int64(0),\n\t\t\"GetData\":                        map[string]string(nil),\n\t\t\"GetDescription\":                 \"\",\n\t\t\"GetEmitMetric\":                  false,\n\t\t\"GetFailoverEndTimestamp\":        zeroUnix,\n\t\t\"GetFailoverNotificationVersion\": int64(0),\n\t\t\"GetFailoverVersion\":             int64(0),\n\t\t\"GetHistoryArchivalStatus\":       int16(0),\n\t\t\"GetHistoryArchivalURI\":          \"\",\n\t\t\"GetLastUpdatedTimestamp\":        time.Time{},\n\t\t\"GetName\":                        \"\",\n\t\t\"GetNotificationVersion\":         int64(0),\n\t\t\"GetOwner\":                       \"\",\n\t\t\"GetPreviousFailoverVersion\":     int64(0),\n\t\t\"GetRetention\":                   time.Duration(0),\n\t\t\"GetStatus\":                      int32(0),\n\t\t\"GetVisibilityArchivalStatus\":    int16(0),\n\t\t\"GetVisibilityArchivalURI\":       \"\",\n\t},\n\t\"*serialization.ShardInfo\": {\n\t\t\"GetClusterReplicationLevel\":               map[string]int64(nil),\n\t\t\"GetClusterTimerAckLevel\":                  map[string]time.Time(nil),\n\t\t\"GetClusterTransferAckLevel\":               map[string]int64(nil),\n\t\t\"GetDomainNotificationVersion\":             int64(0),\n\t\t\"GetOwner\":                                 \"\",\n\t\t\"GetPendingFailoverMarkers\":                []uint8(nil),\n\t\t\"GetPendingFailoverMarkersEncoding\":        \"\",\n\t\t\"GetReplicationAckLevel\":                   int64(0),\n\t\t\"GetReplicationDlqAckLevel\":                map[string]int64(nil),\n\t\t\"GetStolenSinceRenew\":                      int32(0),\n\t\t\"GetTimerAckLevel\":                         time.Time{},\n\t\t\"GetTimerProcessingQueueStates\":            []uint8(nil),\n\t\t\"GetTimerProcessingQueueStatesEncoding\":    \"\",\n\t\t\"GetTransferAckLevel\":                      int64(0),\n\t\t\"GetTransferProcessingQueueStates\":         []uint8(nil),\n\t\t\"GetTransferProcessingQueueStatesEncoding\": \"\",\n\t\t\"GetUpdatedAt\":                             time.Time{},\n\t\t\"GetQueueStates\":                           map[int32]*types.QueueState(nil),\n\t},\n}\n\nvar expectedNonEmpty = map[string]map[string]any{\n\t\"*serialization.WorkflowExecutionInfo\": {\n\t\t\"GetAutoResetPoints\":                    []byte(\"resetpoints\"),\n\t\t\"GetAutoResetPointsEncoding\":            \"\",\n\t\t\"GetCancelRequestID\":                    \"\",\n\t\t\"GetCancelRequested\":                    false,\n\t\t\"GetClientFeatureVersion\":               \"\",\n\t\t\"GetClientImpl\":                         \"\",\n\t\t\"GetClientLibraryVersion\":               \"\",\n\t\t\"GetCloseStatus\":                        int32(6),\n\t\t\"GetCompletionEvent\":                    []byte(\"completionEvent\"),\n\t\t\"GetCompletionEventEncoding\":            \"completionEventEncoding\",\n\t\t\"GetCreateRequestID\":                    \"\",\n\t\t\"GetCompletionEventBatchID\":             int64(2),\n\t\t\"GetCronSchedule\":                       \"\",\n\t\t\"GetCronOverlapPolicy\":                  int32(0),\n\t\t\"GetDecisionAttempt\":                    int64(0),\n\t\t\"GetDecisionOriginalScheduledTimestamp\": time.Time{},\n\t\t\"GetDecisionRequestID\":                  \"\",\n\t\t\"GetDecisionScheduleID\":                 int64(0),\n\t\t\"GetDecisionScheduledTimestamp\":         time.Time{},\n\t\t\"GetDecisionStartedID\":                  int64(0),\n\t\t\"GetDecisionStartedTimestamp\":           time.Time{},\n\t\t\"GetDecisionTaskTimeout\":                time.Duration(0),\n\t\t\"GetDecisionTimeout\":                    time.Duration(4),\n\t\t\"GetLastFirstEventID\":                   int64(7),\n\t\t\"GetLastProcessedEvent\":                 int64(0),\n\t\t\"GetParentWorkflowID\":                   \"parentWorkflowID\",\n\t\t\"GetStartVersion\":                       int64(0),\n\t\t\"GetTaskList\":                           \"taskList\",\n\t\t\"GetTaskListKind\":                       types.TaskListKindEphemeral,\n\t\t\"GetDecisionVersion\":                    int64(0),\n\t\t\"GetEventBranchToken\":                   []uint8(nil),\n\t\t\"GetEventStoreVersion\":                  int32(0),\n\t\t\"GetExecutionContext\":                   []byte(\"executionContext\"),\n\t\t\"GetFirstExecutionRunID\":                []uint8(nil),\n\t\t\"GetHasRetryPolicy\":                     false,\n\t\t\"GetInitiatedID\":                        int64(1),\n\t\t\"GetIsCron\":                             false,\n\t\t\"GetLastEventTaskID\":                    int64(0),\n\t\t\"GetLastUpdatedTimestamp\":               time.Time{},\n\t\t\"GetLastWriteEventID\":                   int64(0),\n\t\t\"GetMemo\":                               map[string][]uint8(nil),\n\t\t\"GetParentDomainID\":                     []byte(parentDomainID),\n\t\t\"GetParentRunID\":                        []byte(parentRunID),\n\t\t\"GetPartitionConfig\":                    map[string]string(nil),\n\t\t\"GetHistorySize\":                        int64(0),\n\t\t\"GetRetryAttempt\":                       int64(0),\n\t\t\"GetRetryBackoffCoefficient\":            float64(0),\n\t\t\"GetRetryExpiration\":                    time.Duration(0),\n\t\t\"GetRetryExpirationTimestamp\":           time.Time{},\n\t\t\"GetRetryInitialInterval\":               time.Duration(0),\n\t\t\"GetRetryMaximumAttempts\":               int32(0),\n\t\t\"GetRetryMaximumInterval\":               time.Duration(0),\n\t\t\"GetRetryNonRetryableErrors\":            []string(nil),\n\t\t\"GetSearchAttributes\": map[string][]uint8{\n\t\t\t\"key\": []byte(\"value\"),\n\t\t},\n\t\t\"GetSignalCount\":                          int64(0),\n\t\t\"GetStartTimestamp\":                       time.Time{},\n\t\t\"GetState\":                                int32(5),\n\t\t\"GetStickyScheduleToStartTimeout\":         time.Duration(0),\n\t\t\"GetStickyTaskList\":                       \"\",\n\t\t\"GetVersionHistories\":                     []uint8(nil),\n\t\t\"GetVersionHistoriesEncoding\":             \"\",\n\t\t\"GetWorkflowTimeout\":                      time.Duration(3),\n\t\t\"GetWorkflowTypeName\":                     \"workflowTypeName\",\n\t\t\"GetChecksum\":                             []uint8(nil),\n\t\t\"GetChecksumEncoding\":                     \"\",\n\t\t\"GetActiveClusterSelectionPolicyEncoding\": \"\",\n\t},\n\t\"*serialization.TransferTaskInfo\": {\n\t\t\"GetDomainID\":                []uint8(taskDomainID),\n\t\t\"GetWorkflowID\":              \"workflowID\",\n\t\t\"GetRunID\":                   []uint8(taskRunID),\n\t\t\"GetTaskType\":                int16(1),\n\t\t\"GetTargetDomainID\":          []uint8(parentDomainID),\n\t\t\"GetTargetDomainIDs\":         map[string]struct{}{parentDomainID.String(): struct{}{}, taskDomainID.String(): struct{}{}},\n\t\t\"GetTargetWorkflowID\":        \"targetID\",\n\t\t\"GetTargetRunID\":             []uint8(parentRunID),\n\t\t\"GetTaskList\":                \"tasklist\",\n\t\t\"GetTargetChildWorkflowOnly\": true,\n\t\t\"GetScheduleID\":              int64(2),\n\t\t\"GetVersion\":                 int64(3),\n\t\t\"GetVisibilityTimestamp\":     taskInfoCreateTime,\n\t\t\"GetOriginalTaskList\":        \"originalTaskList\",\n\t\t\"GetOriginalTaskListKind\":    types.TaskListKindEphemeral,\n\t},\n\t\"*serialization.TimerTaskInfo\": {\n\t\t\"GetDomainID\":        []byte(taskDomainID),\n\t\t\"GetEventID\":         int64(5),\n\t\t\"GetRunID\":           []byte(taskRunID),\n\t\t\"GetScheduleAttempt\": int64(4),\n\t\t\"GetTaskType\":        int16(1),\n\t\t\"GetTimeoutType\":     int16(2),\n\t\t\"GetVersion\":         int64(3),\n\t\t\"GetWorkflowID\":      \"workflowID\",\n\t\t\"GetTaskList\":        \"taskList\",\n\t},\n\t\"*serialization.ReplicationTaskInfo\": {\n\t\t\"GetBranchToken\":             []byte(\"branchToken\"),\n\t\t\"GetCreationTimestamp\":       replicationCreationTimestamp,\n\t\t\"GetDomainID\":                []uint8(replicationTaskDomainID),\n\t\t\"GetEventStoreVersion\":       int32(6),\n\t\t\"GetFirstEventID\":            int64(3),\n\t\t\"GetNewRunBranchToken\":       []byte(\"newRunBranchToken\"),\n\t\t\"GetNewRunEventStoreVersion\": int32(7),\n\t\t\"GetNextEventID\":             int64(4),\n\t\t\"GetRunID\":                   []byte(replicationTaskRunID),\n\t\t\"GetScheduledID\":             int64(5),\n\t\t\"GetTaskType\":                int16(1),\n\t\t\"GetVersion\":                 int64(2),\n\t\t\"GetWorkflowID\":              \"workflowID\",\n\t},\n\t\"*serialization.TaskListInfo\": {\n\t\t\"GetAckLevel\":        int64(2),\n\t\t\"GetExpiryTimestamp\": taskListInfoExpireTime,\n\t\t\"GetKind\":            int16(1),\n\t\t\"GetLastUpdated\":     taskListInfoLastUpdateTime,\n\t},\n\t\"*serialization.TaskInfo\": {\n\t\t\"GetCreatedTimestamp\": taskInfoCreateTime,\n\t\t\"GetExpiryTimestamp\":  taskInfoExpiryTime,\n\t\t\"GetPartitionConfig\":  map[string]string{\"key\": \"value\"},\n\t\t\"GetRunID\":            []byte(taskInfoRunID),\n\t\t\"GetScheduleID\":       int64(1),\n\t\t\"GetWorkflowID\":       \"workflowID\",\n\t},\n\t\"*serialization.TimerInfo\": {\n\t\t\"GetExpiryTimestamp\": timerInfoExpireTime,\n\t\t\"GetStartedID\":       int64(2),\n\t\t\"GetTaskID\":          int64(3),\n\t\t\"GetVersion\":         int64(1),\n\t},\n\t\"*serialization.RequestCancelInfo\": {\n\t\t\"GetCancelRequestID\":       \"cancelRequestID\",\n\t\t\"GetInitiatedEventBatchID\": int64(2),\n\t\t\"GetVersion\":               int64(1),\n\t},\n\t\"*serialization.SignalInfo\": {\n\t\t\"GetControl\":               []byte(\"signalControl\"),\n\t\t\"GetInitiatedEventBatchID\": int64(2),\n\t\t\"GetInput\":                 []byte(\"signalInput\"),\n\t\t\"GetName\":                  \"signalName\",\n\t\t\"GetRequestID\":             \"signalRequestID\",\n\t\t\"GetVersion\":               int64(1),\n\t},\n\t\"*serialization.ChildExecutionInfo\": {\n\t\t\"GetCreateRequestID\":        \"createRequestID\",\n\t\t\"GetDomainID\":               \"domainID\",\n\t\t\"GetDomainNameDEPRECATED\":   \"\",\n\t\t\"GetInitiatedEvent\":         []byte(\"initiatedEvent\"),\n\t\t\"GetInitiatedEventBatchID\":  int64(2),\n\t\t\"GetInitiatedEventEncoding\": \"initiatedEventEncoding\",\n\t\t\"GetParentClosePolicy\":      int32(1),\n\t\t\"GetStartedEvent\":           []byte(\"startedEvent\"),\n\t\t\"GetStartedEventEncoding\":   \"startedEventEncoding\",\n\t\t\"GetStartedID\":              int64(3),\n\t\t\"GetStartedRunID\":           []byte(childExecutionInfoStartedRunID),\n\t\t\"GetStartedWorkflowID\":      \"startedWorkflowID\",\n\t\t\"GetVersion\":                int64(1),\n\t\t\"GetWorkflowTypeName\":       \"workflowTypeName\",\n\t},\n\t\"*serialization.ActivityInfo\": {\n\t\t\"GetActivityID\":               \"activityID\",\n\t\t\"GetAttempt\":                  int32(6),\n\t\t\"GetCancelRequestID\":          int64(4),\n\t\t\"GetCancelRequested\":          true,\n\t\t\"GetHasRetryPolicy\":           true,\n\t\t\"GetHeartbeatTimeout\":         time.Duration(4),\n\t\t\"GetRequestID\":                \"requestID\",\n\t\t\"GetRetryBackoffCoefficient\":  float64(8),\n\t\t\"GetRetryExpirationTimestamp\": activeInfoRetryExpirationTime,\n\t\t\"GetRetryInitialInterval\":     time.Duration(5),\n\t\t\"GetRetryLastFailureDetails\":  []byte(\"retryLastFailureDetails\"),\n\t\t\"GetRetryLastFailureReason\":   \"retryLastFailureReason\",\n\t\t\"GetRetryLastWorkerIdentity\":  \"retryLastWorkerIdentity\",\n\t\t\"GetRetryMaximumAttempts\":     int32(7),\n\t\t\"GetRetryMaximumInterval\":     time.Duration(6),\n\t\t\"GetRetryNonRetryableErrors\":  []string{\"error1\", \"error2\"},\n\t\t\"GetScheduleToCloseTimeout\":   time.Duration(1),\n\t\t\"GetScheduleToStartTimeout\":   time.Duration(2),\n\t\t\"GetScheduledEvent\":           []byte(\"scheduledEvent\"),\n\t\t\"GetScheduledEventBatchID\":    int64(2),\n\t\t\"GetScheduledEventEncoding\":   \"scheduledEventEncoding\",\n\t\t\"GetScheduledTimestamp\":       activityInfoScheduledTime,\n\t\t\"GetStartToCloseTimeout\":      time.Duration(3),\n\t\t\"GetStartedEvent\":             []byte(\"startedEvent\"),\n\t\t\"GetStartedEventEncoding\":     \"startedEventEncoding\",\n\t\t\"GetStartedID\":                int64(3),\n\t\t\"GetStartedIdentity\":          \"startedIdentity\",\n\t\t\"GetStartedTimestamp\":         activeInfoStartedTime,\n\t\t\"GetTaskList\":                 \"taskList\",\n\t\t\"GetTaskListKind\":             types.TaskListKindSticky,\n\n\t\t\"GetTimerTaskStatus\": int32(5),\n\t\t\"GetVersion\":         int64(1),\n\t},\n\t\"*serialization.HistoryTreeInfo\": {\n\t\t\"GetAncestors\": []*types.HistoryBranchRange{\n\t\t\t{\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t},\n\t\t},\n\t\t\"GetCreatedTimestamp\": historyTreeEventCreatedTime,\n\t\t\"GetInfo\":             \"historyTreeInfo\",\n\t},\n\t\"*serialization.DomainInfo\": {\n\t\t\"GetActiveClusterName\":           \"cluster1\",\n\t\t\"GetArchivalBucket\":              \"archivalBucket\",\n\t\t\"GetArchivalStatus\":              int16(2),\n\t\t\"GetBadBinaries\":                 []byte(\"badBinaries\"),\n\t\t\"GetBadBinariesEncoding\":         \"badBinariesEncoding\",\n\t\t\"GetClusters\":                    []string{\"cluster1\", \"cluster2\"},\n\t\t\"GetConfigVersion\":               int64(3),\n\t\t\"GetData\":                        map[string]string{\"datakey\": \"datavalue\"},\n\t\t\"GetDescription\":                 \"description\",\n\t\t\"GetEmitMetric\":                  true,\n\t\t\"GetFailoverEndTimestamp\":        domainInfoFailoverEndTimestamp,\n\t\t\"GetFailoverNotificationVersion\": int64(5),\n\t\t\"GetFailoverVersion\":             int64(6),\n\t\t\"GetHistoryArchivalStatus\":       int16(7),\n\t\t\"GetHistoryArchivalURI\":          \"historyArchivalURI\",\n\t\t\"GetLastUpdatedTimestamp\":        domainInfoLastUpdatedTimestamp,\n\t\t\"GetName\":                        \"name\",\n\t\t\"GetNotificationVersion\":         int64(4),\n\t\t\"GetOwner\":                       \"owner\",\n\t\t\"GetPreviousFailoverVersion\":     int64(9),\n\t\t\"GetRetention\":                   time.Duration(1),\n\t\t\"GetStatus\":                      int32(1),\n\t\t\"GetVisibilityArchivalStatus\":    int16(8),\n\t\t\"GetVisibilityArchivalURI\":       \"visibilityArchivalURI\",\n\t},\n\t\"*serialization.ShardInfo\": {\n\t\t\"GetClusterReplicationLevel\": map[string]int64{\n\t\t\t\"cluster1\": 6,\n\t\t},\n\t\t\"GetClusterTimerAckLevel\": map[string]time.Time{\n\t\t\t\"cluster1\": shardInfoTimerAckLevel,\n\t\t},\n\t\t\"GetClusterTransferAckLevel\": map[string]int64{\n\t\t\t\"cluster1\": 5,\n\t\t},\n\t\t\"GetDomainNotificationVersion\":      int64(4),\n\t\t\"GetOwner\":                          \"owner\",\n\t\t\"GetPendingFailoverMarkers\":         []byte(\"pendingFailoverMarkers\"),\n\t\t\"GetPendingFailoverMarkersEncoding\": \"pendingFailoverMarkersEncoding\",\n\t\t\"GetReplicationAckLevel\":            int64(2),\n\t\t\"GetReplicationDlqAckLevel\": map[string]int64{\n\t\t\t\"cluster1\": 7,\n\t\t},\n\t\t\"GetStolenSinceRenew\":                      int32(1),\n\t\t\"GetTimerAckLevel\":                         shardInfoTimerAckLevel,\n\t\t\"GetTimerProcessingQueueStates\":            []byte(\"timerProcessingQueueStates\"),\n\t\t\"GetTimerProcessingQueueStatesEncoding\":    \"timerProcessingQueueStatesEncoding\",\n\t\t\"GetTransferAckLevel\":                      int64(3),\n\t\t\"GetTransferProcessingQueueStates\":         []byte(\"transferProcessingQueueStates\"),\n\t\t\"GetTransferProcessingQueueStatesEncoding\": \"transferProcessingQueueStatesEncoding\",\n\t\t\"GetUpdatedAt\":                             shardInfoUpdatedTime,\n\t\t\"GetQueueStates\": map[int32]*types.QueueState{\n\t\t\t0: &types.QueueState{\n\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: 2,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t},\n}\n"
  },
  {
    "path": "common/persistence/serialization/getters_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"reflect\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGettersForAllNilInfos(t *testing.T) {\n\tfor _, info := range []any{\n\t\t&WorkflowExecutionInfo{},\n\t\t&TransferTaskInfo{},\n\t\t&TimerTaskInfo{},\n\t\t&ReplicationTaskInfo{},\n\t\t&TaskListInfo{},\n\t\t&TaskInfo{},\n\t\t&TimerInfo{},\n\t\t&RequestCancelInfo{},\n\t\t&SignalInfo{},\n\t\t&ChildExecutionInfo{},\n\t\t&ActivityInfo{},\n\t\t&HistoryTreeInfo{},\n\t\t&DomainInfo{},\n\t\t&ShardInfo{},\n\t} {\n\t\tname := reflect.TypeOf(info).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := nilView(info)\n\t\t\tif diff := cmp.Diff(expectedNil[name], res); diff != \"\" {\n\t\t\t\tt.Errorf(\"nilValue mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGettersForEmptyInfos(t *testing.T) {\n\tfor _, info := range []any{\n\t\t&WorkflowExecutionInfo{},\n\t\t&TransferTaskInfo{},\n\t\t&TimerTaskInfo{},\n\t\t&ReplicationTaskInfo{},\n\t\t&TaskListInfo{},\n\t\t&TaskInfo{},\n\t\t&TimerInfo{},\n\t\t&RequestCancelInfo{},\n\t\t&SignalInfo{},\n\t\t&ChildExecutionInfo{},\n\t\t&ActivityInfo{},\n\t\t&HistoryTreeInfo{},\n\t\t&DomainInfo{},\n\t\t&ShardInfo{},\n\t} {\n\t\tname := reflect.TypeOf(info).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := emptyView(info)\n\t\t\tif diff := cmp.Diff(expectedEmpty[name], res); diff != \"\" {\n\t\t\t\tt.Errorf(\"emptyValue mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nvar (\n\tparentDomainID = MustParseUUID(\"00000000-0000-0000-0000-000000000001\")\n\tparentRunID    = MustParseUUID(\"00000000-0000-0000-0000-000000000002\")\n\n\ttaskDomainID = MustParseUUID(\"00000000-0000-0000-0000-000000000003\")\n\ttaskRunID    = MustParseUUID(\"00000000-0000-0000-0000-000000000004\")\n\n\treplicationTaskDomainID      = MustParseUUID(\"00000000-0000-0000-0000-000000000005\")\n\treplicationTaskRunID         = MustParseUUID(\"00000000-0000-0000-0000-000000000006\")\n\treplicationCreationTimestamp = time.Unix(10, 0)\n\n\ttaskListInfoExpireTime     = time.Unix(20, 0)\n\ttaskListInfoLastUpdateTime = time.Unix(30, 0)\n\n\ttaskInfoRunID      = MustParseUUID(\"00000000-0000-0000-0000-000000000007\")\n\ttaskInfoExpiryTime = time.Unix(40, 0)\n\ttaskInfoCreateTime = time.Unix(50, 0)\n\n\ttimerInfoExpireTime = time.Unix(60, 0)\n\n\tsignalInfoInput   = []byte(\"signalInput\")\n\tsignalInfoControl = []byte(\"signalControl\")\n\n\tchildExecutionInfoInitiatedEvent = []byte(\"initiatedEvent\")\n\tchildExecutionInfoStartedRunID   = MustParseUUID(\"00000000-0000-0000-0000-000000000008\")\n\tchildExecutionInfoStartedEvent   = []byte(\"startedEvent\")\n\n\tactivityInfoScheduledTime     = time.Unix(70, 0)\n\tactiveInfoStartedTime         = time.Unix(80, 0)\n\tactiveInfoRetryExpirationTime = time.Unix(90, 0)\n\n\thistoryTreeEventCreatedTime = time.Unix(100, 0)\n\n\tdomainInfoFailoverEndTimestamp = time.Unix(110, 0)\n\tdomainInfoLastUpdatedTimestamp = time.Unix(120, 0)\n\n\tshardInfoUpdatedTime   = time.Unix(130, 0)\n\tshardInfoTimerAckLevel = time.Unix(140, 0)\n)\n\nfunc TestGettersForInfos(t *testing.T) {\n\tfor _, info := range []any{\n\t\t&WorkflowExecutionInfo{\n\t\t\tParentDomainID:          parentDomainID,\n\t\t\tParentWorkflowID:        \"parentWorkflowID\",\n\t\t\tParentRunID:             parentRunID,\n\t\t\tInitiatedID:             1,\n\t\t\tCompletionEventBatchID:  common.Int64Ptr(2),\n\t\t\tCompletionEvent:         []byte(\"completionEvent\"),\n\t\t\tCompletionEventEncoding: \"completionEventEncoding\",\n\t\t\tTaskList:                \"taskList\",\n\t\t\tTaskListKind:            types.TaskListKindEphemeral,\n\t\t\tWorkflowTypeName:        \"workflowTypeName\",\n\t\t\tWorkflowTimeout:         3,\n\t\t\tDecisionTimeout:         4,\n\t\t\tExecutionContext:        []byte(\"executionContext\"),\n\t\t\tState:                   5,\n\t\t\tCloseStatus:             6,\n\t\t\tLastFirstEventID:        7,\n\t\t\tAutoResetPoints:         []byte(\"resetpoints\"),\n\t\t\tSearchAttributes:        map[string][]byte{\"key\": []byte(\"value\")},\n\t\t},\n\t\t&TransferTaskInfo{\n\t\t\tDomainID:                taskDomainID,\n\t\t\tWorkflowID:              \"workflowID\",\n\t\t\tRunID:                   taskRunID,\n\t\t\tTaskType:                1,\n\t\t\tTargetDomainID:          parentDomainID,\n\t\t\tTargetDomainIDs:         []UUID{parentDomainID, taskDomainID},\n\t\t\tTargetWorkflowID:        \"targetID\",\n\t\t\tTargetRunID:             parentRunID,\n\t\t\tTaskList:                \"tasklist\",\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tScheduleID:              2,\n\t\t\tVersion:                 3,\n\t\t\tVisibilityTimestamp:     taskInfoCreateTime,\n\t\t\tOriginalTaskList:        \"originalTaskList\",\n\t\t\tOriginalTaskListKind:    types.TaskListKindEphemeral,\n\t\t},\n\t\t&TimerTaskInfo{\n\t\t\tDomainID:        taskDomainID,\n\t\t\tWorkflowID:      \"workflowID\",\n\t\t\tRunID:           taskRunID,\n\t\t\tTaskType:        1,\n\t\t\tTimeoutType:     common.Int16Ptr(2),\n\t\t\tVersion:         3,\n\t\t\tScheduleAttempt: 4,\n\t\t\tEventID:         5,\n\t\t\tTaskList:        \"taskList\",\n\t\t},\n\t\t&ReplicationTaskInfo{\n\t\t\tDomainID:                replicationTaskDomainID,\n\t\t\tWorkflowID:              \"workflowID\",\n\t\t\tRunID:                   replicationTaskRunID,\n\t\t\tTaskType:                1,\n\t\t\tVersion:                 2,\n\t\t\tFirstEventID:            3,\n\t\t\tNextEventID:             4,\n\t\t\tScheduledID:             5,\n\t\t\tEventStoreVersion:       6,\n\t\t\tNewRunEventStoreVersion: 7,\n\t\t\tBranchToken:             []byte(\"branchToken\"),\n\t\t\tNewRunBranchToken:       []byte(\"newRunBranchToken\"),\n\t\t\tCreationTimestamp:       replicationCreationTimestamp,\n\t\t},\n\t\t&TaskListInfo{\n\t\t\tKind:            1,\n\t\t\tAckLevel:        2,\n\t\t\tExpiryTimestamp: taskListInfoExpireTime,\n\t\t\tLastUpdated:     taskListInfoLastUpdateTime,\n\t\t},\n\t\t&TaskInfo{\n\t\t\tWorkflowID:       \"workflowID\",\n\t\t\tRunID:            taskInfoRunID,\n\t\t\tScheduleID:       1,\n\t\t\tExpiryTimestamp:  taskInfoExpiryTime,\n\t\t\tCreatedTimestamp: taskInfoCreateTime,\n\t\t\tPartitionConfig: map[string]string{\n\t\t\t\t\"key\": \"value\",\n\t\t\t},\n\t\t},\n\t\t&TimerInfo{\n\t\t\tVersion:         1,\n\t\t\tStartedID:       2,\n\t\t\tExpiryTimestamp: timerInfoExpireTime,\n\t\t\tTaskID:          3,\n\t\t},\n\t\t&RequestCancelInfo{\n\t\t\tVersion:               1,\n\t\t\tInitiatedEventBatchID: 2,\n\t\t\tCancelRequestID:       \"cancelRequestID\",\n\t\t},\n\t\t&SignalInfo{\n\t\t\tVersion:               1,\n\t\t\tInitiatedEventBatchID: 2,\n\t\t\tRequestID:             \"signalRequestID\",\n\t\t\tName:                  \"signalName\",\n\t\t\tInput:                 signalInfoInput,\n\t\t\tControl:               signalInfoControl,\n\t\t},\n\t\t&ChildExecutionInfo{\n\t\t\tVersion:                1,\n\t\t\tInitiatedEventBatchID:  2,\n\t\t\tStartedID:              3,\n\t\t\tInitiatedEvent:         childExecutionInfoInitiatedEvent,\n\t\t\tInitiatedEventEncoding: \"initiatedEventEncoding\",\n\t\t\tStartedWorkflowID:      \"startedWorkflowID\",\n\t\t\tStartedRunID:           childExecutionInfoStartedRunID,\n\t\t\tStartedEvent:           childExecutionInfoStartedEvent,\n\t\t\tStartedEventEncoding:   \"startedEventEncoding\",\n\t\t\tCreateRequestID:        \"createRequestID\",\n\t\t\tDomainID:               \"domainID\",\n\t\t\tWorkflowTypeName:       \"workflowTypeName\",\n\t\t\tParentClosePolicy:      1,\n\t\t},\n\t\t&ActivityInfo{\n\t\t\tVersion:                  1,\n\t\t\tScheduledEventBatchID:    2,\n\t\t\tScheduledEvent:           []byte(\"scheduledEvent\"),\n\t\t\tScheduledEventEncoding:   \"scheduledEventEncoding\",\n\t\t\tScheduledTimestamp:       activityInfoScheduledTime,\n\t\t\tStartedID:                3,\n\t\t\tStartedEvent:             []byte(\"startedEvent\"),\n\t\t\tStartedEventEncoding:     \"startedEventEncoding\",\n\t\t\tStartedTimestamp:         activeInfoStartedTime,\n\t\t\tActivityID:               \"activityID\",\n\t\t\tRequestID:                \"requestID\",\n\t\t\tScheduleToCloseTimeout:   time.Duration(1),\n\t\t\tScheduleToStartTimeout:   time.Duration(2),\n\t\t\tStartToCloseTimeout:      time.Duration(3),\n\t\t\tHeartbeatTimeout:         time.Duration(4),\n\t\t\tCancelRequested:          true,\n\t\t\tCancelRequestID:          4,\n\t\t\tTimerTaskStatus:          5,\n\t\t\tAttempt:                  6,\n\t\t\tTaskList:                 \"taskList\",\n\t\t\tTaskListKind:             types.TaskListKindSticky,\n\t\t\tStartedIdentity:          \"startedIdentity\",\n\t\t\tHasRetryPolicy:           true,\n\t\t\tRetryInitialInterval:     time.Duration(5),\n\t\t\tRetryMaximumInterval:     time.Duration(6),\n\t\t\tRetryMaximumAttempts:     7,\n\t\t\tRetryExpirationTimestamp: activeInfoRetryExpirationTime,\n\t\t\tRetryBackoffCoefficient:  8,\n\t\t\tRetryNonRetryableErrors:  []string{\"error1\", \"error2\"},\n\t\t\tRetryLastWorkerIdentity:  \"retryLastWorkerIdentity\",\n\t\t\tRetryLastFailureReason:   \"retryLastFailureReason\",\n\t\t\tRetryLastFailureDetails:  []byte(\"retryLastFailureDetails\"),\n\t\t},\n\t\t&HistoryTreeInfo{\n\t\t\tCreatedTimestamp: historyTreeEventCreatedTime,\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tInfo: \"historyTreeInfo\",\n\t\t},\n\t\t&DomainInfo{\n\t\t\tName:                         \"name\",\n\t\t\tDescription:                  \"description\",\n\t\t\tOwner:                        \"owner\",\n\t\t\tStatus:                       1,\n\t\t\tRetention:                    time.Duration(1),\n\t\t\tEmitMetric:                   true,\n\t\t\tArchivalBucket:               \"archivalBucket\",\n\t\t\tArchivalStatus:               2,\n\t\t\tConfigVersion:                3,\n\t\t\tNotificationVersion:          4,\n\t\t\tFailoverNotificationVersion:  5,\n\t\t\tFailoverVersion:              6,\n\t\t\tActiveClusterName:            \"cluster1\",\n\t\t\tClusters:                     []string{\"cluster1\", \"cluster2\"},\n\t\t\tActiveClustersConfig:         []byte(\"activeClustersConfig\"),\n\t\t\tActiveClustersConfigEncoding: \"activeClustersConfigEncoding\",\n\t\t\tData: map[string]string{\n\t\t\t\t\"datakey\": \"datavalue\",\n\t\t\t},\n\t\t\tBadBinaries:              []byte(\"badBinaries\"),\n\t\t\tBadBinariesEncoding:      \"badBinariesEncoding\",\n\t\t\tHistoryArchivalStatus:    7,\n\t\t\tHistoryArchivalURI:       \"historyArchivalURI\",\n\t\t\tVisibilityArchivalStatus: 8,\n\t\t\tVisibilityArchivalURI:    \"visibilityArchivalURI\",\n\t\t\tFailoverEndTimestamp:     &domainInfoFailoverEndTimestamp,\n\t\t\tPreviousFailoverVersion:  9,\n\t\t\tLastUpdatedTimestamp:     domainInfoLastUpdatedTimestamp,\n\t\t\tIsolationGroups:          []byte(\"isolationGroups\"),\n\t\t\tIsolationGroupsEncoding:  \"isolationGroupsEncoding\",\n\t\t},\n\t\t&ShardInfo{\n\t\t\tStolenSinceRenew:          1,\n\t\t\tUpdatedAt:                 shardInfoUpdatedTime,\n\t\t\tReplicationAckLevel:       2,\n\t\t\tTransferAckLevel:          3,\n\t\t\tTimerAckLevel:             shardInfoTimerAckLevel,\n\t\t\tDomainNotificationVersion: 4,\n\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\"cluster1\": 5,\n\t\t\t},\n\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\"cluster1\": shardInfoTimerAckLevel,\n\t\t\t},\n\t\t\tOwner: \"owner\",\n\t\t\tClusterReplicationLevel: map[string]int64{\n\t\t\t\t\"cluster1\": 6,\n\t\t\t},\n\t\t\tPendingFailoverMarkers:         []byte(\"pendingFailoverMarkers\"),\n\t\t\tPendingFailoverMarkersEncoding: \"pendingFailoverMarkersEncoding\",\n\t\t\tReplicationDlqAckLevel: map[string]int64{\n\t\t\t\t\"cluster1\": 7,\n\t\t\t},\n\t\t\tTransferProcessingQueueStates:             []byte(\"transferProcessingQueueStates\"),\n\t\t\tTransferProcessingQueueStatesEncoding:     \"transferProcessingQueueStatesEncoding\",\n\t\t\tTimerProcessingQueueStates:                []byte(\"timerProcessingQueueStates\"),\n\t\t\tTimerProcessingQueueStatesEncoding:        \"timerProcessingQueueStatesEncoding\",\n\t\t\tCrossClusterProcessingQueueStates:         []byte(\"crossClusterProcessingQueueStates\"),\n\t\t\tCrossClusterProcessingQueueStatesEncoding: \"crossClusterProcessingQueueStatesEncoding\",\n\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t0: &types.QueueState{\n\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\t\tTaskID: 2,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t} {\n\t\tname := reflect.TypeOf(info).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tval := infoView(info)\n\t\t\trequire.Equal(t, expectedNonEmpty[name], val)\n\t\t})\n\t}\n}\n\nfunc nilView(info any) map[string]any {\n\tinfoVal := reflect.ValueOf(info)\n\tinfoT := reflect.TypeOf(infoVal.Interface())\n\tv := reflect.Zero(infoT)\n\tres := make(map[string]any)\n\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\tmethod := infoT.Method(i)\n\t\tif strings.HasPrefix(method.Name, \"Get\") {\n\t\t\tres[method.Name] = v.MethodByName(method.Name).Call(nil)[0].Interface()\n\t\t}\n\t}\n\n\treturn res\n}\n\nfunc emptyView(info any) map[string]any {\n\tinfoVal := reflect.ValueOf(info)\n\tinfoT := reflect.TypeOf(infoVal.Interface())\n\tres := make(map[string]any)\n\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\tmethod := infoT.Method(i)\n\t\tif strings.HasPrefix(method.Name, \"Get\") {\n\t\t\tres[method.Name] = infoVal.MethodByName(method.Name).Call(nil)[0].Interface()\n\t\t}\n\t}\n\n\treturn res\n}\n\nfunc infoView(info any) map[string]any {\n\tv := reflect.ValueOf(info)\n\tinfoT := reflect.TypeOf(v.Interface())\n\tres := make(map[string]any)\n\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\tmethod := infoT.Method(i)\n\t\tif strings.HasPrefix(method.Name, \"Get\") {\n\t\t\tresVal := v.MethodByName(method.Name).Call(nil)[0]\n\t\t\tres[method.Name] = resVal.Interface()\n\t\t}\n\t}\n\n\treturn res\n}\n"
  },
  {
    "path": "common/persistence/serialization/interfaces.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go -self_package github.com/uber/cadence/common/persistence/serialization\n\npackage serialization\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/thriftrw/protocol/stream\"\n\t\"go.uber.org/thriftrw/wire\"\n\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ShardInfo blob in a serialization agnostic format\n\tShardInfo struct {\n\t\tStolenSinceRenew                          int32\n\t\tUpdatedAt                                 time.Time\n\t\tReplicationAckLevel                       int64\n\t\tTransferAckLevel                          int64\n\t\tTimerAckLevel                             time.Time\n\t\tDomainNotificationVersion                 int64\n\t\tClusterTransferAckLevel                   map[string]int64\n\t\tClusterTimerAckLevel                      map[string]time.Time\n\t\tOwner                                     string\n\t\tClusterReplicationLevel                   map[string]int64\n\t\tPendingFailoverMarkers                    []byte\n\t\tPendingFailoverMarkersEncoding            string\n\t\tReplicationDlqAckLevel                    map[string]int64\n\t\tTransferProcessingQueueStates             []byte\n\t\tTransferProcessingQueueStatesEncoding     string\n\t\tCrossClusterProcessingQueueStates         []byte\n\t\tCrossClusterProcessingQueueStatesEncoding string\n\t\tTimerProcessingQueueStates                []byte\n\t\tTimerProcessingQueueStatesEncoding        string\n\t\tQueueStates                               map[int32]*types.QueueState\n\t}\n\n\t// DomainInfo blob in a serialization agnostic format\n\tDomainInfo struct {\n\t\tName                         string // TODO: This field seems not to be required. We already store domain name in another column.\n\t\tDescription                  string\n\t\tOwner                        string\n\t\tStatus                       int32\n\t\tRetention                    time.Duration\n\t\tEmitMetric                   bool\n\t\tArchivalBucket               string\n\t\tArchivalStatus               int16\n\t\tConfigVersion                int64\n\t\tNotificationVersion          int64\n\t\tFailoverNotificationVersion  int64\n\t\tFailoverVersion              int64\n\t\tActiveClusterName            string\n\t\tActiveClustersConfig         []byte\n\t\tActiveClustersConfigEncoding string\n\t\tClusters                     []string\n\t\tData                         map[string]string\n\t\tBadBinaries                  []byte\n\t\tBadBinariesEncoding          string\n\t\tHistoryArchivalStatus        int16\n\t\tHistoryArchivalURI           string\n\t\tVisibilityArchivalStatus     int16\n\t\tVisibilityArchivalURI        string\n\t\tFailoverEndTimestamp         *time.Time // TODO: There is logic checking if it's nil, should revisit this\n\t\tPreviousFailoverVersion      int64\n\t\tLastUpdatedTimestamp         time.Time\n\t\tIsolationGroups              []byte\n\t\tIsolationGroupsEncoding      string\n\t\tAsyncWorkflowConfig          []byte\n\t\tAsyncWorkflowConfigEncoding  string\n\t}\n\n\t// HistoryBranchRange blob in a serialization agnostic format\n\tHistoryBranchRange struct {\n\t\tBranchID    string\n\t\tBeginNodeID int64\n\t\tEndNodeID   int64\n\t}\n\n\t// HistoryTreeInfo blob in a serialization agnostic format\n\tHistoryTreeInfo struct {\n\t\tCreatedTimestamp time.Time\n\t\tAncestors        []*types.HistoryBranchRange\n\t\tInfo             string\n\t}\n\n\t// WorkflowExecutionInfo blob in a serialization agnostic format\n\tWorkflowExecutionInfo struct {\n\t\tParentDomainID                       UUID\n\t\tParentWorkflowID                     string\n\t\tParentRunID                          UUID\n\t\tInitiatedID                          int64\n\t\tCompletionEventBatchID               *int64 // TODO: This is not updated because of backward compatibility issue. Should revisit it later.\n\t\tCompletionEvent                      []byte\n\t\tCompletionEventEncoding              string\n\t\tTaskList                             string\n\t\tTaskListKind                         types.TaskListKind\n\t\tIsCron                               bool\n\t\tWorkflowTypeName                     string\n\t\tWorkflowTimeout                      time.Duration\n\t\tDecisionTaskTimeout                  time.Duration\n\t\tExecutionContext                     []byte\n\t\tState                                int32\n\t\tCloseStatus                          int32\n\t\tStartVersion                         int64\n\t\tLastWriteEventID                     *int64 // TODO: We have logic checking if LastWriteEventID != nil. The field seems to be deprecated. Should revisit it later.\n\t\tLastEventTaskID                      int64\n\t\tLastFirstEventID                     int64\n\t\tLastProcessedEvent                   int64\n\t\tStartTimestamp                       time.Time\n\t\tLastUpdatedTimestamp                 time.Time\n\t\tDecisionVersion                      int64\n\t\tDecisionScheduleID                   int64\n\t\tDecisionStartedID                    int64\n\t\tDecisionTimeout                      time.Duration\n\t\tDecisionAttempt                      int64\n\t\tDecisionStartedTimestamp             time.Time\n\t\tDecisionScheduledTimestamp           time.Time\n\t\tCancelRequested                      bool\n\t\tDecisionOriginalScheduledTimestamp   time.Time\n\t\tCreateRequestID                      string\n\t\tDecisionRequestID                    string\n\t\tCancelRequestID                      string\n\t\tStickyTaskList                       string\n\t\tStickyScheduleToStartTimeout         time.Duration\n\t\tRetryAttempt                         int64\n\t\tRetryInitialInterval                 time.Duration\n\t\tRetryMaximumInterval                 time.Duration\n\t\tRetryMaximumAttempts                 int32\n\t\tRetryExpiration                      time.Duration\n\t\tRetryBackoffCoefficient              float64\n\t\tRetryExpirationTimestamp             time.Time\n\t\tRetryNonRetryableErrors              []string\n\t\tHasRetryPolicy                       bool\n\t\tCronSchedule                         string\n\t\tCronOverlapPolicy                    types.CronOverlapPolicy\n\t\tEventStoreVersion                    int32\n\t\tEventBranchToken                     []byte\n\t\tSignalCount                          int64\n\t\tHistorySize                          int64\n\t\tClientLibraryVersion                 string\n\t\tClientFeatureVersion                 string\n\t\tClientImpl                           string\n\t\tAutoResetPoints                      []byte\n\t\tAutoResetPointsEncoding              string\n\t\tSearchAttributes                     map[string][]byte\n\t\tMemo                                 map[string][]byte\n\t\tVersionHistories                     []byte\n\t\tVersionHistoriesEncoding             string\n\t\tFirstExecutionRunID                  UUID\n\t\tPartitionConfig                      map[string]string\n\t\tChecksum                             []byte\n\t\tChecksumEncoding                     string\n\t\tActiveClusterSelectionPolicy         []byte\n\t\tActiveClusterSelectionPolicyEncoding string\n\t}\n\n\t// ActivityInfo blob in a serialization agnostic format\n\tActivityInfo struct {\n\t\tVersion                  int64\n\t\tScheduledEventBatchID    int64\n\t\tScheduledEvent           []byte\n\t\tScheduledEventEncoding   string\n\t\tScheduledTimestamp       time.Time\n\t\tStartedID                int64\n\t\tStartedEvent             []byte\n\t\tStartedEventEncoding     string\n\t\tStartedTimestamp         time.Time\n\t\tActivityID               string\n\t\tRequestID                string\n\t\tScheduleToStartTimeout   time.Duration\n\t\tScheduleToCloseTimeout   time.Duration\n\t\tStartToCloseTimeout      time.Duration\n\t\tHeartbeatTimeout         time.Duration\n\t\tCancelRequested          bool\n\t\tCancelRequestID          int64\n\t\tTimerTaskStatus          int32\n\t\tAttempt                  int32\n\t\tTaskList                 string\n\t\tTaskListKind             types.TaskListKind\n\t\tStartedIdentity          string\n\t\tHasRetryPolicy           bool\n\t\tRetryInitialInterval     time.Duration\n\t\tRetryMaximumInterval     time.Duration\n\t\tRetryMaximumAttempts     int32\n\t\tRetryExpirationTimestamp time.Time\n\t\tRetryBackoffCoefficient  float64\n\t\tRetryNonRetryableErrors  []string\n\t\tRetryLastFailureReason   string\n\t\tRetryLastWorkerIdentity  string\n\t\tRetryLastFailureDetails  []byte\n\t}\n\n\t// ChildExecutionInfo blob in a serialization agnostic format\n\tChildExecutionInfo struct {\n\t\tVersion                int64\n\t\tInitiatedEventBatchID  int64\n\t\tStartedID              int64\n\t\tInitiatedEvent         []byte\n\t\tInitiatedEventEncoding string\n\t\tStartedWorkflowID      string\n\t\tStartedRunID           UUID\n\t\tStartedEvent           []byte\n\t\tStartedEventEncoding   string\n\t\tCreateRequestID        string\n\t\tDomainID               string\n\t\tDomainNameDEPRECATED   string\n\t\tWorkflowTypeName       string\n\t\tParentClosePolicy      int32\n\t}\n\n\t// SignalInfo blob in a serialization agnostic format\n\tSignalInfo struct {\n\t\tVersion               int64\n\t\tInitiatedEventBatchID int64\n\t\tRequestID             string\n\t\tName                  string\n\t\tInput                 []byte\n\t\tControl               []byte\n\t}\n\n\t// RequestCancelInfo blob in a serialization agnostic format\n\tRequestCancelInfo struct {\n\t\tVersion               int64\n\t\tInitiatedEventBatchID int64\n\t\tCancelRequestID       string\n\t}\n\n\t// TimerInfo blob in a serialization agnostic format\n\tTimerInfo struct {\n\t\tVersion         int64\n\t\tStartedID       int64\n\t\tExpiryTimestamp time.Time\n\t\tTaskID          int64\n\t}\n\n\t// TaskInfo blob in a serialization agnostic format\n\tTaskInfo struct {\n\t\tWorkflowID       string\n\t\tRunID            UUID\n\t\tScheduleID       int64\n\t\tExpiryTimestamp  time.Time\n\t\tCreatedTimestamp time.Time\n\t\tPartitionConfig  map[string]string\n\t}\n\n\tTaskListPartition struct {\n\t\tIsolationGroups []string\n\t}\n\n\tTaskListPartitionConfig struct {\n\t\tVersion            int64\n\t\tNumReadPartitions  int32\n\t\tNumWritePartitions int32\n\t\tReadPartitions     map[int32]*TaskListPartition\n\t\tWritePartitions    map[int32]*TaskListPartition\n\t}\n\t// TaskListInfo blob in a serialization agnostic format\n\tTaskListInfo struct {\n\t\tKind                    int16\n\t\tAckLevel                int64\n\t\tExpiryTimestamp         time.Time\n\t\tLastUpdated             time.Time\n\t\tAdaptivePartitionConfig *TaskListPartitionConfig\n\t}\n\n\t// TransferTaskInfo blob in a serialization agnostic format\n\tTransferTaskInfo struct {\n\t\tDomainID                UUID\n\t\tWorkflowID              string\n\t\tRunID                   UUID\n\t\tTaskType                int16\n\t\tTargetDomainID          UUID\n\t\tTargetDomainIDs         []UUID\n\t\tTargetWorkflowID        string\n\t\tTargetRunID             UUID\n\t\tTaskList                string\n\t\tTargetChildWorkflowOnly bool\n\t\tScheduleID              int64\n\t\tVersion                 int64\n\t\tVisibilityTimestamp     time.Time\n\t\tOriginalTaskList        string\n\t\tOriginalTaskListKind    types.TaskListKind\n\t}\n\n\t// CrossClusterTaskInfo blob in a serialization agnostic format\n\t// Cross cluster tasks are exactly like transfer tasks so\n\t// instead of creating another struct and duplicating the same\n\t// logic everywhere. We reuse TransferTaskInfo\n\tCrossClusterTaskInfo         = TransferTaskInfo\n\tsqlblobsCrossClusterTaskInfo = sqlblobs.TransferTaskInfo\n\n\t// TimerTaskInfo blob in a serialization agnostic format\n\tTimerTaskInfo struct {\n\t\tDomainID        UUID\n\t\tWorkflowID      string\n\t\tRunID           UUID\n\t\tTaskType        int16\n\t\tTimeoutType     *int16 // TODO: The default value for TimeoutType doesn't make sense. No equivalent value for nil.\n\t\tVersion         int64\n\t\tScheduleAttempt int64\n\t\tEventID         int64\n\t\tTaskList        string\n\t}\n\n\t// ReplicationTaskInfo blob in a serialization agnostic format\n\tReplicationTaskInfo struct {\n\t\tDomainID                UUID\n\t\tWorkflowID              string\n\t\tRunID                   UUID\n\t\tTaskType                int16\n\t\tVersion                 int64\n\t\tFirstEventID            int64\n\t\tNextEventID             int64\n\t\tScheduledID             int64\n\t\tEventStoreVersion       int32\n\t\tNewRunEventStoreVersion int32\n\t\tBranchToken             []byte\n\t\tNewRunBranchToken       []byte\n\t\tCreationTimestamp       time.Time\n\t}\n)\n\ntype (\n\t// Parser is used to do serialization and deserialization. A parser is backed by a\n\t// a single encoder which encodes into one format and a collection of decoders.\n\t// Parser selects the appropriate decoder for the provided blob.\n\tParser interface {\n\t\tShardInfoToBlob(*ShardInfo) (persistence.DataBlob, error)\n\t\tDomainInfoToBlob(*DomainInfo) (persistence.DataBlob, error)\n\t\tHistoryTreeInfoToBlob(*HistoryTreeInfo) (persistence.DataBlob, error)\n\t\tWorkflowExecutionInfoToBlob(*WorkflowExecutionInfo) (persistence.DataBlob, error)\n\t\tActivityInfoToBlob(*ActivityInfo) (persistence.DataBlob, error)\n\t\tChildExecutionInfoToBlob(*ChildExecutionInfo) (persistence.DataBlob, error)\n\t\tSignalInfoToBlob(*SignalInfo) (persistence.DataBlob, error)\n\t\tRequestCancelInfoToBlob(*RequestCancelInfo) (persistence.DataBlob, error)\n\t\tTimerInfoToBlob(*TimerInfo) (persistence.DataBlob, error)\n\t\tTaskInfoToBlob(*TaskInfo) (persistence.DataBlob, error)\n\t\tTaskListInfoToBlob(*TaskListInfo) (persistence.DataBlob, error)\n\t\tTransferTaskInfoToBlob(*TransferTaskInfo) (persistence.DataBlob, error)\n\t\tCrossClusterTaskInfoToBlob(*CrossClusterTaskInfo) (persistence.DataBlob, error)\n\t\tTimerTaskInfoToBlob(*TimerTaskInfo) (persistence.DataBlob, error)\n\t\tReplicationTaskInfoToBlob(*ReplicationTaskInfo) (persistence.DataBlob, error)\n\n\t\tShardInfoFromBlob([]byte, string) (*ShardInfo, error)\n\t\tDomainInfoFromBlob([]byte, string) (*DomainInfo, error)\n\t\tHistoryTreeInfoFromBlob([]byte, string) (*HistoryTreeInfo, error)\n\t\tWorkflowExecutionInfoFromBlob([]byte, string) (*WorkflowExecutionInfo, error)\n\t\tActivityInfoFromBlob([]byte, string) (*ActivityInfo, error)\n\t\tChildExecutionInfoFromBlob([]byte, string) (*ChildExecutionInfo, error)\n\t\tSignalInfoFromBlob([]byte, string) (*SignalInfo, error)\n\t\tRequestCancelInfoFromBlob([]byte, string) (*RequestCancelInfo, error)\n\t\tTimerInfoFromBlob([]byte, string) (*TimerInfo, error)\n\t\tTaskInfoFromBlob([]byte, string) (*TaskInfo, error)\n\t\tTaskListInfoFromBlob([]byte, string) (*TaskListInfo, error)\n\t\tTransferTaskInfoFromBlob([]byte, string) (*TransferTaskInfo, error)\n\t\tCrossClusterTaskInfoFromBlob([]byte, string) (*CrossClusterTaskInfo, error)\n\t\tTimerTaskInfoFromBlob([]byte, string) (*TimerTaskInfo, error)\n\t\tReplicationTaskInfoFromBlob([]byte, string) (*ReplicationTaskInfo, error)\n\t}\n\n\t// encoder is used to serialize structs. Each encoder implementation uses one serialization format.\n\tencoder interface {\n\t\tshardInfoToBlob(*ShardInfo) ([]byte, error)\n\t\tdomainInfoToBlob(*DomainInfo) ([]byte, error)\n\t\thistoryTreeInfoToBlob(*HistoryTreeInfo) ([]byte, error)\n\t\tworkflowExecutionInfoToBlob(*WorkflowExecutionInfo) ([]byte, error)\n\t\tactivityInfoToBlob(*ActivityInfo) ([]byte, error)\n\t\tchildExecutionInfoToBlob(*ChildExecutionInfo) ([]byte, error)\n\t\tsignalInfoToBlob(*SignalInfo) ([]byte, error)\n\t\trequestCancelInfoToBlob(*RequestCancelInfo) ([]byte, error)\n\t\ttimerInfoToBlob(*TimerInfo) ([]byte, error)\n\t\ttaskInfoToBlob(*TaskInfo) ([]byte, error)\n\t\ttaskListInfoToBlob(*TaskListInfo) ([]byte, error)\n\t\ttransferTaskInfoToBlob(*TransferTaskInfo) ([]byte, error)\n\t\tcrossClusterTaskInfoToBlob(*CrossClusterTaskInfo) ([]byte, error)\n\t\ttimerTaskInfoToBlob(*TimerTaskInfo) ([]byte, error)\n\t\treplicationTaskInfoToBlob(*ReplicationTaskInfo) ([]byte, error)\n\t\tencodingType() constants.EncodingType\n\t}\n\n\t// decoder is used to deserialize structs. Each decoder implementation uses one serialization format.\n\tdecoder interface {\n\t\tshardInfoFromBlob([]byte) (*ShardInfo, error)\n\t\tdomainInfoFromBlob([]byte) (*DomainInfo, error)\n\t\thistoryTreeInfoFromBlob([]byte) (*HistoryTreeInfo, error)\n\t\tworkflowExecutionInfoFromBlob([]byte) (*WorkflowExecutionInfo, error)\n\t\tactivityInfoFromBlob([]byte) (*ActivityInfo, error)\n\t\tchildExecutionInfoFromBlob([]byte) (*ChildExecutionInfo, error)\n\t\tsignalInfoFromBlob([]byte) (*SignalInfo, error)\n\t\trequestCancelInfoFromBlob([]byte) (*RequestCancelInfo, error)\n\t\ttimerInfoFromBlob([]byte) (*TimerInfo, error)\n\t\ttaskInfoFromBlob([]byte) (*TaskInfo, error)\n\t\ttaskListInfoFromBlob([]byte) (*TaskListInfo, error)\n\t\ttransferTaskInfoFromBlob([]byte) (*TransferTaskInfo, error)\n\t\tcrossClusterTaskInfoFromBlob([]byte) (*CrossClusterTaskInfo, error)\n\t\ttimerTaskInfoFromBlob([]byte) (*TimerTaskInfo, error)\n\t\treplicationTaskInfoFromBlob([]byte) (*ReplicationTaskInfo, error)\n\t}\n\n\tthriftRWType interface {\n\t\tToWire() (wire.Value, error)\n\t\tFromWire(w wire.Value) error\n\t\tEncode(stream.Writer) error\n\t\tDecode(stream.Reader) error\n\t}\n)\n"
  },
  {
    "path": "common/persistence/serialization/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package serialization -source interfaces.go -destination interfaces_mock.go -self_package github.com/uber/cadence/common/persistence/serialization\n//\n\n// Package serialization is a generated GoMock package.\npackage serialization\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tstream \"go.uber.org/thriftrw/protocol/stream\"\n\twire \"go.uber.org/thriftrw/wire\"\n\n\tconstants \"github.com/uber/cadence/common/constants\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockParser is a mock of Parser interface.\ntype MockParser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockParserMockRecorder\n\tisgomock struct{}\n}\n\n// MockParserMockRecorder is the mock recorder for MockParser.\ntype MockParserMockRecorder struct {\n\tmock *MockParser\n}\n\n// NewMockParser creates a new mock instance.\nfunc NewMockParser(ctrl *gomock.Controller) *MockParser {\n\tmock := &MockParser{ctrl: ctrl}\n\tmock.recorder = &MockParserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockParser) EXPECT() *MockParserMockRecorder {\n\treturn m.recorder\n}\n\n// ActivityInfoFromBlob mocks base method.\nfunc (m *MockParser) ActivityInfoFromBlob(arg0 []byte, arg1 string) (*ActivityInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ActivityInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*ActivityInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ActivityInfoFromBlob indicates an expected call of ActivityInfoFromBlob.\nfunc (mr *MockParserMockRecorder) ActivityInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ActivityInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).ActivityInfoFromBlob), arg0, arg1)\n}\n\n// ActivityInfoToBlob mocks base method.\nfunc (m *MockParser) ActivityInfoToBlob(arg0 *ActivityInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ActivityInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ActivityInfoToBlob indicates an expected call of ActivityInfoToBlob.\nfunc (mr *MockParserMockRecorder) ActivityInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ActivityInfoToBlob\", reflect.TypeOf((*MockParser)(nil).ActivityInfoToBlob), arg0)\n}\n\n// ChildExecutionInfoFromBlob mocks base method.\nfunc (m *MockParser) ChildExecutionInfoFromBlob(arg0 []byte, arg1 string) (*ChildExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ChildExecutionInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*ChildExecutionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ChildExecutionInfoFromBlob indicates an expected call of ChildExecutionInfoFromBlob.\nfunc (mr *MockParserMockRecorder) ChildExecutionInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ChildExecutionInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).ChildExecutionInfoFromBlob), arg0, arg1)\n}\n\n// ChildExecutionInfoToBlob mocks base method.\nfunc (m *MockParser) ChildExecutionInfoToBlob(arg0 *ChildExecutionInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ChildExecutionInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ChildExecutionInfoToBlob indicates an expected call of ChildExecutionInfoToBlob.\nfunc (mr *MockParserMockRecorder) ChildExecutionInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ChildExecutionInfoToBlob\", reflect.TypeOf((*MockParser)(nil).ChildExecutionInfoToBlob), arg0)\n}\n\n// CrossClusterTaskInfoFromBlob mocks base method.\nfunc (m *MockParser) CrossClusterTaskInfoFromBlob(arg0 []byte, arg1 string) (*CrossClusterTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CrossClusterTaskInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*CrossClusterTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CrossClusterTaskInfoFromBlob indicates an expected call of CrossClusterTaskInfoFromBlob.\nfunc (mr *MockParserMockRecorder) CrossClusterTaskInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CrossClusterTaskInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).CrossClusterTaskInfoFromBlob), arg0, arg1)\n}\n\n// CrossClusterTaskInfoToBlob mocks base method.\nfunc (m *MockParser) CrossClusterTaskInfoToBlob(arg0 *CrossClusterTaskInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CrossClusterTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CrossClusterTaskInfoToBlob indicates an expected call of CrossClusterTaskInfoToBlob.\nfunc (mr *MockParserMockRecorder) CrossClusterTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CrossClusterTaskInfoToBlob\", reflect.TypeOf((*MockParser)(nil).CrossClusterTaskInfoToBlob), arg0)\n}\n\n// DomainInfoFromBlob mocks base method.\nfunc (m *MockParser) DomainInfoFromBlob(arg0 []byte, arg1 string) (*DomainInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DomainInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*DomainInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DomainInfoFromBlob indicates an expected call of DomainInfoFromBlob.\nfunc (mr *MockParserMockRecorder) DomainInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DomainInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).DomainInfoFromBlob), arg0, arg1)\n}\n\n// DomainInfoToBlob mocks base method.\nfunc (m *MockParser) DomainInfoToBlob(arg0 *DomainInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DomainInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DomainInfoToBlob indicates an expected call of DomainInfoToBlob.\nfunc (mr *MockParserMockRecorder) DomainInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DomainInfoToBlob\", reflect.TypeOf((*MockParser)(nil).DomainInfoToBlob), arg0)\n}\n\n// HistoryTreeInfoFromBlob mocks base method.\nfunc (m *MockParser) HistoryTreeInfoFromBlob(arg0 []byte, arg1 string) (*HistoryTreeInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HistoryTreeInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*HistoryTreeInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// HistoryTreeInfoFromBlob indicates an expected call of HistoryTreeInfoFromBlob.\nfunc (mr *MockParserMockRecorder) HistoryTreeInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HistoryTreeInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).HistoryTreeInfoFromBlob), arg0, arg1)\n}\n\n// HistoryTreeInfoToBlob mocks base method.\nfunc (m *MockParser) HistoryTreeInfoToBlob(arg0 *HistoryTreeInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HistoryTreeInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// HistoryTreeInfoToBlob indicates an expected call of HistoryTreeInfoToBlob.\nfunc (mr *MockParserMockRecorder) HistoryTreeInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HistoryTreeInfoToBlob\", reflect.TypeOf((*MockParser)(nil).HistoryTreeInfoToBlob), arg0)\n}\n\n// ReplicationTaskInfoFromBlob mocks base method.\nfunc (m *MockParser) ReplicationTaskInfoFromBlob(arg0 []byte, arg1 string) (*ReplicationTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicationTaskInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*ReplicationTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicationTaskInfoFromBlob indicates an expected call of ReplicationTaskInfoFromBlob.\nfunc (mr *MockParserMockRecorder) ReplicationTaskInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicationTaskInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).ReplicationTaskInfoFromBlob), arg0, arg1)\n}\n\n// ReplicationTaskInfoToBlob mocks base method.\nfunc (m *MockParser) ReplicationTaskInfoToBlob(arg0 *ReplicationTaskInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicationTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicationTaskInfoToBlob indicates an expected call of ReplicationTaskInfoToBlob.\nfunc (mr *MockParserMockRecorder) ReplicationTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicationTaskInfoToBlob\", reflect.TypeOf((*MockParser)(nil).ReplicationTaskInfoToBlob), arg0)\n}\n\n// RequestCancelInfoFromBlob mocks base method.\nfunc (m *MockParser) RequestCancelInfoFromBlob(arg0 []byte, arg1 string) (*RequestCancelInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestCancelInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*RequestCancelInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RequestCancelInfoFromBlob indicates an expected call of RequestCancelInfoFromBlob.\nfunc (mr *MockParserMockRecorder) RequestCancelInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).RequestCancelInfoFromBlob), arg0, arg1)\n}\n\n// RequestCancelInfoToBlob mocks base method.\nfunc (m *MockParser) RequestCancelInfoToBlob(arg0 *RequestCancelInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestCancelInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RequestCancelInfoToBlob indicates an expected call of RequestCancelInfoToBlob.\nfunc (mr *MockParserMockRecorder) RequestCancelInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelInfoToBlob\", reflect.TypeOf((*MockParser)(nil).RequestCancelInfoToBlob), arg0)\n}\n\n// ShardInfoFromBlob mocks base method.\nfunc (m *MockParser) ShardInfoFromBlob(arg0 []byte, arg1 string) (*ShardInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ShardInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*ShardInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ShardInfoFromBlob indicates an expected call of ShardInfoFromBlob.\nfunc (mr *MockParserMockRecorder) ShardInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ShardInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).ShardInfoFromBlob), arg0, arg1)\n}\n\n// ShardInfoToBlob mocks base method.\nfunc (m *MockParser) ShardInfoToBlob(arg0 *ShardInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ShardInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ShardInfoToBlob indicates an expected call of ShardInfoToBlob.\nfunc (mr *MockParserMockRecorder) ShardInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ShardInfoToBlob\", reflect.TypeOf((*MockParser)(nil).ShardInfoToBlob), arg0)\n}\n\n// SignalInfoFromBlob mocks base method.\nfunc (m *MockParser) SignalInfoFromBlob(arg0 []byte, arg1 string) (*SignalInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*SignalInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalInfoFromBlob indicates an expected call of SignalInfoFromBlob.\nfunc (mr *MockParserMockRecorder) SignalInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).SignalInfoFromBlob), arg0, arg1)\n}\n\n// SignalInfoToBlob mocks base method.\nfunc (m *MockParser) SignalInfoToBlob(arg0 *SignalInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalInfoToBlob indicates an expected call of SignalInfoToBlob.\nfunc (mr *MockParserMockRecorder) SignalInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalInfoToBlob\", reflect.TypeOf((*MockParser)(nil).SignalInfoToBlob), arg0)\n}\n\n// TaskInfoFromBlob mocks base method.\nfunc (m *MockParser) TaskInfoFromBlob(arg0 []byte, arg1 string) (*TaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*TaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TaskInfoFromBlob indicates an expected call of TaskInfoFromBlob.\nfunc (mr *MockParserMockRecorder) TaskInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).TaskInfoFromBlob), arg0, arg1)\n}\n\n// TaskInfoToBlob mocks base method.\nfunc (m *MockParser) TaskInfoToBlob(arg0 *TaskInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TaskInfoToBlob indicates an expected call of TaskInfoToBlob.\nfunc (mr *MockParserMockRecorder) TaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskInfoToBlob\", reflect.TypeOf((*MockParser)(nil).TaskInfoToBlob), arg0)\n}\n\n// TaskListInfoFromBlob mocks base method.\nfunc (m *MockParser) TaskListInfoFromBlob(arg0 []byte, arg1 string) (*TaskListInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskListInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*TaskListInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TaskListInfoFromBlob indicates an expected call of TaskListInfoFromBlob.\nfunc (mr *MockParserMockRecorder) TaskListInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskListInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).TaskListInfoFromBlob), arg0, arg1)\n}\n\n// TaskListInfoToBlob mocks base method.\nfunc (m *MockParser) TaskListInfoToBlob(arg0 *TaskListInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskListInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TaskListInfoToBlob indicates an expected call of TaskListInfoToBlob.\nfunc (mr *MockParserMockRecorder) TaskListInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskListInfoToBlob\", reflect.TypeOf((*MockParser)(nil).TaskListInfoToBlob), arg0)\n}\n\n// TimerInfoFromBlob mocks base method.\nfunc (m *MockParser) TimerInfoFromBlob(arg0 []byte, arg1 string) (*TimerInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TimerInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*TimerInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TimerInfoFromBlob indicates an expected call of TimerInfoFromBlob.\nfunc (mr *MockParserMockRecorder) TimerInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TimerInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).TimerInfoFromBlob), arg0, arg1)\n}\n\n// TimerInfoToBlob mocks base method.\nfunc (m *MockParser) TimerInfoToBlob(arg0 *TimerInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TimerInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TimerInfoToBlob indicates an expected call of TimerInfoToBlob.\nfunc (mr *MockParserMockRecorder) TimerInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TimerInfoToBlob\", reflect.TypeOf((*MockParser)(nil).TimerInfoToBlob), arg0)\n}\n\n// TimerTaskInfoFromBlob mocks base method.\nfunc (m *MockParser) TimerTaskInfoFromBlob(arg0 []byte, arg1 string) (*TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TimerTaskInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TimerTaskInfoFromBlob indicates an expected call of TimerTaskInfoFromBlob.\nfunc (mr *MockParserMockRecorder) TimerTaskInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TimerTaskInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).TimerTaskInfoFromBlob), arg0, arg1)\n}\n\n// TimerTaskInfoToBlob mocks base method.\nfunc (m *MockParser) TimerTaskInfoToBlob(arg0 *TimerTaskInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TimerTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TimerTaskInfoToBlob indicates an expected call of TimerTaskInfoToBlob.\nfunc (mr *MockParserMockRecorder) TimerTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TimerTaskInfoToBlob\", reflect.TypeOf((*MockParser)(nil).TimerTaskInfoToBlob), arg0)\n}\n\n// TransferTaskInfoFromBlob mocks base method.\nfunc (m *MockParser) TransferTaskInfoFromBlob(arg0 []byte, arg1 string) (*TransferTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TransferTaskInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*TransferTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TransferTaskInfoFromBlob indicates an expected call of TransferTaskInfoFromBlob.\nfunc (mr *MockParserMockRecorder) TransferTaskInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TransferTaskInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).TransferTaskInfoFromBlob), arg0, arg1)\n}\n\n// TransferTaskInfoToBlob mocks base method.\nfunc (m *MockParser) TransferTaskInfoToBlob(arg0 *TransferTaskInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TransferTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TransferTaskInfoToBlob indicates an expected call of TransferTaskInfoToBlob.\nfunc (mr *MockParserMockRecorder) TransferTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TransferTaskInfoToBlob\", reflect.TypeOf((*MockParser)(nil).TransferTaskInfoToBlob), arg0)\n}\n\n// WorkflowExecutionInfoFromBlob mocks base method.\nfunc (m *MockParser) WorkflowExecutionInfoFromBlob(arg0 []byte, arg1 string) (*WorkflowExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WorkflowExecutionInfoFromBlob\", arg0, arg1)\n\tret0, _ := ret[0].(*WorkflowExecutionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WorkflowExecutionInfoFromBlob indicates an expected call of WorkflowExecutionInfoFromBlob.\nfunc (mr *MockParserMockRecorder) WorkflowExecutionInfoFromBlob(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WorkflowExecutionInfoFromBlob\", reflect.TypeOf((*MockParser)(nil).WorkflowExecutionInfoFromBlob), arg0, arg1)\n}\n\n// WorkflowExecutionInfoToBlob mocks base method.\nfunc (m *MockParser) WorkflowExecutionInfoToBlob(arg0 *WorkflowExecutionInfo) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WorkflowExecutionInfoToBlob\", arg0)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WorkflowExecutionInfoToBlob indicates an expected call of WorkflowExecutionInfoToBlob.\nfunc (mr *MockParserMockRecorder) WorkflowExecutionInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WorkflowExecutionInfoToBlob\", reflect.TypeOf((*MockParser)(nil).WorkflowExecutionInfoToBlob), arg0)\n}\n\n// Mockencoder is a mock of encoder interface.\ntype Mockencoder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockencoderMockRecorder\n\tisgomock struct{}\n}\n\n// MockencoderMockRecorder is the mock recorder for Mockencoder.\ntype MockencoderMockRecorder struct {\n\tmock *Mockencoder\n}\n\n// NewMockencoder creates a new mock instance.\nfunc NewMockencoder(ctrl *gomock.Controller) *Mockencoder {\n\tmock := &Mockencoder{ctrl: ctrl}\n\tmock.recorder = &MockencoderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *Mockencoder) EXPECT() *MockencoderMockRecorder {\n\treturn m.recorder\n}\n\n// activityInfoToBlob mocks base method.\nfunc (m *Mockencoder) activityInfoToBlob(arg0 *ActivityInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"activityInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// activityInfoToBlob indicates an expected call of activityInfoToBlob.\nfunc (mr *MockencoderMockRecorder) activityInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"activityInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).activityInfoToBlob), arg0)\n}\n\n// childExecutionInfoToBlob mocks base method.\nfunc (m *Mockencoder) childExecutionInfoToBlob(arg0 *ChildExecutionInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"childExecutionInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// childExecutionInfoToBlob indicates an expected call of childExecutionInfoToBlob.\nfunc (mr *MockencoderMockRecorder) childExecutionInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"childExecutionInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).childExecutionInfoToBlob), arg0)\n}\n\n// crossClusterTaskInfoToBlob mocks base method.\nfunc (m *Mockencoder) crossClusterTaskInfoToBlob(arg0 *CrossClusterTaskInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"crossClusterTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// crossClusterTaskInfoToBlob indicates an expected call of crossClusterTaskInfoToBlob.\nfunc (mr *MockencoderMockRecorder) crossClusterTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"crossClusterTaskInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).crossClusterTaskInfoToBlob), arg0)\n}\n\n// domainInfoToBlob mocks base method.\nfunc (m *Mockencoder) domainInfoToBlob(arg0 *DomainInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"domainInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// domainInfoToBlob indicates an expected call of domainInfoToBlob.\nfunc (mr *MockencoderMockRecorder) domainInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"domainInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).domainInfoToBlob), arg0)\n}\n\n// encodingType mocks base method.\nfunc (m *Mockencoder) encodingType() constants.EncodingType {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"encodingType\")\n\tret0, _ := ret[0].(constants.EncodingType)\n\treturn ret0\n}\n\n// encodingType indicates an expected call of encodingType.\nfunc (mr *MockencoderMockRecorder) encodingType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"encodingType\", reflect.TypeOf((*Mockencoder)(nil).encodingType))\n}\n\n// historyTreeInfoToBlob mocks base method.\nfunc (m *Mockencoder) historyTreeInfoToBlob(arg0 *HistoryTreeInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"historyTreeInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// historyTreeInfoToBlob indicates an expected call of historyTreeInfoToBlob.\nfunc (mr *MockencoderMockRecorder) historyTreeInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"historyTreeInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).historyTreeInfoToBlob), arg0)\n}\n\n// replicationTaskInfoToBlob mocks base method.\nfunc (m *Mockencoder) replicationTaskInfoToBlob(arg0 *ReplicationTaskInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"replicationTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// replicationTaskInfoToBlob indicates an expected call of replicationTaskInfoToBlob.\nfunc (mr *MockencoderMockRecorder) replicationTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"replicationTaskInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).replicationTaskInfoToBlob), arg0)\n}\n\n// requestCancelInfoToBlob mocks base method.\nfunc (m *Mockencoder) requestCancelInfoToBlob(arg0 *RequestCancelInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"requestCancelInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// requestCancelInfoToBlob indicates an expected call of requestCancelInfoToBlob.\nfunc (mr *MockencoderMockRecorder) requestCancelInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"requestCancelInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).requestCancelInfoToBlob), arg0)\n}\n\n// shardInfoToBlob mocks base method.\nfunc (m *Mockencoder) shardInfoToBlob(arg0 *ShardInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"shardInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// shardInfoToBlob indicates an expected call of shardInfoToBlob.\nfunc (mr *MockencoderMockRecorder) shardInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"shardInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).shardInfoToBlob), arg0)\n}\n\n// signalInfoToBlob mocks base method.\nfunc (m *Mockencoder) signalInfoToBlob(arg0 *SignalInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"signalInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// signalInfoToBlob indicates an expected call of signalInfoToBlob.\nfunc (mr *MockencoderMockRecorder) signalInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"signalInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).signalInfoToBlob), arg0)\n}\n\n// taskInfoToBlob mocks base method.\nfunc (m *Mockencoder) taskInfoToBlob(arg0 *TaskInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"taskInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// taskInfoToBlob indicates an expected call of taskInfoToBlob.\nfunc (mr *MockencoderMockRecorder) taskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"taskInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).taskInfoToBlob), arg0)\n}\n\n// taskListInfoToBlob mocks base method.\nfunc (m *Mockencoder) taskListInfoToBlob(arg0 *TaskListInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"taskListInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// taskListInfoToBlob indicates an expected call of taskListInfoToBlob.\nfunc (mr *MockencoderMockRecorder) taskListInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"taskListInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).taskListInfoToBlob), arg0)\n}\n\n// timerInfoToBlob mocks base method.\nfunc (m *Mockencoder) timerInfoToBlob(arg0 *TimerInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"timerInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// timerInfoToBlob indicates an expected call of timerInfoToBlob.\nfunc (mr *MockencoderMockRecorder) timerInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"timerInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).timerInfoToBlob), arg0)\n}\n\n// timerTaskInfoToBlob mocks base method.\nfunc (m *Mockencoder) timerTaskInfoToBlob(arg0 *TimerTaskInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"timerTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// timerTaskInfoToBlob indicates an expected call of timerTaskInfoToBlob.\nfunc (mr *MockencoderMockRecorder) timerTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"timerTaskInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).timerTaskInfoToBlob), arg0)\n}\n\n// transferTaskInfoToBlob mocks base method.\nfunc (m *Mockencoder) transferTaskInfoToBlob(arg0 *TransferTaskInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"transferTaskInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// transferTaskInfoToBlob indicates an expected call of transferTaskInfoToBlob.\nfunc (mr *MockencoderMockRecorder) transferTaskInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"transferTaskInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).transferTaskInfoToBlob), arg0)\n}\n\n// workflowExecutionInfoToBlob mocks base method.\nfunc (m *Mockencoder) workflowExecutionInfoToBlob(arg0 *WorkflowExecutionInfo) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"workflowExecutionInfoToBlob\", arg0)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// workflowExecutionInfoToBlob indicates an expected call of workflowExecutionInfoToBlob.\nfunc (mr *MockencoderMockRecorder) workflowExecutionInfoToBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"workflowExecutionInfoToBlob\", reflect.TypeOf((*Mockencoder)(nil).workflowExecutionInfoToBlob), arg0)\n}\n\n// Mockdecoder is a mock of decoder interface.\ntype Mockdecoder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockdecoderMockRecorder\n\tisgomock struct{}\n}\n\n// MockdecoderMockRecorder is the mock recorder for Mockdecoder.\ntype MockdecoderMockRecorder struct {\n\tmock *Mockdecoder\n}\n\n// NewMockdecoder creates a new mock instance.\nfunc NewMockdecoder(ctrl *gomock.Controller) *Mockdecoder {\n\tmock := &Mockdecoder{ctrl: ctrl}\n\tmock.recorder = &MockdecoderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *Mockdecoder) EXPECT() *MockdecoderMockRecorder {\n\treturn m.recorder\n}\n\n// activityInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) activityInfoFromBlob(arg0 []byte) (*ActivityInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"activityInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*ActivityInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// activityInfoFromBlob indicates an expected call of activityInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) activityInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"activityInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).activityInfoFromBlob), arg0)\n}\n\n// childExecutionInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) childExecutionInfoFromBlob(arg0 []byte) (*ChildExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"childExecutionInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*ChildExecutionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// childExecutionInfoFromBlob indicates an expected call of childExecutionInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) childExecutionInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"childExecutionInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).childExecutionInfoFromBlob), arg0)\n}\n\n// crossClusterTaskInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) crossClusterTaskInfoFromBlob(arg0 []byte) (*CrossClusterTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"crossClusterTaskInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*CrossClusterTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// crossClusterTaskInfoFromBlob indicates an expected call of crossClusterTaskInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) crossClusterTaskInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"crossClusterTaskInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).crossClusterTaskInfoFromBlob), arg0)\n}\n\n// domainInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) domainInfoFromBlob(arg0 []byte) (*DomainInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"domainInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*DomainInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// domainInfoFromBlob indicates an expected call of domainInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) domainInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"domainInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).domainInfoFromBlob), arg0)\n}\n\n// historyTreeInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) historyTreeInfoFromBlob(arg0 []byte) (*HistoryTreeInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"historyTreeInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*HistoryTreeInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// historyTreeInfoFromBlob indicates an expected call of historyTreeInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) historyTreeInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"historyTreeInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).historyTreeInfoFromBlob), arg0)\n}\n\n// replicationTaskInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) replicationTaskInfoFromBlob(arg0 []byte) (*ReplicationTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"replicationTaskInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*ReplicationTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// replicationTaskInfoFromBlob indicates an expected call of replicationTaskInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) replicationTaskInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"replicationTaskInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).replicationTaskInfoFromBlob), arg0)\n}\n\n// requestCancelInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) requestCancelInfoFromBlob(arg0 []byte) (*RequestCancelInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"requestCancelInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*RequestCancelInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// requestCancelInfoFromBlob indicates an expected call of requestCancelInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) requestCancelInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"requestCancelInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).requestCancelInfoFromBlob), arg0)\n}\n\n// shardInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) shardInfoFromBlob(arg0 []byte) (*ShardInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"shardInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*ShardInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// shardInfoFromBlob indicates an expected call of shardInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) shardInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"shardInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).shardInfoFromBlob), arg0)\n}\n\n// signalInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) signalInfoFromBlob(arg0 []byte) (*SignalInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"signalInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*SignalInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// signalInfoFromBlob indicates an expected call of signalInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) signalInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"signalInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).signalInfoFromBlob), arg0)\n}\n\n// taskInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) taskInfoFromBlob(arg0 []byte) (*TaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"taskInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*TaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// taskInfoFromBlob indicates an expected call of taskInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) taskInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"taskInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).taskInfoFromBlob), arg0)\n}\n\n// taskListInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) taskListInfoFromBlob(arg0 []byte) (*TaskListInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"taskListInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*TaskListInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// taskListInfoFromBlob indicates an expected call of taskListInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) taskListInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"taskListInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).taskListInfoFromBlob), arg0)\n}\n\n// timerInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) timerInfoFromBlob(arg0 []byte) (*TimerInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"timerInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*TimerInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// timerInfoFromBlob indicates an expected call of timerInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) timerInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"timerInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).timerInfoFromBlob), arg0)\n}\n\n// timerTaskInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) timerTaskInfoFromBlob(arg0 []byte) (*TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"timerTaskInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// timerTaskInfoFromBlob indicates an expected call of timerTaskInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) timerTaskInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"timerTaskInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).timerTaskInfoFromBlob), arg0)\n}\n\n// transferTaskInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) transferTaskInfoFromBlob(arg0 []byte) (*TransferTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"transferTaskInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*TransferTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// transferTaskInfoFromBlob indicates an expected call of transferTaskInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) transferTaskInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"transferTaskInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).transferTaskInfoFromBlob), arg0)\n}\n\n// workflowExecutionInfoFromBlob mocks base method.\nfunc (m *Mockdecoder) workflowExecutionInfoFromBlob(arg0 []byte) (*WorkflowExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"workflowExecutionInfoFromBlob\", arg0)\n\tret0, _ := ret[0].(*WorkflowExecutionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// workflowExecutionInfoFromBlob indicates an expected call of workflowExecutionInfoFromBlob.\nfunc (mr *MockdecoderMockRecorder) workflowExecutionInfoFromBlob(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"workflowExecutionInfoFromBlob\", reflect.TypeOf((*Mockdecoder)(nil).workflowExecutionInfoFromBlob), arg0)\n}\n\n// MockthriftRWType is a mock of thriftRWType interface.\ntype MockthriftRWType struct {\n\tctrl     *gomock.Controller\n\trecorder *MockthriftRWTypeMockRecorder\n\tisgomock struct{}\n}\n\n// MockthriftRWTypeMockRecorder is the mock recorder for MockthriftRWType.\ntype MockthriftRWTypeMockRecorder struct {\n\tmock *MockthriftRWType\n}\n\n// NewMockthriftRWType creates a new mock instance.\nfunc NewMockthriftRWType(ctrl *gomock.Controller) *MockthriftRWType {\n\tmock := &MockthriftRWType{ctrl: ctrl}\n\tmock.recorder = &MockthriftRWTypeMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockthriftRWType) EXPECT() *MockthriftRWTypeMockRecorder {\n\treturn m.recorder\n}\n\n// Decode mocks base method.\nfunc (m *MockthriftRWType) Decode(arg0 stream.Reader) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Decode\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Decode indicates an expected call of Decode.\nfunc (mr *MockthriftRWTypeMockRecorder) Decode(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Decode\", reflect.TypeOf((*MockthriftRWType)(nil).Decode), arg0)\n}\n\n// Encode mocks base method.\nfunc (m *MockthriftRWType) Encode(arg0 stream.Writer) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Encode\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Encode indicates an expected call of Encode.\nfunc (mr *MockthriftRWTypeMockRecorder) Encode(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Encode\", reflect.TypeOf((*MockthriftRWType)(nil).Encode), arg0)\n}\n\n// FromWire mocks base method.\nfunc (m *MockthriftRWType) FromWire(w wire.Value) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FromWire\", w)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// FromWire indicates an expected call of FromWire.\nfunc (mr *MockthriftRWTypeMockRecorder) FromWire(w any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FromWire\", reflect.TypeOf((*MockthriftRWType)(nil).FromWire), w)\n}\n\n// ToWire mocks base method.\nfunc (m *MockthriftRWType) ToWire() (wire.Value, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToWire\")\n\tret0, _ := ret[0].(wire.Value)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToWire indicates an expected call of ToWire.\nfunc (mr *MockthriftRWTypeMockRecorder) ToWire() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToWire\", reflect.TypeOf((*MockthriftRWType)(nil).ToWire))\n}\n"
  },
  {
    "path": "common/persistence/serialization/parser.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tparser struct {\n\t\tdc       *persistence.DynamicConfiguration\n\t\tencoders map[constants.EncodingType]encoder\n\t\tdecoders map[constants.EncodingType]decoder\n\t}\n)\n\nvar allBlobEncodings = []constants.EncodingType{\n\tconstants.EncodingTypeThriftRW,\n\tconstants.EncodingTypeThriftRWSnappy,\n}\n\n// NewParser constructs a new parser using encoder as specified by encodingType and using decoders specified by decodingTypes\nfunc NewParser(dc *persistence.DynamicConfiguration) (Parser, error) {\n\tencoders := make(map[constants.EncodingType]encoder)\n\tdecoders := make(map[constants.EncodingType]decoder)\n\n\tfor _, dt := range allBlobEncodings {\n\t\tdecoder, err := getDecoder(dt)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdecoders[dt] = decoder\n\n\t\tencoder, err := getEncoder(dt)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tencoders[dt] = encoder\n\t}\n\treturn &parser{\n\t\tdc:       dc,\n\t\tencoders: encoders,\n\t\tdecoders: decoders,\n\t}, nil\n}\n\nfunc (p *parser) ShardInfoToBlob(info *ShardInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.shardInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) DomainInfoToBlob(info *DomainInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.domainInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) HistoryTreeInfoToBlob(info *HistoryTreeInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.historyTreeInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) WorkflowExecutionInfoToBlob(info *WorkflowExecutionInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.workflowExecutionInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) ActivityInfoToBlob(info *ActivityInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.activityInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) ChildExecutionInfoToBlob(info *ChildExecutionInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.childExecutionInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) SignalInfoToBlob(info *SignalInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.signalInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) RequestCancelInfoToBlob(info *RequestCancelInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.requestCancelInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) TimerInfoToBlob(info *TimerInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.timerInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) TaskInfoToBlob(info *TaskInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.taskInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) TaskListInfoToBlob(info *TaskListInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.taskListInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) TransferTaskInfoToBlob(info *TransferTaskInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.transferTaskInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) CrossClusterTaskInfoToBlob(info *CrossClusterTaskInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.crossClusterTaskInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) TimerTaskInfoToBlob(info *TimerTaskInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.timerTaskInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) ReplicationTaskInfoToBlob(info *ReplicationTaskInfo) (persistence.DataBlob, error) {\n\tdb := persistence.DataBlob{}\n\tencoding := p.dc.SerializationEncoding()\n\tencoder, err := p.getCachedEncoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn db, err\n\t}\n\n\tdata, err := encoder.replicationTaskInfoToBlob(info)\n\tif err != nil {\n\t\treturn db, err\n\t}\n\tdb.Data = data\n\tdb.Encoding = encoder.encodingType()\n\treturn db, nil\n}\n\nfunc (p *parser) ShardInfoFromBlob(data []byte, encoding string) (*ShardInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.shardInfoFromBlob(data)\n}\n\nfunc (p *parser) DomainInfoFromBlob(data []byte, encoding string) (*DomainInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.domainInfoFromBlob(data)\n}\n\nfunc (p *parser) HistoryTreeInfoFromBlob(data []byte, encoding string) (*HistoryTreeInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.historyTreeInfoFromBlob(data)\n}\n\nfunc (p *parser) WorkflowExecutionInfoFromBlob(data []byte, encoding string) (*WorkflowExecutionInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.workflowExecutionInfoFromBlob(data)\n}\n\nfunc (p *parser) ActivityInfoFromBlob(data []byte, encoding string) (*ActivityInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.activityInfoFromBlob(data)\n}\n\nfunc (p *parser) ChildExecutionInfoFromBlob(data []byte, encoding string) (*ChildExecutionInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.childExecutionInfoFromBlob(data)\n}\n\nfunc (p *parser) SignalInfoFromBlob(data []byte, encoding string) (*SignalInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.signalInfoFromBlob(data)\n}\n\nfunc (p *parser) RequestCancelInfoFromBlob(data []byte, encoding string) (*RequestCancelInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.requestCancelInfoFromBlob(data)\n}\n\nfunc (p *parser) TimerInfoFromBlob(data []byte, encoding string) (*TimerInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.timerInfoFromBlob(data)\n}\n\nfunc (p *parser) TaskInfoFromBlob(data []byte, encoding string) (*TaskInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.taskInfoFromBlob(data)\n}\n\nfunc (p *parser) TaskListInfoFromBlob(data []byte, encoding string) (*TaskListInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.taskListInfoFromBlob(data)\n}\n\nfunc (p *parser) TransferTaskInfoFromBlob(data []byte, encoding string) (*TransferTaskInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.transferTaskInfoFromBlob(data)\n}\n\nfunc (p *parser) CrossClusterTaskInfoFromBlob(data []byte, encoding string) (*CrossClusterTaskInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.crossClusterTaskInfoFromBlob(data)\n}\n\nfunc (p *parser) TimerTaskInfoFromBlob(data []byte, encoding string) (*TimerTaskInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.timerTaskInfoFromBlob(data)\n}\n\nfunc (p *parser) ReplicationTaskInfoFromBlob(data []byte, encoding string) (*ReplicationTaskInfo, error) {\n\tdecoder, err := p.getCachedDecoder(constants.EncodingType(encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn decoder.replicationTaskInfoFromBlob(data)\n}\n\nfunc (p *parser) getCachedEncoder(encoding constants.EncodingType) (encoder, error) {\n\tencoder, ok := p.encoders[encoding]\n\tif !ok {\n\t\treturn nil, unsupportedEncodingError(encoding)\n\t}\n\treturn encoder, nil\n}\n\nfunc (p *parser) getCachedDecoder(encoding constants.EncodingType) (decoder, error) {\n\tdecoder, ok := p.decoders[encoding]\n\tif !ok {\n\t\treturn nil, unsupportedEncodingError(encoding)\n\t}\n\treturn decoder, nil\n}\n\nfunc getDecoder(encoding constants.EncodingType) (decoder, error) {\n\tswitch encoding {\n\tcase constants.EncodingTypeThriftRW:\n\t\treturn newThriftDecoder(), nil\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\treturn newSnappyThriftDecoder(), nil\n\tdefault:\n\t\treturn nil, unsupportedEncodingError(encoding)\n\t}\n}\n\nfunc getEncoder(encoding constants.EncodingType) (encoder, error) {\n\tswitch encoding {\n\tcase constants.EncodingTypeThriftRW:\n\t\treturn newThriftEncoder(), nil\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\treturn newSnappyThriftEncoder(), nil\n\tdefault:\n\t\treturn nil, unsupportedEncodingError(encoding)\n\t}\n}\n\nfunc unsupportedEncodingError(encoding constants.EncodingType) error {\n\treturn fmt.Errorf(\"invalid encoding type: %v\", encoding)\n}\n"
  },
  {
    "path": "common/persistence/serialization/parser_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestParserRoundTrip(t *testing.T) {\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t}\n\tthriftParser, err := NewParser(dc)\n\tassert.NoError(t, err)\n\tnow := time.Now().Round(time.Second)\n\n\tfor _, testCase := range []any{\n\t\t&ShardInfo{\n\t\t\tStolenSinceRenew:                      1,\n\t\t\tUpdatedAt:                             now,\n\t\t\tReplicationAckLevel:                   1,\n\t\t\tTransferAckLevel:                      1,\n\t\t\tTimerAckLevel:                         now,\n\t\t\tDomainNotificationVersion:             1,\n\t\t\tClusterTransferAckLevel:               map[string]int64{\"test\": 1},\n\t\t\tClusterTimerAckLevel:                  map[string]time.Time{\"test\": now},\n\t\t\tTransferProcessingQueueStates:         []byte{1, 2, 3},\n\t\t\tTimerProcessingQueueStates:            []byte{1, 2, 3},\n\t\t\tOwner:                                 \"owner\",\n\t\t\tClusterReplicationLevel:               map[string]int64{\"test\": 1},\n\t\t\tPendingFailoverMarkers:                []byte{2, 3, 4},\n\t\t\tPendingFailoverMarkersEncoding:        \"\",\n\t\t\tTransferProcessingQueueStatesEncoding: \"\",\n\t\t\tTimerProcessingQueueStatesEncoding:    \"\",\n\t\t},\n\t\t&DomainInfo{\n\t\t\tName:                        \"test\",\n\t\t\tDescription:                 \"test_desc\",\n\t\t\tOwner:                       \"test_owner\",\n\t\t\tStatus:                      1,\n\t\t\tRetention:                   48 * time.Hour,\n\t\t\tEmitMetric:                  true,\n\t\t\tArchivalBucket:              \"test_bucket\",\n\t\t\tArchivalStatus:              1,\n\t\t\tConfigVersion:               1,\n\t\t\tFailoverVersion:             1,\n\t\t\tNotificationVersion:         1,\n\t\t\tFailoverNotificationVersion: 1,\n\t\t\tActiveClusterName:           \"test_active_cluster\",\n\t\t\tClusters:                    []string{\"test_active_cluster\", \"test_standby_cluster\"},\n\t\t\tData:                        map[string]string{\"test_key\": \"test_value\"},\n\t\t\tBadBinaries:                 []byte{1, 2, 3},\n\t\t\tBadBinariesEncoding:         \"\",\n\t\t\tHistoryArchivalStatus:       1,\n\t\t\tHistoryArchivalURI:          \"test_history_archival_uri\",\n\t\t\tVisibilityArchivalStatus:    1,\n\t\t\tVisibilityArchivalURI:       \"test_visibility_archival_uri\",\n\t\t},\n\t\t&HistoryTreeInfo{\n\t\t\tCreatedTimestamp: now,\n\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t{\n\t\t\t\t\tBranchID: \"test_branch_id1\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBranchID: \"test_branch_id2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tInfo: \"test_info\",\n\t\t},\n\t\t&WorkflowExecutionInfo{\n\t\t\tParentDomainID:                     MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tParentWorkflowID:                   \"test_parent_workflow_id\",\n\t\t\tParentRunID:                        MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tInitiatedID:                        1,\n\t\t\tCompletionEventBatchID:             common.Int64Ptr(1),\n\t\t\tCompletionEvent:                    []byte{1, 2, 3},\n\t\t\tCompletionEventEncoding:            \"\",\n\t\t\tTaskList:                           \"test_task_list\",\n\t\t\tWorkflowTypeName:                   \"test_workflow_type_name\",\n\t\t\tWorkflowTimeout:                    48 * time.Hour,\n\t\t\tExecutionContext:                   []byte{1, 2, 3},\n\t\t\tState:                              2,\n\t\t\tCloseStatus:                        1,\n\t\t\tStartVersion:                       1,\n\t\t\tLastWriteEventID:                   common.Int64Ptr(2),\n\t\t\tLastEventTaskID:                    3,\n\t\t\tLastFirstEventID:                   5,\n\t\t\tLastProcessedEvent:                 4,\n\t\t\tStartTimestamp:                     now,\n\t\t\tLastUpdatedTimestamp:               now,\n\t\t\tDecisionVersion:                    1,\n\t\t\tDecisionScheduleID:                 1,\n\t\t\tDecisionStartedID:                  1,\n\t\t\tDecisionRequestID:                  \"test_decision_request_id\",\n\t\t\tDecisionTimeout:                    48 * time.Hour,\n\t\t\tDecisionAttempt:                    1,\n\t\t\tDecisionStartedTimestamp:           now,\n\t\t\tDecisionScheduledTimestamp:         now,\n\t\t\tDecisionOriginalScheduledTimestamp: now,\n\t\t\tCancelRequested:                    true,\n\t\t\tCancelRequestID:                    \"test_cancel_request_id\",\n\t\t\tStickyTaskList:                     \"test_sticky_task_list\",\n\t\t\tStickyScheduleToStartTimeout:       48 * time.Hour,\n\t\t\tClientLibraryVersion:               \"test_client_library_version\",\n\t\t\tClientFeatureVersion:               \"test_client_feature_version\",\n\t\t\tClientImpl:                         \"test_client_impl\",\n\t\t\tSignalCount:                        1,\n\t\t\tHistorySize:                        100,\n\t\t\tAutoResetPoints:                    []byte{1, 2, 3},\n\t\t\tAutoResetPointsEncoding:            \"\",\n\t\t\tMemo:                               map[string][]byte{\"test_memo_key\": {1, 2, 3}},\n\t\t\tSearchAttributes:                   map[string][]byte{\"test_search_attr_key\": {1, 2, 3}},\n\t\t},\n\t\t&ActivityInfo{\n\t\t\tVersion:                  1,\n\t\t\tScheduledEventBatchID:    2,\n\t\t\tScheduledEvent:           []byte{1, 2, 3},\n\t\t\tScheduledEventEncoding:   \"scheduled_event_encoding\",\n\t\t\tScheduledTimestamp:       now,\n\t\t\tStartedID:                1,\n\t\t\tStartedEvent:             []byte{1, 2, 3},\n\t\t\tStartedEventEncoding:     \"started_event_encoding\",\n\t\t\tStartedTimestamp:         now,\n\t\t\tActivityID:               \"test_activity_id\",\n\t\t\tRequestID:                \"test_request_id\",\n\t\t\tScheduleToStartTimeout:   48 * time.Hour,\n\t\t\tScheduleToCloseTimeout:   48 * time.Hour,\n\t\t\tStartToCloseTimeout:      48 * time.Hour,\n\t\t\tHeartbeatTimeout:         48 * time.Hour,\n\t\t\tCancelRequested:          true,\n\t\t\tCancelRequestID:          3,\n\t\t\tTimerTaskStatus:          1,\n\t\t\tAttempt:                  1,\n\t\t\tTaskList:                 \"test_task_list\",\n\t\t\tStartedIdentity:          \"test_started_identity\",\n\t\t\tHasRetryPolicy:           true,\n\t\t\tRetryInitialInterval:     time.Hour,\n\t\t\tRetryBackoffCoefficient:  1.1,\n\t\t\tRetryMaximumInterval:     time.Hour,\n\t\t\tRetryMaximumAttempts:     1,\n\t\t\tRetryExpirationTimestamp: now.Add(time.Hour),\n\t\t\tRetryNonRetryableErrors:  []string{\"test_retry_non_retryable_error\"},\n\t\t\tRetryLastWorkerIdentity:  \"test_retry_last_worker_identity\",\n\t\t},\n\t\t&ChildExecutionInfo{\n\t\t\tVersion:                1,\n\t\t\tInitiatedEventBatchID:  2,\n\t\t\tStartedID:              3,\n\t\t\tInitiatedEvent:         []byte{1, 2, 3},\n\t\t\tInitiatedEventEncoding: \"initiated_event_encoding\",\n\t\t\tStartedWorkflowID:      \"test_started_workflow_id\",\n\t\t\tStartedRunID:           MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tStartedEvent:           []byte{1, 2, 3},\n\t\t\tStartedEventEncoding:   \"started_event_encoding\",\n\t\t\tCreateRequestID:        \"test_create_request_id\",\n\t\t\tDomainID:               \"test_domain_id\",\n\t\t\tWorkflowTypeName:       \"test_workflow_type_name\",\n\t\t\tParentClosePolicy:      1,\n\t\t},\n\t\t&SignalInfo{\n\t\t\tVersion:               1,\n\t\t\tInitiatedEventBatchID: 2,\n\t\t\tRequestID:             \"test_request_id\",\n\t\t\tName:                  \"test_name\",\n\t\t\tInput:                 []byte{1, 2, 3},\n\t\t\tControl:               []byte{1, 2, 3},\n\t\t},\n\t\t&RequestCancelInfo{\n\t\t\tVersion:               1,\n\t\t\tInitiatedEventBatchID: 2,\n\t\t\tCancelRequestID:       \"test_cancel_request_id\",\n\t\t},\n\t\t&TimerInfo{\n\t\t\tVersion:         1,\n\t\t\tStartedID:       2,\n\t\t\tExpiryTimestamp: zeroUnix,\n\t\t\tTaskID:          3,\n\t\t},\n\t\t&TaskInfo{\n\t\t\tWorkflowID:       \"test_workflow_id\",\n\t\t\tRunID:            MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tScheduleID:       1,\n\t\t\tExpiryTimestamp:  now,\n\t\t\tCreatedTimestamp: now,\n\t\t\tPartitionConfig:  map[string]string{\"test_partition_key\": \"test_partition_value\"},\n\t\t},\n\t\t&TaskListInfo{\n\t\t\tKind:            1,\n\t\t\tAckLevel:        2,\n\t\t\tExpiryTimestamp: now,\n\t\t\tLastUpdated:     now,\n\t\t},\n\t\t&TransferTaskInfo{\n\t\t\tDomainID:                MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tWorkflowID:              \"test_workflow_id\",\n\t\t\tRunID:                   MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tTaskType:                1,\n\t\t\tTargetDomainID:          MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tTargetDomainIDs:         []UUID{MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"), MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430a8\")},\n\t\t\tTargetWorkflowID:        \"test_target_workflow_id\",\n\t\t\tTargetRunID:             MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tTaskList:                \"test_task_list\",\n\t\t\tTargetChildWorkflowOnly: true,\n\t\t\tScheduleID:              1,\n\t\t\tVersion:                 2,\n\t\t\tVisibilityTimestamp:     now,\n\t\t},\n\t\t&TimerTaskInfo{\n\t\t\tDomainID:        MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tWorkflowID:      \"test_workflow_id\",\n\t\t\tRunID:           MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tTaskType:        1,\n\t\t\tTimeoutType:     common.Int16Ptr(1),\n\t\t\tVersion:         2,\n\t\t\tScheduleAttempt: 3,\n\t\t\tEventID:         4,\n\t\t},\n\t\t&ReplicationTaskInfo{\n\t\t\tDomainID:                MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tWorkflowID:              \"test_workflow_id\",\n\t\t\tRunID:                   MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\t\t\tTaskType:                1,\n\t\t\tVersion:                 2,\n\t\t\tFirstEventID:            3,\n\t\t\tNextEventID:             4,\n\t\t\tScheduledID:             5,\n\t\t\tEventStoreVersion:       6,\n\t\t\tNewRunEventStoreVersion: 7,\n\t\t\tBranchToken:             []byte{1, 2, 3},\n\t\t\tNewRunBranchToken:       []byte{1, 2, 3},\n\t\t\tCreationTimestamp:       now,\n\t\t},\n\t} {\n\t\tt.Run(reflect.TypeOf(testCase).String(), func(t *testing.T) {\n\t\t\tblob := parse(t, thriftParser, testCase)\n\t\t\tresult := unparse(t, thriftParser, blob, testCase)\n\t\t\tassert.Equal(t, testCase, result)\n\t\t})\n\t}\n}\n\nfunc parse(t *testing.T, parser Parser, data any) persistence.DataBlob {\n\tvar (\n\t\tblob persistence.DataBlob\n\t\terr  error\n\t)\n\tswitch v := data.(type) {\n\tcase *ShardInfo:\n\t\tblob, err = parser.ShardInfoToBlob(v)\n\tcase *DomainInfo:\n\t\tblob, err = parser.DomainInfoToBlob(v)\n\tcase *HistoryTreeInfo:\n\t\tblob, err = parser.HistoryTreeInfoToBlob(v)\n\tcase *WorkflowExecutionInfo:\n\t\tblob, err = parser.WorkflowExecutionInfoToBlob(v)\n\tcase *ActivityInfo:\n\t\tblob, err = parser.ActivityInfoToBlob(v)\n\tcase *ChildExecutionInfo:\n\t\tblob, err = parser.ChildExecutionInfoToBlob(v)\n\tcase *SignalInfo:\n\t\tblob, err = parser.SignalInfoToBlob(v)\n\tcase *RequestCancelInfo:\n\t\tblob, err = parser.RequestCancelInfoToBlob(v)\n\tcase *TimerInfo:\n\t\tblob, err = parser.TimerInfoToBlob(v)\n\tcase *TaskInfo:\n\t\tblob, err = parser.TaskInfoToBlob(v)\n\tcase *TaskListInfo:\n\t\tblob, err = parser.TaskListInfoToBlob(v)\n\tcase *TransferTaskInfo:\n\t\tblob, err = parser.TransferTaskInfoToBlob(v)\n\tcase *TimerTaskInfo:\n\t\tblob, err = parser.TimerTaskInfoToBlob(v)\n\tcase *ReplicationTaskInfo:\n\t\tblob, err = parser.ReplicationTaskInfoToBlob(v)\n\tdefault:\n\t\terr = fmt.Errorf(\"unknown type %T\", v)\n\t}\n\tassert.NoError(t, err)\n\treturn blob\n}\n\nfunc unparse(t *testing.T, parser Parser, blob persistence.DataBlob, result any) any {\n\tvar (\n\t\tdata any\n\t\terr  error\n\t)\n\tswitch v := result.(type) {\n\tcase *ShardInfo:\n\t\tdata, err = parser.ShardInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *DomainInfo:\n\t\tdata, err = parser.DomainInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *HistoryTreeInfo:\n\t\tdata, err = parser.HistoryTreeInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *WorkflowExecutionInfo:\n\t\tdata, err = parser.WorkflowExecutionInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *ActivityInfo:\n\t\tdata, err = parser.ActivityInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *ChildExecutionInfo:\n\t\tdata, err = parser.ChildExecutionInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *SignalInfo:\n\t\tdata, err = parser.SignalInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *RequestCancelInfo:\n\t\tdata, err = parser.RequestCancelInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *TimerInfo:\n\t\tdata, err = parser.TimerInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *TaskInfo:\n\t\tdata, err = parser.TaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *TaskListInfo:\n\t\tdata, err = parser.TaskListInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *TransferTaskInfo:\n\t\tdata, err = parser.TransferTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *TimerTaskInfo:\n\t\tdata, err = parser.TimerTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tcase *ReplicationTaskInfo:\n\t\tdata, err = parser.ReplicationTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tdefault:\n\t\terr = fmt.Errorf(\"unknown type %T\", v)\n\t}\n\tassert.NoError(t, err)\n\treturn data\n}\n\nfunc TestParser_WorkflowExecution_with_cron(t *testing.T) {\n\tinfo := &WorkflowExecutionInfo{\n\t\tCronSchedule: \"@every 1m\",\n\t\tIsCron:       true,\n\t}\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t}\n\tparser, err := NewParser(dc)\n\trequire.NoError(t, err)\n\tblob, err := parser.WorkflowExecutionInfoToBlob(info)\n\trequire.NoError(t, err)\n\tresult, err := parser.WorkflowExecutionInfoFromBlob(blob.Data, string(blob.Encoding))\n\trequire.NoError(t, err)\n\tassert.Equal(t, info, result)\n}\n"
  },
  {
    "path": "common/persistence/serialization/persistence_mapper.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc ToInternalWorkflowExecutionInfo(info *WorkflowExecutionInfo) *persistence.InternalWorkflowExecutionInfo {\n\tresult := &persistence.InternalWorkflowExecutionInfo{\n\t\tCompletionEventBatchID:             constants.EmptyEventID,\n\t\tTaskList:                           info.GetTaskList(),\n\t\tTaskListKind:                       info.GetTaskListKind(),\n\t\tWorkflowTypeName:                   info.GetWorkflowTypeName(),\n\t\tWorkflowTimeout:                    info.GetWorkflowTimeout(),\n\t\tDecisionStartToCloseTimeout:        info.GetDecisionTaskTimeout(),\n\t\tExecutionContext:                   info.GetExecutionContext(),\n\t\tState:                              int(info.GetState()),\n\t\tCloseStatus:                        int(info.GetCloseStatus()),\n\t\tLastFirstEventID:                   info.GetLastFirstEventID(),\n\t\tLastEventTaskID:                    info.GetLastEventTaskID(),\n\t\tLastProcessedEvent:                 info.GetLastProcessedEvent(),\n\t\tStartTimestamp:                     info.GetStartTimestamp(),\n\t\tLastUpdatedTimestamp:               info.GetLastUpdatedTimestamp(),\n\t\tCreateRequestID:                    info.GetCreateRequestID(),\n\t\tSignalCount:                        int32(info.GetSignalCount()),\n\t\tDecisionVersion:                    info.GetDecisionVersion(),\n\t\tDecisionScheduleID:                 info.GetDecisionScheduleID(),\n\t\tDecisionStartedID:                  info.GetDecisionStartedID(),\n\t\tDecisionRequestID:                  info.GetDecisionRequestID(),\n\t\tDecisionTimeout:                    info.GetDecisionTimeout(),\n\t\tDecisionAttempt:                    info.GetDecisionAttempt(),\n\t\tDecisionStartedTimestamp:           info.GetDecisionStartedTimestamp(),\n\t\tDecisionScheduledTimestamp:         info.GetDecisionScheduledTimestamp(),\n\t\tDecisionOriginalScheduledTimestamp: info.GetDecisionOriginalScheduledTimestamp(),\n\t\tStickyTaskList:                     info.GetStickyTaskList(),\n\t\tStickyScheduleToStartTimeout:       info.GetStickyScheduleToStartTimeout(),\n\t\tClientLibraryVersion:               info.GetClientLibraryVersion(),\n\t\tClientFeatureVersion:               info.GetClientFeatureVersion(),\n\t\tClientImpl:                         info.GetClientImpl(),\n\t\tAttempt:                            int32(info.GetRetryAttempt()),\n\t\tHasRetryPolicy:                     info.GetHasRetryPolicy(),\n\t\tInitialInterval:                    info.GetRetryInitialInterval(),\n\t\tBackoffCoefficient:                 info.GetRetryBackoffCoefficient(),\n\t\tMaximumInterval:                    info.GetRetryMaximumInterval(),\n\t\tExpirationTime:                     info.GetRetryExpirationTimestamp(),\n\t\tMaximumAttempts:                    info.GetRetryMaximumAttempts(),\n\t\tNonRetriableErrors:                 info.GetRetryNonRetryableErrors(),\n\t\tBranchToken:                        info.GetEventBranchToken(),\n\t\tCronSchedule:                       info.GetCronSchedule(),\n\t\tExpirationInterval:                 info.GetRetryExpiration(),\n\t\tMemo:                               info.GetMemo(),\n\t\tSearchAttributes:                   info.GetSearchAttributes(),\n\t\tHistorySize:                        info.GetHistorySize(),\n\t\tFirstExecutionRunID:                info.FirstExecutionRunID.String(),\n\t\tPartitionConfig:                    info.PartitionConfig,\n\t\tIsCron:                             info.IsCron,\n\t\tCronOverlapPolicy:                  types.CronOverlapPolicy(info.GetCronOverlapPolicy()),\n\t}\n\tif info.ParentDomainID != nil {\n\t\tresult.ParentDomainID = info.ParentDomainID.String()\n\t\tresult.ParentWorkflowID = info.GetParentWorkflowID()\n\t\tresult.ParentRunID = info.ParentRunID.String()\n\t\tresult.InitiatedID = info.GetInitiatedID()\n\t}\n\n\tif info.GetCancelRequested() {\n\t\tresult.CancelRequested = true\n\t\tresult.CancelRequestID = info.GetCancelRequestID()\n\t}\n\n\tif info.CompletionEventBatchID != nil {\n\t\tresult.CompletionEventBatchID = info.GetCompletionEventBatchID()\n\t}\n\n\tif info.CompletionEvent != nil {\n\t\tresult.CompletionEvent = persistence.NewDataBlob(info.CompletionEvent,\n\t\t\tconstants.EncodingType(info.GetCompletionEventEncoding()))\n\t}\n\n\tif info.AutoResetPoints != nil {\n\t\tresult.AutoResetPoints = persistence.NewDataBlob(info.AutoResetPoints,\n\t\t\tconstants.EncodingType(info.GetAutoResetPointsEncoding()))\n\t}\n\n\tif info.ActiveClusterSelectionPolicy != nil {\n\t\tresult.ActiveClusterSelectionPolicy = persistence.NewDataBlob(info.ActiveClusterSelectionPolicy,\n\t\t\tconstants.EncodingType(info.GetActiveClusterSelectionPolicyEncoding()))\n\t}\n\n\treturn result\n}\n\nfunc FromInternalWorkflowExecutionInfo(executionInfo *persistence.InternalWorkflowExecutionInfo) *WorkflowExecutionInfo {\n\tinfo := &WorkflowExecutionInfo{\n\t\tTaskList:                             executionInfo.TaskList,\n\t\tTaskListKind:                         executionInfo.TaskListKind,\n\t\tWorkflowTypeName:                     executionInfo.WorkflowTypeName,\n\t\tWorkflowTimeout:                      executionInfo.WorkflowTimeout,\n\t\tDecisionTaskTimeout:                  executionInfo.DecisionStartToCloseTimeout,\n\t\tExecutionContext:                     executionInfo.ExecutionContext,\n\t\tState:                                int32(executionInfo.State),\n\t\tCloseStatus:                          int32(executionInfo.CloseStatus),\n\t\tLastFirstEventID:                     executionInfo.LastFirstEventID,\n\t\tLastEventTaskID:                      executionInfo.LastEventTaskID,\n\t\tLastProcessedEvent:                   executionInfo.LastProcessedEvent,\n\t\tStartTimestamp:                       executionInfo.StartTimestamp,\n\t\tLastUpdatedTimestamp:                 executionInfo.LastUpdatedTimestamp,\n\t\tCreateRequestID:                      executionInfo.CreateRequestID,\n\t\tDecisionVersion:                      executionInfo.DecisionVersion,\n\t\tDecisionScheduleID:                   executionInfo.DecisionScheduleID,\n\t\tDecisionStartedID:                    executionInfo.DecisionStartedID,\n\t\tDecisionRequestID:                    executionInfo.DecisionRequestID,\n\t\tDecisionTimeout:                      executionInfo.DecisionTimeout,\n\t\tDecisionAttempt:                      executionInfo.DecisionAttempt,\n\t\tDecisionStartedTimestamp:             executionInfo.DecisionStartedTimestamp,\n\t\tDecisionScheduledTimestamp:           executionInfo.DecisionScheduledTimestamp,\n\t\tDecisionOriginalScheduledTimestamp:   executionInfo.DecisionOriginalScheduledTimestamp,\n\t\tStickyTaskList:                       executionInfo.StickyTaskList,\n\t\tStickyScheduleToStartTimeout:         executionInfo.StickyScheduleToStartTimeout,\n\t\tClientLibraryVersion:                 executionInfo.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 executionInfo.ClientFeatureVersion,\n\t\tClientImpl:                           executionInfo.ClientImpl,\n\t\tSignalCount:                          int64(executionInfo.SignalCount),\n\t\tHistorySize:                          executionInfo.HistorySize,\n\t\tCronSchedule:                         executionInfo.CronSchedule,\n\t\tCompletionEventBatchID:               &executionInfo.CompletionEventBatchID,\n\t\tHasRetryPolicy:                       executionInfo.HasRetryPolicy,\n\t\tRetryAttempt:                         int64(executionInfo.Attempt),\n\t\tRetryInitialInterval:                 executionInfo.InitialInterval,\n\t\tRetryBackoffCoefficient:              executionInfo.BackoffCoefficient,\n\t\tRetryMaximumInterval:                 executionInfo.MaximumInterval,\n\t\tRetryMaximumAttempts:                 executionInfo.MaximumAttempts,\n\t\tRetryExpiration:                      executionInfo.ExpirationInterval,\n\t\tRetryExpirationTimestamp:             executionInfo.ExpirationTime,\n\t\tRetryNonRetryableErrors:              executionInfo.NonRetriableErrors,\n\t\tEventStoreVersion:                    persistence.EventStoreVersion,\n\t\tEventBranchToken:                     executionInfo.BranchToken,\n\t\tAutoResetPoints:                      executionInfo.AutoResetPoints.GetData(),\n\t\tAutoResetPointsEncoding:              string(executionInfo.AutoResetPoints.GetEncoding()),\n\t\tSearchAttributes:                     executionInfo.SearchAttributes,\n\t\tMemo:                                 executionInfo.Memo,\n\t\tCompletionEventEncoding:              string(constants.EncodingTypeEmpty),\n\t\tVersionHistoriesEncoding:             string(constants.EncodingTypeEmpty),\n\t\tInitiatedID:                          constants.EmptyEventID,\n\t\tFirstExecutionRunID:                  MustParseUUID(executionInfo.FirstExecutionRunID),\n\t\tPartitionConfig:                      executionInfo.PartitionConfig,\n\t\tIsCron:                               executionInfo.IsCron,\n\t\tCronOverlapPolicy:                    executionInfo.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:         executionInfo.ActiveClusterSelectionPolicy.GetData(),\n\t\tActiveClusterSelectionPolicyEncoding: string(executionInfo.ActiveClusterSelectionPolicy.GetEncoding()),\n\t}\n\n\tif executionInfo.CompletionEvent != nil {\n\t\tinfo.CompletionEvent = executionInfo.CompletionEvent.Data\n\t\tinfo.CompletionEventEncoding = string(executionInfo.CompletionEvent.Encoding)\n\t}\n\n\tif executionInfo.ParentDomainID != \"\" {\n\t\tinfo.ParentDomainID = MustParseUUID(executionInfo.ParentDomainID)\n\t\tinfo.ParentWorkflowID = executionInfo.ParentWorkflowID\n\t\tinfo.ParentRunID = MustParseUUID(executionInfo.ParentRunID)\n\t\tinfo.InitiatedID = executionInfo.InitiatedID\n\t}\n\n\tif executionInfo.CancelRequested {\n\t\tinfo.CancelRequested = true\n\t\tinfo.CancelRequestID = executionInfo.CancelRequestID\n\t}\n\n\tif executionInfo.ActiveClusterSelectionPolicy != nil {\n\t\tinfo.ActiveClusterSelectionPolicy = executionInfo.ActiveClusterSelectionPolicy.Data\n\t\tinfo.ActiveClusterSelectionPolicyEncoding = string(executionInfo.ActiveClusterSelectionPolicy.Encoding)\n\t}\n\n\treturn info\n}\n"
  },
  {
    "path": "common/persistence/serialization/persistence_mapper_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestInternalWorkflowExecutionInfo(t *testing.T) {\n\texpected := &persistence.InternalWorkflowExecutionInfo{\n\t\tParentDomainID:                     uuid.New(),\n\t\tParentWorkflowID:                   \"ParentWorkflowID\",\n\t\tParentRunID:                        uuid.New(),\n\t\tFirstExecutionRunID:                uuid.New(),\n\t\tInitiatedID:                        int64(rand.Intn(1000)),\n\t\tCompletionEventBatchID:             int64(rand.Intn(1000)),\n\t\tCompletionEvent:                    persistence.NewDataBlob([]byte(`CompletionEvent`), constants.EncodingTypeJSON),\n\t\tTaskList:                           \"TaskList\",\n\t\tTaskListKind:                       types.TaskListKindNormal,\n\t\tWorkflowTypeName:                   \"WorkflowTypeName\",\n\t\tWorkflowTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionStartToCloseTimeout:        time.Minute * time.Duration(rand.Intn(10)),\n\t\tExecutionContext:                   []byte(\"ExecutionContext\"),\n\t\tState:                              rand.Intn(1000),\n\t\tCloseStatus:                        rand.Intn(1000),\n\t\tLastFirstEventID:                   int64(rand.Intn(1000)),\n\t\tLastEventTaskID:                    int64(rand.Intn(1000)),\n\t\tLastProcessedEvent:                 int64(rand.Intn(1000)),\n\t\tStartTimestamp:                     time.UnixMilli(1752018142820),\n\t\tLastUpdatedTimestamp:               time.UnixMilli(1752018142821),\n\t\tCreateRequestID:                    \"CreateRequestID\",\n\t\tSignalCount:                        int32(rand.Intn(1000)),\n\t\tDecisionVersion:                    int64(rand.Intn(1000)),\n\t\tDecisionScheduleID:                 int64(rand.Intn(1000)),\n\t\tDecisionStartedID:                  int64(rand.Intn(1000)),\n\t\tDecisionRequestID:                  \"DecisionRequestID\",\n\t\tDecisionTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionAttempt:                    int64(rand.Intn(1000)),\n\t\tDecisionStartedTimestamp:           time.UnixMilli(1752018142822),\n\t\tDecisionScheduledTimestamp:         time.UnixMilli(1752018142823),\n\t\tDecisionOriginalScheduledTimestamp: time.UnixMilli(1752018142824),\n\t\tCancelRequested:                    true,\n\t\tCancelRequestID:                    \"CancelRequestID\",\n\t\tStickyTaskList:                     \"StickyTaskList\",\n\t\tStickyScheduleToStartTimeout:       time.Minute * time.Duration(rand.Intn(10)),\n\t\tClientLibraryVersion:               \"ClientLibraryVersion\",\n\t\tClientFeatureVersion:               \"ClientFeatureVersion\",\n\t\tClientImpl:                         \"ClientImpl\",\n\t\tAutoResetPoints:                    persistence.NewDataBlob([]byte(\"AutoResetPoints\"), constants.EncodingTypeJSON),\n\t\tAttempt:                            int32(rand.Intn(1000)),\n\t\tHasRetryPolicy:                     true,\n\t\tInitialInterval:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tBackoffCoefficient:                 rand.Float64() * 1000,\n\t\tMaximumInterval:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tExpirationTime:                     time.UnixMilli(1752018142825),\n\t\tMaximumAttempts:                    int32(rand.Intn(1000)),\n\t\tNonRetriableErrors:                 []string{\"RetryNonRetryableErrors\"},\n\t\tBranchToken:                        []byte(\"EventBranchToken\"),\n\t\tCronSchedule:                       \"CronSchedule\",\n\t\tExpirationInterval:                 time.Minute * time.Duration(rand.Intn(10)),\n\t\tMemo:                               map[string][]byte{\"key_1\": []byte(\"Memo\")},\n\t\tSearchAttributes:                   map[string][]byte{\"key_1\": []byte(\"SearchAttributes\")},\n\t\tHistorySize:                        int64(rand.Intn(1000)),\n\t\tPartitionConfig:                    map[string]string{\"zone\": \"dca1\"},\n\t\tIsCron:                             true,\n\t\tActiveClusterSelectionPolicy:       persistence.NewDataBlob([]byte(\"ActiveClusterSelectionPolicy\"), constants.EncodingTypeJSON),\n\t\tCronOverlapPolicy:                  types.CronOverlapPolicySkipped,\n\t}\n\tactual := ToInternalWorkflowExecutionInfo(FromInternalWorkflowExecutionInfo(expected))\n\tassert.Equal(t, expected, actual)\n}\n"
  },
  {
    "path": "common/persistence/serialization/serialization_test_utils.go",
    "content": "package serialization\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar shardInfoTestData = &ShardInfo{\n\tStolenSinceRenew:                      1,\n\tUpdatedAt:                             time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tReplicationAckLevel:                   1,\n\tTransferAckLevel:                      1,\n\tTimerAckLevel:                         time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tDomainNotificationVersion:             1,\n\tClusterTransferAckLevel:               map[string]int64{\"test\": 1},\n\tClusterTimerAckLevel:                  map[string]time.Time{\"test\": time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local)},\n\tTransferProcessingQueueStates:         []byte{1, 2, 3},\n\tTimerProcessingQueueStates:            []byte{1, 2, 3},\n\tOwner:                                 \"owner\",\n\tClusterReplicationLevel:               map[string]int64{\"test\": 1},\n\tPendingFailoverMarkers:                []byte{2, 3, 4},\n\tPendingFailoverMarkersEncoding:        \"\",\n\tTransferProcessingQueueStatesEncoding: \"\",\n\tTimerProcessingQueueStatesEncoding:    \"\",\n}\n\nvar domainInfoTestData = &DomainInfo{\n\tDescription:                 \"test_desc\",\n\tOwner:                       \"test_owner\",\n\tStatus:                      1,\n\tRetention:                   48 * time.Hour,\n\tEmitMetric:                  true,\n\tArchivalBucket:              \"test_bucket\",\n\tArchivalStatus:              1,\n\tConfigVersion:               1,\n\tFailoverVersion:             1,\n\tNotificationVersion:         1,\n\tFailoverNotificationVersion: 1,\n\tActiveClusterName:           \"test_active_cluster\",\n\tClusters:                    []string{\"test_active_cluster\", \"test_standby_cluster\"},\n\tData:                        map[string]string{\"test_key\": \"test_value\"},\n\tBadBinaries:                 []byte{1, 2, 3},\n\tBadBinariesEncoding:         \"\",\n\tHistoryArchivalStatus:       1,\n\tHistoryArchivalURI:          \"test_history_archival_uri\",\n\tVisibilityArchivalStatus:    1,\n\tVisibilityArchivalURI:       \"test_visibility_archival_uri\",\n}\n\nvar historyTreeInfoTestData = &HistoryTreeInfo{\n\tCreatedTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tAncestors: []*types.HistoryBranchRange{\n\t\t{\n\t\t\tBranchID: \"test_branch_id1\",\n\t\t},\n\t},\n}\n\nvar workflowExecutionInfoTestData = &WorkflowExecutionInfo{\n\tParentDomainID:                     MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tParentWorkflowID:                   \"test_parent_workflow_id\",\n\tParentRunID:                        MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tInitiatedID:                        1,\n\tCompletionEventBatchID:             common.Int64Ptr(2),\n\tCompletionEvent:                    []byte(\"test_completion_event\"),\n\tCompletionEventEncoding:            \"test_completion_event_encoding\",\n\tTaskList:                           \"test_task_list\",\n\tWorkflowTypeName:                   \"test_workflow_type\",\n\tWorkflowTimeout:                    10 * time.Second,\n\tDecisionTaskTimeout:                5 * time.Second,\n\tExecutionContext:                   []byte(\"test_execution_context\"),\n\tState:                              1,\n\tCloseStatus:                        1,\n\tStartVersion:                       1,\n\tLastWriteEventID:                   common.Int64Ptr(3),\n\tLastEventTaskID:                    4,\n\tLastFirstEventID:                   5,\n\tLastProcessedEvent:                 6,\n\tStartTimestamp:                     time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tLastUpdatedTimestamp:               time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tCreateRequestID:                    \"test_create_request_id\",\n\tDecisionVersion:                    7,\n\tDecisionScheduleID:                 8,\n\tDecisionStartedID:                  9,\n\tDecisionRequestID:                  \"test_decision_request_id\",\n\tDecisionTimeout:                    3 * time.Second,\n\tDecisionAttempt:                    10,\n\tDecisionStartedTimestamp:           time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tDecisionScheduledTimestamp:         time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tDecisionOriginalScheduledTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tCancelRequested:                    true,\n\tCancelRequestID:                    \"test_cancel_request_id\",\n\tStickyTaskList:                     \"test_sticky_task_list\",\n\tStickyScheduleToStartTimeout:       2 * time.Second,\n\tRetryAttempt:                       11,\n\tRetryInitialInterval:               1 * time.Second,\n\tRetryMaximumInterval:               30 * time.Second,\n\tRetryMaximumAttempts:               3,\n\tRetryExpiration:                    time.Hour,\n\tRetryBackoffCoefficient:            2.0,\n\tRetryExpirationTimestamp:           time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tRetryNonRetryableErrors:            []string{\"test_error\"},\n\tHasRetryPolicy:                     true,\n\tCronSchedule:                       \"test_cron\",\n\tIsCron:                             true,\n\tEventStoreVersion:                  12,\n\tEventBranchToken:                   []byte(\"test_branch_token\"),\n\tSignalCount:                        13,\n\tHistorySize:                        14,\n\tClientLibraryVersion:               \"test_client_version\",\n\tClientFeatureVersion:               \"test_feature_version\",\n\tClientImpl:                         \"test_client_impl\",\n\tAutoResetPoints:                    []byte(\"test_reset_points\"),\n\tAutoResetPointsEncoding:            \"test_reset_points_encoding\",\n\tSearchAttributes:                   map[string][]byte{\"test_key\": []byte(\"test_value\")},\n\tMemo:                               map[string][]byte{\"test_memo\": []byte(\"test_memo_value\")},\n\tVersionHistories:                   []byte(\"test_version_histories\"),\n\tVersionHistoriesEncoding:           \"test_version_histories_encoding\",\n\tFirstExecutionRunID:                MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n}\n\nvar activityInfoTestData = &ActivityInfo{\n\tVersion:                  1,\n\tScheduledEventBatchID:    2,\n\tScheduledEvent:           []byte(\"test_scheduled_event\"),\n\tScheduledEventEncoding:   \"test_scheduled_encoding\",\n\tScheduledTimestamp:       time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tStartedID:                3,\n\tStartedEvent:             []byte(\"test_started_event\"),\n\tStartedEventEncoding:     \"test_started_encoding\",\n\tStartedTimestamp:         time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tActivityID:               \"test_activity_id\",\n\tRequestID:                \"test_request_id\",\n\tScheduleToStartTimeout:   5 * time.Second,\n\tScheduleToCloseTimeout:   10 * time.Second,\n\tStartToCloseTimeout:      8 * time.Second,\n\tHeartbeatTimeout:         3 * time.Second,\n\tCancelRequested:          true,\n\tCancelRequestID:          4,\n\tTimerTaskStatus:          5,\n\tAttempt:                  6,\n\tTaskList:                 \"test_task_list\",\n\tStartedIdentity:          \"test_identity\",\n\tHasRetryPolicy:           true,\n\tRetryInitialInterval:     1 * time.Second,\n\tRetryMaximumInterval:     30 * time.Second,\n\tRetryMaximumAttempts:     3,\n\tRetryExpirationTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tRetryBackoffCoefficient:  2.0,\n\tRetryNonRetryableErrors:  []string{\"test_error\"},\n\tRetryLastFailureReason:   \"test_failure_reason\",\n\tRetryLastWorkerIdentity:  \"test_worker_identity\",\n\tRetryLastFailureDetails:  []byte(\"test_failure_details\"),\n}\n\nvar childExecutionInfoTestData = &ChildExecutionInfo{\n\tVersion:                1,\n\tInitiatedEventBatchID:  2,\n\tStartedID:              3,\n\tInitiatedEvent:         []byte(\"test_initiated_event\"),\n\tInitiatedEventEncoding: \"test_initiated_encoding\",\n\tStartedWorkflowID:      \"test_started_workflow_id\",\n\tStartedRunID:           MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tStartedEvent:           []byte(\"test_started_event\"),\n\tStartedEventEncoding:   \"test_started_encoding\",\n\tCreateRequestID:        \"test_create_request_id\",\n\tDomainID:               \"test_domain_id\",\n\tDomainNameDEPRECATED:   \"test_domain_name\",\n\tWorkflowTypeName:       \"test_workflow_type\",\n\tParentClosePolicy:      4,\n}\n\nvar signalInfoTestData = &SignalInfo{\n\tVersion:               1,\n\tInitiatedEventBatchID: 2,\n\tRequestID:             \"test_request_id\",\n\tName:                  \"test_signal_name\",\n\tInput:                 []byte(\"test_input\"),\n\tControl:               []byte(\"test_control\"),\n}\n\nvar requestCancelInfoTestData = &RequestCancelInfo{\n\tVersion:               1,\n\tInitiatedEventBatchID: 2,\n\tCancelRequestID:       \"test_cancel_request_id\",\n}\n\nvar timerInfoTestData = &TimerInfo{\n\tVersion:         1,\n\tStartedID:       2,\n\tExpiryTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tTaskID:          3,\n}\n\nvar taskInfoTestData = &TaskInfo{\n\tWorkflowID:       \"test_workflow_id\",\n\tRunID:            MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tScheduleID:       1,\n\tExpiryTimestamp:  time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tCreatedTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tPartitionConfig:  map[string]string{\"test_key\": \"test_value\"},\n}\n\nvar taskListInfoTestData = &TaskListInfo{\n\tKind:            1,\n\tAckLevel:        2,\n\tExpiryTimestamp: time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n\tLastUpdated:     time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n}\n\nvar transferTaskInfoTestData = &TransferTaskInfo{\n\tDomainID:                MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tWorkflowID:              \"test_workflow_id\",\n\tRunID:                   MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tTaskType:                1,\n\tTargetDomainID:          MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tTargetDomainIDs:         []UUID{MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\")},\n\tTargetWorkflowID:        \"test_target_workflow_id\",\n\tTargetRunID:             MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tTaskList:                \"test_task_list\",\n\tTargetChildWorkflowOnly: true,\n\tScheduleID:              2,\n\tVersion:                 3,\n\tVisibilityTimestamp:     time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n}\n\nvar timerTaskInfoTestData = &TimerTaskInfo{\n\tDomainID:        MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tWorkflowID:      \"test_workflow_id\",\n\tRunID:           MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tTaskType:        1,\n\tTimeoutType:     common.Int16Ptr(2),\n\tVersion:         3,\n\tScheduleAttempt: 4,\n\tEventID:         5,\n}\n\nvar replicationTaskInfoTestData = &ReplicationTaskInfo{\n\tDomainID:                MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tWorkflowID:              \"test_workflow_id\",\n\tRunID:                   MustParseUUID(\"6ba7b810-9dad-11d1-80b4-00c04fd430c8\"),\n\tTaskType:                1,\n\tVersion:                 2,\n\tFirstEventID:            3,\n\tNextEventID:             4,\n\tScheduledID:             5,\n\tEventStoreVersion:       6,\n\tNewRunEventStoreVersion: 7,\n\tBranchToken:             []byte(\"test_branch_token\"),\n\tNewRunBranchToken:       []byte(\"test_new_run_branch_token\"),\n\tCreationTimestamp:       time.Date(2025, 1, 1, 0, 0, 0, 0, time.Local),\n}\n"
  },
  {
    "path": "common/persistence/serialization/snappy_thrift_decoder.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"bytes\"\n\n\t\"github.com/golang/snappy\"\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n)\n\ntype snappyThriftDecoder struct{}\n\nfunc newSnappyThriftDecoder() decoder {\n\treturn &snappyThriftDecoder{}\n}\n\nfunc (d *snappyThriftDecoder) shardInfoFromBlob(data []byte) (*ShardInfo, error) {\n\tresult := &sqlblobs.ShardInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn shardInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) domainInfoFromBlob(data []byte) (*DomainInfo, error) {\n\tresult := &sqlblobs.DomainInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn domainInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) historyTreeInfoFromBlob(data []byte) (*HistoryTreeInfo, error) {\n\tresult := &sqlblobs.HistoryTreeInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn historyTreeInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) workflowExecutionInfoFromBlob(data []byte) (*WorkflowExecutionInfo, error) {\n\tresult := &sqlblobs.WorkflowExecutionInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn workflowExecutionInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) activityInfoFromBlob(data []byte) (*ActivityInfo, error) {\n\tresult := &sqlblobs.ActivityInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn activityInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) childExecutionInfoFromBlob(data []byte) (*ChildExecutionInfo, error) {\n\tresult := &sqlblobs.ChildExecutionInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn childExecutionInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) signalInfoFromBlob(data []byte) (*SignalInfo, error) {\n\tresult := &sqlblobs.SignalInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn signalInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) requestCancelInfoFromBlob(data []byte) (*RequestCancelInfo, error) {\n\tresult := &sqlblobs.RequestCancelInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn requestCancelInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) timerInfoFromBlob(data []byte) (*TimerInfo, error) {\n\tresult := &sqlblobs.TimerInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn timerInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) taskInfoFromBlob(data []byte) (*TaskInfo, error) {\n\tresult := &sqlblobs.TaskInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn taskInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) taskListInfoFromBlob(data []byte) (*TaskListInfo, error) {\n\tresult := &sqlblobs.TaskListInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn taskListInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) transferTaskInfoFromBlob(data []byte) (*TransferTaskInfo, error) {\n\tresult := &sqlblobs.TransferTaskInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn transferTaskInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) crossClusterTaskInfoFromBlob(data []byte) (*CrossClusterTaskInfo, error) {\n\tresult := &sqlblobs.TransferTaskInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn crossClusterTaskInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) timerTaskInfoFromBlob(data []byte) (*TimerTaskInfo, error) {\n\tresult := &sqlblobs.TimerTaskInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn timerTaskInfoFromThrift(result), nil\n}\n\nfunc (d *snappyThriftDecoder) replicationTaskInfoFromBlob(data []byte) (*ReplicationTaskInfo, error) {\n\tresult := &sqlblobs.ReplicationTaskInfo{}\n\tif err := snappyThriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn replicationTaskInfoFromThrift(result), nil\n}\n\nfunc snappyThriftRWDecode(b []byte, result thriftRWType) error {\n\tdecompressed, err := snappy.Decode(nil, b)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tbuf := bytes.NewReader(decompressed)\n\tsr := binary.Default.Reader(buf)\n\treturn result.Decode(sr)\n}\n"
  },
  {
    "path": "common/persistence/serialization/snappy_thrift_decoder_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/golang/snappy\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestSnappyThriftDecoderRoundTrip(t *testing.T) {\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRWSnappy)),\n\t}\n\tparser, err := NewParser(dc)\n\trequire.NoError(t, err)\n\n\ttestCases := []struct {\n\t\tname string\n\t\tdata interface{}\n\t}{\n\t\t{name: \"ShardInfo\", data: shardInfoTestData},\n\t\t{name: \"DomainInfo\", data: domainInfoTestData},\n\t\t{name: \"HistoryTreeInfo\", data: historyTreeInfoTestData},\n\t\t{name: \"WorkflowExecutionInfo\", data: workflowExecutionInfoTestData},\n\t\t{name: \"ActivityInfo\", data: activityInfoTestData},\n\t\t{name: \"ChildExecutionInfo\", data: childExecutionInfoTestData},\n\t\t{name: \"SignalInfo\", data: signalInfoTestData},\n\t\t{name: \"RequestCancelInfo\", data: requestCancelInfoTestData},\n\t\t{name: \"TimerInfo\", data: timerInfoTestData},\n\t\t{name: \"TaskInfo\", data: taskInfoTestData},\n\t\t{name: \"TaskListInfo\", data: taskListInfoTestData},\n\t\t{name: \"TransferTaskInfo\", data: transferTaskInfoTestData},\n\t\t{name: \"TimerTaskInfo\", data: timerTaskInfoTestData},\n\t\t{name: \"ReplicationTaskInfo\", data: replicationTaskInfoTestData},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tblob := encodeWithParser(t, parser, tc.data)\n\t\t\tdecoded := decodeWithParser(t, parser, blob, tc.data)\n\t\t\tassert.Equal(t, tc.data, decoded)\n\t\t})\n\t}\n}\n\nfunc TestSnappyThriftDecoderErrorHandling(t *testing.T) {\n\tdecoder := newSnappyThriftDecoder()\n\n\ttestCases := []struct {\n\t\tname        string\n\t\tdata        []byte\n\t\tdecodeFunc  func([]byte) (interface{}, error)\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"Invalid snappy data for ShardInfo\",\n\t\t\tdata: []byte(\"invalid snappy data\"),\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.shardInfoFromBlob(data)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Empty data for DomainInfo\",\n\t\t\tdata: []byte{},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.domainInfoFromBlob(data)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Corrupted snappy data for ActivityInfo\",\n\t\t\tdata: []byte{0xff, 0xff, 0xff, 0xff},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.activityInfoFromBlob(data)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Valid snappy but invalid thrift for WorkflowExecutionInfo\",\n\t\t\tdata: func() []byte {\n\t\t\t\t// Create valid snappy compressed data but with invalid thrift content\n\t\t\t\tinvalidThrift := []byte(\"not thrift data\")\n\t\t\t\tcompressed := snappy.Encode(nil, invalidThrift)\n\t\t\t\treturn compressed\n\t\t\t}(),\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.workflowExecutionInfoFromBlob(data)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := tc.decodeFunc(tc.data)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSnappyThriftRWDecode(t *testing.T) {\n\t// Test the low-level snappyThriftRWDecode function directly\n\n\tt.Run(\"Valid data\", func(t *testing.T) {\n\t\t// Create a simple thrift struct and encode it\n\t\tencoder := newSnappyThriftEncoder()\n\t\tshardInfo := &ShardInfo{\n\t\t\tStolenSinceRenew:    1,\n\t\t\tUpdatedAt:           time.Now(),\n\t\t\tReplicationAckLevel: 2,\n\t\t\tTransferAckLevel:    3,\n\t\t}\n\n\t\tencoded, err := encoder.shardInfoToBlob(shardInfo)\n\t\trequire.NoError(t, err)\n\n\t\t// Now decode it back\n\t\tdecoder := newSnappyThriftDecoder()\n\t\tdecoded, err := decoder.shardInfoFromBlob(encoded)\n\t\trequire.NoError(t, err)\n\n\t\tassert.Equal(t, shardInfo.StolenSinceRenew, decoded.StolenSinceRenew)\n\t\tassert.Equal(t, shardInfo.ReplicationAckLevel, decoded.ReplicationAckLevel)\n\t\tassert.Equal(t, shardInfo.TransferAckLevel, decoded.TransferAckLevel)\n\t})\n\n\tt.Run(\"Invalid snappy compression\", func(t *testing.T) {\n\t\tdecoder := newSnappyThriftDecoder()\n\n\t\t// Test with invalid snappy data\n\t\tinvalidData := []byte(\"this is not snappy compressed data\")\n\t\t_, err := decoder.shardInfoFromBlob(invalidData)\n\t\tassert.Error(t, err)\n\t\tassert.Contains(t, err.Error(), \"snappy\")\n\t})\n\n\tt.Run(\"Valid snappy but invalid thrift\", func(t *testing.T) {\n\t\tdecoder := newSnappyThriftDecoder()\n\n\t\t// Create valid snappy data but with invalid thrift content\n\t\tinvalidThrift := []byte(\"not a valid thrift message\")\n\t\tcompressed := snappy.Encode(nil, invalidThrift)\n\n\t\t_, err := decoder.shardInfoFromBlob(compressed)\n\t\tassert.Error(t, err)\n\t})\n}\n\nfunc TestSnappyThriftDecoderInterface(t *testing.T) {\n\t// Verify that snappyThriftDecoder implements the decoder interface\n\tvar _ decoder = (*snappyThriftDecoder)(nil)\n\n\t// Test that newSnappyThriftDecoder returns a valid decoder\n\tdecoder := newSnappyThriftDecoder()\n\tassert.NotNil(t, decoder)\n\tassert.IsType(t, &snappyThriftDecoder{}, decoder)\n}\n\nfunc TestSnappyThriftDecoderNilHandling(t *testing.T) {\n\tdecoder := newSnappyThriftDecoder()\n\n\t// Test each method with nil data\n\ttestCases := []struct {\n\t\tname       string\n\t\tdecodeFunc func() (interface{}, error)\n\t}{\n\t\t{\n\t\t\tname: \"shardInfoFromBlob with nil\",\n\t\t\tdecodeFunc: func() (interface{}, error) {\n\t\t\t\treturn decoder.shardInfoFromBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"domainInfoFromBlob with nil\",\n\t\t\tdecodeFunc: func() (interface{}, error) {\n\t\t\t\treturn decoder.domainInfoFromBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activityInfoFromBlob with nil\",\n\t\t\tdecodeFunc: func() (interface{}, error) {\n\t\t\t\treturn decoder.activityInfoFromBlob(nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := tc.decodeFunc()\n\t\t\tassert.Error(t, err)\n\t\t\tassert.Nil(t, result)\n\t\t})\n\t}\n}\n\n// Helper functions for encoding and decoding with parser\nfunc encodeWithParser(t *testing.T, parser Parser, data interface{}) []byte {\n\tvar blob []byte\n\tvar err error\n\n\tswitch v := data.(type) {\n\tcase *ShardInfo:\n\t\tdb, e := parser.ShardInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *DomainInfo:\n\t\tdb, e := parser.DomainInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *HistoryTreeInfo:\n\t\tdb, e := parser.HistoryTreeInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *WorkflowExecutionInfo:\n\t\tdb, e := parser.WorkflowExecutionInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *ActivityInfo:\n\t\tdb, e := parser.ActivityInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *ChildExecutionInfo:\n\t\tdb, e := parser.ChildExecutionInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *SignalInfo:\n\t\tdb, e := parser.SignalInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *RequestCancelInfo:\n\t\tdb, e := parser.RequestCancelInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *TimerInfo:\n\t\tdb, e := parser.TimerInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *TaskInfo:\n\t\tdb, e := parser.TaskInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *TaskListInfo:\n\t\tdb, e := parser.TaskListInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *TransferTaskInfo:\n\t\tdb, e := parser.TransferTaskInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *TimerTaskInfo:\n\t\tdb, e := parser.TimerTaskInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tcase *ReplicationTaskInfo:\n\t\tdb, e := parser.ReplicationTaskInfoToBlob(v)\n\t\trequire.NoError(t, e)\n\t\tblob = db.Data\n\tdefault:\n\t\tt.Fatalf(\"Unknown type %T\", v)\n\t}\n\n\trequire.NoError(t, err)\n\trequire.NotEmpty(t, blob)\n\treturn blob\n}\n\nfunc decodeWithParser(t *testing.T, parser Parser, blob []byte, data interface{}) interface{} {\n\tvar result interface{}\n\tvar err error\n\n\tencoding := string(constants.EncodingTypeThriftRWSnappy)\n\n\tswitch data.(type) {\n\tcase *ShardInfo:\n\t\tresult, err = parser.ShardInfoFromBlob(blob, encoding)\n\tcase *DomainInfo:\n\t\tresult, err = parser.DomainInfoFromBlob(blob, encoding)\n\tcase *HistoryTreeInfo:\n\t\tresult, err = parser.HistoryTreeInfoFromBlob(blob, encoding)\n\tcase *WorkflowExecutionInfo:\n\t\tresult, err = parser.WorkflowExecutionInfoFromBlob(blob, encoding)\n\tcase *ActivityInfo:\n\t\tresult, err = parser.ActivityInfoFromBlob(blob, encoding)\n\tcase *ChildExecutionInfo:\n\t\tresult, err = parser.ChildExecutionInfoFromBlob(blob, encoding)\n\tcase *SignalInfo:\n\t\tresult, err = parser.SignalInfoFromBlob(blob, encoding)\n\tcase *RequestCancelInfo:\n\t\tresult, err = parser.RequestCancelInfoFromBlob(blob, encoding)\n\tcase *TimerInfo:\n\t\tresult, err = parser.TimerInfoFromBlob(blob, encoding)\n\tcase *TaskInfo:\n\t\tresult, err = parser.TaskInfoFromBlob(blob, encoding)\n\tcase *TaskListInfo:\n\t\tresult, err = parser.TaskListInfoFromBlob(blob, encoding)\n\tcase *TransferTaskInfo:\n\t\tresult, err = parser.TransferTaskInfoFromBlob(blob, encoding)\n\tcase *TimerTaskInfo:\n\t\tresult, err = parser.TimerTaskInfoFromBlob(blob, encoding)\n\tcase *ReplicationTaskInfo:\n\t\tresult, err = parser.ReplicationTaskInfoFromBlob(blob, encoding)\n\tdefault:\n\t\tt.Fatalf(\"Unknown type %T\", data)\n\t}\n\n\trequire.NoError(t, err)\n\trequire.NotNil(t, result)\n\treturn result\n}\n"
  },
  {
    "path": "common/persistence/serialization/snappy_thrift_encoder.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"bytes\"\n\n\t\"github.com/golang/snappy\"\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype snappyThriftEncoder struct{}\n\nfunc newSnappyThriftEncoder() encoder {\n\treturn &snappyThriftEncoder{}\n}\n\nfunc (e *snappyThriftEncoder) shardInfoToBlob(info *ShardInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(shardInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) domainInfoToBlob(info *DomainInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(domainInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) historyTreeInfoToBlob(info *HistoryTreeInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(historyTreeInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) workflowExecutionInfoToBlob(info *WorkflowExecutionInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(workflowExecutionInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) activityInfoToBlob(info *ActivityInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(activityInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) childExecutionInfoToBlob(info *ChildExecutionInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(childExecutionInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) signalInfoToBlob(info *SignalInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(signalInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) requestCancelInfoToBlob(info *RequestCancelInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(requestCancelInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) timerInfoToBlob(info *TimerInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(timerInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) taskInfoToBlob(info *TaskInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(taskInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) taskListInfoToBlob(info *TaskListInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(taskListInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) transferTaskInfoToBlob(info *TransferTaskInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(transferTaskInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) crossClusterTaskInfoToBlob(info *CrossClusterTaskInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(crossClusterTaskInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) timerTaskInfoToBlob(info *TimerTaskInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(timerTaskInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) replicationTaskInfoToBlob(info *ReplicationTaskInfo) ([]byte, error) {\n\treturn snappyThriftRWEncode(replicationTaskInfoToThrift(info))\n}\n\nfunc (e *snappyThriftEncoder) encodingType() constants.EncodingType {\n\treturn constants.EncodingTypeThriftRWSnappy\n}\n\nfunc snappyThriftRWEncode(t thriftRWType) ([]byte, error) {\n\tvar b bytes.Buffer\n\tsw := binary.Default.Writer(&b)\n\tdefer sw.Close()\n\tif err := t.Encode(sw); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn snappy.Encode(nil, b.Bytes()), nil\n}\n"
  },
  {
    "path": "common/persistence/serialization/snappy_thrift_encoder_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/golang/snappy\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestSnappyThriftEncoderRoundTrip(t *testing.T) {\n\tencoder := newSnappyThriftEncoder()\n\tdecoder := newSnappyThriftDecoder()\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tdata       interface{}\n\t\tencodeFunc func(interface{}) ([]byte, error)\n\t\tdecodeFunc func([]byte) (interface{}, error)\n\t}{\n\t\t{\n\t\t\tname: \"ShardInfo\",\n\t\t\tdata: shardInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.shardInfoToBlob(data.(*ShardInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.shardInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"DomainInfo\",\n\t\t\tdata: domainInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.domainInfoToBlob(data.(*DomainInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.domainInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"HistoryTreeInfo\",\n\t\t\tdata: historyTreeInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.historyTreeInfoToBlob(data.(*HistoryTreeInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.historyTreeInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionInfo\",\n\t\t\tdata: workflowExecutionInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.workflowExecutionInfoToBlob(data.(*WorkflowExecutionInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.workflowExecutionInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityInfo\",\n\t\t\tdata: activityInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.activityInfoToBlob(data.(*ActivityInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.activityInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildExecutionInfo\",\n\t\t\tdata: childExecutionInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.childExecutionInfoToBlob(data.(*ChildExecutionInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.childExecutionInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalInfo\",\n\t\t\tdata: signalInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.signalInfoToBlob(data.(*SignalInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.signalInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RequestCancelInfo\",\n\t\t\tdata: requestCancelInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.requestCancelInfoToBlob(data.(*RequestCancelInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.requestCancelInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TimerInfo\",\n\t\t\tdata: timerInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.timerInfoToBlob(data.(*TimerInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.timerInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TaskInfo\",\n\t\t\tdata: taskInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.taskInfoToBlob(data.(*TaskInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.taskInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TaskListInfo\",\n\t\t\tdata: taskListInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.taskListInfoToBlob(data.(*TaskListInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.taskListInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TransferTaskInfo\",\n\t\t\tdata: transferTaskInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.transferTaskInfoToBlob(data.(*TransferTaskInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.transferTaskInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TimerTaskInfo\",\n\t\t\tdata: timerTaskInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.timerTaskInfoToBlob(data.(*TimerTaskInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.timerTaskInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ReplicationTaskInfo\",\n\t\t\tdata: replicationTaskInfoTestData,\n\t\t\tencodeFunc: func(data interface{}) ([]byte, error) {\n\t\t\t\treturn encoder.replicationTaskInfoToBlob(data.(*ReplicationTaskInfo))\n\t\t\t},\n\t\t\tdecodeFunc: func(data []byte) (interface{}, error) {\n\t\t\t\treturn decoder.replicationTaskInfoFromBlob(data)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Encode the data using the encoder\n\t\t\tencoded, err := tc.encodeFunc(tc.data)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.NotEmpty(t, encoded)\n\n\t\t\t// Verify the data is snappy compressed\n\t\t\tassert.True(t, isValidSnappyData(encoded), \"encoded data should be valid snappy compressed data\")\n\n\t\t\t// Decode the data using the decoder and verify it matches\n\t\t\tdecoded, err := tc.decodeFunc(encoded)\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tc.data, decoded)\n\t\t})\n\t}\n}\n\nfunc TestSnappyThriftEncoderEncodingType(t *testing.T) {\n\tencoder := newSnappyThriftEncoder()\n\n\tencodingType := encoder.encodingType()\n\tassert.Equal(t, constants.EncodingTypeThriftRWSnappy, encodingType)\n}\n\nfunc TestSnappyThriftEncoderInterface(t *testing.T) {\n\t// Verify that snappyThriftEncoder implements the encoder interface\n\tvar _ encoder = (*snappyThriftEncoder)(nil)\n\n\t// Test that newSnappyThriftEncoder returns a valid encoder\n\tencoder := newSnappyThriftEncoder()\n\tassert.NotNil(t, encoder)\n\tassert.IsType(t, &snappyThriftEncoder{}, encoder)\n}\n\nfunc TestSnappyThriftEncoderNilHandling(t *testing.T) {\n\tencoder := newSnappyThriftEncoder()\n\n\t// Test each method with nil data\n\ttestCases := []struct {\n\t\tname       string\n\t\tencodeFunc func() ([]byte, error)\n\t}{\n\t\t{\n\t\t\tname: \"shardInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.shardInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"domainInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.domainInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activityInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.activityInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"workflowExecutionInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.workflowExecutionInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"childExecutionInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.childExecutionInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"signalInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.signalInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"requestCancelInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.requestCancelInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timerInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.timerInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"taskInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.taskInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"taskListInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.taskListInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"transferTaskInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.transferTaskInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"crossClusterTaskInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.crossClusterTaskInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timerTaskInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.timerTaskInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"replicationTaskInfoToBlob with nil\",\n\t\t\tencodeFunc: func() ([]byte, error) {\n\t\t\t\treturn encoder.replicationTaskInfoToBlob(nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tvar result []byte\n\t\t\tvar err error\n\t\t\tvar panicRecovered bool\n\n\t\t\t// Use defer/recover to handle panic as expected behavior for nil input\n\t\t\tfunc() {\n\t\t\t\tdefer func() {\n\t\t\t\t\tif r := recover(); r != nil {\n\t\t\t\t\t\tpanicRecovered = true\n\t\t\t\t\t}\n\t\t\t\t}()\n\t\t\t\tresult, err = tc.encodeFunc()\n\t\t\t}()\n\n\t\t\t// Nil input should either produce an error, panic, or valid empty data\n\t\t\tif panicRecovered {\n\t\t\t\t// Panic is expected for nil input\n\t\t\t\tassert.True(t, true, \"nil input caused expected panic\")\n\t\t\t} else if err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\t// If no error, result should be valid snappy data\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tassert.True(t, isValidSnappyData(result), \"nil input should produce valid snappy data\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSnappyThriftRWEncode(t *testing.T) {\n\t// Test the low-level snappyThriftRWEncode function directly\n\n\tt.Run(\"Valid thrift object\", func(t *testing.T) {\n\t\t// Create a simple thrift struct and encode it\n\t\tshardInfo := shardInfoTestData\n\n\t\t// Convert to thrift format\n\t\tthriftStruct := shardInfoToThrift(shardInfo)\n\n\t\t// Encode using the low-level function\n\t\tencoded, err := snappyThriftRWEncode(thriftStruct)\n\t\trequire.NoError(t, err)\n\t\trequire.NotEmpty(t, encoded)\n\n\t\t// Verify it's valid snappy data\n\t\tassert.True(t, isValidSnappyData(encoded))\n\n\t\t// Verify we can decode it back\n\t\tdecoder := newSnappyThriftDecoder()\n\t\tdecoded, err := decoder.shardInfoFromBlob(encoded)\n\t\trequire.NoError(t, err)\n\n\t\tassert.Equal(t, shardInfo.StolenSinceRenew, decoded.StolenSinceRenew)\n\t\tassert.Equal(t, shardInfo.ReplicationAckLevel, decoded.ReplicationAckLevel)\n\t\tassert.Equal(t, shardInfo.TransferAckLevel, decoded.TransferAckLevel)\n\t})\n\n\tt.Run(\"Nil thrift object\", func(t *testing.T) {\n\t\t// Test with nil thrift object (converted from nil input)\n\t\tthriftStruct := shardInfoToThrift(nil)\n\t\tassert.Nil(t, thriftStruct)\n\n\t\tvar err error\n\t\tvar panicRecovered bool\n\n\t\t// Use defer/recover to handle panic as expected behavior for nil input\n\t\tfunc() {\n\t\t\tdefer func() {\n\t\t\t\tif r := recover(); r != nil {\n\t\t\t\t\tpanicRecovered = true\n\t\t\t\t}\n\t\t\t}()\n\t\t\t_, err = snappyThriftRWEncode(thriftStruct)\n\t\t}()\n\n\t\t// Either panic or error is acceptable for nil input\n\t\tif panicRecovered {\n\t\t\tassert.True(t, true, \"nil thrift object caused expected panic\")\n\t\t} else if err != nil {\n\t\t\tassert.Error(t, err)\n\t\t}\n\t})\n}\n\nfunc TestSnappyThriftEncoderWithParser(t *testing.T) {\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRWSnappy)),\n\t}\n\t// Test encoder integration with parser\n\tparser, err := NewParser(dc)\n\trequire.NoError(t, err)\n\n\ttestData := shardInfoTestData\n\n\t// Encode using parser\n\tblob, err := parser.ShardInfoToBlob(testData)\n\trequire.NoError(t, err)\n\tassert.Equal(t, constants.EncodingTypeThriftRWSnappy, blob.Encoding)\n\tassert.NotEmpty(t, blob.Data)\n\tassert.True(t, isValidSnappyData(blob.Data))\n\n\t// Decode using parser\n\tdecoded, err := parser.ShardInfoFromBlob(blob.Data, string(blob.Encoding))\n\trequire.NoError(t, err)\n\tassert.Equal(t, testData, decoded)\n}\n\nfunc TestSnappyThriftEncoderDataCompression(t *testing.T) {\n\tencoder := newSnappyThriftEncoder()\n\n\t// Create a large data structure to test compression\n\tlargeData := &WorkflowExecutionInfo{\n\t\tWorkflowTypeName: \"very_long_workflow_type_name_that_should_compress_well_when_repeated\",\n\t\tTaskList:         \"very_long_task_list_name_that_should_compress_well_when_repeated\",\n\t\tExecutionContext: make([]byte, 1000), // Large byte array\n\t\tSearchAttributes: make(map[string][]byte),\n\t\tMemo:             make(map[string][]byte),\n\t}\n\n\t// Fill with repetitive data that should compress well\n\tfor i := 0; i < 100; i++ {\n\t\tkey := fmt.Sprintf(\"repetitive_key_that_compresses_well_%d\", i)\n\t\tvalue := []byte(\"repetitive_value_that_compresses_well_when_repeated_many_times\")\n\t\tlargeData.SearchAttributes[key] = value\n\t\tlargeData.Memo[key] = value\n\t}\n\n\t// Encode the data\n\tencoded, err := encoder.workflowExecutionInfoToBlob(largeData)\n\trequire.NoError(t, err)\n\trequire.NotEmpty(t, encoded)\n\n\t// Verify it's compressed (should be significantly smaller than uncompressed)\n\tassert.True(t, isValidSnappyData(encoded))\n\n\t// Decode and verify correctness\n\tdecoder := newSnappyThriftDecoder()\n\tdecoded, err := decoder.workflowExecutionInfoFromBlob(encoded)\n\trequire.NoError(t, err)\n\tassert.Equal(t, largeData.WorkflowTypeName, decoded.WorkflowTypeName)\n\tassert.Equal(t, largeData.TaskList, decoded.TaskList)\n\tassert.Len(t, decoded.SearchAttributes, 100)\n\tassert.Len(t, decoded.Memo, 100)\n}\n\n// Helper function to check if data is valid snappy compressed data\nfunc isValidSnappyData(data []byte) bool {\n\t_, err := snappy.Decode(nil, data)\n\treturn err == nil\n}\n"
  },
  {
    "path": "common/persistence/serialization/task_serializer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination task_serializer_mock.go github.com/uber/cadence/common/persistence/serialization TaskSerializer\n\npackage serialization\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tTaskSerializer interface {\n\t\tSerializeTask(persistence.HistoryTaskCategory, persistence.Task) (persistence.DataBlob, error)\n\t\tDeserializeTask(persistence.HistoryTaskCategory, *persistence.DataBlob) (persistence.Task, error)\n\t}\n\n\ttaskSerializerImpl struct {\n\t\tparser Parser\n\t}\n)\n\nfunc NewTaskSerializer(parser Parser) TaskSerializer {\n\treturn &taskSerializerImpl{\n\t\tparser: parser,\n\t}\n}\n\nfunc (s *taskSerializerImpl) SerializeTask(category persistence.HistoryTaskCategory, task persistence.Task) (persistence.DataBlob, error) {\n\tswitch category.ID() {\n\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\treturn s.serializeTransferTask(task)\n\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\treturn s.serializeTimerTask(task)\n\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\treturn s.serializeReplicationTask(task)\n\tdefault:\n\t\treturn persistence.DataBlob{}, fmt.Errorf(\"unknown category ID: %v\", category.ID())\n\t}\n}\n\nfunc (s *taskSerializerImpl) DeserializeTask(category persistence.HistoryTaskCategory, blob *persistence.DataBlob) (persistence.Task, error) {\n\tswitch category.ID() {\n\tcase persistence.HistoryTaskCategoryIDTransfer:\n\t\treturn s.deserializeTransferTask(blob)\n\tcase persistence.HistoryTaskCategoryIDTimer:\n\t\treturn s.deserializeTimerTask(blob)\n\tcase persistence.HistoryTaskCategoryIDReplication:\n\t\treturn s.deserializeReplicationTask(blob)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown category ID: %v\", category.ID())\n\t}\n}\n\nfunc (s *taskSerializerImpl) serializeTransferTask(task persistence.Task) (persistence.DataBlob, error) {\n\tinfo := &TransferTaskInfo{\n\t\tTaskType:            int16(task.GetTaskType()),\n\t\tTargetWorkflowID:    persistence.TransferTaskTransferTargetWorkflowID,\n\t\tVersion:             task.GetVersion(),\n\t\tVisibilityTimestamp: task.GetVisibilityTimestamp(),\n\t}\n\tswitch t := task.(type) {\n\tcase *persistence.ActivityTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TaskList = t.TaskList\n\t\tinfo.ScheduleID = t.ScheduleID\n\tcase *persistence.DecisionTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TaskList = t.TaskList\n\t\tinfo.ScheduleID = t.ScheduleID\n\t\tinfo.OriginalTaskList = t.OriginalTaskList\n\t\tinfo.OriginalTaskListKind = t.OriginalTaskListKind\n\tcase *persistence.CancelExecutionTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TargetWorkflowID = t.TargetWorkflowID\n\t\tif t.TargetRunID != \"\" {\n\t\t\tinfo.TargetRunID = MustParseUUID(t.TargetRunID)\n\t\t}\n\t\tinfo.TargetChildWorkflowOnly = t.TargetChildWorkflowOnly\n\t\tinfo.ScheduleID = t.InitiatedID\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.SignalExecutionTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TargetWorkflowID = t.TargetWorkflowID\n\t\tif t.TargetRunID != \"\" {\n\t\t\tinfo.TargetRunID = MustParseUUID(t.TargetRunID)\n\t\t}\n\t\tinfo.TargetChildWorkflowOnly = t.TargetChildWorkflowOnly\n\t\tinfo.ScheduleID = t.InitiatedID\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.StartChildExecutionTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TargetWorkflowID = t.TargetWorkflowID\n\t\tinfo.ScheduleID = t.InitiatedID\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.RecordChildExecutionCompletedTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TargetDomainID = MustParseUUID(t.TargetDomainID)\n\t\tinfo.TargetWorkflowID = t.TargetWorkflowID\n\t\tif t.TargetRunID != \"\" {\n\t\t\tinfo.TargetRunID = MustParseUUID(t.TargetRunID)\n\t\t}\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.CloseExecutionTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.RecordWorkflowStartedTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.RecordWorkflowClosedTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.ResetWorkflowTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.UpsertWorkflowSearchAttributesTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tdefault:\n\t\treturn persistence.DataBlob{}, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Unknown transfer type: %v\", task.GetTaskType()),\n\t\t}\n\t}\n\treturn s.parser.TransferTaskInfoToBlob(info)\n}\n\nfunc (s *taskSerializerImpl) deserializeTransferTask(blob *persistence.DataBlob) (persistence.Task, error) {\n\tinfo, err := s.parser.TransferTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar task persistence.Task\n\tworkflowIdentifier := persistence.WorkflowIdentifier{\n\t\tDomainID:   info.DomainID.String(),\n\t\tWorkflowID: info.GetWorkflowID(),\n\t\tRunID:      info.RunID.String(),\n\t}\n\ttaskData := persistence.TaskData{\n\t\tVersion:             info.GetVersion(),\n\t\tVisibilityTimestamp: info.GetVisibilityTimestamp(),\n\t}\n\tswitch info.GetTaskType() {\n\tcase persistence.TransferTaskTypeDecisionTask:\n\t\ttask = &persistence.DecisionTask{\n\t\t\tWorkflowIdentifier:   workflowIdentifier,\n\t\t\tTaskData:             taskData,\n\t\t\tTargetDomainID:       info.TargetDomainID.String(),\n\t\t\tTaskList:             info.GetTaskList(),\n\t\t\tScheduleID:           info.GetScheduleID(),\n\t\t\tOriginalTaskList:     info.GetOriginalTaskList(),\n\t\t\tOriginalTaskListKind: info.GetOriginalTaskListKind(),\n\t\t}\n\tcase persistence.TransferTaskTypeActivityTask:\n\t\ttask = &persistence.ActivityTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     info.TargetDomainID.String(),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t\tScheduleID:         info.GetScheduleID(),\n\t\t}\n\tcase persistence.TransferTaskTypeCloseExecution:\n\t\ttask = &persistence.CloseExecutionTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeCancelExecution:\n\t\ttask = &persistence.CancelExecutionTask{\n\t\t\tWorkflowIdentifier:      workflowIdentifier,\n\t\t\tTaskData:                taskData,\n\t\t\tTargetDomainID:          info.TargetDomainID.String(),\n\t\t\tTargetWorkflowID:        info.GetTargetWorkflowID(),\n\t\t\tTargetRunID:             info.TargetRunID.String(),\n\t\t\tTargetChildWorkflowOnly: info.GetTargetChildWorkflowOnly(),\n\t\t\tInitiatedID:             info.GetScheduleID(),\n\t\t\tTaskList:                info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeStartChildExecution:\n\t\ttask = &persistence.StartChildExecutionTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     info.TargetDomainID.String(),\n\t\t\tTargetWorkflowID:   info.GetTargetWorkflowID(),\n\t\t\tInitiatedID:        info.GetScheduleID(),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeSignalExecution:\n\t\ttask = &persistence.SignalExecutionTask{\n\t\t\tWorkflowIdentifier:      workflowIdentifier,\n\t\t\tTaskData:                taskData,\n\t\t\tTargetDomainID:          info.TargetDomainID.String(),\n\t\t\tTargetWorkflowID:        info.GetTargetWorkflowID(),\n\t\t\tTargetRunID:             info.TargetRunID.String(),\n\t\t\tTargetChildWorkflowOnly: info.GetTargetChildWorkflowOnly(),\n\t\t\tInitiatedID:             info.GetScheduleID(),\n\t\t\tTaskList:                info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeRecordWorkflowStarted:\n\t\ttask = &persistence.RecordWorkflowStartedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeResetWorkflow:\n\t\ttask = &persistence.ResetWorkflowTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeUpsertWorkflowSearchAttributes:\n\t\ttask = &persistence.UpsertWorkflowSearchAttributesTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeRecordWorkflowClosed:\n\t\ttask = &persistence.RecordWorkflowClosedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TransferTaskTypeRecordChildExecutionCompleted:\n\t\ttask = &persistence.RecordChildExecutionCompletedTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTargetDomainID:     info.TargetDomainID.String(),\n\t\t\tTargetWorkflowID:   info.GetTargetWorkflowID(),\n\t\t\tTargetRunID:        info.TargetRunID.String(),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown transfer task type: %v\", info.GetTaskType())\n\t}\n\treturn task, nil\n}\n\nfunc (s *taskSerializerImpl) serializeTimerTask(task persistence.Task) (persistence.DataBlob, error) {\n\tinfo := &TimerTaskInfo{\n\t\tTaskType: int16(task.GetTaskType()),\n\t\tVersion:  task.GetVersion(),\n\t\tEventID:  constants.EmptyEventID,\n\t}\n\tswitch t := task.(type) {\n\tcase *persistence.DecisionTimeoutTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.EventID = t.EventID\n\t\tinfo.TimeoutType = common.Int16Ptr(int16(t.TimeoutType))\n\t\tinfo.ScheduleAttempt = t.ScheduleAttempt\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.ActivityTimeoutTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.EventID = t.EventID\n\t\tinfo.TimeoutType = common.Int16Ptr(int16(t.TimeoutType))\n\t\tinfo.ScheduleAttempt = t.Attempt\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.UserTimerTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.EventID = t.EventID\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.ActivityRetryTimerTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.EventID = t.EventID\n\t\tinfo.ScheduleAttempt = int64(t.Attempt)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.WorkflowBackoffTimerTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TimeoutType = common.Int16Ptr(int16(t.TimeoutType))\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.WorkflowTimeoutTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tcase *persistence.DeleteHistoryEventTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.TaskList = t.TaskList\n\tdefault:\n\t\treturn persistence.DataBlob{}, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Unknown timer task: %v\", task.GetTaskType()),\n\t\t}\n\t}\n\treturn s.parser.TimerTaskInfoToBlob(info)\n}\n\nfunc (s *taskSerializerImpl) deserializeTimerTask(blob *persistence.DataBlob) (persistence.Task, error) {\n\tinfo, err := s.parser.TimerTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar task persistence.Task\n\tworkflowIdentifier := persistence.WorkflowIdentifier{\n\t\tDomainID:   info.DomainID.String(),\n\t\tWorkflowID: info.GetWorkflowID(),\n\t\tRunID:      info.RunID.String(),\n\t}\n\ttaskData := persistence.TaskData{\n\t\tVersion: info.GetVersion(),\n\t}\n\tswitch info.GetTaskType() {\n\tcase persistence.TaskTypeDecisionTimeout:\n\t\ttask = &persistence.DecisionTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            info.GetEventID(),\n\t\t\tScheduleAttempt:    info.GetScheduleAttempt(),\n\t\t\tTimeoutType:        int(info.GetTimeoutType()),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeActivityTimeout:\n\t\ttask = &persistence.ActivityTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            info.GetEventID(),\n\t\t\tAttempt:            info.GetScheduleAttempt(),\n\t\t\tTimeoutType:        int(info.GetTimeoutType()),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeUserTimer:\n\t\ttask = &persistence.UserTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            info.GetEventID(),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeWorkflowTimeout:\n\t\ttask = &persistence.WorkflowTimeoutTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeDeleteHistoryEvent:\n\t\ttask = &persistence.DeleteHistoryEventTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeActivityRetryTimer:\n\t\ttask = &persistence.ActivityRetryTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tEventID:            info.GetEventID(),\n\t\t\tAttempt:            info.GetScheduleAttempt(),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tcase persistence.TaskTypeWorkflowBackoffTimer:\n\t\ttask = &persistence.WorkflowBackoffTimerTask{\n\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\tTaskData:           taskData,\n\t\t\tTimeoutType:        int(info.GetTimeoutType()),\n\t\t\tTaskList:           info.GetTaskList(),\n\t\t}\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown timer task type: %v\", info.GetTaskType())\n\t}\n\treturn task, nil\n}\n\nfunc (s *taskSerializerImpl) serializeReplicationTask(task persistence.Task) (persistence.DataBlob, error) {\n\tinfo := &ReplicationTaskInfo{\n\t\tTaskType:                int16(task.GetTaskType()),\n\t\tFirstEventID:            constants.EmptyEventID,\n\t\tNextEventID:             constants.EmptyEventID,\n\t\tVersion:                 task.GetVersion(),\n\t\tScheduledID:             constants.EmptyEventID,\n\t\tEventStoreVersion:       persistence.EventStoreVersion,\n\t\tNewRunEventStoreVersion: persistence.EventStoreVersion,\n\t\tCreationTimestamp:       task.GetVisibilityTimestamp(),\n\t}\n\tswitch t := task.(type) {\n\tcase *persistence.HistoryReplicationTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.FirstEventID = t.FirstEventID\n\t\tinfo.NextEventID = t.NextEventID\n\t\tinfo.BranchToken = t.BranchToken\n\t\tinfo.NewRunBranchToken = t.NewRunBranchToken\n\tcase *persistence.SyncActivityTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\t\tinfo.WorkflowID = t.WorkflowID\n\t\tinfo.RunID = MustParseUUID(t.RunID)\n\t\tinfo.ScheduledID = t.ScheduledID\n\tcase *persistence.FailoverMarkerTask:\n\t\tinfo.DomainID = MustParseUUID(t.DomainID)\n\tdefault:\n\t\treturn persistence.DataBlob{}, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Unknown replication task: %v\", task.GetTaskType()),\n\t\t}\n\t}\n\treturn s.parser.ReplicationTaskInfoToBlob(info)\n}\n\nfunc (s *taskSerializerImpl) deserializeReplicationTask(blob *persistence.DataBlob) (persistence.Task, error) {\n\tinfo, err := s.parser.ReplicationTaskInfoFromBlob(blob.Data, string(blob.Encoding))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar task persistence.Task\n\ttaskData := persistence.TaskData{\n\t\tVersion:             info.GetVersion(),\n\t\tVisibilityTimestamp: info.GetCreationTimestamp(),\n\t}\n\tswitch info.GetTaskType() {\n\tcase persistence.ReplicationTaskTypeHistory:\n\t\ttask = &persistence.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   info.DomainID.String(),\n\t\t\t\tWorkflowID: info.GetWorkflowID(),\n\t\t\t\tRunID:      info.RunID.String(),\n\t\t\t},\n\t\t\tTaskData:          taskData,\n\t\t\tFirstEventID:      info.GetFirstEventID(),\n\t\t\tNextEventID:       info.GetNextEventID(),\n\t\t\tBranchToken:       info.BranchToken,\n\t\t\tNewRunBranchToken: info.NewRunBranchToken,\n\t\t}\n\tcase persistence.ReplicationTaskTypeSyncActivity:\n\t\ttask = &persistence.SyncActivityTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   info.DomainID.String(),\n\t\t\t\tWorkflowID: info.GetWorkflowID(),\n\t\t\t\tRunID:      info.RunID.String(),\n\t\t\t},\n\t\t\tTaskData:    taskData,\n\t\t\tScheduledID: info.GetScheduledID(),\n\t\t}\n\tcase persistence.ReplicationTaskTypeFailoverMarker:\n\t\ttask = &persistence.FailoverMarkerTask{\n\t\t\tDomainID: info.DomainID.String(),\n\t\t\tTaskData: taskData,\n\t\t}\n\t}\n\treturn task, nil\n}\n"
  },
  {
    "path": "common/persistence/serialization/task_serializer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence/serialization (interfaces: TaskSerializer)\n//\n// Generated by this command:\n//\n//\tmockgen -package serialization -destination task_serializer_mock.go github.com/uber/cadence/common/persistence/serialization TaskSerializer\n//\n\n// Package serialization is a generated GoMock package.\npackage serialization\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockTaskSerializer is a mock of TaskSerializer interface.\ntype MockTaskSerializer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskSerializerMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskSerializerMockRecorder is the mock recorder for MockTaskSerializer.\ntype MockTaskSerializerMockRecorder struct {\n\tmock *MockTaskSerializer\n}\n\n// NewMockTaskSerializer creates a new mock instance.\nfunc NewMockTaskSerializer(ctrl *gomock.Controller) *MockTaskSerializer {\n\tmock := &MockTaskSerializer{ctrl: ctrl}\n\tmock.recorder = &MockTaskSerializerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskSerializer) EXPECT() *MockTaskSerializerMockRecorder {\n\treturn m.recorder\n}\n\n// DeserializeTask mocks base method.\nfunc (m *MockTaskSerializer) DeserializeTask(arg0 persistence.HistoryTaskCategory, arg1 *persistence.DataBlob) (persistence.Task, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeTask\", arg0, arg1)\n\tret0, _ := ret[0].(persistence.Task)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeTask indicates an expected call of DeserializeTask.\nfunc (mr *MockTaskSerializerMockRecorder) DeserializeTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeTask\", reflect.TypeOf((*MockTaskSerializer)(nil).DeserializeTask), arg0, arg1)\n}\n\n// SerializeTask mocks base method.\nfunc (m *MockTaskSerializer) SerializeTask(arg0 persistence.HistoryTaskCategory, arg1 persistence.Task) (persistence.DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeTask\", arg0, arg1)\n\tret0, _ := ret[0].(persistence.DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeTask indicates an expected call of SerializeTask.\nfunc (mr *MockTaskSerializerMockRecorder) SerializeTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeTask\", reflect.TypeOf((*MockTaskSerializer)(nil).SerializeTask), arg0, arg1)\n}\n"
  },
  {
    "path": "common/persistence/serialization/task_serializer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestTaskSerializerThriftRW(t *testing.T) {\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t}\n\tparser, err := NewParser(dc)\n\trequire.NoError(t, err)\n\ttaskSerializer := NewTaskSerializer(parser)\n\n\tworkflowIdentifier := persistence.WorkflowIdentifier{\n\t\tDomainID:   \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\tWorkflowID: \"test-workflow\",\n\t\tRunID:      \"4be8a310-7d20-483e-a5d2-48659dc47609\",\n\t}\n\n\ttestCases := []struct {\n\t\tcategory persistence.HistoryTaskCategory\n\t\ttask     persistence.Task\n\t}{\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t},\n\t\t\t\tTargetDomainID: \"0be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\tTaskList:       \"test-tl\",\n\t\t\t\tScheduleID:     1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.ActivityTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             2,\n\t\t\t\t\tTaskID:              2,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(2, 2),\n\t\t\t\t},\n\t\t\t\tTargetDomainID: \"2be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\tTaskList:       \"test-tl2\",\n\t\t\t\tScheduleID:     2,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.CloseExecutionTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             3,\n\t\t\t\t\tTaskID:              3,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(3, 3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.CancelExecutionTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             4,\n\t\t\t\t\tTaskID:              4,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(4, 4),\n\t\t\t\t},\n\t\t\t\tTargetDomainID:          \"3be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\tTargetWorkflowID:        \"target-wf\",\n\t\t\t\tTargetRunID:             \"5be8a310-7d20-483e-a5d2-48659dc47609\",\n\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\tInitiatedID:             4,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.StartChildExecutionTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             5,\n\t\t\t\t\tTaskID:              5,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(5, 5),\n\t\t\t\t},\n\t\t\t\tTargetDomainID:   \"3be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\tTargetWorkflowID: \"target-wf\",\n\t\t\t\tInitiatedID:      5,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.SignalExecutionTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             5,\n\t\t\t\t\tTaskID:              5,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(5, 5),\n\t\t\t\t},\n\t\t\t\tTargetDomainID:          \"4be8a310-7d20-483e-a5d2-48659dc47608\",\n\t\t\t\tTargetWorkflowID:        \"target-wf2\",\n\t\t\t\tTargetRunID:             \"7be8a310-7d20-483e-a5d2-48659dc47606\",\n\t\t\t\tTargetChildWorkflowOnly: true,\n\t\t\t\tInitiatedID:             5,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.RecordWorkflowStartedTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             6,\n\t\t\t\t\tTaskID:              6,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(6, 6),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.ResetWorkflowTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             7,\n\t\t\t\t\tTaskID:              7,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(7, 7),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.UpsertWorkflowSearchAttributesTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             8,\n\t\t\t\t\tTaskID:              8,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(8, 8),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.RecordWorkflowClosedTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             9,\n\t\t\t\t\tTaskID:              9,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(9, 9),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\ttask: &persistence.RecordChildExecutionCompletedTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             10,\n\t\t\t\t\tTaskID:              10,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(10, 10),\n\t\t\t\t},\n\t\t\t\tTargetDomainID:   \"7be8a310-7d20-483e-a5d2-48659dc47608\",\n\t\t\t\tTargetWorkflowID: \"target-wf2\",\n\t\t\t\tTargetRunID:      \"2be8a310-7d20-483e-a5d2-48659dc47606\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.DecisionTimeoutTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             11,\n\t\t\t\t\tTaskID:              11,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 11),\n\t\t\t\t},\n\t\t\t\tEventID:         11,\n\t\t\t\tScheduleAttempt: 11,\n\t\t\t\tTimeoutType:     11,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.ActivityTimeoutTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             12,\n\t\t\t\t\tTaskID:              12,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(12, 12),\n\t\t\t\t},\n\t\t\t\tEventID:     12,\n\t\t\t\tAttempt:     12,\n\t\t\t\tTimeoutType: 12,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             13,\n\t\t\t\t\tTaskID:              13,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(13, 13),\n\t\t\t\t},\n\t\t\t\tEventID: 13,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.WorkflowTimeoutTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             14,\n\t\t\t\t\tTaskID:              14,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(14, 14),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.DeleteHistoryEventTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             15,\n\t\t\t\t\tTaskID:              15,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(15, 15),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.ActivityRetryTimerTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             16,\n\t\t\t\t\tTaskID:              16,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(16, 16),\n\t\t\t\t},\n\t\t\t\tEventID: 16,\n\t\t\t\tAttempt: 16,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\ttask: &persistence.WorkflowBackoffTimerTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             17,\n\t\t\t\t\tTaskID:              17,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(17, 17),\n\t\t\t\t},\n\t\t\t\tTimeoutType: 17,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryReplication,\n\t\t\ttask: &persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             18,\n\t\t\t\t\tTaskID:              18,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(18, 18),\n\t\t\t\t},\n\t\t\t\tFirstEventID:      18,\n\t\t\t\tNextEventID:       18,\n\t\t\t\tBranchToken:       []byte(\"18\"),\n\t\t\t\tNewRunBranchToken: []byte(\"180\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryReplication,\n\t\t\ttask: &persistence.SyncActivityTask{\n\t\t\t\tWorkflowIdentifier: workflowIdentifier,\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             19,\n\t\t\t\t\tTaskID:              19,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(19, 19),\n\t\t\t\t},\n\t\t\t\tScheduledID: 19,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcategory: persistence.HistoryTaskCategoryReplication,\n\t\t\ttask: &persistence.FailoverMarkerTask{\n\t\t\t\tDomainID: \"4be8a310-7d20-483e-a5d2-48659dc47608\",\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion:             20,\n\t\t\t\t\tTaskID:              20,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(20, 20),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tblob, err := taskSerializer.SerializeTask(tc.category, tc.task)\n\t\tassert.NoError(t, err)\n\t\ttask, err := taskSerializer.DeserializeTask(tc.category, &blob)\n\t\tassert.NoError(t, err)\n\t\ttask.SetTaskID(tc.task.GetTaskID())\n\t\tif tc.category.Type() == persistence.HistoryTaskCategoryTypeScheduled {\n\t\t\ttask.SetVisibilityTimestamp(tc.task.GetVisibilityTimestamp())\n\t\t}\n\t\tassert.Equal(t, tc.task, task)\n\t}\n}\n"
  },
  {
    "path": "common/persistence/serialization/thrift_decoder.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"bytes\"\n\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n)\n\ntype (\n\tthriftDecoder struct{}\n)\n\nfunc newThriftDecoder() decoder {\n\treturn &thriftDecoder{}\n}\n\nfunc (d *thriftDecoder) shardInfoFromBlob(data []byte) (*ShardInfo, error) {\n\tresult := &sqlblobs.ShardInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn shardInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) domainInfoFromBlob(data []byte) (*DomainInfo, error) {\n\tresult := &sqlblobs.DomainInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn domainInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) historyTreeInfoFromBlob(data []byte) (*HistoryTreeInfo, error) {\n\tresult := &sqlblobs.HistoryTreeInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn historyTreeInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) workflowExecutionInfoFromBlob(data []byte) (*WorkflowExecutionInfo, error) {\n\tresult := &sqlblobs.WorkflowExecutionInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn workflowExecutionInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) activityInfoFromBlob(data []byte) (*ActivityInfo, error) {\n\tresult := &sqlblobs.ActivityInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn activityInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) childExecutionInfoFromBlob(data []byte) (*ChildExecutionInfo, error) {\n\tresult := &sqlblobs.ChildExecutionInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn childExecutionInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) signalInfoFromBlob(data []byte) (*SignalInfo, error) {\n\tresult := &sqlblobs.SignalInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn signalInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) requestCancelInfoFromBlob(data []byte) (*RequestCancelInfo, error) {\n\tresult := &sqlblobs.RequestCancelInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn requestCancelInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) timerInfoFromBlob(data []byte) (*TimerInfo, error) {\n\tresult := &sqlblobs.TimerInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn timerInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) taskInfoFromBlob(data []byte) (*TaskInfo, error) {\n\tresult := &sqlblobs.TaskInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn taskInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) taskListInfoFromBlob(data []byte) (*TaskListInfo, error) {\n\tresult := &sqlblobs.TaskListInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn taskListInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) transferTaskInfoFromBlob(data []byte) (*TransferTaskInfo, error) {\n\tresult := &sqlblobs.TransferTaskInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn transferTaskInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) crossClusterTaskInfoFromBlob(data []byte) (*CrossClusterTaskInfo, error) {\n\tresult := &sqlblobsCrossClusterTaskInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn crossClusterTaskInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) timerTaskInfoFromBlob(data []byte) (*TimerTaskInfo, error) {\n\tresult := &sqlblobs.TimerTaskInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn timerTaskInfoFromThrift(result), nil\n}\n\nfunc (d *thriftDecoder) replicationTaskInfoFromBlob(data []byte) (*ReplicationTaskInfo, error) {\n\tresult := &sqlblobs.ReplicationTaskInfo{}\n\tif err := thriftRWDecode(data, result); err != nil {\n\t\treturn nil, err\n\t}\n\treturn replicationTaskInfoFromThrift(result), nil\n}\n\nfunc thriftRWDecode(b []byte, result thriftRWType) error {\n\tbuf := bytes.NewReader(b)\n\tsr := binary.Default.Reader(buf)\n\treturn result.Decode(sr)\n}\n"
  },
  {
    "path": "common/persistence/serialization/thrift_encoder.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"bytes\"\n\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype thriftEncoder struct{}\n\nfunc newThriftEncoder() encoder {\n\treturn &thriftEncoder{}\n}\n\nfunc (e *thriftEncoder) shardInfoToBlob(info *ShardInfo) ([]byte, error) {\n\treturn thriftRWEncode(shardInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) domainInfoToBlob(info *DomainInfo) ([]byte, error) {\n\treturn thriftRWEncode(domainInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) historyTreeInfoToBlob(info *HistoryTreeInfo) ([]byte, error) {\n\treturn thriftRWEncode(historyTreeInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) workflowExecutionInfoToBlob(info *WorkflowExecutionInfo) ([]byte, error) {\n\treturn thriftRWEncode(workflowExecutionInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) activityInfoToBlob(info *ActivityInfo) ([]byte, error) {\n\treturn thriftRWEncode(activityInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) childExecutionInfoToBlob(info *ChildExecutionInfo) ([]byte, error) {\n\treturn thriftRWEncode(childExecutionInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) signalInfoToBlob(info *SignalInfo) ([]byte, error) {\n\treturn thriftRWEncode(signalInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) requestCancelInfoToBlob(info *RequestCancelInfo) ([]byte, error) {\n\treturn thriftRWEncode(requestCancelInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) timerInfoToBlob(info *TimerInfo) ([]byte, error) {\n\treturn thriftRWEncode(timerInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) taskInfoToBlob(info *TaskInfo) ([]byte, error) {\n\treturn thriftRWEncode(taskInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) taskListInfoToBlob(info *TaskListInfo) ([]byte, error) {\n\treturn thriftRWEncode(taskListInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) transferTaskInfoToBlob(info *TransferTaskInfo) ([]byte, error) {\n\treturn thriftRWEncode(transferTaskInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) crossClusterTaskInfoToBlob(info *CrossClusterTaskInfo) ([]byte, error) {\n\treturn thriftRWEncode(crossClusterTaskInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) timerTaskInfoToBlob(info *TimerTaskInfo) ([]byte, error) {\n\treturn thriftRWEncode(timerTaskInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) replicationTaskInfoToBlob(info *ReplicationTaskInfo) ([]byte, error) {\n\treturn thriftRWEncode(replicationTaskInfoToThrift(info))\n}\n\nfunc (e *thriftEncoder) encodingType() constants.EncodingType {\n\treturn constants.EncodingTypeThriftRW\n}\n\nfunc thriftRWEncode(t thriftRWType) ([]byte, error) {\n\tvar b bytes.Buffer\n\tsw := binary.Default.Writer(&b)\n\tdefer sw.Close()\n\tif err := t.Encode(sw); err != nil {\n\t\treturn nil, err\n\t}\n\treturn b.Bytes(), nil\n}\n"
  },
  {
    "path": "common/persistence/serialization/thrift_mapper.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc shardInfoToThrift(info *ShardInfo) *sqlblobs.ShardInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\tresult := &sqlblobs.ShardInfo{\n\t\tStolenSinceRenew:                          &info.StolenSinceRenew,\n\t\tReplicationAckLevel:                       &info.ReplicationAckLevel,\n\t\tTransferAckLevel:                          &info.TransferAckLevel,\n\t\tDomainNotificationVersion:                 &info.DomainNotificationVersion,\n\t\tClusterTransferAckLevel:                   info.ClusterTransferAckLevel,\n\t\tOwner:                                     &info.Owner,\n\t\tClusterReplicationLevel:                   info.ClusterReplicationLevel,\n\t\tPendingFailoverMarkers:                    info.PendingFailoverMarkers,\n\t\tPendingFailoverMarkersEncoding:            &info.PendingFailoverMarkersEncoding,\n\t\tReplicationDlqAckLevel:                    info.ReplicationDlqAckLevel,\n\t\tTransferProcessingQueueStates:             info.TransferProcessingQueueStates,\n\t\tTransferProcessingQueueStatesEncoding:     &info.TransferProcessingQueueStatesEncoding,\n\t\tCrossClusterProcessingQueueStates:         info.CrossClusterProcessingQueueStates,\n\t\tCrossClusterProcessingQueueStatesEncoding: &info.CrossClusterProcessingQueueStatesEncoding,\n\t\tTimerProcessingQueueStates:                info.TimerProcessingQueueStates,\n\t\tTimerProcessingQueueStatesEncoding:        &info.TimerProcessingQueueStatesEncoding,\n\t\tUpdatedAtNanos:                            timeToUnixNanoPtr(info.UpdatedAt),\n\t\tTimerAckLevelNanos:                        timeToUnixNanoPtr(info.TimerAckLevel),\n\t}\n\tif info.ClusterTimerAckLevel != nil {\n\t\tresult.ClusterTimerAckLevel = make(map[string]int64, len(info.ClusterTimerAckLevel))\n\t\tfor k, v := range info.ClusterTimerAckLevel {\n\t\t\tresult.ClusterTimerAckLevel[k] = v.UnixNano()\n\t\t}\n\t}\n\tif info.QueueStates != nil {\n\t\tresult.QueueStates = make(map[int32]*shared.QueueState, len(info.QueueStates))\n\t\tfor k, v := range info.QueueStates {\n\t\t\tresult.QueueStates[k] = thrift.FromQueueState(v)\n\t\t}\n\t}\n\treturn result\n}\n\nfunc shardInfoFromThrift(info *sqlblobs.ShardInfo) *ShardInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\n\tresult := &ShardInfo{\n\t\tStolenSinceRenew:                          info.GetStolenSinceRenew(),\n\t\tReplicationAckLevel:                       info.GetReplicationAckLevel(),\n\t\tTransferAckLevel:                          info.GetTransferAckLevel(),\n\t\tDomainNotificationVersion:                 info.GetDomainNotificationVersion(),\n\t\tClusterTransferAckLevel:                   info.ClusterTransferAckLevel,\n\t\tOwner:                                     info.GetOwner(),\n\t\tClusterReplicationLevel:                   info.ClusterReplicationLevel,\n\t\tPendingFailoverMarkers:                    info.PendingFailoverMarkers,\n\t\tPendingFailoverMarkersEncoding:            info.GetPendingFailoverMarkersEncoding(),\n\t\tReplicationDlqAckLevel:                    info.ReplicationDlqAckLevel,\n\t\tTransferProcessingQueueStates:             info.TransferProcessingQueueStates,\n\t\tTransferProcessingQueueStatesEncoding:     info.GetTransferProcessingQueueStatesEncoding(),\n\t\tCrossClusterProcessingQueueStates:         info.CrossClusterProcessingQueueStates,\n\t\tCrossClusterProcessingQueueStatesEncoding: info.GetCrossClusterProcessingQueueStatesEncoding(),\n\t\tTimerProcessingQueueStates:                info.TimerProcessingQueueStates,\n\t\tTimerProcessingQueueStatesEncoding:        info.GetTimerProcessingQueueStatesEncoding(),\n\t\tUpdatedAt:                                 timeFromUnixNano(info.GetUpdatedAtNanos()),\n\t\tTimerAckLevel:                             timeFromUnixNano(info.GetTimerAckLevelNanos()),\n\t}\n\tif info.ClusterTimerAckLevel != nil {\n\t\tresult.ClusterTimerAckLevel = make(map[string]time.Time, len(info.ClusterTimerAckLevel))\n\t\tfor k, v := range info.ClusterTimerAckLevel {\n\t\t\tresult.ClusterTimerAckLevel[k] = time.Unix(0, v)\n\t\t}\n\t}\n\tif info.QueueStates != nil {\n\t\tresult.QueueStates = make(map[int32]*types.QueueState, len(info.QueueStates))\n\t\tfor k, v := range info.QueueStates {\n\t\t\tresult.QueueStates[k] = thrift.ToQueueState(v)\n\t\t}\n\t}\n\treturn result\n}\n\nfunc domainInfoToThrift(info *DomainInfo) *sqlblobs.DomainInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.DomainInfo{\n\t\tName:                                 &info.Name,\n\t\tDescription:                          &info.Description,\n\t\tOwner:                                &info.Owner,\n\t\tStatus:                               &info.Status,\n\t\tEmitMetric:                           &info.EmitMetric,\n\t\tArchivalBucket:                       &info.ArchivalBucket,\n\t\tArchivalStatus:                       &info.ArchivalStatus,\n\t\tConfigVersion:                        &info.ConfigVersion,\n\t\tNotificationVersion:                  &info.NotificationVersion,\n\t\tFailoverNotificationVersion:          &info.FailoverNotificationVersion,\n\t\tFailoverVersion:                      &info.FailoverVersion,\n\t\tActiveClusterName:                    &info.ActiveClusterName,\n\t\tActiveClustersConfiguration:          info.ActiveClustersConfig,\n\t\tActiveClustersConfigurationEncoding:  &info.ActiveClustersConfigEncoding,\n\t\tClusters:                             info.Clusters,\n\t\tData:                                 info.Data,\n\t\tBadBinaries:                          info.BadBinaries,\n\t\tBadBinariesEncoding:                  &info.BadBinariesEncoding,\n\t\tHistoryArchivalStatus:                &info.HistoryArchivalStatus,\n\t\tHistoryArchivalURI:                   &info.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:             &info.VisibilityArchivalStatus,\n\t\tVisibilityArchivalURI:                &info.VisibilityArchivalURI,\n\t\tPreviousFailoverVersion:              &info.PreviousFailoverVersion,\n\t\tRetentionDays:                        durationToDaysInt16Ptr(info.Retention),\n\t\tFailoverEndTime:                      unixNanoPtr(info.FailoverEndTimestamp),\n\t\tLastUpdatedTime:                      timeToUnixNanoPtr(info.LastUpdatedTimestamp),\n\t\tIsolationGroupsConfiguration:         info.IsolationGroups,\n\t\tIsolationGroupsConfigurationEncoding: &info.IsolationGroupsEncoding,\n\t\tAsyncWorkflowConfiguration:           info.AsyncWorkflowConfig,\n\t\tAsyncWorkflowConfigurationEncoding:   &info.AsyncWorkflowConfigEncoding,\n\t}\n}\n\nfunc domainInfoFromThrift(info *sqlblobs.DomainInfo) *DomainInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &DomainInfo{\n\t\tName:                         info.GetName(),\n\t\tDescription:                  info.GetDescription(),\n\t\tOwner:                        info.GetOwner(),\n\t\tStatus:                       info.GetStatus(),\n\t\tEmitMetric:                   info.GetEmitMetric(),\n\t\tArchivalBucket:               info.GetArchivalBucket(),\n\t\tArchivalStatus:               info.GetArchivalStatus(),\n\t\tConfigVersion:                info.GetConfigVersion(),\n\t\tNotificationVersion:          info.GetNotificationVersion(),\n\t\tFailoverNotificationVersion:  info.GetFailoverNotificationVersion(),\n\t\tFailoverVersion:              info.GetFailoverVersion(),\n\t\tActiveClusterName:            info.GetActiveClusterName(),\n\t\tActiveClustersConfig:         info.GetActiveClustersConfiguration(),\n\t\tActiveClustersConfigEncoding: info.GetActiveClustersConfigurationEncoding(),\n\t\tClusters:                     info.Clusters,\n\t\tData:                         info.Data,\n\t\tBadBinaries:                  info.BadBinaries,\n\t\tBadBinariesEncoding:          info.GetBadBinariesEncoding(),\n\t\tHistoryArchivalStatus:        info.GetHistoryArchivalStatus(),\n\t\tHistoryArchivalURI:           info.GetHistoryArchivalURI(),\n\t\tVisibilityArchivalStatus:     info.GetVisibilityArchivalStatus(),\n\t\tVisibilityArchivalURI:        info.GetVisibilityArchivalURI(),\n\t\tPreviousFailoverVersion:      info.GetPreviousFailoverVersion(),\n\t\tRetention:                    common.DaysToDuration(int32(info.GetRetentionDays())),\n\t\tFailoverEndTimestamp:         timePtr(info.FailoverEndTime),\n\t\tLastUpdatedTimestamp:         timeFromUnixNano(info.GetLastUpdatedTime()),\n\t\tIsolationGroups:              info.GetIsolationGroupsConfiguration(),\n\t\tIsolationGroupsEncoding:      info.GetIsolationGroupsConfigurationEncoding(),\n\t\tAsyncWorkflowConfig:          info.AsyncWorkflowConfiguration,\n\t\tAsyncWorkflowConfigEncoding:  info.GetAsyncWorkflowConfigurationEncoding(),\n\t}\n}\n\nfunc historyTreeInfoToThrift(info *HistoryTreeInfo) *sqlblobs.HistoryTreeInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.HistoryTreeInfo{\n\t\tCreatedTimeNanos: timeToUnixNanoPtr(info.CreatedTimestamp),\n\t\tInfo:             &info.Info,\n\t\tAncestors:        thrift.FromHistoryBranchRangeArray(info.Ancestors),\n\t}\n}\n\nfunc historyTreeInfoFromThrift(info *sqlblobs.HistoryTreeInfo) *HistoryTreeInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &HistoryTreeInfo{\n\t\tCreatedTimestamp: timeFromUnixNano(info.GetCreatedTimeNanos()),\n\t\tInfo:             info.GetInfo(),\n\t\tAncestors:        thrift.ToHistoryBranchRangeArray(info.Ancestors),\n\t}\n}\n\nfunc taskListKindFromThrift(kind *shared.TaskListKind) types.TaskListKind {\n\tres := thrift.ToTaskListKind(kind)\n\tif res == nil {\n\t\treturn types.TaskListKindNormal\n\t}\n\treturn *res\n}\n\nfunc workflowExecutionInfoToThrift(info *WorkflowExecutionInfo) *sqlblobs.WorkflowExecutionInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.WorkflowExecutionInfo{\n\t\tParentDomainID:                          info.ParentDomainID,\n\t\tParentWorkflowID:                        &info.ParentWorkflowID,\n\t\tParentRunID:                             info.ParentRunID,\n\t\tInitiatedID:                             &info.InitiatedID,\n\t\tCompletionEventBatchID:                  info.CompletionEventBatchID,\n\t\tCompletionEvent:                         info.CompletionEvent,\n\t\tCompletionEventEncoding:                 &info.CompletionEventEncoding,\n\t\tTaskList:                                &info.TaskList,\n\t\tTaskListKind:                            thrift.FromTaskListKind(&info.TaskListKind),\n\t\tWorkflowTypeName:                        &info.WorkflowTypeName,\n\t\tWorkflowTimeoutSeconds:                  durationToSecondsInt32Ptr(info.WorkflowTimeout),\n\t\tDecisionTaskTimeoutSeconds:              durationToSecondsInt32Ptr(info.DecisionTaskTimeout),\n\t\tExecutionContext:                        info.ExecutionContext,\n\t\tState:                                   &info.State,\n\t\tCloseStatus:                             &info.CloseStatus,\n\t\tStartVersion:                            &info.StartVersion,\n\t\tLastWriteEventID:                        info.LastWriteEventID,\n\t\tLastEventTaskID:                         &info.LastEventTaskID,\n\t\tLastFirstEventID:                        &info.LastFirstEventID,\n\t\tLastProcessedEvent:                      &info.LastProcessedEvent,\n\t\tStartTimeNanos:                          timeToUnixNanoPtr(info.StartTimestamp),\n\t\tLastUpdatedTimeNanos:                    timeToUnixNanoPtr(info.LastUpdatedTimestamp),\n\t\tDecisionVersion:                         &info.DecisionVersion,\n\t\tDecisionScheduleID:                      &info.DecisionScheduleID,\n\t\tDecisionStartedID:                       &info.DecisionStartedID,\n\t\tDecisionTimeout:                         durationToSecondsInt32Ptr(info.DecisionTimeout),\n\t\tDecisionAttempt:                         &info.DecisionAttempt,\n\t\tDecisionStartedTimestampNanos:           timeToUnixNanoPtr(info.DecisionStartedTimestamp),\n\t\tDecisionScheduledTimestampNanos:         timeToUnixNanoPtr(info.DecisionScheduledTimestamp),\n\t\tCancelRequested:                         &info.CancelRequested,\n\t\tDecisionOriginalScheduledTimestampNanos: timeToUnixNanoPtr(info.DecisionOriginalScheduledTimestamp),\n\t\tCreateRequestID:                         &info.CreateRequestID,\n\t\tDecisionRequestID:                       &info.DecisionRequestID,\n\t\tCancelRequestID:                         &info.CancelRequestID,\n\t\tStickyTaskList:                          &info.StickyTaskList,\n\t\tStickyScheduleToStartTimeout:            durationToSecondsInt64Ptr(info.StickyScheduleToStartTimeout),\n\t\tRetryAttempt:                            &info.RetryAttempt,\n\t\tRetryInitialIntervalSeconds:             durationToSecondsInt32Ptr(info.RetryInitialInterval),\n\t\tRetryMaximumIntervalSeconds:             durationToSecondsInt32Ptr(info.RetryMaximumInterval),\n\t\tRetryMaximumAttempts:                    &info.RetryMaximumAttempts,\n\t\tRetryExpirationSeconds:                  durationToSecondsInt32Ptr(info.RetryExpiration),\n\t\tRetryBackoffCoefficient:                 &info.RetryBackoffCoefficient,\n\t\tRetryExpirationTimeNanos:                timeToUnixNanoPtr(info.RetryExpirationTimestamp),\n\t\tRetryNonRetryableErrors:                 info.RetryNonRetryableErrors,\n\t\tHasRetryPolicy:                          &info.HasRetryPolicy,\n\t\tCronSchedule:                            &info.CronSchedule,\n\t\tCronOverlapPolicy:                       thrift.FromCronOverlapPolicy(&info.CronOverlapPolicy),\n\t\tEventStoreVersion:                       &info.EventStoreVersion,\n\t\tEventBranchToken:                        info.EventBranchToken,\n\t\tSignalCount:                             &info.SignalCount,\n\t\tHistorySize:                             &info.HistorySize,\n\t\tClientLibraryVersion:                    &info.ClientLibraryVersion,\n\t\tClientFeatureVersion:                    &info.ClientFeatureVersion,\n\t\tClientImpl:                              &info.ClientImpl,\n\t\tAutoResetPoints:                         info.AutoResetPoints,\n\t\tAutoResetPointsEncoding:                 &info.AutoResetPointsEncoding,\n\t\tSearchAttributes:                        info.SearchAttributes,\n\t\tMemo:                                    info.Memo,\n\t\tVersionHistories:                        info.VersionHistories,\n\t\tVersionHistoriesEncoding:                &info.VersionHistoriesEncoding,\n\t\tFirstExecutionRunID:                     info.FirstExecutionRunID,\n\t\tPartitionConfig:                         info.PartitionConfig,\n\t\tChecksum:                                info.Checksum,\n\t\tChecksumEncoding:                        &info.ChecksumEncoding,\n\t\tActiveClusterSelectionPolicy:            info.ActiveClusterSelectionPolicy,\n\t\tActiveClusterSelectionPolicyEncoding:    &info.ActiveClusterSelectionPolicyEncoding,\n\t}\n}\n\nfunc workflowExecutionInfoFromThrift(info *sqlblobs.WorkflowExecutionInfo) *WorkflowExecutionInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &WorkflowExecutionInfo{\n\t\tParentDomainID:                       info.ParentDomainID,\n\t\tParentWorkflowID:                     info.GetParentWorkflowID(),\n\t\tParentRunID:                          info.ParentRunID,\n\t\tInitiatedID:                          info.GetInitiatedID(),\n\t\tCompletionEventBatchID:               info.CompletionEventBatchID,\n\t\tCompletionEvent:                      info.CompletionEvent,\n\t\tCompletionEventEncoding:              info.GetCompletionEventEncoding(),\n\t\tTaskList:                             info.GetTaskList(),\n\t\tTaskListKind:                         taskListKindFromThrift(info.TaskListKind),\n\t\tWorkflowTypeName:                     info.GetWorkflowTypeName(),\n\t\tWorkflowTimeout:                      common.SecondsToDuration(int64(info.GetWorkflowTimeoutSeconds())),\n\t\tDecisionTaskTimeout:                  common.SecondsToDuration(int64(info.GetDecisionTaskTimeoutSeconds())),\n\t\tExecutionContext:                     info.ExecutionContext,\n\t\tState:                                info.GetState(),\n\t\tCloseStatus:                          info.GetCloseStatus(),\n\t\tStartVersion:                         info.GetStartVersion(),\n\t\tLastWriteEventID:                     info.LastWriteEventID,\n\t\tLastEventTaskID:                      info.GetLastEventTaskID(),\n\t\tLastFirstEventID:                     info.GetLastFirstEventID(),\n\t\tLastProcessedEvent:                   info.GetLastProcessedEvent(),\n\t\tStartTimestamp:                       timeFromUnixNano(info.GetStartTimeNanos()),\n\t\tLastUpdatedTimestamp:                 timeFromUnixNano(info.GetLastUpdatedTimeNanos()),\n\t\tDecisionVersion:                      info.GetDecisionVersion(),\n\t\tDecisionScheduleID:                   info.GetDecisionScheduleID(),\n\t\tDecisionStartedID:                    info.GetDecisionStartedID(),\n\t\tDecisionTimeout:                      common.SecondsToDuration(int64(info.GetDecisionTimeout())),\n\t\tDecisionAttempt:                      info.GetDecisionAttempt(),\n\t\tDecisionStartedTimestamp:             timeFromUnixNano(info.GetDecisionStartedTimestampNanos()),\n\t\tDecisionScheduledTimestamp:           timeFromUnixNano(info.GetDecisionScheduledTimestampNanos()),\n\t\tCancelRequested:                      info.GetCancelRequested(),\n\t\tDecisionOriginalScheduledTimestamp:   timeFromUnixNano(info.GetDecisionOriginalScheduledTimestampNanos()),\n\t\tCreateRequestID:                      info.GetCreateRequestID(),\n\t\tDecisionRequestID:                    info.GetDecisionRequestID(),\n\t\tCancelRequestID:                      info.GetCancelRequestID(),\n\t\tStickyTaskList:                       info.GetStickyTaskList(),\n\t\tStickyScheduleToStartTimeout:         common.SecondsToDuration(info.GetStickyScheduleToStartTimeout()),\n\t\tRetryAttempt:                         info.GetRetryAttempt(),\n\t\tRetryInitialInterval:                 common.SecondsToDuration(int64(info.GetRetryInitialIntervalSeconds())),\n\t\tRetryMaximumInterval:                 common.SecondsToDuration(int64(info.GetRetryMaximumIntervalSeconds())),\n\t\tRetryMaximumAttempts:                 info.GetRetryMaximumAttempts(),\n\t\tRetryExpiration:                      common.SecondsToDuration(int64(info.GetRetryExpirationSeconds())),\n\t\tRetryBackoffCoefficient:              info.GetRetryBackoffCoefficient(),\n\t\tRetryExpirationTimestamp:             timeFromUnixNano(info.GetRetryExpirationTimeNanos()),\n\t\tRetryNonRetryableErrors:              info.RetryNonRetryableErrors,\n\t\tHasRetryPolicy:                       info.GetHasRetryPolicy(),\n\t\tCronSchedule:                         info.GetCronSchedule(),\n\t\tCronOverlapPolicy:                    cronOverlapPolicyFromThrift(info.CronOverlapPolicy),\n\t\tEventStoreVersion:                    info.GetEventStoreVersion(),\n\t\tEventBranchToken:                     info.EventBranchToken,\n\t\tSignalCount:                          info.GetSignalCount(),\n\t\tHistorySize:                          info.GetHistorySize(),\n\t\tClientLibraryVersion:                 info.GetClientLibraryVersion(),\n\t\tClientFeatureVersion:                 info.GetClientFeatureVersion(),\n\t\tClientImpl:                           info.GetClientImpl(),\n\t\tAutoResetPoints:                      info.AutoResetPoints,\n\t\tAutoResetPointsEncoding:              info.GetAutoResetPointsEncoding(),\n\t\tSearchAttributes:                     info.SearchAttributes,\n\t\tMemo:                                 info.Memo,\n\t\tVersionHistories:                     info.VersionHistories,\n\t\tVersionHistoriesEncoding:             info.GetVersionHistoriesEncoding(),\n\t\tFirstExecutionRunID:                  info.FirstExecutionRunID,\n\t\tPartitionConfig:                      info.PartitionConfig,\n\t\tIsCron:                               info.GetCronSchedule() != \"\",\n\t\tChecksum:                             info.Checksum,\n\t\tChecksumEncoding:                     info.GetChecksumEncoding(),\n\t\tActiveClusterSelectionPolicy:         info.ActiveClusterSelectionPolicy,\n\t\tActiveClusterSelectionPolicyEncoding: info.GetActiveClusterSelectionPolicyEncoding(),\n\t}\n}\n\nfunc activityInfoToThrift(info *ActivityInfo) *sqlblobs.ActivityInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.ActivityInfo{\n\t\tVersion:                       &info.Version,\n\t\tScheduledEventBatchID:         &info.ScheduledEventBatchID,\n\t\tScheduledEvent:                info.ScheduledEvent,\n\t\tScheduledEventEncoding:        &info.ScheduledEventEncoding,\n\t\tScheduledTimeNanos:            timeToUnixNanoPtr(info.ScheduledTimestamp),\n\t\tStartedID:                     &info.StartedID,\n\t\tStartedEvent:                  info.StartedEvent,\n\t\tStartedEventEncoding:          &info.StartedEventEncoding,\n\t\tStartedTimeNanos:              timeToUnixNanoPtr(info.StartedTimestamp),\n\t\tActivityID:                    &info.ActivityID,\n\t\tRequestID:                     &info.RequestID,\n\t\tScheduleToStartTimeoutSeconds: durationToSecondsInt32Ptr(info.ScheduleToStartTimeout),\n\t\tScheduleToCloseTimeoutSeconds: durationToSecondsInt32Ptr(info.ScheduleToCloseTimeout),\n\t\tStartToCloseTimeoutSeconds:    durationToSecondsInt32Ptr(info.StartToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:       durationToSecondsInt32Ptr(info.HeartbeatTimeout),\n\t\tCancelRequested:               &info.CancelRequested,\n\t\tCancelRequestID:               &info.CancelRequestID,\n\t\tTimerTaskStatus:               &info.TimerTaskStatus,\n\t\tAttempt:                       &info.Attempt,\n\t\tTaskList:                      &info.TaskList,\n\t\tTaskListKind:                  thrift.FromTaskListKind(&info.TaskListKind),\n\t\tStartedIdentity:               &info.StartedIdentity,\n\t\tHasRetryPolicy:                &info.HasRetryPolicy,\n\t\tRetryInitialIntervalSeconds:   durationToSecondsInt32Ptr(info.RetryInitialInterval),\n\t\tRetryMaximumIntervalSeconds:   durationToSecondsInt32Ptr(info.RetryMaximumInterval),\n\t\tRetryMaximumAttempts:          &info.RetryMaximumAttempts,\n\t\tRetryExpirationTimeNanos:      timeToUnixNanoPtr(info.RetryExpirationTimestamp),\n\t\tRetryBackoffCoefficient:       &info.RetryBackoffCoefficient,\n\t\tRetryNonRetryableErrors:       info.RetryNonRetryableErrors,\n\t\tRetryLastFailureReason:        &info.RetryLastFailureReason,\n\t\tRetryLastWorkerIdentity:       &info.RetryLastWorkerIdentity,\n\t\tRetryLastFailureDetails:       info.RetryLastFailureDetails,\n\t}\n}\n\nfunc activityInfoFromThrift(info *sqlblobs.ActivityInfo) *ActivityInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &ActivityInfo{\n\t\tVersion:                  info.GetVersion(),\n\t\tScheduledEventBatchID:    info.GetScheduledEventBatchID(),\n\t\tScheduledEvent:           info.ScheduledEvent,\n\t\tScheduledEventEncoding:   info.GetScheduledEventEncoding(),\n\t\tScheduledTimestamp:       timeFromUnixNano(info.GetScheduledTimeNanos()),\n\t\tStartedID:                info.GetStartedID(),\n\t\tStartedEvent:             info.StartedEvent,\n\t\tStartedEventEncoding:     info.GetStartedEventEncoding(),\n\t\tStartedTimestamp:         timeFromUnixNano(info.GetStartedTimeNanos()),\n\t\tActivityID:               info.GetActivityID(),\n\t\tRequestID:                info.GetRequestID(),\n\t\tScheduleToStartTimeout:   common.SecondsToDuration(int64(info.GetScheduleToStartTimeoutSeconds())),\n\t\tScheduleToCloseTimeout:   common.SecondsToDuration(int64(info.GetScheduleToCloseTimeoutSeconds())),\n\t\tStartToCloseTimeout:      common.SecondsToDuration(int64(info.GetStartToCloseTimeoutSeconds())),\n\t\tHeartbeatTimeout:         common.SecondsToDuration(int64(info.GetHeartbeatTimeoutSeconds())),\n\t\tCancelRequested:          info.GetCancelRequested(),\n\t\tCancelRequestID:          info.GetCancelRequestID(),\n\t\tTimerTaskStatus:          info.GetTimerTaskStatus(),\n\t\tAttempt:                  info.GetAttempt(),\n\t\tTaskList:                 info.GetTaskList(),\n\t\tTaskListKind:             taskListKindFromThrift(info.TaskListKind),\n\t\tStartedIdentity:          info.GetStartedIdentity(),\n\t\tHasRetryPolicy:           info.GetHasRetryPolicy(),\n\t\tRetryInitialInterval:     common.SecondsToDuration(int64(info.GetRetryInitialIntervalSeconds())),\n\t\tRetryMaximumInterval:     common.SecondsToDuration(int64(info.GetRetryMaximumIntervalSeconds())),\n\t\tRetryMaximumAttempts:     info.GetRetryMaximumAttempts(),\n\t\tRetryExpirationTimestamp: timeFromUnixNano(info.GetRetryExpirationTimeNanos()),\n\t\tRetryBackoffCoefficient:  info.GetRetryBackoffCoefficient(),\n\t\tRetryNonRetryableErrors:  info.RetryNonRetryableErrors,\n\t\tRetryLastFailureReason:   info.GetRetryLastFailureReason(),\n\t\tRetryLastWorkerIdentity:  info.GetRetryLastWorkerIdentity(),\n\t\tRetryLastFailureDetails:  info.RetryLastFailureDetails,\n\t}\n}\n\nfunc childExecutionInfoToThrift(info *ChildExecutionInfo) *sqlblobs.ChildExecutionInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.ChildExecutionInfo{\n\t\tVersion:                &info.Version,\n\t\tInitiatedEventBatchID:  &info.InitiatedEventBatchID,\n\t\tStartedID:              &info.StartedID,\n\t\tInitiatedEvent:         info.InitiatedEvent,\n\t\tInitiatedEventEncoding: &info.InitiatedEventEncoding,\n\t\tStartedWorkflowID:      &info.StartedWorkflowID,\n\t\tStartedRunID:           info.StartedRunID,\n\t\tStartedEvent:           info.StartedEvent,\n\t\tStartedEventEncoding:   &info.StartedEventEncoding,\n\t\tCreateRequestID:        &info.CreateRequestID,\n\t\tDomainID:               &info.DomainID,\n\t\tDomainName:             &info.DomainNameDEPRECATED,\n\t\tWorkflowTypeName:       &info.WorkflowTypeName,\n\t\tParentClosePolicy:      &info.ParentClosePolicy,\n\t}\n}\n\nfunc childExecutionInfoFromThrift(info *sqlblobs.ChildExecutionInfo) *ChildExecutionInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &ChildExecutionInfo{\n\t\tVersion:                info.GetVersion(),\n\t\tInitiatedEventBatchID:  info.GetInitiatedEventBatchID(),\n\t\tStartedID:              info.GetStartedID(),\n\t\tInitiatedEvent:         info.InitiatedEvent,\n\t\tInitiatedEventEncoding: info.GetInitiatedEventEncoding(),\n\t\tStartedWorkflowID:      info.GetStartedWorkflowID(),\n\t\tStartedRunID:           info.GetStartedRunID(),\n\t\tStartedEvent:           info.StartedEvent,\n\t\tStartedEventEncoding:   info.GetStartedEventEncoding(),\n\t\tCreateRequestID:        info.GetCreateRequestID(),\n\t\tDomainID:               info.GetDomainID(),\n\t\tDomainNameDEPRECATED:   info.GetDomainName(),\n\t\tWorkflowTypeName:       info.GetWorkflowTypeName(),\n\t\tParentClosePolicy:      info.GetParentClosePolicy(),\n\t}\n}\n\nfunc signalInfoToThrift(info *SignalInfo) *sqlblobs.SignalInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.SignalInfo{\n\t\tVersion:               &info.Version,\n\t\tInitiatedEventBatchID: &info.InitiatedEventBatchID,\n\t\tRequestID:             &info.RequestID,\n\t\tName:                  &info.Name,\n\t\tInput:                 info.Input,\n\t\tControl:               info.Control,\n\t}\n}\n\nfunc signalInfoFromThrift(info *sqlblobs.SignalInfo) *SignalInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &SignalInfo{\n\t\tVersion:               info.GetVersion(),\n\t\tInitiatedEventBatchID: info.GetInitiatedEventBatchID(),\n\t\tRequestID:             info.GetRequestID(),\n\t\tName:                  info.GetName(),\n\t\tInput:                 info.Input,\n\t\tControl:               info.Control,\n\t}\n}\n\nfunc requestCancelInfoToThrift(info *RequestCancelInfo) *sqlblobs.RequestCancelInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.RequestCancelInfo{\n\t\tVersion:               &info.Version,\n\t\tInitiatedEventBatchID: &info.InitiatedEventBatchID,\n\t\tCancelRequestID:       &info.CancelRequestID,\n\t}\n}\n\nfunc requestCancelInfoFromThrift(info *sqlblobs.RequestCancelInfo) *RequestCancelInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &RequestCancelInfo{\n\t\tVersion:               info.GetVersion(),\n\t\tInitiatedEventBatchID: info.GetInitiatedEventBatchID(),\n\t\tCancelRequestID:       info.GetCancelRequestID(),\n\t}\n}\n\nfunc timerInfoToThrift(info *TimerInfo) *sqlblobs.TimerInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.TimerInfo{\n\t\tVersion:         &info.Version,\n\t\tStartedID:       &info.StartedID,\n\t\tExpiryTimeNanos: timeToUnixNanoPtr(info.ExpiryTimestamp),\n\t\tTaskID:          &info.TaskID,\n\t}\n}\n\nfunc timerInfoFromThrift(info *sqlblobs.TimerInfo) *TimerInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &TimerInfo{\n\t\tVersion:         info.GetVersion(),\n\t\tStartedID:       info.GetStartedID(),\n\t\tExpiryTimestamp: timeFromUnixNano(info.GetExpiryTimeNanos()),\n\t\tTaskID:          info.GetTaskID(),\n\t}\n}\n\nfunc taskInfoToThrift(info *TaskInfo) *sqlblobs.TaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.TaskInfo{\n\t\tWorkflowID:       &info.WorkflowID,\n\t\tRunID:            info.RunID,\n\t\tScheduleID:       &info.ScheduleID,\n\t\tExpiryTimeNanos:  timeToUnixNanoPtr(info.ExpiryTimestamp),\n\t\tCreatedTimeNanos: timeToUnixNanoPtr(info.CreatedTimestamp),\n\t\tPartitionConfig:  info.PartitionConfig,\n\t}\n}\n\nfunc taskInfoFromThrift(info *sqlblobs.TaskInfo) *TaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &TaskInfo{\n\t\tWorkflowID:       info.GetWorkflowID(),\n\t\tRunID:            info.RunID,\n\t\tScheduleID:       info.GetScheduleID(),\n\t\tExpiryTimestamp:  timeFromUnixNano(info.GetExpiryTimeNanos()),\n\t\tCreatedTimestamp: timeFromUnixNano(info.GetCreatedTimeNanos()),\n\t\tPartitionConfig:  info.PartitionConfig,\n\t}\n}\n\nfunc taskListPartitionConfigToThrift(info *TaskListPartitionConfig) *sqlblobs.TaskListPartitionConfig {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.TaskListPartitionConfig{\n\t\tVersion:            &info.Version,\n\t\tNumReadPartitions:  common.Int32Ptr(info.NumReadPartitions),\n\t\tNumWritePartitions: common.Int32Ptr(info.NumWritePartitions),\n\t\tReadPartitions:     taskListPartitionMapToThrift(info.ReadPartitions),\n\t\tWritePartitions:    taskListPartitionMapToThrift(info.WritePartitions),\n\t}\n}\n\nfunc taskListPartitionMapToThrift(m map[int32]*TaskListPartition) map[int32]*sqlblobs.TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int32]*sqlblobs.TaskListPartition)\n\tfor id, p := range m {\n\t\tresult[id] = &sqlblobs.TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\treturn result\n}\n\nfunc taskListPartitionMapFromThrift(m map[int32]*sqlblobs.TaskListPartition) map[int32]*TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int32]*TaskListPartition)\n\tfor id, p := range m {\n\t\tresult[id] = &TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\treturn result\n}\n\nfunc taskListParititionConfigFromThrift(info *sqlblobs.TaskListPartitionConfig) *TaskListPartitionConfig {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &TaskListPartitionConfig{\n\t\tVersion:            info.GetVersion(),\n\t\tNumReadPartitions:  info.GetNumReadPartitions(),\n\t\tNumWritePartitions: info.GetNumWritePartitions(),\n\t\tReadPartitions:     taskListPartitionMapFromThrift(info.ReadPartitions),\n\t\tWritePartitions:    taskListPartitionMapFromThrift(info.WritePartitions),\n\t}\n}\n\nfunc taskListInfoToThrift(info *TaskListInfo) *sqlblobs.TaskListInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.TaskListInfo{\n\t\tKind:                    &info.Kind,\n\t\tAckLevel:                &info.AckLevel,\n\t\tExpiryTimeNanos:         timeToUnixNanoPtr(info.ExpiryTimestamp),\n\t\tLastUpdatedNanos:        timeToUnixNanoPtr(info.LastUpdated),\n\t\tAdaptivePartitionConfig: taskListPartitionConfigToThrift(info.AdaptivePartitionConfig),\n\t}\n}\n\nfunc taskListInfoFromThrift(info *sqlblobs.TaskListInfo) *TaskListInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &TaskListInfo{\n\t\tKind:                    info.GetKind(),\n\t\tAckLevel:                info.GetAckLevel(),\n\t\tExpiryTimestamp:         timeFromUnixNano(info.GetExpiryTimeNanos()),\n\t\tLastUpdated:             timeFromUnixNano(info.GetLastUpdatedNanos()),\n\t\tAdaptivePartitionConfig: taskListParititionConfigFromThrift(info.AdaptivePartitionConfig),\n\t}\n}\n\nfunc transferTaskInfoToThrift(info *TransferTaskInfo) *sqlblobs.TransferTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\tthriftTaskInfo := &sqlblobs.TransferTaskInfo{\n\t\tDomainID:       info.DomainID,\n\t\tWorkflowID:     &info.WorkflowID,\n\t\tRunID:          info.RunID,\n\t\tTaskType:       &info.TaskType,\n\t\tTargetDomainID: info.TargetDomainID,\n\t\t// TargetDomainIDs will be assigned below\n\t\tTargetWorkflowID:         &info.TargetWorkflowID,\n\t\tTargetRunID:              info.TargetRunID,\n\t\tTaskList:                 &info.TaskList,\n\t\tTargetChildWorkflowOnly:  &info.TargetChildWorkflowOnly,\n\t\tScheduleID:               &info.ScheduleID,\n\t\tVersion:                  &info.Version,\n\t\tVisibilityTimestampNanos: timeToUnixNanoPtr(info.VisibilityTimestamp),\n\t\tOriginalTaskList:         &info.OriginalTaskList,\n\t\tOriginalTaskListKind:     thrift.FromTaskListKind(&info.OriginalTaskListKind),\n\t}\n\tif len(info.TargetDomainIDs) > 0 {\n\t\tthriftTaskInfo.TargetDomainIDs = [][]byte{}\n\t\tfor _, domainID := range info.TargetDomainIDs {\n\t\t\tthriftTaskInfo.TargetDomainIDs = append(thriftTaskInfo.TargetDomainIDs, domainID)\n\t\t}\n\t}\n\treturn thriftTaskInfo\n}\n\nfunc transferTaskInfoFromThrift(info *sqlblobs.TransferTaskInfo) *TransferTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\ttransferTaskInfo := &TransferTaskInfo{\n\t\tDomainID:       info.DomainID,\n\t\tWorkflowID:     info.GetWorkflowID(),\n\t\tRunID:          info.RunID,\n\t\tTaskType:       info.GetTaskType(),\n\t\tTargetDomainID: info.TargetDomainID,\n\t\t// TargetDomainIDs will be assigned below\n\t\tTargetWorkflowID:        info.GetTargetWorkflowID(),\n\t\tTargetRunID:             info.TargetRunID,\n\t\tTaskList:                info.GetTaskList(),\n\t\tTargetChildWorkflowOnly: info.GetTargetChildWorkflowOnly(),\n\t\tScheduleID:              info.GetScheduleID(),\n\t\tVersion:                 info.GetVersion(),\n\t\tVisibilityTimestamp:     timeFromUnixNano(info.GetVisibilityTimestampNanos()),\n\t\tOriginalTaskList:        info.GetOriginalTaskList(),\n\t\tOriginalTaskListKind:    taskListKindFromThrift(info.OriginalTaskListKind),\n\t}\n\tif len(info.GetTargetDomainIDs()) > 0 {\n\t\ttransferTaskInfo.TargetDomainIDs = []UUID{}\n\t\tfor _, domainID := range info.GetTargetDomainIDs() {\n\t\t\ttransferTaskInfo.TargetDomainIDs = append(transferTaskInfo.TargetDomainIDs, domainID)\n\t\t}\n\t}\n\treturn transferTaskInfo\n}\n\nfunc crossClusterTaskInfoToThrift(info *CrossClusterTaskInfo) *sqlblobsCrossClusterTaskInfo {\n\treturn transferTaskInfoToThrift(info)\n}\n\nfunc crossClusterTaskInfoFromThrift(info *sqlblobsCrossClusterTaskInfo) *CrossClusterTaskInfo {\n\treturn transferTaskInfoFromThrift(info)\n}\n\nfunc timerTaskInfoToThrift(info *TimerTaskInfo) *sqlblobs.TimerTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.TimerTaskInfo{\n\t\tDomainID:        info.DomainID,\n\t\tWorkflowID:      &info.WorkflowID,\n\t\tRunID:           info.RunID,\n\t\tTaskType:        &info.TaskType,\n\t\tTimeoutType:     info.TimeoutType,\n\t\tVersion:         &info.Version,\n\t\tScheduleAttempt: &info.ScheduleAttempt,\n\t\tEventID:         &info.EventID,\n\t\tTaskList:        &info.TaskList,\n\t}\n}\n\nfunc timerTaskInfoFromThrift(info *sqlblobs.TimerTaskInfo) *TimerTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &TimerTaskInfo{\n\t\tDomainID:        info.DomainID,\n\t\tWorkflowID:      info.GetWorkflowID(),\n\t\tRunID:           info.RunID,\n\t\tTaskType:        info.GetTaskType(),\n\t\tTimeoutType:     info.TimeoutType,\n\t\tVersion:         info.GetVersion(),\n\t\tScheduleAttempt: info.GetScheduleAttempt(),\n\t\tEventID:         info.GetEventID(),\n\t\tTaskList:        info.GetTaskList(),\n\t}\n}\n\nfunc replicationTaskInfoToThrift(info *ReplicationTaskInfo) *sqlblobs.ReplicationTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &sqlblobs.ReplicationTaskInfo{\n\t\tDomainID:                info.DomainID,\n\t\tWorkflowID:              &info.WorkflowID,\n\t\tRunID:                   info.RunID,\n\t\tTaskType:                &info.TaskType,\n\t\tVersion:                 &info.Version,\n\t\tFirstEventID:            &info.FirstEventID,\n\t\tNextEventID:             &info.NextEventID,\n\t\tScheduledID:             &info.ScheduledID,\n\t\tEventStoreVersion:       &info.EventStoreVersion,\n\t\tNewRunEventStoreVersion: &info.NewRunEventStoreVersion,\n\t\tBranchToken:             info.BranchToken,\n\t\tNewRunBranchToken:       info.NewRunBranchToken,\n\t\tCreationTime:            timeToUnixNanoPtr(info.CreationTimestamp),\n\t}\n}\n\nfunc replicationTaskInfoFromThrift(info *sqlblobs.ReplicationTaskInfo) *ReplicationTaskInfo {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &ReplicationTaskInfo{\n\t\tDomainID:                info.DomainID,\n\t\tWorkflowID:              info.GetWorkflowID(),\n\t\tRunID:                   info.RunID,\n\t\tTaskType:                info.GetTaskType(),\n\t\tVersion:                 info.GetVersion(),\n\t\tFirstEventID:            info.GetFirstEventID(),\n\t\tNextEventID:             info.GetNextEventID(),\n\t\tScheduledID:             info.GetScheduledID(),\n\t\tEventStoreVersion:       info.GetEventStoreVersion(),\n\t\tNewRunEventStoreVersion: info.GetNewRunEventStoreVersion(),\n\t\tBranchToken:             info.BranchToken,\n\t\tNewRunBranchToken:       info.NewRunBranchToken,\n\t\tCreationTimestamp:       timeFromUnixNano(info.GetCreationTime()),\n\t}\n}\n\nvar zeroTimeNanos = time.Time{}.UnixNano()\n\nfunc unixNanoPtr(t *time.Time) *int64 {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn common.Int64Ptr(t.UnixNano())\n}\n\nfunc timeToUnixNanoPtr(t time.Time) *int64 {\n\treturn common.Int64Ptr(t.UnixNano())\n}\n\nfunc timePtr(t *int64) *time.Time {\n\tif t == nil {\n\t\treturn nil\n\t}\n\t// Calling UnixNano() on zero time is undefined an results in a number that is converted back to 1754-08-30T22:43:41Z\n\t// Handle such case explicitly\n\tif *t == zeroTimeNanos {\n\t\treturn common.TimePtr(time.Time{})\n\t}\n\treturn common.TimePtr(time.Unix(0, *t))\n}\n\nfunc timeFromUnixNano(t int64) time.Time {\n\t// Calling UnixNano() on zero time is undefined an results in a number that is converted back to 1754-08-30T22:43:41Z\n\t// Handle such case explicitly\n\tif t == zeroTimeNanos {\n\t\treturn time.Time{}\n\t}\n\treturn time.Unix(0, t)\n}\n\nfunc durationToSecondsInt32Ptr(t time.Duration) *int32 {\n\treturn common.Int32Ptr(int32(common.DurationToSeconds(t)))\n}\n\nfunc durationToSecondsInt64Ptr(t time.Duration) *int64 {\n\treturn common.Int64Ptr(common.DurationToSeconds(t))\n}\n\nfunc durationToDaysInt16Ptr(t time.Duration) *int16 {\n\treturn common.Int16Ptr(int16(common.DurationToDays(t)))\n}\n\nfunc cronOverlapPolicyFromThrift(t *shared.CronOverlapPolicy) types.CronOverlapPolicy {\n\tif t == nil {\n\t\treturn types.CronOverlapPolicySkipped\n\t}\n\tif result := thrift.ToCronOverlapPolicy(t); result != nil {\n\t\treturn *result\n\t}\n\treturn types.CronOverlapPolicySkipped\n}\n"
  },
  {
    "path": "common/persistence/serialization/thrift_mapper_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestShardInfo(t *testing.T) {\n\texpected := &ShardInfo{\n\t\tStolenSinceRenew:                      int32(rand.Intn(1000)),\n\t\tUpdatedAt:                             time.Now(),\n\t\tReplicationAckLevel:                   int64(rand.Intn(1000)),\n\t\tTransferAckLevel:                      int64(rand.Intn(1000)),\n\t\tTimerAckLevel:                         time.Now(),\n\t\tDomainNotificationVersion:             int64(rand.Intn(1000)),\n\t\tClusterTransferAckLevel:               map[string]int64{\"key_1\": int64(rand.Intn(1000)), \"key_2\": int64(rand.Intn(1000))},\n\t\tClusterTimerAckLevel:                  map[string]time.Time{\"key_1\": time.Now(), \"key_2\": time.Now()},\n\t\tOwner:                                 \"test_owner\",\n\t\tClusterReplicationLevel:               map[string]int64{\"key_1\": int64(rand.Intn(1000)), \"key_2\": int64(rand.Intn(1000))},\n\t\tPendingFailoverMarkers:                []byte(\"PendingFailoverMarkers\"),\n\t\tPendingFailoverMarkersEncoding:        \"PendingFailoverMarkersEncoding\",\n\t\tReplicationDlqAckLevel:                map[string]int64{\"key_1\": int64(rand.Intn(1000)), \"key_2\": int64(rand.Intn(1000))},\n\t\tTransferProcessingQueueStates:         []byte(\"TransferProcessingQueueStates\"),\n\t\tTransferProcessingQueueStatesEncoding: \"TransferProcessingQueueStatesEncoding\",\n\t\tTimerProcessingQueueStates:            []byte(\"TimerProcessingQueueStates\"),\n\t\tTimerProcessingQueueStatesEncoding:    \"TimerProcessingQueueStatesEncoding\",\n\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t0: &types.QueueState{\n\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: 1000,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\t\t\tTaskID: 2000,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\t\t\tTaskID: 1000,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tactual := shardInfoFromThrift(shardInfoToThrift(expected))\n\tassert.Equal(t, expected.StolenSinceRenew, actual.StolenSinceRenew)\n\tassert.Equal(t, expected.UpdatedAt.Sub(actual.UpdatedAt), time.Duration(0))\n\tassert.Equal(t, expected.ReplicationAckLevel, actual.ReplicationAckLevel)\n\tassert.Equal(t, expected.TransferAckLevel, actual.TransferAckLevel)\n\tassert.Equal(t, expected.TimerAckLevel.Sub(actual.TimerAckLevel), time.Duration(0))\n\tassert.Equal(t, expected.DomainNotificationVersion, actual.DomainNotificationVersion)\n\tassert.Equal(t, expected.ClusterTransferAckLevel, actual.ClusterTransferAckLevel)\n\tassert.Equal(t, expected.Owner, actual.Owner)\n\tassert.Equal(t, expected.ClusterReplicationLevel, actual.ClusterReplicationLevel)\n\tassert.Equal(t, expected.PendingFailoverMarkers, actual.PendingFailoverMarkers)\n\tassert.Equal(t, expected.PendingFailoverMarkersEncoding, actual.PendingFailoverMarkersEncoding)\n\tassert.Equal(t, expected.ReplicationDlqAckLevel, actual.ReplicationDlqAckLevel)\n\tassert.Equal(t, expected.TransferProcessingQueueStates, actual.TransferProcessingQueueStates)\n\tassert.Equal(t, expected.TransferProcessingQueueStatesEncoding, actual.TransferProcessingQueueStatesEncoding)\n\tassert.Equal(t, expected.TimerProcessingQueueStates, actual.TimerProcessingQueueStates)\n\tassert.Equal(t, expected.TimerProcessingQueueStatesEncoding, actual.TimerProcessingQueueStatesEncoding)\n\tassert.Len(t, actual.ClusterTimerAckLevel, 2)\n\tassert.Contains(t, actual.ClusterTimerAckLevel, \"key_1\")\n\tassert.Contains(t, actual.ClusterTimerAckLevel, \"key_2\")\n\tassert.Equal(t, expected.ClusterTimerAckLevel[\"key_1\"].Sub(actual.ClusterTimerAckLevel[\"key_1\"]), time.Duration(0))\n\tassert.Equal(t, expected.ClusterTimerAckLevel[\"key_2\"].Sub(actual.ClusterTimerAckLevel[\"key_2\"]), time.Duration(0))\n\tassert.Nil(t, shardInfoFromThrift(nil))\n\tassert.Nil(t, shardInfoToThrift(nil))\n}\n\nfunc TestDomainInfo(t *testing.T) {\n\texpected := &DomainInfo{\n\t\tName:                         \"domain_name\",\n\t\tDescription:                  \"description\",\n\t\tOwner:                        \"owner\",\n\t\tStatus:                       int32(rand.Intn(1000)),\n\t\tRetention:                    time.Duration(int64(rand.Intn(1000))),\n\t\tEmitMetric:                   true,\n\t\tArchivalBucket:               \"archival_bucket\",\n\t\tArchivalStatus:               int16(rand.Intn(1000)),\n\t\tConfigVersion:                int64(rand.Intn(1000)),\n\t\tNotificationVersion:          int64(rand.Intn(1000)),\n\t\tFailoverNotificationVersion:  int64(rand.Intn(1000)),\n\t\tFailoverVersion:              int64(rand.Intn(1000)),\n\t\tActiveClusterName:            \"ActiveClusterName\",\n\t\tActiveClustersConfig:         []byte(\"activeClustersConfig\"),\n\t\tActiveClustersConfigEncoding: \"activeClustersConfigEncoding\",\n\t\tClusters:                     []string{\"cluster_a\", \"cluster_b\"},\n\t\tData:                         map[string]string{\"key_1\": \"value_1\", \"key_2\": \"value_2\"},\n\t\tBadBinaries:                  []byte(\"BadBinaries\"),\n\t\tBadBinariesEncoding:          \"BadBinariesEncoding\",\n\t\tHistoryArchivalStatus:        int16(rand.Intn(1000)),\n\t\tHistoryArchivalURI:           \"HistoryArchivalURI\",\n\t\tVisibilityArchivalStatus:     int16(rand.Intn(1000)),\n\t\tVisibilityArchivalURI:        \"VisibilityArchivalURI\",\n\t\tFailoverEndTimestamp:         common.TimePtr(time.Now()),\n\t\tPreviousFailoverVersion:      int64(rand.Intn(1000)),\n\t\tLastUpdatedTimestamp:         time.Now(),\n\t}\n\tactual := domainInfoFromThrift(domainInfoToThrift(expected))\n\tassert.Equal(t, expected.Name, actual.Name)\n\tassert.Equal(t, expected.Description, actual.Description)\n\tassert.Equal(t, expected.Owner, actual.Owner)\n\tassert.Equal(t, expected.Status, actual.Status)\n\tassert.True(t, (expected.Retention-actual.Retention) < time.Second)\n\tassert.Equal(t, expected.EmitMetric, actual.EmitMetric)\n\tassert.Equal(t, expected.ArchivalBucket, actual.ArchivalBucket)\n\tassert.Equal(t, expected.ArchivalStatus, actual.ArchivalStatus)\n\tassert.Equal(t, expected.ConfigVersion, actual.ConfigVersion)\n\tassert.Equal(t, expected.NotificationVersion, actual.NotificationVersion)\n\tassert.Equal(t, expected.FailoverNotificationVersion, actual.FailoverNotificationVersion)\n\tassert.Equal(t, expected.ActiveClusterName, actual.ActiveClusterName)\n\tassert.Equal(t, expected.Clusters, actual.Clusters)\n\tassert.Equal(t, expected.ActiveClustersConfig, actual.ActiveClustersConfig)\n\tassert.Equal(t, expected.ActiveClustersConfigEncoding, actual.ActiveClustersConfigEncoding)\n\tassert.Equal(t, expected.Data, actual.Data)\n\tassert.Equal(t, expected.BadBinaries, actual.BadBinaries)\n\tassert.Equal(t, expected.BadBinariesEncoding, actual.BadBinariesEncoding)\n\tassert.Equal(t, expected.HistoryArchivalStatus, actual.HistoryArchivalStatus)\n\tassert.Equal(t, expected.HistoryArchivalURI, actual.HistoryArchivalURI)\n\tassert.Equal(t, expected.VisibilityArchivalStatus, actual.VisibilityArchivalStatus)\n\tassert.Equal(t, expected.VisibilityArchivalURI, actual.VisibilityArchivalURI)\n\tassert.Equal(t, expected.FailoverEndTimestamp.Sub(*actual.FailoverEndTimestamp), time.Duration(0))\n\tassert.Equal(t, expected.PreviousFailoverVersion, actual.PreviousFailoverVersion)\n\tassert.Equal(t, expected.LastUpdatedTimestamp.Sub(actual.LastUpdatedTimestamp), time.Duration(0))\n\tassert.Nil(t, domainInfoFromThrift(nil))\n\tassert.Nil(t, domainInfoToThrift(nil))\n}\n\nfunc TestDomainInfoRoundtripPanictest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin *DomainInfo\n\t}{\n\t\t\"empty roundtrip\": {\n\t\t\tin: &DomainInfo{},\n\t\t},\n\t\t\"nil roundtrip\": {\n\t\t\tin: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.in, domainInfoFromThrift(domainInfoToThrift(td.in)))\n\t\t})\n\t}\n}\n\nfunc TestHistoryTreeInfo(t *testing.T) {\n\texpected := &HistoryTreeInfo{\n\t\tCreatedTimestamp: time.Now(),\n\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t{\n\t\t\t\tBranchID:    \"branch_id\",\n\t\t\t\tBeginNodeID: int64(rand.Intn(1000)),\n\t\t\t\tEndNodeID:   int64(rand.Intn(1000)),\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchID:    \"branch_id\",\n\t\t\t\tBeginNodeID: int64(rand.Intn(1000)),\n\t\t\t\tEndNodeID:   int64(rand.Intn(1000)),\n\t\t\t},\n\t\t},\n\t\tInfo: \"info\",\n\t}\n\tactual := historyTreeInfoFromThrift(historyTreeInfoToThrift(expected))\n\tassert.Equal(t, expected.CreatedTimestamp.Sub(actual.CreatedTimestamp), time.Duration(0))\n\tassert.Equal(t, expected.Ancestors, actual.Ancestors)\n\tassert.Equal(t, expected.Info, actual.Info)\n\tassert.Nil(t, historyTreeInfoFromThrift(nil))\n\tassert.Nil(t, historyTreeInfoToThrift(nil))\n}\n\nfunc TestWorkflowExecutionInfo(t *testing.T) {\n\texpected := &WorkflowExecutionInfo{\n\t\tParentDomainID:                     UUID(uuid.New()),\n\t\tParentWorkflowID:                   \"ParentWorkflowID\",\n\t\tParentRunID:                        UUID(uuid.New()),\n\t\tInitiatedID:                        int64(rand.Intn(1000)),\n\t\tCompletionEventBatchID:             common.Int64Ptr(int64(rand.Intn(1000))),\n\t\tCompletionEvent:                    []byte(\"CompletionEvent\"),\n\t\tCompletionEventEncoding:            \"CompletionEventEncoding\",\n\t\tTaskList:                           \"TaskList\",\n\t\tTaskListKind:                       types.TaskListKindNormal,\n\t\tWorkflowTypeName:                   \"WorkflowTypeName\",\n\t\tWorkflowTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionTaskTimeout:                time.Minute * time.Duration(rand.Intn(10)),\n\t\tExecutionContext:                   []byte(\"ExecutionContext\"),\n\t\tState:                              int32(rand.Intn(1000)),\n\t\tCloseStatus:                        int32(rand.Intn(1000)),\n\t\tStartVersion:                       int64(rand.Intn(1000)),\n\t\tLastWriteEventID:                   common.Int64Ptr(int64(rand.Intn(1000))),\n\t\tLastEventTaskID:                    int64(rand.Intn(1000)),\n\t\tLastFirstEventID:                   int64(rand.Intn(1000)),\n\t\tLastProcessedEvent:                 int64(rand.Intn(1000)),\n\t\tStartTimestamp:                     time.UnixMilli(1752018142820),\n\t\tLastUpdatedTimestamp:               time.UnixMilli(1752018142821),\n\t\tDecisionVersion:                    int64(rand.Intn(1000)),\n\t\tDecisionScheduleID:                 int64(rand.Intn(1000)),\n\t\tDecisionStartedID:                  int64(rand.Intn(1000)),\n\t\tDecisionTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionAttempt:                    int64(rand.Intn(1000)),\n\t\tDecisionStartedTimestamp:           time.UnixMilli(1752018142822),\n\t\tDecisionScheduledTimestamp:         time.UnixMilli(1752018142823),\n\t\tCancelRequested:                    true,\n\t\tDecisionOriginalScheduledTimestamp: time.UnixMilli(1752018142824),\n\t\tCreateRequestID:                    \"CreateRequestID\",\n\t\tDecisionRequestID:                  \"DecisionRequestID\",\n\t\tCancelRequestID:                    \"CancelRequestID\",\n\t\tStickyTaskList:                     \"StickyTaskList\",\n\t\tStickyScheduleToStartTimeout:       time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryAttempt:                       int64(rand.Intn(1000)),\n\t\tRetryInitialInterval:               time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumInterval:               time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumAttempts:               int32(rand.Intn(1000)),\n\t\tRetryExpiration:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryBackoffCoefficient:            rand.Float64() * 1000,\n\t\tRetryExpirationTimestamp:           time.UnixMilli(1752018142825),\n\t\tRetryNonRetryableErrors:            []string{\"RetryNonRetryableErrors\"},\n\t\tHasRetryPolicy:                     true,\n\t\tCronSchedule:                       \"CronSchedule\",\n\t\tCronOverlapPolicy:                  types.CronOverlapPolicySkipped,\n\t\tEventStoreVersion:                  int32(rand.Intn(1000)),\n\t\tEventBranchToken:                   []byte(\"EventBranchToken\"),\n\t\tSignalCount:                        int64(rand.Intn(1000)),\n\t\tHistorySize:                        int64(rand.Intn(1000)),\n\t\tClientLibraryVersion:               \"ClientLibraryVersion\",\n\t\tClientFeatureVersion:               \"ClientFeatureVersion\",\n\t\tClientImpl:                         \"ClientImpl\",\n\t\tAutoResetPoints:                    []byte(\"AutoResetPoints\"),\n\t\tAutoResetPointsEncoding:            \"AutoResetPointsEncoding\",\n\t\tSearchAttributes:                   map[string][]byte{\"key_1\": []byte(\"SearchAttributes\")},\n\t\tMemo:                               map[string][]byte{\"key_1\": []byte(\"Memo\")},\n\t\tVersionHistories:                   []byte(\"VersionHistories\"),\n\t\tVersionHistoriesEncoding:           \"VersionHistoriesEncoding\",\n\t\tFirstExecutionRunID:                UUID(uuid.New()),\n\t\tPartitionConfig:                    map[string]string{\"zone\": \"dca1\"},\n\t\tChecksum:                           []byte(\"Checksum\"),\n\t\tChecksumEncoding:                   \"ChecksumEncoding\",\n\t\tIsCron:                             true,\n\t}\n\tactual := workflowExecutionInfoFromThrift(workflowExecutionInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, workflowExecutionInfoFromThrift(nil))\n\tassert.Nil(t, workflowExecutionInfoToThrift(nil))\n}\n\nfunc TestWorkflowExecutionInfo_NoTaskListKind(t *testing.T) {\n\texpected := &WorkflowExecutionInfo{\n\t\tParentDomainID:                     UUID(uuid.New()),\n\t\tParentWorkflowID:                   \"ParentWorkflowID\",\n\t\tParentRunID:                        UUID(uuid.New()),\n\t\tInitiatedID:                        int64(rand.Intn(1000)),\n\t\tCompletionEventBatchID:             common.Int64Ptr(int64(rand.Intn(1000))),\n\t\tCompletionEvent:                    []byte(\"CompletionEvent\"),\n\t\tCompletionEventEncoding:            \"CompletionEventEncoding\",\n\t\tTaskList:                           \"TaskList\",\n\t\tTaskListKind:                       types.TaskListKindNormal,\n\t\tWorkflowTypeName:                   \"WorkflowTypeName\",\n\t\tWorkflowTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionTaskTimeout:                time.Minute * time.Duration(rand.Intn(10)),\n\t\tExecutionContext:                   []byte(\"ExecutionContext\"),\n\t\tState:                              int32(rand.Intn(1000)),\n\t\tCloseStatus:                        int32(rand.Intn(1000)),\n\t\tStartVersion:                       int64(rand.Intn(1000)),\n\t\tLastWriteEventID:                   common.Int64Ptr(int64(rand.Intn(1000))),\n\t\tLastEventTaskID:                    int64(rand.Intn(1000)),\n\t\tLastFirstEventID:                   int64(rand.Intn(1000)),\n\t\tLastProcessedEvent:                 int64(rand.Intn(1000)),\n\t\tStartTimestamp:                     time.UnixMilli(1752018142820),\n\t\tLastUpdatedTimestamp:               time.UnixMilli(1752018142821),\n\t\tDecisionVersion:                    int64(rand.Intn(1000)),\n\t\tDecisionScheduleID:                 int64(rand.Intn(1000)),\n\t\tDecisionStartedID:                  int64(rand.Intn(1000)),\n\t\tDecisionTimeout:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tDecisionAttempt:                    int64(rand.Intn(1000)),\n\t\tDecisionStartedTimestamp:           time.UnixMilli(1752018142822),\n\t\tDecisionScheduledTimestamp:         time.UnixMilli(1752018142823),\n\t\tCancelRequested:                    true,\n\t\tDecisionOriginalScheduledTimestamp: time.UnixMilli(1752018142824),\n\t\tCreateRequestID:                    \"CreateRequestID\",\n\t\tDecisionRequestID:                  \"DecisionRequestID\",\n\t\tCancelRequestID:                    \"CancelRequestID\",\n\t\tStickyTaskList:                     \"StickyTaskList\",\n\t\tStickyScheduleToStartTimeout:       time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryAttempt:                       int64(rand.Intn(1000)),\n\t\tRetryInitialInterval:               time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumInterval:               time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumAttempts:               int32(rand.Intn(1000)),\n\t\tRetryExpiration:                    time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryBackoffCoefficient:            rand.Float64() * 1000,\n\t\tRetryExpirationTimestamp:           time.UnixMilli(1752018142825),\n\t\tRetryNonRetryableErrors:            []string{\"RetryNonRetryableErrors\"},\n\t\tHasRetryPolicy:                     true,\n\t\tCronSchedule:                       \"CronSchedule\",\n\t\tCronOverlapPolicy:                  types.CronOverlapPolicySkipped,\n\t\tEventStoreVersion:                  int32(rand.Intn(1000)),\n\t\tEventBranchToken:                   []byte(\"EventBranchToken\"),\n\t\tSignalCount:                        int64(rand.Intn(1000)),\n\t\tHistorySize:                        int64(rand.Intn(1000)),\n\t\tClientLibraryVersion:               \"ClientLibraryVersion\",\n\t\tClientFeatureVersion:               \"ClientFeatureVersion\",\n\t\tClientImpl:                         \"ClientImpl\",\n\t\tAutoResetPoints:                    []byte(\"AutoResetPoints\"),\n\t\tAutoResetPointsEncoding:            \"AutoResetPointsEncoding\",\n\t\tSearchAttributes:                   map[string][]byte{\"key_1\": []byte(\"SearchAttributes\")},\n\t\tMemo:                               map[string][]byte{\"key_1\": []byte(\"Memo\")},\n\t\tVersionHistories:                   []byte(\"VersionHistories\"),\n\t\tVersionHistoriesEncoding:           \"VersionHistoriesEncoding\",\n\t\tFirstExecutionRunID:                UUID(uuid.New()),\n\t\tPartitionConfig:                    map[string]string{\"zone\": \"dca1\"},\n\t\tChecksum:                           []byte(\"Checksum\"),\n\t\tChecksumEncoding:                   \"ChecksumEncoding\",\n\t\tIsCron:                             true,\n\t}\n\tthriftVersion := workflowExecutionInfoToThrift(expected)\n\t// Nil should come back as NORMAL, the default value\n\tthriftVersion.TaskListKind = nil\n\tactual := workflowExecutionInfoFromThrift(thriftVersion)\n\tassert.Equal(t, expected, actual)\n}\n\nfunc TestActivityInfo(t *testing.T) {\n\texpected := &ActivityInfo{\n\t\tVersion:                  int64(rand.Intn(1000)),\n\t\tScheduledEventBatchID:    int64(rand.Intn(1000)),\n\t\tScheduledEvent:           []byte(\"ScheduledEvent\"),\n\t\tScheduledEventEncoding:   \"ScheduledEventEncoding\",\n\t\tScheduledTimestamp:       time.UnixMilli(1752018142820),\n\t\tStartedID:                int64(rand.Intn(1000)),\n\t\tStartedEvent:             []byte(\"StartedEvent\"),\n\t\tStartedEventEncoding:     \"StartedEventEncoding\",\n\t\tStartedTimestamp:         time.UnixMilli(1752018142821),\n\t\tActivityID:               \"ActivityID\",\n\t\tRequestID:                \"RequestID\",\n\t\tScheduleToStartTimeout:   time.Minute * time.Duration(rand.Intn(10)),\n\t\tScheduleToCloseTimeout:   time.Minute * time.Duration(rand.Intn(10)),\n\t\tStartToCloseTimeout:      time.Minute * time.Duration(rand.Intn(10)),\n\t\tHeartbeatTimeout:         time.Minute * time.Duration(rand.Intn(10)),\n\t\tCancelRequested:          true,\n\t\tCancelRequestID:          int64(rand.Intn(1000)),\n\t\tTimerTaskStatus:          int32(rand.Intn(1000)),\n\t\tAttempt:                  int32(rand.Intn(1000)),\n\t\tTaskList:                 \"TaskList\",\n\t\tTaskListKind:             types.TaskListKindEphemeral,\n\t\tStartedIdentity:          \"StartedIdentity\",\n\t\tHasRetryPolicy:           true,\n\t\tRetryInitialInterval:     time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumInterval:     time.Minute * time.Duration(rand.Intn(10)),\n\t\tRetryMaximumAttempts:     int32(rand.Intn(1000)),\n\t\tRetryExpirationTimestamp: time.Time{},\n\t\tRetryBackoffCoefficient:  rand.Float64() * 1000,\n\t\tRetryNonRetryableErrors:  []string{\"RetryNonRetryableErrors\"},\n\t\tRetryLastFailureReason:   \"RetryLastFailureReason\",\n\t\tRetryLastWorkerIdentity:  \"RetryLastWorkerIdentity\",\n\t\tRetryLastFailureDetails:  []byte(\"RetryLastFailureDetails\"),\n\t}\n\tactual := activityInfoFromThrift(activityInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, activityInfoFromThrift(nil))\n\tassert.Nil(t, activityInfoToThrift(nil))\n}\n\nfunc TestChildExecutionInfo(t *testing.T) {\n\texpected := &ChildExecutionInfo{\n\t\tVersion:                int64(rand.Intn(1000)),\n\t\tInitiatedEventBatchID:  int64(rand.Intn(1000)),\n\t\tStartedID:              int64(rand.Intn(1000)),\n\t\tInitiatedEvent:         []byte(\"InitiatedEvent\"),\n\t\tInitiatedEventEncoding: \"InitiatedEventEncoding\",\n\t\tStartedWorkflowID:      \"InitiatedEventEncoding\",\n\t\tStartedRunID:           UUID(uuid.New()),\n\t\tStartedEvent:           []byte(\"StartedEvent\"),\n\t\tStartedEventEncoding:   \"StartedEventEncoding\",\n\t\tCreateRequestID:        \"CreateRequestID\",\n\t\tDomainID:               \"DomainID\",\n\t\tDomainNameDEPRECATED:   \"DomainName\",\n\t\tWorkflowTypeName:       \"WorkflowTypeName\",\n\t\tParentClosePolicy:      int32(rand.Intn(1000)),\n\t}\n\tactual := childExecutionInfoFromThrift(childExecutionInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, childExecutionInfoFromThrift(nil))\n\tassert.Nil(t, childExecutionInfoToThrift(nil))\n}\n\nfunc TestSignalInfo(t *testing.T) {\n\texpected := &SignalInfo{\n\t\tVersion:               int64(rand.Intn(1000)),\n\t\tInitiatedEventBatchID: int64(rand.Intn(1000)),\n\t\tRequestID:             \"RequestID\",\n\t\tName:                  \"Name\",\n\t\tInput:                 []byte(\"Input\"),\n\t\tControl:               []byte(\"Control\"),\n\t}\n\tactual := signalInfoFromThrift(signalInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, signalInfoFromThrift(nil))\n\tassert.Nil(t, signalInfoToThrift(nil))\n}\n\nfunc TestRequestCancelInfo(t *testing.T) {\n\texpected := &RequestCancelInfo{\n\t\tVersion:               int64(rand.Intn(1000)),\n\t\tInitiatedEventBatchID: int64(rand.Intn(1000)),\n\t\tCancelRequestID:       \"CancelRequestID\",\n\t}\n\tactual := requestCancelInfoFromThrift(requestCancelInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, requestCancelInfoFromThrift(nil))\n\tassert.Nil(t, requestCancelInfoToThrift(nil))\n}\n\nfunc TestTimerInfo(t *testing.T) {\n\texpected := &TimerInfo{\n\t\tVersion:         int64(rand.Intn(1000)),\n\t\tStartedID:       int64(rand.Intn(1000)),\n\t\tExpiryTimestamp: time.Now(),\n\t\tTaskID:          int64(rand.Intn(1000)),\n\t}\n\tactual := timerInfoFromThrift(timerInfoToThrift(expected))\n\tassert.Equal(t, expected.Version, actual.Version)\n\tassert.Equal(t, expected.StartedID, actual.StartedID)\n\tassert.Equal(t, expected.TaskID, actual.TaskID)\n\tassert.Equal(t, expected.ExpiryTimestamp.Sub(actual.ExpiryTimestamp), time.Duration(0))\n\tassert.Nil(t, timerInfoFromThrift(nil))\n\tassert.Nil(t, timerInfoToThrift(nil))\n}\n\nfunc TestTaskInfo(t *testing.T) {\n\texpected := &TaskInfo{\n\t\tWorkflowID:       \"WorkflowID\",\n\t\tRunID:            UUID(uuid.New()),\n\t\tScheduleID:       int64(rand.Intn(1000)),\n\t\tExpiryTimestamp:  time.Now(),\n\t\tCreatedTimestamp: time.Now(),\n\t\tPartitionConfig:  map[string]string{\"zone\": \"dca1\"},\n\t}\n\tactual := taskInfoFromThrift(taskInfoToThrift(expected))\n\tassert.Equal(t, expected.WorkflowID, actual.WorkflowID)\n\tassert.Equal(t, expected.RunID, actual.RunID)\n\tassert.Equal(t, expected.ScheduleID, actual.ScheduleID)\n\tassert.Equal(t, expected.ExpiryTimestamp.Sub(actual.ExpiryTimestamp), time.Duration(0))\n\tassert.Equal(t, expected.CreatedTimestamp.Sub(actual.CreatedTimestamp), time.Duration(0))\n\tassert.Equal(t, expected.PartitionConfig, actual.PartitionConfig)\n\tassert.Nil(t, taskInfoFromThrift(nil))\n\tassert.Nil(t, taskInfoToThrift(nil))\n}\n\nfunc TestTaskListInfo(t *testing.T) {\n\tcases := []*TaskListInfo{\n\t\tnil,\n\t\t{\n\t\t\tKind:            0,\n\t\t\tAckLevel:        1,\n\t\t\tExpiryTimestamp: time.UnixMicro(2),\n\t\t\tLastUpdated:     time.UnixMicro(3),\n\t\t\tAdaptivePartitionConfig: &TaskListPartitionConfig{\n\t\t\t\tVersion:           4,\n\t\t\t\tNumReadPartitions: 1,\n\t\t\t\tReadPartitions: map[int32]*TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t\tWritePartitions: map[int32]*TaskListPartition{\n\t\t\t\t\t0: {\n\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t},\n\t\t\t\t\t1: {\n\t\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tKind:            0,\n\t\t\tAckLevel:        1,\n\t\t\tExpiryTimestamp: time.UnixMicro(2),\n\t\t\tLastUpdated:     time.UnixMicro(3),\n\t\t\tAdaptivePartitionConfig: &TaskListPartitionConfig{\n\t\t\t\tVersion:            4,\n\t\t\t\tNumReadPartitions:  10,\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t},\n\t\t},\n\t}\n\tfor i, info := range cases {\n\t\tassert.Equal(t, info, taskListInfoFromThrift(taskListInfoToThrift(info)), \"case %d\", i)\n\t}\n}\n\nfunc TestTransferTaskInfo(t *testing.T) {\n\texpected := &TransferTaskInfo{\n\t\tDomainID:                UUID(uuid.New()),\n\t\tWorkflowID:              \"WorkflowID\",\n\t\tRunID:                   UUID(uuid.New()),\n\t\tTaskType:                int16(rand.Intn(1000)),\n\t\tTargetDomainID:          UUID(uuid.New()),\n\t\tTargetDomainIDs:         []UUID{UUID(uuid.New()), UUID(uuid.New())},\n\t\tTargetWorkflowID:        \"TargetWorkflowID\",\n\t\tTargetRunID:             UUID(uuid.New()),\n\t\tTaskList:                \"TaskList\",\n\t\tTargetChildWorkflowOnly: true,\n\t\tScheduleID:              int64(rand.Intn(1000)),\n\t\tVersion:                 int64(rand.Intn(1000)),\n\t\tOriginalTaskList:        \"OriginalTaskList\",\n\t}\n\tactual := transferTaskInfoFromThrift(transferTaskInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, transferTaskInfoFromThrift(nil))\n\tassert.Nil(t, transferTaskInfoToThrift(nil))\n}\n\nfunc TestTimerTaskInfo(t *testing.T) {\n\texpected := &TimerTaskInfo{\n\t\tDomainID:        UUID(uuid.New()),\n\t\tWorkflowID:      \"WorkflowID\",\n\t\tRunID:           UUID(uuid.New()),\n\t\tTaskType:        int16(rand.Intn(1000)),\n\t\tTimeoutType:     common.Int16Ptr(int16(rand.Intn(1000))),\n\t\tVersion:         int64(rand.Intn(1000)),\n\t\tScheduleAttempt: int64(rand.Intn(1000)),\n\t\tEventID:         int64(rand.Intn(1000)),\n\t\tTaskList:        \"TaskList\",\n\t}\n\tactual := timerTaskInfoFromThrift(timerTaskInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, timerTaskInfoFromThrift(nil))\n\tassert.Nil(t, timerTaskInfoToThrift(nil))\n}\n\nfunc TestReplicationTaskInfo(t *testing.T) {\n\texpected := &ReplicationTaskInfo{\n\t\tDomainID:                UUID(uuid.New()),\n\t\tWorkflowID:              \"WorkflowID\",\n\t\tRunID:                   UUID(uuid.New()),\n\t\tTaskType:                int16(rand.Intn(1000)),\n\t\tVersion:                 int64(rand.Intn(1000)),\n\t\tFirstEventID:            int64(rand.Intn(1000)),\n\t\tNextEventID:             int64(rand.Intn(1000)),\n\t\tScheduledID:             int64(rand.Intn(1000)),\n\t\tEventStoreVersion:       int32(rand.Intn(1000)),\n\t\tNewRunEventStoreVersion: int32(rand.Intn(1000)),\n\t\tBranchToken:             []byte(\"BranchToken\"),\n\t\tNewRunBranchToken:       []byte(\"NewRunBranchToken\"),\n\t}\n\tactual := replicationTaskInfoFromThrift(replicationTaskInfoToThrift(expected))\n\tassert.Equal(t, expected, actual)\n\tassert.Nil(t, replicationTaskInfoFromThrift(nil))\n\tassert.Nil(t, replicationTaskInfoToThrift(nil))\n}\n\nfunc TestCronOverlapPolicyFromThrift(t *testing.T) {\n\tcases := []struct {\n\t\tthrift *shared.CronOverlapPolicy\n\t\tactual types.CronOverlapPolicy\n\t}{\n\t\t{nil, types.CronOverlapPolicySkipped},\n\t\t{shared.CronOverlapPolicyBufferone.Ptr(), types.CronOverlapPolicyBufferOne},\n\t\t{shared.CronOverlapPolicySkipped.Ptr(), types.CronOverlapPolicySkipped},\n\t}\n\tfor _, c := range cases {\n\t\tassert.Equal(t, c.actual, cronOverlapPolicyFromThrift(c.thrift))\n\t}\n}\n"
  },
  {
    "path": "common/persistence/serialization/uuid.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage serialization\n\nimport (\n\t\"database/sql/driver\"\n\t\"encoding/hex\"\n\n\t\"github.com/google/uuid\"\n)\n\n// UUID represents a 16-byte universally unique identifier\n// this type is a wrapper around google/uuid with the following differences\n//   - type is a byte slice instead of [16]byte\n//   - db serialization converts uuid to bytes as opposed to string\ntype UUID []byte\n\n// MustParseUUID returns a UUID parsed from the given string representation\n// returns nil if the input is empty string\n// panics if the given input is malformed\nfunc MustParseUUID(s string) UUID {\n\tif s == \"\" {\n\t\treturn nil\n\t}\n\tu := uuid.MustParse(s)\n\treturn u[:]\n}\n\n// UUIDPtr simply returns a pointer for the given value type\nfunc UUIDPtr(u UUID) *UUID {\n\treturn &u\n}\n\n// String returns the 36 byet hexstring representation of this uuid\n// return empty string if this uuid is nil\nfunc (u UUID) String() string {\n\tif len(u) != 16 {\n\t\treturn \"\"\n\t}\n\tvar buf [36]byte\n\tu.encodeHex(buf[:])\n\treturn string(buf[:])\n}\n\n// Scan implements sql.Scanner interface to allow this type to be\n// parsed transparently by database drivers\nfunc (u *UUID) Scan(src interface{}) error {\n\tif src == nil {\n\t\treturn nil\n\t}\n\tguuid := &uuid.UUID{}\n\tif err := guuid.Scan(src); err != nil {\n\t\treturn err\n\t}\n\t*u = (*guuid)[:]\n\treturn nil\n}\n\n// Value implements sql.Valuer so that UUIDs can be written to databases\n// transparently. This method returns a byte slice representation of uuid\nfunc (u UUID) Value() (driver.Value, error) {\n\treturn []byte(u), nil\n}\n\nfunc (u UUID) encodeHex(dst []byte) {\n\thex.Encode(dst, u[:4])\n\tdst[8] = '-'\n\thex.Encode(dst[9:13], u[4:6])\n\tdst[13] = '-'\n\thex.Encode(dst[14:18], u[6:8])\n\tdst[18] = '-'\n\thex.Encode(dst[19:23], u[8:10])\n\tdst[23] = '-'\n\thex.Encode(dst[24:], u[10:])\n}\n"
  },
  {
    "path": "common/persistence/serializer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/golang/snappy\"\n\n\t\"github.com/uber/cadence/.gen/go/config\"\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination serializer_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence PayloadSerializer\n\ntype (\n\t// PayloadSerializer is used by persistence to serialize/deserialize history event(s) and others\n\t// It will only be used inside persistence, so that serialize/deserialize is transparent for application\n\tPayloadSerializer interface {\n\t\t// serialize/deserialize history events\n\t\tSerializeBatchEvents(batch []*types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeBatchEvents(data *DataBlob) ([]*types.HistoryEvent, error)\n\n\t\t// serialize/deserialize a single history event\n\t\tSerializeEvent(event *types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeEvent(data *DataBlob) (*types.HistoryEvent, error)\n\n\t\t// serialize/deserialize visibility memo fields\n\t\tSerializeVisibilityMemo(memo *types.Memo, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeVisibilityMemo(data *DataBlob) (*types.Memo, error)\n\n\t\t// serialize/deserialize reset points\n\t\tSerializeResetPoints(event *types.ResetPoints, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeResetPoints(data *DataBlob) (*types.ResetPoints, error)\n\n\t\t// serialize/deserialize bad binaries\n\t\tSerializeBadBinaries(event *types.BadBinaries, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeBadBinaries(data *DataBlob) (*types.BadBinaries, error)\n\n\t\t// serialize/deserialize version histories\n\t\tSerializeVersionHistories(histories *types.VersionHistories, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeVersionHistories(data *DataBlob) (*types.VersionHistories, error)\n\n\t\t// serialize/deserialize pending failover markers\n\t\tSerializePendingFailoverMarkers(markers []*types.FailoverMarkerAttributes, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializePendingFailoverMarkers(data *DataBlob) ([]*types.FailoverMarkerAttributes, error)\n\n\t\t// serialize/deserialize processing queue statesss\n\t\tSerializeProcessingQueueStates(states *types.ProcessingQueueStates, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeProcessingQueueStates(data *DataBlob) (*types.ProcessingQueueStates, error)\n\n\t\t// serialize/deserialize DynamicConfigBlob\n\t\tSerializeDynamicConfigBlob(blob *types.DynamicConfigBlob, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeDynamicConfigBlob(data *DataBlob) (*types.DynamicConfigBlob, error)\n\n\t\t// serialize/deserialize IsolationGroupConfiguration\n\t\tSerializeIsolationGroups(event *types.IsolationGroupConfiguration, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeIsolationGroups(data *DataBlob) (*types.IsolationGroupConfiguration, error)\n\n\t\t// serialize/deserialize async workflow configuration\n\t\tSerializeAsyncWorkflowsConfig(config *types.AsyncWorkflowConfiguration, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeAsyncWorkflowsConfig(data *DataBlob) (*types.AsyncWorkflowConfiguration, error)\n\n\t\t// serialize/deserialize checksum\n\t\tSerializeChecksum(sum checksum.Checksum, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeChecksum(data *DataBlob) (checksum.Checksum, error)\n\n\t\t// serialize/deserialize active clusters config\n\t\tSerializeActiveClusters(activeClusters *types.ActiveClusters, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeActiveClusters(data *DataBlob) (*types.ActiveClusters, error)\n\n\t\t// serialize/deserialize active cluster selection policy\n\t\tSerializeActiveClusterSelectionPolicy(policy *types.ActiveClusterSelectionPolicy, encodingType constants.EncodingType) (*DataBlob, error)\n\t\tDeserializeActiveClusterSelectionPolicy(data *DataBlob) (*types.ActiveClusterSelectionPolicy, error)\n\t}\n\n\t// CadenceSerializationError is an error type for cadence serialization\n\tCadenceSerializationError struct {\n\t\tmsg string\n\t}\n\n\t// CadenceDeserializationError is an error type for cadence deserialization\n\tCadenceDeserializationError struct {\n\t\tmsg string\n\t}\n\n\t// UnknownEncodingTypeError is an error type for unknown or unsupported encoding type\n\tUnknownEncodingTypeError struct {\n\t\tencodingType constants.EncodingType\n\t}\n\n\tserializerImpl struct {\n\t\tthriftrwEncoder codec.BinaryEncoder\n\t}\n)\n\n// NewPayloadSerializer returns a PayloadSerializer\nfunc NewPayloadSerializer() PayloadSerializer {\n\treturn &serializerImpl{\n\t\tthriftrwEncoder: codec.NewThriftRWEncoder(),\n\t}\n}\n\nfunc (t *serializerImpl) SerializeBatchEvents(events []*types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error) {\n\treturn t.serialize(events, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeBatchEvents(data *DataBlob) ([]*types.HistoryEvent, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\tvar events []*types.HistoryEvent\n\tif data != nil && len(data.Data) == 0 {\n\t\treturn events, nil\n\t}\n\terr := t.deserialize(data, &events)\n\treturn events, err\n}\n\nfunc (t *serializerImpl) SerializeEvent(event *types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif event == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(event, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeEvent(data *DataBlob) (*types.HistoryEvent, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\tvar event types.HistoryEvent\n\terr := t.deserialize(data, &event)\n\treturn &event, err\n}\n\nfunc (t *serializerImpl) SerializeResetPoints(rp *types.ResetPoints, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif rp == nil {\n\t\trp = &types.ResetPoints{}\n\t}\n\treturn t.serialize(rp, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeResetPoints(data *DataBlob) (*types.ResetPoints, error) {\n\tvar rp types.ResetPoints\n\terr := t.deserialize(data, &rp)\n\treturn &rp, err\n}\n\nfunc (t *serializerImpl) SerializeBadBinaries(bb *types.BadBinaries, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif bb == nil {\n\t\tbb = &types.BadBinaries{}\n\t}\n\treturn t.serialize(bb, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeBadBinaries(data *DataBlob) (*types.BadBinaries, error) {\n\tvar bb types.BadBinaries\n\terr := t.deserialize(data, &bb)\n\treturn &bb, err\n}\n\nfunc (t *serializerImpl) SerializeVisibilityMemo(memo *types.Memo, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif memo == nil {\n\t\t// Return nil here to be consistent with Event\n\t\t// This check is not duplicate as check in following serialize\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(memo, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeVisibilityMemo(data *DataBlob) (*types.Memo, error) {\n\tvar memo types.Memo\n\terr := t.deserialize(data, &memo)\n\treturn &memo, err\n}\n\nfunc (t *serializerImpl) SerializeVersionHistories(histories *types.VersionHistories, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif histories == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(histories, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeVersionHistories(data *DataBlob) (*types.VersionHistories, error) {\n\tvar histories types.VersionHistories\n\terr := t.deserialize(data, &histories)\n\treturn &histories, err\n}\n\nfunc (t *serializerImpl) SerializePendingFailoverMarkers(markers []*types.FailoverMarkerAttributes, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif markers == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(markers, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializePendingFailoverMarkers(data *DataBlob) ([]*types.FailoverMarkerAttributes, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\tvar markers []*types.FailoverMarkerAttributes\n\tif data != nil && len(data.Data) == 0 {\n\t\treturn markers, nil\n\t}\n\terr := t.deserialize(data, &markers)\n\treturn markers, err\n}\n\nfunc (t *serializerImpl) SerializeProcessingQueueStates(states *types.ProcessingQueueStates, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif states == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(states, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeProcessingQueueStates(data *DataBlob) (*types.ProcessingQueueStates, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar states types.ProcessingQueueStates\n\tif data != nil && len(data.Data) == 0 {\n\t\treturn &states, nil\n\t}\n\terr := t.deserialize(data, &states)\n\treturn &states, err\n}\n\nfunc (t *serializerImpl) SerializeDynamicConfigBlob(blob *types.DynamicConfigBlob, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif blob == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(blob, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeDynamicConfigBlob(data *DataBlob) (*types.DynamicConfigBlob, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar blob types.DynamicConfigBlob\n\tif len(data.Data) == 0 {\n\t\treturn &blob, nil\n\t}\n\n\terr := t.deserialize(data, &blob)\n\treturn &blob, err\n}\n\nfunc (t *serializerImpl) SerializeIsolationGroups(c *types.IsolationGroupConfiguration, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif c == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(c, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeIsolationGroups(data *DataBlob) (*types.IsolationGroupConfiguration, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar cfg types.IsolationGroupConfiguration\n\tif len(data.Data) == 0 {\n\t\treturn &cfg, nil\n\t}\n\n\terr := t.deserialize(data, &cfg)\n\treturn &cfg, err\n}\n\nfunc (t *serializerImpl) SerializeAsyncWorkflowsConfig(c *types.AsyncWorkflowConfiguration, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif c == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(c, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeAsyncWorkflowsConfig(data *DataBlob) (*types.AsyncWorkflowConfiguration, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar cfg types.AsyncWorkflowConfiguration\n\tif len(data.Data) == 0 {\n\t\treturn &cfg, nil\n\t}\n\n\terr := t.deserialize(data, &cfg)\n\treturn &cfg, err\n}\n\nfunc (t *serializerImpl) SerializeChecksum(sum checksum.Checksum, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif len(sum.Value) == 0 {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(sum, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeChecksum(data *DataBlob) (checksum.Checksum, error) {\n\tif data == nil {\n\t\treturn checksum.Checksum{}, nil\n\t}\n\n\tvar sum checksum.Checksum\n\tif len(data.Data) == 0 {\n\t\treturn sum, nil\n\t}\n\n\terr := t.deserialize(data, &sum)\n\tif err != nil {\n\t\treturn checksum.Checksum{}, err\n\t}\n\treturn sum, err\n}\n\nfunc (t *serializerImpl) SerializeActiveClusters(activeClusters *types.ActiveClusters, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif activeClusters == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(activeClusters, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeActiveClusters(data *DataBlob) (*types.ActiveClusters, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar activeClusters types.ActiveClusters\n\tif len(data.Data) == 0 {\n\t\treturn &activeClusters, nil\n\t}\n\terr := t.deserialize(data, &activeClusters)\n\treturn &activeClusters, err\n}\n\nfunc (t *serializerImpl) SerializeActiveClusterSelectionPolicy(policy *types.ActiveClusterSelectionPolicy, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif policy == nil {\n\t\treturn nil, nil\n\t}\n\treturn t.serialize(policy, encodingType)\n}\n\nfunc (t *serializerImpl) DeserializeActiveClusterSelectionPolicy(data *DataBlob) (*types.ActiveClusterSelectionPolicy, error) {\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar policy types.ActiveClusterSelectionPolicy\n\terr := t.deserialize(data, &policy)\n\treturn &policy, err\n}\n\nfunc (t *serializerImpl) serialize(input interface{}, encodingType constants.EncodingType) (*DataBlob, error) {\n\tif input == nil {\n\t\treturn nil, nil\n\t}\n\n\tvar data []byte\n\tvar err error\n\n\tswitch encodingType {\n\tcase constants.EncodingTypeThriftRW:\n\t\tdata, err = t.thriftrwEncode(input)\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\tdata, err = t.thriftrwsnappyEncode(input)\n\tcase constants.EncodingTypeJSON, constants.EncodingTypeUnknown, constants.EncodingTypeEmpty: // For backward-compatibility\n\t\tencodingType = constants.EncodingTypeJSON\n\t\tdata, err = json.Marshal(input)\n\tdefault:\n\t\treturn nil, NewUnknownEncodingTypeError(encodingType)\n\t}\n\n\tif err != nil {\n\t\treturn nil, NewCadenceSerializationError(err.Error())\n\t}\n\treturn NewDataBlob(data, encodingType), nil\n}\n\nfunc (t *serializerImpl) thriftrwEncode(input interface{}) ([]byte, error) {\n\n\tswitch input := input.(type) {\n\tcase []*types.HistoryEvent:\n\t\treturn t.thriftrwEncoder.Encode(&workflow.History{Events: thrift.FromHistoryEventArray(input)})\n\tcase *types.HistoryEvent:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromHistoryEvent(input))\n\tcase *types.Memo:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromMemo(input))\n\tcase *types.ResetPoints:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromResetPoints(input))\n\tcase *types.BadBinaries:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromBadBinaries(input))\n\tcase *types.VersionHistories:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromVersionHistories(input))\n\tcase []*types.FailoverMarkerAttributes:\n\t\treturn t.thriftrwEncoder.Encode(&replicator.FailoverMarkers{FailoverMarkers: thrift.FromFailoverMarkerAttributesArray(input)})\n\tcase *types.ProcessingQueueStates:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromProcessingQueueStates(input))\n\tcase *types.DynamicConfigBlob:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromDynamicConfigBlob(input))\n\tcase *types.IsolationGroupConfiguration:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromIsolationGroupConfig(input))\n\tcase *types.AsyncWorkflowConfiguration:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromDomainAsyncWorkflowConfiguraton(input))\n\tcase *types.ActiveClusters:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromActiveClusters(input))\n\tcase *types.ActiveClusterSelectionPolicy:\n\t\treturn t.thriftrwEncoder.Encode(thrift.FromActiveClusterSelectionPolicy(input))\n\tdefault:\n\t\treturn nil, nil\n\t}\n}\n\nfunc (t *serializerImpl) deserialize(data *DataBlob, target interface{}) error {\n\tif data == nil {\n\t\treturn nil\n\t}\n\tif len(data.Data) == 0 {\n\t\treturn NewCadenceDeserializationError(\"DeserializeEvent empty data\")\n\t}\n\tvar err error\n\n\tswitch data.GetEncoding() {\n\tcase constants.EncodingTypeThriftRW:\n\t\terr = t.thriftrwDecode(data.Data, target)\n\tcase constants.EncodingTypeThriftRWSnappy:\n\t\terr = t.thriftrwsnappyDecode(data.Data, target)\n\tcase constants.EncodingTypeJSON, constants.EncodingTypeUnknown, constants.EncodingTypeEmpty: // For backward-compatibility\n\t\terr = json.Unmarshal(data.Data, target)\n\tdefault:\n\t\treturn NewUnknownEncodingTypeError(data.GetEncoding())\n\t}\n\n\tif err != nil {\n\t\treturn NewCadenceDeserializationError(fmt.Sprintf(\"DeserializeBatchEvents encoding: \\\"%v\\\", error: %v\", data.Encoding, err.Error()))\n\t}\n\treturn nil\n}\n\nfunc (t *serializerImpl) thriftrwDecode(data []byte, target interface{}) error {\n\tswitch target := target.(type) {\n\tcase *[]*types.HistoryEvent:\n\t\tthriftTarget := workflow.History{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = thrift.ToHistoryEventArray(thriftTarget.GetEvents())\n\t\treturn nil\n\tcase *types.HistoryEvent:\n\t\tthriftTarget := workflow.HistoryEvent{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToHistoryEvent(&thriftTarget)\n\t\treturn nil\n\tcase *types.Memo:\n\t\tthriftTarget := workflow.Memo{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToMemo(&thriftTarget)\n\t\treturn nil\n\tcase *types.ResetPoints:\n\t\tthriftTarget := workflow.ResetPoints{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToResetPoints(&thriftTarget)\n\t\treturn nil\n\tcase *types.BadBinaries:\n\t\tthriftTarget := workflow.BadBinaries{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToBadBinaries(&thriftTarget)\n\t\treturn nil\n\tcase *types.VersionHistories:\n\t\tthriftTarget := workflow.VersionHistories{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToVersionHistories(&thriftTarget)\n\t\treturn nil\n\tcase *[]*types.FailoverMarkerAttributes:\n\t\tthriftTarget := replicator.FailoverMarkers{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = thrift.ToFailoverMarkerAttributesArray(thriftTarget.GetFailoverMarkers())\n\t\treturn nil\n\tcase *types.ProcessingQueueStates:\n\t\tthriftTarget := history.ProcessingQueueStates{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToProcessingQueueStates(&thriftTarget)\n\t\treturn nil\n\tcase *types.DynamicConfigBlob:\n\t\tthriftTarget := config.DynamicConfigBlob{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToDynamicConfigBlob(&thriftTarget)\n\t\treturn nil\n\tcase *types.IsolationGroupConfiguration:\n\t\tthriftTarget := workflow.IsolationGroupConfiguration{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToIsolationGroupConfig(&thriftTarget)\n\t\treturn nil\n\tcase *types.AsyncWorkflowConfiguration:\n\t\tthriftTarget := workflow.AsyncWorkflowConfiguration{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToDomainAsyncWorkflowConfiguraton(&thriftTarget)\n\t\treturn nil\n\tcase *types.ActiveClusters:\n\t\tthriftTarget := workflow.ActiveClusters{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToActiveClusters(&thriftTarget)\n\t\treturn nil\n\tcase *types.ActiveClusterSelectionPolicy:\n\t\tthriftTarget := workflow.ActiveClusterSelectionPolicy{}\n\t\tif err := t.thriftrwEncoder.Decode(data, &thriftTarget); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t*target = *thrift.ToActiveClusterSelectionPolicy(&thriftTarget)\n\t\treturn nil\n\tdefault:\n\t\treturn nil\n\t}\n}\n\nfunc (t *serializerImpl) thriftrwsnappyEncode(input interface{}) ([]byte, error) {\n\tdata, err := t.thriftrwEncode(input)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif data == nil {\n\t\treturn nil, nil\n\t}\n\n\treturn snappy.Encode(nil, data), nil\n}\n\nfunc (t *serializerImpl) thriftrwsnappyDecode(data []byte, target interface{}) error {\n\tdecompressed, err := snappy.Decode(nil, data)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn t.thriftrwDecode(decompressed, target)\n}\n\n// NewUnknownEncodingTypeError returns a new instance of encoding type error\nfunc NewUnknownEncodingTypeError(encodingType constants.EncodingType) error {\n\treturn &UnknownEncodingTypeError{encodingType: encodingType}\n}\n\nfunc (e *UnknownEncodingTypeError) Error() string {\n\treturn fmt.Sprintf(\"unknown or unsupported encoding type %v\", e.encodingType)\n}\n\n// NewCadenceSerializationError returns a CadenceSerializationError\nfunc NewCadenceSerializationError(msg string) *CadenceSerializationError {\n\treturn &CadenceSerializationError{msg: msg}\n}\n\nfunc (e *CadenceSerializationError) Error() string {\n\treturn fmt.Sprintf(\"cadence serialization error: %v\", e.msg)\n}\n\n// NewCadenceDeserializationError returns a CadenceDeserializationError\nfunc NewCadenceDeserializationError(msg string) *CadenceDeserializationError {\n\treturn &CadenceDeserializationError{msg: msg}\n}\n\nfunc (e *CadenceDeserializationError) Error() string {\n\treturn fmt.Sprintf(\"cadence deserialization error: %v\", e.msg)\n}\n"
  },
  {
    "path": "common/persistence/serializer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence (interfaces: PayloadSerializer)\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -destination serializer_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence PayloadSerializer\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tchecksum \"github.com/uber/cadence/common/checksum\"\n\tconstants \"github.com/uber/cadence/common/constants\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockPayloadSerializer is a mock of PayloadSerializer interface.\ntype MockPayloadSerializer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPayloadSerializerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPayloadSerializerMockRecorder is the mock recorder for MockPayloadSerializer.\ntype MockPayloadSerializerMockRecorder struct {\n\tmock *MockPayloadSerializer\n}\n\n// NewMockPayloadSerializer creates a new mock instance.\nfunc NewMockPayloadSerializer(ctrl *gomock.Controller) *MockPayloadSerializer {\n\tmock := &MockPayloadSerializer{ctrl: ctrl}\n\tmock.recorder = &MockPayloadSerializerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPayloadSerializer) EXPECT() *MockPayloadSerializerMockRecorder {\n\treturn m.recorder\n}\n\n// DeserializeActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeActiveClusterSelectionPolicy(data *DataBlob) (*types.ActiveClusterSelectionPolicy, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeActiveClusterSelectionPolicy\", data)\n\tret0, _ := ret[0].(*types.ActiveClusterSelectionPolicy)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeActiveClusterSelectionPolicy indicates an expected call of DeserializeActiveClusterSelectionPolicy.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeActiveClusterSelectionPolicy(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeActiveClusterSelectionPolicy\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeActiveClusterSelectionPolicy), data)\n}\n\n// DeserializeActiveClusters mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeActiveClusters(data *DataBlob) (*types.ActiveClusters, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeActiveClusters\", data)\n\tret0, _ := ret[0].(*types.ActiveClusters)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeActiveClusters indicates an expected call of DeserializeActiveClusters.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeActiveClusters(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeActiveClusters\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeActiveClusters), data)\n}\n\n// DeserializeAsyncWorkflowsConfig mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeAsyncWorkflowsConfig(data *DataBlob) (*types.AsyncWorkflowConfiguration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeAsyncWorkflowsConfig\", data)\n\tret0, _ := ret[0].(*types.AsyncWorkflowConfiguration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeAsyncWorkflowsConfig indicates an expected call of DeserializeAsyncWorkflowsConfig.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeAsyncWorkflowsConfig(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeAsyncWorkflowsConfig\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeAsyncWorkflowsConfig), data)\n}\n\n// DeserializeBadBinaries mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeBadBinaries(data *DataBlob) (*types.BadBinaries, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeBadBinaries\", data)\n\tret0, _ := ret[0].(*types.BadBinaries)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeBadBinaries indicates an expected call of DeserializeBadBinaries.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeBadBinaries(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeBadBinaries\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeBadBinaries), data)\n}\n\n// DeserializeBatchEvents mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeBatchEvents(data *DataBlob) ([]*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeBatchEvents\", data)\n\tret0, _ := ret[0].([]*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeBatchEvents indicates an expected call of DeserializeBatchEvents.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeBatchEvents(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeBatchEvents\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeBatchEvents), data)\n}\n\n// DeserializeChecksum mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeChecksum(data *DataBlob) (checksum.Checksum, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeChecksum\", data)\n\tret0, _ := ret[0].(checksum.Checksum)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeChecksum indicates an expected call of DeserializeChecksum.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeChecksum(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeChecksum\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeChecksum), data)\n}\n\n// DeserializeDynamicConfigBlob mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeDynamicConfigBlob(data *DataBlob) (*types.DynamicConfigBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeDynamicConfigBlob\", data)\n\tret0, _ := ret[0].(*types.DynamicConfigBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeDynamicConfigBlob indicates an expected call of DeserializeDynamicConfigBlob.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeDynamicConfigBlob(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeDynamicConfigBlob\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeDynamicConfigBlob), data)\n}\n\n// DeserializeEvent mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeEvent(data *DataBlob) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeEvent\", data)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeEvent indicates an expected call of DeserializeEvent.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeEvent(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeEvent\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeEvent), data)\n}\n\n// DeserializeIsolationGroups mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeIsolationGroups(data *DataBlob) (*types.IsolationGroupConfiguration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeIsolationGroups\", data)\n\tret0, _ := ret[0].(*types.IsolationGroupConfiguration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeIsolationGroups indicates an expected call of DeserializeIsolationGroups.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeIsolationGroups(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeIsolationGroups\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeIsolationGroups), data)\n}\n\n// DeserializePendingFailoverMarkers mocks base method.\nfunc (m *MockPayloadSerializer) DeserializePendingFailoverMarkers(data *DataBlob) ([]*types.FailoverMarkerAttributes, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializePendingFailoverMarkers\", data)\n\tret0, _ := ret[0].([]*types.FailoverMarkerAttributes)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializePendingFailoverMarkers indicates an expected call of DeserializePendingFailoverMarkers.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializePendingFailoverMarkers(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializePendingFailoverMarkers\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializePendingFailoverMarkers), data)\n}\n\n// DeserializeProcessingQueueStates mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeProcessingQueueStates(data *DataBlob) (*types.ProcessingQueueStates, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeProcessingQueueStates\", data)\n\tret0, _ := ret[0].(*types.ProcessingQueueStates)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeProcessingQueueStates indicates an expected call of DeserializeProcessingQueueStates.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeProcessingQueueStates(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeProcessingQueueStates\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeProcessingQueueStates), data)\n}\n\n// DeserializeResetPoints mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeResetPoints(data *DataBlob) (*types.ResetPoints, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeResetPoints\", data)\n\tret0, _ := ret[0].(*types.ResetPoints)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeResetPoints indicates an expected call of DeserializeResetPoints.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeResetPoints(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeResetPoints\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeResetPoints), data)\n}\n\n// DeserializeVersionHistories mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeVersionHistories(data *DataBlob) (*types.VersionHistories, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeVersionHistories\", data)\n\tret0, _ := ret[0].(*types.VersionHistories)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeVersionHistories indicates an expected call of DeserializeVersionHistories.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeVersionHistories(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeVersionHistories\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeVersionHistories), data)\n}\n\n// DeserializeVisibilityMemo mocks base method.\nfunc (m *MockPayloadSerializer) DeserializeVisibilityMemo(data *DataBlob) (*types.Memo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeVisibilityMemo\", data)\n\tret0, _ := ret[0].(*types.Memo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeVisibilityMemo indicates an expected call of DeserializeVisibilityMemo.\nfunc (mr *MockPayloadSerializerMockRecorder) DeserializeVisibilityMemo(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeVisibilityMemo\", reflect.TypeOf((*MockPayloadSerializer)(nil).DeserializeVisibilityMemo), data)\n}\n\n// SerializeActiveClusterSelectionPolicy mocks base method.\nfunc (m *MockPayloadSerializer) SerializeActiveClusterSelectionPolicy(policy *types.ActiveClusterSelectionPolicy, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeActiveClusterSelectionPolicy\", policy, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeActiveClusterSelectionPolicy indicates an expected call of SerializeActiveClusterSelectionPolicy.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeActiveClusterSelectionPolicy(policy, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeActiveClusterSelectionPolicy\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeActiveClusterSelectionPolicy), policy, encodingType)\n}\n\n// SerializeActiveClusters mocks base method.\nfunc (m *MockPayloadSerializer) SerializeActiveClusters(activeClusters *types.ActiveClusters, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeActiveClusters\", activeClusters, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeActiveClusters indicates an expected call of SerializeActiveClusters.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeActiveClusters(activeClusters, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeActiveClusters\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeActiveClusters), activeClusters, encodingType)\n}\n\n// SerializeAsyncWorkflowsConfig mocks base method.\nfunc (m *MockPayloadSerializer) SerializeAsyncWorkflowsConfig(config *types.AsyncWorkflowConfiguration, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeAsyncWorkflowsConfig\", config, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeAsyncWorkflowsConfig indicates an expected call of SerializeAsyncWorkflowsConfig.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeAsyncWorkflowsConfig(config, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeAsyncWorkflowsConfig\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeAsyncWorkflowsConfig), config, encodingType)\n}\n\n// SerializeBadBinaries mocks base method.\nfunc (m *MockPayloadSerializer) SerializeBadBinaries(event *types.BadBinaries, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeBadBinaries\", event, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeBadBinaries indicates an expected call of SerializeBadBinaries.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeBadBinaries(event, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeBadBinaries\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeBadBinaries), event, encodingType)\n}\n\n// SerializeBatchEvents mocks base method.\nfunc (m *MockPayloadSerializer) SerializeBatchEvents(batch []*types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeBatchEvents\", batch, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeBatchEvents indicates an expected call of SerializeBatchEvents.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeBatchEvents(batch, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeBatchEvents\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeBatchEvents), batch, encodingType)\n}\n\n// SerializeChecksum mocks base method.\nfunc (m *MockPayloadSerializer) SerializeChecksum(sum checksum.Checksum, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeChecksum\", sum, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeChecksum indicates an expected call of SerializeChecksum.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeChecksum(sum, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeChecksum\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeChecksum), sum, encodingType)\n}\n\n// SerializeDynamicConfigBlob mocks base method.\nfunc (m *MockPayloadSerializer) SerializeDynamicConfigBlob(blob *types.DynamicConfigBlob, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeDynamicConfigBlob\", blob, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeDynamicConfigBlob indicates an expected call of SerializeDynamicConfigBlob.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeDynamicConfigBlob(blob, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeDynamicConfigBlob\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeDynamicConfigBlob), blob, encodingType)\n}\n\n// SerializeEvent mocks base method.\nfunc (m *MockPayloadSerializer) SerializeEvent(event *types.HistoryEvent, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeEvent\", event, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeEvent indicates an expected call of SerializeEvent.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeEvent(event, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeEvent\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeEvent), event, encodingType)\n}\n\n// SerializeIsolationGroups mocks base method.\nfunc (m *MockPayloadSerializer) SerializeIsolationGroups(event *types.IsolationGroupConfiguration, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeIsolationGroups\", event, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeIsolationGroups indicates an expected call of SerializeIsolationGroups.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeIsolationGroups(event, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeIsolationGroups\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeIsolationGroups), event, encodingType)\n}\n\n// SerializePendingFailoverMarkers mocks base method.\nfunc (m *MockPayloadSerializer) SerializePendingFailoverMarkers(markers []*types.FailoverMarkerAttributes, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializePendingFailoverMarkers\", markers, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializePendingFailoverMarkers indicates an expected call of SerializePendingFailoverMarkers.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializePendingFailoverMarkers(markers, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializePendingFailoverMarkers\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializePendingFailoverMarkers), markers, encodingType)\n}\n\n// SerializeProcessingQueueStates mocks base method.\nfunc (m *MockPayloadSerializer) SerializeProcessingQueueStates(states *types.ProcessingQueueStates, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeProcessingQueueStates\", states, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeProcessingQueueStates indicates an expected call of SerializeProcessingQueueStates.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeProcessingQueueStates(states, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeProcessingQueueStates\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeProcessingQueueStates), states, encodingType)\n}\n\n// SerializeResetPoints mocks base method.\nfunc (m *MockPayloadSerializer) SerializeResetPoints(event *types.ResetPoints, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeResetPoints\", event, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeResetPoints indicates an expected call of SerializeResetPoints.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeResetPoints(event, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeResetPoints\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeResetPoints), event, encodingType)\n}\n\n// SerializeVersionHistories mocks base method.\nfunc (m *MockPayloadSerializer) SerializeVersionHistories(histories *types.VersionHistories, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeVersionHistories\", histories, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeVersionHistories indicates an expected call of SerializeVersionHistories.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeVersionHistories(histories, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeVersionHistories\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeVersionHistories), histories, encodingType)\n}\n\n// SerializeVisibilityMemo mocks base method.\nfunc (m *MockPayloadSerializer) SerializeVisibilityMemo(memo *types.Memo, encodingType constants.EncodingType) (*DataBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeVisibilityMemo\", memo, encodingType)\n\tret0, _ := ret[0].(*DataBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeVisibilityMemo indicates an expected call of SerializeVisibilityMemo.\nfunc (mr *MockPayloadSerializerMockRecorder) SerializeVisibilityMemo(memo, encodingType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeVisibilityMemo\", reflect.TypeOf((*MockPayloadSerializer)(nil).SerializeVisibilityMemo), memo, encodingType)\n}\n"
  },
  {
    "path": "common/persistence/serializer_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype testDef struct {\n\tname string\n\t// payloads is a map of payload name to payload. \"nil\" is a special name for nil payload which expects to return nil with some exceptions\n\tpayloads      map[string]any\n\tnilHandled    bool\n\tserializeFn   func(any, constants.EncodingType) (*DataBlob, error)\n\tdeserializeFn func(*DataBlob) (any, error)\n}\n\n// key is encoding type, value is whether the encoding type is supported\nvar encodingTypes = map[constants.EncodingType]bool{\n\tconstants.EncodingTypeEmpty:    true,\n\tconstants.EncodingTypeUnknown:  true,\n\tconstants.EncodingTypeJSON:     true,\n\tconstants.EncodingTypeThriftRW: true,\n\tconstants.EncodingTypeGob:      false,\n}\n\ntype runnableTest struct {\n\ttestDef\n\tencoding    constants.EncodingType\n\tsupported   bool\n\tpayloadName string\n\tpayload     any\n}\n\nfunc TestSerializers(t *testing.T) {\n\t// create serializer once and reuse it for all test cases in parallel to catch concurrency issues\n\tserializer := NewPayloadSerializer()\n\n\ttests := []testDef{\n\t\t{\n\t\t\tname: \"history event\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.HistoryEvent)(nil),\n\t\t\t\t\"normal\": generateTestHistoryEvent(1),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeEvent(payload.(*types.HistoryEvent), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeEvent(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"batch history events\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    ([]*types.HistoryEvent)(nil),\n\t\t\t\t\"normal\": generateTestHistoryEventBatch(),\n\t\t\t},\n\t\t\tnilHandled: true,\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeBatchEvents(payload.([]*types.HistoryEvent), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeBatchEvents(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"visibility memo\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.Memo)(nil),\n\t\t\t\t\"normal\": generateVisibilityMemo(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeVisibilityMemo(payload.(*types.Memo), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeVisibilityMemo(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"version histories\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.VersionHistories)(nil),\n\t\t\t\t\"normal\": generateVersionHistories(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeVersionHistories(payload.(*types.VersionHistories), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeVersionHistories(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"reset points\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.ResetPoints)(nil),\n\t\t\t\t\"normal\": generateResetPoints(),\n\t\t\t},\n\t\t\tnilHandled: true,\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeResetPoints(payload.(*types.ResetPoints), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeResetPoints(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"bad binaries\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.BadBinaries)(nil),\n\t\t\t\t\"normal\": generateBadBinaries(),\n\t\t\t},\n\t\t\tnilHandled: true,\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeBadBinaries(payload.(*types.BadBinaries), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeBadBinaries(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"processing queue states\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.ProcessingQueueStates)(nil),\n\t\t\t\t\"normal\": generateProcessingQueueStates(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeProcessingQueueStates(payload.(*types.ProcessingQueueStates), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeProcessingQueueStates(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"dynamic config blob\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.DynamicConfigBlob)(nil),\n\t\t\t\t\"normal\": generateDynamicConfigBlob(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeDynamicConfigBlob(payload.(*types.DynamicConfigBlob), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeDynamicConfigBlob(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"isolation group\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.IsolationGroupConfiguration)(nil),\n\t\t\t\t\"normal\": generateIsolationGroupConfiguration(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeIsolationGroups(payload.(*types.IsolationGroupConfiguration), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeIsolationGroups(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failover markers\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    ([]*types.FailoverMarkerAttributes)(nil),\n\t\t\t\t\"normal\": generateFailoverMarkerAttributes(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializePendingFailoverMarkers(payload.([]*types.FailoverMarkerAttributes), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializePendingFailoverMarkers(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"async workflow config\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.AsyncWorkflowConfiguration)(nil),\n\t\t\t\t\"normal\": generateAsyncWorkflowConfig(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeAsyncWorkflowsConfig(payload.(*types.AsyncWorkflowConfiguration), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeAsyncWorkflowsConfig(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"checksum\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"empty\":  checksum.Checksum{},\n\t\t\t\t\"normal\": generateChecksum(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeChecksum(payload.(checksum.Checksum), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeChecksum(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"active clusters\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.ActiveClusters)(nil),\n\t\t\t\t\"normal\": generateActiveClusters(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeActiveClusters(payload.(*types.ActiveClusters), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeActiveClusters(data)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"active cluster selection policy\",\n\t\t\tpayloads: map[string]any{\n\t\t\t\t\"nil\":    (*types.ActiveClusterSelectionPolicy)(nil),\n\t\t\t\t\"normal\": generateActiveClusterSelectionPolicy(),\n\t\t\t},\n\t\t\tserializeFn: func(payload any, encoding constants.EncodingType) (*DataBlob, error) {\n\t\t\t\treturn serializer.SerializeActiveClusterSelectionPolicy(payload.(*types.ActiveClusterSelectionPolicy), encoding)\n\t\t\t},\n\t\t\tdeserializeFn: func(data *DataBlob) (any, error) {\n\t\t\t\treturn serializer.DeserializeActiveClusterSelectionPolicy(data)\n\t\t\t},\n\t\t},\n\t}\n\n\t// generate runnable test cases here so actual test body is not 3 level nested\n\tvar runnableTests []runnableTest\n\tfor _, td := range tests {\n\t\tfor encoding, supported := range encodingTypes {\n\t\t\tfor payloadName, payload := range td.payloads {\n\t\t\t\tif _, ok := payload.(checksum.Checksum); ok {\n\t\t\t\t\tif encoding != constants.EncodingTypeJSON {\n\t\t\t\t\t\tcontinue\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\trunnableTests = append(runnableTests, runnableTest{\n\t\t\t\t\ttestDef:     td,\n\t\t\t\t\tencoding:    encoding,\n\t\t\t\t\tsupported:   supported,\n\t\t\t\t\tpayloadName: payloadName,\n\t\t\t\t\tpayload:     payload,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t}\n\n\tfor _, tc := range runnableTests {\n\t\ttc := tc\n\t\tt.Run(fmt.Sprintf(\"%s with encoding:%s,payload:%s\", tc.name, tc.encoding, tc.payloadName), func(t *testing.T) {\n\t\t\tt.Parallel()\n\n\t\t\tserialized, err := tc.serializeFn(tc.payload, tc.encoding)\n\t\t\t// expect error if the encoding type is not supported. special case is nil payloads.\n\t\t\t// most of the serialization functions return early for nils but\n\t\t\t// some of them call underlying helper function which checks encoding type and raises error.\n\t\t\twantErr := !tc.supported && !(tc.payloadName == \"nil\" && !tc.nilHandled)\n\t\t\tif wantErr != (err != nil) {\n\t\t\t\tt.Fatalf(\"Got serialization err: %v, want err?: %v\", err, wantErr)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tdeserialized, err := tc.deserializeFn(serialized)\n\t\t\tif err != nil {\n\t\t\t\tt.Fatalf(\"Got serialization err: %v\", err)\n\t\t\t}\n\n\t\t\tif deserialized == nil {\n\t\t\t\tt.Fatalf(\"Got nil deserialized payload\")\n\t\t\t}\n\n\t\t\tif tc.payloadName == \"nil\" {\n\t\t\t\treturn\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDataBlob_GetData(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin          *DataBlob\n\t\texpectedOut []byte\n\t}{\n\t\t\"valid data\": {\n\t\t\tin:          &DataBlob{Data: []byte(\"dat\")},\n\t\t\texpectedOut: []byte(\"dat\"),\n\t\t},\n\t\t\"empty data\": {\n\t\t\tin:          &DataBlob{Data: nil},\n\t\t\texpectedOut: []byte{},\n\t\t},\n\t\t\"empty data 2\": {\n\t\t\tin:          nil,\n\t\t\texpectedOut: []byte{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expectedOut, td.in.GetData())\n\t\t})\n\t}\n}\n\nfunc generateTestHistoryEvent(id int64) *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        id,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\tResult:           []byte(\"result-1-event-1\"),\n\t\t\tScheduledEventID: 4,\n\t\t\tStartedEventID:   5,\n\t\t\tIdentity:         \"event-1\",\n\t\t},\n\t}\n}\n\nfunc generateTestHistoryEventBatch() []*types.HistoryEvent {\n\treturn []*types.HistoryEvent{\n\t\tgenerateTestHistoryEvent(111),\n\t\tgenerateTestHistoryEvent(112),\n\t\tgenerateTestHistoryEvent(113),\n\t}\n}\n\nfunc generateVisibilityMemo() *types.Memo {\n\tmemoFields := map[string][]byte{\n\t\t\"TestField\": []byte(`Test binary`),\n\t}\n\treturn &types.Memo{Fields: memoFields}\n}\n\nfunc generateVersionHistories() *types.VersionHistories {\n\treturn &types.VersionHistories{\n\t\tHistories: []*types.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{1},\n\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchToken: []byte{2},\n\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 3,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateResetPoints() *types.ResetPoints {\n\treturn &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\t{\n\t\t\t\tBinaryChecksum:           \"bad-binary-cs\",\n\t\t\t\tRunID:                    \"test-run-id\",\n\t\t\t\tFirstDecisionCompletedID: 123,\n\t\t\t\tCreatedTimeNano:          common.Int64Ptr(456),\n\t\t\t\tExpiringTimeNano:         common.Int64Ptr(789),\n\t\t\t\tResettable:               true,\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateBadBinaries() *types.BadBinaries {\n\treturn &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"bad-binary-cs\": {\n\t\t\t\tCreatedTimeNano: common.Int64Ptr(456),\n\t\t\t\tOperator:        \"test-operattor\",\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateProcessingQueueStates() *types.ProcessingQueueStates {\n\treturn &types.ProcessingQueueStates{\n\t\tStatesByCluster: map[string][]*types.ProcessingQueueState{\n\t\t\t\"cluster1\": {\n\t\t\t\t&types.ProcessingQueueState{\n\t\t\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\t\t\tAckLevel: common.Int64Ptr(1),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(2),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tDomainIDs:    []string{\"domain1\", \"domain2\"},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"cluster2\": {\n\t\t\t\t&types.ProcessingQueueState{\n\t\t\t\t\tLevel:    common.Int32Ptr(3),\n\t\t\t\t\tAckLevel: common.Int64Ptr(4),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(5),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tDomainIDs:    []string{\"domain3\", \"domain4\"},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateDynamicConfigBlob() *types.DynamicConfigBlob {\n\treturn &types.DynamicConfigBlob{\n\t\tSchemaVersion: 1,\n\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t{\n\t\t\t\tName: dynamicproperties.TestGetBoolPropertyKey.String(),\n\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t{\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         []byte(\"false\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateIsolationGroupConfiguration() *types.IsolationGroupConfiguration {\n\treturn &types.IsolationGroupConfiguration{\n\t\t\"zone-1\": {\n\t\t\tName:  \"zone-1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t\t\"zone-2\": {\n\t\t\tName:  \"zone-2\",\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t}\n}\n\nfunc generateFailoverMarkerAttributes() []*types.FailoverMarkerAttributes {\n\treturn []*types.FailoverMarkerAttributes{\n\t\t{\n\t\t\tDomainID:        \"domain1\",\n\t\t\tFailoverVersion: 123,\n\t\t\tCreationTime:    common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\t{\n\t\t\tDomainID:        \"domain2\",\n\t\t\tFailoverVersion: 456,\n\t\t\tCreationTime:    common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t}\n}\n\nfunc generateAsyncWorkflowConfig() *types.AsyncWorkflowConfiguration {\n\treturn &types.AsyncWorkflowConfiguration{\n\t\tEnabled:             true,\n\t\tPredefinedQueueName: \"test-queue\",\n\t\tQueueType:           \"kafka\",\n\t\tQueueConfig: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         []byte(`{\"key\":\"value\"}`),\n\t\t},\n\t}\n}\n\nfunc generateChecksum() checksum.Checksum {\n\treturn checksum.Checksum{\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tVersion: 1,\n\t\tValue:   []byte(\"test-checksum\"),\n\t}\n}\n\nfunc generateActiveClusters() *types.ActiveClusters {\n\treturn &types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t},\n\t\t\t\t\t\"region2\": {\n\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\tFailoverVersion:   3,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateActiveClusterSelectionPolicy() *types.ActiveClusterSelectionPolicy {\n\treturn &types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\tStickyRegion:                   \"region1\",\n\t\tExternalEntityType:             \"externalEntityType1\",\n\t\tExternalEntityKey:              \"externalEntityKey1\",\n\t}\n}\n"
  },
  {
    "path": "common/persistence/shard_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n)\n\ntype (\n\tshardManager struct {\n\t\tpersistence ShardStore\n\t\tserializer  PayloadSerializer\n\t\ttimeSrc     clock.TimeSource\n\t\tdc          *DynamicConfiguration\n\t}\n)\n\nvar _ ShardManager = (*shardManager)(nil)\n\ntype ShardManagerOption func(manager *shardManager)\n\nfunc WithSerializer(serializer PayloadSerializer) ShardManagerOption {\n\treturn func(manager *shardManager) {\n\t\tif serializer != nil {\n\t\t\tmanager.serializer = serializer\n\t\t}\n\t}\n}\n\n// NewShardManager returns a new ShardManager\nfunc NewShardManager(\n\tpersistence ShardStore,\n\tdc *DynamicConfiguration,\n\toptions ...ShardManagerOption,\n) ShardManager {\n\tmanager := &shardManager{\n\t\tpersistence: persistence,\n\t\tserializer:  NewPayloadSerializer(),\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t\tdc:          dc,\n\t}\n\tfor _, option := range options {\n\t\toption(manager)\n\t}\n\treturn manager\n}\n\nfunc (m *shardManager) GetName() string {\n\treturn m.persistence.GetName()\n}\n\nfunc (m *shardManager) Close() {\n\tm.persistence.Close()\n}\n\nfunc (m *shardManager) CreateShard(ctx context.Context, request *CreateShardRequest) error {\n\tshardInfo, err := m.toInternalShardInfo(request.ShardInfo, constants.EncodingType(m.dc.SerializationEncoding()))\n\tif err != nil {\n\t\treturn err\n\t}\n\tinternalRequest := &InternalCreateShardRequest{\n\t\tShardInfo:        shardInfo,\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\treturn m.persistence.CreateShard(ctx, internalRequest)\n}\n\nfunc (m *shardManager) GetShard(ctx context.Context, request *GetShardRequest) (*GetShardResponse, error) {\n\tinternalRequest := &InternalGetShardRequest{\n\t\tShardID: request.ShardID,\n\t}\n\tinternalResult, err := m.persistence.GetShard(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tshardInfo, err := m.fromInternalShardInfo(internalResult.ShardInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult := &GetShardResponse{\n\t\tShardInfo: shardInfo,\n\t}\n\treturn result, nil\n}\n\nfunc (m *shardManager) UpdateShard(ctx context.Context, request *UpdateShardRequest) error {\n\tshardInfo, err := m.toInternalShardInfo(request.ShardInfo, constants.EncodingType(m.dc.SerializationEncoding()))\n\tif err != nil {\n\t\treturn err\n\t}\n\tinternalRequest := &InternalUpdateShardRequest{\n\t\tShardInfo:        shardInfo,\n\t\tPreviousRangeID:  request.PreviousRangeID,\n\t\tCurrentTimeStamp: m.timeSrc.Now(),\n\t}\n\treturn m.persistence.UpdateShard(ctx, internalRequest)\n}\n\nfunc (m *shardManager) toInternalShardInfo(shardInfo *ShardInfo, encodingType constants.EncodingType) (*InternalShardInfo, error) {\n\tif shardInfo == nil {\n\t\treturn nil, nil\n\t}\n\tserializedTransferProcessingQueueStates, err := m.serializer.SerializeProcessingQueueStates(shardInfo.TransferProcessingQueueStates, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tserializedTimerProcessingQueueStates, err := m.serializer.SerializeProcessingQueueStates(shardInfo.TimerProcessingQueueStates, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpendingFailoverMarker, err := m.serializer.SerializePendingFailoverMarkers(shardInfo.PendingFailoverMarkers, encodingType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &InternalShardInfo{\n\t\tShardID:                       shardInfo.ShardID,\n\t\tOwner:                         shardInfo.Owner,\n\t\tRangeID:                       shardInfo.RangeID,\n\t\tStolenSinceRenew:              shardInfo.StolenSinceRenew,\n\t\tUpdatedAt:                     shardInfo.UpdatedAt,\n\t\tReplicationAckLevel:           shardInfo.ReplicationAckLevel,\n\t\tReplicationDLQAckLevel:        shardInfo.ReplicationDLQAckLevel,\n\t\tTransferAckLevel:              shardInfo.TransferAckLevel,\n\t\tTimerAckLevel:                 shardInfo.TimerAckLevel,\n\t\tClusterTransferAckLevel:       shardInfo.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          shardInfo.ClusterTimerAckLevel,\n\t\tTransferProcessingQueueStates: serializedTransferProcessingQueueStates,\n\t\tTimerProcessingQueueStates:    serializedTimerProcessingQueueStates,\n\t\tClusterReplicationLevel:       shardInfo.ClusterReplicationLevel,\n\t\tDomainNotificationVersion:     shardInfo.DomainNotificationVersion,\n\t\tPendingFailoverMarkers:        pendingFailoverMarker,\n\t\tQueueStates:                   shardInfo.QueueStates,\n\t}, nil\n}\n\nfunc (m *shardManager) fromInternalShardInfo(internalShardInfo *InternalShardInfo) (*ShardInfo, error) {\n\tif internalShardInfo == nil {\n\t\treturn nil, nil\n\t}\n\ttransferProcessingQueueStates, err := m.serializer.DeserializeProcessingQueueStates(internalShardInfo.TransferProcessingQueueStates)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttimerProcessingQueueStates, err := m.serializer.DeserializeProcessingQueueStates(internalShardInfo.TimerProcessingQueueStates)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpendingFailoverMarker, err := m.serializer.DeserializePendingFailoverMarkers(internalShardInfo.PendingFailoverMarkers)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &ShardInfo{\n\t\tShardID:                       internalShardInfo.ShardID,\n\t\tOwner:                         internalShardInfo.Owner,\n\t\tRangeID:                       internalShardInfo.RangeID,\n\t\tStolenSinceRenew:              internalShardInfo.StolenSinceRenew,\n\t\tUpdatedAt:                     internalShardInfo.UpdatedAt,\n\t\tReplicationAckLevel:           internalShardInfo.ReplicationAckLevel,\n\t\tReplicationDLQAckLevel:        internalShardInfo.ReplicationDLQAckLevel,\n\t\tTransferAckLevel:              internalShardInfo.TransferAckLevel,\n\t\tTimerAckLevel:                 internalShardInfo.TimerAckLevel,\n\t\tClusterTransferAckLevel:       internalShardInfo.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          internalShardInfo.ClusterTimerAckLevel,\n\t\tTransferProcessingQueueStates: transferProcessingQueueStates,\n\t\tTimerProcessingQueueStates:    timerProcessingQueueStates,\n\t\tClusterReplicationLevel:       internalShardInfo.ClusterReplicationLevel,\n\t\tDomainNotificationVersion:     internalShardInfo.DomainNotificationVersion,\n\t\tPendingFailoverMarkers:        pendingFailoverMarker,\n\t\tQueueStates:                   internalShardInfo.QueueStates,\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/shard_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestShardManagerGetName(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tstore := NewMockShardStore(ctrl)\n\tstore.EXPECT().GetName().Return(\"name\")\n\n\tmanager := NewShardManager(store, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\n\tassert.Equal(t, \"name\", manager.GetName())\n}\n\nfunc TestShardManagerClose(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tstore := NewMockShardStore(ctrl)\n\tstore.EXPECT().Close().Times(1)\n\n\tmanager := NewShardManager(store, &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\tmanager.Close()\n}\n\nfunc TestShardManagerCreateShard(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest          *CreateShardRequest\n\t\tinternalRequest  *InternalCreateShardRequest\n\t\tserializer       PayloadSerializer\n\t\tinternalResponse error\n\t\texpected         error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest: &CreateShardRequest{\n\t\t\t\tShardInfo: sampleShardInfo(),\n\t\t\t},\n\t\t\tinternalRequest: &InternalCreateShardRequest{\n\t\t\t\tShardInfo: sampleInternalShardInfo(t),\n\t\t\t},\n\t\t\tinternalResponse: nil,\n\t\t\texpected:         nil,\n\t\t},\n\t\t\"Serialization Failure\": {\n\t\t\trequest: &CreateShardRequest{\n\t\t\t\tShardInfo: sampleShardInfo(),\n\t\t\t},\n\t\t\tserializer: failingSerializer(ctrl),\n\t\t\texpected:   fmt.Errorf(\"serialization\"),\n\t\t},\n\t\t\"Error Response\": {\n\t\t\trequest: &CreateShardRequest{\n\t\t\t\tShardInfo: sampleShardInfo(),\n\t\t\t},\n\t\t\tinternalRequest: &InternalCreateShardRequest{\n\t\t\t\tShardInfo: sampleInternalShardInfo(t),\n\t\t\t},\n\t\t\tinternalResponse: fmt.Errorf(\"error\"),\n\t\t\texpected:         fmt.Errorf(\"error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore := NewMockShardStore(ctrl)\n\t\t\tif test.internalRequest != nil {\n\t\t\t\tstore.EXPECT().CreateShard(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDo(func(ctx context.Context, req *InternalCreateShardRequest) {\n\t\t\t\t\t\tassert.Equal(t, test.internalRequest.ShardInfo, req.ShardInfo)\n\n\t\t\t\t\t\tassert.WithinDuration(t, time.Now(), req.CurrentTimeStamp, time.Second)\n\t\t\t\t\t}).Return(test.internalResponse)\n\t\t\t}\n\n\t\t\tmanager := NewShardManager(store, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t}, WithSerializer(test.serializer))\n\n\t\t\tresult := manager.CreateShard(context.Background(), test.request)\n\n\t\t\tassert.Equal(t, test.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestShardManagerGetShard(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := map[string]struct {\n\t\trequest          *GetShardRequest\n\t\tinternalRequest  *InternalGetShardRequest\n\t\tserializer       PayloadSerializer\n\t\tinternalResponse *InternalGetShardResponse\n\t\tinternalErr      error\n\t\texpected         *GetShardResponse\n\t\texpectedErr      error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest: &GetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalRequest: &InternalGetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalResponse: &InternalGetShardResponse{\n\t\t\t\tShardInfo: sampleInternalShardInfo(t),\n\t\t\t},\n\t\t\texpected: &GetShardResponse{\n\t\t\t\tShardInfo: sampleShardInfo(),\n\t\t\t},\n\t\t},\n\t\t\"Nil response\": {\n\t\t\trequest: &GetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalRequest: &InternalGetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalResponse: &InternalGetShardResponse{\n\t\t\t\tShardInfo: nil,\n\t\t\t},\n\t\t\texpected: &GetShardResponse{\n\t\t\t\tShardInfo: nil,\n\t\t\t},\n\t\t},\n\t\t\"Serialization Failure\": {\n\t\t\trequest: &GetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalRequest: &InternalGetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalResponse: &InternalGetShardResponse{\n\t\t\t\tShardInfo: sampleInternalShardInfo(t),\n\t\t\t},\n\t\t\tserializer:  failingSerializer(ctrl),\n\t\t\texpectedErr: fmt.Errorf(\"serialization\"),\n\t\t},\n\t\t\"Error Response\": {\n\t\t\trequest: &GetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalRequest: &InternalGetShardRequest{\n\t\t\t\tShardID: shardID,\n\t\t\t},\n\t\t\tinternalErr: fmt.Errorf(\"error\"),\n\t\t\texpectedErr: fmt.Errorf(\"error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore := NewMockShardStore(ctrl)\n\t\t\tif test.internalRequest != nil {\n\t\t\t\tstore.EXPECT().GetShard(gomock.Any(), gomock.Eq(test.internalRequest)).Return(test.internalResponse, test.internalErr)\n\t\t\t}\n\n\t\t\tmanager := NewShardManager(store, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t}, WithSerializer(test.serializer))\n\n\t\t\tresult, err := manager.GetShard(context.Background(), test.request)\n\t\t\tassert.Equal(t, test.expected, result)\n\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestShardManagerUpdateShard(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest          *UpdateShardRequest\n\t\tinternalRequest  *InternalUpdateShardRequest\n\t\tserializer       PayloadSerializer\n\t\tinternalResponse error\n\t\texpected         error\n\t}{\n\t\t\"Success\": {\n\t\t\trequest: &UpdateShardRequest{\n\t\t\t\tShardInfo:       sampleShardInfo(),\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t},\n\t\t\tinternalRequest: &InternalUpdateShardRequest{\n\t\t\t\tShardInfo:       sampleInternalShardInfo(t),\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t},\n\t\t\tinternalResponse: nil,\n\t\t\texpected:         nil,\n\t\t},\n\t\t\"Serialization Failure\": {\n\t\t\trequest: &UpdateShardRequest{\n\t\t\t\tShardInfo:       sampleShardInfo(),\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t},\n\t\t\tserializer: failingSerializer(ctrl),\n\t\t\texpected:   fmt.Errorf(\"serialization\"),\n\t\t},\n\t\t\"Error Response\": {\n\t\t\trequest: &UpdateShardRequest{\n\t\t\t\tShardInfo:       sampleShardInfo(),\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t},\n\t\t\tinternalRequest: &InternalUpdateShardRequest{\n\t\t\t\tShardInfo:       sampleInternalShardInfo(t),\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t},\n\t\t\tinternalResponse: fmt.Errorf(\"error\"),\n\t\t\texpected:         fmt.Errorf(\"error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore := NewMockShardStore(ctrl)\n\t\t\tif test.internalRequest != nil {\n\t\t\t\tstore.EXPECT().UpdateShard(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDo(func(ctx context.Context, req *InternalUpdateShardRequest) {\n\t\t\t\t\t\tassert.Equal(t, test.internalRequest.PreviousRangeID, req.PreviousRangeID)\n\t\t\t\t\t\tassert.Equal(t, test.internalRequest.ShardInfo, req.ShardInfo)\n\n\t\t\t\t\t\tassert.WithinDuration(t, time.Now(), req.CurrentTimeStamp, time.Second)\n\t\t\t\t\t}).Return(test.internalResponse)\n\t\t\t}\n\n\t\t\tmanager := NewShardManager(store, &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t}, WithSerializer(test.serializer))\n\n\t\t\tresult := manager.UpdateShard(context.Background(), test.request)\n\n\t\t\tassert.Equal(t, test.expected, result)\n\n\t\t})\n\t}\n}\n\nvar (\n\tshardID                      = 1\n\towner                        = \"TestOwner\"\n\trangeID                int64 = 2\n\tstolenSinceRenew             = 3\n\tupdatedAt                    = time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC)\n\treplicationAckLevel    int64 = 4\n\ttimerAckLevel                = time.Date(2020, 2, 1, 0, 0, 0, 0, time.UTC)\n\ttransferAckLevel       int64 = 5\n\treplicationDlqAckLevel       = map[string]int64{\n\t\t\"key\": 123,\n\t}\n\tclusterTransferAckLevel = map[string]int64{\n\t\t\"otherKey\": 124,\n\t}\n\tclusterTimerAckLevel = map[string]time.Time{\n\t\t\"foo\": time.Date(2020, 3, 1, 0, 0, 0, 0, time.UTC),\n\t}\n\ttransferProcessingQueueStates = &types.ProcessingQueueStates{\n\t\tStatesByCluster: map[string][]*types.ProcessingQueueState{\n\t\t\t\"transfer\": {\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(1),\n\t\t\t\t\tAckLevel: common.Int64Ptr(2),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(3),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tDomainIDs:    []string{\"foo\", \"bar\"},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ttimerProcesssingQueueStates = &types.ProcessingQueueStates{\n\t\tStatesByCluster: map[string][]*types.ProcessingQueueState{\n\t\t\t\"timer\": {\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(7),\n\t\t\t\t\tAckLevel: common.Int64Ptr(8),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(9),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tDomainIDs:    []string{\"bar\"},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tclusterReplicationLevel = map[string]int64{\n\t\t\"bar\": 100,\n\t}\n\tdomainNotificationVersion int64 = 6\n\tpendingFailoverMarkers          = []*types.FailoverMarkerAttributes{\n\t\t{\n\t\t\tDomainID:        \"TestDomain\",\n\t\t\tFailoverVersion: 11,\n\t\t\tCreationTime:    common.Int64Ptr(12),\n\t\t},\n\t}\n)\n\nfunc failingSerializer(ctrl *gomock.Controller) PayloadSerializer {\n\tserializer := NewMockPayloadSerializer(ctrl)\n\tserializer.EXPECT().SerializePendingFailoverMarkers(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"serialization\")).AnyTimes()\n\tserializer.EXPECT().DeserializePendingFailoverMarkers(gomock.Any()).Return(nil, fmt.Errorf(\"serialization\")).AnyTimes()\n\tserializer.EXPECT().SerializeProcessingQueueStates(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"serialization\")).AnyTimes()\n\tserializer.EXPECT().DeserializeProcessingQueueStates(gomock.Any()).Return(nil, fmt.Errorf(\"serialization\")).AnyTimes()\n\n\treturn serializer\n}\n\nfunc sampleInternalShardInfo(t *testing.T) *InternalShardInfo {\n\tserializer := NewPayloadSerializer()\n\ttransferProcessingQueueStatesBlob, err := serializer.SerializeProcessingQueueStates(transferProcessingQueueStates, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\ttimerProcessingQueueStatesBlob, err := serializer.SerializeProcessingQueueStates(timerProcesssingQueueStates, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\tpendingFailoverMarkerBlob, err := serializer.SerializePendingFailoverMarkers(pendingFailoverMarkers, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\treturn &InternalShardInfo{\n\t\tShardID:                       shardID,\n\t\tOwner:                         owner,\n\t\tRangeID:                       rangeID,\n\t\tStolenSinceRenew:              3,\n\t\tUpdatedAt:                     updatedAt,\n\t\tReplicationAckLevel:           4,\n\t\tReplicationDLQAckLevel:        replicationDlqAckLevel,\n\t\tTransferAckLevel:              5,\n\t\tTimerAckLevel:                 timerAckLevel,\n\t\tClusterTransferAckLevel:       clusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          clusterTimerAckLevel,\n\t\tTransferProcessingQueueStates: transferProcessingQueueStatesBlob,\n\t\tTimerProcessingQueueStates:    timerProcessingQueueStatesBlob,\n\t\tClusterReplicationLevel:       clusterReplicationLevel,\n\t\tDomainNotificationVersion:     domainNotificationVersion,\n\t\tPendingFailoverMarkers:        pendingFailoverMarkerBlob,\n\t}\n}\n\nfunc sampleShardInfo() *ShardInfo {\n\treturn &ShardInfo{\n\t\tShardID:                       shardID,\n\t\tOwner:                         owner,\n\t\tRangeID:                       rangeID,\n\t\tStolenSinceRenew:              stolenSinceRenew,\n\t\tUpdatedAt:                     updatedAt,\n\t\tReplicationAckLevel:           replicationAckLevel,\n\t\tReplicationDLQAckLevel:        replicationDlqAckLevel,\n\t\tTransferAckLevel:              transferAckLevel,\n\t\tTimerAckLevel:                 timerAckLevel,\n\t\tClusterTransferAckLevel:       clusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          clusterTimerAckLevel,\n\t\tTransferProcessingQueueStates: transferProcessingQueueStates,\n\t\tTimerProcessingQueueStates:    timerProcesssingQueueStates,\n\t\tClusterReplicationLevel:       clusterReplicationLevel,\n\t\tDomainNotificationVersion:     domainNotificationVersion,\n\t\tPendingFailoverMarkers:        pendingFailoverMarkers,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/common.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/binary\"\n\t\"encoding/gob\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlStore struct {\n\tdb     sqlplugin.DB\n\tlogger log.Logger\n\tparser serialization.Parser\n\tdc     *p.DynamicConfiguration\n}\n\nfunc (m *sqlStore) GetName() string {\n\treturn m.db.PluginName()\n}\n\nfunc (m *sqlStore) Close() {\n\tif m.db != nil {\n\t\tm.db.Close()\n\t}\n}\n\nfunc (m *sqlStore) useAsyncTransaction() bool {\n\treturn m.db.SupportsAsyncTransaction() && m.dc != nil && m.dc.EnableSQLAsyncTransaction()\n}\n\nfunc (m *sqlStore) txExecute(ctx context.Context, dbShardID int, operation string, f func(tx sqlplugin.Tx) error) error {\n\ttx, err := m.db.BeginTx(ctx, dbShardID)\n\tif err != nil {\n\t\treturn convertCommonErrors(m.db, operation, \"Failed to start transaction.\", err)\n\t}\n\terr = f(tx)\n\tif err != nil {\n\t\trollBackErr := tx.Rollback()\n\t\tif rollBackErr != nil {\n\t\t\tm.logger.Error(\"transaction rollback error\", tag.Error(rollBackErr))\n\t\t}\n\t\treturn convertCommonErrors(m.db, operation, \"\", err)\n\t}\n\tif err := tx.Commit(); err != nil {\n\t\treturn convertCommonErrors(m.db, operation, \"Failed to commit transaction.\", err)\n\t}\n\treturn nil\n}\n\nfunc gobSerialize(x interface{}) ([]byte, error) {\n\tb := bytes.Buffer{}\n\te := gob.NewEncoder(&b)\n\terr := e.Encode(x)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Error in serialization: %v\", err),\n\t\t}\n\t}\n\treturn b.Bytes(), nil\n}\n\nfunc gobDeserialize(a []byte, x interface{}) error {\n\tb := bytes.NewBuffer(a)\n\td := gob.NewDecoder(b)\n\terr := d.Decode(x)\n\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Error in deserialization: %v\", err),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc serializePageToken(offset int64) []byte {\n\tb := make([]byte, 8)\n\tbinary.LittleEndian.PutUint64(b, uint64(offset))\n\treturn b\n}\n\nfunc deserializePageToken(payload []byte) (int64, error) {\n\tif len(payload) != 8 {\n\t\treturn 0, fmt.Errorf(\"invalid token of %v length\", len(payload))\n\t}\n\treturn int64(binary.LittleEndian.Uint64(payload)), nil\n}\n\nfunc convertCommonErrors(\n\terrChecker sqlplugin.ErrorChecker,\n\toperation, message string,\n\terr error,\n) error {\n\tswitch err.(type) {\n\tcase *persistence.ConditionFailedError,\n\t\t*persistence.CurrentWorkflowConditionFailedError,\n\t\t*persistence.WorkflowExecutionAlreadyStartedError,\n\t\t*persistence.ShardOwnershipLostError,\n\t\t*persistence.TimeoutError,\n\t\t*types.DomainAlreadyExistsError,\n\t\t*types.EntityNotExistsError,\n\t\t*types.ServiceBusyError,\n\t\t*types.InternalServiceError:\n\t\treturn err\n\t}\n\tif errChecker.IsNotFoundError(err) {\n\t\treturn &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\"%v failed. %s Error: %v\", operation, message, err),\n\t\t}\n\t}\n\n\tif errChecker.IsTimeoutError(err) {\n\t\treturn &persistence.TimeoutError{Msg: fmt.Sprintf(\"%v timed out. %s Error: %v\", operation, message, err)}\n\t}\n\n\tif errChecker.IsThrottlingError(err) {\n\t\treturn &types.ServiceBusyError{\n\t\t\tMessage: fmt.Sprintf(\"%v operation failed. %s Error: %v\", operation, message, err),\n\t\t}\n\t}\n\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\"%v operation failed. %s Error: %v\", operation, message, err),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/common_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// MockErrorChecker is a mock implementation of the sqlplugin.ErrorChecker interface\ntype MockErrorChecker struct{}\n\nvar _ sqlplugin.ErrorChecker = (*MockErrorChecker)(nil)\n\nfunc (m *MockErrorChecker) IsNotFoundError(err error) bool {\n\tif strings.Contains(err.Error(), \"not found\") {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc (m *MockErrorChecker) IsTimeoutError(err error) bool {\n\tif strings.Contains(err.Error(), \"timeout\") {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc (m *MockErrorChecker) IsThrottlingError(err error) bool {\n\tif strings.Contains(err.Error(), \"throttle\") {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc (m *MockErrorChecker) IsDupEntryError(err error) bool {\n\treturn false\n}\n\nfunc TestConvertCommonErrors(t *testing.T) {\n\terrChecker := &MockErrorChecker{}\n\ttests := []struct {\n\t\tname      string\n\t\toperation string\n\t\tmessage   string\n\t\terr       error\n\t\twantError error\n\t}{\n\t\t{\n\t\t\tname:      \"ConditionFailedError\",\n\t\t\toperation: \"Create\",\n\t\t\tmessage:   \"creation\",\n\t\t\terr:       &persistence.ConditionFailedError{},\n\t\t\twantError: &persistence.ConditionFailedError{},\n\t\t},\n\t\t{\n\t\t\tname:      \"NotFoundError\",\n\t\t\toperation: \"Get\",\n\t\t\tmessage:   \"retrieval\",\n\t\t\terr:       errors.New(\"not found\"),\n\t\t\twantError: &types.EntityNotExistsError{Message: \"Get failed. retrieval Error: not found\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"TimeoutError\",\n\t\t\toperation: \"Update\",\n\t\t\tmessage:   \"update\",\n\t\t\terr:       errors.New(\"timeout\"),\n\t\t\twantError: &persistence.TimeoutError{Msg: \"Update timed out. update Error: timeout\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"ThrottlingError\",\n\t\t\toperation: \"Delete\",\n\t\t\tmessage:   \"deletion\",\n\t\t\terr:       errors.New(\"throttle\"),\n\t\t\twantError: &types.ServiceBusyError{Message: \"Delete operation failed. deletion Error: throttle\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"InternalServiceError\",\n\t\t\toperation: \"List\",\n\t\t\tmessage:   \"listing\",\n\t\t\terr:       errors.New(\"generic error\"),\n\t\t\twantError: &types.InternalServiceError{Message: \"List operation failed. listing Error: generic error\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgotError := convertCommonErrors(errChecker, tt.operation, tt.message, tt.err)\n\t\t\tif gotError.Error() != tt.wantError.Error() {\n\t\t\t\tt.Errorf(\"convertCommonErrors() error = %v, wantErr %v\", gotError, tt.wantError)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTxExecute(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\toperation string\n\t\tfn        func(sqlplugin.Tx) error\n\t\twantError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\tfn:        func(sqlplugin.Tx) error { return nil },\n\t\t\twantError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\tfn:        func(sqlplugin.Tx) error { return errors.New(\"error\") },\n\t\t\twantError: &types.InternalServiceError{Message: \"Insert operation failed.  Error: error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"BeginTxError\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\tfn:        func(sqlplugin.Tx) error { return nil },\n\t\t\twantError: &types.InternalServiceError{Message: \"Insert operation failed. Failed to start transaction. Error: error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"CommitError\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(errors.New(\"error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\tfn:        func(sqlplugin.Tx) error { return nil },\n\t\t\twantError: &types.InternalServiceError{Message: \"Insert operation failed. Failed to commit transaction. Error: error\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\ttt.mockSetup(mockDB, mockTx)\n\n\t\t\ts := &sqlStore{db: mockDB, logger: testlogger.New(t)}\n\n\t\t\tgotError := s.txExecute(context.Background(), 0, tt.operation, tt.fn)\n\t\t\tassert.Equal(t, tt.wantError, gotError)\n\t\t})\n\t}\n}\n\nfunc TestGobSerialize(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\tinput   interface{}\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname:    \"Serialize string\",\n\t\t\tinput:   \"test string\",\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Serialize struct\",\n\t\t\tinput:   struct{ A int }{1},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Serialize unsupported type\",\n\t\t\tinput:   make(chan int),\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, err := gobSerialize(tt.input)\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Errorf(\"gobSerialize() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif !tt.wantErr && len(got) == 0 {\n\t\t\t\tt.Errorf(\"gobSerialize() returned empty byte slice\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGobDeserialize(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\tinput   []byte\n\t\ttarget  interface{}\n\t\twant    interface{}\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname:    \"Deserialize to string\",\n\t\t\tinput:   mustGobSerialize(\"test string\"),\n\t\t\ttarget:  new(string),\n\t\t\twant:    \"test string\",\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Deserialize to struct\",\n\t\t\tinput:   mustGobSerialize(struct{ A int }{1}),\n\t\t\ttarget:  new(struct{ A int }),\n\t\t\twant:    struct{ A int }{1},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:    \"Deserialize with invalid data\",\n\t\t\tinput:   []byte(\"invalid\"),\n\t\t\ttarget:  new(string),\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\terr := gobDeserialize(tt.input, tt.target)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, reflect.ValueOf(tt.target).Elem().Interface())\n\t\t\t}\n\t\t})\n\t}\n}\n\n// mustGobSerialize is a helper function that panics if serialization fails. Used for setting up tests.\nfunc mustGobSerialize(v interface{}) []byte {\n\tb, err := gobSerialize(v)\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"mustGobSerialize: %v\", err))\n\t}\n\treturn b\n}\n"
  },
  {
    "path": "common/persistence/sql/factory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype (\n\t// Factory vends store objects backed by MySQL\n\tFactory struct {\n\t\tcfg            config.SQL\n\t\tdbConn         dbConn\n\t\tclusterName    string\n\t\tlogger         log.Logger\n\t\tparser         serialization.Parser\n\t\ttaskSerializer serialization.TaskSerializer\n\t\tdc             *p.DynamicConfiguration\n\t}\n\n\t// dbConn represents a logical mysql connection - it's a\n\t// wrapper around the standard sql connection pool with\n\t// additional reference counting\n\tdbConn struct {\n\t\tsync.Mutex\n\t\tsqlplugin.DB\n\t\trefCnt int\n\t\tcfg    *config.SQL\n\t}\n)\n\n// NewFactory returns an instance of a factory object which can be used to create\n// datastores backed by any kind of SQL store\nfunc NewFactory(\n\tcfg config.SQL,\n\tclusterName string,\n\tlogger log.Logger,\n\tparser serialization.Parser,\n\tdc *p.DynamicConfiguration,\n) *Factory {\n\treturn &Factory{\n\t\tcfg:            cfg,\n\t\tclusterName:    clusterName,\n\t\tlogger:         logger,\n\t\tdbConn:         newRefCountedDBConn(&cfg),\n\t\tparser:         parser,\n\t\ttaskSerializer: serialization.NewTaskSerializer(parser),\n\t\tdc:             dc,\n\t}\n}\n\n// NewTaskStore returns a new task store\nfunc (f *Factory) NewTaskStore() (p.TaskStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newTaskPersistence(conn, f.cfg.NumShards, f.logger, f.parser)\n}\n\n// NewShardStore returns a new shard store\nfunc (f *Factory) NewShardStore() (p.ShardStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewShardPersistence(conn, f.clusterName, f.logger, f.parser)\n}\n\n// NewHistoryStore returns a new history store\nfunc (f *Factory) NewHistoryStore() (p.HistoryStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewHistoryV2Persistence(conn, f.logger, f.parser, f.dc)\n}\n\n// NewDomainStore returns a new metadata store\nfunc (f *Factory) NewDomainStore() (p.DomainStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newMetadataPersistenceV2(conn, f.clusterName, f.logger, f.parser)\n}\n\n// NewDomainAuditStore returns a domain audit store\nfunc (f *Factory) NewDomainAuditStore() (p.DomainAuditStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newSQLDomainAuditStore(conn, f.logger, f.parser)\n}\n\n// NewExecutionStore returns an ExecutionStore for a given shardID\nfunc (f *Factory) NewExecutionStore(shardID int) (p.ExecutionStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewSQLExecutionStore(conn, f.logger, shardID, f.parser, f.taskSerializer, f.dc)\n}\n\n// NewVisibilityStore returns a visibility store\n// TODO sortByCloseTime will be removed and implemented for https://github.com/uber/cadence/issues/3621\nfunc (f *Factory) NewVisibilityStore(sortByCloseTime bool) (p.VisibilityStore, error) {\n\treturn NewSQLVisibilityStore(f.cfg, f.logger)\n}\n\n// NewQueue returns a new queue backed by sql\nfunc (f *Factory) NewQueue(queueType p.QueueType) (p.QueueStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn newQueueStore(conn, f.logger, queueType)\n}\n\n// NewConfigStore returns a new config store backed by sql. Not Yet Implemented.\nfunc (f *Factory) NewConfigStore() (p.ConfigStore, error) {\n\tconn, err := f.dbConn.get()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewSQLConfigStore(conn, f.logger, f.parser)\n}\n\n// Close closes the factory\nfunc (f *Factory) Close() {\n\tf.dbConn.forceClose()\n}\n\n// newRefCountedDBConn returns a  logical mysql connection that\n// uses reference counting to decide when to close the\n// underlying connection object. The reference count gets incremented\n// everytime get() is called and decremented everytime Close() is called\nfunc newRefCountedDBConn(cfg *config.SQL) dbConn {\n\treturn dbConn{cfg: cfg}\n}\n\n// get returns a mysql db connection and increments a reference count\n// this method will create a new connection, if an existing connection\n// does not exist\nfunc (c *dbConn) get() (sqlplugin.DB, error) {\n\tc.Lock()\n\tdefer c.Unlock()\n\tif c.refCnt == 0 {\n\t\tconn, err := NewSQLDB(c.cfg)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tc.DB = conn\n\t}\n\tc.refCnt++\n\treturn c, nil\n}\n\n// forceClose ignores reference counts and shutsdown the underlying connection pool\nfunc (c *dbConn) forceClose() {\n\tc.Lock()\n\tdefer c.Unlock()\n\tif c.DB != nil {\n\t\terr := c.DB.Close()\n\t\tif err != nil {\n\t\t\tfmt.Println(\"failed to close database connection, may leak some connection\", err)\n\t\t}\n\n\t}\n\tc.refCnt = 0\n}\n\n// Close closes the underlying connection if the reference count becomes zero\nfunc (c *dbConn) Close() error {\n\tc.Lock()\n\tdefer c.Unlock()\n\tc.refCnt--\n\tif c.refCnt == 0 {\n\t\terr := c.DB.Close()\n\t\tc.DB = nil\n\t\treturn err\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/factory_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n)\n\nfunc TestNewFactory(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tassert.NotNil(t, factory)\n}\n\nfunc TestFactoryNewTaskStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\ttaskStore, err := factory.NewTaskStore()\n\tassert.Nil(t, taskStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\ttaskStore, err = factory.NewTaskStore()\n\tassert.NotNil(t, taskStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewShardStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tshardStore, err := factory.NewShardStore()\n\tassert.Nil(t, shardStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tshardStore, err = factory.NewShardStore()\n\tassert.NotNil(t, shardStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewHistoryV2Store(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\thistoryV2Store, err := factory.NewHistoryStore()\n\tassert.Nil(t, historyV2Store)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\thistoryV2Store, err = factory.NewHistoryStore()\n\tassert.NotNil(t, historyV2Store)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewMetadataV2Store(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tmetadataV2Store, err := factory.NewDomainStore()\n\tassert.Nil(t, metadataV2Store)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tmetadataV2Store, err = factory.NewDomainStore()\n\tassert.NotNil(t, metadataV2Store)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewExecutionStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\texecutionStore, err := factory.NewExecutionStore(0)\n\tassert.Nil(t, executionStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\texecutionStore, err = factory.NewExecutionStore(0)\n\tassert.NotNil(t, executionStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewVisibilityStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tvisibilityStore, err := factory.NewVisibilityStore(true)\n\tassert.Nil(t, visibilityStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tvisibilityStore, err = factory.NewVisibilityStore(true)\n\tassert.NotNil(t, visibilityStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewQueue(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tqueueStore, err := factory.NewQueue(persistence.DomainReplicationQueueType)\n\tassert.Nil(t, queueStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tqueueStore, err = factory.NewQueue(persistence.DomainReplicationQueueType)\n\tassert.NotNil(t, queueStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewConfigStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tconfigStore, err := factory.NewConfigStore()\n\tassert.Nil(t, configStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tconfigStore, err = factory.NewConfigStore()\n\tassert.NotNil(t, configStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n\nfunc TestFactoryNewDomainAuditStore(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\tcfg := config.SQL{}\n\tclusterName := \"test\"\n\tlogger := testlogger.New(t)\n\tmockParser := serialization.NewMockParser(ctrl)\n\tdc := &persistence.DynamicConfiguration{}\n\tfactory := NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tdomainAuditStore, err := factory.NewDomainAuditStore()\n\tassert.Nil(t, domainAuditStore)\n\tassert.Error(t, err)\n\tfactory.Close()\n\n\tcfg.PluginName = \"shared\"\n\tfactory = NewFactory(cfg, clusterName, logger, mockParser, dc)\n\tdomainAuditStore, err = factory.NewDomainAuditStore()\n\tassert.NotNil(t, domainAuditStore)\n\tassert.NoError(t, err)\n\tfactory.Close()\n}\n"
  },
  {
    "path": "common/persistence/sql/main_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"os\"\n\t\"testing\"\n)\n\nfunc TestMain(m *testing.M) {\n\tRegisterPlugin(\"shared\", &fakePlugin{})\n\tos.Exit(m.Run())\n}\n"
  },
  {
    "path": "common/persistence/sql/plugin.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nvar supportedPlugins = map[string]sqlplugin.Plugin{}\n\n// RegisterPlugin will register a SQL plugin\nfunc RegisterPlugin(pluginName string, plugin sqlplugin.Plugin) {\n\tif _, ok := supportedPlugins[pluginName]; ok {\n\t\tpanic(\"plugin \" + pluginName + \" already registered\")\n\t}\n\tsupportedPlugins[pluginName] = plugin\n}\n\n// RegisterPluginIfNotExists will register a SQL plugin only if a plugin with same name has not already been registered\nfunc RegisterPluginIfNotExists(pluginName string, plugin sqlplugin.Plugin) {\n\tif _, ok := supportedPlugins[pluginName]; !ok {\n\t\tsupportedPlugins[pluginName] = plugin\n\t}\n}\n\n// PluginRegistered returns true if plugin with given name has been registered, false otherwise\nfunc PluginRegistered(pluginName string) bool {\n\t_, ok := supportedPlugins[pluginName]\n\treturn ok\n}\n\n// GetRegisteredPluginNames returns the list of registered plugin names\nfunc GetRegisteredPluginNames() []string {\n\tvar plugins []string\n\tfor k := range supportedPlugins {\n\t\tplugins = append(plugins, k)\n\t}\n\tsort.Strings(plugins)\n\treturn plugins\n}\n\n// NewSQLDB creates a returns a reference to a logical connection to the\n// underlying SQL database. The returned object is to tied to a single\n// SQL database and the object can be used to perform CRUD operations on\n// the tables in the database\nfunc NewSQLDB(cfg *config.SQL) (sqlplugin.DB, error) {\n\tplugin, ok := supportedPlugins[cfg.PluginName]\n\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"not supported plugin %v, only supported: %v\", cfg.PluginName, supportedPlugins)\n\t}\n\n\treturn plugin.CreateDB(cfg)\n}\n\n// NewSQLAdminDB returns a AdminDB\nfunc NewSQLAdminDB(cfg *config.SQL) (sqlplugin.AdminDB, error) {\n\tplugin, ok := supportedPlugins[cfg.PluginName]\n\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"not supported plugin %v, only supported: %v\", cfg.PluginName, supportedPlugins)\n\t}\n\n\treturn plugin.CreateAdminDB(cfg)\n}\n"
  },
  {
    "path": "common/persistence/sql/plugin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype fakePlugin struct{}\n\nvar _ sqlplugin.Plugin = (*fakePlugin)(nil)\n\nfunc (f *fakePlugin) CreateDB(cfg *config.SQL) (sqlplugin.DB, error) {\n\treturn nil, nil\n}\n\nfunc (f *fakePlugin) CreateAdminDB(cfg *config.SQL) (sqlplugin.AdminDB, error) {\n\treturn nil, nil\n}\n\nfunc TestPluginRegistration(t *testing.T) {\n\tassert.NotPanics(t, func() {\n\t\tRegisterPlugin(\"fake\", &fakePlugin{})\n\t}, \"RegisterPlugin failed to register a plugin\")\n\n\tassert.Panics(t, func() {\n\t\tRegisterPlugin(\"fake\", &fakePlugin{})\n\t}, \"RegisterPlugin failed to panic when registering a plugin with the same name\")\n\n\tassert.NotPanics(t, func() {\n\t\tRegisterPluginIfNotExists(\"fake\", &fakePlugin{})\n\t}, \"RegisterPluginIfNotExists failed to register a plugin\")\n\n\tassert.NotPanics(t, func() {\n\t\tRegisterPluginIfNotExists(\"fake2\", &fakePlugin{})\n\t}, \"RegisterPluginIfNotExists failed to register a plugin\")\n\n\tassert.True(t, PluginRegistered(\"fake\"), \"PluginRegistered failed to return true for a registered plugin\")\n\tassert.True(t, PluginRegistered(\"fake2\"), \"PluginRegistered failed to return true for a registered plugin\")\n\tassert.False(t, PluginRegistered(\"fake3\"), \"PluginRegistered failed to return false for an unregistered plugin\")\n\n\tassert.Equal(t, []string{\"fake\", \"fake2\", \"shared\"}, GetRegisteredPluginNames(), \"GetRegisteredPluginNames failed to return the correct list of registered plugins\")\n\n\t_, err := NewSQLDB(&config.SQL{PluginName: \"fake\"})\n\tassert.NoError(t, err, \"NewSQLDB failed to create a DB with a registered plugin\")\n\t_, err = NewSQLDB(&config.SQL{PluginName: \"fake2\"})\n\tassert.NoError(t, err, \"NewSQLDB failed to create a DB with a registered plugin\")\n\n\t_, err = NewSQLDB(&config.SQL{PluginName: \"fake3\"})\n\tassert.Error(t, err, \"NewSQLDB failed to return an error with an unregistered plugin\")\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_config_store.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype (\n\tsqlConfigStore struct {\n\t\tsqlStore\n\t}\n)\n\n// NewSQLConfigStore creates a config store for SQL\nfunc NewSQLConfigStore(\n\tdb sqlplugin.DB,\n\tlogger log.Logger,\n\tparser serialization.Parser,\n) (persistence.ConfigStore, error) {\n\treturn &sqlConfigStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t\tparser: parser,\n\t\t},\n\t}, nil\n}\n\nfunc (m *sqlConfigStore) FetchConfig(ctx context.Context, configType persistence.ConfigType) (*persistence.InternalConfigStoreEntry, error) {\n\tentry, err := m.db.SelectLatestConfig(ctx, int(configType))\n\tif m.db.IsNotFoundError(err) {\n\t\treturn nil, nil\n\t}\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"FetchConfig\", \"\", err)\n\t}\n\treturn entry, nil\n}\n\nfunc (m *sqlConfigStore) UpdateConfig(ctx context.Context, value *persistence.InternalConfigStoreEntry) error {\n\terr := m.db.InsertConfig(ctx, value)\n\tif err != nil {\n\t\tif m.db.IsDupEntryError(err) {\n\t\t\treturn &persistence.ConditionFailedError{Msg: fmt.Sprintf(\"Version %v already exists. Condition Failed\", value.Version)}\n\t\t}\n\t\treturn convertCommonErrors(m.db, \"UpdateConfig\", \"\", err)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_config_store_test.go",
    "content": "// Modifications Copyright (c) 2020 Uber Technologies Inc.\n\n// Copyright (c) 2020 Temporal Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestFetchConfig(t *testing.T) {\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tconfigType persistence.ConfigType\n\t\tmockSetup  func(*sqlplugin.MockDB)\n\t\twant       *persistence.InternalConfigStoreEntry\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname:       \"Success case\",\n\t\t\tconfigType: persistence.DynamicConfig,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectLatestConfig(gomock.Any(), gomock.Any()).Return(&persistence.InternalConfigStoreEntry{}, nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(nil).Return(false)\n\t\t\t},\n\t\t\twant:    &persistence.InternalConfigStoreEntry{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:       \"Not found error\",\n\t\t\tconfigType: persistence.DynamicConfig,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tnotFoundErr := errors.New(\"not found\")\n\t\t\t\tmockDB.EXPECT().SelectLatestConfig(gomock.Any(), gomock.Any()).Return(nil, notFoundErr)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(notFoundErr).Return(true)\n\t\t\t},\n\t\t\twant:    nil,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:       \"Database error\",\n\t\t\tconfigType: persistence.DynamicConfig,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"db error\")\n\t\t\t\tmockDB.EXPECT().SelectLatestConfig(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(false)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(err).Return(true)\n\t\t\t},\n\t\t\twant:    nil,\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLConfigStore(mockDB, nil, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql config store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tgot, err := store.FetchConfig(context.Background(), tc.configType)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case: %s\", tc.name)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case: %s\", tc.name)\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case: %s\", tc.name)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateConfig(t *testing.T) {\n\ttestEntry := &persistence.InternalConfigStoreEntry{\n\t\tRowType: int(persistence.DynamicConfig),\n\t\tVersion: 0,\n\t\tValues: &persistence.DataBlob{\n\t\t\tData: []byte(`0x0000`),\n\t\t},\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tvalue     *persistence.InternalConfigStoreEntry\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:  \"Success case\",\n\t\t\tvalue: testEntry,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().InsertConfig(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"Duplicate Entry error\",\n\t\t\tvalue: testEntry,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tdupError := errors.New(\"duplicate entry\")\n\t\t\t\tmockDB.EXPECT().InsertConfig(gomock.Any(), gomock.Any()).Return(dupError)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(dupError).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ConditionFailedError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ConditionFailedError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"Database error\",\n\t\t\tvalue: testEntry,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"db error\")\n\t\t\t\tmockDB.EXPECT().InsertConfig(gomock.Any(), gomock.Any()).Return(err)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(err).Return(false)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLConfigStore(mockDB, nil, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql config store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\terr = store.UpdateConfig(context.Background(), tc.value)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case: %s\", tc.name)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case: %s\", tc.name)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_domain_audit_store.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlDomainAuditStore struct {\n\tsqlStore\n}\n\n// domainAuditLogPageToken is used for pagination\ntype domainAuditLogPageToken struct {\n\tCreatedTime time.Time `json:\"created_time\"`\n\tEventID     string    `json:\"event_id\"`\n}\n\n// newSQLDomainAuditStore creates an instance of sqlDomainAuditStore\nfunc newSQLDomainAuditStore(\n\tdb sqlplugin.DB,\n\tlogger log.Logger,\n\tparser serialization.Parser,\n) (persistence.DomainAuditStore, error) {\n\treturn &sqlDomainAuditStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t\tparser: parser,\n\t\t},\n\t}, nil\n}\n\n// CreateDomainAuditLog creates a new domain audit log entry\nfunc (m *sqlDomainAuditStore) CreateDomainAuditLog(\n\tctx context.Context,\n\trequest *persistence.InternalCreateDomainAuditLogRequest,\n) (*persistence.CreateDomainAuditLogResponse, error) {\n\trow := &sqlplugin.DomainAuditLogRow{\n\t\tDomainID:            request.DomainID,\n\t\tEventID:             request.EventID,\n\t\tStateBefore:         getDataBlobBytes(request.StateBefore),\n\t\tStateBeforeEncoding: getDataBlobEncoding(request.StateBefore),\n\t\tStateAfter:          getDataBlobBytes(request.StateAfter),\n\t\tStateAfterEncoding:  getDataBlobEncoding(request.StateAfter),\n\t\tOperationType:       request.OperationType,\n\t\tCreatedTime:         request.CreatedTime,\n\t\tLastUpdatedTime:     request.LastUpdatedTime,\n\t\tIdentity:            request.Identity,\n\t\tIdentityType:        request.IdentityType,\n\t\tComment:             request.Comment,\n\t}\n\n\t_, err := m.db.InsertIntoDomainAuditLog(ctx, row)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"CreateDomainAuditLog\", \"\", err)\n\t}\n\n\treturn &persistence.CreateDomainAuditLogResponse{\n\t\tEventID: request.EventID,\n\t}, nil\n}\n\n// GetDomainAuditLogs retrieves domain audit logs\nfunc (m *sqlDomainAuditStore) GetDomainAuditLogs(\n\tctx context.Context,\n\trequest *persistence.GetDomainAuditLogsRequest,\n) (*persistence.InternalGetDomainAuditLogsResponse, error) {\n\tminCreatedTime := time.Unix(0, 0)\n\tmaxCreatedTime := time.Now().UTC()\n\tif request.MinCreatedTime != nil {\n\t\tminCreatedTime = *request.MinCreatedTime\n\t}\n\tif request.MaxCreatedTime != nil {\n\t\tmaxCreatedTime = *request.MaxCreatedTime\n\t}\n\n\tpageMaxCreatedTime := maxCreatedTime\n\t// if next page token is not present, set pageMinEventID to largest possible uuid\n\t// to prevent the query from returning rows where created_time is equal to pageMaxCreatedTime\n\tpageMinEventID := \"ffffffff-ffff-ffff-ffff-ffffffffffff\"\n\tif request.NextPageToken != nil {\n\t\tpage := domainAuditLogPageToken{}\n\t\tif err := gobDeserialize(request.NextPageToken, &page); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"unable to decode next page token\")\n\t\t}\n\t\tpageMaxCreatedTime = page.CreatedTime\n\t\tpageMinEventID = page.EventID\n\t}\n\n\tfilter := &sqlplugin.DomainAuditLogFilter{\n\t\tDomainID:           request.DomainID,\n\t\tOperationType:      request.OperationType,\n\t\tMinCreatedTime:     &minCreatedTime,\n\t\tPageSize:           request.PageSize,\n\t\tPageMaxCreatedTime: &pageMaxCreatedTime,\n\t\tPageMinEventID:     &pageMinEventID,\n\t}\n\n\trows, err := m.db.SelectFromDomainAuditLogs(ctx, filter)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetDomainAuditLogs\", \"\", err)\n\t}\n\n\tvar nextPageToken []byte\n\tif request.PageSize > 0 && len(rows) >= request.PageSize {\n\t\t// there could be more results\n\t\tlastRow := rows[request.PageSize-1]\n\t\ttoken := domainAuditLogPageToken{\n\t\t\tCreatedTime: lastRow.CreatedTime,\n\t\t\tEventID:     lastRow.EventID,\n\t\t}\n\t\tnextPageToken, err = gobSerialize(token)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"error serializing nextPageToken:%v\", err)}\n\t\t}\n\t}\n\n\tvar auditLogs []*persistence.InternalDomainAuditLog\n\tfor _, row := range rows {\n\t\tauditLogs = append(auditLogs, deseralizeDomainAuditLogRow(row))\n\t}\n\n\treturn &persistence.InternalGetDomainAuditLogsResponse{\n\t\tAuditLogs:     auditLogs,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc getDataBlobBytes(blob *persistence.DataBlob) []byte {\n\tif blob == nil {\n\t\treturn []byte{}\n\t}\n\treturn blob.Data\n}\n\nfunc getDataBlobEncoding(blob *persistence.DataBlob) constants.EncodingType {\n\tif blob == nil {\n\t\treturn constants.EncodingTypeEmpty\n\t}\n\treturn blob.Encoding\n}\n\nfunc deseralizeDomainAuditLogRow(row *sqlplugin.DomainAuditLogRow) *persistence.InternalDomainAuditLog {\n\tauditLog := &persistence.InternalDomainAuditLog{\n\t\tEventID:         row.EventID,\n\t\tDomainID:        row.DomainID,\n\t\tOperationType:   row.OperationType,\n\t\tCreatedTime:     row.CreatedTime,\n\t\tLastUpdatedTime: row.LastUpdatedTime,\n\t\tIdentity:        row.Identity,\n\t\tIdentityType:    row.IdentityType,\n\t\tComment:         row.Comment,\n\t}\n\n\tif len(row.StateBefore) > 0 {\n\t\tauditLog.StateBefore = &persistence.DataBlob{\n\t\t\tEncoding: row.StateBeforeEncoding,\n\t\t\tData:     row.StateBefore,\n\t\t}\n\t}\n\n\tif len(row.StateAfter) > 0 {\n\t\tauditLog.StateAfter = &persistence.DataBlob{\n\t\t\tEncoding: row.StateAfterEncoding,\n\t\t\tData:     row.StateAfter,\n\t\t}\n\t}\n\n\treturn auditLog\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_domain_audit_store_test.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc setUpMocksForDomainAuditStore(t *testing.T) (*sqlDomainAuditStore, *sqlplugin.MockDB) {\n\tctrl := gomock.NewController(t)\n\tdbMock := sqlplugin.NewMockDB(ctrl)\n\n\tdomainAuditStore := &sqlDomainAuditStore{\n\t\tsqlStore: sqlStore{db: dbMock},\n\t}\n\n\treturn domainAuditStore, dbMock\n}\n\nfunc TestCreateDomainAuditLog(t *testing.T) {\n\tctx := context.Background()\n\tnow := time.Unix(1234567890, 0)\n\n\tstateBeforeBlob := &persistence.DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\tData:     []byte(\"state-before-data\"),\n\t}\n\n\tstateAfterBlob := &persistence.DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\tData:     []byte(\"state-after-data\"),\n\t}\n\n\ttests := map[string]struct {\n\t\tsetupMock   func(*sqlplugin.MockDB)\n\t\trequest     *persistence.InternalCreateDomainAuditLogRequest\n\t\texpectError bool\n\t\texpectedID  string\n\t}{\n\t\t\"success with full data\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\texpectedRow := &sqlplugin.DomainAuditLogRow{\n\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tEventID:             \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tStateBefore:         stateBeforeBlob.Data,\n\t\t\t\t\tStateBeforeEncoding: stateBeforeBlob.Encoding,\n\t\t\t\t\tStateAfter:          stateAfterBlob.Data,\n\t\t\t\t\tStateAfterEncoding:  stateAfterBlob.Encoding,\n\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\tIdentity:            \"test-user\",\n\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\tComment:             \"test comment\",\n\t\t\t\t}\n\t\t\t\tdbMock.EXPECT().InsertIntoDomainAuditLog(ctx, expectedRow).Return(&sqlResult{rowsAffected: 1}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\tEventID:         \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\tStateBefore:     stateBeforeBlob,\n\t\t\t\tStateAfter:      stateAfterBlob,\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"test-user\",\n\t\t\t\tIdentityType:    \"user\",\n\t\t\t\tComment:         \"test comment\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedID:  \"e1111111-1111-1111-1111-111111111111\",\n\t\t},\n\t\t\"success with nil state blobs\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\texpectedRow := &sqlplugin.DomainAuditLogRow{\n\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tEventID:             \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tStateBefore:         []byte{},\n\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeEmpty,\n\t\t\t\t\tStateAfter:          []byte{},\n\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeEmpty,\n\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeCreate,\n\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\tIdentity:            \"system\",\n\t\t\t\t\tIdentityType:        \"system\",\n\t\t\t\t\tComment:             \"\",\n\t\t\t\t}\n\t\t\t\tdbMock.EXPECT().InsertIntoDomainAuditLog(ctx, expectedRow).Return(&sqlResult{rowsAffected: 1}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\tEventID:         \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\tStateBefore:     nil,\n\t\t\t\tStateAfter:      nil,\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeCreate,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"system\",\n\t\t\t\tIdentityType:    \"system\",\n\t\t\t\tComment:         \"\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedID:  \"e1111111-1111-1111-1111-111111111111\",\n\t\t},\n\t\t\"database error\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"database error\")\n\t\t\t\tdbMock.EXPECT().InsertIntoDomainAuditLog(ctx, gomock.Any()).Return(nil, err).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsTimeoutError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsThrottlingError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsDupEntryError(err).Return(false).AnyTimes()\n\t\t\t},\n\t\t\trequest: &persistence.InternalCreateDomainAuditLogRequest{\n\t\t\t\tDomainID:        \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\tEventID:         \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\tOperationType:   persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:     now,\n\t\t\t\tLastUpdatedTime: now,\n\t\t\t\tIdentity:        \"test-user\",\n\t\t\t\tIdentityType:    \"user\",\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore, dbMock := setUpMocksForDomainAuditStore(t)\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\tresp, err := store.CreateDomainAuditLog(ctx, tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\trequire.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, tc.expectedID, resp.EventID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomainAuditLogs(t *testing.T) {\n\tctx := context.Background()\n\tnow := time.Unix(1234567890, 0)\n\tminTime := now.Add(-24 * time.Hour)\n\tmaxTime := now\n\tmaxUUID := \"ffffffff-ffff-ffff-ffff-ffffffffffff\"\n\tpageSize := 2\n\tpageSize2 := 3\n\n\ttests := map[string]struct {\n\t\tsetupMock      func(*sqlplugin.MockDB)\n\t\trequest        *persistence.GetDomainAuditLogsRequest\n\t\texpectError    bool\n\t\texpectedCount  int\n\t\tvalidateResult func(*testing.T, *persistence.InternalGetDomainAuditLogsResponse)\n\t}{\n\t\t\"success - first page with results\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\trows := []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\t\tIdentity:            \"user-1\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 1\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-1\"),\n\t\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-1\"),\n\t\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"e2222222-2222-2222-2222-222222222222\",\n\t\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\t\tIdentity:            \"user-2\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 2\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-2\"),\n\t\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-2\"),\n\t\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeThriftRWSnappy,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\texpectedFilter := &sqlplugin.DomainAuditLogFilter{\n\t\t\t\t\tDomainID:           \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tOperationType:      persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\t\tPageSize:           pageSize,\n\t\t\t\t\tPageMaxCreatedTime: &maxTime,\n\t\t\t\t\tPageMinEventID:     &maxUUID,\n\t\t\t\t}\n\n\t\t\t\tdbMock.EXPECT().SelectFromDomainAuditLogs(ctx, expectedFilter).Return(rows, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:       \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\tOperationType:  persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\tPageSize:       2,\n\t\t\t\tNextPageToken:  nil,\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 2,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\trequire.Len(t, resp.AuditLogs, 2)\n\t\t\t\tassert.Equal(t, \"e1111111-1111-1111-1111-111111111111\", resp.AuditLogs[0].EventID)\n\t\t\t\tassert.Equal(t, \"d1111111-1111-1111-1111-111111111111\", resp.AuditLogs[0].DomainID)\n\t\t\t\tassert.Equal(t, persistence.DomainAuditOperationTypeUpdate, resp.AuditLogs[0].OperationType)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateBefore)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRWSnappy, resp.AuditLogs[0].StateBefore.Encoding)\n\t\t\t\tassert.Equal(t, []byte(\"state-before-1\"), resp.AuditLogs[0].StateBefore.Data)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[0].StateAfter)\n\t\t\t\tassert.Equal(t, constants.EncodingTypeThriftRW, resp.AuditLogs[0].StateAfter.Encoding)\n\n\t\t\t\tassert.Equal(t, \"e2222222-2222-2222-2222-222222222222\", resp.AuditLogs[1].EventID)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[1].StateBefore)\n\t\t\t\tassert.NotNil(t, resp.AuditLogs[1].StateAfter)\n\n\t\t\t\tassert.NotNil(t, resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t\"success - subsequent page with nextPageToken\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\texpectedCreatedTime := now\n\t\t\t\texpectedEventID := \"e1111111-1111-1111-1111-111111111111\"\n\n\t\t\t\trows := []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"e3333333-3333-3333-3333-333333333333\",\n\t\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\t\tCreatedTime:         now.Add(-2 * time.Hour),\n\t\t\t\t\t\tLastUpdatedTime:     now.Add(-2 * time.Hour),\n\t\t\t\t\t\tIdentity:            \"user-3\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 3\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-3\"),\n\t\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-3\"),\n\t\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"e2222222-2222-2222-2222-222222222222\",\n\t\t\t\t\t\tDomainID:            \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\t\tCreatedTime:         expectedCreatedTime,\n\t\t\t\t\t\tLastUpdatedTime:     expectedCreatedTime,\n\t\t\t\t\t\tIdentity:            \"user-2\",\n\t\t\t\t\t\tIdentityType:        \"user\",\n\t\t\t\t\t\tComment:             \"comment 2\",\n\t\t\t\t\t\tStateBefore:         []byte(\"state-before-2\"),\n\t\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\t\t\tStateAfter:          []byte(\"state-after-2\"),\n\t\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\texpectedFilter := &sqlplugin.DomainAuditLogFilter{\n\t\t\t\t\tDomainID:           \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tOperationType:      persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\t\tPageSize:           pageSize2,\n\t\t\t\t\tPageMaxCreatedTime: &expectedCreatedTime,\n\t\t\t\t\tPageMinEventID:     &expectedEventID,\n\t\t\t\t}\n\n\t\t\t\tdbMock.EXPECT().SelectFromDomainAuditLogs(ctx, expectedFilter).Return(rows, nil).Times(1)\n\t\t\t},\n\t\t\trequest: func() *persistence.GetDomainAuditLogsRequest {\n\t\t\t\tpageToken := domainAuditLogPageToken{\n\t\t\t\t\tCreatedTime: now,\n\t\t\t\t\tEventID:     \"e1111111-1111-1111-1111-111111111111\",\n\t\t\t\t}\n\t\t\t\tencodedToken, _ := gobSerialize(pageToken)\n\t\t\t\treturn &persistence.GetDomainAuditLogsRequest{\n\t\t\t\t\tDomainID:       \"d1111111-1111-1111-1111-111111111111\",\n\t\t\t\t\tOperationType:  persistence.DomainAuditOperationTypeUpdate,\n\t\t\t\t\tMinCreatedTime: &minTime,\n\t\t\t\t\tMaxCreatedTime: &maxTime,\n\t\t\t\t\tPageSize:       3,\n\t\t\t\t\tNextPageToken:  encodedToken,\n\t\t\t\t}\n\t\t\t}(),\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 2,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\trequire.Len(t, resp.AuditLogs, 2)\n\t\t\t\tassert.Equal(t, \"e3333333-3333-3333-3333-333333333333\", resp.AuditLogs[0].EventID)\n\t\t\t\tassert.Equal(t, \"e2222222-2222-2222-2222-222222222222\", resp.AuditLogs[1].EventID)\n\t\t\t\tassert.Nil(t, resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t\"success with empty state blobs\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\trows := []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID:             \"e3333333-3333-3333-3333-333333333333\",\n\t\t\t\t\t\tDomainID:            \"d2222222-2222-2222-2222-222222222222\",\n\t\t\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeCreate,\n\t\t\t\t\t\tCreatedTime:         now,\n\t\t\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\t\t\tIdentity:            \"system\",\n\t\t\t\t\t\tIdentityType:        \"system\",\n\t\t\t\t\t\tComment:             \"created\",\n\t\t\t\t\t\tStateBefore:         []byte{}, // Empty slice\n\t\t\t\t\t\tStateBeforeEncoding: constants.EncodingTypeEmpty,\n\t\t\t\t\t\tStateAfter:          nil, // Nil\n\t\t\t\t\t\tStateAfterEncoding:  constants.EncodingTypeEmpty,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tdbMock.EXPECT().SelectFromDomainAuditLogs(ctx, gomock.Any()).Return(rows, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"d2222222-2222-2222-2222-222222222222\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 1,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\trequire.Len(t, resp.AuditLogs, 1)\n\t\t\t\tassert.Equal(t, \"e3333333-3333-3333-3333-333333333333\", resp.AuditLogs[0].EventID)\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateBefore)\n\t\t\t\tassert.Nil(t, resp.AuditLogs[0].StateAfter)\n\t\t\t},\n\t\t},\n\t\t\"success with no results\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\tdbMock.EXPECT().SelectFromDomainAuditLogs(ctx, gomock.Any()).Return([]*sqlplugin.DomainAuditLogRow{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"d3333333-3333-3333-3333-333333333333\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError:   false,\n\t\t\texpectedCount: 0,\n\t\t\tvalidateResult: func(t *testing.T, resp *persistence.InternalGetDomainAuditLogsResponse) {\n\t\t\t\tassert.Empty(t, resp.AuditLogs)\n\t\t\t\tassert.Nil(t, resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t\"invalid page token\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\t// No DB call expected since deserialization should fail first\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID:      \"d3333333-3333-3333-3333-333333333333\",\n\t\t\t\tPageSize:      10,\n\t\t\t\tNextPageToken: []byte(\"invalid-token-data\"),\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"database error\": {\n\t\t\tsetupMock: func(dbMock *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"database error\")\n\t\t\t\tdbMock.EXPECT().SelectFromDomainAuditLogs(ctx, gomock.Any()).Return(nil, err).Times(1)\n\t\t\t\tdbMock.EXPECT().IsNotFoundError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsTimeoutError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsThrottlingError(err).Return(false).AnyTimes()\n\t\t\t\tdbMock.EXPECT().IsDupEntryError(err).Return(false).AnyTimes()\n\t\t\t},\n\t\t\trequest: &persistence.GetDomainAuditLogsRequest{\n\t\t\t\tDomainID: \"d3333333-3333-3333-3333-333333333333\",\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstore, dbMock := setUpMocksForDomainAuditStore(t)\n\t\t\ttc.setupMock(dbMock)\n\n\t\t\tresp, err := store.GetDomainAuditLogs(ctx, tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\trequire.NotNil(t, resp)\n\t\t\t\tassert.Len(t, resp.AuditLogs, tc.expectedCount)\n\t\t\t\tif tc.validateResult != nil {\n\t\t\t\t\ttc.validateResult(t, resp)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDataBlobBytes(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *persistence.DataBlob\n\t\texpected []byte\n\t}{\n\t\t\"nil blob\": {\n\t\t\tinput:    nil,\n\t\t\texpected: []byte{},\n\t\t},\n\t\t\"blob with data\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: []byte(\"test-data\"),\n\t\t},\n\t\t\"blob with empty data\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte{},\n\t\t\t},\n\t\t\texpected: []byte{},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := getDataBlobBytes(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestGetDataBlobEncoding(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput    *persistence.DataBlob\n\t\texpected constants.EncodingType\n\t}{\n\t\t\"nil blob\": {\n\t\t\tinput:    nil,\n\t\t\texpected: constants.EncodingTypeEmpty,\n\t\t},\n\t\t\"blob with ThriftRWSnappy encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRWSnappy,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: constants.EncodingTypeThriftRWSnappy,\n\t\t},\n\t\t\"blob with ThriftRW encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: constants.EncodingTypeThriftRW,\n\t\t},\n\t\t\"blob with empty encoding\": {\n\t\t\tinput: &persistence.DataBlob{\n\t\t\t\tEncoding: \"\",\n\t\t\t\tData:     []byte(\"test-data\"),\n\t\t\t},\n\t\t\texpected: constants.EncodingTypeEmpty,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresult := getDataBlobEncoding(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_domain_store.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlDomainStore struct {\n\tsqlStore\n\tactiveClusterName string\n}\n\n// newMetadataPersistenceV2 creates an instance of sqlDomainStore\nfunc newMetadataPersistenceV2(\n\tdb sqlplugin.DB,\n\tcurrentClusterName string,\n\tlogger log.Logger,\n\tparser serialization.Parser,\n) (persistence.DomainStore, error) {\n\treturn &sqlDomainStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t\tparser: parser,\n\t\t},\n\t\tactiveClusterName: currentClusterName,\n\t}, nil\n}\n\nfunc updateMetadata(ctx context.Context, tx sqlplugin.Tx, oldNotificationVersion int64) error {\n\tresult, err := tx.UpdateDomainMetadata(ctx, &sqlplugin.DomainMetadataRow{NotificationVersion: oldNotificationVersion})\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"updateDomainMetadata\", \"\", err)\n\t}\n\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Could not verify whether domain metadata update occurred. Error: %v\", err),\n\t\t}\n\t} else if rowsAffected != 1 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"Failed to update domain metadata. <>1 rows affected.\",\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc lockMetadata(ctx context.Context, tx sqlplugin.Tx) error {\n\terr := tx.LockDomainMetadata(ctx)\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"lockDomainMetadata\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (m *sqlDomainStore) CreateDomain(\n\tctx context.Context,\n\trequest *persistence.InternalCreateDomainRequest,\n) (*persistence.CreateDomainResponse, error) {\n\tmetadata, err := m.GetMetadata(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tclusters := make([]string, len(request.ReplicationConfig.Clusters))\n\tfor i := range clusters {\n\t\tclusters[i] = request.ReplicationConfig.Clusters[i].ClusterName\n\t}\n\n\tvar badBinaries []byte\n\tbadBinariesEncoding := string(constants.EncodingTypeEmpty)\n\tif request.Config.BadBinaries != nil {\n\t\tbadBinaries = request.Config.BadBinaries.Data\n\t\tbadBinariesEncoding = string(request.Config.BadBinaries.GetEncoding())\n\t}\n\n\tvar isolationGroups []byte\n\tvar isolationGroupsEncoding string\n\tif request.Config != nil && request.Config.IsolationGroups != nil {\n\t\tisolationGroups = request.Config.IsolationGroups.GetData()\n\t\tisolationGroupsEncoding = request.Config.IsolationGroups.GetEncodingString()\n\t}\n\n\tvar asyncWorkflowsCfg []byte\n\tvar asyncWorkflowsEncoding string\n\tif request.Config != nil && request.Config.AsyncWorkflowsConfig != nil {\n\t\tasyncWorkflowsCfg = request.Config.AsyncWorkflowsConfig.GetData()\n\t\tasyncWorkflowsEncoding = request.Config.AsyncWorkflowsConfig.GetEncodingString()\n\t}\n\n\tvar activeClustersConfig []byte\n\tvar activeClustersConfigEncoding string\n\tif request.ReplicationConfig != nil && request.ReplicationConfig.ActiveClustersConfig != nil {\n\t\tactiveClustersConfig = request.ReplicationConfig.ActiveClustersConfig.GetData()\n\t\tactiveClustersConfigEncoding = request.ReplicationConfig.ActiveClustersConfig.GetEncodingString()\n\t}\n\n\tdomainInfo := &serialization.DomainInfo{\n\t\tName:                         request.Info.Name,\n\t\tStatus:                       int32(request.Info.Status),\n\t\tDescription:                  request.Info.Description,\n\t\tOwner:                        request.Info.OwnerEmail,\n\t\tData:                         request.Info.Data,\n\t\tRetention:                    request.Config.Retention,\n\t\tEmitMetric:                   request.Config.EmitMetric,\n\t\tArchivalBucket:               request.Config.ArchivalBucket,\n\t\tArchivalStatus:               int16(request.Config.ArchivalStatus),\n\t\tHistoryArchivalStatus:        int16(request.Config.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:           request.Config.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:     int16(request.Config.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:        request.Config.VisibilityArchivalURI,\n\t\tActiveClusterName:            request.ReplicationConfig.ActiveClusterName,\n\t\tActiveClustersConfig:         activeClustersConfig,\n\t\tActiveClustersConfigEncoding: activeClustersConfigEncoding,\n\t\tClusters:                     clusters,\n\t\tConfigVersion:                request.ConfigVersion,\n\t\tFailoverVersion:              request.FailoverVersion,\n\t\tNotificationVersion:          metadata.NotificationVersion,\n\t\tFailoverNotificationVersion:  persistence.InitialFailoverNotificationVersion,\n\t\tPreviousFailoverVersion:      constants.InitialPreviousFailoverVersion,\n\t\tLastUpdatedTimestamp:         request.LastUpdatedTime,\n\t\tBadBinaries:                  badBinaries,\n\t\tBadBinariesEncoding:          badBinariesEncoding,\n\t\tIsolationGroups:              isolationGroups,\n\t\tIsolationGroupsEncoding:      isolationGroupsEncoding,\n\t\tAsyncWorkflowConfig:          asyncWorkflowsCfg,\n\t\tAsyncWorkflowConfigEncoding:  asyncWorkflowsEncoding,\n\t}\n\n\tblob, err := m.parser.DomainInfoToBlob(domainInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar resp *persistence.CreateDomainResponse\n\terr = m.txExecute(ctx, sqlplugin.DbDefaultShard, \"CreateDomain\", func(tx sqlplugin.Tx) error {\n\t\tif _, err1 := tx.InsertIntoDomain(ctx, &sqlplugin.DomainRow{\n\t\t\tName:         request.Info.Name,\n\t\t\tID:           serialization.MustParseUUID(request.Info.ID),\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\tIsGlobal:     request.IsGlobalDomain,\n\t\t}); err1 != nil {\n\t\t\tif m.db.IsDupEntryError(err1) {\n\t\t\t\treturn &types.DomainAlreadyExistsError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"name: %v\", request.Info.Name),\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn err1\n\t\t}\n\t\tif err1 := lockMetadata(ctx, tx); err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tif err1 := updateMetadata(ctx, tx, metadata.NotificationVersion); err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tresp = &persistence.CreateDomainResponse{ID: request.Info.ID}\n\t\treturn nil\n\t})\n\treturn resp, err\n}\n\nfunc (m *sqlDomainStore) GetDomain(\n\tctx context.Context,\n\trequest *persistence.GetDomainRequest,\n) (*persistence.InternalGetDomainResponse, error) {\n\tfilter := &sqlplugin.DomainFilter{}\n\tswitch {\n\tcase request.Name != \"\" && request.ID != \"\":\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"GetDomain operation failed.  Both ID and Name specified in request.\",\n\t\t}\n\tcase request.Name != \"\":\n\t\tfilter.Name = &request.Name\n\tcase request.ID != \"\":\n\t\tfilter.ID = serialization.UUIDPtr(serialization.MustParseUUID(request.ID))\n\tdefault:\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"GetDomain operation failed.  Both ID and Name are empty.\",\n\t\t}\n\t}\n\n\trows, err := m.db.SelectFromDomain(ctx, filter)\n\tif err != nil {\n\t\tswitch err {\n\t\tcase sql.ErrNoRows:\n\t\t\t// We did not return in the above for-loop because there were no rows.\n\t\t\tidentity := request.Name\n\t\t\tif len(request.ID) > 0 {\n\t\t\t\tidentity = request.ID\n\t\t\t}\n\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Domain %s does not exist.\", identity),\n\t\t\t}\n\t\tdefault:\n\t\t\treturn nil, convertCommonErrors(m.db, \"GetDomain\", \"\", err)\n\t\t}\n\t}\n\n\tresponse, err := m.domainRowToGetDomainResponse(&rows[0])\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc (m *sqlDomainStore) domainRowToGetDomainResponse(row *sqlplugin.DomainRow) (*persistence.InternalGetDomainResponse, error) {\n\tdomainInfo, err := m.parser.DomainInfoFromBlob(row.Data, row.DataEncoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tclusters := make([]*persistence.ClusterReplicationConfig, len(domainInfo.Clusters))\n\tfor i := range domainInfo.Clusters {\n\t\tclusters[i] = &persistence.ClusterReplicationConfig{ClusterName: domainInfo.Clusters[i]}\n\t}\n\n\tvar badBinaries *persistence.DataBlob\n\tif domainInfo.BadBinaries != nil {\n\t\tbadBinaries = persistence.NewDataBlob(domainInfo.BadBinaries, constants.EncodingType(domainInfo.GetBadBinariesEncoding()))\n\t}\n\n\tvar isolationGroups *persistence.DataBlob\n\tif domainInfo.IsolationGroups != nil {\n\t\tisolationGroups = persistence.NewDataBlob(domainInfo.IsolationGroups, constants.EncodingType(domainInfo.IsolationGroupsEncoding))\n\t}\n\n\tvar asyncWorkflowsCfg *persistence.DataBlob\n\tif domainInfo.AsyncWorkflowConfig != nil {\n\t\tasyncWorkflowsCfg = persistence.NewDataBlob(domainInfo.AsyncWorkflowConfig, constants.EncodingType(domainInfo.AsyncWorkflowConfigEncoding))\n\t}\n\n\tactiveClusterName := cluster.GetOrUseDefaultActiveCluster(m.activeClusterName, domainInfo.GetActiveClusterName())\n\n\tvar activeClustersConfig *persistence.DataBlob\n\tif domainInfo.ActiveClustersConfig != nil {\n\t\tactiveClustersConfig = persistence.NewDataBlob(domainInfo.ActiveClustersConfig, constants.EncodingType(domainInfo.ActiveClustersConfigEncoding))\n\t\t// for active-active domains, do not populate active cluster name with current cluster name if it is not set\n\t\tactiveClusterName = domainInfo.GetActiveClusterName()\n\t}\n\n\treturn &persistence.InternalGetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          row.ID.String(),\n\t\t\tName:        row.Name,\n\t\t\tStatus:      int(domainInfo.GetStatus()),\n\t\t\tDescription: domainInfo.GetDescription(),\n\t\t\tOwnerEmail:  domainInfo.GetOwner(),\n\t\t\tData:        domainInfo.GetData(),\n\t\t},\n\t\tConfig: &persistence.InternalDomainConfig{\n\t\t\tRetention:                domainInfo.GetRetention(),\n\t\t\tEmitMetric:               domainInfo.GetEmitMetric(),\n\t\t\tArchivalBucket:           domainInfo.GetArchivalBucket(),\n\t\t\tArchivalStatus:           types.ArchivalStatus(domainInfo.GetArchivalStatus()),\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatus(domainInfo.GetHistoryArchivalStatus()),\n\t\t\tHistoryArchivalURI:       domainInfo.GetHistoryArchivalURI(),\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatus(domainInfo.GetVisibilityArchivalStatus()),\n\t\t\tVisibilityArchivalURI:    domainInfo.GetVisibilityArchivalURI(),\n\t\t\tBadBinaries:              badBinaries,\n\t\t\tIsolationGroups:          isolationGroups,\n\t\t\tAsyncWorkflowsConfig:     asyncWorkflowsCfg,\n\t\t},\n\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\tActiveClusterName:    activeClusterName,\n\t\t\tClusters:             cluster.GetOrUseDefaultClusters(m.activeClusterName, clusters),\n\t\t\tActiveClustersConfig: activeClustersConfig,\n\t\t},\n\t\tIsGlobalDomain:              row.IsGlobal,\n\t\tFailoverVersion:             domainInfo.GetFailoverVersion(),\n\t\tConfigVersion:               domainInfo.GetConfigVersion(),\n\t\tNotificationVersion:         domainInfo.GetNotificationVersion(),\n\t\tFailoverNotificationVersion: domainInfo.GetFailoverNotificationVersion(),\n\t\tPreviousFailoverVersion:     domainInfo.GetPreviousFailoverVersion(),\n\t\tFailoverEndTime:             domainInfo.FailoverEndTimestamp,\n\t\tLastUpdatedTime:             domainInfo.GetLastUpdatedTimestamp(),\n\t}, nil\n}\n\nfunc (m *sqlDomainStore) UpdateDomain(\n\tctx context.Context,\n\trequest *persistence.InternalUpdateDomainRequest,\n) error {\n\n\tclusters := make([]string, len(request.ReplicationConfig.Clusters))\n\tfor i := range clusters {\n\t\tclusters[i] = request.ReplicationConfig.Clusters[i].ClusterName\n\t}\n\n\tvar badBinaries []byte\n\tbadBinariesEncoding := string(constants.EncodingTypeEmpty)\n\tif request.Config.BadBinaries != nil {\n\t\tbadBinaries = request.Config.BadBinaries.Data\n\t\tbadBinariesEncoding = string(request.Config.BadBinaries.GetEncoding())\n\t}\n\n\tvar isolationGroups []byte\n\tisolationGroupsEncoding := string(constants.EncodingTypeEmpty)\n\tif request.Config.IsolationGroups != nil {\n\t\tisolationGroups = request.Config.IsolationGroups.Data\n\t\tisolationGroupsEncoding = request.Config.IsolationGroups.GetEncodingString()\n\t}\n\n\tvar asyncWorkflowsCfg []byte\n\tasyncWorkflowsEncoding := string(constants.EncodingTypeEmpty)\n\tif request.Config.AsyncWorkflowsConfig != nil {\n\t\tasyncWorkflowsCfg = request.Config.AsyncWorkflowsConfig.Data\n\t\tasyncWorkflowsEncoding = request.Config.AsyncWorkflowsConfig.GetEncodingString()\n\t}\n\n\tvar activeClustersConfig []byte\n\tvar activeClustersConfigEncoding string\n\tif request.ReplicationConfig.ActiveClustersConfig != nil {\n\t\tactiveClustersConfig = request.ReplicationConfig.ActiveClustersConfig.Data\n\t\tactiveClustersConfigEncoding = request.ReplicationConfig.ActiveClustersConfig.GetEncodingString()\n\t}\n\n\tdomainInfo := &serialization.DomainInfo{\n\t\tStatus:                       int32(request.Info.Status),\n\t\tDescription:                  request.Info.Description,\n\t\tOwner:                        request.Info.OwnerEmail,\n\t\tData:                         request.Info.Data,\n\t\tRetention:                    request.Config.Retention,\n\t\tEmitMetric:                   request.Config.EmitMetric,\n\t\tArchivalBucket:               request.Config.ArchivalBucket,\n\t\tArchivalStatus:               int16(request.Config.ArchivalStatus),\n\t\tHistoryArchivalStatus:        int16(request.Config.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:           request.Config.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:     int16(request.Config.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:        request.Config.VisibilityArchivalURI,\n\t\tActiveClusterName:            request.ReplicationConfig.ActiveClusterName,\n\t\tActiveClustersConfig:         activeClustersConfig,\n\t\tActiveClustersConfigEncoding: activeClustersConfigEncoding,\n\t\tClusters:                     clusters,\n\t\tConfigVersion:                request.ConfigVersion,\n\t\tFailoverVersion:              request.FailoverVersion,\n\t\tNotificationVersion:          request.NotificationVersion,\n\t\tFailoverNotificationVersion:  request.FailoverNotificationVersion,\n\t\tPreviousFailoverVersion:      request.PreviousFailoverVersion,\n\t\tFailoverEndTimestamp:         request.FailoverEndTime,\n\t\tLastUpdatedTimestamp:         request.LastUpdatedTime,\n\t\tBadBinaries:                  badBinaries,\n\t\tBadBinariesEncoding:          badBinariesEncoding,\n\t\tIsolationGroups:              isolationGroups,\n\t\tIsolationGroupsEncoding:      isolationGroupsEncoding,\n\t\tAsyncWorkflowConfig:          asyncWorkflowsCfg,\n\t\tAsyncWorkflowConfigEncoding:  asyncWorkflowsEncoding,\n\t}\n\n\tblob, err := m.parser.DomainInfoToBlob(domainInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn m.txExecute(ctx, sqlplugin.DbDefaultShard, \"UpdateDomain\", func(tx sqlplugin.Tx) error {\n\t\tresult, err := tx.UpdateDomain(ctx, &sqlplugin.DomainRow{\n\t\t\tName:         request.Info.Name,\n\t\t\tID:           serialization.MustParseUUID(request.Info.ID),\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tnoRowsAffected, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"rowsAffected error: %v\", err)\n\t\t}\n\t\tif noRowsAffected != 1 {\n\t\t\treturn fmt.Errorf(\"%v rows updated instead of one\", noRowsAffected)\n\t\t}\n\t\tif err := lockMetadata(ctx, tx); err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn updateMetadata(ctx, tx, request.NotificationVersion)\n\t})\n}\n\nfunc (m *sqlDomainStore) DeleteDomain(\n\tctx context.Context,\n\trequest *persistence.DeleteDomainRequest,\n) error {\n\treturn m.txExecute(ctx, sqlplugin.DbDefaultShard, \"DeleteDomain\", func(tx sqlplugin.Tx) error {\n\t\t_, err := tx.DeleteFromDomain(ctx, &sqlplugin.DomainFilter{ID: serialization.UUIDPtr(serialization.MustParseUUID(request.ID))})\n\t\treturn err\n\t})\n}\n\nfunc (m *sqlDomainStore) DeleteDomainByName(\n\tctx context.Context,\n\trequest *persistence.DeleteDomainByNameRequest,\n) error {\n\treturn m.txExecute(ctx, sqlplugin.DbDefaultShard, \"DeleteDomainByName\", func(tx sqlplugin.Tx) error {\n\t\t_, err := tx.DeleteFromDomain(ctx, &sqlplugin.DomainFilter{Name: &request.Name})\n\t\treturn err\n\t})\n}\n\nfunc (m *sqlDomainStore) GetMetadata(\n\tctx context.Context,\n) (*persistence.GetMetadataResponse, error) {\n\trow, err := m.db.SelectFromDomainMetadata(ctx)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetMetadata\", \"\", err)\n\t}\n\treturn &persistence.GetMetadataResponse{NotificationVersion: row.NotificationVersion}, nil\n}\n\nfunc (m *sqlDomainStore) ListDomains(\n\tctx context.Context,\n\trequest *persistence.ListDomainsRequest,\n) (*persistence.InternalListDomainsResponse, error) {\n\tvar pageToken *serialization.UUID\n\tif request.NextPageToken != nil {\n\t\ttoken := serialization.UUID(request.NextPageToken)\n\t\tpageToken = &token\n\t}\n\trows, err := m.db.SelectFromDomain(ctx, &sqlplugin.DomainFilter{\n\t\tGreaterThanID: pageToken,\n\t\tPageSize:      &request.PageSize,\n\t})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn &persistence.InternalListDomainsResponse{}, nil\n\t\t}\n\t\treturn nil, convertCommonErrors(m.db, \"ListDomains\", \"Failed to get domain rows.\", err)\n\t}\n\n\tvar domains []*persistence.InternalGetDomainResponse\n\tfor _, row := range rows {\n\t\tresp, err := m.domainRowToGetDomainResponse(&row)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdomains = append(domains, resp)\n\t}\n\n\tresp := &persistence.InternalListDomainsResponse{Domains: domains}\n\tif len(rows) >= request.PageSize {\n\t\tresp.NextPageToken = rows[len(rows)-1].ID\n\t}\n\n\treturn resp, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_domain_store_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlResult struct {\n\tlastInsertID int64\n\trowsAffected int64\n\terr          error\n}\n\nfunc (r *sqlResult) LastInsertId() (int64, error) {\n\tif r.err != nil {\n\t\treturn 0, r.err\n\t}\n\treturn r.lastInsertID, nil\n}\n\nfunc (r *sqlResult) RowsAffected() (int64, error) {\n\tif r.err != nil {\n\t\treturn 0, r.err\n\t}\n\treturn r.rowsAffected, nil\n}\n\nfunc TestGetMetadata(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.GetMetadataResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t},\n\t\t\twant:    &persistence.GetMetadataResponse{NotificationVersion: 2},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore := &sqlDomainStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.GetMetadata(context.Background())\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListDomains(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tactiveClusterName string\n\t\treq               *persistence.ListDomainsRequest\n\t\tmockSetup         func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant              *persistence.InternalListDomainsResponse\n\t\twantErr           bool\n\t}{\n\t\t{\n\t\t\tname:              \"Success case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.ListDomainsRequest{\n\t\t\t\tNextPageToken: []byte(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`),\n\t\t\t\tPageSize:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.UUID(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`)\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tGreaterThanID: &uuid,\n\t\t\t\t\tPageSize:      common.IntPtr(1),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).\n\t\t\t\t\tReturn(&serialization.DomainInfo{}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalListDomainsResponse{\n\t\t\t\tDomains: []*persistence.InternalGetDomainResponse{\n\t\t\t\t\t{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\t\t\tName: \"test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.InternalDomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t\t{ClusterName: \"active\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Success case active-active domain\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.ListDomainsRequest{\n\t\t\t\tNextPageToken: []byte(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`),\n\t\t\t\tPageSize:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.UUID(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`)\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tGreaterThanID: &uuid,\n\t\t\t\t\tPageSize:      common.IntPtr(1),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).\n\t\t\t\t\tReturn(&serialization.DomainInfo{\n\t\t\t\t\t\tActiveClustersConfig:         []byte(`active-clusters-config`),\n\t\t\t\t\t\tActiveClustersConfigEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalListDomainsResponse{\n\t\t\t\tDomains: []*persistence.InternalGetDomainResponse{\n\t\t\t\t\t{\n\t\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\t\t\tName: \"test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tConfig: &persistence.InternalDomainConfig{},\n\t\t\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\t\t\tActiveClusterName: \"\",\n\t\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t\t{ClusterName: \"active\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tActiveClustersConfig: &persistence.DataBlob{\n\t\t\t\t\t\t\t\tData:     []byte(`active-clusters-config`),\n\t\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"No Record case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.ListDomainsRequest{\n\t\t\t\tNextPageToken: []byte(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`),\n\t\t\t\tPageSize:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.UUID(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`)\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tGreaterThanID: &uuid,\n\t\t\t\t\tPageSize:      common.IntPtr(1),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twant:    &persistence.InternalListDomainsResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error from database case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.ListDomainsRequest{\n\t\t\t\tNextPageToken: []byte(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`),\n\t\t\t\tPageSize:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.UUID(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tGreaterThanID: &uuid,\n\t\t\t\t\tPageSize:      common.IntPtr(1),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error from corrupted data case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.ListDomainsRequest{\n\t\t\t\tNextPageToken: []byte(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`),\n\t\t\t\tPageSize:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.UUID(`7a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c`)\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tGreaterThanID: &uuid,\n\t\t\t\t\tPageSize:      common.IntPtr(1),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).Return(nil, errors.New(\"corrupted blob\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlDomainStore{\n\t\t\t\tsqlStore:          sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tactiveClusterName: tc.activeClusterName,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tgot, err := store.ListDomains(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tactiveClusterName string\n\t\treq               *persistence.GetDomainRequest\n\t\tmockSetup         func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant              *persistence.InternalGetDomainResponse\n\t\twantErr           bool\n\t\tassertErr         func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:              \"Success case - get domain by name\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tName: common.StringPtr(\"test\"),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).\n\t\t\t\t\tReturn(&serialization.DomainInfo{}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig: &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Success case - active-active domain - get domain by name\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tName: common.StringPtr(\"test\"),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).\n\t\t\t\t\tReturn(&serialization.DomainInfo{\n\t\t\t\t\t\tActiveClustersConfig:         []byte(`active-clusters-config`),\n\t\t\t\t\t\tActiveClustersConfigEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig: &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClustersConfig: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`active-clusters-config`),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Success case - get domain by ID\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tID: \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tuuid := serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\")\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tID: &uuid,\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).Return(&serialization.DomainInfo{}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetDomainResponse{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig: &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - both domain name and ID are set in request\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {},\n\t\t\twantErr:   true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.BadRequestError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be BadRequestError\")\n\t\t\t\tassert.Contains(t, err.Error(), \"GetDomain operation failed.  Both ID and Name specified in request.\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - both domain name and ID are NOT set in request\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq:               &persistence.GetDomainRequest{},\n\t\t\tmockSetup:         func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {},\n\t\t\twantErr:           true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.BadRequestError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be BadRequestError\")\n\t\t\t\tassert.Contains(t, err.Error(), \"GetDomain operation failed.  Both ID and Name are empty.\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"No Record case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tName: common.StringPtr(\"test\"),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.EntityNotExistsError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be EntityNotExistsError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"Error from database case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tName: common.StringPtr(\"test\"),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error from corrupted data case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.GetDomainRequest{\n\t\t\t\tName: \"test\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomain(gomock.Any(), &sqlplugin.DomainFilter{\n\t\t\t\t\tName: common.StringPtr(\"test\"),\n\t\t\t\t}).Return([]sqlplugin.DomainRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\t\tName:         \"test\",\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoFromBlob([]byte(`aaaa`), string(constants.EncodingTypeThriftRW)).Return(nil, errors.New(\"corrupted blob\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlDomainStore{\n\t\t\t\tsqlStore:          sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tactiveClusterName: tc.activeClusterName,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tgot, err := store.GetDomain(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tactiveClusterName string\n\t\treq               *persistence.InternalCreateDomainRequest\n\t\tmockSetup         func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twant              *persistence.CreateDomainResponse\n\t\twantErr           bool\n\t\tassertErr         func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:              \"Success case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:          \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName:        \"test\",\n\t\t\t\t\tStatus:      1,\n\t\t\t\t\tDescription: \"n/a\",\n\t\t\t\t\tOwnerEmail:  \"abc@xyz.com\",\n\t\t\t\t\tData:        map[string]string{\"k\": \"v\"},\n\t\t\t\t},\n\t\t\t\tConfig: &persistence.InternalDomainConfig{\n\t\t\t\t\tRetention:                time.Hour * 24,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"http://a.b\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"http://x.y\",\n\t\t\t\t\tBadBinaries:              nil,\n\t\t\t\t\tIsolationGroups:          nil,\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClustersConfig: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`active-clusters-config`),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tConfigVersion:   1,\n\t\t\t\tFailoverVersion: 3,\n\t\t\t\tIsGlobalDomain:  true,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(&serialization.DomainInfo{\n\t\t\t\t\tName:                         \"test\",\n\t\t\t\t\tStatus:                       1,\n\t\t\t\t\tDescription:                  \"n/a\",\n\t\t\t\t\tOwner:                        \"abc@xyz.com\",\n\t\t\t\t\tData:                         map[string]string{\"k\": \"v\"},\n\t\t\t\t\tRetention:                    time.Hour * 24,\n\t\t\t\t\tEmitMetric:                   true,\n\t\t\t\t\tHistoryArchivalStatus:        int16(types.ArchivalStatusEnabled),\n\t\t\t\t\tHistoryArchivalURI:           \"http://a.b\",\n\t\t\t\t\tVisibilityArchivalStatus:     int16(types.ArchivalStatusEnabled),\n\t\t\t\t\tVisibilityArchivalURI:        \"http://x.y\",\n\t\t\t\t\tActiveClusterName:            \"active\",\n\t\t\t\t\tClusters:                     []string{\"active\"},\n\t\t\t\t\tConfigVersion:                1,\n\t\t\t\t\tFailoverVersion:              3,\n\t\t\t\t\tNotificationVersion:          2,\n\t\t\t\t\tFailoverNotificationVersion:  persistence.InitialFailoverNotificationVersion,\n\t\t\t\t\tPreviousFailoverVersion:      constants.InitialPreviousFailoverVersion,\n\t\t\t\t\tActiveClustersConfig:         []byte(`active-clusters-config`),\n\t\t\t\t\tActiveClustersConfigEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoDomain(gomock.Any(), &sqlplugin.DomainRow{\n\t\t\t\t\tName:         \"test\",\n\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t\tIsGlobal:     true,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().UpdateDomainMetadata(gomock.Any(), &sqlplugin.DomainMetadataRow{NotificationVersion: 2}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.CreateDomainResponse{ID: \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to get metadata\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq:               &persistence.InternalCreateDomainRequest{},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to encode data\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to insert row\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoDomain(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(err).Return(false)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - domain already exists\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t\tConfigVersion:     1,\n\t\t\t\tFailoverVersion:   3,\n\t\t\t\tIsGlobalDomain:    true,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoDomain(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.DomainAlreadyExistsError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be DomainAlreadyExistsError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to lock metadata\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoDomain(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to update metadata\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromDomainMetadata(gomock.Any()).Return(&sqlplugin.DomainMetadataRow{NotificationVersion: 2}, nil)\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoDomain(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateDomainMetadata(gomock.Any(), &sqlplugin.DomainMetadataRow{NotificationVersion: 2}).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlDomainStore{\n\t\t\t\tsqlStore:          sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tactiveClusterName: tc.activeClusterName,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\t\t\tgot, err := store.CreateDomain(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tactiveClusterName string\n\t\treq               *persistence.InternalUpdateDomainRequest\n\t\tmockSetup         func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr           bool\n\t\tassertErr         func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:              \"Success case\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:          \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName:        \"test\",\n\t\t\t\t\tStatus:      1,\n\t\t\t\t\tDescription: \"n/a\",\n\t\t\t\t\tOwnerEmail:  \"abc@xyz.com\",\n\t\t\t\t\tData:        map[string]string{\"k\": \"v\"},\n\t\t\t\t},\n\t\t\t\tConfig: &persistence.InternalDomainConfig{\n\t\t\t\t\tRetention:                time.Hour * 24,\n\t\t\t\t\tEmitMetric:               true,\n\t\t\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\t\t\tHistoryArchivalURI:       \"http://a.b\",\n\t\t\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\t\t\tVisibilityArchivalURI:    \"http://x.y\",\n\t\t\t\t\tBadBinaries:              nil,\n\t\t\t\t\tIsolationGroups:          nil,\n\t\t\t\t},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tActiveClustersConfig: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`active-clusters-config`),\n\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tConfigVersion:               1,\n\t\t\t\tFailoverVersion:             3,\n\t\t\t\tFailoverNotificationVersion: 4,\n\t\t\t\tPreviousFailoverVersion:     5,\n\t\t\t\tNotificationVersion:         2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(&serialization.DomainInfo{\n\t\t\t\t\tStatus:                       1,\n\t\t\t\t\tDescription:                  \"n/a\",\n\t\t\t\t\tOwner:                        \"abc@xyz.com\",\n\t\t\t\t\tData:                         map[string]string{\"k\": \"v\"},\n\t\t\t\t\tRetention:                    time.Hour * 24,\n\t\t\t\t\tEmitMetric:                   true,\n\t\t\t\t\tHistoryArchivalStatus:        int16(types.ArchivalStatusEnabled),\n\t\t\t\t\tHistoryArchivalURI:           \"http://a.b\",\n\t\t\t\t\tVisibilityArchivalStatus:     int16(types.ArchivalStatusEnabled),\n\t\t\t\t\tVisibilityArchivalURI:        \"http://x.y\",\n\t\t\t\t\tActiveClusterName:            \"active\",\n\t\t\t\t\tClusters:                     []string{\"active\"},\n\t\t\t\t\tConfigVersion:                1,\n\t\t\t\t\tFailoverVersion:              3,\n\t\t\t\t\tNotificationVersion:          2,\n\t\t\t\t\tFailoverNotificationVersion:  4,\n\t\t\t\t\tPreviousFailoverVersion:      5,\n\t\t\t\t\tActiveClustersConfig:         []byte(`active-clusters-config`),\n\t\t\t\t\tActiveClustersConfigEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().UpdateDomain(gomock.Any(), &sqlplugin.DomainRow{\n\t\t\t\t\tName:         \"test\",\n\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().UpdateDomainMetadata(gomock.Any(), &sqlplugin.DomainMetadataRow{NotificationVersion: 2}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to encode data\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to update row\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateDomain(gomock.Any(), &sqlplugin.DomainRow{\n\t\t\t\t\tName:         \"test\",\n\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to lock metadata\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:            &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig: &persistence.InternalDomainReplicationConfig{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().UpdateDomain(gomock.Any(), &sqlplugin.DomainRow{\n\t\t\t\t\tName:         \"test\",\n\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Error case - unable to update metadata\",\n\t\t\tactiveClusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateDomainRequest{\n\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\tID:   \"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\",\n\t\t\t\t\tName: \"test\",\n\t\t\t\t},\n\t\t\t\tConfig:              &persistence.InternalDomainConfig{},\n\t\t\t\tReplicationConfig:   &persistence.InternalDomainReplicationConfig{},\n\t\t\t\tNotificationVersion: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().DomainInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`aaaa`), Encoding: constants.EncodingTypeThriftRW}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().UpdateDomain(gomock.Any(), &sqlplugin.DomainRow{\n\t\t\t\t\tName:         \"test\",\n\t\t\t\t\tID:           serialization.MustParseUUID(\"9a3dc7e2-1e67-41aa-8eaf-6d6e27f7e47c\"),\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: string(constants.EncodingTypeThriftRW),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().LockDomainMetadata(gomock.Any()).Return(nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateDomainMetadata(gomock.Any(), &sqlplugin.DomainMetadataRow{NotificationVersion: 2}).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlDomainStore{\n\t\t\t\tsqlStore:          sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tactiveClusterName: tc.activeClusterName,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\t\t\terr := store.UpdateDomain(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_execution_store.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"math\"\n\t\"runtime/debug\"\n\t\"time\"\n\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\temptyWorkflowID       string = \"\"\n\temptyReplicationRunID string = \"30000000-5000-f000-f000-000000000000\"\n)\n\ntype sqlExecutionStore struct {\n\tsqlStore\n\tshardID                                int\n\ttaskSerializer                         serialization.TaskSerializer\n\ttxExecuteShardLockedFn                 func(context.Context, int, string, int64, func(sqlplugin.Tx) error) error\n\tlockCurrentExecutionIfExistsFn         func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error)\n\tcreateOrUpdateCurrentExecutionFn       func(context.Context, sqlplugin.Tx, p.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error\n\tassertNotCurrentExecutionFn            func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error\n\tassertRunIDAndUpdateCurrentExecutionFn func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error\n\tapplyWorkflowSnapshotTxAsNewFn         func(context.Context, sqlplugin.Tx, int, *p.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n\tapplyWorkflowMutationTxFn              func(context.Context, sqlplugin.Tx, int, *p.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error\n\tapplyWorkflowSnapshotTxAsResetFn       func(context.Context, sqlplugin.Tx, int, *p.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n}\n\nvar _ p.ExecutionStore = (*sqlExecutionStore)(nil)\n\n// NewSQLExecutionStore creates an instance of ExecutionStore\nfunc NewSQLExecutionStore(\n\tdb sqlplugin.DB,\n\tlogger log.Logger,\n\tshardID int,\n\tparser serialization.Parser,\n\ttaskSerializer serialization.TaskSerializer,\n\tdc *p.DynamicConfiguration,\n) (p.ExecutionStore, error) {\n\n\tstore := &sqlExecutionStore{\n\t\tshardID:                                shardID,\n\t\tlockCurrentExecutionIfExistsFn:         lockCurrentExecutionIfExists,\n\t\tcreateOrUpdateCurrentExecutionFn:       createOrUpdateCurrentExecution,\n\t\tassertNotCurrentExecutionFn:            assertNotCurrentExecution,\n\t\tassertRunIDAndUpdateCurrentExecutionFn: assertRunIDAndUpdateCurrentExecution,\n\t\tapplyWorkflowSnapshotTxAsNewFn:         applyWorkflowSnapshotTxAsNew,\n\t\tapplyWorkflowMutationTxFn:              applyWorkflowMutationTx,\n\t\tapplyWorkflowSnapshotTxAsResetFn:       applyWorkflowSnapshotTxAsReset,\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t\tparser: parser,\n\t\t\tdc:     dc,\n\t\t},\n\t\ttaskSerializer: taskSerializer,\n\t}\n\tstore.txExecuteShardLockedFn = store.txExecuteShardLocked\n\treturn store, nil\n}\n\n// txExecuteShardLocked executes f under transaction and with read lock on shard row\nfunc (m *sqlExecutionStore) txExecuteShardLocked(\n\tctx context.Context,\n\tdbShardID int,\n\toperation string,\n\trangeID int64,\n\tfn func(tx sqlplugin.Tx) error,\n) error {\n\n\treturn m.txExecute(ctx, dbShardID, operation, func(tx sqlplugin.Tx) error {\n\t\tif err := readLockShard(ctx, tx, m.shardID, rangeID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr := fn(tx)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t})\n}\n\nfunc (m *sqlExecutionStore) GetShardID() int {\n\treturn m.shardID\n}\n\nfunc (m *sqlExecutionStore) CreateWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalCreateWorkflowExecutionRequest,\n) (response *p.CreateWorkflowExecutionResponse, err error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(m.shardID, m.db.GetTotalNumDBShards())\n\n\terr = m.txExecuteShardLockedFn(ctx, dbShardID, \"CreateWorkflowExecution\", request.RangeID, func(tx sqlplugin.Tx) error {\n\t\tresponse, err = m.createWorkflowExecutionTx(ctx, tx, request)\n\t\treturn err\n\t})\n\treturn\n}\n\nfunc (m *sqlExecutionStore) createWorkflowExecutionTx(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\trequest *p.InternalCreateWorkflowExecutionRequest,\n) (*p.CreateWorkflowExecutionResponse, error) {\n\n\tnewWorkflow := request.NewWorkflowSnapshot\n\texecutionInfo := newWorkflow.ExecutionInfo\n\tstartVersion := newWorkflow.StartVersion\n\tlastWriteVersion := newWorkflow.LastWriteVersion\n\tshardID := m.shardID\n\tdomainID := serialization.MustParseUUID(executionInfo.DomainID)\n\tworkflowID := executionInfo.WorkflowID\n\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\n\tif err := p.ValidateCreateWorkflowModeState(\n\t\trequest.Mode,\n\t\tnewWorkflow,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar err error\n\tvar row *sqlplugin.CurrentExecutionsRow\n\tif row, err = m.lockCurrentExecutionIfExistsFn(ctx, tx, m.shardID, domainID, workflowID); err != nil {\n\t\treturn nil, err\n\t}\n\n\t// current workflow record check\n\tif row != nil {\n\t\t// current run ID, last write version, current workflow state check\n\t\tswitch request.Mode {\n\t\tcase p.CreateWorkflowModeBrandNew:\n\t\t\treturn nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\t\t\tMsg:              fmt.Sprintf(\"Workflow execution already running. WorkflowId: %v\", row.WorkflowID),\n\t\t\t\tStartRequestID:   row.CreateRequestID,\n\t\t\t\tRunID:            row.RunID.String(),\n\t\t\t\tState:            int(row.State),\n\t\t\t\tCloseStatus:      int(row.CloseStatus),\n\t\t\t\tLastWriteVersion: row.LastWriteVersion,\n\t\t\t}\n\n\t\tcase p.CreateWorkflowModeWorkflowIDReuse:\n\t\t\tif request.PreviousLastWriteVersion != row.LastWriteVersion {\n\t\t\t\treturn nil, &p.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, \"+\n\t\t\t\t\t\t\"LastWriteVersion: %v, PreviousLastWriteVersion: %v\",\n\t\t\t\t\t\tworkflowID, row.LastWriteVersion, request.PreviousLastWriteVersion),\n\t\t\t\t}\n\t\t\t}\n\t\t\tif row.State != p.WorkflowStateCompleted {\n\t\t\t\treturn nil, &p.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, \"+\n\t\t\t\t\t\t\"State: %v, Expected: %v\",\n\t\t\t\t\t\tworkflowID, row.State, p.WorkflowStateCompleted),\n\t\t\t\t}\n\t\t\t}\n\t\t\trunIDStr := row.RunID.String()\n\t\t\tif runIDStr != request.PreviousRunID {\n\t\t\t\treturn nil, &p.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, \"+\n\t\t\t\t\t\t\"RunID: %v, PreviousRunID: %v\",\n\t\t\t\t\t\tworkflowID, runIDStr, request.PreviousRunID),\n\t\t\t\t}\n\t\t\t}\n\n\t\tcase p.CreateWorkflowModeZombie:\n\t\t\t// zombie workflow creation with existence of current record, this is a noop\n\t\t\tif err := assertRunIDMismatch(serialization.MustParseUUID(executionInfo.RunID), row.RunID); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase p.CreateWorkflowModeContinueAsNew:\n\t\t\trunIDStr := row.RunID.String()\n\t\t\tif runIDStr != request.PreviousRunID {\n\t\t\t\treturn nil, &p.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: fmt.Sprintf(\"Workflow execution creation condition failed. WorkflowId: %v, \"+\n\t\t\t\t\t\t\"RunID: %v, PreviousRunID: %v\",\n\t\t\t\t\t\tworkflowID, runIDStr, request.PreviousRunID),\n\t\t\t\t}\n\t\t\t}\n\n\t\tdefault:\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\t\"CreteWorkflowExecution: unknown mode: %v\",\n\t\t\t\t\trequest.Mode,\n\t\t\t\t),\n\t\t\t}\n\t\t}\n\t}\n\n\tif err := m.createOrUpdateCurrentExecutionFn(\n\t\tctx,\n\t\ttx,\n\t\trequest.Mode,\n\t\tm.shardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\texecutionInfo.State,\n\t\texecutionInfo.CloseStatus,\n\t\texecutionInfo.CreateRequestID,\n\t\tstartVersion,\n\t\tlastWriteVersion); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := m.applyWorkflowSnapshotTxAsNewFn(ctx, tx, shardID, &request.NewWorkflowSnapshot, m.parser, m.taskSerializer); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &p.CreateWorkflowExecutionResponse{}, nil\n}\n\nfunc (m *sqlExecutionStore) getExecutions(\n\tctx context.Context,\n\trequest *p.InternalGetWorkflowExecutionRequest,\n\tdomainID serialization.UUID,\n\twfID string,\n\trunID serialization.UUID,\n) ([]sqlplugin.ExecutionsRow, error) {\n\texecutions, err := m.db.SelectFromExecutions(ctx, &sqlplugin.ExecutionsFilter{\n\t\tShardID: m.shardID, DomainID: domainID, WorkflowID: wfID, RunID: runID})\n\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\t\"Workflow execution not found.  WorkflowId: %v, RunId: %v\",\n\t\t\t\t\trequest.Execution.GetWorkflowID(),\n\t\t\t\t\trequest.Execution.GetRunID(),\n\t\t\t\t),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(m.db, \"GetWorkflowExecution\", \"\", err)\n\t}\n\n\tif len(executions) == 0 {\n\t\treturn nil, &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\"Workflow execution not found.  WorkflowId: %v, RunId: %v\",\n\t\t\t\trequest.Execution.GetWorkflowID(),\n\t\t\t\trequest.Execution.GetRunID(),\n\t\t\t),\n\t\t}\n\t}\n\n\tif len(executions) != 1 {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"GetWorkflowExecution return more than one results.\",\n\t\t}\n\t}\n\treturn executions, nil\n}\n\nfunc (m *sqlExecutionStore) GetWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalGetWorkflowExecutionRequest,\n) (resp *p.InternalGetWorkflowExecutionResponse, e error) {\n\trecoverPanic := func(recovered interface{}, err *error) {\n\t\tif recovered != nil {\n\t\t\t*err = fmt.Errorf(\"DB operation panicked: %v %s\", recovered, debug.Stack())\n\t\t}\n\t}\n\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\trunID := serialization.MustParseUUID(request.Execution.RunID)\n\twfID := request.Execution.WorkflowID\n\n\tvar executions []sqlplugin.ExecutionsRow\n\tvar activityInfos map[int64]*p.InternalActivityInfo\n\tvar timerInfos map[string]*p.TimerInfo\n\tvar childExecutionInfos map[int64]*p.InternalChildExecutionInfo\n\tvar requestCancelInfos map[int64]*p.RequestCancelInfo\n\tvar signalInfos map[int64]*p.SignalInfo\n\tvar bufferedEvents []*p.DataBlob\n\tvar signalsRequested map[string]struct{}\n\n\tg, childCtx := errgroup.WithContext(ctx)\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\tactivityInfos, e = getActivityInfoMap(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID, m.parser)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\ttimerInfos, e = getTimerInfoMap(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID, m.parser)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\tchildExecutionInfos, e = getChildExecutionInfoMap(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID, m.parser)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\trequestCancelInfos, e = getRequestCancelInfoMap(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID, m.parser)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\tsignalInfos, e = getSignalInfoMap(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID, m.parser)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\tbufferedEvents, e = getBufferedEvents(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID)\n\t\treturn e\n\t})\n\n\tg.Go(func() (e error) {\n\t\tdefer func() { recoverPanic(recover(), &e) }()\n\t\tsignalsRequested, e = getSignalsRequested(\n\t\t\tchildCtx, m.db, m.shardID, domainID, wfID, runID)\n\t\treturn e\n\t})\n\n\terr := g.Wait()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// there is a race condition with delete workflow. What could happen is that\n\t// a delete workflow transaction can be committed between 2 concurrent read operations\n\t// and in that case we can get checksum error because data is partially read.\n\t// Since checksum is stored in the executions table, we make it the last step of reading,\n\t// in this case, either we read full data with checksum or we don't get checksum and return error.\n\texecutions, err = m.getExecutions(ctx, request, domainID, wfID, runID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tstate, err := m.populateWorkflowMutableState(executions[0])\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"GetWorkflowExecution: failed. Error: %v\", err),\n\t\t}\n\t}\n\t// if we have checksum, we need to make sure the rangeID did not change\n\t// if the rangeID changed, it means the shard ownership might have changed\n\t// and the workflow might have been updated when we read the data, so the data\n\t// we read might not be from a consistent view, the checksum validation might fail\n\t// in that case, we clear the checksum data so that we will not perform the validation\n\tif state.ChecksumData != nil {\n\t\trow, err := m.db.SelectFromShards(ctx, &sqlplugin.ShardsFilter{ShardID: int64(m.shardID)})\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"GetWorkflowExecution\", \"\", err)\n\t\t}\n\t\tif row.RangeID != request.RangeID {\n\t\t\t// The GetWorkflowExecution operation will not be impacted by this. ChecksumData is purely for validation purposes.\n\t\t\tm.logger.Warn(\"GetWorkflowExecution's checksum is discarded. The shard might have changed owner.\")\n\t\t\tstate.ChecksumData = nil\n\t\t}\n\t}\n\n\tstate.ActivityInfos = activityInfos\n\tstate.TimerInfos = timerInfos\n\tstate.ChildExecutionInfos = childExecutionInfos\n\tstate.RequestCancelInfos = requestCancelInfos\n\tstate.SignalInfos = signalInfos\n\tstate.BufferedEvents = bufferedEvents\n\tstate.SignalRequestedIDs = signalsRequested\n\n\treturn &p.InternalGetWorkflowExecutionResponse{State: state}, nil\n}\n\nfunc (m *sqlExecutionStore) UpdateWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalUpdateWorkflowExecutionRequest,\n) error {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(m.shardID, m.db.GetTotalNumDBShards())\n\treturn m.txExecuteShardLockedFn(ctx, dbShardID, \"UpdateWorkflowExecution\", request.RangeID, func(tx sqlplugin.Tx) error {\n\t\treturn m.updateWorkflowExecutionTx(ctx, tx, request)\n\t})\n}\n\nfunc (m *sqlExecutionStore) updateWorkflowExecutionTx(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\trequest *p.InternalUpdateWorkflowExecutionRequest,\n) error {\n\n\tupdateWorkflow := request.UpdateWorkflowMutation\n\tnewWorkflow := request.NewWorkflowSnapshot\n\n\texecutionInfo := updateWorkflow.ExecutionInfo\n\tdomainID := serialization.MustParseUUID(executionInfo.DomainID)\n\tworkflowID := executionInfo.WorkflowID\n\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\tshardID := m.shardID\n\n\tif err := p.ValidateUpdateWorkflowModeState(\n\t\trequest.Mode,\n\t\tupdateWorkflow,\n\t\tnewWorkflow,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tswitch request.Mode {\n\tcase p.UpdateWorkflowModeIgnoreCurrent:\n\t\t// no-op\n\tcase p.UpdateWorkflowModeBypassCurrent:\n\t\tif err := m.assertNotCurrentExecutionFn(\n\t\t\tctx,\n\t\t\ttx,\n\t\t\tshardID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID); err != nil {\n\t\t\treturn err\n\t\t}\n\n\tcase p.UpdateWorkflowModeUpdateCurrent:\n\t\tif newWorkflow != nil {\n\t\t\tnewExecutionInfo := newWorkflow.ExecutionInfo\n\t\t\tstartVersion := newWorkflow.StartVersion\n\t\t\tlastWriteVersion := newWorkflow.LastWriteVersion\n\t\t\tnewDomainID := serialization.MustParseUUID(newExecutionInfo.DomainID)\n\t\t\tnewRunID := serialization.MustParseUUID(newExecutionInfo.RunID)\n\n\t\t\tif !bytes.Equal(domainID, newDomainID) {\n\t\t\t\treturn &types.InternalServiceError{\n\t\t\t\t\tMessage: \"UpdateWorkflowExecution: cannot continue as new to another domain\",\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif err := m.assertRunIDAndUpdateCurrentExecutionFn(\n\t\t\t\tctx,\n\t\t\t\ttx,\n\t\t\t\tshardID,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\tnewRunID,\n\t\t\t\trunID,\n\t\t\t\tnewWorkflow.ExecutionInfo.CreateRequestID,\n\t\t\t\tnewWorkflow.ExecutionInfo.State,\n\t\t\t\tnewWorkflow.ExecutionInfo.CloseStatus,\n\t\t\t\tstartVersion,\n\t\t\t\tlastWriteVersion); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\tstartVersion := updateWorkflow.StartVersion\n\t\t\tlastWriteVersion := updateWorkflow.LastWriteVersion\n\t\t\t// this is only to update the current record\n\t\t\tif err := m.assertRunIDAndUpdateCurrentExecutionFn(\n\t\t\t\tctx,\n\t\t\t\ttx,\n\t\t\t\tshardID,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\trunID,\n\t\t\t\trunID,\n\t\t\t\texecutionInfo.CreateRequestID,\n\t\t\t\texecutionInfo.State,\n\t\t\t\texecutionInfo.CloseStatus,\n\t\t\t\tstartVersion,\n\t\t\t\tlastWriteVersion); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"UpdateWorkflowExecution: unknown mode: %v\", request.Mode),\n\t\t}\n\t}\n\n\tif err := m.applyWorkflowMutationTxFn(ctx, tx, shardID, &updateWorkflow, m.parser, m.taskSerializer); err != nil {\n\t\treturn err\n\t}\n\tif newWorkflow != nil {\n\t\tif err := m.applyWorkflowSnapshotTxAsNewFn(ctx, tx, shardID, newWorkflow, m.parser, m.taskSerializer); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m *sqlExecutionStore) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalConflictResolveWorkflowExecutionRequest,\n) error {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(m.shardID, m.db.GetTotalNumDBShards())\n\treturn m.txExecuteShardLockedFn(ctx, dbShardID, \"ConflictResolveWorkflowExecution\", request.RangeID, func(tx sqlplugin.Tx) error {\n\t\treturn m.conflictResolveWorkflowExecutionTx(ctx, tx, request)\n\t})\n}\n\nfunc (m *sqlExecutionStore) conflictResolveWorkflowExecutionTx(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\trequest *p.InternalConflictResolveWorkflowExecutionRequest,\n) error {\n\n\tcurrentWorkflow := request.CurrentWorkflowMutation\n\tresetWorkflow := request.ResetWorkflowSnapshot\n\tnewWorkflow := request.NewWorkflowSnapshot\n\n\tshardID := m.shardID\n\n\tdomainID := serialization.MustParseUUID(resetWorkflow.ExecutionInfo.DomainID)\n\tworkflowID := resetWorkflow.ExecutionInfo.WorkflowID\n\n\tif err := p.ValidateConflictResolveWorkflowModeState(\n\t\trequest.Mode,\n\t\tresetWorkflow,\n\t\tnewWorkflow,\n\t\tcurrentWorkflow,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tswitch request.Mode {\n\tcase p.ConflictResolveWorkflowModeBypassCurrent:\n\t\tif err := m.assertNotCurrentExecutionFn(\n\t\t\tctx,\n\t\t\ttx,\n\t\t\tshardID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tserialization.MustParseUUID(resetWorkflow.ExecutionInfo.RunID)); err != nil {\n\t\t\treturn err\n\t\t}\n\n\tcase p.ConflictResolveWorkflowModeUpdateCurrent:\n\t\texecutionInfo := resetWorkflow.ExecutionInfo\n\t\tstartVersion := resetWorkflow.StartVersion\n\t\tlastWriteVersion := resetWorkflow.LastWriteVersion\n\t\tif newWorkflow != nil {\n\t\t\texecutionInfo = newWorkflow.ExecutionInfo\n\t\t\tstartVersion = newWorkflow.StartVersion\n\t\t\tlastWriteVersion = newWorkflow.LastWriteVersion\n\t\t}\n\t\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\t\tcreateRequestID := executionInfo.CreateRequestID\n\t\tstate := executionInfo.State\n\t\tcloseStatus := executionInfo.CloseStatus\n\n\t\tif currentWorkflow != nil {\n\t\t\tprevRunID := serialization.MustParseUUID(currentWorkflow.ExecutionInfo.RunID)\n\n\t\t\tif err := m.assertRunIDAndUpdateCurrentExecutionFn(\n\t\t\t\tctx,\n\t\t\t\ttx,\n\t\t\t\tm.shardID,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\trunID,\n\t\t\t\tprevRunID,\n\t\t\t\tcreateRequestID,\n\t\t\t\tstate,\n\t\t\t\tcloseStatus,\n\t\t\t\tstartVersion,\n\t\t\t\tlastWriteVersion); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\t// reset workflow is current\n\t\t\tprevRunID := serialization.MustParseUUID(resetWorkflow.ExecutionInfo.RunID)\n\n\t\t\tif err := m.assertRunIDAndUpdateCurrentExecutionFn(\n\t\t\t\tctx,\n\t\t\t\ttx,\n\t\t\t\tm.shardID,\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\trunID,\n\t\t\t\tprevRunID,\n\t\t\t\tcreateRequestID,\n\t\t\t\tstate,\n\t\t\t\tcloseStatus,\n\t\t\t\tstartVersion,\n\t\t\t\tlastWriteVersion); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ConflictResolveWorkflowExecution: unknown mode: %v\", request.Mode),\n\t\t}\n\t}\n\n\tif err := m.applyWorkflowSnapshotTxAsResetFn(ctx, tx, shardID, &resetWorkflow, m.parser, m.taskSerializer); err != nil {\n\t\treturn err\n\t}\n\tif currentWorkflow != nil {\n\t\tif err := m.applyWorkflowMutationTxFn(ctx, tx, shardID, currentWorkflow, m.parser, m.taskSerializer); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tif newWorkflow != nil {\n\t\tif err := m.applyWorkflowSnapshotTxAsNewFn(ctx, tx, shardID, newWorkflow, m.parser, m.taskSerializer); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (m *sqlExecutionStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.DeleteWorkflowExecutionRequest,\n) error {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(m.shardID, m.db.GetTotalNumDBShards())\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\trunID := serialization.MustParseUUID(request.RunID)\n\twfID := request.WorkflowID\n\treturn m.txExecute(ctx, dbShardID, \"DeleteWorkflowExecution\", func(tx sqlplugin.Tx) error {\n\t\tif _, err := tx.DeleteFromExecutions(ctx, &sqlplugin.ExecutionsFilter{\n\t\t\tShardID:    m.shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteWorkflowExecution\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromActivityInfoMaps(ctx, &sqlplugin.ActivityInfoMapsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromActivityInfoMaps\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromTimerInfoMaps(ctx, &sqlplugin.TimerInfoMapsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromTimerInfoMaps\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromChildExecutionInfoMaps(ctx, &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromChildExecutionInfoMaps\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromRequestCancelInfoMaps(ctx, &sqlplugin.RequestCancelInfoMapsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromRequestCancelInfoMaps\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromSignalInfoMaps(ctx, &sqlplugin.SignalInfoMapsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromSignalInfoMaps\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromBufferedEvents(ctx, &sqlplugin.BufferedEventsFilter{\n\t\t\tShardID:    m.shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromBufferedEvents\", \"\", err)\n\t\t}\n\t\tif _, err := tx.DeleteFromSignalsRequestedSets(ctx, &sqlplugin.SignalsRequestedSetsFilter{\n\t\t\tShardID:    int64(m.shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: wfID,\n\t\t\tRunID:      runID,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"DeleteFromSignalsRequestedSets\", \"\", err)\n\t\t}\n\t\treturn nil\n\t})\n}\n\n// its possible for a new run of the same workflow to have started after the run we are deleting\n// here was finished. In that case, current_executions table will have the same workflowID but different\n// runID. The following code will delete the row from current_executions if and only if the runID is\n// same as the one we are trying to delete here\nfunc (m *sqlExecutionStore) DeleteCurrentWorkflowExecution(\n\tctx context.Context,\n\trequest *p.DeleteCurrentWorkflowExecutionRequest,\n) error {\n\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\trunID := serialization.MustParseUUID(request.RunID)\n\t_, err := m.db.DeleteFromCurrentExecutions(ctx, &sqlplugin.CurrentExecutionsFilter{\n\t\tShardID:    int64(m.shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: request.WorkflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(m.db, \"DeleteCurrentWorkflowExecution\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (m *sqlExecutionStore) GetCurrentExecution(\n\tctx context.Context,\n\trequest *p.GetCurrentExecutionRequest,\n) (*p.GetCurrentExecutionResponse, error) {\n\n\trow, err := m.db.SelectFromCurrentExecutions(ctx, &sqlplugin.CurrentExecutionsFilter{\n\t\tShardID:    int64(m.shardID),\n\t\tDomainID:   serialization.MustParseUUID(request.DomainID),\n\t\tWorkflowID: request.WorkflowID,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetCurrentExecution\", \"\", err)\n\t}\n\treturn &p.GetCurrentExecutionResponse{\n\t\tStartRequestID:   row.CreateRequestID,\n\t\tRunID:            row.RunID.String(),\n\t\tState:            int(row.State),\n\t\tCloseStatus:      int(row.CloseStatus),\n\t\tLastWriteVersion: row.LastWriteVersion,\n\t}, nil\n}\n\nfunc (m *sqlExecutionStore) ListCurrentExecutions(\n\t_ context.Context,\n\t_ *p.ListCurrentExecutionsRequest,\n) (*p.ListCurrentExecutionsResponse, error) {\n\treturn nil, &types.InternalServiceError{Message: \"Not yet implemented\"}\n}\n\nfunc (m *sqlExecutionStore) IsWorkflowExecutionExists(\n\t_ context.Context,\n\t_ *p.IsWorkflowExecutionExistsRequest,\n) (*p.IsWorkflowExecutionExistsResponse, error) {\n\treturn nil, &types.InternalServiceError{Message: \"Not yet implemented\"}\n}\n\nfunc (m *sqlExecutionStore) ListConcreteExecutions(\n\tctx context.Context,\n\trequest *p.ListConcreteExecutionsRequest,\n) (*p.InternalListConcreteExecutionsResponse, error) {\n\n\tfilter := &sqlplugin.ExecutionsFilter{}\n\tif len(request.PageToken) > 0 {\n\t\terr := gobDeserialize(request.PageToken, &filter)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"ListConcreteExecutions failed. Error: %v\", err),\n\t\t\t}\n\t\t}\n\t} else {\n\t\tfilter = &sqlplugin.ExecutionsFilter{\n\t\t\tShardID:    m.shardID,\n\t\t\tWorkflowID: \"\",\n\t\t}\n\t}\n\tfilter.Size = request.PageSize\n\n\texecutions, err := m.db.SelectFromExecutions(ctx, filter)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn &p.InternalListConcreteExecutionsResponse{}, nil\n\t\t}\n\t\treturn nil, convertCommonErrors(m.db, \"ListConcreteExecutions\", \"\", err)\n\t}\n\n\tif len(executions) == 0 {\n\t\treturn &p.InternalListConcreteExecutionsResponse{}, nil\n\t}\n\tlastExecution := executions[len(executions)-1]\n\tnextFilter := &sqlplugin.ExecutionsFilter{\n\t\tShardID:    m.shardID,\n\t\tWorkflowID: lastExecution.WorkflowID,\n\t}\n\ttoken, err := gobSerialize(nextFilter)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListConcreteExecutions failed. Error: %v\", err),\n\t\t}\n\t}\n\tconcreteExecutions, err := m.populateInternalListConcreteExecutions(executions)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"ListConcreteExecutions failed. Error: %v\", err),\n\t\t}\n\t}\n\n\treturn &p.InternalListConcreteExecutionsResponse{\n\t\tExecutions:    concreteExecutions,\n\t\tNextPageToken: token,\n\t}, nil\n}\n\nfunc getReadLevels(request *p.GetReplicationTasksFromDLQRequest) (readLevel int64, maxReadLevel int64, err error) {\n\treadLevel = request.ReadLevel\n\tif len(request.NextPageToken) > 0 {\n\t\treadLevel, err = deserializePageToken(request.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn 0, 0, err\n\t\t}\n\t}\n\n\tmaxReadLevel = max(readLevel+int64(request.BatchSize), request.MaxReadLevel)\n\treturn readLevel, maxReadLevel, nil\n}\n\nfunc (m *sqlExecutionStore) GetReplicationTasksFromDLQ(\n\tctx context.Context,\n\trequest *p.GetReplicationTasksFromDLQRequest,\n) (*p.GetHistoryTasksResponse, error) {\n\n\treadLevel, maxReadLevel, err := getReadLevels(request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tfilter := sqlplugin.ReplicationTasksFilter{\n\t\tShardID:            m.shardID,\n\t\tInclusiveMinTaskID: readLevel,\n\t\tExclusiveMaxTaskID: maxReadLevel,\n\t\tPageSize:           request.BatchSize,\n\t}\n\trows, err := m.db.SelectFromReplicationTasksDLQ(ctx, &sqlplugin.ReplicationTasksDLQFilter{\n\t\tReplicationTasksFilter: filter,\n\t\tSourceClusterName:      request.SourceClusterName,\n\t})\n\tif err != nil {\n\t\tif err != sql.ErrNoRows {\n\t\t\treturn nil, convertCommonErrors(m.db, \"GetReplicationTasksFromDLQ\", \"\", err)\n\t\t}\n\t}\n\tvar tasks []p.Task\n\tfor _, row := range rows {\n\t\ttask, err := m.taskSerializer.DeserializeTask(p.HistoryTaskCategoryReplication, p.NewDataBlob(row.Data, constants.EncodingType(row.DataEncoding)))\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"GetReplicationTasksFromDLQ\", \"\", err)\n\t\t}\n\t\ttask.SetTaskID(row.TaskID)\n\t\ttasks = append(tasks, task)\n\t}\n\tresp := &p.GetHistoryTasksResponse{Tasks: tasks}\n\tif len(rows) > 0 {\n\t\tnextTaskID := rows[len(rows)-1].TaskID + 1\n\t\tif nextTaskID < maxReadLevel {\n\t\t\tresp.NextPageToken = serializePageToken(nextTaskID)\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (m *sqlExecutionStore) GetReplicationDLQSize(\n\tctx context.Context,\n\trequest *p.GetReplicationDLQSizeRequest,\n) (*p.GetReplicationDLQSizeResponse, error) {\n\n\tsize, err := m.db.SelectFromReplicationDLQ(ctx, &sqlplugin.ReplicationTaskDLQFilter{\n\t\tSourceClusterName: request.SourceClusterName,\n\t\tShardID:           m.shardID,\n\t})\n\n\tswitch err {\n\tcase nil:\n\t\treturn &p.GetReplicationDLQSizeResponse{\n\t\t\tSize: size,\n\t\t}, nil\n\tcase sql.ErrNoRows:\n\t\treturn &p.GetReplicationDLQSizeResponse{\n\t\t\tSize: 0,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, convertCommonErrors(m.db, \"GetReplicationDLQSize\", \"\", err)\n\t}\n}\n\nfunc (m *sqlExecutionStore) DeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *p.DeleteReplicationTaskFromDLQRequest,\n) error {\n\n\tfilter := sqlplugin.ReplicationTasksFilter{\n\t\tShardID: m.shardID,\n\t\tTaskID:  request.TaskID,\n\t}\n\n\tif _, err := m.db.DeleteMessageFromReplicationTasksDLQ(ctx, &sqlplugin.ReplicationTasksDLQFilter{\n\t\tReplicationTasksFilter: filter,\n\t\tSourceClusterName:      request.SourceClusterName,\n\t}); err != nil {\n\t\treturn convertCommonErrors(m.db, \"DeleteReplicationTaskFromDLQ\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (m *sqlExecutionStore) RangeDeleteReplicationTaskFromDLQ(\n\tctx context.Context,\n\trequest *p.RangeDeleteReplicationTaskFromDLQRequest,\n) (*p.RangeDeleteReplicationTaskFromDLQResponse, error) {\n\tfilter := sqlplugin.ReplicationTasksFilter{\n\t\tShardID:            m.shardID,\n\t\tInclusiveMinTaskID: request.InclusiveBeginTaskID,\n\t\tExclusiveMaxTaskID: request.ExclusiveEndTaskID,\n\t\tPageSize:           request.PageSize,\n\t}\n\tresult, err := m.db.RangeDeleteMessageFromReplicationTasksDLQ(ctx, &sqlplugin.ReplicationTasksDLQFilter{\n\t\tReplicationTasksFilter: filter,\n\t\tSourceClusterName:      request.SourceClusterName,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"RangeDeleteReplicationTaskFromDLQ\", \"\", err)\n\t}\n\trowsDeleted, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"RangeDeleteReplicationTaskFromDLQ\", \"\", err)\n\t}\n\treturn &p.RangeDeleteReplicationTaskFromDLQResponse{TasksCompleted: int(rowsDeleted)}, nil\n}\n\nfunc (m *sqlExecutionStore) CreateFailoverMarkerTasks(\n\tctx context.Context,\n\trequest *p.CreateFailoverMarkersRequest,\n) error {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(m.shardID, m.db.GetTotalNumDBShards())\n\treturn m.txExecuteShardLockedFn(ctx, dbShardID, \"CreateFailoverMarkerTasks\", request.RangeID, func(tx sqlplugin.Tx) error {\n\t\treplicationTasksRows := make([]sqlplugin.ReplicationTasksRow, len(request.Markers))\n\t\tfor i, task := range request.Markers {\n\t\t\tblob, err := m.parser.ReplicationTaskInfoToBlob(&serialization.ReplicationTaskInfo{\n\t\t\t\tDomainID:                serialization.MustParseUUID(task.DomainID),\n\t\t\t\tWorkflowID:              emptyWorkflowID,\n\t\t\t\tRunID:                   serialization.MustParseUUID(emptyReplicationRunID),\n\t\t\t\tTaskType:                int16(task.GetTaskType()),\n\t\t\t\tFirstEventID:            constants.EmptyEventID,\n\t\t\t\tNextEventID:             constants.EmptyEventID,\n\t\t\t\tVersion:                 task.GetVersion(),\n\t\t\t\tScheduledID:             constants.EmptyEventID,\n\t\t\t\tEventStoreVersion:       p.EventStoreVersion,\n\t\t\t\tNewRunEventStoreVersion: p.EventStoreVersion,\n\t\t\t\tBranchToken:             nil,\n\t\t\t\tNewRunBranchToken:       nil,\n\t\t\t\tCreationTimestamp:       task.GetVisibilityTimestamp(),\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\treplicationTasksRows[i].ShardID = m.shardID\n\t\t\treplicationTasksRows[i].TaskID = task.GetTaskID()\n\t\t\treplicationTasksRows[i].Data = blob.Data\n\t\t\treplicationTasksRows[i].DataEncoding = string(blob.Encoding)\n\t\t}\n\t\tresult, err := tx.InsertIntoReplicationTasks(ctx, replicationTasksRows)\n\t\tif err != nil {\n\t\t\treturn convertCommonErrors(tx, \"CreateFailoverMarkerTasks\", \"\", err)\n\t\t}\n\t\trowsAffected, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"CreateFailoverMarkerTasks failed. Could not verify number of rows inserted. Error: %v\", err)}\n\t\t}\n\t\tif int(rowsAffected) != len(replicationTasksRows) {\n\t\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"CreateFailoverMarkerTasks failed. Inserted %v instead of %v rows into replication_tasks.\", rowsAffected, len(replicationTasksRows))}\n\t\t}\n\t\treturn nil\n\t})\n}\n\ntype timerTaskPageToken struct {\n\tTaskID    int64     `json:\"TaskID\"`    // CAUTION: JSON format is used in replication, this should not be changed without great care\n\tTimestamp time.Time `json:\"Timestamp\"` // CAUTION: JSON format is used in replication, this should not be changed without great care\n}\n\nfunc (t *timerTaskPageToken) serialize() ([]byte, error) {\n\treturn json.Marshal(t)\n}\n\nfunc (t *timerTaskPageToken) deserialize(payload []byte) error {\n\treturn json.Unmarshal(payload, t)\n}\n\nfunc (m *sqlExecutionStore) PutReplicationTaskToDLQ(\n\tctx context.Context,\n\trequest *p.InternalPutReplicationTaskToDLQRequest,\n) error {\n\treplicationTask := request.TaskInfo\n\tblob, err := m.parser.ReplicationTaskInfoToBlob(&serialization.ReplicationTaskInfo{\n\t\tDomainID:                serialization.MustParseUUID(replicationTask.DomainID),\n\t\tWorkflowID:              replicationTask.WorkflowID,\n\t\tRunID:                   serialization.MustParseUUID(replicationTask.RunID),\n\t\tTaskType:                int16(replicationTask.TaskType),\n\t\tFirstEventID:            replicationTask.FirstEventID,\n\t\tNextEventID:             replicationTask.NextEventID,\n\t\tVersion:                 replicationTask.Version,\n\t\tScheduledID:             replicationTask.ScheduledID,\n\t\tEventStoreVersion:       p.EventStoreVersion,\n\t\tNewRunEventStoreVersion: p.EventStoreVersion,\n\t\tBranchToken:             replicationTask.BranchToken,\n\t\tNewRunBranchToken:       replicationTask.NewRunBranchToken,\n\t\tCreationTimestamp:       replicationTask.CreationTime,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\trow := &sqlplugin.ReplicationTaskDLQRow{\n\t\tSourceClusterName: request.SourceClusterName,\n\t\tShardID:           m.shardID,\n\t\tTaskID:            replicationTask.TaskID,\n\t\tData:              blob.Data,\n\t\tDataEncoding:      string(blob.Encoding),\n\t}\n\n\t_, err = m.db.InsertIntoReplicationTasksDLQ(ctx, row)\n\n\t// Tasks are immutable. So it's fine if we already persisted it before.\n\t// This can happen when tasks are retried (ack and cleanup can have lag on source side).\n\tif err != nil && !m.db.IsDupEntryError(err) {\n\t\treturn convertCommonErrors(m.db, \"PutReplicationTaskToDLQ\", \"\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (m *sqlExecutionStore) populateWorkflowMutableState(\n\texecution sqlplugin.ExecutionsRow,\n) (*p.InternalWorkflowMutableState, error) {\n\n\tinfo, err := m.parser.WorkflowExecutionInfoFromBlob(execution.Data, execution.DataEncoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tstate := &p.InternalWorkflowMutableState{}\n\tstate.ExecutionInfo = serialization.ToInternalWorkflowExecutionInfo(info)\n\tstate.ExecutionInfo.DomainID = execution.DomainID.String()\n\tstate.ExecutionInfo.WorkflowID = execution.WorkflowID\n\tstate.ExecutionInfo.RunID = execution.RunID.String()\n\tstate.ExecutionInfo.NextEventID = execution.NextEventID\n\t// TODO: remove this after all 2DC workflows complete\n\tif info.LastWriteEventID != nil {\n\t\tstate.ReplicationState = &p.ReplicationState{}\n\t\tstate.ReplicationState.StartVersion = info.GetStartVersion()\n\t\tstate.ReplicationState.LastWriteVersion = execution.LastWriteVersion\n\t\tstate.ReplicationState.LastWriteEventID = info.GetLastWriteEventID()\n\t}\n\n\tif info.GetVersionHistories() != nil {\n\t\tstate.VersionHistories = p.NewDataBlob(\n\t\t\tinfo.GetVersionHistories(),\n\t\t\tconstants.EncodingType(info.GetVersionHistoriesEncoding()),\n\t\t)\n\t}\n\n\tif info.GetChecksum() != nil {\n\t\tstate.ChecksumData = p.NewDataBlob(\n\t\t\tinfo.GetChecksum(),\n\t\t\tconstants.EncodingType(info.GetChecksumEncoding()),\n\t\t)\n\t}\n\n\treturn state, nil\n}\n\nfunc (m *sqlExecutionStore) populateInternalListConcreteExecutions(\n\texecutions []sqlplugin.ExecutionsRow,\n) ([]*p.InternalListConcreteExecutionsEntity, error) {\n\n\tconcreteExecutions := make([]*p.InternalListConcreteExecutionsEntity, 0, len(executions))\n\tfor _, execution := range executions {\n\t\tmutableState, err := m.populateWorkflowMutableState(execution)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tconcreteExecution := &p.InternalListConcreteExecutionsEntity{\n\t\t\tExecutionInfo:    mutableState.ExecutionInfo,\n\t\t\tVersionHistories: mutableState.VersionHistories,\n\t\t}\n\t\tconcreteExecutions = append(concreteExecutions, concreteExecution)\n\t}\n\treturn concreteExecutions, nil\n}\n\nfunc (m *sqlExecutionStore) GetHistoryTasks(\n\tctx context.Context,\n\trequest *p.GetHistoryTasksRequest,\n) (*p.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.Type() {\n\tcase p.HistoryTaskCategoryTypeImmediate:\n\t\treturn m.getImmediateHistoryTasks(ctx, request)\n\tcase p.HistoryTaskCategoryTypeScheduled:\n\t\treturn m.getScheduledHistoryTasks(ctx, request)\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) getImmediateHistoryTasks(\n\tctx context.Context,\n\trequest *p.GetHistoryTasksRequest,\n) (*p.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTransfer:\n\t\tinclusiveMinTaskID := request.InclusiveMinTaskKey.GetTaskID()\n\t\tif len(request.NextPageToken) > 0 {\n\t\t\tvar err error\n\t\t\tinclusiveMinTaskID, err = deserializePageToken(request.NextPageToken)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"GetImmediateHistoryTasks: error deserializing page token: %v\", err)}\n\t\t\t}\n\t\t}\n\t\trows, err := m.db.SelectFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\tShardID:            m.shardID,\n\t\t\tInclusiveMinTaskID: inclusiveMinTaskID,\n\t\t\tExclusiveMaxTaskID: request.ExclusiveMaxTaskKey.GetTaskID(),\n\t\t\tPageSize:           request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\tif err != sql.ErrNoRows {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetImmediateHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t}\n\t\tvar tasks []p.Task\n\t\tfor _, row := range rows {\n\t\t\ttask, err := m.taskSerializer.DeserializeTask(request.TaskCategory, p.NewDataBlob(row.Data, constants.EncodingType(row.DataEncoding)))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetImmediateHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t\ttask.SetTaskID(row.TaskID)\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\t\tresp := &p.GetHistoryTasksResponse{Tasks: tasks}\n\t\tif len(rows) > 0 {\n\t\t\tnextTaskID := rows[len(rows)-1].TaskID + 1\n\t\t\tif nextTaskID < request.ExclusiveMaxTaskKey.GetTaskID() {\n\t\t\t\tresp.NextPageToken = serializePageToken(nextTaskID)\n\t\t\t}\n\t\t}\n\t\treturn resp, nil\n\tcase p.HistoryTaskCategoryIDReplication:\n\t\tinclusiveMinTaskID := request.InclusiveMinTaskKey.GetTaskID()\n\t\texclusiveMaxTaskID := request.ExclusiveMaxTaskKey.GetTaskID()\n\t\tif len(request.NextPageToken) > 0 {\n\t\t\tvar err error\n\t\t\tinclusiveMinTaskID, err = deserializePageToken(request.NextPageToken)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"GetImmediateHistoryTasks: error deserializing page token: %v\", err)}\n\t\t\t}\n\t\t\t// TODO: this doesn't seem right, we should be using the exclusiveMaxTaskID from the request, but keeping the same logic for now and review it later\n\t\t\texclusiveMaxTaskID = max(inclusiveMinTaskID+int64(request.PageSize), exclusiveMaxTaskID)\n\t\t}\n\t\trows, err := m.db.SelectFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\tShardID:            m.shardID,\n\t\t\tInclusiveMinTaskID: inclusiveMinTaskID,\n\t\t\tExclusiveMaxTaskID: exclusiveMaxTaskID,\n\t\t\tPageSize:           request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\tif err != sql.ErrNoRows {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetImmediateHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t}\n\t\tvar tasks []p.Task\n\t\tfor _, row := range rows {\n\t\t\ttask, err := m.taskSerializer.DeserializeTask(request.TaskCategory, p.NewDataBlob(row.Data, constants.EncodingType(row.DataEncoding)))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetImmediateHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t\ttask.SetTaskID(row.TaskID)\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\t\tresp := &p.GetHistoryTasksResponse{Tasks: tasks}\n\t\tif len(rows) > 0 {\n\t\t\tnextTaskID := rows[len(rows)-1].TaskID + 1\n\t\t\tif nextTaskID < request.ExclusiveMaxTaskKey.GetTaskID() {\n\t\t\t\tresp.NextPageToken = serializePageToken(nextTaskID)\n\t\t\t}\n\t\t}\n\t\treturn resp, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) getScheduledHistoryTasks(\n\tctx context.Context,\n\trequest *p.GetHistoryTasksRequest,\n) (*p.GetHistoryTasksResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTimer:\n\t\tpageToken := &timerTaskPageToken{TaskID: math.MinInt64, Timestamp: request.InclusiveMinTaskKey.GetScheduledTime()}\n\t\tif len(request.NextPageToken) > 0 {\n\t\t\tif err := pageToken.deserialize(request.NextPageToken); err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"error deserializing timerTaskPageToken: %v\", err),\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\trows, err := m.db.SelectFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\tShardID:                m.shardID,\n\t\t\tMinVisibilityTimestamp: pageToken.Timestamp,\n\t\t\tTaskID:                 pageToken.TaskID,\n\t\t\tMaxVisibilityTimestamp: request.ExclusiveMaxTaskKey.GetScheduledTime(),\n\t\t\tPageSize:               request.PageSize + 1,\n\t\t})\n\t\tif err != nil {\n\t\t\tif err != sql.ErrNoRows {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetScheduledHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t}\n\t\tvar tasks []p.Task\n\t\tfor _, row := range rows {\n\t\t\ttask, err := m.taskSerializer.DeserializeTask(request.TaskCategory, p.NewDataBlob(row.Data, constants.EncodingType(row.DataEncoding)))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, convertCommonErrors(m.db, \"GetScheduledHistoryTasks\", \"\", err)\n\t\t\t}\n\t\t\ttask.SetTaskID(row.TaskID)\n\t\t\ttask.SetVisibilityTimestamp(row.VisibilityTimestamp)\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\t\tresp := &p.GetHistoryTasksResponse{Tasks: tasks}\n\t\tif len(tasks) > request.PageSize {\n\t\t\tpageToken = &timerTaskPageToken{\n\t\t\t\tTaskID:    tasks[request.PageSize].GetTaskID(),\n\t\t\t\tTimestamp: tasks[request.PageSize].GetVisibilityTimestamp(),\n\t\t\t}\n\t\t\tresp.Tasks = resp.Tasks[:request.PageSize]\n\t\t\tnextToken, err := pageToken.serialize()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"GetScheduledHistoryTasks: error serializing page token: %v\", err),\n\t\t\t\t}\n\t\t\t}\n\t\t\tresp.NextPageToken = nextToken\n\t\t}\n\t\treturn resp, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) CompleteHistoryTask(\n\tctx context.Context,\n\trequest *p.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.Type() {\n\tcase p.HistoryTaskCategoryTypeScheduled:\n\t\treturn m.completeScheduledHistoryTask(ctx, request)\n\tcase p.HistoryTaskCategoryTypeImmediate:\n\t\treturn m.completeImmediateHistoryTask(ctx, request)\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) completeScheduledHistoryTask(\n\tctx context.Context,\n\trequest *p.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTimer:\n\t\tif _, err := m.db.DeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\tShardID:             m.shardID,\n\t\t\tVisibilityTimestamp: request.TaskKey.GetScheduledTime(),\n\t\t\tTaskID:              request.TaskKey.GetTaskID(),\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(m.db, \"CompleteScheduledHistoryTask\", \"\", err)\n\t\t}\n\t\treturn nil\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) completeImmediateHistoryTask(\n\tctx context.Context,\n\trequest *p.CompleteHistoryTaskRequest,\n) error {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTransfer:\n\t\tif _, err := m.db.DeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\tShardID: m.shardID,\n\t\t\tTaskID:  request.TaskKey.GetTaskID(),\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(m.db, \"CompleteImmediateHistoryTask\", \"\", err)\n\t\t}\n\t\treturn nil\n\tcase p.HistoryTaskCategoryIDReplication:\n\t\tif _, err := m.db.DeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\tShardID: m.shardID,\n\t\t\tTaskID:  request.TaskKey.GetTaskID(),\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(m.db, \"CompleteImmediateHistoryTask\", \"\", err)\n\t\t}\n\t\treturn nil\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category ID: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) RangeCompleteHistoryTask(\n\tctx context.Context,\n\trequest *p.RangeCompleteHistoryTaskRequest,\n) (*p.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.Type() {\n\tcase p.HistoryTaskCategoryTypeScheduled:\n\t\treturn m.rangeCompleteScheduledHistoryTask(ctx, request)\n\tcase p.HistoryTaskCategoryTypeImmediate:\n\t\treturn m.rangeCompleteImmediateHistoryTask(ctx, request)\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category type: %v\", request.TaskCategory.Type())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) rangeCompleteScheduledHistoryTask(\n\tctx context.Context,\n\trequest *p.RangeCompleteHistoryTaskRequest,\n) (*p.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTimer:\n\t\tresult, err := m.db.RangeDeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\tShardID:                m.shardID,\n\t\t\tMinVisibilityTimestamp: request.InclusiveMinTaskKey.GetScheduledTime(),\n\t\t\tMaxVisibilityTimestamp: request.ExclusiveMaxTaskKey.GetScheduledTime(),\n\t\t\tPageSize:               request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteTimerTask\", \"\", err)\n\t\t}\n\t\trowsDeleted, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteTimerTask\", \"\", err)\n\t\t}\n\t\treturn &p.RangeCompleteHistoryTaskResponse{TasksCompleted: int(rowsDeleted)}, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) rangeCompleteImmediateHistoryTask(\n\tctx context.Context,\n\trequest *p.RangeCompleteHistoryTaskRequest,\n) (*p.RangeCompleteHistoryTaskResponse, error) {\n\tswitch request.TaskCategory.ID() {\n\tcase p.HistoryTaskCategoryIDTransfer:\n\t\tresult, err := m.db.RangeDeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\tShardID:            m.shardID,\n\t\t\tInclusiveMinTaskID: request.InclusiveMinTaskKey.GetTaskID(),\n\t\t\tExclusiveMaxTaskID: request.ExclusiveMaxTaskKey.GetTaskID(),\n\t\t\tPageSize:           request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteTransferTask\", \"\", err)\n\t\t}\n\t\trowsDeleted, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteTransferTask\", \"\", err)\n\t\t}\n\t\treturn &p.RangeCompleteHistoryTaskResponse{TasksCompleted: int(rowsDeleted)}, nil\n\tcase p.HistoryTaskCategoryIDReplication:\n\t\tresult, err := m.db.RangeDeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\tShardID:            m.shardID,\n\t\t\tExclusiveMaxTaskID: request.ExclusiveMaxTaskKey.GetTaskID(),\n\t\t\tPageSize:           request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteReplicationTask\", \"\", err)\n\t\t}\n\t\trowsDeleted, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"RangeCompleteReplicationTask\", \"\", err)\n\t\t}\n\t\treturn &p.RangeCompleteHistoryTaskResponse{TasksCompleted: int(rowsDeleted)}, nil\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown task category: %v\", request.TaskCategory.ID())}\n\t}\n}\n\nfunc (m *sqlExecutionStore) GetActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, wfID, rID string,\n) (*p.DataBlob, error) {\n\t// TODO(active-active): Active cluster selection policy for SQL stores is not yet implemented\n\t// It requires creating a new table in the database to store the active cluster selection policy\n\treturn nil, &types.InternalServiceError{Message: \"Not yet implemented\"}\n}\n\nfunc (m *sqlExecutionStore) DeleteActiveClusterSelectionPolicy(\n\tctx context.Context,\n\tdomainID, wfID, rID string,\n) error {\n\t// TODO(active-active): Active cluster selection policy for SQL stores is not yet implemented\n\t// It requires creating a new table in the database to store the active cluster selection policy\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_execution_store_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDeleteCurrentWorkflowExecution(t *testing.T) {\n\tshardID := int64(100)\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.DeleteCurrentWorkflowExecutionRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\tRunID:      \"fd65967f-777d-45de-8dee-be49dfda6716\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteFromCurrentExecutions(gomock.Any(), &sqlplugin.CurrentExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"fd65967f-777d-45de-8dee-be49dfda6716\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\tRunID:      \"fd65967f-777d-45de-8dee-be49dfda6716\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteFromCurrentExecutions(gomock.Any(), &sqlplugin.CurrentExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"fd65967f-777d-45de-8dee-be49dfda6716\"),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\terr = store.DeleteCurrentWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCurrentExecution(t *testing.T) {\n\tshardID := int64(100)\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetCurrentExecutionRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.GetCurrentExecutionResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetCurrentExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromCurrentExecutions(gomock.Any(), &sqlplugin.CurrentExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\t}).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:          shardID,\n\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID:       \"aaaa\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"fd65967f-777d-45de-8dee-be49dfda6716\"),\n\t\t\t\t\tCreateRequestID:  \"create\",\n\t\t\t\t\tState:            2,\n\t\t\t\t\tCloseStatus:      3,\n\t\t\t\t\tLastWriteVersion: 9,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tStartRequestID:   \"create\",\n\t\t\t\tRunID:            \"fd65967f-777d-45de-8dee-be49dfda6716\",\n\t\t\t\tState:            2,\n\t\t\t\tCloseStatus:      3,\n\t\t\t\tLastWriteVersion: 9,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.GetCurrentExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromCurrentExecutions(gomock.Any(), &sqlplugin.CurrentExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"aaaa\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.GetCurrentExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetReplicationTasksFromDLQ(t *testing.T) {\n\tshardID := 0\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetReplicationTasksFromDLQRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockTaskSerializer)\n\t\twant      *persistence.GetHistoryTasksResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetReplicationTasksFromDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tNextPageToken:     serializePageToken(100),\n\t\t\t\tMaxReadLevel:      199,\n\t\t\t\tBatchSize:         1000,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\t\tExclusiveMaxTaskID: 1100,\n\t\t\t\t\t\tPageSize:           1000,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return([]sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       100,\n\t\t\t\t\t\tData:         []byte(`replication`),\n\t\t\t\t\t\tDataEncoding: \"replication\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryReplication, persistence.NewDataBlob([]byte(`replication`), \"replication\")).Return(&persistence.HistoryReplicationTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\t\t\tWorkflowID: \"test\",\n\t\t\t\t\t\tRunID:      \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             202,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t},\n\t\t\t\t\tFirstEventID:      10,\n\t\t\t\t\tNextEventID:       101,\n\t\t\t\t\tBranchToken:       []byte(`bt`),\n\t\t\t\t\tNewRunBranchToken: []byte(`nbt`),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetHistoryTasksResponse{\n\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\t\t\t\tWorkflowID: \"test\",\n\t\t\t\t\t\t\tRunID:      \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              100,\n\t\t\t\t\t\t\tVersion:             202,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tFirstEventID:      10,\n\t\t\t\t\t\tNextEventID:       101,\n\t\t\t\t\t\tBranchToken:       []byte(`bt`),\n\t\t\t\t\t\tNewRunBranchToken: []byte(`nbt`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: serializePageToken(101),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to load from database\",\n\t\t\treq: &persistence.GetReplicationTasksFromDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tNextPageToken:     serializePageToken(100),\n\t\t\t\tMaxReadLevel:      199,\n\t\t\t\tBatchSize:         1000,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockTaskSerializer) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\t\tExclusiveMaxTaskID: 1100,\n\t\t\t\t\t\tPageSize:           1000,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to decode data\",\n\t\t\treq: &persistence.GetReplicationTasksFromDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tNextPageToken:     serializePageToken(100),\n\t\t\t\tMaxReadLevel:      199,\n\t\t\t\tBatchSize:         1000,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\t\tExclusiveMaxTaskID: 1100,\n\t\t\t\t\t\tPageSize:           1000,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return([]sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       101,\n\t\t\t\t\t\tData:         []byte(`replication`),\n\t\t\t\t\t\tDataEncoding: \"replication\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryReplication, persistence.NewDataBlob([]byte(`replication`), \"replication\")).Return(nil, errors.New(\"some error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockTaskSerializer(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, mockParser, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tgot, err := store.GetReplicationTasksFromDLQ(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetReplicationDLQSize(t *testing.T) {\n\tshardID := 9\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetReplicationDLQSizeRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.GetReplicationDLQSizeResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetReplicationDLQSizeRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationDLQ(gomock.Any(), &sqlplugin.ReplicationTaskDLQFilter{\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t\tShardID:           shardID,\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t},\n\t\t\twant: &persistence.GetReplicationDLQSizeResponse{\n\t\t\t\tSize: 1,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - no row\",\n\t\t\treq: &persistence.GetReplicationDLQSizeRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationDLQ(gomock.Any(), &sqlplugin.ReplicationTaskDLQFilter{\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t\tShardID:           shardID,\n\t\t\t\t}).Return(int64(0), sql.ErrNoRows)\n\t\t\t},\n\t\t\twant: &persistence.GetReplicationDLQSizeResponse{\n\t\t\t\tSize: 0,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.GetReplicationDLQSizeRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationDLQ(gomock.Any(), &sqlplugin.ReplicationTaskDLQFilter{\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t\tShardID:           shardID,\n\t\t\t\t}).Return(int64(0), err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.GetReplicationDLQSize(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteReplicationTaskFromDLQ(t *testing.T) {\n\tshardID := 100\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.DeleteReplicationTaskFromDLQRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.DeleteReplicationTaskFromDLQRequest{\n\t\t\t\tTaskID:            123,\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteMessageFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID: shardID,\n\t\t\t\t\t\tTaskID:  123,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.DeleteReplicationTaskFromDLQRequest{\n\t\t\t\tTaskID:            123,\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteMessageFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID: shardID,\n\t\t\t\t\t\tTaskID:  123,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\terr = store.DeleteReplicationTaskFromDLQ(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteReplicationTaskFromDLQ(t *testing.T) {\n\tshardID := 100\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.RangeDeleteReplicationTaskFromDLQRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.RangeDeleteReplicationTaskFromDLQResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\t\tInclusiveBeginTaskID: 123,\n\t\t\t\tExclusiveEndTaskID:   345,\n\t\t\t\tPageSize:             10,\n\t\t\t\tSourceClusterName:    \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteMessageFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\t\tInclusiveMinTaskID: 123,\n\t\t\t\t\t\tExclusiveMaxTaskID: 345,\n\t\t\t\t\t\tPageSize:           10,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 10}, nil)\n\t\t\t},\n\t\t\twant: &persistence.RangeDeleteReplicationTaskFromDLQResponse{\n\t\t\t\tTasksCompleted: 10,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\t\tInclusiveBeginTaskID: 123,\n\t\t\t\tExclusiveEndTaskID:   345,\n\t\t\t\tPageSize:             10,\n\t\t\t\tSourceClusterName:    \"source\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().RangeDeleteMessageFromReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTasksDLQFilter{\n\t\t\t\t\tReplicationTasksFilter: sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\t\tInclusiveMinTaskID: 123,\n\t\t\t\t\t\tExclusiveMaxTaskID: 345,\n\t\t\t\t\t\tPageSize:           10,\n\t\t\t\t\t},\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.RangeDeleteReplicationTaskFromDLQ(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPutReplicationTaskToDLQ(t *testing.T) {\n\tshardID := 100\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalPutReplicationTaskToDLQRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.InternalPutReplicationTaskToDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tTaskInfo: &persistence.InternalReplicationTaskInfo{\n\t\t\t\t\tDomainID:          \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\t\tWorkflowID:        \"test\",\n\t\t\t\t\tRunID:             \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\tTaskType:          1,\n\t\t\t\t\tTaskID:            101,\n\t\t\t\t\tVersion:           202,\n\t\t\t\t\tFirstEventID:      10,\n\t\t\t\t\tNextEventID:       101,\n\t\t\t\t\tScheduledID:       19,\n\t\t\t\t\tBranchToken:       []byte(`bt`),\n\t\t\t\t\tNewRunBranchToken: []byte(`nbt`),\n\t\t\t\t\tCreationTime:      time.Unix(1, 1),\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ReplicationTaskInfoToBlob(&serialization.ReplicationTaskInfo{\n\t\t\t\t\tDomainID:                serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID:              \"test\",\n\t\t\t\t\tRunID:                   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a54a\"),\n\t\t\t\t\tTaskType:                1,\n\t\t\t\t\tVersion:                 202,\n\t\t\t\t\tFirstEventID:            10,\n\t\t\t\t\tNextEventID:             101,\n\t\t\t\t\tScheduledID:             19,\n\t\t\t\t\tEventStoreVersion:       persistence.EventStoreVersion,\n\t\t\t\t\tNewRunEventStoreVersion: persistence.EventStoreVersion,\n\t\t\t\t\tBranchToken:             []byte(`bt`),\n\t\t\t\t\tNewRunBranchToken:       []byte(`nbt`),\n\t\t\t\t\tCreationTimestamp:       time.Unix(1, 1),\n\t\t\t\t}).Return(persistence.DataBlob{Data: []byte(`replication`), Encoding: \"replication\"}, nil)\n\t\t\t\tmockDB.EXPECT().InsertIntoReplicationTasksDLQ(gomock.Any(), &sqlplugin.ReplicationTaskDLQRow{\n\t\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\t\tShardID:           shardID,\n\t\t\t\t\tTaskID:            101,\n\t\t\t\t\tData:              []byte(`replication`),\n\t\t\t\t\tDataEncoding:      \"replication\",\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to encode data\",\n\t\t\treq: &persistence.InternalPutReplicationTaskToDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tTaskInfo: &persistence.InternalReplicationTaskInfo{\n\t\t\t\t\tDomainID:          \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\t\tWorkflowID:        \"test\",\n\t\t\t\t\tRunID:             \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\tTaskType:          1,\n\t\t\t\t\tTaskID:            101,\n\t\t\t\t\tVersion:           202,\n\t\t\t\t\tFirstEventID:      10,\n\t\t\t\t\tNextEventID:       101,\n\t\t\t\t\tScheduledID:       19,\n\t\t\t\t\tBranchToken:       []byte(`bt`),\n\t\t\t\t\tNewRunBranchToken: []byte(`nbt`),\n\t\t\t\t\tCreationTime:      time.Unix(1, 1),\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to insert into database\",\n\t\t\treq: &persistence.InternalPutReplicationTaskToDLQRequest{\n\t\t\t\tSourceClusterName: \"source\",\n\t\t\t\tTaskInfo: &persistence.InternalReplicationTaskInfo{\n\t\t\t\t\tDomainID:          \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\t\tWorkflowID:        \"test\",\n\t\t\t\t\tRunID:             \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\tTaskType:          1,\n\t\t\t\t\tTaskID:            101,\n\t\t\t\t\tVersion:           202,\n\t\t\t\t\tFirstEventID:      10,\n\t\t\t\t\tNextEventID:       101,\n\t\t\t\t\tScheduledID:       19,\n\t\t\t\t\tBranchToken:       []byte(`bt`),\n\t\t\t\t\tNewRunBranchToken: []byte(`nbt`),\n\t\t\t\t\tCreationTime:      time.Unix(1, 1),\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{Data: []byte(`replication`), Encoding: \"replication\"}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().InsertIntoReplicationTasksDLQ(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(err).Return(false)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), mockParser, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\terr = store.PutReplicationTaskToDLQ(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteWorkflowExecution(t *testing.T) {\n\tshardID := int64(100)\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.DeleteWorkflowExecutionRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.DeleteWorkflowExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\tRunID:      \"bbdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\t\t\t\tShardID:    int(shardID),\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromActivityInfoMaps(gomock.Any(), &sqlplugin.ActivityInfoMapsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromTimerInfoMaps(gomock.Any(), &sqlplugin.TimerInfoMapsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromChildExecutionInfoMaps(gomock.Any(), &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromRequestCancelInfoMaps(gomock.Any(), &sqlplugin.RequestCancelInfoMapsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromSignalInfoMaps(gomock.Any(), &sqlplugin.SignalInfoMapsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromBufferedEvents(gomock.Any(), &sqlplugin.BufferedEventsFilter{\n\t\t\t\t\tShardID:    int(shardID),\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromSignalsRequestedSets(gomock.Any(), &sqlplugin.SignalsRequestedSetsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to delete from executions\",\n\t\t\treq: &persistence.DeleteWorkflowExecutionRequest{\n\t\t\t\tDomainID:   \"abdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\tRunID:      \"bbdcea69-61d5-44c3-9d55-afe23505a542\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\t\t\t\tShardID:    int(shardID),\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"bbdcea69-61d5-44c3-9d55-afe23505a542\"),\n\t\t\t\t}).Return(nil, errors.New(\"some error\"))\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tstore, err := NewSQLExecutionStore(mockDB, nil, int(shardID), nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"failed to create execution store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx)\n\n\t\t\terr = store.DeleteWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTxExecuteShardLocked(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\toperation string\n\t\trangeID   int64\n\t\tfn        func(sqlplugin.Tx) error\n\t\twantError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), gomock.Any()).Return(11, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\trangeID:   11,\n\t\t\tfn:        func(sqlplugin.Tx) error { return nil },\n\t\t\twantError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), gomock.Any()).Return(11, nil)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsTimeoutError(gomock.Any()).Return(false)\n\t\t\t\tmockDB.EXPECT().IsThrottlingError(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\trangeID:   11,\n\t\t\tfn:        func(sqlplugin.Tx) error { return errors.New(\"error\") },\n\t\t\twantError: &types.InternalServiceError{Message: \"Insert operation failed.  Error: error\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - shard ownership lost\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), gomock.Any()).Return(12, nil)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\toperation: \"Insert\",\n\t\t\trangeID:   11,\n\t\t\tfn:        func(sqlplugin.Tx) error { return errors.New(\"error\") },\n\t\t\twantError: &persistence.ShardOwnershipLostError{ShardID: 0, Msg: \"Failed to lock shard. Previous range ID: 11; new range ID: 12\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\ttt.mockSetup(mockDB, mockTx)\n\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     mockDB,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tgotError := s.txExecuteShardLocked(context.Background(), 0, tt.operation, tt.rangeID, tt.fn)\n\t\t\tassert.Equal(t, tt.wantError, gotError)\n\t\t})\n\t}\n}\n\nfunc TestCreateWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                             string\n\t\treq                              *persistence.InternalCreateWorkflowExecutionRequest\n\t\tlockCurrentExecutionIfExistsFn   func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error)\n\t\tcreateOrUpdateCurrentExecutionFn func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error\n\t\tapplyWorkflowSnapshotTxAsNewFn   func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n\t\twantErr                          bool\n\t\twant                             *persistence.CreateWorkflowExecutionResponse\n\t\tassertErr                        func(t *testing.T, err error)\n\t}{\n\t\t{\n\t\t\tname: \"Success - mode brand new\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn nil, nil\n\t\t\t},\n\t\t\tcreateOrUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twant: &persistence.CreateWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode workflow ID reuse\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tcreateOrUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twant: &persistence.CreateWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode zombie\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeZombie,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateZombie,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tRunID: serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a54a\"),\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tcreateOrUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twant: &persistence.CreateWorkflowExecutionResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode state validation failed\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeZombie,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - lockCurrentExecutionIfExists failed\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn nil, errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode brand new\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tCreateRequestID:  \"test\",\n\t\t\t\t\tWorkflowID:       \"test\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a54a\"),\n\t\t\t\t\tState:            persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:      persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tLastWriteVersion: 10,\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\tMsg:              \"Workflow execution already running. WorkflowId: test\",\n\t\t\t\t\tStartRequestID:   \"test\",\n\t\t\t\t\tRunID:            \"abdcea69-61d5-44c3-9d55-afe23505a54a\",\n\t\t\t\t\tState:            persistence.WorkflowStateCreated,\n\t\t\t\t\tCloseStatus:      persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tLastWriteVersion: 10,\n\t\t\t\t}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode workflow ID reuse, version mismatch\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tState:            persistence.WorkflowStateCompleted,\n\t\t\t\t\tLastWriteVersion: 10,\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: \"Workflow execution creation condition failed. WorkflowId: , LastWriteVersion: 10, PreviousLastWriteVersion: 0\",\n\t\t\t\t}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode workflow ID reuse, state mismatch\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: \"Workflow execution creation condition failed. WorkflowId: , State: 0, Expected: 2\",\n\t\t\t\t}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode workflow ID reuse, run ID mismatch\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\tRunID: serialization.MustParseUUID(\"abdcea69-61d5-44c3-9d55-afe23505a54a\"),\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\t\tMsg: \"Workflow execution creation condition failed. WorkflowId: , RunID: abdcea69-61d5-44c3-9d55-afe23505a54a, PreviousRunID: \",\n\t\t\t\t}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode zombie, run ID match\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeZombie,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateZombie,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn &sqlplugin.CurrentExecutionsRow{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - unknown mode\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowMode(100),\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - createOrUpdateCurrentExecution failed\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn nil, nil\n\t\t\t},\n\t\t\tcreateOrUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowSnapshotTxAsNew failed\",\n\t\t\treq: &persistence.InternalCreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t\t\t\tNewWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlockCurrentExecutionIfExistsFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string) (*sqlplugin.CurrentExecutionsRow, error) {\n\t\t\t\treturn nil, nil\n\t\t\t},\n\t\t\tcreateOrUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, persistence.CreateWorkflowMode, int, serialization.UUID, string, serialization.UUID, int, int, string, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     mockDB,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t},\n\t\t\t\ttxExecuteShardLockedFn: func(_ context.Context, _ int, _ string, _ int64, fn func(sqlplugin.Tx) error) error {\n\t\t\t\t\treturn fn(nil)\n\t\t\t\t},\n\t\t\t\tlockCurrentExecutionIfExistsFn:   tc.lockCurrentExecutionIfExistsFn,\n\t\t\t\tcreateOrUpdateCurrentExecutionFn: tc.createOrUpdateCurrentExecutionFn,\n\t\t\t\tapplyWorkflowSnapshotTxAsNewFn:   tc.applyWorkflowSnapshotTxAsNewFn,\n\t\t\t}\n\n\t\t\tgot, err := s.CreateWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                   string\n\t\treq                                    *persistence.InternalUpdateWorkflowExecutionRequest\n\t\tassertNotCurrentExecutionFn            func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error\n\t\tassertRunIDAndUpdateCurrentExecutionFn func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error\n\t\tapplyWorkflowSnapshotTxAsNewFn         func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n\t\tapplyWorkflowMutationTxFn              func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error\n\t\twantErr                                bool\n\t\tassertErr                              func(t *testing.T, err error)\n\t}{\n\t\t{\n\t\t\tname: \"Success - mode ignore current\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeIgnoreCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{},\n\t\t\t\t},\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode bypass current\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeBypassCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertNotCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode update current, new workflow\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode update current, no new workflow\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode state validation failed\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateZombie,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - assertNotCurrentExecution failed\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeBypassCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertNotCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - domain ID mismatch\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID: \"a8ead65c-9d0d-43a2-a6ad-dd17c99509af\",\n\t\t\t\t\t\tState:    persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID: \"c3fab112-5175-4044-a096-a32e7badd4a8\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{\n\t\t\t\t\tMessage: \"UpdateWorkflowExecution: cannot continue as new to another domain\",\n\t\t\t\t}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - assertRunIDAndUpdateCurrentExecution failed\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowMutationTxFn failed\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowSnapshotTxAsNew failed\",\n\t\t\treq: &persistence.InternalUpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     mockDB,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t},\n\t\t\t\ttxExecuteShardLockedFn: func(_ context.Context, _ int, _ string, _ int64, fn func(sqlplugin.Tx) error) error {\n\t\t\t\t\treturn fn(nil)\n\t\t\t\t},\n\t\t\t\tassertNotCurrentExecutionFn:            tc.assertNotCurrentExecutionFn,\n\t\t\t\tassertRunIDAndUpdateCurrentExecutionFn: tc.assertRunIDAndUpdateCurrentExecutionFn,\n\t\t\t\tapplyWorkflowMutationTxFn:              tc.applyWorkflowMutationTxFn,\n\t\t\t\tapplyWorkflowSnapshotTxAsNewFn:         tc.applyWorkflowSnapshotTxAsNewFn,\n\t\t\t}\n\n\t\t\terr := s.UpdateWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConflictResolveWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                   string\n\t\treq                                    *persistence.InternalConflictResolveWorkflowExecutionRequest\n\t\tassertRunIDAndUpdateCurrentExecutionFn func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error\n\t\tassertNotCurrentExecutionFn            func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error\n\t\tapplyWorkflowMutationTxFn              func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error\n\t\tapplyWorkflowSnapshotTxAsResetFn       func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n\t\tapplyWorkflowSnapshotTxAsNewFn         func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error\n\t\twantErr                                bool\n\t\tassertErr                              func(t *testing.T, err error)\n\t}{\n\t\t{\n\t\t\tname: \"Success - mode bypass current\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertNotCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode update current, current workflow exists\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tCurrentWorkflowMutation: &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - mode update current, no current workflow\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mode state validation failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateZombie,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - assertNotCurrentExecution failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertNotCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - assertRunIDAndUpdateCurrentExecution failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tCurrentWorkflowMutation: &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowResetSnapshotTx failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertNotCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowMutationTxFn failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tCurrentWorkflowMutation: &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - applyWorkflowSnapshotTxAsNew failed\",\n\t\t\treq: &persistence.InternalConflictResolveWorkflowExecutionRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMode:    persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\t\tResetWorkflowSnapshot: persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.InternalWorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tCurrentWorkflowMutation: &persistence.InternalWorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertRunIDAndUpdateCurrentExecutionFn: func(context.Context, sqlplugin.Tx, int, serialization.UUID, string, serialization.UUID, serialization.UUID, string, int, int, int64, int64) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsResetFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowMutationTxFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowMutation, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tapplyWorkflowSnapshotTxAsNewFn: func(context.Context, sqlplugin.Tx, int, *persistence.InternalWorkflowSnapshot, serialization.Parser, serialization.TaskSerializer) error {\n\t\t\t\treturn errors.New(\"some random error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     mockDB,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t},\n\t\t\t\ttxExecuteShardLockedFn: func(_ context.Context, _ int, _ string, _ int64, fn func(sqlplugin.Tx) error) error {\n\t\t\t\t\treturn fn(nil)\n\t\t\t\t},\n\t\t\t\tassertNotCurrentExecutionFn:            tc.assertNotCurrentExecutionFn,\n\t\t\t\tassertRunIDAndUpdateCurrentExecutionFn: tc.assertRunIDAndUpdateCurrentExecutionFn,\n\t\t\t\tapplyWorkflowMutationTxFn:              tc.applyWorkflowMutationTxFn,\n\t\t\t\tapplyWorkflowSnapshotTxAsResetFn:       tc.applyWorkflowSnapshotTxAsResetFn,\n\t\t\t\tapplyWorkflowSnapshotTxAsNewFn:         tc.applyWorkflowSnapshotTxAsNewFn,\n\t\t\t}\n\n\t\t\terr := s.ConflictResolveWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateFailoverMarkerTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.CreateFailoverMarkersRequest\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 12),\n\t\t\t\t\t\t\tVersion:             101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(tx *sqlplugin.MockTx, parser *serialization.MockParser) {\n\t\t\t\tparser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t}, nil)\n\t\t\t\ttx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), []sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{\n\t\t\t\t\trowsAffected: 1,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - ReplicationTaskInfoToBlob failed\",\n\t\t\treq: &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 12),\n\t\t\t\t\t\t\tVersion:             101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(tx *sqlplugin.MockTx, parser *serialization.MockParser) {\n\t\t\t\tparser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - InsertIntoReplicationTasks failed\",\n\t\t\treq: &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 12),\n\t\t\t\t\t\t\tVersion:             101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(tx *sqlplugin.MockTx, parser *serialization.MockParser) {\n\t\t\t\tparser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t}, nil)\n\t\t\t\ttx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), []sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"some random error\"))\n\t\t\t\ttx.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - row affected error\",\n\t\t\treq: &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 12),\n\t\t\t\t\t\t\tVersion:             101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(tx *sqlplugin.MockTx, parser *serialization.MockParser) {\n\t\t\t\tparser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t}, nil)\n\t\t\t\ttx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), []sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{\n\t\t\t\t\terr: errors.New(\"some error\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - row affected number mismatch\",\n\t\t\treq: &persistence.CreateFailoverMarkersRequest{\n\t\t\t\tRangeID: 1,\n\t\t\t\tMarkers: []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\tVisibilityTimestamp: time.Unix(11, 12),\n\t\t\t\t\t\t\tVersion:             101,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(tx *sqlplugin.MockTx, parser *serialization.MockParser) {\n\t\t\t\tparser.EXPECT().ReplicationTaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t}, nil)\n\t\t\t\ttx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), []sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{\n\t\t\t\t\trowsAffected: 0,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdb := sqlplugin.NewMockDB(ctrl)\n\t\t\tdb.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\ttx := sqlplugin.NewMockTx(ctrl)\n\t\t\tparser := serialization.NewMockParser(ctrl)\n\t\t\ttc.mockSetup(tx, parser)\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     db,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t\tparser: parser,\n\t\t\t\t},\n\t\t\t\ttxExecuteShardLockedFn: func(_ context.Context, _ int, _ string, _ int64, fn func(sqlplugin.Tx) error) error {\n\t\t\t\t\treturn fn(tx)\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := s.CreateFailoverMarkerTasks(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalGetWorkflowExecutionRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.InternalGetWorkflowExecutionResponse\n\t\twantErr   bool\n\t\tassertErr func(t *testing.T, err error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t\tRangeID: 1,\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromExecutions(gomock.Any(), gomock.Any()).Return([]sqlplugin.ExecutionsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:          0,\n\t\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\t\t\tRunID:            serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tNextEventID:      101,\n\t\t\t\t\t\tLastWriteVersion: 11,\n\t\t\t\t\t\tData:             []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding:     \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return([]sqlplugin.ActivityInfoMapsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tScheduleID:   101,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return([]sqlplugin.TimerInfoMapsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tTimerID:      \"101\",\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return([]sqlplugin.ChildExecutionInfoMapsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tInitiatedID:  101,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return([]sqlplugin.RequestCancelInfoMapsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tInitiatedID:  101,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return([]sqlplugin.SignalInfoMapsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tInitiatedID:  101,\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return([]sqlplugin.SignalsRequestedSetsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    0,\n\t\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tSignalID:   \"test-signal-id\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return([]sqlplugin.BufferedEventsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding: \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().WorkflowExecutionInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.WorkflowExecutionInfo{\n\t\t\t\t\tParentDomainID:                       serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\tParentWorkflowID:                     \"test-parent-workflow-id\",\n\t\t\t\t\tParentRunID:                          serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\tInitiatedID:                          101,\n\t\t\t\t\tCompletionEventBatchID:               common.Int64Ptr(11),\n\t\t\t\t\tCompletionEvent:                      []byte(\"test completion event\"),\n\t\t\t\t\tCompletionEventEncoding:              \"json\",\n\t\t\t\t\tTaskList:                             \"test-task-list\",\n\t\t\t\t\tIsCron:                               true,\n\t\t\t\t\tWorkflowTypeName:                     \"test-workflow-type\",\n\t\t\t\t\tWorkflowTimeout:                      time.Duration(101),\n\t\t\t\t\tDecisionTaskTimeout:                  time.Duration(102),\n\t\t\t\t\tExecutionContext:                     []byte(\"test execution context\"),\n\t\t\t\t\tState:                                persistence.WorkflowStateCompleted,\n\t\t\t\t\tCloseStatus:                          persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t\tStartVersion:                         111,\n\t\t\t\t\tLastWriteEventID:                     common.Int64Ptr(11),\n\t\t\t\t\tLastEventTaskID:                      12,\n\t\t\t\t\tLastFirstEventID:                     13,\n\t\t\t\t\tLastProcessedEvent:                   14,\n\t\t\t\t\tStartTimestamp:                       time.Unix(11, 12),\n\t\t\t\t\tLastUpdatedTimestamp:                 time.Unix(13, 14),\n\t\t\t\t\tDecisionVersion:                      101,\n\t\t\t\t\tDecisionScheduleID:                   102,\n\t\t\t\t\tDecisionStartedID:                    103,\n\t\t\t\t\tDecisionTimeout:                      time.Duration(104),\n\t\t\t\t\tDecisionAttempt:                      105,\n\t\t\t\t\tDecisionStartedTimestamp:             time.Unix(15, 16),\n\t\t\t\t\tDecisionScheduledTimestamp:           time.Unix(17, 18),\n\t\t\t\t\tCancelRequested:                      true,\n\t\t\t\t\tDecisionOriginalScheduledTimestamp:   time.Unix(19, 20),\n\t\t\t\t\tCreateRequestID:                      \"test-create-request-id\",\n\t\t\t\t\tDecisionRequestID:                    \"test-decision-request-id\",\n\t\t\t\t\tCancelRequestID:                      \"test-cancel-request-id\",\n\t\t\t\t\tStickyTaskList:                       \"test-sticky-task-list\",\n\t\t\t\t\tStickyScheduleToStartTimeout:         time.Duration(106),\n\t\t\t\t\tRetryAttempt:                         107,\n\t\t\t\t\tRetryInitialInterval:                 time.Duration(108),\n\t\t\t\t\tRetryMaximumInterval:                 time.Duration(109),\n\t\t\t\t\tRetryMaximumAttempts:                 110,\n\t\t\t\t\tRetryExpiration:                      time.Duration(111),\n\t\t\t\t\tRetryBackoffCoefficient:              111,\n\t\t\t\t\tRetryExpirationTimestamp:             time.Unix(23, 24),\n\t\t\t\t\tRetryNonRetryableErrors:              []string{\"error1\", \"error2\"},\n\t\t\t\t\tHasRetryPolicy:                       true,\n\t\t\t\t\tCronSchedule:                         \"test-cron-schedule\",\n\t\t\t\t\tEventStoreVersion:                    112,\n\t\t\t\t\tEventBranchToken:                     []byte(\"test-event-branch-token\"),\n\t\t\t\t\tSignalCount:                          113,\n\t\t\t\t\tHistorySize:                          114,\n\t\t\t\t\tClientLibraryVersion:                 \"test-client-library-version\",\n\t\t\t\t\tClientFeatureVersion:                 \"test-client-feature-version\",\n\t\t\t\t\tClientImpl:                           \"test-client-impl\",\n\t\t\t\t\tAutoResetPoints:                      []byte(\"test-auto-reset-points\"),\n\t\t\t\t\tAutoResetPointsEncoding:              \"json\",\n\t\t\t\t\tSearchAttributes:                     map[string][]byte{\"test-key\": []byte(\"test-value\")},\n\t\t\t\t\tMemo:                                 map[string][]byte{\"test-key\": []byte(\"test-value\")},\n\t\t\t\t\tVersionHistories:                     []byte(\"test-version-histories\"),\n\t\t\t\t\tVersionHistoriesEncoding:             \"json\",\n\t\t\t\t\tFirstExecutionRunID:                  serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\tPartitionConfig:                      map[string]string{\"test-key\": \"test-value\"},\n\t\t\t\t\tChecksum:                             []byte(\"test-checksum\"),\n\t\t\t\t\tChecksumEncoding:                     \"test-checksum-encoding\",\n\t\t\t\t\tActiveClusterSelectionPolicy:         []byte(\"ActiveClusterSelectionPolicy\"),\n\t\t\t\t\tActiveClusterSelectionPolicyEncoding: \"json\",\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().ActivityInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.ActivityInfo{\n\t\t\t\t\tVersion:                  101,\n\t\t\t\t\tScheduledEventBatchID:    102,\n\t\t\t\t\tScheduledEvent:           []byte(\"test scheduled event\"),\n\t\t\t\t\tScheduledEventEncoding:   \"json\",\n\t\t\t\t\tScheduledTimestamp:       time.Unix(11, 12),\n\t\t\t\t\tStartedID:                103,\n\t\t\t\t\tStartedEvent:             []byte(\"test started event\"),\n\t\t\t\t\tStartedEventEncoding:     \"json\",\n\t\t\t\t\tStartedTimestamp:         time.Unix(13, 14),\n\t\t\t\t\tActivityID:               \"test-activity-id\",\n\t\t\t\t\tRequestID:                \"test-request-id\",\n\t\t\t\t\tScheduleToStartTimeout:   time.Duration(101),\n\t\t\t\t\tScheduleToCloseTimeout:   time.Duration(102),\n\t\t\t\t\tStartToCloseTimeout:      time.Duration(103),\n\t\t\t\t\tHeartbeatTimeout:         time.Duration(104),\n\t\t\t\t\tCancelRequested:          true,\n\t\t\t\t\tCancelRequestID:          105,\n\t\t\t\t\tTimerTaskStatus:          105,\n\t\t\t\t\tAttempt:                  106,\n\t\t\t\t\tTaskList:                 \"test-task-list\",\n\t\t\t\t\tTaskListKind:             types.TaskListKindEphemeral,\n\t\t\t\t\tStartedIdentity:          \"test-started-identity\",\n\t\t\t\t\tHasRetryPolicy:           true,\n\t\t\t\t\tRetryInitialInterval:     time.Duration(107),\n\t\t\t\t\tRetryMaximumInterval:     time.Duration(108),\n\t\t\t\t\tRetryMaximumAttempts:     109,\n\t\t\t\t\tRetryExpirationTimestamp: time.Unix(15, 16),\n\t\t\t\t\tRetryBackoffCoefficient:  110,\n\t\t\t\t\tRetryNonRetryableErrors:  []string{\"error1\", \"error2\"},\n\t\t\t\t\tRetryLastFailureReason:   \"test-retry-last-failure-reason\",\n\t\t\t\t\tRetryLastWorkerIdentity:  \"test-retry-last-worker-identity\",\n\t\t\t\t\tRetryLastFailureDetails:  []byte(\"test-retry-last-failure-details\"),\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().TimerInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.TimerInfo{\n\t\t\t\t\tVersion:         101,\n\t\t\t\t\tStartedID:       102,\n\t\t\t\t\tExpiryTimestamp: time.Unix(11, 12),\n\t\t\t\t\tTaskID:          103,\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().ChildExecutionInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.ChildExecutionInfo{\n\t\t\t\t\tVersion:                101,\n\t\t\t\t\tInitiatedEventBatchID:  102,\n\t\t\t\t\tInitiatedEvent:         []byte(\"test initiated event\"),\n\t\t\t\t\tInitiatedEventEncoding: \"json\",\n\t\t\t\t\tStartedID:              103,\n\t\t\t\t\tStartedWorkflowID:      \"test-started-workflow-id\",\n\t\t\t\t\tStartedRunID:           serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\tCreateRequestID:        \"test-create-request-id\",\n\t\t\t\t\tStartedEvent:           []byte(\"test started event\"),\n\t\t\t\t\tStartedEventEncoding:   \"json\",\n\t\t\t\t\tDomainID:               \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\tWorkflowTypeName:       \"test-workflow-type\",\n\t\t\t\t\tParentClosePolicy:      101,\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().RequestCancelInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.RequestCancelInfo{\n\t\t\t\t\tVersion:               101,\n\t\t\t\t\tInitiatedEventBatchID: 102,\n\t\t\t\t\tCancelRequestID:       \"test-cancel-request-id\",\n\t\t\t\t}, nil)\n\t\t\t\tparser.EXPECT().SignalInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.SignalInfo{\n\t\t\t\t\tVersion:               101,\n\t\t\t\t\tInitiatedEventBatchID: 102,\n\t\t\t\t\tName:                  \"test-signal-name\",\n\t\t\t\t\tInput:                 []byte(\"test input\"),\n\t\t\t\t\tControl:               []byte(\"test control\"),\n\t\t\t\t\tRequestID:             \"test-signal-request-id\",\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(&sqlplugin.ShardsRow{\n\t\t\t\t\tRangeID: 1,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.InternalWorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:                           \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t\tWorkflowID:                         \"test-workflow-id\",\n\t\t\t\t\t\tRunID:                              \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t\t\tParentDomainID:                     \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t\tParentWorkflowID:                   \"test-parent-workflow-id\",\n\t\t\t\t\t\tParentRunID:                        \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t\t\tInitiatedID:                        101,\n\t\t\t\t\t\tCompletionEventBatchID:             11,\n\t\t\t\t\t\tCompletionEvent:                    persistence.NewDataBlob([]byte(\"test completion event\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\tTaskList:                           \"test-task-list\",\n\t\t\t\t\t\tIsCron:                             true,\n\t\t\t\t\t\tWorkflowTypeName:                   \"test-workflow-type\",\n\t\t\t\t\t\tWorkflowTimeout:                    time.Duration(101),\n\t\t\t\t\t\tDecisionStartToCloseTimeout:        time.Duration(102),\n\t\t\t\t\t\tDecisionTimeout:                    time.Duration(104),\n\t\t\t\t\t\tExecutionContext:                   []byte(\"test execution context\"),\n\t\t\t\t\t\tState:                              persistence.WorkflowStateCompleted,\n\t\t\t\t\t\tCloseStatus:                        persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t\t\tNextEventID:                        101,\n\t\t\t\t\t\tLastEventTaskID:                    12,\n\t\t\t\t\t\tLastFirstEventID:                   13,\n\t\t\t\t\t\tLastProcessedEvent:                 14,\n\t\t\t\t\t\tStartTimestamp:                     time.Unix(11, 12),\n\t\t\t\t\t\tLastUpdatedTimestamp:               time.Unix(13, 14),\n\t\t\t\t\t\tDecisionVersion:                    101,\n\t\t\t\t\t\tDecisionScheduleID:                 102,\n\t\t\t\t\t\tDecisionStartedID:                  103,\n\t\t\t\t\t\tDecisionAttempt:                    105,\n\t\t\t\t\t\tDecisionStartedTimestamp:           time.Unix(15, 16),\n\t\t\t\t\t\tDecisionScheduledTimestamp:         time.Unix(17, 18),\n\t\t\t\t\t\tCancelRequested:                    true,\n\t\t\t\t\t\tDecisionOriginalScheduledTimestamp: time.Unix(19, 20),\n\t\t\t\t\t\tCreateRequestID:                    \"test-create-request-id\",\n\t\t\t\t\t\tDecisionRequestID:                  \"test-decision-request-id\",\n\t\t\t\t\t\tCancelRequestID:                    \"test-cancel-request-id\",\n\t\t\t\t\t\tStickyTaskList:                     \"test-sticky-task-list\",\n\t\t\t\t\t\tStickyScheduleToStartTimeout:       time.Duration(106),\n\t\t\t\t\t\tHasRetryPolicy:                     true,\n\t\t\t\t\t\tCronSchedule:                       \"test-cron-schedule\",\n\t\t\t\t\t\tSignalCount:                        113,\n\t\t\t\t\t\tHistorySize:                        114,\n\t\t\t\t\t\tClientLibraryVersion:               \"test-client-library-version\",\n\t\t\t\t\t\tClientFeatureVersion:               \"test-client-feature-version\",\n\t\t\t\t\t\tClientImpl:                         \"test-client-impl\",\n\t\t\t\t\t\tFirstExecutionRunID:                \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t\t\tPartitionConfig:                    map[string]string{\"test-key\": \"test-value\"},\n\t\t\t\t\t\tAutoResetPoints:                    persistence.NewDataBlob([]byte(\"test-auto-reset-points\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\tAttempt:                            107,\n\t\t\t\t\t\tInitialInterval:                    time.Duration(108),\n\t\t\t\t\t\tBackoffCoefficient:                 111,\n\t\t\t\t\t\tMaximumInterval:                    time.Duration(109),\n\t\t\t\t\t\tExpirationTime:                     time.Unix(23, 24),\n\t\t\t\t\t\tMaximumAttempts:                    110,\n\t\t\t\t\t\tNonRetriableErrors:                 []string{\"error1\", \"error2\"},\n\t\t\t\t\t\tBranchToken:                        []byte(\"test-event-branch-token\"),\n\t\t\t\t\t\tSearchAttributes:                   map[string][]byte{\"test-key\": []byte(\"test-value\")},\n\t\t\t\t\t\tMemo:                               map[string][]byte{\"test-key\": []byte(\"test-value\")},\n\t\t\t\t\t\tExpirationInterval:                 time.Duration(111),\n\t\t\t\t\t\tActiveClusterSelectionPolicy:       persistence.NewDataBlob([]byte(\"ActiveClusterSelectionPolicy\"), constants.EncodingTypeJSON),\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: persistence.NewDataBlob([]byte(\"test-version-histories\"), constants.EncodingTypeJSON),\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tStartVersion:     111,\n\t\t\t\t\t\tLastWriteVersion: 11,\n\t\t\t\t\t\tLastWriteEventID: 11,\n\t\t\t\t\t},\n\t\t\t\t\tActivityInfos: map[int64]*persistence.InternalActivityInfo{\n\t\t\t\t\t\t101: {\n\t\t\t\t\t\t\tVersion:                101,\n\t\t\t\t\t\t\tScheduleID:             101,\n\t\t\t\t\t\t\tScheduledEventBatchID:  102,\n\t\t\t\t\t\t\tScheduledEvent:         persistence.NewDataBlob([]byte(\"test scheduled event\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\t\tScheduledTime:          time.Unix(11, 12),\n\t\t\t\t\t\t\tStartedID:              103,\n\t\t\t\t\t\t\tStartedTime:            time.Unix(13, 14),\n\t\t\t\t\t\t\tStartedEvent:           persistence.NewDataBlob([]byte(\"test started event\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\t\tActivityID:             \"test-activity-id\",\n\t\t\t\t\t\t\tRequestID:              \"test-request-id\",\n\t\t\t\t\t\t\tScheduleToStartTimeout: time.Duration(101),\n\t\t\t\t\t\t\tScheduleToCloseTimeout: time.Duration(102),\n\t\t\t\t\t\t\tStartToCloseTimeout:    time.Duration(103),\n\t\t\t\t\t\t\tHeartbeatTimeout:       time.Duration(104),\n\t\t\t\t\t\t\tCancelRequested:        true,\n\t\t\t\t\t\t\tCancelRequestID:        105,\n\t\t\t\t\t\t\tTimerTaskStatus:        105,\n\t\t\t\t\t\t\tAttempt:                106,\n\t\t\t\t\t\t\tTaskList:               \"test-task-list\",\n\t\t\t\t\t\t\tTaskListKind:           types.TaskListKindEphemeral,\n\t\t\t\t\t\t\tStartedIdentity:        \"test-started-identity\",\n\t\t\t\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\t\t\t\tDomainID:               \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t\t\tInitialInterval:        time.Duration(107),\n\t\t\t\t\t\t\tMaximumInterval:        time.Duration(108),\n\t\t\t\t\t\t\tMaximumAttempts:        109,\n\t\t\t\t\t\t\tExpirationTime:         time.Unix(15, 16),\n\t\t\t\t\t\t\tBackoffCoefficient:     110,\n\t\t\t\t\t\t\tNonRetriableErrors:     []string{\"error1\", \"error2\"},\n\t\t\t\t\t\t\tLastFailureReason:      \"test-retry-last-failure-reason\",\n\t\t\t\t\t\t\tLastWorkerIdentity:     \"test-retry-last-worker-identity\",\n\t\t\t\t\t\t\tLastFailureDetails:     []byte(\"test-retry-last-failure-details\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\t\t\t\"101\": {\n\t\t\t\t\t\t\tVersion:    101,\n\t\t\t\t\t\t\tStartedID:  102,\n\t\t\t\t\t\t\tExpiryTime: time.Unix(11, 12),\n\t\t\t\t\t\t\tTaskStatus: 103,\n\t\t\t\t\t\t\tTimerID:    \"101\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tChildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t\t101: {\n\t\t\t\t\t\t\tVersion:               101,\n\t\t\t\t\t\t\tInitiatedID:           101,\n\t\t\t\t\t\t\tInitiatedEvent:        persistence.NewDataBlob([]byte(\"test initiated event\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\t\tInitiatedEventBatchID: 102,\n\t\t\t\t\t\t\tStartedID:             103,\n\t\t\t\t\t\t\tStartedEvent:          persistence.NewDataBlob([]byte(\"test started event\"), constants.EncodingTypeJSON),\n\t\t\t\t\t\t\tStartedWorkflowID:     \"test-started-workflow-id\",\n\t\t\t\t\t\t\tStartedRunID:          \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t\t\t\tCreateRequestID:       \"test-create-request-id\",\n\t\t\t\t\t\t\tDomainID:              \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t\t\tWorkflowTypeName:      \"test-workflow-type\",\n\t\t\t\t\t\t\tParentClosePolicy:     101,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tRequestCancelInfos: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t\t\t101: {\n\t\t\t\t\t\t\tVersion:               101,\n\t\t\t\t\t\t\tInitiatedID:           101,\n\t\t\t\t\t\t\tInitiatedEventBatchID: 102,\n\t\t\t\t\t\t\tCancelRequestID:       \"test-cancel-request-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tSignalInfos: map[int64]*persistence.SignalInfo{\n\t\t\t\t\t\t101: {\n\t\t\t\t\t\t\tVersion:               101,\n\t\t\t\t\t\t\tInitiatedID:           101,\n\t\t\t\t\t\t\tInitiatedEventBatchID: 102,\n\t\t\t\t\t\t\tSignalName:            \"test-signal-name\",\n\t\t\t\t\t\t\tInput:                 []byte(\"test input\"),\n\t\t\t\t\t\t\tControl:               []byte(\"test control\"),\n\t\t\t\t\t\t\tSignalRequestID:       \"test-signal-request-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tSignalRequestedIDs: map[string]struct{}{\n\t\t\t\t\t\t\"test-signal-id\": {},\n\t\t\t\t\t},\n\t\t\t\t\tBufferedEvents: []*persistence.DataBlob{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\t\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tChecksumData: persistence.NewDataBlob([]byte(\"test-checksum\"), \"test-checksum-encoding\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - Shard owner changed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromExecutions(gomock.Any(), gomock.Any()).Return([]sqlplugin.ExecutionsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:          0,\n\t\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\t\t\tRunID:            serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tNextEventID:      101,\n\t\t\t\t\t\tLastWriteVersion: 11,\n\t\t\t\t\t\tData:             []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding:     \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tparser.EXPECT().WorkflowExecutionInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.WorkflowExecutionInfo{\n\t\t\t\t\tChecksum:         []byte(\"test-checksum\"),\n\t\t\t\t\tChecksumEncoding: \"test-checksum-encoding\",\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(&sqlplugin.ShardsRow{\n\t\t\t\t\tRangeID: 1,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.InternalWorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:               \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\t\t\tWorkflowID:             \"test-workflow-id\",\n\t\t\t\t\t\tRunID:                  \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t\t\tNextEventID:            101,\n\t\t\t\t\t\tCompletionEventBatchID: -23,\n\t\t\t\t\t},\n\t\t\t\t\tActivityInfos:       map[int64]*persistence.InternalActivityInfo{},\n\t\t\t\t\tTimerInfos:          map[string]*persistence.TimerInfo{},\n\t\t\t\t\tChildExecutionInfos: map[int64]*persistence.InternalChildExecutionInfo{},\n\t\t\t\t\tRequestCancelInfos:  map[int64]*persistence.RequestCancelInfo{},\n\t\t\t\t\tSignalInfos:         map[int64]*persistence.SignalInfo{},\n\t\t\t\t\tSignalRequestedIDs:  map[string]struct{}{},\n\t\t\t\t\tChecksumData:        nil,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - failed to get shard\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromExecutions(gomock.Any(), gomock.Any()).Return([]sqlplugin.ExecutionsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:          0,\n\t\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\"),\n\t\t\t\t\t\tWorkflowID:       \"test-workflow-id\",\n\t\t\t\t\t\tRunID:            serialization.MustParseUUID(\"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\"),\n\t\t\t\t\t\tNextEventID:      101,\n\t\t\t\t\t\tLastWriteVersion: 11,\n\t\t\t\t\t\tData:             []byte(\"test data\"),\n\t\t\t\t\t\tDataEncoding:     \"thriftrw\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tparser.EXPECT().WorkflowExecutionInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.WorkflowExecutionInfo{\n\t\t\t\t\tChecksum:         []byte(\"test-checksum\"),\n\t\t\t\t\tChecksumEncoding: \"test-checksum-encoding\",\n\t\t\t\t}, nil)\n\t\t\t\tdb.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromExecutions no row\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromExecutions(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.EntityNotExistsError{}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromExecutions failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromExecutions(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromActivityInfoMaps failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromTimerInfoMaps failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromChildExecutionInfoMaps failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromRequestCancelInfoMaps failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromSignalInfoMaps failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromSignalsRequestedSets failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - SelectFromBufferedEvents failed\",\n\t\t\treq: &persistence.InternalGetWorkflowExecutionRequest{\n\t\t\t\tDomainID: \"ff9c8a3f-0e4f-4d3e-a4d2-6f5f8f3f7d9d\",\n\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"ee8d7b6e-876c-4b1e-9b6e-5e3e3c6b6b3f\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(db *sqlplugin.MockDB, parser *serialization.MockParser) {\n\t\t\t\tdb.EXPECT().SelectFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tdb.EXPECT().SelectFromBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdb.EXPECT().IsNotFoundError(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdb := sqlplugin.NewMockDB(ctrl)\n\t\t\tparser := serialization.NewMockParser(ctrl)\n\t\t\ttc.mockSetup(db, parser)\n\t\t\ts := &sqlExecutionStore{\n\t\t\t\tshardID: 0,\n\t\t\t\tsqlStore: sqlStore{\n\t\t\t\t\tdb:     db,\n\t\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\t\tparser: parser,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tresp, err := s.GetWorkflowExecution(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp, \"Response mismatch\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeCompleteHistoryTask(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *persistence.RangeCompleteHistoryTaskRequest\n\t\tsetupMock     func(*sqlplugin.MockDB)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success - scheduled timer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:                shardID,\n\t\t\t\t\tMinVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\tMaxVisibilityTimestamp: time.Unix(0, 0).Add(time.Minute),\n\t\t\t\t\tPageSize:               1000,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - immediate transfer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - immediate replication task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100), // this is ignored by replication task\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category error\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB *sqlplugin.MockDB) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"database error on timer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:                shardID,\n\t\t\t\t\tMinVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\tMaxVisibilityTimestamp: time.Unix(0, 0).Add(time.Minute),\n\t\t\t\t\tPageSize:               1000,\n\t\t\t\t}).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on transfer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on replication task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"sql result error on timer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute), 0),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:                shardID,\n\t\t\t\t\tMinVisibilityTimestamp: time.Unix(0, 0),\n\t\t\t\t\tMaxVisibilityTimestamp: time.Unix(0, 0).Add(time.Minute),\n\t\t\t\t\tPageSize:               1000,\n\t\t\t\t}).Return(&sqlResult{err: errors.New(\"sql result error\")}, nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"sql result error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"sql result error on transfer task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tInclusiveMinTaskID: 100,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(&sqlResult{err: errors.New(\"sql result error\")}, nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"sql result error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"sql result error on replication task\",\n\t\t\trequest: &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            1000,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           1000,\n\t\t\t\t}).Return(&sqlResult{err: errors.New(\"sql result error\")}, nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"sql result error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(controller)\n\t\t\tstore := &sqlExecutionStore{sqlStore: sqlStore{db: mockDB}, shardID: shardID}\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\tresp, err := store.RangeCompleteHistoryTask(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, 1, resp.TasksCompleted)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetHistoryTasks_SQL(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname                  string\n\t\trequest               *persistence.GetHistoryTasksRequest\n\t\tsetupMock             func(*sqlplugin.MockDB, *serialization.MockTaskSerializer)\n\t\texpectedError         error\n\t\texpectedTasks         []persistence.Task\n\t\texpectedNextPageToken []byte\n\t}{\n\t\t{\n\t\t\tname: \"success - get immediate transfer tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       serializePageToken(101),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tInclusiveMinTaskID: 101,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           10,\n\t\t\t\t}).Return([]sqlplugin.TransferTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       101,\n\t\t\t\t\t\tData:         []byte(`{\"task\": \"transfer\"}`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryTransfer, persistence.NewDataBlob([]byte(`{\"task\": \"transfer\"}`), constants.EncodingTypeJSON)).Return(&persistence.DecisionTask{\n\t\t\t\t\tTaskList: \"test-task-list\",\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\tTaskList: \"test-task-list\",\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID: 101,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextPageToken: serializePageToken(102),\n\t\t},\n\t\t{\n\t\t\tname: \"success - get scheduled timer tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).UTC(), 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0).Add(time.Minute).UTC(), 0),\n\t\t\t\tPageSize:            1,\n\t\t\t\tNextPageToken: func() []byte {\n\t\t\t\t\tti := &timerTaskPageToken{TaskID: 10, Timestamp: time.Unix(0, 1).UTC()}\n\t\t\t\t\ttoken, err := ti.serialize()\n\t\t\t\t\trequire.NoError(t, err, \"failed to serialize timer page token\")\n\t\t\t\t\treturn token\n\t\t\t\t}(),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:                shardID,\n\t\t\t\t\tMinVisibilityTimestamp: time.Unix(0, 1).UTC(),\n\t\t\t\t\tTaskID:                 10,\n\t\t\t\t\tMaxVisibilityTimestamp: time.Unix(0, 0).Add(time.Minute).UTC(),\n\t\t\t\t\tPageSize:               2,\n\t\t\t\t}).Return([]sqlplugin.TimerTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tTaskID:              10,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tData:                []byte(`{\"task\": \"timer\"}`),\n\t\t\t\t\t\tDataEncoding:        \"json\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tTaskID:              101,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tData:                []byte(`{\"task\": \"timer\"}`),\n\t\t\t\t\t\tDataEncoding:        \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryTimer, persistence.NewDataBlob([]byte(`{\"task\": \"timer\"}`), constants.EncodingTypeJSON)).Return(&persistence.UserTimerTask{\n\t\t\t\t\tEventID: 100,\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryTimer, persistence.NewDataBlob([]byte(`{\"task\": \"timer\"}`), constants.EncodingTypeJSON)).Return(&persistence.UserTimerTask{\n\t\t\t\t\tEventID: 101,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\tEventID: 100,\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              10,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextPageToken: func() []byte {\n\t\t\t\tti := &timerTaskPageToken{TaskID: 101, Timestamp: time.Unix(1, 1)}\n\t\t\t\ttoken, err := ti.serialize()\n\t\t\t\trequire.NoError(t, err, \"failed to serialize timer page token\")\n\t\t\t\treturn token\n\t\t\t}(),\n\t\t},\n\t\t{\n\t\t\tname: \"success - get immediate replication tasks\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\tPageSize:            10,\n\t\t\t\tNextPageToken:       serializePageToken(101),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID:            shardID,\n\t\t\t\t\tInclusiveMinTaskID: 101,\n\t\t\t\t\tExclusiveMaxTaskID: 200,\n\t\t\t\t\tPageSize:           10,\n\t\t\t\t}).Return([]sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       101,\n\t\t\t\t\t\tData:         []byte(`{\"task\": \"replication\"}`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().DeserializeTask(persistence.HistoryTaskCategoryReplication, persistence.NewDataBlob([]byte(`{\"task\": \"replication\"}`), constants.EncodingTypeJSON)).Return(&persistence.HistoryReplicationTask{\n\t\t\t\t\tFirstEventID: 100,\n\t\t\t\t\tNextEventID:  200,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t\texpectedTasks: []persistence.Task{\n\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\tFirstEventID: 100,\n\t\t\t\t\tNextEventID:  200,\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID: 101,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextPageToken: serializePageToken(102),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on transfer task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromTransferTasks(ctx, gomock.Any()).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on replication task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromReplicationTasks(ctx, gomock.Any()).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"database error on timer task retrieval\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tPageSize:     10,\n\t\t\t},\n\t\t\tsetupMock: func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockDB.EXPECT().SelectFromTimerTasks(ctx, gomock.Any()).Return(nil, errors.New(\"db error\"))\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category error\",\n\t\t\trequest: &persistence.GetHistoryTasksRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB *sqlplugin.MockDB, mockTaskSerializer *serialization.MockTaskSerializer) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(controller)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(controller)\n\t\t\tstore := &sqlExecutionStore{sqlStore: sqlStore{db: mockDB}, shardID: shardID, taskSerializer: mockTaskSerializer}\n\n\t\t\ttc.setupMock(mockDB, mockTaskSerializer)\n\n\t\t\tresp, err := store.GetHistoryTasks(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedTasks, resp.Tasks)\n\t\t\t\tassert.Equal(t, tc.expectedNextPageToken, resp.NextPageToken)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCompleteHistoryTask(t *testing.T) {\n\tctx := context.Background()\n\tshardID := 1\n\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *persistence.CompleteHistoryTaskRequest\n\t\tsetupMock     func(any)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"success - complete scheduled timer task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(10, 10), 1),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(10, 10),\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - complete immediate transfer task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID: shardID,\n\t\t\t\t\tTaskID:  2,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - complete immediate replication task\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID: shardID,\n\t\t\t\t\tTaskID:  3,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"unknown task category type\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategory{},\n\t\t\t},\n\t\t\tsetupMock:     func(mockDB any) {},\n\t\t\texpectedError: &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"delete timer task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(10, 10), 1),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromTimerTasks(ctx, &sqlplugin.TimerTasksFilter{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(10, 10),\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 0}, errors.New(\"db error\"))\n\t\t\t\tmock.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"delete transfer task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromTransferTasks(ctx, &sqlplugin.TransferTasksFilter{\n\t\t\t\t\tShardID: shardID,\n\t\t\t\t\tTaskID:  2,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 0}, errors.New(\"db error\"))\n\t\t\t\tmock.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"delete replication task error\",\n\t\t\trequest: &persistence.CompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tsetupMock: func(mockDB any) {\n\t\t\t\tmock := mockDB.(*sqlplugin.MockDB)\n\t\t\t\tmock.EXPECT().DeleteFromReplicationTasks(ctx, &sqlplugin.ReplicationTasksFilter{\n\t\t\t\t\tShardID: shardID,\n\t\t\t\t\tTaskID:  3,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 0}, errors.New(\"db error\"))\n\t\t\t\tmock.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\texpectedError: errors.New(\"db error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(controller)\n\t\t\tstore := &sqlExecutionStore{sqlStore: sqlStore{db: mockDB}, shardID: shardID}\n\n\t\t\ttc.setupMock(mockDB)\n\n\t\t\terr := store.CompleteHistoryTask(ctx, tc.request)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_execution_store_util.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\t\"time\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc applyWorkflowMutationTx(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tworkflowMutation *p.InternalWorkflowMutation,\n\tparser serialization.Parser,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\texecutionInfo := workflowMutation.ExecutionInfo\n\tversionHistories := workflowMutation.VersionHistories\n\tworkflowChecksum := workflowMutation.ChecksumData\n\tstartVersion := workflowMutation.StartVersion\n\tlastWriteVersion := workflowMutation.LastWriteVersion\n\tdomainID := serialization.MustParseUUID(executionInfo.DomainID)\n\tworkflowID := executionInfo.WorkflowID\n\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\n\t// TODO Remove me if UPDATE holds the lock to the end of a transaction\n\tif err := lockAndCheckNextEventID(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowMutation.Condition); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateExecution(\n\t\tctx,\n\t\ttx,\n\t\texecutionInfo,\n\t\tversionHistories,\n\t\tworkflowChecksum,\n\t\tstartVersion,\n\t\tlastWriteVersion,\n\t\tshardID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := applyTasks(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tworkflowMutation.TasksByCategory,\n\t\ttaskSerializer,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateActivityInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertActivityInfos,\n\t\tworkflowMutation.DeleteActivityInfos,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateTimerInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertTimerInfos,\n\t\tworkflowMutation.DeleteTimerInfos,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateChildExecutionInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertChildExecutionInfos,\n\t\tworkflowMutation.DeleteChildExecutionInfos,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateRequestCancelInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertRequestCancelInfos,\n\t\tworkflowMutation.DeleteRequestCancelInfos,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateSignalInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertSignalInfos,\n\t\tworkflowMutation.DeleteSignalInfos,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateSignalsRequested(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.UpsertSignalRequestedIDs,\n\t\tworkflowMutation.DeleteSignalRequestedIDs,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif workflowMutation.ClearBufferedEvents {\n\t\tif err := deleteBufferedEvents(\n\t\t\tctx,\n\t\t\ttx,\n\t\t\tshardID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn updateBufferedEvents(\n\t\tctx,\n\t\ttx,\n\t\tworkflowMutation.NewBufferedEvents,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t)\n}\n\nfunc applyWorkflowSnapshotTxAsReset(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tworkflowSnapshot *p.InternalWorkflowSnapshot,\n\tparser serialization.Parser,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\texecutionInfo := workflowSnapshot.ExecutionInfo\n\tversionHistories := workflowSnapshot.VersionHistories\n\tworkflowChecksum := workflowSnapshot.ChecksumData\n\tstartVersion := workflowSnapshot.StartVersion\n\tlastWriteVersion := workflowSnapshot.LastWriteVersion\n\tdomainID := serialization.MustParseUUID(executionInfo.DomainID)\n\tworkflowID := executionInfo.WorkflowID\n\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\n\t// TODO Is there a way to modify the various map tables without fear of other people adding rows after we delete, without locking the executions row?\n\tif err := lockAndCheckNextEventID(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowSnapshot.Condition); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateExecution(\n\t\tctx,\n\t\ttx,\n\t\texecutionInfo,\n\t\tversionHistories,\n\t\tworkflowChecksum,\n\t\tstartVersion,\n\t\tlastWriteVersion,\n\t\tshardID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := applyTasks(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tworkflowSnapshot.TasksByCategory,\n\t\ttaskSerializer,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteActivityInfoMap(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateActivityInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.ActivityInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteTimerInfoMap(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateTimerInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.TimerInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteChildExecutionInfoMap(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateChildExecutionInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.ChildExecutionInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteRequestCancelInfoMap(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateRequestCancelInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.RequestCancelInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteSignalInfoMap(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateSignalInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.SignalInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := deleteSignalsRequestedSet(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateSignalsRequested(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.SignalRequestedIDs,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID); err != nil {\n\t\treturn err\n\t}\n\n\treturn deleteBufferedEvents(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID)\n}\n\nfunc applyWorkflowSnapshotTxAsNew(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tworkflowSnapshot *p.InternalWorkflowSnapshot,\n\tparser serialization.Parser,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\texecutionInfo := workflowSnapshot.ExecutionInfo\n\tversionHistories := workflowSnapshot.VersionHistories\n\tworkflowChecksum := workflowSnapshot.ChecksumData\n\tstartVersion := workflowSnapshot.StartVersion\n\tlastWriteVersion := workflowSnapshot.LastWriteVersion\n\tdomainID := serialization.MustParseUUID(executionInfo.DomainID)\n\tworkflowID := executionInfo.WorkflowID\n\trunID := serialization.MustParseUUID(executionInfo.RunID)\n\n\t// TODO(active-active): store active cluster selection policy row. It requires a new table in sql DB schemas.\n\n\tif err := createExecution(\n\t\tctx,\n\t\ttx,\n\t\texecutionInfo,\n\t\tversionHistories,\n\t\tworkflowChecksum,\n\t\tstartVersion,\n\t\tlastWriteVersion,\n\t\tshardID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := applyTasks(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tworkflowSnapshot.TasksByCategory,\n\t\ttaskSerializer,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateActivityInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.ActivityInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateTimerInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.TimerInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateChildExecutionInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.ChildExecutionInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateRequestCancelInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.RequestCancelInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\tif err := updateSignalInfos(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.SignalInfos,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tparser); err != nil {\n\t\treturn err\n\t}\n\n\treturn updateSignalsRequested(\n\t\tctx,\n\t\ttx,\n\t\tworkflowSnapshot.SignalRequestedIDs,\n\t\tnil,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID)\n}\n\nfunc applyTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\ttasksByCategory map[p.HistoryTaskCategory][]p.Task,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\tvar err error\n\tfor c, tasks := range tasksByCategory {\n\t\tswitch c.Type() {\n\t\tcase p.HistoryTaskCategoryTypeImmediate:\n\t\t\terr = createImmediateTasks(ctx, tx, shardID, c.ID(), tasks, taskSerializer)\n\t\tcase p.HistoryTaskCategoryTypeScheduled:\n\t\t\terr = createScheduledTasks(ctx, tx, shardID, c.ID(), tasks, taskSerializer)\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// lockCurrentExecutionIfExists returns current execution or nil if none is found for the workflowID\n// locking it in the DB\nfunc lockCurrentExecutionIfExists(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n) (*sqlplugin.CurrentExecutionsRow, error) {\n\n\trows, err := tx.LockCurrentExecutionsJoinExecutions(ctx, &sqlplugin.CurrentExecutionsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\tif err != nil {\n\t\tif err != sql.ErrNoRows {\n\t\t\treturn nil, convertCommonErrors(tx, \"lockCurrentExecutionIfExists\", fmt.Sprintf(\"Failed to get current_executions row for (shard,domain,workflow) = (%v, %v, %v).\", shardID, domainID, workflowID), err)\n\t\t}\n\t}\n\tsize := len(rows)\n\tif size > 1 {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"lockCurrentExecutionIfExists failed. Multiple current_executions rows for (shard,domain,workflow) = (%v, %v, %v).\", shardID, domainID, workflowID),\n\t\t}\n\t}\n\tif size == 0 {\n\t\treturn nil, nil\n\t}\n\treturn &rows[0], nil\n}\n\nfunc createOrUpdateCurrentExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tcreateMode p.CreateWorkflowMode,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tstate int,\n\tcloseStatus int,\n\tcreateRequestID string,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n) error {\n\n\trow := sqlplugin.CurrentExecutionsRow{\n\t\tShardID:          int64(shardID),\n\t\tDomainID:         domainID,\n\t\tWorkflowID:       workflowID,\n\t\tRunID:            runID,\n\t\tCreateRequestID:  createRequestID,\n\t\tState:            state,\n\t\tCloseStatus:      closeStatus,\n\t\tStartVersion:     startVersion,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}\n\n\tswitch createMode {\n\tcase p.CreateWorkflowModeContinueAsNew,\n\t\tp.CreateWorkflowModeWorkflowIDReuse:\n\t\tif err := updateCurrentExecution(\n\t\t\tctx,\n\t\t\ttx,\n\t\t\tshardID,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tcreateRequestID,\n\t\t\tstate,\n\t\t\tcloseStatus,\n\t\t\trow.StartVersion,\n\t\t\trow.LastWriteVersion); err != nil {\n\t\t\treturn err\n\t\t}\n\tcase p.CreateWorkflowModeBrandNew:\n\t\tif _, err := tx.InsertIntoCurrentExecutions(ctx, &row); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"createOrUpdateCurrentExecution\", \"Failed to insert into current_executions table.\", err)\n\t\t}\n\tcase p.CreateWorkflowModeZombie:\n\t\t// noop\n\tdefault:\n\t\treturn fmt.Errorf(\"createOrUpdateCurrentExecution failed. Unknown workflow creation mode: %v\", createMode)\n\t}\n\n\treturn nil\n}\n\nfunc lockAndCheckNextEventID(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tcondition int64,\n) error {\n\n\tnextEventID, err := lockNextEventID(\n\t\tctx,\n\t\ttx,\n\t\tshardID,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t)\n\n\tif err != nil {\n\t\treturn err\n\t}\n\tif *nextEventID != condition {\n\t\treturn &p.ConditionFailedError{\n\t\t\tMsg: fmt.Sprintf(\"lockAndCheckNextEventID failed. Next_event_id was %v when it should have been %v.\", nextEventID, condition),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc lockNextEventID(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) (*int64, error) {\n\n\tnextEventID, err := tx.WriteLockExecutions(ctx, &sqlplugin.ExecutionsFilter{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\t\"lockNextEventID failed. Unable to lock executions row with (shard, domain, workflow, run) = (%v,%v,%v,%v) which does not exist.\",\n\t\t\t\t\tshardID,\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(tx, \"lockNextEventID\", \"\", err)\n\t}\n\tresult := int64(nextEventID)\n\treturn &result, nil\n}\n\nfunc createImmediateTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tcategoryID int,\n\ttasks []p.Task,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\tswitch categoryID {\n\tcase p.HistoryTaskCategoryIDTransfer:\n\t\treturn createTransferTasks(ctx, tx, tasks, shardID, taskSerializer)\n\tcase p.HistoryTaskCategoryIDReplication:\n\t\treturn createReplicationTasks(ctx, tx, tasks, shardID, taskSerializer)\n\t}\n\t// TODO: implement creating tasks for other categories\n\treturn nil\n}\n\nfunc createScheduledTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tcategoryID int,\n\ttasks []p.Task,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\tswitch categoryID {\n\tcase p.HistoryTaskCategoryIDTimer:\n\t\treturn createTimerTasks(ctx, tx, tasks, shardID, taskSerializer)\n\t}\n\t// TODO: implement creating tasks for other categories\n\treturn nil\n}\n\nfunc createTransferTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\ttransferTasks []p.Task,\n\tshardID int,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\tif len(transferTasks) == 0 {\n\t\treturn nil\n\t}\n\n\ttransferTasksRows := make([]sqlplugin.TransferTasksRow, len(transferTasks))\n\tfor i, task := range transferTasks {\n\t\tblob, err := taskSerializer.SerializeTask(p.HistoryTaskCategoryTransfer, task)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttransferTasksRows[i].ShardID = shardID\n\t\ttransferTasksRows[i].TaskID = task.GetTaskID()\n\t\ttransferTasksRows[i].Data = blob.Data\n\t\ttransferTasksRows[i].DataEncoding = string(blob.Encoding)\n\t}\n\n\tresult, err := tx.InsertIntoTransferTasks(ctx, transferTasksRows)\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"createTransferTasks\", \"\", err)\n\t}\n\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createTransferTasks failed. Could not verify number of rows inserted. Error: %v\", err),\n\t\t}\n\t}\n\n\tif int(rowsAffected) != len(transferTasks) {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createTransferTasks failed. Inserted %v instead of %v rows into transfer_tasks. Error: %v\", rowsAffected, len(transferTasks), err),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc createReplicationTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\treplicationTasks []p.Task,\n\tshardID int,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\tif len(replicationTasks) == 0 {\n\t\treturn nil\n\t}\n\treplicationTasksRows := make([]sqlplugin.ReplicationTasksRow, len(replicationTasks))\n\n\tfor i, task := range replicationTasks {\n\t\tblob, err := taskSerializer.SerializeTask(p.HistoryTaskCategoryReplication, task)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treplicationTasksRows[i].ShardID = shardID\n\t\treplicationTasksRows[i].TaskID = task.GetTaskID()\n\t\treplicationTasksRows[i].Data = blob.Data\n\t\treplicationTasksRows[i].DataEncoding = string(blob.Encoding)\n\t}\n\n\tresult, err := tx.InsertIntoReplicationTasks(ctx, replicationTasksRows)\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"createReplicationTasks\", \"\", err)\n\t}\n\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createReplicationTasks failed. Could not verify number of rows inserted. Error: %v\", err),\n\t\t}\n\t}\n\n\tif int(rowsAffected) != len(replicationTasks) {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createReplicationTasks failed. Inserted %v instead of %v rows into transfer_tasks. Error: %v\", rowsAffected, len(replicationTasks), err),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc createTimerTasks(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\ttimerTasks []p.Task,\n\tshardID int,\n\ttaskSerializer serialization.TaskSerializer,\n) error {\n\n\tif len(timerTasks) == 0 {\n\t\treturn nil\n\t}\n\n\ttimerTasksRows := make([]sqlplugin.TimerTasksRow, len(timerTasks))\n\n\tfor i, task := range timerTasks {\n\t\tblob, err := taskSerializer.SerializeTask(p.HistoryTaskCategoryTimer, task)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttimerTasksRows[i].ShardID = shardID\n\t\ttimerTasksRows[i].VisibilityTimestamp = task.GetVisibilityTimestamp()\n\t\ttimerTasksRows[i].TaskID = task.GetTaskID()\n\t\ttimerTasksRows[i].Data = blob.Data\n\t\ttimerTasksRows[i].DataEncoding = string(blob.Encoding)\n\t}\n\n\tresult, err := tx.InsertIntoTimerTasks(ctx, timerTasksRows)\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"createTimerTasks\", \"\", err)\n\t}\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createTimerTasks failed. Could not verify number of rows inserted. Error: %v\", err),\n\t\t}\n\t}\n\n\tif int(rowsAffected) != len(timerTasks) {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createTimerTasks failed. Inserted %v instead of %v rows into timer_tasks.\", rowsAffected, len(timerTasks)),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc assertNotCurrentExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\tcurrentRow, err := tx.LockCurrentExecutions(ctx, &sqlplugin.CurrentExecutionsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\t// allow bypassing no current record\n\t\t\treturn nil\n\t\t}\n\t\treturn convertCommonErrors(tx, \"assertCurrentExecution\", \"Unable to load current record.\", err)\n\t}\n\treturn assertRunIDMismatch(runID, currentRow.RunID)\n}\n\nfunc assertRunIDAndUpdateCurrentExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\tnewRunID serialization.UUID,\n\tpreviousRunID serialization.UUID,\n\tcreateRequestID string,\n\tstate int,\n\tcloseStatus int,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n) error {\n\n\tassertFn := func(currentRow *sqlplugin.CurrentExecutionsRow) error {\n\t\tif !bytes.Equal(currentRow.RunID, previousRunID) {\n\t\t\treturn &p.ConditionFailedError{Msg: fmt.Sprintf(\n\t\t\t\t\"assertRunIDAndUpdateCurrentExecution failed. Current run ID was %v, expected %v\",\n\t\t\t\tcurrentRow.RunID,\n\t\t\t\tpreviousRunID,\n\t\t\t)}\n\t\t}\n\t\treturn nil\n\t}\n\tif err := assertCurrentExecution(ctx, tx, shardID, domainID, workflowID, assertFn); err != nil {\n\t\treturn err\n\t}\n\n\treturn updateCurrentExecution(ctx, tx, shardID, domainID, workflowID, newRunID, createRequestID, state, closeStatus, startVersion, lastWriteVersion)\n}\n\nfunc assertCurrentExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\tassertFn func(currentRow *sqlplugin.CurrentExecutionsRow) error,\n) error {\n\n\tcurrentRow, err := tx.LockCurrentExecutions(ctx, &sqlplugin.CurrentExecutionsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"assertCurrentExecution\", \"Unable to load current record.\", err)\n\t}\n\treturn assertFn(currentRow)\n}\n\nfunc assertRunIDMismatch(runID serialization.UUID, currentRunID serialization.UUID) error {\n\t// zombie workflow creation with existence of current record, this is a noop\n\tif bytes.Equal(currentRunID, runID) {\n\t\treturn &p.ConditionFailedError{Msg: fmt.Sprintf(\n\t\t\t\"assertRunIDMismatch failed. Current run ID was %v, input %v\",\n\t\t\tcurrentRunID,\n\t\t\trunID,\n\t\t)}\n\t}\n\treturn nil\n}\n\nfunc updateCurrentExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tcreateRequestID string,\n\tstate int,\n\tcloseStatus int,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n) error {\n\n\tresult, err := tx.UpdateCurrentExecutions(ctx, &sqlplugin.CurrentExecutionsRow{\n\t\tShardID:          int64(shardID),\n\t\tDomainID:         domainID,\n\t\tWorkflowID:       workflowID,\n\t\tRunID:            runID,\n\t\tCreateRequestID:  createRequestID,\n\t\tState:            state,\n\t\tCloseStatus:      closeStatus,\n\t\tStartVersion:     startVersion,\n\t\tLastWriteVersion: lastWriteVersion,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"updateCurrentExecution\", \"\", err)\n\t}\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"updateCurrentExecution failed. Failed to check number of rows updated in current_executions table. Error: %v\", err),\n\t\t}\n\t}\n\tif rowsAffected != 1 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"updateCurrentExecution failed. %v rows of current_executions updated instead of 1.\", rowsAffected),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc buildExecutionRow(\n\texecutionInfo *p.InternalWorkflowExecutionInfo,\n\tversionHistories *p.DataBlob,\n\tworkflowChecksum *p.DataBlob,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n\tshardID int,\n\tparser serialization.Parser,\n) (row *sqlplugin.ExecutionsRow, err error) {\n\n\tinfo := serialization.FromInternalWorkflowExecutionInfo(executionInfo)\n\n\tinfo.StartVersion = startVersion\n\tif versionHistories == nil {\n\t\t// this is allowed\n\t} else {\n\t\tinfo.VersionHistories = versionHistories.Data\n\t\tinfo.VersionHistoriesEncoding = string(versionHistories.GetEncoding())\n\t}\n\tif workflowChecksum != nil {\n\t\tinfo.Checksum = workflowChecksum.Data\n\t\tinfo.ChecksumEncoding = string(workflowChecksum.GetEncoding())\n\t}\n\n\tblob, err := parser.WorkflowExecutionInfoToBlob(info)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &sqlplugin.ExecutionsRow{\n\t\tShardID:          shardID,\n\t\tDomainID:         serialization.MustParseUUID(executionInfo.DomainID),\n\t\tWorkflowID:       executionInfo.WorkflowID,\n\t\tRunID:            serialization.MustParseUUID(executionInfo.RunID),\n\t\tNextEventID:      int64(executionInfo.NextEventID),\n\t\tLastWriteVersion: lastWriteVersion,\n\t\tData:             blob.Data,\n\t\tDataEncoding:     string(blob.Encoding),\n\t}, nil\n}\n\nfunc createExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\texecutionInfo *p.InternalWorkflowExecutionInfo,\n\tversionHistories *p.DataBlob,\n\tworkflowChecksum *p.DataBlob,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n\tshardID int,\n\tparser serialization.Parser,\n) error {\n\n\t// validate workflow state & close status\n\tif err := p.ValidateCreateWorkflowStateCloseStatus(\n\t\texecutionInfo.State,\n\t\texecutionInfo.CloseStatus); err != nil {\n\t\treturn err\n\t}\n\n\tnow := time.Now()\n\t// TODO: this case seems to be always false\n\tif executionInfo.StartTimestamp.IsZero() {\n\t\texecutionInfo.StartTimestamp = now\n\t}\n\n\trow, err := buildExecutionRow(\n\t\texecutionInfo,\n\t\tversionHistories,\n\t\tworkflowChecksum,\n\t\tstartVersion,\n\t\tlastWriteVersion,\n\t\tshardID,\n\t\tparser,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresult, err := tx.InsertIntoExecutions(ctx, row)\n\tif err != nil {\n\t\tif tx.IsDupEntryError(err) {\n\t\t\treturn &p.WorkflowExecutionAlreadyStartedError{\n\t\t\t\tMsg:              fmt.Sprintf(\"Workflow execution already running. WorkflowId: %v\", executionInfo.WorkflowID),\n\t\t\t\tStartRequestID:   executionInfo.CreateRequestID,\n\t\t\t\tRunID:            executionInfo.RunID,\n\t\t\t\tState:            executionInfo.State,\n\t\t\t\tCloseStatus:      executionInfo.CloseStatus,\n\t\t\t\tLastWriteVersion: row.LastWriteVersion,\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(tx, \"createExecution\", \"\", err)\n\t}\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"createExecution failed. Failed to verify number of rows affected. Erorr: %v\", err),\n\t\t}\n\t}\n\tif rowsAffected != 1 {\n\t\treturn &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\"createExecution failed. Affected %v rows updated instead of 1.\", rowsAffected),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc updateExecution(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\texecutionInfo *p.InternalWorkflowExecutionInfo,\n\tversionHistories *p.DataBlob,\n\tworkflowChecksum *p.DataBlob,\n\tstartVersion int64,\n\tlastWriteVersion int64,\n\tshardID int,\n\tparser serialization.Parser,\n) error {\n\n\t// validate workflow state & close status\n\tif err := p.ValidateUpdateWorkflowStateCloseStatus(\n\t\texecutionInfo.State,\n\t\texecutionInfo.CloseStatus); err != nil {\n\t\treturn err\n\t}\n\n\trow, err := buildExecutionRow(\n\t\texecutionInfo,\n\t\tversionHistories,\n\t\tworkflowChecksum,\n\t\tstartVersion,\n\t\tlastWriteVersion,\n\t\tshardID,\n\t\tparser,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresult, err := tx.UpdateExecutions(ctx, row)\n\tif err != nil {\n\t\treturn convertCommonErrors(tx, \"updateExecution\", \"\", err)\n\t}\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"updateExecution failed. Failed to verify number of rows affected. Erorr: %v\", err),\n\t\t}\n\t}\n\tif rowsAffected != 1 {\n\t\treturn &types.EntityNotExistsError{\n\t\t\tMessage: fmt.Sprintf(\"updateExecution failed. Affected %v rows updated instead of 1.\", rowsAffected),\n\t\t}\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_execution_store_util_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc mockSetupLockAndCheckNextEventID(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tcondition int64,\n\twantErr bool,\n) {\n\tvar nextEventID int\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t} else {\n\t\tnextEventID = int(condition)\n\t}\n\tmockTx.EXPECT().WriteLockExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nextEventID, err)\n}\n\nfunc mockCreateExecution(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\tmockTx.EXPECT().InsertIntoExecutions(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 1}, err)\n}\n\nfunc mockUpdateExecution(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\tmockTx.EXPECT().UpdateExecutions(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 1}, err)\n}\n\nfunc mockCreateTransferTasks(\n\tmockTx *sqlplugin.MockTx,\n\tmockTaskSerializer *serialization.MockTaskSerializer,\n\ttasks int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(persistence.DataBlob{}, nil).Times(tasks)\n\tmockTx.EXPECT().InsertIntoTransferTasks(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: int64(tasks)}, err)\n}\n\nfunc mockCreateReplicationTasks(\n\tmockTx *sqlplugin.MockTx,\n\tmockTaskSerializer *serialization.MockTaskSerializer,\n\ttasks int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{}, nil).Times(tasks)\n\tmockTx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: int64(tasks)}, err)\n}\n\nfunc mockCreateTimerTasks(\n\tmockTx *sqlplugin.MockTx,\n\tmockTaskSerializer *serialization.MockTaskSerializer,\n\ttasks int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(persistence.DataBlob{}, nil).Times(tasks)\n\tmockTx.EXPECT().InsertIntoTimerTasks(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: int64(tasks)}, err)\n}\n\nfunc mockApplyTasks(\n\tmockTx *sqlplugin.MockTx,\n\tmockTaskSerializer *serialization.MockTaskSerializer,\n\ttransfer int,\n\ttimer int,\n\treplication int,\n\twantErr bool,\n) {\n\tmockCreateTransferTasks(mockTx, mockTaskSerializer, transfer, wantErr)\n\tif wantErr {\n\t\treturn\n\t}\n\tmockCreateTimerTasks(mockTx, mockTaskSerializer, timer, wantErr)\n\tmockCreateReplicationTasks(mockTx, mockTaskSerializer, replication, wantErr)\n}\n\nfunc mockUpdateActivityInfos(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\tactivityInfos int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().ActivityInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil).Times(activityInfos)\n\tif activityInfos > 0 {\n\t\tmockTx.EXPECT().ReplaceIntoActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromActivityInfoMaps(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockUpdateTimerInfos(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\ttimerInfos int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().TimerInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil).Times(timerInfos)\n\tif timerInfos > 0 {\n\t\tmockTx.EXPECT().ReplaceIntoTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromTimerInfoMaps(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockUpdateChildExecutionInfos(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\tchildExecutionInfos int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().ChildExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil).Times(childExecutionInfos)\n\tif childExecutionInfos > 0 {\n\t\tmockTx.EXPECT().ReplaceIntoChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromChildExecutionInfoMaps(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockUpdateRequestCancelInfos(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\tcancelInfos int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().RequestCancelInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil).Times(cancelInfos)\n\tif cancelInfos > 0 {\n\t\tmockTx.EXPECT().ReplaceIntoRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromRequestCancelInfoMaps(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockUpdateSignalInfos(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\tsignalInfos int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockParser.EXPECT().SignalInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil).Times(signalInfos)\n\tif signalInfos > 0 {\n\t\tmockTx.EXPECT().ReplaceIntoSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromSignalInfoMaps(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockUpdateSignalRequested(\n\tmockTx *sqlplugin.MockTx,\n\tmockParser *serialization.MockParser,\n\tsignalRequested int,\n\tdeleteInfos int,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tif signalRequested > 0 {\n\t\tmockTx.EXPECT().InsertIntoSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t}\n\tif deleteInfos > 0 {\n\t\tmockTx.EXPECT().DeleteFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, err)\n\t}\n}\n\nfunc mockDeleteActivityInfoMap(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromActivityInfoMaps(gomock.Any(), &sqlplugin.ActivityInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteTimerInfoMap(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromTimerInfoMaps(gomock.Any(), &sqlplugin.TimerInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteChildExecutionInfoMap(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromChildExecutionInfoMaps(gomock.Any(), &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteRequestCancelInfoMap(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromRequestCancelInfoMaps(gomock.Any(), &sqlplugin.RequestCancelInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteSignalInfoMap(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromSignalInfoMaps(gomock.Any(), &sqlplugin.SignalInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteSignalRequestedSet(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromSignalsRequestedSets(gomock.Any(), &sqlplugin.SignalsRequestedSetsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc mockDeleteBufferedEvents(\n\tmockTx *sqlplugin.MockTx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\twantErr bool,\n) {\n\tvar err error\n\tif wantErr {\n\t\terr = errors.New(\"some error\")\n\t}\n\tmockTx.EXPECT().DeleteFromBufferedEvents(gomock.Any(), &sqlplugin.BufferedEventsFilter{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).Return(nil, err)\n\tif wantErr {\n\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t}\n}\n\nfunc TestApplyWorkflowMutationTx(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tworkflow  *persistence.InternalWorkflowMutation\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflow: &persistence.InternalWorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t},\n\t\t\t\tCondition: 9,\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer: []persistence.Task{\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tUpsertActivityInfos: []*persistence.InternalActivityInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tDeleteActivityInfos: []int64{1, 2},\n\t\t\t\tUpsertTimerInfos: []*persistence.TimerInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tDeleteTimerInfos: []string{\"a\", \"b\"},\n\t\t\t\tUpsertChildExecutionInfos: []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tDeleteChildExecutionInfos: []int64{1, 2},\n\t\t\t\tUpsertRequestCancelInfos: []*persistence.RequestCancelInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tDeleteRequestCancelInfos: []int64{1, 2},\n\t\t\t\tUpsertSignalInfos: []*persistence.SignalInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tDeleteSignalInfos:        []int64{1, 2},\n\t\t\t\tUpsertSignalRequestedIDs: []string{\"a\", \"b\"},\n\t\t\t\tDeleteSignalRequestedIDs: []string{\"c\", \"d\"},\n\t\t\t\tClearBufferedEvents:      true,\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockSetupLockAndCheckNextEventID(mockTx, shardID, serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"), \"abc\", serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"), 9, false)\n\t\t\t\tmockUpdateExecution(mockTx, mockParser, false)\n\t\t\t\tmockApplyTasks(mockTx, mockTaskSerializer, 1, 3, 4, false)\n\t\t\t\tmockUpdateActivityInfos(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockUpdateTimerInfos(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockUpdateChildExecutionInfos(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockUpdateRequestCancelInfos(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockUpdateSignalInfos(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockUpdateSignalRequested(mockTx, mockParser, 1, 2, false)\n\t\t\t\tmockDeleteBufferedEvents(mockTx, shardID, serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"), \"abc\", serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"), false)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockParser, mockTaskSerializer)\n\n\t\t\terr := applyWorkflowMutationTx(context.Background(), mockTx, shardID, tc.workflow, mockParser, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestApplyWorkflowSnapshotTxAsReset(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tworkflow  *persistence.InternalWorkflowSnapshot\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t},\n\t\t\t\tCondition: 9,\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer: []persistence.Task{\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityInfos: []*persistence.InternalActivityInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tTimerInfos: []*persistence.TimerInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tChildExecutionInfos: []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tRequestCancelInfos: []*persistence.RequestCancelInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tSignalInfos: []*persistence.SignalInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tSignalRequestedIDs: []string{\"a\", \"b\"},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tdomainID := serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\")\n\t\t\t\tworkflowID := \"abc\"\n\t\t\t\trunID := serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\")\n\t\t\t\tmockSetupLockAndCheckNextEventID(mockTx, shardID, domainID, workflowID, runID, 9, false)\n\t\t\t\tmockUpdateExecution(mockTx, mockParser, false)\n\t\t\t\tmockApplyTasks(mockTx, mockTaskSerializer, 1, 3, 4, false)\n\t\t\t\tmockDeleteActivityInfoMap(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateActivityInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteTimerInfoMap(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateTimerInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteChildExecutionInfoMap(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateChildExecutionInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteRequestCancelInfoMap(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateRequestCancelInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteSignalInfoMap(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateSignalInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteSignalRequestedSet(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t\tmockUpdateSignalRequested(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockDeleteBufferedEvents(mockTx, shardID, domainID, workflowID, runID, false)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockParser, mockTaskSerializer)\n\n\t\t\terr := applyWorkflowSnapshotTxAsReset(context.Background(), mockTx, shardID, tc.workflow, mockParser, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestApplyWorkflowSnapshotTxAsNew(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tworkflow  *persistence.InternalWorkflowSnapshot\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t},\n\t\t\t\tCondition: 9,\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer: []persistence.Task{\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityInfos: []*persistence.InternalActivityInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tTimerInfos: []*persistence.TimerInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tChildExecutionInfos: []*persistence.InternalChildExecutionInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tRequestCancelInfos: []*persistence.RequestCancelInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tSignalInfos: []*persistence.SignalInfo{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t\tSignalRequestedIDs: []string{\"a\", \"b\"},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockCreateExecution(mockTx, mockParser, false)\n\t\t\t\tmockApplyTasks(mockTx, mockTaskSerializer, 1, 3, 4, false)\n\t\t\t\tmockUpdateActivityInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockUpdateTimerInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockUpdateChildExecutionInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockUpdateRequestCancelInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockUpdateSignalInfos(mockTx, mockParser, 1, 0, false)\n\t\t\t\tmockUpdateSignalRequested(mockTx, mockParser, 1, 0, false)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockParser, mockTaskSerializer)\n\n\t\t\terr := applyWorkflowSnapshotTxAsNew(context.Background(), mockTx, shardID, tc.workflow, mockParser, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLockAndCheckNextEventID(t *testing.T) {\n\tshardID := 1\n\tdomainID := serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\")\n\tworkflowID := \"abc\"\n\trunID := serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\")\n\ttestCases := []struct {\n\t\tname      string\n\t\tcondition int64\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tcondition: 10,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t}).Return(10, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - entity not exists\",\n\t\t\tcondition: 10,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t}).Return(0, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.EntityNotExistsError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be EntityNotExistsError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - condition failed\",\n\t\t\tcondition: 10,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockExecutions(gomock.Any(), &sqlplugin.ExecutionsFilter{\n\t\t\t\t\tShardID:    shardID,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t}).Return(11, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ConditionFailedError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ConditionFailedError\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\n\t\t\terr := lockAndCheckNextEventID(context.Background(), mockTx, shardID, domainID, workflowID, runID, tc.condition)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateExecution(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tworkflow  *persistence.InternalWorkflowSnapshot\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID:  \"abc\",\n\t\t\t\t\tRunID:       \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t\tNextEventID: 9,\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tStartVersion:     1,\n\t\t\t\tLastWriteVersion: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`workflow`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"workflow\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoExecutions(gomock.Any(), &sqlplugin.ExecutionsRow{\n\t\t\t\t\tShardID:          shardID,\n\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID:       \"abc\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\tNextEventID:      9,\n\t\t\t\t\tLastWriteVersion: 2,\n\t\t\t\t\tData:             []byte(`workflow`),\n\t\t\t\t\tDataEncoding:     \"workflow\",\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - already started\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID:  \"abc\",\n\t\t\t\t\tRunID:       \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t\tNextEventID: 9,\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tStartVersion:     1,\n\t\t\t\tLastWriteVersion: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`workflow`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"workflow\"),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoExecutions(gomock.Any(), &sqlplugin.ExecutionsRow{\n\t\t\t\t\tShardID:          shardID,\n\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID:       \"abc\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\tNextEventID:      9,\n\t\t\t\t\tLastWriteVersion: 2,\n\t\t\t\t\tData:             []byte(`workflow`),\n\t\t\t\t\tDataEncoding:     \"workflow\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsDupEntryError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.WorkflowExecutionAlreadyStartedError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be WorkflowExecutionAlreadyStartedError\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockParser)\n\n\t\t\terr := createExecution(context.Background(), mockTx, tc.workflow.ExecutionInfo, tc.workflow.VersionHistories, tc.workflow.ChecksumData, tc.workflow.StartVersion, tc.workflow.LastWriteVersion, shardID, mockParser)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateExecution(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tworkflow  *persistence.InternalWorkflowSnapshot\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID:  \"abc\",\n\t\t\t\t\tRunID:       \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t\tNextEventID: 9,\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tStartVersion:     1,\n\t\t\t\tLastWriteVersion: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`workflow`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"workflow\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().UpdateExecutions(gomock.Any(), &sqlplugin.ExecutionsRow{\n\t\t\t\t\tShardID:          shardID,\n\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID:       \"abc\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\tNextEventID:      9,\n\t\t\t\t\tLastWriteVersion: 2,\n\t\t\t\t\tData:             []byte(`workflow`),\n\t\t\t\t\tDataEncoding:     \"workflow\",\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - already started\",\n\t\t\tworkflow: &persistence.InternalWorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.InternalWorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    \"8be8a310-7d20-483e-a5d2-48659dc47602\",\n\t\t\t\t\tWorkflowID:  \"abc\",\n\t\t\t\t\tRunID:       \"8be8a310-7d20-483e-a5d2-48659dc47603\",\n\t\t\t\t\tNextEventID: 9,\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.DataBlob{},\n\t\t\t\tStartVersion:     1,\n\t\t\t\tLastWriteVersion: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().WorkflowExecutionInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`workflow`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"workflow\"),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateExecutions(gomock.Any(), &sqlplugin.ExecutionsRow{\n\t\t\t\t\tShardID:          shardID,\n\t\t\t\t\tDomainID:         serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID:       \"abc\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\tNextEventID:      9,\n\t\t\t\t\tLastWriteVersion: 2,\n\t\t\t\t\tData:             []byte(`workflow`),\n\t\t\t\t\tDataEncoding:     \"workflow\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockParser)\n\n\t\t\terr := updateExecution(context.Background(), mockTx, tc.workflow.ExecutionInfo, tc.workflow.VersionHistories, tc.workflow.ChecksumData, tc.workflow.StartVersion, tc.workflow.LastWriteVersion, shardID, mockParser)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateTransferTasks(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\ttasks     []persistence.Task\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"8be8a310-7d20-483e-a5d2-48659dc47609\",\n\t\t\t\t\tTaskList:       \"tl\",\n\t\t\t\t\tScheduleID:     111,\n\t\t\t\t},\n\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             2,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(2, 2),\n\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"7be8a310-7d20-483e-a5d2-48659dc47609\",\n\t\t\t\t\tTaskList:       \"tl2\",\n\t\t\t\t\tScheduleID:     222,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`1`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"1\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`2`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"2\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoTransferTasks(gomock.Any(), []sqlplugin.TransferTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(`1`),\n\t\t\t\t\t\tDataEncoding: \"1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       2,\n\t\t\t\t\t\tData:         []byte(`2`),\n\t\t\t\t\t\tDataEncoding: \"2\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 2}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: \"8be8a310-7d20-483e-a5d2-48659dc47609\",\n\t\t\t\t\tTaskList:       \"tl\",\n\t\t\t\t\tScheduleID:     111,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`1`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"1\"),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoTransferTasks(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockTaskSerializer)\n\n\t\t\terr := createTransferTasks(context.Background(), mockTx, tc.tasks, shardID, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateTimerTasks(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\ttasks     []persistence.Task\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTimeoutTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t},\n\t\t\t\t\tEventID:         1,\n\t\t\t\t\tScheduleAttempt: 1,\n\t\t\t\t\tTimeoutType:     1,\n\t\t\t\t},\n\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             2,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(2, 2),\n\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t},\n\t\t\t\t\tEventID:     2,\n\t\t\t\t\tAttempt:     2,\n\t\t\t\t\tTimeoutType: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`1`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"1\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`2`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"2\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoTimerTasks(gomock.Any(), []sqlplugin.TimerTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tData:                []byte(`1`),\n\t\t\t\t\t\tDataEncoding:        \"1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(2, 2),\n\t\t\t\t\t\tData:                []byte(`2`),\n\t\t\t\t\t\tDataEncoding:        \"2\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 2}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.DecisionTimeoutTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t},\n\t\t\t\t\tEventID:         1,\n\t\t\t\t\tScheduleAttempt: 1,\n\t\t\t\t\tTimeoutType:     1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`1`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"1\"),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoTimerTasks(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockTaskSerializer)\n\n\t\t\terr := createTimerTasks(context.Background(), mockTx, tc.tasks, shardID, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateReplicationTasks(t *testing.T) {\n\tshardID := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\ttasks     []persistence.Task\n\t\tmockSetup func(*sqlplugin.MockTx, *serialization.MockTaskSerializer)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(1, 1),\n\t\t\t\t\t\tVersion:             1,\n\t\t\t\t\t},\n\t\t\t\t\tFirstEventID:      1,\n\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\tBranchToken:       []byte{1},\n\t\t\t\t\tNewRunBranchToken: []byte{2},\n\t\t\t\t},\n\t\t\t\t&persistence.SyncActivityTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(2, 2),\n\t\t\t\t\t\tVersion:             2,\n\t\t\t\t\t},\n\t\t\t\t\tScheduledID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx, mockTaskSerializer *serialization.MockTaskSerializer) {\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`1`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"1\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTaskSerializer.EXPECT().SerializeTask(persistence.HistoryTaskCategoryReplication, gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`2`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"2\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoReplicationTasks(gomock.Any(), []sqlplugin.ReplicationTasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       1,\n\t\t\t\t\t\tData:         []byte(`1`),\n\t\t\t\t\t\tDataEncoding: \"1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tTaskID:       2,\n\t\t\t\t\t\tData:         []byte(`2`),\n\t\t\t\t\t\tDataEncoding: \"2\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 2}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockTaskSerializer := serialization.NewMockTaskSerializer(ctrl)\n\n\t\t\ttc.mockSetup(mockTx, mockTaskSerializer)\n\n\t\t\terr := createReplicationTasks(context.Background(), mockTx, tc.tasks, shardID, mockTaskSerializer)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLockCurrentExecutionIfExists(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t\twant      *sqlplugin.CurrentExecutionsRow\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutionsJoinExecutions(gomock.Any(), gomock.Any()).Return([]sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    1,\n\t\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &sqlplugin.CurrentExecutionsRow{\n\t\t\t\tShardID:    1,\n\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutionsJoinExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Empty result\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutionsJoinExecutions(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple rows\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutionsJoinExecutions(gomock.Any(), gomock.Any()).Return([]sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    1,\n\t\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    1,\n\t\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\t\tWorkflowID: \"def\",\n\t\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47604\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\n\t\t\tgot, err := lockCurrentExecutionIfExists(context.Background(), mockTx, 1, serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"), \"abc\")\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Expected result to match\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateOrUpdateCurrentExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tcreateMode persistence.CreateWorkflowMode\n\t\tmockSetup  func(*sqlplugin.MockTx)\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname:       \"Brand new workflow - success\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeBrandNew,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().InsertIntoCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:       \"Brand new workflow - error\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeBrandNew,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"Update current execution - success\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().UpdateCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:       \"Update current execution - error\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeWorkflowIDReuse,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"Update current execution - no rows affected\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeContinueAsNew,\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().UpdateCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 0}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"Zombie workflow - success\",\n\t\t\tcreateMode: persistence.CreateWorkflowModeZombie,\n\t\t\tmockSetup:  func(mockTx *sqlplugin.MockTx) {},\n\t\t\twantErr:    false,\n\t\t},\n\t\t{\n\t\t\tname:       \"Unknown create mode\",\n\t\t\tcreateMode: persistence.CreateWorkflowMode(100),\n\t\t\tmockSetup:  func(mockTx *sqlplugin.MockTx) {},\n\t\t\twantErr:    true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\n\t\t\terr := createOrUpdateCurrentExecution(\n\t\t\t\tcontext.Background(),\n\t\t\t\tmockTx,\n\t\t\t\ttc.createMode,\n\t\t\t\t1,\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\"abc\",\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t0,\n\t\t\t\t1,\n\t\t\t\t\"request-id\",\n\t\t\t\t11,\n\t\t\t\t12,\n\t\t\t)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAssertNotCurrentExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:    1,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - No rows\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - run ID match\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:    1,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\n\t\t\terr := assertNotCurrentExecution(\n\t\t\t\tcontext.Background(),\n\t\t\t\tmockTx,\n\t\t\t\t1,\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\"abc\",\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAssertRunIDAndUpdateCurrentExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:    1,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47604\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockTx.EXPECT().UpdateCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - update current execution\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:    1,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47604\"),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - run ID mismatch\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(&sqlplugin.CurrentExecutionsRow{\n\t\t\t\t\tShardID:    1,\n\t\t\t\t\tDomainID:   serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\tWorkflowID: \"abc\",\n\t\t\t\t\tRunID:      serialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - unknown error\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockCurrentExecutions(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\n\t\t\terr := assertRunIDAndUpdateCurrentExecution(\n\t\t\t\tcontext.Background(),\n\t\t\t\tmockTx,\n\t\t\t\t1,\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47602\"),\n\t\t\t\t\"abc\",\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47603\"),\n\t\t\t\tserialization.MustParseUUID(\"8be8a310-7d20-483e-a5d2-48659dc47604\"),\n\t\t\t\t\"request-id\",\n\t\t\t\t1,\n\t\t\t\t11,\n\t\t\t\t12,\n\t\t\t\t13,\n\t\t\t)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_history_store.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t_defaultHistoryNodeDeleteBatch = 1000\n)\n\ntype sqlHistoryStore struct {\n\tsqlStore\n}\n\ntype historyTreePageToken struct {\n\tShardID  int\n\tTreeID   serialization.UUID\n\tBranchID serialization.UUID\n}\n\n// NewHistoryV2Persistence creates an instance of HistoryManager\nfunc NewHistoryV2Persistence(\n\tdb sqlplugin.DB,\n\tlogger log.Logger,\n\tparser serialization.Parser,\n\tdc *persistence.DynamicConfiguration,\n) (persistence.HistoryStore, error) {\n\n\treturn &sqlHistoryStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t\tparser: parser,\n\t\t\tdc:     dc,\n\t\t},\n\t}, nil\n}\n\n// AppendHistoryNodes add(or override) a node to a history branch\nfunc (m *sqlHistoryStore) AppendHistoryNodes(\n\tctx context.Context,\n\trequest *persistence.InternalAppendHistoryNodesRequest,\n) error {\n\n\tbranchInfo := request.BranchInfo\n\tbeginNodeID := persistenceutils.GetBeginNodeID(branchInfo)\n\n\tif request.NodeID < beginNodeID {\n\t\treturn &persistence.InvalidPersistenceRequestError{\n\t\t\tMsg: \"cannot append to ancestors' nodes\",\n\t\t}\n\t}\n\n\tnodeRow := &sqlplugin.HistoryNodeRow{\n\t\tTreeID:       serialization.MustParseUUID(branchInfo.TreeID),\n\t\tBranchID:     serialization.MustParseUUID(branchInfo.BranchID),\n\t\tNodeID:       request.NodeID,\n\t\tTxnID:        &request.TransactionID,\n\t\tData:         request.Events.Data,\n\t\tDataEncoding: string(request.Events.Encoding),\n\t\tShardID:      request.ShardID,\n\t}\n\n\tif request.IsNewBranch {\n\t\tvar ancestors []*types.HistoryBranchRange\n\t\tancestors = append(ancestors, branchInfo.Ancestors...)\n\n\t\ttreeInfo := &serialization.HistoryTreeInfo{\n\t\t\tAncestors:        ancestors,\n\t\t\tInfo:             request.Info,\n\t\t\tCreatedTimestamp: request.CurrentTimeStamp,\n\t\t}\n\n\t\tblob, err := m.parser.HistoryTreeInfoToBlob(treeInfo)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\ttreeRow := &sqlplugin.HistoryTreeRow{\n\t\t\tShardID:      request.ShardID,\n\t\t\tTreeID:       serialization.MustParseUUID(branchInfo.TreeID),\n\t\t\tBranchID:     serialization.MustParseUUID(branchInfo.BranchID),\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t}\n\n\t\ttreeUUID := serialization.MustParseUUID(branchInfo.TreeID)\n\t\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(treeUUID, m.db.GetTotalNumDBShards())\n\t\treturn m.txExecute(ctx, dbShardID, \"AppendHistoryNodes\", func(tx sqlplugin.Tx) error {\n\t\t\tresult, err := tx.InsertIntoHistoryNode(ctx, nodeRow)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trowsAffected, err := result.RowsAffected()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif rowsAffected != 1 {\n\t\t\t\treturn fmt.Errorf(\"expected 1 row to be affected for node table, got %v\", rowsAffected)\n\t\t\t}\n\t\t\tresult, err = tx.InsertIntoHistoryTree(ctx, treeRow)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trowsAffected, err = result.RowsAffected()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tif rowsAffected != 1 {\n\t\t\t\treturn fmt.Errorf(\"expected 1 row to be affected for tree table, got %v\", rowsAffected)\n\t\t\t}\n\t\t\treturn nil\n\t\t})\n\t}\n\n\t_, err := m.db.InsertIntoHistoryNode(ctx, nodeRow)\n\tif err != nil {\n\t\tif m.db.IsDupEntryError(err) {\n\t\t\treturn &persistence.ConditionFailedError{Msg: fmt.Sprintf(\"AppendHistoryNodes: row already exist: %v\", err)}\n\t\t}\n\t\treturn convertCommonErrors(m.db, \"AppendHistoryEvents\", \"\", err)\n\t}\n\treturn nil\n}\n\n// ReadHistoryBranch returns history node data for a branch\nfunc (m *sqlHistoryStore) ReadHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalReadHistoryBranchRequest,\n) (*persistence.InternalReadHistoryBranchResponse, error) {\n\n\tminNodeID := request.MinNodeID\n\tmaxNodeID := request.MaxNodeID\n\n\tlastNodeID := request.LastNodeID\n\tlastTxnID := request.LastTransactionID\n\n\tif request.NextPageToken != nil && len(request.NextPageToken) > 0 {\n\t\tvar lastNodeID int64\n\t\tvar err error\n\t\t// TODO the inner pagination token can be replaced by a dummy token\n\t\t//  since lastNodeID & lastTxnID are both provided\n\t\tif lastNodeID, err = deserializePageToken(request.NextPageToken); err != nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"invalid next page token %v\", request.NextPageToken)}\n\t\t}\n\t\tminNodeID = lastNodeID + 1\n\t}\n\n\tfilter := &sqlplugin.HistoryNodeFilter{\n\t\tTreeID:    serialization.MustParseUUID(request.TreeID),\n\t\tBranchID:  serialization.MustParseUUID(request.BranchID),\n\t\tMinNodeID: &minNodeID,\n\t\tMaxNodeID: &maxNodeID,\n\t\tPageSize:  request.PageSize,\n\t\tShardID:   request.ShardID,\n\t}\n\n\trows, err := m.db.SelectFromHistoryNode(ctx, filter)\n\tif err == sql.ErrNoRows || (err == nil && len(rows) == 0) {\n\t\treturn &persistence.InternalReadHistoryBranchResponse{}, nil\n\t}\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"ReadHistoryBranch\", \"\", err)\n\t}\n\n\thistory := make([]*persistence.DataBlob, 0, int(request.PageSize))\n\teventBlob := &persistence.DataBlob{}\n\n\tfor _, row := range rows {\n\t\teventBlob.Data = row.Data\n\t\teventBlob.Encoding = constants.EncodingType(row.DataEncoding)\n\n\t\tif *row.TxnID < lastTxnID {\n\t\t\t// assuming that business logic layer is correct and transaction ID only increase\n\t\t\t// thus, valid event batch will come with increasing transaction ID\n\n\t\t\t// event batches with smaller node ID\n\t\t\t//  -> should not be possible since records are already sorted\n\t\t\t// event batches with same node ID\n\t\t\t//  -> batch with higher transaction ID is valid\n\t\t\t// event batches with larger node ID\n\t\t\t//  -> batch with lower transaction ID is invalid (happens before)\n\t\t\t//  -> batch with higher transaction ID is valid\n\t\t\tif row.NodeID < lastNodeID {\n\t\t\t\treturn nil, &types.InternalDataInconsistencyError{\n\t\t\t\t\tMessage: \"corrupted data, nodeID cannot decrease\",\n\t\t\t\t}\n\t\t\t} else if row.NodeID > lastNodeID {\n\t\t\t\t// update lastNodeID so that our pagination can make progress in the corner case that\n\t\t\t\t// the page are all rows with smaller txnID\n\t\t\t\t// because next page we always have minNodeID = lastNodeID+1\n\t\t\t\tlastNodeID = row.NodeID\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\n\t\tswitch {\n\t\tcase row.NodeID < lastNodeID:\n\t\t\treturn nil, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted data, nodeID cannot decrease\",\n\t\t\t}\n\t\tcase row.NodeID == lastNodeID:\n\t\t\treturn nil, &types.InternalDataInconsistencyError{\n\t\t\t\tMessage: \"corrupted data, same nodeID must have smaller txnID\",\n\t\t\t}\n\t\tdefault: // row.NodeID > lastNodeID:\n\t\t\t// NOTE: when row.nodeID > lastNodeID, we expect the one with largest txnID comes first\n\t\t\tlastTxnID = *row.TxnID\n\t\t\tlastNodeID = row.NodeID\n\t\t\thistory = append(history, eventBlob)\n\t\t\teventBlob = &persistence.DataBlob{}\n\t\t}\n\t}\n\n\tvar pagingToken []byte\n\tif len(rows) >= request.PageSize {\n\t\tpagingToken = serializePageToken(lastNodeID)\n\t}\n\n\treturn &persistence.InternalReadHistoryBranchResponse{\n\t\tHistory:           history,\n\t\tNextPageToken:     pagingToken,\n\t\tLastNodeID:        lastNodeID,\n\t\tLastTransactionID: lastTxnID,\n\t}, nil\n}\n\n// ForkHistoryBranch forks a new branch from an existing branch\n// Note that application must provide a void forking nodeID, it must be a valid nodeID in that branch.\n// A valid forking nodeID can be an ancestor from the existing branch.\n// For example, we have branch B1 with three nodes(1[1,2], 3[3,4,5] and 6[6,7,8]. 1, 3 and 6 are nodeIDs (first eventID of the batch).\n// So B1 looks like this:\n//\n//\t     1[1,2]\n//\t     /\n//\t   3[3,4,5]\n//\t  /\n//\t6[6,7,8]\n//\n// Assuming we have branch B2 which contains one ancestor B1 stopping at 6 (exclusive). So B2 inherit nodeID 1 and 3 from B1, and have its own nodeID 6 and 8.\n// Branch B2 looks like this:\n//\n//\t  1[1,2]\n//\t  /\n//\t3[3,4,5]\n//\t \\\n//\t  6[6,7]\n//\t  \\\n//\t   8[8]\n//\n// Now we want to fork a new branch B3 from B2.\n// The only valid forking nodeIDs are 3,6 or 8.\n// 1 is not valid because we can't fork from first node.\n// 2/4/5 is NOT valid either because they are inside a batch.\n//\n// Case #1: If we fork from nodeID 6, then B3 will have an ancestor B1 which stops at 6(exclusive).\n// As we append a batch of events[6,7,8,9] to B3, it will look like :\n//\n//\t  1[1,2]\n//\t  /\n//\t3[3,4,5]\n//\t \\\n//\t6[6,7,8,9]\n//\n// Case #2: If we fork from node 8, then B3 will have two ancestors: B1 stops at 6(exclusive) and ancestor B2 stops at 8(exclusive)\n// As we append a batch of events[8,9] to B3, it will look like:\n//\n//\t     1[1,2]\n//\t     /\n//\t   3[3,4,5]\n//\t  /\n//\t6[6,7]\n//\t \\\n//\t 8[8,9]\nfunc (m *sqlHistoryStore) ForkHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalForkHistoryBranchRequest,\n) (*persistence.InternalForkHistoryBranchResponse, error) {\n\n\tforkB := request.ForkBranchInfo\n\ttreeID := forkB.TreeID\n\tnewAncestors := make([]*types.HistoryBranchRange, 0, len(forkB.Ancestors)+1)\n\n\tbeginNodeID := persistenceutils.GetBeginNodeID(forkB)\n\tif beginNodeID >= request.ForkNodeID {\n\t\t// this is the case that new branch's ancestors doesn't include the forking branch\n\t\tfor _, br := range forkB.Ancestors {\n\t\t\tif br.EndNodeID >= request.ForkNodeID {\n\t\t\t\tnewAncestors = append(newAncestors, &types.HistoryBranchRange{\n\t\t\t\t\tBranchID:    br.BranchID,\n\t\t\t\t\tBeginNodeID: br.BeginNodeID,\n\t\t\t\t\tEndNodeID:   request.ForkNodeID,\n\t\t\t\t})\n\t\t\t\tbreak\n\t\t\t} else {\n\t\t\t\tnewAncestors = append(newAncestors, br)\n\t\t\t}\n\t\t}\n\t} else {\n\t\t// this is the case the new branch will inherit all ancestors from forking branch\n\t\tnewAncestors = forkB.Ancestors\n\t\tnewAncestors = append(newAncestors, &types.HistoryBranchRange{\n\t\t\tBranchID:    forkB.BranchID,\n\t\t\tBeginNodeID: beginNodeID,\n\t\t\tEndNodeID:   request.ForkNodeID,\n\t\t})\n\t}\n\n\tresp := &persistence.InternalForkHistoryBranchResponse{\n\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:    treeID,\n\t\t\tBranchID:  request.NewBranchID,\n\t\t\tAncestors: newAncestors,\n\t\t}}\n\n\ttreeInfo := &serialization.HistoryTreeInfo{\n\t\tAncestors:        newAncestors,\n\t\tInfo:             request.Info,\n\t\tCreatedTimestamp: request.CurrentTimeStamp,\n\t}\n\n\tblob, err := m.parser.HistoryTreeInfoToBlob(treeInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trow := &sqlplugin.HistoryTreeRow{\n\t\tShardID:      request.ShardID,\n\t\tTreeID:       serialization.MustParseUUID(treeID),\n\t\tBranchID:     serialization.MustParseUUID(request.NewBranchID),\n\t\tData:         blob.Data,\n\t\tDataEncoding: string(blob.Encoding),\n\t}\n\tresult, err := m.db.InsertIntoHistoryTree(ctx, row)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"ForkHistoryBranch\", \"\", err)\n\t}\n\trowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif rowsAffected != 1 {\n\t\treturn nil, types.InternalServiceError{Message: fmt.Sprintf(\"expected 1 row to be affected for tree table, got %v\", rowsAffected)}\n\t}\n\treturn resp, nil\n}\n\n// DeleteHistoryBranch removes a branch\nfunc (m *sqlHistoryStore) DeleteHistoryBranch(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteHistoryBranchRequest,\n) error {\n\n\tbranch := request.BranchInfo\n\ttreeID := branch.TreeID\n\tbrsToDelete := branch.Ancestors\n\tbeginNodeID := persistenceutils.GetBeginNodeID(branch)\n\tbrsToDelete = append(brsToDelete, &types.HistoryBranchRange{\n\t\tBranchID:    branch.BranchID,\n\t\tBeginNodeID: beginNodeID,\n\t})\n\n\trsp, err := m.GetHistoryTree(ctx, &persistence.InternalGetHistoryTreeRequest{\n\t\tTreeID:  treeID,\n\t\tShardID: common.IntPtr(request.ShardID),\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// validBRsMaxEndNode is to for each branch range that is being used, we want to know what is the max nodeID referred by other valid branch\n\tvalidBRsMaxEndNode := persistenceutils.GetBranchesMaxReferredNodeIDs(rsp.Branches)\n\n\ttreeUUID := serialization.MustParseUUID(treeID)\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(treeUUID, m.db.GetTotalNumDBShards())\n\treturn m.txExecute(ctx, dbShardID, \"DeleteHistoryBranch\", func(tx sqlplugin.Tx) error {\n\t\tbranchID := serialization.MustParseUUID(branch.BranchID)\n\t\ttreeFilter := &sqlplugin.HistoryTreeFilter{\n\t\t\tTreeID:   treeUUID,\n\t\t\tBranchID: &branchID,\n\t\t\tShardID:  request.ShardID,\n\t\t}\n\t\t_, err = tx.DeleteFromHistoryTree(ctx, treeFilter)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tdone := false\n\t\t// for each branch range to delete, we iterate from bottom to up, and delete up to the point according to validBRsEndNode\n\t\tfor i := len(brsToDelete) - 1; i >= 0; i-- {\n\t\t\tbr := brsToDelete[i]\n\t\t\tmaxReferredEndNodeID, ok := validBRsMaxEndNode[br.BranchID]\n\t\t\tbatchSize := _defaultHistoryNodeDeleteBatch\n\t\t\tif m.dc != nil {\n\t\t\t\tbatchSize = m.dc.HistoryNodeDeleteBatchSize()\n\t\t\t}\n\t\t\tnodeFilter := &sqlplugin.HistoryNodeFilter{\n\t\t\t\tTreeID:   serialization.MustParseUUID(treeID),\n\t\t\t\tBranchID: serialization.MustParseUUID(br.BranchID),\n\t\t\t\tShardID:  request.ShardID,\n\t\t\t\tPageSize: batchSize,\n\t\t\t}\n\n\t\t\tif ok {\n\t\t\t\t// we can only delete from the maxEndNode and stop here\n\t\t\t\tnodeFilter.MinNodeID = &maxReferredEndNodeID\n\t\t\t\tdone = true\n\t\t\t} else {\n\t\t\t\t// No any branch is using this range, we can delete all of it\n\t\t\t\tnodeFilter.MinNodeID = &br.BeginNodeID\n\t\t\t}\n\t\t\tfor {\n\t\t\t\tresult, err := tx.DeleteFromHistoryNode(ctx, nodeFilter)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\trowsAffected, err := result.RowsAffected()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tif batchSize <= 0 ||\n\t\t\t\t\trowsAffected < int64(batchSize) ||\n\t\t\t\t\trowsAffected == persistence.UnknownNumRowsAffected ||\n\t\t\t\t\trowsAffected > int64(batchSize) {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif done {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t})\n}\n\n// TODO: Limit the underlying query to a specific shard at a time. See https://github.com/uber/cadence/issues/4064\nfunc (m *sqlHistoryStore) GetAllHistoryTreeBranches(\n\tctx context.Context,\n\trequest *persistence.GetAllHistoryTreeBranchesRequest,\n) (*persistence.GetAllHistoryTreeBranchesResponse, error) {\n\tpage := historyTreePageToken{}\n\tif request.NextPageToken != nil {\n\t\tif err := gobDeserialize(request.NextPageToken, &page); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"unable to decode next page token\")\n\t\t}\n\t} else {\n\t\tpage = historyTreePageToken{\n\t\t\tShardID:  0, // First page starting from ShardID 0, and increase if finish reading current shard\n\t\t\tTreeID:   serialization.UUID{},\n\t\t\tBranchID: serialization.UUID{},\n\t\t}\n\t}\n\tfilter := sqlplugin.HistoryTreeFilter{\n\t\tShardID:  page.ShardID,\n\t\tTreeID:   page.TreeID,\n\t\tBranchID: &page.BranchID,\n\t\tPageSize: &request.PageSize,\n\t}\n\trows, err := m.db.GetAllHistoryTreeBranches(ctx, &filter)\n\tif err == sql.ErrNoRows || (err == nil && len(rows) == 0) {\n\t\treturn &persistence.GetAllHistoryTreeBranchesResponse{}, nil\n\t}\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetAllHistoryTreeBranches\", \"\", err)\n\t}\n\tresp := &persistence.GetAllHistoryTreeBranchesResponse{}\n\tresp.Branches = make([]persistence.HistoryBranchDetail, len(rows))\n\tfor i, row := range rows {\n\t\ttreeInfo, err := m.parser.HistoryTreeInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresp.Branches[i].TreeID = row.TreeID.String()\n\t\tresp.Branches[i].BranchID = row.BranchID.String()\n\t\tresp.Branches[i].ForkTime = treeInfo.GetCreatedTimestamp()\n\t\tresp.Branches[i].Info = treeInfo.GetInfo()\n\t}\n\tif len(rows) >= request.PageSize {\n\t\t// there could be more\n\t\tlastRow := &rows[request.PageSize-1]\n\t\tresp.NextPageToken, err = gobSerialize(&historyTreePageToken{\n\t\t\tShardID:  lastRow.ShardID,\n\t\t\tTreeID:   lastRow.TreeID,\n\t\t\tBranchID: lastRow.BranchID,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"error serializing nextPageToken:%v\", err)}\n\t\t}\n\n\t}\n\t// TODO: this is broken for multi-sharding: the shardID should increase if there are less rows than request pageSize,\n\t// until loop over all shards\n\treturn resp, nil\n}\n\n// GetHistoryTree returns all branch information of a tree\nfunc (m *sqlHistoryStore) GetHistoryTree(\n\tctx context.Context,\n\trequest *persistence.InternalGetHistoryTreeRequest,\n) (*persistence.InternalGetHistoryTreeResponse, error) {\n\n\ttreeID := serialization.MustParseUUID(request.TreeID)\n\tbranches := make([]*types.HistoryBranch, 0)\n\n\ttreeFilter := &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:  treeID,\n\t\tShardID: *request.ShardID,\n\t}\n\trows, err := m.db.SelectFromHistoryTree(ctx, treeFilter)\n\tif err == sql.ErrNoRows || (err == nil && len(rows) == 0) {\n\t\treturn &persistence.InternalGetHistoryTreeResponse{}, nil\n\t}\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetHistoryTree\", \"\", err)\n\t}\n\tfor _, row := range rows {\n\t\ttreeInfo, err := m.parser.HistoryTreeInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tbr := &types.HistoryBranch{\n\t\t\tTreeID:    request.TreeID,\n\t\t\tBranchID:  row.BranchID.String(),\n\t\t\tAncestors: treeInfo.Ancestors,\n\t\t}\n\t\tbranches = append(branches, br)\n\t}\n\n\treturn &persistence.InternalGetHistoryTreeResponse{\n\t\tBranches: branches,\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_history_store_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetHistoryTree(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalGetHistoryTreeRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.InternalGetHistoryTreeResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.InternalGetHistoryTreeRequest{\n\t\t\t\tTreeID:  \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tShardID: common.IntPtr(1),\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      1,\n\t\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetHistoryTreeResponse{\n\t\t\t\tBranches: []*types.HistoryBranch{\n\t\t\t\t\t{\n\t\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - no record\",\n\t\t\treq: &persistence.InternalGetHistoryTreeRequest{\n\t\t\t\tTreeID:  \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tShardID: common.IntPtr(1),\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twant:    &persistence.InternalGetHistoryTreeResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - database failure\",\n\t\t\treq: &persistence.InternalGetHistoryTreeRequest{\n\t\t\t\tTreeID:  \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tShardID: common.IntPtr(1),\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - decode error\",\n\t\t\treq: &persistence.InternalGetHistoryTreeRequest{\n\t\t\t\tTreeID:  \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tShardID: common.IntPtr(1),\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      1,\n\t\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql history store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\t\t\tgot, err := store.GetHistoryTree(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalDeleteHistoryBranchRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.InternalDeleteHistoryBranchRequest{\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      1,\n\t\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID: serialization.UUIDPtr(serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\")),\n\t\t\t\t\tShardID:  1,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\t\t\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:  serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID:   1,\n\t\t\t\t\tPageSize:  1000,\n\t\t\t\t\tMinNodeID: common.Int64Ptr(10),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\t\t\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:  serialization.MustParseUUID(\"730ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID:   1,\n\t\t\t\t\tPageSize:  1000,\n\t\t\t\t\tMinNodeID: common.Int64Ptr(3),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 7}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to get history tree\",\n\t\t\treq: &persistence.InternalDeleteHistoryBranchRequest{\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to delete history tree\",\n\t\t\treq: &persistence.InternalDeleteHistoryBranchRequest{\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\t\t\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      1,\n\t\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to delete history node\",\n\t\t\treq: &persistence.InternalDeleteHistoryBranchRequest{\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), gomock.Any()).Return([]sqlplugin.HistoryTreeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      1,\n\t\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.HistoryTreeInfo{\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql history store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\t\t\terr = store.DeleteHistoryBranch(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestForkHistoryBranch(t *testing.T) {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalForkHistoryBranchRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.InternalForkHistoryBranchResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case - case 1\",\n\t\t\treq: &persistence.InternalForkHistoryBranchRequest{\n\t\t\t\tForkBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"830ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 3,\n\t\t\t\t\t\t\tEndNodeID:   5,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tForkNodeID:       4,\n\t\t\t\tNewBranchID:      \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\tInfo:             \"test\",\n\t\t\t\tShardID:          1,\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.HistoryTreeInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.WithinDuration(t, now, info.CreatedTimestamp, time.Second)\n\t\t\t\t\treturn persistence.DataBlob{}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().InsertIntoHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeRow{\n\t\t\t\t\tShardID:  1,\n\t\t\t\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID: serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe699\"),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalForkHistoryBranchResponse{\n\t\t\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"830ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 3,\n\t\t\t\t\t\t\tEndNodeID:   4,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - case 2\",\n\t\t\treq: &persistence.InternalForkHistoryBranchRequest{\n\t\t\t\tForkBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tForkNodeID:  4,\n\t\t\t\tNewBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tShardID:     1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\t\t\t\tmockDB.EXPECT().InsertIntoHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeRow{\n\t\t\t\t\tShardID:  1,\n\t\t\t\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID: serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe699\"),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalForkHistoryBranchResponse{\n\t\t\t\tNewBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 2,\n\t\t\t\t\t\t\tEndNodeID:   4,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to encode blob\",\n\t\t\treq: &persistence.InternalForkHistoryBranchRequest{\n\t\t\t\tForkBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tForkNodeID:  4,\n\t\t\t\tNewBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tShardID:     1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to insert data\",\n\t\t\treq: &persistence.InternalForkHistoryBranchRequest{\n\t\t\t\tForkBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tForkNodeID:  4,\n\t\t\t\tNewBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe699\",\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tShardID:     1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().InsertIntoHistoryTree(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql history store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\t\t\tgot, err := store.ForkHistoryBranch(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAppendHistoryNodes(t *testing.T) {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalAppendHistoryNodesRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case - new branch\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:           11,\n\t\t\t\tEvents:           &persistence.DataBlob{},\n\t\t\t\tTransactionID:    100,\n\t\t\t\tShardID:          1,\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.HistoryTreeInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.WithinDuration(t, now, info.CreatedTimestamp, time.Second)\n\t\t\t\t\treturn persistence.DataBlob{}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeRow{\n\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tNodeID:       11,\n\t\t\t\t\tTxnID:        common.Int64Ptr(100),\n\t\t\t\t\tData:         nil,\n\t\t\t\t\tDataEncoding: \"\",\n\t\t\t\t\tShardID:      1,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeRow{\n\t\t\t\t\tShardID:      1,\n\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tData:         nil,\n\t\t\t\t\tDataEncoding: \"\",\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - old branch\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: false,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        11,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().InsertIntoHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeRow{\n\t\t\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tNodeID:       11,\n\t\t\t\t\tTxnID:        common.Int64Ptr(100),\n\t\t\t\t\tData:         nil,\n\t\t\t\t\tDataEncoding: \"\",\n\t\t\t\t\tShardID:      1,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - new branch, failed to encode data\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        11,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - new branch, failed to insert history node\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        11,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoHistoryNode(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - new branch, failed to insert history tree\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        11,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().HistoryTreeInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, nil)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoHistoryNode(gomock.Any(), gomock.Any()).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoHistoryTree(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - old branch, duplicate entry\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: false,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        11,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().InsertIntoHistoryNode(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsDupEntryError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ConditionFailedError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ConditionFailedError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - invalid history\",\n\t\t\treq: &persistence.InternalAppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: false,\n\t\t\t\tInfo:        \"test\",\n\t\t\t\tBranchInfo: types.HistoryBranch{\n\t\t\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\tAncestors: []*types.HistoryBranchRange{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchID:    \"730ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\t\t\t\tBeginNodeID: 1,\n\t\t\t\t\t\t\tEndNodeID:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNodeID:        5,\n\t\t\t\tEvents:        &persistence.DataBlob{},\n\t\t\t\tTransactionID: 100,\n\t\t\t\tShardID:       1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {},\n\t\t\twantErr:   true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.InvalidPersistenceRequestError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be InvalidPersistenceRequestError\")\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql history store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\t\t\terr = store.AppendHistoryNodes(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadHistoryBranch(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.InternalReadHistoryBranchRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.InternalReadHistoryBranchResponse\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\t\t\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tBranchID:  serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\t\t\tMinNodeID: common.Int64Ptr(201),\n\t\t\t\t\tMaxNodeID: common.Int64Ptr(1000),\n\t\t\t\t\tPageSize:  2,\n\t\t\t\t\tShardID:   1,\n\t\t\t\t}).Return([]sqlplugin.HistoryNodeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       201,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(99),\n\t\t\t\t\t\tData:         []byte(`a`),\n\t\t\t\t\t\tDataEncoding: \"a\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       202,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(101),\n\t\t\t\t\t\tData:         []byte(`b`),\n\t\t\t\t\t\tDataEncoding: \"b\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalReadHistoryBranchResponse{\n\t\t\t\tHistory:           []*persistence.DataBlob{{Data: []byte(`b`), Encoding: constants.EncodingType(\"b\")}},\n\t\t\t\tNextPageToken:     serializePageToken(202),\n\t\t\t\tLastNodeID:        202,\n\t\t\t\tLastTransactionID: 101,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - no row\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t},\n\t\t\twant:    &persistence.InternalReadHistoryBranchResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - corrupted data 1\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), gomock.Any()).Return([]sqlplugin.HistoryNodeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       119,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(99),\n\t\t\t\t\t\tData:         []byte(`a`),\n\t\t\t\t\t\tDataEncoding: \"a\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       202,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(101),\n\t\t\t\t\t\tData:         []byte(`b`),\n\t\t\t\t\t\tDataEncoding: \"b\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.InternalDataInconsistencyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be InternalDataInconsistencyError\")\n\t\t\t\tassert.Contains(t, err.Error(), \"corrupted data, nodeID cannot decrease\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - corrupted data 2\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), gomock.Any()).Return([]sqlplugin.HistoryNodeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       119,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(101),\n\t\t\t\t\t\tData:         []byte(`a`),\n\t\t\t\t\t\tDataEncoding: \"a\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       202,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(101),\n\t\t\t\t\t\tData:         []byte(`b`),\n\t\t\t\t\t\tDataEncoding: \"b\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.InternalDataInconsistencyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be InternalDataInconsistencyError\")\n\t\t\t\tassert.Contains(t, err.Error(), \"corrupted data, nodeID cannot decrease\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - corrupted data 3\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), gomock.Any()).Return([]sqlplugin.HistoryNodeRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tNodeID:       120,\n\t\t\t\t\t\tTxnID:        common.Int64Ptr(101),\n\t\t\t\t\t\tData:         []byte(`a`),\n\t\t\t\t\t\tDataEncoding: \"a\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *types.InternalDataInconsistencyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be InternalDataInconsistencyError\")\n\t\t\t\tassert.Contains(t, err.Error(), \"corrupted data, same nodeID must have smaller txnID\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - database error\",\n\t\t\treq: &persistence.InternalReadHistoryBranchRequest{\n\t\t\t\tTreeID:            \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tBranchID:          \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\t\tMinNodeID:         100,\n\t\t\t\tMaxNodeID:         1000,\n\t\t\t\tPageSize:          2,\n\t\t\t\tNextPageToken:     serializePageToken(200),\n\t\t\t\tLastNodeID:        120,\n\t\t\t\tLastTransactionID: 100,\n\t\t\t\tShardID:           1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromHistoryNode(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := NewHistoryV2Persistence(mockDB, nil, nil, nil)\n\t\t\trequire.NoError(t, err, \"Failed to create sql history store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tgot, err := store.ReadHistoryBranch(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteHistoryBranch_CustomBatchSize(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockDB := sqlplugin.NewMockDB(ctrl)\n\tmockTx := sqlplugin.NewMockTx(ctrl)\n\tmockParser := serialization.NewMockParser(ctrl)\n\n\tcustomBatchSize := 500\n\tdc := &persistence.DynamicConfiguration{\n\t\tHistoryNodeDeleteBatchSize: func(...dynamicproperties.FilterOption) int { return customBatchSize },\n\t}\n\n\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, dc)\n\trequire.NoError(t, err)\n\n\treq := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t},\n\t\tShardID: 1,\n\t}\n\n\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID: 1,\n\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:      1,\n\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tData:         []byte(`aaaa`),\n\t\t\tDataEncoding: \"json\",\n\t\t},\n\t}, nil)\n\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{}, nil)\n\n\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID: serialization.UUIDPtr(serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\")),\n\t\tShardID:  1,\n\t}).Return(nil, nil)\n\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID:  serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID:   1,\n\t\tPageSize:  customBatchSize,\n\t\tMinNodeID: common.Int64Ptr(1),\n\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\tmockTx.EXPECT().Commit().Return(nil)\n\n\terr = store.DeleteHistoryBranch(context.Background(), req)\n\tassert.NoError(t, err)\n}\n\nfunc TestDeleteHistoryBranch_UnboundedBatchSize(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockDB := sqlplugin.NewMockDB(ctrl)\n\tmockTx := sqlplugin.NewMockTx(ctrl)\n\tmockParser := serialization.NewMockParser(ctrl)\n\n\tdc := &persistence.DynamicConfiguration{\n\t\tHistoryNodeDeleteBatchSize: func(...dynamicproperties.FilterOption) int { return 0 },\n\t}\n\n\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, dc)\n\trequire.NoError(t, err)\n\n\treq := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t},\n\t\tShardID: 1,\n\t}\n\n\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID: 1,\n\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:      1,\n\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tData:         []byte(`aaaa`),\n\t\t\tDataEncoding: \"json\",\n\t\t},\n\t}, nil)\n\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{}, nil)\n\n\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID: serialization.UUIDPtr(serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\")),\n\t\tShardID:  1,\n\t}).Return(nil, nil)\n\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID:  serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID:   1,\n\t\tPageSize:  0,\n\t\tMinNodeID: common.Int64Ptr(1),\n\t}).Return(&sqlResult{rowsAffected: 100}, nil)\n\tmockTx.EXPECT().Commit().Return(nil)\n\n\terr = store.DeleteHistoryBranch(context.Background(), req)\n\tassert.NoError(t, err)\n}\n\nfunc TestDeleteHistoryBranch_NegativeBatchSize(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockDB := sqlplugin.NewMockDB(ctrl)\n\tmockTx := sqlplugin.NewMockTx(ctrl)\n\tmockParser := serialization.NewMockParser(ctrl)\n\n\tdc := &persistence.DynamicConfiguration{\n\t\tHistoryNodeDeleteBatchSize: func(...dynamicproperties.FilterOption) int { return -100 },\n\t}\n\n\tstore, err := NewHistoryV2Persistence(mockDB, nil, mockParser, dc)\n\trequire.NoError(t, err)\n\n\treq := &persistence.InternalDeleteHistoryBranchRequest{\n\t\tBranchInfo: types.HistoryBranch{\n\t\t\tTreeID:   \"530ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t\tBranchID: \"630ec3d3-f74b-423f-a138-3b35494fe691\",\n\t\t},\n\t\tShardID: 1,\n\t}\n\n\tmockDB.EXPECT().SelectFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:  serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID: 1,\n\t}).Return([]sqlplugin.HistoryTreeRow{\n\t\t{\n\t\t\tShardID:      1,\n\t\t\tTreeID:       serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tBranchID:     serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\t\tData:         []byte(`aaaa`),\n\t\t\tDataEncoding: \"json\",\n\t\t},\n\t}, nil)\n\tmockParser.EXPECT().HistoryTreeInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.HistoryTreeInfo{}, nil)\n\n\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\tmockTx.EXPECT().DeleteFromHistoryTree(gomock.Any(), &sqlplugin.HistoryTreeFilter{\n\t\tTreeID:   serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID: serialization.UUIDPtr(serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\")),\n\t\tShardID:  1,\n\t}).Return(nil, nil)\n\tmockTx.EXPECT().DeleteFromHistoryNode(gomock.Any(), &sqlplugin.HistoryNodeFilter{\n\t\tTreeID:    serialization.MustParseUUID(\"530ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tBranchID:  serialization.MustParseUUID(\"630ec3d3-f74b-423f-a138-3b35494fe691\"),\n\t\tShardID:   1,\n\t\tPageSize:  -100,\n\t\tMinNodeID: common.Int64Ptr(1),\n\t}).Return(&sqlResult{rowsAffected: 50}, nil)\n\tmockTx.EXPECT().Commit().Return(nil)\n\n\terr = store.DeleteHistoryBranch(context.Background(), req)\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_queue_store.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tsqlQueueStore struct {\n\t\tqueueType persistence.QueueType\n\t\tlogger    log.Logger\n\t\tsqlStore\n\t}\n)\n\nfunc newQueueStore(\n\tdb sqlplugin.DB,\n\tlogger log.Logger,\n\tqueueType persistence.QueueType,\n) (persistence.QueueStore, error) {\n\treturn &sqlQueueStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t},\n\t\tqueueType: queueType,\n\t\tlogger:    logger,\n\t}, nil\n}\n\nfunc (q *sqlQueueStore) EnqueueMessage(ctx context.Context, request *persistence.InternalEnqueueMessageRequest) error {\n\treturn q.txExecute(ctx, sqlplugin.DbDefaultShard, \"EnqueueMessage\", func(tx sqlplugin.Tx) error {\n\t\tlastMessageID, err := tx.GetLastEnqueuedMessageIDForUpdate(ctx, q.queueType)\n\t\tif err != nil {\n\t\t\tif err == sql.ErrNoRows {\n\t\t\t\tlastMessageID = -1\n\t\t\t} else {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tackLevels, err := tx.GetAckLevels(ctx, q.queueType, true)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\t_, err = tx.InsertIntoQueue(ctx, newQueueRow(q.queueType, getNextID(ackLevels, lastMessageID), request.MessagePayload))\n\t\treturn err\n\t})\n}\n\nfunc (q *sqlQueueStore) ReadMessages(\n\tctx context.Context,\n\trequest *persistence.InternalReadMessagesRequest,\n) (*persistence.InternalReadMessagesResponse, error) {\n\n\trows, err := q.db.GetMessagesFromQueue(ctx, q.queueType, request.LastMessageID, request.MaxCount)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"ReadMessages\", \"\", err)\n\t}\n\n\tvar messages []*persistence.InternalQueueMessage\n\tfor _, row := range rows {\n\t\tmessages = append(messages, &persistence.InternalQueueMessage{ID: row.MessageID, Payload: row.MessagePayload})\n\t}\n\treturn &persistence.InternalReadMessagesResponse{Messages: messages}, nil\n}\n\nfunc newQueueRow(\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n\tpayload []byte,\n) *sqlplugin.QueueRow {\n\n\treturn &sqlplugin.QueueRow{QueueType: queueType, MessageID: messageID, MessagePayload: payload}\n}\n\nfunc (q *sqlQueueStore) DeleteMessagesBefore(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteMessagesBeforeRequest,\n) error {\n\n\t_, err := q.db.DeleteMessagesBefore(ctx, q.queueType, request.MessageID)\n\tif err != nil {\n\t\treturn convertCommonErrors(q.db, \"DeleteMessagesBefore\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (q *sqlQueueStore) UpdateAckLevel(ctx context.Context, request *persistence.InternalUpdateAckLevelRequest) error {\n\treturn q.txExecute(ctx, sqlplugin.DbDefaultShard, \"UpdateAckLevel\", func(tx sqlplugin.Tx) error {\n\t\tclusterAckLevels, err := tx.GetAckLevels(ctx, q.queueType, true)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif clusterAckLevels == nil {\n\t\t\treturn tx.InsertAckLevel(ctx, q.queueType, request.MessageID, request.ClusterName)\n\t\t}\n\n\t\t// Ignore possibly delayed message\n\t\tif ackLevel, ok := clusterAckLevels[request.ClusterName]; ok && ackLevel >= request.MessageID {\n\t\t\treturn nil\n\t\t}\n\n\t\tclusterAckLevels[request.ClusterName] = request.MessageID\n\t\treturn tx.UpdateAckLevels(ctx, q.queueType, clusterAckLevels)\n\t})\n}\n\nfunc (q *sqlQueueStore) GetAckLevels(\n\tctx context.Context,\n\t_ *persistence.InternalGetAckLevelsRequest,\n) (*persistence.InternalGetAckLevelsResponse, error) {\n\tresult, err := q.db.GetAckLevels(ctx, q.queueType, false)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"GetAckLevels\", \"\", err)\n\t}\n\treturn &persistence.InternalGetAckLevelsResponse{AckLevels: result}, nil\n}\n\nfunc (q *sqlQueueStore) EnqueueMessageToDLQ(ctx context.Context, request *persistence.InternalEnqueueMessageToDLQRequest) error {\n\treturn q.txExecute(ctx, sqlplugin.DbDefaultShard, \"EnqueueMessageToDLQ\", func(tx sqlplugin.Tx) error {\n\t\tvar err error\n\t\tlastMessageID, err := tx.GetLastEnqueuedMessageIDForUpdate(ctx, q.getDLQTypeFromQueueType())\n\t\tif err != nil {\n\t\t\tif err == sql.ErrNoRows {\n\t\t\t\tlastMessageID = -1\n\t\t\t} else {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\t_, err = tx.InsertIntoQueue(ctx, newQueueRow(q.getDLQTypeFromQueueType(), lastMessageID+1, request.MessagePayload))\n\t\treturn err\n\t})\n}\n\nfunc (q *sqlQueueStore) ReadMessagesFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalReadMessagesFromDLQRequest,\n) (*persistence.InternalReadMessagesFromDLQResponse, error) {\n\n\tfirstMessageID := request.FirstMessageID\n\tif len(request.PageToken) != 0 {\n\t\tlastReadMessageID, err := deserializePageToken(request.PageToken)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"invalid next page token %v\", request.PageToken)}\n\t\t}\n\t\tfirstMessageID = lastReadMessageID\n\t}\n\n\trows, err := q.db.GetMessagesBetween(ctx, q.getDLQTypeFromQueueType(), firstMessageID, request.LastMessageID, request.PageSize)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"ReadMessagesFromDLQ\", \"\", err)\n\t}\n\n\tvar messages []*persistence.InternalQueueMessage\n\tfor _, row := range rows {\n\t\tmessages = append(messages, &persistence.InternalQueueMessage{ID: row.MessageID, Payload: row.MessagePayload})\n\t}\n\n\tvar newPagingToken []byte\n\tif messages != nil && len(messages) >= request.PageSize {\n\t\tlastReadMessageID := messages[len(messages)-1].ID\n\t\tnewPagingToken = serializePageToken(int64(lastReadMessageID))\n\t}\n\treturn &persistence.InternalReadMessagesFromDLQResponse{\n\t\tMessages:      messages,\n\t\tNextPageToken: newPagingToken,\n\t}, nil\n}\n\nfunc (q *sqlQueueStore) DeleteMessageFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalDeleteMessageFromDLQRequest,\n) error {\n\t_, err := q.db.DeleteMessage(ctx, q.getDLQTypeFromQueueType(), request.MessageID)\n\tif err != nil {\n\t\treturn convertCommonErrors(q.db, \"DeleteMessageFromDLQ\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (q *sqlQueueStore) RangeDeleteMessagesFromDLQ(\n\tctx context.Context,\n\trequest *persistence.InternalRangeDeleteMessagesFromDLQRequest,\n) error {\n\t_, err := q.db.RangeDeleteMessages(ctx, q.getDLQTypeFromQueueType(), request.FirstMessageID, request.LastMessageID)\n\tif err != nil {\n\t\treturn convertCommonErrors(q.db, \"RangeDeleteMessagesFromDLQ\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (q *sqlQueueStore) UpdateDLQAckLevel(ctx context.Context, request *persistence.InternalUpdateDLQAckLevelRequest) error {\n\treturn q.txExecute(ctx, sqlplugin.DbDefaultShard, \"UpdateDLQAckLevel\", func(tx sqlplugin.Tx) error {\n\t\tclusterAckLevels, err := tx.GetAckLevels(ctx, q.getDLQTypeFromQueueType(), true)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif clusterAckLevels == nil {\n\t\t\treturn tx.InsertAckLevel(ctx, q.getDLQTypeFromQueueType(), request.MessageID, request.ClusterName)\n\t\t}\n\n\t\t// Ignore possibly delayed message\n\t\tif ackLevel, ok := clusterAckLevels[request.ClusterName]; ok && ackLevel >= request.MessageID {\n\t\t\treturn nil\n\t\t}\n\n\t\tclusterAckLevels[request.ClusterName] = request.MessageID\n\t\treturn tx.UpdateAckLevels(ctx, q.getDLQTypeFromQueueType(), clusterAckLevels)\n\t})\n}\n\nfunc (q *sqlQueueStore) GetDLQAckLevels(\n\tctx context.Context,\n\t_ *persistence.InternalGetDLQAckLevelsRequest,\n) (*persistence.InternalGetDLQAckLevelsResponse, error) {\n\tresult, err := q.db.GetAckLevels(ctx, q.getDLQTypeFromQueueType(), false)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"GetDLQAckLevels\", \"\", err)\n\t}\n\treturn &persistence.InternalGetDLQAckLevelsResponse{AckLevels: result}, nil\n}\n\nfunc (q *sqlQueueStore) GetDLQSize(\n\tctx context.Context,\n\t_ *persistence.InternalGetDLQSizeRequest,\n) (*persistence.InternalGetDLQSizeResponse, error) {\n\tresult, err := q.db.GetQueueSize(ctx, q.getDLQTypeFromQueueType())\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(q.db, \"GetDLQSize\", \"\", err)\n\t}\n\treturn &persistence.InternalGetDLQSizeResponse{Size: result}, nil\n}\n\nfunc (q *sqlQueueStore) getDLQTypeFromQueueType() persistence.QueueType {\n\treturn -q.queueType\n}\n\n// if, for whatever reason, the ack-levels get ahead of the actual messages\n// then ensure the next ID follows\nfunc getNextID(acks map[string]int64, lastMessageID int64) int64 {\n\to := lastMessageID\n\tfor _, v := range acks {\n\t\tif v > o {\n\t\t\to = v\n\t\t}\n\t}\n\treturn o + 1\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_queue_store_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nvar fixedTime = time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)\n\nfunc TestGetNextID(t *testing.T) {\n\ttests := map[string]struct {\n\t\tacks     map[string]int64\n\t\tlastID   int64\n\t\texpected int64\n\t}{\n\t\t\"expected case - last ID is equal to ack-levels\": {\n\t\t\tacks:     map[string]int64{\"a\": 3},\n\t\t\tlastID:   3,\n\t\t\texpected: 4,\n\t\t},\n\t\t\"expected case - last ID is equal to ack-levels haven't caught up\": {\n\t\t\tacks:     map[string]int64{\"a\": 2},\n\t\t\tlastID:   3,\n\t\t\texpected: 4,\n\t\t},\n\t\t\"error case - ack-levels are ahead for some reason\": {\n\t\t\tacks:     map[string]int64{\"a\": 3},\n\t\t\tlastID:   2,\n\t\t\texpected: 4,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, getNextID(td.acks, td.lastID))\n\t\t})\n\t}\n}\n\nfunc TestEnqueueMessage(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), persistence.DomainReplicationQueueType).Return(int64(0), sql.ErrNoRows)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoQueue(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to get last enqueued message ID for update\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), persistence.DomainReplicationQueueType).Return(int64(0), err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to get ack levels\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), persistence.DomainReplicationQueueType).Return(int64(0), sql.ErrNoRows)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to insert into queue\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), persistence.DomainReplicationQueueType).Return(int64(0), sql.ErrNoRows)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoQueue(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx)\n\t\t\terr = store.EnqueueMessage(context.Background(), &persistence.InternalEnqueueMessageRequest{\n\t\t\t\tMessagePayload:   nil,\n\t\t\t\tCurrentTimeStamp: fixedTime,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadMessages(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      []*persistence.InternalQueueMessage\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetMessagesFromQueue(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return([]sqlplugin.QueueRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tQueueType:      persistence.DomainReplicationQueueType,\n\t\t\t\t\t\tMessageID:      123,\n\t\t\t\t\t\tMessagePayload: []byte(`aaaa`),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: []*persistence.InternalQueueMessage{\n\t\t\t\t{\n\t\t\t\t\tID:      123,\n\t\t\t\t\tPayload: []byte(`aaaa`),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetMessagesFromQueue(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tresp, err := store.ReadMessages(context.Background(), &persistence.InternalReadMessagesRequest{\n\t\t\t\tLastMessageID: 0,\n\t\t\t\tMaxCount:      10,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp.Messages, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessagesBefore(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteMessagesBefore(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteMessagesBefore(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\terr = store.DeleteMessagesBefore(context.Background(), &persistence.InternalDeleteMessagesBeforeRequest{\n\t\t\t\tMessageID: 10,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateAckLevel(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tclusterName string\n\t\tmockSetup   func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"Success case - insert\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().InsertAckLevel(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Success case - update\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(map[string]int64{}, nil)\n\t\t\t\tmockTx.EXPECT().UpdateAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Success case - no op\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(map[string]int64{\"abc\": 100}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to get\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to insert\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertAckLevel(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to update\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, true).Return(map[string]int64{}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx)\n\t\t\terr = store.UpdateAckLevel(context.Background(), &persistence.InternalUpdateAckLevelRequest{\n\t\t\t\tMessageID:        0,\n\t\t\t\tClusterName:      tc.clusterName,\n\t\t\t\tCurrentTimeStamp: fixedTime,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetAckLevels(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      map[string]int64\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, false).Return(map[string]int64{\"x\": 9}, nil)\n\t\t\t},\n\t\t\twant:    map[string]int64{\"x\": 9},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetAckLevels(gomock.Any(), persistence.DomainReplicationQueueType, false).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tresp, err := store.GetAckLevels(context.Background(), &persistence.InternalGetAckLevelsRequest{})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp.AckLevels, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestEnqueueMessageToDLQ(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), -1*persistence.DomainReplicationQueueType).Return(int64(0), sql.ErrNoRows)\n\t\t\t\tmockTx.EXPECT().InsertIntoQueue(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to get last enqueued message ID for update\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), -1*persistence.DomainReplicationQueueType).Return(int64(0), err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to insert into queue\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetLastEnqueuedMessageIDForUpdate(gomock.Any(), -1*persistence.DomainReplicationQueueType).Return(int64(0), sql.ErrNoRows)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoQueue(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx)\n\t\t\terr = store.EnqueueMessageToDLQ(context.Background(), &persistence.InternalEnqueueMessageToDLQRequest{\n\t\t\t\tMessagePayload:   nil,\n\t\t\t\tCurrentTimeStamp: fixedTime,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteMessageFromDLQ(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().DeleteMessage(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteMessage(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\terr = store.DeleteMessageFromDLQ(context.Background(), &persistence.InternalDeleteMessageFromDLQRequest{\n\t\t\t\tMessageID: 10,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRangeDeleteMessagesFromDLQ(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().RangeDeleteMessages(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().RangeDeleteMessages(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\terr = store.RangeDeleteMessagesFromDLQ(context.Background(), &persistence.InternalRangeDeleteMessagesFromDLQRequest{\n\t\t\t\tFirstMessageID: 10,\n\t\t\t\tLastMessageID:  100,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDLQAckLevels(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      map[string]int64\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, false).Return(map[string]int64{\"x\": 9}, nil)\n\t\t\t},\n\t\t\twant:    map[string]int64{\"x\": 9},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, false).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tresp, err := store.GetDLQAckLevels(context.Background(), &persistence.InternalGetDLQAckLevelsRequest{})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp.AckLevels, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateDLQAckLevel(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tqueueType   persistence.QueueType\n\t\tclusterName string\n\t\tmockSetup   func(*sqlplugin.MockDB, *sqlplugin.MockTx)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"Success case - insert\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().InsertAckLevel(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Success case - update\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(map[string]int64{}, nil)\n\t\t\t\tmockTx.EXPECT().UpdateAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any()).Return(nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Success case - no op\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(map[string]int64{\"abc\": 100}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to get\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to insert\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertAckLevel(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to update\",\n\t\t\tqueueType:   persistence.DomainReplicationQueueType,\n\t\t\tclusterName: \"abc\",\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), sqlplugin.DbDefaultShard).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().GetAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, true).Return(map[string]int64{}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateAckLevels(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any()).Return(err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx)\n\t\t\terr = store.UpdateDLQAckLevel(context.Background(), &persistence.InternalUpdateDLQAckLevelRequest{\n\t\t\t\tMessageID:        0,\n\t\t\t\tClusterName:      tc.clusterName,\n\t\t\t\tCurrentTimeStamp: fixedTime,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetDLQSize(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      int64\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetQueueSize(gomock.Any(), -1*persistence.DomainReplicationQueueType).Return(int64(1000), nil)\n\t\t\t},\n\t\t\twant:    1000,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetQueueSize(gomock.Any(), -1*persistence.DomainReplicationQueueType).Return(int64(0), err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tresp, err := store.GetDLQSize(context.Background(), &persistence.InternalGetDLQSizeRequest{})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp.Size, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadMessagesFromDLQ(t *testing.T) {\n\tfirstMessageID := int64(0)\n\tlastMessageID := int64(9999)\n\tpageSize := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tqueueType persistence.QueueType\n\t\tpageToken []byte\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      []*persistence.InternalQueueMessage\n\t\twantToken []byte\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:      \"Success case\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tpageToken: serializePageToken(int64(100)),\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetMessagesBetween(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any(), gomock.Any()).Return([]sqlplugin.QueueRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tQueueType:      -1 * persistence.DomainReplicationQueueType,\n\t\t\t\t\t\tMessageID:      123,\n\t\t\t\t\t\tMessagePayload: []byte(`aaaa`),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: []*persistence.InternalQueueMessage{\n\t\t\t\t{\n\t\t\t\t\tID:      123,\n\t\t\t\t\tPayload: []byte(`aaaa`),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantToken: serializePageToken(int64(123)),\n\t\t\twantErr:   false,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to deserialize token\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tpageToken: []byte(`aaa`),\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {},\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Error case - failed to get messages\",\n\t\t\tqueueType: persistence.DomainReplicationQueueType,\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetMessagesBetween(gomock.Any(), -1*persistence.DomainReplicationQueueType, gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore, err := newQueueStore(mockDB, nil, tc.queueType)\n\t\t\trequire.NoError(t, err, \"Failed to create sql queue store\")\n\n\t\t\ttc.mockSetup(mockDB)\n\t\t\tresp, err := store.ReadMessagesFromDLQ(context.Background(), &persistence.InternalReadMessagesFromDLQRequest{\n\t\t\t\tFirstMessageID: firstMessageID,\n\t\t\t\tLastMessageID:  lastMessageID,\n\t\t\t\tPageSize:       pageSize,\n\t\t\t\tPageToken:      tc.pageToken,\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, resp.Messages, \"Unexpected result for test case\")\n\t\t\t\tassert.Equal(t, tc.wantToken, resp.NextPageToken, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_shard_store.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlShardStore struct {\n\tsqlStore\n\tcurrentClusterName string\n}\n\n// NewShardPersistence creates an instance of ShardStore\nfunc NewShardPersistence(\n\tdb sqlplugin.DB,\n\tcurrentClusterName string,\n\tlog log.Logger,\n\tparser serialization.Parser,\n) (persistence.ShardStore, error) {\n\treturn &sqlShardStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: log,\n\t\t\tparser: parser,\n\t\t},\n\t\tcurrentClusterName: currentClusterName,\n\t}, nil\n}\n\nfunc (m *sqlShardStore) CreateShard(\n\tctx context.Context,\n\trequest *persistence.InternalCreateShardRequest,\n) error {\n\tif _, err := m.GetShard(ctx, &persistence.InternalGetShardRequest{\n\t\tShardID: request.ShardInfo.ShardID,\n\t}); err == nil {\n\t\treturn &persistence.ShardAlreadyExistError{\n\t\t\tMsg: fmt.Sprintf(\"CreateShard operation failed. Shard with ID %v already exists.\", request.ShardInfo.ShardID),\n\t\t}\n\t}\n\n\trow, err := shardInfoToShardsRow(*request.ShardInfo, m.parser)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"CreateShard operation failed. Error: %v\", err),\n\t\t}\n\t}\n\n\tif _, err := m.db.InsertIntoShards(ctx, row); err != nil {\n\t\treturn convertCommonErrors(m.db, \"CreateShard\", \"Failed to insert into shards table.\", err)\n\t}\n\n\treturn nil\n}\n\nfunc (m *sqlShardStore) GetShard(\n\tctx context.Context,\n\trequest *persistence.InternalGetShardRequest,\n) (*persistence.InternalGetShardResponse, error) {\n\trow, err := m.db.SelectFromShards(ctx, &sqlplugin.ShardsFilter{ShardID: int64(request.ShardID)})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetShard\", fmt.Sprintf(\"Failed to get shard, ShardId: %v.\", request.ShardID), err)\n\t}\n\n\tshardInfo, err := m.parser.ShardInfoFromBlob(row.Data, row.DataEncoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif len(shardInfo.ClusterTransferAckLevel) == 0 {\n\t\tshardInfo.ClusterTransferAckLevel = map[string]int64{\n\t\t\tm.currentClusterName: shardInfo.GetTransferAckLevel(),\n\t\t}\n\t}\n\n\ttimerAckLevel := make(map[string]time.Time, len(shardInfo.ClusterTimerAckLevel))\n\tfor k, v := range shardInfo.ClusterTimerAckLevel {\n\t\ttimerAckLevel[k] = v\n\t}\n\n\tif len(timerAckLevel) == 0 {\n\t\ttimerAckLevel = map[string]time.Time{\n\t\t\tm.currentClusterName: shardInfo.GetTimerAckLevel(),\n\t\t}\n\t}\n\n\tif shardInfo.ClusterReplicationLevel == nil {\n\t\tshardInfo.ClusterReplicationLevel = make(map[string]int64)\n\t}\n\tif shardInfo.ReplicationDlqAckLevel == nil {\n\t\tshardInfo.ReplicationDlqAckLevel = make(map[string]int64)\n\t}\n\n\tvar transferPQS *persistence.DataBlob\n\tif shardInfo.GetTransferProcessingQueueStates() != nil {\n\t\ttransferPQS = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetTransferProcessingQueueStatesEncoding()),\n\t\t\tData:     shardInfo.GetTransferProcessingQueueStates(),\n\t\t}\n\t}\n\n\tvar timerPQS *persistence.DataBlob\n\tif shardInfo.GetTimerProcessingQueueStates() != nil {\n\t\ttimerPQS = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetTimerProcessingQueueStatesEncoding()),\n\t\t\tData:     shardInfo.GetTimerProcessingQueueStates(),\n\t\t}\n\t}\n\n\tvar pendingFailoverMarkers *persistence.DataBlob\n\tif shardInfo.GetPendingFailoverMarkers() != nil {\n\t\tpendingFailoverMarkers = &persistence.DataBlob{\n\t\t\tEncoding: constants.EncodingType(shardInfo.GetPendingFailoverMarkersEncoding()),\n\t\t\tData:     shardInfo.GetPendingFailoverMarkers(),\n\t\t}\n\t}\n\tresp := &persistence.InternalGetShardResponse{ShardInfo: &persistence.InternalShardInfo{\n\t\tShardID:                       int(row.ShardID),\n\t\tRangeID:                       row.RangeID,\n\t\tOwner:                         shardInfo.GetOwner(),\n\t\tStolenSinceRenew:              int(shardInfo.GetStolenSinceRenew()),\n\t\tUpdatedAt:                     shardInfo.GetUpdatedAt(),\n\t\tReplicationAckLevel:           shardInfo.GetReplicationAckLevel(),\n\t\tTransferAckLevel:              shardInfo.GetTransferAckLevel(),\n\t\tTimerAckLevel:                 shardInfo.GetTimerAckLevel(),\n\t\tClusterTransferAckLevel:       shardInfo.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:          timerAckLevel,\n\t\tTransferProcessingQueueStates: transferPQS,\n\t\tTimerProcessingQueueStates:    timerPQS,\n\t\tDomainNotificationVersion:     shardInfo.GetDomainNotificationVersion(),\n\t\tClusterReplicationLevel:       shardInfo.ClusterReplicationLevel,\n\t\tReplicationDLQAckLevel:        shardInfo.ReplicationDlqAckLevel,\n\t\tPendingFailoverMarkers:        pendingFailoverMarkers,\n\t\tQueueStates:                   shardInfo.GetQueueStates(),\n\t}}\n\n\treturn resp, nil\n}\n\nfunc (m *sqlShardStore) UpdateShard(\n\tctx context.Context,\n\trequest *persistence.InternalUpdateShardRequest,\n) error {\n\trow, err := shardInfoToShardsRow(*request.ShardInfo, m.parser)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"UpdateShard operation failed. Error: %v\", err),\n\t\t}\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(request.ShardInfo.ShardID, m.db.GetTotalNumDBShards())\n\treturn m.txExecute(ctx, dbShardID, \"UpdateShard\", func(tx sqlplugin.Tx) error {\n\t\tif err := lockShard(ctx, tx, request.ShardInfo.ShardID, request.PreviousRangeID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tresult, err := tx.UpdateShards(ctx, row)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\trowsAffected, err := result.RowsAffected()\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"rowsAffected returned error for shardID %v: %v\", request.ShardInfo.ShardID, err)\n\t\t}\n\t\tif rowsAffected != 1 {\n\t\t\treturn fmt.Errorf(\"rowsAffected returned %v shards instead of one\", rowsAffected)\n\t\t}\n\t\treturn nil\n\t})\n}\n\n// initiated by the owning shard\nfunc lockShard(ctx context.Context, tx sqlplugin.Tx, shardID int, oldRangeID int64) error {\n\trangeID, err := tx.WriteLockShards(ctx, &sqlplugin.ShardsFilter{ShardID: int64(shardID)})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Failed to lock shard with ID %v that does not exist.\", shardID),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(tx, \"lockShard\", fmt.Sprintf(\"Failed to lock shard with ID: %v.\", shardID), err)\n\t}\n\n\tif int64(rangeID) != oldRangeID {\n\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\tShardID: shardID,\n\t\t\tMsg:     fmt.Sprintf(\"Failed to update shard. Previous range ID: %v; new range ID: %v\", oldRangeID, rangeID),\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// initiated by the owning shard\nfunc readLockShard(ctx context.Context, tx sqlplugin.Tx, shardID int, oldRangeID int64) error {\n\trangeID, err := tx.ReadLockShards(ctx, &sqlplugin.ShardsFilter{ShardID: int64(shardID)})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Failed to lock shard with ID %v that does not exist.\", shardID),\n\t\t\t}\n\t\t}\n\t\treturn convertCommonErrors(tx, \"readLockShard\", fmt.Sprintf(\"Failed to lock shard with ID: %v.\", shardID), err)\n\t}\n\n\tif int64(rangeID) != oldRangeID {\n\t\treturn &persistence.ShardOwnershipLostError{\n\t\t\tShardID: shardID,\n\t\t\tMsg:     fmt.Sprintf(\"Failed to lock shard. Previous range ID: %v; new range ID: %v\", oldRangeID, rangeID),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc shardInfoToShardsRow(s persistence.InternalShardInfo, parser serialization.Parser) (*sqlplugin.ShardsRow, error) {\n\tvar markerData []byte\n\tmarkerEncoding := string(constants.EncodingTypeEmpty)\n\tif s.PendingFailoverMarkers != nil {\n\t\tmarkerData = s.PendingFailoverMarkers.Data\n\t\tmarkerEncoding = string(s.PendingFailoverMarkers.Encoding)\n\t}\n\n\tvar transferPQSData []byte\n\ttransferPQSEncoding := string(constants.EncodingTypeEmpty)\n\tif s.TransferProcessingQueueStates != nil {\n\t\ttransferPQSData = s.TransferProcessingQueueStates.Data\n\t\ttransferPQSEncoding = string(s.TransferProcessingQueueStates.Encoding)\n\t}\n\n\tvar timerPQSData []byte\n\ttimerPQSEncoding := string(constants.EncodingTypeEmpty)\n\tif s.TimerProcessingQueueStates != nil {\n\t\ttimerPQSData = s.TimerProcessingQueueStates.Data\n\t\ttimerPQSEncoding = string(s.TimerProcessingQueueStates.Encoding)\n\t}\n\n\tshardInfo := &serialization.ShardInfo{\n\t\tStolenSinceRenew:                      int32(s.StolenSinceRenew),\n\t\tUpdatedAt:                             s.UpdatedAt,\n\t\tReplicationAckLevel:                   s.ReplicationAckLevel,\n\t\tTransferAckLevel:                      s.TransferAckLevel,\n\t\tTimerAckLevel:                         s.TimerAckLevel,\n\t\tClusterTransferAckLevel:               s.ClusterTransferAckLevel,\n\t\tClusterTimerAckLevel:                  s.ClusterTimerAckLevel,\n\t\tTransferProcessingQueueStates:         transferPQSData,\n\t\tTransferProcessingQueueStatesEncoding: transferPQSEncoding,\n\t\tTimerProcessingQueueStates:            timerPQSData,\n\t\tTimerProcessingQueueStatesEncoding:    timerPQSEncoding,\n\t\tDomainNotificationVersion:             s.DomainNotificationVersion,\n\t\tOwner:                                 s.Owner,\n\t\tClusterReplicationLevel:               s.ClusterReplicationLevel,\n\t\tReplicationDlqAckLevel:                s.ReplicationDLQAckLevel,\n\t\tPendingFailoverMarkers:                markerData,\n\t\tPendingFailoverMarkersEncoding:        markerEncoding,\n\t\tQueueStates:                           s.QueueStates,\n\t}\n\n\tblob, err := parser.ShardInfoToBlob(shardInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &sqlplugin.ShardsRow{\n\t\tShardID:      int64(s.ShardID),\n\t\tRangeID:      s.RangeID,\n\t\tData:         blob.Data,\n\t\tDataEncoding: string(blob.Encoding),\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_shard_store_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tclusterName string\n\t\treq         *persistence.InternalGetShardRequest\n\t\tmockSetup   func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant        *persistence.InternalGetShardResponse\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"Success case\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalGetShardRequest{\n\t\t\t\tShardID: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 2}).Return(&sqlplugin.ShardsRow{\n\t\t\t\t\tShardID:      2,\n\t\t\t\t\tRangeID:      4,\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().ShardInfoFromBlob([]byte(`aaaa`), \"json\").Return(&serialization.ShardInfo{\n\t\t\t\t\tOwner:                                 \"owner\",\n\t\t\t\t\tStolenSinceRenew:                      1,\n\t\t\t\t\tUpdatedAt:                             time.Unix(100, 10),\n\t\t\t\t\tReplicationAckLevel:                   1001,\n\t\t\t\t\tTransferAckLevel:                      1002,\n\t\t\t\t\tTimerAckLevel:                         time.Unix(2, 1),\n\t\t\t\t\tTransferProcessingQueueStates:         []byte(`transfer`),\n\t\t\t\t\tTransferProcessingQueueStatesEncoding: \"transfer\",\n\t\t\t\t\tTimerProcessingQueueStates:            []byte(`timer`),\n\t\t\t\t\tTimerProcessingQueueStatesEncoding:    \"timer\",\n\t\t\t\t\tDomainNotificationVersion:             99,\n\t\t\t\t\tPendingFailoverMarkers:                []byte(`markers`),\n\t\t\t\t\tPendingFailoverMarkersEncoding:        \"markers\",\n\t\t\t\t\tReplicationDlqAckLevel:                map[string]int64{\"active\": 10},\n\t\t\t\t\tClusterReplicationLevel:               map[string]int64{\"active\": 1002},\n\t\t\t\t\tClusterTimerAckLevel:                  map[string]time.Time{\"active\": time.Unix(2, 1)},\n\t\t\t\t\tClusterTransferAckLevel:               map[string]int64{\"active\": 1002},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.InternalGetShardResponse{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:                 2,\n\t\t\t\t\tRangeID:                 4,\n\t\t\t\t\tOwner:                   \"owner\",\n\t\t\t\t\tStolenSinceRenew:        1,\n\t\t\t\t\tUpdatedAt:               time.Unix(100, 10),\n\t\t\t\t\tReplicationAckLevel:     1001,\n\t\t\t\t\tTransferAckLevel:        1002,\n\t\t\t\t\tTimerAckLevel:           time.Unix(2, 1),\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\"active\": 1002},\n\t\t\t\t\tClusterTimerAckLevel:    map[string]time.Time{\"active\": time.Unix(2, 1)},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"transfer\"),\n\t\t\t\t\t\tData:     []byte(`transfer`),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"timer\"),\n\t\t\t\t\t\tData:     []byte(`timer`),\n\t\t\t\t\t},\n\t\t\t\t\tDomainNotificationVersion: 99,\n\t\t\t\t\tClusterReplicationLevel:   map[string]int64{\"active\": 1002},\n\t\t\t\t\tReplicationDLQAckLevel:    map[string]int64{\"active\": 10},\n\t\t\t\t\tPendingFailoverMarkers: &persistence.DataBlob{\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"markers\"),\n\t\t\t\t\t\tData:     []byte(`markers`),\n\t\t\t\t\t},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to get shard\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalGetShardRequest{\n\t\t\t\tShardID: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to decode data\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalGetShardRequest{\n\t\t\t\tShardID: 2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 2}).Return(&sqlplugin.ShardsRow{\n\t\t\t\t\tShardID:      2,\n\t\t\t\t\tRangeID:      4,\n\t\t\t\t\tData:         []byte(`aaaa`),\n\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockParser.EXPECT().ShardInfoFromBlob(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewShardPersistence(mockDB, tc.clusterName, nil, mockParser)\n\t\t\trequire.NoError(t, err, \"Failed to create sql shard store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\t\t\tgot, err := store.GetShard(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tclusterName string\n\t\treq         *persistence.InternalCreateShardRequest\n\t\tmockSetup   func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twantErr     bool\n\t\tassertErr   func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:        \"Success case\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:                 2,\n\t\t\t\t\tOwner:                   \"owner\",\n\t\t\t\t\tRangeID:                 4,\n\t\t\t\t\tStolenSinceRenew:        1,\n\t\t\t\t\tUpdatedAt:               time.Unix(1, 2),\n\t\t\t\t\tReplicationAckLevel:     9,\n\t\t\t\t\tTransferAckLevel:        99,\n\t\t\t\t\tTimerAckLevel:           time.Unix(9, 9),\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\"a\": 8},\n\t\t\t\t\tClusterTimerAckLevel:    map[string]time.Time{\"b\": time.Unix(8, 8)},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`transfer`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"transfer\"),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`timer`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"timer\"),\n\t\t\t\t\t},\n\t\t\t\t\tDomainNotificationVersion: 101,\n\t\t\t\t\tClusterReplicationLevel:   map[string]int64{\"z\": 199},\n\t\t\t\t\tReplicationDLQAckLevel:    map[string]int64{\"y\": 1111},\n\t\t\t\t\tPendingFailoverMarkers: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`markers`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"markers\"),\n\t\t\t\t\t},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(&serialization.ShardInfo{\n\t\t\t\t\tStolenSinceRenew:                      1,\n\t\t\t\t\tUpdatedAt:                             time.Unix(1, 2),\n\t\t\t\t\tReplicationAckLevel:                   9,\n\t\t\t\t\tTransferAckLevel:                      99,\n\t\t\t\t\tTimerAckLevel:                         time.Unix(9, 9),\n\t\t\t\t\tClusterTransferAckLevel:               map[string]int64{\"a\": 8},\n\t\t\t\t\tClusterTimerAckLevel:                  map[string]time.Time{\"b\": time.Unix(8, 8)},\n\t\t\t\t\tTransferProcessingQueueStates:         []byte(`transfer`),\n\t\t\t\t\tTransferProcessingQueueStatesEncoding: \"transfer\",\n\t\t\t\t\tTimerProcessingQueueStates:            []byte(`timer`),\n\t\t\t\t\tTimerProcessingQueueStatesEncoding:    \"timer\",\n\t\t\t\t\tDomainNotificationVersion:             101,\n\t\t\t\t\tOwner:                                 \"owner\",\n\t\t\t\t\tClusterReplicationLevel:               map[string]int64{\"z\": 199},\n\t\t\t\t\tReplicationDlqAckLevel:                map[string]int64{\"y\": 1111},\n\t\t\t\t\tPendingFailoverMarkers:                []byte(`markers`),\n\t\t\t\t\tPendingFailoverMarkersEncoding:        \"markers\",\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingType(\"shard\"),\n\t\t\t\t\tData:     []byte(`shard`),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().InsertIntoShards(gomock.Any(), &sqlplugin.ShardsRow{\n\t\t\t\t\tShardID:      2,\n\t\t\t\t\tRangeID:      4,\n\t\t\t\t\tData:         []byte(`shard`),\n\t\t\t\t\tDataEncoding: \"shard\",\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - already exists\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(&sqlplugin.ShardsRow{}, nil)\n\t\t\t\tmockParser.EXPECT().ShardInfoFromBlob(gomock.Any(), gomock.Any()).Return(&serialization.ShardInfo{}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ShardAlreadyExistError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ShardAlreadyExistError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to encode\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to insert\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalCreateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromShards(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingType(\"shard\"),\n\t\t\t\t\tData:     []byte(`shard`),\n\t\t\t\t}, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().InsertIntoShards(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewShardPersistence(mockDB, tc.clusterName, nil, mockParser)\n\t\t\trequire.NoError(t, err, \"Failed to create sql shard store\")\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\t\t\terr = store.CreateShard(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateShard(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tclusterName string\n\t\treq         *persistence.InternalUpdateShardRequest\n\t\tmockSetup   func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twantErr     bool\n\t}{\n\t\t{\n\t\t\tname:        \"Success case\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateShardRequest{\n\t\t\t\tPreviousRangeID: 1,\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{\n\t\t\t\t\tShardID:                 2,\n\t\t\t\t\tOwner:                   \"owner\",\n\t\t\t\t\tRangeID:                 4,\n\t\t\t\t\tStolenSinceRenew:        1,\n\t\t\t\t\tUpdatedAt:               time.Unix(1, 2),\n\t\t\t\t\tReplicationAckLevel:     9,\n\t\t\t\t\tTransferAckLevel:        99,\n\t\t\t\t\tTimerAckLevel:           time.Unix(9, 9),\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\"a\": 8},\n\t\t\t\t\tClusterTimerAckLevel:    map[string]time.Time{\"b\": time.Unix(8, 8)},\n\t\t\t\t\tTransferProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`transfer`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"transfer\"),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`timer`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"timer\"),\n\t\t\t\t\t},\n\t\t\t\t\tDomainNotificationVersion: 101,\n\t\t\t\t\tClusterReplicationLevel:   map[string]int64{\"z\": 199},\n\t\t\t\t\tReplicationDLQAckLevel:    map[string]int64{\"y\": 1111},\n\t\t\t\t\tPendingFailoverMarkers: &persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`markers`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"markers\"),\n\t\t\t\t\t},\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(&serialization.ShardInfo{\n\t\t\t\t\tStolenSinceRenew:                      1,\n\t\t\t\t\tUpdatedAt:                             time.Unix(1, 2),\n\t\t\t\t\tReplicationAckLevel:                   9,\n\t\t\t\t\tTransferAckLevel:                      99,\n\t\t\t\t\tTimerAckLevel:                         time.Unix(9, 9),\n\t\t\t\t\tClusterTransferAckLevel:               map[string]int64{\"a\": 8},\n\t\t\t\t\tClusterTimerAckLevel:                  map[string]time.Time{\"b\": time.Unix(8, 8)},\n\t\t\t\t\tTransferProcessingQueueStates:         []byte(`transfer`),\n\t\t\t\t\tTransferProcessingQueueStatesEncoding: \"transfer\",\n\t\t\t\t\tTimerProcessingQueueStates:            []byte(`timer`),\n\t\t\t\t\tTimerProcessingQueueStatesEncoding:    \"timer\",\n\t\t\t\t\tDomainNotificationVersion:             101,\n\t\t\t\t\tOwner:                                 \"owner\",\n\t\t\t\t\tClusterReplicationLevel:               map[string]int64{\"z\": 199},\n\t\t\t\t\tReplicationDlqAckLevel:                map[string]int64{\"y\": 1111},\n\t\t\t\t\tPendingFailoverMarkers:                []byte(`markers`),\n\t\t\t\t\tPendingFailoverMarkersEncoding:        \"markers\",\n\t\t\t\t\tQueueStates: map[int32]*types.QueueState{\n\t\t\t\t\t\t0: &types.QueueState{\n\t\t\t\t\t\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingType(\"shard\"),\n\t\t\t\t\tData:     []byte(`shard`),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 2}).Return(1, nil)\n\t\t\t\tmockTx.EXPECT().UpdateShards(gomock.Any(), &sqlplugin.ShardsRow{\n\t\t\t\t\tShardID:      2,\n\t\t\t\t\tRangeID:      4,\n\t\t\t\t\tData:         []byte(`shard`),\n\t\t\t\t\tDataEncoding: \"shard\",\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to encode\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{}, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to lock\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingType(\"shard\"),\n\t\t\t\t\tData:     []byte(`shard`),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), gomock.Any()).Return(0, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Error case - failed to update\",\n\t\t\tclusterName: \"active\",\n\t\t\treq: &persistence.InternalUpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.InternalShardInfo{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockParser.EXPECT().ShardInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tEncoding: constants.EncodingType(\"shard\"),\n\t\t\t\t\tData:     []byte(`shard`),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), gomock.Any()).Return(0, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateShards(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(gomock.Any()).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore, err := NewShardPersistence(mockDB, tc.clusterName, nil, mockParser)\n\t\t\trequire.NoError(t, err, \"Failed to create sql shard store\")\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\t\t\terr = store.UpdateShard(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLockShard(t *testing.T) {\n\tshardID := 1\n\toldRangeID := int64(99)\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(99, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - not found\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(0, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - database failure\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(0, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rangeID mismatch\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().WriteLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(98, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ShardOwnershipLostError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ShardOwnershipLostError\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\t\t\terr := lockShard(context.Background(), mockTx, shardID, oldRangeID)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReadLockShard(t *testing.T) {\n\tshardID := 1\n\toldRangeID := int64(99)\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(99, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - not found\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(0, sql.ErrNoRows)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - database failure\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(0, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rangeID mismatch\",\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().ReadLockShards(gomock.Any(), &sqlplugin.ShardsFilter{ShardID: 1}).Return(98, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ShardOwnershipLostError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ShardOwnershipLostError\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\t\t\terr := readLockShard(context.Background(), mockTx, shardID, oldRangeID)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_task_store.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype sqlTaskStore struct {\n\tsqlStore\n\tnShards int\n}\n\nvar (\n\ttaskListTTL = time.Hour * 24\n)\n\n// newTaskPersistence creates a new instance of TaskManager\nfunc newTaskPersistence(\n\tdb sqlplugin.DB,\n\tnShards int,\n\tlog log.Logger,\n\tparser serialization.Parser,\n) (persistence.TaskStore, error) {\n\treturn &sqlTaskStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: log,\n\t\t\tparser: parser,\n\t\t},\n\t\tnShards: nShards,\n\t}, nil\n}\n\nfunc (m *sqlTaskStore) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (*persistence.GetTaskListSizeResponse, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskListName, m.db.GetTotalNumDBShards())\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\tsize, err := m.db.GetTasksCount(ctx, &sqlplugin.TasksFilter{\n\t\tShardID:      dbShardID,\n\t\tDomainID:     domainID,\n\t\tTaskListName: request.TaskListName,\n\t\tTaskType:     int64(request.TaskListType),\n\t\tMinTaskID:    &request.AckLevel,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetTaskListSize\", \"\", err)\n\t}\n\treturn &persistence.GetTaskListSizeResponse{Size: size}, nil\n}\n\nfunc (m *sqlTaskStore) LeaseTaskList(\n\tctx context.Context,\n\trequest *persistence.LeaseTaskListRequest,\n) (*persistence.LeaseTaskListResponse, error) {\n\tvar rangeID int64\n\tvar ackLevel int64\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskList, m.db.GetTotalNumDBShards())\n\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\trows, err := m.db.SelectFromTaskLists(ctx, &sqlplugin.TaskListsFilter{\n\t\tShardID:  dbShardID,\n\t\tDomainID: &domainID,\n\t\tName:     &request.TaskList,\n\t\tTaskType: common.Int64Ptr(int64(request.TaskType))})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\ttlInfo := &serialization.TaskListInfo{\n\t\t\t\tAckLevel:        ackLevel,\n\t\t\t\tKind:            int16(request.TaskListKind),\n\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\tLastUpdated:     request.CurrentTimeStamp,\n\t\t\t}\n\t\t\tblob, err := m.parser.TaskListInfoToBlob(tlInfo)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\trow := sqlplugin.TaskListsRow{\n\t\t\t\tShardID:      dbShardID,\n\t\t\t\tDomainID:     domainID,\n\t\t\t\tName:         request.TaskList,\n\t\t\t\tTaskType:     int64(request.TaskType),\n\t\t\t\tData:         blob.Data,\n\t\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\t}\n\t\t\trows = []sqlplugin.TaskListsRow{row}\n\t\t\tif m.db.SupportsTTL() && persistence.TaskListKindHasTTL(request.TaskListKind) {\n\t\t\t\trowWithTTL := sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: row,\n\t\t\t\t\tTTL:          taskListTTL,\n\t\t\t\t}\n\t\t\t\tif _, err := m.db.InsertIntoTaskListsWithTTL(ctx, &rowWithTTL); err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(m.db, \"LeaseTaskListWithTTL\", fmt.Sprintf(\"Failed to make task list %v of type %v.\", request.TaskList, request.TaskType), err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tif _, err := m.db.InsertIntoTaskLists(ctx, &row); err != nil {\n\t\t\t\t\treturn nil, convertCommonErrors(m.db, \"LeaseTaskList\", fmt.Sprintf(\"Failed to make task list %v of type %v.\", request.TaskList, request.TaskType), err)\n\t\t\t\t}\n\t\t\t}\n\t\t} else {\n\t\t\treturn nil, convertCommonErrors(m.db, \"LeaseTaskList\", \"Failed to check if task list existed.\", err)\n\t\t}\n\t}\n\n\trow := rows[0]\n\tif request.RangeID > 0 && request.RangeID != row.RangeID {\n\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\tMsg: fmt.Sprintf(\"leaseTaskList:renew failed:taskList:%v, taskListType:%v, haveRangeID:%v, gotRangeID:%v\",\n\t\t\t\trequest.TaskList, request.TaskType, rangeID, row.RangeID),\n\t\t}\n\t}\n\n\ttlInfo, err := m.parser.TaskListInfoFromBlob(row.Data, row.DataEncoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar resp *persistence.LeaseTaskListResponse\n\terr = m.txExecute(ctx, dbShardID, \"LeaseTaskList\", func(tx sqlplugin.Tx) error {\n\t\trangeID = row.RangeID\n\t\tackLevel = tlInfo.GetAckLevel()\n\t\t// We need to separately check the condition and do the\n\t\t// update because we want to throw different error codes.\n\t\t// Since we need to do things separately (in a transaction), we need to take a lock.\n\t\terr1 := lockTaskList(ctx, tx, dbShardID, domainID, request.TaskList, request.TaskType, rangeID)\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tnow := request.CurrentTimeStamp\n\t\ttlInfo.LastUpdated = now\n\t\tblob, err1 := m.parser.TaskListInfoToBlob(tlInfo)\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\trow := &sqlplugin.TaskListsRow{\n\t\t\tShardID:      dbShardID,\n\t\t\tDomainID:     row.DomainID,\n\t\t\tRangeID:      row.RangeID + 1,\n\t\t\tName:         row.Name,\n\t\t\tTaskType:     row.TaskType,\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t}\n\t\tvar result sql.Result\n\t\tif m.db.SupportsTTL() && persistence.TaskListKindHasTTL(int(tlInfo.GetKind())) {\n\t\t\tresult, err1 = tx.UpdateTaskListsWithTTL(ctx, &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\tTaskListsRow: *row,\n\t\t\t\tTTL:          taskListTTL,\n\t\t\t})\n\t\t} else {\n\t\t\tresult, err1 = tx.UpdateTaskLists(ctx, row)\n\t\t}\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\trowsAffected, err1 := result.RowsAffected()\n\t\tif err1 != nil {\n\t\t\treturn fmt.Errorf(\"rowsAffected error: %v\", err1)\n\t\t}\n\t\tif rowsAffected == 0 {\n\t\t\treturn fmt.Errorf(\"%v rows affected instead of 1\", rowsAffected)\n\t\t}\n\t\tresp = &persistence.LeaseTaskListResponse{TaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID:                request.DomainID,\n\t\t\tName:                    request.TaskList,\n\t\t\tTaskType:                request.TaskType,\n\t\t\tRangeID:                 rangeID + 1,\n\t\t\tAckLevel:                ackLevel,\n\t\t\tKind:                    request.TaskListKind,\n\t\t\tLastUpdated:             now,\n\t\t\tAdaptivePartitionConfig: fromSerializationTaskListPartitionConfig(tlInfo.AdaptivePartitionConfig),\n\t\t}}\n\t\treturn nil\n\t})\n\treturn resp, err\n}\n\nfunc (m *sqlTaskStore) GetTaskList(\n\tctx context.Context,\n\trequest *persistence.GetTaskListRequest,\n) (*persistence.GetTaskListResponse, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskList, m.db.GetTotalNumDBShards())\n\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\trows, err := m.db.SelectFromTaskLists(ctx, &sqlplugin.TaskListsFilter{\n\t\tShardID:  dbShardID,\n\t\tDomainID: &domainID,\n\t\tName:     &request.TaskList,\n\t\tTaskType: common.Int64Ptr(int64(request.TaskType))})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetTaskList\", \"\", err)\n\t}\n\trow := rows[0]\n\ttlInfo, err := m.parser.TaskListInfoFromBlob(row.Data, row.DataEncoding)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &persistence.GetTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID:                request.DomainID,\n\t\t\tName:                    request.TaskList,\n\t\t\tTaskType:                request.TaskType,\n\t\t\tRangeID:                 row.RangeID,\n\t\t\tAckLevel:                tlInfo.AckLevel,\n\t\t\tKind:                    int(tlInfo.Kind),\n\t\t\tExpiry:                  tlInfo.ExpiryTimestamp,\n\t\t\tLastUpdated:             tlInfo.LastUpdated,\n\t\t\tAdaptivePartitionConfig: fromSerializationTaskListPartitionConfig(tlInfo.AdaptivePartitionConfig),\n\t\t},\n\t}, nil\n}\n\nfunc (m *sqlTaskStore) UpdateTaskList(\n\tctx context.Context,\n\trequest *persistence.UpdateTaskListRequest,\n) (*persistence.UpdateTaskListResponse, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.TaskListInfo.DomainID, request.TaskListInfo.Name, m.db.GetTotalNumDBShards())\n\tdomainID := serialization.MustParseUUID(request.TaskListInfo.DomainID)\n\tnow := request.CurrentTimeStamp\n\ttlInfo := &serialization.TaskListInfo{\n\t\tAckLevel:                request.TaskListInfo.AckLevel,\n\t\tKind:                    int16(request.TaskListInfo.Kind),\n\t\tExpiryTimestamp:         time.Unix(0, 0),\n\t\tLastUpdated:             now,\n\t\tAdaptivePartitionConfig: toSerializationTaskListPartitionConfig(request.TaskListInfo.AdaptivePartitionConfig),\n\t}\n\tif persistence.TaskListKindHasTTL(request.TaskListInfo.Kind) {\n\t\ttlInfo.ExpiryTimestamp = now.Add(taskListTTL)\n\t}\n\n\tvar resp *persistence.UpdateTaskListResponse\n\tblob, err := m.parser.TaskListInfoToBlob(tlInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\terr = m.txExecute(ctx, dbShardID, \"UpdateTaskList\", func(tx sqlplugin.Tx) error {\n\t\terr1 := lockTaskList(\n\t\t\tctx, tx, dbShardID, domainID, request.TaskListInfo.Name, request.TaskListInfo.TaskType, request.TaskListInfo.RangeID)\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tvar result sql.Result\n\t\trow := &sqlplugin.TaskListsRow{\n\t\t\tShardID:      dbShardID,\n\t\t\tDomainID:     domainID,\n\t\t\tRangeID:      request.TaskListInfo.RangeID,\n\t\t\tName:         request.TaskListInfo.Name,\n\t\t\tTaskType:     int64(request.TaskListInfo.TaskType),\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t}\n\t\tif m.db.SupportsTTL() && persistence.TaskListKindHasTTL(request.TaskListInfo.Kind) {\n\t\t\tresult, err1 = tx.UpdateTaskListsWithTTL(ctx, &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\tTaskListsRow: *row,\n\t\t\t\tTTL:          taskListTTL,\n\t\t\t})\n\t\t} else {\n\t\t\tresult, err1 = tx.UpdateTaskLists(ctx, row)\n\t\t}\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\trowsAffected, err1 := result.RowsAffected()\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tif rowsAffected != 1 {\n\t\t\treturn fmt.Errorf(\"%v rows were affected instead of 1\", rowsAffected)\n\t\t}\n\t\tresp = &persistence.UpdateTaskListResponse{}\n\t\treturn nil\n\t})\n\treturn resp, err\n}\n\ntype taskListPageToken struct {\n\tShardID  int\n\tDomainID serialization.UUID\n\tName     string\n\tTaskType int64\n}\n\n// ListTaskList lists tasklist from DB\n// DomainID translates into byte array in SQL. The minUUID is not the minimum byte array.\nfunc (m *sqlTaskStore) ListTaskList(\n\tctx context.Context,\n\trequest *persistence.ListTaskListRequest,\n) (*persistence.ListTaskListResponse, error) {\n\tpageToken := taskListPageToken{DomainID: serialization.UUID{}}\n\tif len(request.PageToken) > 0 {\n\t\tif err := gobDeserialize(request.PageToken, &pageToken); err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"error deserializing page token: %v\", err)}\n\t\t}\n\t} else {\n\t\tpageToken = taskListPageToken{TaskType: math.MinInt16, DomainID: serialization.UUID{}}\n\t}\n\tvar err error\n\tvar rows []sqlplugin.TaskListsRow\n\tfor pageToken.ShardID < m.nShards {\n\t\trows, err = m.db.SelectFromTaskLists(ctx, &sqlplugin.TaskListsFilter{\n\t\t\tShardID:             pageToken.ShardID,\n\t\t\tDomainIDGreaterThan: &pageToken.DomainID,\n\t\t\tNameGreaterThan:     &pageToken.Name,\n\t\t\tTaskTypeGreaterThan: &pageToken.TaskType,\n\t\t\tPageSize:            &request.PageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, convertCommonErrors(m.db, \"ListTaskList\", \"\", err)\n\t\t}\n\t\tif len(rows) > 0 {\n\t\t\tbreak\n\t\t}\n\t\tpageToken = taskListPageToken{ShardID: pageToken.ShardID + 1, TaskType: math.MinInt16, DomainID: serialization.UUID{}}\n\t}\n\n\tvar nextPageToken []byte\n\tswitch {\n\tcase len(rows) >= request.PageSize:\n\t\tlastRow := &rows[request.PageSize-1]\n\t\tnextPageToken, err = gobSerialize(&taskListPageToken{\n\t\t\tShardID:  pageToken.ShardID,\n\t\t\tDomainID: lastRow.DomainID,\n\t\t\tName:     lastRow.Name,\n\t\t\tTaskType: lastRow.TaskType,\n\t\t})\n\tcase pageToken.ShardID+1 < m.nShards:\n\t\tnextPageToken, err = gobSerialize(&taskListPageToken{ShardID: pageToken.ShardID + 1, TaskType: math.MinInt16, DomainID: serialization.UUID{}})\n\t}\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"error serializing nextPageToken:%v\", err)}\n\t}\n\n\tresp := &persistence.ListTaskListResponse{\n\t\tItems:         make([]persistence.TaskListInfo, len(rows)),\n\t\tNextPageToken: nextPageToken,\n\t}\n\n\tfor i := range rows {\n\t\tinfo, err := m.parser.TaskListInfoFromBlob(rows[i].Data, rows[i].DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresp.Items[i].DomainID = rows[i].DomainID.String()\n\t\tresp.Items[i].Name = rows[i].Name\n\t\tresp.Items[i].TaskType = int(rows[i].TaskType)\n\t\tresp.Items[i].RangeID = rows[i].RangeID\n\t\tresp.Items[i].Kind = int(info.GetKind())\n\t\tresp.Items[i].AckLevel = info.GetAckLevel()\n\t\tresp.Items[i].Expiry = info.GetExpiryTimestamp()\n\t\tresp.Items[i].LastUpdated = info.GetLastUpdated()\n\t}\n\n\treturn resp, nil\n}\n\nfunc (m *sqlTaskStore) DeleteTaskList(\n\tctx context.Context,\n\trequest *persistence.DeleteTaskListRequest,\n) error {\n\tshardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskListName, m.db.GetTotalNumDBShards())\n\tdomainID := serialization.MustParseUUID(request.DomainID)\n\tresult, err := m.db.DeleteFromTaskLists(ctx, &sqlplugin.TaskListsFilter{\n\t\tShardID:  shardID,\n\t\tDomainID: &domainID,\n\t\tName:     &request.TaskListName,\n\t\tTaskType: common.Int64Ptr(int64(request.TaskListType)),\n\t\tRangeID:  &request.RangeID,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(m.db, \"DeleteTaskList\", \"\", err)\n\t}\n\tnRows, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"rowsAffected returned error:%v\", err)}\n\t}\n\tif nRows != 1 {\n\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"delete failed: %v rows affected instead of 1\", nRows)}\n\t}\n\treturn nil\n}\n\nfunc (m *sqlTaskStore) CreateTasks(\n\tctx context.Context,\n\trequest *persistence.CreateTasksRequest,\n) (*persistence.CreateTasksResponse, error) {\n\tvar tasksRows []sqlplugin.TasksRow\n\tvar tasksRowsWithTTL []sqlplugin.TasksRowWithTTL\n\tif m.db.SupportsTTL() {\n\t\ttasksRowsWithTTL = make([]sqlplugin.TasksRowWithTTL, len(request.Tasks))\n\t} else {\n\t\ttasksRows = make([]sqlplugin.TasksRow, len(request.Tasks))\n\t}\n\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.TaskListInfo.DomainID, request.TaskListInfo.Name, m.db.GetTotalNumDBShards())\n\n\tfor i, v := range request.Tasks {\n\t\tvar expiryTime time.Time\n\t\tvar ttl time.Duration\n\t\tif v.Data.ScheduleToStartTimeoutSeconds > 0 {\n\t\t\tttl = time.Duration(v.Data.ScheduleToStartTimeoutSeconds) * time.Second\n\t\t\tif m.db.SupportsTTL() {\n\t\t\t\tmaxAllowedTTL, err := m.db.MaxAllowedTTL()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\tif ttl > *maxAllowedTTL {\n\t\t\t\t\tttl = *maxAllowedTTL\n\t\t\t\t}\n\t\t\t}\n\t\t\texpiryTime = request.CurrentTimeStamp.Add(ttl)\n\t\t}\n\t\tblob, err := m.parser.TaskInfoToBlob(&serialization.TaskInfo{\n\t\t\tWorkflowID:       v.Data.WorkflowID,\n\t\t\tRunID:            serialization.MustParseUUID(v.Data.RunID),\n\t\t\tScheduleID:       v.Data.ScheduleID,\n\t\t\tExpiryTimestamp:  expiryTime,\n\t\t\tCreatedTimestamp: request.CurrentTimeStamp,\n\t\t\tPartitionConfig:  v.Data.PartitionConfig,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tcurrTasksRow := sqlplugin.TasksRow{\n\t\t\tShardID:      dbShardID,\n\t\t\tDomainID:     serialization.MustParseUUID(v.Data.DomainID),\n\t\t\tTaskListName: request.TaskListInfo.Name,\n\t\t\tTaskType:     int64(request.TaskListInfo.TaskType),\n\t\t\tTaskID:       v.TaskID,\n\t\t\tData:         blob.Data,\n\t\t\tDataEncoding: string(blob.Encoding),\n\t\t}\n\t\tif m.db.SupportsTTL() {\n\t\t\tcurrTasksRowWithTTL := sqlplugin.TasksRowWithTTL{\n\t\t\t\tTasksRow: currTasksRow,\n\t\t\t}\n\t\t\tif ttl > 0 {\n\t\t\t\tcurrTasksRowWithTTL.TTL = &ttl\n\t\t\t}\n\t\t\ttasksRowsWithTTL[i] = currTasksRowWithTTL\n\t\t} else {\n\t\t\ttasksRows[i] = currTasksRow\n\t\t}\n\n\t}\n\tvar resp *persistence.CreateTasksResponse\n\terr := m.txExecute(ctx, dbShardID, \"CreateTasks\", func(tx sqlplugin.Tx) error {\n\t\tif m.db.SupportsTTL() {\n\t\t\tif _, err := tx.InsertIntoTasksWithTTL(ctx, tasksRowsWithTTL); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\tif _, err := tx.InsertIntoTasks(ctx, tasksRows); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\t// Lock task list before committing.\n\t\terr1 := lockTaskList(ctx, tx,\n\t\t\tdbShardID,\n\t\t\tserialization.MustParseUUID(request.TaskListInfo.DomainID),\n\t\t\trequest.TaskListInfo.Name,\n\t\t\trequest.TaskListInfo.TaskType, request.TaskListInfo.RangeID)\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\t\tresp = &persistence.CreateTasksResponse{}\n\t\treturn nil\n\t})\n\treturn resp, err\n}\n\nfunc (m *sqlTaskStore) GetTasks(\n\tctx context.Context,\n\trequest *persistence.GetTasksRequest,\n) (*persistence.GetTasksResponse, error) {\n\tshardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskList, m.db.GetTotalNumDBShards())\n\trows, err := m.db.SelectFromTasks(ctx, &sqlplugin.TasksFilter{\n\t\tShardID:      shardID,\n\t\tDomainID:     serialization.MustParseUUID(request.DomainID),\n\t\tTaskListName: request.TaskList,\n\t\tTaskType:     int64(request.TaskType),\n\t\tMinTaskID:    &request.ReadLevel,\n\t\tMaxTaskID:    request.MaxReadLevel,\n\t\tPageSize:     &request.BatchSize,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetTasks\", \"\", err)\n\t}\n\n\tvar tasks = make([]*persistence.TaskInfo, len(rows))\n\tfor i, v := range rows {\n\t\tinfo, err := m.parser.TaskInfoFromBlob(v.Data, v.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ttasks[i] = &persistence.TaskInfo{\n\t\t\tDomainID:        request.DomainID,\n\t\t\tWorkflowID:      info.GetWorkflowID(),\n\t\t\tRunID:           info.RunID.String(),\n\t\t\tTaskID:          v.TaskID,\n\t\t\tScheduleID:      info.GetScheduleID(),\n\t\t\tExpiry:          info.GetExpiryTimestamp(),\n\t\t\tCreatedTime:     info.GetCreatedTimestamp(),\n\t\t\tPartitionConfig: info.GetPartitionConfig(),\n\t\t}\n\t}\n\n\treturn &persistence.GetTasksResponse{Tasks: tasks}, nil\n}\n\nfunc (m *sqlTaskStore) CompleteTask(\n\tctx context.Context,\n\trequest *persistence.CompleteTaskRequest,\n) error {\n\ttaskID := request.TaskID\n\ttaskList := request.TaskList\n\tshardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(taskList.DomainID, taskList.Name, m.db.GetTotalNumDBShards())\n\t_, err := m.db.DeleteFromTasks(ctx, &sqlplugin.TasksFilter{\n\t\tShardID:      shardID,\n\t\tDomainID:     serialization.MustParseUUID(taskList.DomainID),\n\t\tTaskListName: taskList.Name,\n\t\tTaskType:     int64(taskList.TaskType),\n\t\tTaskID:       &taskID})\n\tif err != nil {\n\t\treturn convertCommonErrors(m.db, \"CompleteTask\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (m *sqlTaskStore) CompleteTasksLessThan(\n\tctx context.Context,\n\trequest *persistence.CompleteTasksLessThanRequest,\n) (*persistence.CompleteTasksLessThanResponse, error) {\n\tshardID := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(request.DomainID, request.TaskListName, m.db.GetTotalNumDBShards())\n\tresult, err := m.db.DeleteFromTasks(ctx, &sqlplugin.TasksFilter{\n\t\tShardID:              shardID,\n\t\tDomainID:             serialization.MustParseUUID(request.DomainID),\n\t\tTaskListName:         request.TaskListName,\n\t\tTaskType:             int64(request.TaskType),\n\t\tTaskIDLessThanEquals: &request.TaskID,\n\t\tLimit:                &request.Limit,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"CompleteTasksLessThan\", \"\", err)\n\t}\n\tnRows, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"rowsAffected returned error: %v\", err),\n\t\t}\n\t}\n\treturn &persistence.CompleteTasksLessThanResponse{TasksCompleted: int(nRows)}, nil\n}\n\n// GetOrphanTasks gets tasks from the tasks table that belong to a task_list no longer present\n// in the task_lists table.\n// TODO: Limit this query to a specific shard at a time. See https://github.com/uber/cadence/issues/4064\nfunc (m *sqlTaskStore) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (*persistence.GetOrphanTasksResponse, error) {\n\trows, err := m.db.GetOrphanTasks(ctx, &sqlplugin.OrphanTasksFilter{\n\t\tLimit: &request.Limit,\n\t})\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(m.db, \"GetOrphanTasks\", \"\", err)\n\t}\n\n\tvar tasks = make([]*persistence.TaskKey, len(rows))\n\tfor i, v := range rows {\n\t\ttasks[i] = &persistence.TaskKey{\n\t\t\tDomainID:     v.DomainID.String(),\n\t\t\tTaskListName: v.TaskListName,\n\t\t\tTaskType:     int(v.TaskType),\n\t\t\tTaskID:       v.TaskID,\n\t\t}\n\t}\n\n\treturn &persistence.GetOrphanTasksResponse{Tasks: tasks}, nil\n}\n\nfunc lockTaskList(ctx context.Context, tx sqlplugin.Tx, shardID int, domainID serialization.UUID, name string, taskListType int, oldRangeID int64) error {\n\trangeID, err := tx.LockTaskLists(ctx, &sqlplugin.TaskListsFilter{\n\t\tShardID: shardID, DomainID: &domainID, Name: &name, TaskType: common.Int64Ptr(int64(taskListType))})\n\n\tswitch err {\n\tcase nil:\n\t\tif rangeID != oldRangeID {\n\t\t\treturn &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"Task list range ID was %v when it was should have been %v\", rangeID, oldRangeID),\n\t\t\t}\n\t\t}\n\t\treturn nil\n\tcase sql.ErrNoRows:\n\t\treturn &persistence.ConditionFailedError{\n\t\t\tMsg: \"Task list does not exist.\",\n\t\t}\n\tdefault:\n\t\treturn convertCommonErrors(tx, \"lockTaskList\", \"\", err)\n\t}\n}\n\nfunc toSerializationTaskListPartitionConfig(c *persistence.TaskListPartitionConfig) *serialization.TaskListPartitionConfig {\n\tif c == nil {\n\t\treturn nil\n\t}\n\treturn &serialization.TaskListPartitionConfig{\n\t\tVersion:            c.Version,\n\t\tNumReadPartitions:  int32(len(c.ReadPartitions)),\n\t\tNumWritePartitions: int32(len(c.WritePartitions)),\n\t\tReadPartitions:     toSerializationTaskListPartitionMap(c.ReadPartitions),\n\t\tWritePartitions:    toSerializationTaskListPartitionMap(c.WritePartitions),\n\t}\n}\n\nfunc toSerializationTaskListPartitionMap(m map[int]*persistence.TaskListPartition) map[int32]*serialization.TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int32]*serialization.TaskListPartition, len(m))\n\tfor id, p := range m {\n\t\tresult[int32(id)] = &serialization.TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\treturn result\n}\n\nfunc fromSerializationTaskListPartitionConfig(c *serialization.TaskListPartitionConfig) *persistence.TaskListPartitionConfig {\n\tif c == nil {\n\t\treturn nil\n\t}\n\tvar read map[int]*persistence.TaskListPartition\n\tif int32(len(c.ReadPartitions)) == c.NumReadPartitions {\n\t\tread = fromSerializationTaskListPartitionMap(c.ReadPartitions)\n\t} else {\n\t\tread = createDefaultPartitions(c.NumReadPartitions)\n\t}\n\tvar write map[int]*persistence.TaskListPartition\n\tif int32(len(c.WritePartitions)) == c.NumWritePartitions {\n\t\twrite = fromSerializationTaskListPartitionMap(c.WritePartitions)\n\t} else {\n\t\twrite = createDefaultPartitions(c.NumWritePartitions)\n\t}\n\treturn &persistence.TaskListPartitionConfig{\n\t\tVersion:         c.Version,\n\t\tReadPartitions:  read,\n\t\tWritePartitions: write,\n\t}\n}\n\nfunc createDefaultPartitions(len int32) map[int]*persistence.TaskListPartition {\n\tpartitions := make(map[int]*persistence.TaskListPartition, len)\n\tfor i := 0; i < int(len); i++ {\n\t\tpartitions[i] = &persistence.TaskListPartition{}\n\t}\n\treturn partitions\n}\n\nfunc fromSerializationTaskListPartitionMap(m map[int32]*serialization.TaskListPartition) map[int]*persistence.TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int]*persistence.TaskListPartition, len(m))\n\tfor id, p := range m {\n\t\tresult[int(id)] = &persistence.TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_task_store_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/base64\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestGetTaskListSize(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetTaskListSizeRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.GetTaskListSizeResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetTaskListSizeRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskListType: 0,\n\t\t\t\tAckLevel:     10,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().GetTasksCount(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tMinTaskID:    common.Int64Ptr(10),\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t},\n\t\t\twant: &persistence.GetTaskListSizeResponse{\n\t\t\t\tSize: 1,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.GetTaskListSizeRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskListType: 0,\n\t\t\t\tAckLevel:     10,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetTasksCount(gomock.Any(), gomock.Any()).Return(int64(0), err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.GetTaskListSize(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLeaseTaskList(t *testing.T) {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.LeaseTaskListRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twant      *persistence.LeaseTaskListResponse\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case - first lease\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:         \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:         \"tl\",\n\t\t\t\tTaskType:         0,\n\t\t\t\tTaskListKind:     persistence.TaskListKindSticky,\n\t\t\t\tRangeID:          0,\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.TaskListInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.Equal(t, int16(persistence.TaskListKindSticky), info.Kind)\n\t\t\t\t\tassert.Equal(t, int64(0), info.AckLevel)\n\t\t\t\t\tassert.WithinDuration(t, now, info.LastUpdated, time.Second)\n\t\t\t\t\treturn persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tAckLevel:        0,\n\t\t\t\t\tKind:            int16(persistence.TaskListKindSticky),\n\t\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\t\tLastUpdated:     time.Unix(0, 1),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), 0).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(0), nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant: &persistence.LeaseTaskListResponse{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindSticky,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - first lease - ephemeral\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:         \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:         \"tl\",\n\t\t\t\tTaskType:         0,\n\t\t\t\tTaskListKind:     persistence.TaskListKindEphemeral,\n\t\t\t\tRangeID:          0,\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.TaskListInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.Equal(t, int16(persistence.TaskListKindEphemeral), info.Kind)\n\t\t\t\t\tassert.Equal(t, int64(0), info.AckLevel)\n\t\t\t\t\tassert.WithinDuration(t, now, info.LastUpdated, time.Second)\n\t\t\t\t\treturn persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tAckLevel:        0,\n\t\t\t\t\tKind:            int16(persistence.TaskListKindEphemeral),\n\t\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\t\tLastUpdated:     time.Unix(0, 1),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), 0).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(0), nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant: &persistence.LeaseTaskListResponse{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindEphemeral,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - first lease - normal tasklist\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindNormal,\n\t\t\t\tRangeID:      0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.TaskListInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.Equal(t, int16(persistence.TaskListKindNormal), info.Kind)\n\t\t\t\t\tassert.Equal(t, int64(0), info.AckLevel)\n\t\t\t\t\treturn persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskLists(gomock.Any(),\n\t\t\t\t\t&sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t).Return(nil, nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tAckLevel:        0,\n\t\t\t\t\tKind:            int16(persistence.TaskListKindSticky),\n\t\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\t\tLastUpdated:     time.Unix(0, 1),\n\t\t\t\t\tAdaptivePartitionConfig: &serialization.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:            0,\n\t\t\t\t\t\tNumReadPartitions:  1,\n\t\t\t\t\t\tReadPartitions:     map[int32]*serialization.TaskListPartition{0: {}},\n\t\t\t\t\t\tNumWritePartitions: 1,\n\t\t\t\t\t\tWritePartitions:    map[int32]*serialization.TaskListPartition{0: {}},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), 0).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(0), nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant: &persistence.LeaseTaskListResponse{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to get tasklist\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindSticky,\n\t\t\t\tRangeID:      0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to insert tasklist\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindSticky,\n\t\t\t\tRangeID:      0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - condition failed\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindSticky,\n\t\t\t\tRangeID:      1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar expectedErr *persistence.ConditionFailedError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"Expected the error to be ConditionFailedError\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to lock tasklist\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindSticky,\n\t\t\t\tRangeID:      0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tAckLevel:        0,\n\t\t\t\t\tKind:            int16(persistence.TaskListKindSticky),\n\t\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\t\tLastUpdated:     time.Unix(0, 1),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), 0).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(0), err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to update tasklists\",\n\t\t\treq: &persistence.LeaseTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskListKind: persistence.TaskListKindSticky,\n\t\t\t\tRangeID:      0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(nil, sql.ErrNoRows)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockDB.EXPECT().InsertIntoTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tAckLevel:        0,\n\t\t\t\t\tKind:            int16(persistence.TaskListKindSticky),\n\t\t\t\t\tExpiryTimestamp: time.Unix(0, 0),\n\t\t\t\t\tLastUpdated:     time.Unix(0, 1),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), 0).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(0), nil)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\n\t\t\tgot, err := store.LeaseTaskList(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tgot.TaskListInfo.LastUpdated = tc.want.TaskListInfo.LastUpdated\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetTaskListRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.GetTaskListResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetTaskListRequest{\n\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList: \"tl\",\n\t\t\t\tTaskType: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\t\t\t\t\tassert.Equal(t, serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"), *filter.DomainID)\n\t\t\t\t\tassert.Equal(t, \"tl\", *filter.Name)\n\t\t\t\t\tassert.Equal(t, int64(1), *filter.TaskType)\n\t\t\t\t\treturn []sqlplugin.TaskListsRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID:      11,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tRangeID:      123,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tKind:            1,\n\t\t\t\t\tAckLevel:        2,\n\t\t\t\t\tExpiryTimestamp: time.Unix(1, 4),\n\t\t\t\t\tLastUpdated:     time.Unix(10, 0),\n\t\t\t\t\tAdaptivePartitionConfig: &serialization.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:            0,\n\t\t\t\t\t\tNumReadPartitions:  1,\n\t\t\t\t\t\tNumWritePartitions: 1,\n\t\t\t\t\t\tReadPartitions: map[int32]*serialization.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tWritePartitions: map[int32]*serialization.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetTaskListResponse{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID:    \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:        \"tl\",\n\t\t\t\t\tTaskType:    1,\n\t\t\t\t\tRangeID:     123,\n\t\t\t\t\tKind:        1,\n\t\t\t\t\tAckLevel:    2,\n\t\t\t\t\tExpiry:      time.Unix(1, 4),\n\t\t\t\t\tLastUpdated: time.Unix(10, 0),\n\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Partition counts instead of data\",\n\t\t\treq: &persistence.GetTaskListRequest{\n\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList: \"tl\",\n\t\t\t\tTaskType: 1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\t\t\t\t\tassert.Equal(t, serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"), *filter.DomainID)\n\t\t\t\t\tassert.Equal(t, \"tl\", *filter.Name)\n\t\t\t\t\tassert.Equal(t, int64(1), *filter.TaskType)\n\t\t\t\t\treturn []sqlplugin.TaskListsRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID:      11,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tRangeID:      123,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tKind:            1,\n\t\t\t\t\tAckLevel:        2,\n\t\t\t\t\tExpiryTimestamp: time.Unix(1, 4),\n\t\t\t\t\tLastUpdated:     time.Unix(10, 0),\n\t\t\t\t\tAdaptivePartitionConfig: &serialization.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:            0,\n\t\t\t\t\t\tNumReadPartitions:  2,\n\t\t\t\t\t\tNumWritePartitions: 2,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetTaskListResponse{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID:    \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:        \"tl\",\n\t\t\t\t\tTaskType:    1,\n\t\t\t\t\tRangeID:     123,\n\t\t\t\t\tKind:        1,\n\t\t\t\t\tAckLevel:    2,\n\t\t\t\t\tExpiry:      time.Unix(1, 4),\n\t\t\t\t\tLastUpdated: time.Unix(10, 0),\n\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t1: {},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t\t1: {},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.GetTaskListRequest{\n\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList: \"tl\",\n\t\t\t\tTaskType: 0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tnShards:  1000,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tgot, err := store.GetTaskList(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTaskList(t *testing.T) {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.UpdateTaskListRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twant      *persistence.UpdateTaskListResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindSticky,\n\t\t\t\t},\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.TaskListInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.WithinDuration(t, now, info.LastUpdated, time.Second)\n\t\t\t\t\treturn persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.UpdateTaskListResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - ephemeral\",\n\t\t\treq: &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindEphemeral,\n\t\t\t\t},\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskListsWithTTL(gomock.Any(), &sqlplugin.TaskListsRowWithTTL{\n\t\t\t\t\tTaskListsRow: sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t\tTTL: taskListTTL,\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.UpdateTaskListResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - normal tasklist\",\n\t\t\treq: &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion: 0,\n\t\t\t\t\t\tReadPartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tWritePartitions: map[int]*persistence.TaskListPartition{\n\t\t\t\t\t\t\t0: {\n\t\t\t\t\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tCurrentTimeStamp: now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).DoAndReturn(func(info *serialization.TaskListInfo) (persistence.DataBlob, error) {\n\t\t\t\t\tassert.Equal(t, int16(persistence.TaskListKindNormal), info.Kind)\n\t\t\t\t\tassert.Equal(t, int64(0), info.AckLevel)\n\t\t\t\t\tassert.WithinDuration(t, now, info.LastUpdated, time.Second)\n\t\t\t\t\tassert.Equal(t, int64(0), info.AdaptivePartitionConfig.Version)\n\t\t\t\t\tassert.Equal(t, int32(1), info.AdaptivePartitionConfig.NumReadPartitions)\n\t\t\t\t\tassert.Equal(t, int32(1), info.AdaptivePartitionConfig.NumWritePartitions)\n\t\t\t\t\tassert.Equal(t, \"foo\", info.AdaptivePartitionConfig.ReadPartitions[0].IsolationGroups[0])\n\t\t\t\t\tassert.Equal(t, \"bar\", info.AdaptivePartitionConfig.WritePartitions[0].IsolationGroups[0])\n\t\t\t\t\treturn persistence.DataBlob{\n\t\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true)\n\t\t\t\tmockTx.EXPECT().UpdateTaskLists(gomock.Any(),\n\t\t\t\t\t&sqlplugin.TaskListsRow{\n\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\tTaskType:     0,\n\t\t\t\t\t\tRangeID:      1,\n\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.UpdateTaskListResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to lock task list\",\n\t\t\treq: &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindSticky,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(1), err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.UpdateTaskListResponse{},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to update task list\",\n\t\t\treq: &persistence.UpdateTaskListRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t\tRangeID:  1,\n\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\tKind:     persistence.TaskListKindSticky,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockParser.EXPECT().TaskListInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t}).Return(int64(1), nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(false)\n\t\t\t\tmockTx.EXPECT().UpdateTaskLists(gomock.Any(), &sqlplugin.TaskListsRow{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tRangeID:      1,\n\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.UpdateTaskListResponse{},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\n\t\t\tgot, err := store.UpdateTaskList(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListTaskList(t *testing.T) {\n\tpageSize := 1\n\ttestCases := []struct {\n\t\tname      string\n\t\tpageToken *taskListPageToken\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.ListTaskListResponse\n\t\twantToken *taskListPageToken\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tpageToken: &taskListPageToken{\n\t\t\t\tShardID:  10,\n\t\t\t\tDomainID: serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\tName:     \"tl\",\n\t\t\t\tTaskType: 0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\t\t\t\t\tassert.Equal(t, 10, filter.ShardID)\n\t\t\t\t\tassert.Equal(t, serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"), *filter.DomainIDGreaterThan)\n\t\t\t\t\tassert.Equal(t, \"tl\", *filter.NameGreaterThan)\n\t\t\t\t\tassert.Equal(t, int64(0), *filter.TaskTypeGreaterThan)\n\t\t\t\t\tassert.Equal(t, 1, *filter.PageSize)\n\t\t\t\t\treturn []sqlplugin.TaskListsRow{}, nil\n\t\t\t\t})\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).DoAndReturn(func(_ context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\t\t\t\t\tassert.Equal(t, 11, filter.ShardID)\n\t\t\t\t\tassert.Equal(t, serialization.UUID{}, *filter.DomainIDGreaterThan)\n\t\t\t\t\tassert.Equal(t, \"\", *filter.NameGreaterThan)\n\t\t\t\t\tassert.Equal(t, int64(math.MinInt16), *filter.TaskTypeGreaterThan)\n\t\t\t\t\tassert.Equal(t, 1, *filter.PageSize)\n\t\t\t\t\treturn []sqlplugin.TaskListsRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID:      11,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tName:         \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tRangeID:      123,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil\n\t\t\t\t})\n\t\t\t\tmockParser.EXPECT().TaskListInfoFromBlob([]byte(`tl`), \"tl\").Return(&serialization.TaskListInfo{\n\t\t\t\t\tKind:            1,\n\t\t\t\t\tAckLevel:        2,\n\t\t\t\t\tExpiryTimestamp: time.Unix(1, 4),\n\t\t\t\t\tLastUpdated:     time.Unix(10, 0),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.ListTaskListResponse{\n\t\t\t\tItems: []persistence.TaskListInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainID:    \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\tName:        \"tl\",\n\t\t\t\t\t\tTaskType:    1,\n\t\t\t\t\t\tRangeID:     123,\n\t\t\t\t\t\tKind:        1,\n\t\t\t\t\t\tAckLevel:    2,\n\t\t\t\t\t\tExpiry:      time.Unix(1, 4),\n\t\t\t\t\t\tLastUpdated: time.Unix(10, 0),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNextPageToken: func() []byte {\n\t\t\t\t\ttoken, err := gobSerialize(&taskListPageToken{\n\t\t\t\t\t\tShardID:  11,\n\t\t\t\t\t\tDomainID: serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\t\tTaskType: 1,\n\t\t\t\t\t})\n\t\t\t\t\trequire.NoError(t, err, \"failed to serialize page token\")\n\t\t\t\t\treturn token\n\t\t\t\t}(),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\tpageToken: &taskListPageToken{\n\t\t\t\tShardID:  10,\n\t\t\t\tDomainID: serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\tName:     \"tl\",\n\t\t\t\tTaskType: 0,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromTaskLists(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t\tnShards:  1000,\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tpageToken, err := gobSerialize(tc.pageToken)\n\t\t\trequire.NoError(t, err, \"invalid pageToken\")\n\t\t\treq := &persistence.ListTaskListRequest{\n\t\t\t\tPageSize:  pageSize,\n\t\t\t\tPageToken: pageToken,\n\t\t\t}\n\t\t\tgot, err := store.ListTaskList(context.Background(), req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.DeleteTaskListRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.DeleteTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskListType: 0,\n\t\t\t\tRangeID:      10,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().DeleteFromTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(0),\n\t\t\t\t\tRangeID:  common.Int64Ptr(10),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.DeleteTaskListRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskListType: 0,\n\t\t\t\tRangeID:      10,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteFromTaskLists(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\terr := store.DeleteTaskList(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.CreateTasksRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *sqlplugin.MockTx, *serialization.MockParser)\n\t\twant      *persistence.CreateTasksResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 1,\n\t\t\t\t\tRangeID:  9,\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\t\t\t\t\tDomainID:                      \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\t\tTaskID:                        999,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskID: 999,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true).Times(4)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().MaxAllowedTTL().Return(common.DurationPtr(time.Hour), nil)\n\t\t\t\tmockParser.EXPECT().TaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoTasksWithTTL(gomock.Any(), []sqlplugin.TasksRowWithTTL{\n\t\t\t\t\t{\n\t\t\t\t\t\tTasksRow: sqlplugin.TasksRow{\n\t\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tTaskID:       999,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTTL: common.DurationPtr(time.Second),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(1),\n\t\t\t\t}).Return(int64(9), nil)\n\t\t\t\tmockTx.EXPECT().Commit().Return(nil)\n\t\t\t},\n\t\t\twant:    &persistence.CreateTasksResponse{},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to insert tasks\",\n\t\t\treq: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 1,\n\t\t\t\t\tRangeID:  9,\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\t\t\t\t\tDomainID:                      \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\t\tTaskID:                        999,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskID: 999,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true).Times(4)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().MaxAllowedTTL().Return(common.DurationPtr(time.Hour), nil)\n\t\t\t\tmockParser.EXPECT().TaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoTasksWithTTL(gomock.Any(), []sqlplugin.TasksRowWithTTL{\n\t\t\t\t\t{\n\t\t\t\t\t\tTasksRow: sqlplugin.TasksRow{\n\t\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tTaskID:       999,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTTL: common.DurationPtr(time.Second),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to lock tasklist\",\n\t\t\treq: &persistence.CreateTasksRequest{\n\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 1,\n\t\t\t\t\tRangeID:  9,\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t},\n\t\t\t\tTasks: []*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tData: &persistence.TaskInfo{\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\t\t\t\t\tDomainID:                      \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\t\tTaskID:                        999,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskID: 999,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockTx *sqlplugin.MockTx, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().SupportsTTL().Return(true).Times(4)\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().MaxAllowedTTL().Return(common.DurationPtr(time.Hour), nil)\n\t\t\t\tmockParser.EXPECT().TaskInfoToBlob(gomock.Any()).Return(persistence.DataBlob{\n\t\t\t\t\tData:     []byte(`tl`),\n\t\t\t\t\tEncoding: constants.EncodingType(\"tl\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockDB.EXPECT().BeginTx(gomock.Any(), gomock.Any()).Return(mockTx, nil)\n\t\t\t\tmockTx.EXPECT().InsertIntoTasksWithTTL(gomock.Any(), []sqlplugin.TasksRowWithTTL{\n\t\t\t\t\t{\n\t\t\t\t\t\tTasksRow: sqlplugin.TasksRow{\n\t\t\t\t\t\t\tShardID:      0,\n\t\t\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\t\tTaskID:       999,\n\t\t\t\t\t\t\tData:         []byte(`tl`),\n\t\t\t\t\t\t\tDataEncoding: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTTL: common.DurationPtr(time.Second),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().LockTaskLists(gomock.Any(), &sqlplugin.TaskListsFilter{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: serialization.UUIDPtr(serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\")),\n\t\t\t\t\tName:     common.StringPtr(\"tl\"),\n\t\t\t\t\tTaskType: common.Int64Ptr(1),\n\t\t\t\t}).Return(int64(0), err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t\tmockTx.EXPECT().Rollback().Return(nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockTx, mockParser)\n\n\t\t\tgot, err := store.CreateTasks(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.GetTasksRequest\n\t\tmockSetup func(*sqlplugin.MockDB, *serialization.MockParser)\n\t\twant      *persistence.GetTasksResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.GetTasksRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tReadLevel:    10,\n\t\t\t\tMaxReadLevel: common.Int64Ptr(9999),\n\t\t\t\tBatchSize:    1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().SelectFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tMinTaskID:    common.Int64Ptr(10),\n\t\t\t\t\tMaxTaskID:    common.Int64Ptr(9999),\n\t\t\t\t\tPageSize:     common.IntPtr(1),\n\t\t\t\t}).Return([]sqlplugin.TasksRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskID:       888,\n\t\t\t\t\t\tData:         []byte(`task`),\n\t\t\t\t\t\tDataEncoding: \"task\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockParser.EXPECT().TaskInfoFromBlob([]byte(`task`), \"task\").Return(&serialization.TaskInfo{\n\t\t\t\t\tWorkflowID:       \"test\",\n\t\t\t\t\tRunID:            serialization.MustParseUUID(\"b9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tScheduleID:       7,\n\t\t\t\t\tExpiryTimestamp:  time.Unix(9, 0),\n\t\t\t\t\tCreatedTimestamp: time.Unix(8, 7),\n\t\t\t\t\tPartitionConfig:  map[string]string{\"a\": \"b\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetTasksResponse{\n\t\t\t\tTasks: []*persistence.TaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainID:        \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\tWorkflowID:      \"test\",\n\t\t\t\t\t\tRunID:           \"b9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\t\tTaskID:          888,\n\t\t\t\t\t\tScheduleID:      7,\n\t\t\t\t\t\tExpiry:          time.Unix(9, 0),\n\t\t\t\t\t\tCreatedTime:     time.Unix(8, 7),\n\t\t\t\t\t\tPartitionConfig: map[string]string{\"a\": \"b\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.GetTasksRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskList:     \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tReadLevel:    10,\n\t\t\t\tMaxReadLevel: common.Int64Ptr(9999),\n\t\t\t\tBatchSize:    1,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB, mockParser *serialization.MockParser) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().SelectFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tMinTaskID:    common.Int64Ptr(10),\n\t\t\t\t\tMaxTaskID:    common.Int64Ptr(9999),\n\t\t\t\t\tPageSize:     common.IntPtr(1),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tmockParser := serialization.NewMockParser(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB, parser: mockParser},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB, mockParser)\n\n\t\t\tgot, err := store.GetTasks(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCompleteTask(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.CompleteTaskRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.CompleteTaskRequest{\n\t\t\t\tTaskList: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t},\n\t\t\t\tTaskID: 1001,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().DeleteFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tTaskID:       common.Int64Ptr(1001),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 1}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.CompleteTaskRequest{\n\t\t\t\tTaskList: &persistence.TaskListInfo{\n\t\t\t\t\tDomainID: \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\tTaskType: 0,\n\t\t\t\t},\n\t\t\t\tTaskID: 1001,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:      0,\n\t\t\t\t\tDomainID:     serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName: \"tl\",\n\t\t\t\t\tTaskType:     0,\n\t\t\t\t\tTaskID:       common.Int64Ptr(1001),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\terr := store.CompleteTask(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCompleteTaskLessThan(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *persistence.CompleteTasksLessThanRequest\n\t\tmockSetup func(*sqlplugin.MockDB)\n\t\twant      *persistence.CompleteTasksLessThanResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\treq: &persistence.CompleteTasksLessThanRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskID:       1001,\n\t\t\t\tLimit:        100,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\tmockDB.EXPECT().DeleteFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:              0,\n\t\t\t\t\tDomainID:             serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName:         \"tl\",\n\t\t\t\t\tTaskType:             0,\n\t\t\t\t\tTaskIDLessThanEquals: common.Int64Ptr(1001),\n\t\t\t\t\tLimit:                common.IntPtr(100),\n\t\t\t\t}).Return(&sqlResult{rowsAffected: 100}, nil)\n\t\t\t},\n\t\t\twant:    &persistence.CompleteTasksLessThanResponse{TasksCompleted: 100},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\treq: &persistence.CompleteTasksLessThanRequest{\n\t\t\t\tDomainID:     \"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\",\n\t\t\t\tTaskListName: \"tl\",\n\t\t\t\tTaskType:     0,\n\t\t\t\tTaskID:       1001,\n\t\t\t\tLimit:        100,\n\t\t\t},\n\t\t\tmockSetup: func(mockDB *sqlplugin.MockDB) {\n\t\t\t\tmockDB.EXPECT().GetTotalNumDBShards().Return(1)\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockDB.EXPECT().DeleteFromTasks(gomock.Any(), &sqlplugin.TasksFilter{\n\t\t\t\t\tShardID:              0,\n\t\t\t\t\tDomainID:             serialization.MustParseUUID(\"c9488dc7-20b2-44c3-b2e4-bfea5af62ac0\"),\n\t\t\t\t\tTaskListName:         \"tl\",\n\t\t\t\t\tTaskType:             0,\n\t\t\t\t\tTaskIDLessThanEquals: common.Int64Ptr(1001),\n\t\t\t\t\tLimit:                common.IntPtr(100),\n\t\t\t\t}).Return(nil, err)\n\t\t\t\tmockDB.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDB := sqlplugin.NewMockDB(ctrl)\n\t\t\tstore := &sqlTaskStore{\n\t\t\t\tsqlStore: sqlStore{db: mockDB},\n\t\t\t}\n\n\t\t\ttc.mockSetup(mockDB)\n\n\t\t\tgot, err := store.CompleteTasksLessThan(context.Background(), tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Unexpected result for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHelp(t *testing.T) {\n\tshard := sqlplugin.GetDBShardIDFromDomainIDAndTasklist(\"c4b5cb22-c213-4812-bb4a-fc1ade5405ef\", \"pgtasklist\", 16384)\n\tprintln(\"shard: \", shard)\n\t// BgAKAAAKAAwAAAAABQ+kWAoADgAAAAAAAAAACgAQGB6uFsU9cvEMABIKAAoAAAAAAAAAAQgADAAAAAAIAA4AAAAAAAA=\n\tdc := &persistence.DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t}\n\tparser, err := serialization.NewParser(dc)\n\trequire.NoError(t, err)\n\tdata, err := base64.StdEncoding.DecodeString(\"BgAKAAAKAAwAAAAABGGFYAoADgAAAAAAAAAACgAQGB6uGPaVqOUMABIKAAoAAAAAAAAAAQgADAAAAAIIAA4AAAACAAA=\")\n\trequire.NoError(t, err)\n\tinfo, err := parser.TaskListInfoFromBlob(data, \"thriftrw\")\n\trequire.NoError(t, err)\n\tfmt.Printf(\"info: %v\", info)\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_test_utils.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"log\"\n\t\"os\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n\t\"github.com/uber/cadence/environment\"\n)\n\n// testCluster allows executing cassandra operations in testing.\ntype testCluster struct {\n\tdbName    string\n\tschemaDir string\n\tcfg       config.SQL\n}\n\nvar _ testcluster.PersistenceTestCluster = (*testCluster)(nil)\n\n// NewTestCluster returns a new SQL test cluster\nfunc NewTestCluster(pluginName, dbName, username, password, host string, port int, schemaDir string) (testcluster.PersistenceTestCluster, error) {\n\tvar result testCluster\n\tvar err error\n\tif port == 0 {\n\t\tport, err = environment.GetMySQLPort()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tif schemaDir == \"\" {\n\t\treturn nil, errors.New(\"schemaDir is empty\")\n\t}\n\tresult.dbName = dbName\n\tresult.schemaDir = schemaDir\n\tresult.cfg = config.SQL{\n\t\tUser:            username,\n\t\tPassword:        password,\n\t\tConnectAddr:     fmt.Sprintf(\"%v:%v\", host, port),\n\t\tConnectProtocol: \"tcp\",\n\t\tPluginName:      pluginName,\n\t\tDatabaseName:    dbName,\n\t\tNumShards:       4,\n\t\tEncodingType:    \"thriftrw\",\n\t\tDecodingTypes:   []string{\"thriftrw\"},\n\t}\n\treturn &result, nil\n}\n\n// DatabaseName from PersistenceTestCluster interface\nfunc (s *testCluster) DatabaseName() string {\n\treturn s.dbName\n}\n\n// SetupTestDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) SetupTestDatabase() {\n\ts.createDatabase()\n\n\tschemaDir := s.schemaDir + \"/\"\n\tif !strings.HasPrefix(schemaDir, \"/\") && !strings.HasPrefix(schemaDir, \"../\") {\n\t\tcadencePackageDir, err := getCadencePackageDir()\n\t\tif err != nil {\n\t\t\tlog.Fatal(err)\n\t\t}\n\t\tschemaDir = cadencePackageDir + schemaDir\n\t}\n\ts.loadSchema([]string{\"schema.sql\"}, schemaDir)\n\ts.loadVisibilitySchema([]string{\"schema.sql\"}, schemaDir)\n}\n\n// Config returns the persistence config for connecting to this test cluster\nfunc (s *testCluster) Config() config.Persistence {\n\tcfg := s.cfg\n\treturn config.Persistence{\n\t\tDefaultStore:    \"test\",\n\t\tVisibilityStore: \"test\",\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"test\": {SQL: &cfg},\n\t\t},\n\t\tTransactionSizeLimit: dynamicproperties.GetIntPropertyFn(constants.DefaultTransactionSizeLimit),\n\t\tErrorInjectionRate:   dynamicproperties.GetFloatPropertyFn(0),\n\t\tNumHistoryShards:     s.cfg.NumShards,\n\t}\n}\n\n// TearDownTestDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) TearDownTestDatabase() {\n\ts.dropDatabase()\n}\n\n// createDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) createDatabase() {\n\tif s.cfg.PluginName == \"sqlite\" {\n\t\t// sqlite doesn't support creating database\n\t\treturn\n\t}\n\n\tcfg2 := s.cfg\n\t// NOTE need to connect with empty name to create new database\n\tcfg2.DatabaseName = \"\"\n\tdb, err := NewSQLAdminDB(&cfg2)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tdefer func() {\n\t\terr := db.Close()\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\t}()\n\n\terr = db.CreateDatabase(s.cfg.DatabaseName)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n}\n\n// dropDatabase from PersistenceTestCluster interface\nfunc (s *testCluster) dropDatabase() {\n\tif s.cfg.PluginName == \"sqlite\" {\n\t\t// sqlite doesn't support dropping database\n\t\treturn\n\t}\n\n\tcfg2 := s.cfg\n\t// NOTE need to connect with empty name to drop the database\n\tcfg2.DatabaseName = \"\"\n\tdb, err := NewSQLAdminDB(&cfg2)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tdefer func() {\n\t\terr := db.Close()\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\t}()\n\terr = db.DropDatabase(s.cfg.DatabaseName)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n}\n\n// loadSchema from PersistenceTestCluster interface\nfunc (s *testCluster) loadSchema(fileNames []string, schemaDir string) {\n\tworkflowSchemaDir := schemaDir + \"/cadence\"\n\terr := s.loadDatabaseSchema(workflowSchemaDir, fileNames, true)\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n}\n\n// loadVisibilitySchema from PersistenceTestCluster interface\nfunc (s *testCluster) loadVisibilitySchema(fileNames []string, schemaDir string) {\n\tworkflowSchemaDir := schemaDir + \"/visibility\"\n\terr := s.loadDatabaseSchema(workflowSchemaDir, fileNames, true)\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n}\n\nfunc getCadencePackageDir() (string, error) {\n\tcadencePackageDir, err := os.Getwd()\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tcadenceIndex := strings.LastIndex(cadencePackageDir, \"cadence/\")\n\tcadencePackageDir = cadencePackageDir[:cadenceIndex+len(\"cadence/\")]\n\treturn cadencePackageDir, err\n}\n\n// loadDatabaseSchema loads the schema from the given .sql files on this database\nfunc (s *testCluster) loadDatabaseSchema(dir string, fileNames []string, override bool) (err error) {\n\tdb, err := NewSQLAdminDB(&s.cfg)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tdefer func() {\n\t\terr := db.Close()\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\t}()\n\n\tfor _, file := range fileNames {\n\t\t// This is only used in tests. Excluding it from security scanners\n\t\t// #nosec\n\t\tcontent, err := ioutil.ReadFile(dir + \"/\" + file)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"error reading contents of file %v:%v\", file, err.Error())\n\t\t}\n\t\terr = db.ExecSchemaOperationQuery(context.Background(), string(content))\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"error loading schema from %v: %v\", file, err.Error())\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sql_visibility_store.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\ntype (\n\tsqlVisibilityStore struct {\n\t\tsqlStore\n\t}\n\n\tvisibilityPageToken struct {\n\t\tTime  time.Time\n\t\tRunID string\n\t}\n)\n\n// NewSQLVisibilityStore creates an instance of ExecutionStore\nfunc NewSQLVisibilityStore(cfg config.SQL, logger log.Logger) (p.VisibilityStore, error) {\n\tdb, err := NewSQLDB(&cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &sqlVisibilityStore{\n\t\tsqlStore: sqlStore{\n\t\t\tdb:     db,\n\t\t\tlogger: logger,\n\t\t},\n\t}, nil\n}\n\nfunc (s *sqlVisibilityStore) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionStartedRequest,\n) error {\n\t_, err := s.db.InsertIntoVisibility(ctx, &sqlplugin.VisibilityRow{\n\t\tDomainID:               request.DomainUUID,\n\t\tWorkflowID:             request.WorkflowID,\n\t\tRunID:                  request.RunID,\n\t\tStartTime:              request.StartTimestamp,\n\t\tExecutionTime:          request.ExecutionTimestamp,\n\t\tWorkflowTypeName:       request.WorkflowTypeName,\n\t\tMemo:                   request.Memo.Data,\n\t\tEncoding:               string(request.Memo.GetEncoding()),\n\t\tIsCron:                 request.IsCron,\n\t\tCronSchedule:           request.CronSchedule,\n\t\tNumClusters:            request.NumClusters,\n\t\tUpdateTime:             request.UpdateTimestamp,\n\t\tShardID:                request.ShardID,\n\t\tExecutionStatus:        int32(request.ExecutionStatus),\n\t\tScheduledExecutionTime: request.ScheduledExecutionTime,\n\t})\n\n\tif err != nil {\n\t\treturn convertCommonErrors(s.db, \"RecordWorkflowExecutionStarted\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (s *sqlVisibilityStore) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionClosedRequest,\n) error {\n\tcloseTime := request.CloseTimestamp\n\n\t// Map CloseStatus to ExecutionStatus\n\texecutionStatus := types.WorkflowExecutionStatusCompleted // default\n\tswitch request.Status {\n\tcase types.WorkflowExecutionCloseStatusCompleted:\n\t\texecutionStatus = types.WorkflowExecutionStatusCompleted\n\tcase types.WorkflowExecutionCloseStatusFailed:\n\t\texecutionStatus = types.WorkflowExecutionStatusFailed\n\tcase types.WorkflowExecutionCloseStatusCanceled:\n\t\texecutionStatus = types.WorkflowExecutionStatusCanceled\n\tcase types.WorkflowExecutionCloseStatusTerminated:\n\t\texecutionStatus = types.WorkflowExecutionStatusTerminated\n\tcase types.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\texecutionStatus = types.WorkflowExecutionStatusContinuedAsNew\n\tcase types.WorkflowExecutionCloseStatusTimedOut:\n\t\texecutionStatus = types.WorkflowExecutionStatusTimedOut\n\t}\n\n\tresult, err := s.db.ReplaceIntoVisibility(ctx, &sqlplugin.VisibilityRow{\n\t\tDomainID:               request.DomainUUID,\n\t\tWorkflowID:             request.WorkflowID,\n\t\tRunID:                  request.RunID,\n\t\tStartTime:              request.StartTimestamp,\n\t\tExecutionTime:          request.ExecutionTimestamp,\n\t\tWorkflowTypeName:       request.WorkflowTypeName,\n\t\tCloseTime:              &closeTime,\n\t\tCloseStatus:            common.Int32Ptr(int32(*thrift.FromWorkflowExecutionCloseStatus(&request.Status))),\n\t\tHistoryLength:          &request.HistoryLength,\n\t\tMemo:                   request.Memo.Data,\n\t\tEncoding:               string(request.Memo.GetEncoding()),\n\t\tIsCron:                 request.IsCron,\n\t\tCronSchedule:           request.CronSchedule,\n\t\tNumClusters:            request.NumClusters,\n\t\tUpdateTime:             request.UpdateTimestamp,\n\t\tShardID:                request.ShardID,\n\t\tExecutionStatus:        int32(executionStatus),\n\t\tScheduledExecutionTime: request.ScheduledExecutionTime,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(s.db, \"RecordWorkflowExecutionClosed\", \"\", err)\n\t}\n\tnoRowsAffected, err := result.RowsAffected()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"RecordWorkflowExecutionClosed rowsAffected error: %v\", err),\n\t\t}\n\t}\n\tif noRowsAffected > 2 { // either adds a new row or deletes old row and adds new row\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"RecordWorkflowExecutionClosed unexpected numRows (%v) updated\", noRowsAffected),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (s *sqlVisibilityStore) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *p.InternalRecordWorkflowExecutionUninitializedRequest,\n) error {\n\t// temporary: not implemented, only implemented for ES\n\treturn nil\n}\n\nfunc (s *sqlVisibilityStore) UpsertWorkflowExecution(\n\t_ context.Context,\n\trequest *p.InternalUpsertWorkflowExecutionRequest,\n) error {\n\tif p.IsNopUpsertWorkflowRequest(request) {\n\t\treturn nil\n\t}\n\treturn p.ErrVisibilityOperationNotSupported\n}\n\nfunc (s *sqlVisibilityStore) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListOpenWorkflowExecutions\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:     request.DomainUUID,\n\t\t\t\tMinStartTime: &request.EarliestTime,\n\t\t\t\tMaxStartTime: &readLevel.Time,\n\t\t\t\tRunID:        &readLevel.RunID,\n\t\t\t\tPageSize:     &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListClosedWorkflowExecutions\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:     request.DomainUUID,\n\t\t\t\tMinStartTime: &request.EarliestTime,\n\t\t\t\tMaxStartTime: &readLevel.Time,\n\t\t\t\tClosed:       true,\n\t\t\t\tRunID:        &readLevel.RunID,\n\t\t\t\tPageSize:     &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByTypeRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListOpenWorkflowExecutionsByType\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:         request.DomainUUID,\n\t\t\t\tMinStartTime:     &request.EarliestTime,\n\t\t\t\tMaxStartTime:     &readLevel.Time,\n\t\t\t\tRunID:            &readLevel.RunID,\n\t\t\t\tWorkflowTypeName: &request.WorkflowTypeName,\n\t\t\t\tPageSize:         &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByTypeRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListClosedWorkflowExecutionsByType\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:         request.DomainUUID,\n\t\t\t\tMinStartTime:     &request.EarliestTime,\n\t\t\t\tMaxStartTime:     &readLevel.Time,\n\t\t\t\tClosed:           true,\n\t\t\t\tRunID:            &readLevel.RunID,\n\t\t\t\tWorkflowTypeName: &request.WorkflowTypeName,\n\t\t\t\tPageSize:         &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListOpenWorkflowExecutionsByWorkflowID\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:     request.DomainUUID,\n\t\t\t\tMinStartTime: &request.EarliestTime,\n\t\t\t\tMaxStartTime: &readLevel.Time,\n\t\t\t\tRunID:        &readLevel.RunID,\n\t\t\t\tWorkflowID:   &request.WorkflowID,\n\t\t\t\tPageSize:     &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *p.InternalListWorkflowExecutionsByWorkflowIDRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListClosedWorkflowExecutionsByWorkflowID\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:     request.DomainUUID,\n\t\t\t\tMinStartTime: &request.EarliestTime,\n\t\t\t\tMaxStartTime: &readLevel.Time,\n\t\t\t\tClosed:       true,\n\t\t\t\tRunID:        &readLevel.RunID,\n\t\t\t\tWorkflowID:   &request.WorkflowID,\n\t\t\t\tPageSize:     &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *p.InternalListClosedWorkflowExecutionsByStatusRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn s.listWorkflowExecutions(\"ListClosedWorkflowExecutionsByStatus\", request.NextPageToken, request.EarliestTime, request.LatestTime,\n\t\tfunc(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error) {\n\t\t\treturn s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\t\t\tDomainID:     request.DomainUUID,\n\t\t\t\tMinStartTime: &request.EarliestTime,\n\t\t\t\tMaxStartTime: &readLevel.Time,\n\t\t\t\tClosed:       true,\n\t\t\t\tRunID:        &readLevel.RunID,\n\t\t\t\tCloseStatus:  common.Int32Ptr(int32(*thrift.FromWorkflowExecutionCloseStatus(&request.Status))),\n\t\t\t\tPageSize:     &request.PageSize,\n\t\t\t})\n\t\t})\n}\n\nfunc (s *sqlVisibilityStore) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.InternalGetClosedWorkflowExecutionRequest,\n) (*p.InternalGetClosedWorkflowExecutionResponse, error) {\n\texecution := request.Execution\n\trows, err := s.db.SelectFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\tDomainID: request.DomainUUID,\n\t\tClosed:   true,\n\t\tRunID:    &execution.RunID,\n\t})\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: fmt.Sprintf(\"Workflow execution not found.  WorkflowId: %v, RunId: %v\",\n\t\t\t\t\texecution.GetWorkflowID(), execution.GetRunID()),\n\t\t\t}\n\t\t}\n\t\treturn nil, convertCommonErrors(s.db, \"GetClosedWorkflowExecution\", \"\", err)\n\t}\n\trows[0].DomainID = request.DomainUUID\n\trows[0].RunID = execution.GetRunID()\n\trows[0].WorkflowID = execution.GetWorkflowID()\n\treturn &p.InternalGetClosedWorkflowExecutionResponse{Execution: s.rowToInfo(&rows[0])}, nil\n}\n\nfunc (s *sqlVisibilityStore) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\t_, err := s.db.DeleteFromVisibility(ctx, &sqlplugin.VisibilityFilter{\n\t\tDomainID: request.DomainID,\n\t\tRunID:    &request.RunID,\n\t})\n\tif err != nil {\n\t\treturn convertCommonErrors(s.db, \"DeleteWorkflowExecution\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc (s *sqlVisibilityStore) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *p.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\t// temporary: not implemented, only implemented for ES\n\treturn nil\n}\n\nfunc (s *sqlVisibilityStore) ListWorkflowExecutions(\n\t_ context.Context,\n\t_ *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn nil, p.ErrVisibilityOperationNotSupported\n}\n\nfunc (s *sqlVisibilityStore) ScanWorkflowExecutions(\n\t_ context.Context,\n\t_ *p.ListWorkflowExecutionsByQueryRequest,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\treturn nil, p.ErrVisibilityOperationNotSupported\n}\n\nfunc (s *sqlVisibilityStore) CountWorkflowExecutions(\n\t_ context.Context,\n\t_ *p.CountWorkflowExecutionsRequest,\n) (*p.CountWorkflowExecutionsResponse, error) {\n\treturn nil, p.ErrVisibilityOperationNotSupported\n}\n\nfunc (s *sqlVisibilityStore) rowToInfo(row *sqlplugin.VisibilityRow) *p.InternalVisibilityWorkflowExecutionInfo {\n\tif row.ExecutionTime.UnixNano() == 0 {\n\t\trow.ExecutionTime = row.StartTime\n\t}\n\tinfo := &p.InternalVisibilityWorkflowExecutionInfo{\n\t\tWorkflowID:             row.WorkflowID,\n\t\tRunID:                  row.RunID,\n\t\tTypeName:               row.WorkflowTypeName,\n\t\tStartTime:              row.StartTime,\n\t\tExecutionTime:          row.ExecutionTime,\n\t\tIsCron:                 row.IsCron,\n\t\tCronSchedule:           row.CronSchedule,\n\t\tNumClusters:            row.NumClusters,\n\t\tMemo:                   p.NewDataBlob(row.Memo, constants.EncodingType(row.Encoding)),\n\t\tUpdateTime:             row.UpdateTime,\n\t\tShardID:                row.ShardID,\n\t\tExecutionStatus:        types.WorkflowExecutionStatus(row.ExecutionStatus),\n\t\tScheduledExecutionTime: row.ScheduledExecutionTime,\n\t}\n\tif row.CloseStatus != nil {\n\t\tstatus := workflow.WorkflowExecutionCloseStatus(*row.CloseStatus)\n\t\tinfo.Status = thrift.ToWorkflowExecutionCloseStatus(&status)\n\t\tinfo.CloseTime = *row.CloseTime\n\t\tinfo.HistoryLength = *row.HistoryLength\n\t}\n\treturn info\n}\n\nfunc (s *sqlVisibilityStore) listWorkflowExecutions(opName string, pageToken []byte, earliestTime time.Time, latestTime time.Time, selectOp func(readLevel *visibilityPageToken) ([]sqlplugin.VisibilityRow, error)) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tvar readLevel *visibilityPageToken\n\tvar err error\n\tif len(pageToken) > 0 {\n\t\treadLevel, err = s.deserializePageToken(pageToken)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\treadLevel = &visibilityPageToken{Time: latestTime, RunID: \"\"}\n\t}\n\trows, err := selectOp(readLevel)\n\tif err != nil {\n\t\treturn nil, convertCommonErrors(s.db, opName, \"\", err)\n\t}\n\tif len(rows) == 0 {\n\t\treturn &p.InternalListWorkflowExecutionsResponse{}, nil\n\t}\n\n\tvar infos = make([]*p.InternalVisibilityWorkflowExecutionInfo, len(rows))\n\tfor i, row := range rows {\n\t\tinfos[i] = s.rowToInfo(&row)\n\t}\n\tvar nextPageToken []byte\n\tlastRow := rows[len(rows)-1]\n\tlastStartTime := lastRow.StartTime\n\tif lastStartTime.Sub(earliestTime).Nanoseconds() > 0 {\n\t\tnextPageToken, err = s.serializePageToken(&visibilityPageToken{\n\t\t\tTime:  lastStartTime,\n\t\t\tRunID: lastRow.RunID,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn &p.InternalListWorkflowExecutionsResponse{\n\t\tExecutions:    infos,\n\t\tNextPageToken: nextPageToken,\n\t}, nil\n}\n\nfunc (s *sqlVisibilityStore) deserializePageToken(data []byte) (*visibilityPageToken, error) {\n\tvar token visibilityPageToken\n\terr := json.Unmarshal(data, &token)\n\treturn &token, err\n}\n\nfunc (s *sqlVisibilityStore) serializePageToken(token *visibilityPageToken) ([]byte, error) {\n\tdata, err := json.Marshal(token)\n\treturn data, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/connections.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqldriver\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\ntype CreateSingleDBConn func(cfg *config.SQL) (*sqlx.DB, error)\n\n// CreateDBConnections returns references to logical connections to the underlying SQL databases.\n// By default when UseMultipleDatabases == false, the returned object is to tied to a single\n// SQL database and the object can be used to perform CRUD operations on the tables in the database.\n// If UseMultipleDatabases == true then return connections to all the databases\nfunc CreateDBConnections(cfg *config.SQL, createConnFunc CreateSingleDBConn) ([]*sqlx.DB, error) {\n\tif !cfg.UseMultipleDatabases {\n\t\txdb, err := createConnFunc(cfg)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn []*sqlx.DB{xdb}, nil\n\t}\n\tif cfg.NumShards <= 1 || len(cfg.MultipleDatabasesConfig) != cfg.NumShards {\n\t\treturn nil, fmt.Errorf(\"invalid SQL config. NumShards should be > 1 and equal to the length of MultipleDatabasesConfig\")\n\t}\n\n\t// recover from the original at the end\n\tdefer func() {\n\t\tcfg.User = \"\"\n\t\tcfg.Password = \"\"\n\t\tcfg.DatabaseName = \"\"\n\t\tcfg.ConnectAddr = \"\"\n\t}()\n\n\txdbs := make([]*sqlx.DB, cfg.NumShards)\n\tfor idx, entry := range cfg.MultipleDatabasesConfig {\n\t\tcfg.User = entry.User\n\t\tcfg.Password = entry.Password\n\t\tcfg.DatabaseName = entry.DatabaseName\n\t\tcfg.ConnectAddr = entry.ConnectAddr\n\t\txdb, err := createConnFunc(cfg)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"got error of %v to connect to %v database with config %v\", err, idx, cfg)\n\t\t}\n\t\txdbs[idx] = xdb\n\t}\n\treturn xdbs, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/driver.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqldriver\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/jmoiron/sqlx\"\n)\n\n// NewDriver returns a driver to SQL, either using singleton Driver or sharded Driver\nfunc NewDriver(xdbs []*sqlx.DB, tx *sqlx.Tx, dbShardID int) (Driver, error) {\n\n\tif len(xdbs) == 1 {\n\t\treturn newSingletonSQLDriver(xdbs[0], tx, dbShardID), nil\n\t}\n\n\tif len(xdbs) <= 1 {\n\t\treturn nil, fmt.Errorf(\"invalid number of connection for sharded SQL driver\")\n\t}\n\t// this is the case of multiple database with sharding\n\treturn newShardedSQLDriver(xdbs, tx, dbShardID), nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/interface.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/sql/sqldriver\n\npackage sqldriver\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/jmoiron/sqlx\"\n)\n\ntype (\n\t// Driver interface is an abstraction to query SQL.\n\t// The layer is added so that we can have a adapter to support multiple SQL databases behind a single Cadence cluster\n\tDriver interface {\n\n\t\t// shared methods are for both non-transactional (using sqlx.DB) and transactional (using sqlx.Tx) operation --\n\t\t// if a transaction is started(using BeginTxx), then query are executed in the transaction mode. Otherwise executed in normal mode.\n\t\tcommonOfDbAndTx\n\n\t\t// BeginTxx starts a new transaction in the shard of dbShardID\n\t\tBeginTxx(ctx context.Context, dbShardID int, opts *sql.TxOptions) (*sqlx.Tx, error)\n\t\t// Commit commits the current transaction(started by BeginTxx)\n\t\tCommit() error\n\t\t// Rollback rollbacks the current transaction(started by BeginTxx)\n\t\tRollback() error\n\t\t// Close closes this driver(and underlying connections)\n\t\tClose() error\n\n\t\t// ExecDDL executes a DDL query\n\t\tExecDDL(ctx context.Context, dbShardID int, query string, args ...interface{}) (sql.Result, error)\n\t\t// SelectForSchemaQuery executes a select query for schema(returning multiple rows).\n\t\tSelectForSchemaQuery(dbShardID int, dest interface{}, query string, args ...interface{}) error\n\t\t// GetForSchemaQuery executes a get query for schema(returning single row).\n\t\tGetForSchemaQuery(dbShardID int, dest interface{}, query string, args ...interface{}) error\n\t}\n\n\t// the methods can be executed from either a started or transaction(then need to call Commit/Rollback), or without a transaction\n\tcommonOfDbAndTx interface {\n\t\tExecContext(ctx context.Context, dbShardID int, query string, args ...interface{}) (sql.Result, error)\n\t\tNamedExecContext(ctx context.Context, dbShardID int, query string, arg interface{}) (sql.Result, error)\n\t\tGetContext(ctx context.Context, dbShardID int, dest interface{}, query string, args ...interface{}) error\n\t\tSelectContext(ctx context.Context, dbShardID int, dest interface{}, query string, args ...interface{}) error\n\t}\n)\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package sqldriver -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/sql/sqldriver\n//\n\n// Package sqldriver is a generated GoMock package.\npackage sqldriver\n\nimport (\n\tcontext \"context\"\n\tsql \"database/sql\"\n\treflect \"reflect\"\n\n\tsqlx \"github.com/jmoiron/sqlx\"\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockDriver is a mock of Driver interface.\ntype MockDriver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDriverMockRecorder\n\tisgomock struct{}\n}\n\n// MockDriverMockRecorder is the mock recorder for MockDriver.\ntype MockDriverMockRecorder struct {\n\tmock *MockDriver\n}\n\n// NewMockDriver creates a new mock instance.\nfunc NewMockDriver(ctrl *gomock.Controller) *MockDriver {\n\tmock := &MockDriver{ctrl: ctrl}\n\tmock.recorder = &MockDriverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDriver) EXPECT() *MockDriverMockRecorder {\n\treturn m.recorder\n}\n\n// BeginTxx mocks base method.\nfunc (m *MockDriver) BeginTxx(ctx context.Context, dbShardID int, opts *sql.TxOptions) (*sqlx.Tx, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"BeginTxx\", ctx, dbShardID, opts)\n\tret0, _ := ret[0].(*sqlx.Tx)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// BeginTxx indicates an expected call of BeginTxx.\nfunc (mr *MockDriverMockRecorder) BeginTxx(ctx, dbShardID, opts any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"BeginTxx\", reflect.TypeOf((*MockDriver)(nil).BeginTxx), ctx, dbShardID, opts)\n}\n\n// Close mocks base method.\nfunc (m *MockDriver) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDriverMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDriver)(nil).Close))\n}\n\n// Commit mocks base method.\nfunc (m *MockDriver) Commit() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Commit\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Commit indicates an expected call of Commit.\nfunc (mr *MockDriverMockRecorder) Commit() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Commit\", reflect.TypeOf((*MockDriver)(nil).Commit))\n}\n\n// ExecContext mocks base method.\nfunc (m *MockDriver) ExecContext(ctx context.Context, dbShardID int, query string, args ...any) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecContext\", varargs...)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ExecContext indicates an expected call of ExecContext.\nfunc (mr *MockDriverMockRecorder) ExecContext(ctx, dbShardID, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecContext\", reflect.TypeOf((*MockDriver)(nil).ExecContext), varargs...)\n}\n\n// ExecDDL mocks base method.\nfunc (m *MockDriver) ExecDDL(ctx context.Context, dbShardID int, query string, args ...any) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecDDL\", varargs...)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ExecDDL indicates an expected call of ExecDDL.\nfunc (mr *MockDriverMockRecorder) ExecDDL(ctx, dbShardID, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecDDL\", reflect.TypeOf((*MockDriver)(nil).ExecDDL), varargs...)\n}\n\n// GetContext mocks base method.\nfunc (m *MockDriver) GetContext(ctx context.Context, dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetContext\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GetContext indicates an expected call of GetContext.\nfunc (mr *MockDriverMockRecorder) GetContext(ctx, dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetContext\", reflect.TypeOf((*MockDriver)(nil).GetContext), varargs...)\n}\n\n// GetForSchemaQuery mocks base method.\nfunc (m *MockDriver) GetForSchemaQuery(dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetForSchemaQuery\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GetForSchemaQuery indicates an expected call of GetForSchemaQuery.\nfunc (mr *MockDriverMockRecorder) GetForSchemaQuery(dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetForSchemaQuery\", reflect.TypeOf((*MockDriver)(nil).GetForSchemaQuery), varargs...)\n}\n\n// NamedExecContext mocks base method.\nfunc (m *MockDriver) NamedExecContext(ctx context.Context, dbShardID int, query string, arg any) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NamedExecContext\", ctx, dbShardID, query, arg)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NamedExecContext indicates an expected call of NamedExecContext.\nfunc (mr *MockDriverMockRecorder) NamedExecContext(ctx, dbShardID, query, arg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NamedExecContext\", reflect.TypeOf((*MockDriver)(nil).NamedExecContext), ctx, dbShardID, query, arg)\n}\n\n// Rollback mocks base method.\nfunc (m *MockDriver) Rollback() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Rollback\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Rollback indicates an expected call of Rollback.\nfunc (mr *MockDriverMockRecorder) Rollback() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Rollback\", reflect.TypeOf((*MockDriver)(nil).Rollback))\n}\n\n// SelectContext mocks base method.\nfunc (m *MockDriver) SelectContext(ctx context.Context, dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SelectContext\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SelectContext indicates an expected call of SelectContext.\nfunc (mr *MockDriverMockRecorder) SelectContext(ctx, dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectContext\", reflect.TypeOf((*MockDriver)(nil).SelectContext), varargs...)\n}\n\n// SelectForSchemaQuery mocks base method.\nfunc (m *MockDriver) SelectForSchemaQuery(dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SelectForSchemaQuery\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SelectForSchemaQuery indicates an expected call of SelectForSchemaQuery.\nfunc (mr *MockDriverMockRecorder) SelectForSchemaQuery(dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectForSchemaQuery\", reflect.TypeOf((*MockDriver)(nil).SelectForSchemaQuery), varargs...)\n}\n\n// MockcommonOfDbAndTx is a mock of commonOfDbAndTx interface.\ntype MockcommonOfDbAndTx struct {\n\tctrl     *gomock.Controller\n\trecorder *MockcommonOfDbAndTxMockRecorder\n\tisgomock struct{}\n}\n\n// MockcommonOfDbAndTxMockRecorder is the mock recorder for MockcommonOfDbAndTx.\ntype MockcommonOfDbAndTxMockRecorder struct {\n\tmock *MockcommonOfDbAndTx\n}\n\n// NewMockcommonOfDbAndTx creates a new mock instance.\nfunc NewMockcommonOfDbAndTx(ctrl *gomock.Controller) *MockcommonOfDbAndTx {\n\tmock := &MockcommonOfDbAndTx{ctrl: ctrl}\n\tmock.recorder = &MockcommonOfDbAndTxMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockcommonOfDbAndTx) EXPECT() *MockcommonOfDbAndTxMockRecorder {\n\treturn m.recorder\n}\n\n// ExecContext mocks base method.\nfunc (m *MockcommonOfDbAndTx) ExecContext(ctx context.Context, dbShardID int, query string, args ...any) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecContext\", varargs...)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ExecContext indicates an expected call of ExecContext.\nfunc (mr *MockcommonOfDbAndTxMockRecorder) ExecContext(ctx, dbShardID, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecContext\", reflect.TypeOf((*MockcommonOfDbAndTx)(nil).ExecContext), varargs...)\n}\n\n// GetContext mocks base method.\nfunc (m *MockcommonOfDbAndTx) GetContext(ctx context.Context, dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"GetContext\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GetContext indicates an expected call of GetContext.\nfunc (mr *MockcommonOfDbAndTxMockRecorder) GetContext(ctx, dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetContext\", reflect.TypeOf((*MockcommonOfDbAndTx)(nil).GetContext), varargs...)\n}\n\n// NamedExecContext mocks base method.\nfunc (m *MockcommonOfDbAndTx) NamedExecContext(ctx context.Context, dbShardID int, query string, arg any) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NamedExecContext\", ctx, dbShardID, query, arg)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NamedExecContext indicates an expected call of NamedExecContext.\nfunc (mr *MockcommonOfDbAndTxMockRecorder) NamedExecContext(ctx, dbShardID, query, arg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NamedExecContext\", reflect.TypeOf((*MockcommonOfDbAndTx)(nil).NamedExecContext), ctx, dbShardID, query, arg)\n}\n\n// SelectContext mocks base method.\nfunc (m *MockcommonOfDbAndTx) SelectContext(ctx context.Context, dbShardID int, dest any, query string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, dbShardID, dest, query}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"SelectContext\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SelectContext indicates an expected call of SelectContext.\nfunc (mr *MockcommonOfDbAndTxMockRecorder) SelectContext(ctx, dbShardID, dest, query any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, dbShardID, dest, query}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectContext\", reflect.TypeOf((*MockcommonOfDbAndTx)(nil).SelectContext), varargs...)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/sharded.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqldriver\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/jmoiron/sqlx\"\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype (\n\t// sharded is the driver querying a group of SQL databases as sharded solution\n\tsharded struct {\n\t\tdbs           []*sqlx.DB // this is for starting a transaction, or executing any non transaction query\n\t\ttx            *sqlx.Tx   // this is a reference of a started transaction\n\t\tuseTx         bool       // if tx is not nil, the methods from commonOfDbAndTx should use tx\n\t\tcurrTxShardID int        // which shard is current tx started from\n\t}\n)\n\n// newShardedSQLDriver returns a driver querying a group of SQL databases as sharded solution.\n// xdbs is the list of connections to the sql instances. The length of the list of the list is the totalNumShards\n// dbShardID is needed when tx is not nil. It means a started transaction in the shard.\nfunc newShardedSQLDriver(xdbs []*sqlx.DB, xtx *sqlx.Tx, dbShardID int) Driver {\n\tdriver := &sharded{\n\t\tdbs: xdbs,\n\t\ttx:  xtx,\n\t}\n\tif xtx != nil {\n\t\tdriver.useTx = true\n\t\tdriver.currTxShardID = dbShardID\n\t}\n\treturn driver\n}\n\n// below are shared by transactional and non-transactional, if s.tx is not nil then use s.tx, otherwise use s.db\n\nfunc (s *sharded) ExecContext(ctx context.Context, dbShardID int, query string, args ...interface{}) (sql.Result, error) {\n\tif dbShardID == sqlplugin.DbShardUndefined || dbShardID == sqlplugin.DbAllShards {\n\t\treturn nil, fmt.Errorf(\"invalid dbShardID %v shouldn't be used to ExecContext, there must be a bug\", dbShardID)\n\t}\n\tif s.useTx {\n\t\tif s.currTxShardID != dbShardID {\n\t\t\treturn nil, getUnmatchedTxnError(dbShardID, s.currTxShardID)\n\t\t}\n\t\treturn s.tx.ExecContext(ctx, query, args...)\n\t}\n\treturn s.dbs[dbShardID].ExecContext(ctx, query, args...)\n}\n\nfunc (s *sharded) NamedExecContext(ctx context.Context, dbShardID int, query string, arg interface{}) (sql.Result, error) {\n\tif dbShardID == sqlplugin.DbShardUndefined || dbShardID == sqlplugin.DbAllShards {\n\t\treturn nil, fmt.Errorf(\"invalid dbShardID %v shouldn't be used to NamedExecContext, there must be a bug\", dbShardID)\n\t}\n\tif s.useTx {\n\t\tif s.currTxShardID != dbShardID {\n\t\t\treturn nil, getUnmatchedTxnError(dbShardID, s.currTxShardID)\n\t\t}\n\t\treturn s.tx.NamedExecContext(ctx, query, arg)\n\t}\n\treturn s.dbs[dbShardID].NamedExecContext(ctx, query, arg)\n}\n\nfunc (s *sharded) GetContext(ctx context.Context, dbShardID int, dest interface{}, query string, args ...interface{}) error {\n\tif dbShardID == sqlplugin.DbShardUndefined || dbShardID == sqlplugin.DbAllShards {\n\t\treturn fmt.Errorf(\"invalid dbShardID %v shouldn't be used to GetContext, there must be a bug\", dbShardID)\n\t}\n\tif s.useTx {\n\t\tif s.currTxShardID != dbShardID {\n\t\t\treturn getUnmatchedTxnError(dbShardID, s.currTxShardID)\n\t\t}\n\t\treturn s.tx.GetContext(ctx, dest, query, args...)\n\t}\n\treturn s.dbs[dbShardID].GetContext(ctx, dest, query, args...)\n}\n\nfunc (s *sharded) SelectContext(ctx context.Context, dbShardID int, dest interface{}, query string, args ...interface{}) error {\n\tif dbShardID == sqlplugin.DbShardUndefined || dbShardID == sqlplugin.DbAllShards {\n\t\treturn fmt.Errorf(\"invalid dbShardID %v shouldn't be used to SelectContext, there must be a bug\", dbShardID)\n\t}\n\tif s.useTx {\n\t\tif s.currTxShardID != dbShardID {\n\t\t\treturn getUnmatchedTxnError(dbShardID, s.currTxShardID)\n\t\t}\n\t\treturn s.tx.SelectContext(ctx, dest, query, args...)\n\t}\n\treturn s.dbs[dbShardID].SelectContext(ctx, dest, query, args...)\n\n}\n\n// below are non-transactional methods only\n\nfunc (s *sharded) ExecDDL(ctx context.Context, dbShardID int, query string, args ...interface{}) (sql.Result, error) {\n\t// sharded SQL driver doesn't implement any schema operation as it's hard to guarantee the correctness.\n\t// schema operation across shards is implemented by application layer\n\treturn nil, fmt.Errorf(\"sharded SQL driver shouldn't be used to ExecDDL, there must be a bug\")\n}\n\nfunc (s *sharded) SelectForSchemaQuery(dbShardID int, dest interface{}, query string, args ...interface{}) error {\n\t// sharded SQL driver doesn't implement any schema operation as it's hard to guarantee the correctness.\n\t// schema operation across shards is implemented by application layer\n\treturn fmt.Errorf(\"sharded SQL driver shouldn't be used to SelectForSchemaQuery, there must be a bug\")\n}\n\nfunc (s *sharded) GetForSchemaQuery(dbShardID int, dest interface{}, query string, args ...interface{}) error {\n\t// sharded SQL driver doesn't implement any schema operation as it's hard to guarantee the correctness.\n\t// schema operation across shards is implemented by application layer\n\treturn fmt.Errorf(\"sharded SQL driver shouldn't be used to GetForSchemaQuery, there must be a bug\")\n}\n\nfunc (s *sharded) BeginTxx(ctx context.Context, dbShardID int, opts *sql.TxOptions) (*sqlx.Tx, error) {\n\tif dbShardID == sqlplugin.DbShardUndefined || dbShardID == sqlplugin.DbAllShards {\n\t\treturn nil, fmt.Errorf(\"invalid dbShardID %v shouldn't be used to BeginTxx, there must be a bug\", dbShardID)\n\t}\n\treturn s.dbs[dbShardID].BeginTxx(ctx, opts)\n}\n\nfunc (s *sharded) Close() error {\n\tvar errs []error\n\tfor _, db := range s.dbs {\n\t\terr := db.Close()\n\t\tif err != nil {\n\t\t\terrs = append(errs, err)\n\t\t}\n\t}\n\tif len(errs) > 0 {\n\t\treturn multierr.Combine(errs...)\n\t}\n\treturn nil\n}\n\n// below are transactional methods only\n\nfunc (s *sharded) Commit() error {\n\treturn s.tx.Commit()\n}\n\nfunc (s *sharded) Rollback() error {\n\treturn s.tx.Rollback()\n}\n\nfunc getUnmatchedTxnError(requestShardID, startedShardID int) error {\n\treturn fmt.Errorf(\"requested dbShardID %v doesn't match with started transaction shardID %v, must be a bug\", requestShardID, startedShardID)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqldriver/singleton.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqldriver\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/jmoiron/sqlx\"\n)\n\ntype (\n\t// singleton is the driver querying a single SQL database, which is the default driver\n\tsingleton struct {\n\t\tdb    *sqlx.DB // this is for starting a transaction, or executing any non transaction query\n\t\ttx    *sqlx.Tx // this is a reference of a started transaction\n\t\tuseTx bool     // if tx is not nil, the methods from commonOfDbAndTx should use tx\n\t}\n)\n\n// newSingletonSQLDriver returns a driver querying a single SQL database, which is the default driver\n// typically dbShardID is needed when tx is not nil, because it means a started transaction in a shard.\n// But this singleton doesn't have sharding so omitting it.\nfunc newSingletonSQLDriver(xdb *sqlx.DB, xtx *sqlx.Tx, _ int) Driver {\n\tdriver := &singleton{\n\t\tdb: xdb,\n\t\ttx: xtx,\n\t}\n\tif xtx != nil {\n\t\tdriver.useTx = true\n\t}\n\treturn driver\n}\n\n// below are shared by transactional and non-transactional, if s.tx is not nil then use s.tx, otherwise use s.db\n\nfunc (s *singleton) ExecContext(ctx context.Context, _ int, query string, args ...interface{}) (sql.Result, error) {\n\tif s.useTx {\n\t\treturn s.tx.ExecContext(ctx, query, args...)\n\t}\n\treturn s.db.ExecContext(ctx, query, args...)\n}\n\nfunc (s *singleton) NamedExecContext(ctx context.Context, _ int, query string, arg interface{}) (sql.Result, error) {\n\tif s.useTx {\n\t\treturn s.tx.NamedExecContext(ctx, query, arg)\n\t}\n\treturn s.db.NamedExecContext(ctx, query, arg)\n}\n\nfunc (s *singleton) GetContext(ctx context.Context, _ int, dest interface{}, query string, args ...interface{}) error {\n\tif s.useTx {\n\t\treturn s.tx.GetContext(ctx, dest, query, args...)\n\t}\n\treturn s.db.GetContext(ctx, dest, query, args...)\n}\n\nfunc (s *singleton) SelectContext(ctx context.Context, _ int, dest interface{}, query string, args ...interface{}) error {\n\tif s.useTx {\n\t\treturn s.tx.SelectContext(ctx, dest, query, args...)\n\t}\n\treturn s.db.SelectContext(ctx, dest, query, args...)\n}\n\n// below are non-transactional methods only\n\nfunc (s *singleton) ExecDDL(ctx context.Context, _ int, query string, args ...interface{}) (sql.Result, error) {\n\treturn s.db.ExecContext(ctx, query, args...)\n}\n\nfunc (s *singleton) SelectForSchemaQuery(_ int, dest interface{}, query string, args ...interface{}) error {\n\treturn s.db.Select(dest, query, args...)\n}\n\nfunc (s *singleton) GetForSchemaQuery(_ int, dest interface{}, query string, args ...interface{}) error {\n\treturn s.db.Get(dest, query, args...)\n}\n\nfunc (s *singleton) BeginTxx(ctx context.Context, _ int, opts *sql.TxOptions) (*sqlx.Tx, error) {\n\treturn s.db.BeginTxx(ctx, opts)\n}\n\nfunc (s *singleton) Close() error {\n\treturn s.db.Close()\n}\n\n// below are transactional methods only\n\nfunc (s *singleton) Commit() error {\n\treturn s.tx.Commit()\n}\n\nfunc (s *singleton) Rollback() error {\n\treturn s.tx.Rollback()\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/dbSharding.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqlplugin\n\nimport (\n\t\"github.com/dgryski/go-farm\"\n\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n)\n\n// This section defines the special dbShardID, they must all below 0\nconst (\n\t// this means the query need to execute in one shard but the shard should be fixed/static, e.g. for domain, queue storage are single shard\n\tDbDefaultShard = 0\n\t// this is should never being used in sharded SQL driver. It is used in admin/schema operation in singleton driver, which ignores all the shardID parameter\n\tDbShardUndefined = -1\n\t// this means the query needs to execute in all dbShards in sharded SQL driver (currently not supported)\n\tDbAllShards = -2\n)\n\n// GetDBShardIDFromHistoryShardID maps  historyShardID to a DBShardID\nfunc GetDBShardIDFromHistoryShardID(historyShardID int, numDBShards int) int {\n\treturn historyShardID % numDBShards\n}\n\n// GetDBShardIDFromDomainIDAndTasklist maps <domainID, tasklistName> to a DBShardID\nfunc GetDBShardIDFromDomainIDAndTasklist(domainID, tasklistName string, numDBShards int) int {\n\thash := farm.Hash32([]byte(domainID+\"_\"+tasklistName)) % uint32(numDBShards)\n\treturn int(hash) % numDBShards\n}\n\n// GetDBShardIDFromDomainID maps domainID to a DBShardID\nfunc GetDBShardIDFromDomainID(domainID string, numDBShards int) int {\n\thash := farm.Hash32([]byte(domainID)) % uint32(numDBShards)\n\treturn int(hash) % numDBShards\n}\n\n// GetDBShardIDFromTreeID maps treeID to a DBShardID\nfunc GetDBShardIDFromTreeID(treeID serialization.UUID, numDBShards int) int {\n\thash := farm.Hash32(treeID) % uint32(numDBShards)\n\treturn int(hash) % numDBShards\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package sqlplugin -source interfaces.go -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/sql/sqlplugin\n//\n\n// Package sqlplugin is a generated GoMock package.\npackage sqlplugin\n\nimport (\n\tcontext \"context\"\n\tsql \"database/sql\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tconfig \"github.com/uber/cadence/common/config\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockPlugin is a mock of Plugin interface.\ntype MockPlugin struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPluginMockRecorder\n\tisgomock struct{}\n}\n\n// MockPluginMockRecorder is the mock recorder for MockPlugin.\ntype MockPluginMockRecorder struct {\n\tmock *MockPlugin\n}\n\n// NewMockPlugin creates a new mock instance.\nfunc NewMockPlugin(ctrl *gomock.Controller) *MockPlugin {\n\tmock := &MockPlugin{ctrl: ctrl}\n\tmock.recorder = &MockPluginMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPlugin) EXPECT() *MockPluginMockRecorder {\n\treturn m.recorder\n}\n\n// CreateAdminDB mocks base method.\nfunc (m *MockPlugin) CreateAdminDB(cfg *config.SQL) (AdminDB, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateAdminDB\", cfg)\n\tret0, _ := ret[0].(AdminDB)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateAdminDB indicates an expected call of CreateAdminDB.\nfunc (mr *MockPluginMockRecorder) CreateAdminDB(cfg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateAdminDB\", reflect.TypeOf((*MockPlugin)(nil).CreateAdminDB), cfg)\n}\n\n// CreateDB mocks base method.\nfunc (m *MockPlugin) CreateDB(cfg *config.SQL) (DB, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDB\", cfg)\n\tret0, _ := ret[0].(DB)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateDB indicates an expected call of CreateDB.\nfunc (mr *MockPluginMockRecorder) CreateDB(cfg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDB\", reflect.TypeOf((*MockPlugin)(nil).CreateDB), cfg)\n}\n\n// MocktableCRUD is a mock of tableCRUD interface.\ntype MocktableCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MocktableCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MocktableCRUDMockRecorder is the mock recorder for MocktableCRUD.\ntype MocktableCRUDMockRecorder struct {\n\tmock *MocktableCRUD\n}\n\n// NewMocktableCRUD creates a new mock instance.\nfunc NewMocktableCRUD(ctrl *gomock.Controller) *MocktableCRUD {\n\tmock := &MocktableCRUD{ctrl: ctrl}\n\tmock.recorder = &MocktableCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MocktableCRUD) EXPECT() *MocktableCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// DeleteFromActivityInfoMaps mocks base method.\nfunc (m *MocktableCRUD) DeleteFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromActivityInfoMaps indicates an expected call of DeleteFromActivityInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromActivityInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromActivityInfoMaps), ctx, filter)\n}\n\n// DeleteFromBufferedEvents mocks base method.\nfunc (m *MocktableCRUD) DeleteFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromBufferedEvents indicates an expected call of DeleteFromBufferedEvents.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromBufferedEvents\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromBufferedEvents), ctx, filter)\n}\n\n// DeleteFromChildExecutionInfoMaps mocks base method.\nfunc (m *MocktableCRUD) DeleteFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromChildExecutionInfoMaps indicates an expected call of DeleteFromChildExecutionInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromChildExecutionInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// DeleteFromCrossClusterTasks mocks base method.\nfunc (m *MocktableCRUD) DeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCrossClusterTasks indicates an expected call of DeleteFromCrossClusterTasks.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCrossClusterTasks\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// DeleteFromCurrentExecutions mocks base method.\nfunc (m *MocktableCRUD) DeleteFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCurrentExecutions indicates an expected call of DeleteFromCurrentExecutions.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCurrentExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromCurrentExecutions), ctx, filter)\n}\n\n// DeleteFromDomain mocks base method.\nfunc (m *MocktableCRUD) DeleteFromDomain(ctx context.Context, filter *DomainFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromDomain\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromDomain indicates an expected call of DeleteFromDomain.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromDomain\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromDomain), ctx, filter)\n}\n\n// DeleteFromExecutions mocks base method.\nfunc (m *MocktableCRUD) DeleteFromExecutions(ctx context.Context, filter *ExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromExecutions indicates an expected call of DeleteFromExecutions.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromExecutions), ctx, filter)\n}\n\n// DeleteFromHistoryNode mocks base method.\nfunc (m *MocktableCRUD) DeleteFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryNode indicates an expected call of DeleteFromHistoryNode.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryNode\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromHistoryNode), ctx, filter)\n}\n\n// DeleteFromHistoryTree mocks base method.\nfunc (m *MocktableCRUD) DeleteFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryTree indicates an expected call of DeleteFromHistoryTree.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTree\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromHistoryTree), ctx, filter)\n}\n\n// DeleteFromReplicationTasks mocks base method.\nfunc (m *MocktableCRUD) DeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromReplicationTasks indicates an expected call of DeleteFromReplicationTasks.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromReplicationTasks\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromReplicationTasks), ctx, filter)\n}\n\n// DeleteFromRequestCancelInfoMaps mocks base method.\nfunc (m *MocktableCRUD) DeleteFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromRequestCancelInfoMaps indicates an expected call of DeleteFromRequestCancelInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromRequestCancelInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalInfoMaps mocks base method.\nfunc (m *MocktableCRUD) DeleteFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalInfoMaps indicates an expected call of DeleteFromSignalInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromSignalInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalsRequestedSets mocks base method.\nfunc (m *MocktableCRUD) DeleteFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalsRequestedSets indicates an expected call of DeleteFromSignalsRequestedSets.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalsRequestedSets\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromSignalsRequestedSets), ctx, filter)\n}\n\n// DeleteFromTaskLists mocks base method.\nfunc (m *MocktableCRUD) DeleteFromTaskLists(ctx context.Context, filter *TaskListsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTaskLists indicates an expected call of DeleteFromTaskLists.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTaskLists\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromTaskLists), ctx, filter)\n}\n\n// DeleteFromTasks mocks base method.\nfunc (m *MocktableCRUD) DeleteFromTasks(ctx context.Context, filter *TasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTasks indicates an expected call of DeleteFromTasks.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTasks\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromTasks), ctx, filter)\n}\n\n// DeleteFromTimerInfoMaps mocks base method.\nfunc (m *MocktableCRUD) DeleteFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerInfoMaps indicates an expected call of DeleteFromTimerInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromTimerInfoMaps), ctx, filter)\n}\n\n// DeleteFromTimerTasks mocks base method.\nfunc (m *MocktableCRUD) DeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerTasks indicates an expected call of DeleteFromTimerTasks.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerTasks\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromTimerTasks), ctx, filter)\n}\n\n// DeleteFromTransferTasks mocks base method.\nfunc (m *MocktableCRUD) DeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTransferTasks indicates an expected call of DeleteFromTransferTasks.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTransferTasks\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromTransferTasks), ctx, filter)\n}\n\n// DeleteFromVisibility mocks base method.\nfunc (m *MocktableCRUD) DeleteFromVisibility(ctx context.Context, filter *VisibilityFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromVisibility indicates an expected call of DeleteFromVisibility.\nfunc (mr *MocktableCRUDMockRecorder) DeleteFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteFromVisibility), ctx, filter)\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MocktableCRUD) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MocktableCRUD) DeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessageFromReplicationTasksDLQ indicates an expected call of DeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MocktableCRUD) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MocktableCRUDMockRecorder) DeleteMessagesBefore(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MocktableCRUD)(nil).DeleteMessagesBefore), ctx, queueType, messageID)\n}\n\n// GetAckLevels mocks base method.\nfunc (m *MocktableCRUD) GetAckLevels(ctx context.Context, queueType persistence.QueueType, forUpdate bool) (map[string]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevels\", ctx, queueType, forUpdate)\n\tret0, _ := ret[0].(map[string]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAckLevels indicates an expected call of GetAckLevels.\nfunc (mr *MocktableCRUDMockRecorder) GetAckLevels(ctx, queueType, forUpdate any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevels\", reflect.TypeOf((*MocktableCRUD)(nil).GetAckLevels), ctx, queueType, forUpdate)\n}\n\n// GetAllHistoryTreeBranches mocks base method.\nfunc (m *MocktableCRUD) GetAllHistoryTreeBranches(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllHistoryTreeBranches\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllHistoryTreeBranches indicates an expected call of GetAllHistoryTreeBranches.\nfunc (mr *MocktableCRUDMockRecorder) GetAllHistoryTreeBranches(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllHistoryTreeBranches\", reflect.TypeOf((*MocktableCRUD)(nil).GetAllHistoryTreeBranches), ctx, filter)\n}\n\n// GetLastEnqueuedMessageIDForUpdate mocks base method.\nfunc (m *MocktableCRUD) GetLastEnqueuedMessageIDForUpdate(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastEnqueuedMessageIDForUpdate\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetLastEnqueuedMessageIDForUpdate indicates an expected call of GetLastEnqueuedMessageIDForUpdate.\nfunc (mr *MocktableCRUDMockRecorder) GetLastEnqueuedMessageIDForUpdate(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastEnqueuedMessageIDForUpdate\", reflect.TypeOf((*MocktableCRUD)(nil).GetLastEnqueuedMessageIDForUpdate), ctx, queueType)\n}\n\n// GetMessagesBetween mocks base method.\nfunc (m *MocktableCRUD) GetMessagesBetween(ctx context.Context, queueType persistence.QueueType, firstMessageID, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesBetween\", ctx, queueType, firstMessageID, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesBetween indicates an expected call of GetMessagesBetween.\nfunc (mr *MocktableCRUDMockRecorder) GetMessagesBetween(ctx, queueType, firstMessageID, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesBetween\", reflect.TypeOf((*MocktableCRUD)(nil).GetMessagesBetween), ctx, queueType, firstMessageID, lastMessageID, maxRows)\n}\n\n// GetMessagesFromQueue mocks base method.\nfunc (m *MocktableCRUD) GetMessagesFromQueue(ctx context.Context, queueType persistence.QueueType, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesFromQueue\", ctx, queueType, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesFromQueue indicates an expected call of GetMessagesFromQueue.\nfunc (mr *MocktableCRUDMockRecorder) GetMessagesFromQueue(ctx, queueType, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesFromQueue\", reflect.TypeOf((*MocktableCRUD)(nil).GetMessagesFromQueue), ctx, queueType, lastMessageID, maxRows)\n}\n\n// GetOrphanTasks mocks base method.\nfunc (m *MocktableCRUD) GetOrphanTasks(ctx context.Context, filter *OrphanTasksFilter) ([]TaskKeyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrphanTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TaskKeyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOrphanTasks indicates an expected call of GetOrphanTasks.\nfunc (mr *MocktableCRUDMockRecorder) GetOrphanTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrphanTasks\", reflect.TypeOf((*MocktableCRUD)(nil).GetOrphanTasks), ctx, filter)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MocktableCRUD) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MocktableCRUDMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MocktableCRUD)(nil).GetQueueSize), ctx, queueType)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MocktableCRUD) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MocktableCRUDMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MocktableCRUD)(nil).GetTasksCount), ctx, filter)\n}\n\n// InsertAckLevel mocks base method.\nfunc (m *MocktableCRUD) InsertAckLevel(ctx context.Context, queueType persistence.QueueType, messageID int64, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertAckLevel\", ctx, queueType, messageID, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertAckLevel indicates an expected call of InsertAckLevel.\nfunc (mr *MocktableCRUDMockRecorder) InsertAckLevel(ctx, queueType, messageID, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertAckLevel\", reflect.TypeOf((*MocktableCRUD)(nil).InsertAckLevel), ctx, queueType, messageID, clusterName)\n}\n\n// InsertConfig mocks base method.\nfunc (m *MocktableCRUD) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MocktableCRUDMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MocktableCRUD)(nil).InsertConfig), ctx, row)\n}\n\n// InsertIntoBufferedEvents mocks base method.\nfunc (m *MocktableCRUD) InsertIntoBufferedEvents(ctx context.Context, rows []BufferedEventsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoBufferedEvents\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoBufferedEvents indicates an expected call of InsertIntoBufferedEvents.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoBufferedEvents(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoBufferedEvents\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoBufferedEvents), ctx, rows)\n}\n\n// InsertIntoCrossClusterTasks mocks base method.\nfunc (m *MocktableCRUD) InsertIntoCrossClusterTasks(ctx context.Context, rows []CrossClusterTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCrossClusterTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCrossClusterTasks indicates an expected call of InsertIntoCrossClusterTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoCrossClusterTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCrossClusterTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoCrossClusterTasks), ctx, rows)\n}\n\n// InsertIntoCurrentExecutions mocks base method.\nfunc (m *MocktableCRUD) InsertIntoCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCurrentExecutions indicates an expected call of InsertIntoCurrentExecutions.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCurrentExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoCurrentExecutions), ctx, row)\n}\n\n// InsertIntoDomain mocks base method.\nfunc (m *MocktableCRUD) InsertIntoDomain(ctx context.Context, rows *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomain\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomain indicates an expected call of InsertIntoDomain.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoDomain(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomain\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoDomain), ctx, rows)\n}\n\n// InsertIntoDomainAuditLog mocks base method.\nfunc (m *MocktableCRUD) InsertIntoDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomainAuditLog indicates an expected call of InsertIntoDomainAuditLog.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomainAuditLog\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoDomainAuditLog), ctx, row)\n}\n\n// InsertIntoExecutions mocks base method.\nfunc (m *MocktableCRUD) InsertIntoExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoExecutions indicates an expected call of InsertIntoExecutions.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoExecutions), ctx, row)\n}\n\n// InsertIntoHistoryNode mocks base method.\nfunc (m *MocktableCRUD) InsertIntoHistoryNode(ctx context.Context, row *HistoryNodeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryNode\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryNode indicates an expected call of InsertIntoHistoryNode.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoHistoryNode(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryNode\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoHistoryNode), ctx, row)\n}\n\n// InsertIntoHistoryTree mocks base method.\nfunc (m *MocktableCRUD) InsertIntoHistoryTree(ctx context.Context, row *HistoryTreeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTree\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryTree indicates an expected call of InsertIntoHistoryTree.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoHistoryTree(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTree\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoHistoryTree), ctx, row)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MocktableCRUD) InsertIntoQueue(ctx context.Context, row *QueueRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertIntoReplicationTasks mocks base method.\nfunc (m *MocktableCRUD) InsertIntoReplicationTasks(ctx context.Context, rows []ReplicationTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasks indicates an expected call of InsertIntoReplicationTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoReplicationTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoReplicationTasks), ctx, rows)\n}\n\n// InsertIntoReplicationTasksDLQ mocks base method.\nfunc (m *MocktableCRUD) InsertIntoReplicationTasksDLQ(ctx context.Context, row *ReplicationTaskDLQRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasksDLQ\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasksDLQ indicates an expected call of InsertIntoReplicationTasksDLQ.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoReplicationTasksDLQ(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasksDLQ\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoReplicationTasksDLQ), ctx, row)\n}\n\n// InsertIntoShards mocks base method.\nfunc (m *MocktableCRUD) InsertIntoShards(ctx context.Context, rows *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoShards\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoShards indicates an expected call of InsertIntoShards.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoShards(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoShards\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoShards), ctx, rows)\n}\n\n// InsertIntoSignalsRequestedSets mocks base method.\nfunc (m *MocktableCRUD) InsertIntoSignalsRequestedSets(ctx context.Context, rows []SignalsRequestedSetsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoSignalsRequestedSets\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoSignalsRequestedSets indicates an expected call of InsertIntoSignalsRequestedSets.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoSignalsRequestedSets(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoSignalsRequestedSets\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoSignalsRequestedSets), ctx, rows)\n}\n\n// InsertIntoTaskLists mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskLists indicates an expected call of InsertIntoTaskLists.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskLists\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTaskLists), ctx, row)\n}\n\n// InsertIntoTaskListsWithTTL mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskListsWithTTL indicates an expected call of InsertIntoTaskListsWithTTL.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskListsWithTTL\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTaskListsWithTTL), ctx, row)\n}\n\n// InsertIntoTasks mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTasks(ctx context.Context, rows []TasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasks indicates an expected call of InsertIntoTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTasks), ctx, rows)\n}\n\n// InsertIntoTasksWithTTL mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTasksWithTTL(ctx context.Context, rows []TasksRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasksWithTTL\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasksWithTTL indicates an expected call of InsertIntoTasksWithTTL.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTasksWithTTL(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasksWithTTL\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTasksWithTTL), ctx, rows)\n}\n\n// InsertIntoTimerTasks mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTimerTasks(ctx context.Context, rows []TimerTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTimerTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTimerTasks indicates an expected call of InsertIntoTimerTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTimerTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTimerTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTimerTasks), ctx, rows)\n}\n\n// InsertIntoTransferTasks mocks base method.\nfunc (m *MocktableCRUD) InsertIntoTransferTasks(ctx context.Context, rows []TransferTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTransferTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTransferTasks indicates an expected call of InsertIntoTransferTasks.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoTransferTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTransferTasks\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoTransferTasks), ctx, rows)\n}\n\n// InsertIntoVisibility mocks base method.\nfunc (m *MocktableCRUD) InsertIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoVisibility indicates an expected call of InsertIntoVisibility.\nfunc (mr *MocktableCRUDMockRecorder) InsertIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).InsertIntoVisibility), ctx, row)\n}\n\n// LockCurrentExecutions mocks base method.\nfunc (m *MocktableCRUD) LockCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutions indicates an expected call of LockCurrentExecutions.\nfunc (mr *MocktableCRUDMockRecorder) LockCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).LockCurrentExecutions), ctx, filter)\n}\n\n// LockCurrentExecutionsJoinExecutions mocks base method.\nfunc (m *MocktableCRUD) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *CurrentExecutionsFilter) ([]CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutionsJoinExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutionsJoinExecutions indicates an expected call of LockCurrentExecutionsJoinExecutions.\nfunc (mr *MocktableCRUDMockRecorder) LockCurrentExecutionsJoinExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutionsJoinExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).LockCurrentExecutionsJoinExecutions), ctx, filter)\n}\n\n// LockDomainMetadata mocks base method.\nfunc (m *MocktableCRUD) LockDomainMetadata(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockDomainMetadata\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// LockDomainMetadata indicates an expected call of LockDomainMetadata.\nfunc (mr *MocktableCRUDMockRecorder) LockDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockDomainMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).LockDomainMetadata), ctx)\n}\n\n// LockTaskLists mocks base method.\nfunc (m *MocktableCRUD) LockTaskLists(ctx context.Context, filter *TaskListsFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockTaskLists indicates an expected call of LockTaskLists.\nfunc (mr *MocktableCRUDMockRecorder) LockTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockTaskLists\", reflect.TypeOf((*MocktableCRUD)(nil).LockTaskLists), ctx, filter)\n}\n\n// MaxAllowedTTL mocks base method.\nfunc (m *MocktableCRUD) MaxAllowedTTL() (*time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MaxAllowedTTL\")\n\tret0, _ := ret[0].(*time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MaxAllowedTTL indicates an expected call of MaxAllowedTTL.\nfunc (mr *MocktableCRUDMockRecorder) MaxAllowedTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaxAllowedTTL\", reflect.TypeOf((*MocktableCRUD)(nil).MaxAllowedTTL))\n}\n\n// RangeDeleteFromCrossClusterTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromCrossClusterTasks indicates an expected call of RangeDeleteFromCrossClusterTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromCrossClusterTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// RangeDeleteFromReplicationTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromReplicationTasks indicates an expected call of RangeDeleteFromReplicationTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromReplicationTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteFromReplicationTasks), ctx, filter)\n}\n\n// RangeDeleteFromTimerTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTimerTasks indicates an expected call of RangeDeleteFromTimerTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTimerTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteFromTimerTasks), ctx, filter)\n}\n\n// RangeDeleteFromTransferTasks mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTransferTasks indicates an expected call of RangeDeleteFromTransferTasks.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTransferTasks\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteFromTransferTasks), ctx, filter)\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ indicates an expected call of RangeDeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// RangeDeleteMessages mocks base method.\nfunc (m *MocktableCRUD) RangeDeleteMessages(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessages\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessages indicates an expected call of RangeDeleteMessages.\nfunc (mr *MocktableCRUDMockRecorder) RangeDeleteMessages(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessages\", reflect.TypeOf((*MocktableCRUD)(nil).RangeDeleteMessages), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// ReadLockExecutions mocks base method.\nfunc (m *MocktableCRUD) ReadLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockExecutions indicates an expected call of ReadLockExecutions.\nfunc (mr *MocktableCRUDMockRecorder) ReadLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).ReadLockExecutions), ctx, filter)\n}\n\n// ReadLockShards mocks base method.\nfunc (m *MocktableCRUD) ReadLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockShards indicates an expected call of ReadLockShards.\nfunc (mr *MocktableCRUDMockRecorder) ReadLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockShards\", reflect.TypeOf((*MocktableCRUD)(nil).ReadLockShards), ctx, filter)\n}\n\n// ReplaceIntoActivityInfoMaps mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoActivityInfoMaps(ctx context.Context, rows []ActivityInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoActivityInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoActivityInfoMaps indicates an expected call of ReplaceIntoActivityInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoActivityInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoActivityInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoActivityInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoChildExecutionInfoMaps mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []ChildExecutionInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoChildExecutionInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoChildExecutionInfoMaps indicates an expected call of ReplaceIntoChildExecutionInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoChildExecutionInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoChildExecutionInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoChildExecutionInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoRequestCancelInfoMaps mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []RequestCancelInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoRequestCancelInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoRequestCancelInfoMaps indicates an expected call of ReplaceIntoRequestCancelInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoRequestCancelInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoRequestCancelInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoRequestCancelInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoSignalInfoMaps mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoSignalInfoMaps(ctx context.Context, rows []SignalInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoSignalInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoSignalInfoMaps indicates an expected call of ReplaceIntoSignalInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoSignalInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoSignalInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoSignalInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoTimerInfoMaps mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoTimerInfoMaps(ctx context.Context, rows []TimerInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoTimerInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoTimerInfoMaps indicates an expected call of ReplaceIntoTimerInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoTimerInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoTimerInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoTimerInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoVisibility mocks base method.\nfunc (m *MocktableCRUD) ReplaceIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoVisibility indicates an expected call of ReplaceIntoVisibility.\nfunc (mr *MocktableCRUDMockRecorder) ReplaceIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).ReplaceIntoVisibility), ctx, row)\n}\n\n// SelectFromActivityInfoMaps mocks base method.\nfunc (m *MocktableCRUD) SelectFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) ([]ActivityInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ActivityInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromActivityInfoMaps indicates an expected call of SelectFromActivityInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromActivityInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromActivityInfoMaps), ctx, filter)\n}\n\n// SelectFromBufferedEvents mocks base method.\nfunc (m *MocktableCRUD) SelectFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) ([]BufferedEventsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].([]BufferedEventsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromBufferedEvents indicates an expected call of SelectFromBufferedEvents.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromBufferedEvents\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromBufferedEvents), ctx, filter)\n}\n\n// SelectFromChildExecutionInfoMaps mocks base method.\nfunc (m *MocktableCRUD) SelectFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) ([]ChildExecutionInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ChildExecutionInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromChildExecutionInfoMaps indicates an expected call of SelectFromChildExecutionInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromChildExecutionInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// SelectFromCrossClusterTasks mocks base method.\nfunc (m *MocktableCRUD) SelectFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) ([]CrossClusterTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].([]CrossClusterTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCrossClusterTasks indicates an expected call of SelectFromCrossClusterTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCrossClusterTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromCrossClusterTasks), ctx, filter)\n}\n\n// SelectFromCurrentExecutions mocks base method.\nfunc (m *MocktableCRUD) SelectFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCurrentExecutions indicates an expected call of SelectFromCurrentExecutions.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCurrentExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromCurrentExecutions), ctx, filter)\n}\n\n// SelectFromDomain mocks base method.\nfunc (m *MocktableCRUD) SelectFromDomain(ctx context.Context, filter *DomainFilter) ([]DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomain\", ctx, filter)\n\tret0, _ := ret[0].([]DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomain indicates an expected call of SelectFromDomain.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomain\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromDomain), ctx, filter)\n}\n\n// SelectFromDomainAuditLogs mocks base method.\nfunc (m *MocktableCRUD) SelectFromDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainAuditLogs indicates an expected call of SelectFromDomainAuditLogs.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainAuditLogs\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromDomainAuditLogs), ctx, filter)\n}\n\n// SelectFromDomainMetadata mocks base method.\nfunc (m *MocktableCRUD) SelectFromDomainMetadata(ctx context.Context) (*DomainMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainMetadata\", ctx)\n\tret0, _ := ret[0].(*DomainMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainMetadata indicates an expected call of SelectFromDomainMetadata.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromDomainMetadata), ctx)\n}\n\n// SelectFromExecutions mocks base method.\nfunc (m *MocktableCRUD) SelectFromExecutions(ctx context.Context, filter *ExecutionsFilter) ([]ExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]ExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromExecutions indicates an expected call of SelectFromExecutions.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromExecutions), ctx, filter)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MocktableCRUD) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]HistoryNodeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryNodeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MocktableCRUD) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// SelectFromReplicationDLQ mocks base method.\nfunc (m *MocktableCRUD) SelectFromReplicationDLQ(ctx context.Context, filter *ReplicationTaskDLQFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationDLQ\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationDLQ indicates an expected call of SelectFromReplicationDLQ.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromReplicationDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationDLQ\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromReplicationDLQ), ctx, filter)\n}\n\n// SelectFromReplicationTasks mocks base method.\nfunc (m *MocktableCRUD) SelectFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasks indicates an expected call of SelectFromReplicationTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromReplicationTasks), ctx, filter)\n}\n\n// SelectFromReplicationTasksDLQ mocks base method.\nfunc (m *MocktableCRUD) SelectFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasksDLQ indicates an expected call of SelectFromReplicationTasksDLQ.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasksDLQ\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromReplicationTasksDLQ), ctx, filter)\n}\n\n// SelectFromRequestCancelInfoMaps mocks base method.\nfunc (m *MocktableCRUD) SelectFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) ([]RequestCancelInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]RequestCancelInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromRequestCancelInfoMaps indicates an expected call of SelectFromRequestCancelInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromRequestCancelInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// SelectFromShards mocks base method.\nfunc (m *MocktableCRUD) SelectFromShards(ctx context.Context, filter *ShardsFilter) (*ShardsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromShards\", ctx, filter)\n\tret0, _ := ret[0].(*ShardsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromShards indicates an expected call of SelectFromShards.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromShards\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromShards), ctx, filter)\n}\n\n// SelectFromSignalInfoMaps mocks base method.\nfunc (m *MocktableCRUD) SelectFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) ([]SignalInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]SignalInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalInfoMaps indicates an expected call of SelectFromSignalInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromSignalInfoMaps), ctx, filter)\n}\n\n// SelectFromSignalsRequestedSets mocks base method.\nfunc (m *MocktableCRUD) SelectFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) ([]SignalsRequestedSetsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].([]SignalsRequestedSetsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalsRequestedSets indicates an expected call of SelectFromSignalsRequestedSets.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalsRequestedSets\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromSignalsRequestedSets), ctx, filter)\n}\n\n// SelectFromTaskLists mocks base method.\nfunc (m *MocktableCRUD) SelectFromTaskLists(ctx context.Context, filter *TaskListsFilter) ([]TaskListsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].([]TaskListsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTaskLists indicates an expected call of SelectFromTaskLists.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTaskLists\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromTaskLists), ctx, filter)\n}\n\n// SelectFromTasks mocks base method.\nfunc (m *MocktableCRUD) SelectFromTasks(ctx context.Context, filter *TasksFilter) ([]TasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTasks indicates an expected call of SelectFromTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromTasks), ctx, filter)\n}\n\n// SelectFromTimerInfoMaps mocks base method.\nfunc (m *MocktableCRUD) SelectFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) ([]TimerInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]TimerInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerInfoMaps indicates an expected call of SelectFromTimerInfoMaps.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerInfoMaps\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromTimerInfoMaps), ctx, filter)\n}\n\n// SelectFromTimerTasks mocks base method.\nfunc (m *MocktableCRUD) SelectFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) ([]TimerTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TimerTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerTasks indicates an expected call of SelectFromTimerTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromTimerTasks), ctx, filter)\n}\n\n// SelectFromTransferTasks mocks base method.\nfunc (m *MocktableCRUD) SelectFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) ([]TransferTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TransferTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTransferTasks indicates an expected call of SelectFromTransferTasks.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTransferTasks\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromTransferTasks), ctx, filter)\n}\n\n// SelectFromVisibility mocks base method.\nfunc (m *MocktableCRUD) SelectFromVisibility(ctx context.Context, filter *VisibilityFilter) ([]VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].([]VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromVisibility indicates an expected call of SelectFromVisibility.\nfunc (mr *MocktableCRUDMockRecorder) SelectFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromVisibility\", reflect.TypeOf((*MocktableCRUD)(nil).SelectFromVisibility), ctx, filter)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MocktableCRUD) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MocktableCRUDMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MocktableCRUD)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// SupportsAsyncTransaction mocks base method.\nfunc (m *MocktableCRUD) SupportsAsyncTransaction() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsAsyncTransaction\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsAsyncTransaction indicates an expected call of SupportsAsyncTransaction.\nfunc (mr *MocktableCRUDMockRecorder) SupportsAsyncTransaction() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsAsyncTransaction\", reflect.TypeOf((*MocktableCRUD)(nil).SupportsAsyncTransaction))\n}\n\n// SupportsTTL mocks base method.\nfunc (m *MocktableCRUD) SupportsTTL() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsTTL\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsTTL indicates an expected call of SupportsTTL.\nfunc (mr *MocktableCRUDMockRecorder) SupportsTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsTTL\", reflect.TypeOf((*MocktableCRUD)(nil).SupportsTTL))\n}\n\n// UpdateAckLevels mocks base method.\nfunc (m *MocktableCRUD) UpdateAckLevels(ctx context.Context, queueType persistence.QueueType, clusterAckLevels map[string]int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevels\", ctx, queueType, clusterAckLevels)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAckLevels indicates an expected call of UpdateAckLevels.\nfunc (mr *MocktableCRUDMockRecorder) UpdateAckLevels(ctx, queueType, clusterAckLevels any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevels\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateAckLevels), ctx, queueType, clusterAckLevels)\n}\n\n// UpdateCurrentExecutions mocks base method.\nfunc (m *MocktableCRUD) UpdateCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateCurrentExecutions indicates an expected call of UpdateCurrentExecutions.\nfunc (mr *MocktableCRUDMockRecorder) UpdateCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateCurrentExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateCurrentExecutions), ctx, row)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MocktableCRUD) UpdateDomain(ctx context.Context, row *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MocktableCRUDMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateDomain), ctx, row)\n}\n\n// UpdateDomainMetadata mocks base method.\nfunc (m *MocktableCRUD) UpdateDomainMetadata(ctx context.Context, row *DomainMetadataRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainMetadata\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainMetadata indicates an expected call of UpdateDomainMetadata.\nfunc (mr *MocktableCRUDMockRecorder) UpdateDomainMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainMetadata\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateDomainMetadata), ctx, row)\n}\n\n// UpdateExecutions mocks base method.\nfunc (m *MocktableCRUD) UpdateExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateExecutions indicates an expected call of UpdateExecutions.\nfunc (mr *MocktableCRUDMockRecorder) UpdateExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateExecutions), ctx, row)\n}\n\n// UpdateShards mocks base method.\nfunc (m *MocktableCRUD) UpdateShards(ctx context.Context, row *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShards\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateShards indicates an expected call of UpdateShards.\nfunc (mr *MocktableCRUDMockRecorder) UpdateShards(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShards\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateShards), ctx, row)\n}\n\n// UpdateTaskLists mocks base method.\nfunc (m *MocktableCRUD) UpdateTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskLists indicates an expected call of UpdateTaskLists.\nfunc (mr *MocktableCRUDMockRecorder) UpdateTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskLists\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateTaskLists), ctx, row)\n}\n\n// UpdateTaskListsWithTTL mocks base method.\nfunc (m *MocktableCRUD) UpdateTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListsWithTTL indicates an expected call of UpdateTaskListsWithTTL.\nfunc (mr *MocktableCRUDMockRecorder) UpdateTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListsWithTTL\", reflect.TypeOf((*MocktableCRUD)(nil).UpdateTaskListsWithTTL), ctx, row)\n}\n\n// WriteLockExecutions mocks base method.\nfunc (m *MocktableCRUD) WriteLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockExecutions indicates an expected call of WriteLockExecutions.\nfunc (mr *MocktableCRUDMockRecorder) WriteLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockExecutions\", reflect.TypeOf((*MocktableCRUD)(nil).WriteLockExecutions), ctx, filter)\n}\n\n// WriteLockShards mocks base method.\nfunc (m *MocktableCRUD) WriteLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockShards indicates an expected call of WriteLockShards.\nfunc (mr *MocktableCRUDMockRecorder) WriteLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockShards\", reflect.TypeOf((*MocktableCRUD)(nil).WriteLockShards), ctx, filter)\n}\n\n// MockadminCRUD is a mock of adminCRUD interface.\ntype MockadminCRUD struct {\n\tctrl     *gomock.Controller\n\trecorder *MockadminCRUDMockRecorder\n\tisgomock struct{}\n}\n\n// MockadminCRUDMockRecorder is the mock recorder for MockadminCRUD.\ntype MockadminCRUDMockRecorder struct {\n\tmock *MockadminCRUD\n}\n\n// NewMockadminCRUD creates a new mock instance.\nfunc NewMockadminCRUD(ctrl *gomock.Controller) *MockadminCRUD {\n\tmock := &MockadminCRUD{ctrl: ctrl}\n\tmock.recorder = &MockadminCRUDMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockadminCRUD) EXPECT() *MockadminCRUDMockRecorder {\n\treturn m.recorder\n}\n\n// CreateDatabase mocks base method.\nfunc (m *MockadminCRUD) CreateDatabase(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDatabase\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateDatabase indicates an expected call of CreateDatabase.\nfunc (mr *MockadminCRUDMockRecorder) CreateDatabase(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDatabase\", reflect.TypeOf((*MockadminCRUD)(nil).CreateDatabase), database)\n}\n\n// CreateSchemaVersionTables mocks base method.\nfunc (m *MockadminCRUD) CreateSchemaVersionTables() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateSchemaVersionTables\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateSchemaVersionTables indicates an expected call of CreateSchemaVersionTables.\nfunc (mr *MockadminCRUDMockRecorder) CreateSchemaVersionTables() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateSchemaVersionTables\", reflect.TypeOf((*MockadminCRUD)(nil).CreateSchemaVersionTables))\n}\n\n// DropAllTables mocks base method.\nfunc (m *MockadminCRUD) DropAllTables(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropAllTables\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropAllTables indicates an expected call of DropAllTables.\nfunc (mr *MockadminCRUDMockRecorder) DropAllTables(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropAllTables\", reflect.TypeOf((*MockadminCRUD)(nil).DropAllTables), database)\n}\n\n// DropDatabase mocks base method.\nfunc (m *MockadminCRUD) DropDatabase(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropDatabase\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropDatabase indicates an expected call of DropDatabase.\nfunc (mr *MockadminCRUDMockRecorder) DropDatabase(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropDatabase\", reflect.TypeOf((*MockadminCRUD)(nil).DropDatabase), database)\n}\n\n// DropTable mocks base method.\nfunc (m *MockadminCRUD) DropTable(table string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropTable\", table)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropTable indicates an expected call of DropTable.\nfunc (mr *MockadminCRUDMockRecorder) DropTable(table any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropTable\", reflect.TypeOf((*MockadminCRUD)(nil).DropTable), table)\n}\n\n// ExecSchemaOperationQuery mocks base method.\nfunc (m *MockadminCRUD) ExecSchemaOperationQuery(ctx context.Context, stmt string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, stmt}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecSchemaOperationQuery\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ExecSchemaOperationQuery indicates an expected call of ExecSchemaOperationQuery.\nfunc (mr *MockadminCRUDMockRecorder) ExecSchemaOperationQuery(ctx, stmt any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, stmt}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecSchemaOperationQuery\", reflect.TypeOf((*MockadminCRUD)(nil).ExecSchemaOperationQuery), varargs...)\n}\n\n// ListTables mocks base method.\nfunc (m *MockadminCRUD) ListTables(database string) ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTables\", database)\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTables indicates an expected call of ListTables.\nfunc (mr *MockadminCRUDMockRecorder) ListTables(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTables\", reflect.TypeOf((*MockadminCRUD)(nil).ListTables), database)\n}\n\n// ReadSchemaVersion mocks base method.\nfunc (m *MockadminCRUD) ReadSchemaVersion(database string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadSchemaVersion\", database)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadSchemaVersion indicates an expected call of ReadSchemaVersion.\nfunc (mr *MockadminCRUDMockRecorder) ReadSchemaVersion(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadSchemaVersion\", reflect.TypeOf((*MockadminCRUD)(nil).ReadSchemaVersion), database)\n}\n\n// UpdateSchemaVersion mocks base method.\nfunc (m *MockadminCRUD) UpdateSchemaVersion(database, newVersion, minCompatibleVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateSchemaVersion\", database, newVersion, minCompatibleVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateSchemaVersion indicates an expected call of UpdateSchemaVersion.\nfunc (mr *MockadminCRUDMockRecorder) UpdateSchemaVersion(database, newVersion, minCompatibleVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateSchemaVersion\", reflect.TypeOf((*MockadminCRUD)(nil).UpdateSchemaVersion), database, newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog mocks base method.\nfunc (m *MockadminCRUD) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteSchemaUpdateLog\", oldVersion, newVersion, manifestMD5, desc)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// WriteSchemaUpdateLog indicates an expected call of WriteSchemaUpdateLog.\nfunc (mr *MockadminCRUDMockRecorder) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteSchemaUpdateLog\", reflect.TypeOf((*MockadminCRUD)(nil).WriteSchemaUpdateLog), oldVersion, newVersion, manifestMD5, desc)\n}\n\n// MockTx is a mock of Tx interface.\ntype MockTx struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTxMockRecorder\n\tisgomock struct{}\n}\n\n// MockTxMockRecorder is the mock recorder for MockTx.\ntype MockTxMockRecorder struct {\n\tmock *MockTx\n}\n\n// NewMockTx creates a new mock instance.\nfunc NewMockTx(ctrl *gomock.Controller) *MockTx {\n\tmock := &MockTx{ctrl: ctrl}\n\tmock.recorder = &MockTxMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTx) EXPECT() *MockTxMockRecorder {\n\treturn m.recorder\n}\n\n// Commit mocks base method.\nfunc (m *MockTx) Commit() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Commit\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Commit indicates an expected call of Commit.\nfunc (mr *MockTxMockRecorder) Commit() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Commit\", reflect.TypeOf((*MockTx)(nil).Commit))\n}\n\n// DeleteFromActivityInfoMaps mocks base method.\nfunc (m *MockTx) DeleteFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromActivityInfoMaps indicates an expected call of DeleteFromActivityInfoMaps.\nfunc (mr *MockTxMockRecorder) DeleteFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromActivityInfoMaps\", reflect.TypeOf((*MockTx)(nil).DeleteFromActivityInfoMaps), ctx, filter)\n}\n\n// DeleteFromBufferedEvents mocks base method.\nfunc (m *MockTx) DeleteFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromBufferedEvents indicates an expected call of DeleteFromBufferedEvents.\nfunc (mr *MockTxMockRecorder) DeleteFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromBufferedEvents\", reflect.TypeOf((*MockTx)(nil).DeleteFromBufferedEvents), ctx, filter)\n}\n\n// DeleteFromChildExecutionInfoMaps mocks base method.\nfunc (m *MockTx) DeleteFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromChildExecutionInfoMaps indicates an expected call of DeleteFromChildExecutionInfoMaps.\nfunc (mr *MockTxMockRecorder) DeleteFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromChildExecutionInfoMaps\", reflect.TypeOf((*MockTx)(nil).DeleteFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// DeleteFromCrossClusterTasks mocks base method.\nfunc (m *MockTx) DeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCrossClusterTasks indicates an expected call of DeleteFromCrossClusterTasks.\nfunc (mr *MockTxMockRecorder) DeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCrossClusterTasks\", reflect.TypeOf((*MockTx)(nil).DeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// DeleteFromCurrentExecutions mocks base method.\nfunc (m *MockTx) DeleteFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCurrentExecutions indicates an expected call of DeleteFromCurrentExecutions.\nfunc (mr *MockTxMockRecorder) DeleteFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCurrentExecutions\", reflect.TypeOf((*MockTx)(nil).DeleteFromCurrentExecutions), ctx, filter)\n}\n\n// DeleteFromDomain mocks base method.\nfunc (m *MockTx) DeleteFromDomain(ctx context.Context, filter *DomainFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromDomain\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromDomain indicates an expected call of DeleteFromDomain.\nfunc (mr *MockTxMockRecorder) DeleteFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromDomain\", reflect.TypeOf((*MockTx)(nil).DeleteFromDomain), ctx, filter)\n}\n\n// DeleteFromExecutions mocks base method.\nfunc (m *MockTx) DeleteFromExecutions(ctx context.Context, filter *ExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromExecutions indicates an expected call of DeleteFromExecutions.\nfunc (mr *MockTxMockRecorder) DeleteFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromExecutions\", reflect.TypeOf((*MockTx)(nil).DeleteFromExecutions), ctx, filter)\n}\n\n// DeleteFromHistoryNode mocks base method.\nfunc (m *MockTx) DeleteFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryNode indicates an expected call of DeleteFromHistoryNode.\nfunc (mr *MockTxMockRecorder) DeleteFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryNode\", reflect.TypeOf((*MockTx)(nil).DeleteFromHistoryNode), ctx, filter)\n}\n\n// DeleteFromHistoryTree mocks base method.\nfunc (m *MockTx) DeleteFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryTree indicates an expected call of DeleteFromHistoryTree.\nfunc (mr *MockTxMockRecorder) DeleteFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTree\", reflect.TypeOf((*MockTx)(nil).DeleteFromHistoryTree), ctx, filter)\n}\n\n// DeleteFromReplicationTasks mocks base method.\nfunc (m *MockTx) DeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromReplicationTasks indicates an expected call of DeleteFromReplicationTasks.\nfunc (mr *MockTxMockRecorder) DeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromReplicationTasks\", reflect.TypeOf((*MockTx)(nil).DeleteFromReplicationTasks), ctx, filter)\n}\n\n// DeleteFromRequestCancelInfoMaps mocks base method.\nfunc (m *MockTx) DeleteFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromRequestCancelInfoMaps indicates an expected call of DeleteFromRequestCancelInfoMaps.\nfunc (mr *MockTxMockRecorder) DeleteFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromRequestCancelInfoMaps\", reflect.TypeOf((*MockTx)(nil).DeleteFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalInfoMaps mocks base method.\nfunc (m *MockTx) DeleteFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalInfoMaps indicates an expected call of DeleteFromSignalInfoMaps.\nfunc (mr *MockTxMockRecorder) DeleteFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalInfoMaps\", reflect.TypeOf((*MockTx)(nil).DeleteFromSignalInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalsRequestedSets mocks base method.\nfunc (m *MockTx) DeleteFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalsRequestedSets indicates an expected call of DeleteFromSignalsRequestedSets.\nfunc (mr *MockTxMockRecorder) DeleteFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalsRequestedSets\", reflect.TypeOf((*MockTx)(nil).DeleteFromSignalsRequestedSets), ctx, filter)\n}\n\n// DeleteFromTaskLists mocks base method.\nfunc (m *MockTx) DeleteFromTaskLists(ctx context.Context, filter *TaskListsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTaskLists indicates an expected call of DeleteFromTaskLists.\nfunc (mr *MockTxMockRecorder) DeleteFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTaskLists\", reflect.TypeOf((*MockTx)(nil).DeleteFromTaskLists), ctx, filter)\n}\n\n// DeleteFromTasks mocks base method.\nfunc (m *MockTx) DeleteFromTasks(ctx context.Context, filter *TasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTasks indicates an expected call of DeleteFromTasks.\nfunc (mr *MockTxMockRecorder) DeleteFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTasks\", reflect.TypeOf((*MockTx)(nil).DeleteFromTasks), ctx, filter)\n}\n\n// DeleteFromTimerInfoMaps mocks base method.\nfunc (m *MockTx) DeleteFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerInfoMaps indicates an expected call of DeleteFromTimerInfoMaps.\nfunc (mr *MockTxMockRecorder) DeleteFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerInfoMaps\", reflect.TypeOf((*MockTx)(nil).DeleteFromTimerInfoMaps), ctx, filter)\n}\n\n// DeleteFromTimerTasks mocks base method.\nfunc (m *MockTx) DeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerTasks indicates an expected call of DeleteFromTimerTasks.\nfunc (mr *MockTxMockRecorder) DeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerTasks\", reflect.TypeOf((*MockTx)(nil).DeleteFromTimerTasks), ctx, filter)\n}\n\n// DeleteFromTransferTasks mocks base method.\nfunc (m *MockTx) DeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTransferTasks indicates an expected call of DeleteFromTransferTasks.\nfunc (mr *MockTxMockRecorder) DeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTransferTasks\", reflect.TypeOf((*MockTx)(nil).DeleteFromTransferTasks), ctx, filter)\n}\n\n// DeleteFromVisibility mocks base method.\nfunc (m *MockTx) DeleteFromVisibility(ctx context.Context, filter *VisibilityFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromVisibility indicates an expected call of DeleteFromVisibility.\nfunc (mr *MockTxMockRecorder) DeleteFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromVisibility\", reflect.TypeOf((*MockTx)(nil).DeleteFromVisibility), ctx, filter)\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MockTx) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MockTxMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MockTx)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MockTx) DeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessageFromReplicationTasksDLQ indicates an expected call of DeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MockTxMockRecorder) DeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MockTx)(nil).DeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MockTx) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MockTxMockRecorder) DeleteMessagesBefore(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MockTx)(nil).DeleteMessagesBefore), ctx, queueType, messageID)\n}\n\n// GetAckLevels mocks base method.\nfunc (m *MockTx) GetAckLevels(ctx context.Context, queueType persistence.QueueType, forUpdate bool) (map[string]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevels\", ctx, queueType, forUpdate)\n\tret0, _ := ret[0].(map[string]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAckLevels indicates an expected call of GetAckLevels.\nfunc (mr *MockTxMockRecorder) GetAckLevels(ctx, queueType, forUpdate any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevels\", reflect.TypeOf((*MockTx)(nil).GetAckLevels), ctx, queueType, forUpdate)\n}\n\n// GetAllHistoryTreeBranches mocks base method.\nfunc (m *MockTx) GetAllHistoryTreeBranches(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllHistoryTreeBranches\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllHistoryTreeBranches indicates an expected call of GetAllHistoryTreeBranches.\nfunc (mr *MockTxMockRecorder) GetAllHistoryTreeBranches(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllHistoryTreeBranches\", reflect.TypeOf((*MockTx)(nil).GetAllHistoryTreeBranches), ctx, filter)\n}\n\n// GetLastEnqueuedMessageIDForUpdate mocks base method.\nfunc (m *MockTx) GetLastEnqueuedMessageIDForUpdate(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastEnqueuedMessageIDForUpdate\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetLastEnqueuedMessageIDForUpdate indicates an expected call of GetLastEnqueuedMessageIDForUpdate.\nfunc (mr *MockTxMockRecorder) GetLastEnqueuedMessageIDForUpdate(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastEnqueuedMessageIDForUpdate\", reflect.TypeOf((*MockTx)(nil).GetLastEnqueuedMessageIDForUpdate), ctx, queueType)\n}\n\n// GetMessagesBetween mocks base method.\nfunc (m *MockTx) GetMessagesBetween(ctx context.Context, queueType persistence.QueueType, firstMessageID, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesBetween\", ctx, queueType, firstMessageID, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesBetween indicates an expected call of GetMessagesBetween.\nfunc (mr *MockTxMockRecorder) GetMessagesBetween(ctx, queueType, firstMessageID, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesBetween\", reflect.TypeOf((*MockTx)(nil).GetMessagesBetween), ctx, queueType, firstMessageID, lastMessageID, maxRows)\n}\n\n// GetMessagesFromQueue mocks base method.\nfunc (m *MockTx) GetMessagesFromQueue(ctx context.Context, queueType persistence.QueueType, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesFromQueue\", ctx, queueType, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesFromQueue indicates an expected call of GetMessagesFromQueue.\nfunc (mr *MockTxMockRecorder) GetMessagesFromQueue(ctx, queueType, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesFromQueue\", reflect.TypeOf((*MockTx)(nil).GetMessagesFromQueue), ctx, queueType, lastMessageID, maxRows)\n}\n\n// GetOrphanTasks mocks base method.\nfunc (m *MockTx) GetOrphanTasks(ctx context.Context, filter *OrphanTasksFilter) ([]TaskKeyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrphanTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TaskKeyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOrphanTasks indicates an expected call of GetOrphanTasks.\nfunc (mr *MockTxMockRecorder) GetOrphanTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrphanTasks\", reflect.TypeOf((*MockTx)(nil).GetOrphanTasks), ctx, filter)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MockTx) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MockTxMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MockTx)(nil).GetQueueSize), ctx, queueType)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MockTx) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MockTxMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MockTx)(nil).GetTasksCount), ctx, filter)\n}\n\n// InsertAckLevel mocks base method.\nfunc (m *MockTx) InsertAckLevel(ctx context.Context, queueType persistence.QueueType, messageID int64, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertAckLevel\", ctx, queueType, messageID, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertAckLevel indicates an expected call of InsertAckLevel.\nfunc (mr *MockTxMockRecorder) InsertAckLevel(ctx, queueType, messageID, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertAckLevel\", reflect.TypeOf((*MockTx)(nil).InsertAckLevel), ctx, queueType, messageID, clusterName)\n}\n\n// InsertConfig mocks base method.\nfunc (m *MockTx) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MockTxMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MockTx)(nil).InsertConfig), ctx, row)\n}\n\n// InsertIntoBufferedEvents mocks base method.\nfunc (m *MockTx) InsertIntoBufferedEvents(ctx context.Context, rows []BufferedEventsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoBufferedEvents\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoBufferedEvents indicates an expected call of InsertIntoBufferedEvents.\nfunc (mr *MockTxMockRecorder) InsertIntoBufferedEvents(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoBufferedEvents\", reflect.TypeOf((*MockTx)(nil).InsertIntoBufferedEvents), ctx, rows)\n}\n\n// InsertIntoCrossClusterTasks mocks base method.\nfunc (m *MockTx) InsertIntoCrossClusterTasks(ctx context.Context, rows []CrossClusterTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCrossClusterTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCrossClusterTasks indicates an expected call of InsertIntoCrossClusterTasks.\nfunc (mr *MockTxMockRecorder) InsertIntoCrossClusterTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCrossClusterTasks\", reflect.TypeOf((*MockTx)(nil).InsertIntoCrossClusterTasks), ctx, rows)\n}\n\n// InsertIntoCurrentExecutions mocks base method.\nfunc (m *MockTx) InsertIntoCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCurrentExecutions indicates an expected call of InsertIntoCurrentExecutions.\nfunc (mr *MockTxMockRecorder) InsertIntoCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCurrentExecutions\", reflect.TypeOf((*MockTx)(nil).InsertIntoCurrentExecutions), ctx, row)\n}\n\n// InsertIntoDomain mocks base method.\nfunc (m *MockTx) InsertIntoDomain(ctx context.Context, rows *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomain\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomain indicates an expected call of InsertIntoDomain.\nfunc (mr *MockTxMockRecorder) InsertIntoDomain(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomain\", reflect.TypeOf((*MockTx)(nil).InsertIntoDomain), ctx, rows)\n}\n\n// InsertIntoDomainAuditLog mocks base method.\nfunc (m *MockTx) InsertIntoDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomainAuditLog indicates an expected call of InsertIntoDomainAuditLog.\nfunc (mr *MockTxMockRecorder) InsertIntoDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomainAuditLog\", reflect.TypeOf((*MockTx)(nil).InsertIntoDomainAuditLog), ctx, row)\n}\n\n// InsertIntoExecutions mocks base method.\nfunc (m *MockTx) InsertIntoExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoExecutions indicates an expected call of InsertIntoExecutions.\nfunc (mr *MockTxMockRecorder) InsertIntoExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoExecutions\", reflect.TypeOf((*MockTx)(nil).InsertIntoExecutions), ctx, row)\n}\n\n// InsertIntoHistoryNode mocks base method.\nfunc (m *MockTx) InsertIntoHistoryNode(ctx context.Context, row *HistoryNodeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryNode\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryNode indicates an expected call of InsertIntoHistoryNode.\nfunc (mr *MockTxMockRecorder) InsertIntoHistoryNode(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryNode\", reflect.TypeOf((*MockTx)(nil).InsertIntoHistoryNode), ctx, row)\n}\n\n// InsertIntoHistoryTree mocks base method.\nfunc (m *MockTx) InsertIntoHistoryTree(ctx context.Context, row *HistoryTreeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTree\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryTree indicates an expected call of InsertIntoHistoryTree.\nfunc (mr *MockTxMockRecorder) InsertIntoHistoryTree(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTree\", reflect.TypeOf((*MockTx)(nil).InsertIntoHistoryTree), ctx, row)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MockTx) InsertIntoQueue(ctx context.Context, row *QueueRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MockTxMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MockTx)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertIntoReplicationTasks mocks base method.\nfunc (m *MockTx) InsertIntoReplicationTasks(ctx context.Context, rows []ReplicationTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasks indicates an expected call of InsertIntoReplicationTasks.\nfunc (mr *MockTxMockRecorder) InsertIntoReplicationTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasks\", reflect.TypeOf((*MockTx)(nil).InsertIntoReplicationTasks), ctx, rows)\n}\n\n// InsertIntoReplicationTasksDLQ mocks base method.\nfunc (m *MockTx) InsertIntoReplicationTasksDLQ(ctx context.Context, row *ReplicationTaskDLQRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasksDLQ\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasksDLQ indicates an expected call of InsertIntoReplicationTasksDLQ.\nfunc (mr *MockTxMockRecorder) InsertIntoReplicationTasksDLQ(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasksDLQ\", reflect.TypeOf((*MockTx)(nil).InsertIntoReplicationTasksDLQ), ctx, row)\n}\n\n// InsertIntoShards mocks base method.\nfunc (m *MockTx) InsertIntoShards(ctx context.Context, rows *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoShards\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoShards indicates an expected call of InsertIntoShards.\nfunc (mr *MockTxMockRecorder) InsertIntoShards(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoShards\", reflect.TypeOf((*MockTx)(nil).InsertIntoShards), ctx, rows)\n}\n\n// InsertIntoSignalsRequestedSets mocks base method.\nfunc (m *MockTx) InsertIntoSignalsRequestedSets(ctx context.Context, rows []SignalsRequestedSetsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoSignalsRequestedSets\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoSignalsRequestedSets indicates an expected call of InsertIntoSignalsRequestedSets.\nfunc (mr *MockTxMockRecorder) InsertIntoSignalsRequestedSets(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoSignalsRequestedSets\", reflect.TypeOf((*MockTx)(nil).InsertIntoSignalsRequestedSets), ctx, rows)\n}\n\n// InsertIntoTaskLists mocks base method.\nfunc (m *MockTx) InsertIntoTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskLists indicates an expected call of InsertIntoTaskLists.\nfunc (mr *MockTxMockRecorder) InsertIntoTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskLists\", reflect.TypeOf((*MockTx)(nil).InsertIntoTaskLists), ctx, row)\n}\n\n// InsertIntoTaskListsWithTTL mocks base method.\nfunc (m *MockTx) InsertIntoTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskListsWithTTL indicates an expected call of InsertIntoTaskListsWithTTL.\nfunc (mr *MockTxMockRecorder) InsertIntoTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskListsWithTTL\", reflect.TypeOf((*MockTx)(nil).InsertIntoTaskListsWithTTL), ctx, row)\n}\n\n// InsertIntoTasks mocks base method.\nfunc (m *MockTx) InsertIntoTasks(ctx context.Context, rows []TasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasks indicates an expected call of InsertIntoTasks.\nfunc (mr *MockTxMockRecorder) InsertIntoTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasks\", reflect.TypeOf((*MockTx)(nil).InsertIntoTasks), ctx, rows)\n}\n\n// InsertIntoTasksWithTTL mocks base method.\nfunc (m *MockTx) InsertIntoTasksWithTTL(ctx context.Context, rows []TasksRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasksWithTTL\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasksWithTTL indicates an expected call of InsertIntoTasksWithTTL.\nfunc (mr *MockTxMockRecorder) InsertIntoTasksWithTTL(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasksWithTTL\", reflect.TypeOf((*MockTx)(nil).InsertIntoTasksWithTTL), ctx, rows)\n}\n\n// InsertIntoTimerTasks mocks base method.\nfunc (m *MockTx) InsertIntoTimerTasks(ctx context.Context, rows []TimerTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTimerTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTimerTasks indicates an expected call of InsertIntoTimerTasks.\nfunc (mr *MockTxMockRecorder) InsertIntoTimerTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTimerTasks\", reflect.TypeOf((*MockTx)(nil).InsertIntoTimerTasks), ctx, rows)\n}\n\n// InsertIntoTransferTasks mocks base method.\nfunc (m *MockTx) InsertIntoTransferTasks(ctx context.Context, rows []TransferTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTransferTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTransferTasks indicates an expected call of InsertIntoTransferTasks.\nfunc (mr *MockTxMockRecorder) InsertIntoTransferTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTransferTasks\", reflect.TypeOf((*MockTx)(nil).InsertIntoTransferTasks), ctx, rows)\n}\n\n// InsertIntoVisibility mocks base method.\nfunc (m *MockTx) InsertIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoVisibility indicates an expected call of InsertIntoVisibility.\nfunc (mr *MockTxMockRecorder) InsertIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoVisibility\", reflect.TypeOf((*MockTx)(nil).InsertIntoVisibility), ctx, row)\n}\n\n// IsDupEntryError mocks base method.\nfunc (m *MockTx) IsDupEntryError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDupEntryError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDupEntryError indicates an expected call of IsDupEntryError.\nfunc (mr *MockTxMockRecorder) IsDupEntryError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDupEntryError\", reflect.TypeOf((*MockTx)(nil).IsDupEntryError), err)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockTx) IsNotFoundError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockTxMockRecorder) IsNotFoundError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockTx)(nil).IsNotFoundError), err)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockTx) IsThrottlingError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockTxMockRecorder) IsThrottlingError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockTx)(nil).IsThrottlingError), err)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockTx) IsTimeoutError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockTxMockRecorder) IsTimeoutError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockTx)(nil).IsTimeoutError), err)\n}\n\n// LockCurrentExecutions mocks base method.\nfunc (m *MockTx) LockCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutions indicates an expected call of LockCurrentExecutions.\nfunc (mr *MockTxMockRecorder) LockCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutions\", reflect.TypeOf((*MockTx)(nil).LockCurrentExecutions), ctx, filter)\n}\n\n// LockCurrentExecutionsJoinExecutions mocks base method.\nfunc (m *MockTx) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *CurrentExecutionsFilter) ([]CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutionsJoinExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutionsJoinExecutions indicates an expected call of LockCurrentExecutionsJoinExecutions.\nfunc (mr *MockTxMockRecorder) LockCurrentExecutionsJoinExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutionsJoinExecutions\", reflect.TypeOf((*MockTx)(nil).LockCurrentExecutionsJoinExecutions), ctx, filter)\n}\n\n// LockDomainMetadata mocks base method.\nfunc (m *MockTx) LockDomainMetadata(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockDomainMetadata\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// LockDomainMetadata indicates an expected call of LockDomainMetadata.\nfunc (mr *MockTxMockRecorder) LockDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockDomainMetadata\", reflect.TypeOf((*MockTx)(nil).LockDomainMetadata), ctx)\n}\n\n// LockTaskLists mocks base method.\nfunc (m *MockTx) LockTaskLists(ctx context.Context, filter *TaskListsFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockTaskLists indicates an expected call of LockTaskLists.\nfunc (mr *MockTxMockRecorder) LockTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockTaskLists\", reflect.TypeOf((*MockTx)(nil).LockTaskLists), ctx, filter)\n}\n\n// MaxAllowedTTL mocks base method.\nfunc (m *MockTx) MaxAllowedTTL() (*time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MaxAllowedTTL\")\n\tret0, _ := ret[0].(*time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MaxAllowedTTL indicates an expected call of MaxAllowedTTL.\nfunc (mr *MockTxMockRecorder) MaxAllowedTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaxAllowedTTL\", reflect.TypeOf((*MockTx)(nil).MaxAllowedTTL))\n}\n\n// RangeDeleteFromCrossClusterTasks mocks base method.\nfunc (m *MockTx) RangeDeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromCrossClusterTasks indicates an expected call of RangeDeleteFromCrossClusterTasks.\nfunc (mr *MockTxMockRecorder) RangeDeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromCrossClusterTasks\", reflect.TypeOf((*MockTx)(nil).RangeDeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// RangeDeleteFromReplicationTasks mocks base method.\nfunc (m *MockTx) RangeDeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromReplicationTasks indicates an expected call of RangeDeleteFromReplicationTasks.\nfunc (mr *MockTxMockRecorder) RangeDeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromReplicationTasks\", reflect.TypeOf((*MockTx)(nil).RangeDeleteFromReplicationTasks), ctx, filter)\n}\n\n// RangeDeleteFromTimerTasks mocks base method.\nfunc (m *MockTx) RangeDeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTimerTasks indicates an expected call of RangeDeleteFromTimerTasks.\nfunc (mr *MockTxMockRecorder) RangeDeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTimerTasks\", reflect.TypeOf((*MockTx)(nil).RangeDeleteFromTimerTasks), ctx, filter)\n}\n\n// RangeDeleteFromTransferTasks mocks base method.\nfunc (m *MockTx) RangeDeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTransferTasks indicates an expected call of RangeDeleteFromTransferTasks.\nfunc (mr *MockTxMockRecorder) RangeDeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTransferTasks\", reflect.TypeOf((*MockTx)(nil).RangeDeleteFromTransferTasks), ctx, filter)\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MockTx) RangeDeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ indicates an expected call of RangeDeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MockTxMockRecorder) RangeDeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MockTx)(nil).RangeDeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// RangeDeleteMessages mocks base method.\nfunc (m *MockTx) RangeDeleteMessages(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessages\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessages indicates an expected call of RangeDeleteMessages.\nfunc (mr *MockTxMockRecorder) RangeDeleteMessages(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessages\", reflect.TypeOf((*MockTx)(nil).RangeDeleteMessages), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// ReadLockExecutions mocks base method.\nfunc (m *MockTx) ReadLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockExecutions indicates an expected call of ReadLockExecutions.\nfunc (mr *MockTxMockRecorder) ReadLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockExecutions\", reflect.TypeOf((*MockTx)(nil).ReadLockExecutions), ctx, filter)\n}\n\n// ReadLockShards mocks base method.\nfunc (m *MockTx) ReadLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockShards indicates an expected call of ReadLockShards.\nfunc (mr *MockTxMockRecorder) ReadLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockShards\", reflect.TypeOf((*MockTx)(nil).ReadLockShards), ctx, filter)\n}\n\n// ReplaceIntoActivityInfoMaps mocks base method.\nfunc (m *MockTx) ReplaceIntoActivityInfoMaps(ctx context.Context, rows []ActivityInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoActivityInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoActivityInfoMaps indicates an expected call of ReplaceIntoActivityInfoMaps.\nfunc (mr *MockTxMockRecorder) ReplaceIntoActivityInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoActivityInfoMaps\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoActivityInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoChildExecutionInfoMaps mocks base method.\nfunc (m *MockTx) ReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []ChildExecutionInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoChildExecutionInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoChildExecutionInfoMaps indicates an expected call of ReplaceIntoChildExecutionInfoMaps.\nfunc (mr *MockTxMockRecorder) ReplaceIntoChildExecutionInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoChildExecutionInfoMaps\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoChildExecutionInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoRequestCancelInfoMaps mocks base method.\nfunc (m *MockTx) ReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []RequestCancelInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoRequestCancelInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoRequestCancelInfoMaps indicates an expected call of ReplaceIntoRequestCancelInfoMaps.\nfunc (mr *MockTxMockRecorder) ReplaceIntoRequestCancelInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoRequestCancelInfoMaps\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoRequestCancelInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoSignalInfoMaps mocks base method.\nfunc (m *MockTx) ReplaceIntoSignalInfoMaps(ctx context.Context, rows []SignalInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoSignalInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoSignalInfoMaps indicates an expected call of ReplaceIntoSignalInfoMaps.\nfunc (mr *MockTxMockRecorder) ReplaceIntoSignalInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoSignalInfoMaps\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoSignalInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoTimerInfoMaps mocks base method.\nfunc (m *MockTx) ReplaceIntoTimerInfoMaps(ctx context.Context, rows []TimerInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoTimerInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoTimerInfoMaps indicates an expected call of ReplaceIntoTimerInfoMaps.\nfunc (mr *MockTxMockRecorder) ReplaceIntoTimerInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoTimerInfoMaps\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoTimerInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoVisibility mocks base method.\nfunc (m *MockTx) ReplaceIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoVisibility indicates an expected call of ReplaceIntoVisibility.\nfunc (mr *MockTxMockRecorder) ReplaceIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoVisibility\", reflect.TypeOf((*MockTx)(nil).ReplaceIntoVisibility), ctx, row)\n}\n\n// Rollback mocks base method.\nfunc (m *MockTx) Rollback() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Rollback\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Rollback indicates an expected call of Rollback.\nfunc (mr *MockTxMockRecorder) Rollback() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Rollback\", reflect.TypeOf((*MockTx)(nil).Rollback))\n}\n\n// SelectFromActivityInfoMaps mocks base method.\nfunc (m *MockTx) SelectFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) ([]ActivityInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ActivityInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromActivityInfoMaps indicates an expected call of SelectFromActivityInfoMaps.\nfunc (mr *MockTxMockRecorder) SelectFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromActivityInfoMaps\", reflect.TypeOf((*MockTx)(nil).SelectFromActivityInfoMaps), ctx, filter)\n}\n\n// SelectFromBufferedEvents mocks base method.\nfunc (m *MockTx) SelectFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) ([]BufferedEventsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].([]BufferedEventsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromBufferedEvents indicates an expected call of SelectFromBufferedEvents.\nfunc (mr *MockTxMockRecorder) SelectFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromBufferedEvents\", reflect.TypeOf((*MockTx)(nil).SelectFromBufferedEvents), ctx, filter)\n}\n\n// SelectFromChildExecutionInfoMaps mocks base method.\nfunc (m *MockTx) SelectFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) ([]ChildExecutionInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ChildExecutionInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromChildExecutionInfoMaps indicates an expected call of SelectFromChildExecutionInfoMaps.\nfunc (mr *MockTxMockRecorder) SelectFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromChildExecutionInfoMaps\", reflect.TypeOf((*MockTx)(nil).SelectFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// SelectFromCrossClusterTasks mocks base method.\nfunc (m *MockTx) SelectFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) ([]CrossClusterTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].([]CrossClusterTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCrossClusterTasks indicates an expected call of SelectFromCrossClusterTasks.\nfunc (mr *MockTxMockRecorder) SelectFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCrossClusterTasks\", reflect.TypeOf((*MockTx)(nil).SelectFromCrossClusterTasks), ctx, filter)\n}\n\n// SelectFromCurrentExecutions mocks base method.\nfunc (m *MockTx) SelectFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCurrentExecutions indicates an expected call of SelectFromCurrentExecutions.\nfunc (mr *MockTxMockRecorder) SelectFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCurrentExecutions\", reflect.TypeOf((*MockTx)(nil).SelectFromCurrentExecutions), ctx, filter)\n}\n\n// SelectFromDomain mocks base method.\nfunc (m *MockTx) SelectFromDomain(ctx context.Context, filter *DomainFilter) ([]DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomain\", ctx, filter)\n\tret0, _ := ret[0].([]DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomain indicates an expected call of SelectFromDomain.\nfunc (mr *MockTxMockRecorder) SelectFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomain\", reflect.TypeOf((*MockTx)(nil).SelectFromDomain), ctx, filter)\n}\n\n// SelectFromDomainAuditLogs mocks base method.\nfunc (m *MockTx) SelectFromDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainAuditLogs indicates an expected call of SelectFromDomainAuditLogs.\nfunc (mr *MockTxMockRecorder) SelectFromDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainAuditLogs\", reflect.TypeOf((*MockTx)(nil).SelectFromDomainAuditLogs), ctx, filter)\n}\n\n// SelectFromDomainMetadata mocks base method.\nfunc (m *MockTx) SelectFromDomainMetadata(ctx context.Context) (*DomainMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainMetadata\", ctx)\n\tret0, _ := ret[0].(*DomainMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainMetadata indicates an expected call of SelectFromDomainMetadata.\nfunc (mr *MockTxMockRecorder) SelectFromDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainMetadata\", reflect.TypeOf((*MockTx)(nil).SelectFromDomainMetadata), ctx)\n}\n\n// SelectFromExecutions mocks base method.\nfunc (m *MockTx) SelectFromExecutions(ctx context.Context, filter *ExecutionsFilter) ([]ExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]ExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromExecutions indicates an expected call of SelectFromExecutions.\nfunc (mr *MockTxMockRecorder) SelectFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromExecutions\", reflect.TypeOf((*MockTx)(nil).SelectFromExecutions), ctx, filter)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MockTx) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]HistoryNodeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryNodeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MockTxMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MockTx)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MockTx) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MockTxMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MockTx)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// SelectFromReplicationDLQ mocks base method.\nfunc (m *MockTx) SelectFromReplicationDLQ(ctx context.Context, filter *ReplicationTaskDLQFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationDLQ\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationDLQ indicates an expected call of SelectFromReplicationDLQ.\nfunc (mr *MockTxMockRecorder) SelectFromReplicationDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationDLQ\", reflect.TypeOf((*MockTx)(nil).SelectFromReplicationDLQ), ctx, filter)\n}\n\n// SelectFromReplicationTasks mocks base method.\nfunc (m *MockTx) SelectFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasks indicates an expected call of SelectFromReplicationTasks.\nfunc (mr *MockTxMockRecorder) SelectFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasks\", reflect.TypeOf((*MockTx)(nil).SelectFromReplicationTasks), ctx, filter)\n}\n\n// SelectFromReplicationTasksDLQ mocks base method.\nfunc (m *MockTx) SelectFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasksDLQ indicates an expected call of SelectFromReplicationTasksDLQ.\nfunc (mr *MockTxMockRecorder) SelectFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasksDLQ\", reflect.TypeOf((*MockTx)(nil).SelectFromReplicationTasksDLQ), ctx, filter)\n}\n\n// SelectFromRequestCancelInfoMaps mocks base method.\nfunc (m *MockTx) SelectFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) ([]RequestCancelInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]RequestCancelInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromRequestCancelInfoMaps indicates an expected call of SelectFromRequestCancelInfoMaps.\nfunc (mr *MockTxMockRecorder) SelectFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromRequestCancelInfoMaps\", reflect.TypeOf((*MockTx)(nil).SelectFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// SelectFromShards mocks base method.\nfunc (m *MockTx) SelectFromShards(ctx context.Context, filter *ShardsFilter) (*ShardsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromShards\", ctx, filter)\n\tret0, _ := ret[0].(*ShardsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromShards indicates an expected call of SelectFromShards.\nfunc (mr *MockTxMockRecorder) SelectFromShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromShards\", reflect.TypeOf((*MockTx)(nil).SelectFromShards), ctx, filter)\n}\n\n// SelectFromSignalInfoMaps mocks base method.\nfunc (m *MockTx) SelectFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) ([]SignalInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]SignalInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalInfoMaps indicates an expected call of SelectFromSignalInfoMaps.\nfunc (mr *MockTxMockRecorder) SelectFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalInfoMaps\", reflect.TypeOf((*MockTx)(nil).SelectFromSignalInfoMaps), ctx, filter)\n}\n\n// SelectFromSignalsRequestedSets mocks base method.\nfunc (m *MockTx) SelectFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) ([]SignalsRequestedSetsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].([]SignalsRequestedSetsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalsRequestedSets indicates an expected call of SelectFromSignalsRequestedSets.\nfunc (mr *MockTxMockRecorder) SelectFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalsRequestedSets\", reflect.TypeOf((*MockTx)(nil).SelectFromSignalsRequestedSets), ctx, filter)\n}\n\n// SelectFromTaskLists mocks base method.\nfunc (m *MockTx) SelectFromTaskLists(ctx context.Context, filter *TaskListsFilter) ([]TaskListsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].([]TaskListsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTaskLists indicates an expected call of SelectFromTaskLists.\nfunc (mr *MockTxMockRecorder) SelectFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTaskLists\", reflect.TypeOf((*MockTx)(nil).SelectFromTaskLists), ctx, filter)\n}\n\n// SelectFromTasks mocks base method.\nfunc (m *MockTx) SelectFromTasks(ctx context.Context, filter *TasksFilter) ([]TasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTasks indicates an expected call of SelectFromTasks.\nfunc (mr *MockTxMockRecorder) SelectFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTasks\", reflect.TypeOf((*MockTx)(nil).SelectFromTasks), ctx, filter)\n}\n\n// SelectFromTimerInfoMaps mocks base method.\nfunc (m *MockTx) SelectFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) ([]TimerInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]TimerInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerInfoMaps indicates an expected call of SelectFromTimerInfoMaps.\nfunc (mr *MockTxMockRecorder) SelectFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerInfoMaps\", reflect.TypeOf((*MockTx)(nil).SelectFromTimerInfoMaps), ctx, filter)\n}\n\n// SelectFromTimerTasks mocks base method.\nfunc (m *MockTx) SelectFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) ([]TimerTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TimerTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerTasks indicates an expected call of SelectFromTimerTasks.\nfunc (mr *MockTxMockRecorder) SelectFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerTasks\", reflect.TypeOf((*MockTx)(nil).SelectFromTimerTasks), ctx, filter)\n}\n\n// SelectFromTransferTasks mocks base method.\nfunc (m *MockTx) SelectFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) ([]TransferTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TransferTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTransferTasks indicates an expected call of SelectFromTransferTasks.\nfunc (mr *MockTxMockRecorder) SelectFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTransferTasks\", reflect.TypeOf((*MockTx)(nil).SelectFromTransferTasks), ctx, filter)\n}\n\n// SelectFromVisibility mocks base method.\nfunc (m *MockTx) SelectFromVisibility(ctx context.Context, filter *VisibilityFilter) ([]VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].([]VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromVisibility indicates an expected call of SelectFromVisibility.\nfunc (mr *MockTxMockRecorder) SelectFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromVisibility\", reflect.TypeOf((*MockTx)(nil).SelectFromVisibility), ctx, filter)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MockTx) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MockTxMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MockTx)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// SupportsAsyncTransaction mocks base method.\nfunc (m *MockTx) SupportsAsyncTransaction() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsAsyncTransaction\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsAsyncTransaction indicates an expected call of SupportsAsyncTransaction.\nfunc (mr *MockTxMockRecorder) SupportsAsyncTransaction() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsAsyncTransaction\", reflect.TypeOf((*MockTx)(nil).SupportsAsyncTransaction))\n}\n\n// SupportsTTL mocks base method.\nfunc (m *MockTx) SupportsTTL() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsTTL\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsTTL indicates an expected call of SupportsTTL.\nfunc (mr *MockTxMockRecorder) SupportsTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsTTL\", reflect.TypeOf((*MockTx)(nil).SupportsTTL))\n}\n\n// UpdateAckLevels mocks base method.\nfunc (m *MockTx) UpdateAckLevels(ctx context.Context, queueType persistence.QueueType, clusterAckLevels map[string]int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevels\", ctx, queueType, clusterAckLevels)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAckLevels indicates an expected call of UpdateAckLevels.\nfunc (mr *MockTxMockRecorder) UpdateAckLevels(ctx, queueType, clusterAckLevels any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevels\", reflect.TypeOf((*MockTx)(nil).UpdateAckLevels), ctx, queueType, clusterAckLevels)\n}\n\n// UpdateCurrentExecutions mocks base method.\nfunc (m *MockTx) UpdateCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateCurrentExecutions indicates an expected call of UpdateCurrentExecutions.\nfunc (mr *MockTxMockRecorder) UpdateCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateCurrentExecutions\", reflect.TypeOf((*MockTx)(nil).UpdateCurrentExecutions), ctx, row)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockTx) UpdateDomain(ctx context.Context, row *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockTxMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockTx)(nil).UpdateDomain), ctx, row)\n}\n\n// UpdateDomainMetadata mocks base method.\nfunc (m *MockTx) UpdateDomainMetadata(ctx context.Context, row *DomainMetadataRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainMetadata\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainMetadata indicates an expected call of UpdateDomainMetadata.\nfunc (mr *MockTxMockRecorder) UpdateDomainMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainMetadata\", reflect.TypeOf((*MockTx)(nil).UpdateDomainMetadata), ctx, row)\n}\n\n// UpdateExecutions mocks base method.\nfunc (m *MockTx) UpdateExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateExecutions indicates an expected call of UpdateExecutions.\nfunc (mr *MockTxMockRecorder) UpdateExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateExecutions\", reflect.TypeOf((*MockTx)(nil).UpdateExecutions), ctx, row)\n}\n\n// UpdateShards mocks base method.\nfunc (m *MockTx) UpdateShards(ctx context.Context, row *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShards\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateShards indicates an expected call of UpdateShards.\nfunc (mr *MockTxMockRecorder) UpdateShards(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShards\", reflect.TypeOf((*MockTx)(nil).UpdateShards), ctx, row)\n}\n\n// UpdateTaskLists mocks base method.\nfunc (m *MockTx) UpdateTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskLists indicates an expected call of UpdateTaskLists.\nfunc (mr *MockTxMockRecorder) UpdateTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskLists\", reflect.TypeOf((*MockTx)(nil).UpdateTaskLists), ctx, row)\n}\n\n// UpdateTaskListsWithTTL mocks base method.\nfunc (m *MockTx) UpdateTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListsWithTTL indicates an expected call of UpdateTaskListsWithTTL.\nfunc (mr *MockTxMockRecorder) UpdateTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListsWithTTL\", reflect.TypeOf((*MockTx)(nil).UpdateTaskListsWithTTL), ctx, row)\n}\n\n// WriteLockExecutions mocks base method.\nfunc (m *MockTx) WriteLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockExecutions indicates an expected call of WriteLockExecutions.\nfunc (mr *MockTxMockRecorder) WriteLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockExecutions\", reflect.TypeOf((*MockTx)(nil).WriteLockExecutions), ctx, filter)\n}\n\n// WriteLockShards mocks base method.\nfunc (m *MockTx) WriteLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockShards indicates an expected call of WriteLockShards.\nfunc (mr *MockTxMockRecorder) WriteLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockShards\", reflect.TypeOf((*MockTx)(nil).WriteLockShards), ctx, filter)\n}\n\n// MockDB is a mock of DB interface.\ntype MockDB struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDBMockRecorder\n\tisgomock struct{}\n}\n\n// MockDBMockRecorder is the mock recorder for MockDB.\ntype MockDBMockRecorder struct {\n\tmock *MockDB\n}\n\n// NewMockDB creates a new mock instance.\nfunc NewMockDB(ctrl *gomock.Controller) *MockDB {\n\tmock := &MockDB{ctrl: ctrl}\n\tmock.recorder = &MockDBMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDB) EXPECT() *MockDBMockRecorder {\n\treturn m.recorder\n}\n\n// BeginTx mocks base method.\nfunc (m *MockDB) BeginTx(ctx context.Context, dbShardID int) (Tx, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"BeginTx\", ctx, dbShardID)\n\tret0, _ := ret[0].(Tx)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// BeginTx indicates an expected call of BeginTx.\nfunc (mr *MockDBMockRecorder) BeginTx(ctx, dbShardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"BeginTx\", reflect.TypeOf((*MockDB)(nil).BeginTx), ctx, dbShardID)\n}\n\n// Close mocks base method.\nfunc (m *MockDB) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockDBMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockDB)(nil).Close))\n}\n\n// DeleteFromActivityInfoMaps mocks base method.\nfunc (m *MockDB) DeleteFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromActivityInfoMaps indicates an expected call of DeleteFromActivityInfoMaps.\nfunc (mr *MockDBMockRecorder) DeleteFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromActivityInfoMaps\", reflect.TypeOf((*MockDB)(nil).DeleteFromActivityInfoMaps), ctx, filter)\n}\n\n// DeleteFromBufferedEvents mocks base method.\nfunc (m *MockDB) DeleteFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromBufferedEvents indicates an expected call of DeleteFromBufferedEvents.\nfunc (mr *MockDBMockRecorder) DeleteFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromBufferedEvents\", reflect.TypeOf((*MockDB)(nil).DeleteFromBufferedEvents), ctx, filter)\n}\n\n// DeleteFromChildExecutionInfoMaps mocks base method.\nfunc (m *MockDB) DeleteFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromChildExecutionInfoMaps indicates an expected call of DeleteFromChildExecutionInfoMaps.\nfunc (mr *MockDBMockRecorder) DeleteFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromChildExecutionInfoMaps\", reflect.TypeOf((*MockDB)(nil).DeleteFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// DeleteFromCrossClusterTasks mocks base method.\nfunc (m *MockDB) DeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCrossClusterTasks indicates an expected call of DeleteFromCrossClusterTasks.\nfunc (mr *MockDBMockRecorder) DeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCrossClusterTasks\", reflect.TypeOf((*MockDB)(nil).DeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// DeleteFromCurrentExecutions mocks base method.\nfunc (m *MockDB) DeleteFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromCurrentExecutions indicates an expected call of DeleteFromCurrentExecutions.\nfunc (mr *MockDBMockRecorder) DeleteFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromCurrentExecutions\", reflect.TypeOf((*MockDB)(nil).DeleteFromCurrentExecutions), ctx, filter)\n}\n\n// DeleteFromDomain mocks base method.\nfunc (m *MockDB) DeleteFromDomain(ctx context.Context, filter *DomainFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromDomain\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromDomain indicates an expected call of DeleteFromDomain.\nfunc (mr *MockDBMockRecorder) DeleteFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromDomain\", reflect.TypeOf((*MockDB)(nil).DeleteFromDomain), ctx, filter)\n}\n\n// DeleteFromExecutions mocks base method.\nfunc (m *MockDB) DeleteFromExecutions(ctx context.Context, filter *ExecutionsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromExecutions indicates an expected call of DeleteFromExecutions.\nfunc (mr *MockDBMockRecorder) DeleteFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromExecutions\", reflect.TypeOf((*MockDB)(nil).DeleteFromExecutions), ctx, filter)\n}\n\n// DeleteFromHistoryNode mocks base method.\nfunc (m *MockDB) DeleteFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryNode indicates an expected call of DeleteFromHistoryNode.\nfunc (mr *MockDBMockRecorder) DeleteFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryNode\", reflect.TypeOf((*MockDB)(nil).DeleteFromHistoryNode), ctx, filter)\n}\n\n// DeleteFromHistoryTree mocks base method.\nfunc (m *MockDB) DeleteFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromHistoryTree indicates an expected call of DeleteFromHistoryTree.\nfunc (mr *MockDBMockRecorder) DeleteFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromHistoryTree\", reflect.TypeOf((*MockDB)(nil).DeleteFromHistoryTree), ctx, filter)\n}\n\n// DeleteFromReplicationTasks mocks base method.\nfunc (m *MockDB) DeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromReplicationTasks indicates an expected call of DeleteFromReplicationTasks.\nfunc (mr *MockDBMockRecorder) DeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromReplicationTasks\", reflect.TypeOf((*MockDB)(nil).DeleteFromReplicationTasks), ctx, filter)\n}\n\n// DeleteFromRequestCancelInfoMaps mocks base method.\nfunc (m *MockDB) DeleteFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromRequestCancelInfoMaps indicates an expected call of DeleteFromRequestCancelInfoMaps.\nfunc (mr *MockDBMockRecorder) DeleteFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromRequestCancelInfoMaps\", reflect.TypeOf((*MockDB)(nil).DeleteFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalInfoMaps mocks base method.\nfunc (m *MockDB) DeleteFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalInfoMaps indicates an expected call of DeleteFromSignalInfoMaps.\nfunc (mr *MockDBMockRecorder) DeleteFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalInfoMaps\", reflect.TypeOf((*MockDB)(nil).DeleteFromSignalInfoMaps), ctx, filter)\n}\n\n// DeleteFromSignalsRequestedSets mocks base method.\nfunc (m *MockDB) DeleteFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromSignalsRequestedSets indicates an expected call of DeleteFromSignalsRequestedSets.\nfunc (mr *MockDBMockRecorder) DeleteFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromSignalsRequestedSets\", reflect.TypeOf((*MockDB)(nil).DeleteFromSignalsRequestedSets), ctx, filter)\n}\n\n// DeleteFromTaskLists mocks base method.\nfunc (m *MockDB) DeleteFromTaskLists(ctx context.Context, filter *TaskListsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTaskLists indicates an expected call of DeleteFromTaskLists.\nfunc (mr *MockDBMockRecorder) DeleteFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTaskLists\", reflect.TypeOf((*MockDB)(nil).DeleteFromTaskLists), ctx, filter)\n}\n\n// DeleteFromTasks mocks base method.\nfunc (m *MockDB) DeleteFromTasks(ctx context.Context, filter *TasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTasks indicates an expected call of DeleteFromTasks.\nfunc (mr *MockDBMockRecorder) DeleteFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTasks\", reflect.TypeOf((*MockDB)(nil).DeleteFromTasks), ctx, filter)\n}\n\n// DeleteFromTimerInfoMaps mocks base method.\nfunc (m *MockDB) DeleteFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerInfoMaps indicates an expected call of DeleteFromTimerInfoMaps.\nfunc (mr *MockDBMockRecorder) DeleteFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerInfoMaps\", reflect.TypeOf((*MockDB)(nil).DeleteFromTimerInfoMaps), ctx, filter)\n}\n\n// DeleteFromTimerTasks mocks base method.\nfunc (m *MockDB) DeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTimerTasks indicates an expected call of DeleteFromTimerTasks.\nfunc (mr *MockDBMockRecorder) DeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTimerTasks\", reflect.TypeOf((*MockDB)(nil).DeleteFromTimerTasks), ctx, filter)\n}\n\n// DeleteFromTransferTasks mocks base method.\nfunc (m *MockDB) DeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromTransferTasks indicates an expected call of DeleteFromTransferTasks.\nfunc (mr *MockDBMockRecorder) DeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromTransferTasks\", reflect.TypeOf((*MockDB)(nil).DeleteFromTransferTasks), ctx, filter)\n}\n\n// DeleteFromVisibility mocks base method.\nfunc (m *MockDB) DeleteFromVisibility(ctx context.Context, filter *VisibilityFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteFromVisibility indicates an expected call of DeleteFromVisibility.\nfunc (mr *MockDBMockRecorder) DeleteFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFromVisibility\", reflect.TypeOf((*MockDB)(nil).DeleteFromVisibility), ctx, filter)\n}\n\n// DeleteMessage mocks base method.\nfunc (m *MockDB) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessage\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessage indicates an expected call of DeleteMessage.\nfunc (mr *MockDBMockRecorder) DeleteMessage(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessage\", reflect.TypeOf((*MockDB)(nil).DeleteMessage), ctx, queueType, messageID)\n}\n\n// DeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MockDB) DeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessageFromReplicationTasksDLQ indicates an expected call of DeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MockDBMockRecorder) DeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MockDB)(nil).DeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// DeleteMessagesBefore mocks base method.\nfunc (m *MockDB) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteMessagesBefore\", ctx, queueType, messageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteMessagesBefore indicates an expected call of DeleteMessagesBefore.\nfunc (mr *MockDBMockRecorder) DeleteMessagesBefore(ctx, queueType, messageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteMessagesBefore\", reflect.TypeOf((*MockDB)(nil).DeleteMessagesBefore), ctx, queueType, messageID)\n}\n\n// GetAckLevels mocks base method.\nfunc (m *MockDB) GetAckLevels(ctx context.Context, queueType persistence.QueueType, forUpdate bool) (map[string]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAckLevels\", ctx, queueType, forUpdate)\n\tret0, _ := ret[0].(map[string]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAckLevels indicates an expected call of GetAckLevels.\nfunc (mr *MockDBMockRecorder) GetAckLevels(ctx, queueType, forUpdate any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAckLevels\", reflect.TypeOf((*MockDB)(nil).GetAckLevels), ctx, queueType, forUpdate)\n}\n\n// GetAllHistoryTreeBranches mocks base method.\nfunc (m *MockDB) GetAllHistoryTreeBranches(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllHistoryTreeBranches\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetAllHistoryTreeBranches indicates an expected call of GetAllHistoryTreeBranches.\nfunc (mr *MockDBMockRecorder) GetAllHistoryTreeBranches(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllHistoryTreeBranches\", reflect.TypeOf((*MockDB)(nil).GetAllHistoryTreeBranches), ctx, filter)\n}\n\n// GetLastEnqueuedMessageIDForUpdate mocks base method.\nfunc (m *MockDB) GetLastEnqueuedMessageIDForUpdate(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastEnqueuedMessageIDForUpdate\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetLastEnqueuedMessageIDForUpdate indicates an expected call of GetLastEnqueuedMessageIDForUpdate.\nfunc (mr *MockDBMockRecorder) GetLastEnqueuedMessageIDForUpdate(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastEnqueuedMessageIDForUpdate\", reflect.TypeOf((*MockDB)(nil).GetLastEnqueuedMessageIDForUpdate), ctx, queueType)\n}\n\n// GetMessagesBetween mocks base method.\nfunc (m *MockDB) GetMessagesBetween(ctx context.Context, queueType persistence.QueueType, firstMessageID, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesBetween\", ctx, queueType, firstMessageID, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesBetween indicates an expected call of GetMessagesBetween.\nfunc (mr *MockDBMockRecorder) GetMessagesBetween(ctx, queueType, firstMessageID, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesBetween\", reflect.TypeOf((*MockDB)(nil).GetMessagesBetween), ctx, queueType, firstMessageID, lastMessageID, maxRows)\n}\n\n// GetMessagesFromQueue mocks base method.\nfunc (m *MockDB) GetMessagesFromQueue(ctx context.Context, queueType persistence.QueueType, lastMessageID int64, maxRows int) ([]QueueRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagesFromQueue\", ctx, queueType, lastMessageID, maxRows)\n\tret0, _ := ret[0].([]QueueRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMessagesFromQueue indicates an expected call of GetMessagesFromQueue.\nfunc (mr *MockDBMockRecorder) GetMessagesFromQueue(ctx, queueType, lastMessageID, maxRows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagesFromQueue\", reflect.TypeOf((*MockDB)(nil).GetMessagesFromQueue), ctx, queueType, lastMessageID, maxRows)\n}\n\n// GetOrphanTasks mocks base method.\nfunc (m *MockDB) GetOrphanTasks(ctx context.Context, filter *OrphanTasksFilter) ([]TaskKeyRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrphanTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TaskKeyRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetOrphanTasks indicates an expected call of GetOrphanTasks.\nfunc (mr *MockDBMockRecorder) GetOrphanTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrphanTasks\", reflect.TypeOf((*MockDB)(nil).GetOrphanTasks), ctx, filter)\n}\n\n// GetQueueSize mocks base method.\nfunc (m *MockDB) GetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueSize\", ctx, queueType)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueSize indicates an expected call of GetQueueSize.\nfunc (mr *MockDBMockRecorder) GetQueueSize(ctx, queueType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueSize\", reflect.TypeOf((*MockDB)(nil).GetQueueSize), ctx, queueType)\n}\n\n// GetTasksCount mocks base method.\nfunc (m *MockDB) GetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasksCount\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasksCount indicates an expected call of GetTasksCount.\nfunc (mr *MockDBMockRecorder) GetTasksCount(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasksCount\", reflect.TypeOf((*MockDB)(nil).GetTasksCount), ctx, filter)\n}\n\n// GetTotalNumDBShards mocks base method.\nfunc (m *MockDB) GetTotalNumDBShards() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTotalNumDBShards\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetTotalNumDBShards indicates an expected call of GetTotalNumDBShards.\nfunc (mr *MockDBMockRecorder) GetTotalNumDBShards() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTotalNumDBShards\", reflect.TypeOf((*MockDB)(nil).GetTotalNumDBShards))\n}\n\n// InsertAckLevel mocks base method.\nfunc (m *MockDB) InsertAckLevel(ctx context.Context, queueType persistence.QueueType, messageID int64, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertAckLevel\", ctx, queueType, messageID, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertAckLevel indicates an expected call of InsertAckLevel.\nfunc (mr *MockDBMockRecorder) InsertAckLevel(ctx, queueType, messageID, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertAckLevel\", reflect.TypeOf((*MockDB)(nil).InsertAckLevel), ctx, queueType, messageID, clusterName)\n}\n\n// InsertConfig mocks base method.\nfunc (m *MockDB) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertConfig\", ctx, row)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// InsertConfig indicates an expected call of InsertConfig.\nfunc (mr *MockDBMockRecorder) InsertConfig(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertConfig\", reflect.TypeOf((*MockDB)(nil).InsertConfig), ctx, row)\n}\n\n// InsertIntoBufferedEvents mocks base method.\nfunc (m *MockDB) InsertIntoBufferedEvents(ctx context.Context, rows []BufferedEventsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoBufferedEvents\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoBufferedEvents indicates an expected call of InsertIntoBufferedEvents.\nfunc (mr *MockDBMockRecorder) InsertIntoBufferedEvents(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoBufferedEvents\", reflect.TypeOf((*MockDB)(nil).InsertIntoBufferedEvents), ctx, rows)\n}\n\n// InsertIntoCrossClusterTasks mocks base method.\nfunc (m *MockDB) InsertIntoCrossClusterTasks(ctx context.Context, rows []CrossClusterTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCrossClusterTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCrossClusterTasks indicates an expected call of InsertIntoCrossClusterTasks.\nfunc (mr *MockDBMockRecorder) InsertIntoCrossClusterTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCrossClusterTasks\", reflect.TypeOf((*MockDB)(nil).InsertIntoCrossClusterTasks), ctx, rows)\n}\n\n// InsertIntoCurrentExecutions mocks base method.\nfunc (m *MockDB) InsertIntoCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoCurrentExecutions indicates an expected call of InsertIntoCurrentExecutions.\nfunc (mr *MockDBMockRecorder) InsertIntoCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoCurrentExecutions\", reflect.TypeOf((*MockDB)(nil).InsertIntoCurrentExecutions), ctx, row)\n}\n\n// InsertIntoDomain mocks base method.\nfunc (m *MockDB) InsertIntoDomain(ctx context.Context, rows *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomain\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomain indicates an expected call of InsertIntoDomain.\nfunc (mr *MockDBMockRecorder) InsertIntoDomain(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomain\", reflect.TypeOf((*MockDB)(nil).InsertIntoDomain), ctx, rows)\n}\n\n// InsertIntoDomainAuditLog mocks base method.\nfunc (m *MockDB) InsertIntoDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoDomainAuditLog\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoDomainAuditLog indicates an expected call of InsertIntoDomainAuditLog.\nfunc (mr *MockDBMockRecorder) InsertIntoDomainAuditLog(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoDomainAuditLog\", reflect.TypeOf((*MockDB)(nil).InsertIntoDomainAuditLog), ctx, row)\n}\n\n// InsertIntoExecutions mocks base method.\nfunc (m *MockDB) InsertIntoExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoExecutions indicates an expected call of InsertIntoExecutions.\nfunc (mr *MockDBMockRecorder) InsertIntoExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoExecutions\", reflect.TypeOf((*MockDB)(nil).InsertIntoExecutions), ctx, row)\n}\n\n// InsertIntoHistoryNode mocks base method.\nfunc (m *MockDB) InsertIntoHistoryNode(ctx context.Context, row *HistoryNodeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryNode\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryNode indicates an expected call of InsertIntoHistoryNode.\nfunc (mr *MockDBMockRecorder) InsertIntoHistoryNode(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryNode\", reflect.TypeOf((*MockDB)(nil).InsertIntoHistoryNode), ctx, row)\n}\n\n// InsertIntoHistoryTree mocks base method.\nfunc (m *MockDB) InsertIntoHistoryTree(ctx context.Context, row *HistoryTreeRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoHistoryTree\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoHistoryTree indicates an expected call of InsertIntoHistoryTree.\nfunc (mr *MockDBMockRecorder) InsertIntoHistoryTree(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoHistoryTree\", reflect.TypeOf((*MockDB)(nil).InsertIntoHistoryTree), ctx, row)\n}\n\n// InsertIntoQueue mocks base method.\nfunc (m *MockDB) InsertIntoQueue(ctx context.Context, row *QueueRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoQueue\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoQueue indicates an expected call of InsertIntoQueue.\nfunc (mr *MockDBMockRecorder) InsertIntoQueue(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoQueue\", reflect.TypeOf((*MockDB)(nil).InsertIntoQueue), ctx, row)\n}\n\n// InsertIntoReplicationTasks mocks base method.\nfunc (m *MockDB) InsertIntoReplicationTasks(ctx context.Context, rows []ReplicationTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasks indicates an expected call of InsertIntoReplicationTasks.\nfunc (mr *MockDBMockRecorder) InsertIntoReplicationTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasks\", reflect.TypeOf((*MockDB)(nil).InsertIntoReplicationTasks), ctx, rows)\n}\n\n// InsertIntoReplicationTasksDLQ mocks base method.\nfunc (m *MockDB) InsertIntoReplicationTasksDLQ(ctx context.Context, row *ReplicationTaskDLQRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoReplicationTasksDLQ\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoReplicationTasksDLQ indicates an expected call of InsertIntoReplicationTasksDLQ.\nfunc (mr *MockDBMockRecorder) InsertIntoReplicationTasksDLQ(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoReplicationTasksDLQ\", reflect.TypeOf((*MockDB)(nil).InsertIntoReplicationTasksDLQ), ctx, row)\n}\n\n// InsertIntoShards mocks base method.\nfunc (m *MockDB) InsertIntoShards(ctx context.Context, rows *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoShards\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoShards indicates an expected call of InsertIntoShards.\nfunc (mr *MockDBMockRecorder) InsertIntoShards(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoShards\", reflect.TypeOf((*MockDB)(nil).InsertIntoShards), ctx, rows)\n}\n\n// InsertIntoSignalsRequestedSets mocks base method.\nfunc (m *MockDB) InsertIntoSignalsRequestedSets(ctx context.Context, rows []SignalsRequestedSetsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoSignalsRequestedSets\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoSignalsRequestedSets indicates an expected call of InsertIntoSignalsRequestedSets.\nfunc (mr *MockDBMockRecorder) InsertIntoSignalsRequestedSets(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoSignalsRequestedSets\", reflect.TypeOf((*MockDB)(nil).InsertIntoSignalsRequestedSets), ctx, rows)\n}\n\n// InsertIntoTaskLists mocks base method.\nfunc (m *MockDB) InsertIntoTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskLists indicates an expected call of InsertIntoTaskLists.\nfunc (mr *MockDBMockRecorder) InsertIntoTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskLists\", reflect.TypeOf((*MockDB)(nil).InsertIntoTaskLists), ctx, row)\n}\n\n// InsertIntoTaskListsWithTTL mocks base method.\nfunc (m *MockDB) InsertIntoTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTaskListsWithTTL indicates an expected call of InsertIntoTaskListsWithTTL.\nfunc (mr *MockDBMockRecorder) InsertIntoTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTaskListsWithTTL\", reflect.TypeOf((*MockDB)(nil).InsertIntoTaskListsWithTTL), ctx, row)\n}\n\n// InsertIntoTasks mocks base method.\nfunc (m *MockDB) InsertIntoTasks(ctx context.Context, rows []TasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasks indicates an expected call of InsertIntoTasks.\nfunc (mr *MockDBMockRecorder) InsertIntoTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasks\", reflect.TypeOf((*MockDB)(nil).InsertIntoTasks), ctx, rows)\n}\n\n// InsertIntoTasksWithTTL mocks base method.\nfunc (m *MockDB) InsertIntoTasksWithTTL(ctx context.Context, rows []TasksRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTasksWithTTL\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTasksWithTTL indicates an expected call of InsertIntoTasksWithTTL.\nfunc (mr *MockDBMockRecorder) InsertIntoTasksWithTTL(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTasksWithTTL\", reflect.TypeOf((*MockDB)(nil).InsertIntoTasksWithTTL), ctx, rows)\n}\n\n// InsertIntoTimerTasks mocks base method.\nfunc (m *MockDB) InsertIntoTimerTasks(ctx context.Context, rows []TimerTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTimerTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTimerTasks indicates an expected call of InsertIntoTimerTasks.\nfunc (mr *MockDBMockRecorder) InsertIntoTimerTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTimerTasks\", reflect.TypeOf((*MockDB)(nil).InsertIntoTimerTasks), ctx, rows)\n}\n\n// InsertIntoTransferTasks mocks base method.\nfunc (m *MockDB) InsertIntoTransferTasks(ctx context.Context, rows []TransferTasksRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoTransferTasks\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoTransferTasks indicates an expected call of InsertIntoTransferTasks.\nfunc (mr *MockDBMockRecorder) InsertIntoTransferTasks(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoTransferTasks\", reflect.TypeOf((*MockDB)(nil).InsertIntoTransferTasks), ctx, rows)\n}\n\n// InsertIntoVisibility mocks base method.\nfunc (m *MockDB) InsertIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"InsertIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// InsertIntoVisibility indicates an expected call of InsertIntoVisibility.\nfunc (mr *MockDBMockRecorder) InsertIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"InsertIntoVisibility\", reflect.TypeOf((*MockDB)(nil).InsertIntoVisibility), ctx, row)\n}\n\n// IsDupEntryError mocks base method.\nfunc (m *MockDB) IsDupEntryError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDupEntryError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDupEntryError indicates an expected call of IsDupEntryError.\nfunc (mr *MockDBMockRecorder) IsDupEntryError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDupEntryError\", reflect.TypeOf((*MockDB)(nil).IsDupEntryError), err)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockDB) IsNotFoundError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockDBMockRecorder) IsNotFoundError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockDB)(nil).IsNotFoundError), err)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockDB) IsThrottlingError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockDBMockRecorder) IsThrottlingError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockDB)(nil).IsThrottlingError), err)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockDB) IsTimeoutError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockDBMockRecorder) IsTimeoutError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockDB)(nil).IsTimeoutError), err)\n}\n\n// LockCurrentExecutions mocks base method.\nfunc (m *MockDB) LockCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutions indicates an expected call of LockCurrentExecutions.\nfunc (mr *MockDBMockRecorder) LockCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutions\", reflect.TypeOf((*MockDB)(nil).LockCurrentExecutions), ctx, filter)\n}\n\n// LockCurrentExecutionsJoinExecutions mocks base method.\nfunc (m *MockDB) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *CurrentExecutionsFilter) ([]CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockCurrentExecutionsJoinExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockCurrentExecutionsJoinExecutions indicates an expected call of LockCurrentExecutionsJoinExecutions.\nfunc (mr *MockDBMockRecorder) LockCurrentExecutionsJoinExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockCurrentExecutionsJoinExecutions\", reflect.TypeOf((*MockDB)(nil).LockCurrentExecutionsJoinExecutions), ctx, filter)\n}\n\n// LockDomainMetadata mocks base method.\nfunc (m *MockDB) LockDomainMetadata(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockDomainMetadata\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// LockDomainMetadata indicates an expected call of LockDomainMetadata.\nfunc (mr *MockDBMockRecorder) LockDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockDomainMetadata\", reflect.TypeOf((*MockDB)(nil).LockDomainMetadata), ctx)\n}\n\n// LockTaskLists mocks base method.\nfunc (m *MockDB) LockTaskLists(ctx context.Context, filter *TaskListsFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LockTaskLists\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LockTaskLists indicates an expected call of LockTaskLists.\nfunc (mr *MockDBMockRecorder) LockTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockTaskLists\", reflect.TypeOf((*MockDB)(nil).LockTaskLists), ctx, filter)\n}\n\n// MaxAllowedTTL mocks base method.\nfunc (m *MockDB) MaxAllowedTTL() (*time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MaxAllowedTTL\")\n\tret0, _ := ret[0].(*time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MaxAllowedTTL indicates an expected call of MaxAllowedTTL.\nfunc (mr *MockDBMockRecorder) MaxAllowedTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaxAllowedTTL\", reflect.TypeOf((*MockDB)(nil).MaxAllowedTTL))\n}\n\n// PluginName mocks base method.\nfunc (m *MockDB) PluginName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PluginName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PluginName indicates an expected call of PluginName.\nfunc (mr *MockDBMockRecorder) PluginName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PluginName\", reflect.TypeOf((*MockDB)(nil).PluginName))\n}\n\n// RangeDeleteFromCrossClusterTasks mocks base method.\nfunc (m *MockDB) RangeDeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromCrossClusterTasks indicates an expected call of RangeDeleteFromCrossClusterTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromCrossClusterTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteFromCrossClusterTasks), ctx, filter)\n}\n\n// RangeDeleteFromReplicationTasks mocks base method.\nfunc (m *MockDB) RangeDeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromReplicationTasks indicates an expected call of RangeDeleteFromReplicationTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromReplicationTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteFromReplicationTasks), ctx, filter)\n}\n\n// RangeDeleteFromTimerTasks mocks base method.\nfunc (m *MockDB) RangeDeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTimerTasks indicates an expected call of RangeDeleteFromTimerTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTimerTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteFromTimerTasks), ctx, filter)\n}\n\n// RangeDeleteFromTransferTasks mocks base method.\nfunc (m *MockDB) RangeDeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteFromTransferTasks indicates an expected call of RangeDeleteFromTransferTasks.\nfunc (mr *MockDBMockRecorder) RangeDeleteFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteFromTransferTasks\", reflect.TypeOf((*MockDB)(nil).RangeDeleteFromTransferTasks), ctx, filter)\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ mocks base method.\nfunc (m *MockDB) RangeDeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessageFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessageFromReplicationTasksDLQ indicates an expected call of RangeDeleteMessageFromReplicationTasksDLQ.\nfunc (mr *MockDBMockRecorder) RangeDeleteMessageFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessageFromReplicationTasksDLQ\", reflect.TypeOf((*MockDB)(nil).RangeDeleteMessageFromReplicationTasksDLQ), ctx, filter)\n}\n\n// RangeDeleteMessages mocks base method.\nfunc (m *MockDB) RangeDeleteMessages(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID, inclusiveEndMessageID int64) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RangeDeleteMessages\", ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RangeDeleteMessages indicates an expected call of RangeDeleteMessages.\nfunc (mr *MockDBMockRecorder) RangeDeleteMessages(ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RangeDeleteMessages\", reflect.TypeOf((*MockDB)(nil).RangeDeleteMessages), ctx, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// ReadLockExecutions mocks base method.\nfunc (m *MockDB) ReadLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockExecutions indicates an expected call of ReadLockExecutions.\nfunc (mr *MockDBMockRecorder) ReadLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockExecutions\", reflect.TypeOf((*MockDB)(nil).ReadLockExecutions), ctx, filter)\n}\n\n// ReadLockShards mocks base method.\nfunc (m *MockDB) ReadLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadLockShards indicates an expected call of ReadLockShards.\nfunc (mr *MockDBMockRecorder) ReadLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLockShards\", reflect.TypeOf((*MockDB)(nil).ReadLockShards), ctx, filter)\n}\n\n// ReplaceIntoActivityInfoMaps mocks base method.\nfunc (m *MockDB) ReplaceIntoActivityInfoMaps(ctx context.Context, rows []ActivityInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoActivityInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoActivityInfoMaps indicates an expected call of ReplaceIntoActivityInfoMaps.\nfunc (mr *MockDBMockRecorder) ReplaceIntoActivityInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoActivityInfoMaps\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoActivityInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoChildExecutionInfoMaps mocks base method.\nfunc (m *MockDB) ReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []ChildExecutionInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoChildExecutionInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoChildExecutionInfoMaps indicates an expected call of ReplaceIntoChildExecutionInfoMaps.\nfunc (mr *MockDBMockRecorder) ReplaceIntoChildExecutionInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoChildExecutionInfoMaps\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoChildExecutionInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoRequestCancelInfoMaps mocks base method.\nfunc (m *MockDB) ReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []RequestCancelInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoRequestCancelInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoRequestCancelInfoMaps indicates an expected call of ReplaceIntoRequestCancelInfoMaps.\nfunc (mr *MockDBMockRecorder) ReplaceIntoRequestCancelInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoRequestCancelInfoMaps\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoRequestCancelInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoSignalInfoMaps mocks base method.\nfunc (m *MockDB) ReplaceIntoSignalInfoMaps(ctx context.Context, rows []SignalInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoSignalInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoSignalInfoMaps indicates an expected call of ReplaceIntoSignalInfoMaps.\nfunc (mr *MockDBMockRecorder) ReplaceIntoSignalInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoSignalInfoMaps\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoSignalInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoTimerInfoMaps mocks base method.\nfunc (m *MockDB) ReplaceIntoTimerInfoMaps(ctx context.Context, rows []TimerInfoMapsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoTimerInfoMaps\", ctx, rows)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoTimerInfoMaps indicates an expected call of ReplaceIntoTimerInfoMaps.\nfunc (mr *MockDBMockRecorder) ReplaceIntoTimerInfoMaps(ctx, rows any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoTimerInfoMaps\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoTimerInfoMaps), ctx, rows)\n}\n\n// ReplaceIntoVisibility mocks base method.\nfunc (m *MockDB) ReplaceIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplaceIntoVisibility\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplaceIntoVisibility indicates an expected call of ReplaceIntoVisibility.\nfunc (mr *MockDBMockRecorder) ReplaceIntoVisibility(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplaceIntoVisibility\", reflect.TypeOf((*MockDB)(nil).ReplaceIntoVisibility), ctx, row)\n}\n\n// SelectFromActivityInfoMaps mocks base method.\nfunc (m *MockDB) SelectFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) ([]ActivityInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromActivityInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ActivityInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromActivityInfoMaps indicates an expected call of SelectFromActivityInfoMaps.\nfunc (mr *MockDBMockRecorder) SelectFromActivityInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromActivityInfoMaps\", reflect.TypeOf((*MockDB)(nil).SelectFromActivityInfoMaps), ctx, filter)\n}\n\n// SelectFromBufferedEvents mocks base method.\nfunc (m *MockDB) SelectFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) ([]BufferedEventsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromBufferedEvents\", ctx, filter)\n\tret0, _ := ret[0].([]BufferedEventsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromBufferedEvents indicates an expected call of SelectFromBufferedEvents.\nfunc (mr *MockDBMockRecorder) SelectFromBufferedEvents(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromBufferedEvents\", reflect.TypeOf((*MockDB)(nil).SelectFromBufferedEvents), ctx, filter)\n}\n\n// SelectFromChildExecutionInfoMaps mocks base method.\nfunc (m *MockDB) SelectFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) ([]ChildExecutionInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromChildExecutionInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]ChildExecutionInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromChildExecutionInfoMaps indicates an expected call of SelectFromChildExecutionInfoMaps.\nfunc (mr *MockDBMockRecorder) SelectFromChildExecutionInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromChildExecutionInfoMaps\", reflect.TypeOf((*MockDB)(nil).SelectFromChildExecutionInfoMaps), ctx, filter)\n}\n\n// SelectFromCrossClusterTasks mocks base method.\nfunc (m *MockDB) SelectFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) ([]CrossClusterTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCrossClusterTasks\", ctx, filter)\n\tret0, _ := ret[0].([]CrossClusterTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCrossClusterTasks indicates an expected call of SelectFromCrossClusterTasks.\nfunc (mr *MockDBMockRecorder) SelectFromCrossClusterTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCrossClusterTasks\", reflect.TypeOf((*MockDB)(nil).SelectFromCrossClusterTasks), ctx, filter)\n}\n\n// SelectFromCurrentExecutions mocks base method.\nfunc (m *MockDB) SelectFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromCurrentExecutions\", ctx, filter)\n\tret0, _ := ret[0].(*CurrentExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromCurrentExecutions indicates an expected call of SelectFromCurrentExecutions.\nfunc (mr *MockDBMockRecorder) SelectFromCurrentExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromCurrentExecutions\", reflect.TypeOf((*MockDB)(nil).SelectFromCurrentExecutions), ctx, filter)\n}\n\n// SelectFromDomain mocks base method.\nfunc (m *MockDB) SelectFromDomain(ctx context.Context, filter *DomainFilter) ([]DomainRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomain\", ctx, filter)\n\tret0, _ := ret[0].([]DomainRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomain indicates an expected call of SelectFromDomain.\nfunc (mr *MockDBMockRecorder) SelectFromDomain(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomain\", reflect.TypeOf((*MockDB)(nil).SelectFromDomain), ctx, filter)\n}\n\n// SelectFromDomainAuditLogs mocks base method.\nfunc (m *MockDB) SelectFromDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainAuditLogs\", ctx, filter)\n\tret0, _ := ret[0].([]*DomainAuditLogRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainAuditLogs indicates an expected call of SelectFromDomainAuditLogs.\nfunc (mr *MockDBMockRecorder) SelectFromDomainAuditLogs(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainAuditLogs\", reflect.TypeOf((*MockDB)(nil).SelectFromDomainAuditLogs), ctx, filter)\n}\n\n// SelectFromDomainMetadata mocks base method.\nfunc (m *MockDB) SelectFromDomainMetadata(ctx context.Context) (*DomainMetadataRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromDomainMetadata\", ctx)\n\tret0, _ := ret[0].(*DomainMetadataRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromDomainMetadata indicates an expected call of SelectFromDomainMetadata.\nfunc (mr *MockDBMockRecorder) SelectFromDomainMetadata(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromDomainMetadata\", reflect.TypeOf((*MockDB)(nil).SelectFromDomainMetadata), ctx)\n}\n\n// SelectFromExecutions mocks base method.\nfunc (m *MockDB) SelectFromExecutions(ctx context.Context, filter *ExecutionsFilter) ([]ExecutionsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromExecutions\", ctx, filter)\n\tret0, _ := ret[0].([]ExecutionsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromExecutions indicates an expected call of SelectFromExecutions.\nfunc (mr *MockDBMockRecorder) SelectFromExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromExecutions\", reflect.TypeOf((*MockDB)(nil).SelectFromExecutions), ctx, filter)\n}\n\n// SelectFromHistoryNode mocks base method.\nfunc (m *MockDB) SelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]HistoryNodeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryNode\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryNodeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryNode indicates an expected call of SelectFromHistoryNode.\nfunc (mr *MockDBMockRecorder) SelectFromHistoryNode(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryNode\", reflect.TypeOf((*MockDB)(nil).SelectFromHistoryNode), ctx, filter)\n}\n\n// SelectFromHistoryTree mocks base method.\nfunc (m *MockDB) SelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromHistoryTree\", ctx, filter)\n\tret0, _ := ret[0].([]HistoryTreeRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromHistoryTree indicates an expected call of SelectFromHistoryTree.\nfunc (mr *MockDBMockRecorder) SelectFromHistoryTree(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromHistoryTree\", reflect.TypeOf((*MockDB)(nil).SelectFromHistoryTree), ctx, filter)\n}\n\n// SelectFromReplicationDLQ mocks base method.\nfunc (m *MockDB) SelectFromReplicationDLQ(ctx context.Context, filter *ReplicationTaskDLQFilter) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationDLQ\", ctx, filter)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationDLQ indicates an expected call of SelectFromReplicationDLQ.\nfunc (mr *MockDBMockRecorder) SelectFromReplicationDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationDLQ\", reflect.TypeOf((*MockDB)(nil).SelectFromReplicationDLQ), ctx, filter)\n}\n\n// SelectFromReplicationTasks mocks base method.\nfunc (m *MockDB) SelectFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasks\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasks indicates an expected call of SelectFromReplicationTasks.\nfunc (mr *MockDBMockRecorder) SelectFromReplicationTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasks\", reflect.TypeOf((*MockDB)(nil).SelectFromReplicationTasks), ctx, filter)\n}\n\n// SelectFromReplicationTasksDLQ mocks base method.\nfunc (m *MockDB) SelectFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) ([]ReplicationTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromReplicationTasksDLQ\", ctx, filter)\n\tret0, _ := ret[0].([]ReplicationTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromReplicationTasksDLQ indicates an expected call of SelectFromReplicationTasksDLQ.\nfunc (mr *MockDBMockRecorder) SelectFromReplicationTasksDLQ(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromReplicationTasksDLQ\", reflect.TypeOf((*MockDB)(nil).SelectFromReplicationTasksDLQ), ctx, filter)\n}\n\n// SelectFromRequestCancelInfoMaps mocks base method.\nfunc (m *MockDB) SelectFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) ([]RequestCancelInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromRequestCancelInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]RequestCancelInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromRequestCancelInfoMaps indicates an expected call of SelectFromRequestCancelInfoMaps.\nfunc (mr *MockDBMockRecorder) SelectFromRequestCancelInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromRequestCancelInfoMaps\", reflect.TypeOf((*MockDB)(nil).SelectFromRequestCancelInfoMaps), ctx, filter)\n}\n\n// SelectFromShards mocks base method.\nfunc (m *MockDB) SelectFromShards(ctx context.Context, filter *ShardsFilter) (*ShardsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromShards\", ctx, filter)\n\tret0, _ := ret[0].(*ShardsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromShards indicates an expected call of SelectFromShards.\nfunc (mr *MockDBMockRecorder) SelectFromShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromShards\", reflect.TypeOf((*MockDB)(nil).SelectFromShards), ctx, filter)\n}\n\n// SelectFromSignalInfoMaps mocks base method.\nfunc (m *MockDB) SelectFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) ([]SignalInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]SignalInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalInfoMaps indicates an expected call of SelectFromSignalInfoMaps.\nfunc (mr *MockDBMockRecorder) SelectFromSignalInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalInfoMaps\", reflect.TypeOf((*MockDB)(nil).SelectFromSignalInfoMaps), ctx, filter)\n}\n\n// SelectFromSignalsRequestedSets mocks base method.\nfunc (m *MockDB) SelectFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) ([]SignalsRequestedSetsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromSignalsRequestedSets\", ctx, filter)\n\tret0, _ := ret[0].([]SignalsRequestedSetsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromSignalsRequestedSets indicates an expected call of SelectFromSignalsRequestedSets.\nfunc (mr *MockDBMockRecorder) SelectFromSignalsRequestedSets(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromSignalsRequestedSets\", reflect.TypeOf((*MockDB)(nil).SelectFromSignalsRequestedSets), ctx, filter)\n}\n\n// SelectFromTaskLists mocks base method.\nfunc (m *MockDB) SelectFromTaskLists(ctx context.Context, filter *TaskListsFilter) ([]TaskListsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTaskLists\", ctx, filter)\n\tret0, _ := ret[0].([]TaskListsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTaskLists indicates an expected call of SelectFromTaskLists.\nfunc (mr *MockDBMockRecorder) SelectFromTaskLists(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTaskLists\", reflect.TypeOf((*MockDB)(nil).SelectFromTaskLists), ctx, filter)\n}\n\n// SelectFromTasks mocks base method.\nfunc (m *MockDB) SelectFromTasks(ctx context.Context, filter *TasksFilter) ([]TasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTasks indicates an expected call of SelectFromTasks.\nfunc (mr *MockDBMockRecorder) SelectFromTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTasks\", reflect.TypeOf((*MockDB)(nil).SelectFromTasks), ctx, filter)\n}\n\n// SelectFromTimerInfoMaps mocks base method.\nfunc (m *MockDB) SelectFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) ([]TimerInfoMapsRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerInfoMaps\", ctx, filter)\n\tret0, _ := ret[0].([]TimerInfoMapsRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerInfoMaps indicates an expected call of SelectFromTimerInfoMaps.\nfunc (mr *MockDBMockRecorder) SelectFromTimerInfoMaps(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerInfoMaps\", reflect.TypeOf((*MockDB)(nil).SelectFromTimerInfoMaps), ctx, filter)\n}\n\n// SelectFromTimerTasks mocks base method.\nfunc (m *MockDB) SelectFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) ([]TimerTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTimerTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TimerTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTimerTasks indicates an expected call of SelectFromTimerTasks.\nfunc (mr *MockDBMockRecorder) SelectFromTimerTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTimerTasks\", reflect.TypeOf((*MockDB)(nil).SelectFromTimerTasks), ctx, filter)\n}\n\n// SelectFromTransferTasks mocks base method.\nfunc (m *MockDB) SelectFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) ([]TransferTasksRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromTransferTasks\", ctx, filter)\n\tret0, _ := ret[0].([]TransferTasksRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromTransferTasks indicates an expected call of SelectFromTransferTasks.\nfunc (mr *MockDBMockRecorder) SelectFromTransferTasks(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromTransferTasks\", reflect.TypeOf((*MockDB)(nil).SelectFromTransferTasks), ctx, filter)\n}\n\n// SelectFromVisibility mocks base method.\nfunc (m *MockDB) SelectFromVisibility(ctx context.Context, filter *VisibilityFilter) ([]VisibilityRow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectFromVisibility\", ctx, filter)\n\tret0, _ := ret[0].([]VisibilityRow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectFromVisibility indicates an expected call of SelectFromVisibility.\nfunc (mr *MockDBMockRecorder) SelectFromVisibility(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectFromVisibility\", reflect.TypeOf((*MockDB)(nil).SelectFromVisibility), ctx, filter)\n}\n\n// SelectLatestConfig mocks base method.\nfunc (m *MockDB) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SelectLatestConfig\", ctx, rowType)\n\tret0, _ := ret[0].(*persistence.InternalConfigStoreEntry)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SelectLatestConfig indicates an expected call of SelectLatestConfig.\nfunc (mr *MockDBMockRecorder) SelectLatestConfig(ctx, rowType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SelectLatestConfig\", reflect.TypeOf((*MockDB)(nil).SelectLatestConfig), ctx, rowType)\n}\n\n// SupportsAsyncTransaction mocks base method.\nfunc (m *MockDB) SupportsAsyncTransaction() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsAsyncTransaction\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsAsyncTransaction indicates an expected call of SupportsAsyncTransaction.\nfunc (mr *MockDBMockRecorder) SupportsAsyncTransaction() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsAsyncTransaction\", reflect.TypeOf((*MockDB)(nil).SupportsAsyncTransaction))\n}\n\n// SupportsTTL mocks base method.\nfunc (m *MockDB) SupportsTTL() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SupportsTTL\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// SupportsTTL indicates an expected call of SupportsTTL.\nfunc (mr *MockDBMockRecorder) SupportsTTL() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SupportsTTL\", reflect.TypeOf((*MockDB)(nil).SupportsTTL))\n}\n\n// UpdateAckLevels mocks base method.\nfunc (m *MockDB) UpdateAckLevels(ctx context.Context, queueType persistence.QueueType, clusterAckLevels map[string]int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevels\", ctx, queueType, clusterAckLevels)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateAckLevels indicates an expected call of UpdateAckLevels.\nfunc (mr *MockDBMockRecorder) UpdateAckLevels(ctx, queueType, clusterAckLevels any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevels\", reflect.TypeOf((*MockDB)(nil).UpdateAckLevels), ctx, queueType, clusterAckLevels)\n}\n\n// UpdateCurrentExecutions mocks base method.\nfunc (m *MockDB) UpdateCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateCurrentExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateCurrentExecutions indicates an expected call of UpdateCurrentExecutions.\nfunc (mr *MockDBMockRecorder) UpdateCurrentExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateCurrentExecutions\", reflect.TypeOf((*MockDB)(nil).UpdateCurrentExecutions), ctx, row)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockDB) UpdateDomain(ctx context.Context, row *DomainRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockDBMockRecorder) UpdateDomain(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockDB)(nil).UpdateDomain), ctx, row)\n}\n\n// UpdateDomainMetadata mocks base method.\nfunc (m *MockDB) UpdateDomainMetadata(ctx context.Context, row *DomainMetadataRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainMetadata\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainMetadata indicates an expected call of UpdateDomainMetadata.\nfunc (mr *MockDBMockRecorder) UpdateDomainMetadata(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainMetadata\", reflect.TypeOf((*MockDB)(nil).UpdateDomainMetadata), ctx, row)\n}\n\n// UpdateExecutions mocks base method.\nfunc (m *MockDB) UpdateExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateExecutions\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateExecutions indicates an expected call of UpdateExecutions.\nfunc (mr *MockDBMockRecorder) UpdateExecutions(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateExecutions\", reflect.TypeOf((*MockDB)(nil).UpdateExecutions), ctx, row)\n}\n\n// UpdateShards mocks base method.\nfunc (m *MockDB) UpdateShards(ctx context.Context, row *ShardsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateShards\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateShards indicates an expected call of UpdateShards.\nfunc (mr *MockDBMockRecorder) UpdateShards(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateShards\", reflect.TypeOf((*MockDB)(nil).UpdateShards), ctx, row)\n}\n\n// UpdateTaskLists mocks base method.\nfunc (m *MockDB) UpdateTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskLists\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskLists indicates an expected call of UpdateTaskLists.\nfunc (mr *MockDBMockRecorder) UpdateTaskLists(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskLists\", reflect.TypeOf((*MockDB)(nil).UpdateTaskLists), ctx, row)\n}\n\n// UpdateTaskListsWithTTL mocks base method.\nfunc (m *MockDB) UpdateTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListsWithTTL\", ctx, row)\n\tret0, _ := ret[0].(sql.Result)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListsWithTTL indicates an expected call of UpdateTaskListsWithTTL.\nfunc (mr *MockDBMockRecorder) UpdateTaskListsWithTTL(ctx, row any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListsWithTTL\", reflect.TypeOf((*MockDB)(nil).UpdateTaskListsWithTTL), ctx, row)\n}\n\n// WriteLockExecutions mocks base method.\nfunc (m *MockDB) WriteLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockExecutions\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockExecutions indicates an expected call of WriteLockExecutions.\nfunc (mr *MockDBMockRecorder) WriteLockExecutions(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockExecutions\", reflect.TypeOf((*MockDB)(nil).WriteLockExecutions), ctx, filter)\n}\n\n// WriteLockShards mocks base method.\nfunc (m *MockDB) WriteLockShards(ctx context.Context, filter *ShardsFilter) (int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteLockShards\", ctx, filter)\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// WriteLockShards indicates an expected call of WriteLockShards.\nfunc (mr *MockDBMockRecorder) WriteLockShards(ctx, filter any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteLockShards\", reflect.TypeOf((*MockDB)(nil).WriteLockShards), ctx, filter)\n}\n\n// MockAdminDB is a mock of AdminDB interface.\ntype MockAdminDB struct {\n\tctrl     *gomock.Controller\n\trecorder *MockAdminDBMockRecorder\n\tisgomock struct{}\n}\n\n// MockAdminDBMockRecorder is the mock recorder for MockAdminDB.\ntype MockAdminDBMockRecorder struct {\n\tmock *MockAdminDB\n}\n\n// NewMockAdminDB creates a new mock instance.\nfunc NewMockAdminDB(ctrl *gomock.Controller) *MockAdminDB {\n\tmock := &MockAdminDB{ctrl: ctrl}\n\tmock.recorder = &MockAdminDBMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockAdminDB) EXPECT() *MockAdminDBMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockAdminDB) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockAdminDBMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockAdminDB)(nil).Close))\n}\n\n// CreateDatabase mocks base method.\nfunc (m *MockAdminDB) CreateDatabase(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateDatabase\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateDatabase indicates an expected call of CreateDatabase.\nfunc (mr *MockAdminDBMockRecorder) CreateDatabase(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateDatabase\", reflect.TypeOf((*MockAdminDB)(nil).CreateDatabase), database)\n}\n\n// CreateSchemaVersionTables mocks base method.\nfunc (m *MockAdminDB) CreateSchemaVersionTables() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateSchemaVersionTables\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateSchemaVersionTables indicates an expected call of CreateSchemaVersionTables.\nfunc (mr *MockAdminDBMockRecorder) CreateSchemaVersionTables() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateSchemaVersionTables\", reflect.TypeOf((*MockAdminDB)(nil).CreateSchemaVersionTables))\n}\n\n// DropAllTables mocks base method.\nfunc (m *MockAdminDB) DropAllTables(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropAllTables\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropAllTables indicates an expected call of DropAllTables.\nfunc (mr *MockAdminDBMockRecorder) DropAllTables(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropAllTables\", reflect.TypeOf((*MockAdminDB)(nil).DropAllTables), database)\n}\n\n// DropDatabase mocks base method.\nfunc (m *MockAdminDB) DropDatabase(database string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropDatabase\", database)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropDatabase indicates an expected call of DropDatabase.\nfunc (mr *MockAdminDBMockRecorder) DropDatabase(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropDatabase\", reflect.TypeOf((*MockAdminDB)(nil).DropDatabase), database)\n}\n\n// DropTable mocks base method.\nfunc (m *MockAdminDB) DropTable(table string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropTable\", table)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropTable indicates an expected call of DropTable.\nfunc (mr *MockAdminDBMockRecorder) DropTable(table any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropTable\", reflect.TypeOf((*MockAdminDB)(nil).DropTable), table)\n}\n\n// ExecSchemaOperationQuery mocks base method.\nfunc (m *MockAdminDB) ExecSchemaOperationQuery(ctx context.Context, stmt string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, stmt}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecSchemaOperationQuery\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ExecSchemaOperationQuery indicates an expected call of ExecSchemaOperationQuery.\nfunc (mr *MockAdminDBMockRecorder) ExecSchemaOperationQuery(ctx, stmt any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, stmt}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecSchemaOperationQuery\", reflect.TypeOf((*MockAdminDB)(nil).ExecSchemaOperationQuery), varargs...)\n}\n\n// ListTables mocks base method.\nfunc (m *MockAdminDB) ListTables(database string) ([]string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTables\", database)\n\tret0, _ := ret[0].([]string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTables indicates an expected call of ListTables.\nfunc (mr *MockAdminDBMockRecorder) ListTables(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTables\", reflect.TypeOf((*MockAdminDB)(nil).ListTables), database)\n}\n\n// PluginName mocks base method.\nfunc (m *MockAdminDB) PluginName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PluginName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// PluginName indicates an expected call of PluginName.\nfunc (mr *MockAdminDBMockRecorder) PluginName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PluginName\", reflect.TypeOf((*MockAdminDB)(nil).PluginName))\n}\n\n// ReadSchemaVersion mocks base method.\nfunc (m *MockAdminDB) ReadSchemaVersion(database string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadSchemaVersion\", database)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadSchemaVersion indicates an expected call of ReadSchemaVersion.\nfunc (mr *MockAdminDBMockRecorder) ReadSchemaVersion(database any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadSchemaVersion\", reflect.TypeOf((*MockAdminDB)(nil).ReadSchemaVersion), database)\n}\n\n// UpdateSchemaVersion mocks base method.\nfunc (m *MockAdminDB) UpdateSchemaVersion(database, newVersion, minCompatibleVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateSchemaVersion\", database, newVersion, minCompatibleVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateSchemaVersion indicates an expected call of UpdateSchemaVersion.\nfunc (mr *MockAdminDBMockRecorder) UpdateSchemaVersion(database, newVersion, minCompatibleVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateSchemaVersion\", reflect.TypeOf((*MockAdminDB)(nil).UpdateSchemaVersion), database, newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog mocks base method.\nfunc (m *MockAdminDB) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteSchemaUpdateLog\", oldVersion, newVersion, manifestMD5, desc)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// WriteSchemaUpdateLog indicates an expected call of WriteSchemaUpdateLog.\nfunc (mr *MockAdminDBMockRecorder) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteSchemaUpdateLog\", reflect.TypeOf((*MockAdminDB)(nil).WriteSchemaUpdateLog), oldVersion, newVersion, manifestMD5, desc)\n}\n\n// MockErrorChecker is a mock of ErrorChecker interface.\ntype MockErrorChecker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockErrorCheckerMockRecorder\n\tisgomock struct{}\n}\n\n// MockErrorCheckerMockRecorder is the mock recorder for MockErrorChecker.\ntype MockErrorCheckerMockRecorder struct {\n\tmock *MockErrorChecker\n}\n\n// NewMockErrorChecker creates a new mock instance.\nfunc NewMockErrorChecker(ctrl *gomock.Controller) *MockErrorChecker {\n\tmock := &MockErrorChecker{ctrl: ctrl}\n\tmock.recorder = &MockErrorCheckerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockErrorChecker) EXPECT() *MockErrorCheckerMockRecorder {\n\treturn m.recorder\n}\n\n// IsDupEntryError mocks base method.\nfunc (m *MockErrorChecker) IsDupEntryError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsDupEntryError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsDupEntryError indicates an expected call of IsDupEntryError.\nfunc (mr *MockErrorCheckerMockRecorder) IsDupEntryError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsDupEntryError\", reflect.TypeOf((*MockErrorChecker)(nil).IsDupEntryError), err)\n}\n\n// IsNotFoundError mocks base method.\nfunc (m *MockErrorChecker) IsNotFoundError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsNotFoundError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsNotFoundError indicates an expected call of IsNotFoundError.\nfunc (mr *MockErrorCheckerMockRecorder) IsNotFoundError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsNotFoundError\", reflect.TypeOf((*MockErrorChecker)(nil).IsNotFoundError), err)\n}\n\n// IsThrottlingError mocks base method.\nfunc (m *MockErrorChecker) IsThrottlingError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsThrottlingError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsThrottlingError indicates an expected call of IsThrottlingError.\nfunc (mr *MockErrorCheckerMockRecorder) IsThrottlingError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsThrottlingError\", reflect.TypeOf((*MockErrorChecker)(nil).IsThrottlingError), err)\n}\n\n// IsTimeoutError mocks base method.\nfunc (m *MockErrorChecker) IsTimeoutError(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsTimeoutError\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsTimeoutError indicates an expected call of IsTimeoutError.\nfunc (mr *MockErrorCheckerMockRecorder) IsTimeoutError(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsTimeoutError\", reflect.TypeOf((*MockErrorChecker)(nil).IsTimeoutError), err)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/interfaces.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/persistence/sql/sqlplugin\n\npackage sqlplugin\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n)\n\nvar (\n\t// ErrTTLNotSupported indicates the sql plugin does not support ttl\n\tErrTTLNotSupported = errors.New(\"plugin implementation does not support ttl\")\n)\n\ntype (\n\t// Plugin defines the interface for any SQL database that needs to implement\n\tPlugin interface {\n\t\tCreateDB(cfg *config.SQL) (DB, error)\n\t\tCreateAdminDB(cfg *config.SQL) (AdminDB, error)\n\t}\n\n\t// DomainRow represents a row in domain table\n\tDomainRow struct {\n\t\tID           serialization.UUID\n\t\tName         string\n\t\tData         []byte\n\t\tDataEncoding string\n\t\tIsGlobal     bool\n\t}\n\n\t// DomainFilter contains the column names within domain table that\n\t// can be used to filter results through a WHERE clause. When ID is not\n\t// nil, it will be used for WHERE condition. If ID is nil and Name is non-nil,\n\t// Name will be used for WHERE condition. When both ID and Name are nil,\n\t// no WHERE clause will be used\n\tDomainFilter struct {\n\t\tID            *serialization.UUID\n\t\tName          *string\n\t\tGreaterThanID *serialization.UUID\n\t\tPageSize      *int\n\t}\n\n\t// DomainMetadataRow represents a row in domain_metadata table\n\tDomainMetadataRow struct {\n\t\tNotificationVersion int64\n\t}\n\n\t// ShardsRow represents a row in shards table\n\tShardsRow struct {\n\t\tShardID      int64\n\t\tRangeID      int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// ShardsFilter contains the column names within shards table that\n\t// can be used to filter results through a WHERE clause\n\tShardsFilter struct {\n\t\tShardID int64\n\t}\n\n\t// TransferTasksRow represents a row in transfer_tasks table\n\tTransferTasksRow struct {\n\t\tShardID      int\n\t\tTaskID       int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// CrossClusterTasksRow represents a row in cross_cluster_tasks table\n\tCrossClusterTasksRow struct {\n\t\tTargetCluster string\n\t\tShardID       int\n\t\tTaskID        int64\n\t\tData          []byte\n\t\tDataEncoding  string\n\t}\n\n\t// TransferTasksFilter contains the column names within transfer_tasks table that\n\t// can be used to filter results through a WHERE clause\n\tTransferTasksFilter struct {\n\t\tShardID            int\n\t\tTaskID             int64\n\t\tInclusiveMinTaskID int64\n\t\tExclusiveMaxTaskID int64\n\t\tPageSize           int\n\t}\n\n\t// CrossClusterTasksFilter contains the column names within cross_cluster_tasks table that\n\t// can be used to filter results through a WHERE clause\n\tCrossClusterTasksFilter struct {\n\t\tTargetCluster string\n\t\tShardID       int\n\t\tTaskID        int64\n\t\tMinTaskID     int64\n\t\tMaxTaskID     int64\n\t\tPageSize      int\n\t}\n\n\t// ExecutionsRow represents a row in executions table\n\tExecutionsRow struct {\n\t\tShardID                  int\n\t\tDomainID                 serialization.UUID\n\t\tWorkflowID               string\n\t\tRunID                    serialization.UUID\n\t\tNextEventID              int64\n\t\tLastWriteVersion         int64\n\t\tData                     []byte\n\t\tDataEncoding             string\n\t\tVersionHistories         []byte\n\t\tVersionHistoriesEncoding string\n\t}\n\n\t// ExecutionsFilter contains the column names within executions table that\n\t// can be used to filter results through a WHERE clause\n\t// To get single row, it requires ShardID, DomainID, WorkflowID, RunID\n\t// To get a list of rows, it requires ShardID, Size.\n\t// The WorkflowID and RunID are optional for listing rows. They work as the start boundary for pagination.\n\tExecutionsFilter struct {\n\t\tShardID    int\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t\tSize       int\n\t}\n\n\t// CurrentExecutionsRow represents a row in current_executions table\n\tCurrentExecutionsRow struct {\n\t\tShardID          int64\n\t\tDomainID         serialization.UUID\n\t\tWorkflowID       string\n\t\tRunID            serialization.UUID\n\t\tCreateRequestID  string\n\t\tState            int\n\t\tCloseStatus      int\n\t\tLastWriteVersion int64\n\t\tStartVersion     int64\n\t}\n\n\t// CurrentExecutionsFilter contains the column names within current_executions table that\n\t// can be used to filter results through a WHERE clause\n\tCurrentExecutionsFilter struct {\n\t\tShardID    int64\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t}\n\n\t// BufferedEventsRow represents a row in buffered_events table\n\tBufferedEventsRow struct {\n\t\tShardID      int\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// BufferedEventsFilter contains the column names within buffered_events table that\n\t// can be used to filter results through a WHERE clause\n\tBufferedEventsFilter struct {\n\t\tShardID    int\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t}\n\n\t// TasksRow represents a row in tasks table\n\tTasksRow struct {\n\t\tShardID      int // this is DBShardID, not historyShardID (TODO: maybe rename it for clarification)\n\t\tDomainID     serialization.UUID\n\t\tTaskType     int64\n\t\tTaskID       int64\n\t\tTaskListName string\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// TaskKeyRow represents a result row giving task keys\n\tTaskKeyRow struct {\n\t\tDomainID     serialization.UUID\n\t\tTaskListName string\n\t\tTaskType     int64\n\t\tTaskID       int64\n\t}\n\n\t// TasksRowWithTTL represents a row in tasks table with a ttl\n\tTasksRowWithTTL struct {\n\t\tTasksRow TasksRow\n\t\t// TTL is optional because InsertIntoTasksWithTTL operates over a slice of TasksRowWithTTL.\n\t\t// Some items in the slice may have a TTL while others do not. It is the responsibility\n\t\t// of the plugin implementation to handle items with TTL set and items with TTL not set.\n\t\tTTL *time.Duration\n\t}\n\n\t// TasksFilter contains the column names within tasks table that\n\t// can be used to filter results through a WHERE clause\n\tTasksFilter struct {\n\t\tShardID              int // this is DBShardID, not historyShardID (TODO: maybe rename it for clarification)\n\t\tDomainID             serialization.UUID\n\t\tTaskListName         string\n\t\tTaskType             int64\n\t\tTaskID               *int64\n\t\tMinTaskID            *int64\n\t\tMaxTaskID            *int64\n\t\tTaskIDLessThanEquals *int64\n\t\tLimit                *int\n\t\tPageSize             *int\n\t}\n\n\t// OrphanTasksFilter contains the parameters controlling orphan deletion\n\tOrphanTasksFilter struct {\n\t\tLimit *int\n\t}\n\n\t// TaskListsRow represents a row in task_lists table\n\tTaskListsRow struct {\n\t\tShardID      int // this is DBShardID, not historyShardID (TODO: maybe rename it for clarification)\n\t\tDomainID     serialization.UUID\n\t\tName         string\n\t\tTaskType     int64\n\t\tRangeID      int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// TaskListsRowWithTTL represents a row in task_lists table with a ttl\n\tTaskListsRowWithTTL struct {\n\t\tTaskListsRow TaskListsRow\n\t\tTTL          time.Duration\n\t}\n\n\t// TaskListsFilter contains the column names within task_lists table that\n\t// can be used to filter results through a WHERE clause\n\tTaskListsFilter struct {\n\t\tShardID             int // this is DBShardID, not historyShardID (TODO: maybe rename it for clarification)\n\t\tDomainID            *serialization.UUID\n\t\tName                *string\n\t\tTaskType            *int64\n\t\tDomainIDGreaterThan *serialization.UUID\n\t\tNameGreaterThan     *string\n\t\tTaskTypeGreaterThan *int64\n\t\tRangeID             *int64\n\t\tPageSize            *int\n\t}\n\n\t// ReplicationTasksRow represents a row in replication_tasks table\n\tReplicationTasksRow struct {\n\t\tShardID      int\n\t\tTaskID       int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// ReplicationTaskDLQRow represents a row in replication_tasks_dlq table\n\tReplicationTaskDLQRow struct {\n\t\tSourceClusterName string\n\t\tShardID           int\n\t\tTaskID            int64\n\t\tData              []byte\n\t\tDataEncoding      string\n\t}\n\n\t// ReplicationTasksFilter contains the column names within replication_tasks table that\n\t// can be used to filter results through a WHERE clause\n\tReplicationTasksFilter struct {\n\t\tShardID            int\n\t\tTaskID             int64\n\t\tInclusiveMinTaskID int64\n\t\tExclusiveMaxTaskID int64\n\t\tPageSize           int\n\t}\n\n\t// ReplicationTasksDLQFilter contains the column names within replication_tasks_dlq table that\n\t// can be used to filter results through a WHERE clause\n\tReplicationTasksDLQFilter struct {\n\t\tReplicationTasksFilter\n\t\tSourceClusterName string\n\t}\n\n\t// ReplicationTaskDLQFilter contains the column names within replication_tasks_dlq table that\n\t// can be used to filter results through a WHERE clause\n\tReplicationTaskDLQFilter struct {\n\t\tSourceClusterName string\n\t\tShardID           int\n\t}\n\n\t// TimerTasksRow represents a row in timer_tasks table\n\tTimerTasksRow struct {\n\t\tShardID             int\n\t\tVisibilityTimestamp time.Time\n\t\tTaskID              int64\n\t\tData                []byte\n\t\tDataEncoding        string\n\t}\n\n\t// TimerTasksFilter contains the column names within timer_tasks table that\n\t// can be used to filter results through a WHERE clause\n\tTimerTasksFilter struct {\n\t\tShardID                int\n\t\tTaskID                 int64\n\t\tVisibilityTimestamp    time.Time\n\t\tMinVisibilityTimestamp time.Time\n\t\tMaxVisibilityTimestamp time.Time\n\t\tPageSize               int\n\t}\n\n\t// EventsRow represents a row in events table\n\tEventsRow struct {\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tFirstEventID int64\n\t\tBatchVersion int64\n\t\tRangeID      int64\n\t\tTxID         int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// EventsFilter contains the column names within events table that\n\t// can be used to filter results through a WHERE clause\n\tEventsFilter struct {\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tFirstEventID *int64\n\t\tNextEventID  *int64\n\t\tPageSize     *int\n\t}\n\n\t// HistoryNodeRow represents a row in history_node table\n\tHistoryNodeRow struct {\n\t\tShardID  int\n\t\tTreeID   serialization.UUID\n\t\tBranchID serialization.UUID\n\t\tNodeID   int64\n\t\t// use pointer so that it's easier to multiple by -1\n\t\tTxnID        *int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// HistoryNodeFilter contains the column names within history_node table that\n\t// can be used to filter results through a WHERE clause\n\tHistoryNodeFilter struct {\n\t\tShardID  int\n\t\tTreeID   serialization.UUID\n\t\tBranchID serialization.UUID\n\t\t// Inclusive\n\t\tMinNodeID *int64\n\t\t// Exclusive\n\t\tMaxNodeID *int64\n\t\tPageSize  int\n\t}\n\n\t// HistoryTreeRow represents a row in history_tree table\n\tHistoryTreeRow struct {\n\t\tShardID      int\n\t\tTreeID       serialization.UUID\n\t\tBranchID     serialization.UUID\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// HistoryTreeFilter contains the column names within history_tree table that\n\t// can be used to filter results through a WHERE clause\n\tHistoryTreeFilter struct {\n\t\tShardID  int\n\t\tTreeID   serialization.UUID\n\t\tBranchID *serialization.UUID\n\t\tPageSize *int\n\t}\n\n\t// ActivityInfoMapsRow represents a row in activity_info_maps table\n\tActivityInfoMapsRow struct {\n\t\tShardID                  int64\n\t\tDomainID                 serialization.UUID\n\t\tWorkflowID               string\n\t\tRunID                    serialization.UUID\n\t\tScheduleID               int64\n\t\tData                     []byte\n\t\tDataEncoding             string\n\t\tLastHeartbeatDetails     []byte\n\t\tLastHeartbeatUpdatedTime time.Time\n\t}\n\n\t// ActivityInfoMapsFilter contains the column names within activity_info_maps table that\n\t// can be used to filter results through a WHERE clause\n\tActivityInfoMapsFilter struct {\n\t\tShardID     int64\n\t\tDomainID    serialization.UUID\n\t\tWorkflowID  string\n\t\tRunID       serialization.UUID\n\t\tScheduleIDs []int64\n\t}\n\n\t// TimerInfoMapsRow represents a row in timer_info_maps table\n\tTimerInfoMapsRow struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tTimerID      string\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// TimerInfoMapsFilter contains the column names within timer_info_maps table that\n\t// can be used to filter results through a WHERE clause\n\tTimerInfoMapsFilter struct {\n\t\tShardID    int64\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t\tTimerIDs   []string\n\t}\n\n\t// ChildExecutionInfoMapsRow represents a row in child_execution_info_maps table\n\tChildExecutionInfoMapsRow struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedID  int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// ChildExecutionInfoMapsFilter contains the column names within child_execution_info_maps table that\n\t// can be used to filter results through a WHERE clause\n\tChildExecutionInfoMapsFilter struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedIDs []int64\n\t}\n\n\t// RequestCancelInfoMapsRow represents a row in request_cancel_info_maps table\n\tRequestCancelInfoMapsRow struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedID  int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// RequestCancelInfoMapsFilter contains the column names within request_cancel_info_maps table that\n\t// can be used to filter results through a WHERE clause\n\tRequestCancelInfoMapsFilter struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedIDs []int64\n\t}\n\n\t// SignalInfoMapsRow represents a row in signal_info_maps table\n\tSignalInfoMapsRow struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedID  int64\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// SignalInfoMapsFilter contains the column names within signal_info_maps table that\n\t// can be used to filter results through a WHERE clause\n\tSignalInfoMapsFilter struct {\n\t\tShardID      int64\n\t\tDomainID     serialization.UUID\n\t\tWorkflowID   string\n\t\tRunID        serialization.UUID\n\t\tInitiatedIDs []int64\n\t}\n\n\t// SignalsRequestedSetsRow represents a row in signals_requested_sets table\n\tSignalsRequestedSetsRow struct {\n\t\tShardID    int64\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t\tSignalID   string\n\t}\n\n\t// SignalsRequestedSetsFilter contains the column names within signals_requested_sets table that\n\t// can be used to filter results through a WHERE clause\n\tSignalsRequestedSetsFilter struct {\n\t\tShardID    int64\n\t\tDomainID   serialization.UUID\n\t\tWorkflowID string\n\t\tRunID      serialization.UUID\n\t\tSignalIDs  []string\n\t}\n\n\t// VisibilityRow represents a row in executions_visibility table\n\tVisibilityRow struct {\n\t\tDomainID               string\n\t\tRunID                  string\n\t\tWorkflowTypeName       string\n\t\tWorkflowID             string\n\t\tStartTime              time.Time\n\t\tExecutionTime          time.Time\n\t\tCloseStatus            *int32\n\t\tCloseTime              *time.Time\n\t\tHistoryLength          *int64\n\t\tMemo                   []byte\n\t\tEncoding               string\n\t\tIsCron                 bool\n\t\tCronSchedule           string\n\t\tNumClusters            int16\n\t\tUpdateTime             time.Time\n\t\tShardID                int16\n\t\tExecutionStatus        int32\n\t\tScheduledExecutionTime time.Time\n\t}\n\n\t// VisibilityFilter contains the column names within executions_visibility table that\n\t// can be used to filter results through a WHERE clause\n\tVisibilityFilter struct {\n\t\tDomainID         string\n\t\tClosed           bool\n\t\tRunID            *string\n\t\tWorkflowID       *string\n\t\tWorkflowTypeName *string\n\t\tCloseStatus      *int32\n\t\tMinStartTime     *time.Time\n\t\tMaxStartTime     *time.Time\n\t\tPageSize         *int\n\t}\n\n\t// QueueRow represents a row in queue table\n\tQueueRow struct {\n\t\tQueueType      persistence.QueueType\n\t\tMessageID      int64\n\t\tMessagePayload []byte\n\t}\n\n\t// QueueMetadataRow represents a row in queue_metadata table\n\tQueueMetadataRow struct {\n\t\tQueueType persistence.QueueType\n\t\tData      []byte\n\t}\n\n\t// ClusterConfigRow represents a row in cluster_config table\n\tClusterConfigRow struct {\n\t\tRowType      int\n\t\tVersion      int64\n\t\tTimestamp    time.Time\n\t\tData         []byte\n\t\tDataEncoding string\n\t}\n\n\t// DomainAuditLogRow represents a row in domain_audit_log table\n\tDomainAuditLogRow struct {\n\t\tDomainID            string\n\t\tEventID             string\n\t\tStateBefore         []byte\n\t\tStateBeforeEncoding constants.EncodingType\n\t\tStateAfter          []byte\n\t\tStateAfterEncoding  constants.EncodingType\n\t\tOperationType       persistence.DomainAuditOperationType\n\t\tCreatedTime         time.Time\n\t\tLastUpdatedTime     time.Time\n\t\tIdentity            string\n\t\tIdentityType        string\n\t\tComment             string\n\t}\n\n\t// DomainAuditLogFilter contains the filter criteria for querying domain audit logs\n\tDomainAuditLogFilter struct {\n\t\tDomainID       string\n\t\tOperationType  persistence.DomainAuditOperationType\n\t\tMinCreatedTime *time.Time\n\t\tPageSize       int\n\t\t// PageMaxCreatedTime and PageMinEventID are used to paginate Select queries\n\t\tPageMaxCreatedTime *time.Time\n\t\tPageMinEventID     *string\n\t}\n\n\t// tableCRUD defines the API for interacting with the database tables\n\ttableCRUD interface {\n\t\tInsertIntoDomain(ctx context.Context, rows *DomainRow) (sql.Result, error)\n\t\tUpdateDomain(ctx context.Context, row *DomainRow) (sql.Result, error)\n\t\t// SelectFromDomain returns domains that match filter criteria. Either ID or\n\t\t// Name can be specified to filter results. If both are not specified, all rows\n\t\t// will be returned\n\t\tSelectFromDomain(ctx context.Context, filter *DomainFilter) ([]DomainRow, error)\n\t\t// DeleteDomain deletes a single row. One of ID or Name MUST be specified\n\t\tDeleteFromDomain(ctx context.Context, filter *DomainFilter) (sql.Result, error)\n\n\t\tLockDomainMetadata(ctx context.Context) error\n\t\tUpdateDomainMetadata(ctx context.Context, row *DomainMetadataRow) (sql.Result, error)\n\t\tSelectFromDomainMetadata(ctx context.Context) (*DomainMetadataRow, error)\n\n\t\tInsertIntoShards(ctx context.Context, rows *ShardsRow) (sql.Result, error)\n\t\tUpdateShards(ctx context.Context, row *ShardsRow) (sql.Result, error)\n\t\tSelectFromShards(ctx context.Context, filter *ShardsFilter) (*ShardsRow, error)\n\t\tReadLockShards(ctx context.Context, filter *ShardsFilter) (int, error)\n\t\tWriteLockShards(ctx context.Context, filter *ShardsFilter) (int, error)\n\n\t\tInsertIntoTasks(ctx context.Context, rows []TasksRow) (sql.Result, error)\n\t\tInsertIntoTasksWithTTL(ctx context.Context, rows []TasksRowWithTTL) (sql.Result, error)\n\t\t// SelectFromTasks retrieves one or more rows from the tasks table\n\t\t// Required filter params - {domainID, tasklistName, taskType, minTaskID, maxTaskID, pageSize}\n\t\tSelectFromTasks(ctx context.Context, filter *TasksFilter) ([]TasksRow, error)\n\t\t// DeleteFromTasks deletes a row from tasks table\n\t\t// Required filter params:\n\t\t//  to delete single row\n\t\t//     - {domainID, tasklistName, taskType, taskID}\n\t\t//  to delete multiple rows\n\t\t//    - {domainID, tasklistName, taskType, taskIDLessThanEquals, limit }\n\t\t//    - this will delete up to limit number of tasks less than or equal to the given task id\n\t\tDeleteFromTasks(ctx context.Context, filter *TasksFilter) (sql.Result, error)\n\t\tGetTasksCount(ctx context.Context, filter *TasksFilter) (int64, error)\n\t\tGetOrphanTasks(ctx context.Context, filter *OrphanTasksFilter) ([]TaskKeyRow, error)\n\n\t\tInsertIntoTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error)\n\t\tInsertIntoTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error)\n\t\tUpdateTaskLists(ctx context.Context, row *TaskListsRow) (sql.Result, error)\n\t\tUpdateTaskListsWithTTL(ctx context.Context, row *TaskListsRowWithTTL) (sql.Result, error)\n\t\t// SelectFromTaskLists returns one or more rows from task_lists table\n\t\t// Required Filter params:\n\t\t//  to read a single row: {shardID, domainID, name, taskType}\n\t\t//  to range read multiple rows: {shardID, domainIDGreaterThan, nameGreaterThan, taskTypeGreaterThan, pageSize}\n\t\tSelectFromTaskLists(ctx context.Context, filter *TaskListsFilter) ([]TaskListsRow, error)\n\t\tDeleteFromTaskLists(ctx context.Context, filter *TaskListsFilter) (sql.Result, error)\n\t\tLockTaskLists(ctx context.Context, filter *TaskListsFilter) (int64, error)\n\n\t\t// eventsV2\n\t\tInsertIntoHistoryNode(ctx context.Context, row *HistoryNodeRow) (sql.Result, error)\n\t\tSelectFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) ([]HistoryNodeRow, error)\n\t\tDeleteFromHistoryNode(ctx context.Context, filter *HistoryNodeFilter) (sql.Result, error)\n\t\tInsertIntoHistoryTree(ctx context.Context, row *HistoryTreeRow) (sql.Result, error)\n\t\tSelectFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error)\n\t\tDeleteFromHistoryTree(ctx context.Context, filter *HistoryTreeFilter) (sql.Result, error)\n\t\tGetAllHistoryTreeBranches(ctx context.Context, filter *HistoryTreeFilter) ([]HistoryTreeRow, error)\n\n\t\tInsertIntoExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error)\n\t\tUpdateExecutions(ctx context.Context, row *ExecutionsRow) (sql.Result, error)\n\t\tSelectFromExecutions(ctx context.Context, filter *ExecutionsFilter) ([]ExecutionsRow, error)\n\t\tDeleteFromExecutions(ctx context.Context, filter *ExecutionsFilter) (sql.Result, error)\n\t\tReadLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error)\n\t\tWriteLockExecutions(ctx context.Context, filter *ExecutionsFilter) (int, error)\n\n\t\tLockCurrentExecutionsJoinExecutions(ctx context.Context, filter *CurrentExecutionsFilter) ([]CurrentExecutionsRow, error)\n\n\t\tInsertIntoCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error)\n\t\tUpdateCurrentExecutions(ctx context.Context, row *CurrentExecutionsRow) (sql.Result, error)\n\t\t// SelectFromCurrentExecutions returns one or more rows from current_executions table\n\t\t// Required params - {shardID, domainID, workflowID}\n\t\tSelectFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error)\n\t\t// DeleteFromCurrentExecutions deletes a single row that matches the filter criteria\n\t\t// If a row exist, that row will be deleted and this method will return success\n\t\t// If there is no row matching the filter criteria, this method will still return success\n\t\t// Callers can check the output of Result.RowsAffected() to see if a row was deleted or not\n\t\t// Required params - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (sql.Result, error)\n\t\tLockCurrentExecutions(ctx context.Context, filter *CurrentExecutionsFilter) (*CurrentExecutionsRow, error)\n\n\t\tInsertIntoTransferTasks(ctx context.Context, rows []TransferTasksRow) (sql.Result, error)\n\t\t// SelectFromTransferTasks returns rows that match filter criteria from transfer_tasks table.\n\t\t// Required filter params - {shardID, minTaskID, maxTaskID}\n\t\tSelectFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) ([]TransferTasksRow, error)\n\t\t// DeleteFromTransferTasks deletes one or more rows from transfer_tasks table.\n\t\t// Required filter params - {shardID, taskID}\n\t\tDeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error)\n\t\t// RangeDeleteFromTransferTasks deletes one or more rows from transfer_tasks table.\n\t\t// Required filter params - {shardID, minTaskID, maxTaskID}\n\t\tRangeDeleteFromTransferTasks(ctx context.Context, filter *TransferTasksFilter) (sql.Result, error)\n\n\t\t// TODO: add cross-cluster tasks methods\n\t\t// InsertIntoCrossClusterTasks adds a new row to the cross_cluster_tasks table\n\t\tInsertIntoCrossClusterTasks(ctx context.Context, rows []CrossClusterTasksRow) (sql.Result, error)\n\t\t// SelectFromCrossClusterTasks returns rows that match filter criteria from cross_cluster_tasks table.\n\t\t// Required filter params - {shardID, minTaskID, maxTaskID}\n\t\tSelectFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) ([]CrossClusterTasksRow, error)\n\t\t// DeleteFromCrossClusterTasks deletes one or more rows from cross_cluster_tasks table.\n\t\t// Required filter params - {shardID, taskID}\n\t\tDeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error)\n\t\t// RangeDeleteFromCrossClusterTasks deletes one or more rows from cross_cluster_tasks table.\n\t\t// Required filter params - {shardID, minTaskID, maxTaskID}\n\t\tRangeDeleteFromCrossClusterTasks(ctx context.Context, filter *CrossClusterTasksFilter) (sql.Result, error)\n\n\t\tInsertIntoTimerTasks(ctx context.Context, rows []TimerTasksRow) (sql.Result, error)\n\t\t// SelectFromTimerTasks returns one or more rows from timer_tasks table\n\t\t// Required filter Params - {shardID, taskID, minVisibilityTimestamp, maxVisibilityTimestamp, pageSize}\n\t\tSelectFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) ([]TimerTasksRow, error)\n\t\t// DeleteFromTimerTasks deletes one or more rows from timer_tasks table\n\t\t// Required filter Params: {shardID, visibilityTimestamp, taskID}\n\t\tDeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error)\n\t\t// RangeDeleteFromTimerTasks deletes one or more rows from timer_tasks table\n\t\t// Required filter Params: {shardID, minVisibilityTimestamp, maxVisibilityTimestamp}\n\t\tRangeDeleteFromTimerTasks(ctx context.Context, filter *TimerTasksFilter) (sql.Result, error)\n\n\t\tInsertIntoBufferedEvents(ctx context.Context, rows []BufferedEventsRow) (sql.Result, error)\n\t\tSelectFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) ([]BufferedEventsRow, error)\n\t\tDeleteFromBufferedEvents(ctx context.Context, filter *BufferedEventsFilter) (sql.Result, error)\n\n\t\tInsertIntoReplicationTasks(ctx context.Context, rows []ReplicationTasksRow) (sql.Result, error)\n\t\t// SelectFromReplicationTasks returns one or more rows from replication_tasks table\n\t\t// Required filter params - {shardID, minTaskID, maxTaskID, pageSize}\n\t\tSelectFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) ([]ReplicationTasksRow, error)\n\t\t// DeleteFromReplicationTasks deletes a row from replication_tasks table\n\t\t// Required filter params - {shardID, taskID}\n\t\tDeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error)\n\t\t// DeleteFromReplicationTasks deletes multi rows from replication_tasks table\n\t\t// Required filter params - {shardID, inclusiveEndTaskID}\n\t\tRangeDeleteFromReplicationTasks(ctx context.Context, filter *ReplicationTasksFilter) (sql.Result, error)\n\t\t// InsertIntoReplicationTasksDLQ puts the replication task into DLQ\n\t\tInsertIntoReplicationTasksDLQ(ctx context.Context, row *ReplicationTaskDLQRow) (sql.Result, error)\n\t\t// SelectFromReplicationTasksDLQ returns one or more rows from replication_tasks_dlq table\n\t\t// Required filter params - {sourceClusterName, shardID, minTaskID, pageSize}\n\t\tSelectFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) ([]ReplicationTasksRow, error)\n\t\t// SelectFromReplicationDLQ returns one row from replication_tasks_dlq table\n\t\t// Required filter params - {sourceClusterName}\n\t\tSelectFromReplicationDLQ(ctx context.Context, filter *ReplicationTaskDLQFilter) (int64, error)\n\t\t// DeleteMessageFromReplicationTasksDLQ deletes one row from replication_tasks_dlq table\n\t\t// Required filter params - {sourceClusterName, shardID, taskID}\n\t\tDeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error)\n\t\t// RangeDeleteMessageFromReplicationTasksDLQ deletes one or more rows from replication_tasks_dlq table\n\t\t// Required filter params - {sourceClusterName, shardID, taskID, inclusiveTaskID}\n\t\tRangeDeleteMessageFromReplicationTasksDLQ(ctx context.Context, filter *ReplicationTasksDLQFilter) (sql.Result, error)\n\n\t\tReplaceIntoActivityInfoMaps(ctx context.Context, rows []ActivityInfoMapsRow) (sql.Result, error)\n\t\t// SelectFromActivityInfoMaps returns one or more rows from activity_info_maps\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) ([]ActivityInfoMapsRow, error)\n\t\t// DeleteFromActivityInfoMaps deletes a row from activity_info_maps table\n\t\t// Required filter params\n\t\t// - one or multiple rows delete - {shardID, domainID, workflowID, runID, scheduleIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromActivityInfoMaps(ctx context.Context, filter *ActivityInfoMapsFilter) (sql.Result, error)\n\n\t\tReplaceIntoTimerInfoMaps(ctx context.Context, rows []TimerInfoMapsRow) (sql.Result, error)\n\t\t// SelectFromTimerInfoMaps returns one or more rows form timer_info_maps table\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) ([]TimerInfoMapsRow, error)\n\t\t// DeleteFromTimerInfoMaps deletes one or more rows from timer_info_maps\n\t\t// Required filter params\n\t\t// - one or multiple rows delete- {shardID, domainID, workflowID, runID, timerIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromTimerInfoMaps(ctx context.Context, filter *TimerInfoMapsFilter) (sql.Result, error)\n\n\t\tReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []ChildExecutionInfoMapsRow) (sql.Result, error)\n\t\t// SelectFromChildExecutionInfoMaps returns one or more rows form child_execution_info_maps table\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) ([]ChildExecutionInfoMapsRow, error)\n\t\t// DeleteFromChildExecutionInfoMaps deletes one or more rows from child_execution_info_maps\n\t\t// Required filter params\n\t\t// - onne or multiple rows delete - {shardID, domainID, workflowID, runID, initiatedIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromChildExecutionInfoMaps(ctx context.Context, filter *ChildExecutionInfoMapsFilter) (sql.Result, error)\n\n\t\tReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []RequestCancelInfoMapsRow) (sql.Result, error)\n\t\t// SelectFromRequestCancelInfoMaps returns one or more rows form request_cancel_info_maps table\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) ([]RequestCancelInfoMapsRow, error)\n\t\t// DeleteFromRequestCancelInfoMaps deletes one or more rows from request_cancel_info_maps\n\t\t// Required filter params\n\t\t// - one or multiple rows delete - {shardID, domainID, workflowID, runID, initiatedIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromRequestCancelInfoMaps(ctx context.Context, filter *RequestCancelInfoMapsFilter) (sql.Result, error)\n\n\t\tReplaceIntoSignalInfoMaps(ctx context.Context, rows []SignalInfoMapsRow) (sql.Result, error)\n\t\t// SelectFromSignalInfoMaps returns one or more rows form signal_info_maps table\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) ([]SignalInfoMapsRow, error)\n\t\t// DeleteFromSignalInfoMaps deletes one or more rows from signal_info_maps table\n\t\t// Required filter params\n\t\t// - one or multiple rows delete - {shardID, domainID, workflowID, runID, initiatedIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromSignalInfoMaps(ctx context.Context, filter *SignalInfoMapsFilter) (sql.Result, error)\n\n\t\tInsertIntoSignalsRequestedSets(ctx context.Context, rows []SignalsRequestedSetsRow) (sql.Result, error)\n\t\t// SelectFromSignalInfoMaps returns one or more rows form singals_requested_sets table\n\t\t// Required filter params - {shardID, domainID, workflowID, runID}\n\t\tSelectFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) ([]SignalsRequestedSetsRow, error)\n\t\t// DeleteFromSignalsRequestedSets deletes one or more rows from signals_requested_sets\n\t\t// Required filter params\n\t\t// - one or multiple rows delete - {shardID, domainID, workflowID, runID, signalIDs}\n\t\t// - range delete - {shardID, domainID, workflowID, runID}\n\t\tDeleteFromSignalsRequestedSets(ctx context.Context, filter *SignalsRequestedSetsFilter) (sql.Result, error)\n\n\t\t// InsertIntoVisibility inserts a row into visibility table. If a row already exist,\n\t\t// no changes will be made by this API\n\t\tInsertIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error)\n\t\t// ReplaceIntoVisibility deletes old row (if it exist) and inserts new row into visibility table\n\t\tReplaceIntoVisibility(ctx context.Context, row *VisibilityRow) (sql.Result, error)\n\t\t// SelectFromVisibility returns one or more rows from visibility table\n\t\t// Required filter params:\n\t\t// - getClosedWorkflowExecution - retrieves single row - {domainID, runID, closed=true}\n\t\t// - All other queries retrieve multiple rows (range):\n\t\t//   - MUST specify following required params:\n\t\t//     - domainID, minStartTime, maxStartTime, runID and pageSize where some or all of these may come from previous page token\n\t\t//   - OPTIONALLY specify one of following params\n\t\t//     - workflowID, workflowTypeName, closeStatus (along with closed=true)\n\t\tSelectFromVisibility(ctx context.Context, filter *VisibilityFilter) ([]VisibilityRow, error)\n\t\tDeleteFromVisibility(ctx context.Context, filter *VisibilityFilter) (sql.Result, error)\n\n\t\tInsertIntoQueue(ctx context.Context, row *QueueRow) (sql.Result, error)\n\t\tGetLastEnqueuedMessageIDForUpdate(ctx context.Context, queueType persistence.QueueType) (int64, error)\n\t\tGetMessagesFromQueue(ctx context.Context, queueType persistence.QueueType, lastMessageID int64, maxRows int) ([]QueueRow, error)\n\t\tGetMessagesBetween(ctx context.Context, queueType persistence.QueueType, firstMessageID int64, lastMessageID int64, maxRows int) ([]QueueRow, error)\n\t\tDeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error)\n\t\tRangeDeleteMessages(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, inclusiveEndMessageID int64) (sql.Result, error)\n\t\tDeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error)\n\t\tInsertAckLevel(ctx context.Context, queueType persistence.QueueType, messageID int64, clusterName string) error\n\t\tUpdateAckLevels(ctx context.Context, queueType persistence.QueueType, clusterAckLevels map[string]int64) error\n\t\tGetAckLevels(ctx context.Context, queueType persistence.QueueType, forUpdate bool) (map[string]int64, error)\n\t\tGetQueueSize(ctx context.Context, queueType persistence.QueueType) (int64, error)\n\n\t\t// InsertConfig insert a config entry with version. Return nosqlplugin.NewConditionFailure if the same version of the row_type is existing\n\t\tInsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error\n\t\t// SelectLatestConfig returns the config entry of the row_type with the largest(latest) version value\n\t\tSelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error)\n\n\t\t// InsertDomainAuditLog inserts a new audit log entry for a domain operation. Returns error if there is any failure\n\t\tInsertIntoDomainAuditLog(ctx context.Context, row *DomainAuditLogRow) (sql.Result, error)\n\t\t// SelectFromDomainAuditLogs returns audit log entries for a domain. Returns paginated results ordered by created_time DESC, event_id ASC\n\t\tSelectFromDomainAuditLogs(ctx context.Context, filter *DomainAuditLogFilter) ([]*DomainAuditLogRow, error)\n\n\t\t// The follow provide information about the underlying sql crud implementation\n\t\tSupportsTTL() bool\n\t\tMaxAllowedTTL() (*time.Duration, error)\n\t\tSupportsAsyncTransaction() bool\n\t}\n\n\t// adminCRUD defines admin operations for CLI and test suites\n\tadminCRUD interface {\n\t\tCreateSchemaVersionTables() error\n\t\tReadSchemaVersion(database string) (string, error)\n\t\tUpdateSchemaVersion(database string, newVersion string, minCompatibleVersion string) error\n\t\tWriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error\n\t\tListTables(database string) ([]string, error)\n\t\tDropTable(table string) error\n\t\tDropAllTables(database string) error\n\t\tCreateDatabase(database string) error\n\t\tDropDatabase(database string) error\n\t\t// ExecSchemaOperationQuery allows passing in any query, but it must be schema operation (DDL)\n\t\tExecSchemaOperationQuery(ctx context.Context, stmt string, args ...interface{}) error\n\t}\n\n\t// Tx defines the API for a SQL transaction\n\tTx interface {\n\t\ttableCRUD\n\t\tErrorChecker\n\n\t\tCommit() error\n\t\tRollback() error\n\t}\n\n\t// DB defines the API for regular SQL operations of a Cadence server\n\tDB interface {\n\t\ttableCRUD\n\t\tErrorChecker\n\n\t\tGetTotalNumDBShards() int\n\t\tBeginTx(ctx context.Context, dbShardID int) (Tx, error)\n\t\tPluginName() string\n\t\tClose() error\n\t}\n\n\t// AdminDB defines the API for admin SQL operations for CLI and testing suites\n\tAdminDB interface {\n\t\tadminCRUD\n\t\tPluginName() string\n\t\tClose() error\n\t}\n\n\tErrorChecker interface {\n\t\tIsDupEntryError(err error) bool\n\t\tIsNotFoundError(err error) bool\n\t\tIsTimeoutError(err error) bool\n\t\tIsThrottlingError(err error) bool\n\t}\n)\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/admin.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\treadSchemaVersionQuery = `SELECT curr_version from schema_version where db_name=?`\n\n\twriteSchemaVersionQuery = `REPLACE into schema_version(db_name, creation_time, curr_version, min_compatible_version) VALUES (?,?,?,?)`\n\n\twriteSchemaUpdateHistoryQuery = `INSERT into schema_update_history(year, month, update_time, old_version, new_version, manifest_md5, description) VALUES(?,?,?,?,?,?,?)`\n\n\tcreateSchemaVersionTableQuery = `CREATE TABLE schema_version(db_name VARCHAR(255) not null PRIMARY KEY, ` +\n\t\t`creation_time DATETIME(6), ` +\n\t\t`curr_version VARCHAR(64), ` +\n\t\t`min_compatible_version VARCHAR(64));`\n\n\tcreateSchemaUpdateHistoryTableQuery = `CREATE TABLE schema_update_history(` +\n\t\t`year int not null, ` +\n\t\t`month int not null, ` +\n\t\t`update_time DATETIME(6) not null, ` +\n\t\t`description VARCHAR(255), ` +\n\t\t`manifest_md5 VARCHAR(64), ` +\n\t\t`new_version VARCHAR(64), ` +\n\t\t`old_version VARCHAR(64), ` +\n\t\t`PRIMARY KEY (year, month, update_time));`\n\n\t// NOTE we have to use %v because somehow mysql doesn't work with ? here\n\tcreateDatabaseQuery = \"CREATE database %v CHARACTER SET UTF8\"\n\n\tdropDatabaseQuery = \"Drop database %v\"\n\n\tlistTablesQuery = \"SHOW TABLES FROM %v\"\n\n\tdropTableQuery = \"DROP TABLE %v\"\n)\n\n// CreateSchemaVersionTables sets up the schema version tables\nfunc (mdb *DB) CreateSchemaVersionTables() error {\n\tif err := mdb.ExecSchemaOperationQuery(context.Background(), createSchemaVersionTableQuery); err != nil {\n\t\treturn err\n\t}\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), createSchemaUpdateHistoryTableQuery)\n}\n\n// ReadSchemaVersion returns the current schema version for the keyspace\nfunc (mdb *DB) ReadSchemaVersion(database string) (string, error) {\n\tvar version string\n\terr := mdb.driver.GetForSchemaQuery(sqlplugin.DbShardUndefined, &version, readSchemaVersionQuery, database)\n\treturn version, err\n}\n\n// UpdateSchemaVersion updates the schema version for the keyspace\nfunc (mdb *DB) UpdateSchemaVersion(database string, newVersion string, minCompatibleVersion string) error {\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), writeSchemaVersionQuery, database, time.Now(), newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog adds an entry to the schema update history table\nfunc (mdb *DB) WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error {\n\tnow := time.Now().UTC()\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), writeSchemaUpdateHistoryQuery, now.Year(), int(now.Month()), now, oldVersion, newVersion, manifestMD5, desc)\n}\n\n// ExecSchemaOperationQuery executes a sql statement for schema ONLY. DO NOT use it in other cases, otherwise it will not work for multiple SQL database.\n// For Sharded SQL, it will execute the statement for all shards\nfunc (mdb *DB) ExecSchemaOperationQuery(ctx context.Context, stmt string, args ...interface{}) error {\n\t_, err := mdb.driver.ExecDDL(ctx, sqlplugin.DbShardUndefined, stmt, args...)\n\treturn err\n}\n\n// ListTables returns a list of tables in this database\nfunc (mdb *DB) ListTables(database string) ([]string, error) {\n\tvar tables []string\n\terr := mdb.driver.SelectForSchemaQuery(sqlplugin.DbShardUndefined, &tables, fmt.Sprintf(listTablesQuery, database))\n\treturn tables, err\n}\n\n// DropTable drops a given table from the database\nfunc (mdb *DB) DropTable(name string) error {\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(dropTableQuery, name))\n}\n\n// DropAllTables drops all tables from this database\nfunc (mdb *DB) DropAllTables(database string) error {\n\ttables, err := mdb.ListTables(database)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, tab := range tables {\n\t\tif err := mdb.DropTable(tab); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// CreateDatabase creates a database if it doesn't exist\nfunc (mdb *DB) CreateDatabase(name string) error {\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(createDatabaseQuery, name))\n}\n\n// DropDatabase drops a database\nfunc (mdb *DB) DropDatabase(name string) error {\n\treturn mdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(dropDatabaseQuery, name))\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/configstore.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc (mdb *DB) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\t_, err := mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, _insertConfigQuery, row.RowType, -1*row.Version, mdb.converter.ToDateTime(row.Timestamp), row.Values.Data, row.Values.Encoding)\n\treturn err\n}\n\nfunc (mdb *DB) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tvar row sqlplugin.ClusterConfigRow\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, _selectLatestConfigQuery, rowType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trow.Version *= -1\n\treturn &persistence.InternalConfigStoreEntry{\n\t\tRowType:   row.RowType,\n\t\tVersion:   row.Version,\n\t\tTimestamp: mdb.converter.FromDateTime(row.Timestamp),\n\t\tValues: &persistence.DataBlob{\n\t\t\tData:     row.Data,\n\t\t\tEncoding: constants.EncodingType(row.DataEncoding),\n\t\t},\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/configstore_sql.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mysql\n\nconst (\n\t_selectLatestConfigQuery = \"SELECT row_type, version, timestamp, data, data_encoding FROM cluster_config WHERE row_type = ? ORDER BY version LIMIT 1;\"\n\n\t_insertConfigQuery = \"INSERT INTO cluster_config (row_type, version, timestamp, data, data_encoding) VALUES(?, ?, ?, ?, ?)\"\n)\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/configstore_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestInsertConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\trow         *persistence.InternalConfigStoreEntry\n\t\tmockSetup   func(*sqldriver.MockDriver)\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\trow: &persistence.InternalConfigStoreEntry{\n\t\t\t\tValues: &persistence.DataBlob{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(gomock.Any(), sqlplugin.DbDefaultShard, _insertConfigQuery, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case\",\n\t\t\trow: &persistence.InternalConfigStoreEntry{\n\t\t\t\tValues: &persistence.DataBlob{},\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(gomock.Any(), sqlplugin.DbDefaultShard, _insertConfigQuery, gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\tmdb := &DB{driver: mockDriver, converter: &converter{}}\n\n\t\t\t// Setup mock expectations\n\t\t\ttc.mockSetup(mockDriver)\n\n\t\t\terr := mdb.InsertConfig(context.Background(), tc.row)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectLatestConfig(t *testing.T) {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\tname        string\n\t\trowType     int\n\t\tsetupMock   func(*sqldriver.MockDriver)\n\t\texpectError bool\n\t\texpectedRow *persistence.InternalConfigStoreEntry\n\t}{\n\t\t{\n\t\t\tname:    \"Success case\",\n\t\t\trowType: 1,\n\t\t\tsetupMock: func(md *sqldriver.MockDriver) {\n\t\t\t\trow := sqlplugin.ClusterConfigRow{\n\t\t\t\t\tRowType:      1,\n\t\t\t\t\tVersion:      -2,\n\t\t\t\t\tTimestamp:    now,\n\t\t\t\t\tData:         []byte(\"test data\"),\n\t\t\t\t\tDataEncoding: \"json\",\n\t\t\t\t}\n\t\t\t\tmd.EXPECT().GetContext(gomock.Any(), sqlplugin.DbDefaultShard, gomock.Any(), _selectLatestConfigQuery, 1).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, shardID int, r *sqlplugin.ClusterConfigRow, query string, args ...interface{}) error {\n\t\t\t\t\t\t*r = row\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpectedRow: &persistence.InternalConfigStoreEntry{\n\t\t\t\tRowType:   1,\n\t\t\t\tVersion:   2,\n\t\t\t\tTimestamp: now,\n\t\t\t\tValues: &persistence.DataBlob{\n\t\t\t\t\tData:     []byte(\"test data\"),\n\t\t\t\t\tEncoding: constants.EncodingType(\"json\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"Error case\",\n\t\t\trowType: 2,\n\t\t\tsetupMock: func(md *sqldriver.MockDriver) {\n\t\t\t\tmd.EXPECT().GetContext(gomock.Any(), sqlplugin.DbDefaultShard, gomock.Any(), _selectLatestConfigQuery, 2).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\tmdb := &DB{driver: mockDriver, converter: &converter{}}\n\n\t\t\ttc.setupMock(mockDriver)\n\n\t\t\trow, err := mdb.SelectLatestConfig(context.Background(), tc.rowType)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t\tassert.Equal(t, tc.expectedRow, row, \"Expected result to be the same for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/db.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"time\"\n\n\t\"github.com/VividCortex/mysqlerr\"\n\t\"github.com/go-sql-driver/mysql\"\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype (\n\tDB struct {\n\t\tconverter   DataConverter\n\t\tdriver      sqldriver.Driver\n\t\toriginalDBs []*sqlx.DB\n\t\tnumDBShards int\n\t}\n)\n\n// NewDB returns an instance of DB, which is a logical\n// connection to the underlying mysql database\n// dbShardID is needed when tx is not nil\nfunc NewDB(xdbs []*sqlx.DB, tx *sqlx.Tx, dbShardID int, numDBShards int, converter DataConverter) (*DB, error) {\n\tdriver, err := sqldriver.NewDriver(xdbs, tx, dbShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdb := &DB{\n\t\tconverter:   converter,\n\t\toriginalDBs: xdbs, // this is kept because NewDB will be called again when starting a transaction\n\t\tdriver:      driver,\n\t\tnumDBShards: numDBShards,\n\t}\n\n\treturn db, nil\n}\n\n// NewDBWithDriver returns an instance of DB with the given driver\nfunc NewDBWithDriver(originalDBs []*sqlx.DB, driver sqldriver.Driver, numDBShards int, converter DataConverter) *DB {\n\treturn &DB{\n\t\tconverter:   converter,\n\t\toriginalDBs: originalDBs,\n\t\tdriver:      driver,\n\t\tnumDBShards: numDBShards,\n\t}\n}\n\nfunc (mdb *DB) GetTotalNumDBShards() int {\n\treturn mdb.numDBShards\n}\n\nvar _ sqlplugin.AdminDB = (*DB)(nil)\nvar _ sqlplugin.DB = (*DB)(nil)\nvar _ sqlplugin.Tx = (*DB)(nil)\n\nfunc (mdb *DB) IsDupEntryError(err error) bool {\n\tsqlErr, ok := err.(*mysql.MySQLError)\n\t// ErrDupEntry MySQL Error 1062 indicates a duplicate primary key i.e. the row already exists,\n\t// so we don't do the insert and return a ConditionalUpdate error.\n\treturn ok && sqlErr.Number == mysqlerr.ER_DUP_ENTRY\n}\n\nfunc (mdb *DB) IsNotFoundError(err error) bool {\n\treturn err == sql.ErrNoRows\n}\n\nfunc (mdb *DB) IsTimeoutError(err error) bool {\n\tif err == context.DeadlineExceeded {\n\t\treturn true\n\t}\n\tsqlErr, ok := err.(*mysql.MySQLError)\n\tif ok {\n\t\tif sqlErr.Number == mysqlerr.ER_NET_READ_INTERRUPTED ||\n\t\t\tsqlErr.Number == mysqlerr.ER_NET_WRITE_INTERRUPTED ||\n\t\t\tsqlErr.Number == mysqlerr.ER_LOCK_WAIT_TIMEOUT ||\n\t\t\tsqlErr.Number == mysqlerr.ER_XA_RBTIMEOUT ||\n\t\t\tsqlErr.Number == mysqlerr.ER_QUERY_TIMEOUT ||\n\t\t\tsqlErr.Number == mysqlerr.ER_LOCKING_SERVICE_TIMEOUT ||\n\t\t\tsqlErr.Number == mysqlerr.ER_REGEXP_TIME_OUT {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc (mdb *DB) IsThrottlingError(err error) bool {\n\tsqlErr, ok := err.(*mysql.MySQLError)\n\tif ok {\n\t\tif sqlErr.Number == mysqlerr.ER_CON_COUNT_ERROR ||\n\t\t\tsqlErr.Number == mysqlerr.ER_TOO_MANY_USER_CONNECTIONS ||\n\t\t\tsqlErr.Number == mysqlerr.ER_TOO_MANY_CONCURRENT_TRXS ||\n\t\t\tsqlErr.Number == mysqlerr.ER_CLONE_TOO_MANY_CONCURRENT_CLONES {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// BeginTx starts a new transaction and returns a reference to the Tx object\nfunc (mdb *DB) BeginTx(ctx context.Context, dbShardID int) (sqlplugin.Tx, error) {\n\txtx, err := mdb.driver.BeginTxx(ctx, dbShardID, nil)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewDB(mdb.originalDBs, xtx, dbShardID, mdb.numDBShards, mdb.converter)\n}\n\n// Commit commits a previously started transaction\nfunc (mdb *DB) Commit() error {\n\treturn mdb.driver.Commit()\n}\n\n// Rollback triggers rollback of a previously started transaction\nfunc (mdb *DB) Rollback() error {\n\treturn mdb.driver.Rollback()\n}\n\n// Close closes the connection to the mysql db\nfunc (mdb *DB) Close() error {\n\treturn mdb.driver.Close()\n}\n\n// PluginName returns the name of the mysql plugin\nfunc (mdb *DB) PluginName() string {\n\treturn PluginName\n}\n\n// SupportsTTL returns weather MySQL supports TTL\nfunc (mdb *DB) SupportsTTL() bool {\n\treturn false\n}\n\n// MaxAllowedTTL returns the max allowed ttl MySQL supports\nfunc (mdb *DB) MaxAllowedTTL() (*time.Duration, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// SupportsTTL returns weather MySQL supports Asynchronous transaction\nfunc (mdb *DB) SupportsAsyncTransaction() bool {\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/domain.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tcreateDomainQuery = `INSERT INTO \n domains (id, name, is_global, data, data_encoding)\n VALUES(?, ?, ?, ?, ?)`\n\n\tupdateDomainQuery = `UPDATE domains \n SET name = ?, data = ?, data_encoding = ?\n WHERE shard_id=54321 AND id = ?`\n\n\tgetDomainPart = `SELECT id, name, is_global, data, data_encoding FROM domains`\n\n\tgetDomainByIDQuery   = getDomainPart + ` WHERE shard_id=? AND id = ?`\n\tgetDomainByNameQuery = getDomainPart + ` WHERE shard_id=? AND name = ?`\n\n\tlistDomainsQuery      = getDomainPart + ` WHERE shard_id=? ORDER BY id LIMIT ?`\n\tlistDomainsRangeQuery = getDomainPart + ` WHERE shard_id=? AND id > ? ORDER BY id LIMIT ?`\n\n\tdeleteDomainByIDQuery   = `DELETE FROM domains WHERE shard_id=? AND id = ?`\n\tdeleteDomainByNameQuery = `DELETE FROM domains WHERE shard_id=? AND name = ?`\n\n\tgetDomainMetadataQuery    = `SELECT notification_version FROM domain_metadata`\n\tlockDomainMetadataQuery   = `SELECT notification_version FROM domain_metadata FOR UPDATE`\n\tupdateDomainMetadataQuery = `UPDATE domain_metadata SET notification_version = ? WHERE notification_version = ?`\n)\n\nconst (\n\tshardID = 54321\n)\n\nvar errMissingArgs = errors.New(\"missing one or more args for API\")\n\n// InsertIntoDomain inserts a single row into domains table\nfunc (mdb *DB) InsertIntoDomain(ctx context.Context, row *sqlplugin.DomainRow) (sql.Result, error) {\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, createDomainQuery, row.ID, row.Name, row.IsGlobal, row.Data, row.DataEncoding)\n}\n\n// UpdateDomain updates a single row in domains table\nfunc (mdb *DB) UpdateDomain(ctx context.Context, row *sqlplugin.DomainRow) (sql.Result, error) {\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, updateDomainQuery, row.Name, row.Data, row.DataEncoding, row.ID)\n}\n\n// SelectFromDomain reads one or more rows from domains table\nfunc (mdb *DB) SelectFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tswitch {\n\tcase filter.ID != nil || filter.Name != nil:\n\t\treturn mdb.selectFromDomain(ctx, filter)\n\tcase filter.PageSize != nil && *filter.PageSize > 0:\n\t\treturn mdb.selectAllFromDomain(ctx, filter)\n\tdefault:\n\t\treturn nil, errMissingArgs\n\t}\n}\n\nfunc (mdb *DB) selectFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tvar err error\n\tvar row sqlplugin.DomainRow\n\tswitch {\n\tcase filter.ID != nil:\n\t\terr = mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, getDomainByIDQuery, shardID, *filter.ID)\n\tcase filter.Name != nil:\n\t\terr = mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, getDomainByNameQuery, shardID, *filter.Name)\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn []sqlplugin.DomainRow{row}, err\n}\n\nfunc (mdb *DB) selectAllFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.DomainRow\n\tswitch {\n\tcase filter.GreaterThanID != nil:\n\t\terr = mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, listDomainsRangeQuery, shardID, *filter.GreaterThanID, *filter.PageSize)\n\tdefault:\n\t\terr = mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, listDomainsQuery, shardID, filter.PageSize)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromDomain deletes a single row in domains table\nfunc (mdb *DB) DeleteFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) (sql.Result, error) {\n\tvar err error\n\tvar result sql.Result\n\tswitch {\n\tcase filter.ID != nil:\n\t\tresult, err = mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, deleteDomainByIDQuery, shardID, filter.ID)\n\tdefault:\n\t\tresult, err = mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, deleteDomainByNameQuery, shardID, filter.Name)\n\t}\n\treturn result, err\n}\n\n// LockDomainMetadata acquires a write lock on a single row in domain_metadata table\nfunc (mdb *DB) LockDomainMetadata(ctx context.Context) error {\n\tvar row sqlplugin.DomainMetadataRow\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row.NotificationVersion, lockDomainMetadataQuery)\n\treturn err\n}\n\n// SelectFromDomainMetadata reads a single row in domain_metadata table\nfunc (mdb *DB) SelectFromDomainMetadata(ctx context.Context) (*sqlplugin.DomainMetadataRow, error) {\n\tvar row sqlplugin.DomainMetadataRow\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row.NotificationVersion, getDomainMetadataQuery)\n\treturn &row, err\n}\n\n// UpdateDomainMetadata updates a single row in domain_metadata table\nfunc (mdb *DB) UpdateDomainMetadata(ctx context.Context, row *sqlplugin.DomainMetadataRow) (sql.Result, error) {\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, updateDomainMetadataQuery, row.NotificationVersion+1, row.NotificationVersion)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/domain_audit_log.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\t_insertDomainAuditLogQuery = `INSERT INTO domain_audit_log (\n\t\tdomain_id, event_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\t) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\t_selectDomainAuditLogsQuery = `SELECT\n\t\tevent_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\tFROM domain_audit_log\n\tWHERE domain_id = ? AND operation_type = ? AND created_time >= ?\n\tAND (created_time < ? OR (created_time = ? AND event_id > ?))\n\tORDER BY created_time DESC, event_id ASC\n\tLIMIT ?`\n\t_selectAllDomainAuditLogsQuery = `SELECT\n\t\tevent_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\tFROM domain_audit_log\n\tWHERE domain_id = ? AND operation_type = ? AND created_time >= ?\n\tAND (created_time < ? OR (created_time = ? AND event_id > ?))\n\tORDER BY created_time DESC, event_id ASC`\n)\n\n// InsertIntoDomainAuditLog inserts a single row into domain_audit_log table\nfunc (mdb *DB) InsertIntoDomainAuditLog(ctx context.Context, row *sqlplugin.DomainAuditLogRow) (sql.Result, error) {\n\treturn mdb.driver.ExecContext(\n\t\tctx,\n\t\tsqlplugin.DbDefaultShard,\n\t\t_insertDomainAuditLogQuery,\n\t\trow.DomainID,\n\t\trow.EventID,\n\t\trow.StateBefore,\n\t\trow.StateBeforeEncoding,\n\t\trow.StateAfter,\n\t\trow.StateAfterEncoding,\n\t\trow.OperationType,\n\t\trow.CreatedTime,\n\t\trow.LastUpdatedTime,\n\t\trow.Identity,\n\t\trow.IdentityType,\n\t\trow.Comment,\n\t)\n}\n\n// SelectFromDomainAuditLogs returns audit log entries for a domain, operation type, and time range\nfunc (mdb *DB) SelectFromDomainAuditLogs(\n\tctx context.Context,\n\tfilter *sqlplugin.DomainAuditLogFilter,\n) ([]*sqlplugin.DomainAuditLogRow, error) {\n\targs := []interface{}{\n\t\tfilter.DomainID,\n\t\tfilter.OperationType,\n\t\t*filter.MinCreatedTime,\n\t\t*filter.PageMaxCreatedTime,\n\t\t*filter.PageMaxCreatedTime,\n\t\t*filter.PageMinEventID,\n\t}\n\n\tvar rows []*sqlplugin.DomainAuditLogRow\n\tif filter.PageSize > 0 {\n\t\targs = append(args, filter.PageSize)\n\t\terr := mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, _selectDomainAuditLogsQuery, args...)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\terr := mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, _selectAllDomainAuditLogsQuery, args...)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn rows, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/domain_audit_log_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestInsertIntoDomainAuditLog(t *testing.T) {\n\tnow := time.Now().UTC()\n\n\ttests := []struct {\n\t\tname      string\n\t\trow       *sqlplugin.DomainAuditLogRow\n\t\tmockSetup func(*sqldriver.MockDriver)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully inserted\",\n\t\t\trow: &sqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:            uuid.New(),\n\t\t\t\tEventID:             uuid.New(),\n\t\t\t\tStateBefore:         []byte(\"state-before\"),\n\t\t\t\tStateBeforeEncoding: constants.EncodingTypeJSON,\n\t\t\t\tStateAfter:          []byte(\"state-after\"),\n\t\t\t\tStateAfterEncoding:  constants.EncodingTypeJSON,\n\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:         now,\n\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\tIdentity:            \"test-identity\",\n\t\t\t\tIdentityType:        \"user\",\n\t\t\t\tComment:             \"test comment\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\t_insertDomainAuditLogQuery,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t[]byte(\"state-before\"),\n\t\t\t\t\tconstants.EncodingTypeJSON,\n\t\t\t\t\t[]byte(\"state-after\"),\n\t\t\t\t\tconstants.EncodingTypeJSON,\n\t\t\t\t\tpersistence.DomainAuditOperationTypeFailover,\n\t\t\t\t\tnow,\n\t\t\t\t\tnow,\n\t\t\t\t\t\"test-identity\",\n\t\t\t\t\t\"user\",\n\t\t\t\t\t\"test comment\",\n\t\t\t\t).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"exec failed\",\n\t\t\trow: &sqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:      uuid.New(),\n\t\t\t\tEventID:       uuid.New(),\n\t\t\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:   now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, errors.New(\"exec failed\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\ttc.mockSetup(mockDriver)\n\n\t\t\tmdb := &DB{\n\t\t\t\tdriver:    mockDriver,\n\t\t\t\tconverter: &converter{},\n\t\t\t}\n\n\t\t\t_, err := mdb.InsertIntoDomainAuditLog(context.Background(), tc.row)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectFromDomainAuditLogs(t *testing.T) {\n\tdomainID := \"d1111111-1111-1111-1111-111111111111\"\n\toperationType := persistence.DomainAuditOperationTypeFailover\n\tminTime := time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)\n\tmaxTime := time.Date(2024, 12, 31, 23, 59, 59, 0, time.UTC)\n\n\t// Create times in descending order\n\tcreatedTime1 := time.Date(2024, 7, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime2 := time.Date(2024, 6, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime3 := time.Date(2024, 5, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime4 := time.Date(2024, 4, 1, 12, 0, 0, 0, time.UTC)\n\n\teventID1 := \"e1111111-1111-1111-1111-111111111111\"\n\teventID2 := \"e2222222-2222-2222-2222-222222222222\"\n\teventID3 := \"e3333333-3333-3333-3333-333333333333\"\n\teventID4 := \"e4444444-4444-4444-4444-444444444444\"\n\n\t// Default page cursor: maxCreatedTime with max UUID (no page token)\n\tdefaultPageMaxCreatedTime := maxTime\n\tdefaultPageMinEventID := \"ffffffff-ffff-ffff-ffff-ffffffffffff\"\n\n\ttests := []struct {\n\t\tname      string\n\t\tfilter    *sqlplugin.DomainAuditLogFilter\n\t\tmockSetup func(*sqldriver.MockDriver)\n\t\twantRows  []*sqlplugin.DomainAuditLogRow\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"pageSize limits number of results; no pageToken\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageSize:           2,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMaxCreatedTime, defaultPageMinEventID, 2,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\trows := dest.(*[]*sqlplugin.DomainAuditLogRow)\n\t\t\t\t\t*rows = []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID1,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime1,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID2,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime2,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID1,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime1,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID2,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime2,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"pageToken filters results by createdTime and eventID; pageSize is 0\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageSize:           0,\n\t\t\t\tPageMaxCreatedTime: &createdTime2,\n\t\t\t\tPageMinEventID:     &eventID2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, createdTime2, createdTime2, eventID2,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\trows := dest.(*[]*sqlplugin.DomainAuditLogRow)\n\t\t\t\t\t*rows = []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID3,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime3,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID4,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime4,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID3,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime3,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID4,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime4,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success with no results\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMaxCreatedTime, defaultPageMinEventID,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{},\n\t\t\twantErr:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"error when select fails\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMaxCreatedTime, defaultPageMinEventID,\n\t\t\t\t).Return(errors.New(\"select failed\"))\n\t\t\t},\n\t\t\twantRows: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\ttc.mockSetup(mockDriver)\n\n\t\t\tmdb := &DB{\n\t\t\t\tdriver:    mockDriver,\n\t\t\t\tconverter: &converter{},\n\t\t\t}\n\n\t\t\trows, err := mdb.SelectFromDomainAuditLogs(context.Background(), tc.filter)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, len(tc.wantRows), len(rows), \"number of rows should match\")\n\n\t\t\tfor i, wantRow := range tc.wantRows {\n\t\t\t\tif i < len(rows) {\n\t\t\t\t\tassert.Equal(t, wantRow.EventID, rows[i].EventID, \"row %d eventID\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.DomainID, rows[i].DomainID, \"row %d domainID\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.OperationType, rows[i].OperationType, \"row %d operationType\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.CreatedTime.Unix(), rows[i].CreatedTime.Unix(), \"row %d createdTime\", i)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/dsn_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"net/url\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\ntype StoreTestSuite struct {\n\tsuite.Suite\n}\n\nfunc TestStoreTestSuite(t *testing.T) {\n\tsuite.Run(t, new(StoreTestSuite))\n}\n\nfunc (s *StoreTestSuite) TestBuildDSN() {\n\ttestCases := []struct {\n\t\tin              config.SQL\n\t\toutURLPath      string\n\t\toutIsolationKey string\n\t\toutIsolationVal string\n\t}{\n\t\t{\n\t\t\tin: config.SQL{\n\t\t\t\tUser:            \"test\",\n\t\t\t\tPassword:        \"pass\",\n\t\t\t\tConnectProtocol: \"tcp\",\n\t\t\t\tConnectAddr:     \"192.168.0.1:3306\",\n\t\t\t\tDatabaseName:    \"db1\",\n\t\t\t\tEncodingType:    \"thriftrw\",\n\t\t\t\tDecodingTypes:   []string{\"thriftrw\"},\n\t\t\t},\n\t\t\toutIsolationKey: \"transaction_isolation\",\n\t\t\toutIsolationVal: \"'READ-COMMITTED'\",\n\t\t\toutURLPath:      \"test:pass@tcp(192.168.0.1:3306)/db1?\",\n\t\t},\n\t\t{\n\t\t\tin: config.SQL{\n\t\t\t\tUser:              \"test\",\n\t\t\t\tPassword:          \"pass\",\n\t\t\t\tConnectProtocol:   \"tcp\",\n\t\t\t\tConnectAddr:       \"192.168.0.1:3306\",\n\t\t\t\tDatabaseName:      \"db1\",\n\t\t\t\tConnectAttributes: map[string]string{\"k1\": \"v1\", \"k2\": \"v2\"},\n\t\t\t\tEncodingType:      \"thriftrw\",\n\t\t\t\tDecodingTypes:     []string{\"thriftrw\"},\n\t\t\t},\n\t\t\toutIsolationKey: \"transaction_isolation\",\n\t\t\toutIsolationVal: \"'READ-COMMITTED'\",\n\t\t\toutURLPath:      \"test:pass@tcp(192.168.0.1:3306)/db1?\",\n\t\t},\n\t\t{\n\t\t\tin: config.SQL{\n\t\t\t\tUser:              \"test\",\n\t\t\t\tPassword:          \"pass\",\n\t\t\t\tConnectProtocol:   \"tcp\",\n\t\t\t\tConnectAddr:       \"192.168.0.1:3306\",\n\t\t\t\tDatabaseName:      \"db1\",\n\t\t\t\tConnectAttributes: map[string]string{\"k1\": \"v1\", \"k2\": \"v2\", \"tx_isolation\": \"'REPEATABLE-READ'\"},\n\t\t\t\tEncodingType:      \"thriftrw\",\n\t\t\t\tDecodingTypes:     []string{\"thriftrw\"},\n\t\t\t},\n\t\t\toutIsolationKey: \"tx_isolation\",\n\t\t\toutIsolationVal: \"'repeatable-read'\",\n\t\t\toutURLPath:      \"test:pass@tcp(192.168.0.1:3306)/db1?\",\n\t\t},\n\t\t{\n\t\t\tin: config.SQL{\n\t\t\t\tUser:              \"test\",\n\t\t\t\tPassword:          \"pass\",\n\t\t\t\tConnectProtocol:   \"tcp\",\n\t\t\t\tConnectAddr:       \"192.168.0.1:3306\",\n\t\t\t\tDatabaseName:      \"db1\",\n\t\t\t\tConnectAttributes: map[string]string{\"k1\": \"v1\", \"k2\": \"v2\", \"tx_isolation\": \"REPEATABLE-READ\"},\n\t\t\t\tEncodingType:      \"thriftrw\",\n\t\t\t\tDecodingTypes:     []string{\"thriftrw\"},\n\t\t\t},\n\t\t\toutIsolationKey: \"tx_isolation\",\n\t\t\toutIsolationVal: \"'repeatable-read'\",\n\t\t\toutURLPath:      \"test:pass@tcp(192.168.0.1:3306)/db1?\",\n\t\t},\n\t\t{\n\t\t\tin: config.SQL{\n\t\t\t\tUser:              \"test\",\n\t\t\t\tPassword:          \"pass\",\n\t\t\t\tConnectProtocol:   \"tcp\",\n\t\t\t\tConnectAddr:       \"192.168.0.1:3306\",\n\t\t\t\tDatabaseName:      \"db1\",\n\t\t\t\tConnectAttributes: map[string]string{\"k1\": \"v1\", \"k2\": \"v2\", \"transaction_isolation\": \"REPEATABLE-READ\"},\n\t\t\t\tEncodingType:      \"thriftrw\",\n\t\t\t\tDecodingTypes:     []string{\"thriftrw\"},\n\t\t\t},\n\t\t\toutIsolationKey: \"transaction_isolation\",\n\t\t\toutIsolationVal: \"'repeatable-read'\",\n\t\t\toutURLPath:      \"test:pass@tcp(192.168.0.1:3306)/db1?\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tout := buildDSN(&tc.in)\n\t\ts.True(strings.HasPrefix(out, tc.outURLPath), \"invalid url path\")\n\t\ttokens := strings.Split(out, \"?\")\n\t\ts.Equal(2, len(tokens), \"invalid url\")\n\t\tqry, err := url.Parse(\"?\" + tokens[1])\n\t\ts.NoError(err)\n\t\twantAttrs := buildExpectedURLParams(tc.in.ConnectAttributes, tc.outIsolationKey, tc.outIsolationVal)\n\t\ts.Equal(wantAttrs, qry.Query(), \"invalid dsn url params\")\n\t}\n}\n\nfunc buildExpectedURLParams(attrs map[string]string, isolationKey string, isolationValue string) url.Values {\n\tresult := make(map[string][]string, len(dsnAttrOverrides)+len(attrs)+1)\n\tfor k, v := range attrs {\n\t\tresult[k] = []string{v}\n\t}\n\tresult[isolationKey] = []string{isolationValue}\n\tfor k, v := range dsnAttrOverrides {\n\t\tresult[k] = []string{v}\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/events.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\t// below are templates for history_node table\n\taddHistoryNodesQuery = `INSERT INTO history_node (` +\n\t\t`shard_id, tree_id, branch_id, node_id, txn_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :tree_id, :branch_id, :node_id, :txn_id, :data, :data_encoding) `\n\n\tgetHistoryNodesQuery = `SELECT node_id, txn_id, data, data_encoding FROM history_node ` +\n\t\t`WHERE shard_id = ? AND tree_id = ? AND branch_id = ? AND node_id >= ? and node_id < ? ORDER BY shard_id, tree_id, branch_id, node_id, txn_id LIMIT ? `\n\n\tdeleteHistoryNodesQuery = `DELETE FROM history_node WHERE shard_id = ? AND tree_id = ? AND branch_id = ? AND node_id >= ? `\n\n\tdeleteHistoryNodesByBatchQuery = `DELETE FROM history_node WHERE shard_id = ? AND tree_id = ? AND branch_id = ? AND node_id >= ? ORDER BY shard_id, tree_id, branch_id, node_id, txn_id LIMIT ? `\n\n\t// below are templates for history_tree table\n\taddHistoryTreeQuery = `INSERT INTO history_tree (` +\n\t\t`shard_id, tree_id, branch_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :tree_id, :branch_id, :data, :data_encoding) `\n\n\tgetHistoryTreeQuery = `SELECT branch_id, data, data_encoding FROM history_tree WHERE shard_id = ? AND tree_id = ? `\n\n\tdeleteHistoryTreeQuery = `DELETE FROM history_tree WHERE shard_id = ? AND tree_id = ? AND branch_id = ? `\n\n\tgetAllHistoryTreeQuery = `SELECT shard_id, tree_id, branch_id, data, data_encoding FROM history_tree WHERE (shard_id = ? AND tree_id = ? AND branch_id > ?) OR (shard_id = ? AND tree_id > ?) OR (shard_id > ?) ORDER BY shard_id, tree_id, branch_id LIMIT ?`\n)\n\n// For history_node table:\n\n// InsertIntoHistoryNode inserts a row into history_node table\nfunc (mdb *DB) InsertIntoHistoryNode(ctx context.Context, row *sqlplugin.HistoryNodeRow) (sql.Result, error) {\n\t// NOTE: Query 5.6 doesn't support clustering order, to workaround, we let txn_id multiple by -1\n\t*row.TxnID *= -1\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(row.TreeID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, addHistoryNodesQuery, row)\n}\n\n// SelectFromHistoryNode reads one or more rows from history_node table\nfunc (mdb *DB) SelectFromHistoryNode(ctx context.Context, filter *sqlplugin.HistoryNodeFilter) ([]sqlplugin.HistoryNodeRow, error) {\n\tvar rows []sqlplugin.HistoryNodeRow\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getHistoryNodesQuery,\n\t\tfilter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID, *filter.MaxNodeID, filter.PageSize)\n\t// NOTE: since we let txn_id multiple by -1 when inserting, we have to revert it back here\n\tfor _, row := range rows {\n\t\t*row.TxnID *= -1\n\t}\n\treturn rows, err\n}\n\n// DeleteFromHistoryNode deletes one or more rows from history_node table\nfunc (mdb *DB) DeleteFromHistoryNode(ctx context.Context, filter *sqlplugin.HistoryNodeFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteHistoryNodesByBatchQuery, filter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteHistoryNodesQuery, filter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID)\n}\n\n// For history_tree table:\n\n// InsertIntoHistoryTree inserts a row into history_tree table\nfunc (mdb *DB) InsertIntoHistoryTree(ctx context.Context, row *sqlplugin.HistoryTreeRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(row.TreeID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, addHistoryTreeQuery, row)\n}\n\n// SelectFromHistoryTree reads one or more rows from history_tree table\nfunc (mdb *DB) SelectFromHistoryTree(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) ([]sqlplugin.HistoryTreeRow, error) {\n\tvar rows []sqlplugin.HistoryTreeRow\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getHistoryTreeQuery, filter.ShardID, filter.TreeID)\n\treturn rows, err\n}\n\n// DeleteFromHistoryTree deletes one or more rows from history_tree table\nfunc (mdb *DB) DeleteFromHistoryTree(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteHistoryTreeQuery, filter.ShardID, filter.TreeID, *filter.BranchID)\n}\n\nfunc (mdb *DB) GetAllHistoryTreeBranches(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) ([]sqlplugin.HistoryTreeRow, error) {\n\tvar rows []sqlplugin.HistoryTreeRow\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getAllHistoryTreeQuery, filter.ShardID, filter.TreeID, *filter.BranchID, filter.ShardID, filter.TreeID, filter.ShardID, filter.PageSize)\n\treturn rows, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/execution.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\texecutionsColumns = `shard_id, domain_id, workflow_id, run_id, next_event_id, last_write_version, data, data_encoding`\n\n\tcreateExecutionQuery = `INSERT INTO executions(` + executionsColumns + `)\n VALUES(:shard_id, :domain_id, :workflow_id, :run_id, :next_event_id, :last_write_version, :data, :data_encoding)`\n\n\tupdateExecutionQuery = `UPDATE executions SET\n next_event_id = :next_event_id, last_write_version = :last_write_version, data = :data, data_encoding = :data_encoding\n WHERE shard_id = :shard_id AND domain_id = :domain_id AND workflow_id = :workflow_id AND run_id = :run_id`\n\n\tgetExecutionQuery = `SELECT ` + executionsColumns + ` FROM executions\n WHERE shard_id = ? AND domain_id = ? AND workflow_id = ? AND run_id = ?`\n\n\tlistExecutionQuery = `SELECT ` + executionsColumns + ` FROM executions\n WHERE shard_id = ? AND workflow_id > ? ORDER BY workflow_id LIMIT ?`\n\n\tdeleteExecutionQuery = `DELETE FROM executions\n WHERE shard_id = ? AND domain_id = ? AND workflow_id = ? AND run_id = ?`\n\n\tlockExecutionQueryBase = `SELECT next_event_id FROM executions\n WHERE shard_id = ? AND domain_id = ? AND workflow_id = ? AND run_id = ?`\n\n\twriteLockExecutionQuery = lockExecutionQueryBase + ` FOR UPDATE`\n\treadLockExecutionQuery  = lockExecutionQueryBase + ` LOCK IN SHARE MODE`\n\n\tcreateCurrentExecutionQuery = `INSERT INTO current_executions\n(shard_id, domain_id, workflow_id, run_id, create_request_id, state, close_status, start_version, last_write_version) VALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :create_request_id, :state, :close_status, :start_version, :last_write_version)`\n\n\tdeleteCurrentExecutionQuery = \"DELETE FROM current_executions WHERE shard_id=? AND domain_id=? AND workflow_id=? AND run_id=?\"\n\n\tgetCurrentExecutionQuery = `SELECT\nshard_id, domain_id, workflow_id, run_id, create_request_id, state, close_status, start_version, last_write_version\nFROM current_executions WHERE shard_id = ? AND domain_id = ? AND workflow_id = ?`\n\n\tlockCurrentExecutionJoinExecutionsQuery = `SELECT\nce.shard_id, ce.domain_id, ce.workflow_id, ce.run_id, ce.create_request_id, ce.state, ce.close_status, ce.start_version, e.last_write_version\nFROM current_executions ce\nINNER JOIN executions e ON e.shard_id = ce.shard_id AND e.domain_id = ce.domain_id AND e.workflow_id = ce.workflow_id AND e.run_id = ce.run_id\nWHERE ce.shard_id = ? AND ce.domain_id = ? AND ce.workflow_id = ? FOR UPDATE`\n\n\tlockCurrentExecutionQuery = getCurrentExecutionQuery + ` FOR UPDATE`\n\n\tupdateCurrentExecutionsQuery = `UPDATE current_executions SET\nrun_id = :run_id,\ncreate_request_id = :create_request_id,\nstate = :state,\nclose_status = :close_status,\nstart_version = :start_version,\nlast_write_version = :last_write_version\nWHERE\nshard_id = :shard_id AND\ndomain_id = :domain_id AND\nworkflow_id = :workflow_id\n`\n\n\tgetTransferTasksQuery = `SELECT task_id, data, data_encoding\nFROM transfer_tasks WHERE shard_id = ? AND task_id >= ? AND task_id < ? ORDER BY shard_id, task_id LIMIT ?`\n\n\tcreateTransferTasksQuery = `INSERT INTO transfer_tasks(shard_id, task_id, data, data_encoding)\n VALUES(:shard_id, :task_id, :data, :data_encoding)`\n\n\tdeleteTransferTaskQuery             = `DELETE FROM transfer_tasks WHERE shard_id = ? AND task_id = ?`\n\trangeDeleteTransferTaskQuery        = `DELETE FROM transfer_tasks WHERE shard_id = ? AND task_id >= ? AND task_id < ?`\n\trangeDeleteTransferTaskByBatchQuery = rangeDeleteTransferTaskQuery + ` ORDER BY task_id LIMIT ?`\n\n\tgetCrossClusterTasksQuery = `SELECT task_id, data, data_encoding\nFROM cross_cluster_tasks WHERE target_cluster = ? AND shard_id = ? AND task_id > ? AND task_id <= ? ORDER BY task_id LIMIT ?`\n\n\tcreateCrossClusterTasksQuery = `INSERT INTO cross_cluster_tasks(target_cluster, shard_id, task_id, data, data_encoding)\nVALUES(:target_cluster, :shard_id, :task_id, :data, :data_encoding)`\n\n\tdeleteCrossClusterTaskQuery             = `DELETE FROM cross_cluster_tasks WHERE target_cluster = ? AND shard_id = ? AND task_id = ?`\n\trangeDeleteCrossClusterTaskQuery        = `DELETE FROM cross_cluster_tasks WHERE target_cluster = ? AND shard_id = ? AND task_id >= ? AND task_id < ?`\n\trangeDeleteCrossClusterTaskByBatchQuery = rangeDeleteCrossClusterTaskQuery + ` ORDER BY task_id LIMIT ?`\n\n\tcreateTimerTasksQuery = `INSERT INTO timer_tasks (shard_id, visibility_timestamp, task_id, data, data_encoding)\n  VALUES (:shard_id, :visibility_timestamp, :task_id, :data, :data_encoding)`\n\n\tgetTimerTasksQuery = `SELECT visibility_timestamp, task_id, data, data_encoding FROM timer_tasks\n  WHERE shard_id = ?\n  AND ((visibility_timestamp >= ? AND task_id >= ?) OR visibility_timestamp > ?)\n  AND visibility_timestamp < ?\n  ORDER BY visibility_timestamp,task_id LIMIT ?`\n\n\tdeleteTimerTaskQuery             = `DELETE FROM timer_tasks WHERE shard_id = ? AND visibility_timestamp = ? AND task_id = ?`\n\trangeDeleteTimerTaskQuery        = `DELETE FROM timer_tasks WHERE shard_id = ? AND visibility_timestamp >= ? AND visibility_timestamp < ?`\n\trangeDeleteTimerTaskByBatchQuery = rangeDeleteTimerTaskQuery + ` ORDER BY visibility_timestamp,task_id LIMIT ?`\n\n\tcreateReplicationTasksQuery = `INSERT INTO replication_tasks (shard_id, task_id, data, data_encoding)\n  VALUES(:shard_id, :task_id, :data, :data_encoding)`\n\n\tgetReplicationTasksQuery = `SELECT task_id, data, data_encoding FROM replication_tasks WHERE\nshard_id = ? AND\ntask_id >= ? AND\ntask_id < ?\nORDER BY task_id LIMIT ?`\n\n\tdeleteReplicationTaskQuery             = `DELETE FROM replication_tasks WHERE shard_id = ? AND task_id = ?`\n\trangeDeleteReplicationTaskQuery        = `DELETE FROM replication_tasks WHERE shard_id = ? AND task_id < ?`\n\trangeDeleteReplicationTaskByBatchQuery = rangeDeleteReplicationTaskQuery + ` ORDER BY task_id LIMIT ?`\n\n\tgetReplicationTasksDLQQuery = `SELECT task_id, data, data_encoding FROM replication_tasks_dlq WHERE\nsource_cluster_name = ? AND\nshard_id = ? AND\ntask_id >= ? AND\ntask_id < ?\nORDER BY task_id LIMIT ?`\n\n\tgetReplicationTaskDLQQuery = `SELECT count(1) as count FROM replication_tasks_dlq WHERE\nsource_cluster_name = ? AND\nshard_id = ?`\n\n\tbufferedEventsColumns     = `shard_id, domain_id, workflow_id, run_id, data, data_encoding`\n\tcreateBufferedEventsQuery = `INSERT INTO buffered_events(` + bufferedEventsColumns + `)\nVALUES (:shard_id, :domain_id, :workflow_id, :run_id, :data, :data_encoding)`\n\n\tdeleteBufferedEventsQuery = `DELETE FROM buffered_events WHERE shard_id=? AND domain_id=? AND workflow_id=? AND run_id=?`\n\tgetBufferedEventsQuery    = `SELECT data, data_encoding FROM buffered_events WHERE\nshard_id=? AND domain_id=? AND workflow_id=? AND run_id=?`\n\n\tinsertReplicationTaskDLQQuery = `\nINSERT INTO replication_tasks_dlq\n            (source_cluster_name,\n             shard_id,\n             task_id,\n             data,\n             data_encoding)\nVALUES     (:source_cluster_name,\n            :shard_id,\n            :task_id,\n            :data,\n            :data_encoding)\n`\n\tdeleteReplicationTaskFromDLQQuery = `\n\tDELETE FROM replication_tasks_dlq\n\t\tWHERE source_cluster_name = ?\n\t\tAND shard_id = ?\n\t\tAND task_id = ?`\n\n\trangeDeleteReplicationTaskFromDLQQuery = `\n\tDELETE FROM replication_tasks_dlq\n\t\tWHERE source_cluster_name = ?\n\t\tAND shard_id = ?\n\t\tAND task_id >= ?\n\t\tAND task_id < ?`\n\trangeDeleteReplicationTaskFromDLQByBatchQuery = rangeDeleteReplicationTaskFromDLQQuery + ` ORDER BY task_id LIMIT ?`\n)\n\n// InsertIntoExecutions inserts a row into executions table\nfunc (mdb *DB) InsertIntoExecutions(ctx context.Context, row *sqlplugin.ExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(row.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createExecutionQuery, row)\n}\n\n// UpdateExecutions updates a single row in executions table\nfunc (mdb *DB) UpdateExecutions(ctx context.Context, row *sqlplugin.ExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(row.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, updateExecutionQuery, row)\n}\n\n// SelectFromExecutions reads a single row from executions table\n// The list execution query result is order by workflow ID only. It may returns duplicate record with pagination.\nfunc (mdb *DB) SelectFromExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) ([]sqlplugin.ExecutionsRow, error) {\n\tvar rows []sqlplugin.ExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tvar err error\n\tif len(filter.DomainID) == 0 && filter.Size > 0 {\n\t\terr = mdb.driver.SelectContext(ctx, dbShardID, &rows, listExecutionQuery, filter.ShardID, filter.WorkflowID, filter.Size)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tvar row sqlplugin.ExecutionsRow\n\t\terr = mdb.driver.GetContext(ctx, dbShardID, &row, getExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trows = append(rows, row)\n\t}\n\n\treturn rows, err\n}\n\n// DeleteFromExecutions deletes a single row from executions table\nfunc (mdb *DB) DeleteFromExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// ReadLockExecutions acquires a write lock on a single row in executions table\nfunc (mdb *DB) ReadLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tvar nextEventID int\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &nextEventID, readLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// WriteLockExecutions acquires a write lock on a single row in executions table\nfunc (mdb *DB) WriteLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tvar nextEventID int\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &nextEventID, writeLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// InsertIntoCurrentExecutions inserts a single row into current_executions table\nfunc (mdb *DB) InsertIntoCurrentExecutions(ctx context.Context, row *sqlplugin.CurrentExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createCurrentExecutionQuery, row)\n}\n\n// UpdateCurrentExecutions updates a single row in current_executions table\nfunc (mdb *DB) UpdateCurrentExecutions(ctx context.Context, row *sqlplugin.CurrentExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, updateCurrentExecutionsQuery, row)\n}\n\n// SelectFromCurrentExecutions reads one or more rows from current_executions table\nfunc (mdb *DB) SelectFromCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (*sqlplugin.CurrentExecutionsRow, error) {\n\tvar row sqlplugin.CurrentExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &row, getCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn &row, err\n}\n\n// DeleteFromCurrentExecutions deletes a single row in current_executions table\nfunc (mdb *DB) DeleteFromCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// LockCurrentExecutions acquires a write lock on a single row in current_executions table\nfunc (mdb *DB) LockCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (*sqlplugin.CurrentExecutionsRow, error) {\n\tvar row sqlplugin.CurrentExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &row, lockCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn &row, err\n}\n\n// LockCurrentExecutionsJoinExecutions joins a row in current_executions with executions table and acquires a\n// write lock on the result\nfunc (mdb *DB) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) ([]sqlplugin.CurrentExecutionsRow, error) {\n\tvar rows []sqlplugin.CurrentExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, lockCurrentExecutionJoinExecutionsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn rows, err\n}\n\n// InsertIntoTransferTasks inserts one or more rows into transfer_tasks table\nfunc (mdb *DB) InsertIntoTransferTasks(ctx context.Context, rows []sqlplugin.TransferTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createTransferTasksQuery, rows)\n}\n\n// SelectFromTransferTasks reads one or more rows from transfer_tasks table\nfunc (mdb *DB) SelectFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) ([]sqlplugin.TransferTasksRow, error) {\n\tvar rows []sqlplugin.TransferTasksRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getTransferTasksQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTransferTasks deletes one row from transfer_tasks table\nfunc (mdb *DB) DeleteFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteTransferTaskQuery, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromTransferTasks deletes multi rows from transfer_tasks table\nfunc (mdb *DB) RangeDeleteFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskByBatchQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID)\n}\n\n// InsertIntoCrossClusterTasks inserts one or more rows into cross_cluster_tasks table\nfunc (mdb *DB) InsertIntoCrossClusterTasks(ctx context.Context, rows []sqlplugin.CrossClusterTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createCrossClusterTasksQuery, rows)\n}\n\n// SelectFromCrossClusterTasks reads one or more rows from cross_cluster_tasks table\nfunc (mdb *DB) SelectFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) ([]sqlplugin.CrossClusterTasksRow, error) {\n\tvar rows []sqlplugin.CrossClusterTasksRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getCrossClusterTasksQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromCrossClusterTasks deletes one row from cross_cluster_tasks table\nfunc (mdb *DB) DeleteFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteCrossClusterTaskQuery, filter.TargetCluster, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromCrossClusterTasks deletes multi rows from cross_cluster_tasks table\nfunc (mdb *DB) RangeDeleteFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteCrossClusterTaskByBatchQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteCrossClusterTaskQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID)\n}\n\n// InsertIntoTimerTasks inserts one or more rows into timer_tasks table\nfunc (mdb *DB) InsertIntoTimerTasks(ctx context.Context, rows []sqlplugin.TimerTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, mdb.GetTotalNumDBShards())\n\tfor i := range rows {\n\t\trows[i].VisibilityTimestamp = mdb.converter.ToDateTime(rows[i].VisibilityTimestamp)\n\t}\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createTimerTasksQuery, rows)\n}\n\n// SelectFromTimerTasks reads one or more rows from timer_tasks table\nfunc (mdb *DB) SelectFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) ([]sqlplugin.TimerTasksRow, error) {\n\tvar rows []sqlplugin.TimerTasksRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tfilter.MinVisibilityTimestamp = mdb.converter.ToDateTime(filter.MinVisibilityTimestamp)\n\tfilter.MaxVisibilityTimestamp = mdb.converter.ToDateTime(filter.MaxVisibilityTimestamp)\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getTimerTasksQuery, filter.ShardID, filter.MinVisibilityTimestamp,\n\t\tfilter.TaskID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].VisibilityTimestamp = mdb.converter.FromDateTime(rows[i].VisibilityTimestamp)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTimerTasks deletes one row from timer_tasks table\nfunc (mdb *DB) DeleteFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) (sql.Result, error) {\n\tfilter.VisibilityTimestamp = mdb.converter.ToDateTime(filter.VisibilityTimestamp)\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteTimerTaskQuery, filter.ShardID, filter.VisibilityTimestamp, filter.TaskID)\n}\n\n// RangeDeleteFromTimerTasks deletes multi rows from timer_tasks table\nfunc (mdb *DB) RangeDeleteFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) (sql.Result, error) {\n\tfilter.MinVisibilityTimestamp = mdb.converter.ToDateTime(filter.MinVisibilityTimestamp)\n\tfilter.MaxVisibilityTimestamp = mdb.converter.ToDateTime(filter.MaxVisibilityTimestamp)\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskByBatchQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp)\n}\n\n// InsertIntoBufferedEvents inserts one or more rows into buffered_events table\nfunc (mdb *DB) InsertIntoBufferedEvents(ctx context.Context, rows []sqlplugin.BufferedEventsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createBufferedEventsQuery, rows)\n}\n\n// SelectFromBufferedEvents reads one or more rows from buffered_events table\nfunc (mdb *DB) SelectFromBufferedEvents(ctx context.Context, filter *sqlplugin.BufferedEventsFilter) ([]sqlplugin.BufferedEventsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.BufferedEventsRow\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getBufferedEventsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t\trows[i].ShardID = filter.ShardID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromBufferedEvents deletes one or more rows from buffered_events table\nfunc (mdb *DB) DeleteFromBufferedEvents(ctx context.Context, filter *sqlplugin.BufferedEventsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteBufferedEventsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// InsertIntoReplicationTasks inserts one or more rows into replication_tasks table\nfunc (mdb *DB) InsertIntoReplicationTasks(ctx context.Context, rows []sqlplugin.ReplicationTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createReplicationTasksQuery, rows)\n}\n\n// SelectFromReplicationTasks reads one or more rows from replication_tasks table\nfunc (mdb *DB) SelectFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) ([]sqlplugin.ReplicationTasksRow, error) {\n\tvar rows []sqlplugin.ReplicationTasksRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getReplicationTasksQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\treturn rows, err\n}\n\n// DeleteFromReplicationTasks deletes one row from replication_tasks table\nfunc (mdb *DB) DeleteFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteReplicationTaskQuery, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromReplicationTasks deletes multi rows from replication_tasks table\nfunc (mdb *DB) RangeDeleteFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskByBatchQuery, filter.ShardID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskQuery, filter.ShardID, filter.ExclusiveMaxTaskID)\n}\n\n// InsertIntoReplicationTasksDLQ inserts one or more rows into replication_tasks_dlq table\nfunc (mdb *DB) InsertIntoReplicationTasksDLQ(ctx context.Context, row *sqlplugin.ReplicationTaskDLQRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(row.ShardID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, insertReplicationTaskDLQQuery, row)\n}\n\n// SelectFromReplicationTasksDLQ reads one or more rows from replication_tasks_dlq table\nfunc (mdb *DB) SelectFromReplicationTasksDLQ(ctx context.Context, filter *sqlplugin.ReplicationTasksDLQFilter) ([]sqlplugin.ReplicationTasksRow, error) {\n\tvar rows []sqlplugin.ReplicationTasksRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(\n\t\tctx,\n\t\tdbShardID,\n\t\t&rows,\n\t\tgetReplicationTasksDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.InclusiveMinTaskID,\n\t\tfilter.ExclusiveMaxTaskID,\n\t\tfilter.PageSize)\n\treturn rows, err\n}\n\n// SelectFromReplicationDLQ reads one row from replication_tasks_dlq table\nfunc (mdb *DB) SelectFromReplicationDLQ(ctx context.Context, filter *sqlplugin.ReplicationTaskDLQFilter) (int64, error) {\n\tvar size []int64\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif err := mdb.driver.SelectContext(\n\t\tctx,\n\t\tdbShardID,\n\t\t&size,\n\t\tgetReplicationTaskDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n\n// DeleteMessageFromReplicationTasksDLQ deletes one row from replication_tasks_dlq table\nfunc (mdb *DB) DeleteMessageFromReplicationTasksDLQ(\n\tctx context.Context,\n\tfilter *sqlplugin.ReplicationTasksDLQFilter,\n) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\n\treturn mdb.driver.ExecContext(\n\t\tctx,\n\t\tdbShardID,\n\t\tdeleteReplicationTaskFromDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.TaskID,\n\t)\n}\n\n// DeleteMessageFromReplicationTasksDLQ deletes one or more rows from replication_tasks_dlq table\nfunc (mdb *DB) RangeDeleteMessageFromReplicationTasksDLQ(\n\tctx context.Context,\n\tfilter *sqlplugin.ReplicationTasksDLQFilter,\n) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(\n\t\t\tctx,\n\t\t\tdbShardID,\n\t\t\trangeDeleteReplicationTaskFromDLQByBatchQuery,\n\t\t\tfilter.SourceClusterName,\n\t\t\tfilter.ShardID,\n\t\t\tfilter.InclusiveMinTaskID,\n\t\t\tfilter.ExclusiveMaxTaskID,\n\t\t\tfilter.PageSize,\n\t\t)\n\t}\n\n\treturn mdb.driver.ExecContext(\n\t\tctx,\n\t\tdbShardID,\n\t\trangeDeleteReplicationTaskFromDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.InclusiveMinTaskID,\n\t\tfilter.ExclusiveMaxTaskID,\n\t)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/execution_maps.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tdeleteMapQryTemplate = `DELETE FROM %v\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ?`\n\n\t// %[2]v is the columns of the value struct (i.e. no primary key columns), comma separated\n\t// %[3]v should be %[2]v with colons prepended.\n\t// i.e. %[3]v = \",\".join(\":\" + s for s in %[2]v)\n\t// So that this query can be used with BindNamed\n\t// %[4]v should be the name of the key associated with the map\n\t// e.g. for ActivityInfo it is \"schedule_id\"\n\tsetKeyInMapQryTemplate = `REPLACE INTO %[1]v\n(shard_id, domain_id, workflow_id, run_id, %[4]v, %[2]v)\nVALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :%[4]v, %[3]v)`\n\n\t// %[2]v is the name of the key\n\tdeleteKeyInMapQryTemplate = `DELETE FROM %[1]v\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ? AND\n%[2]v IN ( ? )`\n\n\t// %[1]v is the name of the table\n\t// %[2]v is the name of the key\n\t// %[3]v is the value columns, separated by commas\n\tgetMapQryTemplate = `SELECT %[2]v, %[3]v FROM %[1]v\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ?`\n)\n\nfunc stringMap(a []string, f func(string) string) []string {\n\tb := make([]string, len(a))\n\tfor i, v := range a {\n\t\tb[i] = f(v)\n\t}\n\treturn b\n}\n\nfunc makeDeleteMapQry(tableName string) string {\n\treturn fmt.Sprintf(deleteMapQryTemplate, tableName)\n}\n\nfunc makeSetKeyInMapQry(tableName string, nonPrimaryKeyColumns []string, mapKeyName string) string {\n\treturn fmt.Sprintf(setKeyInMapQryTemplate,\n\t\ttableName,\n\t\tstrings.Join(nonPrimaryKeyColumns, \",\"),\n\t\tstrings.Join(stringMap(nonPrimaryKeyColumns, func(x string) string {\n\t\t\treturn \":\" + x\n\t\t}), \",\"),\n\t\tmapKeyName)\n}\n\nfunc makeDeleteKeyInMapQry(tableName string, mapKeyName string) string {\n\treturn fmt.Sprintf(deleteKeyInMapQryTemplate,\n\t\ttableName,\n\t\tmapKeyName)\n}\n\nfunc makeGetMapQryTemplate(tableName string, nonPrimaryKeyColumns []string, mapKeyName string) string {\n\treturn fmt.Sprintf(getMapQryTemplate,\n\t\ttableName,\n\t\tmapKeyName,\n\t\tstrings.Join(nonPrimaryKeyColumns, \",\"))\n}\n\nvar (\n\t// Omit shard_id, run_id, domain_id, workflow_id, schedule_id since they're in the primary key\n\tactivityInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t\t\"last_heartbeat_details\",\n\t\t\"last_heartbeat_updated_time\",\n\t}\n\tactivityInfoTableName = \"activity_info_maps\"\n\tactivityInfoKey       = \"schedule_id\"\n\n\tdeleteActivityInfoMapQry      = makeDeleteMapQry(activityInfoTableName)\n\tsetKeyInActivityInfoMapQry    = makeSetKeyInMapQry(activityInfoTableName, activityInfoColumns, activityInfoKey)\n\tdeleteKeyInActivityInfoMapQry = makeDeleteKeyInMapQry(activityInfoTableName, activityInfoKey)\n\tgetActivityInfoMapQry         = makeGetMapQryTemplate(activityInfoTableName, activityInfoColumns, activityInfoKey)\n)\n\n// ReplaceIntoActivityInfoMaps replaces one or more rows in activity_info_maps table\nfunc (mdb *DB) ReplaceIntoActivityInfoMaps(ctx context.Context, rows []sqlplugin.ActivityInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\tfor i := range rows {\n\t\trows[i].LastHeartbeatUpdatedTime = mdb.converter.ToDateTime(rows[i].LastHeartbeatUpdatedTime)\n\t}\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, setKeyInActivityInfoMapQry, rows)\n}\n\n// SelectFromActivityInfoMaps reads one or more rows from activity_info_maps table\nfunc (mdb *DB) SelectFromActivityInfoMaps(ctx context.Context, filter *sqlplugin.ActivityInfoMapsFilter) ([]sqlplugin.ActivityInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ActivityInfoMapsRow\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t\trows[i].LastHeartbeatUpdatedTime = mdb.converter.FromDateTime(rows[i].LastHeartbeatUpdatedTime)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromActivityInfoMaps deletes one or more rows from activity_info_maps table\nfunc (mdb *DB) DeleteFromActivityInfoMaps(ctx context.Context, filter *sqlplugin.ActivityInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.ScheduleIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.ScheduleIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\ttimerInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\ttimerInfoTableName = \"timer_info_maps\"\n\ttimerInfoKey       = \"timer_id\"\n\n\tdeleteTimerInfoMapSQLQuery      = makeDeleteMapQry(timerInfoTableName)\n\tsetKeyInTimerInfoMapSQLQuery    = makeSetKeyInMapQry(timerInfoTableName, timerInfoColumns, timerInfoKey)\n\tdeleteKeyInTimerInfoMapSQLQuery = makeDeleteKeyInMapQry(timerInfoTableName, timerInfoKey)\n\tgetTimerInfoMapSQLQuery         = makeGetMapQryTemplate(timerInfoTableName, timerInfoColumns, timerInfoKey)\n)\n\n// ReplaceIntoTimerInfoMaps replaces one or more rows in timer_info_maps table\nfunc (mdb *DB) ReplaceIntoTimerInfoMaps(ctx context.Context, rows []sqlplugin.TimerInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, setKeyInTimerInfoMapSQLQuery, rows)\n}\n\n// SelectFromTimerInfoMaps reads one or more rows from timer_info_maps table\nfunc (mdb *DB) SelectFromTimerInfoMaps(ctx context.Context, filter *sqlplugin.TimerInfoMapsFilter) ([]sqlplugin.TimerInfoMapsRow, error) {\n\tvar rows []sqlplugin.TimerInfoMapsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTimerInfoMaps deletes one or more rows from timer_info_maps table\nfunc (mdb *DB) DeleteFromTimerInfoMaps(ctx context.Context, filter *sqlplugin.TimerInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.TimerIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.TimerIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\tchildExecutionInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\tchildExecutionInfoTableName = \"child_execution_info_maps\"\n\tchildExecutionInfoKey       = \"initiated_id\"\n\n\tdeleteChildExecutionInfoMapQry      = makeDeleteMapQry(childExecutionInfoTableName)\n\tsetKeyInChildExecutionInfoMapQry    = makeSetKeyInMapQry(childExecutionInfoTableName, childExecutionInfoColumns, childExecutionInfoKey)\n\tdeleteKeyInChildExecutionInfoMapQry = makeDeleteKeyInMapQry(childExecutionInfoTableName, childExecutionInfoKey)\n\tgetChildExecutionInfoMapQry         = makeGetMapQryTemplate(childExecutionInfoTableName, childExecutionInfoColumns, childExecutionInfoKey)\n)\n\n// ReplaceIntoChildExecutionInfoMaps replaces one or more rows in child_execution_info_maps table\nfunc (mdb *DB) ReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []sqlplugin.ChildExecutionInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, setKeyInChildExecutionInfoMapQry, rows)\n}\n\n// SelectFromChildExecutionInfoMaps reads one or more rows from child_execution_info_maps table\nfunc (mdb *DB) SelectFromChildExecutionInfoMaps(ctx context.Context, filter *sqlplugin.ChildExecutionInfoMapsFilter) ([]sqlplugin.ChildExecutionInfoMapsRow, error) {\n\tvar rows []sqlplugin.ChildExecutionInfoMapsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromChildExecutionInfoMaps deletes one or more rows from child_execution_info_maps table\nfunc (mdb *DB) DeleteFromChildExecutionInfoMaps(ctx context.Context, filter *sqlplugin.ChildExecutionInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\trequestCancelInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\trequestCancelInfoTableName = \"request_cancel_info_maps\"\n\trequestCancelInfoKey       = \"initiated_id\"\n\n\tdeleteRequestCancelInfoMapQry      = makeDeleteMapQry(requestCancelInfoTableName)\n\tsetKeyInRequestCancelInfoMapQry    = makeSetKeyInMapQry(requestCancelInfoTableName, requestCancelInfoColumns, requestCancelInfoKey)\n\tdeleteKeyInRequestCancelInfoMapQry = makeDeleteKeyInMapQry(requestCancelInfoTableName, requestCancelInfoKey)\n\tgetRequestCancelInfoMapQry         = makeGetMapQryTemplate(requestCancelInfoTableName, requestCancelInfoColumns, requestCancelInfoKey)\n)\n\n// ReplaceIntoRequestCancelInfoMaps replaces one or more rows in request_cancel_info_maps table\nfunc (mdb *DB) ReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []sqlplugin.RequestCancelInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, setKeyInRequestCancelInfoMapQry, rows)\n}\n\n// SelectFromRequestCancelInfoMaps reads one or more rows from request_cancel_info_maps table\nfunc (mdb *DB) SelectFromRequestCancelInfoMaps(ctx context.Context, filter *sqlplugin.RequestCancelInfoMapsFilter) ([]sqlplugin.RequestCancelInfoMapsRow, error) {\n\tvar rows []sqlplugin.RequestCancelInfoMapsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromRequestCancelInfoMaps deletes one or more rows from request_cancel_info_maps table\nfunc (mdb *DB) DeleteFromRequestCancelInfoMaps(ctx context.Context, filter *sqlplugin.RequestCancelInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\tsignalInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\tsignalInfoTableName = \"signal_info_maps\"\n\tsignalInfoKey       = \"initiated_id\"\n\n\tdeleteSignalInfoMapQry      = makeDeleteMapQry(signalInfoTableName)\n\tsetKeyInSignalInfoMapQry    = makeSetKeyInMapQry(signalInfoTableName, signalInfoColumns, signalInfoKey)\n\tdeleteKeyInSignalInfoMapQry = makeDeleteKeyInMapQry(signalInfoTableName, signalInfoKey)\n\tgetSignalInfoMapQry         = makeGetMapQryTemplate(signalInfoTableName, signalInfoColumns, signalInfoKey)\n)\n\n// ReplaceIntoSignalInfoMaps replaces one or more rows in signal_info_maps table\nfunc (mdb *DB) ReplaceIntoSignalInfoMaps(ctx context.Context, rows []sqlplugin.SignalInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, setKeyInSignalInfoMapQry, rows)\n}\n\n// SelectFromSignalInfoMaps reads one or more rows from signal_info_maps table\nfunc (mdb *DB) SelectFromSignalInfoMaps(ctx context.Context, filter *sqlplugin.SignalInfoMapsFilter) ([]sqlplugin.SignalInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.SignalInfoMapsRow\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromSignalInfoMaps deletes one or more rows from signal_info_maps table\nfunc (mdb *DB) DeleteFromSignalInfoMaps(ctx context.Context, filter *sqlplugin.SignalInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nconst (\n\tdeleteAllSignalsRequestedSetQry = `DELETE FROM signals_requested_sets\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ?\n`\n\n\tcreateSignalsRequestedSetQry = `INSERT IGNORE INTO signals_requested_sets\n(shard_id, domain_id, workflow_id, run_id, signal_id) VALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :signal_id)`\n\n\tdeleteSignalsRequestedSetQry = `DELETE FROM signals_requested_sets\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ? AND\nsignal_id IN ( ? )`\n\n\tgetSignalsRequestedSetQry = `SELECT signal_id FROM signals_requested_sets WHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ?`\n)\n\n// InsertIntoSignalsRequestedSets inserts one or more rows into signals_requested_sets table\nfunc (mdb *DB) InsertIntoSignalsRequestedSets(ctx context.Context, rows []sqlplugin.SignalsRequestedSetsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createSignalsRequestedSetQry, rows)\n}\n\n// SelectFromSignalsRequestedSets reads one or more rows from signals_requested_sets table\nfunc (mdb *DB) SelectFromSignalsRequestedSets(ctx context.Context, filter *sqlplugin.SignalsRequestedSetsFilter) ([]sqlplugin.SignalsRequestedSetsRow, error) {\n\tvar rows []sqlplugin.SignalsRequestedSetsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, getSignalsRequestedSetQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromSignalsRequestedSets deletes one or more rows from signals_requested_sets table\nfunc (mdb *DB) DeleteFromSignalsRequestedSets(ctx context.Context, filter *sqlplugin.SignalsRequestedSetsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tif len(filter.SignalIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteSignalsRequestedSetQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.SignalIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteAllSignalsRequestedSetQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/plugin.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"bytes\"\n\t\"crypto/tls\"\n\t\"crypto/x509\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"net\"\n\t\"net/url\"\n\t\"strings\"\n\n\t\"github.com/go-sql-driver/mysql\"\n\t\"github.com/iancoleman/strcase\"\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/environment\"\n)\n\nconst (\n\t// PluginName is the name of the plugin\n\tPluginName                   = \"mysql\"\n\tdsnFmt                       = \"%s:%s@%v(%v)/%s\"\n\tisolationLevelAttrName       = \"transaction_isolation\"\n\tisolationLevelAttrNameLegacy = \"tx_isolation\"\n\tdefaultIsolationLevel        = \"'READ-COMMITTED'\"\n\t// customTLSName is the name used if a custom tls configuration is created\n\tcustomTLSName = \"tls-custom\"\n)\n\nvar dsnAttrOverrides = map[string]string{\n\t\"parseTime\":       \"true\",\n\t\"clientFoundRows\": \"true\",\n\t\"multiStatements\": \"true\",\n}\n\ntype plugin struct{}\n\nvar _ sqlplugin.Plugin = (*plugin)(nil)\n\nfunc init() {\n\tsql.RegisterPlugin(PluginName, &plugin{})\n}\n\n// CreateDB initialize the DB object\nfunc (p *plugin) CreateDB(cfg *config.SQL) (sqlplugin.DB, error) {\n\treturn p.createDB(cfg)\n}\n\n// CreateAdminDB initialize the adminDb object\nfunc (p *plugin) CreateAdminDB(cfg *config.SQL) (sqlplugin.AdminDB, error) {\n\treturn p.createDB(cfg)\n}\n\nfunc (p *plugin) createDB(cfg *config.SQL) (*DB, error) {\n\tconns, err := sqldriver.CreateDBConnections(cfg, func(cfg *config.SQL) (*sqlx.DB, error) {\n\t\treturn p.createSingleDBConn(cfg)\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewDB(conns, nil, sqlplugin.DbShardUndefined, cfg.NumShards, newConverter())\n}\n\nfunc (p *plugin) createSingleDBConn(cfg *config.SQL) (*sqlx.DB, error) {\n\terr := registerTLSConfig(cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdb, err := sqlx.Connect(PluginName, buildDSN(cfg))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cfg.MaxConns > 0 {\n\t\tdb.SetMaxOpenConns(cfg.MaxConns)\n\t}\n\tif cfg.MaxIdleConns > 0 {\n\t\tdb.SetMaxIdleConns(cfg.MaxIdleConns)\n\t}\n\tif cfg.MaxConnLifetime > 0 {\n\t\tdb.SetConnMaxLifetime(cfg.MaxConnLifetime)\n\t}\n\n\t// Maps struct names in CamelCase to snake without need for DB struct tags.\n\tdb.MapperFunc(strcase.ToSnake)\n\treturn db, nil\n}\n\nfunc registerTLSConfig(cfg *config.SQL) error {\n\tif cfg.TLS == nil || !cfg.TLS.Enabled {\n\t\treturn nil\n\t}\n\n\thost, _, err := net.SplitHostPort(cfg.ConnectAddr)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"error in host port from ConnectAddr: %v\", err)\n\t}\n\n\t// TODO: create a way to set MinVersion and CipherSuites via cfg.\n\ttlsConfig := &tls.Config{\n\t\tServerName:         host,\n\t\tInsecureSkipVerify: !cfg.TLS.EnableHostVerification,\n\t}\n\n\tif cfg.TLS.CaFile != \"\" {\n\t\trootCertPool := x509.NewCertPool()\n\t\tpem, err := ioutil.ReadFile(cfg.TLS.CaFile)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to load CA files: %v\", err)\n\t\t}\n\t\tif ok := rootCertPool.AppendCertsFromPEM(pem); !ok {\n\t\t\treturn fmt.Errorf(\"failed to append CA file\")\n\t\t}\n\t\ttlsConfig.RootCAs = rootCertPool\n\t}\n\n\tif cfg.TLS.CertFile != \"\" && cfg.TLS.KeyFile != \"\" {\n\t\tclientCert := make([]tls.Certificate, 0, 1)\n\t\tcerts, err := tls.LoadX509KeyPair(\n\t\t\tcfg.TLS.CertFile,\n\t\t\tcfg.TLS.KeyFile,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to load tls x509 key pair: %v\", err)\n\t\t}\n\t\tclientCert = append(clientCert, certs)\n\t\ttlsConfig.Certificates = clientCert\n\t}\n\n\t// In order to use the TLS configuration you need to register it. Once registered you use it by specifying\n\t// `tls` in the connect attributes.\n\terr = mysql.RegisterTLSConfig(customTLSName, tlsConfig)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to register tls config: %v\", err)\n\t}\n\n\tif cfg.ConnectAttributes == nil {\n\t\tcfg.ConnectAttributes = map[string]string{}\n\t}\n\n\t// If no `tls` connect attribute is provided then we override it to our newly registered tls config automatically.\n\t// This allows users to simply provide a tls config without needing to remember to also set the connect attribute\n\tif cfg.ConnectAttributes[\"tls\"] == \"\" {\n\t\tif cfg.TLS.SSLMode != \"\" {\n\t\t\tcfg.ConnectAttributes[\"tls\"] = cfg.TLS.SSLMode\n\t\t} else {\n\t\t\tcfg.ConnectAttributes[\"tls\"] = customTLSName\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc buildDSN(cfg *config.SQL) string {\n\tattrs := buildDSNAttrs(cfg)\n\tdsn := fmt.Sprintf(dsnFmt, cfg.User, cfg.Password, cfg.ConnectProtocol, cfg.ConnectAddr, cfg.DatabaseName)\n\tif attrs != \"\" {\n\t\tdsn = dsn + \"?\" + attrs\n\t}\n\treturn dsn\n}\n\nfunc buildDSNAttrs(cfg *config.SQL) string {\n\tattrs := make(map[string]string, len(dsnAttrOverrides)+len(cfg.ConnectAttributes)+1)\n\tfor k, v := range cfg.ConnectAttributes {\n\t\tk1, v1 := sanitizeAttr(k, v)\n\t\tattrs[k1] = v1\n\t}\n\n\t// only override isolation level if not specified\n\tif !hasAttr(attrs, isolationLevelAttrName) &&\n\t\t!hasAttr(attrs, isolationLevelAttrNameLegacy) {\n\t\tattrs[isolationLevelAttrName] = defaultIsolationLevel\n\t}\n\n\t// these attrs are always overriden\n\tfor k, v := range dsnAttrOverrides {\n\t\tattrs[k] = v\n\t}\n\n\tfirst := true\n\tvar buf bytes.Buffer\n\tfor k, v := range attrs {\n\t\tif !first {\n\t\t\tbuf.WriteString(\"&\")\n\t\t}\n\t\tfirst = false\n\t\tbuf.WriteString(k)\n\t\tbuf.WriteString(\"=\")\n\t\tbuf.WriteString(v)\n\t}\n\treturn url.PathEscape(buf.String())\n}\n\nfunc hasAttr(attrs map[string]string, key string) bool {\n\t_, ok := attrs[key]\n\treturn ok\n}\n\nfunc sanitizeAttr(inkey string, invalue string) (string, string) {\n\tkey := strings.ToLower(strings.TrimSpace(inkey))\n\tvalue := strings.ToLower(strings.TrimSpace(invalue))\n\tswitch key {\n\tcase isolationLevelAttrName, isolationLevelAttrNameLegacy:\n\t\tif value[0] != '\\'' { // mysql sys variable values must be enclosed in single quotes\n\t\t\tvalue = \"'\" + value + \"'\"\n\t\t}\n\t\treturn key, value\n\tdefault:\n\t\treturn inkey, invalue\n\t}\n}\n\nconst (\n\ttestSchemaDir = \"schema/mysql/v8\"\n)\n\n// GetTestClusterOption return test options\nfunc GetTestClusterOption() (*pt.TestBaseOptions, error) {\n\tport, err := environment.GetMySQLPort()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &pt.TestBaseOptions{\n\t\tDBPluginName: PluginName,\n\t\tDBUsername:   environment.GetMySQLUser(),\n\t\tDBPassword:   environment.GetMySQLPassword(),\n\t\tDBHost:       environment.GetMySQLAddress(),\n\t\tDBPort:       port,\n\t\tSchemaDir:    testSchemaDir,\n\t\tStoreType:    config.StoreTypeSQL,\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/queue.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateEnqueueMessageQuery            = `INSERT INTO queue (queue_type, message_id, message_payload) VALUES(:queue_type, :message_id, :message_payload)`\n\ttemplateGetLastMessageIDQuery          = `SELECT message_id FROM queue WHERE queue_type=? ORDER BY message_id DESC LIMIT 1 FOR UPDATE`\n\ttemplateGetMessagesQuery               = `SELECT message_id, message_payload FROM queue WHERE queue_type = ? and message_id > ? ORDER BY message_id ASC LIMIT ?`\n\ttemplateGetMessagesBetweenQuery        = `SELECT message_id, message_payload FROM queue WHERE queue_type = ? and message_id > ? and message_id <= ? ORDER BY message_id ASC LIMIT ?`\n\ttemplateDeleteMessagesBeforeQuery      = `DELETE FROM queue WHERE queue_type = ? and message_id < ?`\n\ttemplateRangeDeleteMessagesQuery       = `DELETE FROM queue WHERE queue_type = ? and message_id > ? and message_id <= ?`\n\ttemplateDeleteMessageQuery             = `DELETE FROM queue WHERE queue_type = ? and message_id = ?`\n\ttemplateGetQueueMetadataQuery          = `SELECT data from queue_metadata WHERE queue_type = ?`\n\ttemplateGetQueueMetadataForUpdateQuery = templateGetQueueMetadataQuery + ` FOR UPDATE`\n\ttemplateInsertQueueMetadataQuery       = `INSERT INTO queue_metadata (queue_type, data) VALUES(:queue_type, :data)`\n\ttemplateUpdateQueueMetadataQuery       = `UPDATE queue_metadata SET data = ? WHERE queue_type = ?`\n\ttemplateGetQueueSizeQuery              = `SELECT COUNT(1) AS count FROM queue WHERE queue_type=?`\n)\n\n// InsertIntoQueue inserts a new row into queue table\nfunc (mdb *DB) InsertIntoQueue(\n\tctx context.Context,\n\trow *sqlplugin.QueueRow,\n) (sql.Result, error) {\n\n\treturn mdb.driver.NamedExecContext(ctx, sqlplugin.DbDefaultShard, templateEnqueueMessageQuery, row)\n}\n\n// GetLastEnqueuedMessageIDForUpdate returns the last enqueued message ID\nfunc (mdb *DB) GetLastEnqueuedMessageIDForUpdate(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tvar lastMessageID int64\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &lastMessageID, templateGetLastMessageIDQuery, queueType)\n\treturn lastMessageID, err\n}\n\n// GetMessagesFromQueue retrieves messages from the queue\nfunc (mdb *DB) GetMessagesFromQueue(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tlastMessageID int64,\n\tmaxRows int,\n) ([]sqlplugin.QueueRow, error) {\n\n\tvar rows []sqlplugin.QueueRow\n\terr := mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, templateGetMessagesQuery, queueType, lastMessageID, maxRows)\n\treturn rows, err\n}\n\n// GetMessagesBetween retrieves messages from the queue\nfunc (mdb *DB) GetMessagesBetween(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tfirstMessageID int64,\n\tlastMessageID int64,\n\tmaxRows int,\n) ([]sqlplugin.QueueRow, error) {\n\n\tvar rows []sqlplugin.QueueRow\n\terr := mdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, templateGetMessagesBetweenQuery, queueType, firstMessageID, lastMessageID, maxRows)\n\treturn rows, err\n}\n\n// DeleteMessagesBefore deletes messages before messageID from the queue\nfunc (mdb *DB) DeleteMessagesBefore(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n) (sql.Result, error) {\n\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateDeleteMessagesBeforeQuery, queueType, messageID)\n}\n\n// RangeDeleteMessages deletes messages before messageID from the queue\nfunc (mdb *DB) RangeDeleteMessages(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\texclusiveBeginMessageID int64,\n\tinclusiveEndMessageID int64,\n) (sql.Result, error) {\n\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateRangeDeleteMessagesQuery, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// DeleteMessage deletes message with a messageID from the queue\nfunc (mdb *DB) DeleteMessage(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n) (sql.Result, error) {\n\n\treturn mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateDeleteMessageQuery, queueType, messageID)\n}\n\n// InsertAckLevel inserts ack level\nfunc (mdb *DB) InsertAckLevel(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tmessageID int64,\n\tclusterName string,\n) error {\n\n\tclusterAckLevels := map[string]int64{clusterName: messageID}\n\tdata, err := json.Marshal(clusterAckLevels)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = mdb.driver.NamedExecContext(ctx, sqlplugin.DbDefaultShard, templateInsertQueueMetadataQuery, sqlplugin.QueueMetadataRow{QueueType: queueType, Data: data})\n\treturn err\n\n}\n\n// UpdateAckLevels updates cluster ack levels\nfunc (mdb *DB) UpdateAckLevels(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tclusterAckLevels map[string]int64,\n) error {\n\n\tdata, err := json.Marshal(clusterAckLevels)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = mdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateUpdateQueueMetadataQuery, data, queueType)\n\treturn err\n}\n\n// GetAckLevels returns ack levels for pulling clusters\nfunc (mdb *DB) GetAckLevels(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tforUpdate bool,\n) (map[string]int64, error) {\n\n\tqueryStr := templateGetQueueMetadataQuery\n\tif forUpdate {\n\t\tqueryStr = templateGetQueueMetadataForUpdateQuery\n\t}\n\n\tvar data []byte\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &data, queryStr, queueType)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\n\tvar clusterAckLevels map[string]int64\n\tif err := json.Unmarshal(data, &clusterAckLevels); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn clusterAckLevels, nil\n}\n\n// GetQueueSize returns the queue size\nfunc (mdb *DB) GetQueueSize(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tvar size []int64\n\tif err := mdb.driver.SelectContext(\n\t\tctx,\n\t\tsqlplugin.DbDefaultShard,\n\t\t&size,\n\t\ttemplateGetQueueSizeQuery,\n\t\tqueueType,\n\t); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/shard.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tcreateShardQry = `INSERT INTO\n shards (shard_id, range_id, data, data_encoding) VALUES (?, ?, ?, ?)`\n\n\tgetShardQry = `SELECT\n shard_id, range_id, data, data_encoding\n FROM shards WHERE shard_id = ?`\n\n\tupdateShardQry = `UPDATE shards \n SET range_id = ?, data = ?, data_encoding = ? \n WHERE shard_id = ?`\n\n\tlockShardQry     = `SELECT range_id FROM shards WHERE shard_id = ? FOR UPDATE`\n\treadLockShardQry = `SELECT range_id FROM shards WHERE shard_id = ? LOCK IN SHARE MODE`\n)\n\n// InsertIntoShards inserts one or more rows into shards table\nfunc (mdb *DB) InsertIntoShards(ctx context.Context, row *sqlplugin.ShardsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, createShardQry, row.ShardID, row.RangeID, row.Data, row.DataEncoding)\n}\n\n// UpdateShards updates one or more rows into shards table\nfunc (mdb *DB) UpdateShards(ctx context.Context, row *sqlplugin.ShardsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, updateShardQry, row.RangeID, row.Data, row.DataEncoding, row.ShardID)\n}\n\n// SelectFromShards reads one or more rows from shards table\nfunc (mdb *DB) SelectFromShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (*sqlplugin.ShardsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar row sqlplugin.ShardsRow\n\terr := mdb.driver.GetContext(ctx, dbShardID, &row, getShardQry, filter.ShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &row, err\n}\n\n// ReadLockShards acquires a read lock on a single row in shards table\nfunc (mdb *DB) ReadLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := mdb.driver.GetContext(ctx, dbShardID, &rangeID, readLockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n\n// WriteLockShards acquires a write lock on a single row in shards table\nfunc (mdb *DB) WriteLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := mdb.driver.GetContext(ctx, dbShardID, &rangeID, lockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/task.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttaskListCreatePart = `INTO task_lists(shard_id, domain_id, name, task_type, range_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :domain_id, :name, :task_type, :range_id, :data, :data_encoding)`\n\n\t// (default range ID: initialRangeID == 1)\n\tcreateTaskListQry = `INSERT ` + taskListCreatePart\n\n\tupdateTaskListQry = `UPDATE task_lists SET\nrange_id = :range_id,\ndata = :data,\ndata_encoding = :data_encoding\nWHERE\nshard_id = :shard_id AND\ndomain_id = :domain_id AND\nname = :name AND\ntask_type = :task_type\n`\n\n\t// This query uses pagination that is best understood by analogy to simple numbers.\n\t// Given a list of numbers\n\t// \t111\n\t//\t113\n\t//\t122\n\t//\t211\n\t// where the hundreds digit corresponds to domain_id, the tens digit\n\t// corresponds to name, and the ones digit corresponds to task_type,\n\t// Imagine recurring queries with a limit of 1.\n\t// For the second query to skip the first result and return 112, it must allow equal values in hundreds & tens, but it's OK because the ones digit is higher.\n\t// For the third query, the ones digit is now lower but that's irrelevant because the tens digit is greater.\n\t// For the fourth query, both the tens digit and ones digit are now lower but that's again irrelevant because now the hundreds digit is higher.\n\t// This technique is useful since the size of the table can easily change between calls, making SKIP an unreliable method, while other db-specific things like rowids are not portable\n\tlistTaskListQry = `SELECT domain_id, range_id, name, task_type, data, data_encoding ` +\n\t\t`FROM task_lists ` +\n\t\t`WHERE shard_id = ? AND ((domain_id = ? AND name = ? AND task_type > ?) OR (domain_id=? AND name > ?) OR (domain_id > ?)) ` +\n\t\t`ORDER BY domain_id,name,task_type LIMIT ?`\n\n\tgetTaskListQry = `SELECT domain_id, range_id, name, task_type, data, data_encoding ` +\n\t\t`FROM task_lists ` +\n\t\t`WHERE shard_id = ? AND domain_id = ? AND name = ? AND task_type = ?`\n\n\tdeleteTaskListQry = `DELETE FROM task_lists WHERE shard_id=? AND domain_id=? AND name=? AND task_type=? AND range_id=?`\n\n\tlockTaskListQry = `SELECT range_id FROM task_lists ` +\n\t\t`WHERE shard_id = ? AND domain_id = ? AND name = ? AND task_type = ? FOR UPDATE`\n\n\tgetTaskMinMaxQry = `SELECT task_id, data, data_encoding ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id > ? AND task_id <= ? ` +\n\t\t` ORDER BY task_id LIMIT ?`\n\n\tgetTaskMinQry = `SELECT task_id, data, data_encoding ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id > ? ORDER BY task_id LIMIT ?`\n\n\tgetTasksCountQry = `SELECT count(1) as count ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id > ?`\n\n\tcreateTaskQry = `INSERT INTO ` +\n\t\t`tasks(domain_id, task_list_name, task_type, task_id, data, data_encoding) ` +\n\t\t`VALUES(:domain_id, :task_list_name, :task_type, :task_id, :data, :data_encoding)`\n\n\tdeleteTaskQry = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id = ?`\n\n\trangeDeleteTaskQry = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id <= ? ` +\n\t\t`ORDER BY domain_id,task_list_name,task_type,task_id LIMIT ?`\n\n\tgetOrphanTaskQry = `SELECT task_id, domain_id, task_list_name, task_type FROM tasks AS t ` +\n\t\t`WHERE NOT EXISTS ( ` +\n\t\t`\tSELECT domain_id, name, task_type FROM task_lists AS tl ` +\n\t\t`\tWHERE t.domain_id=tl.domain_id and t.task_list_name=tl.name and t.task_type=tl.task_type ` +\n\t\t`) LIMIT ?;`\n)\n\n// InsertIntoTasks inserts one or more rows into tasks table\nfunc (mdb *DB) InsertIntoTasks(ctx context.Context, rows []sqlplugin.TasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\treturn mdb.driver.NamedExecContext(ctx, rows[0].ShardID, createTaskQry, rows)\n}\n\n// SelectFromTasks reads one or more rows from tasks table\nfunc (mdb *DB) SelectFromTasks(ctx context.Context, filter *sqlplugin.TasksFilter) ([]sqlplugin.TasksRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.TasksRow\n\tswitch {\n\tcase filter.MaxTaskID != nil:\n\t\terr = mdb.driver.SelectContext(ctx, filter.ShardID, &rows, getTaskMinMaxQry, filter.DomainID,\n\t\t\tfilter.TaskListName, filter.TaskType, *filter.MinTaskID, *filter.MaxTaskID, *filter.PageSize)\n\tdefault:\n\t\terr = mdb.driver.SelectContext(ctx, filter.ShardID, &rows, getTaskMinQry, filter.DomainID,\n\t\t\tfilter.TaskListName, filter.TaskType, *filter.MinTaskID, *filter.PageSize)\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTasks deletes one or more rows from tasks table\nfunc (mdb *DB) DeleteFromTasks(ctx context.Context, filter *sqlplugin.TasksFilter) (sql.Result, error) {\n\tif filter.TaskIDLessThanEquals != nil {\n\t\tif filter.Limit == nil || *filter.Limit == 0 {\n\t\t\treturn nil, fmt.Errorf(\"missing limit parameter\")\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, filter.ShardID, rangeDeleteTaskQry,\n\t\t\tfilter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskIDLessThanEquals, *filter.Limit)\n\t}\n\treturn mdb.driver.ExecContext(ctx, filter.ShardID, deleteTaskQry, filter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskID)\n}\n\nfunc (mdb *DB) GetOrphanTasks(ctx context.Context, filter *sqlplugin.OrphanTasksFilter) ([]sqlplugin.TaskKeyRow, error) {\n\tif filter.Limit == nil || *filter.Limit == 0 {\n\t\treturn nil, fmt.Errorf(\"missing limit parameter\")\n\t}\n\tvar rows []sqlplugin.TaskKeyRow\n\n\terr := mdb.driver.SelectContext(ctx, sqlplugin.DbAllShards, &rows, getOrphanTaskQry, *filter.Limit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, nil\n}\n\n// InsertIntoTaskLists inserts one or more rows into task_lists table\nfunc (mdb *DB) InsertIntoTaskLists(ctx context.Context, row *sqlplugin.TaskListsRow) (sql.Result, error) {\n\treturn mdb.driver.NamedExecContext(ctx, row.ShardID, createTaskListQry, row)\n}\n\n// UpdateTaskLists updates a row in task_lists table\nfunc (mdb *DB) UpdateTaskLists(ctx context.Context, row *sqlplugin.TaskListsRow) (sql.Result, error) {\n\treturn mdb.driver.NamedExecContext(ctx, row.ShardID, updateTaskListQry, row)\n}\n\n// SelectFromTaskLists reads one or more rows from task_lists table\nfunc (mdb *DB) SelectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tswitch {\n\tcase filter.DomainID != nil && filter.Name != nil && filter.TaskType != nil:\n\t\treturn mdb.selectFromTaskLists(ctx, filter)\n\tcase filter.DomainIDGreaterThan != nil && filter.NameGreaterThan != nil && filter.TaskTypeGreaterThan != nil && filter.PageSize != nil:\n\t\treturn mdb.rangeSelectFromTaskLists(ctx, filter)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid set of query filter params\")\n\t}\n}\n\nfunc (mdb *DB) selectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tvar err error\n\tvar row sqlplugin.TaskListsRow\n\terr = mdb.driver.GetContext(ctx, filter.ShardID, &row, getTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn []sqlplugin.TaskListsRow{row}, err\n}\n\nfunc (mdb *DB) rangeSelectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.TaskListsRow\n\terr = mdb.driver.SelectContext(ctx, filter.ShardID, &rows, listTaskListQry,\n\t\tfilter.ShardID, *filter.DomainIDGreaterThan, *filter.NameGreaterThan, *filter.TaskTypeGreaterThan, *filter.DomainIDGreaterThan, *filter.NameGreaterThan, *filter.DomainIDGreaterThan, *filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].ShardID = filter.ShardID\n\t}\n\treturn rows, nil\n}\n\n// DeleteFromTaskLists deletes a row from task_lists table\nfunc (mdb *DB) DeleteFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) (sql.Result, error) {\n\treturn mdb.driver.ExecContext(ctx, filter.ShardID, deleteTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType, *filter.RangeID)\n}\n\n// LockTaskLists locks a row in task_lists table\nfunc (mdb *DB) LockTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) (int64, error) {\n\tvar rangeID int64\n\terr := mdb.driver.GetContext(ctx, filter.ShardID, &rangeID, lockTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType)\n\treturn rangeID, err\n}\n\nfunc (mdb *DB) GetTasksCount(ctx context.Context, filter *sqlplugin.TasksFilter) (int64, error) {\n\tvar size []int64\n\tif err := mdb.driver.SelectContext(ctx, filter.ShardID, &size, getTasksCountQry, filter.DomainID, filter.TaskListName, filter.TaskType, *filter.MinTaskID); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n\n// InsertIntoTasksWithTTL is not supported in MySQL\nfunc (mdb *DB) InsertIntoTasksWithTTL(_ context.Context, _ []sqlplugin.TasksRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// InsertIntoTaskListsWithTTL is not supported in MySQL\nfunc (mdb *DB) InsertIntoTaskListsWithTTL(_ context.Context, _ *sqlplugin.TaskListsRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// UpdateTaskListsWithTTL is not supported in MySQL\nfunc (mdb *DB) UpdateTaskListsWithTTL(_ context.Context, _ *sqlplugin.TaskListsRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/typeconv.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport \"time\"\n\n// DataConverter defines the API for conversions to/from\n// go types to mysql datatypes\ntype DataConverter interface {\n\tToDateTime(t time.Time) time.Time\n\tFromDateTime(t time.Time) time.Time\n}\n\ntype converter struct{}\n\nfunc newConverter() *converter {\n\treturn &converter{}\n}\n\nvar minMySQLDateTime = getMinMySQLDateTime()\n\n// ToDateTime converts to time to MySQL datetime\nfunc (c *converter) ToDateTime(t time.Time) time.Time {\n\tif t.IsZero() {\n\t\treturn minMySQLDateTime\n\t}\n\treturn t\n}\n\n// FromDateTime converts mysql datetime and returns go time\nfunc (c *converter) FromDateTime(t time.Time) time.Time {\n\tif t.Equal(minMySQLDateTime) {\n\t\treturn time.Time{}\n\t}\n\treturn t\n}\n\nfunc getMinMySQLDateTime() time.Time {\n\tt, err := time.Parse(time.RFC3339, \"1000-01-01T00:00:00Z\")\n\tif err != nil {\n\t\treturn time.Unix(0, 0)\n\t}\n\treturn t\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/mysql/visibility.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateCreateWorkflowExecutionStarted = `INSERT IGNORE INTO executions_visibility (` +\n\t\t`domain_id, workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time) ` +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\ttemplateCreateWorkflowExecutionClosed = `REPLACE INTO executions_visibility (` +\n\t\t`domain_id, workflow_id, run_id, start_time, execution_time, workflow_type_name, close_time, close_status, history_length, memo, encoding, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time) ` +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n\n\t// RunID condition is needed for correct pagination\n\ttemplateConditions = ` AND domain_id = ?\n\t\t AND start_time >= ?\n\t\t AND start_time <= ?\n \t\t AND (run_id > ? OR start_time < ?)\n         ORDER BY start_time DESC, run_id\n         LIMIT ?`\n\n\ttemplateOpenFieldNames = `workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, is_cron, update_time, shard_id`\n\ttemplateOpenSelect     = `SELECT ` + templateOpenFieldNames + ` FROM executions_visibility WHERE close_status IS NULL `\n\n\ttemplateClosedSelect = `SELECT ` + templateOpenFieldNames + `, close_time, close_status, history_length\n\t\t FROM executions_visibility WHERE close_status IS NOT NULL `\n\n\ttemplateGetOpenWorkflowExecutions = templateOpenSelect + templateConditions\n\n\ttemplateGetClosedWorkflowExecutions = templateClosedSelect + templateConditions\n\n\ttemplateGetOpenWorkflowExecutionsByType = templateOpenSelect + `AND workflow_type_name = ?` + templateConditions\n\n\ttemplateGetClosedWorkflowExecutionsByType = templateClosedSelect + `AND workflow_type_name = ?` + templateConditions\n\n\ttemplateGetOpenWorkflowExecutionsByID = templateOpenSelect + `AND workflow_id = ?` + templateConditions\n\n\ttemplateGetClosedWorkflowExecutionsByID = templateClosedSelect + `AND workflow_id = ?` + templateConditions\n\n\ttemplateGetClosedWorkflowExecutionsByStatus = templateClosedSelect + `AND close_status = ?` + templateConditions\n\n\ttemplateGetClosedWorkflowExecution = `SELECT workflow_id, run_id, start_time, execution_time, memo, encoding, close_time, workflow_type_name, close_status, history_length, is_cron, update_time, shard_id\n\t\t FROM executions_visibility\n\t\t WHERE domain_id = ? AND close_status IS NOT NULL\n\t\t AND run_id = ?`\n\n\ttemplateDeleteWorkflowExecution = \"DELETE FROM executions_visibility WHERE domain_id=? AND run_id=?\"\n)\n\nvar errCloseParams = errors.New(\"missing one of {closeStatus, closeTime, historyLength} params\")\n\n// InsertIntoVisibility inserts a row into visibility table. If an row already exist,\n// its left as such and no update will be made\nfunc (mdb *DB) InsertIntoVisibility(ctx context.Context, row *sqlplugin.VisibilityRow) (sql.Result, error) {\n\trow.StartTime = mdb.converter.ToDateTime(row.StartTime)\n\tscheduledExecutionTime := mdb.converter.ToDateTime(row.ScheduledExecutionTime)\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(row.DomainID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx,\n\t\tdbShardID,\n\t\ttemplateCreateWorkflowExecutionStarted,\n\t\trow.DomainID,\n\t\trow.WorkflowID,\n\t\trow.RunID,\n\t\trow.StartTime,\n\t\trow.ExecutionTime,\n\t\trow.WorkflowTypeName,\n\t\trow.Memo,\n\t\trow.Encoding,\n\t\trow.IsCron,\n\t\trow.NumClusters,\n\t\trow.UpdateTime,\n\t\trow.ShardID,\n\t\trow.ExecutionStatus,\n\t\trow.CronSchedule,\n\t\tscheduledExecutionTime)\n}\n\n// ReplaceIntoVisibility replaces an existing row if it exist or creates a new row in visibility table\nfunc (mdb *DB) ReplaceIntoVisibility(ctx context.Context, row *sqlplugin.VisibilityRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(row.DomainID, mdb.GetTotalNumDBShards())\n\tswitch {\n\tcase row.CloseStatus != nil && row.CloseTime != nil && row.HistoryLength != nil:\n\t\trow.StartTime = mdb.converter.ToDateTime(row.StartTime)\n\t\tcloseTime := mdb.converter.ToDateTime(*row.CloseTime)\n\t\tscheduledExecutionTime := mdb.converter.ToDateTime(row.ScheduledExecutionTime)\n\t\treturn mdb.driver.ExecContext(ctx,\n\t\t\tdbShardID,\n\t\t\ttemplateCreateWorkflowExecutionClosed,\n\t\t\trow.DomainID,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\trow.StartTime,\n\t\t\trow.ExecutionTime,\n\t\t\trow.WorkflowTypeName,\n\t\t\tcloseTime,\n\t\t\t*row.CloseStatus,\n\t\t\t*row.HistoryLength,\n\t\t\trow.Memo,\n\t\t\trow.Encoding,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tscheduledExecutionTime)\n\tdefault:\n\t\treturn nil, errCloseParams\n\t}\n}\n\n// DeleteFromVisibility deletes a row from visibility table if it exist\nfunc (mdb *DB) DeleteFromVisibility(ctx context.Context, filter *sqlplugin.VisibilityFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(filter.DomainID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, templateDeleteWorkflowExecution, filter.DomainID, filter.RunID)\n}\n\n// SelectFromVisibility reads one or more rows from visibility table\nfunc (mdb *DB) SelectFromVisibility(ctx context.Context, filter *sqlplugin.VisibilityFilter) ([]sqlplugin.VisibilityRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(filter.DomainID, mdb.GetTotalNumDBShards())\n\tvar err error\n\tvar rows []sqlplugin.VisibilityRow\n\tif filter.MinStartTime != nil {\n\t\t*filter.MinStartTime = mdb.converter.ToDateTime(*filter.MinStartTime)\n\t}\n\tif filter.MaxStartTime != nil {\n\t\t*filter.MaxStartTime = mdb.converter.ToDateTime(*filter.MaxStartTime)\n\t}\n\tswitch {\n\tcase filter.MinStartTime == nil && filter.RunID != nil && filter.Closed:\n\t\tvar row sqlplugin.VisibilityRow\n\t\terr = mdb.driver.GetContext(ctx, dbShardID, &row, templateGetClosedWorkflowExecution, filter.DomainID, *filter.RunID)\n\t\tif err == nil {\n\t\t\trows = append(rows, row)\n\t\t}\n\tcase filter.MinStartTime != nil && filter.WorkflowID != nil:\n\t\tqry := templateGetOpenWorkflowExecutionsByID\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutionsByID\n\t\t}\n\t\terr = mdb.driver.SelectContext(ctx,\n\t\t\tdbShardID,\n\t\t\t&rows,\n\t\t\tqry,\n\t\t\t*filter.WorkflowID,\n\t\t\tfilter.DomainID,\n\t\t\tmdb.converter.ToDateTime(*filter.MinStartTime),\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\t*filter.MinStartTime,\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil && filter.WorkflowTypeName != nil:\n\t\tqry := templateGetOpenWorkflowExecutionsByType\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutionsByType\n\t\t}\n\t\terr = mdb.driver.SelectContext(ctx,\n\t\t\tdbShardID,\n\t\t\t&rows,\n\t\t\tqry,\n\t\t\t*filter.WorkflowTypeName,\n\t\t\tfilter.DomainID,\n\t\t\tmdb.converter.ToDateTime(*filter.MinStartTime),\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\t*filter.MaxStartTime,\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil && filter.CloseStatus != nil:\n\t\terr = mdb.driver.SelectContext(ctx,\n\t\t\tdbShardID,\n\t\t\t&rows,\n\t\t\ttemplateGetClosedWorkflowExecutionsByStatus,\n\t\t\t*filter.CloseStatus,\n\t\t\tfilter.DomainID,\n\t\t\tmdb.converter.ToDateTime(*filter.MinStartTime),\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil:\n\t\tqry := templateGetOpenWorkflowExecutions\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutions\n\t\t}\n\t\terr = mdb.driver.SelectContext(ctx,\n\t\t\tdbShardID,\n\t\t\t&rows,\n\t\t\tqry,\n\t\t\tfilter.DomainID,\n\t\t\tmdb.converter.ToDateTime(*filter.MinStartTime),\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\tmdb.converter.ToDateTime(*filter.MaxStartTime),\n\t\t\t*filter.PageSize)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid query filter\")\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].StartTime = mdb.converter.FromDateTime(rows[i].StartTime)\n\t\trows[i].ExecutionTime = mdb.converter.FromDateTime(rows[i].ExecutionTime)\n\t\tif rows[i].CloseTime != nil {\n\t\t\tcloseTime := mdb.converter.FromDateTime(*rows[i].CloseTime)\n\t\t\trows[i].CloseTime = &closeTime\n\t\t}\n\t}\n\treturn rows, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/admin.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\treadSchemaVersionQuery = `SELECT curr_version from schema_version where db_name=$1`\n\n\twriteSchemaVersionQuery = `INSERT into schema_version(db_name, creation_time, curr_version, min_compatible_version) VALUES ($1,$2,$3,$4)\n\t\t\t\t\t\t\t\t\t\tON CONFLICT (db_name) DO UPDATE \n\t\t\t\t\t\t\t\t\t\t  SET creation_time = excluded.creation_time,\n\t\t\t\t\t\t\t\t\t\t   \t  curr_version = excluded.curr_version,\n\t\t\t\t\t\t\t\t\t\t      min_compatible_version = excluded.min_compatible_version;`\n\n\twriteSchemaUpdateHistoryQuery = `INSERT into schema_update_history(year, month, update_time, old_version, new_version, manifest_md5, description) VALUES($1,$2,$3,$4,$5,$6,$7)`\n\n\tcreateSchemaVersionTableQuery = `CREATE TABLE schema_version(db_name VARCHAR(255) not null PRIMARY KEY, ` +\n\t\t`creation_time TIMESTAMP, ` +\n\t\t`curr_version VARCHAR(64), ` +\n\t\t`min_compatible_version VARCHAR(64));`\n\n\tcreateSchemaUpdateHistoryTableQuery = `CREATE TABLE schema_update_history(` +\n\t\t`year int not null, ` +\n\t\t`month int not null, ` +\n\t\t`update_time TIMESTAMP not null, ` +\n\t\t`description VARCHAR(255), ` +\n\t\t`manifest_md5 VARCHAR(64), ` +\n\t\t`new_version VARCHAR(64), ` +\n\t\t`old_version VARCHAR(64), ` +\n\t\t`PRIMARY KEY (year, month, update_time));`\n\n\t// NOTE we have to use %v because somehow postgres doesn't work with ? here\n\t// It's a small bug in sqlx library\n\t// TODO https://github.com/uber/cadence/issues/2893\n\tcreateDatabaseQuery = \"CREATE database %v\"\n\n\tdropDatabaseQuery = \"Drop database %v\"\n\n\tlistTablesQuery = \"select table_name from information_schema.tables where table_schema='public'\"\n\n\tdropTableQuery = \"DROP TABLE %v\"\n)\n\n// CreateSchemaVersionTables sets up the schema version tables\nfunc (pdb *db) CreateSchemaVersionTables() error {\n\tif err := pdb.ExecSchemaOperationQuery(context.Background(), createSchemaVersionTableQuery); err != nil {\n\t\treturn err\n\t}\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), createSchemaUpdateHistoryTableQuery)\n}\n\n// ReadSchemaVersion returns the current schema version for the keyspace\nfunc (pdb *db) ReadSchemaVersion(database string) (string, error) {\n\tvar version string\n\terr := pdb.driver.GetForSchemaQuery(sqlplugin.DbShardUndefined, &version, readSchemaVersionQuery, database)\n\treturn version, err\n}\n\n// UpdateSchemaVersion updates the schema version for the keyspace\nfunc (pdb *db) UpdateSchemaVersion(database string, newVersion string, minCompatibleVersion string) error {\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), writeSchemaVersionQuery, database, time.Now(), newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog adds an entry to the schema update history table\nfunc (pdb *db) WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error {\n\tnow := time.Now().UTC()\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), writeSchemaUpdateHistoryQuery, now.Year(), int(now.Month()), now, oldVersion, newVersion, manifestMD5, desc)\n}\n\n// ExecSchemaOperationQuery executes a sql statement for schema ONLY. DO NOT use it in other cases, otherwise it will not work for multiple SQL database.\n// For Sharded SQL, it will execute the statement for all shards\nfunc (pdb *db) ExecSchemaOperationQuery(ctx context.Context, stmt string, args ...interface{}) error {\n\t_, err := pdb.driver.ExecDDL(ctx, sqlplugin.DbShardUndefined, stmt, args...)\n\treturn err\n}\n\n// ListTables returns a list of tables in this database\nfunc (pdb *db) ListTables(database string) ([]string, error) {\n\tvar tables []string\n\terr := pdb.driver.SelectForSchemaQuery(sqlplugin.DbShardUndefined, &tables, listTablesQuery)\n\treturn tables, err\n}\n\n// DropTable drops a given table from the database\nfunc (pdb *db) DropTable(name string) error {\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(dropTableQuery, name))\n}\n\n// DropAllTables drops all tables from this database\nfunc (pdb *db) DropAllTables(database string) error {\n\ttables, err := pdb.ListTables(database)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, tab := range tables {\n\t\tif err := pdb.DropTable(tab); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// CreateDatabase creates a database if it doesn't exist\nfunc (pdb *db) CreateDatabase(name string) error {\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(createDatabaseQuery, name))\n}\n\n// DropDatabase drops a database\nfunc (pdb *db) DropDatabase(name string) error {\n\treturn pdb.ExecSchemaOperationQuery(context.Background(), fmt.Sprintf(dropDatabaseQuery, name))\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/configstore.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\t_selectLatestConfigQuery = \"SELECT row_type, version, timestamp, data, data_encoding FROM cluster_config WHERE row_type = $1 ORDER BY version LIMIT 1;\"\n\n\t_insertConfigQuery = \"INSERT INTO cluster_config (row_type, version, timestamp, data, data_encoding) VALUES($1, $2, $3, $4, $5)\"\n)\n\nfunc (pdb *db) InsertConfig(ctx context.Context, row *persistence.InternalConfigStoreEntry) error {\n\t_, err := pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, _insertConfigQuery, row.RowType, -1*row.Version, pdb.converter.ToPostgresDateTime(row.Timestamp), row.Values.Data, row.Values.Encoding)\n\treturn err\n}\n\nfunc (pdb *db) SelectLatestConfig(ctx context.Context, rowType int) (*persistence.InternalConfigStoreEntry, error) {\n\tvar row sqlplugin.ClusterConfigRow\n\terr := pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, _selectLatestConfigQuery, rowType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trow.Version *= -1\n\treturn &persistence.InternalConfigStoreEntry{\n\t\tRowType:   row.RowType,\n\t\tVersion:   row.Version,\n\t\tTimestamp: pdb.converter.FromPostgresDateTime(row.Timestamp),\n\t\tValues: &persistence.DataBlob{\n\t\t\tData:     row.Data,\n\t\t\tEncoding: constants.EncodingType(row.DataEncoding),\n\t\t},\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/db.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"time\"\n\n\t\"github.com/jmoiron/sqlx\"\n\t\"github.com/lib/pq\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\ntype (\n\tdb struct {\n\t\tconverter   DataConverter\n\t\tdriver      sqldriver.Driver\n\t\toriginalDBs []*sqlx.DB\n\t\tnumDBShards int\n\t}\n)\n\nfunc (pdb *db) GetTotalNumDBShards() int {\n\treturn pdb.numDBShards\n}\n\nvar _ sqlplugin.DB = (*db)(nil)\nvar _ sqlplugin.Tx = (*db)(nil)\n\n// ErrDupEntry indicates a duplicate primary key i.e. the row already exists,\n// check http://www.postgresql.org/docs/9.3/static/errcodes-appendix.html\nconst ErrDupEntry = \"23505\"\n\nconst ErrInsufficientResources = \"53000\"\nconst ErrTooManyConnections = \"53300\"\n\nfunc (pdb *db) IsDupEntryError(err error) bool {\n\tsqlErr, ok := err.(*pq.Error)\n\treturn ok && sqlErr.Code == ErrDupEntry\n}\n\nfunc (pdb *db) IsNotFoundError(err error) bool {\n\treturn err == sql.ErrNoRows\n}\n\nfunc (pdb *db) IsTimeoutError(err error) bool {\n\treturn err == context.DeadlineExceeded\n}\n\nfunc (pdb *db) IsThrottlingError(err error) bool {\n\tsqlErr, ok := err.(*pq.Error)\n\tif ok {\n\t\tif sqlErr.Code == ErrTooManyConnections ||\n\t\t\tsqlErr.Code == ErrInsufficientResources {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// newDB returns an instance of DB, which is a logical\n// connection to the underlying postgres database\n// dbShardID is needed when tx is not nil\nfunc newDB(xdbs []*sqlx.DB, tx *sqlx.Tx, dbShardID int, numDBShards int) (*db, error) {\n\tdriver, err := sqldriver.NewDriver(xdbs, tx, dbShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdb := &db{\n\t\tconverter:   &converter{},\n\t\toriginalDBs: xdbs, // this is kept because newDB will be called again when starting a transaction\n\t\tdriver:      driver,\n\t\tnumDBShards: numDBShards,\n\t}\n\treturn db, nil\n}\n\n// BeginTx starts a new transaction and returns a reference to the Tx object\nfunc (pdb *db) BeginTx(ctx context.Context, dbShardID int) (sqlplugin.Tx, error) {\n\txtx, err := pdb.driver.BeginTxx(ctx, dbShardID, nil)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newDB(pdb.originalDBs, xtx, dbShardID, pdb.numDBShards)\n}\n\n// Commit commits a previously started transaction\nfunc (pdb *db) Commit() error {\n\treturn pdb.driver.Commit()\n}\n\n// Rollback triggers rollback of a previously started transaction\nfunc (pdb *db) Rollback() error {\n\treturn pdb.driver.Rollback()\n}\n\n// Close closes the connection to the mysql db\nfunc (pdb *db) Close() error {\n\treturn pdb.driver.Close()\n}\n\n// PluginName returns the name of the mysql plugin\nfunc (pdb *db) PluginName() string {\n\treturn PluginName\n}\n\n// SupportsTTL returns weather Postgres supports TTL\nfunc (pdb *db) SupportsTTL() bool {\n\treturn false\n}\n\n// MaxAllowedTTL returns the max allowed ttl Postgres supports\nfunc (pdb *db) MaxAllowedTTL() (*time.Duration, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// SupportsTTL returns weather Postgre supports Asynchronous transaction\nfunc (pdb *db) SupportsAsyncTransaction() bool {\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/domain.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tcreateDomainQuery = `INSERT INTO \n domains (id, name, is_global, data, data_encoding)\n VALUES($1, $2, $3, $4, $5)`\n\n\tupdateDomainQuery = `UPDATE domains \n SET name = $1, data = $2, data_encoding = $3\n WHERE shard_id=54321 AND id = $4`\n\n\tgetDomainPart = `SELECT id, name, is_global, data, data_encoding FROM domains`\n\n\tgetDomainByIDQuery   = getDomainPart + ` WHERE shard_id=$1 AND id = $2`\n\tgetDomainByNameQuery = getDomainPart + ` WHERE shard_id=$1 AND name = $2`\n\n\tlistDomainsQuery      = getDomainPart + ` WHERE shard_id=$1 ORDER BY id LIMIT $2`\n\tlistDomainsRangeQuery = getDomainPart + ` WHERE shard_id=$1 AND id > $2 ORDER BY id LIMIT $3`\n\n\tdeleteDomainByIDQuery   = `DELETE FROM domains WHERE shard_id=$1 AND id = $2`\n\tdeleteDomainByNameQuery = `DELETE FROM domains WHERE shard_id=$1 AND name = $2`\n\n\tgetDomainMetadataQuery    = `SELECT notification_version FROM domain_metadata`\n\tlockDomainMetadataQuery   = `SELECT notification_version FROM domain_metadata FOR UPDATE`\n\tupdateDomainMetadataQuery = `UPDATE domain_metadata SET notification_version = $1 WHERE notification_version = $2`\n)\n\nconst (\n\tshardID = 54321\n)\n\nvar errMissingArgs = errors.New(\"missing one or more args for API\")\n\n// InsertIntoDomain inserts a single row into domains table\nfunc (pdb *db) InsertIntoDomain(ctx context.Context, row *sqlplugin.DomainRow) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, createDomainQuery, row.ID, row.Name, row.IsGlobal, row.Data, row.DataEncoding)\n}\n\n// UpdateDomain updates a single row in domains table\nfunc (pdb *db) UpdateDomain(ctx context.Context, row *sqlplugin.DomainRow) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, updateDomainQuery, row.Name, row.Data, row.DataEncoding, row.ID)\n}\n\n// SelectFromDomain reads one or more rows from domains table\nfunc (pdb *db) SelectFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tswitch {\n\tcase filter.ID != nil || filter.Name != nil:\n\t\treturn pdb.selectFromDomain(ctx, filter)\n\tcase filter.PageSize != nil && *filter.PageSize > 0:\n\t\treturn pdb.selectAllFromDomain(ctx, filter)\n\tdefault:\n\t\treturn nil, errMissingArgs\n\t}\n}\n\nfunc (pdb *db) selectFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tvar err error\n\tvar row sqlplugin.DomainRow\n\tswitch {\n\tcase filter.ID != nil:\n\t\terr = pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, getDomainByIDQuery, shardID, *filter.ID)\n\tcase filter.Name != nil:\n\t\terr = pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row, getDomainByNameQuery, shardID, *filter.Name)\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn []sqlplugin.DomainRow{row}, err\n}\n\nfunc (pdb *db) selectAllFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) ([]sqlplugin.DomainRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.DomainRow\n\tswitch {\n\tcase filter.GreaterThanID != nil:\n\t\terr = pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, listDomainsRangeQuery, shardID, *filter.GreaterThanID, *filter.PageSize)\n\tdefault:\n\t\terr = pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, listDomainsQuery, shardID, filter.PageSize)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromDomain deletes a single row in domains table\nfunc (pdb *db) DeleteFromDomain(ctx context.Context, filter *sqlplugin.DomainFilter) (sql.Result, error) {\n\tvar err error\n\tvar result sql.Result\n\tswitch {\n\tcase filter.ID != nil:\n\t\tresult, err = pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, deleteDomainByIDQuery, shardID, filter.ID)\n\tdefault:\n\t\tresult, err = pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, deleteDomainByNameQuery, shardID, filter.Name)\n\t}\n\treturn result, err\n}\n\n// LockDomainMetadata acquires a write lock on a single row in domain_metadata table\nfunc (pdb *db) LockDomainMetadata(ctx context.Context) error {\n\tvar row sqlplugin.DomainMetadataRow\n\terr := pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row.NotificationVersion, lockDomainMetadataQuery)\n\treturn err\n}\n\n// SelectFromDomainMetadata reads a single row in domain_metadata table\nfunc (pdb *db) SelectFromDomainMetadata(ctx context.Context) (*sqlplugin.DomainMetadataRow, error) {\n\tvar row sqlplugin.DomainMetadataRow\n\terr := pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row.NotificationVersion, getDomainMetadataQuery)\n\treturn &row, err\n}\n\n// UpdateDomainMetadata updates a single row in domain_metadata table\nfunc (pdb *db) UpdateDomainMetadata(ctx context.Context, row *sqlplugin.DomainMetadataRow) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, updateDomainMetadataQuery, row.NotificationVersion+1, row.NotificationVersion)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/domain_audit_log.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\t_insertDomainAuditLogQuery = `INSERT INTO domain_audit_log (\n\t\tdomain_id, event_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\t) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`\n\n\t_selectDomainAuditLogsQuery = `SELECT\n\t\tevent_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\tFROM domain_audit_log\n\tWHERE domain_id = $1 AND operation_type = $2 AND created_time >= $3\n\tAND (created_time < $4 OR (created_time = $4 AND event_id > $5))\n\tORDER BY created_time DESC, event_id ASC\n\tLIMIT $6`\n\t_selectAllDomainAuditLogsQuery = `SELECT\n\t\tevent_id, domain_id, state_before, state_before_encoding, state_after, state_after_encoding,\n\t\toperation_type, created_time, last_updated_time, identity, identity_type, comment\n\tFROM domain_audit_log\n\tWHERE domain_id = $1 AND operation_type = $2 AND created_time >= $3\n\tAND (created_time < $4 OR (created_time = $4 AND event_id > $5))\n\tORDER BY created_time DESC, event_id ASC`\n)\n\n// InsertIntoDomainAuditLog inserts a single row into domain_audit_log table\nfunc (pdb *db) InsertIntoDomainAuditLog(ctx context.Context, row *sqlplugin.DomainAuditLogRow) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(\n\t\tctx,\n\t\tsqlplugin.DbDefaultShard,\n\t\t_insertDomainAuditLogQuery,\n\t\trow.DomainID,\n\t\trow.EventID,\n\t\trow.StateBefore,\n\t\trow.StateBeforeEncoding,\n\t\trow.StateAfter,\n\t\trow.StateAfterEncoding,\n\t\trow.OperationType,\n\t\trow.CreatedTime,\n\t\trow.LastUpdatedTime,\n\t\trow.Identity,\n\t\trow.IdentityType,\n\t\trow.Comment,\n\t)\n}\n\n// SelectFromDomainAuditLogs returns audit log entries for a domain, operation type, and time range\nfunc (pdb *db) SelectFromDomainAuditLogs(\n\tctx context.Context,\n\tfilter *sqlplugin.DomainAuditLogFilter,\n) ([]*sqlplugin.DomainAuditLogRow, error) {\n\targs := []interface{}{\n\t\tfilter.DomainID,\n\t\tfilter.OperationType,\n\t\t*filter.MinCreatedTime,\n\t\t*filter.PageMaxCreatedTime,\n\t\t*filter.PageMinEventID,\n\t}\n\n\tvar rows []*sqlplugin.DomainAuditLogRow\n\tif filter.PageSize > 0 {\n\t\targs = append(args, filter.PageSize)\n\t\terr := pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, _selectDomainAuditLogsQuery, args...)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\terr := pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, _selectAllDomainAuditLogsQuery, args...)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn rows, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/domain_audit_log_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestInsertIntoDomainAuditLog(t *testing.T) {\n\tnow := time.Now().UTC()\n\n\ttests := []struct {\n\t\tname      string\n\t\trow       *sqlplugin.DomainAuditLogRow\n\t\tmockSetup func(*sqldriver.MockDriver)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"successfully inserted\",\n\t\t\trow: &sqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:            uuid.New(),\n\t\t\t\tEventID:             uuid.New(),\n\t\t\t\tStateBefore:         []byte(\"state-before\"),\n\t\t\t\tStateBeforeEncoding: constants.EncodingTypeJSON,\n\t\t\t\tStateAfter:          []byte(\"state-after\"),\n\t\t\t\tStateAfterEncoding:  constants.EncodingTypeJSON,\n\t\t\t\tOperationType:       persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:         now,\n\t\t\t\tLastUpdatedTime:     now,\n\t\t\t\tIdentity:            \"test-identity\",\n\t\t\t\tIdentityType:        \"user\",\n\t\t\t\tComment:             \"test comment\",\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\t_insertDomainAuditLogQuery,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t[]byte(\"state-before\"),\n\t\t\t\t\tconstants.EncodingTypeJSON,\n\t\t\t\t\t[]byte(\"state-after\"),\n\t\t\t\t\tconstants.EncodingTypeJSON,\n\t\t\t\t\tpersistence.DomainAuditOperationTypeFailover,\n\t\t\t\t\tnow,\n\t\t\t\t\tnow,\n\t\t\t\t\t\"test-identity\",\n\t\t\t\t\t\"user\",\n\t\t\t\t\t\"test comment\",\n\t\t\t\t).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"exec failed\",\n\t\t\trow: &sqlplugin.DomainAuditLogRow{\n\t\t\t\tDomainID:      uuid.New(),\n\t\t\t\tEventID:       uuid.New(),\n\t\t\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\t\t\tCreatedTime:   now,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().ExecContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, errors.New(\"exec failed\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\ttc.mockSetup(mockDriver)\n\n\t\t\tpdb := &db{\n\t\t\t\tdriver:    mockDriver,\n\t\t\t\tconverter: &converter{},\n\t\t\t}\n\n\t\t\t_, err := pdb.InsertIntoDomainAuditLog(context.Background(), tc.row)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSelectFromDomainAuditLogs(t *testing.T) {\n\tdomainID := \"d1111111-1111-1111-1111-111111111111\"\n\toperationType := persistence.DomainAuditOperationTypeFailover\n\tminTime := time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC)\n\tmaxTime := time.Date(2024, 12, 31, 23, 59, 59, 0, time.UTC)\n\n\t// Create times in descending order\n\tcreatedTime1 := time.Date(2024, 7, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime2 := time.Date(2024, 6, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime3 := time.Date(2024, 5, 1, 12, 0, 0, 0, time.UTC)\n\tcreatedTime4 := time.Date(2024, 4, 1, 12, 0, 0, 0, time.UTC)\n\n\teventID1 := \"e1111111-1111-1111-1111-111111111111\"\n\teventID2 := \"e2222222-2222-2222-2222-222222222222\"\n\teventID3 := \"e3333333-3333-3333-3333-333333333333\"\n\teventID4 := \"e4444444-4444-4444-4444-444444444444\"\n\n\t// Default page cursor: maxCreatedTime with max UUID (no page token)\n\tdefaultPageMaxCreatedTime := maxTime\n\tdefaultPageMinEventID := \"ffffffff-ffff-ffff-ffff-ffffffffffff\"\n\n\ttests := []struct {\n\t\tname      string\n\t\tfilter    *sqlplugin.DomainAuditLogFilter\n\t\tmockSetup func(*sqldriver.MockDriver)\n\t\twantRows  []*sqlplugin.DomainAuditLogRow\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"pageSize limits number of results; no pageToken\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageSize:           2,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMinEventID, 2,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\trows := dest.(*[]*sqlplugin.DomainAuditLogRow)\n\t\t\t\t\t*rows = []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID1,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime1,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID2,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime2,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID1,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime1,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID2,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime2,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"pageToken filters results by createdTime and eventID; pageSize is 0\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageSize:           0,\n\t\t\t\tPageMaxCreatedTime: &createdTime2,\n\t\t\t\tPageMinEventID:     &eventID2,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, createdTime2, eventID2,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\trows := dest.(*[]*sqlplugin.DomainAuditLogRow)\n\t\t\t\t\t*rows = []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID3,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime3,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:       eventID4,\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tCreatedTime:   createdTime4,\n\t\t\t\t\t\t\tOperationType: operationType,\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID3,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime3,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventID:       eventID4,\n\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\tCreatedTime:   createdTime4,\n\t\t\t\t\tOperationType: operationType,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success with no results\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMinEventID,\n\t\t\t\t).DoAndReturn(func(ctx context.Context, shardID int, dest interface{}, query string, args ...interface{}) error {\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t},\n\t\t\twantRows: []*sqlplugin.DomainAuditLogRow{},\n\t\t\twantErr:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"error when select fails\",\n\t\t\tfilter: &sqlplugin.DomainAuditLogFilter{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tOperationType:      operationType,\n\t\t\t\tMinCreatedTime:     &minTime,\n\t\t\t\tPageMaxCreatedTime: &defaultPageMaxCreatedTime,\n\t\t\t\tPageMinEventID:     &defaultPageMinEventID,\n\t\t\t},\n\t\t\tmockSetup: func(mockDriver *sqldriver.MockDriver) {\n\t\t\t\tmockDriver.EXPECT().SelectContext(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tsqlplugin.DbDefaultShard,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t_selectAllDomainAuditLogsQuery,\n\t\t\t\t\tdomainID, operationType, minTime, defaultPageMaxCreatedTime, defaultPageMinEventID,\n\t\t\t\t).Return(errors.New(\"select failed\"))\n\t\t\t},\n\t\t\twantRows: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDriver := sqldriver.NewMockDriver(ctrl)\n\t\t\ttc.mockSetup(mockDriver)\n\n\t\t\tpdb := &db{\n\t\t\t\tdriver:    mockDriver,\n\t\t\t\tconverter: &converter{},\n\t\t\t}\n\n\t\t\trows, err := pdb.SelectFromDomainAuditLogs(context.Background(), tc.filter)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, len(tc.wantRows), len(rows), \"number of rows should match\")\n\n\t\t\tfor i, wantRow := range tc.wantRows {\n\t\t\t\tif i < len(rows) {\n\t\t\t\t\tassert.Equal(t, wantRow.EventID, rows[i].EventID, \"row %d eventID\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.DomainID, rows[i].DomainID, \"row %d domainID\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.OperationType, rows[i].OperationType, \"row %d operationType\", i)\n\t\t\t\t\tassert.Equal(t, wantRow.CreatedTime.Unix(), rows[i].CreatedTime.Unix(), \"row %d createdTime\", i)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/events.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\t// below are templates for history_node table\n\taddHistoryNodesQuery = `INSERT INTO history_node (` +\n\t\t`shard_id, tree_id, branch_id, node_id, txn_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :tree_id, :branch_id, :node_id, :txn_id, :data, :data_encoding) `\n\n\tgetHistoryNodesQuery = `SELECT node_id, txn_id, data, data_encoding FROM history_node ` +\n\t\t`WHERE shard_id = $1 AND tree_id = $2 AND branch_id = $3 AND node_id >= $4 and node_id < $5 ORDER BY shard_id, tree_id, branch_id, node_id, txn_id LIMIT $6 `\n\n\tdeleteHistoryNodesQuery = `DELETE FROM history_node WHERE shard_id = $1 AND tree_id = $2 AND branch_id = $3 AND node_id >= $4`\n\n\tdeleteHistoryNodesByBatchQuery = `DELETE FROM history_node WHERE shard_id = $1 AND tree_id = $2 AND branch_id = $3 AND (node_id,txn_id) IN (SELECT node_id,txn_id FROM\n\t\thistory_node WHERE shard_id = $1 AND tree_id = $2 AND branch_id = $3 AND node_id >= $4 LIMIT $5)`\n\n\t// below are templates for history_tree table\n\taddHistoryTreeQuery = `INSERT INTO history_tree (` +\n\t\t`shard_id, tree_id, branch_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :tree_id, :branch_id, :data, :data_encoding) `\n\n\tgetHistoryTreeQuery = `SELECT branch_id, data, data_encoding FROM history_tree WHERE shard_id = $1 AND tree_id = $2 `\n\n\tdeleteHistoryTreeQuery = `DELETE FROM history_tree WHERE shard_id = $1 AND tree_id = $2 AND branch_id = $3 `\n\n\tgetAllHistoryTreeQuery = `SELECT shard_id, tree_id, branch_id, data, data_encoding FROM history_tree WHERE (shard_id = $1 AND tree_id = $2 AND branch_id > $3) OR (shard_id = $1 AND tree_id > $2) OR (shard_id > $1) ORDER BY shard_id, tree_id, branch_id LIMIT $4`\n)\n\n// For history_node table:\n\n// InsertIntoHistoryNode inserts a row into history_node table\nfunc (pdb *db) InsertIntoHistoryNode(ctx context.Context, row *sqlplugin.HistoryNodeRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(row.TreeID, pdb.GetTotalNumDBShards())\n\t// NOTE: Query 5.6 doesn't support clustering order, to workaround, we let txn_id multiple by -1\n\t*row.TxnID *= -1\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, addHistoryNodesQuery, row)\n}\n\n// SelectFromHistoryNode reads one or more rows from history_node table\nfunc (pdb *db) SelectFromHistoryNode(ctx context.Context, filter *sqlplugin.HistoryNodeFilter) ([]sqlplugin.HistoryNodeRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.HistoryNodeRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getHistoryNodesQuery,\n\t\tfilter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID, *filter.MaxNodeID, filter.PageSize)\n\t// NOTE: since we let txn_id multiple by -1 when inserting, we have to revert it back here\n\tfor _, row := range rows {\n\t\t*row.TxnID *= -1\n\t}\n\treturn rows, err\n}\n\n// DeleteFromHistoryNode deletes one or more rows from history_node table\nfunc (pdb *db) DeleteFromHistoryNode(ctx context.Context, filter *sqlplugin.HistoryNodeFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, pdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteHistoryNodesByBatchQuery, filter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID, filter.PageSize)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteHistoryNodesQuery, filter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID)\n}\n\n// For history_tree table:\n\n// InsertIntoHistoryTree inserts a row into history_tree table\nfunc (pdb *db) InsertIntoHistoryTree(ctx context.Context, row *sqlplugin.HistoryTreeRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(row.TreeID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, addHistoryTreeQuery, row)\n}\n\n// SelectFromHistoryTree reads one or more rows from history_tree table\nfunc (pdb *db) SelectFromHistoryTree(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) ([]sqlplugin.HistoryTreeRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.HistoryTreeRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getHistoryTreeQuery, filter.ShardID, filter.TreeID)\n\treturn rows, err\n}\n\n// DeleteFromHistoryTree deletes one or more rows from history_tree table\nfunc (pdb *db) DeleteFromHistoryTree(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteHistoryTreeQuery, filter.ShardID, filter.TreeID, *filter.BranchID)\n}\n\nfunc (pdb *db) GetAllHistoryTreeBranches(ctx context.Context, filter *sqlplugin.HistoryTreeFilter) ([]sqlplugin.HistoryTreeRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.HistoryTreeRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getAllHistoryTreeQuery, filter.ShardID, filter.TreeID, filter.BranchID, filter.PageSize)\n\treturn rows, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/execution.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\texecutionsColumns = `shard_id, domain_id, workflow_id, run_id, next_event_id, last_write_version, data, data_encoding`\n\n\tcreateExecutionQuery = `INSERT INTO executions(` + executionsColumns + `)\n VALUES(:shard_id, :domain_id, :workflow_id, :run_id, :next_event_id, :last_write_version, :data, :data_encoding)`\n\n\tupdateExecutionQuery = `UPDATE executions SET\n next_event_id = :next_event_id, last_write_version = :last_write_version, data = :data, data_encoding = :data_encoding\n WHERE shard_id = :shard_id AND domain_id = :domain_id AND workflow_id = :workflow_id AND run_id = :run_id`\n\n\tgetExecutionQuery = `SELECT ` + executionsColumns + ` FROM executions\n WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4`\n\n\tlistExecutionQuery = `SELECT ` + executionsColumns + ` FROM executions\n WHERE shard_id = $1 AND workflow_id > $2 ORDER BY workflow_id LIMIT $3`\n\n\tdeleteExecutionQuery = `DELETE FROM executions\n WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4`\n\n\tlockExecutionQueryBase = `SELECT next_event_id FROM executions\n WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4`\n\n\twriteLockExecutionQuery = lockExecutionQueryBase + ` FOR UPDATE`\n\treadLockExecutionQuery  = lockExecutionQueryBase + ` FOR SHARE`\n\n\tcreateCurrentExecutionQuery = `INSERT INTO current_executions\n(shard_id, domain_id, workflow_id, run_id, create_request_id, state, close_status, start_version, last_write_version) VALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :create_request_id, :state, :close_status, :start_version, :last_write_version)`\n\n\tdeleteCurrentExecutionQuery = \"DELETE FROM current_executions WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4\"\n\n\tgetCurrentExecutionQuery = `SELECT\nshard_id, domain_id, workflow_id, run_id, create_request_id, state, close_status, start_version, last_write_version\nFROM current_executions WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3`\n\n\tlockCurrentExecutionJoinExecutionsQuery = `SELECT\nce.shard_id, ce.domain_id, ce.workflow_id, ce.run_id, ce.create_request_id, ce.state, ce.close_status, ce.start_version, e.last_write_version\nFROM current_executions ce\nINNER JOIN executions e ON e.shard_id = ce.shard_id AND e.domain_id = ce.domain_id AND e.workflow_id = ce.workflow_id AND e.run_id = ce.run_id\nWHERE ce.shard_id = $1 AND ce.domain_id = $2 AND ce.workflow_id = $3 FOR UPDATE`\n\n\tlockCurrentExecutionQuery = getCurrentExecutionQuery + ` FOR UPDATE`\n\n\tupdateCurrentExecutionsQuery = `UPDATE current_executions SET\nrun_id = :run_id,\ncreate_request_id = :create_request_id,\nstate = :state,\nclose_status = :close_status,\nstart_version = :start_version,\nlast_write_version = :last_write_version\nWHERE\nshard_id = :shard_id AND\ndomain_id = :domain_id AND\nworkflow_id = :workflow_id\n`\n\n\tgetTransferTasksQuery = `SELECT task_id, data, data_encoding\n FROM transfer_tasks WHERE shard_id = $1 AND task_id >= $2 AND task_id < $3 ORDER BY shard_id, task_id LIMIT $4`\n\n\tcreateTransferTasksQuery = `INSERT INTO transfer_tasks(shard_id, task_id, data, data_encoding)\n VALUES(:shard_id, :task_id, :data, :data_encoding)`\n\n\tdeleteTransferTaskQuery             = `DELETE FROM transfer_tasks WHERE shard_id = $1 AND task_id = $2`\n\trangeDeleteTransferTaskQuery        = `DELETE FROM transfer_tasks WHERE shard_id = $1 AND task_id > $2 AND task_id <= $3`\n\trangeDeleteTransferTaskByBatchQuery = `DELETE FROM transfer_tasks WHERE shard_id = $1 AND task_id IN (SELECT task_id FROM\n\t\ttransfer_tasks WHERE shard_id = $1 AND task_id >= $2 AND task_id < $3 ORDER BY task_id LIMIT $4)`\n\n\tgetCrossClusterTasksQuery = `SELECT task_id, data, data_encoding\n FROM cross_cluster_tasks WHERE target_cluster = $1 AND shard_id = $2 AND task_id > $3 AND task_id <= $4 ORDER BY task_id LIMIT $5`\n\n\tcreateCrossClusterTasksQuery = `INSERT INTO cross_cluster_tasks(target_cluster, shard_id, task_id, data, data_encoding)\n VALUES(:target_cluster, :shard_id, :task_id, :data, :data_encoding)`\n\n\tdeleteCrossClusterTaskQuery             = `DELETE FROM cross_cluster_tasks WHERE target_cluster = $1 AND shard_id = $2 AND task_id = $3`\n\trangeDeleteCrossClusterTaskQuery        = `DELETE FROM cross_cluster_tasks WHERE target_cluster = $1 AND shard_id = $2 AND task_id >= $3 AND task_id < $4`\n\trangeDeleteCrossClusterTaskByBatchQuery = `DELETE FROM cross_cluster_tasks WHERE target_cluster = $1 AND shard_id = $2 AND task_id IN (SELECT task_id FROM\n\t\tcross_cluster_tasks WHERE target_cluster = $1 AND shard_id = $2 AND task_id >= $3 AND task_id < $4 ORDER BY task_id LIMIT $5)`\n\n\tcreateTimerTasksQuery = `INSERT INTO timer_tasks (shard_id, visibility_timestamp, task_id, data, data_encoding)\n  VALUES (:shard_id, :visibility_timestamp, :task_id, :data, :data_encoding)`\n\n\tgetTimerTasksQuery = `SELECT visibility_timestamp, task_id, data, data_encoding FROM timer_tasks\n  WHERE shard_id = $1\n  AND (visibility_timestamp, task_id) >= ($2, $3)\n  AND visibility_timestamp < $4\n  ORDER BY visibility_timestamp,task_id LIMIT $5`\n\n\tdeleteTimerTaskQuery             = `DELETE FROM timer_tasks WHERE shard_id = $1 AND visibility_timestamp = $2 AND task_id = $3`\n\trangeDeleteTimerTaskQuery        = `DELETE FROM timer_tasks WHERE shard_id = $1 AND visibility_timestamp >= $2 AND visibility_timestamp < $3`\n\trangeDeleteTimerTaskByBatchQuery = `DELETE FROM timer_tasks WHERE shard_id = $1 AND (visibility_timestamp,task_id) IN (SELECT visibility_timestamp,task_id FROM\n\t\ttimer_tasks WHERE shard_id = $1 AND visibility_timestamp >= $2 AND visibility_timestamp < $3 ORDER BY visibility_timestamp,task_id LIMIT $4)`\n\n\tcreateReplicationTasksQuery = `INSERT INTO replication_tasks (shard_id, task_id, data, data_encoding)\n  VALUES(:shard_id, :task_id, :data, :data_encoding)`\n\n\tgetReplicationTasksQuery = `SELECT task_id, data, data_encoding FROM replication_tasks WHERE\nshard_id = $1 AND\ntask_id >= $2 AND\ntask_id < $3\nORDER BY task_id LIMIT $4`\n\n\tdeleteReplicationTaskQuery             = `DELETE FROM replication_tasks WHERE shard_id = $1 AND task_id = $2`\n\trangeDeleteReplicationTaskQuery        = `DELETE FROM replication_tasks WHERE shard_id = $1 AND task_id <= $2`\n\trangeDeleteReplicationTaskByBatchQuery = `DELETE FROM replication_tasks WHERE shard_id = $1 AND task_id IN (SELECT task_id FROM\n\t\treplication_tasks WHERE task_id < $2 ORDER BY task_id LIMIT $3)`\n\n\tgetReplicationTasksDLQQuery = `SELECT task_id, data, data_encoding FROM replication_tasks_dlq WHERE\nsource_cluster_name = $1 AND\nshard_id = $2 AND\ntask_id >= $3 AND\ntask_id < $4\nORDER BY task_id LIMIT $5`\n\tgetReplicationTaskDLQQuery = `SELECT count(1) as count FROM replication_tasks_dlq WHERE\nsource_cluster_name = $1 AND\nshard_id = $2`\n\n\tbufferedEventsColumns     = `shard_id, domain_id, workflow_id, run_id, data, data_encoding`\n\tcreateBufferedEventsQuery = `INSERT INTO buffered_events(` + bufferedEventsColumns + `)\nVALUES (:shard_id, :domain_id, :workflow_id, :run_id, :data, :data_encoding)`\n\n\tdeleteBufferedEventsQuery = `DELETE FROM buffered_events WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4`\n\tgetBufferedEventsQuery    = `SELECT data, data_encoding FROM buffered_events WHERE shard_id = $1 AND domain_id = $2 AND workflow_id = $3 AND run_id = $4`\n\n\tinsertReplicationTaskDLQQuery = `\nINSERT INTO replication_tasks_dlq\n            (source_cluster_name,\n             shard_id,\n             task_id,\n             data,\n             data_encoding)\nVALUES     (:source_cluster_name,\n            :shard_id,\n            :task_id,\n            :data,\n            :data_encoding)\n`\n\tdeleteReplicationTaskFromDLQQuery = `\n\tDELETE FROM replication_tasks_dlq\n\t\tWHERE source_cluster_name = $1\n\t\tAND shard_id = $2\n\t\tAND task_id = $3`\n\n\trangeDeleteReplicationTaskFromDLQQuery = `\n\tDELETE FROM replication_tasks_dlq\n\t\tWHERE source_cluster_name = $1\n\t\tAND shard_id = $2\n\t\tAND task_id >= $3\n\t\tAND task_id < $4`\n\trangeDeleteReplicationTaskFromDLQByBatchQuery = `DELETE FROM replication_tasks_dlq WHERE source_cluster_name = $1 AND shard_id = $2 AND task_id IN (SELECT task_id FROM\n\t\treplication_tasks_dlq WHERE source_cluster_name = $1 AND shard_id = $2 AND task_id >= $3 AND task_id < $4 ORDER BY task_id LIMIT $5)`\n)\n\n// InsertIntoExecutions inserts a row into executions table\nfunc (pdb *db) InsertIntoExecutions(ctx context.Context, row *sqlplugin.ExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createExecutionQuery, row)\n}\n\n// UpdateExecutions updates a single row in executions table\nfunc (pdb *db) UpdateExecutions(ctx context.Context, row *sqlplugin.ExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, updateExecutionQuery, row)\n}\n\n// SelectFromExecutions reads a single row from executions table\n// The list execution query result is order by workflow ID only. It may returns duplicate record with pagination.\nfunc (pdb *db) SelectFromExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) ([]sqlplugin.ExecutionsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ExecutionsRow\n\tvar err error\n\tif len(filter.DomainID) == 0 && filter.Size > 0 {\n\t\terr = pdb.driver.SelectContext(ctx, dbShardID, &rows, listExecutionQuery, filter.ShardID, filter.WorkflowID, filter.Size)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tvar row sqlplugin.ExecutionsRow\n\t\terr = pdb.driver.GetContext(ctx, dbShardID, &row, getExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trows = append(rows, row)\n\t}\n\n\treturn rows, err\n}\n\n// DeleteFromExecutions deletes a single row from executions table\nfunc (pdb *db) DeleteFromExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// ReadLockExecutions acquires a write lock on a single row in executions table\nfunc (pdb *db) ReadLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar nextEventID int\n\terr := pdb.driver.GetContext(ctx, dbShardID, &nextEventID, readLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// WriteLockExecutions acquires a write lock on a single row in executions table\nfunc (pdb *db) WriteLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar nextEventID int\n\terr := pdb.driver.GetContext(ctx, dbShardID, &nextEventID, writeLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// InsertIntoCurrentExecutions inserts a single row into current_executions table\nfunc (pdb *db) InsertIntoCurrentExecutions(ctx context.Context, row *sqlplugin.CurrentExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createCurrentExecutionQuery, row)\n}\n\n// UpdateCurrentExecutions updates a single row in current_executions table\nfunc (pdb *db) UpdateCurrentExecutions(ctx context.Context, row *sqlplugin.CurrentExecutionsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, updateCurrentExecutionsQuery, row)\n}\n\n// SelectFromCurrentExecutions reads one or more rows from current_executions table\nfunc (pdb *db) SelectFromCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (*sqlplugin.CurrentExecutionsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar row sqlplugin.CurrentExecutionsRow\n\terr := pdb.driver.GetContext(ctx, dbShardID, &row, getCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn &row, err\n}\n\n// DeleteFromCurrentExecutions deletes a single row in current_executions table\nfunc (pdb *db) DeleteFromCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// LockCurrentExecutions acquires a write lock on a single row in current_executions table\nfunc (pdb *db) LockCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (*sqlplugin.CurrentExecutionsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar row sqlplugin.CurrentExecutionsRow\n\terr := pdb.driver.GetContext(ctx, dbShardID, &row, lockCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn &row, err\n}\n\n// LockCurrentExecutionsJoinExecutions joins a row in current_executions with executions table and acquires a\n// write lock on the result\nfunc (pdb *db) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) ([]sqlplugin.CurrentExecutionsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.CurrentExecutionsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, lockCurrentExecutionJoinExecutionsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn rows, err\n}\n\n// InsertIntoTransferTasks inserts one or more rows into transfer_tasks table\nfunc (pdb *db) InsertIntoTransferTasks(ctx context.Context, rows []sqlplugin.TransferTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createTransferTasksQuery, rows)\n}\n\n// SelectFromTransferTasks reads one or more rows from transfer_tasks table\nfunc (pdb *db) SelectFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) ([]sqlplugin.TransferTasksRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.TransferTasksRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getTransferTasksQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTransferTasks deletes one or more rows from transfer_tasks table\nfunc (pdb *db) DeleteFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteTransferTaskQuery, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromTransferTasks deletes multi rows from transfer_tasks table\nfunc (pdb *db) RangeDeleteFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskByBatchQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID)\n}\n\n// InsertIntoCrossClusterTasks inserts one or more rows into cross_cluster_tasks table\nfunc (pdb *db) InsertIntoCrossClusterTasks(ctx context.Context, rows []sqlplugin.CrossClusterTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createCrossClusterTasksQuery, rows)\n}\n\n// SelectFromCrossClusterTasks reads one or more rows from cross_cluster_tasks table\nfunc (pdb *db) SelectFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) ([]sqlplugin.CrossClusterTasksRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.CrossClusterTasksRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getCrossClusterTasksQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromCrossClusterTasks deletes one or more rows from cross_cluster_tasks table\nfunc (pdb *db) DeleteFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteCrossClusterTaskQuery, filter.TargetCluster, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromCrossClusterTasks deletes multi rows from cross_cluster_tasks table\nfunc (pdb *db) RangeDeleteFromCrossClusterTasks(ctx context.Context, filter *sqlplugin.CrossClusterTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteCrossClusterTaskByBatchQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID, filter.PageSize)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteCrossClusterTaskQuery, filter.TargetCluster, filter.ShardID, filter.MinTaskID, filter.MaxTaskID)\n}\n\n// InsertIntoTimerTasks inserts one or more rows into timer_tasks table\nfunc (pdb *db) InsertIntoTimerTasks(ctx context.Context, rows []sqlplugin.TimerTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, pdb.GetTotalNumDBShards())\n\tfor i := range rows {\n\t\trows[i].VisibilityTimestamp = pdb.converter.ToPostgresDateTime(rows[i].VisibilityTimestamp)\n\t}\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createTimerTasksQuery, rows)\n}\n\n// SelectFromTimerTasks reads one or more rows from timer_tasks table\nfunc (pdb *db) SelectFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) ([]sqlplugin.TimerTasksRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.TimerTasksRow\n\tfilter.MinVisibilityTimestamp = pdb.converter.ToPostgresDateTime(filter.MinVisibilityTimestamp)\n\tfilter.MaxVisibilityTimestamp = pdb.converter.ToPostgresDateTime(filter.MaxVisibilityTimestamp)\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getTimerTasksQuery, filter.ShardID, filter.MinVisibilityTimestamp,\n\t\tfilter.TaskID, filter.MaxVisibilityTimestamp, filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].VisibilityTimestamp = pdb.converter.FromPostgresDateTime(rows[i].VisibilityTimestamp)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTimerTasks deletes one or more rows from timer_tasks table\nfunc (pdb *db) DeleteFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tfilter.VisibilityTimestamp = pdb.converter.ToPostgresDateTime(filter.VisibilityTimestamp)\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteTimerTaskQuery, filter.ShardID, filter.VisibilityTimestamp, filter.TaskID)\n}\n\n// RangeDeleteFromTimerTasks deletes multi rows from timer_tasks table\nfunc (pdb *db) RangeDeleteFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tfilter.MinVisibilityTimestamp = pdb.converter.ToPostgresDateTime(filter.MinVisibilityTimestamp)\n\tfilter.MaxVisibilityTimestamp = pdb.converter.ToPostgresDateTime(filter.MaxVisibilityTimestamp)\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskByBatchQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp, filter.PageSize)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp)\n}\n\n// InsertIntoBufferedEvents inserts one or more rows into buffered_events table\nfunc (pdb *db) InsertIntoBufferedEvents(ctx context.Context, rows []sqlplugin.BufferedEventsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createBufferedEventsQuery, rows)\n}\n\n// SelectFromBufferedEvents reads one or more rows from buffered_events table\nfunc (pdb *db) SelectFromBufferedEvents(ctx context.Context, filter *sqlplugin.BufferedEventsFilter) ([]sqlplugin.BufferedEventsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.BufferedEventsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getBufferedEventsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t\trows[i].ShardID = filter.ShardID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromBufferedEvents deletes one or more rows from buffered_events table\nfunc (pdb *db) DeleteFromBufferedEvents(ctx context.Context, filter *sqlplugin.BufferedEventsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteBufferedEventsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// InsertIntoReplicationTasks inserts one or more rows into replication_tasks table\nfunc (pdb *db) InsertIntoReplicationTasks(ctx context.Context, rows []sqlplugin.ReplicationTasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(rows[0].ShardID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createReplicationTasksQuery, rows)\n}\n\n// SelectFromReplicationTasks reads one or more rows from replication_tasks table\nfunc (pdb *db) SelectFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) ([]sqlplugin.ReplicationTasksRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ReplicationTasksRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getReplicationTasksQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\treturn rows, err\n}\n\n// DeleteFromReplicationTasks deletes one rows from replication_tasks table\nfunc (pdb *db) DeleteFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteReplicationTaskQuery, filter.ShardID, filter.TaskID)\n}\n\n// RangeDeleteFromReplicationTasks deletes multi rows from replication_tasks table\nfunc (pdb *db) RangeDeleteFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskByBatchQuery, filter.ShardID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskQuery, filter.ShardID, filter.ExclusiveMaxTaskID)\n}\n\n// InsertIntoReplicationTasksDLQ inserts one or more rows into replication_tasks_dlq table\nfunc (pdb *db) InsertIntoReplicationTasksDLQ(ctx context.Context, row *sqlplugin.ReplicationTaskDLQRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, insertReplicationTaskDLQQuery, row)\n}\n\n// SelectFromReplicationTasksDLQ reads one or more rows from replication_tasks_dlq table\nfunc (pdb *db) SelectFromReplicationTasksDLQ(ctx context.Context, filter *sqlplugin.ReplicationTasksDLQFilter) ([]sqlplugin.ReplicationTasksRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ReplicationTasksRow\n\terr := pdb.driver.SelectContext(\n\t\tctx,\n\t\tdbShardID,\n\t\t&rows, getReplicationTasksDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.InclusiveMinTaskID,\n\t\tfilter.ExclusiveMaxTaskID,\n\t\tfilter.PageSize)\n\treturn rows, err\n}\n\n// SelectFromReplicationDLQ reads one row from replication_tasks_dlq table\nfunc (pdb *db) SelectFromReplicationDLQ(ctx context.Context, filter *sqlplugin.ReplicationTaskDLQFilter) (int64, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar size []int64\n\tif err := pdb.driver.SelectContext(\n\t\tctx,\n\t\tdbShardID,\n\t\t&size, getReplicationTaskDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n\n// DeleteMessageFromReplicationTasksDLQ deletes one row from replication_tasks_dlq table\nfunc (pdb *db) DeleteMessageFromReplicationTasksDLQ(\n\tctx context.Context,\n\tfilter *sqlplugin.ReplicationTasksDLQFilter,\n) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(\n\t\tctx,\n\t\tdbShardID,\n\t\tdeleteReplicationTaskFromDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.TaskID,\n\t)\n}\n\n// DeleteMessageFromReplicationTasksDLQ deletes one or more rows from replication_tasks_dlq table\nfunc (pdb *db) RangeDeleteMessageFromReplicationTasksDLQ(\n\tctx context.Context,\n\tfilter *sqlplugin.ReplicationTasksDLQFilter,\n) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn pdb.driver.ExecContext(\n\t\t\tctx,\n\t\t\tdbShardID,\n\t\t\trangeDeleteReplicationTaskFromDLQByBatchQuery,\n\t\t\tfilter.SourceClusterName,\n\t\t\tfilter.ShardID,\n\t\t\tfilter.InclusiveMinTaskID,\n\t\t\tfilter.ExclusiveMaxTaskID,\n\t\t\tfilter.PageSize,\n\t\t)\n\t}\n\n\treturn pdb.driver.ExecContext(\n\t\tctx,\n\t\tdbShardID,\n\t\trangeDeleteReplicationTaskFromDLQQuery,\n\t\tfilter.SourceClusterName,\n\t\tfilter.ShardID,\n\t\tfilter.InclusiveMinTaskID,\n\t\tfilter.ExclusiveMaxTaskID,\n\t)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/execution_maps.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tdeleteMapQueryTemplate = `DELETE FROM %v\nWHERE\nshard_id = $1 AND\ndomain_id = $2 AND\nworkflow_id = $3 AND\nrun_id = $4`\n\n\t// %[2]v is the columns of the value struct (i.e. no primary key columns), comma separated\n\t// %[3]v should be %[2]v with colons prepended.\n\t// i.e. %[3]v = \",\".join(\":\" + s for s in %[2]v)\n\t// %[5]v should be %[2]v with \"excluded.\" prepended.\n\t// i.e. %[5]v = \",\".join(\"excluded.\" + s for s in %[2]v)\n\t// So that this query can be used with BindNamed\n\t// %[4]v should be the name of the key associated with the map\n\t// e.g. for ActivityInfo it is \"schedule_id\"\n\tsetKeyInMapQueryTemplate = `INSERT INTO %[1]v\n(shard_id, domain_id, workflow_id, run_id, %[4]v, %[2]v)\nVALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :%[4]v, %[3]v)\nON CONFLICT (shard_id, domain_id, workflow_id, run_id, %[4]v) DO UPDATE\n\tSET (shard_id, domain_id, workflow_id, run_id, %[4]v, %[2]v)\n  \t  = (excluded.shard_id, excluded.domain_id, excluded.workflow_id, excluded.run_id, excluded.%[4]v, %[5]v)`\n\n\t// %[2]v is the name of the key\n\tdeleteKeyInMapQueryTemplate = `DELETE FROM %[1]v\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ? AND\n%[2]v IN ( ? )`\n\n\t// %[1]v is the name of the table\n\t// %[2]v is the name of the key\n\t// %[3]v is the value columns, separated by commas\n\tgetMapQueryTemplate = `SELECT %[2]v, %[3]v FROM %[1]v\nWHERE\nshard_id = $1 AND\ndomain_id = $2 AND\nworkflow_id = $3 AND\nrun_id = $4`\n)\n\nconst (\n\tdeleteAllSignalsRequestedSetQuery = `DELETE FROM signals_requested_sets\nWHERE\nshard_id = $1 AND\ndomain_id = $2 AND\nworkflow_id = $3 AND\nrun_id = $4\n`\n\n\tcreateSignalsRequestedSetQuery = `INSERT INTO signals_requested_sets\n(shard_id, domain_id, workflow_id, run_id, signal_id) VALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :signal_id)\nON CONFLICT (shard_id, domain_id, workflow_id, run_id, signal_id) DO NOTHING`\n\n\tdeleteSignalsRequestedSetQuery = `DELETE FROM signals_requested_sets\nWHERE\nshard_id = ? AND\ndomain_id = ? AND\nworkflow_id = ? AND\nrun_id = ? AND\nsignal_id IN ( ? )`\n\n\tgetSignalsRequestedSetQuery = `SELECT signal_id FROM signals_requested_sets WHERE\nshard_id = $1 AND\ndomain_id = $2 AND\nworkflow_id = $3 AND\nrun_id = $4`\n)\n\nfunc stringMap(a []string, f func(string) string) []string {\n\tb := make([]string, len(a))\n\tfor i, v := range a {\n\t\tb[i] = f(v)\n\t}\n\treturn b\n}\n\nfunc makeDeleteMapQry(tableName string) string {\n\treturn fmt.Sprintf(deleteMapQueryTemplate, tableName)\n}\n\nfunc makeSetKeyInMapQry(tableName string, nonPrimaryKeyColumns []string, mapKeyName string) string {\n\treturn fmt.Sprintf(setKeyInMapQueryTemplate,\n\t\ttableName,\n\t\tstrings.Join(nonPrimaryKeyColumns, \",\"),\n\t\tstrings.Join(stringMap(nonPrimaryKeyColumns, func(x string) string {\n\t\t\treturn \":\" + x\n\t\t}), \",\"),\n\t\tmapKeyName,\n\t\tstrings.Join(stringMap(nonPrimaryKeyColumns, func(x string) string {\n\t\t\treturn \"excluded.\" + x\n\t\t}), \",\"))\n}\n\nfunc makeDeleteKeyInMapQry(tableName string, mapKeyName string) string {\n\treturn fmt.Sprintf(deleteKeyInMapQueryTemplate,\n\t\ttableName,\n\t\tmapKeyName)\n}\n\nfunc makeGetMapQryTemplate(tableName string, nonPrimaryKeyColumns []string, mapKeyName string) string {\n\treturn fmt.Sprintf(getMapQueryTemplate,\n\t\ttableName,\n\t\tmapKeyName,\n\t\tstrings.Join(nonPrimaryKeyColumns, \",\"))\n}\n\nvar (\n\t// Omit shard_id, run_id, domain_id, workflow_id, schedule_id since they're in the primary key\n\tactivityInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t\t\"last_heartbeat_details\",\n\t\t\"last_heartbeat_updated_time\",\n\t}\n\tactivityInfoTableName = \"activity_info_maps\"\n\tactivityInfoKey       = \"schedule_id\"\n\n\tdeleteActivityInfoMapQry      = makeDeleteMapQry(activityInfoTableName)\n\tsetKeyInActivityInfoMapQry    = makeSetKeyInMapQry(activityInfoTableName, activityInfoColumns, activityInfoKey)\n\tdeleteKeyInActivityInfoMapQry = makeDeleteKeyInMapQry(activityInfoTableName, activityInfoKey)\n\tgetActivityInfoMapQry         = makeGetMapQryTemplate(activityInfoTableName, activityInfoColumns, activityInfoKey)\n)\n\n// ReplaceIntoActivityInfoMaps replaces one or more rows in activity_info_maps table\nfunc (pdb *db) ReplaceIntoActivityInfoMaps(ctx context.Context, rows []sqlplugin.ActivityInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\tfor i := range rows {\n\t\trows[i].LastHeartbeatUpdatedTime = pdb.converter.ToPostgresDateTime(rows[i].LastHeartbeatUpdatedTime)\n\t}\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, setKeyInActivityInfoMapQry, rows)\n}\n\n// SelectFromActivityInfoMaps reads one or more rows from activity_info_maps table\nfunc (pdb *db) SelectFromActivityInfoMaps(ctx context.Context, filter *sqlplugin.ActivityInfoMapsFilter) ([]sqlplugin.ActivityInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ActivityInfoMapsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t\trows[i].LastHeartbeatUpdatedTime = pdb.converter.FromPostgresDateTime(rows[i].LastHeartbeatUpdatedTime)\n\t}\n\treturn rows, err\n}\n\n// DeleteFromActivityInfoMaps deletes one or more rows from activity_info_maps table\nfunc (pdb *db) DeleteFromActivityInfoMaps(ctx context.Context, filter *sqlplugin.ActivityInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.ScheduleIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.ScheduleIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteActivityInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\ttimerInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\ttimerInfoTableName = \"timer_info_maps\"\n\ttimerInfoKey       = \"timer_id\"\n\n\tdeleteTimerInfoMapSQLQuery      = makeDeleteMapQry(timerInfoTableName)\n\tsetKeyInTimerInfoMapSQLQuery    = makeSetKeyInMapQry(timerInfoTableName, timerInfoColumns, timerInfoKey)\n\tdeleteKeyInTimerInfoMapSQLQuery = makeDeleteKeyInMapQry(timerInfoTableName, timerInfoKey)\n\tgetTimerInfoMapSQLQuery         = makeGetMapQryTemplate(timerInfoTableName, timerInfoColumns, timerInfoKey)\n)\n\n// ReplaceIntoTimerInfoMaps replaces one or more rows in timer_info_maps table\nfunc (pdb *db) ReplaceIntoTimerInfoMaps(ctx context.Context, rows []sqlplugin.TimerInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, setKeyInTimerInfoMapSQLQuery, rows)\n}\n\n// SelectFromTimerInfoMaps reads one or more rows from timer_info_maps table\nfunc (pdb *db) SelectFromTimerInfoMaps(ctx context.Context, filter *sqlplugin.TimerInfoMapsFilter) ([]sqlplugin.TimerInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.TimerInfoMapsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTimerInfoMaps deletes one or more rows from timer_info_maps table\nfunc (pdb *db) DeleteFromTimerInfoMaps(ctx context.Context, filter *sqlplugin.TimerInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.TimerIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.TimerIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteTimerInfoMapSQLQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\tchildExecutionInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\tchildExecutionInfoTableName = \"child_execution_info_maps\"\n\tchildExecutionInfoKey       = \"initiated_id\"\n\n\tdeleteChildExecutionInfoMapQry      = makeDeleteMapQry(childExecutionInfoTableName)\n\tsetKeyInChildExecutionInfoMapQry    = makeSetKeyInMapQry(childExecutionInfoTableName, childExecutionInfoColumns, childExecutionInfoKey)\n\tdeleteKeyInChildExecutionInfoMapQry = makeDeleteKeyInMapQry(childExecutionInfoTableName, childExecutionInfoKey)\n\tgetChildExecutionInfoMapQry         = makeGetMapQryTemplate(childExecutionInfoTableName, childExecutionInfoColumns, childExecutionInfoKey)\n)\n\n// ReplaceIntoChildExecutionInfoMaps replaces one or more rows in child_execution_info_maps table\nfunc (pdb *db) ReplaceIntoChildExecutionInfoMaps(ctx context.Context, rows []sqlplugin.ChildExecutionInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, setKeyInChildExecutionInfoMapQry, rows)\n}\n\n// SelectFromChildExecutionInfoMaps reads one or more rows from child_execution_info_maps table\nfunc (pdb *db) SelectFromChildExecutionInfoMaps(ctx context.Context, filter *sqlplugin.ChildExecutionInfoMapsFilter) ([]sqlplugin.ChildExecutionInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.ChildExecutionInfoMapsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromChildExecutionInfoMaps deletes one or more rows from child_execution_info_maps table\nfunc (pdb *db) DeleteFromChildExecutionInfoMaps(ctx context.Context, filter *sqlplugin.ChildExecutionInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteChildExecutionInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\trequestCancelInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\trequestCancelInfoTableName = \"request_cancel_info_maps\"\n\trequestCancelInfoKey       = \"initiated_id\"\n\n\tdeleteRequestCancelInfoMapQry      = makeDeleteMapQry(requestCancelInfoTableName)\n\tsetKeyInRequestCancelInfoMapQry    = makeSetKeyInMapQry(requestCancelInfoTableName, requestCancelInfoColumns, requestCancelInfoKey)\n\tdeleteKeyInRequestCancelInfoMapQry = makeDeleteKeyInMapQry(requestCancelInfoTableName, requestCancelInfoKey)\n\tgetRequestCancelInfoMapQry         = makeGetMapQryTemplate(requestCancelInfoTableName, requestCancelInfoColumns, requestCancelInfoKey)\n)\n\n// ReplaceIntoRequestCancelInfoMaps replaces one or more rows in request_cancel_info_maps table\nfunc (pdb *db) ReplaceIntoRequestCancelInfoMaps(ctx context.Context, rows []sqlplugin.RequestCancelInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, setKeyInRequestCancelInfoMapQry, rows)\n}\n\n// SelectFromRequestCancelInfoMaps reads one or more rows from request_cancel_info_maps table\nfunc (pdb *db) SelectFromRequestCancelInfoMaps(ctx context.Context, filter *sqlplugin.RequestCancelInfoMapsFilter) ([]sqlplugin.RequestCancelInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.RequestCancelInfoMapsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromRequestCancelInfoMaps deletes one or more rows from request_cancel_info_maps table\nfunc (pdb *db) DeleteFromRequestCancelInfoMaps(ctx context.Context, filter *sqlplugin.RequestCancelInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteRequestCancelInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\nvar (\n\tsignalInfoColumns = []string{\n\t\t\"data\",\n\t\t\"data_encoding\",\n\t}\n\tsignalInfoTableName = \"signal_info_maps\"\n\tsignalInfoKey       = \"initiated_id\"\n\n\tdeleteSignalInfoMapQry      = makeDeleteMapQry(signalInfoTableName)\n\tsetKeyInSignalInfoMapQry    = makeSetKeyInMapQry(signalInfoTableName, signalInfoColumns, signalInfoKey)\n\tdeleteKeyInSignalInfoMapQry = makeDeleteKeyInMapQry(signalInfoTableName, signalInfoKey)\n\tgetSignalInfoMapQry         = makeGetMapQryTemplate(signalInfoTableName, signalInfoColumns, signalInfoKey)\n)\n\n// ReplaceIntoSignalInfoMaps replaces one or more rows in signal_info_maps table\nfunc (pdb *db) ReplaceIntoSignalInfoMaps(ctx context.Context, rows []sqlplugin.SignalInfoMapsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, setKeyInSignalInfoMapQry, rows)\n}\n\n// SelectFromSignalInfoMaps reads one or more rows from signal_info_maps table\nfunc (pdb *db) SelectFromSignalInfoMaps(ctx context.Context, filter *sqlplugin.SignalInfoMapsFilter) ([]sqlplugin.SignalInfoMapsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.SignalInfoMapsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromSignalInfoMaps deletes one or more rows from signal_info_maps table\nfunc (pdb *db) DeleteFromSignalInfoMaps(ctx context.Context, filter *sqlplugin.SignalInfoMapsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.InitiatedIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteKeyInSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.InitiatedIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteSignalInfoMapQry, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n\n// InsertIntoSignalsRequestedSets inserts one or more rows into signals_requested_sets table\nfunc (pdb *db) InsertIntoSignalsRequestedSets(ctx context.Context, rows []sqlplugin.SignalsRequestedSetsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.NamedExecContext(ctx, dbShardID, createSignalsRequestedSetQuery, rows)\n}\n\n// SelectFromSignalsRequestedSets reads one or more rows from signals_requested_sets table\nfunc (pdb *db) SelectFromSignalsRequestedSets(ctx context.Context, filter *sqlplugin.SignalsRequestedSetsFilter) ([]sqlplugin.SignalsRequestedSetsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rows []sqlplugin.SignalsRequestedSetsRow\n\terr := pdb.driver.SelectContext(ctx, dbShardID, &rows, getSignalsRequestedSetQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\tfor i := 0; i < len(rows); i++ {\n\t\trows[i].ShardID = int64(filter.ShardID)\n\t\trows[i].DomainID = filter.DomainID\n\t\trows[i].WorkflowID = filter.WorkflowID\n\t\trows[i].RunID = filter.RunID\n\t}\n\treturn rows, err\n}\n\n// DeleteFromSignalsRequestedSets deletes one or more rows from signals_requested_sets table\nfunc (pdb *db) DeleteFromSignalsRequestedSets(ctx context.Context, filter *sqlplugin.SignalsRequestedSetsFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tif len(filter.SignalIDs) > 0 {\n\t\tquery, args, err := sqlx.In(deleteSignalsRequestedSetQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID, filter.SignalIDs)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, sqlx.Rebind(sqlx.BindType(PluginName), query), args...)\n\t}\n\treturn pdb.driver.ExecContext(ctx, dbShardID, deleteAllSignalsRequestedSetQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/plugin.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\t\"net/url\"\n\t\"os\"\n\t\"runtime\"\n\n\t\"github.com/iancoleman/strcase\"\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/environment\"\n)\n\nconst (\n\t// PluginName is the name of the plugin\n\tPluginName = \"postgres\"\n\tdsnFmt     = \"postgres://%s@%s:%s/%s\"\n)\n\ntype plugin struct{}\n\nvar _ sqlplugin.Plugin = (*plugin)(nil)\n\nfunc init() {\n\tsql.RegisterPlugin(PluginName, &plugin{})\n}\n\n// CreateDB initialize the db object\nfunc (d *plugin) CreateDB(cfg *config.SQL) (sqlplugin.DB, error) {\n\tconns, err := sqldriver.CreateDBConnections(cfg, func(cfg *config.SQL) (*sqlx.DB, error) {\n\t\treturn d.createSingleDBConn(cfg)\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newDB(conns, nil, sqlplugin.DbShardUndefined, cfg.NumShards)\n}\n\n// CreateAdminDB initialize the adminDB object\nfunc (d *plugin) CreateAdminDB(cfg *config.SQL) (sqlplugin.AdminDB, error) {\n\tconns, err := sqldriver.CreateDBConnections(cfg, func(cfg *config.SQL) (*sqlx.DB, error) {\n\t\treturn d.createSingleDBConn(cfg)\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newDB(conns, nil, sqlplugin.DbShardUndefined, cfg.NumShards)\n}\n\n// CreateDBConnection creates a returns a reference to a logical connection to the\n// underlying SQL database. The returned object is to tied to a single\n// SQL database and the object can be used to perform CRUD operations on\n// the tables in the database\nfunc (d *plugin) createSingleDBConn(cfg *config.SQL) (*sqlx.DB, error) {\n\tparams, err := registerTLSConfig(cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor k, v := range cfg.ConnectAttributes {\n\t\tparams.Set(k, v)\n\t}\n\thost, port, err := net.SplitHostPort(cfg.ConnectAddr)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid connect address, it must be in host:port format, %v, err: %v\", cfg.ConnectAddr, err)\n\t}\n\n\tdb, err := sqlx.Connect(PluginName, buildDSN(cfg, host, port, params))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cfg.MaxConns > 0 {\n\t\tdb.SetMaxOpenConns(cfg.MaxConns)\n\t}\n\tif cfg.MaxIdleConns > 0 {\n\t\tdb.SetMaxIdleConns(cfg.MaxIdleConns)\n\t}\n\tif cfg.MaxConnLifetime > 0 {\n\t\tdb.SetConnMaxLifetime(cfg.MaxConnLifetime)\n\t}\n\n\t// Maps struct names in CamelCase to snake without need for db struct tags.\n\tdb.MapperFunc(strcase.ToSnake)\n\treturn db, nil\n}\n\nfunc buildDSN(cfg *config.SQL, host string, port string, params url.Values) string {\n\tdbName := cfg.DatabaseName\n\t// NOTE: postgres doesn't allow to connect with empty dbName, the admin dbName is \"postgres\"\n\tif dbName == \"\" {\n\t\tdbName = \"postgres\"\n\t}\n\n\tcredentialString := generateCredentialString(cfg.User, cfg.Password)\n\tdsn := fmt.Sprintf(dsnFmt, credentialString, host, port, dbName)\n\tif attrs := params.Encode(); attrs != \"\" {\n\t\tdsn += \"?\" + attrs\n\t}\n\treturn dsn\n}\n\nfunc generateCredentialString(user string, password string) string {\n\tuserPass := url.PathEscape(user)\n\tif password != \"\" {\n\t\tuserPass += \":\" + url.PathEscape(password)\n\t}\n\treturn userPass\n}\n\nfunc registerTLSConfig(cfg *config.SQL) (sslParams url.Values, err error) {\n\tsslParams = url.Values{}\n\tif cfg.TLS != nil && cfg.TLS.Enabled {\n\t\tsslMode := cfg.TLS.SSLMode\n\t\tif sslMode == \"\" {\n\t\t\t// NOTE: Default to require for backward compatibility for Cadence users.\n\t\t\tsslMode = \"require\"\n\t\t}\n\t\tsslParams.Set(\"sslmode\", sslMode)\n\t\tsslParams.Set(\"sslrootcert\", cfg.TLS.CaFile)\n\t\tsslParams.Set(\"sslkey\", cfg.TLS.KeyFile)\n\t\tsslParams.Set(\"sslcert\", cfg.TLS.CertFile)\n\t} else {\n\t\tsslParams.Set(\"sslmode\", \"disable\")\n\t}\n\treturn\n}\n\nconst (\n\ttestSchemaDir = \"schema/postgres\"\n)\n\n// GetTestClusterOption return test options\nfunc GetTestClusterOption() (*pt.TestBaseOptions, error) {\n\ttestUser := \"postgres\"\n\ttestPassword := \"cadence\"\n\n\tif runtime.GOOS == \"darwin\" {\n\t\ttestUser = os.Getenv(\"USER\")\n\t\ttestPassword = \"\"\n\t}\n\n\tif os.Getenv(\"POSTGRES_USER\") != \"\" {\n\t\ttestUser = os.Getenv(\"POSTGRES_USER\")\n\t}\n\n\tif os.Getenv(\"POSTGRES_PASSWORD\") != \"\" {\n\t\ttestPassword = os.Getenv(\"POSTGRES_PASSWORD\")\n\t}\n\tdbPort, err := environment.GetPostgresPort()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &pt.TestBaseOptions{\n\t\tDBPluginName: PluginName,\n\t\tDBUsername:   testUser,\n\t\tDBPassword:   testPassword,\n\t\tDBHost:       environment.GetPostgresAddress(),\n\t\tDBPort:       dbPort,\n\t\tSchemaDir:    testSchemaDir,\n\t}, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/plugin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage postgres\n\nimport \"testing\"\n\nvar testCases = []struct {\n\tname     string\n\tusername string\n\tpassword string\n\twant     string\n}{\n\t{\n\t\tname:     \"default\",\n\t\tusername: \"cadence\",\n\t\tpassword: \"cadence\",\n\t\twant:     \"cadence:cadence\",\n\t},\n\t{\n\t\tname:     \"with forward slash\",\n\t\tusername: \"cadence\",\n\t\tpassword: \"cad/ence\",\n\t\twant:     \"cadence:cad%2Fence\",\n\t},\n\t{\n\t\tname:     \"with question mark\",\n\t\tusername: \"cadence\",\n\t\tpassword: \"cad?ence\",\n\t\twant:     \"cadence:cad%3Fence\",\n\t},\n}\n\nfunc TestGenerateCredentialString(t *testing.T) {\n\tfor _, tc := range testCases {\n\t\tif userPass := generateCredentialString(tc.username, tc.password); userPass != tc.want {\n\t\t\tt.Errorf(\"%v: got %v, want %v\", tc.name, userPass, tc.want)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/queue.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateEnqueueMessageQuery            = `INSERT INTO queue (queue_type, message_id, message_payload) VALUES(:queue_type, :message_id, :message_payload)`\n\ttemplateGetLastMessageIDQuery          = `SELECT message_id FROM queue WHERE queue_type=$1 ORDER BY message_id DESC LIMIT 1 FOR UPDATE`\n\ttemplateGetMessagesQuery               = `SELECT message_id, message_payload FROM queue WHERE queue_type = $1 and message_id > $2 ORDER BY message_id ASC LIMIT $3`\n\ttemplateGetMessagesBetweenQuery        = `SELECT message_id, message_payload FROM queue WHERE queue_type = $1 and message_id > $2 and message_id <= $3 ORDER BY message_id ASC LIMIT $4`\n\ttemplateDeleteMessageQuery             = `DELETE FROM queue WHERE queue_type = $1 and message_id = $2`\n\ttemplateDeleteMessagesBeforeQuery      = `DELETE FROM queue WHERE queue_type = $1 and message_id < $2`\n\ttemplateRangeDeleteMessagesQuery       = `DELETE FROM queue WHERE queue_type = $1 and message_id > $2 and message_id <= $3`\n\ttemplateGetQueueMetadataQuery          = `SELECT data from queue_metadata WHERE queue_type = $1`\n\ttemplateGetQueueMetadataForUpdateQuery = templateGetQueueMetadataQuery + ` FOR UPDATE`\n\ttemplateInsertQueueMetadataQuery       = `INSERT INTO queue_metadata (queue_type, data) VALUES(:queue_type, :data)`\n\ttemplateUpdateQueueMetadataQuery       = `UPDATE queue_metadata SET data = $1 WHERE queue_type = $2`\n\ttemplateGetQueueSizeQuery              = `SELECT COUNT(1) AS count FROM queue WHERE queue_type=$1`\n)\n\n// InsertIntoQueue inserts a new row into queue table\nfunc (pdb *db) InsertIntoQueue(ctx context.Context, row *sqlplugin.QueueRow) (sql.Result, error) {\n\treturn pdb.driver.NamedExecContext(ctx, sqlplugin.DbDefaultShard, templateEnqueueMessageQuery, row)\n}\n\n// GetLastEnqueuedMessageIDForUpdate returns the last enqueued message ID\nfunc (pdb *db) GetLastEnqueuedMessageIDForUpdate(ctx context.Context, queueType persistence.QueueType) (int64, error) {\n\tvar lastMessageID int64\n\terr := pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &lastMessageID, templateGetLastMessageIDQuery, queueType)\n\treturn lastMessageID, err\n}\n\n// GetMessagesFromQueue retrieves messages from the queue\nfunc (pdb *db) GetMessagesFromQueue(ctx context.Context, queueType persistence.QueueType, lastMessageID int64, maxRows int) ([]sqlplugin.QueueRow, error) {\n\tvar rows []sqlplugin.QueueRow\n\terr := pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, templateGetMessagesQuery, queueType, lastMessageID, maxRows)\n\treturn rows, err\n}\n\n// GetMessagesBetween retrieves messages from the queue\nfunc (pdb *db) GetMessagesBetween(ctx context.Context, queueType persistence.QueueType, firstMessageID int64, lastMessageID int64, maxRows int) ([]sqlplugin.QueueRow, error) {\n\tvar rows []sqlplugin.QueueRow\n\terr := pdb.driver.SelectContext(ctx, sqlplugin.DbDefaultShard, &rows, templateGetMessagesBetweenQuery, queueType, firstMessageID, lastMessageID, maxRows)\n\treturn rows, err\n}\n\n// DeleteMessagesBefore deletes messages before messageID from the queue\nfunc (pdb *db) DeleteMessagesBefore(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateDeleteMessagesBeforeQuery, queueType, messageID)\n}\n\n// RangeDeleteMessages deletes messages before messageID from the queue\nfunc (pdb *db) RangeDeleteMessages(ctx context.Context, queueType persistence.QueueType, exclusiveBeginMessageID int64, inclusiveEndMessageID int64) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateRangeDeleteMessagesQuery, queueType, exclusiveBeginMessageID, inclusiveEndMessageID)\n}\n\n// DeleteMessage deletes message with a messageID from the queue\nfunc (pdb *db) DeleteMessage(ctx context.Context, queueType persistence.QueueType, messageID int64) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateDeleteMessageQuery, queueType, messageID)\n}\n\n// InsertAckLevel inserts ack level\nfunc (pdb *db) InsertAckLevel(ctx context.Context, queueType persistence.QueueType, messageID int64, clusterName string) error {\n\tclusterAckLevels := map[string]int64{clusterName: messageID}\n\tdata, err := json.Marshal(clusterAckLevels)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = pdb.driver.NamedExecContext(ctx, sqlplugin.DbDefaultShard, templateInsertQueueMetadataQuery, sqlplugin.QueueMetadataRow{QueueType: queueType, Data: data})\n\treturn err\n\n}\n\n// UpdateAckLevels updates cluster ack levels\nfunc (pdb *db) UpdateAckLevels(ctx context.Context, queueType persistence.QueueType, clusterAckLevels map[string]int64) error {\n\tdata, err := json.Marshal(clusterAckLevels)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t_, err = pdb.driver.ExecContext(ctx, sqlplugin.DbDefaultShard, templateUpdateQueueMetadataQuery, data, queueType)\n\treturn err\n}\n\n// GetAckLevels returns ack levels for pulling clusters\nfunc (pdb *db) GetAckLevels(ctx context.Context, queueType persistence.QueueType, forUpdate bool) (map[string]int64, error) {\n\tqueryStr := templateGetQueueMetadataQuery\n\tif forUpdate {\n\t\tqueryStr = templateGetQueueMetadataForUpdateQuery\n\t}\n\n\tvar data []byte\n\terr := pdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &data, queryStr, queueType)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\n\tvar clusterAckLevels map[string]int64\n\tif err := json.Unmarshal(data, &clusterAckLevels); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn clusterAckLevels, nil\n}\n\n// GetQueueSize returns the queue size\nfunc (pdb *db) GetQueueSize(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tvar size []int64\n\tif err := pdb.driver.SelectContext(\n\t\tctx,\n\t\tsqlplugin.DbDefaultShard,\n\t\t&size,\n\t\ttemplateGetQueueSizeQuery,\n\t\tqueueType,\n\t); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/shard.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tcreateShardQry = `INSERT INTO\n shards (shard_id, range_id, data, data_encoding) VALUES ($1, $2, $3, $4)`\n\n\tgetShardQry = `SELECT\n shard_id, range_id, data, data_encoding\n FROM shards WHERE shard_id = $1`\n\n\tupdateShardQry = `UPDATE shards \n SET range_id = $1, data = $2, data_encoding = $3 \n WHERE shard_id = $4`\n\n\tlockShardQry     = `SELECT range_id FROM shards WHERE shard_id = $1 FOR UPDATE`\n\treadLockShardQry = `SELECT range_id FROM shards WHERE shard_id = $1 FOR SHARE`\n)\n\n// InsertIntoShards inserts one or more rows into shards table\nfunc (pdb *db) InsertIntoShards(ctx context.Context, row *sqlplugin.ShardsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, createShardQry, row.ShardID, row.RangeID, row.Data, row.DataEncoding)\n}\n\n// UpdateShards updates one or more rows into shards table\nfunc (pdb *db) UpdateShards(ctx context.Context, row *sqlplugin.ShardsRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(row.ShardID), pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, updateShardQry, row.RangeID, row.Data, row.DataEncoding, row.ShardID)\n}\n\n// SelectFromShards reads one or more rows from shards table\nfunc (pdb *db) SelectFromShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (*sqlplugin.ShardsRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar row sqlplugin.ShardsRow\n\terr := pdb.driver.GetContext(ctx, dbShardID, &row, getShardQry, filter.ShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &row, err\n}\n\n// ReadLockShards acquires a read lock on a single row in shards table\nfunc (pdb *db) ReadLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := pdb.driver.GetContext(ctx, dbShardID, &rangeID, readLockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n\n// WriteLockShards acquires a write lock on a single row in shards table\nfunc (pdb *db) WriteLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), pdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := pdb.driver.GetContext(ctx, dbShardID, &rangeID, lockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/task.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttaskListCreatePart = `INTO task_lists(shard_id, domain_id, name, task_type, range_id, data, data_encoding) ` +\n\t\t`VALUES (:shard_id, :domain_id, :name, :task_type, :range_id, :data, :data_encoding)`\n\n\t// (default range ID: initialRangeID == 1)\n\tcreateTaskListQry = `INSERT ` + taskListCreatePart\n\n\tupdateTaskListQry = `UPDATE task_lists SET\nrange_id = :range_id,\ndata = :data,\ndata_encoding = :data_encoding\nWHERE\nshard_id = :shard_id AND\ndomain_id = :domain_id AND\nname = :name AND\ntask_type = :task_type\n`\n\n\t// This query uses pagination that is best understood by analogy to simple numbers.\n\t// Given a list of numbers\n\t// \t111\n\t//\t112\n\t//\t121\n\t//\t211\n\t// where the hundreds digit corresponds to domain_id, the tens digit\n\t// corresponds to name, and the ones digit corresponds to task_type,\n\t// Imagine recurring queries with a limit of 1.\n\t// For the second query to skip the first result and return 112, it must allow equal values in hundreds & tens, but it's OK because the ones digit is higher.\n\t// For the third query, the ones digit is now lower but that's irrelevant because the tens digit is greater.\n\t// For the fourth query, the tens digit is now lower but that's again irrelevant because now the hundreds digit is higher.\n\t// This technique is useful since the size of the table can easily change between calls, making SKIP an unreliable method, while other db-specific things like rowids are not portable\n\tlistTaskListQry = `SELECT domain_id, range_id, name, task_type, data, data_encoding ` +\n\t\t`FROM task_lists ` +\n\t\t`WHERE shard_id = $1 AND ((domain_id = $2 AND name = $3 AND task_type > $4) OR (domain_id=$2 AND name > $3) OR (domain_id > $2)) ` +\n\t\t`ORDER BY domain_id,name,task_type LIMIT $5`\n\n\tgetTaskListQry = `SELECT domain_id, range_id, name, task_type, data, data_encoding ` +\n\t\t`FROM task_lists ` +\n\t\t`WHERE shard_id = $1 AND domain_id = $2 AND name = $3 AND task_type = $4`\n\n\tdeleteTaskListQry = `DELETE FROM task_lists WHERE shard_id=$1 AND domain_id=$2 AND name=$3 AND task_type=$4 AND range_id=$5`\n\n\tlockTaskListQry = `SELECT range_id FROM task_lists ` +\n\t\t`WHERE shard_id = $1 AND domain_id = $2 AND name = $3 AND task_type = $4 FOR UPDATE`\n\n\tgetTaskMinMaxQry = `SELECT task_id, data, data_encoding ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id > $4 AND task_id <= $5 ` +\n\t\t` ORDER BY task_id LIMIT $6`\n\n\tgetTaskMinQry = `SELECT task_id, data, data_encoding ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id > $4 ORDER BY task_id LIMIT $5`\n\n\tgetTasksCountQry = `SELECT count(1) as count ` +\n\t\t`FROM tasks ` +\n\t\t`WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id > $4`\n\n\tcreateTaskQry = `INSERT INTO ` +\n\t\t`tasks(domain_id, task_list_name, task_type, task_id, data, data_encoding) ` +\n\t\t`VALUES(:domain_id, :task_list_name, :task_type, :task_id, :data, :data_encoding)`\n\n\tdeleteTaskQry = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id = $4`\n\n\trangeDeleteTaskQry = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id IN (SELECT task_id FROM\n\t\t tasks WHERE domain_id = $1 AND task_list_name = $2 AND task_type = $3 AND task_id <= $4 ` +\n\t\t`ORDER BY domain_id,task_list_name,task_type,task_id LIMIT $5 )`\n\n\tgetOrphanTaskQry = `SELECT task_id, domain_id, task_list_name, task_type FROM tasks AS t ` +\n\t\t`WHERE NOT EXISTS ( ` +\n\t\t`\tSELECT domain_id, name, task_type FROM task_lists AS tl ` +\n\t\t`\tWHERE t.domain_id=tl.domain_id and t.task_list_name=tl.name and t.task_type=tl.task_type ` +\n\t\t`) LIMIT $1;`\n)\n\n// InsertIntoTasks inserts one or more rows into tasks table\nfunc (pdb *db) InsertIntoTasks(ctx context.Context, rows []sqlplugin.TasksRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\treturn pdb.driver.NamedExecContext(ctx, rows[0].ShardID, createTaskQry, rows)\n}\n\n// SelectFromTasks reads one or more rows from tasks table\nfunc (pdb *db) SelectFromTasks(ctx context.Context, filter *sqlplugin.TasksFilter) ([]sqlplugin.TasksRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.TasksRow\n\tswitch {\n\tcase filter.MaxTaskID != nil:\n\t\terr = pdb.driver.SelectContext(ctx, filter.ShardID, &rows, getTaskMinMaxQry, filter.DomainID,\n\t\t\tfilter.TaskListName, filter.TaskType, *filter.MinTaskID, *filter.MaxTaskID, *filter.PageSize)\n\tdefault:\n\t\terr = pdb.driver.SelectContext(ctx, filter.ShardID, &rows, getTaskMinQry, filter.DomainID,\n\t\t\tfilter.TaskListName, filter.TaskType, *filter.MinTaskID, *filter.PageSize)\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, err\n}\n\n// DeleteFromTasks deletes one or more rows from tasks table\nfunc (pdb *db) DeleteFromTasks(ctx context.Context, filter *sqlplugin.TasksFilter) (sql.Result, error) {\n\tif filter.TaskIDLessThanEquals != nil {\n\t\tif filter.Limit == nil || *filter.Limit == 0 {\n\t\t\treturn nil, fmt.Errorf(\"missing limit parameter\")\n\t\t}\n\t\treturn pdb.driver.ExecContext(ctx, filter.ShardID, rangeDeleteTaskQry,\n\t\t\tfilter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskIDLessThanEquals, *filter.Limit)\n\t}\n\treturn pdb.driver.ExecContext(ctx, filter.ShardID, deleteTaskQry, filter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskID)\n}\n\nfunc (pdb *db) GetTasksCount(ctx context.Context, filter *sqlplugin.TasksFilter) (int64, error) {\n\tvar size []int64\n\tif err := pdb.driver.SelectContext(ctx, filter.ShardID, &size, getTasksCountQry, filter.DomainID, filter.TaskListName, filter.TaskType, *filter.MinTaskID); err != nil {\n\t\treturn 0, err\n\t}\n\treturn size[0], nil\n}\n\nfunc (pdb *db) GetOrphanTasks(ctx context.Context, filter *sqlplugin.OrphanTasksFilter) ([]sqlplugin.TaskKeyRow, error) {\n\tif filter.Limit == nil || *filter.Limit == 0 {\n\t\treturn nil, fmt.Errorf(\"missing limit parameter\")\n\t}\n\tvar rows []sqlplugin.TaskKeyRow\n\terr := pdb.driver.SelectContext(ctx, sqlplugin.DbAllShards, &rows, getOrphanTaskQry, *filter.Limit)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn rows, nil\n}\n\n// InsertIntoTaskLists inserts one or more rows into task_lists table\nfunc (pdb *db) InsertIntoTaskLists(ctx context.Context, row *sqlplugin.TaskListsRow) (sql.Result, error) {\n\treturn pdb.driver.NamedExecContext(ctx, row.ShardID, createTaskListQry, row)\n}\n\n// UpdateTaskLists updates a row in task_lists table\nfunc (pdb *db) UpdateTaskLists(ctx context.Context, row *sqlplugin.TaskListsRow) (sql.Result, error) {\n\treturn pdb.driver.NamedExecContext(ctx, row.ShardID, updateTaskListQry, row)\n}\n\n// SelectFromTaskLists reads one or more rows from task_lists table\nfunc (pdb *db) SelectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tswitch {\n\tcase filter.DomainID != nil && filter.Name != nil && filter.TaskType != nil:\n\t\treturn pdb.selectFromTaskLists(ctx, filter)\n\tcase filter.DomainIDGreaterThan != nil && filter.NameGreaterThan != nil && filter.TaskTypeGreaterThan != nil && filter.PageSize != nil:\n\t\treturn pdb.rangeSelectFromTaskLists(ctx, filter)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid set of query filter params\")\n\t}\n}\n\nfunc (pdb *db) selectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tvar err error\n\tvar row sqlplugin.TaskListsRow\n\terr = pdb.driver.GetContext(ctx, filter.ShardID, &row, getTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn []sqlplugin.TaskListsRow{row}, err\n}\n\nfunc (pdb *db) rangeSelectFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) ([]sqlplugin.TaskListsRow, error) {\n\tvar err error\n\tvar rows []sqlplugin.TaskListsRow\n\terr = pdb.driver.SelectContext(ctx, filter.ShardID, &rows, listTaskListQry,\n\t\tfilter.ShardID, *filter.DomainIDGreaterThan, *filter.NameGreaterThan, *filter.TaskTypeGreaterThan, *filter.PageSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].ShardID = filter.ShardID\n\t}\n\treturn rows, nil\n}\n\n// DeleteFromTaskLists deletes a row from task_lists table\nfunc (pdb *db) DeleteFromTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) (sql.Result, error) {\n\treturn pdb.driver.ExecContext(ctx, filter.ShardID, deleteTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType, *filter.RangeID)\n}\n\n// LockTaskLists locks a row in task_lists table\nfunc (pdb *db) LockTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) (int64, error) {\n\tvar rangeID int64\n\terr := pdb.driver.GetContext(ctx, filter.ShardID, &rangeID, lockTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType)\n\treturn rangeID, err\n}\n\n// InsertIntoTasksWithTTL is not supported in Postgres\nfunc (pdb *db) InsertIntoTasksWithTTL(_ context.Context, _ []sqlplugin.TasksRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// InsertIntoTaskListsWithTTL is not supported in Postgres\nfunc (pdb *db) InsertIntoTaskListsWithTTL(_ context.Context, _ *sqlplugin.TaskListsRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n\n// UpdateTaskListsWithTTL is not supported in Postgres\nfunc (pdb *db) UpdateTaskListsWithTTL(_ context.Context, _ *sqlplugin.TaskListsRowWithTTL) (sql.Result, error) {\n\treturn nil, sqlplugin.ErrTTLNotSupported\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/typeconv.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport \"time\"\n\nvar localZone, _ = time.Now().Zone()\nvar localOffset = getLocalOffset()\n\ntype (\n\t// DataConverter defines the API for conversions to/from\n\t// go types to postgres datatypes\n\t// TODO https://github.com/uber/cadence/issues/2892\n\t// There are some reasons:\n\t// r application layer is not consistent with timezone: for example,\n\t// in some case we write timestamp with local timezone but when the time.Time\n\t// is converted from \"JSON\"(from paging token), the timezone is missing\n\tDataConverter interface {\n\t\tToPostgresDateTime(t time.Time) time.Time\n\t\tFromPostgresDateTime(t time.Time) time.Time\n\t}\n\tconverter struct{}\n)\n\n// ToPostgresDateTime converts to time to Postgres datetime\nfunc (c *converter) ToPostgresDateTime(t time.Time) time.Time {\n\tzn, _ := t.Zone()\n\tif zn != localZone {\n\t\tnano := t.UnixNano()\n\t\tt := time.Unix(0, nano)\n\t\treturn t\n\t}\n\treturn t\n}\n\n// FromPostgresDateTime converts postgres datetime and returns go time\nfunc (c *converter) FromPostgresDateTime(t time.Time) time.Time {\n\treturn t.Add(-localOffset)\n}\n\nfunc getLocalOffset() time.Duration {\n\t_, offsetSecs := time.Now().Zone()\n\treturn time.Duration(offsetSecs) * time.Second\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/postgres/visibility.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateCreateWorkflowExecutionStarted = `INSERT INTO executions_visibility (` +\n\t\t`domain_id, workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time) ` +\n\t\t`VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15)\n         ON CONFLICT (domain_id, run_id) DO NOTHING`\n\n\ttemplateCreateWorkflowExecutionClosed = `INSERT INTO executions_visibility (` +\n\t\t`domain_id, workflow_id, run_id, start_time, execution_time, workflow_type_name, close_time, close_status, history_length, memo, encoding, is_cron, num_clusters, update_time, shard_id, execution_status, cron_schedule, scheduled_execution_time) ` +\n\t\t`VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18)\n\t\tON CONFLICT (domain_id, run_id) DO UPDATE\n\t\t  SET workflow_id = excluded.workflow_id,\n\t\t      start_time = excluded.start_time,\n\t\t      execution_time = excluded.execution_time,\n              workflow_type_name = excluded.workflow_type_name,\n\t\t\t  close_time = excluded.close_time,\n\t\t\t  close_status = excluded.close_status,\n\t\t\t  history_length = excluded.history_length,\n\t\t\t  memo = excluded.memo,\n\t\t\t  encoding = excluded.encoding,\n\t\t\t\tis_cron = excluded.is_cron,\n\t\t\t\tnum_clusters = excluded.num_clusters,\n\t\t\t\tupdate_time = excluded.update_time,\n\t\t\t\tshard_id = excluded.shard_id,\n\t\t\t\texecution_status = excluded.execution_status,\n\t\t\t\tcron_schedule = excluded.cron_schedule,\n\t\t\t\tscheduled_execution_time = excluded.scheduled_execution_time`\n\n\t// RunID condition is needed for correct pagination\n\ttemplateConditions1 = ` AND domain_id = $1\n\t\t AND start_time >= $2\n\t\t AND start_time <= $3\n \t\t AND (run_id > $4 OR start_time < $5)\n         ORDER BY start_time DESC, run_id\n         LIMIT $6`\n\n\ttemplateConditions2 = ` AND domain_id = $2\n\t\t AND start_time >= $3\n\t\t AND start_time <= $4\n \t\t AND (run_id > $5 OR start_time < $6)\n         ORDER BY start_time DESC, run_id\n         LIMIT $7`\n\n\ttemplateOpenFieldNames = `workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, is_cron, update_time, shard_id`\n\ttemplateOpenSelect     = `SELECT ` + templateOpenFieldNames + ` FROM executions_visibility WHERE close_status IS NULL `\n\n\ttemplateClosedSelect = `SELECT ` + templateOpenFieldNames + `, close_time, close_status, history_length\n\t\t FROM executions_visibility WHERE close_status IS NOT NULL `\n\n\ttemplateGetOpenWorkflowExecutions = templateOpenSelect + templateConditions1\n\n\ttemplateGetClosedWorkflowExecutions = templateClosedSelect + templateConditions1\n\n\ttemplateGetOpenWorkflowExecutionsByType = templateOpenSelect + `AND workflow_type_name = $1` + templateConditions2\n\n\ttemplateGetClosedWorkflowExecutionsByType = templateClosedSelect + `AND workflow_type_name = $1` + templateConditions2\n\n\ttemplateGetOpenWorkflowExecutionsByID = templateOpenSelect + `AND workflow_id = $1` + templateConditions2\n\n\ttemplateGetClosedWorkflowExecutionsByID = templateClosedSelect + `AND workflow_id = $1` + templateConditions2\n\n\ttemplateGetClosedWorkflowExecutionsByStatus = templateClosedSelect + `AND close_status = $1` + templateConditions2\n\n\ttemplateGetClosedWorkflowExecution = `SELECT workflow_id, run_id, start_time, execution_time, memo, encoding, close_time, workflow_type_name, close_status, history_length, is_cron, update_time, shard_id\n\t\t FROM executions_visibility\n\t\t WHERE domain_id = $1 AND close_status IS NOT NULL\n\t\t AND run_id = $2`\n\n\ttemplateDeleteWorkflowExecution = \"DELETE FROM executions_visibility WHERE domain_id=$1 AND run_id=$2\"\n)\n\nvar errCloseParams = errors.New(\"missing one of {closeStatus, closeTime, historyLength} params\")\n\n// InsertIntoVisibility inserts a row into visibility table. If an row already exist,\n// its left as such and no update will be made\nfunc (pdb *db) InsertIntoVisibility(ctx context.Context, row *sqlplugin.VisibilityRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(row.DomainID, pdb.GetTotalNumDBShards())\n\trow.StartTime = pdb.converter.ToPostgresDateTime(row.StartTime)\n\tscheduledExecutionTime := pdb.converter.ToPostgresDateTime(row.ScheduledExecutionTime)\n\treturn pdb.driver.ExecContext(ctx, dbShardID, templateCreateWorkflowExecutionStarted,\n\t\trow.DomainID,\n\t\trow.WorkflowID,\n\t\trow.RunID,\n\t\trow.StartTime,\n\t\trow.ExecutionTime,\n\t\trow.WorkflowTypeName,\n\t\trow.Memo,\n\t\trow.Encoding,\n\t\trow.IsCron,\n\t\trow.NumClusters,\n\t\trow.UpdateTime,\n\t\trow.ShardID,\n\t\trow.ExecutionStatus,\n\t\trow.CronSchedule,\n\t\tscheduledExecutionTime)\n}\n\n// ReplaceIntoVisibility replaces an existing row if it exist or creates a new row in visibility table\nfunc (pdb *db) ReplaceIntoVisibility(ctx context.Context, row *sqlplugin.VisibilityRow) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(row.DomainID, pdb.GetTotalNumDBShards())\n\tswitch {\n\tcase row.CloseStatus != nil && row.CloseTime != nil && row.HistoryLength != nil:\n\t\trow.StartTime = pdb.converter.ToPostgresDateTime(row.StartTime)\n\t\tcloseTime := pdb.converter.ToPostgresDateTime(*row.CloseTime)\n\t\tscheduledExecutionTime := pdb.converter.ToPostgresDateTime(row.ScheduledExecutionTime)\n\t\treturn pdb.driver.ExecContext(ctx, dbShardID, templateCreateWorkflowExecutionClosed,\n\t\t\trow.DomainID,\n\t\t\trow.WorkflowID,\n\t\t\trow.RunID,\n\t\t\trow.StartTime,\n\t\t\trow.ExecutionTime,\n\t\t\trow.WorkflowTypeName,\n\t\t\tcloseTime,\n\t\t\t*row.CloseStatus,\n\t\t\t*row.HistoryLength,\n\t\t\trow.Memo,\n\t\t\trow.Encoding,\n\t\t\trow.IsCron,\n\t\t\trow.NumClusters,\n\t\t\trow.UpdateTime,\n\t\t\trow.ShardID,\n\t\t\trow.ExecutionStatus,\n\t\t\trow.CronSchedule,\n\t\t\tscheduledExecutionTime)\n\tdefault:\n\t\treturn nil, errCloseParams\n\t}\n}\n\n// DeleteFromVisibility deletes a row from visibility table if it exist\nfunc (pdb *db) DeleteFromVisibility(ctx context.Context, filter *sqlplugin.VisibilityFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(filter.DomainID, pdb.GetTotalNumDBShards())\n\treturn pdb.driver.ExecContext(ctx, dbShardID, templateDeleteWorkflowExecution, filter.DomainID, filter.RunID)\n}\n\n// SelectFromVisibility reads one or more rows from visibility table\nfunc (pdb *db) SelectFromVisibility(ctx context.Context, filter *sqlplugin.VisibilityFilter) ([]sqlplugin.VisibilityRow, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(filter.DomainID, pdb.GetTotalNumDBShards())\n\tvar err error\n\tvar rows []sqlplugin.VisibilityRow\n\tif filter.MinStartTime != nil {\n\t\t*filter.MinStartTime = pdb.converter.ToPostgresDateTime(*filter.MinStartTime)\n\t}\n\tif filter.MaxStartTime != nil {\n\t\t*filter.MaxStartTime = pdb.converter.ToPostgresDateTime(*filter.MaxStartTime)\n\t}\n\tswitch {\n\tcase filter.MinStartTime == nil && filter.RunID != nil && filter.Closed:\n\t\tvar row sqlplugin.VisibilityRow\n\t\terr = pdb.driver.GetContext(ctx, dbShardID, &row, templateGetClosedWorkflowExecution, filter.DomainID, *filter.RunID)\n\t\tif err == nil {\n\t\t\trows = append(rows, row)\n\t\t}\n\tcase filter.MinStartTime != nil && filter.WorkflowID != nil:\n\t\tqry := templateGetOpenWorkflowExecutionsByID\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutionsByID\n\t\t}\n\t\terr = pdb.driver.SelectContext(ctx, dbShardID, &rows,\n\t\t\tqry,\n\t\t\t*filter.WorkflowID,\n\t\t\tfilter.DomainID,\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MinStartTime),\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\t*filter.MinStartTime,\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil && filter.WorkflowTypeName != nil:\n\t\tqry := templateGetOpenWorkflowExecutionsByType\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutionsByType\n\t\t}\n\t\terr = pdb.driver.SelectContext(ctx, dbShardID, &rows,\n\t\t\tqry,\n\t\t\t*filter.WorkflowTypeName,\n\t\t\tfilter.DomainID,\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MinStartTime),\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\t*filter.MaxStartTime,\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil && filter.CloseStatus != nil:\n\t\terr = pdb.driver.SelectContext(ctx, dbShardID, &rows,\n\t\t\ttemplateGetClosedWorkflowExecutionsByStatus,\n\t\t\t*filter.CloseStatus,\n\t\t\tfilter.DomainID,\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MinStartTime),\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MaxStartTime),\n\t\t\t*filter.RunID,\n\t\t\tpdb.converter.ToPostgresDateTime(*filter.MaxStartTime),\n\t\t\t*filter.PageSize)\n\tcase filter.MinStartTime != nil:\n\t\tqry := templateGetOpenWorkflowExecutions\n\t\tif filter.Closed {\n\t\t\tqry = templateGetClosedWorkflowExecutions\n\t\t}\n\t\tminSt := pdb.converter.ToPostgresDateTime(*filter.MinStartTime)\n\t\tmaxSt := pdb.converter.ToPostgresDateTime(*filter.MaxStartTime)\n\t\terr = pdb.driver.SelectContext(ctx, dbShardID, &rows,\n\t\t\tqry,\n\t\t\tfilter.DomainID,\n\t\t\tminSt,\n\t\t\tmaxSt,\n\t\t\t*filter.RunID,\n\t\t\tmaxSt,\n\t\t\t*filter.PageSize)\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid query filter\")\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfor i := range rows {\n\t\trows[i].StartTime = pdb.converter.FromPostgresDateTime(rows[i].StartTime)\n\t\trows[i].ExecutionTime = pdb.converter.FromPostgresDateTime(rows[i].ExecutionTime)\n\t\tif rows[i].CloseTime != nil {\n\t\t\tcloseTime := pdb.converter.FromPostgresDateTime(*rows[i].CloseTime)\n\t\t\trows[i].CloseTime = &closeTime\n\t\t}\n\t\trows[i].RunID = strings.TrimSpace(rows[i].RunID)\n\t\trows[i].WorkflowID = strings.TrimSpace(rows[i].WorkflowID)\n\t}\n\treturn rows, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/admin.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tlistTablesQuery = \"SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'\"\n)\n\n// CreateDatabase is not supported by sqlite\n// each sqlite file is a database\nfunc (mdb *DB) CreateDatabase(name string) error {\n\treturn errors.New(\"sqlite doesn't support creating database\")\n}\n\n// DropDatabase is not supported by sqlite\n// each sqlite file is a database\nfunc (mdb *DB) DropDatabase(name string) error {\n\treturn errors.New(\"sqlite doesn't support dropping database\")\n}\n\n// ListTables returns a list of tables in this database\nfunc (mdb *DB) ListTables(_ string) ([]string, error) {\n\tvar tables []string\n\terr := mdb.driver.SelectForSchemaQuery(sqlplugin.DbShardUndefined, &tables, listTablesQuery)\n\treturn tables, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/db.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"\n\n\t// import sqlite driver\n\t_ \"github.com/ncruces/go-sqlite3/driver\"\n\t// import embed sqlite db\n\t_ \"github.com/ncruces/go-sqlite3/embed\"\n)\n\nvar (\n\t_ sqlplugin.AdminDB = (*DB)(nil)\n\t_ sqlplugin.DB      = (*DB)(nil)\n\t_ sqlplugin.Tx      = (*DB)(nil)\n)\n\n// DB contains methods for managing objects in a sqlite database\n// It inherits methods from the mysql.DB to reuse the implementation of the methods\n// sqlplugin.ErrorChecker is customized for sqlite\ntype DB struct {\n\t*mysql.DB\n\n\tconverter    mysql.DataConverter\n\tdriver       sqldriver.Driver\n\toriginalDBs  []*sqlx.DB\n\tnumDBShards  int\n\tdatabaseName string\n}\n\n// NewDB returns an instance of DB, which contains a new created mysql.DB with sqlite specific methods\nfunc NewDB(xdbs []*sqlx.DB, tx *sqlx.Tx, dbShardID int, numDBShards int, dataConverter mysql.DataConverter, databaseName string) (*DB, error) {\n\tdriver, err := sqldriver.NewDriver(xdbs, tx, dbShardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &DB{\n\t\tDB:           mysql.NewDBWithDriver(xdbs, driver, numDBShards, dataConverter),\n\t\tdriver:       driver,\n\t\toriginalDBs:  xdbs,\n\t\tnumDBShards:  numDBShards,\n\t\tconverter:    dataConverter,\n\t\tdatabaseName: databaseName,\n\t}, nil\n}\n\n// PluginName returns the name of the plugin\nfunc (mdb *DB) PluginName() string {\n\treturn PluginName\n}\n\n// BeginTx starts a new transaction and returns a new Tx\nfunc (mdb *DB) BeginTx(ctx context.Context, dbShardID int) (sqlplugin.Tx, error) {\n\txtx, err := mdb.driver.BeginTxx(ctx, dbShardID, nil)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn NewDB(mdb.originalDBs, xtx, dbShardID, mdb.numDBShards, mdb.converter, mdb.databaseName)\n}\n\nfunc (mdb *DB) Close() error {\n\tif mdb.databaseName == \"\" {\n\t\treturn mdb.DB.Close()\n\t}\n\n\treturn closeSharedDBConn(mdb.databaseName, mdb.DB.Close)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/db_pool.go",
    "content": "package sqlite\n\nimport (\n\t\"sync\"\n\n\t\"github.com/jmoiron/sqlx\"\n)\n\n// SQLite database takes exclusive access to the database file during write operations.\n// If another process attempts to perform a write operation, it receives a \"database is locked\" error.\n// To ensure only one database connection is created per database file within the running process,\n// we use dbPool to track and reuse database connections.\n// When a new database connection is requested, it increments dbPoolCounter and returns the existing sql.DB\n// from dbPool. When a connection is requested to be closed, it decrements dbPoolCounter.\n// Once the counter reaches 0, the database connection is closed as there are no more references to it.\n// Reference: https://github.com/mattn/go-sqlite3/issues/274#issuecomment-232942571\nvar (\n\tdbPool        = make(map[string]*sqlx.DB)\n\tdbPoolCounter = make(map[string]int)\n\tdbPoolMx      sync.Mutex\n)\n\n// createSharedDBConn creates a new database connection in the dbPool if it doesn't exist.\nfunc createSharedDBConn(databaseName string, createDBConnFn func() (*sqlx.DB, error)) (*sqlx.DB, error) {\n\tdbPoolMx.Lock()\n\tdefer dbPoolMx.Unlock()\n\n\tif db, ok := dbPool[databaseName]; ok {\n\t\tdbPoolCounter[databaseName]++\n\t\treturn db, nil\n\t}\n\n\tdb, err := createDBConnFn()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdbPool[databaseName] = db\n\tdbPoolCounter[databaseName]++\n\n\treturn db, nil\n}\n\n// closeSharedDBConn closes the database connection in the dbPool if it exists.\nfunc closeSharedDBConn(databaseName string, closeDBConnFn func() error) error {\n\tdbPoolMx.Lock()\n\tdefer dbPoolMx.Unlock()\n\n\tdbPoolCounter[databaseName]--\n\tif dbPoolCounter[databaseName] != 0 {\n\t\treturn nil\n\t}\n\n\tdelete(dbPool, databaseName)\n\tdelete(dbPoolCounter, databaseName)\n\treturn closeDBConnFn()\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/domain.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tlockDomainMetadataQuery = `SELECT notification_version FROM domain_metadata`\n)\n\n// LockDomainMetadata acquires a write lock on a single row in domain_metadata table\nfunc (mdb *DB) LockDomainMetadata(ctx context.Context) error {\n\tvar row sqlplugin.DomainMetadataRow\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &row.NotificationVersion, lockDomainMetadataQuery)\n\treturn err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/dsn.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"bytes\"\n\t\"fmt\"\n\t\"sort\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nconst (\n\t// if journal mode is not provided, we set it to WAL by default\n\t// WAL mode allows readers and writers from different processes\n\t// to access the database concurrently by default\n\t// https://sqlite.org/pragma.html#pragma_journal_mode\n\t// https://www.sqlite.org/wal.html\n\tpragmaJournalModeAttrName     = \"_pragma.journal_mode\"\n\tpragmaJournalModeDefaultValue = \"WAL\"\n\n\t// if busy_timeout is not provided, we set it to 60 seconds by default\n\t// https://sqlite.org/pragma.html#pragma_busy_timeout\n\tpragmaBusyTimeoutAttrName = \"_pragma.busy_timeout\"\n\tpragmaBusyTimeoutDefault  = \"60000\"\n)\n\nconst (\n\t// pragmaKey is the key is used to set pragma arguments in the DSN\n\tpragmaKey = \"_pragma\"\n\n\t// pragmaPrefix is the prefix used to identify pragma arguments in the config\n\tpragmaPrefix = \"_pragma.\"\n)\n\n// buildDSN builds the data source name for sqlite from config.SQL\n// If DatabaseName is not provided, then sqlite will use in-memory database, otherwise it will use the file as the database\n// All dsn attributes can be set up in the ConnectAttributes field of the config.SQL\n// Available attributes can be found here: https://github.com/ncruces/go-sqlite3/blob/main/driver/driver.go\n// PRAGMA attributes should start with \"_pragma.\" prefix\n// example: \"_pragma.journal_mode\":\"wal\" will be transformed to \"_pragma=journal_mode(wal)\"\n// More about PRAGMA attributes: https://sqlite.org/pragma.html\n// Default PRAGMA values if not provided:\n// - journal_mode: WAL (file only) \t- https://sqlite.org/pragma.html#pragma_journal_mode\n// - busy_timeout: 60 seconds \t\t- https://sqlite.org/pragma.html#pragma_busy_timeout\nfunc buildDSN(cfg *config.SQL) string {\n\n\t// by default, we use in-memory database if no database name is provided\n\tvar dsn = \"file::memory:\"\n\n\t// if database name is provided, then sqlite will use the file as the database\n\tif cfg.DatabaseName != \"\" {\n\t\tdsn = fmt.Sprintf(\"file:%s\", cfg.DatabaseName)\n\n\t}\n\n\tif dsnAttrs := buildDSNAttrs(cfg); dsnAttrs != \"\" {\n\t\tdsn += \"?\" + dsnAttrs\n\t}\n\n\treturn dsn\n}\n\n// buildDSNAttrs builds the data source name attributes for sqlite from config.SQL\nfunc buildDSNAttrs(cfg *config.SQL) string {\n\n\tsanitizedAttrs := sanitizeDSNAttrs(cfg.ConnectAttributes)\n\n\tif cfg.DatabaseName != \"\" {\n\t\tdefaultIfEmpty(sanitizedAttrs, pragmaJournalModeAttrName, pragmaJournalModeDefaultValue)\n\t}\n\n\tdefaultIfEmpty(sanitizedAttrs, pragmaBusyTimeoutAttrName, pragmaBusyTimeoutDefault)\n\treturn joinDSNAttrs(sanitizedAttrs)\n}\n\n// defaultIfEmpty sets the value to the key if the key is not present in the attributes\nfunc defaultIfEmpty(attrs map[string]string, key, value string) {\n\tif hasAttr(attrs, key) {\n\t\treturn\n\t}\n\tattrs[key] = value\n}\n\n// hasAttr checks if the attributes map has any of the keys\nfunc hasAttr(attrs map[string]string, keys ...string) bool {\n\tfor key := range attrs {\n\t\tfor _, k := range keys {\n\t\t\tif key == k {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t}\n\treturn false\n}\n\n// sanitizeDSNAttrs sanitizes the attributes by trimming the keys and values\nfunc sanitizeDSNAttrs(attrs map[string]string) map[string]string {\n\tsanitized := make(map[string]string, len(attrs))\n\n\tfor k, v := range attrs {\n\t\tk, v = sanitizeDSNAttrKey(k), sanitizeDSNAttrValue(v)\n\t\tsanitized[k] = v\n\t}\n\n\treturn sanitized\n}\n\n// isPragmaKey checks if the key is a pragma key\nfunc isPragmaKey(key string) bool {\n\treturn strings.HasPrefix(key, pragmaPrefix)\n}\n\n// transformPragmaArgument transforms the pragma argument to the format that can be used in the DSN\n// example: \"_pragma.journal_mode\":\"wal\" -> \"_pragma=journal_mode(wal)\"\nfunc transformPragmaArgument(key, value string) (newKey, newValue string) {\n\treturn pragmaKey, fmt.Sprintf(\"%s(%s)\", strings.TrimPrefix(key, pragmaPrefix), value)\n}\n\n// sanitizeDSNAttrElem trims the value, lowercases it\nfunc sanitizeDSNAttrKey(v string) string {\n\treturn strings.TrimSpace(strings.ToLower(v))\n}\n\n// sanitizeDSNAttrElem trims the value, lowercases it\nfunc sanitizeDSNAttrValue(v string) string {\n\treturn strings.TrimSpace(v)\n}\n\n// sortedKeys returns the sorted keys of the map\nfunc sortMapKeys(m map[string]string) []string {\n\tkeys := make([]string, 0, len(m))\n\tfor k := range m {\n\t\tkeys = append(keys, k)\n\t}\n\n\tsort.Strings(keys)\n\treturn keys\n}\n\n// joinDSNAttrs joins the attributes into a single string\n// with key=value pairs separated by & and escaped\nfunc joinDSNAttrs(attrs map[string]string) string {\n\tfirst := true\n\tvar buf bytes.Buffer\n\n\t// sort the keys to make the order of the attributes deterministic\n\tsortedKeys := sortMapKeys(attrs)\n\n\tfor _, k := range sortedKeys {\n\t\tv := attrs[k]\n\n\t\t// pragma arguments should be transformed to the format that can be used in the DSN\n\t\tif isPragmaKey(k) {\n\t\t\tk, v = transformPragmaArgument(k, v)\n\t\t}\n\n\t\tif !first {\n\t\t\tbuf.WriteString(\"&\")\n\t\t}\n\t\tfirst = false\n\t\tbuf.WriteString(k)\n\t\tbuf.WriteString(\"=\")\n\t\tbuf.WriteString(v)\n\t}\n\treturn buf.String()\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/dsn_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc Test_buildDSN(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcfg  *config.SQL\n\t\twant string\n\t}{\n\t\t\"empty\": {\n\t\t\tcfg:  &config.SQL{},\n\t\t\twant: \"file::memory:?_pragma=busy_timeout(60000)\",\n\t\t},\n\t\t\"database name only\": {\n\t\t\tcfg: &config.SQL{\n\t\t\t\tDatabaseName: \"cadence.db\",\n\t\t\t},\n\t\t\twant: \"file:cadence.db?_pragma=busy_timeout(60000)&_pragma=journal_mode(WAL)\",\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tdsn := buildDSN(c.cfg)\n\t\t\tassert.Equal(t, c.want, dsn)\n\t\t})\n\t}\n}\n\nfunc Test_buildDSN_attrs(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcfg  *config.SQL\n\t\twant string\n\t}{\n\t\t\"only connection attrs\": {\n\t\t\tcfg: &config.SQL{\n\t\t\t\tConnectAttributes: map[string]string{\n\t\t\t\t\t\"_busy_timeout\": \"10\",\n\t\t\t\t\t\"_FK\":           \"true\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: \"file::memory:?_busy_timeout=10&_fk=true&_pragma=busy_timeout(60000)\",\n\t\t},\n\t\t\"database name and connection attrs\": {\n\t\t\tcfg: &config.SQL{\n\t\t\t\tDatabaseName: \"cadence.db\",\n\t\t\t\tConnectAttributes: map[string]string{\n\t\t\t\t\t\"cache1 \": \"NONe \",\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: \"file:cadence.db?_pragma=busy_timeout(60000)&_pragma=journal_mode(WAL)&cache1=NONe\",\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tdsn := buildDSN(c.cfg)\n\t\t\tassert.Contains(t, c.want, dsn)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/error_checker.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"errors\"\n\n\t\"github.com/ncruces/go-sqlite3\"\n)\n\n// IsDupEntryError verify if the error is a duplicate entry error\nfunc (mdb *DB) IsDupEntryError(err error) bool {\n\tvar sqlErr *sqlite3.Error\n\tif ok := errors.As(err, &sqlErr); !ok {\n\t\treturn false\n\t}\n\n\tswitch sqlErr.ExtendedCode() {\n\tcase\n\t\t// https://sqlite.org/rescode.html#constraint_unique\n\t\tsqlite3.CONSTRAINT_UNIQUE,\n\n\t\t// https://sqlite.org/rescode.html#constraint_primarykey\n\t\tsqlite3.CONSTRAINT_PRIMARYKEY:\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// IsNotFoundError verify if the error is a not found error\nfunc (mdb *DB) IsNotFoundError(err error) bool {\n\treturn errors.Is(err, sql.ErrNoRows)\n}\n\n// IsTimeoutError verify if the error is a timeout error\nfunc (mdb *DB) IsTimeoutError(err error) bool {\n\tif errors.Is(err, context.DeadlineExceeded) {\n\t\treturn true\n\t}\n\n\tvar sqlErr *sqlite3.Error\n\tif ok := errors.As(err, &sqlErr); !ok {\n\t\treturn false\n\t}\n\n\t// https://sqlite.org/rescode.html#busy_timeout\n\tif sqlErr.Timeout() {\n\t\treturn true\n\t}\n\n\t// https://sqlite.org/rescode.html#interrupt\n\tif sqlErr.Code() == sqlite3.INTERRUPT {\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// IsThrottlingError verify if the error is a throttling error\nfunc (mdb *DB) IsThrottlingError(_ error) bool {\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/events.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tdeleteHistoryNodesQuery = `WITH events_to_delete AS (\n    SELECT shard_id, tree_id, branch_id, node_id, txn_id\n    FROM history_node\n    WHERE shard_id = ? AND tree_id = ? AND branch_id = ? AND node_id >= ?\n    ORDER BY shard_id, tree_id, branch_id, node_id, txn_id\n    LIMIT ?\n)\n\nDELETE FROM history_node\nWHERE (shard_id, tree_id, branch_id, node_id, txn_id) IN (SELECT shard_id, tree_id, branch_id, node_id, txn_id FROM events_to_delete);`\n)\n\n// DeleteFromHistoryNode deletes one or more rows from history_node table\nfunc (mdb *DB) DeleteFromHistoryNode(ctx context.Context, filter *sqlplugin.HistoryNodeFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromTreeID(filter.TreeID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx, dbShardID, deleteHistoryNodesQuery, filter.ShardID, filter.TreeID, filter.BranchID, *filter.MinNodeID, filter.PageSize)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/execution.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tlockExecutionQueryBase = `SELECT next_event_id FROM executions\n WHERE shard_id = ? AND domain_id = ? AND workflow_id = ? AND run_id = ?`\n\twriteLockExecutionQuery = lockExecutionQueryBase\n\treadLockExecutionQuery  = lockExecutionQueryBase\n\n\tlockCurrentExecutionJoinExecutionsQuery = `SELECT\nce.shard_id, ce.domain_id, ce.workflow_id, ce.run_id, ce.create_request_id, ce.state, ce.close_status, ce.start_version, e.last_write_version\nFROM current_executions ce\nINNER JOIN executions e ON e.shard_id = ce.shard_id AND e.domain_id = ce.domain_id AND e.workflow_id = ce.workflow_id AND e.run_id = ce.run_id\nWHERE ce.shard_id = ? AND ce.domain_id = ? AND ce.workflow_id = ?`\n\n\tgetCurrentExecutionQuery = `SELECT\nshard_id, domain_id, workflow_id, run_id, create_request_id, state, close_status, start_version, last_write_version\nFROM current_executions WHERE shard_id = ? AND domain_id = ? AND workflow_id = ?`\n\tlockCurrentExecutionQuery = getCurrentExecutionQuery\n\n\trangeDeleteTransferTaskQuery        = `DELETE FROM transfer_tasks WHERE shard_id = ? AND task_id >= ? AND task_id < ?`\n\trangeDeleteTransferTaskByBatchQuery = `WITH tasks_to_delete AS (\n    SELECT shard_id, task_id\n    FROM transfer_tasks\n    WHERE shard_id = ? AND task_id >= ? AND task_id < ?\n    ORDER BY task_id\n    LIMIT ?\n)\n\nDELETE FROM transfer_tasks\nWHERE (shard_id, task_id) IN (SELECT shard_id, task_id FROM tasks_to_delete);`\n\n\trangeDeleteReplicationTaskQuery        = `DELETE FROM replication_tasks WHERE shard_id = ? AND task_id < ?`\n\trangeDeleteReplicationTaskByBatchQuery = `WITH tasks_to_delete AS (\n    SELECT shard_id, task_id\n    FROM replication_tasks\n    WHERE shard_id = ? AND task_id < ?\n    ORDER BY task_id\n    LIMIT ?\n)\n\nDELETE FROM replication_tasks\nWHERE (shard_id, task_id) IN (SELECT shard_id, task_id FROM tasks_to_delete);`\n\n\trangeDeleteTimerTaskQuery        = `DELETE FROM timer_tasks WHERE shard_id = ? AND visibility_timestamp >= ? AND visibility_timestamp < ?`\n\trangeDeleteTimerTaskByBatchQuery = `WITH tasks_to_delete AS (\n    SELECT shard_id, visibility_timestamp, task_id\n    FROM timer_tasks\n    WHERE shard_id = ? AND visibility_timestamp >= ? AND visibility_timestamp < ?\n    ORDER BY visibility_timestamp,task_id\n    LIMIT ?\n)\n\nDELETE FROM timer_tasks\nWHERE (shard_id, visibility_timestamp, task_id) IN (SELECT shard_id, visibility_timestamp, task_id FROM tasks_to_delete);`\n)\n\n// ReadLockExecutions acquires a write lock on a single row in executions table\nfunc (mdb *DB) ReadLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tvar nextEventID int\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &nextEventID, readLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// WriteLockExecutions acquires a write lock on a single row in executions table\nfunc (mdb *DB) WriteLockExecutions(ctx context.Context, filter *sqlplugin.ExecutionsFilter) (int, error) {\n\tvar nextEventID int\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &nextEventID, writeLockExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID, filter.RunID)\n\treturn nextEventID, err\n}\n\n// LockCurrentExecutionsJoinExecutions joins a row in current_executions with executions table and acquires a\n// write lock on the result\nfunc (mdb *DB) LockCurrentExecutionsJoinExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) ([]sqlplugin.CurrentExecutionsRow, error) {\n\tvar rows []sqlplugin.CurrentExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.SelectContext(ctx, dbShardID, &rows, lockCurrentExecutionJoinExecutionsQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn rows, err\n}\n\n// LockCurrentExecutions acquires a write lock on a single row in current_executions table\nfunc (mdb *DB) LockCurrentExecutions(ctx context.Context, filter *sqlplugin.CurrentExecutionsFilter) (*sqlplugin.CurrentExecutionsRow, error) {\n\tvar row sqlplugin.CurrentExecutionsRow\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\terr := mdb.driver.GetContext(ctx, dbShardID, &row, lockCurrentExecutionQuery, filter.ShardID, filter.DomainID, filter.WorkflowID)\n\treturn &row, err\n}\n\n// RangeDeleteFromTransferTasks deletes multi rows from transfer_tasks table\nfunc (mdb *DB) RangeDeleteFromTransferTasks(ctx context.Context, filter *sqlplugin.TransferTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskByBatchQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTransferTaskQuery, filter.ShardID, filter.InclusiveMinTaskID, filter.ExclusiveMaxTaskID)\n}\n\n// RangeDeleteFromReplicationTasks deletes multi rows from replication_tasks table\nfunc (mdb *DB) RangeDeleteFromReplicationTasks(ctx context.Context, filter *sqlplugin.ReplicationTasksFilter) (sql.Result, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskByBatchQuery, filter.ShardID, filter.ExclusiveMaxTaskID, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteReplicationTaskQuery, filter.ShardID, filter.ExclusiveMaxTaskID)\n}\n\n// RangeDeleteFromTimerTasks deletes multi rows from timer_tasks table\nfunc (mdb *DB) RangeDeleteFromTimerTasks(ctx context.Context, filter *sqlplugin.TimerTasksFilter) (sql.Result, error) {\n\tfilter.MinVisibilityTimestamp = mdb.converter.ToDateTime(filter.MinVisibilityTimestamp)\n\tfilter.MaxVisibilityTimestamp = mdb.converter.ToDateTime(filter.MaxVisibilityTimestamp)\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(filter.ShardID, mdb.GetTotalNumDBShards())\n\tif filter.PageSize > 0 {\n\t\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskByBatchQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp, filter.PageSize)\n\t}\n\treturn mdb.driver.ExecContext(ctx, dbShardID, rangeDeleteTimerTaskQuery, filter.ShardID, filter.MinVisibilityTimestamp, filter.MaxVisibilityTimestamp)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/execution_maps.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tcreateSignalsRequestedSetQry = `INSERT OR IGNORE INTO signals_requested_sets\n(shard_id, domain_id, workflow_id, run_id, signal_id) VALUES\n(:shard_id, :domain_id, :workflow_id, :run_id, :signal_id)`\n)\n\n// InsertIntoSignalsRequestedSets inserts one or more rows into signals_requested_sets table\nfunc (mdb *DB) InsertIntoSignalsRequestedSets(ctx context.Context, rows []sqlplugin.SignalsRequestedSetsRow) (sql.Result, error) {\n\tif len(rows) == 0 {\n\t\treturn nil, nil\n\t}\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(rows[0].ShardID), mdb.GetTotalNumDBShards())\n\treturn mdb.driver.NamedExecContext(ctx, dbShardID, createSignalsRequestedSetQry, rows)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/plugin.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"path\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/iancoleman/strcase\"\n\t\"github.com/jmoiron/sqlx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqldriver\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tPluginName = \"sqlite\"\n)\n\n// SQLite plugin provides an sql persistence storage implementation for sqlite database\n// Mostly the implementation reuses the mysql implementation\n// If DatabaseName is not provided, then sqlite will use in-memory database,\n// otherwise it will use the file as the database\ntype plugin struct{}\n\nvar _ sqlplugin.Plugin = (*plugin)(nil)\n\nfunc init() {\n\tsql.RegisterPlugin(PluginName, &plugin{})\n}\n\n// CreateDB wraps createDB to return an instance of sqlplugin.DB\nfunc (p *plugin) CreateDB(cfg *config.SQL) (sqlplugin.DB, error) {\n\treturn p.createDB(cfg)\n}\n\n// CreateAdminDB wraps createDB to return an instance of sqlplugin.AdminDB\nfunc (p *plugin) CreateAdminDB(cfg *config.SQL) (sqlplugin.AdminDB, error) {\n\treturn p.createDB(cfg)\n}\n\n// createDB create a new instance of DB\nfunc (p *plugin) createDB(cfg *config.SQL) (*DB, error) {\n\tconns, err := sqldriver.CreateDBConnections(cfg, p.createSingleDBConn)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn NewDB(conns, nil, sqlplugin.DbShardUndefined, cfg.NumShards, newConverter(), cfg.DatabaseName)\n}\n\n// createSingleDBConn creates a single database connection for sqlite\n// Plugin respects the following arguments MaxConns, MaxIdleConns, MaxConnLifetime\n// Other arguments are used and described in buildDSN function\nfunc (p *plugin) createSingleDBConn(cfg *config.SQL) (*sqlx.DB, error) {\n\tif cfg.DatabaseName == \"\" {\n\t\treturn p.createDBConn(cfg)\n\t}\n\n\treturn createSharedDBConn(cfg.DatabaseName, func() (*sqlx.DB, error) {\n\t\treturn p.createDBConn(cfg)\n\t})\n}\n\nfunc (p *plugin) createDBConn(cfg *config.SQL) (*sqlx.DB, error) {\n\tdb, err := sqlx.Connect(\"sqlite3\", buildDSN(cfg))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create database connection: %v\", err)\n\t}\n\n\tif cfg.MaxConns > 0 {\n\t\tdb.SetMaxOpenConns(cfg.MaxConns)\n\t}\n\tif cfg.MaxIdleConns > 0 {\n\t\tdb.SetMaxIdleConns(cfg.MaxIdleConns)\n\t}\n\tif cfg.MaxConnLifetime > 0 {\n\t\tdb.SetConnMaxLifetime(cfg.MaxConnLifetime)\n\t}\n\n\t// Maps struct names in CamelCase to snake without need for DB struct tags.\n\tdb.MapperFunc(strcase.ToSnake)\n\treturn db, nil\n}\n\nconst testSchemaDir = \"schema/sqlite\"\n\n// GetTestClusterOption returns a test cluster option for sqlite plugin\n// It uses a temporary directory for the database name\nfunc GetTestClusterOption() *pt.TestBaseOptions {\n\treturn &pt.TestBaseOptions{\n\t\tDBPluginName: PluginName,\n\t\tDBName:       path.Join(os.TempDir(), uuid.New().String()),\n\t\tSchemaDir:    testSchemaDir,\n\t\tStoreType:    config.StoreTypeSQL,\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/plugin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"os\"\n\t\"path\"\n\t\"testing\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc TestPlugin_CreateDB(t *testing.T) {\n\tfor name, cfg := range map[string]*config.SQL{\n\t\t\"in-memory\": {},\n\t\t\"temp file\": {DatabaseName: path.Join(os.TempDir(), uuid.New().String())},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tp := &plugin{}\n\t\t\tdb, err := p.CreateDB(cfg)\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, db)\n\t\t})\n\t}\n}\n\nfunc TestPlugin_CreateAdminDB(t *testing.T) {\n\tfor name, cfg := range map[string]*config.SQL{\n\t\t\"in-memory\": {},\n\t\t\"temp file\": {DatabaseName: path.Join(os.TempDir(), uuid.New().String())},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tp := &plugin{}\n\t\t\tdb, err := p.CreateAdminDB(cfg)\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, db)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/queue.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateGetLastMessageIDQuery = `SELECT message_id FROM queue WHERE queue_type=? ORDER BY message_id DESC LIMIT 1`\n\ttemplateGetQueueMetadataQuery = `SELECT data from queue_metadata WHERE queue_type = ?`\n)\n\n// GetLastEnqueuedMessageIDForUpdate returns the last enqueued message ID\nfunc (mdb *DB) GetLastEnqueuedMessageIDForUpdate(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n) (int64, error) {\n\n\tvar lastMessageID int64\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &lastMessageID, templateGetLastMessageIDQuery, queueType)\n\treturn lastMessageID, err\n}\n\n// GetAckLevels returns ack levels for pulling clusters\nfunc (mdb *DB) GetAckLevels(\n\tctx context.Context,\n\tqueueType persistence.QueueType,\n\tforUpdate bool,\n) (map[string]int64, error) {\n\n\tqueryStr := templateGetQueueMetadataQuery\n\tif forUpdate {\n\t\t// FOR UPDATE is not supported in sqlite\n\t}\n\n\tvar data []byte\n\terr := mdb.driver.GetContext(ctx, sqlplugin.DbDefaultShard, &data, queryStr, queueType)\n\tif err != nil {\n\t\tif err == sql.ErrNoRows {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn nil, err\n\t}\n\n\tvar clusterAckLevels map[string]int64\n\tif err := json.Unmarshal(data, &clusterAckLevels); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn clusterAckLevels, nil\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/shard.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tlockShardQry     = `SELECT range_id FROM shards WHERE shard_id = ?`\n\treadLockShardQry = `SELECT range_id FROM shards WHERE shard_id = ?`\n)\n\n// WriteLockShards acquires a write lock on a single row in shards table\nfunc (mdb *DB) WriteLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := mdb.driver.GetContext(ctx, dbShardID, &rangeID, lockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n\n// ReadLockShards acquires a read lock on a single row in shards table\nfunc (mdb *DB) ReadLockShards(ctx context.Context, filter *sqlplugin.ShardsFilter) (int, error) {\n\tdbShardID := sqlplugin.GetDBShardIDFromHistoryShardID(int(filter.ShardID), mdb.GetTotalNumDBShards())\n\tvar rangeID int\n\terr := mdb.driver.GetContext(ctx, dbShardID, &rangeID, readLockShardQry, filter.ShardID)\n\treturn rangeID, err\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/sqlite_persistence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n)\n\nfunc TestSQLiteHistoryV2PersistenceSuite(t *testing.T) {\n\ts := new(pt.HistoryV2PersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteMatchingPersistenceSuite(t *testing.T) {\n\ts := new(pt.MatchingPersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteMetadataPersistenceSuiteV2(t *testing.T) {\n\ts := new(pt.MetadataPersistenceSuiteV2)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteShardPersistenceSuite(t *testing.T) {\n\ts := new(pt.ShardPersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\ntype ExecutionManagerSuite struct {\n\tpt.ExecutionManagerSuite\n}\n\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in sqlite\")\n}\n\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in sqlite\")\n}\n\nfunc TestSQLiteExecutionManagerSuite(t *testing.T) {\n\ts := new(ExecutionManagerSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteExecutionManagerWithEventsV2(t *testing.T) {\n\ts := new(pt.ExecutionManagerSuiteForEventsV2)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteVisibilityPersistenceSuite(t *testing.T) {\n\ts := new(pt.DBVisibilityPersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteQueuePersistence(t *testing.T) {\n\ts := new(pt.QueuePersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteConfigPersistence(t *testing.T) {\n\ts := new(pt.ConfigStorePersistenceSuite)\n\toption := GetTestClusterOption()\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestSQLiteDomainAuditPersistence(t *testing.T) {\n\tt.Skip(\"DomainAuditPersistence is only implemented for NoSQL/Cassandra in this PR\")\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/task.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\tlockTaskListQry = `SELECT range_id FROM task_lists ` +\n\t\t`WHERE shard_id = ? AND domain_id = ? AND name = ? AND task_type = ?`\n\n\tdeleteTaskQry = `DELETE FROM tasks ` +\n\t\t`WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id = ?`\n\n\trangeDeleteTaskQry = `WITH tasks_to_delete AS (\n    SELECT domain_id,task_list_name,task_type,task_id \n    FROM tasks\n    WHERE domain_id = ? AND task_list_name = ? AND task_type = ? AND task_id <= ?\n    ORDER BY domain_id,task_list_name,task_type,task_id\n    LIMIT ?\n)\n\nDELETE FROM tasks\nWHERE (domain_id,task_list_name,task_type,task_id) IN (SELECT domain_id,task_list_name,task_type,task_id FROM tasks_to_delete);`\n)\n\n// LockTaskLists locks a row in task_lists table\nfunc (mdb *DB) LockTaskLists(ctx context.Context, filter *sqlplugin.TaskListsFilter) (int64, error) {\n\tvar rangeID int64\n\terr := mdb.driver.GetContext(ctx, filter.ShardID, &rangeID, lockTaskListQry, filter.ShardID, *filter.DomainID, *filter.Name, *filter.TaskType)\n\treturn rangeID, err\n}\n\n// DeleteFromTasks deletes one or more rows from tasks table\nfunc (mdb *DB) DeleteFromTasks(ctx context.Context, filter *sqlplugin.TasksFilter) (sql.Result, error) {\n\tif filter.TaskIDLessThanEquals != nil {\n\t\tif filter.Limit == nil || *filter.Limit == 0 {\n\t\t\treturn nil, fmt.Errorf(\"missing limit parameter\")\n\t\t}\n\t\treturn mdb.driver.ExecContext(ctx, filter.ShardID, rangeDeleteTaskQry,\n\t\t\tfilter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskIDLessThanEquals, *filter.Limit)\n\t}\n\treturn mdb.driver.ExecContext(ctx, filter.ShardID, deleteTaskQry, filter.DomainID, filter.TaskListName, filter.TaskType, *filter.TaskID)\n}\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/typeconv.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"time\"\n)\n\n// converter implements mysql.DataConverter\n// SQLite does not require any conversion, so this is a no-op implementation\ntype converter struct{}\n\n// newConverter returns a new instance of converter\nfunc newConverter() *converter {\n\treturn &converter{}\n}\n\n// ToDateTime returns the same time\nfunc (c converter) ToDateTime(t time.Time) time.Time { return t }\n\n// FromDateTime returns the same time\nfunc (c converter) FromDateTime(t time.Time) time.Time { return t }\n"
  },
  {
    "path": "common/persistence/sql/sqlplugin/sqlite/visibility.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nconst (\n\ttemplateCreateWorkflowExecutionStarted = `INSERT OR IGNORE INTO executions_visibility (` +\n\t\t`domain_id, workflow_id, run_id, start_time, execution_time, workflow_type_name, memo, encoding, is_cron, num_clusters, update_time, shard_id) ` +\n\t\t`VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`\n)\n\n// InsertIntoVisibility inserts a row into visibility table. If an row already exist,\n// its left as such and no update will be made\nfunc (mdb *DB) InsertIntoVisibility(ctx context.Context, row *sqlplugin.VisibilityRow) (sql.Result, error) {\n\trow.StartTime = mdb.converter.ToDateTime(row.StartTime)\n\tdbShardID := sqlplugin.GetDBShardIDFromDomainID(row.DomainID, mdb.GetTotalNumDBShards())\n\treturn mdb.driver.ExecContext(ctx,\n\t\tdbShardID,\n\t\ttemplateCreateWorkflowExecutionStarted,\n\t\trow.DomainID,\n\t\trow.WorkflowID,\n\t\trow.RunID,\n\t\trow.StartTime,\n\t\trow.ExecutionTime,\n\t\trow.WorkflowTypeName,\n\t\trow.Memo,\n\t\trow.Encoding,\n\t\trow.IsCron,\n\t\trow.NumClusters,\n\t\trow.UpdateTime,\n\t\trow.ShardID)\n}\n"
  },
  {
    "path": "common/persistence/sql/workflow_state_maps.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc updateActivityInfos(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tactivityInfos []*persistence.InternalActivityInfo,\n\tdeleteInfos []int64,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) error {\n\n\tif len(activityInfos) > 0 {\n\t\trows := make([]sqlplugin.ActivityInfoMapsRow, len(activityInfos))\n\t\tfor i, activityInfo := range activityInfos {\n\t\t\tscheduledEvent, scheduledEncoding := persistence.FromDataBlob(activityInfo.ScheduledEvent)\n\t\t\tstartEvent, startEncoding := persistence.FromDataBlob(activityInfo.StartedEvent)\n\n\t\t\tinfo := &serialization.ActivityInfo{\n\t\t\t\tVersion:                  activityInfo.Version,\n\t\t\t\tScheduledEventBatchID:    activityInfo.ScheduledEventBatchID,\n\t\t\t\tScheduledEvent:           scheduledEvent,\n\t\t\t\tScheduledEventEncoding:   scheduledEncoding,\n\t\t\t\tScheduledTimestamp:       activityInfo.ScheduledTime,\n\t\t\t\tStartedID:                activityInfo.StartedID,\n\t\t\t\tStartedEvent:             startEvent,\n\t\t\t\tStartedEventEncoding:     startEncoding,\n\t\t\t\tStartedTimestamp:         activityInfo.StartedTime,\n\t\t\t\tActivityID:               activityInfo.ActivityID,\n\t\t\t\tRequestID:                activityInfo.RequestID,\n\t\t\t\tScheduleToStartTimeout:   activityInfo.ScheduleToStartTimeout,\n\t\t\t\tScheduleToCloseTimeout:   activityInfo.ScheduleToCloseTimeout,\n\t\t\t\tStartToCloseTimeout:      activityInfo.StartToCloseTimeout,\n\t\t\t\tHeartbeatTimeout:         activityInfo.HeartbeatTimeout,\n\t\t\t\tCancelRequested:          activityInfo.CancelRequested,\n\t\t\t\tCancelRequestID:          activityInfo.CancelRequestID,\n\t\t\t\tTimerTaskStatus:          activityInfo.TimerTaskStatus,\n\t\t\t\tAttempt:                  activityInfo.Attempt,\n\t\t\t\tTaskList:                 activityInfo.TaskList,\n\t\t\t\tTaskListKind:             activityInfo.TaskListKind,\n\t\t\t\tStartedIdentity:          activityInfo.StartedIdentity,\n\t\t\t\tHasRetryPolicy:           activityInfo.HasRetryPolicy,\n\t\t\t\tRetryInitialInterval:     activityInfo.InitialInterval,\n\t\t\t\tRetryBackoffCoefficient:  activityInfo.BackoffCoefficient,\n\t\t\t\tRetryMaximumInterval:     activityInfo.MaximumInterval,\n\t\t\t\tRetryExpirationTimestamp: activityInfo.ExpirationTime,\n\t\t\t\tRetryMaximumAttempts:     activityInfo.MaximumAttempts,\n\t\t\t\tRetryNonRetryableErrors:  activityInfo.NonRetriableErrors,\n\t\t\t\tRetryLastFailureReason:   activityInfo.LastFailureReason,\n\t\t\t\tRetryLastWorkerIdentity:  activityInfo.LastWorkerIdentity,\n\t\t\t\tRetryLastFailureDetails:  activityInfo.LastFailureDetails,\n\t\t\t}\n\t\t\tblob, err := parser.ActivityInfoToBlob(info)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trows[i] = sqlplugin.ActivityInfoMapsRow{\n\t\t\t\tShardID:                  int64(shardID),\n\t\t\t\tDomainID:                 domainID,\n\t\t\t\tWorkflowID:               workflowID,\n\t\t\t\tRunID:                    runID,\n\t\t\t\tScheduleID:               activityInfo.ScheduleID,\n\t\t\t\tLastHeartbeatUpdatedTime: activityInfo.LastHeartBeatUpdatedTime,\n\t\t\t\tLastHeartbeatDetails:     activityInfo.Details,\n\t\t\t\tData:                     blob.Data,\n\t\t\t\tDataEncoding:             string(blob.Encoding),\n\t\t\t}\n\t\t}\n\n\t\tif _, err := tx.ReplaceIntoActivityInfoMaps(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateActivityInfos\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteInfos) > 0 {\n\t\tif _, err := tx.DeleteFromActivityInfoMaps(ctx, &sqlplugin.ActivityInfoMapsFilter{\n\t\t\tShardID:     int64(shardID),\n\t\t\tDomainID:    domainID,\n\t\t\tWorkflowID:  workflowID,\n\t\t\tRunID:       runID,\n\t\t\tScheduleIDs: deleteInfos,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateActivityInfos\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getActivityInfoMap(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) (map[int64]*persistence.InternalActivityInfo, error) {\n\n\trows, err := db.SelectFromActivityInfoMaps(ctx, &sqlplugin.ActivityInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getActivityInfoMap\", \"\", err)\n\t}\n\n\tret := make(map[int64]*persistence.InternalActivityInfo)\n\tfor _, row := range rows {\n\t\tdecoded, err := parser.ActivityInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tinfo := &persistence.InternalActivityInfo{\n\t\t\tDomainID:                 row.DomainID.String(),\n\t\t\tScheduleID:               row.ScheduleID,\n\t\t\tDetails:                  row.LastHeartbeatDetails,\n\t\t\tLastHeartBeatUpdatedTime: row.LastHeartbeatUpdatedTime,\n\t\t\tVersion:                  decoded.GetVersion(),\n\t\t\tScheduledEventBatchID:    decoded.GetScheduledEventBatchID(),\n\t\t\tScheduledEvent:           persistence.NewDataBlob(decoded.ScheduledEvent, constants.EncodingType(decoded.GetScheduledEventEncoding())),\n\t\t\tScheduledTime:            decoded.GetScheduledTimestamp(),\n\t\t\tStartedID:                decoded.GetStartedID(),\n\t\t\tStartedTime:              decoded.GetStartedTimestamp(),\n\t\t\tActivityID:               decoded.GetActivityID(),\n\t\t\tRequestID:                decoded.GetRequestID(),\n\t\t\tScheduleToStartTimeout:   decoded.GetScheduleToStartTimeout(),\n\t\t\tScheduleToCloseTimeout:   decoded.GetScheduleToCloseTimeout(),\n\t\t\tStartToCloseTimeout:      decoded.GetStartToCloseTimeout(),\n\t\t\tHeartbeatTimeout:         decoded.GetHeartbeatTimeout(),\n\t\t\tCancelRequested:          decoded.GetCancelRequested(),\n\t\t\tCancelRequestID:          decoded.GetCancelRequestID(),\n\t\t\tTimerTaskStatus:          decoded.GetTimerTaskStatus(),\n\t\t\tAttempt:                  decoded.GetAttempt(),\n\t\t\tStartedIdentity:          decoded.GetStartedIdentity(),\n\t\t\tTaskList:                 decoded.GetTaskList(),\n\t\t\tTaskListKind:             decoded.GetTaskListKind(),\n\t\t\tHasRetryPolicy:           decoded.GetHasRetryPolicy(),\n\t\t\tInitialInterval:          decoded.GetRetryInitialInterval(),\n\t\t\tBackoffCoefficient:       decoded.GetRetryBackoffCoefficient(),\n\t\t\tMaximumInterval:          decoded.GetRetryMaximumInterval(),\n\t\t\tExpirationTime:           decoded.GetRetryExpirationTimestamp(),\n\t\t\tMaximumAttempts:          decoded.GetRetryMaximumAttempts(),\n\t\t\tNonRetriableErrors:       decoded.GetRetryNonRetryableErrors(),\n\t\t\tLastFailureReason:        decoded.GetRetryLastFailureReason(),\n\t\t\tLastWorkerIdentity:       decoded.GetRetryLastWorkerIdentity(),\n\t\t\tLastFailureDetails:       decoded.GetRetryLastFailureDetails(),\n\t\t}\n\t\tif decoded.StartedEvent != nil {\n\t\t\tinfo.StartedEvent = persistence.NewDataBlob(decoded.StartedEvent, constants.EncodingType(decoded.GetStartedEventEncoding()))\n\t\t}\n\t\tret[row.ScheduleID] = info\n\t}\n\n\treturn ret, nil\n}\n\nfunc deleteActivityInfoMap(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromActivityInfoMaps(ctx, &sqlplugin.ActivityInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteActivityInfoMap\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc updateTimerInfos(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\ttimerInfos []*persistence.TimerInfo,\n\tdeleteInfos []string,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) error {\n\n\tif len(timerInfos) > 0 {\n\t\trows := make([]sqlplugin.TimerInfoMapsRow, len(timerInfos))\n\t\tfor i, timerInfo := range timerInfos {\n\t\t\tblob, err := parser.TimerInfoToBlob(&serialization.TimerInfo{\n\t\t\t\tVersion:         timerInfo.Version,\n\t\t\t\tStartedID:       timerInfo.StartedID,\n\t\t\t\tExpiryTimestamp: timerInfo.ExpiryTime,\n\t\t\t\t// TaskID is a misleading variable, it actually serves\n\t\t\t\t// the purpose of indicating whether a timer task is\n\t\t\t\t// generated for this timer info\n\t\t\t\tTaskID: timerInfo.TaskStatus,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trows[i] = sqlplugin.TimerInfoMapsRow{\n\t\t\t\tShardID:      int64(shardID),\n\t\t\t\tDomainID:     domainID,\n\t\t\t\tWorkflowID:   workflowID,\n\t\t\t\tRunID:        runID,\n\t\t\t\tTimerID:      timerInfo.TimerID,\n\t\t\t\tData:         blob.Data,\n\t\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\t}\n\t\t}\n\t\tif _, err := tx.ReplaceIntoTimerInfoMaps(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateTimerInfos\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteInfos) > 0 {\n\t\tif _, err := tx.DeleteFromTimerInfoMaps(ctx, &sqlplugin.TimerInfoMapsFilter{\n\t\t\tShardID:    int64(shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tTimerIDs:   deleteInfos,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateTimerInfos\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getTimerInfoMap(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) (map[string]*persistence.TimerInfo, error) {\n\n\trows, err := db.SelectFromTimerInfoMaps(ctx, &sqlplugin.TimerInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getTimerInfoMap\", \"\", err)\n\t}\n\tret := make(map[string]*persistence.TimerInfo)\n\tfor _, row := range rows {\n\t\tinfo, err := parser.TimerInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tret[row.TimerID] = &persistence.TimerInfo{\n\t\t\tTimerID:    row.TimerID,\n\t\t\tVersion:    info.GetVersion(),\n\t\t\tStartedID:  info.GetStartedID(),\n\t\t\tExpiryTime: info.GetExpiryTimestamp(),\n\t\t\t// TaskID is a misleading variable, it actually serves\n\t\t\t// the purpose of indicating whether a timer task is\n\t\t\t// generated for this timer info\n\t\t\tTaskStatus: info.GetTaskID(),\n\t\t}\n\t}\n\n\treturn ret, nil\n}\n\nfunc deleteTimerInfoMap(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromTimerInfoMaps(ctx, &sqlplugin.TimerInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteTimerInfoMap\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc updateChildExecutionInfos(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tchildExecutionInfos []*persistence.InternalChildExecutionInfo,\n\tdeleteInfos []int64,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) error {\n\n\tif len(childExecutionInfos) > 0 {\n\t\trows := make([]sqlplugin.ChildExecutionInfoMapsRow, len(childExecutionInfos))\n\t\tfor i, childExecutionInfo := range childExecutionInfos {\n\t\t\tinitiateEvent, initiateEncoding := persistence.FromDataBlob(childExecutionInfo.InitiatedEvent)\n\t\t\tstartEvent, startEncoding := persistence.FromDataBlob(childExecutionInfo.StartedEvent)\n\n\t\t\tinfo := &serialization.ChildExecutionInfo{\n\t\t\t\tVersion:                childExecutionInfo.Version,\n\t\t\t\tInitiatedEventBatchID:  childExecutionInfo.InitiatedEventBatchID,\n\t\t\t\tInitiatedEvent:         initiateEvent,\n\t\t\t\tInitiatedEventEncoding: initiateEncoding,\n\t\t\t\tStartedEvent:           startEvent,\n\t\t\t\tStartedEventEncoding:   startEncoding,\n\t\t\t\tStartedID:              childExecutionInfo.StartedID,\n\t\t\t\tStartedWorkflowID:      childExecutionInfo.StartedWorkflowID,\n\t\t\t\tStartedRunID:           serialization.MustParseUUID(childExecutionInfo.StartedRunID),\n\t\t\t\tCreateRequestID:        childExecutionInfo.CreateRequestID,\n\t\t\t\tDomainID:               childExecutionInfo.DomainID,\n\t\t\t\tDomainNameDEPRECATED:   childExecutionInfo.DomainNameDEPRECATED,\n\t\t\t\tWorkflowTypeName:       childExecutionInfo.WorkflowTypeName,\n\t\t\t\tParentClosePolicy:      int32(childExecutionInfo.ParentClosePolicy),\n\t\t\t}\n\t\t\tblob, err := parser.ChildExecutionInfoToBlob(info)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trows[i] = sqlplugin.ChildExecutionInfoMapsRow{\n\t\t\t\tShardID:      int64(shardID),\n\t\t\t\tDomainID:     domainID,\n\t\t\t\tWorkflowID:   workflowID,\n\t\t\t\tRunID:        runID,\n\t\t\t\tInitiatedID:  childExecutionInfo.InitiatedID,\n\t\t\t\tData:         blob.Data,\n\t\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\t}\n\t\t}\n\t\tif _, err := tx.ReplaceIntoChildExecutionInfoMaps(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateChildExecutionInfos\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteInfos) > 0 {\n\t\tif _, err := tx.DeleteFromChildExecutionInfoMaps(ctx, &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\t\tShardID:      int64(shardID),\n\t\t\tDomainID:     domainID,\n\t\t\tWorkflowID:   workflowID,\n\t\t\tRunID:        runID,\n\t\t\tInitiatedIDs: deleteInfos,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateChildExecutionInfos\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getChildExecutionInfoMap(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) (map[int64]*persistence.InternalChildExecutionInfo, error) {\n\n\trows, err := db.SelectFromChildExecutionInfoMaps(ctx, &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getChildExecutionInfoMap\", \"\", err)\n\t}\n\n\tret := make(map[int64]*persistence.InternalChildExecutionInfo)\n\tfor _, row := range rows {\n\t\trowInfo, err := parser.ChildExecutionInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tinfo := &persistence.InternalChildExecutionInfo{\n\t\t\tInitiatedID:           row.InitiatedID,\n\t\t\tInitiatedEventBatchID: rowInfo.GetInitiatedEventBatchID(),\n\t\t\tVersion:               rowInfo.GetVersion(),\n\t\t\tStartedID:             rowInfo.GetStartedID(),\n\t\t\tStartedWorkflowID:     rowInfo.GetStartedWorkflowID(),\n\t\t\tStartedRunID:          serialization.UUID(rowInfo.GetStartedRunID()).String(),\n\t\t\tCreateRequestID:       rowInfo.GetCreateRequestID(),\n\t\t\tDomainID:              rowInfo.GetDomainID(),\n\t\t\tDomainNameDEPRECATED:  rowInfo.GetDomainNameDEPRECATED(),\n\t\t\tWorkflowTypeName:      rowInfo.GetWorkflowTypeName(),\n\t\t\tParentClosePolicy:     types.ParentClosePolicy(rowInfo.GetParentClosePolicy()),\n\t\t}\n\t\tif rowInfo.InitiatedEvent != nil {\n\t\t\tinfo.InitiatedEvent = persistence.NewDataBlob(rowInfo.InitiatedEvent, constants.EncodingType(rowInfo.GetInitiatedEventEncoding()))\n\t\t}\n\t\tif rowInfo.StartedEvent != nil {\n\t\t\tinfo.StartedEvent = persistence.NewDataBlob(rowInfo.StartedEvent, constants.EncodingType(rowInfo.GetStartedEventEncoding()))\n\t\t}\n\t\tret[row.InitiatedID] = info\n\t}\n\n\treturn ret, nil\n}\n\nfunc deleteChildExecutionInfoMap(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromChildExecutionInfoMaps(ctx, &sqlplugin.ChildExecutionInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteChildExecutionInfoMap\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc updateRequestCancelInfos(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\trequestCancelInfos []*persistence.RequestCancelInfo,\n\tdeleteInfos []int64,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) error {\n\n\tif len(requestCancelInfos) > 0 {\n\t\trows := make([]sqlplugin.RequestCancelInfoMapsRow, len(requestCancelInfos))\n\t\tfor i, requestCancelInfo := range requestCancelInfos {\n\t\t\tblob, err := parser.RequestCancelInfoToBlob(&serialization.RequestCancelInfo{\n\t\t\t\tVersion:               requestCancelInfo.Version,\n\t\t\t\tInitiatedEventBatchID: requestCancelInfo.InitiatedEventBatchID,\n\t\t\t\tCancelRequestID:       requestCancelInfo.CancelRequestID,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trows[i] = sqlplugin.RequestCancelInfoMapsRow{\n\t\t\t\tShardID:      int64(shardID),\n\t\t\t\tDomainID:     domainID,\n\t\t\t\tWorkflowID:   workflowID,\n\t\t\t\tRunID:        runID,\n\t\t\t\tInitiatedID:  requestCancelInfo.InitiatedID,\n\t\t\t\tData:         blob.Data,\n\t\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\t}\n\t\t}\n\n\t\tif _, err := tx.ReplaceIntoRequestCancelInfoMaps(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateRequestCancelInfos\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteInfos) > 0 {\n\t\tif _, err := tx.DeleteFromRequestCancelInfoMaps(ctx, &sqlplugin.RequestCancelInfoMapsFilter{\n\t\t\tShardID:      int64(shardID),\n\t\t\tDomainID:     domainID,\n\t\t\tWorkflowID:   workflowID,\n\t\t\tRunID:        runID,\n\t\t\tInitiatedIDs: deleteInfos,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateRequestCancelInfos\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getRequestCancelInfoMap(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) (map[int64]*persistence.RequestCancelInfo, error) {\n\n\trows, err := db.SelectFromRequestCancelInfoMaps(ctx, &sqlplugin.RequestCancelInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getRequestCancelInfoMap\", \"\", err)\n\t}\n\n\tret := make(map[int64]*persistence.RequestCancelInfo)\n\tfor _, row := range rows {\n\t\trowInfo, err := parser.RequestCancelInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tret[row.InitiatedID] = &persistence.RequestCancelInfo{\n\t\t\tVersion:               rowInfo.GetVersion(),\n\t\t\tInitiatedID:           row.InitiatedID,\n\t\t\tInitiatedEventBatchID: rowInfo.GetInitiatedEventBatchID(),\n\t\t\tCancelRequestID:       rowInfo.GetCancelRequestID(),\n\t\t}\n\t}\n\n\treturn ret, nil\n}\n\nfunc deleteRequestCancelInfoMap(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromRequestCancelInfoMaps(ctx, &sqlplugin.RequestCancelInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteRequestCancelInfoMap\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc updateSignalInfos(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tsignalInfos []*persistence.SignalInfo,\n\tdeleteInfos []int64,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) error {\n\n\tif len(signalInfos) > 0 {\n\t\trows := make([]sqlplugin.SignalInfoMapsRow, len(signalInfos))\n\t\tfor i, signalInfo := range signalInfos {\n\t\t\tblob, err := parser.SignalInfoToBlob(&serialization.SignalInfo{\n\t\t\t\tVersion:               signalInfo.Version,\n\t\t\t\tInitiatedEventBatchID: signalInfo.InitiatedEventBatchID,\n\t\t\t\tRequestID:             signalInfo.SignalRequestID,\n\t\t\t\tName:                  signalInfo.SignalName,\n\t\t\t\tInput:                 signalInfo.Input,\n\t\t\t\tControl:               signalInfo.Control,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\trows[i] = sqlplugin.SignalInfoMapsRow{\n\t\t\t\tShardID:      int64(shardID),\n\t\t\t\tDomainID:     domainID,\n\t\t\t\tWorkflowID:   workflowID,\n\t\t\t\tRunID:        runID,\n\t\t\t\tInitiatedID:  signalInfo.InitiatedID,\n\t\t\t\tData:         blob.Data,\n\t\t\t\tDataEncoding: string(blob.Encoding),\n\t\t\t}\n\t\t}\n\n\t\tif _, err := tx.ReplaceIntoSignalInfoMaps(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateSignalInfos\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteInfos) > 0 {\n\t\tif _, err := tx.DeleteFromSignalInfoMaps(ctx, &sqlplugin.SignalInfoMapsFilter{\n\t\t\tShardID:      int64(shardID),\n\t\t\tDomainID:     domainID,\n\t\t\tWorkflowID:   workflowID,\n\t\t\tRunID:        runID,\n\t\t\tInitiatedIDs: deleteInfos,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateSignalInfos\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getSignalInfoMap(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n\tparser serialization.Parser,\n) (map[int64]*persistence.SignalInfo, error) {\n\n\trows, err := db.SelectFromSignalInfoMaps(ctx, &sqlplugin.SignalInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getSignalInfoMap\", \"\", err)\n\t}\n\n\tret := make(map[int64]*persistence.SignalInfo)\n\tfor _, row := range rows {\n\t\trowInfo, err := parser.SignalInfoFromBlob(row.Data, row.DataEncoding)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tret[row.InitiatedID] = &persistence.SignalInfo{\n\t\t\tVersion:               rowInfo.GetVersion(),\n\t\t\tInitiatedID:           row.InitiatedID,\n\t\t\tInitiatedEventBatchID: rowInfo.GetInitiatedEventBatchID(),\n\t\t\tSignalRequestID:       rowInfo.GetRequestID(),\n\t\t\tSignalName:            rowInfo.GetName(),\n\t\t\tInput:                 rowInfo.GetInput(),\n\t\t\tControl:               rowInfo.GetControl(),\n\t\t}\n\t}\n\n\treturn ret, nil\n}\n\nfunc deleteSignalInfoMap(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromSignalInfoMaps(ctx, &sqlplugin.SignalInfoMapsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteSignalInfoMap\", \"\", err)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/workflow_state_non_maps.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"database/sql\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc updateSignalsRequested(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tsignalRequestedIDs []string,\n\tdeleteSignalRequestIDs []string,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif len(signalRequestedIDs) > 0 {\n\t\trows := make([]sqlplugin.SignalsRequestedSetsRow, len(signalRequestedIDs))\n\t\tfor i, v := range signalRequestedIDs {\n\t\t\trows[i] = sqlplugin.SignalsRequestedSetsRow{\n\t\t\t\tShardID:    int64(shardID),\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t\tSignalID:   v,\n\t\t\t}\n\t\t}\n\t\tif _, err := tx.InsertIntoSignalsRequestedSets(ctx, rows); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateSignalsRequested\", \"Failed to execute update query.\", err)\n\t\t}\n\t}\n\n\tif len(deleteSignalRequestIDs) > 0 {\n\t\tif _, err := tx.DeleteFromSignalsRequestedSets(ctx, &sqlplugin.SignalsRequestedSetsFilter{\n\t\t\tShardID:    int64(shardID),\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tSignalIDs:  deleteSignalRequestIDs,\n\t\t}); err != nil {\n\t\t\treturn convertCommonErrors(tx, \"updateSignalsRequested\", \"Failed to execute delete query.\", err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getSignalsRequested(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) (map[string]struct{}, error) {\n\n\trows, err := db.SelectFromSignalsRequestedSets(ctx, &sqlplugin.SignalsRequestedSetsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getSignalsRequested\", \"\", err)\n\t}\n\tvar ret = make(map[string]struct{})\n\tfor _, s := range rows {\n\t\tret[s.SignalID] = struct{}{}\n\t}\n\treturn ret, nil\n}\n\nfunc deleteSignalsRequestedSet(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromSignalsRequestedSets(ctx, &sqlplugin.SignalsRequestedSetsFilter{\n\t\tShardID:    int64(shardID),\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteSignalsRequestedSet\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc updateBufferedEvents(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tbatch *p.DataBlob,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif batch == nil {\n\t\treturn nil\n\t}\n\trow := sqlplugin.BufferedEventsRow{\n\t\tShardID:      shardID,\n\t\tDomainID:     domainID,\n\t\tWorkflowID:   workflowID,\n\t\tRunID:        runID,\n\t\tData:         batch.Data,\n\t\tDataEncoding: string(batch.Encoding),\n\t}\n\n\tif _, err := tx.InsertIntoBufferedEvents(ctx, []sqlplugin.BufferedEventsRow{row}); err != nil {\n\t\treturn convertCommonErrors(tx, \"updateBufferedEvents\", \"\", err)\n\t}\n\treturn nil\n}\n\nfunc getBufferedEvents(\n\tctx context.Context,\n\tdb sqlplugin.DB,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) ([]*p.DataBlob, error) {\n\n\trows, err := db.SelectFromBufferedEvents(ctx, &sqlplugin.BufferedEventsFilter{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tif err != nil && err != sql.ErrNoRows {\n\t\treturn nil, convertCommonErrors(db, \"getBufferedEvents\", \"\", err)\n\t}\n\tvar result []*p.DataBlob\n\tfor _, row := range rows {\n\t\tresult = append(result, p.NewDataBlob(row.Data, constants.EncodingType(row.DataEncoding)))\n\t}\n\treturn result, nil\n}\n\nfunc deleteBufferedEvents(\n\tctx context.Context,\n\ttx sqlplugin.Tx,\n\tshardID int,\n\tdomainID serialization.UUID,\n\tworkflowID string,\n\trunID serialization.UUID,\n) error {\n\n\tif _, err := tx.DeleteFromBufferedEvents(ctx, &sqlplugin.BufferedEventsFilter{\n\t\tShardID:    shardID,\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}); err != nil {\n\t\treturn convertCommonErrors(tx, \"deleteBufferedEvents\", \"\", err)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/sql/workflow_state_non_maps_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/serialization\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n)\n\nfunc TestUpdateSignalsRequested(t *testing.T) {\n\tshardID := 1\n\tdomainID := serialization.MustParseUUID(\"bf2360c6-62c7-48a2-bd95-fe6254bca628\")\n\tworkflowID := \"test\"\n\trunID := serialization.MustParseUUID(\"bf2360c6-a2c7-48a2-bd95-fe6254bca628\")\n\ttestCases := []struct {\n\t\tname                   string\n\t\tsignalRequestedIDs     []string\n\t\tdeleteSignalRequestIDs []string\n\t\tmockSetup              func(*sqlplugin.MockTx)\n\t\twantErr                bool\n\t}{\n\t\t{\n\t\t\tname:                   \"Success case\",\n\t\t\tsignalRequestedIDs:     []string{\"s1\", \"s2\"},\n\t\t\tdeleteSignalRequestIDs: []string{\"s3\", \"s4\"},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().InsertIntoSignalsRequestedSets(gomock.Any(), []sqlplugin.SignalsRequestedSetsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    int64(shardID),\n\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t\tSignalID:   \"s1\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:    int64(shardID),\n\t\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t\tSignalID:   \"s2\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromSignalsRequestedSets(gomock.Any(), &sqlplugin.SignalsRequestedSetsFilter{\n\t\t\t\t\tShardID:    int64(shardID),\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tSignalIDs:  []string{\"s3\", \"s4\"},\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:                   \"Error case - failed to insert\",\n\t\t\tsignalRequestedIDs:     []string{\"s1\", \"s2\"},\n\t\t\tdeleteSignalRequestIDs: []string{\"s3\", \"s4\"},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:                   \"Error case - failed to delete\",\n\t\t\tsignalRequestedIDs:     []string{\"s1\", \"s2\"},\n\t\t\tdeleteSignalRequestIDs: []string{\"s3\", \"s4\"},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\tmockTx.EXPECT().DeleteFromSignalsRequestedSets(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\t\t\terr := updateSignalsRequested(context.Background(), mockTx, tc.signalRequestedIDs, tc.deleteSignalRequestIDs, shardID, domainID, workflowID, runID)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateBufferedEvents(t *testing.T) {\n\tshardID := 1\n\tdomainID := serialization.MustParseUUID(\"bf2360c6-62c7-48a2-bd95-fe6254bca628\")\n\tworkflowID := \"test\"\n\trunID := serialization.MustParseUUID(\"bf2360c6-a2c7-48a2-bd95-fe6254bca628\")\n\ttestCases := []struct {\n\t\tname      string\n\t\tbatch     *persistence.DataBlob\n\t\tmockSetup func(*sqlplugin.MockTx)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:  \"Success case\",\n\t\t\tbatch: &persistence.DataBlob{Data: []byte(`buffer`), Encoding: constants.EncodingType(\"buffer\")},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\tmockTx.EXPECT().InsertIntoBufferedEvents(gomock.Any(), []sqlplugin.BufferedEventsRow{\n\t\t\t\t\t{\n\t\t\t\t\t\tShardID:      shardID,\n\t\t\t\t\t\tDomainID:     domainID,\n\t\t\t\t\t\tWorkflowID:   workflowID,\n\t\t\t\t\t\tRunID:        runID,\n\t\t\t\t\t\tData:         []byte(`buffer`),\n\t\t\t\t\t\tDataEncoding: \"buffer\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:  \"Error case\",\n\t\t\tbatch: &persistence.DataBlob{Data: []byte(`buffer`), Encoding: constants.EncodingType(\"buffer\")},\n\t\t\tmockSetup: func(mockTx *sqlplugin.MockTx) {\n\t\t\t\terr := errors.New(\"some error\")\n\t\t\t\tmockTx.EXPECT().InsertIntoBufferedEvents(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\tmockTx.EXPECT().IsNotFoundError(err).Return(true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockTx := sqlplugin.NewMockTx(ctrl)\n\n\t\t\ttc.mockSetup(mockTx)\n\t\t\terr := updateBufferedEvents(context.Background(), mockTx, tc.batch, shardID, domainID, workflowID, runID)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"Expected an error for test case\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Did not expect an error for test case\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/statsComputer.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\ntype (\n\t// statsComputer is to computing struct sizes after serialization\n\tstatsComputer struct{}\n)\n\nfunc (sc *statsComputer) computeMutableStateStats(req *InternalGetWorkflowExecutionResponse) *MutableStateStats {\n\texecutionInfoSize := computeExecutionInfoSize(req.State.ExecutionInfo)\n\n\tactivityInfoCount := 0\n\tactivityInfoSize := 0\n\tfor _, ai := range req.State.ActivityInfos {\n\t\tactivityInfoCount++\n\t\tactivityInfoSize += computeActivityInfoSize(ai)\n\t}\n\n\ttimerInfoCount := 0\n\ttimerInfoSize := 0\n\tfor _, ti := range req.State.TimerInfos {\n\t\ttimerInfoCount++\n\t\ttimerInfoSize += computeTimerInfoSize(ti)\n\t}\n\n\tchildExecutionInfoCount := 0\n\tchildExecutionInfoSize := 0\n\tfor _, ci := range req.State.ChildExecutionInfos {\n\t\tchildExecutionInfoCount++\n\t\tchildExecutionInfoSize += computeChildInfoSize(ci)\n\t}\n\n\tsignalInfoCount := 0\n\tsignalInfoSize := 0\n\tfor _, si := range req.State.SignalInfos {\n\t\tsignalInfoCount++\n\t\tsignalInfoSize += computeSignalInfoSize(si)\n\t}\n\n\tbufferedEventsCount := 0\n\tbufferedEventsSize := 0\n\n\tfor _, be := range req.State.BufferedEvents {\n\t\tbufferedEventsCount++\n\t\tbufferedEventsSize += len(be.Data)\n\t}\n\n\trequestCancelInfoCount := len(req.State.RequestCancelInfos)\n\n\ttotalSize := executionInfoSize\n\ttotalSize += activityInfoSize\n\ttotalSize += timerInfoSize\n\ttotalSize += childExecutionInfoSize\n\ttotalSize += signalInfoSize\n\ttotalSize += bufferedEventsSize\n\n\treturn &MutableStateStats{\n\t\tMutableStateSize:       totalSize,\n\t\tExecutionInfoSize:      executionInfoSize,\n\t\tActivityInfoSize:       activityInfoSize,\n\t\tTimerInfoSize:          timerInfoSize,\n\t\tChildInfoSize:          childExecutionInfoSize,\n\t\tSignalInfoSize:         signalInfoSize,\n\t\tBufferedEventsSize:     bufferedEventsSize,\n\t\tActivityInfoCount:      activityInfoCount,\n\t\tTimerInfoCount:         timerInfoCount,\n\t\tChildInfoCount:         childExecutionInfoCount,\n\t\tSignalInfoCount:        signalInfoCount,\n\t\tBufferedEventsCount:    bufferedEventsCount,\n\t\tRequestCancelInfoCount: requestCancelInfoCount,\n\t}\n}\n\nfunc (sc *statsComputer) computeMutableStateUpdateStats(req *InternalUpdateWorkflowExecutionRequest) *MutableStateUpdateSessionStats {\n\tif req.NewWorkflowSnapshot != nil {\n\t\treturn mergeMutableStateUpdateSessionStats(sc.computeWorkflowMutationStats(&req.UpdateWorkflowMutation), sc.computeWorkflowSnapshotStats(req.NewWorkflowSnapshot))\n\t}\n\treturn sc.computeWorkflowMutationStats(&req.UpdateWorkflowMutation)\n}\n\nfunc (sc *statsComputer) computeMutableStateCreateStats(req *InternalCreateWorkflowExecutionRequest) *MutableStateUpdateSessionStats {\n\treturn sc.computeWorkflowSnapshotStats(&req.NewWorkflowSnapshot)\n}\n\nfunc (sc *statsComputer) computeMutableStateConflictResolveStats(req *InternalConflictResolveWorkflowExecutionRequest) *MutableStateUpdateSessionStats {\n\tmss := sc.computeWorkflowSnapshotStats(&req.ResetWorkflowSnapshot)\n\tif req.NewWorkflowSnapshot != nil {\n\t\tmss = mergeMutableStateUpdateSessionStats(mss, sc.computeWorkflowSnapshotStats(req.NewWorkflowSnapshot))\n\t}\n\tif req.CurrentWorkflowMutation != nil {\n\t\tmss = mergeMutableStateUpdateSessionStats(mss, sc.computeWorkflowMutationStats(req.CurrentWorkflowMutation))\n\t}\n\treturn mss\n}\n\nfunc (sc *statsComputer) computeWorkflowMutationStats(req *InternalWorkflowMutation) *MutableStateUpdateSessionStats {\n\texecutionInfoSize := computeExecutionInfoSize(req.ExecutionInfo)\n\n\tactivityInfoCount := 0\n\tactivityInfoSize := 0\n\tfor _, ai := range req.UpsertActivityInfos {\n\t\tactivityInfoCount++\n\t\tactivityInfoSize += computeActivityInfoSize(ai)\n\t}\n\n\ttimerInfoCount := 0\n\ttimerInfoSize := 0\n\tfor _, ti := range req.UpsertTimerInfos {\n\t\ttimerInfoCount++\n\t\ttimerInfoSize += computeTimerInfoSize(ti)\n\t}\n\n\tchildExecutionInfoCount := 0\n\tchildExecutionInfoSize := 0\n\tfor _, ci := range req.UpsertChildExecutionInfos {\n\t\tchildExecutionInfoCount++\n\t\tchildExecutionInfoSize += computeChildInfoSize(ci)\n\t}\n\n\tsignalInfoCount := 0\n\tsignalInfoSize := 0\n\tfor _, si := range req.UpsertSignalInfos {\n\t\tsignalInfoCount++\n\t\tsignalInfoSize += computeSignalInfoSize(si)\n\t}\n\n\tbufferedEventsSize := 0\n\tif req.NewBufferedEvents != nil {\n\t\tbufferedEventsSize = len(req.NewBufferedEvents.Data)\n\t}\n\n\trequestCancelInfoCount := len(req.UpsertRequestCancelInfos)\n\n\tdeleteActivityInfoCount := len(req.DeleteActivityInfos)\n\n\tdeleteTimerInfoCount := len(req.DeleteTimerInfos)\n\n\tdeleteChildInfoCount := len(req.DeleteChildExecutionInfos)\n\n\tdeleteSignalInfoCount := len(req.DeleteSignalInfos)\n\n\tdeleteRequestCancelInfoCount := len(req.DeleteRequestCancelInfos)\n\n\ttaskCountByCategory := computeTaskCountByCategory(req.TasksByCategory)\n\n\ttotalSize := executionInfoSize\n\ttotalSize += activityInfoSize\n\ttotalSize += timerInfoSize\n\ttotalSize += childExecutionInfoSize\n\ttotalSize += signalInfoSize\n\ttotalSize += bufferedEventsSize\n\n\treturn &MutableStateUpdateSessionStats{\n\t\tMutableStateSize:             totalSize,\n\t\tExecutionInfoSize:            executionInfoSize,\n\t\tActivityInfoSize:             activityInfoSize,\n\t\tTimerInfoSize:                timerInfoSize,\n\t\tChildInfoSize:                childExecutionInfoSize,\n\t\tSignalInfoSize:               signalInfoSize,\n\t\tBufferedEventsSize:           bufferedEventsSize,\n\t\tActivityInfoCount:            activityInfoCount,\n\t\tTimerInfoCount:               timerInfoCount,\n\t\tChildInfoCount:               childExecutionInfoCount,\n\t\tSignalInfoCount:              signalInfoCount,\n\t\tRequestCancelInfoCount:       requestCancelInfoCount,\n\t\tDeleteActivityInfoCount:      deleteActivityInfoCount,\n\t\tDeleteTimerInfoCount:         deleteTimerInfoCount,\n\t\tDeleteChildInfoCount:         deleteChildInfoCount,\n\t\tDeleteSignalInfoCount:        deleteSignalInfoCount,\n\t\tDeleteRequestCancelInfoCount: deleteRequestCancelInfoCount,\n\t\tTaskCountByCategory:          taskCountByCategory,\n\t}\n}\n\nfunc (sc *statsComputer) computeWorkflowSnapshotStats(req *InternalWorkflowSnapshot) *MutableStateUpdateSessionStats {\n\texecutionInfoSize := computeExecutionInfoSize(req.ExecutionInfo)\n\n\tactivityInfoCount := 0\n\tactivityInfoSize := 0\n\tfor _, ai := range req.ActivityInfos {\n\t\tactivityInfoCount++\n\t\tactivityInfoSize += computeActivityInfoSize(ai)\n\t}\n\n\ttimerInfoCount := 0\n\ttimerInfoSize := 0\n\tfor _, ti := range req.TimerInfos {\n\t\ttimerInfoCount++\n\t\ttimerInfoSize += computeTimerInfoSize(ti)\n\t}\n\n\tchildExecutionInfoCount := 0\n\tchildExecutionInfoSize := 0\n\tfor _, ci := range req.ChildExecutionInfos {\n\t\tchildExecutionInfoCount++\n\t\tchildExecutionInfoSize += computeChildInfoSize(ci)\n\t}\n\n\tsignalInfoCount := 0\n\tsignalInfoSize := 0\n\tfor _, si := range req.SignalInfos {\n\t\tsignalInfoCount++\n\t\tsignalInfoSize += computeSignalInfoSize(si)\n\t}\n\n\trequestCancelInfoCount := len(req.RequestCancelInfos)\n\n\ttaskCountByCategory := computeTaskCountByCategory(req.TasksByCategory)\n\n\ttotalSize := executionInfoSize\n\ttotalSize += activityInfoSize\n\ttotalSize += timerInfoSize\n\ttotalSize += childExecutionInfoSize\n\ttotalSize += signalInfoSize\n\n\treturn &MutableStateUpdateSessionStats{\n\t\tMutableStateSize:       totalSize,\n\t\tExecutionInfoSize:      executionInfoSize,\n\t\tActivityInfoSize:       activityInfoSize,\n\t\tTimerInfoSize:          timerInfoSize,\n\t\tChildInfoSize:          childExecutionInfoSize,\n\t\tSignalInfoSize:         signalInfoSize,\n\t\tActivityInfoCount:      activityInfoCount,\n\t\tTimerInfoCount:         timerInfoCount,\n\t\tChildInfoCount:         childExecutionInfoCount,\n\t\tSignalInfoCount:        signalInfoCount,\n\t\tRequestCancelInfoCount: requestCancelInfoCount,\n\t\tTaskCountByCategory:    taskCountByCategory,\n\t}\n}\n\nfunc mergeMutableStateUpdateSessionStats(stats ...*MutableStateUpdateSessionStats) *MutableStateUpdateSessionStats {\n\tresult := &MutableStateUpdateSessionStats{\n\t\tTaskCountByCategory: make(map[HistoryTaskCategory]int),\n\t}\n\tfor _, s := range stats {\n\t\tresult.MutableStateSize += s.MutableStateSize\n\n\t\tresult.ExecutionInfoSize += s.ExecutionInfoSize\n\t\tresult.ActivityInfoSize += s.ActivityInfoSize\n\t\tresult.TimerInfoSize += s.TimerInfoSize\n\t\tresult.ChildInfoSize += s.ChildInfoSize\n\t\tresult.SignalInfoSize += s.SignalInfoSize\n\t\tresult.BufferedEventsSize += s.BufferedEventsSize\n\n\t\tresult.ActivityInfoCount += s.ActivityInfoCount\n\t\tresult.TimerInfoCount += s.TimerInfoCount\n\t\tresult.ChildInfoCount += s.ChildInfoCount\n\t\tresult.SignalInfoCount += s.SignalInfoCount\n\t\tresult.RequestCancelInfoCount += s.RequestCancelInfoCount\n\n\t\tresult.DeleteActivityInfoCount += s.DeleteActivityInfoCount\n\t\tresult.DeleteTimerInfoCount += s.DeleteTimerInfoCount\n\t\tresult.DeleteChildInfoCount += s.DeleteChildInfoCount\n\t\tresult.DeleteSignalInfoCount += s.DeleteSignalInfoCount\n\t\tresult.DeleteRequestCancelInfoCount += s.DeleteRequestCancelInfoCount\n\n\t\tfor k, v := range s.TaskCountByCategory {\n\t\t\tresult.TaskCountByCategory[k] += v\n\t\t}\n\t}\n\treturn result\n}\n\nfunc computeExecutionInfoSize(executionInfo *InternalWorkflowExecutionInfo) int {\n\tsize := len(executionInfo.WorkflowID)\n\tsize += len(executionInfo.TaskList)\n\tsize += len(executionInfo.WorkflowTypeName)\n\tsize += len(executionInfo.ParentWorkflowID)\n\n\treturn size\n}\n\nfunc computeActivityInfoSize(ai *InternalActivityInfo) int {\n\tsize := len(ai.ActivityID)\n\tif ai.ScheduledEvent != nil {\n\t\tsize += len(ai.ScheduledEvent.Data)\n\t}\n\tif ai.StartedEvent != nil {\n\t\tsize += len(ai.StartedEvent.Data)\n\t}\n\tsize += len(ai.Details)\n\n\treturn size\n}\n\nfunc computeTimerInfoSize(ti *TimerInfo) int {\n\tsize := len(ti.TimerID)\n\n\treturn size\n}\n\nfunc computeChildInfoSize(ci *InternalChildExecutionInfo) int {\n\tsize := 0\n\tif ci.InitiatedEvent != nil {\n\t\tsize += len(ci.InitiatedEvent.Data)\n\t}\n\tif ci.StartedEvent != nil {\n\t\tsize += len(ci.StartedEvent.Data)\n\t}\n\treturn size\n}\n\nfunc computeSignalInfoSize(si *SignalInfo) int {\n\tsize := len(si.SignalName)\n\tsize += len(si.Input)\n\tsize += len(si.Control)\n\n\treturn size\n}\n\nfunc computeTaskCountByCategory(tasks map[HistoryTaskCategory][]Task) map[HistoryTaskCategory]int {\n\ttaskCountByCategory := make(map[HistoryTaskCategory]int, len(tasks))\n\tfor k, v := range tasks {\n\t\ttaskCountByCategory[k] = len(v)\n\t}\n\treturn taskCountByCategory\n}\n"
  },
  {
    "path": "common/persistence/statsComputer_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n)\n\ntype (\n\tstatsComputerSuite struct {\n\t\tsc *statsComputer\n\t\tsuite.Suite\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t}\n)\n\nfunc TestStatsComputerSuite(t *testing.T) {\n\ts := new(statsComputerSuite)\n\tsuite.Run(t, s)\n}\n\n// TODO need to add more tests\nfunc (s *statsComputerSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.sc = &statsComputer{}\n}\n\nfunc (s *statsComputerSuite) createRequest() *InternalUpdateWorkflowExecutionRequest {\n\treturn &InternalUpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: InternalWorkflowMutation{\n\t\t\tExecutionInfo: &InternalWorkflowExecutionInfo{},\n\t\t},\n\t}\n}\n\nfunc (s *statsComputerSuite) TestStatsWithStartedEvent() {\n\tms := s.createRequest()\n\tdomainID := \"A\"\n\texecution := workflow.WorkflowExecution{\n\t\tWorkflowId: common.StringPtr(\"test-workflow-id\"),\n\t\tRunId:      common.StringPtr(\"run_id\"),\n\t}\n\tworkflowType := &workflow.WorkflowType{\n\t\tName: common.StringPtr(\"test-workflow-type-name\"),\n\t}\n\ttaskList := &workflow.TaskList{\n\t\tName: common.StringPtr(\"test-tasklist\"),\n\t}\n\n\tms.UpdateWorkflowMutation.ExecutionInfo.DomainID = domainID\n\tms.UpdateWorkflowMutation.ExecutionInfo.WorkflowID = execution.GetWorkflowId()\n\tms.UpdateWorkflowMutation.ExecutionInfo.RunID = execution.GetRunId()\n\tms.UpdateWorkflowMutation.ExecutionInfo.WorkflowTypeName = workflowType.GetName()\n\tms.UpdateWorkflowMutation.ExecutionInfo.TaskList = taskList.GetName()\n\n\texpectedSize := len(execution.GetWorkflowId()) + len(workflowType.GetName()) + len(taskList.GetName())\n\n\tstats := s.sc.computeMutableStateUpdateStats(ms)\n\ts.Equal(stats.ExecutionInfoSize, expectedSize)\n}\n\nfunc (s *statsComputerSuite) TestComputeWorkflowMutationStats() {\n\ta1 := &InternalActivityInfo{}\n\tt1 := &TimerInfo{}\n\tc1 := &InternalChildExecutionInfo{}\n\ts1 := &SignalInfo{}\n\tr1 := &RequestCancelInfo{}\n\tms := &InternalWorkflowMutation{\n\t\tExecutionInfo:             &InternalWorkflowExecutionInfo{},\n\t\tUpsertActivityInfos:       []*InternalActivityInfo{a1, a1},\n\t\tUpsertTimerInfos:          []*TimerInfo{t1, t1, t1},\n\t\tUpsertChildExecutionInfos: []*InternalChildExecutionInfo{c1, c1, c1, c1},\n\t\tUpsertSignalInfos:         []*SignalInfo{s1},\n\t\tUpsertRequestCancelInfos:  []*RequestCancelInfo{r1},\n\t\tNewBufferedEvents:         &DataBlob{Data: []byte(\"asdfsaf\")},\n\t\tDeleteActivityInfos:       []int64{1, 2, 3, 4},\n\t\tDeleteTimerInfos:          []string{\"asdfa\"},\n\t\tDeleteChildExecutionInfos: []int64{0},\n\t\tDeleteSignalInfos:         nil,\n\t\tDeleteRequestCancelInfos:  []int64{},\n\t\tTasksByCategory: map[HistoryTaskCategory][]Task{\n\t\t\tHistoryTaskCategoryTransfer: []Task{\n\t\t\t\t&ActivityTask{}, &DecisionTask{},\n\t\t\t},\n\t\t\tHistoryTaskCategoryTimer: []Task{\n\t\t\t\t&UserTimerTask{},\n\t\t\t},\n\t\t},\n\t}\n\tstats := s.sc.computeWorkflowMutationStats(ms)\n\ts.Equal(computeExecutionInfoSize(ms.ExecutionInfo), stats.ExecutionInfoSize)\n\ts.Equal(computeActivityInfoSize(a1)*len(ms.UpsertActivityInfos), stats.ActivityInfoSize)\n\ts.Equal(len(ms.UpsertActivityInfos), stats.ActivityInfoCount)\n\ts.Equal(computeTimerInfoSize(t1)*len(ms.UpsertTimerInfos), stats.TimerInfoSize)\n\ts.Equal(len(ms.UpsertTimerInfos), stats.TimerInfoCount)\n\ts.Equal(computeChildInfoSize(c1)*len(ms.UpsertChildExecutionInfos), stats.ChildInfoSize)\n\ts.Equal(len(ms.UpsertChildExecutionInfos), stats.ChildInfoCount)\n\ts.Equal(computeSignalInfoSize(s1)*len(ms.UpsertSignalInfos), stats.SignalInfoSize)\n\ts.Equal(len(ms.UpsertSignalInfos), stats.SignalInfoCount)\n\ts.Equal(len(ms.UpsertRequestCancelInfos), stats.RequestCancelInfoCount)\n\ts.Equal(len(ms.NewBufferedEvents.Data), stats.BufferedEventsSize)\n\ts.Equal(len(ms.DeleteActivityInfos), stats.DeleteActivityInfoCount)\n\ts.Equal(len(ms.DeleteTimerInfos), stats.DeleteTimerInfoCount)\n\ts.Equal(len(ms.DeleteChildExecutionInfos), stats.DeleteChildInfoCount)\n\ts.Equal(len(ms.DeleteSignalInfos), stats.DeleteSignalInfoCount)\n\ts.Equal(len(ms.DeleteRequestCancelInfos), stats.DeleteRequestCancelInfoCount)\n\ts.Equal(stats.ExecutionInfoSize+stats.ActivityInfoSize+stats.TimerInfoSize+stats.ChildInfoSize+stats.SignalInfoSize+stats.BufferedEventsSize, stats.MutableStateSize)\n\tfor c, tasks := range ms.TasksByCategory {\n\t\ts.Equal(len(tasks), stats.TaskCountByCategory[c])\n\t}\n}\n\nfunc (s *statsComputerSuite) TestComputeWorkflowSnapshotStats() {\n\ta1 := &InternalActivityInfo{}\n\tt1 := &TimerInfo{}\n\tc1 := &InternalChildExecutionInfo{}\n\ts1 := &SignalInfo{}\n\tr1 := &RequestCancelInfo{}\n\tms := &InternalWorkflowSnapshot{\n\t\tExecutionInfo:       &InternalWorkflowExecutionInfo{},\n\t\tActivityInfos:       []*InternalActivityInfo{a1, a1},\n\t\tTimerInfos:          []*TimerInfo{t1, t1, t1},\n\t\tChildExecutionInfos: []*InternalChildExecutionInfo{c1, c1, c1, c1},\n\t\tSignalInfos:         []*SignalInfo{s1},\n\t\tRequestCancelInfos:  []*RequestCancelInfo{r1},\n\t\tTasksByCategory: map[HistoryTaskCategory][]Task{\n\t\t\tHistoryTaskCategoryTransfer: []Task{\n\t\t\t\t&ActivityTask{}, &DecisionTask{},\n\t\t\t},\n\t\t\tHistoryTaskCategoryTimer: []Task{\n\t\t\t\t&UserTimerTask{},\n\t\t\t},\n\t\t},\n\t}\n\tstats := s.sc.computeWorkflowSnapshotStats(ms)\n\ts.Equal(computeExecutionInfoSize(ms.ExecutionInfo), stats.ExecutionInfoSize)\n\ts.Equal(computeActivityInfoSize(a1)*len(ms.ActivityInfos), stats.ActivityInfoSize)\n\ts.Equal(len(ms.ActivityInfos), stats.ActivityInfoCount)\n\ts.Equal(computeTimerInfoSize(t1)*len(ms.TimerInfos), stats.TimerInfoSize)\n\ts.Equal(len(ms.TimerInfos), stats.TimerInfoCount)\n\ts.Equal(computeChildInfoSize(c1)*len(ms.ChildExecutionInfos), stats.ChildInfoSize)\n\ts.Equal(len(ms.ChildExecutionInfos), stats.ChildInfoCount)\n\ts.Equal(computeSignalInfoSize(s1)*len(ms.SignalInfos), stats.SignalInfoSize)\n\ts.Equal(len(ms.SignalInfos), stats.SignalInfoCount)\n\ts.Equal(len(ms.RequestCancelInfos), stats.RequestCancelInfoCount)\n\ts.Equal(stats.ExecutionInfoSize+stats.ActivityInfoSize+stats.TimerInfoSize+stats.ChildInfoSize+stats.SignalInfoSize+stats.BufferedEventsSize, stats.MutableStateSize)\n\tfor c, tasks := range ms.TasksByCategory {\n\t\ts.Equal(len(tasks), stats.TaskCountByCategory[c])\n\t}\n}\n"
  },
  {
    "path": "common/persistence/task_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\ttaskManager struct {\n\t\tpersistence TaskStore\n\t\ttimeSrc     clock.TimeSource\n\t}\n)\n\nvar _ TaskManager = (*taskManager)(nil)\n\n// NewTaskManager returns a new TaskManager\nfunc NewTaskManager(\n\tpersistence TaskStore,\n) TaskManager {\n\treturn &taskManager{\n\t\tpersistence: persistence,\n\t\ttimeSrc:     clock.NewRealTimeSource(),\n\t}\n}\n\nfunc (t *taskManager) GetName() string {\n\treturn t.persistence.GetName()\n}\n\nfunc (t *taskManager) Close() {\n\tt.persistence.Close()\n}\n\nfunc (t *taskManager) LeaseTaskList(ctx context.Context, request *LeaseTaskListRequest) (*LeaseTaskListResponse, error) {\n\trequest.CurrentTimeStamp = t.timeSrc.Now()\n\treturn t.persistence.LeaseTaskList(ctx, request)\n}\n\nfunc (t *taskManager) GetTaskList(ctx context.Context, request *GetTaskListRequest) (*GetTaskListResponse, error) {\n\treturn t.persistence.GetTaskList(ctx, request)\n}\n\nfunc (t *taskManager) UpdateTaskList(ctx context.Context, request *UpdateTaskListRequest) (*UpdateTaskListResponse, error) {\n\trequest.CurrentTimeStamp = t.timeSrc.Now()\n\treturn t.persistence.UpdateTaskList(ctx, request)\n}\n\nfunc (t *taskManager) ListTaskList(ctx context.Context, request *ListTaskListRequest) (*ListTaskListResponse, error) {\n\treturn t.persistence.ListTaskList(ctx, request)\n}\n\nfunc (t *taskManager) DeleteTaskList(ctx context.Context, request *DeleteTaskListRequest) error {\n\treturn t.persistence.DeleteTaskList(ctx, request)\n}\n\nfunc (t *taskManager) GetTaskListSize(ctx context.Context, request *GetTaskListSizeRequest) (*GetTaskListSizeResponse, error) {\n\treturn t.persistence.GetTaskListSize(ctx, request)\n}\n\nfunc (t *taskManager) CreateTasks(ctx context.Context, request *CreateTasksRequest) (*CreateTasksResponse, error) {\n\trequest.CurrentTimeStamp = t.timeSrc.Now()\n\treturn t.persistence.CreateTasks(ctx, request)\n}\n\nfunc (t *taskManager) GetTasks(ctx context.Context, request *GetTasksRequest) (*GetTasksResponse, error) {\n\treturn t.persistence.GetTasks(ctx, request)\n}\n\nfunc (t *taskManager) CompleteTask(ctx context.Context, request *CompleteTaskRequest) error {\n\treturn t.persistence.CompleteTask(ctx, request)\n}\n\nfunc (t *taskManager) CompleteTasksLessThan(ctx context.Context, request *CompleteTasksLessThanRequest) (*CompleteTasksLessThanResponse, error) {\n\treturn t.persistence.CompleteTasksLessThan(ctx, request)\n}\n\nfunc (t *taskManager) GetOrphanTasks(ctx context.Context, request *GetOrphanTasksRequest) (*GetOrphanTasksResponse, error) {\n\treturn t.persistence.GetOrphanTasks(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/task_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n)\n\nfunc setUpMocksForTaskManager(t *testing.T) (*taskManager, *MockTaskStore) {\n\tctrl := gomock.NewController(t)\n\tmockTaskStore := NewMockTaskStore(ctrl)\n\n\ttaskManager := NewTaskManager(mockTaskStore).(*taskManager)\n\n\treturn taskManager, mockTaskStore\n}\n\nfunc TestTaskManager_GetName(t *testing.T) {\n\tt.Run(\"get name\", func(t *testing.T) {\n\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t// Set up expectation\n\t\tmockTaskStore.EXPECT().GetName().Return(\"mock-task-store\").Times(1)\n\n\t\t// Call the method\n\t\tname := taskManager.GetName()\n\n\t\t// Validate the result\n\t\tassert.Equal(t, \"mock-task-store\", name)\n\t})\n}\n\nfunc TestTaskManager_Close(t *testing.T) {\n\tt.Run(\"close task store\", func(t *testing.T) {\n\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t// Set up expectation\n\t\tmockTaskStore.EXPECT().Close().Times(1)\n\n\t\t// Call the method\n\t\ttaskManager.Close()\n\n\t\t// No need to assert, just ensuring method is called\n\t})\n}\n\nfunc TestTaskManager_LeaseTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *LeaseTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tLeaseTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&LeaseTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &LeaseTaskListRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tLeaseTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &LeaseTaskListRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.LeaseTaskList(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_GetTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *GetTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&GetTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetTaskListRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetTaskListRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.GetTaskList(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_UpdateTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *UpdateTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tUpdateTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&UpdateTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &UpdateTaskListRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tUpdateTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &UpdateTaskListRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.UpdateTaskList(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_ListTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *ListTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tListTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&ListTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &ListTaskListRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tListTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &ListTaskListRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.ListTaskList(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_DeleteTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *DeleteTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tDeleteTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &DeleteTaskListRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tDeleteTaskList(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &DeleteTaskListRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\terr := taskManager.DeleteTaskList(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_GetTaskListSize(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *GetTaskListSizeRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTaskListSize(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&GetTaskListSizeResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetTaskListSizeRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTaskListSize(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetTaskListSizeRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.GetTaskListSize(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_CreateTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *CreateTasksRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCreateTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&CreateTasksResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &CreateTasksRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCreateTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &CreateTasksRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.CreateTasks(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_GetTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *GetTasksRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&GetTasksResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetTasksRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetTasksRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.GetTasks(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_CompleteTask(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *CompleteTaskRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCompleteTask(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &CompleteTaskRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCompleteTask(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &CompleteTaskRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\terr := taskManager.CompleteTask(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_CompleteTasksLessThan(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *CompleteTasksLessThanRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCompleteTasksLessThan(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&CompleteTasksLessThanResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &CompleteTasksLessThanRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tCompleteTasksLessThan(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &CompleteTasksLessThanRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.CompleteTasksLessThan(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskManager_GetOrphanTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(*MockTaskStore)\n\t\trequest       *GetOrphanTasksRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetOrphanTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&GetOrphanTasksResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\trequest:     &GetOrphanTasksRequest{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"persistence error\",\n\t\t\tsetupMock: func(mockTaskStore *MockTaskStore) {\n\t\t\t\tmockTaskStore.EXPECT().\n\t\t\t\t\tGetOrphanTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"persistence error\")).Times(1)\n\t\t\t},\n\t\t\trequest:       &GetOrphanTasksRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskManager, mockTaskStore := setUpMocksForTaskManager(t)\n\n\t\t\ttc.setupMock(mockTaskStore)\n\n\t\t\t// Call the method\n\t\t\t_, err := taskManager.GetOrphanTasks(context.Background(), tc.request)\n\n\t\t\t// Validate the result\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/tasks.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Task is the generic interface for workflow tasks\ntype Task interface {\n\tGetTaskCategory() HistoryTaskCategory\n\tGetTaskKey() HistoryTaskKey\n\tGetTaskType() int\n\tGetDomainID() string\n\tGetWorkflowID() string\n\tGetRunID() string\n\t// GetTaskList returns the name of the task list the task is currently\n\t// associated with. This may differ from the original task list if the\n\t// task is a sticky decision task.\n\tGetTaskList() string\n\t// GetOriginalTaskList returns the task list on which the task was initially\n\t// scheduled. It is used to enforce rate limits and ensure fair scheduling\n\t// across task lists.\n\tGetOriginalTaskList() string\n\tGetOriginalTaskListKind() types.TaskListKind\n\tGetVersion() int64\n\tSetVersion(version int64)\n\tGetTaskID() int64\n\tSetTaskID(id int64)\n\tGetVisibilityTimestamp() time.Time\n\tSetVisibilityTimestamp(timestamp time.Time)\n\tByteSize() uint64\n\tToTransferTaskInfo() (*TransferTaskInfo, error)\n\tToTimerTaskInfo() (*TimerTaskInfo, error)\n\tToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error)\n}\n\nvar (\n\tMaximumHistoryTaskKey = HistoryTaskKey{\n\t\tscheduledTime: time.Unix(0, math.MaxInt64),\n\t\ttaskID:        math.MaxInt64,\n\t}\n)\n\ntype (\n\tHistoryTaskKey struct {\n\t\tscheduledTime time.Time\n\t\ttaskID        int64\n\t}\n\n\tWorkflowIdentifier struct {\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t}\n\t// TaskData is common attributes for all tasks.\n\tTaskData struct {\n\t\tVersion             int64\n\t\tTaskID              int64\n\t\tVisibilityTimestamp time.Time\n\t}\n\n\t// ActivityTask identifies a transfer task for activity\n\tActivityTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID string\n\t\tTaskList       string\n\t\tScheduleID     int64\n\t}\n\n\t// DecisionTask identifies a transfer task for decision\n\tDecisionTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID       string\n\t\tTaskList             string\n\t\tScheduleID           int64\n\t\tOriginalTaskList     string\n\t\tOriginalTaskListKind types.TaskListKind\n\t}\n\n\t// RecordWorkflowStartedTask identifites a transfer task for writing visibility open execution record\n\tRecordWorkflowStartedTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// ResetWorkflowTask identifites a transfer task to reset workflow\n\tResetWorkflowTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// CloseExecutionTask identifies a transfer task for deletion of execution\n\tCloseExecutionTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// DeleteHistoryEventTask identifies a timer task for deletion of history events of completed execution.\n\tDeleteHistoryEventTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// DecisionTimeoutTask identifies a timeout task.\n\tDecisionTimeoutTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tEventID         int64\n\t\tScheduleAttempt int64\n\t\tTimeoutType     int\n\t\tTaskList        string\n\t}\n\n\t// WorkflowTimeoutTask identifies a timeout task.\n\tWorkflowTimeoutTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// CancelExecutionTask identifies a transfer task for cancel of execution\n\tCancelExecutionTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID          string\n\t\tTargetWorkflowID        string\n\t\tTargetRunID             string\n\t\tTargetChildWorkflowOnly bool\n\t\tInitiatedID             int64\n\t\tTaskList                string\n\t}\n\n\t// SignalExecutionTask identifies a transfer task for signal execution\n\tSignalExecutionTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID          string\n\t\tTargetWorkflowID        string\n\t\tTargetRunID             string\n\t\tTargetChildWorkflowOnly bool\n\t\tInitiatedID             int64\n\t\tTaskList                string\n\t}\n\n\t// UpsertWorkflowSearchAttributesTask identifies a transfer task for upsert search attributes\n\tUpsertWorkflowSearchAttributesTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// StartChildExecutionTask identifies a transfer task for starting child execution\n\tStartChildExecutionTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID   string\n\t\tTargetWorkflowID string\n\t\tInitiatedID      int64\n\t\tTaskList         string\n\t}\n\n\t// RecordWorkflowClosedTask identifies a transfer task for writing visibility close execution record\n\tRecordWorkflowClosedTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTaskList string\n\t}\n\n\t// RecordChildExecutionCompletedTask identifies a task for recording the competion of a child workflow\n\tRecordChildExecutionCompletedTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTargetDomainID   string\n\t\tTargetWorkflowID string\n\t\tTargetRunID      string\n\t\tTaskList         string\n\t}\n\n\t// ActivityTimeoutTask identifies a timeout task.\n\tActivityTimeoutTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTimeoutType int\n\t\tEventID     int64\n\t\tAttempt     int64\n\t\tTaskList    string\n\t}\n\n\t// UserTimerTask identifies a timeout task.\n\tUserTimerTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tEventID  int64\n\t\tTaskList string\n\t}\n\n\t// ActivityRetryTimerTask to schedule a retry task for activity\n\tActivityRetryTimerTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tEventID  int64\n\t\tAttempt  int64\n\t\tTaskList string\n\t}\n\n\t// WorkflowBackoffTimerTask to schedule first decision task for retried workflow\n\tWorkflowBackoffTimerTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tTimeoutType int // 0 for retry, 1 for cron.\n\t\tTaskList    string\n\t}\n\n\t// HistoryReplicationTask is the replication task created for shipping history replication events to other clusters\n\tHistoryReplicationTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tFirstEventID      int64\n\t\tNextEventID       int64\n\t\tBranchToken       []byte\n\t\tNewRunBranchToken []byte\n\t}\n\n\t// SyncActivityTask is the replication task created for shipping activity info to other clusters\n\tSyncActivityTask struct {\n\t\tWorkflowIdentifier\n\t\tTaskData\n\t\tScheduledID int64\n\t}\n\n\t// FailoverMarkerTask is the marker for graceful failover\n\tFailoverMarkerTask struct {\n\t\tTaskData\n\t\tDomainID string\n\t}\n)\n\n// assert all task types implements Task interface\nvar (\n\t_ Task = (*ActivityTask)(nil)\n\t_ Task = (*DecisionTask)(nil)\n\t_ Task = (*RecordWorkflowStartedTask)(nil)\n\t_ Task = (*ResetWorkflowTask)(nil)\n\t_ Task = (*CloseExecutionTask)(nil)\n\t_ Task = (*DeleteHistoryEventTask)(nil)\n\t_ Task = (*DecisionTimeoutTask)(nil)\n\t_ Task = (*WorkflowTimeoutTask)(nil)\n\t_ Task = (*CancelExecutionTask)(nil)\n\t_ Task = (*SignalExecutionTask)(nil)\n\t_ Task = (*RecordChildExecutionCompletedTask)(nil)\n\t_ Task = (*UpsertWorkflowSearchAttributesTask)(nil)\n\t_ Task = (*StartChildExecutionTask)(nil)\n\t_ Task = (*RecordWorkflowClosedTask)(nil)\n\t_ Task = (*ActivityTimeoutTask)(nil)\n\t_ Task = (*UserTimerTask)(nil)\n\t_ Task = (*ActivityRetryTimerTask)(nil)\n\t_ Task = (*WorkflowBackoffTimerTask)(nil)\n\t_ Task = (*HistoryReplicationTask)(nil)\n\t_ Task = (*SyncActivityTask)(nil)\n\t_ Task = (*FailoverMarkerTask)(nil)\n\n\timmediateTaskKeyScheduleTime = time.Unix(0, 0).UTC()\n)\n\nfunc IsTaskCorrupted(task Task) bool {\n\tswitch task.(type) {\n\tcase *FailoverMarkerTask:\n\t\treturn task.GetDomainID() == \"\"\n\tdefault:\n\t\treturn task.GetDomainID() == \"\" || task.GetWorkflowID() == \"\" || task.GetRunID() == \"\"\n\t}\n}\n\nfunc NewImmediateTaskKey(taskID int64) HistoryTaskKey {\n\treturn HistoryTaskKey{\n\t\tscheduledTime: immediateTaskKeyScheduleTime,\n\t\ttaskID:        taskID,\n\t}\n}\n\nfunc NewHistoryTaskKey(scheduledTime time.Time, taskID int64) HistoryTaskKey {\n\treturn HistoryTaskKey{\n\t\tscheduledTime: scheduledTime,\n\t\ttaskID:        taskID,\n\t}\n}\n\nfunc (a HistoryTaskKey) GetTaskID() int64 {\n\treturn a.taskID\n}\n\nfunc (a HistoryTaskKey) GetScheduledTime() time.Time {\n\treturn a.scheduledTime\n}\n\nfunc (a HistoryTaskKey) Compare(b HistoryTaskKey) int {\n\tif a.scheduledTime.Before(b.scheduledTime) {\n\t\treturn -1\n\t} else if a.scheduledTime.After(b.scheduledTime) {\n\t\treturn 1\n\t}\n\tif a.taskID < b.taskID {\n\t\treturn -1\n\t} else if a.taskID > b.taskID {\n\t\treturn 1\n\t}\n\treturn 0\n}\n\nfunc (a HistoryTaskKey) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(map[string]interface{}{\n\t\t\"scheduledTime\": a.scheduledTime.UTC(),\n\t\t\"taskID\":        a.taskID,\n\t})\n}\n\nfunc (a HistoryTaskKey) Next() HistoryTaskKey {\n\tif a.taskID == math.MaxInt64 {\n\t\treturn HistoryTaskKey{\n\t\t\tscheduledTime: a.scheduledTime.Add(time.Nanosecond),\n\t\t\ttaskID:        0,\n\t\t}\n\t}\n\treturn HistoryTaskKey{\n\t\tscheduledTime: a.scheduledTime,\n\t\ttaskID:        a.taskID + 1,\n\t}\n}\n\nfunc MinHistoryTaskKey(a, b HistoryTaskKey) HistoryTaskKey {\n\tif a.Compare(b) < 0 {\n\t\treturn a\n\t}\n\treturn b\n}\n\nfunc MaxHistoryTaskKey(a, b HistoryTaskKey) HistoryTaskKey {\n\tif a.Compare(b) > 0 {\n\t\treturn a\n\t}\n\treturn b\n}\n\nfunc (a *WorkflowIdentifier) GetDomainID() string {\n\treturn a.DomainID\n}\n\nfunc (a *WorkflowIdentifier) GetWorkflowID() string {\n\treturn a.WorkflowID\n}\n\nfunc (a *WorkflowIdentifier) GetRunID() string {\n\treturn a.RunID\n}\n\nfunc (a *WorkflowIdentifier) ByteSize() uint64 {\n\treturn uint64(len(a.DomainID) + len(a.WorkflowID) + len(a.RunID))\n}\n\n// GetVersion returns the version of the task\nfunc (a *TaskData) GetVersion() int64 {\n\treturn a.Version\n}\n\n// SetVersion sets the version of the task\nfunc (a *TaskData) SetVersion(version int64) {\n\ta.Version = version\n}\n\n// GetTaskID returns the sequence ID of the task\nfunc (a *TaskData) GetTaskID() int64 {\n\treturn a.TaskID\n}\n\n// SetTaskID sets the sequence ID of the task\nfunc (a *TaskData) SetTaskID(id int64) {\n\ta.TaskID = id\n}\n\n// GetVisibilityTimestamp get the visibility timestamp\nfunc (a *TaskData) GetVisibilityTimestamp() time.Time {\n\treturn a.VisibilityTimestamp\n}\n\n// SetVisibilityTimestamp set the visibility timestamp\nfunc (a *TaskData) SetVisibilityTimestamp(timestamp time.Time) {\n\ta.VisibilityTimestamp = timestamp\n}\n\nfunc (a *TaskData) ByteSize() uint64 {\n\treturn uint64(8 + 8 + 24) // time.Time is 24 bytes\n}\n\n// GetType returns the type of the activity task\nfunc (a *ActivityTask) GetTaskType() int {\n\treturn TransferTaskTypeActivityTask\n}\n\nfunc (a *ActivityTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (a *ActivityTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *ActivityTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ActivityTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ActivityTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *ActivityTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + uint64(len(a.TargetDomainID)) + uint64(len(a.TaskList)) + 8\n}\n\nfunc (a *ActivityTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeActivityTask,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tTargetDomainID:      a.TargetDomainID,\n\t\tTaskList:            a.TaskList,\n\t\tScheduleID:          a.ScheduleID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (a *ActivityTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity task is not timer task\")\n}\n\nfunc (a *ActivityTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity task is not replication task\")\n}\n\n// GetType returns the type of the decision task\nfunc (d *DecisionTask) GetTaskType() int {\n\treturn TransferTaskTypeDecisionTask\n}\n\nfunc (d *DecisionTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (d *DecisionTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(d.TaskID)\n}\n\nfunc (d *DecisionTask) GetTaskList() string {\n\treturn d.TaskList\n}\n\nfunc (d *DecisionTask) GetOriginalTaskList() string {\n\treturn d.OriginalTaskList\n}\n\nfunc (d *DecisionTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn d.OriginalTaskListKind\n}\n\nfunc (d *DecisionTask) ByteSize() uint64 {\n\treturn d.WorkflowIdentifier.ByteSize() + d.TaskData.ByteSize() + uint64(len(d.TargetDomainID)) + uint64(len(d.TaskList)) + uint64(len(d.OriginalTaskList)) + 16\n}\n\nfunc (d *DecisionTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:             TransferTaskTypeDecisionTask,\n\t\tDomainID:             d.DomainID,\n\t\tWorkflowID:           d.WorkflowID,\n\t\tRunID:                d.RunID,\n\t\tTaskID:               d.TaskID,\n\t\tVisibilityTimestamp:  d.VisibilityTimestamp,\n\t\tVersion:              d.Version,\n\t\tTargetDomainID:       d.TargetDomainID,\n\t\tTaskList:             d.TaskList,\n\t\tScheduleID:           d.ScheduleID,\n\t\tOriginalTaskList:     d.OriginalTaskList,\n\t\tOriginalTaskListKind: d.OriginalTaskListKind,\n\t\tTargetWorkflowID:     TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:          TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (d *DecisionTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"decision task is not timer task\")\n}\n\nfunc (d *DecisionTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"decision task is not replication task\")\n}\n\n// GetType returns the type of the record workflow started task\nfunc (a *RecordWorkflowStartedTask) GetTaskType() int {\n\treturn TransferTaskTypeRecordWorkflowStarted\n}\n\nfunc (a *RecordWorkflowStartedTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (a *RecordWorkflowStartedTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *RecordWorkflowStartedTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *RecordWorkflowStartedTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *RecordWorkflowStartedTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *RecordWorkflowStartedTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + uint64(len(a.TaskList))\n}\n\nfunc (a *RecordWorkflowStartedTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeRecordWorkflowStarted,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tTaskList:            a.TaskList,\n\t\tTargetDomainID:      a.DomainID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (a *RecordWorkflowStartedTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record workflow started task is not timer task\")\n}\n\nfunc (a *RecordWorkflowStartedTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record workflow started task is not replication task\")\n}\n\n// GetType returns the type of the ResetWorkflowTask\nfunc (a *ResetWorkflowTask) GetTaskType() int {\n\treturn TransferTaskTypeResetWorkflow\n}\n\nfunc (a *ResetWorkflowTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (a *ResetWorkflowTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *ResetWorkflowTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ResetWorkflowTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ResetWorkflowTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *ResetWorkflowTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + uint64(len(a.TaskList))\n}\n\nfunc (a *ResetWorkflowTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeResetWorkflow,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tTaskList:            a.TaskList,\n\t\tTargetDomainID:      a.DomainID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (a *ResetWorkflowTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"reset workflow task is not timer task\")\n}\n\nfunc (a *ResetWorkflowTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"reset workflow task is not replication task\")\n}\n\n// GetType returns the type of the close execution task\nfunc (a *CloseExecutionTask) GetTaskType() int {\n\treturn TransferTaskTypeCloseExecution\n}\n\nfunc (a *CloseExecutionTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (a *CloseExecutionTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *CloseExecutionTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *CloseExecutionTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *CloseExecutionTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *CloseExecutionTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + uint64(len(a.TaskList))\n}\n\nfunc (a *CloseExecutionTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeCloseExecution,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tTaskList:            a.TaskList,\n\t\tTargetDomainID:      a.DomainID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (a *CloseExecutionTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"close execution task is not timer task\")\n}\n\nfunc (a *CloseExecutionTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"close execution task is not replication task\")\n}\n\n// GetType returns the type of the delete execution task\nfunc (a *DeleteHistoryEventTask) GetTaskType() int {\n\treturn TaskTypeDeleteHistoryEvent\n}\n\nfunc (a *DeleteHistoryEventTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (a *DeleteHistoryEventTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(a.VisibilityTimestamp, a.TaskID)\n}\n\nfunc (a *DeleteHistoryEventTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *DeleteHistoryEventTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *DeleteHistoryEventTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *DeleteHistoryEventTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + uint64(len(a.TaskList))\n}\n\nfunc (a *DeleteHistoryEventTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"delete history event task is not transfer task\")\n}\n\nfunc (a *DeleteHistoryEventTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeDeleteHistoryEvent,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tTaskList:            a.TaskList,\n\t}, nil\n}\n\nfunc (a *DeleteHistoryEventTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"delete history event task is not replication task\")\n}\n\n// GetType returns the type of the timer task\nfunc (d *DecisionTimeoutTask) GetTaskType() int {\n\treturn TaskTypeDecisionTimeout\n}\n\nfunc (d *DecisionTimeoutTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (d *DecisionTimeoutTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(d.VisibilityTimestamp, d.TaskID)\n}\n\nfunc (d *DecisionTimeoutTask) GetTaskList() string {\n\treturn d.TaskList\n}\n\nfunc (d *DecisionTimeoutTask) GetOriginalTaskList() string {\n\treturn d.TaskList\n}\n\nfunc (d *DecisionTimeoutTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (d *DecisionTimeoutTask) ByteSize() uint64 {\n\treturn d.WorkflowIdentifier.ByteSize() + d.TaskData.ByteSize() + 8 + 8 + 8 + uint64(len(d.TaskList))\n}\n\nfunc (d *DecisionTimeoutTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"decision timeout task is not transfer task\")\n}\n\nfunc (d *DecisionTimeoutTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeDecisionTimeout,\n\t\tDomainID:            d.DomainID,\n\t\tWorkflowID:          d.WorkflowID,\n\t\tRunID:               d.RunID,\n\t\tTaskID:              d.TaskID,\n\t\tVisibilityTimestamp: d.VisibilityTimestamp,\n\t\tVersion:             d.Version,\n\t\tEventID:             d.EventID,\n\t\tScheduleAttempt:     d.ScheduleAttempt,\n\t\tTimeoutType:         d.TimeoutType,\n\t\tTaskList:            d.TaskList,\n\t}, nil\n}\n\nfunc (d *DecisionTimeoutTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"decision timeout task is not replication task\")\n}\n\n// GetType returns the type of the timer task\nfunc (a *ActivityTimeoutTask) GetTaskType() int {\n\treturn TaskTypeActivityTimeout\n}\n\nfunc (a *ActivityTimeoutTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (a *ActivityTimeoutTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(a.VisibilityTimestamp, a.TaskID)\n}\n\nfunc (a *ActivityTimeoutTask) GetTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ActivityTimeoutTask) GetOriginalTaskList() string {\n\treturn a.TaskList\n}\n\nfunc (a *ActivityTimeoutTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *ActivityTimeoutTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + 8 + 8 + 8 + uint64(len(a.TaskList))\n}\n\nfunc (a *ActivityTimeoutTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity timeout task is not transfer task\")\n}\n\nfunc (a *ActivityTimeoutTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeActivityTimeout,\n\t\tDomainID:            a.DomainID,\n\t\tWorkflowID:          a.WorkflowID,\n\t\tRunID:               a.RunID,\n\t\tTaskID:              a.TaskID,\n\t\tVisibilityTimestamp: a.VisibilityTimestamp,\n\t\tVersion:             a.Version,\n\t\tEventID:             a.EventID,\n\t\tScheduleAttempt:     a.Attempt,\n\t\tTimeoutType:         a.TimeoutType,\n\t\tTaskList:            a.TaskList,\n\t}, nil\n}\n\nfunc (a *ActivityTimeoutTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity timeout task is not replication task\")\n}\n\n// GetType returns the type of the timer task\nfunc (u *UserTimerTask) GetTaskType() int {\n\treturn TaskTypeUserTimer\n}\n\nfunc (u *UserTimerTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (u *UserTimerTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(u.VisibilityTimestamp, u.TaskID)\n}\n\nfunc (u *UserTimerTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *UserTimerTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *UserTimerTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *UserTimerTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + 8 + uint64(len(u.TaskList))\n}\n\nfunc (u *UserTimerTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"user timer task is not transfer task\")\n}\n\nfunc (u *UserTimerTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeUserTimer,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tEventID:             u.EventID,\n\t\tTaskList:            u.TaskList,\n\t}, nil\n}\n\nfunc (u *UserTimerTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"user timer task is not replication task\")\n}\n\n// GetType returns the type of the retry timer task\nfunc (r *ActivityRetryTimerTask) GetTaskType() int {\n\treturn TaskTypeActivityRetryTimer\n}\n\nfunc (r *ActivityRetryTimerTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (r *ActivityRetryTimerTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(r.VisibilityTimestamp, r.TaskID)\n}\n\nfunc (r *ActivityRetryTimerTask) GetTaskList() string {\n\treturn r.TaskList\n}\n\nfunc (r *ActivityRetryTimerTask) GetOriginalTaskList() string {\n\treturn r.TaskList\n}\n\nfunc (r *ActivityRetryTimerTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (r *ActivityRetryTimerTask) ByteSize() uint64 {\n\treturn r.WorkflowIdentifier.ByteSize() + r.TaskData.ByteSize() + 8 + 8 + uint64(len(r.TaskList))\n}\n\nfunc (r *ActivityRetryTimerTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity retry timer task is not transfer task\")\n}\n\nfunc (r *ActivityRetryTimerTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeActivityRetryTimer,\n\t\tDomainID:            r.DomainID,\n\t\tWorkflowID:          r.WorkflowID,\n\t\tRunID:               r.RunID,\n\t\tTaskID:              r.TaskID,\n\t\tVisibilityTimestamp: r.VisibilityTimestamp,\n\t\tVersion:             r.Version,\n\t\tEventID:             r.EventID,\n\t\tScheduleAttempt:     r.Attempt,\n\t\tTaskList:            r.TaskList,\n\t}, nil\n}\n\nfunc (r *ActivityRetryTimerTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"activity retry timer task is not replication task\")\n}\n\n// GetType returns the type of the retry timer task\nfunc (r *WorkflowBackoffTimerTask) GetTaskType() int {\n\treturn TaskTypeWorkflowBackoffTimer\n}\n\nfunc (r *WorkflowBackoffTimerTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (r *WorkflowBackoffTimerTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(r.VisibilityTimestamp, r.TaskID)\n}\n\nfunc (r *WorkflowBackoffTimerTask) GetTaskList() string {\n\treturn r.TaskList\n}\n\nfunc (r *WorkflowBackoffTimerTask) GetOriginalTaskList() string {\n\treturn r.TaskList\n}\n\nfunc (r *WorkflowBackoffTimerTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (r *WorkflowBackoffTimerTask) ByteSize() uint64 {\n\treturn r.WorkflowIdentifier.ByteSize() + r.TaskData.ByteSize() + 8 + uint64(len(r.TaskList))\n}\n\nfunc (r *WorkflowBackoffTimerTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"workflow backoff timer task is not transfer task\")\n}\n\nfunc (r *WorkflowBackoffTimerTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeWorkflowBackoffTimer,\n\t\tDomainID:            r.DomainID,\n\t\tWorkflowID:          r.WorkflowID,\n\t\tRunID:               r.RunID,\n\t\tTaskID:              r.TaskID,\n\t\tVisibilityTimestamp: r.VisibilityTimestamp,\n\t\tVersion:             r.Version,\n\t\tTimeoutType:         r.TimeoutType,\n\t\tTaskList:            r.TaskList,\n\t}, nil\n}\n\nfunc (r *WorkflowBackoffTimerTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"workflow backoff timer task is not replication task\")\n}\n\n// GetType returns the type of the timeout task.\nfunc (u *WorkflowTimeoutTask) GetTaskType() int {\n\treturn TaskTypeWorkflowTimeout\n}\n\nfunc (u *WorkflowTimeoutTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTimer\n}\n\nfunc (u *WorkflowTimeoutTask) GetTaskKey() HistoryTaskKey {\n\treturn NewHistoryTaskKey(u.VisibilityTimestamp, u.TaskID)\n}\n\nfunc (u *WorkflowTimeoutTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *WorkflowTimeoutTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *WorkflowTimeoutTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *WorkflowTimeoutTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TaskList))\n}\n\nfunc (u *WorkflowTimeoutTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"workflow timeout task is not transfer task\")\n}\n\nfunc (u *WorkflowTimeoutTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn &TimerTaskInfo{\n\t\tTaskType:            TaskTypeWorkflowTimeout,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tTaskList:            u.TaskList,\n\t}, nil\n}\n\nfunc (u *WorkflowTimeoutTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"workflow timeout task is not replication task\")\n}\n\n// GetType returns the type of the cancel transfer task\nfunc (u *CancelExecutionTask) GetTaskType() int {\n\treturn TransferTaskTypeCancelExecution\n}\n\nfunc (u *CancelExecutionTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *CancelExecutionTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *CancelExecutionTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *CancelExecutionTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *CancelExecutionTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *CancelExecutionTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TargetDomainID)) + uint64(len(u.TargetWorkflowID)) + uint64(len(u.TargetRunID)) + 8 + 1 + uint64(len(u.TaskList))\n}\n\nfunc (u *CancelExecutionTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\ttargetRunID := u.TargetRunID\n\tif u.TargetRunID == \"\" {\n\t\ttargetRunID = TransferTaskTransferTargetRunID\n\t}\n\treturn &TransferTaskInfo{\n\t\tTaskType:                TransferTaskTypeCancelExecution,\n\t\tDomainID:                u.DomainID,\n\t\tWorkflowID:              u.WorkflowID,\n\t\tRunID:                   u.RunID,\n\t\tTaskID:                  u.TaskID,\n\t\tVisibilityTimestamp:     u.VisibilityTimestamp,\n\t\tVersion:                 u.Version,\n\t\tTargetDomainID:          u.TargetDomainID,\n\t\tTargetWorkflowID:        u.TargetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: u.TargetChildWorkflowOnly,\n\t\tScheduleID:              u.InitiatedID,\n\t\tTaskList:                u.TaskList,\n\t}, nil\n}\n\nfunc (u *CancelExecutionTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"cancel execution task is not timer task\")\n}\n\nfunc (u *CancelExecutionTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"cancel execution task is not replication task\")\n}\n\n// GetType returns the type of the signal transfer task\nfunc (u *SignalExecutionTask) GetTaskType() int {\n\treturn TransferTaskTypeSignalExecution\n}\n\nfunc (u *SignalExecutionTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *SignalExecutionTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *SignalExecutionTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *SignalExecutionTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *SignalExecutionTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *SignalExecutionTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TargetDomainID)) + uint64(len(u.TargetWorkflowID)) + uint64(len(u.TargetRunID)) + 8 + 1 + uint64(len(u.TaskList))\n}\n\nfunc (u *SignalExecutionTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\ttargetRunID := u.TargetRunID\n\tif u.TargetRunID == \"\" {\n\t\ttargetRunID = TransferTaskTransferTargetRunID\n\t}\n\treturn &TransferTaskInfo{\n\t\tTaskType:                TransferTaskTypeSignalExecution,\n\t\tDomainID:                u.DomainID,\n\t\tWorkflowID:              u.WorkflowID,\n\t\tRunID:                   u.RunID,\n\t\tTaskID:                  u.TaskID,\n\t\tVisibilityTimestamp:     u.VisibilityTimestamp,\n\t\tVersion:                 u.Version,\n\t\tTargetDomainID:          u.TargetDomainID,\n\t\tTargetWorkflowID:        u.TargetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: u.TargetChildWorkflowOnly,\n\t\tScheduleID:              u.InitiatedID,\n\t\tTaskList:                u.TaskList,\n\t}, nil\n}\n\nfunc (u *SignalExecutionTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"signal execution task is not timer task\")\n}\n\nfunc (u *SignalExecutionTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"signal execution task is not replication task\")\n}\n\n// GetType returns the type of the record child execution completed task\nfunc (u *RecordChildExecutionCompletedTask) GetTaskType() int {\n\treturn TransferTaskTypeRecordChildExecutionCompleted\n}\n\nfunc (u *RecordChildExecutionCompletedTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *RecordChildExecutionCompletedTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *RecordChildExecutionCompletedTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *RecordChildExecutionCompletedTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *RecordChildExecutionCompletedTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *RecordChildExecutionCompletedTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TargetDomainID)) + uint64(len(u.TargetWorkflowID)) + uint64(len(u.TargetRunID)) + uint64(len(u.TaskList))\n}\n\nfunc (u *RecordChildExecutionCompletedTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\ttargetRunID := u.TargetRunID\n\tif u.TargetRunID == \"\" {\n\t\ttargetRunID = TransferTaskTransferTargetRunID\n\t}\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeRecordChildExecutionCompleted,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tTargetDomainID:      u.TargetDomainID,\n\t\tTargetWorkflowID:    u.TargetWorkflowID,\n\t\tTargetRunID:         targetRunID,\n\t\tTaskList:            u.TaskList,\n\t}, nil\n}\n\nfunc (u *RecordChildExecutionCompletedTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record child execution completed task is not timer task\")\n}\n\nfunc (u *RecordChildExecutionCompletedTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record child execution completed task is not replication task\")\n}\n\n// GetType returns the type of the upsert search attributes transfer task\nfunc (u *UpsertWorkflowSearchAttributesTask) GetTaskType() int {\n\treturn TransferTaskTypeUpsertWorkflowSearchAttributes\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TaskList))\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeUpsertWorkflowSearchAttributes,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tTaskList:            u.TaskList,\n\t\tTargetDomainID:      u.DomainID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"upsert workflow search attributes task is not timer task\")\n}\n\nfunc (u *UpsertWorkflowSearchAttributesTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"upsert workflow search attributes task is not replication task\")\n}\n\n// GetType returns the type of the start child transfer task\nfunc (u *StartChildExecutionTask) GetTaskType() int {\n\treturn TransferTaskTypeStartChildExecution\n}\n\nfunc (u *StartChildExecutionTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *StartChildExecutionTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *StartChildExecutionTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *StartChildExecutionTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *StartChildExecutionTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *StartChildExecutionTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TargetDomainID)) + uint64(len(u.TargetWorkflowID)) + 8 + uint64(len(u.TaskList))\n}\n\nfunc (u *StartChildExecutionTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeStartChildExecution,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tTargetDomainID:      u.TargetDomainID,\n\t\tTargetWorkflowID:    u.TargetWorkflowID,\n\t\tScheduleID:          u.InitiatedID,\n\t\tTaskList:            u.TaskList,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (u *StartChildExecutionTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"start child execution task is not timer task\")\n}\n\nfunc (u *StartChildExecutionTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"start child execution task is not replication task\")\n}\n\n// GetType returns the type of the record workflow closed task\nfunc (u *RecordWorkflowClosedTask) GetTaskType() int {\n\treturn TransferTaskTypeRecordWorkflowClosed\n}\n\nfunc (u *RecordWorkflowClosedTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryTransfer\n}\n\nfunc (u *RecordWorkflowClosedTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(u.TaskID)\n}\n\nfunc (u *RecordWorkflowClosedTask) GetTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *RecordWorkflowClosedTask) GetOriginalTaskList() string {\n\treturn u.TaskList\n}\n\nfunc (u *RecordWorkflowClosedTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (u *RecordWorkflowClosedTask) ByteSize() uint64 {\n\treturn u.WorkflowIdentifier.ByteSize() + u.TaskData.ByteSize() + uint64(len(u.TaskList))\n}\n\nfunc (u *RecordWorkflowClosedTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn &TransferTaskInfo{\n\t\tTaskType:            TransferTaskTypeRecordWorkflowClosed,\n\t\tDomainID:            u.DomainID,\n\t\tWorkflowID:          u.WorkflowID,\n\t\tRunID:               u.RunID,\n\t\tTaskID:              u.TaskID,\n\t\tVisibilityTimestamp: u.VisibilityTimestamp,\n\t\tVersion:             u.Version,\n\t\tTaskList:            u.TaskList,\n\t\tTargetDomainID:      u.DomainID,\n\t\tTargetWorkflowID:    TransferTaskTransferTargetWorkflowID,\n\t\tTargetRunID:         TransferTaskTransferTargetRunID,\n\t}, nil\n}\n\nfunc (u *RecordWorkflowClosedTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record workflow closed task is not timer task\")\n}\n\nfunc (u *RecordWorkflowClosedTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"record workflow closed task is not replication task\")\n}\n\n// GetType returns the type of the history replication task\nfunc (a *HistoryReplicationTask) GetTaskType() int {\n\treturn ReplicationTaskTypeHistory\n}\n\nfunc (a *HistoryReplicationTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryReplication\n}\n\nfunc (a *HistoryReplicationTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *HistoryReplicationTask) GetTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *HistoryReplicationTask) GetOriginalTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *HistoryReplicationTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *HistoryReplicationTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + 8 + 8 + uint64(len(a.BranchToken)) + uint64(len(a.NewRunBranchToken))\n}\n\nfunc (a *HistoryReplicationTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"history replication task is not transfer task\")\n}\n\nfunc (a *HistoryReplicationTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"history replication task is not timer task\")\n}\n\nfunc (a *HistoryReplicationTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn &types.ReplicationTaskInfo{\n\t\tDomainID:     a.DomainID,\n\t\tWorkflowID:   a.WorkflowID,\n\t\tRunID:        a.RunID,\n\t\tTaskType:     ReplicationTaskTypeHistory,\n\t\tTaskID:       a.TaskID,\n\t\tVersion:      a.Version,\n\t\tFirstEventID: a.FirstEventID,\n\t\tNextEventID:  a.NextEventID,\n\t\tScheduledID:  constants.EmptyEventID,\n\t}, nil\n}\n\n// GetType returns the type of the sync activity task\nfunc (a *SyncActivityTask) GetTaskType() int {\n\treturn ReplicationTaskTypeSyncActivity\n}\n\nfunc (a *SyncActivityTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryReplication\n}\n\nfunc (a *SyncActivityTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *SyncActivityTask) GetTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *SyncActivityTask) GetOriginalTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *SyncActivityTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *SyncActivityTask) ByteSize() uint64 {\n\treturn a.WorkflowIdentifier.ByteSize() + a.TaskData.ByteSize() + 8\n}\n\nfunc (a *SyncActivityTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn &types.ReplicationTaskInfo{\n\t\tDomainID:     a.DomainID,\n\t\tWorkflowID:   a.WorkflowID,\n\t\tRunID:        a.RunID,\n\t\tTaskType:     ReplicationTaskTypeSyncActivity,\n\t\tTaskID:       a.TaskID,\n\t\tVersion:      a.Version,\n\t\tFirstEventID: constants.EmptyEventID,\n\t\tNextEventID:  constants.EmptyEventID,\n\t\tScheduledID:  a.ScheduledID,\n\t}, nil\n}\n\nfunc (a *SyncActivityTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"sync activity task is not transfer task\")\n}\n\nfunc (a *SyncActivityTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"sync activity task is not timer task\")\n}\n\n// GetType returns the type of the history replication task\nfunc (a *FailoverMarkerTask) GetTaskType() int {\n\treturn ReplicationTaskTypeFailoverMarker\n}\n\nfunc (a *FailoverMarkerTask) GetTaskCategory() HistoryTaskCategory {\n\treturn HistoryTaskCategoryReplication\n}\n\nfunc (a *FailoverMarkerTask) GetTaskKey() HistoryTaskKey {\n\treturn NewImmediateTaskKey(a.TaskID)\n}\n\nfunc (a *FailoverMarkerTask) GetTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *FailoverMarkerTask) GetOriginalTaskList() string {\n\treturn \"\"\n}\n\nfunc (a *FailoverMarkerTask) GetOriginalTaskListKind() types.TaskListKind {\n\treturn types.TaskListKindNormal\n}\n\nfunc (a *FailoverMarkerTask) ByteSize() uint64 {\n\treturn uint64(len(a.DomainID)) + a.TaskData.ByteSize()\n}\n\nfunc (a *FailoverMarkerTask) ToTransferTaskInfo() (*TransferTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"failover marker task is not transfer task\")\n}\n\nfunc (a *FailoverMarkerTask) ToTimerTaskInfo() (*TimerTaskInfo, error) {\n\treturn nil, fmt.Errorf(\"failover marker task is not timer task\")\n}\n\nfunc (a *FailoverMarkerTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\treturn &types.ReplicationTaskInfo{\n\t\tDomainID:     a.DomainID,\n\t\tTaskType:     ReplicationTaskTypeFailoverMarker,\n\t\tTaskID:       a.TaskID,\n\t\tVersion:      a.Version,\n\t\tFirstEventID: constants.EmptyEventID,\n\t\tNextEventID:  constants.EmptyEventID,\n\t\tScheduledID:  constants.EmptyEventID,\n\t}, nil\n}\n\nfunc (a *FailoverMarkerTask) GetDomainID() string {\n\treturn a.DomainID\n}\n\nfunc (a *FailoverMarkerTask) GetWorkflowID() string {\n\treturn \"\"\n}\n\nfunc (a *FailoverMarkerTask) GetRunID() string {\n\treturn \"\"\n}\n"
  },
  {
    "path": "common/persistence/tasks_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestTaskCommonMethods(t *testing.T) {\n\ttimeNow := time.Now()\n\ttasks := []Task{\n\t\t&ActivityTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowStartedTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ResetWorkflowTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CloseExecutionTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DeleteHistoryEventTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTimeoutTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityTimeoutTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UserTimerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityRetryTimerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowBackoffTimerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowTimeoutTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CancelExecutionTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SignalExecutionTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordChildExecutionCompletedTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UpsertWorkflowSearchAttributesTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&StartChildExecutionTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowClosedTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&HistoryReplicationTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SyncActivityTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&FailoverMarkerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t}\n\n\tfor _, task := range tasks {\n\t\tswitch ty := task.(type) {\n\t\tcase *ActivityTask:\n\t\t\tassert.Equal(t, TransferTaskTypeActivityTask, ty.GetTaskType())\n\t\tcase *DecisionTask:\n\t\t\tassert.Equal(t, TransferTaskTypeDecisionTask, ty.GetTaskType())\n\t\tcase *RecordWorkflowStartedTask:\n\t\t\tassert.Equal(t, TransferTaskTypeRecordWorkflowStarted, ty.GetTaskType())\n\t\tcase *ResetWorkflowTask:\n\t\t\tassert.Equal(t, TransferTaskTypeResetWorkflow, ty.GetTaskType())\n\t\tcase *CloseExecutionTask:\n\t\t\tassert.Equal(t, TransferTaskTypeCloseExecution, ty.GetTaskType())\n\t\tcase *DeleteHistoryEventTask:\n\t\t\tassert.Equal(t, TaskTypeDeleteHistoryEvent, ty.GetTaskType())\n\t\tcase *DecisionTimeoutTask:\n\t\t\tassert.Equal(t, TaskTypeDecisionTimeout, ty.GetTaskType())\n\t\tcase *ActivityTimeoutTask:\n\t\t\tassert.Equal(t, TaskTypeActivityTimeout, ty.GetTaskType())\n\t\tcase *UserTimerTask:\n\t\t\tassert.Equal(t, TaskTypeUserTimer, ty.GetTaskType())\n\t\tcase *ActivityRetryTimerTask:\n\t\t\tassert.Equal(t, TaskTypeActivityRetryTimer, ty.GetTaskType())\n\t\tcase *WorkflowBackoffTimerTask:\n\t\t\tassert.Equal(t, TaskTypeWorkflowBackoffTimer, ty.GetTaskType())\n\t\tcase *WorkflowTimeoutTask:\n\t\t\tassert.Equal(t, TaskTypeWorkflowTimeout, ty.GetTaskType())\n\t\tcase *CancelExecutionTask:\n\t\t\tassert.Equal(t, TransferTaskTypeCancelExecution, ty.GetTaskType())\n\t\tcase *SignalExecutionTask:\n\t\t\tassert.Equal(t, TransferTaskTypeSignalExecution, ty.GetTaskType())\n\t\tcase *RecordChildExecutionCompletedTask:\n\t\t\tassert.Equal(t, TransferTaskTypeRecordChildExecutionCompleted, ty.GetTaskType())\n\t\tcase *UpsertWorkflowSearchAttributesTask:\n\t\t\tassert.Equal(t, TransferTaskTypeUpsertWorkflowSearchAttributes, ty.GetTaskType())\n\t\tcase *StartChildExecutionTask:\n\t\t\tassert.Equal(t, TransferTaskTypeStartChildExecution, ty.GetTaskType())\n\t\tcase *RecordWorkflowClosedTask:\n\t\t\tassert.Equal(t, TransferTaskTypeRecordWorkflowClosed, ty.GetTaskType())\n\t\tcase *HistoryReplicationTask:\n\t\t\tassert.Equal(t, ReplicationTaskTypeHistory, ty.GetTaskType())\n\t\tcase *SyncActivityTask:\n\t\t\tassert.Equal(t, ReplicationTaskTypeSyncActivity, ty.GetTaskType())\n\t\tcase *FailoverMarkerTask:\n\t\t\tassert.Equal(t, ReplicationTaskTypeFailoverMarker, ty.GetTaskType())\n\t\tdefault:\n\t\t\tt.Fatalf(\"Unhandled task type: %T\", t)\n\t\t}\n\n\t\t// Test version methods\n\t\tassert.Equal(t, int64(1), task.GetVersion())\n\t\ttask.SetVersion(2)\n\t\tassert.Equal(t, int64(2), task.GetVersion())\n\n\t\t// Test TaskID methods\n\t\tassert.Equal(t, int64(1), task.GetTaskID())\n\t\ttask.SetTaskID(2)\n\t\tassert.Equal(t, int64(2), task.GetTaskID())\n\n\t\t// Test VisibilityTimestamp methods\n\t\tassert.Equal(t, timeNow, task.GetVisibilityTimestamp())\n\t\tnewTime := timeNow.Add(time.Second)\n\t\ttask.SetVisibilityTimestamp(newTime)\n\t\tassert.Equal(t, newTime, task.GetVisibilityTimestamp())\n\n\t\tif task.GetTaskCategory().Type() == HistoryTaskCategoryTypeImmediate {\n\t\t\tassert.Equal(t, NewImmediateTaskKey(task.GetTaskID()), task.GetTaskKey())\n\t\t} else {\n\t\t\tassert.Equal(t, NewHistoryTaskKey(task.GetVisibilityTimestamp(), task.GetTaskID()), task.GetTaskKey())\n\t\t}\n\t}\n}\n\nfunc TestTransferTaskMapping(t *testing.T) {\n\tf := fuzz.New().NilChance(0.0)\n\ttasks := []Task{\n\t\t&ActivityTask{},\n\t\t&DecisionTask{},\n\t\t&RecordWorkflowStartedTask{},\n\t\t&ResetWorkflowTask{},\n\t\t&CloseExecutionTask{},\n\t\t&CancelExecutionTask{},\n\t\t&SignalExecutionTask{},\n\t\t&RecordChildExecutionCompletedTask{},\n\t\t&UpsertWorkflowSearchAttributesTask{},\n\t\t&StartChildExecutionTask{},\n\t\t&RecordWorkflowClosedTask{},\n\t}\n\tfor i := 0; i < 1000; i++ {\n\t\tfor _, task := range tasks {\n\t\t\tf.Fuzz(task)\n\t\t\ttransfer, err := task.ToTransferTaskInfo()\n\t\t\tassert.NoError(t, err)\n\t\t\tt.Log(transfer)\n\t\t\ttask2, err := transfer.ToTask()\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, task, task2)\n\t\t}\n\t}\n}\n\nfunc TestTimerTaskMapping(t *testing.T) {\n\tf := fuzz.New().NilChance(0.0)\n\ttasks := []Task{\n\t\t&DecisionTimeoutTask{},\n\t\t&ActivityTimeoutTask{},\n\t\t&DeleteHistoryEventTask{},\n\t\t&WorkflowTimeoutTask{},\n\t\t&UserTimerTask{},\n\t\t&ActivityRetryTimerTask{},\n\t\t&WorkflowBackoffTimerTask{},\n\t}\n\tfor i := 0; i < 1000; i++ {\n\t\tfor _, task := range tasks {\n\t\t\tf.Fuzz(task)\n\t\t\ttimer, err := task.ToTimerTaskInfo()\n\t\t\tassert.NoError(t, err)\n\t\t\ttask2, err := timer.ToTask()\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, task, task2)\n\t\t}\n\t}\n}\n\nfunc TestHistoryTaskKeyComparison(t *testing.T) {\n\tnow := time.Now()\n\tkey1 := NewHistoryTaskKey(now, 1)\n\tkey2 := NewHistoryTaskKey(now, 2)\n\tkey3 := NewHistoryTaskKey(now.Add(time.Second), 1)\n\n\tassert.Equal(t, 0, key1.Compare(key1))\n\tassert.Equal(t, -1, key1.Compare(key2))\n\tassert.Equal(t, 1, key2.Compare(key1))\n\n\tassert.Equal(t, -1, key2.Compare(key3))\n\tassert.Equal(t, 1, key3.Compare(key2))\n}\n\nfunc TestIsTaskCorrupted(t *testing.T) {\n\ttimeNow := time.Now()\n\n\ttests := []struct {\n\t\tname     string\n\t\ttask     Task\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname: \"Valid ActivityTask\",\n\t\t\ttask: &ActivityTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Valid DecisionTask\",\n\t\t\ttask: &DecisionTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Valid TimerTask\",\n\t\t\ttask: &UserTimerTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tEventID: 123,\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Valid ReplicationTask\",\n\t\t\ttask: &HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tFirstEventID: 1,\n\t\t\t\tNextEventID:  10,\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with empty DomainID\",\n\t\t\ttask: &ActivityTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with empty WorkflowID\",\n\t\t\ttask: &DecisionTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with empty RunID\",\n\t\t\ttask: &UserTimerTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tEventID: 123,\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with all empty identifiers\",\n\t\t\ttask: &ActivityTimeoutTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"\",\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tTimeoutType: 1,\n\t\t\t\tEventID:     123,\n\t\t\t\tAttempt:     1,\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with whitespace-only identifiers\",\n\t\t\ttask: &WorkflowTimeoutTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"   \",\n\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: false, // Whitespace is not empty string\n\t\t},\n\t\t{\n\t\t\tname: \"FailoverMarkerTask with empty DomainID\",\n\t\t\ttask: &FailoverMarkerTask{\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tDomainID: \"\",\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"FailoverMarkerTask with valid DomainID\",\n\t\t\ttask: &FailoverMarkerTask{\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tDomainID: \"test-domain\",\n\t\t\t},\n\t\t\texpected: false, // FailoverMarkerTask only has DomainID, WorkflowID and RunID are empty by design\n\t\t},\n\t\t{\n\t\t\tname: \"Task with mixed empty and non-empty identifiers\",\n\t\t\ttask: &CancelExecutionTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tTargetDomainID:          \"target-domain\",\n\t\t\t\tTargetWorkflowID:        \"target-workflow\",\n\t\t\t\tTargetRunID:             \"target-run\",\n\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\tInitiatedID:             123,\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with very long identifiers\",\n\t\t\ttask: &SignalExecutionTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"very-long-domain-id-that-exceeds-normal-length\",\n\t\t\t\t\tWorkflowID: \"very-long-workflow-id-that-exceeds-normal-length\",\n\t\t\t\t\tRunID:      \"very-long-run-id-that-exceeds-normal-length\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tTargetDomainID:          \"target-domain\",\n\t\t\t\tTargetWorkflowID:        \"target-workflow\",\n\t\t\t\tTargetRunID:             \"target-run\",\n\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\tInitiatedID:             123,\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with special characters in identifiers\",\n\t\t\ttask: &StartChildExecutionTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"domain-with-special-chars-!@#$%^&*()\",\n\t\t\t\t\tWorkflowID: \"workflow-with-special-chars-!@#$%^&*()\",\n\t\t\t\t\tRunID:      \"run-with-special-chars-!@#$%^&*()\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t\tTargetDomainID:   \"target-domain\",\n\t\t\t\tTargetWorkflowID: \"target-workflow\",\n\t\t\t\tInitiatedID:      123,\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Task with unicode characters in identifiers\",\n\t\t\ttask: &RecordWorkflowStartedTask{\n\t\t\t\tWorkflowIdentifier: WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"domain-with-unicode-测试\",\n\t\t\t\t\tWorkflowID: \"workflow-with-unicode-测试\",\n\t\t\t\t\tRunID:      \"run-with-unicode-测试\",\n\t\t\t\t},\n\t\t\t\tTaskData: TaskData{\n\t\t\t\t\tVersion:             1,\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeNow,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := IsTaskCorrupted(tt.task)\n\t\t\tassert.Equal(t, tt.expected, result, \"IsTaskCorrupted() = %v, want %v\", result, tt.expected)\n\t\t})\n\t}\n}\n\nfunc TestIsTaskCorruptedWithAllTaskTypes(t *testing.T) {\n\ttimeNow := time.Now()\n\tvalidIdentifier := WorkflowIdentifier{\n\t\tDomainID:   \"test-domain\",\n\t\tWorkflowID: \"test-workflow\",\n\t\tRunID:      \"test-run\",\n\t}\n\temptyIdentifier := WorkflowIdentifier{\n\t\tDomainID:   \"\",\n\t\tWorkflowID: \"\",\n\t\tRunID:      \"\",\n\t}\n\n\t// Test all task types with valid identifiers\n\tvalidTasks := []Task{\n\t\t&ActivityTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowStartedTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ResetWorkflowTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CloseExecutionTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DeleteHistoryEventTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTimeoutTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowTimeoutTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CancelExecutionTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SignalExecutionTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordChildExecutionCompletedTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UpsertWorkflowSearchAttributesTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&StartChildExecutionTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowClosedTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityTimeoutTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UserTimerTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityRetryTimerTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowBackoffTimerTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&HistoryReplicationTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SyncActivityTask{WorkflowIdentifier: validIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&FailoverMarkerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}, DomainID: \"test-domain\"},\n\t}\n\n\t// Test all task types with empty identifiers\n\tcorruptedTasks := []Task{\n\t\t&ActivityTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowStartedTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ResetWorkflowTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CloseExecutionTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DeleteHistoryEventTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&DecisionTimeoutTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowTimeoutTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&CancelExecutionTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SignalExecutionTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordChildExecutionCompletedTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UpsertWorkflowSearchAttributesTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&StartChildExecutionTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&RecordWorkflowClosedTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityTimeoutTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&UserTimerTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&ActivityRetryTimerTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&WorkflowBackoffTimerTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&HistoryReplicationTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&SyncActivityTask{WorkflowIdentifier: emptyIdentifier, TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}},\n\t\t&FailoverMarkerTask{TaskData: TaskData{Version: 1, TaskID: 1, VisibilityTimestamp: timeNow}, DomainID: \"\"},\n\t}\n\n\tt.Run(\"All task types with valid identifiers should not be corrupted\", func(t *testing.T) {\n\t\tfor i, task := range validTasks {\n\t\t\tresult := IsTaskCorrupted(task)\n\t\t\tassert.False(t, result, \"Task type %T at index %d should not be corrupted\", task, i)\n\t\t}\n\t})\n\n\tt.Run(\"All task types with empty identifiers should be corrupted\", func(t *testing.T) {\n\t\tfor i, task := range corruptedTasks {\n\t\t\tresult := IsTaskCorrupted(task)\n\t\t\tassert.True(t, result, \"Task type %T at index %d should be corrupted\", task, i)\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "common/persistence/versionHistory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"bytes\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// NewVersionHistoryItem create a new version history item\nfunc NewVersionHistoryItem(\n\tinputEventID int64,\n\tinputVersion int64,\n) *VersionHistoryItem {\n\n\tif inputEventID < 0 || (inputVersion < 0 && inputVersion != constants.EmptyVersion) {\n\t\tpanic(fmt.Sprintf(\n\t\t\t\"invalid version history item event ID: %v, version: %v\",\n\t\t\tinputEventID,\n\t\t\tinputVersion,\n\t\t))\n\t}\n\n\treturn &VersionHistoryItem{EventID: inputEventID, Version: inputVersion}\n}\n\n// NewVersionHistoryItemFromInternalType create a new version history item from internal type object\nfunc NewVersionHistoryItemFromInternalType(\n\tinput *types.VersionHistoryItem,\n) *VersionHistoryItem {\n\tif input == nil {\n\t\treturn nil\n\t}\n\treturn NewVersionHistoryItem(input.EventID, input.Version)\n}\n\n// Duplicate duplicate VersionHistoryItem\nfunc (item *VersionHistoryItem) Duplicate() *VersionHistoryItem {\n\n\treturn NewVersionHistoryItem(item.EventID, item.Version)\n}\n\n// ToInternalType return internal format of version history item\nfunc (item *VersionHistoryItem) ToInternalType() *types.VersionHistoryItem {\n\tif item == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistoryItem{\n\t\tEventID: item.EventID,\n\t\tVersion: item.Version,\n\t}\n}\n\n// Equals test if this version history item and input version history item are the same\nfunc (item *VersionHistoryItem) Equals(input *VersionHistoryItem) bool {\n\treturn item.Version == input.Version && item.EventID == input.EventID\n}\n\n// NewVersionHistory create a new version history\nfunc NewVersionHistory(\n\tinputToken []byte,\n\tinputItems []*VersionHistoryItem,\n) *VersionHistory {\n\n\ttoken := make([]byte, len(inputToken))\n\tcopy(token, inputToken)\n\tversionHistory := &VersionHistory{\n\t\tBranchToken: token,\n\t\tItems:       nil,\n\t}\n\n\tfor _, item := range inputItems {\n\t\tif err := versionHistory.AddOrUpdateItem(item.Duplicate()); err != nil {\n\t\t\tpanic(fmt.Sprintf(\"unable to initialize version history: %v\", err))\n\t\t}\n\t}\n\n\treturn versionHistory\n}\n\n// NewVersionHistoryFromInternalType create a new version history from internal type object\nfunc NewVersionHistoryFromInternalType(\n\tinput *types.VersionHistory,\n) *VersionHistory {\n\n\tif input == nil {\n\t\treturn nil\n\t}\n\n\titems := make([]*VersionHistoryItem, 0, len(input.Items))\n\tfor _, item := range input.Items {\n\t\titems = append(items, NewVersionHistoryItemFromInternalType(item))\n\t}\n\treturn NewVersionHistory(input.BranchToken, items)\n}\n\n// Duplicate duplicate VersionHistory\nfunc (v *VersionHistory) Duplicate() *VersionHistory {\n\n\treturn NewVersionHistory(v.BranchToken, v.Items)\n}\n\n// ToInternalType return internal format of version history\nfunc (v *VersionHistory) ToInternalType() *types.VersionHistory {\n\tif v == nil {\n\t\treturn nil\n\t}\n\n\ttoken := make([]byte, len(v.BranchToken))\n\tcopy(token, v.BranchToken)\n\titems := []*types.VersionHistoryItem{}\n\tfor _, item := range v.Items {\n\t\titems = append(items, item.ToInternalType())\n\t}\n\n\ttHistory := &types.VersionHistory{\n\t\tBranchToken: token,\n\t\tItems:       items,\n\t}\n\treturn tHistory\n}\n\n// DuplicateUntilLCAItem duplicate the version history up until LCA item\nfunc (v *VersionHistory) DuplicateUntilLCAItem(\n\tlcaItem *VersionHistoryItem,\n) (*VersionHistory, error) {\n\n\tversionHistory := NewVersionHistory(nil, nil)\n\tnotFoundErr := &types.BadRequestError{\n\t\tMessage: \"version history does not contains the LCA item.\",\n\t}\n\tfor _, item := range v.Items {\n\n\t\tif item.Version < lcaItem.Version {\n\t\t\tif err := versionHistory.AddOrUpdateItem(item); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t} else if item.Version == lcaItem.Version {\n\t\t\tif lcaItem.EventID > item.EventID {\n\t\t\t\treturn nil, notFoundErr\n\t\t\t}\n\t\t\tif err := versionHistory.AddOrUpdateItem(lcaItem); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treturn versionHistory, nil\n\n\t\t} else {\n\t\t\treturn nil, notFoundErr\n\t\t}\n\t}\n\n\treturn nil, notFoundErr\n}\n\n// SetBranchToken the overwrite the branch token\nfunc (v *VersionHistory) SetBranchToken(\n\tinputToken []byte,\n) error {\n\n\ttoken := make([]byte, len(inputToken))\n\tcopy(token, inputToken)\n\tv.BranchToken = token\n\treturn nil\n}\n\n// GetBranchToken return the branch token\nfunc (v *VersionHistory) GetBranchToken() []byte {\n\ttoken := make([]byte, len(v.BranchToken))\n\tcopy(token, v.BranchToken)\n\treturn token\n}\n\n// AddOrUpdateItem updates the versionHistory slice\nfunc (v *VersionHistory) AddOrUpdateItem(\n\titem *VersionHistoryItem,\n) error {\n\n\tif len(v.Items) == 0 {\n\t\tv.Items = []*VersionHistoryItem{item.Duplicate()}\n\t\treturn nil\n\t}\n\n\tlastItem := v.Items[len(v.Items)-1]\n\tif item.Version < lastItem.Version {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\"cannot update version history with a lower version %v. Last version: %v\",\n\t\t\titem.Version, lastItem.Version,\n\t\t)}\n\t}\n\n\tif item.EventID <= lastItem.EventID {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\"cannot add version history with a lower event id %v. Last event id: %v\",\n\t\t\titem.EventID, lastItem.EventID,\n\t\t)}\n\t}\n\n\tif item.Version > lastItem.Version {\n\t\t// Add a new history\n\t\tv.Items = append(v.Items, item.Duplicate())\n\t} else {\n\t\t// item.Version == lastItem.Version && item.EventID > lastItem.EventID\n\t\t// Update event ID\n\t\tlastItem.EventID = item.EventID\n\t}\n\treturn nil\n}\n\n// ContainsItem check whether given version history item is included\nfunc (v *VersionHistory) ContainsItem(\n\titem *VersionHistoryItem,\n) bool {\n\tprevEventID := constants.FirstEventID - 1\n\tfor _, currentItem := range v.Items {\n\t\tif item.Version == currentItem.Version {\n\t\t\tif prevEventID < item.EventID && item.EventID <= currentItem.EventID {\n\t\t\t\treturn true\n\t\t\t}\n\t\t} else if item.Version < currentItem.Version {\n\t\t\treturn false\n\t\t}\n\t\tprevEventID = currentItem.EventID\n\t}\n\treturn false\n}\n\n// FindLCAItem returns the lowest common ancestor version history item\nfunc (v *VersionHistory) FindLCAItem(\n\tremote *VersionHistory,\n) (*VersionHistoryItem, error) {\n\n\tlocalIndex := len(v.Items) - 1\n\tremoteIndex := len(remote.Items) - 1\n\n\tfor localIndex >= 0 && remoteIndex >= 0 {\n\t\tlocalVersionItem := v.Items[localIndex]\n\t\tremoteVersionItem := remote.Items[remoteIndex]\n\n\t\tif localVersionItem.Version == remoteVersionItem.Version {\n\t\t\tif localVersionItem.EventID > remoteVersionItem.EventID {\n\t\t\t\treturn remoteVersionItem.Duplicate(), nil\n\t\t\t}\n\t\t\treturn localVersionItem.Duplicate(), nil\n\t\t} else if localVersionItem.Version > remoteVersionItem.Version {\n\t\t\tlocalIndex--\n\t\t} else {\n\t\t\t// localVersionItem.Version < remoteVersionItem.Version\n\t\t\tremoteIndex--\n\t\t}\n\t}\n\n\treturn nil, &types.BadRequestError{\n\t\tMessage: \"version history is malformed. No joint point found.\",\n\t}\n}\n\n// IsLCAAppendable checks if a LCA version history item is appendable\nfunc (v *VersionHistory) IsLCAAppendable(\n\titem *VersionHistoryItem,\n) bool {\n\n\tif len(v.Items) == 0 {\n\t\tpanic(\"version history not initialized\")\n\t}\n\tif item == nil {\n\t\tpanic(\"version history item is null\")\n\t}\n\n\treturn *v.Items[len(v.Items)-1] == *item\n}\n\n// GetFirstItem return the first version history item\nfunc (v *VersionHistory) GetFirstItem() (*VersionHistoryItem, error) {\n\n\tif len(v.Items) == 0 {\n\t\treturn nil, &types.BadRequestError{Message: \"version history is empty.\"}\n\t}\n\n\treturn v.Items[0].Duplicate(), nil\n}\n\n// GetLastItem return the last version history item\nfunc (v *VersionHistory) GetLastItem() (*VersionHistoryItem, error) {\n\tif len(v.Items) == 0 {\n\t\treturn nil, &types.BadRequestError{Message: \"version history is empty\"}\n\t}\n\treturn v.Items[len(v.Items)-1].Duplicate(), nil\n}\n\n// GetEventVersion return the corresponding event version of an event ID\nfunc (v *VersionHistory) GetEventVersion(\n\teventID int64,\n) (int64, error) {\n\n\tlastItem, err := v.GetLastItem()\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tif eventID < constants.FirstEventID || eventID > lastItem.EventID {\n\t\treturn 0, &types.BadRequestError{Message: \"input event ID is not in range.\"}\n\t}\n\n\t// items are sorted by eventID & version\n\t// so the fist item with item event ID >= input event ID\n\t// the item version is the result\n\tfor _, currentItem := range v.Items {\n\t\tif eventID <= currentItem.EventID {\n\t\t\treturn currentItem.Version, nil\n\t\t}\n\t}\n\treturn 0, &types.BadRequestError{Message: \"input event ID is not in range.\"}\n}\n\n// IsEmpty indicate whether version history is empty\nfunc (v *VersionHistory) IsEmpty() bool {\n\treturn len(v.Items) == 0\n}\n\n// Equals test if this version history and input version history are the same\nfunc (v *VersionHistory) Equals(\n\tinput *VersionHistory,\n) bool {\n\n\tif !bytes.Equal(v.BranchToken, input.BranchToken) {\n\t\treturn false\n\t}\n\n\tif len(v.Items) != len(input.Items) {\n\t\treturn false\n\t}\n\n\tfor index, localItem := range v.Items {\n\t\tincomingItem := input.Items[index]\n\t\tif !localItem.Equals(incomingItem) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\n// NewVersionHistories create a new version histories\nfunc NewVersionHistories(\n\tversionHistory *VersionHistory,\n) *VersionHistories {\n\tif versionHistory == nil {\n\t\treturn nil\n\t}\n\treturn &VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories:                  []*VersionHistory{versionHistory},\n\t}\n}\n\n// NewVersionHistoriesFromInternalType create a new version histories from internal type object\nfunc NewVersionHistoriesFromInternalType(\n\tinput *types.VersionHistories,\n) *VersionHistories {\n\tif input == nil {\n\t\treturn nil\n\t}\n\tif len(input.Histories) == 0 {\n\t\tpanic(\"version histories cannot have empty\")\n\t}\n\n\tcurrentVersionHistoryIndex := int(input.GetCurrentVersionHistoryIndex())\n\n\tversionHistories := NewVersionHistories(NewVersionHistoryFromInternalType(input.Histories[0]))\n\tfor i := 1; i < len(input.Histories); i++ {\n\t\t_, _, err := versionHistories.AddVersionHistory(NewVersionHistoryFromInternalType(input.Histories[i]))\n\t\tif err != nil {\n\t\t\tpanic(fmt.Sprintf(\"unable to initialize version histories: %v\", err))\n\t\t}\n\t}\n\n\tif currentVersionHistoryIndex != versionHistories.CurrentVersionHistoryIndex {\n\t\tpanic(\"unable to initialize version histories: current index mismatch\")\n\t}\n\n\treturn versionHistories\n}\n\n// Duplicate duplicate VersionHistories\nfunc (h *VersionHistories) Duplicate() *VersionHistories {\n\tif h == nil {\n\t\treturn nil\n\t}\n\n\tcurrentVersionHistoryIndex := h.CurrentVersionHistoryIndex\n\thistories := []*VersionHistory{}\n\tfor _, history := range h.Histories {\n\t\thistories = append(histories, history.Duplicate())\n\t}\n\n\treturn &VersionHistories{\n\t\tCurrentVersionHistoryIndex: currentVersionHistoryIndex,\n\t\tHistories:                  histories,\n\t}\n}\n\n// ToInternalType return internal format of version histories\nfunc (h *VersionHistories) ToInternalType() *types.VersionHistories {\n\n\tcurrentVersionHistoryIndex := h.CurrentVersionHistoryIndex\n\thistories := []*types.VersionHistory{}\n\tfor _, history := range h.Histories {\n\t\thistories = append(histories, history.ToInternalType())\n\t}\n\n\treturn &types.VersionHistories{\n\t\tCurrentVersionHistoryIndex: int32(currentVersionHistoryIndex),\n\t\tHistories:                  histories,\n\t}\n}\n\n// GetVersionHistory get the version history according to index provided\nfunc (h *VersionHistories) GetVersionHistory(\n\tbranchIndex int,\n) (*VersionHistory, error) {\n\n\tif branchIndex < 0 || branchIndex >= len(h.Histories) {\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"getting branch index: %d, available branch count: %d\", branchIndex, len(h.Histories))}\n\t}\n\n\treturn h.Histories[branchIndex], nil\n}\n\n// AddVersionHistory add a version history and return the whether current branch is changed\nfunc (h *VersionHistories) AddVersionHistory(\n\tv *VersionHistory,\n) (bool, int, error) {\n\n\tif v == nil {\n\t\treturn false, 0, &types.BadRequestError{Message: \"version histories is null.\"}\n\t}\n\n\t// assuming existing version histories inside are valid\n\tincomingFirstItem, err := v.GetFirstItem()\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\tcurrentVersionHistory, err := h.GetVersionHistory(h.CurrentVersionHistoryIndex)\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\tcurrentFirstItem, err := currentVersionHistory.GetFirstItem()\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\tif incomingFirstItem.Version != currentFirstItem.Version {\n\t\treturn false, 0, &types.BadRequestError{Message: \"version history first item does not match.\"}\n\t}\n\n\t// TODO maybe we need more strict validation\n\n\tnewVersionHistory := v.Duplicate()\n\th.Histories = append(h.Histories, newVersionHistory)\n\tnewVersionHistoryIndex := len(h.Histories) - 1\n\n\t// check if need to switch current branch\n\tnewLastItem, err := newVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\tcurrentLastItem, err := currentVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\tcurrentBranchChanged := false\n\tif newLastItem.Version > currentLastItem.Version {\n\t\tcurrentBranchChanged = true\n\t\th.CurrentVersionHistoryIndex = newVersionHistoryIndex\n\t}\n\treturn currentBranchChanged, newVersionHistoryIndex, nil\n}\n\n// FindLCAVersionHistoryIndexAndItem finds the lowest common ancestor version history index\n// along with corresponding item\nfunc (h *VersionHistories) FindLCAVersionHistoryIndexAndItem(\n\tincomingHistory *VersionHistory,\n) (int, *VersionHistoryItem, error) {\n\n\tvar versionHistoryIndex int\n\tvar versionHistoryLength int\n\tvar versionHistoryItem *VersionHistoryItem\n\n\tfor index, localHistory := range h.Histories {\n\t\titem, err := localHistory.FindLCAItem(incomingHistory)\n\t\tif err != nil {\n\t\t\treturn 0, nil, err\n\t\t}\n\n\t\t// if not set\n\t\tif versionHistoryItem == nil ||\n\t\t\t// if seeing LCA item with higher event ID\n\t\t\titem.EventID > versionHistoryItem.EventID ||\n\t\t\t// if seeing LCA item with equal event ID but shorter history\n\t\t\t(item.EventID == versionHistoryItem.EventID && len(localHistory.Items) < versionHistoryLength) {\n\n\t\t\tversionHistoryIndex = index\n\t\t\tversionHistoryLength = len(localHistory.Items)\n\t\t\tversionHistoryItem = item\n\t\t}\n\t}\n\treturn versionHistoryIndex, versionHistoryItem, nil\n}\n\n// FindFirstVersionHistoryByItem find the first version history index and history which\n// contains the given version history item\nfunc (h *VersionHistories) FindFirstVersionHistoryByItem(\n\titem *VersionHistoryItem,\n) (index int, history *VersionHistory, err error) {\n\n\tfor index, localHistory := range h.Histories {\n\t\tif localHistory.ContainsItem(item) {\n\t\t\treturn index, localHistory, nil\n\t\t}\n\t}\n\treturn 0, nil, &types.BadRequestError{Message: \"version histories does not contains given item.\"}\n}\n\n// IsRebuilt returns true if the current branch index's last write version is not the largest\n// among all branches' last write version\nfunc (h *VersionHistories) IsRebuilt() (bool, error) {\n\n\tcurrentVersionHistory, err := h.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tcurrentLastItem, err := currentVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tfor _, versionHistory := range h.Histories {\n\t\tlastItem, err := versionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn false, err\n\t\t}\n\t\tif lastItem.Version > currentLastItem.Version {\n\t\t\treturn true, nil\n\t\t}\n\t}\n\n\treturn false, nil\n}\n\n// SetCurrentVersionHistoryIndex set the current branch index\nfunc (h *VersionHistories) SetCurrentVersionHistoryIndex(\n\tindex int,\n) error {\n\n\tif index < 0 || index >= len(h.Histories) {\n\t\treturn &types.BadRequestError{Message: \"invalid current branch index.\"}\n\t}\n\n\th.CurrentVersionHistoryIndex = index\n\treturn nil\n}\n\n// GetCurrentVersionHistoryIndex get the current branch index\nfunc (h *VersionHistories) GetCurrentVersionHistoryIndex() int {\n\treturn h.CurrentVersionHistoryIndex\n}\n\n// GetCurrentVersionHistory get the current version history\nfunc (h *VersionHistories) GetCurrentVersionHistory() (*VersionHistory, error) {\n\treturn h.GetVersionHistory(h.GetCurrentVersionHistoryIndex())\n}\n"
  },
  {
    "path": "common/persistence/versionHistory_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tversionHistorySuite struct {\n\t\tsuite.Suite\n\t}\n\n\tversionHistoriesSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestVersionHistorySuite(t *testing.T) {\n\ts := new(versionHistorySuite)\n\tsuite.Run(t, s)\n}\n\nfunc TestVersionHistoriesSuite(t *testing.T) {\n\ts := new(versionHistoriesSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *versionHistorySuite) TestConversion() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\n\thistory := NewVersionHistory(BranchToken, Items)\n\ts.Equal(&VersionHistory{\n\t\tBranchToken: BranchToken,\n\t\tItems:       Items,\n\t}, history)\n\n\ts.Equal(history, NewVersionHistoryFromInternalType(history.ToInternalType()))\n}\n\nfunc (s *versionHistorySuite) TestDuplicateUntilLCAItem_Success() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\tnewHistory, err := history.DuplicateUntilLCAItem(NewVersionHistoryItem(2, 0))\n\ts.NoError(err)\n\tnewBranchToken := []byte(\"other random branch token\")\n\terr = newHistory.SetBranchToken(newBranchToken)\n\ts.NoError(err)\n\ts.Equal(newBranchToken, newHistory.GetBranchToken())\n\ts.Equal(NewVersionHistory(\n\t\tnewBranchToken,\n\t\t[]*VersionHistoryItem{{EventID: 2, Version: 0}},\n\t), newHistory)\n\n\tnewHistory, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(5, 4))\n\ts.NoError(err)\n\tnewBranchToken = []byte(\"another random branch token\")\n\terr = newHistory.SetBranchToken(newBranchToken)\n\ts.NoError(err)\n\ts.Equal(newBranchToken, newHistory.GetBranchToken())\n\ts.Equal(NewVersionHistory(\n\t\tnewBranchToken,\n\t\t[]*VersionHistoryItem{\n\t\t\t{EventID: 3, Version: 0},\n\t\t\t{EventID: 5, Version: 4},\n\t\t},\n\t), newHistory)\n\n\tnewHistory, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(6, 4))\n\ts.NoError(err)\n\tnewBranchToken = []byte(\"yet another random branch token\")\n\terr = newHistory.SetBranchToken(newBranchToken)\n\ts.NoError(err)\n\ts.Equal(newBranchToken, newHistory.GetBranchToken())\n\ts.Equal(NewVersionHistory(\n\t\tnewBranchToken,\n\t\t[]*VersionHistoryItem{\n\t\t\t{EventID: 3, Version: 0},\n\t\t\t{EventID: 6, Version: 4},\n\t\t},\n\t), newHistory)\n}\n\nfunc (s *versionHistorySuite) TestDuplicateUntilLCAItem_Failure() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\t_, err := history.DuplicateUntilLCAItem(NewVersionHistoryItem(4, 0))\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t_, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(2, 1))\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t_, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(5, 3))\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t_, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(7, 5))\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t_, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(4, 0))\n\ts.IsType(&types.BadRequestError{}, err)\n\n\t_, err = history.DuplicateUntilLCAItem(NewVersionHistoryItem(7, 4))\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *versionHistorySuite) TestSetBranchToken() {\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(nil, Items)\n\n\terr := history.SetBranchToken([]byte(\"some random branch token\"))\n\ts.NoError(err)\n}\n\nfunc (s *versionHistorySuite) TestAddOrUpdateItem_VersionIncrease() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\titem := &VersionHistoryItem{\n\t\tEventID: 8,\n\t\tVersion: 5,\n\t}\n\terr := history.AddOrUpdateItem(item)\n\ts.NoError(err)\n\n\ts.Equal(NewVersionHistory(\n\t\tBranchToken,\n\t\t[]*VersionHistoryItem{\n\t\t\t{EventID: 3, Version: 0},\n\t\t\t{EventID: 6, Version: 4},\n\t\t\t{EventID: 8, Version: 5},\n\t\t},\n\t), history)\n\n}\n\nfunc (s *versionHistorySuite) TestAddOrUpdateItem_EventIDIncrease() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\titem := &VersionHistoryItem{\n\t\tEventID: 8,\n\t\tVersion: 4,\n\t}\n\terr := history.AddOrUpdateItem(item)\n\ts.NoError(err)\n\n\ts.Equal(NewVersionHistory(\n\t\tBranchToken,\n\t\t[]*VersionHistoryItem{\n\t\t\t{EventID: 3, Version: 0},\n\t\t\t{EventID: 8, Version: 4},\n\t\t},\n\t), history)\n}\n\nfunc (s *versionHistorySuite) TestAddOrUpdateItem_Failed_LowerVersion() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\terr := history.AddOrUpdateItem(NewVersionHistoryItem(8, 3))\n\ts.Error(err)\n}\n\nfunc (s *versionHistorySuite) TestAddOrUpdateItem_Failed_SameVersion_EventIDNotIncreasing() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\terr := history.AddOrUpdateItem(NewVersionHistoryItem(5, 4))\n\ts.Error(err)\n\n\terr = history.AddOrUpdateItem(NewVersionHistoryItem(6, 4))\n\ts.Error(err)\n}\n\nfunc (s *versionHistorySuite) TestAddOrUpdateItem_Failed_VersionNoIncreasing() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\terr := history.AddOrUpdateItem(NewVersionHistoryItem(6, 3))\n\ts.Error(err)\n\n\terr = history.AddOrUpdateItem(NewVersionHistoryItem(2, 3))\n\ts.Error(err)\n\n\terr = history.AddOrUpdateItem(NewVersionHistoryItem(7, 3))\n\ts.Error(err)\n}\n\nfunc (s *versionHistoriesSuite) TestContainsItem_True() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\tprevEventID := constants.FirstEventID - 1\n\tfor _, item := range Items {\n\t\tfor EventID := prevEventID + 1; EventID <= item.EventID; EventID++ {\n\t\t\ts.True(history.ContainsItem(NewVersionHistoryItem(EventID, item.Version)))\n\t\t}\n\t\tprevEventID = item.EventID\n\t}\n}\n\nfunc (s *versionHistoriesSuite) TestContainsItem_False() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\ts.False(history.ContainsItem(NewVersionHistoryItem(4, 0)))\n\ts.False(history.ContainsItem(NewVersionHistoryItem(3, 1)))\n\n\ts.False(history.ContainsItem(NewVersionHistoryItem(7, 4)))\n\ts.False(history.ContainsItem(NewVersionHistoryItem(6, 5)))\n}\n\nfunc (s *versionHistorySuite) TestIsLCAAppendable_True() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\tret := history.IsLCAAppendable(NewVersionHistoryItem(6, 4))\n\ts.True(ret)\n}\n\nfunc (s *versionHistorySuite) TestIsLCAAppendable_False_VersionNotMatch() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\tret := history.IsLCAAppendable(NewVersionHistoryItem(6, 7))\n\ts.False(ret)\n}\n\nfunc (s *versionHistorySuite) TestIsLCAAppendable_False_EventIDNotMatch() {\n\tBranchToken := []byte(\"some random branch token\")\n\tItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 6, Version: 4},\n\t}\n\thistory := NewVersionHistory(BranchToken, Items)\n\n\tret := history.IsLCAAppendable(NewVersionHistoryItem(7, 4))\n\ts.False(ret)\n}\n\nfunc (s *versionHistorySuite) TestFindLCAItem_ReturnLocal() {\n\tlocalBranchToken := []byte(\"local branch token\")\n\tlocalItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t}\n\tremoteBranchToken := []byte(\"remote branch token\")\n\tremoteItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 7, Version: 4},\n\t\t{EventID: 8, Version: 8},\n\t\t{EventID: 11, Version: 12},\n\t}\n\tlocalVersionHistory := NewVersionHistory(localBranchToken, localItems)\n\tremoteVersionHistory := NewVersionHistory(remoteBranchToken, remoteItems)\n\n\titem, err := localVersionHistory.FindLCAItem(remoteVersionHistory)\n\ts.NoError(err)\n\ts.Equal(NewVersionHistoryItem(5, 4), item)\n}\n\nfunc (s *versionHistorySuite) TestFindLCAItem_ReturnRemote() {\n\tlocalBranchToken := []byte(\"local branch token\")\n\tlocalItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t}\n\tremoteBranchToken := []byte(\"remote branch token\")\n\tremoteItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 6, Version: 6},\n\t\t{EventID: 11, Version: 12},\n\t}\n\tlocalVersionHistory := NewVersionHistory(localBranchToken, localItems)\n\tremoteVersionHistory := NewVersionHistory(remoteBranchToken, remoteItems)\n\n\titem, err := localVersionHistory.FindLCAItem(remoteVersionHistory)\n\ts.NoError(err)\n\ts.Equal(NewVersionHistoryItem(6, 6), item)\n}\n\nfunc (s *versionHistorySuite) TestFindLCAItem_Error_NoLCA() {\n\tlocalBranchToken := []byte(\"local branch token\")\n\tlocalItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t}\n\tremoteBranchToken := []byte(\"remote branch token\")\n\tremoteItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 1},\n\t\t{EventID: 7, Version: 2},\n\t\t{EventID: 8, Version: 3},\n\t}\n\tlocalVersionHistory := NewVersionHistory(localBranchToken, localItems)\n\tremoteVersionHistory := NewVersionHistory(remoteBranchToken, remoteItems)\n\n\t_, err := localVersionHistory.FindLCAItem(remoteVersionHistory)\n\ts.Error(err)\n}\n\nfunc (s *versionHistorySuite) TestGetFirstItem_Success() {\n\tBranchToken := []byte(\"some random branch token\")\n\titem := NewVersionHistoryItem(3, 0)\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{item})\n\n\tfirstItem, err := history.GetFirstItem()\n\ts.NoError(err)\n\ts.Equal(item, firstItem)\n\n\titem = NewVersionHistoryItem(4, 0)\n\terr = history.AddOrUpdateItem(item)\n\ts.NoError(err)\n\n\tfirstItem, err = history.GetFirstItem()\n\ts.NoError(err)\n\ts.Equal(item, firstItem)\n\n\terr = history.AddOrUpdateItem(NewVersionHistoryItem(7, 1))\n\ts.NoError(err)\n\n\tfirstItem, err = history.GetFirstItem()\n\ts.NoError(err)\n\ts.Equal(item, firstItem)\n}\n\nfunc (s *versionHistorySuite) TestGetFirstItem_Failure() {\n\tBranchToken := []byte(\"some random branch token\")\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{})\n\n\t_, err := history.GetFirstItem()\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *versionHistorySuite) TestGetLastItem_Success() {\n\tBranchToken := []byte(\"some random branch token\")\n\titem := NewVersionHistoryItem(3, 0)\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{item})\n\n\tlastItem, err := history.GetLastItem()\n\ts.NoError(err)\n\ts.Equal(item, lastItem)\n\n\titem = NewVersionHistoryItem(4, 0)\n\terr = history.AddOrUpdateItem(item)\n\ts.NoError(err)\n\n\tlastItem, err = history.GetLastItem()\n\ts.NoError(err)\n\ts.Equal(item, lastItem)\n\n\titem = NewVersionHistoryItem(7, 1)\n\terr = history.AddOrUpdateItem(item)\n\ts.NoError(err)\n\n\tlastItem, err = history.GetLastItem()\n\ts.NoError(err)\n\ts.Equal(item, lastItem)\n}\n\nfunc (s *versionHistorySuite) TestGetLastItem_Failure() {\n\tBranchToken := []byte(\"some random branch token\")\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{})\n\n\t_, err := history.GetLastItem()\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *versionHistoriesSuite) TestGetVersion_Success() {\n\tBranchToken := []byte(\"some random branch token\")\n\titem1 := NewVersionHistoryItem(3, 0)\n\titem2 := NewVersionHistoryItem(6, 8)\n\titem3 := NewVersionHistoryItem(8, 12)\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{item1, item2, item3})\n\n\tVersion, err := history.GetEventVersion(1)\n\ts.NoError(err)\n\ts.Equal(item1.Version, Version)\n\tVersion, err = history.GetEventVersion(2)\n\ts.NoError(err)\n\ts.Equal(item1.Version, Version)\n\tVersion, err = history.GetEventVersion(3)\n\ts.NoError(err)\n\ts.Equal(item1.Version, Version)\n\n\tVersion, err = history.GetEventVersion(4)\n\ts.NoError(err)\n\ts.Equal(item2.Version, Version)\n\tVersion, err = history.GetEventVersion(5)\n\ts.NoError(err)\n\ts.Equal(item2.Version, Version)\n\tVersion, err = history.GetEventVersion(6)\n\ts.NoError(err)\n\ts.Equal(item2.Version, Version)\n\n\tVersion, err = history.GetEventVersion(7)\n\ts.NoError(err)\n\ts.Equal(item3.Version, Version)\n\tVersion, err = history.GetEventVersion(8)\n\ts.NoError(err)\n\ts.Equal(item3.Version, Version)\n}\n\nfunc (s *versionHistoriesSuite) TestGetVersion_Failure() {\n\tBranchToken := []byte(\"some random branch token\")\n\titem1 := NewVersionHistoryItem(3, 0)\n\titem2 := NewVersionHistoryItem(6, 8)\n\titem3 := NewVersionHistoryItem(8, 12)\n\thistory := NewVersionHistory(BranchToken, []*VersionHistoryItem{item1, item2, item3})\n\n\t_, err := history.GetEventVersion(0)\n\ts.Error(err)\n\n\t_, err = history.GetEventVersion(9)\n\ts.Error(err)\n}\n\nfunc (s *versionHistorySuite) TestEquals() {\n\tlocalBranchToken := []byte(\"local branch token\")\n\tlocalItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t}\n\tremoteBranchToken := []byte(\"remote branch token\")\n\tremoteItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 1},\n\t\t{EventID: 7, Version: 2},\n\t\t{EventID: 8, Version: 3},\n\t}\n\tlocalVersionHistory := NewVersionHistory(localBranchToken, localItems)\n\tremoteVersionHistory := NewVersionHistory(remoteBranchToken, remoteItems)\n\n\ts.False(localVersionHistory.Equals(remoteVersionHistory))\n\ts.True(localVersionHistory.Equals(localVersionHistory.Duplicate()))\n\ts.True(remoteVersionHistory.Equals(remoteVersionHistory.Duplicate()))\n}\n\nfunc (s *versionHistoriesSuite) TestConversion() {\n\tBranchToken := []byte(\"some random branch token\")\n\tlocalItems := []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t}\n\tversionHistory := NewVersionHistory(BranchToken, localItems)\n\thistories := NewVersionHistories(versionHistory)\n\n\ts.Equal(&VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories:                  []*VersionHistory{versionHistory},\n\t}, histories)\n\n\ts.Equal(histories, NewVersionHistoriesFromInternalType(histories.ToInternalType()))\n}\n\nfunc (s *versionHistoriesSuite) TestAddGetVersionHistory() {\n\tversionHistory1 := NewVersionHistory([]byte(\"branch token 1\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t})\n\tversionHistory2 := NewVersionHistory([]byte(\"branch token 2\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 6, Version: 6},\n\t\t{EventID: 11, Version: 12},\n\t})\n\n\thistories := NewVersionHistories(versionHistory1)\n\ts.Equal(0, histories.CurrentVersionHistoryIndex)\n\n\tcurrentBranchChanged, newVersionHistoryIndex, err := histories.AddVersionHistory(versionHistory2)\n\ts.Nil(err)\n\ts.True(currentBranchChanged)\n\ts.Equal(1, newVersionHistoryIndex)\n\ts.Equal(1, histories.CurrentVersionHistoryIndex)\n\n\tresultVersionHistory1, err := histories.GetVersionHistory(0)\n\ts.Nil(err)\n\ts.Equal(versionHistory1, resultVersionHistory1)\n\n\tresultVersionHistory2, err := histories.GetVersionHistory(1)\n\ts.Nil(err)\n\ts.Equal(versionHistory2, resultVersionHistory2)\n}\n\nfunc (s *versionHistoriesSuite) TestFindLCAVersionHistoryIndexAndItem_LargerEventIDWins() {\n\tversionHistory1 := NewVersionHistory([]byte(\"branch token 1\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t})\n\tversionHistory2 := NewVersionHistory([]byte(\"branch token 2\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 6, Version: 6},\n\t\t{EventID: 11, Version: 12},\n\t})\n\n\thistories := NewVersionHistories(versionHistory1)\n\t_, _, err := histories.AddVersionHistory(versionHistory2)\n\ts.Nil(err)\n\n\tversionHistoryIncoming := NewVersionHistory([]byte(\"branch token incoming\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 8, Version: 6},\n\t\t{EventID: 11, Version: 100},\n\t})\n\n\tindex, item, err := histories.FindLCAVersionHistoryIndexAndItem(versionHistoryIncoming)\n\ts.Nil(err)\n\ts.Equal(0, index)\n\ts.Equal(NewVersionHistoryItem(7, 6), item)\n}\n\nfunc (s *versionHistoriesSuite) TestFindLCAVersionHistoryIndexAndItem_SameEventIDShorterLengthWins() {\n\tversionHistory1 := NewVersionHistory([]byte(\"branch token 1\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t})\n\tversionHistory2 := NewVersionHistory([]byte(\"branch token 2\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t})\n\n\thistories := NewVersionHistories(versionHistory1)\n\t_, _, err := histories.AddVersionHistory(versionHistory2)\n\ts.Nil(err)\n\n\tversionHistoryIncoming := NewVersionHistory([]byte(\"branch token incoming\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 8, Version: 6},\n\t\t{EventID: 11, Version: 100},\n\t})\n\n\tindex, item, err := histories.FindLCAVersionHistoryIndexAndItem(versionHistoryIncoming)\n\ts.Nil(err)\n\ts.Equal(1, index)\n\ts.Equal(NewVersionHistoryItem(7, 6), item)\n}\n\nfunc (s *versionHistoriesSuite) TestFindFirstVersionHistoryByItem() {\n\tversionHistory1 := NewVersionHistory([]byte(\"branch token 1\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t})\n\tversionHistory2 := NewVersionHistory([]byte(\"branch token 2\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t})\n\n\thistories := NewVersionHistories(versionHistory1)\n\t_, _, err := histories.AddVersionHistory(versionHistory2)\n\ts.Nil(err)\n\n\tindex, history, err := histories.FindFirstVersionHistoryByItem(NewVersionHistoryItem(8, 10))\n\ts.NoError(err)\n\ts.Equal(1, index)\n\ts.Equal(versionHistory2, history)\n\n\tindex, history, err = histories.FindFirstVersionHistoryByItem(NewVersionHistoryItem(4, 4))\n\ts.NoError(err)\n\ts.Equal(0, index)\n\ts.Equal(versionHistory1, history)\n\n\t_, _, err = histories.FindFirstVersionHistoryByItem(NewVersionHistoryItem(41, 4))\n\ts.Error(err)\n}\n\nfunc (s *versionHistoriesSuite) TestCurrentVersionHistoryIndexIsInReplay() {\n\tversionHistory1 := NewVersionHistory([]byte(\"branch token 1\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 7, Version: 6},\n\t\t{EventID: 9, Version: 10},\n\t})\n\tversionHistory2 := NewVersionHistory([]byte(\"branch token 2\"), []*VersionHistoryItem{\n\t\t{EventID: 3, Version: 0},\n\t\t{EventID: 5, Version: 4},\n\t\t{EventID: 6, Version: 6},\n\t\t{EventID: 11, Version: 12},\n\t})\n\n\thistories := NewVersionHistories(versionHistory1)\n\ts.Equal(0, histories.CurrentVersionHistoryIndex)\n\n\tcurrentBranchChanged, newVersionHistoryIndex, err := histories.AddVersionHistory(versionHistory2)\n\ts.Nil(err)\n\ts.True(currentBranchChanged)\n\ts.Equal(1, newVersionHistoryIndex)\n\ts.Equal(1, histories.CurrentVersionHistoryIndex)\n\n\tisInReplay, err := histories.IsRebuilt()\n\ts.NoError(err)\n\ts.False(isInReplay)\n\n\terr = histories.SetCurrentVersionHistoryIndex(0)\n\ts.NoError(err)\n\tisInReplay, err = histories.IsRebuilt()\n\ts.NoError(err)\n\ts.True(isInReplay)\n\n\terr = histories.SetCurrentVersionHistoryIndex(1)\n\ts.NoError(err)\n\tisInReplay, err = histories.IsRebuilt()\n\ts.NoError(err)\n\ts.False(isInReplay)\n}\n\nfunc TestNilHandling(t *testing.T) {\n\tassert.Nil(t, NewVersionHistoriesFromInternalType(nil))\n\tassert.Nil(t, NewVersionHistories(nil))\n\tassert.Nil(t, NewVersionHistoryItemFromInternalType(nil))\n\tassert.Nil(t, NewVersionHistoryFromInternalType(nil))\n\tvar vh *VersionHistory\n\tassert.Nil(t, vh.ToInternalType())\n}\n"
  },
  {
    "path": "common/persistence/visibility_hybrid_manager.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tvisibilityHybridManager struct {\n\t\tlogger                    log.Logger\n\t\tvisibilityMgrs            map[string]VisibilityManager\n\t\treadVisibilityStoreName   dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twriteVisibilityStoreName  dynamicproperties.StringPropertyFn\n\t\tlogCustomerQueryParameter dynamicproperties.BoolPropertyFnWithDomainFilter\n\t\tname                      string\n\t}\n)\n\nconst (\n\tContextKey           = ResponseComparatorContextKey(\"visibility-override\")\n\tdbVisStoreName       = \"db\"\n\tadvancedWriteModeOff = \"off\"\n)\n\n// ResponseComparatorContextKey is for Pinot/ES response comparator. This struct will be passed into ctx as a key.\ntype ResponseComparatorContextKey string\n\ntype OperationType string\n\nvar Operation = struct {\n\tLIST  OperationType\n\tCOUNT OperationType\n}{\n\tLIST:  \"list\",\n\tCOUNT: \"count\",\n}\n\nvar _ VisibilityManager = (*visibilityHybridManager)(nil)\n\n// NewVisibilityTripleManager create a visibility manager that operate on DB or advanced visibility based on dynamic config.\n// For Pinot migration, Pinot is the destination visibility manager, ES is the source visibility manager, and DB is the fallback.\n// For OpenSearch migration, OS is the destination visibility manager, ES is the source visibility manager, and DB is the fallback.\nfunc NewVisibilityHybridManager(\n\tvisibilityMgrs map[string]VisibilityManager,\n\treadVisibilityStoreName dynamicproperties.StringPropertyFnWithDomainFilter,\n\twriteVisibilityStoreName dynamicproperties.StringPropertyFn,\n\tlogCustomerQueryParameter dynamicproperties.BoolPropertyFnWithDomainFilter,\n\tname string,\n\tlogger log.Logger,\n) VisibilityManager {\n\tif len(visibilityMgrs) == 0 {\n\t\tlogger.Fatal(\"No visibility managers provided. At least one visibility manager is required.\")\n\t\treturn nil\n\t}\n\n\tif logCustomerQueryParameter == nil {\n\t\tlogCustomerQueryParameter = dynamicproperties.GetBoolPropertyFnFilteredByDomain(false)\n\t}\n\n\treturn &visibilityHybridManager{\n\t\tvisibilityMgrs:            visibilityMgrs,\n\t\treadVisibilityStoreName:   readVisibilityStoreName,\n\t\twriteVisibilityStoreName:  writeVisibilityStoreName,\n\t\tlogger:                    logger,\n\t\tlogCustomerQueryParameter: logCustomerQueryParameter,\n\t\tname:                      name,\n\t}\n}\n\nfunc (v *visibilityHybridManager) Close() {\n\tfor _, mgr := range v.visibilityMgrs {\n\t\tif mgr != nil {\n\t\t\tmgr.Close()\n\t\t}\n\t}\n}\n\nfunc (v *visibilityHybridManager) GetName() string {\n\treturn v.name\n}\n\nfunc (v *visibilityHybridManager) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionStartedRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.RecordWorkflowExecutionStarted(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionClosedRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.RecordWorkflowExecutionClosed(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionUninitializedRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.RecordWorkflowExecutionUninitialized(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.DeleteWorkflowExecution(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.DeleteUninitializedWorkflowExecution(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *UpsertWorkflowExecutionRequest,\n) error {\n\treturn v.chooseVisibilityManagerForWrite(\n\t\tctx,\n\t\tfunc(storeName string) error {\n\t\t\tmgr, ok := v.visibilityMgrs[storeName]\n\t\t\tif !ok || mgr == nil {\n\t\t\t\treturn fmt.Errorf(\"Visibility store manager with name %s not found\", storeName)\n\t\t\t}\n\t\t\treturn mgr.UpsertWorkflowExecution(ctx, request)\n\t\t},\n\t)\n}\n\nfunc (v *visibilityHybridManager) chooseVisibilityModeForAdmin() string {\n\tvar modes []string\n\n\tfor storeName, mgr := range v.visibilityMgrs {\n\t\tif mgr != nil {\n\t\t\tmodes = append(modes, storeName)\n\t\t}\n\t}\n\n\tif len(modes) == 0 {\n\t\treturn \"INVALID_ADMIN_MODE\"\n\t}\n\n\treturn strings.Join(modes, \",\")\n}\n\nfunc (v *visibilityHybridManager) chooseVisibilityManagerForWrite(ctx context.Context, visFunc func(string) error) error {\n\tvar writeMode string\n\tif v.writeVisibilityStoreName != nil {\n\t\twriteMode = v.writeVisibilityStoreName()\n\t} else {\n\t\tkey := VisibilityAdminDeletionKey(\"visibilityAdminDelete\")\n\t\tif value := ctx.Value(key); value != nil && value.(bool) {\n\t\t\twriteMode = v.chooseVisibilityModeForAdmin()\n\t\t}\n\t}\n\tmodes := strings.Split(writeMode, \",\")\n\tfor i := range modes {\n\t\tmodes[i] = strings.ToLower(strings.TrimSpace(modes[i]))\n\t}\n\n\tvar errors []string\n\tfor _, mode := range modes {\n\t\tif mode == advancedWriteModeOff {\n\t\t\tmode = dbVisStoreName\n\t\t}\n\t\tif mgr, ok := v.visibilityMgrs[mode]; ok && mgr != nil {\n\t\t\tif err := visFunc(mode); err != nil {\n\t\t\t\terrors = append(errors, err.Error())\n\t\t\t}\n\t\t} else if mode != dbVisStoreName && !strings.Contains(writeMode, dbVisStoreName) {\n\t\t\t// If requested mode is not available and it's not already \"db\", fall back to \"db\"\n\t\t\t// when write mode already includes db, skip this step since it will perform the write in another loop\n\t\t\tv.logger.Warn(\"requested visibility mode is not available, falling back to db\", tag.Value(mode))\n\t\t\tif err := visFunc(dbVisStoreName); err != nil {\n\t\t\t\terrors = append(errors, err.Error())\n\t\t\t}\n\t\t} else {\n\t\t\t// If the mode is \"db\" but not available, this is an error\n\t\t\t// This is the else case - when mode is \"db\" but the manager is not available\n\t\t\terrors = append(errors, fmt.Sprintf(\"DB visibility mode is not available: %s\", mode))\n\t\t}\n\t}\n\n\tif len(errors) > 0 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Error writing to visibility: %v\", strings.Join(errors, \"; \")),\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// For Pinot Migration uses. It will be a temporary usage\ntype userParameters struct {\n\toperation    string\n\tdomainName   string\n\tworkflowType string\n\tworkflowID   string\n\tcloseStatus  int // if it is -1, then will have --open flag in comparator workflow\n\tcustomQuery  string\n\tearliestTime int64\n\tlatestTime   int64\n}\n\n// For Visibility Migration uses. It will be a temporary usage\n// logUserQueryParameters will log user queries' parameters so that a comparator workflow can consume\nfunc (v *visibilityHybridManager) logUserQueryParameters(userParam userParameters, domain string, override bool) {\n\t// Don't log if it is not enabled\n\t// don't log if it is a call from Pinot Response Comparator workflow\n\tif !v.logCustomerQueryParameter(domain) || override {\n\t\treturn\n\t}\n\n\trandNum := rand.Intn(10)\n\tif randNum != 5 { // Intentionally to have 1/10 chance to log custom query parameters\n\t\treturn\n\t}\n\n\tv.logger.Info(\"Logging user query parameters for visibility migration response comparator...\",\n\t\ttag.OperationName(userParam.operation),\n\t\ttag.WorkflowDomainName(userParam.domainName),\n\t\ttag.WorkflowType(userParam.workflowType),\n\t\ttag.WorkflowID(userParam.workflowID),\n\t\ttag.WorkflowCloseStatus(userParam.closeStatus),\n\t\ttag.VisibilityQuery(filterAttrPrefix(userParam.customQuery)),\n\t\ttag.EarliestTime(userParam.earliestTime),\n\t\ttag.LatestTime(userParam.latestTime))\n\n}\n\n// This is for only logUserQueryParameters (for Pinot Response comparator) usage.\n// Be careful because there's a low possibility that there'll be false positive cases (shown in unit tests)\nfunc filterAttrPrefix(str string) string {\n\tstr = strings.Replace(str, \"`Attr.\", \"\", -1)\n\treturn strings.Replace(str, \"`\", \"\", -1)\n}\n\nfunc (v *visibilityHybridManager) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListOpenWorkflowExecutions, request, v.logger)\n\t}\n\t// return result from primary\n\treturn manager.ListOpenWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListClosedWorkflowExecutions, request, v.logger)\n\t}\n\treturn manager.ListClosedWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByTypeRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListOpenWorkflowExecutionsByType, request, v.logger)\n\t}\n\treturn manager.ListOpenWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByTypeRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListClosedWorkflowExecutionsByType, request, v.logger)\n\t}\n\treturn manager.ListClosedWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByWorkflowIDRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListOpenWorkflowExecutionsByWorkflowID, request, v.logger)\n\t}\n\treturn manager.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByWorkflowIDRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListClosedWorkflowExecutionsByWorkflowID, request, v.logger)\n\t}\n\treturn manager.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *ListClosedWorkflowExecutionsByStatusRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListClosedWorkflowExecutionsByStatus, request, v.logger)\n\t}\n\treturn manager.ListClosedWorkflowExecutionsByStatus(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *GetClosedWorkflowExecutionRequest,\n) (*GetClosedWorkflowExecutionResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.GetClosedWorkflowExecution, request, v.logger)\n\t}\n\treturn manager.GetClosedWorkflowExecution(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByQueryRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ListWorkflowExecutions, request, v.logger)\n\t}\n\treturn manager.ListWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByQueryRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.ScanWorkflowExecutions, request, v.logger)\n\t}\n\treturn manager.ScanWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *CountWorkflowExecutionsRequest,\n) (*CountWorkflowExecutionsResponse, error) {\n\tmanager, shadowMgr := v.chooseVisibilityManagerForRead(ctx, request.Domain)\n\tif shadowMgr != nil {\n\t\tgo shadow(shadowMgr.CountWorkflowExecutions, request, v.logger)\n\t}\n\treturn manager.CountWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityHybridManager) chooseVisibilityManagerForRead(ctx context.Context, domain string) (VisibilityManager, VisibilityManager) {\n\tvar visibilityMgr, shadowMgr VisibilityManager\n\tstores := strings.Split(v.readVisibilityStoreName(domain), \",\")\n\tfor i := range stores {\n\t\tstores[i] = strings.ToLower(strings.TrimSpace(stores[i]))\n\t}\n\treadStore := stores[0] // if read stores have more than 1, the others will go shadow read\n\tif v.visibilityMgrs[readStore] != nil {\n\t\tvisibilityMgr = v.visibilityMgrs[readStore]\n\t} else {\n\t\tv.logger.Warn(\"domain is configured to read from advanced visibility but it's not available, fall back to basic visibility\",\n\t\t\ttag.WorkflowDomainName(domain))\n\t\tvisibilityMgr = v.visibilityMgrs[dbVisStoreName] //db will always be available\n\t}\n\n\tif len(stores) > 1 {\n\t\tshadowMgr = v.visibilityMgrs[stores[1]]\n\t}\n\n\treturn visibilityMgr, shadowMgr\n}\n\nfunc shadow[ReqT any, ResT any](f func(ctx context.Context, request ReqT) (ResT, error), request ReqT, logger log.Logger) {\n\tctxNew, cancel := context.WithTimeout(context.Background(), 30*time.Second) // don't want f to run too long\n\n\tdefer cancel()\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tlogger.Info(fmt.Sprintf(\"Recovered in Shadow function in double read: %v\", r))\n\t\t}\n\t}()\n\n\t_, err := f(ctxNew, request)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\"Error in Shadow function in double read: %s\", err.Error()))\n\t}\n}\n"
  },
  {
    "path": "common/persistence/visibility_hybrid_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\nconst (\n\tdualStoreName          = \"es,pinot\"    // test dual read for es and pinot, with es as primary\n\tdualStorePinotPrimary  = \"pinot,es\"    // test dual read for es and pinot, with pinot as primary\n\ttripleStoreName        = \"es,pinot,db\" // test triple read for es, pinot and db\n\tdualReadStoreName      = \"es,db\"       // test dual read for es and db\n\tdualReadStoreDBPrimary = \"db,es\"       // test dual read for es and db with db as primary\n\tesStoreName            = \"es\"\n\tpinotStoreName         = \"pinot\"\n\ttestStoreName          = \"test\"\n)\n\nfunc TestNewVisibilityHybridManager(t *testing.T) {\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\tmockDBVisibilityManager    VisibilityManager\n\t\tmockESVisibilityManager    VisibilityManager\n\t\tmockPinotVisibilityManager VisibilityManager\n\t}{\n\t\t\"Case1: nil case\": {\n\t\t\tmockDBVisibilityManager:    nil,\n\t\t\tmockESVisibilityManager:    nil,\n\t\t\tmockPinotVisibilityManager: nil,\n\t\t},\n\t\t\"Case2: success case\": {\n\t\t\tmockDBVisibilityManager:    NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManager:    NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t\t}\n\t\t\t\tNewVisibilityHybridManager(visibilityMgrs, nil, dynamicproperties.GetStringPropertyFn(esStoreName), nil, testStoreName, log.NewNoop())\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestNewVisibilityHybridManager_EmptyVisibilityMgr(t *testing.T) {\n\t// put this outside because need to use it as an input of the table tests\n\tassert.NotPanics(t, func() {\n\t\tvisibilityMgrs := map[string]VisibilityManager{}\n\t\tNewVisibilityHybridManager(visibilityMgrs, nil, nil, nil, testStoreName, log.NewNoop())\n\t})\n}\n\nfunc TestVisibilityHybridManagerClose(t *testing.T) {\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().Close().Return().Times(1)\n\t\t\t},\n\t\t},\n\t\t\"Case1-2: success case with ES visibility is not nil\": {\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().Close().Return().Times(1)\n\t\t\t},\n\t\t},\n\t\t\"Case1-3: success case with pinot visibility is not nil\": {\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().Close().Return().Times(1)\n\t\t\t},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, dynamicproperties.GetStringPropertyFn(esStoreName), nil, testStoreName, log.NewNoop())\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tvisibilityManager.Close()\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridManagerGetName(t *testing.T) {\n\tvisibilityMgrs := map[string]VisibilityManager{\n\t\tdbVisStoreName: NewMockVisibilityManager(gomock.NewController(t)),\n\t}\n\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, dynamicproperties.GetStringPropertyFn(dbVisStoreName), nil, testStoreName, log.NewNoop())\n\tassert.Equal(t, testStoreName, visibilityManager.GetName())\n}\n\nfunc TestVisibilityHybridRecordWorkflowExecutionStarted(t *testing.T) {\n\trequest := &RecordWorkflowExecutionStartedRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *RecordWorkflowExecutionStartedRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with ES visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-3: success case with pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(pinotStoreName),\n\t\t},\n\t\t\"Case1-4: success case with ES visibility is nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionStarted(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridRecordWorkflowExecutionClosed(t *testing.T) {\n\trequest := &RecordWorkflowExecutionClosedRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\tcontext                              context.Context\n\t\trequest                              *RecordWorkflowExecutionClosedRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case0-1: success case with writeVisibilityStoreName is nil - should fall back to db\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\texpectedError: nil, // Should succeed by falling back to db\n\t\t},\n\t\t\"Case0-2: error case with ES has errors in dual mode\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(tripleStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case0-3: error case with ES has errors in On mode with Pinot is not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case0-4: error case with Pinot has errors in On mode\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case0-5: error case with Pinot has errors in Dual mode\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case0-6: error case with ES has errors in Dual mode\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with ES visibility is not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t},\n\t\t\"Case1-3: success case with pinot visibility is not nil\": {\n\t\t\tcontext:                    context.Background(),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(pinotStoreName),\n\t\t},\n\t\t\"Case1-4: success case with dual manager\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t},\n\t\t\"Case1-5: success case with triple manager when ES and Pinot are not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(tripleStoreName),\n\t\t},\n\t\t\"Case2-1: choose both when ES is nil, fall back to db\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case2-2: choose both when Pinot is nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case3-1: chooseVisibilityModeForAdmin when ES is nil\": {\n\t\t\tcontext:                 context.WithValue(context.Background(), VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"Case3-2: chooseVisibilityModeForAdmin when DB is nil\": {\n\t\t\tcontext:                    context.WithValue(context.Background(), VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"Case3-3: chooseVisibilityModeForAdmin when both are not nil\": {\n\t\t\tcontext:                 context.WithValue(context.Background(), VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"Case3-3: chooseVisibilityModeForAdmin when triple are not nil\": {\n\t\t\tcontext:                 context.WithValue(context.Background(), VisibilityAdminDeletionKey(\"visibilityAdminDelete\"), true),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t},\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionClosed(test.context, test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// test an edge case\nfunc TestVisibilityHybridChooseVisibilityModeForAdmin(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdbManager := NewMockVisibilityManager(ctrl)\n\tesManager := NewMockVisibilityManager(ctrl)\n\tpntManager := NewMockVisibilityManager(ctrl)\n\tvisibilityMgrs := map[string]VisibilityManager{\n\t\tdbVisStoreName: dbManager,\n\t\tesStoreName:    esManager,\n\t\tpinotStoreName: pntManager,\n\t}\n\tmgr := NewVisibilityHybridManager(visibilityMgrs, nil, nil, nil, testStoreName, log.NewNoop())\n\ttripleManager := mgr.(*visibilityHybridManager)\n\ttripleManager.visibilityMgrs[dbVisStoreName] = nil\n\ttripleManager.visibilityMgrs[esStoreName] = nil\n\ttripleManager.visibilityMgrs[pinotStoreName] = nil\n\tassert.Equal(t, \"INVALID_ADMIN_MODE\", tripleManager.chooseVisibilityModeForAdmin())\n}\n\nfunc TestVisibilityHybridRecordWorkflowExecutionUninitialized(t *testing.T) {\n\trequest := &RecordWorkflowExecutionUninitializedRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *RecordWorkflowExecutionUninitializedRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-3: success case with ES visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionUninitialized(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridUpsertWorkflowExecution(t *testing.T) {\n\trequest := &UpsertWorkflowExecutionRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *UpsertWorkflowExecutionRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(pinotStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-3: success case with ES visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.UpsertWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridDeleteWorkflowExecution(t *testing.T) {\n\trequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *VisibilityDeleteWorkflowExecutionRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(pinotStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-3: success case with ES visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.DeleteWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridDeleteUninitializedWorkflowExecution(t *testing.T) {\n\trequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *VisibilityDeleteWorkflowExecutionRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(mockESVisibilityManager *MockVisibilityManager)\n\t\twriteVisibilityStoreName             dynamicproperties.StringPropertyFn\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(advancedWriteModeOff),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(pinotStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-3: success case with ES visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(esStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t\t\"Case1-4: success case with both are not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\twriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(dualStoreName),\n\t\t\texpectedError:            nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, nil, test.writeVisibilityStoreName, nil, testStoreName, log.NewNoop())\n\n\t\t\terr := visibilityManager.DeleteUninitializedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFilterAttrPrefix(t *testing.T) {\n\ttests := map[string]struct {\n\t\texpectedInput  string\n\t\texpectedOutput string\n\t}{\n\t\t\"Case1: empty input\": {\n\t\t\texpectedInput:  \"\",\n\t\t\texpectedOutput: \"\",\n\t\t},\n\t\t\"Case2: filtered input\": {\n\t\t\texpectedInput:  \"`Attr.CustomIntField` = 12\",\n\t\t\texpectedOutput: \"CustomIntField = 12\",\n\t\t},\n\t\t\"Case3: complex input\": {\n\t\t\texpectedInput:  \"WorkflowID = 'test-wf' and (`Attr.CustomIntField` = 12 or `Attr.CustomStringField` = 'a-b-c' and WorkflowType = 'wf-type')\",\n\t\t\texpectedOutput: \"WorkflowID = 'test-wf' and (CustomIntField = 12 or CustomStringField = 'a-b-c' and WorkflowType = 'wf-type')\",\n\t\t},\n\t\t\"Case4: false positive case\": {\n\t\t\texpectedInput:  \"`Attr.CustomStringField` = '`Attr.ABCtesting'\",\n\t\t\texpectedOutput: \"CustomStringField = 'ABCtesting'\", // this is supposed to be CustomStringField = '`Attr.ABCtesting'\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tactualOutput := filterAttrPrefix(test.expectedInput)\n\t\t\t\tassert.Equal(t, test.expectedOutput, actualOutput)\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListOpenWorkflowExecutions(t *testing.T) {\n\trequest := &ListWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(esStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: Pinot nil case with double read\": {\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-3: Read mode is from ES with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-4: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-5: double read with panic\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\tpanic(\"test panic\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutions(context.Background(), test.request)\n\n\t\t\twg.Wait()\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListClosedWorkflowExecutions(t *testing.T) {\n\trequest := &ListWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\tcontext                              context.Context\n\t\trequest                              *ListWorkflowExecutionsRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\tcontext:                    context.Background(),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-3: success case with ES visibility is not nil\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(esStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with DB visibility is not nil and read Pinot is true\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: success case with DB visibility is not nil and read modes are false\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(\"db\"),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case3-1: read from ES with context key\": {\n\t\t\tcontext:                 context.Background(),\n\t\t\trequest:                 request,\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(esStoreName),\n\t\t\twgCount:                 0,\n\t\t},\n\t\t\"Case3-2: read from Pinot with context key\": {\n\t\t\tcontext:                    context.Background(),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t},\n\t\t\"Case4-1: success case with double read\": {\n\t\t\tcontext:                    context.Background(),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case4-2: double read with an error\": {\n\t\t\tcontext:                    context.Background(),\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutions(test.context, test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListOpenWorkflowExecutionsByType(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"test-domain\",\n\t\t},\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByTypeRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(esStoreName),\n\t\t\texpectedError:           nil,\n\t\t\twgCount:                 0,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListClosedWorkflowExecutionsByType(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByTypeRequest{\n\t\tListWorkflowExecutionsRequest: ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"test-domain\",\n\t\t},\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByTypeRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListOpenWorkflowExecutionsByWorkflowID(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"test-domain\",\n\t\t},\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByWorkflowIDRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListClosedWorkflowExecutionsByWorkflowID(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByWorkflowIDRequest{\n\t\tListWorkflowExecutionsRequest: ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"test-domain\",\n\t\t},\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByWorkflowIDRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListClosedWorkflowExecutionsByStatus(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\trequest := &ListClosedWorkflowExecutionsByStatusRequest{\n\t\tListWorkflowExecutionsRequest: ListWorkflowExecutionsRequest{\n\t\t\tDomain: \"test-domain\",\n\t\t},\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListClosedWorkflowExecutionsByStatusRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListClosedWorkflowExecutionsByStatusRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListClosedWorkflowExecutionsByStatusRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByStatus(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridGetClosedWorkflowExecution(t *testing.T) {\n\trequest := &GetClosedWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *GetClosedWorkflowExecutionRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *GetClosedWorkflowExecutionRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *GetClosedWorkflowExecutionRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.GetClosedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridListWorkflowExecutions(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByQueryRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByQueryRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ListWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridScanWorkflowExecutions(t *testing.T) {\n\trequest := &ListWorkflowExecutionsByQueryRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// put this outside because need to use it as an input of the table tests\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *ListWorkflowExecutionsByQueryRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.ScanWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n\nfunc TestVisibilityHybridCountWorkflowExecutions(t *testing.T) {\n\trequest := &CountWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\tctrl := gomock.NewController(t)\n\n\ttests := map[string]struct {\n\t\trequest                              *CountWorkflowExecutionsRequest\n\t\tmockDBVisibilityManager              VisibilityManager\n\t\tmockESVisibilityManager              VisibilityManager\n\t\tmockPinotVisibilityManager           VisibilityManager\n\t\tmockDBVisibilityManagerAffordance    func(mockDBVisibilityManager *MockVisibilityManager)\n\t\tmockPinotVisibilityManagerAffordance func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager)\n\t\tmockESVisibilityManagerAffordance    func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager)\n\t\treadVisibilityStoreName              dynamicproperties.StringPropertyFnWithDomainFilter\n\t\twgCount                              int\n\t\texpectedError                        error\n\t}{\n\t\t\"Case1-1: success case with DB visibility is not nil\": {\n\t\t\trequest:                 request,\n\t\t\tmockDBVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockDBVisibilityManagerAffordance: func(mockDBVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockDBVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dbVisStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case1-2: success case with Pinot visibility is not nil\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(pinotStoreName),\n\t\t\twgCount:                 0,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-1: success case with double read\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).DoAndReturn(func(\n\t\t\t\t\tctx context.Context, request *CountWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\twg.Done()\n\t\t\t\t\treturn nil, nil\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStorePinotPrimary),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t\t\"Case2-2: double read with an error\": {\n\t\t\trequest:                    request,\n\t\t\tmockPinotVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockPinotVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockPinotVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockPinotVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(\n\t\t\t\t\t\tctx context.Context, request *CountWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\t\t\t\t\t\twg.Done()\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockESVisibilityManager: NewMockVisibilityManager(ctrl),\n\t\t\tmockESVisibilityManagerAffordance: func(wg *sync.WaitGroup, mockESVisibilityManager *MockVisibilityManager) {\n\t\t\t\tmockESVisibilityManager.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\treadVisibilityStoreName: dynamicproperties.GetStringPropertyFnFilteredByDomain(dualStoreName),\n\t\t\twgCount:                 1,\n\t\t\texpectedError:           nil,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twg := sync.WaitGroup{}\n\t\t\twg.Add(test.wgCount)\n\n\t\t\tif test.mockDBVisibilityManager != nil {\n\t\t\t\ttest.mockDBVisibilityManagerAffordance(test.mockDBVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockPinotVisibilityManager != nil {\n\t\t\t\ttest.mockPinotVisibilityManagerAffordance(&wg, test.mockPinotVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\t\t\tif test.mockESVisibilityManager != nil {\n\t\t\t\ttest.mockESVisibilityManagerAffordance(&wg, test.mockESVisibilityManager.(*MockVisibilityManager))\n\t\t\t}\n\n\t\t\tvisibilityMgrs := map[string]VisibilityManager{\n\t\t\t\tdbVisStoreName: test.mockDBVisibilityManager,\n\t\t\t\tesStoreName:    test.mockESVisibilityManager,\n\t\t\t\tpinotStoreName: test.mockPinotVisibilityManager,\n\t\t\t}\n\t\t\tvisibilityManager := NewVisibilityHybridManager(visibilityMgrs, test.readVisibilityStoreName, nil, dynamicproperties.GetBoolPropertyFnFilteredByDomain(true), testStoreName, log.NewNoop())\n\n\t\t\t_, err := visibilityManager.CountWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\twg.Wait()\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/visibility_manager_interfaces.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination visibility_manager_interfaces_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence VisibilityManager\n// Generate rate limiter wrapper.\n//go:generate gowrap gen -g -p . -i VisibilityManager -t ./wrappers/templates/ratelimited.tmpl -o wrappers/ratelimited/visibility_generated.go\n\n// Generate error injection wrapper.\n//go:generate gowrap gen -g -p . -i VisibilityManager -t ./wrappers/templates/errorinjector.tmpl -o wrappers/errorinjectors/visibility_generated.go\n\n// Generate metered wrapper.\n//go:generate gowrap gen -g -p . -i VisibilityManager -t ./wrappers/templates/metered.tmpl -o wrappers/metered/visibility_generated.go\n\npackage persistence\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Interfaces for the Visibility Store.\n// This is a secondary store that is eventually consistent with the main\n// executions store, and stores workflow execution records for visibility\n// purposes.\n\n// ErrVisibilityOperationNotSupported is an error which indicates that operation is not supported in selected persistence\nvar ErrVisibilityOperationNotSupported = &types.BadRequestError{Message: \"Operation is not supported\"}\n\ntype (\n\t// RecordWorkflowExecutionStartedRequest is used to add a record of a newly\n\t// started execution\n\tRecordWorkflowExecutionStartedRequest struct {\n\t\tDomainUUID                  string\n\t\tDomain                      string // not persisted, used as config filter key\n\t\tExecution                   types.WorkflowExecution\n\t\tWorkflowTypeName            string\n\t\tStartTimestamp              int64\n\t\tExecutionTimestamp          int64\n\t\tWorkflowTimeout             int64 // not persisted, used for cassandra ttl\n\t\tTaskID                      int64 // not persisted, used as condition update version for ES\n\t\tMemo                        *types.Memo\n\t\tTaskList                    string\n\t\tIsCron                      bool\n\t\tNumClusters                 int16\n\t\tClusterAttributeScope       string\n\t\tClusterAttributeName        string\n\t\tUpdateTimestamp             int64 // unit is unix nano, consistent with start/execution timestamp, same in other requests\n\t\tSearchAttributes            map[string][]byte\n\t\tShardID                     int16\n\t\tExecutionStatus             types.WorkflowExecutionStatus\n\t\tCronSchedule                string\n\t\tScheduledExecutionTimestamp int64 // unit is unix nano, used to record the actual execution timestamp if it's a cron workflow\n\t}\n\n\t// RecordWorkflowExecutionClosedRequest is used to add a record of a newly\n\t// closed execution\n\tRecordWorkflowExecutionClosedRequest struct {\n\t\tDomainUUID                  string\n\t\tDomain                      string // not persisted, used as config filter key\n\t\tExecution                   types.WorkflowExecution\n\t\tWorkflowTypeName            string\n\t\tStartTimestamp              int64\n\t\tExecutionTimestamp          int64\n\t\tCloseTimestamp              int64\n\t\tStatus                      types.WorkflowExecutionCloseStatus\n\t\tHistoryLength               int64\n\t\tRetentionSeconds            int64\n\t\tTaskID                      int64 // not persisted, used as condition update version for ES\n\t\tMemo                        *types.Memo\n\t\tTaskList                    string\n\t\tIsCron                      bool\n\t\tCronSchedule                string\n\t\tNumClusters                 int16\n\t\tClusterAttributeScope       string\n\t\tClusterAttributeName        string\n\t\tUpdateTimestamp             int64\n\t\tSearchAttributes            map[string][]byte\n\t\tShardID                     int16\n\t\tExecutionStatus             types.WorkflowExecutionStatus\n\t\tScheduledExecutionTimestamp int64\n\t}\n\n\t// RecordWorkflowExecutionUninitializedRequest is used to add a record of a newly uninitialized execution\n\tRecordWorkflowExecutionUninitializedRequest struct {\n\t\tDomainUUID       string\n\t\tDomain           string\n\t\tExecution        types.WorkflowExecution\n\t\tWorkflowTypeName string\n\t\tUpdateTimestamp  int64\n\t\tShardID          int64\n\t}\n\n\t// UpsertWorkflowExecutionRequest is used to upsert workflow execution\n\tUpsertWorkflowExecutionRequest struct {\n\t\tDomainUUID                  string\n\t\tDomain                      string // not persisted, used as config filter key\n\t\tExecution                   types.WorkflowExecution\n\t\tWorkflowTypeName            string\n\t\tStartTimestamp              int64\n\t\tExecutionTimestamp          int64\n\t\tWorkflowTimeout             int64 // not persisted, used for cassandra ttl\n\t\tTaskID                      int64 // not persisted, used as condition update version for ES\n\t\tMemo                        *types.Memo\n\t\tTaskList                    string\n\t\tIsCron                      bool\n\t\tNumClusters                 int16\n\t\tClusterAttributeScope       string\n\t\tClusterAttributeName        string\n\t\tUpdateTimestamp             int64\n\t\tSearchAttributes            map[string][]byte\n\t\tShardID                     int64\n\t\tExecutionStatus             types.WorkflowExecutionStatus\n\t\tCronSchedule                string\n\t\tScheduledExecutionTimestamp int64 // unit is unix nano, used to record the actual execution timestamp if it's a cron workflow\n\t}\n\n\t// ListWorkflowExecutionsRequest is used to list executions in a domain\n\tListWorkflowExecutionsRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\t// The earliest end of the time range\n\t\tEarliestTime int64\n\t\t// The latest end of the time range\n\t\tLatestTime int64\n\t\t// Maximum number of workflow executions per page\n\t\tPageSize int\n\t\t// Token to continue reading next page of workflow executions.\n\t\t// Pass in empty slice for first page.\n\t\tNextPageToken []byte\n\t}\n\n\t// ListWorkflowExecutionsByQueryRequest is used to list executions in a domain\n\tListWorkflowExecutionsByQueryRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\tPageSize   int    // Maximum number of workflow executions per page\n\t\t// Token to continue reading next page of workflow executions.\n\t\t// Pass in empty slice for first page.\n\t\tNextPageToken []byte\n\t\tQuery         string\n\t}\n\n\t// ListWorkflowExecutionsResponse is the response to ListWorkflowExecutionsRequest\n\tListWorkflowExecutionsResponse struct {\n\t\tExecutions []*types.WorkflowExecutionInfo\n\t\t// Token to read next page if there are more workflow executions beyond page size.\n\t\t// Use this to set NextPageToken on ListWorkflowExecutionsRequest to read the next page.\n\t\tNextPageToken []byte\n\t}\n\n\t// CountWorkflowExecutionsRequest is request from CountWorkflowExecutions\n\tCountWorkflowExecutionsRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\tQuery      string\n\t}\n\n\t// CountWorkflowExecutionsResponse is response to CountWorkflowExecutions\n\tCountWorkflowExecutionsResponse struct {\n\t\tCount int64\n\t}\n\n\t// ListWorkflowExecutionsByTypeRequest is used to list executions of\n\t// a specific type in a domain\n\tListWorkflowExecutionsByTypeRequest struct {\n\t\tListWorkflowExecutionsRequest\n\t\tWorkflowTypeName string\n\t}\n\n\t// ListWorkflowExecutionsByWorkflowIDRequest is used to list executions that\n\t// have specific WorkflowID in a domain\n\tListWorkflowExecutionsByWorkflowIDRequest struct {\n\t\tListWorkflowExecutionsRequest\n\t\tWorkflowID string\n\t}\n\n\t// ListClosedWorkflowExecutionsByStatusRequest is used to list executions that\n\t// have specific close status\n\tListClosedWorkflowExecutionsByStatusRequest struct {\n\t\tListWorkflowExecutionsRequest\n\t\tStatus types.WorkflowExecutionCloseStatus\n\t}\n\n\t// GetClosedWorkflowExecutionRequest is used retrieve the record for a specific execution\n\tGetClosedWorkflowExecutionRequest struct {\n\t\tDomainUUID string\n\t\tDomain     string // domain name is not persisted, but used as config filter key\n\t\tExecution  types.WorkflowExecution\n\t}\n\n\t// GetClosedWorkflowExecutionResponse is the response to GetClosedWorkflowExecutionRequest\n\tGetClosedWorkflowExecutionResponse struct {\n\t\tExecution *types.WorkflowExecutionInfo\n\t}\n\n\t// VisibilityDeleteWorkflowExecutionRequest contains the request params for DeleteWorkflowExecution call\n\tVisibilityDeleteWorkflowExecutionRequest struct {\n\t\tDomainID   string\n\t\tDomain     string\n\t\tRunID      string\n\t\tWorkflowID string\n\t\tTaskID     int64\n\t}\n\n\tVisibilityAdminDeletionKey string\n\n\t// VisibilityManager is used to manage the visibility store\n\tVisibilityManager interface {\n\t\tCloseable\n\t\tGetName() string\n\t\tRecordWorkflowExecutionStarted(ctx context.Context, request *RecordWorkflowExecutionStartedRequest) error\n\t\tRecordWorkflowExecutionClosed(ctx context.Context, request *RecordWorkflowExecutionClosedRequest) error\n\t\tRecordWorkflowExecutionUninitialized(ctx context.Context, request *RecordWorkflowExecutionUninitializedRequest) error\n\t\tUpsertWorkflowExecution(ctx context.Context, request *UpsertWorkflowExecutionRequest) error\n\t\tListOpenWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListOpenWorkflowExecutionsByType(ctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByType(ctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutionsByStatus(ctx context.Context, request *ListClosedWorkflowExecutionsByStatusRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tDeleteWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error\n\t\tListWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tScanWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error)\n\t\tCountWorkflowExecutions(ctx context.Context, request *CountWorkflowExecutionsRequest) (*CountWorkflowExecutionsResponse, error)\n\t\t// NOTE: GetClosedWorkflowExecution is only for persistence testing, currently no index is supported for filtering by RunID\n\t\tGetClosedWorkflowExecution(ctx context.Context, request *GetClosedWorkflowExecutionRequest) (*GetClosedWorkflowExecutionResponse, error)\n\t\tDeleteUninitializedWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error\n\t}\n)\n\n// IsNopUpsertWorkflowRequest return whether upsert request should be no-op\nfunc IsNopUpsertWorkflowRequest(request *InternalUpsertWorkflowExecutionRequest) bool {\n\t_, exist := request.SearchAttributes[definition.CadenceChangeVersion]\n\treturn exist\n}\n"
  },
  {
    "path": "common/persistence/visibility_manager_interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence (interfaces: VisibilityManager)\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -destination visibility_manager_interfaces_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence VisibilityManager\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockVisibilityManager is a mock of VisibilityManager interface.\ntype MockVisibilityManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVisibilityManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockVisibilityManagerMockRecorder is the mock recorder for MockVisibilityManager.\ntype MockVisibilityManagerMockRecorder struct {\n\tmock *MockVisibilityManager\n}\n\n// NewMockVisibilityManager creates a new mock instance.\nfunc NewMockVisibilityManager(ctrl *gomock.Controller) *MockVisibilityManager {\n\tmock := &MockVisibilityManager{ctrl: ctrl}\n\tmock.recorder = &MockVisibilityManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVisibilityManager) EXPECT() *MockVisibilityManagerMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockVisibilityManager) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockVisibilityManagerMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockVisibilityManager)(nil).Close))\n}\n\n// CountWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityManager) CountWorkflowExecutions(ctx context.Context, request *CountWorkflowExecutionsRequest) (*CountWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*CountWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountWorkflowExecutions indicates an expected call of CountWorkflowExecutions.\nfunc (mr *MockVisibilityManagerMockRecorder) CountWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountWorkflowExecutions\", reflect.TypeOf((*MockVisibilityManager)(nil).CountWorkflowExecutions), ctx, request)\n}\n\n// DeleteUninitializedWorkflowExecution mocks base method.\nfunc (m *MockVisibilityManager) DeleteUninitializedWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteUninitializedWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteUninitializedWorkflowExecution indicates an expected call of DeleteUninitializedWorkflowExecution.\nfunc (mr *MockVisibilityManagerMockRecorder) DeleteUninitializedWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteUninitializedWorkflowExecution\", reflect.TypeOf((*MockVisibilityManager)(nil).DeleteUninitializedWorkflowExecution), ctx, request)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockVisibilityManager) DeleteWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockVisibilityManagerMockRecorder) DeleteWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockVisibilityManager)(nil).DeleteWorkflowExecution), ctx, request)\n}\n\n// GetClosedWorkflowExecution mocks base method.\nfunc (m *MockVisibilityManager) GetClosedWorkflowExecution(ctx context.Context, request *GetClosedWorkflowExecutionRequest) (*GetClosedWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClosedWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*GetClosedWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetClosedWorkflowExecution indicates an expected call of GetClosedWorkflowExecution.\nfunc (mr *MockVisibilityManagerMockRecorder) GetClosedWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClosedWorkflowExecution\", reflect.TypeOf((*MockVisibilityManager)(nil).GetClosedWorkflowExecution), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockVisibilityManager) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockVisibilityManagerMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockVisibilityManager)(nil).GetName))\n}\n\n// ListClosedWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityManager) ListClosedWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutions indicates an expected call of ListClosedWorkflowExecutions.\nfunc (mr *MockVisibilityManagerMockRecorder) ListClosedWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutions\", reflect.TypeOf((*MockVisibilityManager)(nil).ListClosedWorkflowExecutions), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByStatus mocks base method.\nfunc (m *MockVisibilityManager) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *ListClosedWorkflowExecutionsByStatusRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByStatus\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByStatus indicates an expected call of ListClosedWorkflowExecutionsByStatus.\nfunc (mr *MockVisibilityManagerMockRecorder) ListClosedWorkflowExecutionsByStatus(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByStatus\", reflect.TypeOf((*MockVisibilityManager)(nil).ListClosedWorkflowExecutionsByStatus), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByType mocks base method.\nfunc (m *MockVisibilityManager) ListClosedWorkflowExecutionsByType(ctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByType\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByType indicates an expected call of ListClosedWorkflowExecutionsByType.\nfunc (mr *MockVisibilityManagerMockRecorder) ListClosedWorkflowExecutionsByType(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByType\", reflect.TypeOf((*MockVisibilityManager)(nil).ListClosedWorkflowExecutionsByType), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByWorkflowID mocks base method.\nfunc (m *MockVisibilityManager) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByWorkflowID\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByWorkflowID indicates an expected call of ListClosedWorkflowExecutionsByWorkflowID.\nfunc (mr *MockVisibilityManagerMockRecorder) ListClosedWorkflowExecutionsByWorkflowID(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByWorkflowID\", reflect.TypeOf((*MockVisibilityManager)(nil).ListClosedWorkflowExecutionsByWorkflowID), ctx, request)\n}\n\n// ListOpenWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityManager) ListOpenWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutions indicates an expected call of ListOpenWorkflowExecutions.\nfunc (mr *MockVisibilityManagerMockRecorder) ListOpenWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutions\", reflect.TypeOf((*MockVisibilityManager)(nil).ListOpenWorkflowExecutions), ctx, request)\n}\n\n// ListOpenWorkflowExecutionsByType mocks base method.\nfunc (m *MockVisibilityManager) ListOpenWorkflowExecutionsByType(ctx context.Context, request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutionsByType\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutionsByType indicates an expected call of ListOpenWorkflowExecutionsByType.\nfunc (mr *MockVisibilityManagerMockRecorder) ListOpenWorkflowExecutionsByType(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutionsByType\", reflect.TypeOf((*MockVisibilityManager)(nil).ListOpenWorkflowExecutionsByType), ctx, request)\n}\n\n// ListOpenWorkflowExecutionsByWorkflowID mocks base method.\nfunc (m *MockVisibilityManager) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutionsByWorkflowID\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutionsByWorkflowID indicates an expected call of ListOpenWorkflowExecutionsByWorkflowID.\nfunc (mr *MockVisibilityManagerMockRecorder) ListOpenWorkflowExecutionsByWorkflowID(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutionsByWorkflowID\", reflect.TypeOf((*MockVisibilityManager)(nil).ListOpenWorkflowExecutionsByWorkflowID), ctx, request)\n}\n\n// ListWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityManager) ListWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListWorkflowExecutions indicates an expected call of ListWorkflowExecutions.\nfunc (mr *MockVisibilityManagerMockRecorder) ListWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListWorkflowExecutions\", reflect.TypeOf((*MockVisibilityManager)(nil).ListWorkflowExecutions), ctx, request)\n}\n\n// RecordWorkflowExecutionClosed mocks base method.\nfunc (m *MockVisibilityManager) RecordWorkflowExecutionClosed(ctx context.Context, request *RecordWorkflowExecutionClosedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionClosed\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionClosed indicates an expected call of RecordWorkflowExecutionClosed.\nfunc (mr *MockVisibilityManagerMockRecorder) RecordWorkflowExecutionClosed(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionClosed\", reflect.TypeOf((*MockVisibilityManager)(nil).RecordWorkflowExecutionClosed), ctx, request)\n}\n\n// RecordWorkflowExecutionStarted mocks base method.\nfunc (m *MockVisibilityManager) RecordWorkflowExecutionStarted(ctx context.Context, request *RecordWorkflowExecutionStartedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionStarted\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionStarted indicates an expected call of RecordWorkflowExecutionStarted.\nfunc (mr *MockVisibilityManagerMockRecorder) RecordWorkflowExecutionStarted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionStarted\", reflect.TypeOf((*MockVisibilityManager)(nil).RecordWorkflowExecutionStarted), ctx, request)\n}\n\n// RecordWorkflowExecutionUninitialized mocks base method.\nfunc (m *MockVisibilityManager) RecordWorkflowExecutionUninitialized(ctx context.Context, request *RecordWorkflowExecutionUninitializedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionUninitialized\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionUninitialized indicates an expected call of RecordWorkflowExecutionUninitialized.\nfunc (mr *MockVisibilityManagerMockRecorder) RecordWorkflowExecutionUninitialized(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionUninitialized\", reflect.TypeOf((*MockVisibilityManager)(nil).RecordWorkflowExecutionUninitialized), ctx, request)\n}\n\n// ScanWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityManager) ScanWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ScanWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ScanWorkflowExecutions indicates an expected call of ScanWorkflowExecutions.\nfunc (mr *MockVisibilityManagerMockRecorder) ScanWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScanWorkflowExecutions\", reflect.TypeOf((*MockVisibilityManager)(nil).ScanWorkflowExecutions), ctx, request)\n}\n\n// UpsertWorkflowExecution mocks base method.\nfunc (m *MockVisibilityManager) UpsertWorkflowExecution(ctx context.Context, request *UpsertWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpsertWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpsertWorkflowExecution indicates an expected call of UpsertWorkflowExecution.\nfunc (mr *MockVisibilityManagerMockRecorder) UpsertWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpsertWorkflowExecution\", reflect.TypeOf((*MockVisibilityManager)(nil).UpsertWorkflowExecution), ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/visibility_single_manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tvisibilityManagerImpl struct {\n\t\tserializer  PayloadSerializer\n\t\tpersistence VisibilityStore\n\t\tlogger      log.Logger\n\t\tdc          *DynamicConfiguration\n\t}\n)\n\nvar _ VisibilityManager = (*visibilityManagerImpl)(nil)\n\n// NewVisibilityManagerImpl returns new VisibilityManager via a VisibilityStore\nfunc NewVisibilityManagerImpl(persistence VisibilityStore, logger log.Logger, dc *DynamicConfiguration) VisibilityManager {\n\treturn &visibilityManagerImpl{\n\t\tserializer:  NewPayloadSerializer(),\n\t\tpersistence: persistence,\n\t\tlogger:      logger,\n\t\tdc:          dc,\n\t}\n}\n\nfunc (v *visibilityManagerImpl) Close() {\n\tv.persistence.Close()\n}\n\nfunc (v *visibilityManagerImpl) GetName() string {\n\treturn v.persistence.GetName()\n}\n\nfunc (v *visibilityManagerImpl) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionStartedRequest,\n) error {\n\treq := &InternalRecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID:             request.DomainUUID,\n\t\tWorkflowID:             request.Execution.GetWorkflowID(),\n\t\tRunID:                  request.Execution.GetRunID(),\n\t\tWorkflowTypeName:       request.WorkflowTypeName,\n\t\tStartTimestamp:         time.Unix(0, request.StartTimestamp),\n\t\tExecutionTimestamp:     time.Unix(0, request.ExecutionTimestamp),\n\t\tWorkflowTimeout:        common.SecondsToDuration(request.WorkflowTimeout),\n\t\tTaskID:                 request.TaskID,\n\t\tTaskList:               request.TaskList,\n\t\tIsCron:                 request.IsCron,\n\t\tCronSchedule:           request.CronSchedule,\n\t\tNumClusters:            request.NumClusters,\n\t\tClusterAttributeScope:  request.ClusterAttributeScope,\n\t\tClusterAttributeName:   request.ClusterAttributeName,\n\t\tMemo:                   v.serializeMemo(request.Memo, request.DomainUUID, request.Execution.GetWorkflowID(), request.Execution.GetRunID()),\n\t\tUpdateTimestamp:        time.Unix(0, request.UpdateTimestamp),\n\t\tSearchAttributes:       request.SearchAttributes,\n\t\tShardID:                request.ShardID,\n\t\tExecutionStatus:        request.ExecutionStatus,\n\t\tScheduledExecutionTime: time.Unix(0, request.ScheduledExecutionTimestamp),\n\t}\n\treturn v.persistence.RecordWorkflowExecutionStarted(ctx, req)\n}\n\nfunc (v *visibilityManagerImpl) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionClosedRequest,\n) error {\n\treq := &InternalRecordWorkflowExecutionClosedRequest{\n\t\tDomainUUID:             request.DomainUUID,\n\t\tWorkflowID:             request.Execution.GetWorkflowID(),\n\t\tRunID:                  request.Execution.GetRunID(),\n\t\tWorkflowTypeName:       request.WorkflowTypeName,\n\t\tStartTimestamp:         time.Unix(0, request.StartTimestamp),\n\t\tExecutionTimestamp:     time.Unix(0, request.ExecutionTimestamp),\n\t\tTaskID:                 request.TaskID,\n\t\tMemo:                   v.serializeMemo(request.Memo, request.DomainUUID, request.Execution.GetWorkflowID(), request.Execution.GetRunID()),\n\t\tTaskList:               request.TaskList,\n\t\tClusterAttributeScope:  request.ClusterAttributeScope,\n\t\tClusterAttributeName:   request.ClusterAttributeName,\n\t\tSearchAttributes:       request.SearchAttributes,\n\t\tCloseTimestamp:         time.Unix(0, request.CloseTimestamp),\n\t\tStatus:                 request.Status,\n\t\tHistoryLength:          request.HistoryLength,\n\t\tRetentionPeriod:        common.SecondsToDuration(request.RetentionSeconds),\n\t\tIsCron:                 request.IsCron,\n\t\tCronSchedule:           request.CronSchedule,\n\t\tNumClusters:            request.NumClusters,\n\t\tUpdateTimestamp:        time.Unix(0, request.UpdateTimestamp),\n\t\tShardID:                request.ShardID,\n\t\tExecutionStatus:        request.ExecutionStatus,\n\t\tScheduledExecutionTime: time.Unix(0, request.ScheduledExecutionTimestamp),\n\t}\n\treturn v.persistence.RecordWorkflowExecutionClosed(ctx, req)\n}\n\nfunc (v *visibilityManagerImpl) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *RecordWorkflowExecutionUninitializedRequest,\n) error {\n\treq := &InternalRecordWorkflowExecutionUninitializedRequest{\n\t\tDomainUUID:       request.DomainUUID,\n\t\tWorkflowID:       request.Execution.GetWorkflowID(),\n\t\tRunID:            request.Execution.GetRunID(),\n\t\tWorkflowTypeName: request.WorkflowTypeName,\n\t\tUpdateTimestamp:  time.Unix(0, request.UpdateTimestamp),\n\t\tShardID:          request.ShardID,\n\t}\n\treturn v.persistence.RecordWorkflowExecutionUninitialized(ctx, req)\n}\n\nfunc (v *visibilityManagerImpl) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *UpsertWorkflowExecutionRequest,\n) error {\n\treq := &InternalUpsertWorkflowExecutionRequest{\n\t\tDomainUUID:                  request.DomainUUID,\n\t\tWorkflowID:                  request.Execution.GetWorkflowID(),\n\t\tRunID:                       request.Execution.GetRunID(),\n\t\tWorkflowTypeName:            request.WorkflowTypeName,\n\t\tStartTimestamp:              time.Unix(0, request.StartTimestamp),\n\t\tExecutionTimestamp:          time.Unix(0, request.ExecutionTimestamp),\n\t\tTaskID:                      request.TaskID,\n\t\tMemo:                        v.serializeMemo(request.Memo, request.DomainUUID, request.Execution.GetWorkflowID(), request.Execution.GetRunID()),\n\t\tTaskList:                    request.TaskList,\n\t\tIsCron:                      request.IsCron,\n\t\tCronSchedule:                request.CronSchedule,\n\t\tNumClusters:                 request.NumClusters,\n\t\tClusterAttributeScope:       request.ClusterAttributeScope,\n\t\tClusterAttributeName:        request.ClusterAttributeName,\n\t\tUpdateTimestamp:             time.Unix(0, request.UpdateTimestamp),\n\t\tSearchAttributes:            request.SearchAttributes,\n\t\tShardID:                     request.ShardID,\n\t\tExecutionStatus:             request.ExecutionStatus,\n\t\tScheduledExecutionTimestamp: request.ScheduledExecutionTimestamp,\n\t}\n\treturn v.persistence.UpsertWorkflowExecution(ctx, req)\n}\n\nfunc (v *visibilityManagerImpl) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalResp, err := v.persistence.ListOpenWorkflowExecutions(ctx, v.toInternalListWorkflowExecutionsRequest(request))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalResp, err := v.persistence.ListClosedWorkflowExecutions(ctx, v.toInternalListWorkflowExecutionsRequest(request))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByTypeRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalListRequest := v.toInternalListWorkflowExecutionsRequest(&request.ListWorkflowExecutionsRequest)\n\tinternalRequest := &InternalListWorkflowExecutionsByTypeRequest{\n\t\tWorkflowTypeName: request.WorkflowTypeName,\n\t}\n\tif internalListRequest != nil {\n\t\tinternalRequest.InternalListWorkflowExecutionsRequest = *internalListRequest\n\t}\n\tinternalResp, err := v.persistence.ListOpenWorkflowExecutionsByType(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByTypeRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalListRequest := v.toInternalListWorkflowExecutionsRequest(&request.ListWorkflowExecutionsRequest)\n\tinternalRequest := &InternalListWorkflowExecutionsByTypeRequest{\n\t\tWorkflowTypeName: request.WorkflowTypeName,\n\t}\n\tif internalListRequest != nil {\n\t\tinternalRequest.InternalListWorkflowExecutionsRequest = *internalListRequest\n\t}\n\tinternalResp, err := v.persistence.ListClosedWorkflowExecutionsByType(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByWorkflowIDRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalListRequest := v.toInternalListWorkflowExecutionsRequest(&request.ListWorkflowExecutionsRequest)\n\tinternalRequest := &InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tWorkflowID: request.WorkflowID,\n\t}\n\tif internalListRequest != nil {\n\t\tinternalRequest.InternalListWorkflowExecutionsRequest = *internalListRequest\n\t}\n\tinternalResp, err := v.persistence.ListOpenWorkflowExecutionsByWorkflowID(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByWorkflowIDRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalListRequest := v.toInternalListWorkflowExecutionsRequest(&request.ListWorkflowExecutionsRequest)\n\tinternalRequest := &InternalListWorkflowExecutionsByWorkflowIDRequest{\n\t\tWorkflowID: request.WorkflowID,\n\t}\n\tif internalListRequest != nil {\n\t\tinternalRequest.InternalListWorkflowExecutionsRequest = *internalListRequest\n\t}\n\tinternalResp, err := v.persistence.ListClosedWorkflowExecutionsByWorkflowID(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *ListClosedWorkflowExecutionsByStatusRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalListRequest := v.toInternalListWorkflowExecutionsRequest(&request.ListWorkflowExecutionsRequest)\n\tinternalRequest := &InternalListClosedWorkflowExecutionsByStatusRequest{\n\t\tStatus: request.Status,\n\t}\n\tif internalListRequest != nil {\n\t\tinternalRequest.InternalListWorkflowExecutionsRequest = *internalListRequest\n\t}\n\tinternalResp, err := v.persistence.ListClosedWorkflowExecutionsByStatus(ctx, internalRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *GetClosedWorkflowExecutionRequest,\n) (*GetClosedWorkflowExecutionResponse, error) {\n\tinternalReq := &InternalGetClosedWorkflowExecutionRequest{\n\t\tDomainUUID: request.DomainUUID,\n\t\tDomain:     request.Domain,\n\t\tExecution:  request.Execution,\n\t}\n\tinternalResp, err := v.persistence.GetClosedWorkflowExecution(ctx, internalReq)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalGetResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn v.persistence.DeleteWorkflowExecution(ctx, request)\n}\n\nfunc (v *visibilityManagerImpl) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn v.persistence.DeleteUninitializedWorkflowExecution(ctx, request)\n}\n\nfunc (v *visibilityManagerImpl) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByQueryRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalResp, err := v.persistence.ListWorkflowExecutions(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *ListWorkflowExecutionsByQueryRequest,\n) (*ListWorkflowExecutionsResponse, error) {\n\tinternalResp, err := v.persistence.ScanWorkflowExecutions(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn v.convertInternalListResponse(internalResp), nil\n}\n\nfunc (v *visibilityManagerImpl) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *CountWorkflowExecutionsRequest,\n) (*CountWorkflowExecutionsResponse, error) {\n\treturn v.persistence.CountWorkflowExecutions(ctx, request)\n}\n\nfunc (v *visibilityManagerImpl) convertInternalGetResponse(internalResp *InternalGetClosedWorkflowExecutionResponse) *GetClosedWorkflowExecutionResponse {\n\tif internalResp == nil {\n\t\treturn nil\n\t}\n\n\tresp := &GetClosedWorkflowExecutionResponse{}\n\tresp.Execution = v.convertVisibilityWorkflowExecutionInfo(internalResp.Execution)\n\treturn resp\n}\n\nfunc (v *visibilityManagerImpl) convertInternalListResponse(internalResp *InternalListWorkflowExecutionsResponse) *ListWorkflowExecutionsResponse {\n\tif internalResp == nil {\n\t\treturn nil\n\t}\n\n\tresp := &ListWorkflowExecutionsResponse{}\n\tresp.Executions = make([]*types.WorkflowExecutionInfo, len(internalResp.Executions))\n\tfor i, execution := range internalResp.Executions {\n\t\tresp.Executions[i] = v.convertVisibilityWorkflowExecutionInfo(execution)\n\t}\n\n\tresp.NextPageToken = internalResp.NextPageToken\n\treturn resp\n}\n\nfunc (v *visibilityManagerImpl) getSearchAttributes(attr map[string]interface{}) (*types.SearchAttributes, error) {\n\tindexedFields := make(map[string][]byte)\n\tvar err error\n\tvar valBytes []byte\n\tfor k, val := range attr {\n\t\tvalBytes, err = json.Marshal(val)\n\t\tif err != nil {\n\t\t\tv.logger.Error(\"error when encode search attributes\", tag.Value(val))\n\t\t\tcontinue\n\t\t}\n\t\tindexedFields[k] = valBytes\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.SearchAttributes{\n\t\tIndexedFields: indexedFields,\n\t}, nil\n}\n\nfunc (v *visibilityManagerImpl) convertVisibilityWorkflowExecutionInfo(execution *InternalVisibilityWorkflowExecutionInfo) *types.WorkflowExecutionInfo {\n\t// special handling of ExecutionTime for cron or retry\n\tif execution.ExecutionTime.UnixNano() == 0 {\n\t\texecution.ExecutionTime = execution.StartTime\n\t}\n\n\tmemo, err := v.serializer.DeserializeVisibilityMemo(execution.Memo)\n\tif err != nil {\n\t\tv.logger.Error(\"failed to deserialize memo\",\n\t\t\ttag.WorkflowID(execution.WorkflowID),\n\t\t\ttag.WorkflowRunID(execution.RunID),\n\t\t\ttag.Error(err))\n\t}\n\tsearchAttributes, err := v.getSearchAttributes(execution.SearchAttributes)\n\tif err != nil {\n\t\tv.logger.Error(\"failed to convert search attributes\",\n\t\t\ttag.WorkflowID(execution.WorkflowID),\n\t\t\ttag.WorkflowRunID(execution.RunID),\n\t\t\ttag.Error(err))\n\t}\n\n\tconvertedExecution := &types.WorkflowExecutionInfo{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\tRunID:      execution.RunID,\n\t\t},\n\t\tType: &types.WorkflowType{\n\t\t\tName: execution.TypeName,\n\t\t},\n\t\tStartTime:              common.Int64Ptr(execution.StartTime.UnixNano()),\n\t\tExecutionTime:          common.Int64Ptr(execution.ExecutionTime.UnixNano()),\n\t\tMemo:                   memo,\n\t\tSearchAttributes:       searchAttributes,\n\t\tTaskList:               &types.TaskList{Name: execution.TaskList},\n\t\tIsCron:                 execution.IsCron,\n\t\tCronSchedule:           common.StringPtr(execution.CronSchedule),\n\t\tExecutionStatus:        &execution.ExecutionStatus,\n\t\tScheduledExecutionTime: common.Int64Ptr(execution.ScheduledExecutionTime.UnixNano()),\n\t}\n\n\t// for close records\n\tif execution.Status != nil {\n\t\tconvertedExecution.CloseTime = common.Int64Ptr(execution.CloseTime.UnixNano())\n\t\tconvertedExecution.CloseStatus = execution.Status\n\t\tconvertedExecution.HistoryLength = execution.HistoryLength\n\t}\n\n\treturn convertedExecution\n}\n\nfunc (v *visibilityManagerImpl) toInternalListWorkflowExecutionsRequest(req *ListWorkflowExecutionsRequest) *InternalListWorkflowExecutionsRequest {\n\tif req == nil {\n\t\treturn nil\n\t}\n\treturn &InternalListWorkflowExecutionsRequest{\n\t\tDomainUUID:    req.DomainUUID,\n\t\tDomain:        req.Domain,\n\t\tEarliestTime:  time.Unix(0, req.EarliestTime),\n\t\tLatestTime:    time.Unix(0, req.LatestTime),\n\t\tPageSize:      req.PageSize,\n\t\tNextPageToken: req.NextPageToken,\n\t}\n}\n\nfunc (v *visibilityManagerImpl) serializeMemo(visibilityMemo *types.Memo, domainID, wID, rID string) *DataBlob {\n\tmemo, err := v.serializer.SerializeVisibilityMemo(visibilityMemo, constants.EncodingType(v.dc.SerializationEncoding()))\n\tif err != nil {\n\t\tv.logger.WithTags(\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.Error(err)).\n\t\t\tError(\"Unable to encode visibility memo\")\n\t}\n\tif memo == nil {\n\t\treturn &DataBlob{}\n\t}\n\treturn memo\n}\n"
  },
  {
    "path": "common/persistence/visibility_single_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewVisibilityManagerImpl(t *testing.T) {\n\ttests := map[string]struct {\n\t\tname string\n\t}{\n\t\t\"Case1: success case\": {\n\t\t\tname: \"TestNewVisibilityManagerImpl\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tNewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t\t})\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestClose(t *testing.T) {\n\ttests := map[string]struct {\n\t\tname string\n\t}{\n\t\t\"Case1: success case\": {\n\t\t\tname: \"TestNewVisibilityManagerImpl\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tmockVisibilityStore.EXPECT().Close().Return().Times(1)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tvisibilityManager.Close()\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestGetName(t *testing.T) {\n\ttests := map[string]struct {\n\t\tname string\n\t}{\n\t\t\"Case1: success case\": {\n\t\t\tname: \"TestNewVisibilityManagerImpl\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tmockVisibilityStore.EXPECT().GetName().Return(testTableName).Times(1)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tvisibilityManager.GetName()\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestRecordWorkflowExecutionStarted(t *testing.T) {\n\terrorRequest := &RecordWorkflowExecutionStartedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"err-wid\",\n\t\t\tRunID:      \"err-rid\",\n\t\t},\n\t\tMemo: &types.Memo{\n\t\t\tFields: map[string][]byte{\n\t\t\t\t\"Service\": []byte(\"servername1\"),\n\t\t\t},\n\t\t},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\trequest := &RecordWorkflowExecutionStartedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tMemo: &types.Memo{},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *RecordWorkflowExecutionStartedRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionStarted(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRecordWorkflowExecutionClosed(t *testing.T) {\n\terrorRequest := &RecordWorkflowExecutionClosedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"err-wid\",\n\t\t\tRunID:      \"err-rid\",\n\t\t},\n\t\tMemo: &types.Memo{},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\trequest := &RecordWorkflowExecutionClosedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tMemo: &types.Memo{},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *RecordWorkflowExecutionClosedRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionClosed(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRecordWorkflowExecutionUninitialized(t *testing.T) {\n\terrorRequest := &RecordWorkflowExecutionUninitializedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"err-wid\",\n\t\t\tRunID:      \"err-rid\",\n\t\t},\n\t}\n\n\trequest := &RecordWorkflowExecutionUninitializedRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *RecordWorkflowExecutionUninitializedRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.RecordWorkflowExecutionUninitialized(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpsertWorkflowExecution(t *testing.T) {\n\terrorRequest := &UpsertWorkflowExecutionRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"err-wid\",\n\t\t\tRunID:      \"err-rid\",\n\t\t},\n\t\tMemo: &types.Memo{},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\trequest := &UpsertWorkflowExecutionRequest{\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tMemo: &types.Memo{},\n\t\tSearchAttributes: map[string][]byte{\n\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\trequest                   *UpsertWorkflowExecutionRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.UpsertWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteWorkflowExecution(t *testing.T) {\n\terrorRequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\trequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *VisibilityDeleteWorkflowExecutionRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.DeleteWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteUninitializedWorkflowExecution(t *testing.T) {\n\terrorRequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\trequest := &VisibilityDeleteWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *VisibilityDeleteWorkflowExecutionRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\terr := visibilityManager.DeleteUninitializedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutions(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsRequest{}\n\n\trequest := &ListWorkflowExecutionsRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutions(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsRequest{}\n\n\trequest := &ListWorkflowExecutionsRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsByType(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByTypeRequest{}\n\n\trequest := &ListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByTypeRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTestListClosedWorkflowExecutionsByType(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByTypeRequest{}\n\n\trequest := &ListWorkflowExecutionsByTypeRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByTypeRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByType(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsByWorkflowID(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\trequest := &ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByWorkflowIDRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListOpenWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsByWorkflowID(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\trequest := &ListWorkflowExecutionsByWorkflowIDRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByWorkflowIDRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByWorkflowID(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsByStatus(t *testing.T) {\n\terrorRequest := &ListClosedWorkflowExecutionsByStatusRequest{}\n\n\trequest := &ListClosedWorkflowExecutionsByStatusRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListClosedWorkflowExecutionsByStatusRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListClosedWorkflowExecutionsByStatus(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetClosedWorkflowExecution(t *testing.T) {\n\terrorRequest := &GetClosedWorkflowExecutionRequest{}\n\n\trequest := &GetClosedWorkflowExecutionRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *GetClosedWorkflowExecutionRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&InternalGetClosedWorkflowExecutionResponse{\n\t\t\t\t\t\tExecution: &InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\t\t\t\tMemo: &DataBlob{\n\t\t\t\t\t\t\t\tData: []byte(\"test string\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.GetClosedWorkflowExecution(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListWorkflowExecutions(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByQueryRequest{}\n\n\trequest := &ListWorkflowExecutionsByQueryRequest{}\n\n\ttestStatus := types.WorkflowExecutionCloseStatus(2)\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByQueryRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&InternalListWorkflowExecutionsResponse{\n\t\t\t\t\t\tExecutions: []*InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tStatus: &testStatus,\n\t\t\t\t\t\t\t\tMemo: &DataBlob{\n\t\t\t\t\t\t\t\t\tData: []byte(\"test string\"),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tSearchAttributes: map[string]interface{}{\n\t\t\t\t\t\t\t\t\t\"CustomStringField\": []byte(\"test string\"),\n\t\t\t\t\t\t\t\t\t\"CustomTimeField\":   []byte(\"2020-01-01T00:00:00Z\"),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ListWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScanWorkflowExecutions(t *testing.T) {\n\terrorRequest := &ListWorkflowExecutionsByQueryRequest{}\n\n\trequest := &ListWorkflowExecutionsByQueryRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *ListWorkflowExecutionsByQueryRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.ScanWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCountWorkflowExecutions(t *testing.T) {\n\terrorRequest := &CountWorkflowExecutionsRequest{}\n\n\trequest := &CountWorkflowExecutionsRequest{}\n\n\ttests := map[string]struct {\n\t\trequest                   *CountWorkflowExecutionsRequest\n\t\tvisibilityStoreAffordance func(mockVisibilityStore *MockVisibilityStore)\n\t\texpectedError             error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\trequest: errorRequest,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\trequest: request,\n\t\t\tvisibilityStoreAffordance: func(mockVisibilityStore *MockVisibilityStore) {\n\t\t\t\tmockVisibilityStore.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\n\t\t\ttest.visibilityStoreAffordance(mockVisibilityStore)\n\n\t\t\t_, err := visibilityManager.CountWorkflowExecutions(context.Background(), test.request)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetSearchAttributes(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput          *map[string]interface{}\n\t\texpectedOutput *types.SearchAttributes\n\t\texpectedError  error\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\tinput: &map[string]interface{}{\n\t\t\t\t\"CustomStringField\": make(chan int),\n\t\t\t},\n\t\t\texpectedError: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\tinput:         &map[string]interface{}{},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityManagerImpl := visibilityManager.(*visibilityManagerImpl)\n\n\t\t\tactualOutput, actualErr := visibilityManagerImpl.getSearchAttributes(*test.input)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Error(t, actualErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, actualErr)\n\t\t\t\tassert.NotNil(t, actualOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConvertVisibilityWorkflowExecutionInfo(t *testing.T) {\n\ttestExecutionTime := int64(0)\n\ttestStartTime := time.Now()\n\ttestResultTime := testStartTime.UnixNano()\n\n\ttests := map[string]struct {\n\t\tinput          *InternalVisibilityWorkflowExecutionInfo\n\t\texpectedOutput *types.WorkflowExecutionInfo\n\t}{\n\t\t\"Case1: error case\": {\n\t\t\tinput: &InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tSearchAttributes: map[string]interface{}{\n\t\t\t\t\t\"CustomStringField\": make(chan int),\n\t\t\t\t},\n\t\t\t\tExecutionTime: time.UnixMilli(int64(testExecutionTime)),\n\t\t\t\tStartTime:     testStartTime,\n\t\t\t},\n\t\t\texpectedOutput: &types.WorkflowExecutionInfo{\n\t\t\t\tExecutionTime: &testResultTime,\n\t\t\t},\n\t\t},\n\t\t\"Case2: normal case with Execution Time is 0\": {\n\t\t\tinput: &InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tExecutionTime: time.UnixMilli(int64(testExecutionTime)),\n\t\t\t\tStartTime:     testStartTime,\n\t\t\t},\n\t\t\texpectedOutput: &types.WorkflowExecutionInfo{\n\t\t\t\tExecutionTime: &testResultTime,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\t\t\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\t\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\t})\n\t\t\tvisibilityManagerImpl := visibilityManager.(*visibilityManagerImpl)\n\n\t\t\tactualOutput := visibilityManagerImpl.convertVisibilityWorkflowExecutionInfo(test.input)\n\t\t\tassert.Equal(t, *test.expectedOutput.ExecutionTime, *actualOutput.ExecutionTime)\n\t\t})\n\t}\n}\n\nfunc TestToInternalListWorkflowExecutionsRequest(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\tvisibilityManagerImpl := visibilityManager.(*visibilityManagerImpl)\n\n\tassert.Nil(t, visibilityManagerImpl.toInternalListWorkflowExecutionsRequest(nil))\n}\n\nfunc TestSerializeMemo(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockVisibilityStore := NewMockVisibilityStore(ctrl)\n\tmockPayloadSerializer := NewMockPayloadSerializer(ctrl)\n\tmockPayloadSerializer.EXPECT().SerializeVisibilityMemo(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\tvisibilityManager := NewVisibilityManagerImpl(mockVisibilityStore, log.NewNoop(), &DynamicConfiguration{\n\t\tSerializationEncoding: dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t})\n\tvisibilityManagerImpl := visibilityManager.(*visibilityManagerImpl)\n\tvisibilityManagerImpl.serializer = mockPayloadSerializer\n\tassert.NotPanics(t, func() {\n\t\tvisibilityManagerImpl.serializeMemo(nil, \"testDomain\", \"testWorkflowID\", \"testRunID\")\n\t})\n}\n"
  },
  {
    "path": "common/persistence/visibility_store_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/persistence (interfaces: VisibilityStore)\n//\n// Generated by this command:\n//\n//\tmockgen -package persistence -destination visibility_store_mock.go -self_package github.com/uber/cadence/common/persistence github.com/uber/cadence/common/persistence VisibilityStore\n//\n\n// Package persistence is a generated GoMock package.\npackage persistence\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockVisibilityStore is a mock of VisibilityStore interface.\ntype MockVisibilityStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVisibilityStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockVisibilityStoreMockRecorder is the mock recorder for MockVisibilityStore.\ntype MockVisibilityStoreMockRecorder struct {\n\tmock *MockVisibilityStore\n}\n\n// NewMockVisibilityStore creates a new mock instance.\nfunc NewMockVisibilityStore(ctrl *gomock.Controller) *MockVisibilityStore {\n\tmock := &MockVisibilityStore{ctrl: ctrl}\n\tmock.recorder = &MockVisibilityStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVisibilityStore) EXPECT() *MockVisibilityStoreMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockVisibilityStore) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockVisibilityStoreMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockVisibilityStore)(nil).Close))\n}\n\n// CountWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityStore) CountWorkflowExecutions(ctx context.Context, request *CountWorkflowExecutionsRequest) (*CountWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*CountWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountWorkflowExecutions indicates an expected call of CountWorkflowExecutions.\nfunc (mr *MockVisibilityStoreMockRecorder) CountWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountWorkflowExecutions\", reflect.TypeOf((*MockVisibilityStore)(nil).CountWorkflowExecutions), ctx, request)\n}\n\n// DeleteUninitializedWorkflowExecution mocks base method.\nfunc (m *MockVisibilityStore) DeleteUninitializedWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteUninitializedWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteUninitializedWorkflowExecution indicates an expected call of DeleteUninitializedWorkflowExecution.\nfunc (mr *MockVisibilityStoreMockRecorder) DeleteUninitializedWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteUninitializedWorkflowExecution\", reflect.TypeOf((*MockVisibilityStore)(nil).DeleteUninitializedWorkflowExecution), ctx, request)\n}\n\n// DeleteWorkflowExecution mocks base method.\nfunc (m *MockVisibilityStore) DeleteWorkflowExecution(ctx context.Context, request *VisibilityDeleteWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteWorkflowExecution indicates an expected call of DeleteWorkflowExecution.\nfunc (mr *MockVisibilityStoreMockRecorder) DeleteWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflowExecution\", reflect.TypeOf((*MockVisibilityStore)(nil).DeleteWorkflowExecution), ctx, request)\n}\n\n// GetClosedWorkflowExecution mocks base method.\nfunc (m *MockVisibilityStore) GetClosedWorkflowExecution(ctx context.Context, request *InternalGetClosedWorkflowExecutionRequest) (*InternalGetClosedWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClosedWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*InternalGetClosedWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetClosedWorkflowExecution indicates an expected call of GetClosedWorkflowExecution.\nfunc (mr *MockVisibilityStoreMockRecorder) GetClosedWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClosedWorkflowExecution\", reflect.TypeOf((*MockVisibilityStore)(nil).GetClosedWorkflowExecution), ctx, request)\n}\n\n// GetName mocks base method.\nfunc (m *MockVisibilityStore) GetName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetName indicates an expected call of GetName.\nfunc (mr *MockVisibilityStoreMockRecorder) GetName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetName\", reflect.TypeOf((*MockVisibilityStore)(nil).GetName))\n}\n\n// ListClosedWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityStore) ListClosedWorkflowExecutions(ctx context.Context, request *InternalListWorkflowExecutionsRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutions indicates an expected call of ListClosedWorkflowExecutions.\nfunc (mr *MockVisibilityStoreMockRecorder) ListClosedWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutions\", reflect.TypeOf((*MockVisibilityStore)(nil).ListClosedWorkflowExecutions), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByStatus mocks base method.\nfunc (m *MockVisibilityStore) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *InternalListClosedWorkflowExecutionsByStatusRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByStatus\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByStatus indicates an expected call of ListClosedWorkflowExecutionsByStatus.\nfunc (mr *MockVisibilityStoreMockRecorder) ListClosedWorkflowExecutionsByStatus(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByStatus\", reflect.TypeOf((*MockVisibilityStore)(nil).ListClosedWorkflowExecutionsByStatus), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByType mocks base method.\nfunc (m *MockVisibilityStore) ListClosedWorkflowExecutionsByType(ctx context.Context, request *InternalListWorkflowExecutionsByTypeRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByType\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByType indicates an expected call of ListClosedWorkflowExecutionsByType.\nfunc (mr *MockVisibilityStoreMockRecorder) ListClosedWorkflowExecutionsByType(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByType\", reflect.TypeOf((*MockVisibilityStore)(nil).ListClosedWorkflowExecutionsByType), ctx, request)\n}\n\n// ListClosedWorkflowExecutionsByWorkflowID mocks base method.\nfunc (m *MockVisibilityStore) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *InternalListWorkflowExecutionsByWorkflowIDRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutionsByWorkflowID\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutionsByWorkflowID indicates an expected call of ListClosedWorkflowExecutionsByWorkflowID.\nfunc (mr *MockVisibilityStoreMockRecorder) ListClosedWorkflowExecutionsByWorkflowID(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutionsByWorkflowID\", reflect.TypeOf((*MockVisibilityStore)(nil).ListClosedWorkflowExecutionsByWorkflowID), ctx, request)\n}\n\n// ListOpenWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityStore) ListOpenWorkflowExecutions(ctx context.Context, request *InternalListWorkflowExecutionsRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutions indicates an expected call of ListOpenWorkflowExecutions.\nfunc (mr *MockVisibilityStoreMockRecorder) ListOpenWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutions\", reflect.TypeOf((*MockVisibilityStore)(nil).ListOpenWorkflowExecutions), ctx, request)\n}\n\n// ListOpenWorkflowExecutionsByType mocks base method.\nfunc (m *MockVisibilityStore) ListOpenWorkflowExecutionsByType(ctx context.Context, request *InternalListWorkflowExecutionsByTypeRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutionsByType\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutionsByType indicates an expected call of ListOpenWorkflowExecutionsByType.\nfunc (mr *MockVisibilityStoreMockRecorder) ListOpenWorkflowExecutionsByType(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutionsByType\", reflect.TypeOf((*MockVisibilityStore)(nil).ListOpenWorkflowExecutionsByType), ctx, request)\n}\n\n// ListOpenWorkflowExecutionsByWorkflowID mocks base method.\nfunc (m *MockVisibilityStore) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *InternalListWorkflowExecutionsByWorkflowIDRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutionsByWorkflowID\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutionsByWorkflowID indicates an expected call of ListOpenWorkflowExecutionsByWorkflowID.\nfunc (mr *MockVisibilityStoreMockRecorder) ListOpenWorkflowExecutionsByWorkflowID(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutionsByWorkflowID\", reflect.TypeOf((*MockVisibilityStore)(nil).ListOpenWorkflowExecutionsByWorkflowID), ctx, request)\n}\n\n// ListWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityStore) ListWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListWorkflowExecutions indicates an expected call of ListWorkflowExecutions.\nfunc (mr *MockVisibilityStoreMockRecorder) ListWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListWorkflowExecutions\", reflect.TypeOf((*MockVisibilityStore)(nil).ListWorkflowExecutions), ctx, request)\n}\n\n// RecordWorkflowExecutionClosed mocks base method.\nfunc (m *MockVisibilityStore) RecordWorkflowExecutionClosed(ctx context.Context, request *InternalRecordWorkflowExecutionClosedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionClosed\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionClosed indicates an expected call of RecordWorkflowExecutionClosed.\nfunc (mr *MockVisibilityStoreMockRecorder) RecordWorkflowExecutionClosed(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionClosed\", reflect.TypeOf((*MockVisibilityStore)(nil).RecordWorkflowExecutionClosed), ctx, request)\n}\n\n// RecordWorkflowExecutionStarted mocks base method.\nfunc (m *MockVisibilityStore) RecordWorkflowExecutionStarted(ctx context.Context, request *InternalRecordWorkflowExecutionStartedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionStarted\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionStarted indicates an expected call of RecordWorkflowExecutionStarted.\nfunc (mr *MockVisibilityStoreMockRecorder) RecordWorkflowExecutionStarted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionStarted\", reflect.TypeOf((*MockVisibilityStore)(nil).RecordWorkflowExecutionStarted), ctx, request)\n}\n\n// RecordWorkflowExecutionUninitialized mocks base method.\nfunc (m *MockVisibilityStore) RecordWorkflowExecutionUninitialized(ctx context.Context, request *InternalRecordWorkflowExecutionUninitializedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordWorkflowExecutionUninitialized\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordWorkflowExecutionUninitialized indicates an expected call of RecordWorkflowExecutionUninitialized.\nfunc (mr *MockVisibilityStoreMockRecorder) RecordWorkflowExecutionUninitialized(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordWorkflowExecutionUninitialized\", reflect.TypeOf((*MockVisibilityStore)(nil).RecordWorkflowExecutionUninitialized), ctx, request)\n}\n\n// ScanWorkflowExecutions mocks base method.\nfunc (m *MockVisibilityStore) ScanWorkflowExecutions(ctx context.Context, request *ListWorkflowExecutionsByQueryRequest) (*InternalListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ScanWorkflowExecutions\", ctx, request)\n\tret0, _ := ret[0].(*InternalListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ScanWorkflowExecutions indicates an expected call of ScanWorkflowExecutions.\nfunc (mr *MockVisibilityStoreMockRecorder) ScanWorkflowExecutions(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScanWorkflowExecutions\", reflect.TypeOf((*MockVisibilityStore)(nil).ScanWorkflowExecutions), ctx, request)\n}\n\n// UpsertWorkflowExecution mocks base method.\nfunc (m *MockVisibilityStore) UpsertWorkflowExecution(ctx context.Context, request *InternalUpsertWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpsertWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpsertWorkflowExecution indicates an expected call of UpsertWorkflowExecution.\nfunc (mr *MockVisibilityStoreMockRecorder) UpsertWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpsertWorkflowExecution\", reflect.TypeOf((*MockVisibilityStore)(nil).UpsertWorkflowExecution), ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/workflowStateCloseStatusValidator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tvalidWorkflowStates = map[int]struct{}{\n\t\tWorkflowStateCreated:   {},\n\t\tWorkflowStateRunning:   {},\n\t\tWorkflowStateCompleted: {},\n\t\tWorkflowStateZombie:    {},\n\t\tWorkflowStateCorrupted: {},\n\t}\n\n\tvalidWorkflowCloseStatuses = map[int]struct{}{\n\t\tWorkflowCloseStatusNone:           {},\n\t\tWorkflowCloseStatusCompleted:      {},\n\t\tWorkflowCloseStatusFailed:         {},\n\t\tWorkflowCloseStatusCanceled:       {},\n\t\tWorkflowCloseStatusTerminated:     {},\n\t\tWorkflowCloseStatusContinuedAsNew: {},\n\t\tWorkflowCloseStatusTimedOut:       {},\n\t}\n)\n\n// ValidateCreateWorkflowStateCloseStatus validate workflow state and close status\nfunc ValidateCreateWorkflowStateCloseStatus(\n\tstate int,\n\tcloseStatus int,\n) error {\n\n\tif err := validateWorkflowState(state); err != nil {\n\t\treturn err\n\t}\n\tif err := validateWorkflowCloseStatus(closeStatus); err != nil {\n\t\treturn err\n\t}\n\n\t// validate workflow state & close status\n\tif state == WorkflowStateCompleted || closeStatus != WorkflowCloseStatusNone {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Create workflow with invalid state: %v or close status: %v\",\n\t\t\t\tstate, closeStatus),\n\t\t}\n\t}\n\treturn nil\n}\n\n// ValidateUpdateWorkflowStateCloseStatus validate workflow state and close status\nfunc ValidateUpdateWorkflowStateCloseStatus(\n\tstate int,\n\tcloseStatus int,\n) error {\n\n\tif err := validateWorkflowState(state); err != nil {\n\t\treturn err\n\t}\n\tif err := validateWorkflowCloseStatus(closeStatus); err != nil {\n\t\treturn err\n\t}\n\n\t// validate workflow state & close status\n\tif closeStatus == WorkflowCloseStatusNone {\n\t\tif state == WorkflowStateCompleted {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Update workflow with invalid state: %v or close status: %v\",\n\t\t\t\t\tstate, closeStatus),\n\t\t\t}\n\t\t}\n\t} else {\n\t\t// WorkflowCloseStatusCompleted\n\t\t// WorkflowCloseStatusFailed\n\t\t// WorkflowCloseStatusCanceled\n\t\t// WorkflowCloseStatusTerminated\n\t\t// WorkflowCloseStatusContinuedAsNew\n\t\t// WorkflowCloseStatusTimedOut\n\t\tif state != WorkflowStateCompleted {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Update workflow with invalid state: %v or close status: %v\",\n\t\t\t\t\tstate, closeStatus),\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\n// validateWorkflowState validate workflow state\nfunc validateWorkflowState(\n\tstate int,\n) error {\n\n\tif _, ok := validWorkflowStates[state]; !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Invalid workflow state: %v\", state),\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// validateWorkflowCloseStatus validate workflow close status\nfunc validateWorkflowCloseStatus(\n\tcloseStatus int,\n) error {\n\n\tif _, ok := validWorkflowCloseStatuses[closeStatus]; !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Invalid workflow close status: %v\", closeStatus),\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// ToInternalWorkflowExecutionCloseStatus convert persistence representation of close status to internal representation\nfunc ToInternalWorkflowExecutionCloseStatus(\n\tcloseStatus int,\n) *types.WorkflowExecutionCloseStatus {\n\n\tswitch closeStatus {\n\tcase WorkflowCloseStatusNone:\n\t\treturn nil\n\tcase WorkflowCloseStatusCompleted:\n\t\treturn types.WorkflowExecutionCloseStatusCompleted.Ptr()\n\tcase WorkflowCloseStatusFailed:\n\t\treturn types.WorkflowExecutionCloseStatusFailed.Ptr()\n\tcase WorkflowCloseStatusCanceled:\n\t\treturn types.WorkflowExecutionCloseStatusCanceled.Ptr()\n\tcase WorkflowCloseStatusTerminated:\n\t\treturn types.WorkflowExecutionCloseStatusTerminated.Ptr()\n\tcase WorkflowCloseStatusContinuedAsNew:\n\t\treturn types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr()\n\tcase WorkflowCloseStatusTimedOut:\n\t\treturn types.WorkflowExecutionCloseStatusTimedOut.Ptr()\n\tdefault:\n\t\tpanic(\"Invalid value for enum WorkflowExecutionCloseStatus\")\n\t}\n}\n\n// FromInternalWorkflowExecutionCloseStatus convert internal representation of close status to persistence representation\nfunc FromInternalWorkflowExecutionCloseStatus(\n\tcloseStatus *types.WorkflowExecutionCloseStatus,\n) int {\n\tif closeStatus == nil {\n\t\treturn WorkflowCloseStatusNone\n\t}\n\tswitch *closeStatus {\n\tcase types.WorkflowExecutionCloseStatusCompleted:\n\t\treturn WorkflowCloseStatusCompleted\n\tcase types.WorkflowExecutionCloseStatusFailed:\n\t\treturn WorkflowCloseStatusFailed\n\tcase types.WorkflowExecutionCloseStatusCanceled:\n\t\treturn WorkflowCloseStatusCanceled\n\tcase types.WorkflowExecutionCloseStatusTerminated:\n\t\treturn WorkflowCloseStatusTerminated\n\tcase types.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\treturn WorkflowCloseStatusContinuedAsNew\n\tcase types.WorkflowExecutionCloseStatusTimedOut:\n\t\treturn WorkflowCloseStatusTimedOut\n\tdefault:\n\t\tpanic(\"Invalid value for enum WorkflowExecutionCloseStatus\")\n\t}\n}\n"
  },
  {
    "path": "common/persistence/workflowStateCloseStatusValidator_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tworkflowStateCloseStatusSuite struct {\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestWorkflowStateCloseStatusSuite(t *testing.T) {\n\ts := new(workflowStateCloseStatusSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *workflowStateCloseStatusSuite) SetupSuite() {\n}\n\nfunc (s *workflowStateCloseStatusSuite) TearDownSuite() {\n\n}\n\nfunc (s *workflowStateCloseStatusSuite) SetupTest() {\n\n}\n\nfunc (s *workflowStateCloseStatusSuite) TearDownTest() {\n\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestCreateWorkflowStateCloseStatus_WorkflowStateCreated() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateCreated, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateCreated, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestCreateWorkflowStateCloseStatus_WorkflowStateRunning() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateRunning, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateRunning, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestCreateWorkflowStateCloseStatus_WorkflowStateCompleted() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusNone,\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateCompleted, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestCreateWorkflowStateCloseStatus_WorkflowStateZombie() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateZombie, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateCreateWorkflowStateCloseStatus(WorkflowStateZombie, closeStatus))\n\t}\n}\n\n// TODO\n\nfunc (s *workflowStateCloseStatusSuite) TestUpdateWorkflowStateCloseStatus_WorkflowStateCreated() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateCreated, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateCreated, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestUpdateWorkflowStateCloseStatus_WorkflowStateRunning() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateRunning, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateRunning, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestUpdateWorkflowStateCloseStatus_WorkflowStateCompleted() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.NotNil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateCompleted, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.Nil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateCompleted, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestUpdateWorkflowStateCloseStatus_WorkflowStateZombie() {\n\tcloseStatuses := []int{\n\t\tWorkflowCloseStatusCompleted,\n\t\tWorkflowCloseStatusFailed,\n\t\tWorkflowCloseStatusCanceled,\n\t\tWorkflowCloseStatusTerminated,\n\t\tWorkflowCloseStatusContinuedAsNew,\n\t\tWorkflowCloseStatusTimedOut,\n\t}\n\n\ts.Nil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateZombie, WorkflowCloseStatusNone))\n\n\tfor _, closeStatus := range closeStatuses {\n\t\ts.NotNil(ValidateUpdateWorkflowStateCloseStatus(WorkflowStateZombie, closeStatus))\n\t}\n}\n\nfunc (s *workflowStateCloseStatusSuite) TestInternalMapping() {\n\ts.Nil(ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusNone))\n\ts.Equal(types.WorkflowExecutionCloseStatusCompleted.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusCompleted))\n\ts.Equal(types.WorkflowExecutionCloseStatusFailed.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusFailed))\n\ts.Equal(types.WorkflowExecutionCloseStatusCanceled.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusCanceled))\n\ts.Equal(types.WorkflowExecutionCloseStatusTerminated.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusTerminated))\n\ts.Equal(types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusContinuedAsNew))\n\ts.Equal(types.WorkflowExecutionCloseStatusTimedOut.Ptr(), ToInternalWorkflowExecutionCloseStatus(WorkflowCloseStatusTimedOut))\n\n\ts.Equal(WorkflowCloseStatusNone, FromInternalWorkflowExecutionCloseStatus(nil))\n\ts.Equal(WorkflowCloseStatusCompleted, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusCompleted.Ptr()))\n\ts.Equal(WorkflowCloseStatusFailed, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusFailed.Ptr()))\n\ts.Equal(WorkflowCloseStatusCanceled, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusCanceled.Ptr()))\n\ts.Equal(WorkflowCloseStatusTerminated, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusTerminated.Ptr()))\n\ts.Equal(WorkflowCloseStatusContinuedAsNew, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr()))\n\ts.Equal(WorkflowCloseStatusTimedOut, FromInternalWorkflowExecutionCloseStatus(types.WorkflowExecutionCloseStatusTimedOut.Ptr()))\n}\n"
  },
  {
    "path": "common/persistence/workflow_execution_info.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// SetNextEventID sets the nextEventID\nfunc (e *WorkflowExecutionInfo) SetNextEventID(id int64) {\n\te.NextEventID = id\n}\n\n// IncreaseNextEventID increase the nextEventID by 1\nfunc (e *WorkflowExecutionInfo) IncreaseNextEventID() {\n\te.NextEventID++\n}\n\n// SetLastFirstEventID set the LastFirstEventID\nfunc (e *WorkflowExecutionInfo) SetLastFirstEventID(id int64) {\n\te.LastFirstEventID = id\n}\n\n// UpdateWorkflowStateCloseStatus update the workflow state\nfunc (e *WorkflowExecutionInfo) UpdateWorkflowStateCloseStatus(\n\tstate int,\n\tcloseStatus int,\n) error {\n\n\tswitch e.State {\n\tcase WorkflowStateVoid:\n\t\t// no validation\n\tcase WorkflowStateCreated:\n\t\tswitch state {\n\t\tcase WorkflowStateCreated:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateRunning:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateCompleted:\n\t\t\tif closeStatus != WorkflowCloseStatusTerminated &&\n\t\t\t\tcloseStatus != WorkflowCloseStatusTimedOut &&\n\t\t\t\tcloseStatus != WorkflowCloseStatusContinuedAsNew {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateZombie:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tdefault:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t\t}\n\t\t}\n\tcase WorkflowStateRunning:\n\t\tswitch state {\n\t\tcase WorkflowStateCreated:\n\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\n\t\tcase WorkflowStateRunning:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateCompleted:\n\t\t\tif closeStatus == WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateZombie:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tdefault:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t\t}\n\t\t}\n\tcase WorkflowStateCompleted:\n\t\tswitch state {\n\t\tcase WorkflowStateCreated:\n\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\n\t\tcase WorkflowStateRunning:\n\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\n\t\tcase WorkflowStateCompleted:\n\t\t\tif closeStatus != e.CloseStatus {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\n\t\t\t}\n\t\tcase WorkflowStateZombie:\n\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\n\t\tdefault:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t\t}\n\t\t}\n\tcase WorkflowStateZombie:\n\t\tswitch state {\n\t\tcase WorkflowStateCreated:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateRunning:\n\t\t\tif closeStatus != WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateCompleted:\n\t\t\tif closeStatus == WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tcase WorkflowStateZombie:\n\t\t\tif closeStatus == WorkflowCloseStatusNone {\n\t\t\t\treturn e.createInvalidStateTransitionErr(e.State, state, closeStatus)\n\t\t\t}\n\n\t\tdefault:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t\t}\n\t\t}\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown workflow state: %v\", state),\n\t\t}\n\t}\n\n\te.State = state\n\te.CloseStatus = closeStatus\n\treturn nil\n\n}\n\nfunc (e *WorkflowExecutionInfo) IsRunning() bool {\n\tswitch e.State {\n\tcase WorkflowStateCreated:\n\t\treturn true\n\tcase WorkflowStateRunning:\n\t\treturn true\n\tcase WorkflowStateCompleted:\n\t\treturn false\n\tcase WorkflowStateZombie:\n\t\treturn false\n\tcase WorkflowStateCorrupted:\n\t\treturn false\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown workflow state: %v\", e.State))\n\t}\n}\n\n// UpdateWorkflowStateCloseStatus update the workflow state\nfunc (e *WorkflowExecutionInfo) createInvalidStateTransitionErr(\n\tcurrentState int,\n\ttargetState int,\n\ttargetCloseStatus int,\n) error {\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(invalidStateTransitionMsg, currentState, targetState, targetCloseStatus),\n\t}\n}\n"
  },
  {
    "path": "common/persistence/workflow_execution_info_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage persistence\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nvar (\n\tworkflowStates = map[string]int{\n\t\t\"Void\":      WorkflowStateVoid,\n\t\t\"Created\":   WorkflowStateCreated,\n\t\t\"Running\":   WorkflowStateRunning,\n\t\t\"Completed\": WorkflowStateCompleted,\n\t\t\"Zombie\":    WorkflowStateZombie,\n\t}\n\tworkflowCloseStatuses = map[string]int{\n\t\t\"None\":           WorkflowCloseStatusNone,\n\t\t\"Completed\":      WorkflowCloseStatusCompleted,\n\t\t\"Failed\":         WorkflowCloseStatusFailed,\n\t\t\"Canceled\":       WorkflowCloseStatusCanceled,\n\t\t\"Terminated\":     WorkflowCloseStatusTerminated,\n\t\t\"ContinuedAsNew\": WorkflowCloseStatusContinuedAsNew,\n\t\t\"TimedOut\":       WorkflowCloseStatusTimedOut,\n\t}\n)\n\nfunc TestWorkflowExecutionInfoSetNextEventID(t *testing.T) {\n\tinfo := &WorkflowExecutionInfo{NextEventID: 1}\n\tinfo.SetNextEventID(2)\n\n\tassert.Equal(t, int64(2), info.NextEventID)\n}\n\nfunc TestWorkflowExecutionInfoIncreaseNextEventID(t *testing.T) {\n\tinfo := &WorkflowExecutionInfo{NextEventID: 1}\n\n\tinfo.IncreaseNextEventID()\n\n\tassert.Equal(t, int64(2), info.NextEventID)\n}\n\nfunc TestWorkflowExecutionInfoSetLastFirstEventID(t *testing.T) {\n\tinfo := &WorkflowExecutionInfo{LastFirstEventID: 1}\n\n\tinfo.SetLastFirstEventID(2)\n\n\tassert.Equal(t, int64(2), info.LastFirstEventID)\n}\n\nfunc TestWorkflowExecutionInfoIsRunning(t *testing.T) {\n\ttests := map[string]struct {\n\t\tstate    int\n\t\texpected bool\n\t}{\n\t\t\"Created\": {\n\t\t\tWorkflowStateCreated, true,\n\t\t},\n\t\t\"Running\": {\n\t\t\tWorkflowStateRunning, true,\n\t\t},\n\t\t\"Completed\": {\n\t\t\tWorkflowStateCompleted, false,\n\t\t},\n\t\t\"Zombie\": {\n\t\t\tWorkflowStateZombie, false,\n\t\t},\n\t\t\"Corrupted\": {\n\t\t\tWorkflowStateCorrupted, false,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tinfo := &WorkflowExecutionInfo{\n\t\t\t\tState: test.state,\n\t\t\t}\n\t\t\tassert.Equal(t, test.expected, info.IsRunning())\n\t\t})\n\t}\n}\n\ntype transitionChecker func(executionInfo *WorkflowExecutionInfo, toState, closeStatus int) bool\ntype stateChecker func(executionInfo *WorkflowExecutionInfo, closeStatus int) bool\ntype testCase struct {\n\tname       string\n\tfromState  int\n\tfromStatus int\n\ttoState    int\n\ttoStatus   int\n}\n\nfunc TestWorkflowExecutionInfoUpdateWorkflowStateCloseStatus(t *testing.T) {\n\texpectedTransitions := map[int]transitionChecker{\n\t\tWorkflowStateVoid: toAnyState(),\n\t\tWorkflowStateCreated: toStates(map[int]stateChecker{\n\t\t\tWorkflowStateCreated:   withCloseStatusNone(),\n\t\t\tWorkflowStateRunning:   withCloseStatusNone(),\n\t\t\tWorkflowStateCompleted: withCloseStatus(WorkflowCloseStatusTerminated, WorkflowCloseStatusTimedOut, WorkflowCloseStatusContinuedAsNew),\n\t\t\tWorkflowStateZombie:    withCloseStatusNone(),\n\t\t}),\n\t\tWorkflowStateRunning: toStates(map[int]stateChecker{\n\t\t\tWorkflowStateRunning:   withCloseStatusNone(),\n\t\t\tWorkflowStateCompleted: withAnyCloseStatusExcept(WorkflowCloseStatusNone),\n\t\t\tWorkflowStateZombie:    withCloseStatusNone(),\n\t\t}),\n\t\tWorkflowStateCompleted: toStates(map[int]stateChecker{\n\t\t\tWorkflowStateCompleted: func(executionInfo *WorkflowExecutionInfo, closeStatus int) bool {\n\t\t\t\treturn executionInfo.CloseStatus == closeStatus\n\t\t\t},\n\t\t}),\n\t\tWorkflowStateZombie: toStates(map[int]stateChecker{\n\t\t\tWorkflowStateCreated:   withCloseStatusNone(),\n\t\t\tWorkflowStateRunning:   withCloseStatusNone(),\n\t\t\tWorkflowStateCompleted: withAnyCloseStatusExcept(WorkflowCloseStatusNone),\n\t\t\tWorkflowStateZombie:    withAnyCloseStatusExcept(WorkflowCloseStatusNone),\n\t\t}),\n\t}\n\tvar testCases []*testCase\n\n\t// From every state with a CloseStatus of None to every state and close status\n\tfor fromStateName, fromState := range workflowStates {\n\t\tfor toStateName, toState := range workflowStates {\n\t\t\tfor closeStatusName, closeStatus := range workflowCloseStatuses {\n\t\t\t\ttestCases = append(testCases, &testCase{\n\t\t\t\t\tname:       fmt.Sprintf(\"(%s,None)->(%s,%s)\", fromStateName, toStateName, closeStatusName),\n\t\t\t\t\tfromState:  fromState,\n\t\t\t\t\tfromStatus: WorkflowCloseStatusNone,\n\t\t\t\t\ttoState:    toState,\n\t\t\t\t\ttoStatus:   closeStatus,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t}\n\t// Ensure Completed doesn't allow for changing the Status\n\ttestCases = append(testCases, &testCase{\n\t\tname:       \"(Completed,Failed)->(Completed, Completed)\",\n\t\tfromState:  WorkflowStateCompleted,\n\t\tfromStatus: WorkflowCloseStatusFailed,\n\t\ttoState:    WorkflowStateCompleted,\n\t\ttoStatus:   WorkflowCloseStatusCompleted,\n\t})\n\ttestCases = append(testCases, &testCase{\n\t\tname:       \"(Completed,Completed)->(Completed,Completed)\",\n\t\tfromState:  WorkflowStateCompleted,\n\t\tfromStatus: WorkflowCloseStatusCompleted,\n\t\ttoState:    WorkflowStateCompleted,\n\t\ttoStatus:   WorkflowCloseStatusCompleted,\n\t})\n\t// Invalid states\n\ttestCases = append(testCases, &testCase{\n\t\tname:       \"(Completed,Completed)->(?,Completed)\",\n\t\tfromState:  WorkflowStateCompleted,\n\t\tfromStatus: WorkflowCloseStatusCompleted,\n\t\ttoState:    100000,\n\t\ttoStatus:   WorkflowCloseStatusCompleted,\n\t})\n\ttestCases = append(testCases, &testCase{\n\t\tname:       \"(?,Completed)->(Completed,Completed)\",\n\t\tfromState:  100000,\n\t\tfromStatus: WorkflowCloseStatusCompleted,\n\t\ttoState:    WorkflowStateCompleted,\n\t\ttoStatus:   WorkflowCloseStatusCompleted,\n\t})\n\n\tfor _, test := range testCases {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\texecutionInfo := &WorkflowExecutionInfo{\n\t\t\t\tState:       test.fromState,\n\t\t\t\tCloseStatus: test.fromStatus,\n\t\t\t}\n\n\t\t\tvar noErr bool\n\t\t\tif isValid, ok := expectedTransitions[test.fromState]; ok {\n\t\t\t\tnoErr = isValid(executionInfo, test.toState, test.toStatus)\n\t\t\t} else {\n\t\t\t\tnoErr = false\n\t\t\t}\n\n\t\t\terr := executionInfo.UpdateWorkflowStateCloseStatus(test.toState, test.toStatus)\n\n\t\t\tif noErr {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, test.toState, executionInfo.State)\n\t\t\t\tassert.Equal(t, test.toStatus, executionInfo.CloseStatus)\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\n\t\t})\n\t}\n}\n\nfunc toAnyState() transitionChecker {\n\treturn func(workflowExecutionInfo *WorkflowExecutionInfo, toState, closeStatus int) bool {\n\t\treturn true\n\t}\n}\n\nfunc toStates(validStates map[int]stateChecker) transitionChecker {\n\treturn func(workflowExecutionInfo *WorkflowExecutionInfo, toState, closeStatus int) bool {\n\t\tif checker, ok := validStates[toState]; ok {\n\t\t\treturn checker(workflowExecutionInfo, closeStatus)\n\t\t}\n\t\treturn false\n\t}\n}\n\nfunc withCloseStatusNone() stateChecker {\n\treturn withCloseStatus(WorkflowCloseStatusNone)\n}\n\nfunc withCloseStatus(closeStatuses ...int) stateChecker {\n\treturn func(workflowExecutionInfo *WorkflowExecutionInfo, closeStatus int) bool {\n\t\treturn contains(closeStatuses, closeStatus)\n\t}\n}\n\nfunc withAnyCloseStatusExcept(closeStatuses ...int) stateChecker {\n\treturn func(workflowExecutionInfo *WorkflowExecutionInfo, closeStatus int) bool {\n\t\tfor _, v := range closeStatuses {\n\t\t\tif v == closeStatus {\n\t\t\t\treturn false\n\t\t\t}\n\t\t}\n\t\treturn true\n\t}\n}\n\nfunc contains(slice []int, value int) bool {\n\tfor _, v := range slice {\n\t\tif v == value {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/configstore_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorConfigStoreManager implements persistence.ConfigStoreManager interface instrumented with error injection.\ntype injectorConfigStoreManager struct {\n\twrapped   persistence.ConfigStoreManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewConfigStoreManager creates a new instance of ConfigStoreManager with error injection.\nfunc NewConfigStoreManager(\n\twrapped persistence.ConfigStoreManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.ConfigStoreManager {\n\treturn &injectorConfigStoreManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorConfigStoreManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorConfigStoreManager) FetchDynamicConfig(ctx context.Context, cfgType persistence.ConfigType) (fp1 *persistence.FetchDynamicConfigResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tfp1, err = c.wrapped.FetchDynamicConfig(ctx, cfgType)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ConfigStoreManager.FetchDynamicConfig\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorConfigStoreManager) UpdateDynamicConfig(ctx context.Context, request *persistence.UpdateDynamicConfigRequest, cfgType persistence.ConfigType) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpdateDynamicConfig(ctx, request, cfgType)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ConfigStoreManager.UpdateDynamicConfig\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/domain_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorDomainManager implements persistence.DomainManager interface instrumented with error injection.\ntype injectorDomainManager struct {\n\twrapped   persistence.DomainManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewDomainManager creates a new instance of DomainManager with error injection.\nfunc NewDomainManager(\n\twrapped persistence.DomainManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.DomainManager {\n\treturn &injectorDomainManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorDomainManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorDomainManager) CreateDomain(ctx context.Context, request *persistence.CreateDomainRequest) (cp1 *persistence.CreateDomainResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.CreateDomain(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.CreateDomain\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) DeleteDomain(ctx context.Context, request *persistence.DeleteDomainRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteDomain(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.DeleteDomain\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) DeleteDomainByName(ctx context.Context, request *persistence.DeleteDomainByNameRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteDomainByName(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.DeleteDomainByName\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) GetDomain(ctx context.Context, request *persistence.GetDomainRequest) (gp1 *persistence.GetDomainResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetDomain(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.GetDomain\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) GetMetadata(ctx context.Context) (gp1 *persistence.GetMetadataResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetMetadata(ctx)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.GetMetadata\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorDomainManager) ListDomains(ctx context.Context, request *persistence.ListDomainsRequest) (lp1 *persistence.ListDomainsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListDomains(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.ListDomains\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorDomainManager) UpdateDomain(ctx context.Context, request *persistence.UpdateDomainRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpdateDomain(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"DomainManager.UpdateDomain\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/execution_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// injectorExecutionManager implements persistence.ExecutionManager interface instrumented with error injection.\ntype injectorExecutionManager struct {\n\twrapped   persistence.ExecutionManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewExecutionManager creates a new instance of ExecutionManager with error injection.\nfunc NewExecutionManager(\n\twrapped persistence.ExecutionManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.ExecutionManager {\n\treturn &injectorExecutionManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorExecutionManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorExecutionManager) CompleteHistoryTask(ctx context.Context, request *persistence.CompleteHistoryTaskRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.CompleteHistoryTask(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.CompleteHistoryTask\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) ConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (cp1 *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.ConflictResolveWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.ConflictResolveWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) CreateFailoverMarkerTasks(ctx context.Context, request *persistence.CreateFailoverMarkersRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.CreateFailoverMarkerTasks(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.CreateFailoverMarkerTasks\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) CreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (cp1 *persistence.CreateWorkflowExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.CreateWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.CreateWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID string, workflowID string, runID string) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.DeleteActiveClusterSelectionPolicy\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) DeleteCurrentWorkflowExecution(ctx context.Context, request *persistence.DeleteCurrentWorkflowExecutionRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteCurrentWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.DeleteCurrentWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) DeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.DeleteReplicationTaskFromDLQRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteReplicationTaskFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.DeleteReplicationTaskFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.DeleteWorkflowExecutionRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.DeleteWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetActiveClusterSelectionPolicy(ctx context.Context, domainID string, wfID string, rID string) (ap1 *types.ActiveClusterSelectionPolicy, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tap1, err = c.wrapped.GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetActiveClusterSelectionPolicy\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetCurrentExecution(ctx context.Context, request *persistence.GetCurrentExecutionRequest) (gp1 *persistence.GetCurrentExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetCurrentExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetCurrentExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetHistoryTasks(ctx context.Context, request *persistence.GetHistoryTasksRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetHistoryTasks(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetHistoryTasks\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorExecutionManager) GetReplicationDLQSize(ctx context.Context, request *persistence.GetReplicationDLQSizeRequest) (gp1 *persistence.GetReplicationDLQSizeResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetReplicationDLQSize(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetReplicationDLQSize\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetReplicationTasksFromDLQ(ctx context.Context, request *persistence.GetReplicationTasksFromDLQRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetReplicationTasksFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetReplicationTasksFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) GetShardID() (i1 int) {\n\treturn c.wrapped.GetShardID()\n}\n\nfunc (c *injectorExecutionManager) GetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (gp1 *persistence.GetWorkflowExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.GetWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) IsWorkflowExecutionExists(ctx context.Context, request *persistence.IsWorkflowExecutionExistsRequest) (ip1 *persistence.IsWorkflowExecutionExistsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tip1, err = c.wrapped.IsWorkflowExecutionExists(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.IsWorkflowExecutionExists\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) ListConcreteExecutions(ctx context.Context, request *persistence.ListConcreteExecutionsRequest) (lp1 *persistence.ListConcreteExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListConcreteExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.ListConcreteExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) ListCurrentExecutions(ctx context.Context, request *persistence.ListCurrentExecutionsRequest) (lp1 *persistence.ListCurrentExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListCurrentExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.ListCurrentExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) PutReplicationTaskToDLQ(ctx context.Context, request *persistence.PutReplicationTaskToDLQRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.PutReplicationTaskToDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.PutReplicationTaskToDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) RangeCompleteHistoryTask(ctx context.Context, request *persistence.RangeCompleteHistoryTaskRequest) (rp1 *persistence.RangeCompleteHistoryTaskResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.RangeCompleteHistoryTask(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.RangeCompleteHistoryTask\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.RangeDeleteReplicationTaskFromDLQRequest) (rp1 *persistence.RangeDeleteReplicationTaskFromDLQResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.RangeDeleteReplicationTaskFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.RangeDeleteReplicationTaskFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorExecutionManager) UpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (up1 *persistence.UpdateWorkflowExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tup1, err = c.wrapped.UpdateWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ExecutionManager.UpdateWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/history_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorHistoryManager implements persistence.HistoryManager interface instrumented with error injection.\ntype injectorHistoryManager struct {\n\twrapped   persistence.HistoryManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewHistoryManager creates a new instance of HistoryManager with error injection.\nfunc NewHistoryManager(\n\twrapped persistence.HistoryManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.HistoryManager {\n\treturn &injectorHistoryManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorHistoryManager) AppendHistoryNodes(ctx context.Context, request *persistence.AppendHistoryNodesRequest) (ap1 *persistence.AppendHistoryNodesResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tap1, err = c.wrapped.AppendHistoryNodes(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.AppendHistoryNodes\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorHistoryManager) DeleteHistoryBranch(ctx context.Context, request *persistence.DeleteHistoryBranchRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteHistoryBranch(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.DeleteHistoryBranch\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) ForkHistoryBranch(ctx context.Context, request *persistence.ForkHistoryBranchRequest) (fp1 *persistence.ForkHistoryBranchResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tfp1, err = c.wrapped.ForkHistoryBranch(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.ForkHistoryBranch\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) GetAllHistoryTreeBranches(ctx context.Context, request *persistence.GetAllHistoryTreeBranchesRequest) (gp1 *persistence.GetAllHistoryTreeBranchesResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetAllHistoryTreeBranches(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.GetAllHistoryTreeBranches\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) GetHistoryTree(ctx context.Context, request *persistence.GetHistoryTreeRequest) (gp1 *persistence.GetHistoryTreeResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetHistoryTree(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.GetHistoryTree\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorHistoryManager) ReadHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.ReadHistoryBranch(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.ReadHistoryBranch\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) ReadHistoryBranchByBatch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchByBatchResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.ReadHistoryBranchByBatch(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.ReadHistoryBranchByBatch\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorHistoryManager) ReadRawHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadRawHistoryBranchResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.ReadRawHistoryBranch(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"HistoryManager.ReadRawHistoryBranch\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/injectors_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errorinjectors\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _staticMethods = map[string]bool{\n\t\"Close\":      true,\n\t\"GetName\":    true,\n\t\"GetShardID\": true,\n}\n\nvar wrappers = []any{\n\t&injectorConfigStoreManager{},\n\t&injectorDomainManager{},\n\t&injectorHistoryManager{},\n\t&injectorQueueManager{},\n\t&injectorShardManager{},\n\t&injectorTaskManager{},\n\t&injectorVisibilityManager{},\n\t&injectorExecutionManager{},\n}\n\nfunc TestInjectorsWithoutErrors(t *testing.T) {\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tobject := builderForPassThrough(t, injector, 0, testlogger.New(t), true, nil)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tcallRes := v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\n\t\t\t\t\tassert.Nil(t, resultErr, \"method %v returned error %v\", method.Name, resultErr)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInjectorsWith100ErrorRate(t *testing.T) {\n\toldRandomStubFunc := _randomStubFunc\n\t_randomStubFunc = func() bool {\n\t\treturn false\n\t}\n\tdefer func() { _randomStubFunc = oldRandomStubFunc }()\n\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// We cannot use test logger here, since logger.Error will fail the test.\n\t\t\tobject := builderForPassThrough(t, injector, 1, log.NewNoop(), false, nil)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tvar callRes []reflect.Value\n\t\t\t\t\tassert.NotPanicsf(t, func() {\n\t\t\t\t\t\tcallRes = v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\t}, \"method does not have tag defined\")\n\n\t\t\t\t\tif len(callRes) == 0 {\n\t\t\t\t\t\t// Empty result means that method panicked.\n\t\t\t\t\t\treturn\n\t\t\t\t\t}\n\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\n\t\t\t\t\terr, ok := resultErr.(error)\n\n\t\t\t\t\tassert.True(t, ok, \"method %v must return error\")\n\t\t\t\t\tassert.True(t, isFakeError(err), \"method %v returned not faked error, got %v\", method.Name, err)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInjectorsWithUnderlyingErrors(t *testing.T) {\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\texpectedMethodErr := fmt.Errorf(\"%s: injected error\", name)\n\t\t\tobject := builderForPassThrough(t, injector, 0, testlogger.New(t), true, expectedMethodErr)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tcallRes := v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\t\t\t\t\terr, ok := resultErr.(error)\n\t\t\t\t\trequire.True(t, ok, \"method %v must return error\")\n\t\t\t\t\tassert.Equal(t, expectedMethodErr, err, \"method %v returned different error, expected %v, got %v\", method.Name, expectedMethodErr, err)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc builderForPassThrough(t *testing.T, injector any, errorRate float64, logger log.Logger, expectCalls bool, expectedErr error) (object any) {\n\tctrl := gomock.NewController(t)\n\t// For error injection tests (errorRate > 0), use a starttime far enough in the past\n\t// to bypass the 30-second warmup period in generateFakeError\n\tstarttime := time.Now()\n\tif errorRate > 0 {\n\t\tstarttime = time.Now().Add(-time.Minute) // 60 seconds ago, past the 30s warmup\n\t}\n\tswitch injector.(type) {\n\tcase *injectorConfigStoreManager:\n\t\tmocked := persistence.NewMockConfigStoreManager(ctrl)\n\t\tobject = NewConfigStoreManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().FetchDynamicConfig(gomock.Any(), gomock.Any()).Return(&persistence.FetchDynamicConfigResponse{}, expectedErr)\n\t\t}\n\tcase *injectorDomainManager:\n\t\tmocked := persistence.NewMockDomainManager(ctrl)\n\t\tobject = NewDomainManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteDomainByName(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(&persistence.ListDomainsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{}, expectedErr)\n\t\t}\n\tcase *injectorHistoryManager:\n\t\tmocked := persistence.NewMockHistoryManager(ctrl)\n\t\tobject = NewHistoryManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().AppendHistoryNodes(gomock.Any(), gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadHistoryBranchByBatch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchByBatchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadRawHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadRawHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ForkHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ForkHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteHistoryBranch(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetHistoryTree(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTreeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetAllHistoryTreeBranches(gomock.Any(), gomock.Any()).Return(&persistence.GetAllHistoryTreeBranchesResponse{}, expectedErr)\n\t\t}\n\tcase *injectorQueueManager:\n\t\tmocked := persistence.NewMockQueueManager(ctrl)\n\t\tobject = NewQueueManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().EnqueueMessage(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ReadMessages(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesResponse{Messages: []*persistence.QueueMessage{}}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteMessagesBefore(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().EnqueueMessageToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetDLQAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr)\n\t\t\tmocked.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQSizeResponse{Size: 0}, expectedErr)\n\t\t\tmocked.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ReadMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesFromDLQResponse{Messages: []*persistence.QueueMessage{}}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tcase *injectorShardManager:\n\t\tmocked := persistence.NewMockShardManager(ctrl)\n\t\tobject = NewShardManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().GetShard(gomock.Any(), gomock.Any()).Return(&persistence.GetShardResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateShard(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CreateShard(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tcase *injectorTaskManager:\n\t\tmocked := persistence.NewMockTaskManager(ctrl)\n\t\tobject = NewTaskManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CompleteTasksLessThan(gomock.Any(), gomock.Any()).Return(&persistence.CompleteTasksLessThanResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CompleteTask(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CreateTasks(gomock.Any(), gomock.Any()).Return(&persistence.CreateTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteTaskList(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetOrphanTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetOrphanTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetTaskListSize(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListSizeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().LeaseTaskList(gomock.Any(), gomock.Any()).Return(&persistence.LeaseTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetTaskList(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListTaskList(gomock.Any(), gomock.Any()).Return(&persistence.ListTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any()).Return(&persistence.UpdateTaskListResponse{}, expectedErr)\n\t\t}\n\tcase *injectorVisibilityManager:\n\t\tmocked := persistence.NewMockVisibilityManager(ctrl)\n\t\tobject = NewVisibilityManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.CountWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetClosedWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t}\n\tcase *injectorExecutionManager:\n\t\tmocked := persistence.NewMockExecutionManager(ctrl)\n\t\tobject = NewExecutionManager(mocked, errorRate, logger, starttime)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.CreateWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetCurrentExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.ConflictResolveWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CreateFailoverMarkerTasks(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetReplicationDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetReplicationDLQSizeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetReplicationTasksFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Any()).Return(&persistence.IsWorkflowExecutionExistsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListConcreteExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListCurrentExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().PutReplicationTaskToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(&persistence.RangeCompleteHistoryTaskResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().RangeDeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.RangeDeleteReplicationTaskFromDLQResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterSelectionPolicy{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tdefault:\n\t\tt.Errorf(\"unsupported type %v\", reflect.TypeOf(injector))\n\t\tt.FailNow()\n\t}\n\treturn\n}\n\nfunc TestGenerateFakeErrorWarmupPeriod(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tstarttime      time.Time\n\t\terrorRate      float64\n\t\texpectNoErrors bool\n\t}{\n\t\t{\n\t\t\tname:           \"no errors within 30 second warmup period\",\n\t\t\tstarttime:      time.Now(), // just started\n\t\t\terrorRate:      1.0,        // 100% error rate\n\t\t\texpectNoErrors: true,       // but no errors during warmup\n\t\t},\n\t\t{\n\t\t\tname:           \"no errors at 29 seconds\",\n\t\t\tstarttime:      time.Now().Add(-29 * time.Second),\n\t\t\terrorRate:      1.0,\n\t\t\texpectNoErrors: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"errors generated after 30 seconds\",\n\t\t\tstarttime:      time.Now().Add(-31 * time.Second),\n\t\t\terrorRate:      1.0,\n\t\t\texpectNoErrors: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"no errors when error rate is 0 even after warmup\",\n\t\t\tstarttime:      time.Now().Add(-time.Minute),\n\t\t\terrorRate:      0.0,\n\t\t\texpectNoErrors: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// Call multiple times to account for randomness\n\t\t\tconst iterations = 100\n\t\t\tvar errorCount int\n\t\t\tfor i := 0; i < iterations; i++ {\n\t\t\t\terr := generateFakeError(tc.errorRate, tc.starttime)\n\t\t\t\tif err != nil {\n\t\t\t\t\terrorCount++\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif tc.expectNoErrors {\n\t\t\t\tassert.Equal(t, 0, errorCount,\n\t\t\t\t\t\"expected no errors during warmup period, but got %d errors in %d iterations\",\n\t\t\t\t\terrorCount, iterations)\n\t\t\t} else {\n\t\t\t\t// With 100% error rate and past warmup, we expect errors\n\t\t\t\tassert.Greater(t, errorCount, 0,\n\t\t\t\t\t\"expected errors after warmup period with 100%% error rate, but got none in %d iterations\",\n\t\t\t\t\titerations)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/queue_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorQueueManager implements persistence.QueueManager interface instrumented with error injection.\ntype injectorQueueManager struct {\n\twrapped   persistence.QueueManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewQueueManager creates a new instance of QueueManager with error injection.\nfunc NewQueueManager(\n\twrapped persistence.QueueManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.QueueManager {\n\treturn &injectorQueueManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorQueueManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorQueueManager) DeleteMessageFromDLQ(ctx context.Context, request *persistence.DeleteMessageFromDLQRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteMessageFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.DeleteMessageFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) DeleteMessagesBefore(ctx context.Context, request *persistence.DeleteMessagesBeforeRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteMessagesBefore(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.DeleteMessagesBefore\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) EnqueueMessage(ctx context.Context, request *persistence.EnqueueMessageRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.EnqueueMessage(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.EnqueueMessage\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) EnqueueMessageToDLQ(ctx context.Context, request *persistence.EnqueueMessageToDLQRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.EnqueueMessageToDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.EnqueueMessageToDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) GetAckLevels(ctx context.Context, request *persistence.GetAckLevelsRequest) (gp1 *persistence.GetAckLevelsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetAckLevels(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.GetAckLevels\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) GetDLQAckLevels(ctx context.Context, request *persistence.GetDLQAckLevelsRequest) (gp1 *persistence.GetDLQAckLevelsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetDLQAckLevels(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.GetDLQAckLevels\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) GetDLQSize(ctx context.Context, request *persistence.GetDLQSizeRequest) (gp1 *persistence.GetDLQSizeResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetDLQSize(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.GetDLQSize\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) RangeDeleteMessagesFromDLQ(ctx context.Context, request *persistence.RangeDeleteMessagesFromDLQRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.RangeDeleteMessagesFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.RangeDeleteMessagesFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) ReadMessages(ctx context.Context, request *persistence.ReadMessagesRequest) (rp1 *persistence.ReadMessagesResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.ReadMessages(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.ReadMessages\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) ReadMessagesFromDLQ(ctx context.Context, request *persistence.ReadMessagesFromDLQRequest) (rp1 *persistence.ReadMessagesFromDLQResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\trp1, err = c.wrapped.ReadMessagesFromDLQ(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.ReadMessagesFromDLQ\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) UpdateAckLevel(ctx context.Context, request *persistence.UpdateAckLevelRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpdateAckLevel(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.UpdateAckLevel\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorQueueManager) UpdateDLQAckLevel(ctx context.Context, request *persistence.UpdateDLQAckLevelRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpdateDLQAckLevel(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"QueueManager.UpdateDLQAckLevel\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/shard_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorShardManager implements persistence.ShardManager interface instrumented with error injection.\ntype injectorShardManager struct {\n\twrapped   persistence.ShardManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewShardManager creates a new instance of ShardManager with error injection.\nfunc NewShardManager(\n\twrapped persistence.ShardManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.ShardManager {\n\treturn &injectorShardManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorShardManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorShardManager) CreateShard(ctx context.Context, request *persistence.CreateShardRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.CreateShard(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ShardManager.CreateShard\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorShardManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorShardManager) GetShard(ctx context.Context, request *persistence.GetShardRequest) (gp1 *persistence.GetShardResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetShard(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ShardManager.GetShard\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorShardManager) UpdateShard(ctx context.Context, request *persistence.UpdateShardRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpdateShard(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"ShardManager.UpdateShard\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/task_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorTaskManager implements persistence.TaskManager interface instrumented with error injection.\ntype injectorTaskManager struct {\n\twrapped   persistence.TaskManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewTaskManager creates a new instance of TaskManager with error injection.\nfunc NewTaskManager(\n\twrapped persistence.TaskManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.TaskManager {\n\treturn &injectorTaskManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorTaskManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorTaskManager) CompleteTask(ctx context.Context, request *persistence.CompleteTaskRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.CompleteTask(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.CompleteTask\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) CompleteTasksLessThan(ctx context.Context, request *persistence.CompleteTasksLessThanRequest) (cp1 *persistence.CompleteTasksLessThanResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.CompleteTasksLessThan(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.CompleteTasksLessThan\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) CreateTasks(ctx context.Context, request *persistence.CreateTasksRequest) (cp1 *persistence.CreateTasksResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.CreateTasks(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.CreateTasks\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) DeleteTaskList(ctx context.Context, request *persistence.DeleteTaskListRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteTaskList(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.DeleteTaskList\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorTaskManager) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (gp1 *persistence.GetOrphanTasksResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetOrphanTasks(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.GetOrphanTasks\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) GetTaskList(ctx context.Context, request *persistence.GetTaskListRequest) (gp1 *persistence.GetTaskListResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetTaskList(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.GetTaskList\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (gp1 *persistence.GetTaskListSizeResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetTaskListSize(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.GetTaskListSize\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) GetTasks(ctx context.Context, request *persistence.GetTasksRequest) (gp1 *persistence.GetTasksResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetTasks(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.GetTasks\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) LeaseTaskList(ctx context.Context, request *persistence.LeaseTaskListRequest) (lp1 *persistence.LeaseTaskListResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.LeaseTaskList(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.LeaseTaskList\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) ListTaskList(ctx context.Context, request *persistence.ListTaskListRequest) (lp1 *persistence.ListTaskListResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListTaskList(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.ListTaskList\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorTaskManager) UpdateTaskList(ctx context.Context, request *persistence.UpdateTaskListRequest) (up1 *persistence.UpdateTaskListResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tup1, err = c.wrapped.UpdateTaskList(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"TaskManager.UpdateTaskList\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/utils.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errorinjectors\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// _randomStubFunc is a stub randomized function that could be overriden in tests.\n// It introduces randomness to timeout and unhandled errors, to mimic retriable db isues.\nvar _randomStubFunc = func() bool {\n\t// forward the call with 50% chance\n\treturn rand.Intn(2) == 0\n}\n\nfunc shouldForwardCallToPersistence(\n\terr error,\n) bool {\n\tif err == nil {\n\t\treturn true\n\t}\n\n\tif err == ErrFakeTimeout || err == errors.ErrFakeUnhandled {\n\t\treturn _randomStubFunc()\n\t}\n\n\treturn false\n}\n\nfunc generateFakeError(\n\terrorRate float64,\n\tstarttime time.Time,\n) error {\n\t// don't inject errors before 30 seconds of startup to avoid overwhelming the system\n\t// on startup\n\tif time.Since(starttime) < 30*time.Second {\n\t\treturn nil\n\t}\n\n\trandFl := rand.Float64()\n\tif randFl < errorRate {\n\t\treturn fakeErrors[rand.Intn(len(fakeErrors))]\n\t}\n\n\treturn nil\n}\n\nvar (\n\t// ErrFakeTimeout is a fake persistence timeout error.\n\tErrFakeTimeout = &persistence.TimeoutError{Msg: \"Fake Persistence Timeout Error.\"}\n)\n\nvar (\n\tfakeErrors = []error{\n\t\terrors.ErrFakeServiceBusy,\n\t\terrors.ErrFakeInternalService,\n\t\tErrFakeTimeout,\n\t\terrors.ErrFakeUnhandled,\n\t}\n)\n\nfunc isFakeError(err error) bool {\n\tfor _, fakeErr := range fakeErrors {\n\t\tif err == fakeErr {\n\t\t\treturn true\n\t\t}\n\t}\n\n\treturn false\n}\n\nconst (\n\tmsgInjectedFakeErr = \"Injected fake persistence error\"\n)\n\nfunc logErr(logger log.Logger, objectMethod string, fakeErr error, forwardCall bool, err error) {\n\tlogger.Error(msgInjectedFakeErr,\n\t\tgetOperationFromMethodName(objectMethod),\n\t\ttag.Error(fakeErr),\n\t\ttag.Bool(forwardCall),\n\t\ttag.StoreError(err),\n\t)\n}\n\nfunc getOperationFromMethodName(op string) tag.Tag {\n\tvar t *tag.Tag\n\tswitch {\n\tcase strings.HasPrefix(op, \"ConfigStoreManager\"):\n\t\tt = configManagerTags(op)\n\tcase strings.HasPrefix(op, \"DomainManager\"):\n\t\tt = domainManagerTags(op)\n\tcase strings.HasPrefix(op, \"HistoryManager\"):\n\t\tt = historyManagerTags(op)\n\tcase strings.HasPrefix(op, \"ShardManager\"):\n\t\tt = shardManagerTags(op)\n\tcase strings.HasPrefix(op, \"ExecutionManager\"):\n\t\tt = executionManagerTags(op)\n\tcase strings.HasPrefix(op, \"VisibilityManager\"):\n\t\tt = visibilityManagerTags(op)\n\tcase strings.HasPrefix(op, \"QueueManager\"):\n\t\tt = queueManagerTags(op)\n\tcase strings.HasPrefix(op, \"TaskManager\"):\n\t\tt = taskManagerTags(op)\n\t}\n\tif t == nil {\n\t\tpanic(fmt.Sprintf(\"No tag defined for operation %s\", op))\n\t}\n\treturn *t\n}\n\nfunc configManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"ConfigStoreManager.FetchDynamicConfig\":\n\t\treturn &tag.StoreOperationFetchDynamicConfig\n\tcase \"ConfigStoreManager.UpdateDynamicConfig\":\n\t\treturn &tag.StoreOperationUpdateDynamicConfig\n\t}\n\treturn nil\n}\n\nfunc domainManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"DomainManager.CreateDomain\":\n\t\treturn &tag.StoreOperationCreateDomain\n\tcase \"DomainManager.GetDomain\":\n\t\treturn &tag.StoreOperationGetDomain\n\tcase \"DomainManager.DeleteDomain\":\n\t\treturn &tag.StoreOperationDeleteDomain\n\tcase \"DomainManager.DeleteDomainByName\":\n\t\treturn &tag.StoreOperationDeleteDomainByName\n\tcase \"DomainManager.ListDomains\":\n\t\treturn &tag.StoreOperationListDomains\n\tcase \"DomainManager.GetMetadata\":\n\t\treturn &tag.StoreOperationGetMetadata\n\tcase \"DomainManager.UpdateDomain\":\n\t\treturn &tag.StoreOperationUpdateDomain\n\t}\n\treturn nil\n}\n\nfunc historyManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"HistoryManager.AppendHistoryNodes\":\n\t\treturn &tag.StoreOperationAppendHistoryNodes\n\tcase \"HistoryManager.DeleteHistoryBranch\":\n\t\treturn &tag.StoreOperationDeleteHistoryBranch\n\tcase \"HistoryManager.ForkHistoryBranch\":\n\t\treturn &tag.StoreOperationForkHistoryBranch\n\tcase \"HistoryManager.GetHistoryTree\":\n\t\treturn &tag.StoreOperationGetHistoryTree\n\tcase \"HistoryManager.ReadHistoryBranch\":\n\t\treturn &tag.StoreOperationReadHistoryBranch\n\tcase \"HistoryManager.ReadHistoryBranchByBatch\":\n\t\treturn &tag.StoreOperationReadHistoryBranchByBatch\n\tcase \"HistoryManager.ReadRawHistoryBranch\":\n\t\treturn &tag.StoreOperationReadRawHistoryBranch\n\tcase \"HistoryManager.GetAllHistoryTreeBranches\":\n\t\treturn &tag.StoreOperationGetAllHistoryTreeBranches\n\t}\n\treturn nil\n}\n\nfunc shardManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"ShardManager.CreateShard\":\n\t\treturn &tag.StoreOperationCreateShard\n\tcase \"ShardManager.GetShard\":\n\t\treturn &tag.StoreOperationGetShard\n\tcase \"ShardManager.UpdateShard\":\n\t\treturn &tag.StoreOperationUpdateShard\n\t}\n\treturn nil\n}\n\nfunc executionManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"ExecutionManager.CreateWorkflowExecution\":\n\t\treturn &tag.StoreOperationCreateWorkflowExecution\n\tcase \"ExecutionManager.GetWorkflowExecution\":\n\t\treturn &tag.StoreOperationGetWorkflowExecution\n\tcase \"ExecutionManager.UpdateWorkflowExecution\":\n\t\treturn &tag.StoreOperationUpdateWorkflowExecution\n\tcase \"ExecutionManager.ResetWorkflowExecution\":\n\t\treturn &tag.StoreOperationResetWorkflowExecution\n\tcase \"ExecutionManager.DeleteWorkflowExecution\":\n\t\treturn &tag.StoreOperationDeleteWorkflowExecution\n\tcase \"ExecutionManager.GetCurrentExecution\":\n\t\treturn &tag.StoreOperationGetCurrentExecution\n\tcase \"ExecutionManager.ConflictResolveWorkflowExecution\":\n\t\treturn &tag.StoreOperationConflictResolveWorkflowExecution\n\tcase \"ExecutionManager.DeleteCurrentWorkflowExecution\":\n\t\treturn &tag.StoreOperationDeleteCurrentWorkflowExecution\n\tcase \"ExecutionManager.ListCurrentExecutions\":\n\t\treturn &tag.StoreOperationListCurrentExecution\n\tcase \"ExecutionManager.IsWorkflowExecutionExists\":\n\t\treturn &tag.StoreOperationIsWorkflowExecutionExists\n\tcase \"ExecutionManager.ListConcreteExecutions\":\n\t\treturn &tag.StoreOperationListConcreteExecution\n\tcase \"ExecutionManager.GetTransferTasks\":\n\t\treturn &tag.StoreOperationGetTransferTasks\n\tcase \"ExecutionManager.GetCrossClusterTasks\":\n\t\treturn &tag.StoreOperationGetCrossClusterTasks\n\tcase \"ExecutionManager.GetReplicationTasks\":\n\t\treturn &tag.StoreOperationGetReplicationTasks\n\tcase \"ExecutionManager.CompleteTransferTask\":\n\t\treturn &tag.StoreOperationCompleteTransferTask\n\tcase \"ExecutionManager.GetHistoryTasks\":\n\t\treturn &tag.StoreOperationGetHistoryTasks\n\tcase \"ExecutionManager.CompleteHistoryTask\":\n\t\treturn &tag.StoreOperationCompleteHistoryTask\n\tcase \"ExecutionManager.RangeCompleteHistoryTask\":\n\t\treturn &tag.StoreOperationRangeCompleteHistoryTask\n\tcase \"ExecutionManager.CompleteCrossClusterTask\":\n\t\treturn &tag.StoreOperationCompleteCrossClusterTask\n\tcase \"ExecutionManager.CompleteReplicationTask\":\n\t\treturn &tag.StoreOperationCompleteReplicationTask\n\tcase \"ExecutionManager.PutReplicationTaskToDLQ\":\n\t\treturn &tag.StoreOperationPutReplicationTaskToDLQ\n\tcase \"ExecutionManager.GetReplicationTasksFromDLQ\":\n\t\treturn &tag.StoreOperationGetReplicationTasksFromDLQ\n\tcase \"ExecutionManager.GetReplicationDLQSize\":\n\t\treturn &tag.StoreOperationGetReplicationDLQSize\n\tcase \"ExecutionManager.DeleteReplicationTaskFromDLQ\":\n\t\treturn &tag.StoreOperationDeleteReplicationTaskFromDLQ\n\tcase \"ExecutionManager.RangeDeleteReplicationTaskFromDLQ\":\n\t\treturn &tag.StoreOperationRangeDeleteReplicationTaskFromDLQ\n\tcase \"ExecutionManager.GetTimerIndexTasks\":\n\t\treturn &tag.StoreOperationGetTimerIndexTasks\n\tcase \"ExecutionManager.CompleteTimerTask\":\n\t\treturn &tag.StoreOperationCompleteTimerTask\n\tcase \"ExecutionManager.CreateFailoverMarkerTasks\":\n\t\treturn &tag.StoreOperationCreateFailoverMarkerTasks\n\tcase \"ExecutionManager.GetActiveClusterSelectionPolicy\":\n\t\treturn &tag.StoreOperationGetActiveClusterSelectionPolicy\n\tcase \"ExecutionManager.DeleteActiveClusterSelectionPolicy\":\n\t\treturn &tag.StoreOperationDeleteActiveClusterSelectionPolicy\n\t}\n\treturn nil\n}\n\nfunc visibilityManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"VisibilityManager.RecordWorkflowExecutionStarted\":\n\t\treturn &tag.StoreOperationRecordWorkflowExecutionStarted\n\tcase \"VisibilityManager.RecordWorkflowExecutionClosed\":\n\t\treturn &tag.StoreOperationRecordWorkflowExecutionClosed\n\tcase \"VisibilityManager.UpsertWorkflowExecution\":\n\t\treturn &tag.StoreOperationUpsertWorkflowExecution\n\tcase \"VisibilityManager.ListOpenWorkflowExecutions\":\n\t\treturn &tag.StoreOperationListOpenWorkflowExecutions\n\tcase \"VisibilityManager.ListClosedWorkflowExecutions\":\n\t\treturn &tag.StoreOperationListClosedWorkflowExecutions\n\tcase \"VisibilityManager.ListOpenWorkflowExecutionsByType\":\n\t\treturn &tag.StoreOperationListOpenWorkflowExecutionsByType\n\tcase \"VisibilityManager.ListClosedWorkflowExecutionsByType\":\n\t\treturn &tag.StoreOperationListClosedWorkflowExecutionsByType\n\tcase \"VisibilityManager.ListOpenWorkflowExecutionsByWorkflowID\":\n\t\treturn &tag.StoreOperationListOpenWorkflowExecutionsByWorkflowID\n\tcase \"VisibilityManager.ListClosedWorkflowExecutionsByWorkflowID\":\n\t\treturn &tag.StoreOperationListClosedWorkflowExecutionsByWorkflowID\n\tcase \"VisibilityManager.ListClosedWorkflowExecutionsByStatus\":\n\t\treturn &tag.StoreOperationListClosedWorkflowExecutionsByStatus\n\tcase \"VisibilityManager.GetClosedWorkflowExecution\":\n\t\treturn &tag.StoreOperationGetClosedWorkflowExecution\n\tcase \"VisibilityManager.DeleteWorkflowExecution\":\n\t\treturn &tag.StoreOperationDeleteWorkflowExecution\n\tcase \"VisibilityManager.ListWorkflowExecutions\":\n\t\treturn &tag.StoreOperationListWorkflowExecutions\n\tcase \"VisibilityManager.ScanWorkflowExecutions\":\n\t\treturn &tag.StoreOperationScanWorkflowExecutions\n\tcase \"VisibilityManager.CountWorkflowExecutions\":\n\t\treturn &tag.StoreOperationCountWorkflowExecutions\n\tcase \"VisibilityManager.DeleteUninitializedWorkflowExecution\":\n\t\treturn &tag.StoreOperationDeleteUninitializedWorkflowExecution\n\tcase \"VisibilityManager.RecordWorkflowExecutionUninitialized\":\n\t\treturn &tag.StoreOperationRecordWorkflowExecutionUninitialized\n\t}\n\treturn nil\n}\n\nfunc taskManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"TaskManager.LeaseTaskList\":\n\t\treturn &tag.StoreOperationLeaseTaskList\n\tcase \"TaskManager.GetTaskList\":\n\t\treturn &tag.StoreOperationGetTaskList\n\tcase \"TaskManager.UpdateTaskList\":\n\t\treturn &tag.StoreOperationUpdateTaskList\n\tcase \"TaskManager.CreateTasks\":\n\t\treturn &tag.StoreOperationCreateTasks\n\tcase \"TaskManager.GetTasks\":\n\t\treturn &tag.StoreOperationGetTasks\n\tcase \"TaskManager.CompleteTask\":\n\t\treturn &tag.StoreOperationCompleteTask\n\tcase \"TaskManager.CompleteTasksLessThan\":\n\t\treturn &tag.StoreOperationCompleteTasksLessThan\n\tcase \"TaskManager.DeleteTaskList\":\n\t\treturn &tag.StoreOperationDeleteTaskList\n\tcase \"TaskManager.GetOrphanTasks\":\n\t\treturn &tag.StoreOperationGetOrphanTasks\n\tcase \"TaskManager.GetTaskListSize\":\n\t\treturn &tag.StoreOperationGetTaskListSize\n\tcase \"TaskManager.ListTaskList\":\n\t\treturn &tag.StoreOperationListTaskList\n\t}\n\treturn nil\n}\n\nfunc queueManagerTags(op string) *tag.Tag {\n\tswitch op {\n\tcase \"QueueManager.EnqueueMessage\":\n\t\treturn &tag.StoreOperationEnqueueMessage\n\tcase \"QueueManager.EnqueueMessageToDLQ\":\n\t\treturn &tag.StoreOperationEnqueueMessageToDLQ\n\tcase \"QueueManager.DeleteMessageFromDLQ\":\n\t\treturn &tag.StoreOperationDeleteMessageFromDLQ\n\tcase \"QueueManager.RangeDeleteMessagesFromDLQ\":\n\t\treturn &tag.StoreOperationRangeDeleteMessagesFromDLQ\n\tcase \"QueueManager.UpdateAckLevel\":\n\t\treturn &tag.StoreOperationUpdateAckLevel\n\tcase \"QueueManager.GetAckLevels\":\n\t\treturn &tag.StoreOperationGetAckLevels\n\tcase \"QueueManager.UpdateDLQAckLevel\":\n\t\treturn &tag.StoreOperationUpdateDLQAckLevel\n\tcase \"QueueManager.GetDLQAckLevels\":\n\t\treturn &tag.StoreOperationGetDLQAckLevels\n\tcase \"QueueManager.GetDLQSize\":\n\t\treturn &tag.StoreOperationGetDLQSize\n\tcase \"QueueManager.DeleteMessagesBefore\":\n\t\treturn &tag.StoreOperationDeleteMessagesBefore\n\tcase \"QueueManager.ReadMessages\":\n\t\treturn &tag.StoreOperationReadMessages\n\tcase \"QueueManager.ReadMessagesFromDLQ\":\n\t\treturn &tag.StoreOperationReadMessagesFromDLQ\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/persistence/wrappers/errorinjectors/visibility_generated.go",
    "content": "package errorinjectors\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/errorinjector.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// injectorVisibilityManager implements persistence.VisibilityManager interface instrumented with error injection.\ntype injectorVisibilityManager struct {\n\twrapped   persistence.VisibilityManager\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// NewVisibilityManager creates a new instance of VisibilityManager with error injection.\nfunc NewVisibilityManager(\n\twrapped persistence.VisibilityManager,\n\terrorRate float64,\n\tlogger log.Logger,\n\tstarttime time.Time,\n) persistence.VisibilityManager {\n\treturn &injectorVisibilityManager{\n\t\twrapped:   wrapped,\n\t\tstarttime: starttime,\n\t\terrorRate: errorRate,\n\t\tlogger:    logger,\n\t}\n}\n\nfunc (c *injectorVisibilityManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) CountWorkflowExecutions(ctx context.Context, request *persistence.CountWorkflowExecutionsRequest) (cp1 *persistence.CountWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tcp1, err = c.wrapped.CountWorkflowExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.CountWorkflowExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) DeleteUninitializedWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteUninitializedWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.DeleteUninitializedWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.DeleteWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.DeleteWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) GetClosedWorkflowExecution(ctx context.Context, request *persistence.GetClosedWorkflowExecutionRequest) (gp1 *persistence.GetClosedWorkflowExecutionResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tgp1, err = c.wrapped.GetClosedWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.GetClosedWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *injectorVisibilityManager) ListClosedWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListClosedWorkflowExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *persistence.ListClosedWorkflowExecutionsByStatusRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListClosedWorkflowExecutionsByStatus\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListClosedWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByType(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListClosedWorkflowExecutionsByType\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListClosedWorkflowExecutionsByWorkflowID\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListOpenWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListOpenWorkflowExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListOpenWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutionsByType(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListOpenWorkflowExecutionsByType\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListOpenWorkflowExecutionsByWorkflowID\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ListWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ListWorkflowExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ListWorkflowExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) RecordWorkflowExecutionClosed(ctx context.Context, request *persistence.RecordWorkflowExecutionClosedRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.RecordWorkflowExecutionClosed(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.RecordWorkflowExecutionClosed\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) RecordWorkflowExecutionStarted(ctx context.Context, request *persistence.RecordWorkflowExecutionStartedRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.RecordWorkflowExecutionStarted(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.RecordWorkflowExecutionStarted\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) RecordWorkflowExecutionUninitialized(ctx context.Context, request *persistence.RecordWorkflowExecutionUninitializedRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.RecordWorkflowExecutionUninitialized(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.RecordWorkflowExecutionUninitialized\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) ScanWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\tlp1, err = c.wrapped.ScanWorkflowExecutions(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.ScanWorkflowExecutions\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n\nfunc (c *injectorVisibilityManager) UpsertWorkflowExecution(ctx context.Context, request *persistence.UpsertWorkflowExecutionRequest) (err error) {\n\tfakeErr := generateFakeError(c.errorRate, c.starttime)\n\tvar forwardCall bool\n\tif forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t\terr = c.wrapped.UpsertWorkflowExecution(ctx, request)\n\t}\n\n\tif fakeErr != nil {\n\t\tlogErr(c.logger, \"VisibilityManager.UpsertWorkflowExecution\", fakeErr, forwardCall, err)\n\t\terr = fakeErr\n\t\treturn\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/base.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype retryCountKeyType string\n\nconst retryCountKey = retryCountKeyType(\"retryCount\")\n\ntype base struct {\n\tmetricClient                  metrics.Client\n\tlogger                        log.Logger\n\tenableLatencyHistogramMetrics bool\n\tsampleLoggingRate             dynamicproperties.IntPropertyFn\n\tenableShardIDMetrics          dynamicproperties.BoolPropertyFn\n}\n\nfunc (p *base) updateErrorMetricPerDomain(scope metrics.ScopeIdx, err error, scopeWithDomainTag metrics.Scope, logger log.Logger) {\n\tlogger = logger.Helper()\n\n\tswitch {\n\tcase errors.As(err, new(*types.DomainAlreadyExistsError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrDomainAlreadyExistsCounterPerDomain)\n\tcase errors.As(err, new(*types.BadRequestError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrBadRequestCounterPerDomain)\n\tcase errors.As(err, new(*persistence.WorkflowExecutionAlreadyStartedError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrExecutionAlreadyStartedCounterPerDomain)\n\tcase errors.As(err, new(*persistence.ConditionFailedError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrConditionFailedCounterPerDomain)\n\tcase errors.As(err, new(*persistence.CurrentWorkflowConditionFailedError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrCurrentWorkflowConditionFailedCounterPerDomain)\n\tcase errors.As(err, new(*persistence.ShardAlreadyExistError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrShardExistsCounterPerDomain)\n\tcase errors.As(err, new(*persistence.ShardOwnershipLostError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrShardOwnershipLostCounterPerDomain)\n\tcase errors.As(err, new(*types.EntityNotExistsError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrEntityNotExistsCounterPerDomain)\n\tcase errors.As(err, new(*persistence.DuplicateRequestError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrDuplicateRequestCounterPerDomain)\n\tcase errors.As(err, new(*persistence.TimeoutError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrTimeoutCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceFailuresPerDomain)\n\tcase errors.As(err, new(*types.ServiceBusyError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrBusyCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceFailuresPerDomain)\n\tcase errors.As(err, new(*persistence.DBUnavailableError)):\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceErrDBUnavailableCounterPerDomain)\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceFailuresPerDomain)\n\t\tlogger.Error(\"DBUnavailable Error:\", tag.Error(err), tag.MetricScope(int(scope)))\n\tdefault:\n\t\tlogger.Error(\"Operation failed with internal error.\", tag.Error(err), tag.MetricScope(int(scope)))\n\t\tscopeWithDomainTag.IncCounter(metrics.PersistenceFailuresPerDomain)\n\t}\n}\n\nfunc (p *base) updateErrorMetric(scope metrics.ScopeIdx, err error, metricsScope metrics.Scope, logger log.Logger) {\n\tlogger = logger.Helper()\n\n\tswitch {\n\tcase errors.As(err, new(*types.DomainAlreadyExistsError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrDomainAlreadyExistsCounter)\n\tcase errors.As(err, new(*types.BadRequestError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrBadRequestCounter)\n\tcase errors.As(err, new(*persistence.WorkflowExecutionAlreadyStartedError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrExecutionAlreadyStartedCounter)\n\tcase errors.As(err, new(*persistence.ConditionFailedError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrConditionFailedCounter)\n\tcase errors.As(err, new(*persistence.CurrentWorkflowConditionFailedError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrCurrentWorkflowConditionFailedCounter)\n\tcase errors.As(err, new(*persistence.ShardAlreadyExistError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrShardExistsCounter)\n\tcase errors.As(err, new(*persistence.ShardOwnershipLostError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrShardOwnershipLostCounter)\n\tcase errors.As(err, new(*types.EntityNotExistsError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrEntityNotExistsCounter)\n\tcase errors.As(err, new(*persistence.DuplicateRequestError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrDuplicateRequestCounter)\n\tcase errors.As(err, new(*persistence.TimeoutError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrTimeoutCounter)\n\t\tmetricsScope.IncCounter(metrics.PersistenceFailures)\n\tcase errors.As(err, new(*types.ServiceBusyError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrBusyCounter)\n\t\tmetricsScope.IncCounter(metrics.PersistenceFailures)\n\tcase errors.As(err, new(*persistence.DBUnavailableError)):\n\t\tmetricsScope.IncCounter(metrics.PersistenceErrDBUnavailableCounter)\n\t\tmetricsScope.IncCounter(metrics.PersistenceFailures)\n\t\tlogger.Error(\"DBUnavailable Error:\", tag.Error(err), tag.MetricScope(int(scope)))\n\tdefault:\n\t\tlogger.Error(\"Operation failed with internal error.\", tag.Error(err), tag.MetricScope(int(scope)))\n\t\tmetricsScope.IncCounter(metrics.PersistenceFailures)\n\t}\n}\n\nfunc (p *base) call(scope metrics.ScopeIdx, op func() error, tags ...metrics.Tag) error {\n\tmetricsScope := p.metricClient.Scope(scope, tags...)\n\tif len(tags) > 0 {\n\t\tmetricsScope.IncCounter(metrics.PersistenceRequestsPerDomain)\n\t} else {\n\t\tmetricsScope.IncCounter(metrics.PersistenceRequests)\n\t}\n\tbefore := time.Now()\n\terr := op()\n\tduration := time.Since(before)\n\tif len(tags) > 0 {\n\t\tmetricsScope.RecordTimer(metrics.PersistenceLatencyPerDomain, duration)\n\t} else {\n\t\tmetricsScope.RecordTimer(metrics.PersistenceLatency, duration)\n\t}\n\n\tif p.enableLatencyHistogramMetrics {\n\t\tmetricsScope.RecordHistogramDuration(metrics.PersistenceLatencyHistogram, duration)\n\t}\n\n\tlogger := p.logger.Helper()\n\tif err != nil {\n\t\tif len(tags) > 0 {\n\t\t\tp.updateErrorMetricPerDomain(scope, err, metricsScope, logger)\n\t\t} else {\n\t\t\tp.updateErrorMetric(scope, err, metricsScope, logger)\n\t\t}\n\t}\n\treturn err\n}\n\nfunc (p *base) callWithoutDomainTag(scope metrics.ScopeIdx, op func() error, tags ...metrics.Tag) error {\n\tmetricsScope := p.metricClient.Scope(scope, tags...)\n\tmetricsScope.IncCounter(metrics.PersistenceRequests)\n\tbefore := time.Now()\n\terr := op()\n\tduration := time.Since(before)\n\tmetricsScope.RecordTimer(metrics.PersistenceLatency, duration)\n\n\tif p.enableLatencyHistogramMetrics {\n\t\tmetricsScope.RecordHistogramDuration(metrics.PersistenceLatencyHistogram, duration)\n\t}\n\tif err != nil {\n\t\tp.updateErrorMetric(scope, err, metricsScope, p.logger.Helper())\n\t}\n\treturn err\n}\n\nfunc (p *base) callWithDomainAndShardScope(scope metrics.ScopeIdx, op func() error, domainTag metrics.Tag, shardIDTag metrics.Tag, additionalTags ...metrics.Tag) error {\n\tdomainMetricsScope := p.metricClient.Scope(scope, append([]metrics.Tag{domainTag}, additionalTags...)...)\n\tshardOperationsMetricsScope := p.metricClient.Scope(scope, append([]metrics.Tag{shardIDTag}, additionalTags...)...)\n\tshardOverallMetricsScope := p.metricClient.Scope(metrics.PersistenceShardRequestCountScope, shardIDTag)\n\n\tdomainMetricsScope.IncCounter(metrics.PersistenceRequestsPerDomain)\n\tshardOperationsMetricsScope.IncCounter(metrics.PersistenceRequestsPerShard)\n\tshardOverallMetricsScope.IncCounter(metrics.PersistenceRequestsPerShard)\n\n\tbefore := time.Now()\n\terr := op()\n\tduration := time.Since(before)\n\n\tdomainMetricsScope.RecordTimer(metrics.PersistenceLatencyPerDomain, duration)\n\tshardOperationsMetricsScope.RecordTimer(metrics.PersistenceLatencyPerShard, duration)\n\tshardOverallMetricsScope.RecordTimer(metrics.PersistenceLatencyPerShard, duration)\n\n\tif p.enableLatencyHistogramMetrics {\n\t\tdomainMetricsScope.RecordHistogramDuration(metrics.PersistenceLatencyHistogram, duration)\n\t}\n\tif err != nil {\n\t\tp.updateErrorMetricPerDomain(scope, err, domainMetricsScope, p.logger.Helper())\n\t}\n\treturn err\n}\n\ntype lengther interface {\n\tLen() int\n}\n\ntype sizer interface {\n\tByteSize() uint64\n}\n\ntype taggedRequest interface {\n\tMetricTags() []metrics.Tag\n}\n\ntype extraLogRequest interface {\n\tGetExtraLogTags() []tag.Tag\n}\n\nfunc (p *base) emitRowCountMetrics(methodName string, req any, res any) {\n\tscope, ok := emptyCountedMethods[methodName]\n\tif !ok {\n\t\t// Method is not counted as empty.\n\t\treturn\n\t}\n\n\tresLen, ok := res.(lengther)\n\tif !ok {\n\t\treturn\n\t}\n\n\tmetricScope := p.metricClient.Scope(scope.scope, getCustomMetricTags(req)...)\n\n\tif resLen.Len() == 0 {\n\t\tmetricScope.IncCounter(metrics.PersistenceEmptyResponseCounter)\n\t} else {\n\t\tmetricScope.RecordHistogramValue(metrics.PersistenceResponseRowSize, float64(resLen.Len()))\n\t}\n}\n\nfunc (p *base) emitPayloadSizeMetrics(methodName string, req any, res any) {\n\tscope, ok := payloadSizeEmittingMethods[methodName]\n\tif !ok {\n\t\treturn\n\t}\n\n\tresSize, ok := res.(sizer)\n\tif !ok {\n\t\treturn\n\t}\n\n\tmetricScope := p.metricClient.Scope(scope.scope, getCustomMetricTags(req)...)\n\tmetricScope.RecordHistogramValue(metrics.PersistenceResponsePayloadSize, float64(resSize.ByteSize()))\n}\n\nfunc (p *base) emptyMetric(methodName string, req any, res any, err error) {\n\tif err != nil {\n\t\treturn\n\t}\n\n\tp.emitRowCountMetrics(methodName, req, res)\n\tp.emitPayloadSizeMetrics(methodName, req, res)\n}\n\nvar emptyCountedMethods = map[string]struct {\n\tscope metrics.ScopeIdx\n}{\n\t\"ExecutionManager.ListCurrentExecutions\": {\n\t\tscope: metrics.PersistenceListCurrentExecutionsScope,\n\t},\n\t\"ExecutionManager.GetReplicationTasksFromDLQ\": {\n\t\tscope: metrics.PersistenceGetReplicationTasksFromDLQScope,\n\t},\n\t\"ExecutionManager.GetHistoryTasks\": {\n\t\tscope: metrics.PersistenceGetHistoryTasksScope,\n\t},\n\t\"TaskManager.GetTasks\": {\n\t\tscope: metrics.PersistenceGetTasksScope,\n\t},\n\t\"DomainManager.ListDomains\": {\n\t\tscope: metrics.PersistenceListDomainsScope,\n\t},\n\t\"HistoryManager.ReadHistoryBranch\": {\n\t\tscope: metrics.PersistenceReadHistoryBranchScope,\n\t},\n\t\"HistoryManager.GetAllHistoryTreeBranches\": {\n\t\tscope: metrics.PersistenceGetAllHistoryTreeBranchesScope,\n\t},\n\t\"QueueManager.ReadMessages\": {\n\t\tscope: metrics.PersistenceReadMessagesScope,\n\t},\n}\n\nvar payloadSizeEmittingMethods = map[string]struct {\n\tscope metrics.ScopeIdx\n}{\n\t\"ExecutionManager.ListCurrentExecutions\": {\n\t\tscope: metrics.PersistenceListCurrentExecutionsScope,\n\t},\n\t\"ExecutionManager.GetReplicationTasksFromDLQ\": {\n\t\tscope: metrics.PersistenceGetReplicationTasksFromDLQScope,\n\t},\n\t\"ExecutionManager.GetHistoryTasks\": {\n\t\tscope: metrics.PersistenceGetHistoryTasksScope,\n\t},\n\t\"TaskManager.GetTasks\": {\n\t\tscope: metrics.PersistenceGetTasksScope,\n\t},\n\t\"DomainManager.ListDomains\": {\n\t\tscope: metrics.PersistenceListDomainsScope,\n\t},\n\t\"HistoryManager.ReadRawHistoryBranch\": {\n\t\tscope: metrics.PersistenceReadRawHistoryBranchScope,\n\t},\n\t\"HistoryManager.GetAllHistoryTreeBranches\": {\n\t\tscope: metrics.PersistenceGetAllHistoryTreeBranchesScope,\n\t},\n\t\"QueueManager.ReadMessages\": {\n\t\tscope: metrics.PersistenceReadMessagesScope,\n\t},\n}\n\ntype domainTaggedRequest interface {\n\tGetDomainName() string\n}\n\nfunc getDomainNameFromRequest(req any) (res string, check bool) {\n\td, check := req.(domainTaggedRequest)\n\tif check {\n\t\tres = d.GetDomainName()\n\t}\n\treturn res, check\n}\n\nfunc getCustomLogTags(req any) (res []tag.Tag) {\n\td, check := req.(extraLogRequest)\n\tif check {\n\t\tres = d.GetExtraLogTags()\n\t}\n\treturn res\n}\n\nfunc getCustomMetricTags(req any) (res []metrics.Tag) {\n\td, check := req.(taggedRequest)\n\tif check {\n\t\tres = d.MetricTags()\n\t}\n\treturn res\n}\n\nfunc getRetryCountFromContext(ctx context.Context) int {\n\tif retryCount, ok := ctx.Value(retryCountKey).(int); ok {\n\t\treturn retryCount\n\t}\n\treturn 0\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/configstore_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredConfigStoreManager implements persistence.ConfigStoreManager interface instrumented with rate limiter.\ntype meteredConfigStoreManager struct {\n\tbase\n\twrapped persistence.ConfigStoreManager\n}\n\n// NewConfigStoreManager creates a new instance of ConfigStoreManager with ratelimiter.\nfunc NewConfigStoreManager(\n\twrapped persistence.ConfigStoreManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.ConfigStoreManager {\n\treturn &meteredConfigStoreManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredConfigStoreManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredConfigStoreManager) FetchDynamicConfig(ctx context.Context, cfgType persistence.ConfigType) (fp1 *persistence.FetchDynamicConfigResponse, err error) {\n\top := func() error {\n\t\tfp1, err = c.wrapped.FetchDynamicConfig(ctx, cfgType)\n\t\tc.emptyMetric(\"ConfigStoreManager.FetchDynamicConfig\", cfgType, fp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceFetchDynamicConfigScope, op, getCustomMetricTags(cfgType)...)\n\treturn\n}\n\nfunc (c *meteredConfigStoreManager) UpdateDynamicConfig(ctx context.Context, request *persistence.UpdateDynamicConfigRequest, cfgType persistence.ConfigType) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpdateDynamicConfig(ctx, request, cfgType)\n\t\tc.emptyMetric(\"ConfigStoreManager.UpdateDynamicConfig\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateDynamicConfigScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/domain_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredDomainManager implements persistence.DomainManager interface instrumented with rate limiter.\ntype meteredDomainManager struct {\n\tbase\n\twrapped persistence.DomainManager\n}\n\n// NewDomainManager creates a new instance of DomainManager with ratelimiter.\nfunc NewDomainManager(\n\twrapped persistence.DomainManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.DomainManager {\n\treturn &meteredDomainManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredDomainManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredDomainManager) CreateDomain(ctx context.Context, request *persistence.CreateDomainRequest) (cp1 *persistence.CreateDomainResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.CreateDomain(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.CreateDomain\", request, cp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCreateDomainScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredDomainManager) DeleteDomain(ctx context.Context, request *persistence.DeleteDomainRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteDomain(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.DeleteDomain\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteDomainScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredDomainManager) DeleteDomainByName(ctx context.Context, request *persistence.DeleteDomainByNameRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteDomainByName(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.DeleteDomainByName\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteDomainByNameScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredDomainManager) GetDomain(ctx context.Context, request *persistence.GetDomainRequest) (gp1 *persistence.GetDomainResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetDomain(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.GetDomain\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetDomainScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredDomainManager) GetMetadata(ctx context.Context) (gp1 *persistence.GetMetadataResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetMetadata(ctx)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetMetadataScope, op)\n\treturn\n}\n\nfunc (c *meteredDomainManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredDomainManager) ListDomains(ctx context.Context, request *persistence.ListDomainsRequest) (lp1 *persistence.ListDomainsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListDomains(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.ListDomains\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListDomainsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredDomainManager) UpdateDomain(ctx context.Context, request *persistence.UpdateDomainRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpdateDomain(ctx, request)\n\t\tc.emptyMetric(\"DomainManager.UpdateDomain\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateDomainScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/execution_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered_execution.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// meteredExecutionManager implements persistence.ExecutionManager interface instrumented with rate limiter.\ntype meteredExecutionManager struct {\n\tbase\n\twrapped persistence.ExecutionManager\n}\n\n// NewExecutionManager creates a new instance of ExecutionManager with ratelimiter.\nfunc NewExecutionManager(\n\twrapped persistence.ExecutionManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n\tsampleLoggingRate dynamicproperties.IntPropertyFn,\n\tenableShardIDMetrics dynamicproperties.BoolPropertyFn,\n) persistence.ExecutionManager {\n\treturn &meteredExecutionManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger.WithTags(tag.ShardID(wrapped.GetShardID())),\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t\tsampleLoggingRate:             sampleLoggingRate,\n\t\t\tenableShardIDMetrics:          enableShardIDMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredExecutionManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredExecutionManager) CompleteHistoryTask(ctx context.Context, request *persistence.CompleteHistoryTaskRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.CompleteHistoryTask(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence CompleteHistoryTask called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceCompleteHistoryTaskScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceCompleteHistoryTaskScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceCompleteHistoryTaskScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) ConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (cp1 *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.ConflictResolveWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.ConflictResolveWorkflowExecution\", request, cp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence ConflictResolveWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceConflictResolveWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceConflictResolveWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceConflictResolveWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) CreateFailoverMarkerTasks(ctx context.Context, request *persistence.CreateFailoverMarkersRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.CreateFailoverMarkerTasks(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence CreateFailoverMarkerTasks called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceCreateFailoverMarkerTasksScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceCreateFailoverMarkerTasksScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceCreateFailoverMarkerTasksScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) CreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (cp1 *persistence.CreateWorkflowExecutionResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.CreateWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.CreateWorkflowExecution\", request, cp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence CreateWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceCreateWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceCreateWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceCreateWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID string, workflowID string, runID string) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(domainID); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(domainID)...)\n\t\tc.logger.SampleInfo(\"Persistence DeleteActiveClusterSelectionPolicy called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceDeleteActiveClusterSelectionPolicyScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceDeleteActiveClusterSelectionPolicyScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceDeleteActiveClusterSelectionPolicyScope, op, append(getCustomMetricTags(domainID), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) DeleteCurrentWorkflowExecution(ctx context.Context, request *persistence.DeleteCurrentWorkflowExecutionRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteCurrentWorkflowExecution(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence DeleteCurrentWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceDeleteCurrentWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceDeleteCurrentWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceDeleteCurrentWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) DeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.DeleteReplicationTaskFromDLQRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteReplicationTaskFromDLQ(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence DeleteReplicationTaskFromDLQ called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceDeleteReplicationTaskFromDLQScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceDeleteReplicationTaskFromDLQScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceDeleteReplicationTaskFromDLQScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.DeleteWorkflowExecutionRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteWorkflowExecution(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence DeleteWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceDeleteWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceDeleteWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceDeleteWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetActiveClusterSelectionPolicy(ctx context.Context, domainID string, wfID string, rID string) (ap1 *types.ActiveClusterSelectionPolicy, err error) {\n\top := func() error {\n\t\tap1, err = c.wrapped.GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID)\n\t\tc.emptyMetric(\"ExecutionManager.GetActiveClusterSelectionPolicy\", domainID, ap1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(domainID); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(domainID)...)\n\t\tc.logger.SampleInfo(\"Persistence GetActiveClusterSelectionPolicy called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetActiveClusterSelectionPolicyScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetActiveClusterSelectionPolicyScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetActiveClusterSelectionPolicyScope, op, append(getCustomMetricTags(domainID), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetCurrentExecution(ctx context.Context, request *persistence.GetCurrentExecutionRequest) (gp1 *persistence.GetCurrentExecutionResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetCurrentExecution(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.GetCurrentExecution\", request, gp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence GetCurrentExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetCurrentExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetCurrentExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetCurrentExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetHistoryTasks(ctx context.Context, request *persistence.GetHistoryTasksRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetHistoryTasks(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.GetHistoryTasks\", request, gp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence GetHistoryTasks called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetHistoryTasksScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetHistoryTasksScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetHistoryTasksScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredExecutionManager) GetReplicationDLQSize(ctx context.Context, request *persistence.GetReplicationDLQSizeRequest) (gp1 *persistence.GetReplicationDLQSizeResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetReplicationDLQSize(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.GetReplicationDLQSize\", request, gp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence GetReplicationDLQSize called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetReplicationDLQSizeScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetReplicationDLQSizeScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetReplicationDLQSizeScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetReplicationTasksFromDLQ(ctx context.Context, request *persistence.GetReplicationTasksFromDLQRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetReplicationTasksFromDLQ(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.GetReplicationTasksFromDLQ\", request, gp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence GetReplicationTasksFromDLQ called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetReplicationTasksFromDLQScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetReplicationTasksFromDLQScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetReplicationTasksFromDLQScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) GetShardID() (i1 int) {\n\treturn c.wrapped.GetShardID()\n}\n\nfunc (c *meteredExecutionManager) GetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (gp1 *persistence.GetWorkflowExecutionResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.GetWorkflowExecution\", request, gp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence GetWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceGetWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceGetWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceGetWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) IsWorkflowExecutionExists(ctx context.Context, request *persistence.IsWorkflowExecutionExistsRequest) (ip1 *persistence.IsWorkflowExecutionExistsResponse, err error) {\n\top := func() error {\n\t\tip1, err = c.wrapped.IsWorkflowExecutionExists(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.IsWorkflowExecutionExists\", request, ip1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence IsWorkflowExecutionExists called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceIsWorkflowExecutionExistsScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceIsWorkflowExecutionExistsScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceIsWorkflowExecutionExistsScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) ListConcreteExecutions(ctx context.Context, request *persistence.ListConcreteExecutionsRequest) (lp1 *persistence.ListConcreteExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListConcreteExecutions(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.ListConcreteExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence ListConcreteExecutions called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceListConcreteExecutionsScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceListConcreteExecutionsScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceListConcreteExecutionsScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) ListCurrentExecutions(ctx context.Context, request *persistence.ListCurrentExecutionsRequest) (lp1 *persistence.ListCurrentExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListCurrentExecutions(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.ListCurrentExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence ListCurrentExecutions called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceListCurrentExecutionsScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceListCurrentExecutionsScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceListCurrentExecutionsScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) PutReplicationTaskToDLQ(ctx context.Context, request *persistence.PutReplicationTaskToDLQRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.PutReplicationTaskToDLQ(ctx, request)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence PutReplicationTaskToDLQ called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistencePutReplicationTaskToDLQScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistencePutReplicationTaskToDLQScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistencePutReplicationTaskToDLQScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) RangeCompleteHistoryTask(ctx context.Context, request *persistence.RangeCompleteHistoryTaskRequest) (rp1 *persistence.RangeCompleteHistoryTaskResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.RangeCompleteHistoryTask(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.RangeCompleteHistoryTask\", request, rp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence RangeCompleteHistoryTask called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceRangeCompleteHistoryTaskScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceRangeCompleteHistoryTaskScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceRangeCompleteHistoryTaskScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.RangeDeleteReplicationTaskFromDLQRequest) (rp1 *persistence.RangeDeleteReplicationTaskFromDLQResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.RangeDeleteReplicationTaskFromDLQ(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.RangeDeleteReplicationTaskFromDLQ\", request, rp1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence RangeDeleteReplicationTaskFromDLQ called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceRangeDeleteReplicationTaskFromDLQScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceRangeDeleteReplicationTaskFromDLQScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceRangeDeleteReplicationTaskFromDLQScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n\nfunc (c *meteredExecutionManager) UpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (up1 *persistence.UpdateWorkflowExecutionResponse, err error) {\n\top := func() error {\n\t\tup1, err = c.wrapped.UpdateWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"ExecutionManager.UpdateWorkflowExecution\", request, up1, err)\n\t\treturn err\n\t}\n\n\tretryCount := getRetryCountFromContext(ctx)\n\tif domainName, hasDomainName := getDomainNameFromRequest(request); hasDomainName {\n\t\tlogTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags(request)...)\n\t\tc.logger.SampleInfo(\"Persistence UpdateWorkflowExecution called\", c.sampleLoggingRate(), logTags...)\n\t\tif c.enableShardIDMetrics() {\n\t\t\terr = c.callWithDomainAndShardScope(metrics.PersistenceUpdateWorkflowExecutionScope, op, metrics.DomainTag(domainName),\n\t\t\t\tmetrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n\t\t} else {\n\t\t\terr = c.call(metrics.PersistenceUpdateWorkflowExecutionScope, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n\t\t}\n\t\treturn\n\t}\n\n\terr = c.callWithoutDomainTag(metrics.PersistenceUpdateWorkflowExecutionScope, op, append(getCustomMetricTags(request), metrics.IsRetryTag(retryCount > 0))...)\n\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/history_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredHistoryManager implements persistence.HistoryManager interface instrumented with rate limiter.\ntype meteredHistoryManager struct {\n\tbase\n\twrapped persistence.HistoryManager\n}\n\n// NewHistoryManager creates a new instance of HistoryManager with ratelimiter.\nfunc NewHistoryManager(\n\twrapped persistence.HistoryManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.HistoryManager {\n\treturn &meteredHistoryManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredHistoryManager) AppendHistoryNodes(ctx context.Context, request *persistence.AppendHistoryNodesRequest) (ap1 *persistence.AppendHistoryNodesResponse, err error) {\n\top := func() error {\n\t\tap1, err = c.wrapped.AppendHistoryNodes(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.AppendHistoryNodes\", request, ap1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceAppendHistoryNodesScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredHistoryManager) DeleteHistoryBranch(ctx context.Context, request *persistence.DeleteHistoryBranchRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteHistoryBranch(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.DeleteHistoryBranch\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteHistoryBranchScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) ForkHistoryBranch(ctx context.Context, request *persistence.ForkHistoryBranchRequest) (fp1 *persistence.ForkHistoryBranchResponse, err error) {\n\top := func() error {\n\t\tfp1, err = c.wrapped.ForkHistoryBranch(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.ForkHistoryBranch\", request, fp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceForkHistoryBranchScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) GetAllHistoryTreeBranches(ctx context.Context, request *persistence.GetAllHistoryTreeBranchesRequest) (gp1 *persistence.GetAllHistoryTreeBranchesResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetAllHistoryTreeBranches(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.GetAllHistoryTreeBranches\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetAllHistoryTreeBranchesScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) GetHistoryTree(ctx context.Context, request *persistence.GetHistoryTreeRequest) (gp1 *persistence.GetHistoryTreeResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetHistoryTree(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.GetHistoryTree\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetHistoryTreeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredHistoryManager) ReadHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.ReadHistoryBranch(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.ReadHistoryBranch\", request, rp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceReadHistoryBranchScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) ReadHistoryBranchByBatch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchByBatchResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.ReadHistoryBranchByBatch(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.ReadHistoryBranchByBatch\", request, rp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceReadHistoryBranchByBatchScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredHistoryManager) ReadRawHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadRawHistoryBranchResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.ReadRawHistoryBranch(ctx, request)\n\t\tc.emptyMetric(\"HistoryManager.ReadRawHistoryBranch\", request, rp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceReadRawHistoryBranchScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/metered_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _staticMethods = map[string]bool{\n\t\"Close\":      true,\n\t\"GetName\":    true,\n\t\"GetShardID\": true,\n}\n\nfunc TestWrappersAgainstPreviousImplementation(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname        string\n\t\tprepareMock func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any)\n\t}{\n\t\t{\n\t\t\tname: \"ConfigStoreManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockConfigStoreManager(ctrl)\n\n\t\t\t\tnewObj := NewConfigStoreManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"DomainManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockDomainManager(ctrl)\n\n\t\t\t\tnewObj := NewDomainManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"HistoryManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockHistoryManager(ctrl)\n\n\t\t\t\tnewObj := NewHistoryManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"QueueManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockQueueManager(ctrl)\n\n\t\t\t\tnewObj := NewQueueManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ShardManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockShardManager(ctrl)\n\n\t\t\t\tnewObj := NewShardManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TaskManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockTaskManager(ctrl)\n\n\t\t\t\tnewObj := NewTaskManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"VisibilityManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockVisibilityManager(ctrl)\n\n\t\t\t\tnewObj := NewVisibilityManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true})\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ExecutionManager\",\n\t\t\tprepareMock: func(t *testing.T, ctrl *gomock.Controller, newMetricsClient metrics.Client, newLogger log.Logger) (newManager any, mocked any) {\n\t\t\t\twrapped := persistence.NewMockExecutionManager(ctrl)\n\n\t\t\t\twrapped.EXPECT().GetShardID().Return(0).AnyTimes()\n\n\t\t\t\tnewObj := NewExecutionManager(wrapped, newMetricsClient, newLogger, &config.Persistence{EnablePersistenceLatencyHistogramMetrics: true},\n\t\t\t\t\tdynamicproperties.GetIntPropertyFn(1), dynamicproperties.GetBoolPropertyFn(true))\n\n\t\t\t\treturn newObj, wrapped\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tt.Run(\"without error\", func(t *testing.T) {\n\t\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t\tzapLogger, logs := setupLogsCapture()\n\t\t\t\tmetricScope := tally.NewTestScope(\"\", nil)\n\t\t\t\tmetricsClient := metrics.NewClient(metricScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\t\t\t\tlogger := log.NewLogger(zapLogger)\n\n\t\t\t\twrapper, mocked := tc.prepareMock(t, ctrl, metricsClient, logger)\n\t\t\t\tprepareMockForTest(t, mocked, nil)\n\n\t\t\t\trunScenario(t, wrapper, logs, metricScope)\n\t\t\t})\n\t\t\tt.Run(\"with error\", func(t *testing.T) {\n\t\t\t\tfor _, errorType := range []error{\n\t\t\t\t\t&types.DomainAlreadyExistsError{},\n\t\t\t\t\t&types.BadRequestError{},\n\t\t\t\t\t&types.EntityNotExistsError{},\n\t\t\t\t\t&types.ServiceBusyError{},\n\t\t\t\t\t&persistence.WorkflowExecutionAlreadyStartedError{},\n\t\t\t\t\t&persistence.ConditionFailedError{},\n\t\t\t\t\t&persistence.CurrentWorkflowConditionFailedError{},\n\t\t\t\t\t&persistence.ShardAlreadyExistError{},\n\t\t\t\t\t&persistence.ShardOwnershipLostError{},\n\t\t\t\t\t&persistence.DuplicateRequestError{},\n\t\t\t\t\t&persistence.TimeoutError{},\n\t\t\t\t\t&persistence.DBUnavailableError{},\n\t\t\t\t\terrors.New(\"persistence error\"),\n\t\t\t\t} {\n\t\t\t\t\tt.Run(reflect.TypeOf(errorType).String(), func(t *testing.T) {\n\t\t\t\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t\t\t\tzapLogger, logs := setupLogsCapture()\n\t\t\t\t\t\tmetricScope := tally.NewTestScope(\"\", nil)\n\t\t\t\t\t\tmetricsClient := metrics.NewClient(metricScope, metrics.ServiceIdx(0), metrics.HistogramMigration{})\n\t\t\t\t\t\tlogger := log.NewLogger(zapLogger)\n\n\t\t\t\t\t\tnewObj, mocked := tc.prepareMock(t, ctrl, metricsClient, logger)\n\t\t\t\t\t\tprepareMockForTest(t, mocked, errorType)\n\n\t\t\t\t\t\trunScenario(t, newObj, logs, metricScope)\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc prepareMockForTest(t *testing.T, input interface{}, expectedErr error) {\n\tswitch mocked := input.(type) {\n\tcase *persistence.MockConfigStoreManager:\n\t\tmocked.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().FetchDynamicConfig(gomock.Any(), gomock.Any()).Return(&persistence.FetchDynamicConfigResponse{}, expectedErr).Times(1)\n\tcase *persistence.MockDomainManager:\n\t\tmocked.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteDomainByName(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(&persistence.ListDomainsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{}, expectedErr).Times(1)\n\tcase *persistence.MockHistoryManager:\n\t\tmocked.EXPECT().AppendHistoryNodes(gomock.Any(), gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ReadHistoryBranchByBatch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchByBatchResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ReadRawHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadRawHistoryBranchResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ForkHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ForkHistoryBranchResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteHistoryBranch(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetHistoryTree(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTreeResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetAllHistoryTreeBranches(gomock.Any(), gomock.Any()).Return(&persistence.GetAllHistoryTreeBranchesResponse{}, expectedErr).Times(1)\n\tcase *persistence.MockQueueManager:\n\t\tmocked.EXPECT().EnqueueMessage(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().ReadMessages(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesResponse{Messages: []*persistence.QueueMessage{}}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteMessagesBefore(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().EnqueueMessageToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetDLQAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQSizeResponse{Size: 0}, expectedErr).Times(1)\n\t\tmocked.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().ReadMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesFromDLQResponse{Messages: []*persistence.QueueMessage{}}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\tcase *persistence.MockShardManager:\n\t\tmocked.EXPECT().GetShard(gomock.Any(), gomock.Any()).Return(&persistence.GetShardResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateShard(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().CreateShard(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\tcase *persistence.MockTaskManager:\n\t\tmocked.EXPECT().CompleteTasksLessThan(gomock.Any(), gomock.Any()).Return(&persistence.CompleteTasksLessThanResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().CompleteTask(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().CreateTasks(gomock.Any(), gomock.Any()).Return(&persistence.CreateTasksResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteTaskList(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetTasksResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetOrphanTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetOrphanTasksResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetTaskListSize(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListSizeResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().LeaseTaskList(gomock.Any(), gomock.Any()).Return(&persistence.LeaseTaskListResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetTaskList(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListTaskList(gomock.Any(), gomock.Any()).Return(&persistence.ListTaskListResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any()).Return(&persistence.UpdateTaskListResponse{}, expectedErr).Times(1)\n\tcase *persistence.MockVisibilityManager:\n\t\tmocked.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.CountWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetClosedWorkflowExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr).Times(1)\n\tcase *persistence.MockExecutionManager:\n\t\tmocked.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.CreateWorkflowExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetWorkflowExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetCurrentExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.ConflictResolveWorkflowExecutionResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().CreateFailoverMarkerTasks(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetReplicationDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetReplicationDLQSizeResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetReplicationTasksFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Any()).Return(&persistence.IsWorkflowExecutionExistsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListConcreteExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListCurrentExecutionsResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().PutReplicationTaskToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\t\tmocked.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(&persistence.RangeCompleteHistoryTaskResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().RangeDeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.RangeDeleteReplicationTaskFromDLQResponse{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterSelectionPolicy{}, expectedErr).Times(1)\n\t\tmocked.EXPECT().DeleteActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr).Times(1)\n\tdefault:\n\t\tt.Errorf(\"unsupported type %v\", reflect.TypeOf(input))\n\t\tt.FailNow()\n\t}\n\treturn\n}\n\nfunc setupLogsCapture() (*zap.Logger, *observer.ObservedLogs) {\n\tcore, logs := observer.New(zap.InfoLevel)\n\treturn zap.New(core), logs\n}\n\nfunc runScenario(t *testing.T, newObj any, newLogs *observer.ObservedLogs, newMetrics tally.TestScope) {\n\tnewV := reflect.ValueOf(newObj)\n\tinfoT := reflect.TypeOf(newV.Interface())\n\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\tmethod := infoT.Method(i)\n\t\tif _staticMethods[method.Name] {\n\t\t\t// Skip methods that do not use error injection.\n\t\t\tcontinue\n\t\t}\n\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t// First argument is always context.Context\n\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\tif method.Type.In(i).Kind() == reflect.Ptr {\n\t\t\t\t\tvals = append(vals, reflect.New(method.Type.In(i).Elem()))\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t}\n\n\t\t\tvar newRes []reflect.Value\n\t\t\tassert.NotPanicsf(t, func() {\n\t\t\t\tnewRes = newV.MethodByName(method.Name).Call(vals)\n\t\t\t}, \"method does not have tag defined\")\n\n\t\t\tif len(newRes) == 0 {\n\t\t\t\t// Empty result means that method panicked.\n\t\t\t\treturn\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/queue_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredQueueManager implements persistence.QueueManager interface instrumented with rate limiter.\ntype meteredQueueManager struct {\n\tbase\n\twrapped persistence.QueueManager\n}\n\n// NewQueueManager creates a new instance of QueueManager with ratelimiter.\nfunc NewQueueManager(\n\twrapped persistence.QueueManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.QueueManager {\n\treturn &meteredQueueManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredQueueManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredQueueManager) DeleteMessageFromDLQ(ctx context.Context, request *persistence.DeleteMessageFromDLQRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteMessageFromDLQ(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.DeleteMessageFromDLQ\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteMessageFromDLQScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) DeleteMessagesBefore(ctx context.Context, request *persistence.DeleteMessagesBeforeRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteMessagesBefore(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.DeleteMessagesBefore\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteMessagesBeforeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) EnqueueMessage(ctx context.Context, request *persistence.EnqueueMessageRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.EnqueueMessage(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.EnqueueMessage\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceEnqueueMessageScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) EnqueueMessageToDLQ(ctx context.Context, request *persistence.EnqueueMessageToDLQRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.EnqueueMessageToDLQ(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.EnqueueMessageToDLQ\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceEnqueueMessageToDLQScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) GetAckLevels(ctx context.Context, request *persistence.GetAckLevelsRequest) (gp1 *persistence.GetAckLevelsResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetAckLevels(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.GetAckLevels\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetAckLevelsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) GetDLQAckLevels(ctx context.Context, request *persistence.GetDLQAckLevelsRequest) (gp1 *persistence.GetDLQAckLevelsResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetDLQAckLevels(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.GetDLQAckLevels\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetDLQAckLevelsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) GetDLQSize(ctx context.Context, request *persistence.GetDLQSizeRequest) (gp1 *persistence.GetDLQSizeResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetDLQSize(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.GetDLQSize\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetDLQSizeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) RangeDeleteMessagesFromDLQ(ctx context.Context, request *persistence.RangeDeleteMessagesFromDLQRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.RangeDeleteMessagesFromDLQ(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.RangeDeleteMessagesFromDLQ\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceRangeDeleteMessagesFromDLQScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) ReadMessages(ctx context.Context, request *persistence.ReadMessagesRequest) (rp1 *persistence.ReadMessagesResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.ReadMessages(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.ReadMessages\", request, rp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceReadMessagesScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) ReadMessagesFromDLQ(ctx context.Context, request *persistence.ReadMessagesFromDLQRequest) (rp1 *persistence.ReadMessagesFromDLQResponse, err error) {\n\top := func() error {\n\t\trp1, err = c.wrapped.ReadMessagesFromDLQ(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.ReadMessagesFromDLQ\", request, rp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceReadMessagesFromDLQScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) UpdateAckLevel(ctx context.Context, request *persistence.UpdateAckLevelRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpdateAckLevel(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.UpdateAckLevel\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateAckLevelScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredQueueManager) UpdateDLQAckLevel(ctx context.Context, request *persistence.UpdateDLQAckLevelRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpdateDLQAckLevel(ctx, request)\n\t\tc.emptyMetric(\"QueueManager.UpdateDLQAckLevel\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateDLQAckLevelScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/shard_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredShardManager implements persistence.ShardManager interface instrumented with rate limiter.\ntype meteredShardManager struct {\n\tbase\n\twrapped persistence.ShardManager\n}\n\n// NewShardManager creates a new instance of ShardManager with ratelimiter.\nfunc NewShardManager(\n\twrapped persistence.ShardManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.ShardManager {\n\treturn &meteredShardManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredShardManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredShardManager) CreateShard(ctx context.Context, request *persistence.CreateShardRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.CreateShard(ctx, request)\n\t\tc.emptyMetric(\"ShardManager.CreateShard\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCreateShardScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredShardManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredShardManager) GetShard(ctx context.Context, request *persistence.GetShardRequest) (gp1 *persistence.GetShardResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetShard(ctx, request)\n\t\tc.emptyMetric(\"ShardManager.GetShard\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetShardScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredShardManager) UpdateShard(ctx context.Context, request *persistence.UpdateShardRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpdateShard(ctx, request)\n\t\tc.emptyMetric(\"ShardManager.UpdateShard\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateShardScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/task_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredTaskManager implements persistence.TaskManager interface instrumented with rate limiter.\ntype meteredTaskManager struct {\n\tbase\n\twrapped persistence.TaskManager\n}\n\n// NewTaskManager creates a new instance of TaskManager with ratelimiter.\nfunc NewTaskManager(\n\twrapped persistence.TaskManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.TaskManager {\n\treturn &meteredTaskManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredTaskManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredTaskManager) CompleteTask(ctx context.Context, request *persistence.CompleteTaskRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.CompleteTask(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.CompleteTask\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCompleteTaskScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) CompleteTasksLessThan(ctx context.Context, request *persistence.CompleteTasksLessThanRequest) (cp1 *persistence.CompleteTasksLessThanResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.CompleteTasksLessThan(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.CompleteTasksLessThan\", request, cp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCompleteTasksLessThanScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) CreateTasks(ctx context.Context, request *persistence.CreateTasksRequest) (cp1 *persistence.CreateTasksResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.CreateTasks(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.CreateTasks\", request, cp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCreateTasksScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) DeleteTaskList(ctx context.Context, request *persistence.DeleteTaskListRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteTaskList(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.DeleteTaskList\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteTaskListScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredTaskManager) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (gp1 *persistence.GetOrphanTasksResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetOrphanTasks(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.GetOrphanTasks\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetOrphanTasksScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) GetTaskList(ctx context.Context, request *persistence.GetTaskListRequest) (gp1 *persistence.GetTaskListResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetTaskList(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.GetTaskList\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetTaskListScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (gp1 *persistence.GetTaskListSizeResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetTaskListSize(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.GetTaskListSize\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetTaskListSizeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) GetTasks(ctx context.Context, request *persistence.GetTasksRequest) (gp1 *persistence.GetTasksResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetTasks(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.GetTasks\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetTasksScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) LeaseTaskList(ctx context.Context, request *persistence.LeaseTaskListRequest) (lp1 *persistence.LeaseTaskListResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.LeaseTaskList(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.LeaseTaskList\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceLeaseTaskListScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) ListTaskList(ctx context.Context, request *persistence.ListTaskListRequest) (lp1 *persistence.ListTaskListResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListTaskList(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.ListTaskList\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListTaskListScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredTaskManager) UpdateTaskList(ctx context.Context, request *persistence.UpdateTaskListRequest) (up1 *persistence.UpdateTaskListResponse, err error) {\n\top := func() error {\n\t\tup1, err = c.wrapped.UpdateTaskList(ctx, request)\n\t\tc.emptyMetric(\"TaskManager.UpdateTaskList\", request, up1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpdateTaskListScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/metered/visibility_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// meteredVisibilityManager implements persistence.VisibilityManager interface instrumented with rate limiter.\ntype meteredVisibilityManager struct {\n\tbase\n\twrapped persistence.VisibilityManager\n}\n\n// NewVisibilityManager creates a new instance of VisibilityManager with ratelimiter.\nfunc NewVisibilityManager(\n\twrapped persistence.VisibilityManager,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\tcfg *config.Persistence,\n) persistence.VisibilityManager {\n\treturn &meteredVisibilityManager{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t},\n\t}\n}\n\nfunc (c *meteredVisibilityManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) CountWorkflowExecutions(ctx context.Context, request *persistence.CountWorkflowExecutionsRequest) (cp1 *persistence.CountWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tcp1, err = c.wrapped.CountWorkflowExecutions(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.CountWorkflowExecutions\", request, cp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceCountWorkflowExecutionsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) DeleteUninitializedWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteUninitializedWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.DeleteUninitializedWorkflowExecution\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceDeleteUninitializedWorkflowExecutionScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.DeleteWorkflowExecution\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceVisibilityDeleteWorkflowExecutionScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) GetClosedWorkflowExecution(ctx context.Context, request *persistence.GetClosedWorkflowExecutionRequest) (gp1 *persistence.GetClosedWorkflowExecutionResponse, err error) {\n\top := func() error {\n\t\tgp1, err = c.wrapped.GetClosedWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.GetClosedWorkflowExecution\", request, gp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceGetClosedWorkflowExecutionScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *meteredVisibilityManager) ListClosedWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutions(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListClosedWorkflowExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListClosedWorkflowExecutionsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *persistence.ListClosedWorkflowExecutionsByStatusRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByStatus(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListClosedWorkflowExecutionsByStatus\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListClosedWorkflowExecutionsByStatusScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListClosedWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByType(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListClosedWorkflowExecutionsByType\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListClosedWorkflowExecutionsByTypeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListClosedWorkflowExecutionsByWorkflowID\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListClosedWorkflowExecutionsByWorkflowIDScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListOpenWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutions(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListOpenWorkflowExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListOpenWorkflowExecutionsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListOpenWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutionsByType(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListOpenWorkflowExecutionsByType\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListOpenWorkflowExecutionsByTypeScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListOpenWorkflowExecutionsByWorkflowID\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListOpenWorkflowExecutionsByWorkflowIDScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ListWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ListWorkflowExecutions(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ListWorkflowExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceListWorkflowExecutionsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) RecordWorkflowExecutionClosed(ctx context.Context, request *persistence.RecordWorkflowExecutionClosedRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.RecordWorkflowExecutionClosed(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.RecordWorkflowExecutionClosed\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceRecordWorkflowExecutionClosedScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) RecordWorkflowExecutionStarted(ctx context.Context, request *persistence.RecordWorkflowExecutionStartedRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.RecordWorkflowExecutionStarted(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.RecordWorkflowExecutionStarted\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceRecordWorkflowExecutionStartedScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) RecordWorkflowExecutionUninitialized(ctx context.Context, request *persistence.RecordWorkflowExecutionUninitializedRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.RecordWorkflowExecutionUninitialized(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.RecordWorkflowExecutionUninitialized\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceRecordWorkflowExecutionUninitializedScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) ScanWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\top := func() error {\n\t\tlp1, err = c.wrapped.ScanWorkflowExecutions(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.ScanWorkflowExecutions\", request, lp1, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceScanWorkflowExecutionsScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n\nfunc (c *meteredVisibilityManager) UpsertWorkflowExecution(ctx context.Context, request *persistence.UpsertWorkflowExecutionRequest) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.UpsertWorkflowExecution(ctx, request)\n\t\tc.emptyMetric(\"VisibilityManager.UpsertWorkflowExecution\", request, err, err)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.PersistenceUpsertWorkflowExecutionScope, op, getCustomMetricTags(request)...)\n\treturn\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/configstore_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedConfigStoreManager implements persistence.ConfigStoreManager interface instrumented with rate limiter.\ntype ratelimitedConfigStoreManager struct {\n\twrapped      persistence.ConfigStoreManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewConfigStoreManager creates a new instance of ConfigStoreManager with ratelimiter.\nfunc NewConfigStoreManager(\n\twrapped persistence.ConfigStoreManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.ConfigStoreManager {\n\treturn &ratelimitedConfigStoreManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedConfigStoreManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedConfigStoreManager) FetchDynamicConfig(ctx context.Context, cfgType persistence.ConfigType) (fp1 *persistence.FetchDynamicConfigResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.FetchDynamicConfig(ctx, cfgType)\n}\n\nfunc (c *ratelimitedConfigStoreManager) UpdateDynamicConfig(ctx context.Context, request *persistence.UpdateDynamicConfigRequest, cfgType persistence.ConfigType) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateDynamicConfig(ctx, request, cfgType)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/domain_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedDomainManager implements persistence.DomainManager interface instrumented with rate limiter.\ntype ratelimitedDomainManager struct {\n\twrapped      persistence.DomainManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewDomainManager creates a new instance of DomainManager with ratelimiter.\nfunc NewDomainManager(\n\twrapped persistence.DomainManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.DomainManager {\n\treturn &ratelimitedDomainManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedDomainManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedDomainManager) CreateDomain(ctx context.Context, request *persistence.CreateDomainRequest) (cp1 *persistence.CreateDomainResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CreateDomain(ctx, request)\n}\n\nfunc (c *ratelimitedDomainManager) DeleteDomain(ctx context.Context, request *persistence.DeleteDomainRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteDomain(ctx, request)\n}\n\nfunc (c *ratelimitedDomainManager) DeleteDomainByName(ctx context.Context, request *persistence.DeleteDomainByNameRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteDomainByName(ctx, request)\n}\n\nfunc (c *ratelimitedDomainManager) GetDomain(ctx context.Context, request *persistence.GetDomainRequest) (gp1 *persistence.GetDomainResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetDomain(ctx, request)\n}\n\nfunc (c *ratelimitedDomainManager) GetMetadata(ctx context.Context) (gp1 *persistence.GetMetadataResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetMetadata(ctx)\n}\n\nfunc (c *ratelimitedDomainManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedDomainManager) ListDomains(ctx context.Context, request *persistence.ListDomainsRequest) (lp1 *persistence.ListDomainsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListDomains(ctx, request)\n}\n\nfunc (c *ratelimitedDomainManager) UpdateDomain(ctx context.Context, request *persistence.UpdateDomainRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateDomain(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/errors.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// ErrPersistenceLimitExceeded is the error indicating QPS limit reached.\n\tErrPersistenceLimitExceeded = &types.ServiceBusyError{Message: \"Persistence Max QPS Reached.\"}\n)\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/execution_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ratelimitedExecutionManager implements persistence.ExecutionManager interface instrumented with rate limiter.\ntype ratelimitedExecutionManager struct {\n\twrapped      persistence.ExecutionManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewExecutionManager creates a new instance of ExecutionManager with ratelimiter.\nfunc NewExecutionManager(\n\twrapped persistence.ExecutionManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.ExecutionManager {\n\treturn &ratelimitedExecutionManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedExecutionManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedExecutionManager) CompleteHistoryTask(ctx context.Context, request *persistence.CompleteHistoryTaskRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CompleteHistoryTask(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) ConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (cp1 *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ConflictResolveWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) CreateFailoverMarkerTasks(ctx context.Context, request *persistence.CreateFailoverMarkersRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CreateFailoverMarkerTasks(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) CreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (cp1 *persistence.CreateWorkflowExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CreateWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) DeleteActiveClusterSelectionPolicy(ctx context.Context, domainID string, workflowID string, runID string) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteActiveClusterSelectionPolicy(ctx, domainID, workflowID, runID)\n}\n\nfunc (c *ratelimitedExecutionManager) DeleteCurrentWorkflowExecution(ctx context.Context, request *persistence.DeleteCurrentWorkflowExecutionRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteCurrentWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) DeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.DeleteReplicationTaskFromDLQRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteReplicationTaskFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.DeleteWorkflowExecutionRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) GetActiveClusterSelectionPolicy(ctx context.Context, domainID string, wfID string, rID string) (ap1 *types.ActiveClusterSelectionPolicy, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetActiveClusterSelectionPolicy(ctx, domainID, wfID, rID)\n}\n\nfunc (c *ratelimitedExecutionManager) GetCurrentExecution(ctx context.Context, request *persistence.GetCurrentExecutionRequest) (gp1 *persistence.GetCurrentExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetCurrentExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) GetHistoryTasks(ctx context.Context, request *persistence.GetHistoryTasksRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetHistoryTasks(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedExecutionManager) GetReplicationDLQSize(ctx context.Context, request *persistence.GetReplicationDLQSizeRequest) (gp1 *persistence.GetReplicationDLQSizeResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetReplicationDLQSize(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) GetReplicationTasksFromDLQ(ctx context.Context, request *persistence.GetReplicationTasksFromDLQRequest) (gp1 *persistence.GetHistoryTasksResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetReplicationTasksFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) GetShardID() (i1 int) {\n\treturn c.wrapped.GetShardID()\n}\n\nfunc (c *ratelimitedExecutionManager) GetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (gp1 *persistence.GetWorkflowExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) IsWorkflowExecutionExists(ctx context.Context, request *persistence.IsWorkflowExecutionExistsRequest) (ip1 *persistence.IsWorkflowExecutionExistsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.IsWorkflowExecutionExists(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) ListConcreteExecutions(ctx context.Context, request *persistence.ListConcreteExecutionsRequest) (lp1 *persistence.ListConcreteExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListConcreteExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) ListCurrentExecutions(ctx context.Context, request *persistence.ListCurrentExecutionsRequest) (lp1 *persistence.ListCurrentExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListCurrentExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) PutReplicationTaskToDLQ(ctx context.Context, request *persistence.PutReplicationTaskToDLQRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.PutReplicationTaskToDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) RangeCompleteHistoryTask(ctx context.Context, request *persistence.RangeCompleteHistoryTaskRequest) (rp1 *persistence.RangeCompleteHistoryTaskResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RangeCompleteHistoryTask(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) RangeDeleteReplicationTaskFromDLQ(ctx context.Context, request *persistence.RangeDeleteReplicationTaskFromDLQRequest) (rp1 *persistence.RangeDeleteReplicationTaskFromDLQResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RangeDeleteReplicationTaskFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedExecutionManager) UpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (up1 *persistence.UpdateWorkflowExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateWorkflowExecution(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/history_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedHistoryManager implements persistence.HistoryManager interface instrumented with rate limiter.\ntype ratelimitedHistoryManager struct {\n\twrapped      persistence.HistoryManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewHistoryManager creates a new instance of HistoryManager with ratelimiter.\nfunc NewHistoryManager(\n\twrapped persistence.HistoryManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.HistoryManager {\n\treturn &ratelimitedHistoryManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedHistoryManager) AppendHistoryNodes(ctx context.Context, request *persistence.AppendHistoryNodesRequest) (ap1 *persistence.AppendHistoryNodesResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.AppendHistoryNodes(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedHistoryManager) DeleteHistoryBranch(ctx context.Context, request *persistence.DeleteHistoryBranchRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteHistoryBranch(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) ForkHistoryBranch(ctx context.Context, request *persistence.ForkHistoryBranchRequest) (fp1 *persistence.ForkHistoryBranchResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ForkHistoryBranch(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) GetAllHistoryTreeBranches(ctx context.Context, request *persistence.GetAllHistoryTreeBranchesRequest) (gp1 *persistence.GetAllHistoryTreeBranchesResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetAllHistoryTreeBranches(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) GetHistoryTree(ctx context.Context, request *persistence.GetHistoryTreeRequest) (gp1 *persistence.GetHistoryTreeResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetHistoryTree(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedHistoryManager) ReadHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ReadHistoryBranch(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) ReadHistoryBranchByBatch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadHistoryBranchByBatchResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ReadHistoryBranchByBatch(ctx, request)\n}\n\nfunc (c *ratelimitedHistoryManager) ReadRawHistoryBranch(ctx context.Context, request *persistence.ReadHistoryBranchRequest) (rp1 *persistence.ReadRawHistoryBranchResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ReadRawHistoryBranch(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/queue_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedQueueManager implements persistence.QueueManager interface instrumented with rate limiter.\ntype ratelimitedQueueManager struct {\n\twrapped      persistence.QueueManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewQueueManager creates a new instance of QueueManager with ratelimiter.\nfunc NewQueueManager(\n\twrapped persistence.QueueManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.QueueManager {\n\treturn &ratelimitedQueueManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedQueueManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedQueueManager) DeleteMessageFromDLQ(ctx context.Context, request *persistence.DeleteMessageFromDLQRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteMessageFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) DeleteMessagesBefore(ctx context.Context, request *persistence.DeleteMessagesBeforeRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteMessagesBefore(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) EnqueueMessage(ctx context.Context, request *persistence.EnqueueMessageRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.EnqueueMessage(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) EnqueueMessageToDLQ(ctx context.Context, request *persistence.EnqueueMessageToDLQRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.EnqueueMessageToDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) GetAckLevels(ctx context.Context, request *persistence.GetAckLevelsRequest) (gp1 *persistence.GetAckLevelsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetAckLevels(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) GetDLQAckLevels(ctx context.Context, request *persistence.GetDLQAckLevelsRequest) (gp1 *persistence.GetDLQAckLevelsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetDLQAckLevels(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) GetDLQSize(ctx context.Context, request *persistence.GetDLQSizeRequest) (gp1 *persistence.GetDLQSizeResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetDLQSize(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) RangeDeleteMessagesFromDLQ(ctx context.Context, request *persistence.RangeDeleteMessagesFromDLQRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RangeDeleteMessagesFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) ReadMessages(ctx context.Context, request *persistence.ReadMessagesRequest) (rp1 *persistence.ReadMessagesResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ReadMessages(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) ReadMessagesFromDLQ(ctx context.Context, request *persistence.ReadMessagesFromDLQRequest) (rp1 *persistence.ReadMessagesFromDLQResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ReadMessagesFromDLQ(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) UpdateAckLevel(ctx context.Context, request *persistence.UpdateAckLevelRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateAckLevel(ctx, request)\n}\n\nfunc (c *ratelimitedQueueManager) UpdateDLQAckLevel(ctx context.Context, request *persistence.UpdateDLQAckLevelRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateDLQAckLevel(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/shard_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedShardManager implements persistence.ShardManager interface instrumented with rate limiter.\ntype ratelimitedShardManager struct {\n\twrapped      persistence.ShardManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewShardManager creates a new instance of ShardManager with ratelimiter.\nfunc NewShardManager(\n\twrapped persistence.ShardManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.ShardManager {\n\treturn &ratelimitedShardManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedShardManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedShardManager) CreateShard(ctx context.Context, request *persistence.CreateShardRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CreateShard(ctx, request)\n}\n\nfunc (c *ratelimitedShardManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedShardManager) GetShard(ctx context.Context, request *persistence.GetShardRequest) (gp1 *persistence.GetShardResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetShard(ctx, request)\n}\n\nfunc (c *ratelimitedShardManager) UpdateShard(ctx context.Context, request *persistence.UpdateShardRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateShard(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/task_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedTaskManager implements persistence.TaskManager interface instrumented with rate limiter.\ntype ratelimitedTaskManager struct {\n\twrapped      persistence.TaskManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewTaskManager creates a new instance of TaskManager with ratelimiter.\nfunc NewTaskManager(\n\twrapped persistence.TaskManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.TaskManager {\n\treturn &ratelimitedTaskManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedTaskManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedTaskManager) CompleteTask(ctx context.Context, request *persistence.CompleteTaskRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CompleteTask(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) CompleteTasksLessThan(ctx context.Context, request *persistence.CompleteTasksLessThanRequest) (cp1 *persistence.CompleteTasksLessThanResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CompleteTasksLessThan(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) CreateTasks(ctx context.Context, request *persistence.CreateTasksRequest) (cp1 *persistence.CreateTasksResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CreateTasks(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) DeleteTaskList(ctx context.Context, request *persistence.DeleteTaskListRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteTaskList(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedTaskManager) GetOrphanTasks(ctx context.Context, request *persistence.GetOrphanTasksRequest) (gp1 *persistence.GetOrphanTasksResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetOrphanTasks(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) GetTaskList(ctx context.Context, request *persistence.GetTaskListRequest) (gp1 *persistence.GetTaskListResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetTaskList(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) GetTaskListSize(ctx context.Context, request *persistence.GetTaskListSizeRequest) (gp1 *persistence.GetTaskListSizeResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetTaskListSize(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) GetTasks(ctx context.Context, request *persistence.GetTasksRequest) (gp1 *persistence.GetTasksResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetTasks(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) LeaseTaskList(ctx context.Context, request *persistence.LeaseTaskListRequest) (lp1 *persistence.LeaseTaskListResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.LeaseTaskList(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) ListTaskList(ctx context.Context, request *persistence.ListTaskListRequest) (lp1 *persistence.ListTaskListResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListTaskList(ctx, request)\n}\n\nfunc (c *ratelimitedTaskManager) UpdateTaskList(ctx context.Context, request *persistence.UpdateTaskListRequest) (up1 *persistence.UpdateTaskListResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpdateTaskList(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/utils_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"context\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\ntype limiterAlwaysAllow struct{}\n\nfunc (l limiterAlwaysAllow) Allow() bool {\n\treturn true\n}\n\nfunc (l limiterAlwaysAllow) Wait(ctx context.Context) error {\n\treturn nil\n}\n\nfunc (l limiterAlwaysAllow) Reserve() clock.Reservation {\n\treturn &reservationAlwaysAllow{}\n}\n\nfunc (l limiterAlwaysAllow) Limit() rate.Limit {\n\treturn rate.Inf\n}\n\ntype limiterNeverAllow struct{}\n\nfunc (l limiterNeverAllow) Allow() bool {\n\treturn false\n}\n\nfunc (l limiterNeverAllow) Wait(ctx context.Context) error {\n\t<-ctx.Done()\n\treturn ctx.Err()\n}\n\nfunc (l limiterNeverAllow) Reserve() clock.Reservation {\n\treturn &reservationNeverAllow{}\n}\n\nfunc (l limiterNeverAllow) Limit() rate.Limit {\n\treturn 0\n}\n\ntype reservationAlwaysAllow struct{}\ntype reservationNeverAllow struct{}\n\nfunc (r reservationAlwaysAllow) Allow() bool { return true }\nfunc (r reservationAlwaysAllow) Used(bool)   {}\nfunc (r reservationNeverAllow) Allow() bool  { return false }\nfunc (r reservationNeverAllow) Used(bool)    {}\n\nvar _ quotas.Limiter = (*limiterAlwaysAllow)(nil)\nvar _ quotas.Limiter = (*limiterNeverAllow)(nil)\nvar _ clock.Reservation = (*reservationAlwaysAllow)(nil)\nvar _ clock.Reservation = (*reservationNeverAllow)(nil)\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/visibility_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// ratelimitedVisibilityManager implements persistence.VisibilityManager interface instrumented with rate limiter.\ntype ratelimitedVisibilityManager struct {\n\twrapped      persistence.VisibilityManager\n\trateLimiter  quotas.Limiter\n\tcallerBypass quotas.CallerBypass\n}\n\n// NewVisibilityManager creates a new instance of VisibilityManager with ratelimiter.\nfunc NewVisibilityManager(\n\twrapped persistence.VisibilityManager,\n\trateLimiter quotas.Limiter,\n\tcallerBypass quotas.CallerBypass,\n) persistence.VisibilityManager {\n\treturn &ratelimitedVisibilityManager{\n\t\twrapped:      wrapped,\n\t\trateLimiter:  rateLimiter,\n\t\tcallerBypass: callerBypass,\n\t}\n}\n\nfunc (c *ratelimitedVisibilityManager) Close() {\n\tc.wrapped.Close()\n\treturn\n}\n\nfunc (c *ratelimitedVisibilityManager) CountWorkflowExecutions(ctx context.Context, request *persistence.CountWorkflowExecutionsRequest) (cp1 *persistence.CountWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.CountWorkflowExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) DeleteUninitializedWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteUninitializedWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) DeleteWorkflowExecution(ctx context.Context, request *persistence.VisibilityDeleteWorkflowExecutionRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.DeleteWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) GetClosedWorkflowExecution(ctx context.Context, request *persistence.GetClosedWorkflowExecutionRequest) (gp1 *persistence.GetClosedWorkflowExecutionResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.GetClosedWorkflowExecution(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) GetName() (s1 string) {\n\treturn c.wrapped.GetName()\n}\n\nfunc (c *ratelimitedVisibilityManager) ListClosedWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListClosedWorkflowExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListClosedWorkflowExecutionsByStatus(ctx context.Context, request *persistence.ListClosedWorkflowExecutionsByStatusRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListClosedWorkflowExecutionsByStatus(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListClosedWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListClosedWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListClosedWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListOpenWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListOpenWorkflowExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListOpenWorkflowExecutionsByType(ctx context.Context, request *persistence.ListWorkflowExecutionsByTypeRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListOpenWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListOpenWorkflowExecutionsByWorkflowID(ctx context.Context, request *persistence.ListWorkflowExecutionsByWorkflowIDRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ListWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ListWorkflowExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) RecordWorkflowExecutionClosed(ctx context.Context, request *persistence.RecordWorkflowExecutionClosedRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RecordWorkflowExecutionClosed(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) RecordWorkflowExecutionStarted(ctx context.Context, request *persistence.RecordWorkflowExecutionStartedRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RecordWorkflowExecutionStarted(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) RecordWorkflowExecutionUninitialized(ctx context.Context, request *persistence.RecordWorkflowExecutionUninitializedRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.RecordWorkflowExecutionUninitialized(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) ScanWorkflowExecutions(ctx context.Context, request *persistence.ListWorkflowExecutionsByQueryRequest) (lp1 *persistence.ListWorkflowExecutionsResponse, err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.ScanWorkflowExecutions(ctx, request)\n}\n\nfunc (c *ratelimitedVisibilityManager) UpsertWorkflowExecution(ctx context.Context, request *persistence.UpsertWorkflowExecutionRequest) (err error) {\n\tif !c.callerBypass.AllowLimiter(ctx, c.rateLimiter) {\n\t\terr = ErrPersistenceLimitExceeded\n\t\treturn\n\t}\n\treturn c.wrapped.UpsertWorkflowExecution(ctx, request)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/ratelimited/wrappers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar _staticMethods = map[string]bool{\n\t\"Close\":      true,\n\t\"GetName\":    true,\n\t\"GetShardID\": true,\n}\n\nvar wrappers = []any{\n\t&ratelimitedConfigStoreManager{},\n\t&ratelimitedDomainManager{},\n\t&ratelimitedHistoryManager{},\n\t&ratelimitedQueueManager{},\n\t&ratelimitedShardManager{},\n\t&ratelimitedTaskManager{},\n\t&ratelimitedVisibilityManager{},\n\t&ratelimitedExecutionManager{},\n}\n\nfunc TestClientsRateLimitAlwaysAllow(t *testing.T) {\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tobject := builderForPassThrough(t, injector, &limiterAlwaysAllow{}, true, quotas.CallerBypass{}, nil)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tcallRes := v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\n\t\t\t\t\tassert.Nil(t, resultErr, \"method %v returned error %v\", method.Name, resultErr)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestClientsAlwaysRateLimited(t *testing.T) {\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tobject := builderForPassThrough(t, injector, &limiterNeverAllow{}, false, quotas.CallerBypass{}, nil)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tcallRes := v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\n\t\t\t\t\terr, ok := resultErr.(error)\n\t\t\t\t\trequire.True(t, ok, \"method %v must return error\")\n\t\t\t\t\tvar expectedErr *types.ServiceBusyError\n\t\t\t\t\tassert.True(t, errors.As(err, &expectedErr), \"method %v must return error of type *types.ServiceBusyError\", method.Name)\n\t\t\t\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message, \"method %v returned different error, expected %v, got %v\", method.Name, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInjectorsWithUnderlyingErrors(t *testing.T) {\n\tfor _, injector := range wrappers {\n\t\tname := reflect.TypeOf(injector).String()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\texpectedMethodErr := fmt.Errorf(\"%s: injected error\", name)\n\t\t\tobject := builderForPassThrough(t, injector, &limiterAlwaysAllow{}, true, quotas.CallerBypass{}, expectedMethodErr)\n\t\t\tv := reflect.ValueOf(object)\n\t\t\tinfoT := reflect.TypeOf(v.Interface())\n\t\t\tfor i := 0; i < infoT.NumMethod(); i++ {\n\t\t\t\tmethod := infoT.Method(i)\n\t\t\t\tif _staticMethods[method.Name] {\n\t\t\t\t\t// Skip methods that do not use error injection.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tt.Run(method.Name, func(t *testing.T) {\n\t\t\t\t\tvals := make([]reflect.Value, 0, method.Type.NumIn()-1)\n\t\t\t\t\t// First argument is always context.Context\n\t\t\t\t\tvals = append(vals, reflect.ValueOf(context.Background()))\n\t\t\t\t\tfor i := 2; i < method.Type.NumIn(); i++ {\n\t\t\t\t\t\tvals = append(vals, reflect.Zero(method.Type.In(i)))\n\t\t\t\t\t}\n\n\t\t\t\t\tcallRes := v.MethodByName(method.Name).Call(vals)\n\t\t\t\t\tresultErr := callRes[len(callRes)-1].Interface()\n\t\t\t\t\terr, ok := resultErr.(error)\n\t\t\t\t\trequire.True(t, ok, \"method %v must return error\")\n\t\t\t\t\tassert.Equal(t, expectedMethodErr, err, \"method %v returned different error, expected %v, got %v\", method.Name, expectedMethodErr, err)\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc builderForPassThrough(t *testing.T, injector any, limiter quotas.Limiter, expectCalls bool, callerBypass quotas.CallerBypass, expectedErr error) (object any) {\n\tctrl := gomock.NewController(t)\n\tswitch injector.(type) {\n\tcase *ratelimitedConfigStoreManager:\n\t\tmocked := persistence.NewMockConfigStoreManager(ctrl)\n\t\tobject = NewConfigStoreManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().FetchDynamicConfig(gomock.Any(), gomock.Any()).Return(&persistence.FetchDynamicConfigResponse{}, expectedErr)\n\t\t}\n\tcase *ratelimitedDomainManager:\n\t\tmocked := persistence.NewMockDomainManager(ctrl)\n\t\tobject = NewDomainManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CreateDomain(gomock.Any(), gomock.Any()).Return(&persistence.CreateDomainResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(&persistence.GetDomainResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteDomainByName(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(&persistence.ListDomainsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{}, expectedErr)\n\t\t}\n\tcase *ratelimitedHistoryManager:\n\t\tmocked := persistence.NewMockHistoryManager(ctrl)\n\t\tobject = NewHistoryManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().AppendHistoryNodes(gomock.Any(), gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadHistoryBranchByBatch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchByBatchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ReadRawHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadRawHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ForkHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ForkHistoryBranchResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteHistoryBranch(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetHistoryTree(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTreeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetAllHistoryTreeBranches(gomock.Any(), gomock.Any()).Return(&persistence.GetAllHistoryTreeBranchesResponse{}, expectedErr)\n\t\t}\n\tcase *ratelimitedQueueManager:\n\t\tmocked := persistence.NewMockQueueManager(ctrl)\n\t\tobject = NewQueueManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().EnqueueMessage(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ReadMessages(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesResponse{Messages: []*persistence.QueueMessage{}}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteMessagesBefore(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteMessageFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().EnqueueMessageToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetDLQAckLevels(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQAckLevelsResponse{AckLevels: map[string]int64{}}, expectedErr)\n\t\t\tmocked.EXPECT().GetDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetDLQSizeResponse{Size: 0}, expectedErr)\n\t\t\tmocked.EXPECT().RangeDeleteMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ReadMessagesFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.ReadMessagesFromDLQResponse{Messages: []*persistence.QueueMessage{}}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateDLQAckLevel(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tcase *ratelimitedShardManager:\n\t\tmocked := persistence.NewMockShardManager(ctrl)\n\t\tobject = NewShardManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().GetShard(gomock.Any(), gomock.Any()).Return(&persistence.GetShardResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateShard(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CreateShard(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tcase *ratelimitedTaskManager:\n\t\tmocked := persistence.NewMockTaskManager(ctrl)\n\t\tobject = NewTaskManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CompleteTasksLessThan(gomock.Any(), gomock.Any()).Return(&persistence.CompleteTasksLessThanResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CompleteTask(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CreateTasks(gomock.Any(), gomock.Any()).Return(&persistence.CreateTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteTaskList(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetOrphanTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetOrphanTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetTaskListSize(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListSizeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().LeaseTaskList(gomock.Any(), gomock.Any()).Return(&persistence.LeaseTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetTaskList(gomock.Any(), gomock.Any()).Return(&persistence.GetTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListTaskList(gomock.Any(), gomock.Any()).Return(&persistence.ListTaskListResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any()).Return(&persistence.UpdateTaskListResponse{}, expectedErr)\n\t\t}\n\tcase *ratelimitedVisibilityManager:\n\t\tmocked := persistence.NewMockVisibilityManager(ctrl)\n\t\tobject = NewVisibilityManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().DeleteUninitializedWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.CountWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetClosedWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetClosedWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByStatus(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RecordWorkflowExecutionUninitialized(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, expectedErr)\n\t\t}\n\tcase *ratelimitedExecutionManager:\n\t\tmocked := persistence.NewMockExecutionManager(ctrl)\n\t\tobject = NewExecutionManager(mocked, limiter, callerBypass)\n\t\tif expectCalls {\n\t\t\tmocked.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.CreateWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetCurrentExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.ConflictResolveWorkflowExecutionResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CreateFailoverMarkerTasks(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().DeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetReplicationDLQSize(gomock.Any(), gomock.Any()).Return(&persistence.GetReplicationDLQSizeResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetReplicationTasksFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Any()).Return(&persistence.IsWorkflowExecutionExistsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListConcreteExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListCurrentExecutionsResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().PutReplicationTaskToDLQ(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().CompleteHistoryTask(gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t\tmocked.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(&persistence.RangeCompleteHistoryTaskResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().RangeDeleteReplicationTaskFromDLQ(gomock.Any(), gomock.Any()).Return(&persistence.RangeDeleteReplicationTaskFromDLQResponse{}, expectedErr)\n\t\t\tmocked.EXPECT().GetActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterSelectionPolicy{}, expectedErr)\n\t\t\tmocked.EXPECT().DeleteActiveClusterSelectionPolicy(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(expectedErr)\n\t\t}\n\tdefault:\n\t\tt.Errorf(\"unsupported type %v\", reflect.TypeOf(injector))\n\t\tt.FailNow()\n\t}\n\treturn\n}\n\nfunc TestVisibilityManagerBypassRateLimitForCallerTypes(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tcallerType        types.CallerType\n\t\tbypassCallerTypes []interface{}\n\t\tshouldBypass      bool\n\t}{\n\t\t{\n\t\t\tname:              \"CLI bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"UI bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeUI,\n\t\t\tbypassCallerTypes: []interface{}{\"ui\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"SDK bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\tbypassCallerTypes: []interface{}{\"sdk\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Internal bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeInternal,\n\t\t\tbypassCallerTypes: []interface{}{\"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Multiple types can bypass\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Caller type not in bypass list is rate limited\",\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Unknown does not bypass\",\n\t\t\tcallerType:        types.CallerTypeUnknown,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\", \"sdk\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmocked := persistence.NewMockVisibilityManager(ctrl)\n\n\t\t\tconfigClient := dynamicconfig.NewInMemoryClient()\n\t\t\t_ = configClient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, tt.bypassCallerTypes)\n\n\t\t\tdc := dynamicconfig.NewCollection(\n\t\t\t\tconfigClient,\n\t\t\t\ttestlogger.New(t),\n\t\t\t)\n\n\t\t\tcallerBypass := quotas.NewCallerBypass(dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes))\n\t\t\tvm := NewVisibilityManager(mocked, &limiterNeverAllow{}, callerBypass)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\n\t\t\tif tt.shouldBypass {\n\t\t\t\tmocked.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\terr := vm.RecordWorkflowExecutionStarted(ctx, &persistence.RecordWorkflowExecutionStartedRequest{})\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\terr := vm.RecordWorkflowExecutionStarted(ctx, &persistence.RecordWorkflowExecutionStartedRequest{})\n\t\t\t\tvar expectedErr *types.ServiceBusyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr))\n\t\t\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVisibilityManagerBypassRateLimitWithDynamicConfig(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmocked := persistence.NewMockVisibilityManager(ctrl)\n\n\tconfigClient := dynamicconfig.NewInMemoryClient()\n\tconfigClient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, []interface{}{\"cli\", \"internal\"})\n\n\tdc := dynamicconfig.NewCollection(\n\t\tconfigClient,\n\t\ttestlogger.New(t),\n\t)\n\n\tcallerBypass := quotas.NewCallerBypass(dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes))\n\tvm := NewVisibilityManager(mocked, &limiterNeverAllow{}, callerBypass)\n\n\tt.Run(\"CLI bypasses rate limit\", func(t *testing.T) {\n\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(types.CallerTypeCLI))\n\t\tmocked.EXPECT().RecordWorkflowExecutionStarted(ctx, gomock.Any()).Return(nil)\n\n\t\terr := vm.RecordWorkflowExecutionStarted(ctx, &persistence.RecordWorkflowExecutionStartedRequest{})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"Internal bypasses rate limit\", func(t *testing.T) {\n\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(types.CallerTypeInternal))\n\t\tmocked.EXPECT().RecordWorkflowExecutionClosed(ctx, gomock.Any()).Return(nil)\n\n\t\terr := vm.RecordWorkflowExecutionClosed(ctx, &persistence.RecordWorkflowExecutionClosedRequest{})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"SDK does not bypass rate limit\", func(t *testing.T) {\n\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(types.CallerTypeSDK))\n\n\t\terr := vm.UpsertWorkflowExecution(ctx, &persistence.UpsertWorkflowExecutionRequest{})\n\t\tvar expectedErr *types.ServiceBusyError\n\t\tassert.True(t, errors.As(err, &expectedErr))\n\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t})\n\n\tt.Run(\"No CallerInfo does not bypass rate limit\", func(t *testing.T) {\n\t\tctx := context.Background()\n\n\t\terr := vm.DeleteWorkflowExecution(ctx, &persistence.VisibilityDeleteWorkflowExecutionRequest{})\n\t\tvar expectedErr *types.ServiceBusyError\n\t\tassert.True(t, errors.As(err, &expectedErr))\n\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t})\n}\n\nfunc TestVisibilityManagerNoBypassWithoutDynamicConfig(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmocked := persistence.NewMockVisibilityManager(ctrl)\n\n\tvm := NewVisibilityManager(mocked, &limiterNeverAllow{}, quotas.CallerBypass{})\n\n\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(types.CallerTypeCLI))\n\n\terr := vm.RecordWorkflowExecutionStarted(ctx, &persistence.RecordWorkflowExecutionStartedRequest{})\n\tvar expectedErr *types.ServiceBusyError\n\tassert.True(t, errors.As(err, &expectedErr))\n\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n}\n\nfunc TestShardManagerBypassRateLimitForCallerTypes(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tcallerType        types.CallerType\n\t\tbypassCallerTypes []interface{}\n\t\tshouldBypass      bool\n\t}{\n\t\t{\n\t\t\tname:              \"CLI bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Internal bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeInternal,\n\t\t\tbypassCallerTypes: []interface{}{\"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Multiple types can bypass\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Caller type not in bypass list is rate limited\",\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Unknown does not bypass\",\n\t\t\tcallerType:        types.CallerTypeUnknown,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\", \"sdk\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmocked := persistence.NewMockShardManager(ctrl)\n\n\t\t\tconfigClient := dynamicconfig.NewInMemoryClient()\n\t\t\t_ = configClient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, tt.bypassCallerTypes)\n\n\t\t\tdc := dynamicconfig.NewCollection(\n\t\t\t\tconfigClient,\n\t\t\t\ttestlogger.New(t),\n\t\t\t)\n\n\t\t\tcallerBypass := quotas.NewCallerBypass(dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes))\n\t\t\tsm := NewShardManager(mocked, &limiterNeverAllow{}, callerBypass)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\n\t\t\tif tt.shouldBypass {\n\t\t\t\tmocked.EXPECT().GetShard(gomock.Any(), gomock.Any()).Return(&persistence.GetShardResponse{}, nil)\n\t\t\t\t_, err := sm.GetShard(ctx, &persistence.GetShardRequest{})\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\t_, err := sm.GetShard(ctx, &persistence.GetShardRequest{})\n\t\t\t\tvar expectedErr *types.ServiceBusyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr))\n\t\t\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHistoryManagerBypassRateLimitForCallerTypes(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tcallerType        types.CallerType\n\t\tbypassCallerTypes []interface{}\n\t\tshouldBypass      bool\n\t}{\n\t\t{\n\t\t\tname:              \"CLI bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Internal bypasses when configured\",\n\t\t\tcallerType:        types.CallerTypeInternal,\n\t\t\tbypassCallerTypes: []interface{}{\"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Multiple types can bypass\",\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Caller type not in bypass list is rate limited\",\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Unknown does not bypass\",\n\t\t\tcallerType:        types.CallerTypeUnknown,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\", \"sdk\", \"internal\"},\n\t\t\tshouldBypass:      false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmocked := persistence.NewMockHistoryManager(ctrl)\n\n\t\t\tconfigClient := dynamicconfig.NewInMemoryClient()\n\t\t\t_ = configClient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, tt.bypassCallerTypes)\n\n\t\t\tdc := dynamicconfig.NewCollection(\n\t\t\t\tconfigClient,\n\t\t\t\ttestlogger.New(t),\n\t\t\t)\n\n\t\t\tcallerBypass := quotas.NewCallerBypass(dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes))\n\t\t\thm := NewHistoryManager(mocked, &limiterNeverAllow{}, callerBypass)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\n\t\t\tif tt.shouldBypass {\n\t\t\t\tmocked.EXPECT().AppendHistoryNodes(gomock.Any(), gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\t_, err := hm.AppendHistoryNodes(ctx, &persistence.AppendHistoryNodesRequest{})\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\t_, err := hm.AppendHistoryNodes(ctx, &persistence.AppendHistoryNodesRequest{})\n\t\t\t\tvar expectedErr *types.ServiceBusyError\n\t\t\t\tassert.True(t, errors.As(err, &expectedErr))\n\t\t\t\tassert.Equal(t, ErrPersistenceLimitExceeded.Message, expectedErr.Message)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/persistence/wrappers/sampled/tokenbucketfactory.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sampled\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/tokenbucket\"\n)\n\ntype RateLimiterFactoryFunc func(timeSource clock.TimeSource, numOfPriority int, qpsConfig dynamicproperties.IntPropertyFnWithDomainFilter) RateLimiterFactory\n\ntype RateLimiterFactory interface {\n\tGetRateLimiter(domain string) tokenbucket.PriorityTokenBucket\n}\n\ntype domainToBucketMap struct {\n\tsync.RWMutex\n\ttimeSource    clock.TimeSource\n\tqpsConfig     dynamicproperties.IntPropertyFnWithDomainFilter\n\tnumOfPriority int\n\tmappings      map[string]tokenbucket.PriorityTokenBucket\n}\n\n// NewDomainToBucketMap returns a rate limiter factory.\nfunc NewDomainToBucketMap(timeSource clock.TimeSource, numOfPriority int, qpsConfig dynamicproperties.IntPropertyFnWithDomainFilter) RateLimiterFactory {\n\treturn &domainToBucketMap{\n\t\ttimeSource:    timeSource,\n\t\tqpsConfig:     qpsConfig,\n\t\tnumOfPriority: numOfPriority,\n\t\tmappings:      make(map[string]tokenbucket.PriorityTokenBucket),\n\t}\n}\n\nfunc (m *domainToBucketMap) GetRateLimiter(domain string) tokenbucket.PriorityTokenBucket {\n\tm.RLock()\n\trateLimiter, exist := m.mappings[domain]\n\tm.RUnlock()\n\n\tif exist {\n\t\treturn rateLimiter\n\t}\n\n\tm.Lock()\n\tif rateLimiter, ok := m.mappings[domain]; ok { // read again to ensure no duplicate create\n\t\tm.Unlock()\n\t\treturn rateLimiter\n\t}\n\trateLimiter = tokenbucket.NewFullPriorityTokenBucket(m.numOfPriority, m.qpsConfig(domain), m.timeSource)\n\tm.mappings[domain] = rateLimiter\n\tm.Unlock()\n\treturn rateLimiter\n}\n"
  },
  {
    "path": "common/persistence/wrappers/sampled/tokenbucketfactory_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sampled\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\nfunc TestDomainToBucketMap(t *testing.T) {\n\tmockedTime := clock.NewMockedTimeSource()\n\tfactory := NewDomainToBucketMap(mockedTime, 1, dynamicproperties.GetIntPropertyFilteredByDomain(1))\n\n\t// Test that the factory returns the same bucket for the same domain\n\tbucket1 := factory.GetRateLimiter(\"domain1\")\n\tbucket2 := factory.GetRateLimiter(\"domain1\")\n\tassert.Equal(t, bucket1, bucket2, \"domain bucket should return the same bucket for the same domain\")\n}\n"
  },
  {
    "path": "common/persistence/wrappers/sampled/visibility_manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sampled\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// To sample visibility request, open has only 1 bucket, closed has 2\n\tnumOfPriorityForOpen   = 1\n\tnumOfPriorityForClosed = 2\n\tnumOfPriorityForList   = 1\n)\n\n// errPersistenceLimitExceededForList is the error indicating QPS limit reached for list visibility.\nvar errPersistenceLimitExceededForList = &types.ServiceBusyError{Message: \"Persistence Max QPS Reached for List Operations.\"}\n\ntype visibilityManager struct {\n\trateLimitersForOpen   RateLimiterFactory\n\trateLimitersForClosed RateLimiterFactory\n\trateLimitersForList   RateLimiterFactory\n\tpersistence           persistence.VisibilityManager\n\tmetricClient          metrics.Client\n\tlogger                log.Logger\n}\n\ntype (\n\t// Config is config for visibility\n\tConfig struct {\n\t\tVisibilityOpenMaxQPS dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\t\t// VisibilityClosedMaxQPS max QPS for record closed workflows\n\t\tVisibilityClosedMaxQPS dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\t\t// VisibilityListMaxQPS max QPS for list workflow\n\t\tVisibilityListMaxQPS dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\t}\n)\n\ntype Params struct {\n\tConfig                 *Config\n\tMetricClient           metrics.Client\n\tLogger                 log.Logger\n\tTimeSource             clock.TimeSource\n\tRateLimiterFactoryFunc RateLimiterFactoryFunc\n}\n\n// NewVisibilityManager creates a client to manage visibility with sampling\n// For write requests, it will do sampling which will lose some records\n// For read requests, it will do sampling which will return service busy errors.\n// Note that this is different from NewVisibilityPersistenceRateLimitedClient which is overlapping with the read processing.\nfunc NewVisibilityManager(persistence persistence.VisibilityManager, p Params) persistence.VisibilityManager {\n\treturn &visibilityManager{\n\t\tpersistence:           persistence,\n\t\trateLimitersForOpen:   p.RateLimiterFactoryFunc(p.TimeSource, numOfPriorityForOpen, p.Config.VisibilityOpenMaxQPS),\n\t\trateLimitersForClosed: p.RateLimiterFactoryFunc(p.TimeSource, numOfPriorityForClosed, p.Config.VisibilityClosedMaxQPS),\n\t\trateLimitersForList:   p.RateLimiterFactoryFunc(p.TimeSource, numOfPriorityForList, p.Config.VisibilityListMaxQPS),\n\t\tmetricClient:          p.MetricClient,\n\t\tlogger:                p.Logger,\n\t}\n}\n\nfunc (p *visibilityManager) RecordWorkflowExecutionStarted(\n\tctx context.Context,\n\trequest *persistence.RecordWorkflowExecutionStartedRequest,\n) error {\n\tdomain := request.Domain\n\tdomainID := request.DomainUUID\n\n\trateLimiter := p.rateLimitersForOpen.GetRateLimiter(domain)\n\tif ok, _ := rateLimiter.GetToken(0, 1); ok {\n\t\treturn p.persistence.RecordWorkflowExecutionStarted(ctx, request)\n\t}\n\n\tp.logger.Info(\"Request for open workflow is sampled\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowDomainName(domain),\n\t\ttag.WorkflowType(request.WorkflowTypeName),\n\t\ttag.WorkflowID(request.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.Execution.GetRunID()),\n\t)\n\tp.metricClient.IncCounter(metrics.PersistenceRecordWorkflowExecutionStartedScope, metrics.PersistenceSampledCounter)\n\treturn nil\n}\n\nfunc (p *visibilityManager) RecordWorkflowExecutionClosed(\n\tctx context.Context,\n\trequest *persistence.RecordWorkflowExecutionClosedRequest,\n) error {\n\tdomain := request.Domain\n\tdomainID := request.DomainUUID\n\tpriority := getRequestPriority(request)\n\n\trateLimiter := p.rateLimitersForClosed.GetRateLimiter(domain)\n\tif ok, _ := rateLimiter.GetToken(priority, 1); ok {\n\t\treturn p.persistence.RecordWorkflowExecutionClosed(ctx, request)\n\t}\n\n\tp.logger.Info(\"Request for closed workflow is sampled\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowDomainName(domain),\n\t\ttag.WorkflowType(request.WorkflowTypeName),\n\t\ttag.WorkflowID(request.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.Execution.GetRunID()),\n\t)\n\tp.metricClient.IncCounter(metrics.PersistenceRecordWorkflowExecutionClosedScope, metrics.PersistenceSampledCounter)\n\treturn nil\n}\n\nfunc (p *visibilityManager) UpsertWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.UpsertWorkflowExecutionRequest,\n) error {\n\tdomain := request.Domain\n\tdomainID := request.DomainUUID\n\n\trateLimiter := p.rateLimitersForClosed.GetRateLimiter(domain)\n\tif ok, _ := rateLimiter.GetToken(0, 1); ok {\n\t\treturn p.persistence.UpsertWorkflowExecution(ctx, request)\n\t}\n\n\tp.logger.Info(\"Request for upsert workflow is sampled\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowDomainName(domain),\n\t\ttag.WorkflowType(request.WorkflowTypeName),\n\t\ttag.WorkflowID(request.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.Execution.GetRunID()),\n\t)\n\tp.metricClient.IncCounter(metrics.PersistenceUpsertWorkflowExecutionScope, metrics.PersistenceSampledCounter)\n\treturn nil\n}\n\nfunc (p *visibilityManager) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListOpenWorkflowExecutions\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListOpenWorkflowExecutions(ctx, request)\n}\n\nfunc (p *visibilityManager) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListClosedWorkflowExecutions\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListClosedWorkflowExecutions(ctx, request)\n}\n\nfunc (p *visibilityManager) ListOpenWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByTypeRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListOpenWorkflowExecutionsByType\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListOpenWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (p *visibilityManager) ListClosedWorkflowExecutionsByType(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByTypeRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListClosedWorkflowExecutionsByType\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListClosedWorkflowExecutionsByType(ctx, request)\n}\n\nfunc (p *visibilityManager) ListOpenWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListOpenWorkflowExecutionsByWorkflowID\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListOpenWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (p *visibilityManager) ListClosedWorkflowExecutionsByWorkflowID(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByWorkflowIDRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListClosedWorkflowExecutionsByWorkflowID\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListClosedWorkflowExecutionsByWorkflowID(ctx, request)\n}\n\nfunc (p *visibilityManager) ListClosedWorkflowExecutionsByStatus(\n\tctx context.Context,\n\trequest *persistence.ListClosedWorkflowExecutionsByStatusRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\tif err := p.tryConsumeListToken(request.Domain, \"ListClosedWorkflowExecutionsByStatus\"); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn p.persistence.ListClosedWorkflowExecutionsByStatus(ctx, request)\n}\n\nfunc (p *visibilityManager) RecordWorkflowExecutionUninitialized(\n\tctx context.Context,\n\trequest *persistence.RecordWorkflowExecutionUninitializedRequest,\n) error {\n\treturn p.persistence.RecordWorkflowExecutionUninitialized(ctx, request)\n}\n\nfunc (p *visibilityManager) GetClosedWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.GetClosedWorkflowExecutionRequest,\n) (*persistence.GetClosedWorkflowExecutionResponse, error) {\n\treturn p.persistence.GetClosedWorkflowExecution(ctx, request)\n}\n\nfunc (p *visibilityManager) DeleteWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn p.persistence.DeleteWorkflowExecution(ctx, request)\n}\n\nfunc (p *visibilityManager) DeleteUninitializedWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.VisibilityDeleteWorkflowExecutionRequest,\n) error {\n\treturn p.persistence.DeleteUninitializedWorkflowExecution(ctx, request)\n}\n\nfunc (p *visibilityManager) ListWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByQueryRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\treturn p.persistence.ListWorkflowExecutions(ctx, request)\n}\n\nfunc (p *visibilityManager) ScanWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.ListWorkflowExecutionsByQueryRequest,\n) (*persistence.ListWorkflowExecutionsResponse, error) {\n\treturn p.persistence.ScanWorkflowExecutions(ctx, request)\n}\n\nfunc (p *visibilityManager) CountWorkflowExecutions(\n\tctx context.Context,\n\trequest *persistence.CountWorkflowExecutionsRequest,\n) (*persistence.CountWorkflowExecutionsResponse, error) {\n\treturn p.persistence.CountWorkflowExecutions(ctx, request)\n}\n\nfunc (p *visibilityManager) Close() {\n\tp.persistence.Close()\n}\n\nfunc (p *visibilityManager) GetName() string {\n\treturn p.persistence.GetName()\n}\n\nfunc getRequestPriority(request *persistence.RecordWorkflowExecutionClosedRequest) int {\n\tpriority := 0\n\tif request.Status == types.WorkflowExecutionCloseStatusCompleted {\n\t\tpriority = 1 // low priority for completed workflows\n\t}\n\treturn priority\n}\n\nfunc (p *visibilityManager) tryConsumeListToken(domain, method string) error {\n\trateLimiter := p.rateLimitersForList.GetRateLimiter(domain)\n\tok, _ := rateLimiter.GetToken(0, 1)\n\tif ok {\n\t\tp.logger.Debug(\"List API request consumed QPS token\", tag.WorkflowDomainName(domain), tag.Name(method))\n\t\treturn nil\n\t}\n\tp.logger.Debug(\"List API request is being sampled\", tag.WorkflowDomainName(domain), tag.Name(method))\n\treturn errPersistenceLimitExceededForList\n}\n"
  },
  {
    "path": "common/persistence/wrappers/sampled/visibility_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sampled\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/tokenbucket\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestVisibilityManagerSampledCalls(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname        string\n\t\tpriority    int\n\t\tprepareMock func(*persistence.MockVisibilityManager)\n\t\toperation   func(context.Context, string, persistence.VisibilityManager) error\n\t}{\n\t\t{\n\t\t\tname: \"RecordWorkflowExecutionStarted\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().RecordWorkflowExecutionStarted(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\treturn m.RecordWorkflowExecutionStarted(ctx, &persistence.RecordWorkflowExecutionStartedRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"RecordWorkflowExecutionClosed\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\treturn m.RecordWorkflowExecutionClosed(ctx, &persistence.RecordWorkflowExecutionClosedRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t\tStatus: types.WorkflowExecutionCloseStatusCanceled,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"RecordWorkflowExecutionClosed_Completed\",\n\t\t\tpriority: 1,\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().RecordWorkflowExecutionClosed(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\treturn m.RecordWorkflowExecutionClosed(ctx, &persistence.RecordWorkflowExecutionClosedRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t\tStatus: types.WorkflowExecutionCloseStatusCompleted,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"UpsertWorkflowExecution\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().UpsertWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\treturn m.UpsertWorkflowExecution(ctx, &persistence.UpsertWorkflowExecutionRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedManager := persistence.NewMockVisibilityManager(ctrl)\n\n\t\t\ttestDomain := \"domain1\"\n\n\t\t\tm := NewVisibilityManager(mockedManager, Params{\n\t\t\t\tConfig:       &Config{},\n\t\t\t\tMetricClient: metrics.NewNoopMetricsClient(),\n\t\t\t\tLogger:       testlogger.New(t),\n\t\t\t\tTimeSource:   clock.NewMockedTimeSource(),\n\t\t\t\tRateLimiterFactoryFunc: rateLimiterStubFunc(map[string]tokenbucket.PriorityTokenBucket{\n\t\t\t\t\ttestDomain: &tokenBucketFactoryStub{tokens: map[int]int{tc.priority: 1}},\n\t\t\t\t}),\n\t\t\t})\n\n\t\t\ttc.prepareMock(mockedManager)\n\n\t\t\terr := tc.operation(context.Background(), testDomain, m)\n\t\t\tassert.NoError(t, err, \"first call should succeed\")\n\n\t\t\terr = tc.operation(context.Background(), testDomain, m)\n\t\t\tassert.NoError(t, err, \"second call should not fail, but underlying call should be blocked by rate limiter\")\n\t\t})\n\t}\n}\n\nfunc TestVisibilityManagerListOperations(t *testing.T) {\n\tfor _, tc := range []struct {\n\t\tname        string\n\t\tpriority    int\n\t\tprepareMock func(*persistence.MockVisibilityManager)\n\t\toperation   func(context.Context, string, persistence.VisibilityManager) error\n\t}{\n\t\t{\n\t\t\tname: \"ListOpenWorkflowExecutions\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListOpenWorkflowExecutions(ctx, &persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ListClosedWorkflowExecutions\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListClosedWorkflowExecutions(ctx, &persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\tDomain: domain,\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ListOpenWorkflowExecutionsByType\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListOpenWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListOpenWorkflowExecutionsByType(ctx, &persistence.ListWorkflowExecutionsByTypeRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\t\tDomain: domain,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ListClosedWorkflowExecutionsByType\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListClosedWorkflowExecutionsByType(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListClosedWorkflowExecutionsByType(ctx, &persistence.ListWorkflowExecutionsByTypeRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\t\tDomain: domain,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ListOpenWorkflowExecutionsByWorkflowID\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListOpenWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListOpenWorkflowExecutionsByWorkflowID(ctx, &persistence.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\t\tDomain: domain,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ListClosedWorkflowExecutionsByWorkflowID\",\n\t\t\tprepareMock: func(mock *persistence.MockVisibilityManager) {\n\t\t\t\tmock.EXPECT().ListClosedWorkflowExecutionsByWorkflowID(gomock.Any(), gomock.Any()).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\toperation: func(ctx context.Context, domain string, m persistence.VisibilityManager) error {\n\t\t\t\t_, err := m.ListClosedWorkflowExecutionsByWorkflowID(ctx, &persistence.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: persistence.ListWorkflowExecutionsRequest{\n\t\t\t\t\t\tDomain: domain,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockedManager := persistence.NewMockVisibilityManager(ctrl)\n\n\t\t\ttestDomain := \"domain1\"\n\n\t\t\tm := NewVisibilityManager(mockedManager, Params{\n\t\t\t\tConfig:       &Config{},\n\t\t\t\tMetricClient: metrics.NewNoopMetricsClient(),\n\t\t\t\tLogger:       testlogger.New(t),\n\t\t\t\tTimeSource:   clock.NewMockedTimeSource(),\n\t\t\t\tRateLimiterFactoryFunc: rateLimiterStubFunc(map[string]tokenbucket.PriorityTokenBucket{\n\t\t\t\t\ttestDomain: &tokenBucketFactoryStub{tokens: map[int]int{tc.priority: 1}},\n\t\t\t\t}),\n\t\t\t})\n\n\t\t\ttc.prepareMock(mockedManager)\n\n\t\t\terr := tc.operation(context.Background(), testDomain, m)\n\t\t\tassert.NoError(t, err, \"first call should succeed\")\n\n\t\t\terr = tc.operation(context.Background(), testDomain, m)\n\t\t\tassert.Error(t, err, \"second call should fail since underlying call should be blocked by rate limiter\")\n\t\t})\n\t}\n}\n\nfunc rateLimiterStubFunc(domainData map[string]tokenbucket.PriorityTokenBucket) RateLimiterFactoryFunc {\n\treturn func(timeSource clock.TimeSource, numOfPriority int, qpsConfig dynamicproperties.IntPropertyFnWithDomainFilter) RateLimiterFactory {\n\t\treturn rateLimiterStub{domainData}\n\t}\n}\n\ntype rateLimiterStub struct {\n\tdata map[string]tokenbucket.PriorityTokenBucket\n}\n\nfunc (r rateLimiterStub) GetRateLimiter(domain string) tokenbucket.PriorityTokenBucket {\n\treturn r.data[domain]\n}\n\ntype tokenBucketFactoryStub struct {\n\ttokens map[int]int\n}\n\nfunc (t *tokenBucketFactoryStub) GetToken(priority, count int) (bool, time.Duration) {\n\tval := t.tokens[priority]\n\tif count > val {\n\t\treturn false, time.Duration(0)\n\t}\n\tval -= count\n\tt.tokens[priority] = val\n\treturn true, time.Duration(0)\n}\n"
  },
  {
    "path": "common/persistence/wrappers/templates/errorinjector.tmpl",
    "content": "import (\n\t\"context\"\n\n    \"time\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n{{ $decorator := (printf \"injector%s\" .Interface.Name) }}\n{{ $interfaceName := .Interface.Name }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with error injection.\ntype {{$decorator}} struct {\n    wrapped   {{.Interface.Type}}\n\tstarttime time.Time\n\terrorRate float64\n\tlogger    log.Logger\n}\n\n// New{{.Interface.Name}} creates a new instance of {{.Interface.Name}} with error injection.\nfunc New{{.Interface.Name}}(\n    wrapped   persistence.{{.Interface.Name}},\n\terrorRate float64,\n\tlogger    log.Logger,\n    starttime time.Time,\n) persistence.{{.Interface.Name}} {\n    return &{{$decorator}}{\n        wrapped:   wrapped,\n        starttime: starttime,\n        errorRate: errorRate,\n        logger:    logger,\n    }\n}\n\n{{range $methodName, $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n\t        fakeErr := generateFakeError(c.errorRate, c.starttime)\n\t        var forwardCall bool\n\t        if forwardCall = shouldForwardCallToPersistence(fakeErr); forwardCall {\n\t            {{$method.ResultsNames}} = c.wrapped.{{$method.Call}}\n\t        }\n\n\t        if fakeErr != nil {\n\t            logErr(c.logger, \"{{$interfaceName}}.{{$methodName}}\", fakeErr, forwardCall, err)\n\t            err = fakeErr\n\t            return\n            }\n            return\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "common/persistence/wrappers/templates/metered.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n{{ $decorator := (printf \"metered%s\" .Interface.Name) }}\n{{ $interfaceName := .Interface.Name }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with rate limiter.\ntype {{$decorator}} struct {\n    base\n    wrapped     {{.Interface.Type}}\n}\n\n// New{{.Interface.Name}} creates a new instance of {{.Interface.Name}} with ratelimiter.\nfunc New{{.Interface.Name}}(\n\twrapped       persistence.{{.Interface.Name}},\n\tmetricClient  metrics.Client,\n\tlogger        log.Logger,\n\tcfg           *config.Persistence,\n) persistence.{{.Interface.Name}} {\n    return &{{$decorator}}{\n        wrapped: wrapped,\n        base:       base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger,\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n        },\n    }\n}\n\n{{range $methodName, $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n\t        op := func() error {\n\t\t        {{$method.ResultsNames}} = c.wrapped.{{$method.Call}}\n\t\t        {{ if and (gt (len $method.Params) 1) (gt (len $method.Results) 0) -}}\n\t\t             c.emptyMetric(\"{{$interfaceName}}.{{$methodName}}\", {{(index $method.Params 1).Name}}, {{(index $method.Results 0).Name}}, err)\n\t\t        {{ end -}}\n\t\t        return err\n\t        }\n\t        {{$scopeName := printf \"metrics.Persistence%sScope\" $methodName}}\n            {{ if and (eq $interfaceName \"VisibilityManager\") (eq $methodName \"DeleteWorkflowExecution\") -}}\n                {{ $scopeName = \"metrics.PersistenceVisibilityDeleteWorkflowExecutionScope\" }}\n            {{ end -}}\n\n            {{$extraTags := \"\"}}\n            {{ if gt (len $method.Params) 1 -}}\n               {{ $reqName := (index $method.Params 1).Name }}\n               {{ $extraTags = printf \", getCustomMetricTags(%s)...\" $reqName }}\n            {{ end -}}\n\n\t        err = c.call({{$scopeName}}, op{{$extraTags}})\n\t        return\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "common/persistence/wrappers/templates/metered_execution.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\n{{ $decorator := (printf \"metered%s\" .Interface.Name) }}\n{{ $interfaceName := .Interface.Name }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with rate limiter.\ntype {{$decorator}} struct {\n    base\n    wrapped     {{.Interface.Type}}\n}\n\n// New{{.Interface.Name}} creates a new instance of {{.Interface.Name}} with ratelimiter.\nfunc New{{.Interface.Name}}(\n\twrapped      persistence.{{.Interface.Name}},\n\tmetricClient metrics.Client,\n\tlogger       log.Logger,\n\tcfg          *config.Persistence,\n    sampleLoggingRate dynamicproperties.IntPropertyFn,\n    enableShardIDMetrics dynamicproperties.BoolPropertyFn,\n) persistence.{{.Interface.Name}} {\n    return &{{$decorator}}{\n        wrapped: wrapped,\n        base:       base{\n\t\t\tmetricClient:                  metricClient,\n\t\t\tlogger:                        logger.WithTags(tag.ShardID(wrapped.GetShardID())),\n\t\t\tenableLatencyHistogramMetrics: cfg.EnablePersistenceLatencyHistogramMetrics,\n\t\t\tsampleLoggingRate:             sampleLoggingRate,\n\t\t\tenableShardIDMetrics:          enableShardIDMetrics,\n        },\n    }\n}\n\n{{range $methodName, $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n        \top := func() error {\n        \t    {{$method.ResultsNames}} = c.wrapped.{{$method.Call}}\n        \t\t{{ if gt (len $method.Results) 1 -}}\n        \t\t    c.emptyMetric(\"{{$interfaceName}}.{{$methodName}}\", {{(index $method.Params 1).Name}}, {{(index $method.Results 0).Name}}, err)\n        \t\t{{ end -}}\n        \t\treturn err\n        \t}\n\n        \t{{$scopeName := printf \"metrics.Persistence%sScope\" $methodName}}\n\n            {{ if gt (len $method.Params) 1 -}}\n                {{ $reqName := (index $method.Params 1).Name }}\n                retryCount := getRetryCountFromContext(ctx)\n                if domainName, hasDomainName := getDomainNameFromRequest({{$reqName}}); hasDomainName {\n                    logTags := append([]tag.Tag{tag.WorkflowDomainName(domainName)}, getCustomLogTags({{$reqName}})...)\n                    c.logger.SampleInfo(\"Persistence {{$methodName}} called\", c.sampleLoggingRate(), logTags...)\n                \tif c.enableShardIDMetrics() {\n                \t    err = c.callWithDomainAndShardScope({{$scopeName}}, op, metrics.DomainTag(domainName),\n                \t    metrics.ShardIDTag(c.GetShardID()), metrics.IsRetryTag(retryCount > 0))\n                \t} else {\n                \t\terr = c.call({{$scopeName}}, op, metrics.DomainTag(domainName), metrics.IsRetryTag(retryCount > 0))\n                \t}\n                \treturn\n                }\n            {{ end -}}\n\n             {{$extraTags := \"\"}}\n             {{ if gt (len $method.Params) 1 -}}\n                  {{ $reqName := (index $method.Params 1).Name }}\n                  {{ $extraTags = printf \", append(getCustomMetricTags(%s), metrics.IsRetryTag(retryCount > 0))...\" $reqName }}\n             {{ end -}}\n\n\t        err = c.callWithoutDomainTag({{$scopeName}}, op{{$extraTags}})\n\n\t        return\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "common/persistence/wrappers/templates/ratelimited.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n{{ $decorator := (printf \"ratelimited%s\" .Interface.Name) }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with rate limiter.\ntype {{$decorator}} struct {\n    wrapped      {{.Interface.Type}}\n    rateLimiter  quotas.Limiter\n    callerBypass quotas.CallerBypass\n}\n\n// New{{.Interface.Name}} creates a new instance of {{.Interface.Name}} with ratelimiter.\nfunc New{{.Interface.Name}}(\n    wrapped persistence.{{.Interface.Name}},\n    rateLimiter quotas.Limiter,\n    callerBypass quotas.CallerBypass,\n) persistence.{{.Interface.Name}} {\n    return &{{$decorator}}{\n        wrapped: wrapped,\n        rateLimiter: rateLimiter,\n        callerBypass: callerBypass,\n    }\n}\n\n{{range $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n\t        if !c.callerBypass.AllowLimiter({{(index $method.Params 0).Name}}, c.rateLimiter) {\n\t\t        err = ErrPersistenceLimitExceeded\n\t\t        return\n            }\n            {{ $method.Pass \"c.wrapped.\" }}\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "common/pinot/generic_client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package pinot -source interfaces.go -destination generic_client_mock.go -self_package github.com/uber/cadence/common/pinot\n//\n\n// Package pinot is a generated GoMock package.\npackage pinot\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockGenericClient is a mock of GenericClient interface.\ntype MockGenericClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockGenericClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockGenericClientMockRecorder is the mock recorder for MockGenericClient.\ntype MockGenericClientMockRecorder struct {\n\tmock *MockGenericClient\n}\n\n// NewMockGenericClient creates a new mock instance.\nfunc NewMockGenericClient(ctrl *gomock.Controller) *MockGenericClient {\n\tmock := &MockGenericClient{ctrl: ctrl}\n\tmock.recorder = &MockGenericClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockGenericClient) EXPECT() *MockGenericClientMockRecorder {\n\treturn m.recorder\n}\n\n// CountByQuery mocks base method.\nfunc (m *MockGenericClient) CountByQuery(query string) (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountByQuery\", query)\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountByQuery indicates an expected call of CountByQuery.\nfunc (mr *MockGenericClientMockRecorder) CountByQuery(query any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountByQuery\", reflect.TypeOf((*MockGenericClient)(nil).CountByQuery), query)\n}\n\n// GetTableName mocks base method.\nfunc (m *MockGenericClient) GetTableName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTableName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetTableName indicates an expected call of GetTableName.\nfunc (mr *MockGenericClientMockRecorder) GetTableName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTableName\", reflect.TypeOf((*MockGenericClient)(nil).GetTableName))\n}\n\n// Search mocks base method.\nfunc (m *MockGenericClient) Search(request *SearchRequest) (*SearchResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Search\", request)\n\tret0, _ := ret[0].(*SearchResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Search indicates an expected call of Search.\nfunc (mr *MockGenericClientMockRecorder) Search(request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Search\", reflect.TypeOf((*MockGenericClient)(nil).Search), request)\n}\n\n// SearchAggr mocks base method.\nfunc (m *MockGenericClient) SearchAggr(request *SearchRequest) (AggrResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SearchAggr\", request)\n\tret0, _ := ret[0].(AggrResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SearchAggr indicates an expected call of SearchAggr.\nfunc (mr *MockGenericClientMockRecorder) SearchAggr(request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SearchAggr\", reflect.TypeOf((*MockGenericClient)(nil).SearchAggr), request)\n}\n"
  },
  {
    "path": "common/pinot/interfaces.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination generic_client_mock.go -self_package github.com/uber/cadence/common/pinot\n\npackage pinot\n\nimport p \"github.com/uber/cadence/common/persistence\"\n\ntype (\n\t// GenericClient is a generic interface for all versions of Pinot clients\n\tGenericClient interface {\n\t\t// Search API is only for supporting various List[Open/Closed]WorkflowExecutions(ByXyz).\n\t\t// Use SearchByQuery or ScanByQuery for generic purpose searching.\n\t\tSearch(request *SearchRequest) (*SearchResponse, error)\n\t\tSearchAggr(request *SearchRequest) (AggrResponse, error)\n\t\t// CountByQuery is for returning the count of workflow executions that match the query\n\t\tCountByQuery(query string) (int64, error)\n\t\tGetTableName() string\n\t}\n\n\t// IsRecordValidFilter is a function to filter visibility records\n\tIsRecordValidFilter func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool\n\n\t// SearchRequest is request for Search\n\tSearchRequest struct {\n\t\tQuery           string\n\t\tIsOpen          bool\n\t\tFilter          IsRecordValidFilter\n\t\tMaxResultWindow int\n\t\tListRequest     *p.InternalListWorkflowExecutionsRequest\n\t}\n\n\t// SearchResponse is a response to Search, SearchByQuery and ScanByQuery\n\tSearchResponse = p.InternalListWorkflowExecutionsResponse\n\tAggrResponse   [][]interface{}\n)\n"
  },
  {
    "path": "common/pinot/page_token.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// PinotVisibilityPageToken holds the paging token for Pinot\n\tPinotVisibilityPageToken struct {\n\t\tFrom int\n\t}\n)\n\n// DeserializePageToken return the structural token\nfunc DeserializePageToken(data []byte) (*PinotVisibilityPageToken, error) {\n\tvar token PinotVisibilityPageToken\n\tdec := json.NewDecoder(bytes.NewReader(data))\n\tdec.UseNumber()\n\terr := dec.Decode(&token)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"unable to deserialize page token. err: %v\", err),\n\t\t}\n\t}\n\treturn &token, nil\n}\n\n// SerializePageToken return the token blob\nfunc SerializePageToken(token *PinotVisibilityPageToken) ([]byte, error) {\n\tdata, err := json.Marshal(token)\n\tif err != nil {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"unable to serialize page token. err: %v\", err),\n\t\t}\n\t}\n\treturn data, nil\n}\n\n// GetNextPageToken returns the structural token with nil handling\nfunc GetNextPageToken(token []byte) (*PinotVisibilityPageToken, error) {\n\tvar result *PinotVisibilityPageToken\n\tvar err error\n\tif len(token) > 0 {\n\t\tresult, err = DeserializePageToken(token)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tresult = &PinotVisibilityPageToken{}\n\t}\n\treturn result, nil\n}\n"
  },
  {
    "path": "common/pinot/page_token_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestSerializePageToken(t *testing.T) {\n\ttoken := &PinotVisibilityPageToken{\n\t\tFrom: 1,\n\t}\n\n\ttests := map[string]struct {\n\t\ttoken          *PinotVisibilityPageToken\n\t\texpectedOutput []byte\n\t\texpectedError  error\n\t}{\n\t\t\"Case2: normal case with nil response\": {\n\t\t\ttoken:          token,\n\t\t\texpectedOutput: []byte(`{\"From\":1}`),\n\t\t\texpectedError:  nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactualOutput, err := SerializePageToken(test.token)\n\t\t\tassert.Equal(t, test.expectedOutput, actualOutput)\n\t\t\tassert.Nil(t, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/pinot/pinotQueryValidator.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// VisibilityQueryValidator for sql query validation\ntype VisibilityQueryValidator struct {\n\tvalidSearchAttributes      dynamicproperties.MapPropertyFn\n\tpinotOptimizedQueryColumns dynamicproperties.MapPropertyFn\n}\n\nvar timeSystemKeys = map[string]bool{\n\t\"StartTime\":     true,\n\t\"CloseTime\":     true,\n\t\"ExecutionTime\": true,\n\t\"UpdateTime\":    true,\n}\n\n// NewPinotQueryValidator create VisibilityQueryValidator\nfunc NewPinotQueryValidator(validSearchAttributes dynamicproperties.MapPropertyFn, pinotOptimizedQueryColumns dynamicproperties.MapPropertyFn) *VisibilityQueryValidator {\n\treturn &VisibilityQueryValidator{\n\t\tvalidSearchAttributes:      validSearchAttributes,\n\t\tpinotOptimizedQueryColumns: pinotOptimizedQueryColumns,\n\t}\n}\n\n// ValidateQuery validates that search attributes in the query and returns modified query.\nfunc (qv *VisibilityQueryValidator) ValidateQuery(whereClause string) (string, error) {\n\tif len(whereClause) != 0 {\n\t\t// Build a placeholder query that allows us to easily parse the contents of the where clause.\n\t\t// IMPORTANT: This query is never executed, it is just used to parse and validate whereClause\n\t\tvar placeholderQuery string\n\t\twhereClause := strings.TrimSpace(whereClause)\n\t\tif common.IsJustOrderByClause(whereClause) { // just order by\n\t\t\tplaceholderQuery = fmt.Sprintf(\"SELECT * FROM dummy %s\", whereClause)\n\t\t} else {\n\t\t\tplaceholderQuery = fmt.Sprintf(\"SELECT * FROM dummy WHERE %s\", whereClause)\n\t\t}\n\n\t\tstmt, err := sqlparser.Parse(placeholderQuery)\n\t\tif err != nil {\n\t\t\treturn \"\", &types.BadRequestError{Message: \"Invalid query: \" + err.Error()}\n\t\t}\n\n\t\tsel, ok := stmt.(*sqlparser.Select)\n\t\tif !ok {\n\t\t\treturn \"\", &types.BadRequestError{Message: \"Invalid select query.\"}\n\t\t}\n\t\tbuf := sqlparser.NewTrackedBuffer(nil)\n\t\tres := \"\"\n\t\t// validate where expr\n\t\tif sel.Where != nil {\n\t\t\tres, err = qv.validateWhereExpr(sel.Where.Expr)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", &types.BadRequestError{Message: err.Error()}\n\t\t\t}\n\t\t}\n\n\t\tsel.OrderBy.Format(buf)\n\t\tres += buf.String()\n\t\treturn res, nil\n\t}\n\treturn whereClause, nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateWhereExpr(expr sqlparser.Expr) (string, error) {\n\tif expr == nil {\n\t\treturn \"\", nil\n\t}\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.AndExpr, *sqlparser.OrExpr:\n\t\treturn qv.validateAndOrExpr(expr)\n\tcase *sqlparser.ComparisonExpr:\n\t\treturn qv.validateComparisonExpr(expr)\n\tcase *sqlparser.RangeCond:\n\t\treturn qv.validateRangeExpr(expr)\n\tcase *sqlparser.ParenExpr:\n\t\treturn qv.validateWhereExpr(expr.Expr)\n\tdefault:\n\t\treturn \"\", errors.New(\"invalid where clause\")\n\t}\n}\n\n// for \"between...and...\" only\n// <, >, >=, <= are included in validateComparisonExpr()\nfunc (qv *VisibilityQueryValidator) validateRangeExpr(expr sqlparser.Expr) (string, error) {\n\tbuf := sqlparser.NewTrackedBuffer(nil)\n\trangeCond := expr.(*sqlparser.RangeCond)\n\tcolName, ok := rangeCond.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid range expression: fail to get colname\")\n\t}\n\tcolNameStr := colName.Name.String()\n\n\tif !qv.IsValidSearchAttributes(colNameStr) {\n\t\treturn \"\", fmt.Errorf(\"invalid search attribute %q\", colNameStr)\n\t}\n\n\tif definition.IsSystemIndexedKey(colNameStr) {\n\t\tif _, ok = timeSystemKeys[colNameStr]; ok {\n\t\t\tif lowerBound, ok := rangeCond.From.(*sqlparser.SQLVal); ok {\n\t\t\t\ttrimmed, err := trimTimeFieldValueFromNanoToMilliSeconds(lowerBound)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn \"\", fmt.Errorf(\"trim time field %s got error: %w\", colNameStr, err)\n\t\t\t\t}\n\t\t\t\trangeCond.From = trimmed\n\t\t\t}\n\t\t\tif upperBound, ok := rangeCond.To.(*sqlparser.SQLVal); ok {\n\t\t\t\ttrimmed, err := trimTimeFieldValueFromNanoToMilliSeconds(upperBound)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn \"\", fmt.Errorf(\"trim time field %s got error: %w\", colNameStr, err)\n\t\t\t\t}\n\t\t\t\trangeCond.To = trimmed\n\t\t\t}\n\t\t}\n\t\texpr.Format(buf)\n\t\treturn buf.String(), nil\n\t}\n\n\t// lowerBound, ok := rangeCond.From.(*sqlparser.ColName)\n\tlowerBound, ok := rangeCond.From.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid range expression: fail to get lowerbound\")\n\t}\n\tlowerBoundString := string(lowerBound.Val)\n\n\tupperBound, ok := rangeCond.To.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid range expression: fail to get upperbound\")\n\t}\n\tupperBoundString := string(upperBound.Val)\n\n\treturn fmt.Sprintf(\"(JSON_MATCH(Attr, '\\\"$.%s\\\" is not null') \"+\n\t\t\"AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.%s') AS INT) >= %s \"+\n\t\t\"AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.%s') AS INT) <= %s)\", colNameStr, colNameStr, lowerBoundString, colNameStr, upperBoundString), nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateAndOrExpr(expr sqlparser.Expr) (string, error) {\n\tvar leftExpr sqlparser.Expr\n\tvar rightExpr sqlparser.Expr\n\tisAnd := false\n\n\tswitch expr := expr.(type) {\n\tcase *sqlparser.AndExpr:\n\t\tleftExpr = expr.Left\n\t\trightExpr = expr.Right\n\t\tisAnd = true\n\tcase *sqlparser.OrExpr:\n\t\tleftExpr = expr.Left\n\t\trightExpr = expr.Right\n\t}\n\n\tleftRes, err := qv.validateWhereExpr(leftExpr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\trightRes, err := qv.validateWhereExpr(rightExpr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tif isAnd {\n\t\treturn fmt.Sprintf(\"%s and %s\", leftRes, rightRes), nil\n\t}\n\n\treturn fmt.Sprintf(\"(%s or %s)\", leftRes, rightRes), nil\n}\n\nfunc (qv *VisibilityQueryValidator) validateComparisonExpr(expr sqlparser.Expr) (string, error) {\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\n\tcolName, ok := comparisonExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid comparison expression, left\")\n\t}\n\n\tcolNameStr := colName.Name.String()\n\n\tif !qv.IsValidSearchAttributes(colNameStr) {\n\t\treturn \"\", fmt.Errorf(\"invalid search attribute %q\", colNameStr)\n\t}\n\n\t// Case1: it is system key\n\t// this means that we don't need to change the structure of the query,\n\t// just need to check if a value == \"missing\"\n\tif definition.IsSystemIndexedKey(colNameStr) {\n\t\treturn qv.processSystemKey(expr)\n\t}\n\t// Case2: when a value is not system key\n\t// This means, the value is from Attr so that we need to change the query to be a Json index format\n\treturn qv.processCustomKey(expr)\n}\n\n// IsValidSearchAttributes return true if key is registered\nfunc (qv *VisibilityQueryValidator) IsValidSearchAttributes(key string) bool {\n\tvalidAttr := qv.validSearchAttributes()\n\t_, isValidKey := validAttr[key]\n\treturn isValidKey\n}\n\n// IsOptimizedQueryColumn return true if colNameStr is in the optimized query columns\nfunc (qv *VisibilityQueryValidator) IsOptimizedQueryColumn(colNameStr string) bool {\n\tif qv.pinotOptimizedQueryColumns() != nil {\n\t\toptimizedQueryColumns := qv.pinotOptimizedQueryColumns()\n\t\t_, isOptimizedQueryColumn := optimizedQueryColumns[colNameStr]\n\t\treturn isOptimizedQueryColumn\n\t}\n\treturn false\n}\n\nfunc (qv *VisibilityQueryValidator) processSystemBoolKey(colNameStr string, comparisonExpr sqlparser.ComparisonExpr) (string, error) {\n\t// case1: isCron = false\n\tcolVal, ok := comparisonExpr.Right.(sqlparser.BoolVal)\n\tif !ok {\n\t\t// case2: isCron = \"false\" or isCron = 'false'\n\t\tsqlVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\t\tif !ok {\n\t\t\treturn \"\", fmt.Errorf(\"failed to process a bool key to SQLVal: %v\", comparisonExpr.Right)\n\t\t}\n\t\tcolValStr := string(sqlVal.Val)\n\t\tif strings.ToLower(colValStr) != \"false\" && strings.ToLower(colValStr) != \"true\" {\n\t\t\treturn \"\", fmt.Errorf(\"invalid bool value in pinot_query_validator: %s\", colValStr)\n\t\t}\n\t\treturn fmt.Sprintf(\"%s = %s\", colNameStr, colValStr), nil\n\t}\n\treturn fmt.Sprintf(\"%s = %v\", colNameStr, colVal), nil\n}\n\nfunc (qv *VisibilityQueryValidator) processSystemKey(expr sqlparser.Expr) (string, error) {\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\tbuf := sqlparser.NewTrackedBuffer(nil)\n\n\tcolName, ok := comparisonExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", fmt.Errorf(\"left comparison is invalid: %v\", comparisonExpr.Left)\n\t}\n\tcolNameStr := colName.Name.String()\n\n\t// handle system bool key\n\tif definition.IsSystemBoolKey(colNameStr) {\n\t\treturn qv.processSystemBoolKey(colNameStr, *comparisonExpr)\n\t}\n\n\tif comparisonExpr.Operator == sqlparser.LikeStr {\n\t\tcolVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\t\tif !ok {\n\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid: %v\", comparisonExpr.Right)\n\t\t}\n\t\tcolValStr := string(colVal.Val)\n\t\treturn fmt.Sprintf(\"TEXT_MATCH(%s, '/.*%s.*/')\", colNameStr, colValStr), nil\n\t}\n\n\tif comparisonExpr.Operator != sqlparser.EqualStr && comparisonExpr.Operator != sqlparser.NotEqualStr {\n\t\tif _, ok := timeSystemKeys[colNameStr]; ok {\n\t\t\tsqlVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\t\t\tif !ok {\n\t\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid: %v\", comparisonExpr.Right)\n\t\t\t}\n\t\t\ttrimmed, err := trimTimeFieldValueFromNanoToMilliSeconds(sqlVal)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", fmt.Errorf(\"trim time field %s got error: %w\", colNameStr, err)\n\t\t\t}\n\t\t\tcomparisonExpr.Right = trimmed\n\t\t}\n\n\t\texpr.Format(buf)\n\t\treturn buf.String(), nil\n\t}\n\t// need to deal with missing value e.g. CloseTime = missing\n\t// Question: why is the right side is sometimes a type of \"colName\", and sometimes a type of \"SQLVal\"?\n\t// Answer: for any value, sqlParser will treat any string that doesn't surrounded by single quote as ColName;\n\t// any string that surrounded by single quote as SQLVal\n\t_, ok = comparisonExpr.Right.(*sqlparser.SQLVal)\n\tif !ok { // this means, the value is a string, and not surrounded by single qoute, which means, val = missing\n\t\tcolVal, ok := comparisonExpr.Right.(*sqlparser.ColName)\n\t\tif !ok {\n\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid: %v\", comparisonExpr.Right)\n\t\t}\n\t\tcolValStr := colVal.Name.String()\n\n\t\t// double check if val is not missing\n\t\tif colValStr != \"missing\" {\n\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid string value: %s\", colValStr)\n\t\t}\n\n\t\tvar newColVal string\n\t\tif strings.ToLower(colNameStr) == \"historylength\" {\n\t\t\tnewColVal = \"0\"\n\t\t} else {\n\t\t\tnewColVal = \"-1\" // -1 is the default value for all Closed workflows related fields\n\t\t}\n\t\tcomparisonExpr.Right = &sqlparser.SQLVal{\n\t\t\tType: sqlparser.IntVal, // or sqlparser.StrVal if you need to assign a string\n\t\t\tVal:  []byte(newColVal),\n\t\t}\n\t} else {\n\t\tif _, ok := timeSystemKeys[colNameStr]; ok {\n\t\t\tsqlVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\t\t\tif !ok {\n\t\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid/missing. key %s, right expr %v\", colNameStr, comparisonExpr.Right)\n\t\t\t}\n\t\t\ttrimmed, err := trimTimeFieldValueFromNanoToMilliSeconds(sqlVal)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", fmt.Errorf(\"trim time field %s got error: %w\", colNameStr, err)\n\t\t\t}\n\t\t\tcomparisonExpr.Right = trimmed\n\t\t} else if colNameStr == \"CloseStatus\" {\n\t\t\tsqlVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\t\t\tif !ok {\n\t\t\t\treturn \"\", fmt.Errorf(\"right comparison is invalid: %v\", comparisonExpr.Right)\n\t\t\t}\n\t\t\tcloseStatus, err := parseCloseStatus(sqlVal)\n\t\t\tif err != nil {\n\t\t\t\treturn \"\", fmt.Errorf(\"parse CloseStatus field got error: %w\", err)\n\t\t\t}\n\t\t\tcomparisonExpr.Right = closeStatus\n\t\t}\n\t}\n\n\t// For this branch, we still have a sqlExpr type. So need to use a buf to return the string\n\tcomparisonExpr.Format(buf)\n\treturn buf.String(), nil\n}\n\nfunc (qv *VisibilityQueryValidator) processInClause(expr sqlparser.Expr) (string, error) {\n\tcomparisonExpr, ok := expr.(*sqlparser.ComparisonExpr)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid IN expression\")\n\t}\n\n\tcolName, ok := comparisonExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid IN expression, left\")\n\t}\n\n\tcolNameStr := colName.Name.String()\n\tvalTuple, ok := comparisonExpr.Right.(sqlparser.ValTuple)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid IN expression, right\")\n\t}\n\n\tvalues := make([]string, len(valTuple))\n\tfor i, val := range valTuple {\n\t\tsqlVal, ok := val.(*sqlparser.SQLVal)\n\t\tif !ok {\n\t\t\treturn \"\", errors.New(\"invalid IN expression, value\")\n\t\t}\n\t\tvalues[i] = \"''\" + string(sqlVal.Val) + \"''\"\n\t}\n\n\treturn fmt.Sprintf(\"JSON_MATCH(Attr, '\\\"$.%s\\\" IN (%s)') or JSON_MATCH(Attr, '\\\"$.%s[*]\\\" IN (%s)')\",\n\t\tcolNameStr, strings.Join(values, \",\"), colNameStr, strings.Join(values, \",\")), nil\n}\n\nfunc (qv *VisibilityQueryValidator) processCustomKey(expr sqlparser.Expr) (string, error) {\n\tcomparisonExpr := expr.(*sqlparser.ComparisonExpr)\n\n\tcolName, ok := comparisonExpr.Left.(*sqlparser.ColName)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid comparison expression, left\")\n\t}\n\n\tcolNameStr := colName.Name.String()\n\n\t// check type: if is IndexedValueTypeString, change to like statement for partial match\n\tvalType, ok := qv.validSearchAttributes()[colNameStr]\n\tif !ok {\n\t\treturn \"\", fmt.Errorf(\"invalid search attribute\")\n\t}\n\tuseOptimizedQuery := qv.IsOptimizedQueryColumn(colNameStr)\n\n\t// process IN clause in json indexed col: Attr\n\toperator := strings.ToLower(comparisonExpr.Operator)\n\tif operator == sqlparser.InStr {\n\t\treturn qv.processInClause(expr)\n\t}\n\n\t// get the column value\n\tcolVal, ok := comparisonExpr.Right.(*sqlparser.SQLVal)\n\tif !ok {\n\t\treturn \"\", errors.New(\"invalid comparison expression, right\")\n\t}\n\n\t// get the value type\n\tindexValType := common.ConvertIndexedValueTypeToInternalType(valType, log.NewNoop())\n\tcolValStr := string(colVal.Val)\n\n\tswitch indexValType {\n\tcase types.IndexedValueTypeString:\n\t\treturn processCustomString(operator, colNameStr, colValStr, useOptimizedQuery), nil\n\tcase types.IndexedValueTypeKeyword:\n\t\treturn processCustomKeyword(operator, colNameStr, colValStr, useOptimizedQuery), nil\n\tcase types.IndexedValueTypeDatetime:\n\t\tvar err error\n\t\tcolVal, err = trimTimeFieldValueFromNanoToMilliSeconds(colVal)\n\t\tif err != nil {\n\t\t\treturn \"\", fmt.Errorf(\"trim time field %s got error: %w\", colNameStr, err)\n\t\t}\n\t\tcolValStr := string(colVal.Val)\n\t\treturn processCustomNum(operator, colNameStr, colValStr, \"BIGINT\"), nil\n\tcase types.IndexedValueTypeDouble:\n\t\treturn processCustomNum(operator, colNameStr, colValStr, \"DOUBLE\"), nil\n\tcase types.IndexedValueTypeInt:\n\t\treturn processCustomNum(operator, colNameStr, colValStr, \"INT\"), nil\n\tdefault:\n\t\treturn processEqual(colNameStr, colValStr), nil\n\t}\n}\n\nfunc processCustomNum(operator string, colNameStr string, colValStr string, valType string) string {\n\tif operator == sqlparser.EqualStr {\n\t\treturn processEqual(colNameStr, colValStr)\n\t}\n\treturn fmt.Sprintf(\"(JSON_MATCH(Attr, '\\\"$.%s\\\" is not null') \"+\n\t\t\"AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.%s') AS %s) %s %s)\", colNameStr, colNameStr, valType, operator, colValStr)\n}\n\nfunc processEqual(colNameStr string, colValStr string) string {\n\treturn fmt.Sprintf(\"JSON_MATCH(Attr, '\\\"$.%s\\\"=''%s''')\", colNameStr, colValStr)\n}\n\nfunc processCustomKeyword(operator string, colNameStr string, colValStr string, useOptimizedQuery bool) string {\n\tif useOptimizedQuery {\n\t\t// replace \"-\" with \"\" for optimized query so Pinot can tokenize the whole UUID\n\t\tfilteredColValStr := strings.ReplaceAll(colValStr, \"-\", \"\")\n\t\treturn fmt.Sprintf(\"%s %s '%s'\", colNameStr, operator, filteredColValStr)\n\t}\n\n\tif colValStr == \"\" {\n\t\t// partial match for an empty string (still it will only match empty string)\n\t\t// so it equals to exact match for an empty string\n\t\treturn processCustomString(operator, colNameStr, colValStr, useOptimizedQuery)\n\t}\n\n\tconnector := \"or\"\n\tif operator == \"!=\" {\n\t\tconnector = \"and\"\n\t}\n\treturn fmt.Sprintf(\"(JSON_MATCH(Attr, '\\\"$.%s\\\"%s''%s''') %s JSON_MATCH(Attr, '\\\"$.%s[*]\\\"%s''%s'''))\",\n\t\tcolNameStr, operator, colValStr, connector, colNameStr, operator, colValStr)\n}\n\nfunc processCustomString(operator string, colNameStr string, colValStr string, useOptimizedQuery bool) string {\n\tif useOptimizedQuery {\n\t\t// replace \"-\" with \"\" for optimized query so Pinot can tokenize the whole UUID\n\t\tfilteredColValStr := strings.ReplaceAll(colValStr, \"-\", \"\")\n\t\treturn fmt.Sprintf(\"%s %s '%s'\", colNameStr, operator, filteredColValStr)\n\t}\n\n\tnotEqual := \"\"\n\tif operator == \"!=\" {\n\t\tnotEqual = \"NOT \"\n\t}\n\t// handle edge case\n\tif colValStr == \"\" {\n\t\treturn fmt.Sprintf(\"JSON_MATCH(Attr, '\\\"$.%s\\\" is not null') \"+\n\t\t\t\"AND %sJSON_MATCH(Attr, 'REGEXP_LIKE(\\\"$.%s\\\", ''^$'')')\", colNameStr, notEqual, colNameStr)\n\t}\n\treturn fmt.Sprintf(\"JSON_MATCH(Attr, '\\\"$.%s\\\" is not null') \"+\n\t\t\"AND %sJSON_MATCH(Attr, 'REGEXP_LIKE(\\\"$.%s\\\", ''.*%s.*'')')\", colNameStr, notEqual, colNameStr, colValStr)\n}\n\nfunc trimTimeFieldValueFromNanoToMilliSeconds(original *sqlparser.SQLVal) (*sqlparser.SQLVal, error) {\n\t// Convert the SQLVal to a string\n\tvalStr := string(original.Val)\n\tnewVal, err := parseTime(valStr)\n\tif err != nil {\n\t\treturn original, fmt.Errorf(\"error: failed to parse int from SQLVal %s\", valStr)\n\t}\n\n\t// Convert the new value back to SQLVal\n\treturn &sqlparser.SQLVal{\n\t\tType: sqlparser.IntVal,\n\t\tVal:  []byte(strconv.FormatInt(newVal, 10)),\n\t}, nil\n}\n\nfunc parseTime(timeStr string) (int64, error) {\n\tif len(timeStr) == 0 {\n\t\treturn 0, errors.New(\"invalid time string\")\n\t}\n\n\t// try to parse\n\tparsedTime, err := time.Parse(time.RFC3339, timeStr)\n\tif err == nil {\n\t\treturn parsedTime.UnixMilli(), nil\n\t}\n\n\t// treat as raw time\n\tvalInt, err := strconv.ParseInt(timeStr, 10, 64)\n\tif err == nil {\n\t\tvar newVal int64\n\t\tif valInt < 0 { // exclude open workflow which time field will be -1\n\t\t\tnewVal = valInt\n\t\t} else if len(timeStr) > 13 { // Assuming nanoseconds if more than 13 digits\n\t\t\tnewVal = valInt / 1000000 // Convert time to milliseconds\n\t\t} else {\n\t\t\tnewVal = valInt\n\t\t}\n\t\treturn newVal, nil\n\t}\n\n\treturn 0, errors.New(\"invalid time string\")\n}\n\nfunc parseCloseStatus(original *sqlparser.SQLVal) (*sqlparser.SQLVal, error) {\n\tstatusStr := string(original.Val)\n\n\t// first check if already in int64 format\n\tif status, err := strconv.ParseInt(statusStr, 10, 64); err == nil {\n\t\t// Instead of returning the original value, return a new SQLVal that holds the integer value\n\t\t// Or it will fail the case CloseStatus = '1'\n\t\treturn &sqlparser.SQLVal{\n\t\t\tType: sqlparser.IntVal,\n\t\t\tVal:  []byte(strconv.FormatInt(status, 10)),\n\t\t}, nil\n\t}\n\n\t// try to parse close status string\n\tvar parsedStatus types.WorkflowExecutionCloseStatus\n\terr := parsedStatus.UnmarshalText([]byte(statusStr))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &sqlparser.SQLVal{\n\t\tType: sqlparser.IntVal,\n\t\tVal:  []byte(strconv.FormatInt(int64(parsedStatus), 10)),\n\t}, nil\n}\n"
  },
  {
    "path": "common/pinot/pinotQueryValidator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/xwb1989/sqlparser\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestValidateQuery(t *testing.T) {\n\ttests := map[string]struct {\n\t\tquery     string\n\t\tvalidated string\n\t\terr       string\n\t}{\n\t\t\"Case1: empty query\": {\n\t\t\tquery:     \"\",\n\t\t\tvalidated: \"\",\n\t\t},\n\t\t\"Case2-1: simple query\": {\n\t\t\tquery:     \"WorkflowID = 'wid'\",\n\t\t\tvalidated: \"WorkflowID = 'wid'\",\n\t\t},\n\t\t\"Case2-2: simple query with partial match\": {\n\t\t\tquery:     \"WorkflowID like 'wid'\",\n\t\t\tvalidated: \"TEXT_MATCH(WorkflowID, '/.*wid.*/')\",\n\t\t},\n\t\t\"Case2-3: invalid simple query with partial match\": {\n\t\t\tquery: \"WorkflowID like wid\",\n\t\t\terr:   \"right comparison is invalid: &{<nil> wid { }}\"},\n\t\t\"Case3-1: query with custom field\": {\n\t\t\tquery:     \"CustomStringField = 'custom'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*custom.*'')')`,\n\t\t},\n\t\t\"Case3-2: query with custom field not equal\": {\n\t\t\tquery:     \"CustomStringField != 'custom'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*custom.*'')')`,\n\t\t},\n\t\t\"Case3-3: query with custom field value is empty\": {\n\t\t\tquery:     \"CustomStringField = ''\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''^$'')')`,\n\t\t},\n\t\t\"Case3-4: query with custom field not equal to empty\": {\n\t\t\tquery:     \"CustomStringField != ''\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''^$'')')`,\n\t\t},\n\t\t\"Case4: custom field query with or in string\": {\n\t\t\tquery:     \"CustomStringField='Or'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*Or.*'')')`,\n\t\t},\n\t\t\"Case5: custom keyword field query\": {\n\t\t\tquery:     \"CustomKeywordField = 'custom'\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomKeywordField\"=''custom''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"=''custom'''))`,\n\t\t},\n\t\t\"Case6-1: complex query I: with parenthesis\": {\n\t\t\tquery:     \"(CustomStringField = 'custom and custom2 or custom3 order by') or CustomIntField between 1 and 10\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*custom and custom2 or custom3 order by.*'')') or (JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) >= 1 AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) <= 10))`,\n\t\t},\n\t\t\"Case6-2: complex query II: with only system keys\": {\n\t\t\tquery:     \"DomainID = 'd-id' and (RunID = 'run-id' or WorkflowID = 'wid')\",\n\t\t\tvalidated: \"DomainID = 'd-id' and (RunID = 'run-id' or WorkflowID = 'wid')\",\n\t\t},\n\t\t\"Case6-3: complex query III: operation priorities\": {\n\t\t\tquery:     \"DomainID = 'd-id' or RunID = 'run-id' and WorkflowID = 'wid'\",\n\t\t\tvalidated: \"(DomainID = 'd-id' or RunID = 'run-id' and WorkflowID = 'wid')\",\n\t\t},\n\t\t\"Case6-4: complex query IV\": {\n\t\t\tquery:     \"WorkflowID = 'wid' and (CustomStringField = 'custom and custom2 or custom3 order by' or CustomIntField between 1 and 10)\",\n\t\t\tvalidated: `WorkflowID = 'wid' and (JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*custom and custom2 or custom3 order by.*'')') or (JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) >= 1 AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) <= 10))`,\n\t\t},\n\t\t\"Case6-5: complex query with partial match\": {\n\t\t\tquery:     \"RunID like '123' or WorkflowID like '123'\",\n\t\t\tvalidated: \"(TEXT_MATCH(RunID, '/.*123.*/') or TEXT_MATCH(WorkflowID, '/.*123.*/'))\",\n\t\t},\n\t\t\"Case7: invalid sql query\": {\n\t\t\tquery: \"Invalid SQL\",\n\t\t\terr:   \"Invalid query: syntax error at position 38 near 'sql'\",\n\t\t},\n\t\t\"Case8-1: query with missing val\": {\n\t\t\tquery:     \"CloseTime = missing\",\n\t\t\tvalidated: \"CloseTime = -1\",\n\t\t},\n\t\t\"Case8-2: query with not missing case\": {\n\t\t\tquery:     \"CloseTime != missing\",\n\t\t\tvalidated: \"CloseTime != -1\",\n\t\t},\n\t\t\"Case8-3: query with custom attr\": {\n\t\t\tquery: \"CustomKeywordField = missing\",\n\t\t\terr:   \"invalid comparison expression, right\",\n\t\t},\n\t\t\"Case8-4: query with custom keyword field not equal\": {\n\t\t\tquery:     \"CustomKeywordField != 0\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomKeywordField\"!=''0''') and JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"!=''0'''))`,\n\t\t},\n\t\t\"Case9: invalid where expression\": {\n\t\t\tquery: \"InvalidWhereExpr\",\n\t\t\terr:   \"invalid where clause\",\n\t\t},\n\t\t\"Case10: invalid search attribute\": {\n\t\t\tquery: \"Invalid = 'a' and 1 < 2\",\n\t\t\terr:   `invalid search attribute \"Invalid\"`,\n\t\t},\n\t\t\"Case11-1: order by clause\": {\n\t\t\tquery:     \"order by CloseTime desc\",\n\t\t\tvalidated: \" order by CloseTime desc\",\n\t\t},\n\t\t\"Case11-2: only order by clause with custom field\": {\n\t\t\tquery:     \"order by CustomIntField desc\",\n\t\t\tvalidated: \" order by CustomIntField desc\",\n\t\t},\n\t\t\"Case11-3: order by clause with custom field\": {\n\t\t\tquery:     \"WorkflowID = 'wid' order by CloseTime desc\",\n\t\t\tvalidated: \"WorkflowID = 'wid' order by CloseTime desc\",\n\t\t},\n\t\t\"Case12-1: security SQL injection - with another statement\": {\n\t\t\tquery: \"WorkflowID = 'wid'; SELECT * FROM important_table;\",\n\t\t\terr:   \"Invalid query: syntax error at position 53 near 'select'\",\n\t\t},\n\t\t\"Case12-2: security SQL injection - with union\": {\n\t\t\tquery: \"WorkflowID = 'wid' union select * from dummy\",\n\t\t\terr:   \"Invalid select query.\",\n\t\t},\n\t\t\"Case13: or clause\": {\n\t\t\tquery:     \"CustomIntField = 1 or CustomIntField = 2\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomIntField\"=''1''') or JSON_MATCH(Attr, '\"$.CustomIntField\"=''2'''))`,\n\t\t},\n\t\t\"Case14-1: range query: custom filed\": {\n\t\t\tquery:     \"CustomIntField BETWEEN 1 AND 2\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) >= 1 AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) <= 2)`,\n\t\t},\n\t\t\"Case14-2: range query: system filed\": {\n\t\t\tquery:     \"NumClusters BETWEEN 1 AND 2\",\n\t\t\tvalidated: \"NumClusters between 1 and 2\",\n\t\t},\n\t\t\"Case15-1: custom date attribute less than\": {\n\t\t\tquery:     \"CustomDatetimeField < 1697754674\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomDatetimeField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomDatetimeField') AS BIGINT) < 1697754674)`,\n\t\t},\n\t\t\"Case15-2: custom date attribute greater than or equal to\": {\n\t\t\tquery:     \"CustomDatetimeField >= 1697754674\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomDatetimeField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomDatetimeField') AS BIGINT) >= 1697754674)`,\n\t\t},\n\t\t\"Case15-3: system date attribute greater than or equal to\": {\n\t\t\tquery:     \"StartTime >= 1697754674\",\n\t\t\tvalidated: \"StartTime >= 1697754674\",\n\t\t},\n\t\t\"Case15-4: unix nano converts to milli seconds for equal statements\": {\n\t\t\tquery:     \"StartTime = 1707319950934000128\",\n\t\t\tvalidated: \"StartTime = 1707319950934\",\n\t\t},\n\t\t\"Case15-5: unix nano converts to milli seconds for unequal statements query\": {\n\t\t\tquery:     \"StartTime > 1707319950934000128\",\n\t\t\tvalidated: \"StartTime > 1707319950934\",\n\t\t},\n\t\t\"Case15-6: open workflows\": {\n\t\t\tquery:     \"CloseTime = -1\",\n\t\t\tvalidated: \"CloseTime = -1\",\n\t\t},\n\t\t\"Case15-7: startTime for range query\": {\n\t\t\tquery:     \"StartTime BETWEEN 1707319950934000128 AND 1707319950935000128\",\n\t\t\tvalidated: \"StartTime between 1707319950934 and 1707319950935\",\n\t\t},\n\t\t\"Case15-8: invalid string for trim\": {\n\t\t\tquery:     \"CloseTime = abc\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"right comparison is invalid string value: abc\",\n\t\t},\n\t\t\"Case15-9: invalid value for trim\": {\n\t\t\tquery:     \"CloseTime = 123.45\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"trim time field CloseTime got error: error: failed to parse int from SQLVal 123.45\",\n\t\t},\n\t\t\"Case15-10: invalid from time for range query\": {\n\t\t\tquery:     \"StartTime BETWEEN 17.50 AND 1707319950935000128\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"trim time field StartTime got error: error: failed to parse int from SQLVal 17.50\",\n\t\t},\n\t\t\"Case15-11: invalid to time for range query\": {\n\t\t\tquery:     \"StartTime BETWEEN 1707319950934000128 AND 1707319950935000128.1\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"trim time field StartTime got error: error: failed to parse int from SQLVal 1707319950935000128.1\",\n\t\t},\n\t\t\"Case15-12: value already in milliseconds\": {\n\t\t\tquery:     \"StartTime = 170731995093\",\n\t\t\tvalidated: \"StartTime = 170731995093\",\n\t\t},\n\t\t\"Case15-13: value in raw string for equal statement\": {\n\t\t\tquery:     \"StartTime = '2024-02-07T15:32:30Z'\",\n\t\t\tvalidated: \"StartTime = 1707319950000\",\n\t\t},\n\t\t\"Case15-14: value in raw string for not equal statement\": {\n\t\t\tquery:     \"StartTime > '2024-02-07T15:32:30Z'\",\n\t\t\tvalidated: \"StartTime > 1707319950000\",\n\t\t},\n\t\t\"Case15-15: value in raw string for range statement\": {\n\t\t\tquery:     \"StartTime between '2024-02-07T15:32:30Z' and '2024-02-07T15:33:30Z'\",\n\t\t\tvalidated: \"StartTime between 1707319950000 and 1707320010000\",\n\t\t},\n\t\t\"Case15-16: combined time and missing case\": {\n\t\t\tquery:     \"CloseTime != missing and StartTime >= 1707662555754408145\",\n\t\t\tvalidated: \"CloseTime != -1 and StartTime >= 1707662555754\",\n\t\t},\n\t\t\"Case15-17: CustomDatetimeField with big int type case\": {\n\t\t\tquery:     \"CustomDatetimeField = 1707319950000\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomDatetimeField\"=''1707319950000''')`,\n\t\t},\n\t\t\"Case15-18: CustomDatetimeField with time.Time() type case\": {\n\t\t\tquery:     \"CustomDatetimeField = '2024-02-07T15:32:30Z'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomDatetimeField\"=''1707319950000''')`,\n\t\t},\n\t\t\"Case15-19: CustomDatetimeField with error case\": {\n\t\t\tquery:     \"CustomDatetimeField = 'test'\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"trim time field CustomDatetimeField got error: error: failed to parse int from SQLVal test\",\n\t\t},\n\t\t\"Case16-1: custom int attribute greater than or equal to\": {\n\t\t\tquery:     \"CustomIntField >= 0\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomIntField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomIntField') AS INT) >= 0)`,\n\t\t},\n\t\t\"Case16-2: custom double attribute greater than or equal to\": {\n\t\t\tquery:     \"CustomDoubleField >= 0\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomDoubleField\" is not null') AND CAST(JSON_EXTRACT_SCALAR(Attr, '$.CustomDoubleField') AS DOUBLE) >= 0)`,\n\t\t},\n\t\t\"Case17: custom keyword attribute greater than or equal to. Will return error run time\": {\n\t\t\tquery:     \"CustomKeywordField < 0\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomKeywordField\"<''0''') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\"<''0'''))`,\n\t\t},\n\t\t\"Case18: custom int order by. Will have errors at run time. Doesn't support for now\": {\n\t\t\tquery:     \"CustomIntField = 0 order by CustomIntField desc\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomIntField\"=''0''') order by CustomIntField desc`,\n\t\t},\n\t\t\"case19-1: close status parse string\": {\n\t\t\tquery:     \"CloseStatus = 'CONTINUED_AS_NEW'\",\n\t\t\tvalidated: \"CloseStatus = 4\",\n\t\t},\n\t\t\"case19-2: close status parse number\": {\n\t\t\tquery:     \"CloseStatus = '1'\",\n\t\t\tvalidated: \"CloseStatus = 1\",\n\t\t},\n\t\t\"case19-3: close status parse normal case\": {\n\t\t\tquery:     \"CloseStatus = 1\",\n\t\t\tvalidated: \"CloseStatus = 1\",\n\t\t},\n\t\t\"case20-1: in clause in Attr\": {\n\t\t\tquery:     \"CustomKeywordField in (123)\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" IN (''123'')') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\" IN (''123'')')`,\n\t\t},\n\t\t\"case20-2: in clause in Attr with multiple values\": {\n\t\t\tquery:     \"CustomKeywordField in (123, 456)\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" IN (''123'',''456'')') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\" IN (''123'',''456'')')`,\n\t\t},\n\t\t\"case20-3-1: in clause in Attr with a string value, double quote\": {\n\t\t\tquery:     `CustomKeywordField in (\"abc\")`,\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" IN (''abc'')') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\" IN (''abc'')')`,\n\t\t},\n\t\t\"case20-3-2: in clause in Attr with a string value, single quote\": {\n\t\t\tquery:     \"CustomKeywordField in ('abc')\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" IN (''abc'')') or JSON_MATCH(Attr, '\"$.CustomKeywordField[*]\" IN (''abc'')')`,\n\t\t},\n\t\t\"case20-4: in clause in Attr with invalid IN expression, value\": {\n\t\t\tquery:     \"CustomKeywordField in (abc)\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"invalid IN expression, value\",\n\t\t},\n\t\t\"case21-1: test bool value- system key- no quotes\": {\n\t\t\tquery:     \"IsCron = true\",\n\t\t\tvalidated: \"IsCron = true\",\n\t\t},\n\t\t\"case21-2: test bool value- system key- single quotes\": {\n\t\t\tquery:     \"IsCron = 'true'\",\n\t\t\tvalidated: \"IsCron = true\",\n\t\t},\n\t\t\"case21-3: test bool value- system key- double quotes\": {\n\t\t\tquery:     `IsCron = \"true\"`,\n\t\t\tvalidated: \"IsCron = true\",\n\t\t},\n\t\t\"case21-4: test bool value- system key- invalid value\": {\n\t\t\tquery:     \"IsCron = 1\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"invalid bool value in pinot_query_validator: 1\",\n\t\t},\n\t\t\"case21-5: test bool value- when it is not SQLBool and SQLVAl\": {\n\t\t\tquery:     \"IsCron = abc\",\n\t\t\tvalidated: \"\",\n\t\t\terr:       \"failed to process a bool key to SQLVal: &{<nil> abc { }}\",\n\t\t},\n\t\t\"case22-1: test not equal to a string field\": {\n\t\t\tquery:     \"CustomStringField != 'abc'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')')`,\n\t\t},\n\t\t\"case22-2: test not equal to an empty string\": {\n\t\t\tquery:     \"CustomStringField != ''\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''^$'')')`,\n\t\t},\n\t\t// ES also doesn't support this kind of query\n\t\t\"case22-3: custom string is missing\": {\n\t\t\tquery: \"CustomStringField is missing\",\n\t\t\terr:   \"Invalid query: syntax error at position 55 near 'missing'\",\n\t\t},\n\t\t// ES also doesn't support this kind of query\n\t\t\"case22-4: custom string is not missing\": {\n\t\t\tquery: \"CustomStringField is not missing\",\n\t\t\terr:   \"Invalid query: syntax error at position 59 near 'missing'\",\n\t\t},\n\t\t\"case22-5: 2 custom string not equal with and clause\": {\n\t\t\tquery:     \"CustomStringField != 'abc' AND CustomStringField != 'def'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')')`,\n\t\t},\n\t\t\"case22-6: 2 custom string, equal and not equal with and clause\": {\n\t\t\tquery:     \"CustomStringField = 'abc' AND CustomStringField != 'def'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')')`,\n\t\t},\n\t\t\"case22-7: 2 custom string, not equal and equal with and clause\": {\n\t\t\tquery:     \"CustomStringField != 'abc' AND CustomStringField = 'def'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')')`,\n\t\t},\n\t\t\"case22-8: 2 custom string equal with and clause\": {\n\t\t\tquery:     \"CustomStringField = 'abc' AND CustomStringField = 'def'\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') and JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')')`,\n\t\t},\n\t\t\"case22-9: 2 custom string not equal with or clause\": {\n\t\t\tquery:     \"CustomStringField != 'abc' OR CustomStringField != 'def'\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') or JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')'))`,\n\t\t},\n\t\t\"case22-10: 2 custom string, equal and not equal with or clause\": {\n\t\t\tquery:     \"CustomStringField = 'abc' OR CustomStringField != 'def'\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') or JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')'))`,\n\t\t},\n\t\t\"case22-11: 2 custom string, not equal and equal with or clause\": {\n\t\t\tquery:     \"CustomStringField != 'abc' OR CustomStringField = 'def'\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') or JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')'))`,\n\t\t},\n\t\t\"case22-12: 2 custom string equal with or clause\": {\n\t\t\tquery:     \"CustomStringField = 'abc' OR CustomStringField = 'def'\",\n\t\t\tvalidated: `(JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*abc.*'')') or JSON_MATCH(Attr, '\"$.CustomStringField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomStringField\", ''.*def.*'')'))`,\n\t\t},\n\t\t\"case23-1: custom keyword field is empty case\": {\n\t\t\tquery:     \"CustomKeywordField = ''\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" is not null') AND JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomKeywordField\", ''^$'')')`,\n\t\t},\n\t\t\"case23-2: custom keyword field is not empty case\": {\n\t\t\tquery:     \"CustomKeywordField != ''\",\n\t\t\tvalidated: `JSON_MATCH(Attr, '\"$.CustomKeywordField\" is not null') AND NOT JSON_MATCH(Attr, 'REGEXP_LIKE(\"$.CustomKeywordField\", ''^$'')')`,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tvalidSearchAttr := dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys())\n\t\t\tpinotOptimizedQueryColumns := dynamicproperties.GetMapPropertyFn(map[string]interface{}{\n\t\t\t\t\"CustomTestField\": \"test\",\n\t\t\t})\n\t\t\tqv := NewPinotQueryValidator(validSearchAttr, pinotOptimizedQueryColumns)\n\t\t\tvalidated, err := qv.ValidateQuery(test.query)\n\t\t\tif err != nil {\n\t\t\t\tassert.Equal(t, test.err, err.Error())\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.validated, validated)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestProcessInClause_FailedInputExprCases(t *testing.T) {\n\t// Define test cases\n\ttests := map[string]struct {\n\t\tinputExpr     sqlparser.Expr\n\t\texpectedError string\n\t}{\n\t\t\"case1: 'In' clause in Attr with invalid expr\": {\n\t\t\tinputExpr:     &sqlparser.SQLVal{Type: sqlparser.StrVal, Val: []byte(\"invalid\")},\n\t\t\texpectedError: \"invalid IN expression\",\n\t\t},\n\t\t\"case2: 'In' clause in Attr with invalid expr, left\": {\n\t\t\tinputExpr:     &sqlparser.ComparisonExpr{Operator: sqlparser.InStr},\n\t\t\texpectedError: \"invalid IN expression, left\",\n\t\t},\n\t\t\"case3: 'In' clause in Attr with invalid expr, right\": {\n\t\t\tinputExpr:     &sqlparser.ComparisonExpr{Operator: sqlparser.InStr, Left: &sqlparser.ColName{Name: sqlparser.NewColIdent(\"CustomKeywordField\")}},\n\t\t\texpectedError: \"invalid IN expression, right\",\n\t\t},\n\t}\n\n\t// Create a new VisibilityQueryValidator\n\tvalidSearchAttr := dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys())\n\tpinotOptimizedQueryColumns := dynamicproperties.GetMapPropertyFn(map[string]interface{}{\n\t\t\"CustomTestField\": \"test\",\n\t})\n\tqv := NewPinotQueryValidator(validSearchAttr, pinotOptimizedQueryColumns)\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Call processInClause with the input expression\n\t\t\t_, err := qv.processInClause(test.inputExpr)\n\n\t\t\t// Check that an error was returned and that the error message matches the expected error message\n\t\t\tif assert.Error(t, err) {\n\t\t\t\tassert.Contains(t, err.Error(), test.expectedError)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestParseTime(t *testing.T) {\n\tvar tests = []struct {\n\t\tname     string\n\t\ttimeStr  string\n\t\texpected int64\n\t\thasErr   bool\n\t}{\n\t\t{\"empty string\", \"\", 0, true},\n\t\t{\"valid RFC3339\", \"2024-02-07T15:32:30Z\", 1707319950000, false},\n\t\t{\"valid unix milli string\", \"1707319950000\", 1707319950000, false},\n\t\t{\"valid unix nano string\", \"1707319950000000000\", 1707319950000, false},\n\t\t{\"invalid string\", \"invalid\", 0, true},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tparsed, err := parseTime(tt.timeStr)\n\t\t\tassert.Equal(t, parsed, tt.expected)\n\t\t\tif tt.hasErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestParseCloseStatus(t *testing.T) {\n\ttests := []struct {\n\t\tinput       string\n\t\texpected    *sqlparser.SQLVal\n\t\texpectedErr bool\n\t}{\n\t\t{\n\t\t\tinput:       \"4\",\n\t\t\texpected:    &sqlparser.SQLVal{Type: sqlparser.IntVal, Val: []byte(\"4\")},\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tinput:       \"CANCELED\",\n\t\t\texpected:    &sqlparser.SQLVal{Type: sqlparser.IntVal, Val: []byte(\"2\")},\n\t\t\texpectedErr: false,\n\t\t},\n\t\t{\n\t\t\tinput:       \"invalid\",\n\t\t\texpected:    nil,\n\t\t\texpectedErr: true, // expected error,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.input, func(t *testing.T) {\n\t\t\toriginal := &sqlparser.SQLVal{Type: sqlparser.IntVal, Val: []byte(test.input)}\n\t\t\tresult, err := parseCloseStatus(original)\n\n\t\t\tassert.Equal(t, test.expected, result)\n\t\t\tif test.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsOptimizedQueryColumn(t *testing.T) {\n\ttests := map[string]struct {\n\t\tquery     string\n\t\tvalidated string\n\t\terr       string\n\t}{\n\t\t\"case1-1: customzied search attribute field in pinotOptimizedQueryColumns\": {\n\t\t\tquery:     \"CustomTestKeywordField = 'Test-UUID-1234-5678'\",\n\t\t\tvalidated: `CustomTestKeywordField = 'TestUUID12345678'`,\n\t\t},\n\t\t\"case1-2: customzied search attribute field in pinotOptimizedQueryColumns\": {\n\t\t\tquery:     \"CustomTestStringField = 'Test Test-UUID-1234-5678'\",\n\t\t\tvalidated: `CustomTestStringField = 'Test TestUUID12345678'`,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tsearchAttr := definition.GetDefaultIndexedKeys()\n\t\t\tsearchAttr[\"CustomTestKeywordField\"] = types.IndexedValueTypeKeyword\n\t\t\tsearchAttr[\"CustomTestStringField\"] = types.IndexedValueTypeString\n\t\t\tvalidSearchAttr := dynamicproperties.GetMapPropertyFn(searchAttr)\n\t\t\tpinotOptimizedQueryColumns := dynamicproperties.GetMapPropertyFn(map[string]interface{}{\n\t\t\t\t\"CustomTestKeywordField\": \"test\",\n\t\t\t\t\"CustomTestStringField\":  \"test\",\n\t\t\t})\n\t\t\tqv := NewPinotQueryValidator(validSearchAttr, pinotOptimizedQueryColumns)\n\t\t\tvalidated, err := qv.ValidateQuery(test.query)\n\t\t\tif err != nil {\n\t\t\t\tassert.Equal(t, test.err, err.Error())\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.validated, validated)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/pinot/pinot_client.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/startreedata/pinot-client-go/pinot\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype PinotClient struct {\n\tclient      *pinot.Connection\n\tlogger      log.Logger\n\ttableName   string\n\tserviceName string\n}\n\nfunc NewPinotClient(client *pinot.Connection, logger log.Logger, pinotConfig *config.PinotVisibilityConfig) GenericClient {\n\treturn &PinotClient{\n\t\tclient:      client,\n\t\tlogger:      logger,\n\t\ttableName:   pinotConfig.Table,\n\t\tserviceName: pinotConfig.ServiceName,\n\t}\n}\n\nfunc (c *PinotClient) Search(request *SearchRequest) (*SearchResponse, error) {\n\tresp, err := c.client.ExecuteSQL(c.tableName, request.Query)\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Pinot Search failed, %v\", err),\n\t\t}\n\t}\n\n\ttoken, err := GetNextPageToken(request.ListRequest.NextPageToken)\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Get NextPage token failed, %v\", err),\n\t\t}\n\t}\n\n\treturn c.getInternalListWorkflowExecutionsResponse(resp, request.Filter, token, request.ListRequest.PageSize, request.MaxResultWindow)\n}\n\nfunc (c *PinotClient) SearchAggr(request *SearchRequest) (AggrResponse, error) {\n\tresp, err := c.client.ExecuteSQL(c.tableName, request.Query)\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Pinot SearchAggr failed, %v\", err),\n\t\t}\n\t}\n\n\treturn resp.ResultTable.Rows, nil\n}\n\nfunc (c *PinotClient) CountByQuery(query string) (int64, error) {\n\tresp, err := c.client.ExecuteSQL(c.tableName, query)\n\tif err != nil {\n\t\treturn 0, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"CountWorkflowExecutions ExecuteSQL failed, %v\", err),\n\t\t}\n\t}\n\n\tcount, err := resp.ResultTable.Rows[0][0].(json.Number).Int64()\n\tif err == nil {\n\t\treturn count, nil\n\t}\n\n\treturn -1, &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\"can't convert result to integer!, query = %s, query result = %v, err = %v\", query, resp.ResultTable.Rows[0][0], err),\n\t}\n}\n\nfunc (c *PinotClient) GetTableName() string {\n\treturn c.tableName\n}\n\n// Pinot Response Translator\n// We flattened the search attributes into columns in Pinot table\n// This function converts the search result back to VisibilityRecord\nfunc (c *PinotClient) getInternalListWorkflowExecutionsResponse(\n\tresp *pinot.BrokerResponse,\n\tisRecordValid func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool,\n\ttoken *PinotVisibilityPageToken,\n\tpageSize int,\n\tmaxResultWindow int,\n) (*p.InternalListWorkflowExecutionsResponse, error) {\n\tresponse := &p.InternalListWorkflowExecutionsResponse{}\n\tif resp == nil || resp.ResultTable == nil || resp.ResultTable.GetRowCount() == 0 {\n\t\treturn response, nil\n\t}\n\tschema := resp.ResultTable.DataSchema // get the schema to map results\n\tcolumnNames := schema.ColumnNames\n\tactualHits := resp.ResultTable.Rows\n\tnumOfActualHits := resp.ResultTable.GetRowCount()\n\tresponse.Executions = make([]*p.InternalVisibilityWorkflowExecutionInfo, 0)\n\tfor i := 0; i < numOfActualHits; i++ {\n\t\tworkflowExecutionInfo, err := ConvertSearchResultToVisibilityRecord(actualHits[i], columnNames)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif isRecordValid == nil || isRecordValid(workflowExecutionInfo) {\n\t\t\tresponse.Executions = append(response.Executions, workflowExecutionInfo)\n\t\t}\n\t}\n\n\tif numOfActualHits == pageSize { // this means the response is not the last page\n\t\tvar nextPageToken []byte\n\t\tvar err error\n\n\t\t// ES Search API support pagination using From and PageSize, but has limit that From+PageSize cannot exceed a threshold\n\t\t// In pinot we just skip (previous pages * page limit) items and take the next (number of page limit) items\n\t\tnextPageToken, err = SerializePageToken(&PinotVisibilityPageToken{From: token.From + numOfActualHits})\n\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tresponse.NextPageToken = make([]byte, len(nextPageToken))\n\t\tcopy(response.NextPageToken, nextPageToken)\n\t}\n\treturn response, nil\n}\n\nfunc (c *PinotClient) getInternalGetClosedWorkflowExecutionResponse(resp *pinot.BrokerResponse) (\n\t*p.InternalGetClosedWorkflowExecutionResponse,\n\terror,\n) {\n\tif resp == nil {\n\t\treturn nil, nil\n\t}\n\n\tresponse := &p.InternalGetClosedWorkflowExecutionResponse{}\n\tschema := resp.ResultTable.DataSchema // get the schema to map results\n\tcolumnNames := schema.ColumnNames\n\tactualHits := resp.ResultTable.Rows\n\tvar err error\n\tresponse.Execution, err = ConvertSearchResultToVisibilityRecord(actualHits[0], columnNames)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n"
  },
  {
    "path": "common/pinot/pinot_client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"fmt\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/startreedata/pinot-client-go/pinot\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\ttestIndex        = \"test-index\"\n\ttestDomain       = \"test-domain\"\n\ttestDomainID     = \"bfd5c907-f899-4baf-a7b2-2ab85e623ebd\"\n\ttestPageSize     = 10\n\ttestEarliestTime = int64(1547596872371000000)\n\ttestLatestTime   = int64(2547596872371000000)\n\ttestWorkflowType = \"test-wf-type\"\n\ttestWorkflowID   = \"test-wid\"\n\ttestCloseStatus  = int32(1)\n\n\tclient = PinotClient{\n\t\tclient: nil,\n\t\tlogger: nil,\n\t}\n)\n\nfunc TestSearch(t *testing.T) {\n\tquery := \"select teamID, count(*) as cnt, sum(homeRuns) as sum_homeRuns from baseballStats group by teamID limit 10\"\n\n\trequest := &SearchRequest{\n\t\tQuery: query,\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tNextPageToken: nil,\n\t\t},\n\t}\n\n\terrorRequest := &SearchRequest{\n\t\tQuery: \"error\",\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tNextPageToken: []byte(\"ha-ha\"),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tinputRequest  *SearchRequest\n\t\texpectedError error\n\t\tserver        *httptest.Server\n\t}{\n\t\t\"Case1-1: error internal server case\": {\n\t\t\tinputRequest: errorRequest,\n\t\t\texpectedError: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Pinot Search failed, caught http exception when querying Pinot: 400 Bad Request\"),\n\t\t\t},\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t})),\n\t\t},\n\t\t\"Case1-2: error json conversion case\": {\n\t\t\tinputRequest: errorRequest,\n\t\t\texpectedError: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Get NextPage token failed, unable to deserialize page token. err: invalid character 'h' looking for beginning of value\"),\n\t\t\t},\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tassert.Equal(t, \"POST\", r.Method)\n\t\t\t\tassert.True(t, strings.HasSuffix(r.RequestURI, \"/query/sql\"))\n\t\t\t\tfmt.Fprintln(w, \"{\\\"resultTable\\\":{\\\"dataSchema\\\":{\\\"columnDataTypes\\\":[\\\"LONG\\\"],\\\"columnNames\\\":[\\\"cnt\\\"]},\\\"rows\\\":[[97.889]]},\\\"exceptions\\\":[],\\\"numServersQueried\\\":1,\\\"numServersResponded\\\":1,\\\"numSegmentsQueried\\\":1,\\\"numSegmentsProcessed\\\":1,\\\"numSegmentsMatched\\\":1,\\\"numConsumingSegmentsQueried\\\":0,\\\"numDocsScanned\\\":97889,\\\"numEntriesScannedInFilter\\\":0,\\\"numEntriesScannedPostFilter\\\":0,\\\"numGroupsLimitReached\\\":false,\\\"totalDocs\\\":97889,\\\"timeUsedMs\\\":5,\\\"segmentStatistics\\\":[],\\\"traceInfo\\\":{},\\\"minConsumingFreshnessTimeMs\\\":0}\")\n\t\t\t})),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\tinputRequest:  request,\n\t\t\texpectedError: nil,\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tassert.Equal(t, \"POST\", r.Method)\n\t\t\t\tassert.True(t, strings.HasSuffix(r.RequestURI, \"/query/sql\"))\n\t\t\t\tfmt.Fprintln(w, \"{\\\"resultTable\\\":{\\\"dataSchema\\\":{\\\"columnDataTypes\\\":[\\\"LONG\\\"],\\\"columnNames\\\":[\\\"cnt\\\"]},\\\"rows\\\":[[97889]]},\\\"exceptions\\\":[],\\\"numServersQueried\\\":1,\\\"numServersResponded\\\":1,\\\"numSegmentsQueried\\\":1,\\\"numSegmentsProcessed\\\":1,\\\"numSegmentsMatched\\\":1,\\\"numConsumingSegmentsQueried\\\":0,\\\"numDocsScanned\\\":97889,\\\"numEntriesScannedInFilter\\\":0,\\\"numEntriesScannedPostFilter\\\":0,\\\"numGroupsLimitReached\\\":false,\\\"totalDocs\\\":97889,\\\"timeUsedMs\\\":5,\\\"segmentStatistics\\\":[],\\\"traceInfo\\\":{},\\\"minConsumingFreshnessTimeMs\\\":0}\")\n\t\t\t})),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tts := test.server\n\t\t\tdefer ts.Close()\n\t\t\tpinotConnection, err := pinot.NewFromBrokerList([]string{ts.URL})\n\t\t\tassert.NotNil(t, pinotConnection)\n\t\t\tassert.Nil(t, err)\n\n\t\t\tpinotClient := NewPinotClient(pinotConnection, testlogger.New(t), &config.PinotVisibilityConfig{\n\t\t\t\tTable:       \"\",\n\t\t\t\tServiceName: \"\",\n\t\t\t})\n\n\t\t\tactualOutput, err := pinotClient.Search(test.inputRequest)\n\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Nil(t, actualOutput)\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, actualOutput)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCountByQuery(t *testing.T) {\n\terrorQuery := \"error\"\n\tquery := \"select teamID, count(*) as cnt, sum(homeRuns) as sum_homeRuns from baseballStats group by teamID limit 10\"\n\n\ttests := map[string]struct {\n\t\tinputQuery     string\n\t\texpectedOutput int64\n\t\texpectedError  error\n\t\tserver         *httptest.Server\n\t}{\n\t\t\"Case1-1: error internal server case\": {\n\t\t\tinputQuery:     errorQuery,\n\t\t\texpectedOutput: 0,\n\t\t\texpectedError: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"CountWorkflowExecutions ExecuteSQL failed, caught http exception when querying Pinot: 400 Bad Request\"),\n\t\t\t},\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t})),\n\t\t},\n\t\t\"Case1-2: error json conversion case\": {\n\t\t\tinputQuery:     errorQuery,\n\t\t\texpectedOutput: -1,\n\t\t\texpectedError: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"can't convert result to integer!, query = error, query result = 97.889, err = strconv.ParseInt: parsing \\\"97.889\\\": invalid syntax\"),\n\t\t\t},\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tassert.Equal(t, \"POST\", r.Method)\n\t\t\t\tassert.True(t, strings.HasSuffix(r.RequestURI, \"/query/sql\"))\n\t\t\t\tfmt.Fprintln(w, \"{\\\"resultTable\\\":{\\\"dataSchema\\\":{\\\"columnDataTypes\\\":[\\\"LONG\\\"],\\\"columnNames\\\":[\\\"cnt\\\"]},\\\"rows\\\":[[97.889]]},\\\"exceptions\\\":[],\\\"numServersQueried\\\":1,\\\"numServersResponded\\\":1,\\\"numSegmentsQueried\\\":1,\\\"numSegmentsProcessed\\\":1,\\\"numSegmentsMatched\\\":1,\\\"numConsumingSegmentsQueried\\\":0,\\\"numDocsScanned\\\":97889,\\\"numEntriesScannedInFilter\\\":0,\\\"numEntriesScannedPostFilter\\\":0,\\\"numGroupsLimitReached\\\":false,\\\"totalDocs\\\":97889,\\\"timeUsedMs\\\":5,\\\"segmentStatistics\\\":[],\\\"traceInfo\\\":{},\\\"minConsumingFreshnessTimeMs\\\":0}\")\n\t\t\t})),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\tinputQuery:     query,\n\t\t\texpectedOutput: 97889,\n\t\t\texpectedError:  nil,\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tassert.Equal(t, \"POST\", r.Method)\n\t\t\t\tassert.True(t, strings.HasSuffix(r.RequestURI, \"/query/sql\"))\n\t\t\t\tfmt.Fprintln(w, \"{\\\"resultTable\\\":{\\\"dataSchema\\\":{\\\"columnDataTypes\\\":[\\\"LONG\\\"],\\\"columnNames\\\":[\\\"cnt\\\"]},\\\"rows\\\":[[97889]]},\\\"exceptions\\\":[],\\\"numServersQueried\\\":1,\\\"numServersResponded\\\":1,\\\"numSegmentsQueried\\\":1,\\\"numSegmentsProcessed\\\":1,\\\"numSegmentsMatched\\\":1,\\\"numConsumingSegmentsQueried\\\":0,\\\"numDocsScanned\\\":97889,\\\"numEntriesScannedInFilter\\\":0,\\\"numEntriesScannedPostFilter\\\":0,\\\"numGroupsLimitReached\\\":false,\\\"totalDocs\\\":97889,\\\"timeUsedMs\\\":5,\\\"segmentStatistics\\\":[],\\\"traceInfo\\\":{},\\\"minConsumingFreshnessTimeMs\\\":0}\")\n\t\t\t})),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tts := test.server\n\t\t\tdefer ts.Close()\n\t\t\tpinotConnection, err := pinot.NewFromBrokerList([]string{ts.URL})\n\t\t\tassert.NotNil(t, pinotConnection)\n\t\t\tassert.Nil(t, err)\n\n\t\t\tpinotClient := NewPinotClient(pinotConnection, testlogger.New(t), &config.PinotVisibilityConfig{\n\t\t\t\tTable:       \"\",\n\t\t\t\tServiceName: \"\",\n\t\t\t})\n\n\t\t\tactualOutput, err := pinotClient.CountByQuery(test.inputQuery)\n\t\t\tassert.Equal(t, test.expectedOutput, actualOutput)\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSearchAggr(t *testing.T) {\n\tquery := \"select teamID, count(*) as cnt, sum(homeRuns) as sum_homeRuns from baseballStats group by teamID limit 10\"\n\n\trequest := &SearchRequest{\n\t\tQuery: query,\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tNextPageToken: nil,\n\t\t},\n\t}\n\n\terrorRequest := &SearchRequest{\n\t\tQuery: \"error\",\n\t\tListRequest: &p.InternalListWorkflowExecutionsRequest{\n\t\t\tNextPageToken: []byte(\"ha-ha\"),\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tinputRequest  *SearchRequest\n\t\texpectedError error\n\t\tserver        *httptest.Server\n\t}{\n\t\t\"Case1-1: error internal server case\": {\n\t\t\tinputRequest: errorRequest,\n\t\t\texpectedError: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Pinot SearchAggr failed, caught http exception when querying Pinot: 400 Bad Request\"),\n\t\t\t},\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.WriteHeader(http.StatusBadRequest)\n\t\t\t})),\n\t\t},\n\t\t\"Case2: normal case\": {\n\t\t\tinputRequest:  request,\n\t\t\texpectedError: nil,\n\t\t\tserver: httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\tw.Header().Set(\"Content-Type\", \"application/json\")\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\tassert.Equal(t, \"POST\", r.Method)\n\t\t\t\tassert.True(t, strings.HasSuffix(r.RequestURI, \"/query/sql\"))\n\t\t\t\tfmt.Fprintln(w, \"{\\\"resultTable\\\":{\\\"dataSchema\\\":{\\\"columnDataTypes\\\":[\\\"LONG\\\"],\\\"columnNames\\\":[\\\"cnt\\\"]},\\\"rows\\\":[[\\\"test-domain\\\", 10]]},\\\"exceptions\\\":[],\\\"numServersQueried\\\":1,\\\"numServersResponded\\\":1,\\\"numSegmentsQueried\\\":1,\\\"numSegmentsProcessed\\\":1,\\\"numSegmentsMatched\\\":1,\\\"numConsumingSegmentsQueried\\\":0,\\\"numDocsScanned\\\":97889,\\\"numEntriesScannedInFilter\\\":0,\\\"numEntriesScannedPostFilter\\\":0,\\\"numGroupsLimitReached\\\":false,\\\"totalDocs\\\":97889,\\\"timeUsedMs\\\":5,\\\"segmentStatistics\\\":[],\\\"traceInfo\\\":{},\\\"minConsumingFreshnessTimeMs\\\":0}\")\n\t\t\t})),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tts := test.server\n\t\t\tdefer ts.Close()\n\t\t\tpinotConnection, err := pinot.NewFromBrokerList([]string{ts.URL})\n\t\t\tassert.NotNil(t, pinotConnection)\n\t\t\tassert.Nil(t, err)\n\n\t\t\tpinotClient := NewPinotClient(pinotConnection, testlogger.New(t), &config.PinotVisibilityConfig{\n\t\t\t\tTable:       \"\",\n\t\t\t\tServiceName: \"\",\n\t\t\t})\n\n\t\t\tactualOutput, err := pinotClient.SearchAggr(test.inputRequest)\n\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Nil(t, actualOutput)\n\t\t\t\tassert.Equal(t, test.expectedError.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, actualOutput)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetTableName(t *testing.T) {\n\tassert.Equal(t, \"\", client.GetTableName())\n}\n\nfunc TestBuildMap(t *testing.T) {\n\tcolumnName := []string{\"WorkflowID\", \"RunID\", \"WorkflowType\", \"DomainID\", \"StartTime\", \"ExecutionTime\", \"CloseTime\", \"CloseStatus\", \"HistoryLength\", \"TaskList\", \"IsCron\", \"NumClusters\", \"UpdateTime\", \"CustomIntField\", \"CustomStringField\"}\n\thit := []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"tsklst\", true, 1, testEarliestTime, 1, \"some string\"}\n\n\ttests := map[string]struct {\n\t\tinputColumnNames []string\n\t\tinputHit         []interface{}\n\t\texpectedMap      map[string]interface{}\n\t}{\n\t\t\"Case1: with everything\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit:         hit,\n\t\t\texpectedMap:      map[string]interface{}{\"CloseStatus\": 1, \"CloseTime\": int64(2547596872371000000), \"CustomIntField\": 1, \"CustomStringField\": \"some string\", \"DomainID\": \"domainid\", \"ExecutionTime\": int64(1547596872371000000), \"HistoryLength\": 1, \"IsCron\": true, \"NumClusters\": 1, \"RunID\": \"rid\", \"StartTime\": int64(1547596872371000000), \"TaskList\": \"tsklst\", \"UpdateTime\": int64(1547596872371000000), \"WorkflowID\": \"wfid\", \"WorkflowType\": \"wftype\"},\n\t\t},\n\t\t\"Case2: nil result\": {\n\t\t\tinputColumnNames: nil,\n\t\t\tinputHit:         nil,\n\t\t\texpectedMap:      map[string]interface{}{},\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresMap := buildMap(test.inputHit, test.inputColumnNames)\n\t\t\tassert.Equal(t, test.expectedMap, resMap)\n\t\t})\n\t}\n}\n\nfunc TestGetInternalListWorkflowExecutionsResponse(t *testing.T) {\n\tcolumnName := []string{\"WorkflowID\", \"RunID\", \"WorkflowType\", \"DomainID\", \"StartTime\", \"ExecutionTime\", \"CloseTime\", \"CloseStatus\", \"HistoryLength\", \"Encoding\", \"TaskList\", \"IsCron\", \"NumClusters\", \"UpdateTime\", \"Attr\"}\n\thit1 := []interface{}{\"wfid1\", \"rid1\", \"wftype1\", \"domainid1\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode1\", \"tsklst1\", true, 1, testEarliestTime, \"null\"}\n\thit2 := []interface{}{\"wfid2\", \"rid2\", \"wftype2\", \"domainid2\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode2\", \"tsklst2\", false, 1, testEarliestTime, \"null\"}\n\thit3 := []interface{}{\"wfid3\", \"rid3\", \"wftype3\", \"domainid3\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode3\", \"tsklst3\", false, 1, testEarliestTime, \"null\"}\n\thit4 := []interface{}{\"wfid4\", \"rid4\", \"wftype4\", \"domainid4\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode4\", \"tsklst4\", false, 1, testEarliestTime, \"null\"}\n\thit5 := []interface{}{\"wfid5\", \"rid5\", \"wftype5\", \"domainid5\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode5\", \"tsklst5\", false, 1, testEarliestTime, \"null\"}\n\n\tbrokerResponse := &pinot.BrokerResponse{\n\t\tAggregationResults: nil,\n\t\tSelectionResults:   nil,\n\t\tResultTable: &pinot.ResultTable{\n\t\t\tDataSchema: pinot.RespSchema{\n\t\t\t\tColumnDataTypes: nil,\n\t\t\t\tColumnNames:     columnName,\n\t\t\t},\n\t\t\tRows: [][]interface{}{\n\t\t\t\thit1,\n\t\t\t\thit2,\n\t\t\t\thit3,\n\t\t\t\thit4,\n\t\t\t\thit5,\n\t\t\t},\n\t\t},\n\t\tExceptions:                  nil,\n\t\tTraceInfo:                   nil,\n\t\tNumServersQueried:           1,\n\t\tNumServersResponded:         1,\n\t\tNumSegmentsQueried:          1,\n\t\tNumSegmentsProcessed:        1,\n\t\tNumSegmentsMatched:          1,\n\t\tNumConsumingSegmentsQueried: 1,\n\t\tNumDocsScanned:              10,\n\t\tNumEntriesScannedInFilter:   1,\n\t\tNumEntriesScannedPostFilter: 1,\n\t\tNumGroupsLimitReached:       false,\n\t\tTotalDocs:                   1,\n\t\tTimeUsedMs:                  1,\n\t\tMinConsumingFreshnessTimeMs: 1,\n\t}\n\n\ttoken := &PinotVisibilityPageToken{\n\t\tFrom: 0,\n\t}\n\n\t// Cannot use a table test, because they are not checking the same fields\n\tresult, err := client.getInternalListWorkflowExecutionsResponse(brokerResponse, nil, token, 5, 33)\n\n\tassert.Equal(t, \"wfid1\", result.Executions[0].WorkflowID)\n\tassert.Equal(t, \"rid1\", result.Executions[0].RunID)\n\tassert.Equal(t, \"wftype1\", result.Executions[0].WorkflowType)\n\tassert.Equal(t, \"domainid1\", result.Executions[0].DomainID)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[0].StartTime)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[0].ExecutionTime)\n\tassert.Equal(t, time.UnixMilli(testLatestTime), result.Executions[0].CloseTime)\n\tassert.Equal(t, types.WorkflowExecutionCloseStatus(1), *result.Executions[0].Status)\n\tassert.Equal(t, int64(1), result.Executions[0].HistoryLength)\n\tassert.Equal(t, \"tsklst1\", result.Executions[0].TaskList)\n\tassert.Equal(t, true, result.Executions[0].IsCron)\n\tassert.Equal(t, int16(1), result.Executions[0].NumClusters)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[0].UpdateTime)\n\n\tassert.Equal(t, \"wfid2\", result.Executions[1].WorkflowID)\n\tassert.Equal(t, \"rid2\", result.Executions[1].RunID)\n\tassert.Equal(t, \"wftype2\", result.Executions[1].WorkflowType)\n\tassert.Equal(t, \"domainid2\", result.Executions[1].DomainID)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[1].StartTime)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[1].ExecutionTime)\n\tassert.Equal(t, time.UnixMilli(testLatestTime), result.Executions[1].CloseTime)\n\tassert.Equal(t, types.WorkflowExecutionCloseStatus(1), *result.Executions[1].Status)\n\tassert.Equal(t, int64(1), result.Executions[1].HistoryLength)\n\tassert.Equal(t, \"tsklst2\", result.Executions[1].TaskList)\n\tassert.Equal(t, false, result.Executions[1].IsCron)\n\tassert.Equal(t, int16(1), result.Executions[1].NumClusters)\n\tassert.Equal(t, time.UnixMilli(testEarliestTime), result.Executions[1].UpdateTime)\n\n\tassert.Nil(t, err)\n\n\tresponseToken := result.NextPageToken\n\tunmarshalResponseToken, err := GetNextPageToken(responseToken)\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"Unmarshal error in PinotClient test %s\", err))\n\t}\n\tassert.Equal(t, 5, unmarshalResponseToken.From)\n\n\t// check if record is not valid\n\tisRecordValid := func(rec *p.InternalVisibilityWorkflowExecutionInfo) bool {\n\t\treturn false\n\t}\n\temptyResult, err := client.getInternalListWorkflowExecutionsResponse(brokerResponse, isRecordValid, nil, 10, 33)\n\tassert.Equal(t, 0, len(emptyResult.Executions))\n\tassert.Nil(t, err)\n\n\t// check nil input\n\tnilResult, err := client.getInternalListWorkflowExecutionsResponse(nil, isRecordValid, nil, 10, 33)\n\tassert.Equal(t, &p.InternalListWorkflowExecutionsResponse{}, nilResult)\n\tassert.Nil(t, err)\n}\n\nfunc TestGetInternalGetClosedWorkflowExecutionResponse(t *testing.T) {\n\tcolumnName := []string{\"WorkflowID\", \"RunID\", \"WorkflowType\", \"DomainID\", \"StartTime\", \"ExecutionTime\", \"CloseTime\", \"CloseStatus\", \"HistoryLength\", \"Encoding\", \"TaskList\", \"IsCron\", \"NumClusters\", \"UpdateTime\", \"Attr\"}\n\thit1 := []interface{}{\"wfid1\", \"rid1\", \"wftype1\", \"domainid1\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"encode1\", \"tsklst1\", true, 1, testEarliestTime, \"null\"}\n\n\tbrokerResponse := &pinot.BrokerResponse{\n\t\tAggregationResults: nil,\n\t\tSelectionResults:   nil,\n\t\tResultTable: &pinot.ResultTable{\n\t\t\tDataSchema: pinot.RespSchema{\n\t\t\t\tColumnDataTypes: nil,\n\t\t\t\tColumnNames:     columnName,\n\t\t\t},\n\t\t\tRows: [][]interface{}{\n\t\t\t\thit1,\n\t\t\t},\n\t\t},\n\t\tExceptions:                  nil,\n\t\tTraceInfo:                   nil,\n\t\tNumServersQueried:           1,\n\t\tNumServersResponded:         1,\n\t\tNumSegmentsQueried:          1,\n\t\tNumSegmentsProcessed:        1,\n\t\tNumSegmentsMatched:          1,\n\t\tNumConsumingSegmentsQueried: 1,\n\t\tNumDocsScanned:              1,\n\t\tNumEntriesScannedInFilter:   1,\n\t\tNumEntriesScannedPostFilter: 1,\n\t\tNumGroupsLimitReached:       false,\n\t\tTotalDocs:                   1,\n\t\tTimeUsedMs:                  1,\n\t\tMinConsumingFreshnessTimeMs: 1,\n\t}\n\n\ttests := map[string]struct {\n\t\tinput         *pinot.BrokerResponse\n\t\tisInputEmpty  bool\n\t\texpectedError error\n\t}{\n\t\t\"Case1: empty case\": {\n\t\t\tinput:         nil,\n\t\t\tisInputEmpty:  true,\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case2: with everything\": {\n\t\t\tinput:         brokerResponse,\n\t\t\tisInputEmpty:  false,\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactualOutput, actualError := client.getInternalGetClosedWorkflowExecutionResponse(test.input)\n\t\t\tif test.isInputEmpty {\n\t\t\t\tassert.Nil(t, actualOutput)\n\t\t\t\tassert.Nil(t, actualError)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, \"wfid1\", actualOutput.Execution.WorkflowID)\n\t\t\t\tassert.Equal(t, \"rid1\", actualOutput.Execution.RunID)\n\t\t\t\tassert.Equal(t, \"wftype1\", actualOutput.Execution.WorkflowType)\n\t\t\t\tassert.Equal(t, \"domainid1\", actualOutput.Execution.DomainID)\n\t\t\t\tassert.Equal(t, time.UnixMilli(testEarliestTime), actualOutput.Execution.StartTime)\n\t\t\t\tassert.Equal(t, time.UnixMilli(testEarliestTime), actualOutput.Execution.ExecutionTime)\n\t\t\t\tassert.Equal(t, time.UnixMilli(testLatestTime), actualOutput.Execution.CloseTime)\n\t\t\t\tassert.Equal(t, types.WorkflowExecutionCloseStatus(1), *actualOutput.Execution.Status)\n\t\t\t\tassert.Equal(t, int64(1), actualOutput.Execution.HistoryLength)\n\t\t\t\tassert.Equal(t, \"tsklst1\", actualOutput.Execution.TaskList)\n\t\t\t\tassert.Equal(t, true, actualOutput.Execution.IsCron)\n\t\t\t\tassert.Equal(t, int16(1), actualOutput.Execution.NumClusters)\n\t\t\t\tassert.Equal(t, time.UnixMilli(testEarliestTime), actualOutput.Execution.UpdateTime)\n\t\t\t\tassert.Nil(t, actualError)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/pinot/response_utility.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tMemo = \"Memo\"\n\tAttr = \"Attr\"\n)\n\nfunc buildMap(hit []interface{}, columnNames []string) map[string]interface{} {\n\tsystemKeyMap := make(map[string]interface{})\n\n\tfor i := 0; i < len(columnNames); i++ {\n\t\tkey := columnNames[i]\n\t\tsystemKeyMap[key] = hit[i]\n\t}\n\n\treturn systemKeyMap\n}\n\n// VisibilityRecord is a struct of doc for deserialization\n// this is different from InternalVisibilityWorkflowExecutionInfo\n// use this to deserialize the systemKeyMap from Pinot response\ntype VisibilityRecord struct {\n\tWorkflowID            string\n\tRunID                 string\n\tWorkflowType          string\n\tDomainID              string\n\tStartTime             int64\n\tExecutionTime         int64\n\tCloseTime             int64\n\tCloseStatus           int\n\tHistoryLength         int64\n\tTaskList              string\n\tIsCron                bool\n\tNumClusters           int16\n\tClusterAttributeScope string\n\tClusterAttributeName  string\n\tUpdateTime            int64\n\tShardID               int16\n}\n\nfunc ConvertSearchResultToVisibilityRecord(hit []interface{}, columnNames []string) (*p.InternalVisibilityWorkflowExecutionInfo, error) {\n\tif len(hit) != len(columnNames) {\n\t\treturn nil, fmt.Errorf(\"length of hit (%v) is not equal with length of columnNames(%v)\", len(hit), len(columnNames))\n\t}\n\n\tsystemKeyMap := buildMap(hit, columnNames)\n\n\tjsonSystemKeyMap, err := json.Marshal(systemKeyMap)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"unable to marshal systemKeyMap\")\n\t}\n\n\tattributeMap := make(map[string]interface{})\n\tvar memo *p.DataBlob\n\tif systemKeyMap[Attr] != nil {\n\t\tattrMapStr, ok := systemKeyMap[Attr].(string)\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(`assertion error. Can't convert systemKeyMap[Attr] to string. Found %T`, systemKeyMap[Attr])\n\t\t}\n\n\t\terr = json.Unmarshal([]byte(attrMapStr), &attributeMap)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"unable to Unmarshal searchAttribute map: %s\", err.Error())\n\t\t}\n\n\t\tif attributeMap[Memo] != nil {\n\t\t\tmemo, err = convertMemo(attributeMap[Memo])\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"unable to convert memo: %s\", err.Error())\n\t\t\t}\n\t\t}\n\t\tdelete(attributeMap, Memo) // cleanup after we get memo from search attribute\n\t}\n\n\t// if memo is empty, set it to nil\n\tif memo != nil && len(memo.GetData()) == 0 {\n\t\tmemo = nil\n\t}\n\n\tvar source *VisibilityRecord\n\terr = json.Unmarshal(jsonSystemKeyMap, &source)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"unable to Unmarshal systemKeyMap: %s\", err.Error())\n\t}\n\n\trecord := &p.InternalVisibilityWorkflowExecutionInfo{\n\t\tDomainID:              source.DomainID,\n\t\tWorkflowType:          source.WorkflowType,\n\t\tWorkflowID:            source.WorkflowID,\n\t\tRunID:                 source.RunID,\n\t\tTypeName:              source.WorkflowType,\n\t\tStartTime:             time.UnixMilli(source.StartTime), // be careful: source.StartTime is in milliseconds\n\t\tExecutionTime:         time.UnixMilli(source.ExecutionTime),\n\t\tTaskList:              source.TaskList,\n\t\tIsCron:                source.IsCron,\n\t\tNumClusters:           source.NumClusters,\n\t\tClusterAttributeScope: source.ClusterAttributeScope,\n\t\tClusterAttributeName:  source.ClusterAttributeName,\n\t\tShardID:               source.ShardID,\n\t\tSearchAttributes:      attributeMap,\n\t\tMemo:                  memo,\n\t}\n\tif source.UpdateTime > 0 {\n\t\trecord.UpdateTime = time.UnixMilli(source.UpdateTime)\n\t}\n\tif source.CloseTime > 0 {\n\t\trecord.CloseTime = time.UnixMilli(source.CloseTime)\n\t\trecord.Status = toWorkflowExecutionCloseStatus(source.CloseStatus)\n\t\trecord.HistoryLength = source.HistoryLength\n\t}\n\n\treturn record, nil\n}\n\nfunc convertMemo(memoRaw interface{}) (*p.DataBlob, error) {\n\tdb := p.DataBlob{}\n\n\tmemoRawStr, ok := memoRaw.(string)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"memoRaw is not a String: %T\", memoRaw)\n\t}\n\n\terr := json.Unmarshal([]byte(memoRawStr), &db)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"unable to unmarshal memoRawStr: %s\", err.Error())\n\t}\n\treturn &db, nil\n}\n\nfunc toWorkflowExecutionCloseStatus(status int) *types.WorkflowExecutionCloseStatus {\n\tif status < 0 {\n\t\treturn nil\n\t}\n\tcloseStatus := types.WorkflowExecutionCloseStatus(status)\n\treturn &closeStatus\n}\n"
  },
  {
    "path": "common/pinot/response_utility_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage pinot\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestConvertSearchResultToVisibilityRecord(t *testing.T) {\n\tcolumnName := []string{\"WorkflowID\", \"RunID\", \"WorkflowType\", \"DomainID\", \"StartTime\", \"ExecutionTime\", \"CloseTime\", \"CloseStatus\", \"HistoryLength\", \"TaskList\", \"IsCron\", \"NumClusters\", \"UpdateTime\", \"Attr\", \"ClusterAttributeScope\", \"ClusterAttributeName\"}\n\tcloseStatus := types.WorkflowExecutionCloseStatusFailed\n\n\tsampleRawMemo := &types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t`\"Service\"`: []byte(`\"serverName1\"`),\n\t\t},\n\t}\n\tserializer := p.NewPayloadSerializer()\n\tsampleEncodedMemo, err := serializer.SerializeVisibilityMemo(sampleRawMemo, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\n\terrorMapRaw1 := map[string]interface{}{\"Memo\": 123}\n\terrorMap1, err := json.Marshal(errorMapRaw1)\n\n\terrorMapRaw2 := map[string]interface{}{\"Memo\": \"123\"}\n\terrorMap2, err := json.Marshal(errorMapRaw2)\n\n\ttests := map[string]struct {\n\t\tinputColumnNames         []string\n\t\tinputHit                 []interface{}\n\t\texpectedVisibilityRecord *p.InternalVisibilityWorkflowExecutionInfo\n\t\tmemoCheck                bool\n\t\texpectErr                error\n\t}{\n\t\t\"Case1: nil result\": {\n\t\t\tinputColumnNames:         nil,\n\t\t\tinputHit:                 []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, -1, 1, \"tsklst\", true, 1, testEarliestTime, \"{}\", \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(\"length of hit (16) is not equal with length of columnNames(0)\"),\n\t\t},\n\t\t\"Case2-1: marshal system key error case\": {\n\t\t\tinputColumnNames:         columnName,\n\t\t\tinputHit:                 []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"tsklst\", true, 1, testEarliestTime, make(chan int), \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(\"unable to marshal systemKeyMap\"),\n\t\t},\n\t\t\"Case2-2: unmarshal system key error case\": {\n\t\t\tinputColumnNames:         columnName,\n\t\t\tinputHit:                 []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, 1, \"1\", \"tsklst\", true, 1, testEarliestTime, `{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14}`, \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(\"unable to Unmarshal systemKeyMap: json: cannot unmarshal string into Go struct field VisibilityRecord.HistoryLength of type int64\"),\n\t\t},\n\t\t\"Case2-3: Attr to string error case\": {\n\t\t\tinputColumnNames:         []string{\"Attr\"},\n\t\t\tinputHit:                 []interface{}{123},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(`assertion error. Can't convert systemKeyMap[Attr] to string. Found int`),\n\t\t},\n\t\t\"Case2-4: Attr unmarshal to map error case\": {\n\t\t\tinputColumnNames:         []string{\"Attr\"},\n\t\t\tinputHit:                 []interface{}{\"123\"},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(`unable to Unmarshal searchAttribute map: json: cannot unmarshal number into Go value of type map[string]interface {}`),\n\t\t},\n\t\t\"Case2-4: Memo to string error case\": {\n\t\t\tinputColumnNames:         []string{\"Attr\"},\n\t\t\tinputHit:                 []interface{}{string(errorMap1)},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(`unable to convert memo: memoRaw is not a String: float64`),\n\t\t},\n\t\t\"Case2-5: Memo unmarshal error case\": {\n\t\t\tinputColumnNames:         []string{\"Attr\"},\n\t\t\tinputHit:                 []interface{}{string(errorMap2)},\n\t\t\texpectedVisibilityRecord: nil,\n\t\t\texpectErr:                fmt.Errorf(`unable to convert memo: unable to unmarshal memoRawStr: json: cannot unmarshal number into Go value of type persistence.DataBlob`),\n\t\t},\n\t\t\"Case3-1: closed wf with everything except for an empty Attr\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit:         []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, -1, 1, \"tsklst\", true, 1, testEarliestTime, \"{}\", \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tDomainID:              \"domainid\",\n\t\t\t\tWorkflowType:          \"wftype\",\n\t\t\t\tWorkflowID:            \"wfid\",\n\t\t\t\tRunID:                 \"rid\",\n\t\t\t\tTypeName:              \"wftype\",\n\t\t\t\tStartTime:             time.UnixMilli(testEarliestTime),\n\t\t\t\tExecutionTime:         time.UnixMilli(testEarliestTime),\n\t\t\t\tCloseTime:             time.UnixMilli(testLatestTime),\n\t\t\t\tStatus:                nil,\n\t\t\t\tHistoryLength:         1,\n\t\t\t\tMemo:                  nil,\n\t\t\t\tTaskList:              \"tsklst\",\n\t\t\t\tIsCron:                true,\n\t\t\t\tNumClusters:           1,\n\t\t\t\tClusterAttributeScope: \"region\",\n\t\t\t\tClusterAttributeName:  \"us-east\",\n\t\t\t\tUpdateTime:            time.UnixMilli(testEarliestTime),\n\t\t\t\tSearchAttributes:      map[string]interface{}{},\n\t\t\t\tShardID:               0,\n\t\t\t},\n\t\t},\n\t\t\"Case3-2: closed wf with everything\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit:         []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, testLatestTime, 1, 1, \"tsklst\", true, 1, testEarliestTime, `{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14}`, \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tDomainID:              \"domainid\",\n\t\t\t\tWorkflowType:          \"wftype\",\n\t\t\t\tWorkflowID:            \"wfid\",\n\t\t\t\tRunID:                 \"rid\",\n\t\t\t\tTypeName:              \"wftype\",\n\t\t\t\tStartTime:             time.UnixMilli(testEarliestTime),\n\t\t\t\tExecutionTime:         time.UnixMilli(testEarliestTime),\n\t\t\t\tCloseTime:             time.UnixMilli(testLatestTime),\n\t\t\t\tStatus:                &closeStatus,\n\t\t\t\tHistoryLength:         1,\n\t\t\t\tMemo:                  nil,\n\t\t\t\tTaskList:              \"tsklst\",\n\t\t\t\tIsCron:                true,\n\t\t\t\tNumClusters:           1,\n\t\t\t\tClusterAttributeScope: \"region\",\n\t\t\t\tClusterAttributeName:  \"us-east\",\n\t\t\t\tUpdateTime:            time.UnixMilli(testEarliestTime),\n\t\t\t\tSearchAttributes:      map[string]interface{}{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14},\n\t\t\t\tShardID:               0,\n\t\t\t},\n\t\t},\n\t\t\"Case4: open wf with everything\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit: []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, -1, -1, -1,\n\t\t\t\t\"tsklst\", true, 1, testEarliestTime, `{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14}`, \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tDomainID:              \"domainid\",\n\t\t\t\tWorkflowType:          \"wftype\",\n\t\t\t\tWorkflowID:            \"wfid\",\n\t\t\t\tRunID:                 \"rid\",\n\t\t\t\tTypeName:              \"wftype\",\n\t\t\t\tStartTime:             time.UnixMilli(testEarliestTime),\n\t\t\t\tExecutionTime:         time.UnixMilli(testEarliestTime),\n\t\t\t\tMemo:                  nil,\n\t\t\t\tTaskList:              \"tsklst\",\n\t\t\t\tIsCron:                true,\n\t\t\t\tNumClusters:           1,\n\t\t\t\tClusterAttributeScope: \"region\",\n\t\t\t\tClusterAttributeName:  \"us-east\",\n\t\t\t\tUpdateTime:            time.UnixMilli(testEarliestTime),\n\t\t\t\tSearchAttributes:      map[string]interface{}{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14},\n\t\t\t\tShardID:               0,\n\t\t\t},\n\t\t\tmemoCheck: true,\n\t\t},\n\t\t\"Case5-1: open wf with memo\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit: []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, -1, -1, -1,\n\t\t\t\t\"tsklst\", true, 1, testEarliestTime,\n\t\t\t\t`{\"Memo\":\"{\\\"Encoding\\\":\\\"thriftrw\\\",\\\"Data\\\":\\\"WQ0ACgsLAAAAAQAAAAkiU2VydmljZSIAAAANInNlcnZlck5hbWUxIgA=\\\"}\"}`, \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tDomainID:              \"domainid\",\n\t\t\t\tWorkflowType:          \"wftype\",\n\t\t\t\tWorkflowID:            \"wfid\",\n\t\t\t\tRunID:                 \"rid\",\n\t\t\t\tTypeName:              \"wftype\",\n\t\t\t\tStartTime:             time.UnixMilli(testEarliestTime),\n\t\t\t\tExecutionTime:         time.UnixMilli(testEarliestTime),\n\t\t\t\tMemo:                  sampleEncodedMemo,\n\t\t\t\tTaskList:              \"tsklst\",\n\t\t\t\tIsCron:                true,\n\t\t\t\tNumClusters:           1,\n\t\t\t\tClusterAttributeScope: \"region\",\n\t\t\t\tClusterAttributeName:  \"us-east\",\n\t\t\t\tUpdateTime:            time.UnixMilli(testEarliestTime),\n\t\t\t\tSearchAttributes:      map[string]interface{}{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14},\n\t\t\t\tShardID:               0,\n\t\t\t},\n\t\t\tmemoCheck: true,\n\t\t},\n\t\t\"Case5-2: open wf with empty memo\": {\n\t\t\tinputColumnNames: columnName,\n\t\t\tinputHit: []interface{}{\"wfid\", \"rid\", \"wftype\", \"domainid\", testEarliestTime, testEarliestTime, -1, -1, -1,\n\t\t\t\t\"tsklst\", true, 1, testEarliestTime,\n\t\t\t\t`{\"Memo\":\"{\\\"Encoding\\\":\\\"\\\",\\\"Data\\\":\\\"\\\"}\"}`, \"region\", \"us-east\"},\n\t\t\texpectedVisibilityRecord: &p.InternalVisibilityWorkflowExecutionInfo{\n\t\t\t\tDomainID:              \"domainid\",\n\t\t\t\tWorkflowType:          \"wftype\",\n\t\t\t\tWorkflowID:            \"wfid\",\n\t\t\t\tRunID:                 \"rid\",\n\t\t\t\tTypeName:              \"wftype\",\n\t\t\t\tStartTime:             time.UnixMilli(testEarliestTime),\n\t\t\t\tExecutionTime:         time.UnixMilli(testEarliestTime),\n\t\t\t\tMemo:                  nil,\n\t\t\t\tTaskList:              \"tsklst\",\n\t\t\t\tIsCron:                true,\n\t\t\t\tNumClusters:           1,\n\t\t\t\tClusterAttributeScope: \"region\",\n\t\t\t\tClusterAttributeName:  \"us-east\",\n\t\t\t\tUpdateTime:            time.UnixMilli(testEarliestTime),\n\t\t\t\tSearchAttributes:      map[string]interface{}{\"CustomStringField\": \"customA and customB or customC\", \"CustomDoubleField\": 3.14},\n\t\t\t\tShardID:               0,\n\t\t\t},\n\t\t\tmemoCheck: true,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tvisibilityRecord, err := ConvertSearchResultToVisibilityRecord(test.inputHit, test.inputColumnNames)\n\t\t\t\tif !test.memoCheck {\n\t\t\t\t\tassert.Equal(t, test.expectedVisibilityRecord, visibilityRecord)\n\t\t\t\t\tassert.Equal(t, test.expectErr, err)\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, test.expectedVisibilityRecord.Memo.GetData(), visibilityRecord.Memo.GetData())\n\t\t\t\t\tassert.Equal(t, test.expectedVisibilityRecord.Memo.GetEncoding(), visibilityRecord.Memo.GetEncoding())\n\t\t\t\t}\n\t\t\t})\n\t\t})\n\t}\n}\n\n// This is the process of figuring out how to encode/decode memo for Pinot\nfunc TestDeserializeMemoMockingE2E(t *testing.T) {\n\tsampleRawMemo := &types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t\"Service\": []byte(\"serverName1\"),\n\t\t},\n\t}\n\tserializer := p.NewPayloadSerializer()\n\tsampleEncodedMemo, err := serializer.SerializeVisibilityMemo(sampleRawMemo, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\t// not a human-readable string\n\tassert.Equal(t, \"Y\\r\\x00\\n\\v\\v\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\aService\\x00\\x00\\x00\\vserverName1\\x00\", string(sampleEncodedMemo.GetData()))\n\n\tmarshaledMemo, err := json.Marshal(sampleEncodedMemo)\n\tassert.NoError(t, err)\n\t// after marshal, data becomes a human-readable char array\n\tassert.Equal(t, `{\"Encoding\":\"thriftrw\",\"Data\":\"WQ0ACgsLAAAAAQAAAAdTZXJ2aWNlAAAAC3NlcnZlck5hbWUxAA==\"}`, string(marshaledMemo))\n\n\t// must-do step, to give it a type, or we can't convert it to a string in the reading side.\n\tmarshaledMemoStr := string(marshaledMemo)\n\n\t// mock the reading side\n\tunmarshaledRawData := p.DataBlob{}\n\n\t// marshaledMemoStr still knows that it is a DataBlob type\n\terr = json.Unmarshal([]byte(marshaledMemoStr), &unmarshaledRawData)\n\tassert.NoError(t, err)\n\tsampleDecodedMemo, err := serializer.DeserializeVisibilityMemo(&unmarshaledRawData)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"serverName1\", string(sampleDecodedMemo.Fields[\"Service\"]))\n}\n"
  },
  {
    "path": "common/pprof.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination pprof_mock.go -package common github.com/uber/cadence/common PProfInitializer\n\ntype (\n\t// PProfInitializer initialize the pprof based on config\n\tPProfInitializer interface {\n\t\tStart() error\n\t}\n)\n"
  },
  {
    "path": "common/pprof_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: pprof.go\n//\n// Generated by this command:\n//\n//\tmockgen -package common -source pprof.go -destination pprof_mock.go -package common github.com/uber/cadence/common PProfInitializer\n//\n\n// Package common is a generated GoMock package.\npackage common\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPProfInitializer is a mock of PProfInitializer interface.\ntype MockPProfInitializer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPProfInitializerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPProfInitializerMockRecorder is the mock recorder for MockPProfInitializer.\ntype MockPProfInitializerMockRecorder struct {\n\tmock *MockPProfInitializer\n}\n\n// NewMockPProfInitializer creates a new mock instance.\nfunc NewMockPProfInitializer(ctrl *gomock.Controller) *MockPProfInitializer {\n\tmock := &MockPProfInitializer{ctrl: ctrl}\n\tmock.recorder = &MockPProfInitializerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPProfInitializer) EXPECT() *MockPProfInitializerMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockPProfInitializer) Start() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockPProfInitializerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockPProfInitializer)(nil).Start))\n}\n"
  },
  {
    "path": "common/quotas/caller_bypass.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// CallerBypass encapsulates the logic for bypassing rate limits based on caller type\ntype CallerBypass struct {\n\tbypassCallerTypes dynamicproperties.ListPropertyFn\n}\n\n// NewCallerBypass creates a new CallerBypass with the given bypass caller types configuration\nfunc NewCallerBypass(bypassCallerTypes dynamicproperties.ListPropertyFn) CallerBypass {\n\treturn CallerBypass{\n\t\tbypassCallerTypes: bypassCallerTypes,\n\t}\n}\n\n// AllowLimiter checks if a request should be allowed through a Limiter.\n// It first checks the limiter's Allow() method, and if that returns false,\n// it checks if the caller type should bypass rate limiting.\nfunc (c CallerBypass) AllowLimiter(ctx context.Context, limiter Limiter) bool {\n\tif limiter.Allow() {\n\t\treturn true\n\t}\n\treturn c.ShouldBypass(ctx)\n}\n\n// AllowPolicy checks if a request should be allowed through a Policy.\n// It first checks the policy's Allow() method, and if that returns false,\n// it checks if the caller type should bypass rate limiting.\nfunc (c CallerBypass) AllowPolicy(ctx context.Context, policy Policy, info Info) bool {\n\tif policy.Allow(info) {\n\t\treturn true\n\t}\n\treturn c.ShouldBypass(ctx)\n}\n\n// ShouldBypass checks if the caller type from the context should bypass rate limiting\n// based on the configured bypass caller types.\nfunc (c CallerBypass) ShouldBypass(ctx context.Context) bool {\n\tif c.bypassCallerTypes == nil {\n\t\treturn false\n\t}\n\n\tcallerInfo := types.GetCallerInfoFromContext(ctx)\n\tbypassCallerTypes := c.bypassCallerTypes()\n\n\tfor _, bypassType := range bypassCallerTypes {\n\t\tif bypassTypeStr, ok := bypassType.(string); ok {\n\t\t\tif types.ParseCallerType(bypassTypeStr) == callerInfo.GetCallerType() {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/quotas/caller_bypass_test.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestCallerBypass_AllowLimiter(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tlimiterAllows     bool\n\t\tbypassCallerTypes []interface{}\n\t\tcallerType        types.CallerType\n\t\texpected          bool\n\t}{\n\t\t{\n\t\t\tname:              \"Limiter allows - bypass not checked\",\n\t\t\tlimiterAllows:     true,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Limiter blocks but caller bypasses\",\n\t\t\tlimiterAllows:     false,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Limiter blocks and caller doesn't bypass\",\n\t\t\tlimiterAllows:     false,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\texpected:          false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Empty bypass list blocks\",\n\t\t\tlimiterAllows:     false,\n\t\t\tbypassCallerTypes: []interface{}{},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpected:          false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockLimiter := NewMockLimiter(ctrl)\n\t\t\tmockLimiter.EXPECT().Allow().Return(tt.limiterAllows)\n\n\t\t\tbypassFn := func(...dynamicproperties.FilterOption) []interface{} {\n\t\t\t\treturn tt.bypassCallerTypes\n\t\t\t}\n\t\t\tcallerBypass := NewCallerBypass(bypassFn)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\t\t\tresult := callerBypass.AllowLimiter(ctx, mockLimiter)\n\n\t\t\tif result != tt.expected {\n\t\t\t\tt.Errorf(\"expected %v, got %v\", tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCallerBypass_AllowPolicy(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tpolicyAllows      bool\n\t\tbypassCallerTypes []interface{}\n\t\tcallerType        types.CallerType\n\t\texpected          bool\n\t}{\n\t\t{\n\t\t\tname:              \"Policy allows - bypass not checked\",\n\t\t\tpolicyAllows:      true,\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Policy blocks but caller bypasses\",\n\t\t\tpolicyAllows:      false,\n\t\t\tbypassCallerTypes: []interface{}{\"ui\", \"cli\"},\n\t\t\tcallerType:        types.CallerTypeUI,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Policy blocks and caller doesn't bypass\",\n\t\t\tpolicyAllows:      false,\n\t\t\tbypassCallerTypes: []interface{}{\"internal\"},\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\texpected:          false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Empty bypass list blocks\",\n\t\t\tpolicyAllows:      false,\n\t\t\tbypassCallerTypes: []interface{}{},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpected:          false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockPolicy := NewMockPolicy(ctrl)\n\t\t\tmockPolicy.EXPECT().Allow(Info{Domain: \"test\"}).Return(tt.policyAllows)\n\n\t\t\tbypassFn := func(...dynamicproperties.FilterOption) []interface{} {\n\t\t\t\treturn tt.bypassCallerTypes\n\t\t\t}\n\t\t\tcallerBypass := NewCallerBypass(bypassFn)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\t\t\tresult := callerBypass.AllowPolicy(ctx, mockPolicy, Info{Domain: \"test\"})\n\n\t\t\tif result != tt.expected {\n\t\t\t\tt.Errorf(\"expected %v, got %v\", tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCallerBypass_ShouldBypass(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tbypassCallerTypes []interface{}\n\t\tcallerType        types.CallerType\n\t\texpected          bool\n\t}{\n\t\t{\n\t\t\tname:              \"CLI bypasses when configured\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Multiple types in list\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\", \"internal\"},\n\t\t\tcallerType:        types.CallerTypeInternal,\n\t\t\texpected:          true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Type not in list doesn't bypass\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeSDK,\n\t\t\texpected:          false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Unknown type doesn't bypass\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\"},\n\t\t\tcallerType:        types.CallerTypeUnknown,\n\t\t\texpected:          false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tbypassFn := func(...dynamicproperties.FilterOption) []interface{} {\n\t\t\t\treturn tt.bypassCallerTypes\n\t\t\t}\n\t\t\tcallerBypass := NewCallerBypass(bypassFn)\n\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\t\t\tresult := callerBypass.ShouldBypass(ctx)\n\n\t\t\tif result != tt.expected {\n\t\t\t\tt.Errorf(\"expected %v, got %v\", tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCallerBypass_NilBypassFunction(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tcallerBypass := NewCallerBypass(nil)\n\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(types.CallerTypeCLI))\n\n\tif callerBypass.ShouldBypass(ctx) {\n\t\tt.Error(\"expected ShouldBypass to return false with nil bypass function\")\n\t}\n\n\tmockLimiter := NewMockLimiter(ctrl)\n\tmockLimiter.EXPECT().Allow().Return(false)\n\tif callerBypass.AllowLimiter(ctx, mockLimiter) {\n\t\tt.Error(\"expected AllowLimiter to return false when limiter blocks and bypass function is nil\")\n\t}\n}\n"
  },
  {
    "path": "common/quotas/collection.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package=$GOPACKAGE -destination=collection_mock.go github.com/uber/cadence/common/quotas ICollection\n//go:generate mockgen -package=$GOPACKAGE -destination=limiterfactory_mock.go github.com/uber/cadence/common/quotas LimiterFactory\n\npackage quotas\n\nimport (\n\t\"sync\"\n)\n\n// LimiterFactory is used to create a Limiter for a given domain\ntype LimiterFactory interface {\n\t// GetLimiter returns a new Limiter for the given domain\n\tGetLimiter(domain string) Limiter\n}\n\n// Collection stores a map of limiters by key\ntype Collection struct {\n\tmu       sync.RWMutex\n\tfactory  LimiterFactory\n\tlimiters map[string]Limiter\n}\n\ntype ICollection interface {\n\tFor(key string) Limiter\n}\n\nvar _ ICollection = (*Collection)(nil)\n\n// NewCollection create a new limiter collection.\n// Given factory is called to create new individual limiter.\nfunc NewCollection(factory LimiterFactory) *Collection {\n\treturn &Collection{\n\t\tfactory:  factory,\n\t\tlimiters: make(map[string]Limiter),\n\t}\n}\n\n// For retrieves limiter by a given key.\n// If limiter for such key does not exists, it creates new one with via factory.\nfunc (c *Collection) For(key string) Limiter {\n\tc.mu.RLock()\n\tlimiter, ok := c.limiters[key]\n\tc.mu.RUnlock()\n\n\tif !ok {\n\t\t// create a new limiter\n\t\tnewLimiter := c.factory.GetLimiter(key)\n\n\t\t// verify that it is needed and add to map\n\t\tc.mu.Lock()\n\t\tlimiter, ok = c.limiters[key]\n\t\tif !ok {\n\t\t\tc.limiters[key] = newLimiter\n\t\t\tlimiter = newLimiter\n\t\t}\n\t\tc.mu.Unlock()\n\t}\n\n\treturn limiter\n}\n"
  },
  {
    "path": "common/quotas/collection_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/quotas (interfaces: ICollection)\n//\n// Generated by this command:\n//\n//\tmockgen -package=quotas -destination=collection_mock.go github.com/uber/cadence/common/quotas ICollection\n//\n\n// Package quotas is a generated GoMock package.\npackage quotas\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockICollection is a mock of ICollection interface.\ntype MockICollection struct {\n\tctrl     *gomock.Controller\n\trecorder *MockICollectionMockRecorder\n\tisgomock struct{}\n}\n\n// MockICollectionMockRecorder is the mock recorder for MockICollection.\ntype MockICollectionMockRecorder struct {\n\tmock *MockICollection\n}\n\n// NewMockICollection creates a new mock instance.\nfunc NewMockICollection(ctrl *gomock.Controller) *MockICollection {\n\tmock := &MockICollection{ctrl: ctrl}\n\tmock.recorder = &MockICollectionMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockICollection) EXPECT() *MockICollectionMockRecorder {\n\treturn m.recorder\n}\n\n// For mocks base method.\nfunc (m *MockICollection) For(key string) Limiter {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"For\", key)\n\tret0, _ := ret[0].(Limiter)\n\treturn ret0\n}\n\n// For indicates an expected call of For.\nfunc (mr *MockICollectionMockRecorder) For(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"For\", reflect.TypeOf((*MockICollection)(nil).For), key)\n}\n"
  },
  {
    "path": "common/quotas/dynamicratelimiter.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nconst _ttl = time.Second * 5\nconst _minBurst = 1\n\n// DynamicRateLimiter implements a dynamic config wrapper around the rate limiter,\n// checks for updates to the dynamic config and updates the rate limiter accordingly\ntype DynamicRateLimiter struct {\n\trps            RPSFunc\n\trl             clock.Ratelimiter\n\ttimeSource     clock.TimeSource\n\tttl            time.Duration\n\tlastUpdateTime atomic.Pointer[time.Time]\n\tminBurst       int\n}\n\ntype DynamicRateLimiterOpts struct {\n\tTTL        time.Duration\n\tMinBurst   int\n\tTimeSource clock.TimeSource\n}\n\nvar _defaultOpts = DynamicRateLimiterOpts{\n\tTTL:        _ttl,\n\tMinBurst:   _minBurst,\n\tTimeSource: clock.NewRealTimeSource(),\n}\n\n// NewDynamicRateLimiter returns a rate limiter which handles dynamic config\nfunc NewDynamicRateLimiter(rps RPSFunc) Limiter {\n\treturn NewDynamicRateLimiterWithOpts(rps, _defaultOpts)\n}\n\nfunc NewDynamicRateLimiterWithOpts(rps RPSFunc, opts DynamicRateLimiterOpts) Limiter {\n\tts := opts.TimeSource\n\tif ts == nil {\n\t\tts = _defaultOpts.TimeSource\n\t}\n\tres := &DynamicRateLimiter{\n\t\trps:        rps,\n\t\ttimeSource: ts,\n\t\tttl:        opts.TTL,\n\t\tminBurst:   opts.MinBurst,\n\t}\n\tnow := res.timeSource.Now()\n\tres.lastUpdateTime.Store(&now)\n\tlim, burst := res.getLimitAndBurst()\n\tres.rl = clock.NewRateLimiterWithTimeSource(res.timeSource, lim, burst)\n\treturn res\n}\n\n// Allow immediately returns with true or false indicating if a rate limit\n// token is available or not\nfunc (d *DynamicRateLimiter) Allow() bool {\n\td.maybeRefreshRps()\n\treturn d.rl.Allow()\n}\n\n// Wait waits up till deadline for a rate limit token\nfunc (d *DynamicRateLimiter) Wait(ctx context.Context) error {\n\td.maybeRefreshRps()\n\treturn d.rl.Wait(ctx)\n}\n\n// Reserve reserves a rate limit token\nfunc (d *DynamicRateLimiter) Reserve() clock.Reservation {\n\td.maybeRefreshRps()\n\treturn d.rl.Reserve()\n}\n\nfunc (d *DynamicRateLimiter) Limit() rate.Limit {\n\td.maybeRefreshRps()\n\treturn d.rl.Limit()\n}\n\nfunc (d *DynamicRateLimiter) maybeRefreshRps() {\n\tnow := d.timeSource.Now()\n\tlastUpdated := d.lastUpdateTime.Load()\n\tif now.After(lastUpdated.Add(d.ttl-1)) && d.lastUpdateTime.CompareAndSwap(lastUpdated, &now) {\n\t\td.rl.SetLimitAndBurst(d.getLimitAndBurst())\n\t}\n}\n\nfunc (d *DynamicRateLimiter) getLimitAndBurst() (rate.Limit, int) {\n\trps := d.rps()\n\tburst := max(int(math.Ceil(rps)), d.minBurst)\n\t// If we have 0 rps we have to zero out the burst to immediately cut off new permits\n\tif rps == 0 {\n\t\tburst = 0\n\t}\n\treturn rate.Limit(rps), burst\n}\n"
  },
  {
    "path": "common/quotas/global/algorithm/requestweighted.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n/*\nPackage algorithm contains a running-weighted-average calculator for ratelimits,\nand some associated types to prevent accidental confusion between the various\nfloats/ints/etc involved.\n\nThis package is intentionally unaware of RPC or any desired RPS limits, it just\ntracks the observed rates of requests and returns proportions, so other code can\ndetermine RPS values.\n\nThis is both to simplify testing and to keep its internals as fast as possible, as\nit is relatively CPU heavy and needs to be quick to prevent a snowballing backlog\nof concurrent updates and requests for aggregated data.\n\n# How this fits in the global ratelimiting system\n\nGoing back to the [github.com/uber/cadence/common/quotas/global] package diagram:\neach aggregating host will have one or more instances of this weight-calculator,\nand it will receive a shard worth of request data.\n\nThough this package has no direct connection to RPC structures, the request data is\nexpected to closely match the argument to RequestWeighted.Update, and the response\ndata like the return value of RequestWeighted.HostWeights.\n\nThis makes aggregating hosts intentionally very simple outside of this package:\nthey essentially just forward the request and response, multiplying by dynamicconfig\nintended-ratelimit values (which occurs outside the mutex, to minimize contention).\n\n# Expected use\n\nLimiting hosts collect metrics and submit it to an aggregating host via\nRequestWeighted.Update, through [some kind of RPC setup].\n\nOnce updated, the aggregating host can get the RequestWeighted.HostWeights for\nthat host's ratelimits (== the updated limits), multiply those 0..1 weights by\nthe dynamicconfig configured RPS, and return them to the limiting host that\ntriggered the request.\n\nIf there are unused RPS remaining, the aggregating host *may* increase the RPS\nit returns by [some amount], to allow the limiting host to pre-emptively allow\nmore requests than is \"fair\" before the next update.  Even if it does not, an\nincrease in attempted usage will increase that limiter's weight on the next\nupdate cycle, so this is mostly intended as a tool for reducing incorrectly-rejected\nrequests when a ratelimit's usage is well below its allowed limit.\n\n# Dealing with expired data\n\nAs user calls change, or the aggregating-host ring changes, Limit keys may become\neffectively unused in an instance.  During normal use, any accessed Limit will\nclean up \"expired\" data if it is found, and there is essentially no ongoing\n\"upkeep\" cost for an un-accessed Limit (aside from memory).\n\nHopefully this will be sufficient to keep memory use and performance reasonable.\n\nIf it is not, a trivial goroutine to periodically call RequestWeighted.GC will\nclear *all* old data.  Every minute or so should be more than sufficient.\nThis method returns some simple metrics about how much data exists / was removed,\nso it can be reported to help us estimate how necessary it is in practice.\n\n# Dealing with excessive contention\n\nIn large clusters, there will be likely be many update-requests, many Limit keys,\nand more data to process to return responses (more hosts per limit).  At some point\nthis could become a bottleneck, preventing timely updates.\n\nOn even our largest internal clusters, I do not believe we will run that risk.\nAt least, not with the planned frontend-only limits.  `pprof` should be able\nto easily validate this and let us better estimate headroom for adding more in\nthe future.\n\nIf too much contention does occur, there are 3 relatively simple mitigations:\n 1. turn it completely off, go back to host-local limits\n 2. slow down the update frequency\n 3. use N instances and shard keys across them\n\n1 is pretty obvious and has clear behavior, though it means degrading behavior\nfor our imbalanced-load clusters.\n\n2 is essentially the only short-term and dynamically-apply-able option that retains\nany of the behavior we want.  This will impact how quickly the algorithm converges,\nso you may also want to adjust the new-data weight to be higher, though \"how much\"\ndepends on what kind of behavior you want / can tolerate.\n\n3 offers a linear contention improvement and should be basically trivial to build:\ncreate N instances instead of 1, and shard the Limit keys to each instance.\nSince this is mostly CPU bound and each one would be fully independent, making\n`GOMAXPROCS` instances is an obvious first choice, and this does not need to be\ndynamically reconfigured at runtime so there should be no need to build a \"smart\"\nre-balancing / re-sharding system of any kind.\n\nTests contain a single-threaded benchmark and laptop-local results for estimating\ncontention, and for judging if changes will help or hinder, but real-world use and\ndifferent CPUs will of course be somewhat different.\nPersonally I expect \"few mutexes, GOMAXPROCS instances\" is roughly ideal for CPU\nthroughput with the current setup, but an entirely different internal structure\nmight exceed it.\n*/\npackage algorithm\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/multierr\"\n\t\"golang.org/x/exp/constraints\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\ntype (\n\t// requests holds the running per-second running average request data for a single key from a single host,\n\t// and the last time it was updated.\n\trequests struct {\n\t\t// TODO: add \"first update\" time, don't return to callers until it has passed 1(+?) update cycle as data is incomplete.\n\n\t\tlastUpdate         time.Time // only so we know if elapsed times are exceeded, not used to compute per-second rates\n\t\taccepted, rejected PerSecond // requests received, per second (conceptually divided by update rate)\n\t}\n\n\t// Identity is an arbitrary (stable) identifier for a \"limiting\" host.\n\tIdentity string\n\t// Limit is the key being used to Limit requests.\n\tLimit string\n\t// PerSecond represents already-scaled-to-per-second RPS values\n\tPerSecond float64\n\t// HostWeight is a float between 0 and 1, showing how much of a ratelimit should be allocated to a host.\n\tHostWeight float64\n\n\timpl struct {\n\t\t// intentionally value-typed so caller cannot mutate the fields.\n\t\t// manually copy data if this changes.\n\t\tcfg    Config\n\t\tscope  metrics.Scope\n\t\tlogger log.Logger\n\n\t\t// mut protects usage, as it is the only mutable data\n\t\tmut sync.Mutex\n\t\t// usage data for ratelimits, per host.\n\t\t//\n\t\t// data is first keyed on the limit and then on the host, as it's assumed that there will be\n\t\t// many times more limit keys than limiting hosts, and this will reduce cardinality more rapidly.\n\t\tusage map[Limit]map[Identity]requests\n\n\t\tclock clock.TimeSource\n\t}\n\n\t// Metrics reports overall counts discovered as part of garbage collection.\n\t//\n\t// This is not necessarily worth maintaining verbatim, but while it's easy to\n\t// collect it might give us some insight into overall behavior.\n\tMetrics struct {\n\t\t// HostLimits is the number of per-host limits that remain after cleanup\n\t\tHostLimits int\n\t\t// Limits is the number of cross-host limits that remain after cleanup\n\t\tLimits int\n\n\t\t// RemovedHostLimits is the number of per-host limits that were removed as part of this GC pass\n\t\tRemovedHostLimits int\n\t\t// RemovedLimits is the number of cross-host limits that were removed as part of this GC pass\n\t\tRemovedLimits int\n\t}\n\n\t// Requests contains accepted/rejected request counts for a ratelimit, as part of an update operation.\n\t// Only intended to be used in [RequestWeighted.Update].\n\tRequests struct {\n\t\tAccepted, Rejected int\n\t}\n\n\t// RequestWeighted returns aggregated RPS and distribution data for its Limit keys,\n\t// based on how many requests each Identity has received.\n\tRequestWeighted interface {\n\t\t// Update load-data for this host's requests, given a known elapsed time spent accumulating this load info.\n\t\t//\n\t\t// Elapsed time must be non-zero, but is not otherwise constrained.\n\t\tUpdate(params UpdateParams) error\n\n\t\t// HostUsage returns the per-[Limit] weights for all requested + known keys for this Identity,\n\t\t// as well as the Limit's overall used RPS (to decide RPS to allow for new hosts).\n\t\tHostUsage(host Identity, limits []Limit) (weights map[Limit]HostUsage, err error)\n\n\t\t// GC can be called periodically to pre-emptively prune old ratelimits.\n\t\t//\n\t\t// Limit keys that are accessed normally will automatically garbage-collect themselves and old host data,\n\t\t// and an unused limit only costs memory, so this may prove to be unnecessary.\n\t\tGC() (Metrics, error)\n\t}\n\n\tConfig struct {\n\t\t// How much each update should be weighted vs prior data.\n\t\t// Must be between 0 and 1, recommend starting with 0.5 (4 updates until data has <10% influence)\n\t\tNewDataWeight dynamicproperties.FloatPropertyFn\n\n\t\t// Expected time between updates.  Should match the cluster's config for how often limiters check in,\n\t\t// i.e. this should probably be the same dynamic config value, updated at / near the same time.\n\t\tUpdateInterval dynamicproperties.DurationPropertyFn\n\n\t\t// How long to wait before considering a host-limit's RPS usage \"probably inactive\", rather than\n\t\t// simply delayed.\n\t\t//\n\t\t// Should always be larger than UpdateInterval, as less is meaningless.  Values are reduced based on\n\t\t// missed UpdateInterval multiples, not DecayAfter.\n\t\t// Unsure about a good default (try 2x UpdateInterval?), but larger numbers mean smoother behavior\n\t\t// but longer delays on adjusting to hosts that have disappeared or stopped receiving requests.\n\t\tDecayAfter dynamicproperties.DurationPropertyFn\n\n\t\t// How much time can pass without receiving any update before completely deleting data.\n\t\t//\n\t\t// Due to ever-reducing weight after DecayAfter, this is intended to reduce calculation costs,\n\t\t// not influence behavior / weights returned.  Even extremely-low-weighted hosts will still be retained\n\t\t// as long as they keep reporting \"in use\" (e.g. 1 rps used out of >100,000 is fine and will be tracked).\n\t\t//\n\t\t// \"Good\" values depend on a lot of details, but >=10*UpdateInterval seems reasonably safe for a\n\t\t// NewDataWeight of 0.5, as the latest data will be reduced to only 0.1% and may not be worth keeping.\n\t\tGcAfter dynamicproperties.DurationPropertyFn\n\t}\n\n\t// UpdateParams contains args for calling Update.\n\tUpdateParams struct {\n\t\tID      Identity\n\t\tLoad    map[Limit]Requests\n\t\tElapsed time.Duration\n\t}\n\n\tHostUsage struct {\n\t\tWeight HostWeight\n\t\tUsed   PerSecond\n\t}\n\n\t// configSnapshot holds a non-changing snapshot of the dynamic config values,\n\t// and also provides a validate() method to make sure they're sane.\n\tconfigSnapshot struct {\n\t\tnow        time.Time\n\t\tweight     float64\n\t\trate       time.Duration\n\t\tdecayAfter time.Duration\n\t\tgcAfter    time.Duration\n\t}\n)\n\n// these could be configurable, but it's not expected to be a noticeable performance concern.\nconst (\n\tguessNumKeys   = 1024 // guesstimate at num of ratelimit keys in a cluster\n\tguessHostCount = 32   // guesstimate at num of frontend hosts in a cluster that receive traffic for each key\n\n\t// Some event-RPSes exceed 100k, but none get close to 1m.\n\t// So, to add some buffer, 10m per second is considered \"impossible\".\n\t// This is quite vague and can be changed, it essentially just serves as the logging threshold.\n\tguessImpossibleRps = 10_000_000\n)\n\nfunc (p UpdateParams) Validate() error {\n\tif len(p.ID) == 0 {\n\t\treturn fmt.Errorf(\"empty caller ID\")\n\t}\n\tif p.Elapsed <= 0 {\n\t\treturn fmt.Errorf(\"elapsed time must be positive, got %v\", p.Elapsed)\n\t}\n\treturn nil\n}\n\nfunc (c configSnapshot) validate() error {\n\t// errors are untyped because they should not generally be \"handled\", only returned.\n\t// in principle, they're all \"bad server config\" / 5XX and if sustained will eventually lead to\n\t// the limiting hosts using their fallback limiters.\n\n\tvar err error\n\tif c.weight < 0 {\n\t\t// nonsensical\n\t\terr = multierr.Append(err, fmt.Errorf(\"new data weight cannot be negative: %f\", c.weight))\n\t}\n\tif c.rate <= 0 {\n\t\t// rate is used in division, absolutely must not be zero.\n\t\t// and negatives would just be weird enough to consider an error too.\n\t\terr = multierr.Append(err, fmt.Errorf(\"update rate must be positive: %v\", c.rate))\n\t}\n\n\t// decayAfter and gcAfter should always be larger than rate, but currently no logic misbehaves\n\t// if this is not true, so it is not checked.  this allows e.g. temporary weird mid-rollout combinations\n\t// without breaking.\n\t// negative values cannot ever be correct though, so block them.\n\tif c.decayAfter < 0 {\n\t\terr = multierr.Append(err, fmt.Errorf(\"decay-after cannot be negative: %v\", c.decayAfter))\n\t}\n\tif c.gcAfter < 0 {\n\t\terr = multierr.Append(err, fmt.Errorf(\"gc-after cannot be negative: %v\", c.decayAfter))\n\t}\n\n\tif err != nil {\n\t\treturn multierr.Append(errors.New(\"bad ratelimiter config\"), err)\n\t}\n\n\treturn nil\n}\n\n// shouldGC returns true if data should be garbage collected\nfunc (c configSnapshot) shouldGC(dataAge time.Duration) bool {\n\treturn dataAge >= c.gcAfter\n}\n\n// missedUpdateScalar returns an amount to multiply old RPS data by, to account for missed updates outside SLA.\nfunc (c configSnapshot) missedUpdateScalar(dataAge time.Duration) PerSecond {\n\tif dataAge < c.decayAfter {\n\t\t// new enough to not decay old data\n\t\treturn 1\n\t}\n\t// fast check as exponents are slow and computing past this is unnecessary.\n\t// normally, a `c.shouldGC(dataAge)` check should make this unused.\n\tif dataAge >= c.gcAfter {\n\t\t// old enough to treat old data as nonexistent\n\t\treturn 0\n\t}\n\n\t// somewhere in the middle: account for missed updates by simulating 0-value updates.\n\t// this intentionally floors rather than allowing fractions because:\n\t// - precision isn't important\n\t// - tests are a bit easier (more stable / less crazy-looking values)\n\t// - as a bonus freebie: int division and int exponents are typically faster to compute\n\tmissed := dataAge / c.rate\n\tif missed < 1 {\n\t\t// note: this path should never be used, it's effectively error handling.\n\t\t//\n\t\t// this can only be true if cfg.decayAfter is smaller than cfg.rate, which\n\t\t// should not occur in practice as that would be a nonsensical config combination.\n\t\t//\n\t\t// the types cannot *prevent* it though, so it must be handled, and there's a\n\t\t// reasonable response if it does occur:\n\t\t// don't reduce old data.\n\t\treturn 1\n\t}\n\n\t// missed at least one full update period, calculate an exponential decay multiplier for the old values.\n\treturn PerSecond(math.Pow(1-c.weight, math.Floor(float64(missed))))\n}\n\n// New returns a concurrency-safe host-weight aggregator.\n//\n// This instance is effectively single-threaded, but a small sharding wrapper should allow better concurrent\n// throughput if needed (bound by CPU cores, as it's moderately CPU-costly).\nfunc New(met metrics.Client, logger log.Logger, cfg Config) (RequestWeighted, error) {\n\ti := &impl{\n\t\tcfg:    cfg,\n\t\tscope:  met.Scope(metrics.GlobalRatelimiterAggregator),\n\t\tlogger: logger.WithTags(tag.ComponentGlobalRatelimiter),\n\t\tusage:  make(map[Limit]map[Identity]requests, guessNumKeys), // start out relatively large\n\n\t\tclock: clock.NewRealTimeSource(),\n\t}\n\t_, err := i.snapshot() // validate config by just taking a snapshot\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid config: %w\", err)\n\t}\n\treturn i, nil\n}\n\n// Update performs a weighted update to the running RPS for this host's per-key request data\nfunc (a *impl) Update(p UpdateParams) error {\n\tif err := p.Validate(); err != nil {\n\t\treturn fmt.Errorf(\"bad args to update: %w\", err)\n\t}\n\ta.mut.Lock()\n\tonce := newOnce()\n\tdefer once.Do(a.mut.Unlock)\n\n\tvar initialized, reinitialized, updated, decayed int64\n\n\tsnap, err := a.snapshot()\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor key, req := range p.Load {\n\t\tih := a.usage[key]\n\t\tif ih == nil {\n\t\t\tih = make(map[Identity]requests, guessHostCount)\n\t\t}\n\n\t\tvar next requests\n\t\tprev := ih[p.ID]\n\n\t\t// sanity check: elapsed time should not be less than 1s, so just cap it.\n\t\t// in practice this should always be safe with a >=1s configured rate, as\n\t\t// the caller should not send *more* frequently than every 1s (monotonic time).\n\t\t//\n\t\t// but this is rather easy to trigger in tests / fuzzing,\n\t\t// and extreme values lead to irrational math either way.\n\t\telapsed := math.Max(float64(p.Elapsed), float64(time.Second))\n\t\taps := shared.SanityLogFloat(0, PerSecond(float64(req.Accepted)/(elapsed/float64(time.Second))), guessImpossibleRps, \"accepted rps\", a.logger)\n\t\trps := shared.SanityLogFloat(0, PerSecond(float64(req.Rejected)/(elapsed/float64(time.Second))), guessImpossibleRps, \"rejected rps\", a.logger)\n\n\t\t// zeros are not worth recording, and this also simplifies math elsewhere\n\t\t// for two major reasons:\n\t\t// - it prevents some divide-by-zero scenarios by simply not having actual zeros\n\t\t// - it prevents weights from perpetually lowering if zeros are repeatedly sent, where they may eventually reach zero\n\t\t//\n\t\t// these keys will eventually gc, just leave them alone until that happens.\n\t\t// currently this gc relies on the assumption that HostUsage will be called with the same set of keys \"soon\",\n\t\t// but that is fairly easy to fix if needed.\n\t\tif rps+aps == 0 {\n\t\t\tcontinue\n\t\t}\n\n\t\tif prev.lastUpdate.IsZero() {\n\t\t\tinitialized++\n\t\t\tnext = requests{\n\t\t\t\tlastUpdate: snap.now,\n\t\t\t\taccepted:   aps, // no requests == 100% weight\n\t\t\t\trejected:   rps, // no requests == 100% weight\n\t\t\t}\n\t\t} else {\n\t\t\tage := snap.now.Sub(prev.lastUpdate)\n\t\t\tif snap.shouldGC(age) {\n\t\t\t\treinitialized++\n\t\t\t\t// would have GC'd if we had seen it earlier, so it's the same as the zero state\n\t\t\t\tnext = requests{\n\t\t\t\t\tlastUpdate: snap.now,\n\t\t\t\t\taccepted:   aps, // no requests == 100% weight\n\t\t\t\t\trejected:   rps, // no requests == 100% weight\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tupdated++\n\t\t\t\t// compute the next rolling average step (`*reduce` simulates skipped updates)\n\t\t\t\treduce := snap.missedUpdateScalar(age)\n\t\t\t\tif reduce < 1 {\n\t\t\t\t\tdecayed++\n\t\t\t\t}\n\t\t\t\tnext = requests{\n\t\t\t\t\tlastUpdate: snap.now,\n\t\t\t\t\t// TODO: max(1, actual) so this does not lead to <1 rps allowances?  or maybe just 1+actual and then reduce in used-responses?\n\t\t\t\t\t// otherwise currently this may lead to rare callers getting 0.0001 rps,\n\t\t\t\t\t// and never recovering, despite steady and fair usage.\n\t\t\t\t\taccepted: shared.SanityLogFloat(0, weighted(aps, prev.accepted*reduce, snap.weight), guessImpossibleRps, \"weighted accepted rps\", a.logger),\n\t\t\t\t\trejected: shared.SanityLogFloat(0, weighted(rps, prev.rejected*reduce, snap.weight), guessImpossibleRps, \"weighted rejected rps\", a.logger),\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tih[p.ID] = next\n\t\ta.usage[key] = ih\n\t}\n\n\tonce.Do(a.mut.Unlock) // don't hold the lock while emitting metrics\n\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterInitialized, float64(initialized))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterReinitialized, float64(reinitialized))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterUpdated, float64(updated))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterDecayed, float64(decayed))\n\n\treturn nil\n}\n\n// getWeightsLocked returns the weights of observed hosts (based on ALL requests), and the total number of requests accepted per second.\nfunc (a *impl) getWeightsLocked(key Limit, snap configSnapshot) (weights map[Identity]HostWeight, usedRPS PerSecond, met Metrics) {\n\tir := a.usage[key]\n\tif len(ir) == 0 {\n\t\treturn nil, 0, met\n\t}\n\n\tweights = make(map[Identity]HostWeight, len(ir))\n\ttotal := HostWeight(0.0)\n\tfor id, reqs := range ir {\n\t\t// account for missed updates\n\t\tage := snap.now.Sub(reqs.lastUpdate)\n\t\tif snap.shouldGC(age) {\n\t\t\t// old, clean up\n\t\t\tdelete(ir, id)\n\t\t\tmet.RemovedHostLimits++\n\t\t\tcontinue\n\t\t}\n\n\t\t// should never be zero, `shouldGC` takes care of that.\n\t\treduce := shared.SanityLogFloat(0, snap.missedUpdateScalar(age), 1, \"missed update\", a.logger)\n\t\t// similarly: should never be zero, accepted + rejected must be nonzero or they are not inserted.\n\t\t// this may be reduced to very low values, but still far from == 0.\n\t\tactual := HostWeight((reqs.accepted + reqs.rejected) * reduce)\n\n\t\tweights[id] = actual // populate with the reduced values so it doesn't have to be calculated again\n\t\ttotal += actual      // keep a running total to scale all values when done\n\t\tusedRPS += reqs.accepted * reduce\n\t\tmet.HostLimits++\n\t}\n\n\tif len(ir) == 0 {\n\t\t// completely empty Limit, gc it as well\n\t\tdelete(a.usage, key)\n\t\tmet.RemovedLimits++\n\t\treturn nil, 0, met\n\t}\n\n\t// zeros anywhere here should not be possible - they are prevented from being inserted,\n\t// and anything simply \"losing weight\" will only become \"rather low\", not zero,\n\t// before enough passes have occurred to garbage collect it.\n\t//\n\t// specifically, 1rps -> 0rps takes over 1,000 halving cycles, and a single 1rps event\n\t// during that will immediately re-set it above 0.5 and will need 1,000+ more cycles.\n\t//\n\t// if gc period / weight amount is set extreme enough this is \"possible\",\n\t// but we are unlikely to ever cause it.\n\tfor id := range ir {\n\t\t// normalize by the total.\n\t\t// this also ensures all values are between 0 and 1 (inclusive),\n\t\t// though zero should be impossible.\n\t\tweights[id] = shared.SanityLogFloat(0, weights[id]/total, 1, \"normalized weight\", a.logger)\n\t}\n\tmet.Limits = 1\n\treturn weights, usedRPS, met\n}\n\nfunc (a *impl) HostUsage(host Identity, limits []Limit) (usage map[Limit]HostUsage, err error) {\n\ta.mut.Lock()\n\tonce := newOnce()\n\tdefer once.Do(a.mut.Unlock)\n\n\tsnap, err := a.snapshot()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar cumulative Metrics\n\tusage = make(map[Limit]HostUsage, len(limits))\n\tfor _, lim := range limits {\n\t\thosts, used, met := a.getWeightsLocked(lim, snap)\n\n\t\tcumulative.Limits += met.Limits // always 1 or 0\n\t\tcumulative.HostLimits += met.HostLimits\n\t\tcumulative.RemovedLimits += met.RemovedLimits // always 0 or 1 (opposite Limits)\n\t\tcumulative.RemovedHostLimits += met.RemovedHostLimits\n\n\t\tif len(hosts) > 0 {\n\t\t\tusage[lim] = HostUsage{\n\t\t\t\t// limit is known, has some usage on at least one host.\n\t\t\t\t// usage has an \"upper limit\" because it is only the accepted RPS, not all requests received.\n\t\t\t\tUsed: shared.SanityLogFloat(0, used, guessImpossibleRps, \"used rps\", a.logger),\n\t\t\t\t// either known weight if there is info for this host, or zero if not.\n\t\t\t\t// zeros are interpreted as \"unknown\", the same as \"not present\".\n\t\t\t\tWeight: shared.SanityLogFloat(0, hosts[host], 1, \"computed weight\", a.logger),\n\t\t\t}\n\t\t}\n\t}\n\n\tonce.Do(a.mut.Unlock) // don't hold the lock while emitting metrics\n\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterLimitsQueried, float64(cumulative.Limits))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterHostLimitsQueried, float64(cumulative.HostLimits))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterRemovedLimits, float64(cumulative.RemovedLimits))\n\ta.scope.RecordHistogramValue(metrics.GlobalRatelimiterRemovedHostLimits, float64(cumulative.RemovedHostLimits))\n\n\treturn usage, nil\n}\n\nfunc (a *impl) GC() (Metrics, error) {\n\ta.mut.Lock()\n\tdefer a.mut.Unlock()\n\n\tm := Metrics{}\n\tsnap, err := a.snapshot()\n\tif err != nil {\n\t\treturn Metrics{}, err\n\t}\n\t// TODO: too costly? can check the first-N% each time and it'll eventually visit all keys, demonstrated in tests.\n\tfor lim, dat := range a.usage {\n\t\tfor host, reqs := range dat {\n\t\t\tage := snap.now.Sub(reqs.lastUpdate)\n\t\t\tif snap.shouldGC(age) {\n\t\t\t\t// clean up stale host data within limits\n\t\t\t\tdelete(dat, host)\n\t\t\t\tm.RemovedHostLimits++\n\t\t\t} else {\n\t\t\t\tm.HostLimits++\n\t\t\t}\n\t\t}\n\n\t\t// clean up stale limits\n\t\tif len(dat) == 0 {\n\t\t\tdelete(a.usage, lim)\n\t\t\tm.RemovedLimits++\n\t\t} else {\n\t\t\tm.Limits++\n\t\t}\n\t}\n\n\treturn m, nil\n}\n\n// returns a snapshot of config and \"now\" for easier chaining through calls, and reducing calls to\n// non-trivial field types like dynamic config.\nfunc (a *impl) snapshot() (configSnapshot, error) {\n\tsnap := configSnapshot{\n\t\tnow:        a.clock.Now(),\n\t\tweight:     a.cfg.NewDataWeight(),\n\t\trate:       a.cfg.UpdateInterval(),\n\t\tdecayAfter: a.cfg.DecayAfter(),\n\t\tgcAfter:    a.cfg.GcAfter(),\n\t}\n\treturn snap, snap.validate()\n}\n\nfunc weighted[T numeric](newer, older T, weight float64) T {\n\treturn T((float64(newer) * weight) +\n\t\t(float64(older) * (1 - weight)))\n}\n\ntype numeric interface {\n\tconstraints.Integer | constraints.Float\n}\n\n// non-sync version of sync.Once, for easier unlocking\ntype doOnce bool\n\nfunc newOnce() *doOnce {\n\tvalue := doOnce(false)\n\treturn &value\n}\n\nfunc (o *doOnce) Do(cb func()) {\n\tif *o == false {\n\t\t*o = true\n\t\tcb()\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/algorithm/requestweighted_fuzz_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage algorithm\n\nimport (\n\t\"encoding/binary\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"golang.org/x/exp/maps\"\n\t\"golang.org/x/exp/slices\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\n// Calls Update and HostWeights as many times as there is sufficient data,\n// to make sure multiple updates do not lead to Infs or NaNs, because floating point is hard.\n//\n// This is best to let run for a very long time after making changes, as odd sequences may be necessary.\n// E.g. it took nearly 30 minutes to hit a case where `elapsed` had an `INT_MIN` value from the data bytes,\n// which failed earlier attempts to math.Abs it with `-val` because `-INT_MIN == INT_MIN`.\n//\n// Your *local* fuzz corpus will help re-running to find more interesting things more quickly,\n// but a cleared cache will have to start over and may take a long time.\nfunc FuzzMultiUpdate(f *testing.F) {\n\tf.Fuzz(func(t *testing.T, data []byte) {\n\t\taccept, reject, elapsed, data, msg := ints(data)\n\t\tif msg != \"\" {\n\t\t\tt.Skip(msg)\n\t\t}\n\t\tvar host, key string\n\t\tif len(data) > 1 {\n\t\t\t// single-char strings are probably best, some collision is a good thing.\n\t\t\thost, key = string(data[0]), string(data[1])\n\t\t\t// data = data[2:] // leaving the data in there is fine, it'll just become part of the accept/reject ints.\n\t\t} else {\n\t\t\tt.Skip(\"not enough data for keys\")\n\t\t}\n\n\t\tl, obs := testlogger.NewObserved(t) // failed sanity checks will fail the fuzz test, as they use .Error\n\t\ti, err := New(metrics.NewNoopMetricsClient(), l, Config{\n\t\t\t// TODO: fuzz test with different weights too? though only some human-friendly values are likely, like 0.25..0.75\n\t\t\t// extremely high or low values are kinda-intentionally allowed to misbehave since they're really not rational to set,\n\t\t\t// but it might be a good exercise to make sure the math is reasonable even in those edge cases...\n\t\t\tNewDataWeight:  func(opts ...dynamicproperties.FilterOption) float64 { return 0.5 },\n\t\t\tUpdateInterval: func(opts ...dynamicproperties.FilterOption) time.Duration { return time.Second },\n\t\t\tDecayAfter:     func(opts ...dynamicproperties.FilterOption) time.Duration { return time.Hour },\n\t\t\tGcAfter:        func(opts ...dynamicproperties.FilterOption) time.Duration { return time.Hour },\n\t\t})\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\n\t\t// if it takes more than a couple seconds, fuzz considers it stuck.\n\t\t// 1000 seems too long, 100 is pretty quick, probably just lower if necessary.\n\t\tfor iter := 0; iter < 100; iter++ {\n\t\t\t// send as many updates as we have data for\n\t\t\tt.Logf(\"accept=%d, reject=%d, elapsed=%d, host=%q, key=%q\", accept, reject, elapsed, host, key)\n\n\t\t\terr = i.Update(UpdateParams{\n\t\t\t\tID: Identity(host),\n\t\t\t\tLoad: map[Limit]Requests{\n\t\t\t\t\tLimit(key): {\n\t\t\t\t\t\tAccepted: int(accept),\n\t\t\t\t\t\tRejected: int(reject),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tElapsed: time.Duration(int(elapsed)),\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tt.Fatal(err)\n\t\t\t}\n\n\t\t\tu := i.(*impl).usage\n\n\t\t\t// collect and sort the layers of map keys,\n\t\t\t// as determinism helps the fuzzing system choose branches better.\n\t\t\tkeySet := make(map[Limit]struct{})\n\t\t\tidentSet := make(map[Identity]struct{})\n\t\t\tfor limitKey, hostUsage := range u {\n\t\t\t\tkeySet[limitKey] = struct{}{}\n\t\t\t\tfor ident := range hostUsage {\n\t\t\t\t\tidentSet[ident] = struct{}{}\n\t\t\t\t}\n\t\t\t}\n\t\t\tif accept+reject > 0 && len(keySet) == 0 {\n\t\t\t\tt.Error(\"no identities\")\n\t\t\t}\n\t\t\tif accept+reject > 0 && len(identSet) == 0 {\n\t\t\t\tt.Error(\"no keys\")\n\t\t\t}\n\n\t\t\tkeys := maps.Keys(keySet)\n\t\t\tidents := maps.Keys(identSet)\n\t\t\tslices.Sort(keys)\n\t\t\tslices.Sort(idents)\n\n\t\t\t// scan for NaNs and Infs in internal data\n\t\t\tfor _, k := range keys {\n\t\t\t\thu := u[k]\n\t\t\t\tfor _, ident := range idents {\n\t\t\t\t\tus := hu[ident]\n\t\t\t\t\tif bad := fsane(us.accepted); bad != \"\" {\n\t\t\t\t\t\tt.Errorf(\"%v for accepted rps:%q, key:%q\", bad, k, ident)\n\t\t\t\t\t}\n\t\t\t\t\tif bad := fsane(us.rejected); bad != \"\" {\n\t\t\t\t\t\tt.Errorf(\"%v for rejected rps:%q, key:%q\", bad, k, ident)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// and get all hosts and check the weight calculations for all keys too\n\t\t\tfor _, ident := range idents {\n\t\t\t\tres, err := i.HostUsage(ident, keys)\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Fatal(err)\n\t\t\t\t}\n\t\t\t\tif len(res) == 0 {\n\t\t\t\t\tt.Error(\"no results\")\n\t\t\t\t}\n\t\t\t\tfor _, k := range keys {\n\t\t\t\t\tr, ok := res[k]\n\t\t\t\t\tif !ok {\n\t\t\t\t\t\t// currently all requested keys are expected to be returned\n\t\t\t\t\t\tt.Errorf(\"key not found: %q\", k)\n\t\t\t\t\t}\n\t\t\t\t\tif bad := fsane(r.Used); bad != \"\" {\n\t\t\t\t\t\tt.Error(bad, \"usage\")\n\t\t\t\t\t}\n\t\t\t\t\tif r.Used < 0 {\n\t\t\t\t\t\tt.Error(\"negative usage\")\n\t\t\t\t\t}\n\t\t\t\t\tif bad := fsane(r.Weight); bad != \"\" {\n\t\t\t\t\t\tt.Error(bad, \"weight\")\n\t\t\t\t\t}\n\t\t\t\t\tif r.Weight < 0 {\n\t\t\t\t\t\tt.Error(\"negative weight\")\n\t\t\t\t\t} else if r.Weight > 1 {\n\t\t\t\t\t\tt.Error(\"too much weight\")\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// check for error logs, as this would imply sanity check violations\n\t\t\tshared.AssertNoSanityCheckFailures(t, obs.TakeAll())\n\n\t\t\t// refresh for the next round\n\t\t\taccept, reject, elapsed, data, msg = ints(data)\n\t\t\tif msg != \"\" {\n\t\t\t\tbreak // not enough data for another round\n\t\t\t}\n\t\t\tif len(data) > 1 {\n\t\t\t\thost, key = string(data[0]), string(data[1])\n\t\t\t\t// data = data[2:] // leaving the data in there is fine\n\t\t\t} else {\n\t\t\t\tbreak // not enough data for another round\n\t\t\t}\n\t\t}\n\t})\n}\n\n// float sanity check because it's a lot to write out every time\nfunc fsane[T ~float64](t T) string {\n\tif math.IsNaN(float64(t)) {\n\t\treturn \"NaN\"\n\t}\n\tif math.IsInf(float64(t), 0) {\n\t\treturn \"Inf\"\n\t}\n\treturn \"\"\n}\n\n// helper to pull varints out of a pile of bytes.\n// returns the unused portion of data,\n// and returns a non-empty string if there was a problem\nfunc ints(inData []byte) (accept, reject, elapsed int64, data []byte, err string) {\n\tdata = inData\n\taccept, data, msg := getPositiveInt64(data)\n\tif msg != \"\" {\n\t\treturn 0, 0, 0, nil, \"not enough data\"\n\t}\n\treject, data, msg = getPositiveInt64(data)\n\tif msg != \"\" {\n\t\treturn 0, 0, 0, nil, \"not enough data\"\n\t}\n\telapsed, data, msg = getPositiveInt64(data)\n\tif msg != \"\" {\n\t\treturn 0, 0, 0, nil, \"not enough data\"\n\t}\n\tif elapsed == 0 {\n\t\treturn 0, 0, 0, nil, \"zero elapsed time, cannot use\"\n\t}\n\treturn accept, reject, elapsed, data, \"\"\n}\n\n// helpers need helpers sometimes\nfunc getPositiveInt64(data []byte) (int64, []byte, string) {\n\tval, read := binary.Varint(data)\n\tif read == 0 {\n\t\treturn 0, nil, \"not enough data\"\n\t}\n\tif read > 0 {\n\t\tdata = data[read:] // successfully read an int\n\t}\n\tif read < 0 {\n\t\tdata = data[-read:] // overflowed and stopped reading part way\n\t}\n\tif val < 0 {\n\t\tval = -val\n\t}\n\tif val < 0 {\n\t\t// -INT_MIN == INT_MIN, so the above check may have done nothing.\n\t\t// so special case INT_MIN by rolling it over to INT_MAX.\n\t\tval--\n\t}\n\t// and last but not least: make sure it's below our \"impossible\" value when accept+reject are combined.\n\t// partly this ensures random fuzzed rps doesn't exceed it (which is common),\n\t// and partly it just asserts \"we do not test irrational rates\".\n\tval = val % ((guessImpossibleRps - 1) / 2)\n\treturn val, data, \"\"\n}\n\n// not covered by the larger fuzz test because it pins some \"reasonable\" values.\nfunc FuzzMissedUpdate(f *testing.F) {\n\tf.Fuzz(func(t *testing.T, decay, gc, age, rate int, weight float64) {\n\t\tif decay <= 0 || gc <= 0 || age <= 0 || rate <= 0 {\n\t\t\tt.Skip()\n\t\t}\n\t\tscalar := configSnapshot{\n\t\t\tweight:     weight - math.Floor(weight), // 0..1, idk about negatives but it doesn't seem to break either\n\t\t\trate:       time.Duration(rate),\n\t\t\tdecayAfter: time.Duration(decay),\n\t\t\tgcAfter:    time.Duration(gc),\n\t\t}.missedUpdateScalar(time.Duration(age))\n\t\tif scalar < 0 || scalar > 1 {\n\t\t\tt.Fail()\n\t\t}\n\t\tif bad := fsane(scalar); bad != \"\" {\n\t\t\tt.Error(bad, \"scalar\")\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "common/quotas/global/algorithm/requestweighted_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage algorithm\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/multierr\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\n// just simplifies newForTest usage as most tests only care about rate\nfunc defaultConfig(rate time.Duration) configSnapshot {\n\treturn configSnapshot{\n\t\t// now:    , ignored\n\n\t\t// intentionally avoiding 0.5 because it cannot tell if the code uses\n\t\t// `weight` or `1-weight`, which is usually relevant.\n\t\t//\n\t\t// 0.1 is relatively human-math-friendly for a single step,\n\t\t// but is otherwise arbitrary.\n\t\tweight:     0.1,\n\t\trate:       rate,\n\t\tdecayAfter: 2 * rate,\n\t\tgcAfter:    10 * rate,\n\t}\n}\n\nfunc newValid(t testlogger.TestingT, snap configSnapshot) (*impl, clock.MockedTimeSource) {\n\treturn newForTest(t, snap, true)\n}\n\nfunc newForTest(t testlogger.TestingT, snap configSnapshot, validate bool) (*impl, clock.MockedTimeSource) {\n\tcfg := Config{\n\t\tNewDataWeight: func(_ ...dynamicproperties.FilterOption) float64 {\n\t\t\treturn snap.weight\n\t\t},\n\t\tUpdateInterval: func(_ ...dynamicproperties.FilterOption) time.Duration {\n\t\t\treturn snap.rate\n\t\t},\n\t\tDecayAfter: func(_ ...dynamicproperties.FilterOption) time.Duration {\n\t\t\treturn snap.decayAfter\n\t\t},\n\t\tGcAfter: func(_ ...dynamicproperties.FilterOption) time.Duration {\n\t\t\treturn snap.gcAfter\n\t\t},\n\t}\n\tvar agg *impl\n\n\tif validate {\n\t\tl, obs := testlogger.NewObserved(t)\n\t\tt.Cleanup(func() {\n\t\t\tshared.AssertNoSanityCheckFailures(t, obs.TakeAll())\n\t\t})\n\t\ti, err := New(metrics.NewNoopMetricsClient(), l, cfg)\n\t\trequire.NoError(t, err)\n\t\tagg = i.(*impl)\n\t} else {\n\t\t// need to build by hand, New returns nil on err\n\t\tagg = &impl{\n\t\t\tcfg:   cfg,\n\t\t\tscope: metrics.NewNoopMetricsClient().Scope(metrics.GlobalRatelimiterAggregator),\n\t\t\tusage: make(map[Limit]map[Identity]requests),\n\t\t\tclock: nil,\n\t\t}\n\t}\n\n\tunderlying := agg\n\ttick := clock.NewMockedTimeSource()\n\tunderlying.clock = tick\n\n\t// adjust time to get rid of sub-second output, it's just harder to read.\n\t// doesn't matter if this goes forward or backward.\n\ttick.Advance(tick.Now().Sub(tick.Now().Round(time.Second)))\n\n\treturn underlying, tick\n}\n\nfunc TestEmitsMetrics(t *testing.T) {\n\tt.Cleanup(func() {\n\t\tif t.Failed() {\n\t\t\tt.Log(\"This test is sensitive about bucket sizes, but they aren't actually important.\")\n\t\t\tt.Log(\"If bucket sizes have changed, just update the test to create enough data / use the new values, so each step is unique\")\n\t\t}\n\t})\n\n\tassertHistogramContents := func(name string, snap tally.HistogramSnapshot, expected map[float64]int64) {\n\t\tt.Helper()\n\t\tfor bucket, val := range snap.Values() { // {bucket_boundary: value}\n\t\t\tassert.Equal(t, expected[bucket], val, \"bucket %v has unexpected value for histogram name %v\", bucket, name)\n\t\t}\n\t}\n\tassertAllHistogramContents := func(snap tally.Snapshot, contents map[string]map[float64]int64) {\n\t\tt.Helper()\n\t\tfor _, data := range snap.Histograms() { // {\"full.name+tags\":{tags, values, etc}}\n\t\t\tname := strings.TrimPrefix(data.Name(), \"test.global_ratelimiter_\") // common prefix for this test\n\t\t\texp, ok := contents[name]\n\t\t\tif !ok {\n\t\t\t\t// keys remain between snapshots, so they're a bit of a pain.\n\t\t\t\t// values are zeroed by each snapshot though.\n\t\t\t\tfor bucket, val := range data.Values() {\n\t\t\t\t\tassert.Zerof(t, val, \"ignored key %v (trimmed: %v) in bucket %v has non-zero value, cannot be ignored\", data.Name(), name, bucket)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassertHistogramContents(name, data, exp)\n\t\t\t}\n\t\t}\n\t}\n\n\tagg, _ := newValid(t, defaultConfig(time.Second))\n\tts := tally.NewTestScope(\"test\", nil)\n\tagg.scope = metrics.NewClient(ts, metrics.History, metrics.HistogramMigration{}).Scope(metrics.GlobalRatelimiterAggregator)\n\n\th1, h2 := Identity(\"host 1\"), Identity(\"host 2\")\n\tkey := Limit(\"key\")\n\n\terr := agg.Update(UpdateParams{\n\t\tID:      h1,\n\t\tLoad:    map[Limit]Requests{key: {1, 1}},\n\t\tElapsed: time.Second,\n\t})\n\trequire.NoError(t, err)\n\tsnap := ts.Snapshot()\n\tassertAllHistogramContents(snap, map[string]map[float64]int64{\n\t\t\"initialized\":   {1: 1}, // one key was created\n\t\t\"reinitialized\": {0: 1},\n\t\t\"updated\":       {0: 1},\n\t\t\"decayed\":       {0: 1},\n\t})\n\n\terr = agg.Update(UpdateParams{\n\t\tID:      h2,\n\t\tLoad:    map[Limit]Requests{key: {1, 1}},\n\t\tElapsed: time.Second,\n\t})\n\trequire.NoError(t, err)\n\tsnap = ts.Snapshot()\n\tassertAllHistogramContents(snap, map[string]map[float64]int64{\n\t\t\"initialized\":   {1: 1}, // keys are disjoint, so another key was created\n\t\t\"reinitialized\": {0: 1},\n\t\t\"updated\":       {0: 1},\n\t\t\"decayed\":       {0: 1},\n\t})\n\n\terr = agg.Update(UpdateParams{\n\t\tID:      h1,\n\t\tLoad:    map[Limit]Requests{key: {1, 1}},\n\t\tElapsed: time.Second,\n\t})\n\trequire.NoError(t, err)\n\tsnap = ts.Snapshot()\n\tassertAllHistogramContents(snap, map[string]map[float64]int64{\n\t\t\"initialized\":   {0: 1},\n\t\t\"reinitialized\": {0: 1},\n\t\t\"updated\":       {1: 1}, // h1 was updated\n\t\t\"decayed\":       {0: 1},\n\t})\n\n\t_, err = agg.HostUsage(h1, []Limit{key})\n\trequire.NoError(t, err)\n\tsnap = ts.Snapshot()\n\tassertAllHistogramContents(snap, map[string]map[float64]int64{\n\t\t\"limits_queried\":      {1: 1}, // one limit exists\n\t\t\"host_limits_queried\": {2: 1}, // two hosts have data for that limit\n\t\t\"removed_limits\":      {0: 1}, // none removed\n\t\t\"removed_host_limits\": {0: 1}, // none removed\n\t})\n}\n\nfunc TestMissedUpdateHandling(t *testing.T) {\n\tagg, tick := newValid(t, configSnapshot{\n\t\tweight:     0.1,\n\t\trate:       time.Second,\n\t\tdecayAfter: 2 * time.Second,\n\t\tgcAfter:    10 * time.Second,\n\t})\n\n\th1, h2 := Identity(\"host 1\"), Identity(\"host 2\")\n\tkey := Limit(\"key\")\n\terr := agg.Update(UpdateParams{\n\t\tID:      h1,\n\t\tLoad:    map[Limit]Requests{key: {1, 1}},\n\t\tElapsed: time.Second,\n\t})\n\trequire.NoError(t, err)\n\terr = agg.Update(UpdateParams{\n\t\tID:      h2,\n\t\tLoad:    map[Limit]Requests{key: {1, 1}},\n\t\tElapsed: time.Second,\n\t})\n\trequire.NoError(t, err)\n\n\t// sanity-check the initial values\n\tusage, err := agg.HostUsage(h1, []Limit{key})\n\trequire.NoError(t, err)\n\tassert.Len(t, usage, 1)                        // 1 key known, should always be true\n\tassert.Equal(t, PerSecond(2), usage[key].Used) // only 2 accepted requests\n\n\t// move to 1 second later == expected update rate.\n\t// should still match the original values as it's not excessively delayed.\n\ttick.Advance(time.Second)\n\tusage, err = agg.HostUsage(h1, []Limit{key})\n\tassert.Len(t, usage, 1)\n\tassert.Equal(t, PerSecond(2), usage[key].Used) // still 2 allowed\n\n\t// advance to 1.5 seconds: over the expected rate but not beyond decay-after,\n\t// which should still not change anything.\n\ttick.Advance(time.Second / 2)\n\tusage, err = agg.HostUsage(h1, []Limit{key})\n\tassert.Len(t, usage, 1)\n\tassert.Equal(t, PerSecond(2), usage[key].Used) // still 2 allowed\n\n\t// advance another second, to 2.5s total.\n\t// exactly 2s and 3s are being avoided because nanosecond-equal times are highly unlikely,\n\t// and the behavior at that time doesn't actually matter, so it doesn't matter if it changes.\n\t//\n\t// 2.5s is beyond decayAfter (2s) so it should retroactively count missed updates,\n\t// immediately freeing up RPS as if it was tracking 0s all along, because we're assuming\n\t// it has been inactive for some reason.\n\t//\n\t// 2 full updates have been missed, so: 2 => 1.8 => 1.62\n\ttick.Advance(time.Second)\n\tusage, err = agg.HostUsage(h1, []Limit{key})\n\tassert.Len(t, usage, 1)\n\tassert.Equal(t, PerSecond(1.62), usage[key].Used) // reduced from 2\n\n\t// advance to 3.5 seconds, for 3 total missed updates:\n\t// 2 => 1.8 => 1.62 => 1.4580000000000002\n\ttick.Advance(time.Second)\n\tusage, err = agg.HostUsage(h1, []Limit{key})\n\tassert.Len(t, usage, 1)                                    // still tracking the key / not GC'd\n\tassert.InDelta(t, float64(usage[key].Used), 1.458, 0.0001) // further reduced\n}\n\nfunc TestGC(t *testing.T) {\n\th1, h2 := Identity(\"host 1\"), Identity(\"host 2\")\n\tkey := Limit(\"key\")\n\n\t// creates an aggregator and advances time 9 seconds, ensuring that data still exists.\n\t// advance 1 more second to trigger garbage collection.\n\t// this moves slightly beyond 9s to avoid testing the precise boundary time, as it's not relevant.\n\tsetup := func(t *testing.T) (*impl, clock.MockedTimeSource) {\n\t\tagg, tick := newValid(t, configSnapshot{\n\t\t\trate:    time.Second,\n\t\t\tgcAfter: 10 * time.Second,\n\n\t\t\t// irrelevant for these tests but must be non-zero:\n\t\t\tweight:     0.1,\n\t\t\tdecayAfter: 2 * time.Second,\n\t\t})\n\n\t\terr := agg.Update(UpdateParams{ID: h1, Load: map[Limit]Requests{key: {1, 1}}, Elapsed: time.Second})\n\t\trequire.NoError(t, err)\n\t\terr = agg.Update(UpdateParams{ID: h2, Load: map[Limit]Requests{key: {1, 1}}, Elapsed: time.Second})\n\t\trequire.NoError(t, err)\n\t\tusage, err := agg.HostUsage(h1, []Limit{key})\n\t\trequire.NoError(t, err)\n\t\t// sanity check that we have data\n\t\trequire.Len(t, usage, 1, \"sanity check: should have inserted limit's data\")\n\t\trequire.Equal(t, usage[key].Used, PerSecond(2), \"sanity check: should have inserted usage data\")\n\n\t\t// partially advance, sanity check.\n\t\ttick.Advance(9*time.Second + (time.Second / 10))\n\t\tusage, err = agg.HostUsage(h1, []Limit{key})\n\t\trequire.Len(t, usage, 1, \"sanity check: should have inserted limit's data after 9s\")\n\t\trequire.Equal(t, usage[key].Used, PerSecond(2*math.Pow(0.9, 9)), \"sanity check: should have inserted usage data, reduced after 9s\")\n\n\t\treturn agg, tick\n\t}\n\n\tt.Run(\"no cleanup before expiration\", func(t *testing.T) {\n\t\tagg, _ := setup(t)\n\t\tmet, err := agg.GC()\n\t\trequire.NoError(t, err)\n\t\trequire.Equal(t, Metrics{\n\t\t\tHostLimits:        2,\n\t\t\tLimits:            1,\n\t\t\tRemovedHostLimits: 0,\n\t\t\tRemovedLimits:     0,\n\t\t}, met)\n\t})\n\n\tt.Run(\"cleans up during read\", func(t *testing.T) {\n\t\tagg, tick := setup(t)\n\n\t\ttick.Advance(time.Second) // advance to 10th second\n\t\t// read it out, should detect out-of-date data and clean it up\n\t\tusage, err := agg.HostUsage(h1, []Limit{key})\n\t\trequire.NoError(t, err)\n\t\trequire.Len(t, usage, 0, \"should be no data for h1\")\n\n\t\t// internals should also be empty\n\t\trequire.Len(t, agg.usage, 0) // also frees memory\n\t})\n\tt.Run(\"retains recent data while cleaning\", func(t *testing.T) {\n\t\tagg, tick := setup(t)\n\n\t\t// refresh data for one host\n\t\terr := agg.Update(UpdateParams{ID: h1, Load: map[Limit]Requests{key: {1, 1}}, Elapsed: time.Second})\n\t\trequire.NoError(t, err)\n\t\ttick.Advance(time.Second) // advance to 10th second\n\n\t\t// read both hosts.  h1 should exist, h2 should not\n\t\tusage, err := agg.HostUsage(h1, []Limit{key})\n\t\trequire.NoError(t, err)\n\t\trequire.NotZero(t, usage[key].Weight, \"h1 was refreshed and should remain\")\n\t\trequire.NotZero(t, usage[key].Used, \"h1 was refreshed and usage data should remain\")\n\t\tusage, err = agg.HostUsage(h2, []Limit{key})\n\t\trequire.Zero(t, usage[key].Weight, \"h2 should have no weight at all\")\n\t\trequire.NotZero(t, usage[key].Used, \"h1 was refreshed and usage data should remain\")\n\n\t\t// internals should also be partly emptied\n\t\trequire.Len(t, agg.usage, 1, \"limit should remain\")\n\t})\n\tt.Run(\"cleans up by explicit gc\", func(t *testing.T) {\n\t\tagg, tick := setup(t)\n\t\ttick.Advance(time.Second)\n\t\tmet, err := agg.GC()\n\t\trequire.NoError(t, err)\n\t\trequire.Equal(t, Metrics{\n\t\t\tHostLimits: 0, // none remain\n\t\t\tLimits:     0, // none remain\n\n\t\t\tRemovedHostLimits: 2, // h1 and h2 for the single key, both removed\n\t\t\tRemovedLimits:     1, // single key removed\n\t\t}, met)\n\n\t\t// internals should also be empty\n\t\trequire.Len(t, agg.usage, 0)\n\t})\n}\n\nfunc TestMinorCoverage(t *testing.T) {\n\t// not overly useful tests, but coverage++\n\tt.Run(\"gc\", func(t *testing.T) {\n\t\t// invalid config\n\t\tagg, _ := newForTest(t, configSnapshot{}, false)\n\t\tm, err := agg.GC()\n\t\tassert.Zero(t, m)\n\t\tassert.ErrorContains(t, err, \"bad ratelimiter config\")\n\t})\n\tt.Run(\"update\", func(t *testing.T) {\n\t\t// invalid config\n\t\tagg, _ := newForTest(t, configSnapshot{}, false)\n\t\terr := agg.Update(UpdateParams{ID: \"ignored\", Load: nil, Elapsed: time.Second})\n\t\tassert.ErrorContains(t, err, \"bad ratelimiter config\")\n\t})\n\tt.Run(\"get-weights\", func(t *testing.T) {\n\t\t// invalid config\n\t\tagg, _ := newForTest(t, configSnapshot{}, false)\n\t\tusage, err := agg.HostUsage(\"ignored\", nil)\n\t\tassert.Zero(t, usage)\n\t\tassert.ErrorContains(t, err, \"bad ratelimiter config\")\n\t})\n\n\t// a bit more useful\n\tt.Run(\"config validation\", func(t *testing.T) {\n\t\terr := configSnapshot{\n\t\t\tweight:     -1,\n\t\t\trate:       time.Duration(0),\n\t\t\tdecayAfter: -time.Second,\n\t\t\tgcAfter:    -time.Second,\n\n\t\t\tnow: time.Time{}, // ignored\n\t\t}.validate()\n\t\t// should have the shared error string\n\t\tassert.ErrorContains(t, err, \"bad ratelimiter config\")\n\t\t// should have each sub-error\n\t\tassert.ErrorContains(t, err, \"weight cannot be negative\")\n\t\tassert.ErrorContains(t, err, \"rate must be positive\")\n\t\tassert.ErrorContains(t, err, \"decay-after cannot be negative\")\n\t\tassert.ErrorContains(t, err, \"gc-after cannot be negative\")\n\t\tassert.Len(t, multierr.Errors(err), 5, \"should have 5 errors, 4 details and one general\")\n\t})\n\tt.Run(\"fast scalar path\", func(t *testing.T) {\n\t\tcfg := configSnapshot{\n\t\t\tgcAfter: time.Second,\n\t\t}\n\t\tassert.Zero(t, cfg.missedUpdateScalar(2*time.Second), \"should multiply old data by exactly zero when beyond gc age\")\n\t})\n\n\t// weird config coverage\n\tt.Run(\"irrational decayAfter\", func(t *testing.T) {\n\t\t// specifically: exercises the \"less than one missed update\" branch\n\t\tnow := time.Now().Round(time.Second)\n\t\tcfg := configSnapshot{\n\t\t\tnow:        now,\n\t\t\trate:       time.Second,\n\t\t\tdecayAfter: time.Second / 4, // irrational but allowed: decay faster than expected update rate\n\n\t\t\t// effectively ignored\n\t\t\tweight:  0.1,\n\t\t\tgcAfter: time.Second * 2,\n\t\t}\n\t\tscale := cfg.missedUpdateScalar(time.Second / 2) // between update and decay periods\n\t\tassert.Equal(t, PerSecond(1), scale, \"should not have decayed yet\")\n\t})\n}\n\nfunc TestRapidlyCoalesces(t *testing.T) {\n\t// This test ensures that, regardless of starting weights, the algorithm\n\t// \"rapidly\" achieves near-actual weight distribution after a small number of rounds.\n\t//\n\t// Otherwise, the exact numbers here don't really matter, it's just handy to show the\n\t// behavior in semi-extreme scenarios.  Logs show a quick adjustment which is what we want.\n\t// If you're making changes, check with like 10k rounds to make sure it's stable.\n\t//\n\t// Time is also not advanced because it doesn't actually need to be advanced.\n\t// An update is an update, and the caller's elapsed time is assumed to be correct.\n\tagg, _ := newValid(t, configSnapshot{\n\t\t// Using 0.5 weight because that's what we expect to use IRL, and this test is\n\t\t// ensuring that weight is good enough for the behavior we want.\n\t\t// Weight-math-correctness is ensured by other tests.\n\t\tweight: 0.5,\n\n\t\t// irrelevant / ignored\n\t\trate:       time.Second,\n\t\tdecayAfter: 2 * time.Second,\n\t\tgcAfter:    10 * time.Second,\n\t})\n\tsnapshot := func() configSnapshot {\n\t\tsnap, err := agg.snapshot()\n\t\trequire.NoError(t, err)\n\t\treturn snap\n\t}\n\n\tkey := Limit(\"start workflow\")\n\th1, h2, h3 := Identity(\"one\"), Identity(\"two\"), Identity(\"three\")\n\n\tweights, used, met := agg.getWeightsLocked(key, snapshot())\n\tassert.Zero(t, weights, \"should have no weights\")\n\tassert.Zero(t, used, \"should have no used RPS\")\n\tassert.Zero(t, met, \"should have processed no data while calculating\")\n\n\tpush := func(host Identity, accept, reject int) {\n\t\terr := agg.Update(UpdateParams{\n\t\t\tID: host,\n\t\t\tLoad: map[Limit]Requests{\n\t\t\t\tkey: {\n\t\t\t\t\tAccepted: accept,\n\t\t\t\t\tRejected: reject,\n\t\t\t\t},\n\t\t\t},\n\t\t\tElapsed: time.Second, // 1s just to make rps in == rps out\n\t\t})\n\t\trequire.NoError(t, err)\n\t}\n\n\t// init with anything <~1000, too large and even a small fraction of the original value can be too big.\n\tpush(h1, rand.Intn(1000), rand.Intn(1000))\n\tpush(h2, rand.Intn(1000), rand.Intn(1000))\n\tpush(h3, rand.Intn(1000), rand.Intn(1000))\n\n\t// now update multiple times and make sure it gets to 90% within 4 steps == 12s (normally).\n\t//\n\t// 4 steps with 0.5 weight should mean only 0.5^4 => 6.25% of the original influence remains,\n\t// which feels pretty reasonable: after ~10 seconds (3s updates), the oldest data only has ~10% weight.\n\tconst target = 10 + 200 + 999\n\tfor i := 0; i < 4; i++ {\n\t\tweights, used, met = agg.getWeightsLocked(key, snapshot())\n\t\tt.Log(\"used:\", used, \"of actual:\", target)\n\t\tt.Log(\"weights so far:\", weights)\n\t\tt.Log(\"calculation metrics:\", met)\n\t\tpush(h1, 10, 10)\n\t\tpush(h2, 200, 200)\n\t\tpush(h3, 999, 999)\n\t}\n\tweights, used, met = agg.getWeightsLocked(key, snapshot())\n\tt.Log(\"used:\", used, \"of actual:\", target)\n\tt.Log(\"weights so far:\", weights)\n\tt.Log(\"calculation metrics:\", met)\n\n\t// aggregated allowed-request values should be less than 10% off\n\tassert.InDeltaf(t, target, float64(used), target*0.1, \"should have allowed >90%% of target rps by the 5th round\") // actually ~94%\n\t// also check weights, they should be within 10%\n\tassert.InDeltaMapValues(t, map[Identity]float64{\n\t\th1: 10 / 1209.0,  // 0.07698229407\n\t\th2: 200 / 1209.0, // 0.1539645881\n\t\th3: 999 / 1209.0, // 0.7690531178\n\t}, floaty(weights), 0.1, \"should be close to true load balance\")\n}\n\n// converts for testify/assert as it requires same types, not just same underlying type\nfunc floaty[K comparable, V numeric](m map[K]V) map[K]float64 {\n\tout := make(map[K]float64, len(m))\n\tfor k, v := range m {\n\t\tout[k] = float64(v)\n\t}\n\treturn out\n}\n\nfunc TestConcurrent(t *testing.T) {\n\t// essentially a fuzz-test for race purposes.\n\t// values aren't checked, but it shouldn't panic / shouldn't race / etc.\n\t// low timeout + real time clock to also have wall-clock changes race with logic.\n\t//\n\t// this test frequently reaches 100% coverage all on its own (currently),\n\t// though it's not guaranteed and that is not the intent.\n\t// other tests should cover sufficiently even if this test is skipped:\n\t// t.Skip(\"skipped to check coverage\")\n\n\tconst (\n\t\tupdateRate      = time.Millisecond // fairly arbitrary, max gap between updates\n\t\ttargetDuration  = 100 * updateRate // also minimum number of updates\n\t\tnumHosts        = 10\n\t\tnumUpdaters     = 10           // fairly arbitrary\n\t\tupdatesPerBatch = numHosts / 3 // intentionally below len(hosts) to allow some to gc normally\n\t)\n\n\tagg, _ := newValid(t, configSnapshot{\n\t\trate:       updateRate,\n\t\tdecayAfter: 2 * updateRate,\n\t\tgcAfter:    3 * updateRate, // relatively low to trigger implicit gc, check coverage if changing the values\n\n\t\tweight: 0.1, // irrelevant but must be non-zero\n\t})\n\tagg.clock = clock.NewRealTimeSource()\n\n\tvar hosts []Identity\n\tfor i := 0; i < numHosts; i++ {\n\t\thosts = append(hosts, Identity(fmt.Sprintf(\"host %d\", i)))\n\t}\n\tvar keys []Limit\n\tfor i := 'a'; i <= 'z'; i++ {\n\t\tkeys = append(keys, Limit(i))\n\t}\n\n\tdo := make(chan struct{}, numUpdaters)\n\tvar wg sync.WaitGroup\n\n\t// run some goroutines to update/read in batches\n\tfor i := 0; i < numUpdaters; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\n\t\t\tfor {\n\t\t\t\t_, ok := <-do\n\t\t\t\tif !ok {\n\t\t\t\t\t// chan's closed, stop\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tnum := rand.Intn(len(keys))\n\t\t\t\tupdates := make(map[Limit]Requests, num)\n\t\t\t\tfor i := 0; i < num; i++ {\n\t\t\t\t\tupdates[keys[rand.Intn(len(keys))]] = Requests{\n\t\t\t\t\t\tAccepted: rand.Intn(100),\n\t\t\t\t\t\tRejected: rand.Intn(100),\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\thost := hosts[rand.Intn(len(hosts))]\n\n\t\t\t\t// randomly update or read\n\t\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t\terr := agg.Update(UpdateParams{ID: host, Load: updates, Elapsed: updateRate})\n\t\t\t\t\trequire.NoError(t, err)\n\t\t\t\t} else {\n\t\t\t\t\t_, err := agg.HostUsage(host, maps.Keys(updates))\n\t\t\t\t\trequire.NoError(t, err)\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\n\t// run a \"trigger some work occasionally\" goroutine\n\twg.Add(1)\n\tstart := time.Now()\n\tgo func() {\n\t\tdefer wg.Done()\n\n\t\tfor {\n\t\t\tif time.Since(start) > targetDuration {\n\t\t\t\tclose(do)\n\t\t\t\treturn\n\t\t\t}\n\t\t\t// sleep a random portion of the update rate\n\t\t\ttime.Sleep(time.Duration(rand.Intn(int(updateRate))))\n\t\t\t// allow a random number of updates.\n\t\t\t// should be non-blocking to further encourage racing, when possible\n\t\t\tfor i := rand.Intn(updatesPerBatch); i > 0; i-- {\n\t\t\t\tdo <- struct{}{}\n\t\t\t}\n\t\t}\n\t}()\n\n\tassert.True(t,\n\t\tcommon.AwaitWaitGroup(&wg, 5*targetDuration),\n\t\t\"blocked test? still waiting after %v\", 5*targetDuration)\n\t// non-racy even if waiting failed\n\tm, err := agg.GC()\n\trequire.NoError(t, err)\n\tt.Logf(\"%#v\", m) // should (usually) not be \"full\" + should (usually) remove some data\n}\n\nfunc TestSimulate(t *testing.T) {\n\t// Semi-fuzzy simulated sequence with fully computed values, exercising most behaviors.\n\t//\n\t// Everything about this test is sensitive to changes in behavior,\n\t// so if that occurs just update the values after ensuring they're reasonable.\n\t// Exact matches after changes are not at all important.\n\n\tupdateRate := 3 * time.Second // both expected and duration fed to update\n\tagg, tick := newValid(t, configSnapshot{\n\t\t// now:    , ignored\n\t\tweight:     0.75, // fairly fast adjustment, and semi-human-friendly math\n\t\trate:       updateRate,\n\t\tdecayAfter: 2 * updateRate,\n\t\tgcAfter:    10 * updateRate,\n\t})\n\n\t// keeping var == string simplifies copy/paste as we cannot log the var name\n\tstart, query := Limit(\"start\"), Limit(\"query\")\n\tall := []Limit{start, query}\n\th1, h2, h3 := Identity(\"one\"), Identity(\"two\"), Identity(\"three\")\n\n\tsnap, err := agg.snapshot()\n\trequire.NoError(t, err)\n\tweights, used, met := agg.getWeightsLocked(start, snap)\n\tassert.Zero(t, weights, \"should have no weights\")\n\tassert.Zero(t, used, \"should have no used RPS\")\n\tassert.Zero(t, met, \"should have processed no data while calculating\")\n\n\t// just simplifies arg-construction\n\tpush := func(host Identity, key Limit, accept, reject int) {\n\t\terr := agg.Update(UpdateParams{\n\t\t\tID: host,\n\t\t\tLoad: map[Limit]Requests{\n\t\t\t\tkey: {\n\t\t\t\t\tAccepted: accept,\n\t\t\t\t\tRejected: reject,\n\t\t\t\t},\n\t\t\t},\n\t\t\tElapsed: time.Second,\n\t\t})\n\t\trequire.NoError(t, err)\n\t}\n\n\t// these tests intentionally share data and run sequentially,\n\t// the sub-testing is mostly to help group semantics.\n\n\t// init with Some Numbers.  updates to different keys with the same timestamp\n\t// can be grouped or separate, it doesn't matter - only same-key changes behavior.\n\tpush(h1, query, 5, 5)\n\tpush(h1, start, 5, 5)\n\ttick.Advance(time.Second)\n\tpush(h2, query, 1, 1)\n\tpush(h2, start, 1, 1)\n\ttick.Advance(time.Second)\n\tpush(h3, query, 0, 0)\n\tpush(h3, start, 1, 1)\n\ttick.Advance(time.Second)\n\n\t// changing new-data-weight does not affect this test because\n\t// zero -> nonzero keeps 100% to jump-start the initial state,\n\t// rather than gradually growing from zero (which would be biased\n\t// towards lower values during movement - not bad, but not intended).\n\tt.Run(\"initial weights at 3s\", func(t *testing.T) {\n\t\t// 3s elapsed, all fresh.  h1 is \"old\" but within decayAfter so it's assumed still valid\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\n\t\t// h1 has most of the weight.\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.83, Used: 6},\n\t\t\tstart: {Weight: 0.71, Used: 7}, // h3 uses more start than query, so h1 has less weight for start\n\t\t}, usage)\n\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.17, Used: 6}, // only h1 and h2 called this, so (0.83 + 0.17)==1\n\t\t\tstart: {Weight: 0.14, Used: 7}, // h3 also had a small use, so less than query\n\t\t}, usage)\n\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0},    // no query at all\n\t\t\tstart: {Weight: 0.14}, // same num of starts as h2\n\t\t}, usage)\n\t})\n\tif t.Failed() {\n\t\treturn // later tests likely invalid, verify each step before moving to the next\n\t}\n\n\t// advance to second h2 update, skip the others.\n\t// no data has expired yet, but h1 is now at 5s old (closing in on decayAfter)\n\ttick.Advance(2 * time.Second)\n\tpush(h2, query, 1, 2)\n\tpush(h2, start, 3, 3)\n\n\tt.Run(\"increased h2 weight at 5s\", func(t *testing.T) {\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\t\t// h1's weight reduces due to increased h2 usage\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.78, Used: 6},   // accepted query requests did not change (1 before, 1 after})\n\t\t\tstart: {Weight: 0.59, Used: 8.5}, // but start calls from h2 increased by 2 -> 0.75 weight -> +1.5 total\n\t\t}, usage)\n\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.22, Used: 6},   // h3 has no query, so h1+h2=1.0 but the balance has shifted a bit towards h2\n\t\t\tstart: {Weight: 0.29, Used: 8.5}, // increased over last round due to more calls\n\t\t\t// plus a sanity check: same rps as h1 saw\n\t\t}, usage)\n\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\trequire.NoError(t, err)\n\t\t// h3 is almost idle but 0.1 weight changes slowly\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0, Used: 6},      // never sent any query requests\n\t\t\tstart: {Weight: 0.12, Used: 8.5}, // decreased since last round, as relative usage is lower\n\t\t\t// plus a sanity check: same rps as h1 saw\n\t\t}, usage)\n\t})\n\tif t.Failed() {\n\t\treturn // later tests likely invalid, verify each step before moving to the next\n\t}\n\n\t// advance to 10s.\n\t// this puts h1's original data beyond decayAfter, so it will act as if it\n\t// had received 0-valued updates, to reduce its weight.\n\ttick.Advance(5 * time.Second)\n\tpush(h2, query, 5, 5)\n\tpush(h2, start, 5, 5)\n\tpush(h3, query, 5, 5)\n\tpush(h3, start, 5, 5)\n\n\tt.Run(\"h1 decayed at 10s\", func(t *testing.T) {\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.01, Used: 7.83}, // 10 accepted in last round, but only 0.75 weight, 6.0 -> 7.8 due to older data\n\t\t\tstart: {Weight: 0.01, Used: 8.22}, // similar\n\t\t}, usage)\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\trequire.NoError(t, err)\n\t\t// h2 has slightly over half weight due to greater historical use\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.45},\n\t\t\tstart: {Weight: 0.53}, // still more starts than queries\n\t\t}, usage)\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\trequire.NoError(t, err)\n\t\t// h2 slightly less, due to low historical + some used by h1\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\t// this looks odd, but it's higher due to lower historical *total* queries,\n\t\t\t// leading to a somewhat counter-intuitive:\n\t\t\t// - smaller numerator (lower calls by this host)\n\t\t\t// - smaller denominator (lower total calls)\n\t\t\t// - higher final value (smaller denominator has greater influence)\n\t\t\tquery: {Weight: 0.54},\n\t\t\tstart: {Weight: 0.46},\n\t\t}, usage)\n\t})\n\tif t.Failed() {\n\t\treturn // later tests likely invalid, verify each step before moving to the next\n\t}\n\n\t// update everything, should flatten compared to previous round\n\ttick.Advance(1 * time.Second)\n\tpush(h1, query, 5, 5)\n\tpush(h1, start, 5, 5)\n\tpush(h2, query, 5, 5)\n\tpush(h2, start, 5, 5)\n\tpush(h3, query, 5, 5)\n\tpush(h3, start, 5, 5)\n\n\tt.Run(\"all equal at 11s is relatively flatter than before\", func(t *testing.T) {\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\t\t// historically lower weight = current lower weight, but a big jump from 0.01 before.\n\t\t//\n\t\t// this is likely a faster shift than we want in practice, as it'll make allowed-request\n\t\t// behavior quite jumpy, which is why the initial weight is likely to be around 0.5.\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.28, Used: 13.21}, // used is adjusting towards 15\n\t\t\tstart: {Weight: 0.28, Used: 13.52},\n\t\t}, usage)\n\t\t// and weights are flattening towards 0.33\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.36},\n\t\t\tstart: {Weight: 0.36},\n\t\t}, usage)\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.37},\n\t\t\tstart: {Weight: 0.35},\n\t\t}, usage)\n\t})\n\tif t.Failed() {\n\t\treturn // later tests likely invalid, verify each step before moving to the next\n\t}\n\n\t// do that again, should flatten further\n\ttick.Advance(1 * time.Second)\n\tpush(h1, query, 5, 5)\n\tpush(h1, start, 5, 5)\n\tpush(h2, query, 5, 5)\n\tpush(h2, start, 5, 5)\n\tpush(h3, query, 5, 5)\n\tpush(h3, start, 5, 5)\n\n\tt.Run(\"all equal at 12s is even flatter\", func(t *testing.T) {\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\t\t// still slightly below the others but it hardly matters now\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.32, Used: 14.55}, // Used RPS now quite close to 15\n\t\t\tstart: {Weight: 0.32, Used: 14.58},\n\t\t}, usage)\n\t\t// weights flattening towards 0.33\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.34},\n\t\t\tstart: {Weight: 0.34},\n\t\t}, usage)\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.34},\n\t\t\tstart: {Weight: 0.34},\n\t\t}, usage)\n\t})\n\tif t.Failed() {\n\t\treturn // later tests likely invalid, verify each step before moving to the next\n\t}\n\n\t// make everything fully expired so the \"from expired data\" branch is exercised too.\n\t// specifically this just needs to be beyond gcAfter for all data.\n\ttick.Advance(time.Hour)\n\t// push anything (need all keys to get same HostWeight map keys for `expectSimilarUsage`\n\tpush(h1, query, 5, 5)\n\tpush(h1, start, 5, 5)\n\tpush(h2, query, 5, 5)\n\tpush(h2, start, 5, 5)\n\tpush(h3, query, 5, 5)\n\tpush(h3, start, 5, 5)\n\n\tt.Run(\"updating expired data acts like deleted data\", func(t *testing.T) {\n\t\t// should leap to exact values, not weighted-from-zero.\n\t\tusage, err := agg.HostUsage(h1, all)\n\t\trequire.NoError(t, err)\n\t\t// still slightly below the others but it hardly matters now\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\t// note: used RPS is exactly 15. if weighting from or very near zero, rather than tossing\n\t\t\t// old data entirely, this would be: (0*0.25 + 15*0.75) == 11.25\n\t\t\tquery: {Weight: 0.333, Used: 15},\n\t\t\tstart: {Weight: 0.333, Used: 15},\n\t\t}, usage)\n\t\t// weights flattening towards 0.33\n\t\tusage, err = agg.HostUsage(h2, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.333},\n\t\t\tstart: {Weight: 0.333},\n\t\t}, usage)\n\t\tusage, err = agg.HostUsage(h3, all)\n\t\trequire.NoError(t, err)\n\t\texpectSimilarUsage(t, map[Limit]HostUsage{\n\t\t\tquery: {Weight: 0.333},\n\t\t\tstart: {Weight: 0.333},\n\t\t}, usage)\n\t})\n}\n\nfunc expectSimilarUsage(\n\tt *testing.T,\n\texpected map[Limit]HostUsage,\n\tactual map[Limit]HostUsage) {\n\tt.Helper() // report caller's line as the logger, not this expect-er\n\n\t// check that host weights are similar\n\teWeight := make(map[Limit]float64, len(expected))\n\taWeight := make(map[Limit]float64, len(expected))\n\tfor k, v := range expected {\n\t\teWeight[k] = float64(v.Weight)\n\t}\n\tfor k, v := range actual {\n\t\taWeight[k] = float64(v.Weight)\n\t}\n\n\t// check host weights are similar\n\tif !assert.InDeltaMapValues(t, eWeight, aWeight, 0.01) {\n\t\tt.Logf(\"weight(s) differ by more than 0.01:\\n\"+\n\t\t\t\"expected: %v\\n\"+\n\t\t\t\"actual: %v\", eWeight, aWeight)\n\t\tt.Logf(\"verify the weights by hand, and if it's correct just update them.  \" +\n\t\t\t\"values do not need to stay precise / there is no perfect value, \" +\n\t\t\t\"they just need to behave the way we want and should not change unexpectedly.\")\n\t}\n\n\t// check that the used RPS is identical too, if non-zero\n\teRps := make(map[Limit]float64, len(expected))\n\taRps := make(map[Limit]float64, len(expected))\n\tfor k, v := range expected {\n\t\tif v.Used != 0 {\n\t\t\teRps[k] = float64(v.Used)\n\t\t}\n\t}\n\tfor k, v := range actual {\n\t\tif v, ok := expected[k]; ok && v.Used == 0 {\n\t\t\tcontinue // ignore zeros in the expected map, as long as the key exists\n\t\t}\n\t\taRps[k] = float64(v.Used)\n\t}\n\tif !assert.InDeltaMapValues(t, eWeight, aWeight, 0.01) {\n\t\tt.Logf(\"RPS(s) differ by more than 0.1:\\n\"+\n\t\t\t\"expected: %v\\n\"+\n\t\t\t\"actual: %v\", eRps, aRps)\n\t\tt.Logf(\"verify the RPS values by hand, and if it's correct just update them.  \" +\n\t\t\t\"zero-RPS ignores the value, and can be used where we don't particularly care.\")\n\t}\n}\n\n// fairly fuzzy but somewhat representative of expected use:\n// benchmark \"update a bunch of keys and get my load\" requests, and accumulate data across all iterations.\n// this is roughly what a server update operation will look like.\n// there are intentionally overlaps in both hosts and keys to get multiple hosts per key and updates to existing data.\n//\n// while this obviously has major caveats and fake costs, the benchmark regularly runs several thousand iterations\n// before it settles, filling a moderate amount of data along the way (e.g. consistently all 10k keys, and >200k records in memory).\n//\n// general results imply:\n//   - host-weight reading takes a few times longer than updating, but on my laptop the\n//     bench takes about 3-4ms per loop for the current 100/1000/10000 values, without mutexes.\n//   - this is relatively CPU costly, but even if our largest cluster checks in every 3s to a *single*\n//     host, this should work out with some room to spare... and it will be far more spread\n//     out in practice due to the number of history hosts involved.\n//   - an attempt to cache weight calculations per key until invalidated saved about 1/2 the cpu\n//     with a perfect cache.\n//   - ^ the cache seems not worth the complexity, but it'd be fairly easy to add if we change our minds\n//\n// since there is only a coarse mutex in the implementation itself, the whole aggregator forces itself\n// to be processed serially.  if this turns out to be too costly, the easy fix is probably\n// to shard the keys to e.g. 8 separate aggregators that can be processed concurrently.\n//\n// mutexes could be added to each Limit, but that could end up costing more in memory delays\n// than the fine-grained locking gains us in concurrency flexibility.  benchmark first!\nfunc BenchmarkNormalUse(b *testing.B) {\n\tupdateRate := 3 * time.Second\n\tagg, _ := newValid(b, defaultConfig(updateRate))\n\t// intentionally using real clock source, time-gathering cost is relevant to benchmark\n\tagg.clock = clock.NewRealTimeSource()\n\n\t// aiming for vaguely realistic-ish values\n\tconst (\n\t\thosts       = 100   // num of hosts in pool, if rounds > hosts there will be duplicates (intentional)\n\t\tkeysPerHost = 1000  // more domains than hosts, multiple limits per domain.\n\t\tglobalKeys  = 10000 // each host gets ~10% of requests (not consistent across calls tho)\n\t\trequests    = 1000  // accept/reject counts\n\t)\n\n\t// rand and string-formatting is a non-trivial amount of load,\n\t// so this is prepared up-front so it can be excluded.\n\ttype round struct {\n\t\thost    Identity\n\t\tload    map[Limit]Requests\n\t\tkeys    []Limit\n\t\telapsed time.Duration\n\t}\n\n\trounds := make([]round, 0, b.N)\n\tfor i := 0; i < b.N; i++ {\n\t\tkeys := rand.Intn(keysPerHost)\n\t\treqs := make(map[Limit]Requests, keys)\n\t\tfor len(reqs) < keys {\n\t\t\tkey := Limit(fmt.Sprintf(\"key %d\", rand.Intn(globalKeys)))\n\t\t\tif _, ok := reqs[key]; ok {\n\t\t\t\t// duplicate, just try again.\n\t\t\t\t// dups wouldn't really invalidate the benchmark,\n\t\t\t\t// but it's easy to avoid and should make more stable results.\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\treqs[key] = Requests{\n\t\t\t\t// values don't really matter but there's no need for them all to use the same value\n\t\t\t\tAccepted: rand.Intn(requests),\n\t\t\t\tRejected: rand.Intn(requests),\n\t\t\t}\n\t\t}\n\t\trounds = append(rounds, round{\n\t\t\thost:    Identity(fmt.Sprintf(\"host %d\", rand.Intn(hosts))),\n\t\t\tload:    reqs,\n\t\t\tkeys:    maps.Keys(reqs),\n\t\t\telapsed: time.Duration(rand.Int63n(updateRate.Nanoseconds())),\n\t\t})\n\t}\n\n\tb.ResetTimer()\n\tb.ReportAllocs()\n\n\tsawnonzero := 0\n\tfor _, r := range rounds { // == b.N times\n\t\terr := agg.Update(UpdateParams{ID: r.host, Load: r.load, Elapsed: r.elapsed})\n\t\trequire.NoError(b, err)\n\t\tvar unused map[Limit]PerSecond // ensure non-error second return for test safety\n\t\tusage, err := agg.HostUsage(r.host, r.keys)\n\t\trequire.NoError(b, err)\n\t\t_ = unused // ignore unused rps\n\t\tif len(usage) > 0 {\n\t\t\t// \twrote data and later read it out, benchmark is likely functional\n\t\t\tsawnonzero++\n\t\t}\n\t}\n\tb.StopTimer() // not benchmarking gc, just using it to show sanity-check metrics\n\n\tb.Log(\"N was:\", b.N)\n\tm, err := agg.GC()\n\trequire.NoError(b, err)\n\tb.Log(\"gc metrics:\", m) // shows how many keys / hosts / etc we stored for manual validation\n\n\t// sanity check, as \"all zero\" should mean something like \"always looking at nonexistent keys\" / bad benchmark code.\n\tb.Log(\"nonzero results:\", sawnonzero)\n\tassert.True(b, b.N < 500 || sawnonzero > 0, \"no non-zero result found on a large enough benchmark, likely not benchmarking anything useful\")\n}\n\nfunc TestHashmapIterationIsRandom(t *testing.T) {\n\t// is partial hashmap iteration reliably random each time so we can statistically ensure coverage,\n\t// or is it relatively fixed per map?\n\t// language spec is somewhat vague on the details here, so let's check.\n\t//\n\t// verdict: yes!  looks random each time.\n\t// so we can abuse it for an amortized / interruptible cleanup tool.\n\n\tconst keys = 100\n\tm := map[int]struct{}{}\n\tfor i := 0; i < keys; i++ {\n\t\tm[i] = struct{}{}\n\t}\n\n\tallObserved := make(map[int]struct{}, len(m))\n\tvar orderObserved [][]int\n\tsinglePass := func() {\n\t\tfor i := 0; i < keys; i++ {\n\t\t\tobserved := make([]int, 0, keys/10)\n\t\t\tfor k := range m {\n\t\t\t\tallObserved[k] = struct{}{}\n\t\t\t\tobserved = append(observed, k)\n\t\t\t\tif len(observed) == cap(observed) {\n\t\t\t\t\tbreak // interrupt part way through\n\t\t\t\t}\n\t\t\t}\n\t\t\torderObserved = append(orderObserved, observed)\n\t\t}\n\t}\n\n\t// keep trying up to 10x to make it sufficiently-unlikely that randomness will fail.\n\t// with only a single round it fails like 5% of the time, which is reasonable behavior\n\t// but too much noise to allow to fail the test suite.\n\tfor i := 0; i < 10 && len(allObserved) < keys; i++ {\n\t\tif i > 0 {\n\t\t\tt.Logf(\"insufficient keys observed (%d), trying again in round %d\", len(allObserved), i+1)\n\t\t}\n\t\tsinglePass()\n\t}\n\n\t// complain if it still hasn't observed all keys\n\tif !assert.Len(t, allObserved, keys) {\n\t\t// super noisy when successful, so only log when failing\n\t\tfor idx, pass := range orderObserved {\n\t\t\tt.Log(\"Pass\", idx)\n\t\t\tfor _, keys := range pass {\n\t\t\t\tt.Log(\"\\t\", keys)\n\t\t\t}\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMissedUpdate/264c784f7bafbf5f",
    "content": "go test fuzz v1\nint(0)\nint(60)\nint(0)\nint(0)\nfloat64(0)\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/00649674e28cdc32",
    "content": "go test fuzz v1\n[]byte(\"00\\x0000\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/0bb6b094a7f63d70",
    "content": "go test fuzz v1\n[]byte(\"00000\\x13\\x89\\x8c>\\bەlw)4\\xca|A##X\\x1ca\\xf6O-\\xf3\\x9a:\\xfeS+\\xe5\\xafW\\x1cn\\xcbŴ\\xe3<\\xaf\\xc6\\xec\\xb4$R\\xb3\\xd6Q\\\"\\xb3\\x82\\x10g\\xed$\\xac\\xdd\\xc9\\xfa\\x88\\xb7\\xfe\\xf3z&vY\\n\\x16\\xfdT\\x16\\xcf\\x12T\\xbdv\\x12\\xf9\\xb5~\\xc3Y4@\\x84\\x16\\xfc9$\\xbd\\xf1z\\xff\\x7f\\xff\\xff\\x04\\x04\\xbe\\xdfP\\xee\\xde[\\xbb)\\x86\\v\\xcd_\\xd7\\xca%-zO{\\xccl\\xbe`\\x1d\\xd2\\xe8\\x99o\\xbbO\\f\\x10\\r\\x88\\xaa\\x1eG\\x17\\x19\\x83}\\xcfUR\\xa4\\x18<\\xa4\\xd6zhk\\x8bV(\\x98h\\xee\\xa2\\xe4ns.\\xb7}-u6\\x83\\xfa\\x0f\\a\\x17\\xd8ؑ\\xd2 9ȗ(x\\xa7\\xfe-C\\xe9{[\\x01\\x13\\xbcE\\x845C!)\\x19\\xf6\\x1d۾\\t\\xd92\\xfe<\\xf3\\xcaD2:]\\x03\\xfc\\x16Q\\x15\\xc0\\xd8(\\x1f\\x9bBg\\xc1\\xe8ev\\xe4\\xe7<v\\x10\\xe3Y7\\xee\\xf9\\xb5\\x80T\\x85\\x03\\x00e\\x8f\\x83\\xcfro*\\x0f\\x9eM\\x9a\\xde\\xdbB\\x0e+\\xb16\\x8f\\xed\\x8bm\\xfdUt\\x98\\x872\\xb1\\xf8r\\xeaW\\b\\x19\\xf2n\\x1b\\xa3\\x19\\xb6\\xf6\\xafu\\a\\xed\\x13\\xb2\\x8b\\xe5\\xe9rFD\\xae?c+\\xb8u\\xf4\\xbaЫ\\xaa\\xe9>\\xe7\\xf4\\x8c/\\x18\\x1dڱ\\xfc\\xb1\\xfc,S<\\\"\\x16\\xfcN\\x91\\xc2g\\xc63\\x81\\x87\\xf8\\x84\\x02\\x0e\\xeb\\xedբ\\x99ˎ\\x1a*\\x9d\\xff<)\\xdb \\xf0i\\xa9L\\x9c\\xf5\\x9f\\xaf犨\\xa1\\xbc\\xc4\\xd4\\xef\\x8e>Z\\t\\x83?\\xa0i\\xdeFB\\x83\\x0e\\x05\\x05\\xb2\\xe0%~\\xb0x\\x96\\xe2(FM:\\xc8J\\x98\\xd1\\xfd\\x1e\\x10\\xef\\x1b\\x14\\x92\\x18\\xab\\x89\\tU\\xcc\\x04\\xe3\\xe3C\\x010z\\x99TK\\xf6N\\b\\u0601\\xa5}t\\x1dyQvem\\x9c\\x19]kM\\xa8=\\xb9:\\xfd\\x8e\\x11\\xe8\\x81D\\x1a%M)\\xfa\\xa6~k@˽\\x94 U\\x8d\\xfbF\\x17O\\x905\\xcc\\xf2\\xffMi\\xa3\\xdeeV\\x9b\\xbb\\xec6\\xa9\\x939E\\xc3\\xfa\\f\\xbd\\x92\\b#,1Q\\x89\\xe8\\x15u9\\xb3{\\xed\\x10z\\xc9!\\xdfs\\xe7s\\x18O0\\x1a\\xe9\\x8d\\xf3\\x95:\\\"X\\x86ڰ\\x19\\xcb-\\xfc\\x0f\\xa1EK`\\x1fRew;/\\x80\\x7fP\\xe1G&R\\xc76\\xdc,\\x1a?\\xe8\\xcf\\xc9\\xeaR\\xf5l\\x86\\xa6\\xd3\\tL\\x14>\\x87.B\\x1a\\xbd?Tڧ8\\xa6S\\xf7\\xfb\\x84\\xf6*\\xe3\\xb8\\xf7\\x13\\\"c85\\x93Б$\\xc4{\\xb9\\x85J\\xc0\\xa9\\xaam\\xc5\\xfc\\b\\xe6K\\x90v\\x15'J\\x83\\x0fK!k\\xcaᚲJgw\\xee\\xffGYt\\b\\x94\\xbd\\x9f\\xa9vG\\xf0\\x98\\xe37\\x1c\\xde\\x1a\\xeb)n7\\x8b(3H\\xd1\\v\\xfc\\x9c\\xd0K\\xacCd\\\"\\xe2\\xcd^i\\xab\\xffA\\xd8[\\xad\\xdda\\xba6\\xf4\\xa6\\xd5\\xd8`̓\\xaa\\x04}?%:\\xe1\\x9b6r2|b\\xaf\\x82AP\\xdc\\xc2\\x1ct\\xdc\\x0e\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/356e28f5914a0f16",
    "content": "go test fuzz v1\n[]byte(\"00000000000000000000000000000000\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/449388c309f148fd",
    "content": "go test fuzz v1\n[]byte(\"\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e0\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e\\x9e0000\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/582528ddfad69eb5",
    "content": "go test fuzz v1\n[]byte(\"0\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/754d7f941db60f1d",
    "content": "go test fuzz v1\n[]byte(\"0\\xc8\\xc8\\xc80000\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/84c7bfae679f54a7",
    "content": "go test fuzz v1\n[]byte(\"00000000000000000000000000000001\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/9abdd061069970e6",
    "content": "go test fuzz v1\n[]byte(\"00100\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/c50ed3d1a22fe00d",
    "content": "go test fuzz v1\n[]byte(\"00000000000000000000000000000000000000000000000000000000000000\\bPG\\x12\\r\\x14<\\x9e2\\x8d\\xb2\\f\\xdc:\\xe4\\xd1\\xf0G\\x93z\\xb6O\\xbe\\xec\\xc41\\xf9\\x9eq|\\x1c\\f\\x03\\x83D\\x90\\x18\\xfd\\x96\\xb6Z\\x1a\\xf0\\xc4\\xc9\\\"\\x14\\x15\\xca\\xcaw\\x04\\xd7\\xda\\xce%\\x92&\\x04#\\x8d_e\\xc1\\xa0\\xc4\\x1e\\x82\\x0e\\xe05\\x1a҃4\\xfe1-\\x00}\\xb4\\xb1e\\\\~Ġd\\x83\\x8b\\x93\\x9fl@25\\xeam\\x19\\xb4\\xd5\\x1d\\x9c\\x1b6\\x7f\\x96\\xc0\\x94\\xa4\\xd2\\xdf\\xfc\\x00\\x0e\\xb2AOCm\\xe6L\\xdc~\\x1e\\xeb\\x9a}\\x1co\\xa1\\xc5u/29\\x80JL9\\xf2\\xe8+\\\"\\xe6_\\x99'\\xaajJt\\x99\\x12\\x04\\xfb\\xff&p\\xac3\\x10g<\\x1f\\xfeX\\x87P\\xd8\\x17\\x13ڃ\\xa9&\\x8akQ\\xf3\\x13=!\\xa5\\x8d\\xa4)\\r\\bC^E\\x83'ʔ\\x00\\xac\\x10w\\xc3\\xd9ӧ\\x1e[A\\xffuEQn\\x8c*ar\\x8b49\\xc3\\xce\\x03u\\xf3\\nD\\xc5A\\xa8\\x98\\x14\\xba.\\x98A\\x8b\\x1f=<\\xa2\\x9a\\b\\\\\\x00(>h\\xbc\\x1d\\xbd+/Y\\x88\\xb2h\\x15\\xf5\\xd6z\\x8e`\\xd2[x Ϣ\\xacP3h\\xba\\x9bWs\\a\\xf4'-\\xb3\\x97\\x12g}\\xcf\\xea\\xcarD\\x12=\\xc5\\x16\\xd9\\xedâ[s\\x9f\\x17uX\\xd5\\xfa\\x9ee\\x85\\xd0\\xe0X\\xe6\\x1eS\\xf1\\x80E\\xd79\\xfaQ=\\xc5\\xe22\\x85\\xb1j\\x7fy[+eK{\\xb0\\xaa'\\xe8fA\\x7f\\x1a\\xae\\x8dMDŽ\\xa3\\x93\\x13\\x10͗{Qcs\\xfe\\x8c\\xe6n\\xc9\\xed\\xe9!|\\f/W-k>}$\\xed\\x9c[\\x990\\x18\\bE4|\\t;\\x0e\\xba\\x03\\xa9J\\xf5\\x15u\\xc6=\\xb4\\xc6WCI\\xb4\\xfb\\xab!\\x8eϽ\\xb1\\xad\\x8e\\x13\\xd1\\xecω\\xcej\\xd4\\\"\\x8d2z\\x86t4on\\x189%`\\xb8\\xad\\xbc\\xa0\\xae$\\x81ѾU\\x1fh\\xec\\xe4n\\a\\x10䖸\\x883\\xa9}\\x94\\xe6\\xac:\\xa7\\xfbBy^\\x1b\\xa2\\\"'(e\\xfcK\\xa4\\x01!D\\xb2\\xe4)\\x98Y/{\\x8d\\x82\\x97\\xef\\v\\xe6X\\xe3\\x04տ\\xb4\\x95\\xee\\x8bY\\x9f\\x93a \\xbfIǜ\\xb0\\x17]\\xc1\\xd8T\\xbb\\xd3\\x05\\xb9\\x8eP\\xee?\\xa5Gx\\x80G.&<L.\\xb0\\x89إ@j\\xf2\\xe3\\x92\\x18\\xc7j\\xaab7\\b\\x85\\x00\\xfc\\v\\xd1l\\xaa\\xac\\n\\xdc\\xcf|=6$\\x89\\xde\\xfcp\\xb3bz\\xcd!\\x95>\\x84'n\\x93, \\xdd\\x13b\\xf1\\xb72V/8\\xe0Fb\\x10a\\xe7\\x94\\xdb\\x03k\\x82i\\xf46f=+\\xce\\x13E\\xd2.b\\xb1\\x84ʴ\\x0f\\xf9\\xac\\xf8ʑ\\xef+J\\xe0\\x7f&\\xe4\\xa02<q\\xe7\\xec\\xe0*\\x9f1\\xe9L\\fR\\v>\\xa2\\x04\\xea\\x17k\\xb3T\\xb9\\x8c\\xd1І_o\\xbby|\\xa9\\x96<lk\\x89\\xe6\\xb4P\\x04;8DW\\xff\\xa7\\x92\\xea\\xde\\xc02\\xac{\\x06\\x16.\\x1d\\x89*\\xc6(B\\x88RB\\xc2[u\\x82\\xf7ci\\xf2\\xdbԑ鮟\\xb9RuP\\xe153\\xee\\xc7Y\\x03\\xb1\\x1d\\x9e\\x1e\\x97-\\xd6;6\\xceP\\xbcۄ\\\\\\xdbcC\\xd9>L\\xc3F\\xc7c\\x03\\xf7\\xf6\\x9d\\x86\\xe0'\\x1e\\\\\\x1aXZ\\a\\x9d\\x9d\\x01\\xf3\\xee\\x1bf:W\\x8c\\x83\\xacX\\xb7\\fs@{r\\xa1\\x94\\xb7\\xb9gwX/\\x1c\\x05\\xb4np\\xbfM\\x1bL\\x91\\x89\\xa9\\xf7\\x8bP\\x8e\\x99BA\\xf3ԞU>\\rZ\\xb4S\\xa6\\xf2d\\xf2\\xdbo\\xca\\x1a%\\\"7\\xf7\\xf6\\xb0\\xc8\\xcb\\x11\\x1b\\xcb\\x06Cq\\x85\\x7fW\\xe2\\xc1\\xe99\\xf7\\xc6?<mL\\x1bbe'\\x9b\\x9fn\\x1fܗ\\xb6R\\xaa\\xabH\\xf9\\xb3\\x88\\xff\\xe3t\\xe3]\\xceh;#Rmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmmm6\\xf9\\x17_,\\xff\\v$\\x12\\x18Ҕ9\\xb7k\\xbc\\xd0\\xc7oM\\xb9\\xa3kf\\x87\\xca\\tk\\x99\\x1c\\xcc\\x03\\xd5Af^\\x87ط\\xb9\\xb1%\\xb4\\xbf\\xea(eG\\xf8\\xf5L~\\x80\\x9f\\xd4\\xef\\xd45\\xec0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000\")\n"
  },
  {
    "path": "common/quotas/global/algorithm/testdata/fuzz/FuzzMultiUpdate/f1fafad245481a29",
    "content": "go test fuzz v1\n[]byte(\"00\\xff\\xff\\xff\\xff\\xff\\xff\\xff\\xff\\xff\\x0100\")\n"
  },
  {
    "path": "common/quotas/global/collection/collection.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n/*\nPackage collection contains the limiting-host ratelimit usage tracking and enforcing logic,\nwhich acts as a quotas.Collection.\n\nAt a very high level, this wraps [quotas.Limiter] values to do a few additional things\nin the context of the [github.com/uber/cadence/common/quotas/global] ratelimiter system:\n  - keep track of usage per key (quotas.Limiter does not support this natively)\n  - periodically report usage to each key's \"aggregator\" host (batched and fanned out in parallel)\n  - apply the aggregator's returned per-key RPS limits to future requests\n  - fall back to the wrapped limiter in case of failures (handled internally in internal.FallbackLimiter)\n*/\npackage collection\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/collection/internal\"\n\t\"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\ntype (\n\t// Collection wraps three kinds of ratelimiter collections, and allows choosing/shadowing which one is used per key:\n\t//   1. a \"global\" collection, which tracks usage, sends data to aggregators, and adjusts to match request patterns between hosts.\n\t//   2. a \"local\" collection, which tracks usage, but all decisions stay local (no requests are sent anywhere to share load info).\n\t//   3. a \"disabled\" collection, which does NOT track usage, and is used to bypass as much of this Collection as possible\n\t//\n\t// 1 is the reason this Collection exists - limiter-usage is tracked and submitted to aggregating hosts to\n\t// drive the whole \"global load-balanced ratelimiter\" system.  Internally, this will fall back to a local collection\n\t// if there is insufficient data or too many errors.\n\t//\n\t// 2 is essentially just a pass-through of the \"local\" collection, but with added allow/reject metrics.\n\t// Currently, all of these are our \"target RPS / num hosts in ring\" ratelimiters.\n\t// This is a lower-cost and MUCH less complex system, and it SHOULD be used if your Cadence cluster receives requests\n\t// in a roughly random way (e.g. any client-side request goes to a roughly-fair roughly-random Frontend host).\n\t//\n\t// 3 is a *complete* pass-through of the \"local\" collection (no metrics, no monitoring, nothing), and is intended to be temporary.\n\t// It is meant to be a maximum-safety fallback mode during initial rollout, and should be removed once 2 is demonstrated\n\t// to be safe enough to use in all cases.\n\t//\n\t// And last but not least:\n\t// 1's local-fallback and 2 MUST NOT share ratelimiter instances, or the local instances will be double-counted when shadowing.\n\t// they should likely be configured to behave identically, but they need to be separate instances.\n\tCollection struct {\n\t\tupdateInterval dynamicproperties.DurationPropertyFn\n\n\t\t// targetRPS is a small type-casting wrapper around dynamicconfig.IntPropertyFnWithDomainFilter\n\t\t// to prevent accidentally using the wrong key type.\n\t\ttargetRPS func(lkey shared.LocalKey) int\n\t\t// keyModes is a small type-casting wrapper around dynamicconfig.StringPropertyWithRatelimitKeyFilter\n\t\t// to prevent accidentally using the wrong key type.\n\t\tkeyModes func(gkey shared.GlobalKey) string\n\n\t\tlogger log.Logger\n\t\tscope  metrics.Scope\n\t\taggs   rpc.Client\n\n\t\tglobal   *internal.AtomicMap[shared.LocalKey, *internal.FallbackLimiter]\n\t\tlocal    *internal.AtomicMap[shared.LocalKey, internal.CountedLimiter]\n\t\tdisabled quotas.ICollection\n\t\tkm       shared.KeyMapper\n\n\t\tctx       context.Context // context used for background operations, canceled when stopping\n\t\tctxCancel func()\n\t\tstopped   chan struct{} // closed when stopping is complete\n\n\t\t// now exists largely for tests, elsewhere it is always time.Now\n\t\ttimesource clock.TimeSource\n\t}\n\n\t// basically an enum for key values.\n\t// when plug-in behavior is allowed, this will eventually be from a parsed string,\n\t// and values (aside from disabled) are not known at compile time.\n\tkeyMode string\n)\n\n// current known values.\n//\n// all unknown values become modeDisabled, which should also be used for\n// the default fallback in switch statements.\nvar (\n\tmodeDisabled          keyMode = \"disabled\"\n\tmodeLocal             keyMode = \"local\"\n\tmodeGlobal            keyMode = \"global\"\n\tmodeLocalShadowGlobal keyMode = \"local-shadow-global\"\n\tmodeGlobalShadowLocal keyMode = \"global-shadow-local\"\n)\n\nconst (\n\t// stop sending an idle key after N zero-valued updates in a row.\n\t// this will delete unused limiters locally, keeping collections small,\n\t// and also allow the aggregator to delete the key soon after.\n\t//\n\t// this has two major consequences:\n\t//  1. the next usage will use the local fallback, which may unfairly allow\n\t//     more requests than it should if other hosts are consuming the whole\n\t//     quota.  shorter values increase this effect.\n\t//  2. since zero values are submitted until garbage-collected, this host\n\t//     will repeatedly reduce its weight for a key until GC occurs.  if a\n\t//     request occurs during this window, it will take that many rounds to\n\t//     get back to the weight it left off at, because it is not starting from\n\t//     a zero state.  larger values increase this effect.\n\t//\n\t// currently, it is believed that this should be kept relatively small, to\n\t// get closer to the desired behavior: new frontend requests allow a small\n\t// number through (local fallback) until rebalanced, and idle time between\n\t// those requests does not very strongly penalize an abnormal-host caller\n\t// (e.g. leading to 0.0001 rps or similar).\n\tgcAfterIdle = 5 // TODO: change to time-based, like aggregator?\n)\n\nfunc (m keyMode) isLocalPrimary() bool {\n\treturn m == modeLocal || m == modeLocalShadowGlobal\n}\nfunc (m keyMode) isGlobalPrimary() bool {\n\treturn m == modeGlobal || m == modeGlobalShadowLocal\n}\nfunc (m keyMode) usesLocal() bool {\n\treturn m == modeLocal || m == modeLocalShadowGlobal || m == modeGlobalShadowLocal\n}\nfunc (m keyMode) usesGlobal() bool {\n\treturn m == modeGlobal || m == modeGlobalShadowLocal || m == modeLocalShadowGlobal\n}\n\nfunc New(\n\tname string,\n\t// quotas for \"local only\" behavior.\n\t// used for both \"local\" and \"disabled\" behavior, though \"local\" wraps the values with usage metrics.\n\tlocal quotas.ICollection,\n\t// quotas for the global limiter's internal fallback.\n\t//\n\t// this should be configured the same as the local collection, but it\n\t// MUST NOT actually be the same collection, or shadowing will double-count\n\t// events on the fallback.\n\tglobalFallback quotas.ICollection,\n\tupdateInterval dynamicproperties.DurationPropertyFn,\n\ttargetRPS dynamicproperties.IntPropertyFnWithDomainFilter,\n\tkeyModes dynamicproperties.StringPropertyWithRatelimitKeyFilter,\n\taggs rpc.Client,\n\tlogger log.Logger,\n\tmet metrics.Client,\n) (*Collection, error) {\n\tif local == globalFallback {\n\t\treturn nil, errors.New(\"local and global-fallback collections must be different\")\n\t}\n\n\tglobalCollection := internal.NewAtomicMap(func(key shared.LocalKey) *internal.FallbackLimiter {\n\t\treturn internal.NewFallbackLimiter(globalFallback.For(string(key)))\n\t})\n\tlocalCollection := internal.NewAtomicMap(func(key shared.LocalKey) internal.CountedLimiter {\n\t\treturn internal.NewCountedLimiter(local.For(string(key)))\n\t})\n\tctx, cancel := context.WithCancel(context.Background())\n\tc := &Collection{\n\t\tglobal:         globalCollection,\n\t\tlocal:          localCollection,\n\t\tdisabled:       local,\n\t\taggs:           aggs,\n\t\tupdateInterval: updateInterval,\n\t\ttargetRPS: func(lkey shared.LocalKey) int {\n\t\t\t// wrapper just ensures only local keys are used, as each\n\t\t\t// collection uses a separate dynamic config value.\n\t\t\t//\n\t\t\t// local keys also help ensure that \"local\" and \"disabled\"\n\t\t\t// use the same underlying golang.org/x/time/rate.Limiter,\n\t\t\t// so switching between them is practically a noop.\n\t\t\treturn targetRPS(string(lkey))\n\t\t},\n\n\t\tlogger: logger.WithTags(tag.ComponentGlobalRatelimiter, tag.GlobalRatelimiterCollectionName(name)),\n\t\tscope:  met.Scope(metrics.GlobalRatelimiter).Tagged(metrics.GlobalRatelimiterCollectionName(name)),\n\t\tkeyModes: func(gkey shared.GlobalKey) string {\n\t\t\t// all collections share a single dynamic config key,\n\t\t\t// so they must use the global key to uniquely identify all keys.\n\t\t\t//\n\t\t\t// in the future I think we may want to move the target RPS configs\n\t\t\t// to use this same strategy, to keep user-facing limits together and\n\t\t\t// easier to notice.\n\t\t\treturn keyModes(string(gkey))\n\t\t},\n\t\tkm: shared.PrefixKey(name + \":\"),\n\n\t\tctx:       ctx,\n\t\tctxCancel: cancel,\n\t\tstopped:   make(chan struct{}),\n\n\t\t// override externally in tests\n\t\ttimesource: clock.NewRealTimeSource(),\n\t}\n\treturn c, nil\n}\n\nfunc (c *Collection) TestOverrides(t *testing.T, timesource *clock.MockedTimeSource, km *shared.KeyMapper) {\n\tt.Helper()\n\tif timesource != nil {\n\t\tc.timesource = *timesource\n\t}\n\tif km != nil {\n\t\tc.km = *km\n\t}\n}\n\n// OnStart follows fx's OnStart hook semantics.\nfunc (c *Collection) OnStart(ctx context.Context) error {\n\tgo func() {\n\t\tdefer func() {\n\t\t\t// bad but not worth crashing the process\n\t\t\tlog.CapturePanic(recover(), c.logger, nil)\n\t\t}()\n\t\tdefer func() {\n\t\t\tclose(c.stopped)\n\t\t}()\n\t\tc.backgroundUpdateLoop()\n\t}()\n\n\treturn nil\n}\n\n// OnStop follows fx's OnStop hook semantics.\nfunc (c *Collection) OnStop(ctx context.Context) error {\n\tc.ctxCancel()\n\t// context timeout is for everything to shut down, but this one should be fast.\n\t// don't wait very long.\n\tctx, cancel := context.WithTimeout(ctx, 100*time.Millisecond)\n\tdefer cancel()\n\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn fmt.Errorf(\"collection failed to stop, context canceled: %w\", ctx.Err())\n\tcase <-c.stopped:\n\t\treturn nil\n\t}\n}\n\nfunc (c *Collection) keyMode(key shared.GlobalKey) keyMode {\n\trawMode := keyMode(c.keyModes(key))\n\tswitch rawMode {\n\tcase modeLocal, modeGlobal, modeLocalShadowGlobal, modeGlobalShadowLocal, modeDisabled:\n\t\treturn rawMode\n\tdefault:\n\t\tc.logger.Error(\n\t\t\t\"ratelimiter key forcefully disabled, bad mode\",\n\t\t\ttag.GlobalRatelimiterKey(string(key)),\n\t\t\ttag.GlobalRatelimiterKeyMode(string(rawMode)),\n\t\t)\n\t\treturn modeDisabled\n\t}\n}\n\nfunc (c *Collection) For(key string) quotas.Limiter {\n\tk := shared.LocalKey(key)\n\tgkey := c.km.LocalToGlobal(k)\n\tswitch c.keyMode(gkey) {\n\tcase modeLocal:\n\t\treturn c.local.Load(k)\n\tcase modeGlobal:\n\t\treturn c.global.Load(k)\n\tcase modeLocalShadowGlobal:\n\t\treturn internal.NewShadowedLimiter(c.local.Load(k), c.global.Load(k))\n\tcase modeGlobalShadowLocal:\n\t\treturn internal.NewShadowedLimiter(c.global.Load(k), c.local.Load(k))\n\n\tdefault:\n\t\t// pass through to the disabled collection, as if this fancy collection\n\t\t// wrapper did not exist.\n\t\t// this means usage cannot be collected or shadowed, so changing to or\n\t\t// from \"disabled\" may allow a burst of requests beyond intended limits.\n\t\t//\n\t\t// this is intended for safety during initial rollouts, not normal use,\n\t\t// and will be removed.  normally \"local\" or \"global\" should be used.\n\t\t// \"local\" SHOULD behave the same as \"disabled\", but with added\n\t\t// monitoring, and the ability to warm caches in either direction before\n\t\t// switching.\n\t\treturn c.disabled.For(key)\n\t}\n}\n\nfunc (c *Collection) shouldDeleteKey(mode keyMode, local bool) bool {\n\tif local {\n\t\treturn !mode.usesLocal()\n\t}\n\treturn !mode.usesGlobal()\n}\n\nfunc (c *Collection) backgroundUpdateLoop() {\n\ttickInterval := c.updateInterval()\n\n\tc.logger.Debug(\"update loop starting\")\n\n\tticker := c.timesource.NewTicker(tickInterval)\n\tdefer ticker.Stop()\n\n\tlastGatherTime := c.timesource.Now()\n\tfor {\n\t\tselect {\n\t\tcase <-c.ctx.Done():\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tnow := c.timesource.Now()\n\t\t\tc.logger.Debug(\"update tick\")\n\t\t\t// update interval if it changed\n\t\t\tnewTickInterval := c.updateInterval()\n\t\t\tif tickInterval != newTickInterval {\n\t\t\t\ttickInterval = newTickInterval\n\t\t\t\tticker.Reset(newTickInterval)\n\t\t\t}\n\n\t\t\t// submit local metrics asynchronously, because there's no need to do it synchronously\n\t\t\tlocalMetricsDone := make(chan struct{})\n\t\t\tgo func() {\n\t\t\t\tdefer close(localMetricsDone)\n\t\t\t\tc.local.Range(func(k shared.LocalKey, v internal.CountedLimiter) bool {\n\t\t\t\t\tgkey := c.km.LocalToGlobal(k)\n\t\t\t\t\tcounts := v.Collect()\n\t\t\t\t\tmode := c.keyMode(gkey)\n\t\t\t\t\tif counts.Idle > gcAfterIdle || c.shouldDeleteKey(mode, true) {\n\t\t\t\t\t\tc.logger.Debug(\n\t\t\t\t\t\t\t\"deleting local ratelimiter\",\n\t\t\t\t\t\t\ttag.GlobalRatelimiterKey(string(k)),\n\t\t\t\t\t\t\ttag.GlobalRatelimiterIdleCount(counts.Idle),\n\t\t\t\t\t\t)\n\t\t\t\t\t\tc.local.Delete(k)\n\t\t\t\t\t\treturn true // continue iterating, possibly delete others too\n\t\t\t\t\t}\n\n\t\t\t\t\tc.sendMetrics(k, true, mode, counts)\n\t\t\t\t\treturn true\n\t\t\t\t})\n\t\t\t}()\n\n\t\t\tcapacity := c.global.Len()\n\t\t\tcapacity += capacity / 10 // size may grow while we range, try to avoid reallocating in that case\n\t\t\tusage := make(map[shared.GlobalKey]rpc.Calls, capacity)\n\t\t\tstartups, failings, globals := 0, 0, 0\n\t\t\tc.global.Range(func(k shared.LocalKey, v *internal.FallbackLimiter) bool {\n\t\t\t\tgkey := c.km.LocalToGlobal(k)\n\t\t\t\tcounts, startup, failing := v.Collect()\n\t\t\t\tmode := c.keyMode(gkey)\n\t\t\t\tif counts.Idle > gcAfterIdle || c.shouldDeleteKey(mode, false) {\n\t\t\t\t\tc.logger.Debug(\n\t\t\t\t\t\t\"deleting global ratelimiter\",\n\t\t\t\t\t\ttag.GlobalRatelimiterKey(string(k)),\n\t\t\t\t\t\ttag.GlobalRatelimiterIdleCount(counts.Idle),\n\t\t\t\t\t)\n\t\t\t\t\tc.global.Delete(k)\n\t\t\t\t\treturn true // continue iterating, possibly delete others too\n\t\t\t\t}\n\n\t\t\t\tif startup {\n\t\t\t\t\tstartups++\n\t\t\t\t}\n\t\t\t\tif failing {\n\t\t\t\t\tfailings++\n\t\t\t\t}\n\t\t\t\tif !(startup || failing) {\n\t\t\t\t\tglobals++\n\t\t\t\t}\n\t\t\t\tusage[gkey] = rpc.Calls{\n\t\t\t\t\tAllowed:  counts.Allowed,\n\t\t\t\t\tRejected: counts.Rejected,\n\t\t\t\t}\n\t\t\t\tc.sendMetrics(k, false, mode, counts)\n\n\t\t\t\treturn true\n\t\t\t})\n\n\t\t\t// track how often we're using fallbacks vs non-fallbacks.\n\t\t\t// with per-host metrics this can tell us if a host is abnormal\n\t\t\t// compared to the rest of the cluster, e.g. persistent failing values.\n\t\t\tc.scope.RecordHistogramValue(metrics.GlobalRatelimiterStartupUsageHistogram, float64(startups))\n\t\t\tc.scope.RecordHistogramValue(metrics.GlobalRatelimiterFailingUsageHistogram, float64(failings))\n\t\t\tc.scope.RecordHistogramValue(metrics.GlobalRatelimiterGlobalUsageHistogram, float64(globals))\n\n\t\t\tif len(usage) > 0 {\n\t\t\t\tsw := c.scope.StartTimer(metrics.GlobalRatelimiterUpdateLatency)\n\t\t\t\tc.doUpdate(now.Sub(lastGatherTime), usage)\n\t\t\t\tsw.Stop()\n\t\t\t}\n\n\t\t\t<-localMetricsDone // should be much faster than doUpdate, unless it's no-opped\n\n\t\t\tlastGatherTime = now\n\t\t}\n\t}\n}\n\nfunc (c *Collection) sendMetrics(lkey shared.LocalKey, isLocalLimiter bool, mode keyMode, usage internal.UsageMetrics) {\n\t// emit quota information to make monitoring easier.\n\t// regrettably this will only be emitted when the key is (recently) in use, but\n\t// for active users this is probably sufficient.  other cases will probably need\n\t// a continual \"emit all quotas\" loop somewhere.\n\tc.scope.\n\t\tTagged(metrics.GlobalRatelimiterKeyTag(string(lkey))).\n\t\tUpdateGauge(metrics.GlobalRatelimiterQuota, float64(c.targetRPS(lkey)))\n\n\tlimitType := \"global\"\n\tif isLocalLimiter {\n\t\tlimitType = \"local\"\n\t}\n\tlimitTypeIsPrimary := isLocalLimiter && mode.isLocalPrimary() || !isLocalLimiter && mode.isGlobalPrimary()\n\n\tscope := c.scope.Tagged(\n\t\tmetrics.GlobalRatelimiterKeyTag(string(lkey)),\n\t\tmetrics.GlobalRatelimiterTypeTag(limitType),\n\t\t// useful for being able to tell when a key is \"in use\" or not, e.g. for monitoring purposes\n\t\tmetrics.GlobalRatelimiterIsPrimary(limitTypeIsPrimary),\n\t)\n\tscope.AddCounter(metrics.GlobalRatelimiterAllowedRequestsCount, int64(usage.Allowed))\n\tscope.AddCounter(metrics.GlobalRatelimiterRejectedRequestsCount, int64(usage.Rejected))\n}\n\n// doUpdate pushes usage data to aggregators. mutates `usage` as it runs.\nfunc (c *Collection) doUpdate(since time.Duration, usage map[shared.GlobalKey]rpc.Calls) {\n\tctx, cancel := context.WithTimeout(c.ctx, 10*time.Second) // TODO: configurable?  possibly even worth cutting off after 1s.\n\tdefer cancel()\n\tres := c.aggs.Update(ctx, since, usage)\n\tif res.Err != nil {\n\t\t// should not happen outside pretty major errors, but may recover next time.\n\t\t// if sustained, e.g. due to a bug, this should eventually cause the affected\n\t\t// keys to use their fallback behavior, which is easily alerted on.\n\t\tc.logger.Error(\"aggregator update error\", tag.Error(res.Err))\n\t}\n\t// either way, process all weights we did successfully retrieve.\n\tfor gkey, info := range res.Weights {\n\t\tdelete(usage, gkey) // clean up the list so we know what was missed\n\t\tlkey, err := c.km.GlobalToLocal(gkey)\n\t\tif err != nil {\n\t\t\t// should not happen unless agg returns keys that were not asked for,\n\t\t\t// and are not for this collection\n\t\t\tc.logger.Error(\"bad global key structure returned\", tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\n\t\t// < 0, nan, and inf in `info` are prevented by Update and do not need to be handled here,\n\t\t// though the math below could create new irrational values.\n\t\ttarget := rate.Limit(c.targetRPS(lkey))\n\t\tlimiter := c.global.Load(lkey)\n\t\tfallbackTarget := limiter.FallbackLimit()\n\t\tboosted := boostRPS(target, fallbackTarget, info.Weight, info.UsedRPS)\n\t\tlimiter.Update(boosted)\n\t}\n\n\t// mark all non-returned limits as failures.\n\t// this also handles request errors: all requested values are failed\n\t// because none of them have been deleted above.\n\t//\n\t// aside from request errors this should be rare and might represent a race\n\t// in the aggregator, but the semantics for handling it are pretty clear\n\t// either way.\n\tfor globalkey := range usage {\n\t\tlocalKey, err := c.km.GlobalToLocal(globalkey)\n\t\tif err != nil {\n\t\t\t// should not happen, would require local mapping to be wrong\n\t\t\tc.logger.Error(\"bad global key structure in local-only path\", tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\t\t// requested but not returned, bump the fallback fuse\n\t\tc.global.Load(localKey).FailedUpdate()\n\t}\n}\n\nfunc boostRPS(target, fallback rate.Limit, weight float64, usedRPS float64) rate.Limit {\n\tbaseline := target * rate.Limit(weight)\n\n\t// low weights lead to low per-host overage allowed, and this can lead to\n\t// restricting low-RPS-slightly-bursty requests quite a lot more than intended,\n\t// despite more than enough unused quota remaining at all times.\n\t//\n\t// as a partial mitigation, \"boost\" low-weight values, allowing them to use\n\t// more of the unused RPS than their weight would normally imply, up to the\n\t// fallback's limit.\n\t// as overall usage increases, this \"allowed overage\" will shrink, helping\n\t// ensure it keeps converging towards the global target RPS.\n\tif baseline < fallback {\n\t\t// unused should not go below zero, e.g. if target was lowered,\n\t\t// so this cannot reduce below the fair baseline.\n\t\tunused := math.Max(0, float64(target)-usedRPS)\n\t\tboosted := math.Min(\n\t\t\t// with many bursty low-weight hosts, this may allow too much.\n\t\t\t// currently this isn't really a concern, but this could be adjusted\n\t\t\t// by num-of-low-hosts or something if needed.\n\t\t\tfloat64(baseline)+unused,\n\t\t\t// can't exceed the local fallback value though.\n\t\t\t// this is also what would be allowed if this limit was garbage collected,\n\t\t\t// so it's already established as a \"safe enough\" value.\n\t\t\tfloat64(fallback),\n\t\t)\n\t\treturn rate.Limit(boosted)\n\t}\n\n\t// any host with a weighted target higher than the fallback will already be\n\t// allowing a relatively large \"growth room\" on top of its actual usage, so\n\t// they don't need this boost.\n\treturn baseline\n}\n"
  },
  {
    "path": "common/quotas/global/collection/collection_fuzz_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage collection\n\nimport (\n\t\"math\"\n\t\"testing\"\n\n\t\"golang.org/x/time/rate\"\n)\n\nfunc FuzzBoostRPS(f *testing.F) {\n\tf.Fuzz(func(t *testing.T, target, fallback, weight, used float64) {\n\t\ttarget = math.Abs(target)\n\t\tfallback = math.Abs(fallback)\n\t\tused = math.Abs(used)\n\t\tweight = weight - math.Floor(weight) // trim to 0..1\n\n\t\tif anyInvalid(target, fallback, used, weight) {\n\t\t\tt.Skip(\"bad numbers\")\n\t\t}\n\n\t\tif target < fallback {\n\t\t\t// fallback is always equal or below target, as it's `target / num hosts`.\n\t\t\ttarget, fallback = fallback, target\n\t\t}\n\n\t\tboosted := boostRPS(rate.Limit(target), rate.Limit(fallback), weight, used)\n\n\t\tif boosted > rate.Limit(target) {\n\t\t\t// should never exceed whole-cluster target\n\t\t\tt.Error(\"boosted beyond configured limit\")\n\t\t}\n\t\tif boosted < 0 {\n\t\t\t// should never become negative.\n\t\t\t//\n\t\t\t// the ratelimiter treats negatives as zero, so this is \"fine\",\n\t\t\t// but it's likely a sign of flawed logic.\n\t\t\tt.Error(\"boosted is negative\")\n\t\t}\n\t\tif math.IsNaN(float64(boosted)) {\n\t\t\tt.Error(\"boosted is NaN\")\n\t\t}\n\t\tif math.IsInf(float64(boosted), 0) {\n\t\t\tt.Error(\"boosted is inf\")\n\t\t}\n\t})\n}\n\nfunc anyInvalid(f ...float64) bool {\n\tfor _, v := range f {\n\t\tif math.IsNaN(v) {\n\t\t\treturn true\n\t\t}\n\t\tif math.IsInf(v, 0) {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/quotas/global/collection/collection_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage collection\n\nimport (\n\t\"context\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/atomic\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap/zapcore\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tdynamicquotas \"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/collection/internal\"\n\t\"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\nfunc TestLifecycleBasics(t *testing.T) {\n\tdefer goleak.VerifyNone(t) // must shut down cleanly\n\n\tctrl := gomock.NewController(t)\n\tlogger, logs := testlogger.NewObserved(t)\n\tc, err := New(\n\t\t\"test\",\n\t\tquotas.NewCollection(quotas.NewMockLimiterFactory(ctrl)),\n\t\tquotas.NewCollection(quotas.NewMockLimiterFactory(ctrl)),\n\t\tfunc(opts ...dynamicproperties.FilterOption) time.Duration { return time.Second },\n\t\tnil, // not used\n\t\tfunc(globalRatelimitKey string) string { return string(modeGlobal) },\n\t\tnil, // no rpc expected as there are no metrics to submit\n\t\tlogger,\n\t\tmetrics.NewNoopMetricsClient(),\n\t)\n\trequire.NoError(t, err)\n\tmts := clock.NewMockedTimeSource()\n\tc.TestOverrides(t, &mts, nil)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tdefer cancel()\n\trequire.NoError(t, c.OnStart(ctx))\n\n\tmts.BlockUntil(1)        // created timer in background updater\n\tmts.Advance(time.Second) // trigger the update\n\t// clockwork does not provide a way to wait for \"a ticker or timer event was consumed\",\n\t// so just sleep for a bit to allow that branch of the select statement to be followed.\n\t//\n\t// it seems *possible* for clockwork to provide this, maybe we should build it?\n\ttime.Sleep(time.Millisecond)\n\n\trequire.NoError(t, c.OnStop(ctx))\n\n\t// will fail if the tick did not begin processing, e.g. no tick or it raced with shutdown.\n\t// without a brief sleep after advancing time, this extremely rarely fails (try 10k tests to see)\n\tassert.Len(t, logs.FilterMessage(\"update tick\").All(), 1, \"no update logs, did it process the tick? all: %#v\", logs.All())\n}\n\nfunc TestCollectionLimitersCollectMetrics(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmode   keyMode\n\t\tlocal  internal.UsageMetrics\n\t\tglobal internal.UsageMetrics\n\t}{\n\t\t\"disabled\": {\n\t\t\tmode: modeDisabled,\n\t\t},\n\t\t\"local\": {\n\t\t\tmode:  modeLocal,\n\t\t\tlocal: internal.UsageMetrics{Allowed: 1},\n\t\t},\n\t\t\"global\": {\n\t\t\tmode:   modeGlobal,\n\t\t\tglobal: internal.UsageMetrics{Allowed: 1},\n\t\t},\n\t\t\"local-shadow\": {\n\t\t\tmode:   modeLocalShadowGlobal,\n\t\t\tlocal:  internal.UsageMetrics{Allowed: 1},\n\t\t\tglobal: internal.UsageMetrics{Allowed: 1},\n\t\t},\n\t\t\"global-shadow\": {\n\t\t\tmode:   modeGlobalShadowLocal,\n\t\t\tlocal:  internal.UsageMetrics{Allowed: 1},\n\t\t\tglobal: internal.UsageMetrics{Allowed: 1},\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ttest := test\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tt.Parallel()\n\n\t\t\t// anything non-zero\n\t\t\tlocalLimiters := dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int {\n\t\t\t\treturn 1\n\t\t\t})\n\t\t\tglobalLimiters := dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int {\n\t\t\t\treturn 10\n\t\t\t})\n\n\t\t\tc, err := New(\n\t\t\t\t\"test\",\n\t\t\t\tquotas.NewCollection(localLimiters),\n\t\t\t\tquotas.NewCollection(globalLimiters),\n\t\t\t\tfunc(opts ...dynamicproperties.FilterOption) time.Duration { return time.Second },\n\t\t\t\tfunc(domain string) int { return 5 },\n\t\t\t\tfunc(globalRatelimitKey string) string { return string(test.mode) },\n\t\t\t\tnil,\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t)\n\t\t\trequire.NoError(t, err)\n\t\t\t// not starting, in principle it could Collect() in the background and break this test.\n\n\t\t\t// perform one call\n\t\t\t_ = c.For(\"test\").Allow()\n\n\t\t\t// check counts on limits\n\t\t\tglobalEvents := make(map[shared.LocalKey]internal.UsageMetrics)\n\t\t\tc.global.Range(func(k shared.LocalKey, v *internal.FallbackLimiter) bool {\n\t\t\t\tusage, _, _ := v.Collect()\n\t\t\t\t_, ok := globalEvents[k]\n\t\t\t\trequire.False(t, ok, \"data key should not already exist: %v\", k)\n\t\t\t\tglobalEvents[k] = internal.UsageMetrics{\n\t\t\t\t\tAllowed:  usage.Allowed,\n\t\t\t\t\tRejected: usage.Rejected,\n\t\t\t\t}\n\t\t\t\treturn true\n\t\t\t})\n\t\t\tlocalEvents := make(map[shared.LocalKey]internal.UsageMetrics)\n\t\t\tc.local.Range(func(k shared.LocalKey, v internal.CountedLimiter) bool {\n\t\t\t\t_, ok := localEvents[k]\n\t\t\t\trequire.False(t, ok, \"data key should not already exist: %v\", k)\n\t\t\t\tlocalEvents[k] = v.Collect()\n\t\t\t\treturn true\n\t\t\t})\n\n\t\t\tif reflect.ValueOf(test.global).IsZero() {\n\t\t\t\tassert.Empty(t, globalEvents, \"unexpected global events, should be none\")\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, map[shared.LocalKey]internal.UsageMetrics{\"test\": test.global}, globalEvents, \"incorrect global metrics\")\n\t\t\t}\n\t\t\tif reflect.ValueOf(test.local).IsZero() {\n\t\t\t\tassert.Empty(t, localEvents, \"unexpected local events, should be none\")\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, map[shared.LocalKey]internal.UsageMetrics{\"test\": test.local}, localEvents, \"incorrect local metrics\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCollectionSubmitsDataAndUpdates(t *testing.T) {\n\tdefer goleak.VerifyNone(t) // must shut down cleanly\n\n\tctrl := gomock.NewController(t)\n\t// anything non-zero\n\tlimiters := dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int {\n\t\treturn 1\n\t})\n\tlogger, observed := testlogger.NewObserved(t)\n\taggs := rpc.NewMockClient(ctrl)\n\tc, err := New(\n\t\t\"test\",\n\t\tquotas.NewCollection(limiters),\n\t\tquotas.NewCollection(limiters),\n\t\tfunc(opts ...dynamicproperties.FilterOption) time.Duration { return time.Second },\n\t\tfunc(domain string) int { return 10 }, // target 10 rps / one per 100ms\n\t\tfunc(globalRatelimitKey string) string { return string(modeGlobal) },\n\t\taggs,\n\t\tlogger,\n\t\tmetrics.NewNoopMetricsClient(),\n\t)\n\trequire.NoError(t, err)\n\tmts := clock.NewMockedTimeSource()\n\tc.TestOverrides(t, &mts, nil)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tdefer cancel()\n\trequire.NoError(t, c.OnStart(ctx))\n\n\t// generate some data.\n\t// these start with the collections' limits, i.e. 1 token, so only one request is allowed.\n\tsomeLimiter := c.For(\"something\")\n\tres := someLimiter.Reserve()\n\tassert.True(t, res.Allow(), \"first request should have been allowed\")\n\tres.Used(true)\n\tassert.False(t, someLimiter.Allow(), \"second request on the same domain should have been rejected\")\n\tassert.NoError(t, c.For(\"other\").Wait(ctx), \"request on a different domain should be allowed\")\n\n\t// all limiters are now drained, and will take ~1s to recover normally.\n\n\t// prep for the calls\n\tcalled := make(chan struct{}, 1)\n\taggs.EXPECT().Update(gomock.Any(), gomock.Any(), map[shared.GlobalKey]rpc.Calls{\n\t\t\"test:other\": {\n\t\t\tAllowed:  1,\n\t\t\tRejected: 0,\n\t\t},\n\t\t\"test:something\": {\n\t\t\tAllowed:  1,\n\t\t\tRejected: 1,\n\t\t},\n\t}).DoAndReturn(func(ctx context.Context, period time.Duration, load map[shared.GlobalKey]rpc.Calls) rpc.UpdateResult {\n\t\tcalled <- struct{}{}\n\t\treturn rpc.UpdateResult{\n\t\t\tWeights: map[shared.GlobalKey]rpc.UpdateEntry{\n\t\t\t\t\"test:something\": {Weight: 1, UsedRPS: 2}, // should recover a token in 100ms\n\t\t\t\t// \"test:other\":   // not returned, should not change weight/rps and stay at 1s\n\t\t\t},\n\t\t\tErr: nil,\n\t\t}\n\t})\n\n\tmts.BlockUntil(1)        // need to have created timer in background updater\n\tmts.Advance(time.Second) // trigger the update.  also fills all ratelimiters.\n\n\t// wait until the calls occur\n\tselect {\n\tcase <-called:\n\tcase <-time.After(time.Second / 2):\n\t\t// keep total wait shorter than 1s to avoid refilling the slow token, just in case\n\t\tt.Fatal(\"did not make an rpc call after 1/2s\")\n\t}\n\t// crash if more calls occur\n\tclose(called)\n\n\t// wait for the updates to be sent to the ratelimiters, and for at least one \"something\"'s 100ms token to recover\n\ttime.Sleep(150 * time.Millisecond)\n\n\t// and make sure updates occurred\n\tassert.False(t, c.For(\"other\").Allow(), \"should be no recovered tokens yet on the slow limit\")  // less than 1 second == no tokens\n\tassert.True(t, c.For(\"something\").Allow(), \"should have allowed one request on the fast limit\") // over 100ms (got updated rate) == at least one token\n\t// specifically: because this is the first update to this limiter, it should now allow 10 requests, because the token bucket should be full.\n\t// just check once though, no need to be precise here.\n\tassert.True(t, c.For(\"something\").Allow(), \"after the initial update, the fast limiter should have extra tokens available\")\n\n\tassert.NoError(t, c.OnStop(ctx))\n\n\t// and make sure no non-fatal errors were logged\n\tfor _, log := range observed.All() {\n\t\tt.Logf(\"observed log: %v\", log.Message)\n\t\tassert.Lessf(t, log.Level, zapcore.ErrorLevel, \"should not be any error logs, got: %v\", log.Message)\n\t}\n}\n\nfunc TestTogglingMode(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\taggs := rpc.NewMockClient(gomock.NewController(t))\n\tvar mode atomic.Value\n\tmode.Store(modeDisabled)\n\tc, err := New(\n\t\t\"test\",\n\t\tquotas.NewCollection(dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int { return 1 })),\n\t\tquotas.NewCollection(dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int { return 1 })),\n\t\tfunc(opts ...dynamicproperties.FilterOption) time.Duration { return time.Second }, // update every second\n\t\tfunc(domain string) int { return 1 },\n\t\tfunc(globalRatelimitKey string) string { return string(mode.Load().(keyMode)) },\n\t\taggs,\n\t\ttestlogger.New(t),\n\t\tmetrics.NewNoopMetricsClient(),\n\t)\n\trequire.NoError(t, err)\n\tts := clock.NewMockedTimeSource()\n\tc.TestOverrides(t, &ts, nil)\n\n\trequire.NoError(t, c.OnStart(context.Background()), \"failed to start\")\n\tdefer func() {\n\t\t// reduces goroutine-noise on failure\n\t\trequire.NoError(t, c.OnStop(context.Background()), \"failed to stop\")\n\t}()\n\n\tts.BlockUntil(1) // background ticker created\n\n\t// must tick \"incrementally\" or updates may be lost unrealistically\n\ttick := func(n int) {\n\t\tfor i := 0; i < n; i++ {\n\t\t\tts.Advance(time.Second)\n\t\t\ttime.Sleep(10 * time.Millisecond) // allow code time to run to get back to waiting\n\t\t}\n\t}\n\n\t// a bit of shenaniganry:\n\t// use the same mock for all the sub-tests, but don't make any assertions on the number of calls.\n\t// if you do, it \"blames\" the parent test rather than the sub-test, which is hard to debug.\n\t//\n\t// this could be resolved by rerunning all steps <N for the Nth sub-test, but that's a fair bit\n\t// more boilerplate.\n\tremainingCalls := atomic.NewInt64(int64(0))\n\taggs.EXPECT().\n\t\tUpdate(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(context.Context, time.Duration, map[shared.GlobalKey]rpc.Calls) rpc.UpdateResult {\n\t\t\tremainingCalls.Sub(1)\n\t\t\t// empty because this test doesn't change if global chooses to use its internal fallback\n\t\t\treturn rpc.UpdateResult{}\n\t\t}).\n\t\tAnyTimes()\n\t// enforce actual call counts uniquely per test\n\texpectUpdateCall := func(t *testing.T, times int) {\n\t\tpreviousBudget := remainingCalls.Swap(int64(times))\n\t\trequire.Zero(t, previousBudget, \"should not have any remaining call budget before making an expectation, have %v and tried to add %v\", previousBudget, times)\n\t\tt.Cleanup(func() {\n\t\t\tcalled := remainingCalls.Load()\n\t\t\tassert.Zero(t, called, \"incorrect number of update calls performed, current budget is %v out of %v\", called, times)\n\t\t})\n\t}\n\n\tt.Run(\"disabled should not make any internal limiters or update calls\", func(t *testing.T) {\n\t\tmode.Store(modeDisabled)\n\t\tc.For(\"key\").Allow()\n\t\tassert.Equal(t, c.local.Len(), 0, \"local collection should be empty when disabled\")\n\t\tassert.Equal(t, c.global.Len(), 0, \"global collection should be empty when disabled\")\n\n\t\tassertLimiterKeys(t, c.local, \"local\")\n\t\tassertLimiterKeys(t, c.global, \"global\")\n\t\ttick(1) // ensure no update calls occur\n\t})\n\trequire.False(t, t.Failed(), \"stopping early, already failed\")\n\tt.Run(\"local creates only local keys\", func(t *testing.T) {\n\t\tmode.Store(modeLocal)\n\t\tc.For(\"key\").Allow()\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\tassertLimiterKeys(t, c.global, \"global\") // not in a global-accessing mode\n\t\ttick(1)                                  // ensure no update calls occur\n\t})\n\trequire.False(t, t.Failed(), \"stopping early, already failed\")\n\tt.Run(\"local keys GC\", func(t *testing.T) {\n\t\ttick(1)\n\t\t// not garbage-collected immediately\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\ttick(1)\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\ttick(10)\n\n\t\tassertLimiterKeys(t, c.local, \"local\")\n\t\tassertLimiterKeys(t, c.global, \"global\")\n\t})\n\trequire.False(t, t.Failed(), \"stopping early, already failed\")\n\tt.Run(\"global-local creates both keys\", func(t *testing.T) {\n\t\tmode.Store(modeLocalShadowGlobal)\n\t\tc.For(\"key\").Allow()\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\tassertLimiterKeys(t, c.global, \"global\", \"key\")\n\n\t\texpectUpdateCall(t, 1) // global-touching mode should send an update request\n\t\ttick(1)\n\t})\n\trequire.False(t, t.Failed(), \"stopping early, already failed\")\n\tt.Run(\"collections GC separately\", func(t *testing.T) {\n\t\t// must start with data in both\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\tassertLimiterKeys(t, c.global, \"global\", \"key\")\n\n\t\t// switch modes so only global is retained\n\t\tmode.Store(modeGlobal)\n\t\texpectUpdateCall(t, 1) // global mode should send an update request\n\t\ttick(1)\n\t\tassertLimiterKeys(t, c.local, \"local\") // GC'd because it's no longer a local-touching mode\n\t\tassertLimiterKeys(t, c.global, \"global\", \"key\")\n\t})\n\trequire.False(t, t.Failed(), \"stopping early, already failed\")\n\tt.Run(\"global keys GC\", func(t *testing.T) {\n\t\t// sanity check\n\t\tassertLimiterKeys(t, c.local, \"local\") // GC'd because it's no longer a local-touching mode\n\t\tassertLimiterKeys(t, c.global, \"global\", \"key\")\n\n\t\t// should send ~4 more updates (5 after above), then delete data and stop updating.\n\t\t// exact number doesn't matter, just that it's less than the number of\n\t\t// ticks, to show that empty means \"no update calls\"\n\t\texpectUpdateCall(t, 4)\n\t\ttick(10)\n\t\tassertLimiterKeys(t, c.local, \"local\")\n\t\tassertLimiterKeys(t, c.global, \"global\")\n\t})\n\tt.Run(\"disabling stops updates and GCs immediately\", func(t *testing.T) {\n\t\tmode.Store(modeGlobalShadowLocal)\n\t\tc.For(\"key\").Allow()\n\t\t// sanity check\n\t\tassertLimiterKeys(t, c.local, \"local\", \"key\")\n\t\tassertLimiterKeys(t, c.global, \"global\", \"key\")\n\n\t\tmode.Store(modeDisabled)\n\t\t// should not cause an update despite global keys existing,\n\t\t// disabled mode should delete them before sending\n\t\ttick(1)\n\t\t// should be emptied in a single tick\n\t\tassertLimiterKeys(t, c.local, \"local\")\n\t\tassertLimiterKeys(t, c.global, \"global\")\n\t})\n}\n\nfunc TestBoostRPS(t *testing.T) {\n\tt.Parallel()\n\n\ttests := map[string]struct {\n\t\ttargetLimit, fallbackLimit float64\n\t\tweight                     float64\n\t\tusedRPS                    float64\n\t\tresultLimit                float64\n\t}{\n\t\t\"low weight fully used\": {\n\t\t\ttargetLimit:   10,\n\t\t\tfallbackLimit: 3,\n\t\t\tweight:        0.11,\n\t\t\tusedRPS:       10,\n\t\t\tresultLimit:   1.1, // no unused RPS free for boost, gets fair value\n\t\t},\n\t\t\"low weight mostly unused\": {\n\t\t\ttargetLimit:   10,\n\t\t\tfallbackLimit: 3,\n\t\t\tweight:        0.11,\n\t\t\tusedRPS:       2,\n\t\t\tresultLimit:   3, // min(1.1 + 8, 3)\n\t\t},\n\t\t\"high weight fully used\": {\n\t\t\ttargetLimit:   10,\n\t\t\tfallbackLimit: 3,\n\t\t\tweight:        0.89,\n\t\t\tusedRPS:       10,\n\t\t\tresultLimit:   8.9, // no unused RPS free for boost, gets fair value\n\t\t},\n\t\t\"high weight mostly unused\": {\n\t\t\ttargetLimit:   10,\n\t\t\tfallbackLimit: 3,\n\t\t\tweight:        0.89,\n\t\t\tusedRPS:       2,\n\t\t\tresultLimit:   8.9, // should not be boosted beyond fair, nor limited below it\n\t\t},\n\t\t\"usage over target\": {\n\t\t\ttargetLimit:   10,\n\t\t\tfallbackLimit: 3,\n\t\t\tweight:        0.089,\n\t\t\tusedRPS:       100,\n\t\t\tresultLimit:   0.89, // should be weight-based at worst, and not be reduced by excess usage\n\t\t},\n\t}\n\tfor name, tc := range tests {\n\t\tname, tc := name, tc\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tassert.GreaterOrEqual(t, tc.weight, float64(0), \"sanity check on weight\")\n\t\t\tassert.LessOrEqual(t, tc.weight, float64(1), \"sanity check on weight\")\n\t\t\tresult := boostRPS(rate.Limit(tc.targetLimit), rate.Limit(tc.fallbackLimit), tc.weight, tc.usedRPS)\n\t\t\tassert.InDeltaf(t, tc.resultLimit, float64(result), 0.01, \"computed RPS does not match closely enough, should be %0.2f\", tc.resultLimit)\n\t\t})\n\t}\n}\n\nfunc assertLimiterKeys[V any](t *testing.T, collection *internal.AtomicMap[shared.LocalKey, V], name string, keys ...shared.LocalKey) {\n\tt.Helper()\n\tfound := map[shared.LocalKey]struct{}{}\n\tcollection.Range(func(k shared.LocalKey, v V) bool {\n\t\tfound[k] = struct{}{}\n\t\treturn true\n\t})\n\tkeyMap := map[shared.LocalKey]struct{}{}\n\tfor _, k := range keys {\n\t\tkeyMap[k] = struct{}{}\n\t}\n\tassert.Equal(t, keyMap, found, \"unexpected keys in %v collection\", name)\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/atomicmap.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n)\n\n// AtomicMap adds type safety around a sync.Map (which has atomic-like behavior), and:\n//   - implicitly constructs values as needed, not relying on zero values\n//   - simplifies the API quite a bit because very few methods are in use.\n//     in particular there is no \"Store\" currently because it is not needed.\n//   - tracks length (atomically, so values are only an estimate)\n//\n// Due to length tracking, this is marginally more costly when modifying contents\n// than \"just\" a type-safe sync.Map.  It should only be used when length is needed.\ntype AtomicMap[Key comparable, Value any] struct {\n\tcontents sync.Map\n\tcreate   func(key Key) Value\n\tlen      int64\n}\n\n// NewAtomicMap makes a simplified type-safe [sync.Map] that creates values as needed, and tracks length.\n//\n// The `create` callback will be called when creating a new value, possibly multiple times,\n// without synchronization.\n// It must be concurrency safe and should return ASAP to reduce the window for storage races,\n// so ideally it should be simple and non-blocking, or pulling from a pre-populated cache if not.\n//\n// Due to length tracking, this is marginally more costly when modifying contents\n// than \"just\" a type-safe [sync.Map].  It should only be used when length is needed.\nfunc NewAtomicMap[Key comparable, Value any](create func(key Key) Value) *AtomicMap[Key, Value] {\n\treturn &AtomicMap[Key, Value]{\n\t\tcontents: sync.Map{},\n\t\tcreate:   create,\n\t\tlen:      0,\n\t}\n}\n\n// Load will get the current Value for a Key, initializing it if necessary.\nfunc (t *AtomicMap[Key, Value]) Load(key Key) Value {\n\tval, loaded := t.contents.Load(key)\n\tif loaded {\n\t\treturn val.(Value)\n\t}\n\tcreated := t.create(key)\n\tval, loaded = t.contents.LoadOrStore(key, created)\n\tif !loaded {\n\t\t// stored a new value\n\t\tatomic.AddInt64(&t.len, 1)\n\t}\n\treturn val.(Value)\n}\n\n// Delete removes an entry from the map, and updates the length.\n//\n// Like the underlying [sync.Map.LoadAndDelete], this can be called concurrently with Range.\nfunc (t *AtomicMap[Key, Value]) Delete(k Key) {\n\t// whether used or not, this is included to ensure it is possible to build\n\t// while maintaining length so collections can be pruned later if needed.\n\t_, loaded := t.contents.LoadAndDelete(k)\n\tif loaded {\n\t\tatomic.AddInt64(&t.len, -1)\n\t}\n}\n\n// Range calls [sync.Map.Range] on the underlying [sync.Map], and has the same semantics.\n//\n// This can be used while concurrently modifying the map, and it may result\n// in ranging over more or fewer entries than Len would imply.\nfunc (t *AtomicMap[Key, Value]) Range(f func(k Key, v Value) bool) {\n\tt.contents.Range(func(k, v any) bool {\n\t\treturn f(k.(Key), v.(Value))\n\t})\n}\n\n// Len returns the currently-known size of the collection.  It cannot be guaranteed to\n// be precise, as the collection may change at any time during or after this call.\n//\n// In particular, Range may iterate over more or fewer entries.\nfunc (t *AtomicMap[Key, Value]) Len() int {\n\treturn int(atomic.LoadInt64(&t.len))\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/atomicmap_external_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal_test\n\nimport (\n\t\"math/rand\"\n\t\"runtime\"\n\t\"strconv\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/atomic\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common/quotas/global/collection/internal\"\n)\n\nfunc TestMapBasics(t *testing.T) {\n\tconst (\n\t\tloaded1 = \"four\"\n\t\tloaded2 = \"tenletters\"\n\t\ttried   = \"should not exist\"\n\t)\n\ttype custom struct{ value int }\n\n\tassertContentsEqual := func(t *testing.T, m *internal.AtomicMap[string, *custom], expected map[string]int) {\n\t\tdup := make(map[string]int) // to avoid mutating the original\n\t\tfor k, v := range expected {\n\t\t\tdup[k] = v\n\t\t}\n\t\tm.Range(func(k string, v *custom) bool {\n\t\t\tif _, ok := dup[k]; ok {\n\t\t\t\tdelete(dup, k)\n\t\t\t} else {\n\t\t\t\tt.Errorf(\"ranged over unexpected or duplicate key %q\", k)\n\t\t\t}\n\t\t\treturn true\n\t\t})\n\t\tassert.Empty(t, dup, \"did not find some contents\")\n\t}\n\n\tm := internal.NewAtomicMap(func(key string) *custom {\n\t\treturn &custom{value: len(key)}\n\t})\n\n\tt.Run(\"load should work\", func(t *testing.T) {\n\t\tv := m.Load(loaded1)\n\t\tassert.Equal(t, len(loaded1), v.value, \"should use the constructed value initially, not a zero value\")\n\t\tv.value = 10\n\t\treload := m.Load(loaded1)\n\t\tassert.Equal(t, v, reload, \"should return the same object when loaded more than once\")\n\t})\n\tt.Run(\"range should walk over only the one key\", func(t *testing.T) {\n\t\tassertContentsEqual(t, m, map[string]int{\n\t\t\tloaded1: len(loaded1),\n\t\t})\n\t})\n\tt.Run(\"loading a second value should range over two\", func(t *testing.T) {\n\t\t// init this value too\n\t\tv2 := m.Load(loaded2)\n\t\tassert.Equal(t, len(loaded2), v2.value, \"sanity check: loaded2 should be created correctly, like loaded1\")\n\t\tassertContentsEqual(t, m, map[string]int{\n\t\t\tloaded1: len(loaded1),\n\t\t\tloaded2: len(loaded2),\n\t\t})\n\t})\n}\n\nfunc TestMapNotRacy(t *testing.T) {\n\tcreates := atomic.NewInt64(0)\n\t// using a string pointer just to make things a bit riskier / more sensitive to races since mutation is possible.\n\t// no mutation currently occurs, but it seems slightly safer to leave it here for future changes.\n\tm := internal.NewAtomicMap(func(key string) *string {\n\t\ts := key\n\t\ts += \"-\"\n\t\ts += strconv.Itoa(int(creates.Inc())) // just to be recognizable\n\t\treturn &s\n\t})\n\n\t// call ALL the methods concurrently\n\tvar g errgroup.Group\n\tconst loops = 1000 // using 1,000 because 100 had some coverage flapping\n\tfor i := 0; i < loops; i++ {\n\t\tkey := strconv.Itoa(i)\n\t\tg.Go(func() error {\n\t\t\tv := m.Load(key)\n\t\t\tassert.NotEmpty(t, *v) // \"never nil\" also asserted by crashing\n\t\t\treturn nil\n\t\t})\n\t\t// try to load the same key multiple times\n\t\tg.Go(func() error {\n\t\t\tv := m.Load(key)\n\t\t\tassert.NotEmpty(t, *v)\n\t\t\treturn nil\n\t\t})\n\t\t// range over it while reading/writing\n\t\tg.Go(func() error {\n\t\t\tm.Range(func(k string, v *string) bool {\n\t\t\t\tassert.NotEmpty(t, k)\n\t\t\t\tassert.NotEmpty(t, *v)\n\t\t\t\treturn true\n\t\t\t})\n\t\t\treturn nil\n\t\t})\n\t\tg.Go(func() error {\n\t\t\t_ = m.Len() // value does not matter / hard to check usefully\n\t\t\treturn nil\n\t\t})\n\t\t// delete ~10% of keys to exercise that logic, and mostly ensure coverage\n\t\tif rand.Intn(10) == 0 {\n\t\t\tg.Go(func() error {\n\t\t\t\tm.Delete(key)\n\t\t\t\treturn nil\n\t\t\t})\n\t\t}\n\t}\n\trequire.NoError(t, g.Wait())\n\n\t// sanity-check to show decent concurrency:\n\t// - out-of-order inits (values can be both higher and lower than the key)\n\t// - duplicate inits (values higher than 100)\n\tsame, higher, lower, upper := 0, 0, 0, int64(0)\n\tm.Range(func(k string, v *string) bool {\n\t\tparts := strings.SplitN(*v, \"-\", 2)\n\n\t\t// sanity check that keys and values stay associated\n\t\tassert.Equal(t, k, parts[0], \"key %q and first part of value must match: %q\", k, *v)\n\n\t\tif parts[0] == parts[1] {\n\t\t\tsame++\n\t\t} else if parts[0] < parts[1] {\n\t\t\thigher++\n\t\t} else {\n\t\t\tlower++\n\t\t}\n\n\t\tvint, err := strconv.ParseInt(parts[1], 10, 64)\n\t\tassert.NoError(t, err, \"creates-%v should be parse-able as an int\", parts[1])\n\t\tif vint > upper {\n\t\t\tupper = vint\n\t\t}\n\t\treturn true\n\t})\n\n\tassert.LessOrEqual(t,\n\t\tint64(loops), upper,\n\t\t// regrettably not guaranteed due to deletions, but I have yet to see it.\n\t\t// if this becomes an issue, probably just delete it.\n\t\t\"did not observe a value at least as high as the number of loops.  \"+\n\t\t\t\"not technically impossible, just very unlikely\",\n\t)\n\n\tt.Logf(\n\t\t\"Metrics for cpu %v:\\n\"+\n\t\t\t\"\\tKey == value  (1=>1-1):      %v\\n\"+\n\t\t\t\"\\tValue higher  (5=>5-100):    %v\\n\"+\n\t\t\t\"\\tValue lower   (100=>100-5):  %v\\n\"+\n\t\t\t\"\\tNumber of iterations:        %v\\n\"+\n\t\t\t\"\\tHighest saved create:        %v\\n\"+ // same or higher than iterations\n\t\t\t\"\\tTotal num of creates:        %v\", // same or higher than saved\n\t\truntime.GOMAXPROCS(0), same, higher, lower, loops, upper, creates.Load(),\n\t)\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/counted.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/atomic\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\ntype (\n\tCountedLimiter struct {\n\t\twrapped quotas.Limiter\n\t\tusage   *AtomicUsage\n\t}\n\n\tcountedReservation struct {\n\t\twrapped clock.Reservation\n\t\tusage   *AtomicUsage // reference to the reservation's limiter's usage\n\t}\n\n\tAtomicUsage struct {\n\t\tallowed, rejected, idle atomic.Int64\n\t}\n\n\tUsageMetrics struct {\n\t\tAllowed, Rejected, Idle int\n\t}\n)\n\nvar _ quotas.Limiter = CountedLimiter{}\nvar _ clock.Reservation = countedReservation{}\n\nfunc NewCountedLimiter(limiter quotas.Limiter) CountedLimiter {\n\treturn CountedLimiter{\n\t\twrapped: limiter,\n\t\tusage:   &AtomicUsage{},\n\t}\n}\n\nfunc (c CountedLimiter) Allow() bool {\n\tallowed := c.wrapped.Allow()\n\tc.usage.Count(allowed)\n\treturn allowed\n}\n\nfunc (c CountedLimiter) Wait(ctx context.Context) error {\n\terr := c.wrapped.Wait(ctx)\n\tc.usage.Count(err == nil)\n\treturn err\n}\n\nfunc (c CountedLimiter) Reserve() clock.Reservation {\n\treturn countedReservation{\n\t\twrapped: c.wrapped.Reserve(),\n\t\tusage:   c.usage,\n\t}\n}\n\nfunc (c CountedLimiter) Limit() rate.Limit {\n\treturn c.wrapped.Limit()\n}\n\nfunc (c CountedLimiter) Collect() UsageMetrics {\n\treturn c.usage.Collect()\n}\n\nfunc (c countedReservation) Allow() bool {\n\treturn c.wrapped.Allow()\n}\n\nfunc (c countedReservation) Used(wasUsed bool) {\n\tc.wrapped.Used(wasUsed)\n\tc.usage.idle.Store(0)\n\tif c.Allow() {\n\t\tif wasUsed {\n\t\t\t// only counts as allowed if used, else it is hopefully rolled back.\n\t\t\t// this may or may not restore the token, but it does imply \"this limiter did not limit the event\".\n\t\t\tc.usage.Count(true)\n\t\t}\n\n\t\t// else it was canceled, and not \"used\".\n\t\t//\n\t\t// currently these are not tracked because some other rejection will occur\n\t\t// and be emitted in all our current uses, but with bad enough luck or\n\t\t// latency before canceling it could lead to misleading metrics.\n\t} else {\n\t\t// these reservations cannot be waited on so they cannot become allowed,\n\t\t// and they cannot be returned, so they are always rejected.\n\t\t//\n\t\t// specifically: it is likely that `wasUsed == Allow()`, so false cannot be\n\t\t// trusted to mean \"will not use for some other reason\", and the underlying\n\t\t// rate.Limiter did not change state anyway because it returned the\n\t\t// pending-token before becoming a clock.Reservation.\n\t\tc.usage.Count(false)\n\t}\n}\n\nfunc (a *AtomicUsage) Count(allowed bool) {\n\tif allowed {\n\t\ta.allowed.Add(1)\n\t} else {\n\t\ta.rejected.Add(1)\n\t}\n\ta.idle.Store(0)\n}\n\nfunc (a *AtomicUsage) Collect() UsageMetrics {\n\tm := UsageMetrics{\n\t\tAllowed:  int(a.allowed.Swap(0)),\n\t\tRejected: int(a.rejected.Swap(0)),\n\t}\n\tif m.Allowed+m.Rejected == 0 {\n\t\tm.Idle = int(a.idle.Add(1))\n\t} // else 0\n\treturn m\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/counted_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"golang.org/x/sync/errgroup\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\nfunc TestUsage(t *testing.T) {\n\tt.Run(\"tracks allow\", func(t *testing.T) {\n\t\tts := clock.NewMockedTimeSource()\n\t\tcounted := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 1, 1))\n\n\t\tassert.True(t, counted.Allow(), \"should match wrapped limiter\")\n\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, counted.Collect())\n\n\t\tassert.False(t, counted.Allow(), \"should match wrapped limiter\")\n\t\tassert.Equal(t, UsageMetrics{0, 1, 0}, counted.Collect(), \"previous collect should have reset counts, and should now have just a reject\")\n\t})\n\tt.Run(\"tracks wait\", func(t *testing.T) {\n\t\tts := clock.NewMockedTimeSource()\n\t\tcounted := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 1, 1))\n\n\t\t// consume the available token\n\t\trequireQuickly(t, 100*time.Millisecond, func() {\n\t\t\tassert.NoError(t, counted.Wait(context.Background()), \"should match wrapped limiter\")\n\t\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, counted.Collect())\n\t\t})\n\t\t// give up before the next token arrives\n\t\trequireQuickly(t, 100*time.Millisecond, func() {\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\t\t\tdefer cancel()\n\t\t\tassert.Error(t, counted.Wait(ctx), \"should match wrapped limiter\")\n\t\t\tassert.Equal(t, UsageMetrics{0, 1, 0}, counted.Collect(), \"previous collect should have reset counts, and should now have just a reject\")\n\t\t})\n\t\t// wait for the next token to arrive\n\t\trequireQuickly(t, 100*time.Millisecond, func() {\n\t\t\tvar g errgroup.Group\n\t\t\tg.Go(func() error {\n\t\t\t\t// waits for token to arrive\n\t\t\t\tassert.NoError(t, counted.Wait(context.Background()), \"should match wrapped limiter\")\n\t\t\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, counted.Collect())\n\t\t\t\treturn nil\n\t\t\t})\n\t\t\tg.Go(func() error {\n\t\t\t\ttime.Sleep(time.Millisecond)\n\t\t\t\tts.Advance(time.Second) // recover one token\n\t\t\t\treturn nil\n\t\t\t})\n\t\t\tassert.NoError(t, g.Wait())\n\t\t})\n\t})\n\tt.Run(\"tracks reserve\", func(t *testing.T) {\n\t\tts := clock.NewMockedTimeSource()\n\t\tlim := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 1, 1))\n\n\t\tr := lim.Reserve()\n\t\tassert.True(t, r.Allow(), \"should have used the available burst\")\n\t\tassert.Equal(t, UsageMetrics{0, 0, 1}, lim.Collect(), \"allowed tokens should not be counted until they're used\")\n\n\t\tr.Used(true)\n\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, lim.Collect(), \"using the token should reset idle and count allowed\")\n\n\t\tr = lim.Reserve()\n\t\tassert.False(t, r.Allow(), \"should not have a token available\")\n\t\tr.Used(false)\n\t\tassert.Equal(t, UsageMetrics{0, 1, 0}, lim.Collect(), \"not-allowed reservations count as rejection\")\n\t})\n\t// largely for coverage\n\tt.Run(\"supports Limit\", func(t *testing.T) {\n\t\trps := rate.Limit(1)\n\t\tlim := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(clock.NewMockedTimeSource(), rps, 1))\n\t\tassert.Equal(t, rps, lim.Limit())\n\t})\n}\n\nfunc TestRegression_ReserveCountsCorrectly(t *testing.T) {\n\trun := func(t *testing.T, lim quotas.Limiter, advance func(time.Duration), collect func() UsageMetrics) {\n\t\tallowed, returned, rejected := 0, 0, 0\n\t\tfor i := 0; ; i++ {\n\t\t\tif rejected > 3 {\n\t\t\t\t// normal exit: some rejects occurred.\n\t\t\t\tbreak // just to get more than 1 to be more interesting\n\t\t\t}\n\t\t\tif i > 1_000 {\n\t\t\t\t// infinite loop guard because it's a real mess to debug\n\t\t\t\tt.Error(\"too many attempts, test is not sane. allowed:\", allowed, \"rejected:\", rejected, \"returned:\", returned)\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\tr := lim.Reserve()\n\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t// time advancing before canceling should not affect this test because it is not concurrent,\n\t\t\t\t// so only do it sometimes to make sure that's true\n\t\t\t\tadvance(time.Millisecond)\n\t\t\t}\n\n\t\t\tif r.Allow() {\n\t\t\t\tif i%2 == 0 {\n\t\t\t\t\tallowed++\n\t\t\t\t\tr.Used(true)\n\t\t\t\t} else {\n\t\t\t\t\treturned++\n\t\t\t\t\tr.Used(false)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\trejected++\n\t\t\t\t// try with both true and false.\n\t\t\t\t// expected use is to call with false on all rejects, but it should not be required\n\t\t\t\tr.Used(i%2 == 0)\n\t\t\t}\n\t\t}\n\t\tusage := collect()\n\t\tt.Logf(\"usage: %#v\", usage)\n\t\tassert.NotZero(t, allowed, \"should have allowed some requests\")\n\t\tassert.Equal(t, allowed, usage.Allowed, \"wrong num of requests allowed\")\n\t\tassert.Equal(t, rejected, usage.Rejected, \"wrong num of requests rejected\")\n\t\tassert.Equal(t, 0, usage.Idle, \"limiter should never be idle in this test\")\n\t}\n\n\tt.Run(\"counted\", func(t *testing.T) {\n\t\t// \"base\" counting-limiter should count correctly\n\t\tts := clock.NewMockedTimeSource()\n\t\twrapped := clock.NewRateLimiterWithTimeSource(ts, 1, 100)\n\t\tlim := NewCountedLimiter(wrapped)\n\n\t\trun(t, lim, ts.Advance, lim.Collect)\n\t})\n\tt.Run(\"shadowed\", func(t *testing.T) {\n\t\t// \"shadowed\" should call the primary correctly at the very least\n\t\tts := clock.NewMockedTimeSource()\n\t\twrapped := clock.NewRateLimiterWithTimeSource(ts, 1, 100)\n\t\tcounted := NewCountedLimiter(wrapped)\n\t\tlim := NewShadowedLimiter(counted, allowlimiter{})\n\n\t\trun(t, lim, ts.Advance, counted.Collect)\n\t})\n\tt.Run(\"fallback\", func(t *testing.T) {\n\t\t// \"fallback\" uses a different implementation, but it should count exactly the same.\n\t\t// TODO: ideally it would actually be the same code, but that's a bit awkward due to needing different interfaces.\n\t\tts := clock.NewMockedTimeSource()\n\t\twrapped := clock.NewRateLimiterWithTimeSource(ts, 1, 100)\n\t\tl := NewFallbackLimiter(allowlimiter{})\n\t\tl.Update(1)         // allows using primary, else it calls the fallback\n\t\tl.primary = wrapped // cheat, just swap it out\n\n\t\trun(t, l, ts.Advance, func() UsageMetrics {\n\t\t\tu, _, _ := l.Collect()\n\t\t\treturn u\n\t\t})\n\t})\n}\n\n// Wait-based tests can block forever if there's an issue, better to fail fast.\nfunc requireQuickly(t *testing.T, timeout time.Duration, cb func()) {\n\tdone := make(chan struct{})\n\tgo func() {\n\t\tdefer close(done)\n\t\tcb()\n\t}()\n\twait := time.NewTimer(timeout)\n\tdefer wait.Stop()\n\tselect {\n\tcase <-done:\n\tcase <-wait.C: // should be far faster\n\t\tt.Fatal(\"timed out waiting for callback to return\")\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/fallback.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Package internal protects these types' concurrency primitives and other\n// internals from accidental misuse.\npackage internal\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"sync/atomic\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\ntype (\n\t// FallbackLimiter wraps a \"primary\" [rate.Limiter] with a \"fallback\" Limiter (i.e. a [github.com/uber/cadence/common/quotas.Limiter])\n\t// to use before the primary is fully ready, or after too many attempts to update the \"primary\" limit have failed.\n\t//\n\t// Intended use is:\n\t//   - \"primary\" is the global-load-balanced ratelimiter, with weights updating every few seconds\n\t//   - \"fallback\" is the \"rps / num-of-hosts\" ratelimiters we use many other places, which do not need to exchange data to work\n\t//   - When \"primary\" is not yet initialized (no updates yet at startup) or the global-load-balancing system is broken (too many failures),\n\t//     this limiter internally switches to the \"fallback\" limiter until it recovers.  Otherwise, \"primary\" is preferred.\n\t//   - both limiters are called all the time to keep their used-tokens roughly in sync (via shadowedLimiter), so they\n\t//     can be switched between without allowing bursts due to the \"other\" limit's token bucket filling up.\n\t//\n\t// The owning global/collection.Collection uses this as follows:\n\t//   - every couple seconds, Collect() usage information from all limiters\n\t//   - submit this to aggregator hosts, which return how much weight to allow this host per limiter\n\t//   - per limiter:\n\t//     - if a weight was returned, Update(rps) it so the new value is used in the future\n\t//     - else, call FailedUpdate() to shorten the \"use fallback\" fuse, possibly switching to the fallback by doing so\n\t//\n\t// During this sequence, a requested limit may not be returned by an aggregator for two major reasons,\n\t// and will result in a FailedUpdate() call to shorten a \"use fallback logic\" fuse:\n\t//   1. this limit is legitimately unused (or idle too long) and no data exists for it\n\t//   2. ring re-sharding led to losing track of the limit, and it is now owned by a host with insufficient data\n\t//\n\t// To mitigate the impact of the second case, missing data is temporarily ignored, and the previously-configured\n\t// Update rate is used without any changes.  This gives the aggregating host time to fill in its data, and a later\n\t// cycle should return \"real\" values that match actual usage, Update-ing this limiter.\n\t//\n\t// If no data has been returned for a sufficiently long time, the \"primary\" ratelimit will be ignored, and\n\t// the fallback limit will be used exclusively.  This is intended as a safety fallback, e.g. during initial\n\t// rollout/rollback and outage scenarios, normal use is not expected to rely on it.\n\t//\n\t// -----\n\t//\n\t// Implementation notes:\n\t//   - atomics are used as values instead of pointers simply for data locality, which requires FallbackLimiter\n\t//     to be used as a pointer.  `go vet -copylocks` should be enough to ensure this is done correctly, which is\n\t//     enforced by `make lint`.\n\t//   - the variety of atomics here is almost certainly NOT necessary for performance,\n\t//     it was just possible and simple enough to use, and it makes deadlocks trivially impossible.\n\t//     but rate.Limiter already use mutexes internally, so adding another to track counts / fallback\n\t//     decisions / etc should be entirely fine, if it becomes needed.\n\tFallbackLimiter struct {\n\t\t// number of failed updates, for deciding when to use fallback\n\t\tfailedUpdates atomic.Int64\n\t\t// counts of what was allowed vs not.\n\t\t// this is not collected by a wrapper because it would hide the limit-setting methods,\n\t\t// but it is otherwise identical to a CountedLimiter around this FallbackLimiter.\n\t\tusage AtomicUsage\n\n\t\t// primary / desired limiter, based on load-balanced information.\n\t\t//\n\t\t// note that use and modification is NOT synchronized externally,\n\t\t// so updates and deciding when to use the fallback must be done carefully\n\t\t// to avoid undesired combinations when they interleave.\n\t\t//\n\t\t// if needed or desired, just switch to locks, they should be more than\n\t\t// fast enough because the limiter already uses locking internally.\n\t\tprimary clock.Ratelimiter\n\t\t// fallback used when failedUpdates exceeds maxFailedUpdates,\n\t\t// or prior to the first Update call with globally-adjusted values.\n\t\tfallback quotas.Limiter\n\t}\n)\n\nconst (\n\t// when failed updates exceeds this value, use the fallback\n\tmaxFailedUpdates = 9\n\n\t// at startup / new limits in use, use the fallback logic.\n\t// that's expected / normal behavior as we have no data yet.\n\t//\n\t// positive values risk being interpreted as a \"failure\" though, so start deeply\n\t// negative, so it can be identified as \"still starting up\".\n\t// min-int64 has more than enough room to \"count\" failed updates for eons\n\t// without becoming positive, so it should not risk being misinterpreted.\n\tinitialFailedUpdates = math.MinInt64\n)\n\n// NewFallbackLimiter returns a quotas.Limiter that uses a simpler fallback when necessary,\n// and attempts to keep both the fallback and the \"real\" limiter \"warm\" by mirroring calls\n// between the two regardless of which is being used.\nfunc NewFallbackLimiter(fallback quotas.Limiter) *FallbackLimiter {\n\tl := &FallbackLimiter{\n\t\t// start from 0 as a default, the limiter is unused until it is updated.\n\t\t//\n\t\t// caution: it's important to not call any time-advancing methods on this until\n\t\t// after the first update, so the token bucket will fill properly.\n\t\t//\n\t\t// this (partially) mimics how new ratelimiters start with a full token bucket,\n\t\t// but there does not seem to be any way to perfectly mimic it without using locks,\n\t\t// and hopefully precision is not needed.\n\t\tprimary:  clock.NewRatelimiter(0, 0),\n\t\tfallback: fallback,\n\t}\n\tl.failedUpdates.Store(initialFailedUpdates)\n\treturn l\n}\n\n// Collect returns the current allowed/rejected values and resets them to zero (like CountedLimiter).\n// it also returns info about if this limiter is currently starting up or in a \"use fallback due to failures\" mode.\n// \"starting\" is expected any time a limit is new (or not recently used), \"failing\" may happen rarely\n// but should not ever be a steady event unless something is broken.\nfunc (b *FallbackLimiter) Collect() (usage UsageMetrics, starting, failing bool) {\n\tusage = b.usage.Collect()\n\tstarting, failing = b.mode()\n\treturn usage, starting, failing\n}\n\nfunc (b *FallbackLimiter) mode() (startingUp, tooManyFailures bool) {\n\tfailed := b.failedUpdates.Load()\n\tstartingUp = failed < 0\n\ttooManyFailures = failed > maxFailedUpdates\n\treturn startingUp, tooManyFailures\n}\n\nfunc (b *FallbackLimiter) useFallback() bool {\n\tstarting, fallback := b.mode()\n\treturn starting || fallback\n}\n\n// Update adjusts the underlying \"primary\" ratelimit, and resets the fallback fuse.\n// This implies switching to the \"primary\" limiter - if that is not desired, call Reset() immediately after.\nfunc (b *FallbackLimiter) Update(lim rate.Limit) {\n\t// caution: order here matters, to prevent potentially-old limiter values from being used\n\t// before they are updated.\n\t//\n\t// this is probably not going to be noticeable, but some users are sensitive to ANY\n\t// requests being ratelimited.  updating the fallback fuse last should be more reliable\n\t// in preventing that from happening when they would not otherwise be limited, e.g. with\n\t// the initial value of 0 burst, or if this rps was very low previously and is now high\n\t// (because the token bucket will be empty).\n\tdefer func() {\n\t\t// reset the use-fallback fuse, which may also (re)enable using the \"primary\" limiter\n\t\tb.failedUpdates.Store(0)\n\t}()\n\n\tif b.primary.Limit() == lim {\n\t\treturn\n\t}\n\n\tb.primary.SetLimit(lim)\n\tb.primary.SetBurst(max(1, int(lim))) // 0 burst blocks all requests, so allow at least 1 and rely on rps to fill sanely\n}\n\n// FailedUpdate should be called when a limit fails to update from an aggregator,\n// possibly implying some kind of problem, which may be unique to this limit.\n//\n// After crossing a threshold of failures (currently 10), the fallback will be switched to.\nfunc (b *FallbackLimiter) FailedUpdate() (failures int) {\n\tfailures = int(b.failedUpdates.Add(1)) // always increment the count for monitoring purposes\n\treturn failures\n}\n\n// Reset defers to the fallback limiter until an update is received.\n//\n// This is intended to be used when the current limit is no longer trustworthy for some reason,\n// determined externally rather than from too many FailedUpdate calls.\nfunc (b *FallbackLimiter) Reset() {\n\tb.failedUpdates.Store(initialFailedUpdates)\n}\n\nfunc (b *FallbackLimiter) Allow() bool {\n\tallowed := b.both().Allow()\n\tb.usage.Count(allowed)\n\treturn allowed\n}\n\nfunc (b *FallbackLimiter) Wait(ctx context.Context) error {\n\terr := b.both().Wait(ctx)\n\tb.usage.Count(err == nil)\n\treturn err\n}\n\nfunc (b *FallbackLimiter) Reserve() clock.Reservation {\n\treturn countedReservation{\n\t\twrapped: b.both().Reserve(),\n\t\tusage:   &b.usage,\n\t}\n}\n\nfunc (b *FallbackLimiter) Limit() rate.Limit {\n\tif b.useFallback() {\n\t\treturn b.fallback.Limit()\n\t}\n\treturn b.primary.Limit()\n}\n\nfunc (b *FallbackLimiter) FallbackLimit() rate.Limit {\n\treturn b.fallback.Limit()\n}\n\nfunc (b *FallbackLimiter) both() quotas.Limiter {\n\tstarting, failing := b.mode()\n\tif starting {\n\t\t// don't touch the primary until an update occurs,\n\t\t// to allow the token bucket to fill properly.\n\t\treturn b.fallback\n\t}\n\tif failing {\n\t\t// keep shadowing calls, so the token buckets are similar.\n\t\t// this prevents allowing a full burst when recovering, which seems\n\t\t// reasonable as things were apparently unhealthy.\n\t\treturn NewShadowedLimiter(b.fallback, b.primary)\n\t}\n\treturn NewShadowedLimiter(b.primary, b.fallback)\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/fallback_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\npackage internal\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/sync/errgroup\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\nfunc TestFallbackRegression(t *testing.T) {\n\tt.Run(\"primary should start with fallback limit\", func(t *testing.T) {\n\t\t// checks for an issue in earlier versions, where a newly-enabled primary limit would start with an empty token bucket,\n\t\t// unfairly rejecting requests that other limiters would allow (as they are created with a full bucket at their target rate).\n\t\t//\n\t\t// this can be spotted by doing this sequence:\n\t\t// - create a new limiter, fallback of N per second\n\t\t//   - this starts the \"primary\" limiter with limit=0, burst=0.\n\t\t// - Allow() a request\n\t\t//   - the limiters here may disagree. this is fine, primary is not used yet.\n\t\t//   - however: this advanced the primary's internal \"now\" to now.  (this now happens only after update)\n\t\t// - update the limit to match the fallback\n\t\t//   - primary now has limit==N, burst==N, but tokens=0 and now=now.\n\t\t// - Allow() a request\n\t\t//   - fallback allows due to having N-1 tokens\n\t\t//   - primary rejects due to 0 tokens\n\t\t//     ^ this is the problem.\n\t\t//     ^ this could also happen if calling `ratelimiter.Limit()` sets \"now\" in the future, though this seems unlikely.\n\t\t//\n\t\t// this uses real time because sleeping is not necessary, and it ensures\n\t\t// that any time-advancing calls that occur too early will lead to ~zero tokens.\n\n\t\tlimit := rate.Limit(10) // enough to accept all requests\n\t\trl := NewFallbackLimiter(clock.NewRatelimiter(limit, int(limit)))\n\n\t\t// sanity check: this may be fine to change, but it will mean the test needs to be rewritten because the token bucket may not be empty.\n\t\torig := rl.primary\n\t\tassert.Zero(t, orig.Burst(), \"sanity check: default primary ratelimiter's burst is not zero, this test likely needs a rewrite\")\n\n\t\t// simulate: call while still starting up, it should not touch the primary limiter.\n\t\tallowed := rl.Allow()\n\t\tbefore, starting, failing := rl.Collect()\n\t\tassert.True(t, allowed, \"first request should have been allowed, on the fallback limiter\")\n\t\tassert.True(t, starting, \"should be true == in starting mode\")\n\t\tassert.False(t, failing, \"should be false == not failing\")\n\t\tassert.Equal(t, UsageMetrics{\n\t\t\tAllowed:  1,\n\t\t\tRejected: 0,\n\t\t\tIdle:     0,\n\t\t}, before)\n\n\t\t// update: this should set limits, and either fill the bucket or cause it to be filled on the next request\n\t\trl.Update(limit)\n\n\t\t// call again: should be allowed, as this is the first time-touching request since it was created,\n\t\t// and the token bucket should have filled to match the first-update value.\n\t\tallowed = rl.Allow()\n\t\tafter, starting, failing := rl.Collect()\n\t\tassert.True(t, allowed, \"second request should have been allowed, on the primary limiter\")\n\t\tassert.False(t, starting, \"should be false == not in starting mode (using global)\")\n\t\tassert.False(t, failing, \"should be false == not failing\")\n\t\tassert.Equal(t, UsageMetrics{\n\t\t\tAllowed:  1,\n\t\t\tRejected: 0,\n\t\t\tIdle:     0,\n\t\t}, after)\n\t\tassert.InDeltaf(t,\n\t\t\t// Tokens() advances time, so this will not be precise.\n\t\t\trl.primary.Tokens(), int(limit)-1, 0.1,\n\t\t\t\"should have ~%v tokens: %v from the initial fill, minus 1 for the allow call\",\n\t\t\tint(limit)-1, int(limit),\n\t\t)\n\t})\n}\n\nfunc TestLimiter(t *testing.T) {\n\tt.Run(\"uses fallback initially\", func(t *testing.T) {\n\t\tm := quotas.NewMockLimiter(gomock.NewController(t))\n\t\tm.EXPECT().Allow().Times(1).Return(true)\n\t\tm.EXPECT().Allow().Times(2).Return(false)\n\t\tlim := NewFallbackLimiter(m)\n\n\t\tassert.True(t, lim.Allow(), \"should return fallback's first response\")\n\t\tassert.False(t, lim.Allow(), \"should return fallback's second response\")\n\t\tassert.False(t, lim.Allow(), \"should return fallback's third response\")\n\n\t\tusage, starting, failing := lim.Collect()\n\t\tassert.Equal(t, UsageMetrics{1, 2, 0}, usage, \"usage metrics should match returned values\")\n\t\tassert.True(t, starting, \"should still be starting up\")\n\t\tassert.False(t, failing, \"should not be failing, still starting up\")\n\t})\n\tt.Run(\"uses primary after update\", func(t *testing.T) {\n\t\tlim := NewFallbackLimiter(allowlimiter{})\n\t\tlim.Update(1_000_000) // large enough to allow millisecond sleeps to refill\n\n\t\ttime.Sleep(time.Millisecond) // allow some tokens to fill\n\t\tassert.True(t, lim.Allow(), \"limiter allows after enough time has passed\")\n\t\tassert.True(t, lim.Allow(), \"limiter allows burst too\")\n\n\t\tusage, startup, failing := lim.Collect()\n\t\tassert.False(t, failing, \"should not use fallback limiter after update\")\n\t\tassert.False(t, startup, \"should not be starting up, has had an update\")\n\t\tassert.Equal(t, UsageMetrics{2, 0, 0}, usage, \"usage should match behavior\")\n\t})\n\n\tt.Run(\"collecting usage data resets counts\", func(t *testing.T) {\n\t\tlim := NewFallbackLimiter(allowlimiter{})\n\t\tlim.Update(1)\n\t\tlim.Allow()\n\t\tlimit, _, _ := lim.Collect()\n\t\tassert.Equal(t, 1, limit.Allowed+limit.Rejected, \"should count one request\")\n\t\tlimit, _, _ = lim.Collect()\n\t\tassert.Zero(t, limit.Allowed+limit.Rejected, \"collect should have cleared the counts\")\n\t})\n\n\tt.Run(\"use-fallback fuse\", func(t *testing.T) {\n\t\t// duplicate to allow this test to be external, keep in sync by hand\n\t\tconst maxFailedUpdates = 9\n\t\tt.Cleanup(func() {\n\t\t\tif t.Failed() { // notices sub-test failures\n\t\t\t\tt.Logf(\"maxFailedUpdates may be out of sync (%v), check hardcoded values\", maxFailedUpdates)\n\t\t\t}\n\t\t})\n\n\t\tt.Run(\"falls back after too many failures\", func(t *testing.T) {\n\t\t\tlim := NewFallbackLimiter(allowlimiter{}) // fallback behavior is ignored\n\t\t\tlim.Update(1)\n\t\t\t_, startup, failing := lim.Collect()\n\t\t\trequire.False(t, failing, \"should not be using fallback\")\n\t\t\trequire.False(t, startup, \"should not be starting up, has had an update\")\n\n\t\t\t// the bucket will fill from time 0 on the first update, ensuring the first request is allowed\n\t\t\trequire.True(t, lim.Allow(), \"rate.Limiter should start with a full bucket\")\n\n\t\t\t// fail enough times to trigger a fallback\n\t\t\tfor i := 0; i < maxFailedUpdates; i++ {\n\t\t\t\t// build up to the edge...\n\t\t\t\tlim.FailedUpdate()\n\t\t\t\t_, _, failing = lim.Collect()\n\t\t\t\trequire.False(t, failing, \"should not be using fallback after %n failed updates\", i+1)\n\t\t\t}\n\t\t\tlim.FailedUpdate() // ... and push it over\n\t\t\t_, _, failing = lim.Collect()\n\t\t\trequire.True(t, failing, \"%vth update should switch to fallback\", maxFailedUpdates+1)\n\n\t\t\tassert.True(t, lim.Allow(), \"should return fallback's allowed request\")\n\t\t})\n\t\tt.Run(\"failing many times does not accidentally switch away from startup mode\", func(t *testing.T) {\n\t\t\tlim := NewFallbackLimiter(nil)\n\t\t\tfor i := 0; i < maxFailedUpdates*10; i++ {\n\t\t\t\tlim.FailedUpdate()\n\t\t\t\t_, startup, failing := lim.Collect()\n\t\t\t\trequire.True(t, startup, \"should still be starting up %v failed updates\", i+1)\n\t\t\t\trequire.False(t, failing, \"failing can only happen after startup finishess\")\n\t\t\t}\n\t\t})\n\t})\n\n\tt.Run(\"coverage\", func(t *testing.T) {\n\t\t// easy line to cover to bring to 100%\n\t\tlim := NewFallbackLimiter(nil)\n\t\tlim.Update(1)\n\t\tlim.Update(1) // should go down \"no changes needed, return early\" path\n\t})\n}\n\nfunc TestLimiterNotRacy(t *testing.T) {\n\tlim := NewFallbackLimiter(allowlimiter{})\n\tvar g errgroup.Group\n\tconst loops = 1000\n\tfor i := 0; i < loops; i++ {\n\t\t// clear ~10% of the time\n\t\tif rand.Intn(10) == 0 {\n\t\t\tg.Go(func() error {\n\t\t\t\tlim.Reset()\n\t\t\t\treturn nil\n\t\t\t})\n\t\t}\n\t\t// update ~10% of the time, fail the rest.\n\t\t// this should randomly clear occasionally via failures.\n\t\tif rand.Intn(10) == 0 {\n\t\t\tg.Go(func() error {\n\t\t\t\tlim.Update(rate.Limit(1 / rand.Float64())) // essentially never exercises \"same value, do nothing\" logic\n\t\t\t\treturn nil\n\t\t\t})\n\t\t} else {\n\t\t\tg.Go(func() error {\n\t\t\t\tlim.FailedUpdate()\n\t\t\t\treturn nil\n\t\t\t})\n\t\t}\n\t\t// collect occasionally\n\t\tif rand.Intn(10) == 0 {\n\t\t\tg.Go(func() error {\n\t\t\t\tlim.Collect()\n\t\t\t\treturn nil\n\t\t\t})\n\t\t}\n\t\tg.Go(func() error {\n\t\t\tlim.Allow()\n\t\t\treturn nil\n\t\t})\n\t\tg.Go(func() error {\n\t\t\tlim.Reserve().Used(rand.Int()%2 == 0)\n\t\t\treturn nil\n\t\t})\n\t\tg.Go(func() error {\n\t\t\tlim.Limit()\n\t\t\treturn nil\n\t\t})\n\t\tg.Go(func() error {\n\t\t\tlim.FallbackLimit()\n\t\t\treturn nil\n\t\t})\n\t\tg.Go(func() error {\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Microsecond)\n\t\t\tdefer cancel()\n\t\t\t_ = lim.Wait(ctx)\n\t\t\treturn nil\n\t\t})\n\t}\n}\n\nvar _ quotas.Limiter = allowlimiter{}\nvar _ clock.Reservation = allowres{}\n\ntype allowlimiter struct{}\ntype allowres struct{}\n\nfunc (allowlimiter) Allow() bool                  { return true }\nfunc (a allowlimiter) Wait(context.Context) error { return nil }\nfunc (a allowlimiter) Reserve() clock.Reservation { return allowres{} }\nfunc (a allowlimiter) Limit() rate.Limit          { return rate.Inf }\n\nfunc (a allowres) Allow() bool { return true }\nfunc (a allowres) Used(bool)   {}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/shadowed.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal\n\nimport (\n\t\"context\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\ntype (\n\tshadowedLimiter struct {\n\t\tprimary quotas.Limiter\n\t\tshadow  quotas.Limiter\n\t}\n\tshadowedReservation struct {\n\t\tprimary clock.Reservation\n\t\tshadow  clock.Reservation\n\t}\n)\n\nvar _ quotas.Limiter = shadowedLimiter{}\nvar _ clock.Reservation = shadowedReservation{}\n\n// NewShadowedLimiter mirrors all quotas.Limiter to its two wrapped limiters,\n// but only returns results from / waits as long as the \"primary\" one (first arg).\n//\n// This is intended for when you want to use one limit but also run a second\n// limit at the same time for comparison (to decide before switching) or\n// to warm data (to reduce spikes when switching).\nfunc NewShadowedLimiter(primary, shadow quotas.Limiter) quotas.Limiter {\n\treturn shadowedLimiter{\n\t\tprimary: primary,\n\t\tshadow:  shadow,\n\t}\n}\n\nfunc (s shadowedLimiter) Allow() bool {\n\t_ = s.shadow.Allow()\n\treturn s.primary.Allow()\n}\n\nfunc (s shadowedLimiter) Wait(ctx context.Context) (err error) {\n\tctx, cancel := context.WithCancel(ctx)\n\tdefer cancel()\n\n\tgo func() {\n\t\t_ = s.shadow.Wait(ctx) // not waited for because it should end ~immediately on its own\n\t}()\n\n\treturn s.primary.Wait(ctx)\n}\n\nfunc (s shadowedLimiter) Reserve() clock.Reservation {\n\treturn shadowedReservation{\n\t\tprimary: s.primary.Reserve(),\n\t\tshadow:  s.shadow.Reserve(),\n\t}\n}\n\nfunc (s shadowedLimiter) Limit() rate.Limit {\n\treturn s.primary.Limit()\n}\n\nfunc (s shadowedReservation) Allow() bool {\n\t_ = s.shadow.Allow()\n\treturn s.primary.Allow()\n}\n\nfunc (s shadowedReservation) Used(wasUsed bool) {\n\ts.primary.Used(wasUsed)\n\n\t// Shadow reservations are more complicated than they seem.\n\t//\n\t// Mirroring the *exact* call here will not ever let the shadow allow more RPS\n\t// than the primary, because the primary will reject and not-use requests exceeding its limit.\n\t//\n\t// This means you'll see fewer rejects (because shadow-cancels are not rejects), but:\n\t// - You can't assume each reject would have been an allow because it returned\n\t//   the shadow-token, allowing ~infinite not-counted-reject-RPS on shadows.\n\t// - You can't tell what the shadow's peak RPS actually is, because it will\n\t//   always be limited to match the primary (or lower).\n\t//\n\t// This doesn't really tell you anything useful about the shadow's behavior.\n\t// To improve this, we need to try to differentiate between \"was this\n\t// rejected directly by the primary\" vs \"was this rejected by something else,\n\t// and that's transitively canceling this reservation\".\n\t//\n\t// That's pretty easy: we can tell by the primary's Allow state.\n\n\tif s.primary.Allow() {\n\t\t// If wasUsed: this request was allowed, so the shadow should also try to consume any token\n\t\t// it received (pass `true`).\n\t\t//\n\t\t// If !wasUsed: this request was rejected by something else, e.g. a second-stage limiter.\n\t\t// this second-stage limiter would also have rejected the request if shadow was primary,\n\t\t// so try to return the shadow's token (pass `false`).\n\t\t// This will cap the shadow's allow-RPS to match the later-stage limiter, but that would\n\t\t// still be true if the shadow limiter became the primary, so it's accurate behavior.\n\t\t//\n\t\t// Tn other words: just forward `wasUsed` to shadow.\n\t\ts.shadow.Used(wasUsed)\n\t} else {\n\t\t// No transitive rejection is possible (we do not currently try to get two limiters separately before checking,\n\t\t// so this must be due to the primary rejecting the request), so we must assume that any token available would\n\t\t// have been used.\n\t\t//\n\t\t// This will be misleading about the behavior of later-stage limiters (shadow may allow more RPS than if it was\n\t\t// primary because a later-stage limiter could lead to it canceling reservations, effectively capping its RPS\n\t\t// below its configured RPS), but there isn't much we can do to know about that here.\n\t\t// Metrics on that second-stage limiter should show why a multi-stage operation is being rejected, and we could\n\t\t// start reporting \"allowed but canceled transitively\" metrics to show when these transitive limits are affecting\n\t\t// a limiter, and that's about as good as we can get.\n\t\t//\n\t\t// But regardless, for this shadowed reservation: always try to consume the token.\n\t\t// - This allows shadow-RPS to go above primary-RPS, which is not possible if `false` was forwarded here.\n\t\t// - Shadow-RPS can still go below primary-RPS because it can reject additional requests.\n\t\t//   it'll essentially just be \"primary-rejects + difference-in-allows\" which isn't all that useful, but it's relatively accurate.\n\t\ts.shadow.Used(true)\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/collection/internal/shadowed_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage internal\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\nfunc TestShadowed(t *testing.T) {\n\tt.Run(\"allow\", func(t *testing.T) {\n\t\ttests := map[string]struct {\n\t\t\tprimary bool\n\t\t\tshadow  bool\n\t\t\tallowed bool\n\t\t\tpUsage  UsageMetrics\n\t\t\tsUsage  UsageMetrics\n\t\t}{\n\t\t\t\"only primary allows\": {\n\t\t\t\tprimary: true,\n\t\t\t\tshadow:  false,\n\t\t\t\tallowed: true,\n\t\t\t\tpUsage:  UsageMetrics{Allowed: 1},\n\t\t\t\tsUsage:  UsageMetrics{Rejected: 1},\n\t\t\t},\n\t\t\t\"only shadow allows\": {\n\t\t\t\tprimary: false,\n\t\t\t\tshadow:  true,\n\t\t\t\tallowed: false,\n\t\t\t\tpUsage:  UsageMetrics{Rejected: 1},\n\t\t\t\tsUsage:  UsageMetrics{Allowed: 1},\n\t\t\t},\n\t\t\t\"both allow\": {\n\t\t\t\tprimary: true,\n\t\t\t\tshadow:  true,\n\t\t\t\tallowed: true,\n\t\t\t\tpUsage:  UsageMetrics{Allowed: 1},\n\t\t\t\tsUsage:  UsageMetrics{Allowed: 1},\n\t\t\t},\n\t\t\t\"both reject\": {\n\t\t\t\tprimary: false,\n\t\t\t\tshadow:  false,\n\t\t\t\tallowed: false,\n\t\t\t\tpUsage:  UsageMetrics{Rejected: 1},\n\t\t\t\tsUsage:  UsageMetrics{Rejected: 1},\n\t\t\t},\n\t\t}\n\t\tfor name, test := range tests {\n\t\t\ttest := test\n\t\t\tt.Run(\"allow-\"+name, func(t *testing.T) {\n\t\t\t\tts := clock.NewMockedTimeSource()\n\t\t\t\tprimaryBurst, shadowBurst := 0, 0\n\t\t\t\tif test.primary {\n\t\t\t\t\tprimaryBurst = 1\n\t\t\t\t}\n\t\t\t\tif test.shadow {\n\t\t\t\t\tshadowBurst = 1\n\t\t\t\t}\n\t\t\t\tprimary := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, rate.Every(time.Second), primaryBurst))\n\t\t\t\tshadow := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, rate.Every(time.Second), shadowBurst))\n\t\t\t\ts := NewShadowedLimiter(primary, shadow)\n\n\t\t\t\tassert.Equalf(t, test.allowed, s.Allow(), \"should match primary behavior: %v\", test.allowed)\n\t\t\t\tassert.Equal(t, test.pUsage, primary.Collect(), \"should have called primary\")\n\t\t\t\tassert.Equal(t, test.sUsage, shadow.Collect(), \"should have called shadow\")\n\t\t\t})\n\t\t\tt.Run(\"reserve-\"+name, func(t *testing.T) {\n\t\t\t\tts := clock.NewMockedTimeSource()\n\t\t\t\tprimaryBurst, shadowBurst := 0, 0\n\t\t\t\tif test.primary {\n\t\t\t\t\tprimaryBurst = 1\n\t\t\t\t}\n\t\t\t\tif test.shadow {\n\t\t\t\t\tshadowBurst = 1\n\t\t\t\t}\n\t\t\t\tprimary := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, rate.Every(time.Second), primaryBurst))\n\t\t\t\tshadow := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, rate.Every(time.Second), shadowBurst))\n\t\t\t\ts := NewShadowedLimiter(primary, shadow)\n\n\t\t\t\tres := s.Reserve()\n\t\t\t\tassert.Equalf(t, test.allowed, res.Allow(), \"should match primary behavior: %v\", test.allowed)\n\t\t\t\tres.Used(true)\n\t\t\t\tassert.Equal(t, test.pUsage, primary.Collect(), \"should have called primary\")\n\t\t\t\tassert.Equal(t, test.sUsage, shadow.Collect(), \"should have called shadow\")\n\t\t\t})\n\t\t\tt.Run(\"wait-\"+name, func(t *testing.T) {\n\t\t\t\t// the Wait operation is inherently racy in how it calls both limiters,\n\t\t\t\t// so this test delays the primary's response a bit to stabilize behavior.\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tprimary := quotas.NewMockLimiter(ctrl)\n\t\t\t\tprimary.EXPECT().Wait(gomock.Any()).DoAndReturn(func(ctx context.Context) error {\n\t\t\t\t\tselect {\n\t\t\t\t\tcase <-ctx.Done(): // canceled\n\t\t\t\t\t\treturn ctx.Err()\n\t\t\t\t\tcase <-time.After(10 * time.Millisecond): // give the shadow some time to run\n\t\t\t\t\t\tif test.primary {\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn context.Canceled\n\t\t\t\t\t}\n\t\t\t\t})\n\t\t\t\tcPrimary := NewCountedLimiter(primary)\n\t\t\t\tshadow := quotas.NewMockLimiter(ctrl)\n\t\t\t\tshadow.EXPECT().Wait(gomock.Any()).DoAndReturn(func(ctx context.Context) error {\n\t\t\t\t\t// shadow returns immediately to keep the test's inherent race\n\t\t\t\t\t// more stable.\n\t\t\t\t\tif test.shadow {\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}\n\t\t\t\t\treturn context.Canceled\n\t\t\t\t})\n\t\t\t\tcShadow := NewCountedLimiter(shadow)\n\t\t\t\ts := NewShadowedLimiter(cPrimary, cShadow)\n\n\t\t\t\trequireQuickly(t, 100*time.Millisecond, func() {\n\t\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\t\tgo func() {\n\t\t\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\t\t\t\t\t\tcancel()\n\t\t\t\t\t}()\n\t\t\t\t\terr := s.Wait(ctx)\n\t\t\t\t\tif test.allowed {\n\t\t\t\t\t\tassert.NoError(t, err, \"should have waited until allowed\")\n\t\t\t\t\t} else {\n\t\t\t\t\t\tassert.Error(t, err, \"should have been canceled\")\n\t\t\t\t\t}\n\t\t\t\t})\n\t\t\t\tassert.Equal(t, test.pUsage, cPrimary.Collect(), \"should have called primary\")\n\t\t\t\tassert.Equal(t, test.sUsage, cShadow.Collect(), \"should have called shadow\")\n\t\t\t})\n\t\t}\n\t})\n\tt.Run(\"reserve\", func(t *testing.T) {\n\t\tt.Run(\"only returns primary behavior\", func(t *testing.T) {\n\t\t\tt.Run(\"allowed\", func(t *testing.T) {\n\t\t\t\tts := clock.NewMockedTimeSource()\n\t\t\t\tprimary := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 1, 1)) // allows an event\n\t\t\t\tshadow := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 0, 0))  // always rejects\n\t\t\t\ts := NewShadowedLimiter(primary, shadow)\n\t\t\t\tres := s.Reserve()\n\t\t\t\tres.Used(res.Allow())\n\n\t\t\t\tassert.True(t, res.Allow(), \"should match primary behavior\")\n\t\t\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, primary.Collect(), \"should have called primary\")\n\t\t\t\tassert.Equal(t, UsageMetrics{0, 1, 0}, shadow.Collect(), \"should have called shadow\")\n\t\t\t})\n\t\t\tt.Run(\"rejected\", func(t *testing.T) {\n\t\t\t\tts := clock.NewMockedTimeSource()\n\t\t\t\tprimary := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 0, 0)) // always rejects\n\t\t\t\tshadow := NewCountedLimiter(clock.NewRateLimiterWithTimeSource(ts, 1, 1))  // allow an event\n\t\t\t\ts := NewShadowedLimiter(primary, shadow)\n\t\t\t\tres := s.Reserve()\n\t\t\t\tres.Used(res.Allow())\n\n\t\t\t\tassert.False(t, res.Allow(), \"should match primary behavior\")\n\t\t\t\tassert.Equal(t, UsageMetrics{0, 1, 0}, primary.Collect(), \"should have called primary\")\n\t\t\t\t// this also checks \"shadow can allow higher than primary\".\n\t\t\t\t// this was not true with the original too-simple implementation.\n\t\t\t\tassert.Equal(t, UsageMetrics{1, 0, 0}, shadow.Collect(), \"should have called shadow, and shadow should have allowed despite primary rejecting\")\n\t\t\t})\n\t\t})\n\t})\n\tt.Run(\"limit\", func(t *testing.T) {\n\t\tl := NewShadowedLimiter(&allowlimiter{}, clock.NewRatelimiter(rate.Limit(0), 1))\n\t\tassert.Equal(t, rate.Inf, l.Limit(), \"should return the primary limit, not shadowed\")\n\t})\n}\n"
  },
  {
    "path": "common/quotas/global/collection/testdata/fuzz/FuzzBoostRPS/1b0805e3169f0ae7",
    "content": "go test fuzz v1\nfloat64(-95)\nfloat64(-158)\nfloat64(-52.111111111111114)\nfloat64(0)\n"
  },
  {
    "path": "common/quotas/global/collection/testdata/fuzz/FuzzBoostRPS/216cd14a71215fe2",
    "content": "go test fuzz v1\nfloat64(-95)\nfloat64(0)\nfloat64(0)\nfloat64(0)\n"
  },
  {
    "path": "common/quotas/global/doc.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n/*\nPackage global contains a global-load-balance-aware ratelimiter (when complete).\n\n# High level overview\n\nAt a very high level, this will:\n  - collect usage metrics in-memory in \"limiting\" hosts (which use an in-memory ratelimiter)\n  - asynchronously submit these usage metrics to \"aggregating\" hosts, which will return per-key RPS data\n  - \"limiting\" hosts use this returned value to update their in-memory ratelimiters\n\n\"Aggregating\" hosts will internally:\n  - receive data for key X from all \"limiting\" hosts\n  - compute request-weight per host+key in the cluster\n  - return the \"limiting\"-host's weighted portion of each ratelimit to allow\n\nThe exact logic that the \"aggregating\" hosts use is intentionally hidden from the\n\"limiting\" hosts, so it can be changed without changing how they enforce limits.\n\nThe weight-calculation algorithm will (eventually) be controlled by a dynamicconfig value,\nto allow shadowing and experimenting with different algorithms at runtime, or disabling\nthe global logic altogether.\n\nSee sub-packages for implementation details.\n\n# Data flow in the system\n\nThe overall planned system's data-flow is driven entirely by limiting hosts repeatedly\ncalling an \"update(request, data)\" API on aggregating hosts, and using the return value.\nAggregating hosts make no outbound requests at all:\n\n\tlim: a host limiting e.g. user RPS for domain ASDF\n\tagg: a host aggregating this data, to compute fair RPS allowances\n\t┌───┐              ┌────┐┌────┐┌────┐              ┌────┐\n\t│lim│              │ring││agg1││agg2│              │lim2│\n\t└─┬─┘              └─┬──┘└─┬──┘└─┬──┘              └─┬──┘\n\t  │                  │     │     │                   │\n\t  │ shard(a,b,c)     │     │     │    update(a1,d37) │\n\t  │────────────────>┌┴┐   ┌┴┐<───────────────────────│\n\t  │    [[a], [b,c]] │ │   └┬┘───────────────────────>│\n\t  │<────────────────└┬┘    │     │                  ...\n\t  │                  │     │     │\n\t  │ update(a1)       │     │     │   got 1 request on \"a\" limit\n\t  │──────────────────────>┌┴┐    │\n\t  │                  │    │ │    │\n\t  │ update(b1,c2)    │    │ │    │   1 req on \"b\", 2 on \"c\"\n\t  │────────────────────────────>┌┴┐\n\t  │                  │    │ │   │ │\n\t  │     [b4/s, c9/s] │    │ │   │ │  allow \"b\" 4 RPS, \"c\" gets 9\n\t  │<────────────────────────────└┬┘\n\t  │                  │    │ │    │\n\t  │           [a2/s] │    │ │    │   2 RPS for \"a\"\n\t  │<──────────────────────└┬┘    │\n\t  │                  │     │     │\n\t┌─┴─┐              ┌─┴──┐┌─┴──┐┌─┴──┐\n\t│lim│              │ring││agg1││agg2│\n\t└───┘              └────┘└────┘└────┘\n\nThis package as a whole is only concerned with the high level request pattern in use:\n  - limiters enforce limits and collect data\n  - limiters asynchronously submit that data to all relevant aggregators, sharded by key\n  - aggregators decide per-limiter-and-key limits based on global data, and return it to limiters\n\nCurrently:\n  - limiters are Frontends, but can trivially be any Cadence service as they only send outbound requests\n  - aggregators are arbitrarily in History service because it already has a ring implemented (Frontend does not)\n  - neither of these are fundamentally required, and they may change or be in multiple services later\n\nThe actual contents of the requests and responses and how limits are decided is\nintentionally flexible to allow creating custom algorithms / data / etc externally without\nrequiring internal RPC spec definitions.  To achieve this, we have a custom\n[github.com/uber/cadence/common/types.Any] type that can be used with either Thrift or Protobuf.\nThis type is intentionally not google.protobuf.Any because there is no reason to require\nexchanging Protobuf data (particularly if using Thrift), and because that tends to bind to a\nspecific code generator, but it serves essentially the same purpose.\n\nA built-in version of all this will be coming in later commits, and possibly more in the future\nif we end up needing major changes that are reusable.\n\n# Planned use\n\nThe first use of this is intentionally focused on addressing imbalanced usage\nof per-domain user/worker/ES ratelimits across different frontend hosts, e.g.\ndue to unfair load balancing or per-datacenter affinity or other\noccasionally-changing-else-mostly-stable routing decisions.\nThis will essentially imply 3 keys per domain, and will look something like\n`domain-user-request` / `domain-worker-request` / etc - they just have to be\nunique per conceptual thing-with-a-configured-limit.\n\nThese frontend limits are generally configured up-front and are used to protect\nthe cluster (especially the database) from noisy neighbors and un-planned major\nusage changes.  True precision is not necessary (accepting 110 instead of 100 will\nnot collapse the cluster, but allowing *unlimited* could), but users plan around the\nnumbers we have given them, and significant or sustained differences in reality can\ncause problems for them.  Particularly if it significantly decreases due to a change\nin how our network decides to distribute their requests.\n\nBy detecting this imbalanced load we can fairly allow \"hot\" hosts more RPS than our\n\"limit / N hosts\" limiters allow (essentially stealing unused quota from low-usage\nhosts), and ultimately allow request rates closer to the intended limits as the\nbalance changes over time.  It's better for our users (they get what we told them\nthey get), and better for us (no need to artificially raise the limit to work around\nthe load imbalance, nor keep changing it as load changes).\n\nWeight-data loss is expected due to frontend or agg process losses, and allowed-limits\nnaturally trail real-time use due to the asynchronous nature - this is not *ideal* but\nis expected to be acceptable, as it will adjust / self-repair \"soon\" as normal updates\noccur.\n\nMore concretely, our planned target is something like:\n  - frontends (the limiting hosts) submit usage data every 3s\n  - history's aggregating algorithms \"quickly\" account for new load, and return allowed\n    limits to frontends\n  - within about 10s of a change, we want things to be ~90% accurate or better\n    compared to the \"ground truth\" (this implies ~0.5 weight per 3s update)\n  - with stable traffic, >99% accuracy at allowing the intended RPS of a limit key,\n    regardless of how requests are distributed\n  - briefly going over or under the intended limit during changes or partial data\n    is allowed, but generally requests should be incorrectly rejected over\n    incorrectly allowed\n\nFuture uses (with minor changes) may include things like:\n  - limiting cluster-wide database RPS across all hosts (one key, 50k+ RPS) despite\n    significantly-different load depending on server role and shard distribution\n  - different load-deciding algorithms, e.g. TCP Vegas or some other more sophisticated\n    weight-deciding algorithm (this first one is intentionally simple, and might prove\n    good enough)\n  - closed-source pluggable algorithms that share the same intent and request-flow\n    but little else (i.e. extendable by operators without new RPC definitions.\n    routing and usage patterns are often unique, no one approach can be ideal everywhere)\n\n# Out of scope\n\nUses explicitly out of scope for this whole structure includes things like:\n  - perfect accuracy on RPS targets (requires synchronous decisions)\n  - rapid and significant usage changes, like a domain sending all requests to one\n    frontend host one second, then a different the next, etc.\n  - brief spikes (e.g. sub-10-seconds.  these are broadly undesirable and can\n    behave worse than RPS/hosts would allow)\n  - guaranteed limiting below intended RPS (partial data may lead to split-brain-like\n    behavior, leading to over-allowing)\n  - fully automated target-RPS adjusting (e.g. system-load detecting, not pre-configured RPS.\n    this may be *possible* to implement in this system, but it is not *planned* and\n    may require significant changes)\n*/\npackage global\n"
  },
  {
    "path": "common/quotas/global/rpc/client.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Package rpc contains a concurrent RPC client, and handles mapping to/from the\n// Any-typed request details so other packages do not have to concern themselves\n// with those details.\npackage rpc\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination client_mock.go -self_package github.com/uber/cadence/common/quotas/global/rpc Client\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tCalls struct {\n\t\tAllowed  int\n\t\tRejected int\n\t}\n)\n\ntype (\n\t// UpdateResult holds all successes and errors encountered, to be distinct\n\t// from fatal errors that imply unusable results (i.e. a normal error return).\n\tUpdateResult struct {\n\t\t// all successfully-returned weights.\n\t\t// this may be populated and is safe to use even if Errors is present,\n\t\t// as some shards may have succeeded.\n\t\tWeights map[shared.GlobalKey]UpdateEntry\n\t\t// Any unexpected errors encountered, or nil if none.\n\t\t// Weights is valid even if this is present, though it may be empty.\n\t\t//\n\t\t// This should always be nil aside from major programming or rollout\n\t\t// errors, e.g. failures to serialize or deserialize data (coding errors\n\t\t// or incompatible server versions).\n\t\tErr error\n\t}\n\tUpdateEntry struct {\n\t\tWeight  float64\n\t\tUsedRPS float64\n\t}\n\n\tClient interface {\n\t\t// Update performs concurrent calls to all aggregating peers to send load info\n\t\t// and retrieve new per-key weights from the rest of the cluster.\n\t\t// This is intended to be called periodically in the background per process,\n\t\t// not synchronously or concurrently (per set of keys anyway), as it is fairly\n\t\t// high cost to shard keys and send so many requests.\n\t\t//\n\t\t// Currently, there are no truly fatal errors so this API does not return `error`.\n\t\t// Even in the presence of errors, some successful data may have been loaded,\n\t\t// and that will be part of the UpdateResult struct.\n\t\t//\n\t\t// As part of the contract of this call, UpdateResult will not ever contain\n\t\t// NaN, Inf, or negative values, as these imply fatal calculation problems.\n\t\t// It checks internally and will return an error rather than any value.\n\t\tUpdate(ctx context.Context, period time.Duration, load map[shared.GlobalKey]Calls) UpdateResult\n\t}\n\n\tclient struct {\n\t\thistory  history.Client\n\t\tresolver history.PeerResolver\n\t\tthisHost string\n\n\t\tlogger log.Logger\n\t\tscope  metrics.Scope\n\t}\n)\n\nvar _ Client = (*client)(nil)\n\nfunc New(\n\thistoryClient history.Client,\n\tresolver history.PeerResolver,\n\tlogger log.Logger,\n\tmet metrics.Client,\n) Client {\n\treturn &client{\n\t\thistory:  historyClient,\n\t\tresolver: resolver,\n\t\tthisHost: uuid.NewString(), // TODO: would descriptive be better?  but it works, unique ensures correctness.\n\t\tlogger:   logger,\n\t\tscope:    met.Scope(metrics.GlobalRatelimiter),\n\t}\n}\n\nfunc (c *client) Update(ctx context.Context, period time.Duration, load map[shared.GlobalKey]Calls) UpdateResult {\n\tkeys := make([]string, 0, len(load))\n\tfor k := range load {\n\t\tkeys = append(keys, string(k))\n\t}\n\tpeers, err := c.resolver.GlobalRatelimitPeers(keys)\n\tif err != nil {\n\t\t// should only happen if peers are unavailable, individual requests are handled other ways\n\t\treturn UpdateResult{\n\t\t\tErr: fmt.Errorf(\"unable to shard ratelimit update data: %w\", err),\n\t\t}\n\t}\n\n\tvar mut sync.Mutex\n\tweights := make(map[shared.GlobalKey]UpdateEntry, len(load)) // should get back most or all keys requested\n\n\tvar g errgroup.Group\n\t// could limit max concurrency easily with `g.SetLimit(n)` if desired,\n\t// but as each goes to a different host this seems fine to just blast out all at once,\n\t// and it makes timeouts easy because we don't need to reserve room for queued calls.\n\tfor peer, peerKeys := range peers {\n\t\tpeer, peerKeys := peer, peerKeys // for closure\n\t\tg.Go(func() (err error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), c.logger, &err) }()\n\n\t\t\tresult, err := c.updateSinglePeer(ctx, peer, period, filterKeys(peerKeys, load))\n\t\t\tif err != nil {\n\t\t\t\treturn err // does not stop other calls, nor should it\n\t\t\t}\n\n\t\t\tmut.Lock()\n\t\t\tdefer mut.Unlock()\n\t\t\tfor k, v := range result {\n\t\t\t\tif _, ok := weights[k]; ok {\n\t\t\t\t\t// should not happen as resolver.GlobalRatelimiterPeers ensures disjoint keys\n\t\t\t\t\t// are requested, and only the requested keys are returned.\n\t\t\t\t\treturn fmt.Errorf(\"received duplicate key %q from peer %q\", k, peer)\n\t\t\t\t}\n\t\t\t\tweights[k] = v\n\t\t\t}\n\t\t\treturn nil\n\t\t})\n\t}\n\n\terr = g.Wait()\n\tif err != nil {\n\t\t// wrap it so it's relatively severe-looking, as these should not happen\n\t\terr = fmt.Errorf(\"potentially fatal error during ratelimit update requests: %w\", err)\n\t}\n\treturn UpdateResult{\n\t\tWeights: weights,\n\t\tErr:     err,\n\t}\n}\n\nfunc (c *client) updateSinglePeer(ctx context.Context, peer history.Peer, period time.Duration, load map[shared.GlobalKey]Calls) (map[shared.GlobalKey]UpdateEntry, error) {\n\tanyValue, err := updateToAny(c.thisHost, period, load)\n\tif err != nil {\n\t\t// serialization errors should never happen\n\t\treturn nil, &SerializationError{err}\n\t}\n\n\tresult, err := c.history.RatelimitUpdate(\n\t\tctx,\n\t\t&types.RatelimitUpdateRequest{\n\t\t\tAny: anyValue,\n\t\t},\n\t\tpeer.ToYarpcShardKey(),\n\t)\n\tif err != nil {\n\t\t// client metrics are fine for monitoring, but it does not log errors.\n\t\t// TODO: possibly filter out or aggregate \"peer lost\" logs?  they're expected during deploys, since calls are not implicitly retried.\n\t\tc.logger.Warn(\n\t\t\t\"request failure when updating ratelimits\",\n\t\t\ttag.Error(&RPCError{err}),\n\t\t\ttag.GlobalRatelimiterPeer(string(peer)),\n\t\t)\n\t\treturn nil, nil // rpc errors are essentially expected, and not an \"error\"\n\t}\n\n\tresp, err := anyToWeights(result.Any)\n\tif err != nil {\n\t\t// deserialization errors should never happen\n\t\treturn nil, &SerializationError{err}\n\t}\n\n\t// and check the values.\n\t// - weights must be 0..1 inclusive\n\t// - used RPS must be non-negative\n\t// - NaNs and Infs are always rejected\n\t//\n\t// any failure is fatal to the whole request, that host cannot be trusted,\n\t// and any affected limiters should switch to their fallbacks if the issue\n\t// persists long enough.\n\tfor key, entry := range resp {\n\t\tif msg := shared.SanityCheckFloat(0, entry.Weight, 1, \"weight\"); msg != \"\" {\n\t\t\treturn nil, fmt.Errorf(\"bad value for key %q: from host: %q: %w\", key, peer, errors.New(msg))\n\t\t}\n\t\t// no upper bound, but true infs are not allowed\n\t\tif msg := shared.SanityCheckFloat(0, entry.UsedRPS, math.Inf(1), \"used rps\"); msg != \"\" {\n\t\t\treturn nil, fmt.Errorf(\"bad value for key %q: from host: %q %w\", key, peer, errors.New(msg))\n\t\t}\n\t}\n\n\treturn resp, nil\n}\n\nfunc filterKeys(keys []string, load map[shared.GlobalKey]Calls) map[shared.GlobalKey]Calls {\n\tresult := make(map[shared.GlobalKey]Calls, len(keys))\n\tfor _, k := range keys {\n\t\tresult[shared.GlobalKey(k)] = load[shared.GlobalKey(k)]\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/quotas/global/rpc/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: client.go\n//\n// Generated by this command:\n//\n//\tmockgen -package rpc -source client.go -destination client_mock.go -self_package github.com/uber/cadence/common/quotas/global/rpc Client\n//\n\n// Package rpc is a generated GoMock package.\npackage rpc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tshared \"github.com/uber/cadence/common/quotas/global/shared\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// Update mocks base method.\nfunc (m *MockClient) Update(ctx context.Context, period time.Duration, load map[shared.GlobalKey]Calls) UpdateResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Update\", ctx, period, load)\n\tret0, _ := ret[0].(UpdateResult)\n\treturn ret0\n}\n\n// Update indicates an expected call of Update.\nfunc (mr *MockClientMockRecorder) Update(ctx, period, load any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Update\", reflect.TypeOf((*MockClient)(nil).Update), ctx, period, load)\n}\n"
  },
  {
    "path": "common/quotas/global/rpc/client_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestClient(t *testing.T) {\n\tsetup := func(t *testing.T) (c *client, hc *history.MockClient, pr *history.MockPeerResolver) {\n\t\tctrl := gomock.NewController(t)\n\t\thc = history.NewMockClient(ctrl)\n\t\tpr = history.NewMockPeerResolver(ctrl)\n\t\timpl := New(hc, pr, testlogger.New(t), metrics.NewNoopMetricsClient())\n\t\treturn impl.(*client), hc, pr\n\t}\n\n\tencode := func(t *testing.T, w map[algorithm.Limit]algorithm.HostUsage) (*types.RatelimitUpdateResponse, error) {\n\t\ta, err := AggregatorWeightsToAny(w)\n\t\tassert.NoError(t, err, \"should be impossible: could not encode aggregator response to any\")\n\t\treturn &types.RatelimitUpdateResponse{\n\t\t\tAny: a,\n\t\t}, nil\n\t}\n\n\tt.Run(\"valid request\", func(t *testing.T) {\n\t\tc, hc, pr := setup(t)\n\t\tdata := map[shared.GlobalKey]Calls{\n\t\t\t\"a\": {10, 20},\n\t\t\t\"b\": {3, 5},\n\t\t\t\"c\": {78, 9},\n\t\t}\n\t\t// make a realistic used-RPS value (assuming 1 second elapsed)\n\t\tused := float64(0)\n\t\tfor _, calls := range data {\n\t\t\tused += float64(calls.Allowed + calls.Rejected)\n\t\t}\n\t\tresponse := map[shared.GlobalKey]UpdateEntry{\n\t\t\t\"a\": {Weight: 0.1, UsedRPS: used},\n\t\t\t\"b\": {Weight: 0.2, UsedRPS: used},\n\t\t\t\"c\": {Weight: 0.3, UsedRPS: used},\n\t\t}\n\t\tstringKeys := make([]string, 0, len(data))\n\t\tfor k := range data {\n\t\t\tstringKeys = append(stringKeys, string(k))\n\t\t}\n\t\tpr.EXPECT().GlobalRatelimitPeers(gomock.InAnyOrder(stringKeys)).Return(map[history.Peer][]string{\n\t\t\t\"agg-1\": {\"a\", \"c\"},\n\t\t\t\"agg-2\": {\"b\"},\n\t\t}, nil)\n\t\thc.EXPECT().\n\t\t\tRatelimitUpdate(gomock.Any(), matchrequest{t, []string{\"a\", \"c\"}}, matchyarpc{t, yarpc.WithShardKey(\"agg-1\")}).\n\t\t\tReturn(encode(t, map[algorithm.Limit]algorithm.HostUsage{\"a\": {Weight: 0.1, Used: algorithm.PerSecond(used)}, \"c\": {Weight: 0.3, Used: algorithm.PerSecond(used)}}))\n\t\thc.EXPECT().\n\t\t\tRatelimitUpdate(gomock.Any(), matchrequest{t, []string{\"b\"}}, matchyarpc{t, yarpc.WithShardKey(\"agg-2\")}).\n\t\t\tReturn(encode(t, map[algorithm.Limit]algorithm.HostUsage{\"b\": {Weight: 0.2, Used: algorithm.PerSecond(used)}}))\n\n\t\tresult := c.Update(context.Background(), time.Second, data)\n\t\tassert.NoError(t, result.Err)\n\t\tassert.Equal(t, response, result.Weights)\n\t})\n\tt.Run(\"bad peers\", func(t *testing.T) {\n\t\tc, _, pr := setup(t)\n\t\terr := errors.New(\"peer issue\")\n\t\tpr.EXPECT().GlobalRatelimitPeers(gomock.Any()).Return(nil, err)\n\t\tres := c.Update(context.Background(), time.Second, nil)\n\t\tassert.ErrorContains(t, res.Err, \"unable to shard\")\n\t\tassert.ErrorIs(t, res.Err, err) // should contain the cause\n\t})\n\tt.Run(\"bad math\", func(t *testing.T) {\n\t\t// the client should validate responses to make sure they do not contain\n\t\t// known-to-be-invalid data.\n\t\tc, hc, pr := setup(t)\n\t\tpr.EXPECT().GlobalRatelimitPeers([]string{\"a\"}).\n\t\t\tReturn(map[history.Peer][]string{\"agg-1\": {\"a\"}}, nil)\n\t\thc.EXPECT().RatelimitUpdate(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\tReturn(encode(t, map[algorithm.Limit]algorithm.HostUsage{\"a\": {\n\t\t\t\tWeight: algorithm.HostWeight(math.NaN()), // invalid value from agg, detected after deserializing\n\t\t\t\tUsed:   1,\n\t\t\t}}))\n\n\t\tresult := c.Update(context.Background(), time.Second, map[shared.GlobalKey]Calls{\"a\": {10, 20}})\n\t\tassert.Empty(t, result.Weights, \"bad data should lead to no usable response\")\n\t\tassert.ErrorContains(t, result.Err, `bad value for key \"a\"`, \"failure should mention the key\")\n\t\tassert.ErrorContains(t, result.Err, \"is NaN\", \"failure should mention the bad value type\")\n\t\tassert.ErrorContains(t, result.Err, \"potentially fatal error during ratelimit update requests\", \"failure should look alarming\")\n\t})\n}\n\nfunc stringy[T ~string](in []T) []string {\n\tout := make([]string, len(in))\n\tfor i := range in {\n\t\tout[i] = string(in[i])\n\t}\n\treturn out\n}\n\n// matches a single yarpc option\ntype matchyarpc struct {\n\tt   *testing.T\n\topt yarpc.CallOption\n}\n\nvar _ gomock.Matcher = matchyarpc{}\n\nfunc (m matchyarpc) Matches(x interface{}) bool {\n\topts, ok := x.([]yarpc.CallOption)\n\tif !ok {\n\t\tm.t.Logf(\"not a []yarpc.CallOption: %T\", x)\n\t\treturn false\n\t}\n\tm.t.Logf(\"opts: %#v, should be: %#v\", opts, m.opt)\n\treturn len(opts) == 1 && reflect.DeepEqual(opts[0], m.opt)\n}\n\nfunc (m matchyarpc) String() string {\n\treturn fmt.Sprintf(\"%#v\", m.opt)\n}\n\n// matches load-keys in an update request\ntype matchrequest struct {\n\tt        *testing.T\n\tloadKeys []string\n}\n\nvar _ gomock.Matcher = matchrequest{}\n\nfunc (a matchrequest) Matches(x interface{}) bool {\n\tv, ok := x.(*types.RatelimitUpdateRequest)\n\tif !ok {\n\t\ta.t.Logf(\"not a *types.RatelimitUpdateRequest: %T\", x)\n\t\treturn false\n\t}\n\tup, err := AnyToAggregatorUpdate(v.Any)\n\tif err != nil {\n\t\ta.t.Logf(\"failed to decode to agg update: %v\", err)\n\t\treturn false\n\t}\n\tkeys := map[string]struct{}{}\n\tfor _, k := range a.loadKeys {\n\t\tkeys[k] = struct{}{}\n\t}\n\tgotKeys := map[string]struct{}{}\n\tfor _, k := range stringy(maps.Keys(up.Load)) {\n\t\tgotKeys[k] = struct{}{}\n\t}\n\ta.t.Logf(\"want keys: %v, got keys: %v\", maps.Keys(keys), maps.Keys(gotKeys))\n\treturn reflect.DeepEqual(keys, gotKeys)\n}\n\nfunc (a matchrequest) String() string {\n\treturn fmt.Sprintf(\"request with load keys: %v\", a.loadKeys)\n}\n"
  },
  {
    "path": "common/quotas/global/rpc/error.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rpc\n\n// RPCError wraps errors to mark them as coming from RPC, i.e. basically expected.\ntype RPCError struct {\n\tcause error\n}\n\n// SerializationError wraps errors to mark them as coming from (de)serialization.\n//\n// In practice this should not ever occur, as it should imply one of:\n// - incorrect Any Value/ValueType pairing, which should not pass tests\n// - non-backwards-compatible type changes deployed in an unsafe way\n// - incompatible Thrift-binary changes\ntype SerializationError struct {\n\tcause error\n}\ntype errwrapper interface {\n\terror\n\tUnwrap() error\n}\n\nvar _ errwrapper = (*RPCError)(nil)\nvar _ errwrapper = (*SerializationError)(nil)\n\nfunc (e *RPCError) Error() string           { return e.cause.Error() }\nfunc (e *RPCError) Unwrap() error           { return e.cause }\nfunc (e *SerializationError) Error() string { return e.cause.Error() }\nfunc (e *SerializationError) Unwrap() error { return e.cause }\n"
  },
  {
    "path": "common/quotas/global/rpc/mapping.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\n// funcs are ordered to match how an Update request flows:\n// limiter -> Update on history -> aggregator, and return.\n//\n// 1. updateToAny: limiter-load is gathered and converted to an Any\n// 2. AnyToAggregatorUpdate: the Any is received by a history host and decoded for an aggregator\n// 3. AggregatorWeightsToAny: the aggregator responds, and is converted to another Any\n// 4. anyToWeights: the response is decoded to get the weights to use\n//\n// to make the rpc API generic for other kinds of data / multiple algorithms, just inject\n// these mappers so the request sharding can encode data per peer as needed.\n// e.g. use a mapper with a `peer_callback(keys for peer) Any` so it does not need to\n// know what kind of data is being converted.\n\nfunc updateToAny(host string, elapsed time.Duration, load map[shared.GlobalKey]Calls) (*types.Any, error) {\n\tcalls := make(map[string]*history.WeightedRatelimitCalls, len(load))\n\tfor k, v := range load {\n\t\tcalls[string(k)] = &history.WeightedRatelimitCalls{\n\t\t\tAllowed:  saturatingInt32(v.Allowed),\n\t\t\tRejected: saturatingInt32(v.Rejected),\n\t\t}\n\t}\n\treq := &history.WeightedRatelimitUsage{\n\t\tCaller:    host,\n\t\tElapsedMS: saturatingInt32(elapsed / time.Millisecond),\n\t\tCalls:     calls,\n\t}\n\tbytes, err := thrift.EncodeToBytes(req)\n\tif err != nil {\n\t\t// should be impossible\n\t\treturn nil, &SerializationError{err}\n\t}\n\treturn &types.Any{\n\t\tValueType: history.WeightedRatelimitUsageAnyType,\n\t\tValue:     bytes,\n\t}, nil\n}\n\n// AnyToAggregatorUpdate converts an in-bound ratelimiter's Any-typed data to the\n// structure that [algorithm.RequestWeighted.Update] expects.\nfunc AnyToAggregatorUpdate(request *types.Any) (algorithm.UpdateParams, error) {\n\tif request.ValueType != history.WeightedRatelimitUsageAnyType {\n\t\treturn algorithm.UpdateParams{}, fmt.Errorf(\"unrecognized Any type: %q\", request.ValueType)\n\t}\n\tvar out history.WeightedRatelimitUsage\n\terr := thrift.DecodeStructFromBytes(request.Value, &out)\n\tif err != nil {\n\t\treturn algorithm.UpdateParams{}, &SerializationError{err}\n\t}\n\tload := make(map[algorithm.Limit]algorithm.Requests, len(out.Calls))\n\tfor k, v := range out.Calls {\n\t\tload[algorithm.Limit(k)] = algorithm.Requests{\n\t\t\tAccepted: int(v.Allowed),\n\t\t\tRejected: int(v.Rejected),\n\t\t}\n\t}\n\tpar := algorithm.UpdateParams{\n\t\tID:      algorithm.Identity(out.Caller),\n\t\tElapsed: time.Duration(out.ElapsedMS) * time.Millisecond,\n\t\tLoad:    load,\n\t}\n\treturn par, par.Validate()\n}\n\n// AggregatorWeightsToAny converts the [algorithm.RequestWeighted.HostWeights] response\n// (for an in-bound Update request) to an Any-type compatible with RPC.\nfunc AggregatorWeightsToAny(response map[algorithm.Limit]algorithm.HostUsage) (*types.Any, error) {\n\tquotas := make(map[string]*history.WeightedRatelimitUsageQuotaEntry, len(response))\n\tfor k, v := range response {\n\t\tquotas[string(k)] = &history.WeightedRatelimitUsageQuotaEntry{\n\t\t\tWeight: float64(v.Weight),\n\t\t\tUsed:   float64(v.Used),\n\t\t}\n\t}\n\twrapper := &history.WeightedRatelimitUsageQuotas{\n\t\tQuotas: quotas,\n\t}\n\tdata, err := thrift.EncodeToBytes(wrapper)\n\tif err != nil {\n\t\t// should be impossible\n\t\treturn nil, &SerializationError{err}\n\t}\n\treturn &types.Any{\n\t\tValueType: history.WeightedRatelimitUsageQuotasAnyType,\n\t\tValue:     data,\n\t}, nil\n}\n\nfunc anyToWeights(response *types.Any) (map[shared.GlobalKey]UpdateEntry, error) {\n\tif response.ValueType != history.WeightedRatelimitUsageQuotasAnyType {\n\t\treturn nil, fmt.Errorf(\"unrecognized Any type: %q\", response.ValueType)\n\t}\n\tvar out history.WeightedRatelimitUsageQuotas\n\terr := thrift.DecodeStructFromBytes(response.Value, &out)\n\tif err != nil {\n\t\treturn nil, &SerializationError{err}\n\t}\n\tresult := make(map[shared.GlobalKey]UpdateEntry, len(out.Quotas))\n\tfor k, v := range out.Quotas {\n\t\tresult[shared.GlobalKey(k)] = UpdateEntry{\n\t\t\tWeight:  v.Weight,\n\t\t\tUsedRPS: v.Used,\n\t\t}\n\t}\n\treturn result, nil\n}\n\ntype numeric interface {\n\tint | time.Duration\n}\n\nfunc saturatingInt32[T numeric](i T) int32 {\n\tif i > math.MaxInt32 {\n\t\treturn math.MaxInt32\n\t}\n\treturn int32(i)\n}\n\n// exposed only for testing purposes\n\n// TestUpdateToAny is exposed for handler tests, use updateToAny in internal code instead.\nfunc TestUpdateToAny(t *testing.T, host string, elapsed time.Duration, load map[shared.GlobalKey]Calls) (*types.Any, error) {\n\tt.Helper()\n\treturn updateToAny(host, elapsed, load)\n}\n\n// TestAnyToWeights is exposed for handler tests, use anyToWeights in internal code instead\nfunc TestAnyToWeights(t *testing.T, response *types.Any) (map[shared.GlobalKey]UpdateEntry, error) {\n\tt.Helper()\n\treturn anyToWeights(response)\n}\n"
  },
  {
    "path": "common/quotas/global/rpc/mapping_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestMapping(t *testing.T) {\n\tt.Run(\"request\", func(t *testing.T) {\n\t\thost := uuid.NewString()\n\t\telapsed := 3 * time.Second\n\n\t\tinLoad := map[shared.GlobalKey]Calls{\n\t\t\t\"domain-x-user\":   {1, 2},\n\t\t\t\"domain-y-worker\": {3, 4},\n\t\t}\n\t\t// same contents but different types\n\t\toutLoad := map[algorithm.Limit]algorithm.Requests{\n\t\t\t\"domain-x-user\":   {1, 2},\n\t\t\t\"domain-y-worker\": {3, 4},\n\t\t}\n\n\t\tintermediate, err := updateToAny(host, elapsed, inLoad)\n\t\trequire.NoError(t, err)\n\n\t\tparams, err := AnyToAggregatorUpdate(intermediate)\n\t\trequire.NoError(t, err)\n\t\tassert.EqualValues(t, host, params.ID)\n\t\tassert.Equal(t, elapsed, params.Elapsed)\n\t\tassert.Equal(t, outLoad, params.Load)\n\t})\n\tt.Run(\"response\", func(t *testing.T) {\n\t\tinResponse := map[algorithm.Limit]algorithm.HostUsage{\n\t\t\t\"domain-x-user\":   {Weight: 0.5, Used: 10},\n\t\t\t\"domain-y-worker\": {Weight: 0.3, Used: 20},\n\t\t}\n\t\t// same contents but different types\n\t\toutResponse := map[shared.GlobalKey]UpdateEntry{\n\t\t\t\"domain-x-user\":   {Weight: 0.5, UsedRPS: 10},\n\t\t\t\"domain-y-worker\": {Weight: 0.3, UsedRPS: 20},\n\t\t}\n\n\t\tintermediate, err := AggregatorWeightsToAny(inResponse)\n\t\trequire.NoError(t, err)\n\n\t\tweights, err := anyToWeights(intermediate)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, outResponse, weights)\n\t})\n\tt.Run(\"bad data\", func(t *testing.T) {\n\t\tvalidUpdate, err := updateToAny(uuid.NewString(), time.Second, map[shared.GlobalKey]Calls{\"domain-x-user\": {1, 2}})\n\t\trequire.NoError(t, err)\n\t\tvalidResponse, err := AggregatorWeightsToAny(map[algorithm.Limit]algorithm.HostUsage{\"domain-x-user\": {Weight: 0.7}})\n\t\trequire.NoError(t, err)\n\n\t\t_, err = AnyToAggregatorUpdate(validUpdate)\n\t\trequire.NoError(t, err, \"sanity check failed, starting data should be valid\")\n\t\t_, err = anyToWeights(validResponse)\n\t\trequire.NoError(t, err, \"sanity check failed, starting data should be valid\")\n\n\t\twithType := func(valueType string, in *types.Any) *types.Any {\n\t\t\tdup := *in\n\t\t\tdup.ValueType = valueType\n\t\t\treturn &dup\n\t\t}\n\t\twithData := func(data []byte, in *types.Any) *types.Any {\n\t\t\tdup := *in\n\t\t\tdup.Value = data\n\t\t\treturn &dup\n\t\t}\n\t\ttoUpdate := func(data *types.Any) error {\n\t\t\t_, err := AnyToAggregatorUpdate(data)\n\t\t\treturn err\n\t\t}\n\t\ttoWeights := func(data *types.Any) error {\n\t\t\t_, err := anyToWeights(data)\n\t\t\treturn err\n\t\t}\n\n\t\ttests := map[string]struct {\n\t\t\tvalue       *types.Any\n\t\t\terrContains string\n\t\t\tdo          func(data *types.Any) error\n\t\t}{\n\t\t\t\"wrong value type update\": {\n\t\t\t\tvalue:       withType(\"other\", validUpdate),\n\t\t\t\tdo:          toUpdate,\n\t\t\t\terrContains: \"other\", // should reference the wrong value\n\t\t\t},\n\t\t\t\"wrong value type weights\": {\n\t\t\t\tvalue:       withType(\"other\", validResponse),\n\t\t\t\tdo:          toWeights,\n\t\t\t\terrContains: \"other\", // should reference the wrong value\n\t\t\t},\n\t\t\t\"bad data update\": {\n\t\t\t\tvalue:       withData([]byte(\"bad\"), validUpdate),\n\t\t\t\tdo:          toUpdate,\n\t\t\t\terrContains: \"could not decode\",\n\t\t\t},\n\t\t\t\"bad data weights\": {\n\t\t\t\tvalue:       withData([]byte(\"bad\"), validResponse),\n\t\t\t\tdo:          toWeights,\n\t\t\t\terrContains: \"could not decode\",\n\t\t\t},\n\t\t\t\"empty data update\": {\n\t\t\t\tvalue:       withData(nil, validUpdate),\n\t\t\t\tdo:          toUpdate,\n\t\t\t\terrContains: \"could not decode\",\n\t\t\t},\n\t\t\t\"empty data weights\": {\n\t\t\t\tvalue:       withData(nil, validResponse),\n\t\t\t\tdo:          toWeights,\n\t\t\t\terrContains: \"could not decode\",\n\t\t\t},\n\t\t\t\"swapped types update\": {\n\t\t\t\tvalue:       validUpdate,\n\t\t\t\tdo:          toWeights,\n\t\t\t\terrContains: \"unrecognized Any type\",\n\t\t\t},\n\t\t\t\"swapped types weights\": {\n\t\t\t\tvalue:       validResponse,\n\t\t\t\tdo:          toUpdate,\n\t\t\t\terrContains: \"unrecognized Any type\",\n\t\t\t},\n\t\t}\n\t\tfor name, test := range tests {\n\t\t\ttest := test\n\t\t\tt.Run(name, func(t *testing.T) {\n\t\t\t\trequire.ErrorContains(t, test.do(test.value), test.errContains)\n\t\t\t})\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "common/quotas/global/shared/keymapper.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Package shared holds some types that are used between multiple global-ratelimiter\n// packages, and need to be broken out to prevent import cycles.\npackage shared\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\ntype (\n\t// KeyMapper is used to ensure that all keys that get communicated to\n\t// aggregators are uniquely identifiable, when multiple collections are used.\n\t//\n\t// A unique prefix / pattern per collection should be sufficient.\n\tKeyMapper interface {\n\t\tLocalToGlobal(key LocalKey) GlobalKey\n\t\t// GlobalToLocal does the reverse of LocalToGlobal.\n\t\t//\n\t\t// An error will be returned if the global key does not seem to have\n\t\t// come from this mapper.\n\t\tGlobalToLocal(key GlobalKey) (LocalKey, error)\n\t}\n\tLocalKey  string // LocalKey represents a \"local\" / unique to this collection ratelimit key\n\tGlobalKey string // GlobalKey represents a \"global\" / globally unique ratelimit key (i.e. namespaced by collection)\n\n\tsimplekm struct {\n\t\tprefix string\n\t}\n)\n\nvar _ KeyMapper = simplekm{}\n\nfunc (s simplekm) LocalToGlobal(key LocalKey) GlobalKey {\n\treturn GlobalKey(s.prefix + string(key))\n}\n\nfunc (s simplekm) GlobalToLocal(key GlobalKey) (LocalKey, error) {\n\tk := string(key)\n\tif !strings.HasPrefix(k, s.prefix) {\n\t\treturn \"\", fmt.Errorf(\"missing prefix %q in global key: %q\", s.prefix, key)\n\t}\n\treturn LocalKey(strings.TrimPrefix(k, s.prefix)), nil\n}\n\n// PrefixKey builds a KeyMapper that simply adds (or removes) a fixed prefix to all keys.\n// If the global key does not have this prefix, an error will be returned when mapping to local.\nfunc PrefixKey(prefix string) KeyMapper {\n\treturn simplekm{\n\t\tprefix: prefix,\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/shared/sanity.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shared\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"strings\"\n\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// SanityCheckFloat checks bounds and numerically-valid-ness, and returns an error string if any check fails.\n//\n// This is the \"lower level\" of SanityLogFloat, when invalid means behavior needs to change,\n// and invalid data is an unavoidable possibility.\n//\n// If invalid data should *always* be avoided by earlier code, use SanityLogFloat instead to\n// do a paranoid log of violations without affecting behavior.\nfunc SanityCheckFloat[T ~float64](lower, actual, upper T, what string) (msg string) {\n\tif math.IsNaN(float64(actual)) {\n\t\treturn what + \" is NaN, should be impossible\"\n\t} else if math.IsInf(float64(actual), 0) {\n\t\treturn what + \" is Inf, should be impossible\"\n\t} else if actual < lower {\n\t\treturn what + \" is below lower bound, should be impossible\"\n\t} else if upper < actual {\n\t\treturn what + \" is above upper bound, should be impossible\"\n\t}\n\treturn \"\"\n}\n\n// SanityLogFloat ensures values are within sane bounds (inclusive) and not NaN or Inf,\n// and logs an error if these expectations are violated.\n// This should only be used when a value is believed to *always* be valid, not\n// when it needs to be clamped to a reasonable range.\n//\n// This largely exists because [math.Max] and similar retain NaN/Inf instead of\n// treating them as error states, and that (and other issues) caused problems.\n// There are a lot of potential gotchas with floats, and without good fuzzing\n// you're kinda unlikely to trigger them all, so this helps us be as paranoid as\n// we like with small enough cost to ignore (while also documenting expectations).\n//\n// The \"actual\" value is returned no matter what, to make this easier to use in\n// a logic chain without temp vars.\n//\n// Violations are intentionally NOT fatal, nor is a \"reasonable\" value returned,\n// as they are not expected to occur and code paths involved are not built with\n// that assumption.  Consider it \"undefined behavior\", this log exists only to\n// help troubleshoot the cause later if it happens.\n//\n// ---\n//\n// In tests, always assert that this is not ever triggered, by using the following:\n//\n//\tlogger, obs := logtest.NewObserved(t)\n//\tt.Cleanup(func() { AssertNoSanityCheckFailures(t, obs) })\n//\t// your test\nfunc SanityLogFloat[T ~float64](lower, actual, upper T, what string, logger log.Logger) T {\n\tmsg := SanityCheckFloat(lower, actual, upper, what)\n\tif msg != \"\" {\n\t\tlogger.Error(msg,\n\t\t\ttag.Dynamic(\"lower\", lower),\n\t\t\ttag.Dynamic(\"actual\", actual),\n\t\t\ttag.Dynamic(\"upper\", upper),\n\t\t\ttag.Dynamic(\"actual_type\", fmt.Sprintf(\"%T\", actual)))\n\t}\n\n\treturn actual\n}\n\n// small and easier to mock in tests, as no other methods are needed\ntype testingt interface {\n\tErrorf(string, ...any)\n}\n\nfunc AssertNoSanityCheckFailures(t testingt, entries []observer.LoggedEntry) {\n\tlogged := false\n\tfor _, entry := range entries {\n\t\tif entry.Level.Enabled(zap.ErrorLevel) && strings.Contains(entry.Message, \"impossible\") {\n\t\t\tt.Errorf(\"sanity check log detected: %v: %v\", entry.Message, entry.Context)\n\t\t\tif logged {\n\t\t\t\tt.Errorf(\"ignoring any remaining sanity check failures, comment this out to get all logs\")\n\t\t\t\t// produces 2 logs and then gives up, as large or fuzz tests can trigger thousands and that's very slow\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tlogged = true\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/quotas/global/shared/sanity_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shared\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestSanityChecks(t *testing.T) {\n\ttests := map[string]struct {\n\t\tlower, actual, upper float64\n\t\texpected             string\n\t}{\n\t\t\"nan\":      {lower: 0, actual: math.NaN(), upper: 0, expected: \"is NaN\"},\n\t\t\"pos inf\":  {lower: 0, actual: math.Inf(1), upper: 0, expected: \"is Inf\"},\n\t\t\"neg inf\":  {lower: 0, actual: math.Inf(-1), upper: 0, expected: \"is Inf\"},\n\t\t\"too low\":  {lower: 5, actual: 1, upper: 10, expected: \"below lower bound\"},\n\t\t\"too high\": {lower: 0, actual: 10, upper: 5, expected: \"above upper bound\"},\n\n\t\t\"inclusive lower\": {lower: 0, actual: 0, upper: 10, expected: \"\"},\n\t\t\"valid\":           {lower: 0, actual: 5, upper: 10, expected: \"\"},\n\t\t\"inclusive upper\": {lower: 0, actual: 10, upper: 10, expected: \"\"},\n\t}\n\tfor name, test := range tests {\n\t\tname, test := name, test\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmsg := SanityCheckFloat(test.lower, test.actual, test.upper, \"test\")\n\t\t\tif test.expected == \"\" {\n\t\t\t\tassert.Empty(t, msg, \"should not have failed validation\")\n\t\t\t} else {\n\t\t\t\tassert.Containsf(t, msg, test.expected, \"wrong or missing error case\")\n\t\t\t}\n\t\t})\n\t}\n\n\tt.Run(\"logs\", func(t *testing.T) {\n\t\tl, obs := testlogger.NewObserved(t)\n\t\t_ = SanityLogFloat[float64](5, 1, 10, \"test\", l)\n\t\tlogs := obs.TakeAll()\n\t\trequire.Len(t, logs, 1, \"should have made an error log\")\n\t\tassert.Contains(t, logs[0].Message, \"below lower bound\", \"should log the issue\")\n\t\tassert.Contains(t, logs[0].Message, \"test\", \"should mention the 'what'\")\n\t\tassert.NotZero(t, logs[0].ContextMap()[\"actual\"], \"should log the actual value\")\n\t})\n\tt.Run(\"assert helper\", func(t *testing.T) {\n\t\tl, obs := testlogger.NewObserved(t)\n\t\t_ = SanityLogFloat[float64](5, 1, 10, \"test\", l)\n\t\tlogs := obs.TakeAll()\n\t\trequire.Len(t, logs, 1, \"should have made an error log\")\n\n\t\terrobs := &errorfobserver{}\n\t\tAssertNoSanityCheckFailures(errobs, logs)\n\t\trequire.Len(t, errobs.logs, 1, \"should Errorf that there was a log\")\n\t\tassert.Contains(t, errobs.logs[0], \"sanity check log detected\")\n\t\tassert.Contains(t, errobs.logs[0], \"below lower bound\", \"should contain the original message\")\n\n\t\terrobs = &errorfobserver{}                  // clear the logs\n\t\ttripleLog := append(logs, logs[0], logs[0]) // fake having three logs, details don't matter\n\t\tAssertNoSanityCheckFailures(errobs, tripleLog)\n\t\trequire.Len(t, errobs.logs, 3, \"should have two Errorfs and one giving up\")\n\t\tassert.Contains(t, errobs.logs[0], \"sanity check log detected\")\n\t\tassert.Contains(t, errobs.logs[1], \"sanity check log detected\")\n\t\tassert.Contains(t, errobs.logs[2], \"ignoring any remaining sanity check failures\")\n\t})\n}\n\ntype errorfobserver struct {\n\tlogs []string\n}\n\nfunc (o *errorfobserver) Errorf(format string, args ...interface{}) {\n\to.logs = append(o.logs, fmt.Sprintf(format, args...))\n}\n"
  },
  {
    "path": "common/quotas/interfaces.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package=$GOPACKAGE -destination=limiter_mock.go github.com/uber/cadence/common/quotas Limiter\n//go:generate mockgen -package=$GOPACKAGE -destination=policy_mock.go github.com/uber/cadence/common/quotas Policy\n\npackage quotas\n\nimport (\n\t\"context\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\n// RPSFunc returns a float64 as the RPS\ntype RPSFunc func() float64\n\n// RPSKeyFunc returns a float64 as the RPS for the given key\ntype RPSKeyFunc func(key string) float64\n\n// Info corresponds to information required to determine rate limits\ntype Info struct {\n\tDomain string\n}\n\n// Limiter corresponds to basic rate limiting functionality.\n//\n// TODO: This can likely be replaced with clock.Ratelimiter, now that it exists,\n// but it is being left as a read-only mirror for now as only these methods are\n// currently needed in areas that currently use this Limiter.\ntype Limiter interface {\n\t// Allow attempts to allow a request to go through. The method returns\n\t// immediately with a true or false indicating if the request can make\n\t// progress\n\tAllow() bool\n\n\t// Wait waits till the deadline for a rate limit token to allow the request\n\t// to go through.\n\tWait(ctx context.Context) error\n\n\t// Reserve reserves a rate limit token\n\tReserve() clock.Reservation\n\n\t// Limit returns the current configured ratelimit.\n\t//\n\t// If this Limiter wraps multiple values, this is generally the \"most relevant\" one,\n\t// i.e. the one that is most likely to apply to the next request\n\tLimit() rate.Limit\n}\n\n// Policy corresponds to a quota policy. A policy allows implementing layered\n// and more complex rate limiting functionality.\ntype Policy interface {\n\t// Allow attempts to allow a request to go through. The method returns\n\t// immediately with a true or false indicating if the request can make\n\t// progress\n\tAllow(info Info) bool\n\n\t// Wait waits up till the context deadline for a rate limit token to allow\n\t// the request to go through. Returns nil if request is allowed.\n\tWait(ctx context.Context, info Info) error\n}\n"
  },
  {
    "path": "common/quotas/limiter_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/quotas (interfaces: Limiter)\n//\n// Generated by this command:\n//\n//\tmockgen -package=quotas -destination=limiter_mock.go github.com/uber/cadence/common/quotas Limiter\n//\n\n// Package quotas is a generated GoMock package.\npackage quotas\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\trate \"golang.org/x/time/rate\"\n\n\tclock \"github.com/uber/cadence/common/clock\"\n)\n\n// MockLimiter is a mock of Limiter interface.\ntype MockLimiter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockLimiterMockRecorder\n\tisgomock struct{}\n}\n\n// MockLimiterMockRecorder is the mock recorder for MockLimiter.\ntype MockLimiterMockRecorder struct {\n\tmock *MockLimiter\n}\n\n// NewMockLimiter creates a new mock instance.\nfunc NewMockLimiter(ctrl *gomock.Controller) *MockLimiter {\n\tmock := &MockLimiter{ctrl: ctrl}\n\tmock.recorder = &MockLimiterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockLimiter) EXPECT() *MockLimiterMockRecorder {\n\treturn m.recorder\n}\n\n// Allow mocks base method.\nfunc (m *MockLimiter) Allow() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Allow\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Allow indicates an expected call of Allow.\nfunc (mr *MockLimiterMockRecorder) Allow() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Allow\", reflect.TypeOf((*MockLimiter)(nil).Allow))\n}\n\n// Limit mocks base method.\nfunc (m *MockLimiter) Limit() rate.Limit {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Limit\")\n\tret0, _ := ret[0].(rate.Limit)\n\treturn ret0\n}\n\n// Limit indicates an expected call of Limit.\nfunc (mr *MockLimiterMockRecorder) Limit() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Limit\", reflect.TypeOf((*MockLimiter)(nil).Limit))\n}\n\n// Reserve mocks base method.\nfunc (m *MockLimiter) Reserve() clock.Reservation {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Reserve\")\n\tret0, _ := ret[0].(clock.Reservation)\n\treturn ret0\n}\n\n// Reserve indicates an expected call of Reserve.\nfunc (mr *MockLimiterMockRecorder) Reserve() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Reserve\", reflect.TypeOf((*MockLimiter)(nil).Reserve))\n}\n\n// Wait mocks base method.\nfunc (m *MockLimiter) Wait(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Wait\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Wait indicates an expected call of Wait.\nfunc (mr *MockLimiterMockRecorder) Wait(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Wait\", reflect.TypeOf((*MockLimiter)(nil).Wait), ctx)\n}\n"
  },
  {
    "path": "common/quotas/limiter_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage quotas\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nconst (\n\tdefaultRps    = 2000\n\tdefaultDomain = \"test\"\n)\n\nfunc TestMultiStageRateLimiterBlockedByDomainRps(t *testing.T) {\n\tt.Parallel()\n\tpolicy := newFixedRpsMultiStageRateLimiter(t, 2, 1)\n\tcheck := func(suffix string) {\n\t\tassert.True(t, policy.Allow(Info{Domain: defaultDomain}), \"first should work\"+suffix)\n\t\tassert.False(t, policy.Allow(Info{Domain: defaultDomain}), \"second should be limited\"+suffix) // smaller local limit applies\n\t\tassert.False(t, policy.Allow(Info{Domain: defaultDomain}), \"third should be limited\"+suffix)\n\t}\n\n\tcheck(\"\")\n\t// allow bucket to refill\n\ttime.Sleep(time.Second)\n\tcheck(\" after refresh\")\n}\n\nfunc TestMultiStageRateLimiterBlockedByGlobalRps(t *testing.T) {\n\tt.Parallel()\n\tpolicy := newFixedRpsMultiStageRateLimiter(t, 1, 2)\n\tcheck := func(suffix string) {\n\t\tassert.True(t, policy.Allow(Info{Domain: defaultDomain}), \"first should work\"+suffix)\n\t\tassert.False(t, policy.Allow(Info{Domain: defaultDomain}), \"second should be limited\"+suffix) // smaller global limit applies\n\t\tassert.False(t, policy.Allow(Info{Domain: defaultDomain}), \"third should be limited\"+suffix)\n\t}\n\n\tcheck(\"\")\n\t// allow bucket to refill\n\ttime.Sleep(time.Second)\n\tcheck(\" after refill\")\n}\n\nfunc TestMultiStageRateLimitingMultipleDomains(t *testing.T) {\n\tt.Parallel()\n\tpolicy := newFixedRpsMultiStageRateLimiter(t, 2, 1) // should allow 1/s per domain, 2/s total\n\n\tcheck := func(suffix string) {\n\t\tassert.True(t, policy.Allow(Info{Domain: \"one\"}), \"1:1 should work\"+suffix)\n\t\tassert.False(t, policy.Allow(Info{Domain: \"one\"}), \"1:2 should be limited\"+suffix) // per domain limited\n\n\t\tassert.True(t, policy.Allow(Info{Domain: \"two\"}), \"2:1 should work\"+suffix)\n\t\tassert.False(t, policy.Allow(Info{Domain: \"two\"}), \"2:2 should be limited\"+suffix) // per domain limited + global limited\n\n\t\t// third domain should be entirely cut off by global and cannot perform any requests\n\t\tassert.False(t, policy.Allow(Info{Domain: \"three\"}), \"3:1 should be limited\"+suffix) // allowed by domain, but limited by global\n\t}\n\n\tcheck(\"\")\n\t// allow bucket to refill\n\ttime.Sleep(time.Second)\n\tcheck(\" after refill\")\n}\n\nfunc TestMultiStageRateLimiterWait(t *testing.T) {\n\tt.Parallel()\n\tctx := context.Background()\n\n\tcases := []struct {\n\t\tname        string\n\t\tpolicy      Policy\n\t\tinfo        Info\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tname:        \"both allow\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 2, 1),\n\t\t\tinfo:        Info{Domain: defaultDomain},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"global blocks\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 0, 1),\n\t\t\tinfo:        Info{Domain: defaultDomain},\n\t\t\texpectedErr: clock.ErrCannotWait,\n\t\t},\n\t\t{\n\t\t\tname:        \"domain blocks\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 2, 0),\n\t\t\tinfo:        Info{Domain: defaultDomain},\n\t\t\texpectedErr: clock.ErrCannotWait,\n\t\t},\n\t\t{\n\t\t\tname:        \"both block\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 0, 0),\n\t\t\tinfo:        Info{Domain: defaultDomain},\n\t\t\texpectedErr: clock.ErrCannotWait,\n\t\t},\n\t\t{\n\t\t\tname:        \"empty domain uses global only - allow\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 1, 0),\n\t\t\tinfo:        Info{Domain: \"\"},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"empty domain uses global only - block\",\n\t\t\tpolicy:      newFixedRpsMultiStageRateLimiter(t, 0, 0),\n\t\t\tinfo:        Info{Domain: \"\"},\n\t\t\texpectedErr: clock.ErrCannotWait,\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpolicy := tc.policy\n\t\t\terr := policy.Wait(ctx, tc.info)\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.ErrorIs(t, err, tc.expectedErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc BenchmarkMultiStageRateLimiter(b *testing.B) {\n\tpolicy := newFixedRpsMultiStageRateLimiter(b, defaultRps, defaultRps)\n\tfor n := 0; n < b.N; n++ {\n\t\tpolicy.Allow(Info{Domain: defaultDomain})\n\t}\n}\n\nfunc BenchmarkMultiStageRateLimiter20Domains(b *testing.B) {\n\tnumDomains := 20\n\tpolicy := newFixedRpsMultiStageRateLimiter(b, defaultRps, defaultRps)\n\tdomains := getDomains(numDomains)\n\tfor n := 0; n < b.N; n++ {\n\t\tpolicy.Allow(Info{Domain: domains[n%numDomains]})\n\t}\n}\n\nfunc BenchmarkMultiStageRateLimiter100Domains(b *testing.B) {\n\tnumDomains := 100\n\tpolicy := newFixedRpsMultiStageRateLimiter(b, defaultRps, defaultRps)\n\tdomains := getDomains(numDomains)\n\tfor n := 0; n < b.N; n++ {\n\t\tpolicy.Allow(Info{Domain: domains[n%numDomains]})\n\t}\n}\n\nfunc BenchmarkMultiStageRateLimiter1000Domains(b *testing.B) {\n\tnumDomains := 1000\n\tpolicy := newFixedRpsMultiStageRateLimiter(b, defaultRps, defaultRps)\n\tdomains := getDomains(numDomains)\n\tfor n := 0; n < b.N; n++ {\n\t\tpolicy.Allow(Info{Domain: domains[n%numDomains]})\n\t}\n}\n\nfunc TestDynamicRateLimiter_RepeatedCalls(t *testing.T) {\n\tttl := time.Second\n\tmockTime := clock.NewMockedTimeSource()\n\t// Use the number of times the function has been called as a counter and ensure that we make no extra calls\n\trps := 0\n\trpsFunc := func() float64 {\n\t\tres := rps\n\t\trps = rps + 1\n\t\treturn float64(res)\n\t}\n\tlimiter := NewDynamicRateLimiterWithOpts(rpsFunc, DynamicRateLimiterOpts{\n\t\tTTL:        time.Second,\n\t\tMinBurst:   0,\n\t\tTimeSource: mockTime,\n\t})\n\tassert.Equal(t, limiter.Limit(), rate.Limit(0))\n\tassert.Equal(t, limiter.Limit(), rate.Limit(0))\n\tmockTime.Advance(ttl - 1)\n\tassert.Equal(t, limiter.Limit(), rate.Limit(0))\n\tmockTime.Advance(time.Nanosecond)\n\tassert.Equal(t, limiter.Limit(), rate.Limit(1))\n\tassert.Equal(t, limiter.Limit(), rate.Limit(1))\n\t// Offset the time from the TTL and ensure it's still respected\n\tmockTime.Advance(1500 * time.Millisecond)\n\tassert.Equal(t, limiter.Limit(), rate.Limit(2))\n\tmockTime.Advance(time.Second - 1)\n\tassert.Equal(t, limiter.Limit(), rate.Limit(2))\n\tmockTime.Advance(time.Nanosecond)\n\tassert.Equal(t, limiter.Limit(), rate.Limit(3))\n}\n\nfunc TestDynamicRateLimiter_Allow(t *testing.T) {\n\tmockTime := clock.NewMockedTimeSource()\n\trps := 0\n\tlimiter := NewDynamicRateLimiterWithOpts(func() float64 {\n\t\treturn float64(rps)\n\t}, DynamicRateLimiterOpts{\n\t\tTTL:        time.Second,\n\t\tMinBurst:   1,\n\t\tTimeSource: mockTime,\n\t})\n\tassert.Equal(t, false, limiter.Allow())\n\trps = 1\n\tassert.Equal(t, false, limiter.Allow())\n\tmockTime.Advance(time.Second)\n\tassert.Equal(t, true, limiter.Allow())\n}\n\nfunc TestDynamicRateLimiter_Limit(t *testing.T) {\n\tmockTime := clock.NewMockedTimeSource()\n\trps := 0\n\tlimiter := NewDynamicRateLimiterWithOpts(func() float64 {\n\t\treturn float64(rps)\n\t}, DynamicRateLimiterOpts{\n\t\tTTL:        time.Second,\n\t\tMinBurst:   1,\n\t\tTimeSource: mockTime,\n\t})\n\tassert.Equal(t, rate.Limit(0), limiter.Limit())\n\trps = 1\n\tassert.Equal(t, rate.Limit(0), limiter.Limit())\n\tmockTime.Advance(time.Second)\n\tassert.Equal(t, rate.Limit(1), limiter.Limit())\n}\n\nfunc TestDynamicRateLimiter_Reserve(t *testing.T) {\n\tmockTime := clock.NewMockedTimeSource()\n\trps := 0\n\tlimiter := NewDynamicRateLimiterWithOpts(func() float64 {\n\t\treturn float64(rps)\n\t}, DynamicRateLimiterOpts{\n\t\tTTL:        time.Second,\n\t\tMinBurst:   1,\n\t\tTimeSource: mockTime,\n\t})\n\n\tres := limiter.Reserve()\n\tassert.Equal(t, false, res.Allow())\n\tres.Used(false)\n\n\trps = 1\n\tres = limiter.Reserve()\n\tassert.Equal(t, false, res.Allow())\n\tres.Used(false)\n\n\tmockTime.Advance(time.Second)\n\tres = limiter.Reserve()\n\tassert.Equal(t, true, res.Allow())\n\tres.Used(true)\n}\n\nfunc TestDynamicRateLimiter_Wait(t *testing.T) {\n\tctx, cancelFunc := context.WithTimeout(context.Background(), time.Second)\n\tt.Cleanup(cancelFunc)\n\tmockTime := clock.NewMockedTimeSource()\n\trps := 0\n\tlimiter := NewDynamicRateLimiterWithOpts(func() float64 {\n\t\treturn float64(rps)\n\t}, DynamicRateLimiterOpts{\n\t\tTTL:        time.Second,\n\t\tMinBurst:   1,\n\t\tTimeSource: mockTime,\n\t})\n\n\terr := limiter.Wait(ctx)\n\tassert.ErrorIs(t, err, clock.ErrCannotWait)\n\n\trps = 1\n\terr = limiter.Wait(ctx)\n\tassert.ErrorIs(t, err, clock.ErrCannotWait)\n\n\tmockTime.Advance(time.Second)\n\terr = limiter.Wait(ctx)\n\tassert.NoError(t, err)\n}\n\nfunc newFixedRpsMultiStageRateLimiter(t testing.TB, globalRps float64, domainRps int) Policy {\n\treturn NewMultiStageRateLimiter(\n\t\tNewDynamicRateLimiter(func() float64 {\n\t\t\treturn globalRps\n\t\t}),\n\t\tNewCollection(newStubFactory(t, domainRps)),\n\t)\n}\n\nfunc getDomains(n int) []string {\n\tdomains := make([]string, 0, n)\n\tfor i := 0; i < n; i++ {\n\t\tdomains = append(domains, fmt.Sprintf(\"domains%v\", i))\n\t}\n\treturn domains\n}\n\nfunc newStubFactory(t testing.TB, rps int) *stubLimiterFactory {\n\treturn &stubLimiterFactory{\n\t\tt:   t,\n\t\trps: rps,\n\t}\n}\n\ntype stubLimiterFactory struct {\n\tt   testing.TB\n\trps int\n}\n\nfunc (s *stubLimiterFactory) GetLimiter(domain string) Limiter {\n\treturn clock.NewRatelimiter(rate.Limit(s.rps), s.rps)\n}\n"
  },
  {
    "path": "common/quotas/limiterfactory_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/quotas (interfaces: LimiterFactory)\n//\n// Generated by this command:\n//\n//\tmockgen -package=quotas -destination=limiterfactory_mock.go github.com/uber/cadence/common/quotas LimiterFactory\n//\n\n// Package quotas is a generated GoMock package.\npackage quotas\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockLimiterFactory is a mock of LimiterFactory interface.\ntype MockLimiterFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockLimiterFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockLimiterFactoryMockRecorder is the mock recorder for MockLimiterFactory.\ntype MockLimiterFactoryMockRecorder struct {\n\tmock *MockLimiterFactory\n}\n\n// NewMockLimiterFactory creates a new mock instance.\nfunc NewMockLimiterFactory(ctrl *gomock.Controller) *MockLimiterFactory {\n\tmock := &MockLimiterFactory{ctrl: ctrl}\n\tmock.recorder = &MockLimiterFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockLimiterFactory) EXPECT() *MockLimiterFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// GetLimiter mocks base method.\nfunc (m *MockLimiterFactory) GetLimiter(domain string) Limiter {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLimiter\", domain)\n\tret0, _ := ret[0].(Limiter)\n\treturn ret0\n}\n\n// GetLimiter indicates an expected call of GetLimiter.\nfunc (mr *MockLimiterFactoryMockRecorder) GetLimiter(domain any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLimiter\", reflect.TypeOf((*MockLimiterFactory)(nil).GetLimiter), domain)\n}\n"
  },
  {
    "path": "common/quotas/multistageratelimiter.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage quotas\n\nimport \"context\"\n\n// MultiStageRateLimiter indicates a domain specific rate limit policy\ntype MultiStageRateLimiter struct {\n\tdomainLimiters ICollection\n\tglobalLimiter  Limiter\n}\n\n// NewMultiStageRateLimiter returns a new domain quota rate limiter. This is about\n// an order of magnitude slower than\nfunc NewMultiStageRateLimiter(global Limiter, domainLimiters ICollection) *MultiStageRateLimiter {\n\treturn &MultiStageRateLimiter{\n\t\tdomainLimiters: domainLimiters,\n\t\tglobalLimiter:  global,\n\t}\n}\n\n// Allow attempts to allow a request to go through. The method returns\n// immediately with a true or false indicating if the request can make\n// progress\nfunc (d *MultiStageRateLimiter) Allow(info Info) (allowed bool) {\n\tdomain := info.Domain\n\tif len(domain) == 0 {\n\t\treturn d.globalLimiter.Allow()\n\t}\n\n\t// take a reservation with the domain limiter first\n\trsv := d.domainLimiters.For(domain).Reserve()\n\tdefer func() {\n\t\trsv.Used(allowed) // returns the token if allowed but not used\n\t}()\n\n\tif !rsv.Allow() {\n\t\treturn false\n\t}\n\n\t// ensure that the reservation does not break the global rate limit, if it\n\t// does, cancel the reservation and do not allow to proceed.\n\tif !d.globalLimiter.Allow() {\n\t\treturn false\n\t}\n\treturn true\n}\n\n// Wait waits up till the context deadline for a rate limit token to allow the request\n// to go through. This waits on the per-domain limiter, then waits on the global limiter.\nfunc (d *MultiStageRateLimiter) Wait(ctx context.Context, info Info) error {\n\tdomain := info.Domain\n\tif len(domain) == 0 {\n\t\treturn d.globalLimiter.Wait(ctx)\n\t}\n\n\terr := d.domainLimiters.For(domain).Wait(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// A limitation in this implementation is that when domain limiter\n\t// allows but global limiter does not, we have already consumed a token\n\t// from the domain limiter and cannot return it, because Wait() doesn't\n\t// return a reservation. This should not affect throughput since\n\t// global limiter is the bottleneck in this situation.\n\treturn d.globalLimiter.Wait(ctx)\n}\n"
  },
  {
    "path": "common/quotas/permember/permember.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage permember\n\nimport (\n\t\"math\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\n// PerMember allows creating per instance RPS based on globalRPS averaged by member count for a given service.\n// If member count can not be retrieved or globalRPS is not provided it falls back to instanceRPS.\nfunc PerMember(service string, globalRPS, instanceRPS float64, resolver membership.Resolver) float64 {\n\tif globalRPS <= 0 {\n\t\treturn instanceRPS\n\t}\n\n\tmemberCount, err := resolver.MemberCount(service)\n\tif err != nil || memberCount < 1 {\n\t\treturn instanceRPS\n\t}\n\n\tavgQuota := math.Max(globalRPS/float64(memberCount), 1)\n\treturn math.Min(avgQuota, instanceRPS)\n}\n\n// NewPerMemberDynamicRateLimiterFactory creates a new LimiterFactory which creates\n// a new DynamicRateLimiter for each domain, the RPS for the DynamicRateLimiter is given\n// by the globalRPS and averaged by member count for a given service.\n// instanceRPS is used as a fallback if globalRPS is not provided.\nfunc NewPerMemberDynamicRateLimiterFactory(\n\tservice string,\n\tglobalRPS dynamicproperties.IntPropertyFnWithDomainFilter,\n\tinstanceRPS dynamicproperties.IntPropertyFnWithDomainFilter,\n\tresolver membership.Resolver,\n) quotas.LimiterFactory {\n\treturn perMemberFactory{\n\t\tservice:     service,\n\t\tglobalRPS:   globalRPS,\n\t\tinstanceRPS: instanceRPS,\n\t\tresolver:    resolver,\n\t}\n}\n\ntype perMemberFactory struct {\n\tservice     string\n\tglobalRPS   dynamicproperties.IntPropertyFnWithDomainFilter\n\tinstanceRPS dynamicproperties.IntPropertyFnWithDomainFilter\n\tresolver    membership.Resolver\n}\n\nfunc (f perMemberFactory) GetLimiter(domain string) quotas.Limiter {\n\treturn quotas.NewDynamicRateLimiter(func() float64 {\n\t\treturn PerMember(\n\t\t\tf.service,\n\t\t\tfloat64(f.globalRPS(domain)),\n\t\t\tfloat64(f.instanceRPS(domain)),\n\t\t\tf.resolver,\n\t\t)\n\t})\n}\n"
  },
  {
    "path": "common/quotas/permember/permember_test.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage permember\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/membership\"\n)\n\nfunc Test_PerMember(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tresolver := membership.NewMockResolver(ctrl)\n\tresolver.EXPECT().MemberCount(\"A\").Return(10, nil).MinTimes(1)\n\tresolver.EXPECT().MemberCount(\"X\").Return(0, assert.AnError).MinTimes(1)\n\tresolver.EXPECT().MemberCount(\"Y\").Return(0, nil).MinTimes(1)\n\n\t// Invalid service - fallback to instanceRPS\n\tassert.Equal(t, 3.0, PerMember(\"X\", 20.0, 3.0, resolver))\n\n\t// Invalid member count - fallback to instanceRPS\n\tassert.Equal(t, 3.0, PerMember(\"Y\", 20.0, 3.0, resolver))\n\n\t// GlobalRPS not provided - fallback to instanceRPS\n\tassert.Equal(t, 3.0, PerMember(\"A\", 0, 3.0, resolver))\n\n\t// Calculate average per member RPS (prefer averaged global - lower)\n\tassert.Equal(t, 2.0, PerMember(\"A\", 20.0, 3.0, resolver))\n\n\t// Calculate average per member RPS (prefer instanceRPS - lower)\n\tassert.Equal(t, 3.0, PerMember(\"A\", 100.0, 3.0, resolver))\n}\n\nfunc Test_PerMemberFactory(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tresolver := membership.NewMockResolver(ctrl)\n\tresolver.EXPECT().MemberCount(\"A\").Return(10, nil).MinTimes(1)\n\n\tfactory := NewPerMemberDynamicRateLimiterFactory(\n\t\t\"A\",\n\t\tfunc(string) int { return 20 },\n\t\tfunc(string) int { return 3 },\n\t\tresolver,\n\t)\n\n\tlimiter := factory.GetLimiter(\"TestDomainName\")\n\n\t// The limit is 20 and there are 10 instances, so the per member limit is 2\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, true, limiter.Allow())\n\tassert.Equal(t, false, limiter.Allow())\n}\n"
  },
  {
    "path": "common/quotas/policy_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/common/quotas (interfaces: Policy)\n//\n// Generated by this command:\n//\n//\tmockgen -package=quotas -destination=policy_mock.go github.com/uber/cadence/common/quotas Policy\n//\n\n// Package quotas is a generated GoMock package.\npackage quotas\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPolicy is a mock of Policy interface.\ntype MockPolicy struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPolicyMockRecorder\n\tisgomock struct{}\n}\n\n// MockPolicyMockRecorder is the mock recorder for MockPolicy.\ntype MockPolicyMockRecorder struct {\n\tmock *MockPolicy\n}\n\n// NewMockPolicy creates a new mock instance.\nfunc NewMockPolicy(ctrl *gomock.Controller) *MockPolicy {\n\tmock := &MockPolicy{ctrl: ctrl}\n\tmock.recorder = &MockPolicyMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPolicy) EXPECT() *MockPolicyMockRecorder {\n\treturn m.recorder\n}\n\n// Allow mocks base method.\nfunc (m *MockPolicy) Allow(info Info) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Allow\", info)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Allow indicates an expected call of Allow.\nfunc (mr *MockPolicyMockRecorder) Allow(info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Allow\", reflect.TypeOf((*MockPolicy)(nil).Allow), info)\n}\n\n// Wait mocks base method.\nfunc (m *MockPolicy) Wait(ctx context.Context, info Info) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Wait\", ctx, info)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Wait indicates an expected call of Wait.\nfunc (mr *MockPolicyMockRecorder) Wait(ctx, info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Wait\", reflect.TypeOf((*MockPolicy)(nil).Wait), ctx, info)\n}\n"
  },
  {
    "path": "common/rangeiter/dynamic_config_linear_iterator.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rangeiter\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// DynamicConfigLinearIterator is a LinearIterator that can be reconfigured dynamically.\n// It is safe for concurrent use. Every operation checks the values of min, max, and stepCount in dynamic config\n// and refreshes the iterator if they have changed.\ntype DynamicConfigLinearIterator[T Integer] struct {\n\tmx     sync.Mutex\n\tlogger log.Logger\n\n\titer    *LinearIterator[T]\n\tcurSpec *linearIteratorSpec[T]\n\n\tminFn       func() T\n\tmaxFn       func() T\n\tstepCountFn func() int\n}\n\n// NewDynamicConfigLinearIterator creates a new DynamicConfigLinearIterator.\nfunc NewDynamicConfigLinearIterator[T Integer](min, max func() T, stepCount func() int, logger log.Logger) *DynamicConfigLinearIterator[T] {\n\titer := &DynamicConfigLinearIterator[T]{\n\t\tminFn:       min,\n\t\tmaxFn:       max,\n\t\tstepCountFn: stepCount,\n\t\tlogger:      logger,\n\t}\n\titer.curSpec = iter.newLinearIteratorSpec()\n\titer.iter = iter.curSpec.newIterator()\n\n\titer.logger.Info(\"Dynamic config linear iterator initialized\",\n\t\ttag.DynamicConfigLinearIteratorSpec(iter.curSpec),\n\t)\n\treturn iter\n}\n\n// refreshIterator checks if the spec has changed and refreshes the iterator if it has.\nfunc (d *DynamicConfigLinearIterator[T]) refreshIterator() {\n\tnewSpec := d.newLinearIteratorSpec()\n\tif newSpec.equal(d.curSpec) {\n\t\treturn\n\t}\n\n\td.logger.Info(\"Dynamic config linear iterator spec has changed, refreshing iterator, applying new spec\",\n\t\ttag.DynamicConfigLinearIteratorSpec(newSpec),\n\t)\n\n\td.curSpec = newSpec\n\td.iter = newSpec.newIterator()\n}\n\n// newLinearIteratorSpec creates a new linearIteratorSpec from the current dynamic config.\nfunc (d *DynamicConfigLinearIterator[T]) newLinearIteratorSpec() *linearIteratorSpec[T] {\n\treturn &linearIteratorSpec[T]{\n\t\tmin:       d.minFn(),\n\t\tmax:       d.maxFn(),\n\t\tstepCount: d.stepCountFn(),\n\t}\n}\n\n// Next returns the next value in the iterator.\nfunc (d *DynamicConfigLinearIterator[T]) Next() T {\n\td.mx.Lock()\n\tdefer d.mx.Unlock()\n\n\td.refreshIterator()\n\treturn d.iter.Next()\n}\n\n// Previous returns the previous value in the iterator.\nfunc (d *DynamicConfigLinearIterator[T]) Previous() T {\n\td.mx.Lock()\n\tdefer d.mx.Unlock()\n\n\td.refreshIterator()\n\treturn d.iter.Previous()\n}\n\n// Value returns the current value in the iterator.\nfunc (d *DynamicConfigLinearIterator[T]) Value() T {\n\td.mx.Lock()\n\tdefer d.mx.Unlock()\n\n\td.refreshIterator()\n\treturn d.iter.Value()\n}\n\n// Reset resets the iterator to the beginning.\nfunc (d *DynamicConfigLinearIterator[T]) Reset() {\n\td.mx.Lock()\n\tdefer d.mx.Unlock()\n\n\td.refreshIterator()\n\td.iter.Reset()\n}\n\n// linearIteratorSpec is a specification for a LinearIterator\ntype linearIteratorSpec[T Integer] struct {\n\tmin, max  T\n\tstepCount int\n}\n\n// String returns a string representation of the spec.\nfunc (s *linearIteratorSpec[T]) String() string {\n\treturn fmt.Sprintf(\"min: %v, max: %v, stepCount: %v\", s.min, s.max, s.stepCount)\n}\n\n// newIterator creates a new LinearIterator from the spec.\nfunc (s *linearIteratorSpec[T]) newIterator() *LinearIterator[T] {\n\treturn NewLinearIterator(s.min, s.max, s.stepCount)\n}\n\n// equal returns true if the spec is equal to another spec.\nfunc (s *linearIteratorSpec[T]) equal(b *linearIteratorSpec[T]) bool {\n\treturn s.min == b.min && s.max == b.max && s.stepCount == b.stepCount\n}\n"
  },
  {
    "path": "common/rangeiter/dynamic_config_linear_iterator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rangeiter\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\nfunc TestDynamicConfigLinearIterator_Next(t *testing.T) {\n\tvar (\n\t\tmin       = 10\n\t\tmax       = 20\n\t\tstepCount = 3 // range [10, 15, 20]\n\n\t\tminFn       = func() int { return min }\n\t\tmaxFn       = func() int { return max }\n\t\tstepCountFn = func() int { return stepCount }\n\t)\n\n\titer := NewDynamicConfigLinearIterator(minFn, maxFn, stepCountFn, testlogger.New(t))\n\n\tt.Run(\"initialized values\", func(t *testing.T) {\n\t\t// Start value - 15, next values - 20, 20\n\t\tassert.Equal(t, 20, iter.Next())\n\t\tassert.Equal(t, 20, iter.Next())\n\t})\n\n\tt.Run(\"changed max value\", func(t *testing.T) {\n\t\tmax = 30\n\t\t// range [10, 20, 30]\n\t\t// Start value - 20, next value - 30\n\t\tassert.Equal(t, 30, iter.Next())\n\t})\n\n\tt.Run(\"changed step count value\", func(t *testing.T) {\n\t\tstepCount = 6\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\t// Start value - 18\n\t\tassert.Equal(t, 22, iter.Next())\n\t\tassert.Equal(t, 26, iter.Next())\n\t\tassert.Equal(t, 30, iter.Next())\n\t})\n}\n\nfunc TestDynamicConfigLinearIterator_Previous(t *testing.T) {\n\tvar (\n\t\tmin       = 10\n\t\tmax       = 20\n\t\tstepCount = 3 // range [10, 15, 20]\n\n\t\tminFn       = func() int { return min }\n\t\tmaxFn       = func() int { return max }\n\t\tstepCountFn = func() int { return stepCount }\n\t)\n\n\titer := NewDynamicConfigLinearIterator(minFn, maxFn, stepCountFn, testlogger.New(t))\n\n\tt.Run(\"initialized values\", func(t *testing.T) {\n\t\t// Start value - 15, previous values - 10, 10\n\t\tassert.Equal(t, 10, iter.Previous())\n\t\tassert.Equal(t, 10, iter.Previous())\n\t})\n\n\tt.Run(\"changed max value\", func(t *testing.T) {\n\t\tmax = 30\n\t\t// range [10, 20, 30]\n\t\t// Start value - 20, previous value - 10\n\t\tassert.Equal(t, 10, iter.Previous())\n\t})\n\n\tt.Run(\"changed step count value\", func(t *testing.T) {\n\t\tstepCount = 6\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\t// Start value - 18\n\t\tassert.Equal(t, 14, iter.Previous())\n\t\tassert.Equal(t, 10, iter.Previous())\n\t\tassert.Equal(t, 10, iter.Previous())\n\t})\n}\n\nfunc TestDynamicConfigLinearIterator_Value(t *testing.T) {\n\tvar (\n\t\tmin       = 10\n\t\tmax       = 20\n\t\tstepCount = 3 // range [10, 15, 20]\n\n\t\tminFn       = func() int { return min }\n\t\tmaxFn       = func() int { return max }\n\t\tstepCountFn = func() int { return stepCount }\n\t)\n\n\titer := NewDynamicConfigLinearIterator(minFn, maxFn, stepCountFn, testlogger.New(t))\n\n\tt.Run(\"initialized values\", func(t *testing.T) {\n\t\t// range [10, 15, 20]\n\t\tassert.Equal(t, 15, iter.Value())\n\t})\n\n\tt.Run(\"changed max value\", func(t *testing.T) {\n\t\tmax = 30\n\t\t// range [10, 20, 30]\n\t\tassert.Equal(t, 20, iter.Value())\n\t})\n\n\tt.Run(\"changed step count value\", func(t *testing.T) {\n\t\tstepCount = 6\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\tassert.Equal(t, 18, iter.Value())\n\t})\n}\n\nfunc TestDynamicConfigLinearIterator_Reset(t *testing.T) {\n\tvar (\n\t\tmin       = 10\n\t\tmax       = 20\n\t\tstepCount = 3 // range [10, 15, 20]\n\n\t\tminFn       = func() int { return min }\n\t\tmaxFn       = func() int { return max }\n\t\tstepCountFn = func() int { return stepCount }\n\t)\n\n\titer := NewDynamicConfigLinearIterator(minFn, maxFn, stepCountFn, testlogger.New(t))\n\n\tt.Run(\"initialized values\", func(t *testing.T) {\n\t\t// range [10, 15, 20]\n\t\tassert.Equal(t, 15, iter.Value())\n\t\titer.Reset()\n\t\tassert.Equal(t, 15, iter.Value())\n\t})\n\n\tt.Run(\"changed max value\", func(t *testing.T) {\n\t\tmax = 30\n\t\t// range [10, 20, 30]\n\t\tassert.Equal(t, 20, iter.Value())\n\t\titer.Reset()\n\t\tassert.Equal(t, 20, iter.Value())\n\t})\n\n\tt.Run(\"changed step count value\", func(t *testing.T) {\n\t\tstepCount = 6\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\tassert.Equal(t, 18, iter.Value())\n\t\titer.Reset()\n\t\tassert.Equal(t, 18, iter.Value())\n\t})\n\n\tt.Run(\"reached max and config not changed\", func(t *testing.T) {\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\tassert.Equal(t, 22, iter.Next())\n\t\tassert.Equal(t, 26, iter.Next())\n\t\tassert.Equal(t, 30, iter.Next())\n\n\t\titer.Reset()\n\t\tassert.Equal(t, 18, iter.Value())\n\t})\n\n\tt.Run(\"reached min and config not changed\", func(t *testing.T) {\n\t\t// range [10, 14, 18, 22, 26, 30]\n\t\tassert.Equal(t, 14, iter.Previous())\n\t\tassert.Equal(t, 10, iter.Previous())\n\n\t\titer.Reset()\n\t\tassert.Equal(t, 18, iter.Value())\n\t})\n}\n"
  },
  {
    "path": "common/rangeiter/iterator.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rangeiter\n\nimport \"golang.org/x/exp/constraints\"\n\n// Iterator is an interface for iterating through a range between two values.\n// The range is a range of integers from a minimum to a maximum value (inclusive).\ntype Iterator[T Integer] interface {\n\t// Next returns the next value closer to the max value in the range\n\t// If the current value is the max value, Next will return the max value\n\tNext() T\n\n\t// Previous returns the previous value closer to the min value in the range\n\t// If the current value is the min value, Previous will return the min value\n\tPrevious() T\n\n\t// Value returns the current value in the range\n\tValue() T\n\n\t// Reset resets the Iterator to its initial state\n\tReset()\n}\n\n// Integer is a type constraint for the Iterator interface to ensure that only integer types are used.\ntype Integer = constraints.Integer\n"
  },
  {
    "path": "common/rangeiter/linear_iterator.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rangeiter\n\n// LinearIterator is an Iterator that steps through a range linearly with a fixed step size.\n// The fixed step size is calculated based on the range and the number of steps provided.\ntype LinearIterator[T Integer] struct {\n\tcur   int\n\tsteps []T\n}\n\n// NewLinearIterator returns a new LinearIterator.\n// The range is a range of integers from min to max. The stepCount is the number of steps between min and max (both inclusive).\n// After creating the Iterator, the current value will be the middle value of the range.\n// If stepCount is less or equal than 2 or the difference between min and max is less than 1, the Iterator will only have min and max.\n//   - range [0, 10], \tstepCount 1 -> [0, 10]\n//   - range [10, 10], \tstepCount 2 -> [10, 10]\n//\n// If the difference between min and max is less than stepCount, the Iterator will have all the values between min and max.\n//   - range [0, 5], \tstepCount 10 -> [0, 1, 2, 3, 4, 5]\n//   - range [0, 10], \tstepCount 11 -> [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]\nfunc NewLinearIterator[T Integer](min, max T, stepCount int) *LinearIterator[T] {\n\t// Ensure min <= max\n\tif min > max {\n\t\tmin, max = max, min\n\t}\n\n\ttotalRange := int(max - min)\n\n\t// If stepCount is less than 2 or the difference between min and max is less than 1,\n\t// return a Iterator with only min and max\n\tif stepCount <= 2 || totalRange <= 1 {\n\t\treturn &LinearIterator[T]{\n\t\t\tcur:   0,\n\t\t\tsteps: []T{min, max},\n\t\t}\n\t}\n\n\t// if the total range is less than the step count, return a Iterator with all the values\n\tif totalRange < stepCount {\n\t\tsteps := make([]T, totalRange+1)\n\n\t\tsteps[0], steps[totalRange] = min, max\n\n\t\tfor i := 1; i < totalRange; i++ {\n\t\t\tsteps[i] = min + T(i)\n\t\t}\n\n\t\treturn &LinearIterator[T]{\n\t\t\tcur:   (len(steps) - 1) / 2,\n\t\t\tsteps: steps,\n\t\t}\n\t}\n\n\t// Calculate step size as a float to ensure even distribution\n\tstepSize := float64(totalRange / (stepCount - 1))\n\n\t// Generate steps\n\tsteps := make([]T, stepCount)\n\tfor i := 0; i < stepCount; i++ {\n\t\tsteps[i] = min + T(float64(i)*stepSize)\n\t}\n\n\t// Ensure the last value is exactly max (to avoid floating-point errors)\n\tsteps[stepCount-1] = max\n\n\treturn &LinearIterator[T]{\n\t\tcur:   (len(steps) - 1) / 2,\n\t\tsteps: steps,\n\t}\n}\n\n// Next returns the next value in the range.\nfunc (s *LinearIterator[T]) Next() T {\n\tif s.cur < len(s.steps)-1 {\n\t\ts.cur++\n\t}\n\treturn s.steps[s.cur]\n}\n\n// Previous returns the previous value in the range.\nfunc (s *LinearIterator[T]) Previous() T {\n\tif s.cur > 0 {\n\t\ts.cur--\n\t}\n\treturn s.steps[s.cur]\n}\n\n// Value returns the current value in the range.\nfunc (s *LinearIterator[T]) Value() T {\n\treturn s.steps[s.cur]\n}\n\n// Reset resets the Iterator to the beginning of the range.\nfunc (s *LinearIterator[T]) Reset() {\n\ts.cur = (len(s.steps) - 1) / 2\n}\n"
  },
  {
    "path": "common/rangeiter/linear_iterator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rangeiter\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestNewLinearIterator(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tmin, max  int\n\t\tstepCount int\n\t\twant      *LinearIterator[int]\n\t}{\n\t\t\"min greater max, stepCount greater range\": {\n\t\t\tmin:       10,\n\t\t\tmax:       1,\n\t\t\tstepCount: 10, // range: 10 - 1 = 9\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   4,\n\t\t\t\tsteps: []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, stepCount is one\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 1,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   0,\n\t\t\t\tsteps: []int{0, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, stepCount is two\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 2,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   0,\n\t\t\t\tsteps: []int{0, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, stepCount is three\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 3,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   1,\n\t\t\t\tsteps: []int{0, 5, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, stepCount is four\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 4,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   1,\n\t\t\t\tsteps: []int{0, 3, 6, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, stepCount is six\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 6,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   2,\n\t\t\t\tsteps: []int{0, 2, 4, 6, 8, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min than 1, stepCount is greater range\": {\n\t\t\tmin:       9,\n\t\t\tmax:       10,\n\t\t\tstepCount: 10,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   0,\n\t\t\t\tsteps: []int{9, 10},\n\t\t\t},\n\t\t},\n\t\t\"max equal mim, stepCount is greater range\": {\n\t\t\tmin:       10,\n\t\t\tmax:       10,\n\t\t\tstepCount: 10,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   0,\n\t\t\t\tsteps: []int{10, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, range is less than stepCount\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 20, // range: 10 - 0 = 10\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   5,\n\t\t\t\tsteps: []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10},\n\t\t\t},\n\t\t},\n\t\t\"max greater min, range is equal to stepCount\": {\n\t\t\tmin:       0,\n\t\t\tmax:       10,\n\t\t\tstepCount: 10, // range: 10 - 0 = 10\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   4,\n\t\t\t\tsteps: []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 10},\n\t\t\t},\n\t\t},\n\t\t\"negative max and min, stepCount is two\": {\n\t\t\tmin:       -1000 * 1000,\n\t\t\tmax:       -1000,\n\t\t\tstepCount: 35,\n\t\t\twant: &LinearIterator[int]{\n\t\t\t\tcur:   17,\n\t\t\t\tsteps: []int{-1000000, -970618, -941236, -911854, -882472, -853090, -823708, -794326, -764944, -735562, -706180, -676798, -647416, -618034, -588652, -559270, -529888, -500506, -471124, -441742, -412360, -382978, -353596, -324214, -294832, -265450, -236068, -206686, -177304, -147922, -118540, -89158, -59776, -30394, -1000},\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := NewLinearIterator(c.min, c.max, c.stepCount)\n\t\t\tassert.Equal(t, c.want, got)\n\t\t})\n\t}\n}\n\nfunc TestLinearIterator_Next(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcur   int\n\t\tsteps []int\n\n\t\twantValue int\n\t\twantCur   int\n\t}{\n\t\t\"cur equals to min\": {\n\t\t\tcur:   0,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 1,\n\t\t\twantCur:   1,\n\t\t},\n\t\t\"cur equals to middle\": {\n\t\t\tcur:   3, // value - 3\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 4,\n\t\t\twantCur:   4,\n\t\t},\n\t\t\"cur equals to max\": {\n\t\t\tcur:   5,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 5,\n\t\t\twantCur:   5,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\ts := &LinearIterator[int]{\n\t\t\t\tcur:   c.cur,\n\t\t\t\tsteps: c.steps,\n\t\t\t}\n\t\t\tassert.Equal(t, c.wantValue, s.Next())\n\t\t\tassert.Equal(t, c.wantCur, s.cur)\n\t\t})\n\t}\n}\n\nfunc TestLinearIterator_Previous(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcur   int\n\t\tsteps []int\n\n\t\twantValue int\n\t\twantCur   int\n\t}{\n\t\t\"cur equals to min\": {\n\t\t\tcur:   0,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 0,\n\t\t\twantCur:   0,\n\t\t},\n\t\t\"cur equals to middle\": {\n\t\t\tcur:   3, // value - 3\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 2,\n\t\t\twantCur:   2,\n\t\t},\n\t\t\"cur equals to max\": {\n\t\t\tcur:   6,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 5,\n\t\t\twantCur:   5,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\ts := &LinearIterator[int]{\n\t\t\t\tcur:   c.cur,\n\t\t\t\tsteps: c.steps,\n\t\t\t}\n\t\t\tassert.Equal(t, c.wantValue, s.Previous())\n\t\t\tassert.Equal(t, c.wantCur, s.cur)\n\t\t})\n\t}\n}\n\nfunc TestLinearIterator_Reset(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcur   int\n\t\tsteps []int\n\n\t\twantValue int\n\t\twantCur   int\n\t}{\n\t\t\"cur equals to min\": {\n\t\t\tcur:   0,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 2,\n\t\t\twantCur:   2,\n\t\t},\n\t\t\"cur equals to middle\": {\n\t\t\tcur:   3, // value - 3\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 2,\n\t\t\twantCur:   2,\n\t\t},\n\t\t\"cur equals to max\": {\n\t\t\tcur:   6,\n\t\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\n\t\t\twantValue: 2,\n\t\t\twantCur:   2,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\ts := &LinearIterator[int]{\n\t\t\t\tcur:   c.cur,\n\t\t\t\tsteps: c.steps,\n\t\t\t}\n\n\t\t\ts.Reset()\n\t\t\tassert.Equal(t, c.wantValue, s.Value())\n\t\t\tassert.Equal(t, c.wantCur, s.cur)\n\t\t})\n\t}\n}\n\nfunc TestLinearIterator_Value(t *testing.T) {\n\tIterator := &LinearIterator[int]{\n\t\tcur:   3,\n\t\tsteps: []int{0, 1, 2, 3, 4, 5},\n\t}\n\tassert.Equal(t, 3, Iterator.Value())\n}\n"
  },
  {
    "path": "common/reconciliation/constants.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage reconciliation\n\n// Execution fixer workflow relates\n\nconst (\n\tCheckDataCorruptionWorkflowType                 = \"check-data-corruption-workflow\"\n\tCheckDataCorruptionWorkflowTaskList             = \"check-data-corruption-workflow-tl\"\n\tCheckDataCorruptionWorkflowSignalName           = \"check-data-corruption-workflow-signal\"\n\tCheckDataCorruptionWorkflowID                   = \"check-data-corruption-workflow-id\"\n\tCheckDataCorruptionWorkflowTimeoutInSeconds     = 24 * 60 * 60\n\tCheckDataCorruptionWorkflowTaskTimeoutInSeconds = 60\n)\n"
  },
  {
    "path": "common/reconciliation/entity/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage entity\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// Execution is a base type for executions which should be checked or fixed.\n\tExecution struct {\n\t\tShardID    int\n\t\tDomainID   string\n\t\tWorkflowID string\n\t\tRunID      string\n\t\tState      int\n\t}\n\n\t// ConcreteExecution is a concrete execution.\n\tConcreteExecution struct {\n\t\tBranchToken []byte\n\t\tTreeID      string\n\t\tBranchID    string\n\t\tExecution\n\t}\n\n\t// CurrentExecution is a current execution.\n\tCurrentExecution struct {\n\t\tCurrentRunID string\n\t\tExecution\n\t}\n\n\t// Timer is a timer scheduled to be fired\n\tTimer struct {\n\t\tShardID             int\n\t\tWorkflowID          string\n\t\tDomainID            string\n\t\tRunID               string\n\t\tVisibilityTimestamp time.Time\n\t\tTaskID              int64\n\t\tTaskType            int\n\t\tTimeoutType         int\n\t\tEventID             int64\n\t\tScheduleAttempt     int64\n\t\tVersion             int64\n\t}\n)\n\nfunc (t *Timer) Validate() error {\n\tif t.ShardID < 0 {\n\t\treturn fmt.Errorf(\"invalid ShardID: %v\", t.ShardID)\n\t}\n\tif len(t.DomainID) == 0 {\n\t\treturn errors.New(\"empty DomainID\")\n\t}\n\tif len(t.WorkflowID) == 0 {\n\t\treturn errors.New(\"empty WorkflowID\")\n\t}\n\tif len(t.RunID) == 0 {\n\t\treturn errors.New(\"empty RunID\")\n\t}\n\n\treturn nil\n}\n\nfunc (t *Timer) Clone() Entity {\n\treturn &Timer{}\n}\n\nfunc (t *Timer) GetShardID() int {\n\treturn t.ShardID\n}\n\nfunc (t *Timer) GetDomainID() string {\n\treturn t.DomainID\n}\n\n// ValidateExecution returns an error if Execution is not valid, nil otherwise.\nfunc validateExecution(execution *Execution) error {\n\tif execution.ShardID < 0 {\n\t\treturn fmt.Errorf(\"invalid ShardID: %v\", execution.ShardID)\n\t}\n\tif len(execution.DomainID) == 0 {\n\t\treturn errors.New(\"empty DomainID\")\n\t}\n\tif len(execution.WorkflowID) == 0 {\n\t\treturn errors.New(\"empty WorkflowID\")\n\t}\n\tif len(execution.RunID) == 0 {\n\t\treturn errors.New(\"empty RunID\")\n\t}\n\tif execution.State < persistence.WorkflowStateCreated || execution.State > persistence.WorkflowStateCorrupted {\n\t\treturn fmt.Errorf(\"unknown workflow state: %v\", execution.State)\n\t}\n\treturn nil\n}\n\n// Validate returns an error if ConcreteExecution is not valid, nil otherwise.\nfunc (ce *ConcreteExecution) Validate() error {\n\terr := validateExecution(&ce.Execution)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif len(ce.BranchToken) == 0 {\n\t\treturn errors.New(\"empty BranchToken\")\n\t}\n\tif len(ce.TreeID) == 0 {\n\t\treturn errors.New(\"empty TreeID\")\n\t}\n\tif len(ce.BranchID) == 0 {\n\t\treturn errors.New(\"empty BranchID\")\n\t}\n\treturn nil\n\n}\n\n// Validate returns an error if CurrentExecution is not valid, nil otherwise.\nfunc (curre *CurrentExecution) Validate() error {\n\terr := validateExecution(&curre.Execution)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif len(curre.CurrentRunID) == 0 {\n\t\treturn errors.New(\"empty CurrentRunID\")\n\t}\n\treturn nil\n}\n\n// Clone will return a new copy of ConcreteExecution\nfunc (ConcreteExecution) Clone() Entity {\n\treturn &ConcreteExecution{}\n}\n\n// Clone will return a new copy of CurrentExecution\nfunc (CurrentExecution) Clone() Entity {\n\treturn &CurrentExecution{}\n}\n\n// GetShardID returns shard id\nfunc (ce *ConcreteExecution) GetShardID() int {\n\treturn ce.Execution.ShardID\n}\n\n// GetShardID returns shard id\nfunc (curre *CurrentExecution) GetShardID() int {\n\treturn curre.Execution.ShardID\n}\n\n// GetDomainID returns the domain id\nfunc (ce *ConcreteExecution) GetDomainID() string {\n\treturn ce.DomainID\n}\n\n// GetDomainID returns the domain id\nfunc (curre *CurrentExecution) GetDomainID() string {\n\treturn curre.DomainID\n}\n\n// Entity allows to deserialize and validate different type of executions\ntype Entity interface {\n\tValidate() error\n\tClone() Entity\n\tGetShardID() int\n\tGetDomainID() string\n}\n"
  },
  {
    "path": "common/reconciliation/entity/types_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage entity\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nconst (\n\tdomainID   = \"test-domain-id\"\n\tworkflowID = \"test-workflow-id\"\n\trunID      = \"test-run-id\"\n\ttreeID     = \"test-tree-id\"\n\tbranchID   = \"test-branch-id\"\n)\n\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(TypeSuite))\n}\n\ntype TypeSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc (t *TypeSuite) SetupTest() {\n\tt.Assertions = require.New(t.T())\n}\n\nfunc (t *TypeSuite) TestValidateExecution() {\n\ttestCases := []struct {\n\t\texecution   *ConcreteExecution\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\texecution:   &ConcreteExecution{},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID: -1,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID: 0,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:  0,\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\tTreeID:      treeID,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tState:      persistence.WorkflowStateCreated - 1,\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\tTreeID:      treeID,\n\t\t\t\tBranchID:    branchID,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tState:      persistence.WorkflowStateCorrupted + 1,\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\tTreeID:      treeID,\n\t\t\t\tBranchID:    branchID,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\texecution: &ConcreteExecution{\n\t\t\t\tExecution: Execution{\n\t\t\t\t\tShardID:    0,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tState:      persistence.WorkflowStateCreated,\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\tTreeID:      treeID,\n\t\t\t\tBranchID:    branchID,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\terr := tc.execution.Validate()\n\t\tif tc.expectError {\n\t\t\tt.Error(err)\n\t\t} else {\n\t\t\tt.NoError(err)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/concrete.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ConcreteExecutionIterator is used to retrieve Concrete executions.\nfunc ConcreteExecutionIterator(\n\tctx context.Context,\n\tretryer persistence.Retryer,\n\tpageSize int,\n) pagination.Iterator {\n\treturn pagination.NewIterator(ctx, nil, getConcreteExecutions(retryer, pageSize, codec.NewThriftRWEncoder()))\n}\n\n// ConcreteExecution returns a single ConcreteExecution from persistence\nfunc ConcreteExecution(\n\tctx context.Context,\n\tretryer persistence.Retryer,\n\trequest ExecutionRequest,\n) (entity.Entity, error) {\n\n\treq := persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: request.DomainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: request.WorkflowID,\n\t\t\tRunID:      request.RunID,\n\t\t},\n\t\tDomainName: request.DomainName,\n\t}\n\te, err := retryer.GetWorkflowExecution(ctx, &req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tbranchToken, branch, err := getBranchToken(e.State.ExecutionInfo.BranchToken, e.State.VersionHistories, codec.NewThriftRWEncoder())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &entity.ConcreteExecution{\n\t\tBranchToken: branchToken,\n\t\tTreeID:      branch.GetTreeID(),\n\t\tBranchID:    branch.GetBranchID(),\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    retryer.GetShardID(),\n\t\t\tDomainID:   e.State.ExecutionInfo.DomainID,\n\t\t\tWorkflowID: e.State.ExecutionInfo.WorkflowID,\n\t\t\tRunID:      e.State.ExecutionInfo.RunID,\n\t\t\tState:      e.State.ExecutionInfo.State,\n\t\t},\n\t}, nil\n}\n\nfunc getConcreteExecutions(\n\tpr persistence.Retryer,\n\tpageSize int,\n\tencoder *codec.ThriftRWEncoder,\n) pagination.FetchFn {\n\treturn func(ctx context.Context, token pagination.PageToken) (pagination.Page, error) {\n\t\treq := &persistence.ListConcreteExecutionsRequest{\n\t\t\tPageSize: pageSize,\n\t\t}\n\t\tif token != nil {\n\t\t\treq.PageToken = token.([]byte)\n\t\t}\n\t\tresp, err := pr.ListConcreteExecutions(ctx, req)\n\t\tif err != nil {\n\t\t\treturn pagination.Page{}, err\n\t\t}\n\t\texecutions := make([]pagination.Entity, len(resp.Executions))\n\t\tfor i, e := range resp.Executions {\n\t\t\tbranchToken, branch, err := getBranchToken(e.ExecutionInfo.BranchToken, e.VersionHistories, encoder)\n\t\t\tif err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\t\t\tconcreteExec := &entity.ConcreteExecution{\n\t\t\t\tBranchToken: branchToken,\n\t\t\t\tTreeID:      branch.GetTreeID(),\n\t\t\t\tBranchID:    branch.GetBranchID(),\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tShardID:    pr.GetShardID(),\n\t\t\t\t\tDomainID:   e.ExecutionInfo.DomainID,\n\t\t\t\t\tWorkflowID: e.ExecutionInfo.WorkflowID,\n\t\t\t\t\tRunID:      e.ExecutionInfo.RunID,\n\t\t\t\t\tState:      e.ExecutionInfo.State,\n\t\t\t\t},\n\t\t\t}\n\t\t\tif err := concreteExec.Validate(); err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\t\t\texecutions[i] = concreteExec\n\t\t}\n\t\tvar nextToken interface{} = resp.PageToken\n\t\tif len(resp.PageToken) == 0 {\n\t\t\tnextToken = nil\n\t\t}\n\t\tpage := pagination.Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextToken,\n\t\t\tEntities:     executions,\n\t\t}\n\t\treturn page, nil\n\t}\n}\n\n// getBranchToken returns the branchToken and historyBranch, error on failure.\nfunc getBranchToken(\n\tbranchToken []byte,\n\thistories *persistence.VersionHistories,\n\tdecoder *codec.ThriftRWEncoder,\n) ([]byte, shared.HistoryBranch, error) {\n\tvar branch shared.HistoryBranch\n\tbt := branchToken\n\tif histories != nil {\n\t\tversionHistory, err := histories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, branch, err\n\t\t}\n\t\tbt = versionHistory.GetBranchToken()\n\t}\n\n\tif err := decoder.Decode(bt, &branch); err != nil {\n\t\treturn nil, branch, err\n\t}\n\n\treturn bt, branch, nil\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/concrete_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\nfunc TestConcreteExecutionIterator(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tretryer := persistence.NewMockRetryer(ctrl)\n\tretryer.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).\n\t\tReturn(&persistence.ListConcreteExecutionsResponse{}, nil).\n\t\tTimes(1)\n\n\titerator := ConcreteExecutionIterator(\n\t\tcontext.Background(),\n\t\tretryer,\n\t\t10,\n\t)\n\trequire.NotNil(t, iterator)\n}\n\nfunc TestConcreteExecution(t *testing.T) {\n\tencoder := codec.NewThriftRWEncoder()\n\ttests := []struct {\n\t\tdesc       string\n\t\treq        ExecutionRequest\n\t\tmockFn     func(retryer *persistence.MockRetryer)\n\t\twantEntity entity.Entity\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tdesc: \"success\",\n\t\t\treq: ExecutionRequest{\n\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\tDomainName: \"test-domain-name\",\n\t\t\t},\n\t\t\tmockFn: func(retryer *persistence.MockRetryer) {\n\t\t\t\tretryer.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(\n\t\t\t\t\t\t&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\t\t\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\t\t\t\t\tDomainID:    \"test-domain-id\",\n\t\t\t\t\t\t\t\t\tWorkflowID:  \"test-workflow-id\",\n\t\t\t\t\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tnil,\n\t\t\t\t\t).Times(1)\n\n\t\t\t\tretryer.EXPECT().GetShardID().Return(355).Times(1)\n\t\t\t},\n\t\t\twantEntity: &entity.ConcreteExecution{\n\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\t\tTreeID:      \"test-tree-id\",\n\t\t\t\tBranchID:    \"test-branch-id\",\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tShardID:    355,\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"GetWorkflowExecution failed\",\n\t\t\treq: ExecutionRequest{\n\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\tDomainName: \"test-domain-name\",\n\t\t\t},\n\t\t\tmockFn: func(retryer *persistence.MockRetryer) {\n\t\t\t\tretryer.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tretryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\ttc.mockFn(retryer)\n\n\t\t\tgotEntity, err := ConcreteExecution(context.Background(), retryer, tc.req)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"ConcreteExecution() err: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantEntity, gotEntity); diff != \"\" {\n\t\t\t\tt.Errorf(\"ConcreteExecution() mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetConcreteExecutions(t *testing.T) {\n\tencoder := codec.NewThriftRWEncoder()\n\ttestExecutions := []*persistence.ListConcreteExecutionsEntity{\n\t\t{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id-1\", \"test-branch-id-1\"),\n\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\tDomainID:    \"test-domain-id-1\",\n\t\t\t\tWorkflowID:  \"test-workflow-id-1\",\n\t\t\t\tRunID:       \"test-run-id-1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id-2\", \"test-branch-id-2\"),\n\t\t\t\tState:       persistence.WorkflowStateCompleted,\n\t\t\t\tDomainID:    \"test-domain-id-2\",\n\t\t\t\tWorkflowID:  \"test-workflow-id-2\",\n\t\t\t\tRunID:       \"test-run-id-2\",\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tdesc      string\n\t\tpageSize  int\n\t\tpageToken pagination.PageToken\n\t\tmockFn    func(*testing.T, *persistence.MockRetryer)\n\t\twantPage  pagination.Page\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tdesc:      \"success\",\n\t\t\tpageSize:  2,\n\t\t\tpageToken: []byte(\"test-page-token\"),\n\t\t\tmockFn: func(t *testing.T, retryer *persistence.MockRetryer) {\n\t\t\t\tretryer.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *persistence.ListConcreteExecutionsRequest) (*persistence.ListConcreteExecutionsResponse, error) {\n\t\t\t\t\t\twantReq := &persistence.ListConcreteExecutionsRequest{\n\t\t\t\t\t\t\tPageSize:  2,\n\t\t\t\t\t\t\tPageToken: []byte(\"test-page-token\"),\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, req); diff != \"\" {\n\t\t\t\t\t\t\tt.Errorf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &persistence.ListConcreteExecutionsResponse{\n\t\t\t\t\t\t\tPageToken:  []byte(\"test-next-page-token\"),\n\t\t\t\t\t\t\tExecutions: testExecutions,\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\t// will be called for each execution in the response\n\t\t\t\tretryer.EXPECT().GetShardID().Return(355).Times(2)\n\t\t\t},\n\t\t\twantPage: pagination.Page{\n\t\t\t\tCurrentToken: []byte(\"test-page-token\"),\n\t\t\t\tNextToken:    []byte(\"test-next-page-token\"),\n\t\t\t\tEntities:     concreteExecutionsToEntities(testExecutions, 355, encoder),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tretryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\ttc.mockFn(t, retryer)\n\n\t\t\tfetchFn := getConcreteExecutions(retryer, tc.pageSize, encoder)\n\t\t\tgotPage, err := fetchFn(context.Background(), tc.pageToken)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"ConcreteExecution() err: %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\tif diff := cmp.Diff(tc.wantPage, gotPage); diff != \"\" {\n\t\t\t\tt.Errorf(\"ConcreteExecution() mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetBranchToken(t *testing.T) {\n\tencoder := codec.NewThriftRWEncoder()\n\ttestCases := []struct {\n\t\tname              string\n\t\tentity            *persistence.ListConcreteExecutionsEntity\n\t\twantErr           bool\n\t\twantBranchToken   []byte\n\t\twantHistoryBranch shared.HistoryBranch\n\t}{\n\t\t{\n\t\t\tname: \"valid branch token - no version histories\",\n\t\t\tentity: &persistence.ListConcreteExecutionsEntity{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\twantHistoryBranch: shared.HistoryBranch{\n\t\t\t\tTreeID:   common.StringPtr(\"test-tree-id\"),\n\t\t\t\tBranchID: common.StringPtr(\"test-branch-id\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"valid branch token - with version histories\",\n\t\t\tentity: &persistence.ListConcreteExecutionsEntity{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{}, // this will be ignored because index is 1\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id-from-versionhistory\", \"test-branch-id-from-versionhistory\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id-from-versionhistory\", \"test-branch-id-from-versionhistory\"),\n\t\t\twantHistoryBranch: shared.HistoryBranch{\n\t\t\t\tTreeID:   common.StringPtr(\"test-tree-id-from-versionhistory\"),\n\t\t\t\tBranchID: common.StringPtr(\"test-branch-id-from-versionhistory\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"version history index out of bound\",\n\t\t\tentity: &persistence.ListConcreteExecutionsEntity{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id\", \"test-branch-id\"),\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 2,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: mustGetValidBranchToken(t, encoder, \"test-tree-id-from-versionhistory\", \"test-branch-id-from-versionhistory\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid branch token\",\n\t\t\tentity: &persistence.ListConcreteExecutionsEntity{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tBranchToken: []byte(\"invalid\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tbranchToken, branch, err := getBranchToken(\n\t\t\t\ttc.entity.ExecutionInfo.BranchToken,\n\t\t\t\ttc.entity.VersionHistories,\n\t\t\t\tencoder,\n\t\t\t)\n\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.Nil(t, branchToken)\n\t\t\t\trequire.Empty(t, branch.GetTreeID())\n\t\t\t\trequire.Empty(t, branch.GetBranchID())\n\t\t\t} else {\n\t\t\t\tif diff := cmp.Diff(tc.wantHistoryBranch, branch); diff != \"\" {\n\t\t\t\t\tt.Fatalf(\"HistoryBranch mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t}\n\t\t\t\trequire.Equal(t, tc.wantBranchToken, branchToken)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mustGetValidBranchToken(t *testing.T, encoder *codec.ThriftRWEncoder, treeID, branchID string) []byte {\n\thb := &shared.HistoryBranch{\n\t\tTreeID:   common.StringPtr(treeID),\n\t\tBranchID: common.StringPtr(branchID),\n\t}\n\tbytes, err := encoder.Encode(hb)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to encode branch token: %v\", err)\n\t}\n\n\treturn bytes\n}\n\nfunc concreteExecutionsToEntities(execs []*persistence.ListConcreteExecutionsEntity, shardID int, encoder *codec.ThriftRWEncoder) []pagination.Entity {\n\tentities := make([]pagination.Entity, len(execs))\n\tfor i, e := range execs {\n\t\tbranchToken, branch, err := getBranchToken(e.ExecutionInfo.BranchToken, e.VersionHistories, encoder)\n\t\tif err != nil {\n\t\t\treturn nil\n\t\t}\n\t\tconcreteExec := &entity.ConcreteExecution{\n\t\t\tBranchToken: branchToken,\n\t\t\tTreeID:      branch.GetTreeID(),\n\t\t\tBranchID:    branch.GetBranchID(),\n\t\t\tExecution: entity.Execution{\n\t\t\t\tShardID:    shardID,\n\t\t\t\tDomainID:   e.ExecutionInfo.DomainID,\n\t\t\t\tWorkflowID: e.ExecutionInfo.WorkflowID,\n\t\t\t\tRunID:      e.ExecutionInfo.RunID,\n\t\t\t\tState:      e.ExecutionInfo.State,\n\t\t\t},\n\t\t}\n\t\tentities[i] = concreteExec\n\t}\n\treturn entities\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/current.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\n// CurrentExecutionIterator is used to retrieve Concrete executions.\nfunc CurrentExecutionIterator(\n\tctx context.Context,\n\tretryer persistence.Retryer,\n\tpageSize int,\n) pagination.Iterator {\n\treturn pagination.NewIterator(ctx, nil, getCurrentExecution(retryer, pageSize))\n}\n\n// CurrentExecution returns a single execution\nfunc CurrentExecution(\n\tctx context.Context,\n\tretryer persistence.Retryer,\n\trequest ExecutionRequest,\n) (entity.Entity, error) {\n\treq := persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   request.DomainID,\n\t\tWorkflowID: request.WorkflowID,\n\t\tDomainName: request.DomainName,\n\t}\n\te, err := retryer.GetCurrentExecution(ctx, &req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &entity.CurrentExecution{\n\t\tCurrentRunID: e.RunID,\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    retryer.GetShardID(),\n\t\t\tDomainID:   request.DomainID,\n\t\t\tWorkflowID: request.WorkflowID,\n\t\t\tRunID:      e.RunID,\n\t\t\tState:      e.State,\n\t\t},\n\t}, nil\n}\n\nfunc getCurrentExecution(\n\tpr persistence.Retryer,\n\tpageSize int,\n) pagination.FetchFn {\n\treturn func(ctx context.Context, token pagination.PageToken) (pagination.Page, error) {\n\t\treq := &persistence.ListCurrentExecutionsRequest{\n\t\t\tPageSize: pageSize,\n\t\t}\n\t\tif token != nil {\n\t\t\treq.PageToken = token.([]byte)\n\t\t}\n\t\tresp, err := pr.ListCurrentExecutions(ctx, req)\n\t\tif err != nil {\n\t\t\treturn pagination.Page{}, err\n\t\t}\n\t\texecutions := make([]pagination.Entity, len(resp.Executions))\n\t\tfor i, e := range resp.Executions {\n\t\t\tcurrentExec := &entity.CurrentExecution{\n\t\t\t\tCurrentRunID: e.CurrentRunID,\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tShardID:    pr.GetShardID(),\n\t\t\t\t\tDomainID:   e.DomainID,\n\t\t\t\t\tWorkflowID: e.WorkflowID,\n\t\t\t\t\tRunID:      e.RunID,\n\t\t\t\t\tState:      e.State,\n\t\t\t\t},\n\t\t\t}\n\t\t\tif err := currentExec.Validate(); err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\t\t\texecutions[i] = currentExec\n\t\t}\n\t\tvar nextToken interface{} = resp.PageToken\n\t\tif len(resp.PageToken) == 0 {\n\t\t\tnextToken = nil\n\t\t}\n\t\tpage := pagination.Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextToken,\n\t\t\tEntities:     executions,\n\t\t}\n\t\treturn page, nil\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/current_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\npackage fetcher\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\nfunc TestCurrentExecutionIterator(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tretryer := persistence.NewMockRetryer(ctrl)\n\tretryer.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).\n\t\tReturn(&persistence.ListCurrentExecutionsResponse{}, nil).\n\t\tTimes(1)\n\n\titerator := CurrentExecutionIterator(\n\t\tcontext.Background(),\n\t\tretryer,\n\t\t10,\n\t)\n\trequire.NotNil(t, iterator)\n}\n\nfunc TestCurrentExecution(t *testing.T) {\n\tctx := context.Background()\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMock  func(*persistence.MockRetryer)\n\t\trequest    ExecutionRequest\n\t\twantEntity entity.Entity\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\trequest: ExecutionRequest{\n\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\tDomainName: \"testDomainName\",\n\t\t\t},\n\t\t\tsetupMock: func(mockRetryer *persistence.MockRetryer) {\n\t\t\t\tmockRetryer.EXPECT().GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\tDomainName: \"testDomainName\",\n\t\t\t\t}).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: \"testRunID\",\n\t\t\t\t\tState: persistence.WorkflowStateRunning,\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tmockRetryer.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t},\n\t\t\twantEntity: &entity.CurrentExecution{\n\t\t\t\tCurrentRunID: \"testRunID\",\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tShardID:    123,\n\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetCurrentExecution failed\",\n\t\t\trequest: ExecutionRequest{\n\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\tDomainName: \"testDomainName\",\n\t\t\t},\n\t\t\tsetupMock: func(mockRetryer *persistence.MockRetryer) {\n\t\t\t\tmockRetryer.EXPECT().GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\tDomainName: \"testDomainName\",\n\t\t\t\t}).Return(nil, fmt.Errorf(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\ttc.setupMock(mockRetryer)\n\t\t\tgotEntity, err := CurrentExecution(ctx, mockRetryer, tc.request)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"CurrentExecution() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\trequire.Equal(t, tc.wantEntity, gotEntity)\n\t\t})\n\t}\n}\n\nfunc TestGetCurrentExecution(t *testing.T) {\n\tctx := context.Background()\n\tpageSize := 10\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(*persistence.MockRetryer)\n\t\twantPage  pagination.Page\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(mockRetryer *persistence.MockRetryer) {\n\t\t\t\texecutions := []*persistence.CurrentWorkflowExecution{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainID:     \"testDomainID\",\n\t\t\t\t\t\tWorkflowID:   \"testWorkflowID\",\n\t\t\t\t\t\tRunID:        \"testRunID\",\n\t\t\t\t\t\tState:        persistence.WorkflowStateRunning,\n\t\t\t\t\t\tCurrentRunID: \"testCurrentRunID\", // Setting CurrentRunID\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockRetryer.EXPECT().\n\t\t\t\t\tListCurrentExecutions(ctx, &persistence.ListCurrentExecutionsRequest{\n\t\t\t\t\t\tPageSize: pageSize,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&persistence.ListCurrentExecutionsResponse{\n\t\t\t\t\t\tExecutions: executions,\n\t\t\t\t\t\tPageToken:  nil,\n\t\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tmockRetryer.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t},\n\t\t\twantPage: pagination.Page{\n\t\t\t\tEntities: []pagination.Entity{\n\t\t\t\t\t&entity.CurrentExecution{\n\t\t\t\t\t\tCurrentRunID: \"testCurrentRunID\", // This should match with the mocked data\n\t\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\t\tShardID:    123,\n\t\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"ListCurrentExecutions failed\",\n\t\t\tsetupMock: func(mockRetryer *persistence.MockRetryer) {\n\t\t\t\tmockRetryer.EXPECT().\n\t\t\t\t\tListCurrentExecutions(ctx, &persistence.ListCurrentExecutionsRequest{\n\t\t\t\t\t\tPageSize: pageSize,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"failed\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\ttc.setupMock(mockRetryer)\n\t\t\tfetchFn := getCurrentExecution(mockRetryer, pageSize)\n\t\t\tpage, err := fetchFn(ctx, nil)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"getCurrentExecution() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\n\t\t\trequire.Equal(t, tc.wantPage, page)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/timer.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\n// TimerIterator is used to retrieve Concrete executions.\nfunc TimerIterator(\n\tctx context.Context,\n\tretryer persistence.Retryer,\n\tminTimestamp time.Time,\n\tmaxTimestamp time.Time,\n\tpageSize int,\n) pagination.Iterator {\n\treturn pagination.NewIterator(ctx, nil, getUserTimers(retryer, minTimestamp, maxTimestamp, pageSize))\n}\n\nfunc getUserTimers(\n\tpr persistence.Retryer,\n\tminTimestamp time.Time,\n\tmaxTimestamp time.Time,\n\tpageSize int,\n) pagination.FetchFn {\n\treturn func(ctx context.Context, token pagination.PageToken) (pagination.Page, error) {\n\t\treq := &persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(minTimestamp, 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxTimestamp, 0),\n\t\t\tPageSize:            pageSize,\n\t\t}\n\t\tif token != nil {\n\t\t\treq.NextPageToken = token.([]byte)\n\t\t}\n\t\tresp, err := pr.GetHistoryTasks(ctx, req)\n\n\t\tif err != nil {\n\t\t\treturn pagination.Page{}, err\n\t\t}\n\n\t\tvar timers []pagination.Entity\n\n\t\tfor _, t := range resp.Tasks {\n\t\t\tif t.GetTaskType() != persistence.TaskTypeUserTimer {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\ttimer := &entity.Timer{\n\t\t\t\tShardID:             pr.GetShardID(),\n\t\t\t\tDomainID:            t.GetDomainID(),\n\t\t\t\tWorkflowID:          t.GetWorkflowID(),\n\t\t\t\tRunID:               t.GetRunID(),\n\t\t\t\tTaskType:            t.GetTaskType(),\n\t\t\t\tVisibilityTimestamp: t.GetVisibilityTimestamp(),\n\t\t\t}\n\n\t\t\tif err := timer.Validate(); err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\t\t\ttimers = append(timers, timer)\n\t\t}\n\t\tvar nextToken interface{} = resp.NextPageToken\n\t\tif len(resp.NextPageToken) == 0 {\n\t\t\tnextToken = nil\n\t\t}\n\n\t\tpage := pagination.Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextToken,\n\t\t\tEntities:     timers,\n\t\t}\n\t\treturn page, nil\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/timer_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\nfunc TestTimerIterator(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tretryer := persistence.NewMockRetryer(ctrl)\n\tretryer.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).\n\t\tReturn(&persistence.GetHistoryTasksResponse{}, nil).\n\t\tTimes(1)\n\n\titerator := TimerIterator(\n\t\tcontext.Background(),\n\t\tretryer,\n\t\ttime.Now(),\n\t\ttime.Now(),\n\t\t10,\n\t)\n\trequire.NotNil(t, iterator)\n}\n\nfunc TestGetUserTimers(t *testing.T) {\n\tfixedTimestamp, err := time.Parse(time.RFC3339, \"2023-12-12T22:08:41Z\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to parse timestamp: %v\", err)\n\t}\n\n\tpageSize := 10\n\tminTimestamp := fixedTimestamp.Add(-time.Hour)\n\tmaxTimestamp := fixedTimestamp\n\tnonNilToken := []byte(\"non-nil-token\")\n\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMock     func(ctrl *gomock.Controller) *persistence.MockRetryer\n\t\ttoken         pagination.PageToken\n\t\texpectedPage  pagination.Page\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *persistence.MockRetryer {\n\t\t\t\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\t\t\t\ttimerTasks := []persistence.Task{\n\t\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   \"testDomainID\",\n\t\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVisibilityTimestamp: fixedTimestamp,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockRetryer.EXPECT().\n\t\t\t\t\tGetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(minTimestamp, 0),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxTimestamp, 0),\n\t\t\t\t\t\tPageSize:            pageSize,\n\t\t\t\t\t\tNextPageToken:       nil,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\t\tTasks:         timerTasks,\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t}, nil)\n\n\t\t\t\tmockRetryer.EXPECT().GetShardID().Return(123)\n\n\t\t\t\treturn mockRetryer\n\t\t\t},\n\t\t\ttoken: nil,\n\t\t\texpectedPage: pagination.Page{\n\t\t\t\tEntities: []pagination.Entity{\n\t\t\t\t\t&entity.Timer{\n\t\t\t\t\t\tShardID:             123,\n\t\t\t\t\t\tDomainID:            \"testDomainID\",\n\t\t\t\t\t\tWorkflowID:          \"testWorkflowID\",\n\t\t\t\t\t\tRunID:               \"testRunID\",\n\t\t\t\t\t\tTaskType:            persistence.TaskTypeUserTimer,\n\t\t\t\t\t\tVisibilityTimestamp: fixedTimestamp,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Non-nil Pagination Token Provided\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *persistence.MockRetryer {\n\t\t\t\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\t\tmockRetryer.EXPECT().\n\t\t\t\t\tGetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(minTimestamp, 0),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxTimestamp, 0),\n\t\t\t\t\t\tPageSize:            pageSize,\n\t\t\t\t\t\tNextPageToken:       nonNilToken,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\t\tTasks:         nil,\n\t\t\t\t\t\tNextPageToken: nonNilToken,\n\t\t\t\t\t}, nil)\n\n\t\t\t\treturn mockRetryer\n\t\t\t},\n\t\t\ttoken: nonNilToken,\n\t\t\texpectedPage: pagination.Page{\n\t\t\t\tEntities:     nil,\n\t\t\t\tCurrentToken: nonNilToken,\n\t\t\t\tNextToken:    nonNilToken,\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Invalid Timer Causes Error\",\n\t\t\tsetupMock: func(ctrl *gomock.Controller) *persistence.MockRetryer {\n\t\t\t\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\t\t\t\tinvalidTimer := &persistence.UserTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"\", // Invalid as it's empty\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: fixedTimestamp,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockRetryer.EXPECT().\n\t\t\t\t\tGetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(minTimestamp, 0),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxTimestamp, 0),\n\t\t\t\t\t\tPageSize:            pageSize,\n\t\t\t\t\t\tNextPageToken:       nil,\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\t\tTasks:         []persistence.Task{invalidTimer},\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t}, nil)\n\n\t\t\t\tmockRetryer.EXPECT().GetShardID().Return(123)\n\n\t\t\t\treturn mockRetryer\n\t\t\t},\n\t\t\ttoken:         nil,\n\t\t\texpectedPage:  pagination.Page{},\n\t\t\texpectedError: fmt.Errorf(\"empty DomainID\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockRetryer := tc.setupMock(ctrl)\n\n\t\t\tfetchFn := getUserTimers(mockRetryer, minTimestamp, maxTimestamp, pageSize)\n\t\t\tpage, err := fetchFn(context.Background(), tc.token)\n\n\t\t\tif tc.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.EqualError(t, err, tc.expectedError.Error(), \"Error should match\")\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err, \"No error is expected\")\n\t\t\t}\n\n\t\t\trequire.Equal(t, tc.expectedPage, page, \"Page should match\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/fetcher/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage fetcher\n\n// ExecutionRequest is used to fetch execution from persistence\ntype ExecutionRequest struct {\n\tDomainID   string\n\tWorkflowID string\n\tRunID      string\n\tDomainName string\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/collection_enumer_generated.go",
    "content": "// Code generated by \"enumer -type=Collection -output collection_enumer_generated.go\"; DO NOT EDIT.\n\npackage invariant\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\nconst _CollectionName = \"CollectionMutableStateCollectionHistoryCollectionDomainCollectionStale\"\n\nvar _CollectionIndex = [...]uint8{0, 22, 39, 55, 70}\n\nconst _CollectionLowerName = \"collectionmutablestatecollectionhistorycollectiondomaincollectionstale\"\n\nfunc (i Collection) String() string {\n\tif i < 0 || i >= Collection(len(_CollectionIndex)-1) {\n\t\treturn fmt.Sprintf(\"Collection(%d)\", i)\n\t}\n\treturn _CollectionName[_CollectionIndex[i]:_CollectionIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _CollectionNoOp() {\n\tvar x [1]struct{}\n\t_ = x[CollectionMutableState-(0)]\n\t_ = x[CollectionHistory-(1)]\n\t_ = x[CollectionDomain-(2)]\n\t_ = x[CollectionStale-(3)]\n}\n\nvar _CollectionValues = []Collection{CollectionMutableState, CollectionHistory, CollectionDomain, CollectionStale}\n\nvar _CollectionNameToValueMap = map[string]Collection{\n\t_CollectionName[0:22]:       CollectionMutableState,\n\t_CollectionLowerName[0:22]:  CollectionMutableState,\n\t_CollectionName[22:39]:      CollectionHistory,\n\t_CollectionLowerName[22:39]: CollectionHistory,\n\t_CollectionName[39:55]:      CollectionDomain,\n\t_CollectionLowerName[39:55]: CollectionDomain,\n\t_CollectionName[55:70]:      CollectionStale,\n\t_CollectionLowerName[55:70]: CollectionStale,\n}\n\nvar _CollectionNames = []string{\n\t_CollectionName[0:22],\n\t_CollectionName[22:39],\n\t_CollectionName[39:55],\n\t_CollectionName[55:70],\n}\n\n// CollectionString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc CollectionString(s string) (Collection, error) {\n\tif val, ok := _CollectionNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _CollectionNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to Collection values\", s)\n}\n\n// CollectionValues returns all values of the enum\nfunc CollectionValues() []Collection {\n\treturn _CollectionValues\n}\n\n// CollectionStrings returns a slice of all String values of the enum\nfunc CollectionStrings() []string {\n\tstrs := make([]string, len(_CollectionNames))\n\tcopy(strs, _CollectionNames)\n\treturn strs\n}\n\n// IsACollection returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i Collection) IsACollection() bool {\n\tfor _, v := range _CollectionValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/concrete_execution_exists.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype concreteExecutionExists struct {\n\tpr    persistence.Retryer\n\tcache cache.DomainCache\n}\n\n// NewConcreteExecutionExists returns a new invariant for checking concrete execution\nfunc NewConcreteExecutionExists(pr persistence.Retryer, cache cache.DomainCache) Invariant {\n\treturn &concreteExecutionExists{\n\t\tpr:    pr,\n\t\tcache: cache,\n\t}\n}\n\nfunc (c *concreteExecutionExists) Check(ctx context.Context, execution interface{}) CheckResult {\n\tif checkResult := validateCheckContext(ctx, c.Name()); checkResult != nil {\n\t\treturn *checkResult\n\t}\n\n\tcurrentExecution, ok := execution.(*entity.CurrentExecution)\n\tif !ok {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"failed to check: expected current execution\",\n\t\t}\n\t}\n\n\tif len(currentExecution.CurrentRunID) == 0 {\n\t\t// set the current run id\n\t\tvar runIDCheckResult *CheckResult\n\t\tcurrentExecution, runIDCheckResult = c.validateCurrentRunID(ctx, currentExecution)\n\t\tif runIDCheckResult != nil {\n\t\t\treturn *runIDCheckResult\n\t\t}\n\t}\n\n\tdomainName, err := c.cache.GetDomainName(currentExecution.DomainID)\n\tif err != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"Failed to fetch Domain Name\",\n\t\t\tInfoDetails:     err.Error(),\n\t\t}\n\t}\n\tconcreteExecResp, concreteExecErr := c.pr.IsWorkflowExecutionExists(ctx, &persistence.IsWorkflowExecutionExistsRequest{\n\t\tDomainID:   currentExecution.DomainID,\n\t\tDomainName: domainName,\n\t\tWorkflowID: currentExecution.WorkflowID,\n\t\tRunID:      currentExecution.CurrentRunID,\n\t})\n\tif concreteExecErr != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"failed to check if concrete execution exists\",\n\t\t\tInfoDetails:     concreteExecErr.Error(),\n\t\t}\n\t}\n\tif !concreteExecResp.Exists {\n\t\t// verify if the current execution exists\n\t\t_, checkResult := c.validateCurrentRunID(ctx, currentExecution)\n\t\tif checkResult != nil {\n\t\t\treturn *checkResult\n\t\t}\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"execution is open without having concrete execution\",\n\t\t\tInfoDetails: fmt.Sprintf(\"concrete execution not found. WorkflowId: %v, RunId: %v\",\n\t\t\t\tcurrentExecution.WorkflowID, currentExecution.CurrentRunID),\n\t\t}\n\t}\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   c.Name(),\n\t}\n}\n\nfunc (c *concreteExecutionExists) Fix(ctx context.Context, execution interface{}) FixResult {\n\tif fixResult := validateFixContext(ctx, c.Name()); fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\tcurrentExecution, ok := execution.(*entity.CurrentExecution)\n\tif !ok {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: c.Name(),\n\t\t\tInfo:          \"failed to fix: expected current execution\",\n\t\t}\n\t}\n\n\tvar runIDCheckResult *CheckResult\n\tif len(currentExecution.CurrentRunID) == 0 {\n\t\t// this is to set the current run ID prior to the check and fix operations\n\t\tcurrentExecution, runIDCheckResult = c.validateCurrentRunID(ctx, currentExecution)\n\t\tif runIDCheckResult != nil {\n\t\t\treturn FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tCheckResult:   *runIDCheckResult,\n\t\t\t\tInvariantName: c.Name(),\n\t\t\t}\n\t\t}\n\t}\n\tfixResult, checkResult := checkBeforeFix(ctx, c, currentExecution)\n\tif fixResult != nil {\n\t\treturn *fixResult\n\t}\n\tdomainName, errorDomain := c.cache.GetDomainName(currentExecution.DomainID)\n\tif errorDomain != nil {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: c.Name(),\n\t\t\tInfo:          \"failed to fetch Domain Name\",\n\t\t\tInfoDetails:   errorDomain.Error(),\n\t\t}\n\t}\n\tif err := c.pr.DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\tDomainID:   currentExecution.DomainID,\n\t\tWorkflowID: currentExecution.WorkflowID,\n\t\tRunID:      currentExecution.CurrentRunID,\n\t\tDomainName: domainName,\n\t}); err != nil {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: c.Name(),\n\t\t\tInfo:          \"failed to delete current workflow execution\",\n\t\t\tInfoDetails:   err.Error(),\n\t\t}\n\t}\n\treturn FixResult{\n\t\tFixResultType: FixResultTypeFixed,\n\t\tCheckResult:   *checkResult,\n\t\tInvariantName: c.Name(),\n\t}\n}\n\nfunc (c *concreteExecutionExists) Name() Name {\n\treturn ConcreteExecutionExists\n}\n\nfunc (c *concreteExecutionExists) validateCurrentRunID(\n\tctx context.Context,\n\tcurrentExecution *entity.CurrentExecution,\n) (*entity.CurrentExecution, *CheckResult) {\n\tdomainName, err := c.cache.GetDomainName(currentExecution.DomainID)\n\tif err != nil {\n\t\treturn nil, &CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"failed to fetch domainName\",\n\t\t\tInfoDetails:     err.Error(),\n\t\t}\n\t}\n\tresp, err := c.pr.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   currentExecution.DomainID,\n\t\tWorkflowID: currentExecution.WorkflowID,\n\t\tDomainName: domainName,\n\t})\n\tif err != nil {\n\t\tswitch err.(type) {\n\t\tcase *types.EntityNotExistsError:\n\t\t\treturn nil, &CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   c.Name(),\n\t\t\t\tInfo:            \"current execution does not exist.\",\n\t\t\t\tInfoDetails:     err.Error(),\n\t\t\t}\n\t\tdefault:\n\t\t\treturn nil, &CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   c.Name(),\n\t\t\t\tInfo:            \"failed to get current execution.\",\n\t\t\t\tInfoDetails:     err.Error(),\n\t\t\t}\n\t\t}\n\t}\n\n\tif len(currentExecution.CurrentRunID) == 0 {\n\t\tcurrentExecution.CurrentRunID = resp.RunID\n\t}\n\n\tif currentExecution.CurrentRunID != resp.RunID {\n\t\treturn nil, &CheckResult{\n\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\tInvariantName:   c.Name(),\n\t\t}\n\t}\n\treturn currentExecution, nil\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/concrete_execution_exists_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\tc \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestConcreteExecutionCheckAndFix(t *testing.T) {\n\tnotExistsError := types.EntityNotExistsError{}\n\tunknownError := types.BadRequestError{}\n\ttestCases := []struct {\n\t\tdesc             string\n\t\texecution        any\n\t\tctx              context.Context\n\t\tgetConcreteResp  *persistence.IsWorkflowExecutionExistsResponse\n\t\tgetConcreteErr   error\n\t\tgetCurrentResp   *persistence.GetCurrentExecutionResponse\n\t\tgetCurrentErr    error\n\t\tgetDomainNameErr error\n\t\twantCheckResult  CheckResult\n\t\twantFixResult    FixResult\n\t}{\n\t\t{\n\t\t\tdesc:            \"closed execution with concrete execution\",\n\t\t\texecution:       getClosedCurrentExecution(),\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: true},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: getClosedCurrentExecution().CurrentRunID,\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:           \"failed to get concrete execution\",\n\t\t\texecution:      getOpenCurrentExecution(),\n\t\t\tgetConcreteErr: errors.New(\"error getting concrete execution\"),\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: getOpenCurrentExecution().CurrentRunID,\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"failed to check if concrete execution exists\",\n\t\t\t\tInfoDetails:     \"error getting concrete execution\",\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"failed fix because check failed\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"open execution without concrete execution\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: false},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: getOpenCurrentExecution().CurrentRunID,\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"execution is open without having concrete execution\",\n\t\t\t\tInfoDetails:     fmt.Sprintf(\"concrete execution not found. WorkflowId: %v, RunId: %v\", workflowID, currentRunID),\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"mismatching current runid and concrete execution doesn't exist\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: false},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: uuid.New(),\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"open execution with concrete execution\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteErr:  nil,\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: true},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: getOpenCurrentExecution().CurrentRunID,\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"open execution that is not current\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteErr:  nil,\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: true},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: uuid.New(),\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"concrete exists but current doesn't\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteErr:  nil,\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: true},\n\t\t\tgetCurrentResp:  nil,\n\t\t\tgetCurrentErr:   &notExistsError,\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:            \"concrete exists but failed to get current\",\n\t\t\texecution:       getOpenCurrentExecution(),\n\t\t\tgetConcreteErr:  nil,\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: false},\n\t\t\tgetCurrentResp:  nil,\n\t\t\tgetCurrentErr:   &unknownError,\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"failed to get current execution.\",\n\t\t\t\tInfoDetails:     unknownError.Error(),\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"failed fix because check failed\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"canceled context\",\n\t\t\tctx:  canceledCtx(),\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"failed to check: context expired or cancelled\",\n\t\t\t\tInfoDetails:     \"context canceled\",\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"failed to check: context expired or cancelled\",\n\t\t\t\tInfoDetails:   \"context canceled\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:      \"invalid execution object\",\n\t\t\texecution: &entity.ConcreteExecution{}, // invalid type\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"failed to check: expected current execution\",\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"failed to fix: expected current execution\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"empty run id - found after current lookup\",\n\t\t\texecution: func() *entity.CurrentExecution {\n\t\t\t\te := getOpenCurrentExecution()\n\t\t\t\te.CurrentRunID = \"\"\n\t\t\t\treturn e\n\t\t\t}(),\n\t\t\tgetConcreteResp: &persistence.IsWorkflowExecutionExistsResponse{Exists: true},\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: currentRunID,\n\t\t\t},\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"empty run id - current lookup returns not found\",\n\t\t\texecution: func() *entity.CurrentExecution {\n\t\t\t\te := getOpenCurrentExecution()\n\t\t\t\te.CurrentRunID = \"\"\n\t\t\t\treturn e\n\t\t\t}(),\n\t\t\tgetCurrentErr: &notExistsError,\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"current execution does not exist.\",\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"empty run id - domain name lookup failed\",\n\t\t\texecution: func() *entity.CurrentExecution {\n\t\t\t\te := getOpenCurrentExecution()\n\t\t\t\te.CurrentRunID = \"\"\n\t\t\t\treturn e\n\t\t\t}(),\n\t\t\tgetDomainNameErr: errors.New(\"error getting domain name\"),\n\t\t\twantCheckResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   ConcreteExecutionExists,\n\t\t\t\tInfo:            \"failed to fetch domainName\",\n\t\t\t\tInfoDetails:     \"error getting domain name\",\n\t\t\t},\n\t\t\twantFixResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: ConcreteExecutionExists,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\texecManager := &mocks.ExecutionManager{}\n\t\t\texecManager.On(\"IsWorkflowExecutionExists\", mock.Anything, mock.Anything).Return(tc.getConcreteResp, tc.getConcreteErr)\n\t\t\texecManager.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(tc.getCurrentResp, tc.getCurrentErr)\n\t\t\texecManager.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(nil)\n\t\t\tmockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, tc.getDomainNameErr).AnyTimes()\n\t\t\to := NewConcreteExecutionExists(persistence.NewPersistenceRetryer(execManager, nil, c.CreatePersistenceRetryPolicy()), mockDomainCache)\n\t\t\tctx := tc.ctx\n\t\t\tif ctx == nil {\n\t\t\t\tctx = context.Background()\n\t\t\t}\n\t\t\trequire.Equal(t, tc.wantCheckResult, o.Check(ctx, tc.execution))\n\n\t\t\tgotFixResult := o.Fix(ctx, tc.execution)\n\t\t\tgotFixResult.CheckResult = CheckResult{}\n\t\t\trequire.Equal(t, tc.wantFixResult, gotFixResult)\n\t\t})\n\t}\n}\n\nfunc canceledCtx() context.Context {\n\tctx, cancel := context.WithCancel(context.Background())\n\tcancel()\n\treturn ctx\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/history_exists.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\n\tc \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\thistoryPageSize = 1\n)\n\ntype (\n\thistoryExists struct {\n\t\tpr persistence.Retryer\n\t\tdc cache.DomainCache\n\t}\n)\n\nfunc NewHistoryExists(\n\tpr persistence.Retryer, dc cache.DomainCache,\n) Invariant {\n\treturn &historyExists{\n\t\tpr: pr,\n\t\tdc: dc,\n\t}\n}\n\nfunc (h *historyExists) Check(\n\tctx context.Context,\n\texecution interface{},\n) CheckResult {\n\tif checkResult := validateCheckContext(ctx, h.Name()); checkResult != nil {\n\t\treturn *checkResult\n\t}\n\n\tconcreteExecution, ok := execution.(*entity.ConcreteExecution)\n\tif !ok {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"failed to check: expected concrete execution\",\n\t\t}\n\t}\n\tdomainID := concreteExecution.GetDomainID()\n\tdomainName, errorDomainName := h.dc.GetDomainName(domainID)\n\tif errorDomainName != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"failed to check: expected DomainName\",\n\t\t\tInfoDetails:     errorDomainName.Error(),\n\t\t}\n\t}\n\treadHistoryBranchReq := &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   concreteExecution.BranchToken,\n\t\tMinEventID:    constants.FirstEventID,\n\t\tMaxEventID:    constants.FirstEventID + 1,\n\t\tPageSize:      historyPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       c.IntPtr(concreteExecution.ShardID),\n\t\tDomainName:    domainName,\n\t}\n\treadHistoryBranchResp, readHistoryBranchErr := h.pr.ReadHistoryBranch(ctx, readHistoryBranchReq)\n\tstillExists, existsCheckError := ExecutionStillExists(ctx, &concreteExecution.Execution, h.pr, h.dc)\n\tif existsCheckError != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"failed to check if concrete execution still exists\",\n\t\t\tInfoDetails:     existsCheckError.Error(),\n\t\t}\n\t}\n\tif !stillExists {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"determined execution was healthy because concrete execution no longer exists\",\n\t\t}\n\t}\n\tif readHistoryBranchErr != nil {\n\t\tswitch readHistoryBranchErr.(type) {\n\t\tcase *types.EntityNotExistsError:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   h.Name(),\n\t\t\t\tInfo:            \"concrete execution exists but history does not exist\",\n\t\t\t\tInfoDetails:     readHistoryBranchErr.Error(),\n\t\t\t}\n\t\tdefault:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   h.Name(),\n\t\t\t\tInfo:            \"failed to verify if history exists\",\n\t\t\t\tInfoDetails:     readHistoryBranchErr.Error(),\n\t\t\t}\n\t\t}\n\t}\n\tif readHistoryBranchResp == nil || len(readHistoryBranchResp.HistoryEvents) == 0 {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"concrete execution exists but got empty history\",\n\t\t}\n\t}\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   h.Name(),\n\t}\n}\n\nfunc (h *historyExists) Fix(\n\tctx context.Context,\n\texecution interface{},\n) FixResult {\n\tif fixResult := validateFixContext(ctx, h.Name()); fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\tfixResult, checkResult := checkBeforeFix(ctx, h, execution)\n\tif fixResult != nil {\n\t\treturn *fixResult\n\t}\n\tfixResult = DeleteExecution(ctx, execution, h.pr, h.dc)\n\tfixResult.CheckResult = *checkResult\n\tfixResult.InvariantName = h.Name()\n\treturn *fixResult\n}\n\nfunc (h *historyExists) Name() Name {\n\treturn HistoryExists\n}\n\n// ExecutionStillExists returns true if execution still exists in persistence, false otherwise.\n// Returns error on failure to confirm.\nfunc ExecutionStillExists(\n\tctx context.Context,\n\texec *entity.Execution,\n\tpr persistence.Retryer,\n\tdc cache.DomainCache,\n) (bool, error) {\n\tdomainName, errorDomainName := dc.GetDomainName(exec.DomainID)\n\tif errorDomainName != nil {\n\t\treturn false, errorDomainName\n\t}\n\treq := &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: exec.DomainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: exec.WorkflowID,\n\t\t\tRunID:      exec.RunID,\n\t\t},\n\t\tDomainName: domainName,\n\t}\n\t_, err := pr.GetWorkflowExecution(ctx, req)\n\tif err == nil {\n\t\treturn true, nil\n\t}\n\tswitch err.(type) {\n\tcase *types.EntityNotExistsError:\n\t\treturn false, nil\n\tdefault:\n\t\treturn false, err\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/history_exists_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\tc2 \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype HistoryExistsSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestHistoryExistsSuite(t *testing.T) {\n\tsuite.Run(t, new(HistoryExistsSuite))\n}\n\nfunc (s *HistoryExistsSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *HistoryExistsSuite) TestCheck() {\n\ttestCases := []struct {\n\t\tgetExecErr                error\n\t\tgetExecResp               *persistence.GetWorkflowExecutionResponse\n\t\tgetHistoryErr             error\n\t\tgetHistoryResp            *persistence.ReadHistoryBranchResponse\n\t\texpectedResult            CheckResult\n\t\texpectedResourcePopulated bool\n\t}{\n\t\t{\n\t\t\tgetExecErr:     errors.New(\"got error checking workflow exists\"),\n\t\t\tgetHistoryResp: &persistence.ReadHistoryBranchResponse{},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t\tInfo:            \"failed to check if concrete execution still exists\",\n\t\t\t\tInfoDetails:     \"got error checking workflow exists\",\n\t\t\t},\n\t\t\texpectedResourcePopulated: false,\n\t\t},\n\t\t{\n\t\t\tgetExecErr:     &types.EntityNotExistsError{},\n\t\t\tgetHistoryResp: &persistence.ReadHistoryBranchResponse{},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t\tInfo:            \"determined execution was healthy because concrete execution no longer exists\",\n\t\t\t},\n\t\t\texpectedResourcePopulated: false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp:    &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetHistoryResp: nil,\n\t\t\tgetHistoryErr:  &types.EntityNotExistsError{Message: \"got entity not exists error\"},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t\tInfo:            \"concrete execution exists but history does not exist\",\n\t\t\t\tInfoDetails:     \"got entity not exists error\",\n\t\t\t},\n\t\t\texpectedResourcePopulated: false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp:    &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetHistoryResp: nil,\n\t\t\tgetHistoryErr:  errors.New(\"error fetching history\"),\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t\tInfo:            \"failed to verify if history exists\",\n\t\t\t\tInfoDetails:     \"error fetching history\",\n\t\t\t},\n\t\t\texpectedResourcePopulated: false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp:    &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetHistoryResp: nil,\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t\tInfo:            \"concrete execution exists but got empty history\",\n\t\t\t},\n\t\t\texpectedResourcePopulated: false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetHistoryResp: &persistence.ReadHistoryBranchResponse{\n\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   HistoryExists,\n\t\t\t},\n\t\t\texpectedResourcePopulated: true,\n\t\t},\n\t}\n\n\tctrl := gomock.NewController(s.T())\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\thistoryManager := &mocks.HistoryV2Manager{}\n\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\thistoryManager.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(tc.getHistoryResp, tc.getHistoryErr)\n\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\ti := NewHistoryExists(persistence.NewPersistenceRetryer(execManager, historyManager, c2.CreatePersistenceRetryPolicy()), domainCache)\n\t\tresult := i.Check(context.Background(), getOpenConcreteExecution())\n\t\ts.Equal(tc.expectedResult, result)\n\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/inactive_domain_exists.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\ntype (\n\tinactiveDomainExists struct {\n\t\tpr persistence.Retryer\n\t\tdc cache.DomainCache\n\t}\n)\n\nfunc NewInactiveDomainExists(\n\tpr persistence.Retryer, dc cache.DomainCache,\n) Invariant {\n\treturn &inactiveDomainExists{\n\t\tpr: pr,\n\t\tdc: dc,\n\t}\n}\n\nfunc (idc *inactiveDomainExists) Check(\n\tctx context.Context,\n\texecution interface{},\n) CheckResult {\n\tif checkResult := validateCheckContext(ctx, idc.Name()); checkResult != nil {\n\t\treturn *checkResult\n\t}\n\n\tconcreteExecution, ok := execution.(*entity.ConcreteExecution)\n\tif !ok {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   idc.Name(),\n\t\t\tInfo:            \"failed to check: expected concrete execution\",\n\t\t}\n\t}\n\tdomainID := concreteExecution.GetDomainID()\n\tdomain, err := idc.dc.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   idc.Name(),\n\t\t\tInfo:            \"failed to check: expected Domain\",\n\t\t\tInfoDetails:     err.Error(),\n\t\t}\n\t}\n\tif domain.GetInfo().Status == persistence.DomainStatusDeprecated || domain.GetInfo().Status == persistence.DomainStatusDeleted {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   idc.Name(),\n\t\t\tInfo:            \"failed check: domain is not active\",\n\t\t\tInfoDetails:     \"The domain has been deprecated or deleted\",\n\t\t}\n\t}\n\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   idc.Name(),\n\t\tInfo:            \"workflow's domain is active\",\n\t\tInfoDetails:     \"workflow's domain is active\",\n\t}\n}\n\nfunc (idc *inactiveDomainExists) Fix(\n\tctx context.Context,\n\texecution interface{},\n) FixResult {\n\tif fixResult := validateFixContext(ctx, idc.Name()); fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\tfixResult, checkResult := checkBeforeFix(ctx, idc, execution)\n\tif fixResult != nil {\n\t\treturn *fixResult\n\t}\n\tfixResult = DeleteExecution(ctx, execution, idc.pr, idc.dc)\n\tfixResult.CheckResult = *checkResult\n\tfixResult.InvariantName = idc.Name()\n\treturn *fixResult\n}\n\nfunc (idc *inactiveDomainExists) Name() Name {\n\treturn InactiveDomainExists\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/inactive_domain_exists_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\tc2 \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\ntype InactiveInactiveDomainExistsSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestInactiveInactiveDomainExistsSuite(t *testing.T) {\n\tsuite.Run(t, new(InactiveInactiveDomainExistsSuite))\n}\n\nfunc (s *InactiveInactiveDomainExistsSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *InactiveInactiveDomainExistsSuite) TestCheck() {\n\ttestCases := []struct {\n\t\tgetExecErr     error\n\t\tgetExecResp    *persistence.GetWorkflowExecutionResponse\n\t\tgetDomainErr   error\n\t\texpectedResult CheckResult\n\t}{\n\t\t{\n\t\t\tgetExecErr: &types.EntityNotExistsError{},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   InactiveDomainExists,\n\t\t\t\tInfo:            \"workflow's domain is active\",\n\t\t\t\tInfoDetails:     \"workflow's domain is active\",\n\t\t\t},\n\t\t},\n\t}\n\n\tctrl := gomock.NewController(s.T())\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\tdomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\t\ti := NewInactiveDomainExists(persistence.NewPersistenceRetryer(execManager, nil, c2.CreatePersistenceRetryPolicy()), domainCache)\n\t\tresult := i.Check(context.Background(), getOpenConcreteExecution())\n\t\ts.Equal(tc.expectedResult, result)\n\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/invariant_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport \"context\"\n\ntype (\n\tinvariantManager struct {\n\t\tinvariants []Invariant\n\t}\n)\n\n// NewInvariantManager handles running a collection of invariants according to the invariant collection provided.\nfunc NewInvariantManager(\n\tinvariants []Invariant,\n) Manager {\n\treturn &invariantManager{\n\t\tinvariants: invariants,\n\t}\n}\n\n// RunChecks runs all enabled checks.\nfunc (i *invariantManager) RunChecks(\n\tctx context.Context,\n\texecution interface{},\n) ManagerCheckResult {\n\tresult := ManagerCheckResult{\n\t\tCheckResultType:          CheckResultTypeHealthy,\n\t\tDeterminingInvariantType: nil,\n\t\tCheckResults:             nil,\n\t}\n\tfor _, iv := range i.invariants {\n\t\tcheckResult := iv.Check(ctx, execution)\n\t\tresult.CheckResults = append(result.CheckResults, checkResult)\n\t\tcheckResultType, updated := i.nextCheckResultType(result.CheckResultType, checkResult.CheckResultType)\n\t\tresult.CheckResultType = checkResultType\n\t\tif updated {\n\t\t\tresult.DeterminingInvariantType = &checkResult.InvariantName\n\t\t}\n\t}\n\treturn result\n}\n\n// RunFixes runs all enabled fixes.\nfunc (i *invariantManager) RunFixes(\n\tctx context.Context,\n\texecution interface{}) ManagerFixResult {\n\tresult := ManagerFixResult{\n\t\tFixResultType:            FixResultTypeSkipped,\n\t\tDeterminingInvariantName: nil,\n\t\tFixResults:               nil,\n\t}\n\tfor _, iv := range i.invariants {\n\t\tfixResult := iv.Fix(ctx, execution)\n\t\tresult.FixResults = append(result.FixResults, fixResult)\n\t\tfixResultType, updated := i.nextFixResultType(result.FixResultType, fixResult.FixResultType)\n\t\tresult.FixResultType = fixResultType\n\t\tif updated {\n\t\t\tresult.DeterminingInvariantName = &fixResult.InvariantName\n\t\t}\n\t}\n\treturn result\n}\n\nfunc (i *invariantManager) nextFixResultType(\n\tcurrentState FixResultType,\n\tevent FixResultType,\n) (FixResultType, bool) {\n\tswitch currentState {\n\tcase FixResultTypeSkipped:\n\t\treturn event, event != FixResultTypeSkipped\n\tcase FixResultTypeFixed:\n\t\tif event == FixResultTypeFailed {\n\t\t\treturn event, true\n\t\t}\n\t\treturn currentState, false\n\tcase FixResultTypeFailed:\n\t\treturn currentState, false\n\tdefault:\n\t\tpanic(\"unknown FixResultType\")\n\t}\n}\n\nfunc (i *invariantManager) nextCheckResultType(\n\tcurrentState CheckResultType,\n\tevent CheckResultType,\n) (CheckResultType, bool) {\n\tswitch currentState {\n\tcase CheckResultTypeHealthy:\n\t\treturn event, event != CheckResultTypeHealthy\n\tcase CheckResultTypeCorrupted:\n\t\tif event == CheckResultTypeFailed {\n\t\t\treturn event, true\n\t\t}\n\t\treturn currentState, false\n\tcase CheckResultTypeFailed:\n\t\treturn currentState, false\n\tdefault:\n\t\tpanic(\"unknown CheckResultType\")\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/invariant_manager_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\ntype InvariantManagerSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\tcontroller *gomock.Controller\n}\n\nfunc TestInvariantManagerSuite(t *testing.T) {\n\tsuite.Run(t, new(InvariantManagerSuite))\n}\n\nfunc (s *InvariantManagerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n}\n\nfunc (s *InvariantManagerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *InvariantManagerSuite) TestRunChecks() {\n\ttestCases := []struct {\n\t\tcheckResults []CheckResult\n\t\texpected     ManagerCheckResult\n\t}{\n\t\t{\n\t\t\tcheckResults: nil,\n\t\t\texpected: ManagerCheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tCheckResults:    nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcheckResults: []CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerCheckResult{\n\t\t\t\tCheckResultType:          CheckResultTypeFailed,\n\t\t\t\tDeterminingInvariantType: NamePtr(\"second\"),\n\t\t\t\tCheckResults: []CheckResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcheckResults: []CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerCheckResult{\n\t\t\t\tCheckResultType:          CheckResultTypeCorrupted,\n\t\t\t\tDeterminingInvariantType: NamePtr(\"second\"),\n\t\t\t\tCheckResults: []CheckResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcheckResults: []CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerCheckResult{\n\t\t\t\tCheckResultType:          CheckResultTypeHealthy,\n\t\t\t\tDeterminingInvariantType: nil,\n\t\t\t\tCheckResults: []CheckResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcheckResults: []CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\t\tInvariantName:   Name(\"third\"),\n\t\t\t\t\tInfo:            \"invariant 3 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 3 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   Name(\"forth\"),\n\t\t\t\t\tInfo:            \"invariant 4 info\",\n\t\t\t\t\tInfoDetails:     \"invariant 4 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerCheckResult{\n\t\t\t\tCheckResultType:          CheckResultTypeFailed,\n\t\t\t\tDeterminingInvariantType: NamePtr(\"third\"),\n\t\t\t\tCheckResults: []CheckResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"first\"),\n\t\t\t\t\t\tInfo:            \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInvariantName:   Name(\"second\"),\n\t\t\t\t\t\tInfo:            \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\t\t\tInvariantName:   Name(\"third\"),\n\t\t\t\t\t\tInfo:            \"invariant 3 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 3 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInvariantName:   Name(\"forth\"),\n\t\t\t\t\t\tInfo:            \"invariant 4 info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 4 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tinvariants := make([]Invariant, len(tc.checkResults))\n\t\tfor i := 0; i < len(tc.checkResults); i++ {\n\t\t\tmockInvariant := NewMockInvariant(s.controller)\n\t\t\tmockInvariant.EXPECT().Check(gomock.Any(), gomock.Any()).Return(tc.checkResults[i])\n\t\t\tinvariants[i] = mockInvariant\n\t\t}\n\t\tmanager := &invariantManager{\n\t\t\tinvariants: invariants,\n\t\t}\n\t\ts.Equal(tc.expected, manager.RunChecks(context.Background(), entity.Execution{}))\n\t}\n}\n\nfunc (s *InvariantManagerSuite) TestRunFixes() {\n\ttestCases := []struct {\n\t\tfixResults []FixResult\n\t\texpected   ManagerFixResult\n\t}{\n\t\t{\n\t\t\tfixResults: nil,\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeSkipped,\n\t\t\t\tDeterminingInvariantName: nil,\n\t\t\t\tFixResults:               nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tfixResults: []FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeFailed,\n\t\t\t\tDeterminingInvariantName: NamePtr(\"second\"),\n\t\t\t\tFixResults: []FixResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tfixResults: []FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeSkipped,\n\t\t\t\tDeterminingInvariantName: nil,\n\t\t\t\tFixResults: []FixResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tfixResults: []FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeFixed,\n\t\t\t\tDeterminingInvariantName: NamePtr(\"second\"),\n\t\t\t\tFixResults: []FixResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tfixResults: []FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\tInvariantName: Name(\"forth\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 4 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 4 info\",\n\t\t\t\t\tInfoDetails: \"invariant 4 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeFailed,\n\t\t\t\tDeterminingInvariantName: NamePtr(\"forth\"),\n\t\t\t\tFixResults: []FixResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\t\tInvariantName: Name(\"forth\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 4 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 4 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 4 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tfixResults: []FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 4 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 4 info\",\n\t\t\t\t\tInfoDetails: \"invariant 4 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: Name(\"forth\"),\n\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t},\n\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: ManagerFixResult{\n\t\t\t\tFixResultType:            FixResultTypeFailed,\n\t\t\t\tDeterminingInvariantName: NamePtr(\"second\"),\n\t\t\t\tFixResults: []FixResult{\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"first\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 1 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 1 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 1 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 1 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\t\t\tInvariantName: Name(\"second\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 4 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 4 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 4 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\t\t\tInvariantName: Name(\"third\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\t\t\tInfo:            \"invariant 2 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 2 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 2 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 2 info details\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\t\t\tInvariantName: Name(\"forth\"),\n\t\t\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\t\t\t\tInfo:            \"invariant 3 check info\",\n\t\t\t\t\t\t\tInfoDetails:     \"invariant 3 check info details\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tInfo:        \"invariant 3 info\",\n\t\t\t\t\t\tInfoDetails: \"invariant 3 info details\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tinvariants := make([]Invariant, len(tc.fixResults))\n\t\tfor i := 0; i < len(tc.fixResults); i++ {\n\t\t\tmockInvariant := NewMockInvariant(s.controller)\n\t\t\tmockInvariant.EXPECT().Fix(gomock.Any(), gomock.Any()).Return(tc.fixResults[i])\n\t\t\tinvariants[i] = mockInvariant\n\t\t}\n\t\tmanager := &invariantManager{\n\t\t\tinvariants: invariants,\n\t\t}\n\t\ts.Equal(tc.expected, manager.RunFixes(context.Background(), entity.Execution{}))\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/invariant_test_utils.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\nconst (\n\tdomainID     = \"test-domain-id\"\n\tdomainName   = \"test-domain-name\"\n\tworkflowID   = \"test-workflow-id\"\n\trunID        = \"test-run-id\"\n\tshardID      = 0\n\ttreeID       = \"test-tree-id\"\n\tbranchID     = \"test-branch-id\"\n\topenState    = persistence.WorkflowStateCreated\n\tclosedState  = persistence.WorkflowStateCompleted\n\tcurrentRunID = \"test-current-run-id\"\n)\n\nvar (\n\tbranchToken = []byte{1, 2, 3}\n)\n\nfunc getOpenConcreteExecution() *entity.ConcreteExecution {\n\treturn &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tState:      openState,\n\t\t},\n\t\tBranchToken: branchToken,\n\t\tTreeID:      treeID,\n\t\tBranchID:    branchID,\n\t}\n}\n\nfunc getClosedConcreteExecution() *entity.ConcreteExecution {\n\treturn &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tState:      closedState,\n\t\t},\n\t\tBranchToken: branchToken,\n\t\tTreeID:      treeID,\n\t\tBranchID:    branchID,\n\t}\n}\n\nfunc getOpenCurrentExecution() *entity.CurrentExecution {\n\treturn &entity.CurrentExecution{\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tState:      openState,\n\t\t},\n\t\tCurrentRunID: currentRunID,\n\t}\n}\n\nfunc getClosedCurrentExecution() *entity.CurrentExecution {\n\treturn &entity.CurrentExecution{\n\t\tExecution: entity.Execution{\n\t\t\tShardID:    shardID,\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tState:      closedState,\n\t\t},\n\t\tCurrentRunID: currentRunID,\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/mocks.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: types.go\n//\n// Generated by this command:\n//\n//\tmockgen -package invariant -source types.go -destination mocks.go -self_package github.com/uber/cadence/common/reconciliation/invariant\n//\n\n// Package invariant is a generated GoMock package.\npackage invariant\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockInvariant is a mock of Invariant interface.\ntype MockInvariant struct {\n\tctrl     *gomock.Controller\n\trecorder *MockInvariantMockRecorder\n\tisgomock struct{}\n}\n\n// MockInvariantMockRecorder is the mock recorder for MockInvariant.\ntype MockInvariantMockRecorder struct {\n\tmock *MockInvariant\n}\n\n// NewMockInvariant creates a new mock instance.\nfunc NewMockInvariant(ctrl *gomock.Controller) *MockInvariant {\n\tmock := &MockInvariant{ctrl: ctrl}\n\tmock.recorder = &MockInvariantMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockInvariant) EXPECT() *MockInvariantMockRecorder {\n\treturn m.recorder\n}\n\n// Check mocks base method.\nfunc (m *MockInvariant) Check(arg0 context.Context, arg1 any) CheckResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Check\", arg0, arg1)\n\tret0, _ := ret[0].(CheckResult)\n\treturn ret0\n}\n\n// Check indicates an expected call of Check.\nfunc (mr *MockInvariantMockRecorder) Check(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Check\", reflect.TypeOf((*MockInvariant)(nil).Check), arg0, arg1)\n}\n\n// Fix mocks base method.\nfunc (m *MockInvariant) Fix(arg0 context.Context, arg1 any) FixResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Fix\", arg0, arg1)\n\tret0, _ := ret[0].(FixResult)\n\treturn ret0\n}\n\n// Fix indicates an expected call of Fix.\nfunc (mr *MockInvariantMockRecorder) Fix(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Fix\", reflect.TypeOf((*MockInvariant)(nil).Fix), arg0, arg1)\n}\n\n// Name mocks base method.\nfunc (m *MockInvariant) Name() Name {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Name\")\n\tret0, _ := ret[0].(Name)\n\treturn ret0\n}\n\n// Name indicates an expected call of Name.\nfunc (mr *MockInvariantMockRecorder) Name() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Name\", reflect.TypeOf((*MockInvariant)(nil).Name))\n}\n\n// MockManager is a mock of Manager interface.\ntype MockManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockManagerMockRecorder is the mock recorder for MockManager.\ntype MockManagerMockRecorder struct {\n\tmock *MockManager\n}\n\n// NewMockManager creates a new mock instance.\nfunc NewMockManager(ctrl *gomock.Controller) *MockManager {\n\tmock := &MockManager{ctrl: ctrl}\n\tmock.recorder = &MockManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockManager) EXPECT() *MockManagerMockRecorder {\n\treturn m.recorder\n}\n\n// RunChecks mocks base method.\nfunc (m *MockManager) RunChecks(arg0 context.Context, arg1 any) ManagerCheckResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RunChecks\", arg0, arg1)\n\tret0, _ := ret[0].(ManagerCheckResult)\n\treturn ret0\n}\n\n// RunChecks indicates an expected call of RunChecks.\nfunc (mr *MockManagerMockRecorder) RunChecks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RunChecks\", reflect.TypeOf((*MockManager)(nil).RunChecks), arg0, arg1)\n}\n\n// RunFixes mocks base method.\nfunc (m *MockManager) RunFixes(arg0 context.Context, arg1 any) ManagerFixResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RunFixes\", arg0, arg1)\n\tret0, _ := ret[0].(ManagerFixResult)\n\treturn ret0\n}\n\n// RunFixes indicates an expected call of RunFixes.\nfunc (mr *MockManagerMockRecorder) RunFixes(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RunFixes\", reflect.TypeOf((*MockManager)(nil).RunFixes), arg0, arg1)\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/open_current_execution.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\topenCurrentExecution struct {\n\t\tpr persistence.Retryer\n\t\tdc cache.DomainCache\n\t}\n)\n\n// NewOpenCurrentExecution returns a new invariant for checking open current execution\nfunc NewOpenCurrentExecution(\n\tpr persistence.Retryer, dc cache.DomainCache,\n) Invariant {\n\treturn &openCurrentExecution{\n\t\tpr: pr,\n\t\tdc: dc,\n\t}\n}\n\nfunc (o *openCurrentExecution) Check(\n\tctx context.Context,\n\texecution interface{},\n) CheckResult {\n\tif checkResult := validateCheckContext(ctx, o.Name()); checkResult != nil {\n\t\treturn *checkResult\n\t}\n\n\tconcreteExecution, ok := execution.(*entity.ConcreteExecution)\n\tif !ok {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   o.Name(),\n\t\t\tInfo:            \"failed to check: expected concrete execution\",\n\t\t}\n\t}\n\tif !Open(concreteExecution.State) {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\tInvariantName:   o.Name(),\n\t\t}\n\t}\n\tdomainName, err := o.dc.GetDomainName(concreteExecution.DomainID)\n\tif err != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   o.Name(),\n\t\t\tInfo:            \"failed to fetch Domain Name\",\n\t\t\tInfoDetails:     err.Error(),\n\t\t}\n\t}\n\tcurrentExecResp, currentExecErr := o.pr.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   concreteExecution.DomainID,\n\t\tWorkflowID: concreteExecution.WorkflowID,\n\t\tDomainName: domainName,\n\t})\n\n\tstillOpen, stillOpenErr := ExecutionStillOpen(ctx, &concreteExecution.Execution, o.pr, o.dc)\n\tif stillOpenErr != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   o.Name(),\n\t\t\tInfo:            \"failed to check if concrete execution is still open\",\n\t\t\tInfoDetails:     stillOpenErr.Error(),\n\t\t}\n\t}\n\tif !stillOpen {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\tInvariantName:   o.Name(),\n\t\t}\n\t}\n\tif currentExecErr != nil {\n\t\tswitch currentExecErr.(type) {\n\t\tcase *types.EntityNotExistsError:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   o.Name(),\n\t\t\t\tInfo:            \"execution is open without having current execution\",\n\t\t\t\tInfoDetails:     currentExecErr.Error(),\n\t\t\t}\n\t\tdefault:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   o.Name(),\n\t\t\t\tInfo:            \"failed to check if current execution exists\",\n\t\t\t\tInfoDetails:     currentExecErr.Error(),\n\t\t\t}\n\t\t}\n\t}\n\tif currentExecResp.RunID != concreteExecution.RunID {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   o.Name(),\n\t\t\tInfo:            \"execution is open but current points at a different execution\",\n\t\t\tInfoDetails:     fmt.Sprintf(\"current points at %v\", currentExecResp.RunID),\n\t\t}\n\t}\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   o.Name(),\n\t}\n}\n\nfunc (o *openCurrentExecution) Fix(\n\tctx context.Context,\n\texecution interface{},\n) FixResult {\n\tif fixResult := validateFixContext(ctx, o.Name()); fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\tfixResult, checkResult := checkBeforeFix(ctx, o, execution)\n\tif fixResult != nil {\n\t\treturn *fixResult\n\t}\n\tfixResult = DeleteExecution(ctx, execution, o.pr, o.dc)\n\tfixResult.CheckResult = *checkResult\n\tfixResult.InvariantName = o.Name()\n\treturn *fixResult\n}\n\nfunc (o *openCurrentExecution) Name() Name {\n\treturn OpenCurrentExecution\n}\n\n// ExecutionStillOpen returns true if execution in persistence exists and is open, false otherwise.\n// Returns error on failure to confirm.\nfunc ExecutionStillOpen(\n\tctx context.Context,\n\texec *entity.Execution,\n\tpr persistence.Retryer,\n\tdc cache.DomainCache,\n) (bool, error) {\n\tdomainName, err := dc.GetDomainName(exec.DomainID)\n\tif err != nil {\n\t\treturn false, nil\n\t}\n\treq := &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: exec.DomainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: exec.WorkflowID,\n\t\t\tRunID:      exec.RunID,\n\t\t},\n\t\tDomainName: domainName,\n\t}\n\tresp, err := pr.GetWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\tswitch err.(type) {\n\t\tcase *types.EntityNotExistsError:\n\t\t\treturn false, nil\n\t\tdefault:\n\t\t\treturn false, err\n\t\t}\n\t}\n\treturn Open(resp.State.ExecutionInfo.State), nil\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/open_current_execution_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\tc2 \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype OpenCurrentExecutionSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestOpenCurrentExecutionSuite(t *testing.T) {\n\tsuite.Run(t, new(OpenCurrentExecutionSuite))\n}\n\nfunc (s *OpenCurrentExecutionSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *OpenCurrentExecutionSuite) TestCheck() {\n\ttestCases := []struct {\n\t\texecution       *entity.ConcreteExecution\n\t\tgetCurrentResp  *persistence.GetCurrentExecutionResponse\n\t\tgetCurrentErr   error\n\t\tgetConcreteResp *persistence.GetWorkflowExecutionResponse\n\t\tgetConcreteErr  error\n\t\texpectedResult  CheckResult\n\t}{\n\t\t{\n\t\t\texecution: getClosedConcreteExecution(),\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution:      getOpenConcreteExecution(),\n\t\t\tgetConcreteErr: errors.New(\"got error checking if concrete is open\"),\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t\tInfo:            \"failed to check if concrete execution is still open\",\n\t\t\t\tInfoDetails:     \"got error checking if concrete is open\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution: getOpenConcreteExecution(),\n\t\t\tgetConcreteResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: closedState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetConcreteErr: nil,\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution: getOpenConcreteExecution(),\n\t\t\tgetConcreteResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: openState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetConcreteErr: nil,\n\t\t\tgetCurrentErr:  &types.EntityNotExistsError{},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t\tInfo:            \"execution is open without having current execution\",\n\t\t\t\tInfoDetails:     \"\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution: getOpenConcreteExecution(),\n\t\t\tgetConcreteResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: openState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetConcreteErr: nil,\n\t\t\tgetCurrentErr:  errors.New(\"error getting current execution\"),\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t\tInfo:            \"failed to check if current execution exists\",\n\t\t\t\tInfoDetails:     \"error getting current execution\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution: getOpenConcreteExecution(),\n\t\t\tgetConcreteResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: openState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetConcreteErr: nil,\n\t\t\tgetCurrentErr:  nil,\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: \"not-equal\",\n\t\t\t},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t\tInfo:            \"execution is open but current points at a different execution\",\n\t\t\t\tInfoDetails:     \"current points at not-equal\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texecution: getOpenConcreteExecution(),\n\t\t\tgetConcreteResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: openState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetConcreteErr: nil,\n\t\t\tgetCurrentErr:  nil,\n\t\t\tgetCurrentResp: &persistence.GetCurrentExecutionResponse{\n\t\t\t\tRunID: runID,\n\t\t\t},\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   OpenCurrentExecution,\n\t\t\t},\n\t\t},\n\t}\n\tctrl := gomock.NewController(s.T())\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getConcreteResp, tc.getConcreteErr)\n\t\texecManager.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(tc.getCurrentResp, tc.getCurrentErr)\n\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\to := NewOpenCurrentExecution(persistence.NewPersistenceRetryer(execManager, nil, c2.CreatePersistenceRetryPolicy()), domainCache)\n\t\ts.Equal(tc.expectedResult, o.Check(context.Background(), tc.execution))\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/stale_workflow.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"strings\"\n\t\"time\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// how long to allow things to be beyond when they should have been fully cleaned up, for safety purposes\n\tretentionSafetyMargin = time.Hour * 24 * 10\n\n\t// time.DateOnly, introduced in go1.20\n\t//\n\t// deprecated: use time.DateOnly when able\n\tdateOnly = \"2006-01-02\"\n)\n\n// sane time ranges for workflows.\n// this is largely to protect against accidental conversions from zero values, or incorrect second/nanosecond/etc scales.\nvar (\n\t// too old to be real.  e.g. prior to Cadence existing.\n\t//\n\t// anything \"reasonable\" is likely fine, but this should be substantially newer than 1970 to catch zero times.\n\timpossiblyOld = time.Date(2015, 0, 0, 0, 0, 0, 0, time.UTC)\n\t// too far in the future to be real.\n\t//\n\t// crazy values are technically possible if someone sets a 100-year timeout on a workflow or something,\n\t// but we should probably block those anyway.  hopefully nonexistent or rare enough to handle manually (and correct).\n\timpossiblyFuture = time.Date(2100, 0, 0, 0, 0, 0, 0, time.UTC)\n\t// our internal limits are:\n\t// - practically: 30 days\n\t// - for monthly-job people: 40 days\n\t// - rare exceptions we are trying to eliminate: 90+ days\n\t//\n\t// this should be far beyond that and should imply bad data, not merely an exceptional domain.\n\t//\n\t// note that this is not used to *force* deletion.  workflows in domains beyond this value will fail and be skipped.\n\tmaxRetentionDays = 200\n)\n\ntype staleWorkflowCheck struct {\n\tpr  persistence.Retryer\n\tdc  cache.DomainCache\n\tlog *zap.Logger\n}\n\n// NewStaleWorkflow checks to see if a workflow has out-lived its retention window.\n// This primarily asserts that that now < (min(start + execution timeout, min) + (domain retention*2)).\n// If a workflow fails this check, its data is still lingering well beyond when it should have been cleaned up.\nfunc NewStaleWorkflow(\n\tpr persistence.Retryer,\n\tdc cache.DomainCache,\n\tlog *zap.Logger,\n) Invariant {\n\treturn &staleWorkflowCheck{\n\t\tpr:  pr,\n\t\tdc:  dc,\n\t\tlog: log,\n\t}\n}\n\nfunc (c *staleWorkflowCheck) Check(\n\tctx context.Context,\n\texecution interface{},\n) CheckResult {\n\t_, result := c.check(ctx, execution)\n\treturn result\n}\n\nfunc (c *staleWorkflowCheck) check(\n\tctx context.Context,\n\texecution interface{},\n) (deleteConcrete bool, result CheckResult) {\n\n\tconcreteExecution, ok := execution.(*entity.ConcreteExecution)\n\tif !ok {\n\t\treturn false, c.failed(\"expected concrete execution\", \"\")\n\t}\n\n\tif concreteExecution.WorkflowID == \"\" ||\n\t\tconcreteExecution.RunID == \"\" ||\n\t\tconcreteExecution.DomainID == \"\" {\n\t\treturn false, c.failed(\"missing critical data\", \"\")\n\t}\n\tdomainID := concreteExecution.GetDomainID()\n\n\tconcreteWorkflow, err := c.pr.GetWorkflowExecution(ctx, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: domainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: concreteExecution.WorkflowID,\n\t\t\tRunID:      concreteExecution.RunID,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn false, c.failed(\"failed to get concrete execution record\", err.Error())\n\t}\n\n\tpastExpiration, checkresult := c.CheckAge(concreteWorkflow)\n\tif pastExpiration {\n\t\tif checkresult.CheckResultType == CheckResultTypeCorrupted {\n\t\t\treturn true, checkresult // delete the concrete execution, it's out of retention\n\t\t}\n\t\t// else check the current record, as it may be causing issues for the concrete\n\t}\n\treturn false, checkresult\n\n\t// TODO: ^ similar check for current record?  Bad-current can sometimes block a different concrete.\n}\n\nfunc (c *staleWorkflowCheck) Fix(ctx context.Context, execution interface{}) FixResult {\n\t// essentially checkBeforeFix\n\tdeleteConcrete, checkResult := c.check(ctx, execution)\n\tif checkResult.CheckResultType == CheckResultTypeHealthy {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\tInvariantName: c.Name(),\n\t\t\tInfo:          fmt.Sprintf(\"no need to fix: %s\", checkResult.Info),\n\t\t\tInfoDetails:   checkResult.InfoDetails,\n\t\t}\n\t}\n\n\tif checkResult.CheckResultType != CheckResultTypeCorrupted {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: c.Name(),\n\t\t\tInfo:          fmt.Sprintf(\"unable to determine if beyond expiration: %s\", checkResult.Info),\n\t\t\tInfoDetails:   checkResult.InfoDetails,\n\t\t}\n\t}\n\n\tif deleteConcrete {\n\t\tfixResult := DeleteExecution(ctx, execution, c.pr, c.dc)\n\t\tfixResult.CheckResult = checkResult\n\t\tfixResult.InvariantName = c.Name()\n\t\treturn *fixResult\n\t}\n\n\t// TODO: handle current deletion?\n\t// probably not possible to reach currently because current records are not handled in check.\n\treturn FixResult{\n\t\tFixResultType: FixResultTypeFailed,\n\t\tInvariantName: c.Name(),\n\t\tInfo:          fmt.Sprintf(\"current record deletion not yet built: %v\", checkResult.Info),\n\t\tInfoDetails:   checkResult.InfoDetails,\n\t}\n}\n\nfunc (c *staleWorkflowCheck) Name() Name {\n\treturn StaleWorkflow\n}\n\nfunc (c *staleWorkflowCheck) CheckAge(workflow *persistence.GetWorkflowExecutionResponse) (pastExpiration bool, result CheckResult) {\n\tinfo := workflow.State.ExecutionInfo\n\tretentionNum, domainName, err := c.getDomainInfo(info)\n\tif err != nil {\n\t\treturn false, c.failed(\n\t\t\t\"unable to get domain retention\",\n\t\t\t\"domain: %v, name: %v, err: %v\", info.DomainID, domainName, err)\n\t}\n\tif retentionNum <= 0 {\n\t\treturn false, c.failed(\n\t\t\t\"non-positive retention days in domain\",\n\t\t\t\"domain: %v, name: %v\", info.DomainID, domainName)\n\t}\n\tif retentionNum > int32(maxRetentionDays) {\n\t\treturn false, c.failed(\n\t\t\t\"irrationally-large retention days in domain\",\n\t\t\t\"domain: %v, name: %v, days: %v, max: %v\", info.DomainID, domainName, retentionNum, maxRetentionDays)\n\t}\n\n\tretention := time.Hour * 24 * time.Duration(retentionNum)\n\t// add a safety buffer of 10 days\n\tsafety := time.Hour * 24 * 10\n\tmaxLifespan := retention + safety\n\n\tlogStaleWorkflow := func(state string, expected time.Time) {\n\t\tc.log.Info(\"scanner found stale \"+state+\" workflow\",\n\t\t\tzap.String(\"wid\", info.WorkflowID),\n\t\t\tzap.String(\"rid\", info.RunID),\n\t\t\tzap.String(\"domain_name\", domainName),\n\t\t\tzap.String(\"domain_id\", info.DomainID),\n\t\t\tzap.String(\"started_at\", info.StartTimestamp.Format(dateOnly)),\n\t\t\tzap.String(\"expected_cleanup_at\", expected.Format(dateOnly)),\n\t\t\tzap.String(\"state\", state), // may simplify log processing\n\t\t)\n\t}\n\n\t// treat \"running\" (has had decision tasks) and \"created\" (no decisions, e.g. cron / delay / lost initial tasks)\n\t// as essentially the same thing.  either way there's no close event to check.\n\tif info.State == persistence.WorkflowStateRunning || info.State == persistence.WorkflowStateCreated {\n\t\tstale, expected, result := c.checkRunningAge(workflow, maxLifespan, domainName)\n\t\tif stale {\n\t\t\t// we know these exist, but we don't expect them to be numerous \"recently\".\n\t\t\t// log for verification.\n\t\t\tlogStaleWorkflow(\"running\", expected)\n\t\t}\n\t\treturn stale, result\n\t} else if info.State == persistence.WorkflowStateZombie {\n\t\t// TODO: what exactly is zombie?  seems like probably \"current but no concrete\" but I'm surprised this is in the state.\n\t\t//\n\t\t// https://github.com/uber/cadence/issues/1800\n\t\t// \"The ability to create workflow with zombie status allows replication stack to backfill finished workflow execution sent by remote, while not affecting local running workflow\"\n\t\t// so these are just replication of completed workflows?  why is this a special state?\n\t\tstale, expected, result := c.checkZombieAge(workflow, maxLifespan, domainName)\n\t\tif stale {\n\t\t\t// we know these exist, but we don't have a good feel for the distribution\n\t\t\tlogStaleWorkflow(\"zombie\", expected)\n\t\t}\n\t\treturn stale, result\n\t} else if info.State == persistence.WorkflowStateCompleted {\n\t\tstale, expected, result := c.checkClosedAge(workflow, maxLifespan, domainName)\n\t\tif stale {\n\t\t\t// we know these exist, but we don't have a good feel for the distribution\n\t\t\tlogStaleWorkflow(\"closed\", expected)\n\t\t}\n\t\treturn stale, result\n\t}\n\n\t// currently one of: corrupted, created, void\n\t// but if more are added in the future, this is still the safe option.\n\treturn false, c.failed(\n\t\t\"unhandled workflow state\",\n\t\t\"info.State value: %d\", info.State)\n}\n\nfunc (c *staleWorkflowCheck) checkRunningAge(workflow *persistence.GetWorkflowExecutionResponse, maxLifespan time.Duration, domainName string) (pastExpiration bool, expected time.Time, result CheckResult) {\n\t// workflow-expiration timer is calculated in GenerateWorkflowStartTasks, mimic it here for safety: https://github.com/uber/cadence/blob/master/service/history/execution/mutable_state_task_generator.go#L140-L163\n\t/*\n\t\t// running workflows might contain critical information in their first history record, so it must be retrieved.\n\t\tinfo := workflow.State.ExecutionInfo\n\n\t\tduration := time.Duration(info.WorkflowTimeout) * time.Second\n\t\t// First-decision-backoff is only in first event.  It must be included because the expiration time does not include it.\n\t\t// I really wish it wasn't built this way, this makes it much much harder to scan in bulk.\n\t\ttimeout := info.StartTimestamp.Add(duration + firstdecisionbackoff)\n\n\t\t// ensure that the first attempt does not time out early based on retry policy timeout\n\t\tif attr.Attempt > 0 && !executionInfo.ExpirationTime.IsZero() && workflowTimeoutTimestamp.After(executionInfo.ExpirationTime) {\n\t\t\tworkflowTimeoutTimestamp = executionInfo.ExpirationTime\n\t\t}\n\t*/\n\tfirst, ok, result := c.firstEvent(workflow, domainName)\n\tif !ok {\n\t\t// completely missing history should be caught by the history-exists check, no need to handle here\n\t\treturn false, expected, result\n\t}\n\tstartedAt := workflow.State.ExecutionInfo.StartTimestamp                             // nonzero\n\ttimeout := time.Second * time.Duration(workflow.State.ExecutionInfo.WorkflowTimeout) // nonzero\n\tbackoff := time.Second * time.Duration(first.GetFirstDecisionTaskBackoffSeconds())   // may be zero\n\n\t// comment in GenerateWorkflowStartTasks seems backwards, it clamps to the minimum of timeout and retry-expiration,\n\t// which means we should be able to ignore it and be *safe* if perhaps not *optimal*.\n\t//\n\t// completed workflows seem to have this set, but perhaps not running.\n\t// which can happen here since this method is reused for closed-but-corrupt workflows.\n\t// since I'm not confident on what this field's value is / if it's trustworthy, ignore for now.\n\t//\n\t// expiration := workflow.State.ExecutionInfo.ExpirationTime\n\t// if !expiration.IsZero() {\n\t// \tfmt.Println(\"non-zero expiration time, not sure what this means\")\n\t// }\n\n\ttimesOutAt := startedAt.Add(timeout).Add(backoff)\n\tcleansUpAt := timesOutAt.Add(maxLifespan)\n\t// sanity checks on calculated times\n\tif ok, result := c.checkTimeInSaneRange(timesOutAt, \"workflow timeout\"); !ok {\n\t\treturn false, expected, result\n\t}\n\tif ok, result := c.checkTimeInSaneRange(cleansUpAt, \"workflow cleanup time\"); !ok {\n\t\treturn false, expected, result\n\t}\n\n\tif cleansUpAt.Before(time.Now()) {\n\t\treturn true, cleansUpAt, CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"stale running workflow\",\n\t\t\tInfoDetails:     fmt.Sprintf(\"running workflow should have timed out by %v and been cleaned up by %v, but it still exists\", timesOutAt.Format(dateOnly), cleansUpAt.Format(dateOnly)),\n\t\t}\n\t}\n\treturn false, cleansUpAt, CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   c.Name(),\n\t\tInfo:            \"running workflow is within expiration window\",\n\t}\n}\n\n// intended as defensive programming, in case values are in some wrong scale or some other surprise occurs.\n//\n// these should not trigger in practice, and are intended to catch bugs or logical flaws while developing or changing\n// this fixer, or flawed/missing knowledge - if they occur, that's likely a scenario we should be handling explicitly.\nfunc (c *staleWorkflowCheck) checkTimeInSaneRange(t time.Time, kind string) (ok bool, result CheckResult) {\n\tif t.IsZero() {\n\t\t// will also be before-impossibly-old, but separated for clarity purposes\n\t\treturn false, c.failed(fmt.Sprintf(\"calculated %v is zero, failing\", kind), \"\")\n\t}\n\tif t.Before(impossiblyOld) || t.After(impossiblyFuture) {\n\t\t// something screwed up, time is outside sane bounds\n\t\treturn false, c.failed(\n\t\t\tfmt.Sprintf(\"calculated %v seems insane, failing\", kind),\n\t\t\tfmt.Sprintf(\"expected %v to be within range: %q < actual %q < %q\", kind, impossiblyOld, t, impossiblyFuture))\n\t}\n\treturn true, result\n}\n\nfunc (c *staleWorkflowCheck) checkClosedAge(workflow *persistence.GetWorkflowExecutionResponse, maxLifespan time.Duration, domainName string) (pastExpiration bool, expected time.Time, result CheckResult) {\n\tclosed, ok, result := c.closeEventTime(workflow, domainName)\n\tif !ok {\n\t\t// error of some kind, see if we can consider it stale from just the start info.\n\t\t// this is somewhat common if we are missing the final event in a workflow, e.g. due to broken replication or partial (un)deletion.\n\t\t// running retention will never be shorter than closed retention, so this is always safe.\n\t\trunningStale, expected, runningResult := c.checkRunningAge(workflow, maxLifespan, domainName)\n\t\t// rewrite to mention closed, but interpreted as running.\n\t\t// the reason for failure is still relevant, but this way at least we know it was from the fallback.\n\t\treturn runningStale, expected, CheckResult{\n\t\t\tCheckResultType: runningResult.CheckResultType,\n\t\t\tInvariantName:   runningResult.InvariantName,\n\t\t\tInfo:            strings.Replace(runningResult.Info, \"running workflow\", \"completed workflow (err, treating as running)\", 1) + \". closed info: \" + result.Info,\n\t\t\tInfoDetails:     strings.Replace(runningResult.InfoDetails, \"running workflow\", \"completed workflow (err, treating as running)\", 1) + \". closed info details: \" + result.InfoDetails,\n\t\t}\n\t}\n\tif ok, result := c.checkTimeInSaneRange(closed, \"workflow close\"); !ok {\n\t\t// bad data, zero value, or perhaps a number-scale issue.  either way untrustworthy.\n\t\treturn false, expected, result\n\t}\n\texpected = closed.Add(maxLifespan)\n\tif expected.Before(time.Now()) {\n\t\t// >10 days beyond retention, should be deleted\n\t\treturn true, expected, CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"completed workflow exists beyond retention window\",\n\t\t\tInfoDetails:     fmt.Sprintf(\"completed workflow exists beyond retention window, should have disappeared by %v\", closed.Add(maxLifespan).Format(dateOnly)),\n\t\t}\n\t}\n\n\t// still in retention, it's fine\n\treturn false, expected, CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   c.Name(),\n\t\tInfo:            \"completed workflow still within retention + 10-day buffer\",\n\t\tInfoDetails:     fmt.Sprintf(\"completed workflow still within retention + 10-day buffer, closed %v and allowed to exist until %v\", closed, closed.Add(maxLifespan).Format(dateOnly)),\n\t}\n\n}\n\nfunc (c *staleWorkflowCheck) checkZombieAge(workflow *persistence.GetWorkflowExecutionResponse, maxLifespan time.Duration, domainName string) (pastExpiration bool, expected time.Time, result CheckResult) {\n\t// zombies may or may not have history, and may or may not be \"running\" depending on if enough history has replicated.\n\t// they are not ever \"current\" though apparently.  despite there being a current execution record pointing to them.  wut.\n\t/*\n\t\tfunc (e *mutableStateBuilder) IsCurrentWorkflowGuaranteed() bool {\n\t\t\t// stateInDB is used like a bloom filter:\n\t\t\t//\n\t\t\t// 1. stateInDB being created / running meaning that this workflow must be the current\n\t\t\t//  workflow (assuming there is no rebuild of mutable state).\n\t\t\t// 2. stateInDB being completed does not guarantee this workflow being the current workflow\n\t\t\t// 3. stateInDB being zombie guarantees this workflow not being the current workflow\n\t\t\t// 4. stateInDB cannot be void, void is only possible when mutable state is just initialized\n\t*/\n\n\t// AFAICT zombie workflows are both not running and not completely replicated, i.e. there is no completion info or final completion event.\n\t// so just check them as running, at least until we learn otherwise.\n\t//\n\t// if there are completed ones, just check completed-result if running-age fails.\n\t// the cleanup time is just the minimum of the two (should always be completed if it exists)\n\tcleanup, expected, result := c.checkRunningAge(workflow, maxLifespan, domainName)\n\t// just reword it\n\tresult.Info = strings.Replace(result.Info, \"running workflow\", \"zombie workflow\", 1)\n\tresult.InfoDetails = strings.Replace(result.InfoDetails, \"running workflow\", \"zombie workflow\", 1)\n\treturn cleanup, expected, result\n}\n\n// it's just quite verbose\nfunc (c *staleWorkflowCheck) failed(info string, details string, args ...interface{}) CheckResult {\n\tif details != \"\" {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            info,\n\t\t\tInfoDetails:     fmt.Sprintf(details, args...),\n\t\t}\n\t}\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeFailed,\n\t\tInvariantName:   c.Name(),\n\t\tInfo:            info,\n\t}\n}\n\nfunc (c *staleWorkflowCheck) firstEvent(workflow *persistence.GetWorkflowExecutionResponse, domainName string) (attrs types.WorkflowExecutionStartedEventAttributes, ok bool, result CheckResult) {\n\t// see: func (e *mutableStateBuilder) GetStartEvent(\n\t/*\n\t\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tstartEvent, err := e.eventsCache.GetEvent(\n\t\t\tctx,\n\t\t\te.shard.GetShardID(),\n\t\t\te.executionInfo.DomainID,\n\t\t\te.executionInfo.WorkflowID,\n\t\t\te.executionInfo.RunID,\n\t\t\tcommon.FirstEventID,\n\t\t\tcommon.FirstEventID,\n\t\t\tcurrentBranchToken,\n\t\t)\n\t*/\n\tbranchToken, ok, result := c.getBranchToken(workflow)\n\tif !ok {\n\t\treturn attrs, false, result\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\tdefer cancel()\n\tshard := c.pr.GetShardID()\n\thistory, err := c.pr.ReadHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken: branchToken,\n\t\tMinEventID:  constants.FirstEventID,\n\t\tMaxEventID:  constants.FirstEventID + 1, // exclusive bound\n\t\tShardID:     &shard,\n\t\tPageSize:    1, // just 1 necessary\n\t\tDomainName:  domainName,\n\t})\n\tif err != nil {\n\t\treturn attrs, false, c.failed(\n\t\t\t\"could not read first event history\",\n\t\t\t\"read branch error for shard: %v, token: %s, err: %v\", c.pr.GetShardID(), branchToken, err)\n\t}\n\tif len(history.HistoryEvents) < 1 {\n\t\treturn attrs, false, c.failed(\n\t\t\t\"incomplete history, missing first event?\",\n\t\t\t\"got: %v\", history.HistoryEvents)\n\t}\n\tfirst := history.HistoryEvents[0]\n\tif first.WorkflowExecutionStartedEventAttributes == nil {\n\t\treturn attrs, false, c.failed(\"missing start event attributes\", \"got: %v\", first)\n\t}\n\treturn *first.WorkflowExecutionStartedEventAttributes, true, result\n}\n\nfunc (c *staleWorkflowCheck) getBranchToken(workflow *persistence.GetWorkflowExecutionResponse) (token []byte, ok bool, result CheckResult) {\n\t/*\n\t\tfunc (e *mutableStateBuilder) GetCurrentBranchToken() ([]byte, error) {\n\t\t\tif e.versionHistories != nil {\n\t\t\t\tcurrentVersionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\treturn currentVersionHistory.GetBranchToken(), nil\n\t\t\t}\n\t\t\treturn e.executionInfo.BranchToken, nil\n\t\t}\n\t*/\n\tif workflow.State.VersionHistories == nil {\n\t\t// should imply no history data, should be impossible?\n\t\treturn nil, false, c.failed(\"no version histories\", \"\")\n\t}\n\tcurrentVersionHistory, err := workflow.State.VersionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\t// should be impossible?\n\t\treturn nil, false, c.failed(\"unable to get current version history\", \"err: %v\", err)\n\t}\n\tcurrentBranchToken := currentVersionHistory.GetBranchToken()\n\tif len(currentBranchToken) == 0 {\n\t\t// should be impossible?\n\t\treturn nil, false, c.failed(\"no current branch token\", \"\")\n\t}\n\treturn currentBranchToken, true, result\n}\n\nfunc (c *staleWorkflowCheck) closeEventTime(workflow *persistence.GetWorkflowExecutionResponse, domainName string) (when time.Time, ok bool, result CheckResult) {\n\t// completion event is (sometimes?) nil, pull it from history instead.\n\n\t// strangely, close-event-time is not part of the workflow execution response.\n\t// pull the final event in history to figure it out.\n\tbranchToken, ok, result := c.getBranchToken(workflow)\n\tif !ok {\n\t\treturn when, false, result\n\t}\n\tlast, err := c.getLastEvent(branchToken, domainName)\n\tif err != nil {\n\t\treturn when, false, c.failed(\"failed to get last event\", err.Error())\n\t}\n\tif last == nil {\n\t\treturn when, false, c.failed(\"empty last event, should be impossible\", \"\") // successful totally-empty history seems impossible\n\t}\n\n\t// make extra sure it's a completion event.\n\t// all history events have a timestamp, but it needs to be a final-state event, not a random one.\n\tif !anyPresent(\n\t\tlast.WorkflowExecutionCanceledEventAttributes,\n\t\tlast.WorkflowExecutionCompletedEventAttributes,\n\t\tlast.WorkflowExecutionContinuedAsNewEventAttributes,\n\t\tlast.WorkflowExecutionFailedEventAttributes,\n\t\tlast.WorkflowExecutionTerminatedEventAttributes,\n\t\tlast.WorkflowExecutionTimedOutEventAttributes,\n\t) {\n\t\t// completion event may simply not exist due to incomplete replication.\n\t\t// this may be handled by history_exists or broken-history invariants, and we don't need to handle it here.\n\t\treturn when, false, c.failed(\"last event is not a completion-type\", \"missing data? got: %v\", last)\n\n\t}\n\n\tif last.Timestamp == nil {\n\t\treturn when, false, c.failed(\"last event has a nil timestamp\", \"bad data? got: %v\", last)\n\t}\n\t// sanity check, so far I have not seen any completion events in the top-level state.\n\t// I suspect it's only an old / cache field, but bail if it disagrees, just in case.\n\tif workflow.State.ExecutionInfo.CompletionEvent != nil &&\n\t\tworkflow.State.ExecutionInfo.CompletionEvent.Timestamp != nil &&\n\t\t*last.Timestamp != *workflow.State.ExecutionInfo.CompletionEvent.Timestamp {\n\t\treturn when, false, c.failed(\n\t\t\t\"workflow execution info and final event disagree\",\n\t\t\t\"bad data? got execution-completion-event timestamp: %v and final history event timestamp: %v\",\n\t\t\ttime.Unix(0, *last.Timestamp), time.Unix(0, *workflow.State.ExecutionInfo.CompletionEvent.Timestamp))\n\t}\n\tts := time.Unix(0, *last.Timestamp)\n\tif ok, result := c.checkTimeInSaneRange(ts, \"last event timestamp\"); !ok {\n\t\treturn when, false, result\n\t}\n\treturn ts, true, result\n}\n\nfunc anyPresent(items ...interface{}) bool {\n\tfor _, i := range items {\n\t\t// cannot use `i == nil` because typed nil pointers are boxed into non-nil interface{} due to having a type\n\t\tif !reflect.ValueOf(i).IsNil() {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc (c *staleWorkflowCheck) getLastEvent(branchToken []byte, domainName string) (*types.HistoryEvent, error) {\n\tconst (\n\t\tmaxHistoryLen = 250000 // our internal limits are much smaller, but it would be fine to raise this\n\t\tpageSize      = 1000   // multiple 1000/page limits elsewhere, also fine to change\n\t\tbatchTimeout  = 5 * time.Second\n\t)\n\tshard := c.pr.GetShardID()\n\titer := 0\n\tvar nextPageToken []byte\n\tvar lastEvent *types.HistoryEvent\n\tfor ; iter < (maxHistoryLen / pageSize); iter++ { // loose sanity check\n\t\tctx, cancel := context.WithTimeout(context.Background(), batchTimeout)\n\t\tdefer cancel() // revive:disable-line:defer clean up on panics, dups are just noops\n\n\t\thistory, err := c.pr.ReadHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\t\tBranchToken: branchToken,\n\t\t\t// only interested in the last event, but we have to read in order to get there.\n\t\t\tMinEventID:    constants.FirstEventID,\n\t\t\tMaxEventID:    constants.EndEventID,\n\t\t\tShardID:       &shard,\n\t\t\tPageSize:      pageSize,\n\t\t\tDomainName:    domainName,\n\t\t\tNextPageToken: nextPageToken,\n\t\t})\n\t\tcancel() // handle non-panics\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif len(history.HistoryEvents) > 0 {\n\t\t\t// should only be false if it didn't notice end-of-history in the previous batch (not sure if this happens or not)\n\t\t\tlastEvent = history.HistoryEvents[len(history.HistoryEvents)-1]\n\t\t}\n\t\tif len(history.NextPageToken) == 0 {\n\t\t\t// received last batch\n\t\t\treturn lastEvent, nil\n\t\t}\n\t\tnextPageToken = history.NextPageToken\n\t}\n\n\t// should be unreachable, assuming our limits work\n\treturn nil, fmt.Errorf(\"exceeded max history requests (%v), failing for branch token: %s\", iter, branchToken)\n}\n\nfunc (c *staleWorkflowCheck) getDomainInfo(info *persistence.WorkflowExecutionInfo) (retention int32, name string, err error) {\n\t// domain cache entries have private fields, hence this testable method is necessary to avoid using them\n\n\tdomain, err := c.dc.GetDomainByID(info.DomainID)\n\tif err != nil {\n\t\treturn -1, \"\", err\n\t}\n\tretentionNum := domain.GetRetentionDays(info.WorkflowID) // takes retention-sampling into account\n\treturn retentionNum, domain.GetInfo().Name, nil\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/stale_workflow_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc setup(t *testing.T, retentionDays int) (*persistence.MockRetryer, *staleWorkflowCheck) {\n\tctrl := gomock.NewController(t)\n\tpr := persistence.NewMockRetryer(ctrl)\n\tdc := cache.NewMockDomainCache(ctrl)\n\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t&persistence.DomainConfig{Retention: int32(retentionDays)},\n\t\ttrue,\n\t\tnil,\n\t\t0,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0)\n\tdc.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil).AnyTimes()\n\tdc.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\n\tinvariant := NewStaleWorkflow(pr, dc, testlogger.NewZap(t))\n\timpl, ok := invariant.(*staleWorkflowCheck)\n\tif !ok {\n\t\tt.Fatalf(\"NewStaleWorkflowVersion returned invariant of type %T but want *staleWorkflowCheck\", invariant)\n\t}\n\treturn pr, impl\n}\n\n// some acceptable defaults for tests that don't care about precise windows\nconst (\n\ttestRetentionDays              = 5\n\ttestRetentionAfterSafetyMargin = (time.Duration(testRetentionDays) * 24 * time.Hour) + retentionSafetyMargin\n\toneDay                         = 24 * time.Hour\n)\n\n// This test case validates the Check method for stale workflow invariant is doing the correct thing at high level.\n// Rest of the tests in this file covers the internals of CheckAge which is called by Check.\nfunc TestCheck(t *testing.T) {\n\ttests := []struct {\n\t\tdesc       string\n\t\texecution  any\n\t\tmockFn     func(*persistence.MockRetryer)\n\t\twantResult CheckResult\n\t}{\n\t\t{\n\t\t\tdesc: \"corrupted workflow past expiration\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(execution(domainID, running, time.Now().Add(-testRetentionAfterSafetyMargin-2*time.Hour), time.Hour, nil), nil).Times(1)\n\n\t\t\t\tpr.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t\tpr.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"stale running workflow\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"normal healthy workflow\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(execution(domainID, running, time.Now().Add(-2*time.Hour), time.Hour, nil), nil).Times(1)\n\n\t\t\t\tpr.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t\tpr.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"running workflow is within expiration window\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"zombie healthy workflow\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(execution(domainID, zombie, time.Now().Add(-2*time.Hour), time.Hour, nil), nil).Times(1)\n\n\t\t\t\tpr.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t\tpr.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"zombie workflow is within expiration window\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:      \"invalid execution object type\",\n\t\t\texecution: &entity.Timer{}, // invalid object type\n\t\t\tmockFn:    func(pr *persistence.MockRetryer) {},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"expected concrete execution\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"missing workflow id\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: \"\", // missing workflow id\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"missing critical data\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"get workflow execution failed\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, fmt.Errorf(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\tInfo:            \"failed to get concrete execution record\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tpr, impl := setup(t, testRetentionDays)\n\n\t\t\ttc.mockFn(pr)\n\n\t\t\tgotResult := impl.Check(context.Background(), tc.execution)\n\t\t\tgotResult.InfoDetails = \"\" // ignore details for now\n\t\t\tassert.Equal(t, tc.wantResult, gotResult)\n\t\t})\n\t}\n}\n\nfunc TestFix(t *testing.T) {\n\ttests := []struct {\n\t\tdesc       string\n\t\texecution  any\n\t\tmockFn     func(*persistence.MockRetryer)\n\t\twantResult FixResult\n\t}{\n\t\t{\n\t\t\tdesc: \"healthy workflow skipped\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(execution(domainID, running, time.Now().Add(-2*time.Hour), time.Hour, nil), nil).Times(1)\n\n\t\t\t\tpr.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t\tpr.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twantResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: \"stale_workflow\",\n\t\t\t\tInfo:          \"no need to fix: running workflow is within expiration window\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc: \"corrupted workflow deleted\",\n\t\t\texecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn: func(pr *persistence.MockRetryer) {\n\t\t\t\tpr.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(execution(domainID, running, time.Now().Add(-testRetentionAfterSafetyMargin-2*time.Hour), time.Hour, nil), nil).Times(1)\n\n\t\t\t\tpr.EXPECT().GetShardID().Return(123).Times(1)\n\t\t\t\tpr.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tpr.EXPECT().DeleteWorkflowExecution(gomock.Any(), &persistence.DeleteWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainName: domainName,\n\t\t\t\t}).Return(nil).Times(1)\n\n\t\t\t\tpr.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   domainID,\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t\tDomainName: domainName,\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\tInvariantName: \"stale_workflow\",\n\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   \"stale_workflow\",\n\t\t\t\t\tInfo:            \"stale running workflow\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tpr, impl := setup(t, testRetentionDays)\n\n\t\t\ttc.mockFn(pr)\n\n\t\t\tgotResult := impl.Fix(context.Background(), tc.execution)\n\t\t\tgotResult.CheckResult.InfoDetails = \"\" // ignore details for now\n\t\t\tgotResult.InfoDetails = \"\"             // ignore details for now\n\t\t\tassert.Equal(t, tc.wantResult, gotResult)\n\t\t})\n\t}\n}\n\nfunc TestStillRunning(t *testing.T) {\n\t// lots of similarity in these tests, abstract it a bit.\n\trun := func(created time.Time, timeout, backoff time.Duration) (pastExpiration bool, result CheckResult) {\n\t\t// sanity checks\n\t\tif timeout <= 0 {\n\t\t\tpanic(\"cannot have non-positive timeouts\")\n\t\t}\n\t\tif backoff < 0 {\n\t\t\tpanic(\"cannot have negative backoff\")\n\t\t}\n\t\tif time.Now().Before(created) {\n\t\t\tpanic(\"created cannot be in the future\")\n\t\t}\n\n\t\tpr, impl := setup(t, testRetentionDays)\n\t\twithOpenHistory(\n\t\t\tcreated,\n\t\t\tbackoff,\n\t\t\tpr,\n\t\t)\n\t\twf := execution(\n\t\t\tdomainID,\n\t\t\trunning,\n\t\t\tcreated,\n\t\t\ttimeout,\n\t\t\tnil,\n\t\t)\n\t\treturn impl.CheckAge(wf)\n\t}\n\tassertHealthy := func(t *testing.T, result CheckResult, isPastExpiration bool) {\n\t\tt.Logf(\"result: %#v\", result)\n\t\tassert.Equal(t, CheckResultTypeHealthy, result.CheckResultType, \"result should be healthy\")\n\t\tassert.False(t, isPastExpiration, \"workflow should not be past expiration\")\n\t}\n\n\tt.Run(\"healthy\", func(t *testing.T) {\n\t\tpastExpiration, result := run(\n\t\t\ttime.Now().Add(-time.Hour),\n\t\t\t2*time.Hour,\n\t\t\t0,\n\t\t)\n\n\t\t// still running, still within retention window, \"healthy\"\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"healthy-long-workflow\", func(t *testing.T) {\n\t\t// specifically: this workflow runs longer than the retention window, to make sure it's not prematurely removed\n\t\tpastExpiration, result := run(\n\t\t\ttime.Now().Add(-testRetentionAfterSafetyMargin).Add(-oneDay), // create 1 day before safety margin\n\t\t\ttestRetentionAfterSafetyMargin+(2*oneDay),                    // expires tomorrow\n\t\t\t0,\n\t\t)\n\n\t\t// still running, still within retention window, \"healthy\"\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"healthy-short-workflow-long-backoff\", func(t *testing.T) {\n\t\t// create+timeout alone should have cleaned this up by ~1 day, but backoff means it has not yet done anything\n\t\tpastExpiration, result := run(\n\t\t\ttime.Now().Add(-testRetentionAfterSafetyMargin).Add(-oneDay),\n\t\t\ttime.Minute, // with only create date, should have already been cleaned up\n\t\t\ttestRetentionAfterSafetyMargin+(2*oneDay), // but backoff means it has not actually \"started\" yet\n\t\t)\n\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"bad-running-but-retained\", func(t *testing.T) {\n\t\tpastExpiration, result := run(\n\t\t\ttime.Now().Add(-time.Hour),\n\t\t\ttime.Minute,\n\t\t\t0,\n\t\t)\n\n\t\t// still within the retention window, so it's treated as \"healthy\" for the purposes of this check,\n\t\t// even though it's \"corrupt\" because it should not still be running.\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"bad-running-and-corrupt\", func(t *testing.T) {\n\t\t// created a day before retention margin\n\t\tcreated := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-oneDay)\n\t\tpastExpiration, result := run(\n\t\t\tcreated,     // created a day before retention margin\n\t\t\ttime.Minute, // should have been deleted 23h59m ago at the latest\n\t\t\t0,\n\t\t)\n\n\t\t// should be cleaned up because it's simply too old to reasonably exist\n\t\tt.Logf(\"result: %#v\", result)\n\t\tassert.Equal(t, CheckResultTypeCorrupted, result.CheckResultType, \"result should be corrupted, as it has expired\")\n\t\tassert.True(t, pastExpiration, \"workflow should be past expiration, created: %v, retention days: %v + %v safety, now: %v\", created.Format(dateOnly), testRetentionDays, retentionSafetyMargin, time.Now().Format(dateOnly))\n\t})\n\tt.Run(\"bad-running-and-backoff-and-corrupt\", func(t *testing.T) {\n\t\t// created 2 days before retention margin\n\t\tcreated := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-(2 * oneDay))\n\t\tpastExpiration, result := run(\n\t\t\tcreated,\n\t\t\ttime.Minute,\n\t\t\toneDay, // with one day backoff + timeout, should have been cleaned 23h59m ago\n\t\t)\n\n\t\t// should be cleaned up because it's simply too old to reasonably exist\n\t\tt.Logf(\"result: %#v\", result)\n\t\tassert.Equal(t, CheckResultTypeCorrupted, result.CheckResultType, \"result should be corrupted, as it has expired\")\n\t\tassert.True(t, pastExpiration, \"workflow should be past expiration, created: %v, backoff: %v, retention days: %v + %v safety, now: %v\", created.Format(dateOnly), oneDay, testRetentionDays, retentionSafetyMargin, time.Now().Format(dateOnly))\n\n\t})\n}\n\nfunc TestPresent(t *testing.T) {\n\t// paranoia: nil pointers and interfaces combine in surprising ways in Go.\n\tvar s *string\n\tvar i *int\n\tassert.False(t, anyPresent(s, i), \"null pointers should not be present\")\n\n\tpresent := \"non null\"\n\tassert.True(t, anyPresent(s, i, &present), \"fully populated values should be present\")\n\n\tvar empty string\n\tassert.True(t, anyPresent(s, i, &empty), \"non-null empty values should still be present\")\n}\n\nfunc TestComplete(t *testing.T) {\n\t// lots of similarity in these tests, abstract it a bit.\n\trun := func(t *testing.T, created time.Time, timeout time.Duration, completed time.Time) (pastExpiration bool, result CheckResult) {\n\t\t// sanity checks\n\t\tif timeout <= 0 {\n\t\t\tpanic(\"cannot have non-positive timeouts\")\n\t\t}\n\t\tif completed.Before(created) {\n\t\t\tpanic(\"cannot complete before being created\")\n\t\t}\n\t\tif time.Now().Before(created) {\n\t\t\tpanic(\"created cannot be in the future\")\n\t\t}\n\t\tif created.Add(timeout).Before(completed) {\n\t\t\tpanic(\"timeout would have already completed this workflow\")\n\t\t}\n\n\t\tpr, impl := setup(t, testRetentionDays)\n\t\twithClosedHistory(&completed, pr)\n\n\t\t// randomize completion time vs nil, as both seem important.  log which for reproduction purposes.\n\t\tvar maybeCompleted *time.Time\n\t\tif rand.Intn(2) == 1 {\n\t\t\tt.Log(\"including completed time\")\n\t\t\tmaybeCompleted = &completed\n\t\t} else {\n\t\t\tt.Log(\"leaving completed time as nil\")\n\t\t}\n\n\t\twf := execution(domainID, closed, created, timeout, maybeCompleted)\n\n\t\treturn impl.CheckAge(wf)\n\t}\n\tassertHealthy := func(t *testing.T, result CheckResult, isPastExpiration bool) {\n\t\tt.Logf(\"result: %#v\", result)\n\t\tassert.Equal(t, CheckResultTypeHealthy, result.CheckResultType, \"result should be healthy\")\n\t\tassert.False(t, isPastExpiration, \"workflow should not be past expiration\")\n\t}\n\tassertCorrupt := func(t *testing.T, result CheckResult, isPastExpiration bool) {\n\t\tt.Logf(\"result: %#v\", result)\n\t\tassert.Equal(t, CheckResultTypeCorrupted, result.CheckResultType, \"result should be corrupted, as it has expired\")\n\t\tassert.True(t, isPastExpiration, \"workflow should be past expiration\")\n\t}\n\n\tt.Run(\"recently-complete\", func(t *testing.T) {\n\t\tstartTime := time.Now().Add(-time.Hour)\n\t\ttimeout := time.Hour\n\t\ttimeoutTime := startTime.Add(timeout)\n\t\tpastExpiration, result := run(t, startTime, timeout, timeoutTime)\n\n\t\t// started and completed well within retention\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"long-ago-complete\", func(t *testing.T) {\n\t\tstartTime := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-2 * oneDay)\n\t\ttimeout := time.Hour\n\t\ttimeoutTime := startTime.Add(timeout)\n\t\tpastExpiration, result := run(t, startTime, timeout, timeoutTime)\n\n\t\t// completed long ago, still around\n\t\tassertCorrupt(t, result, pastExpiration)\n\t})\n\tt.Run(\"long-ago-complete-but-not-yet-timeout\", func(t *testing.T) {\n\t\tstartTime := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-2 * oneDay) // same as above\n\t\ttimeout := time.Since(startTime) + oneDay                                     // workflow timeout would fire tomorrow\n\t\ttimeoutTime := startTime.Add(time.Hour)                                       // same as above\n\n\t\tpastExpiration, result := run(t, startTime, timeout, timeoutTime)\n\n\t\t// completed long ago, still around.  timeout would keep it present if it were considered.\n\t\tassertCorrupt(t, result, pastExpiration)\n\t})\n\tt.Run(\"long-ago-started-but-recently-complete\", func(t *testing.T) {\n\t\tstartTime := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-2 * oneDay) // same as above\n\t\ttimeout := time.Since(startTime) - oneDay                                     // workflow timeout occurred yesterday\n\t\ttimeoutTime := startTime.Add(timeout)                                         // and it timed out\n\n\t\tpastExpiration, result := run(t, startTime, timeout, timeoutTime)\n\n\t\t// complete, ran longer than retention, still present\n\t\tassertHealthy(t, result, pastExpiration)\n\t})\n\tt.Run(\"long-long-ago-started-and-long-complete\", func(t *testing.T) {\n\t\tstartTime := time.Now().Add(-(2 * testRetentionAfterSafetyMargin)).Add(-2 * oneDay) // ~2x older\n\t\ttimeout := 10 * testRetentionAfterSafetyMargin                                      // workflow timeout far in the future\n\t\ttimeoutTime := startTime.Add(testRetentionAfterSafetyMargin).Add(oneDay)            // completed retention+oneDay ago\n\n\t\tpastExpiration, result := run(t, startTime, timeout, timeoutTime)\n\n\t\t// ran longer than retention, completed long than retention ago, should be gone\n\t\tassertCorrupt(t, result, pastExpiration)\n\t})\n}\n\nfunc TestCompleteWithRunningFallback(t *testing.T) {\n\tpr, impl := setup(t, testRetentionDays)\n\tstartTime := time.Now().Add(-testRetentionAfterSafetyMargin).Add(-2 * oneDay) // started long ago\n\twithOpenHistoryFallback(startTime, 0, pr)\n\twf := execution(\n\t\tdomainID,\n\t\tclosed,\n\t\tstartTime,\n\t\ttime.Hour, // would have timed out 1 hour after starting, outside retention\n\t\tnil,       // no completed time value\n\t)\n\tpastExpiration, result := impl.CheckAge(wf)\n\n\tt.Logf(\"result: %#v\", result)\n\tassert.Equal(t, CheckResultTypeCorrupted, result.CheckResultType, \"result should be corrupted, as it has expired\")\n\tassert.True(t, pastExpiration, \"workflow should be past expiration\")\n\tassert.Contains(t, result.InfoDetails, \"completed workflow\", \"should have recognized it's completed\")\n\tassert.Contains(t, result.InfoDetails, \"treating as running\", \"should have mentioned that it has fallen back to running\")\n}\n\nfunc withClosedHistory(finish *time.Time, pr *persistence.MockRetryer) {\n\tpr.EXPECT().GetShardID().Return(0).MinTimes(1)\n\n\tfirstPage := &persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tTimestamp:                               nil, // unsure, probably should be start time, but currently unused\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t},\n\t\t\t{\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"not nil\"),\n\t}\n\tvar ts *int64 // always present from samples I've seen, and asserted\n\tif finish != nil {\n\t\tnano := finish.UnixNano()\n\t\tts = &nano\n\t}\n\tfinal := &persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tTimestamp: ts, // always present from samples I've seen, and asserted\n\t\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID: 3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tpr.EXPECT().ReadHistoryBranch(\n\t\tgomock.Any(),\n\t\trequestMaxEvent{constants.EndEventID, \"\"}, // not a first-event-only request\n\t).Return(firstPage, nil).Times(1)\n\tpr.EXPECT().ReadHistoryBranch(\n\t\tgomock.Any(),\n\t\trequestMaxEvent{constants.EndEventID, \"not nil\"}, // token from first page\n\t).Return(final, nil).Times(1)\n}\n\nfunc withOpenHistory(start time.Time, backoff time.Duration, pr *persistence.MockRetryer) {\n\tpr.EXPECT().GetShardID().Return(0).MinTimes(1)\n\n\tseconds := int32(backoff.Seconds())\n\tres := &persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tTimestamp: nil, // unsure, probably should be start time, but currently unused\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: &seconds,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tpr.EXPECT().ReadHistoryBranch(\n\t\tgomock.Any(),\n\t\trequestMaxEvent{2, \"\"}, // does not match a \"read all history\" / \"get last event\" request\n\t).Return(res, nil).Times(1)\n}\n\n// essentially incomplete-closed-then-open, but it's annoying to break them apart reusably so it's duplicated here\nfunc withOpenHistoryFallback(start time.Time, backoff time.Duration, pr *persistence.MockRetryer) {\n\tclosedHistory := &persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tTimestamp:                               nil, // unsure, probably should be start time, but currently unused\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t},\n\t\t\t{\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil, // premature end of history\n\t}\n\tpr.EXPECT().ReadHistoryBranch(\n\t\tgomock.Any(),\n\t\trequestMaxEvent{constants.EndEventID, \"\"}, // not a first-event-only request\n\t).Return(closedHistory, nil).Times(1)\n\n\t// and then it's just another open request, no need to customize\n\twithOpenHistory(start, backoff, pr)\n}\n\ntype requestMaxEvent struct {\n\tmax   int64\n\ttoken string\n}\n\nvar _ gomock.Matcher = requestMaxEvent{}\n\nfunc (m requestMaxEvent) Matches(x interface{}) bool {\n\treq, ok := x.(*persistence.ReadHistoryBranchRequest)\n\tif !ok {\n\t\treturn false\n\t}\n\tif req.MaxEventID > m.max {\n\t\treturn false // wrong max id\n\t}\n\t// nil and empty-array are both empty string, which is close enough\n\treturn m.token == string(req.NextPageToken)\n}\n\nfunc (m requestMaxEvent) String() string {\n\treturn fmt.Sprintf(\"max event ID must be <= %v and next page token must be %q\", m.max, m.token)\n}\n\ntype workflowstate int\n\n// minor type-safety hack, we should change these iotas\nconst (\n\trunning workflowstate = persistence.WorkflowStateRunning\n\tclosed  workflowstate = persistence.WorkflowStateCompleted\n\tzombie  workflowstate = persistence.WorkflowStateZombie\n)\n\nfunc execution(domainid string, state workflowstate, created time.Time, timeout time.Duration, completed *time.Time) *persistence.GetWorkflowExecutionResponse {\n\tres := &persistence.GetWorkflowExecutionResponse{\n\t\tState: &persistence.WorkflowMutableState{\n\t\t\t// needed for many things\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:        domainid,\n\t\t\t\tState:           int(state), /* untyped iota, persistence.WorkflowState* */\n\t\t\t\tStartTimestamp:  created,\n\t\t\t\tWorkflowTimeout: int32(timeout.Seconds()),\n\t\t\t},\n\t\t\t// needed for branch token\n\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t{\n\t\t\t\t\t\tBranchToken: []byte(\"fake-branch-token\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tif completed != nil {\n\t\t// not yet seen IRL, but asserted to match the last event timestamp in full history\n\t\tnano := completed.UnixNano()\n\t\tres.State.ExecutionInfo.CompletionEvent = &types.HistoryEvent{\n\t\t\tTimestamp: &nano,\n\t\t}\n\t}\n\treturn res\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/timer_invalid.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst TimerInvalidName = \"TimerInvalid\"\n\ntype TimerInvalid struct {\n\tpr    persistence.Retryer\n\tcache cache.DomainCache\n}\n\n// NewTimerInvalid returns a new history exists invariant\nfunc NewTimerInvalid(\n\tpr persistence.Retryer, cache cache.DomainCache,\n) Invariant {\n\treturn &TimerInvalid{\n\t\tpr:    pr,\n\t\tcache: cache,\n\t}\n}\n\n// Check checks if timer is scheduled for open execution\nfunc (h *TimerInvalid) Check(\n\tctx context.Context,\n\te interface{},\n) CheckResult {\n\tif checkResult := validateCheckContext(ctx, h.Name()); checkResult != nil {\n\t\treturn *checkResult\n\t}\n\n\ttimer, ok := e.(*entity.Timer)\n\n\tif !ok {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"failed to check: expected timer entity\",\n\t\t}\n\t}\n\tdomainID := timer.DomainID\n\tdomainName, err := h.cache.GetDomainName(timer.DomainID)\n\tif err != nil {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"failed to check: expected Domain Name\",\n\t\t}\n\t}\n\treq := &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: domainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: timer.WorkflowID,\n\t\t\tRunID:      timer.RunID,\n\t\t},\n\t\tDomainName: domainName,\n\t}\n\n\tresp, err := h.pr.GetWorkflowExecution(ctx, req)\n\n\tif err != nil {\n\t\tswitch err.(type) {\n\t\tcase *types.EntityNotExistsError:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   h.Name(),\n\t\t\t\tInfo:            \"timer scheduled for non existing workflow\",\n\t\t\t}\n\t\tdefault:\n\t\t\treturn CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   h.Name(),\n\t\t\t\tInfo:            \"failed to get workflow for timer\",\n\t\t\t}\n\t\t}\n\t}\n\n\tif !Open(resp.State.ExecutionInfo.State) {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   h.Name(),\n\t\t\tInfo:            \"timer scheduled for closed workflow\",\n\t\t}\n\t}\n\n\treturn CheckResult{\n\t\tCheckResultType: CheckResultTypeHealthy,\n\t\tInvariantName:   h.Name(),\n\t}\n}\n\n// Fix will delete invalid timer\nfunc (h *TimerInvalid) Fix(\n\tctx context.Context,\n\te interface{},\n) FixResult {\n\n\tif fixResult := validateFixContext(ctx, h.Name()); fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\tfixResult, checkResult := checkBeforeFix(ctx, h, e)\n\tif fixResult != nil {\n\t\treturn *fixResult\n\t}\n\n\ttimer, _ := e.(*entity.Timer)\n\n\tif timer.TaskType != persistence.TaskTypeUserTimer {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\tInvariantName: h.Name(),\n\t\t\tInfo:          \"timer is not a TaskTypeUserTimer\",\n\t\t}\n\t}\n\n\treq := persistence.CompleteHistoryTaskRequest{\n\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\tTaskKey: persistence.NewHistoryTaskKey(\n\t\t\ttimer.VisibilityTimestamp,\n\t\t\ttimer.TaskID,\n\t\t),\n\t}\n\n\tif err := h.pr.CompleteHistoryTask(ctx, &req); err != nil {\n\t\treturn FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: h.Name(),\n\t\t\tInfo:          err.Error(),\n\t\t}\n\t}\n\n\treturn FixResult{\n\t\tFixResultType: FixResultTypeFixed,\n\t\tInvariantName: h.Name(),\n\t\tCheckResult:   *checkResult,\n\t}\n}\n\nfunc (h *TimerInvalid) Name() Name {\n\treturn TimerInvalidName\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/timer_invalid_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype TimerInvalidTest struct {\n\tsuite.Suite\n}\n\nfunc TestTimerInvalidSuite(t *testing.T) {\n\tsuite.Run(t, new(TimerInvalidTest))\n}\n\nfunc (ts *TimerInvalidTest) TestCheck() {\n\ttestCases := []struct {\n\t\tname           string\n\t\tctxExpired     bool\n\t\tgetExecResp    *persistence.GetWorkflowExecutionResponse\n\t\tgetExecErr     error\n\t\texpectedResult CheckResult\n\t\tentity         interface{}\n\t}{\n\t\t{\n\t\t\tname:       \"Context expired\",\n\t\t\tctxExpired: true,\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\tInfo:            \"failed to check: context expired or cancelled\",\n\t\t\t\tInfoDetails:     \"context deadline exceeded\",\n\t\t\t},\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetExecErr:  errors.New(\"random error\"),\n\t\t\tentity:      &entity.Timer{},\n\t\t},\n\t\t{\n\t\t\tname: \"Check if entity is a timer\",\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\tInfo:            \"failed to check: expected timer entity\",\n\t\t\t\tInfoDetails:     \"\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Check for persistence error\",\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\tInfo:            \"failed to get workflow for timer\",\n\t\t\t\tInfoDetails:     \"\",\n\t\t\t},\n\t\t\tgetExecResp: nil,\n\t\t\tgetExecErr:  errors.New(\"random error\"),\n\t\t\tentity:      &entity.Timer{},\n\t\t},\n\t\t{\n\t\t\tname: \"Workflow not found in persistence\",\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\tInfo:            \"timer scheduled for non existing workflow\",\n\t\t\t\tInfoDetails:     \"\",\n\t\t\t},\n\t\t\tgetExecResp: nil,\n\t\t\tgetExecErr:  &types.EntityNotExistsError{},\n\t\t\tentity:      &entity.Timer{},\n\t\t},\n\t\t{\n\t\t\tname: \"Workflow is closed\",\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\tInfo:            \"timer scheduled for closed workflow\",\n\t\t\t\tInfoDetails:     \"\",\n\t\t\t},\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: closedState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetExecErr: nil,\n\t\t\tentity:     &entity.Timer{},\n\t\t},\n\t\t{\n\t\t\tname: \"Check passed\",\n\t\t\texpectedResult: CheckResult{\n\t\t\t\tCheckResultType: CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t},\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: openState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetExecErr: nil,\n\t\t\tentity:     &entity.Timer{},\n\t\t},\n\t}\n\tctrl := gomock.NewController(ts.T())\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\tts.Run(tc.name, func() {\n\t\t\texecManager := &mocks.ExecutionManager{}\n\t\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\t\ti := NewTimerInvalid(\n\t\t\t\tpersistence.NewPersistenceRetryer(\n\t\t\t\t\texecManager,\n\t\t\t\t\tnil,\n\t\t\t\t\tcommon.CreatePersistenceRetryPolicy(),\n\t\t\t\t),\n\t\t\t\tmockDomainCache,\n\t\t\t)\n\t\t\tctx := context.Background()\n\t\t\tif tc.ctxExpired {\n\t\t\t\tctx, _ = context.WithDeadline(ctx, time.Now())\n\t\t\t}\n\t\t\tresult := i.Check(ctx, tc.entity)\n\t\t\tts.Equal(tc.expectedResult, result)\n\t\t})\n\t}\n}\n\nfunc (ts *TimerInvalidTest) TestFix() {\n\ttestCases := []struct {\n\t\tname           string\n\t\tctxExpired     bool\n\t\tgetExecResp    *persistence.GetWorkflowExecutionResponse\n\t\tgetExecErr     error\n\t\texpectedResult FixResult\n\t\tentity         interface{}\n\t\tttComplete     error\n\t}{\n\t\t{\n\t\t\tname:       \"context expired\",\n\t\t\tctxExpired: true,\n\t\t\texpectedResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: \"TimerInvalid\",\n\t\t\t\tInfo:          \"failed to check: context expired or cancelled\",\n\t\t\t\tInfoDetails:   \"context deadline exceeded\",\n\t\t\t},\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetExecErr:  errors.New(\"random error\"),\n\t\t\tentity:      &entity.Timer{},\n\t\t},\n\t\t{\n\t\t\tname: \"check before fix fails bc it's not a timer\",\n\t\t\texpectedResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\t\tInfo:            \"failed to check: expected timer entity\",\n\t\t\t\t},\n\t\t\t\tInvariantName: \"TimerInvalid\",\n\t\t\t\tInfo:          \"failed fix because check failed\",\n\t\t\t\tInfoDetails:   \"\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timer type is not a user timer\",\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: closedState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\t\tInvariantName: \"TimerInvalid\",\n\t\t\t\tInfo:          \"timer is not a TaskTypeUserTimer\",\n\t\t\t\tInfoDetails:   \"\",\n\t\t\t},\n\t\t\tentity: &entity.Timer{\n\t\t\t\tTaskType: persistence.TaskTypeActivityRetryTimer,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timer deletion fails on persistence\",\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: closedState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInvariantName: \"TimerInvalid\",\n\t\t\t\tInfo:          \"error from persistence\",\n\t\t\t\tInfoDetails:   \"\",\n\t\t\t},\n\t\t\tentity: &entity.Timer{\n\t\t\t\tTaskType: persistence.TaskTypeUserTimer,\n\t\t\t},\n\t\t\tttComplete: errors.New(\"error from persistence\"),\n\t\t},\n\t\t{\n\t\t\tname: \"timer deleted\",\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: closedState,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: FixResult{\n\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t\tInvariantName: \"TimerInvalid\",\n\t\t\t\tInfo:          \"\",\n\t\t\t\tInfoDetails:   \"\",\n\t\t\t\tCheckResult: CheckResult{\n\t\t\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   \"TimerInvalid\",\n\t\t\t\t\tInfo:            \"timer scheduled for closed workflow\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tentity: &entity.Timer{\n\t\t\t\tTaskType: persistence.TaskTypeUserTimer,\n\t\t\t},\n\t\t\tttComplete: nil,\n\t\t},\n\t}\n\tctrl := gomock.NewController(ts.T())\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\tts.Run(tc.name, func() {\n\t\t\texecManager := &mocks.ExecutionManager{}\n\t\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\t\texecManager.On(\"CompleteHistoryTask\", mock.Anything, mock.Anything).Return(tc.ttComplete)\n\t\t\ti := NewTimerInvalid(\n\t\t\t\tpersistence.NewPersistenceRetryer(\n\t\t\t\t\texecManager,\n\t\t\t\t\tnil,\n\t\t\t\t\tcommon.CreatePersistenceRetryPolicy(),\n\t\t\t\t),\n\t\t\t\tmockDomainCache,\n\t\t\t)\n\t\t\tctx := context.Background()\n\t\t\tif tc.ctxExpired {\n\t\t\t\tctx, _ = context.WithDeadline(ctx, time.Now())\n\t\t\t}\n\t\t\tresult := i.Fix(ctx, tc.entity)\n\t\t\tts.Equal(tc.expectedResult, result)\n\t\t})\n\t}\n\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mocks.go -self_package github.com/uber/cadence/common/reconciliation/invariant\n//go:generate enumer -type=Collection -output collection_enumer_generated.go\n\npackage invariant\n\nimport \"context\"\n\nconst (\n\t// CheckResultTypeFailed indicates a failure occurred while attempting to run check\n\tCheckResultTypeFailed CheckResultType = \"failed\"\n\t// CheckResultTypeCorrupted indicates check successfully ran and detected a corruption\n\tCheckResultTypeCorrupted CheckResultType = \"corrupted\"\n\t// CheckResultTypeHealthy indicates check successfully ran and detected no corruption\n\tCheckResultTypeHealthy CheckResultType = \"healthy\"\n\n\t// FixResultTypeSkipped indicates that fix skipped execution\n\tFixResultTypeSkipped FixResultType = \"skipped\"\n\t// FixResultTypeFixed indicates that fix successfully fixed an execution\n\tFixResultTypeFixed FixResultType = \"fixed\"\n\t// FixResultTypeFailed indicates that fix attempted to fix an execution but failed to do so\n\tFixResultTypeFailed FixResultType = \"failed\"\n\n\t// HistoryExists asserts that history must exist if concrete execution exists\n\tHistoryExists Name = \"history_exists\"\n\n\t// InactiveDomainExists asserts that if domain status is not registered that it's inactive\n\tInactiveDomainExists Name = \"inactive_domain_exists\"\n\n\t// OpenCurrentExecution asserts that an open concrete execution must have a valid current execution\n\tOpenCurrentExecution Name = \"open_current_execution\"\n\t// ConcreteExecutionExists asserts that an open current execution must have a valid concrete execution\n\tConcreteExecutionExists Name = \"concrete_execution_exists\"\n\n\t// StaleWorkflow checks for workflows that exist beyond their retention window,\n\t// implying a failed cleanup / lost timers / etc of some kind.\n\tStaleWorkflow Name = \"stale_workflow\"\n\n\t// CollectionMutableState is the collection of invariants relating to mutable state\n\tCollectionMutableState Collection = 0\n\t// CollectionHistory is the collection  of invariants relating to history\n\tCollectionHistory Collection = 1\n\t// CollectionDomain is the collection  of invariants relating to domain status\n\tCollectionDomain Collection = 2\n\t// CollectionStale contains the stale workflow scanner\n\tCollectionStale Collection = 3\n)\n\ntype (\n\t// Name is the name of an invariant\n\tName string\n\n\t// Collection is a type which indicates a collection of invariants\n\tCollection int\n\n\t// CheckResultType is the result type of running an invariant check\n\tCheckResultType string\n\n\t// FixResultType is the result type of running an invariant fix\n\tFixResultType string\n)\n\n// Invariant represents an invariant of a single execution.\n// It can be used to check that the execution satisfies the invariant.\n// It can also be used to fix the invariant for an execution.\ntype Invariant interface {\n\tCheck(context.Context, interface{}) CheckResult\n\tFix(context.Context, interface{}) FixResult\n\tName() Name\n}\n\n// Manager represents a manager of several invariants.\n// It can be used to run a group of invariant checks or fixes.\ntype Manager interface {\n\tRunChecks(context.Context, interface{}) ManagerCheckResult\n\tRunFixes(context.Context, interface{}) ManagerFixResult\n}\n\n// ManagerCheckResult is the result of running a list of checks\ntype ManagerCheckResult struct {\n\tCheckResultType          CheckResultType\n\tDeterminingInvariantType *Name\n\tCheckResults             []CheckResult\n}\n\n// ManagerFixResult is the result of running a list of fixes\ntype ManagerFixResult struct {\n\tFixResultType            FixResultType\n\tDeterminingInvariantName *Name\n\tFixResults               []FixResult\n}\n\n// CheckResult is the result of running Check.\ntype CheckResult struct {\n\tCheckResultType CheckResultType\n\tInvariantName   Name\n\tInfo            string\n\tInfoDetails     string\n}\n\n// FixResult is the result of running Fix.\ntype FixResult struct {\n\tFixResultType FixResultType\n\tInvariantName Name\n\tCheckResult   CheckResult\n\tInfo          string\n\tInfoDetails   string\n}\n\n// NamePtr returns a pointer to Name\nfunc NamePtr(t Name) *Name {\n\treturn &t\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/util.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\nfunc checkBeforeFix(\n\tctx context.Context,\n\tinvariant Invariant,\n\texecution interface{},\n) (*FixResult, *CheckResult) {\n\tcheckResult := invariant.Check(ctx, execution)\n\tif checkResult.CheckResultType == CheckResultTypeHealthy {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeSkipped,\n\t\t\tInvariantName: invariant.Name(),\n\t\t\tCheckResult:   checkResult,\n\t\t\tInfo:          \"skipped fix because execution was healthy\",\n\t\t}, nil\n\t}\n\tif checkResult.CheckResultType == CheckResultTypeFailed {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: invariant.Name(),\n\t\t\tCheckResult:   checkResult,\n\t\t\tInfo:          \"failed fix because check failed\",\n\t\t}, nil\n\t}\n\treturn nil, &checkResult\n}\n\n// Open returns true if workflow state is open false if workflow is closed\nfunc Open(state int) bool {\n\treturn state == persistence.WorkflowStateCreated || state == persistence.WorkflowStateRunning\n}\n\n// ExecutionOpen returns true if execution state is open false if workflow is closed\nfunc ExecutionOpen(execution interface{}) bool {\n\treturn Open(getExecution(execution).State)\n}\n\n// getExecution returns base Execution\nfunc getExecution(execution interface{}) *entity.Execution {\n\tswitch e := execution.(type) {\n\tcase *entity.CurrentExecution:\n\t\treturn &e.Execution\n\tcase *entity.ConcreteExecution:\n\t\treturn &e.Execution\n\tdefault:\n\t\tpanic(\"unexpected execution type\")\n\t}\n}\n\n// DeleteExecution deletes concrete execution and\n// current execution conditionally on matching runID.\nfunc DeleteExecution(\n\tctx context.Context,\n\texec interface{},\n\tpr persistence.Retryer,\n\tdc cache.DomainCache,\n) *FixResult {\n\texecution := getExecution(exec)\n\tdomainName, errorDomainName := dc.GetDomainName(execution.DomainID)\n\tif errorDomainName != nil {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInfo:          \"failed to fetch domainName\",\n\t\t\tInfoDetails:   errorDomainName.Error(),\n\t\t}\n\t}\n\tif err := pr.DeleteWorkflowExecution(ctx, &persistence.DeleteWorkflowExecutionRequest{\n\t\tDomainID:   execution.DomainID,\n\t\tWorkflowID: execution.WorkflowID,\n\t\tRunID:      execution.RunID,\n\t\tDomainName: domainName,\n\t}); err != nil {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInfo:          \"failed to delete concrete workflow execution\",\n\t\t\tInfoDetails:   err.Error(),\n\t\t}\n\t}\n\tif err := pr.DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\tDomainID:   execution.DomainID,\n\t\tWorkflowID: execution.WorkflowID,\n\t\tRunID:      execution.RunID,\n\t\tDomainName: domainName,\n\t}); err != nil {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInfo:          \"failed to delete current workflow execution\",\n\t\t\tInfoDetails:   err.Error(),\n\t\t}\n\t}\n\treturn &FixResult{\n\t\tFixResultType: FixResultTypeFixed,\n\t}\n}\n\nfunc validateCheckContext(\n\tctx context.Context,\n\tinvariantName Name,\n) *CheckResult {\n\tif ctxErr := ctx.Err(); ctxErr != nil {\n\t\treturn &CheckResult{\n\t\t\tCheckResultType: CheckResultTypeFailed,\n\t\t\tInvariantName:   invariantName,\n\t\t\tInfo:            \"failed to check: context expired or cancelled\",\n\t\t\tInfoDetails:     ctxErr.Error(),\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc validateFixContext(\n\tctx context.Context,\n\tinvariantName Name,\n) *FixResult {\n\tif ctxErr := ctx.Err(); ctxErr != nil {\n\t\treturn &FixResult{\n\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\tInvariantName: invariantName,\n\t\t\tInfo:          \"failed to check: context expired or cancelled\",\n\t\t\tInfoDetails:   ctxErr.Error(),\n\t\t}\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "common/reconciliation/invariant/util_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype UtilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(UtilSuite))\n}\n\nfunc (s *UtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *UtilSuite) TestDeleteExecution() {\n\ttestCases := []struct {\n\t\tdeleteConcreteErr error\n\t\tdeleteCurrentErr  error\n\t\texpectedFixResult *FixResult\n\t}{\n\t\t{\n\t\t\tdeleteConcreteErr: errors.New(\"error deleting concrete execution\"),\n\t\t\texpectedFixResult: &FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInfo:          \"failed to delete concrete workflow execution\",\n\t\t\t\tInfoDetails:   \"error deleting concrete execution\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdeleteCurrentErr: errors.New(\"error deleting current execution\"),\n\t\t\texpectedFixResult: &FixResult{\n\t\t\t\tFixResultType: FixResultTypeFailed,\n\t\t\t\tInfo:          \"failed to delete current workflow execution\",\n\t\t\t\tInfoDetails:   \"error deleting current execution\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\texpectedFixResult: &FixResult{\n\t\t\t\tFixResultType: FixResultTypeFixed,\n\t\t\t},\n\t\t},\n\t}\n\tctrl := gomock.NewController(s.T())\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\texecManager.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything, mock.Anything).Return(tc.deleteConcreteErr).Once()\n\t\tif tc.deleteConcreteErr == nil {\n\t\t\texecManager.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.deleteCurrentErr).Once()\n\t\t}\n\t\tpr := persistence.NewPersistenceRetryer(execManager, nil, common.CreatePersistenceRetryPolicy())\n\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\t\tresult := DeleteExecution(context.Background(), &entity.ConcreteExecution{}, pr, mockDomainCache)\n\t\ts.Equal(tc.expectedFixResult, result)\n\t}\n}\n\nfunc (s *UtilSuite) TestExecutionStillOpen() {\n\ttestCases := []struct {\n\t\tgetExecResp *persistence.GetWorkflowExecutionResponse\n\t\tgetExecErr  error\n\t\texpectError bool\n\t\texpectOpen  bool\n\t}{\n\t\t{\n\t\t\tgetExecResp: nil,\n\t\t\tgetExecErr:  &types.EntityNotExistsError{},\n\t\t\texpectError: false,\n\t\t\texpectOpen:  false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp: nil,\n\t\t\tgetExecErr:  errors.New(\"got error\"),\n\t\t\texpectError: true,\n\t\t\texpectOpen:  false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetExecErr:  nil,\n\t\t\texpectError: false,\n\t\t\texpectOpen:  false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tgetExecErr:  nil,\n\t\t\texpectError: false,\n\t\t\texpectOpen:  true,\n\t\t},\n\t}\n\tctrl := gomock.NewController(s.T())\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\tpr := persistence.NewPersistenceRetryer(execManager, nil, common.CreatePersistenceRetryPolicy())\n\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil)\n\t\topen, err := ExecutionStillOpen(context.Background(), &entity.Execution{}, pr, mockDomainCache)\n\t\tif tc.expectError {\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t\tif tc.expectOpen {\n\t\t\ts.True(open)\n\t\t} else {\n\t\t\ts.False(open)\n\t\t}\n\t}\n}\n\nfunc (s *UtilSuite) TestExecutionStillExists() {\n\ttestCases := []struct {\n\t\tgetExecResp  *persistence.GetWorkflowExecutionResponse\n\t\tgetExecErr   error\n\t\texpectError  bool\n\t\texpectExists bool\n\t}{\n\t\t{\n\t\t\tgetExecResp:  &persistence.GetWorkflowExecutionResponse{},\n\t\t\tgetExecErr:   nil,\n\t\t\texpectError:  false,\n\t\t\texpectExists: true,\n\t\t},\n\t\t{\n\t\t\tgetExecResp:  nil,\n\t\t\tgetExecErr:   &types.EntityNotExistsError{},\n\t\t\texpectError:  false,\n\t\t\texpectExists: false,\n\t\t},\n\t\t{\n\t\t\tgetExecResp:  nil,\n\t\t\tgetExecErr:   errors.New(\"got error\"),\n\t\t\texpectError:  true,\n\t\t\texpectExists: false,\n\t\t},\n\t}\n\tctrl := gomock.NewController(s.T())\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tfor _, tc := range testCases {\n\t\texecManager := &mocks.ExecutionManager{}\n\t\texecManager.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(tc.getExecResp, tc.getExecErr)\n\t\tpr := persistence.NewPersistenceRetryer(execManager, nil, common.CreatePersistenceRetryPolicy())\n\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil)\n\t\texists, err := ExecutionStillExists(context.Background(), &entity.Execution{}, pr, mockDomainCache)\n\t\tif tc.expectError {\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t\tif tc.expectExists {\n\t\t\ts.True(exists)\n\t\t} else {\n\t\t\ts.False(exists)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/store/blobstoreIterator.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage store\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n)\n\ntype (\n\tblobstoreIterator struct {\n\t\titr pagination.Iterator\n\t}\n)\n\n// NewBlobstoreIterator constructs a new iterator backed by blobstore.\nfunc NewBlobstoreIterator(\n\tctx context.Context,\n\tclient blobstore.Client,\n\tkeys Keys,\n\tentity entity.Entity,\n) ScanOutputIterator {\n\treturn &blobstoreIterator{\n\t\titr: pagination.NewIterator(ctx, keys.MinPage, getBlobstoreFetchPageFn(client, keys, entity)),\n\t}\n}\n\n// Next returns the next ScanOutputEntity\nfunc (i *blobstoreIterator) Next() (*ScanOutputEntity, error) {\n\texec, err := i.itr.Next()\n\tif exec != nil {\n\t\treturn exec.(*ScanOutputEntity), err\n\t}\n\treturn nil, err\n}\n\n// HasNext returns true if there is a next ScanOutputEntity false otherwise\nfunc (i *blobstoreIterator) HasNext() bool {\n\treturn i.itr.HasNext()\n}\n\nfunc getBlobstoreFetchPageFn(\n\tclient blobstore.Client,\n\tkeys Keys,\n\tentity entity.Entity,\n) pagination.FetchFn {\n\treturn func(ctx context.Context, token pagination.PageToken) (pagination.Page, error) {\n\t\tindex := token.(int)\n\t\tkey := pageNumberToKey(keys.UUID, keys.Extension, index)\n\t\treq := &blobstore.GetRequest{\n\t\t\tKey: key,\n\t\t}\n\t\tresp, err := client.Get(ctx, req)\n\t\tif err != nil {\n\t\t\treturn pagination.Page{}, err\n\t\t}\n\t\tparts := bytes.Split(resp.Blob.Body, SeparatorToken)\n\t\tvar executions []pagination.Entity\n\t\tfor _, p := range parts {\n\t\t\tif len(p) == 0 {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tsoe, err := deserialize(p, entity)\n\t\t\tif err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\t\t\texecutions = append(executions, soe)\n\t\t}\n\t\tvar nextPageToken interface{} = index + 1\n\t\tif nextPageToken.(int) > keys.MaxPage {\n\t\t\tnextPageToken = nil\n\t\t}\n\t\treturn pagination.Page{\n\t\t\tCurrentToken: token,\n\t\t\tNextToken:    nextPageToken,\n\t\t\tEntities:     executions,\n\t\t}, nil\n\t}\n}\n\nfunc deserialize(data []byte, blob entity.Entity) (*ScanOutputEntity, error) {\n\tsoe := &ScanOutputEntity{\n\t\tExecution: blob.Clone(),\n\t}\n\n\tif err := json.Unmarshal(data, soe); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := soe.Execution.(entity.Entity).Validate(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn soe, nil\n}\n"
  },
  {
    "path": "common/reconciliation/store/blobstoreWriter.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage store\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/pagination\"\n)\n\nconst (\n\tmaxRetries        = 3\n\tinitialRetryDelay = 2 * time.Second  // Initial delay between retries\n\tmaxRetryDelay     = 30 * time.Second // Maximum delay between retries\n)\n\ntype (\n\tblobstoreWriter struct {\n\t\twriter    pagination.Writer\n\t\tuuid      string\n\t\textension Extension\n\t}\n)\n\n// NewBlobstoreWriter constructs a new blobstore writer\nfunc NewBlobstoreWriter(\n\tuuid string,\n\textension Extension,\n\tclient blobstore.Client,\n\tflushThreshold int,\n) ExecutionWriter {\n\t// Set a longer expiration interval than timeout for the entire retry process\n\ttotalRetryDuration := 2 * Timeout\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(initialRetryDelay)\n\tretryPolicy.SetMaximumInterval(maxRetryDelay)\n\tretryPolicy.SetExpirationInterval(totalRetryDuration)\n\t// Setting the attempts to 3 as a precaution. If we don't see any significant latency we can remove this config.\n\tretryPolicy.SetMaximumAttempts(maxRetries)\n\n\tthrottlePolicy := backoff.NewExponentialRetryPolicy(initialRetryDelay)\n\tthrottlePolicy.SetMaximumInterval(maxRetryDelay)\n\tthrottlePolicy.SetExpirationInterval(totalRetryDuration)\n\n\treturn &blobstoreWriter{\n\t\twriter: pagination.NewWriter(\n\t\t\tgetBlobstoreWriteFn(uuid, extension, client, retryPolicy, throttlePolicy),\n\t\t\tgetBlobstoreShouldFlushFn(flushThreshold),\n\t\t\t0),\n\t\tuuid:      uuid,\n\t\textension: extension,\n\t}\n}\n\n// Add adds an entity to blobstore writer\nfunc (bw *blobstoreWriter) Add(e interface{}) error {\n\treturn bw.writer.Add(e)\n}\n\n// Flush flushes contents of writer to blobstore.\n// Only triggers flush if page contains some contents.\nfunc (bw *blobstoreWriter) Flush() error {\n\treturn bw.writer.FlushIfNotEmpty()\n}\n\n// FlushedKeys returns the keys that have been successfully flushed.\n// Returns nil if no keys have been flushed.\nfunc (bw *blobstoreWriter) FlushedKeys() *Keys {\n\tif len(bw.writer.FlushedPages()) == 0 {\n\t\treturn nil\n\t}\n\treturn &Keys{\n\t\tUUID:      bw.uuid,\n\t\tMinPage:   bw.writer.FirstFlushedPage().(int),\n\t\tMaxPage:   bw.writer.LastFlushedPage().(int),\n\t\tExtension: bw.extension,\n\t}\n}\n\nfunc getBlobstoreWriteFn(\n\tuuid string,\n\textension Extension,\n\tclient blobstore.Client,\n\tretryPolicy backoff.RetryPolicy,\n\tthrottlePolicy backoff.RetryPolicy,\n) pagination.WriteFn {\n\treturn func(page pagination.Page) (pagination.PageToken, error) {\n\t\tblobIndex := page.CurrentToken.(int)\n\t\tkey := pageNumberToKey(uuid, extension, blobIndex)\n\t\tbuffer := &bytes.Buffer{}\n\t\tfor _, e := range page.Entities {\n\t\t\tdata, err := json.Marshal(e)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tbuffer.Write(data)\n\t\t\tbuffer.Write(SeparatorToken)\n\t\t}\n\t\treq := &blobstore.PutRequest{\n\t\t\tKey: key,\n\t\t\tBlob: blobstore.Blob{\n\t\t\t\tBody: buffer.Bytes(),\n\t\t\t},\n\t\t}\n\n\t\toperation := func(ctx context.Context) error {\n\t\t\tctx, cancel := context.WithTimeout(ctx, Timeout)\n\t\t\tdefer cancel()\n\t\t\t_, err := client.Put(ctx, req)\n\t\t\treturn err\n\t\t}\n\t\t// Using the ThrottleRetry struct and its Do method to implement the retry logic in the getBlobstoreWriteFn.\n\t\t// This struct offers a way to retry operations with a specified policy and also to throttle retries if necessary.\n\t\tthrottleRetry := backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\t\tbackoff.WithThrottlePolicy(throttlePolicy),\n\t\t\tbackoff.WithRetryableError(func(err error) bool {\n\t\t\t\treturn true // assuming all errors are retryable\n\t\t\t}),\n\t\t)\n\n\t\t// The Do method of throttleRetry is used to execute the operation with retries according to the policy.\n\t\terr := throttleRetry.Do(context.Background(), operation)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn blobIndex + 1, nil\n\t}\n}\n\nfunc getBlobstoreShouldFlushFn(\n\tflushThreshold int,\n) pagination.ShouldFlushFn {\n\treturn func(page pagination.Page) bool {\n\t\treturn len(page.Entities) > flushThreshold\n\t}\n}\n\nfunc pageNumberToKey(uuid string, extension Extension, pageNum int) string {\n\treturn fmt.Sprintf(\"%v_%v.%v\", uuid, pageNum, extension)\n}\n"
  },
  {
    "path": "common/reconciliation/store/blobstorewriter_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage store\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/blobstore/filestore\"\n\t\"github.com/uber/cadence/common/config\"\n)\n\nfunc TestBlobstoreWriter(t *testing.T) {\n\ttype testCase struct {\n\t\tname        string\n\t\tinput       string\n\t\tinput2      string\n\t\texpectedErr bool\n\t}\n\n\ttestCases := []testCase{\n\t\t{\n\t\t\tname:        \"Normal case\",\n\t\t\tinput:       \"one\",\n\t\t\tinput2:      \"two\",\n\t\t\texpectedErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tuuid := \"test-uuid\"\n\t\t\textension := Extension(\"test\")\n\t\t\toutputDir := t.TempDir()\n\n\t\t\tcfg := &config.FileBlobstore{\n\t\t\t\tOutputDirectory: outputDir,\n\t\t\t}\n\t\t\t// Reusing the FilestoreClient from the other sister test in the same package.\n\t\t\tblobstoreClient, err := filestore.NewFilestoreClient(cfg)\n\t\t\trequire.NoError(t, err)\n\n\t\t\tblobstoreWriter := NewBlobstoreWriter(uuid, extension, blobstoreClient, 10).(*blobstoreWriter)\n\t\t\t// Add data to the writer\n\t\t\terr = blobstoreWriter.Add(tc.input)\n\t\t\terr = blobstoreWriter.Add(tc.input2)\n\t\t\tif tc.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tassert.NoError(t, err)\n\n\t\t\t// Flush the writer to write data to the blobstore\n\t\t\terr = blobstoreWriter.Flush()\n\t\t\tassert.NoError(t, err)\n\n\t\t\t// Retrieve the keys of flushed data\n\t\t\tflushedKeys := blobstoreWriter.FlushedKeys()\n\t\t\tif flushedKeys == nil {\n\t\t\t\tt.Fatal(\"Expected flushedKeys to be not nil\")\n\t\t\t}\n\n\t\t\t// Read back the data from the blobstore\n\t\t\tkey := pageNumberToKey(uuid, extension, flushedKeys.MinPage)\n\t\t\treq := &blobstore.GetRequest{Key: key}\n\t\t\tctx := context.Background()\n\t\t\tresp, err := blobstoreClient.Get(ctx, req)\n\t\t\tassert.NoError(t, err)\n\n\t\t\t// Verify the contents\n\t\t\tassert.Equal(t, string(resp.Blob.Body), \"\\\"one\\\"\\r\\n\\\"two\\\"\\r\\n\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/reconciliation/store/mocks.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: types.go\n//\n// Generated by this command:\n//\n//\tmockgen -package store -source types.go -destination mocks.go -self_package github.com/uber/cadence/common/reconciliation/store\n//\n\n// Package store is a generated GoMock package.\npackage store\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockScanOutputIterator is a mock of ScanOutputIterator interface.\ntype MockScanOutputIterator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockScanOutputIteratorMockRecorder\n\tisgomock struct{}\n}\n\n// MockScanOutputIteratorMockRecorder is the mock recorder for MockScanOutputIterator.\ntype MockScanOutputIteratorMockRecorder struct {\n\tmock *MockScanOutputIterator\n}\n\n// NewMockScanOutputIterator creates a new mock instance.\nfunc NewMockScanOutputIterator(ctrl *gomock.Controller) *MockScanOutputIterator {\n\tmock := &MockScanOutputIterator{ctrl: ctrl}\n\tmock.recorder = &MockScanOutputIteratorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockScanOutputIterator) EXPECT() *MockScanOutputIteratorMockRecorder {\n\treturn m.recorder\n}\n\n// HasNext mocks base method.\nfunc (m *MockScanOutputIterator) HasNext() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasNext\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasNext indicates an expected call of HasNext.\nfunc (mr *MockScanOutputIteratorMockRecorder) HasNext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasNext\", reflect.TypeOf((*MockScanOutputIterator)(nil).HasNext))\n}\n\n// Next mocks base method.\nfunc (m *MockScanOutputIterator) Next() (*ScanOutputEntity, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Next\")\n\tret0, _ := ret[0].(*ScanOutputEntity)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Next indicates an expected call of Next.\nfunc (mr *MockScanOutputIteratorMockRecorder) Next() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Next\", reflect.TypeOf((*MockScanOutputIterator)(nil).Next))\n}\n\n// MockExecutionWriter is a mock of ExecutionWriter interface.\ntype MockExecutionWriter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutionWriterMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutionWriterMockRecorder is the mock recorder for MockExecutionWriter.\ntype MockExecutionWriterMockRecorder struct {\n\tmock *MockExecutionWriter\n}\n\n// NewMockExecutionWriter creates a new mock instance.\nfunc NewMockExecutionWriter(ctrl *gomock.Controller) *MockExecutionWriter {\n\tmock := &MockExecutionWriter{ctrl: ctrl}\n\tmock.recorder = &MockExecutionWriterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutionWriter) EXPECT() *MockExecutionWriterMockRecorder {\n\treturn m.recorder\n}\n\n// Add mocks base method.\nfunc (m *MockExecutionWriter) Add(arg0 any) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Add\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Add indicates an expected call of Add.\nfunc (mr *MockExecutionWriterMockRecorder) Add(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Add\", reflect.TypeOf((*MockExecutionWriter)(nil).Add), arg0)\n}\n\n// Flush mocks base method.\nfunc (m *MockExecutionWriter) Flush() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Flush\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Flush indicates an expected call of Flush.\nfunc (mr *MockExecutionWriterMockRecorder) Flush() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Flush\", reflect.TypeOf((*MockExecutionWriter)(nil).Flush))\n}\n\n// FlushedKeys mocks base method.\nfunc (m *MockExecutionWriter) FlushedKeys() *Keys {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FlushedKeys\")\n\tret0, _ := ret[0].(*Keys)\n\treturn ret0\n}\n\n// FlushedKeys indicates an expected call of FlushedKeys.\nfunc (mr *MockExecutionWriterMockRecorder) FlushedKeys() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FlushedKeys\", reflect.TypeOf((*MockExecutionWriter)(nil).FlushedKeys))\n}\n"
  },
  {
    "path": "common/reconciliation/store/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mocks.go -self_package github.com/uber/cadence/common/reconciliation/store\n\npackage store\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n)\n\nconst (\n\t// SkippedExtension is the extension for files which contain skips\n\tSkippedExtension Extension = \"skipped\"\n\t// FailedExtension is the extension for files which contain failures\n\tFailedExtension Extension = \"failed\"\n\t// FixedExtension is the extension for files which contain fixes\n\tFixedExtension Extension = \"fixed\"\n\t// CorruptedExtension is the extension for files which contain corruptions\n\tCorruptedExtension Extension = \"corrupted\"\n)\n\nvar (\n\t// SeparatorToken is used to separate entries written to blobstore\n\tSeparatorToken = []byte(\"\\r\\n\")\n\t// Timeout is the timeout used for blobstore requests\n\tTimeout = time.Second * 10\n)\n\ntype (\n\t// Extension is the type which indicates the file extension type\n\tExtension string\n\n\t// Keys indicate the keys which were uploaded during a scan or fix.\n\t// Keys are constructed as uuid_page.extension. MinPage and MaxPage are\n\t// both inclusive and pages are sequential, meaning from this struct all pages can be deterministically constructed.\n\tKeys struct {\n\t\tUUID      string\n\t\tMinPage   int\n\t\tMaxPage   int\n\t\tExtension Extension\n\t}\n)\n\n// The following are serializable types which get output by Scan and Fix to durable sinks.\ntype (\n\t// ScanOutputEntity represents a single execution that should be durably recorded by Scan.\n\tScanOutputEntity struct {\n\t\tExecution interface{}\n\t\tResult    invariant.ManagerCheckResult\n\t}\n\n\t// FixOutputEntity represents a single execution that should be durably recorded by fix.\n\t// It contains the ScanOutputEntity that was given as input to fix.\n\tFixOutputEntity struct {\n\t\tExecution interface{}\n\t\tInput     ScanOutputEntity\n\t\tResult    invariant.ManagerFixResult\n\t}\n)\n\ntype (\n\t// ScanOutputIterator gets ScanOutputEntities from underlying store\n\tScanOutputIterator interface {\n\t\t// Next returns the next ScanOutputEntity found. Any error reading from underlying store\n\t\t// or converting store entry to ScanOutputEntity will result in an error after which iterator cannot be used.\n\t\tNext() (*ScanOutputEntity, error)\n\t\t// HasNext indicates if the iterator has a next element. If HasNext is true it is\n\t\t// guaranteed that Next will return a nil error and non-nil ScanOutputEntity.\n\t\tHasNext() bool\n\t}\n\n\t// ExecutionWriter is used to write entities (FixOutputEntity or ScanOutputEntity) to blobstore\n\tExecutionWriter interface {\n\t\tAdd(interface{}) error\n\t\tFlush() error\n\t\tFlushedKeys() *Keys\n\t}\n)\n"
  },
  {
    "path": "common/reconciliation/store/writerIterator_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\npackage store\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/blobstore/filestore\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/fetcher\"\n)\n\nvar (\n\tvalidBranchToken  = []byte{89, 11, 0, 10, 0, 0, 0, 12, 116, 101, 115, 116, 45, 116, 114, 101, 101, 45, 105, 100, 11, 0, 20, 0, 0, 0, 14, 116, 101, 115, 116, 45, 98, 114, 97, 110, 99, 104, 45, 105, 100, 0}\n\texecutionPageSize = 10\n\ttestShardID       = 1\n)\n\nfunc TestWriterIterator(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tpages         int\n\t\tcountPerPage  int\n\t\texpectedCount int // Add this field\n\t}{\n\t\t{\"StandardCase\", 10, 10, 100},\n\t\t{\"FewPages\", 3, 15, 45}, // Set expectedCount as pages * countPerPage\n\t\t{\"SingleLargePage\", 1, 100, 100},\n\t\t{\"ManySmallPages\", 20, 2, 40},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassertions := require.New(t)\n\t\t\tpr := persistence.NewPersistenceRetryer(getMockExecutionManager(tc.pages, tc.countPerPage), nil, common.CreatePersistenceRetryPolicy())\n\t\t\tpItr := fetcher.ConcreteExecutionIterator(context.Background(), pr, executionPageSize)\n\n\t\t\tuuid := \"uuid\"\n\t\t\textension := Extension(\"test\")\n\t\t\toutputDir := t.TempDir()\n\t\t\tcfg := &config.FileBlobstore{\n\t\t\t\tOutputDirectory: outputDir,\n\t\t\t}\n\t\t\tblobstore, err := filestore.NewFilestoreClient(cfg)\n\t\t\tassertions.NoError(err)\n\t\t\tblobstoreWriter := NewBlobstoreWriter(uuid, extension, blobstore, 10)\n\n\t\t\tvar outputs []*ScanOutputEntity\n\t\t\tfor pItr.HasNext() {\n\t\t\t\texec, err := pItr.Next()\n\t\t\t\tassertions.NoError(err)\n\t\t\t\tsoe := &ScanOutputEntity{\n\t\t\t\t\tExecution: exec,\n\t\t\t\t}\n\t\t\t\toutputs = append(outputs, soe)\n\t\t\t\tassertions.NoError(blobstoreWriter.Add(soe))\n\t\t\t}\n\t\t\tassertions.NoError(blobstoreWriter.Flush())\n\t\t\tassertions.Len(outputs, tc.pages*tc.countPerPage)\n\t\t\tassertions.False(pItr.HasNext())\n\t\t\t_, err = pItr.Next()\n\t\t\tassertions.Equal(pagination.ErrIteratorFinished, err)\n\t\t\tflushedKeys := blobstoreWriter.FlushedKeys()\n\t\t\tassertions.Equal(uuid, flushedKeys.UUID)\n\t\t\tassertions.Equal(0, flushedKeys.MinPage)\n\t\t\tif tc.expectedCount > 0 {\n\t\t\t\tassertions.Equal((tc.expectedCount-1)/10, flushedKeys.MaxPage)\n\t\t\t} else {\n\t\t\t\tassertions.Equal(0, flushedKeys.MaxPage)\n\t\t\t}\n\t\t\tassertions.Equal(Extension(\"test\"), flushedKeys.Extension)\n\n\t\t\tblobstoreItr := NewBlobstoreIterator(context.Background(), blobstore, *flushedKeys, &entity.ConcreteExecution{})\n\t\t\ti := 0\n\t\t\tfor blobstoreItr.HasNext() {\n\t\t\t\tscanOutputEntity, err := blobstoreItr.Next()\n\t\t\t\tif err != nil {\n\t\t\t\t\tassertions.Fail(fmt.Sprintf(\"Error iterating blobstore: %v\", err))\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\tif scanOutputEntity == nil {\n\t\t\t\t\tbreak // No more items\n\t\t\t\t}\n\t\t\t\tif i < len(outputs) {\n\t\t\t\t\t// Compare the Execution field of ScanOutputEntity with the expected ConcreteExecution\n\t\t\t\t\tassertions.Equal(outputs[i].Execution, scanOutputEntity.Execution)\n\t\t\t\t}\n\t\t\t\ti++\n\t\t\t}\n\t\t\tassertions.Equal(tc.expectedCount, i, \"Number of items from blobstore iterator mismatch\")\n\t\t})\n\t}\n\n}\n\nfunc getMockExecutionManager(pages int, countPerPage int) persistence.ExecutionManager {\n\texecManager := &mocks.ExecutionManager{}\n\tfor i := 0; i < pages; i++ {\n\t\treq := &persistence.ListConcreteExecutionsRequest{\n\t\t\tPageToken: []byte(fmt.Sprintf(\"token_%v\", i)),\n\t\t\tPageSize:  executionPageSize,\n\t\t}\n\t\tif i == 0 {\n\t\t\treq.PageToken = nil\n\t\t}\n\t\tresp := &persistence.ListConcreteExecutionsResponse{\n\t\t\tExecutions: getExecutions(countPerPage),\n\t\t\tPageToken:  []byte(fmt.Sprintf(\"token_%v\", i+1)),\n\t\t}\n\t\tif i == pages-1 {\n\t\t\tresp.PageToken = nil\n\t\t}\n\t\texecManager.On(\"ListConcreteExecutions\", mock.Anything, req).Return(resp, nil)\n\t\texecManager.On(\"GetShardID\").Return(testShardID)\n\t}\n\treturn execManager\n}\n\nfunc getExecutions(count int) []*persistence.ListConcreteExecutionsEntity {\n\tvar result []*persistence.ListConcreteExecutionsEntity\n\tfor i := 0; i < count; i++ {\n\t\texecution := &persistence.ListConcreteExecutionsEntity{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:    uuid.New(),\n\t\t\t\tWorkflowID:  uuid.New(),\n\t\t\t\tRunID:       uuid.New(),\n\t\t\t\tBranchToken: validBranchToken,\n\t\t\t\tState:       0,\n\t\t\t},\n\t\t}\n\t\tif i%2 == 0 {\n\t\t\texecution.ExecutionInfo.BranchToken = nil\n\t\t\texecution.VersionHistories = &persistence.VersionHistories{\n\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t{\n\t\t\t\t\t\tBranchToken: validBranchToken,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\t\tresult = append(result, execution)\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/resource/params.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage resource\n\nimport (\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tpersistenceClient \"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\ntype (\n\t// Params holds the set of parameters needed to initialize common service resources\n\tParams struct {\n\t\tName               string\n\t\tInstanceID         string\n\t\tLogger             log.Logger\n\t\tThrottledLogger    log.Logger\n\t\tHostName           string\n\t\tGetIsolationGroups func() []string\n\n\t\tMetricScope        tally.Scope\n\t\tMembershipResolver membership.Resolver\n\t\tHashRings          map[string]membership.SingleProvider\n\t\tRPCFactory         rpc.Factory\n\t\tPProfInitializer   common.PProfInitializer\n\t\tPersistenceConfig  config.Persistence\n\t\tClusterMetadata    cluster.Metadata\n\t\tReplicatorConfig   config.Replicator\n\t\tMetricsClient      metrics.Client\n\t\tMessagingClient    messaging.Client\n\t\tBlobstoreClient    blobstore.Client\n\t\tESClient           es.GenericClient\n\t\tESConfig           *config.ElasticSearchConfig\n\n\t\t// RPC configuration\n\t\tRPCConfig config.RPC\n\n\t\tDynamicConfig              dynamicconfig.Client\n\t\tClusterRedirectionPolicy   *config.ClusterRedirectionPolicy\n\t\tPublicClient               workflowserviceclient.Interface\n\t\tArchivalMetadata           archiver.ArchivalMetadata\n\t\tArchiverProvider           provider.ArchiverProvider\n\t\tAuthorizer                 authorization.Authorizer // NOTE: this can be nil. If nil, AccessControlledHandlerImpl will initiate one with config.Authorization\n\t\tAuthorizationConfig        config.Authorization     // NOTE: empty(default) struct will get a authorization.NoopAuthorizer\n\t\tIsolationGroupStore        configstore.Client       // This can be nil, the default config store will be created if so\n\t\tIsolationGroupState        isolationgroup.State     // This can be nil, the default state store will be chosen if so\n\t\tPinotConfig                *config.PinotVisibilityConfig\n\t\tKafkaConfig                config.KafkaConfig\n\t\tPinotClient                pinot.GenericClient\n\t\tOSClient                   es.GenericClient\n\t\tOSConfig                   *config.ElasticSearchConfig\n\t\tAsyncWorkflowQueueProvider queue.Provider\n\t\tTimeSource                 clock.TimeSource\n\t\t// HistoryClientFn is used by integration tests to mock a history client\n\t\tHistoryClientFn func() history.Client\n\t\t// NewPersistenceBeanFn can be used to override the default persistence bean creation in unit tests to avoid DB setup\n\t\tNewPersistenceBeanFn  func(persistenceClient.Factory, *persistenceClient.Params, *service.Config) (persistenceClient.Bean, error)\n\t\tDiagnosticsInvariants []invariant.Invariant\n\n\t\t// ShardDistributorMatchingConfig is the config for shard distributor executor client in matching service\n\t\tShardDistributorMatchingConfig clientcommon.Config\n\n\t\t// DrainObserver is an optional observer that signals when this instance is\n\t\t// drained from service discovery.\n\t\t// It is used by shard-distributor executor clients to\n\t\t// gracefully stop processing during drains.\n\t\tDrainObserver clientcommon.DrainSignalObserver\n\t}\n)\n"
  },
  {
    "path": "common/resource/resource_impl.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage resource\n\nimport (\n\t\"math/rand\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/retryable\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\tcsc \"github.com/uber/cadence/common/dynamicconfig/configstore/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/isolationgroup/defaultisolationgroupstate\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceClient \"github.com/uber/cadence/common/persistence/client\"\n\tqrpc \"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/quotas/permember\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc NewResourceFactory() ResourceFactory {\n\treturn &resourceImplFactory{}\n}\n\ntype resourceImplFactory struct{}\n\nfunc (*resourceImplFactory) NewResource(\n\tparams *Params,\n\tserviceName string,\n\tserviceConfig *service.Config,\n) (resource Resource, err error) {\n\treturn New(params, serviceName, serviceConfig)\n}\n\n// Impl contains all common resources shared across frontend / matching / history / worker\ntype Impl struct {\n\tstatus int32\n\n\t// static infos\n\tnumShards       int\n\tserviceName     string\n\thostInfo        membership.HostInfo\n\tmetricsScope    tally.Scope\n\tclusterMetadata cluster.Metadata\n\n\t// other common resources\n\n\tdomainCache             cache.DomainCache\n\tdomainMetricsScopeCache cache.DomainMetricsScopeCache\n\tactiveClusterMgr        activecluster.Manager\n\ttimeSource              clock.TimeSource\n\tpayloadSerializer       persistence.PayloadSerializer\n\tmetricsClient           metrics.Client\n\tmessagingClient         messaging.Client\n\tblobstoreClient         blobstore.Client\n\tarchivalMetadata        archiver.ArchivalMetadata\n\tarchiverProvider        provider.ArchiverProvider\n\tdomainReplicationQueue  domain.ReplicationQueue\n\n\t// membership infos\n\n\tmembershipResolver membership.Resolver\n\thashRings          map[string]membership.Ring\n\n\t// internal services clients\n\n\tsdkClient                         workflowserviceclient.Interface\n\tfrontendRawClient                 frontend.Client\n\tfrontendClient                    frontend.Client\n\tmatchingRawClient                 matching.Client\n\tmatchingClient                    matching.Client\n\thistoryRawClient                  history.Client\n\thistoryClient                     history.Client\n\tshardDistributorRawClient         sharddistributor.Client\n\tshardDistributorClient            sharddistributor.Client\n\tshardDistributorExecutorRawClient executorclient.Client\n\tshardDistributorExecutorClient    executorclient.Client\n\tclientBean                        client.Bean\n\n\t// persistence clients\n\tpersistenceBean persistenceClient.Bean\n\n\t// hostName\n\thostName string\n\n\t// loggers\n\tlogger          log.Logger\n\tthrottledLogger log.Logger\n\n\t// for registering handlers\n\tdispatcher *yarpc.Dispatcher\n\n\t// internal vars\n\n\tpprofInitializer       common.PProfInitializer\n\truntimeMetricsReporter *metrics.RuntimeMetricsReporter\n\trpcFactory             rpc.Factory\n\n\tisolationGroups           isolationgroup.State\n\tisolationGroupConfigStore configstore.Client\n\n\tasyncWorkflowQueueProvider queue.Provider\n\n\tratelimiterAggregatorClient qrpc.Client\n}\n\nvar _ Resource = (*Impl)(nil)\n\n// New create a new resource containing common dependencies\nfunc New(\n\tparams *Params,\n\tserviceName string,\n\tserviceConfig *service.Config,\n) (impl *Impl, retError error) {\n\n\thostname := params.HostName\n\n\tlogger := params.Logger\n\tthrottledLogger := log.NewThrottledLogger(logger, serviceConfig.ThrottledLoggerMaxRPS)\n\n\tnumShards := params.PersistenceConfig.NumHistoryShards\n\tdispatcher := params.RPCFactory.GetDispatcher()\n\tmembershipResolver := params.MembershipResolver\n\n\tensureGetAllIsolationGroupsFnIsSet(params)\n\n\tdynamicCollection := dynamicconfig.NewCollection(\n\t\tparams.DynamicConfig,\n\t\tlogger,\n\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t)\n\tclientBean, err := client.NewClientBean(\n\t\tclient.NewRPCClientFactory(\n\t\t\tparams.RPCFactory,\n\t\t\tmembershipResolver,\n\t\t\tparams.MetricsClient,\n\t\t\tdynamicCollection,\n\t\t\tnumShards,\n\t\t\tparams.GetIsolationGroups,\n\t\t\tlogger,\n\t\t),\n\t\tparams.RPCFactory.GetDispatcher(),\n\t\tparams.ClusterMetadata,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnewPersistenceBeanFn := persistenceClient.NewBeanFromFactory\n\tif params.NewPersistenceBeanFn != nil {\n\t\tnewPersistenceBeanFn = params.NewPersistenceBeanFn\n\t}\n\tpersistenceBean, err := newPersistenceBeanFn(persistenceClient.NewFactory(\n\t\t&params.PersistenceConfig,\n\t\tfunc() float64 {\n\t\t\treturn permember.PerMember(\n\t\t\t\tserviceName,\n\t\t\t\tfloat64(serviceConfig.PersistenceGlobalMaxQPS()),\n\t\t\t\tfloat64(serviceConfig.PersistenceMaxQPS()),\n\t\t\t\tmembershipResolver,\n\t\t\t)\n\t\t},\n\t\tparams.ClusterMetadata.GetCurrentClusterName(),\n\t\tparams.MetricsClient,\n\t\tlogger,\n\t\tpersistence.NewDynamicConfiguration(dynamicCollection),\n\t), &persistenceClient.Params{\n\t\tPersistenceConfig: params.PersistenceConfig,\n\t\tMetricsClient:     params.MetricsClient,\n\t\tMessagingClient:   params.MessagingClient,\n\t\tESClient:          params.ESClient,\n\t\tESConfig:          params.ESConfig,\n\t\tPinotConfig:       params.PinotConfig,\n\t\tPinotClient:       params.PinotClient,\n\t\tOSClient:          params.OSClient,\n\t\tOSConfig:          params.OSConfig,\n\t}, serviceConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainCache := cache.NewDomainCache(\n\t\tpersistenceBean.GetDomainManager(),\n\t\tparams.ClusterMetadata,\n\t\tparams.MetricsClient,\n\t\tlogger,\n\t\tcache.WithTimeSource(params.TimeSource),\n\t)\n\n\tactiveClusterMgr, err := activecluster.NewManager(\n\t\tdomainCache.GetDomainByID,\n\t\tparams.MetricsClient,\n\t\tlogger,\n\t\tpersistenceBean,\n\t\tnumShards,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainMetricsScopeCache := cache.NewDomainMetricsScopeCache()\n\tdomainReplicationQueue := domain.NewReplicationQueue(\n\t\tpersistenceBean.GetDomainReplicationQueueManager(),\n\t\tparams.ClusterMetadata.GetCurrentClusterName(),\n\t\tparams.MetricsClient,\n\t\tlogger,\n\t)\n\n\tfrontendRawClient := clientBean.GetFrontendClient()\n\tfrontendClient := retryable.NewFrontendClient(\n\t\tfrontendRawClient,\n\t\tcommon.CreateFrontendServiceRetryPolicy(),\n\t\tserviceConfig.IsErrorRetryableFunction,\n\t)\n\n\tmatchingRawClient, err := clientBean.GetMatchingClient(domainCache.GetDomainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tmatchingClient := retryable.NewMatchingClient(\n\t\tmatchingRawClient,\n\t\tcommon.CreateMatchingServiceRetryPolicy(),\n\t\tserviceConfig.IsErrorRetryableFunction,\n\t)\n\n\tshardDistributorRawClient := clientBean.GetShardDistributorClient()\n\n\t// If the raw client is nil, then the client bean is not configured to provide a shard distributor client, so we\n\t// do not wrap and provide a retryable client\n\tvar shardDistributorClient sharddistributor.Client\n\tif shardDistributorRawClient == nil {\n\t\tshardDistributorClient = nil\n\t} else {\n\t\tshardDistributorClient = retryable.NewShardDistributorClient(\n\t\t\tshardDistributorRawClient,\n\t\t\tcommon.CreateShardDistributorServiceRetryPolicy(),\n\t\t\tserviceConfig.IsErrorRetryableFunction,\n\t\t)\n\t}\n\n\tshardDistributorExecutorRawClient := clientBean.GetShardDistributorExecutorClient()\n\tvar shardDistributorExecutorClient executorclient.Client\n\tif shardDistributorExecutorRawClient == nil {\n\t\tshardDistributorExecutorClient = nil\n\t} else {\n\t\tshardDistributorExecutorClient = retryable.NewShardDistributorExecutorClient(\n\t\t\tshardDistributorExecutorRawClient,\n\t\t\tcommon.CreateShardDistributorServiceRetryPolicy(),\n\t\t\tserviceConfig.IsErrorRetryableFunction,\n\t\t)\n\t}\n\n\tvar historyRawClient history.Client\n\tif params.HistoryClientFn != nil {\n\t\tlogger.Debug(\"Using history client from HistoryClientFn\")\n\t\thistoryRawClient = params.HistoryClientFn()\n\t} else {\n\t\tlogger.Debug(\"Using history client from bean\")\n\t\thistoryRawClient = clientBean.GetHistoryClient()\n\t}\n\thistoryClient := retryable.NewHistoryClient(\n\t\thistoryRawClient,\n\t\tcommon.CreateHistoryServiceRetryPolicy(),\n\t\tserviceConfig.IsErrorRetryableFunction,\n\t)\n\n\thistoryArchiverBootstrapContainer := &archiver.HistoryBootstrapContainer{\n\t\tHistoryV2Manager: persistenceBean.GetHistoryManager(),\n\t\tLogger:           logger,\n\t\tMetricsClient:    params.MetricsClient,\n\t\tClusterMetadata:  params.ClusterMetadata,\n\t\tDomainCache:      domainCache,\n\t}\n\tvisibilityArchiverBootstrapContainer := &archiver.VisibilityBootstrapContainer{\n\t\tLogger:          logger,\n\t\tMetricsClient:   params.MetricsClient,\n\t\tClusterMetadata: params.ClusterMetadata,\n\t\tDomainCache:     domainCache,\n\t}\n\tif err := params.ArchiverProvider.RegisterBootstrapContainer(\n\t\tserviceName,\n\t\thistoryArchiverBootstrapContainer,\n\t\tvisibilityArchiverBootstrapContainer,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tisolationGroupStore := createConfigStoreOrDefault(params, dynamicCollection)\n\n\tisolationGroupState, err := ensureIsolationGroupStateHandlerOrDefault(\n\t\tparams,\n\t\tdynamicCollection,\n\t\tdomainCache,\n\t\tisolationGroupStore,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tratelimiterAggs := qrpc.New(\n\t\thistoryRawClient, // no retries, will retry internally if needed\n\t\tclientBean.GetHistoryPeers(),\n\t\tlogger,\n\t\tparams.MetricsClient,\n\t)\n\n\timpl = &Impl{\n\t\tstatus: common.DaemonStatusInitialized,\n\n\t\t// static infos\n\n\t\tnumShards:       numShards,\n\t\tserviceName:     params.Name,\n\t\tmetricsScope:    params.MetricScope,\n\t\tclusterMetadata: params.ClusterMetadata,\n\n\t\t// other common resources\n\n\t\tdomainCache:             domainCache,\n\t\tdomainMetricsScopeCache: domainMetricsScopeCache,\n\t\tactiveClusterMgr:        activeClusterMgr,\n\t\ttimeSource:              clock.NewRealTimeSource(),\n\t\tpayloadSerializer:       persistence.NewPayloadSerializer(),\n\t\tmetricsClient:           params.MetricsClient,\n\t\tmessagingClient:         params.MessagingClient,\n\t\tblobstoreClient:         params.BlobstoreClient,\n\t\tarchivalMetadata:        params.ArchivalMetadata,\n\t\tarchiverProvider:        params.ArchiverProvider,\n\t\tdomainReplicationQueue:  domainReplicationQueue,\n\n\t\t// membership infos\n\t\tmembershipResolver: membershipResolver,\n\n\t\t// internal services clients\n\n\t\tsdkClient:                         params.PublicClient,\n\t\tfrontendRawClient:                 frontendRawClient,\n\t\tfrontendClient:                    frontendClient,\n\t\tmatchingRawClient:                 matchingRawClient,\n\t\tmatchingClient:                    matchingClient,\n\t\thistoryRawClient:                  historyRawClient,\n\t\thistoryClient:                     historyClient,\n\t\tshardDistributorRawClient:         shardDistributorRawClient,\n\t\tshardDistributorClient:            shardDistributorClient,\n\t\tshardDistributorExecutorRawClient: shardDistributorExecutorRawClient,\n\t\tshardDistributorExecutorClient:    shardDistributorExecutorClient,\n\t\tclientBean:                        clientBean,\n\n\t\t// persistence clients\n\t\tpersistenceBean: persistenceBean,\n\n\t\t// hostname\n\t\thostName: hostname,\n\n\t\t// loggers\n\n\t\tlogger:          logger,\n\t\tthrottledLogger: throttledLogger,\n\n\t\t// for registering handlers\n\t\tdispatcher: dispatcher,\n\n\t\t// internal vars\n\t\tpprofInitializer: params.PProfInitializer,\n\t\truntimeMetricsReporter: metrics.NewRuntimeMetricsReporter(\n\t\t\tparams.MetricScope,\n\t\t\ttime.Minute,\n\t\t\tlogger,\n\t\t\tparams.InstanceID,\n\t\t),\n\t\trpcFactory:                params.RPCFactory,\n\t\tisolationGroups:           isolationGroupState,\n\t\tisolationGroupConfigStore: isolationGroupStore, // can be nil where persistence is not available\n\n\t\tasyncWorkflowQueueProvider: params.AsyncWorkflowQueueProvider,\n\n\t\tratelimiterAggregatorClient: ratelimiterAggs,\n\t}\n\treturn impl, nil\n}\n\n// Start all resources\nfunc (h *Impl) Start() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&h.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\th.metricsScope.Counter(metrics.RestartCount).Inc(1)\n\th.runtimeMetricsReporter.Start()\n\n\tif err := h.pprofInitializer.Start(); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"fail to start PProf\")\n\t}\n\n\tif err := h.rpcFactory.Start(h.membershipResolver); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"fail to start RPC factory\")\n\t}\n\n\tif err := h.dispatcher.Start(); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"fail to start dispatcher\")\n\t}\n\th.membershipResolver.Start()\n\th.domainCache.Start()\n\th.domainMetricsScopeCache.Start()\n\n\thostInfo, err := h.membershipResolver.WhoAmI()\n\tif err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"fail to get host info from membership monitor\")\n\t}\n\th.hostInfo = hostInfo\n\n\tif h.isolationGroupConfigStore != nil {\n\t\th.isolationGroupConfigStore.Start()\n\t}\n\t// The service is now started up\n\th.logger.Info(\"service started\")\n\t// seed the random generator once for this service\n\trand.Seed(time.Now().UTC().UnixNano())\n}\n\n// Stop stops all resources\nfunc (h *Impl) Stop() {\n\tif !atomic.CompareAndSwapInt32(\n\t\t&h.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\th.domainCache.Stop()\n\th.domainMetricsScopeCache.Stop()\n\th.membershipResolver.Stop()\n\n\tif err := h.dispatcher.Stop(); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Error(\"failed to stop dispatcher\")\n\t}\n\th.rpcFactory.Stop()\n\n\th.runtimeMetricsReporter.Stop()\n\th.persistenceBean.Close()\n\tif h.isolationGroupConfigStore != nil {\n\t\th.isolationGroupConfigStore.Stop()\n\t}\n\th.isolationGroups.Stop()\n}\n\n// GetServiceName return service name\nfunc (h *Impl) GetServiceName() string {\n\treturn h.serviceName\n}\n\n// GetHostInfo return host info\nfunc (h *Impl) GetHostInfo() membership.HostInfo {\n\treturn h.hostInfo\n}\n\n// GetClusterMetadata return cluster metadata\nfunc (h *Impl) GetClusterMetadata() cluster.Metadata {\n\treturn h.clusterMetadata\n}\n\n// other common resources\n\n// GetDomainCache return domain cache\nfunc (h *Impl) GetDomainCache() cache.DomainCache {\n\treturn h.domainCache\n}\n\n// GetDomainMetricsScopeCache return domainMetricsScope cache\nfunc (h *Impl) GetDomainMetricsScopeCache() cache.DomainMetricsScopeCache {\n\treturn h.domainMetricsScopeCache\n}\n\n// GetActiveClusterManager return active cluster manager\nfunc (h *Impl) GetActiveClusterManager() activecluster.Manager {\n\treturn h.activeClusterMgr\n}\n\n// GetTimeSource return time source\nfunc (h *Impl) GetTimeSource() clock.TimeSource {\n\treturn h.timeSource\n}\n\n// GetPayloadSerializer return binary payload serializer\nfunc (h *Impl) GetPayloadSerializer() persistence.PayloadSerializer {\n\treturn h.payloadSerializer\n}\n\n// GetMetricsClient return metrics client\nfunc (h *Impl) GetMetricsClient() metrics.Client {\n\treturn h.metricsClient\n}\n\n// GetMessagingClient return messaging client\nfunc (h *Impl) GetMessagingClient() messaging.Client {\n\treturn h.messagingClient\n}\n\n// GetBlobstoreClient returns blobstore client\nfunc (h *Impl) GetBlobstoreClient() blobstore.Client {\n\treturn h.blobstoreClient\n}\n\n// GetArchivalMetadata return archival metadata\nfunc (h *Impl) GetArchivalMetadata() archiver.ArchivalMetadata {\n\treturn h.archivalMetadata\n}\n\n// GetArchiverProvider return archival provider\nfunc (h *Impl) GetArchiverProvider() provider.ArchiverProvider {\n\treturn h.archiverProvider\n}\n\n// GetDomainReplicationQueue return domain replication queue\nfunc (h *Impl) GetDomainReplicationQueue() domain.ReplicationQueue {\n\treturn h.domainReplicationQueue\n}\n\n// GetMembershipResolver return the membership resolver\nfunc (h *Impl) GetMembershipResolver() membership.Resolver {\n\treturn h.membershipResolver\n}\n\n// internal services clients\n\n// GetSDKClient return sdk client\nfunc (h *Impl) GetSDKClient() workflowserviceclient.Interface {\n\treturn h.sdkClient\n}\n\n// GetFrontendRawClient return frontend client without retry policy\nfunc (h *Impl) GetFrontendRawClient() frontend.Client {\n\treturn h.frontendRawClient\n}\n\n// GetFrontendClient return frontend client with retry policy\nfunc (h *Impl) GetFrontendClient() frontend.Client {\n\treturn h.frontendClient\n}\n\n// GetMatchingRawClient return matching client without retry policy\nfunc (h *Impl) GetMatchingRawClient() matching.Client {\n\treturn h.matchingRawClient\n}\n\n// GetMatchingClient return matching client with retry policy\nfunc (h *Impl) GetMatchingClient() matching.Client {\n\treturn h.matchingClient\n}\n\n// GetHistoryRawClient return history client without retry policy\nfunc (h *Impl) GetHistoryRawClient() history.Client {\n\treturn h.historyRawClient\n}\n\n// GetHistoryClient return history client with retry policy\nfunc (h *Impl) GetHistoryClient() history.Client {\n\treturn h.historyClient\n}\n\n// GetShardDistributorExecutorRawClient return client for sharddistributor executor\nfunc (h *Impl) GetShardDistributorExecutorRawClient() executorclient.Client {\n\treturn h.shardDistributorExecutorRawClient\n}\n\n// GetShardDistributorExecutorClient return client for sharddistributor executor\nfunc (h *Impl) GetShardDistributorExecutorClient() executorclient.Client {\n\treturn h.shardDistributorExecutorRawClient\n}\n\nfunc (h *Impl) GetRatelimiterAggregatorsClient() qrpc.Client {\n\treturn h.ratelimiterAggregatorClient\n}\n\n// GetRemoteAdminClient return remote admin client for given cluster name\nfunc (h *Impl) GetRemoteAdminClient(\n\tcluster string,\n) (admin.Client, error) {\n\n\treturn h.clientBean.GetRemoteAdminClient(cluster)\n}\n\n// GetRemoteFrontendClient return remote frontend client for given cluster name\nfunc (h *Impl) GetRemoteFrontendClient(\n\tcluster string,\n) (frontend.Client, error) {\n\n\treturn h.clientBean.GetRemoteFrontendClient(cluster)\n}\n\n// GetClientBean return RPC client bean\nfunc (h *Impl) GetClientBean() client.Bean {\n\treturn h.clientBean\n}\n\n// persistence clients\n\n// GetMetadataManager return metadata manager\nfunc (h *Impl) GetDomainManager() persistence.DomainManager {\n\treturn h.persistenceBean.GetDomainManager()\n}\n\n// GetDomainAuditManager return domain audit manager\nfunc (h *Impl) GetDomainAuditManager() persistence.DomainAuditManager {\n\treturn h.persistenceBean.GetDomainAuditManager()\n}\n\n// GetTaskManager return task manager\nfunc (h *Impl) GetTaskManager() persistence.TaskManager {\n\treturn h.persistenceBean.GetTaskManager()\n}\n\n// GetVisibilityManager return visibility manager\nfunc (h *Impl) GetVisibilityManager() persistence.VisibilityManager {\n\treturn h.persistenceBean.GetVisibilityManager()\n}\n\n// GetShardManager return shard manager\nfunc (h *Impl) GetShardManager() persistence.ShardManager {\n\treturn h.persistenceBean.GetShardManager()\n}\n\n// GetHistoryManager return history manager\nfunc (h *Impl) GetHistoryManager() persistence.HistoryManager {\n\treturn h.persistenceBean.GetHistoryManager()\n}\n\n// GetExecutionManager return execution manager for given shard ID\nfunc (h *Impl) GetExecutionManager(shardID int) (persistence.ExecutionManager, error) {\n\n\treturn h.persistenceBean.GetExecutionManager(shardID)\n}\n\n// GetPersistenceBean return persistence bean\nfunc (h *Impl) GetPersistenceBean() persistenceClient.Bean {\n\treturn h.persistenceBean\n}\n\nfunc (h *Impl) GetHostName() string {\n\treturn h.hostName\n}\n\n// loggers\n\n// GetLogger return logger\nfunc (h *Impl) GetLogger() log.Logger {\n\treturn h.logger\n}\n\n// GetThrottledLogger return throttled logger\nfunc (h *Impl) GetThrottledLogger() log.Logger {\n\treturn h.throttledLogger\n}\n\n// GetDispatcher return YARPC dispatcher, used for registering handlers\nfunc (h *Impl) GetDispatcher() *yarpc.Dispatcher {\n\treturn h.dispatcher\n}\n\n// GetIsolationGroupState returns the isolationGroupState\nfunc (h *Impl) GetIsolationGroupState() isolationgroup.State {\n\treturn h.isolationGroups\n}\n\n// GetIsolationGroupStore returns the isolation group configuration store or nil\nfunc (h *Impl) GetIsolationGroupStore() configstore.Client {\n\treturn h.isolationGroupConfigStore\n}\n\n// GetAsyncWorkflowQueueProvider returns the async workflow queue provider\nfunc (h *Impl) GetAsyncWorkflowQueueProvider() queue.Provider {\n\treturn h.asyncWorkflowQueueProvider\n}\n\n// GetMetricsScope returns the tally scope for metrics reporting\nfunc (h *Impl) GetMetricsScope() tally.Scope {\n\treturn h.metricsScope\n}\n\n// due to the config store being only available for some\n// persistence layers, *both* the configStoreClient and IsolationGroupState\n// will be optionally available\nfunc createConfigStoreOrDefault(\n\tparams *Params,\n\tdc *dynamicconfig.Collection,\n) configstore.Client {\n\n\tif params.IsolationGroupStore != nil {\n\t\treturn params.IsolationGroupStore\n\t}\n\tcscConfig := &csc.ClientConfig{\n\t\tPollInterval:        dc.GetDurationProperty(dynamicproperties.IsolationGroupStateRefreshInterval)(),\n\t\tUpdateRetryAttempts: dc.GetIntProperty(dynamicproperties.IsolationGroupStateUpdateRetryAttempts)(),\n\t\tFetchTimeout:        dc.GetDurationProperty(dynamicproperties.IsolationGroupStateFetchTimeout)(),\n\t\tUpdateTimeout:       dc.GetDurationProperty(dynamicproperties.IsolationGroupStateUpdateTimeout)(),\n\t}\n\tcfgStoreClient, err := configstore.NewConfigStoreClient(cscConfig, &params.PersistenceConfig, params.Logger, params.MetricsClient, persistence.GlobalIsolationGroupConfig)\n\tif err != nil {\n\t\t// not possible to create the client under some persistence configurations, so this is expected\n\t\tparams.Logger.Warn(\"not instantiating Isolation group config store, this feature will not be enabled\", tag.Error(err))\n\t\treturn nil\n\t}\n\treturn cfgStoreClient\n}\n\n// Use the provided IsolationGroupStateHandler or the default one\n// due to the config store being only available for some\n// persistence layers, *both* the configStoreClient and IsolationGroupState\n// will be optionally available\nfunc ensureIsolationGroupStateHandlerOrDefault(\n\tparams *Params,\n\tdc *dynamicconfig.Collection,\n\tdomainCache cache.DomainCache,\n\tisolationGroupStore dynamicconfig.Client,\n) (isolationgroup.State, error) {\n\n\tif params.IsolationGroupState != nil {\n\t\treturn params.IsolationGroupState, nil\n\t}\n\n\treturn defaultisolationgroupstate.NewDefaultIsolationGroupStateWatcherWithConfigStoreClient(\n\t\tparams.Logger,\n\t\tdc,\n\t\tdomainCache,\n\t\tisolationGroupStore,\n\t\tparams.MetricsClient,\n\t\tparams.GetIsolationGroups,\n\t)\n}\n\nfunc ensureGetAllIsolationGroupsFnIsSet(params *Params) {\n\tif params.GetIsolationGroups == nil {\n\t\tparams.GetIsolationGroups = func() []string { return []string{} }\n\t}\n}\n"
  },
  {
    "path": "common/resource/resource_impl_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage resource\n\nimport (\n\t\"runtime/debug\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceClient \"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestStartStop(t *testing.T) {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tt.Fatalf(\"Panic: %v, stacktrace: %s\", r, debug.Stack())\n\t\t}\n\t}()\n\n\tctrl := gomock.NewController(t)\n\tserviceName := \"test-service\"\n\thostName := \"test-host\"\n\tmetricsCl := metrics.NewNoopMetricsClient()\n\tlogger := testlogger.New(t)\n\tdc := dynamicconfig.NewInMemoryClient()\n\n\t// membership resolver mocks\n\tmemberRes := membership.NewMockResolver(ctrl)\n\tmemberRes.EXPECT().MemberCount(gomock.Any()).Return(4, nil).AnyTimes()\n\tmemberRes.EXPECT().Subscribe(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\tmemberRes.EXPECT().Unsubscribe(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\tmemberRes.EXPECT().Start().AnyTimes()\n\tmemberRes.EXPECT().Stop().AnyTimes()\n\tselfHostInfo := membership.NewHostInfo(\"localhost:0\")\n\tmemberRes.EXPECT().WhoAmI().Return(selfHostInfo, nil).AnyTimes()\n\n\t// pprof mocks\n\tpprof := common.NewMockPProfInitializer(ctrl)\n\tpprof.EXPECT().Start().Return(nil).Times(1)\n\n\t// rpc mocks\n\tclusterMetadata := cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: 1,\n\t\t\tPrimaryClusterName:       \"primary-cluster\",\n\t\t\tCurrentClusterName:       \"primary-cluster\",\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\"primary-cluster\":   {InitialFailoverVersion: 1, Enabled: true, RPCTransport: \"tchannel\", RPCAddress: \"localhost:0\"},\n\t\t\t\t\"secondary-cluster\": {InitialFailoverVersion: 1, Enabled: true, RPCTransport: \"tchannel\", RPCAddress: \"localhost:0\"},\n\t\t\t},\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tmetricsCl,\n\t\tlogger,\n\t)\n\tdirectOutboundPCF := rpc.NewDirectPeerChooserFactory(serviceName, logger, metricsCl)\n\tdirectConnRetainFn := func(opts ...dynamicproperties.FilterOption) bool { return false }\n\tpcf := rpc.NewMockPeerChooserFactory(ctrl)\n\tpeerChooser := rpc.NewMockPeerChooser(ctrl)\n\tpeerChooser.EXPECT().Start().Return(nil).AnyTimes()\n\tpeerChooser.EXPECT().Stop().Return(nil).AnyTimes()\n\tpcf.EXPECT().CreatePeerChooser(gomock.Any(), gomock.Any()).Return(peerChooser, nil).AnyTimes()\n\tob := rpc.CombineOutbounds(\n\t\trpc.NewCrossDCOutbounds(clusterMetadata.GetAllClusterInfo(), pcf),\n\t\trpc.NewDirectOutboundBuilder(service.History, true, nil, directOutboundPCF, directConnRetainFn),\n\t\trpc.NewDirectOutboundBuilder(service.Matching, true, nil, directOutboundPCF, directConnRetainFn),\n\t)\n\trpcFac := rpc.NewFactory(logger, rpc.Params{\n\t\tServiceName:      serviceName,\n\t\tTChannelAddress:  \"localhost:0\",\n\t\tGRPCAddress:      \"localhost:0\",\n\t\tOutboundsBuilder: ob,\n\t})\n\n\t// persistence mocks\n\tpersistenceClientBean := persistenceClient.NewMockBean(ctrl)\n\tdomainMgr := persistence.NewMockDomainManager(ctrl)\n\tdomainMgr.EXPECT().GetMetadata(gomock.Any()).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: 2,\n\t}, nil).AnyTimes()\n\tdomain := &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:   \"test-domain-id\",\n\t\t\tName: \"test-domain-name\",\n\t\t\tData: map[string]string{\"k1\": \"v1\"},\n\t\t},\n\t\tConfig:              &persistence.DomainConfig{},\n\t\tReplicationConfig:   &persistence.DomainReplicationConfig{},\n\t\tNotificationVersion: 1, // should be less than notification version for this domain to be loaded by cache.\n\t}\n\tdomainMgr.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(&persistence.ListDomainsResponse{\n\t\tDomains: []*persistence.GetDomainResponse{domain},\n\t}, nil).AnyTimes()\n\tdomainMgr.EXPECT().GetDomain(gomock.Any(), gomock.Any()).Return(domain, nil).AnyTimes()\n\tdomainReplMgr := persistence.NewMockQueueManager(ctrl)\n\thistoryMgr := persistence.NewMockHistoryManager(ctrl)\n\ttaskMgr := persistence.NewMockTaskManager(ctrl)\n\tvisMgr := persistence.NewMockVisibilityManager(ctrl)\n\tshardMgr := persistence.NewMockShardManager(ctrl)\n\texecMgr := persistence.NewMockExecutionManager(ctrl)\n\tpersistenceClientBean.EXPECT().GetDomainManager().Return(domainMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetTaskManager().Return(taskMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetVisibilityManager().Return(visMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetShardManager().Return(shardMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetExecutionManager(gomock.Any()).Return(execMgr, nil).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetDomainReplicationQueueManager().Return(domainReplMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().GetHistoryManager().Return(historyMgr).AnyTimes()\n\tpersistenceClientBean.EXPECT().Close().Times(1)\n\n\t// archiver provider mocks\n\tarchiveProvider := provider.NewMockArchiverProvider(ctrl)\n\tarchiveProvider.EXPECT().RegisterBootstrapContainer(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\n\t// params\n\tparams := &Params{\n\t\tName:     serviceName,\n\t\tHostName: hostName,\n\t\tPersistenceConfig: config.Persistence{\n\t\t\tNumHistoryShards: 1024,\n\t\t\tDefaultStore:     \"nosql-store\",\n\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\"nosql-store\": {\n\t\t\t\t\tNoSQL: &config.NoSQL{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tClusterMetadata:    clusterMetadata,\n\t\tLogger:             logger,\n\t\tMetricScope:        tally.NoopScope,\n\t\tMetricsClient:      metricsCl,\n\t\tRPCFactory:         rpcFac,\n\t\tMembershipResolver: memberRes,\n\t\tDynamicConfig:      dc,\n\t\tTimeSource:         clock.NewRealTimeSource(),\n\t\tPProfInitializer:   pprof,\n\t\tNewPersistenceBeanFn: func(persistenceClient.Factory, *persistenceClient.Params, *service.Config) (persistenceClient.Bean, error) {\n\t\t\treturn persistenceClientBean, nil\n\t\t},\n\t\tArchiverProvider:           archiveProvider,\n\t\tAsyncWorkflowQueueProvider: queue.NewMockProvider(ctrl),\n\t}\n\n\t// bare minimum service config\n\tsvcCfg := &service.Config{\n\t\tThrottledLoggerMaxRPS: func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn 100\n\t\t},\n\t\tPersistenceGlobalMaxQPS: func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn 100\n\t\t},\n\t\tPersistenceMaxQPS: func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn 100\n\t\t},\n\t}\n\n\ti, err := New(params, serviceName, svcCfg)\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\ti.Start()\n\ti.Start() // should be no-op\n\tdefer func() {\n\t\ti.Stop()\n\t\ti.Stop() // should be no-op\n\t\tgoleak.VerifyNone(t)\n\t}()\n\n\ttime.Sleep(100 * time.Millisecond) // wait for go routines to start\n\n\t// validations\n\tassert.Equal(t, serviceName, i.GetServiceName())\n\tassert.Equal(t, selfHostInfo, i.GetHostInfo())\n\tgotDomain, err := i.GetDomainCache().GetDomainByID(\"test-domain-id\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, domain.Info, gotDomain.GetInfo())\n\tassert.NotNil(t, i.GetDomainMetricsScopeCache())\n\tassert.NotNil(t, i.GetTimeSource())\n\tassert.NotNil(t, i.GetPayloadSerializer())\n\tassert.Equal(t, metricsCl, i.GetMetricsClient())\n\tassert.Equal(t, params.MessagingClient, i.GetMessagingClient())\n\tassert.Equal(t, params.BlobstoreClient, i.GetBlobstoreClient())\n\tassert.Equal(t, params.ArchivalMetadata, i.GetArchivalMetadata())\n\tassert.Equal(t, archiveProvider, i.GetArchiverProvider())\n\tassert.NotNil(t, i.GetDomainReplicationQueue())\n\tassert.Equal(t, memberRes, i.GetMembershipResolver())\n\tassert.Equal(t, params.PublicClient, i.GetSDKClient())\n\tassert.NotNil(t, i.GetFrontendRawClient())\n\tassert.NotNil(t, i.GetFrontendClient())\n\tassert.NotNil(t, i.GetMatchingRawClient())\n\tassert.NotNil(t, i.GetMatchingClient())\n\tassert.NotNil(t, i.GetHistoryRawClient())\n\tassert.NotNil(t, i.GetHistoryClient())\n\tassert.NotNil(t, i.GetRatelimiterAggregatorsClient())\n\tadminClient, err := i.GetRemoteAdminClient(\"secondary-cluster\")\n\tassert.NoError(t, err)\n\tassert.NotNil(t, adminClient)\n\tfrontendClient, err := i.GetRemoteFrontendClient(\"secondary-cluster\")\n\tassert.NoError(t, err)\n\tassert.NotNil(t, frontendClient)\n\tassert.NotNil(t, i.GetClientBean())\n\tassert.Equal(t, domainMgr, i.GetDomainManager())\n\tassert.Equal(t, taskMgr, i.GetTaskManager())\n\tassert.Equal(t, visMgr, i.GetVisibilityManager())\n\tassert.Equal(t, shardMgr, i.GetShardManager())\n\tassert.Equal(t, historyMgr, i.GetHistoryManager())\n\tem, err := i.GetExecutionManager(3)\n\tassert.NoError(t, err)\n\tassert.Equal(t, execMgr, em)\n\tassert.Equal(t, persistenceClientBean, i.GetPersistenceBean())\n\tassert.Equal(t, hostName, i.GetHostName())\n\tassert.NotNil(t, i.GetLogger())\n\tassert.NotNil(t, i.GetThrottledLogger())\n\tassert.NotNil(t, i.GetDispatcher())\n\tassert.NotNil(t, i.GetIsolationGroupState())\n\tassert.Nil(t, i.GetIsolationGroupStore())\n\tassert.Equal(t, params.AsyncWorkflowQueueProvider, i.GetAsyncWorkflowQueueProvider())\n}\n"
  },
  {
    "path": "common/resource/resource_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: types.go\n//\n// Generated by this command:\n//\n//\tmockgen -package resource -source types.go -destination resource_mock.go -self_package github.com/uber/cadence/common/resource\n//\n\n// Package resource is a generated GoMock package.\npackage resource\n\nimport (\n\treflect \"reflect\"\n\n\ttally \"github.com/uber-go/tally\"\n\tworkflowserviceclient \"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tclient \"github.com/uber/cadence/client\"\n\tadmin \"github.com/uber/cadence/client/admin\"\n\tfrontend \"github.com/uber/cadence/client/frontend\"\n\thistory \"github.com/uber/cadence/client/history\"\n\tmatching \"github.com/uber/cadence/client/matching\"\n\tactivecluster \"github.com/uber/cadence/common/activecluster\"\n\tarchiver \"github.com/uber/cadence/common/archiver\"\n\tprovider \"github.com/uber/cadence/common/archiver/provider\"\n\tqueue \"github.com/uber/cadence/common/asyncworkflow/queue\"\n\tblobstore \"github.com/uber/cadence/common/blobstore\"\n\tcache \"github.com/uber/cadence/common/cache\"\n\tclock \"github.com/uber/cadence/common/clock\"\n\tcluster \"github.com/uber/cadence/common/cluster\"\n\tdomain \"github.com/uber/cadence/common/domain\"\n\tconfigstore \"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\tisolationgroup \"github.com/uber/cadence/common/isolationgroup\"\n\tlog \"github.com/uber/cadence/common/log\"\n\tmembership \"github.com/uber/cadence/common/membership\"\n\tmessaging \"github.com/uber/cadence/common/messaging\"\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\tclient0 \"github.com/uber/cadence/common/persistence/client\"\n\trpc \"github.com/uber/cadence/common/quotas/global/rpc\"\n\tservice \"github.com/uber/cadence/common/service\"\n\texecutorclient \"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// MockResourceFactory is a mock of ResourceFactory interface.\ntype MockResourceFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockResourceFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockResourceFactoryMockRecorder is the mock recorder for MockResourceFactory.\ntype MockResourceFactoryMockRecorder struct {\n\tmock *MockResourceFactory\n}\n\n// NewMockResourceFactory creates a new mock instance.\nfunc NewMockResourceFactory(ctrl *gomock.Controller) *MockResourceFactory {\n\tmock := &MockResourceFactory{ctrl: ctrl}\n\tmock.recorder = &MockResourceFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockResourceFactory) EXPECT() *MockResourceFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// NewResource mocks base method.\nfunc (m *MockResourceFactory) NewResource(params *Params, serviceName string, serviceConfig *service.Config) (Resource, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewResource\", params, serviceName, serviceConfig)\n\tret0, _ := ret[0].(Resource)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewResource indicates an expected call of NewResource.\nfunc (mr *MockResourceFactoryMockRecorder) NewResource(params, serviceName, serviceConfig any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewResource\", reflect.TypeOf((*MockResourceFactory)(nil).NewResource), params, serviceName, serviceConfig)\n}\n\n// MockResource is a mock of Resource interface.\ntype MockResource struct {\n\tctrl     *gomock.Controller\n\trecorder *MockResourceMockRecorder\n\tisgomock struct{}\n}\n\n// MockResourceMockRecorder is the mock recorder for MockResource.\ntype MockResourceMockRecorder struct {\n\tmock *MockResource\n}\n\n// NewMockResource creates a new mock instance.\nfunc NewMockResource(ctrl *gomock.Controller) *MockResource {\n\tmock := &MockResource{ctrl: ctrl}\n\tmock.recorder = &MockResourceMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockResource) EXPECT() *MockResourceMockRecorder {\n\treturn m.recorder\n}\n\n// GetActiveClusterManager mocks base method.\nfunc (m *MockResource) GetActiveClusterManager() activecluster.Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterManager\")\n\tret0, _ := ret[0].(activecluster.Manager)\n\treturn ret0\n}\n\n// GetActiveClusterManager indicates an expected call of GetActiveClusterManager.\nfunc (mr *MockResourceMockRecorder) GetActiveClusterManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterManager\", reflect.TypeOf((*MockResource)(nil).GetActiveClusterManager))\n}\n\n// GetArchivalMetadata mocks base method.\nfunc (m *MockResource) GetArchivalMetadata() archiver.ArchivalMetadata {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetArchivalMetadata\")\n\tret0, _ := ret[0].(archiver.ArchivalMetadata)\n\treturn ret0\n}\n\n// GetArchivalMetadata indicates an expected call of GetArchivalMetadata.\nfunc (mr *MockResourceMockRecorder) GetArchivalMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetArchivalMetadata\", reflect.TypeOf((*MockResource)(nil).GetArchivalMetadata))\n}\n\n// GetArchiverProvider mocks base method.\nfunc (m *MockResource) GetArchiverProvider() provider.ArchiverProvider {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetArchiverProvider\")\n\tret0, _ := ret[0].(provider.ArchiverProvider)\n\treturn ret0\n}\n\n// GetArchiverProvider indicates an expected call of GetArchiverProvider.\nfunc (mr *MockResourceMockRecorder) GetArchiverProvider() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetArchiverProvider\", reflect.TypeOf((*MockResource)(nil).GetArchiverProvider))\n}\n\n// GetAsyncWorkflowQueueProvider mocks base method.\nfunc (m *MockResource) GetAsyncWorkflowQueueProvider() queue.Provider {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAsyncWorkflowQueueProvider\")\n\tret0, _ := ret[0].(queue.Provider)\n\treturn ret0\n}\n\n// GetAsyncWorkflowQueueProvider indicates an expected call of GetAsyncWorkflowQueueProvider.\nfunc (mr *MockResourceMockRecorder) GetAsyncWorkflowQueueProvider() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAsyncWorkflowQueueProvider\", reflect.TypeOf((*MockResource)(nil).GetAsyncWorkflowQueueProvider))\n}\n\n// GetBlobstoreClient mocks base method.\nfunc (m *MockResource) GetBlobstoreClient() blobstore.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBlobstoreClient\")\n\tret0, _ := ret[0].(blobstore.Client)\n\treturn ret0\n}\n\n// GetBlobstoreClient indicates an expected call of GetBlobstoreClient.\nfunc (mr *MockResourceMockRecorder) GetBlobstoreClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBlobstoreClient\", reflect.TypeOf((*MockResource)(nil).GetBlobstoreClient))\n}\n\n// GetClientBean mocks base method.\nfunc (m *MockResource) GetClientBean() client.Bean {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClientBean\")\n\tret0, _ := ret[0].(client.Bean)\n\treturn ret0\n}\n\n// GetClientBean indicates an expected call of GetClientBean.\nfunc (mr *MockResourceMockRecorder) GetClientBean() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClientBean\", reflect.TypeOf((*MockResource)(nil).GetClientBean))\n}\n\n// GetClusterMetadata mocks base method.\nfunc (m *MockResource) GetClusterMetadata() cluster.Metadata {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClusterMetadata\")\n\tret0, _ := ret[0].(cluster.Metadata)\n\treturn ret0\n}\n\n// GetClusterMetadata indicates an expected call of GetClusterMetadata.\nfunc (mr *MockResourceMockRecorder) GetClusterMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClusterMetadata\", reflect.TypeOf((*MockResource)(nil).GetClusterMetadata))\n}\n\n// GetDispatcher mocks base method.\nfunc (m *MockResource) GetDispatcher() *yarpc.Dispatcher {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDispatcher\")\n\tret0, _ := ret[0].(*yarpc.Dispatcher)\n\treturn ret0\n}\n\n// GetDispatcher indicates an expected call of GetDispatcher.\nfunc (mr *MockResourceMockRecorder) GetDispatcher() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDispatcher\", reflect.TypeOf((*MockResource)(nil).GetDispatcher))\n}\n\n// GetDomainAuditManager mocks base method.\nfunc (m *MockResource) GetDomainAuditManager() persistence.DomainAuditManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAuditManager\")\n\tret0, _ := ret[0].(persistence.DomainAuditManager)\n\treturn ret0\n}\n\n// GetDomainAuditManager indicates an expected call of GetDomainAuditManager.\nfunc (mr *MockResourceMockRecorder) GetDomainAuditManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAuditManager\", reflect.TypeOf((*MockResource)(nil).GetDomainAuditManager))\n}\n\n// GetDomainCache mocks base method.\nfunc (m *MockResource) GetDomainCache() cache.DomainCache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainCache\")\n\tret0, _ := ret[0].(cache.DomainCache)\n\treturn ret0\n}\n\n// GetDomainCache indicates an expected call of GetDomainCache.\nfunc (mr *MockResourceMockRecorder) GetDomainCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainCache\", reflect.TypeOf((*MockResource)(nil).GetDomainCache))\n}\n\n// GetDomainManager mocks base method.\nfunc (m *MockResource) GetDomainManager() persistence.DomainManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainManager\")\n\tret0, _ := ret[0].(persistence.DomainManager)\n\treturn ret0\n}\n\n// GetDomainManager indicates an expected call of GetDomainManager.\nfunc (mr *MockResourceMockRecorder) GetDomainManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainManager\", reflect.TypeOf((*MockResource)(nil).GetDomainManager))\n}\n\n// GetDomainMetricsScopeCache mocks base method.\nfunc (m *MockResource) GetDomainMetricsScopeCache() cache.DomainMetricsScopeCache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainMetricsScopeCache\")\n\tret0, _ := ret[0].(cache.DomainMetricsScopeCache)\n\treturn ret0\n}\n\n// GetDomainMetricsScopeCache indicates an expected call of GetDomainMetricsScopeCache.\nfunc (mr *MockResourceMockRecorder) GetDomainMetricsScopeCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainMetricsScopeCache\", reflect.TypeOf((*MockResource)(nil).GetDomainMetricsScopeCache))\n}\n\n// GetDomainReplicationQueue mocks base method.\nfunc (m *MockResource) GetDomainReplicationQueue() domain.ReplicationQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainReplicationQueue\")\n\tret0, _ := ret[0].(domain.ReplicationQueue)\n\treturn ret0\n}\n\n// GetDomainReplicationQueue indicates an expected call of GetDomainReplicationQueue.\nfunc (mr *MockResourceMockRecorder) GetDomainReplicationQueue() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainReplicationQueue\", reflect.TypeOf((*MockResource)(nil).GetDomainReplicationQueue))\n}\n\n// GetExecutionManager mocks base method.\nfunc (m *MockResource) GetExecutionManager(arg0 int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionManager\", arg0)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetExecutionManager indicates an expected call of GetExecutionManager.\nfunc (mr *MockResourceMockRecorder) GetExecutionManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionManager\", reflect.TypeOf((*MockResource)(nil).GetExecutionManager), arg0)\n}\n\n// GetFrontendClient mocks base method.\nfunc (m *MockResource) GetFrontendClient() frontend.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFrontendClient\")\n\tret0, _ := ret[0].(frontend.Client)\n\treturn ret0\n}\n\n// GetFrontendClient indicates an expected call of GetFrontendClient.\nfunc (mr *MockResourceMockRecorder) GetFrontendClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFrontendClient\", reflect.TypeOf((*MockResource)(nil).GetFrontendClient))\n}\n\n// GetFrontendRawClient mocks base method.\nfunc (m *MockResource) GetFrontendRawClient() frontend.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFrontendRawClient\")\n\tret0, _ := ret[0].(frontend.Client)\n\treturn ret0\n}\n\n// GetFrontendRawClient indicates an expected call of GetFrontendRawClient.\nfunc (mr *MockResourceMockRecorder) GetFrontendRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFrontendRawClient\", reflect.TypeOf((*MockResource)(nil).GetFrontendRawClient))\n}\n\n// GetHistoryClient mocks base method.\nfunc (m *MockResource) GetHistoryClient() history.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryClient\")\n\tret0, _ := ret[0].(history.Client)\n\treturn ret0\n}\n\n// GetHistoryClient indicates an expected call of GetHistoryClient.\nfunc (mr *MockResourceMockRecorder) GetHistoryClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryClient\", reflect.TypeOf((*MockResource)(nil).GetHistoryClient))\n}\n\n// GetHistoryManager mocks base method.\nfunc (m *MockResource) GetHistoryManager() persistence.HistoryManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryManager\")\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\treturn ret0\n}\n\n// GetHistoryManager indicates an expected call of GetHistoryManager.\nfunc (mr *MockResourceMockRecorder) GetHistoryManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryManager\", reflect.TypeOf((*MockResource)(nil).GetHistoryManager))\n}\n\n// GetHistoryRawClient mocks base method.\nfunc (m *MockResource) GetHistoryRawClient() history.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryRawClient\")\n\tret0, _ := ret[0].(history.Client)\n\treturn ret0\n}\n\n// GetHistoryRawClient indicates an expected call of GetHistoryRawClient.\nfunc (mr *MockResourceMockRecorder) GetHistoryRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryRawClient\", reflect.TypeOf((*MockResource)(nil).GetHistoryRawClient))\n}\n\n// GetHostInfo mocks base method.\nfunc (m *MockResource) GetHostInfo() membership.HostInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHostInfo\")\n\tret0, _ := ret[0].(membership.HostInfo)\n\treturn ret0\n}\n\n// GetHostInfo indicates an expected call of GetHostInfo.\nfunc (mr *MockResourceMockRecorder) GetHostInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHostInfo\", reflect.TypeOf((*MockResource)(nil).GetHostInfo))\n}\n\n// GetHostName mocks base method.\nfunc (m *MockResource) GetHostName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHostName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetHostName indicates an expected call of GetHostName.\nfunc (mr *MockResourceMockRecorder) GetHostName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHostName\", reflect.TypeOf((*MockResource)(nil).GetHostName))\n}\n\n// GetIsolationGroupState mocks base method.\nfunc (m *MockResource) GetIsolationGroupState() isolationgroup.State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIsolationGroupState\")\n\tret0, _ := ret[0].(isolationgroup.State)\n\treturn ret0\n}\n\n// GetIsolationGroupState indicates an expected call of GetIsolationGroupState.\nfunc (mr *MockResourceMockRecorder) GetIsolationGroupState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIsolationGroupState\", reflect.TypeOf((*MockResource)(nil).GetIsolationGroupState))\n}\n\n// GetIsolationGroupStore mocks base method.\nfunc (m *MockResource) GetIsolationGroupStore() configstore.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIsolationGroupStore\")\n\tret0, _ := ret[0].(configstore.Client)\n\treturn ret0\n}\n\n// GetIsolationGroupStore indicates an expected call of GetIsolationGroupStore.\nfunc (mr *MockResourceMockRecorder) GetIsolationGroupStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIsolationGroupStore\", reflect.TypeOf((*MockResource)(nil).GetIsolationGroupStore))\n}\n\n// GetLogger mocks base method.\nfunc (m *MockResource) GetLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetLogger indicates an expected call of GetLogger.\nfunc (mr *MockResourceMockRecorder) GetLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLogger\", reflect.TypeOf((*MockResource)(nil).GetLogger))\n}\n\n// GetMatchingClient mocks base method.\nfunc (m *MockResource) GetMatchingClient() matching.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMatchingClient\")\n\tret0, _ := ret[0].(matching.Client)\n\treturn ret0\n}\n\n// GetMatchingClient indicates an expected call of GetMatchingClient.\nfunc (mr *MockResourceMockRecorder) GetMatchingClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMatchingClient\", reflect.TypeOf((*MockResource)(nil).GetMatchingClient))\n}\n\n// GetMatchingRawClient mocks base method.\nfunc (m *MockResource) GetMatchingRawClient() matching.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMatchingRawClient\")\n\tret0, _ := ret[0].(matching.Client)\n\treturn ret0\n}\n\n// GetMatchingRawClient indicates an expected call of GetMatchingRawClient.\nfunc (mr *MockResourceMockRecorder) GetMatchingRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMatchingRawClient\", reflect.TypeOf((*MockResource)(nil).GetMatchingRawClient))\n}\n\n// GetMembershipResolver mocks base method.\nfunc (m *MockResource) GetMembershipResolver() membership.Resolver {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMembershipResolver\")\n\tret0, _ := ret[0].(membership.Resolver)\n\treturn ret0\n}\n\n// GetMembershipResolver indicates an expected call of GetMembershipResolver.\nfunc (mr *MockResourceMockRecorder) GetMembershipResolver() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMembershipResolver\", reflect.TypeOf((*MockResource)(nil).GetMembershipResolver))\n}\n\n// GetMessagingClient mocks base method.\nfunc (m *MockResource) GetMessagingClient() messaging.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagingClient\")\n\tret0, _ := ret[0].(messaging.Client)\n\treturn ret0\n}\n\n// GetMessagingClient indicates an expected call of GetMessagingClient.\nfunc (mr *MockResourceMockRecorder) GetMessagingClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagingClient\", reflect.TypeOf((*MockResource)(nil).GetMessagingClient))\n}\n\n// GetMetricsClient mocks base method.\nfunc (m *MockResource) GetMetricsClient() metrics.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsClient\")\n\tret0, _ := ret[0].(metrics.Client)\n\treturn ret0\n}\n\n// GetMetricsClient indicates an expected call of GetMetricsClient.\nfunc (mr *MockResourceMockRecorder) GetMetricsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsClient\", reflect.TypeOf((*MockResource)(nil).GetMetricsClient))\n}\n\n// GetMetricsScope mocks base method.\nfunc (m *MockResource) GetMetricsScope() tally.Scope {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsScope\")\n\tret0, _ := ret[0].(tally.Scope)\n\treturn ret0\n}\n\n// GetMetricsScope indicates an expected call of GetMetricsScope.\nfunc (mr *MockResourceMockRecorder) GetMetricsScope() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsScope\", reflect.TypeOf((*MockResource)(nil).GetMetricsScope))\n}\n\n// GetPayloadSerializer mocks base method.\nfunc (m *MockResource) GetPayloadSerializer() persistence.PayloadSerializer {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPayloadSerializer\")\n\tret0, _ := ret[0].(persistence.PayloadSerializer)\n\treturn ret0\n}\n\n// GetPayloadSerializer indicates an expected call of GetPayloadSerializer.\nfunc (mr *MockResourceMockRecorder) GetPayloadSerializer() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPayloadSerializer\", reflect.TypeOf((*MockResource)(nil).GetPayloadSerializer))\n}\n\n// GetPersistenceBean mocks base method.\nfunc (m *MockResource) GetPersistenceBean() client0.Bean {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPersistenceBean\")\n\tret0, _ := ret[0].(client0.Bean)\n\treturn ret0\n}\n\n// GetPersistenceBean indicates an expected call of GetPersistenceBean.\nfunc (mr *MockResourceMockRecorder) GetPersistenceBean() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPersistenceBean\", reflect.TypeOf((*MockResource)(nil).GetPersistenceBean))\n}\n\n// GetRatelimiterAggregatorsClient mocks base method.\nfunc (m *MockResource) GetRatelimiterAggregatorsClient() rpc.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRatelimiterAggregatorsClient\")\n\tret0, _ := ret[0].(rpc.Client)\n\treturn ret0\n}\n\n// GetRatelimiterAggregatorsClient indicates an expected call of GetRatelimiterAggregatorsClient.\nfunc (mr *MockResourceMockRecorder) GetRatelimiterAggregatorsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRatelimiterAggregatorsClient\", reflect.TypeOf((*MockResource)(nil).GetRatelimiterAggregatorsClient))\n}\n\n// GetRemoteAdminClient mocks base method.\nfunc (m *MockResource) GetRemoteAdminClient(cluster string) (admin.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteAdminClient\", cluster)\n\tret0, _ := ret[0].(admin.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteAdminClient indicates an expected call of GetRemoteAdminClient.\nfunc (mr *MockResourceMockRecorder) GetRemoteAdminClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteAdminClient\", reflect.TypeOf((*MockResource)(nil).GetRemoteAdminClient), cluster)\n}\n\n// GetRemoteFrontendClient mocks base method.\nfunc (m *MockResource) GetRemoteFrontendClient(cluster string) (frontend.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteFrontendClient\", cluster)\n\tret0, _ := ret[0].(frontend.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteFrontendClient indicates an expected call of GetRemoteFrontendClient.\nfunc (mr *MockResourceMockRecorder) GetRemoteFrontendClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteFrontendClient\", reflect.TypeOf((*MockResource)(nil).GetRemoteFrontendClient), cluster)\n}\n\n// GetSDKClient mocks base method.\nfunc (m *MockResource) GetSDKClient() workflowserviceclient.Interface {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSDKClient\")\n\tret0, _ := ret[0].(workflowserviceclient.Interface)\n\treturn ret0\n}\n\n// GetSDKClient indicates an expected call of GetSDKClient.\nfunc (mr *MockResourceMockRecorder) GetSDKClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSDKClient\", reflect.TypeOf((*MockResource)(nil).GetSDKClient))\n}\n\n// GetServiceName mocks base method.\nfunc (m *MockResource) GetServiceName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetServiceName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetServiceName indicates an expected call of GetServiceName.\nfunc (mr *MockResourceMockRecorder) GetServiceName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetServiceName\", reflect.TypeOf((*MockResource)(nil).GetServiceName))\n}\n\n// GetShardDistributorExecutorClient mocks base method.\nfunc (m *MockResource) GetShardDistributorExecutorClient() executorclient.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardDistributorExecutorClient\")\n\tret0, _ := ret[0].(executorclient.Client)\n\treturn ret0\n}\n\n// GetShardDistributorExecutorClient indicates an expected call of GetShardDistributorExecutorClient.\nfunc (mr *MockResourceMockRecorder) GetShardDistributorExecutorClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardDistributorExecutorClient\", reflect.TypeOf((*MockResource)(nil).GetShardDistributorExecutorClient))\n}\n\n// GetShardManager mocks base method.\nfunc (m *MockResource) GetShardManager() persistence.ShardManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardManager\")\n\tret0, _ := ret[0].(persistence.ShardManager)\n\treturn ret0\n}\n\n// GetShardManager indicates an expected call of GetShardManager.\nfunc (mr *MockResourceMockRecorder) GetShardManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardManager\", reflect.TypeOf((*MockResource)(nil).GetShardManager))\n}\n\n// GetTaskManager mocks base method.\nfunc (m *MockResource) GetTaskManager() persistence.TaskManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskManager\")\n\tret0, _ := ret[0].(persistence.TaskManager)\n\treturn ret0\n}\n\n// GetTaskManager indicates an expected call of GetTaskManager.\nfunc (mr *MockResourceMockRecorder) GetTaskManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskManager\", reflect.TypeOf((*MockResource)(nil).GetTaskManager))\n}\n\n// GetThrottledLogger mocks base method.\nfunc (m *MockResource) GetThrottledLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetThrottledLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetThrottledLogger indicates an expected call of GetThrottledLogger.\nfunc (mr *MockResourceMockRecorder) GetThrottledLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetThrottledLogger\", reflect.TypeOf((*MockResource)(nil).GetThrottledLogger))\n}\n\n// GetTimeSource mocks base method.\nfunc (m *MockResource) GetTimeSource() clock.TimeSource {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTimeSource\")\n\tret0, _ := ret[0].(clock.TimeSource)\n\treturn ret0\n}\n\n// GetTimeSource indicates an expected call of GetTimeSource.\nfunc (mr *MockResourceMockRecorder) GetTimeSource() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTimeSource\", reflect.TypeOf((*MockResource)(nil).GetTimeSource))\n}\n\n// GetVisibilityManager mocks base method.\nfunc (m *MockResource) GetVisibilityManager() persistence.VisibilityManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityManager\")\n\tret0, _ := ret[0].(persistence.VisibilityManager)\n\treturn ret0\n}\n\n// GetVisibilityManager indicates an expected call of GetVisibilityManager.\nfunc (mr *MockResourceMockRecorder) GetVisibilityManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityManager\", reflect.TypeOf((*MockResource)(nil).GetVisibilityManager))\n}\n\n// Start mocks base method.\nfunc (m *MockResource) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockResourceMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockResource)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockResource) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockResourceMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockResource)(nil).Stop))\n}\n"
  },
  {
    "path": "common/resource/resource_test_utils.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage resource\n\nimport (\n\t\"testing\"\n\n\toldgomock \"github.com/golang/mock/gomock\" // client library cannot change from the old gomock\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tpublicservicetest \"go.uber.org/cadence/.gen/go/cadence/workflowservicetest\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceClient \"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/taskvalidator\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype (\n\t// Test is the test implementation used for testing\n\tTest struct {\n\t\tMetricsScope    tally.TestScope\n\t\tClusterMetadata cluster.Metadata\n\n\t\t// other common resources\n\n\t\tDomainCache             *cache.MockDomainCache\n\t\tDomainMetricsScopeCache cache.DomainMetricsScopeCache\n\t\tActiveClusterMgr        *activecluster.MockManager\n\t\tDomainReplicationQueue  *domain.MockReplicationQueue\n\t\tTimeSource              clock.TimeSource\n\t\tPayloadSerializer       persistence.PayloadSerializer\n\t\tMetricsClient           metrics.Client\n\t\tArchivalMetadata        *archiver.MockArchivalMetadata\n\t\tArchiverProvider        *provider.MockArchiverProvider\n\t\tBlobstoreClient         *blobstore.MockClient\n\t\tMockPayloadSerializer   *persistence.MockPayloadSerializer\n\n\t\t// membership infos\n\t\tMembershipResolver *membership.MockResolver\n\n\t\t// internal services clients\n\n\t\tSDKClient                      *publicservicetest.MockClient\n\t\tFrontendClient                 *frontend.MockClient\n\t\tMatchingClient                 *matching.MockClient\n\t\tHistoryClient                  *history.MockClient\n\t\tShardDistributorExecutorClient *sharddistributorexecutor.MockClient\n\t\tRemoteAdminClient              *admin.MockClient\n\t\tRemoteFrontendClient           *frontend.MockClient\n\t\tClientBean                     *client.MockBean\n\n\t\t// persistence clients\n\n\t\tMetadataMgr     *mocks.MetadataManager\n\t\tDomainAuditMgr  *persistence.MockDomainAuditManager\n\t\tTaskMgr         *mocks.TaskManager\n\t\tVisibilityMgr   *mocks.VisibilityManager\n\t\tShardMgr        *mocks.ShardManager\n\t\tHistoryMgr      *mocks.HistoryV2Manager\n\t\tExecutionMgr    *mocks.ExecutionManager\n\t\tPersistenceBean *persistenceClient.MockBean\n\n\t\tIsolationGroups     *isolationgroup.MockState\n\t\tIsolationGroupStore *configstore.MockClient\n\t\tHostName            string\n\t\tLogger              log.Logger\n\t\ttaskvalidator       taskvalidator.Checker\n\n\t\tAsyncWorkflowQueueProvider *queue.MockProvider\n\n\t\tRatelimiterAggregatorClient rpc.Client\n\t}\n)\n\nvar _ Resource = (*Test)(nil)\n\nconst (\n\ttestHostName = \"test_host\"\n)\n\nvar (\n\ttestHostInfo = membership.NewHostInfo(testHostName)\n)\n\n// NewTest returns a new test resource instance\nfunc NewTest(\n\tt *testing.T,\n\tcontroller *gomock.Controller,\n\tserviceMetricsIndex metrics.ServiceIdx,\n) *Test {\n\tlogger := testlogger.New(t)\n\n\tfrontendClient := frontend.NewMockClient(controller)\n\tmatchingClient := matching.NewMockClient(controller)\n\thistoryClient := history.NewMockClient(controller)\n\tremoteAdminClient := admin.NewMockClient(controller)\n\tremoteFrontendClient := frontend.NewMockClient(controller)\n\tclientBean := client.NewMockBean(controller)\n\tclientBean.EXPECT().GetFrontendClient().Return(frontendClient).AnyTimes()\n\tclientBean.EXPECT().GetMatchingClient(gomock.Any()).Return(matchingClient, nil).AnyTimes()\n\tclientBean.EXPECT().GetHistoryClient().Return(historyClient).AnyTimes()\n\tclientBean.EXPECT().GetRemoteAdminClient(gomock.Any()).Return(remoteAdminClient, nil).AnyTimes()\n\tclientBean.EXPECT().GetRemoteFrontendClient(gomock.Any()).Return(remoteFrontendClient, nil).AnyTimes()\n\tshardDistributorExecutorClient := sharddistributorexecutor.NewMockClient(controller)\n\n\tmetadataMgr := &mocks.MetadataManager{}\n\tdomainAuditMgr := persistence.NewMockDomainAuditManager(controller)\n\ttaskMgr := &mocks.TaskManager{}\n\tvisibilityMgr := &mocks.VisibilityManager{}\n\tshardMgr := &mocks.ShardManager{}\n\thistoryMgr := &mocks.HistoryV2Manager{}\n\texecutionMgr := &mocks.ExecutionManager{}\n\tdomainReplicationQueue := domain.NewMockReplicationQueue(controller)\n\tdomainReplicationQueue.EXPECT().Start().AnyTimes()\n\tdomainReplicationQueue.EXPECT().Stop().AnyTimes()\n\tpersistenceBean := persistenceClient.NewMockBean(controller)\n\tpersistenceBean.EXPECT().GetDomainManager().Return(metadataMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetDomainAuditManager().Return(domainAuditMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetTaskManager().Return(taskMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetVisibilityManager().Return(visibilityMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetHistoryManager().Return(historyMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetShardManager().Return(shardMgr).AnyTimes()\n\tpersistenceBean.EXPECT().GetExecutionManager(gomock.Any()).Return(executionMgr, nil).AnyTimes()\n\n\tisolationGroupMock := isolationgroup.NewMockState(controller)\n\tisolationGroupMock.EXPECT().Stop().AnyTimes()\n\n\tscope := tally.NewTestScope(\"test\", nil)\n\n\tasyncWorkflowQueueProvider := queue.NewMockProvider(controller)\n\n\tactiveClusterMgr := activecluster.NewMockManager(controller)\n\n\treturn &Test{\n\t\tMetricsScope: scope,\n\n\t\t// By default tests will run on active cluster unless overridden otherwise\n\t\tClusterMetadata: cluster.TestActiveClusterMetadata,\n\n\t\t// other common resources\n\n\t\tDomainCache:             cache.NewMockDomainCache(controller),\n\t\tDomainMetricsScopeCache: cache.NewDomainMetricsScopeCache(),\n\t\tDomainReplicationQueue:  domainReplicationQueue,\n\t\tActiveClusterMgr:        activeClusterMgr,\n\t\tTimeSource:              clock.NewRealTimeSource(),\n\t\tPayloadSerializer:       persistence.NewPayloadSerializer(),\n\t\tMetricsClient:           metrics.NewClient(scope, serviceMetricsIndex, metrics.HistogramMigration{}),\n\t\tArchivalMetadata:        &archiver.MockArchivalMetadata{},\n\t\tArchiverProvider:        provider.NewMockArchiverProvider(controller),\n\t\tBlobstoreClient:         blobstore.NewMockClient(controller),\n\t\tMockPayloadSerializer:   persistence.NewMockPayloadSerializer(controller),\n\n\t\t// membership infos\n\t\tMembershipResolver: membership.NewMockResolver(controller),\n\n\t\t// internal services clients\n\n\t\tSDKClient:                      publicservicetest.NewMockClient(oldgomock.NewController(t)),\n\t\tFrontendClient:                 frontendClient,\n\t\tMatchingClient:                 matchingClient,\n\t\tHistoryClient:                  historyClient,\n\t\tRemoteAdminClient:              remoteAdminClient,\n\t\tRemoteFrontendClient:           remoteFrontendClient,\n\t\tClientBean:                     clientBean,\n\t\tShardDistributorExecutorClient: shardDistributorExecutorClient,\n\n\t\t// persistence clients\n\n\t\tMetadataMgr:     metadataMgr,\n\t\tDomainAuditMgr:  domainAuditMgr,\n\t\tTaskMgr:         taskMgr,\n\t\tVisibilityMgr:   visibilityMgr,\n\t\tShardMgr:        shardMgr,\n\t\tHistoryMgr:      historyMgr,\n\t\tExecutionMgr:    executionMgr,\n\t\tPersistenceBean: persistenceBean,\n\t\tIsolationGroups: isolationGroupMock,\n\t\t// logger\n\n\t\tLogger: logger,\n\n\t\tAsyncWorkflowQueueProvider: asyncWorkflowQueueProvider,\n\n\t\tRatelimiterAggregatorClient: nil, // TODO: not currently used\n\t}\n}\n\n// Start for testing\nfunc (s *Test) Start() {\n\n}\n\n// Stop for testing\nfunc (s *Test) Stop() {\n\n}\n\n// static infos\n\n// GetServiceName for testing\nfunc (s *Test) GetServiceName() string {\n\tpanic(\"user should implement this method for test\")\n}\n\n// GetHostInfo for testing\nfunc (s *Test) GetHostInfo() membership.HostInfo {\n\treturn testHostInfo\n}\n\n// GetClusterMetadata for testing\nfunc (s *Test) GetClusterMetadata() cluster.Metadata {\n\treturn s.ClusterMetadata\n}\n\n// other common resources\n\n// GetDomainCache for testing\nfunc (s *Test) GetDomainCache() cache.DomainCache {\n\treturn s.DomainCache\n}\n\n// GetDomainMetricsScopeCache for testing\nfunc (s *Test) GetDomainMetricsScopeCache() cache.DomainMetricsScopeCache {\n\treturn s.DomainMetricsScopeCache\n}\n\n// GetDomainReplicationQueue for testing\nfunc (s *Test) GetDomainReplicationQueue() domain.ReplicationQueue {\n\t// user should implement this method for test\n\treturn s.DomainReplicationQueue\n}\n\nfunc (s *Test) GetActiveClusterManager() activecluster.Manager {\n\treturn s.ActiveClusterMgr\n}\n\n// GetTimeSource for testing\nfunc (s *Test) GetTimeSource() clock.TimeSource {\n\treturn s.TimeSource\n}\n\n// GetPayloadSerializer for testing\nfunc (s *Test) GetPayloadSerializer() persistence.PayloadSerializer {\n\treturn s.PayloadSerializer\n}\n\n// GetPayloadSerializer for testing\nfunc (s *Test) GetTaskValidator() taskvalidator.Checker {\n\treturn s.taskvalidator\n}\n\n// GetMetricsClient for testing\nfunc (s *Test) GetMetricsClient() metrics.Client {\n\treturn s.MetricsClient\n}\n\n// GetMetricsScope for testing\nfunc (s *Test) GetMetricsScope() tally.Scope {\n\treturn s.MetricsScope\n}\n\n// GetMessagingClient for testing\nfunc (s *Test) GetMessagingClient() messaging.Client {\n\tpanic(\"user should implement this method for test\")\n}\n\n// GetBlobstoreClient for testing\nfunc (s *Test) GetBlobstoreClient() blobstore.Client {\n\treturn s.BlobstoreClient\n}\n\n// GetArchivalMetadata for testing\nfunc (s *Test) GetArchivalMetadata() archiver.ArchivalMetadata {\n\treturn s.ArchivalMetadata\n}\n\n// GetArchiverProvider for testing\nfunc (s *Test) GetArchiverProvider() provider.ArchiverProvider {\n\treturn s.ArchiverProvider\n}\n\n// GetMembershipResolver for testing\nfunc (s *Test) GetMembershipResolver() membership.Resolver {\n\treturn s.MembershipResolver\n}\n\n// internal services clients\n\n// GetSDKClient for testing\nfunc (s *Test) GetSDKClient() workflowserviceclient.Interface {\n\treturn s.SDKClient\n}\n\n// GetFrontendRawClient for testing\nfunc (s *Test) GetFrontendRawClient() frontend.Client {\n\treturn s.FrontendClient\n}\n\n// GetFrontendClient for testing\nfunc (s *Test) GetFrontendClient() frontend.Client {\n\treturn s.FrontendClient\n}\n\n// GetMatchingRawClient for testing\nfunc (s *Test) GetMatchingRawClient() matching.Client {\n\treturn s.MatchingClient\n}\n\n// GetMatchingClient for testing\nfunc (s *Test) GetMatchingClient() matching.Client {\n\treturn s.MatchingClient\n}\n\n// GetHistoryRawClient for testing\nfunc (s *Test) GetHistoryRawClient() history.Client {\n\treturn s.HistoryClient\n}\n\n// GetHistoryClient for testing\nfunc (s *Test) GetHistoryClient() history.Client {\n\treturn s.HistoryClient\n}\n\nfunc (s *Test) GetShardDistributorExecutorClient() executorclient.Client {\n\treturn s.ShardDistributorExecutorClient\n}\n\n// GetRemoteAdminClient for testing\nfunc (s *Test) GetRemoteAdminClient(\n\tcluster string,\n) (admin.Client, error) {\n\n\treturn s.RemoteAdminClient, nil\n}\n\n// GetRemoteFrontendClient for testing\nfunc (s *Test) GetRemoteFrontendClient(\n\tcluster string,\n) (frontend.Client, error) {\n\n\treturn s.RemoteFrontendClient, nil\n}\n\n// GetClientBean for testing\nfunc (s *Test) GetClientBean() client.Bean {\n\treturn s.ClientBean\n}\n\n// persistence clients\n\n// GetMetadataManager for testing\nfunc (s *Test) GetDomainManager() persistence.DomainManager {\n\treturn s.MetadataMgr\n}\n\n// GetDomainAuditManager for testing\nfunc (s *Test) GetDomainAuditManager() persistence.DomainAuditManager {\n\treturn s.DomainAuditMgr\n}\n\n// GetTaskManager for testing\nfunc (s *Test) GetTaskManager() persistence.TaskManager {\n\treturn s.TaskMgr\n}\n\n// GetVisibilityManager for testing\nfunc (s *Test) GetVisibilityManager() persistence.VisibilityManager {\n\treturn s.VisibilityMgr\n}\n\n// GetShardManager for testing\nfunc (s *Test) GetShardManager() persistence.ShardManager {\n\treturn s.ShardMgr\n}\n\n// GetHistoryManager for testing\nfunc (s *Test) GetHistoryManager() persistence.HistoryManager {\n\treturn s.HistoryMgr\n}\n\n// GetExecutionManager for testing\nfunc (s *Test) GetExecutionManager(\n\tshardID int,\n) (persistence.ExecutionManager, error) {\n\n\treturn s.ExecutionMgr, nil\n}\n\n// GetPersistenceBean for testing\nfunc (s *Test) GetPersistenceBean() persistenceClient.Bean {\n\treturn s.PersistenceBean\n}\n\n// GetHostName for testing\nfunc (s *Test) GetHostName() string {\n\treturn s.HostName\n}\n\n// loggers\n\n// GetLogger for testing\nfunc (s *Test) GetLogger() log.Logger {\n\treturn s.Logger\n}\n\n// GetThrottledLogger for testing\nfunc (s *Test) GetThrottledLogger() log.Logger {\n\treturn s.Logger\n}\n\n// GetDispatcher for testing\nfunc (s *Test) GetDispatcher() *yarpc.Dispatcher {\n\tpanic(\"user should implement this method for test\")\n}\n\n// GetIsolationGroupState returns the isolationGroupState for testing\nfunc (s *Test) GetIsolationGroupState() isolationgroup.State {\n\treturn s.IsolationGroups\n}\n\n// GetIsolationGroupStore returns the config store for their\n// isolation-group stores\nfunc (s *Test) GetIsolationGroupStore() configstore.Client {\n\treturn s.IsolationGroupStore\n}\n\nfunc (s *Test) GetAsyncWorkflowQueueProvider() queue.Provider {\n\treturn s.AsyncWorkflowQueueProvider\n}\n\nfunc (s *Test) GetRatelimiterAggregatorsClient() rpc.Client {\n\treturn s.RatelimiterAggregatorClient\n}\n\n// Finish checks whether expectations are met\nfunc (s *Test) Finish(\n\tt mock.TestingT,\n) {\n\ts.ArchivalMetadata.AssertExpectations(t)\n\n\ts.MetadataMgr.AssertExpectations(t)\n\ts.TaskMgr.AssertExpectations(t)\n\ts.VisibilityMgr.AssertExpectations(t)\n\ts.ShardMgr.AssertExpectations(t)\n\ts.HistoryMgr.AssertExpectations(t)\n\ts.ExecutionMgr.AssertExpectations(t)\n}\n"
  },
  {
    "path": "common/resource/types.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination resource_mock.go -self_package github.com/uber/cadence/common/resource\n\npackage resource\n\nimport (\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceClient \"github.com/uber/cadence/common/persistence/client\"\n\tqrpc \"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype ResourceFactory interface {\n\tNewResource(params *Params,\n\t\tserviceName string,\n\t\tserviceConfig *service.Config,\n\t) (resource Resource, err error)\n}\n\n// Resource is the interface which expose common resources\ntype Resource interface {\n\tcommon.Daemon\n\n\t// static infos\n\n\tGetServiceName() string\n\tGetHostInfo() membership.HostInfo\n\tGetArchivalMetadata() archiver.ArchivalMetadata\n\tGetClusterMetadata() cluster.Metadata\n\n\t// other common resources\n\n\tGetDomainCache() cache.DomainCache\n\tGetDomainMetricsScopeCache() cache.DomainMetricsScopeCache\n\tGetActiveClusterManager() activecluster.Manager\n\tGetTimeSource() clock.TimeSource\n\tGetPayloadSerializer() persistence.PayloadSerializer\n\tGetMetricsClient() metrics.Client\n\tGetArchiverProvider() provider.ArchiverProvider\n\tGetMessagingClient() messaging.Client\n\tGetBlobstoreClient() blobstore.Client\n\tGetDomainReplicationQueue() domain.ReplicationQueue\n\n\t// membership infos\n\tGetMembershipResolver() membership.Resolver\n\n\t// internal services clients\n\n\tGetSDKClient() workflowserviceclient.Interface\n\tGetFrontendRawClient() frontend.Client\n\tGetFrontendClient() frontend.Client\n\tGetMatchingRawClient() matching.Client\n\tGetMatchingClient() matching.Client\n\tGetHistoryRawClient() history.Client\n\tGetHistoryClient() history.Client\n\tGetRatelimiterAggregatorsClient() qrpc.Client\n\tGetRemoteAdminClient(cluster string) (admin.Client, error)\n\tGetRemoteFrontendClient(cluster string) (frontend.Client, error)\n\tGetClientBean() client.Bean\n\tGetShardDistributorExecutorClient() executorclient.Client\n\n\t// persistence clients\n\tGetDomainManager() persistence.DomainManager\n\tGetDomainAuditManager() persistence.DomainAuditManager\n\tGetTaskManager() persistence.TaskManager\n\tGetVisibilityManager() persistence.VisibilityManager\n\tGetShardManager() persistence.ShardManager\n\tGetHistoryManager() persistence.HistoryManager\n\tGetExecutionManager(int) (persistence.ExecutionManager, error)\n\tGetPersistenceBean() persistenceClient.Bean\n\n\t// GetHostName get host name\n\tGetHostName() string\n\n\t// loggers\n\tGetLogger() log.Logger\n\tGetThrottledLogger() log.Logger\n\n\t// for registering handlers\n\tGetDispatcher() *yarpc.Dispatcher\n\n\t// GetIsolationGroupState returns the isolationGroupState\n\tGetIsolationGroupState() isolationgroup.State\n\tGetIsolationGroupStore() configstore.Client\n\n\tGetAsyncWorkflowQueueProvider() queue.Provider\n\n\t// GetMetricsScope returns the tally scope for metrics reporting\n\tGetMetricsScope() tally.Scope\n}\n"
  },
  {
    "path": "common/rpc/direct_peer_chooser.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"runtime/debug\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer/direct\"\n\t\"go.uber.org/yarpc/peer/hostport\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nvar (\n\tnoOpSubscriberInstance = &noOpSubscriber{}\n)\n\n// directPeerChooser is a peer.Chooser that chooses a peer based on the shard key.\n// Peers are managed by the peerList and peers are reused across multiple requests.\ntype directPeerChooser struct {\n\tstatus               int32\n\tserviceName          string\n\tlogger               log.Logger\n\tscope                metrics.Scope\n\tt                    peer.Transport\n\tenableConnRetainMode dynamicproperties.BoolPropertyFn\n\tlegacyChooser        peer.Chooser\n\tlegacyChooserErr     error\n\tmu                   sync.RWMutex\n\tpeers                map[string]peer.Peer\n}\n\nfunc newDirectChooser(\n\tserviceName string,\n\tt peer.Transport,\n\tlogger log.Logger,\n\tmetricsCl metrics.Client,\n\tenableConnRetainMode dynamicproperties.BoolPropertyFn,\n) *directPeerChooser {\n\tdpc := &directPeerChooser{\n\t\tserviceName:          serviceName,\n\t\tlogger:               logger.WithTags(tag.DestService(serviceName)),\n\t\tscope:                metricsCl.Scope(metrics.P2PRPCPeerChooserScope).Tagged(metrics.DestServiceTag(serviceName)),\n\t\tt:                    t,\n\t\tenableConnRetainMode: enableConnRetainMode,\n\t\tpeers:                make(map[string]peer.Peer),\n\t}\n\n\tif dpc.enableConnRetainMode == nil {\n\t\tdpc.enableConnRetainMode = func(opts ...dynamicproperties.FilterOption) bool { return false }\n\t}\n\n\treturn dpc\n}\n\n// Start statisfies the peer.Chooser interface.\nfunc (g *directPeerChooser) Start() (err error) {\n\tif !atomic.CompareAndSwapInt32(&g.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn nil\n\t}\n\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tg.logger.Error(\"direct peer chooser failed to start\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\tg.logger.Info(\"direct peer chooser started\")\n\t}()\n\n\tif !g.enableConnRetainMode() {\n\t\tc, ok := g.getLegacyChooser()\n\t\tif ok {\n\t\t\treturn c.Start()\n\t\t}\n\n\t\treturn fmt.Errorf(\"failed to start direct peer chooser because direct peer chooser initialization failed, err: %v\", g.legacyChooserErr)\n\t}\n\n\treturn nil\n}\n\n// Stop statisfies the peer.Chooser interface.\nfunc (g *directPeerChooser) Stop() error {\n\tif !atomic.CompareAndSwapInt32(&g.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn nil\n\t}\n\n\tvar err error\n\t// Stop legacy chooser if it was initialized\n\tif g.legacyChooser != nil {\n\t\terr = g.legacyChooser.Stop()\n\t}\n\n\t// Release all peers if there's any\n\tg.updatePeersInternal(nil)\n\n\tg.logger.Info(\"direct peer chooser stopped\", tag.Error(err))\n\treturn err\n}\n\n// IsRunning statisfies the peer.Chooser interface.\nfunc (g *directPeerChooser) IsRunning() bool {\n\tif atomic.LoadInt32(&g.status) != common.DaemonStatusStarted {\n\t\treturn false\n\t}\n\n\tif !g.enableConnRetainMode() {\n\t\tc, ok := g.getLegacyChooser()\n\t\tif ok {\n\t\t\treturn c.IsRunning()\n\t\t}\n\t\treturn false\n\t}\n\n\treturn true // no-op\n}\n\n// Choose returns an existing peer for the shard key.\n// ShardKey is {host}:{port} of the peer. It could be tchannel or grpc address.\nfunc (g *directPeerChooser) Choose(ctx context.Context, req *transport.Request) (peer peer.Peer, onFinish func(error), err error) {\n\tif !g.enableConnRetainMode() {\n\t\treturn g.chooseFromLegacyDirectPeerChooser(ctx, req)\n\t}\n\n\tif req.ShardKey == \"\" {\n\t\treturn nil, nil, yarpcerrors.InvalidArgumentErrorf(\"chooser requires ShardKey to be non-empty\")\n\t}\n\n\tg.mu.RLock()\n\tp, ok := g.peers[req.ShardKey]\n\tif ok {\n\t\tg.mu.RUnlock()\n\t\treturn p, func(error) {}, nil\n\t}\n\tg.mu.RUnlock()\n\n\t// peer is not cached, add new peer\n\tp, err = g.addPeer(req.ShardKey)\n\tif err != nil {\n\t\treturn nil, nil, yarpcerrors.InternalErrorf(\"failed to add peer for shard key %v, err: %v\", req.ShardKey, err)\n\t}\n\n\treturn p, func(error) {}, nil\n}\n\n// UpdatePeers removes peers that are not in the members list.\n// Do not create actual yarpc peers for the members. They are created lazily when a request comes in (Choose is called).\nfunc (g *directPeerChooser) UpdatePeers(serviceName string, members []membership.HostInfo) {\n\tif g.serviceName != serviceName {\n\t\tg.logger.Debug(\"This is not the service chooser is created for. Ignore such updates.\", tag.Dynamic(\"members-service\", serviceName))\n\t\treturn\n\t}\n\n\tg.logger.Debug(\"direct peer chooser got a membership update\", tag.Counter(len(members)))\n\n\t// If the chooser is not started, do not act on membership changes.\n\t// If membership updates arrive after chooser is stopped, ignore them.\n\tif atomic.LoadInt32(&g.status) != common.DaemonStatusStarted {\n\t\treturn\n\t}\n\n\tg.updatePeersInternal(members)\n}\n\nfunc (g *directPeerChooser) updatePeersInternal(members []membership.HostInfo) {\n\t// Create a map of valid peer addresses given members list.\n\tvalidPeerAddresses := make(map[string]bool)\n\tfor _, member := range members {\n\t\tfor _, portName := range []string{membership.PortTchannel, membership.PortGRPC} {\n\t\t\taddr, err := member.GetNamedAddress(portName)\n\t\t\tif err != nil {\n\t\t\t\tg.logger.Error(fmt.Sprintf(\"failed to get %s address of member\", portName), tag.Error(err), tag.Address(member.GetAddress()))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tvalidPeerAddresses[addr] = true\n\t\t}\n\t}\n\n\t// Take a copy of the current peers to avoid keeping write lock while removing all peers.\n\tpeers := make(map[string]bool)\n\tg.mu.RLock()\n\tfor addr := range g.peers {\n\t\tpeers[addr] = true\n\t}\n\tg.mu.RUnlock()\n\n\tg.logger.Debugf(\"valid peers: %v, current peers: %v\", validPeerAddresses, peers)\n\n\tfor addr := range peers {\n\t\tif !validPeerAddresses[addr] {\n\t\t\tg.removePeer(addr)\n\t\t}\n\t}\n}\n\nfunc (g *directPeerChooser) removePeer(addr string) {\n\tg.mu.RLock()\n\tpeer, exists := g.peers[addr]\n\tif exists && peer != nil {\n\t\tif err := g.t.ReleasePeer(peer, noOpSubscriberInstance); err != nil {\n\t\t\tg.logger.Error(\"failed to release peer\", tag.Error(err), tag.Address(addr))\n\t\t}\n\t} else {\n\t\tif peer == nil {\n\t\t\tg.logger.Error(\"peer is nil\", tag.Address(addr), tag.Dynamic(\"stack\", string(debug.Stack())))\n\t\t} else {\n\t\t\tg.logger.Error(\"peer is not in the peers map\", tag.Address(addr), tag.Dynamic(\"stack\", string(debug.Stack())))\n\t\t}\n\t}\n\tg.mu.RUnlock()\n\n\tg.mu.Lock()\n\tdefer g.mu.Unlock()\n\n\tdelete(g.peers, addr)\n\tg.logger.Info(\"removed peer from direct peer chooser\", tag.Address(addr))\n\tg.scope.IncCounter(metrics.P2PPeerRemoved)\n\tg.scope.UpdateGauge(metrics.P2PPeersCount, float64(len(g.peers)))\n}\n\nfunc (g *directPeerChooser) addPeer(addr string) (peer.Peer, error) {\n\tg.mu.Lock()\n\tdefer g.mu.Unlock()\n\tif p, ok := g.peers[addr]; ok {\n\t\treturn p, nil\n\t}\n\n\tp, err := g.t.RetainPeer(hostport.Identify(addr), noOpSubscriberInstance)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tg.peers[addr] = p\n\tg.logger.Info(\"added peer to direct peer chooser\", tag.Address(addr))\n\tg.scope.IncCounter(metrics.P2PPeerAdded)\n\tg.scope.UpdateGauge(metrics.P2PPeersCount, float64(len(g.peers)))\n\treturn p, nil\n}\n\nfunc (g *directPeerChooser) chooseFromLegacyDirectPeerChooser(ctx context.Context, req *transport.Request) (peer.Peer, func(error), error) {\n\tc, ok := g.getLegacyChooser()\n\tif !ok {\n\t\treturn nil, nil, yarpcerrors.InternalErrorf(\"failed to get legacy direct peer chooser, err: %v\", g.legacyChooserErr)\n\t}\n\n\treturn c.Choose(ctx, req)\n}\n\nfunc (g *directPeerChooser) getLegacyChooser() (peer.Chooser, bool) {\n\tg.mu.RLock()\n\n\tif g.legacyChooser != nil {\n\t\t// Legacy chooser already created, return it\n\t\tg.mu.RUnlock()\n\t\treturn g.legacyChooser, true\n\t}\n\n\tif g.legacyChooserErr != nil {\n\t\t// There was an error creating the legacy chooser, return false\n\t\tg.mu.RUnlock()\n\t\treturn nil, false\n\t}\n\n\tg.mu.RUnlock()\n\n\tg.mu.Lock()\n\tg.legacyChooser, g.legacyChooserErr = direct.New(direct.Configuration{}, g.t)\n\tg.mu.Unlock()\n\n\tif g.legacyChooserErr != nil {\n\t\tg.logger.Error(\"failed to create legacy direct peer chooser\", tag.Error(g.legacyChooserErr))\n\t\treturn nil, false\n\t}\n\n\tif atomic.LoadInt32(&g.status) == common.DaemonStatusStarted {\n\t\t// Start the legacy chooser if the current chooser is already started\n\t\tif err := g.legacyChooser.Start(); err != nil {\n\t\t\tg.logger.Error(\"failed to start legacy direct peer chooser\", tag.Error(err))\n\t\t\treturn nil, false\n\t\t}\n\t}\n\n\treturn g.legacyChooser, true\n}\n\n// noOpSubscriber is a no-op implementation of peer.Subscriber\ntype noOpSubscriber struct{}\n\nfunc (*noOpSubscriber) NotifyStatusChanged(peer.Identifier) {}\n"
  },
  {
    "path": "common/rpc/direct_peer_chooser_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestDirectChooser_PeerUpdates(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tmetricCl := metrics.NewNoopMetricsClient()\n\tserviceName := \"service\"\n\tdirectConnRetainFn := func(opts ...dynamicproperties.FilterOption) bool { return true }\n\tgrpcTransport := grpc.NewTransport()\n\tchooser := newDirectChooser(serviceName, grpcTransport, logger, metricCl, directConnRetainFn)\n\n\tchoosePeers := func(peers ...string) {\n\t\tt.Helper()\n\t\t// Calling Choose() will create peers and they will be cached\n\t\tfor _, p := range peers {\n\t\t\t_, onFinish, err := chooser.Choose(context.Background(), &transport.Request{\n\t\t\t\tCaller:   \"caller\",\n\t\t\t\tService:  \"service\",\n\t\t\t\tShardKey: p,\n\t\t\t})\n\t\t\tassert.NoError(t, err, \"Choose() failed\")\n\t\t\tonFinish(nil)\n\t\t}\n\t}\n\n\tcurrentPeersMap := func() map[string]bool {\n\t\tt.Helper()\n\t\tchooser.mu.RLock()\n\t\tdefer chooser.mu.RUnlock()\n\t\tpeers := make(map[string]bool, len(chooser.peers))\n\t\tfor p := range chooser.peers {\n\t\t\tpeers[p] = true\n\t\t}\n\t\treturn peers\n\t}\n\n\tnewHost := func(peer string) membership.HostInfo {\n\t\treturn membership.NewDetailedHostInfo(peer+\":80\", peer, membership.PortMap{\n\t\t\tmembership.PortGRPC: 80,\n\t\t})\n\t}\n\n\tt.Run(\"chooser not started so should discard membership updates\", func(t *testing.T) {\n\t\tchoosePeers(\"peer1:80\", \"peer2:80\")\n\t\tchooser.UpdatePeers(serviceName, nil)\n\t\twantPeers := map[string]bool{\"peer1:80\": true, \"peer2:80\": true}\n\t\tgotPeers := currentPeersMap()\n\t\tif diff := cmp.Diff(wantPeers, gotPeers); diff != \"\" {\n\t\t\tt.Fatalf(\"Peers mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t})\n\n\t// Start chooser and do more validations\n\tif err := chooser.Start(); err != nil {\n\t\tt.Fatalf(\"failed to start direct peer chooser: %v\", err)\n\t}\n\n\tdefer chooser.Stop()\n\n\tt.Run(\"peer1 and peer2 are chosen, peer2 is removed from members list\", func(t *testing.T) {\n\t\tchoosePeers(\"peer1:80\", \"peer2:80\")\n\t\tchooser.UpdatePeers(serviceName, []membership.HostInfo{\n\t\t\tnewHost(\"peer1\"),\n\t\t})\n\t\twantPeers := map[string]bool{\"peer1:80\": true}\n\t\tgotPeers := currentPeersMap()\n\t\tif diff := cmp.Diff(wantPeers, gotPeers); diff != \"\" {\n\t\t\tt.Fatalf(\"Peers mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t})\n\n\tt.Run(\"peer3 and peer4 are also chosen, membership list has peer1 and peer4\", func(t *testing.T) {\n\t\tchoosePeers(\"peer3:80\", \"peer4:80\")\n\t\tchooser.UpdatePeers(serviceName, []membership.HostInfo{\n\t\t\tnewHost(\"peer1\"),\n\t\t\tnewHost(\"peer4\"),\n\t\t})\n\t\twantPeers := map[string]bool{\"peer1:80\": true, \"peer4:80\": true}\n\t\tgotPeers := currentPeersMap()\n\t\tif diff := cmp.Diff(wantPeers, gotPeers); diff != \"\" {\n\t\t\tt.Fatalf(\"Peers mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t})\n\n\tt.Run(\"membership list update for another service is ignored, should still keep peer1 and peer4\", func(t *testing.T) {\n\t\tchooser.UpdatePeers(\"another-service\", []membership.HostInfo{\n\t\t\tnewHost(\"peer50\"),\n\t\t})\n\t\twantPeers := map[string]bool{\"peer1:80\": true, \"peer4:80\": true}\n\t\tgotPeers := currentPeersMap()\n\t\tif diff := cmp.Diff(wantPeers, gotPeers); diff != \"\" {\n\t\t\tt.Fatalf(\"Peers mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t})\n}\n\nfunc TestDirectChooser_StartStop(t *testing.T) {\n\tnewReq := func(shardKey string) *transport.Request {\n\t\treturn &transport.Request{\n\t\t\tCaller:   \"caller\",\n\t\t\tService:  \"service\",\n\t\t\tShardKey: shardKey,\n\t\t}\n\t}\n\n\ttests := []struct {\n\t\tdesc           string\n\t\tretainConn     bool\n\t\treq            *transport.Request\n\t\tmultipleChoose bool\n\t\twantChooseErr  bool\n\t}{\n\t\t{\n\t\t\tdesc:       \"legacy chooser\",\n\t\t\tretainConn: false,\n\t\t\treq:        newReq(\"key\"),\n\t\t},\n\t\t{\n\t\t\tdesc:          \"legacy chooser - empty shard key\",\n\t\t\tretainConn:    false,\n\t\t\treq:           newReq(\"\"),\n\t\t\twantChooseErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:       \"connection retain mode\",\n\t\t\tretainConn: true,\n\t\t\treq:        newReq(\"key\"),\n\t\t},\n\t\t{\n\t\t\tdesc:          \"connection retain mode - empty shard key\",\n\t\t\tretainConn:    true,\n\t\t\treq:           newReq(\"\"),\n\t\t\twantChooseErr: true,\n\t\t},\n\t\t{\n\t\t\tdesc:           \"connection retain mode - multiple choose should return chooser from cache\",\n\t\t\tretainConn:     true,\n\t\t\treq:            newReq(\"key\"),\n\t\t\tmultipleChoose: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tdefer goleak.VerifyNone(t)\n\n\t\t\tlogger := testlogger.New(t)\n\t\t\tmetricCl := metrics.NewNoopMetricsClient()\n\t\t\tserviceName := \"service\"\n\t\t\tdirectConnRetainFn := func(opts ...dynamicproperties.FilterOption) bool { return tc.retainConn }\n\t\t\tgrpcTransport := grpc.NewTransport()\n\n\t\t\tchooser := newDirectChooser(serviceName, grpcTransport, logger, metricCl, directConnRetainFn)\n\n\t\t\tassert.False(t, chooser.IsRunning(), \"expected IsRunning()=false before Start()\")\n\n\t\t\tif err := chooser.Start(); err != nil {\n\t\t\t\tt.Fatalf(\"failed to start direct peer chooser: %v\", err)\n\t\t\t}\n\n\t\t\tassert.NoError(t, chooser.Start(), \"starting again should be no-op\")\n\n\t\t\tassert.True(t, chooser.IsRunning())\n\n\t\t\tpeer, onFinish, err := chooser.Choose(context.Background(), tc.req)\n\t\t\tif tc.wantChooseErr != (err != nil) {\n\t\t\t\tt.Fatalf(\"Choose() err = %v, wantChooseErr = %v\", err, tc.wantChooseErr)\n\t\t\t}\n\n\t\t\tif err == nil {\n\t\t\t\tassert.NotNil(t, peer)\n\t\t\t\tassert.NotNil(t, onFinish)\n\n\t\t\t\t// call onFinish will release the peer for legacy chooser\n\t\t\t\tonFinish(nil)\n\t\t\t}\n\n\t\t\tif tc.multipleChoose {\n\t\t\t\tpeer2, onFinish2, err2 := chooser.Choose(context.Background(), tc.req)\n\t\t\t\tassert.NoError(t, err2)\n\t\t\t\tassert.NotNil(t, onFinish2)\n\t\t\t\tassert.Equal(t, peer, peer2)\n\t\t\t\tonFinish2(nil)\n\t\t\t}\n\n\t\t\tif err := chooser.Stop(); err != nil {\n\t\t\t\tt.Fatalf(\"failed to stop direct peer chooser: %v\", err)\n\t\t\t}\n\n\t\t\tassert.NoError(t, chooser.Stop(), \"stopping again should be no-op\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/rpc/dns_updater.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc/api/peer\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype (\n\tdnsUpdater struct {\n\t\tresolver     dnsHostResolver\n\t\tinterval     time.Duration\n\t\tdnsAddress   string\n\t\tport         string\n\t\tcurrentPeers map[string]struct{}\n\t\tlist         peer.List\n\t\tlogger       log.Logger\n\t\twg           sync.WaitGroup\n\t\tctx          context.Context\n\t\tcancel       context.CancelFunc\n\t}\n\tdnsRefreshResult struct {\n\t\tupdates  peer.ListUpdates\n\t\tnewPeers map[string]struct{}\n\t\tchanged  bool\n\t}\n\taPeer struct {\n\t\taddrPort string\n\t}\n)\n\ntype dnsHostResolver interface {\n\tLookupHost(ctx context.Context, host string) (addrs []string, err error)\n}\n\nfunc newDNSUpdater(list peer.List, dnsPort string, interval time.Duration, logger log.Logger) (*dnsUpdater, error) {\n\tss := strings.Split(dnsPort, \":\")\n\tif len(ss) != 2 {\n\t\treturn nil, fmt.Errorf(\"incorrect DNS:Port format\")\n\t}\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &dnsUpdater{\n\t\tresolver:     net.DefaultResolver,\n\t\tinterval:     interval,\n\t\tlogger:       logger,\n\t\tlist:         list,\n\t\tdnsAddress:   ss[0],\n\t\tport:         ss[1],\n\t\tcurrentPeers: make(map[string]struct{}),\n\t\tctx:          ctx,\n\t\tcancel:       cancel,\n\t}, nil\n}\n\nfunc (d *dnsUpdater) Start() {\n\td.wg.Add(1)\n\tgo func() {\n\t\tdefer d.wg.Done()\n\t\tfor {\n\t\t\tnow := time.Now()\n\t\t\tres, err := d.refresh()\n\t\t\tif err != nil {\n\t\t\t\td.logger.Error(\"Failed to update DNS\", tag.Error(err), tag.Address(d.dnsAddress))\n\t\t\t}\n\t\t\tif res != nil && res.changed {\n\t\t\t\tif len(res.updates.Additions) > 0 {\n\t\t\t\t\td.logger.Info(\"Add new peers by DNS lookup\", tag.Address(d.dnsAddress), tag.Addresses(identifiersToStringList(res.updates.Additions)))\n\t\t\t\t}\n\t\t\t\tif len(res.updates.Removals) > 0 {\n\t\t\t\t\td.logger.Info(\"Remove stale peers by DNS lookup\", tag.Address(d.dnsAddress), tag.Addresses(identifiersToStringList(res.updates.Removals)))\n\t\t\t\t}\n\n\t\t\t\terr := d.list.Update(res.updates)\n\t\t\t\tif err != nil {\n\t\t\t\t\td.logger.Error(\"Failed to update peerList\", tag.Error(err), tag.Address(d.dnsAddress))\n\t\t\t\t} else {\n\t\t\t\t\td.currentPeers = res.newPeers\n\t\t\t\t}\n\t\t\t}\n\t\t\tsleepDu := now.Add(d.interval).Sub(now)\n\t\t\tt := time.NewTimer(sleepDu)\n\t\t\tselect {\n\t\t\tcase <-d.ctx.Done():\n\t\t\t\tt.Stop()\n\t\t\t\td.logger.Info(\"DNS updater is stopping so returning from dns update loop\", tag.Address(d.dnsAddress))\n\t\t\t\treturn\n\t\t\tcase <-t.C:\n\t\t\t\tcontinue\n\t\t\t}\n\t\t}\n\t}()\n}\n\nfunc (d *dnsUpdater) Stop() {\n\td.logger.Info(\"DNS updater is stopping\", tag.Address(d.dnsAddress))\n\td.cancel()\n\td.wg.Wait()\n\td.logger.Info(\"DNS updater stopped\", tag.Address(d.dnsAddress))\n}\n\nfunc (d *dnsUpdater) refresh() (*dnsRefreshResult, error) {\n\tips, err := d.resolver.LookupHost(d.ctx, d.dnsAddress)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tnewPeers := map[string]struct{}{}\n\tfor _, ip := range ips {\n\t\tadr := fmt.Sprintf(\"%v:%v\", ip, d.port)\n\t\tnewPeers[adr] = struct{}{}\n\t}\n\n\tupdates := peer.ListUpdates{\n\t\tAdditions: make([]peer.Identifier, 0),\n\t\tRemovals:  make([]peer.Identifier, 0),\n\t}\n\tchanged := false\n\t// remove if it doesn't exist anymore\n\tfor addr := range d.currentPeers {\n\t\tif _, ok := newPeers[addr]; !ok {\n\t\t\tchanged = true\n\t\t\tupdates.Removals = append(\n\t\t\t\tupdates.Removals,\n\t\t\t\taPeer{addrPort: addr},\n\t\t\t)\n\t\t}\n\t}\n\n\t// add if it doesn't exist before\n\tfor addr := range newPeers {\n\t\tif _, ok := d.currentPeers[addr]; !ok {\n\t\t\tchanged = true\n\t\t\tupdates.Additions = append(\n\t\t\t\tupdates.Additions,\n\t\t\t\taPeer{addrPort: addr},\n\t\t\t)\n\t\t}\n\t}\n\n\treturn &dnsRefreshResult{\n\t\tupdates:  updates,\n\t\tnewPeers: newPeers,\n\t\tchanged:  changed,\n\t}, nil\n}\n\nfunc (a aPeer) Identifier() string {\n\treturn a.addrPort\n}\n\nfunc identifiersToStringList(ids []peer.Identifier) []string {\n\tss := make([]string, 0, len(ids))\n\tfor _, id := range ids {\n\t\tss = append(ss, id.Identifier())\n\t}\n\treturn ss\n}\n"
  },
  {
    "path": "common/rpc/dns_updater_test.go",
    "content": "package rpc\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/yarpc/api/peer\"\n\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype mockDNSHostResolver struct {\n\taddrsToReturn []string\n\terrToReturn   error\n}\n\nfunc (m *mockDNSHostResolver) LookupHost(ctx context.Context, host string) ([]string, error) {\n\treturn m.addrsToReturn, m.errToReturn\n}\n\ntype dummyPeerList struct {\n\tupdates     []peer.ListUpdates\n\terrToReturn error\n}\n\nfunc (d *dummyPeerList) Update(updates peer.ListUpdates) error {\n\td.updates = append(d.updates, updates)\n\treturn d.errToReturn\n}\n\nfunc TestDNSUpdater_Refresh(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tinitialPeers   []string\n\t\tnewPeers       []string\n\t\tcurrentPeers   map[string]struct{}\n\t\twantChanged    bool\n\t\twantNewPeers   map[string]struct{}\n\t\twantAdditions  []string\n\t\twantRemovals   []string\n\t\twantDNSFailure bool\n\t}{\n\t\t{\n\t\t\tname:         \"Initial state: two peers\",\n\t\t\tinitialPeers: nil,\n\t\t\tnewPeers:     []string{\"10.0.0.1\", \"10.0.0.2\"},\n\t\t\tcurrentPeers: nil,\n\t\t\twantChanged:  true,\n\t\t\twantNewPeers: map[string]struct{}{\n\t\t\t\t\"10.0.0.1:1234\": {},\n\t\t\t\t\"10.0.0.2:1234\": {},\n\t\t\t},\n\t\t\twantAdditions: []string{\"10.0.0.1:1234\", \"10.0.0.2:1234\"},\n\t\t\twantRemovals:  nil,\n\t\t},\n\t\t{\n\t\t\tname:         \"Change DNS, one removed, one added, one persistent\",\n\t\t\tinitialPeers: []string{\"10.0.0.1\", \"10.0.0.2\"},\n\t\t\tnewPeers:     []string{\"10.0.0.2\", \"10.0.0.3\"},\n\t\t\tcurrentPeers: map[string]struct{}{\n\t\t\t\t\"10.0.0.1:1234\": {},\n\t\t\t\t\"10.0.0.2:1234\": {},\n\t\t\t},\n\t\t\twantChanged: true,\n\t\t\twantNewPeers: map[string]struct{}{\n\t\t\t\t\"10.0.0.2:1234\": {},\n\t\t\t\t\"10.0.0.3:1234\": {},\n\t\t\t},\n\t\t\twantAdditions: []string{\"10.0.0.3:1234\"},\n\t\t\twantRemovals:  []string{\"10.0.0.1:1234\"},\n\t\t},\n\t\t{\n\t\t\tname:         \"No changes\",\n\t\t\tinitialPeers: []string{\"10.0.0.2\", \"10.0.0.3\"},\n\t\t\tnewPeers:     []string{\"10.0.0.2\", \"10.0.0.3\"},\n\t\t\tcurrentPeers: map[string]struct{}{\n\t\t\t\t\"10.0.0.2:1234\": {},\n\t\t\t\t\"10.0.0.3:1234\": {},\n\t\t\t},\n\t\t\twantChanged: false,\n\t\t\twantNewPeers: map[string]struct{}{\n\t\t\t\t\"10.0.0.2:1234\": {},\n\t\t\t\t\"10.0.0.3:1234\": {},\n\t\t\t},\n\t\t\twantAdditions: nil,\n\t\t\twantRemovals:  nil,\n\t\t},\n\t\t{\n\t\t\tname:           \"DNS failure\",\n\t\t\tinitialPeers:   []string{\"10.0.0.1\", \"10.0.0.2\"},\n\t\t\tnewPeers:       nil,\n\t\t\tcurrentPeers:   map[string]struct{}{\"10.0.0.1:1234\": {}, \"10.0.0.2:1234\": {}},\n\t\t\twantChanged:    false,\n\t\t\twantNewPeers:   nil,\n\t\t\twantAdditions:  nil,\n\t\t\twantRemovals:   nil,\n\t\t\twantDNSFailure: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockResolver := &mockDNSHostResolver{}\n\t\t\tdnsAddr := \"test.service.com:1234\"\n\t\t\tinterval := 1 * time.Second\n\t\t\tlogger := log.NewNoop()\n\t\t\ttestList := &dummyPeerList{}\n\n\t\t\tupdater, err := newDNSUpdater(testList, dnsAddr, interval, logger)\n\t\t\trequire.NoError(t, err, \"should create dnsUpdater\")\n\t\t\tupdater.resolver = mockResolver\n\n\t\t\t// Optionally set initial currentPeers\n\t\t\tif tt.currentPeers != nil {\n\t\t\t\tupdater.currentPeers = tt.currentPeers\n\t\t\t}\n\n\t\t\tif tt.wantDNSFailure {\n\t\t\t\tmockResolver.errToReturn = errors.New(\"mock DNS error\")\n\t\t\t} else {\n\t\t\t\tmockResolver.errToReturn = nil\n\t\t\t\tmockResolver.addrsToReturn = tt.newPeers\n\t\t\t}\n\n\t\t\tres, err := updater.refresh()\n\n\t\t\tif tt.wantDNSFailure {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.wantChanged, res.changed)\n\t\t\t\tassert.Equal(t, tt.wantNewPeers, res.newPeers)\n\t\t\t\tactualAdditions := identifiersToStringList(res.updates.Additions)\n\t\t\t\tactualRemovals := identifiersToStringList(res.updates.Removals)\n\t\t\t\tassert.ElementsMatch(t, tt.wantAdditions, actualAdditions)\n\t\t\t\tassert.ElementsMatch(t, tt.wantRemovals, actualRemovals)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDNSUpdater_UpdaterFailure(t *testing.T) {\n\tmockResolver := &mockDNSHostResolver{\n\t\taddrsToReturn: []string{\"10.0.0.3\", \"10.0.0.4\"},\n\t}\n\tlogger := log.NewNoop()\n\ttestList := &dummyPeerList{errToReturn: errors.New(\"mock updater error\")}\n\n\tupdater, err := newDNSUpdater(testList, \"host.test:1234\", 100*time.Millisecond, logger)\n\trequire.NoError(t, err)\n\tupdater.resolver = mockResolver\n\tupdater.currentPeers = map[string]struct{}{\n\t\t\"10.0.0.1:1234\": {},\n\t\t\"10.0.0.2:1234\": {},\n\t}\n\n\tupdater.Start()\n\tdefer updater.Stop()\n\n\ttime.Sleep(1 * time.Second)\n\tassert.Equal(t, map[string]struct{}{\"10.0.0.1:1234\": {}, \"10.0.0.2:1234\": {}}, updater.currentPeers)\n}\n"
  },
  {
    "path": "common/rpc/factory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"crypto/tls\"\n\t\"fmt\"\n\t\"net\"\n\tnethttp \"net/http\"\n\t\"sync\"\n\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\tyarpchttp \"go.uber.org/yarpc/transport/http\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\t\"google.golang.org/grpc/credentials\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\tdefaultGRPCSizeLimit = 4 * 1024 * 1024\n\tfactoryComponentName = \"rpc-factory\"\n)\n\nvar (\n\t// P2P outbounds are only needed for history and matching services\n\tservicesToTalkP2P = []string{service.History, service.Matching}\n)\n\n// Factory is an implementation of rpc.Factory interface\ntype FactoryImpl struct {\n\tstartOnce      sync.Once\n\tstopOnce       sync.Once\n\tmaxMessageSize int\n\tchannel        tchannel.Channel\n\tdispatcher     *yarpc.Dispatcher\n\toutbounds      *Outbounds\n\tlogger         log.Logger\n\tserviceName    string\n\twg             sync.WaitGroup\n\tctx            context.Context\n\tcancelFn       context.CancelFunc\n\tpeerLister     PeerLister\n}\n\n// NewFactory builds a new rpc.Factory\nfunc NewFactory(logger log.Logger, p Params) Factory {\n\tlogger = logger.WithTags(tag.ComponentRPCFactory)\n\n\tinbounds := yarpc.Inbounds{}\n\t// Create TChannel transport\n\t// This is here only because ringpop expects tchannel.ChannelTransport,\n\t// everywhere else we use regular tchannel.Transport.\n\tch, err := tchannel.NewChannelTransport(\n\t\ttchannel.ServiceName(p.ServiceName),\n\t\ttchannel.ListenAddr(p.TChannelAddress))\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to create transport channel\", tag.Error(err))\n\t}\n\ttchannel, err := tchannel.NewTransport(tchannel.ServiceName(p.ServiceName))\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to create tchannel transport\", tag.Error(err))\n\t}\n\n\tinbounds = append(inbounds, ch.NewInbound())\n\tlogger.Info(\"Listening for TChannel requests\", tag.Address(p.TChannelAddress))\n\n\t// Create gRPC transport\n\tvar options []grpc.TransportOption\n\tif p.GRPCMaxMsgSize > 0 {\n\t\toptions = append(options, grpc.ServerMaxRecvMsgSize(p.GRPCMaxMsgSize))\n\t\toptions = append(options, grpc.ClientMaxRecvMsgSize(p.GRPCMaxMsgSize))\n\t}\n\tgrpcTransport := grpc.NewTransport(options...)\n\tif len(p.GRPCAddress) > 0 {\n\t\tlistener, err := net.Listen(\"tcp\", p.GRPCAddress)\n\t\tif err != nil {\n\t\t\tlogger.Fatal(\"Failed to listen on GRPC port\", tag.Error(err))\n\t\t}\n\n\t\tvar inboundOptions []grpc.InboundOption\n\t\tif p.InboundTLS != nil {\n\t\t\tinboundOptions = append(inboundOptions, grpc.InboundCredentials(credentials.NewTLS(p.InboundTLS)))\n\t\t}\n\n\t\tinbounds = append(inbounds, grpcTransport.NewInbound(listener, inboundOptions...))\n\t\tlogger.Info(\"Listening for GRPC requests\", tag.Address(p.GRPCAddress))\n\t}\n\t// Create http inbound if configured\n\tif p.HTTP != nil {\n\t\tinterceptor := func(handler nethttp.Handler) nethttp.Handler {\n\t\t\treturn nethttp.HandlerFunc(func(w nethttp.ResponseWriter, r *nethttp.Request) {\n\t\t\t\tprocedure := r.Header.Get(yarpchttp.ProcedureHeader)\n\t\t\t\tif _, found := p.HTTP.Procedures[procedure]; found {\n\t\t\t\t\thandler.ServeHTTP(w, r)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tnethttp.NotFound(w, r)\n\t\t\t})\n\t\t}\n\n\t\tinboundOptions := []yarpchttp.InboundOption{yarpchttp.Interceptor(interceptor)}\n\n\t\tif p.HTTP.TLS != nil {\n\t\t\tinboundOptions = append(inboundOptions,\n\t\t\t\tyarpchttp.InboundTLSConfiguration(p.HTTP.TLS),\n\t\t\t\tyarpchttp.InboundTLSMode(p.HTTP.Mode))\n\t\t\tlogger.Info(fmt.Sprintf(\"Enabling HTTP TLS with Mode %q\", p.HTTP.Mode))\n\t\t}\n\n\t\thttpinbound := yarpchttp.NewTransport().NewInbound(p.HTTP.Address, inboundOptions...)\n\n\t\tinbounds = append(inbounds, httpinbound)\n\t\tlogger.Info(\"Listening for HTTP requests\", tag.Address(p.HTTP.Address))\n\t}\n\t// Create outbounds\n\toutbounds := &Outbounds{}\n\tif p.OutboundsBuilder != nil {\n\t\toutbounds, err = p.OutboundsBuilder.Build(grpcTransport, tchannel)\n\t\tif err != nil {\n\t\t\tlogger.Fatal(\"Failed to create outbounds\", tag.Error(err))\n\t\t}\n\t}\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName:               p.ServiceName,\n\t\tInbounds:           inbounds,\n\t\tOutbounds:          outbounds.Outbounds,\n\t\tInboundMiddleware:  p.InboundMiddleware,\n\t\tOutboundMiddleware: p.OutboundMiddleware,\n\t})\n\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &FactoryImpl{\n\t\tmaxMessageSize: p.GRPCMaxMsgSize,\n\t\tdispatcher:     dispatcher,\n\t\tchannel:        ch.Channel(),\n\t\toutbounds:      outbounds,\n\t\tserviceName:    p.ServiceName,\n\t\tlogger:         logger,\n\t\tctx:            ctx,\n\t\tcancelFn:       cancel,\n\t}\n}\n\n// GetDispatcher return a cached dispatcher\nfunc (d *FactoryImpl) GetDispatcher() *yarpc.Dispatcher {\n\treturn d.dispatcher\n}\n\n// GetTChannel GetChannel returns Tchannel Channel used by Ringpop\nfunc (d *FactoryImpl) GetTChannel() tchannel.Channel {\n\treturn d.channel\n}\n\nfunc (d *FactoryImpl) GetMaxMessageSize() int {\n\tif d.maxMessageSize == 0 {\n\t\treturn defaultGRPCSizeLimit\n\t}\n\treturn d.maxMessageSize\n}\n\nfunc (d *FactoryImpl) Start(peerLister PeerLister) error {\n\tvar err error\n\td.startOnce.Do(func() {\n\t\td.peerLister = peerLister\n\t\t// subscribe to membership changes for history and matching. This is needed to update the peers for rpc\n\t\tfor _, svc := range servicesToTalkP2P {\n\t\t\tch := make(chan *membership.ChangedEvent, 1)\n\t\t\tif err = d.peerLister.Subscribe(svc, factoryComponentName, ch); err != nil {\n\t\t\t\terr = fmt.Errorf(\"rpc factory failed to subscribe to membership updates for svc: %v, err: %w\", svc, err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\td.wg.Add(1)\n\t\t\tgo d.listenMembershipChanges(svc, ch)\n\t\t}\n\t})\n\treturn err\n}\n\nfunc (d *FactoryImpl) Stop() error {\n\td.stopOnce.Do(func() {\n\t\td.logger.Info(\"stopping rpc factory\")\n\n\t\tfor _, svc := range servicesToTalkP2P {\n\t\t\tif err := d.peerLister.Unsubscribe(svc, factoryComponentName); err != nil {\n\t\t\t\td.logger.Error(\"rpc factory failed to unsubscribe from membership updates\", tag.Error(err), tag.Service(svc))\n\t\t\t}\n\t\t}\n\n\t\td.cancelFn()\n\t\td.wg.Wait()\n\n\t\td.logger.Info(\"stopped rpc factory\")\n\t})\n\treturn nil\n}\n\nfunc (d *FactoryImpl) listenMembershipChanges(svc string, ch chan *membership.ChangedEvent) {\n\tdefer d.wg.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ch:\n\t\t\td.logger.Debug(\"rpc factory received membership changed event\", tag.Service(svc))\n\t\t\tmembers, err := d.peerLister.Members(svc)\n\t\t\tif err != nil {\n\t\t\t\td.logger.Error(\"rpc factory failed to get members from membership resolver\", tag.Error(err), tag.Service(svc))\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\td.outbounds.UpdatePeers(svc, members)\n\t\tcase <-d.ctx.Done():\n\t\t\td.logger.Info(\"rpc factory stopped so listenMembershipChanges returning\", tag.Service(svc))\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc createDialer(transport *grpc.Transport, tlsConfig *tls.Config) *grpc.Dialer {\n\tvar dialOptions []grpc.DialOption\n\tif tlsConfig != nil {\n\t\tdialOptions = append(dialOptions, grpc.DialerCredentials(credentials.NewTLS(tlsConfig)))\n\t}\n\treturn transport.NewDialer(dialOptions...)\n}\n"
  },
  {
    "path": "common/rpc/factory_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: types.go\n//\n// Generated by this command:\n//\n//\tmockgen -package rpc -source types.go -destination factory_mock.go -self_package github.com/uber/cadence/common/rpc\n//\n\n// Package rpc is a generated GoMock package.\npackage rpc\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\ttchannel \"go.uber.org/yarpc/transport/tchannel\"\n\n\tmembership \"github.com/uber/cadence/common/membership\"\n)\n\n// MockFactory is a mock of Factory interface.\ntype MockFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockFactoryMockRecorder is the mock recorder for MockFactory.\ntype MockFactoryMockRecorder struct {\n\tmock *MockFactory\n}\n\n// NewMockFactory creates a new mock instance.\nfunc NewMockFactory(ctrl *gomock.Controller) *MockFactory {\n\tmock := &MockFactory{ctrl: ctrl}\n\tmock.recorder = &MockFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFactory) EXPECT() *MockFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// GetDispatcher mocks base method.\nfunc (m *MockFactory) GetDispatcher() *yarpc.Dispatcher {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDispatcher\")\n\tret0, _ := ret[0].(*yarpc.Dispatcher)\n\treturn ret0\n}\n\n// GetDispatcher indicates an expected call of GetDispatcher.\nfunc (mr *MockFactoryMockRecorder) GetDispatcher() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDispatcher\", reflect.TypeOf((*MockFactory)(nil).GetDispatcher))\n}\n\n// GetMaxMessageSize mocks base method.\nfunc (m *MockFactory) GetMaxMessageSize() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMaxMessageSize\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetMaxMessageSize indicates an expected call of GetMaxMessageSize.\nfunc (mr *MockFactoryMockRecorder) GetMaxMessageSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMaxMessageSize\", reflect.TypeOf((*MockFactory)(nil).GetMaxMessageSize))\n}\n\n// GetTChannel mocks base method.\nfunc (m *MockFactory) GetTChannel() tchannel.Channel {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTChannel\")\n\tret0, _ := ret[0].(tchannel.Channel)\n\treturn ret0\n}\n\n// GetTChannel indicates an expected call of GetTChannel.\nfunc (mr *MockFactoryMockRecorder) GetTChannel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTChannel\", reflect.TypeOf((*MockFactory)(nil).GetTChannel))\n}\n\n// Start mocks base method.\nfunc (m *MockFactory) Start(arg0 PeerLister) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockFactoryMockRecorder) Start(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockFactory)(nil).Start), arg0)\n}\n\n// Stop mocks base method.\nfunc (m *MockFactory) Stop() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Stop\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockFactoryMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockFactory)(nil).Stop))\n}\n\n// MockPeerLister is a mock of PeerLister interface.\ntype MockPeerLister struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerListerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerListerMockRecorder is the mock recorder for MockPeerLister.\ntype MockPeerListerMockRecorder struct {\n\tmock *MockPeerLister\n}\n\n// NewMockPeerLister creates a new mock instance.\nfunc NewMockPeerLister(ctrl *gomock.Controller) *MockPeerLister {\n\tmock := &MockPeerLister{ctrl: ctrl}\n\tmock.recorder = &MockPeerListerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerLister) EXPECT() *MockPeerListerMockRecorder {\n\treturn m.recorder\n}\n\n// Members mocks base method.\nfunc (m *MockPeerLister) Members(service string) ([]membership.HostInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Members\", service)\n\tret0, _ := ret[0].([]membership.HostInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Members indicates an expected call of Members.\nfunc (mr *MockPeerListerMockRecorder) Members(service any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Members\", reflect.TypeOf((*MockPeerLister)(nil).Members), service)\n}\n\n// Subscribe mocks base method.\nfunc (m *MockPeerLister) Subscribe(service, name string, notifyChannel chan<- *membership.ChangedEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Subscribe\", service, name, notifyChannel)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockPeerListerMockRecorder) Subscribe(service, name, notifyChannel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockPeerLister)(nil).Subscribe), service, name, notifyChannel)\n}\n\n// Unsubscribe mocks base method.\nfunc (m *MockPeerLister) Unsubscribe(service, name string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Unsubscribe\", service, name)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Unsubscribe indicates an expected call of Unsubscribe.\nfunc (mr *MockPeerListerMockRecorder) Unsubscribe(service, name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unsubscribe\", reflect.TypeOf((*MockPeerLister)(nil).Unsubscribe), service, name)\n}\n"
  },
  {
    "path": "common/rpc/factory_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestNewFactory(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\tserviceName := \"service\"\n\tob := NewMockOutboundsBuilder(ctrl)\n\tob.EXPECT().Build(gomock.Any(), gomock.Any()).Return(&Outbounds{}, nil).Times(1)\n\tgrpcMsgSize := 4 * 1024 * 1024\n\tf := NewFactory(logger, Params{\n\t\tServiceName:     serviceName,\n\t\tTChannelAddress: \"localhost:0\",\n\t\tGRPCMaxMsgSize:  grpcMsgSize,\n\t\tGRPCAddress:     \"localhost:0\",\n\t\tHTTP: &httpParams{\n\t\t\tAddress: \"localhost:0\",\n\t\t},\n\t\tOutboundsBuilder: ob,\n\t})\n\n\tif f == nil {\n\t\tt.Fatal(\"NewFactory returned nil\")\n\t}\n\n\tassert.NotNil(t, f.GetDispatcher(), \"GetDispatcher returned nil\")\n\tassert.NotNil(t, f.GetTChannel(), \"GetTChannel returned nil\")\n\tassert.Equal(t, grpcMsgSize, f.GetMaxMessageSize(), \"GetMaxMessageSize returned wrong value\")\n}\n\nfunc TestStartStop(t *testing.T) {\n\tmembersBySvc := map[string][]membership.HostInfo{\n\t\tservice.Matching: {\n\t\t\tmembership.NewHostInfo(\"localhost:9191\"),\n\t\t\tmembership.NewHostInfo(\"localhost:9192\"),\n\t\t},\n\t\tservice.History: {\n\t\t\tmembership.NewHostInfo(\"localhost:8585\"),\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tdesc             string\n\t\twantMembersBySvc map[string][]membership.HostInfo\n\t\tmockFn           func(*membership.MockResolver)\n\t\twantStartErr     bool\n\t}{\n\t\t{\n\t\t\tdesc:             \"success\",\n\t\t\twantMembersBySvc: membersBySvc,\n\t\t\tmockFn: func(peerLister *membership.MockResolver) {\n\t\t\t\tfor _, svc := range servicesToTalkP2P {\n\t\t\t\t\tpeerLister.EXPECT().Subscribe(svc, factoryComponentName, gomock.Any()).\n\t\t\t\t\t\tDoAndReturn(func(service, name string, notifyChannel chan<- *membership.ChangedEvent) error {\n\t\t\t\t\t\t\t// Notify the channel once to validate listening logic is working\n\t\t\t\t\t\t\tnotifyChannel <- &membership.ChangedEvent{}\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t}).Times(1)\n\n\t\t\t\t\tpeerLister.EXPECT().Members(svc).Return(membersBySvc[svc], nil).Times(1)\n\t\t\t\t\tpeerLister.EXPECT().Unsubscribe(svc, factoryComponentName).Return(nil).Times(1)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:         \"subscription to membership updates fail\",\n\t\t\twantStartErr: true,\n\t\t\tmockFn: func(peerLister *membership.MockResolver) {\n\t\t\t\tfor i, svc := range servicesToTalkP2P {\n\t\t\t\t\tif i == 0 {\n\t\t\t\t\t\t// subscribe will only be called for the first service and after failing, it should not be called for the rest\n\t\t\t\t\t\tpeerLister.EXPECT().Subscribe(svc, factoryComponentName, gomock.Any()).Return(errors.New(\"failed\")).Times(1)\n\t\t\t\t\t}\n\n\t\t\t\t\t// subscribe will be called for all services during stop\n\t\t\t\t\tpeerLister.EXPECT().Unsubscribe(svc, factoryComponentName).Return(nil).Times(1)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:             \"unsubscirption from membership updates fail\",\n\t\t\twantMembersBySvc: membersBySvc,\n\t\t\tmockFn: func(peerLister *membership.MockResolver) {\n\t\t\t\tfor _, svc := range servicesToTalkP2P {\n\t\t\t\t\tpeerLister.EXPECT().Subscribe(svc, factoryComponentName, gomock.Any()).\n\t\t\t\t\t\tDoAndReturn(func(service, name string, notifyChannel chan<- *membership.ChangedEvent) error {\n\t\t\t\t\t\t\t// Notify the channel once to validate listening logic is working\n\t\t\t\t\t\t\tnotifyChannel <- &membership.ChangedEvent{}\n\t\t\t\t\t\t\treturn nil\n\t\t\t\t\t\t}).Times(1)\n\t\t\t\t\tpeerLister.EXPECT().Members(svc).Return(membersBySvc[svc], nil).Times(1)\n\t\t\t\t\tpeerLister.EXPECT().Unsubscribe(svc, factoryComponentName).Return(errors.New(\"failed\")).Times(1)\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tserviceName := \"service\"\n\t\t\tob := NewMockOutboundsBuilder(ctrl)\n\t\t\tvar mu sync.Mutex\n\t\t\tgotMembers := make(map[string][]membership.HostInfo)\n\t\t\toutbounds := &Outbounds{\n\t\t\t\tonUpdatePeers: func(svc string, members []membership.HostInfo) {\n\t\t\t\t\tmu.Lock()\n\t\t\t\t\tdefer mu.Unlock()\n\t\t\t\t\tgotMembers[svc] = members\n\t\t\t\t},\n\t\t\t}\n\t\t\tob.EXPECT().Build(gomock.Any(), gomock.Any()).Return(outbounds, nil).Times(1)\n\t\t\tgrpcMsgSize := 4 * 1024 * 1024\n\t\t\tf := NewFactory(logger, Params{\n\t\t\t\tServiceName:     serviceName,\n\t\t\t\tTChannelAddress: \"localhost:0\",\n\t\t\t\tGRPCMaxMsgSize:  grpcMsgSize,\n\t\t\t\tGRPCAddress:     \"localhost:0\",\n\t\t\t\tHTTP: &httpParams{\n\t\t\t\t\tAddress: \"localhost:0\",\n\t\t\t\t},\n\t\t\t\tOutboundsBuilder: ob,\n\t\t\t})\n\n\t\t\tpeerLister := membership.NewMockResolver(ctrl)\n\t\t\ttc.mockFn(peerLister)\n\n\t\t\tif err := f.Start(peerLister); err != nil {\n\t\t\t\tif !tc.wantStartErr {\n\t\t\t\t\tt.Fatalf(\"Factory.Start() returned error: %v\", err)\n\t\t\t\t}\n\n\t\t\t\t// start failed expectedly. do not proceed with rest of the validations\n\t\t\t\tf.Stop()\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Wait for membership changes to be processed\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tmu.Lock()\n\t\t\tassert.Equal(t, tc.wantMembersBySvc, gotMembers, \"UpdatePeers not called with expected members\")\n\t\t\tmu.Unlock()\n\n\t\t\tif err := f.Stop(); err != nil {\n\t\t\t\tt.Fatalf(\"Factory.Stop() returned error: %v\", err)\n\t\t\t}\n\n\t\t\tgoleak.VerifyNone(t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/rpc/localip.go",
    "content": "// Copyright (c) 2015 Uber Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\n// ** This code is copied from tchannel, we would like to not take dependency on tchannel code **\n\nimport (\n\t\"errors\"\n\t\"net\"\n)\n\n// scoreAddr scores how likely the given addr is to be a remote address and returns the\n// IP to use when listening. Any address which receives a negative score should not be used.\n// Scores are calculated as:\n// -1 for any unknown IP addreseses.\n// +300 for IPv4 addresses\n// +100 for non-local addresses, extra +100 for \"up\" interaces.\nfunc scoreAddr(iface net.Interface, addr net.Addr) (int, net.IP) {\n\tvar ip net.IP\n\tif netAddr, ok := addr.(*net.IPNet); ok {\n\t\tip = netAddr.IP\n\t} else if netIP, ok := addr.(*net.IPAddr); ok {\n\t\tip = netIP.IP\n\t} else {\n\t\treturn -1, nil\n\t}\n\n\tvar score int\n\tif ip.To4() != nil {\n\t\tscore += 300\n\t}\n\tif iface.Flags&net.FlagLoopback == 0 && !ip.IsLoopback() {\n\t\tscore += 100\n\t\tif iface.Flags&net.FlagUp != 0 {\n\t\t\tscore += 100\n\t\t}\n\t}\n\treturn score, ip\n}\n\n// ListenIP returns the IP to bind to in Listen. It tries to find an IP that can be used\n// by other machines to reach this machine.\nfunc ListenIP() (net.IP, error) {\n\tinterfaces, err := net.Interfaces()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tbestScore := -1\n\tvar bestIP net.IP\n\t// Select the highest scoring IP as the best IP.\n\tfor _, iface := range interfaces {\n\t\taddrs, err := iface.Addrs()\n\t\tif err != nil {\n\t\t\t// Skip this interface if there is an error.\n\t\t\tcontinue\n\t\t}\n\n\t\tfor _, addr := range addrs {\n\t\t\tscore, ip := scoreAddr(iface, addr)\n\t\t\tif score > bestScore {\n\t\t\t\tbestScore = score\n\t\t\t\tbestIP = ip\n\t\t\t}\n\t\t}\n\t}\n\n\tif bestScore == -1 {\n\t\treturn nil, errors.New(\"no addresses to listen on\")\n\t}\n\n\treturn bestIP, nil\n}\n\nfunc mustParseMAC(s string) net.HardwareAddr {\n\taddr, err := net.ParseMAC(s)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn addr\n}\n"
  },
  {
    "path": "common/rpc/localip_test.go",
    "content": "// Copyright (c) 2015 Uber Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\n// ** This code is copied from tchannel, we would like to not take dependency on tchannel code **\n\nimport (\n\t\"net\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestScoreAddr(t *testing.T) {\n\tipv4 := net.ParseIP(\"10.0.1.2\")\n\tipv6 := net.ParseIP(\"2001:db8:a0b:12f0::1\")\n\n\ttests := []struct {\n\t\tmsg    string\n\t\tiface  net.Interface\n\t\taddr   net.Addr\n\t\twant   int\n\t\twantIP net.IP\n\t}{\n\t\t{\n\t\t\tmsg:    \"non-local up ipv4 IPNet address\",\n\t\t\tiface:  net.Interface{Flags: net.FlagUp},\n\t\t\taddr:   &net.IPNet{IP: ipv4},\n\t\t\twant:   500,\n\t\t\twantIP: ipv4,\n\t\t},\n\t\t{\n\t\t\tmsg:    \"non-local up ipv4 IPAddr address\",\n\t\t\tiface:  net.Interface{Flags: net.FlagUp},\n\t\t\taddr:   &net.IPAddr{IP: ipv4},\n\t\t\twant:   500,\n\t\t\twantIP: ipv4,\n\t\t},\n\t\t{\n\t\t\tmsg: \"non-local up ipv4 IPAddr address, docker interface\",\n\t\t\tiface: net.Interface{\n\t\t\t\tFlags:        net.FlagUp,\n\t\t\t\tHardwareAddr: mustParseMAC(\"02:42:ac:11:56:af\"),\n\t\t\t},\n\t\t\taddr:   &net.IPNet{IP: ipv4},\n\t\t\twant:   500,\n\t\t\twantIP: ipv4,\n\t\t},\n\t\t{\n\t\t\tmsg: \"non-local up ipv4 address, local MAC address\",\n\t\t\tiface: net.Interface{\n\t\t\t\tFlags:        net.FlagUp,\n\t\t\t\tHardwareAddr: mustParseMAC(\"02:42:9c:52:fc:86\"),\n\t\t\t},\n\t\t\taddr:   &net.IPNet{IP: ipv4},\n\t\t\twant:   500,\n\t\t\twantIP: ipv4,\n\t\t},\n\t\t{\n\t\t\tmsg:    \"non-local down ipv4 address\",\n\t\t\tiface:  net.Interface{},\n\t\t\taddr:   &net.IPNet{IP: ipv4},\n\t\t\twant:   400,\n\t\t\twantIP: ipv4,\n\t\t},\n\t\t{\n\t\t\tmsg:    \"non-local down ipv6 address\",\n\t\t\tiface:  net.Interface{},\n\t\t\taddr:   &net.IPAddr{IP: ipv6},\n\t\t\twant:   100,\n\t\t\twantIP: ipv6,\n\t\t},\n\t\t{\n\t\t\tmsg:   \"unknown address type\",\n\t\t\tiface: net.Interface{},\n\t\t\taddr:  &net.UnixAddr{Name: \"/tmp/socket\"},\n\t\t\twant:  -1,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tgotScore, gotIP := scoreAddr(tt.iface, tt.addr)\n\t\tassert.Equal(t, tt.want, gotScore, tt.msg)\n\t\tassert.Equal(t, tt.wantIP, gotIP, tt.msg)\n\t}\n}\n"
  },
  {
    "path": "common/rpc/middleware.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"io\"\n\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype authOutboundMiddleware struct {\n\tauthProvider worker.AuthorizationProvider\n}\n\nfunc (m *authOutboundMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\tif m.authProvider == nil {\n\t\treturn out.Call(ctx, request)\n\t}\n\n\ttoken, err := m.authProvider.GetAuthToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trequest.Headers = request.Headers.\n\t\tWith(common.AuthorizationTokenHeaderName, string(token))\n\n\treturn out.Call(ctx, request)\n}\n\ntype contextKey string\n\nconst _responseInfoContextKey = contextKey(\"response-info\")\n\n// ContextWithResponseInfo will create a child context that has ResponseInfo set as value.\n// This value will get filled after the call is made and can be used later to retrieve some info of interest.\nfunc ContextWithResponseInfo(parent context.Context) (context.Context, *ResponseInfo) {\n\tresponseInfo := &ResponseInfo{}\n\treturn context.WithValue(parent, _responseInfoContextKey, responseInfo), responseInfo\n}\n\n// ResponseInfo structure is filled with data after the RPC call.\n// It can be obtained with rpc.ContextWithResponseInfo function.\ntype ResponseInfo struct {\n\tSize int\n}\n\ntype countingReadCloser struct {\n\treader    io.ReadCloser\n\tbytesRead *int\n}\n\nfunc (r *countingReadCloser) Read(p []byte) (n int, err error) {\n\tn, err = r.reader.Read(p)\n\t*r.bytesRead += n\n\treturn n, err\n}\n\nfunc (r *countingReadCloser) Close() (err error) {\n\treturn r.reader.Close()\n}\n\n// ResponseInfoMiddleware populates context with ResponseInfo structure which contains info about response that was received.\n// In particular, it counts the size of the response in bytes. Such information can be useful down the line, where payload are deserialized and no longer have their size.\ntype ResponseInfoMiddleware struct{}\n\nfunc (m *ResponseInfoMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\tresponse, err := out.Call(ctx, request)\n\n\tif value := ctx.Value(_responseInfoContextKey); value != nil {\n\t\tif responseInfo, ok := value.(*ResponseInfo); ok && response != nil {\n\t\t\t// We can not use response.BodySize here, because it is not set on all transports.\n\t\t\t// Instead wrap body reader with counter, that increments responseInfo.Size as it is read.\n\t\t\tresponse.Body = &countingReadCloser{reader: response.Body, bytesRead: &responseInfo.Size}\n\t\t}\n\t}\n\n\treturn response, err\n}\n\n// InboundMetricsMiddleware tags context with additional metric tags from incoming request.\ntype InboundMetricsMiddleware struct{}\n\nfunc (m *InboundMetricsMiddleware) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter, h transport.UnaryHandler) error {\n\tctx = metrics.TagContext(ctx,\n\t\tmetrics.CallerTag(req.Caller),\n\t\tmetrics.TransportTag(req.Transport),\n\t)\n\treturn h.Handle(ctx, req, resw)\n}\n\n// CallerInfoMiddleware extracts caller information from headers and adds it to the context.\ntype CallerInfoMiddleware struct{}\n\nfunc (m *CallerInfoMiddleware) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter, h transport.UnaryHandler) error {\n\tctx = types.GetContextWithCallerInfoFromHeaders(ctx, req.Headers)\n\treturn h.Handle(ctx, req, resw)\n}\n\n// CallerInfoOutboundMiddleware sets the caller type header on outbound calls.\n// If the caller type is not already set in headers (by HeaderForwardingMiddleware),\n// it sets to \"internal\" only if the call originated from internal service logic.\n// External requests without the header are left unset and will be extracted as \"unknown\" by the receiving service.\ntype CallerInfoOutboundMiddleware struct{}\n\nfunc (m *CallerInfoOutboundMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\tif _, ok := request.Headers.Get(types.CallerTypeHeaderName); !ok {\n\t\tif yarpc.CallFromContext(ctx) == nil {\n\t\t\trequest.Headers = request.Headers.With(types.CallerTypeHeaderName, string(types.CallerTypeInternal))\n\t\t}\n\t}\n\treturn out.Call(ctx, request)\n}\n\ntype overrideCallerMiddleware struct {\n\tcaller string\n}\n\nfunc (m *overrideCallerMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\trequest.Caller = m.caller\n\treturn out.Call(ctx, request)\n}\n\n// HeaderForwardingMiddleware forwards headers from current inbound RPC call that is being handled to new outbound calls being made.\n// As this does NOT differentiate between transports or purposes, it generally assumes we are not acting as a true proxy,\n// so things like content lengths and encodings should not be forwarded - they will be provided by the outbound RPC library as needed.\n//\n// Duplicated headers retain the first value only, matching how browsers and Go (afaict) generally behave.\n//\n// This uses overly-simplified rules for choosing which headers are forwarded and which are not, intended to be lightly configurable.\n// For a more in-depth logic review if it becomes needed, check:\n//   - How Go's ReverseProxy deals with headers, e.g. per-protocol and a list of exclusions: https://cs.opensource.google/go/go/+/refs/tags/go1.20.1:src/net/http/httputil/reverseproxy.go;l=332\n//   - HTTP's spec for headers, namely how duplicates and Connection work: https://www.rfc-editor.org/rfc/rfc9110.html#name-header-fields\n//   - Many browsers prefer first-value-wins for unexpected duplicates: https://bugzilla.mozilla.org/show_bug.cgi?id=376756\n//   - But there are MANY map-like implementations that choose last-value wins, and this mismatch is a source of frequent security problems.\n//   - YARPC's `With` only retains the last call's value: https://github.com/yarpc/yarpc-go/blob/8ccd79a2ca696150213faac1d35011c5be52e5fb/api/transport/header.go#L69-L77\n//   - Go's MIMEHeader's Get (used by YARPC) only returns the first value, and does not join duplicates: https://pkg.go.dev/net/textproto#MIMEHeader.Get\n//\n// There is likely no correct choice, as it depends on the recipients' behavior.\n// If we need to support more complex logic, it's likely worth jumping to a fully-controllable thing.\n// Middle-grounds will probably just need to be changed again later.\ntype HeaderForwardingMiddleware struct {\n\t// Rules are applied in order to add or remove headers by regex.\n\t//\n\t// There are no default rules, so by default no headers are copied.\n\t// To include headers by default, Add with a permissive regex and then remove specific ones.\n\tRules []config.HeaderRule\n}\n\nfunc (m *HeaderForwardingMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\tif inboundCall := yarpc.CallFromContext(ctx); inboundCall != nil {\n\t\toutboundHeaders := request.Headers\n\t\tpending := make(map[string][]string, len(inboundCall.HeaderNames()))\n\t\tnames := inboundCall.HeaderNames()\n\t\tfor _, rule := range m.Rules {\n\t\t\tfor _, key := range names {\n\t\t\t\tif !rule.Match.MatchString(key) {\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tif _, ok := outboundHeaders.Get(key); ok {\n\t\t\t\t\tcontinue // do not overwrite existing headers\n\t\t\t\t}\n\t\t\t\tif rule.Add {\n\t\t\t\t\tpending[key] = append(pending[key], inboundCall.Header(key))\n\t\t\t\t} else {\n\t\t\t\t\tdelete(pending, key)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tfor k, vs := range pending {\n\t\t\t// yarpc's Headers.With keeps the LAST value, but we (and browsers) prefer the FIRST,\n\t\t\t// and we do not canonicalize duplicates.\n\t\t\trequest.Headers = request.Headers.With(k, vs[0])\n\t\t}\n\t}\n\n\treturn out.Call(ctx, request)\n}\n\n// ForwardPartitionConfigMiddleware forwards the partition config to remote cluster\n// The middleware should always be applied after any other middleware that inject partition config into the context\n// so that it can overwrites the partition config into the context\n// The purpose of this middleware is to make sure the partition config doesn't change when a request is forwarded from\n// passive cluster to the active cluster\ntype ForwardPartitionConfigMiddleware struct{}\n\nfunc (m *ForwardPartitionConfigMiddleware) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter, h transport.UnaryHandler) error {\n\tif _, ok := req.Headers.Get(common.AutoforwardingClusterHeaderName); ok {\n\t\tvar partitionConfig map[string]string\n\t\tif blob, ok := req.Headers.Get(common.PartitionConfigHeaderName); ok && len(blob) > 0 {\n\t\t\tif err := json.Unmarshal([]byte(blob), &partitionConfig); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tctx = isolationgroup.ContextWithConfig(ctx, partitionConfig)\n\t\tisolationGroup, _ := req.Headers.Get(common.IsolationGroupHeaderName)\n\t\tctx = isolationgroup.ContextWithIsolationGroup(ctx, isolationGroup)\n\t}\n\treturn h.Handle(ctx, req, resw)\n}\n\nfunc (m *ForwardPartitionConfigMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\tif _, ok := request.Headers.Get(common.AutoforwardingClusterHeaderName); ok {\n\t\tpartitionConfig := isolationgroup.ConfigFromContext(ctx)\n\t\tif len(partitionConfig) > 0 {\n\t\t\tblob, err := json.Marshal(partitionConfig)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\trequest.Headers = request.Headers.With(common.PartitionConfigHeaderName, string(blob))\n\t\t} else {\n\t\t\trequest.Headers.Del(common.PartitionConfigHeaderName)\n\t\t}\n\t\tisolationGroup := isolationgroup.IsolationGroupFromContext(ctx)\n\t\tif isolationGroup != \"\" {\n\t\t\trequest.Headers = request.Headers.With(common.IsolationGroupHeaderName, isolationGroup)\n\t\t} else {\n\t\t\trequest.Headers.Del(common.IsolationGroupHeaderName)\n\t\t}\n\t}\n\treturn out.Call(ctx, request)\n}\n\n// ClientPartitionConfigMiddleware stores the partition config and isolation group of the request into the context\n// It reads a header from client request and uses it as the isolation group\ntype ClientPartitionConfigMiddleware struct{}\n\nfunc (m *ClientPartitionConfigMiddleware) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter, h transport.UnaryHandler) error {\n\tzone, _ := req.Headers.Get(common.ClientIsolationGroupHeaderName)\n\tif zone != \"\" {\n\t\tctx = isolationgroup.ContextWithConfig(ctx, map[string]string{\n\t\t\tisolationgroup.GroupKey: zone,\n\t\t})\n\t\tctx = isolationgroup.ContextWithIsolationGroup(ctx, zone)\n\t}\n\treturn h.Handle(ctx, req, resw)\n}\n"
  },
  {
    "path": "common/rpc/middleware_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"regexp\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/yarpctest\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestAuthOubboundMiddleware(t *testing.T) {\n\tm := authOutboundMiddleware{}\n\t_, err := m.Call(context.Background(), &transport.Request{}, &fakeOutbound{verify: func(request *transport.Request) {\n\t\tassert.Empty(t, request.Headers)\n\t}})\n\tassert.NoError(t, err)\n\n\tm = authOutboundMiddleware{fakeAuthProvider{err: assert.AnError}}\n\t_, err = m.Call(context.Background(), &transport.Request{}, &fakeOutbound{})\n\tassert.Error(t, err)\n\n\tm = authOutboundMiddleware{fakeAuthProvider{token: []byte(\"token\")}}\n\t_, err = m.Call(context.Background(), &transport.Request{}, &fakeOutbound{verify: func(request *transport.Request) {\n\t\tassert.Equal(t, \"token\", request.Headers.Items()[common.AuthorizationTokenHeaderName])\n\t}})\n\tassert.NoError(t, err)\n}\n\nfunc TestResponseInfoMiddleware(t *testing.T) {\n\tm := ResponseInfoMiddleware{}\n\tctx, responseInfo := ContextWithResponseInfo(context.Background())\n\tbody := ioutil.NopCloser(bytes.NewReader([]byte{1, 2, 3, 4, 5}))\n\tresponse, err := m.Call(ctx, &transport.Request{}, &fakeOutbound{response: &transport.Response{Body: body}})\n\tassert.NoError(t, err)\n\tioutil.ReadAll(response.Body)\n\tassert.Equal(t, 5, responseInfo.Size)\n}\n\nfunc TestResponseInfoMiddleware_Error(t *testing.T) {\n\tm := ResponseInfoMiddleware{}\n\tctx, responseInfo := ContextWithResponseInfo(context.Background())\n\t_, err := m.Call(ctx, &transport.Request{}, &fakeOutbound{err: fmt.Errorf(\"test\")})\n\tassert.Error(t, err)\n\tassert.Equal(t, 0, responseInfo.Size)\n}\n\nfunc TestInboundMetricsMiddleware(t *testing.T) {\n\tm := InboundMetricsMiddleware{}\n\th := &fakeHandler{}\n\terr := m.Handle(context.Background(), &transport.Request{Transport: \"grpc\", Caller: \"x-caller\"}, nil, h)\n\tassert.NoError(t, err)\n\tassert.ElementsMatch(t, metrics.GetContextTags(h.ctx), []metrics.Tag{\n\t\tmetrics.TransportTag(\"grpc\"),\n\t\tmetrics.CallerTag(\"x-caller\"),\n\t})\n}\n\nfunc TestOverrideCallerMiddleware(t *testing.T) {\n\tm := overrideCallerMiddleware{\"x-caller\"}\n\t_, err := m.Call(context.Background(), &transport.Request{Caller: \"service\"}, &fakeOutbound{verify: func(r *transport.Request) {\n\t\tassert.Equal(t, \"x-caller\", r.Caller)\n\t}})\n\tassert.NoError(t, err)\n}\n\nfunc TestHeaderForwardingMiddleware(t *testing.T) {\n\tinboundHeaders := map[string]string{\n\t\t\"key-a\": \"inbound-value-a\",\n\t\t\"key-b\": \"inbound-value-b\",\n\t\t\"key-x\": \"inbound-value-x\",\n\t}\n\toutboundHeaders := map[string]string{\n\t\t\"key-b\": \"outbound-value-b\",\n\t\t\"key-c\": \"outbound-value-c\",\n\t}\n\tcombinedHeaders := map[string]string{\n\t\t\"key-a\": \"inbound-value-a\",\n\t\t\"key-b\": \"outbound-value-b\",\n\t\t\"key-c\": \"outbound-value-c\",\n\t\t\"key-x\": \"inbound-value-x\",\n\t}\n\tcombinedHeadersWithoutX := map[string]string{\n\t\t\"key-a\": \"inbound-value-a\",\n\t\t\"key-b\": \"outbound-value-b\",\n\t\t\"key-c\": \"outbound-value-c\",\n\t}\n\n\tctxWithInbound := yarpctest.ContextWithCall(context.Background(), &yarpctest.Call{Headers: inboundHeaders})\n\tmakeRequest := func() *transport.Request {\n\t\t// request and headers are mutated by Call, so we must not share data\n\t\treturn &transport.Request{Headers: transport.HeadersFromMap(outboundHeaders)}\n\t}\n\n\tt.Run(\"default rules\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tm := HeaderForwardingMiddleware{\n\t\t\tRules: []config.HeaderRule{\n\t\t\t\t// default\n\t\t\t\t{\n\t\t\t\t\tAdd:   true,\n\t\t\t\t\tMatch: regexp.MustCompile(\"\"),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tt.Run(\"no inbound makes no changes\", func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\t// No ongoing inbound call -> keep existing outbound headers\n\t\t\t_, err := m.Call(context.Background(), makeRequest(), &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\t\tassert.Equal(t, outboundHeaders, r.Headers.Items())\n\t\t\t}})\n\t\t\tassert.NoError(t, err)\n\t\t})\n\n\t\tt.Run(\"inbound headers merge missing\", func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\t// With ongoing inbound call -> forward inbound headers not present in the request\n\t\t\t_, err := m.Call(ctxWithInbound, makeRequest(), &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\t\tassert.Equal(t, combinedHeaders, r.Headers.Items())\n\t\t\t}})\n\t\t\tassert.NoError(t, err)\n\t\t})\n\t})\n\tt.Run(\"can exclude inbound headers\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tm := HeaderForwardingMiddleware{\n\t\t\tRules: []config.HeaderRule{\n\t\t\t\t// add by default, earlier tests ensure this works\n\t\t\t\t{\n\t\t\t\t\tAdd:   true,\n\t\t\t\t\tMatch: regexp.MustCompile(\"\"),\n\t\t\t\t},\n\t\t\t\t// remove X\n\t\t\t\t{\n\t\t\t\t\tAdd:   false,\n\t\t\t\t\tMatch: regexp.MustCompile(\"(?i)Key-x\"),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\t_, err := m.Call(ctxWithInbound, makeRequest(), &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\tassert.Equal(t, combinedHeadersWithoutX, r.Headers.Items())\n\t\t}})\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc TestForwardPartitionConfigMiddleware(t *testing.T) {\n\tt.Run(\"inbound middleware\", func(t *testing.T) {\n\t\tpartitionConfig := map[string]string{\"isolation-group\": \"xyz\"}\n\t\tblob, err := json.Marshal(partitionConfig)\n\t\trequire.NoError(t, err)\n\t\ttestCases := []struct {\n\t\t\tmessage                 string\n\t\t\theaders                 transport.Headers\n\t\t\tctx                     context.Context\n\t\t\texpectedPartitionConfig map[string]string\n\t\t\texpectedIsolationGroup  string\n\t\t}{\n\t\t\t{\n\t\t\t\tmessage: \"it injects partition config into context\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, string(blob)).\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"abc\").\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                     context.Background(),\n\t\t\t\texpectedPartitionConfig: partitionConfig,\n\t\t\t\texpectedIsolationGroup:  \"abc\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it overwrites the existing partition config in the context\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, string(blob)).\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"abc\").\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                     isolationgroup.ContextWithIsolationGroup(isolationgroup.ContextWithConfig(context.Background(), map[string]string{\"z\": \"x\"}), \"fff\"),\n\t\t\t\texpectedPartitionConfig: partitionConfig,\n\t\t\t\texpectedIsolationGroup:  \"abc\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it overwrites the existing partition config in the context with nil config\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                     isolationgroup.ContextWithIsolationGroup(isolationgroup.ContextWithConfig(context.Background(), map[string]string{\"z\": \"x\"}), \"fff\"),\n\t\t\t\texpectedPartitionConfig: nil,\n\t\t\t\texpectedIsolationGroup:  \"\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it injects partition config into context only if the request is an auto-fowarding request\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, string(blob)).\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"abc\"),\n\t\t\t\tctx:                     context.Background(),\n\t\t\t\texpectedPartitionConfig: nil,\n\t\t\t\texpectedIsolationGroup:  \"\",\n\t\t\t},\n\t\t}\n\n\t\tfor _, tt := range testCases {\n\t\t\tt.Run(tt.message, func(t *testing.T) {\n\t\t\t\tm := &ForwardPartitionConfigMiddleware{}\n\t\t\t\th := &fakeHandler{}\n\t\t\t\terr := m.Handle(tt.ctx, &transport.Request{Headers: tt.headers}, nil, h)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedPartitionConfig, isolationgroup.ConfigFromContext(h.ctx))\n\t\t\t\tassert.Equal(t, tt.expectedIsolationGroup, isolationgroup.IsolationGroupFromContext(h.ctx))\n\t\t\t})\n\t\t}\n\t})\n\n\tt.Run(\"outbound middleware\", func(t *testing.T) {\n\t\tpartitionConfig := map[string]string{\"isolation-group\": \"xyz\"}\n\t\tblob, err := json.Marshal(partitionConfig)\n\t\trequire.NoError(t, err)\n\t\ttestCases := []struct {\n\t\t\tmessage                           string\n\t\t\theaders                           transport.Headers\n\t\t\tctx                               context.Context\n\t\t\texpectedSerializedPartitionConfig string\n\t\t\texpectedIsolationGroup            string\n\t\t}{\n\t\t\t{\n\t\t\t\tmessage: \"it retrieves partition config from the context and sets it in the request header\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                               isolationgroup.ContextWithIsolationGroup(isolationgroup.ContextWithConfig(context.Background(), partitionConfig), \"abc\"),\n\t\t\t\texpectedSerializedPartitionConfig: string(blob),\n\t\t\t\texpectedIsolationGroup:            \"abc\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it retrieves partition config from the context and overwrites the existing request header\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"lll\").\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, \"asdfasf\").\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                               isolationgroup.ContextWithIsolationGroup(isolationgroup.ContextWithConfig(context.Background(), partitionConfig), \"abc\"),\n\t\t\t\texpectedSerializedPartitionConfig: string(blob),\n\t\t\t\texpectedIsolationGroup:            \"abc\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it deletes the existing request header if we cannot retrieve partition config from the context\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"lll\").\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, \"asdfasf\").\n\t\t\t\t\tWith(common.AutoforwardingClusterHeaderName, \"cluster0\"),\n\t\t\t\tctx:                               context.Background(),\n\t\t\t\texpectedSerializedPartitionConfig: \"\",\n\t\t\t\texpectedIsolationGroup:            \"\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tmessage: \"it retrieves partition config from the context and sets it in the request header only if the request is an auto-forwarding request\",\n\t\t\t\theaders: transport.NewHeaders().\n\t\t\t\t\tWith(common.IsolationGroupHeaderName, \"lll\").\n\t\t\t\t\tWith(common.PartitionConfigHeaderName, \"asdfasf\"),\n\t\t\t\tctx:                               isolationgroup.ContextWithIsolationGroup(isolationgroup.ContextWithConfig(context.Background(), partitionConfig), \"abc\"),\n\t\t\t\texpectedSerializedPartitionConfig: \"asdfasf\",\n\t\t\t\texpectedIsolationGroup:            \"lll\",\n\t\t\t},\n\t\t}\n\n\t\tfor _, tt := range testCases {\n\t\t\tt.Run(tt.message, func(t *testing.T) {\n\t\t\t\tm := &ForwardPartitionConfigMiddleware{}\n\t\t\t\to := &fakeOutbound{\n\t\t\t\t\tverify: func(r *transport.Request) {\n\t\t\t\t\t\tassert.Equal(t, tt.expectedIsolationGroup, r.Headers.Items()[common.IsolationGroupHeaderName])\n\t\t\t\t\t\tassert.Equal(t, tt.expectedSerializedPartitionConfig, r.Headers.Items()[common.PartitionConfigHeaderName])\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\t_, err := m.Call(tt.ctx, &transport.Request{Headers: tt.headers}, o)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t})\n\t\t}\n\t})\n}\n\nfunc TestClientPartitionConfigMiddleware(t *testing.T) {\n\tt.Run(\"it sets the partition config\", func(t *testing.T) {\n\t\tm := &ClientPartitionConfigMiddleware{}\n\t\th := &fakeHandler{}\n\t\theaders := transport.NewHeaders().\n\t\t\tWith(common.ClientIsolationGroupHeaderName, \"dca1\")\n\t\terr := m.Handle(context.Background(), &transport.Request{Headers: headers}, nil, h)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, map[string]string{isolationgroup.GroupKey: \"dca1\"}, isolationgroup.ConfigFromContext(h.ctx))\n\t\tassert.Equal(t, \"dca1\", isolationgroup.IsolationGroupFromContext(h.ctx))\n\t})\n\n\tt.Run(\"noop when header is empty\", func(t *testing.T) {\n\t\tm := &ClientPartitionConfigMiddleware{}\n\t\th := &fakeHandler{}\n\t\theaders := transport.NewHeaders()\n\t\tctx := context.Background()\n\t\terr := m.Handle(ctx, &transport.Request{Headers: headers}, nil, h)\n\t\tassert.NoError(t, err)\n\t\tassert.Nil(t, isolationgroup.ConfigFromContext(h.ctx))\n\t\tassert.Equal(t, \"\", isolationgroup.IsolationGroupFromContext(h.ctx))\n\t\tassert.Equal(t, ctx, h.ctx)\n\t})\n}\n\nfunc TestCallerInfoMiddleware(t *testing.T) {\n\tt.Run(\"extracts caller type from header\", func(t *testing.T) {\n\t\tm := &CallerInfoMiddleware{}\n\t\th := &fakeHandler{}\n\t\theaders := transport.NewHeaders().With(types.CallerTypeHeaderName, \"cli\")\n\t\terr := m.Handle(context.Background(), &transport.Request{Headers: headers}, nil, h)\n\t\tassert.NoError(t, err)\n\n\t\tcallerInfo := types.GetCallerInfoFromContext(h.ctx)\n\t\tassert.Equal(t, types.CallerTypeCLI, callerInfo.GetCallerType())\n\t})\n\n\tt.Run(\"sets unknown caller type when header is missing\", func(t *testing.T) {\n\t\tm := &CallerInfoMiddleware{}\n\t\th := &fakeHandler{}\n\t\theaders := transport.NewHeaders()\n\t\terr := m.Handle(context.Background(), &transport.Request{Headers: headers}, nil, h)\n\t\tassert.NoError(t, err)\n\n\t\tcallerInfo := types.GetCallerInfoFromContext(h.ctx)\n\t\tassert.Equal(t, types.CallerTypeUnknown, callerInfo.GetCallerType())\n\t})\n\n\tt.Run(\"extracts different caller types\", func(t *testing.T) {\n\t\ttests := []struct {\n\t\t\tname           string\n\t\t\theaderValue    string\n\t\t\texpectedCaller types.CallerType\n\t\t}{\n\t\t\t{\"CLI\", \"cli\", types.CallerTypeCLI},\n\t\t\t{\"UI\", \"ui\", types.CallerTypeUI},\n\t\t\t{\"SDK\", \"sdk\", types.CallerTypeSDK},\n\t\t\t{\"Internal\", \"internal\", types.CallerTypeInternal},\n\t\t\t{\"Empty\", \"\", types.CallerTypeUnknown},\n\t\t}\n\n\t\tfor _, tt := range tests {\n\t\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t\tm := &CallerInfoMiddleware{}\n\t\t\t\th := &fakeHandler{}\n\t\t\t\theaders := transport.NewHeaders().With(types.CallerTypeHeaderName, tt.headerValue)\n\t\t\t\terr := m.Handle(context.Background(), &transport.Request{Headers: headers}, nil, h)\n\t\t\t\tassert.NoError(t, err)\n\n\t\t\t\tcallerInfo := types.GetCallerInfoFromContext(h.ctx)\n\t\t\t\tassert.Equal(t, tt.expectedCaller, callerInfo.GetCallerType())\n\t\t\t})\n\t\t}\n\t})\n}\n\nfunc TestCallerInfoOutboundMiddleware(t *testing.T) {\n\tt.Run(\"sets internal caller type when no inbound call and header is missing\", func(t *testing.T) {\n\t\tm := &CallerInfoOutboundMiddleware{}\n\t\t_, err := m.Call(context.Background(), &transport.Request{}, &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\tcallerType, ok := r.Headers.Get(types.CallerTypeHeaderName)\n\t\t\tassert.True(t, ok)\n\t\t\tassert.Equal(t, string(types.CallerTypeInternal), callerType)\n\t\t}})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"does not set header when inbound call exists without header\", func(t *testing.T) {\n\t\tm := &CallerInfoOutboundMiddleware{}\n\t\tctxWithInbound := yarpctest.ContextWithCall(context.Background(), &yarpctest.Call{})\n\t\t_, err := m.Call(ctxWithInbound, &transport.Request{}, &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\t_, ok := r.Headers.Get(types.CallerTypeHeaderName)\n\t\t\tassert.False(t, ok, \"header should not be set for external requests without header\")\n\t\t}})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"does not override existing caller type header\", func(t *testing.T) {\n\t\tm := &CallerInfoOutboundMiddleware{}\n\t\theaders := transport.NewHeaders().With(types.CallerTypeHeaderName, \"cli\")\n\t\t_, err := m.Call(context.Background(), &transport.Request{Headers: headers}, &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\tcallerType, ok := r.Headers.Get(types.CallerTypeHeaderName)\n\t\t\tassert.True(t, ok)\n\t\t\tassert.Equal(t, \"cli\", callerType)\n\t\t}})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"does not override header forwarded from inbound call\", func(t *testing.T) {\n\t\tm := &CallerInfoOutboundMiddleware{}\n\t\tctxWithInbound := yarpctest.ContextWithCall(context.Background(), &yarpctest.Call{\n\t\t\tHeaders: map[string]string{types.CallerTypeHeaderName: \"ui\"},\n\t\t})\n\t\theaders := transport.NewHeaders().With(types.CallerTypeHeaderName, \"ui\")\n\t\t_, err := m.Call(ctxWithInbound, &transport.Request{Headers: headers}, &fakeOutbound{verify: func(r *transport.Request) {\n\t\t\tcallerType, ok := r.Headers.Get(types.CallerTypeHeaderName)\n\t\t\tassert.True(t, ok)\n\t\t\tassert.Equal(t, \"ui\", callerType, \"should preserve forwarded header\")\n\t\t}})\n\t\tassert.NoError(t, err)\n\t})\n}\n\ntype fakeHandler struct {\n\tctx context.Context\n}\n\nfunc (h *fakeHandler) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter) error {\n\th.ctx = ctx\n\treturn nil\n}\n\ntype fakeOutbound struct {\n\tverify   func(*transport.Request)\n\tresponse *transport.Response\n\terr      error\n}\n\nfunc (o fakeOutbound) Call(ctx context.Context, request *transport.Request) (*transport.Response, error) {\n\tif o.verify != nil {\n\t\to.verify(request)\n\t}\n\treturn o.response, o.err\n}\nfunc (o fakeOutbound) Start() error                      { return nil }\nfunc (o fakeOutbound) Stop() error                       { return nil }\nfunc (o fakeOutbound) IsRunning() bool                   { return true }\nfunc (o fakeOutbound) Transports() []transport.Transport { return nil }\n\ntype fakeAuthProvider struct {\n\ttoken []byte\n\terr   error\n}\n\nfunc (p fakeAuthProvider) GetAuthToken() ([]byte, error) {\n\treturn p.token, p.err\n}\n"
  },
  {
    "path": "common/rpc/outbounds.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination outbounds_mock.go -self_package github.com/uber/cadence/common/rpc\n\npackage rpc\n\nimport (\n\t\"crypto/tls\"\n\t\"fmt\"\n\n\t\"go.uber.org/multierr\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/middleware\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\t// OutboundPublicClient is the name of configured public client outbound\n\tOutboundPublicClient = \"public-client\"\n\n\tcrossDCCaller = \"cadence-xdc-client\"\n)\n\n// OutboundsBuilder allows defining outbounds for the dispatcher\ntype OutboundsBuilder interface {\n\t// Build creates yarpc outbounds given transport instances for either gRPC and TChannel based on the configuration\n\tBuild(*grpc.Transport, *tchannel.Transport) (*Outbounds, error)\n}\n\ntype Outbounds struct {\n\tyarpc.Outbounds\n\tonUpdatePeers func(serviceName string, members []membership.HostInfo)\n}\n\nfunc (o *Outbounds) UpdatePeers(serviceName string, peers []membership.HostInfo) {\n\tif o.onUpdatePeers != nil {\n\t\to.onUpdatePeers(serviceName, peers)\n\t}\n}\n\ntype multiOutboundsBuilder struct {\n\tbuilders []OutboundsBuilder\n}\n\n// CombineOutbounds takes multiple outbound builders and combines them\nfunc CombineOutbounds(builders ...OutboundsBuilder) OutboundsBuilder {\n\treturn multiOutboundsBuilder{builders}\n}\n\nfunc (b multiOutboundsBuilder) Build(grpc *grpc.Transport, tchannel *tchannel.Transport) (*Outbounds, error) {\n\toutbounds := yarpc.Outbounds{}\n\tvar errs error\n\tvar callbacks []func(string, []membership.HostInfo)\n\tfor _, builder := range b.builders {\n\t\tbuilderOutbounds, err := builder.Build(grpc, tchannel)\n\t\tif err != nil {\n\t\t\terrs = multierr.Append(errs, err)\n\t\t\tcontinue\n\t\t}\n\n\t\tif builderOutbounds.onUpdatePeers != nil {\n\t\t\tcallbacks = append(callbacks, builderOutbounds.onUpdatePeers)\n\t\t}\n\n\t\tfor name, outbound := range builderOutbounds.Outbounds {\n\t\t\tif _, exists := outbounds[name]; exists {\n\t\t\t\terrs = multierr.Append(errs, fmt.Errorf(\"outbound %q already configured\", name))\n\t\t\t\tbreak\n\t\t\t}\n\t\t\toutbounds[name] = outbound\n\t\t}\n\t}\n\n\treturn &Outbounds{\n\t\tOutbounds: outbounds,\n\t\tonUpdatePeers: func(serviceName string, members []membership.HostInfo) {\n\t\t\tfor _, callback := range callbacks {\n\t\t\t\tcallback(serviceName, members)\n\t\t\t}\n\t\t},\n\t}, errs\n}\n\ntype publicClientOutbound struct {\n\taddress        string\n\tisGRPC         bool\n\tauthMiddleware middleware.UnaryOutbound\n}\n\nfunc newPublicClientOutbound(config *config.Config) (publicClientOutbound, error) {\n\tif len(config.PublicClient.HostPort) == 0 {\n\t\treturn publicClientOutbound{}, fmt.Errorf(\"need to provide an endpoint config for PublicClient\")\n\t}\n\n\tvar authMiddleware middleware.UnaryOutbound\n\tif config.Authorization.OAuthAuthorizer.Enable {\n\t\tclusterName := config.ClusterGroupMetadata.CurrentClusterName\n\t\tclusterInfo := config.ClusterGroupMetadata.ClusterGroup[clusterName]\n\t\tauthProvider, err := authorization.GetAuthProviderClient(clusterInfo.AuthorizationProvider.PrivateKey)\n\t\tif err != nil {\n\t\t\treturn publicClientOutbound{}, fmt.Errorf(\"create AuthProvider: %v\", err)\n\t\t}\n\t\tauthMiddleware = &authOutboundMiddleware{authProvider}\n\t}\n\n\tisGrpc := config.PublicClient.Transport == grpc.TransportName\n\n\treturn publicClientOutbound{config.PublicClient.HostPort, isGrpc, authMiddleware}, nil\n}\n\nfunc (b publicClientOutbound) Build(grpc *grpc.Transport, tchannel *tchannel.Transport) (*Outbounds, error) {\n\tvar outbound transport.UnaryOutbound\n\tif b.isGRPC {\n\t\toutbound = grpc.NewSingleOutbound(b.address)\n\t} else {\n\t\toutbound = tchannel.NewSingleOutbound(b.address)\n\t}\n\treturn &Outbounds{\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\tOutboundPublicClient: {\n\t\t\t\tServiceName: service.Frontend,\n\t\t\t\tUnary:       middleware.ApplyUnaryOutbound(outbound, b.authMiddleware),\n\t\t\t},\n\t\t},\n\t}, nil\n}\n\ntype crossDCOutbounds struct {\n\tclusterGroup map[string]config.ClusterInformation\n\tpcf          PeerChooserFactory\n}\n\nfunc NewCrossDCOutbounds(clusterGroup map[string]config.ClusterInformation, pcf PeerChooserFactory) OutboundsBuilder {\n\treturn crossDCOutbounds{clusterGroup, pcf}\n}\n\nfunc (b crossDCOutbounds) Build(grpcTransport *grpc.Transport, tchannelTransport *tchannel.Transport) (*Outbounds, error) {\n\toutbounds := yarpc.Outbounds{}\n\tfor clusterName, clusterInfo := range b.clusterGroup {\n\t\tif !clusterInfo.Enabled {\n\t\t\tcontinue\n\t\t}\n\n\t\tvar outbound transport.UnaryOutbound\n\t\tswitch clusterInfo.RPCTransport {\n\t\tcase tchannel.TransportName:\n\t\t\tpeerChooser, err := b.pcf.CreatePeerChooser(tchannelTransport, PeerChooserOptions{Address: clusterInfo.RPCAddress})\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\toutbound = tchannelTransport.NewOutbound(peerChooser)\n\t\tcase grpc.TransportName:\n\t\t\ttlsConfig, err := clusterInfo.TLS.ToTLSConfig()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tpeerChooser, err := b.pcf.CreatePeerChooser(createDialer(grpcTransport, tlsConfig), PeerChooserOptions{Address: clusterInfo.RPCAddress})\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\toutbound = grpcTransport.NewOutbound(peerChooser)\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"unknown cross DC transport type: %s\", clusterInfo.RPCTransport)\n\t\t}\n\n\t\tvar authMiddleware middleware.UnaryOutbound\n\t\tif clusterInfo.AuthorizationProvider.Enable {\n\t\t\tauthProvider, err := authorization.GetAuthProviderClient(clusterInfo.AuthorizationProvider.PrivateKey)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"create AuthProvider: %v\", err)\n\t\t\t}\n\t\t\tauthMiddleware = &authOutboundMiddleware{authProvider}\n\t\t}\n\n\t\toutbounds[clusterName] = transport.Outbounds{\n\t\t\tServiceName: clusterInfo.RPCName,\n\t\t\tUnary: middleware.ApplyUnaryOutbound(outbound, yarpc.UnaryOutboundMiddleware(\n\t\t\t\tauthMiddleware,\n\t\t\t\t&CallerInfoOutboundMiddleware{},\n\t\t\t\t&overrideCallerMiddleware{crossDCCaller},\n\t\t\t)),\n\t\t}\n\t}\n\treturn &Outbounds{Outbounds: outbounds}, nil\n}\n\ntype directOutbound struct {\n\tserviceName          string\n\tgrpcEnabled          bool\n\ttlsConfig            *tls.Config\n\tpcf                  PeerChooserFactory\n\tenableConnRetainMode dynamicproperties.BoolPropertyFn\n}\n\nfunc NewDirectOutboundBuilder(serviceName string, grpcEnabled bool, tlsConfig *tls.Config, pcf PeerChooserFactory, enableConnRetainMode dynamicproperties.BoolPropertyFn) OutboundsBuilder {\n\treturn directOutbound{serviceName, grpcEnabled, tlsConfig, pcf, enableConnRetainMode}\n}\n\nfunc (o directOutbound) Build(grpc *grpc.Transport, tchannel *tchannel.Transport) (*Outbounds, error) {\n\tvar outbound transport.UnaryOutbound\n\tvar streamOutbound transport.StreamOutbound\n\topts := PeerChooserOptions{\n\t\tEnableConnectionRetainingDirectChooser: o.enableConnRetainMode,\n\t\tServiceName:                            o.serviceName,\n\t}\n\n\tvar err error\n\tvar directChooser PeerChooser\n\tif o.grpcEnabled {\n\t\tdirectChooser, err = o.pcf.CreatePeerChooser(createDialer(grpc, o.tlsConfig), opts)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\toutbound = grpc.NewOutbound(directChooser)\n\t\t// Shard manager needs stream outbound, it only supports GRPC, so we don't need to create a tchannel stream outbound\n\t\tstreamOutbound = grpc.NewOutbound(directChooser)\n\t} else {\n\t\tdirectChooser, err = o.pcf.CreatePeerChooser(tchannel, opts)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\toutbound = tchannel.NewOutbound(directChooser)\n\t}\n\n\treturn &Outbounds{\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\to.serviceName: {\n\t\t\t\tServiceName: o.serviceName,\n\t\t\t\tUnary:       middleware.ApplyUnaryOutbound(outbound, yarpc.UnaryOutboundMiddleware(&CallerInfoOutboundMiddleware{}, &ResponseInfoMiddleware{})),\n\t\t\t\tStream:      streamOutbound,\n\t\t\t},\n\t\t},\n\t\tonUpdatePeers: directChooser.UpdatePeers,\n\t}, nil\n}\n\nfunc IsGRPCOutbound(config transport.ClientConfig) bool {\n\tnamer, ok := config.GetUnaryOutbound().(transport.Namer)\n\tif !ok {\n\t\t// This should not happen, unless yarpc older than v1.43.0 is used\n\t\tpanic(\"Outbound does not implement transport.Namer\")\n\t}\n\treturn namer.TransportName() == grpc.TransportName\n}\nfunc NewSingleGRPCOutboundBuilder(outboundName string, serviceName string, address string) OutboundsBuilder {\n\treturn singleGRPCOutbound{outboundName, serviceName, address}\n}\n\ntype singleGRPCOutbound struct {\n\toutboundName string\n\tserviceName  string\n\taddress      string\n}\n\nfunc (b singleGRPCOutbound) Build(grpc *grpc.Transport, _ *tchannel.Transport) (*Outbounds, error) {\n\treturn &Outbounds{\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\tb.outboundName: {\n\t\t\t\tServiceName: b.serviceName,\n\t\t\t\tUnary:       grpc.NewSingleOutbound(b.address),\n\t\t\t\tStream:      grpc.NewSingleOutbound(b.address),\n\t\t\t},\n\t\t},\n\t}, nil\n}\n"
  },
  {
    "path": "common/rpc/outbounds_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: outbounds.go\n//\n// Generated by this command:\n//\n//\tmockgen -package rpc -source outbounds.go -destination outbounds_mock.go -self_package github.com/uber/cadence/common/rpc\n//\n\n// Package rpc is a generated GoMock package.\npackage rpc\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tgrpc \"go.uber.org/yarpc/transport/grpc\"\n\ttchannel \"go.uber.org/yarpc/transport/tchannel\"\n)\n\n// MockOutboundsBuilder is a mock of OutboundsBuilder interface.\ntype MockOutboundsBuilder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockOutboundsBuilderMockRecorder\n\tisgomock struct{}\n}\n\n// MockOutboundsBuilderMockRecorder is the mock recorder for MockOutboundsBuilder.\ntype MockOutboundsBuilderMockRecorder struct {\n\tmock *MockOutboundsBuilder\n}\n\n// NewMockOutboundsBuilder creates a new mock instance.\nfunc NewMockOutboundsBuilder(ctrl *gomock.Controller) *MockOutboundsBuilder {\n\tmock := &MockOutboundsBuilder{ctrl: ctrl}\n\tmock.recorder = &MockOutboundsBuilderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockOutboundsBuilder) EXPECT() *MockOutboundsBuilderMockRecorder {\n\treturn m.recorder\n}\n\n// Build mocks base method.\nfunc (m *MockOutboundsBuilder) Build(arg0 *grpc.Transport, arg1 *tchannel.Transport) (*Outbounds, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Build\", arg0, arg1)\n\tret0, _ := ret[0].(*Outbounds)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Build indicates an expected call of Build.\nfunc (mr *MockOutboundsBuilderMockRecorder) Build(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Build\", reflect.TypeOf((*MockOutboundsBuilder)(nil).Build), arg0, arg1)\n}\n"
  },
  {
    "path": "common/rpc/outbounds_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"errors\"\n\t\"io/ioutil\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer/direct\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestCombineOutbounds(t *testing.T) {\n\tgrpc := &grpc.Transport{}\n\ttchannel := &tchannel.Transport{}\n\n\tcombined := CombineOutbounds()\n\toutbounds, err := combined.Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\tassert.Empty(t, outbounds.Outbounds)\n\n\tcombined = CombineOutbounds(fakeOutboundBuilder{err: errors.New(\"err-A\")})\n\t_, err = combined.Build(grpc, tchannel)\n\tassert.EqualError(t, err, \"err-A\")\n\n\tcombined = CombineOutbounds(\n\t\tfakeOutboundBuilder{outbounds: yarpc.Outbounds{\"A\": {}}},\n\t\tfakeOutboundBuilder{outbounds: yarpc.Outbounds{\"A\": {}}},\n\t)\n\t_, err = combined.Build(grpc, tchannel)\n\tassert.EqualError(t, err, \"outbound \\\"A\\\" already configured\")\n\n\tcombined = CombineOutbounds(\n\t\tfakeOutboundBuilder{err: errors.New(\"err-A\")},\n\t\tfakeOutboundBuilder{err: errors.New(\"err-B\")},\n\t)\n\t_, err = combined.Build(grpc, tchannel)\n\tassert.EqualError(t, err, \"err-A; err-B\")\n\n\tcombined = CombineOutbounds(\n\t\tfakeOutboundBuilder{outbounds: yarpc.Outbounds{\"A\": {}}},\n\t\tfakeOutboundBuilder{outbounds: yarpc.Outbounds{\"B\": {}, \"C\": {}}},\n\t)\n\toutbounds, err = combined.Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\tassert.Equal(t, yarpc.Outbounds{\n\t\t\"A\": {},\n\t\t\"B\": {},\n\t\t\"C\": {},\n\t}, outbounds.Outbounds)\n}\n\nfunc TestPublicClientOutbound(t *testing.T) {\n\tmakeConfig := func(hostPort string, transport string, enableAuth bool, keyPath string) *config.Config {\n\t\treturn &config.Config{\n\t\t\tPublicClient:  config.PublicClient{HostPort: hostPort, Transport: transport},\n\t\t\tAuthorization: config.Authorization{OAuthAuthorizer: config.OAuthAuthorizer{Enable: enableAuth}},\n\t\t\tClusterGroupMetadata: &config.ClusterGroupMetadata{\n\t\t\t\tCurrentClusterName: \"cluster-A\",\n\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\"cluster-A\": {\n\t\t\t\t\t\tAuthorizationProvider: config.AuthorizationProvider{\n\t\t\t\t\t\t\tPrivateKey: keyPath,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t}\n\n\t_, err := newPublicClientOutbound(&config.Config{})\n\trequire.EqualError(t, err, \"need to provide an endpoint config for PublicClient\")\n\n\tbuilder, err := newPublicClientOutbound(makeConfig(\"localhost:1234\", \"tchannel\", false, \"\"))\n\trequire.NoError(t, err)\n\trequire.NotNil(t, builder)\n\trequire.Equal(t, \"localhost:1234\", builder.address)\n\trequire.Equal(t, nil, builder.authMiddleware)\n\trequire.False(t, builder.isGRPC)\n\n\tbuilder, err = newPublicClientOutbound(makeConfig(\"localhost:1234\", \"tchannel\", true, \"invalid\"))\n\trequire.EqualError(t, err, \"create AuthProvider: invalid private key path invalid\")\n\trequire.False(t, builder.isGRPC)\n\n\tbuilder, err = newPublicClientOutbound(makeConfig(\"localhost:1234\", \"grpc\", true, tempFile(t, \"private-key\")))\n\trequire.NoError(t, err)\n\trequire.NotNil(t, builder)\n\trequire.Equal(t, \"localhost:1234\", builder.address)\n\trequire.NotNil(t, builder.authMiddleware)\n\trequire.True(t, builder.isGRPC)\n\n\tgrpc := &grpc.Transport{}\n\ttchannel := &tchannel.Transport{}\n\to, err := builder.Build(grpc, tchannel)\n\trequire.NoError(t, err)\n\toutbounds := o.Outbounds\n\tassert.Equal(t, outbounds[OutboundPublicClient].ServiceName, service.Frontend)\n\tassert.NotNil(t, outbounds[OutboundPublicClient].Unary)\n}\n\nfunc TestCrossDCOutbounds(t *testing.T) {\n\tgrpc := &grpc.Transport{}\n\ttchannel := &tchannel.Transport{}\n\n\tclusterGroup := map[string]config.ClusterInformation{\n\t\t\"cluster-A\": {Enabled: true, RPCName: \"cadence-frontend\", RPCTransport: \"invalid\"},\n\t}\n\t_, err := NewCrossDCOutbounds(clusterGroup, &fakePeerChooserFactory{}).Build(grpc, tchannel)\n\tassert.EqualError(t, err, \"unknown cross DC transport type: invalid\")\n\n\tclusterGroup = map[string]config.ClusterInformation{\n\t\t\"cluster-A\": {Enabled: true, RPCName: \"cadence-frontend\", RPCTransport: \"grpc\", AuthorizationProvider: config.AuthorizationProvider{Enable: true, PrivateKey: \"invalid path\"}},\n\t}\n\t_, err = NewCrossDCOutbounds(clusterGroup, &fakePeerChooserFactory{}).Build(grpc, tchannel)\n\tassert.EqualError(t, err, \"create AuthProvider: invalid private key path invalid path\")\n\n\tclusterGroup = map[string]config.ClusterInformation{\n\t\t\"cluster-A\": {Enabled: true, RPCName: \"cadence-frontend\", RPCAddress: \"address-A\", RPCTransport: \"grpc\", AuthorizationProvider: config.AuthorizationProvider{Enable: true, PrivateKey: tempFile(t, \"key\")}},\n\t\t\"cluster-B\": {Enabled: true, RPCName: \"cadence-frontend\", RPCAddress: \"address-B\", RPCTransport: \"tchannel\"},\n\t\t\"cluster-C\": {Enabled: false},\n\t}\n\to, err := NewCrossDCOutbounds(clusterGroup, &fakePeerChooserFactory{}).Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\toutbounds := o.Outbounds\n\tassert.Equal(t, 2, len(outbounds))\n\tassert.Equal(t, \"cadence-frontend\", outbounds[\"cluster-A\"].ServiceName)\n\tassert.Equal(t, \"cadence-frontend\", outbounds[\"cluster-B\"].ServiceName)\n\tassert.NotNil(t, outbounds[\"cluster-A\"].Unary)\n\tassert.NotNil(t, outbounds[\"cluster-B\"].Unary)\n}\n\nfunc TestDirectOutbound(t *testing.T) {\n\tgrpc := &grpc.Transport{}\n\ttchannel := &tchannel.Transport{}\n\tlogger := testlogger.New(t)\n\tmetricCl := metrics.NewNoopMetricsClient()\n\tfalseFn := func(opts ...dynamicproperties.FilterOption) bool { return false }\n\n\to, err := NewDirectOutboundBuilder(\"cadence-history\", false, nil, NewDirectPeerChooserFactory(\"cadence-history\", logger, metricCl), falseFn).Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\toutbounds := o.Outbounds\n\tassert.Equal(t, \"cadence-history\", outbounds[\"cadence-history\"].ServiceName)\n\tassert.NotNil(t, outbounds[\"cadence-history\"].Unary)\n\n\to, err = NewDirectOutboundBuilder(\"cadence-history\", true, nil, NewDirectPeerChooserFactory(\"cadence-history\", logger, metricCl), falseFn).Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\toutbounds = o.Outbounds\n\tassert.Equal(t, \"cadence-history\", outbounds[\"cadence-history\"].ServiceName)\n\tassert.NotNil(t, outbounds[\"cadence-history\"].Unary)\n}\n\nfunc TestSingleGRPCOutbound(t *testing.T) {\n\tgrpc := &grpc.Transport{}\n\ttchannel := &tchannel.Transport{}\n\n\tbuilder := NewSingleGRPCOutboundBuilder(\"grpc-only-out\", \"grpc-service-name\", \"http://example.com:1234\")\n\n\toutBound, err := builder.Build(grpc, tchannel)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"grpc-service-name\", outBound.Outbounds[\"grpc-only-out\"].ServiceName)\n\tassert.NotNil(t, outBound.Outbounds[\"grpc-only-out\"].Unary)\n}\n\nfunc TestIsGRPCOutboud(t *testing.T) {\n\tassert.True(t, IsGRPCOutbound(&transport.OutboundConfig{Outbounds: transport.Outbounds{Unary: (&grpc.Transport{}).NewSingleOutbound(\"localhost:1234\")}}))\n\tassert.False(t, IsGRPCOutbound(&transport.OutboundConfig{Outbounds: transport.Outbounds{Unary: (&tchannel.Transport{}).NewSingleOutbound(\"localhost:1234\")}}))\n\tassert.Panics(t, func() {\n\t\tIsGRPCOutbound(&transport.OutboundConfig{Outbounds: transport.Outbounds{Unary: fakeOutbound{}}})\n\t})\n}\n\nfunc tempFile(t *testing.T, content string) string {\n\tf, err := ioutil.TempFile(\"\", \"\")\n\trequire.NoError(t, err)\n\n\tf.Write([]byte(content))\n\trequire.NoError(t, err)\n\n\terr = f.Close()\n\trequire.NoError(t, err)\n\n\treturn f.Name()\n}\n\ntype fakeOutboundBuilder struct {\n\toutbounds yarpc.Outbounds\n\terr       error\n}\n\nfunc (b fakeOutboundBuilder) Build(*grpc.Transport, *tchannel.Transport) (*Outbounds, error) {\n\treturn &Outbounds{Outbounds: b.outbounds}, b.err\n}\n\ntype fakePeerChooserFactory struct{}\n\nfunc (f fakePeerChooserFactory) CreatePeerChooser(transport peer.Transport, opts PeerChooserOptions) (PeerChooser, error) {\n\tchooser, err := direct.New(direct.Configuration{}, transport)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &defaultPeerChooser{actual: chooser}, nil\n}\n"
  },
  {
    "path": "common/rpc/params.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"crypto/tls\"\n\t\"errors\"\n\t\"fmt\"\n\t\"net\"\n\t\"regexp\"\n\t\"strconv\"\n\n\t\"go.uber.org/yarpc\"\n\tyarpctls \"go.uber.org/yarpc/api/transport/tls\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// Params allows to configure rpc.Factory\ntype Params struct {\n\tServiceName     string\n\tTChannelAddress string\n\tGRPCAddress     string\n\tGRPCMaxMsgSize  int\n\tHTTP            *httpParams\n\n\tInboundTLS  *tls.Config\n\tOutboundTLS map[string]*tls.Config\n\n\tInboundMiddleware  yarpc.InboundMiddleware\n\tOutboundMiddleware yarpc.OutboundMiddleware\n\n\tOutboundsBuilder OutboundsBuilder\n}\n\ntype httpParams struct {\n\tAddress    string\n\tProcedures map[string]struct{}\n\tTLS        *tls.Config\n\tMode       yarpctls.Mode\n}\n\n// NewParams creates parameters for rpc.Factory from the given config\nfunc NewParams(serviceName string, config *config.Config, dc *dynamicconfig.Collection, logger log.Logger, metricsCl metrics.Client) (Params, error) {\n\tserviceConfig, err := config.GetServiceConfig(serviceName)\n\tif err != nil {\n\t\treturn Params{}, err\n\t}\n\n\tlistenIP, err := GetListenIP(serviceConfig.RPC)\n\tif err != nil {\n\t\treturn Params{}, fmt.Errorf(\"get listen IP: %v\", err)\n\t}\n\n\tinboundTLS, err := serviceConfig.RPC.TLS.ToTLSConfig()\n\tif err != nil {\n\t\treturn Params{}, fmt.Errorf(\"inbound TLS config: %v\", err)\n\t}\n\toutboundTLS := map[string]*tls.Config{}\n\tfor _, outboundServiceName := range service.List {\n\t\toutboundServiceConfig, err := config.GetServiceConfig(outboundServiceName)\n\t\tif err != nil {\n\t\t\tcontinue\n\t\t}\n\t\toutboundTLS[outboundServiceName], err = outboundServiceConfig.RPC.TLS.ToTLSConfig()\n\t\tif err != nil {\n\t\t\treturn Params{}, fmt.Errorf(\"outbound %s TLS config: %v\", outboundServiceName, err)\n\t\t}\n\t}\n\n\tenableGRPCOutbound := dc.GetBoolProperty(dynamicproperties.EnableGRPCOutbound)()\n\n\tpublicClientOutbound, err := newPublicClientOutbound(config)\n\tif err != nil {\n\t\treturn Params{}, fmt.Errorf(\"public client outbound: %v\", err)\n\t}\n\n\tforwardingRules, err := getForwardingRules(dc)\n\tif err != nil {\n\t\treturn Params{}, err\n\t}\n\tif len(forwardingRules) == 0 {\n\t\t// not set, load from static config\n\t\tforwardingRules = config.HeaderForwardingRules\n\t}\n\tvar http *httpParams\n\n\tif serviceConfig.RPC.HTTP != nil {\n\t\tif serviceConfig.RPC.HTTP.Port <= 0 {\n\t\t\treturn Params{}, errors.New(\"HTTP port is not set\")\n\t\t}\n\t\tprocedureMap := map[string]struct{}{}\n\n\t\tfor _, v := range serviceConfig.RPC.HTTP.Procedures {\n\t\t\tprocedureMap[v] = struct{}{}\n\t\t}\n\n\t\thttp = &httpParams{\n\t\t\tAddress:    net.JoinHostPort(listenIP.String(), strconv.Itoa(int(serviceConfig.RPC.HTTP.Port))),\n\t\t\tProcedures: procedureMap,\n\t\t}\n\n\t\tif serviceConfig.RPC.HTTP.TLS.Enabled {\n\t\t\thttptls, err := serviceConfig.RPC.HTTP.TLS.ToTLSConfig()\n\t\t\tif err != nil {\n\t\t\t\treturn Params{}, fmt.Errorf(\"creating TLS config for HTTP: %w\", err)\n\t\t\t}\n\n\t\t\thttp.TLS = httptls\n\t\t\thttp.Mode = serviceConfig.RPC.HTTP.TLSMode\n\t\t}\n\t}\n\n\toutboundsBuilders := []OutboundsBuilder{\n\t\tNewDirectOutboundBuilder(\n\t\t\tservice.History,\n\t\t\tenableGRPCOutbound,\n\t\t\toutboundTLS[service.History],\n\t\t\tNewDirectPeerChooserFactory(service.History, logger, metricsCl),\n\t\t\tdc.GetBoolProperty(dynamicproperties.EnableConnectionRetainingDirectChooser),\n\t\t),\n\t\tNewDirectOutboundBuilder(\n\t\t\tservice.Matching,\n\t\t\tenableGRPCOutbound,\n\t\t\toutboundTLS[service.Matching],\n\t\t\tNewDirectPeerChooserFactory(service.Matching, logger, metricsCl),\n\t\t\tdc.GetBoolProperty(dynamicproperties.EnableConnectionRetainingDirectChooser),\n\t\t),\n\t\tpublicClientOutbound,\n\t}\n\tif config.ShardDistributorClient.HostPort != \"\" {\n\t\toutboundsBuilders = append(outboundsBuilders, NewSingleGRPCOutboundBuilder(\n\t\t\tservice.ShardDistributor,\n\t\t\tservice.ShardDistributor,\n\t\t\tconfig.ShardDistributorClient.HostPort,\n\t\t))\n\t}\n\n\treturn Params{\n\t\tServiceName:      serviceName,\n\t\tHTTP:             http,\n\t\tTChannelAddress:  net.JoinHostPort(listenIP.String(), strconv.Itoa(int(serviceConfig.RPC.Port))),\n\t\tGRPCAddress:      net.JoinHostPort(listenIP.String(), strconv.Itoa(int(serviceConfig.RPC.GRPCPort))),\n\t\tGRPCMaxMsgSize:   serviceConfig.RPC.GRPCMaxMsgSize,\n\t\tOutboundsBuilder: CombineOutbounds(outboundsBuilders...),\n\t\tInboundTLS:       inboundTLS,\n\t\tOutboundTLS:      outboundTLS,\n\t\tInboundMiddleware: yarpc.InboundMiddleware{\n\t\t\t// order matters: ForwardPartitionConfigMiddleware must be applied after ClientPartitionConfigMiddleware\n\t\t\tUnary: yarpc.UnaryInboundMiddleware(&InboundMetricsMiddleware{}, &CallerInfoMiddleware{}, &ClientPartitionConfigMiddleware{}, &ForwardPartitionConfigMiddleware{}),\n\t\t},\n\t\tOutboundMiddleware: yarpc.OutboundMiddleware{\n\t\t\tUnary: yarpc.UnaryOutboundMiddleware(&HeaderForwardingMiddleware{\n\t\t\t\tRules: forwardingRules,\n\t\t\t}, &ForwardPartitionConfigMiddleware{}),\n\t\t},\n\t}, nil\n}\n\nfunc getForwardingRules(dc *dynamicconfig.Collection) ([]config.HeaderRule, error) {\n\tvar forwardingRules []config.HeaderRule\n\tdynForwarding := dc.GetListProperty(dynamicproperties.HeaderForwardingRules)()\n\tif len(dynForwarding) > 0 {\n\t\tfor _, f := range dynForwarding {\n\t\t\tswitch v := f.(type) {\n\t\t\tcase config.HeaderRule: // default or correctly typed value\n\t\t\t\tforwardingRules = append(forwardingRules, v)\n\t\t\tcase map[string]interface{}: // loaded from generic deserialization, compatible with encoding/json\n\t\t\t\tadd, aok := v[\"Add\"].(bool)\n\t\t\t\tm, mok := v[\"Match\"].(string)\n\t\t\t\tif !aok || !mok {\n\t\t\t\t\treturn nil, fmt.Errorf(\"invalid generic types for header forwarding rule: %#v\", v)\n\t\t\t\t}\n\t\t\t\tr, err := regexp.Compile(m)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, fmt.Errorf(\"invalid match regex in header forwarding rule: %q, err: %w\", m, err)\n\t\t\t\t}\n\t\t\t\tforwardingRules = append(forwardingRules, config.HeaderRule{\n\t\t\t\t\tAdd:   add,\n\t\t\t\t\tMatch: r,\n\t\t\t\t})\n\t\t\tdefault: // unknown\n\t\t\t\treturn nil, fmt.Errorf(\"unrecognized dynamic header forwarding type: %T\", v)\n\t\t\t}\n\t\t}\n\t}\n\treturn forwardingRules, nil\n}\n\n// GetListenIP returns the IP address to bind/listen on based on RPC config\n// It respects bindOnLocalHost, bindOnIP, or falls back to auto-detection\nfunc GetListenIP(config config.RPC) (net.IP, error) {\n\tif config.BindOnLocalHost && len(config.BindOnIP) > 0 {\n\t\treturn nil, fmt.Errorf(\"bindOnLocalHost and bindOnIP are mutually exclusive\")\n\t}\n\n\tif config.BindOnLocalHost {\n\t\treturn net.IPv4(127, 0, 0, 1), nil\n\t}\n\n\tif len(config.BindOnIP) > 0 {\n\t\tip := net.ParseIP(config.BindOnIP)\n\t\tif ip != nil && ip.To4() != nil {\n\t\t\treturn ip.To4(), nil\n\t\t}\n\t\tif ip != nil && ip.To16() != nil {\n\t\t\treturn ip.To16(), nil\n\t\t}\n\t\treturn nil, fmt.Errorf(\"unable to parse bindOnIP value or it is not an IPv4 or IPv6 address: %s\", config.BindOnIP)\n\t}\n\treturn ListenIP()\n}\n"
  },
  {
    "path": "common/rpc/params_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"net\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestNewParams(t *testing.T) {\n\tserviceName := service.Frontend\n\tdc := dynamicconfig.NewNopCollection()\n\tmakeConfig := func(svc config.Service) *config.Config {\n\t\treturn &config.Config{\n\t\t\tPublicClient:           config.PublicClient{HostPort: \"localhost:9999\"},\n\t\t\tShardDistributorClient: config.ShardDistributorClient{HostPort: \"localhost:9998\"},\n\t\t\tServices:               map[string]config.Service{\"frontend\": svc}}\n\t}\n\tlogger := testlogger.New(t)\n\tmetricsCl := metrics.NewNoopMetricsClient()\n\n\t_, err := NewParams(serviceName, &config.Config{}, dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"no config section for service: frontend\")\n\n\t_, err = NewParams(serviceName, makeConfig(config.Service{RPC: config.RPC{BindOnLocalHost: true, BindOnIP: \"1.2.3.4\"}}), dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"get listen IP: bindOnLocalHost and bindOnIP are mutually exclusive\")\n\n\t_, err = NewParams(serviceName, makeConfig(config.Service{RPC: config.RPC{BindOnIP: \"invalidIP\"}}), dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"get listen IP: unable to parse bindOnIP value or it is not an IPv4 or IPv6 address: invalidIP\")\n\n\t_, err = NewParams(serviceName, &config.Config{Services: map[string]config.Service{\"frontend\": {}}}, dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"public client outbound: need to provide an endpoint config for PublicClient\")\n\n\tcfg := makeConfig(config.Service{RPC: config.RPC{BindOnLocalHost: true, TLS: config.TLS{Enabled: true, CertFile: \"invalid\", KeyFile: \"invalid\"}}})\n\t_, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"inbound TLS config: open invalid: no such file or directory\")\n\n\tcfg = &config.Config{Services: map[string]config.Service{\n\t\t\"frontend\": {RPC: config.RPC{BindOnLocalHost: true}},\n\t\t\"history\":  {RPC: config.RPC{TLS: config.TLS{Enabled: true, CaFile: \"invalid\"}}},\n\t}}\n\t_, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.EqualError(t, err, \"outbound cadence-history TLS config: open invalid: no such file or directory\")\n\n\tcfg = makeConfig(config.Service{RPC: config.RPC{BindOnLocalHost: true, Port: 1111, GRPCPort: 2222, GRPCMaxMsgSize: 3333}})\n\tparams, err := NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"127.0.0.1:1111\", params.TChannelAddress)\n\tassert.Equal(t, \"127.0.0.1:2222\", params.GRPCAddress)\n\tassert.Equal(t, 3333, params.GRPCMaxMsgSize)\n\tassert.Nil(t, params.InboundTLS)\n\n\tcfg = makeConfig(config.Service{RPC: config.RPC{BindOnLocalHost: true, HTTP: &config.HTTP{Port: 8800}}})\n\tparams, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"127.0.0.1:8800\", params.HTTP.Address)\n\n\tcfg = makeConfig(config.Service{RPC: config.RPC{BindOnLocalHost: true, HTTP: &config.HTTP{}}})\n\tparams, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.Error(t, err)\n\n\tcfg = makeConfig(config.Service{RPC: config.RPC{BindOnIP: \"1.2.3.4\", GRPCPort: 2222}})\n\tparams, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"1.2.3.4:2222\", params.GRPCAddress)\n\n\tcfg = makeConfig(config.Service{RPC: config.RPC{GRPCPort: 2222, TLS: config.TLS{Enabled: true}}})\n\tparams, err = NewParams(serviceName, cfg, dc, logger, metricsCl)\n\tassert.NoError(t, err)\n\tip, port, err := net.SplitHostPort(params.GRPCAddress)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"2222\", port)\n\tassert.NotNil(t, net.ParseIP(ip))\n\tassert.NotNil(t, params.InboundTLS)\n}\n"
  },
  {
    "path": "common/rpc/peer_chooser.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination peer_chooser_mock.go -self_package github.com/uber/cadence/common/rpc\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer/roundrobin\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst defaultDNSRefreshInterval = time.Second * 10\n\ntype (\n\tPeerChooserOptions struct {\n\t\t// Address is the target dns address. Used by dns peer chooser.\n\t\tAddress string\n\n\t\t// ServiceName is the name of service. Used by direct peer chooser.\n\t\tServiceName string\n\n\t\t// EnableConnectionRetainingDirectChooser is used by direct peer chooser.\n\t\t// If false, yarpc's own default direct peer chooser will be used which doesn't retain connections.\n\t\t// If true, cadence's own direct peer chooser will be used which retains connections.\n\t\tEnableConnectionRetainingDirectChooser dynamicproperties.BoolPropertyFn\n\t}\n\tPeerChooserFactory interface {\n\t\tCreatePeerChooser(transport peer.Transport, opts PeerChooserOptions) (PeerChooser, error)\n\t}\n\n\tPeerChooser interface {\n\t\tpeer.Chooser\n\n\t\t// UpdatePeers updates the list of peers if needed.\n\t\tUpdatePeers(serviceName string, members []membership.HostInfo)\n\t}\n\n\tdnsPeerChooserFactory struct {\n\t\tinterval time.Duration\n\t\tlogger   log.Logger\n\t}\n\n\tdirectPeerChooserFactory struct {\n\t\tserviceName string\n\t\tlogger      log.Logger\n\t\tmetricsCl   metrics.Client\n\t\tchoosers    []*directPeerChooser\n\t}\n)\n\ntype defaultPeerChooser struct {\n\tactual peer.Chooser\n\tonStop func()\n}\n\n// UpdatePeers is a no-op for defaultPeerChooser. It is added to satisfy the PeerChooser interface.\nfunc (d *defaultPeerChooser) UpdatePeers(string, []membership.HostInfo) {}\n\n// Choose a Peer for the next call, block until a peer is available (or timeout)\nfunc (d *defaultPeerChooser) Choose(ctx context.Context, req *transport.Request) (peer peer.Peer, onFinish func(error), err error) {\n\treturn d.actual.Choose(ctx, req)\n}\n\nfunc (d *defaultPeerChooser) Start() error {\n\treturn d.actual.Start()\n}\n\nfunc (d *defaultPeerChooser) Stop() error {\n\tif d.onStop != nil {\n\t\td.onStop()\n\t}\n\treturn d.actual.Stop()\n}\n\nfunc (d *defaultPeerChooser) IsRunning() bool {\n\treturn d.actual.IsRunning()\n}\n\nfunc NewDNSPeerChooserFactory(interval time.Duration, logger log.Logger) PeerChooserFactory {\n\tif interval <= 0 {\n\t\tinterval = defaultDNSRefreshInterval\n\t}\n\n\treturn &dnsPeerChooserFactory{\n\t\tinterval: interval,\n\t\tlogger:   logger,\n\t}\n}\n\nfunc (f *dnsPeerChooserFactory) CreatePeerChooser(transport peer.Transport, opts PeerChooserOptions) (PeerChooser, error) {\n\tpeerList := roundrobin.New(transport)\n\tpeerListUpdater, err := newDNSUpdater(peerList, opts.Address, f.interval, f.logger)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpeerListUpdater.Start()\n\treturn &defaultPeerChooser{\n\t\tactual: peerList,\n\t\tonStop: peerListUpdater.Stop,\n\t}, nil\n}\n\nfunc NewDirectPeerChooserFactory(serviceName string, logger log.Logger, metricsCl metrics.Client) PeerChooserFactory {\n\treturn &directPeerChooserFactory{\n\t\tserviceName: serviceName,\n\t\tlogger:      logger,\n\t\tmetricsCl:   metricsCl,\n\t}\n}\n\nfunc (f *directPeerChooserFactory) CreatePeerChooser(transport peer.Transport, opts PeerChooserOptions) (PeerChooser, error) {\n\tc := newDirectChooser(f.serviceName, transport, f.logger, f.metricsCl, opts.EnableConnectionRetainingDirectChooser)\n\tf.choosers = append(f.choosers, c)\n\treturn c, nil\n}\n"
  },
  {
    "path": "common/rpc/peer_chooser_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: peer_chooser.go\n//\n// Generated by this command:\n//\n//\tmockgen -package rpc -source peer_chooser.go -destination peer_chooser_mock.go -self_package github.com/uber/cadence/common/rpc\n//\n\n// Package rpc is a generated GoMock package.\npackage rpc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tpeer \"go.uber.org/yarpc/api/peer\"\n\ttransport \"go.uber.org/yarpc/api/transport\"\n\n\tmembership \"github.com/uber/cadence/common/membership\"\n)\n\n// MockPeerChooserFactory is a mock of PeerChooserFactory interface.\ntype MockPeerChooserFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerChooserFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerChooserFactoryMockRecorder is the mock recorder for MockPeerChooserFactory.\ntype MockPeerChooserFactoryMockRecorder struct {\n\tmock *MockPeerChooserFactory\n}\n\n// NewMockPeerChooserFactory creates a new mock instance.\nfunc NewMockPeerChooserFactory(ctrl *gomock.Controller) *MockPeerChooserFactory {\n\tmock := &MockPeerChooserFactory{ctrl: ctrl}\n\tmock.recorder = &MockPeerChooserFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerChooserFactory) EXPECT() *MockPeerChooserFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// CreatePeerChooser mocks base method.\nfunc (m *MockPeerChooserFactory) CreatePeerChooser(transport peer.Transport, opts PeerChooserOptions) (PeerChooser, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreatePeerChooser\", transport, opts)\n\tret0, _ := ret[0].(PeerChooser)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreatePeerChooser indicates an expected call of CreatePeerChooser.\nfunc (mr *MockPeerChooserFactoryMockRecorder) CreatePeerChooser(transport, opts any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreatePeerChooser\", reflect.TypeOf((*MockPeerChooserFactory)(nil).CreatePeerChooser), transport, opts)\n}\n\n// MockPeerChooser is a mock of PeerChooser interface.\ntype MockPeerChooser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPeerChooserMockRecorder\n\tisgomock struct{}\n}\n\n// MockPeerChooserMockRecorder is the mock recorder for MockPeerChooser.\ntype MockPeerChooserMockRecorder struct {\n\tmock *MockPeerChooser\n}\n\n// NewMockPeerChooser creates a new mock instance.\nfunc NewMockPeerChooser(ctrl *gomock.Controller) *MockPeerChooser {\n\tmock := &MockPeerChooser{ctrl: ctrl}\n\tmock.recorder = &MockPeerChooserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPeerChooser) EXPECT() *MockPeerChooserMockRecorder {\n\treturn m.recorder\n}\n\n// Choose mocks base method.\nfunc (m *MockPeerChooser) Choose(arg0 context.Context, arg1 *transport.Request) (peer.Peer, func(error), error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Choose\", arg0, arg1)\n\tret0, _ := ret[0].(peer.Peer)\n\tret1, _ := ret[1].(func(error))\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// Choose indicates an expected call of Choose.\nfunc (mr *MockPeerChooserMockRecorder) Choose(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Choose\", reflect.TypeOf((*MockPeerChooser)(nil).Choose), arg0, arg1)\n}\n\n// IsRunning mocks base method.\nfunc (m *MockPeerChooser) IsRunning() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsRunning\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsRunning indicates an expected call of IsRunning.\nfunc (mr *MockPeerChooserMockRecorder) IsRunning() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsRunning\", reflect.TypeOf((*MockPeerChooser)(nil).IsRunning))\n}\n\n// Start mocks base method.\nfunc (m *MockPeerChooser) Start() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockPeerChooserMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockPeerChooser)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockPeerChooser) Stop() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Stop\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockPeerChooserMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockPeerChooser)(nil).Stop))\n}\n\n// UpdatePeers mocks base method.\nfunc (m *MockPeerChooser) UpdatePeers(serviceName string, members []membership.HostInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdatePeers\", serviceName, members)\n}\n\n// UpdatePeers indicates an expected call of UpdatePeers.\nfunc (mr *MockPeerChooserMockRecorder) UpdatePeers(serviceName, members any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdatePeers\", reflect.TypeOf((*MockPeerChooser)(nil).UpdatePeers), serviceName, members)\n}\n"
  },
  {
    "path": "common/rpc/peer_chooser_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage rpc\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tfakePeerTransport struct{}\n\tfakePeer          struct{}\n)\n\nfunc (t *fakePeerTransport) RetainPeer(peer.Identifier, peer.Subscriber) (peer.Peer, error) {\n\treturn &fakePeer{}, nil\n}\nfunc (t *fakePeerTransport) ReleasePeer(peer.Identifier, peer.Subscriber) error {\n\treturn nil\n}\n\nfunc (p *fakePeer) Identifier() string  { return \"fakePeer\" }\nfunc (p *fakePeer) Status() peer.Status { return peer.Status{ConnectionStatus: peer.Available} }\nfunc (p *fakePeer) StartRequest()       {}\nfunc (p *fakePeer) EndRequest()         {}\n\nfunc TestDNSPeerChooserFactory(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tlogger := log.NewNoop()\n\tctx := context.Background()\n\tinterval := 10 * time.Millisecond\n\n\tfactory := NewDNSPeerChooserFactory(interval, logger)\n\tpeerTransport := &fakePeerTransport{}\n\n\t// Ensure invalid address returns error\n\t_, err := factory.CreatePeerChooser(peerTransport, PeerChooserOptions{Address: \"invalid address\"})\n\tassert.EqualError(t, err, \"incorrect DNS:Port format\")\n\n\tchooser, err := factory.CreatePeerChooser(peerTransport, PeerChooserOptions{Address: \"localhost:1234\"})\n\trequire.NoError(t, err)\n\n\trequire.NoError(t, chooser.Start())\n\tdefer chooser.Stop()\n\n\trequire.True(t, chooser.IsRunning())\n\n\t// Wait for refresh\n\ttime.Sleep(interval + 50*time.Millisecond)\n\n\tpeer, _, err := chooser.Choose(ctx, &transport.Request{})\n\trequire.NoError(t, err)\n\trequire.NotNil(t, peer)\n\tassert.Equal(t, \"fakePeer\", peer.Identifier())\n}\n\nfunc TestDirectPeerChooserFactory(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tmetricCl := metrics.NewNoopMetricsClient()\n\tserviceName := \"service\"\n\tpcf := NewDirectPeerChooserFactory(serviceName, logger, metricCl)\n\tdirectConnRetainFn := func(opts ...dynamicproperties.FilterOption) bool { return false }\n\tgrpcTransport := grpc.NewTransport()\n\tchooser, err := pcf.CreatePeerChooser(grpcTransport, PeerChooserOptions{\n\t\tServiceName:                            serviceName,\n\t\tEnableConnectionRetainingDirectChooser: directConnRetainFn,\n\t})\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to create direct peer chooser: %v\", err)\n\t}\n\tif chooser == nil {\n\t\tt.Fatal(\"Failed to create direct peer chooser: nil\")\n\t}\n\n\tif _, dc := chooser.(*directPeerChooser); !dc {\n\t\tt.Fatalf(\"Want chooser be of type (*directPeerChooser), got %d\", chooser)\n\t}\n}\n"
  },
  {
    "path": "common/rpc/rpcfx/rpcfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage rpcfx\n\nimport (\n\t\"fmt\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n)\n\n// Module provides rpc.Params and rpc.Factory for fx application.\nvar Module = fx.Module(\"rpcfx\",\n\tfx.Provide(paramsBuilder),\n\tfx.Provide(buildFactory),\n)\n\ntype paramsBuilderParams struct {\n\tfx.In\n\n\tServiceFullName   string `name:\"service-full-name\"`\n\tCfg               config.Config\n\tLogger            log.Logger\n\tDynamicCollection *dynamicconfig.Collection\n\tMetricsClient     metrics.Client\n}\n\nfunc paramsBuilder(p paramsBuilderParams) (rpc.Params, error) {\n\tres, err := rpc.NewParams(p.ServiceFullName, &p.Cfg, p.DynamicCollection, p.Logger, p.MetricsClient)\n\tif err != nil {\n\t\treturn rpc.Params{}, fmt.Errorf(\"create rpc params: %w\", err)\n\t}\n\treturn res, nil\n}\n\ntype factoryParams struct {\n\tfx.In\n\n\tLogger    log.Logger\n\tRPCParams rpc.Params\n\n\tLifecycle fx.Lifecycle\n}\n\nfunc buildFactory(p factoryParams) rpc.Factory {\n\tres := rpc.NewFactory(p.Logger, p.RPCParams)\n\tp.Lifecycle.Append(fx.StartStopHook(startDispatcher(res), rpcStopper(res)))\n\treturn res\n}\n\nfunc startDispatcher(f rpc.Factory) func() error {\n\treturn func() error {\n\t\treturn f.GetDispatcher().Start()\n\t}\n}\n\nfunc rpcStopper(factory rpc.Factory) func() error {\n\treturn func() error {\n\t\terr := factory.GetDispatcher().Stop()\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"dispatcher stop: %w\", err)\n\t\t}\n\t\treturn nil\n\t}\n}\n"
  },
  {
    "path": "common/rpc/types.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination factory_mock.go -self_package github.com/uber/cadence/common/rpc\n\npackage rpc\n\nimport (\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\n\t\"github.com/uber/cadence/common/membership\"\n)\n\n// Factory Creates a dispatcher that knows how to transport requests.\ntype Factory interface {\n\tGetDispatcher() *yarpc.Dispatcher\n\tGetMaxMessageSize() int\n\tStart(PeerLister) error\n\tGetTChannel() tchannel.Channel\n\tStop() error\n}\n\ntype PeerLister interface {\n\tSubscribe(service, name string, notifyChannel chan<- *membership.ChangedEvent) error\n\tUnsubscribe(service, name string) error\n\tMembers(service string) ([]membership.HostInfo, error)\n}\n"
  },
  {
    "path": "common/rsa.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nimport (\n\t\"crypto/rsa\"\n\t\"crypto/x509\"\n\t\"encoding/pem\"\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n)\n\ntype KeyType string\n\nconst (\n\tKeyTypePrivate KeyType = \"private key\"\n\n\tKeyTypePublic KeyType = \"public key\"\n)\n\nfunc loadRSAKey(path string, keyType KeyType) (interface{}, error) {\n\tkeyString, err := os.ReadFile(path)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid %s path %s\", keyType, path)\n\t}\n\tblock, _ := pem.Decode(keyString)\n\tif block == nil || strings.ToLower(block.Type) != strings.ToLower(string(keyType)) {\n\t\treturn nil, fmt.Errorf(\"failed to parse PEM block containing the %s\", keyType)\n\t}\n\n\tswitch keyType {\n\tcase KeyTypePrivate:\n\t\tkey, err := x509.ParsePKCS8PrivateKey(block.Bytes)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to parse DER encoded %s: %s\", keyType, err.Error())\n\t\t}\n\t\treturn key, nil\n\tcase KeyTypePublic:\n\t\tkey, err := x509.ParsePKIXPublicKey(block.Bytes)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to parse DER encoded %s: %s\", keyType, err.Error())\n\t\t}\n\t\treturn key, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"invalid Key Type\")\n\t}\n}\n\nfunc LoadRSAPublicKey(path string) (*rsa.PublicKey, error) {\n\tkey, err := loadRSAKey(path, KeyTypePublic)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn key.(*rsa.PublicKey), err\n}\n\nfunc LoadRSAPrivateKey(path string) (*rsa.PrivateKey, error) {\n\tkey, err := loadRSAKey(path, KeyTypePrivate)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn key.(*rsa.PrivateKey), err\n}\n"
  },
  {
    "path": "common/scripting/exec.go",
    "content": "package scripting\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"io\"\n\t\"os\"\n\t\"os/exec\"\n)\n\n// Executor is a helper for running scripting\ntype Executor interface {\n\tBashExec(ctx context.Context, in string) (stdout string, stderr string, exitErr *exec.ExitError)\n\tExec(ctx context.Context, bin string, args ...string) (stdout string, stderr string, exitErr *exec.ExitError)\n\tQuietBashExec(ctx context.Context, in string) (stdout string, stderr string, exitErr *exec.ExitError)\n\tQuietExec(ctx context.Context, bin string, args ...string) (stdout string, stderr string, exitErr *exec.ExitError)\n}\n\n// New returns a new bash executor\nfunc New() Executor {\n\treturn &execImpl{}\n}\n\ntype execImpl struct{}\n\n// BashExec is a helper function for ensuring the user\n// can read a legible streaming from a subprocess,\n// as well as allowing programmatic access to stdout, stderr and exit codes\n// it's highly unsafe for any kind of untrusted inputs as it's explicitly bypassing\n// go's exec safety args, so it *must not* come into contact with anything untrusted\nfunc (execImpl) BashExec(ctx context.Context, in string) (stdout string, stderr string, exitErr *exec.ExitError) {\n\tcmd := exec.CommandContext(ctx, \"bash\", \"-c\", in)\n\tvar stdBuffer bytes.Buffer\n\tvar stdErrBuffer bytes.Buffer\n\tmw := io.MultiWriter(os.Stdout, &stdBuffer)\n\tmwErr := io.MultiWriter(os.Stderr, &stdErrBuffer)\n\tcmd.Stdout = mw\n\tcmd.Stderr = mwErr\n\terr := cmd.Run()\n\tvar e *exec.ExitError\n\tif errors.As(err, &e) && e.ExitCode() != 0 {\n\t\treturn stdBuffer.String(), stdErrBuffer.String(), e\n\t} else if err != nil {\n\t\tpanic(err)\n\t}\n\treturn stdBuffer.String(), stdErrBuffer.String(), nil\n}\n\n// Exec is a wrapper around exec.Command which adds some convenience\n// functionality to both capture standout/err as well as tee it to the user's UI in real time\n// meaning that the user doesn't need to wait for the command to complete.\n// It's value is fairly marginal and if it presents any problems the user should consider just\n// using exec.Command directly\nfunc (execImpl) Exec(ctx context.Context, bin string, args ...string) (stdout string, stderr string, exitErr *exec.ExitError) {\n\tcmd := exec.CommandContext(ctx, bin, args...)\n\tvar stdBuffer bytes.Buffer\n\tvar stdErrBuffer bytes.Buffer\n\tmw := io.MultiWriter(os.Stdout, &stdBuffer)\n\tmwErr := io.MultiWriter(os.Stderr, &stdErrBuffer)\n\tcmd.Stdout = mw\n\tcmd.Stderr = mwErr\n\terr := cmd.Run()\n\tvar e *exec.ExitError\n\tif errors.As(err, &e) && e.ExitCode() != 0 {\n\t\treturn stdBuffer.String(), stdErrBuffer.String(), e\n\t} else if err != nil {\n\t\tpanic(err)\n\t}\n\treturn stdBuffer.String(), stdErrBuffer.String(), nil\n}\n\n// QuietBashExec ...\nfunc (execImpl) QuietBashExec(ctx context.Context, in string) (stdout string, stderr string, exitErr *exec.ExitError) {\n\tcmd := exec.CommandContext(ctx, \"bash\", \"-c\", in)\n\tvar stdBuffer bytes.Buffer\n\tvar stdErrBuffer bytes.Buffer\n\tcmd.Stdout = &stdBuffer\n\tcmd.Stderr = &stdErrBuffer\n\terr := cmd.Run()\n\tvar e *exec.ExitError\n\tif errors.As(err, &e) && e.ExitCode() != 0 {\n\t\treturn stdBuffer.String(), stdErrBuffer.String(), e\n\t} else if err != nil {\n\t\tpanic(err)\n\t}\n\treturn stdBuffer.String(), stdErrBuffer.String(), nil\n}\n\n// QuietExec ...\nfunc (execImpl) QuietExec(ctx context.Context, bin string, args ...string) (stdout string, stderr string, exitErr *exec.ExitError) {\n\tcmd := exec.CommandContext(ctx, bin, args...)\n\tvar stdBuffer bytes.Buffer\n\tvar stdErrBuffer bytes.Buffer\n\tcmd.Stdout = &stdBuffer\n\tcmd.Stderr = &stdErrBuffer\n\terr := cmd.Run()\n\tvar e *exec.ExitError\n\tif errors.As(err, &e) && e.ExitCode() != 0 {\n\t\treturn stdBuffer.String(), stdErrBuffer.String(), e\n\t} else if err != nil {\n\t\tpanic(err)\n\t}\n\treturn stdBuffer.String(), stdErrBuffer.String(), nil\n}\n"
  },
  {
    "path": "common/scripting/exec_test.go",
    "content": "package scripting\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestBashExec(t *testing.T) {\n\tscript := New()\n\n\tstdout, stderr, err := script.BashExec(context.Background(), \"echo test\")\n\tassert.Equal(t, \"test\\n\", stdout)\n\tassert.Equal(t, \"\", stderr)\n\tassert.Nil(t, err)\n\n\tstdout, stderr, err = script.BashExec(context.Background(), \"echo test 1>&2\")\n\tassert.Equal(t, \"test\\n\", stderr)\n\tassert.Equal(t, \"\", stdout)\n\tassert.Nil(t, err)\n\n\tstdout, stderr, err = script.BashExec(context.Background(), \"false\")\n\tassert.Equal(t, \"\", stderr)\n\tassert.Equal(t, \"\", stdout)\n\tassert.Error(t, err)\n\n\ttests := map[string]struct {\n\t\targs           string\n\t\texpectedStdout string\n\t\texpectedSterr  string\n\t\texpectedErr    bool\n\t}{\n\t\t\"stdout\": {\n\t\t\targs:           \"echo test\",\n\t\t\texpectedStdout: \"test\\n\",\n\t\t\texpectedSterr:  \"\",\n\t\t\texpectedErr:    false,\n\t\t},\n\t\t\"stderr\": {\n\t\t\targs:           \"echo test 1>&2\",\n\t\t\texpectedStdout: \"\",\n\t\t\texpectedSterr:  \"test\\n\",\n\t\t\texpectedErr:    false,\n\t\t},\n\t\t\"error\": {\n\t\t\targs:           \"false\",\n\t\t\texpectedStdout: \"\",\n\t\t\texpectedSterr:  \"\",\n\t\t\texpectedErr:    true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstdout, stderr, err := script.BashExec(context.Background(), td.args)\n\t\t\tassert.Equal(t, td.expectedStdout, stdout)\n\t\t\tassert.Equal(t, td.expectedSterr, stderr)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\n\t\t\tstdout, stderr, err = script.QuietBashExec(context.Background(), td.args)\n\t\t\tassert.Equal(t, td.expectedStdout, stdout)\n\t\t\tassert.Equal(t, td.expectedSterr, stderr)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n\n}\n\nfunc TestExec(t *testing.T) {\n\tscript := New()\n\n\ttests := map[string]struct {\n\t\tbin            string\n\t\targs           []string\n\t\texpectedStdout string\n\t\texpectedSterr  string\n\t\texpectedErr    bool\n\t}{\n\t\t\"stdout\": {\n\t\t\tbin:            \"echo\",\n\t\t\targs:           []string{\"test\"},\n\t\t\texpectedStdout: \"test\\n\",\n\t\t\texpectedSterr:  \"\",\n\t\t\texpectedErr:    false,\n\t\t},\n\t\t\"error\": {\n\t\t\tbin:            \"false\",\n\t\t\targs:           []string{\"\"},\n\t\t\texpectedStdout: \"\",\n\t\t\texpectedSterr:  \"\",\n\t\t\texpectedErr:    true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstdout, stderr, err := script.Exec(context.Background(), td.bin, td.args...)\n\t\t\tassert.Equal(t, td.expectedStdout, stdout)\n\t\t\tassert.Equal(t, td.expectedSterr, stderr)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\n\t\t\tstdout, stderr, err = script.QuietExec(context.Background(), td.bin, td.args...)\n\t\t\tassert.Equal(t, td.expectedStdout, stdout)\n\t\t\tassert.Equal(t, td.expectedSterr, stderr)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/service/config.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage service\n\nimport (\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype (\n\t// Config is a subset of the service dynamic config for single service\n\tConfig struct {\n\t\tPersistenceMaxQPS       dynamicproperties.IntPropertyFn\n\t\tPersistenceGlobalMaxQPS dynamicproperties.IntPropertyFn\n\t\tThrottledLoggerMaxRPS   dynamicproperties.IntPropertyFn\n\n\t\t// WriteVisibilityStoreName is the write mode of visibility\n\t\tWriteVisibilityStoreName dynamicproperties.StringPropertyFn\n\t\t// EnableLogCustomerQueryParameter is to enable log customer parameters\n\t\tEnableLogCustomerQueryParameter dynamicproperties.BoolPropertyFnWithDomainFilter\n\t\t// ReadVisibilityStoreName is the read store for visibility\n\t\tReadVisibilityStoreName dynamicproperties.StringPropertyFnWithDomainFilter\n\n\t\t// configs for db visibility\n\t\tEnableDBVisibilitySampling                  dynamicproperties.BoolPropertyFn                `yaml:\"-\" json:\"-\"`\n\t\tEnableReadDBVisibilityFromClosedExecutionV2 dynamicproperties.BoolPropertyFn                `yaml:\"-\" json:\"-\"`\n\t\tWriteDBVisibilityOpenMaxQPS                 dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\t\tWriteDBVisibilityClosedMaxQPS               dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\t\tDBVisibilityListMaxQPS                      dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\n\t\t// configs for es visibility\n\t\tESIndexMaxResultWindow          dynamicproperties.IntPropertyFn `yaml:\"-\" json:\"-\"`\n\t\tValidSearchAttributes           dynamicproperties.MapPropertyFn `yaml:\"-\" json:\"-\"`\n\t\tPinotOptimizedQueryColumns      dynamicproperties.MapPropertyFn `yaml:\"-\" json:\"-\"`\n\t\tSearchAttributesHiddenValueKeys dynamicproperties.MapPropertyFn `yaml:\"-\" json:\"-\"`\n\t\t// deprecated: never read from, all ES reads and writes erroneously use PersistenceMaxQPS\n\t\tESVisibilityListMaxQPS dynamicproperties.IntPropertyFnWithDomainFilter `yaml:\"-\" json:\"-\"`\n\n\t\tIsErrorRetryableFunction backoff.IsRetryable\n\t}\n)\n"
  },
  {
    "path": "common/service/metrics.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage service\n\nimport (\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// GetMetricsServiceIdx returns the metrics name\nfunc GetMetricsServiceIdx(serviceName string, logger log.Logger) metrics.ServiceIdx {\n\tswitch serviceName {\n\tcase Frontend:\n\t\treturn metrics.Frontend\n\tcase History:\n\t\treturn metrics.History\n\tcase Matching:\n\t\treturn metrics.Matching\n\tcase Worker:\n\t\treturn metrics.Worker\n\tcase ShardDistributor:\n\t\treturn metrics.ShardDistributor\n\tdefault:\n\t\tlogger.Fatal(\"Unknown service name for metrics!\")\n\t}\n\n\t// this should never happen!\n\treturn metrics.NumServices\n}\n"
  },
  {
    "path": "common/service/name.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage service\n\nimport \"strings\"\n\nconst (\n\t_servicePrefix = \"cadence-\"\n\n\t// Frontend is the name of the frontend service\n\tFrontend = \"cadence-frontend\"\n\t// History is the name of the history service\n\tHistory = \"cadence-history\"\n\t// Matching is the name of the matching service\n\tMatching = \"cadence-matching\"\n\t// Worker is the name of the worker service\n\tWorker = \"cadence-worker\"\n\t// ShardDistributor is the name of the shard distributor service\n\tShardDistributor = \"cadence-shard-distributor\"\n)\n\n// ListWithRing contains the list of all cadence services that has a hash ring\nvar ListWithRing = []string{Frontend, History, Matching, Worker}\n\n// List contains the list of all cadence services\nvar List = []string{Frontend, History, Matching, Worker, ShardDistributor}\n\n// ShortName returns cadence service name without \"cadence-\" prefix\nfunc ShortName(name string) string {\n\treturn strings.TrimPrefix(name, _servicePrefix)\n}\n\n// FullName returns cadence service name with \"cadence-\" prefix\nfunc FullName(name string) string {\n\tif strings.HasPrefix(name, _servicePrefix) {\n\t\treturn name\n\t}\n\treturn _servicePrefix + name\n}\n\nfunc ShortNames(names []string) []string {\n\tresult := make([]string, len(names))\n\tfor i := range names {\n\t\tresult[i] = ShortName(names[i])\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "common/service/name_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage service\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestServiceNames(t *testing.T) {\n\tshortName := \"frontend\"\n\tfullName := \"cadence-frontend\"\n\n\tassert.Equal(t, shortName, ShortName(shortName))\n\tassert.Equal(t, shortName, ShortName(fullName))\n\n\tassert.Equal(t, fullName, FullName(shortName))\n\tassert.Equal(t, fullName, FullName(fullName))\n\n\tassert.Equal(t, []string{\"cadence-frontend\", \"cadence-history\", \"cadence-matching\", \"cadence-worker\", \"cadence-shard-distributor\"}, List)\n\tassert.Equal(t, []string{\"frontend\", \"history\", \"matching\", \"worker\", \"shard-distributor\"}, ShortNames(List))\n}\n"
  },
  {
    "path": "common/stats/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package stats -source interfaces.go -destination interface_mock.go -self_package github.com/uber/cadence/common/stats QPSTracker\n//\n\n// Package stats is a generated GoMock package.\npackage stats\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockQPSTracker is a mock of QPSTracker interface.\ntype MockQPSTracker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQPSTrackerMockRecorder\n\tisgomock struct{}\n}\n\n// MockQPSTrackerMockRecorder is the mock recorder for MockQPSTracker.\ntype MockQPSTrackerMockRecorder struct {\n\tmock *MockQPSTracker\n}\n\n// NewMockQPSTracker creates a new mock instance.\nfunc NewMockQPSTracker(ctrl *gomock.Controller) *MockQPSTracker {\n\tmock := &MockQPSTracker{ctrl: ctrl}\n\tmock.recorder = &MockQPSTrackerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQPSTracker) EXPECT() *MockQPSTrackerMockRecorder {\n\treturn m.recorder\n}\n\n// QPS mocks base method.\nfunc (m *MockQPSTracker) QPS() float64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QPS\")\n\tret0, _ := ret[0].(float64)\n\treturn ret0\n}\n\n// QPS indicates an expected call of QPS.\nfunc (mr *MockQPSTrackerMockRecorder) QPS() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QPS\", reflect.TypeOf((*MockQPSTracker)(nil).QPS))\n}\n\n// ReportCounter mocks base method.\nfunc (m *MockQPSTracker) ReportCounter(arg0 int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ReportCounter\", arg0)\n}\n\n// ReportCounter indicates an expected call of ReportCounter.\nfunc (mr *MockQPSTrackerMockRecorder) ReportCounter(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReportCounter\", reflect.TypeOf((*MockQPSTracker)(nil).ReportCounter), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockQPSTracker) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockQPSTrackerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockQPSTracker)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockQPSTracker) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockQPSTrackerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockQPSTracker)(nil).Stop))\n}\n\n// MockQPSTrackerGroup is a mock of QPSTrackerGroup interface.\ntype MockQPSTrackerGroup struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQPSTrackerGroupMockRecorder\n\tisgomock struct{}\n}\n\n// MockQPSTrackerGroupMockRecorder is the mock recorder for MockQPSTrackerGroup.\ntype MockQPSTrackerGroupMockRecorder struct {\n\tmock *MockQPSTrackerGroup\n}\n\n// NewMockQPSTrackerGroup creates a new mock instance.\nfunc NewMockQPSTrackerGroup(ctrl *gomock.Controller) *MockQPSTrackerGroup {\n\tmock := &MockQPSTrackerGroup{ctrl: ctrl}\n\tmock.recorder = &MockQPSTrackerGroupMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQPSTrackerGroup) EXPECT() *MockQPSTrackerGroupMockRecorder {\n\treturn m.recorder\n}\n\n// GroupQPS mocks base method.\nfunc (m *MockQPSTrackerGroup) GroupQPS(group string) float64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GroupQPS\", group)\n\tret0, _ := ret[0].(float64)\n\treturn ret0\n}\n\n// GroupQPS indicates an expected call of GroupQPS.\nfunc (mr *MockQPSTrackerGroupMockRecorder) GroupQPS(group any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GroupQPS\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).GroupQPS), group)\n}\n\n// QPS mocks base method.\nfunc (m *MockQPSTrackerGroup) QPS() float64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QPS\")\n\tret0, _ := ret[0].(float64)\n\treturn ret0\n}\n\n// QPS indicates an expected call of QPS.\nfunc (mr *MockQPSTrackerGroupMockRecorder) QPS() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QPS\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).QPS))\n}\n\n// ReportCounter mocks base method.\nfunc (m *MockQPSTrackerGroup) ReportCounter(arg0 int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ReportCounter\", arg0)\n}\n\n// ReportCounter indicates an expected call of ReportCounter.\nfunc (mr *MockQPSTrackerGroupMockRecorder) ReportCounter(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReportCounter\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).ReportCounter), arg0)\n}\n\n// ReportGroup mocks base method.\nfunc (m *MockQPSTrackerGroup) ReportGroup(group string, amount int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ReportGroup\", group, amount)\n}\n\n// ReportGroup indicates an expected call of ReportGroup.\nfunc (mr *MockQPSTrackerGroupMockRecorder) ReportGroup(group, amount any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReportGroup\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).ReportGroup), group, amount)\n}\n\n// Start mocks base method.\nfunc (m *MockQPSTrackerGroup) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockQPSTrackerGroupMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockQPSTrackerGroup) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockQPSTrackerGroupMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockQPSTrackerGroup)(nil).Stop))\n}\n"
  },
  {
    "path": "common/stats/interfaces.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/stats QPSTracker\n\npackage stats\n\nimport (\n\t\"github.com/uber/cadence/common\"\n)\n\n// QPSTracker is an interface for reporting statistics related to quotas.\ntype QPSTracker interface {\n\tcommon.Daemon\n\t// ReportCounter reports the value of a counter.\n\tReportCounter(int64)\n\n\t// QPS returns the current queries per second (QPS) value.\n\tQPS() float64\n}\n\n// QPSTrackerGroup allows for estimating QPS metrics with an additional dimension\ntype QPSTrackerGroup interface {\n\tQPSTracker\n\n\tReportGroup(group string, amount int64)\n\n\tGroupQPS(group string) float64\n}\n"
  },
  {
    "path": "common/stats/stats.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage stats\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\ntype (\n\t// emaFixedWindowQPSTracker is a QPSTracker that uses a fixed time period to calculate QPS and an exponential moving average algorithm to estimate QPS.\n\temaFixedWindowQPSTracker struct {\n\t\tgroups         sync.Map\n\t\troot           *emaFixedWindowState\n\t\ttimeSource     clock.TimeSource\n\t\texp            float64\n\t\tbucketInterval time.Duration\n\t\twg             sync.WaitGroup\n\t\tdone           chan struct{}\n\t\tstatus         *atomic.Int32\n\t\tbaseEvent      event.E\n\t}\n\n\temaFixedWindowState struct {\n\t\texp                   float64\n\t\tbucketIntervalSeconds float64\n\t\tfirstBucket           bool\n\t\tbaseEvent             event.E\n\t\tqps                   *atomic.Float64\n\t\tcounter               *atomic.Int64\n\t}\n)\n\nfunc NewEmaFixedWindowQPSTracker(timeSource clock.TimeSource, exp float64, bucketInterval time.Duration, baseEvent event.E) QPSTrackerGroup {\n\treturn &emaFixedWindowQPSTracker{\n\t\troot:           newEmaFixedWindowState(exp, bucketInterval, baseEvent),\n\t\ttimeSource:     timeSource,\n\t\texp:            exp,\n\t\tbucketInterval: bucketInterval,\n\t\tdone:           make(chan struct{}),\n\t\tstatus:         atomic.NewInt32(common.DaemonStatusInitialized),\n\t}\n}\n\nfunc (r *emaFixedWindowQPSTracker) Start() {\n\tif !r.status.CompareAndSwap(common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tr.wg.Add(1)\n\tgo r.reportLoop()\n}\n\nfunc (r *emaFixedWindowQPSTracker) reportLoop() {\n\tdefer r.wg.Done()\n\tticker := r.timeSource.NewTicker(r.bucketInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tr.report()\n\t\tcase <-r.done:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (r *emaFixedWindowQPSTracker) report() {\n\tr.root.report()\n\tr.groups.Range(func(key, value any) bool {\n\t\tstate := value.(*emaFixedWindowState)\n\t\tstate.report()\n\t\treturn true\n\t})\n}\n\nfunc (r *emaFixedWindowQPSTracker) Stop() {\n\tif !r.status.CompareAndSwap(common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(r.done)\n\tr.wg.Wait()\n}\n\nfunc (r *emaFixedWindowQPSTracker) ReportCounter(amount int64) {\n\tr.root.add(amount)\n}\n\nfunc (r *emaFixedWindowQPSTracker) ReportGroup(group string, amount int64) {\n\tr.root.add(amount)\n\tr.getOrCreate(group).add(amount)\n}\n\nfunc (r *emaFixedWindowQPSTracker) GroupQPS(group string) float64 {\n\tstate, ok := r.get(group)\n\tif !ok {\n\t\treturn 0\n\t}\n\treturn state.getQPS()\n}\n\nfunc (r *emaFixedWindowQPSTracker) QPS() float64 {\n\treturn r.root.getQPS()\n}\n\nfunc (r *emaFixedWindowQPSTracker) get(group string) (*emaFixedWindowState, bool) {\n\tres, ok := r.groups.Load(group)\n\tif !ok {\n\t\treturn nil, false\n\t}\n\treturn res.(*emaFixedWindowState), true\n}\n\nfunc (r *emaFixedWindowQPSTracker) getOrCreate(group string) *emaFixedWindowState {\n\tres, ok := r.groups.Load(group)\n\tif !ok {\n\t\te := r.baseEvent\n\t\te.Payload = map[string]any{\n\t\t\t\"group\": group,\n\t\t}\n\t\tres, _ = r.groups.LoadOrStore(group, newEmaFixedWindowState(r.exp, r.bucketInterval, e))\n\t}\n\treturn res.(*emaFixedWindowState)\n}\n\nfunc newEmaFixedWindowState(exp float64, bucketInterval time.Duration, baseEvent event.E) *emaFixedWindowState {\n\treturn &emaFixedWindowState{\n\t\texp:                   exp,\n\t\tbucketIntervalSeconds: bucketInterval.Seconds(),\n\t\tfirstBucket:           true,\n\t\tbaseEvent:             baseEvent,\n\t\tqps:                   atomic.NewFloat64(0),\n\t\tcounter:               atomic.NewInt64(0),\n\t}\n}\n\nfunc (s *emaFixedWindowState) report() {\n\tif s.firstBucket {\n\t\tcounter := s.counter.Swap(0)\n\t\ts.store(float64(counter) / s.bucketIntervalSeconds)\n\t\ts.firstBucket = false\n\t\treturn\n\t}\n\tcounter := s.counter.Swap(0)\n\tqps := s.qps.Load()\n\ts.store(qps*(1-s.exp) + float64(counter)*s.exp/s.bucketIntervalSeconds)\n}\n\nfunc (s *emaFixedWindowState) store(qps float64) {\n\ts.qps.Store(qps)\n\te := s.baseEvent\n\te.EventName = \"QPSTrackerUpdate\"\n\tpayload := make(map[string]any, len(e.Payload)+1)\n\tfor k, v := range e.Payload {\n\t\tpayload[k] = v\n\t}\n\tpayload[\"QPS\"] = qps\n\te.Payload = payload\n\tevent.Log(e)\n}\n\nfunc (s *emaFixedWindowState) add(delta int64) {\n\ts.counter.Add(delta)\n}\n\nfunc (s *emaFixedWindowState) getQPS() float64 {\n\treturn s.qps.Load()\n}\n\ntype (\n\tbucket struct {\n\t\tcounter   int64\n\t\ttimeIndex int\n\t}\n\trollingWindowQPSTracker struct {\n\t\tsync.RWMutex\n\t\ttimeSource     clock.TimeSource\n\t\tbucketInterval time.Duration\n\t\tbuckets        []bucket\n\n\t\tcounter   int64\n\t\ttimeIndex int\n\t}\n)\n\nfunc NewRollingWindowQPSTracker(timeSource clock.TimeSource, bucketInterval time.Duration, numBuckets int) QPSTracker {\n\treturn &rollingWindowQPSTracker{\n\t\ttimeSource:     timeSource,\n\t\tbucketInterval: bucketInterval,\n\t\tbuckets:        make([]bucket, numBuckets),\n\t}\n}\n\nfunc (r *rollingWindowQPSTracker) Start() {\n}\n\nfunc (r *rollingWindowQPSTracker) Stop() {\n}\n\nfunc (r *rollingWindowQPSTracker) getCurrentTimeIndex() int {\n\tnow := r.timeSource.Now()\n\treturn int(now.UnixNano() / int64(r.bucketInterval))\n}\n\nfunc (r *rollingWindowQPSTracker) ReportCounter(delta int64) {\n\tr.Lock()\n\tdefer r.Unlock()\n\tcurrentIndex := r.getCurrentTimeIndex()\n\tif currentIndex == r.timeIndex {\n\t\tr.counter += delta\n\t\treturn\n\t}\n\tr.buckets[r.timeIndex%len(r.buckets)] = bucket{\n\t\tcounter:   r.counter,\n\t\ttimeIndex: r.timeIndex,\n\t}\n\tr.timeIndex = currentIndex\n\tr.counter = delta\n}\n\nfunc (r *rollingWindowQPSTracker) QPS() float64 {\n\tr.RLock()\n\tdefer r.RUnlock()\n\tcurrentIndex := r.getCurrentTimeIndex()\n\ttotalCounter := int64(0)\n\tfor _, b := range r.buckets {\n\t\tif currentIndex-b.timeIndex <= len(r.buckets) {\n\t\t\ttotalCounter += b.counter\n\t\t}\n\t}\n\tif currentIndex != r.timeIndex && currentIndex-r.timeIndex <= len(r.buckets) {\n\t\ttotalCounter += r.counter\n\t}\n\treturn float64(totalCounter) / float64(r.bucketInterval) / float64(len(r.buckets)) * float64(time.Second)\n}\n"
  },
  {
    "path": "common/stats/stats_benchmark_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage stats\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\n// Benchmark the ReportCounter function to see how it handles frequent updates\nfunc BenchmarkReportCounter(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter with a smoothing factor and a 1 second bucket interval\n\treporter := NewEmaFixedWindowQPSTracker(timeSource, 0.5, time.Second, event.E{})\n\treporter.Start()\n\n\t// Run the benchmark for b.N iterations\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\treporter.ReportCounter(1)\n\t}\n\n\t// Stop the reporter after the benchmark\n\tb.StopTimer()\n\treporter.Stop()\n}\n\n// Benchmark the QPS calculation function under high load\nfunc BenchmarkQPS(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter\n\treporter := NewEmaFixedWindowQPSTracker(timeSource, 0.5, time.Second, event.E{})\n\treporter.Start()\n\n\t// Simulate a number of report updates before calling QPS\n\tfor i := 0; i < 1000; i++ {\n\t\treporter.ReportCounter(1)\n\t}\n\n\t// Benchmark QPS retrieval\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\t_ = reporter.QPS()\n\t}\n\n\t// Stop the reporter\n\tb.StopTimer()\n\treporter.Stop()\n}\n\n// Benchmark the full reporting loop, simulating a real-time system.\nfunc BenchmarkFullReport(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter\n\treporter := NewEmaFixedWindowQPSTracker(timeSource, 0.5, time.Millisecond*100, event.E{}) // 100ms bucket interval\n\treporter.Start()\n\n\tvar wg sync.WaitGroup\n\t// Number of goroutines for each task\n\tnumReporters := 10\n\tnumQPSQueries := 10\n\tb.ResetTimer()\n\n\tfor i := 0; i < numReporters; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < b.N; j++ {\n\t\t\t\t// Report random counter value (simulate workload)\n\t\t\t\treporter.ReportCounter(1)\n\t\t\t}\n\t\t}()\n\t}\n\tfor i := 0; i < numQPSQueries; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < b.N; j++ {\n\t\t\t\t// Query QPS value (simulate workload)\n\t\t\t\t_ = reporter.QPS()\n\t\t\t}\n\t\t}()\n\t}\n\twg.Wait()\n\t// Stop the reporter after the benchmark\n\tb.StopTimer()\n\treporter.Stop()\n}\n\nfunc BenchmarkReportCounterRollingWindow(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter with a smoothing factor and a 1 second bucket interval\n\treporter := NewRollingWindowQPSTracker(timeSource, time.Second, 10)\n\treporter.Start()\n\n\t// Run the benchmark for b.N iterations\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\treporter.ReportCounter(1)\n\t}\n\n\t// Stop the reporter after the benchmark\n\tb.StopTimer()\n\treporter.Stop()\n}\n\nfunc BenchmarkQPSRollingWindow(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter\n\treporter := NewRollingWindowQPSTracker(timeSource, time.Second, 10)\n\treporter.Start()\n\n\t// Simulate a number of report updates before calling QPS\n\tfor i := 0; i < 1000; i++ {\n\t\treporter.ReportCounter(1)\n\t}\n\n\t// Benchmark QPS retrieval\n\tb.ResetTimer()\n\tfor i := 0; i < b.N; i++ {\n\t\t_ = reporter.QPS()\n\t}\n\n\t// Stop the reporter\n\tb.StopTimer()\n\treporter.Stop()\n}\n\n// Benchmark the full reporting loop, simulating a real-time system.\nfunc BenchmarkFullReportRollingWindow(b *testing.B) {\n\ttimeSource := clock.NewRealTimeSource()\n\t// Initialize the QPS reporter\n\treporter := NewRollingWindowQPSTracker(timeSource, time.Millisecond*100, 10) // 100ms bucket interval\n\treporter.Start()\n\n\tvar wg sync.WaitGroup\n\t// Number of goroutines for each task\n\tnumReporters := 10\n\tnumQPSQueries := 10\n\tb.ResetTimer()\n\n\tfor i := 0; i < numReporters; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < b.N; j++ {\n\t\t\t\t// Report random counter value (simulate workload)\n\t\t\t\treporter.ReportCounter(1)\n\t\t\t}\n\t\t}()\n\t}\n\tfor i := 0; i < numQPSQueries; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < b.N; j++ {\n\t\t\t\t// Query QPS value (simulate workload)\n\t\t\t\t_ = reporter.QPS()\n\t\t\t}\n\t\t}()\n\t}\n\twg.Wait()\n\t// Stop the reporter after the benchmark\n\tb.StopTimer()\n\treporter.Stop()\n}\n"
  },
  {
    "path": "common/stats/stats_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage stats\n\nimport (\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\nconst floatResolution = 1e-6\n\nfunc TestEmaFixedWindowQPSTracker(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\texp := 0.4\n\tbucketInterval := time.Second\n\n\tr := NewEmaFixedWindowQPSTracker(timeSource, exp, bucketInterval, event.E{})\n\tr.Start()\n\tdefer r.Stop()\n\n\t// Test ReportCounter\n\tr.ReportCounter(10)\n\tr.ReportCounter(20)\n\n\tqps := r.QPS()\n\tassert.Equal(t, float64(0), qps)\n\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(bucketInterval)\n\ttime.Sleep(10 * time.Millisecond)\n\t// Test QPS\n\tqps = r.QPS()\n\texpectedQPS := float64(30) / (float64(bucketInterval) / float64(time.Second))\n\tassert.InDelta(t, expectedQPS, qps, floatResolution)\n\n\tr.ReportCounter(10)\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(bucketInterval)\n\ttime.Sleep(10 * time.Millisecond)\n\t// Test QPS\n\tqps = r.QPS()\n\texpectedQPS = float64(22) / (float64(bucketInterval) / float64(time.Second))\n\tassert.InDelta(t, expectedQPS, qps, floatResolution)\n}\n\nfunc TestEmaFixedWindowQPSTracker_Groups(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\texp := 0.5\n\tbucketInterval := time.Second\n\n\tr := NewEmaFixedWindowQPSTracker(timeSource, exp, bucketInterval, event.E{}).(*emaFixedWindowQPSTracker)\n\n\tr.ReportGroup(\"foo\", 10)\n\tr.ReportGroup(\"bar\", 20)\n\tr.report()\n\n\t// Test QPS\n\texpectedQPS := float64(30)\n\texpectedFoo := float64(10)\n\texpectedBar := float64(20)\n\tassert.InDelta(t, expectedQPS, r.QPS(), floatResolution)\n\tassert.InDelta(t, expectedFoo, r.GroupQPS(\"foo\"), floatResolution)\n\tassert.InDelta(t, expectedBar, r.GroupQPS(\"bar\"), floatResolution)\n\tassert.Equal(t, float64(0), r.GroupQPS(\"unknown\"))\n}\n\nfunc TestEmaFixedWindowQPSTracker_MultipleReportCycles(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\texp := 0.5\n\tbucketInterval := time.Second\n\n\tr := NewEmaFixedWindowQPSTracker(timeSource, exp, bucketInterval, event.E{}).(*emaFixedWindowQPSTracker)\n\n\tr.ReportGroup(\"foo\", 10)\n\tr.ReportGroup(\"bar\", 20)\n\tr.report()\n\tr.ReportGroup(\"foo\", 20)\n\tr.ReportGroup(\"bar\", 40)\n\tr.report()\n\n\t// Test QPS\n\texpectedQPS := float64(45)\n\texpectedFoo := float64(15)\n\texpectedBar := float64(30)\n\tassert.InDelta(t, expectedQPS, r.QPS(), floatResolution)\n\tassert.InDelta(t, expectedFoo, r.GroupQPS(\"foo\"), floatResolution)\n\tassert.InDelta(t, expectedBar, r.GroupQPS(\"bar\"), floatResolution)\n\tassert.Equal(t, float64(0), r.GroupQPS(\"unknown\"))\n}\n\nfunc TestRollingWindowQPSTracker(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\tbucketInterval := time.Second\n\n\tr := NewRollingWindowQPSTracker(timeSource, bucketInterval, 10)\n\tr.Start()\n\tdefer r.Stop()\n\n\tr.ReportCounter(10)\n\tqps := r.QPS()\n\tif qps != 0 {\n\t\tt.Errorf(\"QPS mismatch, expected: 0, got: %v\", qps)\n\t}\n\n\ttimeSource.Advance(bucketInterval)\n\n\tqps = r.QPS()\n\texpectedQPS := float64(10) / (float64(bucketInterval) * 10 / float64(time.Second))\n\tif math.Abs(qps-expectedQPS) > floatResolution {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n\n\tr.ReportCounter(20)\n\tqps = r.QPS()\n\tif qps != expectedQPS {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n\n\ttimeSource.Advance(bucketInterval)\n\n\tqps = r.QPS()\n\texpectedQPS = float64(30) / (float64(bucketInterval) * 10 / float64(time.Second))\n\tif math.Abs(qps-expectedQPS) > floatResolution {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n\n\tr.ReportCounter(100)\n\n\ttimeSource.Advance(8 * bucketInterval)\n\n\tqps = r.QPS()\n\texpectedQPS = float64(130) / (float64(bucketInterval) * 10 / float64(time.Second))\n\tif math.Abs(qps-expectedQPS) > floatResolution {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n\n\ttimeSource.Advance(bucketInterval)\n\tqps = r.QPS()\n\texpectedQPS = float64(120) / (float64(bucketInterval) * 10 / float64(time.Second))\n\tif math.Abs(qps-expectedQPS) > floatResolution {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n\n\ttimeSource.Advance(bucketInterval)\n\tqps = r.QPS()\n\texpectedQPS = float64(100) / (float64(bucketInterval) * 10 / float64(time.Second))\n\tif math.Abs(qps-expectedQPS) > floatResolution {\n\t\tt.Errorf(\"QPS mismatch, expected: %v, got: %v\", expectedQPS, qps)\n\t}\n}\n"
  },
  {
    "path": "common/syncmap/syncmap.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage syncmap\n\nimport \"sync\"\n\n// syncmap is a very simple type-safe locked map, with semantics similar to sync.Map,\n// but only supports inserting once and getting.\ntype (\n\tsyncmap[K comparable, V any] struct {\n\t\tmut  sync.Mutex\n\t\tdata map[K]V\n\t}\n\tSyncMap[K comparable, V any] interface {\n\t\tGet(key K) (value V, ok bool)\n\t\tPut(key K, value V) (inserted bool)\n\t}\n)\n\nfunc New[K comparable, V any]() SyncMap[K, V] {\n\treturn &syncmap[K, V]{\n\t\tdata: make(map[K]V),\n\t}\n}\n\nfunc (m *syncmap[K, V]) Get(key K) (value V, ok bool) {\n\tm.mut.Lock()\n\tdefer m.mut.Unlock()\n\tvalue, ok = m.data[key]\n\treturn value, ok\n}\n\nfunc (m *syncmap[K, V]) Put(key K, value V) (inserted bool) {\n\tm.mut.Lock()\n\tdefer m.mut.Unlock()\n\tif _, ok := m.data[key]; ok {\n\t\treturn false\n\t}\n\tm.data[key] = value\n\treturn true\n}\n"
  },
  {
    "path": "common/syncmap/syncmap_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage syncmap\n\nimport (\n\t\"strconv\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\n// exclusively tests for races and crashes\nfunc TestRace(t *testing.T) {\n\tt.Parallel()\n\n\tconst parallel = 100\n\tvar finish sync.WaitGroup\n\tvar start sync.WaitGroup\n\tfinish.Add(parallel)\n\tstart.Add(parallel + 1)\n\n\tm := New[string, int]()\n\tfor i := 0; i < parallel; i++ {\n\t\ti := i\n\t\tgo func() {\n\t\t\tdefer finish.Done()\n\t\t\tstart.Done()\n\t\t\tstart.Wait() // maximize contention\n\t\t\tkey := strconv.Itoa(i % 3)\n\t\t\tif i%2 == 0 {\n\t\t\t\t_ = m.Put(key, i)\n\t\t\t\t_, _ = m.Get(key)\n\t\t\t} else {\n\t\t\t\t// reverse order to thrash a bit harder\n\t\t\t\t_, _ = m.Get(key)\n\t\t\t\t_ = m.Put(key, i)\n\t\t\t}\n\t\t}()\n\t}\n\n\tstart.Done()\n\tfinish.Wait()\n}\n\n// checks get/put behavior.  it's somewhat different and not *expected* to change,\n// but should be totally fine to change to a general map API as long as existing uses are adapted.\nfunc TestBasics(t *testing.T) {\n\tt.Parallel()\n\n\tconst key = \"key\"\n\tconst orig = 5\n\tconst replacement = 10\n\trequire.NotEqual(t, orig, replacement)\n\n\tm := New[string, int]()\n\tval, ok := m.Get(key)\n\tassert.False(t, ok, \"should not get until a value is inserted\")\n\tassert.Zero(t, val, \"should get zero value if nothing inserted\")\n\n\tinserted := m.Put(key, orig)\n\tassert.True(t, inserted, \"should have inserted into empty map\")\n\n\tval, ok = m.Get(key)\n\tassert.True(t, ok, \"should successfully get after insert\")\n\tassert.Equal(t, orig, val, \"should get what was put\")\n\n\treplaced := m.Put(key, replacement)\n\tassert.False(t, replaced, \"should not replace existing values\")\n\n\tval, ok = m.Get(key)\n\tassert.True(t, ok, \"should still get\")\n\tassert.Equal(t, orig, val, \"should get original value, not replaced\")\n}\n"
  },
  {
    "path": "common/task/fifo_task_scheduler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype fifoTaskSchedulerImpl[T Task] struct {\n\tstatus       int32\n\tlogger       log.Logger\n\tmetricsScope metrics.Scope\n\toptions      *FIFOTaskSchedulerOptions\n\tdispatcherWG sync.WaitGroup\n\ttaskCh       chan T\n\tctx          context.Context\n\tcancel       context.CancelFunc\n\n\tprocessor Processor\n}\n\n// NewFIFOTaskScheduler creates a new FIFO task scheduler\n// it's an no-op implementation as it simply copy tasks from\n// one task channel to another task channel.\n// This scheduler is only for development purpose.\nfunc NewFIFOTaskScheduler[T Task](\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\toptions *FIFOTaskSchedulerOptions,\n) Scheduler[T] {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &fifoTaskSchedulerImpl[T]{\n\t\tstatus:       common.DaemonStatusInitialized,\n\t\tlogger:       logger,\n\t\tmetricsScope: metricsClient.Scope(metrics.TaskSchedulerScope),\n\t\toptions:      options,\n\t\ttaskCh:       make(chan T, options.QueueSize),\n\t\tctx:          ctx,\n\t\tcancel:       cancel,\n\t\tprocessor: NewParallelTaskProcessor(\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\t&ParallelTaskProcessorOptions{\n\t\t\t\tQueueSize:   options.QueueSize,\n\t\t\t\tWorkerCount: options.WorkerCount,\n\t\t\t\tRetryPolicy: options.RetryPolicy,\n\t\t\t},\n\t\t),\n\t}\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) Start() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tf.processor.Start()\n\n\tf.dispatcherWG.Add(f.options.DispatcherCount)\n\tfor i := 0; i != f.options.DispatcherCount; i++ {\n\t\tgo f.dispatcher()\n\t}\n\n\tf.logger.Info(\"FIFO task scheduler started.\")\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) Stop() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tf.cancel()\n\n\tf.processor.Stop()\n\n\tf.drainAndNackTasks()\n\n\tif success := common.AwaitWaitGroup(&f.dispatcherWG, time.Minute); !success {\n\t\tf.logger.Warn(\"FIFO task scheduler timedout on shutdown.\")\n\t}\n\n\tf.logger.Info(\"FIFO task scheduler shutdown.\")\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) Submit(task T) error {\n\tf.metricsScope.IncCounter(metrics.ParallelTaskSubmitRequest)\n\tsw := f.metricsScope.StartTimer(metrics.ParallelTaskSubmitLatency)\n\tdefer sw.Stop()\n\n\tif f.isStopped() {\n\t\treturn ErrTaskSchedulerClosed\n\t}\n\n\tselect {\n\tcase f.taskCh <- task:\n\t\tif f.isStopped() {\n\t\t\tf.drainAndNackTasks()\n\t\t}\n\t\treturn nil\n\tcase <-f.ctx.Done():\n\t\treturn ErrTaskSchedulerClosed\n\t}\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) TrySubmit(task T) (bool, error) {\n\tif f.isStopped() {\n\t\treturn false, ErrTaskSchedulerClosed\n\t}\n\n\tselect {\n\tcase f.taskCh <- task:\n\t\tf.metricsScope.IncCounter(metrics.ParallelTaskSubmitRequest)\n\t\tif f.isStopped() {\n\t\t\tf.drainAndNackTasks()\n\t\t}\n\t\treturn true, nil\n\tcase <-f.ctx.Done():\n\t\treturn false, ErrTaskSchedulerClosed\n\tdefault:\n\t\treturn false, nil\n\t}\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) dispatcher() {\n\tdefer f.dispatcherWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase task := <-f.taskCh:\n\t\t\tif err := f.processor.Submit(task); err != nil {\n\t\t\t\tf.logger.Error(\"failed to submit task to processor\", tag.Error(err))\n\t\t\t\ttask.Nack(err)\n\t\t\t}\n\t\tcase <-f.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) isStopped() bool {\n\treturn atomic.LoadInt32(&f.status) == common.DaemonStatusStopped\n}\n\nfunc (f *fifoTaskSchedulerImpl[T]) drainAndNackTasks() {\n\tfor {\n\t\tselect {\n\t\tcase task := <-f.taskCh:\n\t\t\ttask.Nack(nil)\n\t\tdefault:\n\t\t\treturn\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/task/fifo_task_scheduler_options.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\n// FIFOTaskSchedulerOptions configs FIFO task scheduler\ntype FIFOTaskSchedulerOptions struct {\n\tQueueSize       int\n\tWorkerCount     dynamicproperties.IntPropertyFn\n\tDispatcherCount int\n\tRetryPolicy     backoff.RetryPolicy\n}\n\nfunc (o *FIFOTaskSchedulerOptions) String() string {\n\treturn fmt.Sprintf(\"{QueueSize: %v, WorkerCount: %v, DispatcherCount: %v}\", o.QueueSize, o.WorkerCount(), o.DispatcherCount)\n}\n"
  },
  {
    "path": "common/task/fifo_task_scheduler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tfifoTaskSchedulerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tcontroller    *gomock.Controller\n\t\tmockProcessor *MockProcessor\n\n\t\tqueueSize int\n\n\t\tscheduler *fifoTaskSchedulerImpl[PriorityTask]\n\t}\n)\n\nfunc TestFIFOTaskSchedulerSuite(t *testing.T) {\n\ts := new(fifoTaskSchedulerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *fifoTaskSchedulerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockProcessor = NewMockProcessor(s.controller)\n\n\ts.queueSize = 2\n\ts.scheduler = NewFIFOTaskScheduler[PriorityTask](\n\t\ttestlogger.New(s.Suite.T()),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\t&FIFOTaskSchedulerOptions{\n\t\t\tQueueSize:       s.queueSize,\n\t\t\tWorkerCount:     dynamicproperties.GetIntPropertyFn(1),\n\t\t\tDispatcherCount: 1,\n\t\t\tRetryPolicy:     backoff.NewExponentialRetryPolicy(time.Millisecond),\n\t\t},\n\t).(*fifoTaskSchedulerImpl[PriorityTask])\n}\n\nfunc (s *fifoTaskSchedulerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *fifoTaskSchedulerSuite) TestFIFO() {\n\tnumTasks := 5\n\ttasks := []PriorityTask{}\n\tvar taskWG sync.WaitGroup\n\n\tcalls := []any{\n\t\ts.mockProcessor.EXPECT().Start(),\n\t}\n\tmockFn := func(_ Task) error {\n\t\ttaskWG.Done()\n\t\treturn nil\n\t}\n\tfor i := 0; i != numTasks; i++ {\n\t\tmockTask := NewMockPriorityTask(s.controller)\n\t\ttasks = append(tasks, mockTask)\n\t\ttaskWG.Add(1)\n\t\tcalls = append(calls, s.mockProcessor.EXPECT().Submit(newMockPriorityTaskMatcher(mockTask)).DoAndReturn(mockFn))\n\t}\n\tcalls = append(calls, s.mockProcessor.EXPECT().Stop())\n\tgomock.InOrder(calls...)\n\n\ts.scheduler.processor = s.mockProcessor\n\ts.scheduler.Start()\n\tfor _, task := range tasks {\n\t\ts.NoError(s.scheduler.Submit(task))\n\t}\n\ttaskWG.Wait()\n\ts.scheduler.Stop()\n}\n\nfunc (s *fifoTaskSchedulerSuite) TestTrySubmit() {\n\tfor i := 0; i != s.queueSize; i++ {\n\t\tmockTask := NewMockPriorityTask(s.controller)\n\t\tsubmitted, err := s.scheduler.TrySubmit(mockTask)\n\t\ts.NoError(err)\n\t\ts.True(submitted)\n\t}\n\n\t// now the queue is full, submit one more task, should be non-blocking\n\tmockTask := NewMockPriorityTask(s.controller)\n\tsubmitted, err := s.scheduler.TrySubmit(mockTask)\n\ts.NoError(err)\n\ts.False(submitted)\n}\n\nfunc (s *fifoTaskSchedulerSuite) TestSchedulerContract() {\n\ttestSchedulerContract(s.Assertions, s.controller, s.scheduler, nil)\n}\n"
  },
  {
    "path": "common/task/hierarchical_weighted_round_robin_task_pool.go",
    "content": "package task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nvar (\n\terrWeightedKeyWeightMustBeGreaterThanZero = errors.New(\"weight must be greater than 0\")\n)\n\ntype HierarchicalWeightedRoundRobinTaskPoolOptions[K comparable, T Task] struct {\n\tBufferSize           int\n\tTaskToWeightedKeysFn func(T) []WeightedKey[K]\n}\n\ntype hierarchicalWeightedRoundRobinTaskPoolImpl[K comparable, T Task] struct {\n\tsync.Mutex\n\tstatus     int32\n\troot       *iwrrNode[K, T]\n\tbufferSize int\n\tctx        context.Context\n\tcancel     context.CancelFunc\n\toptions    *HierarchicalWeightedRoundRobinTaskPoolOptions[K, T]\n\tlogger     log.Logger\n\ttimeSource clock.TimeSource\n\twg         sync.WaitGroup\n\n\tdoCleanupFn func(now time.Time, ttl time.Duration)\n}\n\n// newHierarchicalWeightedRoundRobinTaskPool creates a new hierarchical WRR task pool\nfunc newHierarchicalWeightedRoundRobinTaskPool[K comparable, T Task](\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n\toptions *HierarchicalWeightedRoundRobinTaskPoolOptions[K, T],\n) *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T] {\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tpool := &hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]{\n\t\tstatus:     common.DaemonStatusInitialized,\n\t\troot:       newiwrrNode[K, T](options.BufferSize),\n\t\tbufferSize: options.BufferSize,\n\t\tctx:        ctx,\n\t\tcancel:     cancel,\n\t\toptions:    options,\n\t\tlogger:     logger,\n\t\ttimeSource: timeSource,\n\t}\n\tpool.doCleanupFn = pool.doCleanup\n\n\treturn pool\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.wg.Add(1)\n\tgo p.cleanupLoop()\n\n\tp.logger.Info(\"Hierarchical weighted round robin task pool started.\")\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tp.cancel()\n\tp.wg.Wait()\n\n\tp.logger.Info(\"Hierarchical weighted round robin task pool stopped.\")\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) cleanupLoop() {\n\tdefer p.wg.Done()\n\n\tticker := p.timeSource.NewTicker(time.Duration((defaultIdleChannelTTLInSeconds / 2)) * time.Second)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tnow := p.timeSource.Now()\n\t\t\tttl := time.Duration(defaultIdleChannelTTLInSeconds) * time.Second\n\t\t\tp.doCleanupFn(now, ttl)\n\t\tcase <-p.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) doCleanup(now time.Time, ttl time.Duration) {\n\tp.root.cleanup(now, ttl)\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) Enqueue(task T) error {\n\tweightedKeys := p.options.TaskToWeightedKeysFn(task)\n\tif err := verifyWeightedKeys(weightedKeys); err != nil {\n\t\treturn err\n\t}\n\tvar err error\n\tp.root.c.IncRef()\n\tdefer p.root.c.DecRef()\n\tp.root.executeAtPath(weightedKeys, p.bufferSize, func(c *TTLChannel[T]) int64 {\n\t\tselect {\n\t\tcase c.Chan() <- task:\n\t\t\tc.UpdateLastWriteTime(p.timeSource.Now())\n\t\t\treturn 1\n\t\tcase <-p.ctx.Done():\n\t\t\terr = p.ctx.Err()\n\t\t\treturn 0\n\t\t}\n\t})\n\treturn err\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) TryEnqueue(task T) (bool, error) {\n\tweightedKeys := p.options.TaskToWeightedKeysFn(task)\n\tif err := verifyWeightedKeys(weightedKeys); err != nil {\n\t\treturn false, err\n\t}\n\tvar err error\n\tp.root.c.IncRef()\n\tdefer p.root.c.DecRef()\n\tdelta := p.root.executeAtPath(weightedKeys, p.bufferSize, func(c *TTLChannel[T]) int64 {\n\t\tselect {\n\t\tcase c.Chan() <- task:\n\t\t\tc.UpdateLastWriteTime(p.timeSource.Now())\n\t\t\treturn 1\n\t\tcase <-p.ctx.Done():\n\t\t\terr = p.ctx.Err()\n\t\t\treturn 0\n\t\tdefault:\n\t\t\treturn 0\n\t\t}\n\t})\n\treturn delta > 0, err\n}\n\nfunc (p *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]) TryDequeue() (T, bool) {\n\tvar zero T\n\titem, ok := p.root.tryGetNextItem()\n\tif !ok {\n\t\treturn zero, false\n\t}\n\treturn item, true\n}\n\nfunc verifyWeightedKeys[K comparable](weightedKeys []WeightedKey[K]) error {\n\tfor _, weightedKey := range weightedKeys {\n\t\tif weightedKey.Weight <= 0 {\n\t\t\treturn errWeightedKeyWeightMustBeGreaterThanZero\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/task/hierarchical_weighted_round_robin_task_pool_test.go",
    "content": "package task\n\nimport (\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// testPriorityTask is a simple test implementation of PriorityTask\ntype testPriorityTask struct {\n\tdomain   string\n\ttasklist string\n\ttenant   string\n\tstate    State\n}\n\nfunc (t *testPriorityTask) Execute() error            { return nil }\nfunc (t *testPriorityTask) HandleErr(err error) error { return err }\nfunc (t *testPriorityTask) RetryErr(err error) bool   { return false }\nfunc (t *testPriorityTask) Ack()                      { t.state = TaskStateAcked }\nfunc (t *testPriorityTask) Nack(err error)            { t.state = TaskStatePending }\nfunc (t *testPriorityTask) Cancel()                   { t.state = TaskStateCanceled }\nfunc (t *testPriorityTask) State() State              { return t.state }\nfunc (t *testPriorityTask) Priority() int             { return 0 }\nfunc (t *testPriorityTask) SetPriority(p int)         {}\n\nfunc TestHierarchicalWRRTaskPool_SingleLevel_IWRROrdering(t *testing.T) {\n\t// Define domain to weight mapping\n\tdomainWeights := map[string]int{\n\t\t\"domain0\": 5,\n\t\t\"domain1\": 3,\n\t\t\"domain2\": 1,\n\t}\n\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 20,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\t// Single-level hierarchy: tasks grouped by domain with different weights\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Create tasks with weights [5, 3, 1]\n\t// Domain0 (weight 5): 5 tasks\n\t// Domain1 (weight 3): 3 tasks\n\t// Domain2 (weight 1): 1 task\n\tvar tasks []*testPriorityTask\n\tfor i := 0; i < 5; i++ {\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain0\"})\n\t}\n\tfor i := 0; i < 3; i++ {\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\"})\n\t}\n\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\"})\n\n\t// Enqueue tasks in parallel using randomly chosen Enqueue or TryEnqueue\n\tvar wg sync.WaitGroup\n\tfor _, task := range tasks {\n\t\twg.Add(1)\n\t\tgo func(t *testPriorityTask) {\n\t\t\tdefer wg.Done()\n\t\t\t// Randomly choose between Enqueue and TryEnqueue\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t_ = pool.Enqueue(t)\n\t\t\t} else {\n\t\t\t\t_, _ = pool.TryEnqueue(t)\n\t\t\t}\n\t\t}(task)\n\t}\n\twg.Wait()\n\n\t// IWRR pattern for weights [5, 3, 1]: domain0, domain0, domain0, domain1, domain0, domain1, domain0, domain1, domain2\n\t// Expected domain sequence: [0, 0, 0, 1, 0, 1, 0, 1, 2]\n\texpectedDomainPattern := []string{\"domain0\", \"domain0\", \"domain0\", \"domain1\", \"domain0\", \"domain1\", \"domain0\", \"domain1\", \"domain2\"}\n\n\tvar actualDomainSequence []string\n\tfor {\n\t\ttask, ok := pool.TryDequeue()\n\t\tif !ok {\n\t\t\tbreak\n\t\t}\n\t\tactualDomainSequence = append(actualDomainSequence, task.domain)\n\t}\n\n\tassert.Equal(t, expectedDomainPattern, actualDomainSequence)\n}\n\nfunc TestHierarchicalWRRTaskPool_TwoLevel_IWRROrdering(t *testing.T) {\n\t// Define domain to weight mapping - different weights\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t\t\"domain2\": 2,\n\t\t\"domain3\": 1,\n\t}\n\n\t// Define tasklist to weight mapping\n\ttasklistWeights := map[string]int{\n\t\t\"tasklist1A\": 2,\n\t\t\"tasklist1B\": 1,\n\t\t\"tasklist2A\": 3,\n\t\t\"tasklist2B\": 1,\n\t\t\"tasklist3A\": 2,\n\t\t\"tasklist3B\": 1,\n\t}\n\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 20,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\t// Two-level hierarchy: domain (level 1) -> tasklist (level 2)\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t\t{Key: task.tasklist, Weight: tasklistWeights[task.tasklist]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Create enough tasks for each domain to complete at least one full tasklist IWRR cycle\n\t// Domain1 tasklist IWRR [2, 1]: [tasklist1A, tasklist1A, tasklist1B] = 3 tasks\n\t// Domain2 tasklist IWRR [3, 1]: [tasklist2A, tasklist2A, tasklist2A, tasklist2B] = 4 tasks\n\t// Domain3 tasklist IWRR [2, 1]: [tasklist3A, tasklist3A, tasklist3B] = 3 tasks\n\ttasks := []*testPriorityTask{\n\t\t// Domain1 tasks (3 tasks for one complete tasklist IWRR)\n\t\t&testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\"},\n\t\t&testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\"},\n\t\t&testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1B\"},\n\t\t// Domain2 tasks (4 tasks for one complete tasklist IWRR)\n\t\t&testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\"},\n\t\t&testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\"},\n\t\t&testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\"},\n\t\t&testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2B\"},\n\t\t// Domain3 tasks (3 tasks for one complete tasklist IWRR)\n\t\t&testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\"},\n\t\t&testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\"},\n\t\t&testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3B\"},\n\t}\n\n\t// Enqueue all tasks in parallel using randomly chosen Enqueue or TryEnqueue\n\tvar wg sync.WaitGroup\n\tfor _, task := range tasks {\n\t\twg.Add(1)\n\t\tgo func(t *testPriorityTask) {\n\t\t\tdefer wg.Done()\n\t\t\t// Randomly choose between Enqueue and TryEnqueue\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t_ = pool.Enqueue(t)\n\t\t\t} else {\n\t\t\t\t_, _ = pool.TryEnqueue(t)\n\t\t\t}\n\t\t}(task)\n\t}\n\twg.Wait()\n\n\t// Domain IWRR pattern for weights [3, 2, 1]: [domain1, domain1, domain2, domain1, domain2, domain3]\n\t// When we dequeue, we follow this domain pattern, and within each domain we follow its tasklist IWRR\n\t//\n\t// Expected dequeue pattern:\n\t// 1. domain1 (1st visit) -> tasklist1A (position 0 in tasklist IWRR)\n\t// 2. domain1 (2nd visit) -> tasklist1A (position 1 in tasklist IWRR)\n\t// 3. domain2 (1st visit) -> tasklist2A (position 0 in tasklist IWRR)\n\t// 4. domain1 (3rd visit) -> tasklist1B (position 2 in tasklist IWRR)\n\t// 5. domain2 (2nd visit) -> tasklist2A (position 1 in tasklist IWRR)\n\t// 6. domain3 (1st visit) -> tasklist3A (position 0 in tasklist IWRR)\n\t// Now we loop back to domain1, but it's exhausted, so we skip it\n\t// 7. domain2 (3rd visit) -> tasklist2A (position 2 in tasklist IWRR)\n\t// 8. domain2 (4th visit) -> tasklist2B (position 3 in tasklist IWRR)\n\t// 9. domain3 (2nd visit) -> tasklist3A (position 1 in tasklist IWRR)\n\t// 10. domain3 (3rd visit) -> tasklist3B (position 2 in tasklist IWRR)\n\ttype expectedTask struct {\n\t\tdomain   string\n\t\ttasklist string\n\t}\n\texpectedPattern := []expectedTask{\n\t\t{\"domain1\", \"tasklist1A\"}, // domain IWRR pos 0, tasklist IWRR pos 0\n\t\t{\"domain1\", \"tasklist1A\"}, // domain IWRR pos 1, tasklist IWRR pos 1\n\t\t{\"domain2\", \"tasklist2A\"}, // domain IWRR pos 2, tasklist IWRR pos 0\n\t\t{\"domain1\", \"tasklist1B\"}, // domain IWRR pos 3, tasklist IWRR pos 2\n\t\t{\"domain2\", \"tasklist2A\"}, // domain IWRR pos 4, tasklist IWRR pos 1\n\t\t{\"domain3\", \"tasklist3A\"}, // domain IWRR pos 5, tasklist IWRR pos 0\n\t\t{\"domain2\", \"tasklist2A\"}, // domain1 exhausted, skip to domain2, tasklist IWRR pos 2\n\t\t{\"domain2\", \"tasklist2B\"}, // continue domain2, tasklist IWRR pos 3\n\t\t{\"domain3\", \"tasklist3A\"}, // domain3, tasklist IWRR pos 1\n\t\t{\"domain3\", \"tasklist3B\"}, // domain3, tasklist IWRR pos 2\n\t}\n\n\t// Dequeue all tasks and verify the order\n\tvar actualPattern []expectedTask\n\tfor {\n\t\ttask, ok := pool.TryDequeue()\n\t\tif !ok {\n\t\t\tbreak\n\t\t}\n\t\tactualPattern = append(actualPattern, expectedTask{\n\t\t\tdomain:   task.domain,\n\t\t\ttasklist: task.tasklist,\n\t\t})\n\t}\n\n\t// Verify exact ordering\n\trequire.Equal(t, len(expectedPattern), len(actualPattern), \"should dequeue all tasks\")\n\tfor i, expected := range expectedPattern {\n\t\tassert.Equal(t, expected.domain, actualPattern[i].domain, \"task %d domain mismatch\", i)\n\t\tassert.Equal(t, expected.tasklist, actualPattern[i].tasklist, \"task %d tasklist mismatch\", i)\n\t}\n}\n\nfunc TestHierarchicalWRRTaskPool_TwoLevel_LargeScale_IWRROrdering(t *testing.T) {\n\t// Define domain to weight mapping\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 5,\n\t\t\"domain2\": 3,\n\t\t\"domain3\": 2,\n\t}\n\n\t// Define tasklist to weight mapping\n\ttasklistWeights := map[string]int{\n\t\t\"tasklist1A\": 3,\n\t\t\"tasklist1B\": 2,\n\t\t\"tasklist1C\": 1,\n\t\t\"tasklist2A\": 4,\n\t\t\"tasklist2B\": 2,\n\t\t\"tasklist3A\": 3,\n\t\t\"tasklist3B\": 1,\n\t}\n\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 100,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t\t{Key: task.tasklist, Weight: tasklistWeights[task.tasklist]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Create a large number of tasks\n\t// Domain1: 30 tasks (5 cycles of tasklist IWRR: [3,2,1] = 6 tasks per cycle)\n\t// Domain2: 30 tasks (5 cycles of tasklist IWRR: [4,2] = 6 tasks per cycle)\n\t// Domain3: 20 tasks (5 cycles of tasklist IWRR: [3,1] = 4 tasks per cycle)\n\tvar tasks []*testPriorityTask\n\n\t// Domain1: 5 cycles of [tasklist1A(3), tasklist1B(2), tasklist1C(1)]\n\tfor cycle := 0; cycle < 5; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1B\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1C\"})\n\t}\n\n\t// Domain2: 5 cycles of [tasklist2A(4), tasklist2B(2)]\n\tfor cycle := 0; cycle < 5; cycle++ {\n\t\tfor i := 0; i < 4; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2B\"})\n\t\t}\n\t}\n\n\t// Domain3: 5 cycles of [tasklist3A(3), tasklist3B(1)]\n\tfor cycle := 0; cycle < 5; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3B\"})\n\t}\n\n\t// Enqueue all tasks in parallel using randomly chosen Enqueue or TryEnqueue\n\tvar wg sync.WaitGroup\n\tfor _, task := range tasks {\n\t\twg.Add(1)\n\t\tgo func(t *testPriorityTask) {\n\t\t\tdefer wg.Done()\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t_ = pool.Enqueue(t)\n\t\t\t} else {\n\t\t\t\t_, _ = pool.TryEnqueue(t)\n\t\t\t}\n\t\t}(task)\n\t}\n\twg.Wait()\n\n\t// Generate expected patterns by level\n\t// Level 1 (domain): IWRR for weights [5, 3, 2]\n\t// Round 4: domain1 (weight 5 > 4)\n\t// Round 3: domain1 (weight 5 > 3)\n\t// Round 2: domain1, domain2 (weights 5,3 > 2)\n\t// Round 1: domain1, domain2, domain3 (weights 5,3,2 > 1)\n\t// Round 0: domain1, domain2, domain3 (weights 5,3,2 > 0)\n\t// Pattern: [domain1, domain1, domain1, domain2, domain1, domain2, domain3, domain1, domain2, domain3]\n\texpectedDomainPattern := []string{\n\t\t\"domain1\", \"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\", \"domain1\", \"domain2\", \"domain3\",\n\t}\n\n\t// Level 2 (tasklist per domain):\n\t// domain1 with weights [3, 2, 1]: [tasklist1A, tasklist1A, tasklist1B, tasklist1A, tasklist1B, tasklist1C]\n\t// domain2 with weights [4, 2]: [tasklist2A, tasklist2A, tasklist2A, tasklist2B, tasklist2A, tasklist2B]\n\t// domain3 with weights [3, 1]: [tasklist3A, tasklist3A, tasklist3A, tasklist3B]\n\texpectedTasklistPatterns := map[string][]string{\n\t\t\"domain1\": {\"tasklist1A\", \"tasklist1A\", \"tasklist1B\", \"tasklist1A\", \"tasklist1B\", \"tasklist1C\"},\n\t\t\"domain2\": {\"tasklist2A\", \"tasklist2A\", \"tasklist2A\", \"tasklist2B\", \"tasklist2A\", \"tasklist2B\"},\n\t\t\"domain3\": {\"tasklist3A\", \"tasklist3A\", \"tasklist3A\", \"tasklist3B\"},\n\t}\n\n\t// Dequeue all tasks\n\tvar dequeuedDomains []string\n\tdequeuedTasklistsByDomain := make(map[string][]string)\n\tfor {\n\t\ttask, ok := pool.TryDequeue()\n\t\tif !ok {\n\t\t\tbreak\n\t\t}\n\t\tdequeuedDomains = append(dequeuedDomains, task.domain)\n\t\tdequeuedTasklistsByDomain[task.domain] = append(\n\t\t\tdequeuedTasklistsByDomain[task.domain],\n\t\t\ttask.tasklist,\n\t\t)\n\t}\n\n\t// Verify we got all 80 tasks\n\trequire.Equal(t, 80, len(dequeuedDomains), \"should dequeue all 80 tasks\")\n\n\t// Verify level 1 (domain) pattern for the first cycle (first 10 tasks)\n\t// After that, domains may exhaust at different rates, so we only verify the first full cycle\n\tfor i := 0; i < len(expectedDomainPattern) && i < len(dequeuedDomains); i++ {\n\t\tassert.Equal(t, expectedDomainPattern[i], dequeuedDomains[i],\n\t\t\t\"domain mismatch at position %d in first cycle\", i)\n\t}\n\n\t// Verify level 2 (tasklist) pattern for each domain\n\t// Each domain should follow its own tasklist IWRR pattern consistently\n\tfor domain, dequeuedTasklists := range dequeuedTasklistsByDomain {\n\t\texpectedPattern := expectedTasklistPatterns[domain]\n\t\tfor i := 0; i < len(dequeuedTasklists); i++ {\n\t\t\texpectedTasklist := expectedPattern[i%len(expectedPattern)]\n\t\t\tassert.Equal(t, expectedTasklist, dequeuedTasklists[i],\n\t\t\t\t\"tasklist mismatch for domain %s at position %d\", domain, i)\n\t\t}\n\t}\n}\n\nfunc TestHierarchicalWRRTaskPool_ThreeLevel_LargeScale_IWRROrdering(t *testing.T) {\n\t// Define domain to weight mapping\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t\t\"domain2\": 2,\n\t\t\"domain3\": 1,\n\t}\n\n\t// Define tasklist to weight mapping\n\ttasklistWeights := map[string]int{\n\t\t\"tasklist1A\": 3,\n\t\t\"tasklist1B\": 1,\n\t\t\"tasklist2A\": 2,\n\t\t\"tasklist2B\": 1,\n\t\t\"tasklist3A\": 2,\n\t\t\"tasklist3B\": 1,\n\t}\n\n\t// Define tenant to weight mapping\n\ttenantWeights := map[string]int{\n\t\t\"tenant1\": 3,\n\t\t\"tenant2\": 2,\n\t\t\"tenant3\": 1,\n\t}\n\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 200,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t\t{Key: task.tasklist, Weight: tasklistWeights[task.tasklist]},\n\t\t\t\t\t{Key: task.tenant, Weight: tenantWeights[task.tenant]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Create a large number of tasks across 3 levels\n\t// Each (domain, tasklist) combination gets multiple tenant cycles\n\t// Tenant IWRR for weights [3, 2, 1]: 6 tasks per cycle\n\tvar tasks []*testPriorityTask\n\n\t// Domain1 - tasklist1A (weight 3) - 3 tenant cycles = 18 tasks\n\tfor cycle := 0; cycle < 3; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1A\", tenant: \"tenant3\"})\n\t}\n\n\t// Domain1 - tasklist1B (weight 1) - 3 tenant cycles = 18 tasks\n\tfor cycle := 0; cycle < 3; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1B\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1B\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain1\", tasklist: \"tasklist1B\", tenant: \"tenant3\"})\n\t}\n\n\t// Domain2 - tasklist2A (weight 2) - 2 tenant cycles = 12 tasks\n\tfor cycle := 0; cycle < 2; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2A\", tenant: \"tenant3\"})\n\t}\n\n\t// Domain2 - tasklist2B (weight 1) - 2 tenant cycles = 12 tasks\n\tfor cycle := 0; cycle < 2; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2B\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2B\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain2\", tasklist: \"tasklist2B\", tenant: \"tenant3\"})\n\t}\n\n\t// Domain3 - tasklist3A (weight 2) - 2 tenant cycles = 12 tasks\n\tfor cycle := 0; cycle < 2; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3A\", tenant: \"tenant3\"})\n\t}\n\n\t// Domain3 - tasklist3B (weight 1) - 2 tenant cycles = 12 tasks\n\tfor cycle := 0; cycle < 2; cycle++ {\n\t\tfor i := 0; i < 3; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3B\", tenant: \"tenant1\"})\n\t\t}\n\t\tfor i := 0; i < 2; i++ {\n\t\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3B\", tenant: \"tenant2\"})\n\t\t}\n\t\ttasks = append(tasks, &testPriorityTask{domain: \"domain3\", tasklist: \"tasklist3B\", tenant: \"tenant3\"})\n\t}\n\n\t// Total: 18+18+12+12+12+12 = 84 tasks\n\n\t// Enqueue all tasks in parallel using randomly chosen Enqueue or TryEnqueue\n\tvar wg sync.WaitGroup\n\tfor _, task := range tasks {\n\t\twg.Add(1)\n\t\tgo func(t *testPriorityTask) {\n\t\t\tdefer wg.Done()\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\t_ = pool.Enqueue(t)\n\t\t\t} else {\n\t\t\t\t_, _ = pool.TryEnqueue(t)\n\t\t\t}\n\t\t}(task)\n\t}\n\twg.Wait()\n\n\t// Dequeue all tasks\n\ttype taskKey struct {\n\t\tdomain   string\n\t\ttasklist string\n\t}\n\tvar dequeuedDomains []string\n\tdequeuedTasklistsByDomain := make(map[string][]string)\n\tdequeuedTenantsByTasklist := make(map[taskKey][]string)\n\n\tfor {\n\t\ttask, ok := pool.TryDequeue()\n\t\tif !ok {\n\t\t\tbreak\n\t\t}\n\t\tdequeuedDomains = append(dequeuedDomains, task.domain)\n\t\tdequeuedTasklistsByDomain[task.domain] = append(\n\t\t\tdequeuedTasklistsByDomain[task.domain],\n\t\t\ttask.tasklist,\n\t\t)\n\t\tkey := taskKey{domain: task.domain, tasklist: task.tasklist}\n\t\tdequeuedTenantsByTasklist[key] = append(\n\t\t\tdequeuedTenantsByTasklist[key],\n\t\t\ttask.tenant,\n\t\t)\n\t}\n\n\t// Verify we got all 84 tasks\n\trequire.Equal(t, 84, len(dequeuedDomains), \"should dequeue all 84 tasks\")\n\n\t// Build expected domain sequence manually with exhaustion tracking\n\t// Domain task counts: domain1=36, domain2=24, domain3=24\n\t// Pattern: [domain1, domain1, domain2, domain1, domain2, domain3] repeats 12 times (72 tasks)\n\t// After 12 cycles: domain1=36 (exhausted), domain2=24 (exhausted), domain3=12\n\t// Then domain3 continues for 12 more tasks\n\texpectedDomainSequence := []string{\n\t\t// 12 full cycles\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t\"domain1\", \"domain1\", \"domain2\", \"domain1\", \"domain2\", \"domain3\",\n\t\t// domain1 and domain2 exhausted, domain3 continues for 12 more\n\t\t\"domain3\", \"domain3\", \"domain3\", \"domain3\", \"domain3\", \"domain3\",\n\t\t\"domain3\", \"domain3\", \"domain3\", \"domain3\", \"domain3\", \"domain3\",\n\t}\n\n\t// Verify full domain sequence\n\trequire.Equal(t, 84, len(expectedDomainSequence), \"expected domain sequence length\")\n\trequire.Equal(t, len(expectedDomainSequence), len(dequeuedDomains),\n\t\t\"domain sequence length mismatch\")\n\tfor i := 0; i < len(expectedDomainSequence); i++ {\n\t\tassert.Equal(t, expectedDomainSequence[i], dequeuedDomains[i],\n\t\t\t\"domain mismatch at position %d\", i)\n\t}\n\n\t// Build expected tasklist sequences manually for each domain\n\texpectedTasklistSequences := map[string][]string{\n\t\t// Domain1: tasklist1A=18, tasklist1B=18\n\t\t// Pattern: [tasklist1A, tasklist1A, tasklist1A, tasklist1B] repeats 6 times (24 tasks)\n\t\t// After 6 cycles: tasklist1A=18 (exhausted), tasklist1B=6\n\t\t// Then tasklist1B continues for 12 more tasks\n\t\t\"domain1\": {\n\t\t\t// 6 full cycles\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t\"tasklist1A\", \"tasklist1A\", \"tasklist1A\", \"tasklist1B\",\n\t\t\t// tasklist1A exhausted, tasklist1B continues for 12 more\n\t\t\t\"tasklist1B\", \"tasklist1B\", \"tasklist1B\", \"tasklist1B\",\n\t\t\t\"tasklist1B\", \"tasklist1B\", \"tasklist1B\", \"tasklist1B\",\n\t\t\t\"tasklist1B\", \"tasklist1B\", \"tasklist1B\", \"tasklist1B\",\n\t\t},\n\t\t// Domain2: tasklist2A=12, tasklist2B=12\n\t\t// Pattern: [tasklist2A, tasklist2A, tasklist2B] repeats 6 times (18 tasks)\n\t\t// After 6 cycles: tasklist2A=12 (exhausted), tasklist2B=6\n\t\t// Then tasklist2B continues for 6 more tasks\n\t\t\"domain2\": {\n\t\t\t// 6 full cycles\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t\"tasklist2A\", \"tasklist2A\", \"tasklist2B\",\n\t\t\t// tasklist2A exhausted, tasklist2B continues for 6 more\n\t\t\t\"tasklist2B\", \"tasklist2B\", \"tasklist2B\",\n\t\t\t\"tasklist2B\", \"tasklist2B\", \"tasklist2B\",\n\t\t},\n\t\t// Domain3: tasklist3A=12, tasklist3B=12\n\t\t// Pattern: [tasklist3A, tasklist3A, tasklist3B] repeats 6 times (18 tasks)\n\t\t// After 6 cycles: tasklist3A=12 (exhausted), tasklist3B=6\n\t\t// Then tasklist3B continues for 6 more tasks\n\t\t\"domain3\": {\n\t\t\t// 6 full cycles\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t\"tasklist3A\", \"tasklist3A\", \"tasklist3B\",\n\t\t\t// tasklist3A exhausted, tasklist3B continues for 6 more\n\t\t\t\"tasklist3B\", \"tasklist3B\", \"tasklist3B\",\n\t\t\t\"tasklist3B\", \"tasklist3B\", \"tasklist3B\",\n\t\t},\n\t}\n\n\t// Verify full tasklist sequence for each domain\n\tfor domain, expectedSequence := range expectedTasklistSequences {\n\t\tdequeuedTasklists := dequeuedTasklistsByDomain[domain]\n\t\trequire.Equal(t, len(expectedSequence), len(dequeuedTasklists),\n\t\t\t\"tasklist sequence length mismatch for domain %s\", domain)\n\t\tfor i := 0; i < len(expectedSequence); i++ {\n\t\t\tassert.Equal(t, expectedSequence[i], dequeuedTasklists[i],\n\t\t\t\t\"tasklist mismatch for domain %s at position %d\", domain, i)\n\t\t}\n\t}\n\n\t// Verify level 3 (tenant) pattern for each (domain, tasklist) combination\n\t// Build expected tenant sequences manually\n\t// Pattern: [tenant1, tenant1, tenant2, tenant1, tenant2, tenant3] - 6 tasks per cycle\n\texpectedTenantSequences := map[taskKey][]string{\n\t\t// Domain1 tasklists: 3 cycles each = 18 tasks\n\t\t{domain: \"domain1\", tasklist: \"tasklist1A\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t\t{domain: \"domain1\", tasklist: \"tasklist1B\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t\t// Domain2 tasklists: 2 cycles each = 12 tasks\n\t\t{domain: \"domain2\", tasklist: \"tasklist2A\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t\t{domain: \"domain2\", tasklist: \"tasklist2B\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t\t// Domain3 tasklists: 2 cycles each = 12 tasks\n\t\t{domain: \"domain3\", tasklist: \"tasklist3A\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t\t{domain: \"domain3\", tasklist: \"tasklist3B\"}: {\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t\t\"tenant1\", \"tenant1\", \"tenant2\", \"tenant1\", \"tenant2\", \"tenant3\",\n\t\t},\n\t}\n\n\tfor key, expectedSequence := range expectedTenantSequences {\n\t\tdequeuedTenants := dequeuedTenantsByTasklist[key]\n\t\trequire.Equal(t, len(expectedSequence), len(dequeuedTenants),\n\t\t\t\"tenant sequence length mismatch for domain=%s, tasklist=%s\",\n\t\t\tkey.domain, key.tasklist)\n\t\tfor i := 0; i < len(expectedSequence); i++ {\n\t\t\tassert.Equal(t, expectedSequence[i], dequeuedTenants[i],\n\t\t\t\t\"tenant mismatch for domain=%s, tasklist=%s at position %d\",\n\t\t\t\tkey.domain, key.tasklist, i)\n\t\t}\n\t}\n}\n\nfunc TestHierarchicalWRRTaskPool_TryDequeue_Empty(t *testing.T) {\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t}\n\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 10,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Try to dequeue from empty pool\n\ttask, ok := pool.TryDequeue()\n\tassert.False(t, ok, \"TryDequeue should return false on empty pool\")\n\tassert.Nil(t, task, \"task should be nil when dequeue fails\")\n\n\t// Enqueue one task and dequeue it\n\ttestTask := &testPriorityTask{domain: \"domain1\"}\n\terr := pool.Enqueue(testTask)\n\trequire.NoError(t, err)\n\n\ttask, ok = pool.TryDequeue()\n\tassert.True(t, ok, \"TryDequeue should return true when task exists\")\n\tassert.NotNil(t, task, \"task should not be nil\")\n\n\t// Try to dequeue again from now-empty pool\n\ttask, ok = pool.TryDequeue()\n\tassert.False(t, ok, \"TryDequeue should return false after pool is emptied\")\n\tassert.Nil(t, task, \"task should be nil when dequeue fails\")\n}\n\nfunc TestHierarchicalWRRTaskPool_TryEnqueue_BufferFull(t *testing.T) {\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t}\n\n\t// Create pool with small buffer size\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 2, // Small buffer to easily fill it\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// Fill the buffer completely\n\tsuccess1, err1 := pool.TryEnqueue(&testPriorityTask{domain: \"domain1\"})\n\tassert.True(t, success1, \"first TryEnqueue should succeed\")\n\tassert.NoError(t, err1)\n\n\tsuccess2, err2 := pool.TryEnqueue(&testPriorityTask{domain: \"domain1\"})\n\tassert.True(t, success2, \"second TryEnqueue should succeed\")\n\tassert.NoError(t, err2)\n\n\t// Try to enqueue when buffer is full\n\tsuccess3, err3 := pool.TryEnqueue(&testPriorityTask{domain: \"domain1\"})\n\tassert.False(t, success3, \"TryEnqueue should return false when buffer is full\")\n\tassert.NoError(t, err3, \"TryEnqueue should not return error even when buffer is full\")\n\n\t// Dequeue one task to free space\n\ttask, ok := pool.TryDequeue()\n\tassert.True(t, ok, \"TryDequeue should succeed\")\n\tassert.NotNil(t, task)\n\n\t// Now TryEnqueue should succeed again\n\tsuccess4, err4 := pool.TryEnqueue(&testPriorityTask{domain: \"domain1\"})\n\tassert.True(t, success4, \"TryEnqueue should succeed after freeing space\")\n\tassert.NoError(t, err4)\n}\n\nfunc TestHierarchicalWRRTaskPool_Enqueue_ContextCancellation(t *testing.T) {\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t}\n\n\t// Create pool with small buffer size\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tclock.NewMockedTimeSource(),\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 1, // Very small buffer\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\tpool.Start()\n\n\t// Fill the buffer\n\terr := pool.Enqueue(&testPriorityTask{domain: \"domain1\"})\n\trequire.NoError(t, err)\n\n\t// Start a goroutine that will block on Enqueue\n\tenqueueDone := make(chan error, 1)\n\tgo func() {\n\t\terr := pool.Enqueue(&testPriorityTask{domain: \"domain1\"})\n\t\tenqueueDone <- err\n\t}()\n\n\t// Stop the pool, which cancels the context\n\tpool.Stop()\n\n\t// The blocked Enqueue should return with context error\n\tselect {\n\tcase err := <-enqueueDone:\n\t\tassert.Error(t, err, \"Enqueue should return error when context is cancelled\")\n\t\tassert.Equal(t, pool.ctx.Err(), err, \"error should be context cancellation error\")\n\tcase <-time.After(1 * time.Second):\n\t\tt.Fatal(\"Enqueue did not unblock after context cancellation\")\n\t}\n}\n\nfunc TestHierarchicalWRRTaskPool_CleanupLoop(t *testing.T) {\n\tdomainWeights := map[string]int{\n\t\t\"domain1\": 3,\n\t}\n\n\ttimeSource := clock.NewMockedTimeSource()\n\tpool := newHierarchicalWeightedRoundRobinTaskPool[string, *testPriorityTask](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\ttimeSource,\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, *testPriorityTask]{\n\t\t\tBufferSize: 10,\n\t\t\tTaskToWeightedKeysFn: func(task *testPriorityTask) []WeightedKey[string] {\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: task.domain, Weight: domainWeights[task.domain]},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\n\t// Replace doCleanupFn with a fake that tracks calls\n\tcleanupCalls := make(chan struct {\n\t\tnow time.Time\n\t\tttl time.Duration\n\t}, 10)\n\tpool.doCleanupFn = func(now time.Time, ttl time.Duration) {\n\t\tcleanupCalls <- struct {\n\t\t\tnow time.Time\n\t\t\tttl time.Duration\n\t\t}{now: now, ttl: ttl}\n\t}\n\n\tpool.Start()\n\n\t// Advance time by cleanup interval (1800 seconds = TTL/2)\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(1800 * time.Second)\n\n\t// Wait for first cleanup call\n\tcall1 := <-cleanupCalls\n\texpectedTTL := 3600 * time.Second\n\trequire.Equal(t, expectedTTL, call1.ttl, \"TTL should be 3600 seconds\")\n\trequire.NotZero(t, call1.now, \"now should be set\")\n\n\t// Advance time by another interval\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(1800 * time.Second)\n\tcall2 := <-cleanupCalls\n\trequire.Equal(t, expectedTTL, call2.ttl, \"TTL should be 3600 seconds\")\n\trequire.True(t, call2.now.After(call1.now), \"second call should have later timestamp\")\n\n\t// Stop the pool\n\tpool.Stop()\n\n\t// Advance time - should NOT trigger cleanup because loop is stopped\n\ttimeSource.Advance(1800 * time.Second)\n\tselect {\n\tcase <-cleanupCalls:\n\t\tt.Fatal(\"cleanup should not be called after Stop()\")\n\tdefault:\n\t\t// Expected - no cleanup call\n\t}\n}\n"
  },
  {
    "path": "common/task/hierarchical_weighted_round_robin_task_scheduler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype hierarchicalWeightedRoundRobinTaskSchedulerImpl[K comparable, T Task] struct {\n\tsync.RWMutex\n\n\tstatus       int32\n\tpool         *hierarchicalWeightedRoundRobinTaskPoolImpl[K, T]\n\tctx          context.Context\n\tcancel       context.CancelFunc\n\tnotifyCh     chan struct{}\n\tdispatcherWG sync.WaitGroup\n\tlogger       log.Logger\n\tmetricsScope metrics.Scope\n\toptions      *HierarchicalWeightedRoundRobinTaskPoolOptions[K, T]\n\n\tprocessor Processor\n}\n\n// NewHierarchicalWeightedRoundRobinTaskScheduler creates a new hierarchical WRR task scheduler\nfunc NewHierarchicalWeightedRoundRobinTaskScheduler[K comparable, T Task](\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n\tprocessor Processor,\n\toptions *HierarchicalWeightedRoundRobinTaskPoolOptions[K, T],\n) (Scheduler[T], error) {\n\tmetricsScope := metricsClient.Scope(metrics.TaskSchedulerScope)\n\tctx, cancel := context.WithCancel(context.Background())\n\tscheduler := &hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]{\n\t\tstatus: common.DaemonStatusInitialized,\n\t\tpool: newHierarchicalWeightedRoundRobinTaskPool[K](\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\ttimeSource,\n\t\t\toptions,\n\t\t),\n\t\tctx:          ctx,\n\t\tcancel:       cancel,\n\t\tnotifyCh:     make(chan struct{}, 1),\n\t\tlogger:       logger,\n\t\tmetricsScope: metricsScope,\n\t\toptions:      options,\n\t\tprocessor:    processor,\n\t}\n\n\treturn scheduler, nil\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) Start() {\n\tif !atomic.CompareAndSwapInt32(&w.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tw.pool.Start()\n\n\tw.dispatcherWG.Add(1)\n\tgo w.dispatcher()\n\tw.logger.Info(\"Hierarchical weighted round robin task scheduler started.\")\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) Stop() {\n\tif !atomic.CompareAndSwapInt32(&w.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tw.cancel()\n\tw.pool.Stop()\n\n\tif success := common.AwaitWaitGroup(&w.dispatcherWG, time.Minute); !success {\n\t\tw.logger.Warn(\"Hierarchical weighted round robin task scheduler timedout on shutdown.\")\n\t}\n\n\tw.drainAndNackTasks()\n\n\tw.logger.Info(\"Hierarchical weighted round robin task scheduler stopped.\")\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) Submit(task T) error {\n\tw.metricsScope.IncCounter(metrics.PriorityTaskSubmitRequest)\n\tsw := w.metricsScope.StartTimer(metrics.PriorityTaskSubmitLatency)\n\tdefer sw.Stop()\n\n\tif w.isStopped() {\n\t\treturn ErrTaskSchedulerClosed\n\t}\n\n\tif err := w.pool.Enqueue(task); err != nil {\n\t\treturn err\n\t}\n\tw.notifyDispatcher()\n\treturn nil\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) TrySubmit(\n\ttask T,\n) (bool, error) {\n\tw.metricsScope.IncCounter(metrics.PriorityTaskSubmitRequest)\n\tsw := w.metricsScope.StartTimer(metrics.PriorityTaskSubmitLatency)\n\tdefer sw.Stop()\n\n\tif w.isStopped() {\n\t\treturn false, ErrTaskSchedulerClosed\n\t}\n\n\tok, err := w.pool.TryEnqueue(task)\n\tif ok {\n\t\tw.notifyDispatcher()\n\t}\n\treturn ok, err\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) dispatcher() {\n\tdefer w.dispatcherWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-w.notifyCh:\n\t\t\tw.dispatchTasks()\n\t\tcase <-w.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) dispatchTasks() {\n\tfor {\n\t\tif w.isStopped() {\n\t\t\treturn\n\t\t}\n\t\ttask, ok := w.pool.TryDequeue()\n\t\tif !ok {\n\t\t\treturn\n\t\t}\n\t\tif err := w.processor.Submit(task); err != nil {\n\t\t\tw.logger.Error(\"fail to submit task to processor\", tag.Error(err))\n\t\t\ttask.Nack(err)\n\t\t}\n\t}\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) notifyDispatcher() {\n\tselect {\n\tcase w.notifyCh <- struct{}{}:\n\t\t// sent a notification to the dispatcher\n\tdefault:\n\t\t// do not block if there's already a notification\n\t}\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) isStopped() bool {\n\treturn atomic.LoadInt32(&w.status) == common.DaemonStatusStopped\n}\n\nfunc (w *hierarchicalWeightedRoundRobinTaskSchedulerImpl[K, T]) drainAndNackTasks() {\n\tfor {\n\t\ttask, ok := w.pool.TryDequeue()\n\t\tif !ok {\n\t\t\treturn\n\t\t}\n\t\ttask.Nack(nil)\n\t}\n}\n"
  },
  {
    "path": "common/task/hierarchical_weighted_round_robin_task_scheduler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestHierarchicalWeightedRoundRobinTaskScheduler_SchedulerContract(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\n\trealProcessor := NewParallelTaskProcessor(\n\t\ttestlogger.New(t),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\t&ParallelTaskProcessorOptions{\n\t\t\tQueueSize:   1,\n\t\t\tWorkerCount: dynamicproperties.GetIntPropertyFn(1),\n\t\t\tRetryPolicy: backoff.NewExponentialRetryPolicy(time.Millisecond),\n\t\t},\n\t)\n\n\t// Create hierarchical scheduler with string keys based on priority\n\tscheduler, err := NewHierarchicalWeightedRoundRobinTaskScheduler(\n\t\ttestlogger.New(t),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\tclock.NewMockedTimeSource(),\n\t\trealProcessor,\n\t\t&HierarchicalWeightedRoundRobinTaskPoolOptions[string, PriorityTask]{\n\t\t\tBufferSize: 1000,\n\t\t\tTaskToWeightedKeysFn: func(task PriorityTask) []WeightedKey[string] {\n\t\t\t\tpriority := task.Priority()\n\t\t\t\t// Create a simple hierarchy: group -> priority\n\t\t\t\t// Groups based on priority ranges with different weights\n\t\t\t\tvar group string\n\t\t\t\tvar groupWeight int\n\t\t\t\tif priority == 0 {\n\t\t\t\t\tgroup = \"group0\"\n\t\t\t\t\tgroupWeight = 3\n\t\t\t\t} else if priority == 1 {\n\t\t\t\t\tgroup = \"group1\"\n\t\t\t\t\tgroupWeight = 2\n\t\t\t\t} else {\n\t\t\t\t\tgroup = \"group2\"\n\t\t\t\t\tgroupWeight = 1\n\t\t\t\t}\n\n\t\t\t\t// Second level: individual priority\n\t\t\t\tpriorityKey := string(rune('0' + priority))\n\t\t\t\tpriorityWeight := 3 - priority\n\t\t\t\tif priorityWeight < 1 {\n\t\t\t\t\tpriorityWeight = 1\n\t\t\t\t}\n\n\t\t\t\treturn []WeightedKey[string]{\n\t\t\t\t\t{Key: group, Weight: groupWeight},\n\t\t\t\t\t{Key: priorityKey, Weight: priorityWeight},\n\t\t\t\t}\n\t\t\t},\n\t\t},\n\t)\n\trequire.NoError(t, err)\n\n\t// Reuse the existing testSchedulerContract function\n\ttestSchedulerContract(require.New(t), controller, scheduler, realProcessor)\n}\n"
  },
  {
    "path": "common/task/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/common/task\n\npackage task\n\nimport \"github.com/uber/cadence/common\"\n\ntype (\n\t// Processor is the generic coroutine pool interface\n\t// which process tasks\n\tProcessor interface {\n\t\tcommon.Daemon\n\t\tSubmit(task Task) error\n\t}\n\n\t// Scheduler is the generic interface for scheduling tasks with priority\n\t// and processing them\n\tScheduler[T Task] interface {\n\t\tcommon.Daemon\n\t\tSubmit(task T) error\n\t\tTrySubmit(task T) (bool, error)\n\t}\n\n\t// SchedulerType respresents the type of the task scheduler implementation\n\tSchedulerType int\n\n\t// State represents the current state of a task\n\tState int\n\n\t// Task is the interface for tasks\n\tTask interface {\n\t\t// Execute process this task\n\t\tExecute() error\n\t\t// HandleErr handle the error returned by Execute\n\t\tHandleErr(err error) error\n\t\t// RetryErr check whether to retry after HandleErr(Execute())\n\t\tRetryErr(err error) bool\n\t\t// Ack marks the task as successful completed\n\t\tAck()\n\t\t// Nack marks the task as unsuccessful completed\n\t\tNack(err error)\n\t\t// Cancel marks the task as canceled\n\t\tCancel()\n\t\t// State returns the current task state\n\t\tState() State\n\t}\n\n\t// PriorityTask is the interface for tasks which have and can be assigned a priority\n\tPriorityTask interface {\n\t\tTask\n\t\t// Priority returns the priority of the task, or noPriority if no priority was previously assigned\n\t\tPriority() int\n\t\t// SetPriority sets the priority of the task\n\t\tSetPriority(int)\n\t}\n\n\t// SequentialTaskQueueFactory is the function which generate a new SequentialTaskQueue\n\t// for a give SequentialTask\n\tSequentialTaskQueueFactory func(task Task) SequentialTaskQueue\n\n\t// SequentialTaskQueue is the generic task queue interface which group\n\t// sequential tasks to be executed one by one\n\tSequentialTaskQueue interface {\n\t\t// QueueID return the ID of the queue, as well as the tasks inside (same)\n\t\tQueueID() interface{}\n\t\t// Add push an task to the task set\n\t\tAdd(task Task)\n\t\t// Remove pop an task from the task set\n\t\tRemove() Task\n\t\t// IsEmpty indicate if the task set is empty\n\t\tIsEmpty() bool\n\t\t// Len return the size of the queue\n\t\tLen() int\n\t}\n\n\t// Schedule represents a stateless schedule definition\n\tSchedule[V any] interface {\n\t\t// NewIterator creates a new stateful iterator for this schedule\n\t\tNewIterator() Iterator[V]\n\n\t\t// Len returns the length of the schedule\n\t\tLen() int\n\t}\n\n\t// Iterator represents a stateful iteration through a schedule\n\tIterator[V any] interface {\n\t\t// Next returns the next value in the iteration\n\t\t// Returns (value, true) if available, (zero value, false) if exhausted\n\t\tTryNext() (V, bool)\n\t}\n)\n\nconst (\n\t// SchedulerTypeFIFO is the scheduler type for FIFO scheduler implementation\n\tSchedulerTypeFIFO SchedulerType = iota + 1\n\t// SchedulerTypeWRR is the scheduler type for weighted round robin scheduler implementation\n\tSchedulerTypeWRR\n)\n\nconst (\n\t// TaskStatePending is the state for a task when it's waiting to be processed or currently being processed\n\tTaskStatePending State = iota + 1\n\t// TaskStateAcked is the state for a task if it has been successfully completed\n\tTaskStateAcked\n\t// TaskStateCanceled is the state for a task if it has been canceled\n\tTaskStateCanceled\n)\n"
  },
  {
    "path": "common/task/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package task -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/common/task\n//\n\n// Package task is a generated GoMock package.\npackage task\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockProcessor is a mock of Processor interface.\ntype MockProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessorMockRecorder is the mock recorder for MockProcessor.\ntype MockProcessorMockRecorder struct {\n\tmock *MockProcessor\n}\n\n// NewMockProcessor creates a new mock instance.\nfunc NewMockProcessor(ctrl *gomock.Controller) *MockProcessor {\n\tmock := &MockProcessor{ctrl: ctrl}\n\tmock.recorder = &MockProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessor) EXPECT() *MockProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockProcessor) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockProcessorMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockProcessor)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockProcessor)(nil).Stop))\n}\n\n// Submit mocks base method.\nfunc (m *MockProcessor) Submit(task Task) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Submit\", task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Submit indicates an expected call of Submit.\nfunc (mr *MockProcessorMockRecorder) Submit(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Submit\", reflect.TypeOf((*MockProcessor)(nil).Submit), task)\n}\n\n// MockScheduler is a mock of Scheduler interface.\ntype MockScheduler[T Task] struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSchedulerMockRecorder[T]\n\tisgomock struct{}\n}\n\n// MockSchedulerMockRecorder is the mock recorder for MockScheduler.\ntype MockSchedulerMockRecorder[T Task] struct {\n\tmock *MockScheduler[T]\n}\n\n// NewMockScheduler creates a new mock instance.\nfunc NewMockScheduler[T Task](ctrl *gomock.Controller) *MockScheduler[T] {\n\tmock := &MockScheduler[T]{ctrl: ctrl}\n\tmock.recorder = &MockSchedulerMockRecorder[T]{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockScheduler[T]) EXPECT() *MockSchedulerMockRecorder[T] {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockScheduler[T]) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockSchedulerMockRecorder[T]) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockScheduler[T])(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockScheduler[T]) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockSchedulerMockRecorder[T]) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockScheduler[T])(nil).Stop))\n}\n\n// Submit mocks base method.\nfunc (m *MockScheduler[T]) Submit(task T) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Submit\", task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Submit indicates an expected call of Submit.\nfunc (mr *MockSchedulerMockRecorder[T]) Submit(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Submit\", reflect.TypeOf((*MockScheduler[T])(nil).Submit), task)\n}\n\n// TrySubmit mocks base method.\nfunc (m *MockScheduler[T]) TrySubmit(task T) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TrySubmit\", task)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TrySubmit indicates an expected call of TrySubmit.\nfunc (mr *MockSchedulerMockRecorder[T]) TrySubmit(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TrySubmit\", reflect.TypeOf((*MockScheduler[T])(nil).TrySubmit), task)\n}\n\n// MockTask is a mock of Task interface.\ntype MockTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskMockRecorder is the mock recorder for MockTask.\ntype MockTaskMockRecorder struct {\n\tmock *MockTask\n}\n\n// NewMockTask creates a new mock instance.\nfunc NewMockTask(ctrl *gomock.Controller) *MockTask {\n\tmock := &MockTask{ctrl: ctrl}\n\tmock.recorder = &MockTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTask) EXPECT() *MockTaskMockRecorder {\n\treturn m.recorder\n}\n\n// Ack mocks base method.\nfunc (m *MockTask) Ack() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Ack\")\n}\n\n// Ack indicates an expected call of Ack.\nfunc (mr *MockTaskMockRecorder) Ack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ack\", reflect.TypeOf((*MockTask)(nil).Ack))\n}\n\n// Cancel mocks base method.\nfunc (m *MockTask) Cancel() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Cancel\")\n}\n\n// Cancel indicates an expected call of Cancel.\nfunc (mr *MockTaskMockRecorder) Cancel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Cancel\", reflect.TypeOf((*MockTask)(nil).Cancel))\n}\n\n// Execute mocks base method.\nfunc (m *MockTask) Execute() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockTaskMockRecorder) Execute() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockTask)(nil).Execute))\n}\n\n// HandleErr mocks base method.\nfunc (m *MockTask) HandleErr(err error) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleErr\", err)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// HandleErr indicates an expected call of HandleErr.\nfunc (mr *MockTaskMockRecorder) HandleErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleErr\", reflect.TypeOf((*MockTask)(nil).HandleErr), err)\n}\n\n// Nack mocks base method.\nfunc (m *MockTask) Nack(err error) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Nack\", err)\n}\n\n// Nack indicates an expected call of Nack.\nfunc (mr *MockTaskMockRecorder) Nack(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Nack\", reflect.TypeOf((*MockTask)(nil).Nack), err)\n}\n\n// RetryErr mocks base method.\nfunc (m *MockTask) RetryErr(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RetryErr\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// RetryErr indicates an expected call of RetryErr.\nfunc (mr *MockTaskMockRecorder) RetryErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RetryErr\", reflect.TypeOf((*MockTask)(nil).RetryErr), err)\n}\n\n// State mocks base method.\nfunc (m *MockTask) State() State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"State\")\n\tret0, _ := ret[0].(State)\n\treturn ret0\n}\n\n// State indicates an expected call of State.\nfunc (mr *MockTaskMockRecorder) State() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"State\", reflect.TypeOf((*MockTask)(nil).State))\n}\n\n// MockPriorityTask is a mock of PriorityTask interface.\ntype MockPriorityTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPriorityTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockPriorityTaskMockRecorder is the mock recorder for MockPriorityTask.\ntype MockPriorityTaskMockRecorder struct {\n\tmock *MockPriorityTask\n}\n\n// NewMockPriorityTask creates a new mock instance.\nfunc NewMockPriorityTask(ctrl *gomock.Controller) *MockPriorityTask {\n\tmock := &MockPriorityTask{ctrl: ctrl}\n\tmock.recorder = &MockPriorityTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPriorityTask) EXPECT() *MockPriorityTaskMockRecorder {\n\treturn m.recorder\n}\n\n// Ack mocks base method.\nfunc (m *MockPriorityTask) Ack() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Ack\")\n}\n\n// Ack indicates an expected call of Ack.\nfunc (mr *MockPriorityTaskMockRecorder) Ack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ack\", reflect.TypeOf((*MockPriorityTask)(nil).Ack))\n}\n\n// Cancel mocks base method.\nfunc (m *MockPriorityTask) Cancel() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Cancel\")\n}\n\n// Cancel indicates an expected call of Cancel.\nfunc (mr *MockPriorityTaskMockRecorder) Cancel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Cancel\", reflect.TypeOf((*MockPriorityTask)(nil).Cancel))\n}\n\n// Execute mocks base method.\nfunc (m *MockPriorityTask) Execute() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockPriorityTaskMockRecorder) Execute() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockPriorityTask)(nil).Execute))\n}\n\n// HandleErr mocks base method.\nfunc (m *MockPriorityTask) HandleErr(err error) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleErr\", err)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// HandleErr indicates an expected call of HandleErr.\nfunc (mr *MockPriorityTaskMockRecorder) HandleErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleErr\", reflect.TypeOf((*MockPriorityTask)(nil).HandleErr), err)\n}\n\n// Nack mocks base method.\nfunc (m *MockPriorityTask) Nack(err error) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Nack\", err)\n}\n\n// Nack indicates an expected call of Nack.\nfunc (mr *MockPriorityTaskMockRecorder) Nack(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Nack\", reflect.TypeOf((*MockPriorityTask)(nil).Nack), err)\n}\n\n// Priority mocks base method.\nfunc (m *MockPriorityTask) Priority() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Priority\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Priority indicates an expected call of Priority.\nfunc (mr *MockPriorityTaskMockRecorder) Priority() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Priority\", reflect.TypeOf((*MockPriorityTask)(nil).Priority))\n}\n\n// RetryErr mocks base method.\nfunc (m *MockPriorityTask) RetryErr(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RetryErr\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// RetryErr indicates an expected call of RetryErr.\nfunc (mr *MockPriorityTaskMockRecorder) RetryErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RetryErr\", reflect.TypeOf((*MockPriorityTask)(nil).RetryErr), err)\n}\n\n// SetPriority mocks base method.\nfunc (m *MockPriorityTask) SetPriority(arg0 int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetPriority\", arg0)\n}\n\n// SetPriority indicates an expected call of SetPriority.\nfunc (mr *MockPriorityTaskMockRecorder) SetPriority(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetPriority\", reflect.TypeOf((*MockPriorityTask)(nil).SetPriority), arg0)\n}\n\n// State mocks base method.\nfunc (m *MockPriorityTask) State() State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"State\")\n\tret0, _ := ret[0].(State)\n\treturn ret0\n}\n\n// State indicates an expected call of State.\nfunc (mr *MockPriorityTaskMockRecorder) State() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"State\", reflect.TypeOf((*MockPriorityTask)(nil).State))\n}\n\n// MockSequentialTaskQueue is a mock of SequentialTaskQueue interface.\ntype MockSequentialTaskQueue struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSequentialTaskQueueMockRecorder\n\tisgomock struct{}\n}\n\n// MockSequentialTaskQueueMockRecorder is the mock recorder for MockSequentialTaskQueue.\ntype MockSequentialTaskQueueMockRecorder struct {\n\tmock *MockSequentialTaskQueue\n}\n\n// NewMockSequentialTaskQueue creates a new mock instance.\nfunc NewMockSequentialTaskQueue(ctrl *gomock.Controller) *MockSequentialTaskQueue {\n\tmock := &MockSequentialTaskQueue{ctrl: ctrl}\n\tmock.recorder = &MockSequentialTaskQueueMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSequentialTaskQueue) EXPECT() *MockSequentialTaskQueueMockRecorder {\n\treturn m.recorder\n}\n\n// Add mocks base method.\nfunc (m *MockSequentialTaskQueue) Add(task Task) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Add\", task)\n}\n\n// Add indicates an expected call of Add.\nfunc (mr *MockSequentialTaskQueueMockRecorder) Add(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Add\", reflect.TypeOf((*MockSequentialTaskQueue)(nil).Add), task)\n}\n\n// IsEmpty mocks base method.\nfunc (m *MockSequentialTaskQueue) IsEmpty() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsEmpty\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsEmpty indicates an expected call of IsEmpty.\nfunc (mr *MockSequentialTaskQueueMockRecorder) IsEmpty() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsEmpty\", reflect.TypeOf((*MockSequentialTaskQueue)(nil).IsEmpty))\n}\n\n// Len mocks base method.\nfunc (m *MockSequentialTaskQueue) Len() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Len\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Len indicates an expected call of Len.\nfunc (mr *MockSequentialTaskQueueMockRecorder) Len() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Len\", reflect.TypeOf((*MockSequentialTaskQueue)(nil).Len))\n}\n\n// QueueID mocks base method.\nfunc (m *MockSequentialTaskQueue) QueueID() any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueueID\")\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// QueueID indicates an expected call of QueueID.\nfunc (mr *MockSequentialTaskQueueMockRecorder) QueueID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueueID\", reflect.TypeOf((*MockSequentialTaskQueue)(nil).QueueID))\n}\n\n// Remove mocks base method.\nfunc (m *MockSequentialTaskQueue) Remove() Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Remove\")\n\tret0, _ := ret[0].(Task)\n\treturn ret0\n}\n\n// Remove indicates an expected call of Remove.\nfunc (mr *MockSequentialTaskQueueMockRecorder) Remove() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Remove\", reflect.TypeOf((*MockSequentialTaskQueue)(nil).Remove))\n}\n\n// MockSchedule is a mock of Schedule interface.\ntype MockSchedule[V any] struct {\n\tctrl     *gomock.Controller\n\trecorder *MockScheduleMockRecorder[V]\n\tisgomock struct{}\n}\n\n// MockScheduleMockRecorder is the mock recorder for MockSchedule.\ntype MockScheduleMockRecorder[V any] struct {\n\tmock *MockSchedule[V]\n}\n\n// NewMockSchedule creates a new mock instance.\nfunc NewMockSchedule[V any](ctrl *gomock.Controller) *MockSchedule[V] {\n\tmock := &MockSchedule[V]{ctrl: ctrl}\n\tmock.recorder = &MockScheduleMockRecorder[V]{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSchedule[V]) EXPECT() *MockScheduleMockRecorder[V] {\n\treturn m.recorder\n}\n\n// Len mocks base method.\nfunc (m *MockSchedule[V]) Len() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Len\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Len indicates an expected call of Len.\nfunc (mr *MockScheduleMockRecorder[V]) Len() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Len\", reflect.TypeOf((*MockSchedule[V])(nil).Len))\n}\n\n// NewIterator mocks base method.\nfunc (m *MockSchedule[V]) NewIterator() Iterator[V] {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewIterator\")\n\tret0, _ := ret[0].(Iterator[V])\n\treturn ret0\n}\n\n// NewIterator indicates an expected call of NewIterator.\nfunc (mr *MockScheduleMockRecorder[V]) NewIterator() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewIterator\", reflect.TypeOf((*MockSchedule[V])(nil).NewIterator))\n}\n\n// MockIterator is a mock of Iterator interface.\ntype MockIterator[V any] struct {\n\tctrl     *gomock.Controller\n\trecorder *MockIteratorMockRecorder[V]\n\tisgomock struct{}\n}\n\n// MockIteratorMockRecorder is the mock recorder for MockIterator.\ntype MockIteratorMockRecorder[V any] struct {\n\tmock *MockIterator[V]\n}\n\n// NewMockIterator creates a new mock instance.\nfunc NewMockIterator[V any](ctrl *gomock.Controller) *MockIterator[V] {\n\tmock := &MockIterator[V]{ctrl: ctrl}\n\tmock.recorder = &MockIteratorMockRecorder[V]{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockIterator[V]) EXPECT() *MockIteratorMockRecorder[V] {\n\treturn m.recorder\n}\n\n// TryNext mocks base method.\nfunc (m *MockIterator[V]) TryNext() (V, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TryNext\")\n\tret0, _ := ret[0].(V)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// TryNext indicates an expected call of TryNext.\nfunc (mr *MockIteratorMockRecorder[V]) TryNext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TryNext\", reflect.TypeOf((*MockIterator[V])(nil).TryNext))\n}\n"
  },
  {
    "path": "common/task/iwrr_node.go",
    "content": "package task\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n)\n\ntype (\n\t// WeightedKey represents a key-weight pair used in hierarchical IWRR scheduling.\n\t// The key identifies a node at a specific level in the hierarchy, and the weight\n\t// determines the relative priority of that node in the IWRR schedule.\n\tWeightedKey[K comparable] struct {\n\t\tKey    K   // the identifier for this level in the hierarchy\n\t\tWeight int // the relative weight/priority for IWRR scheduling (higher weight = more frequent selection)\n\t}\n\n\t// iwrrNode represents a node in the hierarchical IWRR tree structure.\n\t// Each node can contain its own TTL channel for items and/or children nodes that form a subtree.\n\t// The tree structure allows for hierarchical weighted round-robin scheduling where items are\n\t// distributed across multiple levels based on their key paths and weights.\n\tiwrrNode[K comparable, V any] struct {\n\t\tsync.RWMutex\n\t\t// The following fields are protected by the node's RWMutex\n\t\tchildren map[K]weightedContainer[*iwrrNode[K, V]] // child nodes with their associated weights\n\n\t\t// The following fields are immutable after construction\n\t\tc *TTLChannel[V] // TTL channel for storing items at this level\n\n\t\t// The following fields are concurrency-safe atomic fields\n\t\tchildrenItemCount atomic.Int64                                  // total number of items in all children's channels (across entire subtree)\n\t\tdrainSelfFirst    atomic.Bool                                   // when true, this node's channel is drained before children; when false, children are checked first\n\t\tiwrrSchedule      atomic.Pointer[iwrrSchedule[*iwrrNode[K, V]]] // atomic snapshot of all direct children for IWRR scheduling\n\n\t\t// The following fields are NOT concurrency-safe and must only be accessed by the single consumer goroutine\n\t\titer Iterator[*iwrrNode[K, V]] // stateful iterator for traversing the IWRR schedule\n\t}\n)\n\n// newiwrrNode creates a new iwrrNode with the specified buffer size for its TTL channel.\n// The node is initialized with an empty children map and an empty IWRR schedule.\nfunc newiwrrNode[K comparable, V any](bufferSize int) *iwrrNode[K, V] {\n\tnode := &iwrrNode[K, V]{\n\t\tc:        NewTTLChannel[V](bufferSize),\n\t\tchildren: make(map[K]weightedContainer[*iwrrNode[K, V]]),\n\t}\n\tschedule := newIWRRSchedule[K, *iwrrNode[K, V]](nil)\n\tnode.iwrrSchedule.Store(schedule)\n\tnode.iter = schedule.NewIterator()\n\treturn node\n}\n\n// executeAtPath recursively navigates or creates nodes along the hierarchical key path\n// and executes the callback function at the target (leaf) node.\n//\n// The method traverses the tree following the path, creating intermediate nodes as needed.\n// When a child node is created or its weight changes, the parent's IWRR schedule is updated.\n// After executing the callback, the drainSelfFirst flag is set to indicate children should\n// be drained first, and the childrenItemCount is updated with the delta from the callback.\n//\n// Note: The parent node is responsible for incrementing the reference count of the child node's channel,\n// so that the subtree is not cleaned up while the parent is trying to change the subtree structure.\n//\n// Parameters:\n//   - path: slice of WeightedKey representing the hierarchical path to the target node\n//   - bufferSize: buffer size for any newly created nodes\n//   - callback: function to execute on the target node's channel, returns the change in item count\n//\n// Returns: the delta in item count from the callback execution\n//\n// Concurrency: This method is safe for concurrent access from multiple producer goroutines.\nfunc (n *iwrrNode[K, V]) executeAtPath(path []WeightedKey[K], bufferSize int, callback func(c *TTLChannel[V]) int64) int64 {\n\tif len(path) == 0 {\n\t\tn.drainSelfFirst.Store(false)\n\t\tdelta := callback(n.c)\n\t\treturn delta\n\t}\n\n\tkey := path[0].Key\n\tweight := path[0].Weight\n\tneedsUpdate := false\n\n\tn.RLock()\n\tif container, exists := n.children[key]; exists && container.weight == weight {\n\t\t// Increment reference count to prevent cleanup while we're using this child\n\t\tcontainer.item.c.IncRef()\n\t\tdefer container.item.c.DecRef()\n\t\tn.RUnlock()\n\t\tdelta := container.item.executeAtPath(path[1:], bufferSize, callback)\n\t\tn.drainSelfFirst.Store(true)\n\t\tn.childrenItemCount.Add(delta)\n\t\treturn delta\n\t}\n\tn.RUnlock()\n\n\tn.Lock()\n\tcontainer, exists := n.children[key]\n\tif !exists {\n\t\tchild := newiwrrNode[K, V](bufferSize)\n\t\tn.children[key] = weightedContainer[*iwrrNode[K, V]]{\n\t\t\titem:   child,\n\t\t\tweight: weight,\n\t\t}\n\t\tcontainer = n.children[key]\n\t\tneedsUpdate = true\n\t} else if container.weight != weight {\n\t\tn.children[key] = weightedContainer[*iwrrNode[K, V]]{\n\t\t\titem:   container.item,\n\t\t\tweight: weight,\n\t\t}\n\t\tcontainer = n.children[key]\n\t\tneedsUpdate = true\n\t}\n\t// Update this node's schedule on the way back if needed\n\tif needsUpdate {\n\t\tn.updateScheduleLocked()\n\t}\n\t// Increment reference count to prevent cleanup while we're using this child\n\tcontainer.item.c.IncRef()\n\tdefer container.item.c.DecRef()\n\tn.Unlock()\n\t// Recurse to execute at the leaf node\n\tdelta := container.item.executeAtPath(path[1:], bufferSize, callback)\n\tn.drainSelfFirst.Store(true)\n\tn.childrenItemCount.Add(delta)\n\treturn delta\n}\n\n// updateScheduleLocked updates the IWRR schedule for this node based on its current direct children.\n// The new schedule is atomically stored, allowing the consumer goroutine to pick it up when the\n// current iterator naturally exhausts and resets.\n//\n// Concurrency: Must be called with write lock held on this node.\nfunc (n *iwrrNode[K, V]) updateScheduleLocked() {\n\tschedule := newIWRRSchedule[K, *iwrrNode[K, V]](n.children)\n\tn.iwrrSchedule.Store(schedule)\n\t// The iterator will pick up the new schedule when it naturally exhausts and resets\n}\n\n// tryGetNextItem attempts to retrieve the next item from this subtree using IWRR scheduling.\n// The method respects the drainSelfFirst flag to determine priority:\n//   - If drainSelfFirst is true: tries own channel first, then children\n//   - If drainSelfFirst is false: tries children first, then own channel\n//\n// When traversing children, uses the IWRR iterator to select children according to their weights.\n//\n// Returns: the item and true if an item was successfully retrieved, zero value and false otherwise\n//\n// Concurrency: NOT safe for concurrent access. Must only be called by a single consumer goroutine.\nfunc (n *iwrrNode[K, V]) tryGetNextItem() (V, bool) {\n\tvar zero V\n\n\t// Read drainSelfFirst flag\n\tdrainSelf := n.drainSelfFirst.Load()\n\n\t// Check drainSelfFirst flag to determine order\n\tif drainSelf {\n\t\tif item, ok := n.tryOwnChannel(); ok {\n\t\t\treturn item, true\n\t\t}\n\t}\n\n\t// Try children using IWRR iterator\n\tif item, ok := n.tryChildren(); ok {\n\t\treturn item, true\n\t}\n\n\t// Try own channel if we haven't already\n\tif !drainSelf {\n\t\tif item, ok := n.tryOwnChannel(); ok {\n\t\t\treturn item, true\n\t\t}\n\t}\n\n\treturn zero, false\n}\n\n// tryOwnChannel attempts to retrieve an item from this node's own TTL channel.\n// Uses a non-blocking select to avoid waiting if the channel is empty.\n//\n// Returns: the item and true if an item was available, zero value and false if the channel is empty\nfunc (n *iwrrNode[K, V]) tryOwnChannel() (V, bool) {\n\tvar zero V\n\tselect {\n\tcase item := <-n.c.Chan():\n\t\treturn item, true\n\tdefault:\n\t\treturn zero, false\n\t}\n}\n\n// tryChildren attempts to retrieve an item from one of the child nodes using the IWRR iterator.\n// The method loops while childrenItemCount indicates items are available in the subtree.\n// If the iterator is exhausted but items remain, it resets the iterator with a fresh schedule snapshot.\n// When a child returns an item, the childrenItemCount is decremented to reflect the removal.\n//\n// This approach ensures:\n//   - Children are selected according to their IWRR weights\n//   - Schedule updates from producers are eventually picked up\n//   - The method doesn't block indefinitely on empty children\n//\n// Returns: the item and true if an item was retrieved from a child, zero value and false otherwise\nfunc (n *iwrrNode[K, V]) tryChildren() (V, bool) {\n\tvar zero V\n\tfor n.childrenItemCount.Load() > 0 {\n\t\tchild, ok := n.iter.TryNext()\n\t\tif !ok {\n\t\t\t// Iterator exhausted - check if any children have items\n\t\t\tif n.childrenItemCount.Load() == 0 {\n\t\t\t\t// No children have items, give up\n\t\t\t\treturn zero, false\n\t\t\t}\n\n\t\t\t// Some child has items, reset iterator and try another round\n\t\t\tschedule := n.iwrrSchedule.Load()\n\t\t\tn.iter = schedule.NewIterator()\n\t\t\tcontinue\n\t\t}\n\n\t\t// Recursively try to get item from child\n\t\titem, ok := child.tryGetNextItem()\n\t\tif ok {\n\t\t\t// Got an item from child, decrement children count\n\t\t\tn.childrenItemCount.Add(-1)\n\t\t\treturn item, true\n\t\t}\n\t\t// Child had no item, try next child\n\t}\n\treturn zero, false\n}\n\n// cleanup recursively removes idle nodes from the tree based on the TTL.\n// The method performs a depth-first traversal, cleaning up children before checking itself.\n//\n// Process:\n//  1. Take a snapshot of children with read lock\n//  2. Release lock and recursively cleanup all children (allows concurrent executeAtPath)\n//  3. If any children should be removed, acquire write lock and check if the child still qualifies for deletion\n//     - Child still exists and is the same instance (not replaced)\n//     - Child has no children (no descendants were added since cleanup decision)\n//     - Child's TTL channel still qualifies for cleanup (no tasks in own channel)\n//  4. Update the IWRR schedule if any children were removed\n//  5. If this node is now a leaf, check if its own channel should be cleaned up\n//\n// A leaf node is eligible for removal if its TTL channel has been idle beyond the TTL duration.\n//\n// Performance optimization: The parent lock is released during recursive cleanup of children,\n// reducing lock contention and allowing concurrent operations on the parent node.\n//\n// Returns: true if this node should be removed by its parent, false otherwise\n//\n// Concurrency: Safe for concurrent access. Uses read lock for snapshot, write lock only for modifications.\nfunc (n *iwrrNode[K, V]) cleanup(now time.Time, ttl time.Duration) bool {\n\t// Step 1: Take snapshot of children with read lock\n\tn.RLock()\n\tif len(n.children) == 0 {\n\t\tn.RUnlock()\n\t\t// Fast path: leaf node, check if should be cleaned up\n\t\tshouldRemove := n.c.ShouldCleanup(now, ttl)\n\t\treturn shouldRemove\n\t}\n\tchildrenSnapshot := make(map[K]*iwrrNode[K, V], len(n.children))\n\tfor key, container := range n.children {\n\t\tchildrenSnapshot[key] = container.item\n\t}\n\tn.RUnlock()\n\n\t// Step 2: Cleanup children WITHOUT holding parent lock\n\t// This allows concurrent executeAtPath operations on the parent\n\tkeysToRemove := make([]K, 0)\n\tfor key, child := range childrenSnapshot {\n\t\tif child.cleanup(now, ttl) {\n\t\t\tkeysToRemove = append(keysToRemove, key)\n\t\t}\n\t}\n\n\t// Step 3: Fast path if no children need removal\n\tif len(keysToRemove) == 0 {\n\t\treturn false\n\t}\n\n\t// Step 4: Acquire write lock only when we need to modify\n\tn.Lock()\n\tdefer n.Unlock()\n\n\tneedsUpdate := false\n\tfor _, key := range keysToRemove {\n\t\t// Verify the child still qualifies for deletion:\n\t\t// 1. Child still exists and is the same instance (not replaced)\n\t\t// 2. Child's TTL channel still qualifies for cleanup (no tasks in own channel, RefCount == 0)\n\t\t//    If RefCount == 0, executeAtPath has fully returned, meaning any grandchildren it\n\t\t//    created are already committed to child.children — making the hasNoChildren check\n\t\t//    below accurate.\n\t\t// 3. Child has no children (no descendants were added since cleanup decision)\n\t\t//    If no descendants exist, childrenItemCount must be 0 by definition\n\t\t// This protects against the race where executeAtPath adds tasks to descendants\n\t\t// between the cleanup decision and the actual deletion.\n\t\tcontainer, exists := n.children[key]\n\t\tif !exists || container.item != childrenSnapshot[key] {\n\t\t\tcontinue\n\t\t}\n\t\t// Check ShouldCleanup first: if RefCount > 0, an executeAtPath is still in-flight\n\t\t// for this child and may be creating grandchildren, so skip deletion.\n\t\tif !container.item.c.ShouldCleanup(now, ttl) {\n\t\t\tcontinue\n\t\t}\n\t\t// ShouldCleanup passed (RefCount==0, Len==0, TTL expired). Take the read lock only to\n\t\t// guard against a concurrent cleanup goroutine removing grandchildren from this child\n\t\t// and confirm there are truly no descendants before deleting.\n\t\tcontainer.item.RLock()\n\t\thasNoChildren := len(container.item.children) == 0\n\t\tcontainer.item.RUnlock()\n\t\tif hasNoChildren {\n\t\t\tdelete(n.children, key)\n\t\t\tneedsUpdate = true\n\t\t}\n\t}\n\n\t// Update schedule if children were removed\n\tif needsUpdate {\n\t\tn.updateScheduleLocked()\n\t}\n\n\t// If the node became a leaf node, check if it should be removed\n\tif len(n.children) == 0 {\n\t\treturn n.c.ShouldCleanup(now, ttl)\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "common/task/iwrr_node_test.go",
    "content": "package task\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestIwrrNode_ExecuteAtPath(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tsetupFn            func() (*iwrrNode[string, int], *iwrrNode[string, int]) // returns (root, expectedChild)\n\t\tpath               []WeightedKey[string]\n\t\tbufferSize         int\n\t\tcallbackValue      int64\n\t\texpectedDelta      int64\n\t\tverifyFn           func(t *testing.T, root *iwrrNode[string, int], expectedChild *iwrrNode[string, int])\n\t\texpectedChildCount int64\n\t}{\n\t\t{\n\t\t\tname: \"empty_path_executes_on_current_node\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath:          []WeightedKey[string]{},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tassert.False(t, root.drainSelfFirst.Load(), \"drainSelfFirst should be false after execution on self\")\n\t\t\t},\n\t\t\texpectedChildCount: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"single_element_path_creates_child\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"child1\", Weight: 3},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tcontainer := root.children[\"child1\"]\n\t\t\t\trequire.NotNil(t, container.item)\n\t\t\t\tassert.Equal(t, 3, container.weight)\n\t\t\t\tassert.True(t, root.drainSelfFirst.Load(), \"root should have drainSelfFirst set\")\n\t\t\t\tassert.False(t, container.item.drainSelfFirst.Load(), \"target should not have drainSelfFirst set\")\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"multi_element_path_creates_nested_children\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 5},\n\t\t\t\t{Key: \"level2\", Weight: 3},\n\t\t\t\t{Key: \"level3\", Weight: 2},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tlevel1 := root.children[\"level1\"]\n\t\t\t\trequire.NotNil(t, level1.item)\n\t\t\t\tassert.Equal(t, 5, level1.weight)\n\n\t\t\t\tlevel2 := level1.item.children[\"level2\"]\n\t\t\t\trequire.NotNil(t, level2.item)\n\t\t\t\tassert.Equal(t, 3, level2.weight)\n\n\t\t\t\tlevel3 := level2.item.children[\"level3\"]\n\t\t\t\trequire.NotNil(t, level3.item)\n\t\t\t\tassert.Equal(t, 2, level3.weight)\n\n\t\t\t\t// Check childrenItemCount at each level\n\t\t\t\tassert.Equal(t, int64(1), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level1.item.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level2.item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"existing_path_with_same_weight_reuses_nodes\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Pre-create a 3-level path\n\t\t\t\troot.Lock()\n\t\t\t\tlevel1 := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"level1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\tlevel1.Lock()\n\t\t\t\tlevel2 := newiwrrNode[string, int](10)\n\t\t\t\tlevel1.children[\"level2\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level2,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\t\t\t\tlevel1.updateScheduleLocked()\n\t\t\t\tlevel1.Unlock()\n\n\t\t\t\tlevel2.Lock()\n\t\t\t\tlevel3 := newiwrrNode[string, int](10)\n\t\t\t\tlevel2.children[\"level3\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level3,\n\t\t\t\t\tweight: 2,\n\t\t\t\t}\n\t\t\t\tlevel2.updateScheduleLocked()\n\t\t\t\tlevel2.Unlock()\n\n\t\t\t\t// Store the expected nodes by writing unique IDs to their channels\n\t\t\t\tlevel1.c.Chan() <- 111\n\t\t\t\tlevel2.c.Chan() <- 222\n\t\t\t\tlevel3.c.Chan() <- 333\n\n\t\t\t\treturn root, nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 5},\n\t\t\t\t{Key: \"level2\", Weight: 3},\n\t\t\t\t{Key: \"level3\", Weight: 2},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\t// Verify all three levels exist and have correct weights\n\t\t\t\tlevel1Container := root.children[\"level1\"]\n\t\t\t\trequire.NotNil(t, level1Container.item)\n\t\t\t\tassert.Equal(t, 5, level1Container.weight)\n\t\t\t\t// Check it's the same instance by reading the unique ID we stored\n\t\t\t\tid1, ok := <-level1Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 111, id1, \"level1 should be reused (same instance)\")\n\n\t\t\t\tlevel2Container := level1Container.item.children[\"level2\"]\n\t\t\t\trequire.NotNil(t, level2Container.item)\n\t\t\t\tassert.Equal(t, 3, level2Container.weight)\n\t\t\t\tid2, ok := <-level2Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 222, id2, \"level2 should be reused (same instance)\")\n\n\t\t\t\tlevel3Container := level2Container.item.children[\"level3\"]\n\t\t\t\trequire.NotNil(t, level3Container.item)\n\t\t\t\tassert.Equal(t, 2, level3Container.weight)\n\t\t\t\tid3, ok := <-level3Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 333, id3, \"level3 should be reused (same instance)\")\n\n\t\t\t\t// Verify childrenItemCount at each level\n\t\t\t\tassert.Equal(t, int64(1), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level1Container.item.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level2Container.item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"existing_multi_level_path_with_weight_changes_reuses_nodes\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Pre-create a 3-level path with initial weights [5, 3, 2]\n\t\t\t\troot.Lock()\n\t\t\t\tlevel1 := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"level1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\tlevel1.Lock()\n\t\t\t\tlevel2 := newiwrrNode[string, int](10)\n\t\t\t\tlevel1.children[\"level2\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level2,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\t\t\t\tlevel1.updateScheduleLocked()\n\t\t\t\tlevel1.Unlock()\n\n\t\t\t\tlevel2.Lock()\n\t\t\t\tlevel3 := newiwrrNode[string, int](10)\n\t\t\t\tlevel2.children[\"level3\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level3,\n\t\t\t\t\tweight: 2,\n\t\t\t\t}\n\t\t\t\tlevel2.updateScheduleLocked()\n\t\t\t\tlevel2.Unlock()\n\n\t\t\t\t// Store unique IDs to verify same instances\n\t\t\t\tlevel1.c.Chan() <- 111\n\t\t\t\tlevel2.c.Chan() <- 222\n\t\t\t\tlevel3.c.Chan() <- 333\n\n\t\t\t\treturn root, nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 10}, // Changed from 5 to 10\n\t\t\t\t{Key: \"level2\", Weight: 7},  // Changed from 3 to 7\n\t\t\t\t{Key: \"level3\", Weight: 4},  // Changed from 2 to 4\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\t// Verify all three levels exist with UPDATED weights\n\t\t\t\tlevel1Container := root.children[\"level1\"]\n\t\t\t\trequire.NotNil(t, level1Container.item)\n\t\t\t\tassert.Equal(t, 10, level1Container.weight, \"level1 weight should be updated to 10\")\n\t\t\t\t// Check it's the same instance by reading the unique ID we stored\n\t\t\t\tid1, ok := <-level1Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 111, id1, \"level1 should be reused (same instance)\")\n\n\t\t\t\tlevel2Container := level1Container.item.children[\"level2\"]\n\t\t\t\trequire.NotNil(t, level2Container.item)\n\t\t\t\tassert.Equal(t, 7, level2Container.weight, \"level2 weight should be updated to 7\")\n\t\t\t\tid2, ok := <-level2Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 222, id2, \"level2 should be reused (same instance)\")\n\n\t\t\t\tlevel3Container := level2Container.item.children[\"level3\"]\n\t\t\t\trequire.NotNil(t, level3Container.item)\n\t\t\t\tassert.Equal(t, 4, level3Container.weight, \"level3 weight should be updated to 4\")\n\t\t\t\tid3, ok := <-level3Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 333, id3, \"level3 should be reused (same instance)\")\n\n\t\t\t\t// Verify childrenItemCount at each level\n\t\t\t\tassert.Equal(t, int64(1), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level1Container.item.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level2Container.item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"weight_change_triggers_schedule_update\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Pre-create a child with weight 5\n\t\t\t\troot.Lock()\n\t\t\t\tchild := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"child1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   child,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\t\t\t\treturn root, child\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"child1\", Weight: 10}, // Different weight\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], expectedChild *iwrrNode[string, int]) {\n\t\t\t\tcontainer := root.children[\"child1\"]\n\t\t\t\trequire.NotNil(t, container.item)\n\t\t\t\tassert.Equal(t, 10, container.weight, \"weight should be updated\")\n\t\t\t\t// Should reuse the same child node even though weight changed\n\t\t\t\tassert.Same(t, expectedChild, container.item, \"should reuse the same node instance\")\n\n\t\t\t\t// Verify schedule was updated with new weight\n\t\t\t\tschedule := root.iwrrSchedule.Load()\n\t\t\t\trequire.NotNil(t, schedule)\n\t\t\t\t// Schedule length should reflect the updated weight (10, not 5)\n\t\t\t\tassert.Equal(t, 10, schedule.Len(), \"schedule length should reflect updated weight\")\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"partial_path_exists_reuses_top_creates_bottom\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Only create level1, but not level2 or level3\n\t\t\t\troot.Lock()\n\t\t\t\tlevel1 := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"level1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\t// Store unique ID in level1 to verify it's reused\n\t\t\t\tlevel1.c.Chan() <- 111\n\n\t\t\t\treturn root, nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 5}, // Exists - should reuse\n\t\t\t\t{Key: \"level2\", Weight: 3}, // Doesn't exist - should create\n\t\t\t\t{Key: \"level3\", Weight: 2}, // Doesn't exist - should create\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\t// Verify level1 was reused\n\t\t\t\tlevel1Container := root.children[\"level1\"]\n\t\t\t\trequire.NotNil(t, level1Container.item)\n\t\t\t\tassert.Equal(t, 5, level1Container.weight)\n\t\t\t\tid1, ok := <-level1Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 111, id1, \"level1 should be reused (same instance)\")\n\n\t\t\t\t// Verify level2 was created\n\t\t\t\tlevel2Container := level1Container.item.children[\"level2\"]\n\t\t\t\trequire.NotNil(t, level2Container.item)\n\t\t\t\tassert.Equal(t, 3, level2Container.weight)\n\n\t\t\t\t// Verify level3 was created\n\t\t\t\tlevel3Container := level2Container.item.children[\"level3\"]\n\t\t\t\trequire.NotNil(t, level3Container.item)\n\t\t\t\tassert.Equal(t, 2, level3Container.weight)\n\n\t\t\t\t// Verify childrenItemCount propagated correctly\n\t\t\t\tassert.Equal(t, int64(1), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level1Container.item.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level2Container.item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"partial_path_exists_with_weight_change\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Create level1 and level2, but not level3\n\t\t\t\troot.Lock()\n\t\t\t\tlevel1 := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"level1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\tlevel1.Lock()\n\t\t\t\tlevel2 := newiwrrNode[string, int](10)\n\t\t\t\tlevel1.children[\"level2\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level2,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\t\t\t\tlevel1.updateScheduleLocked()\n\t\t\t\tlevel1.Unlock()\n\n\t\t\t\t// Store unique IDs to verify reuse\n\t\t\t\tlevel1.c.Chan() <- 111\n\t\t\t\tlevel2.c.Chan() <- 222\n\n\t\t\t\treturn root, nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 10}, // Exists - reuse with updated weight\n\t\t\t\t{Key: \"level2\", Weight: 3},  // Exists - reuse with same weight\n\t\t\t\t{Key: \"level3\", Weight: 2},  // Doesn't exist - create\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\t// Verify level1 was reused with updated weight\n\t\t\t\tlevel1Container := root.children[\"level1\"]\n\t\t\t\trequire.NotNil(t, level1Container.item)\n\t\t\t\tassert.Equal(t, 10, level1Container.weight, \"level1 weight should be updated\")\n\t\t\t\tid1, ok := <-level1Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 111, id1, \"level1 should be reused (same instance)\")\n\n\t\t\t\t// Verify level2 was reused with same weight\n\t\t\t\tlevel2Container := level1Container.item.children[\"level2\"]\n\t\t\t\trequire.NotNil(t, level2Container.item)\n\t\t\t\tassert.Equal(t, 3, level2Container.weight)\n\t\t\t\tid2, ok := <-level2Container.item.c.Chan()\n\t\t\t\trequire.True(t, ok)\n\t\t\t\tassert.Equal(t, 222, id2, \"level2 should be reused (same instance)\")\n\n\t\t\t\t// Verify level3 was newly created\n\t\t\t\tlevel3Container := level2Container.item.children[\"level3\"]\n\t\t\t\trequire.NotNil(t, level3Container.item)\n\t\t\t\tassert.Equal(t, 2, level3Container.weight)\n\n\t\t\t\t// Verify childrenItemCount propagated correctly\n\t\t\t\tassert.Equal(t, int64(1), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level1Container.item.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(1), level2Container.item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"multiple_children_at_same_level\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"child1\", Weight: 3},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 1,\n\t\t\texpectedDelta: 1,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\t// Execute on another child to verify multiple children work\n\t\t\t\tdelta2 := root.executeAtPath([]WeightedKey[string]{\n\t\t\t\t\t{Key: \"child2\", Weight: 7},\n\t\t\t\t}, 10, func(c *TTLChannel[int]) int64 {\n\t\t\t\t\treturn 2\n\t\t\t\t})\n\t\t\t\tassert.Equal(t, int64(2), delta2)\n\t\t\t\tassert.Len(t, root.children, 2)\n\t\t\t\tassert.Equal(t, 3, root.children[\"child1\"].weight)\n\t\t\t\tassert.Equal(t, 7, root.children[\"child2\"].weight)\n\t\t\t\tassert.Equal(t, int64(3), root.childrenItemCount.Load(), \"should have sum of deltas\")\n\t\t\t},\n\t\t\texpectedChildCount: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"callback_delta_propagates_up\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"level1\", Weight: 5},\n\t\t\t\t{Key: \"level2\", Weight: 3},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 42,\n\t\t\texpectedDelta: 42,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tassert.Equal(t, int64(42), root.childrenItemCount.Load())\n\t\t\t\tassert.Equal(t, int64(42), root.children[\"level1\"].item.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 42,\n\t\t},\n\t\t{\n\t\t\tname: \"zero_delta_from_callback\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\treturn newiwrrNode[string, int](10), nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"child1\", Weight: 3},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: 0,\n\t\t\texpectedDelta: 0,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tassert.Equal(t, int64(0), root.childrenItemCount.Load())\n\t\t\t},\n\t\t\texpectedChildCount: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"negative_delta_from_callback\",\n\t\t\tsetupFn: func() (*iwrrNode[string, int], *iwrrNode[string, int]) {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\t// Pre-populate some count\n\t\t\t\troot.childrenItemCount.Store(10)\n\t\t\t\treturn root, nil\n\t\t\t},\n\t\t\tpath: []WeightedKey[string]{\n\t\t\t\t{Key: \"child1\", Weight: 3},\n\t\t\t},\n\t\t\tbufferSize:    10,\n\t\t\tcallbackValue: -5,\n\t\t\texpectedDelta: -5,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int], _ *iwrrNode[string, int]) {\n\t\t\t\tassert.Equal(t, int64(5), root.childrenItemCount.Load(), \"should subtract from existing count\")\n\t\t\t},\n\t\t\texpectedChildCount: 5,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\troot, expectedChild := tt.setupFn()\n\n\t\t\t// Create callback that returns the expected value\n\t\t\tdelta := root.executeAtPath(tt.path, tt.bufferSize, func(c *TTLChannel[int]) int64 {\n\t\t\t\treturn tt.callbackValue\n\t\t\t})\n\n\t\t\tassert.Equal(t, tt.expectedDelta, delta, \"delta mismatch\")\n\t\t\tassert.Equal(t, tt.expectedChildCount, root.childrenItemCount.Load(), \"childrenItemCount mismatch\")\n\n\t\t\tif tt.verifyFn != nil {\n\t\t\t\ttt.verifyFn(t, root, expectedChild)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIwrrNode_ExecuteAtPath_ConcurrentAccess(t *testing.T) {\n\troot := newiwrrNode[int, int](100)\n\n\t// Concurrently execute on multi-level paths\n\tdone := make(chan bool)\n\tfor i := 0; i < 10; i++ {\n\t\tgo func(id int) {\n\t\t\tfor j := 0; j < 100; j++ {\n\t\t\t\t// Create 3-level paths with varying structures\n\t\t\t\tpath := []WeightedKey[int]{\n\t\t\t\t\t{Key: id % 3, Weight: (id % 3) + 1}, // level1: 3 different keys\n\t\t\t\t\t{Key: id % 5, Weight: (id % 5) + 1}, // level2: 5 different keys\n\t\t\t\t\t{Key: id, Weight: id + 1},           // level3: 10 different keys\n\t\t\t\t}\n\t\t\t\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\t\t\t\t// Just return delta without writing to channel\n\t\t\t\t\t// to avoid blocking when channel is full\n\t\t\t\t\treturn 1\n\t\t\t\t})\n\t\t\t}\n\t\t\tdone <- true\n\t\t}(i)\n\t}\n\n\t// Wait for all goroutines\n\tfor i := 0; i < 10; i++ {\n\t\t<-done\n\t}\n\n\t// Verify tree structure\n\tlevel1Count := len(root.children)\n\tassert.LessOrEqual(t, level1Count, 3, \"should have at most 3 level1 children\")\n\tassert.GreaterOrEqual(t, level1Count, 1, \"should have at least 1 level1 child\")\n\n\t// Verify level2 and level3 exist\n\tfor key1, container1 := range root.children {\n\t\trequire.NotNil(t, container1.item, \"level1 child %d should not be nil\", key1)\n\n\t\tlevel2Count := len(container1.item.children)\n\t\tassert.LessOrEqual(t, level2Count, 5, \"level1[%d] should have at most 5 level2 children\", key1)\n\t\tassert.GreaterOrEqual(t, level2Count, 1, \"level1[%d] should have at least 1 level2 child\", key1)\n\n\t\tfor key2, container2 := range container1.item.children {\n\t\t\trequire.NotNil(t, container2.item, \"level2 child %d->%d should not be nil\", key1, key2)\n\n\t\t\tlevel3Count := len(container2.item.children)\n\t\t\tassert.GreaterOrEqual(t, level3Count, 1, \"level2[%d][%d] should have at least 1 level3 child\", key1, key2)\n\t\t}\n\t}\n\n\t// Total items should be 10 * 100 = 1000\n\tassert.Equal(t, int64(1000), root.childrenItemCount.Load())\n}\n\nfunc TestIwrrNode_Cleanup(t *testing.T) {\n\tbaseTime := time.Unix(1000, 0)\n\tttl := 10 * time.Second\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupFn        func() *iwrrNode[string, int]\n\t\tnow            time.Time\n\t\tttl            time.Duration\n\t\texpectedReturn bool\n\t\tverifyFn       func(t *testing.T, root *iwrrNode[string, int])\n\t}{\n\t\t{\n\t\t\tname: \"leaf_node_should_be_cleaned_up\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\tnode := newiwrrNode[string, int](10)\n\t\t\t\t// Set last write time to old time\n\t\t\t\tnode.c.UpdateLastWriteTime(baseTime)\n\t\t\t\t// Ensure refCount is 0 and channel is empty\n\t\t\t\treturn node\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second), // Past TTL\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: true,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Empty(t, root.children, \"should have no children\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"leaf_node_should_not_be_cleaned_up_not_expired\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\tnode := newiwrrNode[string, int](10)\n\t\t\t\tnode.c.UpdateLastWriteTime(baseTime)\n\t\t\t\treturn node\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl - time.Second), // Before TTL\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Empty(t, root.children, \"should have no children\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"leaf_node_should_not_be_cleaned_up_has_refs\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\tnode := newiwrrNode[string, int](10)\n\t\t\t\tnode.c.UpdateLastWriteTime(baseTime)\n\t\t\t\tnode.c.IncRef() // Add reference\n\t\t\t\treturn node\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second), // Past TTL\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Equal(t, int32(1), root.c.RefCount())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"leaf_node_should_not_be_cleaned_up_has_data\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\tnode := newiwrrNode[string, int](10)\n\t\t\t\tnode.c.UpdateLastWriteTime(baseTime)\n\t\t\t\tnode.c.Chan() <- 42 // Add data to channel\n\t\t\t\treturn node\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second), // Past TTL\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Equal(t, 1, root.c.Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal_node_with_active_children_should_not_be_cleaned_up\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\troot.Lock()\n\t\t\t\tchild := newiwrrNode[string, int](10)\n\t\t\t\tchild.c.UpdateLastWriteTime(baseTime) // Recent write\n\t\t\t\troot.children[\"child1\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   child,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl - time.Second), // Before TTL\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Len(t, root.children, 1, \"child should not be removed\")\n\t\t\t\t_, exists := root.children[\"child1\"]\n\t\t\t\tassert.True(t, exists, \"child should remain\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal_node_removes_expired_children\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\troot.Lock()\n\t\t\t\t// Add two children: one expired, one active\n\t\t\t\texpiredChild := newiwrrNode[string, int](10)\n\t\t\t\texpiredChild.c.UpdateLastWriteTime(baseTime.Add(-2 * ttl)) // Way past TTL\n\t\t\t\troot.children[\"expired\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   expiredChild,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\n\t\t\t\tactiveChild := newiwrrNode[string, int](10)\n\t\t\t\tactiveChild.c.UpdateLastWriteTime(baseTime.Add(ttl + time.Second)) // Before TTL\n\t\t\t\troot.children[\"active\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   activeChild,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second),\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Len(t, root.children, 1, \"should have 1 child remaining\")\n\t\t\t\t_, exists := root.children[\"active\"]\n\t\t\t\tassert.True(t, exists, \"active child should remain\")\n\t\t\t\t_, exists = root.children[\"expired\"]\n\t\t\t\tassert.False(t, exists, \"expired child should be removed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal_node_removes_all_expired_children\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\troot.Lock()\n\t\t\t\t// Add multiple expired children\n\t\t\t\tfor i := 1; i <= 3; i++ {\n\t\t\t\t\tchild := newiwrrNode[string, int](10)\n\t\t\t\t\tchild.c.UpdateLastWriteTime(baseTime.Add(-ttl - time.Second))\n\t\t\t\t\troot.children[string(rune('a'+i-1))] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\t\titem:   child,\n\t\t\t\t\t\tweight: i,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second),\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: true, // All children removed, so node should be removed\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Empty(t, root.children, \"all children should be removed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"schedule_updated_when_children_removed\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\t\t\t\troot.Lock()\n\t\t\t\t// Add two children with different weights\n\t\t\t\tchild1 := newiwrrNode[string, int](10)\n\t\t\t\tchild1.c.UpdateLastWriteTime(baseTime.Add(-ttl - time.Second)) // Expired\n\t\t\t\troot.children[\"expired\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   child1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\n\t\t\t\tchild2 := newiwrrNode[string, int](10)\n\t\t\t\tchild2.c.UpdateLastWriteTime(baseTime.Add(ttl + time.Second)) // Active\n\t\t\t\troot.children[\"active\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   child2,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second),\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Len(t, root.children, 1)\n\t\t\t\t_, exists := root.children[\"expired\"]\n\t\t\t\tassert.False(t, exists, \"expired child should be removed\")\n\t\t\t\t_, exists = root.children[\"active\"]\n\t\t\t\tassert.True(t, exists, \"active child should remain\")\n\n\t\t\t\t// New schedule should only have weight 3\n\t\t\t\tnewSchedule := root.iwrrSchedule.Load()\n\t\t\t\tassert.Equal(t, 3, newSchedule.Len(), \"updated schedule should only have weight 3\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multi_level_recursive_cleanup\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\n\t\t\t\t// Create 3-level tree: root -> level1 -> level2 -> level3\n\t\t\t\troot.Lock()\n\t\t\t\tlevel1 := newiwrrNode[string, int](10)\n\t\t\t\tlevel1.c.UpdateLastWriteTime(baseTime.Add(-ttl - time.Second)) // Expired\n\t\t\t\troot.children[\"expired\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level1,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\tlevel1.Lock()\n\t\t\t\tlevel2 := newiwrrNode[string, int](10)\n\t\t\t\tlevel2.c.UpdateLastWriteTime(baseTime.Add(-ttl - time.Second)) // Expired\n\t\t\t\tlevel1.children[\"expired\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level2,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\t\t\t\tlevel1.updateScheduleLocked()\n\t\t\t\tlevel1.Unlock()\n\n\t\t\t\tlevel2.Lock()\n\t\t\t\tlevel3 := newiwrrNode[string, int](10)\n\t\t\t\tlevel3.c.UpdateLastWriteTime(baseTime.Add(-ttl - time.Second)) // Expired\n\t\t\t\tlevel2.children[\"level3\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   level3,\n\t\t\t\t\tweight: 2,\n\t\t\t\t}\n\t\t\t\tlevel2.updateScheduleLocked()\n\t\t\t\tlevel2.Unlock()\n\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second),\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: true, // All should cascade up and be removed\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Empty(t, root.children, \"all children should be recursively removed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multi_level_partial_cleanup\",\n\t\t\tsetupFn: func() *iwrrNode[string, int] {\n\t\t\t\troot := newiwrrNode[string, int](10)\n\n\t\t\t\t// Create tree where one branch is expired, another is active\n\t\t\t\troot.Lock()\n\t\t\t\texpiredBranch := newiwrrNode[string, int](10)\n\t\t\t\texpiredBranch.c.UpdateLastWriteTime(baseTime.Add(-2 * ttl)) // Way past TTL\n\t\t\t\troot.children[\"expired\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   expiredBranch,\n\t\t\t\t\tweight: 3,\n\t\t\t\t}\n\n\t\t\t\tactiveBranch := newiwrrNode[string, int](10)\n\t\t\t\troot.children[\"active\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   activeBranch,\n\t\t\t\t\tweight: 5,\n\t\t\t\t}\n\t\t\t\troot.updateScheduleLocked()\n\t\t\t\troot.Unlock()\n\n\t\t\t\t// Add active child to active branch\n\t\t\t\tactiveBranch.Lock()\n\t\t\t\tactiveLeaf := newiwrrNode[string, int](10)\n\t\t\t\tactiveLeaf.c.UpdateLastWriteTime(baseTime) // Recent write\n\t\t\t\tactiveLeaf.c.IncRef()                      // Keep it active\n\t\t\t\tactiveBranch.children[\"leaf\"] = weightedContainer[*iwrrNode[string, int]]{\n\t\t\t\t\titem:   activeLeaf,\n\t\t\t\t\tweight: 2,\n\t\t\t\t}\n\t\t\t\tactiveBranch.updateScheduleLocked()\n\t\t\t\tactiveBranch.Unlock()\n\n\t\t\t\treturn root\n\t\t\t},\n\t\t\tnow:            baseTime.Add(ttl + time.Second),\n\t\t\tttl:            ttl,\n\t\t\texpectedReturn: false,\n\t\t\tverifyFn: func(t *testing.T, root *iwrrNode[string, int]) {\n\t\t\t\tassert.Len(t, root.children, 1, \"only active branch should remain\")\n\t\t\t\t_, exists := root.children[\"active\"]\n\t\t\t\tassert.True(t, exists, \"active branch should remain\")\n\n\t\t\t\tactiveBranch := root.children[\"active\"].item\n\t\t\t\tassert.Len(t, activeBranch.children, 1, \"active leaf should remain\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\troot := tt.setupFn()\n\t\t\tshouldRemove := root.cleanup(tt.now, tt.ttl)\n\t\t\tassert.Equal(t, tt.expectedReturn, shouldRemove, \"cleanup return value mismatch\")\n\n\t\t\tif tt.verifyFn != nil {\n\t\t\t\ttt.verifyFn(t, root)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIwrrNode_Cleanup_ConcurrentCleanup(t *testing.T) {\n\t// Test that multiple concurrent cleanup calls don't cause race conditions\n\troot := newiwrrNode[string, int](10)\n\n\t// Create a tree with multiple levels and old timestamps\n\t// root -> level1a -> level2a\n\t//      -> level1b -> level2b\n\t//      -> level1c -> level2c\n\toldTime := time.Now().Add(-2 * time.Hour) // Old enough to be cleaned up\n\tfor i := 0; i < 3; i++ {\n\t\tpath := []WeightedKey[string]{\n\t\t\t{Key: fmt.Sprintf(\"level1_%d\", i), Weight: 1},\n\t\t\t{Key: fmt.Sprintf(\"level2_%d\", i), Weight: 1},\n\t\t}\n\t\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\t\tc.UpdateLastWriteTime(oldTime) // Set old timestamp\n\t\t\treturn 0                       // Don't add items, just create the tree structure\n\t\t})\n\t}\n\n\t// Verify initial state\n\trequire.Equal(t, 3, len(root.children))\n\n\t// Run multiple concurrent cleanup operations\n\tnow := time.Now()\n\tttl := time.Hour\n\n\tvar wg sync.WaitGroup\n\tnumGoroutines := 10\n\tfor i := 0; i < numGoroutines; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\troot.cleanup(now, ttl)\n\t\t}()\n\t}\n\n\twg.Wait()\n\n\t// Verify tree was cleaned up correctly\n\t// All nodes should be removed since they're idle\n\tassert.Equal(t, 0, len(root.children))\n}\n\nfunc TestIwrrNode_Cleanup_ConcurrentWithExecuteAtPath(t *testing.T) {\n\t// Test that cleanup and executeAtPath can run concurrently without deadlock or corruption\n\t// Uses multi-level hierarchy with multiple goroutines for both operations\n\troot := newiwrrNode[string, int](100)\n\n\tvar wg sync.WaitGroup\n\tstopCh := make(chan struct{})\n\tvar tasksAdded atomic.Int64\n\n\t// Multiple goroutines running executeAtPath with multi-level paths\n\tnumExecuteGoroutines := 10\n\tfor g := 0; g < numExecuteGoroutines; g++ {\n\t\twg.Add(1)\n\t\tgo func(goroutineID int) {\n\t\t\tdefer wg.Done()\n\t\t\tcounter := 0\n\t\t\tfor {\n\t\t\t\tselect {\n\t\t\t\tcase <-stopCh:\n\t\t\t\t\treturn\n\t\t\t\tdefault:\n\t\t\t\t\t// Create multi-level paths: domain -> tasklist -> tenant\n\t\t\t\t\tdomainKey := fmt.Sprintf(\"domain_%d\", (goroutineID+counter)%3)\n\t\t\t\t\ttasklistKey := fmt.Sprintf(\"tasklist_%d\", (goroutineID+counter)%5)\n\t\t\t\t\ttenantKey := fmt.Sprintf(\"tenant_%d\", (goroutineID+counter)%7)\n\n\t\t\t\t\tpath := []WeightedKey[string]{\n\t\t\t\t\t\t{Key: domainKey, Weight: ((goroutineID + counter) % 3) + 1},\n\t\t\t\t\t\t{Key: tasklistKey, Weight: ((goroutineID + counter) % 5) + 1},\n\t\t\t\t\t\t{Key: tenantKey, Weight: ((goroutineID + counter) % 7) + 1},\n\t\t\t\t\t}\n\n\t\t\t\t\ttaskValue := goroutineID*10000 + counter\n\t\t\t\t\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\t\t\t\t\tselect {\n\t\t\t\t\t\tcase c.Chan() <- taskValue:\n\t\t\t\t\t\t\tc.UpdateLastWriteTime(time.Now())\n\t\t\t\t\t\t\ttasksAdded.Add(1)\n\t\t\t\t\t\t\treturn 1\n\t\t\t\t\t\tdefault:\n\t\t\t\t\t\t\treturn 0\n\t\t\t\t\t\t}\n\t\t\t\t\t})\n\t\t\t\t\tcounter++\n\t\t\t\t}\n\t\t\t}\n\t\t}(g)\n\t}\n\n\t// Multiple goroutines running cleanup\n\tnumCleanupGoroutines := 5\n\tfor g := 0; g < numCleanupGoroutines; g++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor {\n\t\t\t\tselect {\n\t\t\t\tcase <-stopCh:\n\t\t\t\t\treturn\n\t\t\t\tdefault:\n\t\t\t\t\t// Cleanup with a TTL that will remove idle nodes\n\t\t\t\t\tnow := time.Now()\n\t\t\t\t\tttl := 50 * time.Millisecond\n\t\t\t\t\troot.cleanup(now, ttl)\n\t\t\t\t\ttime.Sleep(time.Millisecond)\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\n\t// Let them run concurrently for a short time\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(stopCh)\n\twg.Wait()\n\n\t// Drain all tasks and verify count\n\tdrainedCount := 0\n\tfor {\n\t\tif _, ok := root.tryGetNextItem(); ok {\n\t\t\tdrainedCount++\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\n\t// All added items should be drained\n\tassert.Equal(t, int(tasksAdded.Load()), drainedCount, \"All enqueued items should be dequeuable\")\n}\n\nfunc TestIwrrNode_Cleanup_PointerEqualityCheck(t *testing.T) {\n\t// Test that cleanup only removes the child it decided to remove,\n\t// not a newly created child with the same key\n\troot := newiwrrNode[string, int](10)\n\n\t// Add an initial child that will be cleaned up\n\tpath := []WeightedKey[string]{\n\t\t{Key: \"child\", Weight: 1},\n\t}\n\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\tc.UpdateLastWriteTime(time.Now().Add(-2 * time.Hour)) // Old timestamp\n\t\treturn 0\n\t})\n\n\t// Start cleanup in background (it will snapshot, then recurse, then try to delete)\n\tcleanupDone := make(chan bool)\n\tgo func() {\n\t\t// Add a small delay to ensure executeAtPath runs during the recursive phase\n\t\tnow := time.Now()\n\t\tttl := time.Hour\n\t\troot.cleanup(now, ttl)\n\t\tcleanupDone <- true\n\t}()\n\n\t// While cleanup is running, add a new child with the same key\n\t// This simulates the race condition the pointer equality check protects against\n\ttime.Sleep(time.Millisecond) // Give cleanup time to snapshot\n\tpath2 := []WeightedKey[string]{\n\t\t{Key: \"child\", Weight: 2}, // Same key, different weight\n\t}\n\troot.executeAtPath(path2, 10, func(c *TTLChannel[int]) int64 {\n\t\tc.UpdateLastWriteTime(time.Now())\n\t\treturn 0\n\t})\n\n\t<-cleanupDone\n\n\t// Verify: the new child should still exist\n\troot.RLock()\n\tcontainer, exists := root.children[\"child\"]\n\troot.RUnlock()\n\n\t// The child should exist (either old or new depending on timing)\n\t// If it's the new child, weight should be 2\n\t// If it's the old child, weight should be 1\n\t// Either way, the tree should be in a consistent state\n\tif exists {\n\t\t// If child exists, verify it's a valid node\n\t\tassert.NotNil(t, container.item)\n\t\tassert.Contains(t, []int{1, 2}, container.weight)\n\t}\n}\n\nfunc TestIwrrNode_Cleanup_DeepHierarchyConcurrency(t *testing.T) {\n\t// Test cleanup performance with deep hierarchy and concurrent operations\n\t// Verifies all tasks can be drained even with concurrent cleanup\n\troot := newiwrrNode[string, int](10)\n\n\t// Track tasks added\n\tvar tasksAdded atomic.Int64\n\n\t// Create a deep hierarchy (5 levels, 3 children per level = 243 leaf paths)\n\tvar createPaths func(depth int, prefix string) [][]WeightedKey[string]\n\tcreatePaths = func(depth int, prefix string) [][]WeightedKey[string] {\n\t\tif depth == 0 {\n\t\t\treturn [][]WeightedKey[string]{{}}\n\t\t}\n\t\tvar paths [][]WeightedKey[string]\n\t\tfor i := 0; i < 3; i++ {\n\t\t\tkey := fmt.Sprintf(\"%s_L%d_C%d\", prefix, 5-depth, i)\n\t\t\tsubPaths := createPaths(depth-1, prefix)\n\t\t\tfor _, subPath := range subPaths {\n\t\t\t\tpath := append([]WeightedKey[string]{{Key: key, Weight: i + 1}}, subPath...)\n\t\t\t\tpaths = append(paths, path)\n\t\t\t}\n\t\t}\n\t\treturn paths\n\t}\n\n\tpaths := createPaths(5, \"node\")\n\tfor i, path := range paths {\n\t\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\t\tselect {\n\t\t\tcase c.Chan() <- i:\n\t\t\t\tc.UpdateLastWriteTime(time.Now())\n\t\t\t\ttasksAdded.Add(1)\n\t\t\t\treturn 1\n\t\t\tdefault:\n\t\t\t\treturn 0\n\t\t\t}\n\t\t})\n\t}\n\n\t// Verify tree was created\n\trequire.Equal(t, 3, len(root.children))\n\n\tvar wg sync.WaitGroup\n\n\t// Start multiple cleanup operations concurrently\n\tfor i := 0; i < 5; i++ {\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tnow := time.Now().Add(2 * time.Hour)\n\t\t\tttl := time.Hour\n\t\t\troot.cleanup(now, ttl)\n\t\t}()\n\t}\n\n\t// Also start some executeAtPath operations to add new nodes/tasks\n\tfor i := 0; i < 5; i++ {\n\t\twg.Add(1)\n\t\tgo func(idx int) {\n\t\t\tdefer wg.Done()\n\t\t\tpath := []WeightedKey[string]{\n\t\t\t\t{Key: fmt.Sprintf(\"new_L0_C%d\", idx), Weight: 1},\n\t\t\t}\n\t\t\troot.executeAtPath(path, 10, func(c *TTLChannel[int]) int64 {\n\t\t\t\tselect {\n\t\t\t\tcase c.Chan() <- idx + 1000:\n\t\t\t\t\tc.UpdateLastWriteTime(time.Now())\n\t\t\t\t\ttasksAdded.Add(1)\n\t\t\t\t\treturn 1\n\t\t\t\tdefault:\n\t\t\t\t\treturn 0\n\t\t\t\t}\n\t\t\t})\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\n\t// Drain all remaining tasks to verify nothing was lost\n\tdrainedCount := 0\n\tfor {\n\t\tif _, ok := root.tryGetNextItem(); ok {\n\t\t\tdrainedCount++\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ttotalAdded := int(tasksAdded.Load())\n\tassert.Equal(t, totalAdded, drainedCount, \"All added tasks should be dequeuable\")\n}\n"
  },
  {
    "path": "common/task/iwrr_schedule.go",
    "content": "package task\n\nimport \"slices\"\n\nvar _ Schedule[any] = &iwrrSchedule[any]{}\n\n// weightedContainer is a container for an item with a weight\ntype weightedContainer[V any] struct {\n\tweight int\n\titem   V\n}\n\n// iwrrSchedule implements Schedule using an efficient interleaved weighted round-robin algorithm\n// ref: https://en.wikipedia.org/wiki/Weighted_round_robin#Interleaved_WRR\n// It is stateless and creates iterators on demand\ntype iwrrSchedule[V any] struct {\n\t// Snapshot of items sorted by weight (ascending)\n\titems []weightedContainer[V]\n\t// Maximum weight among all items\n\tmaxWeight int\n\t// Total virtual length (sum of all weights)\n\ttotalLen int\n}\n\n// iwrrIterator is a stateful iterator that tracks position in the schedule\ntype iwrrIterator[V any] struct {\n\tschedule     *iwrrSchedule[V]\n\tcurrentRound int // Current round (maxWeight-1 down to 0)\n\tcurrentIndex int // Index within current round's qualifying items\n}\n\n// newIWRRSchedule creates a new IWRR schedule from a snapshot of weighted containers\n// Items with weight <= 0 are ignored\nfunc newIWRRSchedule[K comparable, V any](items map[K]weightedContainer[V]) *iwrrSchedule[V] {\n\tif len(items) == 0 {\n\t\treturn &iwrrSchedule[V]{}\n\t}\n\n\t// Filter out items with weight <= 0 and copy to slice\n\titemsCopy := make([]weightedContainer[V], 0, len(items))\n\ttotalLen := 0\n\tfor _, container := range items {\n\t\tif container.weight > 0 {\n\t\t\titemsCopy = append(itemsCopy, container)\n\t\t\ttotalLen += container.weight\n\t\t}\n\t}\n\n\t// Return empty schedule if no valid items\n\tif len(itemsCopy) == 0 {\n\t\treturn &iwrrSchedule[V]{}\n\t}\n\n\t// Sort by weight (ascending)\n\tslices.SortFunc(itemsCopy, func(a, b weightedContainer[V]) int {\n\t\treturn a.weight - b.weight\n\t})\n\n\tmaxWeight := itemsCopy[len(itemsCopy)-1].weight\n\n\treturn &iwrrSchedule[V]{\n\t\titems:     itemsCopy,\n\t\tmaxWeight: maxWeight,\n\t\ttotalLen:  totalLen,\n\t}\n}\n\n// NewIterator creates a new stateful iterator for this schedule\nfunc (s *iwrrSchedule[V]) NewIterator() Iterator[V] {\n\tif len(s.items) == 0 {\n\t\treturn &iwrrIterator[V]{schedule: s}\n\t}\n\treturn &iwrrIterator[V]{\n\t\tschedule:     s,\n\t\tcurrentRound: s.maxWeight - 1,\n\t\tcurrentIndex: len(s.items) - 1,\n\t}\n}\n\n// Len returns the total virtual length of the schedule\nfunc (s *iwrrSchedule[V]) Len() int {\n\treturn s.totalLen\n}\n\n// TryNext returns the next item in the IWRR iteration\n// The algorithm processes rounds from maxWeight-1 down to 0\n// In each round r, items with weight > r are included\n// Returns false when the iteration is exhausted (all rounds completed)\nfunc (it *iwrrIterator[V]) TryNext() (zero V, ok bool) {\n\tif it.schedule == nil || len(it.schedule.items) == 0 {\n\t\treturn\n\t}\n\n\t// Find the next qualifying item\n\tfor it.currentRound >= 0 {\n\t\t// Find items that qualify for current round (weight > round)\n\t\t// We iterate from highest weight to lowest\n\t\tfor it.currentIndex >= 0 && it.schedule.items[it.currentIndex].weight > it.currentRound {\n\t\t\titem := it.schedule.items[it.currentIndex].item\n\t\t\tit.currentIndex--\n\t\t\treturn item, true\n\t\t}\n\n\t\t// Move to next round\n\t\tit.currentRound--\n\t\tit.currentIndex = len(it.schedule.items) - 1\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/task/iwrr_schedule_test.go",
    "content": "package task\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\n// testWeightedItem is a simple implementation for testing\ntype testWeightedItem struct {\n\tid int\n}\n\n// toWeightedMap converts a map of items with weights to a map of weightedContainers\nfunc toWeightedMap(items map[int]*testWeightedItem, weights map[int]int) map[int]weightedContainer[*testWeightedItem] {\n\tresult := make(map[int]weightedContainer[*testWeightedItem])\n\tfor k, item := range items {\n\t\tresult[k] = weightedContainer[*testWeightedItem]{\n\t\t\titem:   item,\n\t\t\tweight: weights[k],\n\t\t}\n\t}\n\treturn result\n}\n\nfunc TestIWRRSchedule_Empty(t *testing.T) {\n\tschedule := newIWRRSchedule[int, *testWeightedItem](nil)\n\n\tassert.Equal(t, 0, schedule.Len())\n\n\titer := schedule.NewIterator()\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_SingleChannel(t *testing.T) {\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t}\n\tweights := map[int]int{\n\t\t0: 3,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\t// Total length should be the weight\n\tassert.Equal(t, 3, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// Should return the item 3 times\n\tfor i := 0; i < 3; i++ {\n\t\titem, ok := iter.TryNext()\n\t\tassert.True(t, ok, \"iteration %d should succeed\", i)\n\t\tassert.Equal(t, items[0], item)\n\t}\n\n\t// Fourth call should return false (exhausted)\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_MultipleChannels_EqualWeights(t *testing.T) {\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t\t1: {id: 1},\n\t\t2: {id: 2},\n\t}\n\tweights := map[int]int{\n\t\t0: 2,\n\t\t1: 2,\n\t\t2: 2,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\tassert.Equal(t, 6, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// With equal weights, each item should appear equal number of times\n\t// Note: the exact order depends on map iteration order\n\tcounts := make(map[*testWeightedItem]int)\n\tfor {\n\t\titem, ok := iter.TryNext()\n\t\tif !ok {\n\t\t\tbreak\n\t\t}\n\t\tcounts[item]++\n\t}\n\n\tassert.Equal(t, 2, counts[items[0]], \"item 0 should appear 2 times\")\n\tassert.Equal(t, 2, counts[items[1]], \"item 1 should appear 2 times\")\n\tassert.Equal(t, 2, counts[items[2]], \"item 2 should appear 2 times\")\n}\n\nfunc TestIWRRSchedule_MultipleChannels_DifferentWeights(t *testing.T) {\n\t// Create items with weights [1, 2, 3]\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t\t1: {id: 1},\n\t\t2: {id: 2},\n\t}\n\tweights := map[int]int{\n\t\t0: 1,\n\t\t1: 2,\n\t\t2: 3,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\tassert.Equal(t, 6, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// IWRR with weights [1, 2, 3] processes rounds from 2 down to 0:\n\t// Round 2: items with weight > 2 → item 2 (weight 3)\n\t// Round 1: items with weight > 1 → items 2, 1 (weights 3, 2)\n\t// Round 0: items with weight > 0 → items 2, 1, 0 (weights 3, 2, 1)\n\t// Result: [2, 2, 1, 2, 1, 0]\n\texpectedSequence := []*testWeightedItem{\n\t\titems[2], // round 2: weight 3\n\t\titems[2], // round 1: weight 3\n\t\titems[1], // round 1: weight 2\n\t\titems[2], // round 0: weight 3\n\t\titems[1], // round 0: weight 2\n\t\titems[0], // round 0: weight 1\n\t}\n\n\tfor i, expected := range expectedSequence {\n\t\titem, ok := iter.TryNext()\n\t\trequire.True(t, ok, \"iteration %d should succeed\", i)\n\t\tassert.Equal(t, expected, item, \"iteration %d\", i)\n\t}\n\n\t// Should be exhausted\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_LargeWeights(t *testing.T) {\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t\t1: {id: 1},\n\t\t2: {id: 2},\n\t}\n\tweights := map[int]int{\n\t\t0: 100,\n\t\t1: 50,\n\t\t2: 25,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\tassert.Equal(t, 175, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// IWRR pattern for [100, 50, 25]:\n\t// Rounds 99-50 (50 rounds): only item 0 (weight 100 > round)\n\t// Rounds 49-25 (25 rounds): items 0, 1 (weights 100, 50 > round)\n\t// Rounds 24-0 (25 rounds): items 0, 1, 2 (all weights > round)\n\tvar expectedSequence []*testWeightedItem\n\n\t// First 50 rounds: only item 0\n\tfor i := 0; i < 50; i++ {\n\t\texpectedSequence = append(expectedSequence, items[0])\n\t}\n\n\t// Next 25 rounds: items 0, 1\n\tfor i := 0; i < 25; i++ {\n\t\texpectedSequence = append(expectedSequence, items[0], items[1])\n\t}\n\n\t// Last 25 rounds: items 0, 1, 2\n\tfor i := 0; i < 25; i++ {\n\t\texpectedSequence = append(expectedSequence, items[0], items[1], items[2])\n\t}\n\n\t// Verify the sequence\n\tfor i, expected := range expectedSequence {\n\t\titem, ok := iter.TryNext()\n\t\trequire.True(t, ok, \"iteration %d should succeed\", i)\n\t\tassert.Equal(t, expected, item, \"iteration %d\", i)\n\t}\n\n\t// Should be exhausted\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_ChannelWithZeroWeight(t *testing.T) {\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t\t1: {id: 1},\n\t}\n\tweights := map[int]int{\n\t\t0: 0,\n\t\t1: 3,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\t// Total length should only count non-zero weights\n\tassert.Equal(t, 3, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// Should only return item with weight 3\n\tfor i := 0; i < 3; i++ {\n\t\titem, ok := iter.TryNext()\n\t\trequire.True(t, ok, \"iteration %d\", i)\n\t\tassert.Equal(t, items[1], item)\n\t}\n\n\t// Should be exhausted\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_WeightedChannelFields(t *testing.T) {\n\t// Verify that the returned item is correct\n\ttestItem := &testWeightedItem{\n\t\tid: 0,\n\t}\n\titems := map[int]*testWeightedItem{\n\t\t0: testItem,\n\t}\n\tweights := map[int]int{\n\t\t0: 10,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\titer := schedule.NewIterator()\n\n\titem, ok := iter.TryNext()\n\trequire.True(t, ok)\n\tassert.Equal(t, testItem, item)\n}\n\nfunc TestIWRRSchedule_ExhaustedSchedule_MultipleCallsReturnFalse(t *testing.T) {\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t}\n\tweights := map[int]int{\n\t\t0: 1,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\titer := schedule.NewIterator()\n\n\t// Exhaust the iterator\n\titem, ok := iter.TryNext()\n\tassert.True(t, ok)\n\tassert.NotNil(t, item)\n\n\t// Multiple calls after exhaustion should all return false\n\tfor i := 0; i < 5; i++ {\n\t\titem, ok := iter.TryNext()\n\t\tassert.False(t, ok, \"call %d after exhaustion\", i)\n\t\tassert.Nil(t, item, \"call %d after exhaustion\", i)\n\t}\n}\n\nfunc TestIWRRSchedule_Ordering_Weights_5_3_1(t *testing.T) {\n\t// Test case from task pool tests: weights [5, 3, 1]\n\titems := map[int]*testWeightedItem{\n\t\t0: {id: 0},\n\t\t1: {id: 1},\n\t\t2: {id: 2},\n\t}\n\tweights := map[int]int{\n\t\t0: 5,\n\t\t1: 3,\n\t\t2: 1,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\tassert.Equal(t, 9, schedule.Len())\n\n\titer := schedule.NewIterator()\n\n\t// IWRR pattern for [5, 3, 1]:\n\t// Round 4: [0]         (weight 5 > 4)\n\t// Round 3: [0]         (weight 5 > 3)\n\t// Round 2: [0, 1]      (weights 5,3 > 2)\n\t// Round 1: [0, 1]      (weights 5,3 > 1)\n\t// Round 0: [0, 1, 2]   (weights 5,3,1 > 0)\n\t// Result: [0, 0, 0, 1, 0, 1, 0, 1, 2]\n\texpectedPattern := []int{0, 0, 0, 1, 0, 1, 0, 1, 2}\n\n\tfor i, expectedIdx := range expectedPattern {\n\t\titem, ok := iter.TryNext()\n\t\trequire.True(t, ok, \"iteration %d\", i)\n\t\tassert.Equal(t, items[expectedIdx], item, \"iteration %d\", i)\n\t}\n\n\t// Exhausted\n\titem, ok := iter.TryNext()\n\tassert.False(t, ok)\n\tassert.Nil(t, item)\n}\n\nfunc TestIWRRSchedule_StatelessSchedule_MultipleIterators(t *testing.T) {\n\t// Test that the schedule is stateless and can create multiple independent iterators\n\titem1 := &testWeightedItem{id: 0}\n\titem2 := &testWeightedItem{id: 1}\n\titems := map[int]*testWeightedItem{\n\t\t0: item1,\n\t\t1: item2,\n\t}\n\tweights := map[int]int{\n\t\t0: 2,\n\t\t1: 1,\n\t}\n\n\tschedule := newIWRRSchedule[int, *testWeightedItem](toWeightedMap(items, weights))\n\n\t// IWRR for weights [2, 1]: [item1, item1, item2]\n\t// Create first iterator and consume partially\n\titer1 := schedule.NewIterator()\n\ti1, ok1 := iter1.TryNext()\n\trequire.True(t, ok1)\n\tassert.Equal(t, item1, i1)\n\n\t// Create second iterator - should start from the beginning\n\titer2 := schedule.NewIterator()\n\ti2, ok2 := iter2.TryNext()\n\trequire.True(t, ok2)\n\tassert.Equal(t, item1, i2, \"second iterator should start from beginning\")\n\n\t// First iterator should continue from where it left off\n\ti1, ok1 = iter1.TryNext()\n\trequire.True(t, ok1)\n\tassert.Equal(t, item1, i1)\n\n\t// Second iterator should be independent and continue its own iteration\n\ti2, ok2 = iter2.TryNext()\n\trequire.True(t, ok2)\n\tassert.Equal(t, item1, i2)\n}\n"
  },
  {
    "path": "common/task/parallel_task_processor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// ParallelTaskProcessorOptions configs PriorityTaskProcessor\n\tParallelTaskProcessorOptions struct {\n\t\tQueueSize   int\n\t\tWorkerCount dynamicproperties.IntPropertyFn\n\t\tRetryPolicy backoff.RetryPolicy\n\t}\n\n\tparallelTaskProcessorImpl struct {\n\t\tstatus           int32\n\t\ttasksCh          chan Task\n\t\tshutdownCh       chan struct{}\n\t\tworkerShutdownCh []chan struct{}\n\t\tshutdownWG       sync.WaitGroup\n\t\tlogger           log.Logger\n\t\tmetricsScope     metrics.Scope\n\t\toptions          *ParallelTaskProcessorOptions\n\t}\n)\n\nconst (\n\tdefaultMonitorTickerDuration = 5 * time.Second\n)\n\nvar (\n\t// ErrTaskProcessorClosed is the error returned when submiting task to a stopped processor\n\tErrTaskProcessorClosed = errors.New(\"task processor has already shutdown\")\n)\n\n// NewParallelTaskProcessor creates a new PriorityTaskProcessor\nfunc NewParallelTaskProcessor(\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\toptions *ParallelTaskProcessorOptions,\n) Processor {\n\treturn &parallelTaskProcessorImpl{\n\t\tstatus:           common.DaemonStatusInitialized,\n\t\ttasksCh:          make(chan Task, options.QueueSize),\n\t\tshutdownCh:       make(chan struct{}),\n\t\tworkerShutdownCh: make([]chan struct{}, 0, options.WorkerCount()),\n\t\tlogger:           logger,\n\t\tmetricsScope:     metricsClient.Scope(metrics.ParallelTaskProcessingScope),\n\t\toptions:          options,\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tinitialWorkerCount := p.options.WorkerCount()\n\n\tp.shutdownWG.Add(initialWorkerCount)\n\tfor i := 0; i < initialWorkerCount; i++ {\n\t\tshutdownCh := make(chan struct{})\n\t\tp.workerShutdownCh = append(p.workerShutdownCh, shutdownCh)\n\t\tgo p.taskWorker(shutdownCh)\n\t}\n\n\tp.shutdownWG.Add(1)\n\tgo p.workerMonitor(defaultMonitorTickerDuration)\n\n\tp.logger.Info(\"Parallel task processor started.\")\n}\n\nfunc (p *parallelTaskProcessorImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(p.shutdownCh)\n\n\tp.drainAndNackTasks()\n\n\tif success := common.AwaitWaitGroup(&p.shutdownWG, time.Minute); !success {\n\t\tp.logger.Warn(\"Parallel task processor timedout on shutdown.\")\n\t}\n\tp.logger.Info(\"Parallel task processor shutdown.\")\n}\n\nfunc (p *parallelTaskProcessorImpl) Submit(task Task) error {\n\tp.metricsScope.IncCounter(metrics.ParallelTaskSubmitRequest)\n\tsw := p.metricsScope.StartTimer(metrics.ParallelTaskSubmitLatency)\n\tdefer sw.Stop()\n\n\tif p.isStopped() {\n\t\treturn ErrTaskProcessorClosed\n\t}\n\n\tselect {\n\tcase p.tasksCh <- task:\n\t\tif p.isStopped() {\n\t\t\tp.drainAndNackTasks()\n\t\t}\n\t\treturn nil\n\tcase <-p.shutdownCh:\n\t\treturn ErrTaskProcessorClosed\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) taskWorker(shutdownCh chan struct{}) {\n\tdefer p.shutdownWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-shutdownCh:\n\t\t\treturn\n\t\tcase task := <-p.tasksCh:\n\t\t\tp.executeTask(task, shutdownCh)\n\t\t}\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) executeTask(task Task, shutdownCh chan struct{}) {\n\tsw := p.metricsScope.StartTimer(metrics.ParallelTaskTaskProcessingLatency)\n\tdefer sw.Stop()\n\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tp.logger.Error(\"recovered panic in task execution\", tag.Dynamic(\"recovered-panic\", r))\n\t\t\ttask.HandleErr(fmt.Errorf(\"recovered panic: %v\", r))\n\t\t\ttask.Nack(nil)\n\t\t}\n\t}()\n\n\top := func(ctx context.Context) error {\n\t\tif err := task.Execute(); err != nil {\n\t\t\treturn task.HandleErr(err)\n\t\t}\n\t\treturn nil\n\t}\n\n\tisRetryable := func(err error) bool {\n\t\tselect {\n\t\tcase <-shutdownCh:\n\t\t\treturn false\n\t\tdefault:\n\t\t}\n\n\t\treturn task.RetryErr(err)\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(p.options.RetryPolicy),\n\t\tbackoff.WithRetryableError(isRetryable),\n\t)\n\n\tif err := throttleRetry.Do(context.Background(), op); err != nil {\n\t\t// non-retryable error or exhausted all retries or worker shutdown\n\t\ttask.Nack(err)\n\t\treturn\n\t}\n\n\t// no error\n\ttask.Ack()\n}\n\nfunc (p *parallelTaskProcessorImpl) workerMonitor(tickerDuration time.Duration) {\n\tdefer p.shutdownWG.Done()\n\n\tticker := time.NewTicker(tickerDuration)\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.shutdownCh:\n\t\t\tticker.Stop()\n\t\t\tp.removeWorker(len(p.workerShutdownCh))\n\t\t\treturn\n\t\tcase <-ticker.C:\n\t\t\ttargetWorkerCount := p.options.WorkerCount()\n\t\t\tcurrentWorkerCount := len(p.workerShutdownCh)\n\t\t\tp.addWorker(targetWorkerCount - currentWorkerCount)\n\t\t\tp.removeWorker(currentWorkerCount - targetWorkerCount)\n\t\t}\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) addWorker(count int) {\n\tfor i := 0; i < count; i++ {\n\t\tshutdownCh := make(chan struct{})\n\t\tp.workerShutdownCh = append(p.workerShutdownCh, shutdownCh)\n\n\t\tp.shutdownWG.Add(1)\n\t\tgo p.taskWorker(shutdownCh)\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) removeWorker(count int) {\n\tif count <= 0 {\n\t\treturn\n\t}\n\n\tcurrentWorkerCount := len(p.workerShutdownCh)\n\tif count > currentWorkerCount {\n\t\tcount = currentWorkerCount\n\t}\n\n\tshutdownChToClose := p.workerShutdownCh[currentWorkerCount-count:]\n\tp.workerShutdownCh = p.workerShutdownCh[:currentWorkerCount-count]\n\n\tfor _, shutdownCh := range shutdownChToClose {\n\t\tclose(shutdownCh)\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) drainAndNackTasks() {\n\tfor {\n\t\tselect {\n\t\tcase task := <-p.tasksCh:\n\t\t\ttask.Nack(nil)\n\t\tdefault:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *parallelTaskProcessorImpl) isStopped() bool {\n\treturn atomic.LoadInt32(&p.status) == common.DaemonStatusStopped\n}\n"
  },
  {
    "path": "common/task/parallel_task_processor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tparallelTaskProcessorSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tcontroller *gomock.Controller\n\n\t\tprocessor *parallelTaskProcessorImpl\n\t}\n)\n\nvar (\n\terrRetryable    = errors.New(\"retryable error\")\n\terrNonRetryable = errors.New(\"non-retryable error\")\n)\n\nfunc TestParallelTaskProcessorSuite(t *testing.T) {\n\ts := new(parallelTaskProcessorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *parallelTaskProcessorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.processor = NewParallelTaskProcessor(\n\t\ttestlogger.New(s.Suite.T()),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\t&ParallelTaskProcessorOptions{\n\t\t\tQueueSize:   0,\n\t\t\tWorkerCount: dynamicproperties.GetIntPropertyFn(1),\n\t\t\tRetryPolicy: backoff.NewExponentialRetryPolicy(time.Millisecond),\n\t\t},\n\t).(*parallelTaskProcessorImpl)\n}\n\nfunc (s *parallelTaskProcessorSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *parallelTaskProcessorSuite) TestSubmit_Success() {\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().Execute().Return(nil).MaxTimes(1)\n\tmockTask.EXPECT().Ack().MaxTimes(1)\n\ts.processor.Start()\n\terr := s.processor.Submit(mockTask)\n\ts.NoError(err)\n\ts.processor.Stop()\n}\n\nfunc (s *parallelTaskProcessorSuite) TestSubmit_Fail() {\n\tmockTask := NewMockTask(s.controller)\n\ts.processor.Start()\n\ts.processor.Stop()\n\terr := s.processor.Submit(mockTask)\n\ts.Equal(ErrTaskProcessorClosed, err)\n}\n\nfunc (s *parallelTaskProcessorSuite) TestTaskWorker() {\n\tnumTasks := 5\n\n\tdone := make(chan struct{})\n\n\ts.processor.shutdownWG.Add(1)\n\tworkerShutdownCh := make(chan struct{})\n\ts.processor.workerShutdownCh = append(s.processor.workerShutdownCh, workerShutdownCh)\n\n\tgo func() {\n\t\tfor i := 0; i != numTasks; i++ {\n\t\t\tmockTask := NewMockTask(s.controller)\n\t\t\tmockTask.EXPECT().Execute().Return(nil).Times(1)\n\t\t\tmockTask.EXPECT().Ack().Times(1)\n\t\t\terr := s.processor.Submit(mockTask)\n\t\t\ts.NoError(err)\n\t\t}\n\t\tclose(workerShutdownCh)\n\t\tclose(done)\n\t}()\n\n\ts.processor.taskWorker(workerShutdownCh)\n\t<-done\n}\n\nfunc (s *parallelTaskProcessorSuite) TestExecuteTask_RetryableError() {\n\tmockTask := NewMockTask(s.controller)\n\tgomock.InOrder(\n\t\tmockTask.EXPECT().Execute().Return(errRetryable),\n\t\tmockTask.EXPECT().HandleErr(errRetryable).Return(errRetryable),\n\t\tmockTask.EXPECT().RetryErr(errRetryable).Return(true),\n\t\tmockTask.EXPECT().Execute().Return(errRetryable),\n\t\tmockTask.EXPECT().HandleErr(errRetryable).Return(errRetryable),\n\t\tmockTask.EXPECT().RetryErr(errRetryable).Return(true),\n\t\tmockTask.EXPECT().Execute().Return(nil),\n\t\tmockTask.EXPECT().Ack(),\n\t)\n\n\ts.processor.executeTask(mockTask, make(chan struct{}))\n}\n\nfunc (s *parallelTaskProcessorSuite) TestExecuteTask_NonRetryableError() {\n\tmockTask := NewMockTask(s.controller)\n\tgomock.InOrder(\n\t\tmockTask.EXPECT().Execute().Return(errNonRetryable),\n\t\tmockTask.EXPECT().HandleErr(errNonRetryable).Return(errNonRetryable),\n\t\tmockTask.EXPECT().RetryErr(errNonRetryable).Return(false).AnyTimes(),\n\t\tmockTask.EXPECT().Nack(gomock.Any()),\n\t)\n\n\ts.processor.executeTask(mockTask, make(chan struct{}))\n}\n\nfunc (s *parallelTaskProcessorSuite) TestExecuteTask_WorkerStopped() {\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().Execute().Return(errRetryable).AnyTimes()\n\tmockTask.EXPECT().HandleErr(errRetryable).Return(errRetryable).AnyTimes()\n\tmockTask.EXPECT().RetryErr(errRetryable).Return(true).AnyTimes()\n\tmockTask.EXPECT().Nack(gomock.Any()).Times(1)\n\n\tdone := make(chan struct{})\n\tworkerShutdownCh := make(chan struct{})\n\tgo func() {\n\t\ts.processor.executeTask(mockTask, workerShutdownCh)\n\t\tclose(done)\n\t}()\n\n\tclose(workerShutdownCh)\n\t<-done\n}\n\nfunc (s *parallelTaskProcessorSuite) TestAddWorker() {\n\tnewWorkerCount := 10\n\n\ts.processor.addWorker(newWorkerCount)\n\ts.Len(s.processor.workerShutdownCh, newWorkerCount)\n\n\ts.processor.addWorker(-1)\n\ts.Len(s.processor.workerShutdownCh, newWorkerCount)\n\n\tfor _, shutdownCh := range s.processor.workerShutdownCh {\n\t\tclose(shutdownCh)\n\t}\n\n\ts.processor.shutdownWG.Wait()\n}\n\nfunc (s *parallelTaskProcessorSuite) TestRemoveWorker() {\n\tcurrentWorkerCount := 10\n\tremoveWorkerCount := 4\n\n\ts.processor.addWorker(currentWorkerCount)\n\n\ts.processor.removeWorker(removeWorkerCount)\n\ts.Len(s.processor.workerShutdownCh, currentWorkerCount-removeWorkerCount)\n\n\ts.processor.removeWorker(-1)\n\ts.Len(s.processor.workerShutdownCh, currentWorkerCount-removeWorkerCount)\n\n\tfor _, shutdownCh := range s.processor.workerShutdownCh {\n\t\tclose(shutdownCh)\n\t}\n\n\ts.processor.shutdownWG.Wait()\n}\n\nfunc (s *parallelTaskProcessorSuite) TestMonitor() {\n\tworkerCount := 5\n\n\ts.processor.shutdownWG.Add(1) // for monitor\n\tdcClient := dynamicconfig.NewInMemoryClient()\n\terr := dcClient.UpdateValue(dynamicproperties.TaskSchedulerWorkerCount, workerCount)\n\ts.NoError(err)\n\tdcCollection := dynamicconfig.NewCollection(dcClient, s.processor.logger)\n\ts.processor.options.WorkerCount = dcCollection.GetIntProperty(dynamicproperties.TaskSchedulerWorkerCount)\n\n\ttestMonitorTickerDuration := 100 * time.Millisecond\n\tgo s.processor.workerMonitor(testMonitorTickerDuration)\n\n\ttime.Sleep(2 * testMonitorTickerDuration)\n\t// note we can't check the length of the workerShutdownCh directly\n\t// as that will lead to race condition. Instead we check the current\n\t// size of shutdownWG chan, which should be workerCount + 1\n\tfor i := 0; i != workerCount+1; i++ {\n\t\ts.processor.shutdownWG.Done()\n\t}\n\ts.processor.shutdownWG.Wait()\n\ts.processor.shutdownWG.Add(workerCount + 1)\n\n\tnewWorkerCount := 3\n\terr = dcClient.UpdateValue(dynamicproperties.TaskSchedulerWorkerCount, newWorkerCount)\n\ts.NoError(err)\n\n\ttime.Sleep(2 * testMonitorTickerDuration)\n\tfor i := 0; i != newWorkerCount+1; i++ {\n\t\ts.processor.shutdownWG.Done()\n\t}\n\ts.processor.shutdownWG.Wait()\n\ts.processor.shutdownWG.Add(newWorkerCount + 1)\n\n\tclose(s.processor.shutdownCh)\n\n\ttime.Sleep(2 * testMonitorTickerDuration)\n\ts.processor.shutdownWG.Wait()\n}\n\nfunc (s *parallelTaskProcessorSuite) TestProcessorContract() {\n\tnumTasks := 10000\n\tvar taskWG sync.WaitGroup\n\n\ttasks := []Task{}\n\ttaskStatusLock := &sync.Mutex{}\n\ttaskStatus := make(map[Task]State)\n\tfor i := 0; i != numTasks; i++ {\n\t\tmockTask := NewMockTask(s.controller)\n\t\ttaskStatus[mockTask] = TaskStatePending\n\t\tmockTask.EXPECT().Execute().Return(nil).MaxTimes(1)\n\t\tmockTask.EXPECT().Ack().Do(func() {\n\t\t\ttaskStatusLock.Lock()\n\t\t\tdefer taskStatusLock.Unlock()\n\n\t\t\ts.Equal(TaskStatePending, taskStatus[mockTask])\n\t\t\ttaskStatus[mockTask] = TaskStateAcked\n\t\t\ttaskWG.Done()\n\t\t}).MaxTimes(1)\n\t\tmockTask.EXPECT().Nack(gomock.Any()).Do(func(err error) {\n\t\t\ttaskStatusLock.Lock()\n\t\t\tdefer taskStatusLock.Unlock()\n\n\t\t\ts.Equal(TaskStatePending, taskStatus[mockTask])\n\t\t\ttaskStatus[mockTask] = State(-1) // set it to whatever state that is not TaskStatePending\n\t\t\ttaskWG.Done()\n\t\t}).MaxTimes(1)\n\t\ttasks = append(tasks, mockTask)\n\t\ttaskWG.Add(1)\n\t}\n\n\tprocessor := NewParallelTaskProcessor(\n\t\ttestlogger.New(s.Suite.T()),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\t&ParallelTaskProcessorOptions{\n\t\t\tQueueSize:   100,\n\t\t\tWorkerCount: dynamicproperties.GetIntPropertyFn(10),\n\t\t\tRetryPolicy: backoff.NewExponentialRetryPolicy(time.Millisecond),\n\t\t},\n\t).(*parallelTaskProcessorImpl)\n\n\tprocessor.Start()\n\tgo func() {\n\t\ttime.Sleep(time.Duration(rand.Intn(50)) * time.Millisecond)\n\t\tprocessor.Stop()\n\t}()\n\n\tfor _, task := range tasks {\n\t\tif err := processor.Submit(task); err != nil {\n\t\t\ttaskWG.Done()\n\t\t\ttaskStatusLock.Lock()\n\t\t\tdelete(taskStatus, task)\n\t\t\ttaskStatusLock.Unlock()\n\t\t}\n\t}\n\ts.True(common.AwaitWaitGroup(&taskWG, 10*time.Second))\n\t<-processor.shutdownCh\n\n\tfor _, status := range taskStatus {\n\t\ts.NotEqual(TaskStatePending, status)\n\t}\n}\n\nfunc (s *parallelTaskProcessorSuite) TestExecuteTask_PanicHandling() {\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().Execute().Do(func() {\n\t\tpanic(\"A panic occurred\")\n\t})\n\tmockTask.EXPECT().HandleErr(gomock.Any()).Return(errRetryable).AnyTimes()\n\tmockTask.EXPECT().Nack(gomock.Any()).Times(1)\n\tdone := make(chan struct{})\n\tworkerShutdownCh := make(chan struct{})\n\tgo func() {\n\t\ts.processor.executeTask(mockTask, workerShutdownCh)\n\t\tclose(done)\n\t}()\n\tclose(workerShutdownCh)\n\t<-done\n}\n"
  },
  {
    "path": "common/task/scheduler_options.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype SchedulerOptions[K comparable, T Task] struct {\n\tSchedulerType        SchedulerType\n\tFIFOSchedulerOptions *FIFOTaskSchedulerOptions\n\tWRRSchedulerOptions  *WeightedRoundRobinTaskSchedulerOptions[K, T]\n}\n\nfunc NewSchedulerOptions[K comparable, T Task](\n\tschedulerType int,\n\tqueueSize int,\n\tworkerCount dynamicproperties.IntPropertyFn,\n\tdispatcherCount int,\n\ttaskToChannelKeyFn func(T) K,\n\tchannelKeyToWeightFn func(K) int,\n) (*SchedulerOptions[K, T], error) {\n\toptions := &SchedulerOptions[K, T]{\n\t\tSchedulerType: SchedulerType(schedulerType),\n\t}\n\tswitch options.SchedulerType {\n\tcase SchedulerTypeFIFO:\n\t\toptions.FIFOSchedulerOptions = &FIFOTaskSchedulerOptions{\n\t\t\tQueueSize:       queueSize,\n\t\t\tWorkerCount:     workerCount,\n\t\t\tDispatcherCount: dispatcherCount,\n\t\t\tRetryPolicy:     common.CreateTaskProcessingRetryPolicy(),\n\t\t}\n\tcase SchedulerTypeWRR:\n\t\toptions.WRRSchedulerOptions = &WeightedRoundRobinTaskSchedulerOptions[K, T]{\n\t\t\tQueueSize:            queueSize,\n\t\t\tDispatcherCount:      dispatcherCount,\n\t\t\tTaskToChannelKeyFn:   taskToChannelKeyFn,\n\t\t\tChannelKeyToWeightFn: channelKeyToWeightFn,\n\t\t}\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown task scheduler type: %v\", schedulerType)\n\t}\n\treturn options, nil\n}\n\nfunc (o *SchedulerOptions[K, T]) String() string {\n\treturn fmt.Sprintf(\"{schedulerType:%v, fifoSchedulerOptions:%s, wrrSchedulerOptions:%s}\",\n\t\to.SchedulerType, o.FIFOSchedulerOptions, o.WRRSchedulerOptions)\n}\n"
  },
  {
    "path": "common/task/scheduler_options_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\nfunc TestSchedulerOptionsString(t *testing.T) {\n\ttests := []struct {\n\t\tdesc            string\n\t\tschedulerType   int\n\t\tqueueSize       int\n\t\tworkerCount     dynamicproperties.IntPropertyFn\n\t\tdispatcherCount int\n\t\twantErr         bool\n\t\twant            string\n\t}{\n\t\t{\n\t\t\tdesc:            \"FIFO\",\n\t\t\tschedulerType:   int(SchedulerTypeFIFO),\n\t\t\tqueueSize:       1,\n\t\t\tworkerCount:     dynamicproperties.GetIntPropertyFn(3),\n\t\t\tdispatcherCount: 1,\n\t\t\twant:            \"{schedulerType:1, fifoSchedulerOptions:{QueueSize: 1, WorkerCount: 3, DispatcherCount: 1}, wrrSchedulerOptions:<nil>}\",\n\t\t},\n\t\t{\n\t\t\tdesc:            \"WRR\",\n\t\t\tschedulerType:   int(SchedulerTypeWRR),\n\t\t\tqueueSize:       3,\n\t\t\tworkerCount:     dynamicproperties.GetIntPropertyFn(4),\n\t\t\tdispatcherCount: 5,\n\t\t\twant:            \"{schedulerType:2, fifoSchedulerOptions:<nil>, wrrSchedulerOptions:{QueueSize: 3, DispatcherCount: 5}}\",\n\t\t},\n\t\t{\n\t\t\tdesc:          \"InvalidSchedulerType\",\n\t\t\tschedulerType: 3,\n\t\t\twantErr:       true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\to, err := NewSchedulerOptions[int, PriorityTask](tc.schedulerType, tc.queueSize, tc.workerCount, tc.dispatcherCount, nil, nil)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif got := o.String(); got != tc.want {\n\t\t\t\tt.Errorf(\"Got: %v, want: %v\", got, tc.want)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/task/sequential_task_processor.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tsequentialTaskProcessorImpl struct {\n\t\tstatus       int32\n\t\tshutdownChan chan struct{}\n\t\twaitGroup    sync.WaitGroup\n\n\t\tcoroutineSize    int\n\t\ttaskqueues       collection.ConcurrentTxMap\n\t\ttaskQueueFactory SequentialTaskQueueFactory\n\t\ttaskqueueChan    chan SequentialTaskQueue\n\n\t\tmetricsScope metrics.Scope\n\t\tlogger       log.Logger\n\t}\n)\n\n// NewSequentialTaskProcessor create a new sequential tasks processor\nfunc NewSequentialTaskProcessor(\n\tcoroutineSize int,\n\ttaskQueueHashFn collection.HashFunc,\n\ttaskQueueFactory SequentialTaskQueueFactory,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) Processor {\n\n\treturn &sequentialTaskProcessorImpl{\n\t\tstatus:           common.DaemonStatusInitialized,\n\t\tshutdownChan:     make(chan struct{}),\n\t\tcoroutineSize:    coroutineSize,\n\t\ttaskqueues:       collection.NewShardedConcurrentTxMap(1024, taskQueueHashFn),\n\t\ttaskQueueFactory: taskQueueFactory,\n\t\ttaskqueueChan:    make(chan SequentialTaskQueue, coroutineSize),\n\t\tmetricsScope:     metricsClient.Scope(metrics.SequentialTaskProcessingScope),\n\t\tlogger:           logger,\n\t}\n}\n\nfunc (t *sequentialTaskProcessorImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tt.waitGroup.Add(t.coroutineSize)\n\tfor i := 0; i < t.coroutineSize; i++ {\n\t\tgo t.pollAndProcessTaskQueue()\n\t}\n\tt.logger.Info(\"Task processor started.\")\n}\n\nfunc (t *sequentialTaskProcessorImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(t.shutdownChan)\n\tif success := common.AwaitWaitGroup(&t.waitGroup, time.Minute); !success {\n\t\tt.logger.Warn(\"Task processor timeout trying to stop.\")\n\t}\n\tt.logger.Info(\"Task processor stopped.\")\n}\n\nfunc (t *sequentialTaskProcessorImpl) Submit(task Task) error {\n\n\tt.metricsScope.IncCounter(metrics.SequentialTaskSubmitRequest)\n\tmetricsTimer := t.metricsScope.StartTimer(metrics.SequentialTaskSubmitLatency)\n\tdefer metricsTimer.Stop()\n\n\ttaskqueue := t.taskQueueFactory(task)\n\ttaskqueue.Add(task)\n\n\t_, fnEvaluated, err := t.taskqueues.PutOrDo(\n\t\ttaskqueue.QueueID(),\n\t\ttaskqueue,\n\t\tfunc(key interface{}, value interface{}) error {\n\t\t\tvalue.(SequentialTaskQueue).Add(task)\n\t\t\treturn nil\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// if function evaluated, meaning that the task set is\n\t// already dispatched\n\tif fnEvaluated {\n\t\tt.metricsScope.IncCounter(metrics.SequentialTaskSubmitRequestTaskQueueExist)\n\t\treturn nil\n\t}\n\n\t// need to dispatch this task set\n\tt.metricsScope.IncCounter(metrics.SequentialTaskSubmitRequestTaskQueueMissing)\n\tselect {\n\tcase <-t.shutdownChan:\n\tcase t.taskqueueChan <- taskqueue:\n\t}\n\treturn nil\n\n}\n\nfunc (t *sequentialTaskProcessorImpl) pollAndProcessTaskQueue() {\n\tdefer t.waitGroup.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownChan:\n\t\t\treturn\n\t\tcase taskqueue := <-t.taskqueueChan:\n\t\t\tmetricsTimer := t.metricsScope.StartTimer(metrics.SequentialTaskQueueProcessingLatency)\n\t\t\tt.processTaskQueue(taskqueue)\n\t\t\tmetricsTimer.Stop()\n\t\t}\n\t}\n}\n\nfunc (t *sequentialTaskProcessorImpl) processTaskQueue(taskqueue SequentialTaskQueue) {\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownChan:\n\t\t\treturn\n\t\tdefault:\n\t\t\tqueueSize := taskqueue.Len()\n\t\t\tt.metricsScope.RecordTimer(metrics.SequentialTaskQueueSize, time.Duration(queueSize))\n\t\t\tif queueSize > 0 {\n\t\t\t\tt.processTaskOnce(taskqueue)\n\t\t\t}\n\t\t\tif taskqueue.IsEmpty() {\n\t\t\t\tdeleted := t.taskqueues.RemoveIf(taskqueue.QueueID(), func(key interface{}, value interface{}) bool {\n\t\t\t\t\treturn value.(SequentialTaskQueue).IsEmpty()\n\t\t\t\t})\n\t\t\t\tif deleted {\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\t// if deletion failed, meaning that task queue is offered with new task\n\t\t\t\t// continue execution\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (t *sequentialTaskProcessorImpl) processTaskOnce(taskqueue SequentialTaskQueue) {\n\tmetricsTimer := t.metricsScope.StartTimer(metrics.SequentialTaskTaskProcessingLatency)\n\tdefer metricsTimer.Stop()\n\n\ttask := taskqueue.Remove()\n\terr := task.Execute()\n\terr = task.HandleErr(err)\n\n\tif err != nil {\n\t\tif task.RetryErr(err) {\n\t\t\ttaskqueue.Add(task)\n\t\t} else {\n\t\t\tt.logger.Error(\"Unable to process task\", tag.Error(err))\n\t\t\ttask.Nack(err)\n\t\t}\n\t} else {\n\t\ttask.Ack()\n\t}\n}\n"
  },
  {
    "path": "common/task/sequential_task_processor_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tSequentialTaskProcessorSuite struct {\n\t\tsuite.Suite\n\t\tprocessor Processor\n\t}\n\n\ttestSequentialTaskQueueImpl struct {\n\t\tid        uint32\n\t\ttaskQueue collection.Queue[Task]\n\t}\n\n\ttestSequentialTaskImpl struct {\n\t\twaitgroup *sync.WaitGroup\n\t\tqueueID   uint32\n\t\ttaskID    uint32\n\n\t\tlock   sync.Mutex\n\t\tacked  int\n\t\tnacked int\n\t}\n)\n\nfunc TestSequentialTaskProcessorSuite(t *testing.T) {\n\tsuite.Run(t, new(SequentialTaskProcessorSuite))\n}\n\nfunc (s *SequentialTaskProcessorSuite) SetupTest() {\n\tlogger := testlogger.New(s.T())\n\ts.processor = NewSequentialTaskProcessor(\n\t\t20,\n\t\tfunc(key interface{}) uint32 {\n\t\t\treturn key.(uint32)\n\t\t},\n\t\tfunc(task Task) SequentialTaskQueue {\n\t\t\ttaskQueue := collection.NewConcurrentPriorityQueue(func(this Task, other Task) bool {\n\t\t\t\treturn this.(*testSequentialTaskImpl).taskID < other.(*testSequentialTaskImpl).taskID\n\t\t\t})\n\n\t\t\treturn &testSequentialTaskQueueImpl{\n\t\t\t\tid:        task.(*testSequentialTaskImpl).queueID,\n\t\t\t\ttaskQueue: taskQueue,\n\t\t\t}\n\t\t},\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\tlogger,\n\t)\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestSubmit_NoPriorTask() {\n\twaitgroup := &sync.WaitGroup{}\n\twaitgroup.Add(1)\n\ttask := newTestSequentialTaskImpl(waitgroup, 4, uint32(1))\n\n\t// do not start the processor\n\ts.Nil(s.processor.Submit(task))\n\tsequentialTaskQueue := <-s.processor.(*sequentialTaskProcessorImpl).taskqueueChan\n\tsequentialTask := sequentialTaskQueue.Remove()\n\ts.True(sequentialTaskQueue.IsEmpty())\n\ts.Equal(task, sequentialTask)\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestSubmit_HasPriorTask() {\n\twaitgroup := &sync.WaitGroup{}\n\ttask1 := newTestSequentialTaskImpl(waitgroup, 4, uint32(1))\n\ttask2 := newTestSequentialTaskImpl(waitgroup, 4, uint32(2))\n\n\t// do not start the processor\n\ts.Nil(s.processor.Submit(task1))\n\ts.Nil(s.processor.Submit(task2))\n\tsequentialTaskQueue := <-s.processor.(*sequentialTaskProcessorImpl).taskqueueChan\n\tsequentialTask1 := sequentialTaskQueue.Remove()\n\tsequentialTask2 := sequentialTaskQueue.Remove()\n\ts.True(sequentialTaskQueue.IsEmpty())\n\ts.Equal(task1, sequentialTask1)\n\ts.Equal(task2, sequentialTask2)\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestProcessTaskQueue_ShutDown() {\n\twaitgroup := &sync.WaitGroup{}\n\twaitgroup.Add(2)\n\ttask1 := newTestSequentialTaskImpl(waitgroup, 4, uint32(1))\n\ttask2 := newTestSequentialTaskImpl(waitgroup, 4, uint32(2))\n\n\t// do not start the processor\n\ts.Nil(s.processor.Submit(task1))\n\ts.Nil(s.processor.Submit(task2))\n\tsequentialTaskQueue := <-s.processor.(*sequentialTaskProcessorImpl).taskqueueChan\n\n\ts.processor.Start()\n\ts.processor.Stop()\n\ts.processor.(*sequentialTaskProcessorImpl).processTaskQueue(sequentialTaskQueue)\n\n\ts.Equal(0, task1.NumAcked())\n\ts.Equal(0, task1.NumNcked())\n\ts.Equal(0, task2.NumAcked())\n\ts.Equal(0, task2.NumNcked())\n\ts.Equal(1, s.processor.(*sequentialTaskProcessorImpl).taskqueues.Len())\n\ts.Equal(2, sequentialTaskQueue.Len())\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestProcessTaskQueue() {\n\twaitgroup := &sync.WaitGroup{}\n\twaitgroup.Add(2)\n\ttask1 := newTestSequentialTaskImpl(waitgroup, 4, uint32(1))\n\ttask2 := newTestSequentialTaskImpl(waitgroup, 4, uint32(2))\n\n\t// do not start the processor\n\ts.Nil(s.processor.Submit(task1))\n\ts.Nil(s.processor.Submit(task2))\n\tsequentialTaskQueue := <-s.processor.(*sequentialTaskProcessorImpl).taskqueueChan\n\n\ts.processor.(*sequentialTaskProcessorImpl).processTaskQueue(sequentialTaskQueue)\n\twaitgroup.Wait()\n\n\ts.Equal(1, task1.NumAcked())\n\ts.Equal(0, task1.NumNcked())\n\ts.Equal(1, task2.NumAcked())\n\ts.Equal(0, task2.NumNcked())\n\ts.Equal(0, s.processor.(*sequentialTaskProcessorImpl).taskqueues.Len())\n\ts.Equal(0, sequentialTaskQueue.Len())\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestSequentialTaskProcessing() {\n\tnumTasks := 100\n\twaitgroup := &sync.WaitGroup{}\n\twaitgroup.Add(numTasks)\n\n\ttasks := []*testSequentialTaskImpl{}\n\tfor i := 0; i < numTasks; i++ {\n\t\ttasks = append(tasks, newTestSequentialTaskImpl(waitgroup, 4, uint32(i)))\n\t}\n\n\ts.processor.Start()\n\tfor _, task := range tasks {\n\t\ts.Nil(s.processor.Submit(task))\n\t}\n\twaitgroup.Wait()\n\ts.processor.Stop()\n\n\tfor _, task := range tasks {\n\t\ts.Equal(1, task.NumAcked())\n\t\ts.Equal(0, task.NumNcked())\n\t}\n\ts.Equal(0, s.processor.(*sequentialTaskProcessorImpl).taskqueues.Len())\n}\n\nfunc (s *SequentialTaskProcessorSuite) TestRandomizedTaskProcessing() {\n\tnumQueues := 100\n\tnumTasks := 1000\n\twaitgroup := &sync.WaitGroup{}\n\twaitgroup.Add(numQueues * numTasks)\n\n\ttasks := make([][]*testSequentialTaskImpl, numQueues)\n\tfor i := 0; i < numQueues; i++ {\n\t\ttasks[i] = make([]*testSequentialTaskImpl, numTasks)\n\n\t\tfor j := 0; j < numTasks; j++ {\n\t\t\ttasks[i][j] = newTestSequentialTaskImpl(waitgroup, uint32(i), uint32(j))\n\t\t}\n\n\t\trandomize(tasks[i])\n\t}\n\n\ts.processor.Start()\n\tstartChan := make(chan struct{})\n\tfor i := 0; i < numQueues; i++ {\n\t\tgo func(i int) {\n\t\t\t<-startChan\n\n\t\t\tfor j := 0; j < numTasks; j++ {\n\t\t\t\ts.Nil(s.processor.Submit(tasks[i][j]))\n\t\t\t}\n\t\t}(i)\n\t}\n\tclose(startChan)\n\twaitgroup.Wait()\n\ts.processor.Stop()\n\n\tfor i := 0; i < numQueues; i++ {\n\t\tfor j := 0; j < numTasks; j++ {\n\t\t\ttask := tasks[i][j]\n\t\t\ts.Equal(1, task.NumAcked())\n\t\t\ts.Equal(0, task.NumNcked())\n\t\t}\n\t}\n\ts.Equal(0, s.processor.(*sequentialTaskProcessorImpl).taskqueues.Len())\n}\n\nfunc randomize(array []*testSequentialTaskImpl) {\n\tfor i := 0; i < len(array); i++ {\n\t\tindex := rand.Int31n(int32(i) + 1)\n\t\tarray[i], array[index] = array[index], array[i]\n\t}\n}\n\nfunc newTestSequentialTaskImpl(waitgroup *sync.WaitGroup, queueID uint32, taskID uint32) *testSequentialTaskImpl {\n\treturn &testSequentialTaskImpl{\n\t\twaitgroup: waitgroup,\n\t\tqueueID:   queueID,\n\t\ttaskID:    taskID,\n\t}\n}\n\nfunc (t *testSequentialTaskImpl) Execute() error {\n\tif rand.Float64() < 0.5 {\n\t\treturn nil\n\t}\n\n\treturn errors.New(\"some random error\")\n}\n\nfunc (t *testSequentialTaskImpl) HandleErr(err error) error {\n\treturn err\n}\n\nfunc (t *testSequentialTaskImpl) RetryErr(err error) bool {\n\treturn true\n}\n\nfunc (t *testSequentialTaskImpl) State() State {\n\tif t.acked > 0 {\n\t\treturn TaskStateAcked\n\t}\n\n\treturn TaskStatePending\n}\n\nfunc (t *testSequentialTaskImpl) Ack() {\n\tt.lock.Lock()\n\tdefer t.lock.Unlock()\n\n\tt.acked++\n\tt.waitgroup.Done()\n}\n\nfunc (t *testSequentialTaskImpl) NumAcked() int {\n\tt.lock.Lock()\n\tdefer t.lock.Unlock()\n\n\treturn t.acked\n}\n\nfunc (t *testSequentialTaskImpl) Nack(err error) {\n\tt.lock.Lock()\n\tdefer t.lock.Unlock()\n\n\tt.nacked++\n\tt.waitgroup.Done()\n}\n\nfunc (t *testSequentialTaskImpl) Cancel() {\n}\n\nfunc (t *testSequentialTaskImpl) NumNcked() int {\n\tt.lock.Lock()\n\tdefer t.lock.Unlock()\n\n\treturn t.nacked\n}\n\nfunc (t *testSequentialTaskQueueImpl) QueueID() interface{} {\n\treturn t.id\n}\n\nfunc (t *testSequentialTaskQueueImpl) Add(task Task) {\n\tt.taskQueue.Add(task)\n}\n\nfunc (t *testSequentialTaskQueueImpl) Remove() Task {\n\titem, _ := t.taskQueue.Remove()\n\treturn item.(Task)\n}\n\nfunc (t *testSequentialTaskQueueImpl) IsEmpty() bool {\n\treturn t.taskQueue.IsEmpty()\n}\n\nfunc (t *testSequentialTaskQueueImpl) Len() int {\n\treturn t.taskQueue.Len()\n}\n"
  },
  {
    "path": "common/task/ttl_channel.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n)\n\n// TTLChannel is a channel that can expire if it is not written to for a given amount of time.\ntype TTLChannel[V any] struct {\n\tc             chan V\n\tlastWriteTime atomic.Int64\n\trefCount      atomic.Int32\n}\n\nfunc NewTTLChannel[V any](bufferSize int) *TTLChannel[V] {\n\treturn &TTLChannel[V]{\n\t\tc: make(chan V, bufferSize),\n\t}\n}\n\nfunc (c *TTLChannel[V]) IncRef() {\n\tc.refCount.Add(1)\n}\n\nfunc (c *TTLChannel[V]) DecRef() {\n\tc.refCount.Add(-1)\n}\n\nfunc (c *TTLChannel[V]) RefCount() int32 {\n\treturn c.refCount.Load()\n}\n\nfunc (c *TTLChannel[V]) LastWriteTime() time.Time {\n\treturn time.Unix(c.lastWriteTime.Load(), 0)\n}\n\nfunc (c *TTLChannel[V]) UpdateLastWriteTime(now time.Time) {\n\tc.lastWriteTime.Store(now.Unix())\n}\n\nfunc (c *TTLChannel[V]) Chan() chan V {\n\treturn c.c\n}\n\nfunc (c *TTLChannel[V]) Len() int {\n\treturn len(c.c)\n}\n\nfunc (c *TTLChannel[V]) Cap() int {\n\treturn cap(c.c)\n}\n\nfunc (c *TTLChannel[V]) ShouldCleanup(now time.Time, ttl time.Duration) bool {\n\treturn now.Sub(c.LastWriteTime()) > ttl && c.Len() == 0 && c.RefCount() == 0\n}\n"
  },
  {
    "path": "common/task/ttl_channel_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestNewTTLChannel(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tbufferSize int\n\t}{\n\t\t{\n\t\t\tname:       \"unbuffered channel\",\n\t\t\tbufferSize: 0,\n\t\t},\n\t\t{\n\t\t\tname:       \"small buffer\",\n\t\t\tbufferSize: 10,\n\t\t},\n\t\t{\n\t\t\tname:       \"large buffer\",\n\t\t\tbufferSize: 1000,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tch := NewTTLChannel[int](tt.bufferSize)\n\t\t\trequire.NotNil(t, ch)\n\t\t\tassert.NotNil(t, ch.Chan())\n\t\t\tassert.Equal(t, tt.bufferSize, ch.Cap())\n\t\t\tassert.Equal(t, 0, ch.Len())\n\t\t\tassert.Equal(t, int32(0), ch.RefCount())\n\t\t})\n\t}\n}\n\nfunc TestTTLChannel_RefCount(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\n\t// Initial ref count should be 0\n\tassert.Equal(t, int32(0), ch.RefCount())\n\n\t// Add refs\n\tch.IncRef()\n\tassert.Equal(t, int32(1), ch.RefCount())\n\n\tch.IncRef()\n\tassert.Equal(t, int32(2), ch.RefCount())\n\n\tch.IncRef()\n\tassert.Equal(t, int32(3), ch.RefCount())\n\n\t// Decrement refs\n\tch.DecRef()\n\tassert.Equal(t, int32(2), ch.RefCount())\n\n\tch.DecRef()\n\tassert.Equal(t, int32(1), ch.RefCount())\n\n\tch.DecRef()\n\tassert.Equal(t, int32(0), ch.RefCount())\n}\n\nfunc TestTTLChannel_RefCount_MultipleAddDec(t *testing.T) {\n\tch := NewTTLChannel[string](5)\n\n\t// Add multiple refs\n\tfor i := 0; i < 100; i++ {\n\t\tch.IncRef()\n\t}\n\tassert.Equal(t, int32(100), ch.RefCount())\n\n\t// Decrement all refs\n\tfor i := 0; i < 100; i++ {\n\t\tch.DecRef()\n\t}\n\tassert.Equal(t, int32(0), ch.RefCount())\n}\n\nfunc TestTTLChannel_LastWriteTime(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\n\t// Initial last write time should be zero (Unix epoch)\n\tassert.Equal(t, time.Unix(0, 0), ch.LastWriteTime())\n\n\t// Update last write time\n\tnow := time.Now()\n\tch.UpdateLastWriteTime(now)\n\n\t// Should return the updated time (truncated to seconds)\n\texpected := time.Unix(now.Unix(), 0)\n\tassert.Equal(t, expected, ch.LastWriteTime())\n\n\t// Update with a different time\n\tlater := now.Add(1 * time.Hour)\n\tch.UpdateLastWriteTime(later)\n\n\texpected = time.Unix(later.Unix(), 0)\n\tassert.Equal(t, expected, ch.LastWriteTime())\n}\n\nfunc TestTTLChannel_Chan(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\n\t// Should return the underlying channel\n\tc := ch.Chan()\n\trequire.NotNil(t, c)\n\n\t// Should be able to send and receive\n\tc <- 42\n\tval := <-c\n\tassert.Equal(t, 42, val)\n}\n\nfunc TestTTLChannel_Len(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\n\t// Initially empty\n\tassert.Equal(t, 0, ch.Len())\n\n\t// Add items\n\tch.Chan() <- 1\n\tassert.Equal(t, 1, ch.Len())\n\n\tch.Chan() <- 2\n\tassert.Equal(t, 2, ch.Len())\n\n\tch.Chan() <- 3\n\tassert.Equal(t, 3, ch.Len())\n\n\t// Remove items\n\t<-ch.Chan()\n\tassert.Equal(t, 2, ch.Len())\n\n\t<-ch.Chan()\n\tassert.Equal(t, 1, ch.Len())\n\n\t<-ch.Chan()\n\tassert.Equal(t, 0, ch.Len())\n}\n\nfunc TestTTLChannel_Cap(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tbufferSize int\n\t}{\n\t\t{\n\t\t\tname:       \"unbuffered\",\n\t\t\tbufferSize: 0,\n\t\t},\n\t\t{\n\t\t\tname:       \"buffered 1\",\n\t\t\tbufferSize: 1,\n\t\t},\n\t\t{\n\t\t\tname:       \"buffered 100\",\n\t\t\tbufferSize: 100,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tch := NewTTLChannel[int](tt.bufferSize)\n\t\t\tassert.Equal(t, tt.bufferSize, ch.Cap())\n\t\t})\n\t}\n}\n\nfunc TestTTLChannel_ShouldCleanup(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\trefCount       int32\n\t\tchannelLen     int\n\t\ttimeSinceWrite time.Duration\n\t\tttl            time.Duration\n\t\texpected       bool\n\t}{\n\t\t{\n\t\t\tname:           \"should cleanup - all conditions met\",\n\t\t\trefCount:       0,\n\t\t\tchannelLen:     0,\n\t\t\ttimeSinceWrite: 2 * time.Hour,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       true,\n\t\t},\n\t\t{\n\t\t\tname:           \"should not cleanup - has references\",\n\t\t\trefCount:       1,\n\t\t\tchannelLen:     0,\n\t\t\ttimeSinceWrite: 2 * time.Hour,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       false,\n\t\t},\n\t\t{\n\t\t\tname:           \"should not cleanup - channel not empty\",\n\t\t\trefCount:       0,\n\t\t\tchannelLen:     1,\n\t\t\ttimeSinceWrite: 2 * time.Hour,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       false,\n\t\t},\n\t\t{\n\t\t\tname:           \"should not cleanup - not expired\",\n\t\t\trefCount:       0,\n\t\t\tchannelLen:     0,\n\t\t\ttimeSinceWrite: 30 * time.Minute,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       false,\n\t\t},\n\t\t{\n\t\t\tname:           \"should not cleanup - just before TTL boundary\",\n\t\t\trefCount:       0,\n\t\t\tchannelLen:     0,\n\t\t\ttimeSinceWrite: 1*time.Hour - 1*time.Second,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       false,\n\t\t},\n\t\t{\n\t\t\tname:           \"should cleanup - slightly past TTL\",\n\t\t\trefCount:       0,\n\t\t\tchannelLen:     0,\n\t\t\ttimeSinceWrite: 1*time.Hour + 1*time.Second,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       true,\n\t\t},\n\t\t{\n\t\t\tname:           \"should not cleanup - multiple issues\",\n\t\t\trefCount:       2,\n\t\t\tchannelLen:     3,\n\t\t\ttimeSinceWrite: 30 * time.Minute,\n\t\t\tttl:            1 * time.Hour,\n\t\t\texpected:       false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tch := NewTTLChannel[int](10)\n\n\t\t\t// Set up ref count\n\t\t\tfor i := int32(0); i < tt.refCount; i++ {\n\t\t\t\tch.IncRef()\n\t\t\t}\n\n\t\t\t// Set up channel length\n\t\t\tfor i := 0; i < tt.channelLen; i++ {\n\t\t\t\tch.Chan() <- i\n\t\t\t}\n\n\t\t\t// Set up last write time\n\t\t\tlastWriteTime := time.Now().Add(-tt.timeSinceWrite)\n\t\t\tch.UpdateLastWriteTime(lastWriteTime)\n\n\t\t\t// Check should cleanup\n\t\t\tnow := time.Now()\n\t\t\tresult := ch.ShouldCleanup(now, tt.ttl)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestTTLChannel_ShouldCleanup_EdgeCases(t *testing.T) {\n\tt.Run(\"zero TTL\", func(t *testing.T) {\n\t\tch := NewTTLChannel[int](10)\n\t\t// Use time truncated to seconds to match what gets stored\n\t\tnow := time.Unix(time.Now().Unix(), 0)\n\t\tch.UpdateLastWriteTime(now)\n\n\t\t// Even with zero TTL, should cleanup only if time has passed\n\t\tassert.False(t, ch.ShouldCleanup(now, 0))\n\t\tassert.True(t, ch.ShouldCleanup(now.Add(1*time.Second), 0))\n\t})\n\n\tt.Run(\"never written\", func(t *testing.T) {\n\t\tch := NewTTLChannel[int](10)\n\t\tnow := time.Now()\n\t\tttl := 1 * time.Hour\n\n\t\t// Channel was never written to (lastWriteTime is Unix epoch)\n\t\t// Should cleanup since time.Now() - Unix(0) > ttl\n\t\tassert.True(t, ch.ShouldCleanup(now, ttl))\n\t})\n\n\tt.Run(\"negative ref count\", func(t *testing.T) {\n\t\tch := NewTTLChannel[int](10)\n\t\tnow := time.Now()\n\t\tch.UpdateLastWriteTime(now.Add(-2 * time.Hour))\n\n\t\t// Manually set negative ref count (shouldn't happen in practice)\n\t\tch.DecRef()\n\t\tassert.Equal(t, int32(-1), ch.RefCount())\n\n\t\t// Should not cleanup with negative ref count\n\t\t// (even though it's an error state, better safe than sorry)\n\t\tassert.False(t, ch.ShouldCleanup(now, 1*time.Hour))\n\t})\n}\n\nfunc TestTTLChannel_ConcurrentRefCount(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\titerations := 1000\n\n\t// Concurrently add refs\n\tdone := make(chan bool)\n\tfor i := 0; i < 10; i++ {\n\t\tgo func() {\n\t\t\tfor j := 0; j < iterations; j++ {\n\t\t\t\tch.IncRef()\n\t\t\t}\n\t\t\tdone <- true\n\t\t}()\n\t}\n\n\t// Wait for all goroutines\n\tfor i := 0; i < 10; i++ {\n\t\t<-done\n\t}\n\n\t// Should have 10 * iterations refs\n\texpected := int32(10 * iterations)\n\tassert.Equal(t, expected, ch.RefCount())\n\n\t// Concurrently decrement refs\n\tfor i := 0; i < 10; i++ {\n\t\tgo func() {\n\t\t\tfor j := 0; j < iterations; j++ {\n\t\t\t\tch.DecRef()\n\t\t\t}\n\t\t\tdone <- true\n\t\t}()\n\t}\n\n\t// Wait for all goroutines\n\tfor i := 0; i < 10; i++ {\n\t\t<-done\n\t}\n\n\t// Should be back to 0\n\tassert.Equal(t, int32(0), ch.RefCount())\n}\n\nfunc TestTTLChannel_ConcurrentTimeUpdate(t *testing.T) {\n\tch := NewTTLChannel[int](10)\n\titerations := 100\n\n\t// Concurrently update time\n\tdone := make(chan bool)\n\tbaseTime := time.Now()\n\n\tfor i := 0; i < 10; i++ {\n\t\tgo func(offset int) {\n\t\t\tfor j := 0; j < iterations; j++ {\n\t\t\t\tch.UpdateLastWriteTime(baseTime.Add(time.Duration(offset+j) * time.Second))\n\t\t\t}\n\t\t\tdone <- true\n\t\t}(i * iterations)\n\t}\n\n\t// Wait for all goroutines\n\tfor i := 0; i < 10; i++ {\n\t\t<-done\n\t}\n\n\t// Should have some valid time (exact value depends on interleaving)\n\tlastTime := ch.LastWriteTime()\n\tassert.True(t, !lastTime.Before(baseTime), \"last write time should not be before base time\")\n}\n"
  },
  {
    "path": "common/task/weighted_channel_pool.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst defaultIdleChannelTTLInSeconds = 3600\n\ntype (\n\tWeightedRoundRobinChannelPoolOptions struct {\n\t\tBufferSize              int\n\t\tIdleChannelTTLInSeconds int64\n\t}\n\n\tWeightedRoundRobinChannelPool[K comparable, V any] struct {\n\t\tsync.RWMutex\n\t\tstatus                  int32\n\t\tshutdownCh              chan struct{}\n\t\tshutdownWG              sync.WaitGroup\n\t\tbufferSize              int\n\t\tidleChannelTTLInSeconds int64\n\t\tlogger                  log.Logger\n\t\tmetricsScope            metrics.Scope\n\t\ttimeSource              clock.TimeSource\n\t\tchannelMap              map[K]weightedContainer[*TTLChannel[V]]\n\n\t\t// a snapshot of the channels to be used for the IWRR schedule\n\t\tiwrrSchedule atomic.Pointer[iwrrSchedule[*TTLChannel[V]]]\n\t}\n)\n\nfunc NewWeightedRoundRobinChannelPool[K comparable, V any](\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n\ttimeSource clock.TimeSource,\n\toptions WeightedRoundRobinChannelPoolOptions,\n) *WeightedRoundRobinChannelPool[K, V] {\n\twrr := &WeightedRoundRobinChannelPool[K, V]{\n\t\tbufferSize:              options.BufferSize,\n\t\tidleChannelTTLInSeconds: options.IdleChannelTTLInSeconds,\n\t\tlogger:                  logger,\n\t\tmetricsScope:            metricsScope,\n\t\ttimeSource:              timeSource,\n\t\tchannelMap:              make(map[K]weightedContainer[*TTLChannel[V]]),\n\t\tshutdownCh:              make(chan struct{}),\n\t}\n\t// Initialize with empty schedule\n\tschedule := newIWRRSchedule[K, *TTLChannel[V]](nil)\n\twrr.iwrrSchedule.Store(schedule)\n\treturn wrr\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.shutdownWG.Add(1)\n\tgo p.cleanupLoop()\n\n\tp.logger.Info(\"Weighted round robin channel pool started.\")\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(p.shutdownCh)\n\tp.shutdownWG.Wait()\n\n\tp.logger.Info(\"Weighted round robin channel pool stopped.\")\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) cleanupLoop() {\n\tdefer p.shutdownWG.Done()\n\tticker := p.timeSource.NewTicker(time.Duration((p.idleChannelTTLInSeconds / 2)) * time.Second)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tp.doCleanup()\n\t\tcase <-p.shutdownCh:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) doCleanup() {\n\tp.Lock()\n\tdefer p.Unlock()\n\tvar channelsToCleanup []K\n\tnow := p.timeSource.Now()\n\tttl := time.Duration(p.idleChannelTTLInSeconds) * time.Second\n\tfor k, container := range p.channelMap {\n\t\tif container.item.ShouldCleanup(now, ttl) {\n\t\t\tchannelsToCleanup = append(channelsToCleanup, k)\n\t\t}\n\t}\n\n\tfor _, k := range channelsToCleanup {\n\t\tdelete(p.channelMap, k)\n\t}\n\n\tif len(channelsToCleanup) > 0 {\n\t\tp.logger.Info(\"clean up idle channels\", tag.Dynamic(\"channels\", channelsToCleanup))\n\t\tp.updateScheduleLocked()\n\t}\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) GetOrCreateChannel(key K, weight int) (chan V, func()) {\n\tp.RLock()\n\tif container, exists := p.channelMap[key]; exists && container.weight == weight {\n\t\tcontainer.item.IncRef()\n\t\tcontainer.item.UpdateLastWriteTime(p.timeSource.Now())\n\t\tp.RUnlock()\n\t\treturn container.item.Chan(), container.item.DecRef\n\t}\n\tp.RUnlock()\n\n\tp.Lock()\n\tdefer p.Unlock()\n\tif container, exists := p.channelMap[key]; exists {\n\t\tcontainer.item.IncRef()\n\t\tcontainer.item.UpdateLastWriteTime(p.timeSource.Now())\n\t\tif container.weight != weight {\n\t\t\tp.channelMap[key] = weightedContainer[*TTLChannel[V]]{\n\t\t\t\titem:   container.item,\n\t\t\t\tweight: weight,\n\t\t\t}\n\t\t\tp.updateScheduleLocked()\n\t\t}\n\t\treturn container.item.Chan(), container.item.DecRef\n\t}\n\n\tc := NewTTLChannel[V](p.bufferSize)\n\tp.channelMap[key] = weightedContainer[*TTLChannel[V]]{\n\t\titem:   c,\n\t\tweight: weight,\n\t}\n\tc.IncRef()\n\tc.UpdateLastWriteTime(p.timeSource.Now())\n\tp.updateScheduleLocked()\n\treturn c.Chan(), c.DecRef\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) GetAllChannels() []chan V {\n\tp.RLock()\n\tdefer p.RUnlock()\n\tallChannels := make([]chan V, 0, len(p.channelMap))\n\tfor _, container := range p.channelMap {\n\t\tallChannels = append(allChannels, container.item.Chan())\n\t}\n\treturn allChannels\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) GetSchedule() Schedule[*TTLChannel[V]] {\n\treturn p.iwrrSchedule.Load()\n}\n\nfunc (p *WeightedRoundRobinChannelPool[K, V]) updateScheduleLocked() {\n\tp.iwrrSchedule.Store(newIWRRSchedule(p.channelMap))\n\n\t// Update memory gauge - now only stores channel references once, not weight times\n\tmemoryBytes := len(p.channelMap) * 16 // channel map entries\n\tp.metricsScope.UpdateGauge(metrics.WeightedChannelPoolSizeGauge, float64(memoryBytes))\n}\n"
  },
  {
    "path": "common/task/weighted_channel_pool_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestGetOrCreateChannel(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tpool := NewWeightedRoundRobinChannelPool[string, int](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopScope,\n\t\ttimeSource,\n\t\tWeightedRoundRobinChannelPoolOptions{\n\t\t\tBufferSize:              1000,\n\t\t\tIdleChannelTTLInSeconds: 10,\n\t\t},\n\t)\n\n\t// First, verify that the method returns the same channel if the key and weight are the same\n\tc1, releaseFn1 := pool.GetOrCreateChannel(\"k1\", 1)\n\tdefer releaseFn1()\n\tc2, releaseFn2 := pool.GetOrCreateChannel(\"k1\", 1)\n\tdefer releaseFn2()\n\tassert.Equal(t, c1, c2)\n\n\t// Next, verify that the methods returns the same channel if the key is the same but weight is different\n\tc3, releaseFn3 := pool.GetOrCreateChannel(\"k1\", 2)\n\tdefer releaseFn3()\n\tassert.Equal(t, c1, c3)\n}\n\nfunc TestGetOrCreateChannelConcurrent(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tpool := NewWeightedRoundRobinChannelPool[string, int](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopScope,\n\t\ttimeSource,\n\t\tWeightedRoundRobinChannelPoolOptions{\n\t\t\tBufferSize:              1000,\n\t\t\tIdleChannelTTLInSeconds: 10,\n\t\t},\n\t)\n\n\tvar wg sync.WaitGroup\n\tvar chMap sync.Map\n\twg.Add(15)\n\n\tfor i := 0; i < 5; i++ {\n\t\tgo func(i int) {\n\t\t\tdefer wg.Done()\n\t\t\tc, releaseFn := pool.GetOrCreateChannel(\"k1\", i+1)\n\t\t\tdefer releaseFn()\n\t\t\tchMap.Store(\"k1\", c)\n\t\t}(i)\n\t\tgo func(i int) {\n\t\t\tdefer wg.Done()\n\t\t\tc, releaseFn := pool.GetOrCreateChannel(\"k2\", i+1)\n\t\t\tdefer releaseFn()\n\t\t\tchMap.Store(\"k2\", c)\n\t\t}(i)\n\t\tgo func(i int) {\n\t\t\tdefer wg.Done()\n\t\t\tc, releaseFn := pool.GetOrCreateChannel(\"k3\", i+1)\n\t\t\tdefer releaseFn()\n\t\t\tchMap.Store(\"k3\", c)\n\t\t}(i)\n\t}\n\twg.Wait()\n\n\tchs := pool.GetAllChannels()\n\tassert.Len(t, chs, 3)\n\tch1, _ := chMap.Load(\"k1\")\n\tch2, _ := chMap.Load(\"k2\")\n\tch3, _ := chMap.Load(\"k3\")\n\texpectedChs := []chan int{ch1.(chan int), ch2.(chan int), ch3.(chan int)}\n\tassert.ElementsMatch(t, expectedChs, chs)\n}\n\nfunc TestGetSchedule(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tpool := NewWeightedRoundRobinChannelPool[string, int](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopScope,\n\t\ttimeSource,\n\t\tWeightedRoundRobinChannelPoolOptions{\n\t\t\tBufferSize:              1000,\n\t\t\tIdleChannelTTLInSeconds: 10,\n\t\t},\n\t)\n\n\tc1, releaseFn1 := pool.GetOrCreateChannel(\"k1\", 1)\n\tdefer releaseFn1()\n\tc2, releaseFn2 := pool.GetOrCreateChannel(\"k2\", 2)\n\tdefer releaseFn2()\n\tc3, releaseFn3 := pool.GetOrCreateChannel(\"k3\", 3)\n\tdefer releaseFn3()\n\n\tschedule := pool.GetSchedule()\n\tassert.Equal(t, 6, schedule.Len())\n\n\titer1 := schedule.NewIterator()\n\n\t// Verify IWRR sequence: [c3, c3, c2, c3, c2, c1]\n\texpectedSequence1 := []chan int{c3, c3, c2, c3, c2, c1}\n\tfor _, expected := range expectedSequence1 {\n\t\tch, ok := iter1.TryNext()\n\t\tassert.True(t, ok)\n\t\tassert.Equal(t, expected, ch.Chan())\n\t}\n\n\tc4, releaseFn4 := pool.GetOrCreateChannel(\"k2\", 4)\n\tdefer releaseFn4()\n\tassert.Equal(t, c2, c4)\n\tschedule = pool.GetSchedule()\n\tassert.Equal(t, 8, schedule.Len())\n\n\titer2 := schedule.NewIterator()\n\n\t// Verify IWRR sequence after weight change: [c2, c2, c3, c2, c3, c2, c3, c1]\n\texpectedSequence2 := []chan int{c2, c2, c3, c2, c3, c2, c3, c1}\n\tfor _, expected := range expectedSequence2 {\n\t\tch, ok := iter2.TryNext()\n\t\tassert.True(t, ok)\n\t\tassert.Equal(t, expected, ch.Chan())\n\t}\n}\n\nfunc TestCleanup(t *testing.T) {\n\ttimeSource := clock.NewRealTimeSource()\n\tpool := NewWeightedRoundRobinChannelPool[string, int](\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopScope,\n\t\ttimeSource,\n\t\tWeightedRoundRobinChannelPoolOptions{\n\t\t\tBufferSize:              1000,\n\t\t\tIdleChannelTTLInSeconds: 2,\n\t\t},\n\t)\n\tpool.Start()\n\tdefer pool.Stop()\n\n\t// First, verify that the method returns the same channel if the key and weight are the same\n\t_, releaseFn1 := pool.GetOrCreateChannel(\"k1\", 1)\n\tch2, releaseFn2 := pool.GetOrCreateChannel(\"k2\", 1)\n\tch3, releaseFn3 := pool.GetOrCreateChannel(\"k3\", 1)\n\tch3 <- 1\n\n\tassert.Len(t, pool.GetAllChannels(), 3)\n\treleaseFn1()\n\treleaseFn3()\n\ttime.Sleep(time.Second * 4)\n\t// only c1 is deleted\n\tchs := pool.GetAllChannels()\n\tassert.ElementsMatch(t, chs, []chan int{ch2, ch3})\n\n\treleaseFn2()\n}\n"
  },
  {
    "path": "common/task/weighted_round_robin_task_scheduler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype weightedRoundRobinTaskSchedulerImpl[K comparable, T Task] struct {\n\tsync.RWMutex\n\n\tstatus       int32\n\tpool         *WeightedRoundRobinChannelPool[K, T]\n\tctx          context.Context\n\tcancel       context.CancelFunc\n\tnotifyCh     chan struct{}\n\tdispatcherWG sync.WaitGroup\n\tlogger       log.Logger\n\tmetricsScope metrics.Scope\n\toptions      *WeightedRoundRobinTaskSchedulerOptions[K, T]\n\n\tprocessor Processor\n}\n\nconst (\n\twRRTaskProcessorQueueSize    = 1\n\tdefaultUpdateWeightsInterval = 5 * time.Second\n)\n\nvar (\n\t// ErrTaskSchedulerClosed is the error returned when submitting task to a stopped scheduler\n\tErrTaskSchedulerClosed = errors.New(\"task scheduler has already shutdown\")\n)\n\n// NewWeightedRoundRobinTaskScheduler creates a new WRR task scheduler\nfunc NewWeightedRoundRobinTaskScheduler[K comparable, T Task](\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n\tprocessor Processor,\n\toptions *WeightedRoundRobinTaskSchedulerOptions[K, T],\n) (Scheduler[T], error) {\n\tmetricsScope := metricsClient.Scope(metrics.TaskSchedulerScope)\n\tctx, cancel := context.WithCancel(context.Background())\n\tscheduler := &weightedRoundRobinTaskSchedulerImpl[K, T]{\n\t\tstatus: common.DaemonStatusInitialized,\n\t\tpool: NewWeightedRoundRobinChannelPool[K, T](\n\t\t\tlogger,\n\t\t\tmetricsScope,\n\t\t\ttimeSource,\n\t\t\tWeightedRoundRobinChannelPoolOptions{\n\t\t\t\tBufferSize:              options.QueueSize,\n\t\t\t\tIdleChannelTTLInSeconds: defaultIdleChannelTTLInSeconds,\n\t\t\t}),\n\t\tctx:          ctx,\n\t\tcancel:       cancel,\n\t\tnotifyCh:     make(chan struct{}, 1),\n\t\tlogger:       logger,\n\t\tmetricsScope: metricsScope,\n\t\toptions:      options,\n\t\tprocessor:    processor,\n\t}\n\n\treturn scheduler, nil\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) Start() {\n\tif !atomic.CompareAndSwapInt32(&w.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tw.dispatcherWG.Add(w.options.DispatcherCount)\n\tfor i := 0; i != w.options.DispatcherCount; i++ {\n\t\tgo w.dispatcher()\n\t}\n\tw.logger.Info(\"Weighted round robin task scheduler started.\")\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) Stop() {\n\tif !atomic.CompareAndSwapInt32(&w.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tw.cancel()\n\n\ttaskChs := w.pool.GetAllChannels()\n\tfor _, taskCh := range taskChs {\n\t\tdrainAndNackPriorityTask(taskCh)\n\t}\n\n\tif success := common.AwaitWaitGroup(&w.dispatcherWG, time.Minute); !success {\n\t\tw.logger.Warn(\"Weighted round robin task scheduler timedout on shutdown.\")\n\t}\n\n\tw.logger.Info(\"Weighted round robin task scheduler shutdown.\")\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) Submit(task T) error {\n\tw.metricsScope.IncCounter(metrics.PriorityTaskSubmitRequest)\n\tsw := w.metricsScope.StartTimer(metrics.PriorityTaskSubmitLatency)\n\tdefer sw.Stop()\n\n\tif w.isStopped() {\n\t\treturn ErrTaskSchedulerClosed\n\t}\n\n\tkey := w.options.TaskToChannelKeyFn(task)\n\tweight := w.options.ChannelKeyToWeightFn(key)\n\ttaskCh, releaseFn := w.pool.GetOrCreateChannel(key, weight)\n\tdefer releaseFn()\n\tselect {\n\tcase taskCh <- task:\n\t\tw.notifyDispatcher()\n\t\tif w.isStopped() {\n\t\t\tdrainAndNackPriorityTask(taskCh)\n\t\t}\n\t\treturn nil\n\tcase <-w.ctx.Done():\n\t\treturn ErrTaskSchedulerClosed\n\t}\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) TrySubmit(\n\ttask T,\n) (bool, error) {\n\tif w.isStopped() {\n\t\treturn false, ErrTaskSchedulerClosed\n\t}\n\n\tkey := w.options.TaskToChannelKeyFn(task)\n\tweight := w.options.ChannelKeyToWeightFn(key)\n\ttaskCh, releaseFn := w.pool.GetOrCreateChannel(key, weight)\n\tdefer releaseFn()\n\n\tselect {\n\tcase taskCh <- task:\n\t\tw.metricsScope.IncCounter(metrics.PriorityTaskSubmitRequest)\n\t\tif w.isStopped() {\n\t\t\tdrainAndNackPriorityTask(taskCh)\n\t\t} else {\n\t\t\tw.notifyDispatcher()\n\t\t}\n\t\treturn true, nil\n\tcase <-w.ctx.Done():\n\t\treturn false, ErrTaskSchedulerClosed\n\tdefault:\n\t\treturn false, nil\n\t}\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) dispatcher() {\n\tdefer w.dispatcherWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-w.notifyCh:\n\t\t\tw.dispatchTasks()\n\t\tcase <-w.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) dispatchTasks() {\n\thasTask := true\n\tfor hasTask {\n\t\thasTask = false\n\t\tschedule := w.pool.GetSchedule()\n\n\t\t// Create a new iterator for this dispatch cycle\n\t\titer := schedule.NewIterator()\n\t\t// Iterate through the schedule using the iterator\n\t\tfor ch, ok := iter.TryNext(); ok; ch, ok = iter.TryNext() {\n\t\t\tselect {\n\t\t\tcase task := <-ch.c:\n\t\t\t\thasTask = true\n\t\t\t\tif err := w.processor.Submit(task); err != nil {\n\t\t\t\t\tw.logger.Error(\"fail to submit task to processor\", tag.Error(err))\n\t\t\t\t\ttask.Nack(err)\n\t\t\t\t}\n\t\t\tcase <-w.ctx.Done():\n\t\t\t\treturn\n\t\t\tdefault:\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) notifyDispatcher() {\n\tselect {\n\tcase w.notifyCh <- struct{}{}:\n\t\t// sent a notification to the dispatcher\n\tdefault:\n\t\t// do not block if there's already a notification\n\t}\n}\n\nfunc (w *weightedRoundRobinTaskSchedulerImpl[K, T]) isStopped() bool {\n\treturn atomic.LoadInt32(&w.status) == common.DaemonStatusStopped\n}\n\nfunc drainAndNackPriorityTask[T Task](taskCh <-chan T) {\n\tfor {\n\t\tselect {\n\t\tcase task := <-taskCh:\n\t\t\ttask.Nack(nil)\n\t\tdefault:\n\t\t\treturn\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/task/weighted_round_robin_task_scheduler_options.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"fmt\"\n)\n\n// WeightedRoundRobinTaskSchedulerOptions configs WRR task scheduler\ntype WeightedRoundRobinTaskSchedulerOptions[K comparable, T Task] struct {\n\tQueueSize            int\n\tDispatcherCount      int\n\tTaskToChannelKeyFn   func(T) K\n\tChannelKeyToWeightFn func(K) int\n}\n\nfunc (o *WeightedRoundRobinTaskSchedulerOptions[K, T]) String() string {\n\treturn fmt.Sprintf(\"{QueueSize: %v, DispatcherCount: %v}\", o.QueueSize, o.DispatcherCount)\n}\n"
  },
  {
    "path": "common/task/weighted_round_robin_task_scheduler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tweightedRoundRobinTaskSchedulerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tcontroller    *gomock.Controller\n\t\tmockProcessor *MockProcessor\n\t\trealProcessor Processor\n\n\t\tqueueSize int\n\n\t\tscheduler *weightedRoundRobinTaskSchedulerImpl[int, PriorityTask]\n\t}\n\n\tmockPriorityTaskMatcher struct {\n\t\ttask *MockPriorityTask\n\t}\n)\n\nvar (\n\ttestSchedulerWeights = dynamicproperties.GetMapPropertyFn(\n\t\tmap[string]interface{}{\n\t\t\t\"0\": 3,\n\t\t\t\"1\": 2,\n\t\t\t\"2\": 1,\n\t\t},\n\t)\n)\n\nfunc TestWeightedRoundRobinTaskSchedulerSuite(t *testing.T) {\n\ts := new(weightedRoundRobinTaskSchedulerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockProcessor = NewMockProcessor(s.controller)\n\n\ts.queueSize = 1000\n\ts.realProcessor = NewParallelTaskProcessor(\n\t\ttestlogger.New(s.Suite.T()),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\t&ParallelTaskProcessorOptions{\n\t\t\tQueueSize:   1,\n\t\t\tWorkerCount: dynamicproperties.GetIntPropertyFn(1),\n\t\t\tRetryPolicy: backoff.NewExponentialRetryPolicy(time.Millisecond),\n\t\t},\n\t)\n\ts.scheduler = s.newTestWeightedRoundRobinTaskScheduler(\n\t\t&WeightedRoundRobinTaskSchedulerOptions[int, PriorityTask]{\n\t\t\tQueueSize:       s.queueSize,\n\t\t\tDispatcherCount: 3,\n\t\t\tTaskToChannelKeyFn: func(task PriorityTask) int {\n\t\t\t\treturn task.Priority()\n\t\t\t},\n\t\t\tChannelKeyToWeightFn: func(key int) int {\n\t\t\t\treturn testSchedulerWeights()[fmt.Sprintf(\"%v\", key)].(int)\n\t\t\t},\n\t\t},\n\t)\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestSubmit_Success() {\n\ttaskPriority := 1\n\tmockTask := NewMockPriorityTask(s.controller)\n\tmockTask.EXPECT().Priority().Return(taskPriority)\n\n\terr := s.scheduler.Submit(mockTask)\n\ts.NoError(err)\n\n\tweight := s.scheduler.options.ChannelKeyToWeightFn(taskPriority)\n\ttaskCh, releaseFn := s.scheduler.pool.GetOrCreateChannel(taskPriority, weight)\n\tdefer releaseFn()\n\ttask := <-taskCh\n\ts.Equal(mockTask, task)\n\ttaskChs := s.scheduler.pool.GetAllChannels()\n\tfor _, taskCh := range taskChs {\n\t\ts.Empty(taskCh)\n\t}\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestSubmit_Fail_SchedulerShutDown() {\n\t// create a new scheduler here with queue size 0, otherwise test is non-deterministic\n\tscheduler := s.newTestWeightedRoundRobinTaskScheduler(\n\t\t&WeightedRoundRobinTaskSchedulerOptions[int, PriorityTask]{\n\t\t\tQueueSize:       0,\n\t\t\tDispatcherCount: 3,\n\t\t},\n\t)\n\n\tmockTask := NewMockPriorityTask(s.controller)\n\tscheduler.Start()\n\tscheduler.Stop()\n\terr := scheduler.Submit(mockTask)\n\ts.Equal(ErrTaskSchedulerClosed, err)\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestTrySubmit() {\n\ttaskPriority := 1\n\tfor i := 0; i != s.queueSize; i++ {\n\t\tmockTask := NewMockPriorityTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(taskPriority)\n\t\tsubmitted, err := s.scheduler.TrySubmit(mockTask)\n\t\ts.NoError(err)\n\t\ts.True(submitted)\n\t}\n\n\t// now the queue is full, submit one more task, should be non-blocking\n\tmockTask := NewMockPriorityTask(s.controller)\n\tmockTask.EXPECT().Priority().Return(taskPriority)\n\tsubmitted, err := s.scheduler.TrySubmit(mockTask)\n\ts.NoError(err)\n\ts.False(submitted)\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestDispatcher_SubmitWithNoError() {\n\tweights, err := dynamicproperties.ConvertDynamicConfigMapPropertyToIntMap(testSchedulerWeights())\n\ts.NoError(err)\n\n\tnumPriorities := len(weights)\n\ttasks := [][]*MockPriorityTask{}\n\tvar taskWG sync.WaitGroup\n\tfor i := 0; i != numPriorities; i++ {\n\t\ttasks = append(tasks, []*MockPriorityTask{})\n\t}\n\n\ttaskPerPriority := 5\n\tnumSubmittedTask := 0\n\ttasksPerRound := []int{6, 5, 2, 1, 1}\n\tround := 0\n\tmockFn := func(_ Task) error {\n\t\tnumSubmittedTask++\n\t\tif numSubmittedTask == tasksPerRound[round] {\n\t\t\tround++\n\t\t\tnumSubmittedTask = 0\n\n\t\t\tfor priority, weight := range weights {\n\t\t\t\texpectedRemainingTasksNum := taskPerPriority - round*weight\n\t\t\t\tif expectedRemainingTasksNum < 0 {\n\t\t\t\t\texpectedRemainingTasksNum = 0\n\t\t\t\t}\n\t\t\t\ttaskCh, releaseFn := s.scheduler.pool.GetOrCreateChannel(priority, weight)\n\t\t\t\ts.Equal(expectedRemainingTasksNum, len(taskCh))\n\t\t\t\treleaseFn()\n\t\t\t}\n\t\t}\n\n\t\ttaskWG.Done()\n\t\treturn nil\n\t}\n\n\tfor priority := range weights {\n\t\tfor i := 0; i != taskPerPriority; i++ {\n\t\t\tmockTask := NewMockPriorityTask(s.controller)\n\t\t\tmockTask.EXPECT().Priority().Return(priority).AnyTimes()\n\t\t\ts.scheduler.Submit(mockTask)\n\t\t\ttasks[priority] = append(tasks[priority], mockTask)\n\t\t\ttaskWG.Add(1)\n\t\t\ts.mockProcessor.EXPECT().Submit(newMockPriorityTaskMatcher(mockTask)).DoAndReturn(mockFn)\n\t\t}\n\t}\n\n\ts.scheduler.processor = s.mockProcessor\n\n\tdoneCh := make(chan struct{})\n\tgo func() {\n\t\ts.scheduler.dispatcherWG.Add(1)\n\t\ts.scheduler.dispatcher()\n\t\tclose(doneCh)\n\t}()\n\n\ttaskWG.Wait()\n\ts.scheduler.cancel()\n\n\t<-doneCh\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestDispatcher_FailToSubmit() {\n\tmockTask := NewMockPriorityTask(s.controller)\n\tmockTask.EXPECT().Priority().Return(0)\n\tmockTask.EXPECT().Nack(gomock.Any())\n\n\tvar taskWG sync.WaitGroup\n\ts.scheduler.Submit(mockTask)\n\ttaskWG.Add(1)\n\n\tmockFn := func(_ Task) error {\n\t\ttaskWG.Done()\n\t\treturn errors.New(\"some random error\")\n\t}\n\ts.mockProcessor.EXPECT().Submit(newMockPriorityTaskMatcher(mockTask)).DoAndReturn(mockFn)\n\ts.scheduler.processor = s.mockProcessor\n\n\tdoneCh := make(chan struct{})\n\tgo func() {\n\t\ts.scheduler.dispatcherWG.Add(1)\n\t\ts.scheduler.dispatcher()\n\t\tclose(doneCh)\n\t}()\n\n\ttaskWG.Wait()\n\ts.scheduler.cancel()\n\n\t<-doneCh\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestWRR() {\n\tnumTasks := 1000\n\tvar taskWG sync.WaitGroup\n\n\ttasks := []PriorityTask{}\n\tmockFn := func(_ Task) error {\n\t\ttaskWG.Done()\n\t\treturn nil\n\t}\n\tfor i := 0; i != numTasks; i++ {\n\t\tmockTask := NewMockPriorityTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(len(testSchedulerWeights()))).Times(1)\n\t\ttasks = append(tasks, mockTask)\n\t\ttaskWG.Add(1)\n\t\ts.mockProcessor.EXPECT().Submit(newMockPriorityTaskMatcher(mockTask)).DoAndReturn(mockFn)\n\t}\n\n\ts.scheduler.processor = s.mockProcessor\n\ts.scheduler.Start()\n\tfor _, task := range tasks {\n\t\tif rand.Intn(2) == 0 {\n\t\t\ts.NoError(s.scheduler.Submit(task))\n\t\t} else {\n\t\t\tsubmitted, err := s.scheduler.TrySubmit(task)\n\t\t\ts.NoError(err)\n\t\t\ts.True(submitted)\n\t\t}\n\t}\n\ttaskWG.Wait()\n\ts.scheduler.Stop()\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) TestSchedulerContract() {\n\ttestSchedulerContract(s.Assertions, s.controller, s.scheduler, s.realProcessor)\n}\n\nfunc (s *weightedRoundRobinTaskSchedulerSuite) newTestWeightedRoundRobinTaskScheduler(\n\toptions *WeightedRoundRobinTaskSchedulerOptions[int, PriorityTask],\n) *weightedRoundRobinTaskSchedulerImpl[int, PriorityTask] {\n\tscheduler, err := NewWeightedRoundRobinTaskScheduler(\n\t\ttestlogger.New(s.Suite.T()),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{}),\n\t\tclock.NewMockedTimeSource(),\n\t\ts.realProcessor,\n\t\toptions,\n\t)\n\ts.NoError(err)\n\treturn scheduler.(*weightedRoundRobinTaskSchedulerImpl[int, PriorityTask])\n}\n\nfunc testSchedulerContract(\n\ts *require.Assertions,\n\tcontroller *gomock.Controller,\n\tscheduler Scheduler[PriorityTask],\n\tprocessor Processor,\n) {\n\tnumTasks := 10000\n\tvar taskWG sync.WaitGroup\n\n\ttasks := []PriorityTask{}\n\ttaskStatusLock := &sync.Mutex{}\n\ttaskStatus := make(map[PriorityTask]State)\n\tfor i := 0; i != numTasks; i++ {\n\t\tmockTask := NewMockPriorityTask(controller)\n\t\ttaskStatus[mockTask] = TaskStatePending\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(len(testSchedulerWeights()))).MaxTimes(1)\n\t\tmockTask.EXPECT().Execute().Return(nil).MaxTimes(1)\n\t\tmockTask.EXPECT().Ack().Do(func() {\n\t\t\ttaskStatusLock.Lock()\n\t\t\tdefer taskStatusLock.Unlock()\n\n\t\t\ts.Equal(TaskStatePending, taskStatus[mockTask])\n\t\t\ttaskStatus[mockTask] = TaskStateAcked\n\t\t\ttaskWG.Done()\n\t\t}).MaxTimes(1)\n\t\tmockTask.EXPECT().Nack(gomock.Any()).Do(func(err error) {\n\t\t\ttaskStatusLock.Lock()\n\t\t\tdefer taskStatusLock.Unlock()\n\n\t\t\ts.Equal(TaskStatePending, taskStatus[mockTask])\n\t\t\ttaskStatus[mockTask] = State(-1) // set it to whatever state that is not TaskStatePending\n\t\t\ttaskWG.Done()\n\t\t}).MaxTimes(1)\n\t\ttasks = append(tasks, mockTask)\n\t\ttaskWG.Add(1)\n\t}\n\n\tif processor != nil {\n\t\tprocessor.Start()\n\t}\n\tscheduler.Start()\n\tgo func() {\n\t\ttime.Sleep(time.Duration(rand.Intn(50)) * time.Millisecond)\n\t\tscheduler.Stop()\n\t\tif processor != nil {\n\t\t\tprocessor.Stop()\n\t\t}\n\t}()\n\n\tfor _, task := range tasks {\n\t\tif rand.Intn(2) == 0 {\n\t\t\tif err := scheduler.Submit(task); err != nil {\n\t\t\t\ttaskWG.Done()\n\t\t\t\ttaskStatusLock.Lock()\n\t\t\t\tdelete(taskStatus, task)\n\t\t\t\ttaskStatusLock.Unlock()\n\t\t\t}\n\t\t} else {\n\t\t\tif submitted, _ := scheduler.TrySubmit(task); !submitted {\n\t\t\t\ttaskWG.Done()\n\t\t\t\ttaskStatusLock.Lock()\n\t\t\t\tdelete(taskStatus, task)\n\t\t\t\ttaskStatusLock.Unlock()\n\t\t\t}\n\t\t}\n\t}\n\ts.True(common.AwaitWaitGroup(&taskWG, 10*time.Second))\n\tswitch schedulerImpl := scheduler.(type) {\n\tcase *fifoTaskSchedulerImpl[PriorityTask]:\n\t\t<-schedulerImpl.ctx.Done()\n\tcase *weightedRoundRobinTaskSchedulerImpl[int, PriorityTask]:\n\t\t<-schedulerImpl.ctx.Done()\n\tcase *hierarchicalWeightedRoundRobinTaskSchedulerImpl[string, PriorityTask]:\n\t\t<-schedulerImpl.ctx.Done()\n\tdefault:\n\t\ts.Fail(\"unknown task scheduler type\")\n\t}\n\n\tfor _, status := range taskStatus {\n\t\ts.NotEqual(TaskStatePending, status)\n\t}\n}\n\nfunc newMockPriorityTaskMatcher(mockTask *MockPriorityTask) gomock.Matcher {\n\treturn &mockPriorityTaskMatcher{\n\t\ttask: mockTask,\n\t}\n}\n\nfunc (m *mockPriorityTaskMatcher) Matches(x interface{}) bool {\n\ttaskPtr, ok := x.(*MockPriorityTask)\n\tif !ok {\n\t\treturn false\n\t}\n\treturn taskPtr == m.task\n}\n\nfunc (m *mockPriorityTaskMatcher) String() string {\n\treturn fmt.Sprintf(\"is equal to %v\", m.task)\n}\n"
  },
  {
    "path": "common/taskTokenSerializerInterfaces.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination taskTokenSerializerInterfaces_mock.go -self_package github.com/uber/cadence/common\n\npackage common\n\ntype (\n\t// TaskTokenSerializer serializes task tokens\n\tTaskTokenSerializer interface {\n\t\tSerialize(token *TaskToken) ([]byte, error)\n\t\tDeserialize(data []byte) (*TaskToken, error)\n\t\tSerializeQueryTaskToken(token *QueryTaskToken) ([]byte, error)\n\t\tDeserializeQueryTaskToken(data []byte) (*QueryTaskToken, error)\n\t}\n\n\t// TaskToken identifies a task\n\tTaskToken struct {\n\t\tDomainID        string `json:\"domainId\"`\n\t\tWorkflowID      string `json:\"workflowId\"`\n\t\tWorkflowType    string `json:\"workflowType\"`\n\t\tRunID           string `json:\"runId\"`\n\t\tScheduleID      int64  `json:\"scheduleId\"`\n\t\tScheduleAttempt int64  `json:\"scheduleAttempt\"`\n\t\tActivityID      string `json:\"activityId\"`\n\t\tActivityType    string `json:\"activityType\"`\n\t}\n\n\t// QueryTaskToken identifies a query task\n\tQueryTaskToken struct {\n\t\tDomainID   string `json:\"domainId\"`\n\t\tWorkflowID string `json:\"workflowId\"`\n\t\tRunID      string `json:\"runId\"`\n\t\tTaskList   string `json:\"taskList\"`\n\t\tTaskID     string `json:\"taskId\"`\n\t}\n)\n\nfunc (t TaskToken) GetDomainID() string {\n\treturn t.DomainID\n}\n\nfunc (t QueryTaskToken) GetDomainID() string {\n\treturn t.DomainID\n}\n"
  },
  {
    "path": "common/taskTokenSerializerInterfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: taskTokenSerializerInterfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package common -source taskTokenSerializerInterfaces.go -destination taskTokenSerializerInterfaces_mock.go -self_package github.com/uber/cadence/common\n//\n\n// Package common is a generated GoMock package.\npackage common\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockTaskTokenSerializer is a mock of TaskTokenSerializer interface.\ntype MockTaskTokenSerializer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskTokenSerializerMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskTokenSerializerMockRecorder is the mock recorder for MockTaskTokenSerializer.\ntype MockTaskTokenSerializerMockRecorder struct {\n\tmock *MockTaskTokenSerializer\n}\n\n// NewMockTaskTokenSerializer creates a new mock instance.\nfunc NewMockTaskTokenSerializer(ctrl *gomock.Controller) *MockTaskTokenSerializer {\n\tmock := &MockTaskTokenSerializer{ctrl: ctrl}\n\tmock.recorder = &MockTaskTokenSerializerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskTokenSerializer) EXPECT() *MockTaskTokenSerializerMockRecorder {\n\treturn m.recorder\n}\n\n// Deserialize mocks base method.\nfunc (m *MockTaskTokenSerializer) Deserialize(data []byte) (*TaskToken, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Deserialize\", data)\n\tret0, _ := ret[0].(*TaskToken)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Deserialize indicates an expected call of Deserialize.\nfunc (mr *MockTaskTokenSerializerMockRecorder) Deserialize(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Deserialize\", reflect.TypeOf((*MockTaskTokenSerializer)(nil).Deserialize), data)\n}\n\n// DeserializeQueryTaskToken mocks base method.\nfunc (m *MockTaskTokenSerializer) DeserializeQueryTaskToken(data []byte) (*QueryTaskToken, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeserializeQueryTaskToken\", data)\n\tret0, _ := ret[0].(*QueryTaskToken)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeserializeQueryTaskToken indicates an expected call of DeserializeQueryTaskToken.\nfunc (mr *MockTaskTokenSerializerMockRecorder) DeserializeQueryTaskToken(data any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeserializeQueryTaskToken\", reflect.TypeOf((*MockTaskTokenSerializer)(nil).DeserializeQueryTaskToken), data)\n}\n\n// Serialize mocks base method.\nfunc (m *MockTaskTokenSerializer) Serialize(token *TaskToken) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Serialize\", token)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Serialize indicates an expected call of Serialize.\nfunc (mr *MockTaskTokenSerializerMockRecorder) Serialize(token any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Serialize\", reflect.TypeOf((*MockTaskTokenSerializer)(nil).Serialize), token)\n}\n\n// SerializeQueryTaskToken mocks base method.\nfunc (m *MockTaskTokenSerializer) SerializeQueryTaskToken(token *QueryTaskToken) ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SerializeQueryTaskToken\", token)\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SerializeQueryTaskToken indicates an expected call of SerializeQueryTaskToken.\nfunc (mr *MockTaskTokenSerializerMockRecorder) SerializeQueryTaskToken(token any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SerializeQueryTaskToken\", reflect.TypeOf((*MockTaskTokenSerializer)(nil).SerializeQueryTaskToken), token)\n}\n"
  },
  {
    "path": "common/taskvalidator/validateworkflow.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n// Package taskvalidator provides a Work in Progress service for workflow validations.\npackage taskvalidator\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"time\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Checker interface for initiating the validation process.\ntype Checker interface {\n\tWorkflowCheckforValidation(ctx context.Context, workflowID string, domainID string, domainName string, runID string) error\n}\n\n// staleChecker interface definition.\ntype staleChecker interface {\n\tCheckAge(response *persistence.GetWorkflowExecutionResponse) (bool, error)\n}\n\n// checkerImpl is the implementation of the Checker interface.\ntype checkerImpl struct {\n\tlogger        *zap.Logger\n\tmetricsClient metrics.Client\n\tdc            cache.DomainCache\n\tpr            persistence.Retryer\n\tstaleCheck    staleChecker\n}\n\n// NewWfChecker creates a new instance of a workflow validation checker.\n// It requires a logger, metrics client, domain cache, persistence retryer,\n// and a stale checker implementation to function.\nfunc NewWfChecker(logger *zap.Logger, metrics metrics.Client, domainCache cache.DomainCache, executionManager persistence.ExecutionManager, historymanager persistence.HistoryManager) (Checker, error) {\n\t// Create the persistence retryer\n\tretryPolicy := backoff.NewExponentialRetryPolicy(100 * time.Millisecond) // Adjust as needed\n\tpr := persistence.NewPersistenceRetryer(executionManager, historymanager, retryPolicy)\n\n\t// Create the stale check instance\n\tstaleCheckInstance := invariant.NewStaleWorkflow(pr, domainCache, logger)\n\tstaleCheck, _ := staleCheckInstance.(staleChecker)\n\n\t// Return the checker implementation\n\treturn &checkerImpl{\n\t\tlogger:        logger,\n\t\tmetricsClient: metrics,\n\t\tdc:            domainCache,\n\t\tpr:            pr,\n\t\tstaleCheck:    staleCheck,\n\t}, nil\n}\n\n// WorkflowCheckforValidation performs workflow validation.\nfunc (w *checkerImpl) WorkflowCheckforValidation(ctx context.Context, workflowID string, domainID string, domainName string, runID string) error {\n\tw.logger.Info(\"WorkflowCheckforValidation\",\n\t\tzap.String(\"WorkflowID\", workflowID),\n\t\tzap.String(\"RunID\", runID),\n\t\tzap.String(\"DomainID\", domainID),\n\t\tzap.String(\"DomainName\", domainName))\n\n\tworkflowResp, err := w.pr.GetWorkflowExecution(ctx, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tDomainName: domainName,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t})\n\tif err != nil {\n\t\tw.logger.Error(\"Error getting workflow execution\", zap.Error(err))\n\t\treturn err\n\t}\n\n\t// Create an instance of ConcreteExecution\n\tconcreteExecution := &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID:   workflowResp.State.ExecutionInfo.DomainID,\n\t\t\tWorkflowID: workflowResp.State.ExecutionInfo.WorkflowID,\n\t\t\tRunID:      workflowResp.State.ExecutionInfo.RunID,\n\t\t\tState:      workflowResp.State.ExecutionInfo.State,\n\t\t},\n\t}\n\n\tif w.isDomainDeprecated(domainID) {\n\t\tinvariant.DeleteExecution(ctx, concreteExecution, w.pr, w.dc)\n\t\treturn nil\n\t}\n\n\tif w.staleCheck != nil {\n\t\tstale, err := w.staleWorkflowCheck(workflowResp)\n\t\tif err != nil {\n\t\t\tw.logger.Error(\"Error checking if the execution is stale\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t\tif stale {\n\t\t\tinvariant.DeleteExecution(ctx, concreteExecution, w.pr, w.dc)\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tw.metricsClient.Scope(metrics.TaskValidatorScope, metrics.DomainTag(domainName)).IncCounter(metrics.ValidatedWorkflowCount)\n\treturn nil\n}\n\n// isDomainDeprecated checks if a domain is deprecated.\nfunc (w *checkerImpl) isDomainDeprecated(domainID string) bool {\n\tdomain, err := w.dc.GetDomainByID(domainID)\n\tif err != nil {\n\t\tw.logger.Error(\"Error getting domain by ID\", zap.Error(err))\n\t\treturn false\n\t}\n\treturn domain.IsDeprecatedOrDeleted()\n}\n\n// staleWorkflowCheck checks if a workflow is stale and returns any errors encountered.\nfunc (w *checkerImpl) staleWorkflowCheck(workflowResp *persistence.GetWorkflowExecutionResponse) (bool, error) {\n\tif workflowResp == nil || workflowResp.State == nil || workflowResp.State.ExecutionInfo == nil {\n\t\tw.logger.Error(\"Invalid workflow execution response received\")\n\t\treturn false, errors.New(\"invalid workflow execution response\")\n\t}\n\tpastExpiration, err := w.staleCheck.CheckAge(workflowResp)\n\tif err != nil {\n\t\tw.logger.Error(\"Error during stale workflow check\", zap.Error(err))\n\t\treturn false, err\n\t}\n\treturn pastExpiration, nil\n}\n"
  },
  {
    "path": "common/taskvalidator/validateworkflow_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage taskvalidator\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\ntype mockStaleChecker struct {\n\tCheckAgeFunc func(response *persistence.GetWorkflowExecutionResponse) (bool, error)\n}\n\nfunc (m *mockStaleChecker) CheckAge(response *persistence.GetWorkflowExecutionResponse) (bool, error) {\n\treturn m.CheckAgeFunc(response)\n}\n\nfunc TestWorkflowCheckforValidation(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tworkflowID    string\n\t\tdomainID      string\n\t\tdomainName    string\n\t\trunID         string\n\t\tisStale       bool\n\t\tsimulateError bool\n\t}{\n\t\t{\"NonStaleWorkflow\", \"workflow-1\", \"domain-1\", \"domain-name-1\", \"run-1\", false, false},\n\t\t{\"StaleWorkflow\", \"workflow-2\", \"domain-2\", \"domain-name-2\", \"run-2\", true, false},\n\t\t{\"ErrorInGetWorkflowExecution\", \"workflow-3\", \"domain-3\", \"domain-name-3\", \"run-3\", false, true},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\n\t\t\tmockLogger := zap.NewNop()\n\t\t\tmockMetricsClient := metrics.NewNoopMetricsClient()\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(mockCtrl)\n\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(mockCtrl)\n\n\t\t\tchecker, err := NewWfChecker(mockLogger, mockMetricsClient, mockDomainCache, mockExecutionManager, mockHistoryManager)\n\t\t\tassert.NoError(t, err, \"Failed to create checker\")\n\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(tc.domainID).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(tc.domainID).Return(tc.domainName, nil).AnyTimes()\n\n\t\t\tif tc.isStale {\n\t\t\t\tmockExecutionManager.EXPECT().DeleteWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t\tmockExecutionManager.EXPECT().DeleteCurrentWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\t\t\t}\n\n\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\tif tc.simulateError {\n\t\t\t\t\treturn nil, errors.New(\"database error\")\n\t\t\t\t}\n\t\t\t\treturn &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t}).AnyTimes()\n\n\t\t\tctx := context.Background()\n\t\t\terr = checker.WorkflowCheckforValidation(ctx, tc.workflowID, tc.domainID, tc.domainName, tc.runID)\n\n\t\t\tif tc.simulateError {\n\t\t\t\tassert.Error(t, err, \"Expected error when GetWorkflowExecution fails\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"Expected no error for valid workflow execution\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/testing/allisset.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testing\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc allIsSet(t *testing.T, err error) {\n\t// All the errors are pointers, so we get the value with .Elem\n\terrValue := reflect.ValueOf(err).Elem()\n\n\tfor i := 0; i < errValue.NumField(); i++ {\n\t\tfield := errValue.Field(i)\n\n\t\t// IsZero checks if the value is the default value (e.g. nil, \"\", 0 etc)\n\t\tassert.True(t, !field.IsZero(), \"Field %s is not set for error %s\", errValue.Type().Field(i).Name, errValue.Type())\n\t}\n}\n"
  },
  {
    "path": "common/testing/allisset_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testing\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestAllFieldsSetInTestErrors(t *testing.T) {\n\tfor _, err := range testdata.Errors {\n\t\tname := reflect.TypeOf(err).Elem().Name()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Test all fields are set in the error\n\t\t\tallIsSet(t, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/testing/event_generator.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testing\n\nimport \"math/rand\"\n\nconst (\n\temptyCandidateIndex = -1\n\tdefaultVersion      = int64(100)\n)\n\nvar (\n\tdefaultBatchFunc = func(batch []Vertex, history []Vertex) bool {\n\t\treturn len(batch) == 0\n\t}\n)\n\ntype (\n\t// HistoryEventVertex represents one type of history event\n\tHistoryEventVertex struct {\n\t\tname                 string\n\t\tisStrictOnNextVertex bool\n\t\tmaxNextGeneration    int\n\t\tdataFunc             func(...interface{}) interface{}\n\t\tdata                 interface{}\n\t}\n\n\t// HistoryEventModel is a graph represents relationships among history event types\n\tHistoryEventModel struct {\n\t\tedges []Edge\n\t}\n\n\t// EventGenerator is a history event generator\n\t// The event generator will generate next history event\n\t// based on the history event transition graph defined in the model\n\tEventGenerator struct {\n\t\tconnections         map[string][]Edge\n\t\tpreviousVertices    []Vertex\n\t\tleafVertices        []Vertex\n\t\tentryVertices       []Vertex\n\t\texitVertices        map[string]bool\n\t\trandomEntryVertices []Vertex\n\t\tdice                *rand.Rand\n\t\tseed                int64\n\t\tcanDoBatch          func([]Vertex, []Vertex) bool\n\t\tversion             int64\n\t}\n\n\t// HistoryEventEdge is the directional edge of two history events\n\tHistoryEventEdge struct {\n\t\tstartVertex Vertex\n\t\tendVertex   Vertex\n\t\tcondition   func(...interface{}) bool\n\t\taction      func()\n\t}\n)\n\n// NewEventGenerator initials the event generator\nfunc NewEventGenerator(\n\tseed int64,\n) Generator {\n\n\treturn &EventGenerator{\n\t\tconnections:         make(map[string][]Edge),\n\t\tpreviousVertices:    make([]Vertex, 0),\n\t\tleafVertices:        make([]Vertex, 0),\n\t\tentryVertices:       make([]Vertex, 0),\n\t\texitVertices:        make(map[string]bool),\n\t\trandomEntryVertices: make([]Vertex, 0),\n\t\tdice:                rand.New(rand.NewSource(seed)),\n\t\tseed:                seed,\n\t\tcanDoBatch:          defaultBatchFunc,\n\t\tversion:             defaultVersion,\n\t}\n}\n\n// AddInitialEntryVertex adds the initial history event vertices\n// Generator will only start from one of the entry vertex\nfunc (g *EventGenerator) AddInitialEntryVertex(\n\tentry ...Vertex,\n) {\n\n\tg.entryVertices = append(\n\t\tg.entryVertices,\n\t\tentry...)\n}\n\n// AddExitVertex adds the terminate history event vertex\nfunc (g *EventGenerator) AddExitVertex(\n\texit ...Vertex,\n) {\n\n\tfor _, v := range exit {\n\t\tg.exitVertices[v.GetName()] = true\n\t}\n}\n\n// AddRandomEntryVertex adds the random history event vertex\nfunc (g *EventGenerator) AddRandomEntryVertex(\n\texit ...Vertex,\n) {\n\n\tg.randomEntryVertices = append(g.randomEntryVertices, exit...)\n}\n\n// AddModel adds a history event model\nfunc (g *EventGenerator) AddModel(\n\tmodel Model,\n) {\n\n\tfor _, e := range model.ListEdges() {\n\t\tif _, ok := g.connections[e.GetStartVertex().GetName()]; !ok {\n\t\t\tg.connections[e.GetStartVertex().GetName()] = make([]Edge, 0)\n\t\t}\n\t\tg.connections[e.GetStartVertex().GetName()] = append(g.connections[e.GetStartVertex().GetName()], e)\n\t}\n}\n\n// ListGeneratedVertices returns all the generated history events\nfunc (g EventGenerator) ListGeneratedVertices() []Vertex {\n\n\treturn g.previousVertices\n}\n\n// HasNextVertex checks if there is accessible history event vertex\nfunc (g *EventGenerator) HasNextVertex() bool {\n\n\tfor _, prev := range g.previousVertices {\n\t\tif _, ok := g.exitVertices[prev.GetName()]; ok {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn len(g.leafVertices) > 0 || (len(g.previousVertices) == 0 && len(g.entryVertices) > 0)\n}\n\n// GetNextVertices generates a batch of history events happened in the same transaction\nfunc (g *EventGenerator) GetNextVertices() []Vertex {\n\n\tif !g.HasNextVertex() {\n\t\tpanic(\"Generator reached to a terminate state.\")\n\t}\n\n\tbatch := make([]Vertex, 0)\n\tfor g.HasNextVertex() && g.canDoBatch(batch, g.previousVertices) {\n\t\tres := g.generateNextEventBatch()\n\t\tg.updateContext(res)\n\t\tbatch = append(batch, res...)\n\t}\n\treturn batch\n}\n\n// Reset reset the generator to the initial state\nfunc (g *EventGenerator) Reset() {\n\n\tg.leafVertices = make([]Vertex, 0)\n\tg.previousVertices = make([]Vertex, 0)\n\tg.version = defaultVersion\n}\n\n// DeepCopy copy a new instance of generator\nfunc (g *EventGenerator) DeepCopy() Generator {\n\n\treturn &EventGenerator{\n\t\tconnections:         copyConnections(g.connections),\n\t\tpreviousVertices:    copyVertex(g.previousVertices),\n\t\tleafVertices:        copyVertex(g.leafVertices),\n\t\tentryVertices:       copyVertex(g.entryVertices),\n\t\texitVertices:        copyExitVertices(g.exitVertices),\n\t\trandomEntryVertices: copyVertex(g.randomEntryVertices),\n\t\tdice:                rand.New(rand.NewSource(g.seed)),\n\t\tseed:                g.seed,\n\t\tcanDoBatch:          g.canDoBatch,\n\t\tversion:             g.version,\n\t}\n}\n\n// SetBatchGenerationRule sets a function to determine next generated batch of history events\nfunc (g *EventGenerator) SetBatchGenerationRule(\n\tcanDoBatchFunc func([]Vertex, []Vertex) bool,\n) {\n\n\tg.canDoBatch = canDoBatchFunc\n}\n\n// SetVersion sets the event version\nfunc (g *EventGenerator) SetVersion(version int64) {\n\n\tg.version = version\n}\n\n// GetVersion returns event version\nfunc (g *EventGenerator) GetVersion() int64 {\n\n\treturn g.version\n}\n\nfunc (g *EventGenerator) generateNextEventBatch() []Vertex {\n\n\tbatch := make([]Vertex, 0)\n\tswitch {\n\tcase len(g.previousVertices) == 0:\n\t\t// Generate for the first time, get the event candidates from entry vertex group\n\t\tbatch = append(batch, g.getEntryVertex())\n\tcase len(g.randomEntryVertices) > 0 && g.dice.Intn(len(g.connections)) == 0:\n\t\t// Get the event candidate from random vertex group\n\t\tbatch = append(batch, g.getRandomVertex())\n\tdefault:\n\t\t// Get the event candidates based on context\n\t\tidx := g.getVertexCandidate()\n\t\tbatch = append(batch, g.randomNextVertex(idx)...)\n\t\tg.leafVertices = append(g.leafVertices[:idx], g.leafVertices[idx+1:]...)\n\t}\n\treturn batch\n}\n\nfunc (g *EventGenerator) updateContext(\n\tbatch []Vertex,\n) {\n\n\tg.leafVertices = append(g.leafVertices, batch...)\n\tg.previousVertices = append(g.previousVertices, batch...)\n}\n\nfunc (g *EventGenerator) getEntryVertex() Vertex {\n\n\tif len(g.entryVertices) == 0 {\n\t\tpanic(\"No possible start vertex to go to next step\")\n\t}\n\tnextRange := len(g.entryVertices)\n\tnextIdx := g.dice.Intn(nextRange)\n\tvertex := g.entryVertices[nextIdx].DeepCopy()\n\tvertex.GenerateData(nil)\n\treturn vertex\n}\n\nfunc (g *EventGenerator) getRandomVertex() Vertex {\n\n\tif len(g.randomEntryVertices) == 0 {\n\t\tpanic(\"No possible vertex to go to next step\")\n\t}\n\tnextRange := len(g.randomEntryVertices)\n\tnextIdx := g.dice.Intn(nextRange)\n\tvertex := g.randomEntryVertices[nextIdx].DeepCopy()\n\tparentEvent := g.previousVertices[len(g.previousVertices)-1]\n\n\tvertex.GenerateData(parentEvent.GetData(), parentEvent.GetData(), g.version)\n\treturn vertex\n}\n\nfunc (g *EventGenerator) getVertexCandidate() int {\n\n\tif len(g.leafVertices) == 0 {\n\t\tpanic(\"No possible vertex to go to next step\")\n\t}\n\tnextRange := len(g.leafVertices)\n\tnotAvailable := make(map[int]bool)\n\tvar nextVertexIdx int\n\tfor len(notAvailable) < nextRange {\n\t\tnextVertexIdx = g.dice.Intn(nextRange)\n\t\t// If the vertex is not accessible at this state, skip it\n\t\tif _, ok := notAvailable[nextVertexIdx]; ok {\n\t\t\tcontinue\n\t\t}\n\t\tisAccessible, nextVertexIdx := g.findAccessibleVertex(nextVertexIdx)\n\t\tif isAccessible {\n\t\t\treturn nextVertexIdx\n\t\t}\n\t\tnotAvailable[nextVertexIdx] = true\n\t}\n\t// If all history event cannot be accessible, which means the model is incorrect\n\tpanic(\"cannot find available history event to proceed. please check your model\")\n}\n\nfunc (g *EventGenerator) findAccessibleVertex(\n\tvertexIndex int,\n) (bool, int) {\n\n\tcandidate := g.leafVertices[vertexIndex]\n\tif g.leafVertices[len(g.leafVertices)-1].IsStrictOnNextVertex() {\n\t\tvertexIndex = len(g.leafVertices) - 1\n\t\tcandidate = g.leafVertices[vertexIndex]\n\t}\n\tneighbors := g.connections[candidate.GetName()]\n\tfor _, nextV := range neighbors {\n\t\tif nextV.GetCondition() == nil || nextV.GetCondition()(g.previousVertices) {\n\t\t\treturn true, vertexIndex\n\t\t}\n\t}\n\treturn false, emptyCandidateIndex\n}\n\nfunc (g *EventGenerator) randomNextVertex(\n\tnextVertexIdx int,\n) []Vertex {\n\n\tnextVertex := g.leafVertices[nextVertexIdx]\n\n\tcount := g.dice.Intn(nextVertex.GetMaxNextVertex()) + 1\n\tres := make([]Vertex, 0)\n\tlatestVertex := g.previousVertices[len(g.previousVertices)-1]\n\tfor i := 0; i < count; i++ {\n\t\tendVertex := g.pickRandomVertex(nextVertex)\n\t\tendVertex.GenerateData(nextVertex.GetData(), latestVertex.GetData(), g.version)\n\t\tlatestVertex = endVertex\n\t\tres = append(res, endVertex)\n\t\tif _, ok := g.exitVertices[endVertex.GetName()]; ok {\n\t\t\tres = []Vertex{endVertex}\n\t\t\treturn res\n\t\t}\n\t}\n\treturn res\n}\n\nfunc (g *EventGenerator) pickRandomVertex(\n\tnextVertex Vertex,\n) Vertex {\n\n\tneighbors := g.connections[nextVertex.GetName()]\n\tneighborsRange := len(neighbors)\n\tnextIdx := g.dice.Intn(neighborsRange)\n\tfor neighbors[nextIdx].GetCondition() != nil && !neighbors[nextIdx].GetCondition()(g.previousVertices) {\n\t\tnextIdx = g.dice.Intn(neighborsRange)\n\t}\n\tnewConnection := neighbors[nextIdx]\n\tendVertex := newConnection.GetEndVertex()\n\tif newConnection.GetAction() != nil {\n\t\tnewConnection.GetAction()()\n\t}\n\treturn endVertex.DeepCopy()\n}\n\n// NewHistoryEventEdge initials a new edge between two HistoryEventVertexes\nfunc NewHistoryEventEdge(\n\tstart Vertex,\n\tend Vertex,\n) Edge {\n\n\treturn &HistoryEventEdge{\n\t\tstartVertex: start,\n\t\tendVertex:   end,\n\t}\n}\n\n// SetStartVertex sets the start vertex\nfunc (c *HistoryEventEdge) SetStartVertex(\n\tstart Vertex,\n) {\n\n\tc.startVertex = start\n}\n\n// GetStartVertex returns the start vertex\nfunc (c HistoryEventEdge) GetStartVertex() Vertex {\n\n\treturn c.startVertex\n}\n\n// SetEndVertex sets the end vertex\nfunc (c *HistoryEventEdge) SetEndVertex(end Vertex) {\n\n\tc.endVertex = end\n}\n\n// GetEndVertex returns the end vertex\nfunc (c HistoryEventEdge) GetEndVertex() Vertex {\n\n\treturn c.endVertex\n}\n\n// SetCondition sets the condition to access this edge\nfunc (c *HistoryEventEdge) SetCondition(\n\tcondition func(...interface{}) bool,\n) {\n\n\tc.condition = condition\n}\n\n// GetCondition returns the condition\nfunc (c HistoryEventEdge) GetCondition() func(...interface{}) bool {\n\n\treturn c.condition\n}\n\n// SetAction sets an action to perform when the end vertex hits\nfunc (c *HistoryEventEdge) SetAction(action func()) {\n\n\tc.action = action\n}\n\n// GetAction returns the action\nfunc (c HistoryEventEdge) GetAction() func() {\n\n\treturn c.action\n}\n\n// DeepCopy copies a new edge\nfunc (c *HistoryEventEdge) DeepCopy() Edge {\n\n\treturn &HistoryEventEdge{\n\t\tstartVertex: c.startVertex.DeepCopy(),\n\t\tendVertex:   c.endVertex.DeepCopy(),\n\t\tcondition:   c.condition,\n\t\taction:      c.action,\n\t}\n}\n\n// NewHistoryEventVertex initials a history event vertex\nfunc NewHistoryEventVertex(\n\tname string,\n) Vertex {\n\n\treturn &HistoryEventVertex{\n\t\tname:                 name,\n\t\tisStrictOnNextVertex: false,\n\t\tmaxNextGeneration:    1,\n\t}\n}\n\n// GetName returns the name\nfunc (he HistoryEventVertex) GetName() string {\n\n\treturn he.name\n}\n\n// SetName sets the name\nfunc (he *HistoryEventVertex) SetName(\n\tname string,\n) {\n\n\the.name = name\n}\n\n// Equals compares two vertex\n// func (he *HistoryEventVertex) Equals(\n//\tv Vertex,\n// ) bool {\n//\n//\treturn strings.EqualFold(he.name, v.GetName()) && he.data == v.GetData()\n// }\n\n// SetIsStrictOnNextVertex sets if a vertex can be added between the current vertex and its child Vertices\nfunc (he *HistoryEventVertex) SetIsStrictOnNextVertex(\n\tisStrict bool,\n) {\n\n\the.isStrictOnNextVertex = isStrict\n}\n\n// IsStrictOnNextVertex returns the isStrict flag\nfunc (he HistoryEventVertex) IsStrictOnNextVertex() bool {\n\n\treturn he.isStrictOnNextVertex\n}\n\n// SetMaxNextVertex sets the max concurrent path can be generated from this vertex\nfunc (he *HistoryEventVertex) SetMaxNextVertex(\n\tmaxNextGeneration int,\n) {\n\n\tif maxNextGeneration < 1 {\n\t\tpanic(\"max next vertex number cannot less than 1\")\n\t}\n\the.maxNextGeneration = maxNextGeneration\n}\n\n// GetMaxNextVertex returns the max concurrent path\nfunc (he HistoryEventVertex) GetMaxNextVertex() int {\n\n\treturn he.maxNextGeneration\n}\n\n// SetDataFunc sets the data generation function\nfunc (he *HistoryEventVertex) SetDataFunc(\n\tdataFunc func(...interface{}) interface{},\n) {\n\n\the.dataFunc = dataFunc\n}\n\n// GetDataFunc returns the data generation function\nfunc (he HistoryEventVertex) GetDataFunc() func(...interface{}) interface{} {\n\n\treturn he.dataFunc\n}\n\n// GenerateData generates the data and return\nfunc (he *HistoryEventVertex) GenerateData(\n\tinput ...interface{},\n) interface{} {\n\n\tif he.dataFunc == nil {\n\t\treturn nil\n\t}\n\n\the.data = he.dataFunc(input...)\n\treturn he.data\n}\n\n// GetData returns the vertex data\nfunc (he HistoryEventVertex) GetData() interface{} {\n\n\treturn he.data\n}\n\n// DeepCopy returns the a deep copy of vertex\nfunc (he HistoryEventVertex) DeepCopy() Vertex {\n\n\treturn &HistoryEventVertex{\n\t\tname:                 he.GetName(),\n\t\tisStrictOnNextVertex: he.IsStrictOnNextVertex(),\n\t\tmaxNextGeneration:    he.GetMaxNextVertex(),\n\t\tdataFunc:             he.GetDataFunc(),\n\t\tdata:                 he.GetData(),\n\t}\n}\n\n// NewHistoryEventModel initials new history event model\nfunc NewHistoryEventModel() Model {\n\n\treturn &HistoryEventModel{\n\t\tedges: make([]Edge, 0),\n\t}\n}\n\n// AddEdge adds an edge to the model\nfunc (m *HistoryEventModel) AddEdge(\n\tedge ...Edge,\n) {\n\n\tm.edges = append(m.edges, edge...)\n}\n\n// ListEdges returns all added edges\nfunc (m HistoryEventModel) ListEdges() []Edge {\n\n\treturn m.edges\n}\n"
  },
  {
    "path": "common/testing/generator_interface.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testing\n\ntype (\n\t// Model represents a state transition graph that contains all the relationships of Vertex\n\tModel interface {\n\t\tAddEdge(...Edge)\n\t\tListEdges() []Edge\n\t}\n\n\t// Generator generates a sequence of vertices based on the defined models\n\t// It must define InitialEntryVertex and ExitVertex\n\t// To use a generator:\n\t// for generator.HasNextVertex {\n\t//     generator.GetNextVertices\n\t// }\n\tGenerator interface {\n\t\t// InitialEntryVertex is the beginning vertices of the graph\n\t\t// Only one vertex will be picked as the entry\n\t\tAddInitialEntryVertex(...Vertex)\n\t\t// ExitVertex is the terminate vertices of the graph\n\t\tAddExitVertex(...Vertex)\n\t\t// RandomEntryVertex is a random entry point which can be access at any state of the generator\n\t\tAddRandomEntryVertex(...Vertex)\n\t\t// AddModel loads model into the generator\n\t\t// AddModel can load multiple models and models will be joint if there is common vertices\n\t\tAddModel(Model)\n\t\t// HasNextVertex determines if there is more vertex to generate\n\t\tHasNextVertex() bool\n\t\t// GetNextVertices generates next vertex batch\n\t\tGetNextVertices() []Vertex\n\t\t// ListGeneratedVertices lists the pasted generated vertices\n\t\tListGeneratedVertices() []Vertex\n\t\t// Reset cleans up all the internal states and reset to a brand new generator\n\t\tReset()\n\t\t// DeepCopy copy a new instance of generator\n\t\tDeepCopy() Generator\n\t\t// SetBatchGenerationRule sets a function that used in GetNextVertex to return batch result\n\t\tSetBatchGenerationRule(func([]Vertex, []Vertex) bool)\n\t\t// SetVersion sets the event version\n\t\tSetVersion(int64)\n\t\t// GetVersion gets the event version\n\t\tGetVersion() int64\n\t}\n\n\t// Vertex represents a state in the model. A state represents a type of an Cadence event\n\tVertex interface {\n\t\t// The name of the vertex. Usually, this will be the Cadence event type\n\t\tSetName(string)\n\t\tGetName() string\n\t\t// Equals(Vertex) bool\n\t\t// IsStrictOnNextVertex means if the vertex must be followed by its children\n\t\t// When IsStrictOnNextVertex set to true, it means this event can only follow by its neighbors\n\t\tSetIsStrictOnNextVertex(bool)\n\t\tIsStrictOnNextVertex() bool\n\t\t// MaxNextVertex means the max neighbors can branch out from this vertex\n\t\tSetMaxNextVertex(int)\n\t\tGetMaxNextVertex() int\n\n\t\t// SetVertexDataFunc sets a function to generate end vertex data\n\t\tSetDataFunc(func(...interface{}) interface{})\n\t\tGetDataFunc() func(...interface{}) interface{}\n\t\tGenerateData(...interface{}) interface{}\n\t\tGetData() interface{}\n\t\tDeepCopy() Vertex\n\t}\n\n\t// Edge is the connection between two vertices\n\tEdge interface {\n\t\t// StartVertex is the head of the connection\n\t\tSetStartVertex(Vertex)\n\t\tGetStartVertex() Vertex\n\t\t// EndVertex is the end of the connection\n\t\tSetEndVertex(Vertex)\n\t\tGetEndVertex() Vertex\n\t\t// Condition defines a function to determine if this connection is accessible\n\t\tSetCondition(func(...interface{}) bool)\n\t\tGetCondition() func(...interface{}) bool\n\t\t// Action defines function to perform when the end vertex reached\n\t\tSetAction(func())\n\t\tGetAction() func()\n\t\tDeepCopy() Edge\n\t}\n)\n"
  },
  {
    "path": "common/testing/history_event_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testing\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\thistoryEventTestSuit struct {\n\t\tsuite.Suite\n\t\tgenerator Generator\n\t}\n)\n\nfunc TestHistoryEventTestSuite(t *testing.T) {\n\tsuite.Run(t, new(historyEventTestSuit))\n}\n\nfunc (s *historyEventTestSuit) SetupSuite() {\n\ts.generator = InitializeHistoryEventGenerator(\"domain\", 1)\n}\n\nfunc (s *historyEventTestSuit) SetupTest() {\n\ts.generator.Reset()\n}\n\n// This is a sample about how to use the generator\nfunc (s *historyEventTestSuit) Test_HistoryEvent_Generator() {\n\tmaxEventID := int64(0)\n\tmaxVersion := int64(1)\n\tmaxTaskID := int64(1)\n\tfor i := 0; i < 10 && s.generator.HasNextVertex(); i++ {\n\t\tevents := s.generator.GetNextVertices()\n\n\t\tfmt.Println(\"########################\")\n\t\tfor _, e := range events {\n\t\t\tevent := e.GetData().(*types.HistoryEvent)\n\t\t\tif maxEventID != event.ID-1 {\n\t\t\t\ts.Fail(\"event id sequence is incorrect\")\n\t\t\t}\n\t\t\tmaxEventID = event.ID\n\t\t\tif maxVersion > event.Version {\n\t\t\t\ts.Fail(\"event version is incorrect\")\n\t\t\t}\n\t\t\tmaxVersion = event.Version\n\t\t\tif maxTaskID > event.TaskID {\n\t\t\t\ts.Fail(\"event task id is incorrect\")\n\t\t\t}\n\t\t\tmaxTaskID = event.TaskID\n\t\t\tfmt.Println(e.GetName())\n\t\t\tfmt.Println(event.ID)\n\t\t}\n\t}\n\ts.NotEmpty(s.generator.ListGeneratedVertices())\n\tfmt.Println(\"==========================\")\n\tbranchGenerator1 := s.generator.DeepCopy()\n\tfor i := 0; i < 10 && branchGenerator1.HasNextVertex(); i++ {\n\t\tevents := branchGenerator1.GetNextVertices()\n\t\tfmt.Println(\"########################\")\n\t\tfor _, e := range events {\n\t\t\tevent := e.GetData().(*types.HistoryEvent)\n\t\t\tif maxEventID != event.ID-1 {\n\t\t\t\ts.Fail(\"event id sequence is incorrect\")\n\t\t\t}\n\t\t\tmaxEventID = event.ID\n\t\t\tif maxVersion > event.Version {\n\t\t\t\ts.Fail(\"event version is incorrect\")\n\t\t\t}\n\t\t\tmaxVersion = event.Version\n\t\t\tif maxTaskID > event.TaskID {\n\t\t\t\ts.Fail(\"event task id is incorrect\")\n\t\t\t}\n\t\t\tmaxTaskID = event.TaskID\n\t\t\tfmt.Println(e.GetName())\n\t\t\tfmt.Println(event.ID)\n\t\t}\n\t}\n\tfmt.Println(\"==========================\")\n\thistory := s.generator.ListGeneratedVertices()\n\tmaxEventID = history[len(history)-1].GetData().(*types.HistoryEvent).ID\n\tfor i := 0; i < 10 && s.generator.HasNextVertex(); i++ {\n\t\tevents := s.generator.GetNextVertices()\n\t\tfmt.Println(\"########################\")\n\t\tfor _, e := range events {\n\t\t\tevent := e.GetData().(*types.HistoryEvent)\n\t\t\tif maxEventID != event.ID-1 {\n\t\t\t\ts.Fail(\"event id sequence is incorrect\")\n\t\t\t}\n\t\t\tmaxEventID = event.ID\n\t\t\tif maxVersion > event.Version {\n\t\t\t\ts.Fail(\"event version is incorrect\")\n\t\t\t}\n\t\t\tmaxVersion = event.Version\n\t\t\tif maxTaskID > event.TaskID {\n\t\t\t\ts.Fail(\"event task id is incorrect\")\n\t\t\t}\n\t\t\tmaxTaskID = event.TaskID\n\t\t\tfmt.Println(e.GetName())\n\t\t\tfmt.Println(event.ID)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "common/testing/history_event_util.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testing\n\nimport (\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttimeout              = int32(10000)\n\tsignal               = \"NDC signal\"\n\tchecksum             = \"NDC checksum\"\n\tchildWorkflowPrefix  = \"child-\"\n\treason               = \"NDC reason\"\n\tworkflowType         = \"test-workflow-type\"\n\ttaskList             = \"taskList\"\n\tidentity             = \"identity\"\n\tdecisionTaskAttempts = 0\n\tchildWorkflowID      = \"child-WorkflowID\"\n\texternalWorkflowID   = \"external-WorkflowID\"\n)\n\nvar (\n\tglobalTaskID int64 = 1\n)\n\n// InitializeHistoryEventGenerator initializes the history event generator\nfunc InitializeHistoryEventGenerator(\n\tdomain string,\n\tdefaultVersion int64,\n) Generator {\n\n\tgenerator := NewEventGenerator(time.Now().UnixNano())\n\tgenerator.SetVersion(defaultVersion)\n\t// Functions\n\tnotPendingDecisionTask := func(input ...interface{}) bool {\n\t\tcount := 0\n\t\thistory := input[0].([]Vertex)\n\t\tfor _, e := range history {\n\t\t\tswitch e.GetName() {\n\t\t\tcase types.EventTypeDecisionTaskScheduled.String():\n\t\t\t\tcount++\n\t\t\tcase types.EventTypeDecisionTaskCompleted.String(),\n\t\t\t\ttypes.EventTypeDecisionTaskFailed.String(),\n\t\t\t\ttypes.EventTypeDecisionTaskTimedOut.String():\n\t\t\t\tcount--\n\t\t\t}\n\t\t}\n\t\treturn count <= 0\n\t}\n\tcontainActivityComplete := func(input ...interface{}) bool {\n\t\thistory := input[0].([]Vertex)\n\t\tfor _, e := range history {\n\t\t\tif e.GetName() == types.EventTypeActivityTaskCompleted.String() {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t\treturn false\n\t}\n\thasPendingActivity := func(input ...interface{}) bool {\n\t\tcount := 0\n\t\thistory := input[0].([]Vertex)\n\t\tfor _, e := range history {\n\t\t\tswitch e.GetName() {\n\t\t\tcase types.EventTypeActivityTaskScheduled.String():\n\t\t\t\tcount++\n\t\t\tcase types.EventTypeActivityTaskCanceled.String(),\n\t\t\t\ttypes.EventTypeActivityTaskFailed.String(),\n\t\t\t\ttypes.EventTypeActivityTaskTimedOut.String(),\n\t\t\t\ttypes.EventTypeActivityTaskCompleted.String():\n\t\t\t\tcount--\n\t\t\t}\n\t\t}\n\t\treturn count > 0\n\t}\n\tcanDoBatch := func(currentBatch []Vertex, history []Vertex) bool {\n\t\tif len(currentBatch) == 0 {\n\t\t\treturn true\n\t\t}\n\n\t\thasPendingDecisionTask := false\n\t\tfor _, event := range history {\n\t\t\tswitch event.GetName() {\n\t\t\tcase types.EventTypeDecisionTaskScheduled.String():\n\t\t\t\thasPendingDecisionTask = true\n\t\t\tcase types.EventTypeDecisionTaskCompleted.String(),\n\t\t\t\ttypes.EventTypeDecisionTaskFailed.String(),\n\t\t\t\ttypes.EventTypeDecisionTaskTimedOut.String():\n\t\t\t\thasPendingDecisionTask = false\n\t\t\t}\n\t\t}\n\t\tif hasPendingDecisionTask {\n\t\t\treturn false\n\t\t}\n\t\tif currentBatch[len(currentBatch)-1].GetName() == types.EventTypeDecisionTaskScheduled.String() {\n\t\t\treturn false\n\t\t}\n\t\tif currentBatch[0].GetName() == types.EventTypeDecisionTaskCompleted.String() {\n\t\t\treturn len(currentBatch) == 1\n\t\t}\n\t\treturn true\n\t}\n\n\t// Setup decision task model\n\tdecisionModel := NewHistoryEventModel()\n\tdecisionSchedule := NewHistoryEventVertex(types.EventTypeDecisionTaskScheduled.String())\n\tdecisionSchedule.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeDecisionTaskScheduled.Ptr()\n\t\thistoryEvent.DecisionTaskScheduledEventAttributes = &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tAttempt:                    decisionTaskAttempts,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionStart := NewHistoryEventVertex(types.EventTypeDecisionTaskStarted.String())\n\tdecisionStart.SetIsStrictOnNextVertex(true)\n\tdecisionStart.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeDecisionTaskStarted.Ptr()\n\t\thistoryEvent.DecisionTaskStartedEventAttributes = &types.DecisionTaskStartedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.ID,\n\t\t\tIdentity:         identity,\n\t\t\tRequestID:        uuid.New(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionFail := NewHistoryEventVertex(types.EventTypeDecisionTaskFailed.String())\n\tdecisionFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeDecisionTaskFailed.Ptr()\n\t\thistoryEvent.DecisionTaskFailedEventAttributes = &types.DecisionTaskFailedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetDecisionTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tCause:            types.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\t\tIdentity:         identity,\n\t\t\tForkEventVersion: version,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionTimedOut := NewHistoryEventVertex(types.EventTypeDecisionTaskTimedOut.String())\n\tdecisionTimedOut.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeDecisionTaskTimedOut.Ptr()\n\t\thistoryEvent.DecisionTaskTimedOutEventAttributes = &types.DecisionTaskTimedOutEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetDecisionTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tTimeoutType:      types.TimeoutTypeScheduleToStart.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionComplete := NewHistoryEventVertex(types.EventTypeDecisionTaskCompleted.String())\n\tdecisionComplete.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeDecisionTaskCompleted.Ptr()\n\t\thistoryEvent.DecisionTaskCompletedEventAttributes = &types.DecisionTaskCompletedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetDecisionTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tIdentity:         identity,\n\t\t\tBinaryChecksum:   checksum,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionComplete.SetIsStrictOnNextVertex(true)\n\tdecisionComplete.SetMaxNextVertex(2)\n\tdecisionScheduleToStart := NewHistoryEventEdge(decisionSchedule, decisionStart)\n\tdecisionStartToComplete := NewHistoryEventEdge(decisionStart, decisionComplete)\n\tdecisionStartToFail := NewHistoryEventEdge(decisionStart, decisionFail)\n\tdecisionStartToTimedOut := NewHistoryEventEdge(decisionStart, decisionTimedOut)\n\tdecisionFailToSchedule := NewHistoryEventEdge(decisionFail, decisionSchedule)\n\tdecisionFailToSchedule.SetCondition(notPendingDecisionTask)\n\tdecisionTimedOutToSchedule := NewHistoryEventEdge(decisionTimedOut, decisionSchedule)\n\tdecisionTimedOutToSchedule.SetCondition(notPendingDecisionTask)\n\tdecisionModel.AddEdge(decisionScheduleToStart, decisionStartToComplete, decisionStartToFail, decisionStartToTimedOut,\n\t\tdecisionFailToSchedule, decisionTimedOutToSchedule)\n\n\t// Setup workflow model\n\tworkflowModel := NewHistoryEventModel()\n\n\tworkflowStart := NewHistoryEventVertex(types.EventTypeWorkflowExecutionStarted.String())\n\tworkflowStart.SetDataFunc(func(input ...interface{}) interface{} {\n\t\thistoryEvent := getDefaultHistoryEvent(1, defaultVersion)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionStarted.Ptr()\n\t\thistoryEvent.WorkflowExecutionStartedEventAttributes = &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: workflowType,\n\t\t\t},\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(timeout),\n\t\t\tIdentity:                            identity,\n\t\t\tFirstExecutionRunID:                 uuid.New(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowSignal := NewHistoryEventVertex(types.EventTypeWorkflowExecutionSignaled.String())\n\tworkflowSignal.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionSignaled.Ptr()\n\t\thistoryEvent.WorkflowExecutionSignaledEventAttributes = &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tSignalName: signal,\n\t\t\tIdentity:   identity,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowComplete := NewHistoryEventVertex(types.EventTypeWorkflowExecutionCompleted.String())\n\tworkflowComplete.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tEventID := lastEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionCompleted.Ptr()\n\t\thistoryEvent.WorkflowExecutionCompletedEventAttributes = &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tcontinueAsNew := NewHistoryEventVertex(types.EventTypeWorkflowExecutionContinuedAsNew.String())\n\tcontinueAsNew.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tEventID := lastEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionContinuedAsNew.Ptr()\n\t\thistoryEvent.WorkflowExecutionContinuedAsNewEventAttributes = &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tNewExecutionRunID: uuid.New(),\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: workflowType,\n\t\t\t},\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(timeout),\n\t\t\tDecisionTaskCompletedEventID:        EventID - 1,\n\t\t\tInitiator:                           types.ContinueAsNewInitiatorDecider.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowFail := NewHistoryEventVertex(types.EventTypeWorkflowExecutionFailed.String())\n\tworkflowFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tEventID := lastEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionFailed.Ptr()\n\t\thistoryEvent.WorkflowExecutionFailedEventAttributes = &types.WorkflowExecutionFailedEventAttributes{\n\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowCancel := NewHistoryEventVertex(types.EventTypeWorkflowExecutionCanceled.String())\n\tworkflowCancel.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionCanceled.Ptr()\n\t\thistoryEvent.WorkflowExecutionCanceledEventAttributes = &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowCancelRequest := NewHistoryEventVertex(types.EventTypeWorkflowExecutionCancelRequested.String())\n\tworkflowCancelRequest.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionCancelRequested.Ptr()\n\t\thistoryEvent.WorkflowExecutionCancelRequestedEventAttributes = &types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\t\tCause:                    \"\",\n\t\t\tExternalInitiatedEventID: common.Int64Ptr(1),\n\t\t\tExternalWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: externalWorkflowID,\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowTerminate := NewHistoryEventVertex(types.EventTypeWorkflowExecutionTerminated.String())\n\tworkflowTerminate.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tEventID := lastEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionTerminated.Ptr()\n\t\thistoryEvent.WorkflowExecutionTerminatedEventAttributes = &types.WorkflowExecutionTerminatedEventAttributes{\n\t\t\tIdentity: identity,\n\t\t\tReason:   reason,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowTimedOut := NewHistoryEventVertex(types.EventTypeWorkflowExecutionTimedOut.String())\n\tworkflowTimedOut.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tEventID := lastEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeWorkflowExecutionTimedOut.Ptr()\n\t\thistoryEvent.WorkflowExecutionTimedOutEventAttributes = &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tworkflowStartToSignal := NewHistoryEventEdge(workflowStart, workflowSignal)\n\tworkflowStartToDecisionSchedule := NewHistoryEventEdge(workflowStart, decisionSchedule)\n\tworkflowStartToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tworkflowSignalToDecisionSchedule := NewHistoryEventEdge(workflowSignal, decisionSchedule)\n\tworkflowSignalToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tdecisionCompleteToWorkflowComplete := NewHistoryEventEdge(decisionComplete, workflowComplete)\n\tdecisionCompleteToWorkflowComplete.SetCondition(containActivityComplete)\n\tdecisionCompleteToWorkflowFailed := NewHistoryEventEdge(decisionComplete, workflowFail)\n\tdecisionCompleteToWorkflowFailed.SetCondition(containActivityComplete)\n\tdecisionCompleteToCAN := NewHistoryEventEdge(decisionComplete, continueAsNew)\n\tdecisionCompleteToCAN.SetCondition(containActivityComplete)\n\tworkflowCancelRequestToCancel := NewHistoryEventEdge(workflowCancelRequest, workflowCancel)\n\tworkflowModel.AddEdge(workflowStartToSignal, workflowStartToDecisionSchedule, workflowSignalToDecisionSchedule,\n\t\tdecisionCompleteToCAN, decisionCompleteToWorkflowComplete, decisionCompleteToWorkflowFailed, workflowCancelRequestToCancel)\n\n\t// Setup activity model\n\tactivityModel := NewHistoryEventModel()\n\tactivitySchedule := NewHistoryEventVertex(types.EventTypeActivityTaskScheduled.String())\n\tactivitySchedule.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskScheduled.Ptr()\n\t\thistoryEvent.ActivityTaskScheduledEventAttributes = &types.ActivityTaskScheduledEventAttributes{\n\t\t\tActivityID: uuid.New(),\n\t\t\tActivityType: &types.ActivityType{\n\t\t\t\tName: \"activity\",\n\t\t\t},\n\t\t\tDomain: common.StringPtr(domain),\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(timeout),\n\t\t\tDecisionTaskCompletedEventID:  lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityStart := NewHistoryEventVertex(types.EventTypeActivityTaskStarted.String())\n\tactivityStart.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskStarted.Ptr()\n\t\thistoryEvent.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.ID,\n\t\t\tIdentity:         identity,\n\t\t\tRequestID:        uuid.New(),\n\t\t\tAttempt:          0,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityComplete := NewHistoryEventVertex(types.EventTypeActivityTaskCompleted.String())\n\tactivityComplete.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskCompleted.Ptr()\n\t\thistoryEvent.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetActivityTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tIdentity:         identity,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityFail := NewHistoryEventVertex(types.EventTypeActivityTaskFailed.String())\n\tactivityFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskFailed.Ptr()\n\t\thistoryEvent.ActivityTaskFailedEventAttributes = &types.ActivityTaskFailedEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetActivityTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tIdentity:         identity,\n\t\t\tReason:           common.StringPtr(reason),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityTimedOut := NewHistoryEventVertex(types.EventTypeActivityTaskTimedOut.String())\n\tactivityTimedOut.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskTimedOut.Ptr()\n\t\thistoryEvent.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\tScheduledEventID: lastEvent.GetActivityTaskStartedEventAttributes().ScheduledEventID,\n\t\t\tStartedEventID:   lastEvent.ID,\n\t\t\tTimeoutType:      types.TimeoutTypeScheduleToClose.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityCancelRequest := NewHistoryEventVertex(types.EventTypeActivityTaskCancelRequested.String())\n\tactivityCancelRequest.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskCancelRequested.Ptr()\n\t\thistoryEvent.ActivityTaskCancelRequestedEventAttributes = &types.ActivityTaskCancelRequestedEventAttributes{\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetActivityTaskScheduledEventAttributes().DecisionTaskCompletedEventID,\n\t\t\tActivityID:                   lastEvent.GetActivityTaskScheduledEventAttributes().ActivityID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityCancel := NewHistoryEventVertex(types.EventTypeActivityTaskCanceled.String())\n\tactivityCancel.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeActivityTaskCanceled.Ptr()\n\t\thistoryEvent.ActivityTaskCanceledEventAttributes = &types.ActivityTaskCanceledEventAttributes{\n\t\t\tLatestCancelRequestedEventID: lastEvent.ID,\n\t\t\tScheduledEventID:             lastEvent.ID,\n\t\t\tStartedEventID:               lastEvent.ID,\n\t\t\tIdentity:                     identity,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tactivityCancelRequestFail := NewHistoryEventVertex(types.EventTypeRequestCancelActivityTaskFailed.String())\n\tactivityCancelRequestFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversionBump := input[2].(int64)\n\t\tsubVersion := input[3].(int64)\n\t\tversion := lastGeneratedEvent.Version + versionBump + subVersion\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeRequestCancelActivityTaskFailed.Ptr()\n\t\thistoryEvent.RequestCancelActivityTaskFailedEventAttributes = &types.RequestCancelActivityTaskFailedEventAttributes{\n\t\t\tActivityID:                   uuid.New(),\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetActivityTaskCancelRequestedEventAttributes().DecisionTaskCompletedEventID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionCompleteToATSchedule := NewHistoryEventEdge(decisionComplete, activitySchedule)\n\n\tactivityScheduleToStart := NewHistoryEventEdge(activitySchedule, activityStart)\n\tactivityScheduleToStart.SetCondition(hasPendingActivity)\n\n\tactivityStartToComplete := NewHistoryEventEdge(activityStart, activityComplete)\n\tactivityStartToComplete.SetCondition(hasPendingActivity)\n\n\tactivityStartToFail := NewHistoryEventEdge(activityStart, activityFail)\n\tactivityStartToFail.SetCondition(hasPendingActivity)\n\n\tactivityStartToTimedOut := NewHistoryEventEdge(activityStart, activityTimedOut)\n\tactivityStartToTimedOut.SetCondition(hasPendingActivity)\n\n\tactivityCompleteToDecisionSchedule := NewHistoryEventEdge(activityComplete, decisionSchedule)\n\tactivityCompleteToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tactivityFailToDecisionSchedule := NewHistoryEventEdge(activityFail, decisionSchedule)\n\tactivityFailToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tactivityTimedOutToDecisionSchedule := NewHistoryEventEdge(activityTimedOut, decisionSchedule)\n\tactivityTimedOutToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tactivityCancelToDecisionSchedule := NewHistoryEventEdge(activityCancel, decisionSchedule)\n\tactivityCancelToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\n\t// TODO: bypass activity cancel request event. Support this event later.\n\t// activityScheduleToActivityCancelRequest := NewHistoryEventEdge(activitySchedule, activityCancelRequest)\n\t// activityScheduleToActivityCancelRequest.SetCondition(hasPendingActivity)\n\tactivityCancelReqToCancel := NewHistoryEventEdge(activityCancelRequest, activityCancel)\n\tactivityCancelReqToCancel.SetCondition(hasPendingActivity)\n\n\tactivityCancelReqToCancelFail := NewHistoryEventEdge(activityCancelRequest, activityCancelRequestFail)\n\tactivityCancelRequestFailToDecisionSchedule := NewHistoryEventEdge(activityCancelRequestFail, decisionSchedule)\n\tactivityCancelRequestFailToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\n\tactivityModel.AddEdge(decisionCompleteToATSchedule, activityScheduleToStart, activityStartToComplete,\n\t\tactivityStartToFail, activityStartToTimedOut, decisionCompleteToATSchedule, activityCompleteToDecisionSchedule,\n\t\tactivityFailToDecisionSchedule, activityTimedOutToDecisionSchedule, activityCancelReqToCancel,\n\t\tactivityCancelReqToCancelFail, activityCancelToDecisionSchedule, activityCancelRequestFailToDecisionSchedule)\n\n\t// Setup timer model\n\ttimerModel := NewHistoryEventModel()\n\ttimerStart := NewHistoryEventVertex(types.EventTypeTimerStarted.String())\n\ttimerStart.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeTimerStarted.Ptr()\n\t\thistoryEvent.TimerStartedEventAttributes = &types.TimerStartedEventAttributes{\n\t\t\tTimerID:                      uuid.New(),\n\t\t\tStartToFireTimeoutSeconds:    common.Int64Ptr(10),\n\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\ttimerFired := NewHistoryEventVertex(types.EventTypeTimerFired.String())\n\ttimerFired.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeTimerFired.Ptr()\n\t\thistoryEvent.TimerFiredEventAttributes = &types.TimerFiredEventAttributes{\n\t\t\tTimerID:        lastEvent.GetTimerStartedEventAttributes().TimerID,\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\ttimerCancel := NewHistoryEventVertex(types.EventTypeTimerCanceled.String())\n\ttimerCancel.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeTimerCanceled.Ptr()\n\t\thistoryEvent.TimerCanceledEventAttributes = &types.TimerCanceledEventAttributes{\n\t\t\tTimerID:                      lastEvent.GetTimerStartedEventAttributes().TimerID,\n\t\t\tStartedEventID:               lastEvent.ID,\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetTimerStartedEventAttributes().DecisionTaskCompletedEventID,\n\t\t\tIdentity:                     identity,\n\t\t}\n\t\treturn historyEvent\n\t})\n\ttimerStartToFire := NewHistoryEventEdge(timerStart, timerFired)\n\ttimerStartToCancel := NewHistoryEventEdge(timerStart, timerCancel)\n\n\tdecisionCompleteToTimerStart := NewHistoryEventEdge(decisionComplete, timerStart)\n\ttimerFiredToDecisionSchedule := NewHistoryEventEdge(timerFired, decisionSchedule)\n\ttimerFiredToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\ttimerCancelToDecisionSchedule := NewHistoryEventEdge(timerCancel, decisionSchedule)\n\ttimerCancelToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\ttimerModel.AddEdge(timerStartToFire, timerStartToCancel, decisionCompleteToTimerStart, timerFiredToDecisionSchedule, timerCancelToDecisionSchedule)\n\n\t// Setup child workflow model\n\tchildWorkflowModel := NewHistoryEventModel()\n\tchildWorkflowInitial := NewHistoryEventVertex(types.EventTypeStartChildWorkflowExecutionInitiated.String())\n\tchildWorkflowInitial.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeStartChildWorkflowExecutionInitiated.Ptr()\n\t\thistoryEvent.StartChildWorkflowExecutionInitiatedEventAttributes = &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain:     domain,\n\t\t\tWorkflowID: childWorkflowID,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(timeout),\n\t\t\tDecisionTaskCompletedEventID:        lastEvent.ID,\n\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyRejectDuplicate.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowInitialFail := NewHistoryEventVertex(types.EventTypeStartChildWorkflowExecutionFailed.String())\n\tchildWorkflowInitialFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeStartChildWorkflowExecutionFailed.Ptr()\n\t\thistoryEvent.StartChildWorkflowExecutionFailedEventAttributes = &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\t\tDomain:     domain,\n\t\t\tWorkflowID: childWorkflowID,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tCause:                        types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t\t\tInitiatedEventID:             lastEvent.ID,\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetStartChildWorkflowExecutionInitiatedEventAttributes().DecisionTaskCompletedEventID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowStart := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionStarted.String())\n\tchildWorkflowStart.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionStarted.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.ID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowCancel := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionCanceled.String())\n\tchildWorkflowCancel.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionCanceled.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionCanceledEventAttributes = &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.GetChildWorkflowExecutionStartedEventAttributes().InitiatedEventID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      lastEvent.GetChildWorkflowExecutionStartedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowComplete := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionCompleted.String())\n\tchildWorkflowComplete.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionCompleted.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.GetChildWorkflowExecutionStartedEventAttributes().InitiatedEventID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      lastEvent.GetChildWorkflowExecutionStartedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowFail := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionFailed.String())\n\tchildWorkflowFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionFailed.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionFailedEventAttributes = &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.GetChildWorkflowExecutionStartedEventAttributes().InitiatedEventID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      lastEvent.GetChildWorkflowExecutionStartedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowTerminate := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionTerminated.String())\n\tchildWorkflowTerminate.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionTerminated.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionTerminatedEventAttributes = &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.GetChildWorkflowExecutionStartedEventAttributes().InitiatedEventID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      lastEvent.GetChildWorkflowExecutionStartedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\tchildWorkflowTimedOut := NewHistoryEventVertex(types.EventTypeChildWorkflowExecutionTimedOut.String())\n\tchildWorkflowTimedOut.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeChildWorkflowExecutionTimedOut.Ptr()\n\t\thistoryEvent.ChildWorkflowExecutionTimedOutEventAttributes = &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\tDomain: domain,\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: childWorkflowPrefix + workflowType,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.GetChildWorkflowExecutionStartedEventAttributes().InitiatedEventID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: childWorkflowID,\n\t\t\t\tRunID:      lastEvent.GetChildWorkflowExecutionStartedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tStartedEventID: lastEvent.ID,\n\t\t\tTimeoutType:    types.TimeoutTypeScheduleToClose.Ptr(),\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionCompleteToChildWorkflowInitial := NewHistoryEventEdge(decisionComplete, childWorkflowInitial)\n\tchildWorkflowInitialToFail := NewHistoryEventEdge(childWorkflowInitial, childWorkflowInitialFail)\n\tchildWorkflowInitialToStart := NewHistoryEventEdge(childWorkflowInitial, childWorkflowStart)\n\tchildWorkflowStartToCancel := NewHistoryEventEdge(childWorkflowStart, childWorkflowCancel)\n\tchildWorkflowStartToFail := NewHistoryEventEdge(childWorkflowStart, childWorkflowFail)\n\tchildWorkflowStartToComplete := NewHistoryEventEdge(childWorkflowStart, childWorkflowComplete)\n\tchildWorkflowStartToTerminate := NewHistoryEventEdge(childWorkflowStart, childWorkflowTerminate)\n\tchildWorkflowStartToTimedOut := NewHistoryEventEdge(childWorkflowStart, childWorkflowTimedOut)\n\tchildWorkflowCancelToDecisionSchedule := NewHistoryEventEdge(childWorkflowCancel, decisionSchedule)\n\tchildWorkflowCancelToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowFailToDecisionSchedule := NewHistoryEventEdge(childWorkflowFail, decisionSchedule)\n\tchildWorkflowFailToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowCompleteToDecisionSchedule := NewHistoryEventEdge(childWorkflowComplete, decisionSchedule)\n\tchildWorkflowCompleteToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowTerminateToDecisionSchedule := NewHistoryEventEdge(childWorkflowTerminate, decisionSchedule)\n\tchildWorkflowTerminateToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowTimedOutToDecisionSchedule := NewHistoryEventEdge(childWorkflowTimedOut, decisionSchedule)\n\tchildWorkflowTimedOutToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowInitialFailToDecisionSchedule := NewHistoryEventEdge(childWorkflowInitialFail, decisionSchedule)\n\tchildWorkflowInitialFailToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\tchildWorkflowModel.AddEdge(decisionCompleteToChildWorkflowInitial, childWorkflowInitialToFail, childWorkflowInitialToStart,\n\t\tchildWorkflowStartToCancel, childWorkflowStartToFail, childWorkflowStartToComplete, childWorkflowStartToTerminate,\n\t\tchildWorkflowStartToTimedOut, childWorkflowCancelToDecisionSchedule, childWorkflowFailToDecisionSchedule,\n\t\tchildWorkflowCompleteToDecisionSchedule, childWorkflowTerminateToDecisionSchedule, childWorkflowTimedOutToDecisionSchedule,\n\t\tchildWorkflowInitialFailToDecisionSchedule)\n\n\t// Setup external workflow model\n\texternalWorkflowModel := NewHistoryEventModel()\n\texternalWorkflowSignal := NewHistoryEventVertex(types.EventTypeSignalExternalWorkflowExecutionInitiated.String())\n\texternalWorkflowSignal.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeSignalExternalWorkflowExecutionInitiated.Ptr()\n\t\thistoryEvent.SignalExternalWorkflowExecutionInitiatedEventAttributes = &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t\tDomain:                       domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: externalWorkflowID,\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tSignalName:        \"signal\",\n\t\t\tChildWorkflowOnly: false,\n\t\t}\n\t\treturn historyEvent\n\t})\n\texternalWorkflowSignalFailed := NewHistoryEventVertex(types.EventTypeSignalExternalWorkflowExecutionFailed.String())\n\texternalWorkflowSignalFailed.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeSignalExternalWorkflowExecutionFailed.Ptr()\n\t\thistoryEvent.SignalExternalWorkflowExecutionFailedEventAttributes = &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\t\tCause:                        types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetSignalExternalWorkflowExecutionInitiatedEventAttributes().DecisionTaskCompletedEventID,\n\t\t\tDomain:                       domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: lastEvent.GetSignalExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().WorkflowID,\n\t\t\t\tRunID:      lastEvent.GetSignalExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\texternalWorkflowSignaled := NewHistoryEventVertex(types.EventTypeExternalWorkflowExecutionSignaled.String())\n\texternalWorkflowSignaled.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeExternalWorkflowExecutionSignaled.Ptr()\n\t\thistoryEvent.ExternalWorkflowExecutionSignaledEventAttributes = &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\t\tInitiatedEventID: lastEvent.ID,\n\t\t\tDomain:           domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: lastEvent.GetSignalExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().WorkflowID,\n\t\t\t\tRunID:      lastEvent.GetSignalExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t}\n\t\treturn historyEvent\n\t})\n\texternalWorkflowCancel := NewHistoryEventVertex(types.EventTypeRequestCancelExternalWorkflowExecutionInitiated.String())\n\texternalWorkflowCancel.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr()\n\t\thistoryEvent.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes =\n\t\t\t&types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\tDecisionTaskCompletedEventID: lastEvent.ID,\n\t\t\t\tDomain:                       domain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: externalWorkflowID,\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tChildWorkflowOnly: false,\n\t\t\t}\n\t\treturn historyEvent\n\t})\n\texternalWorkflowCancelFail := NewHistoryEventVertex(types.EventTypeRequestCancelExternalWorkflowExecutionFailed.String())\n\texternalWorkflowCancelFail.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionFailed.Ptr()\n\t\thistoryEvent.RequestCancelExternalWorkflowExecutionFailedEventAttributes = &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\t\tCause:                        types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t\tDecisionTaskCompletedEventID: lastEvent.GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes().DecisionTaskCompletedEventID,\n\t\t\tDomain:                       domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: lastEvent.GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().WorkflowID,\n\t\t\t\tRunID:      lastEvent.GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t\tInitiatedEventID: lastEvent.ID,\n\t\t}\n\t\treturn historyEvent\n\t})\n\texternalWorkflowCanceled := NewHistoryEventVertex(types.EventTypeExternalWorkflowExecutionCancelRequested.String())\n\texternalWorkflowCanceled.SetDataFunc(func(input ...interface{}) interface{} {\n\t\tlastEvent := input[0].(*types.HistoryEvent)\n\t\tlastGeneratedEvent := input[1].(*types.HistoryEvent)\n\t\tEventID := lastGeneratedEvent.ID + 1\n\t\tversion := input[2].(int64)\n\t\thistoryEvent := getDefaultHistoryEvent(EventID, version)\n\t\thistoryEvent.EventType = types.EventTypeExternalWorkflowExecutionCancelRequested.Ptr()\n\t\thistoryEvent.ExternalWorkflowExecutionCancelRequestedEventAttributes = &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\t\tInitiatedEventID: lastEvent.ID,\n\t\t\tDomain:           domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: lastEvent.GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().WorkflowID,\n\t\t\t\tRunID:      lastEvent.GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes().GetWorkflowExecution().RunID,\n\t\t\t},\n\t\t}\n\t\treturn historyEvent\n\t})\n\tdecisionCompleteToExternalWorkflowSignal := NewHistoryEventEdge(decisionComplete, externalWorkflowSignal)\n\tdecisionCompleteToExternalWorkflowCancel := NewHistoryEventEdge(decisionComplete, externalWorkflowCancel)\n\texternalWorkflowSignalToFail := NewHistoryEventEdge(externalWorkflowSignal, externalWorkflowSignalFailed)\n\texternalWorkflowSignalToSignaled := NewHistoryEventEdge(externalWorkflowSignal, externalWorkflowSignaled)\n\texternalWorkflowCancelToFail := NewHistoryEventEdge(externalWorkflowCancel, externalWorkflowCancelFail)\n\texternalWorkflowCancelToCanceled := NewHistoryEventEdge(externalWorkflowCancel, externalWorkflowCanceled)\n\texternalWorkflowSignaledToDecisionSchedule := NewHistoryEventEdge(externalWorkflowSignaled, decisionSchedule)\n\texternalWorkflowSignaledToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\texternalWorkflowSignalFailedToDecisionSchedule := NewHistoryEventEdge(externalWorkflowSignalFailed, decisionSchedule)\n\texternalWorkflowSignalFailedToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\texternalWorkflowCanceledToDecisionSchedule := NewHistoryEventEdge(externalWorkflowCanceled, decisionSchedule)\n\texternalWorkflowCanceledToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\texternalWorkflowCancelFailToDecisionSchedule := NewHistoryEventEdge(externalWorkflowCancelFail, decisionSchedule)\n\texternalWorkflowCancelFailToDecisionSchedule.SetCondition(notPendingDecisionTask)\n\texternalWorkflowModel.AddEdge(decisionCompleteToExternalWorkflowSignal, decisionCompleteToExternalWorkflowCancel,\n\t\texternalWorkflowSignalToFail, externalWorkflowSignalToSignaled, externalWorkflowCancelToFail, externalWorkflowCancelToCanceled,\n\t\texternalWorkflowSignaledToDecisionSchedule, externalWorkflowSignalFailedToDecisionSchedule,\n\t\texternalWorkflowCanceledToDecisionSchedule, externalWorkflowCancelFailToDecisionSchedule)\n\n\t// Config event generator\n\tgenerator.SetBatchGenerationRule(canDoBatch)\n\tgenerator.AddInitialEntryVertex(workflowStart)\n\tgenerator.AddExitVertex(workflowComplete, workflowFail, workflowTerminate, workflowTimedOut, continueAsNew)\n\t// generator.AddRandomEntryVertex(workflowSignal, workflowTerminate, workflowTimedOut)\n\tgenerator.AddModel(decisionModel)\n\tgenerator.AddModel(workflowModel)\n\tgenerator.AddModel(activityModel)\n\tgenerator.AddModel(timerModel)\n\tgenerator.AddModel(childWorkflowModel)\n\tgenerator.AddModel(externalWorkflowModel)\n\treturn generator\n}\n\nfunc getDefaultHistoryEvent(\n\tEventID int64,\n\tversion int64,\n) *types.HistoryEvent {\n\n\tglobalTaskID++\n\treturn &types.HistoryEvent{\n\t\tID:        EventID,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tTaskID:    globalTaskID,\n\t\tVersion:   version,\n\t}\n}\n\nfunc copyConnections(\n\toriginalMap map[string][]Edge,\n) map[string][]Edge {\n\n\tnewMap := make(map[string][]Edge)\n\tfor key, value := range originalMap {\n\t\tnewMap[key] = copyEdges(value)\n\t}\n\treturn newMap\n}\n\nfunc copyExitVertices(\n\toriginalMap map[string]bool,\n) map[string]bool {\n\n\tnewMap := make(map[string]bool)\n\tfor key, value := range originalMap {\n\t\tnewMap[key] = value\n\t}\n\treturn newMap\n}\n\nfunc copyVertex(vertex []Vertex) []Vertex {\n\tnewVertex := make([]Vertex, len(vertex))\n\tfor idx, v := range vertex {\n\t\tnewVertex[idx] = v.DeepCopy()\n\t}\n\treturn newVertex\n}\n\nfunc copyEdges(edges []Edge) []Edge {\n\tnewEdges := make([]Edge, len(edges))\n\tfor idx, e := range edges {\n\t\tnewEdges[idx] = e.DeepCopy()\n\t}\n\treturn newEdges\n}\n"
  },
  {
    "path": "common/testing/testdatagen/fuzzer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdatagen\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n)\n\n// NewFuzzer creates a new fuzzer, notes down the deterministic seed\nfunc New(t *testing.T, generatorFuncs ...interface{}) *fuzz.Fuzzer {\n\treturn NewWithNilChance(t, time.Now().UnixNano(), 0.2, generatorFuncs...)\n}\n\n// NewFuzzer creates a new fuzzer, notes down the deterministic seed\nfunc NewWithNilChance(t *testing.T, seed int64, nilchance float32, generatorFuncs ...interface{}) *fuzz.Fuzzer {\n\tt.Log(\"Fuzz Seed:\", seed)\n\treturn fuzz.NewWithSeed(seed).Funcs(generatorFuncs...).NilChance(float64(nilchance))\n}\n"
  },
  {
    "path": "common/testing/testdatagen/idlfuzzedtestdata/history.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage idlfuzzedtestdata\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\n// NewFuzzerWithIDLTypes creates a new fuzzer, notes down the deterministic seed\n// this particular invocation is preconfigured to be able to handle idl structs\n// correctly without generating completely invalid data (which, while good to test for\n// in the context of an application is too wide a search to be useful)\nfunc NewFuzzerWithIDLTypes(t *testing.T) *fuzz.Fuzzer {\n\treturn testdatagen.New(t,\n\t\t// USE THESE VERY SPARINGLY, ONLY WHEN YOU MUST!\n\t\t//\n\t\t// The goal of providing these generators for specific types should be\n\t\t// to use them as little as possible, as they are fixed test data\n\t\t// which will not evolve with the idl or functions, therefore\n\t\t// the main benefit of fuzzing - evolving tests to handle all new fields in place -\n\t\t// will be defeated.\n\t\t//\n\t\t// for example, for mappers, if you add a new field that needs to be\n\t\t// mapped from protobuf to a native-go type (from the types folder)\n\t\t// and the testdata is fixed here *and not updated*, then the issue\n\t\t// will not be caught by any roundtrip tests.\n\t\tGenHistoryEvent,\n\t)\n}\n\n// GenHistoryEvent is a function to use with gofuzz which\n// skips the majority of difficult to generate values\n// for the sake of simplicity in testing. Use it with the fuzz.Funcs(...) generation function\nfunc GenHistoryEvent(o *types.HistoryEvent, c fuzz.Continue) {\n\t// todo (david.porter) setup an assertion to ensure this list is exhaustive\n\ti := c.Rand.Intn(len(testdata.HistoryEventArray) - 1)\n\to = testdata.HistoryEventArray[i]\n\treturn\n}\n"
  },
  {
    "path": "common/tokenbucket/tb.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tokenbucket\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype (\n\t// TokenBucket is the interface for any implementation of a token bucket rate limiter\n\tTokenBucket interface {\n\t\t// TryConsume attempts to take count tokens from the\n\t\t// bucket. Returns true on success, false\n\t\t// otherwise along with the duration for the next refill\n\t\tTryConsume(count int) (bool, time.Duration)\n\t\t// Consume waits up to timeout duration to take count\n\t\t// tokens from the bucket. Returns true if count\n\t\t// tokens were acquired before timeout, false\n\t\t// otherwise\n\t\tConsume(count int, timeout time.Duration) bool\n\t}\n\n\t// PriorityTokenBucket is the interface for rate limiter with priority\n\tPriorityTokenBucket interface {\n\t\t// GetToken attempts to take count tokens from the\n\t\t// bucket with that priority. Priority 0 is highest.\n\t\t// Returns true on success, false\n\t\t// otherwise along with the duration for the next refill\n\t\tGetToken(priority, count int) (bool, time.Duration)\n\t}\n\n\ttokenBucketImpl struct {\n\t\tsync.Mutex\n\t\ttokens       int\n\t\tfillRate     int           // amount of tokens to add every interval\n\t\tfillInterval time.Duration // time between refills\n\t\t// Because we divide the per-second quota equally\n\t\t// every 100 millis, there could be a remainder when\n\t\t// the desired rate is not a multiple 10 (1second/100Millis)\n\t\t// To overcome this, we keep track of left over remainder\n\t\t// and distribute this evenly during every fillInterval\n\t\toverflowRps            int\n\t\toverflowTokens         int\n\t\tnextRefillTime         time.Time\n\t\tnextOverflowRefillTime time.Time\n\t\tclock                  clock.TimeSource\n\t}\n\n\tdynamicTokenBucketImpl struct {\n\t\ttb         *tokenBucketImpl\n\t\tcurrentRPS int32\n\t\trps        dynamicproperties.IntPropertyFn\n\t}\n\n\tpriorityTokenBucketImpl struct {\n\t\tsync.Mutex\n\t\ttokens         []int\n\t\tfillRate       int\n\t\tnextRefillTime time.Time\n\t\t// Because we divide the per-second quota equally\n\t\t// every 100 millis, there could be a remainder when\n\t\t// the desired rate is not a multiple 10 (1second/100Millis)\n\t\t// To overcome this, we keep track of left over remainder\n\t\t// and distribute this evenly during every fillInterval\n\t\toverflowRps            int\n\t\toverflowTokens         int\n\t\tnextOverflowRefillTime time.Time\n\t\ttimeSource             clock.TimeSource\n\t}\n)\n\nconst (\n\tmillisPerSecond = 1000\n\tbackoffInterval = 10 * time.Millisecond\n\trefillRate      = 100 * time.Millisecond\n)\n\n// New creates and returns a\n// new token bucket rate limiter that\n// replenishes the bucket every 100\n// milliseconds. Thread safe.\n//\n// @param rps\n//\n//\tDesired rate per second\n//\n// Golang.org has an alternative implementation\n// of the rate limiter. On benchmarking, golang's\n// implementation was order of magnitude slower.\n// In addition, it does a lot more than what we\n// need. These are the benchmarks under different\n// scenarios\n//\n// BenchmarkTokenBucketParallel\t50000000\t        40.7 ns/op\n// BenchmarkGolangRateParallel \t10000000\t       150 ns/op\n// BenchmarkTokenBucketParallel-8\t20000000\t       124 ns/op\n// BenchmarkGolangRateParallel-8 \t10000000\t       208 ns/op\n// BenchmarkTokenBucketParallel\t50000000\t        37.8 ns/op\n// BenchmarkGolangRateParallel \t10000000\t       153 ns/op\n// BenchmarkTokenBucketParallel-8\t10000000\t       129 ns/op\n// BenchmarkGolangRateParallel-8 \t10000000\t       208 ns/op\nfunc New(rps int, timeSource clock.TimeSource) TokenBucket {\n\treturn newTokenBucket(rps, timeSource)\n}\n\nfunc newTokenBucket(rps int, timeSource clock.TimeSource) *tokenBucketImpl {\n\ttb := new(tokenBucketImpl)\n\ttb.clock = timeSource\n\ttb.reset(rps)\n\treturn tb\n}\n\nfunc (tb *tokenBucketImpl) TryConsume(count int) (bool, time.Duration) {\n\tnow := tb.clock.Now()\n\ttb.Lock()\n\ttb.refill(now)\n\tnextRefillTime := tb.nextRefillTime.Sub(now)\n\tif tb.tokens < count {\n\t\ttb.Unlock()\n\t\treturn false, nextRefillTime\n\t}\n\ttb.tokens -= count\n\ttb.Unlock()\n\treturn true, nextRefillTime\n}\n\nfunc (tb *tokenBucketImpl) Consume(count int, timeout time.Duration) bool {\n\n\tvar remTime = timeout\n\tvar expiryTime = tb.clock.Now().Add(timeout)\n\n\tfor {\n\n\t\tif ok, _ := tb.TryConsume(count); ok {\n\t\t\treturn true\n\t\t}\n\n\t\tif remTime < backoffInterval {\n\t\t\ttb.clock.Sleep(remTime)\n\t\t} else {\n\t\t\ttb.clock.Sleep(backoffInterval)\n\t\t}\n\n\t\tnow := tb.clock.Now()\n\t\tif now.Compare(expiryTime) >= 0 {\n\t\t\treturn false\n\t\t}\n\n\t\tremTime = expiryTime.Sub(now)\n\t}\n}\n\nfunc (tb *tokenBucketImpl) reset(rps int) {\n\ttb.Lock()\n\ttb.fillInterval = refillRate\n\ttb.fillRate = (rps * 100) / millisPerSecond\n\ttb.overflowRps = rps - (10 * tb.fillRate)\n\ttb.nextOverflowRefillTime = time.Time{}\n\ttb.Unlock()\n}\n\nfunc (tb *tokenBucketImpl) refill(now time.Time) {\n\ttb.refillOverFlow(now)\n\tif tb.isRefillDue(now) {\n\t\ttb.tokens = tb.fillRate\n\t\tif tb.overflowTokens > 0 {\n\t\t\ttb.tokens++\n\t\t\ttb.overflowTokens--\n\t\t}\n\t\ttb.nextRefillTime = now.Add(tb.fillInterval)\n\t}\n}\n\nfunc (tb *tokenBucketImpl) refillOverFlow(now time.Time) {\n\tif tb.overflowRps < 1 {\n\t\treturn\n\t}\n\tif tb.isOverflowRefillDue(now) {\n\t\ttb.overflowTokens = tb.overflowRps\n\t\ttb.nextOverflowRefillTime = now.Add(time.Second)\n\t}\n}\n\nfunc (tb *tokenBucketImpl) isRefillDue(now time.Time) bool {\n\treturn now.Compare(tb.nextRefillTime) >= 0\n}\n\nfunc (tb *tokenBucketImpl) isOverflowRefillDue(now time.Time) bool {\n\treturn now.Compare(tb.nextOverflowRefillTime) >= 0\n}\n\n// NewDynamicTokenBucket creates and returns a token bucket\n// rate limiter that supports dynamic change of RPS. Thread safe.\n// @param rps\n//\n//\tDynamic config function for rate per second\nfunc NewDynamicTokenBucket(rps dynamicproperties.IntPropertyFn, timeSource clock.TimeSource) TokenBucket {\n\tinitialRPS := rps()\n\treturn &dynamicTokenBucketImpl{\n\t\trps:        rps,\n\t\tcurrentRPS: int32(initialRPS),\n\t\ttb:         newTokenBucket(initialRPS, timeSource),\n\t}\n}\n\nfunc (dtb *dynamicTokenBucketImpl) TryConsume(count int) (bool, time.Duration) {\n\tdtb.resetRateIfChanged(dtb.rps())\n\treturn dtb.tb.TryConsume(count)\n}\n\nfunc (dtb *dynamicTokenBucketImpl) Consume(count int, timeout time.Duration) bool {\n\tdtb.resetRateIfChanged(dtb.rps())\n\treturn dtb.tb.Consume(count, timeout)\n}\n\n// resetLimitIfChanged resets the underlying token bucket if the\n// current rps quota is different from the actual rps quota obtained\n// from dynamic config\nfunc (dtb *dynamicTokenBucketImpl) resetRateIfChanged(newRPS int) {\n\tcurrentRPS := atomic.LoadInt32(&dtb.currentRPS)\n\tif int(currentRPS) == newRPS {\n\t\treturn\n\t}\n\tif atomic.CompareAndSwapInt32(&dtb.currentRPS, currentRPS, int32(newRPS)) {\n\t\tdtb.tb.reset(newRPS)\n\t}\n}\n\n// NewPriorityTokenBucket creates and returns a\n// new token bucket rate limiter support priority.\n// There are n buckets for n priorities. It\n// replenishes the top priority bucket every 100\n// milliseconds, unused tokens flows to next bucket.\n// The idea comes from Dual Token Bucket Algorithms.\n// Thread safe.\n//\n// @param numOfPriority\n//\n//\tNumber of priorities\n//\n// @param rps\n//\n//\tDesired rate per second\nfunc NewPriorityTokenBucket(numOfPriority, rps int, timeSource clock.TimeSource) PriorityTokenBucket {\n\ttb := new(priorityTokenBucketImpl)\n\ttb.tokens = make([]int, numOfPriority)\n\ttb.timeSource = timeSource\n\ttb.fillRate = (rps * 100) / millisPerSecond\n\ttb.overflowRps = rps - (10 * tb.fillRate)\n\ttb.refill(tb.timeSource.Now())\n\treturn tb\n}\n\n// NewFullPriorityTokenBucket creates and returns a new priority token bucket with all bucket init with full tokens.\n// With all buckets full, get tokens from low priority buckets won't be missed initially, but may caused bursts.\nfunc NewFullPriorityTokenBucket(numOfPriority, rps int, timeSource clock.TimeSource) PriorityTokenBucket {\n\ttb := new(priorityTokenBucketImpl)\n\ttb.tokens = make([]int, numOfPriority)\n\ttb.timeSource = timeSource\n\ttb.fillRate = (rps * 100) / millisPerSecond\n\ttb.overflowRps = rps - (10 * tb.fillRate)\n\ttb.refill(tb.timeSource.Now())\n\tfor i := 1; i < numOfPriority; i++ {\n\t\ttb.nextRefillTime = time.Time{}\n\t\ttb.refill(tb.timeSource.Now())\n\t}\n\treturn tb\n}\n\nfunc (tb *priorityTokenBucketImpl) GetToken(priority, count int) (bool, time.Duration) {\n\tnow := tb.timeSource.Now()\n\ttb.Lock()\n\ttb.refill(now)\n\tnextRefillTime := tb.nextRefillTime.Sub(now)\n\tif tb.tokens[priority] < count {\n\t\ttb.Unlock()\n\t\treturn false, nextRefillTime\n\t}\n\ttb.tokens[priority] -= count\n\ttb.Unlock()\n\treturn true, nextRefillTime\n}\n\nfunc (tb *priorityTokenBucketImpl) refill(now time.Time) {\n\ttb.refillOverFlow(now)\n\tif tb.isRefillDue(now) {\n\t\tmore := tb.fillRate\n\t\tfor i := 0; i < len(tb.tokens); i++ {\n\t\t\ttb.tokens[i] += more\n\t\t\tif tb.tokens[i] > tb.fillRate {\n\t\t\t\tmore = tb.tokens[i] - tb.fillRate\n\t\t\t\ttb.tokens[i] = tb.fillRate\n\t\t\t} else {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tif tb.overflowTokens > 0 {\n\t\t\ttb.tokens[0]++\n\t\t\ttb.overflowTokens--\n\t\t}\n\t\ttb.nextRefillTime = now.Add(refillRate)\n\t}\n}\n\nfunc (tb *priorityTokenBucketImpl) refillOverFlow(now time.Time) {\n\tif tb.overflowRps < 1 {\n\t\treturn\n\t}\n\tif tb.isOverflowRefillDue(now) {\n\t\ttb.overflowTokens = tb.overflowRps\n\t\ttb.nextOverflowRefillTime = now.Add(time.Second)\n\t}\n}\n\nfunc (tb *priorityTokenBucketImpl) isRefillDue(now time.Time) bool {\n\treturn now.Compare(tb.nextRefillTime) >= 0\n}\n\nfunc (tb *priorityTokenBucketImpl) isOverflowRefillDue(now time.Time) bool {\n\treturn now.Compare(tb.nextOverflowRefillTime) >= 0\n}\n"
  },
  {
    "path": "common/tokenbucket/tb_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tokenbucket\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\nfunc TestRpsEnforced(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := New(99, ts)\n\tfor i := 0; i < 2; i++ {\n\t\ttotal, attempts := testRpsEnforcedHelper(tb, ts, 11, 90, 10)\n\t\tassert.Equal(t, 90, total, \"Token bucket failed to enforce limit\")\n\t\tassert.Equal(t, 9, attempts, \"Token bucket gave out tokens too quickly\")\n\t\tts.Advance(time.Millisecond * 101)\n\t\tok, _ := tb.TryConsume(9)\n\t\tassert.True(t, ok, \"Token bucket failed to enforce limit\")\n\t\tok, _ = tb.TryConsume(1)\n\t\tassert.False(t, ok, \"Token bucket failed to enforce limit\")\n\t\tts.Advance(time.Second)\n\t}\n}\n\nfunc TestLowRpsEnforced(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := New(3, ts)\n\n\ttotal, attempts := testRpsEnforcedHelper(tb, ts, 10, 3, 1)\n\tassert.Equal(t, 3, total, \"Token bucket failed to enforce limit\")\n\tassert.Equal(t, 3, attempts, \"Token bucket gave out tokens too quickly\")\n}\n\nfunc TestDynamicRpsEnforced(t *testing.T) {\n\trpsConfigFn, rpsPtr := getTestRPSConfigFn(99)\n\tts := clock.NewMockedTimeSource()\n\tdtb := NewDynamicTokenBucket(rpsConfigFn, ts)\n\ttotal, attempts := testRpsEnforcedHelper(dtb, ts, 11, 90, 10)\n\tassert.Equal(t, 90, total, \"Token bucket failed to enforce limit\")\n\tassert.Equal(t, 9, attempts, \"Token bucket gave out tokens too quickly\")\n\tts.Advance(time.Second)\n\n\t*rpsPtr = 3\n\ttotal, attempts = testRpsEnforcedHelper(dtb, ts, 10, 3, 1)\n\tassert.Equal(t, 3, total, \"Token bucket failed to enforce limit\")\n\tassert.Equal(t, 3, attempts, \"Token bucket gave out tokens too quickly\")\n}\n\nfunc testRpsEnforcedHelper(tb TokenBucket, ts clock.MockedTimeSource, maxAttempts, tokenNeeded, consumeRate int) (total, attempts int) {\n\ttotal = 0\n\tattempts = 1\n\tfor ; attempts < maxAttempts+1; attempts++ {\n\t\tfor c := 0; c < 2; c++ {\n\t\t\tif ok, _ := tb.TryConsume(consumeRate); ok {\n\t\t\t\ttotal += consumeRate\n\t\t\t}\n\t\t}\n\t\tif total >= tokenNeeded {\n\t\t\tbreak\n\t\t}\n\t\tts.Advance(time.Millisecond * 101)\n\t}\n\treturn\n}\n\nfunc getTestRPSConfigFn(defaultValue int) (dynamicproperties.IntPropertyFn, *int) {\n\trps := defaultValue\n\treturn func(_ ...dynamicproperties.FilterOption) int {\n\t\treturn rps\n\t}, &rps\n}\n\nfunc TestPriorityRpsEnforced(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := NewPriorityTokenBucket(1, 99, ts) // behavior same to tokenBucketImpl\n\n\tfor i := 0; i < 2; i++ {\n\t\ttotal := 0\n\t\tattempts := 1\n\t\tfor ; attempts < 11; attempts++ {\n\t\t\tfor c := 0; c < 2; c++ {\n\t\t\t\tif ok, _ := tb.GetToken(0, 10); ok {\n\t\t\t\t\ttotal += 10\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif total >= 90 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tts.Advance(time.Millisecond * 101)\n\t\t}\n\t\tassert.Equal(t, 90, total, \"Token bucket failed to enforce limit\")\n\t\tassert.Equal(t, 9, attempts, \"Token bucket gave out tokens too quickly\")\n\n\t\tts.Advance(time.Millisecond * 101)\n\t\tok, _ := tb.GetToken(0, 9)\n\t\tassert.True(t, ok, \"Token bucket failed to enforce limit\")\n\t\tok, _ = tb.GetToken(0, 1)\n\t\tassert.False(t, ok, \"Token bucket failed to enforce limit\")\n\t\tts.Advance(time.Second)\n\t}\n}\n\nfunc TestPriorityLowRpsEnforced(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := NewPriorityTokenBucket(1, 3, ts) // behavior same to tokenBucketImpl\n\n\ttotal := 0\n\tattempts := 1\n\tfor ; attempts < 10; attempts++ {\n\t\tfor c := 0; c < 2; c++ {\n\t\t\tif ok, _ := tb.GetToken(0, 1); ok {\n\t\t\t\ttotal++\n\t\t\t}\n\t\t}\n\t\tif total >= 3 {\n\t\t\tbreak\n\t\t}\n\t\tts.Advance(time.Millisecond * 101)\n\t}\n\tassert.Equal(t, 3, total, \"Token bucket failed to enforce limit\")\n\tassert.Equal(t, 3, attempts, \"Token bucket gave out tokens too quickly\")\n}\n\nfunc TestPriorityTokenBucket(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := NewPriorityTokenBucket(2, 100, ts)\n\n\tfor i := 0; i < 2; i++ {\n\t\tok2, _ := tb.GetToken(1, 1)\n\t\tassert.False(t, ok2)\n\t\tok, _ := tb.GetToken(0, 10)\n\t\tassert.True(t, ok)\n\t\tts.Advance(time.Millisecond * 101)\n\t}\n\n\tfor i := 0; i < 2; i++ {\n\t\tok, _ := tb.GetToken(0, 9)\n\t\tassert.True(t, ok) // 1 token remaining in 1st bucket, 0 in 2nd\n\t\tok2, _ := tb.GetToken(1, 1)\n\t\tassert.False(t, ok2)\n\t\tts.Advance(time.Millisecond * 101)\n\t\tok2, _ = tb.GetToken(1, 2)\n\t\tassert.False(t, ok2)\n\t\tok2, _ = tb.GetToken(1, 1)\n\t\tassert.True(t, ok2)\n\t}\n}\n\nfunc TestFullPriorityTokenBucket(t *testing.T) {\n\tts := clock.NewMockedTimeSource()\n\ttb := NewFullPriorityTokenBucket(2, 100, ts)\n\n\tok2, _ := tb.GetToken(1, 10)\n\tassert.True(t, ok2)\n\n\tfor i := 0; i < 2; i++ {\n\t\tok2, _ := tb.GetToken(1, 1)\n\t\tassert.False(t, ok2)\n\t\tok, _ := tb.GetToken(0, 10)\n\t\tassert.True(t, ok)\n\t\tts.Advance(time.Millisecond * 101)\n\t}\n\n\tok2, _ = tb.GetToken(1, 1)\n\tassert.False(t, ok2)\n\tts.Advance(time.Millisecond * 101)\n\tok2, _ = tb.GetToken(1, 5)\n\tassert.True(t, ok2)\n\tts.Advance(time.Millisecond * 101)\n\tok2, _ = tb.GetToken(1, 15)\n\tassert.False(t, ok2)\n\tok2, _ = tb.GetToken(1, 10)\n\tassert.True(t, ok2)\n\tok, _ := tb.GetToken(0, 10)\n\tassert.True(t, ok)\n}\n\nfunc TestTokenBucketConsume(t *testing.T) {\n\tt.Run(\"consume_wait\", func(t *testing.T) {\n\t\t// make sure we don't deadlock inside the test.\n\t\tgo panicOnTimeout(time.Minute)()\n\n\t\tts := clock.NewMockedTimeSource()\n\n\t\t// I provide 10 rps, which means I can consume 10 tokens per second.\n\t\t// tokenBucketImpl will fill 1 token and then refill it every 100 milliseconds.\n\t\ttb := New(10, ts)\n\n\t\t// I consume 1 token, so the next Consume call will block until refill.\n\t\tok, _ := tb.TryConsume(1)\n\t\tassert.True(t, ok)\n\n\t\tconsumeFinished := make(chan struct{})\n\t\tgo func() {\n\t\t\tsuccess := tb.Consume(1, 2*refillRate)\n\t\t\tassert.True(t, success, \"Consume must acquire token successfully\")\n\t\t\tclose(consumeFinished)\n\t\t}()\n\n\t\t// I need to make sure that goroutine with Consume call is blocked on time.Sleep.\n\t\t// BlockUntil awaits for at least 1 goroutine to be blocked on time.Sleep/timer/ticker.\n\t\tts.BlockUntil(1)\n\t\t// Checking that consume is still blocked.\n\t\tselect {\n\t\tcase <-consumeFinished:\n\t\t\tassert.Fail(t, \"Consume returned before refill\")\n\t\tdefault:\n\t\t}\n\t\t// tb has internal retries to acquire tokens which happens at most backoffInterval (10 * time.Millisecond).\n\t\t// I advance time once, so the next iteration is still blocked.\n\t\tts.Advance(backoffInterval + 1)\n\t\t// wait until Consume did one iteration and stopped on time.Sleep\n\t\tts.BlockUntil(1)\n\t\t// Checking that consume is still blocked.\n\t\tselect {\n\t\tcase <-consumeFinished:\n\t\t\tassert.Fail(t, \"Consume returned before refill\")\n\t\tdefault:\n\t\t}\n\t\t// Advance time to the next refill time.\n\t\tts.Advance(refillRate - backoffInterval + 1)\n\t\t// Advance should unblock Consume and it should return afterwards.\n\t\t// I don't provide a timeout, since I have a goroutine to panic on timeout.\n\t\t<-consumeFinished\n\t})\n\tt.Run(\"timeout\", func(t *testing.T) {\n\t\t// make sure we don't deadlock inside the test.\n\t\tgo panicOnTimeout(time.Minute)()\n\n\t\tts := clock.NewMockedTimeSource()\n\t\t// I provide 10 rps, which means I can consume 10 tokens per second.\n\t\t// tokenBucketImpl will fill 1 token and then refill it every 100 milliseconds.\n\t\ttb := New(10, ts)\n\n\t\t// I consume 1 token, so the next Consume call will block until refill.\n\t\tok, _ := tb.TryConsume(1)\n\t\tassert.True(t, ok)\n\n\t\t// This time in consume I provide a small timeout of backoffInterval/2.\n\t\t// So Consume will do only one internal iteration and then return before refill.\n\t\tconsumeFinished := make(chan struct{})\n\t\tgo func() {\n\t\t\tsuccess := tb.Consume(1, backoffInterval/2)\n\t\t\tassert.False(t, success, \"Consume must fail to acquire token\")\n\t\t\tclose(consumeFinished)\n\t\t}()\n\n\t\t// I need to make sure that goroutine with Consume call is blocked on time.Sleep.\n\t\t// BlockUntil awaits for at least 1 goroutine to be blocked on time.Sleep/timer/ticker.\n\t\tts.BlockUntil(1)\n\t\tselect {\n\t\tcase <-consumeFinished:\n\t\t\tassert.Fail(t, \"Consume returned before refill\")\n\t\tdefault:\n\t\t}\n\n\t\t// advance time to wake Consume up and make it return before refill.\n\t\tts.Advance(backoffInterval/2 + 1)\n\t\t<-consumeFinished\n\t})\n}\n\nfunc panicOnTimeout(d time.Duration) (cancel func()) {\n\tstop := make(chan struct{})\n\tgo func() {\n\t\tselect {\n\t\tcase <-time.After(d):\n\t\t\tpanic(\"timeout\")\n\t\tcase <-stop: // noop\n\t\t}\n\t}()\n\treturn func() { close(stop) }\n}\n"
  },
  {
    "path": "common/types/admin.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"sort\"\n\t\"unsafe\"\n)\n\n// AddSearchAttributeRequest is an internal type (TBD...)\ntype AddSearchAttributeRequest struct {\n\tSearchAttribute map[string]IndexedValueType `json:\"searchAttribute,omitempty\"`\n\tSecurityToken   string                      `json:\"securityToken,omitempty\"`\n}\n\n// GetSearchAttribute is an internal getter (TBD...)\nfunc (v *AddSearchAttributeRequest) GetSearchAttribute() (o map[string]IndexedValueType) {\n\tif v != nil && v.SearchAttribute != nil {\n\t\treturn v.SearchAttribute\n\t}\n\treturn\n}\n\n// DescribeClusterResponse is an internal type (TBD...)\ntype DescribeClusterResponse struct {\n\tSupportedClientVersions *SupportedClientVersions    `json:\"supportedClientVersions,omitempty\"`\n\tMembershipInfo          *MembershipInfo             `json:\"membershipInfo,omitempty\"`\n\tPersistenceInfo         map[string]*PersistenceInfo `json:\"persistenceInfo,omitempty\"`\n}\n\n// AdminDescribeWorkflowExecutionRequest is an internal type (TBD...)\ntype AdminDescribeWorkflowExecutionRequest struct {\n\tDomain    string             `json:\"domain,omitempty\"`\n\tExecution *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *AdminDescribeWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// AdminDescribeWorkflowExecutionResponse is an internal type (TBD...)\ntype AdminDescribeWorkflowExecutionResponse struct {\n\tShardID                string `json:\"shardId,omitempty\"`\n\tHistoryAddr            string `json:\"historyAddr,omitempty\"`\n\tMutableStateInCache    string `json:\"mutableStateInCache,omitempty\"`\n\tMutableStateInDatabase string `json:\"mutableStateInDatabase,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *AdminDescribeWorkflowExecutionResponse) GetShardID() (o string) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetMutableStateInDatabase is an internal getter (TBD...)\nfunc (v *AdminDescribeWorkflowExecutionResponse) GetMutableStateInDatabase() (o string) {\n\tif v != nil {\n\t\treturn v.MutableStateInDatabase\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionRawHistoryV2Request is an internal type (TBD...)\ntype GetWorkflowExecutionRawHistoryV2Request struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tExecution         *WorkflowExecution `json:\"execution,omitempty\"`\n\tStartEventID      *int64             `json:\"startEventId,omitempty\"`\n\tStartEventVersion *int64             `json:\"startEventVersion,omitempty\"`\n\tEndEventID        *int64             `json:\"endEventId,omitempty\"`\n\tEndEventVersion   *int64             `json:\"endEventVersion,omitempty\"`\n\tMaximumPageSize   int32              `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken     []byte             `json:\"nextPageToken,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetStartEventID is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetStartEventID() (o int64) {\n\tif v != nil && v.StartEventID != nil {\n\t\treturn *v.StartEventID\n\t}\n\treturn\n}\n\n// GetStartEventVersion is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetStartEventVersion() (o int64) {\n\tif v != nil && v.StartEventVersion != nil {\n\t\treturn *v.StartEventVersion\n\t}\n\treturn\n}\n\n// GetEndEventID is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetEndEventID() (o int64) {\n\tif v != nil && v.EndEventID != nil {\n\t\treturn *v.EndEventID\n\t}\n\treturn\n}\n\n// GetEndEventVersion is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetEndEventVersion() (o int64) {\n\tif v != nil && v.EndEventVersion != nil {\n\t\treturn *v.EndEventVersion\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Request) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionRawHistoryV2Response is an internal type (TBD...)\ntype GetWorkflowExecutionRawHistoryV2Response struct {\n\tNextPageToken  []byte          `json:\"nextPageToken,omitempty\"`\n\tHistoryBatches []*DataBlob     `json:\"historyBatches,omitempty\"`\n\tVersionHistory *VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\n// GetHistoryBatches is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) GetHistoryBatches() (o []*DataBlob) {\n\tif v != nil && v.HistoryBatches != nil {\n\t\treturn v.HistoryBatches\n\t}\n\treturn\n}\n\n// GetVersionHistory is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionRawHistoryV2Response) GetVersionHistory() (o *VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\treturn\n}\n\n// HostInfo is an internal type (TBD...)\ntype HostInfo struct {\n\tIdentity string `json:\"Identity,omitempty\"`\n}\n\n// MembershipInfo is an internal type (TBD...)\ntype MembershipInfo struct {\n\tCurrentHost      *HostInfo   `json:\"currentHost,omitempty\"`\n\tReachableMembers []string    `json:\"reachableMembers,omitempty\"`\n\tRings            []*RingInfo `json:\"rings,omitempty\"`\n}\n\n// PersistenceSetting is used to expose persistence engine settings\ntype PersistenceSetting struct {\n\tKey   string `json:\"key\"`\n\tValue string `json:\"value\"`\n}\n\n// PersistenceFeature is used to expose store specific feature.\n// Feature can be cadence or store specific.\ntype PersistenceFeature struct {\n\tKey     string `json:\"key\"`\n\tEnabled bool   `json:\"enabled\"`\n}\n\n// PersistenceInfo is used to expose store configuration\ntype PersistenceInfo struct {\n\tBackend  string                `json:\"backend\"`\n\tSettings []*PersistenceSetting `json:\"settings,omitempty\"`\n\tFeatures []*PersistenceFeature `json:\"features,omitempty\"`\n}\n\n// ResendReplicationTasksRequest is an internal type (TBD...)\ntype ResendReplicationTasksRequest struct {\n\tDomainID      string `json:\"domainID,omitempty\"`\n\tWorkflowID    string `json:\"workflowID,omitempty\"`\n\tRunID         string `json:\"runID,omitempty\"`\n\tRemoteCluster string `json:\"remoteCluster,omitempty\"`\n\tStartEventID  *int64 `json:\"startEventID,omitempty\"`\n\tStartVersion  *int64 `json:\"startVersion,omitempty\"`\n\tEndEventID    *int64 `json:\"endEventID,omitempty\"`\n\tEndVersion    *int64 `json:\"endVersion,omitempty\"`\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *ResendReplicationTasksRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *ResendReplicationTasksRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetRemoteCluster is an internal getter (TBD...)\nfunc (v *ResendReplicationTasksRequest) GetRemoteCluster() (o string) {\n\tif v != nil {\n\t\treturn v.RemoteCluster\n\t}\n\treturn\n}\n\n// RingInfo is an internal type (TBD...)\ntype RingInfo struct {\n\tRole        string      `json:\"role,omitempty\"`\n\tMemberCount int32       `json:\"memberCount,omitempty\"`\n\tMembers     []*HostInfo `json:\"members,omitempty\"`\n}\n\ntype GetDynamicConfigRequest struct {\n\tConfigName string                 `json:\"configName,omitempty\"`\n\tFilters    []*DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\ntype GetDynamicConfigResponse struct {\n\tValue *DataBlob `json:\"value,omitempty\"`\n}\n\ntype UpdateDynamicConfigRequest struct {\n\tConfigName   string                `json:\"configName,omitempty\"`\n\tConfigValues []*DynamicConfigValue `json:\"configValues,omitempty\"`\n}\n\ntype RestoreDynamicConfigRequest struct {\n\tConfigName string                 `json:\"configName,omitempty\"`\n\tFilters    []*DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\n// AdminDeleteWorkflowRequest is an internal type (TBD...)\ntype AdminDeleteWorkflowRequest struct {\n\tDomain     string             `json:\"domain,omitempty\"`\n\tExecution  *WorkflowExecution `json:\"execution,omitempty\"`\n\tSkipErrors bool               `json:\"skipErrors,omitempty\"`\n}\n\nfunc (v *AdminDeleteWorkflowRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *AdminDeleteWorkflowRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\nfunc (v *AdminDeleteWorkflowRequest) GetSkipErrors() (o bool) {\n\tif v != nil {\n\t\treturn v.SkipErrors\n\t}\n\treturn\n}\n\ntype AdminDeleteWorkflowResponse struct {\n\tHistoryDeleted    bool `json:\"historyDeleted,omitempty\"`\n\tExecutionsDeleted bool `json:\"executionsDeleted,omitempty\"`\n\tVisibilityDeleted bool `json:\"visibilityDeleted,omitempty\"`\n}\n\ntype AdminMaintainWorkflowRequest = AdminDeleteWorkflowRequest\ntype AdminMaintainWorkflowResponse = AdminDeleteWorkflowResponse\n\ntype ListDynamicConfigRequest struct {\n\tConfigName string `json:\"configName,omitempty\"`\n}\n\ntype ListDynamicConfigResponse struct {\n\tEntries []*DynamicConfigEntry `json:\"entries,omitempty\"`\n}\n\ntype IsolationGroupState int\n\nconst (\n\tIsolationGroupStateInvalid IsolationGroupState = iota\n\tIsolationGroupStateHealthy\n\tIsolationGroupStateDrained\n)\n\ntype IsolationGroupPartition struct {\n\tName  string\n\tState IsolationGroupState\n}\n\n// ByteSize returns an approximate size of the object in bytes\nfunc (i IsolationGroupPartition) ByteSize() uint64 {\n\tvar size uint64\n\tsize += uint64(unsafe.Sizeof(i))\n\tsize += uint64(len(i.Name))\n\treturn size\n}\n\n// IsolationGroupConfiguration is an internal representation of a set of\n// isolation-groups as a mapping and may refer to either globally or per-domain (or both) configurations.\n// and their statuses. It's redundantly indexed by IsolationGroup name to simplify lookups.\n//\n// For example: This might be a global configuration persisted\n// in the config store and look like this:\n//\n//\tIsolationGroupConfiguration{\n//\t  \"isolationGroup1234\": {Name: \"isolationGroup1234\", Status: IsolationGroupStatusDrained},\n//\t}\n//\n// Indicating that task processing isn't to occur within this isolationGroup anymore, but all others are ok.\ntype IsolationGroupConfiguration map[string]IsolationGroupPartition\n\n// ToPartitionList Renders the isolation group to the less complicated and confusing simple list of isolation groups\nfunc (i IsolationGroupConfiguration) ToPartitionList() []IsolationGroupPartition {\n\tout := []IsolationGroupPartition{}\n\tfor _, v := range i {\n\t\tout = append(out, v)\n\t}\n\t// ensure determinitism in list ordering for convenience\n\tsort.Slice(out, func(i, j int) bool {\n\t\treturn out[i].Name < out[j].Name\n\t})\n\treturn out\n}\n\nfunc (i IsolationGroupConfiguration) DeepCopy() IsolationGroupConfiguration {\n\tif i == nil {\n\t\treturn nil\n\t}\n\n\tout := IsolationGroupConfiguration{}\n\tfor k, v := range i {\n\t\tout[k] = v\n\t}\n\treturn out\n}\n\n// ByteSize returns an approximate size of the object in bytes\nfunc (i *IsolationGroupConfiguration) ByteSize() uint64 {\n\tif i == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*i))\n\tfor k, v := range *i {\n\t\tsize += uint64(len(k)) + uint64(len(v.Name))\n\t}\n\treturn size\n}\n\n// FromIsolationGroupPartitionList maps a list of isolation to the internal IsolationGroup configuration type\n// whose map keys tend to be used more for set operations\nfunc FromIsolationGroupPartitionList(in []IsolationGroupPartition) IsolationGroupConfiguration {\n\tif len(in) == 0 {\n\t\treturn IsolationGroupConfiguration{}\n\t}\n\tout := IsolationGroupConfiguration{}\n\tfor _, v := range in {\n\t\tout[v.Name] = v\n\t}\n\treturn out\n}\n\ntype GetGlobalIsolationGroupsRequest struct{}\n\ntype GetGlobalIsolationGroupsResponse struct {\n\tIsolationGroups IsolationGroupConfiguration\n}\n\ntype UpdateGlobalIsolationGroupsRequest struct {\n\tIsolationGroups IsolationGroupConfiguration\n}\n\ntype UpdateGlobalIsolationGroupsResponse struct{}\n\ntype GetDomainIsolationGroupsRequest struct {\n\tDomain string\n}\n\ntype GetDomainIsolationGroupsResponse struct {\n\tIsolationGroups IsolationGroupConfiguration\n}\n\ntype UpdateDomainIsolationGroupsRequest struct {\n\tDomain          string\n\tIsolationGroups IsolationGroupConfiguration\n}\n\ntype UpdateDomainIsolationGroupsResponse struct{}\n\ntype GetDomainAsyncWorkflowConfiguratonRequest struct {\n\tDomain string\n}\n\ntype GetDomainAsyncWorkflowConfiguratonResponse struct {\n\tConfiguration *AsyncWorkflowConfiguration\n}\n\ntype AsyncWorkflowConfiguration struct {\n\tEnabled             bool\n\tPredefinedQueueName string\n\tQueueType           string\n\tQueueConfig         *DataBlob\n}\n\nfunc (c AsyncWorkflowConfiguration) DeepCopy() AsyncWorkflowConfiguration {\n\tres := AsyncWorkflowConfiguration{\n\t\tEnabled:             c.Enabled,\n\t\tPredefinedQueueName: c.PredefinedQueueName,\n\t\tQueueType:           c.QueueType,\n\t\tQueueConfig:         c.QueueConfig.DeepCopy(),\n\t}\n\n\treturn res\n}\n\n// ByteSize returns an approximate size of the object in bytes\nfunc (c *AsyncWorkflowConfiguration) ByteSize() uint64 {\n\tif c == nil {\n\t\treturn 0\n\t}\n\tsize := uint64(unsafe.Sizeof(*c))\n\tsize += uint64(len(c.PredefinedQueueName))\n\tsize += uint64(len(c.QueueType))\n\tsize += c.QueueConfig.ByteSize()\n\treturn size\n}\n\ntype UpdateDomainAsyncWorkflowConfiguratonRequest struct {\n\tDomain        string\n\tConfiguration *AsyncWorkflowConfiguration\n}\n\ntype UpdateDomainAsyncWorkflowConfiguratonResponse struct {\n}\n\ntype UpdateTaskListPartitionConfigRequest struct {\n\tDomain          string\n\tTaskList        *TaskList\n\tTaskListType    *TaskListType\n\tPartitionConfig *TaskListPartitionConfig\n}\n\ntype UpdateTaskListPartitionConfigResponse struct{}\n"
  },
  {
    "path": "common/types/admin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestIsolationGroupConfiguration_ToPartitionList(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin          *IsolationGroupConfiguration\n\t\texpectedOut []IsolationGroupPartition\n\t}{\n\t\t\"valid value\": {\n\t\t\tin: &IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t\"zone-3\": {\n\t\t\t\t\tName:  \"zone-3\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOut: []IsolationGroupPartition{\n\t\t\t\t{\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:  \"zone-3\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expectedOut, td.in.ToPartitionList())\n\t\t})\n\t}\n}\n\nfunc TestIsolationGroupConfigurationDeepCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput IsolationGroupConfiguration\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: IsolationGroupConfiguration{},\n\t\t},\n\t\t{\n\t\t\tname: \"single group\",\n\t\t\tinput: IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple groups\",\n\t\t\tinput: IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t\"zone-3\": {\n\t\t\t\t\tName:  \"zone-3\",\n\t\t\t\t\tState: IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\ttc := tc\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot := tc.input.DeepCopy()\n\t\t\tassert.Equal(t, tc.input, got)\n\n\t\t\tif tc.input == nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\ttc.input[\"new\"] = IsolationGroupPartition{Name: \"new\", State: IsolationGroupStateHealthy}\n\t\t\tassert.NotEqual(t, tc.input, got)\n\t\t})\n\t}\n}\n\nfunc TestAsyncWorkflowConfigurationDeepCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput AsyncWorkflowConfiguration\n\t}{\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: AsyncWorkflowConfiguration{},\n\t\t},\n\t\t{\n\t\t\tname: \"predefined queue\",\n\t\t\tinput: AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             true,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"custom queue\",\n\t\t\tinput: AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:   true,\n\t\t\t\tQueueType: \"custom\",\n\t\t\t\tQueueConfig: &DataBlob{\n\t\t\t\t\tEncodingType: EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\tData:         []byte(\"test-async-wf-queue\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ttc := tc\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot := tc.input.DeepCopy()\n\t\t\tassert.Equal(t, tc.input, got)\n\t\t\tif tc.input.QueueConfig != nil {\n\t\t\t\t// assert that queue configs look the same but underlying slice is different\n\t\t\t\tassert.Equal(t, tc.input.QueueConfig, got.QueueConfig)\n\t\t\t\tif tc.input.QueueConfig.Data != nil && identicalByteArray(tc.input.QueueConfig.Data, got.QueueConfig.Data) {\n\t\t\t\t\tt.Error(\"expected DeepCopy to return a new QueueConfig.Data\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAddSearchAttributeRequest_GetSearchAttribute(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *AddSearchAttributeRequest\n\t\twant    map[string]IndexedValueType\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Nil SearchAttribute\",\n\t\t\trequest: &AddSearchAttributeRequest{\n\t\t\t\tSearchAttribute: nil,\n\t\t\t},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"With SearchAttribute\",\n\t\t\trequest: &AddSearchAttributeRequest{\n\t\t\t\tSearchAttribute: map[string]IndexedValueType{\n\t\t\t\t\t\"attr1\": 1,\n\t\t\t\t\t\"attr2\": 2,\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: map[string]IndexedValueType{\n\t\t\t\t\"attr1\": 1,\n\t\t\t\t\"attr2\": 2,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetSearchAttribute()\n\t\t\tassert.Equal(t, tt.want, got, \"GetSearchAttribute() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeWorkflowExecutionRequest_GetDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *AdminDescribeWorkflowExecutionRequest\n\t\twant    string\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"Empty Domain\",\n\t\t\trequest: &AdminDescribeWorkflowExecutionRequest{},\n\t\t\twant:    \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With Domain\",\n\t\t\trequest: &AdminDescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetDomain()\n\t\t\tassert.Equal(t, tt.want, got, \"GetDomain() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeWorkflowExecutionResponse_GetShardID(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tresponse *AdminDescribeWorkflowExecutionResponse\n\t\twant     string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil response\",\n\t\t\tresponse: nil,\n\t\t\twant:     \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty ShardID\",\n\t\t\tresponse: &AdminDescribeWorkflowExecutionResponse{},\n\t\t\twant:     \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With ShardID\",\n\t\t\tresponse: &AdminDescribeWorkflowExecutionResponse{\n\t\t\t\tShardID: \"shard-123\",\n\t\t\t},\n\t\t\twant: \"shard-123\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.response.GetShardID()\n\t\t\tassert.Equal(t, tt.want, got, \"GetShardID() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeWorkflowExecutionResponse_GetMutableStateInDatabase(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tresponse *AdminDescribeWorkflowExecutionResponse\n\t\twant     string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil response\",\n\t\t\tresponse: nil,\n\t\t\twant:     \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty MutableStateInDatabase\",\n\t\t\tresponse: &AdminDescribeWorkflowExecutionResponse{},\n\t\t\twant:     \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With MutableStateInDatabase\",\n\t\t\tresponse: &AdminDescribeWorkflowExecutionResponse{\n\t\t\t\tMutableStateInDatabase: \"state-data\",\n\t\t\t},\n\t\t\twant: \"state-data\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.response.GetMutableStateInDatabase()\n\t\t\tassert.Equal(t, tt.want, got, \"GetMutableStateInDatabase() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    string\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"Empty Domain\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With Domain\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetDomain()\n\t\t\tassert.Equal(t, tt.want, got, \"GetDomain() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetStartEventID(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    int64\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname:    \"Nil StartEventID\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname: \"With StartEventID\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tStartEventID: ptrInt64(100),\n\t\t\t},\n\t\t\twant: 100,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetStartEventID()\n\t\t\tassert.Equal(t, tt.want, got, \"GetStartEventID() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetMaximumPageSize(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    int32\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname:    \"Default MaximumPageSize\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname: \"With MaximumPageSize\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tMaximumPageSize: 100,\n\t\t\t},\n\t\t\twant: 100,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetMaximumPageSize()\n\t\t\tassert.Equal(t, tt.want, got, \"GetMaximumPageSize() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetStartEventVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    int64\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname:    \"Nil StartEventVersion\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname: \"With StartEventVersion\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tStartEventVersion: ptrInt64(123),\n\t\t\t},\n\t\t\twant: 123,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetStartEventVersion()\n\t\t\tassert.Equal(t, tt.want, got, \"GetStartEventVersion() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetEndEventID(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    int64\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname:    \"Nil EndEventID\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname: \"With EndEventID\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tEndEventID: ptrInt64(200),\n\t\t\t},\n\t\t\twant: 200,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetEndEventID()\n\t\t\tassert.Equal(t, tt.want, got, \"GetEndEventID() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Request_GetEndEventVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\trequest *GetWorkflowExecutionRawHistoryV2Request\n\t\twant    int64\n\t}{\n\t\t{\n\t\t\tname:    \"Nil request\",\n\t\t\trequest: nil,\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname:    \"Nil EndEventVersion\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{},\n\t\t\twant:    0,\n\t\t},\n\t\t{\n\t\t\tname: \"With EndEventVersion\",\n\t\t\trequest: &GetWorkflowExecutionRawHistoryV2Request{\n\t\t\t\tEndEventVersion: ptrInt64(456),\n\t\t\t},\n\t\t\twant: 456,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetEndEventVersion()\n\t\t\tassert.Equal(t, tt.want, got, \"GetEndEventVersion() result mismatch\")\n\t\t})\n\t}\n}\nfunc TestGetWorkflowExecutionRawHistoryV2Response_GetHistoryBatches(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tresponse *GetWorkflowExecutionRawHistoryV2Response\n\t\twant     []*DataBlob\n\t}{\n\t\t{\n\t\t\tname:     \"Nil response\",\n\t\t\tresponse: nil,\n\t\t\twant:     nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty HistoryBatches\",\n\t\t\tresponse: &GetWorkflowExecutionRawHistoryV2Response{},\n\t\t\twant:     nil,\n\t\t},\n\t\t{\n\t\t\tname: \"With HistoryBatches\",\n\t\t\tresponse: &GetWorkflowExecutionRawHistoryV2Response{\n\t\t\t\tHistoryBatches: []*DataBlob{\n\t\t\t\t\t{\n\t\t\t\t\t\tEncodingType: EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"data1\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEncodingType: EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"data2\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: []*DataBlob{\n\t\t\t\t{\n\t\t\t\t\tEncodingType: EncodingTypeJSON.Ptr(),\n\t\t\t\t\tData:         []byte(\"data1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEncodingType: EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\tData:         []byte(\"data2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.response.GetHistoryBatches()\n\t\t\tassert.Equal(t, tt.want, got, \"GetHistoryBatches() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionRawHistoryV2Response_GetVersionHistory(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tresponse *GetWorkflowExecutionRawHistoryV2Response\n\t\twant     *VersionHistory\n\t}{\n\t\t{\n\t\t\tname:     \"Nil response\",\n\t\t\tresponse: nil,\n\t\t\twant:     nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty VersionHistory\",\n\t\t\tresponse: &GetWorkflowExecutionRawHistoryV2Response{},\n\t\t\twant:     nil,\n\t\t},\n\t\t{\n\t\t\tname: \"With VersionHistory\",\n\t\t\tresponse: &GetWorkflowExecutionRawHistoryV2Response{\n\t\t\t\tVersionHistory: &VersionHistory{\n\t\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &VersionHistory{\n\t\t\t\tItems: []*VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.response.GetVersionHistory()\n\t\t\tassert.Equal(t, tt.want, got, \"GetVersionHistory() result mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestResendReplicationTasksRequest_GetWorkflowID(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *ResendReplicationTasksRequest\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty WorkflowID\",\n\t\t\trequest:  &ResendReplicationTasksRequest{},\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With WorkflowID\",\n\t\t\trequest: &ResendReplicationTasksRequest{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t},\n\t\t\texpected: \"test-workflow-id\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetWorkflowID()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestResendReplicationTasksRequest_GetRunID(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *ResendReplicationTasksRequest\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty RunID\",\n\t\t\trequest:  &ResendReplicationTasksRequest{},\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With RunID\",\n\t\t\trequest: &ResendReplicationTasksRequest{\n\t\t\t\tRunID: \"test-run-id\",\n\t\t\t},\n\t\t\texpected: \"test-run-id\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetRunID()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestResendReplicationTasksRequest_GetRemoteCluster(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *ResendReplicationTasksRequest\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty RemoteCluster\",\n\t\t\trequest:  &ResendReplicationTasksRequest{},\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With RemoteCluster\",\n\t\t\trequest: &ResendReplicationTasksRequest{\n\t\t\t\tRemoteCluster: \"test-remote-cluster\",\n\t\t\t},\n\t\t\texpected: \"test-remote-cluster\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetRemoteCluster()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestAdminDeleteWorkflowRequest_GetDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *AdminDeleteWorkflowRequest\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname:     \"Empty Domain\",\n\t\t\trequest:  &AdminDeleteWorkflowRequest{},\n\t\t\texpected: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"With Domain\",\n\t\t\trequest: &AdminDeleteWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\texpected: \"test-domain\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetDomain()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestAdminDeleteWorkflowRequest_GetExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *AdminDeleteWorkflowRequest\n\t\texpected *WorkflowExecution\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"Nil Execution\",\n\t\t\trequest:  &AdminDeleteWorkflowRequest{},\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"With Execution\",\n\t\t\trequest: &AdminDeleteWorkflowRequest{\n\t\t\t\tExecution: &WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetExecution()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestAdminDeleteWorkflowRequest_GetSkipErrors(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\trequest  *AdminDeleteWorkflowRequest\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname:     \"Nil request\",\n\t\t\trequest:  nil,\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"SkipErrors is false\",\n\t\t\trequest:  &AdminDeleteWorkflowRequest{SkipErrors: false},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"SkipErrors is true\",\n\t\t\trequest:  &AdminDeleteWorkflowRequest{SkipErrors: true},\n\t\t\texpected: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.request.GetSkipErrors()\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n\nfunc TestFromIsolationGroupPartitionList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tin   []IsolationGroupPartition\n\t\twant IsolationGroupConfiguration\n\t}{\n\t\t{\n\t\t\tname: \"empty list\",\n\t\t\tin:   []IsolationGroupPartition{},\n\t\t\twant: IsolationGroupConfiguration{},\n\t\t},\n\t\t{\n\t\t\tname: \"single group\",\n\t\t\tin: []IsolationGroupPartition{\n\t\t\t\t{Name: \"zone-1\", State: IsolationGroupStateHealthy},\n\t\t\t},\n\t\t\twant: IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: IsolationGroupStateHealthy},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple groups\",\n\t\t\tin: []IsolationGroupPartition{\n\t\t\t\t{Name: \"zone-1\", State: IsolationGroupStateHealthy},\n\t\t\t\t{Name: \"zone-2\", State: IsolationGroupStateDrained},\n\t\t\t},\n\t\t\twant: IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: IsolationGroupStateHealthy},\n\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: IsolationGroupStateDrained},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := FromIsolationGroupPartitionList(tt.in)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc ptrInt64(i int64) *int64 {\n\treturn &i\n}\n"
  },
  {
    "path": "common/types/caller.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"context\"\n)\n\nconst (\n\tCallerTypeHeaderName = \"cadence-caller-type\" // need to define it here due to circular dependency with common\n)\n\ntype CallerType string\n\nconst (\n\tCallerTypeUnknown  CallerType = \"unknown\"\n\tCallerTypeCLI      CallerType = \"cli\"\n\tCallerTypeUI       CallerType = \"ui\"\n\tCallerTypeSDK      CallerType = \"sdk\"\n\tCallerTypeInternal CallerType = \"internal\"\n)\n\n// CallerInfo captures request source information for observability and resource management.\n//\n// Intent:\n//   - Track the source/origin/actor of API requests (CLI, UI, SDK, internal service calls, etc.)\n//   - Enable client-specific behavior and resource allocation decisions\n//   - Support future extensibility for additional caller metadata (e.g., identity, version)\n//\n// Consumers:\n//   - Logging and audit systems for request attribution\n//   - Metrics and monitoring for client-specific observability\n//   - Rate limiting and resource management based on caller information\n//\n// Lifecycle:\n//   - Should be set early in request processing, typically after authentication\n//   - Expected for external API calls (CLI, UI, SDK)\n//   - May be absent for internal service-to-service calls or unauthenticated endpoints\n//   - Set by authentication/authorization middleware or API gateway components\ntype CallerInfo struct {\n\tcallerType CallerType\n}\n\n// NewCallerInfo creates a new CallerInfo\nfunc NewCallerInfo(callerType CallerType) CallerInfo {\n\treturn CallerInfo{callerType: callerType}\n}\n\n// GetCallerType returns the CallerType\nfunc (c CallerInfo) GetCallerType() CallerType {\n\treturn c.callerType\n}\n\ntype callerInfoContextKey string\n\nconst callerInfoKey = callerInfoContextKey(\"caller-info\")\n\nfunc (c CallerType) String() string {\n\treturn string(c)\n}\n\n// ParseCallerType converts a string to CallerType\n// Returns CallerTypeUnknown if s is empty\nfunc ParseCallerType(s string) CallerType {\n\tif s == \"\" {\n\t\treturn CallerTypeUnknown\n\t}\n\treturn CallerType(s)\n}\n\n// ContextWithCallerInfo adds CallerInfo to context\nfunc ContextWithCallerInfo(ctx context.Context, callerInfo CallerInfo) context.Context {\n\treturn context.WithValue(ctx, callerInfoKey, callerInfo)\n}\n\n// GetCallerInfoFromContext retrieves CallerInfo from context\n// Returns CallerInfo with CallerTypeUnknown if not set in context\nfunc GetCallerInfoFromContext(ctx context.Context) CallerInfo {\n\tif ctx == nil {\n\t\treturn NewCallerInfo(CallerTypeUnknown)\n\t}\n\tif callerInfo, ok := ctx.Value(callerInfoKey).(CallerInfo); ok {\n\t\treturn callerInfo\n\t}\n\treturn NewCallerInfo(CallerTypeUnknown)\n}\n\n// NewCallerInfoFromTransportHeaders extracts CallerInfo from transport headers\n// This is used by middleware to extract caller information from incoming requests\nfunc NewCallerInfoFromTransportHeaders(headers interface{ Get(string) (string, bool) }) CallerInfo {\n\tcallerTypeStr, _ := headers.Get(CallerTypeHeaderName)\n\n\t// Future: add more header extractions here\n\t// version, _ := headers.Get(\"cadence-client-version\")\n\t// identity, _ := headers.Get(\"cadence-client-identity\")\n\n\treturn NewCallerInfo(ParseCallerType(callerTypeStr))\n}\n\n// GetContextWithCallerInfoFromHeaders extracts CallerInfo from transport headers and adds it to the context\nfunc GetContextWithCallerInfoFromHeaders(ctx context.Context, headers interface{ Get(string) (string, bool) }) context.Context {\n\treturn ContextWithCallerInfo(ctx, NewCallerInfoFromTransportHeaders(headers))\n}\n"
  },
  {
    "path": "common/types/caller_test.go",
    "content": "// Copyright (c) 2025 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestCallerType_String(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tcallerType CallerType\n\t\twant       string\n\t}{\n\t\t{\"CLI\", CallerTypeCLI, \"cli\"},\n\t\t{\"UI\", CallerTypeUI, \"ui\"},\n\t\t{\"SDK\", CallerTypeSDK, \"sdk\"},\n\t\t{\"Internal\", CallerTypeInternal, \"internal\"},\n\t\t{\"Unknown\", CallerTypeUnknown, \"unknown\"},\n\t\t{\"Empty string\", CallerType(\"\"), \"\"},\n\t\t{\"Invalid\", CallerType(\"invalid\"), \"invalid\"},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tt.want, tt.callerType.String())\n\t\t})\n\t}\n}\n\nfunc TestParseCallerType(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput string\n\t\twant  CallerType\n\t}{\n\t\t{\"cli\", \"cli\", CallerTypeCLI},\n\t\t{\"ui\", \"ui\", CallerTypeUI},\n\t\t{\"sdk\", \"sdk\", CallerTypeSDK},\n\t\t{\"internal\", \"internal\", CallerTypeInternal},\n\t\t{\"unknown\", \"unknown\", CallerTypeUnknown},\n\t\t{\"empty\", \"\", CallerTypeUnknown},\n\t\t{\"custom value\", \"my-custom-tool\", CallerType(\"my-custom-tool\")},\n\t\t{\"uppercase\", \"CLI\", CallerType(\"CLI\")},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tt.want, ParseCallerType(tt.input))\n\t\t})\n\t}\n}\n\nfunc TestCallerTypeRoundTrip(t *testing.T) {\n\ttests := []CallerType{\n\t\tCallerTypeCLI,\n\t\tCallerTypeUI,\n\t\tCallerTypeSDK,\n\t\tCallerTypeInternal,\n\t\tCallerTypeUnknown,\n\t}\n\n\tfor _, ct := range tests {\n\t\tt.Run(ct.String(), func(t *testing.T) {\n\t\t\tstr := ct.String()\n\t\t\tparsed := ParseCallerType(str)\n\t\t\tassert.Equal(t, ct, parsed)\n\t\t})\n\t}\n}\n\nfunc TestNewCallerInfo(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tcallerType CallerType\n\t\twant       CallerType\n\t}{\n\t\t{\n\t\t\tname:       \"CLI\",\n\t\t\tcallerType: CallerTypeCLI,\n\t\t\twant:       CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname:       \"SDK\",\n\t\t\tcallerType: CallerTypeSDK,\n\t\t\twant:       CallerTypeSDK,\n\t\t},\n\t\t{\n\t\t\tname:       \"Unknown\",\n\t\t\tcallerType: CallerTypeUnknown,\n\t\t\twant:       CallerTypeUnknown,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tinfo := NewCallerInfo(tt.callerType)\n\t\t\tassert.Equal(t, tt.want, info.GetCallerType())\n\t\t})\n\t}\n}\n\nfunc TestCallerInfo_GetCallerType(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tinfo CallerInfo\n\t\twant CallerType\n\t}{\n\t\t{\n\t\t\tname: \"CLI CallerInfo\",\n\t\t\tinfo: NewCallerInfo(CallerTypeCLI),\n\t\t\twant: CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname: \"SDK CallerInfo\",\n\t\t\tinfo: NewCallerInfo(CallerTypeSDK),\n\t\t\twant: CallerTypeSDK,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tt.want, tt.info.GetCallerType())\n\t\t})\n\t}\n}\n\nfunc TestContextWithCallerInfo(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tcallerType CallerType\n\t\twant       CallerType\n\t}{\n\t\t{\n\t\t\tname:       \"CLI caller info\",\n\t\t\tcallerType: CallerTypeCLI,\n\t\t\twant:       CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname:       \"SDK caller info\",\n\t\t\tcallerType: CallerTypeSDK,\n\t\t\twant:       CallerTypeSDK,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctx := context.Background()\n\t\t\tctx = ContextWithCallerInfo(ctx, NewCallerInfo(tt.callerType))\n\n\t\t\tgot := GetCallerInfoFromContext(ctx)\n\t\t\tassert.Equal(t, tt.want, got.GetCallerType())\n\t\t})\n\t}\n}\n\nfunc TestGetCallerInfoFromContext(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tctx        context.Context\n\t\twantCaller CallerType\n\t}{\n\t\t{\n\t\t\tname:       \"nil context\",\n\t\t\tctx:        nil,\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t\t{\n\t\t\tname:       \"context without caller info\",\n\t\t\tctx:        context.Background(),\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t\t{\n\t\t\tname:       \"context with CLI caller info\",\n\t\t\tctx:        ContextWithCallerInfo(context.Background(), NewCallerInfo(CallerTypeCLI)),\n\t\t\twantCaller: CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname:       \"context with SDK caller info\",\n\t\t\tctx:        ContextWithCallerInfo(context.Background(), NewCallerInfo(CallerTypeSDK)),\n\t\t\twantCaller: CallerTypeSDK,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := GetCallerInfoFromContext(tt.ctx)\n\t\t\tassert.Equal(t, tt.wantCaller, got.GetCallerType())\n\t\t})\n\t}\n}\n\ntype mockHeaders map[string]string\n\nfunc (m mockHeaders) Get(key string) (string, bool) {\n\tval, ok := m[key]\n\treturn val, ok\n}\n\nfunc TestNewCallerInfoFromTransportHeaders(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\theaders    mockHeaders\n\t\twantCaller CallerType\n\t}{\n\t\t{\n\t\t\tname:       \"CLI caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"cli\"},\n\t\t\twantCaller: CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname:       \"UI caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"ui\"},\n\t\t\twantCaller: CallerTypeUI,\n\t\t},\n\t\t{\n\t\t\tname:       \"SDK caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"sdk\"},\n\t\t\twantCaller: CallerTypeSDK,\n\t\t},\n\t\t{\n\t\t\tname:       \"Internal caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"internal\"},\n\t\t\twantCaller: CallerTypeInternal,\n\t\t},\n\t\t{\n\t\t\tname:       \"empty caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"\"},\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t\t{\n\t\t\tname:       \"missing caller-type header\",\n\t\t\theaders:    mockHeaders{},\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := NewCallerInfoFromTransportHeaders(tt.headers)\n\t\t\tassert.Equal(t, tt.wantCaller, got.GetCallerType())\n\t\t})\n\t}\n}\n\nfunc TestGetContextWithCallerInfoFromHeaders(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\theaders    mockHeaders\n\t\twantCaller CallerType\n\t}{\n\t\t{\n\t\t\tname:       \"CLI caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"cli\"},\n\t\t\twantCaller: CallerTypeCLI,\n\t\t},\n\t\t{\n\t\t\tname:       \"SDK caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"sdk\"},\n\t\t\twantCaller: CallerTypeSDK,\n\t\t},\n\t\t{\n\t\t\tname:       \"empty caller-type header\",\n\t\t\theaders:    mockHeaders{CallerTypeHeaderName: \"\"},\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t\t{\n\t\t\tname:       \"missing caller-type header\",\n\t\t\theaders:    mockHeaders{},\n\t\t\twantCaller: CallerTypeUnknown,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresultCtx := GetContextWithCallerInfoFromHeaders(context.Background(), tt.headers)\n\n\t\t\tcallerInfo := GetCallerInfoFromContext(resultCtx)\n\t\t\tassert.Equal(t, tt.wantCaller, callerInfo.GetCallerType())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/configStore.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\ntype DynamicConfigBlob struct {\n\tSchemaVersion int64                 `json:\"schemaVersion,omitempty\"`\n\tEntries       []*DynamicConfigEntry `json:\"entries,omitempty\"`\n}\n\ntype DynamicConfigEntry struct {\n\tName   string                `json:\"name,omitempty\"`\n\tValues []*DynamicConfigValue `json:\"values,omitempty\"`\n}\n\ntype DynamicConfigValue struct {\n\tValue   *DataBlob              `json:\"value,omitempty\"`\n\tFilters []*DynamicConfigFilter `json:\"filters,omitempty\"`\n}\n\ntype DynamicConfigFilter struct {\n\tName  string    `json:\"name,omitempty\"`\n\tValue *DataBlob `json:\"value,omitempty\"`\n}\n\nfunc (dcf *DynamicConfigFilter) Copy() *DynamicConfigFilter {\n\tif dcf == nil {\n\t\treturn nil\n\t}\n\treturn &DynamicConfigFilter{\n\t\tName:  dcf.Name,\n\t\tValue: dcf.Value.DeepCopy(),\n\t}\n}\n\nfunc (dcv *DynamicConfigValue) Copy() *DynamicConfigValue {\n\tif dcv == nil {\n\t\treturn nil\n\t}\n\n\tvar newFilters []*DynamicConfigFilter\n\tif dcv.Filters != nil {\n\t\tnewFilters = make([]*DynamicConfigFilter, 0, len(dcv.Filters))\n\t\tfor _, filter := range dcv.Filters {\n\t\t\tnewFilters = append(newFilters, filter.Copy())\n\t\t}\n\t}\n\n\treturn &DynamicConfigValue{\n\t\tValue:   dcv.Value.DeepCopy(),\n\t\tFilters: newFilters,\n\t}\n}\n\nfunc (dce *DynamicConfigEntry) Copy() *DynamicConfigEntry {\n\tif dce == nil {\n\t\treturn nil\n\t}\n\n\tvar newValues []*DynamicConfigValue\n\tif dce.Values != nil {\n\t\tnewValues = make([]*DynamicConfigValue, 0, len(dce.Values))\n\t\tfor _, value := range dce.Values {\n\t\t\tnewValues = append(newValues, value.Copy())\n\t\t}\n\t}\n\n\treturn &DynamicConfigEntry{\n\t\tName:   dce.Name,\n\t\tValues: newValues,\n\t}\n}\n"
  },
  {
    "path": "common/types/configStore_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestDynamicConfigEntryCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput *DynamicConfigEntry\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: &DynamicConfigEntry{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil values\",\n\t\t\tinput: &DynamicConfigEntry{\n\t\t\t\tName:   \"test-2\",\n\t\t\t\tValues: nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil values\",\n\t\t\tinput: &DynamicConfigEntry{\n\t\t\t\tName: \"test-2\",\n\t\t\t\tValues: []*DynamicConfigValue{\n\t\t\t\t\t&DynamicConfigValue{\n\t\t\t\t\t\tValue:   &DataBlob{Data: []byte(\"data\")},\n\t\t\t\t\t\tFilters: nil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tvalCopy := tc.input.Copy()\n\t\t\tassert.Equal(t, tc.input, valCopy)\n\t\t\t// check if modifying the copy does not modify the original\n\t\t\tif valCopy != nil && valCopy.Values != nil {\n\t\t\t\tvalCopy.Values[0].Value.Data = []byte(\"modified\")\n\t\t\t\tassert.NotEqual(t, tc.input, valCopy)\n\t\t\t}\n\t\t\tassert.Equal(t, tc.input, tc.input.Copy())\n\t\t})\n\t}\n}\n\nfunc TestDynamicConfigFilterCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput *DynamicConfigFilter\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: &DynamicConfigFilter{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil value\",\n\t\t\tinput: &DynamicConfigFilter{\n\t\t\t\tName:  \"test-2\",\n\t\t\t\tValue: nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil values\",\n\t\t\tinput: &DynamicConfigFilter{\n\t\t\t\tName:  \"test-2\",\n\t\t\t\tValue: &DataBlob{Data: []byte(\"data\")},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tvalCopy := tc.input.Copy()\n\t\t\tassert.Equal(t, tc.input, valCopy)\n\t\t\t// check if modifying the copy does not modify the original\n\t\t\tif valCopy != nil {\n\t\t\t\tvalCopy.Name = \"modified\"\n\t\t\t\tassert.NotEqual(t, tc.input, valCopy)\n\t\t\t}\n\t\t})\n\t}\n}\nfunc TestDynamicConfigValueCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput *DynamicConfigValue\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: &DynamicConfigValue{},\n\t\t},\n\n\t\t{\n\t\t\tname: \"non-nil values\",\n\t\t\tinput: &DynamicConfigValue{\n\t\t\t\tValue: &DataBlob{Data: []byte(\"data\")},\n\t\t\t\tFilters: []*DynamicConfigFilter{\n\t\t\t\t\t&DynamicConfigFilter{\n\t\t\t\t\t\tName:  \"filter-1\",\n\t\t\t\t\t\tValue: &DataBlob{Data: []byte(\"data\")},\n\t\t\t\t\t}},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// check if modifying the copy does not modify the original\n\t\t\tvalCopy := tc.input.Copy()\n\t\t\tassert.Equal(t, tc.input, valCopy)\n\t\t\tif valCopy != nil && valCopy.Value != nil {\n\t\t\t\tvalCopy.Value.Data = []byte(\"modified\")\n\t\t\t\tassert.NotEqual(t, tc.input, valCopy)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/enums.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\n// EventTypeValues returns all recognized values of EventType.\nfunc EventTypeValues() []EventType {\n\treturn []EventType{\n\t\tEventTypeWorkflowExecutionStarted,\n\t\tEventTypeWorkflowExecutionCompleted,\n\t\tEventTypeWorkflowExecutionFailed,\n\t\tEventTypeWorkflowExecutionTimedOut,\n\t\tEventTypeDecisionTaskScheduled,\n\t\tEventTypeDecisionTaskStarted,\n\t\tEventTypeDecisionTaskCompleted,\n\t\tEventTypeDecisionTaskTimedOut,\n\t\tEventTypeDecisionTaskFailed,\n\t\tEventTypeActivityTaskScheduled,\n\t\tEventTypeActivityTaskStarted,\n\t\tEventTypeActivityTaskCompleted,\n\t\tEventTypeActivityTaskFailed,\n\t\tEventTypeActivityTaskTimedOut,\n\t\tEventTypeActivityTaskCancelRequested,\n\t\tEventTypeRequestCancelActivityTaskFailed,\n\t\tEventTypeActivityTaskCanceled,\n\t\tEventTypeTimerStarted,\n\t\tEventTypeTimerFired,\n\t\tEventTypeCancelTimerFailed,\n\t\tEventTypeTimerCanceled,\n\t\tEventTypeWorkflowExecutionCancelRequested,\n\t\tEventTypeWorkflowExecutionCanceled,\n\t\tEventTypeRequestCancelExternalWorkflowExecutionInitiated,\n\t\tEventTypeRequestCancelExternalWorkflowExecutionFailed,\n\t\tEventTypeExternalWorkflowExecutionCancelRequested,\n\t\tEventTypeMarkerRecorded,\n\t\tEventTypeWorkflowExecutionSignaled,\n\t\tEventTypeWorkflowExecutionTerminated,\n\t\tEventTypeWorkflowExecutionContinuedAsNew,\n\t\tEventTypeStartChildWorkflowExecutionInitiated,\n\t\tEventTypeStartChildWorkflowExecutionFailed,\n\t\tEventTypeChildWorkflowExecutionStarted,\n\t\tEventTypeChildWorkflowExecutionCompleted,\n\t\tEventTypeChildWorkflowExecutionFailed,\n\t\tEventTypeChildWorkflowExecutionCanceled,\n\t\tEventTypeChildWorkflowExecutionTimedOut,\n\t\tEventTypeChildWorkflowExecutionTerminated,\n\t\tEventTypeSignalExternalWorkflowExecutionInitiated,\n\t\tEventTypeSignalExternalWorkflowExecutionFailed,\n\t\tEventTypeExternalWorkflowExecutionSignaled,\n\t\tEventTypeUpsertWorkflowSearchAttributes,\n\t}\n}\n\n// DecisionTypeValues returns all recognized values of DecisionType.\nfunc DecisionTypeValues() []DecisionType {\n\treturn []DecisionType{\n\t\tDecisionTypeScheduleActivityTask,\n\t\tDecisionTypeRequestCancelActivityTask,\n\t\tDecisionTypeStartTimer,\n\t\tDecisionTypeCompleteWorkflowExecution,\n\t\tDecisionTypeFailWorkflowExecution,\n\t\tDecisionTypeCancelTimer,\n\t\tDecisionTypeCancelWorkflowExecution,\n\t\tDecisionTypeRequestCancelExternalWorkflowExecution,\n\t\tDecisionTypeRecordMarker,\n\t\tDecisionTypeContinueAsNewWorkflowExecution,\n\t\tDecisionTypeStartChildWorkflowExecution,\n\t\tDecisionTypeSignalExternalWorkflowExecution,\n\t\tDecisionTypeUpsertWorkflowSearchAttributes,\n\t}\n}\n"
  },
  {
    "path": "common/types/enums_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc Test_EventTypeValues(t *testing.T) {\n\tresult := EventTypeValues()\n\trequire.Equal(t, 42, len(result))\n}\n\nfunc Test_DecisionTypeValues(t *testing.T) {\n\tresult := DecisionTypeValues()\n\trequire.Equal(t, 13, len(result))\n}\n"
  },
  {
    "path": "common/types/errors.go",
    "content": "// Copyright (c) 2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"go.uber.org/zap/zapcore\"\n)\n\n// EventAlreadyStartedError is an internal type (TBD...)\ntype EventAlreadyStartedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\nfunc (err AccessDeniedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err BadRequestError) Error() string {\n\treturn err.Message\n}\n\nfunc (err CancellationAlreadyRequestedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err ClientVersionNotSupportedError) Error() string {\n\treturn \"client version not supported\"\n}\n\nfunc (err ClientVersionNotSupportedError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"feature-version\", err.FeatureVersion)\n\tenc.AddString(\"client-implementation\", err.ClientImpl)\n\tenc.AddString(\"supported-versions\", err.SupportedVersions)\n\treturn nil\n}\n\nfunc (err FeatureNotEnabledError) Error() string {\n\treturn \"feature not enabled\"\n}\n\nfunc (err FeatureNotEnabledError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"feature-flag\", err.FeatureFlag)\n\treturn nil\n}\n\nfunc (err CurrentBranchChangedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err CurrentBranchChangedError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"wf-branch-token\", string(err.CurrentBranchToken))\n\treturn nil\n}\n\nfunc (err DomainAlreadyExistsError) Error() string {\n\treturn err.Message\n}\n\nfunc (err DomainNotActiveError) Error() string {\n\treturn err.Message\n}\n\nfunc (err DomainNotActiveError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"domain-name\", err.DomainName)\n\tenc.AddString(\"current-cluster\", err.CurrentCluster)\n\tenc.AddString(\"active-cluster\", err.ActiveCluster)\n\treturn nil\n}\n\nfunc (err EntityNotExistsError) Error() string {\n\treturn err.Message\n}\n\nfunc (err WorkflowExecutionAlreadyCompletedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err InternalDataInconsistencyError) Error() string {\n\treturn err.Message\n}\n\nfunc (err InternalServiceError) Error() string {\n\treturn err.Message\n}\n\nfunc (err LimitExceededError) Error() string {\n\treturn err.Message\n}\n\nfunc (err QueryFailedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err RemoteSyncMatchedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err RetryTaskV2Error) Error() string {\n\treturn err.Message\n}\n\nfunc (err RetryTaskV2Error) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"domain-id\", err.DomainID)\n\tenc.AddString(\"workflow-id\", err.WorkflowID)\n\tenc.AddString(\"run-id\", err.RunID)\n\tif err.StartEventID != nil {\n\t\tenc.AddInt64(\"start-event-id\", *err.StartEventID)\n\t}\n\tif err.StartEventVersion != nil {\n\t\tenc.AddInt64(\"start-event-version\", *err.StartEventVersion)\n\t}\n\tif err.EndEventID != nil {\n\t\tenc.AddInt64(\"end-event-id\", *err.EndEventID)\n\t}\n\tif err.EndEventVersion != nil {\n\t\tenc.AddInt64(\"end-event-version\", *err.EndEventVersion)\n\t}\n\treturn nil\n}\n\nfunc (err ServiceBusyError) Error() string {\n\treturn err.Message\n}\n\nfunc (err WorkflowExecutionAlreadyStartedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err WorkflowExecutionAlreadyStartedError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"start-request-id\", err.StartRequestID)\n\tenc.AddString(\"run-id\", err.RunID)\n\treturn nil\n}\n\nfunc (err ShardOwnershipLostError) Error() string {\n\treturn err.Message\n}\n\nfunc (err ShardOwnershipLostError) MarshalLogObject(enc zapcore.ObjectEncoder) error {\n\tenc.AddString(\"shard-owner\", err.Owner)\n\treturn nil\n}\n\nfunc (err EventAlreadyStartedError) Error() string {\n\treturn err.Message\n}\n\nfunc (err StickyWorkerUnavailableError) Error() string {\n\treturn err.Message\n}\n\nfunc (err ReadOnlyPartitionError) Error() string {\n\treturn err.Message\n}\n"
  },
  {
    "path": "common/types/errors_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/zap/zapcore\"\n)\n\nfunc Test_Error(t *testing.T) {\n\terrMessage := \"test\"\n\ttests := []struct {\n\t\tname string\n\t\terr  error\n\t}{\n\t\t{\n\t\t\terr: AccessDeniedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: BadRequestError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: CancellationAlreadyRequestedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: DomainAlreadyExistsError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: EntityNotExistsError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: InternalDataInconsistencyError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: WorkflowExecutionAlreadyCompletedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: LimitExceededError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: QueryFailedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: RemoteSyncMatchedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: ServiceBusyError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: EventAlreadyStartedError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: StickyWorkerUnavailableError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: ReadOnlyPartitionError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\terr: InternalServiceError{\n\t\t\t\tMessage: errMessage,\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(reflect.TypeOf(tt.err).String(), func(t *testing.T) {\n\t\t\trequire.Equal(t, errMessage, tt.err.Error())\n\t\t})\n\t}\n}\n\nfunc Test_ClientVersionNotSupportedError(t *testing.T) {\n\terr := ClientVersionNotSupportedError{\n\t\tFeatureVersion:    \"1.0\",\n\t\tClientImpl:        \"1.0\",\n\t\tSupportedVersions: \"1.2\",\n\t}\n\trequire.Equal(t, \"client version not supported\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_FeatureNotEnabledError(t *testing.T) {\n\terr := FeatureNotEnabledError{FeatureFlag: \"test\"}\n\trequire.Equal(t, \"feature not enabled\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_CurrentBranchChangedError(t *testing.T) {\n\terr := CurrentBranchChangedError{Message: \"test\", CurrentBranchToken: []byte{}}\n\trequire.Equal(t, \"test\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_DomainNotActiveError(t *testing.T) {\n\terr := DomainNotActiveError{Message: \"test\", DomainName: \"test-domain\"}\n\trequire.Equal(t, \"test\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_RetryTaskV2Error(t *testing.T) {\n\ttestID := int64(1)\n\ttestVersion := int64(1.0)\n\terr := RetryTaskV2Error{\n\t\tMessage:           \"test\",\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"wid\",\n\t\tRunID:             \"rid\",\n\t\tStartEventID:      &testID,\n\t\tStartEventVersion: &testVersion,\n\t\tEndEventID:        &testID,\n\t\tEndEventVersion:   &testVersion,\n\t}\n\trequire.Equal(t, \"test\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_WorkflowExecutionAlreadyStartedError(t *testing.T) {\n\terr := WorkflowExecutionAlreadyStartedError{Message: \"test\"}\n\trequire.Equal(t, \"test\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n\nfunc Test_ShardOwnershipLostError(t *testing.T) {\n\terr := ShardOwnershipLostError{Message: \"test\"}\n\trequire.Equal(t, \"test\", err.Error())\n\trequire.NoError(t, err.MarshalLogObject(zapcore.NewMapObjectEncoder()))\n}\n"
  },
  {
    "path": "common/types/health.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\n// HealthStatus is an internal type (TBD...)\ntype HealthStatus struct {\n\tOk  bool   `json:\"ok,required\"`\n\tMsg string `json:\"msg,omitempty\"`\n}\n"
  },
  {
    "path": "common/types/history.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\n// DescribeMutableStateRequest is an internal type (TBD...)\ntype DescribeMutableStateRequest struct {\n\tDomainUUID string             `json:\"domainUUID,omitempty\"`\n\tExecution  *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *DescribeMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// DescribeMutableStateResponse is an internal type (TBD...)\ntype DescribeMutableStateResponse struct {\n\tMutableStateInCache    string `json:\"mutableStateInCache,omitempty\"`\n\tMutableStateInDatabase string `json:\"mutableStateInDatabase,omitempty\"`\n}\n\n// HistoryDescribeWorkflowExecutionRequest is an internal type (TBD...)\ntype HistoryDescribeWorkflowExecutionRequest struct {\n\tDomainUUID string                            `json:\"domainUUID,omitempty\"`\n\tRequest    *DescribeWorkflowExecutionRequest `json:\"request,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryDescribeWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// DomainFilter is an internal type (TBD...)\ntype DomainFilter struct {\n\tDomainIDs    []string `json:\"domainIDs,omitempty\"`\n\tReverseMatch bool     `json:\"reverseMatch,omitempty\"`\n}\n\n// GetDomainIDs is an internal getter (TBD...)\nfunc (v *DomainFilter) GetDomainIDs() (o []string) {\n\tif v != nil && v.DomainIDs != nil {\n\t\treturn v.DomainIDs\n\t}\n\treturn\n}\n\n// GetReverseMatch is an internal getter (TBD...)\nfunc (v *DomainFilter) GetReverseMatch() (o bool) {\n\tif v != nil {\n\t\treturn v.ReverseMatch\n\t}\n\treturn\n}\n\n// FailoverMarkerToken is an internal type (TBD...)\ntype FailoverMarkerToken struct {\n\tShardIDs       []int32                   `json:\"shardIDs,omitempty\"`\n\tFailoverMarker *FailoverMarkerAttributes `json:\"failoverMarker,omitempty\"`\n}\n\n// GetShardIDs is an internal getter (TBD...)\nfunc (v *FailoverMarkerToken) GetShardIDs() (o []int32) {\n\tif v != nil && v.ShardIDs != nil {\n\t\treturn v.ShardIDs\n\t}\n\treturn\n}\n\n// GetFailoverMarker is an internal getter (TBD...)\nfunc (v *FailoverMarkerToken) GetFailoverMarker() (o *FailoverMarkerAttributes) {\n\tif v != nil && v.FailoverMarker != nil {\n\t\treturn v.FailoverMarker\n\t}\n\treturn\n}\n\n// GetMutableStateRequest is an internal type (TBD...)\ntype GetMutableStateRequest struct {\n\tDomainUUID          string              `json:\"domainUUID,omitempty\"`\n\tExecution           *WorkflowExecution  `json:\"execution,omitempty\"`\n\tExpectedNextEventID int64               `json:\"expectedNextEventId,omitempty\"`\n\tCurrentBranchToken  []byte              `json:\"currentBranchToken,omitempty\"`\n\tVersionHistoryItem  *VersionHistoryItem `json:\"versionHistoryItem,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *GetMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetExpectedNextEventID is an internal getter (TBD...)\nfunc (v *GetMutableStateRequest) GetExpectedNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ExpectedNextEventID\n\t}\n\treturn\n}\n\n// GetMutableStateResponse is an internal type (TBD...)\ntype GetMutableStateResponse struct {\n\tExecution                            *WorkflowExecution `json:\"execution,omitempty\"`\n\tWorkflowType                         *WorkflowType      `json:\"workflowType,omitempty\"`\n\tNextEventID                          int64              `json:\"NextEventId,omitempty\"`\n\tPreviousStartedEventID               *int64             `json:\"PreviousStartedEventId,omitempty\"`\n\tLastFirstEventID                     int64              `json:\"LastFirstEventId,omitempty\"`\n\tTaskList                             *TaskList          `json:\"taskList,omitempty\"`\n\tStickyTaskList                       *TaskList          `json:\"stickyTaskList,omitempty\"`\n\tClientLibraryVersion                 string             `json:\"clientLibraryVersion,omitempty\"`\n\tClientFeatureVersion                 string             `json:\"clientFeatureVersion,omitempty\"`\n\tClientImpl                           string             `json:\"clientImpl,omitempty\"`\n\tIsWorkflowRunning                    bool               `json:\"isWorkflowRunning,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *int32             `json:\"stickyTaskListScheduleToStartTimeout,omitempty\"`\n\tEventStoreVersion                    int32              `json:\"eventStoreVersion,omitempty\"`\n\tCurrentBranchToken                   []byte             `json:\"currentBranchToken,omitempty\"`\n\tWorkflowState                        *int32             `json:\"workflowState,omitempty\"`\n\tWorkflowCloseState                   *int32             `json:\"workflowCloseState,omitempty\"`\n\tVersionHistories                     *VersionHistories  `json:\"versionHistories,omitempty\"`\n\tIsStickyTaskListEnabled              bool               `json:\"isStickyTaskListEnabled,omitempty\"`\n\tHistorySize                          int64              `json:\"historySize,omitempty\"`\n}\n\n// GetNextEventID is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.NextEventID\n\t}\n\treturn\n}\n\n// GetPreviousStartedEventID is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetPreviousStartedEventID() (o int64) {\n\tif v != nil && v.PreviousStartedEventID != nil {\n\t\treturn *v.PreviousStartedEventID\n\t}\n\treturn\n}\n\n// GetStickyTaskList is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetStickyTaskList() (o *TaskList) {\n\tif v != nil && v.StickyTaskList != nil {\n\t\treturn v.StickyTaskList\n\t}\n\treturn\n}\n\n// GetClientFeatureVersion is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetClientFeatureVersion() (o string) {\n\tif v != nil {\n\t\treturn v.ClientFeatureVersion\n\t}\n\treturn\n}\n\n// GetClientImpl is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetClientImpl() (o string) {\n\tif v != nil {\n\t\treturn v.ClientImpl\n\t}\n\treturn\n}\n\n// GetIsWorkflowRunning is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetIsWorkflowRunning() (o bool) {\n\tif v != nil {\n\t\treturn v.IsWorkflowRunning\n\t}\n\treturn\n}\n\n// GetStickyTaskListScheduleToStartTimeout is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetStickyTaskListScheduleToStartTimeout() (o int32) {\n\tif v != nil && v.StickyTaskListScheduleToStartTimeout != nil {\n\t\treturn *v.StickyTaskListScheduleToStartTimeout\n\t}\n\treturn\n}\n\n// GetCurrentBranchToken is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\treturn\n}\n\n// GetWorkflowCloseState is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetWorkflowCloseState() (o int32) {\n\tif v != nil && v.WorkflowCloseState != nil {\n\t\treturn *v.WorkflowCloseState\n\t}\n\treturn\n}\n\n// GetVersionHistories is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetVersionHistories() (o *VersionHistories) {\n\tif v != nil && v.VersionHistories != nil {\n\t\treturn v.VersionHistories\n\t}\n\treturn\n}\n\n// GetIsStickyTaskListEnabled is an internal getter (TBD...)\nfunc (v *GetMutableStateResponse) GetIsStickyTaskListEnabled() (o bool) {\n\tif v != nil {\n\t\treturn v.IsStickyTaskListEnabled\n\t}\n\treturn\n}\n\n// NotifyFailoverMarkersRequest is an internal type (TBD...)\ntype NotifyFailoverMarkersRequest struct {\n\tFailoverMarkerTokens []*FailoverMarkerToken `json:\"failoverMarkerTokens,omitempty\"`\n}\n\n// GetFailoverMarkerTokens is an internal getter (TBD...)\nfunc (v *NotifyFailoverMarkersRequest) GetFailoverMarkerTokens() (o []*FailoverMarkerToken) {\n\tif v != nil && v.FailoverMarkerTokens != nil {\n\t\treturn v.FailoverMarkerTokens\n\t}\n\treturn\n}\n\n// ParentExecutionInfo is an internal type (TBD...)\ntype ParentExecutionInfo struct {\n\tDomainUUID  string             `json:\"domainUUID,omitempty\"`\n\tDomain      string             `json:\"domain,omitempty\"`\n\tExecution   *WorkflowExecution `json:\"execution,omitempty\"`\n\tInitiatedID int64              `json:\"initiatedId,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ParentExecutionInfo) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *ParentExecutionInfo) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// PollMutableStateRequest is an internal type (TBD...)\ntype PollMutableStateRequest struct {\n\tDomainUUID          string              `json:\"domainUUID,omitempty\"`\n\tExecution           *WorkflowExecution  `json:\"execution,omitempty\"`\n\tExpectedNextEventID int64               `json:\"expectedNextEventId,omitempty\"`\n\tCurrentBranchToken  []byte              `json:\"currentBranchToken,omitempty\"`\n\tVersionHistoryItem  *VersionHistoryItem `json:\"versionHistoryItem,omitempty\"`\n}\n\nfunc (p *PollMutableStateRequest) GetVersionHistoryItem() *VersionHistoryItem {\n\tif p.VersionHistoryItem == nil {\n\t\treturn nil\n\t}\n\treturn p.VersionHistoryItem\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (p *PollMutableStateRequest) GetDomainUUID() (o string) {\n\tif p != nil {\n\t\treturn p.DomainUUID\n\t}\n\treturn\n}\n\n// PollMutableStateResponse is an internal type (TBD...)\ntype PollMutableStateResponse struct {\n\tExecution                            *WorkflowExecution `json:\"execution,omitempty\"`\n\tWorkflowType                         *WorkflowType      `json:\"workflowType,omitempty\"`\n\tNextEventID                          int64              `json:\"NextEventId,omitempty\"`\n\tPreviousStartedEventID               *int64             `json:\"PreviousStartedEventId,omitempty\"`\n\tLastFirstEventID                     int64              `json:\"LastFirstEventId,omitempty\"`\n\tTaskList                             *TaskList          `json:\"taskList,omitempty\"`\n\tStickyTaskList                       *TaskList          `json:\"stickyTaskList,omitempty\"`\n\tClientLibraryVersion                 string             `json:\"clientLibraryVersion,omitempty\"`\n\tClientFeatureVersion                 string             `json:\"clientFeatureVersion,omitempty\"`\n\tClientImpl                           string             `json:\"clientImpl,omitempty\"`\n\tStickyTaskListScheduleToStartTimeout *int32             `json:\"stickyTaskListScheduleToStartTimeout,omitempty\"`\n\tCurrentBranchToken                   []byte             `json:\"currentBranchToken,omitempty\"`\n\tVersionHistories                     *VersionHistories  `json:\"versionHistories,omitempty\"`\n\tWorkflowState                        *int32             `json:\"workflowState,omitempty\"`\n\tWorkflowCloseState                   *int32             `json:\"workflowCloseState,omitempty\"`\n}\n\n// GetNextEventID is an internal getter (TBD...)\nfunc (v *PollMutableStateResponse) GetNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.NextEventID\n\t}\n\treturn\n}\n\n// GetLastFirstEventID is an internal getter (TBD...)\nfunc (v *PollMutableStateResponse) GetLastFirstEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.LastFirstEventID\n\t}\n\treturn\n}\n\n// GetWorkflowCloseState is an internal getter (TBD...)\nfunc (v *PollMutableStateResponse) GetWorkflowCloseState() (o int32) {\n\tif v != nil && v.WorkflowCloseState != nil {\n\t\treturn *v.WorkflowCloseState\n\t}\n\treturn\n}\n\ntype QueueState struct {\n\tVirtualQueueStates    map[int64]*VirtualQueueState\n\tExclusiveMaxReadLevel *TaskKey\n}\n\nfunc (q *QueueState) Copy() (o *QueueState) {\n\tif q == nil {\n\t\treturn\n\t}\n\tvar virtualQueueStates map[int64]*VirtualQueueState\n\tif q.VirtualQueueStates != nil {\n\t\tvirtualQueueStates = make(map[int64]*VirtualQueueState)\n\t\tfor key, value := range q.VirtualQueueStates {\n\t\t\tvirtualQueueStates[key] = value.Copy()\n\t\t}\n\t}\n\to = &QueueState{\n\t\tVirtualQueueStates:    virtualQueueStates,\n\t\tExclusiveMaxReadLevel: q.ExclusiveMaxReadLevel.Copy(),\n\t}\n\treturn\n}\n\ntype VirtualQueueState struct {\n\tVirtualSliceStates []*VirtualSliceState\n}\n\nfunc (v *VirtualQueueState) Copy() (o *VirtualQueueState) {\n\tif v == nil {\n\t\treturn\n\t}\n\tvar virtualSliceStates []*VirtualSliceState\n\tif v.VirtualSliceStates != nil {\n\t\tvirtualSliceStates = make([]*VirtualSliceState, len(v.VirtualSliceStates))\n\t\tfor i, slice := range v.VirtualSliceStates {\n\t\t\tvirtualSliceStates[i] = slice.Copy()\n\t\t}\n\t}\n\to = &VirtualQueueState{\n\t\tVirtualSliceStates: virtualSliceStates,\n\t}\n\treturn\n}\n\ntype VirtualSliceState struct {\n\tTaskRange *TaskRange\n\tPredicate *Predicate\n}\n\nfunc (v *VirtualSliceState) Copy() (o *VirtualSliceState) {\n\tif v == nil {\n\t\treturn\n\t}\n\to = &VirtualSliceState{\n\t\tTaskRange: v.TaskRange.Copy(),\n\t\tPredicate: v.Predicate.Copy(),\n\t}\n\treturn\n}\n\ntype TaskRange struct {\n\tInclusiveMin *TaskKey\n\tExclusiveMax *TaskKey\n}\n\nfunc (t *TaskRange) Copy() (o *TaskRange) {\n\tif t == nil {\n\t\treturn\n\t}\n\to = &TaskRange{\n\t\tInclusiveMin: t.InclusiveMin.Copy(),\n\t\tExclusiveMax: t.ExclusiveMax.Copy(),\n\t}\n\treturn\n}\n\ntype TaskKey struct {\n\tScheduledTimeNano int64\n\tTaskID            int64\n}\n\nfunc (t *TaskKey) Copy() (o *TaskKey) {\n\tif t == nil {\n\t\treturn\n\t}\n\to = &TaskKey{\n\t\tScheduledTimeNano: t.ScheduledTimeNano,\n\t\tTaskID:            t.TaskID,\n\t}\n\treturn\n}\n\n// ProcessingQueueState is an internal type (TBD...)\ntype ProcessingQueueState struct {\n\tLevel        *int32        `json:\"level,omitempty\"`\n\tAckLevel     *int64        `json:\"ackLevel,omitempty\"`\n\tMaxLevel     *int64        `json:\"maxLevel,omitempty\"`\n\tDomainFilter *DomainFilter `json:\"domainFilter,omitempty\"`\n}\n\n// GetLevel is an internal getter (TBD...)\nfunc (v *ProcessingQueueState) GetLevel() (o int32) {\n\tif v != nil && v.Level != nil {\n\t\treturn *v.Level\n\t}\n\treturn\n}\n\n// GetAckLevel is an internal getter (TBD...)\nfunc (v *ProcessingQueueState) GetAckLevel() (o int64) {\n\tif v != nil && v.AckLevel != nil {\n\t\treturn *v.AckLevel\n\t}\n\treturn\n}\n\n// GetMaxLevel is an internal getter (TBD...)\nfunc (v *ProcessingQueueState) GetMaxLevel() (o int64) {\n\tif v != nil && v.MaxLevel != nil {\n\t\treturn *v.MaxLevel\n\t}\n\treturn\n}\n\n// GetDomainFilter is an internal getter (TBD...)\nfunc (v *ProcessingQueueState) GetDomainFilter() (o *DomainFilter) {\n\tif v != nil && v.DomainFilter != nil {\n\t\treturn v.DomainFilter\n\t}\n\treturn\n}\n\n// ProcessingQueueStates is an internal type (TBD...)\ntype ProcessingQueueStates struct {\n\tStatesByCluster map[string][]*ProcessingQueueState `json:\"statesByCluster,omitempty\"`\n}\n\n// HistoryQueryWorkflowRequest is an internal type (TBD...)\ntype HistoryQueryWorkflowRequest struct {\n\tDomainUUID string                `json:\"domainUUID,omitempty\"`\n\tRequest    *QueryWorkflowRequest `json:\"request,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryQueryWorkflowRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetRequest is an internal getter (TBD...)\nfunc (v *HistoryQueryWorkflowRequest) GetRequest() (o *QueryWorkflowRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\treturn\n}\n\n// HistoryQueryWorkflowResponse is an internal type (TBD...)\ntype HistoryQueryWorkflowResponse struct {\n\tResponse *QueryWorkflowResponse `json:\"response,omitempty\"`\n}\n\n// GetResponse is an internal getter (TBD...)\nfunc (v *HistoryQueryWorkflowResponse) GetResponse() (o *QueryWorkflowResponse) {\n\tif v != nil && v.Response != nil {\n\t\treturn v.Response\n\t}\n\treturn\n}\n\n// HistoryReapplyEventsRequest is an internal type (TBD...)\ntype HistoryReapplyEventsRequest struct {\n\tDomainUUID string                `json:\"domainUUID,omitempty\"`\n\tRequest    *ReapplyEventsRequest `json:\"request,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryReapplyEventsRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetRequest is an internal getter (TBD...)\nfunc (v *HistoryReapplyEventsRequest) GetRequest() (o *ReapplyEventsRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\treturn\n}\n\n// HistoryRecordActivityTaskHeartbeatRequest is an internal type (TBD...)\ntype HistoryRecordActivityTaskHeartbeatRequest struct {\n\tDomainUUID       string                              `json:\"domainUUID,omitempty\"`\n\tHeartbeatRequest *RecordActivityTaskHeartbeatRequest `json:\"heartbeatRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRecordActivityTaskHeartbeatRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// RecordActivityTaskStartedRequest is an internal type (TBD...)\ntype RecordActivityTaskStartedRequest struct {\n\tDomainUUID        string                      `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *WorkflowExecution          `json:\"workflowExecution,omitempty\"`\n\tScheduleID        int64                       `json:\"scheduleId,omitempty\"`\n\tTaskID            int64                       `json:\"taskId,omitempty\"`\n\tRequestID         string                      `json:\"requestId,omitempty\"`\n\tPollRequest       *PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetScheduleID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedRequest) GetScheduleID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedRequest) GetTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// RecordActivityTaskStartedResponse is an internal type (TBD...)\ntype RecordActivityTaskStartedResponse struct {\n\tScheduledEvent                  *HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedTimestamp                *int64        `json:\"startedTimestamp,omitempty\"`\n\tAttempt                         int64         `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64        `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte        `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *WorkflowType `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  string        `json:\"workflowDomain,omitempty\"`\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedResponse) GetAttempt() (o int64) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetScheduledTimestampOfThisAttempt is an internal getter (TBD...)\nfunc (v *RecordActivityTaskStartedResponse) GetScheduledTimestampOfThisAttempt() (o int64) {\n\tif v != nil && v.ScheduledTimestampOfThisAttempt != nil {\n\t\treturn *v.ScheduledTimestampOfThisAttempt\n\t}\n\treturn\n}\n\n// RecordChildExecutionCompletedRequest is an internal type (TBD...)\ntype RecordChildExecutionCompletedRequest struct {\n\tDomainUUID         string             `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution  *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tInitiatedID        int64              `json:\"initiatedId,omitempty\"`\n\tCompletedExecution *WorkflowExecution `json:\"completedExecution,omitempty\"`\n\tCompletionEvent    *HistoryEvent      `json:\"completionEvent,omitempty\"`\n\tStartedID          int64              `json:\"startedId,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *RecordChildExecutionCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// RecordDecisionTaskStartedRequest is an internal type (TBD...)\ntype RecordDecisionTaskStartedRequest struct {\n\tDomainUUID        string                      `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *WorkflowExecution          `json:\"workflowExecution,omitempty\"`\n\tScheduleID        int64                       `json:\"scheduleId,omitempty\"`\n\tTaskID            int64                       `json:\"taskId,omitempty\"`\n\tRequestID         string                      `json:\"requestId,omitempty\"`\n\tPollRequest       *PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetScheduleID is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedRequest) GetScheduleID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// RecordDecisionTaskStartedResponse is an internal type (TBD...)\ntype RecordDecisionTaskStartedResponse struct {\n\tWorkflowType              *WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventID    *int64                    `json:\"previousStartedEventId,omitempty\"`\n\tScheduledEventID          int64                     `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID            int64                     `json:\"startedEventId,omitempty\"`\n\tNextEventID               int64                     `json:\"nextEventId,omitempty\"`\n\tAttempt                   int64                     `json:\"attempt,omitempty\"`\n\tStickyExecutionEnabled    bool                      `json:\"stickyExecutionEnabled,omitempty\"`\n\tDecisionInfo              *TransientDecisionInfo    `json:\"decisionInfo,omitempty\"`\n\tWorkflowExecutionTaskList *TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tEventStoreVersion         int32                     `json:\"eventStoreVersion,omitempty\"`\n\tBranchToken               []byte                    `json:\"branchToken,omitempty\"`\n\tScheduledTimestamp        *int64                    `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                    `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*WorkflowQuery `json:\"queries,omitempty\"`\n\tHistorySize               int64                     `json:\"historySize,omitempty\"`\n}\n\n// GetPreviousStartedEventID is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedResponse) GetPreviousStartedEventID() (o int64) {\n\tif v != nil && v.PreviousStartedEventID != nil {\n\t\treturn *v.PreviousStartedEventID\n\t}\n\treturn\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedResponse) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *RecordDecisionTaskStartedResponse) GetAttempt() (o int64) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// HistoryRefreshWorkflowTasksRequest is an internal type (TBD...)\ntype HistoryRefreshWorkflowTasksRequest struct {\n\tDomainUIID string                       `json:\"domainUIID,omitempty\"`\n\tRequest    *RefreshWorkflowTasksRequest `json:\"request,omitempty\"`\n}\n\n// GetRequest is an internal getter (TBD...)\nfunc (v *HistoryRefreshWorkflowTasksRequest) GetRequest() (o *RefreshWorkflowTasksRequest) {\n\tif v != nil && v.Request != nil {\n\t\treturn v.Request\n\t}\n\treturn\n}\n\n// RemoveSignalMutableStateRequest is an internal type (TBD...)\ntype RemoveSignalMutableStateRequest struct {\n\tDomainUUID        string             `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tRequestID         string             `json:\"requestId,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *RemoveSignalMutableStateRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *RemoveSignalMutableStateRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// ReplicateEventsV2Request is an internal type (TBD...)\ntype ReplicateEventsV2Request struct {\n\tDomainUUID          string                `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution   *WorkflowExecution    `json:\"workflowExecution,omitempty\"`\n\tVersionHistoryItems []*VersionHistoryItem `json:\"versionHistoryItems,omitempty\"`\n\tEvents              *DataBlob             `json:\"events,omitempty\"`\n\tNewRunEvents        *DataBlob             `json:\"newRunEvents,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *ReplicateEventsV2Request) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryRequestCancelWorkflowExecutionRequest is an internal type (TBD...)\ntype HistoryRequestCancelWorkflowExecutionRequest struct {\n\tDomainUUID                string                                 `json:\"domainUUID,omitempty\"`\n\tCancelRequest             *RequestCancelWorkflowExecutionRequest `json:\"cancelRequest,omitempty\"`\n\tExternalInitiatedEventID  *int64                                 `json:\"externalInitiatedEventId,omitempty\"`\n\tExternalWorkflowExecution *WorkflowExecution                     `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         bool                                   `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRequestCancelWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetCancelRequest is an internal getter (TBD...)\nfunc (v *HistoryRequestCancelWorkflowExecutionRequest) GetCancelRequest() (o *RequestCancelWorkflowExecutionRequest) {\n\tif v != nil && v.CancelRequest != nil {\n\t\treturn v.CancelRequest\n\t}\n\treturn\n}\n\n// GetExternalWorkflowExecution is an internal getter (TBD...)\nfunc (v *HistoryRequestCancelWorkflowExecutionRequest) GetExternalWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\treturn\n}\n\n// GetChildWorkflowOnly is an internal getter (TBD...)\nfunc (v *HistoryRequestCancelWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil {\n\t\treturn v.ChildWorkflowOnly\n\t}\n\treturn\n}\n\n// HistoryResetStickyTaskListRequest is an internal type (TBD...)\ntype HistoryResetStickyTaskListRequest struct {\n\tDomainUUID string             `json:\"domainUUID,omitempty\"`\n\tExecution  *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryResetStickyTaskListRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryResetStickyTaskListResponse is an internal type (TBD...)\ntype HistoryResetStickyTaskListResponse struct {\n}\n\n// HistoryResetWorkflowExecutionRequest is an internal type (TBD...)\ntype HistoryResetWorkflowExecutionRequest struct {\n\tDomainUUID   string                         `json:\"domainUUID,omitempty\"`\n\tResetRequest *ResetWorkflowExecutionRequest `json:\"resetRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryResetWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryRespondActivityTaskCanceledRequest is an internal type (TBD...)\ntype HistoryRespondActivityTaskCanceledRequest struct {\n\tDomainUUID    string                              `json:\"domainUUID,omitempty\"`\n\tCancelRequest *RespondActivityTaskCanceledRequest `json:\"cancelRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRespondActivityTaskCanceledRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryRespondActivityTaskCompletedRequest is an internal type (TBD...)\ntype HistoryRespondActivityTaskCompletedRequest struct {\n\tDomainUUID      string                               `json:\"domainUUID,omitempty\"`\n\tCompleteRequest *RespondActivityTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRespondActivityTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryRespondActivityTaskFailedRequest is an internal type (TBD...)\ntype HistoryRespondActivityTaskFailedRequest struct {\n\tDomainUUID    string                            `json:\"domainUUID,omitempty\"`\n\tFailedRequest *RespondActivityTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRespondActivityTaskFailedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// HistoryRespondDecisionTaskCompletedRequest is an internal type (TBD...)\ntype HistoryRespondDecisionTaskCompletedRequest struct {\n\tDomainUUID      string                               `json:\"domainUUID,omitempty\"`\n\tCompleteRequest *RespondDecisionTaskCompletedRequest `json:\"completeRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRespondDecisionTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetCompleteRequest is an internal getter (TBD...)\nfunc (v *HistoryRespondDecisionTaskCompletedRequest) GetCompleteRequest() (o *RespondDecisionTaskCompletedRequest) {\n\tif v != nil && v.CompleteRequest != nil {\n\t\treturn v.CompleteRequest\n\t}\n\treturn\n}\n\n// HistoryRespondDecisionTaskCompletedResponse is an internal type (TBD...)\ntype HistoryRespondDecisionTaskCompletedResponse struct {\n\tStartedResponse             *RecordDecisionTaskStartedResponse    `json:\"startedResponse,omitempty\"`\n\tActivitiesToDispatchLocally map[string]*ActivityLocalDispatchInfo `json:\"activitiesToDispatchLocally,omitempty\"`\n}\n\n// HistoryRespondDecisionTaskFailedRequest is an internal type (TBD...)\ntype HistoryRespondDecisionTaskFailedRequest struct {\n\tDomainUUID    string                            `json:\"domainUUID,omitempty\"`\n\tFailedRequest *RespondDecisionTaskFailedRequest `json:\"failedRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryRespondDecisionTaskFailedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// ScheduleDecisionTaskRequest is an internal type (TBD...)\ntype ScheduleDecisionTaskRequest struct {\n\tDomainUUID        string             `json:\"domainUUID,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIsFirstDecision   bool               `json:\"isFirstDecision,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *ScheduleDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// ShardOwnershipLostError is an internal type (TBD...)\ntype ShardOwnershipLostError struct {\n\tMessage string `json:\"message,omitempty\"`\n\tOwner   string `json:\"owner,omitempty\"`\n}\n\n// GetOwner is an internal getter (TBD...)\nfunc (v *ShardOwnershipLostError) GetOwner() (o string) {\n\tif v != nil {\n\t\treturn v.Owner\n\t}\n\treturn\n}\n\n// HistorySignalWithStartWorkflowExecutionRequest is an internal type (TBD...)\ntype HistorySignalWithStartWorkflowExecutionRequest struct {\n\tDomainUUID             string                                   `json:\"domainUUID,omitempty\"`\n\tSignalWithStartRequest *SignalWithStartWorkflowExecutionRequest `json:\"signalWithStartRequest,omitempty\"`\n\tPartitionConfig        map[string]string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistorySignalWithStartWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *HistorySignalWithStartWorkflowExecutionRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// HistorySignalWorkflowExecutionRequest is an internal type (TBD...)\ntype HistorySignalWorkflowExecutionRequest struct {\n\tDomainUUID                string                          `json:\"domainUUID,omitempty\"`\n\tSignalRequest             *SignalWorkflowExecutionRequest `json:\"signalRequest,omitempty\"`\n\tExternalWorkflowExecution *WorkflowExecution              `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         bool                            `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistorySignalWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetChildWorkflowOnly is an internal getter (TBD...)\nfunc (v *HistorySignalWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil {\n\t\treturn v.ChildWorkflowOnly\n\t}\n\treturn\n}\n\n// HistoryStartWorkflowExecutionRequest is an internal type (TBD...)\ntype HistoryStartWorkflowExecutionRequest struct {\n\tDomainUUID                      string                         `json:\"domainUUID,omitempty\"`\n\tStartRequest                    *StartWorkflowExecutionRequest `json:\"startRequest,omitempty\"`\n\tParentExecutionInfo             *ParentExecutionInfo           `json:\"parentExecutionInfo,omitempty\"`\n\tAttempt                         int32                          `json:\"attempt,omitempty\"`\n\tExpirationTimestamp             *int64                         `json:\"expirationTimestamp,omitempty\"`\n\tContinueAsNewInitiator          *ContinueAsNewInitiator        `json:\"continueAsNewInitiator,omitempty\"`\n\tContinuedFailureReason          *string                        `json:\"continuedFailureReason,omitempty\"`\n\tContinuedFailureDetails         []byte                         `json:\"continuedFailureDetails,omitempty\"`\n\tLastCompletionResult            []byte                         `json:\"lastCompletionResult,omitempty\"`\n\tFirstDecisionTaskBackoffSeconds *int32                         `json:\"firstDecisionTaskBackoffSeconds,omitempty\"`\n\tPartitionConfig                 map[string]string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryStartWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *HistoryStartWorkflowExecutionRequest) GetAttempt() (o int32) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetExpirationTimestamp is an internal getter (TBD...)\nfunc (v *HistoryStartWorkflowExecutionRequest) GetExpirationTimestamp() (o int64) {\n\tif v != nil && v.ExpirationTimestamp != nil {\n\t\treturn *v.ExpirationTimestamp\n\t}\n\treturn\n}\n\n// GetFirstDecisionTaskBackoffSeconds is an internal getter (TBD...)\nfunc (v *HistoryStartWorkflowExecutionRequest) GetFirstDecisionTaskBackoffSeconds() (o int32) {\n\tif v != nil && v.FirstDecisionTaskBackoffSeconds != nil {\n\t\treturn *v.FirstDecisionTaskBackoffSeconds\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *HistoryStartWorkflowExecutionRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// SyncActivityRequest is an internal type (TBD...)\ntype SyncActivityRequest struct {\n\tDomainID           string          `json:\"domainId,omitempty\"`\n\tWorkflowID         string          `json:\"workflowId,omitempty\"`\n\tRunID              string          `json:\"runId,omitempty\"`\n\tVersion            int64           `json:\"version,omitempty\"`\n\tScheduledID        int64           `json:\"scheduledId,omitempty\"`\n\tScheduledTime      *int64          `json:\"scheduledTime,omitempty\"`\n\tStartedID          int64           `json:\"startedId,omitempty\"`\n\tStartedTime        *int64          `json:\"startedTime,omitempty\"`\n\tLastHeartbeatTime  *int64          `json:\"lastHeartbeatTime,omitempty\"`\n\tDetails            []byte          `json:\"details,omitempty\"`\n\tAttempt            int32           `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string         `json:\"lastFailureReason,omitempty\"`\n\tLastWorkerIdentity string          `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails []byte          `json:\"lastFailureDetails,omitempty\"`\n\tVersionHistory     *VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetVersion is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.Version\n\t}\n\treturn\n}\n\n// GetScheduledID is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetScheduledID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledID\n\t}\n\treturn\n}\n\n// GetScheduledTime is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetScheduledTime() (o int64) {\n\tif v != nil && v.ScheduledTime != nil {\n\t\treturn *v.ScheduledTime\n\t}\n\treturn\n}\n\n// GetStartedID is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetStartedID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedID\n\t}\n\treturn\n}\n\n// GetStartedTime is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetStartedTime() (o int64) {\n\tif v != nil && v.StartedTime != nil {\n\t\treturn *v.StartedTime\n\t}\n\treturn\n}\n\n// GetLastHeartbeatTime is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetLastHeartbeatTime() (o int64) {\n\tif v != nil && v.LastHeartbeatTime != nil {\n\t\treturn *v.LastHeartbeatTime\n\t}\n\treturn\n}\n\n// GetDetails is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetAttempt() (o int32) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetLastFailureReason is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\treturn\n}\n\n// GetLastWorkerIdentity is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetLastWorkerIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.LastWorkerIdentity\n\t}\n\treturn\n}\n\n// GetLastFailureDetails is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\treturn\n}\n\n// GetVersionHistory is an internal getter (TBD...)\nfunc (v *SyncActivityRequest) GetVersionHistory() (o *VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\treturn\n}\n\n// SyncShardStatusRequest is an internal type (TBD...)\ntype SyncShardStatusRequest struct {\n\tSourceCluster string `json:\"sourceCluster,omitempty\"`\n\tShardID       int64  `json:\"shardId,omitempty\"`\n\tTimestamp     *int64 `json:\"timestamp,omitempty\"`\n}\n\n// GetSourceCluster is an internal getter (TBD...)\nfunc (v *SyncShardStatusRequest) GetSourceCluster() (o string) {\n\tif v != nil {\n\t\treturn v.SourceCluster\n\t}\n\treturn\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *SyncShardStatusRequest) GetShardID() (o int64) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetTimestamp is an internal getter (TBD...)\nfunc (v *SyncShardStatusRequest) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\treturn\n}\n\n// HistoryTerminateWorkflowExecutionRequest is an internal type (TBD...)\ntype HistoryTerminateWorkflowExecutionRequest struct {\n\tDomainUUID                string                             `json:\"domainUUID,omitempty\"`\n\tTerminateRequest          *TerminateWorkflowExecutionRequest `json:\"terminateRequest,omitempty\"`\n\tExternalWorkflowExecution *WorkflowExecution                 `json:\"externalWorkflowExecution,omitempty\"`\n\tChildWorkflowOnly         bool                               `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *HistoryTerminateWorkflowExecutionRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetTerminateRequest is an internal getter (TBD...)\nfunc (v *HistoryTerminateWorkflowExecutionRequest) GetTerminateRequest() (o *TerminateWorkflowExecutionRequest) {\n\tif v != nil && v.TerminateRequest != nil {\n\t\treturn v.TerminateRequest\n\t}\n\treturn\n}\n\n// GetExternalWorkflowExecution is an internal getter (TBD...)\nfunc (v *HistoryTerminateWorkflowExecutionRequest) GetExternalWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.ExternalWorkflowExecution != nil {\n\t\treturn v.ExternalWorkflowExecution\n\t}\n\treturn\n}\n\n// GetChildWorkflowOnly is an internal getter (TBD...)\nfunc (v *HistoryTerminateWorkflowExecutionRequest) GetChildWorkflowOnly() (o bool) {\n\tif v != nil {\n\t\treturn v.ChildWorkflowOnly\n\t}\n\treturn\n}\n\n// GetFailoverInfoRequest is an internal type (TBD...)\ntype GetFailoverInfoRequest struct {\n\tDomainID string `json:\"domainID,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *GetFailoverInfoRequest) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetFailoverInfoResponse is an internal type (TBD...)\ntype GetFailoverInfoResponse struct {\n\tCompletedShardCount int32   `json:\"completedShardCount,omitempty\"`\n\tPendingShards       []int32 `json:\"pendingShards,omitempty\"`\n}\n\n// GetCompletedShardCount is an internal getter (TBD...)\nfunc (v *GetFailoverInfoResponse) GetCompletedShardCount() (o int32) {\n\tif v != nil {\n\t\treturn v.CompletedShardCount\n\t}\n\treturn\n}\n\n// GetPendingShards is an internal getter (TBD...)\nfunc (v *GetFailoverInfoResponse) GetPendingShards() (o []int32) {\n\tif v != nil {\n\t\treturn v.PendingShards\n\t}\n\treturn\n}\n\ntype RatelimitUpdateRequest struct {\n\tAny *Any `json:\"any\"`\n}\n\ntype RatelimitUpdateResponse struct {\n\tAny *Any `json:\"any\"`\n}\n"
  },
  {
    "path": "common/types/history_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nvar (\n\tdomainUUID                = \"test-domainUUID\"\n\tmutableStateInCache       = \"test-cache\"\n\tmutableStateInDatabase    = \"test-database\"\n\tdomainIDs                 = []string{\"id1\", \"id2\", \"id3\"}\n\tshardIDs                  = []int32{1, 2, 3}\n\texpectedNextEventID       = int64(123)\n\tnextEventID               = int64(456)\n\tpreviousStartedEventID    = int64(789)\n\tstickyTaskListTimeout     = int32(100)\n\tworkflowCloseState        = int32(2)\n\tclientFeatureVersion      = \"v1.0\"\n\tclientImpl                = \"test-client\"\n\tcurrentBranchToken        = []byte(\"branch-token\")\n\trequestID                 = \"test-requestID\"\n\tscheduleID                = int64(100)\n\ttaskID                    = int64(200)\n\tstartedTimestamp          = int64(123456789)\n\tattempt                   = int64(3)\n\tattempt32                 = int32(3)\n\texpirationTimestamp       = int64(1234567890)\n\tfirstDecisionBackoff      = int32(60)\n\tpartitionConfig           = map[string]string{\"partition1\": \"value1\"}\n\tscheduledTime             = int64(1617181920)\n\tstartedTime               = int64(1617182920)\n\tlastHeartbeatTime         = int64(1617183920)\n\tlastFailureReason         = \"test-failure\"\n\tlastWorkerIdentity        = \"test-worker\"\n\tdetails                   = []byte(\"test-details\")\n\tlastFailureDetails        = []byte(\"failure-details\")\n\tversionHistory            = &VersionHistory{}\n\tsourceCluster             = \"source-cluster\"\n\tshardID                   = int64(1000)\n\tterminateRequest          = &TerminateWorkflowExecutionRequest{}\n\texternalWorkflowExecution = &WorkflowExecution{}\n\tchildWorkflowOnly         = true\n\tpendingShards             = []int32{1, 2, 3}\n\tcompletedShardCount       = int32(5)\n\tworkflowID                = \"test-workflow-id\"\n\trunID                     = \"test-run-id\"\n\tscheduledID               = int64(202)\n\tstartedID                 = int64(303)\n\tversion                   = int64(10)\n)\n\nfunc TestDescribeMutableStateRequest(t *testing.T) {\n\ttestStruct := DescribeMutableStateRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *DescribeMutableStateRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res) // Should return empty string when struct is nil\n}\n\nfunc TestDescribeMutableStateResponse(t *testing.T) {\n\ttestStruct := DescribeMutableStateResponse{\n\t\tMutableStateInCache:    mutableStateInCache,\n\t\tMutableStateInDatabase: mutableStateInDatabase,\n\t}\n\n\tassert.Equal(t, mutableStateInCache, testStruct.MutableStateInCache)\n\tassert.Equal(t, mutableStateInDatabase, testStruct.MutableStateInDatabase)\n}\n\nfunc TestHistoryDescribeWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryDescribeWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryDescribeWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res) // Should return empty string when struct is nil\n}\n\nfunc TestDomainFilter_GetDomainIDs(t *testing.T) {\n\ttestStruct := DomainFilter{\n\t\tDomainIDs: domainIDs,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainIDs()\n\tassert.Equal(t, domainIDs, res)\n\n\t// Nil struct test\n\tvar nilStruct *DomainFilter\n\tres = nilStruct.GetDomainIDs()\n\tassert.Empty(t, res) // Should return empty slice when struct is nil\n}\n\nfunc TestDomainFilter_GetReverseMatch(t *testing.T) {\n\ttestStruct := DomainFilter{\n\t\tReverseMatch: true,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetReverseMatch()\n\tassert.True(t, res)\n\n\t// Nil struct test\n\tvar nilStruct *DomainFilter\n\tres = nilStruct.GetReverseMatch()\n\tassert.False(t, res) // Should return false when struct is nil\n}\n\nfunc TestFailoverMarkerToken_GetShardIDs(t *testing.T) {\n\ttestStruct := FailoverMarkerToken{\n\t\tShardIDs: shardIDs,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetShardIDs()\n\tassert.Equal(t, shardIDs, res)\n\n\t// Nil struct test\n\tvar nilStruct *FailoverMarkerToken\n\tres = nilStruct.GetShardIDs()\n\tassert.Empty(t, res) // Should return empty slice when struct is nil\n}\n\nfunc TestFailoverMarkerToken_GetFailoverMarker(t *testing.T) {\n\tfailoverMarker := &FailoverMarkerAttributes{}\n\n\ttestStruct := FailoverMarkerToken{\n\t\tFailoverMarker: failoverMarker,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetFailoverMarker()\n\tassert.Equal(t, failoverMarker, res)\n\n\t// Nil struct test\n\tvar nilStruct *FailoverMarkerToken\n\tres = nilStruct.GetFailoverMarker()\n\tassert.Nil(t, res) // Should return nil when struct is nil\n}\n\nfunc TestGetMutableStateRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := GetMutableStateRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetMutableStateRequest_GetExpectedNextEventID(t *testing.T) {\n\ttestStruct := GetMutableStateRequest{\n\t\tExpectedNextEventID: expectedNextEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetExpectedNextEventID()\n\tassert.Equal(t, expectedNextEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateRequest\n\tres = nilStruct.GetExpectedNextEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetMutableStateResponse_GetNextEventID(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tNextEventID: nextEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetNextEventID()\n\tassert.Equal(t, nextEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetNextEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetMutableStateResponse_GetPreviousStartedEventID(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tPreviousStartedEventID: &previousStartedEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetPreviousStartedEventID()\n\tassert.Equal(t, previousStartedEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetPreviousStartedEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetMutableStateResponse_GetStickyTaskList(t *testing.T) {\n\tstickyTaskList := &TaskList{}\n\ttestStruct := GetMutableStateResponse{\n\t\tStickyTaskList: stickyTaskList,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetStickyTaskList()\n\tassert.Equal(t, stickyTaskList, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetStickyTaskList()\n\tassert.Nil(t, res)\n}\n\nfunc TestGetMutableStateResponse_GetClientFeatureVersion(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tClientFeatureVersion: clientFeatureVersion,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetClientFeatureVersion()\n\tassert.Equal(t, clientFeatureVersion, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetClientFeatureVersion()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetMutableStateResponse_GetClientImpl(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tClientImpl: clientImpl,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetClientImpl()\n\tassert.Equal(t, clientImpl, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetClientImpl()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetMutableStateResponse_GetIsWorkflowRunning(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tIsWorkflowRunning: true,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetIsWorkflowRunning()\n\tassert.True(t, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetIsWorkflowRunning()\n\tassert.False(t, res)\n}\n\nfunc TestGetMutableStateResponse_GetStickyTaskListScheduleToStartTimeout(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tStickyTaskListScheduleToStartTimeout: &stickyTaskListTimeout,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetStickyTaskListScheduleToStartTimeout()\n\tassert.Equal(t, stickyTaskListTimeout, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetStickyTaskListScheduleToStartTimeout()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestGetMutableStateResponse_GetCurrentBranchToken(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tCurrentBranchToken: currentBranchToken,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetCurrentBranchToken()\n\tassert.Equal(t, currentBranchToken, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetCurrentBranchToken()\n\tassert.Nil(t, res)\n}\n\nfunc TestGetMutableStateResponse_GetWorkflowCloseState(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tWorkflowCloseState: &workflowCloseState,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetWorkflowCloseState()\n\tassert.Equal(t, workflowCloseState, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetWorkflowCloseState()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestGetMutableStateResponse_GetVersionHistories(t *testing.T) {\n\tversionHistories := &VersionHistories{}\n\ttestStruct := GetMutableStateResponse{\n\t\tVersionHistories: versionHistories,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetVersionHistories()\n\tassert.Equal(t, versionHistories, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetVersionHistories()\n\tassert.Nil(t, res)\n}\n\nfunc TestGetMutableStateResponse_GetIsStickyTaskListEnabled(t *testing.T) {\n\ttestStruct := GetMutableStateResponse{\n\t\tIsStickyTaskListEnabled: true,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetIsStickyTaskListEnabled()\n\tassert.True(t, res)\n\n\t// Nil struct test\n\tvar nilStruct *GetMutableStateResponse\n\tres = nilStruct.GetIsStickyTaskListEnabled()\n\tassert.False(t, res)\n}\n\nfunc TestNotifyFailoverMarkersRequest_GetFailoverMarkerTokens(t *testing.T) {\n\tfailoverMarkerTokens := []*FailoverMarkerToken{}\n\ttestStruct := NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: failoverMarkerTokens,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetFailoverMarkerTokens()\n\tassert.Equal(t, failoverMarkerTokens, res)\n\n\t// Nil struct test\n\tvar nilStruct *NotifyFailoverMarkersRequest\n\tres = nilStruct.GetFailoverMarkerTokens()\n\tassert.Nil(t, res)\n}\n\nfunc TestParentExecutionInfo_GetDomain(t *testing.T) {\n\ttestStruct := ParentExecutionInfo{\n\t\tDomain: \"test-domain\",\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomain()\n\tassert.Equal(t, \"test-domain\", res)\n\n\t// Nil struct test\n\tvar nilStruct *ParentExecutionInfo\n\tres = nilStruct.GetDomain()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestParentExecutionInfo_GetExecution(t *testing.T) {\n\texecution := &WorkflowExecution{}\n\ttestStruct := ParentExecutionInfo{\n\t\tExecution: execution,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetExecution()\n\tassert.Equal(t, execution, res)\n\n\t// Nil struct test\n\tvar nilStruct *ParentExecutionInfo\n\tres = nilStruct.GetExecution()\n\tassert.Nil(t, res)\n}\n\nfunc TestPollMutableStateRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := PollMutableStateRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *PollMutableStateRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestPollMutableStateResponse_GetNextEventID(t *testing.T) {\n\ttestStruct := PollMutableStateResponse{\n\t\tNextEventID: nextEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetNextEventID()\n\tassert.Equal(t, nextEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *PollMutableStateResponse\n\tres = nilStruct.GetNextEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestPollMutableStateResponse_GetLastFirstEventID(t *testing.T) {\n\ttestStruct := PollMutableStateResponse{\n\t\tLastFirstEventID: nextEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetLastFirstEventID()\n\tassert.Equal(t, nextEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *PollMutableStateResponse\n\tres = nilStruct.GetLastFirstEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestPollMutableStateResponse_GetWorkflowCloseState(t *testing.T) {\n\ttestStruct := PollMutableStateResponse{\n\t\tWorkflowCloseState: &workflowCloseState,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetWorkflowCloseState()\n\tassert.Equal(t, workflowCloseState, res)\n\n\t// Nil struct test\n\tvar nilStruct *PollMutableStateResponse\n\tres = nilStruct.GetWorkflowCloseState()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestProcessingQueueState_GetLevel(t *testing.T) {\n\tlevel := int32(2)\n\ttestStruct := ProcessingQueueState{\n\t\tLevel: &level,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetLevel()\n\tassert.Equal(t, level, res)\n\n\t// Nil struct test\n\tvar nilStruct *ProcessingQueueState\n\tres = nilStruct.GetLevel()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestProcessingQueueState_GetAckLevel(t *testing.T) {\n\tackLevel := int64(1111)\n\ttestStruct := ProcessingQueueState{\n\t\tAckLevel: &ackLevel,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetAckLevel()\n\tassert.Equal(t, ackLevel, res)\n\n\t// Nil struct test\n\tvar nilStruct *ProcessingQueueState\n\tres = nilStruct.GetAckLevel()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestProcessingQueueState_GetMaxLevel(t *testing.T) {\n\tmaxLevel := int64(2222)\n\ttestStruct := ProcessingQueueState{\n\t\tMaxLevel: &maxLevel,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetMaxLevel()\n\tassert.Equal(t, maxLevel, res)\n\n\t// Nil struct test\n\tvar nilStruct *ProcessingQueueState\n\tres = nilStruct.GetMaxLevel()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestProcessingQueueState_GetDomainFilter(t *testing.T) {\n\tdomainFilter := &DomainFilter{}\n\ttestStruct := ProcessingQueueState{\n\t\tDomainFilter: domainFilter,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainFilter()\n\tassert.Equal(t, domainFilter, res)\n\n\t// Nil struct test\n\tvar nilStruct *ProcessingQueueState\n\tres = nilStruct.GetDomainFilter()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryQueryWorkflowRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryQueryWorkflowRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryQueryWorkflowRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryQueryWorkflowRequest_GetRequest(t *testing.T) {\n\trequest := &QueryWorkflowRequest{}\n\ttestStruct := HistoryQueryWorkflowRequest{\n\t\tRequest: request,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetRequest()\n\tassert.Equal(t, request, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryQueryWorkflowRequest\n\tres = nilStruct.GetRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryQueryWorkflowResponse_GetResponse(t *testing.T) {\n\tresponse := &QueryWorkflowResponse{}\n\ttestStruct := HistoryQueryWorkflowResponse{\n\t\tResponse: response,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetResponse()\n\tassert.Equal(t, response, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryQueryWorkflowResponse\n\tres = nilStruct.GetResponse()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryReapplyEventsRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryReapplyEventsRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryReapplyEventsRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryReapplyEventsRequest_GetRequest(t *testing.T) {\n\trequest := &ReapplyEventsRequest{}\n\ttestStruct := HistoryReapplyEventsRequest{\n\t\tRequest: request,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetRequest()\n\tassert.Equal(t, request, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryReapplyEventsRequest\n\tres = nilStruct.GetRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryRecordActivityTaskHeartbeatRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *HistoryRecordActivityTaskHeartbeatRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordActivityTaskStartedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordActivityTaskStartedRequest_GetScheduleID(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedRequest{\n\t\tScheduleID: scheduleID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetScheduleID()\n\tassert.Equal(t, scheduleID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedRequest\n\tres = nilStruct.GetScheduleID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordActivityTaskStartedRequest_GetTaskID(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedRequest{\n\t\tTaskID: taskID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetTaskID()\n\tassert.Equal(t, taskID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedRequest\n\tres = nilStruct.GetTaskID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordActivityTaskStartedRequest_GetRequestID(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedRequest{\n\t\tRequestID: requestID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetRequestID()\n\tassert.Equal(t, requestID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedRequest\n\tres = nilStruct.GetRequestID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordActivityTaskStartedResponse_GetAttempt(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedResponse{\n\t\tAttempt: attempt,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetAttempt()\n\tassert.Equal(t, attempt, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedResponse\n\tres = nilStruct.GetAttempt()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordActivityTaskStartedResponse_GetScheduledTimestampOfThisAttempt(t *testing.T) {\n\ttestStruct := RecordActivityTaskStartedResponse{\n\t\tScheduledTimestampOfThisAttempt: &startedTimestamp,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetScheduledTimestampOfThisAttempt()\n\tassert.Equal(t, startedTimestamp, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordActivityTaskStartedResponse\n\tres = nilStruct.GetScheduledTimestampOfThisAttempt()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordChildExecutionCompletedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := RecordChildExecutionCompletedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordChildExecutionCompletedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordDecisionTaskStartedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := RecordDecisionTaskStartedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordDecisionTaskStartedRequest_GetScheduleID(t *testing.T) {\n\ttestStruct := RecordDecisionTaskStartedRequest{\n\t\tScheduleID: scheduleID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetScheduleID()\n\tassert.Equal(t, scheduleID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedRequest\n\tres = nilStruct.GetScheduleID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordDecisionTaskStartedRequest_GetRequestID(t *testing.T) {\n\ttestStruct := RecordDecisionTaskStartedRequest{\n\t\tRequestID: requestID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetRequestID()\n\tassert.Equal(t, requestID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedRequest\n\tres = nilStruct.GetRequestID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRecordDecisionTaskStartedResponse_GetPreviousStartedEventID(t *testing.T) {\n\ttestStruct := RecordDecisionTaskStartedResponse{\n\t\tPreviousStartedEventID: &previousStartedEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetPreviousStartedEventID()\n\tassert.Equal(t, previousStartedEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedResponse\n\tres = nilStruct.GetPreviousStartedEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordDecisionTaskStartedResponse_GetScheduledEventID(t *testing.T) {\n\tscheduledEventID := int64(1001)\n\ttestStruct := RecordDecisionTaskStartedResponse{\n\t\tScheduledEventID: scheduledEventID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetScheduledEventID()\n\tassert.Equal(t, scheduledEventID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedResponse\n\tres = nilStruct.GetScheduledEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRecordDecisionTaskStartedResponse_GetAttempt(t *testing.T) {\n\ttestStruct := RecordDecisionTaskStartedResponse{\n\t\tAttempt: attempt,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetAttempt()\n\tassert.Equal(t, attempt, res)\n\n\t// Nil struct test\n\tvar nilStruct *RecordDecisionTaskStartedResponse\n\tres = nilStruct.GetAttempt()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestRemoveSignalMutableStateRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := RemoveSignalMutableStateRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RemoveSignalMutableStateRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestRemoveSignalMutableStateRequest_GetRequestID(t *testing.T) {\n\ttestStruct := RemoveSignalMutableStateRequest{\n\t\tRequestID: requestID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetRequestID()\n\tassert.Equal(t, requestID, res)\n\n\t// Nil struct test\n\tvar nilStruct *RemoveSignalMutableStateRequest\n\tres = nilStruct.GetRequestID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestReplicateEventsV2Request_GetDomainUUID(t *testing.T) {\n\ttestStruct := ReplicateEventsV2Request{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\t// Non-nil struct test\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\t// Nil struct test\n\tvar nilStruct *ReplicateEventsV2Request\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRefreshWorkflowTasksRequest_GetRequest(t *testing.T) {\n\t// Define a sample RefreshWorkflowTasksRequest\n\tsampleRequest := &RefreshWorkflowTasksRequest{}\n\n\t// Test when the struct is non-nil and Request is non-nil\n\ttestStruct := HistoryRefreshWorkflowTasksRequest{\n\t\tRequest: sampleRequest,\n\t}\n\n\tres := testStruct.GetRequest()\n\tassert.Equal(t, sampleRequest, res)\n\n\t// Test when the struct is non-nil but Request is nil\n\ttestStructWithNilRequest := HistoryRefreshWorkflowTasksRequest{\n\t\tRequest: nil,\n\t}\n\n\tres = testStructWithNilRequest.GetRequest()\n\tassert.Nil(t, res)\n\n\t// Test when the struct itself is nil\n\tvar nilStruct *HistoryRefreshWorkflowTasksRequest\n\tres = nilStruct.GetRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryRequestCancelWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRequestCancelWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRequestCancelWorkflowExecutionRequest_GetCancelRequest(t *testing.T) {\n\tcancelRequest := &RequestCancelWorkflowExecutionRequest{}\n\ttestStruct := HistoryRequestCancelWorkflowExecutionRequest{\n\t\tCancelRequest: cancelRequest,\n\t}\n\n\tres := testStruct.GetCancelRequest()\n\tassert.Equal(t, cancelRequest, res)\n\n\tvar nilStruct *HistoryRequestCancelWorkflowExecutionRequest\n\tres = nilStruct.GetCancelRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryRequestCancelWorkflowExecutionRequest_GetExternalWorkflowExecution(t *testing.T) {\n\texternalWorkflowExecution := &WorkflowExecution{}\n\ttestStruct := HistoryRequestCancelWorkflowExecutionRequest{\n\t\tExternalWorkflowExecution: externalWorkflowExecution,\n\t}\n\n\tres := testStruct.GetExternalWorkflowExecution()\n\tassert.Equal(t, externalWorkflowExecution, res)\n\n\tvar nilStruct *HistoryRequestCancelWorkflowExecutionRequest\n\tres = nilStruct.GetExternalWorkflowExecution()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryRequestCancelWorkflowExecutionRequest_GetChildWorkflowOnly(t *testing.T) {\n\ttestStruct := HistoryRequestCancelWorkflowExecutionRequest{\n\t\tChildWorkflowOnly: true,\n\t}\n\n\tres := testStruct.GetChildWorkflowOnly()\n\tassert.True(t, res)\n\n\tvar nilStruct *HistoryRequestCancelWorkflowExecutionRequest\n\tres = nilStruct.GetChildWorkflowOnly()\n\tassert.False(t, res)\n}\n\nfunc TestHistoryResetStickyTaskListRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryResetStickyTaskListRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryResetWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryResetWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRespondActivityTaskCanceledRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRespondActivityTaskCanceledRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRespondActivityTaskCompletedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRespondActivityTaskCompletedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRespondActivityTaskFailedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRespondActivityTaskFailedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRespondDecisionTaskCompletedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRespondDecisionTaskCompletedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryRespondDecisionTaskCompletedRequest_GetCompleteRequest(t *testing.T) {\n\tcompleteRequest := &RespondDecisionTaskCompletedRequest{}\n\ttestStruct := HistoryRespondDecisionTaskCompletedRequest{\n\t\tCompleteRequest: completeRequest,\n\t}\n\n\tres := testStruct.GetCompleteRequest()\n\tassert.Equal(t, completeRequest, res)\n\n\tvar nilStruct *HistoryRespondDecisionTaskCompletedRequest\n\tres = nilStruct.GetCompleteRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryRespondDecisionTaskFailedRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryRespondDecisionTaskFailedRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestScheduleDecisionTaskRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := ScheduleDecisionTaskRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *ScheduleDecisionTaskRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestShardOwnershipLostError_GetOwner(t *testing.T) {\n\towner := \"test-owner\"\n\ttestStruct := ShardOwnershipLostError{\n\t\tOwner: owner,\n\t}\n\n\tres := testStruct.GetOwner()\n\tassert.Equal(t, owner, res)\n\n\tvar nilStruct *ShardOwnershipLostError\n\tres = nilStruct.GetOwner()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistorySignalWithStartWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistorySignalWithStartWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistorySignalWithStartWorkflowExecutionRequest_GetPartitionConfig(t *testing.T) {\n\tpartitionConfig := map[string]string{\"partition1\": \"value1\"}\n\ttestStruct := HistorySignalWithStartWorkflowExecutionRequest{\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tres := testStruct.GetPartitionConfig()\n\tassert.Equal(t, partitionConfig, res)\n\n\tvar nilStruct *HistorySignalWithStartWorkflowExecutionRequest\n\tres = nilStruct.GetPartitionConfig()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistorySignalWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistorySignalWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistorySignalWorkflowExecutionRequest_GetChildWorkflowOnly(t *testing.T) {\n\ttestStruct := HistorySignalWorkflowExecutionRequest{\n\t\tChildWorkflowOnly: true,\n\t}\n\n\tres := testStruct.GetChildWorkflowOnly()\n\tassert.True(t, res)\n\n\tvar nilStruct *HistorySignalWorkflowExecutionRequest\n\tres = nilStruct.GetChildWorkflowOnly()\n\tassert.False(t, res)\n}\n\nfunc TestHistoryStartWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryStartWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryStartWorkflowExecutionRequest_GetAttempt(t *testing.T) {\n\ttestStruct := HistoryStartWorkflowExecutionRequest{\n\t\tAttempt: attempt32,\n\t}\n\n\tres := testStruct.GetAttempt()\n\tassert.Equal(t, attempt32, res)\n\n\tvar nilStruct *HistoryStartWorkflowExecutionRequest\n\tres = nilStruct.GetAttempt()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestHistoryStartWorkflowExecutionRequest_GetExpirationTimestamp(t *testing.T) {\n\ttestStruct := HistoryStartWorkflowExecutionRequest{\n\t\tExpirationTimestamp: &expirationTimestamp,\n\t}\n\n\tres := testStruct.GetExpirationTimestamp()\n\tassert.Equal(t, expirationTimestamp, res)\n\n\tvar nilStruct *HistoryStartWorkflowExecutionRequest\n\tres = nilStruct.GetExpirationTimestamp()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestHistoryStartWorkflowExecutionRequest_GetFirstDecisionTaskBackoffSeconds(t *testing.T) {\n\ttestStruct := HistoryStartWorkflowExecutionRequest{\n\t\tFirstDecisionTaskBackoffSeconds: &firstDecisionBackoff,\n\t}\n\n\tres := testStruct.GetFirstDecisionTaskBackoffSeconds()\n\tassert.Equal(t, firstDecisionBackoff, res)\n\n\tvar nilStruct *HistoryStartWorkflowExecutionRequest\n\tres = nilStruct.GetFirstDecisionTaskBackoffSeconds()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestHistoryStartWorkflowExecutionRequest_GetPartitionConfig(t *testing.T) {\n\ttestStruct := HistoryStartWorkflowExecutionRequest{\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tres := testStruct.GetPartitionConfig()\n\tassert.Equal(t, partitionConfig, res)\n\n\tvar nilStruct *HistoryStartWorkflowExecutionRequest\n\tres = nilStruct.GetPartitionConfig()\n\tassert.Nil(t, res)\n}\n\nfunc TestSyncActivityRequest_GetDomainID(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tDomainID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityRequest_GetScheduledTime(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tScheduledTime: &scheduledTime,\n\t}\n\n\tres := testStruct.GetScheduledTime()\n\tassert.Equal(t, scheduledTime, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetScheduledTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetStartedTime(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tStartedTime: &startedTime,\n\t}\n\n\tres := testStruct.GetStartedTime()\n\tassert.Equal(t, startedTime, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetStartedTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetLastHeartbeatTime(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tLastHeartbeatTime: &lastHeartbeatTime,\n\t}\n\n\tres := testStruct.GetLastHeartbeatTime()\n\tassert.Equal(t, lastHeartbeatTime, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetLastHeartbeatTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetDetails(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tDetails: details,\n\t}\n\n\tres := testStruct.GetDetails()\n\tassert.Equal(t, details, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetDetails()\n\tassert.Nil(t, res)\n}\n\nfunc TestSyncActivityRequest_GetLastFailureReason(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tLastFailureReason: &lastFailureReason,\n\t}\n\n\tres := testStruct.GetLastFailureReason()\n\tassert.Equal(t, lastFailureReason, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetLastFailureReason()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityRequest_GetLastWorkerIdentity(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tLastWorkerIdentity: lastWorkerIdentity,\n\t}\n\n\tres := testStruct.GetLastWorkerIdentity()\n\tassert.Equal(t, lastWorkerIdentity, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetLastWorkerIdentity()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityRequest_GetLastFailureDetails(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tLastFailureDetails: lastFailureDetails,\n\t}\n\n\tres := testStruct.GetLastFailureDetails()\n\tassert.Equal(t, lastFailureDetails, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetLastFailureDetails()\n\tassert.Nil(t, res)\n}\n\nfunc TestSyncActivityRequest_GetAttempt(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tAttempt: attempt32,\n\t}\n\n\tres := testStruct.GetAttempt()\n\tassert.Equal(t, attempt32, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetAttempt()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestSyncActivityRequest_GetWorkflowID(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tWorkflowID: workflowID,\n\t}\n\n\tres := testStruct.GetWorkflowID()\n\tassert.Equal(t, workflowID, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetWorkflowID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityRequest_GetRunID(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tRunID: runID,\n\t}\n\n\tres := testStruct.GetRunID()\n\tassert.Equal(t, runID, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetRunID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityRequest_GetVersion(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tVersion: version,\n\t}\n\n\tres := testStruct.GetVersion()\n\tassert.Equal(t, version, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetScheduledID(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tScheduledID: scheduledID,\n\t}\n\n\tres := testStruct.GetScheduledID()\n\tassert.Equal(t, scheduledID, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetScheduledID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetStartedID(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tStartedID: startedID,\n\t}\n\n\tres := testStruct.GetStartedID()\n\tassert.Equal(t, startedID, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetStartedID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityRequest_GetVersionHistory(t *testing.T) {\n\ttestStruct := SyncActivityRequest{\n\t\tVersionHistory: versionHistory,\n\t}\n\n\tres := testStruct.GetVersionHistory()\n\tassert.Equal(t, versionHistory, res)\n\n\tvar nilStruct *SyncActivityRequest\n\tres = nilStruct.GetVersionHistory()\n\tassert.Nil(t, res)\n}\n\nfunc TestSyncShardStatusRequest_GetSourceCluster(t *testing.T) {\n\ttestStruct := SyncShardStatusRequest{\n\t\tSourceCluster: sourceCluster,\n\t}\n\n\tres := testStruct.GetSourceCluster()\n\tassert.Equal(t, sourceCluster, res)\n\n\tvar nilStruct *SyncShardStatusRequest\n\tres = nilStruct.GetSourceCluster()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncShardStatusRequest_GetShardID(t *testing.T) {\n\ttestStruct := SyncShardStatusRequest{\n\t\tShardID: shardID,\n\t}\n\n\tres := testStruct.GetShardID()\n\tassert.Equal(t, shardID, res)\n\n\tvar nilStruct *SyncShardStatusRequest\n\tres = nilStruct.GetShardID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncShardStatusRequest_GetTimestamp(t *testing.T) {\n\ttestStruct := SyncShardStatusRequest{\n\t\tTimestamp: &startedTimestamp,\n\t}\n\n\tres := testStruct.GetTimestamp()\n\tassert.Equal(t, startedTimestamp, res)\n\n\tvar nilStruct *SyncShardStatusRequest\n\tres = nilStruct.GetTimestamp()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestHistoryTerminateWorkflowExecutionRequest_GetDomainUUID(t *testing.T) {\n\ttestStruct := HistoryTerminateWorkflowExecutionRequest{\n\t\tDomainUUID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainUUID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *HistoryTerminateWorkflowExecutionRequest\n\tres = nilStruct.GetDomainUUID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryTerminateWorkflowExecutionRequest_GetTerminateRequest(t *testing.T) {\n\ttestStruct := HistoryTerminateWorkflowExecutionRequest{\n\t\tTerminateRequest: terminateRequest,\n\t}\n\n\tres := testStruct.GetTerminateRequest()\n\tassert.Equal(t, terminateRequest, res)\n\n\tvar nilStruct *HistoryTerminateWorkflowExecutionRequest\n\tres = nilStruct.GetTerminateRequest()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryTerminateWorkflowExecutionRequest_GetExternalWorkflowExecution(t *testing.T) {\n\ttestStruct := HistoryTerminateWorkflowExecutionRequest{\n\t\tExternalWorkflowExecution: externalWorkflowExecution,\n\t}\n\n\tres := testStruct.GetExternalWorkflowExecution()\n\tassert.Equal(t, externalWorkflowExecution, res)\n\n\tvar nilStruct *HistoryTerminateWorkflowExecutionRequest\n\tres = nilStruct.GetExternalWorkflowExecution()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryTerminateWorkflowExecutionRequest_GetChildWorkflowOnly(t *testing.T) {\n\ttestStruct := HistoryTerminateWorkflowExecutionRequest{\n\t\tChildWorkflowOnly: childWorkflowOnly,\n\t}\n\n\tres := testStruct.GetChildWorkflowOnly()\n\tassert.Equal(t, childWorkflowOnly, res)\n\n\tvar nilStruct *HistoryTerminateWorkflowExecutionRequest\n\tres = nilStruct.GetChildWorkflowOnly()\n\tassert.False(t, res)\n}\n\nfunc TestGetFailoverInfoRequest_GetDomainID(t *testing.T) {\n\ttestStruct := GetFailoverInfoRequest{\n\t\tDomainID: domainUUID,\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, domainUUID, res)\n\n\tvar nilStruct *GetFailoverInfoRequest\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetFailoverInfoResponse_GetCompletedShardCount(t *testing.T) {\n\ttestStruct := GetFailoverInfoResponse{\n\t\tCompletedShardCount: completedShardCount,\n\t}\n\n\tres := testStruct.GetCompletedShardCount()\n\tassert.Equal(t, completedShardCount, res)\n\n\tvar nilStruct *GetFailoverInfoResponse\n\tres = nilStruct.GetCompletedShardCount()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestGetFailoverInfoResponse_GetPendingShards(t *testing.T) {\n\ttestStruct := GetFailoverInfoResponse{\n\t\tPendingShards: pendingShards,\n\t}\n\n\tres := testStruct.GetPendingShards()\n\tassert.Equal(t, pendingShards, res)\n\n\tvar nilStruct *GetFailoverInfoResponse\n\tres = nilStruct.GetPendingShards()\n\tassert.Nil(t, res)\n}\n\nfunc TestQueueState_Copy(t *testing.T) {\n\tf := fuzz.New().NilChance(0.1)\n\tinfo := &QueueState{}\n\tfor i := 0; i < 1000; i++ {\n\t\tf.Fuzz(info)\n\t\tinfoCopy := info.Copy()\n\t\tassert.Equal(t, info, infoCopy)\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/errorutils/convert.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errorutils\n\nimport \"errors\"\n\n// ConvertError checks if an error is of type T and if so, converts it using f.\nfunc ConvertError[T, V error](err error, fn func(T) V) (bool, V) {\n\tvar (\n\t\te   T\n\t\tres V\n\t)\n\tif !errors.As(err, &e) {\n\t\treturn false, res\n\t}\n\treturn true, fn(e)\n}\n"
  },
  {
    "path": "common/types/mapper/errorutils/convert_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage errorutils\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestConvertError(t *testing.T) {\n\tt.Run(\"sample error\", func(t *testing.T) {\n\t\terr := &sampleError{\n\t\t\tmessage: \"test\",\n\t\t}\n\t\tisError, converted := ConvertError(err, sampleErrorConvertor)\n\t\tassert.True(t, isError, \"is error\")\n\t\tassert.Error(t, converted, \"converted error\")\n\t\tassert.Equal(t, err.message, converted.message)\n\t})\n\tt.Run(\"nil error is propagated as nil\", func(t *testing.T) {\n\t\tisError, converted := ConvertError(nil, sampleErrorConvertor)\n\t\tassert.False(t, isError, \"is error\")\n\t\tassert.Nil(t, converted, \"converted error\")\n\t})\n}\n\ntype sampleError struct {\n\tmessage string\n}\n\nfunc (s *sampleError) Error() string {\n\treturn \"sample error\"\n}\n\ntype convertedError struct {\n\tmessage string\n}\n\nfunc (c *convertedError) Error() string {\n\treturn \"converted error\"\n}\n\nfunc sampleErrorConvertor(e *sampleError) *convertedError {\n\treturn &convertedError{\n\t\tmessage: e.message,\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/admin.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"sort\"\n\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromAdminAddSearchAttributeRequest(t *types.AddSearchAttributeRequest) *adminv1.AddSearchAttributeRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.AddSearchAttributeRequest{\n\t\tSearchAttribute: FromIndexedValueTypeMap(t.SearchAttribute),\n\t\tSecurityToken:   t.SecurityToken,\n\t}\n}\n\nfunc ToAdminAddSearchAttributeRequest(t *adminv1.AddSearchAttributeRequest) *types.AddSearchAttributeRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddSearchAttributeRequest{\n\t\tSearchAttribute: ToIndexedValueTypeMap(t.SearchAttribute),\n\t\tSecurityToken:   t.SecurityToken,\n\t}\n}\n\nfunc FromAdminCloseShardRequest(t *types.CloseShardRequest) *adminv1.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CloseShardRequest{\n\t\tShardId: t.ShardID,\n\t}\n}\n\nfunc ToAdminCloseShardRequest(t *adminv1.CloseShardRequest) *types.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CloseShardRequest{\n\t\tShardID: t.ShardId,\n\t}\n}\n\nfunc FromAdminDescribeClusterResponse(t *types.DescribeClusterResponse) *adminv1.DescribeClusterResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeClusterResponse{\n\t\tSupportedClientVersions: FromSupportedClientVersions(t.SupportedClientVersions),\n\t\tMembershipInfo:          FromMembershipInfo(t.MembershipInfo),\n\t\tPersistenceInfo:         FromPersistenceInfoMap(t.PersistenceInfo),\n\t}\n}\n\nfunc ToAdminDescribeClusterResponse(t *adminv1.DescribeClusterResponse) *types.DescribeClusterResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeClusterResponse{\n\t\tSupportedClientVersions: ToSupportedClientVersions(t.SupportedClientVersions),\n\t\tMembershipInfo:          ToMembershipInfo(t.MembershipInfo),\n\t\tPersistenceInfo:         ToPersistenceInfoMap(t.PersistenceInfo),\n\t}\n}\n\nfunc FromAdminDescribeShardDistributionRequest(t *types.DescribeShardDistributionRequest) *adminv1.DescribeShardDistributionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeShardDistributionRequest{\n\t\tPageSize: t.PageSize,\n\t\tPageId:   t.PageID,\n\t}\n}\n\nfunc FromAdminDescribeHistoryHostRequest(t *types.DescribeHistoryHostRequest) *adminv1.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.HostAddress != nil {\n\t\treturn &adminv1.DescribeHistoryHostRequest{\n\t\t\tDescribeBy: &adminv1.DescribeHistoryHostRequest_HostAddress{HostAddress: *t.HostAddress},\n\t\t}\n\t}\n\tif t.ShardIDForHost != nil {\n\t\treturn &adminv1.DescribeHistoryHostRequest{\n\t\t\tDescribeBy: &adminv1.DescribeHistoryHostRequest_ShardId{ShardId: *t.ShardIDForHost},\n\t\t}\n\t}\n\tif t.ExecutionForHost != nil {\n\t\treturn &adminv1.DescribeHistoryHostRequest{\n\t\t\tDescribeBy: &adminv1.DescribeHistoryHostRequest_WorkflowExecution{WorkflowExecution: FromWorkflowExecution(t.ExecutionForHost)},\n\t\t}\n\t}\n\tpanic(\"neither oneof field is set for DescribeHistoryHostRequest\")\n}\n\nfunc ToAdminDescribeShardDistributionRequest(t *adminv1.DescribeShardDistributionRequest) *types.DescribeShardDistributionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeShardDistributionRequest{\n\t\tPageSize: t.PageSize,\n\t\tPageID:   t.PageId,\n\t}\n}\n\nfunc ToAdminDescribeHistoryHostRequest(t *adminv1.DescribeHistoryHostRequest) *types.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch describeBy := t.DescribeBy.(type) {\n\tcase *adminv1.DescribeHistoryHostRequest_HostAddress:\n\t\treturn &types.DescribeHistoryHostRequest{HostAddress: &describeBy.HostAddress}\n\tcase *adminv1.DescribeHistoryHostRequest_ShardId:\n\t\treturn &types.DescribeHistoryHostRequest{ShardIDForHost: &describeBy.ShardId}\n\tcase *adminv1.DescribeHistoryHostRequest_WorkflowExecution:\n\t\treturn &types.DescribeHistoryHostRequest{ExecutionForHost: ToWorkflowExecution(describeBy.WorkflowExecution)}\n\t}\n\tpanic(\"neither oneof field is set for DescribeHistoryHostRequest\")\n}\n\nfunc FromAdminDescribeShardDistributionResponse(t *types.DescribeShardDistributionResponse) *adminv1.DescribeShardDistributionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeShardDistributionResponse{\n\t\tNumberOfShards: t.NumberOfShards,\n\t\tShards:         t.Shards,\n\t}\n}\n\nfunc FromAdminDescribeHistoryHostResponse(t *types.DescribeHistoryHostResponse) *adminv1.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        t.NumberOfShards,\n\t\tShardIds:              t.ShardIDs,\n\t\tDomainCache:           FromDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: t.ShardControllerStatus,\n\t\tAddress:               t.Address,\n\t}\n}\n\nfunc ToAdminDescribeShardDistributionResponse(t *adminv1.DescribeShardDistributionResponse) *types.DescribeShardDistributionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeShardDistributionResponse{\n\t\tNumberOfShards: t.NumberOfShards,\n\t\tShards:         t.Shards,\n\t}\n}\n\nfunc ToAdminDescribeHistoryHostResponse(t *adminv1.DescribeHistoryHostResponse) *types.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        t.NumberOfShards,\n\t\tShardIDs:              t.ShardIds,\n\t\tDomainCache:           ToDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: t.ShardControllerStatus,\n\t\tAddress:               t.Address,\n\t}\n}\n\nfunc FromAdminDescribeQueueRequest(t *types.DescribeQueueRequest) *adminv1.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeQueueRequest{\n\t\tShardId:     t.ShardID,\n\t\tClusterName: t.ClusterName,\n\t\tTaskType:    FromTaskType(t.Type),\n\t}\n}\n\nfunc ToAdminDescribeQueueRequest(t *adminv1.DescribeQueueRequest) *types.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueRequest{\n\t\tShardID:     t.ShardId,\n\t\tClusterName: t.ClusterName,\n\t\tType:        ToTaskType(t.TaskType),\n\t}\n}\n\nfunc FromAdminDescribeQueueResponse(t *types.DescribeQueueResponse) *adminv1.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\nfunc ToAdminDescribeQueueResponse(t *adminv1.DescribeQueueResponse) *types.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\nfunc FromAdminDescribeWorkflowExecutionRequest(t *types.AdminDescribeWorkflowExecutionRequest) *adminv1.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeWorkflowExecutionRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\nfunc ToAdminDescribeWorkflowExecutionRequest(t *adminv1.DescribeWorkflowExecutionRequest) *types.AdminDescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDescribeWorkflowExecutionRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromAdminDescribeWorkflowExecutionResponse(t *types.AdminDescribeWorkflowExecutionResponse) *adminv1.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DescribeWorkflowExecutionResponse{\n\t\tShardId:                stringToInt32(t.ShardID),\n\t\tHistoryAddr:            t.HistoryAddr,\n\t\tMutableStateInCache:    t.MutableStateInCache,\n\t\tMutableStateInDatabase: t.MutableStateInDatabase,\n\t}\n}\n\nfunc ToAdminDescribeWorkflowExecutionResponse(t *adminv1.DescribeWorkflowExecutionResponse) *types.AdminDescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDescribeWorkflowExecutionResponse{\n\t\tShardID:                int32ToString(t.ShardId),\n\t\tHistoryAddr:            t.HistoryAddr,\n\t\tMutableStateInCache:    t.MutableStateInCache,\n\t\tMutableStateInDatabase: t.MutableStateInDatabase,\n\t}\n}\n\nfunc FromAdminGetDLQReplicationMessagesRequest(t *types.GetDLQReplicationMessagesRequest) *adminv1.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: FromReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\nfunc ToAdminGetDLQReplicationMessagesRequest(t *adminv1.GetDLQReplicationMessagesRequest) *types.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: ToReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\nfunc FromAdminGetDLQReplicationMessagesResponse(t *types.GetDLQReplicationMessagesResponse) *adminv1.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: FromReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\nfunc ToAdminGetDLQReplicationMessagesResponse(t *adminv1.GetDLQReplicationMessagesResponse) *types.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: ToReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\nfunc FromAdminGetDomainReplicationMessagesRequest(t *types.GetDomainReplicationMessagesRequest) *adminv1.GetDomainReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageId: fromInt64Value(t.LastRetrievedMessageID),\n\t\tLastProcessedMessageId: fromInt64Value(t.LastProcessedMessageID),\n\t\tClusterName:            t.ClusterName,\n\t}\n}\n\nfunc ToAdminGetDomainReplicationMessagesRequest(t *adminv1.GetDomainReplicationMessagesRequest) *types.GetDomainReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: toInt64Value(t.LastRetrievedMessageId),\n\t\tLastProcessedMessageID: toInt64Value(t.LastProcessedMessageId),\n\t\tClusterName:            t.ClusterName,\n\t}\n}\n\nfunc FromAdminGetDomainReplicationMessagesResponse(t *types.GetDomainReplicationMessagesResponse) *adminv1.GetDomainReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDomainReplicationMessagesResponse{\n\t\tMessages: FromReplicationMessages(t.Messages),\n\t}\n}\n\nfunc ToAdminGetDomainReplicationMessagesResponse(t *adminv1.GetDomainReplicationMessagesResponse) *types.GetDomainReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: ToReplicationMessages(t.Messages),\n\t}\n}\n\nfunc FromAdminGetReplicationMessagesRequest(t *types.GetReplicationMessagesRequest) *adminv1.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetReplicationMessagesRequest{\n\t\tTokens:      FromReplicationTokenArray(t.Tokens),\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc ToAdminGetReplicationMessagesRequest(t *adminv1.GetReplicationMessagesRequest) *types.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesRequest{\n\t\tTokens:      ToReplicationTokenArray(t.Tokens),\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc FromAdminGetReplicationMessagesResponse(t *types.GetReplicationMessagesResponse) *adminv1.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetReplicationMessagesResponse{\n\t\tShardMessages: FromReplicationMessagesMap(t.MessagesByShard),\n\t}\n}\n\nfunc ToAdminGetReplicationMessagesResponse(t *adminv1.GetReplicationMessagesResponse) *types.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: ToReplicationMessagesMap(t.ShardMessages),\n\t}\n}\n\nfunc FromAdminGetWorkflowExecutionRawHistoryV2Request(t *types.GetWorkflowExecutionRawHistoryV2Request) *adminv1.GetWorkflowExecutionRawHistoryV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t\tStartEvent:        FromEventIDVersionPair(t.StartEventID, t.StartEventVersion),\n\t\tEndEvent:          FromEventIDVersionPair(t.EndEventID, t.EndEventVersion),\n\t\tPageSize:          t.MaximumPageSize,\n\t\tNextPageToken:     t.NextPageToken,\n\t}\n}\n\nfunc ToAdminGetWorkflowExecutionRawHistoryV2Request(t *adminv1.GetWorkflowExecutionRawHistoryV2Request) *types.GetWorkflowExecutionRawHistoryV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain:            t.Domain,\n\t\tExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tStartEventID:      ToEventID(t.StartEvent),\n\t\tStartEventVersion: ToEventVersion(t.StartEvent),\n\t\tEndEventID:        ToEventID(t.EndEvent),\n\t\tEndEventVersion:   ToEventVersion(t.EndEvent),\n\t\tMaximumPageSize:   t.PageSize,\n\t\tNextPageToken:     t.NextPageToken,\n\t}\n}\n\nfunc FromAdminGetWorkflowExecutionRawHistoryV2Response(t *types.GetWorkflowExecutionRawHistoryV2Response) *adminv1.GetWorkflowExecutionRawHistoryV2Response {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetWorkflowExecutionRawHistoryV2Response{\n\t\tNextPageToken:  t.NextPageToken,\n\t\tHistoryBatches: FromDataBlobArray(t.HistoryBatches),\n\t\tVersionHistory: FromVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc ToAdminGetWorkflowExecutionRawHistoryV2Response(t *adminv1.GetWorkflowExecutionRawHistoryV2Response) *types.GetWorkflowExecutionRawHistoryV2Response {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tNextPageToken:  t.NextPageToken,\n\t\tHistoryBatches: ToDataBlobArray(t.HistoryBatches),\n\t\tVersionHistory: ToVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc FromAdminCountDLQMessagesRequest(t *types.CountDLQMessagesRequest) *adminv1.CountDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CountDLQMessagesRequest{\n\t\tForceFetch: t.ForceFetch,\n\t}\n}\n\nfunc ToAdminCountDLQMessagesRequest(t *adminv1.CountDLQMessagesRequest) *types.CountDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountDLQMessagesRequest{\n\t\tForceFetch: t.ForceFetch,\n\t}\n}\n\nfunc FromAdminCountDLQMessagesResponse(t *types.CountDLQMessagesResponse) *adminv1.CountDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CountDLQMessagesResponse{\n\t\tHistory: FromHistoryDLQCountEntryMap(t.History),\n\t\tDomain:  t.Domain,\n\t}\n}\n\nfunc ToAdminCountDLQMessagesResponse(t *adminv1.CountDLQMessagesResponse) *types.CountDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountDLQMessagesResponse{\n\t\tHistory: ToHistoryDLQCountEntryMap(t.History),\n\t\tDomain:  t.Domain,\n\t}\n}\n\nfunc FromAdminMergeDLQMessagesRequest(t *types.MergeDLQMessagesRequest) *adminv1.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.MergeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t\tPageSize:              t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc ToAdminMergeDLQMessagesRequest(t *adminv1.MergeDLQMessagesRequest) *types.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t\tMaximumPageSize:       t.PageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc FromAdminMergeDLQMessagesResponse(t *types.MergeDLQMessagesResponse) *adminv1.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToAdminMergeDLQMessagesResponse(t *adminv1.MergeDLQMessagesResponse) *types.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromAdminPurgeDLQMessagesRequest(t *types.PurgeDLQMessagesRequest) *adminv1.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.PurgeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t}\n}\n\nfunc ToAdminPurgeDLQMessagesRequest(t *adminv1.PurgeDLQMessagesRequest) *types.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PurgeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t}\n}\n\nfunc FromAdminReadDLQMessagesRequest(t *types.ReadDLQMessagesRequest) *adminv1.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReadDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t\tPageSize:              t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc ToAdminReadDLQMessagesRequest(t *adminv1.ReadDLQMessagesRequest) *types.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t\tMaximumPageSize:       t.PageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc FromAdminReadDLQMessagesResponse(t *types.ReadDLQMessagesResponse) *adminv1.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReadDLQMessagesResponse{\n\t\tType:                 FromDLQType(t.Type),\n\t\tReplicationTasks:     FromReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: FromReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\nfunc ToAdminReadDLQMessagesResponse(t *adminv1.ReadDLQMessagesResponse) *types.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesResponse{\n\t\tType:                 ToDLQType(t.Type),\n\t\tReplicationTasks:     ToReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: ToReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\nfunc FromAdminReapplyEventsRequest(t *types.ReapplyEventsRequest) *adminv1.ReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReapplyEventsRequest{\n\t\tDomain:            t.DomainName,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tEvents:            FromDataBlob(t.Events),\n\t}\n}\n\nfunc ToAdminReapplyEventsRequest(t *adminv1.ReapplyEventsRequest) *types.ReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReapplyEventsRequest{\n\t\tDomainName:        t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tEvents:            ToDataBlob(t.Events),\n\t}\n}\n\nfunc FromAdminRefreshWorkflowTasksRequest(t *types.RefreshWorkflowTasksRequest) *adminv1.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RefreshWorkflowTasksRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\nfunc ToAdminRefreshWorkflowTasksRequest(t *adminv1.RefreshWorkflowTasksRequest) *types.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RefreshWorkflowTasksRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromAdminRemoveTaskRequest(t *types.RemoveTaskRequest) *adminv1.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RemoveTaskRequest{\n\t\tShardId:        t.ShardID,\n\t\tTaskType:       FromTaskType(t.Type),\n\t\tTaskId:         t.TaskID,\n\t\tVisibilityTime: unixNanoToTime(t.VisibilityTimestamp),\n\t\tClusterName:    t.ClusterName,\n\t}\n}\n\nfunc ToAdminRemoveTaskRequest(t *adminv1.RemoveTaskRequest) *types.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoveTaskRequest{\n\t\tShardID:             t.ShardId,\n\t\tType:                ToTaskType(t.TaskType),\n\t\tTaskID:              t.TaskId,\n\t\tVisibilityTimestamp: timeToUnixNano(t.VisibilityTime),\n\t\tClusterName:         t.ClusterName,\n\t}\n}\n\nfunc FromAdminResendReplicationTasksRequest(t *types.ResendReplicationTasksRequest) *adminv1.ResendReplicationTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ResendReplicationTasksRequest{\n\t\tDomainId:          t.DomainID,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tRemoteCluster:     t.RemoteCluster,\n\t\tStartEvent:        FromEventIDVersionPair(t.StartEventID, t.StartVersion),\n\t\tEndEvent:          FromEventIDVersionPair(t.EndEventID, t.EndVersion),\n\t}\n}\n\nfunc ToAdminResendReplicationTasksRequest(t *adminv1.ResendReplicationTasksRequest) *types.ResendReplicationTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResendReplicationTasksRequest{\n\t\tDomainID:      t.DomainId,\n\t\tWorkflowID:    ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:         ToRunID(t.WorkflowExecution),\n\t\tRemoteCluster: t.RemoteCluster,\n\t\tStartEventID:  ToEventID(t.StartEvent),\n\t\tStartVersion:  ToEventVersion(t.StartEvent),\n\t\tEndEventID:    ToEventID(t.EndEvent),\n\t\tEndVersion:    ToEventVersion(t.EndEvent),\n\t}\n}\n\nfunc FromAdminResetQueueRequest(t *types.ResetQueueRequest) *adminv1.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ResetQueueRequest{\n\t\tShardId:     t.ShardID,\n\t\tClusterName: t.ClusterName,\n\t\tTaskType:    FromTaskType(t.Type),\n\t}\n}\n\nfunc ToAdminResetQueueRequest(t *adminv1.ResetQueueRequest) *types.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetQueueRequest{\n\t\tShardID:     t.ShardId,\n\t\tClusterName: t.ClusterName,\n\t\tType:        ToTaskType(t.TaskType),\n\t}\n}\n\n// FromAdminGetCrossClusterTasksRequest converts internal GetCrossClusterTasksRequest type to proto\nfunc FromAdminGetCrossClusterTasksRequest(t *types.GetCrossClusterTasksRequest) *adminv1.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetCrossClusterTasksRequest{\n\t\tShardIds:      t.ShardIDs,\n\t\tTargetCluster: t.TargetCluster,\n\t}\n}\n\n// ToAdminGetCrossClusterTasksRequest converts proto GetCrossClusterTasksRequest type to internal\nfunc ToAdminGetCrossClusterTasksRequest(t *adminv1.GetCrossClusterTasksRequest) *types.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksRequest{\n\t\tShardIDs:      t.ShardIds,\n\t\tTargetCluster: t.TargetCluster,\n\t}\n}\n\n// FromAdminGetCrossClusterTasksResponse converts internal GetCrossClusterTasksResponse type to proto\nfunc FromAdminGetCrossClusterTasksResponse(t *types.GetCrossClusterTasksResponse) *adminv1.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       FromCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: FromGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// ToAdminGetCrossClusterTasksResponse converts proto GetCrossClusterTasksResponse type to internal\nfunc ToAdminGetCrossClusterTasksResponse(t *adminv1.GetCrossClusterTasksResponse) *types.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       ToCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: ToGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// FromAdminRespondCrossClusterTasksCompletedRequest converts internal RespondCrossClusterTasksCompletedRequest type to thrift\nfunc FromAdminRespondCrossClusterTasksCompletedRequest(t *types.RespondCrossClusterTasksCompletedRequest) *adminv1.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RespondCrossClusterTasksCompletedRequest{\n\t\tShardId:       t.ShardID,\n\t\tTargetCluster: t.TargetCluster,\n\t\tTaskResponses: FromCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: t.FetchNewTasks,\n\t}\n}\n\n// ToAdminRespondCrossClusterTasksCompletedRequest converts thrift RespondCrossClusterTasksCompletedRequest type to internal\nfunc ToAdminRespondCrossClusterTasksCompletedRequest(t *adminv1.RespondCrossClusterTasksCompletedRequest) *types.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedRequest{\n\t\tShardID:       t.ShardId,\n\t\tTargetCluster: t.TargetCluster,\n\t\tTaskResponses: ToCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: t.FetchNewTasks,\n\t}\n}\n\n// FromAdminRespondCrossClusterTasksCompletedResponse converts internal RespondCrossClusterTasksCompletedResponse type to thrift\nfunc FromAdminRespondCrossClusterTasksCompletedResponse(t *types.RespondCrossClusterTasksCompletedResponse) *adminv1.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: FromCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\n// ToAdminRespondCrossClusterTasksCompletedResponse converts thrift RespondCrossClusterTasksCompletedResponse type to internal\nfunc ToAdminRespondCrossClusterTasksCompletedResponse(t *adminv1.RespondCrossClusterTasksCompletedResponse) *types.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: ToCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\n// FromAdminGetDynamicConfigRequest converts internal GetDynamicConfigRequest type to proto\nfunc FromAdminGetDynamicConfigRequest(t *types.GetDynamicConfigRequest) *adminv1.GetDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t\tFilters:    FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToAdminGetDynamicConfigRequest converts proto GetDynamicConfigRequest type to internal\nfunc ToAdminGetDynamicConfigRequest(t *adminv1.GetDynamicConfigRequest) *types.GetDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t\tFilters:    ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromAdminGetDynamicConfigResponse converts internal GetDynamicConfigResponse type to proto\nfunc FromAdminGetDynamicConfigResponse(t *types.GetDynamicConfigResponse) *adminv1.GetDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDynamicConfigResponse{\n\t\tValue: FromDataBlob(t.Value),\n\t}\n}\n\n// ToAdminGetDynamicConfigResponse converts proto GetDynamicConfigResponse type to internal\nfunc ToAdminGetDynamicConfigResponse(t *adminv1.GetDynamicConfigResponse) *types.GetDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDynamicConfigResponse{\n\t\tValue: ToDataBlob(t.Value),\n\t}\n}\n\n// FromAdminUpdateDynamicConfigRequest converts internal UpdateDynamicConfigRequest type to proto\nfunc FromAdminUpdateDynamicConfigRequest(t *types.UpdateDynamicConfigRequest) *adminv1.UpdateDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateDynamicConfigRequest{\n\t\tConfigName:   t.ConfigName,\n\t\tConfigValues: FromDynamicConfigValueArray(t.ConfigValues),\n\t}\n}\n\n// ToAdminUpdateDynamicConfigRequest converts proto UpdateDynamicConfigRequest type to internal\nfunc ToAdminUpdateDynamicConfigRequest(t *adminv1.UpdateDynamicConfigRequest) *types.UpdateDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDynamicConfigRequest{\n\t\tConfigName:   t.ConfigName,\n\t\tConfigValues: ToDynamicConfigValueArray(t.ConfigValues),\n\t}\n}\n\n// FromAdminRestoreDynamicConfigRequest converts internal RestoreDynamicConfigRequest type to proto\nfunc FromAdminRestoreDynamicConfigRequest(t *types.RestoreDynamicConfigRequest) *adminv1.RestoreDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RestoreDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t\tFilters:    FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToAdminRestoreDynamicConfigRequest converts proto RestoreDynamicConfigRequest type to internal\nfunc ToAdminRestoreDynamicConfigRequest(t *adminv1.RestoreDynamicConfigRequest) *types.RestoreDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestoreDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t\tFilters:    ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromAdminDeleteWorkflowRequest converts internal AdminDeleteWorkflowRequest type to proto\nfunc FromAdminDeleteWorkflowRequest(t *types.AdminDeleteWorkflowRequest) *adminv1.DeleteWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DeleteWorkflowRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToAdminDeleteWorkflowRequest converts proto AdminDeleteWorkflowRequest type to internal\nfunc ToAdminDeleteWorkflowRequest(t *adminv1.DeleteWorkflowRequest) *types.AdminDeleteWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDeleteWorkflowRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\n// FromAdminDeleteWorkflowResponse converts internal AdminDeleteWorkflowRequest type to proto\nfunc FromAdminDeleteWorkflowResponse(t *types.AdminDeleteWorkflowResponse) *adminv1.DeleteWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DeleteWorkflowResponse{\n\t\tHistoryDeleted:    t.HistoryDeleted,\n\t\tExecutionsDeleted: t.ExecutionsDeleted,\n\t\tVisibilityDeleted: t.VisibilityDeleted,\n\t}\n}\n\n// ToAdminDeleteWorkflowResponse converts proto AdminDeleteWorkflowResponse type to internal\nfunc ToAdminDeleteWorkflowResponse(t *adminv1.DeleteWorkflowResponse) *types.AdminDeleteWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDeleteWorkflowResponse{\n\t\tHistoryDeleted:    t.HistoryDeleted,\n\t\tExecutionsDeleted: t.ExecutionsDeleted,\n\t\tVisibilityDeleted: t.VisibilityDeleted,\n\t}\n}\n\n// FromAdminMaintainCorruptWorkflowRequest converts internal AdminMaintainWorkflowRequest type to proto\nfunc FromAdminMaintainCorruptWorkflowRequest(t *types.AdminMaintainWorkflowRequest) *adminv1.MaintainCorruptWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.MaintainCorruptWorkflowRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToAdminMaintainCorruptWorkflowRequest converts proto AdminMaintainWorkflowRequest type to internal\nfunc ToAdminMaintainCorruptWorkflowRequest(t *adminv1.MaintainCorruptWorkflowRequest) *types.AdminMaintainWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminMaintainWorkflowRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\n// FromAdminMaintainCorruptWorkflowResponse converts internal AdminMaintainWorkflowResponse type to proto\nfunc FromAdminMaintainCorruptWorkflowResponse(t *types.AdminMaintainWorkflowResponse) *adminv1.MaintainCorruptWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.MaintainCorruptWorkflowResponse{\n\t\tHistoryDeleted:    t.HistoryDeleted,\n\t\tExecutionsDeleted: t.ExecutionsDeleted,\n\t\tVisibilityDeleted: t.VisibilityDeleted,\n\t}\n}\n\n// ToAdminMaintainCorruptWorkflowResponse converts proto AdminMaintainWorkflowResponse type to internal\nfunc ToAdminMaintainCorruptWorkflowResponse(t *adminv1.MaintainCorruptWorkflowResponse) *types.AdminMaintainWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminMaintainWorkflowResponse{\n\t\tHistoryDeleted:    t.HistoryDeleted,\n\t\tExecutionsDeleted: t.ExecutionsDeleted,\n\t\tVisibilityDeleted: t.VisibilityDeleted,\n\t}\n}\n\n// FromAdminListDynamicConfigRequest converts internal ListDynamicConfigRequest type to proto\nfunc FromAdminListDynamicConfigRequest(t *types.ListDynamicConfigRequest) *adminv1.ListDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ListDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t}\n}\n\n// ToAdminListDynamicConfigRequest converts proto ListDynamicConfigRequest type to internal\nfunc ToAdminListDynamicConfigRequest(t *adminv1.ListDynamicConfigRequest) *types.ListDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDynamicConfigRequest{\n\t\tConfigName: t.ConfigName,\n\t}\n}\n\n// FromAdminListDynamicConfigResponse converts internal ListDynamicConfigResponse type to proto\nfunc FromAdminListDynamicConfigResponse(t *types.ListDynamicConfigResponse) *adminv1.ListDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ListDynamicConfigResponse{\n\t\tEntries: FromDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\n// ToAdminListDynamicConfigResponse converts proto ListDynamicConfigResponse type to internal\nfunc ToAdminListDynamicConfigResponse(t *adminv1.ListDynamicConfigResponse) *types.ListDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDynamicConfigResponse{\n\t\tEntries: ToDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\n// FromDynamicConfigEntryArray converts internal DynamicConfigEntry array type to proto\nfunc FromDynamicConfigEntryArray(t []*types.DynamicConfigEntry) []*adminv1.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.DynamicConfigEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigEntry(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigEntryArray converts proto DynamicConfigEntry array type to internal\nfunc ToDynamicConfigEntryArray(t []*adminv1.DynamicConfigEntry) []*types.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigEntry(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigEntry converts internal DynamicConfigEntry type to proto\nfunc FromDynamicConfigEntry(t *types.DynamicConfigEntry) *adminv1.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DynamicConfigEntry{\n\t\tName:   t.Name,\n\t\tValues: FromDynamicConfigValueArray(t.Values),\n\t}\n}\n\n// ToDynamicConfigEntry converts proto DynamicConfigEntry type to internal\nfunc ToDynamicConfigEntry(t *adminv1.DynamicConfigEntry) *types.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigEntry{\n\t\tName:   t.Name,\n\t\tValues: ToDynamicConfigValueArray(t.Values),\n\t}\n}\n\n// FromDynamicConfigValueArray converts internal DynamicConfigValue array type to proto\nfunc FromDynamicConfigValueArray(t []*types.DynamicConfigValue) []*adminv1.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.DynamicConfigValue, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigValue(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigValueArray converts proto DynamicConfigValue array type to internal\nfunc ToDynamicConfigValueArray(t []*adminv1.DynamicConfigValue) []*types.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigValue, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigValue(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigValue converts internal DynamicConfigValue type to proto\nfunc FromDynamicConfigValue(t *types.DynamicConfigValue) *adminv1.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DynamicConfigValue{\n\t\tValue:   FromDataBlob(t.Value),\n\t\tFilters: FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToDynamicConfigValue converts proto DynamicConfigValue type to internal\nfunc ToDynamicConfigValue(t *adminv1.DynamicConfigValue) *types.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigValue{\n\t\tValue:   ToDataBlob(t.Value),\n\t\tFilters: ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromDynamicConfigFilterArray converts internal DynamicConfigFilter array type to proto\nfunc FromDynamicConfigFilterArray(t []*types.DynamicConfigFilter) []*adminv1.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.DynamicConfigFilter, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigFilter(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigFilterArray converts proto DynamicConfigFilter array type to internal\nfunc ToDynamicConfigFilterArray(t []*adminv1.DynamicConfigFilter) []*types.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigFilter, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigFilter(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigFilter converts internal DynamicConfigFilter type to proto\nfunc FromDynamicConfigFilter(t *types.DynamicConfigFilter) *adminv1.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DynamicConfigFilter{\n\t\tName:  t.Name,\n\t\tValue: FromDataBlob(t.Value),\n\t}\n}\n\n// ToDynamicConfigFilter converts thrift DynamicConfigFilter type to internal\nfunc ToDynamicConfigFilter(t *adminv1.DynamicConfigFilter) *types.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigFilter{\n\t\tName:  t.Name,\n\t\tValue: ToDataBlob(t.Value),\n\t}\n}\n\nfunc FromAdminGetGlobalIsolationGroupsResponse(t *types.GetGlobalIsolationGroupsResponse) *adminv1.GetGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetGlobalIsolationGroupsResponse{\n\t\tIsolationGroups: FromIsolationGroupConfig(&t.IsolationGroups),\n\t}\n}\n\nfunc FromAdminUpdateGlobalIsolationGroupsRequest(t *types.UpdateGlobalIsolationGroupsRequest) *adminv1.UpdateGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: FromIsolationGroupConfig(&t.IsolationGroups),\n\t}\n}\n\nfunc FromAdminUpdateDomainIsolationGroupsRequest(t *types.UpdateDomainIsolationGroupsRequest) *adminv1.UpdateDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          t.Domain,\n\t\tIsolationGroups: FromIsolationGroupConfig(&t.IsolationGroups),\n\t}\n}\n\nfunc FromAdminGetGlobalIsolationGroupsRequest(t *types.GetGlobalIsolationGroupsRequest) *adminv1.GetGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetGlobalIsolationGroupsRequest{}\n}\n\nfunc FromAdminGetDomainIsolationGroupsRequest(t *types.GetDomainIsolationGroupsRequest) *adminv1.GetDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDomainIsolationGroupsRequest{\n\t\tDomain: t.Domain,\n\t}\n}\n\nfunc ToAdminGetGlobalIsolationGroupsRequest(t *adminv1.GetGlobalIsolationGroupsRequest) *types.GetGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetGlobalIsolationGroupsRequest{}\n}\n\nfunc ToAdminGetGlobalIsolationGroupsResponse(t *adminv1.GetGlobalIsolationGroupsResponse) *types.GetGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tig := ToIsolationGroupConfig(t.IsolationGroups)\n\tif ig == nil {\n\t\treturn &types.GetGlobalIsolationGroupsResponse{}\n\t}\n\treturn &types.GetGlobalIsolationGroupsResponse{\n\t\tIsolationGroups: *ig,\n\t}\n}\n\nfunc ToAdminGetDomainIsolationGroupsResponse(t *adminv1.GetDomainIsolationGroupsResponse) *types.GetDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tig := ToIsolationGroupConfig(t.IsolationGroups)\n\tif ig == nil {\n\t\treturn &types.GetDomainIsolationGroupsResponse{}\n\t}\n\treturn &types.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: *ig,\n\t}\n}\n\nfunc FromAdminGetDomainIsolationGroupsResponse(t *types.GetDomainIsolationGroupsResponse) *adminv1.GetDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := FromIsolationGroupConfig(&t.IsolationGroups)\n\treturn &adminv1.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: cfg,\n\t}\n}\n\nfunc ToAdminGetDomainIsolationGroupsRequest(t *adminv1.GetDomainIsolationGroupsRequest) *types.GetDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainIsolationGroupsRequest{Domain: t.Domain}\n}\n\nfunc FromAdminUpdateGlobalIsolationGroupsResponse(t *types.UpdateGlobalIsolationGroupsResponse) *adminv1.UpdateGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateGlobalIsolationGroupsResponse{}\n}\n\nfunc ToAdminUpdateGlobalIsolationGroupsRequest(t *adminv1.UpdateGlobalIsolationGroupsRequest) *types.UpdateGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := ToIsolationGroupConfig(t.IsolationGroups)\n\tif cfg == nil {\n\t\treturn &types.UpdateGlobalIsolationGroupsRequest{}\n\t}\n\treturn &types.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: *cfg,\n\t}\n}\n\nfunc ToAdminUpdateGlobalIsolationGroupsResponse(t *adminv1.UpdateGlobalIsolationGroupsResponse) *types.UpdateGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateGlobalIsolationGroupsResponse{}\n}\n\nfunc ToAdminUpdateDomainIsolationGroupsResponse(t *adminv1.UpdateDomainIsolationGroupsResponse) *types.UpdateDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainIsolationGroupsResponse{}\n}\n\nfunc FromAdminUpdateDomainIsolationGroupsResponse(t *types.UpdateDomainIsolationGroupsResponse) *adminv1.UpdateDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateDomainIsolationGroupsResponse{}\n}\n\nfunc ToAdminUpdateDomainIsolationGroupsRequest(t *adminv1.UpdateDomainIsolationGroupsRequest) *types.UpdateDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := ToIsolationGroupConfig(t.IsolationGroups)\n\tif cfg == nil {\n\t\treturn &types.UpdateDomainIsolationGroupsRequest{\n\t\t\tDomain: t.Domain,\n\t\t}\n\t}\n\treturn &types.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          t.Domain,\n\t\tIsolationGroups: *cfg,\n\t}\n}\n\nfunc FromIsolationGroupConfig(in *types.IsolationGroupConfiguration) *apiv1.IsolationGroupConfiguration {\n\tif in == nil || *in == nil {\n\t\treturn nil\n\t}\n\tvar out []*apiv1.IsolationGroupPartition\n\tfor _, v := range *in {\n\t\tout = append(out, &apiv1.IsolationGroupPartition{\n\t\t\tName:  v.Name,\n\t\t\tState: apiv1.IsolationGroupState(v.State),\n\t\t})\n\t}\n\tsort.Slice(out, func(i, j int) bool {\n\t\tif out[i] == nil || out[j] == nil {\n\t\t\treturn false\n\t\t}\n\t\treturn out[i].Name < out[j].Name\n\t})\n\treturn &apiv1.IsolationGroupConfiguration{\n\t\tIsolationGroups: out,\n\t}\n}\n\nfunc ToIsolationGroupConfig(in *apiv1.IsolationGroupConfiguration) *types.IsolationGroupConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\tout := make(types.IsolationGroupConfiguration)\n\tfor v := range in.IsolationGroups {\n\t\tout[in.IsolationGroups[v].Name] = types.IsolationGroupPartition{\n\t\t\tName:  in.IsolationGroups[v].Name,\n\t\t\tState: types.IsolationGroupState(in.IsolationGroups[v].State),\n\t\t}\n\t}\n\treturn &out\n}\n\nfunc ToAdminGetDomainAsyncWorkflowConfiguratonRequest(in *adminv1.GetDomainAsyncWorkflowConfiguratonRequest) *types.GetDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain: in.Domain,\n\t}\n}\n\nfunc FromAdminGetDomainAsyncWorkflowConfiguratonResponse(in *types.GetDomainAsyncWorkflowConfiguratonResponse) *adminv1.GetDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\tConfiguration: FromDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc FromDomainAsyncWorkflowConfiguraton(in *types.AsyncWorkflowConfiguration) *apiv1.AsyncWorkflowConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\n\treturn &apiv1.AsyncWorkflowConfiguration{\n\t\tEnabled:             in.Enabled,\n\t\tPredefinedQueueName: in.PredefinedQueueName,\n\t\tQueueType:           in.QueueType,\n\t\tQueueConfig:         FromDataBlob(in.QueueConfig),\n\t}\n}\n\nfunc ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(in *adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest) *types.UpdateDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain:        in.Domain,\n\t\tConfiguration: ToDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc ToDomainAsyncWorkflowConfiguraton(in *apiv1.AsyncWorkflowConfiguration) *types.AsyncWorkflowConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.AsyncWorkflowConfiguration{\n\t\tEnabled:             in.Enabled,\n\t\tPredefinedQueueName: in.PredefinedQueueName,\n\t\tQueueType:           in.QueueType,\n\t\tQueueConfig:         ToDataBlob(in.QueueConfig),\n\t}\n}\n\nfunc FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(in *types.UpdateDomainAsyncWorkflowConfiguratonResponse) *adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse{}\n}\n\nfunc FromAdminGetDomainAsyncWorkflowConfiguratonRequest(in *types.GetDomainAsyncWorkflowConfiguratonRequest) *adminv1.GetDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain: in.Domain,\n\t}\n}\n\nfunc ToAdminGetDomainAsyncWorkflowConfiguratonResponse(in *adminv1.GetDomainAsyncWorkflowConfiguratonResponse) *types.GetDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\tConfiguration: ToDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(in *types.UpdateDomainAsyncWorkflowConfiguratonRequest) *adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain:        in.Domain,\n\t\tConfiguration: FromDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(in *adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse) *types.UpdateDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainAsyncWorkflowConfiguratonResponse{}\n}\n\nfunc FromAdminUpdateTaskListPartitionConfigRequest(in *types.UpdateTaskListPartitionConfigRequest) *adminv1.UpdateTaskListPartitionConfigRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateTaskListPartitionConfigRequest{\n\t\tDomain:          in.Domain,\n\t\tTaskList:        FromTaskList(in.TaskList),\n\t\tTaskListType:    FromTaskListType(in.TaskListType),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(in.PartitionConfig),\n\t}\n}\n\nfunc ToAdminUpdateTaskListPartitionConfigRequest(in *adminv1.UpdateTaskListPartitionConfigRequest) *types.UpdateTaskListPartitionConfigRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateTaskListPartitionConfigRequest{\n\t\tDomain:          in.Domain,\n\t\tTaskList:        ToTaskList(in.TaskList),\n\t\tTaskListType:    ToTaskListType(in.TaskListType),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(in.PartitionConfig),\n\t}\n}\n\nfunc FromAdminUpdateTaskListPartitionConfigResponse(t *types.UpdateTaskListPartitionConfigResponse) *adminv1.UpdateTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.UpdateTaskListPartitionConfigResponse{}\n}\n\nfunc ToAdminUpdateTaskListPartitionConfigResponse(t *adminv1.UpdateTaskListPartitionConfigResponse) *types.UpdateTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateTaskListPartitionConfigResponse{}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/admin_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/testutils\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestAdminAddSearchAttributeRequest(t *testing.T) {\n\tfor _, item := range []*types.AddSearchAttributeRequest{nil, {}, &testdata.AdminAddSearchAttributeRequest} {\n\t\tassert.Equal(t, item, ToAdminAddSearchAttributeRequest(FromAdminAddSearchAttributeRequest(item)))\n\t}\n}\nfunc TestAdminCloseShardRequest(t *testing.T) {\n\tfor _, item := range []*types.CloseShardRequest{nil, {}, &testdata.AdminCloseShardRequest} {\n\t\tassert.Equal(t, item, ToAdminCloseShardRequest(FromAdminCloseShardRequest(item)))\n\t}\n}\nfunc TestAdminDeleteWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminDeleteWorkflowRequest{nil, {}, &testdata.AdminDeleteWorkflowRequest} {\n\t\tassert.Equal(t, item, ToAdminDeleteWorkflowRequest(FromAdminDeleteWorkflowRequest(item)))\n\t}\n}\nfunc TestAdminDeleteWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminDeleteWorkflowResponse{nil, {}, &testdata.AdminDeleteWorkflowResponse} {\n\t\tassert.Equal(t, item, ToAdminDeleteWorkflowResponse(FromAdminDeleteWorkflowResponse(item)))\n\t}\n}\nfunc TestAdminDescribeClusterResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeClusterResponse{nil, {}, &testdata.AdminDescribeClusterResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeClusterResponse(FromAdminDescribeClusterResponse(item)))\n\t}\n}\nfunc TestAdminDescribeHistoryHostRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostRequest{\n\t\tnil,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByHost,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByShard,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByExecution,\n\t} {\n\t\tassert.Equal(t, item, ToAdminDescribeHistoryHostRequest(FromAdminDescribeHistoryHostRequest(item)))\n\t}\n\tassert.Panics(t, func() { ToAdminDescribeHistoryHostRequest(&adminv1.DescribeHistoryHostRequest{}) })\n\tassert.Panics(t, func() { FromAdminDescribeHistoryHostRequest(&types.DescribeHistoryHostRequest{}) })\n}\nfunc TestAdminDescribeHistoryHostResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostResponse{nil, {}, &testdata.AdminDescribeHistoryHostResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeHistoryHostResponse(FromAdminDescribeHistoryHostResponse(item)))\n\t}\n}\nfunc TestAdminDescribeQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueRequest{nil, {}, &testdata.AdminDescribeQueueRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeQueueRequest(FromAdminDescribeQueueRequest(item)))\n\t}\n}\nfunc TestAdminDescribeQueueResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueResponse{nil, {}, &testdata.AdminDescribeQueueResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeQueueResponse(FromAdminDescribeQueueResponse(item)))\n\t}\n}\nfunc TestAdminDescribeShardDistributionRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeShardDistributionRequest{nil, {}, &testdata.AdminDescribeShardDistributionRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeShardDistributionRequest(FromAdminDescribeShardDistributionRequest(item)))\n\t}\n}\nfunc TestAdminDescribeShardDistributionResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeShardDistributionResponse{nil, {}, &testdata.AdminDescribeShardDistributionResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeShardDistributionResponse(FromAdminDescribeShardDistributionResponse(item)))\n\t}\n}\nfunc TestAdminDescribeWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminDescribeWorkflowExecutionRequest{nil, {}, &testdata.AdminDescribeWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeWorkflowExecutionRequest(FromAdminDescribeWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestAdminDescribeWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminDescribeWorkflowExecutionResponse{nil, {ShardID: \"0\"}, &testdata.AdminDescribeWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeWorkflowExecutionResponse(FromAdminDescribeWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestAdminGetDLQReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDLQReplicationMessagesRequest{nil, {}, &testdata.AdminGetDLQReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDLQReplicationMessagesRequest(FromAdminGetDLQReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestAdminGetDLQReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDLQReplicationMessagesResponse{nil, {}, &testdata.AdminGetDLQReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDLQReplicationMessagesResponse(FromAdminGetDLQReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestAdminGetDomainIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDomainIsolationGroupsRequest{nil, {}, &testdata.AdminGetDomainIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDomainIsolationGroupsRequest(FromAdminGetDomainIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminGetDomainIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDomainIsolationGroupsResponse{nil, {}, &testdata.AdminGetDomainIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDomainIsolationGroupsResponse(FromAdminGetDomainIsolationGroupsResponse(item)))\n\t}\n}\nfunc TestAdminGetDomainReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDomainReplicationMessagesRequest{nil, {}, &testdata.AdminGetDomainReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDomainReplicationMessagesRequest(FromAdminGetDomainReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestAdminGetDomainReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDomainReplicationMessagesResponse{nil, {}, &testdata.AdminGetDomainReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDomainReplicationMessagesResponse(FromAdminGetDomainReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestAdminGetDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDynamicConfigRequest{nil, {}, &testdata.AdminGetDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDynamicConfigRequest(FromAdminGetDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminGetDynamicConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDynamicConfigResponse{nil, {}, &testdata.AdminGetDynamicConfigResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDynamicConfigResponse(FromAdminGetDynamicConfigResponse(item)))\n\t}\n}\nfunc TestAdminGetGlobalIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.GetGlobalIsolationGroupsRequest{nil, {}, &testdata.AdminGetGlobalIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminGetGlobalIsolationGroupsRequest(FromAdminGetGlobalIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminGetReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesRequest{nil, {}, &testdata.AdminGetReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminGetReplicationMessagesRequest(FromAdminGetReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestAdminGetReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesResponse{nil, {}, &testdata.AdminGetReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminGetReplicationMessagesResponse(FromAdminGetReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestAdminGetWorkflowExecutionRawHistoryV2Request(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionRawHistoryV2Request{nil, {}, &testdata.AdminGetWorkflowExecutionRawHistoryV2Request} {\n\t\tassert.Equal(t, item, ToAdminGetWorkflowExecutionRawHistoryV2Request(FromAdminGetWorkflowExecutionRawHistoryV2Request(item)))\n\t}\n}\nfunc TestAdminGetWorkflowExecutionRawHistoryV2Response(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionRawHistoryV2Response{nil, {}, &testdata.AdminGetWorkflowExecutionRawHistoryV2Response} {\n\t\tassert.Equal(t, item, ToAdminGetWorkflowExecutionRawHistoryV2Response(FromAdminGetWorkflowExecutionRawHistoryV2Response(item)))\n\t}\n}\nfunc TestAdminCountDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.CountDLQMessagesRequest{nil, {}, &testdata.AdminCountDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminCountDLQMessagesRequest(FromAdminCountDLQMessagesRequest(item)))\n\t}\n}\nfunc TestAdminCountDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.CountDLQMessagesResponse{nil, {}, &testdata.AdminCountDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminCountDLQMessagesResponse(FromAdminCountDLQMessagesResponse(item)))\n\t}\n}\nfunc TestAdminListDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.ListDynamicConfigRequest{nil, {}, &testdata.AdminListDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminListDynamicConfigRequest(FromAdminListDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminListDynamicConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.ListDynamicConfigResponse{nil, {}, &testdata.AdminListDynamicConfigResponse} {\n\t\tassert.Equal(t, item, ToAdminListDynamicConfigResponse(FromAdminListDynamicConfigResponse(item)))\n\t}\n}\nfunc TestAdminMaintainCorruptWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminMaintainWorkflowRequest{nil, {}, &testdata.AdminMaintainCorruptWorkflowRequest} {\n\t\tassert.Equal(t, item, ToAdminMaintainCorruptWorkflowRequest(FromAdminMaintainCorruptWorkflowRequest(item)))\n\t}\n}\nfunc TestAdminMaintainCorruptWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminMaintainWorkflowResponse{nil, {}, &testdata.AdminMaintainCorruptWorkflowResponse} {\n\t\tassert.Equal(t, item, ToAdminMaintainCorruptWorkflowResponse(FromAdminMaintainCorruptWorkflowResponse(item)))\n\t}\n}\nfunc TestAdminMergeDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.MergeDLQMessagesRequest{nil, {}, &testdata.AdminMergeDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminMergeDLQMessagesRequest(FromAdminMergeDLQMessagesRequest(item)))\n\t}\n}\nfunc TestAdminMergeDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.MergeDLQMessagesResponse{nil, {}, &testdata.AdminMergeDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminMergeDLQMessagesResponse(FromAdminMergeDLQMessagesResponse(item)))\n\t}\n}\nfunc TestAdminPurgeDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.PurgeDLQMessagesRequest{nil, {}, &testdata.AdminPurgeDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminPurgeDLQMessagesRequest(FromAdminPurgeDLQMessagesRequest(item)))\n\t}\n}\nfunc TestAdminReadDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.ReadDLQMessagesRequest{nil, {}, &testdata.AdminReadDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminReadDLQMessagesRequest(FromAdminReadDLQMessagesRequest(item)))\n\t}\n}\nfunc TestAdminReadDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.ReadDLQMessagesResponse{nil, {}, &testdata.AdminReadDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminReadDLQMessagesResponse(FromAdminReadDLQMessagesResponse(item)))\n\t}\n}\nfunc TestAdminReapplyEventsRequest(t *testing.T) {\n\tfor _, item := range []*types.ReapplyEventsRequest{nil, {}, &testdata.AdminReapplyEventsRequest} {\n\t\tassert.Equal(t, item, ToAdminReapplyEventsRequest(FromAdminReapplyEventsRequest(item)))\n\t}\n}\nfunc TestAdminRefreshWorkflowTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.RefreshWorkflowTasksRequest{nil, {}, &testdata.AdminRefreshWorkflowTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminRefreshWorkflowTasksRequest(FromAdminRefreshWorkflowTasksRequest(item)))\n\t}\n}\nfunc TestAdminRemoveTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.RemoveTaskRequest{nil, {}, &testdata.AdminRemoveTaskRequest} {\n\t\tassert.Equal(t, item, ToAdminRemoveTaskRequest(FromAdminRemoveTaskRequest(item)))\n\t}\n}\nfunc TestAdminResendReplicationTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.ResendReplicationTasksRequest{nil, {}, &testdata.AdminResendReplicationTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminResendReplicationTasksRequest(FromAdminResendReplicationTasksRequest(item)))\n\t}\n}\nfunc TestAdminResetQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.ResetQueueRequest{nil, {}, &testdata.AdminResetQueueRequest} {\n\t\tassert.Equal(t, item, ToAdminResetQueueRequest(FromAdminResetQueueRequest(item)))\n\t}\n}\n\nfunc TestAdminGetCrossClusterTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksRequest{nil, {}, &testdata.AdminGetCrossClusterTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksRequest(FromAdminGetCrossClusterTasksRequest(item)))\n\t}\n}\n\nfunc TestAdminGetCrossClusterTasksResponse(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksResponse{nil, {}, &testdata.AdminGetCrossClusterTasksResponse} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksResponse(FromAdminGetCrossClusterTasksResponse(item)))\n\t}\n}\n\nfunc TestAdminRespondCrossClusterTasksCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedRequest{nil, {}, &testdata.AdminRespondCrossClusterTasksCompletedRequest} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedRequest(FromAdminRespondCrossClusterTasksCompletedRequest(item)))\n\t}\n}\n\nfunc TestAdminRespondCrossClusterTasksCompletedResponse(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedResponse{nil, {}, &testdata.AdminRespondCrossClusterTasksCompletedResponse} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedResponse(FromAdminRespondCrossClusterTasksCompletedResponse(item)))\n\t}\n}\nfunc TestAdminUpdateDomainIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainIsolationGroupsRequest{nil, {}, &testdata.AdminUpdateDomainIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminUpdateDomainIsolationGroupsRequest(FromAdminUpdateDomainIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminUpdateDomainIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainIsolationGroupsResponse{nil, {}, &testdata.AdminUpdateDomainIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminUpdateDomainIsolationGroupsResponse(FromAdminUpdateDomainIsolationGroupsResponse(item)))\n\t}\n}\nfunc TestAdminRestoreDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.RestoreDynamicConfigRequest{nil, {}, &testdata.AdminRestoreDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminRestoreDynamicConfigRequest(FromAdminRestoreDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminUpdateDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateDynamicConfigRequest{nil, {}, &testdata.AdminUpdateDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminUpdateDynamicConfigRequest(FromAdminUpdateDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminUpdateGlobalIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateGlobalIsolationGroupsResponse{nil, {}, &testdata.AdminUpdateGlobalIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminUpdateGlobalIsolationGroupsResponse(FromAdminUpdateGlobalIsolationGroupsResponse(item)))\n\t}\n}\n\nfunc TestFromAdminGetGlobalIsolationGroupsResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetGlobalIsolationGroupsResponse\n\t\texpected *adminv1.GetGlobalIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 0\": {\n\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 1\": {\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*v1.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_HEALTHY,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_DRAINED,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil - 1\": {\n\t\t\tin: &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t\texpected: &adminv1.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t},\n\t\t\"nil - 2\": {\n\t\t\texpected: nil,\n\t\t},\n\t\t\"nil - 3\": {\n\t\t\tin: &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: nil,\n\t\t\t},\n\t\t\texpected: &adminv1.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: nil,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := FromAdminGetGlobalIsolationGroupsResponse(td.in)\n\t\t\tassert.Equal(t, td.expected, res, \"mapping\")\n\t\t\troundTrip := ToAdminGetGlobalIsolationGroupsResponse(res)\n\t\t\tif td.in != nil {\n\t\t\t\tassert.Equal(t, td.in, roundTrip, \"roundtrip\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetGlobalIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *adminv1.GetGlobalIsolationGroupsRequest\n\t\texpected *types.GetGlobalIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin:       &adminv1.GetGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.GetGlobalIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil - 2\": {\n\t\t\tin:       &adminv1.GetGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.GetGlobalIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetGlobalIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminGetDomainIsolationGroupsResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetDomainIsolationGroupsResponse\n\t\texpected *adminv1.GetDomainIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &types.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 0\": {\n\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 1\": {\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*v1.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_HEALTHY,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_DRAINED,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin: &types.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t\texpected: &adminv1.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*v1.IsolationGroupPartition{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := FromAdminGetDomainIsolationGroupsResponse(td.in)\n\t\t\t// map iteration is nondeterministic\n\t\t\tsort.Slice(res.IsolationGroups.IsolationGroups, func(i int, j int) bool {\n\t\t\t\treturn res.IsolationGroups.IsolationGroups[i].Name > res.IsolationGroups.IsolationGroups[j].Name\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetDomainIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *adminv1.GetDomainIsolationGroupsRequest\n\t\texpected *types.GetDomainIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &adminv1.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"domain123\",\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"domain123\",\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.GetDomainIsolationGroupsRequest{},\n\t\t\texpected: &types.GetDomainIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetDomainIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateGlobalIsolationGroupsResponse(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *types.UpdateGlobalIsolationGroupsResponse\n\t\texpected *adminv1.UpdateGlobalIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {},\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateGlobalIsolationGroupsResponse{},\n\t\t\texpected: &adminv1.UpdateGlobalIsolationGroupsResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateGlobalIsolationGroupsResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminUpdateGlobalIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *adminv1.UpdateGlobalIsolationGroupsRequest\n\t\texpected *types.UpdateGlobalIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &adminv1.UpdateGlobalIsolationGroupsRequest{\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*v1.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_HEALTHY,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone 2\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_DRAINED,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateGlobalIsolationGroupsRequest{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 1\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 2\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone 2\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.UpdateGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.UpdateGlobalIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := ToAdminUpdateGlobalIsolationGroupsRequest(td.in)\n\t\t\tassert.Equal(t, td.expected, res, \"conversion\")\n\t\t\troundTrip := FromAdminUpdateGlobalIsolationGroupsRequest(res)\n\t\t\tif td.in != nil {\n\t\t\t\tassert.Equal(t, td.in, roundTrip, \"roundtrip\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateDomainIsolationGroupsResponse(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *types.UpdateDomainIsolationGroupsResponse\n\t\texpected *adminv1.UpdateDomainIsolationGroupsResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateDomainIsolationGroupsResponse{},\n\t\t\texpected: &adminv1.UpdateDomainIsolationGroupsResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateDomainIsolationGroupsResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToUpdateDomainIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *adminv1.UpdateDomainIsolationGroupsRequest\n\t\texpected *types.UpdateDomainIsolationGroupsRequest\n\t}{\n\t\t\"valid\": {\n\t\t\tin: &adminv1.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tIsolationGroups: &v1.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*v1.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_HEALTHY,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\t\tState: v1.IsolationGroupState_ISOLATION_GROUP_STATE_DRAINED,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminUpdateDomainIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *adminv1.GetDomainAsyncWorkflowConfiguratonRequest\n\t\texpected *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\texpected: &types.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"valid\": {\n\t\t\tin: &adminv1.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\texpected: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetDomainAsyncWorkflowConfiguratonRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminGetDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t\texpected *adminv1.GetDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\texpected: &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tin: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tin: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &v1.DataBlob{\n\t\t\t\t\t\tEncodingType: v1.EncodingType_ENCODING_TYPE_JSON,\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminGetDomainAsyncWorkflowConfiguratonResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\texpected *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\texpected: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tin: &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             true,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tin: &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &v1.DataBlob{\n\t\t\t\t\t\tEncodingType: v1.EncodingType_ENCODING_TYPE_JSON,\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\texpected *adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\texpected: &adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminGetDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t\texpected *adminv1.GetDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &types.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\texpected: &adminv1.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"valid\": {\n\t\t\tin: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\texpected: &adminv1.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminGetDomainAsyncWorkflowConfiguratonRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *adminv1.GetDomainAsyncWorkflowConfiguratonResponse\n\t\texpected *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\texpected: &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tin: &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tin: &adminv1.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &v1.DataBlob{\n\t\t\t\t\t\tEncodingType: v1.EncodingType_ENCODING_TYPE_JSON,\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetDomainAsyncWorkflowConfiguratonResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\texpected *adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\texpected: &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tin: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"kafka inline queue\": {\n\t\t\tin: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &v1.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   true,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &v1.DataBlob{\n\t\t\t\t\t\tEncodingType: v1.EncodingType_ENCODING_TYPE_JSON,\n\t\t\t\t\t\tData:         []byte(`{\"topic\":\"test-topic\",\"dlq_topic\":\"test-dlq-topic\",\"consumer_group\":\"test-consumer-group\",\"brokers\":[\"test-broker-1\",\"test-broker-2\"],\"properties\":{\"test-key-1\":\"test-value-1\"}}`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\texpected *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\texpected: &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateTaskListPartitionConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateTaskListPartitionConfigRequest{nil, {}, &testdata.AdminUpdateTaskListPartitionConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminUpdateTaskListPartitionConfigRequest(FromAdminUpdateTaskListPartitionConfigRequest(item)))\n\t}\n}\n\nfunc TestAdminUpdateTaskListPartitionConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateTaskListPartitionConfigResponse{nil, {}} {\n\t\tassert.Equal(t, item, ToAdminUpdateTaskListPartitionConfigResponse(FromAdminUpdateTaskListPartitionConfigResponse(item)))\n\t}\n}\n\nfunc TestAdminCloseShardRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminCloseShardRequest, ToAdminCloseShardRequest)\n}\n\nfunc TestAdminResetQueueRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminResetQueueRequest, ToAdminResetQueueRequest,\n\t\ttestutils.WithCustomFuncs(func(e *types.ResetQueueRequest, c fuzz.Continue) {\n\t\t\t// This function needs to exist because we can't independently fuzz the type field\n\t\t\t// as both shardID and type have the same type signature.\n\t\t\tc.Fuzz(e)\n\n\t\t\ttestutils.TaskTypeFuzzer(e.Type, c)\n\t\t}),\n\t)\n}\n\nfunc TestAdminUpdateDynamicConfigRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateDynamicConfigRequest, ToAdminUpdateDynamicConfigRequest,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminUpdateTaskListPartitionConfigRequestFuzz(t *testing.T) {\n\t// ReadPartitions and WritePartitions are tested in api_test.go\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateTaskListPartitionConfigRequest, ToAdminUpdateTaskListPartitionConfigRequest,\n\t\ttestutils.WithExcludedFields(\"ReadPartitions\", \"WritePartitions\"),\n\t)\n}\n\nfunc TestAdminGetDLQReplicationMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDLQReplicationMessagesRequest, ToAdminGetDLQReplicationMessagesRequest)\n}\n\nfunc TestAdminMergeDLQMessagesResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminMergeDLQMessagesResponse, ToAdminMergeDLQMessagesResponse)\n}\n\nfunc TestAdminDescribeWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeWorkflowExecutionRequest, ToAdminDescribeWorkflowExecutionRequest)\n}\n\nfunc TestAdminCountDLQMessagesResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminCountDLQMessagesResponse, ToAdminCountDLQMessagesResponse)\n}\n\nfunc TestAdminReapplyEventsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminReapplyEventsRequest, ToAdminReapplyEventsRequest,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminRespondCrossClusterTasksCompletedRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminRespondCrossClusterTasksCompletedRequest, ToAdminRespondCrossClusterTasksCompletedRequest,\n\t\ttestutils.WithExcludedFields(\n\t\t\t\"TaskResponses\", // Excluded and tested in TestFromCrossClusterTaskResponseFuzz in shared\n\t\t),\n\t)\n}\n\nfunc TestAdminGetDomainReplicationMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDomainReplicationMessagesRequest, ToAdminGetDomainReplicationMessagesRequest)\n}\n\nfunc TestAdminUpdateTaskListPartitionConfigResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateTaskListPartitionConfigResponse, ToAdminUpdateTaskListPartitionConfigResponse)\n}\n\nfunc TestAdminRefreshWorkflowTasksRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminRefreshWorkflowTasksRequest, ToAdminRefreshWorkflowTasksRequest)\n}\n\nfunc TestAdminDescribeHistoryHostRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeHistoryHostRequest, ToAdminDescribeHistoryHostRequest,\n\t\ttestutils.WithCustomFuncs(DescribeHistoryHostRequestFuzzer),\n\t)\n}\n\nfunc TestAdminGetWorkflowExecutionRawHistoryV2ResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetWorkflowExecutionRawHistoryV2Response, ToAdminGetWorkflowExecutionRawHistoryV2Response,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminRemoveTaskRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminRemoveTaskRequest, ToAdminRemoveTaskRequest,\n\t\ttestutils.WithCustomFuncs(func(e *types.RemoveTaskRequest, c fuzz.Continue) {\n\t\t\tc.Fuzz(e)\n\t\t\ttestutils.TaskTypeFuzzer(e.Type, c)\n\t\t}),\n\t)\n}\n\nfunc TestAdminResendReplicationTasksRequestFuzz(t *testing.T) {\n\t// [BUG] When ID or Version is nil for a given pair round-trip serialization fails. This is a mapper bug - should handle errors gracefully.\n\ttestutils.RunMapperFuzzTest(t, FromAdminResendReplicationTasksRequest, ToAdminResendReplicationTasksRequest,\n\t\ttestutils.WithCustomFuncs(func(r *types.ResendReplicationTasksRequest, c fuzz.Continue) {\n\t\t\tc.FuzzNoCustom(r)\n\t\t\tr.StartEventID, r.StartVersion = EventIDVersionPairFuzzer(c)\n\t\t\tr.EndEventID, r.EndVersion = EventIDVersionPairFuzzer(c)\n\t\t}),\n\t)\n}\n\nfunc TestAdminAddSearchAttributeRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminAddSearchAttributeRequest, ToAdminAddSearchAttributeRequest)\n}\n\nfunc TestAdminDescribeClusterResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeClusterResponse, ToAdminDescribeClusterResponse)\n}\n\nfunc TestAdminGetDynamicConfigResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDynamicConfigResponse, ToAdminGetDynamicConfigResponse,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminMaintainCorruptWorkflowRequestFuzz(t *testing.T) {\n\t// [BUG] SkipErrors field is missing from the mapper - not included in proto conversion\n\ttestutils.RunMapperFuzzTest(t, FromAdminMaintainCorruptWorkflowRequest, ToAdminMaintainCorruptWorkflowRequest,\n\t\ttestutils.WithExcludedFields(\"SkipErrors\"),\n\t)\n}\n\nfunc TestDynamicConfigFilterArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigFilterArray, ToDynamicConfigFilterArray, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminGetDomainAsyncWorkflowConfiguratonResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDomainAsyncWorkflowConfiguratonResponse, ToAdminGetDomainAsyncWorkflowConfiguratonResponse,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminDescribeWorkflowExecutionResponseFuzz(t *testing.T) {\n\t// [BUG] ShardID is a string that gets converted to int32 using stringToInt32()\n\t// which panics on invalid strings. This is a mapper bug - should handle errors gracefully.\n\t// Using custom fuzzer to ensure ShardID is a valid int32 string or empty.\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeWorkflowExecutionResponse, ToAdminDescribeWorkflowExecutionResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(r *types.AdminDescribeWorkflowExecutionResponse, c fuzz.Continue) {\n\t\t\t\tc.FuzzNoCustom(r)\n\n\t\t\t\t// Always generate a valid ShardID (int32 as string) to avoid stringToInt32 panics\n\t\t\t\tshardID := c.Int31()\n\t\t\t\tr.ShardID = fmt.Sprintf(\"%d\", shardID)\n\t\t\t},\n\t\t\ttestutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminDescribeShardDistributionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeShardDistributionResponse, ToAdminDescribeShardDistributionResponse)\n}\n\nfunc TestAdminMergeDLQMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminMergeDLQMessagesRequest, ToAdminMergeDLQMessagesRequest,\n\t\ttestutils.WithCustomFuncs(testutils.DLQTypeFuzzer),\n\t)\n}\n\nfunc TestAdminGetReplicationMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetReplicationMessagesRequest, ToAdminGetReplicationMessagesRequest)\n}\n\nfunc TestAdminListDynamicConfigResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminListDynamicConfigResponse, ToAdminListDynamicConfigResponse,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminGetGlobalIsolationGroupsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetGlobalIsolationGroupsRequest, ToAdminGetGlobalIsolationGroupsRequest)\n}\n\nfunc TestAdminUpdateDomainAsyncWorkflowConfiguratonResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse, ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse)\n}\n\nfunc TestDynamicConfigFilterFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigFilter, ToDynamicConfigFilter, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminDescribeHistoryHostResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeHistoryHostResponse, ToAdminDescribeHistoryHostResponse)\n}\n\nfunc TestAdminGetWorkflowExecutionRawHistoryV2RequestFuzz(t *testing.T) {\n\t// [BUG] When ID or Version is nil for a given pair round-trip serialization fails. This is a mapper bug - should handle errors gracefully.\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetWorkflowExecutionRawHistoryV2Request, ToAdminGetWorkflowExecutionRawHistoryV2Request,\n\t\ttestutils.WithCustomFuncs(func(r *types.GetWorkflowExecutionRawHistoryV2Request, c fuzz.Continue) {\n\t\t\tc.FuzzNoCustom(r)\n\t\t\tr.StartEventID, r.StartEventVersion = EventIDVersionPairFuzzer(c)\n\t\t\tr.EndEventID, r.EndEventVersion = EventIDVersionPairFuzzer(c)\n\t\t}),\n\t)\n}\n\nfunc TestAdminRestoreDynamicConfigRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminRestoreDynamicConfigRequest, ToAdminRestoreDynamicConfigRequest,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminDeleteWorkflowRequestFuzz(t *testing.T) {\n\t// [BUG] SkipErrors is not mapped\n\ttestutils.RunMapperFuzzTest(t, FromAdminDeleteWorkflowRequest, ToAdminDeleteWorkflowRequest,\n\t\ttestutils.WithExcludedFields(\"SkipErrors\"),\n\t)\n}\n\nfunc TestDynamicConfigEntryFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigEntry, ToDynamicConfigEntry, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestDynamicConfigValueArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigValueArray, ToDynamicConfigValueArray, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminDescribeQueueResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeQueueResponse, ToAdminDescribeQueueResponse)\n}\n\nfunc TestAdminReadDLQMessagesResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminReadDLQMessagesResponse, ToAdminReadDLQMessagesResponse,\n\t\ttestutils.WithExcludedFields(\"ReplicationTasks\"), // TODO(c-warren): Test ReplicationTasks in shared_test.go\n\t\ttestutils.WithCustomFuncs(testutils.DLQTypeFuzzer),\n\t)\n}\n\nfunc TestAdminGetDynamicConfigRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDynamicConfigRequest, ToAdminGetDynamicConfigRequest,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestDynamicConfigValueFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigValue, ToDynamicConfigValue, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminCountDLQMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminCountDLQMessagesRequest, ToAdminCountDLQMessagesRequest)\n}\n\nfunc TestAdminUpdateDomainAsyncWorkflowConfiguratonRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest, ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestAdminDescribeQueueRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeQueueRequest, ToAdminDescribeQueueRequest,\n\t\ttestutils.WithCustomFuncs(func(r *types.DescribeQueueRequest, c fuzz.Continue) {\n\t\t\tc.FuzzNoCustom(r)\n\t\t\ttestutils.TaskTypeFuzzer(r.Type, c)\n\t\t}),\n\t)\n}\n\nfunc TestAdminMaintainCorruptWorkflowResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminMaintainCorruptWorkflowResponse, ToAdminMaintainCorruptWorkflowResponse)\n}\n\nfunc TestAdminUpdateGlobalIsolationGroupsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateGlobalIsolationGroupsResponse, ToAdminUpdateGlobalIsolationGroupsResponse)\n}\n\nfunc TestDomainAsyncWorkflowConfiguratonFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDomainAsyncWorkflowConfiguraton, ToDomainAsyncWorkflowConfiguraton, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminDescribeShardDistributionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDescribeShardDistributionRequest, ToAdminDescribeShardDistributionRequest)\n}\n\nfunc TestAdminPurgeDLQMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminPurgeDLQMessagesRequest, ToAdminPurgeDLQMessagesRequest,\n\t\ttestutils.WithCustomFuncs(testutils.DLQTypeFuzzer),\n\t)\n}\n\nfunc TestAdminReadDLQMessagesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminReadDLQMessagesRequest, ToAdminReadDLQMessagesRequest,\n\t\ttestutils.WithCustomFuncs(testutils.DLQTypeFuzzer),\n\t)\n}\n\nfunc TestAdminGetCrossClusterTasksRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetCrossClusterTasksRequest, ToAdminGetCrossClusterTasksRequest)\n}\n\nfunc TestDynamicConfigEntryArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDynamicConfigEntryArray, ToDynamicConfigEntryArray, testutils.WithCustomFuncs(testutils.EncodingTypeFuzzer))\n}\n\nfunc TestAdminUpdateDomainIsolationGroupsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateDomainIsolationGroupsResponse, ToAdminUpdateDomainIsolationGroupsResponse)\n}\n\nfunc TestAdminUpdateDomainIsolationGroupsRequestFuzz(t *testing.T) {\n\t// [BUG] IsolationGroups map keys are normalized during round-trip causing map size changes\n\t// Excluding IsolationGroups field to avoid non-deterministic map key handling\n\ttestutils.RunMapperFuzzTest(t, FromAdminUpdateDomainIsolationGroupsRequest, ToAdminUpdateDomainIsolationGroupsRequest,\n\t\ttestutils.WithExcludedFields(\"IsolationGroups\"), // IsolationGroups rely on testdata testing\n\t)\n}\n\nfunc TestAdminGetDomainIsolationGroupsResponseFuzz(t *testing.T) {\n\t// [BUG] IsolationGroups map keys are normalized during round-trip causing map size changes\n\t// Excluding IsolationGroups field to avoid non-deterministic map key handling\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDomainIsolationGroupsResponse, ToAdminGetDomainIsolationGroupsResponse,\n\t\ttestutils.WithExcludedFields(\"IsolationGroups\"), // IsolationGroups rely on testdata testing\n\t)\n}\n\nfunc TestAdminGetDomainAsyncWorkflowConfiguratonRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDomainAsyncWorkflowConfiguratonRequest, ToAdminGetDomainAsyncWorkflowConfiguratonRequest)\n}\n\nfunc TestAdminDeleteWorkflowResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminDeleteWorkflowResponse, ToAdminDeleteWorkflowResponse)\n}\n\nfunc TestAdminListDynamicConfigRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminListDynamicConfigRequest, ToAdminListDynamicConfigRequest)\n}\n\nfunc TestAdminGetGlobalIsolationGroupsResponseFuzz(t *testing.T) {\n\t// [BUG] IsolationGroups map keys are normalized during round-trip causing map size changes\n\t// Excluding IsolationGroups field to avoid non-deterministic map key handling\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetGlobalIsolationGroupsResponse, ToAdminGetGlobalIsolationGroupsResponse,\n\t\ttestutils.WithExcludedFields(\"IsolationGroups\"),\n\t)\n}\n\nfunc TestAdminGetDomainIsolationGroupsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAdminGetDomainIsolationGroupsRequest, ToAdminGetDomainIsolationGroupsRequest)\n}\n\n// DescribeHistoryHostRequestFuzzer ensures only one of the oneof fields is set\nfunc DescribeHistoryHostRequestFuzzer(r *types.DescribeHistoryHostRequest, c fuzz.Continue) {\n\tchoice := c.Intn(3)\n\tswitch choice {\n\tcase 0:\n\t\t// Set HostAddress only\n\t\tvar addr string\n\t\tc.Fuzz(&addr)\n\t\tr.HostAddress = &addr\n\t\tr.ShardIDForHost = nil\n\t\tr.ExecutionForHost = nil\n\tcase 1:\n\t\t// Set ShardIDForHost only\n\t\tvar shardID int32\n\t\tc.Fuzz(&shardID)\n\t\tr.ShardIDForHost = &shardID\n\t\tr.HostAddress = nil\n\t\tr.ExecutionForHost = nil\n\tcase 2:\n\t\t// Set ExecutionForHost only\n\t\tvar exec types.WorkflowExecution\n\t\tc.Fuzz(&exec)\n\t\tr.ExecutionForHost = &exec\n\t\tr.HostAddress = nil\n\t\tr.ShardIDForHost = nil\n\t}\n}\n\nfunc EventIDVersionPairFuzzer(c fuzz.Continue) (*int64, *int64) {\n\tid := c.Int63()\n\tver := c.Int63()\n\treturn &id, &ver\n}\n"
  },
  {
    "path": "common/types/mapper/proto/api.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromActivityLocalDispatchInfo(t *types.ActivityLocalDispatchInfo) *apiv1.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityLocalDispatchInfo{\n\t\tActivityId:                 t.ActivityID,\n\t\tScheduledTime:              unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:                unixNanoToTime(t.StartedTimestamp),\n\t\tScheduledTimeOfThisAttempt: unixNanoToTime(t.ScheduledTimestampOfThisAttempt),\n\t\tTaskToken:                  t.TaskToken,\n\t}\n}\n\nfunc ToActivityLocalDispatchInfo(t *apiv1.ActivityLocalDispatchInfo) *types.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityLocalDispatchInfo{\n\t\tActivityID:                      t.ActivityId,\n\t\tScheduledTimestamp:              timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:                timeToUnixNano(t.StartedTime),\n\t\tScheduledTimestampOfThisAttempt: timeToUnixNano(t.ScheduledTimeOfThisAttempt),\n\t\tTaskToken:                       t.TaskToken,\n\t}\n}\n\nfunc FromActivityTaskCancelRequestedEventAttributes(t *types.ActivityTaskCancelRequestedEventAttributes) *apiv1.ActivityTaskCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityId:                   t.ActivityID,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToActivityTaskCancelRequestedEventAttributes(t *apiv1.ActivityTaskCancelRequestedEventAttributes) *types.ActivityTaskCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityID:                   t.ActivityId,\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nfunc FromActivityTaskCanceledEventAttributes(t *types.ActivityTaskCanceledEventAttributes) *apiv1.ActivityTaskCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskCanceledEventAttributes{\n\t\tDetails:                      FromPayload(t.Details),\n\t\tLatestCancelRequestedEventId: t.LatestCancelRequestedEventID,\n\t\tScheduledEventId:             t.ScheduledEventID,\n\t\tStartedEventId:               t.StartedEventID,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc ToActivityTaskCanceledEventAttributes(t *apiv1.ActivityTaskCanceledEventAttributes) *types.ActivityTaskCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCanceledEventAttributes{\n\t\tDetails:                      ToPayload(t.Details),\n\t\tLatestCancelRequestedEventID: t.LatestCancelRequestedEventId,\n\t\tScheduledEventID:             t.ScheduledEventId,\n\t\tStartedEventID:               t.StartedEventId,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc FromActivityTaskCompletedEventAttributes(t *types.ActivityTaskCompletedEventAttributes) *apiv1.ActivityTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskCompletedEventAttributes{\n\t\tResult:           FromPayload(t.Result),\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tIdentity:         t.Identity,\n\t}\n}\n\nfunc ToActivityTaskCompletedEventAttributes(t *apiv1.ActivityTaskCompletedEventAttributes) *types.ActivityTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCompletedEventAttributes{\n\t\tResult:           ToPayload(t.Result),\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tStartedEventID:   t.StartedEventId,\n\t\tIdentity:         t.Identity,\n\t}\n}\n\nfunc FromActivityTaskFailedEventAttributes(t *types.ActivityTaskFailedEventAttributes) *apiv1.ActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskFailedEventAttributes{\n\t\tFailure:          FromFailure(t.Reason, t.Details),\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tIdentity:         t.Identity,\n\t}\n}\n\nfunc ToActivityTaskFailedEventAttributes(t *apiv1.ActivityTaskFailedEventAttributes) *types.ActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskFailedEventAttributes{\n\t\tReason:           ToFailureReason(t.Failure),\n\t\tDetails:          ToFailureDetails(t.Failure),\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tStartedEventID:   t.StartedEventId,\n\t\tIdentity:         t.Identity,\n\t}\n}\n\nfunc FromActivityTaskScheduledEventAttributes(t *types.ActivityTaskScheduledEventAttributes) *apiv1.ActivityTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskScheduledEventAttributes{\n\t\tActivityId:                   t.ActivityID,\n\t\tActivityType:                 FromActivityType(t.ActivityType),\n\t\tDomain:                       t.GetDomain(),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tScheduleToCloseTimeout:       secondsToDuration(t.ScheduleToCloseTimeoutSeconds),\n\t\tScheduleToStartTimeout:       secondsToDuration(t.ScheduleToStartTimeoutSeconds),\n\t\tStartToCloseTimeout:          secondsToDuration(t.StartToCloseTimeoutSeconds),\n\t\tHeartbeatTimeout:             secondsToDuration(t.HeartbeatTimeoutSeconds),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tHeader:                       FromHeader(t.Header),\n\t}\n}\n\nfunc ToActivityTaskScheduledEventAttributes(t *apiv1.ActivityTaskScheduledEventAttributes) *types.ActivityTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskScheduledEventAttributes{\n\t\tActivityID:                    t.ActivityId,\n\t\tActivityType:                  ToActivityType(t.ActivityType),\n\t\tDomain:                        &t.Domain,\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tInput:                         ToPayload(t.Input),\n\t\tScheduleToCloseTimeoutSeconds: durationToSeconds(t.ScheduleToCloseTimeout),\n\t\tScheduleToStartTimeoutSeconds: durationToSeconds(t.ScheduleToStartTimeout),\n\t\tStartToCloseTimeoutSeconds:    durationToSeconds(t.StartToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:       durationToSeconds(t.HeartbeatTimeout),\n\t\tDecisionTaskCompletedEventID:  t.DecisionTaskCompletedEventId,\n\t\tRetryPolicy:                   ToRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        ToHeader(t.Header),\n\t}\n}\n\nfunc FromActivityTaskStartedEventAttributes(t *types.ActivityTaskStartedEventAttributes) *apiv1.ActivityTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tIdentity:         t.Identity,\n\t\tRequestId:        t.RequestID,\n\t\tAttempt:          t.Attempt,\n\t\tLastFailure:      FromFailure(t.LastFailureReason, t.LastFailureDetails),\n\t}\n}\n\nfunc ToActivityTaskStartedEventAttributes(t *apiv1.ActivityTaskStartedEventAttributes) *types.ActivityTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventID:   t.ScheduledEventId,\n\t\tIdentity:           t.Identity,\n\t\tRequestID:          t.RequestId,\n\t\tAttempt:            t.Attempt,\n\t\tLastFailureReason:  ToFailureReason(t.LastFailure),\n\t\tLastFailureDetails: ToFailureDetails(t.LastFailure),\n\t}\n}\n\nfunc FromActivityTaskTimedOutEventAttributes(t *types.ActivityTaskTimedOutEventAttributes) *apiv1.ActivityTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityTaskTimedOutEventAttributes{\n\t\tDetails:          FromPayload(t.Details),\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tTimeoutType:      FromTimeoutType(t.TimeoutType),\n\t\tLastFailure:      FromFailure(t.LastFailureReason, t.LastFailureDetails),\n\t}\n}\n\nfunc ToActivityTaskTimedOutEventAttributes(t *apiv1.ActivityTaskTimedOutEventAttributes) *types.ActivityTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskTimedOutEventAttributes{\n\t\tDetails:            ToPayload(t.Details),\n\t\tScheduledEventID:   t.ScheduledEventId,\n\t\tStartedEventID:     t.StartedEventId,\n\t\tTimeoutType:        ToTimeoutType(t.TimeoutType),\n\t\tLastFailureReason:  ToFailureReason(t.LastFailure),\n\t\tLastFailureDetails: ToFailureDetails(t.LastFailure),\n\t}\n}\n\nfunc FromActivityType(t *types.ActivityType) *apiv1.ActivityType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActivityType{\n\t\tName: t.Name,\n\t}\n}\n\nfunc ToActivityType(t *apiv1.ActivityType) *types.ActivityType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityType{\n\t\tName: t.Name,\n\t}\n}\n\nfunc FromArchivalStatus(t *types.ArchivalStatus) apiv1.ArchivalStatus {\n\tif t == nil {\n\t\treturn apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID\n\t}\n\tswitch *t {\n\tcase types.ArchivalStatusDisabled:\n\t\treturn apiv1.ArchivalStatus_ARCHIVAL_STATUS_DISABLED\n\tcase types.ArchivalStatusEnabled:\n\t\treturn apiv1.ArchivalStatus_ARCHIVAL_STATUS_ENABLED\n\t}\n\treturn apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID\n}\n\nfunc ToArchivalStatus(t apiv1.ArchivalStatus) *types.ArchivalStatus {\n\tswitch t {\n\tcase apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID:\n\t\treturn nil\n\tcase apiv1.ArchivalStatus_ARCHIVAL_STATUS_DISABLED:\n\t\treturn types.ArchivalStatusDisabled.Ptr()\n\tcase apiv1.ArchivalStatus_ARCHIVAL_STATUS_ENABLED:\n\t\treturn types.ArchivalStatusEnabled.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromBadBinaries(t *types.BadBinaries) *apiv1.BadBinaries {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.BadBinaries{\n\t\tBinaries: FromBadBinaryInfoMap(t.Binaries),\n\t}\n}\n\nfunc ToBadBinaries(t *apiv1.BadBinaries) *types.BadBinaries {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BadBinaries{\n\t\tBinaries: ToBadBinaryInfoMap(t.Binaries),\n\t}\n}\n\nfunc FromBadBinaryInfo(t *types.BadBinaryInfo) *apiv1.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.BadBinaryInfo{\n\t\tReason:      t.Reason,\n\t\tOperator:    t.Operator,\n\t\tCreatedTime: unixNanoToTime(t.CreatedTimeNano),\n\t}\n}\n\nfunc ToBadBinaryInfo(t *apiv1.BadBinaryInfo) *types.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BadBinaryInfo{\n\t\tReason:          t.Reason,\n\t\tOperator:        t.Operator,\n\t\tCreatedTimeNano: timeToUnixNano(t.CreatedTime),\n\t}\n}\n\nfunc FromCancelExternalWorkflowExecutionFailedCause(t *types.CancelExternalWorkflowExecutionFailedCause) apiv1.CancelExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\treturn apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\n\tcase types.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\treturn apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED\n\t}\n\treturn apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n}\n\nfunc ToCancelExternalWorkflowExecutionFailedCause(t apiv1.CancelExternalWorkflowExecutionFailedCause) *types.CancelExternalWorkflowExecutionFailedCause {\n\tswitch t {\n\tcase apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION:\n\t\treturn types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr()\n\tcase apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED:\n\t\treturn types.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromCancelTimerDecisionAttributes(t *types.CancelTimerDecisionAttributes) *apiv1.CancelTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CancelTimerDecisionAttributes{\n\t\tTimerId: t.TimerID,\n\t}\n}\n\nfunc ToCancelTimerDecisionAttributes(t *apiv1.CancelTimerDecisionAttributes) *types.CancelTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelTimerDecisionAttributes{\n\t\tTimerID: t.TimerId,\n\t}\n}\n\nfunc FromCancelTimerFailedEventAttributes(t *types.CancelTimerFailedEventAttributes) *apiv1.CancelTimerFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CancelTimerFailedEventAttributes{\n\t\tTimerId:                      t.TimerID,\n\t\tCause:                        t.Cause,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc ToCancelTimerFailedEventAttributes(t *apiv1.CancelTimerFailedEventAttributes) *types.CancelTimerFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelTimerFailedEventAttributes{\n\t\tTimerID:                      t.TimerId,\n\t\tCause:                        t.Cause,\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc FromCancelWorkflowExecutionDecisionAttributes(t *types.CancelWorkflowExecutionDecisionAttributes) *apiv1.CancelWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: FromPayload(t.Details),\n\t}\n}\n\nfunc ToCancelWorkflowExecutionDecisionAttributes(t *apiv1.CancelWorkflowExecutionDecisionAttributes) *types.CancelWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: ToPayload(t.Details),\n\t}\n}\n\nfunc FromChildWorkflowExecutionCanceledEventAttributes(t *types.ChildWorkflowExecutionCanceledEventAttributes) *apiv1.ChildWorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tStartedEventId:    t.StartedEventID,\n\t\tDetails:           FromPayload(t.Details),\n\t}\n}\n\nfunc ToChildWorkflowExecutionCanceledEventAttributes(t *apiv1.ChildWorkflowExecutionCanceledEventAttributes) *types.ChildWorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tStartedEventID:    t.StartedEventId,\n\t\tDetails:           ToPayload(t.Details),\n\t}\n}\n\nfunc FromChildWorkflowExecutionCompletedEventAttributes(t *types.ChildWorkflowExecutionCompletedEventAttributes) *apiv1.ChildWorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tStartedEventId:    t.StartedEventID,\n\t\tResult:            FromPayload(t.Result),\n\t}\n}\n\nfunc ToChildWorkflowExecutionCompletedEventAttributes(t *apiv1.ChildWorkflowExecutionCompletedEventAttributes) *types.ChildWorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tStartedEventID:    t.StartedEventId,\n\t\tResult:            ToPayload(t.Result),\n\t}\n}\n\nfunc FromChildWorkflowExecutionFailedCause(t *types.ChildWorkflowExecutionFailedCause) apiv1.ChildWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning:\n\t\treturn apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING\n\t}\n\treturn apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n}\n\nfunc ToChildWorkflowExecutionFailedCause(t apiv1.ChildWorkflowExecutionFailedCause) *types.ChildWorkflowExecutionFailedCause {\n\tswitch t {\n\tcase apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING:\n\t\treturn types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromChildWorkflowExecutionFailedEventAttributes(t *types.ChildWorkflowExecutionFailedEventAttributes) *apiv1.ChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tStartedEventId:    t.StartedEventID,\n\t\tFailure:           FromFailure(t.Reason, t.Details),\n\t}\n}\n\nfunc ToChildWorkflowExecutionFailedEventAttributes(t *apiv1.ChildWorkflowExecutionFailedEventAttributes) *types.ChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tStartedEventID:    t.StartedEventId,\n\t\tReason:            ToFailureReason(t.Failure),\n\t\tDetails:           ToFailureDetails(t.Failure),\n\t}\n}\n\nfunc FromChildWorkflowExecutionStartedEventAttributes(t *types.ChildWorkflowExecutionStartedEventAttributes) *apiv1.ChildWorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tHeader:            FromHeader(t.Header),\n\t}\n}\n\nfunc ToChildWorkflowExecutionStartedEventAttributes(t *apiv1.ChildWorkflowExecutionStartedEventAttributes) *types.ChildWorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tHeader:            ToHeader(t.Header),\n\t}\n}\n\nfunc FromChildWorkflowExecutionTerminatedEventAttributes(t *types.ChildWorkflowExecutionTerminatedEventAttributes) *apiv1.ChildWorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tStartedEventId:    t.StartedEventID,\n\t}\n}\n\nfunc ToChildWorkflowExecutionTerminatedEventAttributes(t *apiv1.ChildWorkflowExecutionTerminatedEventAttributes) *types.ChildWorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tStartedEventID:    t.StartedEventId,\n\t}\n}\n\nfunc FromChildWorkflowExecutionTimedOutEventAttributes(t *types.ChildWorkflowExecutionTimedOutEventAttributes) *apiv1.ChildWorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tStartedEventId:    t.StartedEventID,\n\t\tTimeoutType:       FromTimeoutType(t.TimeoutType),\n\t}\n}\n\nfunc ToChildWorkflowExecutionTimedOutEventAttributes(t *apiv1.ChildWorkflowExecutionTimedOutEventAttributes) *types.ChildWorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tStartedEventID:    t.StartedEventId,\n\t\tTimeoutType:       ToTimeoutType(t.TimeoutType),\n\t}\n}\n\nfunc FromClusterReplicationConfiguration(t *types.ClusterReplicationConfiguration) *apiv1.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ClusterReplicationConfiguration{\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc ToClusterReplicationConfiguration(t *apiv1.ClusterReplicationConfiguration) *types.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterReplicationConfiguration{\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc FromCompleteWorkflowExecutionDecisionAttributes(t *types.CompleteWorkflowExecutionDecisionAttributes) *apiv1.CompleteWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: FromPayload(t.Result),\n\t}\n}\n\nfunc ToCompleteWorkflowExecutionDecisionAttributes(t *apiv1.CompleteWorkflowExecutionDecisionAttributes) *types.CompleteWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: ToPayload(t.Result),\n\t}\n}\n\nfunc FromContinueAsNewInitiator(t *types.ContinueAsNewInitiator) apiv1.ContinueAsNewInitiator {\n\tif t == nil {\n\t\treturn apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID\n\t}\n\tswitch *t {\n\tcase types.ContinueAsNewInitiatorDecider:\n\t\treturn apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_DECIDER\n\tcase types.ContinueAsNewInitiatorRetryPolicy:\n\t\treturn apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_RETRY_POLICY\n\tcase types.ContinueAsNewInitiatorCronSchedule:\n\t\treturn apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_CRON_SCHEDULE\n\t}\n\treturn apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID\n}\n\nfunc ToContinueAsNewInitiator(t apiv1.ContinueAsNewInitiator) *types.ContinueAsNewInitiator {\n\tswitch t {\n\tcase apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID:\n\t\treturn nil\n\tcase apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_DECIDER:\n\t\treturn types.ContinueAsNewInitiatorDecider.Ptr()\n\tcase apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_RETRY_POLICY:\n\t\treturn types.ContinueAsNewInitiatorRetryPolicy.Ptr()\n\tcase apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_CRON_SCHEDULE:\n\t\treturn types.ContinueAsNewInitiatorCronSchedule.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromContinueAsNewWorkflowExecutionDecisionAttributes(t *types.ContinueAsNewWorkflowExecutionDecisionAttributes) *apiv1.ContinueAsNewWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tBackoffStartInterval:         secondsToDuration(t.BackoffStartIntervalInSeconds),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tInitiator:                    FromContinueAsNewInitiator(t.Initiator),\n\t\tFailure:                      FromFailure(t.FailureReason, t.FailureDetails),\n\t\tLastCompletionResult:         FromPayload(t.LastCompletionResult),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tHeader:                       FromHeader(t.Header),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tJitterStart:                  secondsToDuration(t.JitterStartSeconds),\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc ToContinueAsNewWorkflowExecutionDecisionAttributes(t *apiv1.ContinueAsNewWorkflowExecutionDecisionAttributes) *types.ContinueAsNewWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tBackoffStartIntervalInSeconds:       durationToSeconds(t.BackoffStartInterval),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       ToFailureReason(t.Failure),\n\t\tFailureDetails:                      ToFailureDetails(t.Failure),\n\t\tLastCompletionResult:                ToPayload(t.LastCompletionResult),\n\t\tCronSchedule:                        t.CronSchedule,\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tJitterStartSeconds:                  durationToSeconds(t.JitterStart),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromCountWorkflowExecutionsRequest(t *types.CountWorkflowExecutionsRequest) *apiv1.CountWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CountWorkflowExecutionsRequest{\n\t\tDomain: t.Domain,\n\t\tQuery:  t.Query,\n\t}\n}\n\nfunc ToCountWorkflowExecutionsRequest(t *apiv1.CountWorkflowExecutionsRequest) *types.CountWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountWorkflowExecutionsRequest{\n\t\tDomain: t.Domain,\n\t\tQuery:  t.Query,\n\t}\n}\n\nfunc FromCountWorkflowExecutionsResponse(t *types.CountWorkflowExecutionsResponse) *apiv1.CountWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CountWorkflowExecutionsResponse{\n\t\tCount: t.Count,\n\t}\n}\n\nfunc ToCountWorkflowExecutionsResponse(t *apiv1.CountWorkflowExecutionsResponse) *types.CountWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountWorkflowExecutionsResponse{\n\t\tCount: t.Count,\n\t}\n}\n\nfunc FromDataBlob(t *types.DataBlob) *apiv1.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DataBlob{\n\t\tEncodingType: FromEncodingType(t.EncodingType),\n\t\tData:         t.Data,\n\t}\n}\n\nfunc ToDataBlob(t *apiv1.DataBlob) *types.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DataBlob{\n\t\tEncodingType: ToEncodingType(t.EncodingType),\n\t\tData:         t.Data,\n\t}\n}\n\nfunc FromDecisionTaskCompletedEventAttributes(t *types.DecisionTaskCompletedEventAttributes) *apiv1.DecisionTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DecisionTaskCompletedEventAttributes{\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tIdentity:         t.Identity,\n\t\tBinaryChecksum:   t.BinaryChecksum,\n\t\tExecutionContext: t.ExecutionContext,\n\t}\n}\n\nfunc ToDecisionTaskCompletedEventAttributes(t *apiv1.DecisionTaskCompletedEventAttributes) *types.DecisionTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskCompletedEventAttributes{\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tStartedEventID:   t.StartedEventId,\n\t\tIdentity:         t.Identity,\n\t\tBinaryChecksum:   t.BinaryChecksum,\n\t\tExecutionContext: t.ExecutionContext,\n\t}\n}\n\nfunc FromDecisionTaskFailedCause(t *types.DecisionTaskFailedCause) apiv1.DecisionTaskFailedCause {\n\tif t == nil {\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.DecisionTaskFailedCauseUnhandledDecision:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_UNHANDLED_DECISION\n\tcase types.DecisionTaskFailedCauseBadScheduleActivityAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadRequestCancelActivityAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadStartTimerAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_START_TIMER_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadCancelTimerAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_TIMER_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadRecordMarkerAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_RECORD_MARKER_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadContinueAsNewAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CONTINUE_AS_NEW_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseStartTimerDuplicateID:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_START_TIMER_DUPLICATE_ID\n\tcase types.DecisionTaskFailedCauseResetStickyTasklist:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_RESET_STICKY_TASK_LIST\n\tcase types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_WORKFLOW_WORKER_UNHANDLED_FAILURE\n\tcase types.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseBadStartChildExecutionAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_START_CHILD_EXECUTION_ATTRIBUTES\n\tcase types.DecisionTaskFailedCauseForceCloseDecision:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_FORCE_CLOSE_DECISION\n\tcase types.DecisionTaskFailedCauseFailoverCloseDecision:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_FAILOVER_CLOSE_DECISION\n\tcase types.DecisionTaskFailedCauseBadSignalInputSize:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_INPUT_SIZE\n\tcase types.DecisionTaskFailedCauseResetWorkflow:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_RESET_WORKFLOW\n\tcase types.DecisionTaskFailedCauseBadBinary:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_BINARY\n\tcase types.DecisionTaskFailedCauseScheduleActivityDuplicateID:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_SCHEDULE_ACTIVITY_DUPLICATE_ID\n\tcase types.DecisionTaskFailedCauseBadSearchAttributes:\n\t\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SEARCH_ATTRIBUTES\n\t}\n\treturn apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID\n}\n\nfunc ToDecisionTaskFailedCause(t apiv1.DecisionTaskFailedCause) *types.DecisionTaskFailedCause {\n\tswitch t {\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_UNHANDLED_DECISION:\n\t\treturn types.DecisionTaskFailedCauseUnhandledDecision.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SCHEDULE_ACTIVITY_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadScheduleActivityAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadRequestCancelActivityAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_START_TIMER_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadStartTimerAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_TIMER_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadCancelTimerAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_RECORD_MARKER_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadRecordMarkerAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_CONTINUE_AS_NEW_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadContinueAsNewAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_START_TIMER_DUPLICATE_ID:\n\t\treturn types.DecisionTaskFailedCauseStartTimerDuplicateID.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_RESET_STICKY_TASK_LIST:\n\t\treturn types.DecisionTaskFailedCauseResetStickyTasklist.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_WORKFLOW_WORKER_UNHANDLED_FAILURE:\n\t\treturn types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_START_CHILD_EXECUTION_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadStartChildExecutionAttributes.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_FORCE_CLOSE_DECISION:\n\t\treturn types.DecisionTaskFailedCauseForceCloseDecision.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_FAILOVER_CLOSE_DECISION:\n\t\treturn types.DecisionTaskFailedCauseFailoverCloseDecision.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_INPUT_SIZE:\n\t\treturn types.DecisionTaskFailedCauseBadSignalInputSize.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_RESET_WORKFLOW:\n\t\treturn types.DecisionTaskFailedCauseResetWorkflow.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_BINARY:\n\t\treturn types.DecisionTaskFailedCauseBadBinary.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_SCHEDULE_ACTIVITY_DUPLICATE_ID:\n\t\treturn types.DecisionTaskFailedCauseScheduleActivityDuplicateID.Ptr()\n\tcase apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_BAD_SEARCH_ATTRIBUTES:\n\t\treturn types.DecisionTaskFailedCauseBadSearchAttributes.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromDecisionTaskFailedEventAttributes(t *types.DecisionTaskFailedEventAttributes) *apiv1.DecisionTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tCause:            FromDecisionTaskFailedCause(t.Cause),\n\t\tFailure:          FromFailure(t.Reason, t.Details),\n\t\tIdentity:         t.Identity,\n\t\tBaseRunId:        t.BaseRunID,\n\t\tNewRunId:         t.NewRunID,\n\t\tForkEventVersion: t.ForkEventVersion,\n\t\tBinaryChecksum:   t.BinaryChecksum,\n\t\tRequestId:        t.RequestID,\n\t}\n}\n\nfunc ToDecisionTaskFailedEventAttributes(t *apiv1.DecisionTaskFailedEventAttributes) *types.DecisionTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tStartedEventID:   t.StartedEventId,\n\t\tCause:            ToDecisionTaskFailedCause(t.Cause),\n\t\tReason:           ToFailureReason(t.Failure),\n\t\tDetails:          ToFailureDetails(t.Failure),\n\t\tIdentity:         t.Identity,\n\t\tBaseRunID:        t.BaseRunId,\n\t\tNewRunID:         t.NewRunId,\n\t\tForkEventVersion: t.ForkEventVersion,\n\t\tBinaryChecksum:   t.BinaryChecksum,\n\t\tRequestID:        t.RequestId,\n\t}\n}\n\nfunc FromDecisionTaskScheduledEventAttributes(t *types.DecisionTaskScheduledEventAttributes) *apiv1.DecisionTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DecisionTaskScheduledEventAttributes{\n\t\tTaskList:            FromTaskList(t.TaskList),\n\t\tStartToCloseTimeout: secondsToDuration(t.StartToCloseTimeoutSeconds),\n\t\tAttempt:             int32(t.Attempt),\n\t}\n}\n\nfunc ToDecisionTaskScheduledEventAttributes(t *apiv1.DecisionTaskScheduledEventAttributes) *types.DecisionTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskScheduledEventAttributes{\n\t\tTaskList:                   ToTaskList(t.TaskList),\n\t\tStartToCloseTimeoutSeconds: durationToSeconds(t.StartToCloseTimeout),\n\t\tAttempt:                    int64(t.Attempt),\n\t}\n}\n\nfunc FromDecisionTaskStartedEventAttributes(t *types.DecisionTaskStartedEventAttributes) *apiv1.DecisionTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tIdentity:         t.Identity,\n\t\tRequestId:        t.RequestID,\n\t}\n}\n\nfunc ToDecisionTaskStartedEventAttributes(t *apiv1.DecisionTaskStartedEventAttributes) *types.DecisionTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tIdentity:         t.Identity,\n\t\tRequestID:        t.RequestId,\n\t}\n}\n\nfunc FromDecisionTaskTimedOutEventAttributes(t *types.DecisionTaskTimedOutEventAttributes) *apiv1.DecisionTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventId: t.ScheduledEventID,\n\t\tStartedEventId:   t.StartedEventID,\n\t\tTimeoutType:      FromTimeoutType(t.TimeoutType),\n\t\tBaseRunId:        t.BaseRunID,\n\t\tNewRunId:         t.NewRunID,\n\t\tForkEventVersion: t.ForkEventVersion,\n\t\tReason:           t.Reason,\n\t\tCause:            FromDecisionTaskTimedOutCause(t.Cause),\n\t\tRequestId:        t.RequestID,\n\t}\n}\n\nfunc ToDecisionTaskTimedOutEventAttributes(t *apiv1.DecisionTaskTimedOutEventAttributes) *types.DecisionTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventID: t.ScheduledEventId,\n\t\tStartedEventID:   t.StartedEventId,\n\t\tTimeoutType:      ToTimeoutType(t.TimeoutType),\n\t\tBaseRunID:        t.BaseRunId,\n\t\tNewRunID:         t.NewRunId,\n\t\tForkEventVersion: t.ForkEventVersion,\n\t\tReason:           t.Reason,\n\t\tCause:            ToDecisionTaskTimedOutCause(t.Cause),\n\t\tRequestID:        t.RequestId,\n\t}\n}\n\nfunc FromDeleteDomainRequest(t *types.DeleteDomainRequest) *apiv1.DeleteDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DeleteDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n}\n\nfunc ToDeleteDomainRequest(t *apiv1.DeleteDomainRequest) *types.DeleteDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeleteDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n}\n\nfunc FromDeprecateDomainRequest(t *types.DeprecateDomainRequest) *apiv1.DeprecateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DeprecateDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n}\n\nfunc ToDeprecateDomainRequest(t *apiv1.DeprecateDomainRequest) *types.DeprecateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeprecateDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n}\n\nfunc FromDescribeDomainRequest(t *types.DescribeDomainRequest) *apiv1.DescribeDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.UUID != nil {\n\t\treturn &apiv1.DescribeDomainRequest{DescribeBy: &apiv1.DescribeDomainRequest_Id{Id: *t.UUID}}\n\t}\n\tif t.Name != nil {\n\t\treturn &apiv1.DescribeDomainRequest{DescribeBy: &apiv1.DescribeDomainRequest_Name{Name: *t.Name}}\n\t}\n\t// TODO: Remove this panic and decide on an error behaviour\n\tpanic(\"neither oneof field is set for DescribeDomainRequest\")\n}\n\nfunc ToDescribeDomainRequest(t *apiv1.DescribeDomainRequest) *types.DescribeDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch describeBy := t.DescribeBy.(type) {\n\tcase *apiv1.DescribeDomainRequest_Id:\n\t\treturn &types.DescribeDomainRequest{UUID: common.StringPtr(describeBy.Id)}\n\tcase *apiv1.DescribeDomainRequest_Name:\n\t\treturn &types.DescribeDomainRequest{Name: common.StringPtr(describeBy.Name)}\n\t}\n\t// TODO: Remove this panic and decide on an error behaviour\n\tpanic(\"neither oneof field is set for DescribeDomainRequest\")\n}\n\nfunc FromDescribeDomainResponseDomain(t *types.DescribeDomainResponse) *apiv1.Domain {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tdomain := apiv1.Domain{\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tIsGlobalDomain:  t.IsGlobalDomain,\n\t}\n\tif info := t.DomainInfo; info != nil {\n\t\tdomain.Id = info.UUID\n\t\tdomain.Name = info.Name\n\t\tdomain.Status = FromDomainStatus(info.Status)\n\t\tdomain.Description = info.Description\n\t\tdomain.OwnerEmail = info.OwnerEmail\n\t\tdomain.Data = info.Data\n\t}\n\tif config := t.Configuration; config != nil {\n\t\tdomain.IsolationGroups = FromIsolationGroupConfig(config.IsolationGroups)\n\t\tdomain.WorkflowExecutionRetentionPeriod = daysToDuration(&config.WorkflowExecutionRetentionPeriodInDays)\n\t\tdomain.BadBinaries = FromBadBinaries(config.BadBinaries)\n\t\tdomain.HistoryArchivalStatus = FromArchivalStatus(config.HistoryArchivalStatus)\n\t\tdomain.HistoryArchivalUri = config.HistoryArchivalURI\n\t\tdomain.VisibilityArchivalStatus = FromArchivalStatus(config.VisibilityArchivalStatus)\n\t\tdomain.VisibilityArchivalUri = config.VisibilityArchivalURI\n\t\tdomain.AsyncWorkflowConfig = FromDomainAsyncWorkflowConfiguraton(config.AsyncWorkflowConfig)\n\t}\n\tif repl := t.ReplicationConfiguration; repl != nil {\n\t\tdomain.ActiveClusterName = repl.ActiveClusterName\n\t\tdomain.Clusters = FromClusterReplicationConfigurationArray(repl.Clusters)\n\t\tdomain.ActiveClusters = FromActiveClusters(repl.ActiveClusters)\n\t}\n\tif info := t.GetFailoverInfo(); info != nil {\n\t\tdomain.FailoverInfo = FromFailoverInfo(t.GetFailoverInfo())\n\t}\n\treturn &domain\n}\n\nfunc FromDescribeDomainResponse(t *types.DescribeDomainResponse) *apiv1.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeDomainResponse{\n\t\tDomain: FromDescribeDomainResponseDomain(t),\n\t}\n}\n\nfunc ToDescribeDomainResponseDomain(t *apiv1.Domain) *types.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:        t.Name,\n\t\t\tStatus:      ToDomainStatus(t.Status),\n\t\t\tDescription: t.Description,\n\t\t\tOwnerEmail:  t.OwnerEmail,\n\t\t\tData:        t.Data,\n\t\t\tUUID:        t.Id,\n\t\t},\n\t\tConfiguration: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Default(durationToDays(t.WorkflowExecutionRetentionPeriod)),\n\t\t\tEmitMetric:                             true,\n\t\t\tBadBinaries:                            ToBadBinaries(t.BadBinaries),\n\t\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.HistoryArchivalStatus),\n\t\t\tHistoryArchivalURI:                     t.HistoryArchivalUri,\n\t\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.VisibilityArchivalStatus),\n\t\t\tVisibilityArchivalURI:                  t.VisibilityArchivalUri,\n\t\t\tIsolationGroups:                        ToIsolationGroupConfig(t.IsolationGroups),\n\t\t\tAsyncWorkflowConfig:                    ToDomainAsyncWorkflowConfiguraton(t.AsyncWorkflowConfig),\n\t\t},\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: t.ActiveClusterName,\n\t\t\tClusters:          ToClusterReplicationConfigurationArray(t.Clusters),\n\t\t\tActiveClusters:    ToActiveClusters(t.ActiveClusters),\n\t\t},\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tIsGlobalDomain:  t.IsGlobalDomain,\n\t\tFailoverInfo:    ToFailoverInfo(t.FailoverInfo),\n\t}\n}\n\nfunc ToDescribeDomainResponse(t *apiv1.DescribeDomainResponse) *types.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tresponse := ToDescribeDomainResponseDomain(t.Domain)\n\treturn response\n}\n\nfunc FromFailoverInfo(t *types.FailoverInfo) *apiv1.FailoverInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.FailoverInfo{\n\t\tFailoverVersion:         t.GetFailoverVersion(),\n\t\tFailoverStartTimestamp:  unixNanoToTime(&t.FailoverStartTimestamp),\n\t\tFailoverExpireTimestamp: unixNanoToTime(&t.FailoverExpireTimestamp),\n\t\tCompletedShardCount:     t.GetCompletedShardCount(),\n\t\tPendingShards:           t.GetPendingShards(),\n\t}\n}\n\nfunc ToFailoverInfo(t *apiv1.FailoverInfo) *types.FailoverInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tstartTs := timeToUnixNano(t.GetFailoverStartTimestamp())\n\texpireTs := timeToUnixNano(t.GetFailoverExpireTimestamp())\n\tif startTs == nil || expireTs == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverInfo{\n\t\tFailoverVersion:         t.GetFailoverVersion(),\n\t\tFailoverStartTimestamp:  *startTs,\n\t\tFailoverExpireTimestamp: *expireTs,\n\t\tCompletedShardCount:     t.GetCompletedShardCount(),\n\t\tPendingShards:           t.GetPendingShards(),\n\t}\n}\n\nfunc FromDescribeTaskListRequest(t *types.DescribeTaskListRequest) *apiv1.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeTaskListRequest{\n\t\tDomain:                t.Domain,\n\t\tTaskList:              FromTaskList(t.TaskList),\n\t\tTaskListType:          FromTaskListType(t.TaskListType),\n\t\tIncludeTaskListStatus: t.IncludeTaskListStatus,\n\t}\n}\n\nfunc ToDescribeTaskListRequest(t *apiv1.DescribeTaskListRequest) *types.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeTaskListRequest{\n\t\tDomain:                t.Domain,\n\t\tTaskList:              ToTaskList(t.TaskList),\n\t\tTaskListType:          ToTaskListType(t.TaskListType),\n\t\tIncludeTaskListStatus: t.IncludeTaskListStatus,\n\t}\n}\n\nfunc FromDescribeTaskListResponse(t *types.DescribeTaskListResponse) *apiv1.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeTaskListResponse{\n\t\tPollers:         FromPollerInfoArray(t.Pollers),\n\t\tTaskListStatus:  FromTaskListStatus(t.TaskListStatus),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(t.PartitionConfig),\n\t\tTaskList:        FromTaskList(t.TaskList),\n\t}\n}\n\nfunc ToDescribeTaskListResponse(t *apiv1.DescribeTaskListResponse) *types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers:         ToPollerInfoArray(t.Pollers),\n\t\tTaskListStatus:  ToTaskListStatus(t.TaskListStatus),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(t.PartitionConfig),\n\t\tTaskList:        ToTaskList(t.TaskList),\n\t}\n}\n\nfunc FromDescribeWorkflowExecutionRequest(t *types.DescribeWorkflowExecutionRequest) *apiv1.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeWorkflowExecutionRequest{\n\t\tDomain:                t.Domain,\n\t\tWorkflowExecution:     FromWorkflowExecution(t.Execution),\n\t\tQueryConsistencyLevel: FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc ToDescribeWorkflowExecutionRequest(t *apiv1.DescribeWorkflowExecutionRequest) *types.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeWorkflowExecutionRequest{\n\t\tDomain:                t.Domain,\n\t\tExecution:             ToWorkflowExecution(t.WorkflowExecution),\n\t\tQueryConsistencyLevel: ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc FromDescribeWorkflowExecutionResponse(t *types.DescribeWorkflowExecutionResponse) *apiv1.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: FromWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  FromWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      FromPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        FromPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        FromPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\nfunc ToDescribeWorkflowExecutionResponse(t *apiv1.DescribeWorkflowExecutionResponse) *types.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: ToWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  ToWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      ToPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        ToPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        ToPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\nfunc FromDiagnoseWorkflowExecutionRequest(t *types.DiagnoseWorkflowExecutionRequest) *apiv1.DiagnoseWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DiagnoseWorkflowExecutionRequest{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: FromWorkflowExecution(t.GetWorkflowExecution()),\n\t\tIdentity:          t.GetIdentity(),\n\t}\n}\n\nfunc ToDiagnoseWorkflowExecutionRequest(t *apiv1.DiagnoseWorkflowExecutionRequest) *types.DiagnoseWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.GetWorkflowExecution()),\n\t\tIdentity:          t.GetIdentity(),\n\t}\n}\n\nfunc FromDiagnoseWorkflowExecutionResponse(t *types.DiagnoseWorkflowExecutionResponse) *apiv1.DiagnoseWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DiagnoseWorkflowExecutionResponse{\n\t\tDomain:                      t.GetDomain(),\n\t\tDiagnosticWorkflowExecution: FromWorkflowExecution(t.GetDiagnosticWorkflowExecution()),\n\t}\n}\n\nfunc ToDiagnoseWorkflowExecutionResponse(t *apiv1.DiagnoseWorkflowExecutionResponse) *types.DiagnoseWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain:                      t.GetDomain(),\n\t\tDiagnosticWorkflowExecution: ToWorkflowExecution(t.GetDiagnosticWorkflowExecution()),\n\t}\n}\n\nfunc FromDomainStatus(t *types.DomainStatus) apiv1.DomainStatus {\n\tif t == nil {\n\t\treturn apiv1.DomainStatus_DOMAIN_STATUS_INVALID\n\t}\n\tswitch *t {\n\tcase types.DomainStatusRegistered:\n\t\treturn apiv1.DomainStatus_DOMAIN_STATUS_REGISTERED\n\tcase types.DomainStatusDeprecated:\n\t\treturn apiv1.DomainStatus_DOMAIN_STATUS_DEPRECATED\n\tcase types.DomainStatusDeleted:\n\t\treturn apiv1.DomainStatus_DOMAIN_STATUS_DELETED\n\t}\n\treturn apiv1.DomainStatus_DOMAIN_STATUS_INVALID\n}\n\nfunc ToDomainStatus(t apiv1.DomainStatus) *types.DomainStatus {\n\tswitch t {\n\tcase apiv1.DomainStatus_DOMAIN_STATUS_INVALID:\n\t\treturn nil\n\tcase apiv1.DomainStatus_DOMAIN_STATUS_REGISTERED:\n\t\treturn types.DomainStatusRegistered.Ptr()\n\tcase apiv1.DomainStatus_DOMAIN_STATUS_DEPRECATED:\n\t\treturn types.DomainStatusDeprecated.Ptr()\n\tcase apiv1.DomainStatus_DOMAIN_STATUS_DELETED:\n\t\treturn types.DomainStatusDeleted.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromEncodingType(t *types.EncodingType) apiv1.EncodingType {\n\tif t == nil {\n\t\treturn apiv1.EncodingType_ENCODING_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.EncodingTypeThriftRW:\n\t\treturn apiv1.EncodingType_ENCODING_TYPE_THRIFTRW\n\tcase types.EncodingTypeJSON:\n\t\treturn apiv1.EncodingType_ENCODING_TYPE_JSON\n\t}\n\treturn apiv1.EncodingType_ENCODING_TYPE_INVALID\n}\n\nfunc ToEncodingType(t apiv1.EncodingType) *types.EncodingType {\n\tswitch t {\n\tcase apiv1.EncodingType_ENCODING_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.EncodingType_ENCODING_TYPE_THRIFTRW:\n\t\treturn types.EncodingTypeThriftRW.Ptr()\n\tcase apiv1.EncodingType_ENCODING_TYPE_JSON:\n\t\treturn types.EncodingTypeJSON.Ptr()\n\tcase apiv1.EncodingType_ENCODING_TYPE_PROTO3:\n\t\treturn nil\n\t}\n\treturn nil\n}\n\nfunc FromEventFilterType(t *types.HistoryEventFilterType) apiv1.EventFilterType {\n\tif t == nil {\n\t\treturn apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.HistoryEventFilterTypeAllEvent:\n\t\treturn apiv1.EventFilterType_EVENT_FILTER_TYPE_ALL_EVENT\n\tcase types.HistoryEventFilterTypeCloseEvent:\n\t\treturn apiv1.EventFilterType_EVENT_FILTER_TYPE_CLOSE_EVENT\n\t}\n\treturn apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID\n}\n\nfunc ToEventFilterType(t apiv1.EventFilterType) *types.HistoryEventFilterType {\n\tswitch t {\n\tcase apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.EventFilterType_EVENT_FILTER_TYPE_ALL_EVENT:\n\t\treturn types.HistoryEventFilterTypeAllEvent.Ptr()\n\tcase apiv1.EventFilterType_EVENT_FILTER_TYPE_CLOSE_EVENT:\n\t\treturn types.HistoryEventFilterTypeCloseEvent.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromExternalWorkflowExecutionCancelRequestedEventAttributes(t *types.ExternalWorkflowExecutionCancelRequestedEventAttributes) *apiv1.ExternalWorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc ToExternalWorkflowExecutionCancelRequestedEventAttributes(t *apiv1.ExternalWorkflowExecutionCancelRequestedEventAttributes) *types.ExternalWorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromExternalWorkflowExecutionSignaledEventAttributes(t *types.ExternalWorkflowExecutionSignaledEventAttributes) *apiv1.ExternalWorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventId:  t.InitiatedEventID,\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tControl:           t.Control,\n\t}\n}\n\nfunc ToExternalWorkflowExecutionSignaledEventAttributes(t *apiv1.ExternalWorkflowExecutionSignaledEventAttributes) *types.ExternalWorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tControl:           t.Control,\n\t}\n}\n\nfunc FromFailWorkflowExecutionDecisionAttributes(t *types.FailWorkflowExecutionDecisionAttributes) *apiv1.FailWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.FailWorkflowExecutionDecisionAttributes{\n\t\tFailure: FromFailure(t.Reason, t.Details),\n\t}\n}\n\nfunc ToFailWorkflowExecutionDecisionAttributes(t *apiv1.FailWorkflowExecutionDecisionAttributes) *types.FailWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  ToFailureReason(t.Failure),\n\t\tDetails: ToFailureDetails(t.Failure),\n\t}\n}\n\nfunc FromGetClusterInfoResponse(t *types.ClusterInfo) *apiv1.GetClusterInfoResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.GetClusterInfoResponse{\n\t\tSupportedClientVersions: FromSupportedClientVersions(t.SupportedClientVersions),\n\t}\n}\n\nfunc ToGetClusterInfoResponse(t *apiv1.GetClusterInfoResponse) *types.ClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterInfo{\n\t\tSupportedClientVersions: ToSupportedClientVersions(t.SupportedClientVersions),\n\t}\n}\n\nfunc FromGetSearchAttributesResponse(t *types.GetSearchAttributesResponse) *apiv1.GetSearchAttributesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.GetSearchAttributesResponse{\n\t\tKeys: FromIndexedValueTypeMap(t.Keys),\n\t}\n}\n\nfunc ToGetSearchAttributesResponse(t *apiv1.GetSearchAttributesResponse) *types.GetSearchAttributesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetSearchAttributesResponse{\n\t\tKeys: ToIndexedValueTypeMap(t.Keys),\n\t}\n}\n\nfunc FromGetWorkflowExecutionHistoryRequest(t *types.GetWorkflowExecutionHistoryRequest) *apiv1.GetWorkflowExecutionHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 t.Domain,\n\t\tWorkflowExecution:      FromWorkflowExecution(t.Execution),\n\t\tPageSize:               t.MaximumPageSize,\n\t\tNextPageToken:          t.NextPageToken,\n\t\tWaitForNewEvent:        t.WaitForNewEvent,\n\t\tHistoryEventFilterType: FromEventFilterType(t.HistoryEventFilterType),\n\t\tSkipArchival:           t.SkipArchival,\n\t\tQueryConsistencyLevel:  FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc ToGetWorkflowExecutionHistoryRequest(t *apiv1.GetWorkflowExecutionHistoryRequest) *types.GetWorkflowExecutionHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 t.Domain,\n\t\tExecution:              ToWorkflowExecution(t.WorkflowExecution),\n\t\tMaximumPageSize:        t.PageSize,\n\t\tNextPageToken:          t.NextPageToken,\n\t\tWaitForNewEvent:        t.WaitForNewEvent,\n\t\tHistoryEventFilterType: ToEventFilterType(t.HistoryEventFilterType),\n\t\tSkipArchival:           t.SkipArchival,\n\t\tQueryConsistencyLevel:  ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc FromGetWorkflowExecutionHistoryResponse(t *types.GetWorkflowExecutionHistoryResponse) *apiv1.GetWorkflowExecutionHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       FromHistory(t.History),\n\t\tRawHistory:    FromDataBlobArray(t.RawHistory),\n\t\tNextPageToken: t.NextPageToken,\n\t\tArchived:      t.Archived,\n\t}\n}\n\nfunc ToGetWorkflowExecutionHistoryResponse(t *apiv1.GetWorkflowExecutionHistoryResponse) *types.GetWorkflowExecutionHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       ToHistory(t.History),\n\t\tRawHistory:    ToDataBlobArray(t.RawHistory),\n\t\tNextPageToken: t.NextPageToken,\n\t\tArchived:      t.Archived,\n\t}\n}\n\nfunc FromHeader(t *types.Header) *apiv1.Header {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.Header{\n\t\tFields: FromPayloadMap(t.Fields),\n\t}\n}\n\nfunc ToHeader(t *apiv1.Header) *types.Header {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Header{\n\t\tFields: ToPayloadMap(t.Fields),\n\t}\n}\n\nfunc FromHealthResponse(t *types.HealthStatus) *apiv1.HealthResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.HealthResponse{\n\t\tOk:      t.Ok,\n\t\tMessage: t.Msg,\n\t}\n}\n\nfunc ToHealthResponse(t *apiv1.HealthResponse) *types.HealthStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HealthStatus{\n\t\tOk:  t.Ok,\n\t\tMsg: t.Message,\n\t}\n}\n\nfunc FromHistory(t *types.History) *apiv1.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.History{\n\t\tEvents: FromHistoryEventArray(t.Events),\n\t}\n}\n\nfunc ToHistory(t *apiv1.History) *types.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.History{\n\t\tEvents: ToHistoryEventArray(t.Events),\n\t}\n}\n\nfunc FromIndexedValueType(t types.IndexedValueType) apiv1.IndexedValueType {\n\tswitch t {\n\tcase types.IndexedValueTypeString:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_STRING\n\tcase types.IndexedValueTypeKeyword:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_KEYWORD\n\tcase types.IndexedValueTypeInt:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INT\n\tcase types.IndexedValueTypeDouble:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_DOUBLE\n\tcase types.IndexedValueTypeBool:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_BOOL\n\tcase types.IndexedValueTypeDatetime:\n\t\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_DATETIME\n\t}\n\treturn apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INVALID\n}\n\nfunc ToIndexedValueType(t apiv1.IndexedValueType) types.IndexedValueType {\n\tswitch t {\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INVALID:\n\t\treturn types.IndexedValueTypeString\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_STRING:\n\t\treturn types.IndexedValueTypeString\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_KEYWORD:\n\t\treturn types.IndexedValueTypeKeyword\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INT:\n\t\treturn types.IndexedValueTypeInt\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_DOUBLE:\n\t\treturn types.IndexedValueTypeDouble\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_BOOL:\n\t\treturn types.IndexedValueTypeBool\n\tcase apiv1.IndexedValueType_INDEXED_VALUE_TYPE_DATETIME:\n\t\treturn types.IndexedValueTypeDatetime\n\t}\n\treturn types.IndexedValueTypeString\n}\n\nfunc FromListArchivedWorkflowExecutionsRequest(t *types.ListArchivedWorkflowExecutionsRequest) *apiv1.ListArchivedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc ToListArchivedWorkflowExecutionsRequest(t *apiv1.ListArchivedWorkflowExecutionsRequest) *types.ListArchivedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc FromListArchivedWorkflowExecutionsResponse(t *types.ListArchivedWorkflowExecutionsResponse) *apiv1.ListArchivedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListArchivedWorkflowExecutionsResponse(t *apiv1.ListArchivedWorkflowExecutionsResponse) *types.ListArchivedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListClosedWorkflowExecutionsResponse(t *types.ListClosedWorkflowExecutionsResponse) *apiv1.ListClosedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListClosedWorkflowExecutionsResponse(t *apiv1.ListClosedWorkflowExecutionsResponse) *types.ListClosedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListDomainsRequest(t *types.ListDomainsRequest) *apiv1.ListDomainsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListDomainsRequest{\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListDomainsRequest(t *apiv1.ListDomainsRequest) *types.ListDomainsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDomainsRequest{\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListDomainsResponse(t *types.ListDomainsResponse) *apiv1.ListDomainsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListDomainsResponse{\n\t\tDomains:       FromDescribeDomainResponseArray(t.Domains),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListDomainsResponse(t *apiv1.ListDomainsResponse) *types.ListDomainsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDomainsResponse{\n\t\tDomains:       ToDescribeDomainResponseArray(t.Domains),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListOpenWorkflowExecutionsResponse(t *types.ListOpenWorkflowExecutionsResponse) *apiv1.ListOpenWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListOpenWorkflowExecutionsResponse(t *apiv1.ListOpenWorkflowExecutionsResponse) *types.ListOpenWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListTaskListPartitionsRequest(t *types.ListTaskListPartitionsRequest) *apiv1.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListTaskListPartitionsRequest{\n\t\tDomain:   t.Domain,\n\t\tTaskList: FromTaskList(t.TaskList),\n\t}\n}\n\nfunc ToListTaskListPartitionsRequest(t *apiv1.ListTaskListPartitionsRequest) *types.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListTaskListPartitionsRequest{\n\t\tDomain:   t.Domain,\n\t\tTaskList: ToTaskList(t.TaskList),\n\t}\n}\n\nfunc FromListTaskListPartitionsResponse(t *types.ListTaskListPartitionsResponse) *apiv1.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: FromTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: FromTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\nfunc ToListTaskListPartitionsResponse(t *apiv1.ListTaskListPartitionsResponse) *types.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: ToTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: ToTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\nfunc FromGetTaskListsByDomainRequest(t *types.GetTaskListsByDomainRequest) *apiv1.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.GetTaskListsByDomainRequest{\n\t\tDomain: t.Domain,\n\t}\n}\n\nfunc ToGetTaskListsByDomainRequest(t *apiv1.GetTaskListsByDomainRequest) *types.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetTaskListsByDomainRequest{\n\t\tDomain: t.Domain,\n\t}\n}\n\nfunc FromGetTaskListsByDomainResponse(t *types.GetTaskListsByDomainResponse) *apiv1.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &apiv1.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: FromDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: FromDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\nfunc ToGetTaskListsByDomainResponse(t *apiv1.GetTaskListsByDomainResponse) *types.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: ToDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: ToDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\nfunc FromDescribeTaskListResponseMap(t map[string]*types.DescribeTaskListResponse) map[string]*apiv1.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*apiv1.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = FromDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\nfunc ToDescribeTaskListResponseMap(t map[string]*apiv1.DescribeTaskListResponse) map[string]*types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*types.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = ToDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\nfunc FromListWorkflowExecutionsRequest(t *types.ListWorkflowExecutionsRequest) *apiv1.ListWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc ToListWorkflowExecutionsRequest(t *apiv1.ListWorkflowExecutionsRequest) *types.ListWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc FromListWorkflowExecutionsResponse(t *types.ListWorkflowExecutionsResponse) *apiv1.ListWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListWorkflowExecutionsResponse(t *apiv1.ListWorkflowExecutionsResponse) *types.ListWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromMarkerRecordedEventAttributes(t *types.MarkerRecordedEventAttributes) *apiv1.MarkerRecordedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   t.MarkerName,\n\t\tDetails:                      FromPayload(t.Details),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tHeader:                       FromHeader(t.Header),\n\t}\n}\n\nfunc ToMarkerRecordedEventAttributes(t *apiv1.MarkerRecordedEventAttributes) *types.MarkerRecordedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   t.MarkerName,\n\t\tDetails:                      ToPayload(t.Details),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tHeader:                       ToHeader(t.Header),\n\t}\n}\n\nfunc FromMemo(t *types.Memo) *apiv1.Memo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.Memo{\n\t\tFields: FromPayloadMap(t.Fields),\n\t}\n}\n\nfunc ToMemo(t *apiv1.Memo) *types.Memo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Memo{\n\t\tFields: ToPayloadMap(t.Fields),\n\t}\n}\n\nfunc FromParentClosePolicy(t *types.ParentClosePolicy) apiv1.ParentClosePolicy {\n\tif t == nil {\n\t\treturn apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID\n\t}\n\tswitch *t {\n\tcase types.ParentClosePolicyAbandon:\n\t\treturn apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_ABANDON\n\tcase types.ParentClosePolicyRequestCancel:\n\t\treturn apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_REQUEST_CANCEL\n\tcase types.ParentClosePolicyTerminate:\n\t\treturn apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_TERMINATE\n\t}\n\treturn apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID\n}\n\nfunc ToParentClosePolicy(t apiv1.ParentClosePolicy) *types.ParentClosePolicy {\n\tswitch t {\n\tcase apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID:\n\t\treturn nil\n\tcase apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_ABANDON:\n\t\treturn types.ParentClosePolicyAbandon.Ptr()\n\tcase apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_REQUEST_CANCEL:\n\t\treturn types.ParentClosePolicyRequestCancel.Ptr()\n\tcase apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_TERMINATE:\n\t\treturn types.ParentClosePolicyTerminate.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromPendingActivityInfo(t *types.PendingActivityInfo) *apiv1.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PendingActivityInfo{\n\t\tActivityId:            t.ActivityID,\n\t\tActivityType:          FromActivityType(t.ActivityType),\n\t\tState:                 FromPendingActivityState(t.State),\n\t\tHeartbeatDetails:      FromPayload(t.HeartbeatDetails),\n\t\tLastHeartbeatTime:     unixNanoToTime(t.LastHeartbeatTimestamp),\n\t\tLastStartedTime:       unixNanoToTime(t.LastStartedTimestamp),\n\t\tAttempt:               t.Attempt,\n\t\tMaximumAttempts:       t.MaximumAttempts,\n\t\tScheduledTime:         unixNanoToTime(t.ScheduledTimestamp),\n\t\tExpirationTime:        unixNanoToTime(t.ExpirationTimestamp),\n\t\tLastFailure:           FromFailure(t.LastFailureReason, t.LastFailureDetails),\n\t\tLastWorkerIdentity:    t.LastWorkerIdentity,\n\t\tStartedWorkerIdentity: t.StartedWorkerIdentity,\n\t\tScheduleId:            t.ScheduleID,\n\t}\n}\n\nfunc ToPendingActivityInfo(t *apiv1.PendingActivityInfo) *types.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingActivityInfo{\n\t\tActivityID:             t.ActivityId,\n\t\tActivityType:           ToActivityType(t.ActivityType),\n\t\tState:                  ToPendingActivityState(t.State),\n\t\tHeartbeatDetails:       ToPayload(t.HeartbeatDetails),\n\t\tLastHeartbeatTimestamp: timeToUnixNano(t.LastHeartbeatTime),\n\t\tLastStartedTimestamp:   timeToUnixNano(t.LastStartedTime),\n\t\tAttempt:                t.Attempt,\n\t\tMaximumAttempts:        t.MaximumAttempts,\n\t\tScheduledTimestamp:     timeToUnixNano(t.ScheduledTime),\n\t\tExpirationTimestamp:    timeToUnixNano(t.ExpirationTime),\n\t\tLastFailureReason:      ToFailureReason(t.LastFailure),\n\t\tLastFailureDetails:     ToFailureDetails(t.LastFailure),\n\t\tLastWorkerIdentity:     t.LastWorkerIdentity,\n\t\tStartedWorkerIdentity:  t.StartedWorkerIdentity,\n\t\tScheduleID:             t.ScheduleId,\n\t}\n}\n\nfunc FromPendingActivityState(t *types.PendingActivityState) apiv1.PendingActivityState {\n\tif t == nil {\n\t\treturn apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID\n\t}\n\tswitch *t {\n\tcase types.PendingActivityStateScheduled:\n\t\treturn apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_SCHEDULED\n\tcase types.PendingActivityStateStarted:\n\t\treturn apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_STARTED\n\tcase types.PendingActivityStateCancelRequested:\n\t\treturn apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_CANCEL_REQUESTED\n\t}\n\treturn apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID\n}\n\nfunc ToPendingActivityState(t apiv1.PendingActivityState) *types.PendingActivityState {\n\tswitch t {\n\tcase apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID:\n\t\treturn nil\n\tcase apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_SCHEDULED:\n\t\treturn types.PendingActivityStateScheduled.Ptr()\n\tcase apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_STARTED:\n\t\treturn types.PendingActivityStateStarted.Ptr()\n\tcase apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_CANCEL_REQUESTED:\n\t\treturn types.PendingActivityStateCancelRequested.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromPendingChildExecutionInfo(t *types.PendingChildExecutionInfo) *apiv1.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PendingChildExecutionInfo{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tWorkflowTypeName:  t.WorkflowTypeName,\n\t\tInitiatedId:       t.InitiatedID,\n\t\tParentClosePolicy: FromParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\nfunc ToPendingChildExecutionInfo(t *apiv1.PendingChildExecutionInfo) *types.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingChildExecutionInfo{\n\t\tDomain:            t.Domain,\n\t\tWorkflowID:        ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:             ToRunID(t.WorkflowExecution),\n\t\tWorkflowTypeName:  t.WorkflowTypeName,\n\t\tInitiatedID:       t.InitiatedId,\n\t\tParentClosePolicy: ToParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\nfunc FromPendingDecisionInfo(t *types.PendingDecisionInfo) *apiv1.PendingDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PendingDecisionInfo{\n\t\tState:                 FromPendingDecisionState(t.State),\n\t\tScheduledTime:         unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:           unixNanoToTime(t.StartedTimestamp),\n\t\tAttempt:               int32(t.Attempt),\n\t\tOriginalScheduledTime: unixNanoToTime(t.OriginalScheduledTimestamp),\n\t\tScheduleId:            t.ScheduleID,\n\t}\n}\n\nfunc ToPendingDecisionInfo(t *apiv1.PendingDecisionInfo) *types.PendingDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingDecisionInfo{\n\t\tState:                      ToPendingDecisionState(t.State),\n\t\tScheduledTimestamp:         timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:           timeToUnixNano(t.StartedTime),\n\t\tAttempt:                    int64(t.Attempt),\n\t\tOriginalScheduledTimestamp: timeToUnixNano(t.OriginalScheduledTime),\n\t\tScheduleID:                 t.ScheduleId,\n\t}\n}\n\nfunc FromPendingDecisionState(t *types.PendingDecisionState) apiv1.PendingDecisionState {\n\tif t == nil {\n\t\treturn apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID\n\t}\n\tswitch *t {\n\tcase types.PendingDecisionStateScheduled:\n\t\treturn apiv1.PendingDecisionState_PENDING_DECISION_STATE_SCHEDULED\n\tcase types.PendingDecisionStateStarted:\n\t\treturn apiv1.PendingDecisionState_PENDING_DECISION_STATE_STARTED\n\t}\n\treturn apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID\n}\n\nfunc ToPendingDecisionState(t apiv1.PendingDecisionState) *types.PendingDecisionState {\n\tswitch t {\n\tcase apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID:\n\t\treturn nil\n\tcase apiv1.PendingDecisionState_PENDING_DECISION_STATE_SCHEDULED:\n\t\treturn types.PendingDecisionStateScheduled.Ptr()\n\tcase apiv1.PendingDecisionState_PENDING_DECISION_STATE_STARTED:\n\t\treturn types.PendingDecisionStateStarted.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromPollForActivityTaskRequest(t *types.PollForActivityTaskRequest) *apiv1.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PollForActivityTaskRequest{\n\t\tDomain:           t.Domain,\n\t\tTaskList:         FromTaskList(t.TaskList),\n\t\tIdentity:         t.Identity,\n\t\tTaskListMetadata: FromTaskListMetadata(t.TaskListMetadata),\n\t}\n}\n\nfunc ToPollForActivityTaskRequest(t *apiv1.PollForActivityTaskRequest) *types.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForActivityTaskRequest{\n\t\tDomain:           t.Domain,\n\t\tTaskList:         ToTaskList(t.TaskList),\n\t\tIdentity:         t.Identity,\n\t\tTaskListMetadata: ToTaskListMetadata(t.TaskListMetadata),\n\t}\n}\n\nfunc FromPollForActivityTaskResponse(t *types.PollForActivityTaskResponse) *apiv1.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PollForActivityTaskResponse{\n\t\tTaskToken:                  t.TaskToken,\n\t\tWorkflowExecution:          FromWorkflowExecution(t.WorkflowExecution),\n\t\tActivityId:                 t.ActivityID,\n\t\tActivityType:               FromActivityType(t.ActivityType),\n\t\tInput:                      FromPayload(t.Input),\n\t\tScheduledTime:              unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:                unixNanoToTime(t.StartedTimestamp),\n\t\tScheduleToCloseTimeout:     secondsToDuration(t.ScheduleToCloseTimeoutSeconds),\n\t\tStartToCloseTimeout:        secondsToDuration(t.StartToCloseTimeoutSeconds),\n\t\tHeartbeatTimeout:           secondsToDuration(t.HeartbeatTimeoutSeconds),\n\t\tAttempt:                    t.Attempt,\n\t\tScheduledTimeOfThisAttempt: unixNanoToTime(t.ScheduledTimestampOfThisAttempt),\n\t\tHeartbeatDetails:           FromPayload(t.HeartbeatDetails),\n\t\tWorkflowType:               FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:             t.WorkflowDomain,\n\t\tHeader:                     FromHeader(t.Header),\n\t\tAutoConfigHint:             FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc ToPollForActivityTaskResponse(t *apiv1.PollForActivityTaskResponse) *types.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               ToWorkflowExecution(t.WorkflowExecution),\n\t\tActivityID:                      t.ActivityId,\n\t\tActivityType:                    ToActivityType(t.ActivityType),\n\t\tInput:                           ToPayload(t.Input),\n\t\tScheduledTimestamp:              timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:                timeToUnixNano(t.StartedTime),\n\t\tScheduleToCloseTimeoutSeconds:   durationToSeconds(t.ScheduleToCloseTimeout),\n\t\tStartToCloseTimeoutSeconds:      durationToSeconds(t.StartToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:         durationToSeconds(t.HeartbeatTimeout),\n\t\tAttempt:                         t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: timeToUnixNano(t.ScheduledTimeOfThisAttempt),\n\t\tHeartbeatDetails:                ToPayload(t.HeartbeatDetails),\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.WorkflowDomain,\n\t\tHeader:                          ToHeader(t.Header),\n\t\tAutoConfigHint:                  ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc FromPollForDecisionTaskRequest(t *types.PollForDecisionTaskRequest) *apiv1.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PollForDecisionTaskRequest{\n\t\tDomain:         t.Domain,\n\t\tTaskList:       FromTaskList(t.TaskList),\n\t\tIdentity:       t.Identity,\n\t\tBinaryChecksum: t.BinaryChecksum,\n\t}\n}\n\nfunc ToPollForDecisionTaskRequest(t *apiv1.PollForDecisionTaskRequest) *types.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForDecisionTaskRequest{\n\t\tDomain:         t.Domain,\n\t\tTaskList:       ToTaskList(t.TaskList),\n\t\tIdentity:       t.Identity,\n\t\tBinaryChecksum: t.BinaryChecksum,\n\t}\n}\n\nfunc FromPollForDecisionTaskResponse(t *types.PollForDecisionTaskResponse) *apiv1.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    fromInt64Value(t.PreviousStartedEventID),\n\t\tStartedEventId:            t.StartedEventID,\n\t\tAttempt:                   t.Attempt,\n\t\tBacklogCountHint:          t.BacklogCountHint,\n\t\tHistory:                   FromHistory(t.History),\n\t\tNextPageToken:             t.NextPageToken,\n\t\tQuery:                     FromWorkflowQuery(t.Query),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tScheduledTime:             unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:               unixNanoToTime(t.StartedTimestamp),\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tNextEventId:               t.NextEventID,\n\t\tTotalHistoryBytes:         t.TotalHistoryBytes,\n\t\tAutoConfigHint:            FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc ToPollForDecisionTaskResponse(t *apiv1.PollForDecisionTaskResponse) *types.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    toInt64Value(t.PreviousStartedEventId),\n\t\tStartedEventID:            t.StartedEventId,\n\t\tAttempt:                   t.Attempt,\n\t\tBacklogCountHint:          t.BacklogCountHint,\n\t\tHistory:                   ToHistory(t.History),\n\t\tNextPageToken:             t.NextPageToken,\n\t\tQuery:                     ToWorkflowQuery(t.Query),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tScheduledTimestamp:        timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:          timeToUnixNano(t.StartedTime),\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tNextEventID:               t.NextEventId,\n\t\tTotalHistoryBytes:         t.TotalHistoryBytes,\n\t\tAutoConfigHint:            ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc FromPollerInfo(t *types.PollerInfo) *apiv1.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PollerInfo{\n\t\tLastAccessTime: unixNanoToTime(t.LastAccessTime),\n\t\tIdentity:       t.Identity,\n\t\tRatePerSecond:  t.RatePerSecond,\n\t}\n}\n\nfunc ToPollerInfo(t *apiv1.PollerInfo) *types.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollerInfo{\n\t\tLastAccessTime: timeToUnixNano(t.LastAccessTime),\n\t\tIdentity:       t.Identity,\n\t\tRatePerSecond:  t.RatePerSecond,\n\t}\n}\n\nfunc FromQueryConsistencyLevel(t *types.QueryConsistencyLevel) apiv1.QueryConsistencyLevel {\n\tif t == nil {\n\t\treturn apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID\n\t}\n\tswitch *t {\n\tcase types.QueryConsistencyLevelEventual:\n\t\treturn apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_EVENTUAL\n\tcase types.QueryConsistencyLevelStrong:\n\t\treturn apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_STRONG\n\t}\n\treturn apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID\n}\n\nfunc ToQueryConsistencyLevel(t apiv1.QueryConsistencyLevel) *types.QueryConsistencyLevel {\n\tswitch t {\n\tcase apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID:\n\t\treturn nil\n\tcase apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_EVENTUAL:\n\t\treturn types.QueryConsistencyLevelEventual.Ptr()\n\tcase apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_STRONG:\n\t\treturn types.QueryConsistencyLevelStrong.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromQueryRejectCondition(t *types.QueryRejectCondition) apiv1.QueryRejectCondition {\n\tif t == nil {\n\t\treturn apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID\n\t}\n\tswitch *t {\n\tcase types.QueryRejectConditionNotOpen:\n\t\treturn apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_OPEN\n\tcase types.QueryRejectConditionNotCompletedCleanly:\n\t\treturn apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_COMPLETED_CLEANLY\n\t}\n\treturn apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID\n}\n\nfunc ToQueryRejectCondition(t apiv1.QueryRejectCondition) *types.QueryRejectCondition {\n\tswitch t {\n\tcase apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID:\n\t\treturn nil\n\tcase apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_OPEN:\n\t\treturn types.QueryRejectConditionNotOpen.Ptr()\n\tcase apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_COMPLETED_CLEANLY:\n\t\treturn types.QueryRejectConditionNotCompletedCleanly.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromQueryRejected(t *types.QueryRejected) *apiv1.QueryRejected {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.QueryRejected{\n\t\tCloseStatus: FromWorkflowExecutionCloseStatus(t.CloseStatus),\n\t}\n}\n\nfunc ToQueryRejected(t *apiv1.QueryRejected) *types.QueryRejected {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryRejected{\n\t\tCloseStatus: ToWorkflowExecutionCloseStatus(t.CloseStatus),\n\t}\n}\n\nfunc FromQueryResultType(t *types.QueryResultType) apiv1.QueryResultType {\n\tif t == nil {\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.QueryResultTypeAnswered:\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED\n\tcase types.QueryResultTypeFailed:\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_FAILED\n\t}\n\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID\n}\n\nfunc ToQueryResultType(t apiv1.QueryResultType) *types.QueryResultType {\n\tswitch t {\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED:\n\t\treturn types.QueryResultTypeAnswered.Ptr()\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_FAILED:\n\t\treturn types.QueryResultTypeFailed.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromQueryWorkflowRequest(t *types.QueryWorkflowRequest) *apiv1.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.QueryWorkflowRequest{\n\t\tDomain:                t.Domain,\n\t\tWorkflowExecution:     FromWorkflowExecution(t.Execution),\n\t\tQuery:                 FromWorkflowQuery(t.Query),\n\t\tQueryRejectCondition:  FromQueryRejectCondition(t.QueryRejectCondition),\n\t\tQueryConsistencyLevel: FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc ToQueryWorkflowRequest(t *apiv1.QueryWorkflowRequest) *types.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryWorkflowRequest{\n\t\tDomain:                t.Domain,\n\t\tExecution:             ToWorkflowExecution(t.WorkflowExecution),\n\t\tQuery:                 ToWorkflowQuery(t.Query),\n\t\tQueryRejectCondition:  ToQueryRejectCondition(t.QueryRejectCondition),\n\t\tQueryConsistencyLevel: ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\nfunc FromQueryWorkflowResponse(t *types.QueryWorkflowResponse) *apiv1.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.QueryWorkflowResponse{\n\t\tQueryResult:   FromPayload(t.QueryResult),\n\t\tQueryRejected: FromQueryRejected(t.QueryRejected),\n\t}\n}\n\nfunc ToQueryWorkflowResponse(t *apiv1.QueryWorkflowResponse) *types.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryWorkflowResponse{\n\t\tQueryResult:   ToPayload(t.QueryResult),\n\t\tQueryRejected: ToQueryRejected(t.QueryRejected),\n\t}\n}\n\nfunc FromRecordActivityTaskHeartbeatByIDRequest(t *types.RecordActivityTaskHeartbeatByIDRequest) *apiv1.RecordActivityTaskHeartbeatByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tActivityId:        t.ActivityID,\n\t\tDetails:           FromPayload(t.Details),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc ToRecordActivityTaskHeartbeatByIDRequest(t *apiv1.RecordActivityTaskHeartbeatByIDRequest) *types.RecordActivityTaskHeartbeatByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     t.Domain,\n\t\tWorkflowID: ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:      ToRunID(t.WorkflowExecution),\n\t\tActivityID: t.ActivityId,\n\t\tDetails:    ToPayload(t.Details),\n\t\tIdentity:   t.Identity,\n\t}\n}\n\nfunc FromRecordActivityTaskHeartbeatByIDResponse(t *types.RecordActivityTaskHeartbeatResponse) *apiv1.RecordActivityTaskHeartbeatByIDResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RecordActivityTaskHeartbeatByIDResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc ToRecordActivityTaskHeartbeatByIDResponse(t *apiv1.RecordActivityTaskHeartbeatByIDResponse) *types.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc FromRecordActivityTaskHeartbeatRequest(t *types.RecordActivityTaskHeartbeatRequest) *apiv1.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   FromPayload(t.Details),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc ToRecordActivityTaskHeartbeatRequest(t *apiv1.RecordActivityTaskHeartbeatRequest) *types.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   ToPayload(t.Details),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc FromRecordActivityTaskHeartbeatResponse(t *types.RecordActivityTaskHeartbeatResponse) *apiv1.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc ToRecordActivityTaskHeartbeatResponse(t *apiv1.RecordActivityTaskHeartbeatResponse) *types.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc FromRecordMarkerDecisionAttributes(t *types.RecordMarkerDecisionAttributes) *apiv1.RecordMarkerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RecordMarkerDecisionAttributes{\n\t\tMarkerName: t.MarkerName,\n\t\tDetails:    FromPayload(t.Details),\n\t\tHeader:     FromHeader(t.Header),\n\t}\n}\n\nfunc ToRecordMarkerDecisionAttributes(t *apiv1.RecordMarkerDecisionAttributes) *types.RecordMarkerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordMarkerDecisionAttributes{\n\t\tMarkerName: t.MarkerName,\n\t\tDetails:    ToPayload(t.Details),\n\t\tHeader:     ToHeader(t.Header),\n\t}\n}\n\nfunc FromRegisterDomainRequest(t *types.RegisterDomainRequest) *apiv1.RegisterDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RegisterDomainRequest{\n\t\tName:                             t.Name,\n\t\tDescription:                      t.Description,\n\t\tOwnerEmail:                       t.OwnerEmail,\n\t\tWorkflowExecutionRetentionPeriod: daysToDuration(&t.WorkflowExecutionRetentionPeriodInDays),\n\t\tClusters:                         FromClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusterName:                t.ActiveClusterName,\n\t\tActiveClusters:                   FromActiveClusters(t.ActiveClusters),\n\t\tData:                             t.Data,\n\t\tSecurityToken:                    t.SecurityToken,\n\t\tIsGlobalDomain:                   t.IsGlobalDomain,\n\t\tHistoryArchivalStatus:            FromArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalUri:               t.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:         FromArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalUri:            t.VisibilityArchivalURI,\n\t}\n}\n\nfunc ToRegisterDomainRequest(t *apiv1.RegisterDomainRequest) *types.RegisterDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tdays := durationToDays(t.WorkflowExecutionRetentionPeriod)\n\tif days == nil {\n\t\tdays = common.Int32Ptr(3) // default to 3 days if not set\n\t}\n\treturn &types.RegisterDomainRequest{\n\t\tName:                                   t.Name,\n\t\tDescription:                            t.Description,\n\t\tOwnerEmail:                             t.OwnerEmail,\n\t\tWorkflowExecutionRetentionPeriodInDays: *days,\n\t\tEmitMetric:                             common.BoolPtr(true), // this is a legacy field that doesn't exist in proto and probably can be removed\n\t\tClusters:                               ToClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusterName:                      t.ActiveClusterName,\n\t\tActiveClusters:                         ToActiveClusters(t.ActiveClusters),\n\t\tData:                                   t.Data,\n\t\tSecurityToken:                          t.SecurityToken,\n\t\tIsGlobalDomain:                         t.IsGlobalDomain,\n\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:                     t.HistoryArchivalUri,\n\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:                  t.VisibilityArchivalUri,\n\t}\n}\n\nfunc FromRequestCancelActivityTaskDecisionAttributes(t *types.RequestCancelActivityTaskDecisionAttributes) *apiv1.RequestCancelActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelActivityTaskDecisionAttributes{\n\t\tActivityId: t.ActivityID,\n\t}\n}\n\nfunc ToRequestCancelActivityTaskDecisionAttributes(t *apiv1.RequestCancelActivityTaskDecisionAttributes) *types.RequestCancelActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelActivityTaskDecisionAttributes{\n\t\tActivityID: t.ActivityId,\n\t}\n}\n\nfunc FromRequestCancelActivityTaskFailedEventAttributes(t *types.RequestCancelActivityTaskFailedEventAttributes) *apiv1.RequestCancelActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityId:                   t.ActivityID,\n\t\tCause:                        t.Cause,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToRequestCancelActivityTaskFailedEventAttributes(t *apiv1.RequestCancelActivityTaskFailedEventAttributes) *types.RequestCancelActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityID:                   t.ActivityId,\n\t\tCause:                        t.Cause,\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nfunc FromRequestCancelExternalWorkflowExecutionDecisionAttributes(t *types.RequestCancelExternalWorkflowExecutionDecisionAttributes) *apiv1.RequestCancelExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToRequestCancelExternalWorkflowExecutionDecisionAttributes(t *apiv1.RequestCancelExternalWorkflowExecutionDecisionAttributes) *types.RequestCancelExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowID:        ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:             ToRunID(t.WorkflowExecution),\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(t *types.RequestCancelExternalWorkflowExecutionFailedEventAttributes) *apiv1.RequestCancelExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        FromCancelExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventId:             t.InitiatedEventID,\n\t\tControl:                      t.Control,\n\t}\n}\n\nfunc ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(t *apiv1.RequestCancelExternalWorkflowExecutionFailedEventAttributes) *types.RequestCancelExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        ToCancelExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventID:             t.InitiatedEventId,\n\t\tControl:                      t.Control,\n\t}\n}\n\nfunc FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t *types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) *apiv1.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t *apiv1.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) *types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromRequestCancelWorkflowExecutionRequest(t *types.RequestCancelWorkflowExecutionRequest) *apiv1.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RequestCancelWorkflowExecutionRequest{\n\t\tDomain:              t.Domain,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:            t.Identity,\n\t\tRequestId:           t.RequestID,\n\t\tCause:               t.Cause,\n\t\tFirstExecutionRunId: t.FirstExecutionRunID,\n\t}\n}\n\nfunc ToRequestCancelWorkflowExecutionRequest(t *apiv1.RequestCancelWorkflowExecutionRequest) *types.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain:              t.Domain,\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:            t.Identity,\n\t\tRequestID:           t.RequestId,\n\t\tCause:               t.Cause,\n\t\tFirstExecutionRunID: t.FirstExecutionRunId,\n\t}\n}\n\nfunc FromResetPointInfo(t *types.ResetPointInfo) *apiv1.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetPointInfo{\n\t\tBinaryChecksum:           t.BinaryChecksum,\n\t\tRunId:                    t.RunID,\n\t\tFirstDecisionCompletedId: t.FirstDecisionCompletedID,\n\t\tCreatedTime:              unixNanoToTime(t.CreatedTimeNano),\n\t\tExpiringTime:             unixNanoToTime(t.ExpiringTimeNano),\n\t\tResettable:               t.Resettable,\n\t}\n}\n\nfunc ToResetPointInfo(t *apiv1.ResetPointInfo) *types.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetPointInfo{\n\t\tBinaryChecksum:           t.BinaryChecksum,\n\t\tRunID:                    t.RunId,\n\t\tFirstDecisionCompletedID: t.FirstDecisionCompletedId,\n\t\tCreatedTimeNano:          timeToUnixNano(t.CreatedTime),\n\t\tExpiringTimeNano:         timeToUnixNano(t.ExpiringTime),\n\t\tResettable:               t.Resettable,\n\t}\n}\n\nfunc FromResetPoints(t *types.ResetPoints) *apiv1.ResetPoints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetPoints{\n\t\tPoints: FromResetPointInfoArray(t.Points),\n\t}\n}\n\nfunc ToResetPoints(t *apiv1.ResetPoints) *types.ResetPoints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetPoints{\n\t\tPoints: ToResetPointInfoArray(t.Points),\n\t}\n}\n\nfunc FromResetStickyTaskListRequest(t *types.ResetStickyTaskListRequest) *apiv1.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetStickyTaskListRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\nfunc ToResetStickyTaskListRequest(t *apiv1.ResetStickyTaskListRequest) *types.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetStickyTaskListRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromResetWorkflowExecutionRequest(t *types.ResetWorkflowExecutionRequest) *apiv1.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetWorkflowExecutionRequest{\n\t\tDomain:                t.Domain,\n\t\tWorkflowExecution:     FromWorkflowExecution(t.WorkflowExecution),\n\t\tReason:                t.Reason,\n\t\tDecisionFinishEventId: t.DecisionFinishEventID,\n\t\tRequestId:             t.RequestID,\n\t\tSkipSignalReapply:     t.SkipSignalReapply,\n\t}\n}\n\nfunc ToResetWorkflowExecutionRequest(t *apiv1.ResetWorkflowExecutionRequest) *types.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetWorkflowExecutionRequest{\n\t\tDomain:                t.Domain,\n\t\tWorkflowExecution:     ToWorkflowExecution(t.WorkflowExecution),\n\t\tReason:                t.Reason,\n\t\tDecisionFinishEventID: t.DecisionFinishEventId,\n\t\tRequestID:             t.RequestId,\n\t\tSkipSignalReapply:     t.SkipSignalReapply,\n\t}\n}\n\nfunc FromResetWorkflowExecutionResponse(t *types.ResetWorkflowExecutionResponse) *apiv1.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToResetWorkflowExecutionResponse(t *apiv1.ResetWorkflowExecutionResponse) *types.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc ToRefreshWorkflowTasksRequest(t *apiv1.RefreshWorkflowTasksRequest) *types.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RefreshWorkflowTasksRequest{\n\t\tDomain:    t.Domain,\n\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromRefreshWorkflowTasksRequest(t *types.RefreshWorkflowTasksRequest) *apiv1.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RefreshWorkflowTasksRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\nfunc FromRespondActivityTaskCanceledByIDRequest(t *types.RespondActivityTaskCanceledByIDRequest) *apiv1.RespondActivityTaskCanceledByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tActivityId:        t.ActivityID,\n\t\tDetails:           FromPayload(t.Details),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskCanceledByIDRequest(t *apiv1.RespondActivityTaskCanceledByIDRequest) *types.RespondActivityTaskCanceledByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     t.Domain,\n\t\tWorkflowID: ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:      ToRunID(t.WorkflowExecution),\n\t\tActivityID: t.ActivityId,\n\t\tDetails:    ToPayload(t.Details),\n\t\tIdentity:   t.Identity,\n\t}\n}\n\nfunc FromRespondActivityTaskCanceledRequest(t *types.RespondActivityTaskCanceledRequest) *apiv1.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   FromPayload(t.Details),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskCanceledRequest(t *apiv1.RespondActivityTaskCanceledRequest) *types.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   ToPayload(t.Details),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc FromRespondActivityTaskCompletedByIDRequest(t *types.RespondActivityTaskCompletedByIDRequest) *apiv1.RespondActivityTaskCompletedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tActivityId:        t.ActivityID,\n\t\tResult:            FromPayload(t.Result),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskCompletedByIDRequest(t *apiv1.RespondActivityTaskCompletedByIDRequest) *types.RespondActivityTaskCompletedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     t.Domain,\n\t\tWorkflowID: ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:      ToRunID(t.WorkflowExecution),\n\t\tActivityID: t.ActivityId,\n\t\tResult:     ToPayload(t.Result),\n\t\tIdentity:   t.Identity,\n\t}\n}\n\nfunc FromRespondActivityTaskCompletedRequest(t *types.RespondActivityTaskCompletedRequest) *apiv1.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tResult:    FromPayload(t.Result),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskCompletedRequest(t *apiv1.RespondActivityTaskCompletedRequest) *types.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tResult:    ToPayload(t.Result),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc FromRespondActivityTaskFailedByIDRequest(t *types.RespondActivityTaskFailedByIDRequest) *apiv1.RespondActivityTaskFailedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tActivityId:        t.ActivityID,\n\t\tFailure:           FromFailure(t.Reason, t.Details),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskFailedByIDRequest(t *apiv1.RespondActivityTaskFailedByIDRequest) *types.RespondActivityTaskFailedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     t.Domain,\n\t\tWorkflowID: ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:      ToRunID(t.WorkflowExecution),\n\t\tActivityID: t.ActivityId,\n\t\tReason:     ToFailureReason(t.Failure),\n\t\tDetails:    ToFailureDetails(t.Failure),\n\t\tIdentity:   t.Identity,\n\t}\n}\n\nfunc FromRespondActivityTaskFailedRequest(t *types.RespondActivityTaskFailedRequest) *apiv1.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondActivityTaskFailedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tFailure:   FromFailure(t.Reason, t.Details),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc ToRespondActivityTaskFailedRequest(t *apiv1.RespondActivityTaskFailedRequest) *types.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tReason:    ToFailureReason(t.Failure),\n\t\tDetails:   ToFailureDetails(t.Failure),\n\t\tIdentity:  t.Identity,\n\t}\n}\n\nfunc FromRespondDecisionTaskCompletedRequest(t *types.RespondDecisionTaskCompletedRequest) *apiv1.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  t.TaskToken,\n\t\tDecisions:                  FromDecisionArray(t.Decisions),\n\t\tExecutionContext:           t.ExecutionContext,\n\t\tIdentity:                   t.Identity,\n\t\tStickyAttributes:           FromStickyExecutionAttributes(t.StickyAttributes),\n\t\tReturnNewDecisionTask:      t.ReturnNewDecisionTask,\n\t\tForceCreateNewDecisionTask: t.ForceCreateNewDecisionTask,\n\t\tBinaryChecksum:             t.BinaryChecksum,\n\t\tQueryResults:               FromWorkflowQueryResultMap(t.QueryResults),\n\t}\n}\n\nfunc ToRespondDecisionTaskCompletedRequest(t *apiv1.RespondDecisionTaskCompletedRequest) *types.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  t.TaskToken,\n\t\tDecisions:                  ToDecisionArray(t.Decisions),\n\t\tExecutionContext:           t.ExecutionContext,\n\t\tIdentity:                   t.Identity,\n\t\tStickyAttributes:           ToStickyExecutionAttributes(t.StickyAttributes),\n\t\tReturnNewDecisionTask:      t.ReturnNewDecisionTask,\n\t\tForceCreateNewDecisionTask: t.ForceCreateNewDecisionTask,\n\t\tBinaryChecksum:             t.BinaryChecksum,\n\t\tQueryResults:               ToWorkflowQueryResultMap(t.QueryResults),\n\t}\n}\n\nfunc FromRespondDecisionTaskCompletedResponse(t *types.RespondDecisionTaskCompletedResponse) *apiv1.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondDecisionTaskCompletedResponse{\n\t\tDecisionTask:                FromPollForDecisionTaskResponse(t.DecisionTask),\n\t\tActivitiesToDispatchLocally: FromActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\nfunc ToRespondDecisionTaskCompletedResponse(t *apiv1.RespondDecisionTaskCompletedResponse) *types.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskCompletedResponse{\n\t\tDecisionTask:                ToPollForDecisionTaskResponse(t.DecisionTask),\n\t\tActivitiesToDispatchLocally: ToActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\nfunc FromRespondDecisionTaskFailedRequest(t *types.RespondDecisionTaskFailedRequest) *apiv1.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondDecisionTaskFailedRequest{\n\t\tTaskToken:      t.TaskToken,\n\t\tCause:          FromDecisionTaskFailedCause(t.Cause),\n\t\tDetails:        FromPayload(t.Details),\n\t\tIdentity:       t.Identity,\n\t\tBinaryChecksum: t.BinaryChecksum,\n\t}\n}\n\nfunc ToRespondDecisionTaskFailedRequest(t *apiv1.RespondDecisionTaskFailedRequest) *types.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken:      t.TaskToken,\n\t\tCause:          ToDecisionTaskFailedCause(t.Cause),\n\t\tDetails:        ToPayload(t.Details),\n\t\tIdentity:       t.Identity,\n\t\tBinaryChecksum: t.BinaryChecksum,\n\t}\n}\n\nfunc FromRespondQueryTaskCompletedRequest(t *types.RespondQueryTaskCompletedRequest) *apiv1.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RespondQueryTaskCompletedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tResult: &apiv1.WorkflowQueryResult{\n\t\t\tResultType:   FromQueryTaskCompletedType(t.CompletedType),\n\t\t\tAnswer:       FromPayload(t.QueryResult),\n\t\t\tErrorMessage: t.ErrorMessage,\n\t\t},\n\t\tWorkerVersionInfo: FromWorkerVersionInfo(t.WorkerVersionInfo),\n\t}\n}\n\nfunc ToRespondQueryTaskCompletedRequest(t *apiv1.RespondQueryTaskCompletedRequest) *types.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondQueryTaskCompletedRequest{\n\t\tTaskToken:         t.TaskToken,\n\t\tCompletedType:     ToQueryTaskCompletedType(t.Result.ResultType),\n\t\tQueryResult:       ToPayload(t.Result.Answer),\n\t\tErrorMessage:      t.Result.ErrorMessage,\n\t\tWorkerVersionInfo: ToWorkerVersionInfo(t.WorkerVersionInfo),\n\t}\n}\n\nfunc FromQueryTaskCompletedType(t *types.QueryTaskCompletedType) apiv1.QueryResultType {\n\tif t == nil {\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.QueryTaskCompletedTypeCompleted:\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED\n\tcase types.QueryTaskCompletedTypeFailed:\n\t\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_FAILED\n\t}\n\treturn apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID\n}\n\nfunc ToQueryTaskCompletedType(t apiv1.QueryResultType) *types.QueryTaskCompletedType {\n\tswitch t {\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED:\n\t\treturn types.QueryTaskCompletedTypeCompleted.Ptr()\n\tcase apiv1.QueryResultType_QUERY_RESULT_TYPE_FAILED:\n\t\treturn types.QueryTaskCompletedTypeFailed.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromRetryPolicy(t *types.RetryPolicy) *apiv1.RetryPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RetryPolicy{\n\t\tInitialInterval:          secondsToDuration(common.Int32Ptr(t.InitialIntervalInSeconds)),\n\t\tBackoffCoefficient:       t.BackoffCoefficient,\n\t\tMaximumInterval:          secondsToDuration(common.Int32Ptr(t.MaximumIntervalInSeconds)),\n\t\tMaximumAttempts:          t.MaximumAttempts,\n\t\tNonRetryableErrorReasons: t.NonRetriableErrorReasons,\n\t\tExpirationInterval:       secondsToDuration(common.Int32Ptr(t.ExpirationIntervalInSeconds)),\n\t}\n}\n\nfunc ToRetryPolicy(t *apiv1.RetryPolicy) *types.RetryPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    common.Int32Default(durationToSeconds(t.InitialInterval)),\n\t\tBackoffCoefficient:          t.BackoffCoefficient,\n\t\tMaximumIntervalInSeconds:    common.Int32Default(durationToSeconds(t.MaximumInterval)),\n\t\tMaximumAttempts:             t.MaximumAttempts,\n\t\tNonRetriableErrorReasons:    t.NonRetryableErrorReasons,\n\t\tExpirationIntervalInSeconds: common.Int32Default(durationToSeconds(t.ExpirationInterval)),\n\t}\n}\n\nfunc FromRestartWorkflowExecutionResponse(t *types.RestartWorkflowExecutionResponse) *apiv1.RestartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RestartWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToRestartWorkflowExecutionRequest(t *apiv1.RestartWorkflowExecutionRequest) *types.RestartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestartWorkflowExecutionRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc FromScanWorkflowExecutionsRequest(t *types.ListWorkflowExecutionsRequest) *apiv1.ScanWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScanWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc ToScanWorkflowExecutionsRequest(t *apiv1.ScanWorkflowExecutionsRequest) *types.ListWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.Query,\n\t}\n}\n\nfunc FromScanWorkflowExecutionsResponse(t *types.ListWorkflowExecutionsResponse) *apiv1.ScanWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScanWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToScanWorkflowExecutionsResponse(t *apiv1.ScanWorkflowExecutionsResponse) *types.ListWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromScheduleActivityTaskDecisionAttributes(t *types.ScheduleActivityTaskDecisionAttributes) *apiv1.ScheduleActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityId:             t.ActivityID,\n\t\tActivityType:           FromActivityType(t.ActivityType),\n\t\tDomain:                 t.Domain,\n\t\tTaskList:               FromTaskList(t.TaskList),\n\t\tInput:                  FromPayload(t.Input),\n\t\tScheduleToCloseTimeout: secondsToDuration(t.ScheduleToCloseTimeoutSeconds),\n\t\tScheduleToStartTimeout: secondsToDuration(t.ScheduleToStartTimeoutSeconds),\n\t\tStartToCloseTimeout:    secondsToDuration(t.StartToCloseTimeoutSeconds),\n\t\tHeartbeatTimeout:       secondsToDuration(t.HeartbeatTimeoutSeconds),\n\t\tRetryPolicy:            FromRetryPolicy(t.RetryPolicy),\n\t\tHeader:                 FromHeader(t.Header),\n\t\tRequestLocalDispatch:   t.RequestLocalDispatch,\n\t}\n}\n\nfunc ToScheduleActivityTaskDecisionAttributes(t *apiv1.ScheduleActivityTaskDecisionAttributes) *types.ScheduleActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    t.ActivityId,\n\t\tActivityType:                  ToActivityType(t.ActivityType),\n\t\tDomain:                        t.Domain,\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tInput:                         ToPayload(t.Input),\n\t\tScheduleToCloseTimeoutSeconds: durationToSeconds(t.ScheduleToCloseTimeout),\n\t\tScheduleToStartTimeoutSeconds: durationToSeconds(t.ScheduleToStartTimeout),\n\t\tStartToCloseTimeoutSeconds:    durationToSeconds(t.StartToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:       durationToSeconds(t.HeartbeatTimeout),\n\t\tRetryPolicy:                   ToRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        ToHeader(t.Header),\n\t\tRequestLocalDispatch:          t.RequestLocalDispatch,\n\t}\n}\n\nfunc FromSearchAttributes(t *types.SearchAttributes) *apiv1.SearchAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SearchAttributes{\n\t\tIndexedFields: FromPayloadMap(t.IndexedFields),\n\t}\n}\n\nfunc ToSearchAttributes(t *apiv1.SearchAttributes) *types.SearchAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SearchAttributes{\n\t\tIndexedFields: ToPayloadMap(t.IndexedFields),\n\t}\n}\n\nfunc FromSignalExternalWorkflowExecutionDecisionAttributes(t *types.SignalExternalWorkflowExecutionDecisionAttributes) *apiv1.SignalExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t\tSignalName:        t.SignalName,\n\t\tInput:             FromPayload(t.Input),\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToSignalExternalWorkflowExecutionDecisionAttributes(t *apiv1.SignalExternalWorkflowExecutionDecisionAttributes) *types.SignalExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.Domain,\n\t\tExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:        t.SignalName,\n\t\tInput:             ToPayload(t.Input),\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromSignalExternalWorkflowExecutionFailedCause(t *types.SignalExternalWorkflowExecutionFailedCause) apiv1.SignalExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\treturn apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\n\tcase types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\treturn apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED\n\t}\n\treturn apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID\n}\n\nfunc ToSignalExternalWorkflowExecutionFailedCause(t apiv1.SignalExternalWorkflowExecutionFailedCause) *types.SignalExternalWorkflowExecutionFailedCause {\n\tswitch t {\n\tcase apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION:\n\t\treturn types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr()\n\tcase apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED:\n\t\treturn types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromSignalExternalWorkflowExecutionFailedEventAttributes(t *types.SignalExternalWorkflowExecutionFailedEventAttributes) *apiv1.SignalExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        FromSignalExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventId:             t.InitiatedEventID,\n\t\tControl:                      t.Control,\n\t}\n}\n\nfunc ToSignalExternalWorkflowExecutionFailedEventAttributes(t *apiv1.SignalExternalWorkflowExecutionFailedEventAttributes) *types.SignalExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        ToSignalExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventID:             t.InitiatedEventId,\n\t\tControl:                      t.Control,\n\t}\n}\n\nfunc FromSignalExternalWorkflowExecutionInitiatedEventAttributes(t *types.SignalExternalWorkflowExecutionInitiatedEventAttributes) *apiv1.SignalExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:                   t.SignalName,\n\t\tInput:                        FromPayload(t.Input),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToSignalExternalWorkflowExecutionInitiatedEventAttributes(t *apiv1.SignalExternalWorkflowExecutionInitiatedEventAttributes) *types.SignalExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tDomain:                       t.Domain,\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:                   t.SignalName,\n\t\tInput:                        ToPayload(t.Input),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionRequest(t *types.SignalWithStartWorkflowExecutionRequest) *apiv1.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalWithStartWorkflowExecutionRequest{\n\t\tStartRequest: &apiv1.StartWorkflowExecutionRequest{\n\t\t\tDomain:                       t.Domain,\n\t\t\tWorkflowId:                   t.WorkflowID,\n\t\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\t\tInput:                        FromPayload(t.Input),\n\t\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\t\tIdentity:                     t.Identity,\n\t\t\tRequestId:                    t.RequestID,\n\t\t\tWorkflowIdReusePolicy:        FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\t\tCronSchedule:                 t.CronSchedule,\n\t\t\tMemo:                         FromMemo(t.Memo),\n\t\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\t\tHeader:                       FromHeader(t.Header),\n\t\t\tDelayStart:                   secondsToDuration(t.DelayStartSeconds),\n\t\t\tJitterStart:                  secondsToDuration(t.JitterStartSeconds),\n\t\t\tFirstRunAt:                   unixNanoToTime(t.FirstRunAtTimestamp),\n\t\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\t},\n\t\tSignalName:  t.SignalName,\n\t\tSignalInput: FromPayload(t.SignalInput),\n\t\tControl:     t.Control,\n\t}\n}\n\nfunc ToSignalWithStartWorkflowExecutionRequest(t *apiv1.SignalWithStartWorkflowExecutionRequest) *types.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              t.StartRequest.Domain,\n\t\tWorkflowID:                          t.StartRequest.WorkflowId,\n\t\tWorkflowType:                        ToWorkflowType(t.StartRequest.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.StartRequest.TaskList),\n\t\tInput:                               ToPayload(t.StartRequest.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.StartRequest.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.StartRequest.TaskStartToCloseTimeout),\n\t\tIdentity:                            t.StartRequest.Identity,\n\t\tRequestID:                           t.StartRequest.RequestId,\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.StartRequest.WorkflowIdReusePolicy),\n\t\tSignalName:                          t.SignalName,\n\t\tSignalInput:                         ToPayload(t.SignalInput),\n\t\tControl:                             t.Control,\n\t\tRetryPolicy:                         ToRetryPolicy(t.StartRequest.RetryPolicy),\n\t\tCronSchedule:                        t.StartRequest.CronSchedule,\n\t\tMemo:                                ToMemo(t.StartRequest.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.StartRequest.SearchAttributes),\n\t\tHeader:                              ToHeader(t.StartRequest.Header),\n\t\tDelayStartSeconds:                   durationToSeconds(t.StartRequest.DelayStart),\n\t\tJitterStartSeconds:                  durationToSeconds(t.StartRequest.JitterStart),\n\t\tFirstRunAtTimestamp:                 timeToUnixNano(t.StartRequest.FirstRunAt),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.StartRequest.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.StartRequest.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionResponse(t *types.StartWorkflowExecutionResponse) *apiv1.SignalWithStartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalWithStartWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToSignalWithStartWorkflowExecutionResponse(t *apiv1.SignalWithStartWorkflowExecutionResponse) *types.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromSignalWorkflowExecutionRequest(t *types.SignalWorkflowExecutionRequest) *apiv1.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalWorkflowExecutionRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:        t.SignalName,\n\t\tSignalInput:       FromPayload(t.Input),\n\t\tIdentity:          t.Identity,\n\t\tRequestId:         t.RequestID,\n\t\tControl:           t.Control,\n\t}\n}\n\nfunc ToSignalWorkflowExecutionRequest(t *apiv1.SignalWorkflowExecutionRequest) *types.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:        t.SignalName,\n\t\tInput:             ToPayload(t.SignalInput),\n\t\tIdentity:          t.Identity,\n\t\tRequestID:         t.RequestId,\n\t\tControl:           t.Control,\n\t}\n}\n\nfunc FromStartChildWorkflowExecutionDecisionAttributes(t *types.StartChildWorkflowExecutionDecisionAttributes) *apiv1.StartChildWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                       t.Domain,\n\t\tWorkflowId:                   t.WorkflowID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tParentClosePolicy:            FromParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                      t.Control,\n\t\tWorkflowIdReusePolicy:        FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tHeader:                       FromHeader(t.Header),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc ToStartChildWorkflowExecutionDecisionAttributes(t *apiv1.StartChildWorkflowExecutionDecisionAttributes) *types.StartChildWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                              t.Domain,\n\t\tWorkflowID:                          t.WorkflowId,\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tParentClosePolicy:                   ToParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.CronSchedule,\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromStartChildWorkflowExecutionFailedEventAttributes(t *types.StartChildWorkflowExecutionFailedEventAttributes) *apiv1.StartChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       t.Domain,\n\t\tWorkflowId:                   t.WorkflowID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tCause:                        FromChildWorkflowExecutionFailedCause(t.Cause),\n\t\tControl:                      t.Control,\n\t\tInitiatedEventId:             t.InitiatedEventID,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToStartChildWorkflowExecutionFailedEventAttributes(t *apiv1.StartChildWorkflowExecutionFailedEventAttributes) *types.StartChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       t.Domain,\n\t\tWorkflowID:                   t.WorkflowId,\n\t\tWorkflowType:                 ToWorkflowType(t.WorkflowType),\n\t\tCause:                        ToChildWorkflowExecutionFailedCause(t.Cause),\n\t\tControl:                      t.Control,\n\t\tInitiatedEventID:             t.InitiatedEventId,\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nfunc FromStartChildWorkflowExecutionInitiatedEventAttributes(t *types.StartChildWorkflowExecutionInitiatedEventAttributes) *apiv1.StartChildWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                       t.Domain,\n\t\tWorkflowId:                   t.WorkflowID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tParentClosePolicy:            FromParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                      t.Control,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tWorkflowIdReusePolicy:        FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tHeader:                       FromHeader(t.Header),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tDelayStart:                   secondsToDuration(t.DelayStartSeconds),\n\t\tJitterStart:                  secondsToDuration(t.JitterStartSeconds),\n\t\tFirstRunAt:                   unixNanoToTime(t.FirstRunAtTimestamp),\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc ToStartChildWorkflowExecutionInitiatedEventAttributes(t *apiv1.StartChildWorkflowExecutionInitiatedEventAttributes) *types.StartChildWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              t.Domain,\n\t\tWorkflowID:                          t.WorkflowId,\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tParentClosePolicy:                   ToParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tDecisionTaskCompletedEventID:        t.DecisionTaskCompletedEventId,\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.CronSchedule,\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tDelayStartSeconds:                   durationToSeconds(t.DelayStart),\n\t\tJitterStartSeconds:                  durationToSeconds(t.JitterStart),\n\t\tFirstRunAtTimestamp:                 timeToUnixNano(t.FirstRunAt),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromStartTimeFilter(t *types.StartTimeFilter) *apiv1.StartTimeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartTimeFilter{\n\t\tEarliestTime: unixNanoToTime(t.EarliestTime),\n\t\tLatestTime:   unixNanoToTime(t.LatestTime),\n\t}\n}\n\nfunc ToStartTimeFilter(t *apiv1.StartTimeFilter) *types.StartTimeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartTimeFilter{\n\t\tEarliestTime: timeToUnixNano(t.EarliestTime),\n\t\tLatestTime:   timeToUnixNano(t.LatestTime),\n\t}\n}\n\nfunc FromStartTimerDecisionAttributes(t *types.StartTimerDecisionAttributes) *apiv1.StartTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartTimerDecisionAttributes{\n\t\tTimerId:            t.TimerID,\n\t\tStartToFireTimeout: secondsToDuration(int64To32(t.StartToFireTimeoutSeconds)),\n\t}\n}\n\nfunc ToStartTimerDecisionAttributes(t *apiv1.StartTimerDecisionAttributes) *types.StartTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartTimerDecisionAttributes{\n\t\tTimerID:                   t.TimerId,\n\t\tStartToFireTimeoutSeconds: int32To64(durationToSeconds(t.StartToFireTimeout)),\n\t}\n}\n\nfunc FromRestartWorkflowExecutionRequest(t *types.RestartWorkflowExecutionRequest) *apiv1.RestartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.RestartWorkflowExecutionRequest{\n\t\tDomain:            t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:          t.Identity,\n\t}\n}\n\nfunc ToRestartWorkflowExecutionResponse(t *apiv1.RestartWorkflowExecutionResponse) *types.RestartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestartWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionAsyncRequest(t *types.SignalWithStartWorkflowExecutionAsyncRequest) *apiv1.SignalWithStartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tRequest: FromSignalWithStartWorkflowExecutionRequest(t.SignalWithStartWorkflowExecutionRequest),\n\t}\n}\n\nfunc ToSignalWithStartWorkflowExecutionAsyncRequest(t *apiv1.SignalWithStartWorkflowExecutionAsyncRequest) *types.SignalWithStartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tSignalWithStartWorkflowExecutionRequest: ToSignalWithStartWorkflowExecutionRequest(t.Request),\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionAsyncResponse(t *types.SignalWithStartWorkflowExecutionAsyncResponse) *apiv1.SignalWithStartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SignalWithStartWorkflowExecutionAsyncResponse{}\n}\n\nfunc ToSignalWithStartWorkflowExecutionAsyncResponse(t *apiv1.SignalWithStartWorkflowExecutionAsyncResponse) *types.SignalWithStartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionAsyncResponse{}\n}\n\nfunc FromStartWorkflowExecutionAsyncRequest(t *types.StartWorkflowExecutionAsyncRequest) *apiv1.StartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartWorkflowExecutionAsyncRequest{\n\t\tRequest: FromStartWorkflowExecutionRequest(t.StartWorkflowExecutionRequest),\n\t}\n}\n\nfunc ToStartWorkflowExecutionAsyncRequest(t *apiv1.StartWorkflowExecutionAsyncRequest) *types.StartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionAsyncRequest{\n\t\tStartWorkflowExecutionRequest: ToStartWorkflowExecutionRequest(t.Request),\n\t}\n}\n\nfunc FromStartWorkflowExecutionAsyncResponse(t *types.StartWorkflowExecutionAsyncResponse) *apiv1.StartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartWorkflowExecutionAsyncResponse{}\n}\n\nfunc ToStartWorkflowExecutionAsyncResponse(t *apiv1.StartWorkflowExecutionAsyncResponse) *types.StartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionAsyncResponse{}\n}\n\nfunc FromStartWorkflowExecutionRequest(t *types.StartWorkflowExecutionRequest) *apiv1.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartWorkflowExecutionRequest{\n\t\tDomain:                       t.Domain,\n\t\tWorkflowId:                   t.WorkflowID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tIdentity:                     t.Identity,\n\t\tRequestId:                    t.RequestID,\n\t\tWorkflowIdReusePolicy:        FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tHeader:                       FromHeader(t.Header),\n\t\tDelayStart:                   secondsToDuration(t.DelayStartSeconds),\n\t\tJitterStart:                  secondsToDuration(t.JitterStartSeconds),\n\t\tFirstRunAt:                   unixNanoToTime(t.FirstRunAtTimeStamp),\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc ToStartWorkflowExecutionRequest(t *apiv1.StartWorkflowExecutionRequest) *types.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              t.Domain,\n\t\tWorkflowID:                          t.WorkflowId,\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tIdentity:                            t.Identity,\n\t\tRequestID:                           t.RequestId,\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.CronSchedule,\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tDelayStartSeconds:                   durationToSeconds(t.DelayStart),\n\t\tJitterStartSeconds:                  durationToSeconds(t.JitterStart),\n\t\tFirstRunAtTimeStamp:                 timeToUnixNano(t.FirstRunAt),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromStartWorkflowExecutionResponse(t *types.StartWorkflowExecutionResponse) *apiv1.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StartWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToStartWorkflowExecutionResponse(t *apiv1.StartWorkflowExecutionResponse) *types.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromStatusFilter(t *types.WorkflowExecutionCloseStatus) *apiv1.StatusFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StatusFilter{\n\t\tStatus: FromWorkflowExecutionCloseStatus(t),\n\t}\n}\n\nfunc ToStatusFilter(t *apiv1.StatusFilter) *types.WorkflowExecutionCloseStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn ToWorkflowExecutionCloseStatus(t.Status)\n}\n\nfunc FromStickyExecutionAttributes(t *types.StickyExecutionAttributes) *apiv1.StickyExecutionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.StickyExecutionAttributes{\n\t\tWorkerTaskList:         FromTaskList(t.WorkerTaskList),\n\t\tScheduleToStartTimeout: secondsToDuration(t.ScheduleToStartTimeoutSeconds),\n\t}\n}\n\nfunc ToStickyExecutionAttributes(t *apiv1.StickyExecutionAttributes) *types.StickyExecutionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StickyExecutionAttributes{\n\t\tWorkerTaskList:                ToTaskList(t.WorkerTaskList),\n\t\tScheduleToStartTimeoutSeconds: durationToSeconds(t.ScheduleToStartTimeout),\n\t}\n}\n\nfunc FromSupportedClientVersions(t *types.SupportedClientVersions) *apiv1.SupportedClientVersions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SupportedClientVersions{\n\t\tGoSdk:   t.GoSdk,\n\t\tJavaSdk: t.JavaSdk,\n\t}\n}\n\nfunc ToSupportedClientVersions(t *apiv1.SupportedClientVersions) *types.SupportedClientVersions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SupportedClientVersions{\n\t\tGoSdk:   t.GoSdk,\n\t\tJavaSdk: t.JavaSdk,\n\t}\n}\n\nfunc FromTaskIDBlock(t *types.TaskIDBlock) *apiv1.TaskIDBlock {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskIDBlock{\n\t\tStartId: t.StartID,\n\t\tEndId:   t.EndID,\n\t}\n}\n\nfunc ToTaskIDBlock(t *apiv1.TaskIDBlock) *types.TaskIDBlock {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskIDBlock{\n\t\tStartID: t.StartId,\n\t\tEndID:   t.EndId,\n\t}\n}\n\nfunc MigrateTaskList(name string, t *apiv1.TaskList) *types.TaskList {\n\tif t == nil && name != \"\" {\n\t\treturn &types.TaskList{Name: name, Kind: types.TaskListKindNormal.Ptr()}\n\t}\n\treturn ToTaskList(t)\n}\n\nfunc FromTaskList(t *types.TaskList) *apiv1.TaskList {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskList{\n\t\tName: t.Name,\n\t\tKind: FromTaskListKind(t.Kind),\n\t}\n}\n\nfunc ToTaskList(t *apiv1.TaskList) *types.TaskList {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskList{\n\t\tName: t.Name,\n\t\tKind: ToTaskListKind(t.Kind),\n\t}\n}\n\nfunc FromTaskListKind(t *types.TaskListKind) apiv1.TaskListKind {\n\tif t == nil {\n\t\treturn apiv1.TaskListKind_TASK_LIST_KIND_INVALID\n\t}\n\tswitch *t {\n\tcase types.TaskListKindNormal:\n\t\treturn apiv1.TaskListKind_TASK_LIST_KIND_NORMAL\n\tcase types.TaskListKindSticky:\n\t\treturn apiv1.TaskListKind_TASK_LIST_KIND_STICKY\n\tcase types.TaskListKindEphemeral:\n\t\treturn apiv1.TaskListKind_TASK_LIST_KIND_EPHEMERAL\n\t}\n\treturn apiv1.TaskListKind_TASK_LIST_KIND_INVALID\n}\n\nfunc ToTaskListKind(t apiv1.TaskListKind) *types.TaskListKind {\n\tswitch t {\n\tcase apiv1.TaskListKind_TASK_LIST_KIND_INVALID:\n\t\treturn nil\n\tcase apiv1.TaskListKind_TASK_LIST_KIND_NORMAL:\n\t\treturn types.TaskListKindNormal.Ptr()\n\tcase apiv1.TaskListKind_TASK_LIST_KIND_STICKY:\n\t\treturn types.TaskListKindSticky.Ptr()\n\tcase apiv1.TaskListKind_TASK_LIST_KIND_EPHEMERAL:\n\t\treturn types.TaskListKindEphemeral.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromTaskListMetadata(t *types.TaskListMetadata) *apiv1.TaskListMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskListMetadata{\n\t\tMaxTasksPerSecond: fromDoubleValue(t.MaxTasksPerSecond),\n\t}\n}\n\nfunc ToTaskListMetadata(t *apiv1.TaskListMetadata) *types.TaskListMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListMetadata{\n\t\tMaxTasksPerSecond: toDoubleValue(t.MaxTasksPerSecond),\n\t}\n}\n\nfunc FromTaskListPartitionMetadata(t *types.TaskListPartitionMetadata) *apiv1.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskListPartitionMetadata{\n\t\tKey:           t.Key,\n\t\tOwnerHostName: t.OwnerHostName,\n\t}\n}\n\nfunc ToTaskListPartitionMetadata(t *apiv1.TaskListPartitionMetadata) *types.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartitionMetadata{\n\t\tKey:           t.Key,\n\t\tOwnerHostName: t.OwnerHostName,\n\t}\n}\n\nfunc FromTaskListStatus(t *types.TaskListStatus) *apiv1.TaskListStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskListStatus{\n\t\tBacklogCountHint:      t.BacklogCountHint,\n\t\tReadLevel:             t.ReadLevel,\n\t\tAckLevel:              t.AckLevel,\n\t\tRatePerSecond:         t.RatePerSecond,\n\t\tTaskIdBlock:           FromTaskIDBlock(t.TaskIDBlock),\n\t\tIsolationGroupMetrics: FromIsolationGroupMetricsMap(t.IsolationGroupMetrics),\n\t\tNewTasksPerSecond:     t.NewTasksPerSecond,\n\t\tEmpty:                 t.Empty,\n\t}\n}\n\nfunc ToTaskListStatus(t *apiv1.TaskListStatus) *types.TaskListStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListStatus{\n\t\tBacklogCountHint:      t.BacklogCountHint,\n\t\tReadLevel:             t.ReadLevel,\n\t\tAckLevel:              t.AckLevel,\n\t\tRatePerSecond:         t.RatePerSecond,\n\t\tTaskIDBlock:           ToTaskIDBlock(t.TaskIdBlock),\n\t\tIsolationGroupMetrics: ToIsolationGroupMetricsMap(t.IsolationGroupMetrics),\n\t\tNewTasksPerSecond:     t.NewTasksPerSecond,\n\t\tEmpty:                 t.Empty,\n\t}\n}\n\nfunc FromIsolationGroupMetrics(t *types.IsolationGroupMetrics) *apiv1.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.IsolationGroupMetrics{\n\t\tNewTasksPerSecond: t.NewTasksPerSecond,\n\t\tPollerCount:       t.PollerCount,\n\t}\n}\n\nfunc ToIsolationGroupMetrics(t *apiv1.IsolationGroupMetrics) *types.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.IsolationGroupMetrics{\n\t\tNewTasksPerSecond: t.NewTasksPerSecond,\n\t\tPollerCount:       t.PollerCount,\n\t}\n}\n\nfunc FromTaskListType(t *types.TaskListType) apiv1.TaskListType {\n\tif t == nil {\n\t\treturn apiv1.TaskListType_TASK_LIST_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.TaskListTypeDecision:\n\t\treturn apiv1.TaskListType_TASK_LIST_TYPE_DECISION\n\tcase types.TaskListTypeActivity:\n\t\treturn apiv1.TaskListType_TASK_LIST_TYPE_ACTIVITY\n\t}\n\treturn apiv1.TaskListType_TASK_LIST_TYPE_INVALID\n}\n\nfunc ToTaskListType(t apiv1.TaskListType) *types.TaskListType {\n\tswitch t {\n\tcase apiv1.TaskListType_TASK_LIST_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.TaskListType_TASK_LIST_TYPE_DECISION:\n\t\treturn types.TaskListTypeDecision.Ptr()\n\tcase apiv1.TaskListType_TASK_LIST_TYPE_ACTIVITY:\n\t\treturn types.TaskListTypeActivity.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromTerminateWorkflowExecutionRequest(t *types.TerminateWorkflowExecutionRequest) *apiv1.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TerminateWorkflowExecutionRequest{\n\t\tDomain:              t.Domain,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tReason:              t.Reason,\n\t\tDetails:             FromPayload(t.Details),\n\t\tIdentity:            t.Identity,\n\t\tFirstExecutionRunId: t.FirstExecutionRunID,\n\t}\n}\n\nfunc ToTerminateWorkflowExecutionRequest(t *apiv1.TerminateWorkflowExecutionRequest) *types.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:              t.Domain,\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tReason:              t.Reason,\n\t\tDetails:             ToPayload(t.Details),\n\t\tIdentity:            t.Identity,\n\t\tFirstExecutionRunID: t.FirstExecutionRunId,\n\t}\n}\n\nfunc FromTimeoutType(t *types.TimeoutType) apiv1.TimeoutType {\n\tif t == nil {\n\t\treturn apiv1.TimeoutType_TIMEOUT_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.TimeoutTypeStartToClose:\n\t\treturn apiv1.TimeoutType_TIMEOUT_TYPE_START_TO_CLOSE\n\tcase types.TimeoutTypeScheduleToStart:\n\t\treturn apiv1.TimeoutType_TIMEOUT_TYPE_SCHEDULE_TO_START\n\tcase types.TimeoutTypeScheduleToClose:\n\t\treturn apiv1.TimeoutType_TIMEOUT_TYPE_SCHEDULE_TO_CLOSE\n\tcase types.TimeoutTypeHeartbeat:\n\t\treturn apiv1.TimeoutType_TIMEOUT_TYPE_HEARTBEAT\n\t}\n\treturn apiv1.TimeoutType_TIMEOUT_TYPE_INVALID\n}\n\nfunc ToTimeoutType(t apiv1.TimeoutType) *types.TimeoutType {\n\tswitch t {\n\tcase apiv1.TimeoutType_TIMEOUT_TYPE_INVALID:\n\t\treturn nil\n\tcase apiv1.TimeoutType_TIMEOUT_TYPE_START_TO_CLOSE:\n\t\treturn types.TimeoutTypeStartToClose.Ptr()\n\tcase apiv1.TimeoutType_TIMEOUT_TYPE_SCHEDULE_TO_START:\n\t\treturn types.TimeoutTypeScheduleToStart.Ptr()\n\tcase apiv1.TimeoutType_TIMEOUT_TYPE_SCHEDULE_TO_CLOSE:\n\t\treturn types.TimeoutTypeScheduleToClose.Ptr()\n\tcase apiv1.TimeoutType_TIMEOUT_TYPE_HEARTBEAT:\n\t\treturn types.TimeoutTypeHeartbeat.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromDecisionTaskTimedOutCause(t *types.DecisionTaskTimedOutCause) apiv1.DecisionTaskTimedOutCause {\n\tif t == nil {\n\t\treturn apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.DecisionTaskTimedOutCauseTimeout:\n\t\treturn apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_TIMEOUT\n\tcase types.DecisionTaskTimedOutCauseReset:\n\t\treturn apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_RESET\n\t}\n\treturn apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID\n}\n\nfunc ToDecisionTaskTimedOutCause(t apiv1.DecisionTaskTimedOutCause) *types.DecisionTaskTimedOutCause {\n\tswitch t {\n\tcase apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID:\n\t\treturn nil\n\tcase apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_TIMEOUT:\n\t\treturn types.DecisionTaskTimedOutCauseTimeout.Ptr()\n\tcase apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_RESET:\n\t\treturn types.DecisionTaskTimedOutCauseReset.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromTimerCanceledEventAttributes(t *types.TimerCanceledEventAttributes) *apiv1.TimerCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TimerCanceledEventAttributes{\n\t\tTimerId:                      t.TimerID,\n\t\tStartedEventId:               t.StartedEventID,\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc ToTimerCanceledEventAttributes(t *apiv1.TimerCanceledEventAttributes) *types.TimerCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerCanceledEventAttributes{\n\t\tTimerID:                      t.TimerId,\n\t\tStartedEventID:               t.StartedEventId,\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tIdentity:                     t.Identity,\n\t}\n}\n\nfunc FromTimerFiredEventAttributes(t *types.TimerFiredEventAttributes) *apiv1.TimerFiredEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TimerFiredEventAttributes{\n\t\tTimerId:        t.TimerID,\n\t\tStartedEventId: t.StartedEventID,\n\t}\n}\n\nfunc ToTimerFiredEventAttributes(t *apiv1.TimerFiredEventAttributes) *types.TimerFiredEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerFiredEventAttributes{\n\t\tTimerID:        t.TimerId,\n\t\tStartedEventID: t.StartedEventId,\n\t}\n}\n\nfunc FromTimerStartedEventAttributes(t *types.TimerStartedEventAttributes) *apiv1.TimerStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TimerStartedEventAttributes{\n\t\tTimerId:                      t.TimerID,\n\t\tStartToFireTimeout:           secondsToDuration(int64To32(t.StartToFireTimeoutSeconds)),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToTimerStartedEventAttributes(t *apiv1.TimerStartedEventAttributes) *types.TimerStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerStartedEventAttributes{\n\t\tTimerID:                      t.TimerId,\n\t\tStartToFireTimeoutSeconds:    int32To64(durationToSeconds(t.StartToFireTimeout)),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nconst (\n\tDomainUpdateDescriptionField              = \"description\"\n\tDomainUpdateOwnerEmailField               = \"owner_email\"\n\tDomainUpdateDataField                     = \"data\"\n\tDomainUpdateRetentionPeriodField          = \"workflow_execution_retention_period\"\n\tDomainUpdateBadBinariesField              = \"bad_binaries\"\n\tDomainUpdateHistoryArchivalStatusField    = \"history_archival_status\"\n\tDomainUpdateHistoryArchivalURIField       = \"history_archival_uri\"\n\tDomainUpdateVisibilityArchivalStatusField = \"visibility_archival_status\"\n\tDomainUpdateVisibilityArchivalURIField    = \"visibility_archival_uri\"\n\tDomainUpdateActiveClusterNameField        = \"active_cluster_name\"\n\tDomainUpdateActiveClustersField           = \"active_clusters\"\n\tDomainUpdateClustersField                 = \"clusters\"\n\tDomainUpdateDeleteBadBinaryField          = \"delete_bad_binary\"\n\tDomainUpdateFailoverTimeoutField          = \"failover_timeout\"\n)\n\nfunc FromUpdateDomainRequest(t *types.UpdateDomainRequest) *apiv1.UpdateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := apiv1.UpdateDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n\tfields := []string{}\n\n\tif t.Description != nil {\n\t\trequest.Description = *t.Description\n\t\tfields = append(fields, DomainUpdateDescriptionField)\n\t}\n\tif t.OwnerEmail != nil {\n\t\trequest.OwnerEmail = *t.OwnerEmail\n\t\tfields = append(fields, DomainUpdateOwnerEmailField)\n\t}\n\tif t.Data != nil {\n\t\trequest.Data = t.Data\n\t\tfields = append(fields, DomainUpdateDataField)\n\t}\n\tif t.WorkflowExecutionRetentionPeriodInDays != nil {\n\t\trequest.WorkflowExecutionRetentionPeriod = daysToDuration(t.WorkflowExecutionRetentionPeriodInDays)\n\t\tfields = append(fields, DomainUpdateRetentionPeriodField)\n\t}\n\t// if t.EmitMetric != nil {} - DEPRECATED\n\tif t.BadBinaries != nil {\n\t\trequest.BadBinaries = FromBadBinaries(t.BadBinaries)\n\t\tfields = append(fields, DomainUpdateBadBinariesField)\n\t}\n\tif t.HistoryArchivalStatus != nil {\n\t\trequest.HistoryArchivalStatus = FromArchivalStatus(t.HistoryArchivalStatus)\n\t\tfields = append(fields, DomainUpdateHistoryArchivalStatusField)\n\t}\n\tif t.HistoryArchivalURI != nil {\n\t\trequest.HistoryArchivalUri = *t.HistoryArchivalURI\n\t\tfields = append(fields, DomainUpdateHistoryArchivalURIField)\n\t}\n\tif t.VisibilityArchivalStatus != nil {\n\t\trequest.VisibilityArchivalStatus = FromArchivalStatus(t.VisibilityArchivalStatus)\n\t\tfields = append(fields, DomainUpdateVisibilityArchivalStatusField)\n\t}\n\tif t.VisibilityArchivalURI != nil {\n\t\trequest.VisibilityArchivalUri = *t.VisibilityArchivalURI\n\t\tfields = append(fields, DomainUpdateVisibilityArchivalURIField)\n\t}\n\tif t.ActiveClusterName != nil {\n\t\trequest.ActiveClusterName = *t.ActiveClusterName\n\t\tfields = append(fields, DomainUpdateActiveClusterNameField)\n\t}\n\tif t.ActiveClusters != nil {\n\t\trequest.ActiveClusters = FromActiveClusters(t.ActiveClusters)\n\t\tfields = append(fields, DomainUpdateActiveClustersField)\n\t}\n\tif t.Clusters != nil {\n\t\trequest.Clusters = FromClusterReplicationConfigurationArray(t.Clusters)\n\t\tfields = append(fields, DomainUpdateClustersField)\n\t}\n\tif t.DeleteBadBinary != nil {\n\t\trequest.DeleteBadBinary = *t.DeleteBadBinary\n\t\tfields = append(fields, DomainUpdateDeleteBadBinaryField)\n\t}\n\tif t.FailoverTimeoutInSeconds != nil {\n\t\trequest.FailoverTimeout = secondsToDuration(t.FailoverTimeoutInSeconds)\n\t\tfields = append(fields, DomainUpdateFailoverTimeoutField)\n\t}\n\n\trequest.UpdateMask = newFieldMask(fields)\n\n\treturn &request\n}\n\nfunc ToUpdateDomainRequest(t *apiv1.UpdateDomainRequest) *types.UpdateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := types.UpdateDomainRequest{\n\t\tName:          t.Name,\n\t\tSecurityToken: t.SecurityToken,\n\t}\n\tfs := newFieldSet(t.UpdateMask)\n\n\tif fs.isSet(DomainUpdateDescriptionField) {\n\t\trequest.Description = common.StringPtr(t.Description)\n\t}\n\tif fs.isSet(DomainUpdateOwnerEmailField) {\n\t\trequest.OwnerEmail = common.StringPtr(t.OwnerEmail)\n\t}\n\tif fs.isSet(DomainUpdateDataField) {\n\t\trequest.Data = t.Data\n\t}\n\tif fs.isSet(DomainUpdateRetentionPeriodField) {\n\t\trequest.WorkflowExecutionRetentionPeriodInDays = durationToDays(t.WorkflowExecutionRetentionPeriod)\n\t}\n\tif fs.isSet(DomainUpdateBadBinariesField) {\n\t\trequest.BadBinaries = ToBadBinaries(t.BadBinaries)\n\t}\n\tif fs.isSet(DomainUpdateHistoryArchivalStatusField) {\n\t\trequest.HistoryArchivalStatus = ToArchivalStatus(t.HistoryArchivalStatus)\n\t}\n\tif fs.isSet(DomainUpdateHistoryArchivalURIField) {\n\t\trequest.HistoryArchivalURI = common.StringPtr(t.HistoryArchivalUri)\n\t}\n\tif fs.isSet(DomainUpdateVisibilityArchivalStatusField) {\n\t\trequest.VisibilityArchivalStatus = ToArchivalStatus(t.VisibilityArchivalStatus)\n\t}\n\tif fs.isSet(DomainUpdateVisibilityArchivalURIField) {\n\t\trequest.VisibilityArchivalURI = common.StringPtr(t.VisibilityArchivalUri)\n\t}\n\tif fs.isSet(DomainUpdateActiveClusterNameField) {\n\t\trequest.ActiveClusterName = common.StringPtr(t.ActiveClusterName)\n\t}\n\tif fs.isSet(DomainUpdateActiveClustersField) {\n\t\trequest.ActiveClusters = ToActiveClusters(t.ActiveClusters)\n\t}\n\tif fs.isSet(DomainUpdateClustersField) {\n\t\trequest.Clusters = ToClusterReplicationConfigurationArray(t.Clusters)\n\t}\n\tif fs.isSet(DomainUpdateDeleteBadBinaryField) {\n\t\trequest.DeleteBadBinary = common.StringPtr(t.DeleteBadBinary)\n\t}\n\tif fs.isSet(DomainUpdateFailoverTimeoutField) {\n\t\trequest.FailoverTimeoutInSeconds = durationToSeconds(t.FailoverTimeout)\n\t}\n\n\treturn &request\n}\n\nfunc FromUpdateDomainResponse(t *types.UpdateDomainResponse) *apiv1.UpdateDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tdomain := &apiv1.Domain{\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tIsGlobalDomain:  t.IsGlobalDomain,\n\t}\n\tif info := t.DomainInfo; info != nil {\n\t\tdomain.Id = info.UUID\n\t\tdomain.Name = info.Name\n\t\tdomain.Status = FromDomainStatus(info.Status)\n\t\tdomain.Description = info.Description\n\t\tdomain.OwnerEmail = info.OwnerEmail\n\t\tdomain.Data = info.Data\n\t}\n\tif config := t.Configuration; config != nil {\n\t\tdomain.IsolationGroups = FromIsolationGroupConfig(config.IsolationGroups)\n\t\tdomain.WorkflowExecutionRetentionPeriod = daysToDuration(&config.WorkflowExecutionRetentionPeriodInDays)\n\t\tdomain.BadBinaries = FromBadBinaries(config.BadBinaries)\n\t\tdomain.HistoryArchivalStatus = FromArchivalStatus(config.HistoryArchivalStatus)\n\t\tdomain.HistoryArchivalUri = config.HistoryArchivalURI\n\t\tdomain.VisibilityArchivalStatus = FromArchivalStatus(config.VisibilityArchivalStatus)\n\t\tdomain.VisibilityArchivalUri = config.VisibilityArchivalURI\n\t\tdomain.AsyncWorkflowConfig = FromDomainAsyncWorkflowConfiguraton(config.AsyncWorkflowConfig)\n\t}\n\tif repl := t.ReplicationConfiguration; repl != nil {\n\t\tdomain.ActiveClusterName = repl.ActiveClusterName\n\t\tdomain.Clusters = FromClusterReplicationConfigurationArray(repl.Clusters)\n\t\tdomain.ActiveClusters = FromActiveClusters(repl.ActiveClusters)\n\t}\n\treturn &apiv1.UpdateDomainResponse{\n\t\tDomain: domain,\n\t}\n}\n\nfunc ToUpdateDomainResponse(t *apiv1.UpdateDomainResponse) *types.UpdateDomainResponse {\n\tif t == nil || t.Domain == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:        t.Domain.Name,\n\t\t\tStatus:      ToDomainStatus(t.Domain.Status),\n\t\t\tDescription: t.Domain.Description,\n\t\t\tOwnerEmail:  t.Domain.OwnerEmail,\n\t\t\tData:        t.Domain.Data,\n\t\t\tUUID:        t.Domain.Id,\n\t\t},\n\t\tConfiguration: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Default(durationToDays(t.Domain.WorkflowExecutionRetentionPeriod)),\n\t\t\tEmitMetric:                             true,\n\t\t\tBadBinaries:                            ToBadBinaries(t.Domain.BadBinaries),\n\t\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.Domain.HistoryArchivalStatus),\n\t\t\tHistoryArchivalURI:                     t.Domain.HistoryArchivalUri,\n\t\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.Domain.VisibilityArchivalStatus),\n\t\t\tVisibilityArchivalURI:                  t.Domain.VisibilityArchivalUri,\n\t\t\tIsolationGroups:                        ToIsolationGroupConfig(t.Domain.IsolationGroups),\n\t\t\tAsyncWorkflowConfig:                    ToDomainAsyncWorkflowConfiguraton(t.Domain.AsyncWorkflowConfig),\n\t\t},\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: t.Domain.ActiveClusterName,\n\t\t\tClusters:          ToClusterReplicationConfigurationArray(t.Domain.Clusters),\n\t\t\tActiveClusters:    ToActiveClusters(t.Domain.ActiveClusters),\n\t\t},\n\t\tFailoverVersion: t.Domain.FailoverVersion,\n\t\tIsGlobalDomain:  t.Domain.IsGlobalDomain,\n\t}\n}\n\nfunc FromFailoverDomainRequest(t *types.FailoverDomainRequest) *apiv1.FailoverDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.FailoverDomainRequest{\n\t\tDomainName:              t.DomainName,\n\t\tDomainActiveClusterName: t.GetDomainActiveClusterName(),\n\t\tActiveClusters:          FromActiveClusters(t.ActiveClusters),\n\t}\n}\n\nfunc ToFailoverDomainRequest(t *apiv1.FailoverDomainRequest) *types.FailoverDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar domainActiveClusterName *string\n\tif t.DomainActiveClusterName != \"\" {\n\t\tdomainActiveClusterName = common.StringPtr(t.DomainActiveClusterName)\n\t}\n\treturn &types.FailoverDomainRequest{\n\t\tDomainName:              t.DomainName,\n\t\tDomainActiveClusterName: domainActiveClusterName,\n\t\tActiveClusters:          ToActiveClusters(t.ActiveClusters),\n\t}\n}\n\nfunc FromFailoverDomainResponse(t *types.FailoverDomainResponse) *apiv1.FailoverDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tdomain := &apiv1.Domain{\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tIsGlobalDomain:  t.IsGlobalDomain,\n\t}\n\tif info := t.DomainInfo; info != nil {\n\t\tdomain.Id = info.UUID\n\t\tdomain.Name = info.Name\n\t\tdomain.Status = FromDomainStatus(info.Status)\n\t\tdomain.Description = info.Description\n\t\tdomain.OwnerEmail = info.OwnerEmail\n\t\tdomain.Data = info.Data\n\t}\n\tif config := t.Configuration; config != nil {\n\t\tdomain.IsolationGroups = FromIsolationGroupConfig(config.IsolationGroups)\n\t\tdomain.WorkflowExecutionRetentionPeriod = daysToDuration(&config.WorkflowExecutionRetentionPeriodInDays)\n\t\tdomain.BadBinaries = FromBadBinaries(config.BadBinaries)\n\t\tdomain.HistoryArchivalStatus = FromArchivalStatus(config.HistoryArchivalStatus)\n\t\tdomain.HistoryArchivalUri = config.HistoryArchivalURI\n\t\tdomain.VisibilityArchivalStatus = FromArchivalStatus(config.VisibilityArchivalStatus)\n\t\tdomain.VisibilityArchivalUri = config.VisibilityArchivalURI\n\t\tdomain.AsyncWorkflowConfig = FromDomainAsyncWorkflowConfiguraton(config.AsyncWorkflowConfig)\n\t}\n\tif repl := t.ReplicationConfiguration; repl != nil {\n\t\tdomain.ActiveClusterName = repl.ActiveClusterName\n\t\tdomain.Clusters = FromClusterReplicationConfigurationArray(repl.Clusters)\n\t\tdomain.ActiveClusters = FromActiveClusters(repl.ActiveClusters)\n\t}\n\n\treturn &apiv1.FailoverDomainResponse{\n\t\tDomain: domain,\n\t}\n}\n\nfunc ToFailoverDomainResponse(t *apiv1.FailoverDomainResponse) *types.FailoverDomainResponse {\n\tif t == nil || t.Domain == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:        t.Domain.Name,\n\t\t\tStatus:      ToDomainStatus(t.Domain.Status),\n\t\t\tDescription: t.Domain.Description,\n\t\t\tOwnerEmail:  t.Domain.OwnerEmail,\n\t\t\tData:        t.Domain.Data,\n\t\t\tUUID:        t.Domain.Id,\n\t\t},\n\t\tConfiguration: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Default(durationToDays(t.Domain.WorkflowExecutionRetentionPeriod)),\n\t\t\tEmitMetric:                             true,\n\t\t\tBadBinaries:                            ToBadBinaries(t.Domain.BadBinaries),\n\t\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.Domain.HistoryArchivalStatus),\n\t\t\tHistoryArchivalURI:                     t.Domain.HistoryArchivalUri,\n\t\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.Domain.VisibilityArchivalStatus),\n\t\t\tVisibilityArchivalURI:                  t.Domain.VisibilityArchivalUri,\n\t\t\tIsolationGroups:                        ToIsolationGroupConfig(t.Domain.IsolationGroups),\n\t\t\tAsyncWorkflowConfig:                    ToDomainAsyncWorkflowConfiguraton(t.Domain.AsyncWorkflowConfig),\n\t\t},\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: t.Domain.ActiveClusterName,\n\t\t\tClusters:          ToClusterReplicationConfigurationArray(t.Domain.Clusters),\n\t\t\tActiveClusters:    ToActiveClusters(t.Domain.ActiveClusters),\n\t\t},\n\t\tFailoverVersion: t.Domain.FailoverVersion,\n\t\tIsGlobalDomain:  t.Domain.IsGlobalDomain,\n\t}\n}\n\nfunc FromListFailoverHistoryRequest(t *types.ListFailoverHistoryRequest) *apiv1.ListFailoverHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListFailoverHistoryRequest{\n\t\tFilters:    FromListFailoverHistoryRequestFilters(t.Filters),\n\t\tPagination: FromPaginationOptions(t.Pagination),\n\t}\n}\n\nfunc ToListFailoverHistoryRequest(t *apiv1.ListFailoverHistoryRequest) *types.ListFailoverHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListFailoverHistoryRequest{\n\t\tFilters:    ToListFailoverHistoryRequestFilters(t.Filters),\n\t\tPagination: ToPaginationOptions(t.Pagination),\n\t}\n}\n\nfunc ToListFailoverHistoryResponse(t *apiv1.ListFailoverHistoryResponse) *types.ListFailoverHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListFailoverHistoryResponse{\n\t\tFailoverEvents: ToFailoverEventArray(t.FailoverEvents),\n\t\tNextPageToken:  t.NextPageToken,\n\t}\n}\n\nfunc FromListFailoverHistoryResponse(t *types.ListFailoverHistoryResponse) *apiv1.ListFailoverHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListFailoverHistoryResponse{\n\t\tFailoverEvents: FromFailoverEventArray(t.FailoverEvents),\n\t\tNextPageToken:  t.NextPageToken,\n\t}\n}\n\nfunc FromListFailoverHistoryRequestFilters(t *types.ListFailoverHistoryRequestFilters) *apiv1.ListFailoverHistoryRequestFilters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListFailoverHistoryRequestFilters{\n\t\tDomainId: t.DomainID,\n\t}\n}\n\nfunc ToListFailoverHistoryRequestFilters(t *apiv1.ListFailoverHistoryRequestFilters) *types.ListFailoverHistoryRequestFilters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListFailoverHistoryRequestFilters{\n\t\tDomainID: t.DomainId,\n\t}\n}\n\nfunc FromPaginationOptions(t *types.PaginationOptions) *apiv1.PaginationOptions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tpageSize := int32(0)\n\tif t.PageSize != nil {\n\t\tpageSize = *t.PageSize\n\t}\n\treturn &apiv1.PaginationOptions{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToPaginationOptions(t *apiv1.PaginationOptions) *types.PaginationOptions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar pageSize *int32\n\tif t.PageSize != 0 {\n\t\tpageSize = common.Int32Ptr(t.PageSize)\n\t}\n\treturn &types.PaginationOptions{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromFailoverEvent(t *types.FailoverEvent) *apiv1.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.FailoverEvent{\n\t\tId:               t.GetID(),\n\t\tCreatedTime:      unixNanoToTime(t.CreatedTime),\n\t\tFailoverType:     FromFailoverType(t.FailoverType),\n\t\tClusterFailovers: FromClusterFailoverArray(t.ClusterFailovers),\n\t}\n}\n\nfunc ToFailoverEvent(t *apiv1.FailoverEvent) *types.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar id *string\n\tif t.Id != \"\" {\n\t\tid = common.StringPtr(t.Id)\n\t}\n\treturn &types.FailoverEvent{\n\t\tID:               id,\n\t\tCreatedTime:      timeToUnixNano(t.CreatedTime),\n\t\tFailoverType:     ToFailoverType(t.FailoverType),\n\t\tClusterFailovers: ToClusterFailoverArray(t.ClusterFailovers),\n\t}\n}\n\nfunc FromFailoverEventArray(t []*types.FailoverEvent) []*apiv1.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.FailoverEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = FromFailoverEvent(t[i])\n\t}\n\treturn v\n}\n\nfunc ToFailoverEventArray(t []*apiv1.FailoverEvent) []*types.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.FailoverEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = ToFailoverEvent(t[i])\n\t}\n\treturn v\n}\n\nfunc FromClusterFailover(t *types.ClusterFailover) *apiv1.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ClusterFailover{\n\t\tFromCluster:      FromActiveClusterInfo(t.FromCluster),\n\t\tToCluster:        FromActiveClusterInfo(t.ToCluster),\n\t\tClusterAttribute: FromClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\nfunc ToClusterFailover(t *apiv1.ClusterFailover) *types.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterFailover{\n\t\tFromCluster:      ToActiveClusterInfo(t.FromCluster),\n\t\tToCluster:        ToActiveClusterInfo(t.ToCluster),\n\t\tClusterAttribute: ToClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\nfunc FromClusterFailoverArray(t []*types.ClusterFailover) []*apiv1.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.ClusterFailover, len(t))\n\tfor i := range t {\n\t\tv[i] = FromClusterFailover(t[i])\n\t}\n\treturn v\n}\n\nfunc ToClusterFailoverArray(t []*apiv1.ClusterFailover) []*types.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ClusterFailover, len(t))\n\tfor i := range t {\n\t\tv[i] = ToClusterFailover(t[i])\n\t}\n\treturn v\n}\n\nfunc FromActiveClusterInfo(t *types.ActiveClusterInfo) *apiv1.ActiveClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ActiveClusterInfo{\n\t\tActiveClusterName: t.ActiveClusterName,\n\t\tFailoverVersion:   t.FailoverVersion,\n\t}\n}\n\nfunc ToActiveClusterInfo(t *apiv1.ActiveClusterInfo) *types.ActiveClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActiveClusterInfo{\n\t\tActiveClusterName: t.ActiveClusterName,\n\t\tFailoverVersion:   t.FailoverVersion,\n\t}\n}\n\nfunc FromFailoverType(t *types.FailoverType) apiv1.FailoverType {\n\tif t == nil {\n\t\treturn apiv1.FailoverType_FAILOVER_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.FailoverTypeForce:\n\t\treturn apiv1.FailoverType_FAILOVER_TYPE_FORCE\n\tcase types.FailoverTypeGraceful:\n\t\treturn apiv1.FailoverType_FAILOVER_TYPE_GRACEFUL\n\t}\n\treturn apiv1.FailoverType_FAILOVER_TYPE_INVALID\n}\n\nfunc ToFailoverType(t apiv1.FailoverType) *types.FailoverType {\n\tswitch t {\n\tcase apiv1.FailoverType_FAILOVER_TYPE_FORCE:\n\t\treturn types.FailoverTypeForce.Ptr()\n\tcase apiv1.FailoverType_FAILOVER_TYPE_GRACEFUL:\n\t\treturn types.FailoverTypeGraceful.Ptr()\n\tdefault:\n\t\t// For FAILOVER_TYPE_INVALID and unknown values, return nil\n\t\treturn nil\n\t}\n}\n\nfunc FromUpsertWorkflowSearchAttributesDecisionAttributes(t *types.UpsertWorkflowSearchAttributesDecisionAttributes) *apiv1.UpsertWorkflowSearchAttributesDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tSearchAttributes: FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToUpsertWorkflowSearchAttributesDecisionAttributes(t *apiv1.UpsertWorkflowSearchAttributesDecisionAttributes) *types.UpsertWorkflowSearchAttributesDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tSearchAttributes: ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromUpsertWorkflowSearchAttributesEventAttributes(t *types.UpsertWorkflowSearchAttributesEventAttributes) *apiv1.UpsertWorkflowSearchAttributesEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToUpsertWorkflowSearchAttributesEventAttributes(t *apiv1.UpsertWorkflowSearchAttributesEventAttributes) *types.UpsertWorkflowSearchAttributesEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tSearchAttributes:             ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromWorkerVersionInfo(t *types.WorkerVersionInfo) *apiv1.WorkerVersionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkerVersionInfo{\n\t\tImpl:           t.Impl,\n\t\tFeatureVersion: t.FeatureVersion,\n\t}\n}\n\nfunc ToWorkerVersionInfo(t *apiv1.WorkerVersionInfo) *types.WorkerVersionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkerVersionInfo{\n\t\tImpl:           t.Impl,\n\t\tFeatureVersion: t.FeatureVersion,\n\t}\n}\n\nfunc FromWorkflowRunPair(workflowID, runID string) *apiv1.WorkflowExecution {\n\treturn &apiv1.WorkflowExecution{\n\t\tWorkflowId: workflowID,\n\t\tRunId:      runID,\n\t}\n}\n\nfunc ToWorkflowID(t *apiv1.WorkflowExecution) string {\n\tif t == nil {\n\t\treturn \"\"\n\t}\n\treturn t.WorkflowId\n}\n\nfunc ToRunID(t *apiv1.WorkflowExecution) string {\n\tif t == nil {\n\t\treturn \"\"\n\t}\n\treturn t.RunId\n}\n\nfunc FromWorkflowExecution(t *types.WorkflowExecution) *apiv1.WorkflowExecution {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecution{\n\t\tWorkflowId: t.WorkflowID,\n\t\tRunId:      t.RunID,\n\t}\n}\n\nfunc ToWorkflowExecution(t *apiv1.WorkflowExecution) *types.WorkflowExecution {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecution{\n\t\tWorkflowID: t.WorkflowId,\n\t\tRunID:      t.RunId,\n\t}\n}\n\nfunc FromExternalExecutionInfoFields(we *types.WorkflowExecution, initiatedID *int64) *apiv1.ExternalExecutionInfo {\n\tif we == nil && initiatedID == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ExternalExecutionInfo{\n\t\tWorkflowExecution: FromWorkflowExecution(we),\n\t\tInitiatedId:       common.Int64Default(initiatedID), // TODO: we need to figure out whetherrr this field is needed or not\n\t}\n}\n\nfunc ToExternalWorkflowExecution(t *apiv1.ExternalExecutionInfo) *types.WorkflowExecution {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn ToWorkflowExecution(t.WorkflowExecution)\n}\n\nfunc ToExternalInitiatedID(t *apiv1.ExternalExecutionInfo) *int64 {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &t.InitiatedId\n}\n\nfunc FromWorkflowExecutionCancelRequestedEventAttributes(t *types.WorkflowExecutionCancelRequestedEventAttributes) *apiv1.WorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                 t.Cause,\n\t\tExternalExecutionInfo: FromExternalExecutionInfoFields(t.ExternalWorkflowExecution, t.ExternalInitiatedEventID),\n\t\tIdentity:              t.Identity,\n\t\tRequestId:             t.RequestID,\n\t}\n}\n\nfunc ToWorkflowExecutionCancelRequestedEventAttributes(t *apiv1.WorkflowExecutionCancelRequestedEventAttributes) *types.WorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                     t.Cause,\n\t\tExternalInitiatedEventID:  ToExternalInitiatedID(t.ExternalExecutionInfo),\n\t\tExternalWorkflowExecution: ToExternalWorkflowExecution(t.ExternalExecutionInfo),\n\t\tIdentity:                  t.Identity,\n\t\tRequestID:                 t.RequestId,\n\t}\n}\n\nfunc FromWorkflowExecutionCanceledEventAttributes(t *types.WorkflowExecutionCanceledEventAttributes) *apiv1.WorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tDetails:                      FromPayload(t.Details),\n\t}\n}\n\nfunc ToWorkflowExecutionCanceledEventAttributes(t *apiv1.WorkflowExecutionCanceledEventAttributes) *types.WorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t\tDetails:                      ToPayload(t.Details),\n\t}\n}\n\nfunc FromWorkflowExecutionCloseStatus(t *types.WorkflowExecutionCloseStatus) apiv1.WorkflowExecutionCloseStatus {\n\tif t == nil {\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID\n\t}\n\tswitch *t {\n\tcase types.WorkflowExecutionCloseStatusCompleted:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_COMPLETED\n\tcase types.WorkflowExecutionCloseStatusFailed:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_FAILED\n\tcase types.WorkflowExecutionCloseStatusCanceled:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_CANCELED\n\tcase types.WorkflowExecutionCloseStatusTerminated:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_TERMINATED\n\tcase types.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_CONTINUED_AS_NEW\n\tcase types.WorkflowExecutionCloseStatusTimedOut:\n\t\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_TIMED_OUT\n\t}\n\treturn apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID\n}\n\nfunc ToWorkflowExecutionCloseStatus(t apiv1.WorkflowExecutionCloseStatus) *types.WorkflowExecutionCloseStatus {\n\tswitch t {\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID:\n\t\treturn nil\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_COMPLETED:\n\t\treturn types.WorkflowExecutionCloseStatusCompleted.Ptr()\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_FAILED:\n\t\treturn types.WorkflowExecutionCloseStatusFailed.Ptr()\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_CANCELED:\n\t\treturn types.WorkflowExecutionCloseStatusCanceled.Ptr()\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_TERMINATED:\n\t\treturn types.WorkflowExecutionCloseStatusTerminated.Ptr()\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_CONTINUED_AS_NEW:\n\t\treturn types.WorkflowExecutionCloseStatusContinuedAsNew.Ptr()\n\tcase apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_TIMED_OUT:\n\t\treturn types.WorkflowExecutionCloseStatusTimedOut.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromWorkflowExecutionStatus(t *types.WorkflowExecutionStatus) apiv1.WorkflowExecutionStatus {\n\tif t == nil {\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_INVALID\n\t}\n\tswitch *t {\n\tcase types.WorkflowExecutionStatusPending:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_PENDING\n\tcase types.WorkflowExecutionStatusStarted:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_STARTED\n\tcase types.WorkflowExecutionStatusCompleted:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_COMPLETED\n\tcase types.WorkflowExecutionStatusFailed:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_FAILED\n\tcase types.WorkflowExecutionStatusCanceled:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_CANCELED\n\tcase types.WorkflowExecutionStatusTerminated:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_TERMINATED\n\tcase types.WorkflowExecutionStatusContinuedAsNew:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_CONTINUED_AS_NEW\n\tcase types.WorkflowExecutionStatusTimedOut:\n\t\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_TIMED_OUT\n\t}\n\treturn apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_INVALID\n}\n\nfunc ToWorkflowExecutionStatus(t apiv1.WorkflowExecutionStatus) *types.WorkflowExecutionStatus {\n\tswitch t {\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_INVALID:\n\t\treturn nil\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_PENDING:\n\t\treturn types.WorkflowExecutionStatusPending.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_STARTED:\n\t\treturn types.WorkflowExecutionStatusStarted.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_COMPLETED:\n\t\treturn types.WorkflowExecutionStatusCompleted.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_FAILED:\n\t\treturn types.WorkflowExecutionStatusFailed.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_CANCELED:\n\t\treturn types.WorkflowExecutionStatusCanceled.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_TERMINATED:\n\t\treturn types.WorkflowExecutionStatusTerminated.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_CONTINUED_AS_NEW:\n\t\treturn types.WorkflowExecutionStatusContinuedAsNew.Ptr()\n\tcase apiv1.WorkflowExecutionStatus_WORKFLOW_EXECUTION_STATUS_TIMED_OUT:\n\t\treturn types.WorkflowExecutionStatusTimedOut.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromWorkflowExecutionCompletedEventAttributes(t *types.WorkflowExecutionCompletedEventAttributes) *apiv1.WorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       FromPayload(t.Result),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToWorkflowExecutionCompletedEventAttributes(t *apiv1.WorkflowExecutionCompletedEventAttributes) *types.WorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       ToPayload(t.Result),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nfunc FromWorkflowExecutionConfiguration(t *types.WorkflowExecutionConfiguration) *apiv1.WorkflowExecutionConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionConfiguration{\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t}\n}\n\nfunc ToWorkflowExecutionConfiguration(t *apiv1.WorkflowExecutionConfiguration) *types.WorkflowExecutionConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionConfiguration{\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t}\n}\n\nfunc FromWorkflowExecutionContinuedAsNewEventAttributes(t *types.WorkflowExecutionContinuedAsNewEventAttributes) *apiv1.WorkflowExecutionContinuedAsNewEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunId:            t.NewExecutionRunID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t\tBackoffStartInterval:         secondsToDuration(t.BackoffStartIntervalInSeconds),\n\t\tInitiator:                    FromContinueAsNewInitiator(t.Initiator),\n\t\tFailure:                      FromFailure(t.FailureReason, t.FailureDetails),\n\t\tLastCompletionResult:         FromPayload(t.LastCompletionResult),\n\t\tHeader:                       FromHeader(t.Header),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\nfunc ToWorkflowExecutionContinuedAsNewEventAttributes(t *apiv1.WorkflowExecutionContinuedAsNewEventAttributes) *types.WorkflowExecutionContinuedAsNewEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunID:                   t.NewExecutionRunId,\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tDecisionTaskCompletedEventID:        t.DecisionTaskCompletedEventId,\n\t\tBackoffStartIntervalInSeconds:       durationToSeconds(t.BackoffStartInterval),\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       ToFailureReason(t.Failure),\n\t\tFailureDetails:                      ToFailureDetails(t.Failure),\n\t\tLastCompletionResult:                ToPayload(t.LastCompletionResult),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\nfunc FromWorkflowExecutionFailedEventAttributes(t *types.WorkflowExecutionFailedEventAttributes) *apiv1.WorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionFailedEventAttributes{\n\t\tFailure:                      FromFailure(t.Reason, t.Details),\n\t\tDecisionTaskCompletedEventId: t.DecisionTaskCompletedEventID,\n\t}\n}\n\nfunc ToWorkflowExecutionFailedEventAttributes(t *apiv1.WorkflowExecutionFailedEventAttributes) *types.WorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionFailedEventAttributes{\n\t\tReason:                       ToFailureReason(t.Failure),\n\t\tDetails:                      ToFailureDetails(t.Failure),\n\t\tDecisionTaskCompletedEventID: t.DecisionTaskCompletedEventId,\n\t}\n}\n\nfunc FromWorkflowExecutionFilter(t *types.WorkflowExecutionFilter) *apiv1.WorkflowExecutionFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionFilter{\n\t\tWorkflowId: t.WorkflowID,\n\t\tRunId:      t.RunID,\n\t}\n}\n\nfunc ToWorkflowExecutionFilter(t *apiv1.WorkflowExecutionFilter) *types.WorkflowExecutionFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionFilter{\n\t\tWorkflowID: t.WorkflowId,\n\t\tRunID:      t.RunId,\n\t}\n}\n\nfunc FromParentExecutionInfo(t *types.ParentExecutionInfo) *apiv1.ParentExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ParentExecutionInfo{\n\t\tDomainId:          t.DomainUUID,\n\t\tDomainName:        t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t\tInitiatedId:       t.InitiatedID,\n\t}\n}\n\nfunc ToParentExecutionInfo(t *apiv1.ParentExecutionInfo) *types.ParentExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ParentExecutionInfo{\n\t\tDomainUUID:  t.DomainId,\n\t\tDomain:      t.DomainName,\n\t\tExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedID: t.InitiatedId,\n\t}\n}\n\nfunc FromParentExecutionInfoFields(domainID, domainName *string, we *types.WorkflowExecution, initiatedID *int64) *apiv1.ParentExecutionInfo {\n\tif domainID == nil && domainName == nil && we == nil && initiatedID == nil {\n\t\treturn nil\n\t}\n\n\t// ParentExecutionInfo wrapper was added to unify parent related fields.\n\t// However some fields may not be present:\n\t// - on older histories\n\t// - if conversion involves thrift data types\n\t// Fallback to zero values in those cases\n\treturn &apiv1.ParentExecutionInfo{\n\t\tDomainId:          common.StringDefault(domainID),\n\t\tDomainName:        common.StringDefault(domainName),\n\t\tWorkflowExecution: FromWorkflowExecution(we),\n\t\tInitiatedId:       common.Int64Default(initiatedID),\n\t}\n}\n\nfunc ToParentDomainID(pei *apiv1.ParentExecutionInfo) *string {\n\tif pei == nil {\n\t\treturn nil\n\t}\n\treturn &pei.DomainId\n}\n\nfunc ToParentDomainName(pei *apiv1.ParentExecutionInfo) *string {\n\tif pei == nil {\n\t\treturn nil\n\t}\n\treturn &pei.DomainName\n}\n\nfunc ToParentWorkflowExecution(pei *apiv1.ParentExecutionInfo) *types.WorkflowExecution {\n\tif pei == nil {\n\t\treturn nil\n\t}\n\treturn ToWorkflowExecution(pei.WorkflowExecution)\n}\n\nfunc ToParentInitiatedID(pei *apiv1.ParentExecutionInfo) *int64 {\n\tif pei == nil {\n\t\treturn nil\n\t}\n\treturn &pei.InitiatedId\n}\n\nfunc FromWorkflowExecutionInfo(t *types.WorkflowExecutionInfo) *apiv1.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttlName := \"\"\n\tif t.TaskList != nil {\n\t\ttlName = t.TaskList.Name\n\t}\n\tcronSchedule := \"\"\n\tif t.CronSchedule != nil {\n\t\tcronSchedule = *t.CronSchedule\n\t}\n\treturn &apiv1.WorkflowExecutionInfo{\n\t\tWorkflowExecution:            FromWorkflowExecution(t.Execution),\n\t\tType:                         FromWorkflowType(t.Type),\n\t\tStartTime:                    unixNanoToTime(t.StartTime),\n\t\tCloseTime:                    unixNanoToTime(t.CloseTime),\n\t\tCloseStatus:                  FromWorkflowExecutionCloseStatus(t.CloseStatus),\n\t\tHistoryLength:                t.HistoryLength,\n\t\tParentExecutionInfo:          FromParentExecutionInfoFields(t.ParentDomainID, t.ParentDomain, t.ParentExecution, t.ParentInitiatedID),\n\t\tExecutionTime:                unixNanoToTime(t.ExecutionTime),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tAutoResetPoints:              FromResetPoints(t.AutoResetPoints),\n\t\tTaskList:                     tlName,\n\t\tTaskListInfo:                 FromTaskList(t.TaskList),\n\t\tPartitionConfig:              t.PartitionConfig,\n\t\tIsCron:                       t.IsCron,\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronSchedule:                 cronSchedule,\n\t\tExecutionStatus:              FromWorkflowExecutionStatus(t.ExecutionStatus),\n\t\tScheduledExecutionTime:       unixNanoToTime(t.ScheduledExecutionTime),\n\t}\n}\n\nfunc ToWorkflowExecutionInfo(t *apiv1.WorkflowExecutionInfo) *types.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar cronSchedule *string\n\tif t.CronSchedule != \"\" {\n\t\tcronSchedule = &t.CronSchedule\n\t}\n\treturn &types.WorkflowExecutionInfo{\n\t\tExecution:                    ToWorkflowExecution(t.WorkflowExecution),\n\t\tType:                         ToWorkflowType(t.Type),\n\t\tStartTime:                    timeToUnixNano(t.StartTime),\n\t\tCloseTime:                    timeToUnixNano(t.CloseTime),\n\t\tCloseStatus:                  ToWorkflowExecutionCloseStatus(t.CloseStatus),\n\t\tHistoryLength:                t.HistoryLength,\n\t\tParentDomainID:               ToParentDomainID(t.ParentExecutionInfo),\n\t\tParentDomain:                 ToParentDomainName(t.ParentExecutionInfo),\n\t\tParentExecution:              ToParentWorkflowExecution(t.ParentExecutionInfo),\n\t\tParentInitiatedID:            ToParentInitiatedID(t.ParentExecutionInfo),\n\t\tExecutionTime:                timeToUnixNano(t.ExecutionTime),\n\t\tMemo:                         ToMemo(t.Memo),\n\t\tSearchAttributes:             ToSearchAttributes(t.SearchAttributes),\n\t\tAutoResetPoints:              ToResetPoints(t.AutoResetPoints),\n\t\tTaskList:                     MigrateTaskList(t.TaskList, t.TaskListInfo),\n\t\tPartitionConfig:              t.PartitionConfig,\n\t\tIsCron:                       t.IsCron,\n\t\tCronOverlapPolicy:            ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronSchedule:                 cronSchedule,\n\t\tExecutionStatus:              ToWorkflowExecutionStatus(t.ExecutionStatus),\n\t\tScheduledExecutionTime:       timeToUnixNano(t.ScheduledExecutionTime),\n\t}\n}\n\nfunc FromWorkflowExecutionSignaledEventAttributes(t *types.WorkflowExecutionSignaledEventAttributes) *apiv1.WorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: t.SignalName,\n\t\tInput:      FromPayload(t.Input),\n\t\tIdentity:   t.Identity,\n\t\tRequestId:  t.RequestID,\n\t}\n}\n\nfunc ToWorkflowExecutionSignaledEventAttributes(t *apiv1.WorkflowExecutionSignaledEventAttributes) *types.WorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: t.SignalName,\n\t\tInput:      ToPayload(t.Input),\n\t\tIdentity:   t.Identity,\n\t\tRequestID:  t.RequestId,\n\t}\n}\n\nfunc FromWorkflowExecutionStartedEventAttributes(t *types.WorkflowExecutionStartedEventAttributes) *apiv1.WorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &apiv1.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tParentExecutionInfo:          FromParentExecutionInfoFields(t.ParentWorkflowDomainID, t.ParentWorkflowDomain, t.ParentWorkflowExecution, t.ParentInitiatedEventID),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tContinuedExecutionRunId:      t.ContinuedExecutionRunID,\n\t\tInitiator:                    FromContinueAsNewInitiator(t.Initiator),\n\t\tContinuedFailure:             FromFailure(t.ContinuedFailureReason, t.ContinuedFailureDetails),\n\t\tLastCompletionResult:         FromPayload(t.LastCompletionResult),\n\t\tOriginalExecutionRunId:       t.OriginalExecutionRunID,\n\t\tIdentity:                     t.Identity,\n\t\tFirstExecutionRunId:          t.FirstExecutionRunID,\n\t\tFirstScheduledTime:           timeToTimestamp(t.FirstScheduleTime),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tAttempt:                      t.Attempt,\n\t\tExpirationTime:               unixNanoToTime(t.ExpirationTimestamp),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tFirstDecisionTaskBackoff:     secondsToDuration(t.FirstDecisionTaskBackoffSeconds),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tPrevAutoResetPoints:          FromResetPoints(t.PrevAutoResetPoints),\n\t\tHeader:                       FromHeader(t.Header),\n\t\tPartitionConfig:              t.PartitionConfig,\n\t\tRequestId:                    t.RequestID,\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc ToWorkflowExecutionStartedEventAttributes(t *apiv1.WorkflowExecutionStartedEventAttributes) *types.WorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tParentWorkflowDomainID:              ToParentDomainID(t.ParentExecutionInfo),\n\t\tParentWorkflowDomain:                ToParentDomainName(t.ParentExecutionInfo),\n\t\tParentWorkflowExecution:             ToParentWorkflowExecution(t.ParentExecutionInfo),\n\t\tParentInitiatedEventID:              ToParentInitiatedID(t.ParentExecutionInfo),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tContinuedExecutionRunID:             t.ContinuedExecutionRunId,\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tContinuedFailureReason:              ToFailureReason(t.ContinuedFailure),\n\t\tContinuedFailureDetails:             ToFailureDetails(t.ContinuedFailure),\n\t\tLastCompletionResult:                ToPayload(t.LastCompletionResult),\n\t\tOriginalExecutionRunID:              t.OriginalExecutionRunId,\n\t\tIdentity:                            t.Identity,\n\t\tFirstExecutionRunID:                 t.FirstExecutionRunId,\n\t\tFirstScheduleTime:                   timestampToTime(t.FirstScheduledTime),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tAttempt:                             t.Attempt,\n\t\tExpirationTimestamp:                 timeToUnixNano(t.ExpirationTime),\n\t\tCronSchedule:                        t.CronSchedule,\n\t\tFirstDecisionTaskBackoffSeconds:     durationToSeconds(t.FirstDecisionTaskBackoff),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tPrevAutoResetPoints:                 ToResetPoints(t.PrevAutoResetPoints),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tPartitionConfig:                     t.PartitionConfig,\n\t\tRequestID:                           t.RequestId,\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\nfunc FromWorkflowExecutionTerminatedEventAttributes(t *types.WorkflowExecutionTerminatedEventAttributes) *apiv1.WorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   t.Reason,\n\t\tDetails:  FromPayload(t.Details),\n\t\tIdentity: t.Identity,\n\t}\n}\n\nfunc ToWorkflowExecutionTerminatedEventAttributes(t *apiv1.WorkflowExecutionTerminatedEventAttributes) *types.WorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   t.Reason,\n\t\tDetails:  ToPayload(t.Details),\n\t\tIdentity: t.Identity,\n\t}\n}\n\nfunc FromWorkflowExecutionTimedOutEventAttributes(t *types.WorkflowExecutionTimedOutEventAttributes) *apiv1.WorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: FromTimeoutType(t.TimeoutType),\n\t}\n}\n\nfunc ToWorkflowExecutionTimedOutEventAttributes(t *apiv1.WorkflowExecutionTimedOutEventAttributes) *types.WorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: ToTimeoutType(t.TimeoutType),\n\t}\n}\n\nfunc FromWorkflowIDReusePolicy(t *types.WorkflowIDReusePolicy) apiv1.WorkflowIdReusePolicy {\n\tif t == nil {\n\t\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID\n\t}\n\tswitch *t {\n\tcase types.WorkflowIDReusePolicyAllowDuplicateFailedOnly:\n\t\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE_FAILED_ONLY\n\tcase types.WorkflowIDReusePolicyAllowDuplicate:\n\t\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE\n\tcase types.WorkflowIDReusePolicyRejectDuplicate:\n\t\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_REJECT_DUPLICATE\n\tcase types.WorkflowIDReusePolicyTerminateIfRunning:\n\t\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_TERMINATE_IF_RUNNING\n\t}\n\treturn apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID\n}\n\nfunc ToWorkflowIDReusePolicy(t apiv1.WorkflowIdReusePolicy) *types.WorkflowIDReusePolicy {\n\tswitch t {\n\tcase apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID:\n\t\treturn nil\n\tcase apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE_FAILED_ONLY:\n\t\treturn types.WorkflowIDReusePolicyAllowDuplicateFailedOnly.Ptr()\n\tcase apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE:\n\t\treturn types.WorkflowIDReusePolicyAllowDuplicate.Ptr()\n\tcase apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_REJECT_DUPLICATE:\n\t\treturn types.WorkflowIDReusePolicyRejectDuplicate.Ptr()\n\tcase apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_TERMINATE_IF_RUNNING:\n\t\treturn types.WorkflowIDReusePolicyTerminateIfRunning.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromWorkflowQuery(t *types.WorkflowQuery) *apiv1.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowQuery{\n\t\tQueryType: t.QueryType,\n\t\tQueryArgs: FromPayload(t.QueryArgs),\n\t}\n}\n\nfunc ToWorkflowQuery(t *apiv1.WorkflowQuery) *types.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowQuery{\n\t\tQueryType: t.QueryType,\n\t\tQueryArgs: ToPayload(t.QueryArgs),\n\t}\n}\n\nfunc FromWorkflowQueryResult(t *types.WorkflowQueryResult) *apiv1.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowQueryResult{\n\t\tResultType:   FromQueryResultType(t.ResultType),\n\t\tAnswer:       FromPayload(t.Answer),\n\t\tErrorMessage: t.ErrorMessage,\n\t}\n}\n\nfunc ToWorkflowQueryResult(t *apiv1.WorkflowQueryResult) *types.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowQueryResult{\n\t\tResultType:   ToQueryResultType(t.ResultType),\n\t\tAnswer:       ToPayload(t.Answer),\n\t\tErrorMessage: t.ErrorMessage,\n\t}\n}\n\nfunc FromWorkflowType(t *types.WorkflowType) *apiv1.WorkflowType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowType{\n\t\tName: t.Name,\n\t}\n}\n\nfunc ToWorkflowType(t *apiv1.WorkflowType) *types.WorkflowType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowType{\n\t\tName: t.Name,\n\t}\n}\n\nfunc FromWorkflowTypeFilter(t *types.WorkflowTypeFilter) *apiv1.WorkflowTypeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.WorkflowTypeFilter{\n\t\tName: t.Name,\n\t}\n}\n\nfunc ToWorkflowTypeFilter(t *apiv1.WorkflowTypeFilter) *types.WorkflowTypeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowTypeFilter{\n\t\tName: t.Name,\n\t}\n}\n\nfunc FromDataBlobArray(t []*types.DataBlob) []*apiv1.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.DataBlob, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDataBlob(t[i])\n\t}\n\treturn v\n}\n\nfunc ToDataBlobArray(t []*apiv1.DataBlob) []*types.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DataBlob, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDataBlob(t[i])\n\t}\n\treturn v\n}\n\nfunc FromHistoryEventArray(t []*types.HistoryEvent) []*apiv1.HistoryEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.HistoryEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHistoryEvent(t[i])\n\t}\n\treturn v\n}\n\nfunc ToHistoryEventArray(t []*apiv1.HistoryEvent) []*types.HistoryEvent {\n\tif t == nil {\n\t\treturn []*types.HistoryEvent{}\n\t}\n\tv := make([]*types.HistoryEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHistoryEvent(t[i])\n\t}\n\treturn v\n}\n\nfunc FromTaskListPartitionMetadataArray(t []*types.TaskListPartitionMetadata) []*apiv1.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.TaskListPartitionMetadata, len(t))\n\tfor i := range t {\n\t\tv[i] = FromTaskListPartitionMetadata(t[i])\n\t}\n\treturn v\n}\n\nfunc ToTaskListPartitionMetadataArray(t []*apiv1.TaskListPartitionMetadata) []*types.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.TaskListPartitionMetadata, len(t))\n\tfor i := range t {\n\t\tv[i] = ToTaskListPartitionMetadata(t[i])\n\t}\n\treturn v\n}\n\nfunc FromDecisionArray(t []*types.Decision) []*apiv1.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.Decision, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDecision(t[i])\n\t}\n\treturn v\n}\n\nfunc ToDecisionArray(t []*apiv1.Decision) []*types.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.Decision, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDecision(t[i])\n\t}\n\treturn v\n}\n\nfunc FromPollerInfoArray(t []*types.PollerInfo) []*apiv1.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.PollerInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPollerInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToPollerInfoArray(t []*apiv1.PollerInfo) []*types.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PollerInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPollerInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromPendingChildExecutionInfoArray(t []*types.PendingChildExecutionInfo) []*apiv1.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.PendingChildExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPendingChildExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToPendingChildExecutionInfoArray(t []*apiv1.PendingChildExecutionInfo) []*types.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PendingChildExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPendingChildExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromWorkflowExecutionInfoArray(t []*types.WorkflowExecutionInfo) []*apiv1.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.WorkflowExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromWorkflowExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToWorkflowExecutionInfoArray(t []*apiv1.WorkflowExecutionInfo) []*types.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.WorkflowExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToWorkflowExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromDescribeDomainResponseArray(t []*types.DescribeDomainResponse) []*apiv1.Domain {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.Domain, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDescribeDomainResponseDomain(t[i])\n\t}\n\treturn v\n}\n\nfunc ToDescribeDomainResponseArray(t []*apiv1.Domain) []*types.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DescribeDomainResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDescribeDomainResponseDomain(t[i])\n\t}\n\treturn v\n}\n\nfunc FromResetPointInfoArray(t []*types.ResetPointInfo) []*apiv1.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.ResetPointInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromResetPointInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToResetPointInfoArray(t []*apiv1.ResetPointInfo) []*types.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ResetPointInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToResetPointInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromPendingActivityInfoArray(t []*types.PendingActivityInfo) []*apiv1.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.PendingActivityInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPendingActivityInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToPendingActivityInfoArray(t []*apiv1.PendingActivityInfo) []*types.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PendingActivityInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPendingActivityInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromClusterReplicationConfigurationArray(t []*types.ClusterReplicationConfiguration) []*apiv1.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.ClusterReplicationConfiguration, len(t))\n\tfor i := range t {\n\t\tv[i] = FromClusterReplicationConfiguration(t[i])\n\t}\n\treturn v\n}\n\nfunc ToClusterReplicationConfigurationArray(t []*apiv1.ClusterReplicationConfiguration) []*types.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ClusterReplicationConfiguration, len(t))\n\tfor i := range t {\n\t\tv[i] = ToClusterReplicationConfiguration(t[i])\n\t}\n\treturn v\n}\n\nfunc FromActiveClusters(t *types.ActiveClusters) *apiv1.ActiveClusters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar activeClustersByClusterAttribute map[string]*apiv1.ClusterAttributeScope\n\n\tif t.AttributeScopes != nil {\n\t\tactiveClustersByClusterAttribute = make(map[string]*apiv1.ClusterAttributeScope)\n\t\tfor scopeType, scope := range t.AttributeScopes {\n\t\t\tactiveClustersByClusterAttribute[scopeType] = FromClusterAttributeScope(&scope)\n\t\t}\n\t}\n\n\treturn &apiv1.ActiveClusters{\n\t\tActiveClustersByClusterAttribute: activeClustersByClusterAttribute,\n\t}\n}\n\nfunc ToActiveClusters(t *apiv1.ActiveClusters) *types.ActiveClusters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar attributeScopes map[string]types.ClusterAttributeScope\n\n\tif t.ActiveClustersByClusterAttribute != nil {\n\t\tattributeScopes = make(map[string]types.ClusterAttributeScope)\n\t\tfor scopeType, scope := range t.ActiveClustersByClusterAttribute {\n\t\t\tif converted := ToClusterAttributeScope(scope); converted != nil {\n\t\t\t\tattributeScopes[scopeType] = *converted\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.ActiveClusters{\n\t\tAttributeScopes: attributeScopes,\n\t}\n}\n\nfunc FromClusterAttributeScope(t *types.ClusterAttributeScope) *apiv1.ClusterAttributeScope {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar clusterAttributes map[string]*apiv1.ActiveClusterInfo\n\tif len(t.ClusterAttributes) > 0 {\n\t\tclusterAttributes = make(map[string]*apiv1.ActiveClusterInfo)\n\t\tfor name, clusterInfo := range t.ClusterAttributes {\n\t\t\tclusterAttributes[name] = &apiv1.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: clusterInfo.ActiveClusterName,\n\t\t\t\tFailoverVersion:   clusterInfo.FailoverVersion,\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &apiv1.ClusterAttributeScope{\n\t\tClusterAttributes: clusterAttributes,\n\t}\n}\n\nfunc ToClusterAttributeScope(t *apiv1.ClusterAttributeScope) *types.ClusterAttributeScope {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar clusterAttributes map[string]types.ActiveClusterInfo\n\tif len(t.ClusterAttributes) > 0 {\n\t\tclusterAttributes = make(map[string]types.ActiveClusterInfo)\n\t\tfor name, clusterInfo := range t.ClusterAttributes {\n\t\t\tif clusterInfo != nil {\n\t\t\t\tclusterAttributes[name] = types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: clusterInfo.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   clusterInfo.FailoverVersion,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.ClusterAttributeScope{\n\t\tClusterAttributes: clusterAttributes,\n\t}\n}\n\nfunc FromActivityLocalDispatchInfoMap(t map[string]*types.ActivityLocalDispatchInfo) map[string]*apiv1.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.ActivityLocalDispatchInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromActivityLocalDispatchInfo(t[key])\n\t}\n\treturn v\n}\n\nfunc ToActivityLocalDispatchInfoMap(t map[string]*apiv1.ActivityLocalDispatchInfo) map[string]*types.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.ActivityLocalDispatchInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToActivityLocalDispatchInfo(t[key])\n\t}\n\treturn v\n}\n\nfunc FromBadBinaryInfoMap(t map[string]*types.BadBinaryInfo) map[string]*apiv1.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.BadBinaryInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromBadBinaryInfo(t[key])\n\t}\n\treturn v\n}\n\nfunc ToBadBinaryInfoMap(t map[string]*apiv1.BadBinaryInfo) map[string]*types.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.BadBinaryInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToBadBinaryInfo(t[key])\n\t}\n\treturn v\n}\n\nfunc FromIndexedValueTypeMap(t map[string]types.IndexedValueType) map[string]apiv1.IndexedValueType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]apiv1.IndexedValueType, len(t))\n\tfor key := range t {\n\t\tv[key] = FromIndexedValueType(t[key])\n\t}\n\treturn v\n}\n\nfunc ToIndexedValueTypeMap(t map[string]apiv1.IndexedValueType) map[string]types.IndexedValueType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]types.IndexedValueType, len(t))\n\tfor key := range t {\n\t\tv[key] = ToIndexedValueType(t[key])\n\t}\n\treturn v\n}\n\nfunc FromWorkflowQueryMap(t map[string]*types.WorkflowQuery) map[string]*apiv1.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.WorkflowQuery, len(t))\n\tfor key := range t {\n\t\tv[key] = FromWorkflowQuery(t[key])\n\t}\n\treturn v\n}\n\nfunc ToWorkflowQueryMap(t map[string]*apiv1.WorkflowQuery) map[string]*types.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.WorkflowQuery, len(t))\n\tfor key := range t {\n\t\tv[key] = ToWorkflowQuery(t[key])\n\t}\n\treturn v\n}\n\nfunc FromWorkflowQueryResultMap(t map[string]*types.WorkflowQueryResult) map[string]*apiv1.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.WorkflowQueryResult, len(t))\n\tfor key := range t {\n\t\tv[key] = FromWorkflowQueryResult(t[key])\n\t}\n\treturn v\n}\n\nfunc ToWorkflowQueryResultMap(t map[string]*apiv1.WorkflowQueryResult) map[string]*types.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.WorkflowQueryResult, len(t))\n\tfor key := range t {\n\t\tv[key] = ToWorkflowQueryResult(t[key])\n\t}\n\treturn v\n}\n\nfunc FromIsolationGroupMetricsMap(t map[string]*types.IsolationGroupMetrics) map[string]*apiv1.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.IsolationGroupMetrics, len(t))\n\tfor key := range t {\n\t\tv[key] = FromIsolationGroupMetrics(t[key])\n\t}\n\treturn v\n}\n\nfunc ToIsolationGroupMetricsMap(t map[string]*apiv1.IsolationGroupMetrics) map[string]*types.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.IsolationGroupMetrics, len(t))\n\tfor key := range t {\n\t\tv[key] = ToIsolationGroupMetrics(t[key])\n\t}\n\treturn v\n}\n\nfunc FromPayload(data []byte) *apiv1.Payload {\n\tif data == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.Payload{\n\t\tData: data,\n\t}\n}\n\nfunc ToPayload(p *apiv1.Payload) []byte {\n\tif p == nil {\n\t\treturn nil\n\t}\n\tif p.Data == nil {\n\t\t// FromPayload will not generate this case\n\t\t// however, Data field will be dropped by the encoding if it's empty\n\t\t// and receiver side will see nil for the Data field\n\t\t// since we already know p is not nil, Data field must be an empty byte array\n\t\treturn []byte{}\n\t}\n\treturn p.Data\n}\n\nfunc FromPayloadMap(t map[string][]byte) map[string]*apiv1.Payload {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*apiv1.Payload, len(t))\n\tfor key := range t {\n\t\tv[key] = FromPayload(t[key])\n\t}\n\treturn v\n}\n\nfunc ToPayloadMap(t map[string]*apiv1.Payload) map[string][]byte {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string][]byte, len(t))\n\tfor key := range t {\n\t\tv[key] = ToPayload(t[key])\n\t}\n\treturn v\n}\n\nfunc FromFailure(reason *string, details []byte) *apiv1.Failure {\n\tif reason == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.Failure{\n\t\tReason:  *reason,\n\t\tDetails: details,\n\t}\n}\n\nfunc ToFailureReason(failure *apiv1.Failure) *string {\n\tif failure == nil {\n\t\treturn nil\n\t}\n\treturn &failure.Reason\n}\n\nfunc ToFailureDetails(failure *apiv1.Failure) []byte {\n\tif failure == nil {\n\t\treturn nil\n\t}\n\treturn failure.Details\n}\n\nfunc FromHistoryEvent(e *types.HistoryEvent) *apiv1.HistoryEvent {\n\tif e == nil {\n\t\treturn nil\n\t}\n\tevent := apiv1.HistoryEvent{\n\t\tEventId:   e.ID,\n\t\tEventTime: unixNanoToTime(e.Timestamp),\n\t\tVersion:   e.Version,\n\t\tTaskId:    e.TaskID,\n\t}\n\tswitch *e.EventType {\n\tcase types.EventTypeWorkflowExecutionStarted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowExecutionStartedEventAttributes: FromWorkflowExecutionStartedEventAttributes(e.WorkflowExecutionStartedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionCompletedEventAttributes{\n\t\t\tWorkflowExecutionCompletedEventAttributes: FromWorkflowExecutionCompletedEventAttributes(e.WorkflowExecutionCompletedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionFailedEventAttributes{\n\t\t\tWorkflowExecutionFailedEventAttributes: FromWorkflowExecutionFailedEventAttributes(e.WorkflowExecutionFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionTimedOutEventAttributes{\n\t\t\tWorkflowExecutionTimedOutEventAttributes: FromWorkflowExecutionTimedOutEventAttributes(e.WorkflowExecutionTimedOutEventAttributes),\n\t\t}\n\tcase types.EventTypeDecisionTaskScheduled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_DecisionTaskScheduledEventAttributes{\n\t\t\tDecisionTaskScheduledEventAttributes: FromDecisionTaskScheduledEventAttributes(e.DecisionTaskScheduledEventAttributes),\n\t\t}\n\tcase types.EventTypeDecisionTaskStarted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_DecisionTaskStartedEventAttributes{\n\t\t\tDecisionTaskStartedEventAttributes: FromDecisionTaskStartedEventAttributes(e.DecisionTaskStartedEventAttributes),\n\t\t}\n\tcase types.EventTypeDecisionTaskCompleted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_DecisionTaskCompletedEventAttributes{\n\t\t\tDecisionTaskCompletedEventAttributes: FromDecisionTaskCompletedEventAttributes(e.DecisionTaskCompletedEventAttributes),\n\t\t}\n\tcase types.EventTypeDecisionTaskTimedOut:\n\t\tevent.Attributes = &apiv1.HistoryEvent_DecisionTaskTimedOutEventAttributes{\n\t\t\tDecisionTaskTimedOutEventAttributes: FromDecisionTaskTimedOutEventAttributes(e.DecisionTaskTimedOutEventAttributes),\n\t\t}\n\tcase types.EventTypeDecisionTaskFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_DecisionTaskFailedEventAttributes{\n\t\t\tDecisionTaskFailedEventAttributes: FromDecisionTaskFailedEventAttributes(e.DecisionTaskFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskScheduled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskScheduledEventAttributes{\n\t\t\tActivityTaskScheduledEventAttributes: FromActivityTaskScheduledEventAttributes(e.ActivityTaskScheduledEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskStarted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskStartedEventAttributes{\n\t\t\tActivityTaskStartedEventAttributes: FromActivityTaskStartedEventAttributes(e.ActivityTaskStartedEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskCompleted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskCompletedEventAttributes{\n\t\t\tActivityTaskCompletedEventAttributes: FromActivityTaskCompletedEventAttributes(e.ActivityTaskCompletedEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskFailedEventAttributes{\n\t\t\tActivityTaskFailedEventAttributes: FromActivityTaskFailedEventAttributes(e.ActivityTaskFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskTimedOut:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskTimedOutEventAttributes{\n\t\t\tActivityTaskTimedOutEventAttributes: FromActivityTaskTimedOutEventAttributes(e.ActivityTaskTimedOutEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskCancelRequested:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskCancelRequestedEventAttributes{\n\t\t\tActivityTaskCancelRequestedEventAttributes: FromActivityTaskCancelRequestedEventAttributes(e.ActivityTaskCancelRequestedEventAttributes),\n\t\t}\n\tcase types.EventTypeRequestCancelActivityTaskFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_RequestCancelActivityTaskFailedEventAttributes{\n\t\t\tRequestCancelActivityTaskFailedEventAttributes: FromRequestCancelActivityTaskFailedEventAttributes(e.RequestCancelActivityTaskFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeActivityTaskCanceled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ActivityTaskCanceledEventAttributes{\n\t\t\tActivityTaskCanceledEventAttributes: FromActivityTaskCanceledEventAttributes(e.ActivityTaskCanceledEventAttributes),\n\t\t}\n\tcase types.EventTypeTimerStarted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_TimerStartedEventAttributes{\n\t\t\tTimerStartedEventAttributes: FromTimerStartedEventAttributes(e.TimerStartedEventAttributes),\n\t\t}\n\tcase types.EventTypeTimerFired:\n\t\tevent.Attributes = &apiv1.HistoryEvent_TimerFiredEventAttributes{\n\t\t\tTimerFiredEventAttributes: FromTimerFiredEventAttributes(e.TimerFiredEventAttributes),\n\t\t}\n\tcase types.EventTypeTimerCanceled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_TimerCanceledEventAttributes{\n\t\t\tTimerCanceledEventAttributes: FromTimerCanceledEventAttributes(e.TimerCanceledEventAttributes),\n\t\t}\n\tcase types.EventTypeCancelTimerFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_CancelTimerFailedEventAttributes{\n\t\t\tCancelTimerFailedEventAttributes: FromCancelTimerFailedEventAttributes(e.CancelTimerFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionCancelRequested:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionCancelRequestedEventAttributes{\n\t\t\tWorkflowExecutionCancelRequestedEventAttributes: FromWorkflowExecutionCancelRequestedEventAttributes(e.WorkflowExecutionCancelRequestedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionCanceledEventAttributes{\n\t\t\tWorkflowExecutionCanceledEventAttributes: FromWorkflowExecutionCanceledEventAttributes(e.WorkflowExecutionCanceledEventAttributes),\n\t\t}\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\tevent.Attributes = &apiv1.HistoryEvent_RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(e.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes),\n\t\t}\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes: FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(e.RequestCancelExternalWorkflowExecutionFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeExternalWorkflowExecutionCancelRequested:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\t\tExternalWorkflowExecutionCancelRequestedEventAttributes: FromExternalWorkflowExecutionCancelRequestedEventAttributes(e.ExternalWorkflowExecutionCancelRequestedEventAttributes),\n\t\t}\n\tcase types.EventTypeMarkerRecorded:\n\t\tevent.Attributes = &apiv1.HistoryEvent_MarkerRecordedEventAttributes{\n\t\t\tMarkerRecordedEventAttributes: FromMarkerRecordedEventAttributes(e.MarkerRecordedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionSignaledEventAttributes{\n\t\t\tWorkflowExecutionSignaledEventAttributes: FromWorkflowExecutionSignaledEventAttributes(e.WorkflowExecutionSignaledEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionTerminatedEventAttributes{\n\t\t\tWorkflowExecutionTerminatedEventAttributes: FromWorkflowExecutionTerminatedEventAttributes(e.WorkflowExecutionTerminatedEventAttributes),\n\t\t}\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tevent.Attributes = &apiv1.HistoryEvent_WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: FromWorkflowExecutionContinuedAsNewEventAttributes(e.WorkflowExecutionContinuedAsNewEventAttributes),\n\t\t}\n\tcase types.EventTypeStartChildWorkflowExecutionInitiated:\n\t\tevent.Attributes = &apiv1.HistoryEvent_StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: FromStartChildWorkflowExecutionInitiatedEventAttributes(e.StartChildWorkflowExecutionInitiatedEventAttributes),\n\t\t}\n\tcase types.EventTypeStartChildWorkflowExecutionFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_StartChildWorkflowExecutionFailedEventAttributes{\n\t\t\tStartChildWorkflowExecutionFailedEventAttributes: FromStartChildWorkflowExecutionFailedEventAttributes(e.StartChildWorkflowExecutionFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionStartedEventAttributes{\n\t\t\tChildWorkflowExecutionStartedEventAttributes: FromChildWorkflowExecutionStartedEventAttributes(e.ChildWorkflowExecutionStartedEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\tChildWorkflowExecutionCompletedEventAttributes: FromChildWorkflowExecutionCompletedEventAttributes(e.ChildWorkflowExecutionCompletedEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionFailedEventAttributes{\n\t\t\tChildWorkflowExecutionFailedEventAttributes: FromChildWorkflowExecutionFailedEventAttributes(e.ChildWorkflowExecutionFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\tChildWorkflowExecutionCanceledEventAttributes: FromChildWorkflowExecutionCanceledEventAttributes(e.ChildWorkflowExecutionCanceledEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\tChildWorkflowExecutionTimedOutEventAttributes: FromChildWorkflowExecutionTimedOutEventAttributes(e.ChildWorkflowExecutionTimedOutEventAttributes),\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\tChildWorkflowExecutionTerminatedEventAttributes: FromChildWorkflowExecutionTerminatedEventAttributes(e.ChildWorkflowExecutionTerminatedEventAttributes),\n\t\t}\n\tcase types.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\tevent.Attributes = &apiv1.HistoryEvent_SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes: FromSignalExternalWorkflowExecutionInitiatedEventAttributes(e.SignalExternalWorkflowExecutionInitiatedEventAttributes),\n\t\t}\n\tcase types.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\tevent.Attributes = &apiv1.HistoryEvent_SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\t\tSignalExternalWorkflowExecutionFailedEventAttributes: FromSignalExternalWorkflowExecutionFailedEventAttributes(e.SignalExternalWorkflowExecutionFailedEventAttributes),\n\t\t}\n\tcase types.EventTypeExternalWorkflowExecutionSignaled:\n\t\tevent.Attributes = &apiv1.HistoryEvent_ExternalWorkflowExecutionSignaledEventAttributes{\n\t\t\tExternalWorkflowExecutionSignaledEventAttributes: FromExternalWorkflowExecutionSignaledEventAttributes(e.ExternalWorkflowExecutionSignaledEventAttributes),\n\t\t}\n\tcase types.EventTypeUpsertWorkflowSearchAttributes:\n\t\tevent.Attributes = &apiv1.HistoryEvent_UpsertWorkflowSearchAttributesEventAttributes{\n\t\t\tUpsertWorkflowSearchAttributesEventAttributes: FromUpsertWorkflowSearchAttributesEventAttributes(e.UpsertWorkflowSearchAttributesEventAttributes),\n\t\t}\n\t}\n\treturn &event\n}\n\nfunc ToHistoryEvent(e *apiv1.HistoryEvent) *types.HistoryEvent {\n\tif e == nil {\n\t\treturn nil\n\t}\n\tevent := types.HistoryEvent{\n\t\tID:        e.EventId,\n\t\tTimestamp: timeToUnixNano(e.EventTime),\n\t\tVersion:   e.Version,\n\t\tTaskID:    e.TaskId,\n\t}\n\tswitch attr := e.Attributes.(type) {\n\tcase *apiv1.HistoryEvent_WorkflowExecutionStartedEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionStarted.Ptr()\n\t\tevent.WorkflowExecutionStartedEventAttributes = ToWorkflowExecutionStartedEventAttributes(attr.WorkflowExecutionStartedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionCompletedEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionCompleted.Ptr()\n\t\tevent.WorkflowExecutionCompletedEventAttributes = ToWorkflowExecutionCompletedEventAttributes(attr.WorkflowExecutionCompletedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionFailed.Ptr()\n\t\tevent.WorkflowExecutionFailedEventAttributes = ToWorkflowExecutionFailedEventAttributes(attr.WorkflowExecutionFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionTimedOutEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionTimedOut.Ptr()\n\t\tevent.WorkflowExecutionTimedOutEventAttributes = ToWorkflowExecutionTimedOutEventAttributes(attr.WorkflowExecutionTimedOutEventAttributes)\n\tcase *apiv1.HistoryEvent_DecisionTaskScheduledEventAttributes:\n\t\tevent.EventType = types.EventTypeDecisionTaskScheduled.Ptr()\n\t\tevent.DecisionTaskScheduledEventAttributes = ToDecisionTaskScheduledEventAttributes(attr.DecisionTaskScheduledEventAttributes)\n\tcase *apiv1.HistoryEvent_DecisionTaskStartedEventAttributes:\n\t\tevent.EventType = types.EventTypeDecisionTaskStarted.Ptr()\n\t\tevent.DecisionTaskStartedEventAttributes = ToDecisionTaskStartedEventAttributes(attr.DecisionTaskStartedEventAttributes)\n\tcase *apiv1.HistoryEvent_DecisionTaskCompletedEventAttributes:\n\t\tevent.EventType = types.EventTypeDecisionTaskCompleted.Ptr()\n\t\tevent.DecisionTaskCompletedEventAttributes = ToDecisionTaskCompletedEventAttributes(attr.DecisionTaskCompletedEventAttributes)\n\tcase *apiv1.HistoryEvent_DecisionTaskTimedOutEventAttributes:\n\t\tevent.EventType = types.EventTypeDecisionTaskTimedOut.Ptr()\n\t\tevent.DecisionTaskTimedOutEventAttributes = ToDecisionTaskTimedOutEventAttributes(attr.DecisionTaskTimedOutEventAttributes)\n\tcase *apiv1.HistoryEvent_DecisionTaskFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeDecisionTaskFailed.Ptr()\n\t\tevent.DecisionTaskFailedEventAttributes = ToDecisionTaskFailedEventAttributes(attr.DecisionTaskFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskScheduledEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskScheduled.Ptr()\n\t\tevent.ActivityTaskScheduledEventAttributes = ToActivityTaskScheduledEventAttributes(attr.ActivityTaskScheduledEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskStartedEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskStarted.Ptr()\n\t\tevent.ActivityTaskStartedEventAttributes = ToActivityTaskStartedEventAttributes(attr.ActivityTaskStartedEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskCompletedEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskCompleted.Ptr()\n\t\tevent.ActivityTaskCompletedEventAttributes = ToActivityTaskCompletedEventAttributes(attr.ActivityTaskCompletedEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskFailed.Ptr()\n\t\tevent.ActivityTaskFailedEventAttributes = ToActivityTaskFailedEventAttributes(attr.ActivityTaskFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskTimedOutEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskTimedOut.Ptr()\n\t\tevent.ActivityTaskTimedOutEventAttributes = ToActivityTaskTimedOutEventAttributes(attr.ActivityTaskTimedOutEventAttributes)\n\tcase *apiv1.HistoryEvent_TimerStartedEventAttributes:\n\t\tevent.EventType = types.EventTypeTimerStarted.Ptr()\n\t\tevent.TimerStartedEventAttributes = ToTimerStartedEventAttributes(attr.TimerStartedEventAttributes)\n\tcase *apiv1.HistoryEvent_TimerFiredEventAttributes:\n\t\tevent.EventType = types.EventTypeTimerFired.Ptr()\n\t\tevent.TimerFiredEventAttributes = ToTimerFiredEventAttributes(attr.TimerFiredEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskCancelRequestedEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskCancelRequested.Ptr()\n\t\tevent.ActivityTaskCancelRequestedEventAttributes = ToActivityTaskCancelRequestedEventAttributes(attr.ActivityTaskCancelRequestedEventAttributes)\n\tcase *apiv1.HistoryEvent_RequestCancelActivityTaskFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeRequestCancelActivityTaskFailed.Ptr()\n\t\tevent.RequestCancelActivityTaskFailedEventAttributes = ToRequestCancelActivityTaskFailedEventAttributes(attr.RequestCancelActivityTaskFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ActivityTaskCanceledEventAttributes:\n\t\tevent.EventType = types.EventTypeActivityTaskCanceled.Ptr()\n\t\tevent.ActivityTaskCanceledEventAttributes = ToActivityTaskCanceledEventAttributes(attr.ActivityTaskCanceledEventAttributes)\n\tcase *apiv1.HistoryEvent_TimerCanceledEventAttributes:\n\t\tevent.EventType = types.EventTypeTimerCanceled.Ptr()\n\t\tevent.TimerCanceledEventAttributes = ToTimerCanceledEventAttributes(attr.TimerCanceledEventAttributes)\n\tcase *apiv1.HistoryEvent_CancelTimerFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeCancelTimerFailed.Ptr()\n\t\tevent.CancelTimerFailedEventAttributes = ToCancelTimerFailedEventAttributes(attr.CancelTimerFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_MarkerRecordedEventAttributes:\n\t\tevent.EventType = types.EventTypeMarkerRecorded.Ptr()\n\t\tevent.MarkerRecordedEventAttributes = ToMarkerRecordedEventAttributes(attr.MarkerRecordedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionSignaledEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionSignaled.Ptr()\n\t\tevent.WorkflowExecutionSignaledEventAttributes = ToWorkflowExecutionSignaledEventAttributes(attr.WorkflowExecutionSignaledEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionTerminatedEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionTerminated.Ptr()\n\t\tevent.WorkflowExecutionTerminatedEventAttributes = ToWorkflowExecutionTerminatedEventAttributes(attr.WorkflowExecutionTerminatedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionCancelRequestedEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionCancelRequested.Ptr()\n\t\tevent.WorkflowExecutionCancelRequestedEventAttributes = ToWorkflowExecutionCancelRequestedEventAttributes(attr.WorkflowExecutionCancelRequestedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionCanceledEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionCanceled.Ptr()\n\t\tevent.WorkflowExecutionCanceledEventAttributes = ToWorkflowExecutionCanceledEventAttributes(attr.WorkflowExecutionCanceledEventAttributes)\n\tcase *apiv1.HistoryEvent_RequestCancelExternalWorkflowExecutionInitiatedEventAttributes:\n\t\tevent.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr()\n\t\tevent.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes = ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(attr.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes)\n\tcase *apiv1.HistoryEvent_RequestCancelExternalWorkflowExecutionFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionFailed.Ptr()\n\t\tevent.RequestCancelExternalWorkflowExecutionFailedEventAttributes = ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(attr.RequestCancelExternalWorkflowExecutionFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ExternalWorkflowExecutionCancelRequestedEventAttributes:\n\t\tevent.EventType = types.EventTypeExternalWorkflowExecutionCancelRequested.Ptr()\n\t\tevent.ExternalWorkflowExecutionCancelRequestedEventAttributes = ToExternalWorkflowExecutionCancelRequestedEventAttributes(attr.ExternalWorkflowExecutionCancelRequestedEventAttributes)\n\tcase *apiv1.HistoryEvent_WorkflowExecutionContinuedAsNewEventAttributes:\n\t\tevent.EventType = types.EventTypeWorkflowExecutionContinuedAsNew.Ptr()\n\t\tevent.WorkflowExecutionContinuedAsNewEventAttributes = ToWorkflowExecutionContinuedAsNewEventAttributes(attr.WorkflowExecutionContinuedAsNewEventAttributes)\n\tcase *apiv1.HistoryEvent_StartChildWorkflowExecutionInitiatedEventAttributes:\n\t\tevent.EventType = types.EventTypeStartChildWorkflowExecutionInitiated.Ptr()\n\t\tevent.StartChildWorkflowExecutionInitiatedEventAttributes = ToStartChildWorkflowExecutionInitiatedEventAttributes(attr.StartChildWorkflowExecutionInitiatedEventAttributes)\n\tcase *apiv1.HistoryEvent_StartChildWorkflowExecutionFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeStartChildWorkflowExecutionFailed.Ptr()\n\t\tevent.StartChildWorkflowExecutionFailedEventAttributes = ToStartChildWorkflowExecutionFailedEventAttributes(attr.StartChildWorkflowExecutionFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionStartedEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionStarted.Ptr()\n\t\tevent.ChildWorkflowExecutionStartedEventAttributes = ToChildWorkflowExecutionStartedEventAttributes(attr.ChildWorkflowExecutionStartedEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionCompletedEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionCompleted.Ptr()\n\t\tevent.ChildWorkflowExecutionCompletedEventAttributes = ToChildWorkflowExecutionCompletedEventAttributes(attr.ChildWorkflowExecutionCompletedEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionFailed.Ptr()\n\t\tevent.ChildWorkflowExecutionFailedEventAttributes = ToChildWorkflowExecutionFailedEventAttributes(attr.ChildWorkflowExecutionFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionCanceledEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionCanceled.Ptr()\n\t\tevent.ChildWorkflowExecutionCanceledEventAttributes = ToChildWorkflowExecutionCanceledEventAttributes(attr.ChildWorkflowExecutionCanceledEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionTimedOutEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionTimedOut.Ptr()\n\t\tevent.ChildWorkflowExecutionTimedOutEventAttributes = ToChildWorkflowExecutionTimedOutEventAttributes(attr.ChildWorkflowExecutionTimedOutEventAttributes)\n\tcase *apiv1.HistoryEvent_ChildWorkflowExecutionTerminatedEventAttributes:\n\t\tevent.EventType = types.EventTypeChildWorkflowExecutionTerminated.Ptr()\n\t\tevent.ChildWorkflowExecutionTerminatedEventAttributes = ToChildWorkflowExecutionTerminatedEventAttributes(attr.ChildWorkflowExecutionTerminatedEventAttributes)\n\tcase *apiv1.HistoryEvent_SignalExternalWorkflowExecutionInitiatedEventAttributes:\n\t\tevent.EventType = types.EventTypeSignalExternalWorkflowExecutionInitiated.Ptr()\n\t\tevent.SignalExternalWorkflowExecutionInitiatedEventAttributes = ToSignalExternalWorkflowExecutionInitiatedEventAttributes(attr.SignalExternalWorkflowExecutionInitiatedEventAttributes)\n\tcase *apiv1.HistoryEvent_SignalExternalWorkflowExecutionFailedEventAttributes:\n\t\tevent.EventType = types.EventTypeSignalExternalWorkflowExecutionFailed.Ptr()\n\t\tevent.SignalExternalWorkflowExecutionFailedEventAttributes = ToSignalExternalWorkflowExecutionFailedEventAttributes(attr.SignalExternalWorkflowExecutionFailedEventAttributes)\n\tcase *apiv1.HistoryEvent_ExternalWorkflowExecutionSignaledEventAttributes:\n\t\tevent.EventType = types.EventTypeExternalWorkflowExecutionSignaled.Ptr()\n\t\tevent.ExternalWorkflowExecutionSignaledEventAttributes = ToExternalWorkflowExecutionSignaledEventAttributes(attr.ExternalWorkflowExecutionSignaledEventAttributes)\n\tcase *apiv1.HistoryEvent_UpsertWorkflowSearchAttributesEventAttributes:\n\t\tevent.EventType = types.EventTypeUpsertWorkflowSearchAttributes.Ptr()\n\t\tevent.UpsertWorkflowSearchAttributesEventAttributes = ToUpsertWorkflowSearchAttributesEventAttributes(attr.UpsertWorkflowSearchAttributesEventAttributes)\n\t}\n\treturn &event\n}\n\nfunc FromDecision(d *types.Decision) *apiv1.Decision {\n\tif d == nil {\n\t\treturn nil\n\t}\n\tdecision := apiv1.Decision{}\n\tswitch *d.DecisionType {\n\tcase types.DecisionTypeScheduleActivityTask:\n\t\tdecision.Attributes = &apiv1.Decision_ScheduleActivityTaskDecisionAttributes{\n\t\t\tScheduleActivityTaskDecisionAttributes: FromScheduleActivityTaskDecisionAttributes(d.ScheduleActivityTaskDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeRequestCancelActivityTask:\n\t\tdecision.Attributes = &apiv1.Decision_RequestCancelActivityTaskDecisionAttributes{\n\t\t\tRequestCancelActivityTaskDecisionAttributes: FromRequestCancelActivityTaskDecisionAttributes(d.RequestCancelActivityTaskDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeStartTimer:\n\t\tdecision.Attributes = &apiv1.Decision_StartTimerDecisionAttributes{\n\t\t\tStartTimerDecisionAttributes: FromStartTimerDecisionAttributes(d.StartTimerDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeCompleteWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_CompleteWorkflowExecutionDecisionAttributes{\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: FromCompleteWorkflowExecutionDecisionAttributes(d.CompleteWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeFailWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_FailWorkflowExecutionDecisionAttributes{\n\t\t\tFailWorkflowExecutionDecisionAttributes: FromFailWorkflowExecutionDecisionAttributes(d.FailWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeCancelTimer:\n\t\tdecision.Attributes = &apiv1.Decision_CancelTimerDecisionAttributes{\n\t\t\tCancelTimerDecisionAttributes: FromCancelTimerDecisionAttributes(d.CancelTimerDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeCancelWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_CancelWorkflowExecutionDecisionAttributes{\n\t\t\tCancelWorkflowExecutionDecisionAttributes: FromCancelWorkflowExecutionDecisionAttributes(d.CancelWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeRequestCancelExternalWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: FromRequestCancelExternalWorkflowExecutionDecisionAttributes(d.RequestCancelExternalWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeRecordMarker:\n\t\tdecision.Attributes = &apiv1.Decision_RecordMarkerDecisionAttributes{\n\t\t\tRecordMarkerDecisionAttributes: FromRecordMarkerDecisionAttributes(d.RecordMarkerDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeContinueAsNewWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: FromContinueAsNewWorkflowExecutionDecisionAttributes(d.ContinueAsNewWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeStartChildWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tStartChildWorkflowExecutionDecisionAttributes: FromStartChildWorkflowExecutionDecisionAttributes(d.StartChildWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeSignalExternalWorkflowExecution:\n\t\tdecision.Attributes = &apiv1.Decision_SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: FromSignalExternalWorkflowExecutionDecisionAttributes(d.SignalExternalWorkflowExecutionDecisionAttributes),\n\t\t}\n\tcase types.DecisionTypeUpsertWorkflowSearchAttributes:\n\t\tdecision.Attributes = &apiv1.Decision_UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: FromUpsertWorkflowSearchAttributesDecisionAttributes(d.UpsertWorkflowSearchAttributesDecisionAttributes),\n\t\t}\n\t}\n\treturn &decision\n}\n\nfunc ToDecision(d *apiv1.Decision) *types.Decision {\n\tif d == nil {\n\t\treturn nil\n\t}\n\tdecision := types.Decision{}\n\tswitch attr := d.Attributes.(type) {\n\tcase *apiv1.Decision_ScheduleActivityTaskDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeScheduleActivityTask.Ptr()\n\t\tdecision.ScheduleActivityTaskDecisionAttributes = ToScheduleActivityTaskDecisionAttributes(attr.ScheduleActivityTaskDecisionAttributes)\n\tcase *apiv1.Decision_StartTimerDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeStartTimer.Ptr()\n\t\tdecision.StartTimerDecisionAttributes = ToStartTimerDecisionAttributes(attr.StartTimerDecisionAttributes)\n\tcase *apiv1.Decision_CompleteWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeCompleteWorkflowExecution.Ptr()\n\t\tdecision.CompleteWorkflowExecutionDecisionAttributes = ToCompleteWorkflowExecutionDecisionAttributes(attr.CompleteWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_FailWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeFailWorkflowExecution.Ptr()\n\t\tdecision.FailWorkflowExecutionDecisionAttributes = ToFailWorkflowExecutionDecisionAttributes(attr.FailWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_RequestCancelActivityTaskDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeRequestCancelActivityTask.Ptr()\n\t\tdecision.RequestCancelActivityTaskDecisionAttributes = ToRequestCancelActivityTaskDecisionAttributes(attr.RequestCancelActivityTaskDecisionAttributes)\n\tcase *apiv1.Decision_CancelTimerDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeCancelTimer.Ptr()\n\t\tdecision.CancelTimerDecisionAttributes = ToCancelTimerDecisionAttributes(attr.CancelTimerDecisionAttributes)\n\tcase *apiv1.Decision_CancelWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeCancelWorkflowExecution.Ptr()\n\t\tdecision.CancelWorkflowExecutionDecisionAttributes = ToCancelWorkflowExecutionDecisionAttributes(attr.CancelWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_RequestCancelExternalWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeRequestCancelExternalWorkflowExecution.Ptr()\n\t\tdecision.RequestCancelExternalWorkflowExecutionDecisionAttributes = ToRequestCancelExternalWorkflowExecutionDecisionAttributes(attr.RequestCancelExternalWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_RecordMarkerDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeRecordMarker.Ptr()\n\t\tdecision.RecordMarkerDecisionAttributes = ToRecordMarkerDecisionAttributes(attr.RecordMarkerDecisionAttributes)\n\tcase *apiv1.Decision_ContinueAsNewWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeContinueAsNewWorkflowExecution.Ptr()\n\t\tdecision.ContinueAsNewWorkflowExecutionDecisionAttributes = ToContinueAsNewWorkflowExecutionDecisionAttributes(attr.ContinueAsNewWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_StartChildWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeStartChildWorkflowExecution.Ptr()\n\t\tdecision.StartChildWorkflowExecutionDecisionAttributes = ToStartChildWorkflowExecutionDecisionAttributes(attr.StartChildWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_SignalExternalWorkflowExecutionDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeSignalExternalWorkflowExecution.Ptr()\n\t\tdecision.SignalExternalWorkflowExecutionDecisionAttributes = ToSignalExternalWorkflowExecutionDecisionAttributes(attr.SignalExternalWorkflowExecutionDecisionAttributes)\n\tcase *apiv1.Decision_UpsertWorkflowSearchAttributesDecisionAttributes:\n\t\tdecision.DecisionType = types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr()\n\t\tdecision.UpsertWorkflowSearchAttributesDecisionAttributes = ToUpsertWorkflowSearchAttributesDecisionAttributes(attr.UpsertWorkflowSearchAttributesDecisionAttributes)\n\t}\n\treturn &decision\n}\n\nfunc FromListClosedWorkflowExecutionsRequest(r *types.ListClosedWorkflowExecutionsRequest) *apiv1.ListClosedWorkflowExecutionsRequest {\n\tif r == nil {\n\t\treturn nil\n\t}\n\trequest := apiv1.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          r.Domain,\n\t\tPageSize:        r.MaximumPageSize,\n\t\tNextPageToken:   r.NextPageToken,\n\t\tStartTimeFilter: FromStartTimeFilter(r.StartTimeFilter),\n\t}\n\tif r.ExecutionFilter != nil {\n\t\trequest.Filters = &apiv1.ListClosedWorkflowExecutionsRequest_ExecutionFilter{\n\t\t\tExecutionFilter: FromWorkflowExecutionFilter(r.ExecutionFilter),\n\t\t}\n\t}\n\tif r.TypeFilter != nil {\n\t\trequest.Filters = &apiv1.ListClosedWorkflowExecutionsRequest_TypeFilter{\n\t\t\tTypeFilter: FromWorkflowTypeFilter(r.TypeFilter),\n\t\t}\n\t}\n\tif r.StatusFilter != nil {\n\t\trequest.Filters = &apiv1.ListClosedWorkflowExecutionsRequest_StatusFilter{\n\t\t\tStatusFilter: FromStatusFilter(r.StatusFilter),\n\t\t}\n\t}\n\n\treturn &request\n}\n\nfunc ToListClosedWorkflowExecutionsRequest(r *apiv1.ListClosedWorkflowExecutionsRequest) *types.ListClosedWorkflowExecutionsRequest {\n\tif r == nil {\n\t\treturn nil\n\t}\n\trequest := types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          r.Domain,\n\t\tMaximumPageSize: r.PageSize,\n\t\tNextPageToken:   r.NextPageToken,\n\t\tStartTimeFilter: ToStartTimeFilter(r.StartTimeFilter),\n\t}\n\tswitch filters := r.Filters.(type) {\n\tcase *apiv1.ListClosedWorkflowExecutionsRequest_ExecutionFilter:\n\t\trequest.ExecutionFilter = ToWorkflowExecutionFilter(filters.ExecutionFilter)\n\tcase *apiv1.ListClosedWorkflowExecutionsRequest_TypeFilter:\n\t\trequest.TypeFilter = ToWorkflowTypeFilter(filters.TypeFilter)\n\tcase *apiv1.ListClosedWorkflowExecutionsRequest_StatusFilter:\n\t\trequest.StatusFilter = ToStatusFilter(filters.StatusFilter)\n\t}\n\n\treturn &request\n}\n\nfunc FromListOpenWorkflowExecutionsRequest(r *types.ListOpenWorkflowExecutionsRequest) *apiv1.ListOpenWorkflowExecutionsRequest {\n\tif r == nil {\n\t\treturn nil\n\t}\n\trequest := apiv1.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          r.Domain,\n\t\tPageSize:        r.MaximumPageSize,\n\t\tNextPageToken:   r.NextPageToken,\n\t\tStartTimeFilter: FromStartTimeFilter(r.StartTimeFilter),\n\t}\n\tif r.ExecutionFilter != nil {\n\t\trequest.Filters = &apiv1.ListOpenWorkflowExecutionsRequest_ExecutionFilter{\n\t\t\tExecutionFilter: FromWorkflowExecutionFilter(r.ExecutionFilter),\n\t\t}\n\t}\n\tif r.TypeFilter != nil {\n\t\trequest.Filters = &apiv1.ListOpenWorkflowExecutionsRequest_TypeFilter{\n\t\t\tTypeFilter: FromWorkflowTypeFilter(r.TypeFilter),\n\t\t}\n\t}\n\n\treturn &request\n}\n\nfunc ToListOpenWorkflowExecutionsRequest(r *apiv1.ListOpenWorkflowExecutionsRequest) *types.ListOpenWorkflowExecutionsRequest {\n\tif r == nil {\n\t\treturn nil\n\t}\n\trequest := types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          r.Domain,\n\t\tMaximumPageSize: r.PageSize,\n\t\tNextPageToken:   r.NextPageToken,\n\t\tStartTimeFilter: ToStartTimeFilter(r.StartTimeFilter),\n\t}\n\tswitch filters := r.Filters.(type) {\n\tcase *apiv1.ListOpenWorkflowExecutionsRequest_ExecutionFilter:\n\t\trequest.ExecutionFilter = ToWorkflowExecutionFilter(filters.ExecutionFilter)\n\tcase *apiv1.ListOpenWorkflowExecutionsRequest_TypeFilter:\n\t\trequest.TypeFilter = ToWorkflowTypeFilter(filters.TypeFilter)\n\t}\n\treturn &request\n}\n\nfunc FromResetStickyTaskListResponse(t *types.ResetStickyTaskListResponse) *apiv1.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ResetStickyTaskListResponse{}\n}\n\nfunc ToResetStickyTaskListResponse(t *apiv1.ResetStickyTaskListResponse) *types.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetStickyTaskListResponse{}\n}\n\nfunc FromAPITaskListPartitionConfig(t *types.TaskListPartitionConfig) *apiv1.TaskListPartitionConfig {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskListPartitionConfig{\n\t\tVersion:            t.Version,\n\t\tNumReadPartitions:  int32(len(t.ReadPartitions)),\n\t\tNumWritePartitions: int32(len(t.WritePartitions)),\n\t\tReadPartitions:     FromAPITaskListPartitionsMap(t.ReadPartitions),\n\t\tWritePartitions:    FromAPITaskListPartitionsMap(t.WritePartitions),\n\t}\n}\n\nfunc ToAPITaskListPartitionConfig(t *apiv1.TaskListPartitionConfig) *types.TaskListPartitionConfig {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartitionConfig{\n\t\tVersion:         t.Version,\n\t\tReadPartitions:  ToAPITaskListPartitionsMap(t.NumReadPartitions, t.ReadPartitions),\n\t\tWritePartitions: ToAPITaskListPartitionsMap(t.NumWritePartitions, t.WritePartitions),\n\t}\n}\n\nfunc FromAPITaskListPartition(t *types.TaskListPartition) *apiv1.TaskListPartition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.TaskListPartition{\n\t\tIsolationGroups: t.IsolationGroups,\n\t}\n}\n\nfunc ToAPITaskListPartition(t *apiv1.TaskListPartition) *types.TaskListPartition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartition{\n\t\tIsolationGroups: t.IsolationGroups,\n\t}\n}\n\nfunc FromAPITaskListPartitionsMap(m map[int]*types.TaskListPartition) map[int32]*apiv1.TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int32]*apiv1.TaskListPartition, len(m))\n\tfor id, p := range m {\n\t\tresult[int32(id)] = FromAPITaskListPartition(p)\n\t}\n\treturn result\n}\n\nfunc ToAPITaskListPartitionsMap(numPartitions int32, m map[int32]*apiv1.TaskListPartition) map[int]*types.TaskListPartition {\n\tif m == nil && numPartitions == 0 {\n\t\treturn nil\n\t}\n\tresult := make(map[int]*types.TaskListPartition, len(m))\n\tif numPartitions != int32(len(m)) {\n\t\tfor i := int32(0); i < numPartitions; i++ {\n\t\t\tresult[int(i)] = &types.TaskListPartition{}\n\t\t}\n\t} else {\n\t\tfor id, p := range m {\n\t\t\tresult[int(id)] = ToAPITaskListPartition(p)\n\t\t}\n\t}\n\treturn result\n}\n\nfunc FromAutoConfigHint(t *types.AutoConfigHint) *apiv1.AutoConfigHint {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.AutoConfigHint{\n\t\tPollerWaitTimeInMs: t.PollerWaitTimeInMs,\n\t\tEnableAutoConfig:   t.EnableAutoConfig,\n\t}\n}\n\nfunc ToAutoConfigHint(t *apiv1.AutoConfigHint) *types.AutoConfigHint {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AutoConfigHint{\n\t\tPollerWaitTimeInMs: t.PollerWaitTimeInMs,\n\t\tEnableAutoConfig:   t.EnableAutoConfig,\n\t}\n}\n\nfunc FromCronOverlapPolicy(p *types.CronOverlapPolicy) apiv1.CronOverlapPolicy {\n\tif p == nil {\n\t\treturn apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_INVALID\n\t}\n\tswitch *p {\n\tcase types.CronOverlapPolicyBufferOne:\n\t\treturn apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_BUFFER_ONE\n\tcase types.CronOverlapPolicySkipped:\n\t\treturn apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_SKIPPED\n\t}\n\treturn apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_INVALID\n}\n\nfunc ToCronOverlapPolicy(p apiv1.CronOverlapPolicy) *types.CronOverlapPolicy {\n\tswitch p {\n\tcase apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_BUFFER_ONE:\n\t\treturn types.CronOverlapPolicyBufferOne.Ptr()\n\tcase apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_SKIPPED:\n\t\treturn types.CronOverlapPolicySkipped.Ptr()\n\tcase apiv1.CronOverlapPolicy_CRON_OVERLAP_POLICY_INVALID:\n\t\treturn nil\n\t}\n\treturn nil\n}\n\nfunc FromClusterAttribute(c *types.ClusterAttribute) *apiv1.ClusterAttribute {\n\tif c == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ClusterAttribute{\n\t\tScope: c.Scope,\n\t\tName:  c.Name,\n\t}\n}\n\nfunc ToClusterAttribute(c *apiv1.ClusterAttribute) *types.ClusterAttribute {\n\tif c == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterAttribute{\n\t\tScope: c.Scope,\n\t\tName:  c.Name,\n\t}\n}\n\nfunc FromActiveClusterSelectionPolicy(p *types.ActiveClusterSelectionPolicy) *apiv1.ActiveClusterSelectionPolicy {\n\tif p == nil {\n\t\treturn nil\n\t}\n\t// TODO(active-active): Remove the switch statement once the strategy is removed\n\tif p.ActiveClusterSelectionStrategy == nil {\n\t\treturn &apiv1.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: FromClusterAttribute(p.ClusterAttribute),\n\t\t}\n\t}\n\tswitch *p.ActiveClusterSelectionStrategy {\n\tcase types.ActiveClusterSelectionStrategyRegionSticky:\n\t\treturn &apiv1.ActiveClusterSelectionPolicy{\n\t\t\tStrategy: apiv1.ActiveClusterSelectionStrategy_ACTIVE_CLUSTER_SELECTION_STRATEGY_REGION_STICKY,\n\t\t\tStrategyConfig: &apiv1.ActiveClusterSelectionPolicy_ActiveClusterStickyRegionConfig{\n\t\t\t\tActiveClusterStickyRegionConfig: &apiv1.ActiveClusterStickyRegionConfig{\n\t\t\t\t\tStickyRegion: p.StickyRegion,\n\t\t\t\t},\n\t\t\t},\n\t\t\tClusterAttribute: FromClusterAttribute(p.ClusterAttribute),\n\t\t}\n\tcase types.ActiveClusterSelectionStrategyExternalEntity:\n\t\treturn &apiv1.ActiveClusterSelectionPolicy{\n\t\t\tStrategy: apiv1.ActiveClusterSelectionStrategy_ACTIVE_CLUSTER_SELECTION_STRATEGY_EXTERNAL_ENTITY,\n\t\t\tStrategyConfig: &apiv1.ActiveClusterSelectionPolicy_ActiveClusterExternalEntityConfig{\n\t\t\t\tActiveClusterExternalEntityConfig: &apiv1.ActiveClusterExternalEntityConfig{\n\t\t\t\t\tExternalEntityType: p.ExternalEntityType,\n\t\t\t\t\tExternalEntityKey:  p.ExternalEntityKey,\n\t\t\t\t},\n\t\t\t},\n\t\t\tClusterAttribute: FromClusterAttribute(p.ClusterAttribute),\n\t\t}\n\t}\n\treturn &apiv1.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: FromClusterAttribute(p.ClusterAttribute),\n\t}\n}\n\nfunc ToActiveClusterSelectionPolicy(p *apiv1.ActiveClusterSelectionPolicy) *types.ActiveClusterSelectionPolicy {\n\tif p == nil {\n\t\treturn nil\n\t}\n\t// TODO(active-active): Remove the switch statement once the strategy is removed\n\tswitch p.Strategy {\n\tcase apiv1.ActiveClusterSelectionStrategy_ACTIVE_CLUSTER_SELECTION_STRATEGY_REGION_STICKY:\n\t\treturn &types.ActiveClusterSelectionPolicy{\n\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\t\tStickyRegion:                   p.StrategyConfig.(*apiv1.ActiveClusterSelectionPolicy_ActiveClusterStickyRegionConfig).ActiveClusterStickyRegionConfig.StickyRegion,\n\t\t\tClusterAttribute:               ToClusterAttribute(p.ClusterAttribute),\n\t\t}\n\tcase apiv1.ActiveClusterSelectionStrategy_ACTIVE_CLUSTER_SELECTION_STRATEGY_EXTERNAL_ENTITY:\n\t\treturn &types.ActiveClusterSelectionPolicy{\n\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyExternalEntity.Ptr(),\n\t\t\tExternalEntityType:             p.StrategyConfig.(*apiv1.ActiveClusterSelectionPolicy_ActiveClusterExternalEntityConfig).ActiveClusterExternalEntityConfig.ExternalEntityType,\n\t\t\tExternalEntityKey:              p.StrategyConfig.(*apiv1.ActiveClusterSelectionPolicy_ActiveClusterExternalEntityConfig).ActiveClusterExternalEntityConfig.ExternalEntityKey,\n\t\t\tClusterAttribute:               ToClusterAttribute(p.ClusterAttribute),\n\t\t}\n\t}\n\treturn &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: ToClusterAttribute(p.ClusterAttribute),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/api_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/testutils\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestActivityLocalDispatchInfo(t *testing.T) {\n\tfor _, item := range []*types.ActivityLocalDispatchInfo{nil, {}, &testdata.ActivityLocalDispatchInfo} {\n\t\tassert.Equal(t, item, ToActivityLocalDispatchInfo(FromActivityLocalDispatchInfo(item)))\n\t}\n}\nfunc TestActivityTaskCancelRequestedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskCancelRequestedEventAttributes{nil, {}, &testdata.ActivityTaskCancelRequestedEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskCancelRequestedEventAttributes(FromActivityTaskCancelRequestedEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskCanceledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskCanceledEventAttributes{nil, {}, &testdata.ActivityTaskCanceledEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskCanceledEventAttributes(FromActivityTaskCanceledEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskCompletedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskCompletedEventAttributes{nil, {}, &testdata.ActivityTaskCompletedEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskCompletedEventAttributes(FromActivityTaskCompletedEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskFailedEventAttributes{nil, {}, &testdata.ActivityTaskFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskFailedEventAttributes(FromActivityTaskFailedEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskScheduledEventAttributes(t *testing.T) {\n\t// since proto definition for Domain field doesn't have pointer, To(From(item)) won't be equal to item when item's Domain is a nil pointer\n\t// this is fine as the code using this field will check both if the field is a nil pointer and if it's a pointer to an empty string.\n\tfor _, item := range []*types.ActivityTaskScheduledEventAttributes{nil, {Domain: common.StringPtr(\"\")}, &testdata.ActivityTaskScheduledEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskScheduledEventAttributes(FromActivityTaskScheduledEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskStartedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskStartedEventAttributes{nil, {}, &testdata.ActivityTaskStartedEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskStartedEventAttributes(FromActivityTaskStartedEventAttributes(item)))\n\t}\n}\nfunc TestActivityTaskTimedOutEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ActivityTaskTimedOutEventAttributes{nil, {}, &testdata.ActivityTaskTimedOutEventAttributes} {\n\t\tassert.Equal(t, item, ToActivityTaskTimedOutEventAttributes(FromActivityTaskTimedOutEventAttributes(item)))\n\t}\n}\nfunc TestActivityType(t *testing.T) {\n\tfor _, item := range []*types.ActivityType{nil, {}, &testdata.ActivityType} {\n\t\tassert.Equal(t, item, ToActivityType(FromActivityType(item)))\n\t}\n}\nfunc TestBadBinaries(t *testing.T) {\n\tfor _, item := range []*types.BadBinaries{nil, {}, &testdata.BadBinaries} {\n\t\tassert.Equal(t, item, ToBadBinaries(FromBadBinaries(item)))\n\t}\n}\nfunc TestBadBinaryInfo(t *testing.T) {\n\tfor _, item := range []*types.BadBinaryInfo{nil, {}, &testdata.BadBinaryInfo} {\n\t\tassert.Equal(t, item, ToBadBinaryInfo(FromBadBinaryInfo(item)))\n\t}\n}\nfunc TestCancelTimerDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.CancelTimerDecisionAttributes{nil, {}, &testdata.CancelTimerDecisionAttributes} {\n\t\tassert.Equal(t, item, ToCancelTimerDecisionAttributes(FromCancelTimerDecisionAttributes(item)))\n\t}\n}\nfunc TestCancelTimerFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.CancelTimerFailedEventAttributes{nil, {}, &testdata.CancelTimerFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToCancelTimerFailedEventAttributes(FromCancelTimerFailedEventAttributes(item)))\n\t}\n}\nfunc TestCancelWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.CancelWorkflowExecutionDecisionAttributes{nil, {}, &testdata.CancelWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToCancelWorkflowExecutionDecisionAttributes(FromCancelWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionCanceledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionCanceledEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionCanceledEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionCanceledEventAttributes(FromChildWorkflowExecutionCanceledEventAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionCompletedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionCompletedEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionCompletedEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionCompletedEventAttributes(FromChildWorkflowExecutionCompletedEventAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionFailedEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionFailedEventAttributes(FromChildWorkflowExecutionFailedEventAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionStartedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionStartedEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionStartedEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionStartedEventAttributes(FromChildWorkflowExecutionStartedEventAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionTerminatedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionTerminatedEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionTerminatedEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionTerminatedEventAttributes(FromChildWorkflowExecutionTerminatedEventAttributes(item)))\n\t}\n}\nfunc TestChildWorkflowExecutionTimedOutEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionTimedOutEventAttributes{nil, {}, &testdata.ChildWorkflowExecutionTimedOutEventAttributes} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionTimedOutEventAttributes(FromChildWorkflowExecutionTimedOutEventAttributes(item)))\n\t}\n}\nfunc TestClusterReplicationConfiguration(t *testing.T) {\n\tfor _, item := range []*types.ClusterReplicationConfiguration{nil, {}, &testdata.ClusterReplicationConfiguration} {\n\t\tassert.Equal(t, item, ToClusterReplicationConfiguration(FromClusterReplicationConfiguration(item)))\n\t}\n}\nfunc TestCompleteWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.CompleteWorkflowExecutionDecisionAttributes{nil, {}, &testdata.CompleteWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToCompleteWorkflowExecutionDecisionAttributes(FromCompleteWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestContinueAsNewWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.ContinueAsNewWorkflowExecutionDecisionAttributes{nil, {}, &testdata.ContinueAsNewWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToContinueAsNewWorkflowExecutionDecisionAttributes(FromContinueAsNewWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestCountWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.CountWorkflowExecutionsRequest{nil, {}, &testdata.CountWorkflowExecutionsRequest} {\n\t\tassert.Equal(t, item, ToCountWorkflowExecutionsRequest(FromCountWorkflowExecutionsRequest(item)))\n\t}\n}\nfunc TestCountWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.CountWorkflowExecutionsResponse{nil, {}, &testdata.CountWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToCountWorkflowExecutionsResponse(FromCountWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestDataBlob(t *testing.T) {\n\tfor _, item := range []*types.DataBlob{nil, {}, &testdata.DataBlob} {\n\t\tassert.Equal(t, item, ToDataBlob(FromDataBlob(item)))\n\t}\n}\nfunc TestDecisionTaskCompletedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskCompletedEventAttributes{nil, {}, &testdata.DecisionTaskCompletedEventAttributes} {\n\t\tassert.Equal(t, item, ToDecisionTaskCompletedEventAttributes(FromDecisionTaskCompletedEventAttributes(item)))\n\t}\n}\nfunc TestDecisionTaskFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskFailedEventAttributes{nil, {}, &testdata.DecisionTaskFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToDecisionTaskFailedEventAttributes(FromDecisionTaskFailedEventAttributes(item)))\n\t}\n}\nfunc TestDecisionTaskScheduledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskScheduledEventAttributes{nil, {}, &testdata.DecisionTaskScheduledEventAttributes} {\n\t\tassert.Equal(t, item, ToDecisionTaskScheduledEventAttributes(FromDecisionTaskScheduledEventAttributes(item)))\n\t}\n}\nfunc TestDecisionTaskStartedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskStartedEventAttributes{nil, {}, &testdata.DecisionTaskStartedEventAttributes} {\n\t\tassert.Equal(t, item, ToDecisionTaskStartedEventAttributes(FromDecisionTaskStartedEventAttributes(item)))\n\t}\n}\nfunc TestDecisionTaskTimedOutEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskTimedOutEventAttributes{nil, {}, &testdata.DecisionTaskTimedOutEventAttributes} {\n\t\tassert.Equal(t, item, ToDecisionTaskTimedOutEventAttributes(FromDecisionTaskTimedOutEventAttributes(item)))\n\t}\n}\nfunc TestDeleteDomainRequest(t *testing.T) {\n\tfor _, item := range []*types.DeleteDomainRequest{nil, {}, &testdata.DeleteDomainRequest} {\n\t\tassert.Equal(t, item, ToDeleteDomainRequest(FromDeleteDomainRequest(item)))\n\t}\n}\nfunc TestDeprecateDomainRequest(t *testing.T) {\n\tfor _, item := range []*types.DeprecateDomainRequest{nil, {}, &testdata.DeprecateDomainRequest} {\n\t\tassert.Equal(t, item, ToDeprecateDomainRequest(FromDeprecateDomainRequest(item)))\n\t}\n}\nfunc TestDescribeDomainRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeDomainRequest{\n\t\t&testdata.DescribeDomainRequest_ID,\n\t\t&testdata.DescribeDomainRequest_Name,\n\t} {\n\t\tassert.Equal(t, item, ToDescribeDomainRequest(FromDescribeDomainRequest(item)))\n\t}\n\tassert.Nil(t, ToDescribeDomainRequest(nil))\n\tassert.Nil(t, FromDescribeDomainRequest(nil))\n\tassert.Panics(t, func() { ToDescribeDomainRequest(&apiv1.DescribeDomainRequest{}) })\n\tassert.Panics(t, func() { FromDescribeDomainRequest(&types.DescribeDomainRequest{}) })\n}\nfunc TestDescribeDomainResponse_Domain(t *testing.T) {\n\tfor _, item := range []*types.DescribeDomainResponse{nil, &testdata.DescribeDomainResponse} {\n\t\tassert.Equal(t, item, ToDescribeDomainResponseDomain(FromDescribeDomainResponseDomain(item)))\n\t}\n}\nfunc TestDescribeDomainResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeDomainResponse{nil, &testdata.DescribeDomainResponse} {\n\t\tassert.Equal(t, item, ToDescribeDomainResponse(FromDescribeDomainResponse(item)))\n\t}\n}\nfunc TestDescribeTaskListRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeTaskListRequest{nil, {}, &testdata.DescribeTaskListRequest} {\n\t\tassert.Equal(t, item, ToDescribeTaskListRequest(FromDescribeTaskListRequest(item)))\n\t}\n}\nfunc TestDescribeTaskListResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeTaskListResponse{nil, {}, &testdata.DescribeTaskListResponse} {\n\t\tassert.Equal(t, item, ToDescribeTaskListResponse(FromDescribeTaskListResponse(item)))\n\t}\n}\nfunc TestDescribeWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeWorkflowExecutionRequest{nil, {}, &testdata.DescribeWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToDescribeWorkflowExecutionRequest(FromDescribeWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestDescribeWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeWorkflowExecutionResponse{nil, {}, &testdata.DescribeWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToDescribeWorkflowExecutionResponse(FromDescribeWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestDiagnoseWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.DiagnoseWorkflowExecutionRequest{nil, {}, &testdata.DiagnoseWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToDiagnoseWorkflowExecutionRequest(FromDiagnoseWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestDiagnoseWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.DiagnoseWorkflowExecutionResponse{nil, {}, &testdata.DiagnoseWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToDiagnoseWorkflowExecutionResponse(FromDiagnoseWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestExternalWorkflowExecutionCancelRequestedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ExternalWorkflowExecutionCancelRequestedEventAttributes{nil, {}, &testdata.ExternalWorkflowExecutionCancelRequestedEventAttributes} {\n\t\tassert.Equal(t, item, ToExternalWorkflowExecutionCancelRequestedEventAttributes(FromExternalWorkflowExecutionCancelRequestedEventAttributes(item)))\n\t}\n}\nfunc TestExternalWorkflowExecutionSignaledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.ExternalWorkflowExecutionSignaledEventAttributes{nil, {}, &testdata.ExternalWorkflowExecutionSignaledEventAttributes} {\n\t\tassert.Equal(t, item, ToExternalWorkflowExecutionSignaledEventAttributes(FromExternalWorkflowExecutionSignaledEventAttributes(item)))\n\t}\n}\nfunc TestFailWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.FailWorkflowExecutionDecisionAttributes{nil, {}, &testdata.FailWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToFailWorkflowExecutionDecisionAttributes(FromFailWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestGetClusterInfoResponse(t *testing.T) {\n\tfor _, item := range []*types.ClusterInfo{nil, {}, &testdata.ClusterInfo} {\n\t\tassert.Equal(t, item, ToGetClusterInfoResponse(FromGetClusterInfoResponse(item)))\n\t}\n}\nfunc TestGetSearchAttributesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetSearchAttributesResponse{nil, {}, &testdata.GetSearchAttributesResponse} {\n\t\tassert.Equal(t, item, ToGetSearchAttributesResponse(FromGetSearchAttributesResponse(item)))\n\t}\n}\nfunc TestGetWorkflowExecutionHistoryRequest(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionHistoryRequest{nil, {}, &testdata.GetWorkflowExecutionHistoryRequest} {\n\t\tassert.Equal(t, item, ToGetWorkflowExecutionHistoryRequest(FromGetWorkflowExecutionHistoryRequest(item)))\n\t}\n}\nfunc TestGetWorkflowExecutionHistoryResponse(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionHistoryResponse{nil, {}, &testdata.GetWorkflowExecutionHistoryResponse} {\n\t\tassert.Equal(t, item, ToGetWorkflowExecutionHistoryResponse(FromGetWorkflowExecutionHistoryResponse(item)))\n\t}\n}\nfunc TestHeader(t *testing.T) {\n\tfor _, item := range []*types.Header{nil, {}, &testdata.Header} {\n\t\tassert.Equal(t, item, ToHeader(FromHeader(item)))\n\t}\n}\nfunc TestHealthResponse(t *testing.T) {\n\tfor _, item := range []*types.HealthStatus{nil, {}, &testdata.HealthStatus} {\n\t\tassert.Equal(t, item, ToHealthResponse(FromHealthResponse(item)))\n\t}\n}\nfunc TestHistory(t *testing.T) {\n\tfor _, item := range []*types.History{nil, &testdata.History} {\n\t\tassert.Equal(t, item, ToHistory(FromHistory(item)))\n\t}\n}\nfunc TestListArchivedWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListArchivedWorkflowExecutionsRequest{nil, {}, &testdata.ListArchivedWorkflowExecutionsRequest} {\n\t\tassert.Equal(t, item, ToListArchivedWorkflowExecutionsRequest(FromListArchivedWorkflowExecutionsRequest(item)))\n\t}\n}\nfunc TestListArchivedWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListArchivedWorkflowExecutionsResponse{nil, {}, &testdata.ListArchivedWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToListArchivedWorkflowExecutionsResponse(FromListArchivedWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestListClosedWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListClosedWorkflowExecutionsResponse{nil, {}, &testdata.ListClosedWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToListClosedWorkflowExecutionsResponse(FromListClosedWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestListDomainsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListDomainsRequest{nil, {}, &testdata.ListDomainsRequest} {\n\t\tassert.Equal(t, item, ToListDomainsRequest(FromListDomainsRequest(item)))\n\t}\n}\nfunc TestListDomainsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListDomainsResponse{nil, {}, &testdata.ListDomainsResponse} {\n\t\tassert.Equal(t, item, ToListDomainsResponse(FromListDomainsResponse(item)))\n\t}\n}\nfunc TestListOpenWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListOpenWorkflowExecutionsResponse{nil, {}, &testdata.ListOpenWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToListOpenWorkflowExecutionsResponse(FromListOpenWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestListTaskListPartitionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListTaskListPartitionsRequest{nil, {}, &testdata.ListTaskListPartitionsRequest} {\n\t\tassert.Equal(t, item, ToListTaskListPartitionsRequest(FromListTaskListPartitionsRequest(item)))\n\t}\n}\nfunc TestListTaskListPartitionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListTaskListPartitionsResponse{nil, {}, &testdata.ListTaskListPartitionsResponse} {\n\t\tassert.Equal(t, item, ToListTaskListPartitionsResponse(FromListTaskListPartitionsResponse(item)))\n\t}\n}\nfunc TestListWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListWorkflowExecutionsRequest{nil, {}, &testdata.ListWorkflowExecutionsRequest} {\n\t\tassert.Equal(t, item, ToListWorkflowExecutionsRequest(FromListWorkflowExecutionsRequest(item)))\n\t}\n}\nfunc TestListWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListWorkflowExecutionsResponse{nil, {}, &testdata.ListWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToListWorkflowExecutionsResponse(FromListWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestMarkerRecordedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.MarkerRecordedEventAttributes{nil, {}, &testdata.MarkerRecordedEventAttributes} {\n\t\tassert.Equal(t, item, ToMarkerRecordedEventAttributes(FromMarkerRecordedEventAttributes(item)))\n\t}\n}\nfunc TestMemo(t *testing.T) {\n\tfor _, item := range []*types.Memo{nil, {}, &testdata.Memo} {\n\t\tassert.Equal(t, item, ToMemo(FromMemo(item)))\n\t}\n}\nfunc TestPendingActivityInfo(t *testing.T) {\n\tfor _, item := range []*types.PendingActivityInfo{nil, {}, &testdata.PendingActivityInfo} {\n\t\tassert.Equal(t, item, ToPendingActivityInfo(FromPendingActivityInfo(item)))\n\t}\n}\nfunc TestPendingChildExecutionInfo(t *testing.T) {\n\tfor _, item := range []*types.PendingChildExecutionInfo{nil, {}, &testdata.PendingChildExecutionInfo} {\n\t\tassert.Equal(t, item, ToPendingChildExecutionInfo(FromPendingChildExecutionInfo(item)))\n\t}\n}\nfunc TestPendingDecisionInfo(t *testing.T) {\n\tfor _, item := range []*types.PendingDecisionInfo{nil, {}, &testdata.PendingDecisionInfo} {\n\t\tassert.Equal(t, item, ToPendingDecisionInfo(FromPendingDecisionInfo(item)))\n\t}\n}\nfunc TestPollForActivityTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.PollForActivityTaskRequest{nil, {}, &testdata.PollForActivityTaskRequest} {\n\t\tassert.Equal(t, item, ToPollForActivityTaskRequest(FromPollForActivityTaskRequest(item)))\n\t}\n}\nfunc TestPollForActivityTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.PollForActivityTaskResponse{nil, {}, &testdata.PollForActivityTaskResponse} {\n\t\tassert.Equal(t, item, ToPollForActivityTaskResponse(FromPollForActivityTaskResponse(item)))\n\t}\n}\nfunc TestPollForDecisionTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.PollForDecisionTaskRequest{nil, {}, &testdata.PollForDecisionTaskRequest} {\n\t\tassert.Equal(t, item, ToPollForDecisionTaskRequest(FromPollForDecisionTaskRequest(item)))\n\t}\n}\nfunc TestPollForDecisionTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.PollForDecisionTaskResponse{nil, {}, &testdata.PollForDecisionTaskResponse} {\n\t\tassert.Equal(t, item, ToPollForDecisionTaskResponse(FromPollForDecisionTaskResponse(item)))\n\t}\n}\nfunc TestPollerInfo(t *testing.T) {\n\tfor _, item := range []*types.PollerInfo{nil, {}, &testdata.PollerInfo} {\n\t\tassert.Equal(t, item, ToPollerInfo(FromPollerInfo(item)))\n\t}\n}\nfunc TestQueryRejected(t *testing.T) {\n\tfor _, item := range []*types.QueryRejected{nil, {}, &testdata.QueryRejected} {\n\t\tassert.Equal(t, item, ToQueryRejected(FromQueryRejected(item)))\n\t}\n}\nfunc TestQueryWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.QueryWorkflowRequest{nil, {}, &testdata.QueryWorkflowRequest} {\n\t\tassert.Equal(t, item, ToQueryWorkflowRequest(FromQueryWorkflowRequest(item)))\n\t}\n}\nfunc TestQueryWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.QueryWorkflowResponse{nil, {}, &testdata.QueryWorkflowResponse} {\n\t\tassert.Equal(t, item, ToQueryWorkflowResponse(FromQueryWorkflowResponse(item)))\n\t}\n}\nfunc TestRecordActivityTaskHeartbeatByIDRequest(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskHeartbeatByIDRequest{nil, {}, &testdata.RecordActivityTaskHeartbeatByIDRequest} {\n\t\tassert.Equal(t, item, ToRecordActivityTaskHeartbeatByIDRequest(FromRecordActivityTaskHeartbeatByIDRequest(item)))\n\t}\n}\nfunc TestRecordActivityTaskHeartbeatByIDResponse(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskHeartbeatResponse{nil, {}, &testdata.RecordActivityTaskHeartbeatResponse} {\n\t\tassert.Equal(t, item, ToRecordActivityTaskHeartbeatByIDResponse(FromRecordActivityTaskHeartbeatByIDResponse(item)))\n\t}\n}\nfunc TestRecordActivityTaskHeartbeatRequest(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskHeartbeatRequest{nil, {}, &testdata.RecordActivityTaskHeartbeatRequest} {\n\t\tassert.Equal(t, item, ToRecordActivityTaskHeartbeatRequest(FromRecordActivityTaskHeartbeatRequest(item)))\n\t}\n}\nfunc TestRecordActivityTaskHeartbeatResponse(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskHeartbeatResponse{nil, {}, &testdata.RecordActivityTaskHeartbeatResponse} {\n\t\tassert.Equal(t, item, ToRecordActivityTaskHeartbeatResponse(FromRecordActivityTaskHeartbeatResponse(item)))\n\t}\n}\nfunc TestRecordMarkerDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.RecordMarkerDecisionAttributes{nil, {}, &testdata.RecordMarkerDecisionAttributes} {\n\t\tassert.Equal(t, item, ToRecordMarkerDecisionAttributes(FromRecordMarkerDecisionAttributes(item)))\n\t}\n}\n\nfunc TestRegisterDomainRequestFuzz(t *testing.T) {\n\tt.Run(\"round trip from internal\", func(t *testing.T) {\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"nil\", \"empty\", \"filled\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\t// Configure fuzzer to generate valid enum values and reasonable day ranges\n\t\t\tfuzzer := f.Funcs(\n\t\t\t\tfunc(e *types.ArchivalStatus, c fuzz.Continue) {\n\t\t\t\t\t*e = types.ArchivalStatus(c.Intn(2)) // 0-1 are valid values (Disabled=0, Enabled=1)\n\t\t\t\t},\n\t\t\t\tfunc(days *int32, c fuzz.Continue) {\n\t\t\t\t\t// Generate reasonable retention period values to avoid precision loss in conversion\n\t\t\t\t\t*days = int32(c.Intn(10000)) // 0-9999 days is reasonable range\n\t\t\t\t},\n\t\t\t).NilChance(0.3)\n\n\t\t\tvar orig *types.RegisterDomainRequest\n\t\t\tfuzzer.Fuzz(&orig)\n\t\t\tout := ToRegisterDomainRequest(FromRegisterDomainRequest(orig))\n\n\t\t\t// Proto RegisterDomainRequest doesn't support EmitMetric field, it's always fixed on\n\t\t\tif orig != nil {\n\t\t\t\texpected := *orig                          // Copy the struct\n\t\t\t\texpected.EmitMetric = common.BoolPtr(true) // this is a legacy field which is always true. It's probably safe to remove\n\t\t\t\tassert.Equal(t, &expected, out, \"RegisterDomainRequest did not survive round-tripping\")\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, orig, out, \"RegisterDomainRequest did not survive round-tripping\")\n\t\t\t}\n\n\t\t\tif orig == nil {\n\t\t\t\treturn \"nil\"\n\t\t\t}\n\t\t\tif orig.Name == \"\" && orig.ActiveClusterName == \"\" && orig.ActiveClusters == nil {\n\t\t\t\treturn \"empty\"\n\t\t\t}\n\t\t\treturn \"filled\"\n\t\t})\n\t})\n}\nfunc TestRequestCancelActivityTaskDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelActivityTaskDecisionAttributes{nil, {}, &testdata.RequestCancelActivityTaskDecisionAttributes} {\n\t\tassert.Equal(t, item, ToRequestCancelActivityTaskDecisionAttributes(FromRequestCancelActivityTaskDecisionAttributes(item)))\n\t}\n}\nfunc TestRequestCancelActivityTaskFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelActivityTaskFailedEventAttributes{nil, {}, &testdata.RequestCancelActivityTaskFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToRequestCancelActivityTaskFailedEventAttributes(FromRequestCancelActivityTaskFailedEventAttributes(item)))\n\t}\n}\nfunc TestRequestCancelExternalWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelExternalWorkflowExecutionDecisionAttributes{nil, {}, &testdata.RequestCancelExternalWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToRequestCancelExternalWorkflowExecutionDecisionAttributes(FromRequestCancelExternalWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestRequestCancelExternalWorkflowExecutionFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{nil, {}, &testdata.RequestCancelExternalWorkflowExecutionFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(item)))\n\t}\n}\nfunc TestRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{nil, {}, &testdata.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes} {\n\t\tassert.Equal(t, item, ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(item)))\n\t}\n}\nfunc TestRequestCancelWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.RequestCancelWorkflowExecutionRequest{nil, {}, &testdata.RequestCancelWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToRequestCancelWorkflowExecutionRequest(FromRequestCancelWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestResetPointInfo(t *testing.T) {\n\tfor _, item := range []*types.ResetPointInfo{nil, {}, &testdata.ResetPointInfo} {\n\t\tassert.Equal(t, item, ToResetPointInfo(FromResetPointInfo(item)))\n\t}\n}\nfunc TestResetPoints(t *testing.T) {\n\tfor _, item := range []*types.ResetPoints{nil, {}, &testdata.ResetPoints} {\n\t\tassert.Equal(t, item, ToResetPoints(FromResetPoints(item)))\n\t}\n}\nfunc TestResetStickyTaskListRequest(t *testing.T) {\n\tfor _, item := range []*types.ResetStickyTaskListRequest{nil, {}, &testdata.ResetStickyTaskListRequest} {\n\t\tassert.Equal(t, item, ToResetStickyTaskListRequest(FromResetStickyTaskListRequest(item)))\n\t}\n}\nfunc TestResetWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.ResetWorkflowExecutionRequest{nil, {}, &testdata.ResetWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToResetWorkflowExecutionRequest(FromResetWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestResetWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.ResetWorkflowExecutionResponse{nil, {}, &testdata.ResetWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToResetWorkflowExecutionResponse(FromResetWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestRespondActivityTaskCanceledByIDRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskCanceledByIDRequest{nil, {}, &testdata.RespondActivityTaskCanceledByIDRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskCanceledByIDRequest(FromRespondActivityTaskCanceledByIDRequest(item)))\n\t}\n}\nfunc TestRespondActivityTaskCanceledRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskCanceledRequest{nil, {}, &testdata.RespondActivityTaskCanceledRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskCanceledRequest(FromRespondActivityTaskCanceledRequest(item)))\n\t}\n}\nfunc TestRespondActivityTaskCompletedByIDRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskCompletedByIDRequest{nil, {}, &testdata.RespondActivityTaskCompletedByIDRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskCompletedByIDRequest(FromRespondActivityTaskCompletedByIDRequest(item)))\n\t}\n}\nfunc TestRespondActivityTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskCompletedRequest{nil, {}, &testdata.RespondActivityTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskCompletedRequest(FromRespondActivityTaskCompletedRequest(item)))\n\t}\n}\nfunc TestRespondActivityTaskFailedByIDRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskFailedByIDRequest{nil, {}, &testdata.RespondActivityTaskFailedByIDRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskFailedByIDRequest(FromRespondActivityTaskFailedByIDRequest(item)))\n\t}\n}\nfunc TestRespondActivityTaskFailedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondActivityTaskFailedRequest{nil, {}, &testdata.RespondActivityTaskFailedRequest} {\n\t\tassert.Equal(t, item, ToRespondActivityTaskFailedRequest(FromRespondActivityTaskFailedRequest(item)))\n\t}\n}\nfunc TestRespondDecisionTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondDecisionTaskCompletedRequest{nil, {}, &testdata.RespondDecisionTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToRespondDecisionTaskCompletedRequest(FromRespondDecisionTaskCompletedRequest(item)))\n\t}\n}\nfunc TestRespondDecisionTaskCompletedResponse(t *testing.T) {\n\tfor _, item := range []*types.RespondDecisionTaskCompletedResponse{nil, {}, &testdata.RespondDecisionTaskCompletedResponse} {\n\t\tassert.Equal(t, item, ToRespondDecisionTaskCompletedResponse(FromRespondDecisionTaskCompletedResponse(item)))\n\t}\n}\nfunc TestRespondDecisionTaskFailedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondDecisionTaskFailedRequest{nil, {}, &testdata.RespondDecisionTaskFailedRequest} {\n\t\tassert.Equal(t, item, ToRespondDecisionTaskFailedRequest(FromRespondDecisionTaskFailedRequest(item)))\n\t}\n}\nfunc TestRespondQueryTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondQueryTaskCompletedRequest{nil, {}, &testdata.RespondQueryTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToRespondQueryTaskCompletedRequest(FromRespondQueryTaskCompletedRequest(item)))\n\t}\n}\nfunc TestRetryPolicy(t *testing.T) {\n\tfor _, item := range []*types.RetryPolicy{nil, {}, &testdata.RetryPolicy} {\n\t\tassert.Equal(t, item, ToRetryPolicy(FromRetryPolicy(item)))\n\t}\n}\nfunc TestScanWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListWorkflowExecutionsRequest{nil, {}, &testdata.ListWorkflowExecutionsRequest} {\n\t\tassert.Equal(t, item, ToScanWorkflowExecutionsRequest(FromScanWorkflowExecutionsRequest(item)))\n\t}\n}\nfunc TestScanWorkflowExecutionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListWorkflowExecutionsResponse{nil, {}, &testdata.ListWorkflowExecutionsResponse} {\n\t\tassert.Equal(t, item, ToScanWorkflowExecutionsResponse(FromScanWorkflowExecutionsResponse(item)))\n\t}\n}\nfunc TestScheduleActivityTaskDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.ScheduleActivityTaskDecisionAttributes{nil, {}, &testdata.ScheduleActivityTaskDecisionAttributes} {\n\t\tassert.Equal(t, item, ToScheduleActivityTaskDecisionAttributes(FromScheduleActivityTaskDecisionAttributes(item)))\n\t}\n}\nfunc TestSearchAttributes(t *testing.T) {\n\tfor _, item := range []*types.SearchAttributes{nil, {}, &testdata.SearchAttributes} {\n\t\tassert.Equal(t, item, ToSearchAttributes(FromSearchAttributes(item)))\n\t}\n}\nfunc TestSignalExternalWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.SignalExternalWorkflowExecutionDecisionAttributes{nil, {}, &testdata.SignalExternalWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToSignalExternalWorkflowExecutionDecisionAttributes(FromSignalExternalWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestSignalExternalWorkflowExecutionFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.SignalExternalWorkflowExecutionFailedEventAttributes{nil, {}, &testdata.SignalExternalWorkflowExecutionFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToSignalExternalWorkflowExecutionFailedEventAttributes(FromSignalExternalWorkflowExecutionFailedEventAttributes(item)))\n\t}\n}\nfunc TestSignalExternalWorkflowExecutionInitiatedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.SignalExternalWorkflowExecutionInitiatedEventAttributes{nil, {}, &testdata.SignalExternalWorkflowExecutionInitiatedEventAttributes} {\n\t\tassert.Equal(t, item, ToSignalExternalWorkflowExecutionInitiatedEventAttributes(FromSignalExternalWorkflowExecutionInitiatedEventAttributes(item)))\n\t}\n}\nfunc TestSignalWithStartWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.SignalWithStartWorkflowExecutionRequest{nil, {}, &testdata.SignalWithStartWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToSignalWithStartWorkflowExecutionRequest(FromSignalWithStartWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestSignalWithStartWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionResponse{nil, {}, &testdata.StartWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToSignalWithStartWorkflowExecutionResponse(FromSignalWithStartWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestSignalWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.SignalWorkflowExecutionRequest{nil, {}, &testdata.SignalWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToSignalWorkflowExecutionRequest(FromSignalWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestStartChildWorkflowExecutionDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.StartChildWorkflowExecutionDecisionAttributes{nil, {}, &testdata.StartChildWorkflowExecutionDecisionAttributes} {\n\t\tassert.Equal(t, item, ToStartChildWorkflowExecutionDecisionAttributes(FromStartChildWorkflowExecutionDecisionAttributes(item)))\n\t}\n}\nfunc TestStartChildWorkflowExecutionFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.StartChildWorkflowExecutionFailedEventAttributes{nil, {}, &testdata.StartChildWorkflowExecutionFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToStartChildWorkflowExecutionFailedEventAttributes(FromStartChildWorkflowExecutionFailedEventAttributes(item)))\n\t}\n}\nfunc TestStartChildWorkflowExecutionInitiatedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.StartChildWorkflowExecutionInitiatedEventAttributes{nil, {}, &testdata.StartChildWorkflowExecutionInitiatedEventAttributes} {\n\t\tassert.Equal(t, item, ToStartChildWorkflowExecutionInitiatedEventAttributes(FromStartChildWorkflowExecutionInitiatedEventAttributes(item)))\n\t}\n}\nfunc TestStartTimeFilter(t *testing.T) {\n\tfor _, item := range []*types.StartTimeFilter{nil, {}, &testdata.StartTimeFilter} {\n\t\tassert.Equal(t, item, ToStartTimeFilter(FromStartTimeFilter(item)))\n\t}\n}\nfunc TestStartTimerDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.StartTimerDecisionAttributes{nil, {}, &testdata.StartTimerDecisionAttributes} {\n\t\tassert.Equal(t, item, ToStartTimerDecisionAttributes(FromStartTimerDecisionAttributes(item)))\n\t}\n}\nfunc TestStartWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionRequest{nil, {}, &testdata.StartWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToStartWorkflowExecutionRequest(FromStartWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestStartWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionResponse{nil, {}, &testdata.StartWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToStartWorkflowExecutionResponse(FromStartWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestStartWorkflowExecutionAsyncRequest(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionAsyncRequest{nil, {}, &testdata.StartWorkflowExecutionAsyncRequest} {\n\t\tassert.Equal(t, item, ToStartWorkflowExecutionAsyncRequest(FromStartWorkflowExecutionAsyncRequest(item)))\n\t}\n}\nfunc TestStartWorkflowExecutionAsyncResponse(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionAsyncResponse{nil, {}, &testdata.StartWorkflowExecutionAsyncResponse} {\n\t\tassert.Equal(t, item, ToStartWorkflowExecutionAsyncResponse(FromStartWorkflowExecutionAsyncResponse(item)))\n\t}\n}\nfunc TestSignalWithStartWorkflowExecutionAsyncRequest(t *testing.T) {\n\tfor _, item := range []*types.SignalWithStartWorkflowExecutionAsyncRequest{nil, {}, &testdata.SignalWithStartWorkflowExecutionAsyncRequest} {\n\t\tassert.Equal(t, item, ToSignalWithStartWorkflowExecutionAsyncRequest(FromSignalWithStartWorkflowExecutionAsyncRequest(item)))\n\t}\n}\nfunc TestSignalWithStartWorkflowExecutionAsyncResponse(t *testing.T) {\n\tfor _, item := range []*types.SignalWithStartWorkflowExecutionAsyncResponse{nil, {}, &testdata.SignalWithStartWorkflowExecutionAsyncResponse} {\n\t\tassert.Equal(t, item, ToSignalWithStartWorkflowExecutionAsyncResponse(FromSignalWithStartWorkflowExecutionAsyncResponse(item)))\n\t}\n}\nfunc TestStatusFilter(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionCloseStatus{nil, &testdata.WorkflowExecutionCloseStatus} {\n\t\tassert.Equal(t, item, ToStatusFilter(FromStatusFilter(item)))\n\t}\n}\nfunc TestStickyExecutionAttributes(t *testing.T) {\n\tfor _, item := range []*types.StickyExecutionAttributes{nil, {}, &testdata.StickyExecutionAttributes} {\n\t\tassert.Equal(t, item, ToStickyExecutionAttributes(FromStickyExecutionAttributes(item)))\n\t}\n}\nfunc TestSupportedClientVersions(t *testing.T) {\n\tfor _, item := range []*types.SupportedClientVersions{nil, {}, &testdata.SupportedClientVersions} {\n\t\tassert.Equal(t, item, ToSupportedClientVersions(FromSupportedClientVersions(item)))\n\t}\n}\nfunc TestTaskIDBlock(t *testing.T) {\n\tfor _, item := range []*types.TaskIDBlock{nil, {}, &testdata.TaskIDBlock} {\n\t\tassert.Equal(t, item, ToTaskIDBlock(FromTaskIDBlock(item)))\n\t}\n}\nfunc TestTaskList(t *testing.T) {\n\tfor _, item := range []*types.TaskList{nil, {}, &testdata.TaskList} {\n\t\tassert.Equal(t, item, ToTaskList(FromTaskList(item)))\n\t}\n}\nfunc TestTaskListMetadata(t *testing.T) {\n\tfor _, item := range []*types.TaskListMetadata{nil, {}, &testdata.TaskListMetadata} {\n\t\tassert.Equal(t, item, ToTaskListMetadata(FromTaskListMetadata(item)))\n\t}\n}\nfunc TestTaskListPartitionMetadata(t *testing.T) {\n\tfor _, item := range []*types.TaskListPartitionMetadata{nil, {}, &testdata.TaskListPartitionMetadata} {\n\t\tassert.Equal(t, item, ToTaskListPartitionMetadata(FromTaskListPartitionMetadata(item)))\n\t}\n}\nfunc TestTaskListStatus(t *testing.T) {\n\tfor _, item := range []*types.TaskListStatus{nil, {}, &testdata.TaskListStatus} {\n\t\tassert.Equal(t, item, ToTaskListStatus(FromTaskListStatus(item)))\n\t}\n}\nfunc TestTerminateWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.TerminateWorkflowExecutionRequest{nil, {}, &testdata.TerminateWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToTerminateWorkflowExecutionRequest(FromTerminateWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestTimerCanceledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.TimerCanceledEventAttributes{nil, {}, &testdata.TimerCanceledEventAttributes} {\n\t\tassert.Equal(t, item, ToTimerCanceledEventAttributes(FromTimerCanceledEventAttributes(item)))\n\t}\n}\nfunc TestTimerFiredEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.TimerFiredEventAttributes{nil, {}, &testdata.TimerFiredEventAttributes} {\n\t\tassert.Equal(t, item, ToTimerFiredEventAttributes(FromTimerFiredEventAttributes(item)))\n\t}\n}\nfunc TestTimerStartedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.TimerStartedEventAttributes{nil, {}, &testdata.TimerStartedEventAttributes} {\n\t\tassert.Equal(t, item, ToTimerStartedEventAttributes(FromTimerStartedEventAttributes(item)))\n\t}\n}\nfunc TestUpdateDomainRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainRequest{nil, {}, &testdata.UpdateDomainRequest} {\n\t\tassert.Equal(t, item, ToUpdateDomainRequest(FromUpdateDomainRequest(item)))\n\t}\n}\nfunc TestFailoverDomainRequest(t *testing.T) {\n\t// Test round-trip conversion for standard testdata\n\tfor _, item := range []*types.FailoverDomainRequest{nil, {}, &testdata.FailoverDomainRequest, &testdata.FailoverDomainRequest_OnlyActiveClusters} {\n\t\tassert.Equal(t, item, ToFailoverDomainRequest(FromFailoverDomainRequest(item)))\n\t}\n\n\t// Test specific edge cases for proto3 empty string handling\n\tt.Run(\"empty DomainActiveClusterName should map to nil pointer\", func(t *testing.T) {\n\t\tinput := &apiv1.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: \"\",\n\t\t\tActiveClusters:          nil,\n\t\t}\n\t\texpected := &types.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: nil,\n\t\t\tActiveClusters:          nil,\n\t\t}\n\t\tresult := ToFailoverDomainRequest(input)\n\t\tassert.Equal(t, expected, result)\n\t\tassert.Nil(t, result.DomainActiveClusterName,\n\t\t\t\"DomainActiveClusterName should be nil when proto field is empty string, not pointer to empty string\")\n\t})\n\n\tt.Run(\"non-empty DomainActiveClusterName should map to pointer\", func(t *testing.T) {\n\t\tinput := &apiv1.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: \"cluster1\",\n\t\t\tActiveClusters:          nil,\n\t\t}\n\t\texpected := &types.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: common.StringPtr(\"cluster1\"),\n\t\t\tActiveClusters:          nil,\n\t\t}\n\t\tassert.Equal(t, expected, ToFailoverDomainRequest(input))\n\t})\n\n\tt.Run(\"with ActiveClusters and empty DomainActiveClusterName\", func(t *testing.T) {\n\t\tinput := &apiv1.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: \"\",\n\t\t\tActiveClusters: &apiv1.ActiveClusters{\n\t\t\t\tActiveClustersByClusterAttribute: map[string]*apiv1.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]*apiv1.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"london\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\texpected := &types.FailoverDomainRequest{\n\t\t\tDomainName:              \"test-domain\",\n\t\t\tDomainActiveClusterName: nil,\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"location\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"london\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tresult := ToFailoverDomainRequest(input)\n\t\tassert.Equal(t, expected, result)\n\t\tassert.Nil(t, result.DomainActiveClusterName,\n\t\t\t\"DomainActiveClusterName should be nil when proto field is empty string\")\n\t})\n}\nfunc TestUpdateDomainResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainResponse{nil, &testdata.UpdateDomainResponse} {\n\t\tassert.Equal(t, item, ToUpdateDomainResponse(FromUpdateDomainResponse(item)))\n\t}\n}\nfunc TestUpsertWorkflowSearchAttributesDecisionAttributes(t *testing.T) {\n\tfor _, item := range []*types.UpsertWorkflowSearchAttributesDecisionAttributes{nil, {}, &testdata.UpsertWorkflowSearchAttributesDecisionAttributes} {\n\t\tassert.Equal(t, item, ToUpsertWorkflowSearchAttributesDecisionAttributes(FromUpsertWorkflowSearchAttributesDecisionAttributes(item)))\n\t}\n}\nfunc TestUpsertWorkflowSearchAttributesEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.UpsertWorkflowSearchAttributesEventAttributes{nil, {}, &testdata.UpsertWorkflowSearchAttributesEventAttributes} {\n\t\tassert.Equal(t, item, ToUpsertWorkflowSearchAttributesEventAttributes(FromUpsertWorkflowSearchAttributesEventAttributes(item)))\n\t}\n}\nfunc TestWorkerVersionInfo(t *testing.T) {\n\tfor _, item := range []*types.WorkerVersionInfo{nil, {}, &testdata.WorkerVersionInfo} {\n\t\tassert.Equal(t, item, ToWorkerVersionInfo(FromWorkerVersionInfo(item)))\n\t}\n}\nfunc TestWorkflowExecution(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecution{nil, {}, &testdata.WorkflowExecution} {\n\t\tassert.Equal(t, item, ToWorkflowExecution(FromWorkflowExecution(item)))\n\t}\n\tassert.Empty(t, ToWorkflowID(nil))\n\tassert.Empty(t, ToRunID(nil))\n}\nfunc TestExternalExecutionInfo(t *testing.T) {\n\tassert.Nil(t, FromExternalExecutionInfoFields(nil, nil))\n\tassert.Nil(t, ToExternalWorkflowExecution(nil))\n\tassert.Nil(t, ToExternalInitiatedID(nil))\n\n\tinfo := FromExternalExecutionInfoFields(nil, common.Int64Ptr(testdata.EventID1))\n\tassert.Nil(t, ToExternalWorkflowExecution(nil))\n\tassert.Equal(t, testdata.EventID1, *ToExternalInitiatedID(info))\n\n\tinfo = FromExternalExecutionInfoFields(&testdata.WorkflowExecution, nil)\n\tassert.Equal(t, testdata.WorkflowExecution, *ToExternalWorkflowExecution(info))\n\tassert.Equal(t, int64(0), *ToExternalInitiatedID(info))\n\n\tinfo = FromExternalExecutionInfoFields(&testdata.WorkflowExecution, common.Int64Ptr(testdata.EventID1))\n\tassert.Equal(t, testdata.WorkflowExecution, *ToExternalWorkflowExecution(info))\n\tassert.Equal(t, testdata.EventID1, *ToExternalInitiatedID(info))\n}\nfunc TestWorkflowExecutionCancelRequestedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionCancelRequestedEventAttributes{nil, {}, &testdata.WorkflowExecutionCancelRequestedEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionCancelRequestedEventAttributes(FromWorkflowExecutionCancelRequestedEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionCanceledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionCanceledEventAttributes{nil, {}, &testdata.WorkflowExecutionCanceledEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionCanceledEventAttributes(FromWorkflowExecutionCanceledEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionCompletedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionCompletedEventAttributes{nil, {}, &testdata.WorkflowExecutionCompletedEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionCompletedEventAttributes(FromWorkflowExecutionCompletedEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionConfiguration(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionConfiguration{nil, {}, &testdata.WorkflowExecutionConfiguration} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionConfiguration(FromWorkflowExecutionConfiguration(item)))\n\t}\n}\nfunc TestWorkflowExecutionContinuedAsNewEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionContinuedAsNewEventAttributes{nil, {}, &testdata.WorkflowExecutionContinuedAsNewEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionContinuedAsNewEventAttributes(FromWorkflowExecutionContinuedAsNewEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionFailedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionFailedEventAttributes{nil, {}, &testdata.WorkflowExecutionFailedEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionFailedEventAttributes(FromWorkflowExecutionFailedEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionFilter(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionFilter{nil, {}, &testdata.WorkflowExecutionFilter} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionFilter(FromWorkflowExecutionFilter(item)))\n\t}\n}\nfunc TestParentExecutionInfo(t *testing.T) {\n\tfor _, item := range []*types.ParentExecutionInfo{nil, {}, &testdata.ParentExecutionInfo} {\n\t\tassert.Equal(t, item, ToParentExecutionInfo(FromParentExecutionInfo(item)))\n\t}\n}\nfunc TestParentExecutionInfoFields(t *testing.T) {\n\tassert.Nil(t, FromParentExecutionInfoFields(nil, nil, nil, nil))\n\tinfo := FromParentExecutionInfoFields(nil, nil, testdata.ParentExecutionInfo.Execution, nil)\n\tassert.Equal(t, \"\", *ToParentDomainID(info))\n\tassert.Equal(t, \"\", *ToParentDomainName(info))\n\tassert.Equal(t, testdata.ParentExecutionInfo.Execution, ToParentWorkflowExecution(info))\n\tassert.Equal(t, int64(0), *ToParentInitiatedID(info))\n\tinfo = FromParentExecutionInfoFields(\n\t\t&testdata.ParentExecutionInfo.DomainUUID,\n\t\t&testdata.ParentExecutionInfo.Domain,\n\t\ttestdata.ParentExecutionInfo.Execution,\n\t\t&testdata.ParentExecutionInfo.InitiatedID)\n\tassert.Equal(t, testdata.ParentExecutionInfo.DomainUUID, *ToParentDomainID(info))\n\tassert.Equal(t, testdata.ParentExecutionInfo.Domain, *ToParentDomainName(info))\n\tassert.Equal(t, testdata.ParentExecutionInfo.Execution, ToParentWorkflowExecution(info))\n\tassert.Equal(t, testdata.ParentExecutionInfo.InitiatedID, *ToParentInitiatedID(info))\n}\nfunc TestWorkflowExecutionInfo(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionInfo{nil, {}, &testdata.WorkflowExecutionInfo, &testdata.CronWorkflowExecutionInfo, &testdata.WorkflowExecutionInfoEphemeral} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionInfo(FromWorkflowExecutionInfo(item)))\n\t}\n}\n\nfunc TestWorkflowExecutionInfo_MigrateTaskList(t *testing.T) {\n\ttlName := \"foo\"\n\totherName := \"bar\"\n\tcases := []struct {\n\t\tname string\n\t\tin   *apiv1.WorkflowExecutionInfo\n\t\tout  *types.WorkflowExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"nil\",\n\t\t\tin:   &apiv1.WorkflowExecutionInfo{},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"name only\",\n\t\t\tin: &apiv1.WorkflowExecutionInfo{\n\t\t\t\tTaskList: tlName,\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"tl only\",\n\t\t\tin: &apiv1.WorkflowExecutionInfo{\n\t\t\t\tTaskListInfo: &apiv1.TaskList{Name: tlName, Kind: apiv1.TaskListKind_TASK_LIST_KIND_NORMAL},\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"both\",\n\t\t\tin: &apiv1.WorkflowExecutionInfo{\n\t\t\t\tTaskList:     otherName,\n\t\t\t\tTaskListInfo: &apiv1.TaskList{Name: tlName, Kind: apiv1.TaskListKind_TASK_LIST_KIND_NORMAL},\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.out, ToWorkflowExecutionInfo(tc.in))\n\t\t})\n\t}\n}\n\nfunc TestWorkflowExecutionSignaledEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionSignaledEventAttributes{nil, {}, &testdata.WorkflowExecutionSignaledEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionSignaledEventAttributes(FromWorkflowExecutionSignaledEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionStartedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionStartedEventAttributes{nil, {}, &testdata.WorkflowExecutionStartedEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionStartedEventAttributes(FromWorkflowExecutionStartedEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionTerminatedEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionTerminatedEventAttributes{nil, {}, &testdata.WorkflowExecutionTerminatedEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionTerminatedEventAttributes(FromWorkflowExecutionTerminatedEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowExecutionTimedOutEventAttributes(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionTimedOutEventAttributes{nil, {}, &testdata.WorkflowExecutionTimedOutEventAttributes} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionTimedOutEventAttributes(FromWorkflowExecutionTimedOutEventAttributes(item)))\n\t}\n}\nfunc TestWorkflowQuery(t *testing.T) {\n\tfor _, item := range []*types.WorkflowQuery{nil, {}, &testdata.WorkflowQuery} {\n\t\tassert.Equal(t, item, ToWorkflowQuery(FromWorkflowQuery(item)))\n\t}\n}\nfunc TestWorkflowQueryResult(t *testing.T) {\n\tfor _, item := range []*types.WorkflowQueryResult{nil, {}, &testdata.WorkflowQueryResult} {\n\t\tassert.Equal(t, item, ToWorkflowQueryResult(FromWorkflowQueryResult(item)))\n\t}\n}\nfunc TestWorkflowType(t *testing.T) {\n\tfor _, item := range []*types.WorkflowType{nil, {}, &testdata.WorkflowType} {\n\t\tassert.Equal(t, item, ToWorkflowType(FromWorkflowType(item)))\n\t}\n}\nfunc TestWorkflowTypeFilter(t *testing.T) {\n\tfor _, item := range []*types.WorkflowTypeFilter{nil, {}, &testdata.WorkflowTypeFilter} {\n\t\tassert.Equal(t, item, ToWorkflowTypeFilter(FromWorkflowTypeFilter(item)))\n\t}\n}\nfunc TestDataBlobArray(t *testing.T) {\n\tfor _, item := range [][]*types.DataBlob{nil, {}, testdata.DataBlobArray} {\n\t\tassert.Equal(t, item, ToDataBlobArray(FromDataBlobArray(item)))\n\t}\n}\nfunc TestHistoryEventArray(t *testing.T) {\n\tfor _, item := range [][]*types.HistoryEvent{{}, testdata.HistoryEventArray} {\n\t\tassert.Equal(t, item, ToHistoryEventArray(FromHistoryEventArray(item)))\n\t}\n}\nfunc TestTaskListPartitionMetadataArray(t *testing.T) {\n\tfor _, item := range [][]*types.TaskListPartitionMetadata{nil, {}, testdata.TaskListPartitionMetadataArray} {\n\t\tassert.Equal(t, item, ToTaskListPartitionMetadataArray(FromTaskListPartitionMetadataArray(item)))\n\t}\n}\nfunc TestDecisionArray(t *testing.T) {\n\tfor _, item := range [][]*types.Decision{nil, {}, testdata.DecisionArray} {\n\t\tassert.Equal(t, item, ToDecisionArray(FromDecisionArray(item)))\n\t}\n}\nfunc TestPollerInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.PollerInfo{nil, {}, testdata.PollerInfoArray} {\n\t\tassert.Equal(t, item, ToPollerInfoArray(FromPollerInfoArray(item)))\n\t}\n}\nfunc TestPendingChildExecutionInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.PendingChildExecutionInfo{nil, {}, testdata.PendingChildExecutionInfoArray} {\n\t\tassert.Equal(t, item, ToPendingChildExecutionInfoArray(FromPendingChildExecutionInfoArray(item)))\n\t}\n}\nfunc TestWorkflowExecutionInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.WorkflowExecutionInfo{nil, {}, testdata.WorkflowExecutionInfoArray} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionInfoArray(FromWorkflowExecutionInfoArray(item)))\n\t}\n}\nfunc TestDescribeDomainResponseArray(t *testing.T) {\n\tfor _, item := range [][]*types.DescribeDomainResponse{nil, {}, testdata.DescribeDomainResponseArray} {\n\t\tassert.Equal(t, item, ToDescribeDomainResponseArray(FromDescribeDomainResponseArray(item)))\n\t}\n}\nfunc TestResetPointInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.ResetPointInfo{nil, {}, testdata.ResetPointInfoArray} {\n\t\tassert.Equal(t, item, ToResetPointInfoArray(FromResetPointInfoArray(item)))\n\t}\n}\nfunc TestPendingActivityInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.PendingActivityInfo{nil, {}, testdata.PendingActivityInfoArray} {\n\t\tassert.Equal(t, item, ToPendingActivityInfoArray(FromPendingActivityInfoArray(item)))\n\t}\n}\nfunc TestClusterReplicationConfigurationArray(t *testing.T) {\n\tfor _, item := range [][]*types.ClusterReplicationConfiguration{nil, {}, testdata.ClusterReplicationConfigurationArray} {\n\t\tassert.Equal(t, item, ToClusterReplicationConfigurationArray(FromClusterReplicationConfigurationArray(item)))\n\t}\n}\nfunc TestActivityLocalDispatchInfoMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.ActivityLocalDispatchInfo{nil, {}, testdata.ActivityLocalDispatchInfoMap} {\n\t\tassert.Equal(t, item, ToActivityLocalDispatchInfoMap(FromActivityLocalDispatchInfoMap(item)))\n\t}\n}\nfunc TestBadBinaryInfoMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.BadBinaryInfo{nil, {}, testdata.BadBinaryInfoMap} {\n\t\tassert.Equal(t, item, ToBadBinaryInfoMap(FromBadBinaryInfoMap(item)))\n\t}\n}\nfunc TestIndexedValueTypeMap(t *testing.T) {\n\tfor _, item := range []map[string]types.IndexedValueType{nil, {}, testdata.IndexedValueTypeMap} {\n\t\tassert.Equal(t, item, ToIndexedValueTypeMap(FromIndexedValueTypeMap(item)))\n\t}\n}\nfunc TestWorkflowQueryMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.WorkflowQuery{nil, {}, testdata.WorkflowQueryMap} {\n\t\tassert.Equal(t, item, ToWorkflowQueryMap(FromWorkflowQueryMap(item)))\n\t}\n}\nfunc TestWorkflowQueryResultMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.WorkflowQueryResult{nil, {}, testdata.WorkflowQueryResultMap} {\n\t\tassert.Equal(t, item, ToWorkflowQueryResultMap(FromWorkflowQueryResultMap(item)))\n\t}\n}\nfunc TestPayload(t *testing.T) {\n\tfor _, item := range [][]byte{nil, {}, testdata.Payload1} {\n\t\tassert.Equal(t, item, ToPayload(FromPayload(item)))\n\t}\n\n\tassert.Equal(t, []byte{}, ToPayload(&apiv1.Payload{\n\t\tData: nil,\n\t}))\n}\nfunc TestPayloadMap(t *testing.T) {\n\tfor _, item := range []map[string][]byte{nil, {}, testdata.PayloadMap} {\n\t\tassert.Equal(t, item, ToPayloadMap(FromPayloadMap(item)))\n\t}\n}\nfunc TestFailure(t *testing.T) {\n\tassert.Nil(t, FromFailure(nil, nil))\n\tassert.Nil(t, ToFailureReason(nil))\n\tassert.Nil(t, ToFailureDetails(nil))\n\tfailure := FromFailure(&testdata.FailureReason, testdata.FailureDetails)\n\tassert.Equal(t, testdata.FailureReason, *ToFailureReason(failure))\n\tassert.Equal(t, testdata.FailureDetails, ToFailureDetails(failure))\n}\nfunc TestHistoryEvent(t *testing.T) {\n\tfor _, item := range []*types.HistoryEvent{\n\t\tnil,\n\t\t&testdata.HistoryEvent_WorkflowExecutionStarted,\n\t\t&testdata.HistoryEvent_WorkflowExecutionCompleted,\n\t\t&testdata.HistoryEvent_WorkflowExecutionFailed,\n\t\t&testdata.HistoryEvent_WorkflowExecutionTimedOut,\n\t\t&testdata.HistoryEvent_DecisionTaskScheduled,\n\t\t&testdata.HistoryEvent_DecisionTaskStarted,\n\t\t&testdata.HistoryEvent_DecisionTaskCompleted,\n\t\t&testdata.HistoryEvent_DecisionTaskTimedOut,\n\t\t&testdata.HistoryEvent_DecisionTaskFailed,\n\t\t&testdata.HistoryEvent_ActivityTaskScheduled,\n\t\t&testdata.HistoryEvent_ActivityTaskStarted,\n\t\t&testdata.HistoryEvent_ActivityTaskCompleted,\n\t\t&testdata.HistoryEvent_ActivityTaskFailed,\n\t\t&testdata.HistoryEvent_ActivityTaskTimedOut,\n\t\t&testdata.HistoryEvent_ActivityTaskCancelRequested,\n\t\t&testdata.HistoryEvent_RequestCancelActivityTaskFailed,\n\t\t&testdata.HistoryEvent_ActivityTaskCanceled,\n\t\t&testdata.HistoryEvent_TimerStarted,\n\t\t&testdata.HistoryEvent_TimerFired,\n\t\t&testdata.HistoryEvent_CancelTimerFailed,\n\t\t&testdata.HistoryEvent_TimerCanceled,\n\t\t&testdata.HistoryEvent_WorkflowExecutionCancelRequested,\n\t\t&testdata.HistoryEvent_WorkflowExecutionCanceled,\n\t\t&testdata.HistoryEvent_RequestCancelExternalWorkflowExecutionInitiated,\n\t\t&testdata.HistoryEvent_RequestCancelExternalWorkflowExecutionFailed,\n\t\t&testdata.HistoryEvent_ExternalWorkflowExecutionCancelRequested,\n\t\t&testdata.HistoryEvent_MarkerRecorded,\n\t\t&testdata.HistoryEvent_WorkflowExecutionSignaled,\n\t\t&testdata.HistoryEvent_WorkflowExecutionTerminated,\n\t\t&testdata.HistoryEvent_WorkflowExecutionContinuedAsNew,\n\t\t&testdata.HistoryEvent_StartChildWorkflowExecutionInitiated,\n\t\t&testdata.HistoryEvent_StartChildWorkflowExecutionFailed,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionStarted,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionCompleted,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionFailed,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionCanceled,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionTimedOut,\n\t\t&testdata.HistoryEvent_ChildWorkflowExecutionTerminated,\n\t\t&testdata.HistoryEvent_SignalExternalWorkflowExecutionInitiated,\n\t\t&testdata.HistoryEvent_SignalExternalWorkflowExecutionFailed,\n\t\t&testdata.HistoryEvent_ExternalWorkflowExecutionSignaled,\n\t\t&testdata.HistoryEvent_UpsertWorkflowSearchAttributes,\n\t} {\n\t\tassert.Equal(t, item, ToHistoryEvent(FromHistoryEvent(item)))\n\t}\n\tassert.Panics(t, func() { FromHistoryEvent(&types.HistoryEvent{}) })\n}\nfunc TestDecision(t *testing.T) {\n\tfor _, item := range []*types.Decision{\n\t\tnil,\n\t\t&testdata.Decision_CancelTimer,\n\t\t&testdata.Decision_CancelWorkflowExecution,\n\t\t&testdata.Decision_CompleteWorkflowExecution,\n\t\t&testdata.Decision_ContinueAsNewWorkflowExecution,\n\t\t&testdata.Decision_FailWorkflowExecution,\n\t\t&testdata.Decision_RecordMarker,\n\t\t&testdata.Decision_RequestCancelActivityTask,\n\t\t&testdata.Decision_RequestCancelExternalWorkflowExecution,\n\t\t&testdata.Decision_ScheduleActivityTask,\n\t\t&testdata.Decision_SignalExternalWorkflowExecution,\n\t\t&testdata.Decision_StartChildWorkflowExecution,\n\t\t&testdata.Decision_StartTimer,\n\t\t&testdata.Decision_UpsertWorkflowSearchAttributes,\n\t} {\n\t\tassert.Equal(t, item, ToDecision(FromDecision(item)))\n\t}\n\tassert.Panics(t, func() { FromDecision(&types.Decision{}) })\n}\nfunc TestListClosedWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListClosedWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListClosedWorkflowExecutionsRequest_ExecutionFilter,\n\t\t&testdata.ListClosedWorkflowExecutionsRequest_StatusFilter,\n\t\t&testdata.ListClosedWorkflowExecutionsRequest_TypeFilter,\n\t} {\n\t\tassert.Equal(t, item, ToListClosedWorkflowExecutionsRequest(FromListClosedWorkflowExecutionsRequest(item)))\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsRequest(t *testing.T) {\n\tfor _, item := range []*types.ListOpenWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListOpenWorkflowExecutionsRequest_ExecutionFilter,\n\t\t&testdata.ListOpenWorkflowExecutionsRequest_TypeFilter,\n\t} {\n\t\tassert.Equal(t, item, ToListOpenWorkflowExecutionsRequest(FromListOpenWorkflowExecutionsRequest(item)))\n\t}\n}\n\nfunc TestGetTaskListsByDomainResponse(t *testing.T) {\n\tfor _, item := range []*types.GetTaskListsByDomainResponse{nil, {}, &testdata.GetTaskListsByDomainResponse} {\n\t\tassert.Equal(t, item, ToMatchingGetTaskListsByDomainResponse(FromMatchingGetTaskListsByDomainResponse(item)))\n\t}\n}\n\nfunc TestFailoverInfo(t *testing.T) {\n\tfor _, item := range []*types.FailoverInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.FailoverInfo,\n\t} {\n\t\tassert.Equal(t, item, ToFailoverInfo(FromFailoverInfo(item)))\n\t}\n}\n\nfunc TestDescribeTaskListResponseMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.DescribeTaskListResponse{nil, {}, testdata.DescribeTaskListResponseMap} {\n\t\tassert.Equal(t, item, ToDescribeTaskListResponseMap(FromDescribeTaskListResponseMap(item)))\n\t}\n}\n\nfunc TestAPITaskListPartitionConfig(t *testing.T) {\n\tfor _, item := range []*types.TaskListPartitionConfig{nil, {}, &testdata.TaskListPartitionConfig} {\n\t\tassert.Equal(t, item, ToAPITaskListPartitionConfig(FromAPITaskListPartitionConfig(item)))\n\t}\n}\n\nfunc TestToAPITaskListPartitionConfig(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tconfig   *apiv1.TaskListPartitionConfig\n\t\texpected *types.TaskListPartitionConfig\n\t}{\n\t\t{\n\t\t\tname: \"happy path\",\n\t\t\tconfig: &apiv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t\tReadPartitions: map[int32]*apiv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int32]*apiv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"numbers only\",\n\t\t\tconfig: &apiv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"number mismatch\",\n\t\t\tconfig: &apiv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 1,\n\t\t\t\tReadPartitions: map[int32]*apiv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int32]*apiv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tactual := ToAPITaskListPartitionConfig(tc.config)\n\t\t\tassert.Equal(t, tc.expected, actual)\n\t\t})\n\t}\n}\n\nfunc TestActiveClustersConversion(t *testing.T) {\n\ttestCases := []*types.ActiveClusters{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"datacenter\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tprotoObj := FromActiveClusters(original)\n\t\troundTripObj := ToActiveClusters(protoObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestClusterAttribute(t *testing.T) {\n\tfor _, item := range []*types.ClusterAttribute{nil, {}, &testdata.ClusterAttribute} {\n\t\tassert.Equal(t, item, ToClusterAttribute(FromClusterAttribute(item)))\n\t}\n}\n\nfunc TestActiveClusterSelectionPolicy(t *testing.T) {\n\tfor _, item := range []*types.ActiveClusterSelectionPolicy{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActiveClusterSelectionPolicyWithClusterAttribute,\n\t\t&testdata.ActiveClusterSelectionPolicyRegionSticky,\n\t\t&testdata.ActiveClusterSelectionPolicyExternalEntity,\n\t} {\n\t\tassert.Equal(t, item, ToActiveClusterSelectionPolicy(FromActiveClusterSelectionPolicy(item)))\n\t}\n}\n\nfunc TestClusterAttributeScopeConversion(t *testing.T) {\n\ttestCases := []*types.ClusterAttributeScope{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\"us-west-1\": {\n\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t},\n\t\t\t\t\"us-east-1\": {\n\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tprotoObj := FromClusterAttributeScope(original)\n\t\troundTripObj := ToClusterAttributeScope(protoObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPaginationOptions(t *testing.T) {\n\tfor _, item := range []*types.PaginationOptions{nil, {}, &testdata.PaginationOptions} {\n\t\tassert.Equal(t, item, ToPaginationOptions(FromPaginationOptions(item)))\n\t}\n}\n\nfunc TestListFailoverHistoryRequestFilters(t *testing.T) {\n\tfor _, item := range []*types.ListFailoverHistoryRequestFilters{nil, {}, &testdata.ListFailoverHistoryRequestFilters} {\n\t\tassert.Equal(t, item, ToListFailoverHistoryRequestFilters(FromListFailoverHistoryRequestFilters(item)))\n\t}\n}\n\nfunc TestListFailoverHistoryRequest(t *testing.T) {\n\tfor _, item := range []*types.ListFailoverHistoryRequest{nil, {}, &testdata.ListFailoverHistoryRequest} {\n\t\tassert.Equal(t, item, ToListFailoverHistoryRequest(FromListFailoverHistoryRequest(item)))\n\t}\n}\n\nfunc TestListFailoverHistoryResponse(t *testing.T) {\n\tfor _, item := range []*types.ListFailoverHistoryResponse{nil, {}, &testdata.ListFailoverHistoryResponse} {\n\t\tassert.Equal(t, item, ToListFailoverHistoryResponse(FromListFailoverHistoryResponse(item)))\n\t}\n}\n\nfunc TestFailoverEvent(t *testing.T) {\n\tfor _, item := range []*types.FailoverEvent{nil, {}, &testdata.FailoverEvent} {\n\t\tassert.Equal(t, item, ToFailoverEvent(FromFailoverEvent(item)))\n\t}\n}\n\nfunc TestFailoverEventArray(t *testing.T) {\n\ttestCases := [][]*types.FailoverEvent{\n\t\tnil,\n\t\t{},\n\t\t{nil},\n\t\t{&testdata.FailoverEvent},\n\t\t{&testdata.FailoverEvent, nil, &testdata.FailoverEvent},\n\t}\n\tfor _, item := range testCases {\n\t\tassert.Equal(t, item, ToFailoverEventArray(FromFailoverEventArray(item)))\n\t}\n}\n\nfunc TestListFailoverHistoryResponseMapping(t *testing.T) {\n\tfuzzer := testdatagen.New(t,\n\t\tfunc(v *types.FailoverEvent, c fuzz.Continue) {\n\t\t\tc.Fuzz(v)\n\t\t\t// Don't allow empty strings for ID - use nil or a non-empty string\n\t\t\tif v.ID != nil && *v.ID == \"\" {\n\t\t\t\tv.ID = nil\n\t\t\t}\n\t\t})\n\tfor i := 0; i < 100; i++ {\n\t\tvar response types.ListFailoverHistoryResponse\n\t\tfuzzer.Fuzz(&response)\n\t\tprotoResponse := FromListFailoverHistoryResponse(&response)\n\t\tassert.Equal(t, &response, ToListFailoverHistoryResponse(protoResponse))\n\t}\n}\n\nfunc TestClusterFailover(t *testing.T) {\n\tfor _, item := range []*types.ClusterFailover{nil, {}, &testdata.ClusterFailover} {\n\t\tassert.Equal(t, item, ToClusterFailover(FromClusterFailover(item)))\n\t}\n}\n\nfunc TestClusterFailoverArray(t *testing.T) {\n\ttestCases := [][]*types.ClusterFailover{\n\t\tnil,\n\t\t{},\n\t\t{nil},\n\t\t{&testdata.ClusterFailover},\n\t\t{&testdata.ClusterFailover, nil, &testdata.ClusterFailover},\n\t}\n\tfor _, item := range testCases {\n\t\tassert.Equal(t, item, ToClusterFailoverArray(FromClusterFailoverArray(item)))\n\t}\n}\n\nfunc TestActiveClusterInfo(t *testing.T) {\n\tfor _, item := range []*types.ActiveClusterInfo{nil, {}, &testdata.ActiveClusterInfo1, &testdata.ActiveClusterInfo2} {\n\t\tassert.Equal(t, item, ToActiveClusterInfo(FromActiveClusterInfo(item)))\n\t}\n}\n\nfunc TestFailoverType(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\tinput    *types.FailoverType\n\t\texpected apiv1.FailoverType\n\t}{\n\t\t{\n\t\t\tname:     \"nil\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.FailoverType_FAILOVER_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"force\",\n\t\t\tinput:    types.FailoverTypeForce.Ptr(),\n\t\t\texpected: apiv1.FailoverType_FAILOVER_TYPE_FORCE,\n\t\t},\n\t\t{\n\t\t\tname:     \"graceful\",\n\t\t\tinput:    types.FailoverTypeGraceful.Ptr(),\n\t\t\texpected: apiv1.FailoverType_FAILOVER_TYPE_GRACEFUL,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult := FromFailoverType(tc.input)\n\t\t\tassert.Equal(t, tc.expected, result)\n\n\t\t\t// Test round-trip\n\t\t\tif tc.input != nil {\n\t\t\t\troundTrip := ToFailoverType(result)\n\t\t\t\tassert.Equal(t, tc.input, roundTrip)\n\t\t\t}\n\t\t})\n\t}\n\n\t// Test ToFailoverType for all enum values\n\tassert.Equal(t, types.FailoverTypeForce.Ptr(), ToFailoverType(apiv1.FailoverType_FAILOVER_TYPE_FORCE))\n\tassert.Equal(t, types.FailoverTypeGraceful.Ptr(), ToFailoverType(apiv1.FailoverType_FAILOVER_TYPE_GRACEFUL))\n\tassert.Nil(t, ToFailoverType(apiv1.FailoverType_FAILOVER_TYPE_INVALID))\n\tassert.Nil(t, ToFailoverType(apiv1.FailoverType(999))) // Unknown value\n}\n\nfunc QueryConsistencyLevelFuzzer(e *types.QueryConsistencyLevel, c fuzz.Continue) {\n\t*e = types.QueryConsistencyLevel(c.Intn(2)) // 0-1: Eventual, Strong\n}\n\nfunc SignalExternalWorkflowExecutionFailedCauseFuzzer(e *types.SignalExternalWorkflowExecutionFailedCause, c fuzz.Continue) {\n\t*e = types.SignalExternalWorkflowExecutionFailedCause(c.Intn(2)) // 0-1: UnknownExternalWorkflowExecution, WorkflowAlreadyCompleted\n}\n\nfunc WorkflowIDReusePolicyFuzzer(e *types.WorkflowIDReusePolicy, c fuzz.Continue) {\n\t*e = types.WorkflowIDReusePolicy(c.Intn(4)) // 0-3: AllowDuplicateFailedOnly, AllowDuplicate, RejectDuplicate, TerminateIfRunning\n}\n\nfunc ActiveClusterSelectionStrategyFuzzer(e *types.ActiveClusterSelectionStrategy, c fuzz.Continue) {\n\t*e = types.ActiveClusterSelectionStrategy(c.Intn(2)) // 0-1: RegionSticky, ExternalEntity\n}\n\nfunc CronOverlapPolicyFuzzer(e *types.CronOverlapPolicy, c fuzz.Continue) {\n\t*e = types.CronOverlapPolicy(c.Intn(2)) // 0-1: Skipped, BufferOne\n}\n\nfunc DecisionTypeFuzzer(e *types.DecisionType, c fuzz.Continue) {\n\t*e = types.DecisionType(c.Intn(13))\n}\n\nfunc EventTypeFuzzer(e *types.EventType, c fuzz.Continue) {\n\t*e = types.EventType(c.Intn(42))\n}\n\nfunc TimeoutTypeFuzzer(e *types.TimeoutType, c fuzz.Continue) {\n\t*e = types.TimeoutType(c.Intn(4)) // 0-3: StartToClose, ScheduleToStart, ScheduleToClose, Heartbeat\n}\n\nfunc TaskListKindFuzzer(e *types.TaskListKind, c fuzz.Continue) {\n\t*e = types.TaskListKind(c.Intn(3)) // 0-2: Normal, Sticky, Ephemeral\n}\n\nfunc FailoverTypeFuzzer(e *types.FailoverType, c fuzz.Continue) {\n\t*e = types.FailoverType(c.Intn(2) + 1) // 1-2: Force, Graceful (skip 0=Invalid which maps to nil)\n}\n\nfunc ArchivalStatusFuzzer(e *types.ArchivalStatus, c fuzz.Continue) {\n\t*e = types.ArchivalStatus(c.Intn(2)) // 0-1: Disabled, Enabled\n}\n\nfunc ContinueAsNewInitiatorFuzzer(e *types.ContinueAsNewInitiator, c fuzz.Continue) {\n\t*e = types.ContinueAsNewInitiator(c.Intn(3)) // 0-2: Decider, RetryPolicy, CronSchedule\n}\n\nfunc CancelExternalWorkflowExecutionFailedCauseFuzzer(e *types.CancelExternalWorkflowExecutionFailedCause, c fuzz.Continue) {\n\t*e = types.CancelExternalWorkflowExecutionFailedCause(c.Intn(2)) // 0-1: UnknownExternalWorkflowExecution, WorkflowAlreadyCompleted\n}\n\nfunc ChildWorkflowExecutionFailedCauseFuzzer(e *types.ChildWorkflowExecutionFailedCause, c fuzz.Continue) {\n\t*e = types.ChildWorkflowExecutionFailedCause(c.Intn(2)) // 0-1: WorkflowAlreadyRunning, DeprecatedDomain\n}\n\nfunc WorkflowExecutionCloseStatusFuzzer(e *types.WorkflowExecutionCloseStatus, c fuzz.Continue) {\n\t*e = types.WorkflowExecutionCloseStatus(c.Intn(6)) // 0-5: Completed, Failed, Canceled, Terminated, ContinuedAsNew, TimedOut\n}\n\nfunc ParentClosePolicyFuzzer(e *types.ParentClosePolicy, c fuzz.Continue) {\n\t*e = types.ParentClosePolicy(c.Intn(3)) // 0-2: Abandon, RequestCancel, Terminate\n}\n\nfunc DecisionTaskTimedOutCauseFuzzer(e *types.DecisionTaskTimedOutCause, c fuzz.Continue) {\n\t*e = types.DecisionTaskTimedOutCause(c.Intn(2)) // 0-1: Timeout, Reset\n}\n\nfunc DecisionTaskFailedCauseFuzzer(e *types.DecisionTaskFailedCause, c fuzz.Continue) {\n\t*e = types.DecisionTaskFailedCause(c.Intn(23)) // 0-22: All DecisionTaskFailedCause values\n}\n\nfunc QueryRejectConditionFuzzer(e *types.QueryRejectCondition, c fuzz.Continue) {\n\t*e = types.QueryRejectCondition(c.Intn(2)) // 0-1: NotOpen, NotCompletedCleanly\n}\n\nfunc PendingActivityStateFuzzer(e *types.PendingActivityState, c fuzz.Continue) {\n\t*e = types.PendingActivityState(c.Intn(3)) // 0-2: Scheduled, Started, CancelRequested\n}\n\nfunc HistoryEventFilterTypeFuzzer(e *types.HistoryEventFilterType, c fuzz.Continue) {\n\t*e = types.HistoryEventFilterType(c.Intn(2)) // 0-1: AllEvent, CloseEvent\n}\n\nfunc IndexedValueTypeFuzzer(e *types.IndexedValueType, c fuzz.Continue) {\n\t*e = types.IndexedValueType(c.Intn(6)) // 0-5: String, Keyword, Int, Double, Bool, Datetime\n}\n\nfunc CompletedTypeFuzzer(e *types.QueryTaskCompletedType, c fuzz.Continue) {\n\t*e = types.QueryTaskCompletedType(c.Intn(2)) // 0-1: Completed, Failed\n}\n\nfunc ActiveClusterSelectionPolicyFuzzerClearAttribute(p *types.ActiveClusterSelectionPolicy, c fuzz.Continue) {\n\t// ActiveClusterSelectionPolicy requires string fields to match strategy\n\t// When strategy is nil, all string fields must be empty (mapper uses ClusterAttribute)\n\t// When strategy is set, only the relevant string fields should be set\n\tc.Fuzz(&p.ActiveClusterSelectionStrategy)\n\tif p.ActiveClusterSelectionStrategy == nil {\n\t\tp.StickyRegion = \"\"\n\t\tp.ExternalEntityType = \"\"\n\t\tp.ExternalEntityKey = \"\"\n\t} else {\n\t\tswitch *p.ActiveClusterSelectionStrategy {\n\t\tcase types.ActiveClusterSelectionStrategyRegionSticky:\n\t\t\tc.Fuzz(&p.StickyRegion)\n\t\t\tp.ExternalEntityType = \"\"\n\t\t\tp.ExternalEntityKey = \"\"\n\t\tcase types.ActiveClusterSelectionStrategyExternalEntity:\n\t\t\tc.Fuzz(&p.ExternalEntityType)\n\t\t\tc.Fuzz(&p.ExternalEntityKey)\n\t\t\tp.StickyRegion = \"\"\n\t\t}\n\t}\n\t// ClusterAttribute is always cleared (mapper uses strategy+strings)\n\tp.ClusterAttribute = nil\n}\n\nfunc ActiveClusterSelectionPolicyFuzzerWithAttribute(p *types.ActiveClusterSelectionPolicy, c fuzz.Continue) {\n\t// ActiveClusterSelectionPolicy requires string fields to match strategy\n\t// When strategy is nil, all string fields must be empty (mapper uses ClusterAttribute)\n\t// When strategy is set, only the relevant string fields should be set\n\tc.Fuzz(&p.ActiveClusterSelectionStrategy)\n\tif p.ActiveClusterSelectionStrategy == nil {\n\t\tp.StickyRegion = \"\"\n\t\tp.ExternalEntityType = \"\"\n\t\tp.ExternalEntityKey = \"\"\n\t} else {\n\t\tswitch *p.ActiveClusterSelectionStrategy {\n\t\tcase types.ActiveClusterSelectionStrategyRegionSticky:\n\t\t\tc.Fuzz(&p.StickyRegion)\n\t\t\tp.ExternalEntityType = \"\"\n\t\t\tp.ExternalEntityKey = \"\"\n\t\tcase types.ActiveClusterSelectionStrategyExternalEntity:\n\t\t\tp.StickyRegion = \"\"\n\t\t\tc.Fuzz(&p.ExternalEntityType)\n\t\t\tc.Fuzz(&p.ExternalEntityKey)\n\t\t}\n\t}\n\tc.Fuzz(&p.ClusterAttribute)\n}\n\nfunc ActiveClusterSelectionPolicyFuzzerNoCustom(p *types.ActiveClusterSelectionPolicy, c fuzz.Continue) {\n\t// Fuzz all fields first without custom fuzzers\n\tc.FuzzNoCustom(p)\n\t// Then clear mutually exclusive fields based on strategy\n\tif p.ActiveClusterSelectionStrategy != nil {\n\t\tswitch *p.ActiveClusterSelectionStrategy {\n\t\tcase types.ActiveClusterSelectionStrategyRegionSticky:\n\t\t\tp.ExternalEntityType = \"\"\n\t\t\tp.ExternalEntityKey = \"\"\n\t\tcase types.ActiveClusterSelectionStrategyExternalEntity:\n\t\t\tp.StickyRegion = \"\"\n\t\t}\n\t} else {\n\t\tp.StickyRegion = \"\"\n\t\tp.ExternalEntityType = \"\"\n\t\tp.ExternalEntityKey = \"\"\n\t}\n}\n\nfunc TestDataBlobArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDataBlobArray, ToDataBlobArray,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestPendingActivityInfoArrayFuzz(t *testing.T) {\n\t// LastFailureReason and LastFailureDetails have asymmetric mapping:\n\t// FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\t// Excluding both fields from comparison to handle this asymmetry\n\ttestutils.RunMapperFuzzTest(t, FromPendingActivityInfoArray, ToPendingActivityInfoArray,\n\t\ttestutils.WithExcludedFields(\"LastFailureReason\", \"LastFailureDetails\"),\n\t)\n}\n\nfunc TestDeleteDomainRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDeleteDomainRequest, ToDeleteDomainRequest)\n}\n\nfunc TestDescribeWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDescribeWorkflowExecutionRequest, ToDescribeWorkflowExecutionRequest,\n\t\ttestutils.WithCustomFuncs(QueryConsistencyLevelFuzzer),\n\t)\n}\n\nfunc TestRequestCancelWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelWorkflowExecutionRequest, ToRequestCancelWorkflowExecutionRequest)\n}\n\nfunc TestRespondActivityTaskFailedByIDRequestFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskFailedByIDRequest, ToRespondActivityTaskFailedByIDRequest,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestSignalExternalWorkflowExecutionFailedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalExternalWorkflowExecutionFailedEventAttributes, ToSignalExternalWorkflowExecutionFailedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tSignalExternalWorkflowExecutionFailedCauseFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestClusterReplicationConfigurationArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterReplicationConfigurationArray, ToClusterReplicationConfigurationArray)\n}\n\nfunc TestIndexedValueTypeMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromIndexedValueTypeMap, ToIndexedValueTypeMap,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tIndexedValueTypeFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestPollForActivityTaskRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPollForActivityTaskRequest, ToPollForActivityTaskRequest)\n}\n\nfunc TestResetPointInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetPointInfo, ToResetPointInfo)\n}\n\nfunc TestCancelTimerDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCancelTimerDecisionAttributes, ToCancelTimerDecisionAttributes)\n}\n\nfunc TestListDomainsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListDomainsRequest, ToListDomainsRequest)\n}\n\nfunc TestPollForActivityTaskResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPollForActivityTaskResponse, ToPollForActivityTaskResponse)\n}\n\nfunc TestTaskListMetadataFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskListMetadata, ToTaskListMetadata)\n}\n\nfunc TestListFailoverHistoryResponseFuzz(t *testing.T) {\n\t// FailoverEvent.ID: empty string is normalized to nil during conversion (ToFailoverEvent only sets ID if non-empty)\n\ttestutils.RunMapperFuzzTest(t, FromListFailoverHistoryResponse, ToListFailoverHistoryResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tFailoverTypeFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"ID\"),\n\t)\n}\n\nfunc TestDescribeTaskListResponseFuzz(t *testing.T) {\n\t// TaskListPartitionConfig has map[int] fields that get truncated to map[int32] in proto\n\t// From and To are non-invertable operations, so we rely on testdata to verify the mapping is correct\n\ttestutils.RunMapperFuzzTest(t, FromDescribeTaskListResponse, ToDescribeTaskListResponse,\n\t\ttestutils.WithExcludedFields(\"ReadPartitions\", \"WritePartitions\"),\n\t)\n}\n\nfunc TestStartWorkflowExecutionAsyncRequestFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy: string fields must match strategy\n\ttestutils.RunMapperFuzzTest(t, FromStartWorkflowExecutionAsyncRequest, ToStartWorkflowExecutionAsyncRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerClearAttribute,\n\t\t),\n\t)\n}\n\nfunc TestStartWorkflowExecutionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartWorkflowExecutionResponse, ToStartWorkflowExecutionResponse)\n}\n\nfunc TestListFailoverHistoryRequestFiltersFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListFailoverHistoryRequestFilters, ToListFailoverHistoryRequestFilters)\n}\n\nfunc TestChildWorkflowExecutionTerminatedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionTerminatedEventAttributes, ToChildWorkflowExecutionTerminatedEventAttributes)\n}\n\nfunc TestCountWorkflowExecutionsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCountWorkflowExecutionsRequest, ToCountWorkflowExecutionsRequest)\n}\n\nfunc TestRequestCancelExternalWorkflowExecutionFailedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelExternalWorkflowExecutionFailedEventAttributes, ToRequestCancelExternalWorkflowExecutionFailedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tCancelExternalWorkflowExecutionFailedCauseFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestRequestCancelExternalWorkflowExecutionInitiatedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes, ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes)\n}\n\nfunc TestRespondDecisionTaskCompletedRequestFuzz(t *testing.T) {\n\t// Excluding both complex fields - tested separately in TestDecisionArrayFuzz and TestWorkflowQueryResultMapFuzz\n\ttestutils.RunMapperFuzzTest(t, FromRespondDecisionTaskCompletedRequest, ToRespondDecisionTaskCompletedRequest,\n\t\ttestutils.WithExcludedFields(\"Decisions\", \"QueryResults\"),\n\t)\n}\n\nfunc TestWorkflowQueryResultMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowQueryResultMap, ToWorkflowQueryResultMap)\n}\n\nfunc TestDecisionTaskStartedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecisionTaskStartedEventAttributes, ToDecisionTaskStartedEventAttributes)\n}\n\nfunc TestResetWorkflowExecutionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetWorkflowExecutionResponse, ToResetWorkflowExecutionResponse)\n}\n\nfunc TestRespondActivityTaskCompletedByIDRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskCompletedByIDRequest, ToRespondActivityTaskCompletedByIDRequest)\n}\n\nfunc TestIsolationGroupMetricsFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromIsolationGroupMetrics, ToIsolationGroupMetrics)\n}\n\nfunc TestFailoverDomainResponseFuzz(t *testing.T) {\n\t// [BUG?] EmitMetric is always set to true in the To* mapper, not set in From mapper\n\t// WorkflowExecutionRetentionPeriodInDays loses precision in days→duration→days conversion\n\ttestutils.RunMapperFuzzTest(t, FromFailoverDomainResponse, ToFailoverDomainResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(r *types.FailoverDomainResponse, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// Mapper always creates these nested structs, never nil\n\t\t\t\tif r.DomainInfo == nil {\n\t\t\t\t\tr.DomainInfo = &types.DomainInfo{}\n\t\t\t\t}\n\t\t\t\tif r.Configuration == nil {\n\t\t\t\t\tr.Configuration = &types.DomainConfiguration{}\n\t\t\t\t}\n\t\t\t\tif r.ReplicationConfiguration == nil {\n\t\t\t\t\tr.ReplicationConfiguration = &types.DomainReplicationConfiguration{}\n\t\t\t\t}\n\t\t\t},\n\t\t\ttestutils.EncodingTypeFuzzer,\n\t\t\ttestutils.IsolationGroupStateFuzzer,\n\t\t\ttestutils.DomainStatusFuzzer,\n\t\t\tArchivalStatusFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"EmitMetric\", \"WorkflowExecutionRetentionPeriodInDays\"),\n\t)\n}\n\nfunc TestFailoverEventArrayFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An empty string ID becomes nil, but the return trip translates it back to nil\n\ttestutils.RunMapperFuzzTest(t, FromFailoverEventArray, ToFailoverEventArray,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tFailoverTypeFuzzer,\n\t\t\tfunc(e *types.FailoverEvent, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(e)\n\t\t\t\t// Empty string for ID should be nil\n\t\t\t\tif e.ID != nil && *e.ID == \"\" {\n\t\t\t\t\te.ID = nil\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestWorkflowQueryMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowQueryMap, ToWorkflowQueryMap)\n}\n\nfunc TestActivityTaskCompletedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskCompletedEventAttributes, ToActivityTaskCompletedEventAttributes)\n}\n\nfunc TestHealthResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromHealthResponse, ToHealthResponse)\n}\n\nfunc TestListArchivedWorkflowExecutionsResponseFuzz(t *testing.T) {\n\t// Executions is tested in FromWorkflowExecutionInfoFuzz test\n\ttestutils.RunMapperFuzzTest(t, FromListArchivedWorkflowExecutionsResponse, ToListArchivedWorkflowExecutionsResponse,\n\t\ttestutils.WithExcludedFields(\"Executions\"),\n\t)\n}\n\nfunc TestListOpenWorkflowExecutionsResponseFuzz(t *testing.T) {\n\t// Executions is tested in FromWorkflowExecutionInfoFuzz test\n\ttestutils.RunMapperFuzzTest(t, FromListOpenWorkflowExecutionsResponse, ToListOpenWorkflowExecutionsResponse,\n\t\ttestutils.WithExcludedFields(\"Executions\"),\n\t)\n}\n\nfunc TestRespondDecisionTaskFailedRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondDecisionTaskFailedRequest, ToRespondDecisionTaskFailedRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tDecisionTaskFailedCauseFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestDescribeWorkflowExecutionResponseFuzz(t *testing.T) {\n\tt.Skip(\"All subfields are tested in dedicated fuzz tests\")\n}\n\nfunc TestExternalWorkflowExecutionCancelRequestedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromExternalWorkflowExecutionCancelRequestedEventAttributes, ToExternalWorkflowExecutionCancelRequestedEventAttributes)\n}\n\nfunc TestSignalExternalWorkflowExecutionInitiatedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalExternalWorkflowExecutionInitiatedEventAttributes, ToSignalExternalWorkflowExecutionInitiatedEventAttributes)\n}\n\nfunc TestStartChildWorkflowExecutionInitiatedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartChildWorkflowExecutionInitiatedEventAttributes, ToStartChildWorkflowExecutionInitiatedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerWithAttribute,\n\t\t),\n\t)\n}\n\nfunc TestWorkflowExecutionTerminatedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionTerminatedEventAttributes, ToWorkflowExecutionTerminatedEventAttributes)\n}\n\nfunc TestWorkerVersionInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkerVersionInfo, ToWorkerVersionInfo)\n}\n\nfunc TestListOpenWorkflowExecutionsRequestFuzz(t *testing.T) {\n\t// ExecutionFilter and TypeFilter map to a proto oneof - only one can be set\n\t// If both are set, TypeFilter wins and ExecutionFilter is lost\n\ttestutils.RunMapperFuzzTest(t, FromListOpenWorkflowExecutionsRequest, ToListOpenWorkflowExecutionsRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(r *types.ListOpenWorkflowExecutionsRequest, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// If both filters are set, clear ExecutionFilter (TypeFilter takes precedence in mapper)\n\t\t\t\tif r.ExecutionFilter != nil && r.TypeFilter != nil {\n\t\t\t\t\tr.ExecutionFilter = nil\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestGetTaskListsByDomainResponseFuzz(t *testing.T) {\n\t// SKIPPED: PartitionConfig is nested inside map values (map[string]*DescribeTaskListResponse)\n\t// clearFieldsIf cannot modify fields inside map values due to Go reflection limitations\n\t// TODO(c-warren): Implement map rebuilding in clearFieldsIf to support this pattern\n\tt.Skip(\"Map value field exclusion not yet supported\")\n\ttestutils.RunMapperFuzzTest(t, FromGetTaskListsByDomainResponse, ToGetTaskListsByDomainResponse)\n}\n\nfunc TestRefreshWorkflowTasksRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRefreshWorkflowTasksRequest, ToRefreshWorkflowTasksRequest)\n}\n\nfunc TestRestartWorkflowExecutionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRestartWorkflowExecutionResponse, ToRestartWorkflowExecutionResponse)\n}\n\nfunc TestScanWorkflowExecutionsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScanWorkflowExecutionsRequest, ToScanWorkflowExecutionsRequest)\n}\n\nfunc TestUpsertWorkflowSearchAttributesDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUpsertWorkflowSearchAttributesDecisionAttributes, ToUpsertWorkflowSearchAttributesDecisionAttributes)\n}\n\nfunc TestWorkflowExecutionCompletedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionCompletedEventAttributes, ToWorkflowExecutionCompletedEventAttributes)\n}\n\nfunc TestWorkflowQueryFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowQuery, ToWorkflowQuery)\n}\n\nfunc TestHistoryEventFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromHistoryEvent, ToHistoryEvent,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tEventTypeFuzzer,\n\t\t\tDecisionTypeFuzzer,\n\t\t\tTimeoutTypeFuzzer,\n\t\t\tTaskListKindFuzzer,\n\t\t\tContinueAsNewInitiatorFuzzer,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerWithAttribute,\n\t\t\tfunc(h *types.HistoryEvent, c fuzz.Continue) {\n\t\t\t\t// Fuzz all fields first\n\t\t\t\tc.Fuzz(h)\n\t\t\t\t// Ensure EventType is always non-nil (required by mapper switch statement)\n\t\t\t\tif h.EventType == nil {\n\t\t\t\t\teventType := types.EventType(c.Intn(42)) // 0-41: All EventType values\n\t\t\t\t\th.EventType = &eventType\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestActivityTaskFailedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskFailedEventAttributes, ToActivityTaskFailedEventAttributes,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestHistoryFuzz(t *testing.T) {\n\tt.Skip(\"HistoryEvent array testing covered by individual event attribute fuzz tests\")\n}\n\nfunc TestResetWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetWorkflowExecutionRequest, ToResetWorkflowExecutionRequest)\n}\n\nfunc TestRespondActivityTaskCanceledByIDRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskCanceledByIDRequest, ToRespondActivityTaskCanceledByIDRequest)\n}\n\nfunc TestTimerCanceledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTimerCanceledEventAttributes, ToTimerCanceledEventAttributes)\n}\n\nfunc TestDiagnoseWorkflowExecutionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDiagnoseWorkflowExecutionResponse, ToDiagnoseWorkflowExecutionResponse)\n}\n\nfunc TestListArchivedWorkflowExecutionsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListArchivedWorkflowExecutionsRequest, ToListArchivedWorkflowExecutionsRequest)\n}\n\nfunc TestChildWorkflowExecutionCanceledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionCanceledEventAttributes, ToChildWorkflowExecutionCanceledEventAttributes)\n}\n\nfunc TestRequestCancelExternalWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelExternalWorkflowExecutionDecisionAttributes, ToRequestCancelExternalWorkflowExecutionDecisionAttributes)\n}\n\nfunc TestDecisionFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecision, ToDecision,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(d *types.Decision, c fuzz.Continue) {\n\t\t\t\t// Fuzz all fields first\n\t\t\t\tc.Fuzz(d)\n\t\t\t\t// Ensure DecisionType is always non-nil (required by mapper switch statement)\n\t\t\t\tif d.DecisionType == nil {\n\t\t\t\t\tdecisionType := types.DecisionType(c.Intn(13)) // 0-12: All DecisionType values\n\t\t\t\t\td.DecisionType = &decisionType\n\t\t\t\t}\n\t\t\t},\n\t\t\tDecisionTypeFuzzer,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerWithAttribute,\n\t\t),\n\t)\n}\n\nfunc TestDeprecateDomainRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDeprecateDomainRequest, ToDeprecateDomainRequest)\n}\n\nfunc TestFailoverDomainRequestFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An empty string DomainActiveClusterName becomes nil, but the return trip translates it back to nil\n\t// [Missing] Reason is not yet implemented in the mapper\n\ttestutils.RunMapperFuzzTest(t, FromFailoverDomainRequest, ToFailoverDomainRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tFailoverTypeFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"DomainActiveClusterName\", \"Reason\"),\n\t)\n}\n\nfunc TestChildWorkflowExecutionStartedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionStartedEventAttributes, ToChildWorkflowExecutionStartedEventAttributes)\n}\n\nfunc TestPollForDecisionTaskResponseFuzz(t *testing.T) {\n\t// History.Events: nil vs empty slice (mapper creates empty when nil)\n\t// Contains HistoryEvent array which needs comprehensive enum fuzzers\n\ttestutils.RunMapperFuzzTest(t, FromPollForDecisionTaskResponse, ToPollForDecisionTaskResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(h *types.History, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(h)\n\t\t\t\t// Ensure Events is never nil to avoid nil vs empty slice mismatch\n\t\t\t\tif h.Events == nil {\n\t\t\t\t\th.Events = []*types.HistoryEvent{}\n\t\t\t\t}\n\t\t\t},\n\t\t\tfunc(h *types.HistoryEvent, c fuzz.Continue) {\n\t\t\t\t// Fuzz all fields first\n\t\t\t\tc.Fuzz(h)\n\t\t\t\t// Ensure EventType is always non-nil (required by mapper switch statement)\n\t\t\t\tif h.EventType == nil {\n\t\t\t\t\teventType := types.EventType(c.Intn(42)) // 0-41: All EventType values\n\t\t\t\t\th.EventType = &eventType\n\t\t\t\t}\n\t\t\t},\n\t\t\tEventTypeFuzzer,\n\t\t\tDecisionTypeFuzzer,\n\t\t\tTimeoutTypeFuzzer,\n\t\t\tTaskListKindFuzzer,\n\t\t\tContinueAsNewInitiatorFuzzer,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerClearAttribute,\n\t\t\tSignalExternalWorkflowExecutionFailedCauseFuzzer,\n\t\t\tCancelExternalWorkflowExecutionFailedCauseFuzzer,\n\t\t\tChildWorkflowExecutionFailedCauseFuzzer,\n\t\t\tWorkflowExecutionCloseStatusFuzzer,\n\t\t\tParentClosePolicyFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestDecisionArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecisionArray, ToDecisionArray,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(d *types.Decision, c fuzz.Continue) {\n\t\t\t\t// Fuzz all fields first\n\t\t\t\tc.Fuzz(d)\n\t\t\t\t// Ensure DecisionType is always non-nil (required by mapper switch statement)\n\t\t\t\tif d.DecisionType == nil {\n\t\t\t\t\tdecisionType := types.DecisionType(c.Intn(13)) // 0-12: All DecisionType values\n\t\t\t\t\td.DecisionType = &decisionType\n\t\t\t\t}\n\t\t\t},\n\t\t\tDecisionTypeFuzzer,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerWithAttribute,\n\t\t),\n\t)\n}\n\nfunc TestClusterAttributeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterAttribute, ToClusterAttribute)\n}\n\nfunc TestDecisionTaskTimedOutEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecisionTaskTimedOutEventAttributes, ToDecisionTaskTimedOutEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tDecisionTaskTimedOutCauseFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestIndexedValueTypeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t,\n\t\tfunc(v types.IndexedValueType) apiv1.IndexedValueType {\n\t\t\treturn FromIndexedValueType(v)\n\t\t},\n\t\tfunc(v apiv1.IndexedValueType) types.IndexedValueType {\n\t\t\treturn ToIndexedValueType(v)\n\t\t},\n\t)\n}\n\nfunc TestDescribeTaskListRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDescribeTaskListRequest, ToDescribeTaskListRequest)\n}\n\nfunc TestPollerInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPollerInfo, ToPollerInfo)\n}\n\nfunc TestTerminateWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTerminateWorkflowExecutionRequest, ToTerminateWorkflowExecutionRequest)\n}\n\nfunc TestListFailoverHistoryRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListFailoverHistoryRequest, ToListFailoverHistoryRequest)\n}\n\nfunc TestListClosedWorkflowExecutionsRequestFuzz(t *testing.T) {\n\t// ExecutionFilter, TypeFilter, and StatusFilter map to a proto oneof - only one can be set\n\t// If multiple are set, StatusFilter wins (checked last), then TypeFilter, then ExecutionFilter\n\ttestutils.RunMapperFuzzTest(t, FromListClosedWorkflowExecutionsRequest, ToListClosedWorkflowExecutionsRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(r *types.ListClosedWorkflowExecutionsRequest, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// Clear filters that would be lost in the oneof mapping\n\t\t\t\tif r.StatusFilter != nil {\n\t\t\t\t\t// StatusFilter wins - clear the others\n\t\t\t\t\tr.ExecutionFilter = nil\n\t\t\t\t\tr.TypeFilter = nil\n\t\t\t\t} else if r.TypeFilter != nil {\n\t\t\t\t\t// TypeFilter wins over ExecutionFilter\n\t\t\t\t\tr.ExecutionFilter = nil\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestChildWorkflowExecutionFailedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionFailedEventAttributes, ToChildWorkflowExecutionFailedEventAttributes,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestListClosedWorkflowExecutionsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListClosedWorkflowExecutionsResponse, ToListClosedWorkflowExecutionsResponse,\n\t\ttestutils.WithExcludedFields(\"Executions\"), // Executions is tested in FromWorkflowExecutionInfoFuzz test\n\t)\n}\n\nfunc TestRestartWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRestartWorkflowExecutionRequest, ToRestartWorkflowExecutionRequest)\n}\n\nfunc TestActivityTaskCanceledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskCanceledEventAttributes, ToActivityTaskCanceledEventAttributes)\n}\n\nfunc TestListDomainsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListDomainsResponse, ToListDomainsResponse,\n\t\ttestutils.WithExcludedFields(\"Domains\"), // Domains is tested in TestDescribeDomainResponseDomainFuzz test\n\t)\n}\n\nfunc TestMarkerRecordedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromMarkerRecordedEventAttributes, ToMarkerRecordedEventAttributes)\n}\n\nfunc TestTimerFiredEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTimerFiredEventAttributes, ToTimerFiredEventAttributes)\n}\n\nfunc TestUpdateDomainRequestFuzz(t *testing.T) {\n\t// [BUG?/DEPRECATED] EmitMetric is always set to true in the To* mapper, not set in From mapper\n\t// [Missing] FailoverReason is not yet implemented in the mapper\n\t// [BUG] WorkflowExecutionRetentionPeriodInDays loses precision in days→duration→days conversion\n\t// HistoryArchivalStatus, VisibilityArchivalStatus: only included in field mask if corresponding URI is non-nil\n\ttestutils.RunMapperFuzzTest(t, FromUpdateDomainRequest, ToUpdateDomainRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.EncodingTypeFuzzer,\n\t\t\ttestutils.IsolationGroupStateFuzzer,\n\t\t\ttestutils.DomainStatusFuzzer,\n\t\t\tArchivalStatusFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"EmitMetric\", \"FailoverReason\", \"WorkflowExecutionRetentionPeriodInDays\", \"HistoryArchivalStatus\", \"VisibilityArchivalStatus\"),\n\t)\n}\n\nfunc TestActivityTaskTimedOutEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskTimedOutEventAttributes, ToActivityTaskTimedOutEventAttributes,\n\t\ttestutils.WithExcludedFields(\"LastFailureDetails\"),\n\t)\n}\n\nfunc TestChildWorkflowExecutionTimedOutEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionTimedOutEventAttributes, ToChildWorkflowExecutionTimedOutEventAttributes)\n}\n\nfunc TestDecisionTaskCompletedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecisionTaskCompletedEventAttributes, ToDecisionTaskCompletedEventAttributes)\n}\n\nfunc TestGetWorkflowExecutionHistoryResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetWorkflowExecutionHistoryResponse, ToGetWorkflowExecutionHistoryResponse,\n\t\ttestutils.WithExcludedFields(\"History\"), // History is tested in TestHistoryEventFuzz test\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.EncodingTypeFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestQueryRejectedFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromQueryRejected, ToQueryRejected)\n}\n\nfunc TestResetPointsFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetPoints, ToResetPoints)\n}\n\nfunc TestSignalWithStartWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalWithStartWorkflowExecutionRequest, ToSignalWithStartWorkflowExecutionRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t\tCronOverlapPolicyFuzzer,\n\t\t\tActiveClusterSelectionStrategyFuzzer,\n\t\t\tActiveClusterSelectionPolicyFuzzerWithAttribute,\n\t\t),\n\t)\n}\n\nfunc TestWorkflowExecutionFailedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionFailedEventAttributes, ToWorkflowExecutionFailedEventAttributes,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestPendingDecisionInfoFuzz(t *testing.T) {\n\t// [BUG] Attempt field is truncated from int64 to int32 in proto mapper\n\ttestutils.RunMapperFuzzTest(t, FromPendingDecisionInfo, ToPendingDecisionInfo,\n\t\ttestutils.WithExcludedFields(\"Attempt\"),\n\t)\n}\n\nfunc TestStickyExecutionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStickyExecutionAttributes, ToStickyExecutionAttributes)\n}\n\nfunc TestWorkflowExecutionFilterFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionFilter, ToWorkflowExecutionFilter)\n}\n\nfunc TestActivityTaskStartedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskStartedEventAttributes, ToActivityTaskStartedEventAttributes,\n\t\ttestutils.WithExcludedFields(\"LastFailureDetails\"),\n\t)\n}\n\nfunc TestDecisionTaskScheduledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDecisionTaskScheduledEventAttributes, ToDecisionTaskScheduledEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(v *types.DecisionTaskScheduledEventAttributes, c fuzz.Continue) {\n\t\t\t\tv.TaskList = &types.TaskList{}\n\t\t\t\tc.Fuzz(v.TaskList)\n\t\t\t\tif c.RandBool() {\n\t\t\t\t\ttimeout := c.Int31()\n\t\t\t\t\tv.StartToCloseTimeoutSeconds = &timeout\n\t\t\t\t}\n\t\t\t\t// Attempt is int64 but proto is int32, so limit to int32 range\n\t\t\t\tv.Attempt = int64(c.Int31())\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestQueryWorkflowResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromQueryWorkflowResponse, ToQueryWorkflowResponse)\n}\n\nfunc TestSignalWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalWorkflowExecutionRequest, ToSignalWorkflowExecutionRequest)\n}\n\nfunc TestPendingChildExecutionInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPendingChildExecutionInfo, ToPendingChildExecutionInfo)\n}\n\nfunc TestPollForDecisionTaskRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPollForDecisionTaskRequest, ToPollForDecisionTaskRequest)\n}\n\nfunc TestRecordMarkerDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRecordMarkerDecisionAttributes, ToRecordMarkerDecisionAttributes)\n}\n\nfunc TestWorkflowQueryResultFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowQueryResult, ToWorkflowQueryResult)\n}\n\nfunc TestStartChildWorkflowExecutionFailedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartChildWorkflowExecutionFailedEventAttributes, ToStartChildWorkflowExecutionFailedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(e *types.ChildWorkflowExecutionFailedCause, c fuzz.Continue) {\n\t\t\t\t*e = types.ChildWorkflowExecutionFailedCause(0) // 0: WorkflowAlreadyRunning\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestTaskListStatusFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskListStatus, ToTaskListStatus)\n}\n\nfunc TestTaskListPartitionMetadataArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskListPartitionMetadataArray, ToTaskListPartitionMetadataArray)\n}\n\nfunc TestDecisionTaskFailedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if Reason is non-nil, so Details without Reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromDecisionTaskFailedEventAttributes, ToDecisionTaskFailedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tDecisionTaskFailedCauseFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestDescribeDomainRequestFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An empty string ID becomes nil, but the return trip translates it back to nil\n\ttestutils.RunMapperFuzzTest(t, FromDescribeDomainRequest, ToDescribeDomainRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(r *types.DescribeDomainRequest, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// Must have at least one of Name or UUID set (oneof field)\n\t\t\t\tif r.Name == nil && r.UUID == nil {\n\t\t\t\t\tif c.RandBool() {\n\t\t\t\t\t\tr.Name = new(string)\n\t\t\t\t\t\t*r.Name = c.RandString()\n\t\t\t\t\t} else {\n\t\t\t\t\t\tr.UUID = new(string)\n\t\t\t\t\t\t*r.UUID = c.RandString()\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t\ttestutils.WithExcludedFields(\"ID\"),\n\t)\n}\n\nfunc TestResetStickyTaskListRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetStickyTaskListRequest, ToResetStickyTaskListRequest)\n}\n\nfunc TestActivityTaskCancelRequestedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskCancelRequestedEventAttributes, ToActivityTaskCancelRequestedEventAttributes)\n}\n\nfunc TestBadBinaryInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBadBinaryInfo, ToBadBinaryInfo)\n}\n\nfunc TestCancelTimerFailedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCancelTimerFailedEventAttributes, ToCancelTimerFailedEventAttributes)\n}\n\nfunc TestRespondActivityTaskCompletedRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskCompletedRequest, ToRespondActivityTaskCompletedRequest)\n}\n\nfunc TestPaginationOptionsFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPaginationOptions, ToPaginationOptions)\n}\n\nfunc TestCancelWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCancelWorkflowExecutionDecisionAttributes, ToCancelWorkflowExecutionDecisionAttributes)\n}\n\nfunc TestFailoverInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromFailoverInfo, ToFailoverInfo)\n}\n\nfunc TestRespondActivityTaskFailedRequestFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskFailedRequest, ToRespondActivityTaskFailedRequest,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestScanWorkflowExecutionsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScanWorkflowExecutionsResponse, ToScanWorkflowExecutionsResponse,\n\t\ttestutils.WithExcludedFields(\"Executions\"), // Executions is tested in FromWorkflowExecutionInfoFuzz test\n\t)\n}\n\nfunc TestScheduleActivityTaskDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleActivityTaskDecisionAttributes, ToScheduleActivityTaskDecisionAttributes)\n}\n\nfunc TestTaskListFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskList, ToTaskList)\n}\n\nfunc TestMemoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromMemo, ToMemo)\n}\n\nfunc TestResetPointInfoArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetPointInfoArray, ToResetPointInfoArray)\n}\n\nfunc TestSearchAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSearchAttributes, ToSearchAttributes)\n}\n\nfunc TestTaskListPartitionMetadataFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskListPartitionMetadata, ToTaskListPartitionMetadata)\n}\n\nfunc TestWorkflowExecutionSignaledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionSignaledEventAttributes, ToWorkflowExecutionSignaledEventAttributes)\n}\n\nfunc TestHistoryEventArrayFuzz(t *testing.T) {\n\tt.Skip(\"Tested in TestHistoryEventFuzz test\")\n}\n\nfunc TestActivityLocalDispatchInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityLocalDispatchInfo, ToActivityLocalDispatchInfo)\n}\n\nfunc TestStartWorkflowExecutionRequestFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields\n\ttestutils.RunMapperFuzzTest(t, FromStartWorkflowExecutionRequest, ToStartWorkflowExecutionRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestUpsertWorkflowSearchAttributesEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUpsertWorkflowSearchAttributesEventAttributes, ToUpsertWorkflowSearchAttributesEventAttributes)\n}\n\nfunc TestExternalWorkflowExecutionSignaledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromExternalWorkflowExecutionSignaledEventAttributes, ToExternalWorkflowExecutionSignaledEventAttributes)\n}\n\nfunc TestWorkflowTypeFilterFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowTypeFilter, ToWorkflowTypeFilter)\n}\n\nfunc TestDescribeTaskListResponseMapFuzz(t *testing.T) {\n\t// Map[int] with int64 keys don't roundtrip correctly through proto int32\n\t// Use custom fuzzer to nil out ReadPartitions/WritePartitions in all map values\n\ttestutils.RunMapperFuzzTest(t, FromDescribeTaskListResponseMap, ToDescribeTaskListResponseMap,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(m *map[string]*types.DescribeTaskListResponse, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(m)\n\t\t\t\t// Exclude map[int] fields that don't roundtrip through proto\n\t\t\t\tfor _, resp := range *m {\n\t\t\t\t\tif resp != nil && resp.PartitionConfig != nil {\n\t\t\t\t\t\tresp.PartitionConfig.ReadPartitions = nil\n\t\t\t\t\t\tresp.PartitionConfig.WritePartitions = nil\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestTimerStartedEventAttributesFuzz(t *testing.T) {\n\t// StartToFireTimeoutSeconds is int64 but proto converts through int32, so limit to int32 range\n\ttestutils.RunMapperFuzzTest(t, FromTimerStartedEventAttributes, ToTimerStartedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(v *types.TimerStartedEventAttributes, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(v)\n\t\t\t\tif v.StartToFireTimeoutSeconds != nil {\n\t\t\t\t\t*v.StartToFireTimeoutSeconds = int64(c.Int31())\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestActivityTypeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityType, ToActivityType)\n}\n\nfunc TestWorkflowExecutionFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecution, ToWorkflowExecution)\n}\n\nfunc TestCompleteWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCompleteWorkflowExecutionDecisionAttributes, ToCompleteWorkflowExecutionDecisionAttributes)\n}\n\nfunc TestListTaskListPartitionsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListTaskListPartitionsRequest, ToListTaskListPartitionsRequest)\n}\n\nfunc TestCountWorkflowExecutionsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCountWorkflowExecutionsResponse, ToCountWorkflowExecutionsResponse)\n}\n\nfunc TestDataBlobFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDataBlob, ToDataBlob,\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.EncodingTypeFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestActiveClusterInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActiveClusterInfo, ToActiveClusterInfo)\n}\n\nfunc TestActiveClusterSelectionPolicyFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields based on Strategy\n\ttestutils.RunMapperFuzzTest(t, FromActiveClusterSelectionPolicy, ToActiveClusterSelectionPolicy,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t),\n\t)\n}\n\nfunc TestRespondActivityTaskCanceledRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondActivityTaskCanceledRequest, ToRespondActivityTaskCanceledRequest)\n}\n\nfunc TestPendingChildExecutionInfoArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPendingChildExecutionInfoArray, ToPendingChildExecutionInfoArray)\n}\n\nfunc TestRecordActivityTaskHeartbeatResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRecordActivityTaskHeartbeatResponse, ToRecordActivityTaskHeartbeatResponse)\n}\n\nfunc TestStartTimerDecisionAttributesFuzz(t *testing.T) {\n\t// StartToFireTimeoutSeconds is int64 but proto converts through int32\n\ttestutils.RunMapperFuzzTest(t, FromStartTimerDecisionAttributes, ToStartTimerDecisionAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(v *types.StartTimerDecisionAttributes, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(v)\n\t\t\t\tif v.StartToFireTimeoutSeconds != nil {\n\t\t\t\t\tval := int64(c.Int31())\n\t\t\t\t\tv.StartToFireTimeoutSeconds = &val\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestSupportedClientVersionsFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSupportedClientVersions, ToSupportedClientVersions)\n}\n\nfunc TestClusterFailoverArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterFailoverArray, ToClusterFailoverArray)\n}\n\nfunc TestActiveClustersFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActiveClusters, ToActiveClusters)\n}\n\nfunc TestChildWorkflowExecutionCompletedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromChildWorkflowExecutionCompletedEventAttributes, ToChildWorkflowExecutionCompletedEventAttributes)\n}\n\nfunc TestClusterReplicationConfigurationFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterReplicationConfiguration, ToClusterReplicationConfiguration)\n}\n\nfunc TestWorkflowExecutionInfoFuzz(t *testing.T) {\n\t// [BUG] UpdateTime missing from mapper (not sent over proto)\n\t// [Intended] ParentDomainID, ParentDomain: converted to empty string instead of nil when ParentExecutionInfo exists but field is empty\n\t// [Intended] ParentInitiatedID: converted to 0 instead of nil when ParentExecutionInfo exists but field is 0\n\t// [BUG] CronSchedule is not round trip safe with an empty string\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionInfo, ToWorkflowExecutionInfo,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom, // ActiveClusterSelectionPolicy has mutually exclusive fields based on Strategy\n\t\t),\n\t\ttestutils.WithExcludedFields(\"UpdateTime\", \"ParentDomainID\", \"ParentDomain\", \"ParentInitiatedID\", \"CronSchedule\"),\n\t)\n}\n\nfunc TestIsolationGroupMetricsMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromIsolationGroupMetricsMap, ToIsolationGroupMetricsMap)\n}\n\nfunc TestSignalExternalWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalExternalWorkflowExecutionDecisionAttributes, ToSignalExternalWorkflowExecutionDecisionAttributes)\n}\n\nfunc TestSignalWithStartWorkflowExecutionAsyncRequestFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields, WorkflowIDReusePolicy enum\n\ttestutils.RunMapperFuzzTest(t, FromSignalWithStartWorkflowExecutionAsyncRequest, ToSignalWithStartWorkflowExecutionAsyncRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestPollerInfoArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPollerInfoArray, ToPollerInfoArray)\n}\n\nfunc TestResetStickyTaskListResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromResetStickyTaskListResponse, ToResetStickyTaskListResponse)\n}\n\nfunc TestAPITaskListPartitionFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAPITaskListPartition, ToAPITaskListPartition)\n}\n\nfunc TestListWorkflowExecutionsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListWorkflowExecutionsResponse, ToListWorkflowExecutionsResponse,\n\t\ttestutils.WithExcludedFields(\"Executions\"), // Executions is tested in FromWorkflowExecutionInfoFuzz test\n\t)\n}\n\nfunc TestParentExecutionInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromParentExecutionInfo, ToParentExecutionInfo)\n}\n\nfunc TestFailWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromFailWorkflowExecutionDecisionAttributes, ToFailWorkflowExecutionDecisionAttributes,\n\t\ttestutils.WithExcludedFields(\"Details\"),\n\t)\n}\n\nfunc TestGetClusterInfoResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetClusterInfoResponse, ToGetClusterInfoResponse)\n}\n\nfunc TestSignalWithStartWorkflowExecutionAsyncResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalWithStartWorkflowExecutionAsyncResponse, ToSignalWithStartWorkflowExecutionAsyncResponse)\n}\n\nfunc TestDescribeDomainResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDescribeDomainResponse, ToDescribeDomainResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\ttestutils.DomainStatusFuzzer,\n\t\t\tArchivalStatusFuzzer,\n\t\t\tfunc(r *types.DescribeDomainResponse, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// [BUG] On a round-trip, if DomainInfo, Configuration, or ReplicationConfiguration are nil, they become non-nil\n\t\t\t\tif r.DomainInfo == nil {\n\t\t\t\t\tr.DomainInfo = &types.DomainInfo{}\n\t\t\t\t}\n\t\t\t\tif r.Configuration == nil {\n\t\t\t\t\tr.Configuration = &types.DomainConfiguration{}\n\t\t\t\t}\n\t\t\t\tif r.ReplicationConfiguration == nil {\n\t\t\t\t\tr.ReplicationConfiguration = &types.DomainReplicationConfiguration{}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t\ttestutils.WithExcludedFields(\"EmitMetric\"), // EmitMetric is deprecated and permanently set to true\n\t)\n}\n\nfunc TestListWorkflowExecutionsRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListWorkflowExecutionsRequest, ToListWorkflowExecutionsRequest)\n}\n\nfunc TestRequestCancelActivityTaskFailedEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelActivityTaskFailedEventAttributes, ToRequestCancelActivityTaskFailedEventAttributes)\n}\n\nfunc TestStatusFilterFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStatusFilter, ToStatusFilter)\n}\n\nfunc TestPayloadFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPayload, ToPayload)\n}\n\nfunc TestRespondDecisionTaskCompletedResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondDecisionTaskCompletedResponse, ToRespondDecisionTaskCompletedResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tfunc(h *types.History, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(h)\n\t\t\t\t// nil Events slice becomes empty after proto roundtrip\n\t\t\t\tif h.Events == nil {\n\t\t\t\t\th.Events = []*types.HistoryEvent{}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestStartTimeFilterFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartTimeFilter, ToStartTimeFilter)\n}\n\nfunc TestWorkflowExecutionContinuedAsNewEventAttributesFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields\n\t// JitterStartSeconds don't roundtrip correctly\n\t// [BUG] FailureDetails requires FailureReason to be set\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionContinuedAsNewEventAttributes, ToWorkflowExecutionContinuedAsNewEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tContinueAsNewInitiatorFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"JitterStartSeconds\", \"FailureDetails\"),\n\t)\n}\n\nfunc TestGetSearchAttributesResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetSearchAttributesResponse, ToGetSearchAttributesResponse)\n}\n\nfunc TestRecordActivityTaskHeartbeatByIDRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRecordActivityTaskHeartbeatByIDRequest, ToRecordActivityTaskHeartbeatByIDRequest)\n}\n\nfunc TestTaskIDBlockFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromTaskIDBlock, ToTaskIDBlock)\n}\n\nfunc TestActivityLocalDispatchInfoMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromActivityLocalDispatchInfoMap, ToActivityLocalDispatchInfoMap)\n}\n\nfunc TestHeaderFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromHeader, ToHeader)\n}\n\nfunc TestQueryWorkflowRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromQueryWorkflowRequest, ToQueryWorkflowRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tQueryRejectConditionFuzzer,\n\t\t\tQueryConsistencyLevelFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestRecordActivityTaskHeartbeatByIDResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRecordActivityTaskHeartbeatByIDResponse, ToRecordActivityTaskHeartbeatByIDResponse)\n}\n\nfunc TestWorkflowExecutionConfigurationFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionConfiguration, ToWorkflowExecutionConfiguration)\n}\n\nfunc TestPendingActivityInfoFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\ttestutils.RunMapperFuzzTest(t, FromPendingActivityInfo, ToPendingActivityInfo,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tPendingActivityStateFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"LastFailureDetails\"),\n\t)\n}\n\nfunc TestDiagnoseWorkflowExecutionRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDiagnoseWorkflowExecutionRequest, ToDiagnoseWorkflowExecutionRequest)\n}\n\nfunc TestWorkflowExecutionStartedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields\n\t// JitterStartSeconds don't roundtrip\n\t// Empty string fields become nil\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionStartedEventAttributes, ToWorkflowExecutionStartedEventAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tfunc(e *types.WorkflowExecutionStartedEventAttributes, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(e)\n\t\t\t\t// Empty strings become nil after proto roundtrip\n\t\t\t\tif e.ParentWorkflowDomain != nil && *e.ParentWorkflowDomain == \"\" {\n\t\t\t\t\te.ParentWorkflowDomain = nil\n\t\t\t\t}\n\t\t\t\tif e.ParentWorkflowDomainID != nil && *e.ParentWorkflowDomainID == \"\" {\n\t\t\t\t\te.ParentWorkflowDomainID = nil\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t\ttestutils.WithExcludedFields(\"JitterStartSeconds\"),\n\t)\n}\n\nfunc TestGetWorkflowExecutionHistoryRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetWorkflowExecutionHistoryRequest, ToGetWorkflowExecutionHistoryRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tHistoryEventFilterTypeFuzzer,\n\t\t\tQueryConsistencyLevelFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestRequestCancelActivityTaskDecisionAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRequestCancelActivityTaskDecisionAttributes, ToRequestCancelActivityTaskDecisionAttributes)\n}\n\nfunc TestClusterFailoverFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterFailover, ToClusterFailover)\n}\n\nfunc TestDescribeDomainResponseDomainFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields based on Strategy\n\t// WorkflowExecutionRetentionPeriodInDays: nil→0 conversion\n\t// [BUG] DomainInfo, Configuration, ReplicationConfiguration must be non-nil\n\ttestutils.RunMapperFuzzTest(t, FromDescribeDomainResponseDomain, ToDescribeDomainResponseDomain,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\ttestutils.DomainStatusFuzzer,\n\t\t\tArchivalStatusFuzzer,\n\t\t\tfunc(r *types.DescribeDomainResponse, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// Proto mapper requires these to be non-nil\n\t\t\t\tif r.DomainInfo == nil {\n\t\t\t\t\tr.DomainInfo = &types.DomainInfo{}\n\t\t\t\t}\n\t\t\t\tif r.Configuration == nil {\n\t\t\t\t\tr.Configuration = &types.DomainConfiguration{}\n\t\t\t\t}\n\t\t\t\tif r.ReplicationConfiguration == nil {\n\t\t\t\t\tr.ReplicationConfiguration = &types.DomainReplicationConfiguration{}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t\ttestutils.WithExcludedFields(\"WorkflowExecutionRetentionPeriodInDays\", \"EmitMetric\"),\n\t)\n}\n\nfunc TestRetryPolicyFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRetryPolicy, ToRetryPolicy)\n}\n\nfunc TestStartWorkflowExecutionAsyncResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartWorkflowExecutionAsyncResponse, ToStartWorkflowExecutionAsyncResponse)\n}\n\nfunc TestAPITaskListPartitionConfigFuzz(t *testing.T) {\n\t// From and To are non-invertable operations, so we rely on testdata to verify the mapping is correct\n\ttestutils.RunMapperFuzzTest(t, FromAPITaskListPartitionConfig, ToAPITaskListPartitionConfig,\n\t\ttestutils.WithExcludedFields(\"ReadPartitions\", \"WritePartitions\"),\n\t)\n}\n\nfunc TestRespondQueryTaskCompletedRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRespondQueryTaskCompletedRequest, ToRespondQueryTaskCompletedRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tCompletedTypeFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestListTaskListPartitionsResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListTaskListPartitionsResponse, ToListTaskListPartitionsResponse)\n}\n\nfunc TestWorkflowTypeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowType, ToWorkflowType)\n}\n\nfunc TestWorkflowExecutionInfoArrayFuzz(t *testing.T) {\n\tt.Skip(\"Tested in FromWorkflowExecutionInfoFuzz test\")\n}\n\nfunc TestRecordActivityTaskHeartbeatRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRecordActivityTaskHeartbeatRequest, ToRecordActivityTaskHeartbeatRequest)\n}\n\nfunc TestWorkflowExecutionCancelRequestedEventAttributesFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An external initiated event ID of 0 becomes nil, but the return trip translates it back to nil\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionCancelRequestedEventAttributes, ToWorkflowExecutionCancelRequestedEventAttributes,\n\t\ttestutils.WithExcludedFields(\"ExternalInitiatedEventID\"),\n\t)\n}\n\nfunc TestUpdateDomainResponseFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields based on Strategy\n\t// WorkflowExecutionRetentionPeriodInDays: nil→0 conversion\n\t// DomainInfo, Configuration, ReplicationConfiguration must be non-nil\n\ttestutils.RunMapperFuzzTest(t, FromUpdateDomainResponse, ToUpdateDomainResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\ttestutils.DomainStatusFuzzer,\n\t\t\tArchivalStatusFuzzer,\n\t\t\tfunc(r *types.UpdateDomainResponse, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(r)\n\t\t\t\t// Proto mapper requires these to be non-nil\n\t\t\t\tif r.DomainInfo == nil {\n\t\t\t\t\tr.DomainInfo = &types.DomainInfo{}\n\t\t\t\t}\n\t\t\t\tif r.Configuration == nil {\n\t\t\t\t\tr.Configuration = &types.DomainConfiguration{}\n\t\t\t\t}\n\t\t\t\tif r.ReplicationConfiguration == nil {\n\t\t\t\t\tr.ReplicationConfiguration = &types.DomainReplicationConfiguration{}\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t\ttestutils.WithExcludedFields(\"WorkflowExecutionRetentionPeriodInDays\", \"EmitMetric\"),\n\t)\n}\n\nfunc TestWorkflowExecutionCanceledEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionCanceledEventAttributes, ToWorkflowExecutionCanceledEventAttributes)\n}\n\nfunc TestWorkflowExecutionTimedOutEventAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromWorkflowExecutionTimedOutEventAttributes, ToWorkflowExecutionTimedOutEventAttributes)\n}\n\nfunc TestPayloadMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPayloadMap, ToPayloadMap)\n}\n\nfunc TestAutoConfigHintFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromAutoConfigHint, ToAutoConfigHint)\n}\n\nfunc TestBadBinariesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBadBinaries, ToBadBinaries)\n}\n\nfunc TestSignalWithStartWorkflowExecutionResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSignalWithStartWorkflowExecutionResponse, ToSignalWithStartWorkflowExecutionResponse)\n}\n\nfunc TestStartChildWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields, WorkflowIDReusePolicy enum\n\ttestutils.RunMapperFuzzTest(t, FromStartChildWorkflowExecutionDecisionAttributes, ToStartChildWorkflowExecutionDecisionAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tWorkflowIDReusePolicyFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestActivityTaskScheduledEventAttributesFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An empty string domain becomes nil, but the return trip translates it back to nil - not empty string\n\ttestutils.RunMapperFuzzTest(t, FromActivityTaskScheduledEventAttributes, ToActivityTaskScheduledEventAttributes,\n\t\ttestutils.WithExcludedFields(\"Domain\"),\n\t)\n}\n\nfunc TestGetTaskListsByDomainRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetTaskListsByDomainRequest, ToGetTaskListsByDomainRequest)\n}\n\nfunc TestFailoverEventFuzz(t *testing.T) {\n\t// [BUG] Non-symmetric mapping: An empty string ID becomes nil, but the return trip translates it back to nil\n\ttestutils.RunMapperFuzzTest(t, FromFailoverEvent, ToFailoverEvent,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tFailoverTypeFuzzer,\n\t\t\tfunc(e *types.FailoverEvent, c fuzz.Continue) {\n\t\t\t\tc.Fuzz(e)\n\t\t\t\t// Empty string ID becomes nil\n\t\t\t\tif e.ID != nil && *e.ID == \"\" {\n\t\t\t\t\te.ID = nil\n\t\t\t\t}\n\t\t\t},\n\t\t),\n\t)\n}\n\nfunc TestContinueAsNewWorkflowExecutionDecisionAttributesFuzz(t *testing.T) {\n\t// [BUG] FromFailure only creates a Failure object if reason is non-nil, so details without reason are dropped\n\t// ActiveClusterSelectionPolicy has mutually exclusive fields\n\t// JitterStartSeconds doesn't roundtrip correctly\n\ttestutils.RunMapperFuzzTest(t, FromContinueAsNewWorkflowExecutionDecisionAttributes, ToContinueAsNewWorkflowExecutionDecisionAttributes,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tActiveClusterSelectionPolicyFuzzerNoCustom,\n\t\t\tContinueAsNewInitiatorFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\"JitterStartSeconds\", \"FailureDetails\"),\n\t)\n}\n\nfunc TestClusterAttributeScopeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromClusterAttributeScope, ToClusterAttributeScope)\n}\n\nfunc TestBadBinaryInfoMapFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBadBinaryInfoMap, ToBadBinaryInfoMap)\n}\n"
  },
  {
    "path": "common/types/mapper/proto/enum_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst UnknownValue = 9999\n\nfunc TestTaskSource(t *testing.T) {\n\tfor _, item := range []*types.TaskSource{\n\t\tnil,\n\t\ttypes.TaskSourceHistory.Ptr(),\n\t\ttypes.TaskSourceDbBacklog.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToTaskSource(FromTaskSource(item)))\n\t}\n}\n\nfunc TestToTaskSource(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    sharedv1.TaskSource\n\t\texpected *types.TaskSource\n\t}{\n\t\t{\n\t\t\tname:  \"when input is invalid it should return nil\",\n\t\t\tinput: sharedv1.TaskSource_TASK_SOURCE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:  \"when input is out of range it should return nil\",\n\t\t\tinput: UnknownValue,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    sharedv1.TaskSource_TASK_SOURCE_HISTORY,\n\t\t\texpected: types.TaskSourceHistory.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToTaskSource(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromTaskSource(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.TaskSource\n\t\texpected sharedv1.TaskSource\n\t}{\n\t\t{\n\t\t\tname:  \"when input is nil it should return INVALID\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.TaskSourceHistory.Ptr(),\n\t\t\texpected: sharedv1.TaskSource_TASK_SOURCE_HISTORY,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.TaskSource(UnknownValue).Ptr(),\n\t\t\texpected: sharedv1.TaskSource_TASK_SOURCE_INVALID,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromTaskSource(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestDLQType(t *testing.T) {\n\tfor _, item := range []*types.DLQType{\n\t\tnil,\n\t\ttypes.DLQTypeReplication.Ptr(),\n\t\ttypes.DLQTypeDomain.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToDLQType(FromDLQType(item)))\n\t}\n}\n\nfunc TestToDLQType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    adminv1.DLQType\n\t\texpected *types.DLQType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    adminv1.DLQType_DLQ_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    UnknownValue,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    adminv1.DLQType_DLQ_TYPE_REPLICATION,\n\t\t\texpected: types.DLQTypeReplication.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToDLQType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromDLQType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.DLQType\n\t\texpected adminv1.DLQType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: adminv1.DLQType_DLQ_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.DLQType(UnknownValue).Ptr(),\n\t\t\texpected: adminv1.DLQType_DLQ_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.DLQTypeReplication.Ptr(),\n\t\t\texpected: adminv1.DLQType_DLQ_TYPE_REPLICATION,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromDLQType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestDomainOperation(t *testing.T) {\n\tfor _, item := range []*types.DomainOperation{\n\t\tnil,\n\t\ttypes.DomainOperationCreate.Ptr(),\n\t\ttypes.DomainOperationUpdate.Ptr(),\n\t\ttypes.DomainOperationDelete.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToDomainOperation(FromDomainOperation(item)))\n\t}\n}\n\nfunc TestToDomainOperation(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    adminv1.DomainOperation\n\t\texpected *types.DomainOperation\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    adminv1.DomainOperation_DOMAIN_OPERATION_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    UnknownValue,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    adminv1.DomainOperation_DOMAIN_OPERATION_CREATE,\n\t\t\texpected: types.DomainOperationCreate.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToDomainOperation(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromDomainOperation(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.DomainOperation\n\t\texpected adminv1.DomainOperation\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: adminv1.DomainOperation_DOMAIN_OPERATION_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.DomainOperation(UnknownValue).Ptr(),\n\t\t\texpected: adminv1.DomainOperation_DOMAIN_OPERATION_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.DomainOperationCreate.Ptr(),\n\t\t\texpected: adminv1.DomainOperation_DOMAIN_OPERATION_CREATE,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromDomainOperation(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestReplicationTaskType(t *testing.T) {\n\tfor _, item := range []*types.ReplicationTaskType{\n\t\tnil,\n\t\ttypes.ReplicationTaskTypeDomain.Ptr(),\n\t\ttypes.ReplicationTaskTypeHistory.Ptr(),\n\t\ttypes.ReplicationTaskTypeSyncShardStatus.Ptr(),\n\t\ttypes.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\ttypes.ReplicationTaskTypeHistoryMetadata.Ptr(),\n\t\ttypes.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\ttypes.ReplicationTaskTypeFailoverMarker.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToReplicationTaskType(FromReplicationTaskType(item)))\n\t}\n}\n\nfunc TestToReplicationTaskType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    adminv1.ReplicationTaskType\n\t\texpected *types.ReplicationTaskType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    adminv1.ReplicationTaskType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_DOMAIN,\n\t\t\texpected: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToReplicationTaskType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromReplicationTaskType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.ReplicationTaskType\n\t\texpected adminv1.ReplicationTaskType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.ReplicationTaskType(UnknownValue).Ptr(),\n\t\t\texpected: adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\texpected: adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_DOMAIN,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromReplicationTaskType(tc.input))\n\t\t})\n\t}\n}\nfunc TestArchivalStatus(t *testing.T) {\n\tfor _, item := range []*types.ArchivalStatus{\n\t\tnil,\n\t\ttypes.ArchivalStatusDisabled.Ptr(),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToArchivalStatus(FromArchivalStatus(item)))\n\t}\n}\n\nfunc TestToArchivalStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.ArchivalStatus\n\t\texpected *types.ArchivalStatus\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.ArchivalStatus(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is disabled it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.ArchivalStatus_ARCHIVAL_STATUS_DISABLED,\n\t\t\texpected: types.ArchivalStatusDisabled.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToArchivalStatus(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromArchivalStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.ArchivalStatus\n\t\texpected apiv1.ArchivalStatus\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.ArchivalStatus(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.ArchivalStatus_ARCHIVAL_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is disabled it should return the correctly mapped value\",\n\t\t\tinput:    types.ArchivalStatusDisabled.Ptr(),\n\t\t\texpected: apiv1.ArchivalStatus_ARCHIVAL_STATUS_DISABLED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromArchivalStatus(tc.input))\n\t\t})\n\t}\n}\nfunc TestCancelExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tfor _, item := range []*types.CancelExternalWorkflowExecutionFailedCause{\n\t\tnil,\n\t\ttypes.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\ttypes.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToCancelExternalWorkflowExecutionFailedCause(FromCancelExternalWorkflowExecutionFailedCause(item)))\n\t}\n}\n\nfunc TestToCancelExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.CancelExternalWorkflowExecutionFailedCause\n\t\texpected *types.CancelExternalWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.CancelExternalWorkflowExecutionFailedCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unknown external workflow execution it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\n\t\t\texpected: types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToCancelExternalWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromCancelExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.CancelExternalWorkflowExecutionFailedCause\n\t\texpected apiv1.CancelExternalWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.CancelExternalWorkflowExecutionFailedCause(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unknown external workflow execution it should return the correctly mapped value\",\n\t\t\tinput:    types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t\texpected: apiv1.CancelExternalWorkflowExecutionFailedCause_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromCancelExternalWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\nfunc TestChildWorkflowExecutionFailedCause(t *testing.T) {\n\tfor _, item := range []*types.ChildWorkflowExecutionFailedCause{\n\t\tnil,\n\t\ttypes.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToChildWorkflowExecutionFailedCause(FromChildWorkflowExecutionFailedCause(item)))\n\t}\n}\n\nfunc TestToChildWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.ChildWorkflowExecutionFailedCause\n\t\texpected *types.ChildWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.ChildWorkflowExecutionFailedCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is workflow already running it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING,\n\t\t\texpected: types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToChildWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromChildWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.ChildWorkflowExecutionFailedCause\n\t\texpected apiv1.ChildWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.ChildWorkflowExecutionFailedCause(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is workflow already running it should return the correctly mapped value\",\n\t\t\tinput:    types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t\t\texpected: apiv1.ChildWorkflowExecutionFailedCause_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromChildWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\nfunc TestContinueAsNewInitiator(t *testing.T) {\n\tfor _, item := range []*types.ContinueAsNewInitiator{\n\t\tnil,\n\t\ttypes.ContinueAsNewInitiatorDecider.Ptr(),\n\t\ttypes.ContinueAsNewInitiatorRetryPolicy.Ptr(),\n\t\ttypes.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToContinueAsNewInitiator(FromContinueAsNewInitiator(item)))\n\t}\n}\n\nfunc TestToContinueAsNewInitiator(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.ContinueAsNewInitiator\n\t\texpected *types.ContinueAsNewInitiator\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.ContinueAsNewInitiator(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is decider it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_DECIDER,\n\t\t\texpected: types.ContinueAsNewInitiatorDecider.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToContinueAsNewInitiator(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromContinueAsNewInitiator(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.ContinueAsNewInitiator\n\t\texpected apiv1.ContinueAsNewInitiator\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.ContinueAsNewInitiator(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is decider it should return the correctly mapped value\",\n\t\t\tinput:    types.ContinueAsNewInitiatorDecider.Ptr(),\n\t\t\texpected: apiv1.ContinueAsNewInitiator_CONTINUE_AS_NEW_INITIATOR_DECIDER,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromContinueAsNewInitiator(tc.input))\n\t\t})\n\t}\n}\nfunc TestCrossClusterTaskFailedCause(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskFailedCause{\n\t\tnil,\n\t\ttypes.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t\ttypes.CrossClusterTaskFailedCauseDomainNotExists.Ptr(),\n\t\ttypes.CrossClusterTaskFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t\ttypes.CrossClusterTaskFailedCauseWorkflowNotExists.Ptr(),\n\t\ttypes.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted.Ptr(),\n\t\ttypes.CrossClusterTaskFailedCauseUncategorized.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskFailedCause(FromCrossClusterTaskFailedCause(item)))\n\t}\n}\n\nfunc TestToCrossClusterTaskFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    adminv1.CrossClusterTaskFailedCause\n\t\texpected *types.CrossClusterTaskFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    adminv1.CrossClusterTaskFailedCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is domain not active it should return the correctly mapped value\",\n\t\t\tinput:    adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_ACTIVE,\n\t\t\texpected: types.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToCrossClusterTaskFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromCrossClusterTaskFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.CrossClusterTaskFailedCause\n\t\texpected adminv1.CrossClusterTaskFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.CrossClusterTaskFailedCause(UnknownValue).Ptr(),\n\t\t\texpected: adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is domain not active it should return the correctly mapped value\",\n\t\t\tinput:    types.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t\t\texpected: adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_ACTIVE,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromCrossClusterTaskFailedCause(tc.input))\n\t\t})\n\t}\n}\nfunc TestDecisionTaskFailedCause(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskFailedCause{\n\t\tnil,\n\t\ttypes.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadScheduleActivityAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelActivityAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadStartTimerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCancelTimerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRecordMarkerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadContinueAsNewAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseStartTimerDuplicateID.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseResetStickyTasklist.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadStartChildExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseForceCloseDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseFailoverCloseDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSignalInputSize.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseResetWorkflow.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadBinary.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseScheduleActivityDuplicateID.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSearchAttributes.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToDecisionTaskFailedCause(FromDecisionTaskFailedCause(item)))\n\t}\n}\n\nfunc TestToDecisionTaskFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.DecisionTaskFailedCause\n\t\texpected *types.DecisionTaskFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.DecisionTaskFailedCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unhandled decision it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_UNHANDLED_DECISION,\n\t\t\texpected: types.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToDecisionTaskFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromDecisionTaskFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.DecisionTaskFailedCause\n\t\texpected apiv1.DecisionTaskFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.DecisionTaskFailedCause(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unhandled decision it should return the correctly mapped value\",\n\t\t\tinput:    types.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\t\texpected: apiv1.DecisionTaskFailedCause_DECISION_TASK_FAILED_CAUSE_UNHANDLED_DECISION,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromDecisionTaskFailedCause(tc.input))\n\t\t})\n\t}\n}\nfunc TestDomainStatus(t *testing.T) {\n\tfor _, item := range []*types.DomainStatus{\n\t\tnil,\n\t\ttypes.DomainStatusRegistered.Ptr(),\n\t\ttypes.DomainStatusDeprecated.Ptr(),\n\t\ttypes.DomainStatusDeleted.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToDomainStatus(FromDomainStatus(item)))\n\t}\n}\n\nfunc TestToDomainStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.DomainStatus\n\t\texpected *types.DomainStatus\n\t}{\n\t\t{\n\t\t\tname:  \"when input is invalid it should return nil\",\n\t\t\tinput: apiv1.DomainStatus_DOMAIN_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:  \"when input is out of range it should return nil\",\n\t\t\tinput: apiv1.DomainStatus(UnknownValue),\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.DomainStatus_DOMAIN_STATUS_REGISTERED,\n\t\t\texpected: types.DomainStatusRegistered.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToDomainStatus(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromDomainStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.DomainStatus\n\t\texpected apiv1.DomainStatus\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.DomainStatus_DOMAIN_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.DomainStatus(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.DomainStatus_DOMAIN_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.DomainStatusRegistered.Ptr(),\n\t\t\texpected: apiv1.DomainStatus_DOMAIN_STATUS_REGISTERED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromDomainStatus(tc.input))\n\t\t})\n\t}\n}\nfunc TestEncodingType(t *testing.T) {\n\tfor _, item := range []*types.EncodingType{\n\t\tnil,\n\t\ttypes.EncodingTypeThriftRW.Ptr(),\n\t\ttypes.EncodingTypeJSON.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToEncodingType(FromEncodingType(item)))\n\t}\n}\n\nfunc TestToEncodingType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.EncodingType\n\t\texpected *types.EncodingType\n\t}{\n\t\t{\n\t\t\tname:  \"when input is invalid it should return nil\",\n\t\t\tinput: apiv1.EncodingType_ENCODING_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:  \"when input is out of range it should return nil\",\n\t\t\tinput: apiv1.EncodingType(UnknownValue),\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.EncodingType_ENCODING_TYPE_THRIFTRW,\n\t\t\texpected: types.EncodingTypeThriftRW.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToEncodingType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromEncodingType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.EncodingType\n\t\texpected apiv1.EncodingType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.EncodingType_ENCODING_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.EncodingType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.EncodingType_ENCODING_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.EncodingTypeThriftRW.Ptr(),\n\t\t\texpected: apiv1.EncodingType_ENCODING_TYPE_THRIFTRW,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromEncodingType(tc.input))\n\t\t})\n\t}\n}\nfunc TestEventFilterType(t *testing.T) {\n\tfor _, item := range []*types.HistoryEventFilterType{\n\t\tnil,\n\t\ttypes.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\ttypes.HistoryEventFilterTypeCloseEvent.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToEventFilterType(FromEventFilterType(item)))\n\t}\n}\n\nfunc TestToEventFilterType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.EventFilterType\n\t\texpected *types.HistoryEventFilterType\n\t}{\n\t\t{\n\t\t\tname:  \"when input is invalid it should return nil\",\n\t\t\tinput: apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:  \"when input is out of range it should return nil\",\n\t\t\tinput: apiv1.EventFilterType(UnknownValue),\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.EventFilterType_EVENT_FILTER_TYPE_ALL_EVENT,\n\t\t\texpected: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToEventFilterType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromEventFilterType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.HistoryEventFilterType\n\t\texpected apiv1.EventFilterType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.HistoryEventFilterType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.EventFilterType_EVENT_FILTER_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t\texpected: apiv1.EventFilterType_EVENT_FILTER_TYPE_ALL_EVENT,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromEventFilterType(tc.input))\n\t\t})\n\t}\n}\nfunc TestIndexedValueType(t *testing.T) {\n\tfor _, item := range []types.IndexedValueType{\n\t\ttypes.IndexedValueTypeString,\n\t\ttypes.IndexedValueTypeKeyword,\n\t\ttypes.IndexedValueTypeInt,\n\t\ttypes.IndexedValueTypeDouble,\n\t\ttypes.IndexedValueTypeBool,\n\t\ttypes.IndexedValueTypeDatetime,\n\t} {\n\t\tassert.Equal(t, item, ToIndexedValueType(FromIndexedValueType(item)))\n\t}\n}\n\nfunc TestToIndexedValueType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.IndexedValueType\n\t\texpected types.IndexedValueType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return string\",\n\t\t\tinput:    apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INVALID,\n\t\t\texpected: types.IndexedValueTypeString,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return string\",\n\t\t\tinput:    apiv1.IndexedValueType(UnknownValue),\n\t\t\texpected: types.IndexedValueTypeString,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.IndexedValueType_INDEXED_VALUE_TYPE_STRING,\n\t\t\texpected: types.IndexedValueTypeString,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToIndexedValueType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromIndexedValueType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    types.IndexedValueType\n\t\texpected apiv1.IndexedValueType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.IndexedValueType(UnknownValue),\n\t\t\texpected: apiv1.IndexedValueType_INDEXED_VALUE_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is valid it should return the correctly mapped value\",\n\t\t\tinput:    types.IndexedValueTypeString,\n\t\t\texpected: apiv1.IndexedValueType_INDEXED_VALUE_TYPE_STRING,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromIndexedValueType(tc.input))\n\t\t})\n\t}\n}\nfunc TestParentClosePolicy(t *testing.T) {\n\tfor _, item := range []*types.ParentClosePolicy{\n\t\tnil,\n\t\ttypes.ParentClosePolicyAbandon.Ptr(),\n\t\ttypes.ParentClosePolicyRequestCancel.Ptr(),\n\t\ttypes.ParentClosePolicyTerminate.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToParentClosePolicy(FromParentClosePolicy(item)))\n\t}\n}\n\nfunc TestToParentClosePolicy(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.ParentClosePolicy\n\t\texpected *types.ParentClosePolicy\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.ParentClosePolicy(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is abandon it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_ABANDON,\n\t\t\texpected: types.ParentClosePolicyAbandon.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToParentClosePolicy(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromParentClosePolicy(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.ParentClosePolicy\n\t\texpected apiv1.ParentClosePolicy\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.ParentClosePolicy(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is abandon it should return the correctly mapped value\",\n\t\t\tinput:    types.ParentClosePolicyAbandon.Ptr(),\n\t\t\texpected: apiv1.ParentClosePolicy_PARENT_CLOSE_POLICY_ABANDON,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromParentClosePolicy(tc.input))\n\t\t})\n\t}\n}\nfunc TestPendingActivityState(t *testing.T) {\n\tfor _, item := range []*types.PendingActivityState{\n\t\tnil,\n\t\ttypes.PendingActivityStateScheduled.Ptr(),\n\t\ttypes.PendingActivityStateStarted.Ptr(),\n\t\ttypes.PendingActivityStateCancelRequested.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToPendingActivityState(FromPendingActivityState(item)))\n\t}\n}\n\nfunc TestToPendingActivityState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.PendingActivityState\n\t\texpected *types.PendingActivityState\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.PendingActivityState(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is scheduled it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_SCHEDULED,\n\t\t\texpected: types.PendingActivityStateScheduled.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToPendingActivityState(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromPendingActivityState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.PendingActivityState\n\t\texpected apiv1.PendingActivityState\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.PendingActivityState(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is scheduled it should return the correctly mapped value\",\n\t\t\tinput:    types.PendingActivityStateScheduled.Ptr(),\n\t\t\texpected: apiv1.PendingActivityState_PENDING_ACTIVITY_STATE_SCHEDULED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromPendingActivityState(tc.input))\n\t\t})\n\t}\n}\nfunc TestPendingDecisionState(t *testing.T) {\n\tfor _, item := range []*types.PendingDecisionState{\n\t\tnil,\n\t\ttypes.PendingDecisionStateScheduled.Ptr(),\n\t\ttypes.PendingDecisionStateStarted.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToPendingDecisionState(FromPendingDecisionState(item)))\n\t}\n}\n\nfunc TestToPendingDecisionState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.PendingDecisionState\n\t\texpected *types.PendingDecisionState\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.PendingDecisionState(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is scheduled it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.PendingDecisionState_PENDING_DECISION_STATE_SCHEDULED,\n\t\t\texpected: types.PendingDecisionStateScheduled.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToPendingDecisionState(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromPendingDecisionState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.PendingDecisionState\n\t\texpected apiv1.PendingDecisionState\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.PendingDecisionState(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.PendingDecisionState_PENDING_DECISION_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is scheduled it should return the correctly mapped value\",\n\t\t\tinput:    types.PendingDecisionStateScheduled.Ptr(),\n\t\t\texpected: apiv1.PendingDecisionState_PENDING_DECISION_STATE_SCHEDULED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromPendingDecisionState(tc.input))\n\t\t})\n\t}\n}\nfunc TestQueryConsistencyLevel(t *testing.T) {\n\tfor _, item := range []*types.QueryConsistencyLevel{\n\t\tnil,\n\t\ttypes.QueryConsistencyLevelEventual.Ptr(),\n\t\ttypes.QueryConsistencyLevelStrong.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToQueryConsistencyLevel(FromQueryConsistencyLevel(item)))\n\t}\n}\n\nfunc TestToQueryConsistencyLevel(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.QueryConsistencyLevel\n\t\texpected *types.QueryConsistencyLevel\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.QueryConsistencyLevel(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is eventual it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_EVENTUAL,\n\t\t\texpected: types.QueryConsistencyLevelEventual.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToQueryConsistencyLevel(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromQueryConsistencyLevel(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.QueryConsistencyLevel\n\t\texpected apiv1.QueryConsistencyLevel\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.QueryConsistencyLevel(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is eventual it should return the correctly mapped value\",\n\t\t\tinput:    types.QueryConsistencyLevelEventual.Ptr(),\n\t\t\texpected: apiv1.QueryConsistencyLevel_QUERY_CONSISTENCY_LEVEL_EVENTUAL,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromQueryConsistencyLevel(tc.input))\n\t\t})\n\t}\n}\nfunc TestQueryRejectCondition(t *testing.T) {\n\tfor _, item := range []*types.QueryRejectCondition{\n\t\tnil,\n\t\ttypes.QueryRejectConditionNotOpen.Ptr(),\n\t\ttypes.QueryRejectConditionNotCompletedCleanly.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToQueryRejectCondition(FromQueryRejectCondition(item)))\n\t}\n}\n\nfunc TestToQueryRejectCondition(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.QueryRejectCondition\n\t\texpected *types.QueryRejectCondition\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.QueryRejectCondition(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is not open it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_OPEN,\n\t\t\texpected: types.QueryRejectConditionNotOpen.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToQueryRejectCondition(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromQueryRejectCondition(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.QueryRejectCondition\n\t\texpected apiv1.QueryRejectCondition\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.QueryRejectCondition(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is not open it should return the correctly mapped value\",\n\t\t\tinput:    types.QueryRejectConditionNotOpen.Ptr(),\n\t\t\texpected: apiv1.QueryRejectCondition_QUERY_REJECT_CONDITION_NOT_OPEN,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromQueryRejectCondition(tc.input))\n\t\t})\n\t}\n}\nfunc TestQueryResultType(t *testing.T) {\n\tfor _, item := range []*types.QueryResultType{\n\t\tnil,\n\t\ttypes.QueryResultTypeAnswered.Ptr(),\n\t\ttypes.QueryResultTypeFailed.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToQueryResultType(FromQueryResultType(item)))\n\t}\n}\n\nfunc TestToQueryResultType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.QueryResultType\n\t\texpected *types.QueryResultType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.QueryResultType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is answered it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED,\n\t\t\texpected: types.QueryResultTypeAnswered.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToQueryResultType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromQueryResultType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.QueryResultType\n\t\texpected apiv1.QueryResultType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.QueryResultType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is answered it should return the correctly mapped value\",\n\t\t\tinput:    types.QueryResultTypeAnswered.Ptr(),\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromQueryResultType(tc.input))\n\t\t})\n\t}\n}\nfunc TestQueryTaskCompletedType(t *testing.T) {\n\tfor _, item := range []*types.QueryTaskCompletedType{\n\t\tnil,\n\t\ttypes.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\ttypes.QueryTaskCompletedTypeFailed.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToQueryTaskCompletedType(FromQueryTaskCompletedType(item)))\n\t}\n}\n\nfunc TestToQueryTaskCompletedType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.QueryResultType\n\t\texpected *types.QueryTaskCompletedType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.QueryResultType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is completed it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED,\n\t\t\texpected: types.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToQueryTaskCompletedType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromQueryTaskCompletedType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.QueryTaskCompletedType\n\t\texpected apiv1.QueryResultType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.QueryTaskCompletedType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is completed it should return the correctly mapped value\",\n\t\t\tinput:    types.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\t\texpected: apiv1.QueryResultType_QUERY_RESULT_TYPE_ANSWERED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromQueryTaskCompletedType(tc.input))\n\t\t})\n\t}\n}\nfunc TestSignalExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tfor _, item := range []*types.SignalExternalWorkflowExecutionFailedCause{\n\t\tnil,\n\t\ttypes.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\ttypes.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToSignalExternalWorkflowExecutionFailedCause(FromSignalExternalWorkflowExecutionFailedCause(item)))\n\t}\n}\n\nfunc TestToSignalExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.SignalExternalWorkflowExecutionFailedCause\n\t\texpected *types.SignalExternalWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.SignalExternalWorkflowExecutionFailedCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unknown external workflow execution it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\n\t\t\texpected: types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToSignalExternalWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromSignalExternalWorkflowExecutionFailedCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.SignalExternalWorkflowExecutionFailedCause\n\t\texpected apiv1.SignalExternalWorkflowExecutionFailedCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.SignalExternalWorkflowExecutionFailedCause(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is unknown external workflow execution it should return the correctly mapped value\",\n\t\t\tinput:    types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\t\texpected: apiv1.SignalExternalWorkflowExecutionFailedCause_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromSignalExternalWorkflowExecutionFailedCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestTaskListKind(t *testing.T) {\n\tfor _, item := range []*types.TaskListKind{\n\t\tnil,\n\t\ttypes.TaskListKindNormal.Ptr(),\n\t\ttypes.TaskListKindSticky.Ptr(),\n\t\ttypes.TaskListKindEphemeral.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToTaskListKind(FromTaskListKind(item)))\n\t}\n}\n\nfunc TestToTaskListKind(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.TaskListKind\n\t\texpected *types.TaskListKind\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.TaskListKind_TASK_LIST_KIND_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.TaskListKind(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is normal it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.TaskListKind_TASK_LIST_KIND_NORMAL,\n\t\t\texpected: types.TaskListKindNormal.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToTaskListKind(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromTaskListKind(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.TaskListKind\n\t\texpected apiv1.TaskListKind\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.TaskListKind_TASK_LIST_KIND_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.TaskListKind(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.TaskListKind_TASK_LIST_KIND_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is normal it should return the correctly mapped value\",\n\t\t\tinput:    types.TaskListKindNormal.Ptr(),\n\t\t\texpected: apiv1.TaskListKind_TASK_LIST_KIND_NORMAL,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromTaskListKind(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestTaskListType(t *testing.T) {\n\tfor _, item := range []*types.TaskListType{\n\t\tnil,\n\t\ttypes.TaskListTypeDecision.Ptr(),\n\t\ttypes.TaskListTypeActivity.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToTaskListType(FromTaskListType(item)))\n\t}\n}\n\nfunc TestToTaskListType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.TaskListType\n\t\texpected *types.TaskListType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.TaskListType_TASK_LIST_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.TaskListType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is decision it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.TaskListType_TASK_LIST_TYPE_DECISION,\n\t\t\texpected: types.TaskListTypeDecision.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToTaskListType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromTaskListType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.TaskListType\n\t\texpected apiv1.TaskListType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.TaskListType_TASK_LIST_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.TaskListType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.TaskListType_TASK_LIST_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is decision it should return the correctly mapped value\",\n\t\t\tinput:    types.TaskListTypeDecision.Ptr(),\n\t\t\texpected: apiv1.TaskListType_TASK_LIST_TYPE_DECISION,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromTaskListType(tc.input))\n\t\t})\n\t}\n}\nfunc TestTimeoutType(t *testing.T) {\n\tfor _, item := range []*types.TimeoutType{\n\t\tnil,\n\t\ttypes.TimeoutTypeStartToClose.Ptr(),\n\t\ttypes.TimeoutTypeScheduleToStart.Ptr(),\n\t\ttypes.TimeoutTypeScheduleToClose.Ptr(),\n\t\ttypes.TimeoutTypeHeartbeat.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToTimeoutType(FromTimeoutType(item)))\n\t}\n}\n\nfunc TestToTimeoutType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.TimeoutType\n\t\texpected *types.TimeoutType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.TimeoutType_TIMEOUT_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.TimeoutType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is start to close it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.TimeoutType_TIMEOUT_TYPE_START_TO_CLOSE,\n\t\t\texpected: types.TimeoutTypeStartToClose.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToTimeoutType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromTimeoutType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.TimeoutType\n\t\texpected apiv1.TimeoutType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.TimeoutType_TIMEOUT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.TimeoutType(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.TimeoutType_TIMEOUT_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is start to close it should return the correctly mapped value\",\n\t\t\tinput:    types.TimeoutTypeStartToClose.Ptr(),\n\t\t\texpected: apiv1.TimeoutType_TIMEOUT_TYPE_START_TO_CLOSE,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromTimeoutType(tc.input))\n\t\t})\n\t}\n}\nfunc TestDecisionTaskTimedOutCause(t *testing.T) {\n\tfor _, item := range []*types.DecisionTaskTimedOutCause{\n\t\tnil,\n\t\ttypes.DecisionTaskTimedOutCauseTimeout.Ptr(),\n\t\ttypes.DecisionTaskTimedOutCauseReset.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToDecisionTaskTimedOutCause(FromDecisionTaskTimedOutCause(item)))\n\t}\n}\n\nfunc TestToDecisionTaskTimedOutCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.DecisionTaskTimedOutCause\n\t\texpected *types.DecisionTaskTimedOutCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.DecisionTaskTimedOutCause(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is timeout it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_TIMEOUT,\n\t\t\texpected: types.DecisionTaskTimedOutCauseTimeout.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToDecisionTaskTimedOutCause(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromDecisionTaskTimedOutCause(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.DecisionTaskTimedOutCause\n\t\texpected apiv1.DecisionTaskTimedOutCause\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.DecisionTaskTimedOutCause(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is timeout it should return the correctly mapped value\",\n\t\t\tinput:    types.DecisionTaskTimedOutCauseTimeout.Ptr(),\n\t\t\texpected: apiv1.DecisionTaskTimedOutCause_DECISION_TASK_TIMED_OUT_CAUSE_TIMEOUT,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromDecisionTaskTimedOutCause(tc.input))\n\t\t})\n\t}\n}\nfunc TestWorkflowExecutionCloseStatus(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionCloseStatus{\n\t\tnil,\n\t\ttypes.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusCanceled.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusTerminated.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusTimedOut.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionCloseStatus(FromWorkflowExecutionCloseStatus(item)))\n\t}\n}\n\nfunc TestToWorkflowExecutionCloseStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.WorkflowExecutionCloseStatus\n\t\texpected *types.WorkflowExecutionCloseStatus\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.WorkflowExecutionCloseStatus(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is completed it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_COMPLETED,\n\t\t\texpected: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToWorkflowExecutionCloseStatus(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromWorkflowExecutionCloseStatus(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.WorkflowExecutionCloseStatus\n\t\texpected apiv1.WorkflowExecutionCloseStatus\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.WorkflowExecutionCloseStatus(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is completed it should return the correctly mapped value\",\n\t\t\tinput:    types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\texpected: apiv1.WorkflowExecutionCloseStatus_WORKFLOW_EXECUTION_CLOSE_STATUS_COMPLETED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromWorkflowExecutionCloseStatus(tc.input))\n\t\t})\n\t}\n}\nfunc TestWorkflowIDReusePolicy(t *testing.T) {\n\tfor _, item := range []*types.WorkflowIDReusePolicy{\n\t\tnil,\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicateFailedOnly.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyRejectDuplicate.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t} {\n\t\tassert.Equal(t, item, ToWorkflowIDReusePolicy(FromWorkflowIDReusePolicy(item)))\n\t}\n}\n\nfunc TestToWorkflowIDReusePolicy(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    apiv1.WorkflowIdReusePolicy\n\t\texpected *types.WorkflowIDReusePolicy\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    apiv1.WorkflowIdReusePolicy(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is allow duplicate failed only it should return the correctly mapped value\",\n\t\t\tinput:    apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE_FAILED_ONLY,\n\t\t\texpected: types.WorkflowIDReusePolicyAllowDuplicateFailedOnly.Ptr(),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToWorkflowIDReusePolicy(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromWorkflowIDReusePolicy(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *types.WorkflowIDReusePolicy\n\t\texpected apiv1.WorkflowIdReusePolicy\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    types.WorkflowIDReusePolicy(UnknownValue).Ptr(),\n\t\t\texpected: apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is allow duplicate failed only it should return the correctly mapped value\",\n\t\t\tinput:    types.WorkflowIDReusePolicyAllowDuplicateFailedOnly.Ptr(),\n\t\t\texpected: apiv1.WorkflowIdReusePolicy_WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE_FAILED_ONLY,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromWorkflowIDReusePolicy(tc.input))\n\t\t})\n\t}\n}\nfunc TestWorkflowState(t *testing.T) {\n\tfor _, item := range []*int32{\n\t\tnil,\n\t\tcommon.Int32Ptr(persistence.WorkflowStateCreated),\n\t\tcommon.Int32Ptr(persistence.WorkflowStateRunning),\n\t\tcommon.Int32Ptr(persistence.WorkflowStateCompleted),\n\t\tcommon.Int32Ptr(persistence.WorkflowStateZombie),\n\t\tcommon.Int32Ptr(persistence.WorkflowStateVoid),\n\t\tcommon.Int32Ptr(persistence.WorkflowStateCorrupted),\n\t} {\n\t\tassert.Equal(t, item, ToWorkflowState(FromWorkflowState(item)))\n\t}\n}\n\nfunc TestToWorkflowState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    sharedv1.WorkflowState\n\t\texpected *int32\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    sharedv1.WorkflowState_WORKFLOW_STATE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    sharedv1.WorkflowState(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is created it should return the correctly mapped value\",\n\t\t\tinput:    sharedv1.WorkflowState_WORKFLOW_STATE_CREATED,\n\t\t\texpected: common.Int32Ptr(persistence.WorkflowStateCreated),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToWorkflowState(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromWorkflowState(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *int32\n\t\texpected sharedv1.WorkflowState\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: sharedv1.WorkflowState_WORKFLOW_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    common.Int32Ptr(UnknownValue),\n\t\t\texpected: sharedv1.WorkflowState_WORKFLOW_STATE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is created it should return the correctly mapped value\",\n\t\t\tinput:    common.Int32Ptr(persistence.WorkflowStateCreated),\n\t\t\texpected: sharedv1.WorkflowState_WORKFLOW_STATE_CREATED,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromWorkflowState(tc.input))\n\t\t})\n\t}\n}\nfunc TestTaskType(t *testing.T) {\n\tfor _, item := range []*int32{\n\t\tnil,\n\t\tcommon.Int32Ptr(int32(constants.TaskTypeTransfer)),\n\t\tcommon.Int32Ptr(int32(constants.TaskTypeTimer)),\n\t\tcommon.Int32Ptr(int32(constants.TaskTypeReplication)),\n\t\tcommon.Int32Ptr(int32(constants.TaskTypeCrossCluster)),\n\t} {\n\t\tassert.Equal(t, item, ToTaskType(FromTaskType(item)))\n\t}\n}\n\nfunc TestToTaskType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    adminv1.TaskType\n\t\texpected *int32\n\t}{\n\t\t{\n\t\t\tname:     \"when input is invalid it should return nil\",\n\t\t\tinput:    adminv1.TaskType_TASK_TYPE_INVALID,\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return nil\",\n\t\t\tinput:    adminv1.TaskType(UnknownValue),\n\t\t\texpected: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is transfer it should return the correctly mapped value\",\n\t\t\tinput:    adminv1.TaskType_TASK_TYPE_TRANSFER,\n\t\t\texpected: common.Int32Ptr(int32(constants.TaskTypeTransfer)),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, ToTaskType(tc.input))\n\t\t})\n\t}\n}\n\nfunc TestFromTaskType(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tinput    *int32\n\t\texpected adminv1.TaskType\n\t}{\n\t\t{\n\t\t\tname:     \"when input is nil it should return INVALID\",\n\t\t\tinput:    nil,\n\t\t\texpected: adminv1.TaskType_TASK_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is out of range it should return INVALID\",\n\t\t\tinput:    common.Int32Ptr(UnknownValue),\n\t\t\texpected: adminv1.TaskType_TASK_TYPE_INVALID,\n\t\t},\n\t\t{\n\t\t\tname:     \"when input is transfer it should return the correctly mapped value\",\n\t\t\tinput:    common.Int32Ptr(int32(constants.TaskTypeTransfer)),\n\t\t\texpected: adminv1.TaskType_TASK_TYPE_TRANSFER,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.expected, FromTaskType(tc.input))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/errors.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc/encoding/protobuf\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/errorutils\"\n)\n\nfunc FromError(err error) error {\n\tif err == nil {\n\t\treturn protobuf.NewError(yarpcerrors.CodeOK, \"\")\n\t}\n\n\tvar (\n\t\tok       bool\n\t\ttypedErr error\n\t)\n\tif ok, typedErr = errorutils.ConvertError(err, fromAccessDeniedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromInternalServiceError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromEntityNotExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromWorkflowExecutionAlreadyCompletedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromBadRequestError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromQueryFailedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromShardOwnershipLostError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromTaskListNotOwnedByHostError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromCurrentBranchChangedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromRetryTaskV2Error); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromCancellationAlreadyRequestedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromDomainAlreadyExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromEventAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromWorkflowExecutionAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromClientVersionNotSupportedErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromFeatureNotEnabledErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromDomainNotActive); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromInternalDataInconsistencyErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromLimitExceededErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromServiceBusyErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromRemoteSyncMatchedErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromStickyWorkerUnavailableErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromReadOnlyPartitionErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromNamespaceNotFoundErr); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, fromShardNotFoundErr); ok {\n\t\treturn typedErr\n\t}\n\n\treturn protobuf.NewError(yarpcerrors.CodeUnknown, err.Error())\n}\n\nfunc ToError(err error) error {\n\tstatus := yarpcerrors.FromError(err)\n\tif status == nil || status.Code() == yarpcerrors.CodeOK {\n\t\treturn nil\n\t}\n\n\tswitch status.Code() {\n\tcase yarpcerrors.CodePermissionDenied:\n\t\treturn &types.AccessDeniedError{\n\t\t\tMessage: status.Message(),\n\t\t}\n\tcase yarpcerrors.CodeInternal:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: status.Message(),\n\t\t}\n\tcase yarpcerrors.CodeNotFound:\n\t\tswitch details := getErrorDetails(err).(type) {\n\t\tcase *apiv1.EntityNotExistsError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.EntityNotExistsError{\n\t\t\t\t\tMessage:        status.Message(),\n\t\t\t\t\tCurrentCluster: details.CurrentCluster,\n\t\t\t\t\tActiveCluster:  details.ActiveCluster,\n\t\t\t\t\tActiveClusters: details.ActiveClusters,\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn &types.EntityNotExistsError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn &types.WorkflowExecutionAlreadyCompletedError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *sharddistributorv1.NamespaceNotFoundError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.NamespaceNotFoundError{\n\t\t\t\t\tNamespace: details.Namespace,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *sharddistributorv1.ShardNotFoundError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.ShardNotFoundError{\n\t\t\t\t\tNamespace: details.Namespace,\n\t\t\t\t\tShardKey:  details.ShardKey,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeInvalidArgument:\n\t\tswitch getErrorDetails(err).(type) {\n\t\tcase nil:\n\t\t\treturn &types.BadRequestError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.QueryFailedError:\n\t\t\treturn &types.QueryFailedError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeAborted:\n\t\tswitch details := getErrorDetails(err).(type) {\n\t\tcase *sharedv1.ShardOwnershipLostError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.ShardOwnershipLostError{\n\t\t\t\t\tMessage: status.Message(),\n\t\t\t\t\tOwner:   details.Owner,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *sharedv1.TaskListNotOwnedByHostError:\n\t\t\tif details != nil {\n\t\t\t\treturn &cadence_errors.TaskListNotOwnedByHostError{\n\t\t\t\t\tOwnedByIdentity: details.OwnedByIdentity,\n\t\t\t\t\tMyIdentity:      details.MyIdentity,\n\t\t\t\t\tTasklistName:    details.TaskListName,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *sharedv1.CurrentBranchChangedError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.CurrentBranchChangedError{\n\t\t\t\t\tMessage:            status.Message(),\n\t\t\t\t\tCurrentBranchToken: details.CurrentBranchToken,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *sharedv1.RetryTaskV2Error:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.RetryTaskV2Error{\n\t\t\t\t\tMessage:           status.Message(),\n\t\t\t\t\tDomainID:          details.DomainId,\n\t\t\t\t\tWorkflowID:        ToWorkflowID(details.WorkflowExecution),\n\t\t\t\t\tRunID:             ToRunID(details.WorkflowExecution),\n\t\t\t\t\tStartEventID:      ToEventID(details.StartEvent),\n\t\t\t\t\tStartEventVersion: ToEventVersion(details.StartEvent),\n\t\t\t\t\tEndEventID:        ToEventID(details.EndEvent),\n\t\t\t\t\tEndEventVersion:   ToEventVersion(details.EndEvent),\n\t\t\t\t}\n\t\t\t}\n\t\tcase *apiv1.ReadOnlyPartitionError:\n\t\t\treturn &types.ReadOnlyPartitionError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeAlreadyExists:\n\t\tswitch details := getErrorDetails(err).(type) {\n\t\tcase *apiv1.CancellationAlreadyRequestedError:\n\t\t\treturn &types.CancellationAlreadyRequestedError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.DomainAlreadyExistsError:\n\t\t\treturn &types.DomainAlreadyExistsError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *sharedv1.EventAlreadyStartedError:\n\t\t\treturn &types.EventAlreadyStartedError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.WorkflowExecutionAlreadyStartedError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\tMessage:        status.Message(),\n\t\t\t\t\tStartRequestID: details.StartRequestId,\n\t\t\t\t\tRunID:          details.RunId,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeDataLoss:\n\t\treturn &types.InternalDataInconsistencyError{\n\t\t\tMessage: status.Message(),\n\t\t}\n\tcase yarpcerrors.CodeFailedPrecondition:\n\t\tswitch details := getErrorDetails(err).(type) {\n\t\tcase *apiv1.ClientVersionNotSupportedError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.ClientVersionNotSupportedError{\n\t\t\t\t\tFeatureVersion:    details.FeatureVersion,\n\t\t\t\t\tClientImpl:        details.ClientImpl,\n\t\t\t\t\tSupportedVersions: details.SupportedVersions,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *apiv1.FeatureNotEnabledError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.FeatureNotEnabledError{\n\t\t\t\t\tFeatureFlag: details.FeatureFlag,\n\t\t\t\t}\n\t\t\t}\n\t\tcase *apiv1.DomainNotActiveError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.DomainNotActiveError{\n\t\t\t\t\tMessage:        status.Message(),\n\t\t\t\t\tDomainName:     details.Domain,\n\t\t\t\t\tCurrentCluster: details.CurrentCluster,\n\t\t\t\t\tActiveCluster:  details.ActiveCluster,\n\t\t\t\t\tActiveClusters: details.ActiveClusters,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeResourceExhausted:\n\t\tswitch details := getErrorDetails(err).(type) {\n\t\tcase *apiv1.LimitExceededError:\n\t\t\treturn &types.LimitExceededError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.ServiceBusyError:\n\t\t\tif details != nil {\n\t\t\t\treturn &types.ServiceBusyError{\n\t\t\t\t\tMessage: status.Message(),\n\t\t\t\t\tReason:  details.Reason,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\tcase yarpcerrors.CodeUnavailable:\n\t\tswitch getErrorDetails(err).(type) {\n\t\tcase *sharedv1.RemoteSyncMatchedError:\n\t\t\treturn &types.RemoteSyncMatchedError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\tcase *apiv1.StickyWorkerUnavailableError:\n\t\t\treturn &types.StickyWorkerUnavailableError{\n\t\t\t\tMessage: status.Message(),\n\t\t\t}\n\t\t}\n\t}\n\n\t// If error does not match anything, return raw yarpc status error\n\t// There are some code that casts error to yarpc status to check for deadline exceeded status\n\treturn status\n}\n\nfunc getErrorDetails(err error) interface{} {\n\tdetails := protobuf.GetErrorDetails(err)\n\tif len(details) > 0 {\n\t\treturn details[0]\n\t}\n\treturn nil\n}\n\nfunc fromAccessDeniedError(e *types.AccessDeniedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodePermissionDenied, e.Message)\n}\n\nfunc fromInternalServiceError(e *types.InternalServiceError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeInternal, e.Message)\n}\n\nfunc fromEntityNotExistsError(e *types.EntityNotExistsError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeNotFound, e.Message, protobuf.WithErrorDetails(&apiv1.EntityNotExistsError{\n\t\tCurrentCluster: e.CurrentCluster,\n\t\tActiveCluster:  e.ActiveCluster,\n\t\tActiveClusters: e.ActiveClusters,\n\t}))\n}\n\nfunc fromWorkflowExecutionAlreadyCompletedError(e *types.WorkflowExecutionAlreadyCompletedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeNotFound, e.Message, protobuf.WithErrorDetails(&apiv1.WorkflowExecutionAlreadyCompletedError{}))\n}\n\nfunc fromBadRequestError(e *types.BadRequestError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeInvalidArgument, e.Message)\n}\n\nfunc fromQueryFailedError(e *types.QueryFailedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeInvalidArgument, e.Message, protobuf.WithErrorDetails(&apiv1.QueryFailedError{}))\n}\n\nfunc fromShardOwnershipLostError(e *types.ShardOwnershipLostError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAborted, e.Message, protobuf.WithErrorDetails(&sharedv1.ShardOwnershipLostError{\n\t\tOwner: e.Owner,\n\t}))\n}\n\nfunc fromTaskListNotOwnedByHostError(e *cadence_errors.TaskListNotOwnedByHostError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAborted, e.Error(), protobuf.WithErrorDetails(&sharedv1.TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: e.OwnedByIdentity,\n\t\tMyIdentity:      e.MyIdentity,\n\t\tTaskListName:    e.TasklistName,\n\t}))\n}\n\nfunc fromCurrentBranchChangedError(e *types.CurrentBranchChangedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAborted, e.Message, protobuf.WithErrorDetails(&sharedv1.CurrentBranchChangedError{\n\t\tCurrentBranchToken: e.GetCurrentBranchToken(),\n\t}))\n}\n\nfunc fromRetryTaskV2Error(e *types.RetryTaskV2Error) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAborted, e.Message, protobuf.WithErrorDetails(&sharedv1.RetryTaskV2Error{\n\t\tDomainId:          e.DomainID,\n\t\tWorkflowExecution: FromWorkflowRunPair(e.WorkflowID, e.RunID),\n\t\tStartEvent:        FromEventIDVersionPair(e.StartEventID, e.StartEventVersion),\n\t\tEndEvent:          FromEventIDVersionPair(e.EndEventID, e.EndEventVersion),\n\t}))\n}\n\nfunc fromCancellationAlreadyRequestedError(e *types.CancellationAlreadyRequestedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAlreadyExists, e.Message, protobuf.WithErrorDetails(&apiv1.CancellationAlreadyRequestedError{}))\n}\n\nfunc fromDomainAlreadyExistsError(e *types.DomainAlreadyExistsError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAlreadyExists, e.Message, protobuf.WithErrorDetails(&apiv1.DomainAlreadyExistsError{}))\n}\n\nfunc fromEventAlreadyStartedError(e *types.EventAlreadyStartedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAlreadyExists, e.Message, protobuf.WithErrorDetails(&sharedv1.EventAlreadyStartedError{}))\n}\n\nfunc fromWorkflowExecutionAlreadyStartedError(e *types.WorkflowExecutionAlreadyStartedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAlreadyExists, e.Message, protobuf.WithErrorDetails(&apiv1.WorkflowExecutionAlreadyStartedError{\n\t\tStartRequestId: e.StartRequestID,\n\t\tRunId:          e.RunID,\n\t}))\n}\n\nfunc fromClientVersionNotSupportedErr(e *types.ClientVersionNotSupportedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeFailedPrecondition, \"Client version not supported\", protobuf.WithErrorDetails(&apiv1.ClientVersionNotSupportedError{\n\t\tFeatureVersion:    e.FeatureVersion,\n\t\tClientImpl:        e.ClientImpl,\n\t\tSupportedVersions: e.SupportedVersions,\n\t}))\n}\n\nfunc fromFeatureNotEnabledErr(e *types.FeatureNotEnabledError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeFailedPrecondition, \"Feature flag not enabled\", protobuf.WithErrorDetails(&apiv1.FeatureNotEnabledError{\n\t\tFeatureFlag: e.FeatureFlag,\n\t}))\n}\n\nfunc fromDomainNotActive(e *types.DomainNotActiveError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeFailedPrecondition, e.Message, protobuf.WithErrorDetails(&apiv1.DomainNotActiveError{\n\t\tDomain:         e.DomainName,\n\t\tCurrentCluster: e.CurrentCluster,\n\t\tActiveCluster:  e.ActiveCluster,\n\t\tActiveClusters: e.ActiveClusters,\n\t}))\n}\n\nfunc fromInternalDataInconsistencyErr(e *types.InternalDataInconsistencyError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeDataLoss, e.Message, protobuf.WithErrorDetails(&sharedv1.InternalDataInconsistencyError{}))\n}\n\nfunc fromLimitExceededErr(e *types.LimitExceededError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeResourceExhausted, e.Message, protobuf.WithErrorDetails(&apiv1.LimitExceededError{}))\n}\n\nfunc fromServiceBusyErr(e *types.ServiceBusyError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeResourceExhausted, e.Message, protobuf.WithErrorDetails(&apiv1.ServiceBusyError{\n\t\tReason: e.Reason,\n\t}))\n}\n\nfunc fromRemoteSyncMatchedErr(e *types.RemoteSyncMatchedError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeUnavailable, e.Message, protobuf.WithErrorDetails(&sharedv1.RemoteSyncMatchedError{}))\n}\n\nfunc fromStickyWorkerUnavailableErr(e *types.StickyWorkerUnavailableError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeUnavailable, e.Message, protobuf.WithErrorDetails(&apiv1.StickyWorkerUnavailableError{}))\n}\n\nfunc fromReadOnlyPartitionErr(e *types.ReadOnlyPartitionError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeAborted, e.Message, protobuf.WithErrorDetails(&apiv1.ReadOnlyPartitionError{}))\n}\n\nfunc fromNamespaceNotFoundErr(e *types.NamespaceNotFoundError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeNotFound, e.Error(), protobuf.WithErrorDetails(&sharddistributorv1.NamespaceNotFoundError{\n\t\tNamespace: e.Namespace,\n\t}))\n}\n\nfunc fromShardNotFoundErr(e *types.ShardNotFoundError) error {\n\treturn protobuf.NewError(yarpcerrors.CodeNotFound, e.Error(), protobuf.WithErrorDetails(&sharddistributorv1.ShardNotFoundError{\n\t\tNamespace: e.Namespace,\n\t\tShardKey:  e.ShardKey,\n\t}))\n}\n"
  },
  {
    "path": "common/types/mapper/proto/errors_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"errors\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestErrors(t *testing.T) {\n\tfor _, err := range testdata.Errors {\n\t\tname := reflect.TypeOf(err).Elem().Name()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Test that the mappings does not lose information\n\t\t\tassert.Equal(t, err, ToError(FromError(err)))\n\t\t})\n\t}\n}\n\nfunc TestNilMapsToOK(t *testing.T) {\n\tprotoNoError := FromError(nil)\n\tassert.Equal(t, yarpcerrors.CodeOK, yarpcerrors.FromError(protoNoError).Code())\n\tassert.Nil(t, ToError(protoNoError))\n}\n\nfunc TestFromUnknownErrorMapsToUnknownError(t *testing.T) {\n\terr := errors.New(\"unknown error\")\n\tprotobufErr := FromError(err)\n\tassert.True(t, yarpcerrors.IsUnknown(protobufErr))\n\n\tclientErr := ToError(protobufErr)\n\n\tassert.True(t, yarpcerrors.IsUnknown(clientErr))\n\tassert.ErrorContains(t, clientErr, err.Error())\n}\n\nfunc TestToDeadlineExceededMapsToItself(t *testing.T) {\n\ttimeout := yarpcerrors.DeadlineExceededErrorf(\"timeout\")\n\tassert.Equal(t, timeout, ToError(timeout))\n}\n"
  },
  {
    "path": "common/types/mapper/proto/helpers.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"strconv\"\n\t\"time\"\n\n\tgogo \"github.com/gogo/protobuf/types\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\nfunc fromDoubleValue(v *float64) *gogo.DoubleValue {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn &gogo.DoubleValue{Value: *v}\n}\n\nfunc toDoubleValue(v *gogo.DoubleValue) *float64 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn common.Float64Ptr(v.Value)\n}\n\nfunc fromInt64Value(v *int64) *gogo.Int64Value {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn &gogo.Int64Value{Value: *v}\n}\n\nfunc toInt64Value(v *gogo.Int64Value) *int64 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn common.Int64Ptr(v.Value)\n}\n\nfunc unixNanoToTime(t *int64) *gogo.Timestamp {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttime, err := gogo.TimestampProto(time.Unix(0, *t))\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn time\n}\n\nfunc timeToUnixNano(t *gogo.Timestamp) *int64 {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttimestamp, err := gogo.TimestampFromProto(t)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn common.Int64Ptr(timestamp.UnixNano())\n}\n\nfunc timeToTimestamp(t *time.Time) *gogo.Timestamp {\n\tif t == nil || t.IsZero() {\n\t\treturn nil\n\t}\n\n\tnanos := t.UnixNano()\n\treturn unixNanoToTime(&nanos)\n}\n\nfunc timestampToTime(t *gogo.Timestamp) *time.Time {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\ttime := time.Unix(t.GetSeconds(), int64(t.GetNanos()))\n\treturn &time\n}\n\nfunc timestampToTimeVal(t *gogo.Timestamp) time.Time {\n\tif t == nil {\n\t\treturn time.Time{}\n\t}\n\treturn time.Unix(t.GetSeconds(), int64(t.GetNanos())).UTC()\n}\n\nfunc durationToDurationProto(d time.Duration) *gogo.Duration {\n\tif d == 0 {\n\t\treturn nil\n\t}\n\treturn gogo.DurationProto(d)\n}\n\nfunc durationProtoToDuration(d *gogo.Duration) time.Duration {\n\tif d == nil {\n\t\treturn 0\n\t}\n\tdur, err := gogo.DurationFromProto(d)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn dur\n}\n\nfunc daysToDuration(d *int32) *gogo.Duration {\n\tif d == nil {\n\t\treturn nil\n\t}\n\treturn gogo.DurationProto(common.DaysToDuration(*d))\n}\n\nfunc durationToDays(d *gogo.Duration) *int32 {\n\tif d == nil {\n\t\treturn nil\n\t}\n\tduration, err := gogo.DurationFromProto(d)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn common.Int32Ptr(common.DurationToDays(duration))\n}\n\nfunc secondsToDuration(d *int32) *gogo.Duration {\n\tif d == nil {\n\t\treturn nil\n\t}\n\treturn gogo.DurationProto(common.SecondsToDuration(int64(*d)))\n}\n\nfunc durationToSeconds(d *gogo.Duration) *int32 {\n\tif d == nil {\n\t\treturn nil\n\t}\n\tduration, err := gogo.DurationFromProto(d)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn common.Int32Ptr(int32(common.DurationToSeconds(duration)))\n}\n\nfunc int32To64(v *int32) *int64 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn common.Int64Ptr(int64(*v))\n}\n\nfunc int64To32(v *int64) *int32 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn common.Int32Ptr(int32(*v))\n}\n\nfunc stringToInt32(s string) int32 {\n\ti, err := strconv.Atoi(s)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn int32(i)\n}\n\nfunc int32ToString(i int32) string {\n\ts := strconv.Itoa(int(i))\n\treturn s\n}\n\ntype fieldSet map[string]struct{}\n\nfunc newFieldSet(mask *gogo.FieldMask) fieldSet {\n\tif mask == nil {\n\t\treturn nil\n\t}\n\tfs := map[string]struct{}{}\n\tfor _, field := range mask.Paths {\n\t\tfs[field] = struct{}{}\n\t}\n\treturn fs\n}\n\nfunc (fs fieldSet) isSet(field string) bool {\n\tif fs == nil {\n\t\treturn true\n\t}\n\t_, ok := fs[field]\n\treturn ok\n}\n\nfunc newFieldMask(fields []string) *gogo.FieldMask {\n\treturn &gogo.FieldMask{Paths: fields}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/helpers_test.go",
    "content": "// Copyright (c) 2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tgogo \"github.com/gogo/protobuf/types\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestTimeToTimestamp(t *testing.T) {\n\ttestTime := time.Unix(10, 10)\n\ttimestamp := timeToTimestamp(&testTime)\n\tassert.Equal(t, gogo.Timestamp{Seconds: 10, Nanos: 10}, *timestamp)\n}\n\nfunc TestTimeToTimestampNil(t *testing.T) {\n\tresult := timeToTimestamp(nil)\n\tassert.Nil(t, result)\n}\n"
  },
  {
    "path": "common/types/mapper/proto/history.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromHistoryCloseShardRequest(t *types.CloseShardRequest) *historyv1.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.CloseShardRequest{\n\t\tShardId: t.ShardID,\n\t}\n}\n\nfunc ToHistoryCloseShardRequest(t *historyv1.CloseShardRequest) *types.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CloseShardRequest{\n\t\tShardID: t.ShardId,\n\t}\n}\n\nfunc FromHistoryDescribeHistoryHostRequest(t *types.DescribeHistoryHostRequest) *historyv1.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeHistoryHostRequest{}\n}\n\nfunc ToHistoryDescribeHistoryHostRequest(t *historyv1.DescribeHistoryHostRequest) *types.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeHistoryHostRequest{}\n}\n\nfunc FromHistoryDescribeHistoryHostResponse(t *types.DescribeHistoryHostResponse) *historyv1.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        t.NumberOfShards,\n\t\tShardIds:              t.ShardIDs,\n\t\tDomainCache:           FromDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: t.ShardControllerStatus,\n\t\tAddress:               t.Address,\n\t}\n}\n\nfunc ToHistoryDescribeHistoryHostResponse(t *historyv1.DescribeHistoryHostResponse) *types.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        t.NumberOfShards,\n\t\tShardIDs:              t.ShardIds,\n\t\tDomainCache:           ToDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: t.ShardControllerStatus,\n\t\tAddress:               t.Address,\n\t}\n}\n\nfunc FromHistoryDescribeMutableStateRequest(t *types.DescribeMutableStateRequest) *historyv1.DescribeMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeMutableStateRequest{\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\nfunc ToHistoryDescribeMutableStateRequest(t *historyv1.DescribeMutableStateRequest) *types.DescribeMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeMutableStateRequest{\n\t\tDomainUUID: t.DomainId,\n\t\tExecution:  ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\nfunc FromHistoryDescribeMutableStateResponse(t *types.DescribeMutableStateResponse) *historyv1.DescribeMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeMutableStateResponse{\n\t\tMutableStateInCache:    t.MutableStateInCache,\n\t\tMutableStateInDatabase: t.MutableStateInDatabase,\n\t}\n}\n\nfunc ToHistoryDescribeMutableStateResponse(t *historyv1.DescribeMutableStateResponse) *types.DescribeMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeMutableStateResponse{\n\t\tMutableStateInCache:    t.MutableStateInCache,\n\t\tMutableStateInDatabase: t.MutableStateInDatabase,\n\t}\n}\n\nfunc FromHistoryDescribeQueueRequest(t *types.DescribeQueueRequest) *historyv1.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeQueueRequest{\n\t\tShardId:     t.ShardID,\n\t\tClusterName: t.ClusterName,\n\t\tTaskType:    FromTaskType(t.Type),\n\t}\n}\n\nfunc ToHistoryDescribeQueueRequest(t *historyv1.DescribeQueueRequest) *types.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueRequest{\n\t\tShardID:     t.ShardId,\n\t\tClusterName: t.ClusterName,\n\t\tType:        ToTaskType(t.TaskType),\n\t}\n}\n\nfunc FromHistoryDescribeQueueResponse(t *types.DescribeQueueResponse) *historyv1.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\nfunc ToHistoryDescribeQueueResponse(t *historyv1.DescribeQueueResponse) *types.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\nfunc FromHistoryDescribeWorkflowExecutionRequest(t *types.HistoryDescribeWorkflowExecutionRequest) *historyv1.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeWorkflowExecutionRequest{\n\t\tRequest:  FromDescribeWorkflowExecutionRequest(t.Request),\n\t\tDomainId: t.DomainUUID,\n\t}\n}\n\nfunc ToHistoryDescribeWorkflowExecutionRequest(t *historyv1.DescribeWorkflowExecutionRequest) *types.HistoryDescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryDescribeWorkflowExecutionRequest{\n\t\tRequest:    ToDescribeWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID: t.DomainId,\n\t}\n}\n\nfunc FromHistoryDescribeWorkflowExecutionResponse(t *types.DescribeWorkflowExecutionResponse) *historyv1.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: FromWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  FromWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      FromPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        FromPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        FromPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\nfunc ToHistoryDescribeWorkflowExecutionResponse(t *historyv1.DescribeWorkflowExecutionResponse) *types.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: ToWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  ToWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      ToPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        ToPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        ToPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\nfunc FromHistoryGetDLQReplicationMessagesRequest(t *types.GetDLQReplicationMessagesRequest) *historyv1.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: FromReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\nfunc ToHistoryGetDLQReplicationMessagesRequest(t *historyv1.GetDLQReplicationMessagesRequest) *types.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: ToReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\nfunc FromHistoryGetDLQReplicationMessagesResponse(t *types.GetDLQReplicationMessagesResponse) *historyv1.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: FromReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\nfunc ToHistoryGetDLQReplicationMessagesResponse(t *historyv1.GetDLQReplicationMessagesResponse) *types.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: ToReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\nfunc FromHistoryGetFailoverInfoRequest(t *types.GetFailoverInfoRequest) *historyv1.GetFailoverInfoRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetFailoverInfoRequest{\n\t\tDomainId: t.GetDomainID(),\n\t}\n}\n\nfunc ToHistoryGetFailoverInfoRequest(t *historyv1.GetFailoverInfoRequest) *types.GetFailoverInfoRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetFailoverInfoRequest{\n\t\tDomainID: t.GetDomainId(),\n\t}\n}\n\nfunc FromHistoryGetFailoverInfoResponse(t *types.GetFailoverInfoResponse) *historyv1.GetFailoverInfoResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetFailoverInfoResponse{\n\t\tCompletedShardCount: t.GetCompletedShardCount(),\n\t\tPendingShards:       t.GetPendingShards(),\n\t}\n}\n\nfunc ToHistoryGetFailoverInfoResponse(t *historyv1.GetFailoverInfoResponse) *types.GetFailoverInfoResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetFailoverInfoResponse{\n\t\tCompletedShardCount: t.GetCompletedShardCount(),\n\t\tPendingShards:       t.GetPendingShards(),\n\t}\n}\n\nfunc FromHistoryGetMutableStateRequest(t *types.GetMutableStateRequest) *historyv1.GetMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetMutableStateRequest{\n\t\tDomainId:            t.DomainUUID,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.Execution),\n\t\tExpectedNextEventId: t.ExpectedNextEventID,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t\tVersionHistoryItem:  FromVersionHistoryItem(t.VersionHistoryItem),\n\t}\n}\n\nfunc ToHistoryGetMutableStateRequest(t *historyv1.GetMutableStateRequest) *types.GetMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetMutableStateRequest{\n\t\tDomainUUID:          t.DomainId,\n\t\tExecution:           ToWorkflowExecution(t.WorkflowExecution),\n\t\tExpectedNextEventID: t.ExpectedNextEventId,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t\tVersionHistoryItem:  ToVersionHistoryItem(t.VersionHistoryItem),\n\t}\n}\n\nfunc FromHistoryGetMutableStateResponse(t *types.GetMutableStateResponse) *historyv1.GetMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar workflowCloseState *types.WorkflowExecutionCloseStatus\n\tif t.WorkflowCloseState != nil {\n\t\tworkflowCloseState = persistence.ToInternalWorkflowExecutionCloseStatus(int(*t.WorkflowCloseState))\n\t}\n\treturn &historyv1.GetMutableStateResponse{\n\t\tWorkflowExecution:                    FromWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         FromWorkflowType(t.WorkflowType),\n\t\tNextEventId:                          t.NextEventID,\n\t\tPreviousStartedEventId:               fromInt64Value(t.PreviousStartedEventID),\n\t\tLastFirstEventId:                     t.LastFirstEventID,\n\t\tTaskList:                             FromTaskList(t.TaskList),\n\t\tStickyTaskList:                       FromTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 t.ClientFeatureVersion,\n\t\tClientImpl:                           t.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: secondsToDuration(t.StickyTaskListScheduleToStartTimeout),\n\t\tEventStoreVersion:                    t.EventStoreVersion,\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tWorkflowState:                        FromWorkflowState(t.WorkflowState),\n\t\tWorkflowCloseState:                   FromWorkflowExecutionCloseStatus(workflowCloseState),\n\t\tVersionHistories:                     FromVersionHistories(t.VersionHistories),\n\t\tIsStickyTaskListEnabled:              t.IsStickyTaskListEnabled,\n\t\tHistorySize:                          t.HistorySize,\n\t}\n}\n\nfunc ToHistoryGetMutableStateResponse(t *historyv1.GetMutableStateResponse) *types.GetMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetMutableStateResponse{\n\t\tExecution:                            ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:                         ToWorkflowType(t.WorkflowType),\n\t\tNextEventID:                          t.NextEventId,\n\t\tPreviousStartedEventID:               toInt64Value(t.PreviousStartedEventId),\n\t\tLastFirstEventID:                     t.LastFirstEventId,\n\t\tTaskList:                             ToTaskList(t.TaskList),\n\t\tStickyTaskList:                       ToTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 t.ClientFeatureVersion,\n\t\tClientImpl:                           t.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: durationToSeconds(t.StickyTaskListScheduleToStartTimeout),\n\t\tEventStoreVersion:                    t.EventStoreVersion,\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tWorkflowState:                        ToWorkflowState(t.WorkflowState),\n\t\tWorkflowCloseState:                   common.Int32Ptr(int32(persistence.FromInternalWorkflowExecutionCloseStatus(ToWorkflowExecutionCloseStatus(t.WorkflowCloseState)))),\n\t\tVersionHistories:                     ToVersionHistories(t.VersionHistories),\n\t\tIsStickyTaskListEnabled:              t.IsStickyTaskListEnabled,\n\t\tIsWorkflowRunning:                    t.WorkflowState == sharedv1.WorkflowState_WORKFLOW_STATE_RUNNING,\n\t\tHistorySize:                          t.HistorySize,\n\t}\n}\n\nfunc FromHistoryGetReplicationMessagesRequest(t *types.GetReplicationMessagesRequest) *historyv1.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetReplicationMessagesRequest{\n\t\tTokens:      FromReplicationTokenArray(t.Tokens),\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc ToHistoryGetReplicationMessagesRequest(t *historyv1.GetReplicationMessagesRequest) *types.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesRequest{\n\t\tTokens:      ToReplicationTokenArray(t.Tokens),\n\t\tClusterName: t.ClusterName,\n\t}\n}\n\nfunc FromHistoryGetReplicationMessagesResponse(t *types.GetReplicationMessagesResponse) *historyv1.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetReplicationMessagesResponse{\n\t\tShardMessages: FromReplicationMessagesMap(t.MessagesByShard),\n\t}\n}\n\nfunc ToHistoryGetReplicationMessagesResponse(t *historyv1.GetReplicationMessagesResponse) *types.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: ToReplicationMessagesMap(t.ShardMessages),\n\t}\n}\n\nfunc FromHistoryCountDLQMessagesRequest(t *types.CountDLQMessagesRequest) *historyv1.CountDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.CountDLQMessagesRequest{\n\t\tForceFetch: t.ForceFetch,\n\t}\n}\n\nfunc ToHistoryCountDLQMessagesRequest(t *historyv1.CountDLQMessagesRequest) *types.CountDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountDLQMessagesRequest{\n\t\tForceFetch: t.ForceFetch,\n\t}\n}\n\nfunc FromHistoryCountDLQMessagesResponse(t *types.HistoryCountDLQMessagesResponse) *historyv1.CountDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.CountDLQMessagesResponse{\n\t\tEntries: FromHistoryDLQCountEntryMap(t.Entries),\n\t}\n}\n\nfunc ToHistoryCountDLQMessagesResponse(t *historyv1.CountDLQMessagesResponse) *types.HistoryCountDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryCountDLQMessagesResponse{\n\t\tEntries: ToHistoryDLQCountEntryMap(t.Entries),\n\t}\n}\n\nfunc FromHistoryMergeDLQMessagesRequest(t *types.MergeDLQMessagesRequest) *historyv1.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.MergeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t\tPageSize:              t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc ToHistoryMergeDLQMessagesRequest(t *historyv1.MergeDLQMessagesRequest) *types.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t\tMaximumPageSize:       t.PageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc FromHistoryMergeDLQMessagesResponse(t *types.MergeDLQMessagesResponse) *historyv1.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToHistoryMergeDLQMessagesResponse(t *historyv1.MergeDLQMessagesResponse) *types.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromHistoryNotifyFailoverMarkersRequest(t *types.NotifyFailoverMarkersRequest) *historyv1.NotifyFailoverMarkersRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: FromFailoverMarkerTokenArray(t.FailoverMarkerTokens),\n\t}\n}\n\nfunc ToHistoryNotifyFailoverMarkersRequest(t *historyv1.NotifyFailoverMarkersRequest) *types.NotifyFailoverMarkersRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: ToFailoverMarkerTokenArray(t.FailoverMarkerTokens),\n\t}\n}\n\nfunc FromHistoryPollMutableStateRequest(t *types.PollMutableStateRequest) *historyv1.PollMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.PollMutableStateRequest{\n\t\tDomainId:            t.DomainUUID,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.Execution),\n\t\tExpectedNextEventId: t.ExpectedNextEventID,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t}\n}\n\nfunc ToHistoryPollMutableStateRequest(t *historyv1.PollMutableStateRequest) *types.PollMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollMutableStateRequest{\n\t\tDomainUUID:          t.DomainId,\n\t\tExecution:           ToWorkflowExecution(t.WorkflowExecution),\n\t\tExpectedNextEventID: t.ExpectedNextEventId,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t}\n}\n\nfunc FromHistoryPollMutableStateResponse(t *types.PollMutableStateResponse) *historyv1.PollMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar workflowCloseState *types.WorkflowExecutionCloseStatus\n\tif t.WorkflowCloseState != nil {\n\t\tworkflowCloseState = persistence.ToInternalWorkflowExecutionCloseStatus(int(*t.WorkflowCloseState))\n\t}\n\treturn &historyv1.PollMutableStateResponse{\n\t\tWorkflowExecution:                    FromWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         FromWorkflowType(t.WorkflowType),\n\t\tNextEventId:                          t.NextEventID,\n\t\tPreviousStartedEventId:               fromInt64Value(t.PreviousStartedEventID),\n\t\tLastFirstEventId:                     t.LastFirstEventID,\n\t\tTaskList:                             FromTaskList(t.TaskList),\n\t\tStickyTaskList:                       FromTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 t.ClientFeatureVersion,\n\t\tClientImpl:                           t.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: secondsToDuration(t.StickyTaskListScheduleToStartTimeout),\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tVersionHistories:                     FromVersionHistories(t.VersionHistories),\n\t\tWorkflowState:                        FromWorkflowState(t.WorkflowState),\n\t\tWorkflowCloseState:                   FromWorkflowExecutionCloseStatus(workflowCloseState),\n\t}\n}\n\nfunc ToHistoryPollMutableStateResponse(t *historyv1.PollMutableStateResponse) *types.PollMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.PollMutableStateResponse{\n\t\tExecution:                            ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:                         ToWorkflowType(t.WorkflowType),\n\t\tNextEventID:                          t.NextEventId,\n\t\tPreviousStartedEventID:               toInt64Value(t.PreviousStartedEventId),\n\t\tLastFirstEventID:                     t.LastFirstEventId,\n\t\tTaskList:                             ToTaskList(t.TaskList),\n\t\tStickyTaskList:                       ToTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 t.ClientFeatureVersion,\n\t\tClientImpl:                           t.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: durationToSeconds(t.StickyTaskListScheduleToStartTimeout),\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tVersionHistories:                     ToVersionHistories(t.VersionHistories),\n\t\tWorkflowState:                        ToWorkflowState(t.WorkflowState),\n\t\tWorkflowCloseState:                   common.Int32Ptr(int32(persistence.FromInternalWorkflowExecutionCloseStatus(ToWorkflowExecutionCloseStatus(t.WorkflowCloseState)))),\n\t}\n}\n\nfunc FromHistoryPurgeDLQMessagesRequest(t *types.PurgeDLQMessagesRequest) *historyv1.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.PurgeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t}\n}\n\nfunc ToHistoryPurgeDLQMessagesRequest(t *historyv1.PurgeDLQMessagesRequest) *types.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PurgeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t}\n}\n\nfunc FromHistoryQueryWorkflowRequest(t *types.HistoryQueryWorkflowRequest) *historyv1.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.QueryWorkflowRequest{\n\t\tRequest:  FromQueryWorkflowRequest(t.Request),\n\t\tDomainId: t.DomainUUID,\n\t}\n}\n\nfunc ToHistoryQueryWorkflowRequest(t *historyv1.QueryWorkflowRequest) *types.HistoryQueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryQueryWorkflowRequest{\n\t\tRequest:    ToQueryWorkflowRequest(t.Request),\n\t\tDomainUUID: t.DomainId,\n\t}\n}\n\nfunc FromHistoryQueryWorkflowResponse(t *types.HistoryQueryWorkflowResponse) *historyv1.QueryWorkflowResponse {\n\tif t == nil || t.Response == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.QueryWorkflowResponse{\n\t\tQueryResult:   FromPayload(t.Response.QueryResult),\n\t\tQueryRejected: FromQueryRejected(t.Response.QueryRejected),\n\t}\n}\n\nfunc ToHistoryQueryWorkflowResponse(t *historyv1.QueryWorkflowResponse) *types.HistoryQueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryQueryWorkflowResponse{\n\t\tResponse: &types.QueryWorkflowResponse{\n\t\t\tQueryResult:   ToPayload(t.QueryResult),\n\t\t\tQueryRejected: ToQueryRejected(t.QueryRejected),\n\t\t},\n\t}\n}\n\nfunc FromHistoryReadDLQMessagesRequest(t *types.ReadDLQMessagesRequest) *historyv1.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ReadDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardId:               t.ShardID,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageId: fromInt64Value(t.InclusiveEndMessageID),\n\t\tPageSize:              t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc ToHistoryReadDLQMessagesRequest(t *historyv1.ReadDLQMessagesRequest) *types.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.ShardId,\n\t\tSourceCluster:         t.SourceCluster,\n\t\tInclusiveEndMessageID: toInt64Value(t.InclusiveEndMessageId),\n\t\tMaximumPageSize:       t.PageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\nfunc FromHistoryReadDLQMessagesResponse(t *types.ReadDLQMessagesResponse) *historyv1.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ReadDLQMessagesResponse{\n\t\tType:                 FromDLQType(t.Type),\n\t\tReplicationTasks:     FromReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: FromReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\nfunc ToHistoryReadDLQMessagesResponse(t *historyv1.ReadDLQMessagesResponse) *types.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesResponse{\n\t\tType:                 ToDLQType(t.Type),\n\t\tReplicationTasks:     ToReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: ToReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\nfunc FromHistoryReapplyEventsRequest(t *types.HistoryReapplyEventsRequest) *historyv1.ReapplyEventsRequest {\n\tif t == nil || t.Request == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ReapplyEventsRequest{\n\t\tDomain:            t.Request.DomainName,\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Request.WorkflowExecution),\n\t\tEvents:            FromDataBlob(t.Request.Events),\n\t}\n}\n\nfunc ToHistoryReapplyEventsRequest(t *historyv1.ReapplyEventsRequest) *types.HistoryReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryReapplyEventsRequest{\n\t\tDomainUUID: t.DomainId,\n\t\tRequest: &types.ReapplyEventsRequest{\n\t\t\tDomainName:        t.Domain,\n\t\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\t\tEvents:            ToDataBlob(t.Events),\n\t\t},\n\t}\n}\n\nfunc FromHistoryRecordActivityTaskHeartbeatRequest(t *types.HistoryRecordActivityTaskHeartbeatRequest) *historyv1.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordActivityTaskHeartbeatRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRecordActivityTaskHeartbeatRequest(t.HeartbeatRequest),\n\t}\n}\n\nfunc ToHistoryRecordActivityTaskHeartbeatRequest(t *historyv1.RecordActivityTaskHeartbeatRequest) *types.HistoryRecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID:       t.DomainId,\n\t\tHeartbeatRequest: ToRecordActivityTaskHeartbeatRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryRecordActivityTaskHeartbeatResponse(t *types.RecordActivityTaskHeartbeatResponse) *historyv1.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc ToHistoryRecordActivityTaskHeartbeatResponse(t *historyv1.RecordActivityTaskHeartbeatResponse) *types.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.CancelRequested,\n\t}\n}\n\nfunc FromHistoryRecordActivityTaskStartedRequest(t *types.RecordActivityTaskStartedRequest) *historyv1.RecordActivityTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordActivityTaskStartedRequest{\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleId:        t.ScheduleID,\n\t\tTaskId:            t.TaskID,\n\t\tRequestId:         t.RequestID,\n\t\tPollRequest:       FromPollForActivityTaskRequest(t.PollRequest),\n\t}\n}\n\nfunc ToHistoryRecordActivityTaskStartedRequest(t *historyv1.RecordActivityTaskStartedRequest) *types.RecordActivityTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        t.DomainId,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleID:        t.ScheduleId,\n\t\tTaskID:            t.TaskId,\n\t\tRequestID:         t.RequestId,\n\t\tPollRequest:       ToPollForActivityTaskRequest(t.PollRequest),\n\t}\n}\n\nfunc FromHistoryRecordActivityTaskStartedResponse(t *types.RecordActivityTaskStartedResponse) *historyv1.RecordActivityTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordActivityTaskStartedResponse{\n\t\tScheduledEvent:             FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedTime:                unixNanoToTime(t.StartedTimestamp),\n\t\tAttempt:                    int32(t.Attempt),\n\t\tScheduledTimeOfThisAttempt: unixNanoToTime(t.ScheduledTimestampOfThisAttempt),\n\t\tHeartbeatDetails:           FromPayload(t.HeartbeatDetails),\n\t\tWorkflowType:               FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:             t.WorkflowDomain,\n\t}\n}\n\nfunc ToHistoryRecordActivityTaskStartedResponse(t *historyv1.RecordActivityTaskStartedResponse) *types.RecordActivityTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskStartedResponse{\n\t\tScheduledEvent:                  ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                timeToUnixNano(t.StartedTime),\n\t\tAttempt:                         int64(t.Attempt),\n\t\tScheduledTimestampOfThisAttempt: timeToUnixNano(t.ScheduledTimeOfThisAttempt),\n\t\tHeartbeatDetails:                ToPayload(t.HeartbeatDetails),\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.WorkflowDomain,\n\t}\n}\n\nfunc FromHistoryRecordChildExecutionCompletedRequest(t *types.RecordChildExecutionCompletedRequest) *historyv1.RecordChildExecutionCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordChildExecutionCompletedRequest{\n\t\tDomainId:           t.DomainUUID,\n\t\tWorkflowExecution:  FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedId:        t.InitiatedID,\n\t\tCompletedExecution: FromWorkflowExecution(t.CompletedExecution),\n\t\tCompletionEvent:    FromHistoryEvent(t.CompletionEvent),\n\t\tStartedId:          t.StartedID,\n\t}\n}\n\nfunc ToHistoryRecordChildExecutionCompletedRequest(t *historyv1.RecordChildExecutionCompletedRequest) *types.RecordChildExecutionCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordChildExecutionCompletedRequest{\n\t\tDomainUUID:         t.DomainId,\n\t\tWorkflowExecution:  ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedID:        t.InitiatedId,\n\t\tCompletedExecution: ToWorkflowExecution(t.CompletedExecution),\n\t\tCompletionEvent:    ToHistoryEvent(t.CompletionEvent),\n\t\tStartedID:          t.StartedId,\n\t}\n}\n\nfunc FromHistoryRecordDecisionTaskStartedRequest(t *types.RecordDecisionTaskStartedRequest) *historyv1.RecordDecisionTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordDecisionTaskStartedRequest{\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleId:        t.ScheduleID,\n\t\tTaskId:            t.TaskID,\n\t\tRequestId:         t.RequestID,\n\t\tPollRequest:       FromPollForDecisionTaskRequest(t.PollRequest),\n\t}\n}\n\nfunc ToHistoryRecordDecisionTaskStartedRequest(t *historyv1.RecordDecisionTaskStartedRequest) *types.RecordDecisionTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        t.DomainId,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleID:        t.ScheduleId,\n\t\tTaskID:            t.TaskId,\n\t\tRequestID:         t.RequestId,\n\t\tPollRequest:       ToPollForDecisionTaskRequest(t.PollRequest),\n\t}\n}\n\nfunc FromHistoryRecordDecisionTaskStartedResponse(t *types.RecordDecisionTaskStartedResponse) *historyv1.RecordDecisionTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RecordDecisionTaskStartedResponse{\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    fromInt64Value(t.PreviousStartedEventID),\n\t\tScheduledEventId:          t.ScheduledEventID,\n\t\tStartedEventId:            t.StartedEventID,\n\t\tNextEventId:               t.NextEventID,\n\t\tAttempt:                   int32(t.Attempt),\n\t\tStickyExecutionEnabled:    t.StickyExecutionEnabled,\n\t\tDecisionInfo:              FromTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTime:             unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:               unixNanoToTime(t.StartedTimestamp),\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tHistorySize:               t.HistorySize,\n\t}\n}\n\nfunc ToHistoryRecordDecisionTaskStartedResponse(t *historyv1.RecordDecisionTaskStartedResponse) *types.RecordDecisionTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordDecisionTaskStartedResponse{\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    toInt64Value(t.PreviousStartedEventId),\n\t\tScheduledEventID:          t.ScheduledEventId,\n\t\tStartedEventID:            t.StartedEventId,\n\t\tNextEventID:               t.NextEventId,\n\t\tAttempt:                   int64(t.Attempt),\n\t\tStickyExecutionEnabled:    t.StickyExecutionEnabled,\n\t\tDecisionInfo:              ToTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:          timeToUnixNano(t.StartedTime),\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tHistorySize:               t.HistorySize,\n\t}\n}\n\nfunc FromHistoryRefreshWorkflowTasksRequest(t *types.HistoryRefreshWorkflowTasksRequest) *historyv1.RefreshWorkflowTasksRequest {\n\tif t == nil || t.Request == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RefreshWorkflowTasksRequest{\n\t\tDomainId:          t.DomainUIID,\n\t\tDomain:            t.Request.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.Request.Execution),\n\t}\n}\n\nfunc ToHistoryRefreshWorkflowTasksRequest(t *historyv1.RefreshWorkflowTasksRequest) *types.HistoryRefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRefreshWorkflowTasksRequest{\n\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\tDomain:    t.Domain,\n\t\t\tExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\t},\n\t\tDomainUIID: t.DomainId,\n\t}\n}\n\nfunc FromHistoryRemoveSignalMutableStateRequest(t *types.RemoveSignalMutableStateRequest) *historyv1.RemoveSignalMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RemoveSignalMutableStateRequest{\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tRequestId:         t.RequestID,\n\t}\n}\n\nfunc ToHistoryRemoveSignalMutableStateRequest(t *historyv1.RemoveSignalMutableStateRequest) *types.RemoveSignalMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID:        t.DomainId,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tRequestID:         t.RequestId,\n\t}\n}\n\nfunc FromHistoryRemoveTaskRequest(t *types.RemoveTaskRequest) *historyv1.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RemoveTaskRequest{\n\t\tShardId:        t.ShardID,\n\t\tTaskType:       FromTaskType(t.Type),\n\t\tTaskId:         t.TaskID,\n\t\tVisibilityTime: unixNanoToTime(t.VisibilityTimestamp),\n\t\tClusterName:    t.ClusterName,\n\t}\n}\n\nfunc ToHistoryRemoveTaskRequest(t *historyv1.RemoveTaskRequest) *types.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoveTaskRequest{\n\t\tShardID:             t.ShardId,\n\t\tType:                ToTaskType(t.TaskType),\n\t\tTaskID:              t.TaskId,\n\t\tVisibilityTimestamp: timeToUnixNano(t.VisibilityTime),\n\t\tClusterName:         t.ClusterName,\n\t}\n}\n\nfunc FromHistoryReplicateEventsV2Request(t *types.ReplicateEventsV2Request) *historyv1.ReplicateEventsV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ReplicateEventsV2Request{\n\t\tDomainId:            t.DomainUUID,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tVersionHistoryItems: FromVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              FromDataBlob(t.Events),\n\t\tNewRunEvents:        FromDataBlob(t.NewRunEvents),\n\t}\n}\n\nfunc ToHistoryReplicateEventsV2Request(t *historyv1.ReplicateEventsV2Request) *types.ReplicateEventsV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicateEventsV2Request{\n\t\tDomainUUID:          t.DomainId,\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tVersionHistoryItems: ToVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              ToDataBlob(t.Events),\n\t\tNewRunEvents:        ToDataBlob(t.NewRunEvents),\n\t}\n}\n\nfunc FromHistoryRequestCancelWorkflowExecutionRequest(t *types.HistoryRequestCancelWorkflowExecutionRequest) *historyv1.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RequestCancelWorkflowExecutionRequest{\n\t\tDomainId:              t.DomainUUID,\n\t\tCancelRequest:         FromRequestCancelWorkflowExecutionRequest(t.CancelRequest),\n\t\tExternalExecutionInfo: FromExternalExecutionInfoFields(t.ExternalWorkflowExecution, t.ExternalInitiatedEventID),\n\t\tChildWorkflowOnly:     t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToHistoryRequestCancelWorkflowExecutionRequest(t *historyv1.RequestCancelWorkflowExecutionRequest) *types.HistoryRequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID:                t.DomainId,\n\t\tCancelRequest:             ToRequestCancelWorkflowExecutionRequest(t.CancelRequest),\n\t\tExternalInitiatedEventID:  ToExternalInitiatedID(t.ExternalExecutionInfo),\n\t\tExternalWorkflowExecution: ToExternalWorkflowExecution(t.ExternalExecutionInfo),\n\t\tChildWorkflowOnly:         t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromHistoryResetQueueRequest(t *types.ResetQueueRequest) *historyv1.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ResetQueueRequest{\n\t\tShardId:     t.ShardID,\n\t\tClusterName: t.ClusterName,\n\t\tTaskType:    FromTaskType(t.Type),\n\t}\n}\n\nfunc ToHistoryResetQueueRequest(t *historyv1.ResetQueueRequest) *types.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetQueueRequest{\n\t\tShardID:     t.ShardId,\n\t\tClusterName: t.ClusterName,\n\t\tType:        ToTaskType(t.TaskType),\n\t}\n}\n\nfunc FromHistoryResetStickyTaskListRequest(t *types.HistoryResetStickyTaskListRequest) *historyv1.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ResetStickyTaskListRequest{\n\t\tRequest: &apiv1.ResetStickyTaskListRequest{\n\t\t\tDomain:            \"\",\n\t\t\tWorkflowExecution: FromWorkflowExecution(t.Execution),\n\t\t},\n\t\tDomainId: t.DomainUUID,\n\t}\n}\n\nfunc ToHistoryResetStickyTaskListRequest(t *historyv1.ResetStickyTaskListRequest) *types.HistoryResetStickyTaskListRequest {\n\tif t == nil || t.Request == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: t.DomainId,\n\t\tExecution:  ToWorkflowExecution(t.Request.WorkflowExecution),\n\t}\n}\n\nfunc FromHistoryResetWorkflowExecutionRequest(t *types.HistoryResetWorkflowExecutionRequest) *historyv1.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ResetWorkflowExecutionRequest{\n\t\tRequest:  FromResetWorkflowExecutionRequest(t.ResetRequest),\n\t\tDomainId: t.DomainUUID,\n\t}\n}\n\nfunc ToHistoryResetWorkflowExecutionRequest(t *historyv1.ResetWorkflowExecutionRequest) *types.HistoryResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetWorkflowExecutionRequest{\n\t\tResetRequest: ToResetWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID:   t.DomainId,\n\t}\n}\n\nfunc FromHistoryResetWorkflowExecutionResponse(t *types.ResetWorkflowExecutionResponse) *historyv1.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ResetWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToHistoryResetWorkflowExecutionResponse(t *historyv1.ResetWorkflowExecutionResponse) *types.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromHistoryRespondActivityTaskCanceledRequest(t *types.HistoryRespondActivityTaskCanceledRequest) *historyv1.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondActivityTaskCanceledRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRespondActivityTaskCanceledRequest(t.CancelRequest),\n\t}\n}\n\nfunc ToHistoryRespondActivityTaskCanceledRequest(t *historyv1.RespondActivityTaskCanceledRequest) *types.HistoryRespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID:    t.DomainId,\n\t\tCancelRequest: ToRespondActivityTaskCanceledRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryRespondActivityTaskCompletedRequest(t *types.HistoryRespondActivityTaskCompletedRequest) *historyv1.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondActivityTaskCompletedRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRespondActivityTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\nfunc ToHistoryRespondActivityTaskCompletedRequest(t *historyv1.RespondActivityTaskCompletedRequest) *types.HistoryRespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID:      t.DomainId,\n\t\tCompleteRequest: ToRespondActivityTaskCompletedRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryRespondActivityTaskFailedRequest(t *types.HistoryRespondActivityTaskFailedRequest) *historyv1.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondActivityTaskFailedRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRespondActivityTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\nfunc ToHistoryRespondActivityTaskFailedRequest(t *historyv1.RespondActivityTaskFailedRequest) *types.HistoryRespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID:    t.DomainId,\n\t\tFailedRequest: ToRespondActivityTaskFailedRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryRespondDecisionTaskCompletedRequest(t *types.HistoryRespondDecisionTaskCompletedRequest) *historyv1.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondDecisionTaskCompletedRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRespondDecisionTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\nfunc ToHistoryRespondDecisionTaskCompletedRequest(t *historyv1.RespondDecisionTaskCompletedRequest) *types.HistoryRespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID:      t.DomainId,\n\t\tCompleteRequest: ToRespondDecisionTaskCompletedRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryRespondDecisionTaskCompletedResponse(t *types.HistoryRespondDecisionTaskCompletedResponse) *historyv1.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondDecisionTaskCompletedResponse{\n\t\tStartedResponse:             FromHistoryRecordDecisionTaskStartedResponse(t.StartedResponse),\n\t\tActivitiesToDispatchLocally: FromActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\nfunc ToHistoryRespondDecisionTaskCompletedResponse(t *historyv1.RespondDecisionTaskCompletedResponse) *types.HistoryRespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tStartedResponse:             ToHistoryRecordDecisionTaskStartedResponse(t.StartedResponse),\n\t\tActivitiesToDispatchLocally: ToActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\nfunc FromHistoryRespondDecisionTaskFailedRequest(t *types.HistoryRespondDecisionTaskFailedRequest) *historyv1.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondDecisionTaskFailedRequest{\n\t\tDomainId: t.DomainUUID,\n\t\tRequest:  FromRespondDecisionTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\nfunc ToHistoryRespondDecisionTaskFailedRequest(t *historyv1.RespondDecisionTaskFailedRequest) *types.HistoryRespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID:    t.DomainId,\n\t\tFailedRequest: ToRespondDecisionTaskFailedRequest(t.Request),\n\t}\n}\n\nfunc FromHistoryScheduleDecisionTaskRequest(t *types.ScheduleDecisionTaskRequest) *historyv1.ScheduleDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ScheduleDecisionTaskRequest{\n\t\tDomainId:          t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tIsFirstDecision:   t.IsFirstDecision,\n\t}\n}\n\nfunc ToHistoryScheduleDecisionTaskRequest(t *historyv1.ScheduleDecisionTaskRequest) *types.ScheduleDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleDecisionTaskRequest{\n\t\tDomainUUID:        t.DomainId,\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tIsFirstDecision:   t.IsFirstDecision,\n\t}\n}\n\nfunc FromHistorySignalWithStartWorkflowExecutionRequest(t *types.HistorySignalWithStartWorkflowExecutionRequest) *historyv1.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.SignalWithStartWorkflowExecutionRequest{\n\t\tRequest:         FromSignalWithStartWorkflowExecutionRequest(t.SignalWithStartRequest),\n\t\tDomainId:        t.DomainUUID,\n\t\tPartitionConfig: t.PartitionConfig,\n\t}\n}\n\nfunc ToHistorySignalWithStartWorkflowExecutionRequest(t *historyv1.SignalWithStartWorkflowExecutionRequest) *types.HistorySignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tSignalWithStartRequest: ToSignalWithStartWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID:             t.DomainId,\n\t\tPartitionConfig:        t.PartitionConfig,\n\t}\n}\n\nfunc FromHistorySignalWithStartWorkflowExecutionResponse(t *types.StartWorkflowExecutionResponse) *historyv1.SignalWithStartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.SignalWithStartWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToHistorySignalWithStartWorkflowExecutionResponse(t *historyv1.SignalWithStartWorkflowExecutionResponse) *types.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromHistorySignalWorkflowExecutionRequest(t *types.HistorySignalWorkflowExecutionRequest) *historyv1.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.SignalWorkflowExecutionRequest{\n\t\tRequest:                   FromSignalWorkflowExecutionRequest(t.SignalRequest),\n\t\tDomainId:                  t.DomainUUID,\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToHistorySignalWorkflowExecutionRequest(t *historyv1.SignalWorkflowExecutionRequest) *types.HistorySignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistorySignalWorkflowExecutionRequest{\n\t\tSignalRequest:             ToSignalWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID:                t.DomainId,\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.ChildWorkflowOnly,\n\t}\n}\n\nfunc FromHistoryStartWorkflowExecutionRequest(t *types.HistoryStartWorkflowExecutionRequest) *historyv1.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.StartWorkflowExecutionRequest{\n\t\tRequest:                  FromStartWorkflowExecutionRequest(t.StartRequest),\n\t\tDomainId:                 t.DomainUUID,\n\t\tParentExecutionInfo:      FromParentExecutionInfo(t.ParentExecutionInfo),\n\t\tAttempt:                  t.Attempt,\n\t\tExpirationTime:           unixNanoToTime(t.ExpirationTimestamp),\n\t\tContinueAsNewInitiator:   FromContinueAsNewInitiator(t.ContinueAsNewInitiator),\n\t\tContinuedFailure:         FromFailure(t.ContinuedFailureReason, t.ContinuedFailureDetails),\n\t\tLastCompletionResult:     FromPayload(t.LastCompletionResult),\n\t\tFirstDecisionTaskBackoff: secondsToDuration(t.FirstDecisionTaskBackoffSeconds),\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\nfunc ToHistoryStartWorkflowExecutionRequest(t *historyv1.StartWorkflowExecutionRequest) *types.HistoryStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryStartWorkflowExecutionRequest{\n\t\tStartRequest:                    ToStartWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID:                      t.DomainId,\n\t\tParentExecutionInfo:             ToParentExecutionInfo(t.ParentExecutionInfo),\n\t\tAttempt:                         t.Attempt,\n\t\tExpirationTimestamp:             timeToUnixNano(t.ExpirationTime),\n\t\tContinueAsNewInitiator:          ToContinueAsNewInitiator(t.ContinueAsNewInitiator),\n\t\tContinuedFailureReason:          ToFailureReason(t.ContinuedFailure),\n\t\tContinuedFailureDetails:         ToFailureDetails(t.ContinuedFailure),\n\t\tLastCompletionResult:            ToPayload(t.LastCompletionResult),\n\t\tFirstDecisionTaskBackoffSeconds: durationToSeconds(t.FirstDecisionTaskBackoff),\n\t\tPartitionConfig:                 t.PartitionConfig,\n\t}\n}\n\nfunc FromHistoryStartWorkflowExecutionResponse(t *types.StartWorkflowExecutionResponse) *historyv1.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.StartWorkflowExecutionResponse{\n\t\tRunId: t.RunID,\n\t}\n}\n\nfunc ToHistoryStartWorkflowExecutionResponse(t *historyv1.StartWorkflowExecutionResponse) *types.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: t.RunId,\n\t}\n}\n\nfunc FromHistorySyncActivityRequest(t *types.SyncActivityRequest) *historyv1.SyncActivityRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.SyncActivityRequest{\n\t\tDomainId:           t.DomainID,\n\t\tWorkflowExecution:  FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tVersion:            t.Version,\n\t\tScheduledId:        t.ScheduledID,\n\t\tScheduledTime:      unixNanoToTime(t.ScheduledTime),\n\t\tStartedId:          t.StartedID,\n\t\tStartedTime:        unixNanoToTime(t.StartedTime),\n\t\tLastHeartbeatTime:  unixNanoToTime(t.LastHeartbeatTime),\n\t\tDetails:            FromPayload(t.Details),\n\t\tAttempt:            t.Attempt,\n\t\tLastFailure:        FromFailure(t.LastFailureReason, t.LastFailureDetails),\n\t\tLastWorkerIdentity: t.LastWorkerIdentity,\n\t\tVersionHistory:     FromVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc ToHistorySyncActivityRequest(t *historyv1.SyncActivityRequest) *types.SyncActivityRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncActivityRequest{\n\t\tDomainID:           t.DomainId,\n\t\tWorkflowID:         ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:              ToRunID(t.WorkflowExecution),\n\t\tVersion:            t.Version,\n\t\tScheduledID:        t.ScheduledId,\n\t\tScheduledTime:      timeToUnixNano(t.ScheduledTime),\n\t\tStartedID:          t.StartedId,\n\t\tStartedTime:        timeToUnixNano(t.StartedTime),\n\t\tLastHeartbeatTime:  timeToUnixNano(t.LastHeartbeatTime),\n\t\tDetails:            ToPayload(t.Details),\n\t\tAttempt:            t.Attempt,\n\t\tLastFailureReason:  ToFailureReason(t.LastFailure),\n\t\tLastFailureDetails: ToFailureDetails(t.LastFailure),\n\t\tLastWorkerIdentity: t.LastWorkerIdentity,\n\t\tVersionHistory:     ToVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc FromHistorySyncShardStatusRequest(t *types.SyncShardStatusRequest) *historyv1.SyncShardStatusRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.SyncShardStatusRequest{\n\t\tSourceCluster: t.SourceCluster,\n\t\tShardId:       int32(t.ShardID),\n\t\tTime:          unixNanoToTime(t.Timestamp),\n\t}\n}\n\nfunc ToHistorySyncShardStatusRequest(t *historyv1.SyncShardStatusRequest) *types.SyncShardStatusRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatusRequest{\n\t\tSourceCluster: t.SourceCluster,\n\t\tShardID:       int64(t.ShardId),\n\t\tTimestamp:     timeToUnixNano(t.Time),\n\t}\n}\n\nfunc FromHistoryTerminateWorkflowExecutionRequest(t *types.HistoryTerminateWorkflowExecutionRequest) *historyv1.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.TerminateWorkflowExecutionRequest{\n\t\tRequest:                   FromTerminateWorkflowExecutionRequest(t.TerminateRequest),\n\t\tDomainId:                  t.DomainUUID,\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.ChildWorkflowOnly,\n\t}\n}\n\nfunc ToHistoryTerminateWorkflowExecutionRequest(t *historyv1.TerminateWorkflowExecutionRequest) *types.HistoryTerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryTerminateWorkflowExecutionRequest{\n\t\tTerminateRequest:          ToTerminateWorkflowExecutionRequest(t.Request),\n\t\tDomainUUID:                t.DomainId,\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.ChildWorkflowOnly,\n\t}\n}\n\n// FromHistoryGetCrossClusterTasksRequest converts internal GetCrossClusterTasksRequest type to proto\nfunc FromHistoryGetCrossClusterTasksRequest(t *types.GetCrossClusterTasksRequest) *historyv1.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetCrossClusterTasksRequest{\n\t\tShardIds:      t.ShardIDs,\n\t\tTargetCluster: t.TargetCluster,\n\t}\n}\n\n// ToHistoryGetCrossClusterTasksRequest converts proto GetCrossClusterTasksRequest type to internal\nfunc ToHistoryGetCrossClusterTasksRequest(t *historyv1.GetCrossClusterTasksRequest) *types.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksRequest{\n\t\tShardIDs:      t.ShardIds,\n\t\tTargetCluster: t.TargetCluster,\n\t}\n}\n\n// FromHistoryGetCrossClusterTasksResponse converts internal GetCrossClusterTasksResponse type to proto\nfunc FromHistoryGetCrossClusterTasksResponse(t *types.GetCrossClusterTasksResponse) *historyv1.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       FromCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: FromGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// ToHistoryGetCrossClusterTasksResponse converts proto GetCrossClusterTasksResponse type to internal\nfunc ToHistoryGetCrossClusterTasksResponse(t *historyv1.GetCrossClusterTasksResponse) *types.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       ToCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: ToGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// FromHistoryRespondCrossClusterTasksCompletedRequest converts internal RespondCrossClusterTasksCompletedRequest type to thrift\nfunc FromHistoryRespondCrossClusterTasksCompletedRequest(t *types.RespondCrossClusterTasksCompletedRequest) *historyv1.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondCrossClusterTasksCompletedRequest{\n\t\tShardId:       t.ShardID,\n\t\tTargetCluster: t.TargetCluster,\n\t\tTaskResponses: FromCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: t.FetchNewTasks,\n\t}\n}\n\n// ToHistoryRespondCrossClusterTasksCompletedRequest converts thrift RespondCrossClusterTasksCompletedRequest type to internal\nfunc ToHistoryRespondCrossClusterTasksCompletedRequest(t *historyv1.RespondCrossClusterTasksCompletedRequest) *types.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedRequest{\n\t\tShardID:       t.ShardId,\n\t\tTargetCluster: t.TargetCluster,\n\t\tTaskResponses: ToCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: t.FetchNewTasks,\n\t}\n}\n\n// FromHistoryRespondCrossClusterTasksCompletedResponse converts internal RespondCrossClusterTasksCompletedResponse type to thrift\nfunc FromHistoryRespondCrossClusterTasksCompletedResponse(t *types.RespondCrossClusterTasksCompletedResponse) *historyv1.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: FromCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\n// ToHistoryRespondCrossClusterTasksCompletedResponse converts thrift RespondCrossClusterTasksCompletedResponse type to internal\nfunc ToHistoryRespondCrossClusterTasksCompletedResponse(t *historyv1.RespondCrossClusterTasksCompletedResponse) *types.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: ToCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\nfunc FromHistoryResetStickyTaskListResponse(t *types.HistoryResetStickyTaskListResponse) *historyv1.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.ResetStickyTaskListResponse{}\n}\n\nfunc ToHistoryResetStickyTaskListResponse(t *historyv1.ResetStickyTaskListResponse) *types.HistoryResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetStickyTaskListResponse{}\n}\n\nfunc FromHistoryRatelimitUpdateRequest(t *types.RatelimitUpdateRequest) *historyv1.RatelimitUpdateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RatelimitUpdateRequest{\n\t\tData: FromAny(t.Any),\n\t}\n}\nfunc ToHistoryRatelimitUpdateRequest(t *historyv1.RatelimitUpdateRequest) *types.RatelimitUpdateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RatelimitUpdateRequest{\n\t\tAny: ToAny(t.Data),\n\t}\n}\n\nfunc FromHistoryRatelimitUpdateResponse(t *types.RatelimitUpdateResponse) *historyv1.RatelimitUpdateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &historyv1.RatelimitUpdateResponse{\n\t\tData: FromAny(t.Any),\n\t}\n}\n\nfunc ToHistoryRatelimitUpdateResponse(t *historyv1.RatelimitUpdateResponse) *types.RatelimitUpdateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RatelimitUpdateResponse{\n\t\tAny: ToAny(t.Data),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/history_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestHistoryCloseShardRequest(t *testing.T) {\n\tfor _, item := range []*types.CloseShardRequest{nil, {}, &testdata.HistoryCloseShardRequest} {\n\t\tassert.Equal(t, item, ToHistoryCloseShardRequest(FromHistoryCloseShardRequest(item)))\n\t}\n}\nfunc TestHistoryDescribeHistoryHostRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostRequest{nil, {}, &testdata.HistoryDescribeHistoryHostRequest} {\n\t\tassert.Equal(t, item, ToHistoryDescribeHistoryHostRequest(FromHistoryDescribeHistoryHostRequest(item)))\n\t}\n}\nfunc TestHistoryDescribeHistoryHostResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostResponse{nil, {}, &testdata.HistoryDescribeHistoryHostResponse} {\n\t\tassert.Equal(t, item, ToHistoryDescribeHistoryHostResponse(FromHistoryDescribeHistoryHostResponse(item)))\n\t}\n}\nfunc TestHistoryDescribeMutableStateRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeMutableStateRequest{nil, {}, &testdata.HistoryDescribeMutableStateRequest} {\n\t\tassert.Equal(t, item, ToHistoryDescribeMutableStateRequest(FromHistoryDescribeMutableStateRequest(item)))\n\t}\n}\nfunc TestHistoryDescribeMutableStateResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeMutableStateResponse{nil, {}, &testdata.HistoryDescribeMutableStateResponse} {\n\t\tassert.Equal(t, item, ToHistoryDescribeMutableStateResponse(FromHistoryDescribeMutableStateResponse(item)))\n\t}\n}\nfunc TestHistoryDescribeQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueRequest{nil, {}, &testdata.HistoryDescribeQueueRequest} {\n\t\tassert.Equal(t, item, ToHistoryDescribeQueueRequest(FromHistoryDescribeQueueRequest(item)))\n\t}\n}\nfunc TestHistoryDescribeQueueResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueResponse{nil, {}, &testdata.HistoryDescribeQueueResponse} {\n\t\tassert.Equal(t, item, ToHistoryDescribeQueueResponse(FromHistoryDescribeQueueResponse(item)))\n\t}\n}\nfunc TestHistoryDescribeWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryDescribeWorkflowExecutionRequest{nil, {}, &testdata.HistoryDescribeWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryDescribeWorkflowExecutionRequest(FromHistoryDescribeWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistoryDescribeWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeWorkflowExecutionResponse{nil, {}, &testdata.HistoryDescribeWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToHistoryDescribeWorkflowExecutionResponse(FromHistoryDescribeWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestHistoryGetDLQReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDLQReplicationMessagesRequest{nil, {}, &testdata.HistoryGetDLQReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryGetDLQReplicationMessagesRequest(FromHistoryGetDLQReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryGetDLQReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDLQReplicationMessagesResponse{nil, {}, &testdata.HistoryGetDLQReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToHistoryGetDLQReplicationMessagesResponse(FromHistoryGetDLQReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestHistoryGetMutableStateRequest(t *testing.T) {\n\tfor _, item := range []*types.GetMutableStateRequest{nil, {}, &testdata.HistoryGetMutableStateRequest} {\n\t\tassert.Equal(t, item, ToHistoryGetMutableStateRequest(FromHistoryGetMutableStateRequest(item)))\n\t}\n}\nfunc TestHistoryGetMutableStateResponse(t *testing.T) {\n\tfor _, item := range []*types.GetMutableStateResponse{nil, &testdata.HistoryGetMutableStateResponse} {\n\t\tassert.Equal(t, item, ToHistoryGetMutableStateResponse(FromHistoryGetMutableStateResponse(item)))\n\t}\n}\nfunc TestHistoryGetReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesRequest{nil, {}, &testdata.HistoryGetReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryGetReplicationMessagesRequest(FromHistoryGetReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryGetReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesResponse{nil, {}, &testdata.HistoryGetReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToHistoryGetReplicationMessagesResponse(FromHistoryGetReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestHistoryCountDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.CountDLQMessagesRequest{nil, {}, &testdata.HistoryCountDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryCountDLQMessagesRequest(FromHistoryCountDLQMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryCountDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.HistoryCountDLQMessagesResponse{nil, {}, &testdata.HistoryCountDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToHistoryCountDLQMessagesResponse(FromHistoryCountDLQMessagesResponse(item)))\n\t}\n}\nfunc TestHistoryMergeDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.MergeDLQMessagesRequest{nil, {}, &testdata.HistoryMergeDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryMergeDLQMessagesRequest(FromHistoryMergeDLQMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryMergeDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.MergeDLQMessagesResponse{nil, {}, &testdata.HistoryMergeDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToHistoryMergeDLQMessagesResponse(FromHistoryMergeDLQMessagesResponse(item)))\n\t}\n}\nfunc TestHistoryNotifyFailoverMarkersRequest(t *testing.T) {\n\tfor _, item := range []*types.NotifyFailoverMarkersRequest{nil, {}, &testdata.HistoryNotifyFailoverMarkersRequest} {\n\t\tassert.Equal(t, item, ToHistoryNotifyFailoverMarkersRequest(FromHistoryNotifyFailoverMarkersRequest(item)))\n\t}\n}\nfunc TestHistoryPollMutableStateRequest(t *testing.T) {\n\tfor _, item := range []*types.PollMutableStateRequest{nil, {}, &testdata.HistoryPollMutableStateRequest} {\n\t\tassert.Equal(t, item, ToHistoryPollMutableStateRequest(FromHistoryPollMutableStateRequest(item)))\n\t}\n}\nfunc TestHistoryPollMutableStateResponse(t *testing.T) {\n\tfor _, item := range []*types.PollMutableStateResponse{nil, &testdata.HistoryPollMutableStateResponse} {\n\t\tassert.Equal(t, item, ToHistoryPollMutableStateResponse(FromHistoryPollMutableStateResponse(item)))\n\t}\n}\nfunc TestHistoryPurgeDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.PurgeDLQMessagesRequest{nil, {}, &testdata.HistoryPurgeDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryPurgeDLQMessagesRequest(FromHistoryPurgeDLQMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryQueryWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryQueryWorkflowRequest{nil, {}, &testdata.HistoryQueryWorkflowRequest} {\n\t\tassert.Equal(t, item, ToHistoryQueryWorkflowRequest(FromHistoryQueryWorkflowRequest(item)))\n\t}\n}\nfunc TestHistoryQueryWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.HistoryQueryWorkflowResponse{nil, &testdata.HistoryQueryWorkflowResponse} {\n\t\tassert.Equal(t, item, ToHistoryQueryWorkflowResponse(FromHistoryQueryWorkflowResponse(item)))\n\t}\n\tassert.Nil(t, FromHistoryQueryWorkflowResponse(&types.HistoryQueryWorkflowResponse{}))\n}\nfunc TestHistoryReadDLQMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.ReadDLQMessagesRequest{nil, {}, &testdata.HistoryReadDLQMessagesRequest} {\n\t\tassert.Equal(t, item, ToHistoryReadDLQMessagesRequest(FromHistoryReadDLQMessagesRequest(item)))\n\t}\n}\nfunc TestHistoryReadDLQMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.ReadDLQMessagesResponse{nil, {}, &testdata.HistoryReadDLQMessagesResponse} {\n\t\tassert.Equal(t, item, ToHistoryReadDLQMessagesResponse(FromHistoryReadDLQMessagesResponse(item)))\n\t}\n}\nfunc TestHistoryReapplyEventsRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryReapplyEventsRequest{nil, &testdata.HistoryReapplyEventsRequest} {\n\t\tassert.Equal(t, item, ToHistoryReapplyEventsRequest(FromHistoryReapplyEventsRequest(item)))\n\t}\n\tassert.Nil(t, FromHistoryReapplyEventsRequest(&types.HistoryReapplyEventsRequest{}))\n}\nfunc TestHistoryRecordActivityTaskHeartbeatRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRecordActivityTaskHeartbeatRequest{nil, {}, &testdata.HistoryRecordActivityTaskHeartbeatRequest} {\n\t\tassert.Equal(t, item, ToHistoryRecordActivityTaskHeartbeatRequest(FromHistoryRecordActivityTaskHeartbeatRequest(item)))\n\t}\n}\nfunc TestHistoryRecordActivityTaskHeartbeatResponse(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskHeartbeatResponse{nil, {}, &testdata.HistoryRecordActivityTaskHeartbeatResponse} {\n\t\tassert.Equal(t, item, ToHistoryRecordActivityTaskHeartbeatResponse(FromHistoryRecordActivityTaskHeartbeatResponse(item)))\n\t}\n}\nfunc TestHistoryRecordActivityTaskStartedRequest(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskStartedRequest{nil, {}, &testdata.HistoryRecordActivityTaskStartedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRecordActivityTaskStartedRequest(FromHistoryRecordActivityTaskStartedRequest(item)))\n\t}\n}\nfunc TestHistoryRecordActivityTaskStartedResponse(t *testing.T) {\n\tfor _, item := range []*types.RecordActivityTaskStartedResponse{nil, {}, &testdata.HistoryRecordActivityTaskStartedResponse} {\n\t\tassert.Equal(t, item, ToHistoryRecordActivityTaskStartedResponse(FromHistoryRecordActivityTaskStartedResponse(item)))\n\t}\n}\nfunc TestHistoryRecordChildExecutionCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RecordChildExecutionCompletedRequest{nil, {}, &testdata.HistoryRecordChildExecutionCompletedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRecordChildExecutionCompletedRequest(FromHistoryRecordChildExecutionCompletedRequest(item)))\n\t}\n}\nfunc TestHistoryRecordDecisionTaskStartedRequest(t *testing.T) {\n\tfor _, item := range []*types.RecordDecisionTaskStartedRequest{nil, {}, &testdata.HistoryRecordDecisionTaskStartedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRecordDecisionTaskStartedRequest(FromHistoryRecordDecisionTaskStartedRequest(item)))\n\t}\n}\nfunc TestHistoryRecordDecisionTaskStartedResponse(t *testing.T) {\n\tfor _, item := range []*types.RecordDecisionTaskStartedResponse{nil, {}, &testdata.HistoryRecordDecisionTaskStartedResponse} {\n\t\tassert.Equal(t, item, ToHistoryRecordDecisionTaskStartedResponse(FromHistoryRecordDecisionTaskStartedResponse(item)))\n\t}\n}\nfunc TestHistoryRefreshWorkflowTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRefreshWorkflowTasksRequest{nil, &testdata.HistoryRefreshWorkflowTasksRequest} {\n\t\tassert.Equal(t, item, ToHistoryRefreshWorkflowTasksRequest(FromHistoryRefreshWorkflowTasksRequest(item)))\n\t}\n\tassert.Nil(t, FromHistoryRefreshWorkflowTasksRequest(&types.HistoryRefreshWorkflowTasksRequest{}))\n}\nfunc TestHistoryRemoveSignalMutableStateRequest(t *testing.T) {\n\tfor _, item := range []*types.RemoveSignalMutableStateRequest{nil, {}, &testdata.HistoryRemoveSignalMutableStateRequest} {\n\t\tassert.Equal(t, item, ToHistoryRemoveSignalMutableStateRequest(FromHistoryRemoveSignalMutableStateRequest(item)))\n\t}\n}\nfunc TestHistoryRemoveTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.RemoveTaskRequest{nil, {}, &testdata.HistoryRemoveTaskRequest} {\n\t\tassert.Equal(t, item, ToHistoryRemoveTaskRequest(FromHistoryRemoveTaskRequest(item)))\n\t}\n}\nfunc TestHistoryReplicateEventsV2Request(t *testing.T) {\n\tfor _, item := range []*types.ReplicateEventsV2Request{nil, {}, &testdata.HistoryReplicateEventsV2Request} {\n\t\tassert.Equal(t, item, ToHistoryReplicateEventsV2Request(FromHistoryReplicateEventsV2Request(item)))\n\t}\n}\nfunc TestHistoryRequestCancelWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRequestCancelWorkflowExecutionRequest{nil, {}, &testdata.HistoryRequestCancelWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryRequestCancelWorkflowExecutionRequest(FromHistoryRequestCancelWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistoryResetQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.ResetQueueRequest{nil, {}, &testdata.HistoryResetQueueRequest} {\n\t\tassert.Equal(t, item, ToHistoryResetQueueRequest(FromHistoryResetQueueRequest(item)))\n\t}\n}\nfunc TestHistoryResetStickyTaskListRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryResetStickyTaskListRequest{nil, {}, &testdata.HistoryResetStickyTaskListRequest} {\n\t\tassert.Equal(t, item, ToHistoryResetStickyTaskListRequest(FromHistoryResetStickyTaskListRequest(item)))\n\t}\n}\nfunc TestHistoryResetWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryResetWorkflowExecutionRequest{nil, {}, &testdata.HistoryResetWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryResetWorkflowExecutionRequest(FromHistoryResetWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistoryResetWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.ResetWorkflowExecutionResponse{nil, {}, &testdata.HistoryResetWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToHistoryResetWorkflowExecutionResponse(FromHistoryResetWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestHistoryRespondActivityTaskCanceledRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondActivityTaskCanceledRequest{nil, {}, &testdata.HistoryRespondActivityTaskCanceledRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondActivityTaskCanceledRequest(FromHistoryRespondActivityTaskCanceledRequest(item)))\n\t}\n}\nfunc TestHistoryRespondActivityTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondActivityTaskCompletedRequest{nil, {}, &testdata.HistoryRespondActivityTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondActivityTaskCompletedRequest(FromHistoryRespondActivityTaskCompletedRequest(item)))\n\t}\n}\nfunc TestHistoryRespondActivityTaskFailedRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondActivityTaskFailedRequest{nil, {}, &testdata.HistoryRespondActivityTaskFailedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondActivityTaskFailedRequest(FromHistoryRespondActivityTaskFailedRequest(item)))\n\t}\n}\nfunc TestHistoryRespondDecisionTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondDecisionTaskCompletedRequest{nil, {}, &testdata.HistoryRespondDecisionTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondDecisionTaskCompletedRequest(FromHistoryRespondDecisionTaskCompletedRequest(item)))\n\t}\n}\nfunc TestHistoryRespondDecisionTaskCompletedResponse(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondDecisionTaskCompletedResponse{nil, {}, &testdata.HistoryRespondDecisionTaskCompletedResponse} {\n\t\tassert.Equal(t, item, ToHistoryRespondDecisionTaskCompletedResponse(FromHistoryRespondDecisionTaskCompletedResponse(item)))\n\t}\n}\nfunc TestHistoryRespondDecisionTaskFailedRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryRespondDecisionTaskFailedRequest{nil, {}, &testdata.HistoryRespondDecisionTaskFailedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondDecisionTaskFailedRequest(FromHistoryRespondDecisionTaskFailedRequest(item)))\n\t}\n}\nfunc TestHistoryScheduleDecisionTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.ScheduleDecisionTaskRequest{nil, {}, &testdata.HistoryScheduleDecisionTaskRequest} {\n\t\tassert.Equal(t, item, ToHistoryScheduleDecisionTaskRequest(FromHistoryScheduleDecisionTaskRequest(item)))\n\t}\n}\nfunc TestHistorySignalWithStartWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistorySignalWithStartWorkflowExecutionRequest{nil, {}, &testdata.HistorySignalWithStartWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistorySignalWithStartWorkflowExecutionRequest(FromHistorySignalWithStartWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistorySignalWithStartWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionResponse{nil, {}, &testdata.HistorySignalWithStartWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToHistorySignalWithStartWorkflowExecutionResponse(FromHistorySignalWithStartWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestHistorySignalWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistorySignalWorkflowExecutionRequest{nil, {}, &testdata.HistorySignalWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistorySignalWorkflowExecutionRequest(FromHistorySignalWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistoryStartWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryStartWorkflowExecutionRequest{nil, {}, &testdata.HistoryStartWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryStartWorkflowExecutionRequest(FromHistoryStartWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestHistoryStartWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowExecutionResponse{nil, {}, &testdata.HistoryStartWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToHistoryStartWorkflowExecutionResponse(FromHistoryStartWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestHistorySyncActivityRequest(t *testing.T) {\n\tfor _, item := range []*types.SyncActivityRequest{nil, {}, &testdata.HistorySyncActivityRequest} {\n\t\tassert.Equal(t, item, ToHistorySyncActivityRequest(FromHistorySyncActivityRequest(item)))\n\t}\n}\nfunc TestHistorySyncShardStatusRequest(t *testing.T) {\n\tfor _, item := range []*types.SyncShardStatusRequest{nil, {}, &testdata.HistorySyncShardStatusRequest} {\n\t\tassert.Equal(t, item, ToHistorySyncShardStatusRequest(FromHistorySyncShardStatusRequest(item)))\n\t}\n}\nfunc TestHistoryTerminateWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.HistoryTerminateWorkflowExecutionRequest{nil, {}, &testdata.HistoryTerminateWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryTerminateWorkflowExecutionRequest(FromHistoryTerminateWorkflowExecutionRequest(item)))\n\t}\n}\n\nfunc TestHistoryGetCrossClusterTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksRequest{nil, {}, &testdata.HistoryGetCrossClusterTasksRequest} {\n\t\tassert.Equal(t, item, ToHistoryGetCrossClusterTasksRequest(FromHistoryGetCrossClusterTasksRequest(item)))\n\t}\n}\n\nfunc TestHistoryGetCrossClusterTasksResponse(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksResponse{nil, {}, &testdata.HistoryGetCrossClusterTasksResponse} {\n\t\tassert.Equal(t, item, ToHistoryGetCrossClusterTasksResponse(FromHistoryGetCrossClusterTasksResponse(item)))\n\t}\n}\n\nfunc TestHistoryRespondCrossClusterTasksCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedRequest{nil, {}, &testdata.HistoryRespondCrossClusterTasksCompletedRequest} {\n\t\tassert.Equal(t, item, ToHistoryRespondCrossClusterTasksCompletedRequest(FromHistoryRespondCrossClusterTasksCompletedRequest(item)))\n\t}\n}\n\nfunc TestHistoryRespondCrossClusterTasksCompletedResponse(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedResponse{nil, {}, &testdata.HistoryRespondCrossClusterTasksCompletedResponse} {\n\t\tassert.Equal(t, item, ToHistoryRespondCrossClusterTasksCompletedResponse(FromHistoryRespondCrossClusterTasksCompletedResponse(item)))\n\t}\n}\n\nfunc TestHistoryGetTaskInfoRequest(t *testing.T) {\n\tfor _, item := range []*types.GetFailoverInfoRequest{nil, {}, &testdata.GetFailoverInfoRequest} {\n\t\tassert.Equal(t, item, ToHistoryGetFailoverInfoRequest(FromHistoryGetFailoverInfoRequest(item)))\n\t}\n}\n\nfunc TestHistoryGetTaskInfoResponse(t *testing.T) {\n\tfor _, item := range []*types.GetFailoverInfoResponse{nil, {}, &testdata.GetFailoverInfoResponse} {\n\t\tassert.Equal(t, item, ToHistoryGetFailoverInfoResponse(FromHistoryGetFailoverInfoResponse(item)))\n\t}\n}\n\nfunc TestRatelimitUpdate(t *testing.T) {\n\tt.Run(\"round trip\", func(t *testing.T) {\n\t\tfor _, item := range []*types.RatelimitUpdateResponse{nil, {}, &testdata.RatelimitUpdateResponse} {\n\t\t\tassert.Equal(t, item, ToHistoryRatelimitUpdateResponse(FromHistoryRatelimitUpdateResponse(item)))\n\t\t}\n\t\tfor _, item := range []*types.RatelimitUpdateRequest{nil, {}, &testdata.RatelimitUpdateRequest} {\n\t\t\tassert.Equal(t, item, ToHistoryRatelimitUpdateRequest(FromHistoryRatelimitUpdateRequest(item)))\n\t\t}\n\t})\n\n\tt.Run(\"nil\", func(t *testing.T) {\n\t\tassert.Nil(t, ToHistoryRatelimitUpdateRequest(nil), \"request to internal\")\n\t\tassert.Nil(t, FromHistoryRatelimitUpdateResponse(nil), \"response from internal\")\n\t})\n\n\tt.Run(\"nil Any contents\", func(t *testing.T) {\n\t\tassert.Equal(t, &types.RatelimitUpdateRequest{Any: nil}, ToHistoryRatelimitUpdateRequest(&historyv1.RatelimitUpdateRequest{Data: nil}), \"request to internal\")\n\t\tassert.Equal(t, &historyv1.RatelimitUpdateResponse{Data: nil}, FromHistoryRatelimitUpdateResponse(&types.RatelimitUpdateResponse{Any: nil}), \"response from internal\")\n\t})\n\n\tt.Run(\"with Any contents\", func(t *testing.T) {\n\t\tinternal := &types.Any{ValueType: \"test\", Value: []byte(`test data`)}\n\t\tproto := &sharedv1.Any{ValueType: \"test\", Value: []byte(`test data`)}\n\t\tassert.Equal(t, &types.RatelimitUpdateRequest{Any: internal}, ToHistoryRatelimitUpdateRequest(&historyv1.RatelimitUpdateRequest{Data: proto}), \"request to internal\")\n\t\tassert.Equal(t, &historyv1.RatelimitUpdateResponse{Data: proto}, FromHistoryRatelimitUpdateResponse(&types.RatelimitUpdateResponse{Any: internal}), \"response from internal\")\n\t})\n}\n"
  },
  {
    "path": "common/types/mapper/proto/matching.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromMatchingAddActivityTaskRequest(t *types.AddActivityTaskRequest) *matchingv1.AddActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.AddActivityTaskRequest{\n\t\tDomainId:                 t.DomainUUID,\n\t\tWorkflowExecution:        FromWorkflowExecution(t.Execution),\n\t\tSourceDomainId:           t.SourceDomainUUID,\n\t\tTaskList:                 FromTaskList(t.TaskList),\n\t\tScheduleId:               t.ScheduleID,\n\t\tScheduleToStartTimeout:   secondsToDuration(t.ScheduleToStartTimeoutSeconds),\n\t\tSource:                   FromTaskSource(t.Source),\n\t\tForwardedFrom:            t.ForwardedFrom,\n\t\tActivityTaskDispatchInfo: FromActivityTaskDispatchInfo(t.ActivityTaskDispatchInfo),\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\nfunc ToMatchingAddActivityTaskRequest(t *matchingv1.AddActivityTaskRequest) *types.AddActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddActivityTaskRequest{\n\t\tDomainUUID:                    t.DomainId,\n\t\tExecution:                     ToWorkflowExecution(t.WorkflowExecution),\n\t\tSourceDomainUUID:              t.SourceDomainId,\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tScheduleID:                    t.ScheduleId,\n\t\tScheduleToStartTimeoutSeconds: durationToSeconds(t.ScheduleToStartTimeout),\n\t\tSource:                        ToTaskSource(t.Source),\n\t\tForwardedFrom:                 t.ForwardedFrom,\n\t\tActivityTaskDispatchInfo:      ToActivityTaskDispatchInfo(t.ActivityTaskDispatchInfo),\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\nfunc FromMatchingAddDecisionTaskRequest(t *types.AddDecisionTaskRequest) *matchingv1.AddDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.AddDecisionTaskRequest{\n\t\tDomainId:               t.DomainUUID,\n\t\tWorkflowExecution:      FromWorkflowExecution(t.Execution),\n\t\tTaskList:               FromTaskList(t.TaskList),\n\t\tScheduleId:             t.ScheduleID,\n\t\tScheduleToStartTimeout: secondsToDuration(t.ScheduleToStartTimeoutSeconds),\n\t\tSource:                 FromTaskSource(t.Source),\n\t\tForwardedFrom:          t.ForwardedFrom,\n\t\tPartitionConfig:        t.PartitionConfig,\n\t}\n}\n\nfunc ToMatchingAddDecisionTaskRequest(t *matchingv1.AddDecisionTaskRequest) *types.AddDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddDecisionTaskRequest{\n\t\tDomainUUID:                    t.DomainId,\n\t\tExecution:                     ToWorkflowExecution(t.WorkflowExecution),\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tScheduleID:                    t.ScheduleId,\n\t\tScheduleToStartTimeoutSeconds: durationToSeconds(t.ScheduleToStartTimeout),\n\t\tSource:                        ToTaskSource(t.Source),\n\t\tForwardedFrom:                 t.ForwardedFrom,\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\nfunc FromMatchingAddActivityTaskResponse(t *types.AddActivityTaskResponse) *matchingv1.AddActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.AddActivityTaskResponse{\n\t\tPartitionConfig: FromTaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc ToMatchingAddActivityTaskResponse(t *matchingv1.AddActivityTaskResponse) *types.AddActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddActivityTaskResponse{\n\t\tPartitionConfig: ToTaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc FromMatchingAddDecisionTaskResponse(t *types.AddDecisionTaskResponse) *matchingv1.AddDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.AddDecisionTaskResponse{\n\t\tPartitionConfig: FromTaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc ToMatchingAddDecisionTaskResponse(t *matchingv1.AddDecisionTaskResponse) *types.AddDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddDecisionTaskResponse{\n\t\tPartitionConfig: ToTaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc FromActivityTaskDispatchInfo(t *types.ActivityTaskDispatchInfo) *matchingv1.ActivityTaskDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.ActivityTaskDispatchInfo{\n\t\tScheduledEvent:             FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedTime:                unixNanoToTime(t.StartedTimestamp),\n\t\tAttempt:                    int32(common.Int64Default(t.Attempt)),\n\t\tScheduledTimeOfThisAttempt: unixNanoToTime(t.ScheduledTimestampOfThisAttempt),\n\t\tHeartbeatDetails:           FromPayload(t.HeartbeatDetails),\n\t\tWorkflowType:               FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:             t.WorkflowDomain,\n\t}\n}\n\nfunc ToActivityTaskDispatchInfo(t *matchingv1.ActivityTaskDispatchInfo) *types.ActivityTaskDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskDispatchInfo{\n\t\tScheduledEvent:                  ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                timeToUnixNano(t.StartedTime),\n\t\tAttempt:                         common.Int64Ptr(int64(t.Attempt)),\n\t\tScheduledTimestampOfThisAttempt: timeToUnixNano(t.ScheduledTimeOfThisAttempt),\n\t\tHeartbeatDetails:                ToPayload(t.HeartbeatDetails),\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.WorkflowDomain,\n\t}\n}\n\nfunc FromMatchingCancelOutstandingPollRequest(t *types.CancelOutstandingPollRequest) *matchingv1.CancelOutstandingPollRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar taskListType *types.TaskListType\n\tif t.TaskListType != nil {\n\t\ttaskListType = types.TaskListType(*t.TaskListType).Ptr()\n\t}\n\treturn &matchingv1.CancelOutstandingPollRequest{\n\t\tDomainId:     t.DomainUUID,\n\t\tPollerId:     t.PollerID,\n\t\tTaskListType: FromTaskListType(taskListType),\n\t\tTaskList:     FromTaskList(t.TaskList),\n\t}\n}\n\nfunc ToMatchingCancelOutstandingPollRequest(t *matchingv1.CancelOutstandingPollRequest) *types.CancelOutstandingPollRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tvar taskListType *int32\n\tif tlt := ToTaskListType(t.TaskListType); tlt != nil {\n\t\ttaskListType = common.Int32Ptr(int32(*tlt))\n\t}\n\treturn &types.CancelOutstandingPollRequest{\n\t\tDomainUUID:   t.DomainId,\n\t\tPollerID:     t.PollerId,\n\t\tTaskListType: taskListType,\n\t\tTaskList:     ToTaskList(t.TaskList),\n\t}\n}\n\nfunc FromMatchingDescribeTaskListRequest(t *types.MatchingDescribeTaskListRequest) *matchingv1.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.DescribeTaskListRequest{\n\t\tRequest:  FromDescribeTaskListRequest(t.DescRequest),\n\t\tDomainId: t.DomainUUID,\n\t}\n}\n\nfunc ToMatchingDescribeTaskListRequest(t *matchingv1.DescribeTaskListRequest) *types.MatchingDescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingDescribeTaskListRequest{\n\t\tDescRequest: ToDescribeTaskListRequest(t.Request),\n\t\tDomainUUID:  t.DomainId,\n\t}\n}\n\nfunc FromMatchingDescribeTaskListResponse(t *types.DescribeTaskListResponse) *matchingv1.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.DescribeTaskListResponse{\n\t\tPollers:         FromPollerInfoArray(t.Pollers),\n\t\tTaskListStatus:  FromTaskListStatus(t.TaskListStatus),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(t.PartitionConfig),\n\t\tTaskList:        FromTaskList(t.TaskList),\n\t}\n}\n\nfunc ToMatchingDescribeTaskListResponse(t *matchingv1.DescribeTaskListResponse) *types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers:         ToPollerInfoArray(t.Pollers),\n\t\tTaskListStatus:  ToTaskListStatus(t.TaskListStatus),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(t.PartitionConfig),\n\t\tTaskList:        ToTaskList(t.TaskList),\n\t}\n}\n\nfunc FromMatchingListTaskListPartitionsRequest(t *types.MatchingListTaskListPartitionsRequest) *matchingv1.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.ListTaskListPartitionsRequest{\n\t\tDomain:   t.Domain,\n\t\tTaskList: FromTaskList(t.TaskList),\n\t}\n}\n\nfunc ToMatchingListTaskListPartitionsRequest(t *matchingv1.ListTaskListPartitionsRequest) *types.MatchingListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingListTaskListPartitionsRequest{\n\t\tDomain:   t.Domain,\n\t\tTaskList: ToTaskList(t.TaskList),\n\t}\n}\n\nfunc FromMatchingListTaskListPartitionsResponse(t *types.ListTaskListPartitionsResponse) *matchingv1.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: FromTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: FromTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\nfunc ToMatchingListTaskListPartitionsResponse(t *matchingv1.ListTaskListPartitionsResponse) *types.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: ToTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: ToTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\nfunc FromMatchingGetTaskListsByDomainRequest(t *types.GetTaskListsByDomainRequest) *matchingv1.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.GetTaskListsByDomainRequest{\n\t\tDomain: t.Domain,\n\t}\n}\n\nfunc ToMatchingGetTaskListsByDomainRequest(t *matchingv1.GetTaskListsByDomainRequest) *types.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetTaskListsByDomainRequest{\n\t\tDomain: t.Domain,\n\t}\n}\n\nfunc FromMatchingGetTaskListsByDomainResponse(t *types.GetTaskListsByDomainResponse) *matchingv1.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &matchingv1.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: FromMatchingDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: FromMatchingDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\nfunc ToMatchingGetTaskListsByDomainResponse(t *matchingv1.GetTaskListsByDomainResponse) *types.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: ToMatchingDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: ToMatchingDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\nfunc FromMatchingDescribeTaskListResponseMap(t map[string]*types.DescribeTaskListResponse) map[string]*matchingv1.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*matchingv1.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = FromMatchingDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\nfunc ToMatchingDescribeTaskListResponseMap(t map[string]*matchingv1.DescribeTaskListResponse) map[string]*types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*types.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = ToMatchingDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\nfunc FromMatchingPollForActivityTaskRequest(t *types.MatchingPollForActivityTaskRequest) *matchingv1.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.PollForActivityTaskRequest{\n\t\tRequest:        FromPollForActivityTaskRequest(t.PollRequest),\n\t\tDomainId:       t.DomainUUID,\n\t\tPollerId:       t.PollerID,\n\t\tForwardedFrom:  t.ForwardedFrom,\n\t\tIsolationGroup: t.IsolationGroup,\n\t}\n}\n\nfunc ToMatchingPollForActivityTaskRequest(t *matchingv1.PollForActivityTaskRequest) *types.MatchingPollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForActivityTaskRequest{\n\t\tPollRequest:    ToPollForActivityTaskRequest(t.Request),\n\t\tDomainUUID:     t.DomainId,\n\t\tPollerID:       t.PollerId,\n\t\tForwardedFrom:  t.ForwardedFrom,\n\t\tIsolationGroup: t.IsolationGroup,\n\t}\n}\n\nfunc FromTaskListPartitionConfig(t *types.TaskListPartitionConfig) *matchingv1.TaskListPartitionConfig {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.TaskListPartitionConfig{\n\t\tVersion:            t.Version,\n\t\tNumReadPartitions:  int32(len(t.ReadPartitions)),\n\t\tNumWritePartitions: int32(len(t.WritePartitions)),\n\t\tReadPartitions:     FromMatchingTaskListPartitionsMap(t.ReadPartitions),\n\t\tWritePartitions:    FromMatchingTaskListPartitionsMap(t.WritePartitions),\n\t}\n}\n\nfunc ToTaskListPartitionConfig(t *matchingv1.TaskListPartitionConfig) *types.TaskListPartitionConfig {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartitionConfig{\n\t\tVersion:         t.Version,\n\t\tReadPartitions:  ToMatchingTaskListPartitionsMap(t.NumReadPartitions, t.ReadPartitions),\n\t\tWritePartitions: ToMatchingTaskListPartitionsMap(t.NumWritePartitions, t.WritePartitions),\n\t}\n}\n\nfunc FromMatchingTaskListPartition(t *types.TaskListPartition) *matchingv1.TaskListPartition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.TaskListPartition{\n\t\tIsolationGroups: t.IsolationGroups,\n\t}\n}\n\nfunc ToMatchingTaskListPartition(t *matchingv1.TaskListPartition) *types.TaskListPartition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartition{\n\t\tIsolationGroups: t.IsolationGroups,\n\t}\n}\n\nfunc FromMatchingTaskListPartitionsMap(m map[int]*types.TaskListPartition) map[int32]*matchingv1.TaskListPartition {\n\tif m == nil {\n\t\treturn nil\n\t}\n\tresult := make(map[int32]*matchingv1.TaskListPartition, len(m))\n\tfor id, p := range m {\n\t\tresult[int32(id)] = FromMatchingTaskListPartition(p)\n\t}\n\treturn result\n}\n\nfunc ToMatchingTaskListPartitionsMap(numPartitions int32, m map[int32]*matchingv1.TaskListPartition) map[int]*types.TaskListPartition {\n\tif m == nil && numPartitions == 0 {\n\t\treturn nil\n\t}\n\tresult := make(map[int]*types.TaskListPartition, len(m))\n\tif numPartitions != int32(len(m)) {\n\t\tfor i := int32(0); i < numPartitions; i++ {\n\t\t\tresult[int(i)] = &types.TaskListPartition{}\n\t\t}\n\t} else {\n\t\tfor id, p := range m {\n\t\t\tresult[int(id)] = ToMatchingTaskListPartition(p)\n\t\t}\n\t}\n\treturn result\n}\n\nfunc FromMatchingPollForActivityTaskResponse(t *types.MatchingPollForActivityTaskResponse) *matchingv1.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.PollForActivityTaskResponse{\n\t\tTaskToken:                  t.TaskToken,\n\t\tWorkflowExecution:          FromWorkflowExecution(t.WorkflowExecution),\n\t\tActivityId:                 t.ActivityID,\n\t\tActivityType:               FromActivityType(t.ActivityType),\n\t\tInput:                      FromPayload(t.Input),\n\t\tScheduledTime:              unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:                unixNanoToTime(t.StartedTimestamp),\n\t\tScheduleToCloseTimeout:     secondsToDuration(t.ScheduleToCloseTimeoutSeconds),\n\t\tStartToCloseTimeout:        secondsToDuration(t.StartToCloseTimeoutSeconds),\n\t\tHeartbeatTimeout:           secondsToDuration(t.HeartbeatTimeoutSeconds),\n\t\tAttempt:                    t.Attempt,\n\t\tScheduledTimeOfThisAttempt: unixNanoToTime(t.ScheduledTimestampOfThisAttempt),\n\t\tHeartbeatDetails:           FromPayload(t.HeartbeatDetails),\n\t\tWorkflowType:               FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:             t.WorkflowDomain,\n\t\tHeader:                     FromHeader(t.Header),\n\t\tPartitionConfig:            FromTaskListPartitionConfig(t.PartitionConfig),\n\t\tLoadBalancerHints:          FromLoadBalancerHints(t.LoadBalancerHints),\n\t\tAutoConfigHint:             FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc ToMatchingPollForActivityTaskResponse(t *matchingv1.PollForActivityTaskResponse) *types.MatchingPollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               ToWorkflowExecution(t.WorkflowExecution),\n\t\tActivityID:                      t.ActivityId,\n\t\tActivityType:                    ToActivityType(t.ActivityType),\n\t\tInput:                           ToPayload(t.Input),\n\t\tScheduledTimestamp:              timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:                timeToUnixNano(t.StartedTime),\n\t\tScheduleToCloseTimeoutSeconds:   durationToSeconds(t.ScheduleToCloseTimeout),\n\t\tStartToCloseTimeoutSeconds:      durationToSeconds(t.StartToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:         durationToSeconds(t.HeartbeatTimeout),\n\t\tAttempt:                         t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: timeToUnixNano(t.ScheduledTimeOfThisAttempt),\n\t\tHeartbeatDetails:                ToPayload(t.HeartbeatDetails),\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.WorkflowDomain,\n\t\tHeader:                          ToHeader(t.Header),\n\t\tPartitionConfig:                 ToTaskListPartitionConfig(t.PartitionConfig),\n\t\tLoadBalancerHints:               ToLoadBalancerHints(t.LoadBalancerHints),\n\t\tAutoConfigHint:                  ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc FromMatchingPollForDecisionTaskRequest(t *types.MatchingPollForDecisionTaskRequest) *matchingv1.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.PollForDecisionTaskRequest{\n\t\tRequest:        FromPollForDecisionTaskRequest(t.PollRequest),\n\t\tDomainId:       t.DomainUUID,\n\t\tPollerId:       t.PollerID,\n\t\tForwardedFrom:  t.ForwardedFrom,\n\t\tIsolationGroup: t.IsolationGroup,\n\t}\n}\n\nfunc ToMatchingPollForDecisionTaskRequest(t *matchingv1.PollForDecisionTaskRequest) *types.MatchingPollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForDecisionTaskRequest{\n\t\tPollRequest:    ToPollForDecisionTaskRequest(t.Request),\n\t\tDomainUUID:     t.DomainId,\n\t\tPollerID:       t.PollerId,\n\t\tForwardedFrom:  t.ForwardedFrom,\n\t\tIsolationGroup: t.IsolationGroup,\n\t}\n}\n\nfunc FromMatchingPollForDecisionTaskResponse(t *types.MatchingPollForDecisionTaskResponse) *matchingv1.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    fromInt64Value(t.PreviousStartedEventID),\n\t\tStartedEventId:            t.StartedEventID,\n\t\tAttempt:                   int32(t.Attempt),\n\t\tNextEventId:               t.NextEventID,\n\t\tBacklogCountHint:          t.BacklogCountHint,\n\t\tStickyExecutionEnabled:    t.StickyExecutionEnabled,\n\t\tQuery:                     FromWorkflowQuery(t.Query),\n\t\tDecisionInfo:              FromTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTime:             unixNanoToTime(t.ScheduledTimestamp),\n\t\tStartedTime:               unixNanoToTime(t.StartedTimestamp),\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tTotalHistoryBytes:         t.TotalHistoryBytes,\n\t\tPartitionConfig:           FromTaskListPartitionConfig(t.PartitionConfig),\n\t\tLoadBalancerHints:         FromLoadBalancerHints(t.LoadBalancerHints),\n\t\tAutoConfigHint:            FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc ToMatchingPollForDecisionTaskResponse(t *matchingv1.PollForDecisionTaskResponse) *types.MatchingPollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    toInt64Value(t.PreviousStartedEventId),\n\t\tStartedEventID:            t.StartedEventId,\n\t\tAttempt:                   int64(t.Attempt),\n\t\tNextEventID:               t.NextEventId,\n\t\tBacklogCountHint:          t.BacklogCountHint,\n\t\tStickyExecutionEnabled:    t.StickyExecutionEnabled,\n\t\tQuery:                     ToWorkflowQuery(t.Query),\n\t\tDecisionInfo:              ToTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        timeToUnixNano(t.ScheduledTime),\n\t\tStartedTimestamp:          timeToUnixNano(t.StartedTime),\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tTotalHistoryBytes:         t.TotalHistoryBytes,\n\t\tPartitionConfig:           ToTaskListPartitionConfig(t.PartitionConfig),\n\t\tLoadBalancerHints:         ToLoadBalancerHints(t.LoadBalancerHints),\n\t\tAutoConfigHint:            ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc FromMatchingQueryWorkflowRequest(t *types.MatchingQueryWorkflowRequest) *matchingv1.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.QueryWorkflowRequest{\n\t\tRequest:       FromQueryWorkflowRequest(t.QueryRequest),\n\t\tDomainId:      t.DomainUUID,\n\t\tTaskList:      FromTaskList(t.TaskList),\n\t\tForwardedFrom: t.ForwardedFrom,\n\t}\n}\n\nfunc ToMatchingQueryWorkflowRequest(t *matchingv1.QueryWorkflowRequest) *types.MatchingQueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingQueryWorkflowRequest{\n\t\tQueryRequest:  ToQueryWorkflowRequest(t.Request),\n\t\tDomainUUID:    t.DomainId,\n\t\tTaskList:      ToTaskList(t.TaskList),\n\t\tForwardedFrom: t.ForwardedFrom,\n\t}\n}\n\nfunc FromMatchingQueryWorkflowResponse(t *types.MatchingQueryWorkflowResponse) *matchingv1.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.QueryWorkflowResponse{\n\t\tQueryResult:     FromPayload(t.QueryResult),\n\t\tQueryRejected:   FromQueryRejected(t.QueryRejected),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc ToMatchingQueryWorkflowResponse(t *matchingv1.QueryWorkflowResponse) *types.MatchingQueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingQueryWorkflowResponse{\n\t\tQueryResult:     ToPayload(t.QueryResult),\n\t\tQueryRejected:   ToQueryRejected(t.QueryRejected),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc FromMatchingRespondQueryTaskCompletedRequest(t *types.MatchingRespondQueryTaskCompletedRequest) *matchingv1.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.RespondQueryTaskCompletedRequest{\n\t\tRequest:  FromRespondQueryTaskCompletedRequest(t.CompletedRequest),\n\t\tDomainId: t.DomainUUID,\n\t\tTaskList: FromTaskList(t.TaskList),\n\t\tTaskId:   t.TaskID,\n\t}\n}\n\nfunc ToMatchingRespondQueryTaskCompletedRequest(t *matchingv1.RespondQueryTaskCompletedRequest) *types.MatchingRespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingRespondQueryTaskCompletedRequest{\n\t\tCompletedRequest: ToRespondQueryTaskCompletedRequest(t.Request),\n\t\tDomainUUID:       t.DomainId,\n\t\tTaskList:         ToTaskList(t.TaskList),\n\t\tTaskID:           t.TaskId,\n\t}\n}\n\nfunc FromMatchingUpdateTaskListPartitionConfigRequest(t *types.MatchingUpdateTaskListPartitionConfigRequest) *matchingv1.UpdateTaskListPartitionConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.UpdateTaskListPartitionConfigRequest{\n\t\tDomainId:        t.DomainUUID,\n\t\tTaskList:        FromTaskList(t.TaskList),\n\t\tTaskListType:    FromTaskListType(t.TaskListType),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc ToMatchingUpdateTaskListPartitionConfigRequest(t *matchingv1.UpdateTaskListPartitionConfigRequest) *types.MatchingUpdateTaskListPartitionConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\tDomainUUID:      t.DomainId,\n\t\tTaskList:        ToTaskList(t.TaskList),\n\t\tTaskListType:    ToTaskListType(t.TaskListType),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc FromMatchingRefreshTaskListPartitionConfigRequest(t *types.MatchingRefreshTaskListPartitionConfigRequest) *matchingv1.RefreshTaskListPartitionConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.RefreshTaskListPartitionConfigRequest{\n\t\tDomainId:        t.DomainUUID,\n\t\tTaskList:        FromTaskList(t.TaskList),\n\t\tTaskListType:    FromTaskListType(t.TaskListType),\n\t\tPartitionConfig: FromAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc ToMatchingRefreshTaskListPartitionConfigRequest(t *matchingv1.RefreshTaskListPartitionConfigRequest) *types.MatchingRefreshTaskListPartitionConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\tDomainUUID:      t.DomainId,\n\t\tTaskList:        ToTaskList(t.TaskList),\n\t\tTaskListType:    ToTaskListType(t.TaskListType),\n\t\tPartitionConfig: ToAPITaskListPartitionConfig(t.PartitionConfig),\n\t}\n}\n\nfunc FromMatchingUpdateTaskListPartitionConfigResponse(t *types.MatchingUpdateTaskListPartitionConfigResponse) *matchingv1.UpdateTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.UpdateTaskListPartitionConfigResponse{}\n}\n\nfunc ToMatchingUpdateTaskListPartitionConfigResponse(t *matchingv1.UpdateTaskListPartitionConfigResponse) *types.MatchingUpdateTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingUpdateTaskListPartitionConfigResponse{}\n}\n\nfunc FromMatchingRefreshTaskListPartitionConfigResponse(t *types.MatchingRefreshTaskListPartitionConfigResponse) *matchingv1.RefreshTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.RefreshTaskListPartitionConfigResponse{}\n}\n\nfunc ToMatchingRefreshTaskListPartitionConfigResponse(t *matchingv1.RefreshTaskListPartitionConfigResponse) *types.MatchingRefreshTaskListPartitionConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingRefreshTaskListPartitionConfigResponse{}\n}\n\nfunc FromLoadBalancerHints(t *types.LoadBalancerHints) *matchingv1.LoadBalancerHints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matchingv1.LoadBalancerHints{\n\t\tBacklogCount:  t.BacklogCount,\n\t\tRatePerSecond: t.RatePerSecond,\n\t}\n}\n\nfunc ToLoadBalancerHints(t *matchingv1.LoadBalancerHints) *types.LoadBalancerHints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.LoadBalancerHints{\n\t\tBacklogCount:  t.BacklogCount,\n\t\tRatePerSecond: t.RatePerSecond,\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/matching_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestMatchingAddActivityTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.AddActivityTaskRequest{nil, {}, &testdata.MatchingAddActivityTaskRequest} {\n\t\tassert.Equal(t, item, ToMatchingAddActivityTaskRequest(FromMatchingAddActivityTaskRequest(item)))\n\t}\n}\n\nfunc TestMatchingAddDecisionTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.AddDecisionTaskRequest{nil, {}, &testdata.MatchingAddDecisionTaskRequest} {\n\t\tassert.Equal(t, item, ToMatchingAddDecisionTaskRequest(FromMatchingAddDecisionTaskRequest(item)))\n\t}\n}\n\nfunc TestMatchingAddActivityTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.AddActivityTaskResponse{nil, {}, &testdata.MatchingAddActivityTaskResponse} {\n\t\tassert.Equal(t, item, ToMatchingAddActivityTaskResponse(FromMatchingAddActivityTaskResponse(item)))\n\t}\n}\n\nfunc TestMatchingAddDecisionTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.AddDecisionTaskResponse{nil, {}, &testdata.MatchingAddDecisionTaskResponse} {\n\t\tassert.Equal(t, item, ToMatchingAddDecisionTaskResponse(FromMatchingAddDecisionTaskResponse(item)))\n\t}\n}\n\nfunc TestMatchingCancelOutstandingPollRequest(t *testing.T) {\n\tfor _, item := range []*types.CancelOutstandingPollRequest{nil, {}, &testdata.MatchingCancelOutstandingPollRequest} {\n\t\tassert.Equal(t, item, ToMatchingCancelOutstandingPollRequest(FromMatchingCancelOutstandingPollRequest(item)))\n\t}\n}\n\nfunc TestMatchingDescribeTaskListRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingDescribeTaskListRequest{nil, {}, &testdata.MatchingDescribeTaskListRequest} {\n\t\tassert.Equal(t, item, ToMatchingDescribeTaskListRequest(FromMatchingDescribeTaskListRequest(item)))\n\t}\n}\n\nfunc TestMatchingDescribeTaskListResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeTaskListResponse{nil, {}, &testdata.MatchingDescribeTaskListResponse} {\n\t\tassert.Equal(t, item, ToMatchingDescribeTaskListResponse(FromMatchingDescribeTaskListResponse(item)))\n\t}\n}\n\nfunc TestMatchingDescribeTaskListResponseMap(t *testing.T) {\n\tfor _, item := range []map[string]*types.DescribeTaskListResponse{nil, {}, testdata.DescribeTaskListResponseMap} {\n\t\tassert.Equal(t, item, ToMatchingDescribeTaskListResponseMap(FromMatchingDescribeTaskListResponseMap(item)))\n\t}\n}\n\nfunc TestMatchingListTaskListPartitionsRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingListTaskListPartitionsRequest{nil, {}, &testdata.MatchingListTaskListPartitionsRequest} {\n\t\tassert.Equal(t, item, ToMatchingListTaskListPartitionsRequest(FromMatchingListTaskListPartitionsRequest(item)))\n\t}\n}\n\nfunc TestMatchingListTaskListPartitionsResponse(t *testing.T) {\n\tfor _, item := range []*types.ListTaskListPartitionsResponse{nil, {}, &testdata.MatchingListTaskListPartitionsResponse} {\n\t\tassert.Equal(t, item, ToMatchingListTaskListPartitionsResponse(FromMatchingListTaskListPartitionsResponse(item)))\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingPollForActivityTaskRequest{nil, {}, &testdata.MatchingPollForActivityTaskRequest} {\n\t\tassert.Equal(t, item, ToMatchingPollForActivityTaskRequest(FromMatchingPollForActivityTaskRequest(item)))\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.MatchingPollForActivityTaskResponse{nil, {}, &testdata.MatchingPollForActivityTaskResponse} {\n\t\tassert.Equal(t, item, ToMatchingPollForActivityTaskResponse(FromMatchingPollForActivityTaskResponse(item)))\n\t}\n}\n\nfunc TestTaskListPartitionConfig(t *testing.T) {\n\tfor _, item := range []*types.TaskListPartitionConfig{nil, {}, &testdata.TaskListPartitionConfig} {\n\t\tassert.Equal(t, item, ToTaskListPartitionConfig(FromTaskListPartitionConfig(item)))\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingPollForDecisionTaskRequest{nil, {}, &testdata.MatchingPollForDecisionTaskRequest} {\n\t\tassert.Equal(t, item, ToMatchingPollForDecisionTaskRequest(FromMatchingPollForDecisionTaskRequest(item)))\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.MatchingPollForDecisionTaskResponse{nil, {}, &testdata.MatchingPollForDecisionTaskResponse} {\n\t\tassert.Equal(t, item, ToMatchingPollForDecisionTaskResponse(FromMatchingPollForDecisionTaskResponse(item)))\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingQueryWorkflowRequest{nil, {}, &testdata.MatchingQueryWorkflowRequest} {\n\t\tassert.Equal(t, item, ToMatchingQueryWorkflowRequest(FromMatchingQueryWorkflowRequest(item)))\n\t}\n}\n\nfunc TestMatchingQueryWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.MatchingQueryWorkflowResponse{nil, {}, &testdata.MatchingQueryWorkflowResponse} {\n\t\tassert.Equal(t, item, ToMatchingQueryWorkflowResponse(FromMatchingQueryWorkflowResponse(item)))\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingRespondQueryTaskCompletedRequest{nil, {}, &testdata.MatchingRespondQueryTaskCompletedRequest} {\n\t\tassert.Equal(t, item, ToMatchingRespondQueryTaskCompletedRequest(FromMatchingRespondQueryTaskCompletedRequest(item)))\n\t}\n}\n\nfunc TestMatchingGetTaskListsByDomainRequest(t *testing.T) {\n\tfor _, item := range []*types.GetTaskListsByDomainRequest{nil, {}, &testdata.MatchingGetTaskListsByDomainRequest} {\n\t\tassert.Equal(t, item, ToMatchingGetTaskListsByDomainRequest(FromMatchingGetTaskListsByDomainRequest(item)))\n\t}\n}\n\nfunc TestMatchingGetTaskListsByDomainResponse(t *testing.T) {\n\tfor _, item := range []*types.GetTaskListsByDomainResponse{nil, {}, &testdata.GetTaskListsByDomainResponse} {\n\t\tassert.Equal(t, item, ToMatchingGetTaskListsByDomainResponse(FromMatchingGetTaskListsByDomainResponse(item)))\n\t}\n}\n\nfunc TestMatchingUpdateTaskListPartitionConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingUpdateTaskListPartitionConfigRequest{nil, {}, &testdata.MatchingUpdateTaskListPartitionConfigRequest} {\n\t\tassert.Equal(t, item, ToMatchingUpdateTaskListPartitionConfigRequest(FromMatchingUpdateTaskListPartitionConfigRequest(item)))\n\t}\n}\n\nfunc TestMatchingUpdateTaskListPartitionConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.MatchingUpdateTaskListPartitionConfigResponse{nil, {}} {\n\t\tassert.Equal(t, item, ToMatchingUpdateTaskListPartitionConfigResponse(FromMatchingUpdateTaskListPartitionConfigResponse(item)))\n\t}\n}\n\nfunc TestMatchingRefreshTaskListPartitionConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.MatchingRefreshTaskListPartitionConfigRequest{nil, {}, &testdata.MatchingRefreshTaskListPartitionConfigRequest} {\n\t\tassert.Equal(t, item, ToMatchingRefreshTaskListPartitionConfigRequest(FromMatchingRefreshTaskListPartitionConfigRequest(item)))\n\t}\n}\n\nfunc TestMatchingRefreshTaskListPartitionConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.MatchingRefreshTaskListPartitionConfigResponse{nil, {}} {\n\t\tassert.Equal(t, item, ToMatchingRefreshTaskListPartitionConfigResponse(FromMatchingRefreshTaskListPartitionConfigResponse(item)))\n\t}\n}\n\nfunc TestToMatchingTaskListPartitionConfig(t *testing.T) {\n\tcases := []struct {\n\t\tname     string\n\t\tconfig   *matchingv1.TaskListPartitionConfig\n\t\texpected *types.TaskListPartitionConfig\n\t}{\n\t\t{\n\t\t\tname: \"happy path\",\n\t\t\tconfig: &matchingv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t\tReadPartitions: map[int32]*matchingv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int32]*matchingv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"numbers only\",\n\t\t\tconfig: &matchingv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 2,\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t\t1: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"number mismatch\",\n\t\t\tconfig: &matchingv1.TaskListPartitionConfig{\n\t\t\t\tVersion:            1,\n\t\t\t\tNumReadPartitions:  2,\n\t\t\t\tNumWritePartitions: 1,\n\t\t\t\tReadPartitions: map[int32]*matchingv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int32]*matchingv1.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"baz\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.TaskListPartitionConfig{\n\t\t\t\tVersion: 1,\n\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {IsolationGroups: []string{\"foo\"}},\n\t\t\t\t\t1: {IsolationGroups: []string{\"bar\"}},\n\t\t\t\t},\n\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t0: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tactual := ToTaskListPartitionConfig(tc.config)\n\t\t\tassert.Equal(t, tc.expected, actual)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/schedule.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage proto\n\nimport (\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// --- Enum mappers ---\n\nfunc FromScheduleOverlapPolicy(p types.ScheduleOverlapPolicy) apiv1.ScheduleOverlapPolicy {\n\tswitch p {\n\tcase types.ScheduleOverlapPolicySkipNew:\n\t\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_SKIP_NEW\n\tcase types.ScheduleOverlapPolicyBuffer:\n\t\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_BUFFER\n\tcase types.ScheduleOverlapPolicyConcurrent:\n\t\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_CONCURRENT\n\tcase types.ScheduleOverlapPolicyCancelPrevious:\n\t\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_CANCEL_PREVIOUS\n\tcase types.ScheduleOverlapPolicyTerminatePrevious:\n\t\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_TERMINATE_PREVIOUS\n\t}\n\treturn apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_INVALID\n}\n\nfunc ToScheduleOverlapPolicy(p apiv1.ScheduleOverlapPolicy) types.ScheduleOverlapPolicy {\n\tswitch p {\n\tcase apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_SKIP_NEW:\n\t\treturn types.ScheduleOverlapPolicySkipNew\n\tcase apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_BUFFER:\n\t\treturn types.ScheduleOverlapPolicyBuffer\n\tcase apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_CONCURRENT:\n\t\treturn types.ScheduleOverlapPolicyConcurrent\n\tcase apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_CANCEL_PREVIOUS:\n\t\treturn types.ScheduleOverlapPolicyCancelPrevious\n\tcase apiv1.ScheduleOverlapPolicy_SCHEDULE_OVERLAP_POLICY_TERMINATE_PREVIOUS:\n\t\treturn types.ScheduleOverlapPolicyTerminatePrevious\n\t}\n\treturn types.ScheduleOverlapPolicyInvalid\n}\n\nfunc FromScheduleCatchUpPolicy(p types.ScheduleCatchUpPolicy) apiv1.ScheduleCatchUpPolicy {\n\tswitch p {\n\tcase types.ScheduleCatchUpPolicySkip:\n\t\treturn apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_SKIP\n\tcase types.ScheduleCatchUpPolicyOne:\n\t\treturn apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_ONE\n\tcase types.ScheduleCatchUpPolicyAll:\n\t\treturn apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_ALL\n\t}\n\treturn apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_INVALID\n}\n\nfunc ToScheduleCatchUpPolicy(p apiv1.ScheduleCatchUpPolicy) types.ScheduleCatchUpPolicy {\n\tswitch p {\n\tcase apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_SKIP:\n\t\treturn types.ScheduleCatchUpPolicySkip\n\tcase apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_ONE:\n\t\treturn types.ScheduleCatchUpPolicyOne\n\tcase apiv1.ScheduleCatchUpPolicy_SCHEDULE_CATCH_UP_POLICY_ALL:\n\t\treturn types.ScheduleCatchUpPolicyAll\n\t}\n\treturn types.ScheduleCatchUpPolicyInvalid\n}\n\n// --- Core type mappers ---\n\nfunc FromScheduleSpec(t *types.ScheduleSpec) *apiv1.ScheduleSpec {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleSpec{\n\t\tCronExpression: t.CronExpression,\n\t\tStartTime:      timeToTimestamp(&t.StartTime),\n\t\tEndTime:        timeToTimestamp(&t.EndTime),\n\t\tJitter:         durationToDurationProto(t.Jitter),\n\t}\n}\n\nfunc ToScheduleSpec(t *apiv1.ScheduleSpec) *types.ScheduleSpec {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleSpec{\n\t\tCronExpression: t.CronExpression,\n\t\tStartTime:      timestampToTimeVal(t.StartTime),\n\t\tEndTime:        timestampToTimeVal(t.EndTime),\n\t\tJitter:         durationProtoToDuration(t.Jitter),\n\t}\n}\n\nfunc FromStartWorkflowAction(t *types.StartWorkflowAction) *apiv1.ScheduleAction_StartWorkflowAction {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleAction_StartWorkflowAction{\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                     FromTaskList(t.TaskList),\n\t\tInput:                        FromPayload(t.Input),\n\t\tWorkflowIdPrefix:             t.WorkflowIDPrefix,\n\t\tExecutionStartToCloseTimeout: secondsToDuration(t.ExecutionStartToCloseTimeoutSeconds),\n\t\tTaskStartToCloseTimeout:      secondsToDuration(t.TaskStartToCloseTimeoutSeconds),\n\t\tRetryPolicy:                  FromRetryPolicy(t.RetryPolicy),\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToStartWorkflowAction(t *apiv1.ScheduleAction_StartWorkflowAction) *types.StartWorkflowAction {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowAction{\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               ToPayload(t.Input),\n\t\tWorkflowIDPrefix:                    t.WorkflowIdPrefix,\n\t\tExecutionStartToCloseTimeoutSeconds: durationToSeconds(t.ExecutionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      durationToSeconds(t.TaskStartToCloseTimeout),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromScheduleAction(t *types.ScheduleAction) *apiv1.ScheduleAction {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleAction{\n\t\tStartWorkflow: FromStartWorkflowAction(t.StartWorkflow),\n\t}\n}\n\nfunc ToScheduleAction(t *apiv1.ScheduleAction) *types.ScheduleAction {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleAction{\n\t\tStartWorkflow: ToStartWorkflowAction(t.StartWorkflow),\n\t}\n}\n\nfunc FromSchedulePolicies(t *types.SchedulePolicies) *apiv1.SchedulePolicies {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SchedulePolicies{\n\t\tOverlapPolicy:    FromScheduleOverlapPolicy(t.OverlapPolicy),\n\t\tCatchUpPolicy:    FromScheduleCatchUpPolicy(t.CatchUpPolicy),\n\t\tCatchUpWindow:    durationToDurationProto(t.CatchUpWindow),\n\t\tPauseOnFailure:   t.PauseOnFailure,\n\t\tBufferLimit:      t.BufferLimit,\n\t\tConcurrencyLimit: t.ConcurrencyLimit,\n\t}\n}\n\nfunc ToSchedulePolicies(t *apiv1.SchedulePolicies) *types.SchedulePolicies {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SchedulePolicies{\n\t\tOverlapPolicy:    ToScheduleOverlapPolicy(t.OverlapPolicy),\n\t\tCatchUpPolicy:    ToScheduleCatchUpPolicy(t.CatchUpPolicy),\n\t\tCatchUpWindow:    durationProtoToDuration(t.CatchUpWindow),\n\t\tPauseOnFailure:   t.PauseOnFailure,\n\t\tBufferLimit:      t.BufferLimit,\n\t\tConcurrencyLimit: t.ConcurrencyLimit,\n\t}\n}\n\n// --- State/info type mappers ---\n\nfunc FromSchedulePauseInfo(t *types.SchedulePauseInfo) *apiv1.SchedulePauseInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.SchedulePauseInfo{\n\t\tReason:   t.Reason,\n\t\tPausedAt: timeToTimestamp(&t.PausedAt),\n\t\tPausedBy: t.PausedBy,\n\t}\n}\n\nfunc ToSchedulePauseInfo(t *apiv1.SchedulePauseInfo) *types.SchedulePauseInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SchedulePauseInfo{\n\t\tReason:   t.Reason,\n\t\tPausedAt: timestampToTimeVal(t.PausedAt),\n\t\tPausedBy: t.PausedBy,\n\t}\n}\n\nfunc FromScheduleState(t *types.ScheduleState) *apiv1.ScheduleState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleState{\n\t\tPaused:    t.Paused,\n\t\tPauseInfo: FromSchedulePauseInfo(t.PauseInfo),\n\t}\n}\n\nfunc ToScheduleState(t *apiv1.ScheduleState) *types.ScheduleState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleState{\n\t\tPaused:    t.Paused,\n\t\tPauseInfo: ToSchedulePauseInfo(t.PauseInfo),\n\t}\n}\n\nfunc FromBackfillInfo(t *types.BackfillInfo) *apiv1.BackfillInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.BackfillInfo{\n\t\tBackfillId:    t.BackfillID,\n\t\tStartTime:     timeToTimestamp(&t.StartTime),\n\t\tEndTime:       timeToTimestamp(&t.EndTime),\n\t\tRunsCompleted: t.RunsCompleted,\n\t\tRunsTotal:     t.RunsTotal,\n\t}\n}\n\nfunc ToBackfillInfo(t *apiv1.BackfillInfo) *types.BackfillInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BackfillInfo{\n\t\tBackfillID:    t.BackfillId,\n\t\tStartTime:     timestampToTimeVal(t.StartTime),\n\t\tEndTime:       timestampToTimeVal(t.EndTime),\n\t\tRunsCompleted: t.RunsCompleted,\n\t\tRunsTotal:     t.RunsTotal,\n\t}\n}\n\nfunc FromBackfillInfoArray(t []*types.BackfillInfo) []*apiv1.BackfillInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.BackfillInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromBackfillInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToBackfillInfoArray(t []*apiv1.BackfillInfo) []*types.BackfillInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.BackfillInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToBackfillInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromScheduleInfo(t *types.ScheduleInfo) *apiv1.ScheduleInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleInfo{\n\t\tLastRunTime:      timeToTimestamp(&t.LastRunTime),\n\t\tNextRunTime:      timeToTimestamp(&t.NextRunTime),\n\t\tTotalRuns:        t.TotalRuns,\n\t\tCreateTime:       timeToTimestamp(&t.CreateTime),\n\t\tLastUpdateTime:   timeToTimestamp(&t.LastUpdateTime),\n\t\tOngoingBackfills: FromBackfillInfoArray(t.OngoingBackfills),\n\t}\n}\n\nfunc ToScheduleInfo(t *apiv1.ScheduleInfo) *types.ScheduleInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleInfo{\n\t\tLastRunTime:      timestampToTimeVal(t.LastRunTime),\n\t\tNextRunTime:      timestampToTimeVal(t.NextRunTime),\n\t\tTotalRuns:        t.TotalRuns,\n\t\tCreateTime:       timestampToTimeVal(t.CreateTime),\n\t\tLastUpdateTime:   timestampToTimeVal(t.LastUpdateTime),\n\t\tOngoingBackfills: ToBackfillInfoArray(t.OngoingBackfills),\n\t}\n}\n\nfunc FromScheduleListEntry(t *types.ScheduleListEntry) *apiv1.ScheduleListEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ScheduleListEntry{\n\t\tScheduleId:     t.ScheduleID,\n\t\tWorkflowType:   FromWorkflowType(t.WorkflowType),\n\t\tState:          FromScheduleState(t.State),\n\t\tCronExpression: t.CronExpression,\n\t}\n}\n\nfunc ToScheduleListEntry(t *apiv1.ScheduleListEntry) *types.ScheduleListEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleListEntry{\n\t\tScheduleID:     t.ScheduleId,\n\t\tWorkflowType:   ToWorkflowType(t.WorkflowType),\n\t\tState:          ToScheduleState(t.State),\n\t\tCronExpression: t.CronExpression,\n\t}\n}\n\nfunc FromScheduleListEntryArray(t []*types.ScheduleListEntry) []*apiv1.ScheduleListEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*apiv1.ScheduleListEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = FromScheduleListEntry(t[i])\n\t}\n\treturn v\n}\n\nfunc ToScheduleListEntryArray(t []*apiv1.ScheduleListEntry) []*types.ScheduleListEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ScheduleListEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = ToScheduleListEntry(t[i])\n\t}\n\treturn v\n}\n\n// --- CRUD request/response mappers ---\n\nfunc FromCreateScheduleRequest(t *types.CreateScheduleRequest) *apiv1.CreateScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CreateScheduleRequest{\n\t\tDomain:           t.Domain,\n\t\tScheduleId:       t.ScheduleID,\n\t\tSpec:             FromScheduleSpec(t.Spec),\n\t\tAction:           FromScheduleAction(t.Action),\n\t\tPolicies:         FromSchedulePolicies(t.Policies),\n\t\tMemo:             FromMemo(t.Memo),\n\t\tSearchAttributes: FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToCreateScheduleRequest(t *apiv1.CreateScheduleRequest) *types.CreateScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CreateScheduleRequest{\n\t\tDomain:           t.Domain,\n\t\tScheduleID:       t.ScheduleId,\n\t\tSpec:             ToScheduleSpec(t.Spec),\n\t\tAction:           ToScheduleAction(t.Action),\n\t\tPolicies:         ToSchedulePolicies(t.Policies),\n\t\tMemo:             ToMemo(t.Memo),\n\t\tSearchAttributes: ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromCreateScheduleResponse(t *types.CreateScheduleResponse) *apiv1.CreateScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.CreateScheduleResponse{}\n}\n\nfunc ToCreateScheduleResponse(t *apiv1.CreateScheduleResponse) *types.CreateScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CreateScheduleResponse{}\n}\n\nfunc FromDescribeScheduleRequest(t *types.DescribeScheduleRequest) *apiv1.DescribeScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleId: t.ScheduleID,\n\t}\n}\n\nfunc ToDescribeScheduleRequest(t *apiv1.DescribeScheduleRequest) *types.DescribeScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleID: t.ScheduleId,\n\t}\n}\n\nfunc FromDescribeScheduleResponse(t *types.DescribeScheduleResponse) *apiv1.DescribeScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DescribeScheduleResponse{\n\t\tSpec:             FromScheduleSpec(t.Spec),\n\t\tAction:           FromScheduleAction(t.Action),\n\t\tPolicies:         FromSchedulePolicies(t.Policies),\n\t\tState:            FromScheduleState(t.State),\n\t\tInfo:             FromScheduleInfo(t.Info),\n\t\tMemo:             FromMemo(t.Memo),\n\t\tSearchAttributes: FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToDescribeScheduleResponse(t *apiv1.DescribeScheduleResponse) *types.DescribeScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeScheduleResponse{\n\t\tSpec:             ToScheduleSpec(t.Spec),\n\t\tAction:           ToScheduleAction(t.Action),\n\t\tPolicies:         ToSchedulePolicies(t.Policies),\n\t\tState:            ToScheduleState(t.State),\n\t\tInfo:             ToScheduleInfo(t.Info),\n\t\tMemo:             ToMemo(t.Memo),\n\t\tSearchAttributes: ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromUpdateScheduleRequest(t *types.UpdateScheduleRequest) *apiv1.UpdateScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UpdateScheduleRequest{\n\t\tDomain:           t.Domain,\n\t\tScheduleId:       t.ScheduleID,\n\t\tSpec:             FromScheduleSpec(t.Spec),\n\t\tAction:           FromScheduleAction(t.Action),\n\t\tPolicies:         FromSchedulePolicies(t.Policies),\n\t\tSearchAttributes: FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc ToUpdateScheduleRequest(t *apiv1.UpdateScheduleRequest) *types.UpdateScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateScheduleRequest{\n\t\tDomain:           t.Domain,\n\t\tScheduleID:       t.ScheduleId,\n\t\tSpec:             ToScheduleSpec(t.Spec),\n\t\tAction:           ToScheduleAction(t.Action),\n\t\tPolicies:         ToSchedulePolicies(t.Policies),\n\t\tSearchAttributes: ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\nfunc FromUpdateScheduleResponse(t *types.UpdateScheduleResponse) *apiv1.UpdateScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UpdateScheduleResponse{}\n}\n\nfunc ToUpdateScheduleResponse(t *apiv1.UpdateScheduleResponse) *types.UpdateScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateScheduleResponse{}\n}\n\nfunc FromDeleteScheduleRequest(t *types.DeleteScheduleRequest) *apiv1.DeleteScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DeleteScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleId: t.ScheduleID,\n\t}\n}\n\nfunc ToDeleteScheduleRequest(t *apiv1.DeleteScheduleRequest) *types.DeleteScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeleteScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleID: t.ScheduleId,\n\t}\n}\n\nfunc FromDeleteScheduleResponse(t *types.DeleteScheduleResponse) *apiv1.DeleteScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.DeleteScheduleResponse{}\n}\n\nfunc ToDeleteScheduleResponse(t *apiv1.DeleteScheduleResponse) *types.DeleteScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeleteScheduleResponse{}\n}\n\n// --- Action request/response mappers ---\n\nfunc FromPauseScheduleRequest(t *types.PauseScheduleRequest) *apiv1.PauseScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PauseScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleId: t.ScheduleID,\n\t\tReason:     t.Reason,\n\t}\n}\n\nfunc ToPauseScheduleRequest(t *apiv1.PauseScheduleRequest) *types.PauseScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PauseScheduleRequest{\n\t\tDomain:     t.Domain,\n\t\tScheduleID: t.ScheduleId,\n\t\tReason:     t.Reason,\n\t}\n}\n\nfunc FromPauseScheduleResponse(t *types.PauseScheduleResponse) *apiv1.PauseScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.PauseScheduleResponse{}\n}\n\nfunc ToPauseScheduleResponse(t *apiv1.PauseScheduleResponse) *types.PauseScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PauseScheduleResponse{}\n}\n\nfunc FromUnpauseScheduleRequest(t *types.UnpauseScheduleRequest) *apiv1.UnpauseScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UnpauseScheduleRequest{\n\t\tDomain:        t.Domain,\n\t\tScheduleId:    t.ScheduleID,\n\t\tReason:        t.Reason,\n\t\tCatchUpPolicy: FromScheduleCatchUpPolicy(t.CatchUpPolicy),\n\t}\n}\n\nfunc ToUnpauseScheduleRequest(t *apiv1.UnpauseScheduleRequest) *types.UnpauseScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UnpauseScheduleRequest{\n\t\tDomain:        t.Domain,\n\t\tScheduleID:    t.ScheduleId,\n\t\tReason:        t.Reason,\n\t\tCatchUpPolicy: ToScheduleCatchUpPolicy(t.CatchUpPolicy),\n\t}\n}\n\nfunc FromUnpauseScheduleResponse(t *types.UnpauseScheduleResponse) *apiv1.UnpauseScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.UnpauseScheduleResponse{}\n}\n\nfunc ToUnpauseScheduleResponse(t *apiv1.UnpauseScheduleResponse) *types.UnpauseScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UnpauseScheduleResponse{}\n}\n\nfunc FromListSchedulesRequest(t *types.ListSchedulesRequest) *apiv1.ListSchedulesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListSchedulesRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListSchedulesRequest(t *apiv1.ListSchedulesRequest) *types.ListSchedulesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListSchedulesRequest{\n\t\tDomain:        t.Domain,\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromListSchedulesResponse(t *types.ListSchedulesResponse) *apiv1.ListSchedulesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.ListSchedulesResponse{\n\t\tSchedules:     FromScheduleListEntryArray(t.Schedules),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc ToListSchedulesResponse(t *apiv1.ListSchedulesResponse) *types.ListSchedulesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListSchedulesResponse{\n\t\tSchedules:     ToScheduleListEntryArray(t.Schedules),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\nfunc FromBackfillScheduleRequest(t *types.BackfillScheduleRequest) *apiv1.BackfillScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.BackfillScheduleRequest{\n\t\tDomain:        t.Domain,\n\t\tScheduleId:    t.ScheduleID,\n\t\tStartTime:     timeToTimestamp(&t.StartTime),\n\t\tEndTime:       timeToTimestamp(&t.EndTime),\n\t\tOverlapPolicy: FromScheduleOverlapPolicy(t.OverlapPolicy),\n\t\tBackfillId:    t.BackfillID,\n\t}\n}\n\nfunc ToBackfillScheduleRequest(t *apiv1.BackfillScheduleRequest) *types.BackfillScheduleRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BackfillScheduleRequest{\n\t\tDomain:        t.Domain,\n\t\tScheduleID:    t.ScheduleId,\n\t\tStartTime:     timestampToTimeVal(t.StartTime),\n\t\tEndTime:       timestampToTimeVal(t.EndTime),\n\t\tOverlapPolicy: ToScheduleOverlapPolicy(t.OverlapPolicy),\n\t\tBackfillID:    t.BackfillId,\n\t}\n}\n\nfunc FromBackfillScheduleResponse(t *types.BackfillScheduleResponse) *apiv1.BackfillScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &apiv1.BackfillScheduleResponse{}\n}\n\nfunc ToBackfillScheduleResponse(t *apiv1.BackfillScheduleResponse) *types.BackfillScheduleResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BackfillScheduleResponse{}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/schedule_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/testutils\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestScheduleSpec(t *testing.T) {\n\tfor _, item := range []*types.ScheduleSpec{nil, {}, &testdata.ScheduleSpec} {\n\t\tassert.Equal(t, item, ToScheduleSpec(FromScheduleSpec(item)))\n\t}\n}\n\nfunc TestStartWorkflowAction(t *testing.T) {\n\tfor _, item := range []*types.StartWorkflowAction{nil, {}, &testdata.ScheduleStartWorkflowAction} {\n\t\tassert.Equal(t, item, ToStartWorkflowAction(FromStartWorkflowAction(item)))\n\t}\n}\n\nfunc TestScheduleAction(t *testing.T) {\n\tfor _, item := range []*types.ScheduleAction{nil, {}, &testdata.ScheduleAction} {\n\t\tassert.Equal(t, item, ToScheduleAction(FromScheduleAction(item)))\n\t}\n}\n\nfunc TestSchedulePolicies(t *testing.T) {\n\tfor _, item := range []*types.SchedulePolicies{nil, {}, &testdata.SchedulePolicies} {\n\t\tassert.Equal(t, item, ToSchedulePolicies(FromSchedulePolicies(item)))\n\t}\n}\n\nfunc TestSchedulePauseInfo(t *testing.T) {\n\tfor _, item := range []*types.SchedulePauseInfo{nil, {}, &testdata.SchedulePauseInfo} {\n\t\tassert.Equal(t, item, ToSchedulePauseInfo(FromSchedulePauseInfo(item)))\n\t}\n}\n\nfunc TestScheduleState(t *testing.T) {\n\tfor _, item := range []*types.ScheduleState{nil, {}, &testdata.ScheduleState} {\n\t\tassert.Equal(t, item, ToScheduleState(FromScheduleState(item)))\n\t}\n}\n\nfunc TestBackfillInfo(t *testing.T) {\n\tfor _, item := range []*types.BackfillInfo{nil, {}, &testdata.ScheduleBackfillInfo} {\n\t\tassert.Equal(t, item, ToBackfillInfo(FromBackfillInfo(item)))\n\t}\n}\n\nfunc TestScheduleInfo(t *testing.T) {\n\tfor _, item := range []*types.ScheduleInfo{nil, {}, &testdata.ScheduleInfo} {\n\t\tassert.Equal(t, item, ToScheduleInfo(FromScheduleInfo(item)))\n\t}\n}\n\nfunc TestScheduleListEntry(t *testing.T) {\n\tfor _, item := range []*types.ScheduleListEntry{nil, {}, &testdata.ScheduleListEntry} {\n\t\tassert.Equal(t, item, ToScheduleListEntry(FromScheduleListEntry(item)))\n\t}\n}\n\nfunc TestScheduleSpecFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleSpec, ToScheduleSpec,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestStartWorkflowActionFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromStartWorkflowAction, ToStartWorkflowAction,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleActionFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleAction, ToScheduleAction,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestSchedulePoliciesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSchedulePolicies, ToSchedulePolicies,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestSchedulePauseInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSchedulePauseInfo, ToSchedulePauseInfo,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleStateFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleState, ToScheduleState,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestBackfillInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBackfillInfo, ToBackfillInfo,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleInfo, ToScheduleInfo,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleListEntryFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleListEntry, ToScheduleListEntry,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\n// WithScheduleEnumFuzzers adds fuzzers for Schedule-specific enum types\nfunc WithScheduleEnumFuzzers() testutils.FuzzOption {\n\treturn testutils.WithCustomFuncs(\n\t\tfunc(e *types.ScheduleOverlapPolicy, c fuzz.Continue) {\n\t\t\t*e = types.ScheduleOverlapPolicy(c.Intn(6)) // 0-5: Invalid through TerminatePrevious\n\t\t},\n\t\tfunc(e *types.ScheduleCatchUpPolicy, c fuzz.Continue) {\n\t\t\t*e = types.ScheduleCatchUpPolicy(c.Intn(4)) // 0-3: Invalid through All\n\t\t},\n\t)\n}\n\n// --- CRUD request/response deterministic tests ---\n\nfunc TestCreateScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.CreateScheduleRequest{nil, {}, &testdata.CreateScheduleRequest} {\n\t\tassert.Equal(t, item, ToCreateScheduleRequest(FromCreateScheduleRequest(item)))\n\t}\n}\n\nfunc TestCreateScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.CreateScheduleResponse{nil, {}, &testdata.CreateScheduleResponse} {\n\t\tassert.Equal(t, item, ToCreateScheduleResponse(FromCreateScheduleResponse(item)))\n\t}\n}\n\nfunc TestDescribeScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeScheduleRequest{nil, {}, &testdata.DescribeScheduleRequest} {\n\t\tassert.Equal(t, item, ToDescribeScheduleRequest(FromDescribeScheduleRequest(item)))\n\t}\n}\n\nfunc TestDescribeScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeScheduleResponse{nil, {}, &testdata.DescribeScheduleResponse} {\n\t\tassert.Equal(t, item, ToDescribeScheduleResponse(FromDescribeScheduleResponse(item)))\n\t}\n}\n\nfunc TestUpdateScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateScheduleRequest{nil, {}, &testdata.UpdateScheduleRequest} {\n\t\tassert.Equal(t, item, ToUpdateScheduleRequest(FromUpdateScheduleRequest(item)))\n\t}\n}\n\nfunc TestUpdateScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateScheduleResponse{nil, {}, &testdata.UpdateScheduleResponse} {\n\t\tassert.Equal(t, item, ToUpdateScheduleResponse(FromUpdateScheduleResponse(item)))\n\t}\n}\n\nfunc TestDeleteScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.DeleteScheduleRequest{nil, {}, &testdata.DeleteScheduleRequest} {\n\t\tassert.Equal(t, item, ToDeleteScheduleRequest(FromDeleteScheduleRequest(item)))\n\t}\n}\n\nfunc TestDeleteScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.DeleteScheduleResponse{nil, {}, &testdata.DeleteScheduleResponse} {\n\t\tassert.Equal(t, item, ToDeleteScheduleResponse(FromDeleteScheduleResponse(item)))\n\t}\n}\n\n// --- CRUD request/response fuzz tests ---\n\nfunc TestCreateScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCreateScheduleRequest, ToCreateScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestDescribeScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDescribeScheduleRequest, ToDescribeScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestDescribeScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDescribeScheduleResponse, ToDescribeScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestUpdateScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUpdateScheduleRequest, ToUpdateScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestDeleteScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDeleteScheduleRequest, ToDeleteScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\n// --- Action request/response deterministic tests ---\n\nfunc TestPauseScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.PauseScheduleRequest{nil, {}, &testdata.PauseScheduleRequest} {\n\t\tassert.Equal(t, item, ToPauseScheduleRequest(FromPauseScheduleRequest(item)))\n\t}\n}\n\nfunc TestPauseScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.PauseScheduleResponse{nil, {}, &testdata.PauseScheduleResponse} {\n\t\tassert.Equal(t, item, ToPauseScheduleResponse(FromPauseScheduleResponse(item)))\n\t}\n}\n\nfunc TestUnpauseScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.UnpauseScheduleRequest{nil, {}, &testdata.UnpauseScheduleRequest} {\n\t\tassert.Equal(t, item, ToUnpauseScheduleRequest(FromUnpauseScheduleRequest(item)))\n\t}\n}\n\nfunc TestUnpauseScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.UnpauseScheduleResponse{nil, {}, &testdata.UnpauseScheduleResponse} {\n\t\tassert.Equal(t, item, ToUnpauseScheduleResponse(FromUnpauseScheduleResponse(item)))\n\t}\n}\n\nfunc TestListSchedulesRequest(t *testing.T) {\n\tfor _, item := range []*types.ListSchedulesRequest{nil, {}, &testdata.ListSchedulesRequest} {\n\t\tassert.Equal(t, item, ToListSchedulesRequest(FromListSchedulesRequest(item)))\n\t}\n}\n\nfunc TestListSchedulesResponse(t *testing.T) {\n\tfor _, item := range []*types.ListSchedulesResponse{nil, {}, &testdata.ListSchedulesResponse} {\n\t\tassert.Equal(t, item, ToListSchedulesResponse(FromListSchedulesResponse(item)))\n\t}\n}\n\nfunc TestBackfillScheduleRequest(t *testing.T) {\n\tfor _, item := range []*types.BackfillScheduleRequest{nil, {}, &testdata.BackfillScheduleRequest} {\n\t\tassert.Equal(t, item, ToBackfillScheduleRequest(FromBackfillScheduleRequest(item)))\n\t}\n}\n\nfunc TestBackfillScheduleResponse(t *testing.T) {\n\tfor _, item := range []*types.BackfillScheduleResponse{nil, {}, &testdata.BackfillScheduleResponse} {\n\t\tassert.Equal(t, item, ToBackfillScheduleResponse(FromBackfillScheduleResponse(item)))\n\t}\n}\n\n// --- Action request/response fuzz tests ---\n\nfunc TestPauseScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPauseScheduleRequest, ToPauseScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestUnpauseScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUnpauseScheduleRequest, ToUnpauseScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestListSchedulesRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListSchedulesRequest, ToListSchedulesRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestListSchedulesResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromListSchedulesResponse, ToListSchedulesResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestBackfillScheduleRequestFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBackfillScheduleRequest, ToBackfillScheduleRequest,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestCreateScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCreateScheduleResponse, ToCreateScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestDeleteScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDeleteScheduleResponse, ToDeleteScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleCatchUpPolicyFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleCatchUpPolicy, ToScheduleCatchUpPolicy,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleOverlapPolicyFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t,\n\t\tFromScheduleOverlapPolicy, ToScheduleOverlapPolicy,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestScheduleListEntryArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromScheduleListEntryArray, ToScheduleListEntryArray,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestUpdateScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUpdateScheduleResponse, ToUpdateScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestBackfillScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBackfillScheduleResponse, ToBackfillScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestPauseScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPauseScheduleResponse, ToPauseScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestUnpauseScheduleResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromUnpauseScheduleResponse, ToUnpauseScheduleResponse,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n\nfunc TestBackfillInfoArrayFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromBackfillInfoArray, ToBackfillInfoArray,\n\t\tWithScheduleEnumFuzzers(),\n\t)\n}\n"
  },
  {
    "path": "common/types/mapper/proto/sharddistributor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// FromShardDistributorGetShardOwnerRequest converts a types.GetShardOwnerRequest to a sharddistributor.GetShardOwnerRequest\nfunc FromShardDistributorGetShardOwnerRequest(t *types.GetShardOwnerRequest) *sharddistributorv1.GetShardOwnerRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharddistributorv1.GetShardOwnerRequest{\n\t\tShardKey:  t.GetShardKey(),\n\t\tNamespace: t.GetNamespace(),\n\t}\n}\n\n// ToShardDistributorGetShardOwnerRequest converts a sharddistributor.GetShardOwnerRequest to a types.GetShardOwnerRequest\nfunc ToShardDistributorGetShardOwnerRequest(t *sharddistributorv1.GetShardOwnerRequest) *types.GetShardOwnerRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetShardOwnerRequest{\n\t\tShardKey:  t.GetShardKey(),\n\t\tNamespace: t.GetNamespace(),\n\t}\n}\n\n// FromShardDistributorGetShardOwnerResponse converts a types.GetShardOwnerResponse to a sharddistributor.GetShardOwnerResponse\nfunc FromShardDistributorGetShardOwnerResponse(t *types.GetShardOwnerResponse) *sharddistributorv1.GetShardOwnerResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharddistributorv1.GetShardOwnerResponse{\n\t\tOwner:     t.GetOwner(),\n\t\tNamespace: t.GetNamespace(),\n\t\tMetadata:  t.GetMetadata(),\n\t}\n}\n\n// ToShardDistributorGetShardOwnerResponse converts a sharddistributor.GetShardOwnerResponse to a types.GetShardOwnerResponse\nfunc ToShardDistributorGetShardOwnerResponse(t *sharddistributorv1.GetShardOwnerResponse) *types.GetShardOwnerResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetShardOwnerResponse{\n\t\tOwner:     t.GetOwner(),\n\t\tNamespace: t.GetNamespace(),\n\t\tMetadata:  t.GetMetadata(),\n\t}\n}\n\nfunc FromShardDistributorExecutorHeartbeatRequest(t *types.ExecutorHeartbeatRequest) *sharddistributorv1.HeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\t// Convert the ExecutorStatus enum\n\tvar status sharddistributorv1.ExecutorStatus\n\tswitch t.GetStatus() {\n\tcase types.ExecutorStatusINVALID:\n\t\tstatus = sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_INVALID\n\tcase types.ExecutorStatusACTIVE:\n\t\tstatus = sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_ACTIVE\n\tcase types.ExecutorStatusDRAINING:\n\t\tstatus = sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_DRAINING\n\tcase types.ExecutorStatusDRAINED:\n\t\tstatus = sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_DRAINED\n\tdefault:\n\t\tstatus = sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_INVALID\n\t}\n\n\t// Convert the ShardStatusReports\n\tvar shardStatusReports map[string]*sharddistributorv1.ShardStatusReport\n\tif t.GetShardStatusReports() != nil {\n\t\tshardStatusReports = make(map[string]*sharddistributorv1.ShardStatusReport)\n\n\t\tfor shardKey, shardStatusReport := range t.GetShardStatusReports() {\n\n\t\t\tvar status sharddistributorv1.ShardStatus\n\t\t\tswitch shardStatusReport.GetStatus() {\n\t\t\tcase types.ShardStatusINVALID:\n\t\t\t\tstatus = sharddistributorv1.ShardStatus_SHARD_STATUS_INVALID\n\t\t\tcase types.ShardStatusREADY:\n\t\t\t\tstatus = sharddistributorv1.ShardStatus_SHARD_STATUS_READY\n\t\t\tcase types.ShardStatusDONE:\n\t\t\t\tstatus = sharddistributorv1.ShardStatus_SHARD_STATUS_DONE\n\t\t\tdefault:\n\t\t\t\tstatus = sharddistributorv1.ShardStatus_SHARD_STATUS_INVALID\n\t\t\t}\n\n\t\t\tshardStatusReports[shardKey] = &sharddistributorv1.ShardStatusReport{\n\t\t\t\tStatus:    status,\n\t\t\t\tShardLoad: shardStatusReport.GetShardLoad(),\n\t\t\t}\n\t\t}\n\t}\n\treturn &sharddistributorv1.HeartbeatRequest{\n\t\tNamespace:          t.GetNamespace(),\n\t\tExecutorId:         t.GetExecutorID(),\n\t\tStatus:             status,\n\t\tShardStatusReports: shardStatusReports,\n\t\tMetadata:           t.GetMetadata(),\n\t}\n}\n\nfunc ToShardDistributorExecutorHeartbeatRequest(t *sharddistributorv1.HeartbeatRequest) *types.ExecutorHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\t// Convert the ExecutorStatus enum\n\tvar status types.ExecutorStatus\n\tswitch t.GetStatus() {\n\tcase sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_INVALID:\n\t\tstatus = types.ExecutorStatusINVALID\n\tcase sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_ACTIVE:\n\t\tstatus = types.ExecutorStatusACTIVE\n\tcase sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_DRAINING:\n\t\tstatus = types.ExecutorStatusDRAINING\n\tcase sharddistributorv1.ExecutorStatus_EXECUTOR_STATUS_DRAINED:\n\t\tstatus = types.ExecutorStatusDRAINED\n\tdefault:\n\t\tstatus = types.ExecutorStatusINVALID\n\t}\n\n\t// Convert the ShardStatusReports\n\tvar shardStatusReports map[string]*types.ShardStatusReport\n\tif t.GetShardStatusReports() != nil {\n\t\tshardStatusReports = make(map[string]*types.ShardStatusReport)\n\n\t\tfor shardKey, shardStatusReport := range t.GetShardStatusReports() {\n\n\t\t\tvar status types.ShardStatus\n\t\t\tswitch shardStatusReport.GetStatus() {\n\t\t\tcase sharddistributorv1.ShardStatus_SHARD_STATUS_INVALID:\n\t\t\t\tstatus = types.ShardStatusINVALID\n\t\t\tcase sharddistributorv1.ShardStatus_SHARD_STATUS_READY:\n\t\t\t\tstatus = types.ShardStatusREADY\n\t\t\tcase sharddistributorv1.ShardStatus_SHARD_STATUS_DONE:\n\t\t\t\tstatus = types.ShardStatusDONE\n\t\t\t}\n\n\t\t\tshardStatusReports[shardKey] = &types.ShardStatusReport{\n\t\t\t\tStatus:    status,\n\t\t\t\tShardLoad: shardStatusReport.GetShardLoad(),\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.ExecutorHeartbeatRequest{\n\t\tNamespace:          t.GetNamespace(),\n\t\tExecutorID:         t.GetExecutorId(),\n\t\tStatus:             status,\n\t\tShardStatusReports: shardStatusReports,\n\t\tMetadata:           t.GetMetadata(),\n\t}\n}\n\nfunc FromShardDistributorExecutorHeartbeatResponse(t *types.ExecutorHeartbeatResponse) *sharddistributorv1.HeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\t// Convert the ShardAssignments\n\tvar shardAssignments map[string]*sharddistributorv1.ShardAssignment\n\tmigrationMode := toMigrationMode(t.GetMigrationMode())\n\tif t.GetShardAssignments() != nil {\n\t\tshardAssignments = make(map[string]*sharddistributorv1.ShardAssignment)\n\n\t\tfor shardKey, shardAssignment := range t.GetShardAssignments() {\n\t\t\tvar status sharddistributorv1.AssignmentStatus\n\t\t\tswitch shardAssignment.GetStatus() {\n\t\t\tcase types.AssignmentStatusINVALID:\n\t\t\t\tstatus = sharddistributorv1.AssignmentStatus_ASSIGNMENT_STATUS_INVALID\n\t\t\tcase types.AssignmentStatusREADY:\n\t\t\t\tstatus = sharddistributorv1.AssignmentStatus_ASSIGNMENT_STATUS_READY\n\t\t\t}\n\t\t\tshardAssignments[shardKey] = &sharddistributorv1.ShardAssignment{\n\t\t\t\tStatus: status,\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &sharddistributorv1.HeartbeatResponse{\n\t\tShardAssignments: shardAssignments,\n\t\tMigrationMode:    migrationMode,\n\t}\n}\n\nfunc ToShardDistributorExecutorHeartbeatResponse(t *sharddistributorv1.HeartbeatResponse) *types.ExecutorHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\t// Convert the ShardAssignments\n\tvar shardAssignments map[string]*types.ShardAssignment\n\tvar migrationMode types.MigrationMode\n\tif t.GetShardAssignments() != nil {\n\t\tshardAssignments = make(map[string]*types.ShardAssignment)\n\n\t\tfor shardKey, shardAssignment := range t.GetShardAssignments() {\n\t\t\tvar status types.AssignmentStatus\n\t\t\tswitch shardAssignment.GetStatus() {\n\t\t\tcase sharddistributorv1.AssignmentStatus_ASSIGNMENT_STATUS_INVALID:\n\t\t\t\tstatus = types.AssignmentStatusINVALID\n\t\t\tcase sharddistributorv1.AssignmentStatus_ASSIGNMENT_STATUS_READY:\n\t\t\t\tstatus = types.AssignmentStatusREADY\n\t\t\t}\n\t\t\tshardAssignments[shardKey] = &types.ShardAssignment{\n\t\t\t\tStatus: status,\n\t\t\t}\n\t\t}\n\t}\n\tmigrationMode = getMigrationModeFromProto(t.GetMigrationMode())\n\n\treturn &types.ExecutorHeartbeatResponse{\n\t\tShardAssignments: shardAssignments,\n\t\tMigrationMode:    migrationMode,\n\t}\n}\n\nfunc getMigrationModeFromProto(protoMigrationMode sharddistributorv1.MigrationMode) types.MigrationMode {\n\tvar mode types.MigrationMode\n\tswitch protoMigrationMode {\n\tcase sharddistributorv1.MigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH:\n\t\tmode = types.MigrationModeLOCALPASSTHROUGH\n\tcase sharddistributorv1.MigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW:\n\t\tmode = types.MigrationModeLOCALPASSTHROUGHSHADOW\n\tcase sharddistributorv1.MigrationMode_MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH:\n\t\tmode = types.MigrationModeDISTRIBUTEDPASSTHROUGH\n\tcase sharddistributorv1.MigrationMode_MIGRATION_MODE_ONBOARDED:\n\t\tmode = types.MigrationModeONBOARDED\n\tdefault:\n\t\tmode = types.MigrationModeINVALID\n\t}\n\treturn mode\n}\n\nfunc toMigrationMode(modeSD types.MigrationMode) sharddistributorv1.MigrationMode {\n\tvar mode sharddistributorv1.MigrationMode\n\tswitch modeSD {\n\tcase types.MigrationModeINVALID:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_INVALID\n\tcase types.MigrationModeLOCALPASSTHROUGH:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH\n\tcase types.MigrationModeLOCALPASSTHROUGHSHADOW:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW\n\tcase types.MigrationModeDISTRIBUTEDPASSTHROUGH:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH\n\tcase types.MigrationModeONBOARDED:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_ONBOARDED\n\tdefault:\n\t\tmode = sharddistributorv1.MigrationMode_MIGRATION_MODE_INVALID\n\t}\n\treturn mode\n}\n\n// FromShardDistributorWatchNamespaceStateRequest converts a types.WatchNamespaceStateRequest to a sharddistributor.WatchNamespaceStateRequest\nfunc FromShardDistributorWatchNamespaceStateRequest(t *types.WatchNamespaceStateRequest) *sharddistributorv1.WatchNamespaceStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharddistributorv1.WatchNamespaceStateRequest{\n\t\tNamespace: t.GetNamespace(),\n\t}\n}\n\n// ToShardDistributorWatchNamespaceStateRequest converts a sharddistributor.WatchNamespaceStateRequest to a types.WatchNamespaceStateRequest\nfunc ToShardDistributorWatchNamespaceStateRequest(t *sharddistributorv1.WatchNamespaceStateRequest) *types.WatchNamespaceStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WatchNamespaceStateRequest{\n\t\tNamespace: t.GetNamespace(),\n\t}\n}\n\n// FromShardDistributorWatchNamespaceStateResponse converts a types.WatchNamespaceStateResponse to a sharddistributor.WatchNamespaceStateResponse\nfunc FromShardDistributorWatchNamespaceStateResponse(t *types.WatchNamespaceStateResponse) *sharddistributorv1.WatchNamespaceStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar executors []*sharddistributorv1.ExecutorInfo\n\n\tfor _, executor := range t.GetExecutors() {\n\t\t// Convert the Shards\n\t\tshards := make([]*sharddistributorv1.Shard, 0, len(executor.GetAssignedShards()))\n\t\tfor _, shard := range executor.GetAssignedShards() {\n\t\t\tshards = append(shards, &sharddistributorv1.Shard{\n\t\t\t\tShardKey: shard.GetShardKey(),\n\t\t\t})\n\t\t}\n\t\texecutors = append(executors, &sharddistributorv1.ExecutorInfo{\n\t\t\tExecutorId: executor.GetExecutorID(),\n\t\t\tMetadata:   executor.GetMetadata(),\n\t\t\tShards:     shards,\n\t\t})\n\t}\n\n\treturn &sharddistributorv1.WatchNamespaceStateResponse{\n\t\tExecutors: executors,\n\t}\n}\n\n// ToShardDistributorWatchNamespaceStateResponse converts a sharddistributor.WatchNamespaceStateResponse to a types.WatchNamespaceStateResponse\nfunc ToShardDistributorWatchNamespaceStateResponse(t *sharddistributorv1.WatchNamespaceStateResponse) *types.WatchNamespaceStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar executors []*types.ExecutorShardAssignment\n\tif t.GetExecutors() != nil {\n\t\texecutors = make([]*types.ExecutorShardAssignment, 0, len(t.GetExecutors()))\n\t\tfor _, executor := range t.GetExecutors() {\n\t\t\t// Convert the Shards\n\t\t\tshards := make([]*types.Shard, 0, len(executor.GetShards()))\n\t\t\tfor _, shard := range executor.GetShards() {\n\t\t\t\tshards = append(shards, &types.Shard{\n\t\t\t\t\tShardKey: shard.GetShardKey(),\n\t\t\t\t})\n\t\t\t}\n\n\t\t\texecutors = append(executors, &types.ExecutorShardAssignment{\n\t\t\t\tExecutorID:     executor.GetExecutorId(),\n\t\t\t\tMetadata:       executor.GetMetadata(),\n\t\t\t\tAssignedShards: shards,\n\t\t\t})\n\t\t}\n\t}\n\n\treturn &types.WatchNamespaceStateResponse{\n\t\tExecutors: executors,\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/sharddistributor_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestFromShardDistributorGetShardOwnerRequest(t *testing.T) {\n\tfor _, item := range []*types.GetShardOwnerRequest{nil, {}, &testdata.ShardDistributorGetShardOwnerRequest} {\n\t\tassert.Equal(t, item, ToShardDistributorGetShardOwnerRequest(FromShardDistributorGetShardOwnerRequest(item)))\n\t}\n}\n\nfunc TestFromShardDistributorGetShardOwnerResponse(t *testing.T) {\n\tfor _, item := range []*types.GetShardOwnerResponse{nil, {}, &testdata.ShardDistributorGetShardOwnerResponse} {\n\t\tassert.Equal(t, item, ToShardDistributorGetShardOwnerResponse(FromShardDistributorGetShardOwnerResponse(item)))\n\t}\n}\n\nfunc TestFromShardDistributorExecutorHeartbeatRequest(t *testing.T) {\n\tfor _, item := range []*types.ExecutorHeartbeatRequest{nil, {}, &testdata.ShardDistributorExecutorHeartbeatRequest} {\n\t\tassert.Equal(t, item, ToShardDistributorExecutorHeartbeatRequest(FromShardDistributorExecutorHeartbeatRequest(item)))\n\t}\n}\n\nfunc TestToShardDistributorExecutorHeartbeatResponse(t *testing.T) {\n\tfor _, item := range []*types.ExecutorHeartbeatResponse{nil, {}, &testdata.ShardDistributorExecutorHeartbeatResponse} {\n\t\tassert.Equal(t, item, ToShardDistributorExecutorHeartbeatResponse(FromShardDistributorExecutorHeartbeatResponse(item)))\n\t}\n}\n\nfunc TestFromShardDistributorWatchNamespaceStateRequest(t *testing.T) {\n\tfor _, item := range []*types.WatchNamespaceStateRequest{nil, {}, &testdata.ShardDistributorWatchNamespaceStateRequest} {\n\t\tassert.Equal(t, item, ToShardDistributorWatchNamespaceStateRequest(FromShardDistributorWatchNamespaceStateRequest(item)))\n\t}\n}\n\nfunc TestFromShardDistributorWatchNamespaceStateResponse(t *testing.T) {\n\tfor _, item := range []*types.WatchNamespaceStateResponse{nil, {}, &testdata.ShardDistributorWatchNamespaceStateResponse} {\n\t\tassert.Equal(t, item, ToShardDistributorWatchNamespaceStateResponse(FromShardDistributorWatchNamespaceStateResponse(item)))\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/shared.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromHostInfo(t *types.HostInfo) *adminv1.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.HostInfo{\n\t\tIdentity: t.Identity,\n\t}\n}\n\nfunc ToHostInfo(t *adminv1.HostInfo) *types.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HostInfo{\n\t\tIdentity: t.Identity,\n\t}\n}\n\nfunc FromMembershipInfo(t *types.MembershipInfo) *adminv1.MembershipInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.MembershipInfo{\n\t\tCurrentHost:      FromHostInfo(t.CurrentHost),\n\t\tReachableMembers: t.ReachableMembers,\n\t\tRings:            FromRingInfoArray(t.Rings),\n\t}\n}\n\nfunc FromPersistenceSettings(t []*types.PersistenceSetting) []*adminv1.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.PersistenceSetting, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPersistenceSetting(t[i])\n\t}\n\treturn v\n}\n\nfunc FromPersistenceSetting(t *types.PersistenceSetting) *adminv1.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.PersistenceSetting{\n\t\tKey:   t.Key,\n\t\tValue: t.Value,\n\t}\n}\n\nfunc FromPersistenceFeatures(t []*types.PersistenceFeature) []*adminv1.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.PersistenceFeature, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPersistenceFeature(t[i])\n\t}\n\treturn v\n}\n\nfunc FromPersistenceFeature(t *types.PersistenceFeature) *adminv1.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.PersistenceFeature{\n\t\tKey:     t.Key,\n\t\tEnabled: t.Enabled,\n\t}\n}\n\nfunc FromPersistenceInfoMap(t map[string]*types.PersistenceInfo) map[string]*adminv1.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*adminv1.PersistenceInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromPersistenceInfo(t[key])\n\t}\n\n\treturn v\n}\n\nfunc FromPersistenceInfo(t *types.PersistenceInfo) *adminv1.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &adminv1.PersistenceInfo{\n\t\tBackend:  t.Backend,\n\t\tSettings: FromPersistenceSettings(t.Settings),\n\t\tFeatures: FromPersistenceFeatures(t.Features),\n\t}\n}\n\nfunc ToPersistenceSettings(t []*adminv1.PersistenceSetting) []*types.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PersistenceSetting, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPersistenceSetting(t[i])\n\t}\n\treturn v\n}\n\nfunc ToPersistenceSetting(t *adminv1.PersistenceSetting) *types.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PersistenceSetting{\n\t\tKey:   t.Key,\n\t\tValue: t.Value,\n\t}\n}\n\nfunc ToPersistenceFeatures(t []*adminv1.PersistenceFeature) []*types.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PersistenceFeature, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPersistenceFeature(t[i])\n\t}\n\treturn v\n}\n\nfunc ToPersistenceFeature(t *adminv1.PersistenceFeature) *types.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PersistenceFeature{\n\t\tKey:     t.Key,\n\t\tEnabled: t.Enabled,\n\t}\n}\n\nfunc ToPersistenceInfoMap(t map[string]*adminv1.PersistenceInfo) map[string]*types.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.PersistenceInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToPersistenceInfo(t[key])\n\t}\n\n\treturn v\n}\n\nfunc ToPersistenceInfo(t *adminv1.PersistenceInfo) *types.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.PersistenceInfo{\n\t\tBackend:  t.Backend,\n\t\tSettings: ToPersistenceSettings(t.Settings),\n\t\tFeatures: ToPersistenceFeatures(t.Features),\n\t}\n}\n\nfunc ToMembershipInfo(t *adminv1.MembershipInfo) *types.MembershipInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MembershipInfo{\n\t\tCurrentHost:      ToHostInfo(t.CurrentHost),\n\t\tReachableMembers: t.ReachableMembers,\n\t\tRings:            ToRingInfoArray(t.Rings),\n\t}\n}\n\nfunc FromDomainCacheInfo(t *types.DomainCacheInfo) *adminv1.DomainCacheInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DomainCacheInfo{\n\t\tNumOfItemsInCacheById:   t.NumOfItemsInCacheByID,\n\t\tNumOfItemsInCacheByName: t.NumOfItemsInCacheByName,\n\t}\n}\n\nfunc ToDomainCacheInfo(t *adminv1.DomainCacheInfo) *types.DomainCacheInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainCacheInfo{\n\t\tNumOfItemsInCacheByID:   t.NumOfItemsInCacheById,\n\t\tNumOfItemsInCacheByName: t.NumOfItemsInCacheByName,\n\t}\n}\n\nfunc FromRingInfo(t *types.RingInfo) *adminv1.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.RingInfo{\n\t\tRole:        t.Role,\n\t\tMemberCount: t.MemberCount,\n\t\tMembers:     FromHostInfoArray(t.Members),\n\t}\n}\n\nfunc ToRingInfo(t *adminv1.RingInfo) *types.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RingInfo{\n\t\tRole:        t.Role,\n\t\tMemberCount: t.MemberCount,\n\t\tMembers:     ToHostInfoArray(t.Members),\n\t}\n}\n\nfunc FromTaskSource(t *types.TaskSource) sharedv1.TaskSource {\n\tif t == nil {\n\t\treturn sharedv1.TaskSource_TASK_SOURCE_INVALID\n\t}\n\tswitch *t {\n\tcase types.TaskSourceHistory:\n\t\treturn sharedv1.TaskSource_TASK_SOURCE_HISTORY\n\tcase types.TaskSourceDbBacklog:\n\t\treturn sharedv1.TaskSource_TASK_SOURCE_DB_BACKLOG\n\t}\n\treturn sharedv1.TaskSource_TASK_SOURCE_INVALID\n}\n\nfunc ToTaskSource(t sharedv1.TaskSource) *types.TaskSource {\n\tswitch t {\n\tcase sharedv1.TaskSource_TASK_SOURCE_INVALID:\n\t\treturn nil\n\tcase sharedv1.TaskSource_TASK_SOURCE_HISTORY:\n\t\treturn types.TaskSourceHistory.Ptr()\n\tcase sharedv1.TaskSource_TASK_SOURCE_DB_BACKLOG:\n\t\treturn types.TaskSourceDbBacklog.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromTransientDecisionInfo(t *types.TransientDecisionInfo) *sharedv1.TransientDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharedv1.TransientDecisionInfo{\n\t\tScheduledEvent: FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedEvent:   FromHistoryEvent(t.StartedEvent),\n\t}\n}\n\nfunc ToTransientDecisionInfo(t *sharedv1.TransientDecisionInfo) *types.TransientDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TransientDecisionInfo{\n\t\tScheduledEvent: ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedEvent:   ToHistoryEvent(t.StartedEvent),\n\t}\n}\n\nfunc FromVersionHistories(t *types.VersionHistories) *sharedv1.VersionHistories {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharedv1.VersionHistories{\n\t\tCurrentVersionHistoryIndex: t.CurrentVersionHistoryIndex,\n\t\tHistories:                  FromVersionHistoryArray(t.Histories),\n\t}\n}\n\nfunc ToVersionHistories(t *sharedv1.VersionHistories) *types.VersionHistories {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistories{\n\t\tCurrentVersionHistoryIndex: t.CurrentVersionHistoryIndex,\n\t\tHistories:                  ToVersionHistoryArray(t.Histories),\n\t}\n}\n\nfunc FromVersionHistory(t *types.VersionHistory) *adminv1.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.VersionHistory{\n\t\tBranchToken: t.BranchToken,\n\t\tItems:       FromVersionHistoryItemArray(t.Items),\n\t}\n}\n\nfunc ToVersionHistory(t *adminv1.VersionHistory) *types.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistory{\n\t\tBranchToken: t.BranchToken,\n\t\tItems:       ToVersionHistoryItemArray(t.Items),\n\t}\n}\n\nfunc FromVersionHistoryItem(t *types.VersionHistoryItem) *adminv1.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.VersionHistoryItem{\n\t\tEventId: t.EventID,\n\t\tVersion: t.Version,\n\t}\n}\n\nfunc ToVersionHistoryItem(t *adminv1.VersionHistoryItem) *types.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistoryItem{\n\t\tEventID: t.EventId,\n\t\tVersion: t.Version,\n\t}\n}\n\nfunc FromWorkflowState(t *int32) sharedv1.WorkflowState {\n\tif t == nil {\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_INVALID\n\t}\n\tswitch *t {\n\tcase persistence.WorkflowStateCreated:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_CREATED\n\tcase persistence.WorkflowStateRunning:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_RUNNING\n\tcase persistence.WorkflowStateCompleted:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_COMPLETED\n\tcase persistence.WorkflowStateZombie:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_ZOMBIE\n\tcase persistence.WorkflowStateVoid:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_VOID\n\tcase persistence.WorkflowStateCorrupted:\n\t\treturn sharedv1.WorkflowState_WORKFLOW_STATE_CORRUPTED\n\t}\n\treturn sharedv1.WorkflowState_WORKFLOW_STATE_INVALID\n}\n\nfunc ToWorkflowState(t sharedv1.WorkflowState) *int32 {\n\tswitch t {\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_INVALID:\n\t\treturn nil\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_CREATED:\n\t\tv := int32(persistence.WorkflowStateCreated)\n\t\treturn &v\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_RUNNING:\n\t\tv := int32(persistence.WorkflowStateRunning)\n\t\treturn &v\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_COMPLETED:\n\t\tv := int32(persistence.WorkflowStateCompleted)\n\t\treturn &v\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_ZOMBIE:\n\t\tv := int32(persistence.WorkflowStateZombie)\n\t\treturn &v\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_VOID:\n\t\tv := int32(persistence.WorkflowStateVoid)\n\t\treturn &v\n\tcase sharedv1.WorkflowState_WORKFLOW_STATE_CORRUPTED:\n\t\tv := int32(persistence.WorkflowStateCorrupted)\n\t\treturn &v\n\t}\n\treturn nil\n}\n\nfunc FromHostInfoArray(t []*types.HostInfo) []*adminv1.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.HostInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHostInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToHostInfoArray(t []*adminv1.HostInfo) []*types.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.HostInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHostInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromVersionHistoryArray(t []*types.VersionHistory) []*adminv1.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.VersionHistory, len(t))\n\tfor i := range t {\n\t\tv[i] = FromVersionHistory(t[i])\n\t}\n\treturn v\n}\n\nfunc ToVersionHistoryArray(t []*adminv1.VersionHistory) []*types.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.VersionHistory, len(t))\n\tfor i := range t {\n\t\tv[i] = ToVersionHistory(t[i])\n\t}\n\treturn v\n}\n\nfunc FromRingInfoArray(t []*types.RingInfo) []*adminv1.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.RingInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromRingInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToRingInfoArray(t []*adminv1.RingInfo) []*types.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.RingInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToRingInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromDLQType(t *types.DLQType) adminv1.DLQType {\n\tif t == nil {\n\t\treturn adminv1.DLQType_DLQ_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.DLQTypeReplication:\n\t\treturn adminv1.DLQType_DLQ_TYPE_REPLICATION\n\tcase types.DLQTypeDomain:\n\t\treturn adminv1.DLQType_DLQ_TYPE_DOMAIN\n\t}\n\treturn adminv1.DLQType_DLQ_TYPE_INVALID\n}\n\nfunc ToDLQType(t adminv1.DLQType) *types.DLQType {\n\tswitch t {\n\tcase adminv1.DLQType_DLQ_TYPE_INVALID:\n\t\treturn nil\n\tcase adminv1.DLQType_DLQ_TYPE_REPLICATION:\n\t\treturn types.DLQTypeReplication.Ptr()\n\tcase adminv1.DLQType_DLQ_TYPE_DOMAIN:\n\t\treturn types.DLQTypeDomain.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromDomainOperation(t *types.DomainOperation) adminv1.DomainOperation {\n\tif t == nil {\n\t\treturn adminv1.DomainOperation_DOMAIN_OPERATION_INVALID\n\t}\n\tswitch *t {\n\tcase types.DomainOperationCreate:\n\t\treturn adminv1.DomainOperation_DOMAIN_OPERATION_CREATE\n\tcase types.DomainOperationUpdate:\n\t\treturn adminv1.DomainOperation_DOMAIN_OPERATION_UPDATE\n\tcase types.DomainOperationDelete:\n\t\treturn adminv1.DomainOperation_DOMAIN_OPERATION_DELETE\n\t}\n\treturn adminv1.DomainOperation_DOMAIN_OPERATION_INVALID\n}\n\nfunc ToDomainOperation(t adminv1.DomainOperation) *types.DomainOperation {\n\tswitch t {\n\tcase adminv1.DomainOperation_DOMAIN_OPERATION_INVALID:\n\t\treturn nil\n\tcase adminv1.DomainOperation_DOMAIN_OPERATION_CREATE:\n\t\treturn types.DomainOperationCreate.Ptr()\n\tcase adminv1.DomainOperation_DOMAIN_OPERATION_UPDATE:\n\t\treturn types.DomainOperationUpdate.Ptr()\n\tcase adminv1.DomainOperation_DOMAIN_OPERATION_DELETE:\n\t\treturn types.DomainOperationDelete.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromDomainTaskAttributes(t *types.DomainTaskAttributes) *adminv1.DomainTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.DomainTaskAttributes{\n\t\tDomainOperation: FromDomainOperation(t.DomainOperation),\n\t\tId:              t.ID,\n\t\tDomain: FromDescribeDomainResponseDomain(&types.DescribeDomainResponse{\n\t\t\tDomainInfo:               t.Info,\n\t\t\tConfiguration:            t.Config,\n\t\t\tReplicationConfiguration: t.ReplicationConfig,\n\t\t}),\n\t\tConfigVersion:           t.ConfigVersion,\n\t\tFailoverVersion:         t.FailoverVersion,\n\t\tPreviousFailoverVersion: t.PreviousFailoverVersion,\n\t}\n}\n\nfunc ToDomainTaskAttributes(t *adminv1.DomainTaskAttributes) *types.DomainTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tdomain := ToDescribeDomainResponseDomain(t.Domain)\n\treturn &types.DomainTaskAttributes{\n\t\tDomainOperation:         ToDomainOperation(t.DomainOperation),\n\t\tID:                      t.Id,\n\t\tInfo:                    domain.DomainInfo,\n\t\tConfig:                  domain.Configuration,\n\t\tReplicationConfig:       domain.ReplicationConfiguration,\n\t\tConfigVersion:           t.ConfigVersion,\n\t\tFailoverVersion:         t.FailoverVersion,\n\t\tPreviousFailoverVersion: t.PreviousFailoverVersion,\n\t}\n}\n\nfunc FromFailoverMarkerAttributes(t *types.FailoverMarkerAttributes) *adminv1.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.FailoverMarkerAttributes{\n\t\tDomainId:        t.DomainID,\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tCreationTime:    unixNanoToTime(t.CreationTime),\n\t}\n}\n\nfunc ToFailoverMarkerAttributes(t *adminv1.FailoverMarkerAttributes) *types.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverMarkerAttributes{\n\t\tDomainID:        t.DomainId,\n\t\tFailoverVersion: t.FailoverVersion,\n\t\tCreationTime:    timeToUnixNano(t.CreationTime),\n\t}\n}\n\nfunc FromFailoverMarkerToken(t *types.FailoverMarkerToken) *adminv1.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.FailoverMarkerToken{\n\t\tShardIds:       t.ShardIDs,\n\t\tFailoverMarker: FromFailoverMarkerAttributes(t.FailoverMarker),\n\t}\n}\n\nfunc ToFailoverMarkerToken(t *adminv1.FailoverMarkerToken) *types.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverMarkerToken{\n\t\tShardIDs:       t.ShardIds,\n\t\tFailoverMarker: ToFailoverMarkerAttributes(t.FailoverMarker),\n\t}\n}\n\nfunc FromHistoryTaskV2Attributes(t *types.HistoryTaskV2Attributes) *adminv1.HistoryTaskV2Attributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.HistoryTaskV2Attributes{\n\t\tDomainId:            t.DomainID,\n\t\tWorkflowExecution:   FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tVersionHistoryItems: FromVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              FromDataBlob(t.Events),\n\t\tNewRunEvents:        FromDataBlob(t.NewRunEvents),\n\t}\n}\n\nfunc ToHistoryTaskV2Attributes(t *adminv1.HistoryTaskV2Attributes) *types.HistoryTaskV2Attributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryTaskV2Attributes{\n\t\tDomainID:            t.DomainId,\n\t\tWorkflowID:          ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:               ToRunID(t.WorkflowExecution),\n\t\tVersionHistoryItems: ToVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              ToDataBlob(t.Events),\n\t\tNewRunEvents:        ToDataBlob(t.NewRunEvents),\n\t}\n}\n\nfunc FromReplicationMessages(t *types.ReplicationMessages) *adminv1.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReplicationMessages{\n\t\tReplicationTasks:       FromReplicationTaskArray(t.ReplicationTasks),\n\t\tLastRetrievedMessageId: t.LastRetrievedMessageID,\n\t\tHasMore:                t.HasMore,\n\t\tSyncShardStatus:        FromSyncShardStatus(t.SyncShardStatus),\n\t}\n}\n\nfunc ToReplicationMessages(t *adminv1.ReplicationMessages) *types.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationMessages{\n\t\tReplicationTasks:       ToReplicationTaskArray(t.ReplicationTasks),\n\t\tLastRetrievedMessageID: t.LastRetrievedMessageId,\n\t\tHasMore:                t.HasMore,\n\t\tSyncShardStatus:        ToSyncShardStatus(t.SyncShardStatus),\n\t}\n}\n\n// ReplicationMessagesSize returns the size (in bytes) of the types.ReplicationMessages\nfunc ReplicationMessagesSize(t *types.ReplicationMessages) int {\n\tif t == nil {\n\t\treturn 0\n\t}\n\treturn FromReplicationMessages(t).Size()\n}\n\nfunc FromReplicationTaskInfo(t *types.ReplicationTaskInfo) *adminv1.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReplicationTaskInfo{\n\t\tDomainId:          t.DomainID,\n\t\tWorkflowExecution: FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tTaskType:          int32(t.TaskType),\n\t\tTaskId:            t.TaskID,\n\t\tVersion:           t.Version,\n\t\tFirstEventId:      t.FirstEventID,\n\t\tNextEventId:       t.NextEventID,\n\t\tScheduledId:       t.ScheduledID,\n\t}\n}\n\nfunc ToReplicationTaskInfo(t *adminv1.ReplicationTaskInfo) *types.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationTaskInfo{\n\t\tDomainID:     t.DomainId,\n\t\tWorkflowID:   ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:        ToRunID(t.WorkflowExecution),\n\t\tTaskType:     int16(t.TaskType),\n\t\tTaskID:       t.TaskId,\n\t\tVersion:      t.Version,\n\t\tFirstEventID: t.FirstEventId,\n\t\tNextEventID:  t.NextEventId,\n\t\tScheduledID:  t.ScheduledId,\n\t}\n}\n\nfunc FromReplicationTaskType(t *types.ReplicationTaskType) adminv1.ReplicationTaskType {\n\tif t == nil {\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.ReplicationTaskTypeDomain:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_DOMAIN\n\tcase types.ReplicationTaskTypeHistory:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY\n\tcase types.ReplicationTaskTypeSyncShardStatus:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_SYNC_SHARD_STATUS\n\tcase types.ReplicationTaskTypeSyncActivity:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_SYNC_ACTIVITY\n\tcase types.ReplicationTaskTypeHistoryMetadata:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY_METADATA\n\tcase types.ReplicationTaskTypeHistoryV2:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY_V2\n\tcase types.ReplicationTaskTypeFailoverMarker:\n\t\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_FAILOVER_MARKER\n\t}\n\treturn adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID\n}\n\nfunc ToReplicationTaskType(t adminv1.ReplicationTaskType) *types.ReplicationTaskType {\n\tswitch t {\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_INVALID:\n\t\treturn nil\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_DOMAIN:\n\t\treturn types.ReplicationTaskTypeDomain.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY:\n\t\treturn types.ReplicationTaskTypeHistory.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_SYNC_SHARD_STATUS:\n\t\treturn types.ReplicationTaskTypeSyncShardStatus.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_SYNC_ACTIVITY:\n\t\treturn types.ReplicationTaskTypeSyncActivity.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY_METADATA:\n\t\treturn types.ReplicationTaskTypeHistoryMetadata.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_HISTORY_V2:\n\t\treturn types.ReplicationTaskTypeHistoryV2.Ptr()\n\tcase adminv1.ReplicationTaskType_REPLICATION_TASK_TYPE_FAILOVER_MARKER:\n\t\treturn types.ReplicationTaskTypeFailoverMarker.Ptr()\n\t}\n\treturn nil\n}\n\nfunc FromReplicationToken(t *types.ReplicationToken) *adminv1.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ReplicationToken{\n\t\tShardId:                t.ShardID,\n\t\tLastRetrievedMessageId: t.LastRetrievedMessageID,\n\t\tLastProcessedMessageId: t.LastProcessedMessageID,\n\t}\n}\n\nfunc ToReplicationToken(t *adminv1.ReplicationToken) *types.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationToken{\n\t\tShardID:                t.ShardId,\n\t\tLastRetrievedMessageID: t.LastRetrievedMessageId,\n\t\tLastProcessedMessageID: t.LastProcessedMessageId,\n\t}\n}\n\nfunc FromSyncActivityTaskAttributes(t *types.SyncActivityTaskAttributes) *adminv1.SyncActivityTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.SyncActivityTaskAttributes{\n\t\tDomainId:           t.DomainID,\n\t\tWorkflowExecution:  FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tVersion:            t.Version,\n\t\tScheduledId:        t.ScheduledID,\n\t\tScheduledTime:      unixNanoToTime(t.ScheduledTime),\n\t\tStartedId:          t.StartedID,\n\t\tStartedTime:        unixNanoToTime(t.StartedTime),\n\t\tLastHeartbeatTime:  unixNanoToTime(t.LastHeartbeatTime),\n\t\tDetails:            FromPayload(t.Details),\n\t\tAttempt:            t.Attempt,\n\t\tLastFailure:        FromFailure(t.LastFailureReason, t.LastFailureDetails),\n\t\tLastWorkerIdentity: t.LastWorkerIdentity,\n\t\tVersionHistory:     FromVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc ToSyncActivityTaskAttributes(t *adminv1.SyncActivityTaskAttributes) *types.SyncActivityTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncActivityTaskAttributes{\n\t\tDomainID:           t.DomainId,\n\t\tWorkflowID:         ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:              ToRunID(t.WorkflowExecution),\n\t\tVersion:            t.Version,\n\t\tScheduledID:        t.ScheduledId,\n\t\tScheduledTime:      timeToUnixNano(t.ScheduledTime),\n\t\tStartedID:          t.StartedId,\n\t\tStartedTime:        timeToUnixNano(t.StartedTime),\n\t\tLastHeartbeatTime:  timeToUnixNano(t.LastHeartbeatTime),\n\t\tDetails:            ToPayload(t.Details),\n\t\tAttempt:            t.Attempt,\n\t\tLastFailureReason:  ToFailureReason(t.LastFailure),\n\t\tLastFailureDetails: ToFailureDetails(t.LastFailure),\n\t\tLastWorkerIdentity: t.LastWorkerIdentity,\n\t\tVersionHistory:     ToVersionHistory(t.VersionHistory),\n\t}\n}\n\nfunc FromSyncShardStatus(t *types.SyncShardStatus) *adminv1.SyncShardStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.SyncShardStatus{\n\t\tTimestamp: unixNanoToTime(t.Timestamp),\n\t}\n}\n\nfunc ToSyncShardStatus(t *adminv1.SyncShardStatus) *types.SyncShardStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatus{\n\t\tTimestamp: timeToUnixNano(t.Timestamp),\n\t}\n}\n\nfunc FromSyncShardStatusTaskAttributes(t *types.SyncShardStatusTaskAttributes) *adminv1.SyncShardStatusTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.SyncShardStatusTaskAttributes{\n\t\tSourceCluster: t.SourceCluster,\n\t\tShardId:       int32(t.ShardID),\n\t\tTimestamp:     unixNanoToTime(t.Timestamp),\n\t}\n}\n\nfunc ToSyncShardStatusTaskAttributes(t *adminv1.SyncShardStatusTaskAttributes) *types.SyncShardStatusTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatusTaskAttributes{\n\t\tSourceCluster: t.SourceCluster,\n\t\tShardID:       int64(t.ShardId),\n\t\tTimestamp:     timeToUnixNano(t.Timestamp),\n\t}\n}\n\nfunc FromReplicationTaskInfoArray(t []*types.ReplicationTaskInfo) []*adminv1.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.ReplicationTaskInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationTaskInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc ToReplicationTaskInfoArray(t []*adminv1.ReplicationTaskInfo) []*types.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationTaskInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationTaskInfo(t[i])\n\t}\n\treturn v\n}\n\nfunc FromReplicationTaskArray(t []*types.ReplicationTask) []*adminv1.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.ReplicationTask, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationTask(t[i])\n\t}\n\treturn v\n}\n\nfunc ToReplicationTaskArray(t []*adminv1.ReplicationTask) []*types.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationTask, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationTask(t[i])\n\t}\n\treturn v\n}\n\nfunc FromReplicationTokenArray(t []*types.ReplicationToken) []*adminv1.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.ReplicationToken, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationToken(t[i])\n\t}\n\treturn v\n}\n\nfunc ToReplicationTokenArray(t []*adminv1.ReplicationToken) []*types.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationToken, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationToken(t[i])\n\t}\n\treturn v\n}\n\nfunc FromReplicationMessagesMap(t map[int32]*types.ReplicationMessages) map[int32]*adminv1.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]*adminv1.ReplicationMessages, len(t))\n\tfor key := range t {\n\t\tv[key] = FromReplicationMessages(t[key])\n\t}\n\treturn v\n}\n\nfunc ToReplicationMessagesMap(t map[int32]*adminv1.ReplicationMessages) map[int32]*types.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]*types.ReplicationMessages, len(t))\n\tfor key := range t {\n\t\tv[key] = ToReplicationMessages(t[key])\n\t}\n\treturn v\n}\n\nfunc FromReplicationTask(t *types.ReplicationTask) *adminv1.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttask := adminv1.ReplicationTask{\n\t\tTaskType:     FromReplicationTaskType(t.TaskType),\n\t\tSourceTaskId: t.SourceTaskID,\n\t\tCreationTime: unixNanoToTime(t.CreationTime),\n\t}\n\tif t.DomainTaskAttributes != nil {\n\t\ttask.Attributes = &adminv1.ReplicationTask_DomainTaskAttributes{\n\t\t\tDomainTaskAttributes: FromDomainTaskAttributes(t.DomainTaskAttributes),\n\t\t}\n\t}\n\tif t.SyncShardStatusTaskAttributes != nil {\n\t\ttask.Attributes = &adminv1.ReplicationTask_SyncShardStatusTaskAttributes{\n\t\t\tSyncShardStatusTaskAttributes: FromSyncShardStatusTaskAttributes(t.SyncShardStatusTaskAttributes),\n\t\t}\n\t}\n\tif t.SyncActivityTaskAttributes != nil {\n\t\ttask.Attributes = &adminv1.ReplicationTask_SyncActivityTaskAttributes{\n\t\t\tSyncActivityTaskAttributes: FromSyncActivityTaskAttributes(t.SyncActivityTaskAttributes),\n\t\t}\n\t}\n\tif t.HistoryTaskV2Attributes != nil {\n\t\ttask.Attributes = &adminv1.ReplicationTask_HistoryTaskV2Attributes{\n\t\t\tHistoryTaskV2Attributes: FromHistoryTaskV2Attributes(t.HistoryTaskV2Attributes),\n\t\t}\n\t}\n\tif t.FailoverMarkerAttributes != nil {\n\t\ttask.Attributes = &adminv1.ReplicationTask_FailoverMarkerAttributes{\n\t\t\tFailoverMarkerAttributes: FromFailoverMarkerAttributes(t.FailoverMarkerAttributes),\n\t\t}\n\t}\n\n\treturn &task\n}\n\nfunc ToReplicationTask(t *adminv1.ReplicationTask) *types.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttask := types.ReplicationTask{\n\t\tTaskType:     ToReplicationTaskType(t.TaskType),\n\t\tSourceTaskID: t.SourceTaskId,\n\t\tCreationTime: timeToUnixNano(t.CreationTime),\n\t}\n\n\tswitch attr := t.Attributes.(type) {\n\tcase *adminv1.ReplicationTask_DomainTaskAttributes:\n\t\ttask.DomainTaskAttributes = ToDomainTaskAttributes(attr.DomainTaskAttributes)\n\tcase *adminv1.ReplicationTask_SyncShardStatusTaskAttributes:\n\t\ttask.SyncShardStatusTaskAttributes = ToSyncShardStatusTaskAttributes(attr.SyncShardStatusTaskAttributes)\n\tcase *adminv1.ReplicationTask_SyncActivityTaskAttributes:\n\t\ttask.SyncActivityTaskAttributes = ToSyncActivityTaskAttributes(attr.SyncActivityTaskAttributes)\n\tcase *adminv1.ReplicationTask_HistoryTaskV2Attributes:\n\t\ttask.HistoryTaskV2Attributes = ToHistoryTaskV2Attributes(attr.HistoryTaskV2Attributes)\n\tcase *adminv1.ReplicationTask_FailoverMarkerAttributes:\n\t\ttask.FailoverMarkerAttributes = ToFailoverMarkerAttributes(attr.FailoverMarkerAttributes)\n\t}\n\treturn &task\n}\n\nfunc FromTaskType(t *int32) adminv1.TaskType {\n\tif t == nil {\n\t\treturn adminv1.TaskType_TASK_TYPE_INVALID\n\t}\n\tswitch constants.TaskType(*t) {\n\tcase constants.TaskTypeTransfer:\n\t\treturn adminv1.TaskType_TASK_TYPE_TRANSFER\n\tcase constants.TaskTypeTimer:\n\t\treturn adminv1.TaskType_TASK_TYPE_TIMER\n\tcase constants.TaskTypeReplication:\n\t\treturn adminv1.TaskType_TASK_TYPE_REPLICATION\n\tcase constants.TaskTypeCrossCluster:\n\t\treturn adminv1.TaskType_TASK_TYPE_CROSS_CLUSTER\n\t}\n\treturn adminv1.TaskType_TASK_TYPE_INVALID\n}\n\nfunc ToTaskType(t adminv1.TaskType) *int32 {\n\tswitch t {\n\tcase adminv1.TaskType_TASK_TYPE_INVALID:\n\t\treturn nil\n\tcase adminv1.TaskType_TASK_TYPE_TRANSFER:\n\t\treturn common.Int32Ptr(int32(constants.TaskTypeTransfer))\n\tcase adminv1.TaskType_TASK_TYPE_TIMER:\n\t\treturn common.Int32Ptr(int32(constants.TaskTypeTimer))\n\tcase adminv1.TaskType_TASK_TYPE_REPLICATION:\n\t\treturn common.Int32Ptr(int32(constants.TaskTypeReplication))\n\tcase adminv1.TaskType_TASK_TYPE_CROSS_CLUSTER:\n\t\treturn common.Int32Ptr(int32(constants.TaskTypeCrossCluster))\n\t}\n\treturn nil\n}\n\nfunc FromFailoverMarkerTokenArray(t []*types.FailoverMarkerToken) []*adminv1.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.FailoverMarkerToken, len(t))\n\tfor i := range t {\n\t\tv[i] = FromFailoverMarkerToken(t[i])\n\t}\n\treturn v\n}\n\nfunc ToFailoverMarkerTokenArray(t []*adminv1.FailoverMarkerToken) []*types.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.FailoverMarkerToken, len(t))\n\tfor i := range t {\n\t\tv[i] = ToFailoverMarkerToken(t[i])\n\t}\n\treturn v\n}\n\nfunc FromVersionHistoryItemArray(t []*types.VersionHistoryItem) []*adminv1.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.VersionHistoryItem, len(t))\n\tfor i := range t {\n\t\tv[i] = FromVersionHistoryItem(t[i])\n\t}\n\treturn v\n}\n\nfunc ToVersionHistoryItemArray(t []*adminv1.VersionHistoryItem) []*types.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.VersionHistoryItem, len(t))\n\tfor i := range t {\n\t\tv[i] = ToVersionHistoryItem(t[i])\n\t}\n\treturn v\n}\n\nfunc FromEventIDVersionPair(id, version *int64) *adminv1.VersionHistoryItem {\n\tif id == nil || version == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.VersionHistoryItem{\n\t\tEventId: *id,\n\t\tVersion: *version,\n\t}\n}\n\nfunc ToEventID(item *adminv1.VersionHistoryItem) *int64 {\n\tif item == nil {\n\t\treturn nil\n\t}\n\treturn common.Int64Ptr(item.EventId)\n}\n\nfunc ToEventVersion(item *adminv1.VersionHistoryItem) *int64 {\n\tif item == nil {\n\t\treturn nil\n\t}\n\treturn common.Int64Ptr(item.Version)\n}\n\n// FromCrossClusterTaskType converts internal CrossClusterTaskType type to proto\nfunc FromCrossClusterTaskType(t *types.CrossClusterTaskType) adminv1.CrossClusterTaskType {\n\tif t == nil {\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_INVALID\n\t}\n\tswitch *t {\n\tcase types.CrossClusterTaskTypeStartChildExecution:\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_START_CHILD_EXECUTION\n\tcase types.CrossClusterTaskTypeCancelExecution:\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_CANCEL_EXECUTION\n\tcase types.CrossClusterTaskTypeSignalExecution:\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_SIGNAL_EXECUTION\n\tcase types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete:\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_RECORD_CHILD_WORKKLOW_EXECUTION_COMPLETE\n\tcase types.CrossClusterTaskTypeApplyParentPolicy:\n\t\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_APPLY_PARENT_CLOSE_POLICY\n\t}\n\treturn adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_INVALID\n}\n\n// ToCrossClusterTaskType converts proto CrossClusterTaskType type to internal\nfunc ToCrossClusterTaskType(t adminv1.CrossClusterTaskType) *types.CrossClusterTaskType {\n\tswitch t {\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_INVALID:\n\t\treturn nil\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_START_CHILD_EXECUTION:\n\t\treturn types.CrossClusterTaskTypeStartChildExecution.Ptr()\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_CANCEL_EXECUTION:\n\t\treturn types.CrossClusterTaskTypeCancelExecution.Ptr()\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_SIGNAL_EXECUTION:\n\t\treturn types.CrossClusterTaskTypeSignalExecution.Ptr()\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_RECORD_CHILD_WORKKLOW_EXECUTION_COMPLETE:\n\t\treturn types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete.Ptr()\n\tcase adminv1.CrossClusterTaskType_CROSS_CLUSTER_TASK_TYPE_APPLY_PARENT_CLOSE_POLICY:\n\t\treturn types.CrossClusterTaskTypeApplyParentPolicy.Ptr()\n\t}\n\treturn nil\n}\n\n// FromCrossClusterTaskFailedCause converts internal CrossClusterTaskFailedCause type to proto\nfunc FromCrossClusterTaskFailedCause(t *types.CrossClusterTaskFailedCause) adminv1.CrossClusterTaskFailedCause {\n\tif t == nil {\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.CrossClusterTaskFailedCauseDomainNotActive:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_ACTIVE\n\tcase types.CrossClusterTaskFailedCauseDomainNotExists:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_EXISTS\n\tcase types.CrossClusterTaskFailedCauseWorkflowAlreadyRunning:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING\n\tcase types.CrossClusterTaskFailedCauseWorkflowNotExists:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_NOT_EXISTS\n\tcase types.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED\n\tcase types.CrossClusterTaskFailedCauseUncategorized:\n\t\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_UNCATEGORIZED\n\t}\n\treturn adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID\n}\n\n// ToCrossClusterTaskFailedCause converts proto CrossClusterTaskFailedCause type to internal\nfunc ToCrossClusterTaskFailedCause(t adminv1.CrossClusterTaskFailedCause) *types.CrossClusterTaskFailedCause {\n\tswitch t {\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_ACTIVE:\n\t\treturn types.CrossClusterTaskFailedCauseDomainNotActive.Ptr()\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_DOMAIN_NOT_EXISTS:\n\t\treturn types.CrossClusterTaskFailedCauseDomainNotExists.Ptr()\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_ALREADY_RUNNING:\n\t\treturn types.CrossClusterTaskFailedCauseWorkflowAlreadyRunning.Ptr()\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_NOT_EXISTS:\n\t\treturn types.CrossClusterTaskFailedCauseWorkflowNotExists.Ptr()\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_WORKFLOW_ALREADY_COMPLETED:\n\t\treturn types.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted.Ptr()\n\tcase adminv1.CrossClusterTaskFailedCause_CROSS_CLUSTER_TASK_FAILED_CAUSE_UNCATEGORIZED:\n\t\treturn types.CrossClusterTaskFailedCauseUncategorized.Ptr()\n\n\t}\n\treturn nil\n}\n\n// FromGetTaskFailedCause converts internal GetTaskFailedCause type to proto\nfunc FromGetTaskFailedCause(t *types.GetTaskFailedCause) adminv1.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_INVALID\n\t}\n\tswitch *t {\n\tcase types.GetTaskFailedCauseServiceBusy:\n\t\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_SERVICE_BUSY\n\tcase types.GetTaskFailedCauseTimeout:\n\t\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_TIMEOUT\n\tcase types.GetTaskFailedCauseShardOwnershipLost:\n\t\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_SHARD_OWNERSHIP_LOST\n\tcase types.GetTaskFailedCauseUncategorized:\n\t\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_UNCATEGORIZED\n\t}\n\treturn adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_INVALID\n}\n\n// ToGetTaskFailedCause converts proto GetTaskFailedCause type to internal\nfunc ToGetTaskFailedCause(t adminv1.GetTaskFailedCause) *types.GetTaskFailedCause {\n\tswitch t {\n\tcase adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_INVALID:\n\t\treturn nil\n\tcase adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_SERVICE_BUSY:\n\t\treturn types.GetTaskFailedCauseServiceBusy.Ptr()\n\tcase adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_TIMEOUT:\n\t\treturn types.GetTaskFailedCauseTimeout.Ptr()\n\tcase adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_SHARD_OWNERSHIP_LOST:\n\t\treturn types.GetTaskFailedCauseShardOwnershipLost.Ptr()\n\tcase adminv1.GetTaskFailedCause_GET_TASK_FAILED_CAUSE_UNCATEGORIZED:\n\t\treturn types.GetTaskFailedCauseUncategorized.Ptr()\n\t}\n\treturn nil\n}\n\n// FromCrossClusterTaskInfo converts internal CrossClusterTaskInfo type to proto\nfunc FromCrossClusterTaskInfo(t *types.CrossClusterTaskInfo) *adminv1.CrossClusterTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterTaskInfo{\n\t\tDomainId:            t.DomainID,\n\t\tWorkflowExecution:   FromWorkflowRunPair(t.WorkflowID, t.RunID),\n\t\tTaskType:            FromCrossClusterTaskType(t.TaskType),\n\t\tTaskState:           int32(t.TaskState),\n\t\tTaskId:              t.TaskID,\n\t\tVisibilityTimestamp: unixNanoToTime(t.VisibilityTimestamp),\n\t}\n}\n\n// ToCrossClusterTaskInfo converts proto CrossClusterTaskInfo type to internal\nfunc ToCrossClusterTaskInfo(t *adminv1.CrossClusterTaskInfo) *types.CrossClusterTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterTaskInfo{\n\t\tDomainID:            t.DomainId,\n\t\tWorkflowID:          ToWorkflowID(t.WorkflowExecution),\n\t\tRunID:               ToRunID(t.WorkflowExecution),\n\t\tTaskType:            ToCrossClusterTaskType(t.TaskType),\n\t\tTaskState:           int16(t.TaskState),\n\t\tTaskID:              t.TaskId,\n\t\tVisibilityTimestamp: timeToUnixNano(t.VisibilityTimestamp),\n\t}\n}\n\n// FromCrossClusterStartChildExecutionRequestAttributes converts internal CrossClusterStartChildExecutionRequestAttributes type to proto\nfunc FromCrossClusterStartChildExecutionRequestAttributes(t *types.CrossClusterStartChildExecutionRequestAttributes) *adminv1.CrossClusterStartChildExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterStartChildExecutionRequestAttributes{\n\t\tTargetDomainId:           t.TargetDomainID,\n\t\tRequestId:                t.RequestID,\n\t\tInitiatedEventId:         t.InitiatedEventID,\n\t\tInitiatedEventAttributes: FromStartChildWorkflowExecutionInitiatedEventAttributes(t.InitiatedEventAttributes),\n\t\tTargetRunId:              t.GetTargetRunID(),\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\n// ToCrossClusterStartChildExecutionRequestAttributes converts proto CrossClusterStartChildExecutionRequestAttributes type to internal\nfunc ToCrossClusterStartChildExecutionRequestAttributes(t *adminv1.CrossClusterStartChildExecutionRequestAttributes) *types.CrossClusterStartChildExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterStartChildExecutionRequestAttributes{\n\t\tTargetDomainID:           t.TargetDomainId,\n\t\tRequestID:                t.RequestId,\n\t\tInitiatedEventID:         t.InitiatedEventId,\n\t\tInitiatedEventAttributes: ToStartChildWorkflowExecutionInitiatedEventAttributes(t.InitiatedEventAttributes),\n\t\tTargetRunID:              &t.TargetRunId,\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\n// FromCrossClusterStartChildExecutionResponseAttributes converts internal CrossClusterStartChildExecutionResponseAttributes type to proto\nfunc FromCrossClusterStartChildExecutionResponseAttributes(t *types.CrossClusterStartChildExecutionResponseAttributes) *adminv1.CrossClusterStartChildExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterStartChildExecutionResponseAttributes{\n\t\tRunId: t.RunID,\n\t}\n}\n\n// ToCrossClusterStartChildExecutionResponseAttributes converts proto CrossClusterStartChildExecutionResponseAttributes type to internal\nfunc ToCrossClusterStartChildExecutionResponseAttributes(t *adminv1.CrossClusterStartChildExecutionResponseAttributes) *types.CrossClusterStartChildExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterStartChildExecutionResponseAttributes{\n\t\tRunID: t.RunId,\n\t}\n}\n\n// FromCrossClusterCancelExecutionRequestAttributes converts internal CrossClusterCancelExecutionRequestAttributes type to proto\nfunc FromCrossClusterCancelExecutionRequestAttributes(t *types.CrossClusterCancelExecutionRequestAttributes) *adminv1.CrossClusterCancelExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterCancelExecutionRequestAttributes{\n\t\tTargetDomainId:          t.TargetDomainID,\n\t\tTargetWorkflowExecution: FromWorkflowRunPair(t.TargetWorkflowID, t.TargetRunID),\n\t\tRequestId:               t.RequestID,\n\t\tInitiatedEventId:        t.InitiatedEventID,\n\t\tChildWorkflowOnly:       t.ChildWorkflowOnly,\n\t}\n}\n\n// ToCrossClusterCancelExecutionRequestAttributes converts proto CrossClusterCancelExecutionRequestAttributes type to internal\nfunc ToCrossClusterCancelExecutionRequestAttributes(t *adminv1.CrossClusterCancelExecutionRequestAttributes) *types.CrossClusterCancelExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterCancelExecutionRequestAttributes{\n\t\tTargetDomainID:    t.TargetDomainId,\n\t\tTargetWorkflowID:  ToWorkflowID(t.TargetWorkflowExecution),\n\t\tTargetRunID:       ToRunID(t.TargetWorkflowExecution),\n\t\tRequestID:         t.RequestId,\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t}\n}\n\n// FromCrossClusterCancelExecutionResponseAttributes converts internal CrossClusterCancelExecutionResponseAttributes type to proto\nfunc FromCrossClusterCancelExecutionResponseAttributes(t *types.CrossClusterCancelExecutionResponseAttributes) *adminv1.CrossClusterCancelExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterCancelExecutionResponseAttributes{}\n}\n\n// ToCrossClusterCancelExecutionResponseAttributes converts proto CrossClusterCancelExecutionResponseAttributes type to internal\nfunc ToCrossClusterCancelExecutionResponseAttributes(t *adminv1.CrossClusterCancelExecutionResponseAttributes) *types.CrossClusterCancelExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterCancelExecutionResponseAttributes{}\n}\n\n// FromCrossClusterSignalExecutionRequestAttributes converts internal CrossClusterSignalExecutionRequestAttributes type to proto\nfunc FromCrossClusterSignalExecutionRequestAttributes(t *types.CrossClusterSignalExecutionRequestAttributes) *adminv1.CrossClusterSignalExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterSignalExecutionRequestAttributes{\n\t\tTargetDomainId:          t.TargetDomainID,\n\t\tTargetWorkflowExecution: FromWorkflowRunPair(t.TargetWorkflowID, t.TargetRunID),\n\t\tRequestId:               t.RequestID,\n\t\tInitiatedEventId:        t.InitiatedEventID,\n\t\tChildWorkflowOnly:       t.ChildWorkflowOnly,\n\t\tSignalName:              t.SignalName,\n\t\tSignalInput:             FromPayload(t.SignalInput),\n\t\tControl:                 t.Control,\n\t}\n}\n\n// ToCrossClusterSignalExecutionRequestAttributes converts proto CrossClusterSignalExecutionRequestAttributes type to internal\nfunc ToCrossClusterSignalExecutionRequestAttributes(t *adminv1.CrossClusterSignalExecutionRequestAttributes) *types.CrossClusterSignalExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterSignalExecutionRequestAttributes{\n\t\tTargetDomainID:    t.TargetDomainId,\n\t\tTargetWorkflowID:  ToWorkflowID(t.TargetWorkflowExecution),\n\t\tTargetRunID:       ToRunID(t.TargetWorkflowExecution),\n\t\tRequestID:         t.RequestId,\n\t\tInitiatedEventID:  t.InitiatedEventId,\n\t\tChildWorkflowOnly: t.ChildWorkflowOnly,\n\t\tSignalName:        t.SignalName,\n\t\tSignalInput:       ToPayload(t.SignalInput),\n\t\tControl:           t.Control,\n\t}\n}\n\n// FromCrossClusterSignalExecutionResponseAttributes converts internal CrossClusterSignalExecutionResponseAttributes type to proto\nfunc FromCrossClusterSignalExecutionResponseAttributes(t *types.CrossClusterSignalExecutionResponseAttributes) *adminv1.CrossClusterSignalExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterSignalExecutionResponseAttributes{}\n}\n\n// ToCrossClusterSignalExecutionResponseAttributes converts proto CrossClusterSignalExecutionResponseAttributes type to internal\nfunc ToCrossClusterSignalExecutionResponseAttributes(t *adminv1.CrossClusterSignalExecutionResponseAttributes) *types.CrossClusterSignalExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterSignalExecutionResponseAttributes{}\n}\n\n// FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes converts internal CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes type to proto\nfunc FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(t *types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) *adminv1.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\tTargetDomainId:          t.TargetDomainID,\n\t\tTargetWorkflowExecution: FromWorkflowRunPair(t.TargetWorkflowID, t.TargetRunID),\n\t\tInitiatedEventId:        t.InitiatedEventID,\n\t\tCompletionEvent:         FromHistoryEvent(t.CompletionEvent),\n\t}\n}\n\n// ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes converts proto CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes type to internal\nfunc ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(t *adminv1.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes) *types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\tTargetDomainID:   t.TargetDomainId,\n\t\tTargetWorkflowID: ToWorkflowID(t.TargetWorkflowExecution),\n\t\tTargetRunID:      ToRunID(t.TargetWorkflowExecution),\n\t\tInitiatedEventID: t.InitiatedEventId,\n\t\tCompletionEvent:  ToHistoryEvent(t.CompletionEvent),\n\t}\n}\n\n// FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes converts internal CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes type to proto\nfunc FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(t *types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) *adminv1.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{}\n}\n\n// ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes converts proto CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes type to internal\nfunc ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(t *adminv1.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes) *types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{}\n}\n\n// FromApplyParentClosePolicyStatus converts internal ApplyParentClosePolicyStatus type to proto\nfunc FromApplyParentClosePolicyStatus(t *types.ApplyParentClosePolicyStatus) *adminv1.ApplyParentClosePolicyStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ApplyParentClosePolicyStatus{\n\t\tCompleted:   t.Completed,\n\t\tFailedCause: FromCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// ToApplyParentClosePolicyStatus converts proto ApplyParentClosePolicyStatus type to internal\nfunc ToApplyParentClosePolicyStatus(t *adminv1.ApplyParentClosePolicyStatus) *types.ApplyParentClosePolicyStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyStatus{\n\t\tCompleted:   t.Completed,\n\t\tFailedCause: ToCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// FromApplyParentClosePolicyAttributes converts internal ApplyParentClosePolicyAttributes type to proto\nfunc FromApplyParentClosePolicyAttributes(t *types.ApplyParentClosePolicyAttributes) *adminv1.ApplyParentClosePolicyAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ApplyParentClosePolicyAttributes{\n\t\tChildDomainId:     t.ChildDomainID,\n\t\tChildWorkflowId:   t.ChildWorkflowID,\n\t\tChildRunId:        t.ChildRunID,\n\t\tParentClosePolicy: FromParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// ToApplyParentClosePolicyAttributes converts proto ApplyParentClosePolicyAttributes type to internal\nfunc ToApplyParentClosePolicyAttributes(t *adminv1.ApplyParentClosePolicyAttributes) *types.ApplyParentClosePolicyAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyAttributes{\n\t\tChildDomainID:     t.ChildDomainId,\n\t\tChildWorkflowID:   t.ChildWorkflowId,\n\t\tChildRunID:        t.ChildRunId,\n\t\tParentClosePolicy: ToParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// FromApplyParentClosePolicyResult converts proto ApplyParentClosePolicyResult type to internal\nfunc FromApplyParentClosePolicyResult(t *types.ApplyParentClosePolicyResult) *adminv1.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &adminv1.ApplyParentClosePolicyResult{\n\t\tChild:       FromApplyParentClosePolicyAttributes(t.Child),\n\t\tFailedCause: FromCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// ToApplyParentClosePolicyResult converts proto ApplyParentClosePolicyResult type to internal\nfunc ToApplyParentClosePolicyResult(t *adminv1.ApplyParentClosePolicyResult) *types.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyResult{\n\t\tChild:       ToApplyParentClosePolicyAttributes(t.Child),\n\t\tFailedCause: ToCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// FromCrossClusterApplyParentClosePolicyRequestAttributes converts internal CrossClusterApplyParentClosePolicyRequestAttributes type to proto\nfunc FromCrossClusterApplyParentClosePolicyRequestAttributes(t *types.CrossClusterApplyParentClosePolicyRequestAttributes) *adminv1.CrossClusterApplyParentClosePolicyRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequestAttributes := &adminv1.CrossClusterApplyParentClosePolicyRequestAttributes{}\n\tfor _, execution := range t.Children {\n\t\trequestAttributes.Children = append(\n\t\t\trequestAttributes.Children,\n\t\t\t&adminv1.ApplyParentClosePolicyRequest{\n\t\t\t\tChild:  FromApplyParentClosePolicyAttributes(execution.Child),\n\t\t\t\tStatus: FromApplyParentClosePolicyStatus(execution.Status),\n\t\t\t},\n\t\t)\n\t}\n\treturn requestAttributes\n}\n\n// ToCrossClusterApplyParentClosePolicyRequestAttributes converts proto CrossClusterApplyParentClosePolicyRequestAttributes type to internal\nfunc ToCrossClusterApplyParentClosePolicyRequestAttributes(t *adminv1.CrossClusterApplyParentClosePolicyRequestAttributes) *types.CrossClusterApplyParentClosePolicyRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequestAttributes := &types.CrossClusterApplyParentClosePolicyRequestAttributes{}\n\tfor _, execution := range t.Children {\n\t\trequestAttributes.Children = append(\n\t\t\trequestAttributes.Children,\n\t\t\t&types.ApplyParentClosePolicyRequest{\n\t\t\t\tChild:  ToApplyParentClosePolicyAttributes(execution.Child),\n\t\t\t\tStatus: ToApplyParentClosePolicyStatus(execution.Status),\n\t\t\t},\n\t\t)\n\t}\n\treturn requestAttributes\n}\n\n// FromCrossClusterApplyParentClosePolicyResponseAttributes converts internal CrossClusterApplyParentClosePolicyResponseAttributes type to proto\nfunc FromCrossClusterApplyParentClosePolicyResponseAttributes(t *types.CrossClusterApplyParentClosePolicyResponseAttributes) *adminv1.CrossClusterApplyParentClosePolicyResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tresponse := &adminv1.CrossClusterApplyParentClosePolicyResponseAttributes{}\n\tfor _, childStatus := range t.ChildrenStatus {\n\t\tresponse.ChildrenStatus = append(\n\t\t\tresponse.ChildrenStatus,\n\t\t\tFromApplyParentClosePolicyResult(childStatus),\n\t\t)\n\t}\n\treturn response\n}\n\n// ToCrossClusterApplyParentClosePolicyResponseAttributes converts proto CrossClusterApplyParentClosePolicyResponseAttributes type to internal\nfunc ToCrossClusterApplyParentClosePolicyResponseAttributes(t *adminv1.CrossClusterApplyParentClosePolicyResponseAttributes) *types.CrossClusterApplyParentClosePolicyResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tresponse := &types.CrossClusterApplyParentClosePolicyResponseAttributes{}\n\tfor _, childStatus := range t.ChildrenStatus {\n\t\tresponse.ChildrenStatus = append(\n\t\t\tresponse.ChildrenStatus,\n\t\t\tToApplyParentClosePolicyResult(childStatus),\n\t\t)\n\t}\n\treturn response\n}\n\n// FromCrossClusterTaskRequest converts internal CrossClusterTaskRequest type to proto\nfunc FromCrossClusterTaskRequest(t *types.CrossClusterTaskRequest) *adminv1.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := adminv1.CrossClusterTaskRequest{\n\t\tTaskInfo: FromCrossClusterTaskInfo(t.TaskInfo),\n\t}\n\tif t.StartChildExecutionAttributes != nil {\n\t\trequest.Attributes = &adminv1.CrossClusterTaskRequest_StartChildExecutionAttributes{\n\t\t\tStartChildExecutionAttributes: FromCrossClusterStartChildExecutionRequestAttributes(t.StartChildExecutionAttributes),\n\t\t}\n\t}\n\tif t.CancelExecutionAttributes != nil {\n\t\trequest.Attributes = &adminv1.CrossClusterTaskRequest_CancelExecutionAttributes{\n\t\t\tCancelExecutionAttributes: FromCrossClusterCancelExecutionRequestAttributes(t.CancelExecutionAttributes),\n\t\t}\n\t}\n\tif t.SignalExecutionAttributes != nil {\n\t\trequest.Attributes = &adminv1.CrossClusterTaskRequest_SignalExecutionAttributes{\n\t\t\tSignalExecutionAttributes: FromCrossClusterSignalExecutionRequestAttributes(t.SignalExecutionAttributes),\n\t\t}\n\t}\n\tif t.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\trequest.Attributes = &adminv1.CrossClusterTaskRequest_RecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\t\tRecordChildWorkflowExecutionCompleteRequestAttributes: FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\t}\n\t}\n\tif t.ApplyParentClosePolicyAttributes != nil {\n\t\trequest.Attributes = &adminv1.CrossClusterTaskRequest_ApplyParentClosePolicyRequestAttributes{\n\t\t\tApplyParentClosePolicyRequestAttributes: FromCrossClusterApplyParentClosePolicyRequestAttributes(t.ApplyParentClosePolicyAttributes),\n\t\t}\n\t}\n\treturn &request\n}\n\n// ToCrossClusterTaskRequest converts proto CrossClusterTaskRequest type to internal\nfunc ToCrossClusterTaskRequest(t *adminv1.CrossClusterTaskRequest) *types.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := types.CrossClusterTaskRequest{\n\t\tTaskInfo: ToCrossClusterTaskInfo(t.TaskInfo),\n\t}\n\tswitch attr := t.Attributes.(type) {\n\tcase *adminv1.CrossClusterTaskRequest_StartChildExecutionAttributes:\n\t\trequest.StartChildExecutionAttributes = ToCrossClusterStartChildExecutionRequestAttributes(attr.StartChildExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskRequest_CancelExecutionAttributes:\n\t\trequest.CancelExecutionAttributes = ToCrossClusterCancelExecutionRequestAttributes(attr.CancelExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskRequest_SignalExecutionAttributes:\n\t\trequest.SignalExecutionAttributes = ToCrossClusterSignalExecutionRequestAttributes(attr.SignalExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskRequest_RecordChildWorkflowExecutionCompleteRequestAttributes:\n\t\trequest.RecordChildWorkflowExecutionCompleteAttributes = ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(attr.RecordChildWorkflowExecutionCompleteRequestAttributes)\n\tcase *adminv1.CrossClusterTaskRequest_ApplyParentClosePolicyRequestAttributes:\n\t\trequest.ApplyParentClosePolicyAttributes = ToCrossClusterApplyParentClosePolicyRequestAttributes(attr.ApplyParentClosePolicyRequestAttributes)\n\t}\n\treturn &request\n}\n\n// FromCrossClusterTaskResponse converts internal CrossClusterTaskResponse type to proto\nfunc FromCrossClusterTaskResponse(t *types.CrossClusterTaskResponse) *adminv1.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tresponse := adminv1.CrossClusterTaskResponse{\n\t\tTaskId:      t.TaskID,\n\t\tTaskType:    FromCrossClusterTaskType(t.TaskType),\n\t\tTaskState:   int32(t.TaskState),\n\t\tFailedCause: FromCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n\tif t.StartChildExecutionAttributes != nil {\n\t\tresponse.Attributes = &adminv1.CrossClusterTaskResponse_StartChildExecutionAttributes{\n\t\t\tStartChildExecutionAttributes: FromCrossClusterStartChildExecutionResponseAttributes(t.StartChildExecutionAttributes),\n\t\t}\n\t}\n\tif t.CancelExecutionAttributes != nil {\n\t\tresponse.Attributes = &adminv1.CrossClusterTaskResponse_CancelExecutionAttributes{\n\t\t\tCancelExecutionAttributes: FromCrossClusterCancelExecutionResponseAttributes(t.CancelExecutionAttributes),\n\t\t}\n\t}\n\tif t.SignalExecutionAttributes != nil {\n\t\tresponse.Attributes = &adminv1.CrossClusterTaskResponse_SignalExecutionAttributes{\n\t\t\tSignalExecutionAttributes: FromCrossClusterSignalExecutionResponseAttributes(t.SignalExecutionAttributes),\n\t\t}\n\t}\n\tif t.RecordChildWorkflowExecutionCompleteAttributes != nil {\n\t\tresponse.Attributes = &adminv1.CrossClusterTaskResponse_RecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\t\tRecordChildWorkflowExecutionCompleteRequestAttributes: FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\t}\n\t}\n\tif t.ApplyParentClosePolicyAttributes != nil {\n\t\tresponse.Attributes = &adminv1.CrossClusterTaskResponse_ApplyParentClosePolicyResponseAttributes{\n\t\t\tApplyParentClosePolicyResponseAttributes: FromCrossClusterApplyParentClosePolicyResponseAttributes(t.ApplyParentClosePolicyAttributes),\n\t\t}\n\t}\n\treturn &response\n}\n\n// ToCrossClusterTaskResponse converts proto CrossClusterTaskResponse type to internal\nfunc ToCrossClusterTaskResponse(t *adminv1.CrossClusterTaskResponse) *types.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tresponse := types.CrossClusterTaskResponse{\n\t\tTaskID:      t.TaskId,\n\t\tTaskType:    ToCrossClusterTaskType(t.TaskType),\n\t\tTaskState:   int16(t.TaskState),\n\t\tFailedCause: ToCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n\tswitch attr := t.Attributes.(type) {\n\tcase *adminv1.CrossClusterTaskResponse_StartChildExecutionAttributes:\n\t\tresponse.StartChildExecutionAttributes = ToCrossClusterStartChildExecutionResponseAttributes(attr.StartChildExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskResponse_CancelExecutionAttributes:\n\t\tresponse.CancelExecutionAttributes = ToCrossClusterCancelExecutionResponseAttributes(attr.CancelExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskResponse_SignalExecutionAttributes:\n\t\tresponse.SignalExecutionAttributes = ToCrossClusterSignalExecutionResponseAttributes(attr.SignalExecutionAttributes)\n\tcase *adminv1.CrossClusterTaskResponse_RecordChildWorkflowExecutionCompleteRequestAttributes:\n\t\tresponse.RecordChildWorkflowExecutionCompleteAttributes = ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(attr.RecordChildWorkflowExecutionCompleteRequestAttributes)\n\tcase *adminv1.CrossClusterTaskResponse_ApplyParentClosePolicyResponseAttributes:\n\t\tresponse.ApplyParentClosePolicyAttributes = ToCrossClusterApplyParentClosePolicyResponseAttributes(attr.ApplyParentClosePolicyResponseAttributes)\n\t}\n\treturn &response\n}\n\n// FromCrossClusterTaskRequestArray converts internal CrossClusterTaskRequest type array to proto\nfunc FromCrossClusterTaskRequestArray(t []*types.CrossClusterTaskRequest) *adminv1.CrossClusterTaskRequests {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.CrossClusterTaskRequest, len(t))\n\tfor i := range t {\n\t\tv[i] = FromCrossClusterTaskRequest(t[i])\n\t}\n\treturn &adminv1.CrossClusterTaskRequests{\n\t\tTaskRequests: v,\n\t}\n}\n\n// ToCrossClusterTaskRequestArray converts proto CrossClusterTaskRequest type array to internal\nfunc ToCrossClusterTaskRequestArray(t *adminv1.CrossClusterTaskRequests) []*types.CrossClusterTaskRequest {\n\tif t == nil || t.TaskRequests == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.CrossClusterTaskRequest, len(t.TaskRequests))\n\tfor i := range t.TaskRequests {\n\t\tv[i] = ToCrossClusterTaskRequest(t.TaskRequests[i])\n\t}\n\treturn v\n}\n\n// FromCrossClusterTaskRequestMap converts internal CrossClusterTaskRequest type map to proto\nfunc FromCrossClusterTaskRequestMap(t map[int32][]*types.CrossClusterTaskRequest) map[int32]*adminv1.CrossClusterTaskRequests {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]*adminv1.CrossClusterTaskRequests, len(t))\n\tfor key := range t {\n\t\tv[key] = FromCrossClusterTaskRequestArray(t[key])\n\t}\n\treturn v\n}\n\n// ToCrossClusterTaskRequestMap converts proto CrossClusterTaskRequest type map to internal\nfunc ToCrossClusterTaskRequestMap(t map[int32]*adminv1.CrossClusterTaskRequests) map[int32][]*types.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32][]*types.CrossClusterTaskRequest, len(t))\n\tfor key := range t {\n\t\tvalue := ToCrossClusterTaskRequestArray(t[key])\n\t\tif value == nil {\n\t\t\t// grpc can't differentiate between empty array or nil array\n\t\t\t// our application logic ensure no nil array will be returned\n\t\t\t// for CrossClusterTaskRequest, so always convert to empty array\n\t\t\t// we only need the special handling here as this array is used\n\t\t\t// as a map value in GetCrossClusterTasksResponse,\n\t\t\t// and if the map value is nil, THRIFT won't be able to encode the value\n\t\t\t// this may happen when we are using grpc within a cluster but thrift across cluster\n\t\t\tvalue = []*types.CrossClusterTaskRequest{}\n\t\t}\n\t\tv[key] = value\n\t}\n\treturn v\n}\n\n// FromGetTaskFailedCauseMap converts internal GetTaskFailedCause type map to proto\nfunc FromGetTaskFailedCauseMap(t map[int32]types.GetTaskFailedCause) map[int32]adminv1.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]adminv1.GetTaskFailedCause, len(t))\n\tfor key, value := range t {\n\t\tv[key] = FromGetTaskFailedCause(&value)\n\t}\n\treturn v\n}\n\n// ToGetTaskFailedCauseMap converts proto GetTaskFailedCause type map to internal\nfunc ToGetTaskFailedCauseMap(t map[int32]adminv1.GetTaskFailedCause) map[int32]types.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]types.GetTaskFailedCause, len(t))\n\tfor key := range t {\n\t\tif internalValue := ToGetTaskFailedCause(t[key]); internalValue != nil {\n\t\t\tv[key] = *internalValue\n\t\t}\n\t}\n\treturn v\n}\n\n// FromCrossClusterTaskResponseArray converts internal CrossClusterTaskResponse type array to proto\nfunc FromCrossClusterTaskResponseArray(t []*types.CrossClusterTaskResponse) []*adminv1.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*adminv1.CrossClusterTaskResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = FromCrossClusterTaskResponse(t[i])\n\t}\n\treturn v\n}\n\n// ToCrossClusterTaskResponseArray converts proto CrossClusterTaskResponse type array to internal\nfunc ToCrossClusterTaskResponseArray(t []*adminv1.CrossClusterTaskResponse) []*types.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.CrossClusterTaskResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = ToCrossClusterTaskResponse(t[i])\n\t}\n\treturn v\n}\n\nfunc FromHistoryDLQCountEntryMap(t map[types.HistoryDLQCountKey]int64) []*adminv1.HistoryDLQCountEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tentries := make([]*adminv1.HistoryDLQCountEntry, 0, len(t))\n\tfor key, count := range t {\n\t\tentries = append(entries, &adminv1.HistoryDLQCountEntry{\n\t\t\tShardId:       key.ShardID,\n\t\t\tSourceCluster: key.SourceCluster,\n\t\t\tCount:         count,\n\t\t})\n\t}\n\treturn entries\n}\n\nfunc ToHistoryDLQCountEntryMap(t []*adminv1.HistoryDLQCountEntry) map[types.HistoryDLQCountKey]int64 {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tentries := make(map[types.HistoryDLQCountKey]int64, len(t))\n\tfor _, entry := range t {\n\t\tkey := types.HistoryDLQCountKey{ShardID: entry.ShardId, SourceCluster: entry.SourceCluster}\n\t\tentries[key] = entry.Count\n\t}\n\treturn entries\n}\n\n// ToAny converts thrift Any type to internal\nfunc ToAny(t *sharedv1.Any) *types.Any {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Any{\n\t\tValueType: t.GetValueType(),\n\t\tValue:     t.Value,\n\t}\n}\n\n// FromAny converts internal Any type to thrift\nfunc FromAny(t *types.Any) *sharedv1.Any {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &sharedv1.Any{\n\t\tValueType: t.ValueType,\n\t\tValue:     t.Value,\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/proto/shared_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage proto\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/gogo/protobuf/proto\"\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\tsharedv1 \"github.com/uber/cadence/.gen/proto/shared/v1\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/testutils\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestHostInfo(t *testing.T) {\n\tfor _, item := range []*types.HostInfo{nil, {}, &testdata.HostInfo} {\n\t\tassert.Equal(t, item, ToHostInfo(FromHostInfo(item)))\n\t}\n}\nfunc TestMembershipInfo(t *testing.T) {\n\tfor _, item := range []*types.MembershipInfo{nil, {}, &testdata.MembershipInfo} {\n\t\tassert.Equal(t, item, ToMembershipInfo(FromMembershipInfo(item)))\n\t}\n}\nfunc TestDomainCacheInfo(t *testing.T) {\n\tfor _, item := range []*types.DomainCacheInfo{nil, {}, &testdata.DomainCacheInfo} {\n\t\tassert.Equal(t, item, ToDomainCacheInfo(FromDomainCacheInfo(item)))\n\t}\n}\nfunc TestRingInfo(t *testing.T) {\n\tfor _, item := range []*types.RingInfo{nil, {}, &testdata.RingInfo} {\n\t\tassert.Equal(t, item, ToRingInfo(FromRingInfo(item)))\n\t}\n}\nfunc TestTransientDecisionInfo(t *testing.T) {\n\tfor _, item := range []*types.TransientDecisionInfo{nil, {}, &testdata.TransientDecisionInfo} {\n\t\tassert.Equal(t, item, ToTransientDecisionInfo(FromTransientDecisionInfo(item)))\n\t}\n}\nfunc TestVersionHistories(t *testing.T) {\n\tfor _, item := range []*types.VersionHistories{nil, {}, &testdata.VersionHistories} {\n\t\tassert.Equal(t, item, ToVersionHistories(FromVersionHistories(item)))\n\t}\n}\nfunc TestVersionHistory(t *testing.T) {\n\tfor _, item := range []*types.VersionHistory{nil, {}, &testdata.VersionHistory} {\n\t\tassert.Equal(t, item, ToVersionHistory(FromVersionHistory(item)))\n\t}\n}\nfunc TestVersionHistoryItem(t *testing.T) {\n\tfor _, item := range []*types.VersionHistoryItem{nil, {}, &testdata.VersionHistoryItem} {\n\t\tassert.Equal(t, item, ToVersionHistoryItem(FromVersionHistoryItem(item)))\n\t}\n}\nfunc TestHostInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.HostInfo{nil, {}, testdata.HostInfoArray} {\n\t\tassert.Equal(t, item, ToHostInfoArray(FromHostInfoArray(item)))\n\t}\n}\nfunc TestVersionHistoryArray(t *testing.T) {\n\tfor _, item := range [][]*types.VersionHistory{nil, {}, testdata.VersionHistoryArray} {\n\t\tassert.Equal(t, item, ToVersionHistoryArray(FromVersionHistoryArray(item)))\n\t}\n}\nfunc TestRingInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.RingInfo{nil, {}, testdata.RingInfoArray} {\n\t\tassert.Equal(t, item, ToRingInfoArray(FromRingInfoArray(item)))\n\t}\n}\nfunc TestDomainTaskAttributes(t *testing.T) {\n\tfor _, item := range []*types.DomainTaskAttributes{nil, &testdata.DomainTaskAttributes} {\n\t\tassert.Equal(t, item, ToDomainTaskAttributes(FromDomainTaskAttributes(item)))\n\t}\n}\nfunc TestFailoverMarkerAttributes(t *testing.T) {\n\tfor _, item := range []*types.FailoverMarkerAttributes{nil, {}, &testdata.FailoverMarkerAttributes} {\n\t\tassert.Equal(t, item, ToFailoverMarkerAttributes(FromFailoverMarkerAttributes(item)))\n\t}\n}\nfunc TestFailoverMarkerToken(t *testing.T) {\n\tfor _, item := range []*types.FailoverMarkerToken{nil, {}, &testdata.FailoverMarkerToken} {\n\t\tassert.Equal(t, item, ToFailoverMarkerToken(FromFailoverMarkerToken(item)))\n\t}\n}\nfunc TestHistoryTaskV2Attributes(t *testing.T) {\n\tfor _, item := range []*types.HistoryTaskV2Attributes{nil, {}, &testdata.HistoryTaskV2Attributes} {\n\t\tassert.Equal(t, item, ToHistoryTaskV2Attributes(FromHistoryTaskV2Attributes(item)))\n\t}\n}\nfunc TestReplicationMessages(t *testing.T) {\n\tfor _, item := range []*types.ReplicationMessages{nil, {}, &testdata.ReplicationMessages} {\n\t\tassert.Equal(t, item, ToReplicationMessages(FromReplicationMessages(item)))\n\t}\n}\nfunc TestReplicationTaskInfo(t *testing.T) {\n\tfor _, item := range []*types.ReplicationTaskInfo{nil, {}, &testdata.ReplicationTaskInfo} {\n\t\tassert.Equal(t, item, ToReplicationTaskInfo(FromReplicationTaskInfo(item)))\n\t}\n}\nfunc TestReplicationToken(t *testing.T) {\n\tfor _, item := range []*types.ReplicationToken{nil, {}, &testdata.ReplicationToken} {\n\t\tassert.Equal(t, item, ToReplicationToken(FromReplicationToken(item)))\n\t}\n}\nfunc TestSyncActivityTaskAttributes(t *testing.T) {\n\tfor _, item := range []*types.SyncActivityTaskAttributes{nil, {}, &testdata.SyncActivityTaskAttributes} {\n\t\tassert.Equal(t, item, ToSyncActivityTaskAttributes(FromSyncActivityTaskAttributes(item)))\n\t}\n}\nfunc TestSyncShardStatus(t *testing.T) {\n\tfor _, item := range []*types.SyncShardStatus{nil, {}, &testdata.SyncShardStatus} {\n\t\tassert.Equal(t, item, ToSyncShardStatus(FromSyncShardStatus(item)))\n\t}\n}\nfunc TestSyncShardStatusTaskAttributes(t *testing.T) {\n\tfor _, item := range []*types.SyncShardStatusTaskAttributes{nil, {}, &testdata.SyncShardStatusTaskAttributes} {\n\t\tassert.Equal(t, item, ToSyncShardStatusTaskAttributes(FromSyncShardStatusTaskAttributes(item)))\n\t}\n}\nfunc TestReplicationTaskInfoArray(t *testing.T) {\n\tfor _, item := range [][]*types.ReplicationTaskInfo{nil, {}, testdata.ReplicationTaskInfoArray} {\n\t\tassert.Equal(t, item, ToReplicationTaskInfoArray(FromReplicationTaskInfoArray(item)))\n\t}\n}\nfunc TestReplicationTaskArray(t *testing.T) {\n\tfor _, item := range [][]*types.ReplicationTask{nil, {}, testdata.ReplicationTaskArray} {\n\t\tassert.Equal(t, item, ToReplicationTaskArray(FromReplicationTaskArray(item)))\n\t}\n}\nfunc TestReplicationTokenArray(t *testing.T) {\n\tfor _, item := range [][]*types.ReplicationToken{nil, {}, testdata.ReplicationTokenArray} {\n\t\tassert.Equal(t, item, ToReplicationTokenArray(FromReplicationTokenArray(item)))\n\t}\n}\nfunc TestReplicationMessagesMap(t *testing.T) {\n\tfor _, item := range []map[int32]*types.ReplicationMessages{nil, {}, testdata.ReplicationMessagesMap} {\n\t\tassert.Equal(t, item, ToReplicationMessagesMap(FromReplicationMessagesMap(item)))\n\t}\n}\nfunc TestReplicationTask(t *testing.T) {\n\tfor _, item := range []*types.ReplicationTask{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ReplicationTask_Domain,\n\t\t&testdata.ReplicationTask_Failover,\n\t\t&testdata.ReplicationTask_History,\n\t\t&testdata.ReplicationTask_SyncActivity,\n\t\t&testdata.ReplicationTask_SyncShard,\n\t} {\n\t\tassert.Equal(t, item, ToReplicationTask(FromReplicationTask(item)))\n\t}\n}\nfunc TestFailoverMarkerTokenArray(t *testing.T) {\n\tfor _, item := range [][]*types.FailoverMarkerToken{nil, {}, testdata.FailoverMarkerTokenArray} {\n\t\tassert.Equal(t, item, ToFailoverMarkerTokenArray(FromFailoverMarkerTokenArray(item)))\n\t}\n}\nfunc TestVersionHistoryItemArray(t *testing.T) {\n\tfor _, item := range [][]*types.VersionHistoryItem{nil, {}, testdata.VersionHistoryItemArray} {\n\t\tassert.Equal(t, item, ToVersionHistoryItemArray(FromVersionHistoryItemArray(item)))\n\t}\n}\nfunc TestEventIDVersionPair(t *testing.T) {\n\tassert.Nil(t, FromEventIDVersionPair(nil, nil))\n\tassert.Nil(t, ToEventID(nil))\n\tassert.Nil(t, ToEventVersion(nil))\n\n\tpair := FromEventIDVersionPair(common.Int64Ptr(testdata.EventID1), common.Int64Ptr(testdata.Version1))\n\tassert.Equal(t, testdata.EventID1, *ToEventID(pair))\n\tassert.Equal(t, testdata.Version1, *ToEventVersion(pair))\n}\n\nfunc TestCrossClusterTaskInfo(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskInfo{nil, {}, &testdata.CrossClusterTaskInfo} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskInfo(FromCrossClusterTaskInfo(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CrossClusterTaskRequestStartChildExecution,\n\t\t&testdata.CrossClusterTaskRequestCancelExecution,\n\t\t&testdata.CrossClusterTaskRequestSignalExecution,\n\t} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequest(FromCrossClusterTaskRequest(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskResponse(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CrossClusterTaskResponseStartChildExecution,\n\t\t&testdata.CrossClusterTaskResponseCancelExecution,\n\t\t&testdata.CrossClusterTaskResponseSignalExecution,\n\t} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskResponse(FromCrossClusterTaskResponse(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequestArray(t *testing.T) {\n\tfor _, item := range [][]*types.CrossClusterTaskRequest{nil, {}, testdata.CrossClusterTaskRequestArray} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequestArray(FromCrossClusterTaskRequestArray(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskResponseArray(t *testing.T) {\n\tfor _, item := range [][]*types.CrossClusterTaskResponse{nil, {}, testdata.CrossClusterTaskResponseArray} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskResponseArray(FromCrossClusterTaskResponseArray(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequestMap(t *testing.T) {\n\tfor _, item := range []map[int32][]*types.CrossClusterTaskRequest{nil, {}, testdata.CrossClusterTaskRequestMap} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequestMap(FromCrossClusterTaskRequestMap(item)))\n\t}\n\tassert.Equal(\n\t\tt,\n\t\tmap[int32][]*types.CrossClusterTaskRequest{\n\t\t\t0: {},\n\t\t},\n\t\tToCrossClusterTaskRequestMap(FromCrossClusterTaskRequestMap(\n\t\t\tmap[int32][]*types.CrossClusterTaskRequest{\n\t\t\t\t0: nil,\n\t\t\t},\n\t\t)),\n\t)\n}\n\nfunc TestGetTaskFailedCauseMap(t *testing.T) {\n\tfor _, item := range []map[int32]types.GetTaskFailedCause{nil, {}, testdata.GetCrossClusterTaskFailedCauseMap} {\n\t\tassert.Equal(t, item, ToGetTaskFailedCauseMap(FromGetTaskFailedCauseMap(item)))\n\t}\n}\n\nfunc TestCrossClusterApplyParentClosePolicyRequestAttributes(t *testing.T) {\n\titem := testdata.CrossClusterApplyParentClosePolicyRequestAttributes\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToCrossClusterApplyParentClosePolicyRequestAttributes(\n\t\t\tFromCrossClusterApplyParentClosePolicyRequestAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestApplyParentClosePolicyAttributes(t *testing.T) {\n\titem := testdata.ApplyParentClosePolicyAttributes\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToApplyParentClosePolicyAttributes(\n\t\t\tFromApplyParentClosePolicyAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestApplyParentClosePolicyResult(t *testing.T) {\n\titem := testdata.ApplyParentClosePolicyResult\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToApplyParentClosePolicyResult(\n\t\t\tFromApplyParentClosePolicyResult(&item),\n\t\t),\n\t)\n}\n\nfunc TestCrossClusterApplyParentClosePolicyResponse(t *testing.T) {\n\titem := testdata.CrossClusterApplyParentClosePolicyResponseWithChildren\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToCrossClusterApplyParentClosePolicyResponseAttributes(\n\t\t\tFromCrossClusterApplyParentClosePolicyResponseAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestAny(t *testing.T) {\n\tt.Run(\"sanity check\", func(t *testing.T) {\n\t\tinternal := types.Any{\n\t\t\tValueType: \"testing\",\n\t\t\tValue:     []byte(`test`),\n\t\t}\n\t\trpc := sharedv1.Any{\n\t\t\tValueType: \"testing\",\n\t\t\tValue:     []byte(`test`),\n\t\t}\n\t\trequire.Equal(t, &rpc, FromAny(&internal))\n\t\trequire.Equal(t, &internal, ToAny(&rpc))\n\t})\n\n\tt.Run(\"round trip nils\", func(t *testing.T) {\n\t\t// somewhat annoying in fuzzing and there are few possibilities, so tested separately\n\t\tassert.Nil(t, FromAny(ToAny(nil)), \"nil proto -> internal -> proto => should result in nil\")\n\t\tassert.Nil(t, ToAny(FromAny(nil)), \"nil internal -> proto -> internal => should result in nil\")\n\t})\n\n\tt.Run(\"round trip from internal\", func(t *testing.T) {\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"empty data\", \"filled data\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\tvar orig types.Any\n\t\t\tf.Fuzz(&orig)\n\t\t\tout := ToAny(FromAny(&orig))\n\t\t\tassert.Equal(t, &orig, out, \"did not survive round-tripping\")\n\t\t\t// report what branch of behavior was fuzzed occurred\n\t\t\tif len(orig.Value) == 0 {\n\t\t\t\treturn \"empty data\" // ignoring nil vs empty difference\n\t\t\t}\n\t\t\treturn \"filled data\"\n\t\t})\n\t})\n\tt.Run(\"round trip from proto\", func(t *testing.T) {\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"empty data\", \"filled data\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\t// unfortunately:\n\t\t\t// - gofuzz panics when it encounters interface fields (so this cannot be done on oneof fields)\n\t\t\t//   - not directly relevant for Any, but causes issues for fuzzing other types\n\t\t\t// - it populates the XXX_ fields and these can be hard to clear\n\t\t\t// so this is fuzz-filled by hand with specific fields rather than as a whole.\n\t\t\tvar orig sharedv1.Any\n\t\t\tf.Fuzz(&orig.ValueType)\n\t\t\tf.Fuzz(&orig.Value)\n\t\t\tout := FromAny(ToAny(&orig))\n\t\t\tassert.Equal(t, &orig, out, \"did not survive round-tripping\")\n\t\t\tif len(orig.Value) == 0 {\n\t\t\t\treturn \"empty data\" // ignoring nil vs empty difference\n\t\t\t}\n\t\t\treturn \"filled data\"\n\t\t})\n\t})\n\n\tt.Run(\"can contain proto data\", func(t *testing.T) {\n\t\t// pushing thrift or proto data through this type is very much expected,\n\t\t// so this is both evidence that it's possible and an example for how to do so.\n\t\t//\n\t\t// note that there is an equivalent test in mapper/thrift/shared_test.go.\n\t\t// they are structurally the same, but encoding/decoding details vary a bit.\n\n\t\t// some helpers because it's a bit verbose.\n\t\t//\n\t\t// note that these work for both types in this test: they can encode and decode *any* proto data,\n\t\t// you just have to make sure you pass the same type to both encode and decode via some other\n\t\t// source of info (i.e. the ValueType field).\n\t\tencode := func(thing proto.Marshaler) []byte {\n\t\t\tdata, err := thing.Marshal()\n\t\t\trequire.NoErrorf(t, err, \"could not Marshal the target type: %T\", thing)\n\t\t\treturn data\n\t\t}\n\t\tdecode := func(data []byte, target proto.Unmarshaler) {\n\t\t\terr := target.Unmarshal(data)\n\t\t\trequire.NoErrorf(t, err, \"could not Unmarshal to the target type: %T\", target)\n\t\t}\n\n\t\t// --- create the original data, a proto object, and encode it by hand\n\t\torig := &apiv1.WorkflowExecution{\n\t\t\tWorkflowId: testdata.WorkflowID,\n\t\t\tRunId:      testdata.RunID,\n\t\t}\n\t\tinternalBytes := encode(orig)\n\n\t\t// --- put that data into the custom Any type\n\t\t// proto (unfortunately) maintains a type-registry, which means we can look up the unique name for this type...\n\t\t//\n\t\t// ...BUT this is potentially unsafe in normal code: proto type names can be changed without breaking binary compatibility.\n\t\t// we are unlikely to ever do so, but for this reason, hard-coding is actually a bit safer.\n\t\tvar typeName = \"uber.cadence.api.v1.WorkflowExecution\" // current value of proto.MessageName(orig)\n\t\tanyVal := &types.Any{\n\t\t\tValueType: typeName,\n\t\t\tValue:     internalBytes, // store proto bytes in the Any\n\t\t}\n\n\t\t// --- convert the whole container to proto (mimics making a call via yarpc)\n\t\tprotoAny := FromAny(anyVal)      // we map to the rpc type\n\t\tnetworkBytes := encode(protoAny) // yarpc does this\n\t\t// ^ this is what's sent over the network.\n\n\t\t// as a side note:\n\t\t// the final data is not double-encoded, so this \"encode -> wrap -> encode\" process is reasonably efficient.\n\t\t//\n\t\t// Thrift and Proto can efficiently move around binary blobs like this, as it's essentially just a memcpy between\n\t\t// the input and the output, and there's no `\\0` to `\\\\0` escaping or base64 encoding or whatever needed.\n\t\t//\n\t\t// no behavior depends on this, it's just presented here as evidence that this Any-wrapper does not meaningfully\n\t\t// change any RPC design concerns: anything you would do with normal RPC can be done through an Any if you need\n\t\t// loose typing, the change-stability / performance / etc is entirely unaffected.\n\t\t//\n\t\t// compare via a sliding window to find the place it overlaps, to prove that this is true:\n\t\tfound := false\n\t\tfor i := 0; i <= len(networkBytes)-len(internalBytes); i++ {\n\t\t\tif reflect.DeepEqual(internalBytes, networkBytes[i:i+len(internalBytes)]) {\n\t\t\t\tfound = true\n\t\t\t\tt.Logf(\"Found matching bytes at index %v\", i) // currently at index 41\n\t\t\t}\n\t\t}\n\t\t// *should* be true for efficiency's sake, but is not truly necessary for correct behavior\n\t\tassert.Truef(t, found, \"did not find internal bytes within network bytes, might be paying double-encoding costs:\\n\\tinternal: %v\\n\\tnetwork:  %v\", internalBytes, networkBytes)\n\n\t\t// --- the network pushes the data to a new location ---\n\n\t\t// --- on the receiving side, we map to internal types like normal\n\t\tvar outAny sharedv1.Any\n\t\tdecode(networkBytes, &outAny) // yarpc does this\n\t\toutAnyVal := ToAny(&outAny)   // we map to internal types\n\n\t\t// --- and finally decode the any-typed data by hand\n\t\trequire.Equal(t, typeName, outAnyVal.ValueType, \"type name through RPC should match the original type name\")\n\t\tvar outOrig apiv1.WorkflowExecution // selected based on the ValueType contents\n\t\tdecode(outAnyVal.Value, &outOrig)   // do the actual custom decoding\n\t\tassert.NotEmpty(t, outOrig, \"sanity check, decoded value should not be empty\")\n\t\tassert.Equal(t, orig, &outOrig, \"final round-tripped Any-contained data should be identical to the original object\")\n\t})\n}\n\nfunc TestFromCrossClusterTaskResponseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterTaskResponse, ToCrossClusterTaskResponse,\n\t\ttestutils.WithCustomFuncs(\n\t\t\tCrossClusterTaskResponseFuzzer,\n\t\t\ttestutils.CrossClusterTaskTypeFuzzer,\n\t\t\ttestutils.CrossClusterTaskFailedCauseFuzzer,\n\t\t),\n\t)\n}\n\n// CrossClusterTaskResponseFuzzer ensures only one attribute field is set based on TaskType (oneof constraint)\nfunc CrossClusterTaskResponseFuzzer(r *types.CrossClusterTaskResponse, c fuzz.Continue) {\n\tc.FuzzNoCustom(r)\n\n\t// Ensure TaskType is set to a valid value (0-4) using the enum fuzzer\n\tif r.TaskType == nil {\n\t\tvar taskType types.CrossClusterTaskType\n\t\ttestutils.CrossClusterTaskTypeFuzzer(&taskType, c)\n\t\tr.TaskType = &taskType\n\t} else {\n\t\ttestutils.CrossClusterTaskTypeFuzzer(r.TaskType, c)\n\t}\n\n\t// Ensure FailedCause is valid if set (0-5) using the enum fuzzer\n\tif r.FailedCause != nil {\n\t\ttestutils.CrossClusterTaskFailedCauseFuzzer(r.FailedCause, c)\n\t}\n\n\t// Based on TaskType, clear all attributes except the matching one\n\tswitch *r.TaskType {\n\tcase 0: // StartChildExecution\n\t\tr.CancelExecutionAttributes = nil\n\t\tr.SignalExecutionAttributes = nil\n\t\tr.RecordChildWorkflowExecutionCompleteAttributes = nil\n\t\tr.ApplyParentClosePolicyAttributes = nil\n\tcase 1: // CancelExecution\n\t\tr.StartChildExecutionAttributes = nil\n\t\tr.SignalExecutionAttributes = nil\n\t\tr.RecordChildWorkflowExecutionCompleteAttributes = nil\n\t\tr.ApplyParentClosePolicyAttributes = nil\n\tcase 2: // SignalExecution\n\t\tr.StartChildExecutionAttributes = nil\n\t\tr.CancelExecutionAttributes = nil\n\t\tr.RecordChildWorkflowExecutionCompleteAttributes = nil\n\t\tr.ApplyParentClosePolicyAttributes = nil\n\tcase 3: // RecordChildWorkflowExecutionComplete\n\t\tr.StartChildExecutionAttributes = nil\n\t\tr.CancelExecutionAttributes = nil\n\t\tr.SignalExecutionAttributes = nil\n\t\tr.ApplyParentClosePolicyAttributes = nil\n\tcase 4: // ApplyParentClosePolicy\n\t\tr.StartChildExecutionAttributes = nil\n\t\tr.CancelExecutionAttributes = nil\n\t\tr.SignalExecutionAttributes = nil\n\t\tr.RecordChildWorkflowExecutionCompleteAttributes = nil\n\t}\n}\n\nfunc TestHostInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromHostInfo, ToHostInfo)\n}\n\nfunc TestMembershipInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromMembershipInfo, ToMembershipInfo)\n}\n\nfunc TestDomainCacheInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromDomainCacheInfo, ToDomainCacheInfo)\n}\n\nfunc TestRingInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromRingInfo, ToRingInfo)\n}\n\nfunc TestVersionHistoriesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromVersionHistories, ToVersionHistories)\n}\n\nfunc TestVersionHistoryFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromVersionHistory, ToVersionHistory)\n}\n\nfunc TestVersionHistoryItemFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromVersionHistoryItem, ToVersionHistoryItem)\n}\n\nfunc TestPersistenceSettingFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPersistenceSetting, ToPersistenceSetting)\n}\n\nfunc TestPersistenceFeatureFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPersistenceFeature, ToPersistenceFeature)\n}\n\nfunc TestPersistenceInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromPersistenceInfo, ToPersistenceInfo)\n}\n\nfunc TestCrossClusterTaskTypeFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterTaskType, ToCrossClusterTaskType,\n\t\ttestutils.WithCustomFuncs(testutils.CrossClusterTaskTypeFuzzer),\n\t)\n}\n\nfunc TestGetTaskFailedCauseFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromGetTaskFailedCause, ToGetTaskFailedCause,\n\t\ttestutils.WithCustomFuncs(testutils.GetTaskFailedCauseFuzzer),\n\t)\n}\n\nfunc TestApplyParentClosePolicyStatusFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromApplyParentClosePolicyStatus, ToApplyParentClosePolicyStatus,\n\t\ttestutils.WithCustomFuncs(testutils.CrossClusterTaskFailedCauseFuzzer),\n\t)\n}\n\nfunc TestDomainTaskAttributesFuzz(t *testing.T) {\n\t// Info, Config, and ReplicationConfig are built into the 'Domain' field.\n\t// This mapping is tested in TestDescribeDomainResponseDomainFuzz\n\ttestutils.RunMapperFuzzTest(t, FromDomainTaskAttributes, ToDomainTaskAttributes,\n\t\ttestutils.WithCustomFuncs(testutils.DomainOperationFuzzer, testutils.DomainStatusFuzzer),\n\t\ttestutils.WithExcludedFields(\"Info\", \"Config\", \"ReplicationConfig\"),\n\t)\n}\n\nfunc TestFailoverMarkerAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromFailoverMarkerAttributes, ToFailoverMarkerAttributes)\n}\n\nfunc TestFailoverMarkerTokenFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromFailoverMarkerToken, ToFailoverMarkerToken)\n}\n\nfunc TestHistoryTaskV2AttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromHistoryTaskV2Attributes, ToHistoryTaskV2Attributes,\n\t\ttestutils.WithCustomFuncs(testutils.EncodingTypeFuzzer),\n\t)\n}\n\nfunc TestSyncActivityTaskAttributesFuzz(t *testing.T) {\n\t// [BUG] FailureDetails is not round-trip compatible\n\ttestutils.RunMapperFuzzTest(t, FromSyncActivityTaskAttributes, ToSyncActivityTaskAttributes,\n\t\ttestutils.WithCommonEnumFuzzers(),\n\t\ttestutils.WithExcludedFields(\"LastFailureDetails\"),\n\t)\n}\n\nfunc TestSyncShardStatusFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromSyncShardStatus, ToSyncShardStatus)\n}\n\nfunc TestSyncShardStatusTaskAttributesFuzz(t *testing.T) {\n\t// [BUG] ShardID gets truncated (int64 -> int32 issue)\n\ttestutils.RunMapperFuzzTest(t, FromSyncShardStatusTaskAttributes, ToSyncShardStatusTaskAttributes,\n\t\ttestutils.WithExcludedFields(\"ShardID\"),\n\t)\n}\n\nfunc TestReplicationMessagesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromReplicationMessages, ToReplicationMessages,\n\t\ttestutils.WithCustomFuncs(testutils.ReplicationTaskTypeFuzzer, testutils.DomainOperationFuzzer),\n\t\ttestutils.WithExcludedFields(\"ReplicationTasks\"), // Tested separately in TestReplicationTaskFuzz\n\t)\n}\n\nfunc TestReplicationTaskInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromReplicationTaskInfo, ToReplicationTaskInfo)\n}\n\nfunc TestReplicationTokenFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromReplicationToken, ToReplicationToken)\n}\n\nfunc TestCrossClusterTaskInfoFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterTaskInfo, ToCrossClusterTaskInfo,\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.CrossClusterTaskTypeFuzzer,\n\t\t\ttestutils.CrossClusterTaskFailedCauseFuzzer,\n\t\t),\n\t)\n}\n\nfunc TestCrossClusterStartChildExecutionRequestAttributesFuzz(t *testing.T) {\n\t// [BUG] TargetRunID is not round-trip compatible, when it is set to an empty string, it becomes nil\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterStartChildExecutionRequestAttributes, ToCrossClusterStartChildExecutionRequestAttributes,\n\t\ttestutils.WithCommonEnumFuzzers(),\n\t\ttestutils.WithExcludedFields(\"InitiatedEventAttributes\", \"TargetRunID\"), // InitiatedEventAttributes is tested in FromStartChildWorkflowExecutionInitiatedEventAttributesFuzz\n\t)\n}\n\nfunc TestCrossClusterStartChildExecutionResponseAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterStartChildExecutionResponseAttributes, ToCrossClusterStartChildExecutionResponseAttributes)\n}\n\nfunc TestCrossClusterCancelExecutionRequestAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterCancelExecutionRequestAttributes, ToCrossClusterCancelExecutionRequestAttributes)\n}\n\nfunc TestCrossClusterCancelExecutionResponseAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterCancelExecutionResponseAttributes, ToCrossClusterCancelExecutionResponseAttributes)\n}\n\nfunc TestCrossClusterSignalExecutionRequestAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterSignalExecutionRequestAttributes, ToCrossClusterSignalExecutionRequestAttributes)\n}\n\nfunc TestCrossClusterSignalExecutionResponseAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterSignalExecutionResponseAttributes, ToCrossClusterSignalExecutionResponseAttributes)\n}\n\nfunc TestCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes, ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes,\n\t\ttestutils.WithCommonEnumFuzzers(),\n\t\ttestutils.WithExcludedFields(\"CompletionEvent\"), // CompletionEvent uses FromHistoryEvent which is tested in TestHistoryEventFuzz\n\t)\n}\n\nfunc TestCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes, ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes)\n}\n\nfunc TestCrossClusterApplyParentClosePolicyResponseAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterApplyParentClosePolicyResponseAttributes, ToCrossClusterApplyParentClosePolicyResponseAttributes,\n\t\ttestutils.WithCustomFuncs(testutils.CrossClusterTaskFailedCauseFuzzer),\n\t)\n}\n\nfunc TestApplyParentClosePolicyAttributesFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromApplyParentClosePolicyAttributes, ToApplyParentClosePolicyAttributes)\n}\n\nfunc TestApplyParentClosePolicyResultFuzz(t *testing.T) {\n\ttestutils.RunMapperFuzzTest(t, FromApplyParentClosePolicyResult, ToApplyParentClosePolicyResult,\n\t\ttestutils.WithCustomFuncs(testutils.CrossClusterTaskFailedCauseFuzzer),\n\t)\n}\n\nfunc TestCrossClusterTaskRequestFuzz(t *testing.T) {\n\t// This type has a oneof constraint, which doesn't lend itself well to fuzzing\n\t// All child attributes are tested separately, so we exclude them here and test only TaskInfo\n\ttestutils.RunMapperFuzzTest(t, FromCrossClusterTaskRequest, ToCrossClusterTaskRequest,\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.CrossClusterTaskTypeFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\n\t\t\t\"StartChildExecutionAttributes\",                  // Tested in TestCrossClusterStartChildExecutionRequestAttributesFuzz\n\t\t\t\"CancelExecutionAttributes\",                      // Tested in TestCrossClusterCancelExecutionRequestAttributesFuzz\n\t\t\t\"SignalExecutionAttributes\",                      // Tested in TestCrossClusterSignalExecutionRequestAttributesFuzz\n\t\t\t\"RecordChildWorkflowExecutionCompleteAttributes\", // Tested in TestCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributesFuzz\n\t\t\t\"ApplyParentClosePolicyAttributes\",               // Tested in TestCrossClusterApplyParentClosePolicyRequestAttributesFuzz\n\t\t),\n\t)\n}\n\nfunc TestReplicationTaskFuzz(t *testing.T) {\n\t// This type has a oneof constraint, which doesn't lend itself well to fuzzing\n\t// All child attributes are tested separately, so we exclude them here and test only TaskInfo\n\ttestutils.RunMapperFuzzTest(t, FromReplicationTask, ToReplicationTask,\n\t\ttestutils.WithCustomFuncs(\n\t\t\ttestutils.ReplicationTaskTypeFuzzer,\n\t\t),\n\t\ttestutils.WithExcludedFields(\n\t\t\t\"DomainTaskAttributes\",          // Tested in TestDomainTaskAttributesFuzz\n\t\t\t\"SyncShardStatusTaskAttributes\", // Tested in TestSyncShardStatusTaskAttributesFuzz\n\t\t\t\"SyncActivityTaskAttributes\",    // Tested in TestSyncActivityTaskAttributesFuzz\n\t\t\t\"HistoryTaskV2Attributes\",       // Tested in TestHistoryTaskV2AttributesFuzz\n\t\t\t\"FailoverMarkerAttributes\",      // Tested in TestFailoverMarkerAttributesFuzz\n\t\t),\n\t)\n}\n"
  },
  {
    "path": "common/types/mapper/testutils/common_fuzzers.go",
    "content": "package testutils\n\nimport (\n\tfuzz \"github.com/google/gofuzz\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// WithCommonEnumFuzzers adds fuzzers for common Cadence enum types\n// to ensure only valid enum values are generated\nfunc WithCommonEnumFuzzers() FuzzOption {\n\treturn WithCustomFuncs(\n\t\tfunc(e *types.WorkflowExecutionCloseStatus, c fuzz.Continue) {\n\t\t\t*e = types.WorkflowExecutionCloseStatus(c.Intn(6)) // 0-5\n\t\t},\n\t\tfunc(e *types.TaskListKind, c fuzz.Continue) {\n\t\t\t*e = types.TaskListKind(c.Intn(3)) // 0-2: Normal, Sticky, Ephemeral\n\t\t},\n\t\tfunc(e *types.TaskListType, c fuzz.Continue) {\n\t\t\t*e = types.TaskListType(c.Intn(2)) // 0-1: Decision, Activity\n\t\t},\n\t\tfunc(e *types.TimeoutType, c fuzz.Continue) {\n\t\t\t*e = types.TimeoutType(c.Intn(4)) // 0-3: StartToClose, ScheduleToStart, ScheduleToClose, Heartbeat\n\t\t},\n\t\tfunc(e *types.ParentClosePolicy, c fuzz.Continue) {\n\t\t\t*e = types.ParentClosePolicy(c.Intn(3)) // 0-2\n\t\t},\n\t\tfunc(e *types.PendingActivityState, c fuzz.Continue) {\n\t\t\t*e = types.PendingActivityState(c.Intn(3)) // 0-2\n\t\t},\n\t\tfunc(e *types.PendingDecisionState, c fuzz.Continue) {\n\t\t\t*e = types.PendingDecisionState(c.Intn(2)) // 0-1\n\t\t},\n\t\tfunc(e *types.QueryTaskCompletedType, c fuzz.Continue) {\n\t\t\t*e = types.QueryTaskCompletedType(c.Intn(3)) // 0-2\n\t\t},\n\t\tfunc(e *types.QueryResultType, c fuzz.Continue) {\n\t\t\t*e = types.QueryResultType(c.Intn(2)) // 0-1: Answered, Failed\n\t\t},\n\t\tfunc(e *types.IndexedValueType, c fuzz.Continue) {\n\t\t\t*e = types.IndexedValueType(c.Intn(6)) // 0-5: String, Keyword, Int, Double, Bool, Datetime\n\t\t},\n\t\tfunc(e *types.CronOverlapPolicy, c fuzz.Continue) {\n\t\t\t*e = types.CronOverlapPolicy(c.Intn(2)) // 0-1: Skipped, BufferOne\n\t\t},\n\t\tfunc(e *types.WorkflowExecutionStatus, c fuzz.Continue) {\n\t\t\t*e = types.WorkflowExecutionStatus(c.Intn(8)) // 0-7: Pending, Started, Completed, Failed, Canceled, Terminated, ContinuedAsNew, TimedOut\n\t\t},\n\t\tfunc(e *types.ActiveClusterSelectionStrategy, c fuzz.Continue) {\n\t\t\t*e = types.ActiveClusterSelectionStrategy(c.Intn(2)) // 0-1: RegionSticky, ExternalEntity\n\t\t},\n\t)\n}\n\nfunc EncodingTypeFuzzer(e *types.EncodingType, c fuzz.Continue) {\n\t*e = types.EncodingType(c.Intn(2)) // 0-1: ThriftRW, JSON\n}\n\nfunc CrossClusterTaskFailedCauseFuzzer(e *types.CrossClusterTaskFailedCause, c fuzz.Continue) {\n\t*e = types.CrossClusterTaskFailedCause(c.Intn(6)) // 0-5: DomainNotActive, DomainNotExists, WorkflowAlreadyRunning, WorkflowNotExists, WorkflowAlreadyCompleted, Uncategorized\n}\n\nfunc CrossClusterTaskTypeFuzzer(e *types.CrossClusterTaskType, c fuzz.Continue) {\n\t*e = types.CrossClusterTaskType(c.Intn(5)) // 0-4: StartChildExecution, CancelExecution, SignalExecution, RecordChildWorkflowExecutionComplete, ApplyParentClosePolicy\n}\n\nfunc DLQTypeFuzzer(e *types.DLQType, c fuzz.Continue) {\n\t*e = types.DLQType(c.Intn(2)) // 0-1: Replication, Domain\n}\n\nfunc DomainOperationFuzzer(e *types.DomainOperation, c fuzz.Continue) {\n\t*e = types.DomainOperation(c.Intn(3)) // 0-2: Create, Update, Delete\n}\n\nfunc ReplicationTaskTypeFuzzer(e *types.ReplicationTaskType, c fuzz.Continue) {\n\t*e = types.ReplicationTaskType(c.Intn(7)) // 0-6: Domain, History, SyncShardStatus, SyncActivity, HistoryMetadata, HistoryV2, FailoverMarker\n}\n\nfunc WorkflowStateFuzzer(e *int32, c fuzz.Continue) {\n\t// WorkflowState values: 0-2 (Created, Running, Completed)\n\t// Note: This operates on *int32 because the types package uses *int32 for WorkflowState\n\tif e != nil {\n\t\t*e = int32(c.Intn(3))\n\t}\n}\n\nfunc GetTaskFailedCauseFuzzer(e *types.GetTaskFailedCause, c fuzz.Continue) {\n\t*e = types.GetTaskFailedCause(c.Intn(4)) // 0-3: ServiceBusy, Timeout, ShardOwnershipLost, Uncategorized\n}\n\nfunc IsolationGroupStateFuzzer(e *types.IsolationGroupState, c fuzz.Continue) {\n\t*e = types.IsolationGroupState(c.Intn(3)) // 0-2: Invalid, Healthy, Drained\n}\n\nfunc TaskTypeFuzzer(e *int32, c fuzz.Continue) {\n\t// TaskType internal constant values (from common/constants/constants.go):\n\t// 2: TaskTypeTransfer\n\t// 3: TaskTypeTimer\n\t// 4: TaskTypeReplication\n\t// 6: TaskTypeCrossCluster (deprecated but still valid)\n\t// Note: Value 0 maps to nil/INVALID, so we always generate a valid value\n\tif e != nil {\n\t\t// Generate one of the valid constant values: 2, 3, 4, or 6\n\t\tvalidValues := []int32{2, 3, 4, 6}\n\t\t*e = validValues[c.Intn(len(validValues))]\n\t}\n}\n\nfunc DomainStatusFuzzer(e *types.DomainStatus, c fuzz.Continue) {\n\t*e = types.DomainStatus(c.Intn(3)) // 0-2: Registered, Deprecated, Deleted\n}\n"
  },
  {
    "path": "common/types/mapper/testutils/fuzz_mapper.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testutils\n\nimport (\n\t\"reflect\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nconst (\n\t// DefaultNilChance is the default probability of generating nil values\n\tDefaultNilChance = 0.3\n\t// DefaultIterations is the default number of fuzzing iterations\n\tDefaultIterations = 100\n\t// MaxSafeTimestampSeconds is the maximum safe Unix timestamp (year 2100)\n\tMaxSafeTimestampSeconds = 4102444800\n\t// NanosecondsPerSecond is the number of nanoseconds in a second\n\tNanosecondsPerSecond = 1000000000\n\t// MaxDurationSeconds is the maximum duration in seconds (24 hours)\n\tMaxDurationSeconds = 86400\n)\n\n// FuzzOptions configures the behavior of mapper fuzz tests\ntype FuzzOptions struct {\n\t// CustomFuncs are custom fuzzer functions to apply for specific types\n\tCustomFuncs []interface{}\n\t// ExcludedFields are field names to exclude from comparison (set to zero value)\n\tExcludedFields []string\n\t// NilChance is the probability of setting pointer/slice fields to nil (default DefaultNilChance)\n\tNilChance float64\n\t// Iterations is the number of fuzzing iterations to run (default DefaultIterations)\n\tIterations int\n}\n\n// FuzzOption is a functional option for configuring FuzzOptions\ntype FuzzOption func(*FuzzOptions)\n\n// WithCustomFuncs adds custom fuzzer functions for specific types\nfunc WithCustomFuncs(funcs ...interface{}) FuzzOption {\n\treturn func(opts *FuzzOptions) {\n\t\topts.CustomFuncs = append(opts.CustomFuncs, funcs...)\n\t}\n}\n\n// WithExcludedFields specifies field names to exclude from comparison\nfunc WithExcludedFields(fields ...string) FuzzOption {\n\treturn func(opts *FuzzOptions) {\n\t\topts.ExcludedFields = append(opts.ExcludedFields, fields...)\n\t}\n}\n\n// WithNilChance sets the probability of generating nil values\nfunc WithNilChance(chance float64) FuzzOption {\n\treturn func(opts *FuzzOptions) {\n\t\topts.NilChance = chance\n\t}\n}\n\n// WithIterations sets the number of fuzzing iterations\nfunc WithIterations(iterations int) FuzzOption {\n\treturn func(opts *FuzzOptions) {\n\t\topts.Iterations = iterations\n\t}\n}\n\n// WithTimeFuzzers adds custom fuzzers for time.Time and time.Duration\n// to ensure safe timestamp bounds and proper UTC normalization\nfunc WithTimeFuzzers() FuzzOption {\n\treturn WithCustomFuncs(\n\t\t// time.Time with safe bounds (Unix epoch to year 2100) and UTC normalization\n\t\tfunc(t *time.Time, c fuzz.Continue) {\n\t\t\tif c.Intn(10) < 3 {\n\t\t\t\t*t = time.Time{} // Zero time\n\t\t\t\treturn\n\t\t\t}\n\t\t\t*t = time.Unix(c.Int63n(MaxSafeTimestampSeconds), c.Int63n(NanosecondsPerSecond)).UTC()\n\t\t},\n\t\t// time.Duration with safe bounds (up to 24 hours)\n\t\tfunc(d *time.Duration, c fuzz.Continue) {\n\t\t\tif c.Intn(10) < 3 {\n\t\t\t\t*d = 0 // Zero duration\n\t\t\t\treturn\n\t\t\t}\n\t\t\t*d = time.Duration(c.Int63n(MaxDurationSeconds * NanosecondsPerSecond))\n\t\t},\n\t)\n}\n\n// WithDefaultFuzzers applies the standard set of fuzzers for time and common enums.\n// This is applied automatically by RunMapperFuzzTest unless overridden.\nfunc WithDefaultFuzzers() FuzzOption {\n\treturn func(opts *FuzzOptions) {\n\t\tWithTimeFuzzers()(opts)\n\t\tWithCommonEnumFuzzers()(opts)\n\t}\n}\n\n// RunMapperFuzzTest runs a fuzz test for a mapper pair (From/To functions).\n// It generates random values, converts them through the mapper pair, and verifies\n// the round-trip preserves the original value.\n//\n// Example usage:\n//\n//\tfunc TestActivityTypeFuzz(t *testing.T) {\n//\t    RunMapperFuzzTest(t, FromActivityType, ToActivityType)\n//\t}\nfunc RunMapperFuzzTest[TInternal any, TExternal any](\n\tt *testing.T,\n\tfromFunc func(TInternal) TExternal,\n\ttoFunc func(TExternal) TInternal,\n\toptions ...FuzzOption,\n) {\n\tt.Helper()\n\n\t// Apply default options\n\topts := FuzzOptions{\n\t\tNilChance:  DefaultNilChance,\n\t\tIterations: DefaultIterations,\n\t}\n\n\t// Apply default fuzzers first (time and common enums)\n\tWithDefaultFuzzers()(&opts)\n\n\t// Then apply user-provided options (can override defaults)\n\tfor _, opt := range options {\n\t\topt(&opts)\n\t}\n\n\t// Create fuzzer with seed for reproducibility\n\tseed := time.Now().UnixNano()\n\tfuzzer := fuzz.NewWithSeed(seed).\n\t\tFuncs(opts.CustomFuncs...).\n\t\tNilChance(opts.NilChance)\n\n\t// Log seed on failure for debugging\n\tdefer func() {\n\t\tif t.Failed() {\n\t\t\tt.Logf(\"fuzz seed: %v\", seed)\n\t\t}\n\t}()\n\n\t// Run fuzzing iterations\n\tfor i := 0; i < opts.Iterations; i++ {\n\t\tvar orig TInternal\n\t\tfuzzer.Fuzz(&orig)\n\n\t\t// Clear excluded fields before conversion\n\t\tclearExcludedFields(&orig, opts.ExcludedFields)\n\n\t\t// Test round trip: Internal -> External -> Internal\n\t\texternal := fromFunc(orig)\n\t\tresult := toFunc(external)\n\n\t\t// Clear excluded fields in result for comparison\n\t\tclearExcludedFields(&result, opts.ExcludedFields)\n\n\t\tassert.Equal(t, orig, result, \"Round trip failed at iteration %d\", i)\n\t}\n}\n\n// clearFieldsIf recursively traverses an object and clears fields that match the predicate function.\n// This is used to exclude certain fields from round-trip comparison.\nfunc clearFieldsIf(obj interface{}, shouldClear func(fieldName string) bool) {\n\tif obj == nil {\n\t\treturn\n\t}\n\n\tv := reflect.ValueOf(obj)\n\t// Dereference through multiple pointer levels to handle cases like **types.ScheduleSpec\n\t// This happens when TInternal is a pointer type and we pass &orig to clearExcludedFields\n\tfor v.Kind() == reflect.Ptr {\n\t\tif v.IsNil() {\n\t\t\treturn\n\t\t}\n\t\tv = v.Elem()\n\t}\n\n\t// Handle slices at top level (e.g., when TInternal is []*SomeType)\n\tif v.Kind() == reflect.Slice {\n\t\tfor j := 0; j < v.Len(); j++ {\n\t\t\telem := v.Index(j)\n\t\t\tif elem.CanInterface() {\n\t\t\t\tif elem.Kind() == reflect.Struct && elem.CanAddr() {\n\t\t\t\t\tclearFieldsIf(elem.Addr().Interface(), shouldClear)\n\t\t\t\t} else {\n\t\t\t\t\tclearFieldsIf(elem.Interface(), shouldClear)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\treturn\n\t}\n\n\tif v.Kind() != reflect.Struct {\n\t\treturn\n\t}\n\n\tfor i := 0; i < v.NumField(); i++ {\n\t\tfield := v.Field(i)\n\t\tfieldName := v.Type().Field(i).Name\n\n\t\tif shouldClear(fieldName) && field.CanSet() {\n\t\t\tfield.Set(reflect.Zero(field.Type()))\n\t\t}\n\n\t\t// Recursively clear fields in nested structs and slices\n\t\tif field.CanInterface() {\n\t\t\tswitch field.Kind() {\n\t\t\tcase reflect.Ptr:\n\t\t\t\tif !field.IsNil() {\n\t\t\t\t\tclearFieldsIf(field.Interface(), shouldClear)\n\t\t\t\t}\n\t\t\tcase reflect.Struct:\n\t\t\t\tif field.CanAddr() {\n\t\t\t\t\tclearFieldsIf(field.Addr().Interface(), shouldClear)\n\t\t\t\t}\n\t\t\tcase reflect.Slice:\n\t\t\t\tfor j := 0; j < field.Len(); j++ {\n\t\t\t\t\telem := field.Index(j)\n\t\t\t\t\tif elem.CanInterface() {\n\t\t\t\t\t\t// For struct elements, pass the address to ensure modifications work\n\t\t\t\t\t\t// For pointer elements, pass the interface directly\n\t\t\t\t\t\tif elem.Kind() == reflect.Struct && elem.CanAddr() {\n\t\t\t\t\t\t\tclearFieldsIf(elem.Addr().Interface(), shouldClear)\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tclearFieldsIf(elem.Interface(), shouldClear)\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\tcase reflect.Map:\n\t\t\t\t// Map values cannot be modified through reflection in Go.\n\t\t\t\t// Calling clearFieldsIf on map values would operate on copies that are discarded.\n\t\t\t\t// To properly handle maps with struct values, the map would need to be rebuilt\n\t\t\t\t// with cleared values, which is beyond the scope of this utility.\n\t\t\t\t// Since Cadence types primarily use slices and pointers rather than maps of structs,\n\t\t\t\t// this limitation has minimal practical impact.\n\t\t\t}\n\t\t}\n\t}\n}\n\n// clearExcludedFields clears both protobuf internal fields and user-specified excluded fields.\n// Protobuf internal fields (XXX_*, sizeCache, unknownFields) are always cleared to avoid comparison issues.\n// state is not excluded as it is generic enough it may be used by a mapper in future\nfunc clearExcludedFields(obj interface{}, excludedFields []string) {\n\t// Create a map for O(1) lookup of excluded fields\n\texcludedMap := make(map[string]bool)\n\tfor _, field := range excludedFields {\n\t\texcludedMap[field] = true\n\t}\n\n\tclearFieldsIf(obj, func(fieldName string) bool {\n\t\t// Clear if it's a protobuf internal field OR if it's in the excluded list\n\t\treturn strings.HasPrefix(fieldName, \"XXX_\") ||\n\t\t\tfieldName == \"sizeCache\" ||\n\t\t\tfieldName == \"unknownFields\" ||\n\t\t\texcludedMap[fieldName]\n\t})\n}\n"
  },
  {
    "path": "common/types/mapper/testutils/fuzz_mapper_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testutils\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\n// Test types for clearFieldsIf\n// Note: Real protobuf fields like sizeCache and unknownFields are typically unexported,\n// but we use exported versions here to test the clearing logic. The function will check\n// field names and clear them if they match the pattern, regardless of export status.\ntype testStructWithExcludedFields struct {\n\tKeepMe      string\n\tXXX_ClearMe string // revive:disable-line:var-naming Protobuf internal field pattern (XXX_ prefix)\n\tAlsoClearMe string // User-specified excluded field\n}\n\ntype testStructWithSlices struct {\n\tValueStructs []testStructWithExcludedFields  // Slice of value-type structs\n\tPointerSlice []*testStructWithExcludedFields // Slice of pointers\n\tXXX_Internal string                          // revive:disable-line:var-naming Protobuf internal field pattern (XXX_ prefix)\n}\n\ntype testStructWithNestedStruct struct {\n\tNested       testStructWithExcludedFields // Nested struct\n\tXXX_Internal string                       // revive:disable-line:var-naming Protobuf internal field pattern (XXX_ prefix)\n}\n\nfunc TestClearFieldsIf_ValueStructSlice(t *testing.T) {\n\t// This test verifies the fix for slice struct elements (issue #1 from code review)\n\t// Previously, clearFieldsIf would operate on copies for slice struct elements,\n\t// causing fields to not actually be cleared.\n\tobj := &testStructWithSlices{\n\t\tValueStructs: []testStructWithExcludedFields{\n\t\t\t{KeepMe: \"keep1\", XXX_ClearMe: \"clear1\", AlsoClearMe: \"clear1\"},\n\t\t\t{KeepMe: \"keep2\", XXX_ClearMe: \"clear2\", AlsoClearMe: \"clear2\"},\n\t\t},\n\t\tPointerSlice: []*testStructWithExcludedFields{\n\t\t\t{KeepMe: \"keep3\", XXX_ClearMe: \"clear3\", AlsoClearMe: \"clear3\"},\n\t\t},\n\t\tXXX_Internal: \"internal\",\n\t}\n\n\tclearExcludedFields(obj, []string{\"AlsoClearMe\"})\n\n\t// Top-level fields should be cleared\n\tassert.Equal(t, \"\", obj.XXX_Internal, \"XXX_ field should be cleared\")\n\n\t// Value struct slice elements should have their excluded fields cleared\n\t// This is the key test - before the fix, these would NOT be cleared\n\tassert.Equal(t, \"keep1\", obj.ValueStructs[0].KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", obj.ValueStructs[0].XXX_ClearMe, \"XXX_ClearMe should be cleared in value struct slice\")\n\tassert.Equal(t, \"\", obj.ValueStructs[0].AlsoClearMe, \"AlsoClearMe should be cleared in value struct slice\")\n\n\tassert.Equal(t, \"keep2\", obj.ValueStructs[1].KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", obj.ValueStructs[1].XXX_ClearMe, \"XXX_ClearMe should be cleared in value struct slice\")\n\tassert.Equal(t, \"\", obj.ValueStructs[1].AlsoClearMe, \"AlsoClearMe should be cleared in value struct slice\")\n\n\t// Pointer slice elements should also have their excluded fields cleared\n\tassert.Equal(t, \"keep3\", obj.PointerSlice[0].KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", obj.PointerSlice[0].XXX_ClearMe, \"XXX_ClearMe should be cleared in pointer slice\")\n\tassert.Equal(t, \"\", obj.PointerSlice[0].AlsoClearMe, \"AlsoClearMe should be cleared in pointer slice\")\n}\n\nfunc TestClearFieldsIf_NestedStruct(t *testing.T) {\n\tobj := &testStructWithNestedStruct{\n\t\tNested: testStructWithExcludedFields{\n\t\t\tKeepMe:      \"keep\",\n\t\t\tXXX_ClearMe: \"clear\",\n\t\t\tAlsoClearMe: \"clear\",\n\t\t},\n\t\tXXX_Internal: \"internal\",\n\t}\n\n\tclearExcludedFields(obj, []string{\"AlsoClearMe\"})\n\n\t// Top-level excluded field should be cleared\n\tassert.Equal(t, \"\", obj.XXX_Internal, \"XXX_Internal should be cleared\")\n\n\t// Nested struct fields should be cleared\n\tassert.Equal(t, \"keep\", obj.Nested.KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", obj.Nested.XXX_ClearMe, \"XXX_ClearMe should be cleared in nested struct\")\n\tassert.Equal(t, \"\", obj.Nested.AlsoClearMe, \"AlsoClearMe should be cleared in nested struct\")\n}\n\nfunc TestClearFieldsIf_DoesNotClearLegitimateStateField(t *testing.T) {\n\t// This test verifies issue #2 from code review: we don't clear legitimate fields named \"state\"\n\t// The previous implementation cleared ANY field named \"state\", which was too broad.\n\ttype legitimateStruct struct {\n\t\tState        string // Legitimate business logic field\n\t\tXXX_Internal string // revive:disable-line:var-naming Protobuf internal field pattern (XXX_ prefix)\n\t}\n\n\tobj := &legitimateStruct{\n\t\tState:        \"active\",\n\t\tXXX_Internal: \"internal\",\n\t}\n\n\tclearExcludedFields(obj, nil)\n\n\t// Legitimate \"State\" field should NOT be cleared (issue #2 fix)\n\tassert.Equal(t, \"active\", obj.State, \"Legitimate State field should not be cleared\")\n\t// But XXX_ field should still be cleared\n\tassert.Equal(t, \"\", obj.XXX_Internal, \"XXX_ field should be cleared\")\n}\n\nfunc TestClearFieldsIf_ProtobufFields(t *testing.T) {\n\t// Test that common protobuf internal field patterns are cleared\n\t// Note: Real protobuf internal fields are often unexported and thus cannot be\n\t// cleared via reflection, but we test the logic with exported fields.\n\ttype protobufLikeStruct struct {\n\t\tBusinessField    string\n\t\tXXX_NoUnkeyedLit struct{} // revive:disable-line:var-naming XXX_ prefixed fields should be cleared\n\t\tXXX_unrecognized []byte   // revive:disable-line:var-naming XXX_ prefixed fields should be cleared\n\t}\n\n\tobj := &protobufLikeStruct{\n\t\tBusinessField:    \"keep\",\n\t\tXXX_NoUnkeyedLit: struct{}{},\n\t\tXXX_unrecognized: []byte(\"data\"),\n\t}\n\n\tclearExcludedFields(obj, nil)\n\n\t// Business field should be kept\n\tassert.Equal(t, \"keep\", obj.BusinessField, \"Business field should not be cleared\")\n\n\t// All XXX_ prefixed fields should be cleared\n\tassert.Equal(t, struct{}{}, obj.XXX_NoUnkeyedLit, \"XXX_NoUnkeyedLit should be cleared\")\n\tassert.Nil(t, obj.XXX_unrecognized, \"XXX_unrecognized should be cleared\")\n}\n\nfunc TestClearFieldsIf_DoublePointer(t *testing.T) {\n\t// This test verifies the fix for the double-pointer bug (code review issue #3)\n\t// When TInternal is a pointer type (like *types.ScheduleSpec), RunMapperFuzzTest\n\t// calls clearExcludedFields(&orig, ...), passing a **types.ScheduleSpec.\n\t// The function must dereference through multiple pointer levels.\n\ttype innerStruct struct {\n\t\tKeepMe      string\n\t\tXXX_ClearMe string // revive:disable-line:var-naming Protobuf field\n\t\tAlsoClearMe string\n\t}\n\n\t// Simulate the actual usage pattern in RunMapperFuzzTest\n\torig := &innerStruct{\n\t\tKeepMe:      \"keep\",\n\t\tXXX_ClearMe: \"clear\",\n\t\tAlsoClearMe: \"clear\",\n\t}\n\n\t// This is what RunMapperFuzzTest does: passes &orig where orig is already a pointer\n\t// So we're passing **innerStruct to clearExcludedFields\n\tclearExcludedFields(&orig, []string{\"AlsoClearMe\"})\n\n\t// Fields should be cleared even through the double pointer\n\tassert.Equal(t, \"keep\", orig.KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", orig.XXX_ClearMe, \"XXX_ClearMe should be cleared through double pointer\")\n\tassert.Equal(t, \"\", orig.AlsoClearMe, \"AlsoClearMe should be cleared through double pointer\")\n}\n\nfunc TestClearFieldsIf_TriplePointer(t *testing.T) {\n\t// Edge case: ensure we handle arbitrary pointer depth\n\ttype innerStruct struct {\n\t\tXXX_ClearMe string // revive:disable-line:var-naming Protobuf field\n\t}\n\n\tlevel1 := &innerStruct{XXX_ClearMe: \"clear\"}\n\tlevel2 := &level1\n\tlevel3 := &level2\n\n\tclearExcludedFields(level3, nil)\n\n\tassert.Equal(t, \"\", level1.XXX_ClearMe, \"Should clear through triple pointer\")\n}\n\nfunc TestClearFieldsIf_WithExcludedFields(t *testing.T) {\n\t// This test verifies that WithExcludedFields works correctly with pointer types\n\t// Previously, this would silently do nothing due to the double-pointer bug\n\ttype testStruct struct {\n\t\tKeepMe        string\n\t\tExcludeMe     string\n\t\tAlsoExcludeMe string\n\t}\n\n\torig := &testStruct{\n\t\tKeepMe:        \"keep\",\n\t\tExcludeMe:     \"exclude1\",\n\t\tAlsoExcludeMe: \"exclude2\",\n\t}\n\n\t// Simulate RunMapperFuzzTest usage pattern\n\tclearExcludedFields(&orig, []string{\"ExcludeMe\", \"AlsoExcludeMe\"})\n\n\tassert.Equal(t, \"keep\", orig.KeepMe, \"KeepMe should not be cleared\")\n\tassert.Equal(t, \"\", orig.ExcludeMe, \"ExcludeMe should be cleared via WithExcludedFields\")\n\tassert.Equal(t, \"\", orig.AlsoExcludeMe, \"AlsoExcludeMe should be cleared via WithExcludedFields\")\n}\n"
  },
  {
    "path": "common/types/mapper/testutils/fuzz_test_utils.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testutils\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\tfuzz \"github.com/google/gofuzz\"\n)\n\nfunc EnsureFuzzCoverage(t *testing.T, expected []string, cb func(t *testing.T, f *fuzz.Fuzzer) string) {\n\tt.Helper()\n\n\tvar details []string\n\tresults := make(map[string]bool, len(expected))\n\tfor _, e := range expected {\n\t\tresults[e] = false\n\t}\n\tseed := time.Now().UnixNano()\n\tf := fuzz.NewWithSeed(seed) // helps with troubleshooting\n\n\tdefer func() {\n\t\tif t.Failed() { // else a bit noisy\n\t\t\tt.Logf(\"expected to see:  %#v\", expected)\n\t\t\tt.Logf(\"observed results: %#v\", results)\n\t\t\tt.Logf(\"detailed results: %#v\", details)\n\t\t\tt.Logf(\"fuzz seed: %v\", seed)\n\t\t}\n\t}()\n\n\tfor tries := 0; tries < 100; tries++ { // retry a few times if needed\n\t\tfor i := 0; i < 100; i++ { // always fuzz a moderate amount, don't stop immediately\n\t\t\tres := cb(t, f)\n\t\t\tdetails = append(details, res)\n\t\t\tif res == \"\" {\n\t\t\t\tt.Errorf(\"invalid empty response from fuzzing callback on iteration %v\", (tries*100)+i)\n\t\t\t}\n\t\t\tif _, ok := results[res]; !ok {\n\t\t\t\tt.Errorf(\"unrecognized response from fuzzing callback on iteration %v: %v\", (tries*100)+i, res)\n\t\t\t}\n\t\t\tif t.Failed() {\n\t\t\t\treturn // already failed either internally or in the callback, either way stop trying\n\t\t\t}\n\t\t\tresults[res] = true\n\t\t}\n\t\tstop := true\n\t\tfor _, v := range results {\n\t\t\tstop = stop && v\n\t\t}\n\t\tif stop {\n\t\t\treturn // covered all expected values, stop retrying\n\t\t}\n\t}\n\tmissing := make([]string, 0, len(results))\n\tfor _, v := range expected {\n\t\tif !results[v] {\n\t\t\tmissing = append(missing, v)\n\t\t}\n\t}\n\tt.Errorf(\"fuzzy coverage func did not check enough cases after 10k attempts, missing: %#v\", missing)\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/admin.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"sort\"\n\n\t\"github.com/uber/cadence/.gen/go/admin\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tFromAdminRefreshWorkflowTasksRequest = FromRefreshWorkflowTasksRequest\n\tToAdminRefreshWorkflowTasksRequest   = ToRefreshWorkflowTasksRequest\n)\n\n// FromAdminAddSearchAttributeRequest converts internal AddSearchAttributeRequest type to thrift\nfunc FromAdminAddSearchAttributeRequest(t *types.AddSearchAttributeRequest) *admin.AddSearchAttributeRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.AddSearchAttributeRequest{\n\t\tSearchAttribute: FromIndexedValueTypeMap(t.SearchAttribute),\n\t\tSecurityToken:   &t.SecurityToken,\n\t}\n}\n\n// ToAdminAddSearchAttributeRequest converts thrift AddSearchAttributeRequest type to internal\nfunc ToAdminAddSearchAttributeRequest(t *admin.AddSearchAttributeRequest) *types.AddSearchAttributeRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddSearchAttributeRequest{\n\t\tSearchAttribute: ToIndexedValueTypeMap(t.SearchAttribute),\n\t\tSecurityToken:   t.GetSecurityToken(),\n\t}\n}\n\n// FromAdminDescribeClusterResponse converts internal DescribeClusterResponse type to thrift\nfunc FromAdminDescribeClusterResponse(t *types.DescribeClusterResponse) *admin.DescribeClusterResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.DescribeClusterResponse{\n\t\tSupportedClientVersions: FromSupportedClientVersions(t.SupportedClientVersions),\n\t\tMembershipInfo:          FromMembershipInfo(t.MembershipInfo),\n\t\tPersistenceInfo:         FromPersistenceInfoMap(t.PersistenceInfo),\n\t}\n}\n\n// ToAdminDescribeClusterResponse converts thrift DescribeClusterResponse type to internal\nfunc ToAdminDescribeClusterResponse(t *admin.DescribeClusterResponse) *types.DescribeClusterResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeClusterResponse{\n\t\tSupportedClientVersions: ToSupportedClientVersions(t.SupportedClientVersions),\n\t\tMembershipInfo:          ToMembershipInfo(t.MembershipInfo),\n\t\tPersistenceInfo:         ToPersistenceInfoMap(t.PersistenceInfo),\n\t}\n}\n\n// FromAdminDescribeWorkflowExecutionRequest converts internal DescribeWorkflowExecutionRequest type to thrift\nfunc FromAdminDescribeWorkflowExecutionRequest(t *types.AdminDescribeWorkflowExecutionRequest) *admin.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.DescribeWorkflowExecutionRequest{\n\t\tDomain:    &t.Domain,\n\t\tExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToAdminDescribeWorkflowExecutionRequest converts thrift DescribeWorkflowExecutionRequest type to internal\nfunc ToAdminDescribeWorkflowExecutionRequest(t *admin.DescribeWorkflowExecutionRequest) *types.AdminDescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDescribeWorkflowExecutionRequest{\n\t\tDomain:    t.GetDomain(),\n\t\tExecution: ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromAdminDescribeWorkflowExecutionResponse converts internal DescribeWorkflowExecutionResponse type to thrift\nfunc FromAdminDescribeWorkflowExecutionResponse(t *types.AdminDescribeWorkflowExecutionResponse) *admin.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.DescribeWorkflowExecutionResponse{\n\t\tShardId:                &t.ShardID,\n\t\tHistoryAddr:            &t.HistoryAddr,\n\t\tMutableStateInCache:    &t.MutableStateInCache,\n\t\tMutableStateInDatabase: &t.MutableStateInDatabase,\n\t}\n}\n\n// ToAdminDescribeWorkflowExecutionResponse converts thrift DescribeWorkflowExecutionResponse type to internal\nfunc ToAdminDescribeWorkflowExecutionResponse(t *admin.DescribeWorkflowExecutionResponse) *types.AdminDescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDescribeWorkflowExecutionResponse{\n\t\tShardID:                t.GetShardId(),\n\t\tHistoryAddr:            t.GetHistoryAddr(),\n\t\tMutableStateInCache:    t.GetMutableStateInCache(),\n\t\tMutableStateInDatabase: t.GetMutableStateInDatabase(),\n\t}\n}\n\n// FromAdminGetWorkflowExecutionRawHistoryV2Request converts internal GetWorkflowExecutionRawHistoryV2Request type to thrift\nfunc FromAdminGetWorkflowExecutionRawHistoryV2Request(t *types.GetWorkflowExecutionRawHistoryV2Request) *admin.GetWorkflowExecutionRawHistoryV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain:            &t.Domain,\n\t\tExecution:         FromWorkflowExecution(t.Execution),\n\t\tStartEventId:      t.StartEventID,\n\t\tStartEventVersion: t.StartEventVersion,\n\t\tEndEventId:        t.EndEventID,\n\t\tEndEventVersion:   t.EndEventVersion,\n\t\tMaximumPageSize:   &t.MaximumPageSize,\n\t\tNextPageToken:     t.NextPageToken,\n\t}\n}\n\n// ToAdminGetWorkflowExecutionRawHistoryV2Request converts thrift GetWorkflowExecutionRawHistoryV2Request type to internal\nfunc ToAdminGetWorkflowExecutionRawHistoryV2Request(t *admin.GetWorkflowExecutionRawHistoryV2Request) *types.GetWorkflowExecutionRawHistoryV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain:            t.GetDomain(),\n\t\tExecution:         ToWorkflowExecution(t.Execution),\n\t\tStartEventID:      t.StartEventId,\n\t\tStartEventVersion: t.StartEventVersion,\n\t\tEndEventID:        t.EndEventId,\n\t\tEndEventVersion:   t.EndEventVersion,\n\t\tMaximumPageSize:   t.GetMaximumPageSize(),\n\t\tNextPageToken:     t.NextPageToken,\n\t}\n}\n\n// FromAdminGetWorkflowExecutionRawHistoryV2Response converts internal GetWorkflowExecutionRawHistoryV2Response type to thrift\nfunc FromAdminGetWorkflowExecutionRawHistoryV2Response(t *types.GetWorkflowExecutionRawHistoryV2Response) *admin.GetWorkflowExecutionRawHistoryV2Response {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetWorkflowExecutionRawHistoryV2Response{\n\t\tNextPageToken:  t.NextPageToken,\n\t\tHistoryBatches: FromDataBlobArray(t.HistoryBatches),\n\t\tVersionHistory: FromVersionHistory(t.VersionHistory),\n\t}\n}\n\n// ToAdminGetWorkflowExecutionRawHistoryV2Response converts thrift GetWorkflowExecutionRawHistoryV2Response type to internal\nfunc ToAdminGetWorkflowExecutionRawHistoryV2Response(t *admin.GetWorkflowExecutionRawHistoryV2Response) *types.GetWorkflowExecutionRawHistoryV2Response {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tNextPageToken:  t.NextPageToken,\n\t\tHistoryBatches: ToDataBlobArray(t.HistoryBatches),\n\t\tVersionHistory: ToVersionHistory(t.VersionHistory),\n\t}\n}\n\n// FromHostInfo converts internal HostInfo type to thrift\nfunc FromHostInfo(t *types.HostInfo) *admin.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.HostInfo{\n\t\tIdentity: &t.Identity,\n\t}\n}\n\n// ToHostInfo converts thrift HostInfo type to internal\nfunc ToHostInfo(t *admin.HostInfo) *types.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HostInfo{\n\t\tIdentity: t.GetIdentity(),\n\t}\n}\n\n// FromMembershipInfo converts internal MembershipInfo type to thrift\nfunc FromMembershipInfo(t *types.MembershipInfo) *admin.MembershipInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.MembershipInfo{\n\t\tCurrentHost:      FromHostInfo(t.CurrentHost),\n\t\tReachableMembers: t.ReachableMembers,\n\t\tRings:            FromRingInfoArray(t.Rings),\n\t}\n}\n\n// ToMembershipInfo converts thrift MembershipInfo type to internal\nfunc ToMembershipInfo(t *admin.MembershipInfo) *types.MembershipInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MembershipInfo{\n\t\tCurrentHost:      ToHostInfo(t.CurrentHost),\n\t\tReachableMembers: t.ReachableMembers,\n\t\tRings:            ToRingInfoArray(t.Rings),\n\t}\n}\n\n// FromPersistenceInfoMap converts internal map[string]*types.PersistenceInfo type to thrift\nfunc FromPersistenceInfoMap(t map[string]*types.PersistenceInfo) map[string]*admin.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*admin.PersistenceInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromPersistenceInfo(t[key])\n\t}\n\treturn v\n}\n\n// FromPersistenceInfo converts internal PersistenceInfo type to thrift\nfunc FromPersistenceInfo(t *types.PersistenceInfo) *admin.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.PersistenceInfo{\n\t\tBackend:  &t.Backend,\n\t\tSettings: FromPersistenceSettings(t.Settings),\n\t\tFeatures: FromPersistenceFeatures(t.Features),\n\t}\n}\n\n// FromPersistenceSettings converts internal []*types.PersistenceSetting type to thrift\nfunc FromPersistenceSettings(t []*types.PersistenceSetting) []*admin.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*admin.PersistenceSetting, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPersistenceSetting(t[i])\n\t}\n\treturn v\n}\n\n// FromPersistenceSetting converts internal PersistenceSetting type to thrift\nfunc FromPersistenceSetting(t *types.PersistenceSetting) *admin.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.PersistenceSetting{\n\t\tKey:   &t.Key,\n\t\tValue: &t.Value,\n\t}\n}\n\n// FromPersistenceFeatures converts internal []*types.PersistenceFeature type to thrift\nfunc FromPersistenceFeatures(t []*types.PersistenceFeature) []*admin.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*admin.PersistenceFeature, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPersistenceFeature(t[i])\n\t}\n\treturn v\n}\n\n// FromPersistenceFeature converts internal PersistenceFeature type to thrift\nfunc FromPersistenceFeature(t *types.PersistenceFeature) *admin.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.PersistenceFeature{\n\t\tKey:     &t.Key,\n\t\tEnabled: &t.Enabled,\n\t}\n}\n\n// ToPersistenceInfoMap converts thrift to internal map[string]*types.PersistenceInfo type\nfunc ToPersistenceInfoMap(t map[string]*admin.PersistenceInfo) map[string]*types.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.PersistenceInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToPersistenceInfo(t[key])\n\t}\n\treturn v\n}\n\n// ToPersistenceInfo converts thrift to internal PersistenceInfo type\nfunc ToPersistenceInfo(t *admin.PersistenceInfo) *types.PersistenceInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PersistenceInfo{\n\t\tBackend:  t.GetBackend(),\n\t\tSettings: ToPersistenceSettings(t.Settings),\n\t\tFeatures: ToPersistenceFeatures(t.Features),\n\t}\n}\n\n// ToPersistenceSettings converts thrift to internal []*types.PersistenceSetting type\nfunc ToPersistenceSettings(t []*admin.PersistenceSetting) []*types.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PersistenceSetting, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPersistenceSetting(t[i])\n\t}\n\treturn v\n}\n\n// ToPersistenceSetting converts from thrift to internal PersistenceSetting type\nfunc ToPersistenceSetting(t *admin.PersistenceSetting) *types.PersistenceSetting {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PersistenceSetting{\n\t\tKey:   t.GetKey(),\n\t\tValue: t.GetValue(),\n\t}\n}\n\n// ToPersistenceFeatures converts from thrift to internal []*types.PersistenceFeature type\nfunc ToPersistenceFeatures(t []*admin.PersistenceFeature) []*types.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PersistenceFeature, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPersistenceFeature(t[i])\n\t}\n\treturn v\n}\n\n// ToPersistenceFeature converts from thrift to internal PersistenceFeature type\nfunc ToPersistenceFeature(t *admin.PersistenceFeature) *types.PersistenceFeature {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PersistenceFeature{\n\t\tKey:     t.GetKey(),\n\t\tEnabled: t.GetEnabled(),\n\t}\n}\n\n// FromAdminResendReplicationTasksRequest converts internal ResendReplicationTasksRequest type to thrift\nfunc FromAdminResendReplicationTasksRequest(t *types.ResendReplicationTasksRequest) *admin.ResendReplicationTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.ResendReplicationTasksRequest{\n\t\tDomainID:      &t.DomainID,\n\t\tWorkflowID:    &t.WorkflowID,\n\t\tRunID:         &t.RunID,\n\t\tRemoteCluster: &t.RemoteCluster,\n\t\tStartEventID:  t.StartEventID,\n\t\tStartVersion:  t.StartVersion,\n\t\tEndEventID:    t.EndEventID,\n\t\tEndVersion:    t.EndVersion,\n\t}\n}\n\n// ToAdminResendReplicationTasksRequest converts thrift ResendReplicationTasksRequest type to internal\nfunc ToAdminResendReplicationTasksRequest(t *admin.ResendReplicationTasksRequest) *types.ResendReplicationTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResendReplicationTasksRequest{\n\t\tDomainID:      t.GetDomainID(),\n\t\tWorkflowID:    t.GetWorkflowID(),\n\t\tRunID:         t.GetRunID(),\n\t\tRemoteCluster: t.GetRemoteCluster(),\n\t\tStartEventID:  t.StartEventID,\n\t\tStartVersion:  t.StartVersion,\n\t\tEndEventID:    t.EndEventID,\n\t\tEndVersion:    t.EndVersion,\n\t}\n}\n\n// FromRingInfo converts internal RingInfo type to thrift\nfunc FromRingInfo(t *types.RingInfo) *admin.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.RingInfo{\n\t\tRole:        &t.Role,\n\t\tMemberCount: &t.MemberCount,\n\t\tMembers:     FromHostInfoArray(t.Members),\n\t}\n}\n\n// ToRingInfo converts thrift RingInfo type to internal\nfunc ToRingInfo(t *admin.RingInfo) *types.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RingInfo{\n\t\tRole:        t.GetRole(),\n\t\tMemberCount: t.GetMemberCount(),\n\t\tMembers:     ToHostInfoArray(t.Members),\n\t}\n}\n\n// FromRingInfoArray converts internal RingInfo type array to thrift\nfunc FromRingInfoArray(t []*types.RingInfo) []*admin.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*admin.RingInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromRingInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToRingInfoArray converts thrift RingInfo type array to internal\nfunc ToRingInfoArray(t []*admin.RingInfo) []*types.RingInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.RingInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToRingInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromHostInfoArray converts internal HostInfo type array to thrift\nfunc FromHostInfoArray(t []*types.HostInfo) []*admin.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*admin.HostInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHostInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToHostInfoArray converts thrift HostInfo type array to internal\nfunc ToHostInfoArray(t []*admin.HostInfo) []*types.HostInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.HostInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHostInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromAdminGetDynamicConfigRequest converts internal GetDynamicConfigRequest type to thrift\nfunc FromAdminGetDynamicConfigRequest(t *types.GetDynamicConfigRequest) *admin.GetDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetDynamicConfigRequest{\n\t\tConfigName: &t.ConfigName,\n\t\tFilters:    FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToAdminGetDynamicConfigRequest converts thrift GetDynamicConfigRequest type to internal\nfunc ToAdminGetDynamicConfigRequest(t *admin.GetDynamicConfigRequest) *types.GetDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDynamicConfigRequest{\n\t\tConfigName: t.GetConfigName(),\n\t\tFilters:    ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromAdminGetDynamicConfigResponse converts internal GetDynamicConfigResponse type to thrift\nfunc FromAdminGetDynamicConfigResponse(t *types.GetDynamicConfigResponse) *admin.GetDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetDynamicConfigResponse{\n\t\tValue: FromDataBlob(t.Value),\n\t}\n}\n\n// ToAdminGetDynamicConfigResponse converts thrift GetDynamicConfigResponse type to internal\nfunc ToAdminGetDynamicConfigResponse(t *admin.GetDynamicConfigResponse) *types.GetDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDynamicConfigResponse{\n\t\tValue: ToDataBlob(t.Value),\n\t}\n}\n\n// FromAdminUpdateDynamicConfigRequest converts internal UpdateDynamicConfigRequest type to thrift\nfunc FromAdminUpdateDynamicConfigRequest(t *types.UpdateDynamicConfigRequest) *admin.UpdateDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.UpdateDynamicConfigRequest{\n\t\tConfigName:   &t.ConfigName,\n\t\tConfigValues: FromDynamicConfigValueArray(t.ConfigValues),\n\t}\n}\n\n// ToAdminUpdateDynamicConfigRequest converts thrift UpdateDynamicConfigRequest type to internal\nfunc ToAdminUpdateDynamicConfigRequest(t *admin.UpdateDynamicConfigRequest) *types.UpdateDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDynamicConfigRequest{\n\t\tConfigName:   t.GetConfigName(),\n\t\tConfigValues: ToDynamicConfigValueArray(t.ConfigValues),\n\t}\n}\n\n// FromAdminRestoreDynamicConfigRequest converts internal RestoreDynamicConfigRequest type to thrift\nfunc FromAdminRestoreDynamicConfigRequest(t *types.RestoreDynamicConfigRequest) *admin.RestoreDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.RestoreDynamicConfigRequest{\n\t\tConfigName: &t.ConfigName,\n\t\tFilters:    FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToAdminRestoreDynamicConfigRequest converts thrift RestoreDynamicConfigRequest type to internal\nfunc ToAdminRestoreDynamicConfigRequest(t *admin.RestoreDynamicConfigRequest) *types.RestoreDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestoreDynamicConfigRequest{\n\t\tConfigName: t.GetConfigName(),\n\t\tFilters:    ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromAdminDeleteWorkflowRequest converts internal AdminDeleteWorkflowRequest type to thrift\nfunc FromAdminDeleteWorkflowRequest(t *types.AdminDeleteWorkflowRequest) *admin.AdminDeleteWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.AdminDeleteWorkflowRequest{\n\t\tDomain:    &t.Domain,\n\t\tExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToAdminDeleteWorkflowRequest converts thrift AdminDeleteWorkflowRequest type to internal\nfunc ToAdminDeleteWorkflowRequest(t *admin.AdminDeleteWorkflowRequest) *types.AdminDeleteWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDeleteWorkflowRequest{\n\t\tDomain:    t.GetDomain(),\n\t\tExecution: ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromAdminDeleteWorkflowResponse converts internal AdminDeleteWorkflowResponse type to thrift\nfunc FromAdminDeleteWorkflowResponse(t *types.AdminDeleteWorkflowResponse) *admin.AdminDeleteWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.AdminDeleteWorkflowResponse{\n\t\tHistoryDeleted:    &t.HistoryDeleted,\n\t\tExecutionsDeleted: &t.ExecutionsDeleted,\n\t\tVisibilityDeleted: &t.VisibilityDeleted,\n\t}\n}\n\n// ToAdminDeleteWorkflowResponse converts thrift AdminDeleteWorkflowResponse type to internal\nfunc ToAdminDeleteWorkflowResponse(t *admin.AdminDeleteWorkflowResponse) *types.AdminDeleteWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminDeleteWorkflowResponse{\n\t\tHistoryDeleted:    *t.HistoryDeleted,\n\t\tExecutionsDeleted: *t.ExecutionsDeleted,\n\t\tVisibilityDeleted: *t.VisibilityDeleted,\n\t}\n}\n\n// FromAdminMaintainCorruptWorkflowRequest converts internal AdminMaintainWorkflowRequest type to thrift\nfunc FromAdminMaintainCorruptWorkflowRequest(t *types.AdminMaintainWorkflowRequest) *admin.AdminMaintainWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.AdminMaintainWorkflowRequest{\n\t\tDomain:    &t.Domain,\n\t\tExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToAdminMaintainCorruptWorkflowRequest converts thrift AdminMaintainWorkflowRequest type to internal\nfunc ToAdminMaintainCorruptWorkflowRequest(t *admin.AdminMaintainWorkflowRequest) *types.AdminMaintainWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminMaintainWorkflowRequest{\n\t\tDomain:    t.GetDomain(),\n\t\tExecution: ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromAdminMaintainCorruptWorkflowResponse converts internal AdminMaintainWorkflowResponse type to thrift\nfunc FromAdminMaintainCorruptWorkflowResponse(t *types.AdminMaintainWorkflowResponse) *admin.AdminMaintainWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.AdminMaintainWorkflowResponse{\n\t\tHistoryDeleted:    &t.HistoryDeleted,\n\t\tExecutionsDeleted: &t.ExecutionsDeleted,\n\t\tVisibilityDeleted: &t.VisibilityDeleted,\n\t}\n}\n\n// ToAdminMaintainCorruptWorkflowResponse converts thrift AdminMaintainWorkflowResponse type to internal\nfunc ToAdminMaintainCorruptWorkflowResponse(t *admin.AdminMaintainWorkflowResponse) *types.AdminMaintainWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AdminMaintainWorkflowResponse{\n\t\tHistoryDeleted:    *t.HistoryDeleted,\n\t\tExecutionsDeleted: *t.ExecutionsDeleted,\n\t\tVisibilityDeleted: *t.VisibilityDeleted,\n\t}\n}\n\n// FromAdminListDynamicConfigResponse converts internal ListDynamicConfigResponse type to thrift\nfunc FromAdminListDynamicConfigResponse(t *types.ListDynamicConfigResponse) *admin.ListDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.ListDynamicConfigResponse{\n\t\tEntries: FromDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\n// FromAdminListDynamicConfigRequest converts internal ListDynamicConfigRequest type to thrift\nfunc FromAdminListDynamicConfigRequest(t *types.ListDynamicConfigRequest) *admin.ListDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.ListDynamicConfigRequest{\n\t\tConfigName: &t.ConfigName,\n\t}\n}\n\n// ToAdminListDynamicConfigRequest converts thrift ListDynamicConfigRequest type to internal\nfunc ToAdminListDynamicConfigRequest(t *admin.ListDynamicConfigRequest) *types.ListDynamicConfigRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDynamicConfigRequest{\n\t\tConfigName: t.GetConfigName(),\n\t}\n}\n\n// ToAdminListDynamicConfigResponse converts thrift ListDynamicConfigResponse type to internal\nfunc ToAdminListDynamicConfigResponse(t *admin.ListDynamicConfigResponse) *types.ListDynamicConfigResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDynamicConfigResponse{\n\t\tEntries: ToDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\nfunc FromAdminGetGlobalIsolationGroupsRequest(t *types.GetGlobalIsolationGroupsRequest) *admin.GetGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetGlobalIsolationGroupsRequest{}\n}\n\nfunc FromAdminGetDomainIsolationGroupsRequest(t *types.GetDomainIsolationGroupsRequest) *admin.GetDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetDomainIsolationGroupsRequest{\n\t\tDomain: &t.Domain,\n\t}\n}\n\nfunc FromAdminGetGlobalIsolationGroupsResponse(t *types.GetGlobalIsolationGroupsResponse) *admin.GetGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := FromIsolationGroupConfig(&t.IsolationGroups)\n\tif cfg == nil || cfg.GetIsolationGroups() == nil {\n\t\treturn &admin.GetGlobalIsolationGroupsResponse{}\n\t}\n\treturn &admin.GetGlobalIsolationGroupsResponse{\n\t\tIsolationGroups: cfg,\n\t}\n}\n\nfunc ToAdminGetGlobalIsolationGroupsResponse(t *admin.GetGlobalIsolationGroupsResponse) *types.GetGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.IsolationGroups == nil {\n\t\treturn &types.GetGlobalIsolationGroupsResponse{}\n\t}\n\tig := ToIsolationGroupConfig(t.IsolationGroups)\n\tif ig == nil || len(*ig) == 0 {\n\t\treturn &types.GetGlobalIsolationGroupsResponse{}\n\t}\n\treturn &types.GetGlobalIsolationGroupsResponse{\n\t\tIsolationGroups: *ig,\n\t}\n}\n\nfunc ToAdminGetDomainIsolationGroupsResponse(t *admin.GetDomainIsolationGroupsResponse) *types.GetDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.IsolationGroups == nil {\n\t\treturn &types.GetDomainIsolationGroupsResponse{}\n\t}\n\tig := ToIsolationGroupConfig(t.IsolationGroups)\n\tif ig == nil || len(*ig) == 0 {\n\t\treturn &types.GetDomainIsolationGroupsResponse{}\n\t}\n\treturn &types.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: *ig,\n\t}\n}\n\nfunc ToAdminGetGlobalIsolationGroupsRequest(t *admin.GetGlobalIsolationGroupsRequest) *types.GetGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetGlobalIsolationGroupsRequest{}\n}\n\nfunc FromAdminGetDomainIsolationGroupsResponse(t *types.GetDomainIsolationGroupsResponse) *admin.GetDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := FromIsolationGroupConfig(&t.IsolationGroups)\n\treturn &admin.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: cfg,\n\t}\n}\n\nfunc ToAdminGetDomainIsolationGroupsRequest(t *admin.GetDomainIsolationGroupsRequest) *types.GetDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainIsolationGroupsRequest{Domain: t.GetDomain()}\n}\n\nfunc FromAdminUpdateGlobalIsolationGroupsResponse(t *types.UpdateGlobalIsolationGroupsResponse) *admin.UpdateGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.UpdateGlobalIsolationGroupsResponse{}\n}\n\nfunc FromAdminUpdateGlobalIsolationGroupsRequest(t *types.UpdateGlobalIsolationGroupsRequest) *admin.UpdateGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.IsolationGroups == nil {\n\t\treturn &admin.UpdateGlobalIsolationGroupsRequest{}\n\t}\n\treturn &admin.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: FromIsolationGroupConfig(&t.IsolationGroups),\n\t}\n}\n\nfunc FromAdminUpdateDomainIsolationGroupsRequest(t *types.UpdateDomainIsolationGroupsRequest) *admin.UpdateDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tif t.IsolationGroups == nil {\n\t\treturn &admin.UpdateDomainIsolationGroupsRequest{}\n\t}\n\treturn &admin.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          &t.Domain,\n\t\tIsolationGroups: FromIsolationGroupConfig(&t.IsolationGroups),\n\t}\n}\n\nfunc ToAdminUpdateGlobalIsolationGroupsRequest(t *admin.UpdateGlobalIsolationGroupsRequest) *types.UpdateGlobalIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := ToIsolationGroupConfig(t.IsolationGroups)\n\tif cfg == nil {\n\t\treturn &types.UpdateGlobalIsolationGroupsRequest{}\n\t}\n\treturn &types.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: *cfg,\n\t}\n}\n\nfunc ToAdminUpdateGlobalIsolationGroupsResponse(t *admin.UpdateGlobalIsolationGroupsResponse) *types.UpdateGlobalIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateGlobalIsolationGroupsResponse{}\n}\n\nfunc ToAdminUpdateDomainIsolationGroupsResponse(t *admin.UpdateDomainIsolationGroupsResponse) *types.UpdateDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainIsolationGroupsResponse{}\n}\n\nfunc FromAdminUpdateDomainIsolationGroupsResponse(t *types.UpdateDomainIsolationGroupsResponse) *admin.UpdateDomainIsolationGroupsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &admin.UpdateDomainIsolationGroupsResponse{}\n}\n\nfunc ToAdminUpdateDomainIsolationGroupsRequest(t *admin.UpdateDomainIsolationGroupsRequest) *types.UpdateDomainIsolationGroupsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tcfg := ToIsolationGroupConfig(t.IsolationGroups)\n\tif cfg == nil {\n\t\treturn &types.UpdateDomainIsolationGroupsRequest{\n\t\t\tDomain: t.GetDomain(),\n\t\t}\n\t}\n\treturn &types.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          t.GetDomain(),\n\t\tIsolationGroups: *cfg,\n\t}\n}\n\nfunc FromIsolationGroupConfig(in *types.IsolationGroupConfiguration) *shared.IsolationGroupConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\tvar out []*shared.IsolationGroupPartition\n\tfor _, v := range *in {\n\t\tout = append(out, &shared.IsolationGroupPartition{\n\t\t\tName:  strPtr(v.Name),\n\t\t\tState: igStatePtr(shared.IsolationGroupState(v.State)),\n\t\t})\n\t}\n\tsort.Slice(out, func(i, j int) bool {\n\t\tif out[i] == nil || out[j] == nil {\n\t\t\treturn false\n\t\t}\n\t\treturn *out[i].Name < *out[j].Name\n\t})\n\treturn &shared.IsolationGroupConfiguration{\n\t\tIsolationGroups: out,\n\t}\n}\n\nfunc ToIsolationGroupConfig(in *shared.IsolationGroupConfiguration) *types.IsolationGroupConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\tout := make(types.IsolationGroupConfiguration)\n\tfor v := range in.IsolationGroups {\n\t\tout[in.IsolationGroups[v].GetName()] = types.IsolationGroupPartition{\n\t\t\tName:  in.IsolationGroups[v].GetName(),\n\t\t\tState: types.IsolationGroupState(in.IsolationGroups[v].GetState()),\n\t\t}\n\t}\n\treturn &out\n}\n\nfunc ToAdminGetDomainAsyncWorkflowConfiguratonRequest(in *admin.GetDomainAsyncWorkflowConfiguratonRequest) *types.GetDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain: in.GetDomain(),\n\t}\n}\n\nfunc FromAdminGetDomainAsyncWorkflowConfiguratonResponse(in *types.GetDomainAsyncWorkflowConfiguratonResponse) *admin.GetDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\tConfiguration: FromDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc FromDomainAsyncWorkflowConfiguraton(in *types.AsyncWorkflowConfiguration) *shared.AsyncWorkflowConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\n\treturn &shared.AsyncWorkflowConfiguration{\n\t\tEnabled:             &in.Enabled,\n\t\tPredefinedQueueName: strPtr(in.PredefinedQueueName),\n\t\tQueueType:           strPtr(in.QueueType),\n\t\tQueueConfig:         FromDataBlob(in.QueueConfig),\n\t}\n}\n\nfunc ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(in *admin.UpdateDomainAsyncWorkflowConfiguratonRequest) *types.UpdateDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain:        in.GetDomain(),\n\t\tConfiguration: ToDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc ToDomainAsyncWorkflowConfiguraton(in *shared.AsyncWorkflowConfiguration) *types.AsyncWorkflowConfiguration {\n\tif in == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.AsyncWorkflowConfiguration{\n\t\tEnabled:             in.GetEnabled(),\n\t\tPredefinedQueueName: in.GetPredefinedQueueName(),\n\t\tQueueType:           in.GetQueueType(),\n\t\tQueueConfig:         ToDataBlob(in.GetQueueConfig()),\n\t}\n}\n\nfunc FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(in *types.UpdateDomainAsyncWorkflowConfiguratonResponse) *admin.UpdateDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &admin.UpdateDomainAsyncWorkflowConfiguratonResponse{}\n}\n\nfunc FromAdminGetDomainAsyncWorkflowConfiguratonRequest(in *types.GetDomainAsyncWorkflowConfiguratonRequest) *admin.GetDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &admin.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain: strPtr(in.Domain),\n\t}\n}\n\nfunc ToAdminGetDomainAsyncWorkflowConfiguratonResponse(in *admin.GetDomainAsyncWorkflowConfiguratonResponse) *types.GetDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\tConfiguration: ToDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(in *types.UpdateDomainAsyncWorkflowConfiguratonRequest) *admin.UpdateDomainAsyncWorkflowConfiguratonRequest {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain:        strPtr(in.Domain),\n\t\tConfiguration: FromDomainAsyncWorkflowConfiguraton(in.Configuration),\n\t}\n}\n\nfunc ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(in *admin.UpdateDomainAsyncWorkflowConfiguratonResponse) *types.UpdateDomainAsyncWorkflowConfiguratonResponse {\n\tif in == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainAsyncWorkflowConfiguratonResponse{}\n}\n\nfunc strPtr(s string) *string                                             { return &s }\nfunc igStatePtr(s shared.IsolationGroupState) *shared.IsolationGroupState { return &s }\n"
  },
  {
    "path": "common/types/mapper/thrift/admin_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"sort\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/admin\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestAdminAddSearchAttributeRequest(t *testing.T) {\n\tfor _, item := range []*types.AddSearchAttributeRequest{nil, {}, &testdata.AdminAddSearchAttributeRequest} {\n\t\tassert.Equal(t, item, ToAdminAddSearchAttributeRequest(FromAdminAddSearchAttributeRequest(item)))\n\t}\n}\nfunc TestAdminCloseShardRequest(t *testing.T) {\n\tfor _, item := range []*types.CloseShardRequest{nil, {}, &testdata.AdminCloseShardRequest} {\n\t\tassert.Equal(t, item, ToAdminCloseShardRequest(FromAdminCloseShardRequest(item)))\n\t}\n}\nfunc TestAdminDeleteWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminDeleteWorkflowRequest{nil, {}, &testdata.AdminDeleteWorkflowRequest} {\n\t\tassert.Equal(t, item, ToAdminDeleteWorkflowRequest(FromAdminDeleteWorkflowRequest(item)))\n\t}\n}\nfunc TestAdminDeleteWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminDeleteWorkflowResponse{nil, {}, &testdata.AdminDeleteWorkflowResponse} {\n\t\tassert.Equal(t, item, ToAdminDeleteWorkflowResponse(FromAdminDeleteWorkflowResponse(item)))\n\t}\n}\nfunc TestAdminDescribeClusterResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeClusterResponse{nil, {}, &testdata.AdminDescribeClusterResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeClusterResponse(FromAdminDescribeClusterResponse(item)))\n\t}\n}\nfunc TestAdminDescribeHistoryHostRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostRequest{\n\t\tnil,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByHost,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByShard,\n\t\t&testdata.AdminDescribeHistoryHostRequest_ByExecution,\n\t} {\n\t\tassert.Equal(t, item, ToAdminDescribeHistoryHostRequest(FromAdminDescribeHistoryHostRequest(item)))\n\t}\n}\nfunc TestAdminDescribeHistoryHostResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeHistoryHostResponse{nil, {}, &testdata.AdminDescribeHistoryHostResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeHistoryHostResponse(FromAdminDescribeHistoryHostResponse(item)))\n\t}\n}\nfunc TestAdminDescribeQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueRequest{nil, {}, &testdata.AdminDescribeQueueRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeQueueRequest(FromAdminDescribeQueueRequest(item)))\n\t}\n}\nfunc TestAdminDescribeQueueResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeQueueResponse{nil, {}, &testdata.AdminDescribeQueueResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeQueueResponse(FromAdminDescribeQueueResponse(item)))\n\t}\n}\nfunc TestAdminDescribeShardDistributionRequest(t *testing.T) {\n\tfor _, item := range []*types.DescribeShardDistributionRequest{nil, {}, &testdata.AdminDescribeShardDistributionRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeShardDistributionRequest(FromAdminDescribeShardDistributionRequest(item)))\n\t}\n}\nfunc TestAdminDescribeShardDistributionResponse(t *testing.T) {\n\tfor _, item := range []*types.DescribeShardDistributionResponse{nil, {}, &testdata.AdminDescribeShardDistributionResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeShardDistributionResponse(FromAdminDescribeShardDistributionResponse(item)))\n\t}\n}\nfunc TestAdminDescribeWorkflowExecutionRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminDescribeWorkflowExecutionRequest{nil, {}, &testdata.AdminDescribeWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToAdminDescribeWorkflowExecutionRequest(FromAdminDescribeWorkflowExecutionRequest(item)))\n\t}\n}\nfunc TestAdminDescribeWorkflowExecutionResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminDescribeWorkflowExecutionResponse{nil, {ShardID: \"0\"}, &testdata.AdminDescribeWorkflowExecutionResponse} {\n\t\tassert.Equal(t, item, ToAdminDescribeWorkflowExecutionResponse(FromAdminDescribeWorkflowExecutionResponse(item)))\n\t}\n}\nfunc TestAdminGetDomainIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDomainIsolationGroupsRequest{nil, {}, &testdata.AdminGetDomainIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDomainIsolationGroupsRequest(FromAdminGetDomainIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminGetDomainIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDomainIsolationGroupsResponse{nil, {}, &testdata.AdminGetDomainIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDomainIsolationGroupsResponse(FromAdminGetDomainIsolationGroupsResponse(item)))\n\t}\n}\nfunc TestAdminGetDomainReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDomainReplicationMessagesRequest{nil, {}, &testdata.AdminGetDomainReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDomainReplicationMessagesRequest(FromAdminGetDomainReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestAdminGetDomainReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDomainReplicationMessagesResponse{nil, {}, &testdata.AdminGetDomainReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDomainReplicationMessagesResponse(FromAdminGetDomainReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestAdminGetDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.GetDynamicConfigRequest{nil, {}, &testdata.AdminGetDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminGetDynamicConfigRequest(FromAdminGetDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminGetDynamicConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.GetDynamicConfigResponse{nil, {}, &testdata.AdminGetDynamicConfigResponse} {\n\t\tassert.Equal(t, item, ToAdminGetDynamicConfigResponse(FromAdminGetDynamicConfigResponse(item)))\n\t}\n}\nfunc TestAdminGetGlobalIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.GetGlobalIsolationGroupsRequest{nil, {}, &testdata.AdminGetGlobalIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminGetGlobalIsolationGroupsRequest(FromAdminGetGlobalIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminGetReplicationMessagesRequest(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesRequest{nil, {}, &testdata.AdminGetReplicationMessagesRequest} {\n\t\tassert.Equal(t, item, ToAdminGetReplicationMessagesRequest(FromAdminGetReplicationMessagesRequest(item)))\n\t}\n}\nfunc TestAdminGetReplicationMessagesResponse(t *testing.T) {\n\tfor _, item := range []*types.GetReplicationMessagesResponse{nil, {}, &testdata.AdminGetReplicationMessagesResponse} {\n\t\tassert.Equal(t, item, ToAdminGetReplicationMessagesResponse(FromAdminGetReplicationMessagesResponse(item)))\n\t}\n}\nfunc TestAdminGetWorkflowExecutionRawHistoryV2Request(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionRawHistoryV2Request{nil, {}, &testdata.AdminGetWorkflowExecutionRawHistoryV2Request} {\n\t\tassert.Equal(t, item, ToAdminGetWorkflowExecutionRawHistoryV2Request(FromAdminGetWorkflowExecutionRawHistoryV2Request(item)))\n\t}\n}\nfunc TestAdminGetWorkflowExecutionRawHistoryV2Response(t *testing.T) {\n\tfor _, item := range []*types.GetWorkflowExecutionRawHistoryV2Response{nil, {}, &testdata.AdminGetWorkflowExecutionRawHistoryV2Response} {\n\t\tassert.Equal(t, item, ToAdminGetWorkflowExecutionRawHistoryV2Response(FromAdminGetWorkflowExecutionRawHistoryV2Response(item)))\n\t}\n}\nfunc TestAdminListDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.ListDynamicConfigRequest{nil, {}, &testdata.AdminListDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminListDynamicConfigRequest(FromAdminListDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminListDynamicConfigResponse(t *testing.T) {\n\tfor _, item := range []*types.ListDynamicConfigResponse{nil, {}, &testdata.AdminListDynamicConfigResponse} {\n\t\tassert.Equal(t, item, ToAdminListDynamicConfigResponse(FromAdminListDynamicConfigResponse(item)))\n\t}\n}\nfunc TestAdminMaintainCorruptWorkflowRequest(t *testing.T) {\n\tfor _, item := range []*types.AdminMaintainWorkflowRequest{nil, {}, &testdata.AdminMaintainCorruptWorkflowRequest} {\n\t\tassert.Equal(t, item, ToAdminMaintainCorruptWorkflowRequest(FromAdminMaintainCorruptWorkflowRequest(item)))\n\t}\n}\nfunc TestAdminMaintainCorruptWorkflowResponse(t *testing.T) {\n\tfor _, item := range []*types.AdminMaintainWorkflowResponse{nil, {}, &testdata.AdminMaintainCorruptWorkflowResponse} {\n\t\tassert.Equal(t, item, ToAdminMaintainCorruptWorkflowResponse(FromAdminMaintainCorruptWorkflowResponse(item)))\n\t}\n}\nfunc TestAdminReapplyEventsRequest(t *testing.T) {\n\tfor _, item := range []*types.ReapplyEventsRequest{nil, {}, &testdata.AdminReapplyEventsRequest} {\n\t\tassert.Equal(t, item, ToAdminReapplyEventsRequest(FromAdminReapplyEventsRequest(item)))\n\t}\n}\nfunc TestAdminRefreshWorkflowTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.RefreshWorkflowTasksRequest{nil, {}, &testdata.AdminRefreshWorkflowTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminRefreshWorkflowTasksRequest(FromAdminRefreshWorkflowTasksRequest(item)))\n\t}\n}\nfunc TestAdminRemoveTaskRequest(t *testing.T) {\n\tfor _, item := range []*types.RemoveTaskRequest{nil, {}, &testdata.AdminRemoveTaskRequest} {\n\t\tassert.Equal(t, item, ToAdminRemoveTaskRequest(FromAdminRemoveTaskRequest(item)))\n\t}\n}\nfunc TestAdminResendReplicationTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.ResendReplicationTasksRequest{nil, {}, &testdata.AdminResendReplicationTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminResendReplicationTasksRequest(FromAdminResendReplicationTasksRequest(item)))\n\t}\n}\nfunc TestAdminResetQueueRequest(t *testing.T) {\n\tfor _, item := range []*types.ResetQueueRequest{nil, {}, &testdata.AdminResetQueueRequest} {\n\t\tassert.Equal(t, item, ToAdminResetQueueRequest(FromAdminResetQueueRequest(item)))\n\t}\n}\n\nfunc TestAdminGetCrossClusterTasksRequest(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksRequest{nil, {}, &testdata.AdminGetCrossClusterTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksRequest(FromAdminGetCrossClusterTasksRequest(item)))\n\t}\n}\n\nfunc TestAdminGetCrossClusterTasksResponse(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksResponse{nil, {}, &testdata.AdminGetCrossClusterTasksResponse} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksResponse(FromAdminGetCrossClusterTasksResponse(item)))\n\t}\n}\n\nfunc TestAdminRespondCrossClusterTasksCompletedRequest(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedRequest{nil, {}, &testdata.AdminRespondCrossClusterTasksCompletedRequest} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedRequest(FromAdminRespondCrossClusterTasksCompletedRequest(item)))\n\t}\n}\n\nfunc TestAdminRespondCrossClusterTasksCompletedResponse(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedResponse{nil, {}, &testdata.AdminRespondCrossClusterTasksCompletedResponse} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedResponse(FromAdminRespondCrossClusterTasksCompletedResponse(item)))\n\t}\n}\nfunc TestAdminUpdateDomainIsolationGroupsRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainIsolationGroupsRequest{nil, {}, &testdata.AdminUpdateDomainIsolationGroupsRequest} {\n\t\tassert.Equal(t, item, ToAdminUpdateDomainIsolationGroupsRequest(FromAdminUpdateDomainIsolationGroupsRequest(item)))\n\t}\n}\nfunc TestAdminUpdateDomainIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateDomainIsolationGroupsResponse{nil, {}, &testdata.AdminUpdateDomainIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminUpdateDomainIsolationGroupsResponse(FromAdminUpdateDomainIsolationGroupsResponse(item)))\n\t}\n}\nfunc TestAdminRestoreDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.RestoreDynamicConfigRequest{nil, {}, &testdata.AdminRestoreDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminRestoreDynamicConfigRequest(FromAdminRestoreDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminUpdateDynamicConfigRequest(t *testing.T) {\n\tfor _, item := range []*types.UpdateDynamicConfigRequest{nil, {}, &testdata.AdminUpdateDynamicConfigRequest} {\n\t\tassert.Equal(t, item, ToAdminUpdateDynamicConfigRequest(FromAdminUpdateDynamicConfigRequest(item)))\n\t}\n}\nfunc TestAdminUpdateGlobalIsolationGroupsResponse(t *testing.T) {\n\tfor _, item := range []*types.UpdateGlobalIsolationGroupsResponse{nil, {}, &testdata.AdminUpdateGlobalIsolationGroupsResponse} {\n\t\tassert.Equal(t, item, ToAdminUpdateGlobalIsolationGroupsResponse(FromAdminUpdateGlobalIsolationGroupsResponse(item)))\n\t}\n}\n\nfunc TestFromGetGlobalIsolationGroupsResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetGlobalIsolationGroupsResponse\n\t\texpected *admin.GetGlobalIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 0\": {\n\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 1\": {\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &admin.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 0\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateHealthy),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 1\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateDrained),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil - 1\": {\n\t\t\tin:       &types.GetGlobalIsolationGroupsResponse{},\n\t\t\texpected: &admin.GetGlobalIsolationGroupsResponse{},\n\t\t},\n\t\t\"nil - 2\": {\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := FromAdminGetGlobalIsolationGroupsResponse(td.in)\n\t\t\tif res != nil && res.IsolationGroups != nil {\n\t\t\t\tsort.Slice(res.IsolationGroups.IsolationGroups, func(i int, j int) bool {\n\t\t\t\t\treturn *res.IsolationGroups.IsolationGroups[i].Name < *res.IsolationGroups.IsolationGroups[j].Name\n\t\t\t\t})\n\t\t\t}\n\t\t\tassert.Equal(t, td.expected, res, \"expected value\")\n\t\t\troundTrip := ToAdminGetGlobalIsolationGroupsResponse(res)\n\t\t\tassert.Equal(t, td.in, roundTrip, \"roundtrip value\")\n\t\t})\n\t}\n}\n\nfunc TestToGetGlobalIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *admin.GetGlobalIsolationGroupsRequest\n\t\texpected *types.GetGlobalIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin:       &admin.GetGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.GetGlobalIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil - 2\": {\n\t\t\tin:       &admin.GetGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.GetGlobalIsolationGroupsRequest{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetGlobalIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromGetDomainIsolationGroupsResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tin       *types.GetDomainIsolationGroupsResponse\n\t\texpected *admin.GetDomainIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &types.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 0\": {\n\t\t\t\t\t\tName:  \"zone 0\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 1\": {\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &admin.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 0\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateHealthy),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 1\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateDrained),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin: &types.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{},\n\t\t\t},\n\t\t\texpected: &admin.GetDomainIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := FromAdminGetDomainIsolationGroupsResponse(td.in)\n\t\t\t// map iteration is nondeterministic\n\t\t\tsort.Slice(res.IsolationGroups.IsolationGroups, func(i int, j int) bool {\n\t\t\t\treturn *res.IsolationGroups.IsolationGroups[i].Name > *res.IsolationGroups.IsolationGroups[j].Name\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc TestToGetDomainIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *admin.GetDomainIsolationGroupsRequest\n\t\texpected *types.GetDomainIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &admin.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: strPtr(\"domain123\"),\n\t\t\t},\n\t\t\texpected: &types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"domain123\",\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &admin.GetDomainIsolationGroupsRequest{},\n\t\t\texpected: &types.GetDomainIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetDomainIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestFromUpdateGlobalIsolationGroupsResponse(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *types.UpdateGlobalIsolationGroupsResponse\n\t\texpected *admin.UpdateGlobalIsolationGroupsResponse\n\t}{\n\t\t\"Valid mapping\": {},\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateGlobalIsolationGroupsResponse{},\n\t\t\texpected: &admin.UpdateGlobalIsolationGroupsResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateGlobalIsolationGroupsResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToUpdateGlobalIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *admin.UpdateGlobalIsolationGroupsRequest\n\t\texpected *types.UpdateGlobalIsolationGroupsRequest\n\t}{\n\t\t\"Valid mapping\": {\n\t\t\tin: &admin.UpdateGlobalIsolationGroupsRequest{\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 1\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateHealthy),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone 2\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateDrained),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateGlobalIsolationGroupsRequest{\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone 1\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone 1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone 2\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone 2\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty\": {\n\t\t\tin:       &admin.UpdateGlobalIsolationGroupsRequest{},\n\t\t\texpected: &types.UpdateGlobalIsolationGroupsRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := ToAdminUpdateGlobalIsolationGroupsRequest(td.in)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\troundTrip := FromAdminUpdateGlobalIsolationGroupsRequest(res)\n\t\t\tif td.in != nil {\n\t\t\t\tassert.Equal(t, td.in, roundTrip)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestFromUpdateDomainIsolationGroupsResponse(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *types.UpdateDomainIsolationGroupsResponse\n\t\texpected *admin.UpdateDomainIsolationGroupsResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tin:       &types.UpdateDomainIsolationGroupsResponse{},\n\t\t\texpected: &admin.UpdateDomainIsolationGroupsResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, FromAdminUpdateDomainIsolationGroupsResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToUpdateDomainIsolationGroupsRequest(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *admin.UpdateDomainIsolationGroupsRequest\n\t\texpected *types.UpdateDomainIsolationGroupsRequest\n\t}{\n\t\t\"valid\": {\n\t\t\tin: &admin.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone-1\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateHealthy),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone-2\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateDrained),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.UpdateDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"nil\": {\n\t\t\tin:       nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminUpdateDomainIsolationGroupsRequest(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToGetGlobalIsolationGroupsResponse(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tin       *admin.GetGlobalIsolationGroupsResponse\n\t\texpected *types.GetGlobalIsolationGroupsResponse\n\t}{\n\t\t\"valid\": {\n\t\t\tin: &admin.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: &shared.IsolationGroupConfiguration{\n\t\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone-1\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateDrained),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName:  strPtr(\"zone-2\"),\n\t\t\t\t\t\t\tState: igStatePtr(shared.IsolationGroupStateHealthy),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\tIsolationGroups: map[string]types.IsolationGroupPartition{\n\t\t\t\t\t\"zone-1\": {\n\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t},\n\t\t\t\t\t\"zone-2\": {\n\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"no groups\": {\n\t\t\tin:       &admin.GetGlobalIsolationGroupsResponse{},\n\t\t\texpected: &types.GetGlobalIsolationGroupsResponse{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expected, ToAdminGetGlobalIsolationGroupsResponse(td.in))\n\t\t})\n\t}\n}\n\nfunc TestToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\tenabled := true\n\ttests := map[string]struct {\n\t\tinput *admin.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\twant  *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\twant:  &types.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tinput: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"test-queue\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             enabled,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tinput: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   &enabled,\n\t\t\t\t\tQueueType: strPtr(\"kafka\"),\n\t\t\t\t\tQueueConfig: &shared.DataBlob{\n\t\t\t\t\t\tEncodingType: shared.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   enabled,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(td.input))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\twant  *admin.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\twant:  &admin.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(td.input))\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *admin.GetDomainAsyncWorkflowConfiguratonRequest\n\t\twant  *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &admin.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\twant:  &types.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"valid\": {\n\t\t\tinput: &admin.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t},\n\t\t\twant: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, ToAdminGetDomainAsyncWorkflowConfiguratonRequest(td.input))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminGetDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\tenabled := true\n\ttests := map[string]struct {\n\t\tinput *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t\twant  *admin.GetDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\twant:  &admin.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             enabled,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &admin.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"test-queue\"),\n\t\t\t\t\tQueueType:           strPtr(\"\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   enabled,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &admin.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"\"),\n\t\t\t\t\tQueueType:           strPtr(\"kafka\"),\n\t\t\t\t\tQueueConfig: &shared.DataBlob{\n\t\t\t\t\t\tEncodingType: shared.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, FromAdminGetDomainAsyncWorkflowConfiguratonResponse(td.input))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminGetDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t\twant  *admin.GetDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\twant: &admin.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"\"),\n\t\t\t},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"valid\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\twant: &admin.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, FromAdminGetDomainAsyncWorkflowConfiguratonRequest(td.input))\n\t\t})\n\t}\n}\n\nfunc TestToAdminGetDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\tenabled := true\n\ttests := map[string]struct {\n\t\tinput *admin.GetDomainAsyncWorkflowConfiguratonResponse\n\t\twant  *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &admin.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\twant:  &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tinput: &admin.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"test-queue\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             enabled,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tinput: &admin.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   &enabled,\n\t\t\t\t\tQueueType: strPtr(\"kafka\"),\n\t\t\t\t\tQueueConfig: &shared.DataBlob{\n\t\t\t\t\t\tEncodingType: shared.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   enabled,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, ToAdminGetDomainAsyncWorkflowConfiguratonResponse(td.input))\n\t\t})\n\t}\n}\n\nfunc TestFromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(t *testing.T) {\n\tenabled := true\n\ttests := map[string]struct {\n\t\tinput *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\twant  *admin.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{},\n\t\t\twant: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"\"),\n\t\t\t},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t\t\"predefined queue\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             enabled,\n\t\t\t\t\tPredefinedQueueName: \"test-queue\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"test-queue\"),\n\t\t\t\t\tQueueType:           strPtr(\"\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"inline queue\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:   enabled,\n\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: &admin.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\tDomain: strPtr(\"test-domain\"),\n\t\t\t\tConfiguration: &shared.AsyncWorkflowConfiguration{\n\t\t\t\t\tEnabled:             &enabled,\n\t\t\t\t\tPredefinedQueueName: strPtr(\"\"),\n\t\t\t\t\tQueueType:           strPtr(\"kafka\"),\n\t\t\t\t\tQueueConfig: &shared.DataBlob{\n\t\t\t\t\t\tEncodingType: shared.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, FromAdminUpdateDomainAsyncWorkflowConfiguratonRequest(td.input))\n\t\t})\n\t}\n}\n\nfunc TestToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput *admin.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\twant  *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t}{\n\t\t\"empty\": {\n\t\t\tinput: &admin.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t\twant:  &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil\": {\n\t\t\tinput: nil,\n\t\t\twant:  nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.want, ToAdminUpdateDomainAsyncWorkflowConfiguratonResponse(td.input))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/any.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"bytes\"\n\t\"fmt\"\n\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\t\"go.uber.org/thriftrw/wire\"\n)\n\n// any.go contains Any-helpers for encoding/decoding thrift data\n\n// ThriftObject is satisfied by any thrift type, as they all have To/FromWire methods.\n// go.uber.org/cadence has exactly this in an internal API, it's just copied here for simplicity.\ntype ThriftObject interface {\n\tFromWire(w wire.Value) error\n\tToWire() (wire.Value, error)\n}\n\nfunc EncodeToBytes(obj ThriftObject) ([]byte, error) {\n\t// get the intermediate format, ready for byte-encoding\n\twireValue, err := obj.ToWire()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to produce intermediate wire.Value from type %T\", obj)\n\t}\n\t// write to binary\n\tvar writer bytes.Buffer\n\terr = binary.Default.Encode(wireValue, &writer)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to encode wire.Value from type %T\", obj)\n\t}\n\t// and return the bytes\n\treturn writer.Bytes(), nil\n}\n\nfunc DecodeStructFromBytes(data []byte, target ThriftObject) error {\n\t// decode to the intermediate format, like encoding.\n\t// all top-most types likely *should* be a wire.TStruct, but if Any is at some point used\n\t// to contain a map/int/etc, just make sure to use wire.TMap / wire.TI64 / etc as is appropriate.\n\tintermediate, err := binary.Default.Decode(bytes.NewReader(data), wire.TStruct)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"could not decode thrift bytes to intermediate wire.Value for type %T\", target)\n\t}\n\t// and populate the target with the wire.Value contents\n\terr = target.FromWire(intermediate)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"could not FromWire to the target type %T\", target)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/any_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestThriftInAny(t *testing.T) {\n\t// pushing thrift or proto data through this type is very much expected,\n\t// so this is both evidence that it's possible and an example for how to do so.\n\t//\n\t// note that there is an equivalent test in mapper/proto/shared_test.go.\n\t// they are structurally the same, but encoding/decoding details vary a bit.\n\n\t// --- create the original data, a thrift object, and encode it by hand\n\torig := &shared.WorkflowExecution{\n\t\tWorkflowId: common.StringPtr(testdata.WorkflowID),\n\t\tRunId:      common.StringPtr(testdata.RunID),\n\t}\n\tinternalBytes, err := EncodeToBytes(orig)\n\trequire.NoError(t, err)\n\n\t// --- put that data into the custom Any type\n\t// thrift has no registry like proto has, so there is no canonical name.\n\t// no problem, just generate one.\n\tvar typeName = reflect.TypeOf(orig).PkgPath() + reflect.TypeOf(orig).Name()\n\tanyVal := &types.Any{\n\t\tValueType: typeName,\n\t\tValue:     internalBytes, // store thrift bytes in the Any\n\t}\n\n\t// --- convert the whole container to thrift (mimics making a call via yarpc)\n\tthriftAny := FromAny(anyVal)                  // we map to the rpc type\n\tnetworkBytes, err := EncodeToBytes(thriftAny) // yarpc does this\n\trequire.NoError(t, err)\n\t// ^ this is what's sent over the network.\n\n\t// as a side note:\n\t// the final data is not double-encoded, so this \"encode -> wrap -> encode\" process is reasonably efficient.\n\t//\n\t// Thrift and Proto can efficiently move around binary blobs like this, as it's essentially just a memcpy between\n\t// the input and the output, and there's no `\\0` to `\\\\0` escaping or base64 encoding or whatever needed.\n\t//\n\t// no behavior depends on this, it's just presented here as evidence that this Any-wrapper does not meaningfully\n\t// change any RPC design concerns: anything you would do with normal RPC can be done through an Any if you need\n\t// loose typing, the change-stability / performance / etc is entirely unaffected.\n\t//\n\t// compare via a sliding window to find the place it overlaps, to prove that this is true:\n\tfound := false\n\tfor i := 0; i <= len(networkBytes)-len(internalBytes); i++ {\n\t\tif reflect.DeepEqual(internalBytes, networkBytes[i:i+len(internalBytes)]) {\n\t\t\tfound = true\n\t\t\tt.Logf(\"Found matching bytes at index %v\", i) // currently at index 14\n\t\t}\n\t}\n\t// *should* be true for efficiency's sake, but is not truly necessary for correct behavior\n\tassert.Truef(t, found, \"did not find internal bytes within network bytes, might be paying double-encoding costs:\\n\\tinternal: %v\\n\\tnetwork:  %v\", internalBytes, networkBytes)\n\n\t// --- the network pushes the data to a new location ---\n\n\t// --- on the receiving side, we map to internal types like normal\n\tvar outAny shared.Any\n\terr = DecodeStructFromBytes(networkBytes, &outAny) // yarpc does this\n\trequire.NoError(t, err)\n\toutAnyVal := ToAny(&outAny) // we map to internal types\n\n\t// --- and finally decode the any-typed data by hand\n\trequire.Equal(t, typeName, outAnyVal.ValueType, \"type name through RPC should match the original type name\")\n\tvar outOrig shared.WorkflowExecution                   // selected based on the ValueType contents\n\terr = DecodeStructFromBytes(outAnyVal.Value, &outOrig) // do the actual custom decoding\n\trequire.NoError(t, err)\n\tassert.NotEmpty(t, outOrig, \"sanity check, decoded value should not be empty\")\n\tassert.Equal(t, orig, &outOrig, \"final round-tripped Any-contained data should be identical to the original object\")\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/configStore.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/config\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// FromDynamicConfigBlob converts internal DynamicConfigBlob type to thrift\nfunc FromDynamicConfigBlob(t *types.DynamicConfigBlob) *config.DynamicConfigBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &config.DynamicConfigBlob{\n\t\tSchemaVersion: &t.SchemaVersion,\n\t\tEntries:       FromDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\n// ToDynamicConfigBlob converts thrift DynamicConfigBlob type to internal\nfunc ToDynamicConfigBlob(t *config.DynamicConfigBlob) *types.DynamicConfigBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigBlob{\n\t\tSchemaVersion: t.GetSchemaVersion(),\n\t\tEntries:       ToDynamicConfigEntryArray(t.Entries),\n\t}\n}\n\n// FromDynamicConfigEntryArray converts internal DynamicConfigEntry array type to thrift\nfunc FromDynamicConfigEntryArray(t []*types.DynamicConfigEntry) []*config.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*config.DynamicConfigEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigEntry(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigEntryArray converts thrift DynamicConfigEntry array type to internal\nfunc ToDynamicConfigEntryArray(t []*config.DynamicConfigEntry) []*types.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigEntry, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigEntry(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigEntry converts internal DynamicConfigEntry type to thrift\nfunc FromDynamicConfigEntry(t *types.DynamicConfigEntry) *config.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &config.DynamicConfigEntry{\n\t\tName:   &t.Name,\n\t\tValues: FromDynamicConfigValueArray(t.Values),\n\t}\n}\n\n// ToDynamicConfigEntry converts thrift DynamicConfigEntry type to internal\nfunc ToDynamicConfigEntry(t *config.DynamicConfigEntry) *types.DynamicConfigEntry {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigEntry{\n\t\tName:   t.GetName(),\n\t\tValues: ToDynamicConfigValueArray(t.Values),\n\t}\n}\n\n// FromDynamicConfigValueArray converts internal DynamicConfigValue array type to thrift\nfunc FromDynamicConfigValueArray(t []*types.DynamicConfigValue) []*config.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*config.DynamicConfigValue, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigValue(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigValueArray converts thrift DynamicConfigValue array type to internal\nfunc ToDynamicConfigValueArray(t []*config.DynamicConfigValue) []*types.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigValue, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigValue(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigValue converts internal DynamicConfigValue type to thrift\nfunc FromDynamicConfigValue(t *types.DynamicConfigValue) *config.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &config.DynamicConfigValue{\n\t\tValue:   FromDataBlob(t.Value),\n\t\tFilters: FromDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// ToDynamicConfigValue converts thrift DynamicConfigValue type to internal\nfunc ToDynamicConfigValue(t *config.DynamicConfigValue) *types.DynamicConfigValue {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigValue{\n\t\tValue:   ToDataBlob(t.Value),\n\t\tFilters: ToDynamicConfigFilterArray(t.Filters),\n\t}\n}\n\n// FromDynamicConfigFilterArray converts internal DynamicConfigFilter array type to thrift\nfunc FromDynamicConfigFilterArray(t []*types.DynamicConfigFilter) []*config.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*config.DynamicConfigFilter, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDynamicConfigFilter(t[i])\n\t}\n\treturn v\n}\n\n// ToDynamicConfigFilterArray converts thrift DynamicConfigFilter array type to internal\nfunc ToDynamicConfigFilterArray(t []*config.DynamicConfigFilter) []*types.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DynamicConfigFilter, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDynamicConfigFilter(t[i])\n\t}\n\treturn v\n}\n\n// FromDynamicConfigFilter converts internal DynamicConfigFilter type to thrift\nfunc FromDynamicConfigFilter(t *types.DynamicConfigFilter) *config.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &config.DynamicConfigFilter{\n\t\tName:  &t.Name,\n\t\tValue: FromDataBlob(t.Value),\n\t}\n}\n\n// ToDynamicConfigFilter converts thrift DynamicConfigFilter type to internal\nfunc ToDynamicConfigFilter(t *config.DynamicConfigFilter) *types.DynamicConfigFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DynamicConfigFilter{\n\t\tName:  t.GetName(),\n\t\tValue: ToDataBlob(t.Value),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/config_store_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestDynamicConfigBlob(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DynamicConfigBlob\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.DynamicConfigBlob,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.DynamicConfigBlob{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigBlob(tc.input)\n\t\troundTripObj := ToDynamicConfigBlob(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigEntryArray(t *testing.T) {\n\ttestCase := []struct {\n\t\tdesc  string\n\t\tinput []*types.DynamicConfigEntry\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: []*types.DynamicConfigEntry{&testdata.DynamicConfigEntry},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.DynamicConfigEntry{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCase {\n\t\tthriftObj := FromDynamicConfigEntryArray(tc.input)\n\t\troundTripObj := ToDynamicConfigEntryArray(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigEntry(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DynamicConfigEntry\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.DynamicConfigEntry,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.DynamicConfigEntry{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigEntry(tc.input)\n\t\troundTripObj := ToDynamicConfigEntry(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigValueArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.DynamicConfigValue\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: []*types.DynamicConfigValue{&testdata.DynamicConfigValue},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.DynamicConfigValue{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigValueArray(tc.input)\n\t\troundTripObj := ToDynamicConfigValueArray(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigValue(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DynamicConfigValue\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.DynamicConfigValue,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.DynamicConfigValue{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigValue(tc.input)\n\t\troundTripObj := ToDynamicConfigValue(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigFilterArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.DynamicConfigFilter\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: []*types.DynamicConfigFilter{&testdata.DynamicConfigFilter},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.DynamicConfigFilter{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigFilterArray(tc.input)\n\t\troundTripObj := ToDynamicConfigFilterArray(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestDynamicConfigFilter(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DynamicConfigFilter\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.DynamicConfigFilter,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.DynamicConfigFilter{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromDynamicConfigFilter(tc.input)\n\t\troundTripObj := ToDynamicConfigFilter(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/errors.go",
    "content": "// Copyright (c) 2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/common/types/mapper/errorutils\"\n)\n\n// FromError convert error to Thrift type if it comes as its internal equivalent\nfunc FromError(err error) error {\n\tif err == nil {\n\t\treturn nil\n\t}\n\n\tvar (\n\t\tok       bool\n\t\ttypedErr error\n\t)\n\tif ok, typedErr = errorutils.ConvertError(err, FromAccessDeniedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromBadRequestError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromCancellationAlreadyRequestedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromClientVersionNotSupportedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromFeatureNotEnabledError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromCurrentBranchChangedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromDomainAlreadyExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromDomainNotActiveError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromEntityNotExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromWorkflowExecutionAlreadyCompletedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromInternalDataInconsistencyError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromInternalServiceError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromLimitExceededError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromQueryFailedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromRemoteSyncMatchedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromRetryTaskV2Error); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromServiceBusyError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromWorkflowExecutionAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromShardOwnershipLostError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromEventAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromStickyWorkerUnavailableError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, FromTaskListNotOwnedByHostError); ok {\n\t\treturn typedErr\n\t}\n\n\treturn err\n}\n\n// ToError convert error to internal type if it comes as its thrift equivalent\nfunc ToError(err error) error {\n\tif err == nil {\n\t\treturn nil\n\t}\n\n\tvar (\n\t\tok       bool\n\t\ttypedErr error\n\t)\n\tif ok, typedErr = errorutils.ConvertError(err, ToAccessDeniedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToBadRequestError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToCancellationAlreadyRequestedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToClientVersionNotSupportedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToFeatureNotEnabledError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToCurrentBranchChangedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToDomainAlreadyExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToDomainNotActiveError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToEntityNotExistsError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToWorkflowExecutionAlreadyCompletedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToInternalDataInconsistencyError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToInternalServiceError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToLimitExceededError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToQueryFailedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToRemoteSyncMatchedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToRetryTaskV2Error); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToServiceBusyError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToWorkflowExecutionAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToShardOwnershipLostError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToEventAlreadyStartedError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToStickyWorkerUnavailableError); ok {\n\t\treturn typedErr\n\t} else if ok, typedErr = errorutils.ConvertError(err, ToTaskListNotOwnedByHostError); ok {\n\t\treturn typedErr\n\t}\n\n\treturn err\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/errors_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"errors\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestErrors(t *testing.T) {\n\tfor _, err := range testdata.Errors {\n\t\tname := reflect.TypeOf(err).Elem().Name()\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Test that the mappings does not lose information\n\t\t\tassert.Equal(t, err, ToError(FromError(err)))\n\t\t})\n\t}\n}\n\nfunc TestNilMapsToNil(t *testing.T) {\n\tassert.Nil(t, FromError(nil))\n\tassert.Nil(t, ToError(nil))\n}\n\nfunc TestFromUnknownErrorMapsToItself(t *testing.T) {\n\terr := errors.New(\"unknown error\")\n\tassert.Equal(t, err, FromError(err))\n}\n\nfunc TestToUnknownErrorMapsToItself(t *testing.T) {\n\terr := yarpcerrors.DeadlineExceededErrorf(\"timeout\")\n\tassert.Equal(t, err, ToError(err))\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/health.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/health\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// FromHealthStatus converts internal HealthStatus type to thrift\nfunc FromHealthStatus(t *types.HealthStatus) *health.HealthStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &health.HealthStatus{\n\t\tOk:  t.Ok,\n\t\tMsg: &t.Msg,\n\t}\n}\n\n// ToHealthStatus converts thrift HealthStatus type to internal\nfunc ToHealthStatus(t *health.HealthStatus) *types.HealthStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HealthStatus{\n\t\tOk:  t.Ok,\n\t\tMsg: t.GetMsg(),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/helpers.go",
    "content": "// Copyright (c) 2017-2022 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\nfunc timeToNano(t *time.Time) *int64 {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn common.Int64Ptr(t.UnixNano())\n}\n\nfunc nanoToTime(nanos *int64) *time.Time {\n\tif nanos == nil {\n\t\treturn nil\n\t}\n\n\tresult := time.Unix(0, *nanos)\n\treturn &result\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/helpers_test.go",
    "content": "// Copyright (c) 2017-2022 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestTimeToNano(t *testing.T) {\n\tunixTime := time.Unix(1, 1)\n\tresult := timeToNano(&unixTime)\n\tassert.Equal(t, int64(1000000001), *result)\n}\n\nfunc TestTimeToNanoNil(t *testing.T) {\n\tresult := timeToNano(nil)\n\tassert.Nil(t, result)\n}\n\nfunc TestNanoToTime(t *testing.T) {\n\tnanos := int64(1000000001)\n\tresult := nanoToTime(&nanos)\n\tassert.True(t, time.Unix(1, 1).Equal(*result))\n}\n\nfunc TestNanoToTimeNil(t *testing.T) {\n\tresult := nanoToTime(nil)\n\tassert.Nil(t, result)\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/history.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tFromHistoryDescribeHistoryHostRequest                = FromAdminDescribeHistoryHostRequest\n\tToHistoryDescribeHistoryHostRequest                  = ToAdminDescribeHistoryHostRequest\n\tFromHistoryDescribeHistoryHostResponse               = FromAdminDescribeHistoryHostResponse\n\tToHistoryDescribeHistoryHostResponse                 = ToAdminDescribeHistoryHostResponse\n\tFromHistoryCloseShardRequest                         = FromAdminCloseShardRequest\n\tToHistoryCloseShardRequest                           = ToAdminCloseShardRequest\n\tFromHistoryDescribeQueueRequest                      = FromAdminDescribeQueueRequest\n\tToHistoryDescribeQueueRequest                        = ToAdminDescribeQueueRequest\n\tFromHistoryDescribeQueueResponse                     = FromAdminDescribeQueueResponse\n\tToHistoryDescribeQueueResponse                       = ToAdminDescribeQueueResponse\n\tFromHistoryDescribeWorkflowExecutionResponse         = FromDescribeWorkflowExecutionResponse\n\tToHistoryDescribeWorkflowExecutionResponse           = ToDescribeWorkflowExecutionResponse\n\tFromHistoryGetCrossClusterTasksRequest               = FromAdminGetCrossClusterTasksRequest\n\tToHistoryGetCrossClusterTasksRequest                 = ToAdminGetCrossClusterTasksRequest\n\tFromHistoryGetCrossClusterTasksResponse              = FromAdminGetCrossClusterTasksResponse\n\tToHistoryGetCrossClusterTasksResponse                = ToAdminGetCrossClusterTasksResponse\n\tFromHistoryGetDLQReplicationMessagesRequest          = FromAdminGetDLQReplicationMessagesRequest\n\tToHistoryGetDLQReplicationMessagesRequest            = ToAdminGetDLQReplicationMessagesRequest\n\tFromHistoryGetDLQReplicationMessagesResponse         = FromAdminGetDLQReplicationMessagesResponse\n\tToHistoryGetDLQReplicationMessagesResponse           = ToAdminGetDLQReplicationMessagesResponse\n\tFromHistoryGetFailoverInfoRequest                    = FromGetFailoverInfoRequest\n\tToHistoryGetFailoverInfoRequest                      = ToGetFailoverInfoRequest\n\tFromHistoryGetFailoverInfoResponse                   = FromGetFailoverInfoResponse\n\tToHistoryGetFailoverInfoResponse                     = ToGetFailoverInfoResponse\n\tFromHistoryGetReplicationMessagesRequest             = FromAdminGetReplicationMessagesRequest\n\tToHistoryGetReplicationMessagesRequest               = ToAdminGetReplicationMessagesRequest\n\tFromHistoryGetReplicationMessagesResponse            = FromAdminGetReplicationMessagesResponse\n\tToHistoryGetReplicationMessagesResponse              = ToAdminGetReplicationMessagesResponse\n\tFromHistoryMergeDLQMessagesRequest                   = FromAdminMergeDLQMessagesRequest\n\tToHistoryMergeDLQMessagesRequest                     = ToAdminMergeDLQMessagesRequest\n\tFromHistoryMergeDLQMessagesResponse                  = FromAdminMergeDLQMessagesResponse\n\tToHistoryMergeDLQMessagesResponse                    = ToAdminMergeDLQMessagesResponse\n\tFromHistoryPurgeDLQMessagesRequest                   = FromAdminPurgeDLQMessagesRequest\n\tToHistoryPurgeDLQMessagesRequest                     = ToAdminPurgeDLQMessagesRequest\n\tFromHistoryReadDLQMessagesRequest                    = FromAdminReadDLQMessagesRequest\n\tToHistoryReadDLQMessagesRequest                      = ToAdminReadDLQMessagesRequest\n\tFromHistoryReadDLQMessagesResponse                   = FromAdminReadDLQMessagesResponse\n\tToHistoryReadDLQMessagesResponse                     = ToAdminReadDLQMessagesResponse\n\tFromHistoryRecordActivityTaskHeartbeatResponse       = FromRecordActivityTaskHeartbeatResponse\n\tToHistoryRecordActivityTaskHeartbeatResponse         = ToRecordActivityTaskHeartbeatResponse\n\tFromHistoryRecordActivityTaskStartedRequest          = FromRecordActivityTaskStartedRequest\n\tToHistoryRecordActivityTaskStartedRequest            = ToRecordActivityTaskStartedRequest\n\tFromHistoryRecordActivityTaskStartedResponse         = FromRecordActivityTaskStartedResponse\n\tToHistoryRecordActivityTaskStartedResponse           = ToRecordActivityTaskStartedResponse\n\tFromHistoryRecordChildExecutionCompletedRequest      = FromRecordChildExecutionCompletedRequest\n\tToHistoryRecordChildExecutionCompletedRequest        = ToRecordChildExecutionCompletedRequest\n\tFromHistoryRecordDecisionTaskStartedRequest          = FromRecordDecisionTaskStartedRequest\n\tToHistoryRecordDecisionTaskStartedRequest            = ToRecordDecisionTaskStartedRequest\n\tFromHistoryRecordDecisionTaskStartedResponse         = FromRecordDecisionTaskStartedResponse\n\tToHistoryRecordDecisionTaskStartedResponse           = ToRecordDecisionTaskStartedResponse\n\tFromHistoryRemoveTaskRequest                         = FromAdminRemoveTaskRequest\n\tToHistoryRemoveTaskRequest                           = ToAdminRemoveTaskRequest\n\tFromHistoryResetQueueRequest                         = FromAdminResetQueueRequest\n\tToHistoryResetQueueRequest                           = ToAdminResetQueueRequest\n\tFromHistoryResetWorkflowExecutionResponse            = FromResetWorkflowExecutionResponse\n\tToHistoryResetWorkflowExecutionResponse              = ToResetWorkflowExecutionResponse\n\tFromHistoryRespondCrossClusterTasksCompletedRequest  = FromAdminRespondCrossClusterTasksCompletedRequest\n\tToHistoryRespondCrossClusterTasksCompletedRequest    = ToAdminRespondCrossClusterTasksCompletedRequest\n\tFromHistoryRespondCrossClusterTasksCompletedResponse = FromAdminRespondCrossClusterTasksCompletedResponse\n\tToHistoryRespondCrossClusterTasksCompletedResponse   = ToAdminRespondCrossClusterTasksCompletedResponse\n\tFromHistorySignalWithStartWorkflowExecutionResponse  = FromStartWorkflowExecutionResponse\n\tToHistorySignalWithStartWorkflowExecutionResponse    = ToStartWorkflowExecutionResponse\n\tFromHistoryStartWorkflowExecutionResponse            = FromStartWorkflowExecutionResponse\n\tToHistoryStartWorkflowExecutionResponse              = ToStartWorkflowExecutionResponse\n)\n\n// FromHistoryDescribeMutableStateRequest converts internal DescribeMutableStateRequest type to thrift\nfunc FromHistoryDescribeMutableStateRequest(t *types.DescribeMutableStateRequest) *history.DescribeMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.DescribeMutableStateRequest{\n\t\tDomainUUID: &t.DomainUUID,\n\t\tExecution:  FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToHistoryDescribeMutableStateRequest converts thrift DescribeMutableStateRequest type to internal\nfunc ToHistoryDescribeMutableStateRequest(t *history.DescribeMutableStateRequest) *types.DescribeMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeMutableStateRequest{\n\t\tDomainUUID: t.GetDomainUUID(),\n\t\tExecution:  ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromHistoryDescribeMutableStateResponse converts internal DescribeMutableStateResponse type to thrift\nfunc FromHistoryDescribeMutableStateResponse(t *types.DescribeMutableStateResponse) *history.DescribeMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.DescribeMutableStateResponse{\n\t\tMutableStateInCache:    &t.MutableStateInCache,\n\t\tMutableStateInDatabase: &t.MutableStateInDatabase,\n\t}\n}\n\n// ToHistoryDescribeMutableStateResponse converts thrift DescribeMutableStateResponse type to internal\nfunc ToHistoryDescribeMutableStateResponse(t *history.DescribeMutableStateResponse) *types.DescribeMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeMutableStateResponse{\n\t\tMutableStateInCache:    t.GetMutableStateInCache(),\n\t\tMutableStateInDatabase: t.GetMutableStateInDatabase(),\n\t}\n}\n\n// FromHistoryDescribeWorkflowExecutionRequest converts internal DescribeWorkflowExecutionRequest type to thrift\nfunc FromHistoryDescribeWorkflowExecutionRequest(t *types.HistoryDescribeWorkflowExecutionRequest) *history.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.DescribeWorkflowExecutionRequest{\n\t\tDomainUUID: &t.DomainUUID,\n\t\tRequest:    FromDescribeWorkflowExecutionRequest(t.Request),\n\t}\n}\n\n// ToHistoryDescribeWorkflowExecutionRequest converts thrift DescribeWorkflowExecutionRequest type to internal\nfunc ToHistoryDescribeWorkflowExecutionRequest(t *history.DescribeWorkflowExecutionRequest) *types.HistoryDescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryDescribeWorkflowExecutionRequest{\n\t\tDomainUUID: t.GetDomainUUID(),\n\t\tRequest:    ToDescribeWorkflowExecutionRequest(t.Request),\n\t}\n}\n\n// FromDomainFilter converts internal DomainFilter type to thrift\nfunc FromDomainFilter(t *types.DomainFilter) *history.DomainFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.DomainFilter{\n\t\tDomainIDs:    t.DomainIDs,\n\t\tReverseMatch: &t.ReverseMatch,\n\t}\n}\n\n// ToDomainFilter converts thrift DomainFilter type to internal\nfunc ToDomainFilter(t *history.DomainFilter) *types.DomainFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainFilter{\n\t\tDomainIDs:    t.DomainIDs,\n\t\tReverseMatch: t.GetReverseMatch(),\n\t}\n}\n\n// FromEventAlreadyStartedError converts internal EventAlreadyStartedError type to thrift\nfunc FromEventAlreadyStartedError(t *types.EventAlreadyStartedError) *history.EventAlreadyStartedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.EventAlreadyStartedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToEventAlreadyStartedError converts thrift EventAlreadyStartedError type to internal\nfunc ToEventAlreadyStartedError(t *history.EventAlreadyStartedError) *types.EventAlreadyStartedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.EventAlreadyStartedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromFailoverMarkerToken converts internal FailoverMarkerToken type to thrift\nfunc FromFailoverMarkerToken(t *types.FailoverMarkerToken) *history.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.FailoverMarkerToken{\n\t\tShardIDs:       t.ShardIDs,\n\t\tFailoverMarker: FromFailoverMarkerAttributes(t.FailoverMarker),\n\t}\n}\n\n// ToFailoverMarkerToken converts thrift FailoverMarkerToken type to internal\nfunc ToFailoverMarkerToken(t *history.FailoverMarkerToken) *types.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverMarkerToken{\n\t\tShardIDs:       t.ShardIDs,\n\t\tFailoverMarker: ToFailoverMarkerAttributes(t.FailoverMarker),\n\t}\n}\n\n// FromHistoryGetMutableStateRequest converts internal GetMutableStateRequest type to thrift\nfunc FromHistoryGetMutableStateRequest(t *types.GetMutableStateRequest) *history.GetMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.GetMutableStateRequest{\n\t\tDomainUUID:          &t.DomainUUID,\n\t\tExecution:           FromWorkflowExecution(t.Execution),\n\t\tExpectedNextEventId: &t.ExpectedNextEventID,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t\tVersionHistoryItem:  FromVersionHistoryItem(t.VersionHistoryItem),\n\t}\n}\n\n// ToHistoryGetMutableStateRequest converts thrift GetMutableStateRequest type to internal\nfunc ToHistoryGetMutableStateRequest(t *history.GetMutableStateRequest) *types.GetMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetMutableStateRequest{\n\t\tDomainUUID:          t.GetDomainUUID(),\n\t\tExecution:           ToWorkflowExecution(t.Execution),\n\t\tExpectedNextEventID: t.GetExpectedNextEventId(),\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t\tVersionHistoryItem:  ToVersionHistoryItem(t.VersionHistoryItem),\n\t}\n}\n\n// FromHistoryGetMutableStateResponse converts internal GetMutableStateResponse type to thrift\nfunc FromHistoryGetMutableStateResponse(t *types.GetMutableStateResponse) *history.GetMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.GetMutableStateResponse{\n\t\tExecution:                            FromWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         FromWorkflowType(t.WorkflowType),\n\t\tNextEventId:                          &t.NextEventID,\n\t\tPreviousStartedEventId:               t.PreviousStartedEventID,\n\t\tLastFirstEventId:                     &t.LastFirstEventID,\n\t\tTaskList:                             FromTaskList(t.TaskList),\n\t\tStickyTaskList:                       FromTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 &t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 &t.ClientFeatureVersion,\n\t\tClientImpl:                           &t.ClientImpl,\n\t\tIsWorkflowRunning:                    &t.IsWorkflowRunning,\n\t\tStickyTaskListScheduleToStartTimeout: t.StickyTaskListScheduleToStartTimeout,\n\t\tEventStoreVersion:                    &t.EventStoreVersion,\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tWorkflowState:                        t.WorkflowState,\n\t\tWorkflowCloseState:                   t.WorkflowCloseState,\n\t\tVersionHistories:                     FromVersionHistories(t.VersionHistories),\n\t\tIsStickyTaskListEnabled:              &t.IsStickyTaskListEnabled,\n\t\tHistorySize:                          &t.HistorySize,\n\t}\n}\n\n// ToHistoryGetMutableStateResponse converts thrift GetMutableStateResponse type to internal\nfunc ToHistoryGetMutableStateResponse(t *history.GetMutableStateResponse) *types.GetMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetMutableStateResponse{\n\t\tExecution:                            ToWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         ToWorkflowType(t.WorkflowType),\n\t\tNextEventID:                          t.GetNextEventId(),\n\t\tPreviousStartedEventID:               t.PreviousStartedEventId,\n\t\tLastFirstEventID:                     t.GetLastFirstEventId(),\n\t\tTaskList:                             ToTaskList(t.TaskList),\n\t\tStickyTaskList:                       ToTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.GetClientLibraryVersion(),\n\t\tClientFeatureVersion:                 t.GetClientFeatureVersion(),\n\t\tClientImpl:                           t.GetClientImpl(),\n\t\tIsWorkflowRunning:                    t.GetIsWorkflowRunning(),\n\t\tStickyTaskListScheduleToStartTimeout: t.StickyTaskListScheduleToStartTimeout,\n\t\tEventStoreVersion:                    t.GetEventStoreVersion(),\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tWorkflowState:                        t.WorkflowState,\n\t\tWorkflowCloseState:                   t.WorkflowCloseState,\n\t\tVersionHistories:                     ToVersionHistories(t.VersionHistories),\n\t\tIsStickyTaskListEnabled:              t.GetIsStickyTaskListEnabled(),\n\t\tHistorySize:                          t.GetHistorySize(),\n\t}\n}\n\n// FromHistoryNotifyFailoverMarkersRequest converts internal NotifyFailoverMarkersRequest type to thrift\nfunc FromHistoryNotifyFailoverMarkersRequest(t *types.NotifyFailoverMarkersRequest) *history.NotifyFailoverMarkersRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: FromFailoverMarkerTokenArray(t.FailoverMarkerTokens),\n\t}\n}\n\n// ToHistoryNotifyFailoverMarkersRequest converts thrift NotifyFailoverMarkersRequest type to internal\nfunc ToHistoryNotifyFailoverMarkersRequest(t *history.NotifyFailoverMarkersRequest) *types.NotifyFailoverMarkersRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: ToFailoverMarkerTokenArray(t.FailoverMarkerTokens),\n\t}\n}\n\n// FromHistoryParentExecutionInfo converts internal ParentExecutionInfo type to thrift\nfunc FromParentExecutionInfo(t *types.ParentExecutionInfo) *history.ParentExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ParentExecutionInfo{\n\t\tDomainUUID:  &t.DomainUUID,\n\t\tDomain:      &t.Domain,\n\t\tExecution:   FromWorkflowExecution(t.Execution),\n\t\tInitiatedId: &t.InitiatedID,\n\t}\n}\n\n// ToParentExecutionInfo converts thrift ParentExecutionInfo type to internal\nfunc ToParentExecutionInfo(t *history.ParentExecutionInfo) *types.ParentExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ParentExecutionInfo{\n\t\tDomainUUID:  t.GetDomainUUID(),\n\t\tDomain:      t.GetDomain(),\n\t\tExecution:   ToWorkflowExecution(t.Execution),\n\t\tInitiatedID: t.GetInitiatedId(),\n\t}\n}\n\n// FromHistoryPollMutableStateRequest converts internal PollMutableStateRequest type to thrift\nfunc FromHistoryPollMutableStateRequest(t *types.PollMutableStateRequest) *history.PollMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.PollMutableStateRequest{\n\t\tDomainUUID:          &t.DomainUUID,\n\t\tExecution:           FromWorkflowExecution(t.Execution),\n\t\tExpectedNextEventId: &t.ExpectedNextEventID,\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t}\n}\n\n// ToHistoryPollMutableStateRequest converts thrift PollMutableStateRequest type to internal\nfunc ToHistoryPollMutableStateRequest(t *history.PollMutableStateRequest) *types.PollMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollMutableStateRequest{\n\t\tDomainUUID:          t.GetDomainUUID(),\n\t\tExecution:           ToWorkflowExecution(t.Execution),\n\t\tExpectedNextEventID: t.GetExpectedNextEventId(),\n\t\tCurrentBranchToken:  t.CurrentBranchToken,\n\t}\n}\n\n// FromHistoryPollMutableStateResponse converts internal PollMutableStateResponse type to thrift\nfunc FromHistoryPollMutableStateResponse(t *types.PollMutableStateResponse) *history.PollMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.PollMutableStateResponse{\n\t\tExecution:                            FromWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         FromWorkflowType(t.WorkflowType),\n\t\tNextEventId:                          &t.NextEventID,\n\t\tPreviousStartedEventId:               t.PreviousStartedEventID,\n\t\tLastFirstEventId:                     &t.LastFirstEventID,\n\t\tTaskList:                             FromTaskList(t.TaskList),\n\t\tStickyTaskList:                       FromTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 &t.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 &t.ClientFeatureVersion,\n\t\tClientImpl:                           &t.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: t.StickyTaskListScheduleToStartTimeout,\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tVersionHistories:                     FromVersionHistories(t.VersionHistories),\n\t\tWorkflowState:                        t.WorkflowState,\n\t\tWorkflowCloseState:                   t.WorkflowCloseState,\n\t}\n}\n\n// ToHistoryPollMutableStateResponse converts thrift PollMutableStateResponse type to internal\nfunc ToHistoryPollMutableStateResponse(t *history.PollMutableStateResponse) *types.PollMutableStateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollMutableStateResponse{\n\t\tExecution:                            ToWorkflowExecution(t.Execution),\n\t\tWorkflowType:                         ToWorkflowType(t.WorkflowType),\n\t\tNextEventID:                          t.GetNextEventId(),\n\t\tPreviousStartedEventID:               t.PreviousStartedEventId,\n\t\tLastFirstEventID:                     t.GetLastFirstEventId(),\n\t\tTaskList:                             ToTaskList(t.TaskList),\n\t\tStickyTaskList:                       ToTaskList(t.StickyTaskList),\n\t\tClientLibraryVersion:                 t.GetClientLibraryVersion(),\n\t\tClientFeatureVersion:                 t.GetClientFeatureVersion(),\n\t\tClientImpl:                           t.GetClientImpl(),\n\t\tStickyTaskListScheduleToStartTimeout: t.StickyTaskListScheduleToStartTimeout,\n\t\tCurrentBranchToken:                   t.CurrentBranchToken,\n\t\tVersionHistories:                     ToVersionHistories(t.VersionHistories),\n\t\tWorkflowState:                        t.WorkflowState,\n\t\tWorkflowCloseState:                   t.WorkflowCloseState,\n\t}\n}\n\n// FromProcessingQueueState converts internal ProcessingQueueState type to thrift\nfunc FromProcessingQueueState(t *types.ProcessingQueueState) *history.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ProcessingQueueState{\n\t\tLevel:        t.Level,\n\t\tAckLevel:     t.AckLevel,\n\t\tMaxLevel:     t.MaxLevel,\n\t\tDomainFilter: FromDomainFilter(t.DomainFilter),\n\t}\n}\n\n// ToProcessingQueueState converts thrift ProcessingQueueState type to internal\nfunc ToProcessingQueueState(t *history.ProcessingQueueState) *types.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ProcessingQueueState{\n\t\tLevel:        t.Level,\n\t\tAckLevel:     t.AckLevel,\n\t\tMaxLevel:     t.MaxLevel,\n\t\tDomainFilter: ToDomainFilter(t.DomainFilter),\n\t}\n}\n\n// FromProcessingQueueStates converts internal ProcessingQueueStates type to thrift\nfunc FromProcessingQueueStates(t *types.ProcessingQueueStates) *history.ProcessingQueueStates {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ProcessingQueueStates{\n\t\tStatesByCluster: FromProcessingQueueStateArrayMap(t.StatesByCluster),\n\t}\n}\n\n// ToProcessingQueueStates converts thrift ProcessingQueueStates type to internal\nfunc ToProcessingQueueStates(t *history.ProcessingQueueStates) *types.ProcessingQueueStates {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ProcessingQueueStates{\n\t\tStatesByCluster: ToProcessingQueueStateArrayMap(t.StatesByCluster),\n\t}\n}\n\n// FromHistoryQueryWorkflowRequest converts internal QueryWorkflowRequest type to thrift\nfunc FromHistoryQueryWorkflowRequest(t *types.HistoryQueryWorkflowRequest) *history.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.QueryWorkflowRequest{\n\t\tDomainUUID: &t.DomainUUID,\n\t\tRequest:    FromQueryWorkflowRequest(t.Request),\n\t}\n}\n\n// ToHistoryQueryWorkflowRequest converts thrift QueryWorkflowRequest type to internal\nfunc ToHistoryQueryWorkflowRequest(t *history.QueryWorkflowRequest) *types.HistoryQueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: t.GetDomainUUID(),\n\t\tRequest:    ToQueryWorkflowRequest(t.Request),\n\t}\n}\n\n// FromHistoryQueryWorkflowResponse converts internal QueryWorkflowResponse type to thrift\nfunc FromHistoryQueryWorkflowResponse(t *types.HistoryQueryWorkflowResponse) *history.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.QueryWorkflowResponse{\n\t\tResponse: FromQueryWorkflowResponse(t.Response),\n\t}\n}\n\n// ToHistoryQueryWorkflowResponse converts thrift QueryWorkflowResponse type to internal\nfunc ToHistoryQueryWorkflowResponse(t *history.QueryWorkflowResponse) *types.HistoryQueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryQueryWorkflowResponse{\n\t\tResponse: ToQueryWorkflowResponse(t.Response),\n\t}\n}\n\n// FromHistoryReapplyEventsRequest converts internal ReapplyEventsRequest type to thrift\nfunc FromHistoryReapplyEventsRequest(t *types.HistoryReapplyEventsRequest) *history.ReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ReapplyEventsRequest{\n\t\tDomainUUID: &t.DomainUUID,\n\t\tRequest:    FromAdminReapplyEventsRequest(t.Request),\n\t}\n}\n\n// ToHistoryReapplyEventsRequest converts thrift ReapplyEventsRequest type to internal\nfunc ToHistoryReapplyEventsRequest(t *history.ReapplyEventsRequest) *types.HistoryReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryReapplyEventsRequest{\n\t\tDomainUUID: t.GetDomainUUID(),\n\t\tRequest:    ToAdminReapplyEventsRequest(t.Request),\n\t}\n}\n\n// FromHistoryRecordActivityTaskHeartbeatRequest converts internal RecordActivityTaskHeartbeatRequest type to thrift\nfunc FromHistoryRecordActivityTaskHeartbeatRequest(t *types.HistoryRecordActivityTaskHeartbeatRequest) *history.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID:       &t.DomainUUID,\n\t\tHeartbeatRequest: FromRecordActivityTaskHeartbeatRequest(t.HeartbeatRequest),\n\t}\n}\n\n// ToHistoryRecordActivityTaskHeartbeatRequest converts thrift RecordActivityTaskHeartbeatRequest type to internal\nfunc ToHistoryRecordActivityTaskHeartbeatRequest(t *history.RecordActivityTaskHeartbeatRequest) *types.HistoryRecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID:       t.GetDomainUUID(),\n\t\tHeartbeatRequest: ToRecordActivityTaskHeartbeatRequest(t.HeartbeatRequest),\n\t}\n}\n\n// FromRecordActivityTaskStartedRequest converts internal RecordActivityTaskStartedRequest type to thrift\nfunc FromRecordActivityTaskStartedRequest(t *types.RecordActivityTaskStartedRequest) *history.RecordActivityTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        &t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleId:        &t.ScheduleID,\n\t\tTaskId:            &t.TaskID,\n\t\tRequestId:         &t.RequestID,\n\t\tPollRequest:       FromPollForActivityTaskRequest(t.PollRequest),\n\t}\n}\n\n// ToRecordActivityTaskStartedRequest converts thrift RecordActivityTaskStartedRequest type to internal\nfunc ToRecordActivityTaskStartedRequest(t *history.RecordActivityTaskStartedRequest) *types.RecordActivityTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        t.GetDomainUUID(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleID:        t.GetScheduleId(),\n\t\tTaskID:            t.GetTaskId(),\n\t\tRequestID:         t.GetRequestId(),\n\t\tPollRequest:       ToPollForActivityTaskRequest(t.PollRequest),\n\t}\n}\n\n// FromRecordActivityTaskStartedResponse converts internal RecordActivityTaskStartedResponse type to thrift\nfunc FromRecordActivityTaskStartedResponse(t *types.RecordActivityTaskStartedResponse) *history.RecordActivityTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordActivityTaskStartedResponse{\n\t\tScheduledEvent:                  FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tAttempt:                         &t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  &t.WorkflowDomain,\n\t}\n}\n\n// ToRecordActivityTaskStartedResponse converts thrift RecordActivityTaskStartedResponse type to internal\nfunc ToRecordActivityTaskStartedResponse(t *history.RecordActivityTaskStartedResponse) *types.RecordActivityTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskStartedResponse{\n\t\tScheduledEvent:                  ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tAttempt:                         t.GetAttempt(),\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.GetWorkflowDomain(),\n\t}\n}\n\n// FromRecordChildExecutionCompletedRequest converts internal RecordChildExecutionCompletedRequest type to thrift\nfunc FromRecordChildExecutionCompletedRequest(t *types.RecordChildExecutionCompletedRequest) *history.RecordChildExecutionCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordChildExecutionCompletedRequest{\n\t\tDomainUUID:         &t.DomainUUID,\n\t\tWorkflowExecution:  FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedId:        &t.InitiatedID,\n\t\tCompletedExecution: FromWorkflowExecution(t.CompletedExecution),\n\t\tCompletionEvent:    FromHistoryEvent(t.CompletionEvent),\n\t\tStartedId:          &t.StartedID,\n\t}\n}\n\n// ToRecordChildExecutionCompletedRequest converts thrift RecordChildExecutionCompletedRequest type to internal\nfunc ToRecordChildExecutionCompletedRequest(t *history.RecordChildExecutionCompletedRequest) *types.RecordChildExecutionCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordChildExecutionCompletedRequest{\n\t\tDomainUUID:         t.GetDomainUUID(),\n\t\tWorkflowExecution:  ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedID:        t.GetInitiatedId(),\n\t\tCompletedExecution: ToWorkflowExecution(t.CompletedExecution),\n\t\tCompletionEvent:    ToHistoryEvent(t.CompletionEvent),\n\t\tStartedID:          t.GetStartedId(),\n\t}\n}\n\n// FromRecordDecisionTaskStartedRequest converts internal RecordDecisionTaskStartedRequest type to thrift\nfunc FromRecordDecisionTaskStartedRequest(t *types.RecordDecisionTaskStartedRequest) *history.RecordDecisionTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        &t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleId:        &t.ScheduleID,\n\t\tTaskId:            &t.TaskID,\n\t\tRequestId:         &t.RequestID,\n\t\tPollRequest:       FromPollForDecisionTaskRequest(t.PollRequest),\n\t}\n}\n\n// ToRecordDecisionTaskStartedRequest converts thrift RecordDecisionTaskStartedRequest type to internal\nfunc ToRecordDecisionTaskStartedRequest(t *history.RecordDecisionTaskStartedRequest) *types.RecordDecisionTaskStartedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        t.GetDomainUUID(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tScheduleID:        t.GetScheduleId(),\n\t\tTaskID:            t.GetTaskId(),\n\t\tRequestID:         t.GetRequestId(),\n\t\tPollRequest:       ToPollForDecisionTaskRequest(t.PollRequest),\n\t}\n}\n\n// FromRecordDecisionTaskStartedResponse converts internal RecordDecisionTaskStartedResponse type to thrift\nfunc FromRecordDecisionTaskStartedResponse(t *types.RecordDecisionTaskStartedResponse) *history.RecordDecisionTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RecordDecisionTaskStartedResponse{\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    t.PreviousStartedEventID,\n\t\tScheduledEventId:          &t.ScheduledEventID,\n\t\tStartedEventId:            &t.StartedEventID,\n\t\tNextEventId:               &t.NextEventID,\n\t\tAttempt:                   &t.Attempt,\n\t\tStickyExecutionEnabled:    &t.StickyExecutionEnabled,\n\t\tDecisionInfo:              FromTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         &t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tHistorySize:               &t.HistorySize,\n\t}\n}\n\n// ToRecordDecisionTaskStartedResponse converts thrift RecordDecisionTaskStartedResponse type to internal\nfunc ToRecordDecisionTaskStartedResponse(t *history.RecordDecisionTaskStartedResponse) *types.RecordDecisionTaskStartedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordDecisionTaskStartedResponse{\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    t.PreviousStartedEventId,\n\t\tScheduledEventID:          t.GetScheduledEventId(),\n\t\tStartedEventID:            t.GetStartedEventId(),\n\t\tNextEventID:               t.GetNextEventId(),\n\t\tAttempt:                   t.GetAttempt(),\n\t\tStickyExecutionEnabled:    t.GetStickyExecutionEnabled(),\n\t\tDecisionInfo:              ToTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.GetEventStoreVersion(),\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tHistorySize:               t.GetHistorySize(),\n\t}\n}\n\n// FromHistoryRefreshWorkflowTasksRequest converts internal RefreshWorkflowTasksRequest type to thrift\nfunc FromHistoryRefreshWorkflowTasksRequest(t *types.HistoryRefreshWorkflowTasksRequest) *history.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RefreshWorkflowTasksRequest{\n\t\tDomainUIID: &t.DomainUIID,\n\t\tRequest:    FromAdminRefreshWorkflowTasksRequest(t.Request),\n\t}\n}\n\n// ToHistoryRefreshWorkflowTasksRequest converts thrift RefreshWorkflowTasksRequest type to internal\nfunc ToHistoryRefreshWorkflowTasksRequest(t *history.RefreshWorkflowTasksRequest) *types.HistoryRefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: t.GetDomainUIID(),\n\t\tRequest:    ToAdminRefreshWorkflowTasksRequest(t.Request),\n\t}\n}\n\n// FromHistoryRemoveSignalMutableStateRequest converts internal RemoveSignalMutableStateRequest type to thrift\nfunc FromHistoryRemoveSignalMutableStateRequest(t *types.RemoveSignalMutableStateRequest) *history.RemoveSignalMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RemoveSignalMutableStateRequest{\n\t\tDomainUUID:        &t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tRequestId:         &t.RequestID,\n\t}\n}\n\n// ToHistoryRemoveSignalMutableStateRequest converts thrift RemoveSignalMutableStateRequest type to internal\nfunc ToHistoryRemoveSignalMutableStateRequest(t *history.RemoveSignalMutableStateRequest) *types.RemoveSignalMutableStateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID:        t.GetDomainUUID(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tRequestID:         t.GetRequestId(),\n\t}\n}\n\n// FromHistoryReplicateEventsV2Request converts internal ReplicateEventsV2Request type to thrift\nfunc FromHistoryReplicateEventsV2Request(t *types.ReplicateEventsV2Request) *history.ReplicateEventsV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ReplicateEventsV2Request{\n\t\tDomainUUID:          &t.DomainUUID,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tVersionHistoryItems: FromVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              FromDataBlob(t.Events),\n\t\tNewRunEvents:        FromDataBlob(t.NewRunEvents),\n\t}\n}\n\n// ToHistoryReplicateEventsV2Request converts thrift ReplicateEventsV2Request type to internal\nfunc ToHistoryReplicateEventsV2Request(t *history.ReplicateEventsV2Request) *types.ReplicateEventsV2Request {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicateEventsV2Request{\n\t\tDomainUUID:          t.GetDomainUUID(),\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tVersionHistoryItems: ToVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              ToDataBlob(t.Events),\n\t\tNewRunEvents:        ToDataBlob(t.NewRunEvents),\n\t}\n}\n\n// FromHistoryRequestCancelWorkflowExecutionRequest converts internal RequestCancelWorkflowExecutionRequest type to thrift\nfunc FromHistoryRequestCancelWorkflowExecutionRequest(t *types.HistoryRequestCancelWorkflowExecutionRequest) *history.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID:                &t.DomainUUID,\n\t\tCancelRequest:             FromRequestCancelWorkflowExecutionRequest(t.CancelRequest),\n\t\tExternalInitiatedEventId:  t.ExternalInitiatedEventID,\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToHistoryRequestCancelWorkflowExecutionRequest converts thrift RequestCancelWorkflowExecutionRequest type to internal\nfunc ToHistoryRequestCancelWorkflowExecutionRequest(t *history.RequestCancelWorkflowExecutionRequest) *types.HistoryRequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID:                t.GetDomainUUID(),\n\t\tCancelRequest:             ToRequestCancelWorkflowExecutionRequest(t.CancelRequest),\n\t\tExternalInitiatedEventID:  t.ExternalInitiatedEventId,\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromHistoryResetStickyTaskListRequest converts internal ResetStickyTaskListRequest type to thrift\nfunc FromHistoryResetStickyTaskListRequest(t *types.HistoryResetStickyTaskListRequest) *history.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ResetStickyTaskListRequest{\n\t\tDomainUUID: &t.DomainUUID,\n\t\tExecution:  FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToHistoryResetStickyTaskListRequest converts thrift ResetStickyTaskListRequest type to internal\nfunc ToHistoryResetStickyTaskListRequest(t *history.ResetStickyTaskListRequest) *types.HistoryResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: t.GetDomainUUID(),\n\t\tExecution:  ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromHistoryResetStickyTaskListResponse converts internal ResetStickyTaskListResponse type to thrift\nfunc FromHistoryResetStickyTaskListResponse(t *types.HistoryResetStickyTaskListResponse) *history.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ResetStickyTaskListResponse{}\n}\n\n// ToHistoryResetStickyTaskListResponse converts thrift ResetStickyTaskListResponse type to internal\nfunc ToHistoryResetStickyTaskListResponse(t *history.ResetStickyTaskListResponse) *types.HistoryResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetStickyTaskListResponse{}\n}\n\n// FromHistoryResetWorkflowExecutionRequest converts internal ResetWorkflowExecutionRequest type to thrift\nfunc FromHistoryResetWorkflowExecutionRequest(t *types.HistoryResetWorkflowExecutionRequest) *history.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ResetWorkflowExecutionRequest{\n\t\tDomainUUID:   &t.DomainUUID,\n\t\tResetRequest: FromResetWorkflowExecutionRequest(t.ResetRequest),\n\t}\n}\n\n// ToHistoryResetWorkflowExecutionRequest converts thrift ResetWorkflowExecutionRequest type to internal\nfunc ToHistoryResetWorkflowExecutionRequest(t *history.ResetWorkflowExecutionRequest) *types.HistoryResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID:   t.GetDomainUUID(),\n\t\tResetRequest: ToResetWorkflowExecutionRequest(t.ResetRequest),\n\t}\n}\n\n// FromHistoryRespondActivityTaskCanceledRequest converts internal RespondActivityTaskCanceledRequest type to thrift\nfunc FromHistoryRespondActivityTaskCanceledRequest(t *types.HistoryRespondActivityTaskCanceledRequest) *history.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondActivityTaskCanceledRequest{\n\t\tDomainUUID:    &t.DomainUUID,\n\t\tCancelRequest: FromRespondActivityTaskCanceledRequest(t.CancelRequest),\n\t}\n}\n\n// ToHistoryRespondActivityTaskCanceledRequest converts thrift RespondActivityTaskCanceledRequest type to internal\nfunc ToHistoryRespondActivityTaskCanceledRequest(t *history.RespondActivityTaskCanceledRequest) *types.HistoryRespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID:    t.GetDomainUUID(),\n\t\tCancelRequest: ToRespondActivityTaskCanceledRequest(t.CancelRequest),\n\t}\n}\n\n// FromHistoryRespondActivityTaskCompletedRequest converts internal RespondActivityTaskCompletedRequest type to thrift\nfunc FromHistoryRespondActivityTaskCompletedRequest(t *types.HistoryRespondActivityTaskCompletedRequest) *history.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondActivityTaskCompletedRequest{\n\t\tDomainUUID:      &t.DomainUUID,\n\t\tCompleteRequest: FromRespondActivityTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\n// ToHistoryRespondActivityTaskCompletedRequest converts thrift RespondActivityTaskCompletedRequest type to internal\nfunc ToHistoryRespondActivityTaskCompletedRequest(t *history.RespondActivityTaskCompletedRequest) *types.HistoryRespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID:      t.GetDomainUUID(),\n\t\tCompleteRequest: ToRespondActivityTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\n// FromHistoryRespondActivityTaskFailedRequest converts internal RespondActivityTaskFailedRequest type to thrift\nfunc FromHistoryRespondActivityTaskFailedRequest(t *types.HistoryRespondActivityTaskFailedRequest) *history.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondActivityTaskFailedRequest{\n\t\tDomainUUID:    &t.DomainUUID,\n\t\tFailedRequest: FromRespondActivityTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\n// ToHistoryRespondActivityTaskFailedRequest converts thrift RespondActivityTaskFailedRequest type to internal\nfunc ToHistoryRespondActivityTaskFailedRequest(t *history.RespondActivityTaskFailedRequest) *types.HistoryRespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID:    t.GetDomainUUID(),\n\t\tFailedRequest: ToRespondActivityTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\n// FromHistoryRespondDecisionTaskCompletedRequest converts internal RespondDecisionTaskCompletedRequest type to thrift\nfunc FromHistoryRespondDecisionTaskCompletedRequest(t *types.HistoryRespondDecisionTaskCompletedRequest) *history.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondDecisionTaskCompletedRequest{\n\t\tDomainUUID:      &t.DomainUUID,\n\t\tCompleteRequest: FromRespondDecisionTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\n// ToHistoryRespondDecisionTaskCompletedRequest converts thrift RespondDecisionTaskCompletedRequest type to internal\nfunc ToHistoryRespondDecisionTaskCompletedRequest(t *history.RespondDecisionTaskCompletedRequest) *types.HistoryRespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID:      t.GetDomainUUID(),\n\t\tCompleteRequest: ToRespondDecisionTaskCompletedRequest(t.CompleteRequest),\n\t}\n}\n\n// FromHistoryRespondDecisionTaskCompletedResponse converts internal RespondDecisionTaskCompletedResponse type to thrift\nfunc FromHistoryRespondDecisionTaskCompletedResponse(t *types.HistoryRespondDecisionTaskCompletedResponse) *history.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondDecisionTaskCompletedResponse{\n\t\tStartedResponse:             FromRecordDecisionTaskStartedResponse(t.StartedResponse),\n\t\tActivitiesToDispatchLocally: FromActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\n// ToHistoryRespondDecisionTaskCompletedResponse converts thrift RespondDecisionTaskCompletedResponse type to internal\nfunc ToHistoryRespondDecisionTaskCompletedResponse(t *history.RespondDecisionTaskCompletedResponse) *types.HistoryRespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tStartedResponse:             ToRecordDecisionTaskStartedResponse(t.StartedResponse),\n\t\tActivitiesToDispatchLocally: ToActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\n// FromHistoryRespondDecisionTaskFailedRequest converts internal RespondDecisionTaskFailedRequest type to thrift\nfunc FromHistoryRespondDecisionTaskFailedRequest(t *types.HistoryRespondDecisionTaskFailedRequest) *history.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RespondDecisionTaskFailedRequest{\n\t\tDomainUUID:    &t.DomainUUID,\n\t\tFailedRequest: FromRespondDecisionTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\n// ToHistoryRespondDecisionTaskFailedRequest converts thrift RespondDecisionTaskFailedRequest type to internal\nfunc ToHistoryRespondDecisionTaskFailedRequest(t *history.RespondDecisionTaskFailedRequest) *types.HistoryRespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID:    t.GetDomainUUID(),\n\t\tFailedRequest: ToRespondDecisionTaskFailedRequest(t.FailedRequest),\n\t}\n}\n\n// FromHistoryScheduleDecisionTaskRequest converts internal ScheduleDecisionTaskRequest type to thrift\nfunc FromHistoryScheduleDecisionTaskRequest(t *types.ScheduleDecisionTaskRequest) *history.ScheduleDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ScheduleDecisionTaskRequest{\n\t\tDomainUUID:        &t.DomainUUID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tIsFirstDecision:   &t.IsFirstDecision,\n\t}\n}\n\n// ToHistoryScheduleDecisionTaskRequest converts thrift ScheduleDecisionTaskRequest type to internal\nfunc ToHistoryScheduleDecisionTaskRequest(t *history.ScheduleDecisionTaskRequest) *types.ScheduleDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleDecisionTaskRequest{\n\t\tDomainUUID:        t.GetDomainUUID(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tIsFirstDecision:   t.GetIsFirstDecision(),\n\t}\n}\n\n// FromShardOwnershipLostError converts internal ShardOwnershipLostError type to thrift\nfunc FromShardOwnershipLostError(t *types.ShardOwnershipLostError) *history.ShardOwnershipLostError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.ShardOwnershipLostError{\n\t\tMessage: &t.Message,\n\t\tOwner:   &t.Owner,\n\t}\n}\n\n// ToShardOwnershipLostError converts thrift ShardOwnershipLostError type to internal\nfunc ToShardOwnershipLostError(t *history.ShardOwnershipLostError) *types.ShardOwnershipLostError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ShardOwnershipLostError{\n\t\tMessage: t.GetMessage(),\n\t\tOwner:   t.GetOwner(),\n\t}\n}\n\n// FromHistorySignalWithStartWorkflowExecutionRequest converts internal SignalWithStartWorkflowExecutionRequest type to thrift\nfunc FromHistorySignalWithStartWorkflowExecutionRequest(t *types.HistorySignalWithStartWorkflowExecutionRequest) *history.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.SignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID:             &t.DomainUUID,\n\t\tSignalWithStartRequest: FromSignalWithStartWorkflowExecutionRequest(t.SignalWithStartRequest),\n\t\tPartitionConfig:        t.PartitionConfig,\n\t}\n}\n\n// ToHistorySignalWithStartWorkflowExecutionRequest converts thrift SignalWithStartWorkflowExecutionRequest type to internal\nfunc ToHistorySignalWithStartWorkflowExecutionRequest(t *history.SignalWithStartWorkflowExecutionRequest) *types.HistorySignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID:             t.GetDomainUUID(),\n\t\tSignalWithStartRequest: ToSignalWithStartWorkflowExecutionRequest(t.SignalWithStartRequest),\n\t\tPartitionConfig:        t.PartitionConfig,\n\t}\n}\n\n// FromHistorySignalWorkflowExecutionRequest converts internal SignalWorkflowExecutionRequest type to thrift\nfunc FromHistorySignalWorkflowExecutionRequest(t *types.HistorySignalWorkflowExecutionRequest) *history.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.SignalWorkflowExecutionRequest{\n\t\tDomainUUID:                &t.DomainUUID,\n\t\tSignalRequest:             FromSignalWorkflowExecutionRequest(t.SignalRequest),\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToHistorySignalWorkflowExecutionRequest converts thrift SignalWorkflowExecutionRequest type to internal\nfunc ToHistorySignalWorkflowExecutionRequest(t *history.SignalWorkflowExecutionRequest) *types.HistorySignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID:                t.GetDomainUUID(),\n\t\tSignalRequest:             ToSignalWorkflowExecutionRequest(t.SignalRequest),\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromHistoryStartWorkflowExecutionRequest converts internal StartWorkflowExecutionRequest type to thrift\nfunc FromHistoryStartWorkflowExecutionRequest(t *types.HistoryStartWorkflowExecutionRequest) *history.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.StartWorkflowExecutionRequest{\n\t\tDomainUUID:                      &t.DomainUUID,\n\t\tStartRequest:                    FromStartWorkflowExecutionRequest(t.StartRequest),\n\t\tParentExecutionInfo:             FromParentExecutionInfo(t.ParentExecutionInfo),\n\t\tAttempt:                         &t.Attempt,\n\t\tExpirationTimestamp:             t.ExpirationTimestamp,\n\t\tContinueAsNewInitiator:          FromContinueAsNewInitiator(t.ContinueAsNewInitiator),\n\t\tContinuedFailureReason:          t.ContinuedFailureReason,\n\t\tContinuedFailureDetails:         t.ContinuedFailureDetails,\n\t\tLastCompletionResult:            t.LastCompletionResult,\n\t\tFirstDecisionTaskBackoffSeconds: t.FirstDecisionTaskBackoffSeconds,\n\t\tPartitionConfig:                 t.PartitionConfig,\n\t}\n}\n\n// ToHistoryStartWorkflowExecutionRequest converts thrift StartWorkflowExecutionRequest type to internal\nfunc ToHistoryStartWorkflowExecutionRequest(t *history.StartWorkflowExecutionRequest) *types.HistoryStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID:                      t.GetDomainUUID(),\n\t\tStartRequest:                    ToStartWorkflowExecutionRequest(t.StartRequest),\n\t\tParentExecutionInfo:             ToParentExecutionInfo(t.ParentExecutionInfo),\n\t\tAttempt:                         t.GetAttempt(),\n\t\tExpirationTimestamp:             t.ExpirationTimestamp,\n\t\tContinueAsNewInitiator:          ToContinueAsNewInitiator(t.ContinueAsNewInitiator),\n\t\tContinuedFailureReason:          t.ContinuedFailureReason,\n\t\tContinuedFailureDetails:         t.ContinuedFailureDetails,\n\t\tLastCompletionResult:            t.LastCompletionResult,\n\t\tFirstDecisionTaskBackoffSeconds: t.FirstDecisionTaskBackoffSeconds,\n\t\tPartitionConfig:                 t.PartitionConfig,\n\t}\n}\n\n// FromHistorySyncActivityRequest converts internal SyncActivityRequest type to thrift\nfunc FromHistorySyncActivityRequest(t *types.SyncActivityRequest) *history.SyncActivityRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.SyncActivityRequest{\n\t\tDomainId:           &t.DomainID,\n\t\tWorkflowId:         &t.WorkflowID,\n\t\tRunId:              &t.RunID,\n\t\tVersion:            &t.Version,\n\t\tScheduledId:        &t.ScheduledID,\n\t\tScheduledTime:      t.ScheduledTime,\n\t\tStartedId:          &t.StartedID,\n\t\tStartedTime:        t.StartedTime,\n\t\tLastHeartbeatTime:  t.LastHeartbeatTime,\n\t\tDetails:            t.Details,\n\t\tAttempt:            &t.Attempt,\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastWorkerIdentity: &t.LastWorkerIdentity,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t\tVersionHistory:     FromVersionHistory(t.VersionHistory),\n\t}\n}\n\n// ToHistorySyncActivityRequest converts thrift SyncActivityRequest type to internal\nfunc ToHistorySyncActivityRequest(t *history.SyncActivityRequest) *types.SyncActivityRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncActivityRequest{\n\t\tDomainID:           t.GetDomainId(),\n\t\tWorkflowID:         t.GetWorkflowId(),\n\t\tRunID:              t.GetRunId(),\n\t\tVersion:            t.GetVersion(),\n\t\tScheduledID:        t.GetScheduledId(),\n\t\tScheduledTime:      t.ScheduledTime,\n\t\tStartedID:          t.GetStartedId(),\n\t\tStartedTime:        t.StartedTime,\n\t\tLastHeartbeatTime:  t.LastHeartbeatTime,\n\t\tDetails:            t.Details,\n\t\tAttempt:            t.GetAttempt(),\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastWorkerIdentity: t.GetLastWorkerIdentity(),\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t\tVersionHistory:     ToVersionHistory(t.VersionHistory),\n\t}\n}\n\n// FromHistorySyncShardStatusRequest converts internal SyncShardStatusRequest type to thrift\nfunc FromHistorySyncShardStatusRequest(t *types.SyncShardStatusRequest) *history.SyncShardStatusRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.SyncShardStatusRequest{\n\t\tSourceCluster: &t.SourceCluster,\n\t\tShardId:       &t.ShardID,\n\t\tTimestamp:     t.Timestamp,\n\t}\n}\n\n// ToHistorySyncShardStatusRequest converts thrift SyncShardStatusRequest type to internal\nfunc ToHistorySyncShardStatusRequest(t *history.SyncShardStatusRequest) *types.SyncShardStatusRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatusRequest{\n\t\tSourceCluster: t.GetSourceCluster(),\n\t\tShardID:       t.GetShardId(),\n\t\tTimestamp:     t.Timestamp,\n\t}\n}\n\n// FromHistoryTerminateWorkflowExecutionRequest converts internal TerminateWorkflowExecutionRequest type to thrift\nfunc FromHistoryTerminateWorkflowExecutionRequest(t *types.HistoryTerminateWorkflowExecutionRequest) *history.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.TerminateWorkflowExecutionRequest{\n\t\tDomainUUID:                &t.DomainUUID,\n\t\tTerminateRequest:          FromTerminateWorkflowExecutionRequest(t.TerminateRequest),\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToHistoryTerminateWorkflowExecutionRequest converts thrift TerminateWorkflowExecutionRequest type to internal\nfunc ToHistoryTerminateWorkflowExecutionRequest(t *history.TerminateWorkflowExecutionRequest) *types.HistoryTerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryTerminateWorkflowExecutionRequest{\n\t\tDomainUUID:                t.GetDomainUUID(),\n\t\tTerminateRequest:          ToTerminateWorkflowExecutionRequest(t.TerminateRequest),\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tChildWorkflowOnly:         t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromFailoverMarkerTokenArray converts internal FailoverMarkerToken type array to thrift\nfunc FromFailoverMarkerTokenArray(t []*types.FailoverMarkerToken) []*history.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*history.FailoverMarkerToken, len(t))\n\tfor i := range t {\n\t\tv[i] = FromFailoverMarkerToken(t[i])\n\t}\n\treturn v\n}\n\n// ToFailoverMarkerTokenArray converts thrift FailoverMarkerToken type array to internal\nfunc ToFailoverMarkerTokenArray(t []*history.FailoverMarkerToken) []*types.FailoverMarkerToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.FailoverMarkerToken, len(t))\n\tfor i := range t {\n\t\tv[i] = ToFailoverMarkerToken(t[i])\n\t}\n\treturn v\n}\n\n// FromProcessingQueueStateArray converts internal ProcessingQueueState type array to thrift\nfunc FromProcessingQueueStateArray(t []*types.ProcessingQueueState) []*history.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*history.ProcessingQueueState, len(t))\n\tfor i := range t {\n\t\tv[i] = FromProcessingQueueState(t[i])\n\t}\n\treturn v\n}\n\n// ToProcessingQueueStateArray converts thrift ProcessingQueueState type array to internal\nfunc ToProcessingQueueStateArray(t []*history.ProcessingQueueState) []*types.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ProcessingQueueState, len(t))\n\tfor i := range t {\n\t\tv[i] = ToProcessingQueueState(t[i])\n\t}\n\treturn v\n}\n\n// FromProcessingQueueStateArrayMap converts internal ProcessingQueueState array map to thrift\nfunc FromProcessingQueueStateArrayMap(t map[string][]*types.ProcessingQueueState) map[string][]*history.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string][]*history.ProcessingQueueState, len(t))\n\tfor key := range t {\n\t\tv[key] = FromProcessingQueueStateArray(t[key])\n\t}\n\treturn v\n}\n\n// ToProcessingQueueStateArrayMap converts thrift ProcessingQueueState array map to internal\nfunc ToProcessingQueueStateArrayMap(t map[string][]*history.ProcessingQueueState) map[string][]*types.ProcessingQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string][]*types.ProcessingQueueState, len(t))\n\tfor key := range t {\n\t\tv[key] = ToProcessingQueueStateArray(t[key])\n\t}\n\treturn v\n}\n\n// FromGetFailoverInfoRequest converts internal GetFailoverInfoRequest type to thrift\nfunc FromGetFailoverInfoRequest(t *types.GetFailoverInfoRequest) *history.GetFailoverInfoRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.GetFailoverInfoRequest{\n\t\tDomainID: &t.DomainID,\n\t}\n}\n\n// ToGetFailoverInfoRequest converts thrift GetFailoverInfoRequest type to internal\nfunc ToGetFailoverInfoRequest(t *history.GetFailoverInfoRequest) *types.GetFailoverInfoRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetFailoverInfoRequest{\n\t\tDomainID: t.GetDomainID(),\n\t}\n}\n\n// FromGetFailoverInfoResponse converts internal GetFailoverInfoRequest type to thrift\nfunc FromGetFailoverInfoResponse(t *types.GetFailoverInfoResponse) *history.GetFailoverInfoResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.GetFailoverInfoResponse{\n\t\tCompletedShardCount: &t.CompletedShardCount,\n\t\tPendingShards:       t.GetPendingShards(),\n\t}\n}\n\n// ToGetFailoverInfoResponse converts thrift GetFailoverInfoResponse type to internal\nfunc ToGetFailoverInfoResponse(t *history.GetFailoverInfoResponse) *types.GetFailoverInfoResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetFailoverInfoResponse{\n\t\tCompletedShardCount: t.GetCompletedShardCount(),\n\t\tPendingShards:       t.GetPendingShards(),\n\t}\n}\n\nfunc FromHistoryRatelimitUpdateRequest(t *types.RatelimitUpdateRequest) *history.RatelimitUpdateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RatelimitUpdateRequest{\n\t\tData: FromAny(t.Any),\n\t}\n}\nfunc ToHistoryRatelimitUpdateRequest(t *history.RatelimitUpdateRequest) *types.RatelimitUpdateRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RatelimitUpdateRequest{\n\t\tAny: ToAny(t.Data),\n\t}\n}\nfunc FromHistoryRatelimitUpdateResponse(t *types.RatelimitUpdateResponse) *history.RatelimitUpdateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &history.RatelimitUpdateResponse{\n\t\tData: FromAny(t.Any),\n\t}\n}\nfunc ToHistoryRatelimitUpdateResponse(t *history.RatelimitUpdateResponse) *types.RatelimitUpdateResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RatelimitUpdateResponse{\n\t\tAny: ToAny(t.Data),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/history_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/google/go-cmp/cmp/cmpopts\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestHistoryTerminateWorkflowExecutionRequestConversion(t *testing.T) {\n\tfor _, item := range []*types.HistoryTerminateWorkflowExecutionRequest{nil, {}, &testdata.HistoryTerminateWorkflowExecutionRequest} {\n\t\tassert.Equal(t, item, ToHistoryTerminateWorkflowExecutionRequest(FromHistoryTerminateWorkflowExecutionRequest(item)))\n\t}\n}\n\nfunc TestDescribeMutableStateRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeMutableStateRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryDescribeMutableStateRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryDescribeMutableStateRequest(original)\n\t\troundTripObj := ToHistoryDescribeMutableStateRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeMutableStateResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeMutableStateResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryDescribeMutableStateResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryDescribeMutableStateResponse(original)\n\t\troundTripObj := ToHistoryDescribeMutableStateResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryDescribeWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryDescribeWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryDescribeWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryDescribeWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistoryDescribeWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainFilterConversion(t *testing.T) {\n\ttestCases := []*types.DomainFilter{\n\t\tnil,\n\t\t{},\n\t\t{DomainIDs: []string{\"test\"}, ReverseMatch: true},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainFilter(original)\n\t\troundTripObj := ToDomainFilter(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestEventAlreadyStartedErrorConversion(t *testing.T) {\n\ttestCases := []*types.EventAlreadyStartedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.EventAlreadyStartedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromEventAlreadyStartedError(original)\n\t\troundTripObj := ToEventAlreadyStartedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestFailoverMarkerTokenConversion(t *testing.T) {\n\ttestCases := []*types.FailoverMarkerToken{\n\t\tnil,\n\t\t{},\n\t\t&testdata.FailoverMarkerToken,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromFailoverMarkerToken(original)\n\t\troundTripObj := ToFailoverMarkerToken(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetMutableStateRequestConversion(t *testing.T) {\n\ttestCases := []*types.GetMutableStateRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryGetMutableStateRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryGetMutableStateRequest(original)\n\t\troundTripObj := ToHistoryGetMutableStateRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetMutableStateResponseConversion(t *testing.T) {\n\ttestCases := []*types.GetMutableStateResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryGetMutableStateResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryGetMutableStateResponse(original)\n\t\troundTripObj := ToHistoryGetMutableStateResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestNotifyFailoverMarkersRequestConversion(t *testing.T) {\n\ttestCases := []*types.NotifyFailoverMarkersRequest{\n\t\tnil,\n\t\t{},\n\t\t{FailoverMarkerTokens: []*types.FailoverMarkerToken{&testdata.FailoverMarkerToken}},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryNotifyFailoverMarkersRequest(original)\n\t\troundTripObj := ToHistoryNotifyFailoverMarkersRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestParentExecutionInfoConversion(t *testing.T) {\n\ttestCases := []*types.ParentExecutionInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ParentExecutionInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromParentExecutionInfo(original)\n\t\troundTripObj := ToParentExecutionInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollMutableStateRequestConversion(t *testing.T) {\n\ttestCases := []*types.PollMutableStateRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryPollMutableStateRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryPollMutableStateRequest(original)\n\t\troundTripObj := ToHistoryPollMutableStateRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollMutableStateResponseConversion(t *testing.T) {\n\ttestCases := []*types.PollMutableStateResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryPollMutableStateResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryPollMutableStateResponse(original)\n\t\troundTripObj := ToHistoryPollMutableStateResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestProcessingQueueStateConversion(t *testing.T) {\n\ttestCases := []*types.ProcessingQueueState{\n\t\tnil,\n\t\t{},\n\t\t{Level: common.Int32Ptr(1), AckLevel: common.Int64Ptr(1), MaxLevel: common.Int64Ptr(1), DomainFilter: &types.DomainFilter{DomainIDs: []string{\"test\"}, ReverseMatch: true}},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromProcessingQueueState(original)\n\t\troundTripObj := ToProcessingQueueState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestProcessingQueueStatesConversion(t *testing.T) {\n\ttestCases := []*types.ProcessingQueueStates{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tStatesByCluster: map[string][]*types.ProcessingQueueState{\n\t\t\t\t\"test\": {\n\t\t\t\t\t{Level: common.Int32Ptr(1), AckLevel: common.Int64Ptr(1), MaxLevel: common.Int64Ptr(1), DomainFilter: &types.DomainFilter{DomainIDs: []string{\"test\"}, ReverseMatch: true}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromProcessingQueueStates(original)\n\t\troundTripObj := ToProcessingQueueStates(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryQueryWorkflowRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryQueryWorkflowRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryQueryWorkflowRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryQueryWorkflowRequest(original)\n\t\troundTripObj := ToHistoryQueryWorkflowRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryQueryWorkflowResponseConversion(t *testing.T) {\n\ttestCases := []*types.HistoryQueryWorkflowResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryQueryWorkflowResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryQueryWorkflowResponse(original)\n\t\troundTripObj := ToHistoryQueryWorkflowResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryReapplyEventsRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryReapplyEventsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryReapplyEventsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryReapplyEventsRequest(original)\n\t\troundTripObj := ToHistoryReapplyEventsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\nfunc TestHistoryRecordActivityTaskHeartbeatRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordActivityTaskHeartbeatRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRecordActivityTaskHeartbeatRequest(original)\n\t\troundTripObj := ToHistoryRecordActivityTaskHeartbeatRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordActivityTaskStartedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RecordActivityTaskStartedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordActivityTaskStartedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordActivityTaskStartedRequest(original)\n\t\troundTripObj := ToRecordActivityTaskStartedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordActivityTaskStartedResponseConversion(t *testing.T) {\n\ttestCases := []*types.RecordActivityTaskStartedResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordActivityTaskStartedResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordActivityTaskStartedResponse(original)\n\t\troundTripObj := ToRecordActivityTaskStartedResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestRecordChildExecutionCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RecordChildExecutionCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordChildExecutionCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordChildExecutionCompletedRequest(original)\n\t\troundTripObj := ToRecordChildExecutionCompletedRequest(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestRecordDecisionTaskStartedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RecordDecisionTaskStartedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordDecisionTaskStartedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordDecisionTaskStartedRequest(original)\n\t\troundTripObj := ToRecordDecisionTaskStartedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordDecisionTaskStartedResponseConversion(t *testing.T) {\n\ttestCases := []*types.RecordDecisionTaskStartedResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRecordDecisionTaskStartedResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordDecisionTaskStartedResponse(original)\n\t\troundTripObj := ToRecordDecisionTaskStartedResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestHistoryRefreshWorkflowTasksRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRefreshWorkflowTasksRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRefreshWorkflowTasksRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRefreshWorkflowTasksRequest(original)\n\t\troundTripObj := ToHistoryRefreshWorkflowTasksRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRemoveSignalMutableStateRequestConversion(t *testing.T) {\n\ttestCases := []*types.RemoveSignalMutableStateRequest{\n\t\tnil,\n\t\t{},\n\t\t{DomainUUID: \"test-uuid\", WorkflowExecution: &types.WorkflowExecution{WorkflowID: \"test\", RunID: \"test\"}, RequestID: \"test-req\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRemoveSignalMutableStateRequest(original)\n\t\troundTripObj := ToHistoryRemoveSignalMutableStateRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestReplicateEventsV2RequestConversion(t *testing.T) {\n\ttestCases := []*types.ReplicateEventsV2Request{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryReplicateEventsV2Request,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryReplicateEventsV2Request(original)\n\t\troundTripObj := ToHistoryReplicateEventsV2Request(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRequestCancelWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRequestCancelWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRequestCancelWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistoryRequestCancelWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryResetStickyTaskListRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryResetStickyTaskListRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryResetStickyTaskListRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryResetStickyTaskListRequest(original)\n\t\troundTripObj := ToHistoryResetStickyTaskListRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryResetStickyTaskListResponseConversion(t *testing.T) {\n\ttestCases := []*types.HistoryResetStickyTaskListResponse{\n\t\tnil,\n\t\t{},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryResetStickyTaskListResponse(original)\n\t\troundTripObj := ToHistoryResetStickyTaskListResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryResetWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryResetWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryResetWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryResetWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistoryResetWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRespondActivityTaskCanceledRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondActivityTaskCanceledRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondActivityTaskCanceledRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondActivityTaskCanceledRequest(original)\n\t\troundTripObj := ToHistoryRespondActivityTaskCanceledRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRespondActivityTaskCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondActivityTaskCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondActivityTaskCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondActivityTaskCompletedRequest(original)\n\t\troundTripObj := ToHistoryRespondActivityTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRespondActivityTaskFailedRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondActivityTaskFailedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondActivityTaskFailedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondActivityTaskFailedRequest(original)\n\t\troundTripObj := ToHistoryRespondActivityTaskFailedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRespondDecisionTaskCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondDecisionTaskCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondDecisionTaskCompletedRequest(original)\n\t\troundTripObj := ToHistoryRespondDecisionTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryRespondDecisionTaskCompletedResponseConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondDecisionTaskCompletedResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondDecisionTaskCompletedResponse(original)\n\t\troundTripObj := ToHistoryRespondDecisionTaskCompletedResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestHistoryRespondDecisionTaskFailedRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryRespondDecisionTaskFailedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryRespondDecisionTaskFailedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryRespondDecisionTaskFailedRequest(original)\n\t\troundTripObj := ToHistoryRespondDecisionTaskFailedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestScheduleDecisionTaskRequestConversion(t *testing.T) {\n\ttestCases := []*types.ScheduleDecisionTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryScheduleDecisionTaskRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryScheduleDecisionTaskRequest(original)\n\t\troundTripObj := ToHistoryScheduleDecisionTaskRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestShardOwnershipLostErrorConversion(t *testing.T) {\n\ttestCases := []*types.ShardOwnershipLostError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ShardOwnershipLostError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromShardOwnershipLostError(original)\n\t\troundTripObj := ToShardOwnershipLostError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistorySignalWithStartWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistorySignalWithStartWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistorySignalWithStartWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistorySignalWithStartWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistorySignalWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistorySignalWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistorySignalWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistorySignalWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistorySignalWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryStartWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.HistoryStartWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryStartWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryStartWorkflowExecutionRequest(original)\n\t\troundTripObj := ToHistoryStartWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSyncActivityRequestConversion(t *testing.T) {\n\ttestCases := []*types.SyncActivityRequest{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tDomainID:           \"test-domain-id\",\n\t\t\tWorkflowID:         \"test-workflow-id\",\n\t\t\tRunID:              \"test-run-id\",\n\t\t\tVersion:            1,\n\t\t\tScheduledID:        1,\n\t\t\tScheduledTime:      common.Int64Ptr(98765),\n\t\t\tStartedID:          1,\n\t\t\tStartedTime:        common.Int64Ptr(98765),\n\t\t\tLastHeartbeatTime:  common.Int64Ptr(98765),\n\t\t\tDetails:            []byte(\"test-details\"),\n\t\t\tAttempt:            3,\n\t\t\tLastFailureReason:  common.StringPtr(\"test-last-failure-reason\"),\n\t\t\tLastWorkerIdentity: \"test-last-worker-identity\",\n\t\t\tLastFailureDetails: []byte(\"test-last-failure-details\"),\n\t\t\tVersionHistory:     &testdata.VersionHistory,\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistorySyncActivityRequest(original)\n\t\troundTripObj := ToHistorySyncActivityRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSyncShardStatusRequestConversion(t *testing.T) {\n\ttestCases := []*types.SyncShardStatusRequest{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tSourceCluster: \"test-source-cluster\",\n\t\t\tShardID:       1,\n\t\t\tTimestamp:     common.Int64Ptr(98765),\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistorySyncShardStatusRequest(original)\n\t\troundTripObj := ToHistorySyncShardStatusRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRatelimitUpdate(t *testing.T) {\n\tt.Run(\"nil\", func(t *testing.T) {\n\t\tassert.Nil(t, ToHistoryRatelimitUpdateRequest(nil), \"request to internal\")\n\t\tassert.Nil(t, FromHistoryRatelimitUpdateResponse(nil), \"response from internal\")\n\t})\n\n\tt.Run(\"nil Any contents\", func(t *testing.T) {\n\t\tassert.Equal(t, &types.RatelimitUpdateRequest{Any: nil}, ToHistoryRatelimitUpdateRequest(&history.RatelimitUpdateRequest{Data: nil}), \"request to internal\")\n\t\tassert.Equal(t, &history.RatelimitUpdateResponse{Data: nil}, FromHistoryRatelimitUpdateResponse(&types.RatelimitUpdateResponse{Any: nil}), \"response from internal\")\n\t})\n\n\tt.Run(\"with Any contents\", func(t *testing.T) {\n\t\tinternal := &types.Any{ValueType: \"test\", Value: []byte(`test data`)}\n\t\tthrift := &shared.Any{ValueType: common.StringPtr(\"test\"), Value: []byte(`test data`)}\n\t\tassert.Equal(t, &types.RatelimitUpdateRequest{Any: internal}, ToHistoryRatelimitUpdateRequest(&history.RatelimitUpdateRequest{Data: thrift}), \"request to internal\")\n\t\tassert.Equal(t, &history.RatelimitUpdateResponse{Data: thrift}, FromHistoryRatelimitUpdateResponse(&types.RatelimitUpdateResponse{Any: internal}), \"response from internal\")\n\t})\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/matching.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/matching\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tFromMatchingDescribeTaskListResponse       = FromDescribeTaskListResponse\n\tToMatchingDescribeTaskListResponse         = ToDescribeTaskListResponse\n\tFromMatchingGetTaskListsByDomainRequest    = FromGetTaskListsByDomainRequest\n\tToMatchingGetTaskListsByDomainRequest      = ToGetTaskListsByDomainRequest\n\tFromMatchingGetTaskListsByDomainResponse   = FromGetTaskListsByDomainResponse\n\tToMatchingGetTaskListsByDomainResponse     = ToGetTaskListsByDomainResponse\n\tFromMatchingListTaskListPartitionsResponse = FromListTaskListPartitionsResponse\n\tToMatchingListTaskListPartitionsResponse   = ToListTaskListPartitionsResponse\n)\n\n// FromMatchingAddActivityTaskRequest converts internal AddActivityTaskRequest type to thrift\nfunc FromMatchingAddActivityTaskRequest(t *types.AddActivityTaskRequest) *matching.AddActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.AddActivityTaskRequest{\n\t\tDomainUUID:                    &t.DomainUUID,\n\t\tExecution:                     FromWorkflowExecution(t.Execution),\n\t\tSourceDomainUUID:              &t.SourceDomainUUID,\n\t\tTaskList:                      FromTaskList(t.TaskList),\n\t\tScheduleId:                    &t.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tSource:                        FromTaskSource(t.Source),\n\t\tForwardedFrom:                 &t.ForwardedFrom,\n\t\tActivityTaskDispatchInfo:      FromActivityTaskDispatchInfo(t.ActivityTaskDispatchInfo),\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\n// ToMatchingAddActivityTaskRequest converts thrift AddActivityTaskRequest type to internal\nfunc ToMatchingAddActivityTaskRequest(t *matching.AddActivityTaskRequest) *types.AddActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddActivityTaskRequest{\n\t\tDomainUUID:                    t.GetDomainUUID(),\n\t\tExecution:                     ToWorkflowExecution(t.Execution),\n\t\tSourceDomainUUID:              t.GetSourceDomainUUID(),\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tScheduleID:                    t.GetScheduleId(),\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tSource:                        ToTaskSource(t.Source),\n\t\tForwardedFrom:                 t.GetForwardedFrom(),\n\t\tActivityTaskDispatchInfo:      ToActivityTaskDispatchInfo(t.ActivityTaskDispatchInfo),\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\nfunc FromActivityTaskDispatchInfo(t *types.ActivityTaskDispatchInfo) *matching.ActivityTaskDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.ActivityTaskDispatchInfo{\n\t\tScheduledEvent:                  FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tAttempt:                         t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  &t.WorkflowDomain,\n\t}\n}\n\n// ToRecordActivityTaskStartedResponse converts thrift RecordActivityTaskStartedResponse type to internal\nfunc ToActivityTaskDispatchInfo(t *matching.ActivityTaskDispatchInfo) *types.ActivityTaskDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskDispatchInfo{\n\t\tScheduledEvent:                  ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tAttempt:                         t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  common.StringDefault(t.WorkflowDomain),\n\t}\n}\n\n// FromMatchingAddDecisionTaskRequest converts internal AddDecisionTaskRequest type to thrift\nfunc FromMatchingAddDecisionTaskRequest(t *types.AddDecisionTaskRequest) *matching.AddDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.AddDecisionTaskRequest{\n\t\tDomainUUID:                    &t.DomainUUID,\n\t\tExecution:                     FromWorkflowExecution(t.Execution),\n\t\tTaskList:                      FromTaskList(t.TaskList),\n\t\tScheduleId:                    &t.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tSource:                        FromTaskSource(t.Source),\n\t\tForwardedFrom:                 &t.ForwardedFrom,\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\n// ToMatchingAddDecisionTaskRequest converts thrift AddDecisionTaskRequest type to internal\nfunc ToMatchingAddDecisionTaskRequest(t *matching.AddDecisionTaskRequest) *types.AddDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AddDecisionTaskRequest{\n\t\tDomainUUID:                    t.GetDomainUUID(),\n\t\tExecution:                     ToWorkflowExecution(t.Execution),\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tScheduleID:                    t.GetScheduleId(),\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tSource:                        ToTaskSource(t.Source),\n\t\tForwardedFrom:                 t.GetForwardedFrom(),\n\t\tPartitionConfig:               t.PartitionConfig,\n\t}\n}\n\n// FromMatchingCancelOutstandingPollRequest converts internal CancelOutstandingPollRequest type to thrift\nfunc FromMatchingCancelOutstandingPollRequest(t *types.CancelOutstandingPollRequest) *matching.CancelOutstandingPollRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.CancelOutstandingPollRequest{\n\t\tDomainUUID:   &t.DomainUUID,\n\t\tTaskListType: t.TaskListType,\n\t\tTaskList:     FromTaskList(t.TaskList),\n\t\tPollerID:     &t.PollerID,\n\t}\n}\n\n// ToMatchingCancelOutstandingPollRequest converts thrift CancelOutstandingPollRequest type to internal\nfunc ToMatchingCancelOutstandingPollRequest(t *matching.CancelOutstandingPollRequest) *types.CancelOutstandingPollRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelOutstandingPollRequest{\n\t\tDomainUUID:   t.GetDomainUUID(),\n\t\tTaskListType: t.TaskListType,\n\t\tTaskList:     ToTaskList(t.TaskList),\n\t\tPollerID:     t.GetPollerID(),\n\t}\n}\n\n// FromMatchingDescribeTaskListRequest converts internal DescribeTaskListRequest type to thrift\nfunc FromMatchingDescribeTaskListRequest(t *types.MatchingDescribeTaskListRequest) *matching.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.DescribeTaskListRequest{\n\t\tDomainUUID:  &t.DomainUUID,\n\t\tDescRequest: FromDescribeTaskListRequest(t.DescRequest),\n\t}\n}\n\n// ToMatchingDescribeTaskListRequest converts thrift DescribeTaskListRequest type to internal\nfunc ToMatchingDescribeTaskListRequest(t *matching.DescribeTaskListRequest) *types.MatchingDescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID:  t.GetDomainUUID(),\n\t\tDescRequest: ToDescribeTaskListRequest(t.DescRequest),\n\t}\n}\n\n// FromMatchingListTaskListPartitionsRequest converts internal ListTaskListPartitionsRequest type to thrift\nfunc FromMatchingListTaskListPartitionsRequest(t *types.MatchingListTaskListPartitionsRequest) *matching.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.ListTaskListPartitionsRequest{\n\t\tDomain:   &t.Domain,\n\t\tTaskList: FromTaskList(t.TaskList),\n\t}\n}\n\n// ToMatchingListTaskListPartitionsRequest converts thrift ListTaskListPartitionsRequest type to internal\nfunc ToMatchingListTaskListPartitionsRequest(t *matching.ListTaskListPartitionsRequest) *types.MatchingListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingListTaskListPartitionsRequest{\n\t\tDomain:   t.GetDomain(),\n\t\tTaskList: ToTaskList(t.TaskList),\n\t}\n}\n\n// FromMatchingPollForActivityTaskRequest converts internal PollForActivityTaskRequest type to thrift\nfunc FromMatchingPollForActivityTaskRequest(t *types.MatchingPollForActivityTaskRequest) *matching.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.PollForActivityTaskRequest{\n\t\tDomainUUID:     &t.DomainUUID,\n\t\tPollerID:       &t.PollerID,\n\t\tPollRequest:    FromPollForActivityTaskRequest(t.PollRequest),\n\t\tForwardedFrom:  &t.ForwardedFrom,\n\t\tIsolationGroup: &t.IsolationGroup,\n\t}\n}\n\n// ToMatchingPollForActivityTaskRequest converts thrift PollForActivityTaskRequest type to internal\nfunc ToMatchingPollForActivityTaskRequest(t *matching.PollForActivityTaskRequest) *types.MatchingPollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForActivityTaskRequest{\n\t\tDomainUUID:     t.GetDomainUUID(),\n\t\tPollerID:       t.GetPollerID(),\n\t\tPollRequest:    ToPollForActivityTaskRequest(t.PollRequest),\n\t\tForwardedFrom:  t.GetForwardedFrom(),\n\t\tIsolationGroup: t.GetIsolationGroup(),\n\t}\n}\n\n// FromMatchingPollForDecisionTaskRequest converts internal PollForDecisionTaskRequest type to thrift\nfunc FromMatchingPollForDecisionTaskRequest(t *types.MatchingPollForDecisionTaskRequest) *matching.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.PollForDecisionTaskRequest{\n\t\tDomainUUID:     &t.DomainUUID,\n\t\tPollerID:       &t.PollerID,\n\t\tPollRequest:    FromPollForDecisionTaskRequest(t.PollRequest),\n\t\tForwardedFrom:  &t.ForwardedFrom,\n\t\tIsolationGroup: &t.IsolationGroup,\n\t}\n}\n\n// ToMatchingPollForDecisionTaskRequest converts thrift PollForDecisionTaskRequest type to internal\nfunc ToMatchingPollForDecisionTaskRequest(t *matching.PollForDecisionTaskRequest) *types.MatchingPollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForDecisionTaskRequest{\n\t\tDomainUUID:     t.GetDomainUUID(),\n\t\tPollerID:       t.GetPollerID(),\n\t\tPollRequest:    ToPollForDecisionTaskRequest(t.PollRequest),\n\t\tForwardedFrom:  t.GetForwardedFrom(),\n\t\tIsolationGroup: t.GetIsolationGroup(),\n\t}\n}\n\n// FromMatchingPollForDecisionTaskResponse converts internal PollForDecisionTaskResponse type to thrift\nfunc FromMatchingPollForDecisionTaskResponse(t *types.MatchingPollForDecisionTaskResponse) *matching.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    t.PreviousStartedEventID,\n\t\tStartedEventId:            &t.StartedEventID,\n\t\tAttempt:                   &t.Attempt,\n\t\tNextEventId:               &t.NextEventID,\n\t\tBacklogCountHint:          &t.BacklogCountHint,\n\t\tStickyExecutionEnabled:    &t.StickyExecutionEnabled,\n\t\tQuery:                     FromWorkflowQuery(t.Query),\n\t\tDecisionInfo:              FromTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         &t.EventStoreVersion,\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tTotalHistoryBytes:         &t.TotalHistoryBytes,\n\t\tAutoConfigHint:            FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// ToMatchingPollForDecisionTaskResponse converts thrift PollForDecisionTaskResponse type to internal\nfunc ToMatchingPollForDecisionTaskResponse(t *matching.PollForDecisionTaskResponse) *types.MatchingPollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    t.PreviousStartedEventId,\n\t\tStartedEventID:            t.GetStartedEventId(),\n\t\tAttempt:                   t.GetAttempt(),\n\t\tNextEventID:               t.GetNextEventId(),\n\t\tBacklogCountHint:          t.GetBacklogCountHint(),\n\t\tStickyExecutionEnabled:    t.GetStickyExecutionEnabled(),\n\t\tQuery:                     ToWorkflowQuery(t.Query),\n\t\tDecisionInfo:              ToTransientDecisionInfo(t.DecisionInfo),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tEventStoreVersion:         t.GetEventStoreVersion(),\n\t\tBranchToken:               t.BranchToken,\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tTotalHistoryBytes:         t.GetTotalHistoryBytes(),\n\t\tAutoConfigHint:            ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc FromMatchingPollForActivityTaskResponse(t *types.MatchingPollForActivityTaskResponse) *shared.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               FromWorkflowExecution(t.WorkflowExecution),\n\t\tActivityId:                      &t.ActivityID,\n\t\tActivityType:                    FromActivityType(t.ActivityType),\n\t\tInput:                           t.Input,\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tScheduleToCloseTimeoutSeconds:   t.ScheduleToCloseTimeoutSeconds,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tStartToCloseTimeoutSeconds:      t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:         t.HeartbeatTimeoutSeconds,\n\t\tAttempt:                         &t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  &t.WorkflowDomain,\n\t\tHeader:                          FromHeader(t.Header),\n\t\tAutoConfigHint:                  FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\nfunc ToMatchingPollForActivityTaskResponse(t *shared.PollForActivityTaskResponse) *types.MatchingPollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingPollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               ToWorkflowExecution(t.WorkflowExecution),\n\t\tActivityID:                      t.GetActivityId(),\n\t\tActivityType:                    ToActivityType(t.ActivityType),\n\t\tInput:                           t.Input,\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tScheduleToCloseTimeoutSeconds:   t.ScheduleToCloseTimeoutSeconds,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tStartToCloseTimeoutSeconds:      t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:         t.HeartbeatTimeoutSeconds,\n\t\tAttempt:                         t.GetAttempt(),\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.GetWorkflowDomain(),\n\t\tHeader:                          ToHeader(t.Header),\n\t\tAutoConfigHint:                  ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// FromMatchingQueryWorkflowRequest converts internal QueryWorkflowRequest type to thrift\nfunc FromMatchingQueryWorkflowRequest(t *types.MatchingQueryWorkflowRequest) *matching.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.QueryWorkflowRequest{\n\t\tDomainUUID:    &t.DomainUUID,\n\t\tTaskList:      FromTaskList(t.TaskList),\n\t\tQueryRequest:  FromQueryWorkflowRequest(t.QueryRequest),\n\t\tForwardedFrom: &t.ForwardedFrom,\n\t}\n}\n\n// ToMatchingQueryWorkflowRequest converts thrift QueryWorkflowRequest type to internal\nfunc ToMatchingQueryWorkflowRequest(t *matching.QueryWorkflowRequest) *types.MatchingQueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID:    t.GetDomainUUID(),\n\t\tTaskList:      ToTaskList(t.TaskList),\n\t\tQueryRequest:  ToQueryWorkflowRequest(t.QueryRequest),\n\t\tForwardedFrom: t.GetForwardedFrom(),\n\t}\n}\n\n// FromMatchingRespondQueryTaskCompletedRequest converts internal RespondQueryTaskCompletedRequest type to thrift\nfunc FromMatchingRespondQueryTaskCompletedRequest(t *types.MatchingRespondQueryTaskCompletedRequest) *matching.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &matching.RespondQueryTaskCompletedRequest{\n\t\tDomainUUID:       &t.DomainUUID,\n\t\tTaskList:         FromTaskList(t.TaskList),\n\t\tTaskID:           &t.TaskID,\n\t\tCompletedRequest: FromRespondQueryTaskCompletedRequest(t.CompletedRequest),\n\t}\n}\n\n// ToMatchingRespondQueryTaskCompletedRequest converts thrift RespondQueryTaskCompletedRequest type to internal\nfunc ToMatchingRespondQueryTaskCompletedRequest(t *matching.RespondQueryTaskCompletedRequest) *types.MatchingRespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingRespondQueryTaskCompletedRequest{\n\t\tDomainUUID:       t.GetDomainUUID(),\n\t\tTaskList:         ToTaskList(t.TaskList),\n\t\tTaskID:           t.GetTaskID(),\n\t\tCompletedRequest: ToRespondQueryTaskCompletedRequest(t.CompletedRequest),\n\t}\n}\n\n// FromTaskSource converts internal TaskSource type to thrift\nfunc FromTaskSource(t *types.TaskSource) *matching.TaskSource {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.TaskSourceHistory:\n\t\tv := matching.TaskSourceHistory\n\t\treturn &v\n\tcase types.TaskSourceDbBacklog:\n\t\tv := matching.TaskSourceDbBacklog\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToTaskSource converts thrift TaskSource type to internal\nfunc ToTaskSource(t *matching.TaskSource) *types.TaskSource {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase matching.TaskSourceHistory:\n\t\tv := types.TaskSourceHistory\n\t\treturn &v\n\tcase matching.TaskSourceDbBacklog:\n\t\tv := types.TaskSourceDbBacklog\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\nfunc FromMatchingQueryWorkflowResponse(t *types.MatchingQueryWorkflowResponse) *shared.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueryWorkflowResponse{\n\t\tQueryResult:   t.QueryResult,\n\t\tQueryRejected: FromQueryRejected(t.QueryRejected),\n\t}\n}\n\nfunc ToMatchingQueryWorkflowResponse(t *shared.QueryWorkflowResponse) *types.MatchingQueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MatchingQueryWorkflowResponse{\n\t\tQueryResult:   t.QueryResult,\n\t\tQueryRejected: ToQueryRejected(t.QueryRejected),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/matching_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/google/go-cmp/cmp/cmpopts\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestMatchingAddActivityTaskRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.AddActivityTaskRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingAddActivityTaskRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.AddActivityTaskRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingAddActivityTaskRequest(tc.input)\n\t\troundTripObj := ToMatchingAddActivityTaskRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskDispatchInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ActivityTaskDispatchInfo\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingActivityTaskDispatchInfo,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ActivityTaskDispatchInfo{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromActivityTaskDispatchInfo(tc.input)\n\t\troundTripObj := ToActivityTaskDispatchInfo(thriftObj)\n\t\topts := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(tc.input, roundTripObj, opts); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestMatchingAddDecisionTaskRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.AddDecisionTaskRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingAddDecisionTaskRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.AddDecisionTaskRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingAddDecisionTaskRequest(tc.input)\n\t\troundTripObj := ToMatchingAddDecisionTaskRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingCancelOutStandingPollRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.CancelOutstandingPollRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingCancelOutstandingPollRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.CancelOutstandingPollRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingCancelOutstandingPollRequest(tc.input)\n\t\troundTripObj := ToMatchingCancelOutstandingPollRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingDescribeTaskListRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingDescribeTaskListRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingDescribeTaskListRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingDescribeTaskListRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingDescribeTaskListRequest(tc.input)\n\t\troundTripObj := ToMatchingDescribeTaskListRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingListTaskListPartitionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingListTaskListPartitionsRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingListTaskListPartitionsRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingListTaskListPartitionsRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingListTaskListPartitionsRequest(tc.input)\n\t\troundTripObj := ToMatchingListTaskListPartitionsRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingPollForActivityRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingPollForActivityTaskRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingPollForActivityTaskRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingPollForActivityTaskRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingPollForActivityTaskRequest(tc.input)\n\t\troundTripObj := ToMatchingPollForActivityTaskRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingPollForDecisionRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingPollForDecisionTaskRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingPollForDecisionTaskRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingPollForDecisionTaskRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingPollForDecisionTaskRequest(tc.input)\n\t\troundTripObj := ToMatchingPollForDecisionTaskRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingPollForDecisionResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingPollForDecisionTaskResponse\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingPollForDecisionTaskResponse,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingPollForDecisionTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingPollForDecisionTaskResponse(tc.input)\n\t\troundTripObj := ToMatchingPollForDecisionTaskResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\topt2 := cmpopts.IgnoreFields(types.MatchingPollForDecisionTaskResponse{}, \"PartitionConfig\")\n\t\topt3 := cmpopts.IgnoreFields(types.MatchingPollForDecisionTaskResponse{}, \"LoadBalancerHints\")\n\t\tif diff := cmp.Diff(tc.input, roundTripObj, opt, opt2, opt3); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingPollForActivityTaskResponse\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingPollForActivityTaskResponse,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingPollForActivityTaskResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingPollForActivityTaskResponse(tc.input)\n\t\troundTripObj := ToMatchingPollForActivityTaskResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.MatchingPollForActivityTaskResponse{}, \"PartitionConfig\")\n\t\topt2 := cmpopts.IgnoreFields(types.MatchingPollForActivityTaskResponse{}, \"LoadBalancerHints\")\n\t\tif diff := cmp.Diff(tc.input, roundTripObj, opt, opt2); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingQueryWorkflowRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingQueryWorkflowRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingQueryWorkflowRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingQueryWorkflowRequest(tc.input)\n\t\troundTripObj := ToMatchingQueryWorkflowRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MatchingRespondQueryTaskCompletedRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MatchingRespondQueryTaskCompletedRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MatchingRespondQueryTaskCompletedRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromMatchingRespondQueryTaskCompletedRequest(tc.input)\n\t\troundTripObj := ToMatchingRespondQueryTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n\nfunc TestTaskSource(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.TaskSource\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil TaskSourceHistory input test\",\n\t\t\tinput: types.TaskSourceHistory.Ptr(),\n\t\t},\n\t\t{\n\t\t\tdesc:  \"non-nil TaskSourceDBBacklog input test\",\n\t\t\tinput: types.TaskSourceDbBacklog.Ptr(),\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tthriftObj := FromTaskSource(tc.input)\n\t\troundTripObj := ToTaskSource(thriftObj)\n\t\tassert.Equal(t, tc.input, roundTripObj)\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/predicate.go",
    "content": "package thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromPredicateType(t types.PredicateType) *shared.PredicateType {\n\tv := shared.PredicateTypeUniversal\n\tswitch t {\n\tcase types.PredicateTypeUniversal:\n\t\treturn &v\n\tcase types.PredicateTypeEmpty:\n\t\tv = shared.PredicateTypeEmpty\n\t\treturn &v\n\tcase types.PredicateTypeDomainID:\n\t\tv = shared.PredicateTypeDomainID\n\t\treturn &v\n\t}\n\t// default to universal\n\treturn &v\n}\n\nfunc ToPredicateType(t *shared.PredicateType) types.PredicateType {\n\tif t == nil {\n\t\treturn types.PredicateTypeUniversal\n\t}\n\tswitch *t {\n\tcase shared.PredicateTypeUniversal:\n\t\treturn types.PredicateTypeUniversal\n\tcase shared.PredicateTypeEmpty:\n\t\treturn types.PredicateTypeEmpty\n\tcase shared.PredicateTypeDomainID:\n\t\treturn types.PredicateTypeDomainID\n\t}\n\t// default to universal\n\treturn types.PredicateTypeUniversal\n}\n\nfunc FromUniversalPredicateAttributes(a *types.UniversalPredicateAttributes) *shared.UniversalPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &shared.UniversalPredicateAttributes{}\n}\n\nfunc ToUniversalPredicateAttributes(a *shared.UniversalPredicateAttributes) *types.UniversalPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &types.UniversalPredicateAttributes{}\n}\n\nfunc FromEmptyPredicateAttributes(a *types.EmptyPredicateAttributes) *shared.EmptyPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &shared.EmptyPredicateAttributes{}\n}\n\nfunc ToEmptyPredicateAttributes(a *shared.EmptyPredicateAttributes) *types.EmptyPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &types.EmptyPredicateAttributes{}\n}\n\nfunc FromDomainIDPredicateAttributes(a *types.DomainIDPredicateAttributes) *shared.DomainIDPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainIDPredicateAttributes{\n\t\tDomainIDs:   a.DomainIDs,\n\t\tIsExclusive: a.IsExclusive,\n\t}\n}\n\nfunc ToDomainIDPredicateAttributes(a *shared.DomainIDPredicateAttributes) *types.DomainIDPredicateAttributes {\n\tif a == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainIDPredicateAttributes{\n\t\tDomainIDs:   a.DomainIDs,\n\t\tIsExclusive: a.IsExclusive,\n\t}\n}\n\nfunc FromPredicate(p *types.Predicate) *shared.Predicate {\n\tif p == nil {\n\t\treturn nil\n\t}\n\treturn &shared.Predicate{\n\t\tPredicateType:                FromPredicateType(p.PredicateType),\n\t\tUniversalPredicateAttributes: FromUniversalPredicateAttributes(p.UniversalPredicateAttributes),\n\t\tEmptyPredicateAttributes:     FromEmptyPredicateAttributes(p.EmptyPredicateAttributes),\n\t\tDomainIDPredicateAttributes:  FromDomainIDPredicateAttributes(p.DomainIDPredicateAttributes),\n\t}\n}\n\nfunc ToPredicate(p *shared.Predicate) *types.Predicate {\n\tif p == nil {\n\t\treturn nil\n\t}\n\treturn &types.Predicate{\n\t\tPredicateType:                ToPredicateType(p.PredicateType),\n\t\tUniversalPredicateAttributes: ToUniversalPredicateAttributes(p.UniversalPredicateAttributes),\n\t\tEmptyPredicateAttributes:     ToEmptyPredicateAttributes(p.EmptyPredicateAttributes),\n\t\tDomainIDPredicateAttributes:  ToDomainIDPredicateAttributes(p.DomainIDPredicateAttributes),\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/predicate_test.go",
    "content": "// Copyright (c) 2017-2022 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc predicateTypeFuzzGenerator(t *types.PredicateType, c fuzz.Continue) {\n\tswitch c.Intn(int(types.NumPredicateTypes)) {\n\tcase 0:\n\t\t*t = types.PredicateTypeUniversal\n\tcase 1:\n\t\t*t = types.PredicateTypeEmpty\n\tcase 2:\n\t\t*t = types.PredicateTypeDomainID\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc predicateTypeSharedFuzzGenerator(t **shared.PredicateType, c fuzz.Continue) {\n\tswitch c.Intn(int(types.NumPredicateTypes)) {\n\tcase 0:\n\t\t*t = common.Ptr(shared.PredicateTypeUniversal)\n\tcase 1:\n\t\t*t = common.Ptr(shared.PredicateTypeEmpty)\n\tcase 2:\n\t\t*t = common.Ptr(shared.PredicateTypeDomainID)\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc predicateFuzzGenerator(t *types.Predicate, c fuzz.Continue) {\n\tswitch c.Intn(int(types.NumPredicateTypes)) {\n\tcase 0:\n\t\tt.PredicateType = types.PredicateTypeUniversal\n\t\tc.Fuzz(&t.UniversalPredicateAttributes)\n\tcase 1:\n\t\tt.PredicateType = types.PredicateTypeEmpty\n\t\tc.Fuzz(&t.EmptyPredicateAttributes)\n\tcase 2:\n\t\tt.PredicateType = types.PredicateTypeDomainID\n\t\tc.Fuzz(&t.DomainIDPredicateAttributes)\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc predicateSharedFuzzGenerator(t *shared.Predicate, c fuzz.Continue) {\n\tswitch c.Intn(int(types.NumPredicateTypes)) {\n\tcase 0:\n\t\tt.PredicateType = common.Ptr(shared.PredicateTypeUniversal)\n\t\tc.Fuzz(&t.UniversalPredicateAttributes)\n\tcase 1:\n\t\tt.PredicateType = common.Ptr(shared.PredicateTypeEmpty)\n\t\tc.Fuzz(&t.EmptyPredicateAttributes)\n\tcase 2:\n\t\tt.PredicateType = common.Ptr(shared.PredicateTypeDomainID)\n\t\tc.Fuzz(&t.DomainIDPredicateAttributes)\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc TestFuzzPredicateType_Roundtrip_FromTypes(t *testing.T) {\n\tf := fuzz.New().Funcs(\n\t\tpredicateTypeFuzzGenerator,\n\t)\n\n\tfor i := 0; i < 1000; i++ {\n\t\t// Generate only valid enum values\n\t\tvar original types.PredicateType\n\t\tf.Fuzz(&original)\n\n\t\t// types → shared → types\n\t\tshared := FromPredicateType(original)\n\t\tconverted := ToPredicateType(shared)\n\n\t\tassert.Equal(t, original, converted, \"PredicateType roundtrip failed: types → shared → types\")\n\t}\n}\n\nfunc TestFuzzPredicateType_Roundtrip_FromShared(t *testing.T) {\n\tf := fuzz.New().Funcs(\n\t\tpredicateTypeSharedFuzzGenerator,\n\t)\n\tfor i := 0; i < 1000; i++ {\n\t\t// Generate only valid enum values\n\t\tvar original *shared.PredicateType\n\t\tf.Fuzz(&original)\n\n\t\t// shared → types → shared\n\t\ttypes := ToPredicateType(original)\n\t\tconverted := FromPredicateType(types)\n\n\t\tassert.Equal(t, original, converted, \"PredicateType roundtrip failed: shared → types → shared\")\n\t}\n}\n\nfunc TestFuzzUniversalPredicateAttributes_Roundtrip_FromTypes(t *testing.T) {\n\tf := fuzz.New()\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original types.UniversalPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// types → shared → types\n\t\tshared := FromUniversalPredicateAttributes(&original)\n\t\tconverted := ToUniversalPredicateAttributes(shared)\n\n\t\tassert.Equal(t, original, *converted, \"UniversalPredicateAttributes roundtrip failed: types → shared → types\")\n\t}\n}\n\nfunc TestFuzzUniversalPredicateAttributes_Roundtrip_FromShared(t *testing.T) {\n\tf := fuzz.New()\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original shared.UniversalPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// shared → types → shared\n\t\ttypes := ToUniversalPredicateAttributes(&original)\n\t\tconverted := FromUniversalPredicateAttributes(types)\n\n\t\tassert.Equal(t, original, *converted, \"UniversalPredicateAttributes roundtrip failed: shared → types → shared\")\n\t}\n}\n\nfunc TestFuzzEmptyPredicateAttributes_Roundtrip_FromTypes(t *testing.T) {\n\tf := fuzz.New()\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original types.EmptyPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// types → shared → types\n\t\tshared := FromEmptyPredicateAttributes(&original)\n\t\tconverted := ToEmptyPredicateAttributes(shared)\n\n\t\tassert.Equal(t, original, *converted, \"EmptyPredicateAttributes roundtrip failed: types → shared → types\")\n\t}\n}\n\nfunc TestFuzzEmptyPredicateAttributes_Roundtrip_FromShared(t *testing.T) {\n\tf := fuzz.New()\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original shared.EmptyPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// shared → types → shared\n\t\ttypes := ToEmptyPredicateAttributes(&original)\n\t\tconverted := FromEmptyPredicateAttributes(types)\n\n\t\tassert.Equal(t, original, *converted, \"EmptyPredicateAttributes roundtrip failed: shared → types → shared\")\n\t}\n}\n\nfunc TestFuzzDomainIDPredicateAttributes_Roundtrip_FromTypes(t *testing.T) {\n\tf := fuzz.New().NilChance(0.1)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original types.DomainIDPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// types → shared → types\n\t\tshared := FromDomainIDPredicateAttributes(&original)\n\t\tconverted := ToDomainIDPredicateAttributes(shared)\n\n\t\tassert.Equal(t, original, *converted, \"DomainIDPredicateAttributes roundtrip failed: types → shared → types\")\n\t}\n}\n\nfunc TestFuzzDomainIDPredicateAttributes_Roundtrip_FromShared(t *testing.T) {\n\tf := fuzz.New().NilChance(0.1)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original shared.DomainIDPredicateAttributes\n\t\tf.Fuzz(&original)\n\n\t\t// shared → types → shared\n\t\ttypes := ToDomainIDPredicateAttributes(&original)\n\t\tconverted := FromDomainIDPredicateAttributes(types)\n\n\t\tassert.Equal(t, original, *converted, \"DomainIDPredicateAttributes roundtrip failed: shared → types → shared\")\n\t}\n}\n\nfunc TestFuzzPredicate_Roundtrip_FromTypes(t *testing.T) {\n\tf := fuzz.New().NilChance(0.2).Funcs(\n\t\tpredicateFuzzGenerator,\n\t).MaxDepth(15) // Limit depth to avoid infinite recursion\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original types.Predicate\n\t\tf.Fuzz(&original)\n\n\t\t// types → shared → types\n\t\tshared := FromPredicate(&original)\n\t\tconverted := ToPredicate(shared)\n\n\t\tassert.Equal(t, original, *converted, \"Predicate roundtrip failed: types → shared → types\")\n\t}\n}\n\nfunc TestFuzzPredicate_Roundtrip_FromShared(t *testing.T) {\n\tf := fuzz.New().NilChance(0.2).Funcs(\n\t\tpredicateSharedFuzzGenerator,\n\t).MaxDepth(15) // Limit depth to avoid infinite recursion\n\tfor i := 0; i < 1000; i++ {\n\t\tvar original shared.Predicate\n\t\tf.Fuzz(&original)\n\n\t\t// shared → types → shared\n\t\ttypes := ToPredicate(&original)\n\t\tconverted := FromPredicate(types)\n\n\t\tassert.Equal(t, original, *converted, \"Predicate roundtrip failed: shared → types → shared\")\n\t}\n}\n\nfunc TestFuzzPredicate_NilHandling(t *testing.T) {\n\t// Test nil handling\n\tassert.Nil(t, FromPredicate(nil))\n\tassert.Nil(t, ToPredicate(nil))\n\tassert.Nil(t, FromDomainIDPredicateAttributes(nil))\n\tassert.Nil(t, ToDomainIDPredicateAttributes(nil))\n}\n\nfunc TestPredicateType_InvalidValues(t *testing.T) {\n\tassert.Equal(t, shared.PredicateTypeUniversal, *FromPredicateType(types.PredicateType(100)))\n\tassert.Equal(t, types.PredicateTypeUniversal, ToPredicateType(nil))\n\tassert.Equal(t, types.PredicateTypeUniversal, ToPredicateType(common.Ptr(shared.PredicateType(100))))\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/replicator.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// FromDLQType converts internal DLQType type to thrift\nfunc FromDLQType(t *types.DLQType) *replicator.DLQType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DLQTypeReplication:\n\t\tv := replicator.DLQTypeReplication\n\t\treturn &v\n\tcase types.DLQTypeDomain:\n\t\tv := replicator.DLQTypeDomain\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDLQType converts thrift DLQType type to internal\nfunc ToDLQType(t *replicator.DLQType) *types.DLQType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase replicator.DLQTypeReplication:\n\t\tv := types.DLQTypeReplication\n\t\treturn &v\n\tcase replicator.DLQTypeDomain:\n\t\tv := types.DLQTypeDomain\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromDomainOperation converts internal DomainOperation type to thrift\nfunc FromDomainOperation(t *types.DomainOperation) *replicator.DomainOperation {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DomainOperationCreate:\n\t\tv := replicator.DomainOperationCreate\n\t\treturn &v\n\tcase types.DomainOperationUpdate:\n\t\tv := replicator.DomainOperationUpdate\n\t\treturn &v\n\tcase types.DomainOperationDelete:\n\t\tv := replicator.DomainOperationDelete\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDomainOperation converts thrift DomainOperation type to internal\nfunc ToDomainOperation(t *replicator.DomainOperation) *types.DomainOperation {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase replicator.DomainOperationCreate:\n\t\tv := types.DomainOperationCreate\n\t\treturn &v\n\tcase replicator.DomainOperationUpdate:\n\t\tv := types.DomainOperationUpdate\n\t\treturn &v\n\tcase replicator.DomainOperationDelete:\n\t\tv := types.DomainOperationDelete\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromDomainTaskAttributes converts internal DomainTaskAttributes type to thrift\nfunc FromDomainTaskAttributes(t *types.DomainTaskAttributes) *replicator.DomainTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.DomainTaskAttributes{\n\t\tDomainOperation:         FromDomainOperation(t.DomainOperation),\n\t\tID:                      &t.ID,\n\t\tInfo:                    FromDomainInfo(t.Info),\n\t\tConfig:                  FromDomainConfiguration(t.Config),\n\t\tReplicationConfig:       FromDomainReplicationConfiguration(t.ReplicationConfig),\n\t\tConfigVersion:           &t.ConfigVersion,\n\t\tFailoverVersion:         &t.FailoverVersion,\n\t\tPreviousFailoverVersion: &t.PreviousFailoverVersion,\n\t}\n}\n\n// ToDomainTaskAttributes converts thrift DomainTaskAttributes type to internal\nfunc ToDomainTaskAttributes(t *replicator.DomainTaskAttributes) *types.DomainTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainTaskAttributes{\n\t\tDomainOperation:         ToDomainOperation(t.DomainOperation),\n\t\tID:                      t.GetID(),\n\t\tInfo:                    ToDomainInfo(t.Info),\n\t\tConfig:                  ToDomainConfiguration(t.Config),\n\t\tReplicationConfig:       ToDomainReplicationConfiguration(t.ReplicationConfig),\n\t\tConfigVersion:           t.GetConfigVersion(),\n\t\tFailoverVersion:         t.GetFailoverVersion(),\n\t\tPreviousFailoverVersion: t.GetPreviousFailoverVersion(),\n\t}\n}\n\n// FromFailoverMarkerAttributes converts internal FailoverMarkerAttributes type to thrift\nfunc FromFailoverMarkerAttributes(t *types.FailoverMarkerAttributes) *replicator.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.FailoverMarkerAttributes{\n\t\tDomainID:        &t.DomainID,\n\t\tFailoverVersion: &t.FailoverVersion,\n\t\tCreationTime:    t.CreationTime,\n\t}\n}\n\n// ToFailoverMarkerAttributes converts thrift FailoverMarkerAttributes type to internal\nfunc ToFailoverMarkerAttributes(t *replicator.FailoverMarkerAttributes) *types.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverMarkerAttributes{\n\t\tDomainID:        t.GetDomainID(),\n\t\tFailoverVersion: t.GetFailoverVersion(),\n\t\tCreationTime:    t.CreationTime,\n\t}\n}\n\n// FromFailoverMarkers converts internal FailoverMarkers type to thrift\nfunc FromFailoverMarkers(t *types.FailoverMarkers) *replicator.FailoverMarkers {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.FailoverMarkers{\n\t\tFailoverMarkers: FromFailoverMarkerAttributesArray(t.FailoverMarkers),\n\t}\n}\n\n// ToFailoverMarkers converts thrift FailoverMarkers type to internal\nfunc ToFailoverMarkers(t *replicator.FailoverMarkers) *types.FailoverMarkers {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverMarkers{\n\t\tFailoverMarkers: ToFailoverMarkerAttributesArray(t.FailoverMarkers),\n\t}\n}\n\n// FromAdminGetDLQReplicationMessagesRequest converts internal GetDLQReplicationMessagesRequest type to thrift\nfunc FromAdminGetDLQReplicationMessagesRequest(t *types.GetDLQReplicationMessagesRequest) *replicator.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: FromReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\n// ToAdminGetDLQReplicationMessagesRequest converts thrift GetDLQReplicationMessagesRequest type to internal\nfunc ToAdminGetDLQReplicationMessagesRequest(t *replicator.GetDLQReplicationMessagesRequest) *types.GetDLQReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: ToReplicationTaskInfoArray(t.TaskInfos),\n\t}\n}\n\n// FromAdminGetDLQReplicationMessagesResponse converts internal GetDLQReplicationMessagesResponse type to thrift\nfunc FromAdminGetDLQReplicationMessagesResponse(t *types.GetDLQReplicationMessagesResponse) *replicator.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: FromReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\n// ToAdminGetDLQReplicationMessagesResponse converts thrift GetDLQReplicationMessagesResponse type to internal\nfunc ToAdminGetDLQReplicationMessagesResponse(t *replicator.GetDLQReplicationMessagesResponse) *types.GetDLQReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: ToReplicationTaskArray(t.ReplicationTasks),\n\t}\n}\n\n// FromAdminGetDomainReplicationMessagesRequest converts internal GetDomainReplicationMessagesRequest type to thrift\nfunc FromAdminGetDomainReplicationMessagesRequest(t *types.GetDomainReplicationMessagesRequest) *replicator.GetDomainReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageId: t.LastRetrievedMessageID,\n\t\tLastProcessedMessageId: t.LastProcessedMessageID,\n\t\tClusterName:            &t.ClusterName,\n\t}\n}\n\n// ToAdminGetDomainReplicationMessagesRequest converts thrift GetDomainReplicationMessagesRequest type to internal\nfunc ToAdminGetDomainReplicationMessagesRequest(t *replicator.GetDomainReplicationMessagesRequest) *types.GetDomainReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: t.LastRetrievedMessageId,\n\t\tLastProcessedMessageID: t.LastProcessedMessageId,\n\t\tClusterName:            t.GetClusterName(),\n\t}\n}\n\n// FromAdminGetDomainReplicationMessagesResponse converts internal GetDomainReplicationMessagesResponse type to thrift\nfunc FromAdminGetDomainReplicationMessagesResponse(t *types.GetDomainReplicationMessagesResponse) *replicator.GetDomainReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetDomainReplicationMessagesResponse{\n\t\tMessages: FromReplicationMessages(t.Messages),\n\t}\n}\n\n// ToAdminGetDomainReplicationMessagesResponse converts thrift GetDomainReplicationMessagesResponse type to internal\nfunc ToAdminGetDomainReplicationMessagesResponse(t *replicator.GetDomainReplicationMessagesResponse) *types.GetDomainReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: ToReplicationMessages(t.Messages),\n\t}\n}\n\n// FromAdminGetReplicationMessagesRequest converts internal GetReplicationMessagesRequest type to thrift\nfunc FromAdminGetReplicationMessagesRequest(t *types.GetReplicationMessagesRequest) *replicator.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetReplicationMessagesRequest{\n\t\tTokens:      FromReplicationTokenArray(t.Tokens),\n\t\tClusterName: &t.ClusterName,\n\t}\n}\n\n// ToAdminGetReplicationMessagesRequest converts thrift GetReplicationMessagesRequest type to internal\nfunc ToAdminGetReplicationMessagesRequest(t *replicator.GetReplicationMessagesRequest) *types.GetReplicationMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesRequest{\n\t\tTokens:      ToReplicationTokenArray(t.Tokens),\n\t\tClusterName: t.GetClusterName(),\n\t}\n}\n\n// FromAdminGetReplicationMessagesResponse converts internal GetReplicationMessagesResponse type to thrift\nfunc FromAdminGetReplicationMessagesResponse(t *types.GetReplicationMessagesResponse) *replicator.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.GetReplicationMessagesResponse{\n\t\tMessagesByShard: FromReplicationMessagesMap(t.MessagesByShard),\n\t}\n}\n\n// ToAdminGetReplicationMessagesResponse converts thrift GetReplicationMessagesResponse type to internal\nfunc ToAdminGetReplicationMessagesResponse(t *replicator.GetReplicationMessagesResponse) *types.GetReplicationMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: ToReplicationMessagesMap(t.MessagesByShard),\n\t}\n}\n\n// FromHistoryTaskV2Attributes converts internal HistoryTaskV2Attributes type to thrift\nfunc FromHistoryTaskV2Attributes(t *types.HistoryTaskV2Attributes) *replicator.HistoryTaskV2Attributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.HistoryTaskV2Attributes{\n\t\tDomainId:            &t.DomainID,\n\t\tWorkflowId:          &t.WorkflowID,\n\t\tRunId:               &t.RunID,\n\t\tVersionHistoryItems: FromVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              FromDataBlob(t.Events),\n\t\tNewRunEvents:        FromDataBlob(t.NewRunEvents),\n\t}\n}\n\n// ToHistoryTaskV2Attributes converts thrift HistoryTaskV2Attributes type to internal\nfunc ToHistoryTaskV2Attributes(t *replicator.HistoryTaskV2Attributes) *types.HistoryTaskV2Attributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryTaskV2Attributes{\n\t\tDomainID:            t.GetDomainId(),\n\t\tWorkflowID:          t.GetWorkflowId(),\n\t\tRunID:               t.GetRunId(),\n\t\tVersionHistoryItems: ToVersionHistoryItemArray(t.VersionHistoryItems),\n\t\tEvents:              ToDataBlob(t.Events),\n\t\tNewRunEvents:        ToDataBlob(t.NewRunEvents),\n\t}\n}\n\n// FromAdminMergeDLQMessagesRequest converts internal MergeDLQMessagesRequest type to thrift\nfunc FromAdminMergeDLQMessagesRequest(t *types.MergeDLQMessagesRequest) *replicator.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.MergeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardID:               &t.ShardID,\n\t\tSourceCluster:         &t.SourceCluster,\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t\tMaximumPageSize:       &t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\n// ToAdminMergeDLQMessagesRequest converts thrift MergeDLQMessagesRequest type to internal\nfunc ToAdminMergeDLQMessagesRequest(t *replicator.MergeDLQMessagesRequest) *types.MergeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.GetShardID(),\n\t\tSourceCluster:         t.GetSourceCluster(),\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t\tMaximumPageSize:       t.GetMaximumPageSize(),\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\n// FromAdminMergeDLQMessagesResponse converts internal MergeDLQMessagesResponse type to thrift\nfunc FromAdminMergeDLQMessagesResponse(t *types.MergeDLQMessagesResponse) *replicator.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToAdminMergeDLQMessagesResponse converts thrift MergeDLQMessagesResponse type to internal\nfunc ToAdminMergeDLQMessagesResponse(t *replicator.MergeDLQMessagesResponse) *types.MergeDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MergeDLQMessagesResponse{\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromAdminPurgeDLQMessagesRequest converts internal PurgeDLQMessagesRequest type to thrift\nfunc FromAdminPurgeDLQMessagesRequest(t *types.PurgeDLQMessagesRequest) *replicator.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.PurgeDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardID:               &t.ShardID,\n\t\tSourceCluster:         &t.SourceCluster,\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t}\n}\n\n// ToAdminPurgeDLQMessagesRequest converts thrift PurgeDLQMessagesRequest type to internal\nfunc ToAdminPurgeDLQMessagesRequest(t *replicator.PurgeDLQMessagesRequest) *types.PurgeDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PurgeDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.GetShardID(),\n\t\tSourceCluster:         t.GetSourceCluster(),\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t}\n}\n\n// FromAdminReadDLQMessagesRequest converts internal ReadDLQMessagesRequest type to thrift\nfunc FromAdminReadDLQMessagesRequest(t *types.ReadDLQMessagesRequest) *replicator.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReadDLQMessagesRequest{\n\t\tType:                  FromDLQType(t.Type),\n\t\tShardID:               &t.ShardID,\n\t\tSourceCluster:         &t.SourceCluster,\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t\tMaximumPageSize:       &t.MaximumPageSize,\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\n// ToAdminReadDLQMessagesRequest converts thrift ReadDLQMessagesRequest type to internal\nfunc ToAdminReadDLQMessagesRequest(t *replicator.ReadDLQMessagesRequest) *types.ReadDLQMessagesRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesRequest{\n\t\tType:                  ToDLQType(t.Type),\n\t\tShardID:               t.GetShardID(),\n\t\tSourceCluster:         t.GetSourceCluster(),\n\t\tInclusiveEndMessageID: t.InclusiveEndMessageID,\n\t\tMaximumPageSize:       t.GetMaximumPageSize(),\n\t\tNextPageToken:         t.NextPageToken,\n\t}\n}\n\n// FromAdminReadDLQMessagesResponse converts internal ReadDLQMessagesResponse type to thrift\nfunc FromAdminReadDLQMessagesResponse(t *types.ReadDLQMessagesResponse) *replicator.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReadDLQMessagesResponse{\n\t\tType:                 FromDLQType(t.Type),\n\t\tReplicationTasks:     FromReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: FromReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\n// ToAdminReadDLQMessagesResponse converts thrift ReadDLQMessagesResponse type to internal\nfunc ToAdminReadDLQMessagesResponse(t *replicator.ReadDLQMessagesResponse) *types.ReadDLQMessagesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReadDLQMessagesResponse{\n\t\tType:                 ToDLQType(t.Type),\n\t\tReplicationTasks:     ToReplicationTaskArray(t.ReplicationTasks),\n\t\tReplicationTasksInfo: ToReplicationTaskInfoArray(t.ReplicationTasksInfo),\n\t\tNextPageToken:        t.NextPageToken,\n\t}\n}\n\n// FromReplicationMessages converts internal ReplicationMessages type to thrift\nfunc FromReplicationMessages(t *types.ReplicationMessages) *replicator.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReplicationMessages{\n\t\tReplicationTasks:       FromReplicationTaskArray(t.ReplicationTasks),\n\t\tLastRetrievedMessageId: &t.LastRetrievedMessageID,\n\t\tHasMore:                &t.HasMore,\n\t\tSyncShardStatus:        FromSyncShardStatus(t.SyncShardStatus),\n\t}\n}\n\n// ToReplicationMessages converts thrift ReplicationMessages type to internal\nfunc ToReplicationMessages(t *replicator.ReplicationMessages) *types.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationMessages{\n\t\tReplicationTasks:       ToReplicationTaskArray(t.ReplicationTasks),\n\t\tLastRetrievedMessageID: t.GetLastRetrievedMessageId(),\n\t\tHasMore:                t.GetHasMore(),\n\t\tSyncShardStatus:        ToSyncShardStatus(t.SyncShardStatus),\n\t}\n}\n\n// FromReplicationTask converts internal ReplicationTask type to thrift\nfunc FromReplicationTask(t *types.ReplicationTask) *replicator.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReplicationTask{\n\t\tTaskType:                      FromReplicationTaskType(t.TaskType),\n\t\tSourceTaskId:                  &t.SourceTaskID,\n\t\tDomainTaskAttributes:          FromDomainTaskAttributes(t.DomainTaskAttributes),\n\t\tSyncShardStatusTaskAttributes: FromSyncShardStatusTaskAttributes(t.SyncShardStatusTaskAttributes),\n\t\tSyncActivityTaskAttributes:    FromSyncActivityTaskAttributes(t.SyncActivityTaskAttributes),\n\t\tHistoryTaskV2Attributes:       FromHistoryTaskV2Attributes(t.HistoryTaskV2Attributes),\n\t\tFailoverMarkerAttributes:      FromFailoverMarkerAttributes(t.FailoverMarkerAttributes),\n\t\tCreationTime:                  t.CreationTime,\n\t}\n}\n\n// ToReplicationTask converts thrift ReplicationTask type to internal\nfunc ToReplicationTask(t *replicator.ReplicationTask) *types.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationTask{\n\t\tTaskType:                      ToReplicationTaskType(t.TaskType),\n\t\tSourceTaskID:                  t.GetSourceTaskId(),\n\t\tDomainTaskAttributes:          ToDomainTaskAttributes(t.DomainTaskAttributes),\n\t\tSyncShardStatusTaskAttributes: ToSyncShardStatusTaskAttributes(t.SyncShardStatusTaskAttributes),\n\t\tSyncActivityTaskAttributes:    ToSyncActivityTaskAttributes(t.SyncActivityTaskAttributes),\n\t\tHistoryTaskV2Attributes:       ToHistoryTaskV2Attributes(t.HistoryTaskV2Attributes),\n\t\tFailoverMarkerAttributes:      ToFailoverMarkerAttributes(t.FailoverMarkerAttributes),\n\t\tCreationTime:                  t.CreationTime,\n\t}\n}\n\n// FromReplicationTaskInfo converts internal ReplicationTaskInfo type to thrift\nfunc FromReplicationTaskInfo(t *types.ReplicationTaskInfo) *replicator.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReplicationTaskInfo{\n\t\tDomainID:     &t.DomainID,\n\t\tWorkflowID:   &t.WorkflowID,\n\t\tRunID:        &t.RunID,\n\t\tTaskType:     &t.TaskType,\n\t\tTaskID:       &t.TaskID,\n\t\tVersion:      &t.Version,\n\t\tFirstEventID: &t.FirstEventID,\n\t\tNextEventID:  &t.NextEventID,\n\t\tScheduledID:  &t.ScheduledID,\n\t}\n}\n\n// ToReplicationTaskInfo converts thrift ReplicationTaskInfo type to internal\nfunc ToReplicationTaskInfo(t *replicator.ReplicationTaskInfo) *types.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationTaskInfo{\n\t\tDomainID:     t.GetDomainID(),\n\t\tWorkflowID:   t.GetWorkflowID(),\n\t\tRunID:        t.GetRunID(),\n\t\tTaskType:     t.GetTaskType(),\n\t\tTaskID:       t.GetTaskID(),\n\t\tVersion:      t.GetVersion(),\n\t\tFirstEventID: t.GetFirstEventID(),\n\t\tNextEventID:  t.GetNextEventID(),\n\t\tScheduledID:  t.GetScheduledID(),\n\t}\n}\n\n// FromReplicationTaskType converts internal ReplicationTaskType type to thrift\nfunc FromReplicationTaskType(t *types.ReplicationTaskType) *replicator.ReplicationTaskType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ReplicationTaskTypeDomain:\n\t\tv := replicator.ReplicationTaskTypeDomain\n\t\treturn &v\n\tcase types.ReplicationTaskTypeHistory:\n\t\tv := replicator.ReplicationTaskTypeHistory\n\t\treturn &v\n\tcase types.ReplicationTaskTypeSyncShardStatus:\n\t\tv := replicator.ReplicationTaskTypeSyncShardStatus\n\t\treturn &v\n\tcase types.ReplicationTaskTypeSyncActivity:\n\t\tv := replicator.ReplicationTaskTypeSyncActivity\n\t\treturn &v\n\tcase types.ReplicationTaskTypeHistoryMetadata:\n\t\tv := replicator.ReplicationTaskTypeHistoryMetadata\n\t\treturn &v\n\tcase types.ReplicationTaskTypeHistoryV2:\n\t\tv := replicator.ReplicationTaskTypeHistoryV2\n\t\treturn &v\n\tcase types.ReplicationTaskTypeFailoverMarker:\n\t\tv := replicator.ReplicationTaskTypeFailoverMarker\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToReplicationTaskType converts thrift ReplicationTaskType type to internal\nfunc ToReplicationTaskType(t *replicator.ReplicationTaskType) *types.ReplicationTaskType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase replicator.ReplicationTaskTypeDomain:\n\t\tv := types.ReplicationTaskTypeDomain\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeHistory:\n\t\tv := types.ReplicationTaskTypeHistory\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeSyncShardStatus:\n\t\tv := types.ReplicationTaskTypeSyncShardStatus\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeSyncActivity:\n\t\tv := types.ReplicationTaskTypeSyncActivity\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeHistoryMetadata:\n\t\tv := types.ReplicationTaskTypeHistoryMetadata\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeHistoryV2:\n\t\tv := types.ReplicationTaskTypeHistoryV2\n\t\treturn &v\n\tcase replicator.ReplicationTaskTypeFailoverMarker:\n\t\tv := types.ReplicationTaskTypeFailoverMarker\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromReplicationToken converts internal ReplicationToken type to thrift\nfunc FromReplicationToken(t *types.ReplicationToken) *replicator.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.ReplicationToken{\n\t\tShardID:                &t.ShardID,\n\t\tLastRetrievedMessageId: &t.LastRetrievedMessageID,\n\t\tLastProcessedMessageId: &t.LastProcessedMessageID,\n\t}\n}\n\n// ToReplicationToken converts thrift ReplicationToken type to internal\nfunc ToReplicationToken(t *replicator.ReplicationToken) *types.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReplicationToken{\n\t\tShardID:                t.GetShardID(),\n\t\tLastRetrievedMessageID: t.GetLastRetrievedMessageId(),\n\t\tLastProcessedMessageID: t.GetLastProcessedMessageId(),\n\t}\n}\n\n// FromSyncActivityTaskAttributes converts internal SyncActivityTaskAttributes type to thrift\nfunc FromSyncActivityTaskAttributes(t *types.SyncActivityTaskAttributes) *replicator.SyncActivityTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.SyncActivityTaskAttributes{\n\t\tDomainId:           &t.DomainID,\n\t\tWorkflowId:         &t.WorkflowID,\n\t\tRunId:              &t.RunID,\n\t\tVersion:            &t.Version,\n\t\tScheduledId:        &t.ScheduledID,\n\t\tScheduledTime:      t.ScheduledTime,\n\t\tStartedId:          &t.StartedID,\n\t\tStartedTime:        t.StartedTime,\n\t\tLastHeartbeatTime:  t.LastHeartbeatTime,\n\t\tDetails:            t.Details,\n\t\tAttempt:            &t.Attempt,\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastWorkerIdentity: &t.LastWorkerIdentity,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t\tVersionHistory:     FromVersionHistory(t.VersionHistory),\n\t}\n}\n\n// ToSyncActivityTaskAttributes converts thrift SyncActivityTaskAttributes type to internal\nfunc ToSyncActivityTaskAttributes(t *replicator.SyncActivityTaskAttributes) *types.SyncActivityTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncActivityTaskAttributes{\n\t\tDomainID:           t.GetDomainId(),\n\t\tWorkflowID:         t.GetWorkflowId(),\n\t\tRunID:              t.GetRunId(),\n\t\tVersion:            t.GetVersion(),\n\t\tScheduledID:        t.GetScheduledId(),\n\t\tScheduledTime:      t.ScheduledTime,\n\t\tStartedID:          t.GetStartedId(),\n\t\tStartedTime:        t.StartedTime,\n\t\tLastHeartbeatTime:  t.LastHeartbeatTime,\n\t\tDetails:            t.Details,\n\t\tAttempt:            t.GetAttempt(),\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastWorkerIdentity: t.GetLastWorkerIdentity(),\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t\tVersionHistory:     ToVersionHistory(t.VersionHistory),\n\t}\n}\n\n// FromSyncShardStatus converts internal SyncShardStatus type to thrift\nfunc FromSyncShardStatus(t *types.SyncShardStatus) *replicator.SyncShardStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.SyncShardStatus{\n\t\tTimestamp: t.Timestamp,\n\t}\n}\n\n// ToSyncShardStatus converts thrift SyncShardStatus type to internal\nfunc ToSyncShardStatus(t *replicator.SyncShardStatus) *types.SyncShardStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatus{\n\t\tTimestamp: t.Timestamp,\n\t}\n}\n\n// FromSyncShardStatusTaskAttributes converts internal SyncShardStatusTaskAttributes type to thrift\nfunc FromSyncShardStatusTaskAttributes(t *types.SyncShardStatusTaskAttributes) *replicator.SyncShardStatusTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &replicator.SyncShardStatusTaskAttributes{\n\t\tSourceCluster: &t.SourceCluster,\n\t\tShardId:       &t.ShardID,\n\t\tTimestamp:     t.Timestamp,\n\t}\n}\n\n// ToSyncShardStatusTaskAttributes converts thrift SyncShardStatusTaskAttributes type to internal\nfunc ToSyncShardStatusTaskAttributes(t *replicator.SyncShardStatusTaskAttributes) *types.SyncShardStatusTaskAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SyncShardStatusTaskAttributes{\n\t\tSourceCluster: t.GetSourceCluster(),\n\t\tShardID:       t.GetShardId(),\n\t\tTimestamp:     t.Timestamp,\n\t}\n}\n\n// FromFailoverMarkerAttributesArray converts internal FailoverMarkerAttributes type array to thrift\nfunc FromFailoverMarkerAttributesArray(t []*types.FailoverMarkerAttributes) []*replicator.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*replicator.FailoverMarkerAttributes, len(t))\n\tfor i := range t {\n\t\tv[i] = FromFailoverMarkerAttributes(t[i])\n\t}\n\treturn v\n}\n\n// ToFailoverMarkerAttributesArray converts thrift FailoverMarkerAttributes type array to internal\nfunc ToFailoverMarkerAttributesArray(t []*replicator.FailoverMarkerAttributes) []*types.FailoverMarkerAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.FailoverMarkerAttributes, len(t))\n\tfor i := range t {\n\t\tv[i] = ToFailoverMarkerAttributes(t[i])\n\t}\n\treturn v\n}\n\n// FromReplicationTaskInfoArray converts internal ReplicationTaskInfo type array to thrift\nfunc FromReplicationTaskInfoArray(t []*types.ReplicationTaskInfo) []*replicator.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*replicator.ReplicationTaskInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationTaskInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToReplicationTaskInfoArray converts thrift ReplicationTaskInfo type array to internal\nfunc ToReplicationTaskInfoArray(t []*replicator.ReplicationTaskInfo) []*types.ReplicationTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationTaskInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationTaskInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromReplicationTaskArray converts internal ReplicationTask type array to thrift\nfunc FromReplicationTaskArray(t []*types.ReplicationTask) []*replicator.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*replicator.ReplicationTask, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationTask(t[i])\n\t}\n\treturn v\n}\n\n// ToReplicationTaskArray converts thrift ReplicationTask type array to internal\nfunc ToReplicationTaskArray(t []*replicator.ReplicationTask) []*types.ReplicationTask {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationTask, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationTask(t[i])\n\t}\n\treturn v\n}\n\n// FromReplicationTokenArray converts internal ReplicationToken type array to thrift\nfunc FromReplicationTokenArray(t []*types.ReplicationToken) []*replicator.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*replicator.ReplicationToken, len(t))\n\tfor i := range t {\n\t\tv[i] = FromReplicationToken(t[i])\n\t}\n\treturn v\n}\n\n// ToReplicationTokenArray converts thrift ReplicationToken type array to internal\nfunc ToReplicationTokenArray(t []*replicator.ReplicationToken) []*types.ReplicationToken {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ReplicationToken, len(t))\n\tfor i := range t {\n\t\tv[i] = ToReplicationToken(t[i])\n\t}\n\treturn v\n}\n\n// FromReplicationMessagesMap converts internal ReplicationMessages type map to thrift\nfunc FromReplicationMessagesMap(t map[int32]*types.ReplicationMessages) map[int32]*replicator.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]*replicator.ReplicationMessages, len(t))\n\tfor key := range t {\n\t\tv[key] = FromReplicationMessages(t[key])\n\t}\n\treturn v\n}\n\n// ToReplicationMessagesMap converts thrift ReplicationMessages type map to internal\nfunc ToReplicationMessagesMap(t map[int32]*replicator.ReplicationMessages) map[int32]*types.ReplicationMessages {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]*types.ReplicationMessages, len(t))\n\tfor key := range t {\n\t\tv[key] = ToReplicationMessages(t[key])\n\t}\n\treturn v\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/replicator_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestDLQType(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DLQType\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: types.DLQTypeReplication.Ptr(),\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromDLQType(tc.input)\n\t\t\troundTripObj := ToDLQType(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestDomainOperation(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DomainOperation\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: types.DomainOperationCreate.Ptr(),\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromDomainOperation(tc.input)\n\t\t\troundTripObj := ToDomainOperation(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestDomainTaskAttributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.DomainTaskAttributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.DomainTaskAttributes,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.DomainTaskAttributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromDomainTaskAttributes(tc.input)\n\t\t\troundTripObj := ToDomainTaskAttributes(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestFailoverMarkerAttributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.FailoverMarkerAttributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.FailoverMarkerAttributes,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.FailoverMarkerAttributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromFailoverMarkerAttributes(tc.input)\n\t\t\troundTripObj := ToFailoverMarkerAttributes(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestFailoverMarkers(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.FailoverMarkers\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &types.FailoverMarkers{FailoverMarkers: testdata.FailoverMarkerAttributesArray},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.FailoverMarkers{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromFailoverMarkers(tc.input)\n\t\t\troundTripObj := ToFailoverMarkers(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDLQReplicationMessagesRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetDLQReplicationMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.GetDLQReplicationMessagesRequest{\n\t\t\t\tTaskInfos: testdata.ReplicationTaskInfoArray,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetDLQReplicationMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetDLQReplicationMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminGetDLQReplicationMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDLQReplicationMessagesResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetDLQReplicationMessagesResponse\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.GetDLQReplicationMessagesResponse{\n\t\t\t\tReplicationTasks: testdata.ReplicationTaskArray,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetDLQReplicationMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetDLQReplicationMessagesResponse(tc.input)\n\t\t\troundTripObj := ToAdminGetDLQReplicationMessagesResponse(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDomainReplicationMessageRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetDomainReplicationMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.GetDomainReplicationMessagesRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetDomainReplicationMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetDomainReplicationMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminGetDomainReplicationMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDomainReplicationMessageResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetDomainReplicationMessagesResponse\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.GetDomainReplicationMessagesResponse{\n\t\t\t\tMessages: &testdata.ReplicationMessages,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetDomainReplicationMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetDomainReplicationMessagesResponse(tc.input)\n\t\t\troundTripObj := ToAdminGetDomainReplicationMessagesResponse(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetReplicationMessageRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetReplicationMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.GetReplicationMessagesRequest{\n\t\t\t\tTokens:      testdata.ReplicationTokenArray,\n\t\t\t\tClusterName: testdata.ClusterName1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetReplicationMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetReplicationMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminGetReplicationMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminGetReplicationMessageResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.GetReplicationMessagesResponse\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: testdata.ReplicationMessagesMap,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.GetReplicationMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminGetReplicationMessagesResponse(tc.input)\n\t\t\troundTripObj := ToAdminGetReplicationMessagesResponse(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestHistoryTaskV2Attributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.HistoryTaskV2Attributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.HistoryTaskV2Attributes,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.HistoryTaskV2Attributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromHistoryTaskV2Attributes(tc.input)\n\t\t\troundTripObj := ToHistoryTaskV2Attributes(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminMergeDLQMessagesRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MergeDLQMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.MergeDLQMessagesRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MergeDLQMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminMergeDLQMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminMergeDLQMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminMergeDLQMessagesResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.MergeDLQMessagesResponse\n\t}{\n\t\t{\n\t\t\tdesc: \"non-nil input test\",\n\t\t\tinput: &types.MergeDLQMessagesResponse{\n\t\t\t\tNextPageToken: testdata.Token1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.MergeDLQMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminMergeDLQMessagesResponse(tc.input)\n\t\t\troundTripObj := ToAdminMergeDLQMessagesResponse(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminPurgeDLQMessagesRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.PurgeDLQMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.PurgeDLQMessagesRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.PurgeDLQMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminPurgeDLQMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminPurgeDLQMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminReadDLQMessagesRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReadDLQMessagesRequest\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReadDLQMessagesRequest,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReadDLQMessagesRequest{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminReadDLQMessagesRequest(tc.input)\n\t\t\troundTripObj := ToAdminReadDLQMessagesRequest(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestAdminReadDLQMessagesResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReadDLQMessagesResponse\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReadDLQMessagesResponse,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReadDLQMessagesResponse{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromAdminReadDLQMessagesResponse(tc.input)\n\t\t\troundTripObj := ToAdminReadDLQMessagesResponse(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationMessages(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReplicationMessages\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReplicationMessages,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReplicationMessages{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationMessages(tc.input)\n\t\t\troundTripObj := ToReplicationMessages(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTask(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReplicationTask\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReplicationTask_Domain,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReplicationTask{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTask(tc.input)\n\t\t\troundTripObj := ToReplicationTask(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTaskInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReplicationTaskInfo\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReplicationTaskInfo,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReplicationTaskInfo{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTaskInfo(tc.input)\n\t\t\troundTripObj := ToReplicationTaskInfo(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTaskType(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReplicationTaskType\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTaskType(tc.input)\n\t\t\troundTripObj := ToReplicationTaskType(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationToken(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.ReplicationToken\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.ReplicationToken,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.ReplicationToken{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationToken(tc.input)\n\t\t\troundTripObj := ToReplicationToken(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestSyncActivityTaskAttributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.SyncActivityTaskAttributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.SyncActivityTaskAttributes,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.SyncActivityTaskAttributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromSyncActivityTaskAttributes(tc.input)\n\t\t\troundTripObj := ToSyncActivityTaskAttributes(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestSyncShardStatus(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.SyncShardStatus\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.SyncShardStatus,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.SyncShardStatus{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromSyncShardStatus(tc.input)\n\t\t\troundTripObj := ToSyncShardStatus(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestSyncShardStatusTaskAttributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput *types.SyncShardStatusTaskAttributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: &testdata.SyncShardStatusTaskAttributes,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: &types.SyncShardStatusTaskAttributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromSyncShardStatusTaskAttributes(tc.input)\n\t\t\troundTripObj := ToSyncShardStatusTaskAttributes(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestFailoverMarkerAttributesArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.FailoverMarkerAttributes\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: testdata.FailoverMarkerAttributesArray,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.FailoverMarkerAttributes{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromFailoverMarkerAttributesArray(tc.input)\n\t\t\troundTripObj := ToFailoverMarkerAttributesArray(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTaskInfoArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.ReplicationTaskInfo\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: testdata.ReplicationTaskInfoArray,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.ReplicationTaskInfo{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTaskInfoArray(tc.input)\n\t\t\troundTripObj := ToReplicationTaskInfoArray(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTaskArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.ReplicationTask\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: testdata.ReplicationTaskArray,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.ReplicationTask{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTaskArray(tc.input)\n\t\t\troundTripObj := ToReplicationTaskArray(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationTokenArray(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput []*types.ReplicationToken\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: testdata.ReplicationTokenArray,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: []*types.ReplicationToken{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationTokenArray(tc.input)\n\t\t\troundTripObj := ToReplicationTokenArray(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n\nfunc TestReplicationMessagesMap(t *testing.T) {\n\ttestCases := []struct {\n\t\tdesc  string\n\t\tinput map[int32]*types.ReplicationMessages\n\t}{\n\t\t{\n\t\t\tdesc:  \"non-nil input test\",\n\t\t\tinput: testdata.ReplicationMessagesMap,\n\t\t},\n\t\t{\n\t\t\tdesc:  \"empty input test\",\n\t\t\tinput: map[int32]*types.ReplicationMessages{},\n\t\t},\n\t\t{\n\t\t\tdesc:  \"nil input test\",\n\t\t\tinput: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tthriftObj := FromReplicationMessagesMap(tc.input)\n\t\t\troundTripObj := ToReplicationMessagesMap(thriftObj)\n\t\t\tassert.Equal(t, tc.input, roundTripObj)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/shared.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tFromScanWorkflowExecutionsRequest            = FromListWorkflowExecutionsRequest\n\tToScanWorkflowExecutionsRequest              = ToListWorkflowExecutionsRequest\n\tFromRecordActivityTaskHeartbeatByIDResponse  = FromRecordActivityTaskHeartbeatResponse\n\tToRecordActivityTaskHeartbeatByIDResponse    = ToRecordActivityTaskHeartbeatResponse\n\tFromScanWorkflowExecutionsResponse           = FromListWorkflowExecutionsResponse\n\tToScanWorkflowExecutionsResponse             = ToListWorkflowExecutionsResponse\n\tFromSignalWithStartWorkflowExecutionResponse = FromStartWorkflowExecutionResponse\n\tToSignalWithStartWorkflowExecutionResponse   = ToStartWorkflowExecutionResponse\n)\n\n// FromAccessDeniedError converts internal AccessDeniedError type to thrift\nfunc FromAccessDeniedError(t *types.AccessDeniedError) *shared.AccessDeniedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.AccessDeniedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToAccessDeniedError converts thrift AccessDeniedError type to internal\nfunc ToAccessDeniedError(t *shared.AccessDeniedError) *types.AccessDeniedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AccessDeniedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromActivityLocalDispatchInfo converts internal ActivityLocalDispatchInfo type to thrift\nfunc FromActivityLocalDispatchInfo(t *types.ActivityLocalDispatchInfo) *shared.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityLocalDispatchInfo{\n\t\tActivityId:                      &t.ActivityID,\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tTaskToken:                       t.TaskToken,\n\t}\n}\n\n// ToActivityLocalDispatchInfo converts thrift ActivityLocalDispatchInfo type to internal\nfunc ToActivityLocalDispatchInfo(t *shared.ActivityLocalDispatchInfo) *types.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityLocalDispatchInfo{\n\t\tActivityID:                      t.GetActivityId(),\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tTaskToken:                       t.TaskToken,\n\t}\n}\n\n// FromActivityTaskCancelRequestedEventAttributes converts internal ActivityTaskCancelRequestedEventAttributes type to thrift\nfunc FromActivityTaskCancelRequestedEventAttributes(t *types.ActivityTaskCancelRequestedEventAttributes) *shared.ActivityTaskCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityId:                   &t.ActivityID,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToActivityTaskCancelRequestedEventAttributes converts thrift ActivityTaskCancelRequestedEventAttributes type to internal\nfunc ToActivityTaskCancelRequestedEventAttributes(t *shared.ActivityTaskCancelRequestedEventAttributes) *types.ActivityTaskCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityID:                   t.GetActivityId(),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromActivityTaskCanceledEventAttributes converts internal ActivityTaskCanceledEventAttributes type to thrift\nfunc FromActivityTaskCanceledEventAttributes(t *types.ActivityTaskCanceledEventAttributes) *shared.ActivityTaskCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskCanceledEventAttributes{\n\t\tDetails:                      t.Details,\n\t\tLatestCancelRequestedEventId: &t.LatestCancelRequestedEventID,\n\t\tScheduledEventId:             &t.ScheduledEventID,\n\t\tStartedEventId:               &t.StartedEventID,\n\t\tIdentity:                     &t.Identity,\n\t}\n}\n\n// ToActivityTaskCanceledEventAttributes converts thrift ActivityTaskCanceledEventAttributes type to internal\nfunc ToActivityTaskCanceledEventAttributes(t *shared.ActivityTaskCanceledEventAttributes) *types.ActivityTaskCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCanceledEventAttributes{\n\t\tDetails:                      t.Details,\n\t\tLatestCancelRequestedEventID: t.GetLatestCancelRequestedEventId(),\n\t\tScheduledEventID:             t.GetScheduledEventId(),\n\t\tStartedEventID:               t.GetStartedEventId(),\n\t\tIdentity:                     t.GetIdentity(),\n\t}\n}\n\n// FromActivityTaskCompletedEventAttributes converts internal ActivityTaskCompletedEventAttributes type to thrift\nfunc FromActivityTaskCompletedEventAttributes(t *types.ActivityTaskCompletedEventAttributes) *shared.ActivityTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskCompletedEventAttributes{\n\t\tResult:           t.Result,\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tStartedEventId:   &t.StartedEventID,\n\t\tIdentity:         &t.Identity,\n\t}\n}\n\n// ToActivityTaskCompletedEventAttributes converts thrift ActivityTaskCompletedEventAttributes type to internal\nfunc ToActivityTaskCompletedEventAttributes(t *shared.ActivityTaskCompletedEventAttributes) *types.ActivityTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskCompletedEventAttributes{\n\t\tResult:           t.Result,\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tStartedEventID:   t.GetStartedEventId(),\n\t\tIdentity:         t.GetIdentity(),\n\t}\n}\n\n// FromActivityTaskFailedEventAttributes converts internal ActivityTaskFailedEventAttributes type to thrift\nfunc FromActivityTaskFailedEventAttributes(t *types.ActivityTaskFailedEventAttributes) *shared.ActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskFailedEventAttributes{\n\t\tReason:           t.Reason,\n\t\tDetails:          t.Details,\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tStartedEventId:   &t.StartedEventID,\n\t\tIdentity:         &t.Identity,\n\t}\n}\n\n// ToActivityTaskFailedEventAttributes converts thrift ActivityTaskFailedEventAttributes type to internal\nfunc ToActivityTaskFailedEventAttributes(t *shared.ActivityTaskFailedEventAttributes) *types.ActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskFailedEventAttributes{\n\t\tReason:           t.Reason,\n\t\tDetails:          t.Details,\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tStartedEventID:   t.GetStartedEventId(),\n\t\tIdentity:         t.GetIdentity(),\n\t}\n}\n\n// FromActivityTaskScheduledEventAttributes converts internal ActivityTaskScheduledEventAttributes type to thrift\nfunc FromActivityTaskScheduledEventAttributes(t *types.ActivityTaskScheduledEventAttributes) *shared.ActivityTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskScheduledEventAttributes{\n\t\tActivityId:                    &t.ActivityID,\n\t\tActivityType:                  FromActivityType(t.ActivityType),\n\t\tDomain:                        t.Domain,\n\t\tTaskList:                      FromTaskList(t.TaskList),\n\t\tInput:                         t.Input,\n\t\tScheduleToCloseTimeoutSeconds: t.ScheduleToCloseTimeoutSeconds,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       t.HeartbeatTimeoutSeconds,\n\t\tDecisionTaskCompletedEventId:  &t.DecisionTaskCompletedEventID,\n\t\tRetryPolicy:                   FromRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        FromHeader(t.Header),\n\t}\n}\n\n// ToActivityTaskScheduledEventAttributes converts thrift ActivityTaskScheduledEventAttributes type to internal\nfunc ToActivityTaskScheduledEventAttributes(t *shared.ActivityTaskScheduledEventAttributes) *types.ActivityTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskScheduledEventAttributes{\n\t\tActivityID:                    t.GetActivityId(),\n\t\tActivityType:                  ToActivityType(t.ActivityType),\n\t\tDomain:                        t.Domain,\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tInput:                         t.Input,\n\t\tScheduleToCloseTimeoutSeconds: t.ScheduleToCloseTimeoutSeconds,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       t.HeartbeatTimeoutSeconds,\n\t\tDecisionTaskCompletedEventID:  t.GetDecisionTaskCompletedEventId(),\n\t\tRetryPolicy:                   ToRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        ToHeader(t.Header),\n\t}\n}\n\n// FromActivityTaskStartedEventAttributes converts internal ActivityTaskStartedEventAttributes type to thrift\nfunc FromActivityTaskStartedEventAttributes(t *types.ActivityTaskStartedEventAttributes) *shared.ActivityTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventId:   &t.ScheduledEventID,\n\t\tIdentity:           &t.Identity,\n\t\tRequestId:          &t.RequestID,\n\t\tAttempt:            &t.Attempt,\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t}\n}\n\n// ToActivityTaskStartedEventAttributes converts thrift ActivityTaskStartedEventAttributes type to internal\nfunc ToActivityTaskStartedEventAttributes(t *shared.ActivityTaskStartedEventAttributes) *types.ActivityTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventID:   t.GetScheduledEventId(),\n\t\tIdentity:           t.GetIdentity(),\n\t\tRequestID:          t.GetRequestId(),\n\t\tAttempt:            t.GetAttempt(),\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t}\n}\n\n// FromActivityTaskTimedOutEventAttributes converts internal ActivityTaskTimedOutEventAttributes type to thrift\nfunc FromActivityTaskTimedOutEventAttributes(t *types.ActivityTaskTimedOutEventAttributes) *shared.ActivityTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityTaskTimedOutEventAttributes{\n\t\tDetails:            t.Details,\n\t\tScheduledEventId:   &t.ScheduledEventID,\n\t\tStartedEventId:     &t.StartedEventID,\n\t\tTimeoutType:        FromTimeoutType(t.TimeoutType),\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t}\n}\n\n// ToActivityTaskTimedOutEventAttributes converts thrift ActivityTaskTimedOutEventAttributes type to internal\nfunc ToActivityTaskTimedOutEventAttributes(t *shared.ActivityTaskTimedOutEventAttributes) *types.ActivityTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityTaskTimedOutEventAttributes{\n\t\tDetails:            t.Details,\n\t\tScheduledEventID:   t.GetScheduledEventId(),\n\t\tStartedEventID:     t.GetStartedEventId(),\n\t\tTimeoutType:        ToTimeoutType(t.TimeoutType),\n\t\tLastFailureReason:  t.LastFailureReason,\n\t\tLastFailureDetails: t.LastFailureDetails,\n\t}\n}\n\n// FromActivityType converts internal ActivityType type to thrift\nfunc FromActivityType(t *types.ActivityType) *shared.ActivityType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActivityType{\n\t\tName: &t.Name,\n\t}\n}\n\n// ToActivityType converts thrift ActivityType type to internal\nfunc ToActivityType(t *shared.ActivityType) *types.ActivityType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActivityType{\n\t\tName: t.GetName(),\n\t}\n}\n\n// FromArchivalStatus converts internal ArchivalStatus type to thrift\nfunc FromArchivalStatus(t *types.ArchivalStatus) *shared.ArchivalStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ArchivalStatusDisabled:\n\t\tv := shared.ArchivalStatusDisabled\n\t\treturn &v\n\tcase types.ArchivalStatusEnabled:\n\t\tv := shared.ArchivalStatusEnabled\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToArchivalStatus converts thrift ArchivalStatus type to internal\nfunc ToArchivalStatus(t *shared.ArchivalStatus) *types.ArchivalStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.ArchivalStatusDisabled:\n\t\tv := types.ArchivalStatusDisabled\n\t\treturn &v\n\tcase shared.ArchivalStatusEnabled:\n\t\tv := types.ArchivalStatusEnabled\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromBadBinaries converts internal BadBinaries type to thrift\nfunc FromBadBinaries(t *types.BadBinaries) *shared.BadBinaries {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.BadBinaries{\n\t\tBinaries: FromBadBinaryInfoMap(t.Binaries),\n\t}\n}\n\n// ToBadBinaries converts thrift BadBinaries type to internal\nfunc ToBadBinaries(t *shared.BadBinaries) *types.BadBinaries {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BadBinaries{\n\t\tBinaries: ToBadBinaryInfoMap(t.Binaries),\n\t}\n}\n\n// FromBadBinaryInfo converts internal BadBinaryInfo type to thrift\nfunc FromBadBinaryInfo(t *types.BadBinaryInfo) *shared.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.BadBinaryInfo{\n\t\tReason:          &t.Reason,\n\t\tOperator:        &t.Operator,\n\t\tCreatedTimeNano: t.CreatedTimeNano,\n\t}\n}\n\n// ToBadBinaryInfo converts thrift BadBinaryInfo type to internal\nfunc ToBadBinaryInfo(t *shared.BadBinaryInfo) *types.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BadBinaryInfo{\n\t\tReason:          t.GetReason(),\n\t\tOperator:        t.GetOperator(),\n\t\tCreatedTimeNano: t.CreatedTimeNano,\n\t}\n}\n\n// FromBadRequestError converts internal BadRequestError type to thrift\nfunc FromBadRequestError(t *types.BadRequestError) *shared.BadRequestError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.BadRequestError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToBadRequestError converts thrift BadRequestError type to internal\nfunc ToBadRequestError(t *shared.BadRequestError) *types.BadRequestError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.BadRequestError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromCancelExternalWorkflowExecutionFailedCause converts internal CancelExternalWorkflowExecutionFailedCause type to thrift\nfunc FromCancelExternalWorkflowExecutionFailedCause(t *types.CancelExternalWorkflowExecutionFailedCause) *shared.CancelExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\tv := shared.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn &v\n\tcase types.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\tv := shared.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToCancelExternalWorkflowExecutionFailedCause converts thrift CancelExternalWorkflowExecutionFailedCause type to internal\nfunc ToCancelExternalWorkflowExecutionFailedCause(t *shared.CancelExternalWorkflowExecutionFailedCause) *types.CancelExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\tv := types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn &v\n\tcase shared.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\tv := types.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromCancelTimerDecisionAttributes converts internal CancelTimerDecisionAttributes type to thrift\nfunc FromCancelTimerDecisionAttributes(t *types.CancelTimerDecisionAttributes) *shared.CancelTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CancelTimerDecisionAttributes{\n\t\tTimerId: &t.TimerID,\n\t}\n}\n\n// ToCancelTimerDecisionAttributes converts thrift CancelTimerDecisionAttributes type to internal\nfunc ToCancelTimerDecisionAttributes(t *shared.CancelTimerDecisionAttributes) *types.CancelTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelTimerDecisionAttributes{\n\t\tTimerID: t.GetTimerId(),\n\t}\n}\n\n// FromCancelTimerFailedEventAttributes converts internal CancelTimerFailedEventAttributes type to thrift\nfunc FromCancelTimerFailedEventAttributes(t *types.CancelTimerFailedEventAttributes) *shared.CancelTimerFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CancelTimerFailedEventAttributes{\n\t\tTimerId:                      &t.TimerID,\n\t\tCause:                        &t.Cause,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tIdentity:                     &t.Identity,\n\t}\n}\n\n// ToCancelTimerFailedEventAttributes converts thrift CancelTimerFailedEventAttributes type to internal\nfunc ToCancelTimerFailedEventAttributes(t *shared.CancelTimerFailedEventAttributes) *types.CancelTimerFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelTimerFailedEventAttributes{\n\t\tTimerID:                      t.GetTimerId(),\n\t\tCause:                        t.GetCause(),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tIdentity:                     t.GetIdentity(),\n\t}\n}\n\n// FromCancelWorkflowExecutionDecisionAttributes converts internal CancelWorkflowExecutionDecisionAttributes type to thrift\nfunc FromCancelWorkflowExecutionDecisionAttributes(t *types.CancelWorkflowExecutionDecisionAttributes) *shared.CancelWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: t.Details,\n\t}\n}\n\n// ToCancelWorkflowExecutionDecisionAttributes converts thrift CancelWorkflowExecutionDecisionAttributes type to internal\nfunc ToCancelWorkflowExecutionDecisionAttributes(t *shared.CancelWorkflowExecutionDecisionAttributes) *types.CancelWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: t.Details,\n\t}\n}\n\n// FromCancellationAlreadyRequestedError converts internal CancellationAlreadyRequestedError type to thrift\nfunc FromCancellationAlreadyRequestedError(t *types.CancellationAlreadyRequestedError) *shared.CancellationAlreadyRequestedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CancellationAlreadyRequestedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToCancellationAlreadyRequestedError converts thrift CancellationAlreadyRequestedError type to internal\nfunc ToCancellationAlreadyRequestedError(t *shared.CancellationAlreadyRequestedError) *types.CancellationAlreadyRequestedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CancellationAlreadyRequestedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromChildWorkflowExecutionCanceledEventAttributes converts internal ChildWorkflowExecutionCanceledEventAttributes type to thrift\nfunc FromChildWorkflowExecutionCanceledEventAttributes(t *types.ChildWorkflowExecutionCanceledEventAttributes) *shared.ChildWorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDetails:           t.Details,\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tStartedEventId:    &t.StartedEventID,\n\t}\n}\n\n// ToChildWorkflowExecutionCanceledEventAttributes converts thrift ChildWorkflowExecutionCanceledEventAttributes type to internal\nfunc ToChildWorkflowExecutionCanceledEventAttributes(t *shared.ChildWorkflowExecutionCanceledEventAttributes) *types.ChildWorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDetails:           t.Details,\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tStartedEventID:    t.GetStartedEventId(),\n\t}\n}\n\n// FromChildWorkflowExecutionCompletedEventAttributes converts internal ChildWorkflowExecutionCompletedEventAttributes type to thrift\nfunc FromChildWorkflowExecutionCompletedEventAttributes(t *types.ChildWorkflowExecutionCompletedEventAttributes) *shared.ChildWorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tResult:            t.Result,\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tStartedEventId:    &t.StartedEventID,\n\t}\n}\n\n// ToChildWorkflowExecutionCompletedEventAttributes converts thrift ChildWorkflowExecutionCompletedEventAttributes type to internal\nfunc ToChildWorkflowExecutionCompletedEventAttributes(t *shared.ChildWorkflowExecutionCompletedEventAttributes) *types.ChildWorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tResult:            t.Result,\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tStartedEventID:    t.GetStartedEventId(),\n\t}\n}\n\n// FromChildWorkflowExecutionFailedCause converts internal ChildWorkflowExecutionFailedCause type to thrift\nfunc FromChildWorkflowExecutionFailedCause(t *types.ChildWorkflowExecutionFailedCause) *shared.ChildWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning:\n\t\tv := shared.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToChildWorkflowExecutionFailedCause converts thrift ChildWorkflowExecutionFailedCause type to internal\nfunc ToChildWorkflowExecutionFailedCause(t *shared.ChildWorkflowExecutionFailedCause) *types.ChildWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning:\n\t\tv := types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromChildWorkflowExecutionFailedEventAttributes converts internal ChildWorkflowExecutionFailedEventAttributes type to thrift\nfunc FromChildWorkflowExecutionFailedEventAttributes(t *types.ChildWorkflowExecutionFailedEventAttributes) *shared.ChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionFailedEventAttributes{\n\t\tReason:            t.Reason,\n\t\tDetails:           t.Details,\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tStartedEventId:    &t.StartedEventID,\n\t}\n}\n\n// ToChildWorkflowExecutionFailedEventAttributes converts thrift ChildWorkflowExecutionFailedEventAttributes type to internal\nfunc ToChildWorkflowExecutionFailedEventAttributes(t *shared.ChildWorkflowExecutionFailedEventAttributes) *types.ChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\tReason:            t.Reason,\n\t\tDetails:           t.Details,\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tStartedEventID:    t.GetStartedEventId(),\n\t}\n}\n\n// FromChildWorkflowExecutionStartedEventAttributes converts internal ChildWorkflowExecutionStartedEventAttributes type to thrift\nfunc FromChildWorkflowExecutionStartedEventAttributes(t *types.ChildWorkflowExecutionStartedEventAttributes) *shared.ChildWorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            &t.Domain,\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tHeader:            FromHeader(t.Header),\n\t}\n}\n\n// ToChildWorkflowExecutionStartedEventAttributes converts thrift ChildWorkflowExecutionStartedEventAttributes type to internal\nfunc ToChildWorkflowExecutionStartedEventAttributes(t *shared.ChildWorkflowExecutionStartedEventAttributes) *types.ChildWorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            t.GetDomain(),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tHeader:            ToHeader(t.Header),\n\t}\n}\n\n// FromChildWorkflowExecutionTerminatedEventAttributes converts internal ChildWorkflowExecutionTerminatedEventAttributes type to thrift\nfunc FromChildWorkflowExecutionTerminatedEventAttributes(t *types.ChildWorkflowExecutionTerminatedEventAttributes) *shared.ChildWorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tStartedEventId:    &t.StartedEventID,\n\t}\n}\n\n// ToChildWorkflowExecutionTerminatedEventAttributes converts thrift ChildWorkflowExecutionTerminatedEventAttributes type to internal\nfunc ToChildWorkflowExecutionTerminatedEventAttributes(t *shared.ChildWorkflowExecutionTerminatedEventAttributes) *types.ChildWorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tStartedEventID:    t.GetStartedEventId(),\n\t}\n}\n\n// FromChildWorkflowExecutionTimedOutEventAttributes converts internal ChildWorkflowExecutionTimedOutEventAttributes type to thrift\nfunc FromChildWorkflowExecutionTimedOutEventAttributes(t *types.ChildWorkflowExecutionTimedOutEventAttributes) *shared.ChildWorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType:       FromTimeoutType(t.TimeoutType),\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      FromWorkflowType(t.WorkflowType),\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tStartedEventId:    &t.StartedEventID,\n\t}\n}\n\n// ToChildWorkflowExecutionTimedOutEventAttributes converts thrift ChildWorkflowExecutionTimedOutEventAttributes type to internal\nfunc ToChildWorkflowExecutionTimedOutEventAttributes(t *shared.ChildWorkflowExecutionTimedOutEventAttributes) *types.ChildWorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType:       ToTimeoutType(t.TimeoutType),\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:      ToWorkflowType(t.WorkflowType),\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tStartedEventID:    t.GetStartedEventId(),\n\t}\n}\n\n// FromClientVersionNotSupportedError converts internal ClientVersionNotSupportedError type to thrift\nfunc FromClientVersionNotSupportedError(t *types.ClientVersionNotSupportedError) *shared.ClientVersionNotSupportedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ClientVersionNotSupportedError{\n\t\tFeatureVersion:    t.FeatureVersion,\n\t\tClientImpl:        t.ClientImpl,\n\t\tSupportedVersions: t.SupportedVersions,\n\t}\n}\n\n// ToClientVersionNotSupportedError converts thrift ClientVersionNotSupportedError type to internal\nfunc ToClientVersionNotSupportedError(t *shared.ClientVersionNotSupportedError) *types.ClientVersionNotSupportedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClientVersionNotSupportedError{\n\t\tFeatureVersion:    t.FeatureVersion,\n\t\tClientImpl:        t.ClientImpl,\n\t\tSupportedVersions: t.SupportedVersions,\n\t}\n}\n\n// FromFeatureNotEnabledError converts internal FeatureNotEnabledError type to thrift\nfunc FromFeatureNotEnabledError(t *types.FeatureNotEnabledError) *shared.FeatureNotEnabledError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FeatureNotEnabledError{\n\t\tFeatureFlag: t.FeatureFlag,\n\t}\n}\n\n// ToFeatureNotEnabledError converts thrift FeatureNotEnabledError type to internal\nfunc ToFeatureNotEnabledError(t *shared.FeatureNotEnabledError) *types.FeatureNotEnabledError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FeatureNotEnabledError{\n\t\tFeatureFlag: t.FeatureFlag,\n\t}\n}\n\n// FromAdminCloseShardRequest converts internal CloseShardRequest type to thrift\nfunc FromAdminCloseShardRequest(t *types.CloseShardRequest) *shared.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CloseShardRequest{\n\t\tShardID: &t.ShardID,\n\t}\n}\n\n// ToAdminCloseShardRequest converts thrift CloseShardRequest type to internal\nfunc ToAdminCloseShardRequest(t *shared.CloseShardRequest) *types.CloseShardRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CloseShardRequest{\n\t\tShardID: t.GetShardID(),\n\t}\n}\n\n// FromGetClusterInfoResponse converts internal ClusterInfo type to thrift\nfunc FromGetClusterInfoResponse(t *types.ClusterInfo) *shared.ClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ClusterInfo{\n\t\tSupportedClientVersions: FromSupportedClientVersions(t.SupportedClientVersions),\n\t}\n}\n\n// ToGetClusterInfoResponse converts thrift ClusterInfo type to internal\nfunc ToGetClusterInfoResponse(t *shared.ClusterInfo) *types.ClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterInfo{\n\t\tSupportedClientVersions: ToSupportedClientVersions(t.SupportedClientVersions),\n\t}\n}\n\n// FromClusterReplicationConfiguration converts internal ClusterReplicationConfiguration type to thrift\nfunc FromClusterReplicationConfiguration(t *types.ClusterReplicationConfiguration) *shared.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ClusterReplicationConfiguration{\n\t\tClusterName: &t.ClusterName,\n\t}\n}\n\n// ToClusterReplicationConfiguration converts thrift ClusterReplicationConfiguration type to internal\nfunc ToClusterReplicationConfiguration(t *shared.ClusterReplicationConfiguration) *types.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterReplicationConfiguration{\n\t\tClusterName: t.GetClusterName(),\n\t}\n}\n\n// FromCompleteWorkflowExecutionDecisionAttributes converts internal CompleteWorkflowExecutionDecisionAttributes type to thrift\nfunc FromCompleteWorkflowExecutionDecisionAttributes(t *types.CompleteWorkflowExecutionDecisionAttributes) *shared.CompleteWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: t.Result,\n\t}\n}\n\n// ToCompleteWorkflowExecutionDecisionAttributes converts thrift CompleteWorkflowExecutionDecisionAttributes type to internal\nfunc ToCompleteWorkflowExecutionDecisionAttributes(t *shared.CompleteWorkflowExecutionDecisionAttributes) *types.CompleteWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: t.Result,\n\t}\n}\n\n// FromContinueAsNewInitiator converts internal ContinueAsNewInitiator type to thrift\nfunc FromContinueAsNewInitiator(t *types.ContinueAsNewInitiator) *shared.ContinueAsNewInitiator {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ContinueAsNewInitiatorDecider:\n\t\tv := shared.ContinueAsNewInitiatorDecider\n\t\treturn &v\n\tcase types.ContinueAsNewInitiatorRetryPolicy:\n\t\tv := shared.ContinueAsNewInitiatorRetryPolicy\n\t\treturn &v\n\tcase types.ContinueAsNewInitiatorCronSchedule:\n\t\tv := shared.ContinueAsNewInitiatorCronSchedule\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToContinueAsNewInitiator converts thrift ContinueAsNewInitiator type to internal\nfunc ToContinueAsNewInitiator(t *shared.ContinueAsNewInitiator) *types.ContinueAsNewInitiator {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.ContinueAsNewInitiatorDecider:\n\t\tv := types.ContinueAsNewInitiatorDecider\n\t\treturn &v\n\tcase shared.ContinueAsNewInitiatorRetryPolicy:\n\t\tv := types.ContinueAsNewInitiatorRetryPolicy\n\t\treturn &v\n\tcase shared.ContinueAsNewInitiatorCronSchedule:\n\t\tv := types.ContinueAsNewInitiatorCronSchedule\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromContinueAsNewWorkflowExecutionDecisionAttributes converts internal ContinueAsNewWorkflowExecutionDecisionAttributes type to thrift\nfunc FromContinueAsNewWorkflowExecutionDecisionAttributes(t *types.ContinueAsNewWorkflowExecutionDecisionAttributes) *shared.ContinueAsNewWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tBackoffStartIntervalInSeconds:       t.BackoffStartIntervalInSeconds,\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tInitiator:                           FromContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       t.FailureReason,\n\t\tFailureDetails:                      t.FailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tHeader:                              FromHeader(t.Header),\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\n// ToContinueAsNewWorkflowExecutionDecisionAttributes converts thrift ContinueAsNewWorkflowExecutionDecisionAttributes type to internal\nfunc ToContinueAsNewWorkflowExecutionDecisionAttributes(t *shared.ContinueAsNewWorkflowExecutionDecisionAttributes) *types.ContinueAsNewWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tBackoffStartIntervalInSeconds:       t.BackoffStartIntervalInSeconds,\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       t.FailureReason,\n\t\tFailureDetails:                      t.FailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\n// FromCountWorkflowExecutionsRequest converts internal CountWorkflowExecutionsRequest type to thrift\nfunc FromCountWorkflowExecutionsRequest(t *types.CountWorkflowExecutionsRequest) *shared.CountWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CountWorkflowExecutionsRequest{\n\t\tDomain: &t.Domain,\n\t\tQuery:  &t.Query,\n\t}\n}\n\n// ToCountWorkflowExecutionsRequest converts thrift CountWorkflowExecutionsRequest type to internal\nfunc ToCountWorkflowExecutionsRequest(t *shared.CountWorkflowExecutionsRequest) *types.CountWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountWorkflowExecutionsRequest{\n\t\tDomain: t.GetDomain(),\n\t\tQuery:  t.GetQuery(),\n\t}\n}\n\n// FromCountWorkflowExecutionsResponse converts internal CountWorkflowExecutionsResponse type to thrift\nfunc FromCountWorkflowExecutionsResponse(t *types.CountWorkflowExecutionsResponse) *shared.CountWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CountWorkflowExecutionsResponse{\n\t\tCount: &t.Count,\n\t}\n}\n\n// ToCountWorkflowExecutionsResponse converts thrift CountWorkflowExecutionsResponse type to internal\nfunc ToCountWorkflowExecutionsResponse(t *shared.CountWorkflowExecutionsResponse) *types.CountWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CountWorkflowExecutionsResponse{\n\t\tCount: t.GetCount(),\n\t}\n}\n\n// FromCurrentBranchChangedError converts internal CurrentBranchChangedError type to thrift\nfunc FromCurrentBranchChangedError(t *types.CurrentBranchChangedError) *shared.CurrentBranchChangedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CurrentBranchChangedError{\n\t\tMessage:            t.Message,\n\t\tCurrentBranchToken: t.CurrentBranchToken,\n\t}\n}\n\n// ToCurrentBranchChangedError converts thrift CurrentBranchChangedError type to internal\nfunc ToCurrentBranchChangedError(t *shared.CurrentBranchChangedError) *types.CurrentBranchChangedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CurrentBranchChangedError{\n\t\tMessage:            t.Message,\n\t\tCurrentBranchToken: t.CurrentBranchToken,\n\t}\n}\n\n// FromDataBlob converts internal DataBlob type to thrift\nfunc FromDataBlob(t *types.DataBlob) *shared.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DataBlob{\n\t\tEncodingType: FromEncodingType(t.EncodingType),\n\t\tData:         t.Data,\n\t}\n}\n\n// ToDataBlob converts thrift DataBlob type to internal\nfunc ToDataBlob(t *shared.DataBlob) *types.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DataBlob{\n\t\tEncodingType: ToEncodingType(t.EncodingType),\n\t\tData:         t.Data,\n\t}\n}\n\n// FromDecision converts internal Decision type to thrift\nfunc FromDecision(t *types.Decision) *shared.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.Decision{\n\t\tDecisionType:                                             FromDecisionType(t.DecisionType),\n\t\tScheduleActivityTaskDecisionAttributes:                   FromScheduleActivityTaskDecisionAttributes(t.ScheduleActivityTaskDecisionAttributes),\n\t\tStartTimerDecisionAttributes:                             FromStartTimerDecisionAttributes(t.StartTimerDecisionAttributes),\n\t\tCompleteWorkflowExecutionDecisionAttributes:              FromCompleteWorkflowExecutionDecisionAttributes(t.CompleteWorkflowExecutionDecisionAttributes),\n\t\tFailWorkflowExecutionDecisionAttributes:                  FromFailWorkflowExecutionDecisionAttributes(t.FailWorkflowExecutionDecisionAttributes),\n\t\tRequestCancelActivityTaskDecisionAttributes:              FromRequestCancelActivityTaskDecisionAttributes(t.RequestCancelActivityTaskDecisionAttributes),\n\t\tCancelTimerDecisionAttributes:                            FromCancelTimerDecisionAttributes(t.CancelTimerDecisionAttributes),\n\t\tCancelWorkflowExecutionDecisionAttributes:                FromCancelWorkflowExecutionDecisionAttributes(t.CancelWorkflowExecutionDecisionAttributes),\n\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: FromRequestCancelExternalWorkflowExecutionDecisionAttributes(t.RequestCancelExternalWorkflowExecutionDecisionAttributes),\n\t\tRecordMarkerDecisionAttributes:                           FromRecordMarkerDecisionAttributes(t.RecordMarkerDecisionAttributes),\n\t\tContinueAsNewWorkflowExecutionDecisionAttributes:         FromContinueAsNewWorkflowExecutionDecisionAttributes(t.ContinueAsNewWorkflowExecutionDecisionAttributes),\n\t\tStartChildWorkflowExecutionDecisionAttributes:            FromStartChildWorkflowExecutionDecisionAttributes(t.StartChildWorkflowExecutionDecisionAttributes),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes:        FromSignalExternalWorkflowExecutionDecisionAttributes(t.SignalExternalWorkflowExecutionDecisionAttributes),\n\t\tUpsertWorkflowSearchAttributesDecisionAttributes:         FromUpsertWorkflowSearchAttributesDecisionAttributes(t.UpsertWorkflowSearchAttributesDecisionAttributes),\n\t}\n}\n\n// ToDecision converts thrift Decision type to internal\nfunc ToDecision(t *shared.Decision) *types.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Decision{\n\t\tDecisionType:                                             ToDecisionType(t.DecisionType),\n\t\tScheduleActivityTaskDecisionAttributes:                   ToScheduleActivityTaskDecisionAttributes(t.ScheduleActivityTaskDecisionAttributes),\n\t\tStartTimerDecisionAttributes:                             ToStartTimerDecisionAttributes(t.StartTimerDecisionAttributes),\n\t\tCompleteWorkflowExecutionDecisionAttributes:              ToCompleteWorkflowExecutionDecisionAttributes(t.CompleteWorkflowExecutionDecisionAttributes),\n\t\tFailWorkflowExecutionDecisionAttributes:                  ToFailWorkflowExecutionDecisionAttributes(t.FailWorkflowExecutionDecisionAttributes),\n\t\tRequestCancelActivityTaskDecisionAttributes:              ToRequestCancelActivityTaskDecisionAttributes(t.RequestCancelActivityTaskDecisionAttributes),\n\t\tCancelTimerDecisionAttributes:                            ToCancelTimerDecisionAttributes(t.CancelTimerDecisionAttributes),\n\t\tCancelWorkflowExecutionDecisionAttributes:                ToCancelWorkflowExecutionDecisionAttributes(t.CancelWorkflowExecutionDecisionAttributes),\n\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: ToRequestCancelExternalWorkflowExecutionDecisionAttributes(t.RequestCancelExternalWorkflowExecutionDecisionAttributes),\n\t\tRecordMarkerDecisionAttributes:                           ToRecordMarkerDecisionAttributes(t.RecordMarkerDecisionAttributes),\n\t\tContinueAsNewWorkflowExecutionDecisionAttributes:         ToContinueAsNewWorkflowExecutionDecisionAttributes(t.ContinueAsNewWorkflowExecutionDecisionAttributes),\n\t\tStartChildWorkflowExecutionDecisionAttributes:            ToStartChildWorkflowExecutionDecisionAttributes(t.StartChildWorkflowExecutionDecisionAttributes),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes:        ToSignalExternalWorkflowExecutionDecisionAttributes(t.SignalExternalWorkflowExecutionDecisionAttributes),\n\t\tUpsertWorkflowSearchAttributesDecisionAttributes:         ToUpsertWorkflowSearchAttributesDecisionAttributes(t.UpsertWorkflowSearchAttributesDecisionAttributes),\n\t}\n}\n\n// FromDecisionTaskCompletedEventAttributes converts internal DecisionTaskCompletedEventAttributes type to thrift\nfunc FromDecisionTaskCompletedEventAttributes(t *types.DecisionTaskCompletedEventAttributes) *shared.DecisionTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DecisionTaskCompletedEventAttributes{\n\t\tExecutionContext: t.ExecutionContext,\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tStartedEventId:   &t.StartedEventID,\n\t\tIdentity:         &t.Identity,\n\t\tBinaryChecksum:   &t.BinaryChecksum,\n\t}\n}\n\n// ToDecisionTaskCompletedEventAttributes converts thrift DecisionTaskCompletedEventAttributes type to internal\nfunc ToDecisionTaskCompletedEventAttributes(t *shared.DecisionTaskCompletedEventAttributes) *types.DecisionTaskCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskCompletedEventAttributes{\n\t\tExecutionContext: t.ExecutionContext,\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tStartedEventID:   t.GetStartedEventId(),\n\t\tIdentity:         t.GetIdentity(),\n\t\tBinaryChecksum:   t.GetBinaryChecksum(),\n\t}\n}\n\n// FromDecisionTaskFailedCause converts internal DecisionTaskFailedCause type to thrift\nfunc FromDecisionTaskFailedCause(t *types.DecisionTaskFailedCause) *shared.DecisionTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DecisionTaskFailedCauseUnhandledDecision:\n\t\tv := shared.DecisionTaskFailedCauseUnhandledDecision\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadScheduleActivityAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadScheduleActivityAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadRequestCancelActivityAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadRequestCancelActivityAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadStartTimerAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadStartTimerAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadCancelTimerAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadCancelTimerAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadRecordMarkerAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadRecordMarkerAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadContinueAsNewAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadContinueAsNewAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseStartTimerDuplicateID:\n\t\tv := shared.DecisionTaskFailedCauseStartTimerDuplicateID\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseResetStickyTasklist:\n\t\tv := shared.DecisionTaskFailedCauseResetStickyTasklist\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure:\n\t\tv := shared.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadStartChildExecutionAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadStartChildExecutionAttributes\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseForceCloseDecision:\n\t\tv := shared.DecisionTaskFailedCauseForceCloseDecision\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseFailoverCloseDecision:\n\t\tv := shared.DecisionTaskFailedCauseFailoverCloseDecision\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadSignalInputSize:\n\t\tv := shared.DecisionTaskFailedCauseBadSignalInputSize\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseResetWorkflow:\n\t\tv := shared.DecisionTaskFailedCauseResetWorkflow\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadBinary:\n\t\tv := shared.DecisionTaskFailedCauseBadBinary\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseScheduleActivityDuplicateID:\n\t\tv := shared.DecisionTaskFailedCauseScheduleActivityDuplicateID\n\t\treturn &v\n\tcase types.DecisionTaskFailedCauseBadSearchAttributes:\n\t\tv := shared.DecisionTaskFailedCauseBadSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDecisionTaskFailedCause converts thrift DecisionTaskFailedCause type to internal\nfunc ToDecisionTaskFailedCause(t *shared.DecisionTaskFailedCause) *types.DecisionTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.DecisionTaskFailedCauseUnhandledDecision:\n\t\tv := types.DecisionTaskFailedCauseUnhandledDecision\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadScheduleActivityAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadScheduleActivityAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadRequestCancelActivityAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadRequestCancelActivityAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadStartTimerAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadStartTimerAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadCancelTimerAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadCancelTimerAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadRecordMarkerAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadRecordMarkerAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadContinueAsNewAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadContinueAsNewAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseStartTimerDuplicateID:\n\t\tv := types.DecisionTaskFailedCauseStartTimerDuplicateID\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseResetStickyTasklist:\n\t\tv := types.DecisionTaskFailedCauseResetStickyTasklist\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure:\n\t\tv := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadStartChildExecutionAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadStartChildExecutionAttributes\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseForceCloseDecision:\n\t\tv := types.DecisionTaskFailedCauseForceCloseDecision\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseFailoverCloseDecision:\n\t\tv := types.DecisionTaskFailedCauseFailoverCloseDecision\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadSignalInputSize:\n\t\tv := types.DecisionTaskFailedCauseBadSignalInputSize\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseResetWorkflow:\n\t\tv := types.DecisionTaskFailedCauseResetWorkflow\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadBinary:\n\t\tv := types.DecisionTaskFailedCauseBadBinary\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseScheduleActivityDuplicateID:\n\t\tv := types.DecisionTaskFailedCauseScheduleActivityDuplicateID\n\t\treturn &v\n\tcase shared.DecisionTaskFailedCauseBadSearchAttributes:\n\t\tv := types.DecisionTaskFailedCauseBadSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromDecisionTaskFailedEventAttributes converts internal DecisionTaskFailedEventAttributes type to thrift\nfunc FromDecisionTaskFailedEventAttributes(t *types.DecisionTaskFailedEventAttributes) *shared.DecisionTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tStartedEventId:   &t.StartedEventID,\n\t\tCause:            FromDecisionTaskFailedCause(t.Cause),\n\t\tDetails:          t.Details,\n\t\tIdentity:         &t.Identity,\n\t\tReason:           t.Reason,\n\t\tBaseRunId:        &t.BaseRunID,\n\t\tNewRunId:         &t.NewRunID,\n\t\tForkEventVersion: &t.ForkEventVersion,\n\t\tBinaryChecksum:   &t.BinaryChecksum,\n\t\tRequestId:        &t.RequestID,\n\t}\n}\n\n// ToDecisionTaskFailedEventAttributes converts thrift DecisionTaskFailedEventAttributes type to internal\nfunc ToDecisionTaskFailedEventAttributes(t *shared.DecisionTaskFailedEventAttributes) *types.DecisionTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tStartedEventID:   t.GetStartedEventId(),\n\t\tCause:            ToDecisionTaskFailedCause(t.Cause),\n\t\tDetails:          t.Details,\n\t\tIdentity:         t.GetIdentity(),\n\t\tReason:           t.Reason,\n\t\tBaseRunID:        t.GetBaseRunId(),\n\t\tNewRunID:         t.GetNewRunId(),\n\t\tForkEventVersion: t.GetForkEventVersion(),\n\t\tBinaryChecksum:   t.GetBinaryChecksum(),\n\t\tRequestID:        t.GetRequestId(),\n\t}\n}\n\n// FromDecisionTaskScheduledEventAttributes converts internal DecisionTaskScheduledEventAttributes type to thrift\nfunc FromDecisionTaskScheduledEventAttributes(t *types.DecisionTaskScheduledEventAttributes) *shared.DecisionTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DecisionTaskScheduledEventAttributes{\n\t\tTaskList:                   FromTaskList(t.TaskList),\n\t\tStartToCloseTimeoutSeconds: t.StartToCloseTimeoutSeconds,\n\t\tAttempt:                    &t.Attempt,\n\t}\n}\n\n// ToDecisionTaskScheduledEventAttributes converts thrift DecisionTaskScheduledEventAttributes type to internal\nfunc ToDecisionTaskScheduledEventAttributes(t *shared.DecisionTaskScheduledEventAttributes) *types.DecisionTaskScheduledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskScheduledEventAttributes{\n\t\tTaskList:                   ToTaskList(t.TaskList),\n\t\tStartToCloseTimeoutSeconds: t.StartToCloseTimeoutSeconds,\n\t\tAttempt:                    t.GetAttempt(),\n\t}\n}\n\n// FromDecisionTaskStartedEventAttributes converts internal DecisionTaskStartedEventAttributes type to thrift\nfunc FromDecisionTaskStartedEventAttributes(t *types.DecisionTaskStartedEventAttributes) *shared.DecisionTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tIdentity:         &t.Identity,\n\t\tRequestId:        &t.RequestID,\n\t}\n}\n\n// ToDecisionTaskStartedEventAttributes converts thrift DecisionTaskStartedEventAttributes type to internal\nfunc ToDecisionTaskStartedEventAttributes(t *shared.DecisionTaskStartedEventAttributes) *types.DecisionTaskStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tIdentity:         t.GetIdentity(),\n\t\tRequestID:        t.GetRequestId(),\n\t}\n}\n\n// FromDecisionTaskTimedOutCause converts internal DecisionTaskTimedOutCause type to thrift\nfunc FromDecisionTaskTimedOutCause(t *types.DecisionTaskTimedOutCause) *shared.DecisionTaskTimedOutCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DecisionTaskTimedOutCauseTimeout:\n\t\tv := shared.DecisionTaskTimedOutCauseTimeout\n\t\treturn &v\n\tcase types.DecisionTaskTimedOutCauseReset:\n\t\tv := shared.DecisionTaskTimedOutCauseReset\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDecisionTaskTimedOutCause converts thrift DecisionTaskTimedOutCause type to internal\nfunc ToDecisionTaskTimedOutCause(t *shared.DecisionTaskTimedOutCause) *types.DecisionTaskTimedOutCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.DecisionTaskTimedOutCauseTimeout:\n\t\tv := types.DecisionTaskTimedOutCauseTimeout\n\t\treturn &v\n\tcase shared.DecisionTaskTimedOutCauseReset:\n\t\tv := types.DecisionTaskTimedOutCauseReset\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromDecisionTaskTimedOutEventAttributes converts internal DecisionTaskTimedOutEventAttributes type to thrift\nfunc FromDecisionTaskTimedOutEventAttributes(t *types.DecisionTaskTimedOutEventAttributes) *shared.DecisionTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventId: &t.ScheduledEventID,\n\t\tStartedEventId:   &t.StartedEventID,\n\t\tTimeoutType:      FromTimeoutType(t.TimeoutType),\n\t\tBaseRunId:        &t.BaseRunID,\n\t\tNewRunId:         &t.NewRunID,\n\t\tForkEventVersion: &t.ForkEventVersion,\n\t\tReason:           &t.Reason,\n\t\tCause:            FromDecisionTaskTimedOutCause(t.Cause),\n\t\tRequestId:        &t.RequestID,\n\t}\n}\n\n// ToDecisionTaskTimedOutEventAttributes converts thrift DecisionTaskTimedOutEventAttributes type to internal\nfunc ToDecisionTaskTimedOutEventAttributes(t *shared.DecisionTaskTimedOutEventAttributes) *types.DecisionTaskTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventID: t.GetScheduledEventId(),\n\t\tStartedEventID:   t.GetStartedEventId(),\n\t\tTimeoutType:      ToTimeoutType(t.TimeoutType),\n\t\tBaseRunID:        t.GetBaseRunId(),\n\t\tNewRunID:         t.GetNewRunId(),\n\t\tForkEventVersion: t.GetForkEventVersion(),\n\t\tReason:           t.GetReason(),\n\t\tCause:            ToDecisionTaskTimedOutCause(t.Cause),\n\t\tRequestID:        t.GetRequestId(),\n\t}\n}\n\n// FromDecisionType converts internal DecisionType type to thrift\nfunc FromDecisionType(t *types.DecisionType) *shared.DecisionType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DecisionTypeScheduleActivityTask:\n\t\tv := shared.DecisionTypeScheduleActivityTask\n\t\treturn &v\n\tcase types.DecisionTypeRequestCancelActivityTask:\n\t\tv := shared.DecisionTypeRequestCancelActivityTask\n\t\treturn &v\n\tcase types.DecisionTypeStartTimer:\n\t\tv := shared.DecisionTypeStartTimer\n\t\treturn &v\n\tcase types.DecisionTypeCompleteWorkflowExecution:\n\t\tv := shared.DecisionTypeCompleteWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeFailWorkflowExecution:\n\t\tv := shared.DecisionTypeFailWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeCancelTimer:\n\t\tv := shared.DecisionTypeCancelTimer\n\t\treturn &v\n\tcase types.DecisionTypeCancelWorkflowExecution:\n\t\tv := shared.DecisionTypeCancelWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeRequestCancelExternalWorkflowExecution:\n\t\tv := shared.DecisionTypeRequestCancelExternalWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeRecordMarker:\n\t\tv := shared.DecisionTypeRecordMarker\n\t\treturn &v\n\tcase types.DecisionTypeContinueAsNewWorkflowExecution:\n\t\tv := shared.DecisionTypeContinueAsNewWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeStartChildWorkflowExecution:\n\t\tv := shared.DecisionTypeStartChildWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeSignalExternalWorkflowExecution:\n\t\tv := shared.DecisionTypeSignalExternalWorkflowExecution\n\t\treturn &v\n\tcase types.DecisionTypeUpsertWorkflowSearchAttributes:\n\t\tv := shared.DecisionTypeUpsertWorkflowSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDecisionType converts thrift DecisionType type to internal\nfunc ToDecisionType(t *shared.DecisionType) *types.DecisionType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.DecisionTypeScheduleActivityTask:\n\t\tv := types.DecisionTypeScheduleActivityTask\n\t\treturn &v\n\tcase shared.DecisionTypeRequestCancelActivityTask:\n\t\tv := types.DecisionTypeRequestCancelActivityTask\n\t\treturn &v\n\tcase shared.DecisionTypeStartTimer:\n\t\tv := types.DecisionTypeStartTimer\n\t\treturn &v\n\tcase shared.DecisionTypeCompleteWorkflowExecution:\n\t\tv := types.DecisionTypeCompleteWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeFailWorkflowExecution:\n\t\tv := types.DecisionTypeFailWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeCancelTimer:\n\t\tv := types.DecisionTypeCancelTimer\n\t\treturn &v\n\tcase shared.DecisionTypeCancelWorkflowExecution:\n\t\tv := types.DecisionTypeCancelWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeRequestCancelExternalWorkflowExecution:\n\t\tv := types.DecisionTypeRequestCancelExternalWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeRecordMarker:\n\t\tv := types.DecisionTypeRecordMarker\n\t\treturn &v\n\tcase shared.DecisionTypeContinueAsNewWorkflowExecution:\n\t\tv := types.DecisionTypeContinueAsNewWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeStartChildWorkflowExecution:\n\t\tv := types.DecisionTypeStartChildWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeSignalExternalWorkflowExecution:\n\t\tv := types.DecisionTypeSignalExternalWorkflowExecution\n\t\treturn &v\n\tcase shared.DecisionTypeUpsertWorkflowSearchAttributes:\n\t\tv := types.DecisionTypeUpsertWorkflowSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromDeleteDomainRequest converts internal DeleteDomainRequest type to thrift\nfunc FromDeleteDomainRequest(t *types.DeleteDomainRequest) *shared.DeleteDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DeleteDomainRequest{\n\t\tName:          &t.Name,\n\t\tSecurityToken: &t.SecurityToken,\n\t}\n}\n\n// ToDeleteDomainRequest converts thrift DeleteDomainRequest type to internal\nfunc ToDeleteDomainRequest(t *shared.DeleteDomainRequest) *types.DeleteDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeleteDomainRequest{\n\t\tName:          t.GetName(),\n\t\tSecurityToken: t.GetSecurityToken(),\n\t}\n}\n\n// FromDeprecateDomainRequest converts internal DeprecateDomainRequest type to thrift\nfunc FromDeprecateDomainRequest(t *types.DeprecateDomainRequest) *shared.DeprecateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DeprecateDomainRequest{\n\t\tName:          &t.Name,\n\t\tSecurityToken: &t.SecurityToken,\n\t}\n}\n\n// ToDeprecateDomainRequest converts thrift DeprecateDomainRequest type to internal\nfunc ToDeprecateDomainRequest(t *shared.DeprecateDomainRequest) *types.DeprecateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DeprecateDomainRequest{\n\t\tName:          t.GetName(),\n\t\tSecurityToken: t.GetSecurityToken(),\n\t}\n}\n\n// FromDescribeDomainRequest converts internal DescribeDomainRequest type to thrift\nfunc FromDescribeDomainRequest(t *types.DescribeDomainRequest) *shared.DescribeDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeDomainRequest{\n\t\tName: t.Name,\n\t\tUUID: t.UUID,\n\t}\n}\n\n// ToDescribeDomainRequest converts thrift DescribeDomainRequest type to internal\nfunc ToDescribeDomainRequest(t *shared.DescribeDomainRequest) *types.DescribeDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeDomainRequest{\n\t\tName: t.Name,\n\t\tUUID: t.UUID,\n\t}\n}\n\n// FromDescribeDomainResponse converts internal DescribeDomainResponse type to thrift\nfunc FromDescribeDomainResponse(t *types.DescribeDomainResponse) *shared.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeDomainResponse{\n\t\tDomainInfo:               FromDomainInfo(t.DomainInfo),\n\t\tConfiguration:            FromDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: FromDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          &t.FailoverVersion,\n\t\tIsGlobalDomain:           &t.IsGlobalDomain,\n\t\tFailoverInfo:             FromFailoverInfo(t.GetFailoverInfo()),\n\t}\n}\n\n// ToDescribeDomainResponse converts thrift DescribeDomainResponse type to internal\nfunc ToDescribeDomainResponse(t *shared.DescribeDomainResponse) *types.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeDomainResponse{\n\t\tDomainInfo:               ToDomainInfo(t.DomainInfo),\n\t\tConfiguration:            ToDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: ToDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          t.GetFailoverVersion(),\n\t\tIsGlobalDomain:           t.GetIsGlobalDomain(),\n\t\tFailoverInfo:             ToFailoverInfo(t.FailoverInfo),\n\t}\n}\n\n// FromFailoverDomainRequest converts internal FailoverDomainRequest type to thrift\nfunc FromFailoverDomainRequest(t *types.FailoverDomainRequest) *shared.FailoverDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FailoverDomainRequest{\n\t\tDomainName:              &t.DomainName,\n\t\tDomainActiveClusterName: t.DomainActiveClusterName,\n\t\tActiveClusters:          FromActiveClusters(t.ActiveClusters),\n\t}\n}\n\n// ToFailoverDomainRequest converts thrift FailoverDomainRequest type to internal\nfunc ToFailoverDomainRequest(t *shared.FailoverDomainRequest) *types.FailoverDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverDomainRequest{\n\t\tDomainName:              t.GetDomainName(),\n\t\tDomainActiveClusterName: t.DomainActiveClusterName,\n\t\tActiveClusters:          ToActiveClusters(t.ActiveClusters),\n\t}\n}\n\n// FromFailoverDomainResponse converts internal FailoverDomainResponse type to thrift\nfunc FromFailoverDomainResponse(t *types.FailoverDomainResponse) *shared.FailoverDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FailoverDomainResponse{\n\t\tDomainInfo:               FromDomainInfo(t.DomainInfo),\n\t\tConfiguration:            FromDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: FromDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          &t.FailoverVersion,\n\t\tIsGlobalDomain:           &t.IsGlobalDomain,\n\t}\n}\n\n// ToFailoverDomainResponse converts thrift FailoverDomainResponse type to internal\nfunc ToFailoverDomainResponse(t *shared.FailoverDomainResponse) *types.FailoverDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverDomainResponse{\n\t\tDomainInfo:               ToDomainInfo(t.DomainInfo),\n\t\tConfiguration:            ToDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: ToDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          t.GetFailoverVersion(),\n\t\tIsGlobalDomain:           t.GetIsGlobalDomain(),\n\t}\n}\n\n// FromAdminDescribeHistoryHostRequest converts internal DescribeHistoryHostRequest type to thrift\nfunc FromAdminDescribeHistoryHostRequest(t *types.DescribeHistoryHostRequest) *shared.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeHistoryHostRequest{\n\t\tHostAddress:      t.HostAddress,\n\t\tShardIdForHost:   t.ShardIDForHost,\n\t\tExecutionForHost: FromWorkflowExecution(t.ExecutionForHost),\n\t}\n}\n\n// ToAdminDescribeHistoryHostRequest converts thrift DescribeHistoryHostRequest type to internal\nfunc ToAdminDescribeHistoryHostRequest(t *shared.DescribeHistoryHostRequest) *types.DescribeHistoryHostRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeHistoryHostRequest{\n\t\tHostAddress:      t.HostAddress,\n\t\tShardIDForHost:   t.ShardIdForHost,\n\t\tExecutionForHost: ToWorkflowExecution(t.ExecutionForHost),\n\t}\n}\n\n// FromAdminDescribeShardDistributionRequest converts internal DescribeHistoryHostRequest type to thrift\nfunc FromAdminDescribeShardDistributionRequest(t *types.DescribeShardDistributionRequest) *shared.DescribeShardDistributionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeShardDistributionRequest{\n\t\tPageSize: &t.PageSize,\n\t\tPageID:   &t.PageID,\n\t}\n}\n\n// ToAdminDescribeShardDistributionRequest converts thrift DescribeHistoryHostRequest type to internal\nfunc ToAdminDescribeShardDistributionRequest(t *shared.DescribeShardDistributionRequest) *types.DescribeShardDistributionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeShardDistributionRequest{\n\t\tPageSize: t.GetPageSize(),\n\t\tPageID:   t.GetPageID(),\n\t}\n}\n\n// FromAdminDescribeHistoryHostResponse converts internal DescribeHistoryHostResponse type to thrift\nfunc FromAdminDescribeHistoryHostResponse(t *types.DescribeHistoryHostResponse) *shared.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        &t.NumberOfShards,\n\t\tShardIDs:              t.ShardIDs,\n\t\tDomainCache:           FromDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: &t.ShardControllerStatus,\n\t\tAddress:               &t.Address,\n\t}\n}\n\n// ToAdminDescribeHistoryHostResponse converts thrift DescribeHistoryHostResponse type to internal\nfunc ToAdminDescribeHistoryHostResponse(t *shared.DescribeHistoryHostResponse) *types.DescribeHistoryHostResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        t.GetNumberOfShards(),\n\t\tShardIDs:              t.ShardIDs,\n\t\tDomainCache:           ToDomainCacheInfo(t.DomainCache),\n\t\tShardControllerStatus: t.GetShardControllerStatus(),\n\t\tAddress:               t.GetAddress(),\n\t}\n}\n\n// FromAdminDescribeShardDistributionResponse converts internal DescribeHistoryHostResponse type to thrift\nfunc FromAdminDescribeShardDistributionResponse(t *types.DescribeShardDistributionResponse) *shared.DescribeShardDistributionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeShardDistributionResponse{\n\t\tNumberOfShards: &t.NumberOfShards,\n\t\tShards:         t.Shards,\n\t}\n}\n\n// ToAdminDescribeShardDistributionResponse converts thrift DescribeHistoryHostResponse type to internal\nfunc ToAdminDescribeShardDistributionResponse(t *shared.DescribeShardDistributionResponse) *types.DescribeShardDistributionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeShardDistributionResponse{\n\t\tNumberOfShards: t.GetNumberOfShards(),\n\t\tShards:         t.Shards,\n\t}\n}\n\n// FromAdminDescribeQueueRequest converts internal DescribeQueueRequest type to thrift\nfunc FromAdminDescribeQueueRequest(t *types.DescribeQueueRequest) *shared.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeQueueRequest{\n\t\tShardID:     &t.ShardID,\n\t\tClusterName: &t.ClusterName,\n\t\tType:        t.Type,\n\t}\n}\n\n// ToAdminDescribeQueueRequest converts thrift DescribeQueueRequest type to internal\nfunc ToAdminDescribeQueueRequest(t *shared.DescribeQueueRequest) *types.DescribeQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueRequest{\n\t\tShardID:     t.GetShardID(),\n\t\tClusterName: t.GetClusterName(),\n\t\tType:        t.Type,\n\t}\n}\n\n// FromAdminDescribeQueueResponse converts internal DescribeQueueResponse type to thrift\nfunc FromAdminDescribeQueueResponse(t *types.DescribeQueueResponse) *shared.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\n// ToAdminDescribeQueueResponse converts thrift DescribeQueueResponse type to internal\nfunc ToAdminDescribeQueueResponse(t *shared.DescribeQueueResponse) *types.DescribeQueueResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeQueueResponse{\n\t\tProcessingQueueStates: t.ProcessingQueueStates,\n\t}\n}\n\n// FromDescribeTaskListRequest converts internal DescribeTaskListRequest type to thrift\nfunc FromDescribeTaskListRequest(t *types.DescribeTaskListRequest) *shared.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeTaskListRequest{\n\t\tDomain:                &t.Domain,\n\t\tTaskList:              FromTaskList(t.TaskList),\n\t\tTaskListType:          FromTaskListType(t.TaskListType),\n\t\tIncludeTaskListStatus: &t.IncludeTaskListStatus,\n\t}\n}\n\n// ToDescribeTaskListRequest converts thrift DescribeTaskListRequest type to internal\nfunc ToDescribeTaskListRequest(t *shared.DescribeTaskListRequest) *types.DescribeTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeTaskListRequest{\n\t\tDomain:                t.GetDomain(),\n\t\tTaskList:              ToTaskList(t.TaskList),\n\t\tTaskListType:          ToTaskListType(t.TaskListType),\n\t\tIncludeTaskListStatus: t.GetIncludeTaskListStatus(),\n\t}\n}\n\n// FromDescribeTaskListResponse converts internal DescribeTaskListResponse type to thrift\nfunc FromDescribeTaskListResponse(t *types.DescribeTaskListResponse) *shared.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeTaskListResponse{\n\t\tPollers:        FromPollerInfoArray(t.Pollers),\n\t\tTaskListStatus: FromTaskListStatus(t.TaskListStatus),\n\t\tTaskList:       FromTaskList(t.TaskList),\n\t}\n}\n\n// ToDescribeTaskListResponse converts thrift DescribeTaskListResponse type to internal\nfunc ToDescribeTaskListResponse(t *shared.DescribeTaskListResponse) *types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers:        ToPollerInfoArray(t.Pollers),\n\t\tTaskListStatus: ToTaskListStatus(t.TaskListStatus),\n\t\tTaskList:       ToTaskList(t.TaskList),\n\t}\n}\n\n// FromDescribeWorkflowExecutionRequest converts internal DescribeWorkflowExecutionRequest type to thrift\nfunc FromDescribeWorkflowExecutionRequest(t *types.DescribeWorkflowExecutionRequest) *shared.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeWorkflowExecutionRequest{\n\t\tDomain:                &t.Domain,\n\t\tExecution:             FromWorkflowExecution(t.Execution),\n\t\tQueryConsistencyLevel: FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// ToDescribeWorkflowExecutionRequest converts thrift DescribeWorkflowExecutionRequest type to internal\nfunc ToDescribeWorkflowExecutionRequest(t *shared.DescribeWorkflowExecutionRequest) *types.DescribeWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeWorkflowExecutionRequest{\n\t\tDomain:                t.GetDomain(),\n\t\tExecution:             ToWorkflowExecution(t.Execution),\n\t\tQueryConsistencyLevel: ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// FromDescribeWorkflowExecutionResponse converts internal DescribeWorkflowExecutionResponse type to thrift\nfunc FromDescribeWorkflowExecutionResponse(t *types.DescribeWorkflowExecutionResponse) *shared.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: FromWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  FromWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      FromPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        FromPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        FromPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\n// ToDescribeWorkflowExecutionResponse converts thrift DescribeWorkflowExecutionResponse type to internal\nfunc ToDescribeWorkflowExecutionResponse(t *shared.DescribeWorkflowExecutionResponse) *types.DescribeWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: ToWorkflowExecutionConfiguration(t.ExecutionConfiguration),\n\t\tWorkflowExecutionInfo:  ToWorkflowExecutionInfo(t.WorkflowExecutionInfo),\n\t\tPendingActivities:      ToPendingActivityInfoArray(t.PendingActivities),\n\t\tPendingChildren:        ToPendingChildExecutionInfoArray(t.PendingChildren),\n\t\tPendingDecision:        ToPendingDecisionInfo(t.PendingDecision),\n\t}\n}\n\n// FromDiagnoseWorkflowExecutionRequest converts internal DiagnoseWorkflowExecutionRequest type to thrift\nfunc FromDiagnoseWorkflowExecutionRequest(t *types.DiagnoseWorkflowExecutionRequest) *shared.DiagnoseWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DiagnoseWorkflowExecutionRequest{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.GetWorkflowExecution()),\n\t\tIdentity:          &t.Identity,\n\t}\n}\n\n// ToDiagnoseWorkflowExecutionRequest converts thrift DiagnoseWorkflowExecutionRequest type to internal\nfunc ToDiagnoseWorkflowExecutionRequest(t *shared.DiagnoseWorkflowExecutionRequest) *types.DiagnoseWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.GetWorkflowExecution()),\n\t\tIdentity:          t.GetIdentity(),\n\t}\n}\n\n// FromDiagnoseWorkflowExecutionResponse converts internal DiagnoseWorkflowExecutionResponse type to thrift\nfunc FromDiagnoseWorkflowExecutionResponse(t *types.DiagnoseWorkflowExecutionResponse) *shared.DiagnoseWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DiagnoseWorkflowExecutionResponse{\n\t\tDomain:                      &t.Domain,\n\t\tDiagnosticWorkflowExecution: FromWorkflowExecution(t.GetDiagnosticWorkflowExecution()),\n\t}\n}\n\n// ToDiagnoseWorkflowExecutionResponse converts thrift DiagnoseeWorkflowExecutionResponse type to internal\nfunc ToDiagnoseWorkflowExecutionResponse(t *shared.DiagnoseWorkflowExecutionResponse) *types.DiagnoseWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain:                      t.GetDomain(),\n\t\tDiagnosticWorkflowExecution: ToWorkflowExecution(t.GetDiagnosticWorkflowExecution()),\n\t}\n}\n\n// FromDomainAlreadyExistsError converts internal DomainAlreadyExistsError type to thrift\nfunc FromDomainAlreadyExistsError(t *types.DomainAlreadyExistsError) *shared.DomainAlreadyExistsError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainAlreadyExistsError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToDomainAlreadyExistsError converts thrift DomainAlreadyExistsError type to internal\nfunc ToDomainAlreadyExistsError(t *shared.DomainAlreadyExistsError) *types.DomainAlreadyExistsError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainAlreadyExistsError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromDomainCacheInfo converts internal DomainCacheInfo type to thrift\nfunc FromDomainCacheInfo(t *types.DomainCacheInfo) *shared.DomainCacheInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainCacheInfo{\n\t\tNumOfItemsInCacheByID:   &t.NumOfItemsInCacheByID,\n\t\tNumOfItemsInCacheByName: &t.NumOfItemsInCacheByName,\n\t}\n}\n\n// ToDomainCacheInfo converts thrift DomainCacheInfo type to internal\nfunc ToDomainCacheInfo(t *shared.DomainCacheInfo) *types.DomainCacheInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainCacheInfo{\n\t\tNumOfItemsInCacheByID:   t.GetNumOfItemsInCacheByID(),\n\t\tNumOfItemsInCacheByName: t.GetNumOfItemsInCacheByName(),\n\t}\n}\n\n// FromDomainConfiguration converts internal DomainConfiguration type to thrift\nfunc FromDomainConfiguration(t *types.DomainConfiguration) *shared.DomainConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: &t.WorkflowExecutionRetentionPeriodInDays,\n\t\tEmitMetric:                             &t.EmitMetric,\n\t\tBadBinaries:                            FromBadBinaries(t.BadBinaries),\n\t\tHistoryArchivalStatus:                  FromArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:                     &t.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:               FromArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:                  &t.VisibilityArchivalURI,\n\t\tIsolationgroups:                        FromIsolationGroupConfig(t.IsolationGroups),\n\t\tAsyncWorkflowConfiguration:             FromDomainAsyncWorkflowConfiguraton(t.AsyncWorkflowConfig),\n\t}\n}\n\n// ToDomainConfiguration converts thrift DomainConfiguration type to internal\nfunc ToDomainConfiguration(t *shared.DomainConfiguration) *types.DomainConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: t.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\tEmitMetric:                             t.GetEmitMetric(),\n\t\tBadBinaries:                            ToBadBinaries(t.BadBinaries),\n\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:                     t.GetHistoryArchivalURI(),\n\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:                  t.GetVisibilityArchivalURI(),\n\t\tIsolationGroups:                        ToIsolationGroupConfig(t.Isolationgroups),\n\t\tAsyncWorkflowConfig:                    ToDomainAsyncWorkflowConfiguraton(t.AsyncWorkflowConfiguration),\n\t}\n}\n\n// FromDomainInfo converts internal DomainInfo type to thrift\nfunc FromDomainInfo(t *types.DomainInfo) *shared.DomainInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainInfo{\n\t\tName:        &t.Name,\n\t\tStatus:      FromDomainStatus(t.Status),\n\t\tDescription: &t.Description,\n\t\tOwnerEmail:  &t.OwnerEmail,\n\t\tData:        t.Data,\n\t\tUUID:        &t.UUID,\n\t}\n}\n\n// ToDomainInfo converts thrift DomainInfo type to internal\nfunc ToDomainInfo(t *shared.DomainInfo) *types.DomainInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainInfo{\n\t\tName:        t.GetName(),\n\t\tStatus:      ToDomainStatus(t.Status),\n\t\tDescription: t.GetDescription(),\n\t\tOwnerEmail:  t.GetOwnerEmail(),\n\t\tData:        t.Data,\n\t\tUUID:        t.GetUUID(),\n\t}\n}\n\n// FromFailoverInfo converts internal FailoverInfo type to thrift\nfunc FromFailoverInfo(t *types.FailoverInfo) *shared.FailoverInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FailoverInfo{\n\t\tFailoverVersion:         &t.FailoverVersion,\n\t\tFailoverStartTimestamp:  &t.FailoverStartTimestamp,\n\t\tFailoverExpireTimestamp: &t.FailoverExpireTimestamp,\n\t\tCompletedShardCount:     &t.CompletedShardCount,\n\t\tPendingShards:           t.GetPendingShards(),\n\t}\n}\n\n// ToFailoverInfo converts thrift FailoverInfo type to internal\nfunc ToFailoverInfo(t *shared.FailoverInfo) *types.FailoverInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverInfo{\n\t\tFailoverVersion:         t.GetFailoverVersion(),\n\t\tFailoverStartTimestamp:  t.GetFailoverStartTimestamp(),\n\t\tFailoverExpireTimestamp: t.GetFailoverExpireTimestamp(),\n\t\tCompletedShardCount:     t.GetCompletedShardCount(),\n\t\tPendingShards:           t.GetPendingShards(),\n\t}\n}\n\n// FromDomainNotActiveError converts internal DomainNotActiveError type to thrift\nfunc FromDomainNotActiveError(t *types.DomainNotActiveError) *shared.DomainNotActiveError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainNotActiveError{\n\t\tMessage:        t.Message,\n\t\tDomainName:     t.DomainName,\n\t\tCurrentCluster: t.CurrentCluster,\n\t\tActiveCluster:  t.ActiveCluster,\n\t\tActiveClusters: t.ActiveClusters,\n\t}\n}\n\n// ToDomainNotActiveError converts thrift DomainNotActiveError type to internal\nfunc ToDomainNotActiveError(t *shared.DomainNotActiveError) *types.DomainNotActiveError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainNotActiveError{\n\t\tMessage:        t.Message,\n\t\tDomainName:     t.DomainName,\n\t\tCurrentCluster: t.CurrentCluster,\n\t\tActiveCluster:  t.ActiveCluster,\n\t\tActiveClusters: t.ActiveClusters,\n\t}\n}\n\n// FromDomainReplicationConfiguration converts internal DomainReplicationConfiguration type to thrift\nfunc FromDomainReplicationConfiguration(t *types.DomainReplicationConfiguration) *shared.DomainReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.DomainReplicationConfiguration{\n\t\tActiveClusterName: &t.ActiveClusterName,\n\t\tClusters:          FromClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusters:    FromActiveClusters(t.ActiveClusters),\n\t}\n}\n\n// ToDomainReplicationConfiguration converts thrift DomainReplicationConfiguration type to internal\nfunc ToDomainReplicationConfiguration(t *shared.DomainReplicationConfiguration) *types.DomainReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.DomainReplicationConfiguration{\n\t\tActiveClusterName: t.GetActiveClusterName(),\n\t\tClusters:          ToClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusters:    ToActiveClusters(t.ActiveClusters),\n\t}\n}\n\n// FromActiveClusters converts internal ActiveClusters type to thrift\nfunc FromActiveClusters(t *types.ActiveClusters) *shared.ActiveClusters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar activeClustersByClusterAttribute map[string]*shared.ClusterAttributeScope\n\tif t.AttributeScopes != nil {\n\t\tactiveClustersByClusterAttribute = make(map[string]*shared.ClusterAttributeScope)\n\t\tfor scopeType, scope := range t.AttributeScopes {\n\t\t\tactiveClustersByClusterAttribute[scopeType] = FromClusterAttributeScope(&scope)\n\t\t}\n\t}\n\n\treturn &shared.ActiveClusters{\n\t\tActiveClustersByClusterAttribute: activeClustersByClusterAttribute,\n\t}\n}\n\n// ToActiveClusters converts thrift ActiveClusters type to internal\nfunc ToActiveClusters(t *shared.ActiveClusters) *types.ActiveClusters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar attributeScopes map[string]types.ClusterAttributeScope\n\tif t.ActiveClustersByClusterAttribute != nil {\n\t\tattributeScopes = make(map[string]types.ClusterAttributeScope)\n\t\tfor scopeType, scope := range t.ActiveClustersByClusterAttribute {\n\t\t\tif converted := ToClusterAttributeScope(scope); converted != nil {\n\t\t\t\tattributeScopes[scopeType] = *converted\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.ActiveClusters{\n\t\tAttributeScopes: attributeScopes,\n\t}\n}\n\n// FromClusterAttributeScope converts internal ClusterAttributeScope type to thrift\nfunc FromClusterAttributeScope(t *types.ClusterAttributeScope) *shared.ClusterAttributeScope {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar clusterAttributes map[string]*shared.ActiveClusterInfo\n\tif len(t.ClusterAttributes) > 0 {\n\t\tclusterAttributes = make(map[string]*shared.ActiveClusterInfo)\n\t\tfor name, clusterInfo := range t.ClusterAttributes {\n\t\t\tclusterAttributes[name] = &shared.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: &clusterInfo.ActiveClusterName,\n\t\t\t\tFailoverVersion:   &clusterInfo.FailoverVersion,\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &shared.ClusterAttributeScope{\n\t\tClusterAttributes: clusterAttributes,\n\t}\n}\n\n// ToClusterAttributeScope converts thrift ClusterAttributeScope type to internal\nfunc ToClusterAttributeScope(t *shared.ClusterAttributeScope) *types.ClusterAttributeScope {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\tvar clusterAttributes map[string]types.ActiveClusterInfo\n\tif len(t.ClusterAttributes) > 0 {\n\t\tclusterAttributes = make(map[string]types.ActiveClusterInfo)\n\t\tfor name, clusterInfo := range t.ClusterAttributes {\n\t\t\tif clusterInfo != nil {\n\t\t\t\tclusterAttributes[name] = types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: *clusterInfo.ActiveClusterName,\n\t\t\t\t\tFailoverVersion:   *clusterInfo.FailoverVersion,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn &types.ClusterAttributeScope{\n\t\tClusterAttributes: clusterAttributes,\n\t}\n}\n\n// FromDomainStatus converts internal DomainStatus type to thrift\nfunc FromDomainStatus(t *types.DomainStatus) *shared.DomainStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.DomainStatusRegistered:\n\t\tv := shared.DomainStatusRegistered\n\t\treturn &v\n\tcase types.DomainStatusDeprecated:\n\t\tv := shared.DomainStatusDeprecated\n\t\treturn &v\n\tcase types.DomainStatusDeleted:\n\t\tv := shared.DomainStatusDeleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToDomainStatus converts thrift DomainStatus type to internal\nfunc ToDomainStatus(t *shared.DomainStatus) *types.DomainStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.DomainStatusRegistered:\n\t\tv := types.DomainStatusRegistered\n\t\treturn &v\n\tcase shared.DomainStatusDeprecated:\n\t\tv := types.DomainStatusDeprecated\n\t\treturn &v\n\tcase shared.DomainStatusDeleted:\n\t\tv := types.DomainStatusDeleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromEncodingType converts internal EncodingType type to thrift\nfunc FromEncodingType(t *types.EncodingType) *shared.EncodingType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.EncodingTypeThriftRW:\n\t\tv := shared.EncodingTypeThriftRW\n\t\treturn &v\n\tcase types.EncodingTypeJSON:\n\t\tv := shared.EncodingTypeJSON\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToEncodingType converts thrift EncodingType type to internal\nfunc ToEncodingType(t *shared.EncodingType) *types.EncodingType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.EncodingTypeThriftRW:\n\t\tv := types.EncodingTypeThriftRW\n\t\treturn &v\n\tcase shared.EncodingTypeJSON:\n\t\tv := types.EncodingTypeJSON\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromEntityNotExistsError converts internal EntityNotExistsError type to thrift\nfunc FromEntityNotExistsError(t *types.EntityNotExistsError) *shared.EntityNotExistsError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.EntityNotExistsError{\n\t\tMessage:        t.Message,\n\t\tCurrentCluster: &t.CurrentCluster,\n\t\tActiveCluster:  &t.ActiveCluster,\n\t\tActiveClusters: t.ActiveClusters,\n\t}\n}\n\n// ToEntityNotExistsError converts thrift EntityNotExistsError type to internal\nfunc ToEntityNotExistsError(t *shared.EntityNotExistsError) *types.EntityNotExistsError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.EntityNotExistsError{\n\t\tMessage:        t.Message,\n\t\tCurrentCluster: t.GetCurrentCluster(),\n\t\tActiveCluster:  t.GetActiveCluster(),\n\t\tActiveClusters: t.GetActiveClusters(),\n\t}\n}\n\n// FromWorkflowExecutionAlreadyCompletedError converts internal WorkflowExecutionAlreadyCompletedError type to thrift\nfunc FromWorkflowExecutionAlreadyCompletedError(t *types.WorkflowExecutionAlreadyCompletedError) *shared.WorkflowExecutionAlreadyCompletedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionAlreadyCompletedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToWorkflowExecutionAlreadyCompletedError converts thrift WorkflowExecutionAlreadyCompletedError type to internal\nfunc ToWorkflowExecutionAlreadyCompletedError(t *shared.WorkflowExecutionAlreadyCompletedError) *types.WorkflowExecutionAlreadyCompletedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionAlreadyCompletedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromEventType converts internal EventType type to thrift\nfunc FromEventType(t *types.EventType) *shared.EventType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.EventTypeWorkflowExecutionStarted:\n\t\tv := shared.EventTypeWorkflowExecutionStarted\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tv := shared.EventTypeWorkflowExecutionCompleted\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tv := shared.EventTypeWorkflowExecutionFailed\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tv := shared.EventTypeWorkflowExecutionTimedOut\n\t\treturn &v\n\tcase types.EventTypeDecisionTaskScheduled:\n\t\tv := shared.EventTypeDecisionTaskScheduled\n\t\treturn &v\n\tcase types.EventTypeDecisionTaskStarted:\n\t\tv := shared.EventTypeDecisionTaskStarted\n\t\treturn &v\n\tcase types.EventTypeDecisionTaskCompleted:\n\t\tv := shared.EventTypeDecisionTaskCompleted\n\t\treturn &v\n\tcase types.EventTypeDecisionTaskTimedOut:\n\t\tv := shared.EventTypeDecisionTaskTimedOut\n\t\treturn &v\n\tcase types.EventTypeDecisionTaskFailed:\n\t\tv := shared.EventTypeDecisionTaskFailed\n\t\treturn &v\n\tcase types.EventTypeActivityTaskScheduled:\n\t\tv := shared.EventTypeActivityTaskScheduled\n\t\treturn &v\n\tcase types.EventTypeActivityTaskStarted:\n\t\tv := shared.EventTypeActivityTaskStarted\n\t\treturn &v\n\tcase types.EventTypeActivityTaskCompleted:\n\t\tv := shared.EventTypeActivityTaskCompleted\n\t\treturn &v\n\tcase types.EventTypeActivityTaskFailed:\n\t\tv := shared.EventTypeActivityTaskFailed\n\t\treturn &v\n\tcase types.EventTypeActivityTaskTimedOut:\n\t\tv := shared.EventTypeActivityTaskTimedOut\n\t\treturn &v\n\tcase types.EventTypeActivityTaskCancelRequested:\n\t\tv := shared.EventTypeActivityTaskCancelRequested\n\t\treturn &v\n\tcase types.EventTypeRequestCancelActivityTaskFailed:\n\t\tv := shared.EventTypeRequestCancelActivityTaskFailed\n\t\treturn &v\n\tcase types.EventTypeActivityTaskCanceled:\n\t\tv := shared.EventTypeActivityTaskCanceled\n\t\treturn &v\n\tcase types.EventTypeTimerStarted:\n\t\tv := shared.EventTypeTimerStarted\n\t\treturn &v\n\tcase types.EventTypeTimerFired:\n\t\tv := shared.EventTypeTimerFired\n\t\treturn &v\n\tcase types.EventTypeCancelTimerFailed:\n\t\tv := shared.EventTypeCancelTimerFailed\n\t\treturn &v\n\tcase types.EventTypeTimerCanceled:\n\t\tv := shared.EventTypeTimerCanceled\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionCancelRequested:\n\t\tv := shared.EventTypeWorkflowExecutionCancelRequested\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tv := shared.EventTypeWorkflowExecutionCanceled\n\t\treturn &v\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\tv := shared.EventTypeRequestCancelExternalWorkflowExecutionInitiated\n\t\treturn &v\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\tv := shared.EventTypeRequestCancelExternalWorkflowExecutionFailed\n\t\treturn &v\n\tcase types.EventTypeExternalWorkflowExecutionCancelRequested:\n\t\tv := shared.EventTypeExternalWorkflowExecutionCancelRequested\n\t\treturn &v\n\tcase types.EventTypeMarkerRecorded:\n\t\tv := shared.EventTypeMarkerRecorded\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\tv := shared.EventTypeWorkflowExecutionSignaled\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tv := shared.EventTypeWorkflowExecutionTerminated\n\t\treturn &v\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tv := shared.EventTypeWorkflowExecutionContinuedAsNew\n\t\treturn &v\n\tcase types.EventTypeStartChildWorkflowExecutionInitiated:\n\t\tv := shared.EventTypeStartChildWorkflowExecutionInitiated\n\t\treturn &v\n\tcase types.EventTypeStartChildWorkflowExecutionFailed:\n\t\tv := shared.EventTypeStartChildWorkflowExecutionFailed\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\tv := shared.EventTypeChildWorkflowExecutionStarted\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\tv := shared.EventTypeChildWorkflowExecutionCompleted\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\tv := shared.EventTypeChildWorkflowExecutionFailed\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\tv := shared.EventTypeChildWorkflowExecutionCanceled\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\tv := shared.EventTypeChildWorkflowExecutionTimedOut\n\t\treturn &v\n\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\t\tv := shared.EventTypeChildWorkflowExecutionTerminated\n\t\treturn &v\n\tcase types.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\tv := shared.EventTypeSignalExternalWorkflowExecutionInitiated\n\t\treturn &v\n\tcase types.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\tv := shared.EventTypeSignalExternalWorkflowExecutionFailed\n\t\treturn &v\n\tcase types.EventTypeExternalWorkflowExecutionSignaled:\n\t\tv := shared.EventTypeExternalWorkflowExecutionSignaled\n\t\treturn &v\n\tcase types.EventTypeUpsertWorkflowSearchAttributes:\n\t\tv := shared.EventTypeUpsertWorkflowSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToEventType converts thrift EventType type to internal\nfunc ToEventType(t *shared.EventType) *types.EventType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.EventTypeWorkflowExecutionStarted:\n\t\tv := types.EventTypeWorkflowExecutionStarted\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionCompleted:\n\t\tv := types.EventTypeWorkflowExecutionCompleted\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionFailed:\n\t\tv := types.EventTypeWorkflowExecutionFailed\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionTimedOut:\n\t\tv := types.EventTypeWorkflowExecutionTimedOut\n\t\treturn &v\n\tcase shared.EventTypeDecisionTaskScheduled:\n\t\tv := types.EventTypeDecisionTaskScheduled\n\t\treturn &v\n\tcase shared.EventTypeDecisionTaskStarted:\n\t\tv := types.EventTypeDecisionTaskStarted\n\t\treturn &v\n\tcase shared.EventTypeDecisionTaskCompleted:\n\t\tv := types.EventTypeDecisionTaskCompleted\n\t\treturn &v\n\tcase shared.EventTypeDecisionTaskTimedOut:\n\t\tv := types.EventTypeDecisionTaskTimedOut\n\t\treturn &v\n\tcase shared.EventTypeDecisionTaskFailed:\n\t\tv := types.EventTypeDecisionTaskFailed\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskScheduled:\n\t\tv := types.EventTypeActivityTaskScheduled\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskStarted:\n\t\tv := types.EventTypeActivityTaskStarted\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskCompleted:\n\t\tv := types.EventTypeActivityTaskCompleted\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskFailed:\n\t\tv := types.EventTypeActivityTaskFailed\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskTimedOut:\n\t\tv := types.EventTypeActivityTaskTimedOut\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskCancelRequested:\n\t\tv := types.EventTypeActivityTaskCancelRequested\n\t\treturn &v\n\tcase shared.EventTypeRequestCancelActivityTaskFailed:\n\t\tv := types.EventTypeRequestCancelActivityTaskFailed\n\t\treturn &v\n\tcase shared.EventTypeActivityTaskCanceled:\n\t\tv := types.EventTypeActivityTaskCanceled\n\t\treturn &v\n\tcase shared.EventTypeTimerStarted:\n\t\tv := types.EventTypeTimerStarted\n\t\treturn &v\n\tcase shared.EventTypeTimerFired:\n\t\tv := types.EventTypeTimerFired\n\t\treturn &v\n\tcase shared.EventTypeCancelTimerFailed:\n\t\tv := types.EventTypeCancelTimerFailed\n\t\treturn &v\n\tcase shared.EventTypeTimerCanceled:\n\t\tv := types.EventTypeTimerCanceled\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionCancelRequested:\n\t\tv := types.EventTypeWorkflowExecutionCancelRequested\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionCanceled:\n\t\tv := types.EventTypeWorkflowExecutionCanceled\n\t\treturn &v\n\tcase shared.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\tv := types.EventTypeRequestCancelExternalWorkflowExecutionInitiated\n\t\treturn &v\n\tcase shared.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\tv := types.EventTypeRequestCancelExternalWorkflowExecutionFailed\n\t\treturn &v\n\tcase shared.EventTypeExternalWorkflowExecutionCancelRequested:\n\t\tv := types.EventTypeExternalWorkflowExecutionCancelRequested\n\t\treturn &v\n\tcase shared.EventTypeMarkerRecorded:\n\t\tv := types.EventTypeMarkerRecorded\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionSignaled:\n\t\tv := types.EventTypeWorkflowExecutionSignaled\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionTerminated:\n\t\tv := types.EventTypeWorkflowExecutionTerminated\n\t\treturn &v\n\tcase shared.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tv := types.EventTypeWorkflowExecutionContinuedAsNew\n\t\treturn &v\n\tcase shared.EventTypeStartChildWorkflowExecutionInitiated:\n\t\tv := types.EventTypeStartChildWorkflowExecutionInitiated\n\t\treturn &v\n\tcase shared.EventTypeStartChildWorkflowExecutionFailed:\n\t\tv := types.EventTypeStartChildWorkflowExecutionFailed\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionStarted:\n\t\tv := types.EventTypeChildWorkflowExecutionStarted\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionCompleted:\n\t\tv := types.EventTypeChildWorkflowExecutionCompleted\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionFailed:\n\t\tv := types.EventTypeChildWorkflowExecutionFailed\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionCanceled:\n\t\tv := types.EventTypeChildWorkflowExecutionCanceled\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionTimedOut:\n\t\tv := types.EventTypeChildWorkflowExecutionTimedOut\n\t\treturn &v\n\tcase shared.EventTypeChildWorkflowExecutionTerminated:\n\t\tv := types.EventTypeChildWorkflowExecutionTerminated\n\t\treturn &v\n\tcase shared.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\tv := types.EventTypeSignalExternalWorkflowExecutionInitiated\n\t\treturn &v\n\tcase shared.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\tv := types.EventTypeSignalExternalWorkflowExecutionFailed\n\t\treturn &v\n\tcase shared.EventTypeExternalWorkflowExecutionSignaled:\n\t\tv := types.EventTypeExternalWorkflowExecutionSignaled\n\t\treturn &v\n\tcase shared.EventTypeUpsertWorkflowSearchAttributes:\n\t\tv := types.EventTypeUpsertWorkflowSearchAttributes\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromExternalWorkflowExecutionCancelRequestedEventAttributes converts internal ExternalWorkflowExecutionCancelRequestedEventAttributes type to thrift\nfunc FromExternalWorkflowExecutionCancelRequestedEventAttributes(t *types.ExternalWorkflowExecutionCancelRequestedEventAttributes) *shared.ExternalWorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\n// ToExternalWorkflowExecutionCancelRequestedEventAttributes converts thrift ExternalWorkflowExecutionCancelRequestedEventAttributes type to internal\nfunc ToExternalWorkflowExecutionCancelRequestedEventAttributes(t *shared.ExternalWorkflowExecutionCancelRequestedEventAttributes) *types.ExternalWorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t}\n}\n\n// FromExternalWorkflowExecutionSignaledEventAttributes converts internal ExternalWorkflowExecutionSignaledEventAttributes type to thrift\nfunc FromExternalWorkflowExecutionSignaledEventAttributes(t *types.ExternalWorkflowExecutionSignaledEventAttributes) *shared.ExternalWorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventId:  &t.InitiatedEventID,\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tControl:           t.Control,\n\t}\n}\n\n// ToExternalWorkflowExecutionSignaledEventAttributes converts thrift ExternalWorkflowExecutionSignaledEventAttributes type to internal\nfunc ToExternalWorkflowExecutionSignaledEventAttributes(t *shared.ExternalWorkflowExecutionSignaledEventAttributes) *types.ExternalWorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventID:  t.GetInitiatedEventId(),\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tControl:           t.Control,\n\t}\n}\n\n// FromFailWorkflowExecutionDecisionAttributes converts internal FailWorkflowExecutionDecisionAttributes type to thrift\nfunc FromFailWorkflowExecutionDecisionAttributes(t *types.FailWorkflowExecutionDecisionAttributes) *shared.FailWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  t.Reason,\n\t\tDetails: t.Details,\n\t}\n}\n\n// ToFailWorkflowExecutionDecisionAttributes converts thrift FailWorkflowExecutionDecisionAttributes type to internal\nfunc ToFailWorkflowExecutionDecisionAttributes(t *shared.FailWorkflowExecutionDecisionAttributes) *types.FailWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  t.Reason,\n\t\tDetails: t.Details,\n\t}\n}\n\n// FromGetSearchAttributesResponse converts internal GetSearchAttributesResponse type to thrift\nfunc FromGetSearchAttributesResponse(t *types.GetSearchAttributesResponse) *shared.GetSearchAttributesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetSearchAttributesResponse{\n\t\tKeys: FromIndexedValueTypeMap(t.Keys),\n\t}\n}\n\n// ToGetSearchAttributesResponse converts thrift GetSearchAttributesResponse type to internal\nfunc ToGetSearchAttributesResponse(t *shared.GetSearchAttributesResponse) *types.GetSearchAttributesResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetSearchAttributesResponse{\n\t\tKeys: ToIndexedValueTypeMap(t.Keys),\n\t}\n}\n\n// FromGetWorkflowExecutionHistoryRequest converts internal GetWorkflowExecutionHistoryRequest type to thrift\nfunc FromGetWorkflowExecutionHistoryRequest(t *types.GetWorkflowExecutionHistoryRequest) *shared.GetWorkflowExecutionHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 &t.Domain,\n\t\tExecution:              FromWorkflowExecution(t.Execution),\n\t\tMaximumPageSize:        &t.MaximumPageSize,\n\t\tNextPageToken:          t.NextPageToken,\n\t\tWaitForNewEvent:        &t.WaitForNewEvent,\n\t\tHistoryEventFilterType: FromHistoryEventFilterType(t.HistoryEventFilterType),\n\t\tSkipArchival:           &t.SkipArchival,\n\t\tQueryConsistencyLevel:  FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// ToGetWorkflowExecutionHistoryRequest converts thrift GetWorkflowExecutionHistoryRequest type to internal\nfunc ToGetWorkflowExecutionHistoryRequest(t *shared.GetWorkflowExecutionHistoryRequest) *types.GetWorkflowExecutionHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 t.GetDomain(),\n\t\tExecution:              ToWorkflowExecution(t.Execution),\n\t\tMaximumPageSize:        t.GetMaximumPageSize(),\n\t\tNextPageToken:          t.NextPageToken,\n\t\tWaitForNewEvent:        t.GetWaitForNewEvent(),\n\t\tHistoryEventFilterType: ToHistoryEventFilterType(t.HistoryEventFilterType),\n\t\tSkipArchival:           t.GetSkipArchival(),\n\t\tQueryConsistencyLevel:  ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// FromGetWorkflowExecutionHistoryResponse converts internal GetWorkflowExecutionHistoryResponse type to thrift\nfunc FromGetWorkflowExecutionHistoryResponse(t *types.GetWorkflowExecutionHistoryResponse) *shared.GetWorkflowExecutionHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       FromHistory(t.History),\n\t\tRawHistory:    FromDataBlobArray(t.RawHistory),\n\t\tNextPageToken: t.NextPageToken,\n\t\tArchived:      &t.Archived,\n\t}\n}\n\n// ToGetWorkflowExecutionHistoryResponse converts thrift GetWorkflowExecutionHistoryResponse type to internal\nfunc ToGetWorkflowExecutionHistoryResponse(t *shared.GetWorkflowExecutionHistoryResponse) *types.GetWorkflowExecutionHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       ToHistory(t.History),\n\t\tRawHistory:    ToDataBlobArray(t.RawHistory),\n\t\tNextPageToken: t.NextPageToken,\n\t\tArchived:      t.GetArchived(),\n\t}\n}\n\n// FromHeader converts internal Header type to thrift\nfunc FromHeader(t *types.Header) *shared.Header {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.Header{\n\t\tFields: t.Fields,\n\t}\n}\n\n// ToHeader converts thrift Header type to internal\nfunc ToHeader(t *shared.Header) *types.Header {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Header{\n\t\tFields: t.Fields,\n\t}\n}\n\n// FromHistory converts internal History type to thrift\nfunc FromHistory(t *types.History) *shared.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.History{\n\t\tEvents: FromHistoryEventArray(t.Events),\n\t}\n}\n\n// ToHistory converts thrift History type to internal\nfunc ToHistory(t *shared.History) *types.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.History{\n\t\tEvents: ToHistoryEventArray(t.Events),\n\t}\n}\n\n// FromHistoryBranch converts internal HistoryBranch type to thrift\nfunc FromHistoryBranch(t *types.HistoryBranch) *shared.HistoryBranch {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.HistoryBranch{\n\t\tTreeID:    &t.TreeID,\n\t\tBranchID:  &t.BranchID,\n\t\tAncestors: FromHistoryBranchRangeArray(t.Ancestors),\n\t}\n}\n\n// ToHistoryBranch converts thrift HistoryBranch type to internal\nfunc ToHistoryBranch(t *shared.HistoryBranch) *types.HistoryBranch {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryBranch{\n\t\tTreeID:    *t.TreeID,\n\t\tBranchID:  *t.BranchID,\n\t\tAncestors: ToHistoryBranchRangeArray(t.Ancestors),\n\t}\n}\n\n// FromHistoryBranchRange converts internal HistoryBranchRange type to thrift\nfunc FromHistoryBranchRange(t *types.HistoryBranchRange) *shared.HistoryBranchRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.HistoryBranchRange{\n\t\tBranchID:    &t.BranchID,\n\t\tBeginNodeID: &t.BeginNodeID,\n\t\tEndNodeID:   &t.EndNodeID,\n\t}\n}\n\n// ToHistoryBranchRange converts thrift HistoryBranchRange type to internal\nfunc ToHistoryBranchRange(t *shared.HistoryBranchRange) *types.HistoryBranchRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryBranchRange{\n\t\tBranchID:    *t.BranchID,\n\t\tBeginNodeID: *t.BeginNodeID,\n\t\tEndNodeID:   *t.EndNodeID,\n\t}\n}\n\n// FromHistoryEvent converts internal HistoryEvent type to thrift\nfunc FromHistoryEvent(t *types.HistoryEvent) *shared.HistoryEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.HistoryEvent{\n\t\tEventId:                                 &t.ID,\n\t\tTimestamp:                               t.Timestamp,\n\t\tEventType:                               FromEventType(t.EventType),\n\t\tVersion:                                 &t.Version,\n\t\tTaskId:                                  &t.TaskID,\n\t\tWorkflowExecutionStartedEventAttributes: FromWorkflowExecutionStartedEventAttributes(t.WorkflowExecutionStartedEventAttributes),\n\t\tWorkflowExecutionCompletedEventAttributes:                      FromWorkflowExecutionCompletedEventAttributes(t.WorkflowExecutionCompletedEventAttributes),\n\t\tWorkflowExecutionFailedEventAttributes:                         FromWorkflowExecutionFailedEventAttributes(t.WorkflowExecutionFailedEventAttributes),\n\t\tWorkflowExecutionTimedOutEventAttributes:                       FromWorkflowExecutionTimedOutEventAttributes(t.WorkflowExecutionTimedOutEventAttributes),\n\t\tDecisionTaskScheduledEventAttributes:                           FromDecisionTaskScheduledEventAttributes(t.DecisionTaskScheduledEventAttributes),\n\t\tDecisionTaskStartedEventAttributes:                             FromDecisionTaskStartedEventAttributes(t.DecisionTaskStartedEventAttributes),\n\t\tDecisionTaskCompletedEventAttributes:                           FromDecisionTaskCompletedEventAttributes(t.DecisionTaskCompletedEventAttributes),\n\t\tDecisionTaskTimedOutEventAttributes:                            FromDecisionTaskTimedOutEventAttributes(t.DecisionTaskTimedOutEventAttributes),\n\t\tDecisionTaskFailedEventAttributes:                              FromDecisionTaskFailedEventAttributes(t.DecisionTaskFailedEventAttributes),\n\t\tActivityTaskScheduledEventAttributes:                           FromActivityTaskScheduledEventAttributes(t.ActivityTaskScheduledEventAttributes),\n\t\tActivityTaskStartedEventAttributes:                             FromActivityTaskStartedEventAttributes(t.ActivityTaskStartedEventAttributes),\n\t\tActivityTaskCompletedEventAttributes:                           FromActivityTaskCompletedEventAttributes(t.ActivityTaskCompletedEventAttributes),\n\t\tActivityTaskFailedEventAttributes:                              FromActivityTaskFailedEventAttributes(t.ActivityTaskFailedEventAttributes),\n\t\tActivityTaskTimedOutEventAttributes:                            FromActivityTaskTimedOutEventAttributes(t.ActivityTaskTimedOutEventAttributes),\n\t\tTimerStartedEventAttributes:                                    FromTimerStartedEventAttributes(t.TimerStartedEventAttributes),\n\t\tTimerFiredEventAttributes:                                      FromTimerFiredEventAttributes(t.TimerFiredEventAttributes),\n\t\tActivityTaskCancelRequestedEventAttributes:                     FromActivityTaskCancelRequestedEventAttributes(t.ActivityTaskCancelRequestedEventAttributes),\n\t\tRequestCancelActivityTaskFailedEventAttributes:                 FromRequestCancelActivityTaskFailedEventAttributes(t.RequestCancelActivityTaskFailedEventAttributes),\n\t\tActivityTaskCanceledEventAttributes:                            FromActivityTaskCanceledEventAttributes(t.ActivityTaskCanceledEventAttributes),\n\t\tTimerCanceledEventAttributes:                                   FromTimerCanceledEventAttributes(t.TimerCanceledEventAttributes),\n\t\tCancelTimerFailedEventAttributes:                               FromCancelTimerFailedEventAttributes(t.CancelTimerFailedEventAttributes),\n\t\tMarkerRecordedEventAttributes:                                  FromMarkerRecordedEventAttributes(t.MarkerRecordedEventAttributes),\n\t\tWorkflowExecutionSignaledEventAttributes:                       FromWorkflowExecutionSignaledEventAttributes(t.WorkflowExecutionSignaledEventAttributes),\n\t\tWorkflowExecutionTerminatedEventAttributes:                     FromWorkflowExecutionTerminatedEventAttributes(t.WorkflowExecutionTerminatedEventAttributes),\n\t\tWorkflowExecutionCancelRequestedEventAttributes:                FromWorkflowExecutionCancelRequestedEventAttributes(t.WorkflowExecutionCancelRequestedEventAttributes),\n\t\tWorkflowExecutionCanceledEventAttributes:                       FromWorkflowExecutionCanceledEventAttributes(t.WorkflowExecutionCanceledEventAttributes),\n\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes),\n\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes:    FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(t.RequestCancelExternalWorkflowExecutionFailedEventAttributes),\n\t\tExternalWorkflowExecutionCancelRequestedEventAttributes:        FromExternalWorkflowExecutionCancelRequestedEventAttributes(t.ExternalWorkflowExecutionCancelRequestedEventAttributes),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes:                 FromWorkflowExecutionContinuedAsNewEventAttributes(t.WorkflowExecutionContinuedAsNewEventAttributes),\n\t\tStartChildWorkflowExecutionInitiatedEventAttributes:            FromStartChildWorkflowExecutionInitiatedEventAttributes(t.StartChildWorkflowExecutionInitiatedEventAttributes),\n\t\tStartChildWorkflowExecutionFailedEventAttributes:               FromStartChildWorkflowExecutionFailedEventAttributes(t.StartChildWorkflowExecutionFailedEventAttributes),\n\t\tChildWorkflowExecutionStartedEventAttributes:                   FromChildWorkflowExecutionStartedEventAttributes(t.ChildWorkflowExecutionStartedEventAttributes),\n\t\tChildWorkflowExecutionCompletedEventAttributes:                 FromChildWorkflowExecutionCompletedEventAttributes(t.ChildWorkflowExecutionCompletedEventAttributes),\n\t\tChildWorkflowExecutionFailedEventAttributes:                    FromChildWorkflowExecutionFailedEventAttributes(t.ChildWorkflowExecutionFailedEventAttributes),\n\t\tChildWorkflowExecutionCanceledEventAttributes:                  FromChildWorkflowExecutionCanceledEventAttributes(t.ChildWorkflowExecutionCanceledEventAttributes),\n\t\tChildWorkflowExecutionTimedOutEventAttributes:                  FromChildWorkflowExecutionTimedOutEventAttributes(t.ChildWorkflowExecutionTimedOutEventAttributes),\n\t\tChildWorkflowExecutionTerminatedEventAttributes:                FromChildWorkflowExecutionTerminatedEventAttributes(t.ChildWorkflowExecutionTerminatedEventAttributes),\n\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes:        FromSignalExternalWorkflowExecutionInitiatedEventAttributes(t.SignalExternalWorkflowExecutionInitiatedEventAttributes),\n\t\tSignalExternalWorkflowExecutionFailedEventAttributes:           FromSignalExternalWorkflowExecutionFailedEventAttributes(t.SignalExternalWorkflowExecutionFailedEventAttributes),\n\t\tExternalWorkflowExecutionSignaledEventAttributes:               FromExternalWorkflowExecutionSignaledEventAttributes(t.ExternalWorkflowExecutionSignaledEventAttributes),\n\t\tUpsertWorkflowSearchAttributesEventAttributes:                  FromUpsertWorkflowSearchAttributesEventAttributes(t.UpsertWorkflowSearchAttributesEventAttributes),\n\t}\n}\n\n// ToHistoryEvent converts thrift HistoryEvent type to internal\nfunc ToHistoryEvent(t *shared.HistoryEvent) *types.HistoryEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.HistoryEvent{\n\t\tID:                                      t.GetEventId(),\n\t\tTimestamp:                               t.Timestamp,\n\t\tEventType:                               ToEventType(t.EventType),\n\t\tVersion:                                 t.GetVersion(),\n\t\tTaskID:                                  t.GetTaskId(),\n\t\tWorkflowExecutionStartedEventAttributes: ToWorkflowExecutionStartedEventAttributes(t.WorkflowExecutionStartedEventAttributes),\n\t\tWorkflowExecutionCompletedEventAttributes:                      ToWorkflowExecutionCompletedEventAttributes(t.WorkflowExecutionCompletedEventAttributes),\n\t\tWorkflowExecutionFailedEventAttributes:                         ToWorkflowExecutionFailedEventAttributes(t.WorkflowExecutionFailedEventAttributes),\n\t\tWorkflowExecutionTimedOutEventAttributes:                       ToWorkflowExecutionTimedOutEventAttributes(t.WorkflowExecutionTimedOutEventAttributes),\n\t\tDecisionTaskScheduledEventAttributes:                           ToDecisionTaskScheduledEventAttributes(t.DecisionTaskScheduledEventAttributes),\n\t\tDecisionTaskStartedEventAttributes:                             ToDecisionTaskStartedEventAttributes(t.DecisionTaskStartedEventAttributes),\n\t\tDecisionTaskCompletedEventAttributes:                           ToDecisionTaskCompletedEventAttributes(t.DecisionTaskCompletedEventAttributes),\n\t\tDecisionTaskTimedOutEventAttributes:                            ToDecisionTaskTimedOutEventAttributes(t.DecisionTaskTimedOutEventAttributes),\n\t\tDecisionTaskFailedEventAttributes:                              ToDecisionTaskFailedEventAttributes(t.DecisionTaskFailedEventAttributes),\n\t\tActivityTaskScheduledEventAttributes:                           ToActivityTaskScheduledEventAttributes(t.ActivityTaskScheduledEventAttributes),\n\t\tActivityTaskStartedEventAttributes:                             ToActivityTaskStartedEventAttributes(t.ActivityTaskStartedEventAttributes),\n\t\tActivityTaskCompletedEventAttributes:                           ToActivityTaskCompletedEventAttributes(t.ActivityTaskCompletedEventAttributes),\n\t\tActivityTaskFailedEventAttributes:                              ToActivityTaskFailedEventAttributes(t.ActivityTaskFailedEventAttributes),\n\t\tActivityTaskTimedOutEventAttributes:                            ToActivityTaskTimedOutEventAttributes(t.ActivityTaskTimedOutEventAttributes),\n\t\tTimerStartedEventAttributes:                                    ToTimerStartedEventAttributes(t.TimerStartedEventAttributes),\n\t\tTimerFiredEventAttributes:                                      ToTimerFiredEventAttributes(t.TimerFiredEventAttributes),\n\t\tActivityTaskCancelRequestedEventAttributes:                     ToActivityTaskCancelRequestedEventAttributes(t.ActivityTaskCancelRequestedEventAttributes),\n\t\tRequestCancelActivityTaskFailedEventAttributes:                 ToRequestCancelActivityTaskFailedEventAttributes(t.RequestCancelActivityTaskFailedEventAttributes),\n\t\tActivityTaskCanceledEventAttributes:                            ToActivityTaskCanceledEventAttributes(t.ActivityTaskCanceledEventAttributes),\n\t\tTimerCanceledEventAttributes:                                   ToTimerCanceledEventAttributes(t.TimerCanceledEventAttributes),\n\t\tCancelTimerFailedEventAttributes:                               ToCancelTimerFailedEventAttributes(t.CancelTimerFailedEventAttributes),\n\t\tMarkerRecordedEventAttributes:                                  ToMarkerRecordedEventAttributes(t.MarkerRecordedEventAttributes),\n\t\tWorkflowExecutionSignaledEventAttributes:                       ToWorkflowExecutionSignaledEventAttributes(t.WorkflowExecutionSignaledEventAttributes),\n\t\tWorkflowExecutionTerminatedEventAttributes:                     ToWorkflowExecutionTerminatedEventAttributes(t.WorkflowExecutionTerminatedEventAttributes),\n\t\tWorkflowExecutionCancelRequestedEventAttributes:                ToWorkflowExecutionCancelRequestedEventAttributes(t.WorkflowExecutionCancelRequestedEventAttributes),\n\t\tWorkflowExecutionCanceledEventAttributes:                       ToWorkflowExecutionCanceledEventAttributes(t.WorkflowExecutionCanceledEventAttributes),\n\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes),\n\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes:    ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(t.RequestCancelExternalWorkflowExecutionFailedEventAttributes),\n\t\tExternalWorkflowExecutionCancelRequestedEventAttributes:        ToExternalWorkflowExecutionCancelRequestedEventAttributes(t.ExternalWorkflowExecutionCancelRequestedEventAttributes),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes:                 ToWorkflowExecutionContinuedAsNewEventAttributes(t.WorkflowExecutionContinuedAsNewEventAttributes),\n\t\tStartChildWorkflowExecutionInitiatedEventAttributes:            ToStartChildWorkflowExecutionInitiatedEventAttributes(t.StartChildWorkflowExecutionInitiatedEventAttributes),\n\t\tStartChildWorkflowExecutionFailedEventAttributes:               ToStartChildWorkflowExecutionFailedEventAttributes(t.StartChildWorkflowExecutionFailedEventAttributes),\n\t\tChildWorkflowExecutionStartedEventAttributes:                   ToChildWorkflowExecutionStartedEventAttributes(t.ChildWorkflowExecutionStartedEventAttributes),\n\t\tChildWorkflowExecutionCompletedEventAttributes:                 ToChildWorkflowExecutionCompletedEventAttributes(t.ChildWorkflowExecutionCompletedEventAttributes),\n\t\tChildWorkflowExecutionFailedEventAttributes:                    ToChildWorkflowExecutionFailedEventAttributes(t.ChildWorkflowExecutionFailedEventAttributes),\n\t\tChildWorkflowExecutionCanceledEventAttributes:                  ToChildWorkflowExecutionCanceledEventAttributes(t.ChildWorkflowExecutionCanceledEventAttributes),\n\t\tChildWorkflowExecutionTimedOutEventAttributes:                  ToChildWorkflowExecutionTimedOutEventAttributes(t.ChildWorkflowExecutionTimedOutEventAttributes),\n\t\tChildWorkflowExecutionTerminatedEventAttributes:                ToChildWorkflowExecutionTerminatedEventAttributes(t.ChildWorkflowExecutionTerminatedEventAttributes),\n\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes:        ToSignalExternalWorkflowExecutionInitiatedEventAttributes(t.SignalExternalWorkflowExecutionInitiatedEventAttributes),\n\t\tSignalExternalWorkflowExecutionFailedEventAttributes:           ToSignalExternalWorkflowExecutionFailedEventAttributes(t.SignalExternalWorkflowExecutionFailedEventAttributes),\n\t\tExternalWorkflowExecutionSignaledEventAttributes:               ToExternalWorkflowExecutionSignaledEventAttributes(t.ExternalWorkflowExecutionSignaledEventAttributes),\n\t\tUpsertWorkflowSearchAttributesEventAttributes:                  ToUpsertWorkflowSearchAttributesEventAttributes(t.UpsertWorkflowSearchAttributesEventAttributes),\n\t}\n}\n\n// FromHistoryEventFilterType converts internal HistoryEventFilterType type to thrift\nfunc FromHistoryEventFilterType(t *types.HistoryEventFilterType) *shared.HistoryEventFilterType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.HistoryEventFilterTypeAllEvent:\n\t\tv := shared.HistoryEventFilterTypeAllEvent\n\t\treturn &v\n\tcase types.HistoryEventFilterTypeCloseEvent:\n\t\tv := shared.HistoryEventFilterTypeCloseEvent\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToHistoryEventFilterType converts thrift HistoryEventFilterType type to internal\nfunc ToHistoryEventFilterType(t *shared.HistoryEventFilterType) *types.HistoryEventFilterType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.HistoryEventFilterTypeAllEvent:\n\t\tv := types.HistoryEventFilterTypeAllEvent\n\t\treturn &v\n\tcase shared.HistoryEventFilterTypeCloseEvent:\n\t\tv := types.HistoryEventFilterTypeCloseEvent\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromIndexedValueType converts internal IndexedValueType type to thrift\nfunc FromIndexedValueType(t types.IndexedValueType) shared.IndexedValueType {\n\n\tswitch t {\n\tcase types.IndexedValueTypeString:\n\t\treturn shared.IndexedValueTypeString\n\tcase types.IndexedValueTypeKeyword:\n\t\treturn shared.IndexedValueTypeKeyword\n\tcase types.IndexedValueTypeInt:\n\t\treturn shared.IndexedValueTypeInt\n\tcase types.IndexedValueTypeDouble:\n\t\treturn shared.IndexedValueTypeDouble\n\tcase types.IndexedValueTypeBool:\n\t\treturn shared.IndexedValueTypeBool\n\tcase types.IndexedValueTypeDatetime:\n\t\treturn shared.IndexedValueTypeDatetime\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToIndexedValueType converts thrift IndexedValueType type to internal\nfunc ToIndexedValueType(t shared.IndexedValueType) types.IndexedValueType {\n\n\tswitch t {\n\tcase shared.IndexedValueTypeString:\n\t\treturn types.IndexedValueTypeString\n\tcase shared.IndexedValueTypeKeyword:\n\t\treturn types.IndexedValueTypeKeyword\n\tcase shared.IndexedValueTypeInt:\n\t\treturn types.IndexedValueTypeInt\n\tcase shared.IndexedValueTypeDouble:\n\t\treturn types.IndexedValueTypeDouble\n\tcase shared.IndexedValueTypeBool:\n\t\treturn types.IndexedValueTypeBool\n\tcase shared.IndexedValueTypeDatetime:\n\t\treturn types.IndexedValueTypeDatetime\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromInternalDataInconsistencyError converts internal InternalDataInconsistencyError type to thrift\nfunc FromInternalDataInconsistencyError(t *types.InternalDataInconsistencyError) *shared.InternalDataInconsistencyError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.InternalDataInconsistencyError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToInternalDataInconsistencyError converts thrift InternalDataInconsistencyError type to internal\nfunc ToInternalDataInconsistencyError(t *shared.InternalDataInconsistencyError) *types.InternalDataInconsistencyError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.InternalDataInconsistencyError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromInternalServiceError converts internal InternalServiceError type to thrift\nfunc FromInternalServiceError(t *types.InternalServiceError) *shared.InternalServiceError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.InternalServiceError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToInternalServiceError converts thrift InternalServiceError type to internal\nfunc ToInternalServiceError(t *shared.InternalServiceError) *types.InternalServiceError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.InternalServiceError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromLimitExceededError converts internal LimitExceededError type to thrift\nfunc FromLimitExceededError(t *types.LimitExceededError) *shared.LimitExceededError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.LimitExceededError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToLimitExceededError converts thrift LimitExceededError type to internal\nfunc ToLimitExceededError(t *shared.LimitExceededError) *types.LimitExceededError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.LimitExceededError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromListArchivedWorkflowExecutionsRequest converts internal ListArchivedWorkflowExecutionsRequest type to thrift\nfunc FromListArchivedWorkflowExecutionsRequest(t *types.ListArchivedWorkflowExecutionsRequest) *shared.ListArchivedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:        &t.Domain,\n\t\tPageSize:      &t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         &t.Query,\n\t}\n}\n\n// ToListArchivedWorkflowExecutionsRequest converts thrift ListArchivedWorkflowExecutionsRequest type to internal\nfunc ToListArchivedWorkflowExecutionsRequest(t *shared.ListArchivedWorkflowExecutionsRequest) *types.ListArchivedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:        t.GetDomain(),\n\t\tPageSize:      t.GetPageSize(),\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.GetQuery(),\n\t}\n}\n\n// FromListArchivedWorkflowExecutionsResponse converts internal ListArchivedWorkflowExecutionsResponse type to thrift\nfunc FromListArchivedWorkflowExecutionsResponse(t *types.ListArchivedWorkflowExecutionsResponse) *shared.ListArchivedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListArchivedWorkflowExecutionsResponse converts thrift ListArchivedWorkflowExecutionsResponse type to internal\nfunc ToListArchivedWorkflowExecutionsResponse(t *shared.ListArchivedWorkflowExecutionsResponse) *types.ListArchivedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromListClosedWorkflowExecutionsRequest converts internal ListClosedWorkflowExecutionsRequest type to thrift\nfunc FromListClosedWorkflowExecutionsRequest(t *types.ListClosedWorkflowExecutionsRequest) *shared.ListClosedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          &t.Domain,\n\t\tMaximumPageSize: &t.MaximumPageSize,\n\t\tNextPageToken:   t.NextPageToken,\n\t\tStartTimeFilter: FromStartTimeFilter(t.StartTimeFilter),\n\t\tExecutionFilter: FromWorkflowExecutionFilter(t.ExecutionFilter),\n\t\tTypeFilter:      FromWorkflowTypeFilter(t.TypeFilter),\n\t\tStatusFilter:    FromWorkflowExecutionCloseStatus(t.StatusFilter),\n\t}\n}\n\n// ToListClosedWorkflowExecutionsRequest converts thrift ListClosedWorkflowExecutionsRequest type to internal\nfunc ToListClosedWorkflowExecutionsRequest(t *shared.ListClosedWorkflowExecutionsRequest) *types.ListClosedWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          t.GetDomain(),\n\t\tMaximumPageSize: t.GetMaximumPageSize(),\n\t\tNextPageToken:   t.NextPageToken,\n\t\tStartTimeFilter: ToStartTimeFilter(t.StartTimeFilter),\n\t\tExecutionFilter: ToWorkflowExecutionFilter(t.ExecutionFilter),\n\t\tTypeFilter:      ToWorkflowTypeFilter(t.TypeFilter),\n\t\tStatusFilter:    ToWorkflowExecutionCloseStatus(t.StatusFilter),\n\t}\n}\n\n// FromListClosedWorkflowExecutionsResponse converts internal ListClosedWorkflowExecutionsResponse type to thrift\nfunc FromListClosedWorkflowExecutionsResponse(t *types.ListClosedWorkflowExecutionsResponse) *shared.ListClosedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListClosedWorkflowExecutionsResponse converts thrift ListClosedWorkflowExecutionsResponse type to internal\nfunc ToListClosedWorkflowExecutionsResponse(t *shared.ListClosedWorkflowExecutionsResponse) *types.ListClosedWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromListDomainsRequest converts internal ListDomainsRequest type to thrift\nfunc FromListDomainsRequest(t *types.ListDomainsRequest) *shared.ListDomainsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListDomainsRequest{\n\t\tPageSize:      &t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListDomainsRequest converts thrift ListDomainsRequest type to internal\nfunc ToListDomainsRequest(t *shared.ListDomainsRequest) *types.ListDomainsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDomainsRequest{\n\t\tPageSize:      t.GetPageSize(),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromListDomainsResponse converts internal ListDomainsResponse type to thrift\nfunc FromListDomainsResponse(t *types.ListDomainsResponse) *shared.ListDomainsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListDomainsResponse{\n\t\tDomains:       FromDescribeDomainResponseArray(t.Domains),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListDomainsResponse converts thrift ListDomainsResponse type to internal\nfunc ToListDomainsResponse(t *shared.ListDomainsResponse) *types.ListDomainsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListDomainsResponse{\n\t\tDomains:       ToDescribeDomainResponseArray(t.Domains),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListFailoverHistoryRequest converts thrift ListFailoverHistoryRequest type to internal\nfunc ToListFailoverHistoryRequest(t *shared.ListFailoverHistoryRequest) *types.ListFailoverHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.ListFailoverHistoryRequest{\n\t\tFilters:    ToListFailoverHistoryRequestFilters(t.Filters),\n\t\tPagination: ToPaginationOptions(t.Pagination),\n\t}\n}\n\n// FromListFailoverHistoryRequest converts internal ListFailoverHistoryRequest type to thrift\nfunc FromListFailoverHistoryRequest(t *types.ListFailoverHistoryRequest) *shared.ListFailoverHistoryRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &shared.ListFailoverHistoryRequest{\n\t\tFilters:    FromListFailoverHistoryRequestFilters(t.Filters),\n\t\tPagination: FromPaginationOptions(t.Pagination),\n\t}\n}\n\n// FromListFailoverHistoryResponse converts internal ListFailoverHistoryResponse type to thrift\nfunc FromListFailoverHistoryResponse(t *types.ListFailoverHistoryResponse) *shared.ListFailoverHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListFailoverHistoryResponse{\n\t\tFailoverEvents: FromFailoverEventArray(t.FailoverEvents),\n\t\tNextPageToken:  t.NextPageToken,\n\t}\n}\n\n// ToListFailoverHistoryResponse converts thrift ListFailoverHistoryResponse type to internal\nfunc ToListFailoverHistoryResponse(t *shared.ListFailoverHistoryResponse) *types.ListFailoverHistoryResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListFailoverHistoryResponse{\n\t\tFailoverEvents: ToFailoverEventArray(t.FailoverEvents),\n\t\tNextPageToken:  t.NextPageToken,\n\t}\n}\n\n// ToListFailoverHistoryRequestFilters converts thrift ListFailoverHistoryRequestFilters type to internal\nfunc ToListFailoverHistoryRequestFilters(t *shared.ListFailoverHistoryRequestFilters) *types.ListFailoverHistoryRequestFilters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListFailoverHistoryRequestFilters{\n\t\tDomainID: t.GetDomainID(),\n\t}\n}\n\n// FromListFailoverHistoryRequestFilters converts internal ListFailoverHistoryRequestFilters type to thrift\nfunc FromListFailoverHistoryRequestFilters(t *types.ListFailoverHistoryRequestFilters) *shared.ListFailoverHistoryRequestFilters {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListFailoverHistoryRequestFilters{\n\t\tDomainID: &t.DomainID,\n\t}\n}\n\n// ToPaginationOptions converts thrift PaginationOptions type to internal\nfunc ToPaginationOptions(t *shared.PaginationOptions) *types.PaginationOptions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PaginationOptions{\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromPaginationOptions converts internal PaginationOptions type to thrift\nfunc FromPaginationOptions(t *types.PaginationOptions) *shared.PaginationOptions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PaginationOptions{\n\t\tPageSize:      t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToFailoverEvent converts thrift FailoverEvent type to internal\nfunc ToFailoverEvent(t *shared.FailoverEvent) *types.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.FailoverEvent{\n\t\tID:               t.ID,\n\t\tCreatedTime:      t.CreatedTime,\n\t\tFailoverType:     ToFailoverType(t.FailoverType),\n\t\tClusterFailovers: ToClusterFailoverArray(t.ClusterFailovers),\n\t}\n}\n\n// FromFailoverEvent converts internal FailoverEvent type to thrift\nfunc FromFailoverEvent(t *types.FailoverEvent) *shared.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.FailoverEvent{\n\t\tID:               t.ID,\n\t\tCreatedTime:      t.CreatedTime,\n\t\tFailoverType:     FromFailoverType(t.FailoverType),\n\t\tClusterFailovers: FromClusterFailoverArray(t.ClusterFailovers),\n\t}\n}\n\n// ToFailoverEventArray converts thrift FailoverEvent array type to internal\nfunc ToFailoverEventArray(t []*shared.FailoverEvent) []*types.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.FailoverEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = ToFailoverEvent(t[i])\n\t}\n\treturn v\n}\n\n// FromFailoverEventArray converts internal FailoverEvent array type to thrift\nfunc FromFailoverEventArray(t []*types.FailoverEvent) []*shared.FailoverEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.FailoverEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = FromFailoverEvent(t[i])\n\t}\n\treturn v\n}\n\n// ToFailoverType converts thrift FailoverType type to internal\nfunc ToFailoverType(t *shared.FailoverType) *types.FailoverType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := types.FailoverType(*t)\n\treturn &v\n}\n\n// FromFailoverType converts internal FailoverType type to thrift\nfunc FromFailoverType(t *types.FailoverType) *shared.FailoverType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := shared.FailoverType(*t)\n\treturn &v\n}\n\n// ToClusterFailover converts thrift ClusterFailover type to internal\nfunc ToClusterFailover(t *shared.ClusterFailover) *types.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterFailover{\n\t\tFromCluster:      ToActiveClusterInfo(t.FromCluster),\n\t\tToCluster:        ToActiveClusterInfo(t.ToCluster),\n\t\tClusterAttribute: ToClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\n// FromClusterFailover converts internal ClusterFailover type to thrift\nfunc FromClusterFailover(t *types.ClusterFailover) *shared.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ClusterFailover{\n\t\tFromCluster:      FromActiveClusterInfo(t.FromCluster),\n\t\tToCluster:        FromActiveClusterInfo(t.ToCluster),\n\t\tClusterAttribute: FromClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\n// ToClusterFailoverArray converts thrift ClusterFailover array type to internal\nfunc ToClusterFailoverArray(t []*shared.ClusterFailover) []*types.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ClusterFailover, len(t))\n\tfor i := range t {\n\t\tv[i] = ToClusterFailover(t[i])\n\t}\n\treturn v\n}\n\n// FromClusterFailoverArray converts internal ClusterFailover array type to thrift\nfunc FromClusterFailoverArray(t []*types.ClusterFailover) []*shared.ClusterFailover {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.ClusterFailover, len(t))\n\tfor i := range t {\n\t\tv[i] = FromClusterFailover(t[i])\n\t}\n\treturn v\n}\n\n// ToActiveClusterInfo converts thrift ActiveClusterInfo type to internal\nfunc ToActiveClusterInfo(t *shared.ActiveClusterInfo) *types.ActiveClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActiveClusterInfo{\n\t\tActiveClusterName: t.GetActiveClusterName(),\n\t\tFailoverVersion:   t.GetFailoverVersion(),\n\t}\n}\n\n// FromActiveClusterInfo converts internal ActiveClusterInfo type to thrift\nfunc FromActiveClusterInfo(t *types.ActiveClusterInfo) *shared.ActiveClusterInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActiveClusterInfo{\n\t\tActiveClusterName: &t.ActiveClusterName,\n\t\tFailoverVersion:   &t.FailoverVersion,\n\t}\n}\n\n// FromListOpenWorkflowExecutionsRequest converts internal ListOpenWorkflowExecutionsRequest type to thrift\nfunc FromListOpenWorkflowExecutionsRequest(t *types.ListOpenWorkflowExecutionsRequest) *shared.ListOpenWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          &t.Domain,\n\t\tMaximumPageSize: &t.MaximumPageSize,\n\t\tNextPageToken:   t.NextPageToken,\n\t\tStartTimeFilter: FromStartTimeFilter(t.StartTimeFilter),\n\t\tExecutionFilter: FromWorkflowExecutionFilter(t.ExecutionFilter),\n\t\tTypeFilter:      FromWorkflowTypeFilter(t.TypeFilter),\n\t}\n}\n\n// ToListOpenWorkflowExecutionsRequest converts thrift ListOpenWorkflowExecutionsRequest type to internal\nfunc ToListOpenWorkflowExecutionsRequest(t *shared.ListOpenWorkflowExecutionsRequest) *types.ListOpenWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          t.GetDomain(),\n\t\tMaximumPageSize: t.GetMaximumPageSize(),\n\t\tNextPageToken:   t.NextPageToken,\n\t\tStartTimeFilter: ToStartTimeFilter(t.StartTimeFilter),\n\t\tExecutionFilter: ToWorkflowExecutionFilter(t.ExecutionFilter),\n\t\tTypeFilter:      ToWorkflowTypeFilter(t.TypeFilter),\n\t}\n}\n\n// FromListOpenWorkflowExecutionsResponse converts internal ListOpenWorkflowExecutionsResponse type to thrift\nfunc FromListOpenWorkflowExecutionsResponse(t *types.ListOpenWorkflowExecutionsResponse) *shared.ListOpenWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListOpenWorkflowExecutionsResponse converts thrift ListOpenWorkflowExecutionsResponse type to internal\nfunc ToListOpenWorkflowExecutionsResponse(t *shared.ListOpenWorkflowExecutionsResponse) *types.ListOpenWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromListTaskListPartitionsRequest converts internal ListTaskListPartitionsRequest type to thrift\nfunc FromListTaskListPartitionsRequest(t *types.ListTaskListPartitionsRequest) *shared.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListTaskListPartitionsRequest{\n\t\tDomain:   &t.Domain,\n\t\tTaskList: FromTaskList(t.TaskList),\n\t}\n}\n\n// ToListTaskListPartitionsRequest converts thrift ListTaskListPartitionsRequest type to internal\nfunc ToListTaskListPartitionsRequest(t *shared.ListTaskListPartitionsRequest) *types.ListTaskListPartitionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListTaskListPartitionsRequest{\n\t\tDomain:   t.GetDomain(),\n\t\tTaskList: ToTaskList(t.TaskList),\n\t}\n}\n\n// FromListTaskListPartitionsResponse converts internal ListTaskListPartitionsResponse type to thrift\nfunc FromListTaskListPartitionsResponse(t *types.ListTaskListPartitionsResponse) *shared.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: FromTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: FromTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\n// ToListTaskListPartitionsResponse converts thrift ListTaskListPartitionsResponse type to internal\nfunc ToListTaskListPartitionsResponse(t *shared.ListTaskListPartitionsResponse) *types.ListTaskListPartitionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: ToTaskListPartitionMetadataArray(t.ActivityTaskListPartitions),\n\t\tDecisionTaskListPartitions: ToTaskListPartitionMetadataArray(t.DecisionTaskListPartitions),\n\t}\n}\n\n// FromGetTaskListsByDomainRequest converts internal GetTaskListsByDomainRequest type to thrift\nfunc FromGetTaskListsByDomainRequest(t *types.GetTaskListsByDomainRequest) *shared.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetTaskListsByDomainRequest{\n\t\tDomainName: &t.Domain,\n\t}\n}\n\n// ToGetTaskListsByDomainRequest converts thrift GetTaskListsByDomainRequest type to internal\nfunc ToGetTaskListsByDomainRequest(t *shared.GetTaskListsByDomainRequest) *types.GetTaskListsByDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetTaskListsByDomainRequest{\n\t\tDomain: t.GetDomainName(),\n\t}\n}\n\n// FromGetTaskListsByDomainResponse converts internal GetTaskListsByDomainResponse type to thrift\nfunc FromGetTaskListsByDomainResponse(t *types.GetTaskListsByDomainResponse) *shared.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &shared.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: FromDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: FromDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\n// ToGetTaskListsByDomainResponse converts thrift GetTaskListsByDomainResponse type to internal\nfunc ToGetTaskListsByDomainResponse(t *shared.GetTaskListsByDomainResponse) *types.GetTaskListsByDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: ToDescribeTaskListResponseMap(t.GetDecisionTaskListMap()),\n\t\tActivityTaskListMap: ToDescribeTaskListResponseMap(t.GetActivityTaskListMap()),\n\t}\n}\n\n// FromDescribeTaskListResponseMap converts internal DescribeTaskListResponse map type to thrift\nfunc FromDescribeTaskListResponseMap(t map[string]*types.DescribeTaskListResponse) map[string]*shared.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*shared.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = FromDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\n// ToDescribeTaskListResponseMap converts thrift DescribeTaskListResponse map type to internal\nfunc ToDescribeTaskListResponseMap(t map[string]*shared.DescribeTaskListResponse) map[string]*types.DescribeTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttaskListMap := make(map[string]*types.DescribeTaskListResponse, len(t))\n\tfor key, value := range t {\n\t\ttaskListMap[key] = ToDescribeTaskListResponse(value)\n\t}\n\treturn taskListMap\n}\n\n// FromListWorkflowExecutionsRequest converts internal ListWorkflowExecutionsRequest type to thrift\nfunc FromListWorkflowExecutionsRequest(t *types.ListWorkflowExecutionsRequest) *shared.ListWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListWorkflowExecutionsRequest{\n\t\tDomain:        &t.Domain,\n\t\tPageSize:      &t.PageSize,\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         &t.Query,\n\t}\n}\n\n// ToListWorkflowExecutionsRequest converts thrift ListWorkflowExecutionsRequest type to internal\nfunc ToListWorkflowExecutionsRequest(t *shared.ListWorkflowExecutionsRequest) *types.ListWorkflowExecutionsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        t.GetDomain(),\n\t\tPageSize:      t.GetPageSize(),\n\t\tNextPageToken: t.NextPageToken,\n\t\tQuery:         t.GetQuery(),\n\t}\n}\n\n// FromListWorkflowExecutionsResponse converts internal ListWorkflowExecutionsResponse type to thrift\nfunc FromListWorkflowExecutionsResponse(t *types.ListWorkflowExecutionsResponse) *shared.ListWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ListWorkflowExecutionsResponse{\n\t\tExecutions:    FromWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// ToListWorkflowExecutionsResponse converts thrift ListWorkflowExecutionsResponse type to internal\nfunc ToListWorkflowExecutionsResponse(t *shared.ListWorkflowExecutionsResponse) *types.ListWorkflowExecutionsResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ListWorkflowExecutionsResponse{\n\t\tExecutions:    ToWorkflowExecutionInfoArray(t.Executions),\n\t\tNextPageToken: t.NextPageToken,\n\t}\n}\n\n// FromMarkerRecordedEventAttributes converts internal MarkerRecordedEventAttributes type to thrift\nfunc FromMarkerRecordedEventAttributes(t *types.MarkerRecordedEventAttributes) *shared.MarkerRecordedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   &t.MarkerName,\n\t\tDetails:                      t.Details,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tHeader:                       FromHeader(t.Header),\n\t}\n}\n\n// ToMarkerRecordedEventAttributes converts thrift MarkerRecordedEventAttributes type to internal\nfunc ToMarkerRecordedEventAttributes(t *shared.MarkerRecordedEventAttributes) *types.MarkerRecordedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   t.GetMarkerName(),\n\t\tDetails:                      t.Details,\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tHeader:                       ToHeader(t.Header),\n\t}\n}\n\n// FromMemo converts internal Memo type to thrift\nfunc FromMemo(t *types.Memo) *shared.Memo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.Memo{\n\t\tFields: t.Fields,\n\t}\n}\n\n// ToMemo converts thrift Memo type to internal\nfunc ToMemo(t *shared.Memo) *types.Memo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Memo{\n\t\tFields: t.Fields,\n\t}\n}\n\n// FromParentClosePolicy converts internal ParentClosePolicy type to thrift\nfunc FromParentClosePolicy(t *types.ParentClosePolicy) *shared.ParentClosePolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ParentClosePolicyAbandon:\n\t\tv := shared.ParentClosePolicyAbandon\n\t\treturn &v\n\tcase types.ParentClosePolicyRequestCancel:\n\t\tv := shared.ParentClosePolicyRequestCancel\n\t\treturn &v\n\tcase types.ParentClosePolicyTerminate:\n\t\tv := shared.ParentClosePolicyTerminate\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToParentClosePolicy converts thrift ParentClosePolicy type to internal\nfunc ToParentClosePolicy(t *shared.ParentClosePolicy) *types.ParentClosePolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.ParentClosePolicyAbandon:\n\t\tv := types.ParentClosePolicyAbandon\n\t\treturn &v\n\tcase shared.ParentClosePolicyRequestCancel:\n\t\tv := types.ParentClosePolicyRequestCancel\n\t\treturn &v\n\tcase shared.ParentClosePolicyTerminate:\n\t\tv := types.ParentClosePolicyTerminate\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromPendingActivityInfo converts internal PendingActivityInfo type to thrift\nfunc FromPendingActivityInfo(t *types.PendingActivityInfo) *shared.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PendingActivityInfo{\n\t\tActivityID:             &t.ActivityID,\n\t\tActivityType:           FromActivityType(t.ActivityType),\n\t\tState:                  FromPendingActivityState(t.State),\n\t\tHeartbeatDetails:       t.HeartbeatDetails,\n\t\tLastHeartbeatTimestamp: t.LastHeartbeatTimestamp,\n\t\tLastStartedTimestamp:   t.LastStartedTimestamp,\n\t\tAttempt:                &t.Attempt,\n\t\tMaximumAttempts:        &t.MaximumAttempts,\n\t\tScheduledTimestamp:     t.ScheduledTimestamp,\n\t\tExpirationTimestamp:    t.ExpirationTimestamp,\n\t\tLastFailureReason:      t.LastFailureReason,\n\t\tLastWorkerIdentity:     &t.LastWorkerIdentity,\n\t\tLastFailureDetails:     t.LastFailureDetails,\n\t\tStartedWorkerIdentity:  &t.StartedWorkerIdentity,\n\t\tScheduleID:             &t.ScheduleID,\n\t}\n}\n\n// ToPendingActivityInfo converts thrift PendingActivityInfo type to internal\nfunc ToPendingActivityInfo(t *shared.PendingActivityInfo) *types.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingActivityInfo{\n\t\tActivityID:             t.GetActivityID(),\n\t\tActivityType:           ToActivityType(t.ActivityType),\n\t\tState:                  ToPendingActivityState(t.State),\n\t\tHeartbeatDetails:       t.HeartbeatDetails,\n\t\tLastHeartbeatTimestamp: t.LastHeartbeatTimestamp,\n\t\tLastStartedTimestamp:   t.LastStartedTimestamp,\n\t\tAttempt:                t.GetAttempt(),\n\t\tMaximumAttempts:        t.GetMaximumAttempts(),\n\t\tScheduledTimestamp:     t.ScheduledTimestamp,\n\t\tExpirationTimestamp:    t.ExpirationTimestamp,\n\t\tLastFailureReason:      t.LastFailureReason,\n\t\tLastWorkerIdentity:     t.GetLastWorkerIdentity(),\n\t\tLastFailureDetails:     t.LastFailureDetails,\n\t\tStartedWorkerIdentity:  t.GetStartedWorkerIdentity(),\n\t\tScheduleID:             t.GetScheduleID(),\n\t}\n}\n\n// FromPendingActivityState converts internal PendingActivityState type to thrift\nfunc FromPendingActivityState(t *types.PendingActivityState) *shared.PendingActivityState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.PendingActivityStateScheduled:\n\t\tv := shared.PendingActivityStateScheduled\n\t\treturn &v\n\tcase types.PendingActivityStateStarted:\n\t\tv := shared.PendingActivityStateStarted\n\t\treturn &v\n\tcase types.PendingActivityStateCancelRequested:\n\t\tv := shared.PendingActivityStateCancelRequested\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToPendingActivityState converts thrift PendingActivityState type to internal\nfunc ToPendingActivityState(t *shared.PendingActivityState) *types.PendingActivityState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.PendingActivityStateScheduled:\n\t\tv := types.PendingActivityStateScheduled\n\t\treturn &v\n\tcase shared.PendingActivityStateStarted:\n\t\tv := types.PendingActivityStateStarted\n\t\treturn &v\n\tcase shared.PendingActivityStateCancelRequested:\n\t\tv := types.PendingActivityStateCancelRequested\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromPendingChildExecutionInfo converts internal PendingChildExecutionInfo type to thrift\nfunc FromPendingChildExecutionInfo(t *types.PendingChildExecutionInfo) *shared.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PendingChildExecutionInfo{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowID:        &t.WorkflowID,\n\t\tRunID:             &t.RunID,\n\t\tWorkflowTypName:   &t.WorkflowTypeName,\n\t\tInitiatedID:       &t.InitiatedID,\n\t\tParentClosePolicy: FromParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// ToPendingChildExecutionInfo converts thrift PendingChildExecutionInfo type to internal\nfunc ToPendingChildExecutionInfo(t *shared.PendingChildExecutionInfo) *types.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingChildExecutionInfo{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowID:        t.GetWorkflowID(),\n\t\tRunID:             t.GetRunID(),\n\t\tWorkflowTypeName:  t.GetWorkflowTypName(),\n\t\tInitiatedID:       t.GetInitiatedID(),\n\t\tParentClosePolicy: ToParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// FromPendingDecisionInfo converts internal PendingDecisionInfo type to thrift\nfunc FromPendingDecisionInfo(t *types.PendingDecisionInfo) *shared.PendingDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PendingDecisionInfo{\n\t\tState:                      FromPendingDecisionState(t.State),\n\t\tScheduledTimestamp:         t.ScheduledTimestamp,\n\t\tStartedTimestamp:           t.StartedTimestamp,\n\t\tAttempt:                    &t.Attempt,\n\t\tOriginalScheduledTimestamp: t.OriginalScheduledTimestamp,\n\t\tScheduleID:                 &t.ScheduleID,\n\t}\n}\n\n// ToPendingDecisionInfo converts thrift PendingDecisionInfo type to internal\nfunc ToPendingDecisionInfo(t *shared.PendingDecisionInfo) *types.PendingDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PendingDecisionInfo{\n\t\tState:                      ToPendingDecisionState(t.State),\n\t\tScheduledTimestamp:         t.ScheduledTimestamp,\n\t\tStartedTimestamp:           t.StartedTimestamp,\n\t\tAttempt:                    t.GetAttempt(),\n\t\tOriginalScheduledTimestamp: t.OriginalScheduledTimestamp,\n\t\tScheduleID:                 t.GetScheduleID(),\n\t}\n}\n\n// FromPendingDecisionState converts internal PendingDecisionState type to thrift\nfunc FromPendingDecisionState(t *types.PendingDecisionState) *shared.PendingDecisionState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.PendingDecisionStateScheduled:\n\t\tv := shared.PendingDecisionStateScheduled\n\t\treturn &v\n\tcase types.PendingDecisionStateStarted:\n\t\tv := shared.PendingDecisionStateStarted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToPendingDecisionState converts thrift PendingDecisionState type to internal\nfunc ToPendingDecisionState(t *shared.PendingDecisionState) *types.PendingDecisionState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.PendingDecisionStateScheduled:\n\t\tv := types.PendingDecisionStateScheduled\n\t\treturn &v\n\tcase shared.PendingDecisionStateStarted:\n\t\tv := types.PendingDecisionStateStarted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromPollForActivityTaskRequest converts internal PollForActivityTaskRequest type to thrift\nfunc FromPollForActivityTaskRequest(t *types.PollForActivityTaskRequest) *shared.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollForActivityTaskRequest{\n\t\tDomain:           &t.Domain,\n\t\tTaskList:         FromTaskList(t.TaskList),\n\t\tIdentity:         &t.Identity,\n\t\tTaskListMetadata: FromTaskListMetadata(t.TaskListMetadata),\n\t}\n}\n\n// ToPollForActivityTaskRequest converts thrift PollForActivityTaskRequest type to internal\nfunc ToPollForActivityTaskRequest(t *shared.PollForActivityTaskRequest) *types.PollForActivityTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForActivityTaskRequest{\n\t\tDomain:           t.GetDomain(),\n\t\tTaskList:         ToTaskList(t.TaskList),\n\t\tIdentity:         t.GetIdentity(),\n\t\tTaskListMetadata: ToTaskListMetadata(t.TaskListMetadata),\n\t}\n}\n\n// FromPollForActivityTaskResponse converts internal PollForActivityTaskResponse type to thrift\nfunc FromPollForActivityTaskResponse(t *types.PollForActivityTaskResponse) *shared.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               FromWorkflowExecution(t.WorkflowExecution),\n\t\tActivityId:                      &t.ActivityID,\n\t\tActivityType:                    FromActivityType(t.ActivityType),\n\t\tInput:                           t.Input,\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tScheduleToCloseTimeoutSeconds:   t.ScheduleToCloseTimeoutSeconds,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tStartToCloseTimeoutSeconds:      t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:         t.HeartbeatTimeoutSeconds,\n\t\tAttempt:                         &t.Attempt,\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    FromWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  &t.WorkflowDomain,\n\t\tHeader:                          FromHeader(t.Header),\n\t\tAutoConfigHint:                  FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// ToPollForActivityTaskResponse converts thrift PollForActivityTaskResponse type to internal\nfunc ToPollForActivityTaskResponse(t *shared.PollForActivityTaskResponse) *types.PollForActivityTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForActivityTaskResponse{\n\t\tTaskToken:                       t.TaskToken,\n\t\tWorkflowExecution:               ToWorkflowExecution(t.WorkflowExecution),\n\t\tActivityID:                      t.GetActivityId(),\n\t\tActivityType:                    ToActivityType(t.ActivityType),\n\t\tInput:                           t.Input,\n\t\tScheduledTimestamp:              t.ScheduledTimestamp,\n\t\tScheduleToCloseTimeoutSeconds:   t.ScheduleToCloseTimeoutSeconds,\n\t\tStartedTimestamp:                t.StartedTimestamp,\n\t\tStartToCloseTimeoutSeconds:      t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:         t.HeartbeatTimeoutSeconds,\n\t\tAttempt:                         t.GetAttempt(),\n\t\tScheduledTimestampOfThisAttempt: t.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                t.HeartbeatDetails,\n\t\tWorkflowType:                    ToWorkflowType(t.WorkflowType),\n\t\tWorkflowDomain:                  t.GetWorkflowDomain(),\n\t\tHeader:                          ToHeader(t.Header),\n\t\tAutoConfigHint:                  ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// FromPollForDecisionTaskRequest converts internal PollForDecisionTaskRequest type to thrift\nfunc FromPollForDecisionTaskRequest(t *types.PollForDecisionTaskRequest) *shared.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollForDecisionTaskRequest{\n\t\tDomain:         &t.Domain,\n\t\tTaskList:       FromTaskList(t.TaskList),\n\t\tIdentity:       &t.Identity,\n\t\tBinaryChecksum: &t.BinaryChecksum,\n\t}\n}\n\n// ToPollForDecisionTaskRequest converts thrift PollForDecisionTaskRequest type to internal\nfunc ToPollForDecisionTaskRequest(t *shared.PollForDecisionTaskRequest) *types.PollForDecisionTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForDecisionTaskRequest{\n\t\tDomain:         t.GetDomain(),\n\t\tTaskList:       ToTaskList(t.TaskList),\n\t\tIdentity:       t.GetIdentity(),\n\t\tBinaryChecksum: t.GetBinaryChecksum(),\n\t}\n}\n\n// FromPollForDecisionTaskResponse converts internal PollForDecisionTaskResponse type to thrift\nfunc FromPollForDecisionTaskResponse(t *types.PollForDecisionTaskResponse) *shared.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         FromWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              FromWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventId:    t.PreviousStartedEventID,\n\t\tStartedEventId:            &t.StartedEventID,\n\t\tAttempt:                   &t.Attempt,\n\t\tBacklogCountHint:          &t.BacklogCountHint,\n\t\tHistory:                   FromHistory(t.History),\n\t\tNextPageToken:             t.NextPageToken,\n\t\tQuery:                     FromWorkflowQuery(t.Query),\n\t\tWorkflowExecutionTaskList: FromTaskList(t.WorkflowExecutionTaskList),\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   FromWorkflowQueryMap(t.Queries),\n\t\tNextEventId:               &t.NextEventID,\n\t\tTotalHistoryBytes:         &t.TotalHistoryBytes,\n\t\tAutoConfigHint:            FromAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// ToPollForDecisionTaskResponse converts thrift PollForDecisionTaskResponse type to internal\nfunc ToPollForDecisionTaskResponse(t *shared.PollForDecisionTaskResponse) *types.PollForDecisionTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollForDecisionTaskResponse{\n\t\tTaskToken:                 t.TaskToken,\n\t\tWorkflowExecution:         ToWorkflowExecution(t.WorkflowExecution),\n\t\tWorkflowType:              ToWorkflowType(t.WorkflowType),\n\t\tPreviousStartedEventID:    t.PreviousStartedEventId,\n\t\tStartedEventID:            t.GetStartedEventId(),\n\t\tAttempt:                   t.GetAttempt(),\n\t\tBacklogCountHint:          t.GetBacklogCountHint(),\n\t\tHistory:                   ToHistory(t.History),\n\t\tNextPageToken:             t.NextPageToken,\n\t\tQuery:                     ToWorkflowQuery(t.Query),\n\t\tWorkflowExecutionTaskList: ToTaskList(t.WorkflowExecutionTaskList),\n\t\tScheduledTimestamp:        t.ScheduledTimestamp,\n\t\tStartedTimestamp:          t.StartedTimestamp,\n\t\tQueries:                   ToWorkflowQueryMap(t.Queries),\n\t\tNextEventID:               t.GetNextEventId(),\n\t\tTotalHistoryBytes:         t.GetTotalHistoryBytes(),\n\t\tAutoConfigHint:            ToAutoConfigHint(t.AutoConfigHint),\n\t}\n}\n\n// FromPollerInfo converts internal PollerInfo type to thrift\nfunc FromPollerInfo(t *types.PollerInfo) *shared.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.PollerInfo{\n\t\tLastAccessTime: t.LastAccessTime,\n\t\tIdentity:       &t.Identity,\n\t\tRatePerSecond:  &t.RatePerSecond,\n\t}\n}\n\n// ToPollerInfo converts thrift PollerInfo type to internal\nfunc ToPollerInfo(t *shared.PollerInfo) *types.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.PollerInfo{\n\t\tLastAccessTime: t.LastAccessTime,\n\t\tIdentity:       t.GetIdentity(),\n\t\tRatePerSecond:  t.GetRatePerSecond(),\n\t}\n}\n\n// FromQueryConsistencyLevel converts internal QueryConsistencyLevel type to thrift\nfunc FromQueryConsistencyLevel(t *types.QueryConsistencyLevel) *shared.QueryConsistencyLevel {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.QueryConsistencyLevelEventual:\n\t\tv := shared.QueryConsistencyLevelEventual\n\t\treturn &v\n\tcase types.QueryConsistencyLevelStrong:\n\t\tv := shared.QueryConsistencyLevelStrong\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToQueryConsistencyLevel converts thrift QueryConsistencyLevel type to internal\nfunc ToQueryConsistencyLevel(t *shared.QueryConsistencyLevel) *types.QueryConsistencyLevel {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.QueryConsistencyLevelEventual:\n\t\tv := types.QueryConsistencyLevelEventual\n\t\treturn &v\n\tcase shared.QueryConsistencyLevelStrong:\n\t\tv := types.QueryConsistencyLevelStrong\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromQueryFailedError converts internal QueryFailedError type to thrift\nfunc FromQueryFailedError(t *types.QueryFailedError) *shared.QueryFailedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueryFailedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToQueryFailedError converts thrift QueryFailedError type to internal\nfunc ToQueryFailedError(t *shared.QueryFailedError) *types.QueryFailedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryFailedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromQueryRejectCondition converts internal QueryRejectCondition type to thrift\nfunc FromQueryRejectCondition(t *types.QueryRejectCondition) *shared.QueryRejectCondition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.QueryRejectConditionNotOpen:\n\t\tv := shared.QueryRejectConditionNotOpen\n\t\treturn &v\n\tcase types.QueryRejectConditionNotCompletedCleanly:\n\t\tv := shared.QueryRejectConditionNotCompletedCleanly\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToQueryRejectCondition converts thrift QueryRejectCondition type to internal\nfunc ToQueryRejectCondition(t *shared.QueryRejectCondition) *types.QueryRejectCondition {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.QueryRejectConditionNotOpen:\n\t\tv := types.QueryRejectConditionNotOpen\n\t\treturn &v\n\tcase shared.QueryRejectConditionNotCompletedCleanly:\n\t\tv := types.QueryRejectConditionNotCompletedCleanly\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromQueryRejected converts internal QueryRejected type to thrift\nfunc FromQueryRejected(t *types.QueryRejected) *shared.QueryRejected {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueryRejected{\n\t\tCloseStatus: FromWorkflowExecutionCloseStatus(t.CloseStatus),\n\t}\n}\n\n// ToQueryRejected converts thrift QueryRejected type to internal\nfunc ToQueryRejected(t *shared.QueryRejected) *types.QueryRejected {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryRejected{\n\t\tCloseStatus: ToWorkflowExecutionCloseStatus(t.CloseStatus),\n\t}\n}\n\n// FromQueryResultType converts internal QueryResultType type to thrift\nfunc FromQueryResultType(t *types.QueryResultType) *shared.QueryResultType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.QueryResultTypeAnswered:\n\t\tv := shared.QueryResultTypeAnswered\n\t\treturn &v\n\tcase types.QueryResultTypeFailed:\n\t\tv := shared.QueryResultTypeFailed\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToQueryResultType converts thrift QueryResultType type to internal\nfunc ToQueryResultType(t *shared.QueryResultType) *types.QueryResultType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.QueryResultTypeAnswered:\n\t\tv := types.QueryResultTypeAnswered\n\t\treturn &v\n\tcase shared.QueryResultTypeFailed:\n\t\tv := types.QueryResultTypeFailed\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromQueryTaskCompletedType converts internal QueryTaskCompletedType type to thrift\nfunc FromQueryTaskCompletedType(t *types.QueryTaskCompletedType) *shared.QueryTaskCompletedType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.QueryTaskCompletedTypeCompleted:\n\t\tv := shared.QueryTaskCompletedTypeCompleted\n\t\treturn &v\n\tcase types.QueryTaskCompletedTypeFailed:\n\t\tv := shared.QueryTaskCompletedTypeFailed\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToQueryTaskCompletedType converts thrift QueryTaskCompletedType type to internal\nfunc ToQueryTaskCompletedType(t *shared.QueryTaskCompletedType) *types.QueryTaskCompletedType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.QueryTaskCompletedTypeCompleted:\n\t\tv := types.QueryTaskCompletedTypeCompleted\n\t\treturn &v\n\tcase shared.QueryTaskCompletedTypeFailed:\n\t\tv := types.QueryTaskCompletedTypeFailed\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromQueryWorkflowRequest converts internal QueryWorkflowRequest type to thrift\nfunc FromQueryWorkflowRequest(t *types.QueryWorkflowRequest) *shared.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueryWorkflowRequest{\n\t\tDomain:                &t.Domain,\n\t\tExecution:             FromWorkflowExecution(t.Execution),\n\t\tQuery:                 FromWorkflowQuery(t.Query),\n\t\tQueryRejectCondition:  FromQueryRejectCondition(t.QueryRejectCondition),\n\t\tQueryConsistencyLevel: FromQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// ToQueryWorkflowRequest converts thrift QueryWorkflowRequest type to internal\nfunc ToQueryWorkflowRequest(t *shared.QueryWorkflowRequest) *types.QueryWorkflowRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryWorkflowRequest{\n\t\tDomain:                t.GetDomain(),\n\t\tExecution:             ToWorkflowExecution(t.Execution),\n\t\tQuery:                 ToWorkflowQuery(t.Query),\n\t\tQueryRejectCondition:  ToQueryRejectCondition(t.QueryRejectCondition),\n\t\tQueryConsistencyLevel: ToQueryConsistencyLevel(t.QueryConsistencyLevel),\n\t}\n}\n\n// FromQueryWorkflowResponse converts internal QueryWorkflowResponse type to thrift\nfunc FromQueryWorkflowResponse(t *types.QueryWorkflowResponse) *shared.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueryWorkflowResponse{\n\t\tQueryResult:   t.QueryResult,\n\t\tQueryRejected: FromQueryRejected(t.QueryRejected),\n\t}\n}\n\n// ToQueryWorkflowResponse converts thrift QueryWorkflowResponse type to internal\nfunc ToQueryWorkflowResponse(t *shared.QueryWorkflowResponse) *types.QueryWorkflowResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueryWorkflowResponse{\n\t\tQueryResult:   t.QueryResult,\n\t\tQueryRejected: ToQueryRejected(t.QueryRejected),\n\t}\n}\n\n// FromAdminReapplyEventsRequest converts internal ReapplyEventsRequest type to thrift\nfunc FromAdminReapplyEventsRequest(t *types.ReapplyEventsRequest) *shared.ReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ReapplyEventsRequest{\n\t\tDomainName:        &t.DomainName,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tEvents:            FromDataBlob(t.Events),\n\t}\n}\n\n// ToAdminReapplyEventsRequest converts thrift ReapplyEventsRequest type to internal\nfunc ToAdminReapplyEventsRequest(t *shared.ReapplyEventsRequest) *types.ReapplyEventsRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ReapplyEventsRequest{\n\t\tDomainName:        t.GetDomainName(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tEvents:            ToDataBlob(t.Events),\n\t}\n}\n\n// FromRecordActivityTaskHeartbeatByIDRequest converts internal RecordActivityTaskHeartbeatByIDRequest type to thrift\nfunc FromRecordActivityTaskHeartbeatByIDRequest(t *types.RecordActivityTaskHeartbeatByIDRequest) *shared.RecordActivityTaskHeartbeatByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     &t.Domain,\n\t\tWorkflowID: &t.WorkflowID,\n\t\tRunID:      &t.RunID,\n\t\tActivityID: &t.ActivityID,\n\t\tDetails:    t.Details,\n\t\tIdentity:   &t.Identity,\n\t}\n}\n\n// ToRecordActivityTaskHeartbeatByIDRequest converts thrift RecordActivityTaskHeartbeatByIDRequest type to internal\nfunc ToRecordActivityTaskHeartbeatByIDRequest(t *shared.RecordActivityTaskHeartbeatByIDRequest) *types.RecordActivityTaskHeartbeatByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     t.GetDomain(),\n\t\tWorkflowID: t.GetWorkflowID(),\n\t\tRunID:      t.GetRunID(),\n\t\tActivityID: t.GetActivityID(),\n\t\tDetails:    t.Details,\n\t\tIdentity:   t.GetIdentity(),\n\t}\n}\n\n// FromRecordActivityTaskHeartbeatRequest converts internal RecordActivityTaskHeartbeatRequest type to thrift\nfunc FromRecordActivityTaskHeartbeatRequest(t *types.RecordActivityTaskHeartbeatRequest) *shared.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   t.Details,\n\t\tIdentity:  &t.Identity,\n\t}\n}\n\n// ToRecordActivityTaskHeartbeatRequest converts thrift RecordActivityTaskHeartbeatRequest type to internal\nfunc ToRecordActivityTaskHeartbeatRequest(t *shared.RecordActivityTaskHeartbeatRequest) *types.RecordActivityTaskHeartbeatRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   t.Details,\n\t\tIdentity:  t.GetIdentity(),\n\t}\n}\n\n// FromRecordActivityTaskHeartbeatResponse converts internal RecordActivityTaskHeartbeatResponse type to thrift\nfunc FromRecordActivityTaskHeartbeatResponse(t *types.RecordActivityTaskHeartbeatResponse) *shared.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: &t.CancelRequested,\n\t}\n}\n\n// ToRecordActivityTaskHeartbeatResponse converts thrift RecordActivityTaskHeartbeatResponse type to internal\nfunc ToRecordActivityTaskHeartbeatResponse(t *shared.RecordActivityTaskHeartbeatResponse) *types.RecordActivityTaskHeartbeatResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: t.GetCancelRequested(),\n\t}\n}\n\n// FromRecordMarkerDecisionAttributes converts internal RecordMarkerDecisionAttributes type to thrift\nfunc FromRecordMarkerDecisionAttributes(t *types.RecordMarkerDecisionAttributes) *shared.RecordMarkerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RecordMarkerDecisionAttributes{\n\t\tMarkerName: &t.MarkerName,\n\t\tDetails:    t.Details,\n\t\tHeader:     FromHeader(t.Header),\n\t}\n}\n\n// ToRecordMarkerDecisionAttributes converts thrift RecordMarkerDecisionAttributes type to internal\nfunc ToRecordMarkerDecisionAttributes(t *shared.RecordMarkerDecisionAttributes) *types.RecordMarkerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RecordMarkerDecisionAttributes{\n\t\tMarkerName: t.GetMarkerName(),\n\t\tDetails:    t.Details,\n\t\tHeader:     ToHeader(t.Header),\n\t}\n}\n\n// FromRefreshWorkflowTasksRequest converts internal RefreshWorkflowTasksRequest type to thrift\nfunc FromRefreshWorkflowTasksRequest(t *types.RefreshWorkflowTasksRequest) *shared.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RefreshWorkflowTasksRequest{\n\t\tDomain:    &t.Domain,\n\t\tExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToRefreshWorkflowTasksRequest converts thrift RefreshWorkflowTasksRequest type to internal\nfunc ToRefreshWorkflowTasksRequest(t *shared.RefreshWorkflowTasksRequest) *types.RefreshWorkflowTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RefreshWorkflowTasksRequest{\n\t\tDomain:    t.GetDomain(),\n\t\tExecution: ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromRegisterDomainRequest converts internal RegisterDomainRequest type to thrift\nfunc FromRegisterDomainRequest(t *types.RegisterDomainRequest) *shared.RegisterDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RegisterDomainRequest{\n\t\tName:                                   &t.Name,\n\t\tDescription:                            &t.Description,\n\t\tOwnerEmail:                             &t.OwnerEmail,\n\t\tWorkflowExecutionRetentionPeriodInDays: &t.WorkflowExecutionRetentionPeriodInDays,\n\t\tEmitMetric:                             t.EmitMetric,\n\t\tClusters:                               FromClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusterName:                      &t.ActiveClusterName,\n\t\tActiveClusters:                         FromActiveClusters(t.ActiveClusters),\n\t\tData:                                   t.Data,\n\t\tSecurityToken:                          &t.SecurityToken,\n\t\tIsGlobalDomain:                         &t.IsGlobalDomain,\n\t\tHistoryArchivalStatus:                  FromArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:                     &t.HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:               FromArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:                  &t.VisibilityArchivalURI,\n\t}\n}\n\n// ToRegisterDomainRequest converts thrift RegisterDomainRequest type to internal\nfunc ToRegisterDomainRequest(t *shared.RegisterDomainRequest) *types.RegisterDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RegisterDomainRequest{\n\t\tName:                                   t.GetName(),\n\t\tDescription:                            t.GetDescription(),\n\t\tOwnerEmail:                             t.GetOwnerEmail(),\n\t\tWorkflowExecutionRetentionPeriodInDays: t.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\tEmitMetric:                             t.EmitMetric,\n\t\tClusters:                               ToClusterReplicationConfigurationArray(t.Clusters),\n\t\tActiveClusterName:                      t.GetActiveClusterName(),\n\t\tActiveClusters:                         ToActiveClusters(t.ActiveClusters),\n\t\tData:                                   t.Data,\n\t\tSecurityToken:                          t.GetSecurityToken(),\n\t\tIsGlobalDomain:                         t.GetIsGlobalDomain(),\n\t\tHistoryArchivalStatus:                  ToArchivalStatus(t.HistoryArchivalStatus),\n\t\tHistoryArchivalURI:                     t.GetHistoryArchivalURI(),\n\t\tVisibilityArchivalStatus:               ToArchivalStatus(t.VisibilityArchivalStatus),\n\t\tVisibilityArchivalURI:                  t.GetVisibilityArchivalURI(),\n\t}\n}\n\n// FromRemoteSyncMatchedError converts internal RemoteSyncMatchedError type to thrift\nfunc FromRemoteSyncMatchedError(t *types.RemoteSyncMatchedError) *shared.RemoteSyncMatchedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RemoteSyncMatchedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToRemoteSyncMatchedError converts thrift RemoteSyncMatchedError type to internal\nfunc ToRemoteSyncMatchedError(t *shared.RemoteSyncMatchedError) *types.RemoteSyncMatchedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoteSyncMatchedError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromAdminRemoveTaskRequest converts internal RemoveTaskRequest type to thrift\nfunc FromAdminRemoveTaskRequest(t *types.RemoveTaskRequest) *shared.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RemoveTaskRequest{\n\t\tShardID:             &t.ShardID,\n\t\tType:                t.Type,\n\t\tTaskID:              &t.TaskID,\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t\tClusterName:         &t.ClusterName,\n\t}\n}\n\n// ToAdminRemoveTaskRequest converts thrift RemoveTaskRequest type to internal\nfunc ToAdminRemoveTaskRequest(t *shared.RemoveTaskRequest) *types.RemoveTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RemoveTaskRequest{\n\t\tShardID:             t.GetShardID(),\n\t\tType:                t.Type,\n\t\tTaskID:              t.GetTaskID(),\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t\tClusterName:         t.GetClusterName(),\n\t}\n}\n\n// FromRequestCancelActivityTaskDecisionAttributes converts internal RequestCancelActivityTaskDecisionAttributes type to thrift\nfunc FromRequestCancelActivityTaskDecisionAttributes(t *types.RequestCancelActivityTaskDecisionAttributes) *shared.RequestCancelActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelActivityTaskDecisionAttributes{\n\t\tActivityId: &t.ActivityID,\n\t}\n}\n\n// ToRequestCancelActivityTaskDecisionAttributes converts thrift RequestCancelActivityTaskDecisionAttributes type to internal\nfunc ToRequestCancelActivityTaskDecisionAttributes(t *shared.RequestCancelActivityTaskDecisionAttributes) *types.RequestCancelActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelActivityTaskDecisionAttributes{\n\t\tActivityID: t.GetActivityId(),\n\t}\n}\n\n// FromRequestCancelActivityTaskFailedEventAttributes converts internal RequestCancelActivityTaskFailedEventAttributes type to thrift\nfunc FromRequestCancelActivityTaskFailedEventAttributes(t *types.RequestCancelActivityTaskFailedEventAttributes) *shared.RequestCancelActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityId:                   &t.ActivityID,\n\t\tCause:                        &t.Cause,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToRequestCancelActivityTaskFailedEventAttributes converts thrift RequestCancelActivityTaskFailedEventAttributes type to internal\nfunc ToRequestCancelActivityTaskFailedEventAttributes(t *shared.RequestCancelActivityTaskFailedEventAttributes) *types.RequestCancelActivityTaskFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityID:                   t.GetActivityId(),\n\t\tCause:                        t.GetCause(),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromRequestCancelExternalWorkflowExecutionDecisionAttributes converts internal RequestCancelExternalWorkflowExecutionDecisionAttributes type to thrift\nfunc FromRequestCancelExternalWorkflowExecutionDecisionAttributes(t *types.RequestCancelExternalWorkflowExecutionDecisionAttributes) *shared.RequestCancelExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowId:        &t.WorkflowID,\n\t\tRunId:             &t.RunID,\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToRequestCancelExternalWorkflowExecutionDecisionAttributes converts thrift RequestCancelExternalWorkflowExecutionDecisionAttributes type to internal\nfunc ToRequestCancelExternalWorkflowExecutionDecisionAttributes(t *shared.RequestCancelExternalWorkflowExecutionDecisionAttributes) *types.RequestCancelExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowID:        t.GetWorkflowId(),\n\t\tRunID:             t.GetRunId(),\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromRequestCancelExternalWorkflowExecutionFailedEventAttributes converts internal RequestCancelExternalWorkflowExecutionFailedEventAttributes type to thrift\nfunc FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(t *types.RequestCancelExternalWorkflowExecutionFailedEventAttributes) *shared.RequestCancelExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        FromCancelExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tDomain:                       &t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventId:             &t.InitiatedEventID,\n\t\tControl:                      t.Control,\n\t}\n}\n\n// ToRequestCancelExternalWorkflowExecutionFailedEventAttributes converts thrift RequestCancelExternalWorkflowExecutionFailedEventAttributes type to internal\nfunc ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(t *shared.RequestCancelExternalWorkflowExecutionFailedEventAttributes) *types.RequestCancelExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        ToCancelExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tDomain:                       t.GetDomain(),\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventID:             t.GetInitiatedEventId(),\n\t\tControl:                      t.Control,\n\t}\n}\n\n// FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes converts internal RequestCancelExternalWorkflowExecutionInitiatedEventAttributes type to thrift\nfunc FromRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t *types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) *shared.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tDomain:                       &t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes converts thrift RequestCancelExternalWorkflowExecutionInitiatedEventAttributes type to internal\nfunc ToRequestCancelExternalWorkflowExecutionInitiatedEventAttributes(t *shared.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) *types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tDomain:                       t.GetDomain(),\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromRequestCancelWorkflowExecutionRequest converts internal RequestCancelWorkflowExecutionRequest type to thrift\nfunc FromRequestCancelWorkflowExecutionRequest(t *types.RequestCancelWorkflowExecutionRequest) *shared.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RequestCancelWorkflowExecutionRequest{\n\t\tDomain:              &t.Domain,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:            &t.Identity,\n\t\tRequestId:           &t.RequestID,\n\t\tCause:               &t.Cause,\n\t\tFirstExecutionRunID: &t.FirstExecutionRunID,\n\t}\n}\n\n// ToRequestCancelWorkflowExecutionRequest converts thrift RequestCancelWorkflowExecutionRequest type to internal\nfunc ToRequestCancelWorkflowExecutionRequest(t *shared.RequestCancelWorkflowExecutionRequest) *types.RequestCancelWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain:              t.GetDomain(),\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:            t.GetIdentity(),\n\t\tRequestID:           t.GetRequestId(),\n\t\tCause:               t.GetCause(),\n\t\tFirstExecutionRunID: t.GetFirstExecutionRunID(),\n\t}\n}\n\n// FromResetPointInfo converts internal ResetPointInfo type to thrift\nfunc FromResetPointInfo(t *types.ResetPointInfo) *shared.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetPointInfo{\n\t\tBinaryChecksum:           &t.BinaryChecksum,\n\t\tRunId:                    &t.RunID,\n\t\tFirstDecisionCompletedId: &t.FirstDecisionCompletedID,\n\t\tCreatedTimeNano:          t.CreatedTimeNano,\n\t\tExpiringTimeNano:         t.ExpiringTimeNano,\n\t\tResettable:               &t.Resettable,\n\t}\n}\n\n// ToResetPointInfo converts thrift ResetPointInfo type to internal\nfunc ToResetPointInfo(t *shared.ResetPointInfo) *types.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetPointInfo{\n\t\tBinaryChecksum:           t.GetBinaryChecksum(),\n\t\tRunID:                    t.GetRunId(),\n\t\tFirstDecisionCompletedID: t.GetFirstDecisionCompletedId(),\n\t\tCreatedTimeNano:          t.CreatedTimeNano,\n\t\tExpiringTimeNano:         t.ExpiringTimeNano,\n\t\tResettable:               t.GetResettable(),\n\t}\n}\n\n// FromResetPoints converts internal ResetPoints type to thrift\nfunc FromResetPoints(t *types.ResetPoints) *shared.ResetPoints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetPoints{\n\t\tPoints: FromResetPointInfoArray(t.Points),\n\t}\n}\n\n// ToResetPoints converts thrift ResetPoints type to internal\nfunc ToResetPoints(t *shared.ResetPoints) *types.ResetPoints {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetPoints{\n\t\tPoints: ToResetPointInfoArray(t.Points),\n\t}\n}\n\n// FromAdminResetQueueRequest converts internal ResetQueueRequest type to thrift\nfunc FromAdminResetQueueRequest(t *types.ResetQueueRequest) *shared.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetQueueRequest{\n\t\tShardID:     &t.ShardID,\n\t\tClusterName: &t.ClusterName,\n\t\tType:        t.Type,\n\t}\n}\n\n// ToAdminResetQueueRequest converts thrift ResetQueueRequest type to internal\nfunc ToAdminResetQueueRequest(t *shared.ResetQueueRequest) *types.ResetQueueRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetQueueRequest{\n\t\tShardID:     t.GetShardID(),\n\t\tClusterName: t.GetClusterName(),\n\t\tType:        t.Type,\n\t}\n}\n\n// FromResetStickyTaskListRequest converts internal ResetStickyTaskListRequest type to thrift\nfunc FromResetStickyTaskListRequest(t *types.ResetStickyTaskListRequest) *shared.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetStickyTaskListRequest{\n\t\tDomain:    &t.Domain,\n\t\tExecution: FromWorkflowExecution(t.Execution),\n\t}\n}\n\n// ToResetStickyTaskListRequest converts thrift ResetStickyTaskListRequest type to internal\nfunc ToResetStickyTaskListRequest(t *shared.ResetStickyTaskListRequest) *types.ResetStickyTaskListRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetStickyTaskListRequest{\n\t\tDomain:    t.GetDomain(),\n\t\tExecution: ToWorkflowExecution(t.Execution),\n\t}\n}\n\n// FromResetStickyTaskListResponse converts internal ResetStickyTaskListResponse type to thrift\nfunc FromResetStickyTaskListResponse(t *types.ResetStickyTaskListResponse) *shared.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetStickyTaskListResponse{}\n}\n\n// ToResetStickyTaskListResponse converts thrift ResetStickyTaskListResponse type to internal\nfunc ToResetStickyTaskListResponse(t *shared.ResetStickyTaskListResponse) *types.ResetStickyTaskListResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetStickyTaskListResponse{}\n}\n\n// FromResetWorkflowExecutionRequest converts internal ResetWorkflowExecutionRequest type to thrift\nfunc FromResetWorkflowExecutionRequest(t *types.ResetWorkflowExecutionRequest) *shared.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetWorkflowExecutionRequest{\n\t\tDomain:                &t.Domain,\n\t\tWorkflowExecution:     FromWorkflowExecution(t.WorkflowExecution),\n\t\tReason:                &t.Reason,\n\t\tDecisionFinishEventId: &t.DecisionFinishEventID,\n\t\tRequestId:             &t.RequestID,\n\t\tSkipSignalReapply:     &t.SkipSignalReapply,\n\t}\n}\n\n// ToResetWorkflowExecutionRequest converts thrift ResetWorkflowExecutionRequest type to internal\nfunc ToResetWorkflowExecutionRequest(t *shared.ResetWorkflowExecutionRequest) *types.ResetWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetWorkflowExecutionRequest{\n\t\tDomain:                t.GetDomain(),\n\t\tWorkflowExecution:     ToWorkflowExecution(t.WorkflowExecution),\n\t\tReason:                t.GetReason(),\n\t\tDecisionFinishEventID: t.GetDecisionFinishEventId(),\n\t\tRequestID:             t.GetRequestId(),\n\t\tSkipSignalReapply:     t.GetSkipSignalReapply(),\n\t}\n}\n\n// FromResetWorkflowExecutionResponse converts internal ResetWorkflowExecutionResponse type to thrift\nfunc FromResetWorkflowExecutionResponse(t *types.ResetWorkflowExecutionResponse) *shared.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ResetWorkflowExecutionResponse{\n\t\tRunId: &t.RunID,\n\t}\n}\n\n// ToResetWorkflowExecutionResponse converts thrift ResetWorkflowExecutionResponse type to internal\nfunc ToResetWorkflowExecutionResponse(t *shared.ResetWorkflowExecutionResponse) *types.ResetWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ResetWorkflowExecutionResponse{\n\t\tRunID: t.GetRunId(),\n\t}\n}\n\n// FromRespondActivityTaskCanceledByIDRequest converts internal RespondActivityTaskCanceledByIDRequest type to thrift\nfunc FromRespondActivityTaskCanceledByIDRequest(t *types.RespondActivityTaskCanceledByIDRequest) *shared.RespondActivityTaskCanceledByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     &t.Domain,\n\t\tWorkflowID: &t.WorkflowID,\n\t\tRunID:      &t.RunID,\n\t\tActivityID: &t.ActivityID,\n\t\tDetails:    t.Details,\n\t\tIdentity:   &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskCanceledByIDRequest converts thrift RespondActivityTaskCanceledByIDRequest type to internal\nfunc ToRespondActivityTaskCanceledByIDRequest(t *shared.RespondActivityTaskCanceledByIDRequest) *types.RespondActivityTaskCanceledByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     t.GetDomain(),\n\t\tWorkflowID: t.GetWorkflowID(),\n\t\tRunID:      t.GetRunID(),\n\t\tActivityID: t.GetActivityID(),\n\t\tDetails:    t.Details,\n\t\tIdentity:   t.GetIdentity(),\n\t}\n}\n\n// FromRespondActivityTaskCanceledRequest converts internal RespondActivityTaskCanceledRequest type to thrift\nfunc FromRespondActivityTaskCanceledRequest(t *types.RespondActivityTaskCanceledRequest) *shared.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   t.Details,\n\t\tIdentity:  &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskCanceledRequest converts thrift RespondActivityTaskCanceledRequest type to internal\nfunc ToRespondActivityTaskCanceledRequest(t *shared.RespondActivityTaskCanceledRequest) *types.RespondActivityTaskCanceledRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tDetails:   t.Details,\n\t\tIdentity:  t.GetIdentity(),\n\t}\n}\n\n// FromRespondActivityTaskCompletedByIDRequest converts internal RespondActivityTaskCompletedByIDRequest type to thrift\nfunc FromRespondActivityTaskCompletedByIDRequest(t *types.RespondActivityTaskCompletedByIDRequest) *shared.RespondActivityTaskCompletedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     &t.Domain,\n\t\tWorkflowID: &t.WorkflowID,\n\t\tRunID:      &t.RunID,\n\t\tActivityID: &t.ActivityID,\n\t\tResult:     t.Result,\n\t\tIdentity:   &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskCompletedByIDRequest converts thrift RespondActivityTaskCompletedByIDRequest type to internal\nfunc ToRespondActivityTaskCompletedByIDRequest(t *shared.RespondActivityTaskCompletedByIDRequest) *types.RespondActivityTaskCompletedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     t.GetDomain(),\n\t\tWorkflowID: t.GetWorkflowID(),\n\t\tRunID:      t.GetRunID(),\n\t\tActivityID: t.GetActivityID(),\n\t\tResult:     t.Result,\n\t\tIdentity:   t.GetIdentity(),\n\t}\n}\n\n// FromRespondActivityTaskCompletedRequest converts internal RespondActivityTaskCompletedRequest type to thrift\nfunc FromRespondActivityTaskCompletedRequest(t *types.RespondActivityTaskCompletedRequest) *shared.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tResult:    t.Result,\n\t\tIdentity:  &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskCompletedRequest converts thrift RespondActivityTaskCompletedRequest type to internal\nfunc ToRespondActivityTaskCompletedRequest(t *shared.RespondActivityTaskCompletedRequest) *types.RespondActivityTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tResult:    t.Result,\n\t\tIdentity:  t.GetIdentity(),\n\t}\n}\n\n// FromRespondActivityTaskFailedByIDRequest converts internal RespondActivityTaskFailedByIDRequest type to thrift\nfunc FromRespondActivityTaskFailedByIDRequest(t *types.RespondActivityTaskFailedByIDRequest) *shared.RespondActivityTaskFailedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     &t.Domain,\n\t\tWorkflowID: &t.WorkflowID,\n\t\tRunID:      &t.RunID,\n\t\tActivityID: &t.ActivityID,\n\t\tReason:     t.Reason,\n\t\tDetails:    t.Details,\n\t\tIdentity:   &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskFailedByIDRequest converts thrift RespondActivityTaskFailedByIDRequest type to internal\nfunc ToRespondActivityTaskFailedByIDRequest(t *shared.RespondActivityTaskFailedByIDRequest) *types.RespondActivityTaskFailedByIDRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     t.GetDomain(),\n\t\tWorkflowID: t.GetWorkflowID(),\n\t\tRunID:      t.GetRunID(),\n\t\tActivityID: t.GetActivityID(),\n\t\tReason:     t.Reason,\n\t\tDetails:    t.Details,\n\t\tIdentity:   t.GetIdentity(),\n\t}\n}\n\n// FromRespondActivityTaskFailedRequest converts internal RespondActivityTaskFailedRequest type to thrift\nfunc FromRespondActivityTaskFailedRequest(t *types.RespondActivityTaskFailedRequest) *shared.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondActivityTaskFailedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tReason:    t.Reason,\n\t\tDetails:   t.Details,\n\t\tIdentity:  &t.Identity,\n\t}\n}\n\n// ToRespondActivityTaskFailedRequest converts thrift RespondActivityTaskFailedRequest type to internal\nfunc ToRespondActivityTaskFailedRequest(t *shared.RespondActivityTaskFailedRequest) *types.RespondActivityTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: t.TaskToken,\n\t\tReason:    t.Reason,\n\t\tDetails:   t.Details,\n\t\tIdentity:  t.GetIdentity(),\n\t}\n}\n\n// FromRespondDecisionTaskCompletedRequest converts internal RespondDecisionTaskCompletedRequest type to thrift\nfunc FromRespondDecisionTaskCompletedRequest(t *types.RespondDecisionTaskCompletedRequest) *shared.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  t.TaskToken,\n\t\tDecisions:                  FromDecisionArray(t.Decisions),\n\t\tExecutionContext:           t.ExecutionContext,\n\t\tIdentity:                   &t.Identity,\n\t\tStickyAttributes:           FromStickyExecutionAttributes(t.StickyAttributes),\n\t\tReturnNewDecisionTask:      &t.ReturnNewDecisionTask,\n\t\tForceCreateNewDecisionTask: &t.ForceCreateNewDecisionTask,\n\t\tBinaryChecksum:             &t.BinaryChecksum,\n\t\tQueryResults:               FromWorkflowQueryResultMap(t.QueryResults),\n\t}\n}\n\n// ToRespondDecisionTaskCompletedRequest converts thrift RespondDecisionTaskCompletedRequest type to internal\nfunc ToRespondDecisionTaskCompletedRequest(t *shared.RespondDecisionTaskCompletedRequest) *types.RespondDecisionTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  t.TaskToken,\n\t\tDecisions:                  ToDecisionArray(t.Decisions),\n\t\tExecutionContext:           t.ExecutionContext,\n\t\tIdentity:                   t.GetIdentity(),\n\t\tStickyAttributes:           ToStickyExecutionAttributes(t.StickyAttributes),\n\t\tReturnNewDecisionTask:      t.GetReturnNewDecisionTask(),\n\t\tForceCreateNewDecisionTask: t.GetForceCreateNewDecisionTask(),\n\t\tBinaryChecksum:             t.GetBinaryChecksum(),\n\t\tQueryResults:               ToWorkflowQueryResultMap(t.QueryResults),\n\t}\n}\n\n// FromRespondDecisionTaskCompletedResponse converts internal RespondDecisionTaskCompletedResponse type to thrift\nfunc FromRespondDecisionTaskCompletedResponse(t *types.RespondDecisionTaskCompletedResponse) *shared.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondDecisionTaskCompletedResponse{\n\t\tDecisionTask:                FromPollForDecisionTaskResponse(t.DecisionTask),\n\t\tActivitiesToDispatchLocally: FromActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\n// ToRespondDecisionTaskCompletedResponse converts thrift RespondDecisionTaskCompletedResponse type to internal\nfunc ToRespondDecisionTaskCompletedResponse(t *shared.RespondDecisionTaskCompletedResponse) *types.RespondDecisionTaskCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskCompletedResponse{\n\t\tDecisionTask:                ToPollForDecisionTaskResponse(t.DecisionTask),\n\t\tActivitiesToDispatchLocally: ToActivityLocalDispatchInfoMap(t.ActivitiesToDispatchLocally),\n\t}\n}\n\n// FromRespondDecisionTaskFailedRequest converts internal RespondDecisionTaskFailedRequest type to thrift\nfunc FromRespondDecisionTaskFailedRequest(t *types.RespondDecisionTaskFailedRequest) *shared.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondDecisionTaskFailedRequest{\n\t\tTaskToken:      t.TaskToken,\n\t\tCause:          FromDecisionTaskFailedCause(t.Cause),\n\t\tDetails:        t.Details,\n\t\tIdentity:       &t.Identity,\n\t\tBinaryChecksum: &t.BinaryChecksum,\n\t}\n}\n\n// ToRespondDecisionTaskFailedRequest converts thrift RespondDecisionTaskFailedRequest type to internal\nfunc ToRespondDecisionTaskFailedRequest(t *shared.RespondDecisionTaskFailedRequest) *types.RespondDecisionTaskFailedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken:      t.TaskToken,\n\t\tCause:          ToDecisionTaskFailedCause(t.Cause),\n\t\tDetails:        t.Details,\n\t\tIdentity:       t.GetIdentity(),\n\t\tBinaryChecksum: t.GetBinaryChecksum(),\n\t}\n}\n\n// FromRespondQueryTaskCompletedRequest converts internal RespondQueryTaskCompletedRequest type to thrift\nfunc FromRespondQueryTaskCompletedRequest(t *types.RespondQueryTaskCompletedRequest) *shared.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondQueryTaskCompletedRequest{\n\t\tTaskToken:         t.TaskToken,\n\t\tCompletedType:     FromQueryTaskCompletedType(t.CompletedType),\n\t\tQueryResult:       t.QueryResult,\n\t\tErrorMessage:      &t.ErrorMessage,\n\t\tWorkerVersionInfo: FromWorkerVersionInfo(t.WorkerVersionInfo),\n\t}\n}\n\n// ToRespondQueryTaskCompletedRequest converts thrift RespondQueryTaskCompletedRequest type to internal\nfunc ToRespondQueryTaskCompletedRequest(t *shared.RespondQueryTaskCompletedRequest) *types.RespondQueryTaskCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondQueryTaskCompletedRequest{\n\t\tTaskToken:         t.TaskToken,\n\t\tCompletedType:     ToQueryTaskCompletedType(t.CompletedType),\n\t\tQueryResult:       t.QueryResult,\n\t\tErrorMessage:      t.GetErrorMessage(),\n\t\tWorkerVersionInfo: ToWorkerVersionInfo(t.WorkerVersionInfo),\n\t}\n}\n\n// FromRetryPolicy converts internal RetryPolicy type to thrift\nfunc FromRetryPolicy(t *types.RetryPolicy) *shared.RetryPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RetryPolicy{\n\t\tInitialIntervalInSeconds:    &t.InitialIntervalInSeconds,\n\t\tBackoffCoefficient:          &t.BackoffCoefficient,\n\t\tMaximumIntervalInSeconds:    &t.MaximumIntervalInSeconds,\n\t\tMaximumAttempts:             &t.MaximumAttempts,\n\t\tNonRetriableErrorReasons:    t.NonRetriableErrorReasons,\n\t\tExpirationIntervalInSeconds: &t.ExpirationIntervalInSeconds,\n\t}\n}\n\n// ToRetryPolicy converts thrift RetryPolicy type to internal\nfunc ToRetryPolicy(t *shared.RetryPolicy) *types.RetryPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    t.GetInitialIntervalInSeconds(),\n\t\tBackoffCoefficient:          t.GetBackoffCoefficient(),\n\t\tMaximumIntervalInSeconds:    t.GetMaximumIntervalInSeconds(),\n\t\tMaximumAttempts:             t.GetMaximumAttempts(),\n\t\tNonRetriableErrorReasons:    t.NonRetriableErrorReasons,\n\t\tExpirationIntervalInSeconds: t.GetExpirationIntervalInSeconds(),\n\t}\n}\n\n// FromRetryTaskV2Error converts internal RetryTaskV2Error type to thrift\nfunc FromRetryTaskV2Error(t *types.RetryTaskV2Error) *shared.RetryTaskV2Error {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RetryTaskV2Error{\n\t\tMessage:           t.Message,\n\t\tDomainId:          &t.DomainID,\n\t\tWorkflowId:        &t.WorkflowID,\n\t\tRunId:             &t.RunID,\n\t\tStartEventId:      t.StartEventID,\n\t\tStartEventVersion: t.StartEventVersion,\n\t\tEndEventId:        t.EndEventID,\n\t\tEndEventVersion:   t.EndEventVersion,\n\t}\n}\n\n// ToRetryTaskV2Error converts thrift RetryTaskV2Error type to internal\nfunc ToRetryTaskV2Error(t *shared.RetryTaskV2Error) *types.RetryTaskV2Error {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RetryTaskV2Error{\n\t\tMessage:           t.Message,\n\t\tDomainID:          t.GetDomainId(),\n\t\tWorkflowID:        t.GetWorkflowId(),\n\t\tRunID:             t.GetRunId(),\n\t\tStartEventID:      t.StartEventId,\n\t\tStartEventVersion: t.StartEventVersion,\n\t\tEndEventID:        t.EndEventId,\n\t\tEndEventVersion:   t.EndEventVersion,\n\t}\n}\n\n// FromRestartWorkflowExecutionRequest converts internal RestartWorkflowExecutionRequest type to thrift\nfunc FromRestartWorkflowExecutionRequest(t *types.RestartWorkflowExecutionRequest) *shared.RestartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RestartWorkflowExecutionRequest{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:          &t.Identity,\n\t}\n}\n\n// ToRestartWorkflowExecutionRequest converts thrift RestartWorkflowExecutionRequest type to internal\nfunc ToRestartWorkflowExecutionRequest(t *shared.RestartWorkflowExecutionRequest) *types.RestartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestartWorkflowExecutionRequest{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tIdentity:          t.GetIdentity(),\n\t}\n}\n\n// FromScheduleActivityTaskDecisionAttributes converts internal ScheduleActivityTaskDecisionAttributes type to thrift\nfunc FromScheduleActivityTaskDecisionAttributes(t *types.ScheduleActivityTaskDecisionAttributes) *shared.ScheduleActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityId:                    &t.ActivityID,\n\t\tActivityType:                  FromActivityType(t.ActivityType),\n\t\tDomain:                        &t.Domain,\n\t\tTaskList:                      FromTaskList(t.TaskList),\n\t\tInput:                         t.Input,\n\t\tScheduleToCloseTimeoutSeconds: t.ScheduleToCloseTimeoutSeconds,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       t.HeartbeatTimeoutSeconds,\n\t\tRetryPolicy:                   FromRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        FromHeader(t.Header),\n\t\tRequestLocalDispatch:          &t.RequestLocalDispatch,\n\t}\n}\n\n// ToScheduleActivityTaskDecisionAttributes converts thrift ScheduleActivityTaskDecisionAttributes type to internal\nfunc ToScheduleActivityTaskDecisionAttributes(t *shared.ScheduleActivityTaskDecisionAttributes) *types.ScheduleActivityTaskDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    t.GetActivityId(),\n\t\tActivityType:                  ToActivityType(t.ActivityType),\n\t\tDomain:                        t.GetDomain(),\n\t\tTaskList:                      ToTaskList(t.TaskList),\n\t\tInput:                         t.Input,\n\t\tScheduleToCloseTimeoutSeconds: t.ScheduleToCloseTimeoutSeconds,\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    t.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       t.HeartbeatTimeoutSeconds,\n\t\tRetryPolicy:                   ToRetryPolicy(t.RetryPolicy),\n\t\tHeader:                        ToHeader(t.Header),\n\t\tRequestLocalDispatch:          t.GetRequestLocalDispatch(),\n\t}\n}\n\n// FromSearchAttributes converts internal SearchAttributes type to thrift\nfunc FromSearchAttributes(t *types.SearchAttributes) *shared.SearchAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SearchAttributes{\n\t\tIndexedFields: t.IndexedFields,\n\t}\n}\n\n// ToSearchAttributes converts thrift SearchAttributes type to internal\nfunc ToSearchAttributes(t *shared.SearchAttributes) *types.SearchAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SearchAttributes{\n\t\tIndexedFields: t.IndexedFields,\n\t}\n}\n\n// FromServiceBusyError converts internal ServiceBusyError type to thrift\nfunc FromServiceBusyError(t *types.ServiceBusyError) *shared.ServiceBusyError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ServiceBusyError{\n\t\tMessage: t.Message,\n\t\tReason:  &t.Reason,\n\t}\n}\n\n// ToServiceBusyError converts thrift ServiceBusyError type to internal\nfunc ToServiceBusyError(t *shared.ServiceBusyError) *types.ServiceBusyError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ServiceBusyError{\n\t\tMessage: t.Message,\n\t\tReason:  t.GetReason(),\n\t}\n}\n\n// FromSignalExternalWorkflowExecutionDecisionAttributes converts internal SignalExternalWorkflowExecutionDecisionAttributes type to thrift\nfunc FromSignalExternalWorkflowExecutionDecisionAttributes(t *types.SignalExternalWorkflowExecutionDecisionAttributes) *shared.SignalExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            &t.Domain,\n\t\tExecution:         FromWorkflowExecution(t.Execution),\n\t\tSignalName:        &t.SignalName,\n\t\tInput:             t.Input,\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToSignalExternalWorkflowExecutionDecisionAttributes converts thrift SignalExternalWorkflowExecutionDecisionAttributes type to internal\nfunc ToSignalExternalWorkflowExecutionDecisionAttributes(t *shared.SignalExternalWorkflowExecutionDecisionAttributes) *types.SignalExternalWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            t.GetDomain(),\n\t\tExecution:         ToWorkflowExecution(t.Execution),\n\t\tSignalName:        t.GetSignalName(),\n\t\tInput:             t.Input,\n\t\tControl:           t.Control,\n\t\tChildWorkflowOnly: t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromSignalExternalWorkflowExecutionFailedCause converts internal SignalExternalWorkflowExecutionFailedCause type to thrift\nfunc FromSignalExternalWorkflowExecutionFailedCause(t *types.SignalExternalWorkflowExecutionFailedCause) *shared.SignalExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\tv := shared.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn &v\n\tcase types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\tv := shared.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToSignalExternalWorkflowExecutionFailedCause converts thrift SignalExternalWorkflowExecutionFailedCause type to internal\nfunc ToSignalExternalWorkflowExecutionFailedCause(t *shared.SignalExternalWorkflowExecutionFailedCause) *types.SignalExternalWorkflowExecutionFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution:\n\t\tv := types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn &v\n\tcase shared.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted:\n\t\tv := types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromSignalExternalWorkflowExecutionFailedEventAttributes converts internal SignalExternalWorkflowExecutionFailedEventAttributes type to thrift\nfunc FromSignalExternalWorkflowExecutionFailedEventAttributes(t *types.SignalExternalWorkflowExecutionFailedEventAttributes) *shared.SignalExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        FromSignalExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tDomain:                       &t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventId:             &t.InitiatedEventID,\n\t\tControl:                      t.Control,\n\t}\n}\n\n// ToSignalExternalWorkflowExecutionFailedEventAttributes converts thrift SignalExternalWorkflowExecutionFailedEventAttributes type to internal\nfunc ToSignalExternalWorkflowExecutionFailedEventAttributes(t *shared.SignalExternalWorkflowExecutionFailedEventAttributes) *types.SignalExternalWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        ToSignalExternalWorkflowExecutionFailedCause(t.Cause),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tDomain:                       t.GetDomain(),\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tInitiatedEventID:             t.GetInitiatedEventId(),\n\t\tControl:                      t.Control,\n\t}\n}\n\n// FromSignalExternalWorkflowExecutionInitiatedEventAttributes converts internal SignalExternalWorkflowExecutionInitiatedEventAttributes type to thrift\nfunc FromSignalExternalWorkflowExecutionInitiatedEventAttributes(t *types.SignalExternalWorkflowExecutionInitiatedEventAttributes) *shared.SignalExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tDomain:                       &t.Domain,\n\t\tWorkflowExecution:            FromWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:                   &t.SignalName,\n\t\tInput:                        t.Input,\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToSignalExternalWorkflowExecutionInitiatedEventAttributes converts thrift SignalExternalWorkflowExecutionInitiatedEventAttributes type to internal\nfunc ToSignalExternalWorkflowExecutionInitiatedEventAttributes(t *shared.SignalExternalWorkflowExecutionInitiatedEventAttributes) *types.SignalExternalWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tDomain:                       t.GetDomain(),\n\t\tWorkflowExecution:            ToWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:                   t.GetSignalName(),\n\t\tInput:                        t.Input,\n\t\tControl:                      t.Control,\n\t\tChildWorkflowOnly:            t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromSignalWithStartWorkflowExecutionRequest converts internal SignalWithStartWorkflowExecutionRequest type to thrift\nfunc FromSignalWithStartWorkflowExecutionRequest(t *types.SignalWithStartWorkflowExecutionRequest) *shared.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              &t.Domain,\n\t\tWorkflowId:                          &t.WorkflowID,\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            &t.Identity,\n\t\tRequestId:                           &t.RequestID,\n\t\tWorkflowIdReusePolicy:               FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tSignalName:                          &t.SignalName,\n\t\tSignalInput:                         t.SignalInput,\n\t\tControl:                             t.Control,\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tHeader:                              FromHeader(t.Header),\n\t\tFirstRunAtTimestamp:                 t.FirstRunAtTimestamp,\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// ToSignalWithStartWorkflowExecutionRequest converts thrift SignalWithStartWorkflowExecutionRequest type to internal\nfunc ToSignalWithStartWorkflowExecutionRequest(t *shared.SignalWithStartWorkflowExecutionRequest) *types.SignalWithStartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              t.GetDomain(),\n\t\tWorkflowID:                          t.GetWorkflowId(),\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            t.GetIdentity(),\n\t\tRequestID:                           t.GetRequestId(),\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tSignalName:                          t.GetSignalName(),\n\t\tSignalInput:                         t.SignalInput,\n\t\tControl:                             t.Control,\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tFirstRunAtTimestamp:                 t.FirstRunAtTimestamp,\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// FromSignalWorkflowExecutionRequest converts internal SignalWorkflowExecutionRequest type to thrift\nfunc FromSignalWorkflowExecutionRequest(t *types.SignalWorkflowExecutionRequest) *shared.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalWorkflowExecutionRequest{\n\t\tDomain:            &t.Domain,\n\t\tWorkflowExecution: FromWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:        &t.SignalName,\n\t\tInput:             t.Input,\n\t\tIdentity:          &t.Identity,\n\t\tRequestId:         &t.RequestID,\n\t\tControl:           t.Control,\n\t}\n}\n\n// ToSignalWorkflowExecutionRequest converts thrift SignalWorkflowExecutionRequest type to internal\nfunc ToSignalWorkflowExecutionRequest(t *shared.SignalWorkflowExecutionRequest) *types.SignalWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            t.GetDomain(),\n\t\tWorkflowExecution: ToWorkflowExecution(t.WorkflowExecution),\n\t\tSignalName:        t.GetSignalName(),\n\t\tInput:             t.Input,\n\t\tIdentity:          t.GetIdentity(),\n\t\tRequestID:         t.GetRequestId(),\n\t\tControl:           t.Control,\n\t}\n}\n\n// FromStartChildWorkflowExecutionDecisionAttributes converts internal StartChildWorkflowExecutionDecisionAttributes type to thrift\nfunc FromStartChildWorkflowExecutionDecisionAttributes(t *types.StartChildWorkflowExecutionDecisionAttributes) *shared.StartChildWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                              &t.Domain,\n\t\tWorkflowId:                          &t.WorkflowID,\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tParentClosePolicy:                   FromParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tWorkflowIdReusePolicy:               FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tHeader:                              FromHeader(t.Header),\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// ToStartChildWorkflowExecutionDecisionAttributes converts thrift StartChildWorkflowExecutionDecisionAttributes type to internal\nfunc ToStartChildWorkflowExecutionDecisionAttributes(t *shared.StartChildWorkflowExecutionDecisionAttributes) *types.StartChildWorkflowExecutionDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                              t.GetDomain(),\n\t\tWorkflowID:                          t.GetWorkflowId(),\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tParentClosePolicy:                   ToParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// FromStartChildWorkflowExecutionFailedEventAttributes converts internal StartChildWorkflowExecutionFailedEventAttributes type to thrift\nfunc FromStartChildWorkflowExecutionFailedEventAttributes(t *types.StartChildWorkflowExecutionFailedEventAttributes) *shared.StartChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       &t.Domain,\n\t\tWorkflowId:                   &t.WorkflowID,\n\t\tWorkflowType:                 FromWorkflowType(t.WorkflowType),\n\t\tCause:                        FromChildWorkflowExecutionFailedCause(t.Cause),\n\t\tControl:                      t.Control,\n\t\tInitiatedEventId:             &t.InitiatedEventID,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToStartChildWorkflowExecutionFailedEventAttributes converts thrift StartChildWorkflowExecutionFailedEventAttributes type to internal\nfunc ToStartChildWorkflowExecutionFailedEventAttributes(t *shared.StartChildWorkflowExecutionFailedEventAttributes) *types.StartChildWorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       t.GetDomain(),\n\t\tWorkflowID:                   t.GetWorkflowId(),\n\t\tWorkflowType:                 ToWorkflowType(t.WorkflowType),\n\t\tCause:                        ToChildWorkflowExecutionFailedCause(t.Cause),\n\t\tControl:                      t.Control,\n\t\tInitiatedEventID:             t.GetInitiatedEventId(),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromStartChildWorkflowExecutionInitiatedEventAttributes converts internal StartChildWorkflowExecutionInitiatedEventAttributes type to thrift\nfunc FromStartChildWorkflowExecutionInitiatedEventAttributes(t *types.StartChildWorkflowExecutionInitiatedEventAttributes) *shared.StartChildWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              &t.Domain,\n\t\tWorkflowId:                          &t.WorkflowID,\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tParentClosePolicy:                   FromParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tDecisionTaskCompletedEventId:        &t.DecisionTaskCompletedEventID,\n\t\tWorkflowIdReusePolicy:               FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tHeader:                              FromHeader(t.Header),\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tDelayStartSeconds:                   t.DelayStartSeconds,\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tFirstRunAtTimestamp:                 t.FirstRunAtTimestamp,\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// ToStartChildWorkflowExecutionInitiatedEventAttributes converts thrift StartChildWorkflowExecutionInitiatedEventAttributes type to internal\nfunc ToStartChildWorkflowExecutionInitiatedEventAttributes(t *shared.StartChildWorkflowExecutionInitiatedEventAttributes) *types.StartChildWorkflowExecutionInitiatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              t.GetDomain(),\n\t\tWorkflowID:                          t.GetWorkflowId(),\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tParentClosePolicy:                   ToParentClosePolicy(t.ParentClosePolicy),\n\t\tControl:                             t.Control,\n\t\tDecisionTaskCompletedEventID:        t.GetDecisionTaskCompletedEventId(),\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tDelayStartSeconds:                   t.DelayStartSeconds,\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tFirstRunAtTimestamp:                 t.FirstRunAtTimestamp,\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// FromStartTimeFilter converts internal StartTimeFilter type to thrift\nfunc FromStartTimeFilter(t *types.StartTimeFilter) *shared.StartTimeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartTimeFilter{\n\t\tEarliestTime: t.EarliestTime,\n\t\tLatestTime:   t.LatestTime,\n\t}\n}\n\n// ToStartTimeFilter converts thrift StartTimeFilter type to internal\nfunc ToStartTimeFilter(t *shared.StartTimeFilter) *types.StartTimeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartTimeFilter{\n\t\tEarliestTime: t.EarliestTime,\n\t\tLatestTime:   t.LatestTime,\n\t}\n}\n\n// FromStartTimerDecisionAttributes converts internal StartTimerDecisionAttributes type to thrift\nfunc FromStartTimerDecisionAttributes(t *types.StartTimerDecisionAttributes) *shared.StartTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartTimerDecisionAttributes{\n\t\tTimerId:                   &t.TimerID,\n\t\tStartToFireTimeoutSeconds: t.StartToFireTimeoutSeconds,\n\t}\n}\n\n// ToStartTimerDecisionAttributes converts thrift StartTimerDecisionAttributes type to internal\nfunc ToStartTimerDecisionAttributes(t *shared.StartTimerDecisionAttributes) *types.StartTimerDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartTimerDecisionAttributes{\n\t\tTimerID:                   t.GetTimerId(),\n\t\tStartToFireTimeoutSeconds: t.StartToFireTimeoutSeconds,\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionAsyncRequest(t *types.SignalWithStartWorkflowExecutionAsyncRequest) *shared.SignalWithStartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tRequest: FromSignalWithStartWorkflowExecutionRequest(t.SignalWithStartWorkflowExecutionRequest),\n\t}\n}\n\nfunc ToSignalWithStartWorkflowExecutionAsyncRequest(t *shared.SignalWithStartWorkflowExecutionAsyncRequest) *types.SignalWithStartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tSignalWithStartWorkflowExecutionRequest: ToSignalWithStartWorkflowExecutionRequest(t.Request),\n\t}\n}\n\nfunc FromSignalWithStartWorkflowExecutionAsyncResponse(t *types.SignalWithStartWorkflowExecutionAsyncResponse) *shared.SignalWithStartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SignalWithStartWorkflowExecutionAsyncResponse{}\n}\n\nfunc ToSignalWithStartWorkflowExecutionAsyncResponse(t *shared.SignalWithStartWorkflowExecutionAsyncResponse) *types.SignalWithStartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionAsyncResponse{}\n}\n\nfunc FromStartWorkflowExecutionAsyncRequest(t *types.StartWorkflowExecutionAsyncRequest) *shared.StartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartWorkflowExecutionAsyncRequest{\n\t\tRequest: FromStartWorkflowExecutionRequest(t.StartWorkflowExecutionRequest),\n\t}\n}\n\nfunc ToStartWorkflowExecutionAsyncRequest(t *shared.StartWorkflowExecutionAsyncRequest) *types.StartWorkflowExecutionAsyncRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionAsyncRequest{\n\t\tStartWorkflowExecutionRequest: ToStartWorkflowExecutionRequest(t.Request),\n\t}\n}\n\nfunc FromStartWorkflowExecutionAsyncResponse(t *types.StartWorkflowExecutionAsyncResponse) *shared.StartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartWorkflowExecutionAsyncResponse{}\n}\n\nfunc ToStartWorkflowExecutionAsyncResponse(t *shared.StartWorkflowExecutionAsyncResponse) *types.StartWorkflowExecutionAsyncResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionAsyncResponse{}\n}\n\n// FromStartWorkflowExecutionRequest converts internal StartWorkflowExecutionRequest type to thrift\nfunc FromStartWorkflowExecutionRequest(t *types.StartWorkflowExecutionRequest) *shared.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tthriftPolicy := FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy)\n\treturn &shared.StartWorkflowExecutionRequest{\n\t\tDomain:                              &t.Domain,\n\t\tWorkflowId:                          &t.WorkflowID,\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            &t.Identity,\n\t\tRequestId:                           &t.RequestID,\n\t\tWorkflowIdReusePolicy:               FromWorkflowIDReusePolicy(t.WorkflowIDReusePolicy),\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tHeader:                              FromHeader(t.Header),\n\t\tDelayStartSeconds:                   t.DelayStartSeconds,\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tFirstRunAtTimestamp:                 t.FirstRunAtTimeStamp,\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        thriftPolicy,\n\t}\n}\n\n// ToStartWorkflowExecutionRequest converts thrift StartWorkflowExecutionRequest type to internal\nfunc ToStartWorkflowExecutionRequest(t *shared.StartWorkflowExecutionRequest) *types.StartWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              t.GetDomain(),\n\t\tWorkflowID:                          t.GetWorkflowId(),\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            t.GetIdentity(),\n\t\tRequestID:                           t.GetRequestId(),\n\t\tWorkflowIDReusePolicy:               ToWorkflowIDReusePolicy(t.WorkflowIdReusePolicy),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tDelayStartSeconds:                   t.DelayStartSeconds,\n\t\tJitterStartSeconds:                  t.JitterStartSeconds,\n\t\tFirstRunAtTimeStamp:                 t.FirstRunAtTimestamp,\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t}\n}\n\n// FromRestartWorkflowExecutionResponse converts internal RestartWorkflowExecutionResponse type to thrift\nfunc FromRestartWorkflowExecutionResponse(t *types.RestartWorkflowExecutionResponse) *shared.RestartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RestartWorkflowExecutionResponse{\n\t\tRunId: &t.RunID,\n\t}\n}\n\n// FromStartWorkflowExecutionResponse converts internal StartWorkflowExecutionResponse type to thrift\nfunc FromStartWorkflowExecutionResponse(t *types.StartWorkflowExecutionResponse) *shared.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StartWorkflowExecutionResponse{\n\t\tRunId: &t.RunID,\n\t}\n}\n\n// ToStartWorkflowExecutionResponse converts thrift StartWorkflowExecutionResponse type to internal\nfunc ToStartWorkflowExecutionResponse(t *shared.StartWorkflowExecutionResponse) *types.StartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: t.GetRunId(),\n\t}\n}\n\n// FromStickyExecutionAttributes converts internal StickyExecutionAttributes type to thrift\nfunc FromStickyExecutionAttributes(t *types.StickyExecutionAttributes) *shared.StickyExecutionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StickyExecutionAttributes{\n\t\tWorkerTaskList:                FromTaskList(t.WorkerTaskList),\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t}\n}\n\n// ToStickyExecutionAttributes converts thrift StickyExecutionAttributes type to internal\nfunc ToStickyExecutionAttributes(t *shared.StickyExecutionAttributes) *types.StickyExecutionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StickyExecutionAttributes{\n\t\tWorkerTaskList:                ToTaskList(t.WorkerTaskList),\n\t\tScheduleToStartTimeoutSeconds: t.ScheduleToStartTimeoutSeconds,\n\t}\n}\n\n// FromSupportedClientVersions converts internal SupportedClientVersions type to thrift\nfunc FromSupportedClientVersions(t *types.SupportedClientVersions) *shared.SupportedClientVersions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.SupportedClientVersions{\n\t\tGoSdk:   &t.GoSdk,\n\t\tJavaSdk: &t.JavaSdk,\n\t}\n}\n\n// ToSupportedClientVersions converts thrift SupportedClientVersions type to internal\nfunc ToSupportedClientVersions(t *shared.SupportedClientVersions) *types.SupportedClientVersions {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.SupportedClientVersions{\n\t\tGoSdk:   t.GetGoSdk(),\n\t\tJavaSdk: t.GetJavaSdk(),\n\t}\n}\n\n// FromTaskIDBlock converts internal TaskIDBlock type to thrift\nfunc FromTaskIDBlock(t *types.TaskIDBlock) *shared.TaskIDBlock {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskIDBlock{\n\t\tStartID: &t.StartID,\n\t\tEndID:   &t.EndID,\n\t}\n}\n\n// ToTaskIDBlock converts thrift TaskIDBlock type to internal\nfunc ToTaskIDBlock(t *shared.TaskIDBlock) *types.TaskIDBlock {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskIDBlock{\n\t\tStartID: t.GetStartID(),\n\t\tEndID:   t.GetEndID(),\n\t}\n}\n\n// FromTaskList converts internal TaskList type to thrift\nfunc FromTaskList(t *types.TaskList) *shared.TaskList {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskList{\n\t\tName: &t.Name,\n\t\tKind: FromTaskListKind(t.Kind),\n\t}\n}\n\nfunc MigrateTaskList(name string, t *shared.TaskList) *types.TaskList {\n\tif t == nil && name != \"\" {\n\t\treturn &types.TaskList{Name: name, Kind: types.TaskListKindNormal.Ptr()}\n\t}\n\treturn ToTaskList(t)\n}\n\n// ToTaskList converts thrift TaskList type to internal\nfunc ToTaskList(t *shared.TaskList) *types.TaskList {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskList{\n\t\tName: t.GetName(),\n\t\tKind: ToTaskListKind(t.Kind),\n\t}\n}\n\n// FromTaskListKind converts internal TaskListKind type to thrift\nfunc FromTaskListKind(t *types.TaskListKind) *shared.TaskListKind {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.TaskListKindNormal:\n\t\treturn shared.TaskListKindNormal.Ptr()\n\tcase types.TaskListKindSticky:\n\t\treturn shared.TaskListKindSticky.Ptr()\n\tcase types.TaskListKindEphemeral:\n\t\treturn shared.TaskListKindEphemeral.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToTaskListKind converts thrift TaskListKind type to internal\nfunc ToTaskListKind(t *shared.TaskListKind) *types.TaskListKind {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.TaskListKindNormal:\n\t\treturn types.TaskListKindNormal.Ptr()\n\tcase shared.TaskListKindSticky:\n\t\treturn types.TaskListKindSticky.Ptr()\n\tcase shared.TaskListKindEphemeral:\n\t\treturn types.TaskListKindEphemeral.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromTaskListMetadata converts internal TaskListMetadata type to thrift\nfunc FromTaskListMetadata(t *types.TaskListMetadata) *shared.TaskListMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskListMetadata{\n\t\tMaxTasksPerSecond: t.MaxTasksPerSecond,\n\t}\n}\n\n// ToTaskListMetadata converts thrift TaskListMetadata type to internal\nfunc ToTaskListMetadata(t *shared.TaskListMetadata) *types.TaskListMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListMetadata{\n\t\tMaxTasksPerSecond: t.MaxTasksPerSecond,\n\t}\n}\n\n// FromTaskListPartitionMetadata converts internal TaskListPartitionMetadata type to thrift\nfunc FromTaskListPartitionMetadata(t *types.TaskListPartitionMetadata) *shared.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskListPartitionMetadata{\n\t\tKey:           &t.Key,\n\t\tOwnerHostName: &t.OwnerHostName,\n\t}\n}\n\n// ToTaskListPartitionMetadata converts thrift TaskListPartitionMetadata type to internal\nfunc ToTaskListPartitionMetadata(t *shared.TaskListPartitionMetadata) *types.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListPartitionMetadata{\n\t\tKey:           t.GetKey(),\n\t\tOwnerHostName: t.GetOwnerHostName(),\n\t}\n}\n\n// FromTaskListStatus converts internal TaskListStatus type to thrift\nfunc FromTaskListStatus(t *types.TaskListStatus) *shared.TaskListStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskListStatus{\n\t\tBacklogCountHint:      &t.BacklogCountHint,\n\t\tReadLevel:             &t.ReadLevel,\n\t\tAckLevel:              &t.AckLevel,\n\t\tRatePerSecond:         &t.RatePerSecond,\n\t\tTaskIDBlock:           FromTaskIDBlock(t.TaskIDBlock),\n\t\tIsolationGroupMetrics: FromIsolationGroupMetricsMap(t.IsolationGroupMetrics),\n\t\tNewTasksPerSecond:     &t.NewTasksPerSecond,\n\t\tEmpty:                 &t.Empty,\n\t}\n}\n\n// ToTaskListStatus converts thrift TaskListStatus type to internal\nfunc ToTaskListStatus(t *shared.TaskListStatus) *types.TaskListStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskListStatus{\n\t\tBacklogCountHint:      t.GetBacklogCountHint(),\n\t\tReadLevel:             t.GetReadLevel(),\n\t\tAckLevel:              t.GetAckLevel(),\n\t\tRatePerSecond:         t.GetRatePerSecond(),\n\t\tTaskIDBlock:           ToTaskIDBlock(t.TaskIDBlock),\n\t\tIsolationGroupMetrics: ToIsolationGroupMetricsMap(t.GetIsolationGroupMetrics()),\n\t\tNewTasksPerSecond:     t.GetNewTasksPerSecond(),\n\t\tEmpty:                 t.GetEmpty(),\n\t}\n}\n\nfunc FromIsolationGroupMetrics(t *types.IsolationGroupMetrics) *shared.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.IsolationGroupMetrics{\n\t\tNewTasksPerSecond: &t.NewTasksPerSecond,\n\t\tPollerCount:       &t.PollerCount,\n\t}\n}\n\nfunc ToIsolationGroupMetrics(t *shared.IsolationGroupMetrics) *types.IsolationGroupMetrics {\n\treturn &types.IsolationGroupMetrics{\n\t\tNewTasksPerSecond: t.GetNewTasksPerSecond(),\n\t\tPollerCount:       t.GetPollerCount(),\n\t}\n}\n\n// FromTaskListType converts internal TaskListType type to thrift\nfunc FromTaskListType(t *types.TaskListType) *shared.TaskListType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.TaskListTypeDecision:\n\t\tv := shared.TaskListTypeDecision\n\t\treturn &v\n\tcase types.TaskListTypeActivity:\n\t\tv := shared.TaskListTypeActivity\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToTaskListType converts thrift TaskListType type to internal\nfunc ToTaskListType(t *shared.TaskListType) *types.TaskListType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.TaskListTypeDecision:\n\t\tv := types.TaskListTypeDecision\n\t\treturn &v\n\tcase shared.TaskListTypeActivity:\n\t\tv := types.TaskListTypeActivity\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToRestartWorkflowExecutionResponse converts thrift RestartWorkflowExecutionResponse type to internal\nfunc ToRestartWorkflowExecutionResponse(t *shared.RestartWorkflowExecutionResponse) *types.RestartWorkflowExecutionResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RestartWorkflowExecutionResponse{\n\t\tRunID: t.GetRunId(),\n\t}\n}\n\n// FromTerminateWorkflowExecutionRequest converts internal TerminateWorkflowExecutionRequest type to thrift\nfunc FromTerminateWorkflowExecutionRequest(t *types.TerminateWorkflowExecutionRequest) *shared.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TerminateWorkflowExecutionRequest{\n\t\tDomain:              &t.Domain,\n\t\tWorkflowExecution:   FromWorkflowExecution(t.WorkflowExecution),\n\t\tReason:              &t.Reason,\n\t\tDetails:             t.Details,\n\t\tIdentity:            &t.Identity,\n\t\tFirstExecutionRunID: &t.FirstExecutionRunID,\n\t}\n}\n\n// ToTerminateWorkflowExecutionRequest converts thrift TerminateWorkflowExecutionRequest type to internal\nfunc ToTerminateWorkflowExecutionRequest(t *shared.TerminateWorkflowExecutionRequest) *types.TerminateWorkflowExecutionRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:              t.GetDomain(),\n\t\tWorkflowExecution:   ToWorkflowExecution(t.WorkflowExecution),\n\t\tReason:              t.GetReason(),\n\t\tDetails:             t.Details,\n\t\tIdentity:            t.GetIdentity(),\n\t\tFirstExecutionRunID: t.GetFirstExecutionRunID(),\n\t}\n}\n\n// FromTimeoutType converts internal TimeoutType type to thrift\nfunc FromTimeoutType(t *types.TimeoutType) *shared.TimeoutType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.TimeoutTypeStartToClose:\n\t\tv := shared.TimeoutTypeStartToClose\n\t\treturn &v\n\tcase types.TimeoutTypeScheduleToStart:\n\t\tv := shared.TimeoutTypeScheduleToStart\n\t\treturn &v\n\tcase types.TimeoutTypeScheduleToClose:\n\t\tv := shared.TimeoutTypeScheduleToClose\n\t\treturn &v\n\tcase types.TimeoutTypeHeartbeat:\n\t\tv := shared.TimeoutTypeHeartbeat\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToTimeoutType converts thrift TimeoutType type to internal\nfunc ToTimeoutType(t *shared.TimeoutType) *types.TimeoutType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.TimeoutTypeStartToClose:\n\t\tv := types.TimeoutTypeStartToClose\n\t\treturn &v\n\tcase shared.TimeoutTypeScheduleToStart:\n\t\tv := types.TimeoutTypeScheduleToStart\n\t\treturn &v\n\tcase shared.TimeoutTypeScheduleToClose:\n\t\tv := types.TimeoutTypeScheduleToClose\n\t\treturn &v\n\tcase shared.TimeoutTypeHeartbeat:\n\t\tv := types.TimeoutTypeHeartbeat\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromTimerCanceledEventAttributes converts internal TimerCanceledEventAttributes type to thrift\nfunc FromTimerCanceledEventAttributes(t *types.TimerCanceledEventAttributes) *shared.TimerCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TimerCanceledEventAttributes{\n\t\tTimerId:                      &t.TimerID,\n\t\tStartedEventId:               &t.StartedEventID,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tIdentity:                     &t.Identity,\n\t}\n}\n\n// ToTimerCanceledEventAttributes converts thrift TimerCanceledEventAttributes type to internal\nfunc ToTimerCanceledEventAttributes(t *shared.TimerCanceledEventAttributes) *types.TimerCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerCanceledEventAttributes{\n\t\tTimerID:                      t.GetTimerId(),\n\t\tStartedEventID:               t.GetStartedEventId(),\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tIdentity:                     t.GetIdentity(),\n\t}\n}\n\n// FromTimerFiredEventAttributes converts internal TimerFiredEventAttributes type to thrift\nfunc FromTimerFiredEventAttributes(t *types.TimerFiredEventAttributes) *shared.TimerFiredEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TimerFiredEventAttributes{\n\t\tTimerId:        &t.TimerID,\n\t\tStartedEventId: &t.StartedEventID,\n\t}\n}\n\n// ToTimerFiredEventAttributes converts thrift TimerFiredEventAttributes type to internal\nfunc ToTimerFiredEventAttributes(t *shared.TimerFiredEventAttributes) *types.TimerFiredEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerFiredEventAttributes{\n\t\tTimerID:        t.GetTimerId(),\n\t\tStartedEventID: t.GetStartedEventId(),\n\t}\n}\n\n// FromTimerStartedEventAttributes converts internal TimerStartedEventAttributes type to thrift\nfunc FromTimerStartedEventAttributes(t *types.TimerStartedEventAttributes) *shared.TimerStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TimerStartedEventAttributes{\n\t\tTimerId:                      &t.TimerID,\n\t\tStartToFireTimeoutSeconds:    t.StartToFireTimeoutSeconds,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToTimerStartedEventAttributes converts thrift TimerStartedEventAttributes type to internal\nfunc ToTimerStartedEventAttributes(t *shared.TimerStartedEventAttributes) *types.TimerStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TimerStartedEventAttributes{\n\t\tTimerID:                      t.GetTimerId(),\n\t\tStartToFireTimeoutSeconds:    t.StartToFireTimeoutSeconds,\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromTransientDecisionInfo converts internal TransientDecisionInfo type to thrift\nfunc FromTransientDecisionInfo(t *types.TransientDecisionInfo) *shared.TransientDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TransientDecisionInfo{\n\t\tScheduledEvent: FromHistoryEvent(t.ScheduledEvent),\n\t\tStartedEvent:   FromHistoryEvent(t.StartedEvent),\n\t}\n}\n\n// ToTransientDecisionInfo converts thrift TransientDecisionInfo type to internal\nfunc ToTransientDecisionInfo(t *shared.TransientDecisionInfo) *types.TransientDecisionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TransientDecisionInfo{\n\t\tScheduledEvent: ToHistoryEvent(t.ScheduledEvent),\n\t\tStartedEvent:   ToHistoryEvent(t.StartedEvent),\n\t}\n}\n\n// FromUpdateDomainRequest converts internal UpdateDomainRequest type to thrift\nfunc FromUpdateDomainRequest(t *types.UpdateDomainRequest) *shared.UpdateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := shared.UpdateDomainRequest{\n\t\tName:                     &t.Name,\n\t\tSecurityToken:            &t.SecurityToken,\n\t\tDeleteBadBinary:          t.DeleteBadBinary,\n\t\tFailoverTimeoutInSeconds: t.FailoverTimeoutInSeconds,\n\t}\n\tif t.Description != nil || t.OwnerEmail != nil || t.Data != nil {\n\t\trequest.UpdatedInfo = &shared.UpdateDomainInfo{\n\t\t\tDescription: t.Description,\n\t\t\tOwnerEmail:  t.OwnerEmail,\n\t\t\tData:        t.Data,\n\t\t}\n\t}\n\tif t.WorkflowExecutionRetentionPeriodInDays != nil ||\n\t\tt.EmitMetric != nil ||\n\t\tt.BadBinaries != nil ||\n\t\tt.HistoryArchivalStatus != nil ||\n\t\tt.HistoryArchivalURI != nil ||\n\t\tt.VisibilityArchivalStatus != nil ||\n\t\tt.VisibilityArchivalURI != nil {\n\t\trequest.Configuration = &shared.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: t.WorkflowExecutionRetentionPeriodInDays,\n\t\t\tEmitMetric:                             t.EmitMetric,\n\t\t\tBadBinaries:                            FromBadBinaries(t.BadBinaries),\n\t\t\tHistoryArchivalStatus:                  FromArchivalStatus(t.HistoryArchivalStatus),\n\t\t\tHistoryArchivalURI:                     t.HistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus:               FromArchivalStatus(t.VisibilityArchivalStatus),\n\t\t\tVisibilityArchivalURI:                  t.VisibilityArchivalURI,\n\t\t}\n\t}\n\tif t.ActiveClusterName != nil || t.Clusters != nil || t.ActiveClusters != nil {\n\t\trequest.ReplicationConfiguration = &shared.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: t.ActiveClusterName,\n\t\t\tClusters:          FromClusterReplicationConfigurationArray(t.Clusters),\n\t\t\tActiveClusters:    FromActiveClusters(t.ActiveClusters),\n\t\t}\n\t}\n\treturn &request\n}\n\n// ToUpdateDomainRequest converts thrift UpdateDomainRequest type to internal\nfunc ToUpdateDomainRequest(t *shared.UpdateDomainRequest) *types.UpdateDomainRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\trequest := types.UpdateDomainRequest{\n\t\tName:                     t.GetName(),\n\t\tSecurityToken:            t.GetSecurityToken(),\n\t\tDeleteBadBinary:          t.DeleteBadBinary,\n\t\tFailoverTimeoutInSeconds: t.FailoverTimeoutInSeconds,\n\t}\n\tif t.UpdatedInfo != nil {\n\t\trequest.Description = t.UpdatedInfo.Description\n\t\trequest.OwnerEmail = t.UpdatedInfo.OwnerEmail\n\t\trequest.Data = t.UpdatedInfo.Data\n\t}\n\tif t.Configuration != nil {\n\t\trequest.WorkflowExecutionRetentionPeriodInDays = t.Configuration.WorkflowExecutionRetentionPeriodInDays\n\t\trequest.EmitMetric = t.Configuration.EmitMetric\n\t\trequest.BadBinaries = ToBadBinaries(t.Configuration.BadBinaries)\n\t\trequest.HistoryArchivalStatus = ToArchivalStatus(t.Configuration.HistoryArchivalStatus)\n\t\trequest.HistoryArchivalURI = t.Configuration.HistoryArchivalURI\n\t\trequest.VisibilityArchivalStatus = ToArchivalStatus(t.Configuration.VisibilityArchivalStatus)\n\t\trequest.VisibilityArchivalURI = t.Configuration.VisibilityArchivalURI\n\t}\n\tif t.ReplicationConfiguration != nil {\n\t\trequest.ActiveClusterName = t.ReplicationConfiguration.ActiveClusterName\n\t\trequest.Clusters = ToClusterReplicationConfigurationArray(t.ReplicationConfiguration.Clusters)\n\t\trequest.ActiveClusters = ToActiveClusters(t.ReplicationConfiguration.ActiveClusters)\n\t}\n\treturn &request\n}\n\n// FromUpdateDomainResponse converts internal UpdateDomainResponse type to thrift\nfunc FromUpdateDomainResponse(t *types.UpdateDomainResponse) *shared.UpdateDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.UpdateDomainResponse{\n\t\tDomainInfo:               FromDomainInfo(t.DomainInfo),\n\t\tConfiguration:            FromDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: FromDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          &t.FailoverVersion,\n\t\tIsGlobalDomain:           &t.IsGlobalDomain,\n\t}\n}\n\n// ToUpdateDomainResponse converts thrift UpdateDomainResponse type to internal\nfunc ToUpdateDomainResponse(t *shared.UpdateDomainResponse) *types.UpdateDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpdateDomainResponse{\n\t\tDomainInfo:               ToDomainInfo(t.DomainInfo),\n\t\tConfiguration:            ToDomainConfiguration(t.Configuration),\n\t\tReplicationConfiguration: ToDomainReplicationConfiguration(t.ReplicationConfiguration),\n\t\tFailoverVersion:          t.GetFailoverVersion(),\n\t\tIsGlobalDomain:           t.GetIsGlobalDomain(),\n\t}\n}\n\n// FromUpsertWorkflowSearchAttributesDecisionAttributes converts internal UpsertWorkflowSearchAttributesDecisionAttributes type to thrift\nfunc FromUpsertWorkflowSearchAttributesDecisionAttributes(t *types.UpsertWorkflowSearchAttributesDecisionAttributes) *shared.UpsertWorkflowSearchAttributesDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tSearchAttributes: FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\n// ToUpsertWorkflowSearchAttributesDecisionAttributes converts thrift UpsertWorkflowSearchAttributesDecisionAttributes type to internal\nfunc ToUpsertWorkflowSearchAttributesDecisionAttributes(t *shared.UpsertWorkflowSearchAttributesDecisionAttributes) *types.UpsertWorkflowSearchAttributesDecisionAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tSearchAttributes: ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\n// FromUpsertWorkflowSearchAttributesEventAttributes converts internal UpsertWorkflowSearchAttributesEventAttributes type to thrift\nfunc FromUpsertWorkflowSearchAttributesEventAttributes(t *types.UpsertWorkflowSearchAttributesEventAttributes) *shared.UpsertWorkflowSearchAttributesEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t}\n}\n\n// ToUpsertWorkflowSearchAttributesEventAttributes converts thrift UpsertWorkflowSearchAttributesEventAttributes type to internal\nfunc ToUpsertWorkflowSearchAttributesEventAttributes(t *shared.UpsertWorkflowSearchAttributesEventAttributes) *types.UpsertWorkflowSearchAttributesEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tSearchAttributes:             ToSearchAttributes(t.SearchAttributes),\n\t}\n}\n\n// FromVersionHistories converts internal VersionHistories type to thrift\nfunc FromVersionHistories(t *types.VersionHistories) *shared.VersionHistories {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.VersionHistories{\n\t\tCurrentVersionHistoryIndex: &t.CurrentVersionHistoryIndex,\n\t\tHistories:                  FromVersionHistoryArray(t.Histories),\n\t}\n}\n\n// ToVersionHistories converts thrift VersionHistories type to internal\nfunc ToVersionHistories(t *shared.VersionHistories) *types.VersionHistories {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistories{\n\t\tCurrentVersionHistoryIndex: t.GetCurrentVersionHistoryIndex(),\n\t\tHistories:                  ToVersionHistoryArray(t.Histories),\n\t}\n}\n\n// FromVersionHistory converts internal VersionHistory type to thrift\nfunc FromVersionHistory(t *types.VersionHistory) *shared.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.VersionHistory{\n\t\tBranchToken: t.BranchToken,\n\t\tItems:       FromVersionHistoryItemArray(t.Items),\n\t}\n}\n\n// ToVersionHistory converts thrift VersionHistory type to internal\nfunc ToVersionHistory(t *shared.VersionHistory) *types.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistory{\n\t\tBranchToken: t.BranchToken,\n\t\tItems:       ToVersionHistoryItemArray(t.Items),\n\t}\n}\n\n// FromVersionHistoryItem converts internal VersionHistoryItem type to thrift\nfunc FromVersionHistoryItem(t *types.VersionHistoryItem) *shared.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.VersionHistoryItem{\n\t\tEventID: &t.EventID,\n\t\tVersion: &t.Version,\n\t}\n}\n\n// ToVersionHistoryItem converts thrift VersionHistoryItem type to internal\nfunc ToVersionHistoryItem(t *shared.VersionHistoryItem) *types.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VersionHistoryItem{\n\t\tEventID: t.GetEventID(),\n\t\tVersion: t.GetVersion(),\n\t}\n}\n\n// FromWorkerVersionInfo converts internal WorkerVersionInfo type to thrift\nfunc FromWorkerVersionInfo(t *types.WorkerVersionInfo) *shared.WorkerVersionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkerVersionInfo{\n\t\tImpl:           &t.Impl,\n\t\tFeatureVersion: &t.FeatureVersion,\n\t}\n}\n\n// ToWorkerVersionInfo converts thrift WorkerVersionInfo type to internal\nfunc ToWorkerVersionInfo(t *shared.WorkerVersionInfo) *types.WorkerVersionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkerVersionInfo{\n\t\tImpl:           t.GetImpl(),\n\t\tFeatureVersion: t.GetFeatureVersion(),\n\t}\n}\n\n// FromWorkflowExecution converts internal WorkflowExecution type to thrift\nfunc FromWorkflowExecution(t *types.WorkflowExecution) *shared.WorkflowExecution {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecution{\n\t\tWorkflowId: &t.WorkflowID,\n\t\tRunId:      &t.RunID,\n\t}\n}\n\n// ToWorkflowExecution converts thrift WorkflowExecution type to internal\nfunc ToWorkflowExecution(t *shared.WorkflowExecution) *types.WorkflowExecution {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecution{\n\t\tWorkflowID: t.GetWorkflowId(),\n\t\tRunID:      t.GetRunId(),\n\t}\n}\n\n// FromWorkflowExecutionAlreadyStartedError converts internal WorkflowExecutionAlreadyStartedError type to thrift\nfunc FromWorkflowExecutionAlreadyStartedError(t *types.WorkflowExecutionAlreadyStartedError) *shared.WorkflowExecutionAlreadyStartedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionAlreadyStartedError{\n\t\tMessage:        &t.Message,\n\t\tStartRequestId: &t.StartRequestID,\n\t\tRunId:          &t.RunID,\n\t}\n}\n\n// ToWorkflowExecutionAlreadyStartedError converts thrift WorkflowExecutionAlreadyStartedError type to internal\nfunc ToWorkflowExecutionAlreadyStartedError(t *shared.WorkflowExecutionAlreadyStartedError) *types.WorkflowExecutionAlreadyStartedError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionAlreadyStartedError{\n\t\tMessage:        t.GetMessage(),\n\t\tStartRequestID: t.GetStartRequestId(),\n\t\tRunID:          t.GetRunId(),\n\t}\n}\n\n// FromWorkflowExecutionCancelRequestedEventAttributes converts internal WorkflowExecutionCancelRequestedEventAttributes type to thrift\nfunc FromWorkflowExecutionCancelRequestedEventAttributes(t *types.WorkflowExecutionCancelRequestedEventAttributes) *shared.WorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                     &t.Cause,\n\t\tExternalInitiatedEventId:  t.ExternalInitiatedEventID,\n\t\tExternalWorkflowExecution: FromWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tIdentity:                  &t.Identity,\n\t\tRequestId:                 &t.RequestID,\n\t}\n}\n\n// ToWorkflowExecutionCancelRequestedEventAttributes converts thrift WorkflowExecutionCancelRequestedEventAttributes type to internal\nfunc ToWorkflowExecutionCancelRequestedEventAttributes(t *shared.WorkflowExecutionCancelRequestedEventAttributes) *types.WorkflowExecutionCancelRequestedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                     t.GetCause(),\n\t\tExternalInitiatedEventID:  t.ExternalInitiatedEventId,\n\t\tExternalWorkflowExecution: ToWorkflowExecution(t.ExternalWorkflowExecution),\n\t\tIdentity:                  t.GetIdentity(),\n\t\tRequestID:                 t.GetRequestId(),\n\t}\n}\n\n// FromWorkflowExecutionCanceledEventAttributes converts internal WorkflowExecutionCanceledEventAttributes type to thrift\nfunc FromWorkflowExecutionCanceledEventAttributes(t *types.WorkflowExecutionCanceledEventAttributes) *shared.WorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t\tDetails:                      t.Details,\n\t}\n}\n\n// ToWorkflowExecutionCanceledEventAttributes converts thrift WorkflowExecutionCanceledEventAttributes type to internal\nfunc ToWorkflowExecutionCanceledEventAttributes(t *shared.WorkflowExecutionCanceledEventAttributes) *types.WorkflowExecutionCanceledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t\tDetails:                      t.Details,\n\t}\n}\n\n// FromWorkflowExecutionCloseStatus converts internal WorkflowExecutionCloseStatus type to thrift\nfunc FromWorkflowExecutionCloseStatus(t *types.WorkflowExecutionCloseStatus) *shared.WorkflowExecutionCloseStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.WorkflowExecutionCloseStatusCompleted:\n\t\tv := shared.WorkflowExecutionCloseStatusCompleted\n\t\treturn &v\n\tcase types.WorkflowExecutionCloseStatusFailed:\n\t\tv := shared.WorkflowExecutionCloseStatusFailed\n\t\treturn &v\n\tcase types.WorkflowExecutionCloseStatusCanceled:\n\t\tv := shared.WorkflowExecutionCloseStatusCanceled\n\t\treturn &v\n\tcase types.WorkflowExecutionCloseStatusTerminated:\n\t\tv := shared.WorkflowExecutionCloseStatusTerminated\n\t\treturn &v\n\tcase types.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\tv := shared.WorkflowExecutionCloseStatusContinuedAsNew\n\t\treturn &v\n\tcase types.WorkflowExecutionCloseStatusTimedOut:\n\t\tv := shared.WorkflowExecutionCloseStatusTimedOut\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToWorkflowExecutionCloseStatus converts thrift WorkflowExecutionCloseStatus type to internal\nfunc ToWorkflowExecutionCloseStatus(t *shared.WorkflowExecutionCloseStatus) *types.WorkflowExecutionCloseStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.WorkflowExecutionCloseStatusCompleted:\n\t\tv := types.WorkflowExecutionCloseStatusCompleted\n\t\treturn &v\n\tcase shared.WorkflowExecutionCloseStatusFailed:\n\t\tv := types.WorkflowExecutionCloseStatusFailed\n\t\treturn &v\n\tcase shared.WorkflowExecutionCloseStatusCanceled:\n\t\tv := types.WorkflowExecutionCloseStatusCanceled\n\t\treturn &v\n\tcase shared.WorkflowExecutionCloseStatusTerminated:\n\t\tv := types.WorkflowExecutionCloseStatusTerminated\n\t\treturn &v\n\tcase shared.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\tv := types.WorkflowExecutionCloseStatusContinuedAsNew\n\t\treturn &v\n\tcase shared.WorkflowExecutionCloseStatusTimedOut:\n\t\tv := types.WorkflowExecutionCloseStatusTimedOut\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromWorkflowExecutionStatus converts internal WorkflowExecutionStatus type to thrift\nfunc FromWorkflowExecutionStatus(t *types.WorkflowExecutionStatus) *shared.WorkflowExecutionStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.WorkflowExecutionStatusPending:\n\t\tv := shared.WorkflowExecutionStatusPending\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusStarted:\n\t\tv := shared.WorkflowExecutionStatusStarted\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusCompleted:\n\t\tv := shared.WorkflowExecutionStatusCompleted\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusFailed:\n\t\tv := shared.WorkflowExecutionStatusFailed\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusCanceled:\n\t\tv := shared.WorkflowExecutionStatusCanceled\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusTerminated:\n\t\tv := shared.WorkflowExecutionStatusTerminated\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusContinuedAsNew:\n\t\tv := shared.WorkflowExecutionStatusContinuedAsNew\n\t\treturn &v\n\tcase types.WorkflowExecutionStatusTimedOut:\n\t\tv := shared.WorkflowExecutionStatusTimedOut\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToWorkflowExecutionStatus converts thrift WorkflowExecutionStatus type to internal\nfunc ToWorkflowExecutionStatus(t *shared.WorkflowExecutionStatus) *types.WorkflowExecutionStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.WorkflowExecutionStatusPending:\n\t\tv := types.WorkflowExecutionStatusPending\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusStarted:\n\t\tv := types.WorkflowExecutionStatusStarted\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusCompleted:\n\t\tv := types.WorkflowExecutionStatusCompleted\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusFailed:\n\t\tv := types.WorkflowExecutionStatusFailed\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusCanceled:\n\t\tv := types.WorkflowExecutionStatusCanceled\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusTerminated:\n\t\tv := types.WorkflowExecutionStatusTerminated\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusContinuedAsNew:\n\t\tv := types.WorkflowExecutionStatusContinuedAsNew\n\t\treturn &v\n\tcase shared.WorkflowExecutionStatusTimedOut:\n\t\tv := types.WorkflowExecutionStatusTimedOut\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromWorkflowExecutionCompletedEventAttributes converts internal WorkflowExecutionCompletedEventAttributes type to thrift\nfunc FromWorkflowExecutionCompletedEventAttributes(t *types.WorkflowExecutionCompletedEventAttributes) *shared.WorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       t.Result,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToWorkflowExecutionCompletedEventAttributes converts thrift WorkflowExecutionCompletedEventAttributes type to internal\nfunc ToWorkflowExecutionCompletedEventAttributes(t *shared.WorkflowExecutionCompletedEventAttributes) *types.WorkflowExecutionCompletedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       t.Result,\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromWorkflowExecutionConfiguration converts internal WorkflowExecutionConfiguration type to thrift\nfunc FromWorkflowExecutionConfiguration(t *types.WorkflowExecutionConfiguration) *shared.WorkflowExecutionConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionConfiguration{\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t}\n}\n\n// ToWorkflowExecutionConfiguration converts thrift WorkflowExecutionConfiguration type to internal\nfunc ToWorkflowExecutionConfiguration(t *shared.WorkflowExecutionConfiguration) *types.WorkflowExecutionConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionConfiguration{\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t}\n}\n\n// FromWorkflowExecutionContinuedAsNewEventAttributes converts internal WorkflowExecutionContinuedAsNewEventAttributes type to thrift\nfunc FromWorkflowExecutionContinuedAsNewEventAttributes(t *types.WorkflowExecutionContinuedAsNewEventAttributes) *shared.WorkflowExecutionContinuedAsNewEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunId:                   &t.NewExecutionRunID,\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tDecisionTaskCompletedEventId:        &t.DecisionTaskCompletedEventID,\n\t\tBackoffStartIntervalInSeconds:       t.BackoffStartIntervalInSeconds,\n\t\tInitiator:                           FromContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       t.FailureReason,\n\t\tFailureDetails:                      t.FailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tHeader:                              FromHeader(t.Header),\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\n// ToWorkflowExecutionContinuedAsNewEventAttributes converts thrift WorkflowExecutionContinuedAsNewEventAttributes type to internal\nfunc ToWorkflowExecutionContinuedAsNewEventAttributes(t *shared.WorkflowExecutionContinuedAsNewEventAttributes) *types.WorkflowExecutionContinuedAsNewEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunID:                   t.GetNewExecutionRunId(),\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tDecisionTaskCompletedEventID:        t.GetDecisionTaskCompletedEventId(),\n\t\tBackoffStartIntervalInSeconds:       t.BackoffStartIntervalInSeconds,\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tFailureReason:                       t.FailureReason,\n\t\tFailureDetails:                      t.FailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tHeader:                              ToHeader(t.Header),\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\n// FromWorkflowExecutionFailedEventAttributes converts internal WorkflowExecutionFailedEventAttributes type to thrift\nfunc FromWorkflowExecutionFailedEventAttributes(t *types.WorkflowExecutionFailedEventAttributes) *shared.WorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionFailedEventAttributes{\n\t\tReason:                       t.Reason,\n\t\tDetails:                      t.Details,\n\t\tDecisionTaskCompletedEventId: &t.DecisionTaskCompletedEventID,\n\t}\n}\n\n// ToWorkflowExecutionFailedEventAttributes converts thrift WorkflowExecutionFailedEventAttributes type to internal\nfunc ToWorkflowExecutionFailedEventAttributes(t *shared.WorkflowExecutionFailedEventAttributes) *types.WorkflowExecutionFailedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionFailedEventAttributes{\n\t\tReason:                       t.Reason,\n\t\tDetails:                      t.Details,\n\t\tDecisionTaskCompletedEventID: t.GetDecisionTaskCompletedEventId(),\n\t}\n}\n\n// FromWorkflowExecutionFilter converts internal WorkflowExecutionFilter type to thrift\nfunc FromWorkflowExecutionFilter(t *types.WorkflowExecutionFilter) *shared.WorkflowExecutionFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionFilter{\n\t\tWorkflowId: &t.WorkflowID,\n\t\tRunId:      &t.RunID,\n\t}\n}\n\n// ToWorkflowExecutionFilter converts thrift WorkflowExecutionFilter type to internal\nfunc ToWorkflowExecutionFilter(t *shared.WorkflowExecutionFilter) *types.WorkflowExecutionFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionFilter{\n\t\tWorkflowID: t.GetWorkflowId(),\n\t\tRunID:      t.GetRunId(),\n\t}\n}\n\n// FromWorkflowExecutionInfo converts internal WorkflowExecutionInfo type to thrift\nfunc FromWorkflowExecutionInfo(t *types.WorkflowExecutionInfo) *shared.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttlName := \"\"\n\tif t.TaskList != nil {\n\t\ttlName = t.TaskList.Name\n\t}\n\treturn &shared.WorkflowExecutionInfo{\n\t\tExecution:                    FromWorkflowExecution(t.Execution),\n\t\tType:                         FromWorkflowType(t.Type),\n\t\tStartTime:                    t.StartTime,\n\t\tCloseTime:                    t.CloseTime,\n\t\tCloseStatus:                  FromWorkflowExecutionCloseStatus(t.CloseStatus),\n\t\tHistoryLength:                &t.HistoryLength,\n\t\tParentDomainId:               t.ParentDomainID,\n\t\tParentDomainName:             t.ParentDomain,\n\t\tParentInitatedId:             t.ParentInitiatedID,\n\t\tParentExecution:              FromWorkflowExecution(t.ParentExecution),\n\t\tExecutionTime:                t.ExecutionTime,\n\t\tMemo:                         FromMemo(t.Memo),\n\t\tSearchAttributes:             FromSearchAttributes(t.SearchAttributes),\n\t\tAutoResetPoints:              FromResetPoints(t.AutoResetPoints),\n\t\tTaskList:                     &tlName,\n\t\tTaskListInfo:                 FromTaskList(t.TaskList),\n\t\tIsCron:                       &t.IsCron,\n\t\tUpdateTime:                   t.UpdateTime,\n\t\tPartitionConfig:              t.PartitionConfig,\n\t\tCronOverlapPolicy:            FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tExecutionStatus:              FromWorkflowExecutionStatus(t.ExecutionStatus),\n\t\tScheduledExecutionTime:       t.ScheduledExecutionTime,\n\t}\n}\n\n// ToWorkflowExecutionInfo converts thrift WorkflowExecutionInfo type to internal\nfunc ToWorkflowExecutionInfo(t *shared.WorkflowExecutionInfo) *types.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionInfo{\n\t\tExecution:                    ToWorkflowExecution(t.Execution),\n\t\tType:                         ToWorkflowType(t.Type),\n\t\tStartTime:                    t.StartTime,\n\t\tCloseTime:                    t.CloseTime,\n\t\tCloseStatus:                  ToWorkflowExecutionCloseStatus(t.CloseStatus),\n\t\tHistoryLength:                t.GetHistoryLength(),\n\t\tParentDomainID:               t.ParentDomainId,\n\t\tParentDomain:                 t.ParentDomainName,\n\t\tParentInitiatedID:            t.ParentInitatedId,\n\t\tParentExecution:              ToWorkflowExecution(t.ParentExecution),\n\t\tExecutionTime:                t.ExecutionTime,\n\t\tMemo:                         ToMemo(t.Memo),\n\t\tSearchAttributes:             ToSearchAttributes(t.SearchAttributes),\n\t\tAutoResetPoints:              ToResetPoints(t.AutoResetPoints),\n\t\tTaskList:                     MigrateTaskList(t.GetTaskList(), t.TaskListInfo),\n\t\tIsCron:                       t.GetIsCron(),\n\t\tUpdateTime:                   t.UpdateTime,\n\t\tPartitionConfig:              t.PartitionConfig,\n\t\tCronOverlapPolicy:            ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t\tActiveClusterSelectionPolicy: ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronSchedule:                 t.CronSchedule,\n\t\tExecutionStatus:              ToWorkflowExecutionStatus(t.ExecutionStatus),\n\t\tScheduledExecutionTime:       t.ScheduledExecutionTime,\n\t}\n}\n\n// FromWorkflowExecutionSignaledEventAttributes converts internal WorkflowExecutionSignaledEventAttributes type to thrift\nfunc FromWorkflowExecutionSignaledEventAttributes(t *types.WorkflowExecutionSignaledEventAttributes) *shared.WorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: &t.SignalName,\n\t\tInput:      t.Input,\n\t\tIdentity:   &t.Identity,\n\t\tRequestId:  &t.RequestID,\n\t}\n}\n\n// ToWorkflowExecutionSignaledEventAttributes converts thrift WorkflowExecutionSignaledEventAttributes type to internal\nfunc ToWorkflowExecutionSignaledEventAttributes(t *shared.WorkflowExecutionSignaledEventAttributes) *types.WorkflowExecutionSignaledEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: t.GetSignalName(),\n\t\tInput:      t.Input,\n\t\tIdentity:   t.GetIdentity(),\n\t\tRequestID:  t.GetRequestId(),\n\t}\n}\n\n// FromWorkflowExecutionStartedEventAttributes converts internal WorkflowExecutionStartedEventAttributes type to thrift\nfunc FromWorkflowExecutionStartedEventAttributes(t *types.WorkflowExecutionStartedEventAttributes) *shared.WorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &shared.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        FromWorkflowType(t.WorkflowType),\n\t\tParentWorkflowDomain:                t.ParentWorkflowDomain,\n\t\tParentWorkflowExecution:             FromWorkflowExecution(t.ParentWorkflowExecution),\n\t\tParentInitiatedEventId:              t.ParentInitiatedEventID,\n\t\tTaskList:                            FromTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tContinuedExecutionRunId:             &t.ContinuedExecutionRunID,\n\t\tInitiator:                           FromContinueAsNewInitiator(t.Initiator),\n\t\tContinuedFailureReason:              t.ContinuedFailureReason,\n\t\tContinuedFailureDetails:             t.ContinuedFailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tOriginalExecutionRunId:              &t.OriginalExecutionRunID,\n\t\tIdentity:                            &t.Identity,\n\t\tFirstExecutionRunId:                 &t.FirstExecutionRunID,\n\t\tFirstScheduledTimeNano:              timeToNano(t.FirstScheduleTime),\n\t\tRetryPolicy:                         FromRetryPolicy(t.RetryPolicy),\n\t\tAttempt:                             &t.Attempt,\n\t\tExpirationTimestamp:                 t.ExpirationTimestamp,\n\t\tCronSchedule:                        &t.CronSchedule,\n\t\tFirstDecisionTaskBackoffSeconds:     t.FirstDecisionTaskBackoffSeconds,\n\t\tMemo:                                FromMemo(t.Memo),\n\t\tSearchAttributes:                    FromSearchAttributes(t.SearchAttributes),\n\t\tPrevAutoResetPoints:                 FromResetPoints(t.PrevAutoResetPoints),\n\t\tHeader:                              FromHeader(t.Header),\n\t\tPartitionConfig:                     t.PartitionConfig,\n\t\tRequestId:                           &t.RequestID,\n\t\tActiveClusterSelectionPolicy:        FromActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   FromCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\n// ToWorkflowExecutionStartedEventAttributes converts thrift WorkflowExecutionStartedEventAttributes type to internal\nfunc ToWorkflowExecutionStartedEventAttributes(t *shared.WorkflowExecutionStartedEventAttributes) *types.WorkflowExecutionStartedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\n\treturn &types.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        ToWorkflowType(t.WorkflowType),\n\t\tParentWorkflowDomain:                t.ParentWorkflowDomain,\n\t\tParentWorkflowExecution:             ToWorkflowExecution(t.ParentWorkflowExecution),\n\t\tParentInitiatedEventID:              t.ParentInitiatedEventId,\n\t\tTaskList:                            ToTaskList(t.TaskList),\n\t\tInput:                               t.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: t.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      t.TaskStartToCloseTimeoutSeconds,\n\t\tContinuedExecutionRunID:             t.GetContinuedExecutionRunId(),\n\t\tInitiator:                           ToContinueAsNewInitiator(t.Initiator),\n\t\tContinuedFailureReason:              t.ContinuedFailureReason,\n\t\tContinuedFailureDetails:             t.ContinuedFailureDetails,\n\t\tLastCompletionResult:                t.LastCompletionResult,\n\t\tOriginalExecutionRunID:              t.GetOriginalExecutionRunId(),\n\t\tIdentity:                            t.GetIdentity(),\n\t\tFirstExecutionRunID:                 t.GetFirstExecutionRunId(),\n\t\tFirstScheduleTime:                   nanoToTime(t.FirstScheduledTimeNano),\n\t\tRetryPolicy:                         ToRetryPolicy(t.RetryPolicy),\n\t\tAttempt:                             t.GetAttempt(),\n\t\tExpirationTimestamp:                 t.ExpirationTimestamp,\n\t\tCronSchedule:                        t.GetCronSchedule(),\n\t\tFirstDecisionTaskBackoffSeconds:     t.FirstDecisionTaskBackoffSeconds,\n\t\tMemo:                                ToMemo(t.Memo),\n\t\tSearchAttributes:                    ToSearchAttributes(t.SearchAttributes),\n\t\tPrevAutoResetPoints:                 ToResetPoints(t.PrevAutoResetPoints),\n\t\tHeader:                              ToHeader(t.Header),\n\t\tPartitionConfig:                     t.PartitionConfig,\n\t\tRequestID:                           t.GetRequestId(),\n\t\tActiveClusterSelectionPolicy:        ToActiveClusterSelectionPolicy(t.ActiveClusterSelectionPolicy),\n\t\tCronOverlapPolicy:                   ToCronOverlapPolicy(t.CronOverlapPolicy),\n\t}\n}\n\nfunc FromClusterAttribute(t *types.ClusterAttribute) *shared.ClusterAttribute {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ClusterAttribute{\n\t\tScope: &t.Scope,\n\t\tName:  &t.Name,\n\t}\n}\n\nfunc ToClusterAttribute(t *shared.ClusterAttribute) *types.ClusterAttribute {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ClusterAttribute{\n\t\tScope: t.GetScope(),\n\t\tName:  t.GetName(),\n\t}\n}\n\nfunc FromActiveClusterSelectionPolicy(t *types.ActiveClusterSelectionPolicy) *shared.ActiveClusterSelectionPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ActiveClusterSelectionPolicy{\n\t\tStrategy:           FromActiveClusterSelectionStrategy(t.ActiveClusterSelectionStrategy),\n\t\tExternalEntityType: &t.ExternalEntityType,\n\t\tExternalEntityKey:  &t.ExternalEntityKey,\n\t\tStickyRegion:       &t.StickyRegion,\n\t\tClusterAttribute:   FromClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\nfunc ToActiveClusterSelectionPolicy(t *shared.ActiveClusterSelectionPolicy) *types.ActiveClusterSelectionPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: ToActiveClusterSelectionStrategy(t.Strategy),\n\t\tExternalEntityType:             t.GetExternalEntityType(),\n\t\tExternalEntityKey:              t.GetExternalEntityKey(),\n\t\tStickyRegion:                   t.GetStickyRegion(),\n\t\tClusterAttribute:               ToClusterAttribute(t.ClusterAttribute),\n\t}\n}\n\nfunc FromActiveClusterSelectionStrategy(t *types.ActiveClusterSelectionStrategy) *shared.ActiveClusterSelectionStrategy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.ActiveClusterSelectionStrategyRegionSticky:\n\t\treturn shared.ActiveClusterSelectionStrategyRegionSticky.Ptr()\n\tcase types.ActiveClusterSelectionStrategyExternalEntity:\n\t\treturn shared.ActiveClusterSelectionStrategyExternalEntity.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\nfunc ToActiveClusterSelectionStrategy(t *shared.ActiveClusterSelectionStrategy) *types.ActiveClusterSelectionStrategy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.ActiveClusterSelectionStrategyRegionSticky:\n\t\treturn types.ActiveClusterSelectionStrategyRegionSticky.Ptr()\n\tcase shared.ActiveClusterSelectionStrategyExternalEntity:\n\t\treturn types.ActiveClusterSelectionStrategyExternalEntity.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromWorkflowExecutionTerminatedEventAttributes converts internal WorkflowExecutionTerminatedEventAttributes type to thrift\nfunc FromWorkflowExecutionTerminatedEventAttributes(t *types.WorkflowExecutionTerminatedEventAttributes) *shared.WorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   &t.Reason,\n\t\tDetails:  t.Details,\n\t\tIdentity: &t.Identity,\n\t}\n}\n\n// ToWorkflowExecutionTerminatedEventAttributes converts thrift WorkflowExecutionTerminatedEventAttributes type to internal\nfunc ToWorkflowExecutionTerminatedEventAttributes(t *shared.WorkflowExecutionTerminatedEventAttributes) *types.WorkflowExecutionTerminatedEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   t.GetReason(),\n\t\tDetails:  t.Details,\n\t\tIdentity: t.GetIdentity(),\n\t}\n}\n\n// FromWorkflowExecutionTimedOutEventAttributes converts internal WorkflowExecutionTimedOutEventAttributes type to thrift\nfunc FromWorkflowExecutionTimedOutEventAttributes(t *types.WorkflowExecutionTimedOutEventAttributes) *shared.WorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: FromTimeoutType(t.TimeoutType),\n\t}\n}\n\n// ToWorkflowExecutionTimedOutEventAttributes converts thrift WorkflowExecutionTimedOutEventAttributes type to internal\nfunc ToWorkflowExecutionTimedOutEventAttributes(t *shared.WorkflowExecutionTimedOutEventAttributes) *types.WorkflowExecutionTimedOutEventAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: ToTimeoutType(t.TimeoutType),\n\t}\n}\n\n// FromWorkflowIDReusePolicy converts internal WorkflowIDReusePolicy type to thrift\nfunc FromWorkflowIDReusePolicy(t *types.WorkflowIDReusePolicy) *shared.WorkflowIdReusePolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.WorkflowIDReusePolicyAllowDuplicateFailedOnly:\n\t\tv := shared.WorkflowIdReusePolicyAllowDuplicateFailedOnly\n\t\treturn &v\n\tcase types.WorkflowIDReusePolicyAllowDuplicate:\n\t\tv := shared.WorkflowIdReusePolicyAllowDuplicate\n\t\treturn &v\n\tcase types.WorkflowIDReusePolicyRejectDuplicate:\n\t\tv := shared.WorkflowIdReusePolicyRejectDuplicate\n\t\treturn &v\n\tcase types.WorkflowIDReusePolicyTerminateIfRunning:\n\t\tv := shared.WorkflowIdReusePolicyTerminateIfRunning\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToWorkflowIDReusePolicy converts thrift WorkflowIdReusePolicy type to internal\nfunc ToWorkflowIDReusePolicy(t *shared.WorkflowIdReusePolicy) *types.WorkflowIDReusePolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.WorkflowIdReusePolicyAllowDuplicateFailedOnly:\n\t\tv := types.WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\t\treturn &v\n\tcase shared.WorkflowIdReusePolicyAllowDuplicate:\n\t\tv := types.WorkflowIDReusePolicyAllowDuplicate\n\t\treturn &v\n\tcase shared.WorkflowIdReusePolicyRejectDuplicate:\n\t\tv := types.WorkflowIDReusePolicyRejectDuplicate\n\t\treturn &v\n\tcase shared.WorkflowIdReusePolicyTerminateIfRunning:\n\t\tv := types.WorkflowIDReusePolicyTerminateIfRunning\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromWorkflowQuery converts internal WorkflowQuery type to thrift\nfunc FromWorkflowQuery(t *types.WorkflowQuery) *shared.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowQuery{\n\t\tQueryType: &t.QueryType,\n\t\tQueryArgs: t.QueryArgs,\n\t}\n}\n\n// ToWorkflowQuery converts thrift WorkflowQuery type to internal\nfunc ToWorkflowQuery(t *shared.WorkflowQuery) *types.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowQuery{\n\t\tQueryType: t.GetQueryType(),\n\t\tQueryArgs: t.QueryArgs,\n\t}\n}\n\n// FromWorkflowQueryResult converts internal WorkflowQueryResult type to thrift\nfunc FromWorkflowQueryResult(t *types.WorkflowQueryResult) *shared.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowQueryResult{\n\t\tResultType:   FromQueryResultType(t.ResultType),\n\t\tAnswer:       t.Answer,\n\t\tErrorMessage: &t.ErrorMessage,\n\t}\n}\n\n// ToWorkflowQueryResult converts thrift WorkflowQueryResult type to internal\nfunc ToWorkflowQueryResult(t *shared.WorkflowQueryResult) *types.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowQueryResult{\n\t\tResultType:   ToQueryResultType(t.ResultType),\n\t\tAnswer:       t.Answer,\n\t\tErrorMessage: t.GetErrorMessage(),\n\t}\n}\n\n// FromWorkflowType converts internal WorkflowType type to thrift\nfunc FromWorkflowType(t *types.WorkflowType) *shared.WorkflowType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowType{\n\t\tName: &t.Name,\n\t}\n}\n\n// ToWorkflowType converts thrift WorkflowType type to internal\nfunc ToWorkflowType(t *shared.WorkflowType) *types.WorkflowType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowType{\n\t\tName: t.GetName(),\n\t}\n}\n\n// FromWorkflowTypeFilter converts internal WorkflowTypeFilter type to thrift\nfunc FromWorkflowTypeFilter(t *types.WorkflowTypeFilter) *shared.WorkflowTypeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.WorkflowTypeFilter{\n\t\tName: &t.Name,\n\t}\n}\n\n// ToWorkflowTypeFilter converts thrift WorkflowTypeFilter type to internal\nfunc ToWorkflowTypeFilter(t *shared.WorkflowTypeFilter) *types.WorkflowTypeFilter {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.WorkflowTypeFilter{\n\t\tName: t.GetName(),\n\t}\n}\n\n// FromPendingActivityInfoArray converts internal PendingActivityInfo type array to thrift\nfunc FromPendingActivityInfoArray(t []*types.PendingActivityInfo) []*shared.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.PendingActivityInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPendingActivityInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToPendingActivityInfoArray converts thrift PendingActivityInfo type array to internal\nfunc ToPendingActivityInfoArray(t []*shared.PendingActivityInfo) []*types.PendingActivityInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PendingActivityInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPendingActivityInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromHistoryEventArray converts internal HistoryEvent type array to thrift\nfunc FromHistoryEventArray(t []*types.HistoryEvent) []*shared.HistoryEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.HistoryEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHistoryEvent(t[i])\n\t}\n\treturn v\n}\n\n// ToHistoryEventArray converts thrift HistoryEvent type array to internal\nfunc ToHistoryEventArray(t []*shared.HistoryEvent) []*types.HistoryEvent {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.HistoryEvent, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHistoryEvent(t[i])\n\t}\n\treturn v\n}\n\n// FromWorkflowExecutionInfoArray converts internal WorkflowExecutionInfo type array to thrift\nfunc FromWorkflowExecutionInfoArray(t []*types.WorkflowExecutionInfo) []*shared.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.WorkflowExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromWorkflowExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToWorkflowExecutionInfoArray converts thrift WorkflowExecutionInfo type array to internal\nfunc ToWorkflowExecutionInfoArray(t []*shared.WorkflowExecutionInfo) []*types.WorkflowExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.WorkflowExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToWorkflowExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromPollerInfoArray converts internal PollerInfo type array to thrift\nfunc FromPollerInfoArray(t []*types.PollerInfo) []*shared.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.PollerInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPollerInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToPollerInfoArray converts thrift PollerInfo type array to internal\nfunc ToPollerInfoArray(t []*shared.PollerInfo) []*types.PollerInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PollerInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPollerInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromClusterReplicationConfigurationArray converts internal ClusterReplicationConfiguration type array to thrift\nfunc FromClusterReplicationConfigurationArray(t []*types.ClusterReplicationConfiguration) []*shared.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.ClusterReplicationConfiguration, len(t))\n\tfor i := range t {\n\t\tv[i] = FromClusterReplicationConfiguration(t[i])\n\t}\n\treturn v\n}\n\n// ToClusterReplicationConfigurationArray converts thrift ClusterReplicationConfiguration type array to internal\nfunc ToClusterReplicationConfigurationArray(t []*shared.ClusterReplicationConfiguration) []*types.ClusterReplicationConfiguration {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ClusterReplicationConfiguration, len(t))\n\tfor i := range t {\n\t\tv[i] = ToClusterReplicationConfiguration(t[i])\n\t}\n\treturn v\n}\n\n// FromDataBlobArray converts internal DataBlob type array to thrift\nfunc FromDataBlobArray(t []*types.DataBlob) []*shared.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.DataBlob, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDataBlob(t[i])\n\t}\n\treturn v\n}\n\n// ToDataBlobArray converts thrift DataBlob type array to internal\nfunc ToDataBlobArray(t []*shared.DataBlob) []*types.DataBlob {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DataBlob, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDataBlob(t[i])\n\t}\n\treturn v\n}\n\n// FromDescribeDomainResponseArray converts internal DescribeDomainResponse type array to thrift\nfunc FromDescribeDomainResponseArray(t []*types.DescribeDomainResponse) []*shared.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.DescribeDomainResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDescribeDomainResponse(t[i])\n\t}\n\treturn v\n}\n\n// ToDescribeDomainResponseArray converts thrift DescribeDomainResponse type array to internal\nfunc ToDescribeDomainResponseArray(t []*shared.DescribeDomainResponse) []*types.DescribeDomainResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.DescribeDomainResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDescribeDomainResponse(t[i])\n\t}\n\treturn v\n}\n\n// FromDecisionArray converts internal Decision type array to thrift\nfunc FromDecisionArray(t []*types.Decision) []*shared.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.Decision, len(t))\n\tfor i := range t {\n\t\tv[i] = FromDecision(t[i])\n\t}\n\treturn v\n}\n\n// ToDecisionArray converts thrift Decision type array to internal\nfunc ToDecisionArray(t []*shared.Decision) []*types.Decision {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.Decision, len(t))\n\tfor i := range t {\n\t\tv[i] = ToDecision(t[i])\n\t}\n\treturn v\n}\n\n// FromVersionHistoryArray converts internal VersionHistory type array to thrift\nfunc FromVersionHistoryArray(t []*types.VersionHistory) []*shared.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.VersionHistory, len(t))\n\tfor i := range t {\n\t\tv[i] = FromVersionHistory(t[i])\n\t}\n\treturn v\n}\n\n// ToVersionHistoryArray converts thrift VersionHistory type array to internal\nfunc ToVersionHistoryArray(t []*shared.VersionHistory) []*types.VersionHistory {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.VersionHistory, len(t))\n\tfor i := range t {\n\t\tv[i] = ToVersionHistory(t[i])\n\t}\n\treturn v\n}\n\n// FromVersionHistoryItemArray converts internal VersionHistoryItem type array to thrift\nfunc FromVersionHistoryItemArray(t []*types.VersionHistoryItem) []*shared.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.VersionHistoryItem, len(t))\n\tfor i := range t {\n\t\tv[i] = FromVersionHistoryItem(t[i])\n\t}\n\treturn v\n}\n\n// ToVersionHistoryItemArray converts thrift VersionHistoryItem type array to internal\nfunc ToVersionHistoryItemArray(t []*shared.VersionHistoryItem) []*types.VersionHistoryItem {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.VersionHistoryItem, len(t))\n\tfor i := range t {\n\t\tv[i] = ToVersionHistoryItem(t[i])\n\t}\n\treturn v\n}\n\n// FromTaskListPartitionMetadataArray converts internal TaskListPartitionMetadata type array to thrift\nfunc FromTaskListPartitionMetadataArray(t []*types.TaskListPartitionMetadata) []*shared.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.TaskListPartitionMetadata, len(t))\n\tfor i := range t {\n\t\tv[i] = FromTaskListPartitionMetadata(t[i])\n\t}\n\treturn v\n}\n\n// ToTaskListPartitionMetadataArray converts thrift TaskListPartitionMetadata type array to internal\nfunc ToTaskListPartitionMetadataArray(t []*shared.TaskListPartitionMetadata) []*types.TaskListPartitionMetadata {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.TaskListPartitionMetadata, len(t))\n\tfor i := range t {\n\t\tv[i] = ToTaskListPartitionMetadata(t[i])\n\t}\n\treturn v\n}\n\n// FromResetPointInfoArray converts internal ResetPointInfo type array to thrift\nfunc FromResetPointInfoArray(t []*types.ResetPointInfo) []*shared.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.ResetPointInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromResetPointInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToResetPointInfoArray converts thrift ResetPointInfo type array to internal\nfunc ToResetPointInfoArray(t []*shared.ResetPointInfo) []*types.ResetPointInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ResetPointInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToResetPointInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromPendingChildExecutionInfoArray converts internal PendingChildExecutionInfo type array to thrift\nfunc FromPendingChildExecutionInfoArray(t []*types.PendingChildExecutionInfo) []*shared.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.PendingChildExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = FromPendingChildExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\n// ToPendingChildExecutionInfoArray converts thrift PendingChildExecutionInfo type array to internal\nfunc ToPendingChildExecutionInfoArray(t []*shared.PendingChildExecutionInfo) []*types.PendingChildExecutionInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.PendingChildExecutionInfo, len(t))\n\tfor i := range t {\n\t\tv[i] = ToPendingChildExecutionInfo(t[i])\n\t}\n\treturn v\n}\n\n// FromHistoryBranchRangeArray converts internal HistoryBranchRange type array to thrift\nfunc FromHistoryBranchRangeArray(t []*types.HistoryBranchRange) []*shared.HistoryBranchRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.HistoryBranchRange, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHistoryBranchRange(t[i])\n\t}\n\treturn v\n}\n\n// ToHistoryBranchRangeArray converts thrift HistoryBranchRange type array to internal\nfunc ToHistoryBranchRangeArray(t []*shared.HistoryBranchRange) []*types.HistoryBranchRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.HistoryBranchRange, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHistoryBranchRange(t[i])\n\t}\n\treturn v\n}\n\n// FromIndexedValueTypeMap converts internal IndexedValueType type map to thrift\nfunc FromIndexedValueTypeMap(t map[string]types.IndexedValueType) map[string]shared.IndexedValueType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]shared.IndexedValueType, len(t))\n\tfor key := range t {\n\t\tv[key] = FromIndexedValueType(t[key])\n\t}\n\treturn v\n}\n\n// ToIndexedValueTypeMap converts thrift IndexedValueType type map to internal\nfunc ToIndexedValueTypeMap(t map[string]shared.IndexedValueType) map[string]types.IndexedValueType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]types.IndexedValueType, len(t))\n\tfor key := range t {\n\t\tv[key] = ToIndexedValueType(t[key])\n\t}\n\treturn v\n}\n\n// FromWorkflowQueryMap converts internal WorkflowQuery type map to thrift\nfunc FromWorkflowQueryMap(t map[string]*types.WorkflowQuery) map[string]*shared.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*shared.WorkflowQuery, len(t))\n\tfor key := range t {\n\t\tv[key] = FromWorkflowQuery(t[key])\n\t}\n\treturn v\n}\n\n// ToWorkflowQueryMap converts thrift WorkflowQuery type map to internal\nfunc ToWorkflowQueryMap(t map[string]*shared.WorkflowQuery) map[string]*types.WorkflowQuery {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.WorkflowQuery, len(t))\n\tfor key := range t {\n\t\tv[key] = ToWorkflowQuery(t[key])\n\t}\n\treturn v\n}\n\n// FromWorkflowQueryResultMap converts internal WorkflowQueryResult type map to thrift\nfunc FromWorkflowQueryResultMap(t map[string]*types.WorkflowQueryResult) map[string]*shared.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*shared.WorkflowQueryResult, len(t))\n\tfor key := range t {\n\t\tv[key] = FromWorkflowQueryResult(t[key])\n\t}\n\treturn v\n}\n\n// ToWorkflowQueryResultMap converts thrift WorkflowQueryResult type map to internal\nfunc ToWorkflowQueryResultMap(t map[string]*shared.WorkflowQueryResult) map[string]*types.WorkflowQueryResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.WorkflowQueryResult, len(t))\n\tfor key := range t {\n\t\tv[key] = ToWorkflowQueryResult(t[key])\n\t}\n\treturn v\n}\n\n// FromActivityLocalDispatchInfoMap converts internal ActivityLocalDispatchInfo type map to thrift\nfunc FromActivityLocalDispatchInfoMap(t map[string]*types.ActivityLocalDispatchInfo) map[string]*shared.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*shared.ActivityLocalDispatchInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromActivityLocalDispatchInfo(t[key])\n\t}\n\treturn v\n}\n\n// ToActivityLocalDispatchInfoMap converts thrift ActivityLocalDispatchInfo type map to internal\nfunc ToActivityLocalDispatchInfoMap(t map[string]*shared.ActivityLocalDispatchInfo) map[string]*types.ActivityLocalDispatchInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.ActivityLocalDispatchInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToActivityLocalDispatchInfo(t[key])\n\t}\n\treturn v\n}\n\n// FromBadBinaryInfoMap converts internal BadBinaryInfo type map to thrift\nfunc FromBadBinaryInfoMap(t map[string]*types.BadBinaryInfo) map[string]*shared.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*shared.BadBinaryInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = FromBadBinaryInfo(t[key])\n\t}\n\treturn v\n}\n\n// ToBadBinaryInfoMap converts thrift BadBinaryInfo type map to internal\nfunc ToBadBinaryInfoMap(t map[string]*shared.BadBinaryInfo) map[string]*types.BadBinaryInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.BadBinaryInfo, len(t))\n\tfor key := range t {\n\t\tv[key] = ToBadBinaryInfo(t[key])\n\t}\n\treturn v\n}\n\nfunc FromIsolationGroupMetricsMap(t map[string]*types.IsolationGroupMetrics) map[string]*shared.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*shared.IsolationGroupMetrics, len(t))\n\tfor key := range t {\n\t\tv[key] = FromIsolationGroupMetrics(t[key])\n\t}\n\treturn v\n}\n\nfunc ToIsolationGroupMetricsMap(t map[string]*shared.IsolationGroupMetrics) map[string]*types.IsolationGroupMetrics {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[string]*types.IsolationGroupMetrics, len(t))\n\tfor key := range t {\n\t\tv[key] = ToIsolationGroupMetrics(t[key])\n\t}\n\treturn v\n}\n\n// FromHistoryArray converts internal History array to thrift\nfunc FromHistoryArray(t []*types.History) []*shared.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.History, len(t))\n\tfor i := range t {\n\t\tv[i] = FromHistory(t[i])\n\t}\n\treturn v\n}\n\n// ToHistoryArray converts thrift History array to internal\nfunc ToHistoryArray(t []*shared.History) []*types.History {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.History, len(t))\n\tfor i := range t {\n\t\tv[i] = ToHistory(t[i])\n\t}\n\treturn v\n}\n\n// FromCrossClusterTaskType converts internal CrossClusterTaskType type to thrift\nfunc FromCrossClusterTaskType(t *types.CrossClusterTaskType) *shared.CrossClusterTaskType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.CrossClusterTaskTypeStartChildExecution:\n\t\tv := shared.CrossClusterTaskTypeStartChildExecution\n\t\treturn &v\n\tcase types.CrossClusterTaskTypeCancelExecution:\n\t\tv := shared.CrossClusterTaskTypeCancelExecution\n\t\treturn &v\n\tcase types.CrossClusterTaskTypeSignalExecution:\n\t\tv := shared.CrossClusterTaskTypeSignalExecution\n\t\treturn &v\n\tcase types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete:\n\t\tv := shared.CrossClusterTaskTypeRecordChildWorkflowExecutionComplete\n\t\treturn &v\n\tcase types.CrossClusterTaskTypeApplyParentPolicy:\n\t\tv := shared.CrossClusterTaskTypeApplyParentClosePolicy\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToCrossClusterTaskType converts thrift CrossClusterTaskType type to internal\nfunc ToCrossClusterTaskType(t *shared.CrossClusterTaskType) *types.CrossClusterTaskType {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.CrossClusterTaskTypeStartChildExecution:\n\t\tv := types.CrossClusterTaskTypeStartChildExecution\n\t\treturn &v\n\tcase shared.CrossClusterTaskTypeCancelExecution:\n\t\tv := types.CrossClusterTaskTypeCancelExecution\n\t\treturn &v\n\tcase shared.CrossClusterTaskTypeSignalExecution:\n\t\tv := types.CrossClusterTaskTypeSignalExecution\n\t\treturn &v\n\tcase shared.CrossClusterTaskTypeRecordChildWorkflowExecutionComplete:\n\t\tv := types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete\n\t\treturn &v\n\tcase shared.CrossClusterTaskTypeApplyParentClosePolicy:\n\t\tv := types.CrossClusterTaskTypeApplyParentPolicy\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromCrossClusterTaskFailedCause converts internal CrossClusterTaskFailedCause type to thrift\nfunc FromCrossClusterTaskFailedCause(t *types.CrossClusterTaskFailedCause) *shared.CrossClusterTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.CrossClusterTaskFailedCauseDomainNotActive:\n\t\tv := shared.CrossClusterTaskFailedCauseDomainNotActive\n\t\treturn &v\n\tcase types.CrossClusterTaskFailedCauseDomainNotExists:\n\t\tv := shared.CrossClusterTaskFailedCauseDomainNotExists\n\t\treturn &v\n\tcase types.CrossClusterTaskFailedCauseWorkflowAlreadyRunning:\n\t\tv := shared.CrossClusterTaskFailedCauseWorkflowAlreadyRunning\n\t\treturn &v\n\tcase types.CrossClusterTaskFailedCauseWorkflowNotExists:\n\t\tv := shared.CrossClusterTaskFailedCauseWorkflowNotExists\n\t\treturn &v\n\tcase types.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted:\n\t\tv := shared.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\tcase types.CrossClusterTaskFailedCauseUncategorized:\n\t\tv := shared.CrossClusterTaskFailedCauseUncategorized\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToCrossClusterTaskFailedCause converts thrift CrossClusterTaskFailedCause type to internal\nfunc ToCrossClusterTaskFailedCause(t *shared.CrossClusterTaskFailedCause) *types.CrossClusterTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.CrossClusterTaskFailedCauseDomainNotActive:\n\t\tv := types.CrossClusterTaskFailedCauseDomainNotActive\n\t\treturn &v\n\tcase shared.CrossClusterTaskFailedCauseDomainNotExists:\n\t\tv := types.CrossClusterTaskFailedCauseDomainNotExists\n\t\treturn &v\n\tcase shared.CrossClusterTaskFailedCauseWorkflowAlreadyRunning:\n\t\tv := types.CrossClusterTaskFailedCauseWorkflowAlreadyRunning\n\t\treturn &v\n\tcase shared.CrossClusterTaskFailedCauseWorkflowNotExists:\n\t\tv := types.CrossClusterTaskFailedCauseWorkflowNotExists\n\t\treturn &v\n\tcase shared.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted:\n\t\tv := types.CrossClusterTaskFailedCauseWorkflowAlreadyCompleted\n\t\treturn &v\n\tcase shared.CrossClusterTaskFailedCauseUncategorized:\n\t\tv := types.CrossClusterTaskFailedCauseUncategorized\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromGetTaskFailedCause converts internal GetTaskFailedCause to thrift\nfunc FromGetTaskFailedCause(t *types.GetTaskFailedCause) *shared.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.GetTaskFailedCauseServiceBusy:\n\t\tv := shared.GetTaskFailedCauseServiceBusy\n\t\treturn &v\n\tcase types.GetTaskFailedCauseTimeout:\n\t\tv := shared.GetTaskFailedCauseTimeout\n\t\treturn &v\n\tcase types.GetTaskFailedCauseShardOwnershipLost:\n\t\tv := shared.GetTaskFailedCauseShardOwnershipLost\n\t\treturn &v\n\tcase types.GetTaskFailedCauseUncategorized:\n\t\tv := shared.GetTaskFailedCauseUncategorized\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// ToGetCrossClusterTaskFailedCause converts thrift GetTaskFailedCause type to internal\nfunc ToGetCrossClusterTaskFailedCause(t *shared.GetTaskFailedCause) *types.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.GetTaskFailedCauseServiceBusy:\n\t\tv := types.GetTaskFailedCauseServiceBusy\n\t\treturn &v\n\tcase shared.GetTaskFailedCauseTimeout:\n\t\tv := types.GetTaskFailedCauseTimeout\n\t\treturn &v\n\tcase shared.GetTaskFailedCauseShardOwnershipLost:\n\t\tv := types.GetTaskFailedCauseShardOwnershipLost\n\t\treturn &v\n\tcase shared.GetTaskFailedCauseUncategorized:\n\t\tv := types.GetTaskFailedCauseUncategorized\n\t\treturn &v\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\n// FromCrossClusterTaskInfo converts internal CrossClusterTaskInfo type to thrift\nfunc FromCrossClusterTaskInfo(t *types.CrossClusterTaskInfo) *shared.CrossClusterTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterTaskInfo{\n\t\tDomainID:            &t.DomainID,\n\t\tWorkflowID:          &t.WorkflowID,\n\t\tRunID:               &t.RunID,\n\t\tTaskType:            FromCrossClusterTaskType(t.TaskType),\n\t\tTaskState:           &t.TaskState,\n\t\tTaskID:              &t.TaskID,\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t}\n}\n\n// ToCrossClusterTaskInfo converts thrift CrossClusterTaskInfo type to internal\nfunc ToCrossClusterTaskInfo(t *shared.CrossClusterTaskInfo) *types.CrossClusterTaskInfo {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterTaskInfo{\n\t\tDomainID:            t.GetDomainID(),\n\t\tWorkflowID:          t.GetWorkflowID(),\n\t\tRunID:               t.GetRunID(),\n\t\tTaskType:            ToCrossClusterTaskType(t.TaskType),\n\t\tTaskState:           t.GetTaskState(),\n\t\tTaskID:              t.GetTaskID(),\n\t\tVisibilityTimestamp: t.VisibilityTimestamp,\n\t}\n}\n\n// FromCrossClusterStartChildExecutionRequestAttributes converts internal CrossClusterStartChildExecutionRequestAttributes type to thrift\nfunc FromCrossClusterStartChildExecutionRequestAttributes(t *types.CrossClusterStartChildExecutionRequestAttributes) *shared.CrossClusterStartChildExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterStartChildExecutionRequestAttributes{\n\t\tTargetDomainID:           &t.TargetDomainID,\n\t\tRequestID:                &t.RequestID,\n\t\tInitiatedEventID:         &t.InitiatedEventID,\n\t\tInitiatedEventAttributes: FromStartChildWorkflowExecutionInitiatedEventAttributes(t.InitiatedEventAttributes),\n\t\tTargetRunID:              t.TargetRunID,\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\n// ToCrossClusterStartChildExecutionRequestAttributes converts thrift CrossClusterStartChildExecutionRequestAttributes type to internal\nfunc ToCrossClusterStartChildExecutionRequestAttributes(t *shared.CrossClusterStartChildExecutionRequestAttributes) *types.CrossClusterStartChildExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterStartChildExecutionRequestAttributes{\n\t\tTargetDomainID:           t.GetTargetDomainID(),\n\t\tRequestID:                t.GetRequestID(),\n\t\tInitiatedEventID:         t.GetInitiatedEventID(),\n\t\tInitiatedEventAttributes: ToStartChildWorkflowExecutionInitiatedEventAttributes(t.InitiatedEventAttributes),\n\t\tTargetRunID:              t.TargetRunID,\n\t\tPartitionConfig:          t.PartitionConfig,\n\t}\n}\n\n// FromCrossClusterStartChildExecutionResponseAttributes converts internal CrossClusterStartChildExecutionResponseAttributes type to thrift\nfunc FromCrossClusterStartChildExecutionResponseAttributes(t *types.CrossClusterStartChildExecutionResponseAttributes) *shared.CrossClusterStartChildExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterStartChildExecutionResponseAttributes{\n\t\tRunID: &t.RunID,\n\t}\n}\n\n// ToCrossClusterStartChildExecutionResponseAttributes converts thrift CrossClusterStartChildExecutionResponseAttributes type to internal\nfunc ToCrossClusterStartChildExecutionResponseAttributes(t *shared.CrossClusterStartChildExecutionResponseAttributes) *types.CrossClusterStartChildExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterStartChildExecutionResponseAttributes{\n\t\tRunID: t.GetRunID(),\n\t}\n}\n\n// FromCrossClusterCancelExecutionRequestAttributes converts internal CrossClusterCancelExecutionRequestAttributes type to thrift\nfunc FromCrossClusterCancelExecutionRequestAttributes(t *types.CrossClusterCancelExecutionRequestAttributes) *shared.CrossClusterCancelExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterCancelExecutionRequestAttributes{\n\t\tTargetDomainID:    &t.TargetDomainID,\n\t\tTargetWorkflowID:  &t.TargetWorkflowID,\n\t\tTargetRunID:       &t.TargetRunID,\n\t\tRequestID:         &t.RequestID,\n\t\tInitiatedEventID:  &t.InitiatedEventID,\n\t\tChildWorkflowOnly: &t.ChildWorkflowOnly,\n\t}\n}\n\n// ToCrossClusterCancelExecutionRequestAttributes converts thrift CrossClusterCancelExecutionRequestAttributes type to internal\nfunc ToCrossClusterCancelExecutionRequestAttributes(t *shared.CrossClusterCancelExecutionRequestAttributes) *types.CrossClusterCancelExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterCancelExecutionRequestAttributes{\n\t\tTargetDomainID:    t.GetTargetDomainID(),\n\t\tTargetWorkflowID:  t.GetTargetWorkflowID(),\n\t\tTargetRunID:       t.GetTargetRunID(),\n\t\tRequestID:         t.GetRequestID(),\n\t\tInitiatedEventID:  t.GetInitiatedEventID(),\n\t\tChildWorkflowOnly: t.GetChildWorkflowOnly(),\n\t}\n}\n\n// FromCrossClusterCancelExecutionResponseAttributes converts internal CrossClusterCancelExecutionResponseAttributes type to thrift\nfunc FromCrossClusterCancelExecutionResponseAttributes(t *types.CrossClusterCancelExecutionResponseAttributes) *shared.CrossClusterCancelExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterCancelExecutionResponseAttributes{}\n}\n\n// ToCrossClusterCancelExecutionResponseAttributes converts thrift CrossClusterCancelExecutionResponseAttributes type to internal\nfunc ToCrossClusterCancelExecutionResponseAttributes(t *shared.CrossClusterCancelExecutionResponseAttributes) *types.CrossClusterCancelExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterCancelExecutionResponseAttributes{}\n}\n\n// FromCrossClusterSignalExecutionRequestAttributes converts internal CrossClusterSignalExecutionRequestAttributes type to thrift\nfunc FromCrossClusterSignalExecutionRequestAttributes(t *types.CrossClusterSignalExecutionRequestAttributes) *shared.CrossClusterSignalExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterSignalExecutionRequestAttributes{\n\t\tTargetDomainID:    &t.TargetDomainID,\n\t\tTargetWorkflowID:  &t.TargetWorkflowID,\n\t\tTargetRunID:       &t.TargetRunID,\n\t\tRequestID:         &t.RequestID,\n\t\tInitiatedEventID:  &t.InitiatedEventID,\n\t\tChildWorkflowOnly: &t.ChildWorkflowOnly,\n\t\tSignalName:        &t.SignalName,\n\t\tSignalInput:       t.SignalInput,\n\t\tControl:           t.Control,\n\t}\n}\n\n// ToCrossClusterSignalExecutionRequestAttributes converts thrift CrossClusterSignalExecutionRequestAttributes type to internal\nfunc ToCrossClusterSignalExecutionRequestAttributes(t *shared.CrossClusterSignalExecutionRequestAttributes) *types.CrossClusterSignalExecutionRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterSignalExecutionRequestAttributes{\n\t\tTargetDomainID:    t.GetTargetDomainID(),\n\t\tTargetWorkflowID:  t.GetTargetWorkflowID(),\n\t\tTargetRunID:       t.GetTargetRunID(),\n\t\tRequestID:         t.GetRequestID(),\n\t\tInitiatedEventID:  t.GetInitiatedEventID(),\n\t\tChildWorkflowOnly: t.GetChildWorkflowOnly(),\n\t\tSignalName:        t.GetSignalName(),\n\t\tSignalInput:       t.SignalInput,\n\t\tControl:           t.Control,\n\t}\n}\n\n// FromCrossClusterSignalExecutionResponseAttributes converts internal CrossClusterSignalExecutionResponseAttributes type to thrift\nfunc FromCrossClusterSignalExecutionResponseAttributes(t *types.CrossClusterSignalExecutionResponseAttributes) *shared.CrossClusterSignalExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterSignalExecutionResponseAttributes{}\n}\n\n// ToCrossClusterSignalExecutionResponseAttributes converts thrift CrossClusterSignalExecutionResponseAttributes type to internal\nfunc ToCrossClusterSignalExecutionResponseAttributes(t *shared.CrossClusterSignalExecutionResponseAttributes) *types.CrossClusterSignalExecutionResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterSignalExecutionResponseAttributes{}\n}\n\n// FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes converts internal\n// CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes type to thrift\nfunc FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(\n\tt *types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes,\n) *shared.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\tTargetDomainID:   &t.TargetDomainID,\n\t\tTargetWorkflowID: &t.TargetWorkflowID,\n\t\tTargetRunID:      &t.TargetRunID,\n\t\tInitiatedEventID: &t.InitiatedEventID,\n\t\tCompletionEvent:  FromHistoryEvent(t.CompletionEvent),\n\t}\n}\n\n// ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes converts thrift\n// CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes type to internal\nfunc ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(\n\tt *shared.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes,\n) *types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\tTargetDomainID:   t.GetTargetDomainID(),\n\t\tTargetWorkflowID: t.GetTargetWorkflowID(),\n\t\tTargetRunID:      t.GetTargetRunID(),\n\t\tInitiatedEventID: t.GetInitiatedEventID(),\n\t\tCompletionEvent:  ToHistoryEvent(t.CompletionEvent),\n\t}\n}\n\n// FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes converts internal\n// CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes type to thrift\nfunc FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(\n\tt *types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes,\n) *shared.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{}\n}\n\n// ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes converts thrift\n// CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes type to internal\nfunc ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(\n\tt *shared.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes,\n) *types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{}\n}\n\n// FromApplyParentClosePolicyAttributes converts internal\n// ApplyParentClosePolicyAttributes types to thrift\nfunc FromApplyParentClosePolicyAttributes(\n\tt *types.ApplyParentClosePolicyAttributes,\n) *shared.ApplyParentClosePolicyAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ApplyParentClosePolicyAttributes{\n\t\tChildDomainID:     &t.ChildDomainID,\n\t\tChildWorkflowID:   &t.ChildWorkflowID,\n\t\tChildRunID:        &t.ChildRunID,\n\t\tParentClosePolicy: FromParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// ToApplyParentClosePolicyAttributes converts thrift\n// ApplyParentClosePolicyAttributes types to internal\nfunc ToApplyParentClosePolicyAttributes(\n\tt *shared.ApplyParentClosePolicyAttributes,\n) *types.ApplyParentClosePolicyAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyAttributes{\n\t\tChildDomainID:     t.GetChildDomainID(),\n\t\tChildWorkflowID:   t.GetChildWorkflowID(),\n\t\tChildRunID:        t.GetChildRunID(),\n\t\tParentClosePolicy: ToParentClosePolicy(t.ParentClosePolicy),\n\t}\n}\n\n// FromApplyParentClosePolicyStatus converts thrift\n// ApplyParentClosePolicyStatus types to internal\nfunc FromApplyParentClosePolicyStatus(\n\tt *types.ApplyParentClosePolicyStatus,\n) *shared.ApplyParentClosePolicyStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ApplyParentClosePolicyStatus{\n\t\tCompleted:   &t.Completed,\n\t\tFailedCause: FromCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// ToApplyParentClosePolicyStatus converts thrift\n// ApplyParentClosePolicyStatus types to internal\nfunc ToApplyParentClosePolicyStatus(\n\tt *shared.ApplyParentClosePolicyStatus,\n) *types.ApplyParentClosePolicyStatus {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyStatus{\n\t\tCompleted:   t.GetCompleted(),\n\t\tFailedCause: ToCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// FromApplyParentClosePolicyRequest converts thrift\n// ApplyParentClosePolicyRequest types to internal\nfunc FromApplyParentClosePolicyRequest(\n\tt *types.ApplyParentClosePolicyRequest,\n) *shared.ApplyParentClosePolicyRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ApplyParentClosePolicyRequest{\n\t\tChild:  FromApplyParentClosePolicyAttributes(t.Child),\n\t\tStatus: FromApplyParentClosePolicyStatus(t.Status),\n\t}\n}\n\n// ToApplyParentClosePolicyRequest converts thrift\n// ApplyParentClosePolicyRequest types to internal\nfunc ToApplyParentClosePolicyRequest(\n\tt *shared.ApplyParentClosePolicyRequest,\n) *types.ApplyParentClosePolicyRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyRequest{\n\t\tChild:  ToApplyParentClosePolicyAttributes(t.Child),\n\t\tStatus: ToApplyParentClosePolicyStatus(t.Status),\n\t}\n}\n\n// FromApplyParentClosePolicyRequestArray converts thrift\n// ApplyParentClosePolicyRequestArray types to internal\nfunc FromApplyParentClosePolicyRequestArray(\n\tt []*types.ApplyParentClosePolicyRequest,\n) []*shared.ApplyParentClosePolicyRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.ApplyParentClosePolicyRequest, len(t))\n\tfor i := range t {\n\t\tv[i] = FromApplyParentClosePolicyRequest(t[i])\n\t}\n\treturn v\n}\n\n// ToApplyParentClosePolicyRequestArray converts internal\n// ApplyParentClosePolicyRequestArray types to thrift\nfunc ToApplyParentClosePolicyRequestArray(\n\tt []*shared.ApplyParentClosePolicyRequest,\n) []*types.ApplyParentClosePolicyRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ApplyParentClosePolicyRequest, len(t))\n\tfor i := range t {\n\t\tv[i] = ToApplyParentClosePolicyRequest(t[i])\n\t}\n\treturn v\n}\n\n// FromApplyParentClosePolicyResult converts thrift\n// ApplyParentClosePolicyResult types to internal\nfunc FromApplyParentClosePolicyResult(\n\tt *types.ApplyParentClosePolicyResult,\n) *shared.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.ApplyParentClosePolicyResult{\n\t\tChild:       FromApplyParentClosePolicyAttributes(t.Child),\n\t\tFailedCause: FromCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// ToApplyParentClosePolicyResult converts thrift\n// ApplyParentClosePolicyResult types to internal\nfunc ToApplyParentClosePolicyResult(\n\tt *shared.ApplyParentClosePolicyResult,\n) *types.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.ApplyParentClosePolicyResult{\n\t\tChild:       ToApplyParentClosePolicyAttributes(t.Child),\n\t\tFailedCause: ToCrossClusterTaskFailedCause(t.FailedCause),\n\t}\n}\n\n// FromApplyParentClosePolicyResultArray converts internal\n// ApplyParentClosePolicyResultArray types to internal\nfunc FromApplyParentClosePolicyResultArray(\n\tt []*types.ApplyParentClosePolicyResult,\n) []*shared.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.ApplyParentClosePolicyResult, len(t))\n\tfor i := range t {\n\t\tv[i] = FromApplyParentClosePolicyResult(t[i])\n\t}\n\treturn v\n}\n\n// ToApplyParentClosePolicyResultArray converts internal\n// ApplyParentClosePolicyResultArray types to thrift\nfunc ToApplyParentClosePolicyResultArray(\n\tt []*shared.ApplyParentClosePolicyResult,\n) []*types.ApplyParentClosePolicyResult {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.ApplyParentClosePolicyResult, len(t))\n\tfor i := range t {\n\t\tv[i] = ToApplyParentClosePolicyResult(t[i])\n\t}\n\treturn v\n}\n\n// FromCrossClusterApplyParentClosePolicyRequestAttributes converts internal\n// CrossClusterApplyParentClosePolicyRequestAttributes types to thrift\nfunc FromCrossClusterApplyParentClosePolicyRequestAttributes(\n\tt *types.CrossClusterApplyParentClosePolicyRequestAttributes,\n) *shared.CrossClusterApplyParentClosePolicyRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterApplyParentClosePolicyRequestAttributes{\n\t\tChildren: FromApplyParentClosePolicyRequestArray(t.Children),\n\t}\n}\n\n// ToCrossClusterApplyParentClosePolicyRequestAttributes converts thrift\n// CrossClusterApplyParentClosePolicyRequestAttributes types to internal\nfunc ToCrossClusterApplyParentClosePolicyRequestAttributes(\n\tt *shared.CrossClusterApplyParentClosePolicyRequestAttributes,\n) *types.CrossClusterApplyParentClosePolicyRequestAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterApplyParentClosePolicyRequestAttributes{\n\t\tChildren: ToApplyParentClosePolicyRequestArray(t.Children),\n\t}\n}\n\n// FromCrossClusterApplyParentClosePolicyResponseAttributes converts internal\n// CrossClusterApplyParentClosePolicyResponseAttributes type to thrift\nfunc FromCrossClusterApplyParentClosePolicyResponseAttributes(\n\tt *types.CrossClusterApplyParentClosePolicyResponseAttributes,\n) *shared.CrossClusterApplyParentClosePolicyResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterApplyParentClosePolicyResponseAttributes{\n\t\tChildrenStatus: FromApplyParentClosePolicyResultArray(t.ChildrenStatus),\n\t}\n}\n\n// ToCrossClusterApplyParentClosePolicyResponseAttributes converts thrift\n// CrossClusterApplyParentClosePolicyResponseAttributes type to internal\nfunc ToCrossClusterApplyParentClosePolicyResponseAttributes(\n\tt *shared.CrossClusterApplyParentClosePolicyResponseAttributes,\n) *types.CrossClusterApplyParentClosePolicyResponseAttributes {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterApplyParentClosePolicyResponseAttributes{\n\t\tChildrenStatus: ToApplyParentClosePolicyResultArray(t.ChildrenStatus),\n\t}\n}\n\n// FromCrossClusterTaskRequest converts internal CrossClusterTaskRequest type to thrift\nfunc FromCrossClusterTaskRequest(t *types.CrossClusterTaskRequest) *shared.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterTaskRequest{\n\t\tTaskInfo:                                       FromCrossClusterTaskInfo(t.TaskInfo),\n\t\tStartChildExecutionAttributes:                  FromCrossClusterStartChildExecutionRequestAttributes(t.StartChildExecutionAttributes),\n\t\tCancelExecutionAttributes:                      FromCrossClusterCancelExecutionRequestAttributes(t.CancelExecutionAttributes),\n\t\tSignalExecutionAttributes:                      FromCrossClusterSignalExecutionRequestAttributes(t.SignalExecutionAttributes),\n\t\tRecordChildWorkflowExecutionCompleteAttributes: FromCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\tApplyParentClosePolicyAttributes:               FromCrossClusterApplyParentClosePolicyRequestAttributes(t.ApplyParentClosePolicyAttributes),\n\t}\n}\n\n// ToCrossClusterTaskRequest converts thrift CrossClusterTaskRequest type to internal\nfunc ToCrossClusterTaskRequest(t *shared.CrossClusterTaskRequest) *types.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterTaskRequest{\n\t\tTaskInfo:                                       ToCrossClusterTaskInfo(t.TaskInfo),\n\t\tStartChildExecutionAttributes:                  ToCrossClusterStartChildExecutionRequestAttributes(t.StartChildExecutionAttributes),\n\t\tCancelExecutionAttributes:                      ToCrossClusterCancelExecutionRequestAttributes(t.CancelExecutionAttributes),\n\t\tSignalExecutionAttributes:                      ToCrossClusterSignalExecutionRequestAttributes(t.SignalExecutionAttributes),\n\t\tRecordChildWorkflowExecutionCompleteAttributes: ToCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\tApplyParentClosePolicyAttributes:               ToCrossClusterApplyParentClosePolicyRequestAttributes(t.ApplyParentClosePolicyAttributes),\n\t}\n}\n\n// FromCrossClusterTaskResponse converts internal CrossClusterTaskResponse type to thrift\nfunc FromCrossClusterTaskResponse(t *types.CrossClusterTaskResponse) *shared.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.CrossClusterTaskResponse{\n\t\tTaskID:                        &t.TaskID,\n\t\tTaskType:                      FromCrossClusterTaskType(t.TaskType),\n\t\tTaskState:                     &t.TaskState,\n\t\tFailedCause:                   FromCrossClusterTaskFailedCause(t.FailedCause),\n\t\tStartChildExecutionAttributes: FromCrossClusterStartChildExecutionResponseAttributes(t.StartChildExecutionAttributes),\n\t\tCancelExecutionAttributes:     FromCrossClusterCancelExecutionResponseAttributes(t.CancelExecutionAttributes),\n\t\tSignalExecutionAttributes:     FromCrossClusterSignalExecutionResponseAttributes(t.SignalExecutionAttributes),\n\t\tRecordChildWorkflowExecutionCompleteAttributes: FromCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\tApplyParentClosePolicyAttributes:               FromCrossClusterApplyParentClosePolicyResponseAttributes(t.ApplyParentClosePolicyAttributes),\n\t}\n}\n\n// ToCrossClusterTaskResponse converts thrift CrossClusterTaskResponse type to internal\nfunc ToCrossClusterTaskResponse(t *shared.CrossClusterTaskResponse) *types.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.CrossClusterTaskResponse{\n\t\tTaskID:                        t.GetTaskID(),\n\t\tTaskType:                      ToCrossClusterTaskType(t.TaskType),\n\t\tTaskState:                     t.GetTaskState(),\n\t\tFailedCause:                   ToCrossClusterTaskFailedCause(t.FailedCause),\n\t\tStartChildExecutionAttributes: ToCrossClusterStartChildExecutionResponseAttributes(t.StartChildExecutionAttributes),\n\t\tCancelExecutionAttributes:     ToCrossClusterCancelExecutionResponseAttributes(t.CancelExecutionAttributes),\n\t\tSignalExecutionAttributes:     ToCrossClusterSignalExecutionResponseAttributes(t.SignalExecutionAttributes),\n\t\tRecordChildWorkflowExecutionCompleteAttributes: ToCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes(t.RecordChildWorkflowExecutionCompleteAttributes),\n\t\tApplyParentClosePolicyAttributes:               ToCrossClusterApplyParentClosePolicyResponseAttributes(t.ApplyParentClosePolicyAttributes),\n\t}\n}\n\n// FromAdminGetCrossClusterTasksRequest converts internal GetCrossClusterTasksRequest type to thrift\nfunc FromAdminGetCrossClusterTasksRequest(t *types.GetCrossClusterTasksRequest) *shared.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetCrossClusterTasksRequest{\n\t\tShardIDs:      t.ShardIDs,\n\t\tTargetCluster: &t.TargetCluster,\n\t}\n}\n\n// ToAdminGetCrossClusterTasksRequest converts thrift GetCrossClusterTasksRequest type to internal\nfunc ToAdminGetCrossClusterTasksRequest(t *shared.GetCrossClusterTasksRequest) *types.GetCrossClusterTasksRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksRequest{\n\t\tShardIDs:      t.ShardIDs,\n\t\tTargetCluster: t.GetTargetCluster(),\n\t}\n}\n\n// FromCrossClusterTaskRequestArray converts internal CrossClusterTaskRequest type array to thrift\nfunc FromCrossClusterTaskRequestArray(t []*types.CrossClusterTaskRequest) []*shared.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.CrossClusterTaskRequest, len(t))\n\tfor i := range t {\n\t\tv[i] = FromCrossClusterTaskRequest(t[i])\n\t}\n\treturn v\n}\n\n// ToCrossClusterTaskRequestArray converts thrift CrossClusterTaskRequest type array to internal\nfunc ToCrossClusterTaskRequestArray(t []*shared.CrossClusterTaskRequest) []*types.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.CrossClusterTaskRequest, len(t))\n\tfor i := range t {\n\t\tv[i] = ToCrossClusterTaskRequest(t[i])\n\t}\n\treturn v\n}\n\n// FromCrossClusterTaskRequestMap converts internal CrossClusterTaskRequest type map to thrift\nfunc FromCrossClusterTaskRequestMap(t map[int32][]*types.CrossClusterTaskRequest) map[int32][]*shared.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32][]*shared.CrossClusterTaskRequest)\n\tfor key := range t {\n\t\tv[key] = FromCrossClusterTaskRequestArray(t[key])\n\t}\n\treturn v\n}\n\n// ToCrossClusterTaskRequestMap converts thrift CrossClusterTaskRequest type map to internal\nfunc ToCrossClusterTaskRequestMap(t map[int32][]*shared.CrossClusterTaskRequest) map[int32][]*types.CrossClusterTaskRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32][]*types.CrossClusterTaskRequest)\n\tfor key := range t {\n\t\tv[key] = ToCrossClusterTaskRequestArray(t[key])\n\t}\n\treturn v\n}\n\n// FromGetTaskFailedCauseMap converts internal GetTaskFailedCause type map to thrift\nfunc FromGetTaskFailedCauseMap(t map[int32]types.GetTaskFailedCause) map[int32]shared.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]shared.GetTaskFailedCause)\n\tfor key, value := range t {\n\t\tv[key] = *FromGetTaskFailedCause(&value)\n\t}\n\treturn v\n}\n\n// ToGetTaskFailedCauseMap converts thrift GetTaskFailedCause type map to internal\nfunc ToGetTaskFailedCauseMap(t map[int32]shared.GetTaskFailedCause) map[int32]types.GetTaskFailedCause {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int32]types.GetTaskFailedCause)\n\tfor key, value := range t {\n\t\tv[key] = *ToGetCrossClusterTaskFailedCause(&value)\n\t}\n\treturn v\n}\n\n// FromAdminGetCrossClusterTasksResponse converts internal GetCrossClusterTasksResponse type to thrift\nfunc FromAdminGetCrossClusterTasksResponse(t *types.GetCrossClusterTasksResponse) *shared.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       FromCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: FromGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// ToAdminGetCrossClusterTasksResponse converts thrift GetCrossClusterTasksResponse type to internal\nfunc ToAdminGetCrossClusterTasksResponse(t *shared.GetCrossClusterTasksResponse) *types.GetCrossClusterTasksResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       ToCrossClusterTaskRequestMap(t.TasksByShard),\n\t\tFailedCauseByShard: ToGetTaskFailedCauseMap(t.FailedCauseByShard),\n\t}\n}\n\n// FromCrossClusterTaskResponseArray converts internal CrossClusterTaskResponse type array to thrift\nfunc FromCrossClusterTaskResponseArray(t []*types.CrossClusterTaskResponse) []*shared.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.CrossClusterTaskResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = FromCrossClusterTaskResponse(t[i])\n\t}\n\treturn v\n}\n\n// ToCrossClusterTaskResponseArray converts thrift CrossClusterTaskResponse type array to internal\nfunc ToCrossClusterTaskResponseArray(t []*shared.CrossClusterTaskResponse) []*types.CrossClusterTaskResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.CrossClusterTaskResponse, len(t))\n\tfor i := range t {\n\t\tv[i] = ToCrossClusterTaskResponse(t[i])\n\t}\n\treturn v\n}\n\n// FromAdminRespondCrossClusterTasksCompletedRequest converts internal RespondCrossClusterTasksCompletedRequest type to thrift\nfunc FromAdminRespondCrossClusterTasksCompletedRequest(t *types.RespondCrossClusterTasksCompletedRequest) *shared.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondCrossClusterTasksCompletedRequest{\n\t\tShardID:       &t.ShardID,\n\t\tTargetCluster: &t.TargetCluster,\n\t\tTaskResponses: FromCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: &t.FetchNewTasks,\n\t}\n}\n\n// ToAdminRespondCrossClusterTasksCompletedRequest converts thrift RespondCrossClusterTasksCompletedRequest type to internal\nfunc ToAdminRespondCrossClusterTasksCompletedRequest(t *shared.RespondCrossClusterTasksCompletedRequest) *types.RespondCrossClusterTasksCompletedRequest {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedRequest{\n\t\tShardID:       t.GetShardID(),\n\t\tTargetCluster: t.GetTargetCluster(),\n\t\tTaskResponses: ToCrossClusterTaskResponseArray(t.TaskResponses),\n\t\tFetchNewTasks: t.GetFetchNewTasks(),\n\t}\n}\n\n// FromAdminRespondCrossClusterTasksCompletedResponse converts internal RespondCrossClusterTasksCompletedResponse type to thrift\nfunc FromAdminRespondCrossClusterTasksCompletedResponse(t *types.RespondCrossClusterTasksCompletedResponse) *shared.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: FromCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\n// ToAdminRespondCrossClusterTasksCompletedResponse converts thrift RespondCrossClusterTasksCompletedResponse type to internal\nfunc ToAdminRespondCrossClusterTasksCompletedResponse(t *shared.RespondCrossClusterTasksCompletedResponse) *types.RespondCrossClusterTasksCompletedResponse {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: ToCrossClusterTaskRequestArray(t.Tasks),\n\t}\n}\n\n// FromStickyWorkerUnavailableError converts internal StickyWorkerUnavailableError type to thrift\nfunc FromStickyWorkerUnavailableError(t *types.StickyWorkerUnavailableError) *shared.StickyWorkerUnavailableError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.StickyWorkerUnavailableError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// ToStickyWorkerUnavailableError converts thrift StickyWorkerUnavailableError type to internal\nfunc ToStickyWorkerUnavailableError(t *shared.StickyWorkerUnavailableError) *types.StickyWorkerUnavailableError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.StickyWorkerUnavailableError{\n\t\tMessage: t.Message,\n\t}\n}\n\n// FromTaskListNotOwnedByHostError converts internal TaskListNotOwnedByHostError type to thrift\nfunc FromTaskListNotOwnedByHostError(t *cadence_errors.TaskListNotOwnedByHostError) *shared.TaskListNotOwnedByHostError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: t.OwnedByIdentity,\n\t\tMyIdentity:      t.MyIdentity,\n\t\tTasklistName:    t.TasklistName,\n\t}\n}\n\n// ToTaskListNotOwnedByHostError converts thrift TaskListNotOwnedByHostError type to internal\nfunc ToTaskListNotOwnedByHostError(t *shared.TaskListNotOwnedByHostError) *cadence_errors.TaskListNotOwnedByHostError {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &cadence_errors.TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: t.OwnedByIdentity,\n\t\tMyIdentity:      t.MyIdentity,\n\t\tTasklistName:    t.TasklistName,\n\t}\n}\n\n// ToAny converts thrift Any type to internal\nfunc ToAny(t *shared.Any) *types.Any {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.Any{\n\t\tValueType: t.GetValueType(),\n\t\tValue:     t.Value,\n\t}\n}\n\n// FromAny converts internal Any type to thrift\nfunc FromAny(t *types.Any) *shared.Any {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tdup := t.ValueType\n\treturn &shared.Any{\n\t\tValueType: &dup,\n\t\tValue:     t.Value,\n\t}\n}\n\n// FromWorkflowQuery converts internal WorkflowQuery type to thrift\nfunc FromAutoConfigHint(t *types.AutoConfigHint) *shared.AutoConfigHint {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.AutoConfigHint{\n\t\tPollerWaitTimeInMs: &t.PollerWaitTimeInMs,\n\t\tEnableAutoConfig:   &t.EnableAutoConfig,\n\t}\n}\n\n// ToWorkflowQuery converts thrift WorkflowQuery type to internal\nfunc ToAutoConfigHint(t *shared.AutoConfigHint) *types.AutoConfigHint {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.AutoConfigHint{\n\t\tPollerWaitTimeInMs: *t.PollerWaitTimeInMs,\n\t\tEnableAutoConfig:   *t.EnableAutoConfig,\n\t}\n}\n\nfunc FromTaskKey(t *types.TaskKey) *shared.TaskKey {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskKey{\n\t\tTaskID:            common.Ptr(t.TaskID),\n\t\tScheduledTimeNano: common.Ptr(t.ScheduledTimeNano),\n\t}\n}\n\nfunc ToTaskKey(t *shared.TaskKey) *types.TaskKey {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskKey{\n\t\tTaskID:            t.GetTaskID(),\n\t\tScheduledTimeNano: t.GetScheduledTimeNano(),\n\t}\n}\n\nfunc FromTaskRange(t *types.TaskRange) *shared.TaskRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.TaskRange{\n\t\tInclusiveMin: FromTaskKey(t.InclusiveMin),\n\t\tExclusiveMax: FromTaskKey(t.ExclusiveMax),\n\t}\n}\n\nfunc ToTaskRange(t *shared.TaskRange) *types.TaskRange {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.TaskRange{\n\t\tInclusiveMin: ToTaskKey(t.InclusiveMin),\n\t\tExclusiveMax: ToTaskKey(t.ExclusiveMax),\n\t}\n}\n\nfunc FromVirtualSliceState(t *types.VirtualSliceState) *shared.VirtualSliceState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.VirtualSliceState{\n\t\tTaskRange: FromTaskRange(t.TaskRange),\n\t\tPredicate: FromPredicate(t.Predicate),\n\t}\n}\n\nfunc ToVirtualSliceState(t *shared.VirtualSliceState) *types.VirtualSliceState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VirtualSliceState{\n\t\tTaskRange: ToTaskRange(t.TaskRange),\n\t\tPredicate: ToPredicate(t.Predicate),\n\t}\n}\n\nfunc FromVirtualSliceStateArray(t []*types.VirtualSliceState) []*shared.VirtualSliceState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*shared.VirtualSliceState, len(t))\n\tfor i := range t {\n\t\tv[i] = FromVirtualSliceState(t[i])\n\t}\n\treturn v\n}\n\nfunc ToVirtualSliceStateArray(t []*shared.VirtualSliceState) []*types.VirtualSliceState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make([]*types.VirtualSliceState, len(t))\n\tfor i := range t {\n\t\tv[i] = ToVirtualSliceState(t[i])\n\t}\n\treturn v\n}\n\nfunc FromVirtualQueueState(t *types.VirtualQueueState) *shared.VirtualQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.VirtualQueueState{\n\t\tVirtualSliceStates: FromVirtualSliceStateArray(t.VirtualSliceStates),\n\t}\n}\n\nfunc ToVirtualQueueState(t *shared.VirtualQueueState) *types.VirtualQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.VirtualQueueState{\n\t\tVirtualSliceStates: ToVirtualSliceStateArray(t.VirtualSliceStates),\n\t}\n}\n\nfunc FromVirtualQueueStateMap(t map[int64]*types.VirtualQueueState) map[int64]*shared.VirtualQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int64]*shared.VirtualQueueState, len(t))\n\tfor key := range t {\n\t\tv[key] = FromVirtualQueueState(t[key])\n\t}\n\treturn v\n}\n\nfunc ToVirtualQueueStateMap(t map[int64]*shared.VirtualQueueState) map[int64]*types.VirtualQueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tv := make(map[int64]*types.VirtualQueueState, len(t))\n\tfor key := range t {\n\t\tv[key] = ToVirtualQueueState(t[key])\n\t}\n\treturn v\n}\n\nfunc FromQueueState(t *types.QueueState) *shared.QueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &shared.QueueState{\n\t\tVirtualQueueStates:    FromVirtualQueueStateMap(t.VirtualQueueStates),\n\t\tExclusiveMaxReadLevel: FromTaskKey(t.ExclusiveMaxReadLevel),\n\t}\n}\n\nfunc ToQueueState(t *shared.QueueState) *types.QueueState {\n\tif t == nil {\n\t\treturn nil\n\t}\n\treturn &types.QueueState{\n\t\tVirtualQueueStates:    ToVirtualQueueStateMap(t.VirtualQueueStates),\n\t\tExclusiveMaxReadLevel: ToTaskKey(t.ExclusiveMaxReadLevel),\n\t}\n}\n\nfunc FromCronOverlapPolicy(t *types.CronOverlapPolicy) *shared.CronOverlapPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase types.CronOverlapPolicyBufferOne:\n\t\treturn shared.CronOverlapPolicyBufferone.Ptr()\n\tcase types.CronOverlapPolicySkipped:\n\t\treturn shared.CronOverlapPolicySkipped.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n\nfunc ToCronOverlapPolicy(t *shared.CronOverlapPolicy) *types.CronOverlapPolicy {\n\tif t == nil {\n\t\treturn nil\n\t}\n\tswitch *t {\n\tcase shared.CronOverlapPolicyBufferone:\n\t\treturn types.CronOverlapPolicyBufferOne.Ptr()\n\tcase shared.CronOverlapPolicySkipped:\n\t\treturn types.CronOverlapPolicySkipped.Ptr()\n\t}\n\tpanic(\"unexpected enum value\")\n}\n"
  },
  {
    "path": "common/types/mapper/thrift/shared_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/google/go-cmp/cmp/cmpopts\"\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/testutils\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n)\n\nfunc TestCrossClusterTaskInfoConversion(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskInfo{nil, {}, &testdata.CrossClusterTaskInfo} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskInfo(FromCrossClusterTaskInfo(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequestConversion(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CrossClusterTaskRequestStartChildExecution,\n\t\t&testdata.CrossClusterTaskRequestCancelExecution,\n\t\t&testdata.CrossClusterTaskRequestSignalExecution,\n\t} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequest(FromCrossClusterTaskRequest(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskResponseConversion(t *testing.T) {\n\tfor _, item := range []*types.CrossClusterTaskResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CrossClusterTaskResponseStartChildExecution,\n\t\t&testdata.CrossClusterTaskResponseCancelExecution,\n\t\t&testdata.CrossClusterTaskResponseSignalExecution,\n\t} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskResponse(FromCrossClusterTaskResponse(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequestArrayConversion(t *testing.T) {\n\tfor _, item := range [][]*types.CrossClusterTaskRequest{nil, {}, testdata.CrossClusterTaskRequestArray} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequestArray(FromCrossClusterTaskRequestArray(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskResponseArrayConversion(t *testing.T) {\n\tfor _, item := range [][]*types.CrossClusterTaskResponse{nil, {}, testdata.CrossClusterTaskResponseArray} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskResponseArray(FromCrossClusterTaskResponseArray(item)))\n\t}\n}\n\nfunc TestCrossClusterTaskRequestMapConversion(t *testing.T) {\n\tfor _, item := range []map[int32][]*types.CrossClusterTaskRequest{nil, {}, testdata.CrossClusterTaskRequestMap} {\n\t\tassert.Equal(t, item, ToCrossClusterTaskRequestMap(FromCrossClusterTaskRequestMap(item)))\n\t}\n}\n\nfunc TestGetTaskFailedCauseMapConversion(t *testing.T) {\n\tfor _, item := range []map[int32]types.GetTaskFailedCause{nil, {}, testdata.GetCrossClusterTaskFailedCauseMap} {\n\t\tassert.Equal(t, item, ToGetTaskFailedCauseMap(FromGetTaskFailedCauseMap(item)))\n\t}\n}\n\nfunc TestGetCrossClusterTasksRequestConversion(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksRequest{nil, {}, &testdata.GetCrossClusterTasksRequest} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksRequest(FromAdminGetCrossClusterTasksRequest(item)))\n\t}\n}\n\nfunc TestGetCrossClusterTasksResponseConversion(t *testing.T) {\n\tfor _, item := range []*types.GetCrossClusterTasksResponse{nil, {}, &testdata.GetCrossClusterTasksResponse} {\n\t\tassert.Equal(t, item, ToAdminGetCrossClusterTasksResponse(FromAdminGetCrossClusterTasksResponse(item)))\n\t}\n}\n\nfunc TestRespondCrossClusterTasksCompletedRequestConversion(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedRequest{nil, {}, &testdata.RespondCrossClusterTasksCompletedRequest} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedRequest(FromAdminRespondCrossClusterTasksCompletedRequest(item)))\n\t}\n}\n\nfunc TestRespondCrossClusterTasksCompletedResponseConversion(t *testing.T) {\n\tfor _, item := range []*types.RespondCrossClusterTasksCompletedResponse{nil, {}, &testdata.RespondCrossClusterTasksCompletedResponse} {\n\t\tassert.Equal(t, item, ToAdminRespondCrossClusterTasksCompletedResponse(FromAdminRespondCrossClusterTasksCompletedResponse(item)))\n\t}\n}\n\nfunc TestGetFailoverInfoRequestConversion(t *testing.T) {\n\tfor _, item := range []*types.GetFailoverInfoRequest{nil, {}, &testdata.GetFailoverInfoRequest} {\n\t\tassert.Equal(t, item, ToGetFailoverInfoRequest(FromGetFailoverInfoRequest(item)))\n\t}\n}\n\nfunc TestGetFailoverInfoResponseConversion(t *testing.T) {\n\tfor _, item := range []*types.GetFailoverInfoResponse{nil, {}, &testdata.GetFailoverInfoResponse} {\n\t\tassert.Equal(t, item, ToGetFailoverInfoResponse(FromGetFailoverInfoResponse(item)))\n\t}\n}\n\nfunc TestCrossClusterApplyParentClosePolicyRequestAttributesConversion(t *testing.T) {\n\titem := testdata.CrossClusterApplyParentClosePolicyRequestAttributes\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToCrossClusterApplyParentClosePolicyRequestAttributes(\n\t\t\tFromCrossClusterApplyParentClosePolicyRequestAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestApplyParentClosePolicyAttributesConversion(t *testing.T) {\n\titem := testdata.ApplyParentClosePolicyAttributes\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToApplyParentClosePolicyAttributes(\n\t\t\tFromApplyParentClosePolicyAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestApplyParentClosePolicyResultConversion(t *testing.T) {\n\titem := testdata.ApplyParentClosePolicyResult\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToApplyParentClosePolicyResult(\n\t\t\tFromApplyParentClosePolicyResult(&item),\n\t\t),\n\t)\n}\n\nfunc TestCrossClusterApplyParentClosePolicyResponseConversion(t *testing.T) {\n\titem := testdata.CrossClusterApplyParentClosePolicyResponseWithChildren\n\tassert.Equal(\n\t\tt,\n\t\t&item,\n\t\tToCrossClusterApplyParentClosePolicyResponseAttributes(\n\t\t\tFromCrossClusterApplyParentClosePolicyResponseAttributes(&item),\n\t\t),\n\t)\n}\n\nfunc TestIsolationGroupToDomainBlobConversion(t *testing.T) {\n\tzone1 := \"zone-1\"\n\tzone2 := \"zone-2\"\n\tdrained := shared.IsolationGroupStateDrained\n\thealthy := shared.IsolationGroupStateHealthy\n\n\ttests := map[string]struct {\n\t\tin          *types.IsolationGroupConfiguration\n\t\texpectedOut *shared.IsolationGroupConfiguration\n\t}{\n\t\t\"valid input\": {\n\t\t\tin: &types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  zone1,\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  zone2,\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOut: &shared.IsolationGroupConfiguration{\n\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t{\n\t\t\t\t\t\tName:  &zone1,\n\t\t\t\t\t\tState: &drained,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tName:  &zone2,\n\t\t\t\t\t\tState: &healthy,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty input\": {\n\t\t\tin:          &types.IsolationGroupConfiguration{},\n\t\t\texpectedOut: &shared.IsolationGroupConfiguration{},\n\t\t},\n\t\t\"nil input\": {\n\t\t\tin:          nil,\n\t\t\texpectedOut: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tout := FromIsolationGroupConfig(td.in)\n\t\t\tassert.Equal(t, td.expectedOut, out)\n\t\t\troundTrip := ToIsolationGroupConfig(out)\n\t\t\tassert.Equal(t, td.in, roundTrip)\n\t\t})\n\t}\n}\n\nfunc TestIsolationGroupFromDomainBlobConversion(t *testing.T) {\n\tzone1 := \"zone-1\"\n\tzone2 := \"zone-2\"\n\tdrained := shared.IsolationGroupStateDrained\n\thealthy := shared.IsolationGroupStateHealthy\n\n\ttests := map[string]struct {\n\t\tin          *shared.IsolationGroupConfiguration\n\t\texpectedOut *types.IsolationGroupConfiguration\n\t}{\n\t\t\"valid input\": {\n\t\t\tin: &shared.IsolationGroupConfiguration{\n\t\t\t\tIsolationGroups: []*shared.IsolationGroupPartition{\n\t\t\t\t\t{\n\t\t\t\t\t\tName:  &zone1,\n\t\t\t\t\t\tState: &drained,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tName:  &zone2,\n\t\t\t\t\t\tState: &healthy,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOut: &types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  zone1,\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  zone2,\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"empty input\": {\n\t\t\tin:          &shared.IsolationGroupConfiguration{},\n\t\t\texpectedOut: &types.IsolationGroupConfiguration{},\n\t\t},\n\t\t\"nil input\": {\n\t\t\tin:          nil,\n\t\t\texpectedOut: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tout := ToIsolationGroupConfig(td.in)\n\t\t\tassert.Equal(t, td.expectedOut, out)\n\t\t\troundTrip := FromIsolationGroupConfig(out)\n\t\t\tassert.Equal(t, td.in, roundTrip)\n\t\t})\n\t}\n}\n\nfunc TestWorkflowExecutionInfoConversion(t *testing.T) {\n\tfor _, item := range []*types.WorkflowExecutionInfo{nil, {}, &testdata.WorkflowExecutionInfo, &testdata.CronWorkflowExecutionInfo, &testdata.WorkflowExecutionInfoEphemeral} {\n\t\tassert.Equal(t, item, ToWorkflowExecutionInfo(FromWorkflowExecutionInfo(item)))\n\t}\n}\n\nfunc TestWorkflowExecutionInfo_MigrateTaskList(t *testing.T) {\n\ttlName := \"foo\"\n\totherName := \"bar\"\n\tcases := []struct {\n\t\tname string\n\t\tin   *shared.WorkflowExecutionInfo\n\t\tout  *types.WorkflowExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"nil\",\n\t\t\tin:   &shared.WorkflowExecutionInfo{},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"name only\",\n\t\t\tin: &shared.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &tlName,\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"tl only\",\n\t\t\tin: &shared.WorkflowExecutionInfo{\n\t\t\t\tTaskListInfo: &shared.TaskList{Name: &tlName, Kind: shared.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"both\",\n\t\t\tin: &shared.WorkflowExecutionInfo{\n\t\t\t\tTaskList:     &otherName,\n\t\t\t\tTaskListInfo: &shared.TaskList{Name: &tlName, Kind: shared.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t\tout: &types.WorkflowExecutionInfo{\n\t\t\t\tTaskList: &types.TaskList{Name: tlName, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tassert.Equal(t, tc.out, ToWorkflowExecutionInfo(tc.in))\n\t\t})\n\t}\n}\n\nfunc TestActivityLocalDispatchInfoConversion(t *testing.T) {\n\ttestCases := []*types.ActivityLocalDispatchInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityLocalDispatchInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityLocalDispatchInfo(original)\n\t\troundTripObj := ToActivityLocalDispatchInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskCancelRequestedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskCancelRequestedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskCancelRequestedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskCancelRequestedEventAttributes(original)\n\t\troundTripObj := ToActivityTaskCancelRequestedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskCanceledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskCanceledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskCanceledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskCanceledEventAttributes(original)\n\t\troundTripObj := ToActivityTaskCanceledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestAccessDeniedErrorConversion(t *testing.T) {\n\ttestCases := []*types.AccessDeniedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.AccessDeniedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAccessDeniedError(original)\n\t\troundTripObj := ToAccessDeniedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskCompletedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskCompletedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskCompletedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskCompletedEventAttributes(original)\n\t\troundTripObj := ToActivityTaskCompletedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskFailedEventAttributes(original)\n\t\troundTripObj := ToActivityTaskFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskScheduledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskScheduledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskScheduledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskScheduledEventAttributes(original)\n\t\troundTripObj := ToActivityTaskScheduledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskStartedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskStartedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskStartedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskStartedEventAttributes(original)\n\t\troundTripObj := ToActivityTaskStartedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityTaskTimedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ActivityTaskTimedOutEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActivityTaskTimedOutEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityTaskTimedOutEventAttributes(original)\n\t\troundTripObj := ToActivityTaskTimedOutEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestArchivalStatusConversion(t *testing.T) {\n\tenabledStatus := types.ArchivalStatus(1)\n\tdisabledStatus := types.ArchivalStatus(0)\n\ttestCases := []*types.ArchivalStatus{\n\t\tnil,\n\t\t&enabledStatus,\n\t\t&disabledStatus,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromArchivalStatus(original)\n\t\troundTripObj := ToArchivalStatus(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestBadBinariesConversion(t *testing.T) {\n\ttestCases := []*types.BadBinaries{\n\t\tnil,\n\t\t{},\n\t\t&testdata.BadBinaries,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromBadBinaries(original)\n\t\troundTripObj := ToBadBinaries(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestBadBinaryInfoConversion(t *testing.T) {\n\ttestCases := []*types.BadBinaryInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.BadBinaryInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromBadBinaryInfo(original)\n\t\troundTripObj := ToBadBinaryInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestBadRequestErrorConversion(t *testing.T) {\n\ttestCases := []*types.BadRequestError{\n\t\tnil,\n\t\t{},\n\t\t{Message: \"Error message for bad request\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromBadRequestError(original)\n\t\troundTripObj := ToBadRequestError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCancelExternalWorkflowExecutionFailedCauseConversion(t *testing.T) {\n\ttestCases := []*types.CancelExternalWorkflowExecutionFailedCause{\n\t\tnil,\n\t\ttypes.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\ttypes.CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCancelExternalWorkflowExecutionFailedCause(original)\n\t\troundTripObj := ToCancelExternalWorkflowExecutionFailedCause(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCancelTimerDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.CancelTimerDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CancelTimerDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCancelTimerDecisionAttributes(original)\n\t\troundTripObj := ToCancelTimerDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCancelTimerFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.CancelTimerFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CancelTimerFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCancelTimerFailedEventAttributes(original)\n\t\troundTripObj := ToCancelTimerFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCancelWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.CancelWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CancelWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCancelWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToCancelWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCancellationAlreadyRequestedErrorConversion(t *testing.T) {\n\ttestCases := []*types.CancellationAlreadyRequestedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CancellationAlreadyRequestedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCancellationAlreadyRequestedError(original)\n\t\troundTripObj := ToCancellationAlreadyRequestedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionCanceledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionCanceledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionCanceledEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionCanceledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionCompletedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionCompletedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionCompletedEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionCompletedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionFailedEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionStartedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionStartedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionStartedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionStartedEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionStartedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionTerminatedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionTerminatedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionTerminatedEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionTerminatedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestChildWorkflowExecutionTimedOutEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ChildWorkflowExecutionTimedOutEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromChildWorkflowExecutionTimedOutEventAttributes(original)\n\t\troundTripObj := ToChildWorkflowExecutionTimedOutEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestClientVersionNotSupportedErrorConversion(t *testing.T) {\n\ttestCases := []*types.ClientVersionNotSupportedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ClientVersionNotSupportedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromClientVersionNotSupportedError(original)\n\t\troundTripObj := ToClientVersionNotSupportedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestFeatureNotEnabledErrorConversion(t *testing.T) {\n\ttestCases := []*types.FeatureNotEnabledError{\n\t\tnil,\n\t\t{},\n\t\t{FeatureFlag: \"test-feature-flag\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromFeatureNotEnabledError(original)\n\t\troundTripObj := ToFeatureNotEnabledError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCloseShardRequestConversion(t *testing.T) {\n\ttestCases := []*types.CloseShardRequest{\n\t\tnil,\n\t\t{},\n\t\t{ShardID: 5},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminCloseShardRequest(original)\n\t\troundTripObj := ToAdminCloseShardRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestClusterInfoConversion(t *testing.T) {\n\ttestCases := []*types.ClusterInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ClusterInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetClusterInfoResponse(original)\n\t\troundTripObj := ToGetClusterInfoResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestClusterReplicationConfigurationConversion(t *testing.T) {\n\ttestCases := []*types.ClusterReplicationConfiguration{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ClusterReplicationConfiguration,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromClusterReplicationConfiguration(original)\n\t\troundTripObj := ToClusterReplicationConfiguration(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCompleteWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CompleteWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCompleteWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToCompleteWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestContinueAsNewInitiatorConversion(t *testing.T) {\n\ttestCases := []*types.ContinueAsNewInitiator{\n\t\tnil,\n\t\ttypes.ContinueAsNewInitiatorDecider.Ptr(),\n\t\ttypes.ContinueAsNewInitiatorRetryPolicy.Ptr(),\n\t\ttypes.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromContinueAsNewInitiator(original)\n\t\troundTripObj := ToContinueAsNewInitiator(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestContinueAsNewWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ContinueAsNewWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromContinueAsNewWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToContinueAsNewWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCountWorkflowExecutionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.CountWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CountWorkflowExecutionsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCountWorkflowExecutionsRequest(original)\n\t\troundTripObj := ToCountWorkflowExecutionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCountWorkflowExecutionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.CountWorkflowExecutionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CountWorkflowExecutionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCountWorkflowExecutionsResponse(original)\n\t\troundTripObj := ToCountWorkflowExecutionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestCurrentBranchChangedErrorConversion(t *testing.T) {\n\ttestCases := []*types.CurrentBranchChangedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.CurrentBranchChangedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromCurrentBranchChangedError(original)\n\t\troundTripObj := ToCurrentBranchChangedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDataBlobConversion(t *testing.T) {\n\ttestCases := []*types.DataBlob{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DataBlob,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDataBlob(original)\n\t\troundTripObj := ToDataBlob(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionConverson(t *testing.T) {\n\ttestCases := []*types.Decision{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tDecisionType:                           types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\tScheduleActivityTaskDecisionAttributes: &testdata.ScheduleActivityTaskDecisionAttributes,\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecision(original)\n\t\troundTripObj := ToDecision(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskCompletedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskCompletedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DecisionTaskCompletedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskCompletedEventAttributes(original)\n\t\troundTripObj := ToDecisionTaskCompletedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskFailedCauseConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskFailedCause{\n\t\tnil,\n\t\ttypes.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadScheduleActivityAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelActivityAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadStartTimerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCancelTimerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRecordMarkerAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadContinueAsNewAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseStartTimerDuplicateID.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseResetStickyTasklist.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadStartChildExecutionAttributes.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseForceCloseDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseFailoverCloseDecision.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSignalInputSize.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseResetWorkflow.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadBinary.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseScheduleActivityDuplicateID.Ptr(),\n\t\ttypes.DecisionTaskFailedCauseBadSearchAttributes.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskFailedCause(original)\n\t\troundTripObj := ToDecisionTaskFailedCause(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DecisionTaskFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskFailedEventAttributes(original)\n\t\troundTripObj := ToDecisionTaskFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskScheduledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskScheduledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DecisionTaskScheduledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskScheduledEventAttributes(original)\n\t\troundTripObj := ToDecisionTaskScheduledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskStartedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskStartedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DecisionTaskStartedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskStartedEventAttributes(original)\n\t\troundTripObj := ToDecisionTaskStartedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskTimedOutCauseConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskTimedOutCause{\n\t\tnil,\n\t\ttypes.DecisionTaskTimedOutCauseTimeout.Ptr(),\n\t\ttypes.DecisionTaskTimedOutCauseReset.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskTimedOutCause(original)\n\t\troundTripObj := ToDecisionTaskTimedOutCause(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTaskTimedOutEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.DecisionTaskTimedOutEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DecisionTaskTimedOutEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionTaskTimedOutEventAttributes(original)\n\t\troundTripObj := ToDecisionTaskTimedOutEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecisionTypeConversion(t *testing.T) {\n\ttestCases := []*types.DecisionType{\n\t\tnil,\n\t\ttypes.DecisionTypeScheduleActivityTask.Ptr(),\n\t\ttypes.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\ttypes.DecisionTypeStartTimer.Ptr(),\n\t\ttypes.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeCancelTimer.Ptr(),\n\t\ttypes.DecisionTypeCancelWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeRequestCancelExternalWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeRecordMarker.Ptr(),\n\t\ttypes.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\ttypes.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDecisionType(original)\n\t\troundTripObj := ToDecisionType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDeleteDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.DeleteDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DeleteDomainRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDeleteDomainRequest(original)\n\t\troundTripObj := ToDeleteDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDeprecateDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.DeprecateDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DeprecateDomainRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDeprecateDomainRequest(original)\n\t\troundTripObj := ToDeprecateDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t{Name: common.StringPtr(\"test-name\"), UUID: common.StringPtr(\"test-uuid\")},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeDomainRequest(original)\n\t\troundTripObj := ToDescribeDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeDomainResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeDomainResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DescribeDomainResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeDomainResponse(original)\n\t\troundTripObj := ToDescribeDomainResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDecribeHistoryHostRequstConversion(t *testing.T) {\n\ttestCases := []*types.DescribeHistoryHostRequest{\n\t\tnil,\n\t\t{},\n\t\t{HostAddress: common.StringPtr(\"test-host-address\")},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeHistoryHostRequest(original)\n\t\troundTripObj := ToAdminDescribeHistoryHostRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeShardDistributionRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeShardDistributionRequest{\n\t\tnil,\n\t\t{},\n\t\t{PageSize: 100},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeShardDistributionRequest(original)\n\t\troundTripObj := ToAdminDescribeShardDistributionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeHistoryHostResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeHistoryHostResponse{\n\t\tnil,\n\t\t{},\n\t\t{NumberOfShards: 10, ShardIDs: []int32{1, 2, 3}, Address: \"test-address\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeHistoryHostResponse(original)\n\t\troundTripObj := ToAdminDescribeHistoryHostResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeShardDistributionResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeShardDistributionResponse{\n\t\tnil,\n\t\t{},\n\t\t{NumberOfShards: 10, Shards: map[int32]string{1: \"test-host-1\", 2: \"test-host-2\"}},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeShardDistributionResponse(original)\n\t\troundTripObj := ToAdminDescribeShardDistributionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeQueueRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeQueueRequest{\n\t\tnil,\n\t\t{},\n\t\t{ShardID: 10},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeQueueRequest(original)\n\t\troundTripObj := ToAdminDescribeQueueRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeQueueResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeQueueResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.HistoryDescribeQueueResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminDescribeQueueResponse(original)\n\t\troundTripObj := ToAdminDescribeQueueResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeTaskListRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeTaskListRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DescribeTaskListRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeTaskListRequest(original)\n\t\troundTripObj := ToDescribeTaskListRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeTaskListResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeTaskListResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DescribeTaskListResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeTaskListResponse(original)\n\t\troundTripObj := ToDescribeTaskListResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.DescribeTaskListResponse{}, \"PartitionConfig\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestDescribeWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.DescribeWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DescribeWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeWorkflowExecutionRequest(original)\n\t\troundTripObj := ToDescribeWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeWorkflowExecutionResponseConversion(t *testing.T) {\n\ttestCases := []*types.DescribeWorkflowExecutionResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DescribeWorkflowExecutionResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeWorkflowExecutionResponse(original)\n\t\troundTripObj := ToDescribeWorkflowExecutionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDiagnoseWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.DiagnoseWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DiagnoseWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDiagnoseWorkflowExecutionRequest(original)\n\t\troundTripObj := ToDiagnoseWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDiagnoseWorkflowExecutionResponseConversion(t *testing.T) {\n\ttestCases := []*types.DiagnoseWorkflowExecutionResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DiagnoseWorkflowExecutionResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDiagnoseWorkflowExecutionResponse(original)\n\t\troundTripObj := ToDiagnoseWorkflowExecutionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainAlreadyExistsErrorConversion(t *testing.T) {\n\ttestCases := []*types.DomainAlreadyExistsError{\n\t\tnil,\n\t\t{},\n\t\t{Message: \"test-message\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainAlreadyExistsError(original)\n\t\troundTripObj := ToDomainAlreadyExistsError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainCacheInfoConversion(t *testing.T) {\n\ttestCases := []*types.DomainCacheInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DomainCacheInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainCacheInfo(original)\n\t\troundTripObj := ToDomainCacheInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainConfigurationConversion(t *testing.T) {\n\ttestCases := []*types.DomainConfiguration{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DomainConfiguration,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainConfiguration(original)\n\t\troundTripObj := ToDomainConfiguration(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainInfoConversion(t *testing.T) {\n\ttestCases := []*types.DomainInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DomainInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainInfo(original)\n\t\troundTripObj := ToDomainInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestFailoverInfoConversion(t *testing.T) {\n\ttestCases := []*types.FailoverInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.FailoverInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromFailoverInfo(original)\n\t\troundTripObj := ToFailoverInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainNotActiveErrorConversion(t *testing.T) {\n\ttestCases := []*types.DomainNotActiveError{\n\t\tnil,\n\t\t{},\n\t\t{Message: \"test-message\"},\n\t\t{Message: \"test-message\", DomainName: \"test-domain\", CurrentCluster: \"test-current-cluster\", ActiveCluster: \"test-active-cluster\"},\n\t\t{Message: \"test-message\", DomainName: \"test-domain\", CurrentCluster: \"test-current-cluster\", ActiveClusters: []string{\"test-active-cluster-1\", \"test-active-cluster-2\"}},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainNotActiveError(original)\n\t\troundTripObj := ToDomainNotActiveError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainReplicationConfigurationConversion(t *testing.T) {\n\ttestCases := []*types.DomainReplicationConfiguration{\n\t\tnil,\n\t\t{},\n\t\t&testdata.DomainReplicationConfiguration,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainReplicationConfiguration(original)\n\t\troundTripObj := ToDomainReplicationConfiguration(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDomainStatusConversion(t *testing.T) {\n\ttestCases := []*types.DomainStatus{\n\t\tnil,\n\t\ttypes.DomainStatusRegistered.Ptr(),\n\t\ttypes.DomainStatusDeprecated.Ptr(),\n\t\ttypes.DomainStatusDeleted.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDomainStatus(original)\n\t\troundTripObj := ToDomainStatus(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestEncodingTypeConversion(t *testing.T) {\n\ttestCases := []*types.EncodingType{\n\t\tnil,\n\t\ttypes.EncodingTypeThriftRW.Ptr(),\n\t\ttypes.EncodingTypeJSON.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromEncodingType(original)\n\t\troundTripObj := ToEncodingType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestEntityNotExistsErrorConversion(t *testing.T) {\n\ttestCases := []*types.EntityNotExistsError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.EntityNotExistsError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromEntityNotExistsError(original)\n\t\troundTripObj := ToEntityNotExistsError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionAlreadyCompletedErrorConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionAlreadyCompletedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecutionAlreadyCompletedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionAlreadyCompletedError(original)\n\t\troundTripObj := ToWorkflowExecutionAlreadyCompletedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestEventTypeConversion(t *testing.T) {\n\ttestCases := []*types.EventType{\n\t\tnil,\n\t\ttypes.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionTimedOut.Ptr(),\n\t\ttypes.EventTypeDecisionTaskScheduled.Ptr(),\n\t\ttypes.EventTypeDecisionTaskStarted.Ptr(),\n\t\ttypes.EventTypeDecisionTaskCompleted.Ptr(),\n\t\ttypes.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\ttypes.EventTypeDecisionTaskFailed.Ptr(),\n\t\ttypes.EventTypeActivityTaskScheduled.Ptr(),\n\t\ttypes.EventTypeActivityTaskStarted.Ptr(),\n\t\ttypes.EventTypeActivityTaskCompleted.Ptr(),\n\t\ttypes.EventTypeActivityTaskFailed.Ptr(),\n\t\ttypes.EventTypeActivityTaskTimedOut.Ptr(),\n\t\ttypes.EventTypeActivityTaskCancelRequested.Ptr(),\n\t\ttypes.EventTypeActivityTaskCanceled.Ptr(),\n\t\ttypes.EventTypeTimerStarted.Ptr(),\n\t\ttypes.EventTypeTimerFired.Ptr(),\n\t\ttypes.EventTypeCancelTimerFailed.Ptr(),\n\t\ttypes.EventTypeTimerCanceled.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionCancelRequested.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionCanceled.Ptr(),\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr(),\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionFailed.Ptr(),\n\t\ttypes.EventTypeExternalWorkflowExecutionCancelRequested.Ptr(),\n\t\ttypes.EventTypeMarkerRecorded.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionTerminated.Ptr(),\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\ttypes.EventTypeStartChildWorkflowExecutionInitiated.Ptr(),\n\t\ttypes.EventTypeStartChildWorkflowExecutionFailed.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionCompleted.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionFailed.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionCanceled.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionTimedOut.Ptr(),\n\t\ttypes.EventTypeChildWorkflowExecutionTerminated.Ptr(),\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionInitiated.Ptr(),\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionFailed.Ptr(),\n\t\ttypes.EventTypeExternalWorkflowExecutionSignaled.Ptr(),\n\t\ttypes.EventTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromEventType(original)\n\t\troundTripObj := ToEventType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestExternalWorkflowExecutionSignaledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ExternalWorkflowExecutionSignaledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromExternalWorkflowExecutionSignaledEventAttributes(original)\n\t\troundTripObj := ToExternalWorkflowExecutionSignaledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestFailWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.FailWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.FailWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromFailWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToFailWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetSearchAttributesResponseConversion(t *testing.T) {\n\ttestCases := []*types.GetSearchAttributesResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.GetSearchAttributesResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetSearchAttributesResponse(original)\n\t\troundTripObj := ToGetSearchAttributesResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetWorkflowExecutionHistoryRequestConversion(t *testing.T) {\n\ttestCases := []*types.GetWorkflowExecutionHistoryRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.GetWorkflowExecutionHistoryRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetWorkflowExecutionHistoryRequest(original)\n\t\troundTripObj := ToGetWorkflowExecutionHistoryRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetWorkflowExecutionHistoryResponseConvesion(t *testing.T) {\n\ttestCases := []*types.GetWorkflowExecutionHistoryResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.GetWorkflowExecutionHistoryResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetWorkflowExecutionHistoryResponse(original)\n\t\troundTripObj := ToGetWorkflowExecutionHistoryResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestHeaderConversion(t *testing.T) {\n\ttestCases := []*types.Header{\n\t\tnil,\n\t\t{},\n\t\t&testdata.Header,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHeader(original)\n\t\troundTripObj := ToHeader(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryConversion(t *testing.T) {\n\ttestCases := []*types.History{\n\t\tnil,\n\t\t{},\n\t\t&testdata.History,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistory(original)\n\t\troundTripObj := ToHistory(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestHistoryBranchConversion(t *testing.T) {\n\ttestCases := []*types.HistoryBranch{\n\t\tnil,\n\t\t{},\n\t\t{TreeID: \"test-tree-id\", BranchID: \"test-branch-id\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryBranch(original)\n\t\troundTripObj := ToHistoryBranch(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryBranchRangeConversion(t *testing.T) {\n\ttestCases := []*types.HistoryBranchRange{\n\t\tnil,\n\t\t{},\n\t\t{BranchID: \"test-branch-id\", BeginNodeID: 5, EndNodeID: 10},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryBranchRange(original)\n\t\troundTripObj := ToHistoryBranchRange(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryEventConversion(t *testing.T) {\n\ttestCases := []*types.HistoryEvent{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tID:                                      5,\n\t\t\tTimestamp:                               common.Int64Ptr(10),\n\t\t\tEventType:                               types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\tWorkflowExecutionStartedEventAttributes: &testdata.WorkflowExecutionStartedEventAttributes,\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryEvent(original)\n\t\troundTripObj := ToHistoryEvent(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestHistoryEventFilterTypeConversion(t *testing.T) {\n\ttestCases := []*types.HistoryEventFilterType{\n\t\tnil,\n\t\ttypes.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\ttypes.HistoryEventFilterTypeCloseEvent.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryEventFilterType(original)\n\t\troundTripObj := ToHistoryEventFilterType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestIndexedValueTypeConversion(t *testing.T) {\n\ttestCases := []types.IndexedValueType{\n\t\ttypes.IndexedValueTypeString,\n\t\ttypes.IndexedValueTypeKeyword,\n\t\ttypes.IndexedValueTypeInt,\n\t\ttypes.IndexedValueTypeDouble,\n\t\ttypes.IndexedValueTypeBool,\n\t\ttypes.IndexedValueTypeDatetime,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromIndexedValueType(original)\n\t\troundTripObj := ToIndexedValueType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestInternalDataInconsistencyErrorConversion(t *testing.T) {\n\ttestCases := []*types.InternalDataInconsistencyError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.InternalDataInconsistencyError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromInternalDataInconsistencyError(original)\n\t\troundTripObj := ToInternalDataInconsistencyError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestInternalServiceErrorConversion(t *testing.T) {\n\ttestCases := []*types.InternalServiceError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.InternalServiceError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromInternalServiceError(original)\n\t\troundTripObj := ToInternalServiceError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestLimitExceededErrorConversion(t *testing.T) {\n\ttestCases := []*types.LimitExceededError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.LimitExceededError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromLimitExceededError(original)\n\t\troundTripObj := ToLimitExceededError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListArchivedWorkflowExecutionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListArchivedWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListArchivedWorkflowExecutionsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListArchivedWorkflowExecutionsRequest(original)\n\t\troundTripObj := ToListArchivedWorkflowExecutionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListArchivedWorkflowExecutionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListArchivedWorkflowExecutionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListArchivedWorkflowExecutionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListArchivedWorkflowExecutionsResponse(original)\n\t\troundTripObj := ToListArchivedWorkflowExecutionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListClosedWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t{Domain: \"test-domain\", MaximumPageSize: 100, NextPageToken: []byte(\"test-next-page-token\")},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListClosedWorkflowExecutionsRequest(original)\n\t\troundTripObj := ToListClosedWorkflowExecutionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListClosedWorkflowExecutionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListClosedWorkflowExecutionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListClosedWorkflowExecutionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListClosedWorkflowExecutionsResponse(original)\n\t\troundTripObj := ToListClosedWorkflowExecutionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListDomainsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListDomainsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListDomainsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListDomainsRequest(original)\n\t\troundTripObj := ToListDomainsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListDomainsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListDomainsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListDomainsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListDomainsResponse(original)\n\t\troundTripObj := ToListDomainsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListOpenWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t{Domain: \"test-domain\", MaximumPageSize: 100, NextPageToken: []byte(\"test-next-page-token\")},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListOpenWorkflowExecutionsRequest(original)\n\t\troundTripObj := ToListOpenWorkflowExecutionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListOpenWorkflowExecutionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListOpenWorkflowExecutionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListOpenWorkflowExecutionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListOpenWorkflowExecutionsResponse(original)\n\t\troundTripObj := ToListOpenWorkflowExecutionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListTaskListPartitionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListTaskListPartitionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListTaskListPartitionsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListTaskListPartitionsRequest(original)\n\t\troundTripObj := ToListTaskListPartitionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListTaskListPartitionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListTaskListPartitionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListTaskListPartitionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListTaskListPartitionsResponse(original)\n\t\troundTripObj := ToListTaskListPartitionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetTaskListsByDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.GetTaskListsByDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.MatchingGetTaskListsByDomainRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetTaskListsByDomainRequest(original)\n\t\troundTripObj := ToGetTaskListsByDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestGetTaskListsByDomainResponseConversion(t *testing.T) {\n\ttestCases := []*types.GetTaskListsByDomainResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.GetTaskListsByDomainResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromGetTaskListsByDomainResponse(original)\n\t\troundTripObj := ToGetTaskListsByDomainResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestDescribeTaskListResponseMapConversion(t *testing.T) {\n\ttestCases := []map[string]*types.DescribeTaskListResponse{\n\t\tnil,\n\t\t{},\n\t\t{\"test-key\": &testdata.DescribeTaskListResponse},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromDescribeTaskListResponseMap(original)\n\t\troundTripObj := ToDescribeTaskListResponseMap(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.DescribeTaskListResponse{}, \"PartitionConfig\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestListWorkflowExecutionsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ListWorkflowExecutionsRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListWorkflowExecutionsRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListWorkflowExecutionsRequest(original)\n\t\troundTripObj := ToListWorkflowExecutionsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListWorkflowExecutionsResponseConversion(t *testing.T) {\n\ttestCases := []*types.ListWorkflowExecutionsResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ListWorkflowExecutionsResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromListWorkflowExecutionsResponse(original)\n\t\troundTripObj := ToListWorkflowExecutionsResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestMarkerRecordedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.MarkerRecordedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.MarkerRecordedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromMarkerRecordedEventAttributes(original)\n\t\troundTripObj := ToMarkerRecordedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestMemoConversion(t *testing.T) {\n\ttestCases := []*types.Memo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.Memo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromMemo(original)\n\t\troundTripObj := ToMemo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestParentClosePolicyConversion(t *testing.T) {\n\ttestCases := []*types.ParentClosePolicy{\n\t\tnil,\n\t\ttypes.ParentClosePolicyAbandon.Ptr(),\n\t\ttypes.ParentClosePolicyRequestCancel.Ptr(),\n\t\ttypes.ParentClosePolicyTerminate.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromParentClosePolicy(original)\n\t\troundTripObj := ToParentClosePolicy(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPendingActivityInfoConversion(t *testing.T) {\n\ttestCases := []*types.PendingActivityInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PendingActivityInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPendingActivityInfo(original)\n\t\troundTripObj := ToPendingActivityInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPendingActivityStateConversion(t *testing.T) {\n\ttestCases := []*types.PendingActivityState{\n\t\tnil,\n\t\ttypes.PendingActivityStateScheduled.Ptr(),\n\t\ttypes.PendingActivityStateStarted.Ptr(),\n\t\ttypes.PendingActivityStateCancelRequested.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPendingActivityState(original)\n\t\troundTripObj := ToPendingActivityState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPendingChildExecutionInfoConversion(t *testing.T) {\n\ttestCases := []*types.PendingChildExecutionInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PendingChildExecutionInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPendingChildExecutionInfo(original)\n\t\troundTripObj := ToPendingChildExecutionInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPendingDecisionInfoConversion(t *testing.T) {\n\ttestCases := []*types.PendingDecisionInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PendingDecisionInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPendingDecisionInfo(original)\n\t\troundTripObj := ToPendingDecisionInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPendingDecisionStateConversion(t *testing.T) {\n\ttestCases := []*types.PendingDecisionState{\n\t\tnil,\n\t\ttypes.PendingDecisionStateScheduled.Ptr(),\n\t\ttypes.PendingDecisionStateStarted.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPendingDecisionState(original)\n\t\troundTripObj := ToPendingDecisionState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollForActivityTaskRequestConversion(t *testing.T) {\n\ttestCases := []*types.PollForActivityTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PollForActivityTaskRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPollForActivityTaskRequest(original)\n\t\troundTripObj := ToPollForActivityTaskRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollForActivityTaskResponseConversion(t *testing.T) {\n\ttestCases := []*types.PollForActivityTaskResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PollForActivityTaskResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPollForActivityTaskResponse(original)\n\t\troundTripObj := ToPollForActivityTaskResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollForDecisionTaskRequestConversion(t *testing.T) {\n\ttestCases := []*types.PollForDecisionTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PollForDecisionTaskRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPollForDecisionTaskRequest(original)\n\t\troundTripObj := ToPollForDecisionTaskRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestPollForDecisionTaskResponseConversion(t *testing.T) {\n\ttestCases := []*types.PollForDecisionTaskResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PollForDecisionTaskResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPollForDecisionTaskResponse(original)\n\t\troundTripObj := ToPollForDecisionTaskResponse(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestPollerInfoConversion(t *testing.T) {\n\ttestCases := []*types.PollerInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.PollerInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromPollerInfo(original)\n\t\troundTripObj := ToPollerInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryConsistencyLevelConversion(t *testing.T) {\n\ttestCases := []*types.QueryConsistencyLevel{\n\t\tnil,\n\t\ttypes.QueryConsistencyLevelEventual.Ptr(),\n\t\ttypes.QueryConsistencyLevelStrong.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryConsistencyLevel(original)\n\t\troundTripObj := ToQueryConsistencyLevel(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryFailedErrorConversion(t *testing.T) {\n\ttestCases := []*types.QueryFailedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.QueryFailedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryFailedError(original)\n\t\troundTripObj := ToQueryFailedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryRejectConditionConversion(t *testing.T) {\n\ttestCases := []*types.QueryRejectCondition{\n\t\tnil,\n\t\ttypes.QueryRejectConditionNotOpen.Ptr(),\n\t\ttypes.QueryRejectConditionNotCompletedCleanly.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryRejectCondition(original)\n\t\troundTripObj := ToQueryRejectCondition(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryRejectedConversion(t *testing.T) {\n\ttestCases := []*types.QueryRejected{\n\t\tnil,\n\t\t{},\n\t\t&testdata.QueryRejected,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryRejected(original)\n\t\troundTripObj := ToQueryRejected(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryResultTypeConversion(t *testing.T) {\n\ttestCases := []*types.QueryResultType{\n\t\tnil,\n\t\ttypes.QueryResultTypeAnswered.Ptr(),\n\t\ttypes.QueryResultTypeFailed.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryResultType(original)\n\t\troundTripObj := ToQueryResultType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryTaskCompletedTypeConversion(t *testing.T) {\n\ttestCases := []*types.QueryTaskCompletedType{\n\t\tnil,\n\t\ttypes.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\ttypes.QueryTaskCompletedTypeFailed.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryTaskCompletedType(original)\n\t\troundTripObj := ToQueryTaskCompletedType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryWorkflowRequestConversion(t *testing.T) {\n\ttestCases := []*types.QueryWorkflowRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.QueryWorkflowRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryWorkflowRequest(original)\n\t\troundTripObj := ToQueryWorkflowRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueryWorkflowResponseConversion(t *testing.T) {\n\ttestCases := []*types.QueryWorkflowResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.QueryWorkflowResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueryWorkflowResponse(original)\n\t\troundTripObj := ToQueryWorkflowResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestReapplyEventsRequestConversion(t *testing.T) {\n\ttestCases := []*types.ReapplyEventsRequest{\n\t\tnil,\n\t\t{},\n\t\t{DomainName: \"test-domain-name\", WorkflowExecution: &testdata.WorkflowExecution},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminReapplyEventsRequest(original)\n\t\troundTripObj := ToAdminReapplyEventsRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordActivityTaskHeartbeatByIDRequestConversion(t *testing.T) {\n\ttestCases := []*types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RecordActivityTaskHeartbeatByIDRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordActivityTaskHeartbeatByIDRequest(original)\n\t\troundTripObj := ToRecordActivityTaskHeartbeatByIDRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordActivityTaskHeartbeatRequestConversion(t *testing.T) {\n\ttestCases := []*types.RecordActivityTaskHeartbeatRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RecordActivityTaskHeartbeatRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordActivityTaskHeartbeatRequest(original)\n\t\troundTripObj := ToRecordActivityTaskHeartbeatRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordActivityTaskHeartbeatResponseConversion(t *testing.T) {\n\ttestCases := []*types.RecordActivityTaskHeartbeatResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RecordActivityTaskHeartbeatResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordActivityTaskHeartbeatResponse(original)\n\t\troundTripObj := ToRecordActivityTaskHeartbeatResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRecordMarkerDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.RecordMarkerDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RecordMarkerDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRecordMarkerDecisionAttributes(original)\n\t\troundTripObj := ToRecordMarkerDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRefreshWorkflowTasksRequestConversion(t *testing.T) {\n\ttestCases := []*types.RefreshWorkflowTasksRequest{\n\t\tnil,\n\t\t{},\n\t\t{Domain: \"test-domain\", Execution: &testdata.WorkflowExecution},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRefreshWorkflowTasksRequest(original)\n\t\troundTripObj := ToRefreshWorkflowTasksRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRegisterDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.RegisterDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RegisterDomainRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRegisterDomainRequest(original)\n\t\troundTripObj := ToRegisterDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRegisterDomainRequestFuzz(t *testing.T) {\n\tt.Run(\"round trip from internal\", func(t *testing.T) {\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"nil\", \"empty\", \"filled\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\t// Configure fuzzer to generate valid enum values\n\t\t\tfuzzer := f.Funcs(\n\t\t\t\tfunc(e *types.ArchivalStatus, c fuzz.Continue) {\n\t\t\t\t\t*e = types.ArchivalStatus(c.Intn(2)) // 0-1 are valid values (Disabled=0, Enabled=1)\n\t\t\t\t},\n\t\t\t).NilChance(0.3)\n\n\t\t\tvar orig *types.RegisterDomainRequest\n\t\t\tfuzzer.Fuzz(&orig)\n\t\t\tout := ToRegisterDomainRequest(FromRegisterDomainRequest(orig))\n\t\t\tassert.Equal(t, orig, out, \"RegisterDomainRequest did not survive round-tripping\")\n\n\t\t\tif orig == nil {\n\t\t\t\treturn \"nil\"\n\t\t\t}\n\t\t\tif orig.Name == \"\" && orig.ActiveClusterName == \"\" && orig.ActiveClusters == nil {\n\t\t\t\treturn \"empty\"\n\t\t\t}\n\t\t\treturn \"filled\"\n\t\t})\n\t})\n}\n\nfunc TestRemoteSyncMatchedErrorConversion(t *testing.T) {\n\ttestCases := []*types.RemoteSyncMatchedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RemoteSyncMatchedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRemoteSyncMatchedError(original)\n\t\troundTripObj := ToRemoteSyncMatchedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRemoveTaskRequestConversion(t *testing.T) {\n\ttestCases := []*types.RemoveTaskRequest{\n\t\tnil,\n\t\t{},\n\t\t{ShardID: 10, Type: common.Int32Ptr(1), TaskID: 5, ClusterName: \"test-cluster-name\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminRemoveTaskRequest(original)\n\t\troundTripObj := ToAdminRemoveTaskRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRequestCancelActivityTaskDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.RequestCancelActivityTaskDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RequestCancelActivityTaskDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRequestCancelActivityTaskDecisionAttributes(original)\n\t\troundTripObj := ToRequestCancelActivityTaskDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRequestCancelActivityTaskFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.RequestCancelActivityTaskFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RequestCancelActivityTaskFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRequestCancelActivityTaskFailedEventAttributes(original)\n\t\troundTripObj := ToRequestCancelActivityTaskFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRequestCancelExternalWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RequestCancelExternalWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRequestCancelExternalWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToRequestCancelExternalWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRequestCancelExternalWorkflowExecutionFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RequestCancelExternalWorkflowExecutionFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRequestCancelExternalWorkflowExecutionFailedEventAttributes(original)\n\t\troundTripObj := ToRequestCancelExternalWorkflowExecutionFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRequestCancelWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.RequestCancelWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RequestCancelWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRequestCancelWorkflowExecutionRequest(original)\n\t\troundTripObj := ToRequestCancelWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetPointInfoConversion(t *testing.T) {\n\ttestCases := []*types.ResetPointInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetPointInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetPointInfo(original)\n\t\troundTripObj := ToResetPointInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetPointsConversion(t *testing.T) {\n\ttestCases := []*types.ResetPoints{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetPoints,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetPoints(original)\n\t\troundTripObj := ToResetPoints(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetQueueRequestConversion(t *testing.T) {\n\ttestCases := []*types.ResetQueueRequest{\n\t\tnil,\n\t\t{},\n\t\t{ShardID: 10, ClusterName: \"test-cluster-name\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromAdminResetQueueRequest(original)\n\t\troundTripObj := ToAdminResetQueueRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetStickyTaskListRequestConversion(t *testing.T) {\n\ttestCases := []*types.ResetStickyTaskListRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetStickyTaskListRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetStickyTaskListRequest(original)\n\t\troundTripObj := ToResetStickyTaskListRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetStickyTaskListResponseConversion(t *testing.T) {\n\ttestCases := []*types.ResetStickyTaskListResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetStickyTaskListResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetStickyTaskListResponse(original)\n\t\troundTripObj := ToResetStickyTaskListResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.ResetWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetWorkflowExecutionRequest(original)\n\t\troundTripObj := ToResetWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestResetWorkflowExecutionResponseConversion(t *testing.T) {\n\ttestCases := []*types.ResetWorkflowExecutionResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ResetWorkflowExecutionResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromResetWorkflowExecutionResponse(original)\n\t\troundTripObj := ToResetWorkflowExecutionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskCanceledByIDRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskCanceledByIDRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskCanceledByIDRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskCanceledByIDRequest(original)\n\t\troundTripObj := ToRespondActivityTaskCanceledByIDRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskCanceledRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskCanceledRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskCanceledRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskCanceledRequest(original)\n\t\troundTripObj := ToRespondActivityTaskCanceledRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskCompletedByIDRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskCompletedByIDRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskCompletedByIDRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskCompletedByIDRequest(original)\n\t\troundTripObj := ToRespondActivityTaskCompletedByIDRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskCompletedRequest(original)\n\t\troundTripObj := ToRespondActivityTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskFailedByIDRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskFailedByIDRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskFailedByIDRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskFailedByIDRequest(original)\n\t\troundTripObj := ToRespondActivityTaskFailedByIDRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondActivityTaskFailedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondActivityTaskFailedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondActivityTaskFailedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondActivityTaskFailedRequest(original)\n\t\troundTripObj := ToRespondActivityTaskFailedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondDecisionTaskCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondDecisionTaskCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondDecisionTaskCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondDecisionTaskCompletedRequest(original)\n\t\troundTripObj := ToRespondDecisionTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondDecisionTaskFailedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondDecisionTaskFailedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondDecisionTaskFailedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondDecisionTaskFailedRequest(original)\n\t\troundTripObj := ToRespondDecisionTaskFailedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRespondQueryTaskCompletedRequestConversion(t *testing.T) {\n\ttestCases := []*types.RespondQueryTaskCompletedRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RespondQueryTaskCompletedRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRespondQueryTaskCompletedRequest(original)\n\t\troundTripObj := ToRespondQueryTaskCompletedRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRetryPolicyConversion(t *testing.T) {\n\ttestCases := []*types.RetryPolicy{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RetryPolicy,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRetryPolicy(original)\n\t\troundTripObj := ToRetryPolicy(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRetryTaskV2ErrorConversion(t *testing.T) {\n\ttestCases := []*types.RetryTaskV2Error{\n\t\tnil,\n\t\t{},\n\t\t&testdata.RetryTaskV2Error,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRetryTaskV2Error(original)\n\t\troundTripObj := ToRetryTaskV2Error(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRestartWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.RestartWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t{Domain: \"test-domain\", WorkflowExecution: &testdata.WorkflowExecution},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRestartWorkflowExecutionRequest(original)\n\t\troundTripObj := ToRestartWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestScheduleActivityTaskDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.ScheduleActivityTaskDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ScheduleActivityTaskDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromScheduleActivityTaskDecisionAttributes(original)\n\t\troundTripObj := ToScheduleActivityTaskDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSearchAttributesConversion(t *testing.T) {\n\ttestCases := []*types.SearchAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SearchAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSearchAttributes(original)\n\t\troundTripObj := ToSearchAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestServiceBusyErrorConversion(t *testing.T) {\n\ttestCases := []*types.ServiceBusyError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ServiceBusyError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromServiceBusyError(original)\n\t\troundTripObj := ToServiceBusyError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalExternalWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalExternalWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalExternalWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToSignalExternalWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalExternalWorkflowExecutionFailedCauseConversion(t *testing.T) {\n\ttestCases := []*types.SignalExternalWorkflowExecutionFailedCause{\n\t\tnil,\n\t\ttypes.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution.Ptr(),\n\t\ttypes.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalExternalWorkflowExecutionFailedCause(original)\n\t\troundTripObj := ToSignalExternalWorkflowExecutionFailedCause(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalExternalWorkflowExecutionFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalExternalWorkflowExecutionFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalExternalWorkflowExecutionFailedEventAttributes(original)\n\t\troundTripObj := ToSignalExternalWorkflowExecutionFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalExternalWorkflowExecutionInitiatedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalExternalWorkflowExecutionInitiatedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalExternalWorkflowExecutionInitiatedEventAttributes(original)\n\t\troundTripObj := ToSignalExternalWorkflowExecutionInitiatedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.SignalWithStartWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalWithStartWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalWithStartWorkflowExecutionRequest(original)\n\t\troundTripObj := ToSignalWithStartWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.SignalWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalWorkflowExecutionRequest(original)\n\t\troundTripObj := ToSignalWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartChildWorkflowExecutionDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartChildWorkflowExecutionDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartChildWorkflowExecutionDecisionAttributes(original)\n\t\troundTripObj := ToStartChildWorkflowExecutionDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartChildWorkflowExecutionFailedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartChildWorkflowExecutionFailedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartChildWorkflowExecutionFailedEventAttributes(original)\n\t\troundTripObj := ToStartChildWorkflowExecutionFailedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartChildWorkflowExecutionInitiatedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartChildWorkflowExecutionInitiatedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartChildWorkflowExecutionInitiatedEventAttributes(original)\n\t\troundTripObj := ToStartChildWorkflowExecutionInitiatedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartTimeFilterConversion(t *testing.T) {\n\ttestCases := []*types.StartTimeFilter{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartTimeFilter,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartTimeFilter(original)\n\t\troundTripObj := ToStartTimeFilter(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartTimerDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.StartTimerDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartTimerDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartTimerDecisionAttributes(original)\n\t\troundTripObj := ToStartTimerDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartWorkflowExecutionRequestConversion(t *testing.T) {\n\ttestCases := []*types.StartWorkflowExecutionRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartWorkflowExecutionRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartWorkflowExecutionRequest(original)\n\t\troundTripObj := ToStartWorkflowExecutionRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartWorkflowExecutionAsyncRequestConversion(t *testing.T) {\n\ttestCases := []*types.StartWorkflowExecutionAsyncRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartWorkflowExecutionAsyncRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartWorkflowExecutionAsyncRequest(original)\n\t\troundTripObj := ToStartWorkflowExecutionAsyncRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecutionAsyncRequestConversion(t *testing.T) {\n\ttestCases := []*types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalWithStartWorkflowExecutionAsyncRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalWithStartWorkflowExecutionAsyncRequest(original)\n\t\troundTripObj := ToSignalWithStartWorkflowExecutionAsyncRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestRestartWorkflowExecutionResponseConversion(t *testing.T) {\n\ttestCases := []*types.RestartWorkflowExecutionResponse{\n\t\tnil,\n\t\t{},\n\t\t{RunID: \"test-run-id\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromRestartWorkflowExecutionResponse(original)\n\t\troundTripObj := ToRestartWorkflowExecutionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartWorkflowExecutionResponseConversion(t *testing.T) {\n\ttestCases := []*types.StartWorkflowExecutionResponse{\n\t\tnil,\n\t\t{},\n\t\t{RunID: \"test-run-id\"},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartWorkflowExecutionResponse(original)\n\t\troundTripObj := ToStartWorkflowExecutionResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStartWorkflowExecutionAsyncResponseConversion(t *testing.T) {\n\ttestCases := []*types.StartWorkflowExecutionAsyncResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StartWorkflowExecutionAsyncResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStartWorkflowExecutionAsyncResponse(original)\n\t\troundTripObj := ToStartWorkflowExecutionAsyncResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecutionAsyncResponseConversion(t *testing.T) {\n\ttestCases := []*types.SignalWithStartWorkflowExecutionAsyncResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SignalWithStartWorkflowExecutionAsyncResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSignalWithStartWorkflowExecutionAsyncResponse(original)\n\t\troundTripObj := ToSignalWithStartWorkflowExecutionAsyncResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStickyExecutionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.StickyExecutionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StickyExecutionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStickyExecutionAttributes(original)\n\t\troundTripObj := ToStickyExecutionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestSupportedClientVersionsConversion(t *testing.T) {\n\ttestCases := []*types.SupportedClientVersions{\n\t\tnil,\n\t\t{},\n\t\t&testdata.SupportedClientVersions,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromSupportedClientVersions(original)\n\t\troundTripObj := ToSupportedClientVersions(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskIDBlockConversion(t *testing.T) {\n\ttestCases := []*types.TaskIDBlock{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TaskIDBlock,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskIDBlock(original)\n\t\troundTripObj := ToTaskIDBlock(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskListConversion(t *testing.T) {\n\ttestCases := []*types.TaskList{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TaskList,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskList(original)\n\t\troundTripObj := ToTaskList(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskListMetadataConversion(t *testing.T) {\n\ttestCases := []*types.TaskListMetadata{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TaskListMetadata,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskListMetadata(original)\n\t\troundTripObj := ToTaskListMetadata(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskListPartitionMetadataConversion(t *testing.T) {\n\ttestCases := []*types.TaskListPartitionMetadata{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TaskListPartitionMetadata,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskListPartitionMetadata(original)\n\t\troundTripObj := ToTaskListPartitionMetadata(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskListStatusConversion(t *testing.T) {\n\ttestCases := []*types.TaskListStatus{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TaskListStatus,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskListStatus(original)\n\t\troundTripObj := ToTaskListStatus(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskListTypeConversion(t *testing.T) {\n\ttestCases := []*types.TaskListType{\n\t\tnil,\n\t\ttypes.TaskListTypeDecision.Ptr(),\n\t\ttypes.TaskListTypeActivity.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskListType(original)\n\t\troundTripObj := ToTaskListType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTimeoutTypeConversion(t *testing.T) {\n\ttestCases := []*types.TimeoutType{\n\t\tnil,\n\t\ttypes.TimeoutTypeStartToClose.Ptr(),\n\t\ttypes.TimeoutTypeScheduleToStart.Ptr(),\n\t\ttypes.TimeoutTypeScheduleToClose.Ptr(),\n\t\ttypes.TimeoutTypeHeartbeat.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTimeoutType(original)\n\t\troundTripObj := ToTimeoutType(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTimerCanceledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.TimerCanceledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TimerCanceledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTimerCanceledEventAttributes(original)\n\t\troundTripObj := ToTimerCanceledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTimerFiredEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.TimerFiredEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TimerFiredEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTimerFiredEventAttributes(original)\n\t\troundTripObj := ToTimerFiredEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTimerStartedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.TimerStartedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TimerStartedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTimerStartedEventAttributes(original)\n\t\troundTripObj := ToTimerStartedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTransientDecisionInfoConversion(t *testing.T) {\n\ttestCases := []*types.TransientDecisionInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.TransientDecisionInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTransientDecisionInfo(original)\n\t\troundTripObj := ToTransientDecisionInfo(thriftObj)\n\t\topt := cmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"ParentWorkflowDomainID\")\n\t\tif diff := cmp.Diff(original, roundTripObj, opt); diff != \"\" {\n\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t}\n\t}\n}\n\nfunc TestUpdateDomainRequestConversion(t *testing.T) {\n\ttestCases := []*types.UpdateDomainRequest{\n\t\tnil,\n\t\t{},\n\t\t&testdata.UpdateDomainRequest,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromUpdateDomainRequest(original)\n\t\troundTripObj := ToUpdateDomainRequest(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestUpdateDomainResponseConversion(t *testing.T) {\n\ttestCases := []*types.UpdateDomainResponse{\n\t\tnil,\n\t\t{},\n\t\t&testdata.UpdateDomainResponse,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromUpdateDomainResponse(original)\n\t\troundTripObj := ToUpdateDomainResponse(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestUpsertWorkflowSearchAttributesDecisionAttributesConversion(t *testing.T) {\n\ttestCases := []*types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.UpsertWorkflowSearchAttributesDecisionAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromUpsertWorkflowSearchAttributesDecisionAttributes(original)\n\t\troundTripObj := ToUpsertWorkflowSearchAttributesDecisionAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestUpsertWorkflowSearchAttributesEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.UpsertWorkflowSearchAttributesEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromUpsertWorkflowSearchAttributesEventAttributes(original)\n\t\troundTripObj := ToUpsertWorkflowSearchAttributesEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestVersionHistoriesConversion(t *testing.T) {\n\ttestCases := []*types.VersionHistories{\n\t\tnil,\n\t\t{},\n\t\t&testdata.VersionHistories,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromVersionHistories(original)\n\t\troundTripObj := ToVersionHistories(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestVersionHistoryConversion(t *testing.T) {\n\ttestCases := []*types.VersionHistory{\n\t\tnil,\n\t\t{},\n\t\t&testdata.VersionHistory,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromVersionHistory(original)\n\t\troundTripObj := ToVersionHistory(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestVersionHistoryItemConversion(t *testing.T) {\n\ttestCases := []*types.VersionHistoryItem{\n\t\tnil,\n\t\t{},\n\t\t&testdata.VersionHistoryItem,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromVersionHistoryItem(original)\n\t\troundTripObj := ToVersionHistoryItem(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkerVersionInfoConversion(t *testing.T) {\n\ttestCases := []*types.WorkerVersionInfo{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkerVersionInfo,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkerVersionInfo(original)\n\t\troundTripObj := ToWorkerVersionInfo(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecution{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecution,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecution(original)\n\t\troundTripObj := ToWorkflowExecution(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionAlreadyStartedErrorConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionAlreadyStartedError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecutionAlreadyStartedError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionAlreadyStartedError(original)\n\t\troundTripObj := ToWorkflowExecutionAlreadyStartedError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionCancelRequestedEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecutionCancelRequestedEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionCancelRequestedEventAttributes(original)\n\t\troundTripObj := ToWorkflowExecutionCancelRequestedEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionCanceledEventAttributesConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionCanceledEventAttributes{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecutionCanceledEventAttributes,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionCanceledEventAttributes(original)\n\t\troundTripObj := ToWorkflowExecutionCanceledEventAttributes(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionCloseStatusConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionCloseStatus{\n\t\tnil,\n\t\ttypes.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusCanceled.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusTerminated.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(),\n\t\ttypes.WorkflowExecutionCloseStatusTimedOut.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionCloseStatus(original)\n\t\troundTripObj := ToWorkflowExecutionCloseStatus(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowExecutionFilterConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowExecutionFilter{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowExecutionFilter,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowExecutionFilter(original)\n\t\troundTripObj := ToWorkflowExecutionFilter(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowIDReusePolicyConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowIDReusePolicy{\n\t\tnil,\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicateFailedOnly.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyRejectDuplicate.Ptr(),\n\t\ttypes.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowIDReusePolicy(original)\n\t\troundTripObj := ToWorkflowIDReusePolicy(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestWorkflowTypeFilterConversion(t *testing.T) {\n\ttestCases := []*types.WorkflowTypeFilter{\n\t\tnil,\n\t\t{},\n\t\t&testdata.WorkflowTypeFilter,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromWorkflowTypeFilter(original)\n\t\troundTripObj := ToWorkflowTypeFilter(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryBranchRangeArrayConversion(t *testing.T) {\n\ttestCases := [][]*types.HistoryBranchRange{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\t{BranchID: \"test-branch-id-1\", BeginNodeID: 1, EndNodeID: 2},\n\t\t\t{BranchID: \"test-branch-id-2\", BeginNodeID: 5, EndNodeID: 6},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryBranchRangeArray(original)\n\t\troundTripObj := ToHistoryBranchRangeArray(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestActivityLocalDispatchInfoMapConversion(t *testing.T) {\n\ttestCases := []map[string]*types.ActivityLocalDispatchInfo{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\t\"test-key-1\": &testdata.ActivityLocalDispatchInfo,\n\t\t\t\"test-key-2\": &testdata.ActivityLocalDispatchInfo,\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActivityLocalDispatchInfoMap(original)\n\t\troundTripObj := ToActivityLocalDispatchInfoMap(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestHistoryArrayConversion(t *testing.T) {\n\ttestCases := [][]*types.History{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\t{Events: []*types.HistoryEvent{{ID: 1}, {ID: 2}}},\n\t\t\t{Events: []*types.HistoryEvent{{ID: 3}, {ID: 4}}},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromHistoryArray(original)\n\t\troundTripObj := ToHistoryArray(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestStickyWorkerUnavailableErrorConversion(t *testing.T) {\n\ttestCases := []*types.StickyWorkerUnavailableError{\n\t\tnil,\n\t\t{},\n\t\t&testdata.StickyWorkerUnavailableError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromStickyWorkerUnavailableError(original)\n\t\troundTripObj := ToStickyWorkerUnavailableError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestFromTaskListNotOwnedByHostError(t *testing.T) {\n\ttestCases := []*cadence_errors.TaskListNotOwnedByHostError{\n\t\tnil,\n\t\t&testdata.TaskListNotOwnedByHostError,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskListNotOwnedByHostError(original)\n\t\troundTripObj := ToTaskListNotOwnedByHostError(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestAny(t *testing.T) {\n\tt.Run(\"sanity check\", func(t *testing.T) {\n\t\t// test the main fields are mapped correctly\n\t\tinternal := types.Any{\n\t\t\tValueType: \"testing\",\n\t\t\tValue:     []byte(`test`),\n\t\t}\n\t\trpc := shared.Any{\n\t\t\tValueType: common.StringPtr(\"testing\"),\n\t\t\tValue:     []byte(`test`),\n\t\t}\n\t\tassert.Equal(t, &rpc, FromAny(&internal))\n\t\tassert.Equal(t, &internal, ToAny(&rpc))\n\t})\n\tt.Run(\"round trip nils\", func(t *testing.T) {\n\t\t// somewhat annoying in fuzzing and there are few possibilities, so tested separately\n\t\tassert.Nil(t, FromAny(ToAny(nil)), \"nil thrift -> internal -> thrift => should result in nil\")\n\t\tassert.Nil(t, ToAny(FromAny(nil)), \"nil internal -> thrift -> internal => should result in nil\")\n\t})\n\n\tt.Run(\"round trip from internal\", func(t *testing.T) {\n\t\t// fuzz test to check round trip data works as it should when starting from internal types\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"empty data\", \"filled data\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\tvar orig types.Any\n\t\t\tf.Fuzz(&orig)\n\t\t\tout := ToAny(FromAny(&orig))\n\t\t\tassert.Equal(t, &orig, out, \"did not survive round-tripping\")\n\n\t\t\t// report what branch of behavior was executed so we can ensure stable coverage\n\t\t\tif len(orig.Value) == 0 {\n\t\t\t\treturn \"empty data\" // ignoring nil vs empty difference\n\t\t\t}\n\t\t\treturn \"filled data\"\n\t\t})\n\t})\n\tt.Run(\"round trip from thrift\", func(t *testing.T) {\n\t\t// fuzz test to check round trip data works as it should\n\t\ttestutils.EnsureFuzzCoverage(t, []string{\n\t\t\t\"nil id\", \"empty data\", \"filled data\",\n\t\t}, func(t *testing.T, f *fuzz.Fuzzer) string {\n\t\t\tvar orig shared.Any\n\t\t\tf.Fuzz(&orig)\n\t\t\tout := FromAny(ToAny(&orig))\n\n\t\t\tif orig.ValueType == nil {\n\t\t\t\t// internal type does not support nil strings, so it will be transformed to an empty string.\n\t\t\t\tassert.Equal(t, \"\", *out.ValueType, \"empty value type did not survive round-tripping\")\n\t\t\t\tassert.Equal(t, orig.Value, out.Value, \"value did not survive round-tripping\")\n\t\t\t\treturn \"nil id\"\n\t\t\t}\n\t\t\t// non-nil ValueType can check all fields\n\t\t\tassert.Equal(t, &orig, out, \"did not survive round-tripping\")\n\t\t\tif len(orig.Value) == 0 {\n\t\t\t\treturn \"empty data\" // ignoring nil vs empty difference\n\t\t\t}\n\t\t\treturn \"filled data\"\n\t\t})\n\t})\n}\n\nfunc TestTaskKeyConversion(t *testing.T) {\n\ttestCases := []*types.TaskKey{\n\t\tnil,\n\t\t&types.TaskKey{},\n\t\t&testdata.TaskKey,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskKey(original)\n\t\troundTripObj := ToTaskKey(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestTaskRangeConversion(t *testing.T) {\n\ttestCases := []*types.TaskRange{\n\t\tnil,\n\t\t&types.TaskRange{},\n\t\t&testdata.TaskRange,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromTaskRange(original)\n\t\troundTripObj := ToTaskRange(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestVirtualSliceStateConversion(t *testing.T) {\n\ttestCases := []*types.VirtualSliceState{\n\t\tnil,\n\t\t&types.VirtualSliceState{},\n\t\t&testdata.VirtualSliceState,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromVirtualSliceState(original)\n\t\troundTripObj := ToVirtualSliceState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestVirtualQueueStateConversion(t *testing.T) {\n\ttestCases := []*types.VirtualQueueState{\n\t\tnil,\n\t\t&types.VirtualQueueState{},\n\t\t&testdata.VirtualQueueState,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromVirtualQueueState(original)\n\t\troundTripObj := ToVirtualQueueState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestQueueStateConversion(t *testing.T) {\n\ttestCases := []*types.QueueState{\n\t\tnil,\n\t\t&types.QueueState{},\n\t\t&testdata.QueueState,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromQueueState(original)\n\t\troundTripObj := ToQueueState(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\n// TODO: The way we're testing mappers doesn't have good coverage. The round trip tests don't cover serialization and deserialization of thrift.\n// We should also generate test data in thrift types and do round trip tests from thrift to internal and back to thrift.\nfunc TestActiveClusterSelectionPolicyConversion(t *testing.T) {\n\ttestCases := []*types.ActiveClusterSelectionPolicy{\n\t\tnil,\n\t\t{},\n\t\t&testdata.ActiveClusterSelectionPolicyExternalEntity,\n\t\t&testdata.ActiveClusterSelectionPolicyRegionSticky,\n\t\t&testdata.ActiveClusterSelectionPolicyWithClusterAttribute,\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActiveClusterSelectionPolicy(original)\n\t\troundTripObj := ToActiveClusterSelectionPolicy(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n\n}\n\nfunc TestActiveClustersConversion(t *testing.T) {\n\ttestCases := []*types.ActiveClusters{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"datacenter\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"dc1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"us-west-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromActiveClusters(original)\n\t\troundTripObj := ToActiveClusters(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestClusterAttributeScopeConversion(t *testing.T) {\n\ttestCases := []*types.ClusterAttributeScope{\n\t\tnil,\n\t\t{},\n\t\t{\n\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\"us-west-1\": {\n\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t},\n\t\t\t\t\"us-east-1\": {\n\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, original := range testCases {\n\t\tthriftObj := FromClusterAttributeScope(original)\n\t\troundTripObj := ToClusterAttributeScope(thriftObj)\n\t\tassert.Equal(t, original, roundTripObj)\n\t}\n}\n\nfunc TestListFailoverHistoryResponseConversion(t *testing.T) {\n\tfuzzer := testdatagen.New(t)\n\tfor i := 0; i < 100; i++ {\n\t\tvar response types.ListFailoverHistoryResponse\n\t\tfuzzer.Fuzz(&response)\n\t\tthriftResponse := FromListFailoverHistoryResponse(&response)\n\n\t\ttoResponse := ToListFailoverHistoryResponse(thriftResponse)\n\t\tassert.Equal(t, &response, toResponse)\n\t}\n}\n"
  },
  {
    "path": "common/types/matching.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n)\n\n// AddActivityTaskRequest is an internal type (TBD...)\ntype AddActivityTaskRequest struct {\n\tDomainUUID                    string                    `json:\"domainUUID,omitempty\"`\n\tExecution                     *WorkflowExecution        `json:\"execution,omitempty\"`\n\tSourceDomainUUID              string                    `json:\"sourceDomainUUID,omitempty\"`\n\tTaskList                      *TaskList                 `json:\"taskList,omitempty\"`\n\tScheduleID                    int64                     `json:\"scheduleId,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32                    `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tSource                        *TaskSource               `json:\"source,omitempty\"`\n\tForwardedFrom                 string                    `json:\"forwardedFrom,omitempty\"`\n\tActivityTaskDispatchInfo      *ActivityTaskDispatchInfo `json:\"activityTaskDispatchInfo,omitempty\"`\n\tPartitionConfig               map[string]string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetSourceDomainUUID is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetSourceDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.SourceDomainUUID\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetScheduleID is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetScheduleID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// GetScheduleToStartTimeoutSeconds is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetSource is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetSource() (o TaskSource) {\n\tif v != nil && v.Source != nil {\n\t\treturn *v.Source\n\t}\n\treturn\n}\n\n// GetForwardedFrom is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil {\n\t\treturn v.ForwardedFrom\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *AddActivityTaskRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// ActivityTaskDispatchInfo is an internal type (TBD...)\ntype ActivityTaskDispatchInfo struct {\n\tScheduledEvent                  *HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedTimestamp                *int64        `json:\"startedTimestamp,omitempty\"`\n\tAttempt                         *int64        `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64        `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte        `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *WorkflowType `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  string        `json:\"workflowDomain,omitempty\"`\n}\n\n// AddDecisionTaskRequest is an internal type (TBD...)\ntype AddDecisionTaskRequest struct {\n\tDomainUUID                    string             `json:\"domainUUID,omitempty\"`\n\tExecution                     *WorkflowExecution `json:\"execution,omitempty\"`\n\tTaskList                      *TaskList          `json:\"taskList,omitempty\"`\n\tScheduleID                    int64              `json:\"scheduleId,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32             `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tSource                        *TaskSource        `json:\"source,omitempty\"`\n\tForwardedFrom                 string             `json:\"forwardedFrom,omitempty\"`\n\tPartitionConfig               map[string]string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetScheduleID is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetScheduleID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// GetScheduleToStartTimeoutSeconds is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetSource is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetSource() (o TaskSource) {\n\tif v != nil && v.Source != nil {\n\t\treturn *v.Source\n\t}\n\treturn\n}\n\n// GetForwardedFrom is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil {\n\t\treturn v.ForwardedFrom\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *AddDecisionTaskRequest) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\ntype AddActivityTaskResponse struct {\n\tPartitionConfig *TaskListPartitionConfig\n}\n\ntype AddDecisionTaskResponse struct {\n\tPartitionConfig *TaskListPartitionConfig\n}\n\n// CancelOutstandingPollRequest is an internal type (TBD...)\ntype CancelOutstandingPollRequest struct {\n\tDomainUUID   string    `json:\"domainUUID,omitempty\"`\n\tTaskListType *int32    `json:\"taskListType,omitempty\"`\n\tTaskList     *TaskList `json:\"taskList,omitempty\"`\n\tPollerID     string    `json:\"pollerID,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *CancelOutstandingPollRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetTaskListType is an internal getter (TBD...)\nfunc (v *CancelOutstandingPollRequest) GetTaskListType() (o int32) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *CancelOutstandingPollRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetPollerID is an internal getter (TBD...)\nfunc (v *CancelOutstandingPollRequest) GetPollerID() (o string) {\n\tif v != nil {\n\t\treturn v.PollerID\n\t}\n\treturn\n}\n\n// MatchingDescribeTaskListRequest is an internal type (TBD...)\ntype MatchingDescribeTaskListRequest struct {\n\tDomainUUID  string                   `json:\"domainUUID,omitempty\"`\n\tDescRequest *DescribeTaskListRequest `json:\"descRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *MatchingDescribeTaskListRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetDescRequest is an internal getter (TBD...)\nfunc (v *MatchingDescribeTaskListRequest) GetDescRequest() (o *DescribeTaskListRequest) {\n\tif v != nil && v.DescRequest != nil {\n\t\treturn v.DescRequest\n\t}\n\treturn\n}\n\n// MatchingListTaskListPartitionsRequest is an internal type (TBD...)\ntype MatchingListTaskListPartitionsRequest struct {\n\tDomain   string    `json:\"domain,omitempty\"`\n\tTaskList *TaskList `json:\"taskList,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *MatchingListTaskListPartitionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *MatchingListTaskListPartitionsRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// MatchingGetTaskListsByDomainRequest is an internal type (TBD...)\ntype MatchingGetTaskListsByDomainRequest struct {\n\tDomain string `json:\"domain,omitempty\"`\n}\n\n// GetDomainName is an internal getter (TBD...)\nfunc (v *MatchingGetTaskListsByDomainRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// MatchingPollForActivityTaskRequest is an internal type (TBD...)\ntype MatchingPollForActivityTaskRequest struct {\n\tDomainUUID     string                      `json:\"domainUUID,omitempty\"`\n\tPollerID       string                      `json:\"pollerID,omitempty\"`\n\tPollRequest    *PollForActivityTaskRequest `json:\"pollRequest,omitempty\"`\n\tForwardedFrom  string                      `json:\"forwardedFrom,omitempty\"`\n\tIsolationGroup string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *MatchingPollForActivityTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetPollerID is an internal getter (TBD...)\nfunc (v *MatchingPollForActivityTaskRequest) GetPollerID() (o string) {\n\tif v != nil {\n\t\treturn v.PollerID\n\t}\n\treturn\n}\n\n// GetPollRequest is an internal getter (TBD...)\nfunc (v *MatchingPollForActivityTaskRequest) GetPollRequest() (o *PollForActivityTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter.\nfunc (v *MatchingPollForActivityTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest.TaskList\n\t}\n\treturn\n}\n\n// GetForwardedFrom is an internal getter (TBD...)\nfunc (v *MatchingPollForActivityTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil {\n\t\treturn v.ForwardedFrom\n\t}\n\treturn\n}\n\n// GetIsolatioonGroup is an internal getter (TBD...)\nfunc (v *MatchingPollForActivityTaskRequest) GetIsolationGroup() (o string) {\n\tif v != nil {\n\t\treturn v.IsolationGroup\n\t}\n\treturn\n}\n\n// MatchingPollForDecisionTaskRequest is an internal type (TBD...)\ntype MatchingPollForDecisionTaskRequest struct {\n\tDomainUUID     string                      `json:\"domainUUID,omitempty\"`\n\tPollerID       string                      `json:\"pollerID,omitempty\"`\n\tPollRequest    *PollForDecisionTaskRequest `json:\"pollRequest,omitempty\"`\n\tForwardedFrom  string                      `json:\"forwardedFrom,omitempty\"`\n\tIsolationGroup string\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetPollerID is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskRequest) GetPollerID() (o string) {\n\tif v != nil {\n\t\treturn v.PollerID\n\t}\n\treturn\n}\n\n// GetPollRequest is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskRequest) GetPollRequest() (o *PollForDecisionTaskRequest) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter.\nfunc (v *MatchingPollForDecisionTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.PollRequest != nil {\n\t\treturn v.PollRequest.TaskList\n\t}\n\treturn\n}\n\n// GetForwardedFrom is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskRequest) GetForwardedFrom() (o string) {\n\tif v != nil {\n\t\treturn v.ForwardedFrom\n\t}\n\treturn\n}\n\n// GetIsolatioonGroup is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskRequest) GetIsolationGroup() (o string) {\n\tif v != nil {\n\t\treturn v.IsolationGroup\n\t}\n\treturn\n}\n\ntype TaskListPartition struct {\n\tIsolationGroups []string\n}\n\ntype TaskListPartitionConfig struct {\n\tVersion         int64\n\tReadPartitions  map[int]*TaskListPartition\n\tWritePartitions map[int]*TaskListPartition\n}\n\n// MatchingPollForDecisionTaskResponse is an internal type (TBD...)\ntype MatchingPollForDecisionTaskResponse struct {\n\tTaskToken                 []byte                    `json:\"taskToken,omitempty\"`\n\tWorkflowExecution         *WorkflowExecution        `json:\"workflowExecution,omitempty\"`\n\tWorkflowType              *WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventID    *int64                    `json:\"previousStartedEventId,omitempty\"`\n\tStartedEventID            int64                     `json:\"startedEventId,omitempty\"`\n\tAttempt                   int64                     `json:\"attempt,omitempty\"`\n\tNextEventID               int64                     `json:\"nextEventId,omitempty\"`\n\tBacklogCountHint          int64                     `json:\"backlogCountHint,omitempty\"`\n\tStickyExecutionEnabled    bool                      `json:\"stickyExecutionEnabled,omitempty\"`\n\tQuery                     *WorkflowQuery            `json:\"query,omitempty\"`\n\tDecisionInfo              *TransientDecisionInfo    `json:\"decisionInfo,omitempty\"`\n\tWorkflowExecutionTaskList *TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tEventStoreVersion         int32                     `json:\"eventStoreVersion,omitempty\"`\n\tBranchToken               []byte                    `json:\"branchToken,omitempty\"`\n\tScheduledTimestamp        *int64                    `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                    `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*WorkflowQuery `json:\"queries,omitempty\"`\n\tTotalHistoryBytes         int64                     `json:\"currentHistorySize,omitempty\"`\n\tPartitionConfig           *TaskListPartitionConfig\n\tLoadBalancerHints         *LoadBalancerHints\n\tAutoConfigHint            *AutoConfigHint\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskResponse) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetPreviousStartedEventID is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskResponse) GetPreviousStartedEventID() (o int64) {\n\tif v != nil && v.PreviousStartedEventID != nil {\n\t\treturn *v.PreviousStartedEventID\n\t}\n\treturn\n}\n\n// GetNextEventID is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskResponse) GetNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.NextEventID\n\t}\n\treturn\n}\n\n// GetStickyExecutionEnabled is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskResponse) GetStickyExecutionEnabled() (o bool) {\n\tif v != nil {\n\t\treturn v.StickyExecutionEnabled\n\t}\n\treturn\n}\n\n// GetBranchToken is an internal getter (TBD...)\nfunc (v *MatchingPollForDecisionTaskResponse) GetBranchToken() (o []byte) {\n\tif v != nil && v.BranchToken != nil {\n\t\treturn v.BranchToken\n\t}\n\treturn\n}\n\n// GetTotalHistoryBytes is an internal getter of returning the history size in bytes\nfunc (v *MatchingPollForDecisionTaskResponse) GetTotalHistoryBytes() (o int64) {\n\tif v != nil {\n\t\treturn v.TotalHistoryBytes\n\t}\n\treturn\n}\n\ntype MatchingPollForActivityTaskResponse struct {\n\tTaskToken                       []byte             `json:\"taskToken,omitempty\"`\n\tWorkflowExecution               *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tActivityID                      string             `json:\"activityId,omitempty\"`\n\tActivityType                    *ActivityType      `json:\"activityType,omitempty\"`\n\tInput                           []byte             `json:\"input,omitempty\"`\n\tScheduledTimestamp              *int64             `json:\"scheduledTimestamp,omitempty\"`\n\tScheduleToCloseTimeoutSeconds   *int32             `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tStartedTimestamp                *int64             `json:\"startedTimestamp,omitempty\"`\n\tStartToCloseTimeoutSeconds      *int32             `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds         *int32             `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tAttempt                         int32              `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64             `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte             `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *WorkflowType      `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  string             `json:\"workflowDomain,omitempty\"`\n\tHeader                          *Header            `json:\"header,omitempty\"`\n\tBacklogCountHint                int64              `json:\"backlogCountHint,omitempty\"`\n\tPartitionConfig                 *TaskListPartitionConfig\n\tLoadBalancerHints               *LoadBalancerHints\n\tAutoConfigHint                  *AutoConfigHint\n}\n\n// MatchingQueryWorkflowRequest is an internal type (TBD...)\ntype MatchingQueryWorkflowRequest struct {\n\tDomainUUID    string                `json:\"domainUUID,omitempty\"`\n\tTaskList      *TaskList             `json:\"taskList,omitempty\"`\n\tQueryRequest  *QueryWorkflowRequest `json:\"queryRequest,omitempty\"`\n\tForwardedFrom string                `json:\"forwardedFrom,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetQueryRequest is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowRequest) GetQueryRequest() (o *QueryWorkflowRequest) {\n\tif v != nil && v.QueryRequest != nil {\n\t\treturn v.QueryRequest\n\t}\n\treturn\n}\n\n// GetForwardedFrom is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowRequest) GetForwardedFrom() (o string) {\n\tif v != nil {\n\t\treturn v.ForwardedFrom\n\t}\n\treturn\n}\n\ntype MatchingQueryWorkflowResponse struct {\n\tQueryResult     []byte\n\tQueryRejected   *QueryRejected\n\tPartitionConfig *TaskListPartitionConfig\n}\n\n// GetQueryResult is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowResponse) GetQueryResult() (o []byte) {\n\tif v != nil {\n\t\treturn v.QueryResult\n\t}\n\treturn\n}\n\n// GetQueryRejected is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowResponse) GetQueryRejected() (o *QueryRejected) {\n\tif v != nil {\n\t\treturn v.QueryRejected\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *MatchingQueryWorkflowResponse) GetPartitionConfig() (o *TaskListPartitionConfig) {\n\tif v != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// MatchingRespondQueryTaskCompletedRequest is an internal type (TBD...)\ntype MatchingRespondQueryTaskCompletedRequest struct {\n\tDomainUUID       string                            `json:\"domainUUID,omitempty\"`\n\tTaskList         *TaskList                         `json:\"taskList,omitempty\"`\n\tTaskID           string                            `json:\"taskID,omitempty\"`\n\tCompletedRequest *RespondQueryTaskCompletedRequest `json:\"completedRequest,omitempty\"`\n}\n\n// GetDomainUUID is an internal getter (TBD...)\nfunc (v *MatchingRespondQueryTaskCompletedRequest) GetDomainUUID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainUUID\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *MatchingRespondQueryTaskCompletedRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *MatchingRespondQueryTaskCompletedRequest) GetTaskID() (o string) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetCompletedRequest is an internal getter (TBD...)\nfunc (v *MatchingRespondQueryTaskCompletedRequest) GetCompletedRequest() (o *RespondQueryTaskCompletedRequest) {\n\tif v != nil && v.CompletedRequest != nil {\n\t\treturn v.CompletedRequest\n\t}\n\treturn\n}\n\n// TaskSource is an internal type (TBD...)\ntype TaskSource int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e TaskSource) Ptr() *TaskSource {\n\treturn &e\n}\n\n// String returns a readable string representation of TaskSource.\nfunc (e TaskSource) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"HISTORY\"\n\tcase 1:\n\t\treturn \"DB_BACKLOG\"\n\t}\n\treturn fmt.Sprintf(\"TaskSource(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *TaskSource) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"HISTORY\":\n\t\t*e = TaskSourceHistory\n\t\treturn nil\n\tcase \"DB_BACKLOG\":\n\t\t*e = TaskSourceDbBacklog\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskSource\", err)\n\t\t}\n\t\t*e = TaskSource(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskSource to text.\nfunc (e TaskSource) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// TaskSourceHistory is an option for TaskSource\n\tTaskSourceHistory TaskSource = iota\n\t// TaskSourceDbBacklog is an option for TaskSource\n\tTaskSourceDbBacklog\n)\n\ntype MatchingUpdateTaskListPartitionConfigRequest struct {\n\tDomainUUID      string\n\tTaskList        *TaskList\n\tTaskListType    *TaskListType\n\tPartitionConfig *TaskListPartitionConfig\n}\n\nfunc (v *MatchingUpdateTaskListPartitionConfigRequest) GetTaskListType() (o TaskListType) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\treturn\n}\n\ntype MatchingUpdateTaskListPartitionConfigResponse struct{}\n\ntype MatchingRefreshTaskListPartitionConfigRequest struct {\n\tDomainUUID      string\n\tTaskList        *TaskList\n\tTaskListType    *TaskListType\n\tPartitionConfig *TaskListPartitionConfig\n}\n\nfunc (v *MatchingRefreshTaskListPartitionConfigRequest) GetTaskListType() (o TaskListType) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\treturn\n}\n\ntype MatchingRefreshTaskListPartitionConfigResponse struct{}\n\ntype LoadBalancerHints struct {\n\tBacklogCount  int64\n\tRatePerSecond float64\n}\n"
  },
  {
    "path": "common/types/matching_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestAddActivityTaskRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &AddActivityTaskRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant *WorkflowExecution\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty execution\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"execution\",\n\t\t\treq:  &AddActivityTaskRequest{Execution: &WorkflowExecution{WorkflowID: \"test-workflow-id\"}},\n\t\t\twant: &WorkflowExecution{WorkflowID: \"test-workflow-id\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetExecution()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetSourceDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty source domain\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"source domain\",\n\t\t\treq:  &AddActivityTaskRequest{SourceDomainUUID: \"test-source-domain\"},\n\t\t\twant: \"test-source-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetSourceDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &AddActivityTaskRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetScheduleID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant int64\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty schedule id\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"schedule id\",\n\t\t\treq:  &AddActivityTaskRequest{ScheduleID: 123},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetScheduleID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetScheduleToStartTimeoutSeconds(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant int32\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty schedule to start timeout\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"schedule to start timeout\",\n\t\t\treq:  &AddActivityTaskRequest{ScheduleToStartTimeoutSeconds: func() *int32 { i := int32(123); return &i }()},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetScheduleToStartTimeoutSeconds()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetSource(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant TaskSource\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: TaskSource(0),\n\t\t},\n\t\t{\n\t\t\tname: \"empty source\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: TaskSource(0),\n\t\t},\n\t\t{\n\t\t\tname: \"source\",\n\t\t\treq:  &AddActivityTaskRequest{Source: TaskSource(123).Ptr()},\n\t\t\twant: TaskSource(123),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetSource()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetForwardedFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty forwarded from\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"forwarded from\",\n\t\t\treq:  &AddActivityTaskRequest{ForwardedFrom: \"test-forwarded-from\"},\n\t\t\twant: \"test-forwarded-from\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetForwardedFrom()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddActivityTaskRequest_GetPartitionConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddActivityTaskRequest\n\t\twant map[string]string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: func() map[string]string { return nil }(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty partition config\",\n\t\t\treq:  &AddActivityTaskRequest{},\n\t\t\twant: func() map[string]string { return nil }(),\n\t\t},\n\t\t{\n\t\t\tname: \"partition config\",\n\t\t\treq:  &AddActivityTaskRequest{PartitionConfig: map[string]string{\"test-key\": \"test-value\"}},\n\t\t\twant: map[string]string{\"test-key\": \"test-value\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPartitionConfig()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &AddDecisionTaskRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant *WorkflowExecution\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty execution\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"execution\",\n\t\t\treq:  &AddDecisionTaskRequest{Execution: &WorkflowExecution{WorkflowID: \"test-workflow-id\"}},\n\t\t\twant: &WorkflowExecution{WorkflowID: \"test-workflow-id\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetExecution()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &AddDecisionTaskRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetScheduleID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant int64\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty schedule id\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"schedule id\",\n\t\t\treq:  &AddDecisionTaskRequest{ScheduleID: 123},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetScheduleID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetScheduleToStartTimeoutSeconds(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant int32\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty schedule to start timeout\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"schedule to start timeout\",\n\t\t\treq:  &AddDecisionTaskRequest{ScheduleToStartTimeoutSeconds: func() *int32 { i := int32(123); return &i }()},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetScheduleToStartTimeoutSeconds()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetSource(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant TaskSource\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: TaskSource(0),\n\t\t},\n\t\t{\n\t\t\tname: \"empty source\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: TaskSource(0),\n\t\t},\n\t\t{\n\t\t\tname: \"source\",\n\t\t\treq:  &AddDecisionTaskRequest{Source: TaskSource(123).Ptr()},\n\t\t\twant: TaskSource(123),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetSource()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetForwardedFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty forwarded from\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"forwarded from\",\n\t\t\treq:  &AddDecisionTaskRequest{ForwardedFrom: \"test-forwarded-from\"},\n\t\t\twant: \"test-forwarded-from\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetForwardedFrom()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskRequest_GetPartitionConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *AddDecisionTaskRequest\n\t\twant map[string]string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: func() map[string]string { return nil }(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty partition config\",\n\t\t\treq:  &AddDecisionTaskRequest{},\n\t\t\twant: func() map[string]string { return nil }(),\n\t\t},\n\t\t{\n\t\t\tname: \"partition config\",\n\t\t\treq:  &AddDecisionTaskRequest{PartitionConfig: map[string]string{\"test-key\": \"test-value\"}},\n\t\t\twant: map[string]string{\"test-key\": \"test-value\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPartitionConfig()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestCancelOutstandingPollRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *CancelOutstandingPollRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &CancelOutstandingPollRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &CancelOutstandingPollRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestCancelOutstandingPollRequest_GetTaskListType(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *CancelOutstandingPollRequest\n\t\twant int32\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list type\",\n\t\t\treq:  &CancelOutstandingPollRequest{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"task list type\",\n\t\t\treq:  &CancelOutstandingPollRequest{TaskListType: func() *int32 { var v int32; v = int32(TaskListTypeActivity); return &v }()},\n\t\t\twant: int32(TaskListTypeActivity),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskListType()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestCancelOutstandingPollRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *CancelOutstandingPollRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &CancelOutstandingPollRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &CancelOutstandingPollRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestCancelOutstandingPollRequest_GetPollerID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *CancelOutstandingPollRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty poller id\",\n\t\t\treq:  &CancelOutstandingPollRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"poller id\",\n\t\t\treq:  &CancelOutstandingPollRequest{PollerID: \"test-poller-id\"},\n\t\t\twant: \"test-poller-id\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPollerID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingDescribeTaskListRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingDescribeTaskListRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingDescribeTaskListRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingDescribeTaskListRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingDescribeTaskListRequest_GetDescRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingDescribeTaskListRequest\n\t\twant *DescribeTaskListRequest\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty desc request\",\n\t\t\treq:  &MatchingDescribeTaskListRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"desc request\",\n\t\t\treq:  &MatchingDescribeTaskListRequest{DescRequest: &DescribeTaskListRequest{Domain: \"test-domain\"}},\n\t\t\twant: &DescribeTaskListRequest{Domain: \"test-domain\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDescRequest()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingListTaskListPartitionsRequest_GetDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingListTaskListPartitionsRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingListTaskListPartitionsRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingListTaskListPartitionsRequest{Domain: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomain()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingListTaskListPartitionsRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingListTaskListPartitionsRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &MatchingListTaskListPartitionsRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &MatchingListTaskListPartitionsRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingGetTaskListsByDomainRequest_GetDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingGetTaskListsByDomainRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingGetTaskListsByDomainRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingGetTaskListsByDomainRequest{Domain: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomain()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetPollerID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty poller id\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"poller id\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{PollerID: \"test-poller-id\"},\n\t\t\twant: \"test-poller-id\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPollerID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetPollRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant *PollForActivityTaskRequest\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty poll request\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"poll request\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{PollRequest: &PollForActivityTaskRequest{Domain: \"test-domain\"}},\n\t\t\twant: &PollForActivityTaskRequest{Domain: \"test-domain\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPollRequest()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{PollRequest: &PollForActivityTaskRequest{TaskList: &TaskList{Name: \"test-task-list\"}}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetForwardedFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty forwarded from\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"forwarded from\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{ForwardedFrom: \"test-forwarded-from\"},\n\t\t\twant: \"test-forwarded-from\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetForwardedFrom()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForActivityTaskRequest_GetIsolationGroup(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForActivityTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty isolation group\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"isolation group\",\n\t\t\treq:  &MatchingPollForActivityTaskRequest{IsolationGroup: \"test-isolation-group\"},\n\t\t\twant: \"test-isolation-group\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetIsolationGroup()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetPollerID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty poller id\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"poller id\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{PollerID: \"test-poller-id\"},\n\t\t\twant: \"test-poller-id\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPollerID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetPollRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant *PollForDecisionTaskRequest\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty poll request\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"poll request\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{PollRequest: &PollForDecisionTaskRequest{Domain: \"test-domain\"}},\n\t\t\twant: &PollForDecisionTaskRequest{Domain: \"test-domain\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetPollRequest()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{PollRequest: &PollForDecisionTaskRequest{TaskList: &TaskList{Name: \"test-task-list\"}}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetForwardedFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty forwarded from\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"forwarded from\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{ForwardedFrom: \"test-forwarded-from\"},\n\t\t\twant: \"test-forwarded-from\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetForwardedFrom()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskRequest_GetIsolationGroup(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingPollForDecisionTaskRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty isolation group\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"isolation group\",\n\t\t\treq:  &MatchingPollForDecisionTaskRequest{IsolationGroup: \"test-isolation-group\"},\n\t\t\twant: \"test-isolation-group\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetIsolationGroup()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant *WorkflowExecution\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty workflow execution\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"workflow execution\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{WorkflowExecution: &WorkflowExecution{WorkflowID: \"test-workflow-id\"}},\n\t\t\twant: &WorkflowExecution{WorkflowID: \"test-workflow-id\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetWorkflowExecution()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetPreviousStartedEventID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant int64\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty previous started event id\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"previous started event id\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{PreviousStartedEventID: func() *int64 { v := int64(123); return &v }()},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetPreviousStartedEventID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetNextEventID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant int64\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty next event id\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"next event id\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{NextEventID: int64(123)},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetNextEventID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetStickyExecutionEnabled(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant bool\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"empty sticky execution enabled\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"sticky execution enabled\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{StickyExecutionEnabled: true},\n\t\t\twant: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetStickyExecutionEnabled()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetBranchToken(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant []byte\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty branch token\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"branch token\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{BranchToken: []byte(\"test-branch-token\")},\n\t\t\twant: []byte(\"test-branch-token\"),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetBranchToken()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingPollForDecisionTaskResponse_GetTotalHistoryBytes(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *MatchingPollForDecisionTaskResponse\n\t\twant int64\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"empty total history bytes\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{},\n\t\t\twant: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"total history bytes\",\n\t\t\tresp: &MatchingPollForDecisionTaskResponse{TotalHistoryBytes: 123},\n\t\t\twant: 123,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetTotalHistoryBytes()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingQueryWorkflowRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingQueryWorkflowRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest_GetQueryRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingQueryWorkflowRequest\n\t\twant *QueryWorkflowRequest\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty query request\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"query request\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{QueryRequest: &QueryWorkflowRequest{Domain: \"test-domain\"}},\n\t\t\twant: &QueryWorkflowRequest{Domain: \"test-domain\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetQueryRequest()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingQueryWorkflowRequest_GetForwardedFrom(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingQueryWorkflowRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty forwarded from\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"forwarded from\",\n\t\t\treq:  &MatchingQueryWorkflowRequest{ForwardedFrom: \"test-forwarded-from\"},\n\t\t\twant: \"test-forwarded-from\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetForwardedFrom()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest_GetDomainUUID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingRespondQueryTaskCompletedRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty domain\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{DomainUUID: \"test-domain\"},\n\t\t\twant: \"test-domain\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetDomainUUID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest_GetTaskList(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingRespondQueryTaskCompletedRequest\n\t\twant *TaskList\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"task list\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{TaskList: &TaskList{Name: \"test-task-list\"}},\n\t\t\twant: &TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskList()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest_GetTaskID(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingRespondQueryTaskCompletedRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty task id\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"task id\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{TaskID: \"test-task-id\"},\n\t\t\twant: \"test-task-id\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskID()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingRespondQueryTaskCompletedRequest_GetCompletedRequest(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingRespondQueryTaskCompletedRequest\n\t\twant *RespondQueryTaskCompletedRequest\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"empty completed request\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{},\n\t\t\twant: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"completed request\",\n\t\t\treq:  &MatchingRespondQueryTaskCompletedRequest{CompletedRequest: &RespondQueryTaskCompletedRequest{ErrorMessage: \"test-error-message\"}},\n\t\t\twant: &RespondQueryTaskCompletedRequest{ErrorMessage: \"test-error-message\"},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetCompletedRequest()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestTaskSource_Ptr(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ts    TaskSource\n\t\twant *TaskSource\n\t}{\n\t\t{\n\t\t\tname: \"nil\",\n\t\t\ts:    TaskSource(0),\n\t\t\twant: func() *TaskSource { i := TaskSource(0); return &i }(),\n\t\t},\n\t\t{\n\t\t\tname: \"non-nil\",\n\t\t\ts:    TaskSource(123),\n\t\t\twant: func() *TaskSource { i := TaskSource(123); return &i }(),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.s.Ptr()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestTaskSource_String(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ts    TaskSource\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"history\",\n\t\t\ts:    TaskSource(0),\n\t\t\twant: \"HISTORY\",\n\t\t},\n\t\t{\n\t\t\tname: \"db-backlog\",\n\t\t\ts:    TaskSource(1),\n\t\t\twant: \"DB_BACKLOG\",\n\t\t},\n\t\t{\n\t\t\tname: \"default\",\n\t\t\ts:    TaskSource(123),\n\t\t\twant: \"TaskSource(123)\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.s.String()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestTaskSource_UnmarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ttext string\n\t\twant TaskSource\n\t\terr  error\n\t}{\n\t\t{\n\t\t\tname: \"history\",\n\t\t\ttext: \"HISTORY\",\n\t\t\twant: TaskSource(0),\n\t\t\terr:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"db_backlog\",\n\t\t\ttext: \"DB_BACKLOG\",\n\t\t\twant: TaskSource(1),\n\t\t\terr:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"number in string\",\n\t\t\ttext: \"123\",\n\t\t\twant: TaskSource(123),\n\t\t\terr:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\ttext: \"unknown\",\n\t\t\terr:  errors.New(\"unknown enum value \\\"UNKNOWN\\\" for \\\"TaskSource\\\": strconv.ParseInt: parsing \\\"UNKNOWN\\\": invalid syntax\"),\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar s TaskSource\n\t\t\terr := s.UnmarshalText([]byte(tt.text))\n\t\t\tif tt.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tt.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, s)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTaskSource_MarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ts    TaskSource\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"history\",\n\t\t\ts:    TaskSource(0),\n\t\t\twant: \"HISTORY\",\n\t\t},\n\t\t{\n\t\t\tname: \"db-backlog\",\n\t\t\ts:    TaskSource(1),\n\t\t\twant: \"DB_BACKLOG\",\n\t\t},\n\t\t{\n\t\t\tname: \"default\",\n\t\t\ts:    TaskSource(123),\n\t\t\twant: \"TaskSource(123)\",\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tb, err := tt.s.MarshalText()\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tt.want, string(b))\n\t\t})\n\t}\n}\n\nfunc TestMatchingUpdateTaskListPartitionConfigRequest_GetTaskListType(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingUpdateTaskListPartitionConfigRequest\n\t\twant TaskListType\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: TaskListTypeDecision,\n\t\t},\n\t\t{\n\t\t\tname: \"empty request\",\n\t\t\treq:  &MatchingUpdateTaskListPartitionConfigRequest{},\n\t\t\twant: TaskListTypeDecision,\n\t\t},\n\t\t{\n\t\t\tname: \"non empty request\",\n\t\t\treq:  &MatchingUpdateTaskListPartitionConfigRequest{TaskListType: TaskListTypeActivity.Ptr()},\n\t\t\twant: TaskListTypeActivity,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskListType()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestMatchingRefreshTaskListPartitionConfigRequest_GetTaskListType(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *MatchingRefreshTaskListPartitionConfigRequest\n\t\twant TaskListType\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: TaskListTypeDecision,\n\t\t},\n\t\t{\n\t\t\tname: \"empty request\",\n\t\t\treq:  &MatchingRefreshTaskListPartitionConfigRequest{},\n\t\t\twant: TaskListTypeDecision,\n\t\t},\n\t\t{\n\t\t\tname: \"non empty request\",\n\t\t\treq:  &MatchingRefreshTaskListPartitionConfigRequest{TaskListType: TaskListTypeActivity.Ptr()},\n\t\t\twant: TaskListTypeActivity,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetTaskListType()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/predicate.go",
    "content": "package types\n\ntype PredicateType int32\n\nfunc (p PredicateType) Ptr() *PredicateType {\n\treturn &p\n}\n\nconst (\n\tPredicateTypeUniversal PredicateType = iota\n\tPredicateTypeEmpty\n\tPredicateTypeDomainID\n\n\tNumPredicateTypes\n)\n\ntype UniversalPredicateAttributes struct{}\n\nfunc (u *UniversalPredicateAttributes) Copy() *UniversalPredicateAttributes {\n\tif u == nil {\n\t\treturn nil\n\t}\n\treturn &UniversalPredicateAttributes{}\n}\n\ntype EmptyPredicateAttributes struct{}\n\nfunc (e *EmptyPredicateAttributes) Copy() *EmptyPredicateAttributes {\n\tif e == nil {\n\t\treturn nil\n\t}\n\treturn &EmptyPredicateAttributes{}\n}\n\ntype DomainIDPredicateAttributes struct {\n\tDomainIDs   []string\n\tIsExclusive *bool\n}\n\nfunc (d *DomainIDPredicateAttributes) Copy() *DomainIDPredicateAttributes {\n\tif d == nil {\n\t\treturn nil\n\t}\n\treturn &DomainIDPredicateAttributes{\n\t\tDomainIDs:   d.DomainIDs,\n\t\tIsExclusive: d.IsExclusive,\n\t}\n}\n\nfunc (d *DomainIDPredicateAttributes) GetIsExclusive() bool {\n\tif d.IsExclusive == nil {\n\t\treturn false\n\t}\n\treturn *d.IsExclusive\n}\n\ntype Predicate struct {\n\tPredicateType                PredicateType\n\tUniversalPredicateAttributes *UniversalPredicateAttributes\n\tEmptyPredicateAttributes     *EmptyPredicateAttributes\n\tDomainIDPredicateAttributes  *DomainIDPredicateAttributes\n}\n\nfunc (p *Predicate) Copy() *Predicate {\n\tif p == nil {\n\t\treturn nil\n\t}\n\treturn &Predicate{\n\t\tPredicateType:                p.PredicateType,\n\t\tUniversalPredicateAttributes: p.UniversalPredicateAttributes.Copy(),\n\t\tEmptyPredicateAttributes:     p.EmptyPredicateAttributes.Copy(),\n\t\tDomainIDPredicateAttributes:  p.DomainIDPredicateAttributes.Copy(),\n\t}\n}\n\nfunc (p *Predicate) GetDomainIDPredicateAttributes() *DomainIDPredicateAttributes {\n\tif p == nil {\n\t\treturn nil\n\t}\n\tif p.PredicateType != PredicateTypeDomainID {\n\t\treturn nil\n\t}\n\treturn p.DomainIDPredicateAttributes\n}\n"
  },
  {
    "path": "common/types/replicator.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"unsafe\"\n)\n\n// DLQType is an internal type (TBD...)\ntype DLQType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DLQType) Ptr() *DLQType {\n\treturn &e\n}\n\n// String returns a readable string representation of DLQType.\nfunc (e DLQType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Replication\"\n\tcase 1:\n\t\treturn \"Domain\"\n\t}\n\treturn fmt.Sprintf(\"DLQType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DLQType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"REPLICATION\":\n\t\t*e = DLQTypeReplication\n\t\treturn nil\n\tcase \"DOMAIN\":\n\t\t*e = DLQTypeDomain\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DLQType\", err)\n\t\t}\n\t\t*e = DLQType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DLQType to text.\nfunc (e DLQType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// DLQTypeReplication is an option for DLQType\n\tDLQTypeReplication DLQType = iota\n\t// DLQTypeDomain is an option for DLQType\n\tDLQTypeDomain\n)\n\n// DomainOperation is an internal type (TBD...)\ntype DomainOperation int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DomainOperation) Ptr() *DomainOperation {\n\treturn &e\n}\n\n// String returns a readable string representation of DomainOperation.\nfunc (e DomainOperation) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Create\"\n\tcase 1:\n\t\treturn \"Update\"\n\tcase 2:\n\t\treturn \"Delete\"\n\t}\n\treturn fmt.Sprintf(\"DomainOperation(%d)\", w)\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (e *DomainOperation) ByteSize() uint64 {\n\tif e == nil {\n\t\treturn 0\n\t}\n\treturn uint64(unsafe.Sizeof(*e))\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DomainOperation) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"CREATE\":\n\t\t*e = DomainOperationCreate\n\t\treturn nil\n\tcase \"UPDATE\":\n\t\t*e = DomainOperationUpdate\n\t\treturn nil\n\tcase \"DELETE\":\n\t\t*e = DomainOperationDelete\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DomainOperation\", err)\n\t\t}\n\t\t*e = DomainOperation(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DomainOperation to text.\nfunc (e DomainOperation) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// DomainOperationCreate is an option for DomainOperation\n\tDomainOperationCreate DomainOperation = iota\n\t// DomainOperationUpdate is an option for DomainOperation\n\tDomainOperationUpdate\n\t// DomainOperationDelete is an option for DomainOperation\n\tDomainOperationDelete\n)\n\n// DomainTaskAttributes is an internal type (TBD...)\ntype DomainTaskAttributes struct {\n\tDomainOperation         *DomainOperation                `json:\"domainOperation,omitempty\"`\n\tID                      string                          `json:\"id,omitempty\"`\n\tInfo                    *DomainInfo                     `json:\"info,omitempty\"`\n\tConfig                  *DomainConfiguration            `json:\"config,omitempty\"`\n\tReplicationConfig       *DomainReplicationConfiguration `json:\"replicationConfig,omitempty\"`\n\tConfigVersion           int64                           `json:\"configVersion,omitempty\"`\n\tFailoverVersion         int64                           `json:\"failoverVersion,omitempty\"`\n\tPreviousFailoverVersion int64                           `json:\"previousFailoverVersion,omitempty\"`\n}\n\n// GetDomainOperation is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetDomainOperation() (o DomainOperation) {\n\tif v != nil && v.DomainOperation != nil {\n\t\treturn *v.DomainOperation\n\t}\n\treturn\n}\n\n// GetID is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetID() (o string) {\n\tif v != nil {\n\t\treturn v.ID\n\t}\n\treturn\n}\n\n// GetInfo is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetInfo() (o *DomainInfo) {\n\tif v != nil && v.Info != nil {\n\t\treturn v.Info\n\t}\n\treturn\n}\n\n// GetConfigVersion is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetConfigVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.ConfigVersion\n\t}\n\treturn\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetPreviousFailoverVersion is an internal getter (TBD...)\nfunc (v *DomainTaskAttributes) GetPreviousFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.PreviousFailoverVersion\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *DomainTaskAttributes) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += v.DomainOperation.ByteSize()\n\tsize += uint64(len(v.ID))\n\tsize += v.Info.ByteSize()\n\tsize += v.Config.ByteSize()\n\tsize += v.ReplicationConfig.ByteSize()\n\n\treturn size\n}\n\n// FailoverMarkerAttributes is an internal type (TBD...)\ntype FailoverMarkerAttributes struct {\n\tDomainID        string `json:\"domainID,omitempty\"`\n\tFailoverVersion int64  `json:\"failoverVersion,omitempty\"`\n\tCreationTime    *int64 `json:\"creationTime,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *FailoverMarkerAttributes) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *FailoverMarkerAttributes) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetCreationTime is an internal getter (TBD...)\nfunc (v *FailoverMarkerAttributes) GetCreationTime() (o int64) {\n\tif v != nil && v.CreationTime != nil {\n\t\treturn *v.CreationTime\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *FailoverMarkerAttributes) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.DomainID))\n\tif v.CreationTime != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.CreationTime))\n\t}\n\n\treturn size\n}\n\n// FailoverMarkers is an internal type (TBD...)\ntype FailoverMarkers struct {\n\tFailoverMarkers []*FailoverMarkerAttributes `json:\"failoverMarkers,omitempty\"`\n}\n\n// GetDLQReplicationMessagesRequest is an internal type (TBD...)\ntype GetDLQReplicationMessagesRequest struct {\n\tTaskInfos []*ReplicationTaskInfo `json:\"taskInfos,omitempty\"`\n}\n\n// GetTaskInfos is an internal getter (TBD...)\nfunc (v *GetDLQReplicationMessagesRequest) GetTaskInfos() (o []*ReplicationTaskInfo) {\n\tif v != nil && v.TaskInfos != nil {\n\t\treturn v.TaskInfos\n\t}\n\treturn\n}\n\n// GetDLQReplicationMessagesResponse is an internal type (TBD...)\ntype GetDLQReplicationMessagesResponse struct {\n\tReplicationTasks []*ReplicationTask `json:\"replicationTasks,omitempty\"`\n}\n\n// GetDomainReplicationMessagesRequest is an internal type (TBD...)\ntype GetDomainReplicationMessagesRequest struct {\n\tLastRetrievedMessageID *int64 `json:\"lastRetrievedMessageId,omitempty\"`\n\tLastProcessedMessageID *int64 `json:\"lastProcessedMessageId,omitempty\"`\n\tClusterName            string `json:\"clusterName,omitempty\"`\n}\n\n// GetLastRetrievedMessageID is an internal getter (TBD...)\nfunc (v *GetDomainReplicationMessagesRequest) GetLastRetrievedMessageID() (o int64) {\n\tif v != nil && v.LastRetrievedMessageID != nil {\n\t\treturn *v.LastRetrievedMessageID\n\t}\n\treturn\n}\n\n// GetLastProcessedMessageID is an internal getter (TBD...)\nfunc (v *GetDomainReplicationMessagesRequest) GetLastProcessedMessageID() (o int64) {\n\tif v != nil && v.LastProcessedMessageID != nil {\n\t\treturn *v.LastProcessedMessageID\n\t}\n\treturn\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *GetDomainReplicationMessagesRequest) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// GetDomainReplicationMessagesResponse is an internal type (TBD...)\ntype GetDomainReplicationMessagesResponse struct {\n\tMessages *ReplicationMessages `json:\"messages,omitempty\"`\n}\n\n// GetReplicationMessagesRequest is an internal type (TBD...)\ntype GetReplicationMessagesRequest struct {\n\tTokens      []*ReplicationToken `json:\"tokens,omitempty\"`\n\tClusterName string              `json:\"clusterName,omitempty\"`\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *GetReplicationMessagesRequest) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// GetReplicationMessagesResponse is an internal type (TBD...)\ntype GetReplicationMessagesResponse struct {\n\tMessagesByShard map[int32]*ReplicationMessages `json:\"messagesByShard,omitempty\"`\n}\n\n// GetMessagesByShard is an internal getter (TBD...)\nfunc (v *GetReplicationMessagesResponse) GetMessagesByShard() (o map[int32]*ReplicationMessages) {\n\tif v != nil && v.MessagesByShard != nil {\n\t\treturn v.MessagesByShard\n\t}\n\treturn\n}\n\n// GetEarliestCreationTime returns the earliest creation time of replication tasks if it exists, otherwise return nil\nfunc (v *GetReplicationMessagesResponse) GetEarliestCreationTime() *int64 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\n\tvar earliestTime *int64\n\tfor _, messages := range v.MessagesByShard {\n\n\t\tcreationTime := messages.GetEarliestCreationTime()\n\t\tif creationTime == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tif earliestTime == nil || *creationTime < *earliestTime {\n\t\t\tearliestTime = creationTime\n\t\t}\n\t}\n\n\tif earliestTime == nil {\n\t\treturn nil\n\t}\n\n\t// avoid returning a pointer to the internal value\n\t// for immutability\n\tresult := *earliestTime\n\treturn &result\n}\n\n// HistoryTaskV2Attributes is an internal type (TBD...)\ntype HistoryTaskV2Attributes struct {\n\tDomainID            string                `json:\"domainId,omitempty\"`\n\tWorkflowID          string                `json:\"workflowId,omitempty\"`\n\tRunID               string                `json:\"runId,omitempty\"`\n\tVersionHistoryItems []*VersionHistoryItem `json:\"versionHistoryItems,omitempty\"`\n\tEvents              *DataBlob             `json:\"events,omitempty\"`\n\tNewRunEvents        *DataBlob             `json:\"newRunEvents,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetVersionHistoryItems is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetVersionHistoryItems() (o []*VersionHistoryItem) {\n\tif v != nil && v.VersionHistoryItems != nil {\n\t\treturn v.VersionHistoryItems\n\t}\n\treturn\n}\n\n// GetEvents is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetEvents() (o *DataBlob) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\treturn\n}\n\n// GetNewRunEvents is an internal getter (TBD...)\nfunc (v *HistoryTaskV2Attributes) GetNewRunEvents() (o *DataBlob) {\n\tif v != nil && v.NewRunEvents != nil {\n\t\treturn v.NewRunEvents\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *HistoryTaskV2Attributes) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.DomainID))\n\tsize += uint64(len(v.WorkflowID))\n\tsize += uint64(len(v.RunID))\n\n\tsize += uint64(len(v.VersionHistoryItems)) * uint64(unsafe.Sizeof((*VersionHistoryItem)(nil)))\n\tfor _, item := range v.VersionHistoryItems {\n\t\tsize += item.ByteSize()\n\t}\n\n\tsize += v.Events.ByteSize()\n\tsize += v.NewRunEvents.ByteSize()\n\n\treturn size\n}\n\ntype CountDLQMessagesRequest struct {\n\t// ForceFetch will force fetching current values from DB\n\t// instead of using cached values used for emitting metrics.\n\tForceFetch bool\n}\n\ntype CountDLQMessagesResponse struct {\n\tHistory map[HistoryDLQCountKey]int64\n\tDomain  int64\n}\n\ntype HistoryDLQCountKey struct {\n\tShardID       int32\n\tSourceCluster string\n}\n\ntype HistoryCountDLQMessagesResponse struct {\n\tEntries map[HistoryDLQCountKey]int64\n}\n\n// MergeDLQMessagesRequest is an internal type (TBD...)\ntype MergeDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               int32    `json:\"shardID,omitempty\"`\n\tSourceCluster         string   `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n\tMaximumPageSize       int32    `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken         []byte   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetSourceCluster is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil {\n\t\treturn v.SourceCluster\n\t}\n\treturn\n}\n\n// GetInclusiveEndMessageID is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *MergeDLQMessagesRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// MergeDLQMessagesResponse is an internal type (TBD...)\ntype MergeDLQMessagesResponse struct {\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// PurgeDLQMessagesRequest is an internal type (TBD...)\ntype PurgeDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               int32    `json:\"shardID,omitempty\"`\n\tSourceCluster         string   `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *PurgeDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *PurgeDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetSourceCluster is an internal getter (TBD...)\nfunc (v *PurgeDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil {\n\t\treturn v.SourceCluster\n\t}\n\treturn\n}\n\n// GetInclusiveEndMessageID is an internal getter (TBD...)\nfunc (v *PurgeDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\treturn\n}\n\n// ReadDLQMessagesRequest is an internal type (TBD...)\ntype ReadDLQMessagesRequest struct {\n\tType                  *DLQType `json:\"type,omitempty\"`\n\tShardID               int32    `json:\"shardID,omitempty\"`\n\tSourceCluster         string   `json:\"sourceCluster,omitempty\"`\n\tInclusiveEndMessageID *int64   `json:\"inclusiveEndMessageID,omitempty\"`\n\tMaximumPageSize       int32    `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken         []byte   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetType() (o DLQType) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetSourceCluster is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetSourceCluster() (o string) {\n\tif v != nil {\n\t\treturn v.SourceCluster\n\t}\n\treturn\n}\n\n// GetInclusiveEndMessageID is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetInclusiveEndMessageID() (o int64) {\n\tif v != nil && v.InclusiveEndMessageID != nil {\n\t\treturn *v.InclusiveEndMessageID\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *ReadDLQMessagesRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// ReadDLQMessagesResponse is an internal type (TBD...)\ntype ReadDLQMessagesResponse struct {\n\tType                 *DLQType               `json:\"type,omitempty\"`\n\tReplicationTasks     []*ReplicationTask     `json:\"replicationTasks,omitempty\"`\n\tReplicationTasksInfo []*ReplicationTaskInfo `json:\"replicationTasksInfo,omitempty\"`\n\tNextPageToken        []byte                 `json:\"nextPageToken,omitempty\"`\n}\n\n// ReplicationMessages is an internal type (TBD...)\ntype ReplicationMessages struct {\n\tReplicationTasks       []*ReplicationTask `json:\"replicationTasks,omitempty\"`\n\tLastRetrievedMessageID int64              `json:\"lastRetrievedMessageId,omitempty\"`\n\tHasMore                bool               `json:\"hasMore,omitempty\"`\n\tSyncShardStatus        *SyncShardStatus   `json:\"syncShardStatus,omitempty\"`\n}\n\n// GetReplicationTasks is an internal getter (TBD...)\nfunc (v *ReplicationMessages) GetReplicationTasks() (o []*ReplicationTask) {\n\tif v != nil && v.ReplicationTasks != nil {\n\t\treturn v.ReplicationTasks\n\t}\n\treturn\n}\n\n// GetLastRetrievedMessageID is an internal getter (TBD...)\nfunc (v *ReplicationMessages) GetLastRetrievedMessageID() (o int64) {\n\tif v != nil {\n\t\treturn v.LastRetrievedMessageID\n\t}\n\treturn\n}\n\n// GetHasMore is an internal getter (TBD...)\nfunc (v *ReplicationMessages) GetHasMore() (o bool) {\n\tif v != nil {\n\t\treturn v.HasMore\n\t}\n\treturn\n}\n\n// GetSyncShardStatus is an internal getter (TBD...)\nfunc (v *ReplicationMessages) GetSyncShardStatus() (o *SyncShardStatus) {\n\tif v != nil && v.SyncShardStatus != nil {\n\t\treturn v.SyncShardStatus\n\t}\n\treturn\n}\n\n// GetEarliestCreationTime returns the earliest message time in the replication tasks if it exists\n// otherwise return nil\nfunc (v *ReplicationMessages) GetEarliestCreationTime() *int64 {\n\tif v == nil {\n\t\treturn nil\n\t}\n\n\tvar earliestTime *int64\n\tfor _, task := range v.GetReplicationTasks() {\n\t\tif task.CreationTime == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tif earliestTime == nil || *task.CreationTime < *earliestTime {\n\t\t\tearliestTime = task.CreationTime\n\t\t}\n\t}\n\n\tif earliestTime == nil {\n\t\treturn nil\n\t}\n\n\t// avoid returning a pointer to the internal value\n\t// for immutability\n\tresult := *earliestTime\n\treturn &result\n}\n\n// ReplicationMessagesSizeFn is a function type to calculate the size of ReplicationMessages\ntype ReplicationMessagesSizeFn func(v *ReplicationMessages) int\n\n// ReplicationTaskSizeFn is a function type to calculate the size of a single ReplicationTask\ntype ReplicationTaskSizeFn func(v *ReplicationTask) int\n\n// ReplicationTask is an internal type (TBD...)\ntype ReplicationTask struct {\n\tTaskType                      *ReplicationTaskType           `json:\"taskType,omitempty\"`\n\tSourceTaskID                  int64                          `json:\"sourceTaskId,omitempty\"`\n\tDomainTaskAttributes          *DomainTaskAttributes          `json:\"domainTaskAttributes,omitempty\"`\n\tSyncShardStatusTaskAttributes *SyncShardStatusTaskAttributes `json:\"syncShardStatusTaskAttributes,omitempty\"`\n\tSyncActivityTaskAttributes    *SyncActivityTaskAttributes    `json:\"syncActivityTaskAttributes,omitempty\"`\n\tHistoryTaskV2Attributes       *HistoryTaskV2Attributes       `json:\"historyTaskV2Attributes,omitempty\"`\n\tFailoverMarkerAttributes      *FailoverMarkerAttributes      `json:\"failoverMarkerAttributes,omitempty\"`\n\tCreationTime                  *int64                         `json:\"creationTime,omitempty\"`\n}\n\n// GetTaskType is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetTaskType() (o ReplicationTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\treturn\n}\n\n// GetSourceTaskID is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetSourceTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.SourceTaskID\n\t}\n\treturn\n}\n\n// GetSequenceID implements cache.AckCacheItem interface by delegating to GetSourceTaskID\nfunc (v *ReplicationTask) GetSequenceID() int64 {\n\treturn v.GetSourceTaskID()\n}\n\n// GetDomainTaskAttributes is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetDomainTaskAttributes() (o *DomainTaskAttributes) {\n\tif v != nil && v.DomainTaskAttributes != nil {\n\t\treturn v.DomainTaskAttributes\n\t}\n\treturn\n}\n\n// GetSyncActivityTaskAttributes is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetSyncActivityTaskAttributes() (o *SyncActivityTaskAttributes) {\n\tif v != nil && v.SyncActivityTaskAttributes != nil {\n\t\treturn v.SyncActivityTaskAttributes\n\t}\n\treturn\n}\n\n// GetHistoryTaskV2Attributes is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetHistoryTaskV2Attributes() (o *HistoryTaskV2Attributes) {\n\tif v != nil && v.HistoryTaskV2Attributes != nil {\n\t\treturn v.HistoryTaskV2Attributes\n\t}\n\treturn\n}\n\n// GetFailoverMarkerAttributes is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetFailoverMarkerAttributes() (o *FailoverMarkerAttributes) {\n\tif v != nil && v.FailoverMarkerAttributes != nil {\n\t\treturn v.FailoverMarkerAttributes\n\t}\n\treturn\n}\n\n// GetCreationTime is an internal getter (TBD...)\nfunc (v *ReplicationTask) GetCreationTime() (o int64) {\n\tif v != nil && v.CreationTime != nil {\n\t\treturn *v.CreationTime\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *ReplicationTask) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += v.TaskType.ByteSize()\n\tsize += v.DomainTaskAttributes.ByteSize()\n\tsize += v.SyncShardStatusTaskAttributes.ByteSize()\n\tsize += v.SyncActivityTaskAttributes.ByteSize()\n\tsize += v.HistoryTaskV2Attributes.ByteSize()\n\tsize += v.FailoverMarkerAttributes.ByteSize()\n\tif v.CreationTime != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.CreationTime))\n\t}\n\n\treturn size\n}\n\n// ReplicationTaskInfo is an internal type (TBD...)\ntype ReplicationTaskInfo struct {\n\tDomainID     string `json:\"domainID,omitempty\"`\n\tWorkflowID   string `json:\"workflowID,omitempty\"`\n\tRunID        string `json:\"runID,omitempty\"`\n\tTaskType     int16  `json:\"taskType,omitempty\"`\n\tTaskID       int64  `json:\"taskID,omitempty\"`\n\tVersion      int64  `json:\"version,omitempty\"`\n\tFirstEventID int64  `json:\"firstEventID,omitempty\"`\n\tNextEventID  int64  `json:\"nextEventID,omitempty\"`\n\tScheduledID  int64  `json:\"scheduledID,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetTaskType is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetTaskType() (o int16) {\n\tif v != nil {\n\t\treturn v.TaskType\n\t}\n\treturn\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetVersion is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.Version\n\t}\n\treturn\n}\n\n// GetFirstEventID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetFirstEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.FirstEventID\n\t}\n\treturn\n}\n\n// GetNextEventID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.NextEventID\n\t}\n\treturn\n}\n\n// GetScheduledID is an internal getter (TBD...)\nfunc (v *ReplicationTaskInfo) GetScheduledID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledID\n\t}\n\treturn\n}\n\n// ReplicationTaskType is an internal type (TBD...)\ntype ReplicationTaskType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e ReplicationTaskType) Ptr() *ReplicationTaskType {\n\treturn &e\n}\n\n// String returns a readable string representation of ReplicationTaskType.\nfunc (e ReplicationTaskType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Domain\"\n\tcase 1:\n\t\treturn \"History\"\n\tcase 2:\n\t\treturn \"SyncShardStatus\"\n\tcase 3:\n\t\treturn \"SyncActivity\"\n\tcase 4:\n\t\treturn \"HistoryMetadata\"\n\tcase 5:\n\t\treturn \"HistoryV2\"\n\tcase 6:\n\t\treturn \"FailoverMarker\"\n\t}\n\treturn fmt.Sprintf(\"ReplicationTaskType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *ReplicationTaskType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"DOMAIN\":\n\t\t*e = ReplicationTaskTypeDomain\n\t\treturn nil\n\tcase \"HISTORY\":\n\t\t*e = ReplicationTaskTypeHistory\n\t\treturn nil\n\tcase \"SYNCSHARDSTATUS\":\n\t\t*e = ReplicationTaskTypeSyncShardStatus\n\t\treturn nil\n\tcase \"SYNCACTIVITY\":\n\t\t*e = ReplicationTaskTypeSyncActivity\n\t\treturn nil\n\tcase \"HISTORYMETADATA\":\n\t\t*e = ReplicationTaskTypeHistoryMetadata\n\t\treturn nil\n\tcase \"HISTORYV2\":\n\t\t*e = ReplicationTaskTypeHistoryV2\n\t\treturn nil\n\tcase \"FAILOVERMARKER\":\n\t\t*e = ReplicationTaskTypeFailoverMarker\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ReplicationTaskType\", err)\n\t\t}\n\t\t*e = ReplicationTaskType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ReplicationTaskType to text.\nfunc (e ReplicationTaskType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// ReplicationTaskTypeDomain is an option for ReplicationTaskType\n\tReplicationTaskTypeDomain ReplicationTaskType = iota\n\t// ReplicationTaskTypeHistory is an option for ReplicationTaskType\n\tReplicationTaskTypeHistory\n\t// ReplicationTaskTypeSyncShardStatus is an option for ReplicationTaskType\n\tReplicationTaskTypeSyncShardStatus\n\t// ReplicationTaskTypeSyncActivity is an option for ReplicationTaskType\n\tReplicationTaskTypeSyncActivity\n\t// ReplicationTaskTypeHistoryMetadata is an option for ReplicationTaskType\n\tReplicationTaskTypeHistoryMetadata\n\t// ReplicationTaskTypeHistoryV2 is an option for ReplicationTaskType\n\tReplicationTaskTypeHistoryV2\n\t// ReplicationTaskTypeFailoverMarker is an option for ReplicationTaskType\n\tReplicationTaskTypeFailoverMarker\n)\n\n// ByteSize returns the approximate memory used in bytes\nfunc (e *ReplicationTaskType) ByteSize() uint64 {\n\tif e == nil {\n\t\treturn 0\n\t}\n\treturn uint64(unsafe.Sizeof(*e))\n}\n\n// ReplicationToken is an internal type (TBD...)\ntype ReplicationToken struct {\n\tShardID                int32 `json:\"shardID,omitempty\"`\n\tLastRetrievedMessageID int64 `json:\"lastRetrievedMessageId,omitempty\"`\n\tLastProcessedMessageID int64 `json:\"lastProcessedMessageId,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *ReplicationToken) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetLastRetrievedMessageID is an internal getter (TBD...)\nfunc (v *ReplicationToken) GetLastRetrievedMessageID() (o int64) {\n\tif v != nil {\n\t\treturn v.LastRetrievedMessageID\n\t}\n\treturn\n}\n\n// GetLastProcessedMessageID is an internal getter (TBD...)\nfunc (v *ReplicationToken) GetLastProcessedMessageID() (o int64) {\n\tif v != nil {\n\t\treturn v.LastProcessedMessageID\n\t}\n\treturn\n}\n\n// SyncActivityTaskAttributes is an internal type (TBD...)\ntype SyncActivityTaskAttributes struct {\n\tDomainID           string          `json:\"domainId,omitempty\"`\n\tWorkflowID         string          `json:\"workflowId,omitempty\"`\n\tRunID              string          `json:\"runId,omitempty\"`\n\tVersion            int64           `json:\"version,omitempty\"`\n\tScheduledID        int64           `json:\"scheduledId,omitempty\"`\n\tScheduledTime      *int64          `json:\"scheduledTime,omitempty\"`\n\tStartedID          int64           `json:\"startedId,omitempty\"`\n\tStartedTime        *int64          `json:\"startedTime,omitempty\"`\n\tLastHeartbeatTime  *int64          `json:\"lastHeartbeatTime,omitempty\"`\n\tDetails            []byte          `json:\"details,omitempty\"`\n\tAttempt            int32           `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string         `json:\"lastFailureReason,omitempty\"`\n\tLastWorkerIdentity string          `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails []byte          `json:\"lastFailureDetails,omitempty\"`\n\tVersionHistory     *VersionHistory `json:\"versionHistory,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetVersion is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.Version\n\t}\n\treturn\n}\n\n// GetScheduledID is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetScheduledID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledID\n\t}\n\treturn\n}\n\n// GetScheduledTime is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetScheduledTime() (o int64) {\n\tif v != nil && v.ScheduledTime != nil {\n\t\treturn *v.ScheduledTime\n\t}\n\treturn\n}\n\n// GetStartedID is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetStartedID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedID\n\t}\n\treturn\n}\n\n// GetStartedTime is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetStartedTime() (o int64) {\n\tif v != nil && v.StartedTime != nil {\n\t\treturn *v.StartedTime\n\t}\n\treturn\n}\n\n// GetLastHeartbeatTime is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetLastHeartbeatTime() (o int64) {\n\tif v != nil && v.LastHeartbeatTime != nil {\n\t\treturn *v.LastHeartbeatTime\n\t}\n\treturn\n}\n\n// GetVersionHistory is an internal getter (TBD...)\nfunc (v *SyncActivityTaskAttributes) GetVersionHistory() (o *VersionHistory) {\n\tif v != nil && v.VersionHistory != nil {\n\t\treturn v.VersionHistory\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *SyncActivityTaskAttributes) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.DomainID))\n\tsize += uint64(len(v.WorkflowID))\n\tsize += uint64(len(v.RunID))\n\tif v.ScheduledTime != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.ScheduledTime))\n\t}\n\tif v.StartedTime != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.StartedTime))\n\t}\n\tif v.LastHeartbeatTime != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.LastHeartbeatTime))\n\t}\n\tsize += uint64(len(v.Details))\n\tif v.LastFailureReason != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.LastFailureReason))\n\t\tsize += uint64(len(*v.LastFailureReason))\n\t}\n\tsize += uint64(len(v.LastWorkerIdentity))\n\tsize += uint64(len(v.LastFailureDetails))\n\tsize += v.VersionHistory.ByteSize()\n\n\treturn size\n}\n\n// SyncShardStatus is an internal type (TBD...)\ntype SyncShardStatus struct {\n\tTimestamp *int64 `json:\"timestamp,omitempty\"`\n}\n\n// GetTimestamp is an internal getter (TBD...)\nfunc (v *SyncShardStatus) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\treturn\n}\n\n// SyncShardStatusTaskAttributes is an internal type (TBD...)\ntype SyncShardStatusTaskAttributes struct {\n\tSourceCluster string `json:\"sourceCluster,omitempty\"`\n\tShardID       int64  `json:\"shardId,omitempty\"`\n\tTimestamp     *int64 `json:\"timestamp,omitempty\"`\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *SyncShardStatusTaskAttributes) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.SourceCluster))\n\tif v.Timestamp != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.Timestamp))\n\t}\n\n\treturn size\n}\n"
  },
  {
    "path": "common/types/replicator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/thriftrw/ptr\"\n)\n\nfunc TestDLQType_Ptr(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\tptr := dlqType.Ptr()\n\n\tassert.Equal(t, &dlqType, ptr)\n}\n\nfunc TestDLQType_String(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\tassert.Equal(t, \"Replication\", dlqType.String())\n\n\tdlqType = DLQTypeDomain\n\tassert.Equal(t, \"Domain\", dlqType.String())\n\n\tdlqType = 2\n\tassert.Equal(t, \"DLQType(2)\", dlqType.String())\n}\n\nfunc TestDLQType_UnmarshalText(t *testing.T) {\n\tvar dlqType DLQType\n\terr := dlqType.UnmarshalText([]byte(\"Replication\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DLQTypeReplication, dlqType)\n\n\terr = dlqType.UnmarshalText([]byte(\"Domain\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DLQTypeDomain, dlqType)\n\n\terr = dlqType.UnmarshalText([]byte(\"2\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DLQType(2), dlqType)\n\n\terr = dlqType.UnmarshalText([]byte(\"Invalid\"))\n\tassert.Error(t, err)\n}\n\nfunc TestDLQType_MarshalText(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\ttext, err := dlqType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"Replication\"), text)\n\n\tdlqType = DLQTypeDomain\n\ttext, err = dlqType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"Domain\"), text)\n}\n\nfunc TestDomainOperation_Ptr(t *testing.T) {\n\tdomainOp := DomainOperationCreate\n\tptr := domainOp.Ptr()\n\n\tassert.Equal(t, &domainOp, ptr)\n}\n\nfunc TestDomainOperation_String(t *testing.T) {\n\tdomainOp := DomainOperationCreate\n\tassert.Equal(t, \"Create\", domainOp.String())\n\n\tdomainOp = DomainOperationUpdate\n\tassert.Equal(t, \"Update\", domainOp.String())\n\n\tdomainOp = DomainOperationDelete\n\tassert.Equal(t, \"Delete\", domainOp.String())\n\n\tdomainOp = 3\n\tassert.Equal(t, \"DomainOperation(3)\", domainOp.String())\n}\n\nfunc TestDomainOperation_UnmarshalText(t *testing.T) {\n\tvar domainOp DomainOperation\n\terr := domainOp.UnmarshalText([]byte(\"Create\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DomainOperationCreate, domainOp)\n\n\terr = domainOp.UnmarshalText([]byte(\"Update\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DomainOperationUpdate, domainOp)\n\n\terr = domainOp.UnmarshalText([]byte(\"2\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, DomainOperation(2), domainOp)\n\n\terr = domainOp.UnmarshalText([]byte(\"Invalid\"))\n\tassert.Error(t, err)\n}\n\nfunc TestDomainOperation_MarshalText(t *testing.T) {\n\tdomainOp := DomainOperationCreate\n\ttext, err := domainOp.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"Create\"), text)\n\n\tdomainOp = DomainOperationUpdate\n\ttext, err = domainOp.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"Update\"), text)\n}\n\nfunc TestDomainTaskAttributes_GetDomainOperation(t *testing.T) {\n\tdomainOp := DomainOperationCreate\n\ttestStruct := DomainTaskAttributes{\n\t\tDomainOperation: &domainOp,\n\t}\n\n\tres := testStruct.GetDomainOperation()\n\tassert.Equal(t, domainOp, res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetDomainOperation()\n\tassert.Equal(t, DomainOperation(0), res)\n}\n\nfunc TestDomainTaskAttributes_GetID(t *testing.T) {\n\ttestStruct := DomainTaskAttributes{\n\t\tID: \"test-id\",\n\t}\n\n\tres := testStruct.GetID()\n\tassert.Equal(t, \"test-id\", res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestDomainTaskAttributes_GetInfo(t *testing.T) {\n\tdomainInfo := &DomainInfo{}\n\ttestStruct := DomainTaskAttributes{\n\t\tInfo: domainInfo,\n\t}\n\n\tres := testStruct.GetInfo()\n\tassert.Equal(t, domainInfo, res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetInfo()\n\tassert.Nil(t, res)\n}\n\nfunc TestDomainTaskAttributes_GetConfigVersion(t *testing.T) {\n\ttestStruct := DomainTaskAttributes{\n\t\tConfigVersion: 123,\n\t}\n\n\tres := testStruct.GetConfigVersion()\n\tassert.Equal(t, int64(123), res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetConfigVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestDomainTaskAttributes_GetFailoverVersion(t *testing.T) {\n\ttestStruct := DomainTaskAttributes{\n\t\tFailoverVersion: 456,\n\t}\n\n\tres := testStruct.GetFailoverVersion()\n\tassert.Equal(t, int64(456), res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetFailoverVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestDomainTaskAttributes_GetPreviousFailoverVersion(t *testing.T) {\n\ttestStruct := DomainTaskAttributes{\n\t\tPreviousFailoverVersion: 789,\n\t}\n\n\tres := testStruct.GetPreviousFailoverVersion()\n\tassert.Equal(t, int64(789), res)\n\n\tvar nilStruct *DomainTaskAttributes\n\tres = nilStruct.GetPreviousFailoverVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestFailoverMarkerAttributes_GetDomainID(t *testing.T) {\n\ttestStruct := FailoverMarkerAttributes{\n\t\tDomainID: \"domain-id\",\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, \"domain-id\", res)\n\n\tvar nilStruct *FailoverMarkerAttributes\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestFailoverMarkerAttributes_GetFailoverVersion(t *testing.T) {\n\ttestStruct := FailoverMarkerAttributes{\n\t\tFailoverVersion: 1234,\n\t}\n\n\tres := testStruct.GetFailoverVersion()\n\tassert.Equal(t, int64(1234), res)\n\n\tvar nilStruct *FailoverMarkerAttributes\n\tres = nilStruct.GetFailoverVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestFailoverMarkerAttributes_GetCreationTime(t *testing.T) {\n\tcreationTime := int64(5678)\n\ttestStruct := FailoverMarkerAttributes{\n\t\tCreationTime: &creationTime,\n\t}\n\n\tres := testStruct.GetCreationTime()\n\tassert.Equal(t, creationTime, res)\n\n\tvar nilStruct *FailoverMarkerAttributes\n\tres = nilStruct.GetCreationTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestMergeDLQMessagesRequest_GetType(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\ttestStruct := MergeDLQMessagesRequest{\n\t\tType: &dlqType,\n\t}\n\n\tres := testStruct.GetType()\n\tassert.Equal(t, dlqType, res)\n\n\tvar nilStruct *MergeDLQMessagesRequest\n\tres = nilStruct.GetType()\n\tassert.Equal(t, DLQType(0), res)\n}\n\nfunc TestMergeDLQMessagesRequest_GetShardID(t *testing.T) {\n\ttestStruct := MergeDLQMessagesRequest{\n\t\tShardID: 101,\n\t}\n\n\tres := testStruct.GetShardID()\n\tassert.Equal(t, int32(101), res)\n\n\tvar nilStruct *MergeDLQMessagesRequest\n\tres = nilStruct.GetShardID()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestMergeDLQMessagesRequest_GetSourceCluster(t *testing.T) {\n\ttestStruct := MergeDLQMessagesRequest{\n\t\tSourceCluster: \"cluster-1\",\n\t}\n\n\tres := testStruct.GetSourceCluster()\n\tassert.Equal(t, \"cluster-1\", res)\n\n\tvar nilStruct *MergeDLQMessagesRequest\n\tres = nilStruct.GetSourceCluster()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestMergeDLQMessagesRequest_GetInclusiveEndMessageID(t *testing.T) {\n\tendMessageID := int64(102)\n\ttestStruct := MergeDLQMessagesRequest{\n\t\tInclusiveEndMessageID: &endMessageID,\n\t}\n\n\tres := testStruct.GetInclusiveEndMessageID()\n\tassert.Equal(t, endMessageID, res)\n\n\tvar nilStruct *MergeDLQMessagesRequest\n\tres = nilStruct.GetInclusiveEndMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetDLQReplicationMessagesRequest_GetTaskInfos(t *testing.T) {\n\ttaskInfos := []*ReplicationTaskInfo{{}, {}}\n\ttestStruct := GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: taskInfos,\n\t}\n\n\tres := testStruct.GetTaskInfos()\n\tassert.Equal(t, taskInfos, res)\n\n\tvar nilStruct *GetDLQReplicationMessagesRequest\n\tres = nilStruct.GetTaskInfos()\n\tassert.Nil(t, res)\n}\n\nfunc TestGetDomainReplicationMessagesRequest_GetLastRetrievedMessageID(t *testing.T) {\n\tlastRetrievedMessageID := int64(12345)\n\ttestStruct := GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: &lastRetrievedMessageID,\n\t}\n\n\tres := testStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, lastRetrievedMessageID, res)\n\n\tvar nilStruct *GetDomainReplicationMessagesRequest\n\tres = nilStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetDomainReplicationMessagesRequest_GetLastProcessedMessageID(t *testing.T) {\n\tlastProcessedMessageID := int64(67890)\n\ttestStruct := GetDomainReplicationMessagesRequest{\n\t\tLastProcessedMessageID: &lastProcessedMessageID,\n\t}\n\n\tres := testStruct.GetLastProcessedMessageID()\n\tassert.Equal(t, lastProcessedMessageID, res)\n\n\tvar nilStruct *GetDomainReplicationMessagesRequest\n\tres = nilStruct.GetLastProcessedMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestGetDomainReplicationMessagesRequest_GetClusterName(t *testing.T) {\n\ttestStruct := GetDomainReplicationMessagesRequest{\n\t\tClusterName: \"test-cluster\",\n\t}\n\n\tres := testStruct.GetClusterName()\n\tassert.Equal(t, \"test-cluster\", res)\n\n\tvar nilStruct *GetDomainReplicationMessagesRequest\n\tres = nilStruct.GetClusterName()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetReplicationMessagesRequest_GetClusterName(t *testing.T) {\n\ttestStruct := GetReplicationMessagesRequest{\n\t\tClusterName: \"test-cluster\",\n\t}\n\n\tres := testStruct.GetClusterName()\n\tassert.Equal(t, \"test-cluster\", res)\n\n\tvar nilStruct *GetReplicationMessagesRequest\n\tres = nilStruct.GetClusterName()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestGetReplicationMessagesResponse_GetMessagesByShard(t *testing.T) {\n\tmessagesByShard := map[int32]*ReplicationMessages{\n\t\t1: {},\n\t\t2: {},\n\t}\n\ttestStruct := GetReplicationMessagesResponse{\n\t\tMessagesByShard: messagesByShard,\n\t}\n\n\tres := testStruct.GetMessagesByShard()\n\tassert.Equal(t, messagesByShard, res)\n\n\tvar nilStruct *GetReplicationMessagesResponse\n\tres = nilStruct.GetMessagesByShard()\n\tassert.Nil(t, res)\n}\n\nfunc TestGetReplicationMessagesResponse_GetEarliestCreationTime(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tresponse *GetReplicationMessagesResponse\n\t\twant     *int64\n\t}{\n\t\t\"nil\":           {response: nil, want: nil},\n\t\t\"zero messages\": {response: &GetReplicationMessagesResponse{}, want: nil},\n\t\t\"no messages with creation time\": {\n\t\t\tresponse: &GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*ReplicationMessages{\n\t\t\t\t\t1: {ReplicationTasks: []*ReplicationTask{{}, {}, {}}},\n\t\t\t\t\t2: {ReplicationTasks: []*ReplicationTask{{}, {}, {}}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"a few messages with creation time\": {\n\t\t\tresponse: &GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*ReplicationMessages{\n\t\t\t\t\t1: {ReplicationTasks: []*ReplicationTask{\n\t\t\t\t\t\t{}, {CreationTime: ptr.Int64(20)}, {CreationTime: ptr.Int64(1000)}},\n\t\t\t\t\t},\n\t\t\t\t\t2: {ReplicationTasks: []*ReplicationTask{\n\t\t\t\t\t\t{CreationTime: ptr.Int64(10)}, {}, {CreationTime: ptr.Int64(12000)}}},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: ptrInt64(10),\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, c.want, c.response.GetEarliestCreationTime())\n\t\t})\n\t}\n}\n\nfunc TestCountDLQMessagesResponse(t *testing.T) {\n\thistory := map[HistoryDLQCountKey]int64{\n\t\t{ShardID: 1, SourceCluster: \"cluster-1\"}: 100,\n\t}\n\ttestStruct := CountDLQMessagesResponse{\n\t\tHistory: history,\n\t\tDomain:  200,\n\t}\n\n\tassert.Equal(t, history, testStruct.History)\n\tassert.Equal(t, int64(200), testStruct.Domain)\n\n\t// Test for empty history\n\temptyStruct := CountDLQMessagesResponse{}\n\tassert.Nil(t, emptyStruct.History)\n\tassert.Equal(t, int64(0), emptyStruct.Domain)\n}\n\nfunc TestHistoryCountDLQMessagesResponse(t *testing.T) {\n\tentries := map[HistoryDLQCountKey]int64{\n\t\t{ShardID: 1, SourceCluster: \"cluster-1\"}: 100,\n\t}\n\ttestStruct := HistoryCountDLQMessagesResponse{\n\t\tEntries: entries,\n\t}\n\n\tassert.Equal(t, entries, testStruct.Entries)\n\n\t// Test for empty entries\n\temptyStruct := HistoryCountDLQMessagesResponse{}\n\tassert.Nil(t, emptyStruct.Entries)\n}\n\nfunc TestMergeDLQMessagesRequest_Getters(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\tendMessageID := int64(102)\n\tnextPageToken := []byte(\"token\")\n\ttestStruct := MergeDLQMessagesRequest{\n\t\tType:                  &dlqType,\n\t\tShardID:               101,\n\t\tSourceCluster:         \"cluster-1\",\n\t\tInclusiveEndMessageID: &endMessageID,\n\t\tMaximumPageSize:       50,\n\t\tNextPageToken:         nextPageToken,\n\t}\n\n\tassert.Equal(t, dlqType, testStruct.GetType())\n\tassert.Equal(t, int32(101), testStruct.GetShardID())\n\tassert.Equal(t, \"cluster-1\", testStruct.GetSourceCluster())\n\tassert.Equal(t, endMessageID, testStruct.GetInclusiveEndMessageID())\n\tassert.Equal(t, int32(50), testStruct.GetMaximumPageSize())\n\tassert.Equal(t, nextPageToken, testStruct.GetNextPageToken())\n\n\t// Test for nil values\n\tvar nilStruct *MergeDLQMessagesRequest\n\tassert.Equal(t, DLQType(0), nilStruct.GetType())\n\tassert.Equal(t, int32(0), nilStruct.GetShardID())\n\tassert.Equal(t, \"\", nilStruct.GetSourceCluster())\n\tassert.Equal(t, int64(0), nilStruct.GetInclusiveEndMessageID())\n\tassert.Equal(t, int32(0), nilStruct.GetMaximumPageSize())\n\tassert.Nil(t, nilStruct.GetNextPageToken())\n}\n\nfunc TestHistoryTaskV2Attributes_GetDomainID(t *testing.T) {\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tDomainID: \"domain-id\",\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, \"domain-id\", res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryTaskV2Attributes_GetWorkflowID(t *testing.T) {\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tWorkflowID: \"workflow-id\",\n\t}\n\n\tres := testStruct.GetWorkflowID()\n\tassert.Equal(t, \"workflow-id\", res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetWorkflowID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryTaskV2Attributes_GetRunID(t *testing.T) {\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tRunID: \"run-id\",\n\t}\n\n\tres := testStruct.GetRunID()\n\tassert.Equal(t, \"run-id\", res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetRunID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestHistoryTaskV2Attributes_GetVersionHistoryItems(t *testing.T) {\n\tversionHistoryItems := []*VersionHistoryItem{{}, {}}\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tVersionHistoryItems: versionHistoryItems,\n\t}\n\n\tres := testStruct.GetVersionHistoryItems()\n\tassert.Equal(t, versionHistoryItems, res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetVersionHistoryItems()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryTaskV2Attributes_GetEvents(t *testing.T) {\n\tevents := &DataBlob{}\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tEvents: events,\n\t}\n\n\tres := testStruct.GetEvents()\n\tassert.Equal(t, events, res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetEvents()\n\tassert.Nil(t, res)\n}\n\nfunc TestHistoryTaskV2Attributes_GetNewRunEvents(t *testing.T) {\n\tnewRunEvents := &DataBlob{}\n\ttestStruct := HistoryTaskV2Attributes{\n\t\tNewRunEvents: newRunEvents,\n\t}\n\n\tres := testStruct.GetNewRunEvents()\n\tassert.Equal(t, newRunEvents, res)\n\n\tvar nilStruct *HistoryTaskV2Attributes\n\tres = nilStruct.GetNewRunEvents()\n\tassert.Nil(t, res)\n}\n\nfunc TestPurgeDLQMessagesRequest_Getters(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\tendMessageID := int64(12345)\n\ttestStruct := PurgeDLQMessagesRequest{\n\t\tType:                  &dlqType,\n\t\tShardID:               101,\n\t\tSourceCluster:         \"test-cluster\",\n\t\tInclusiveEndMessageID: &endMessageID,\n\t}\n\n\tassert.Equal(t, dlqType, testStruct.GetType())\n\tassert.Equal(t, int32(101), testStruct.GetShardID())\n\tassert.Equal(t, \"test-cluster\", testStruct.GetSourceCluster())\n\tassert.Equal(t, endMessageID, testStruct.GetInclusiveEndMessageID())\n\n\t// Test nil case\n\tvar nilStruct *PurgeDLQMessagesRequest\n\tassert.Equal(t, DLQType(0), nilStruct.GetType())\n\tassert.Equal(t, int32(0), nilStruct.GetShardID())\n\tassert.Equal(t, \"\", nilStruct.GetSourceCluster())\n\tassert.Equal(t, int64(0), nilStruct.GetInclusiveEndMessageID())\n}\n\nfunc TestReadDLQMessagesRequest_Getters(t *testing.T) {\n\tdlqType := DLQTypeReplication\n\tendMessageID := int64(12345)\n\tnextPageToken := []byte(\"token\")\n\ttestStruct := ReadDLQMessagesRequest{\n\t\tType:                  &dlqType,\n\t\tShardID:               101,\n\t\tSourceCluster:         \"test-cluster\",\n\t\tInclusiveEndMessageID: &endMessageID,\n\t\tMaximumPageSize:       50,\n\t\tNextPageToken:         nextPageToken,\n\t}\n\n\tassert.Equal(t, dlqType, testStruct.GetType())\n\tassert.Equal(t, int32(101), testStruct.GetShardID())\n\tassert.Equal(t, \"test-cluster\", testStruct.GetSourceCluster())\n\tassert.Equal(t, endMessageID, testStruct.GetInclusiveEndMessageID())\n\tassert.Equal(t, int32(50), testStruct.GetMaximumPageSize())\n\tassert.Equal(t, nextPageToken, testStruct.GetNextPageToken())\n\n\t// Test nil case\n\tvar nilStruct *ReadDLQMessagesRequest\n\tassert.Equal(t, DLQType(0), nilStruct.GetType())\n\tassert.Equal(t, int32(0), nilStruct.GetShardID())\n\tassert.Equal(t, \"\", nilStruct.GetSourceCluster())\n\tassert.Equal(t, int64(0), nilStruct.GetInclusiveEndMessageID())\n\tassert.Equal(t, int32(0), nilStruct.GetMaximumPageSize())\n\tassert.Nil(t, nilStruct.GetNextPageToken())\n}\n\nfunc TestReplicationMessages_GetReplicationTasks(t *testing.T) {\n\ttasks := []*ReplicationTask{{}, {}}\n\ttestStruct := ReplicationMessages{\n\t\tReplicationTasks: tasks,\n\t}\n\n\tres := testStruct.GetReplicationTasks()\n\tassert.Equal(t, tasks, res)\n\n\tvar nilStruct *ReplicationMessages\n\tres = nilStruct.GetReplicationTasks()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationMessages_GetLastRetrievedMessageID(t *testing.T) {\n\ttestStruct := ReplicationMessages{\n\t\tLastRetrievedMessageID: 12345,\n\t}\n\n\tres := testStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, int64(12345), res)\n\n\tvar nilStruct *ReplicationMessages\n\tres = nilStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationMessages_GetHasMore(t *testing.T) {\n\ttestStruct := ReplicationMessages{\n\t\tHasMore: true,\n\t}\n\n\tres := testStruct.GetHasMore()\n\tassert.True(t, res)\n\n\tvar nilStruct *ReplicationMessages\n\tres = nilStruct.GetHasMore()\n\tassert.False(t, res)\n}\n\nfunc TestReplicationMessages_GetSyncShardStatus(t *testing.T) {\n\tstatus := &SyncShardStatus{}\n\ttestStruct := ReplicationMessages{\n\t\tSyncShardStatus: status,\n\t}\n\n\tres := testStruct.GetSyncShardStatus()\n\tassert.Equal(t, status, res)\n\n\tvar nilStruct *ReplicationMessages\n\tres = nilStruct.GetSyncShardStatus()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationMessages_GetEarliestCreationTime(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tmsgs *ReplicationMessages\n\t\twant *int64\n\t}{\n\t\t\"nil\":        {msgs: nil, want: nil},\n\t\t\"zero tasks\": {msgs: &ReplicationMessages{}, want: nil},\n\t\t\"no tasks with creation time\": {\n\t\t\tmsgs: &ReplicationMessages{\n\t\t\t\tReplicationTasks: []*ReplicationTask{\n\t\t\t\t\t{}, {}, {},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: nil,\n\t\t},\n\t\t\"a few tasks with creation time\": {\n\t\t\tmsgs: &ReplicationMessages{\n\t\t\t\tReplicationTasks: []*ReplicationTask{\n\t\t\t\t\t{CreationTime: ptr.Int64(50)},\n\t\t\t\t\t{CreationTime: ptr.Int64(20)},\n\t\t\t\t\t{}, {}, {},\n\t\t\t\t\t{CreationTime: ptr.Int64(100)},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: ptr.Int64(20),\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, c.want, c.msgs.GetEarliestCreationTime())\n\t\t})\n\t}\n}\n\nfunc TestReplicationTask_GetTaskType(t *testing.T) {\n\ttaskType := ReplicationTaskTypeDomain\n\ttestStruct := ReplicationTask{\n\t\tTaskType: &taskType,\n\t}\n\n\tres := testStruct.GetTaskType()\n\tassert.Equal(t, taskType, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetTaskType()\n\tassert.Equal(t, ReplicationTaskType(0), res)\n}\n\nfunc TestReplicationTask_GetSourceTaskID(t *testing.T) {\n\ttestStruct := ReplicationTask{\n\t\tSourceTaskID: 12345,\n\t}\n\n\tres := testStruct.GetSourceTaskID()\n\tassert.Equal(t, int64(12345), res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetSourceTaskID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTask_GetSequenceID(t *testing.T) {\n\ttestStruct := ReplicationTask{\n\t\tSourceTaskID: 12345,\n\t}\n\n\tres := testStruct.GetSequenceID()\n\tassert.Equal(t, int64(12345), res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetSequenceID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTask_GetDomainTaskAttributes(t *testing.T) {\n\tdomainTask := &DomainTaskAttributes{}\n\ttestStruct := ReplicationTask{\n\t\tDomainTaskAttributes: domainTask,\n\t}\n\n\tres := testStruct.GetDomainTaskAttributes()\n\tassert.Equal(t, domainTask, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetDomainTaskAttributes()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationTask_GetSyncActivityTaskAttributes(t *testing.T) {\n\tsyncActivityTask := &SyncActivityTaskAttributes{}\n\ttestStruct := ReplicationTask{\n\t\tSyncActivityTaskAttributes: syncActivityTask,\n\t}\n\n\tres := testStruct.GetSyncActivityTaskAttributes()\n\tassert.Equal(t, syncActivityTask, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetSyncActivityTaskAttributes()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationTask_GetHistoryTaskV2Attributes(t *testing.T) {\n\thistoryTaskV2 := &HistoryTaskV2Attributes{}\n\ttestStruct := ReplicationTask{\n\t\tHistoryTaskV2Attributes: historyTaskV2,\n\t}\n\n\tres := testStruct.GetHistoryTaskV2Attributes()\n\tassert.Equal(t, historyTaskV2, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetHistoryTaskV2Attributes()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationTask_GetFailoverMarkerAttributes(t *testing.T) {\n\tfailoverMarker := &FailoverMarkerAttributes{}\n\ttestStruct := ReplicationTask{\n\t\tFailoverMarkerAttributes: failoverMarker,\n\t}\n\n\tres := testStruct.GetFailoverMarkerAttributes()\n\tassert.Equal(t, failoverMarker, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetFailoverMarkerAttributes()\n\tassert.Nil(t, res)\n}\n\nfunc TestReplicationTask_GetCreationTime(t *testing.T) {\n\tcreationTime := int64(123456)\n\ttestStruct := ReplicationTask{\n\t\tCreationTime: &creationTime,\n\t}\n\n\tres := testStruct.GetCreationTime()\n\tassert.Equal(t, creationTime, res)\n\n\tvar nilStruct *ReplicationTask\n\tres = nilStruct.GetCreationTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetDomainID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tDomainID: \"domain-id\",\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, \"domain-id\", res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestReplicationTaskInfo_GetWorkflowID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tWorkflowID: \"workflow-id\",\n\t}\n\n\tres := testStruct.GetWorkflowID()\n\tassert.Equal(t, \"workflow-id\", res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetWorkflowID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestReplicationTaskInfo_GetRunID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tRunID: \"run-id\",\n\t}\n\n\tres := testStruct.GetRunID()\n\tassert.Equal(t, \"run-id\", res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetRunID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestReplicationTaskInfo_GetTaskType(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tTaskType: 1,\n\t}\n\n\tres := testStruct.GetTaskType()\n\tassert.Equal(t, int16(1), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetTaskType()\n\tassert.Equal(t, int16(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetTaskID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tTaskID: 12345,\n\t}\n\n\tres := testStruct.GetTaskID()\n\tassert.Equal(t, int64(12345), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetTaskID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetVersion(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tVersion: 100,\n\t}\n\n\tres := testStruct.GetVersion()\n\tassert.Equal(t, int64(100), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetFirstEventID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tFirstEventID: 11111,\n\t}\n\n\tres := testStruct.GetFirstEventID()\n\tassert.Equal(t, int64(11111), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetFirstEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetNextEventID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tNextEventID: 22222,\n\t}\n\n\tres := testStruct.GetNextEventID()\n\tassert.Equal(t, int64(22222), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetNextEventID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskInfo_GetScheduledID(t *testing.T) {\n\ttestStruct := ReplicationTaskInfo{\n\t\tScheduledID: 33333,\n\t}\n\n\tres := testStruct.GetScheduledID()\n\tassert.Equal(t, int64(33333), res)\n\n\tvar nilStruct *ReplicationTaskInfo\n\tres = nilStruct.GetScheduledID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationToken_GetShardID(t *testing.T) {\n\ttestStruct := ReplicationToken{\n\t\tShardID: 1,\n\t}\n\n\tres := testStruct.GetShardID()\n\tassert.Equal(t, int32(1), res)\n\n\tvar nilStruct *ReplicationToken\n\tres = nilStruct.GetShardID()\n\tassert.Equal(t, int32(0), res)\n}\n\nfunc TestReplicationToken_GetLastRetrievedMessageID(t *testing.T) {\n\ttestStruct := ReplicationToken{\n\t\tLastRetrievedMessageID: 123456,\n\t}\n\n\tres := testStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, int64(123456), res)\n\n\tvar nilStruct *ReplicationToken\n\tres = nilStruct.GetLastRetrievedMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationToken_GetLastProcessedMessageID(t *testing.T) {\n\ttestStruct := ReplicationToken{\n\t\tLastProcessedMessageID: 654321,\n\t}\n\n\tres := testStruct.GetLastProcessedMessageID()\n\tassert.Equal(t, int64(654321), res)\n\n\tvar nilStruct *ReplicationToken\n\tres = nilStruct.GetLastProcessedMessageID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTaskType_Ptr(t *testing.T) {\n\ttaskType := ReplicationTaskTypeDomain\n\tptr := taskType.Ptr()\n\n\tassert.Equal(t, &taskType, ptr)\n}\n\nfunc TestReplicationTaskType_String(t *testing.T) {\n\tvar taskType ReplicationTaskType\n\n\ttaskType = ReplicationTaskTypeDomain\n\tassert.Equal(t, \"Domain\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeHistory\n\tassert.Equal(t, \"History\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeSyncShardStatus\n\tassert.Equal(t, \"SyncShardStatus\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeSyncActivity\n\tassert.Equal(t, \"SyncActivity\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeHistoryMetadata\n\tassert.Equal(t, \"HistoryMetadata\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeHistoryV2\n\tassert.Equal(t, \"HistoryV2\", taskType.String())\n\n\ttaskType = ReplicationTaskTypeFailoverMarker\n\tassert.Equal(t, \"FailoverMarker\", taskType.String())\n\n\t// Test unknown task type\n\ttaskType = ReplicationTaskType(10)\n\tassert.Equal(t, \"ReplicationTaskType(10)\", taskType.String())\n}\n\nfunc TestReplicationTaskType_UnmarshalText(t *testing.T) {\n\tvar taskType ReplicationTaskType\n\n\terr := taskType.UnmarshalText([]byte(\"DOMAIN\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeDomain, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"HISTORY\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeHistory, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"SYNCSHARDSTATUS\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeSyncShardStatus, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"SYNCACTIVITY\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeSyncActivity, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"HISTORYMETADATA\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeHistoryMetadata, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"HISTORYV2\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeHistoryV2, taskType)\n\n\terr = taskType.UnmarshalText([]byte(\"FAILOVERMARKER\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskTypeFailoverMarker, taskType)\n\n\t// Test unknown string value\n\terr = taskType.UnmarshalText([]byte(\"UNKNOWN\"))\n\tassert.Error(t, err)\n\n\t// Test numeric value parsing\n\terr = taskType.UnmarshalText([]byte(\"10\"))\n\tassert.NoError(t, err)\n\tassert.Equal(t, ReplicationTaskType(10), taskType)\n}\n\nfunc TestReplicationTaskType_MarshalText(t *testing.T) {\n\tvar taskType ReplicationTaskType\n\n\ttaskType = ReplicationTaskTypeDomain\n\ttext, err := taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"Domain\"), text)\n\n\ttaskType = ReplicationTaskTypeHistory\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"History\"), text)\n\n\ttaskType = ReplicationTaskTypeSyncShardStatus\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"SyncShardStatus\"), text)\n\n\ttaskType = ReplicationTaskTypeSyncActivity\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"SyncActivity\"), text)\n\n\ttaskType = ReplicationTaskTypeHistoryMetadata\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"HistoryMetadata\"), text)\n\n\ttaskType = ReplicationTaskTypeHistoryV2\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"HistoryV2\"), text)\n\n\ttaskType = ReplicationTaskTypeFailoverMarker\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"FailoverMarker\"), text)\n\n\t// Test unknown task type\n\ttaskType = ReplicationTaskType(10)\n\ttext, err = taskType.MarshalText()\n\tassert.NoError(t, err)\n\tassert.Equal(t, []byte(\"ReplicationTaskType(10)\"), text)\n}\n\nfunc TestSyncActivityTaskAttributes_GetDomainID(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tDomainID: \"domain-id\",\n\t}\n\n\tres := testStruct.GetDomainID()\n\tassert.Equal(t, \"domain-id\", res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetDomainID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetWorkflowID(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tWorkflowID: \"workflow-id\",\n\t}\n\n\tres := testStruct.GetWorkflowID()\n\tassert.Equal(t, \"workflow-id\", res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetWorkflowID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetRunID(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tRunID: \"run-id\",\n\t}\n\n\tres := testStruct.GetRunID()\n\tassert.Equal(t, \"run-id\", res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetRunID()\n\tassert.Equal(t, \"\", res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetVersion(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tVersion: 12345,\n\t}\n\n\tres := testStruct.GetVersion()\n\tassert.Equal(t, int64(12345), res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetVersion()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetScheduledID(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tScheduledID: 67890,\n\t}\n\n\tres := testStruct.GetScheduledID()\n\tassert.Equal(t, int64(67890), res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetScheduledID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetScheduledTime(t *testing.T) {\n\tscheduledTime := int64(1234567890)\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tScheduledTime: &scheduledTime,\n\t}\n\n\tres := testStruct.GetScheduledTime()\n\tassert.Equal(t, scheduledTime, res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetScheduledTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetStartedID(t *testing.T) {\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tStartedID: 11111,\n\t}\n\n\tres := testStruct.GetStartedID()\n\tassert.Equal(t, int64(11111), res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetStartedID()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetStartedTime(t *testing.T) {\n\tstartedTime := int64(1234567890)\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tStartedTime: &startedTime,\n\t}\n\n\tres := testStruct.GetStartedTime()\n\tassert.Equal(t, startedTime, res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetStartedTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetLastHeartbeatTime(t *testing.T) {\n\tlastHeartbeatTime := int64(9876543210)\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tLastHeartbeatTime: &lastHeartbeatTime,\n\t}\n\n\tres := testStruct.GetLastHeartbeatTime()\n\tassert.Equal(t, lastHeartbeatTime, res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetLastHeartbeatTime()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestSyncActivityTaskAttributes_GetVersionHistory(t *testing.T) {\n\tversionHistory := &VersionHistory{}\n\ttestStruct := SyncActivityTaskAttributes{\n\t\tVersionHistory: versionHistory,\n\t}\n\n\tres := testStruct.GetVersionHistory()\n\tassert.Equal(t, versionHistory, res)\n\n\tvar nilStruct *SyncActivityTaskAttributes\n\tres = nilStruct.GetVersionHistory()\n\tassert.Nil(t, res)\n}\n\nfunc TestSyncShardStatus_GetTimestamp(t *testing.T) {\n\ttimestamp := int64(9876543210)\n\ttestStruct := SyncShardStatus{\n\t\tTimestamp: &timestamp,\n\t}\n\n\tres := testStruct.GetTimestamp()\n\tassert.Equal(t, timestamp, res)\n\n\tvar nilStruct *SyncShardStatus\n\tres = nilStruct.GetTimestamp()\n\tassert.Equal(t, int64(0), res)\n}\n\nfunc TestReplicationTask_ByteSize(t *testing.T) {\n\tAssertReachablesImplementByteSize(t, (*ReplicationTask)(nil))\n\tAssertByteSizeMatchesReflect(t, &ReplicationTask{})\n}\n"
  },
  {
    "path": "common/types/schedule.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n)\n\n// --- Enums ---\n\n// ScheduleOverlapPolicy defines behavior when a scheduled run overlaps with a previous run.\ntype ScheduleOverlapPolicy int32\n\nconst (\n\tScheduleOverlapPolicyInvalid           ScheduleOverlapPolicy = iota\n\tScheduleOverlapPolicySkipNew                                 // Skip if previous still running\n\tScheduleOverlapPolicyBuffer                                  // Buffer and execute sequentially\n\tScheduleOverlapPolicyConcurrent                              // Allow concurrent execution\n\tScheduleOverlapPolicyCancelPrevious                          // Cancel previous, start new\n\tScheduleOverlapPolicyTerminatePrevious                       // Terminate previous, start new\n)\n\nfunc (e ScheduleOverlapPolicy) Ptr() *ScheduleOverlapPolicy { return &e }\n\nfunc (e ScheduleOverlapPolicy) String() string {\n\tswitch e {\n\tcase ScheduleOverlapPolicyInvalid:\n\t\treturn \"INVALID\"\n\tcase ScheduleOverlapPolicySkipNew:\n\t\treturn \"SKIP_NEW\"\n\tcase ScheduleOverlapPolicyBuffer:\n\t\treturn \"BUFFER\"\n\tcase ScheduleOverlapPolicyConcurrent:\n\t\treturn \"CONCURRENT\"\n\tcase ScheduleOverlapPolicyCancelPrevious:\n\t\treturn \"CANCEL_PREVIOUS\"\n\tcase ScheduleOverlapPolicyTerminatePrevious:\n\t\treturn \"TERMINATE_PREVIOUS\"\n\t}\n\treturn fmt.Sprintf(\"ScheduleOverlapPolicy(%d)\", int32(e))\n}\n\nfunc (e *ScheduleOverlapPolicy) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"INVALID\":\n\t\t*e = ScheduleOverlapPolicyInvalid\n\tcase \"SKIP_NEW\":\n\t\t*e = ScheduleOverlapPolicySkipNew\n\tcase \"BUFFER\":\n\t\t*e = ScheduleOverlapPolicyBuffer\n\tcase \"CONCURRENT\":\n\t\t*e = ScheduleOverlapPolicyConcurrent\n\tcase \"CANCEL_PREVIOUS\":\n\t\t*e = ScheduleOverlapPolicyCancelPrevious\n\tcase \"TERMINATE_PREVIOUS\":\n\t\t*e = ScheduleOverlapPolicyTerminatePrevious\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ScheduleOverlapPolicy\", err)\n\t\t}\n\t\t*e = ScheduleOverlapPolicy(val)\n\t}\n\treturn nil\n}\n\nfunc (e ScheduleOverlapPolicy) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\n// ScheduleCatchUpPolicy defines how missed runs are handled on unpause or system recovery.\ntype ScheduleCatchUpPolicy int32\n\nconst (\n\tScheduleCatchUpPolicyInvalid ScheduleCatchUpPolicy = iota\n\tScheduleCatchUpPolicySkip                          // Skip all missed runs\n\tScheduleCatchUpPolicyOne                           // Run only the most recent missed time\n\tScheduleCatchUpPolicyAll                           // Run for each missed (up to window)\n)\n\nfunc (e ScheduleCatchUpPolicy) Ptr() *ScheduleCatchUpPolicy { return &e }\n\nfunc (e ScheduleCatchUpPolicy) String() string {\n\tswitch e {\n\tcase ScheduleCatchUpPolicyInvalid:\n\t\treturn \"INVALID\"\n\tcase ScheduleCatchUpPolicySkip:\n\t\treturn \"SKIP\"\n\tcase ScheduleCatchUpPolicyOne:\n\t\treturn \"ONE\"\n\tcase ScheduleCatchUpPolicyAll:\n\t\treturn \"ALL\"\n\t}\n\treturn fmt.Sprintf(\"ScheduleCatchUpPolicy(%d)\", int32(e))\n}\n\nfunc (e *ScheduleCatchUpPolicy) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"INVALID\":\n\t\t*e = ScheduleCatchUpPolicyInvalid\n\tcase \"SKIP\":\n\t\t*e = ScheduleCatchUpPolicySkip\n\tcase \"ONE\":\n\t\t*e = ScheduleCatchUpPolicyOne\n\tcase \"ALL\":\n\t\t*e = ScheduleCatchUpPolicyAll\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ScheduleCatchUpPolicy\", err)\n\t\t}\n\t\t*e = ScheduleCatchUpPolicy(val)\n\t}\n\treturn nil\n}\n\nfunc (e ScheduleCatchUpPolicy) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\n// --- Core Types ---\n\n// ScheduleSpec defines when a schedule should trigger.\ntype ScheduleSpec struct {\n\tCronExpression string        `json:\"cronExpression,omitempty\"`\n\tStartTime      time.Time     `json:\"startTime,omitempty\"`\n\tEndTime        time.Time     `json:\"endTime,omitempty\"`\n\tJitter         time.Duration `json:\"jitter,omitempty\"`\n}\n\nfunc (v *ScheduleSpec) GetCronExpression() (o string) {\n\tif v != nil {\n\t\treturn v.CronExpression\n\t}\n\treturn\n}\n\nfunc (v *ScheduleSpec) GetStartTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.StartTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleSpec) GetEndTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.EndTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleSpec) GetJitter() (o time.Duration) {\n\tif v != nil {\n\t\treturn v.Jitter\n\t}\n\treturn\n}\n\n// StartWorkflowAction defines a workflow to start when the schedule triggers.\ntype StartWorkflowAction struct {\n\tWorkflowType                        *WorkflowType     `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList         `json:\"taskList,omitempty\"`\n\tInput                               []byte            `json:\"-\"` // Potential PII\n\tWorkflowIDPrefix                    string            `json:\"workflowIdPrefix,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32            `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32            `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tRetryPolicy                         *RetryPolicy      `json:\"retryPolicy,omitempty\"`\n\tMemo                                *Memo             `json:\"-\"` // Filtering PII\n\tSearchAttributes                    *SearchAttributes `json:\"-\"` // Filtering PII\n}\n\nfunc (v *StartWorkflowAction) GetWorkflowType() *WorkflowType {\n\tif v != nil {\n\t\treturn v.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (v *StartWorkflowAction) GetTaskList() *TaskList {\n\tif v != nil {\n\t\treturn v.TaskList\n\t}\n\treturn nil\n}\n\nfunc (v *StartWorkflowAction) GetWorkflowIDPrefix() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowIDPrefix\n\t}\n\treturn\n}\n\nfunc (v *StartWorkflowAction) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\nfunc (v *StartWorkflowAction) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// ScheduleAction defines the action to take when the schedule triggers.\n// Exactly one action field must be set.\ntype ScheduleAction struct {\n\tStartWorkflow *StartWorkflowAction `json:\"startWorkflow,omitempty\"`\n}\n\nfunc (v *ScheduleAction) GetStartWorkflow() *StartWorkflowAction {\n\tif v != nil {\n\t\treturn v.StartWorkflow\n\t}\n\treturn nil\n}\n\n// SchedulePolicies configures schedule behavior.\ntype SchedulePolicies struct {\n\tOverlapPolicy    ScheduleOverlapPolicy `json:\"overlapPolicy,omitempty\"`\n\tCatchUpPolicy    ScheduleCatchUpPolicy `json:\"catchUpPolicy,omitempty\"`\n\tCatchUpWindow    time.Duration         `json:\"catchUpWindow,omitempty\"`\n\tPauseOnFailure   bool                  `json:\"pauseOnFailure,omitempty\"`\n\tBufferLimit      int32                 `json:\"bufferLimit,omitempty\"`\n\tConcurrencyLimit int32                 `json:\"concurrencyLimit,omitempty\"`\n}\n\nfunc (v *SchedulePolicies) GetOverlapPolicy() (o ScheduleOverlapPolicy) {\n\tif v != nil {\n\t\treturn v.OverlapPolicy\n\t}\n\treturn\n}\n\nfunc (v *SchedulePolicies) GetCatchUpPolicy() (o ScheduleCatchUpPolicy) {\n\tif v != nil {\n\t\treturn v.CatchUpPolicy\n\t}\n\treturn\n}\n\nfunc (v *SchedulePolicies) GetCatchUpWindow() (o time.Duration) {\n\tif v != nil {\n\t\treturn v.CatchUpWindow\n\t}\n\treturn\n}\n\nfunc (v *SchedulePolicies) GetPauseOnFailure() (o bool) {\n\tif v != nil {\n\t\treturn v.PauseOnFailure\n\t}\n\treturn\n}\n\nfunc (v *SchedulePolicies) GetBufferLimit() (o int32) {\n\tif v != nil {\n\t\treturn v.BufferLimit\n\t}\n\treturn\n}\n\nfunc (v *SchedulePolicies) GetConcurrencyLimit() (o int32) {\n\tif v != nil {\n\t\treturn v.ConcurrencyLimit\n\t}\n\treturn\n}\n\n// SchedulePauseInfo captures the state of a paused schedule (response-only, server-populated).\ntype SchedulePauseInfo struct {\n\tReason   string    `json:\"reason,omitempty\"`\n\tPausedAt time.Time `json:\"pausedAt,omitempty\"`\n\tPausedBy string    `json:\"pausedBy,omitempty\"`\n}\n\nfunc (v *SchedulePauseInfo) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\nfunc (v *SchedulePauseInfo) GetPausedAt() (o time.Time) {\n\tif v != nil {\n\t\treturn v.PausedAt\n\t}\n\treturn\n}\n\nfunc (v *SchedulePauseInfo) GetPausedBy() (o string) {\n\tif v != nil {\n\t\treturn v.PausedBy\n\t}\n\treturn\n}\n\n// ScheduleState represents the current runtime state of a schedule.\ntype ScheduleState struct {\n\tPaused    bool               `json:\"paused,omitempty\"`\n\tPauseInfo *SchedulePauseInfo `json:\"pauseInfo,omitempty\"`\n}\n\nfunc (v *ScheduleState) GetPaused() (o bool) {\n\tif v != nil {\n\t\treturn v.Paused\n\t}\n\treturn\n}\n\nfunc (v *ScheduleState) GetPauseInfo() *SchedulePauseInfo {\n\tif v != nil {\n\t\treturn v.PauseInfo\n\t}\n\treturn nil\n}\n\n// BackfillInfo tracks the progress of an ongoing backfill operation.\ntype BackfillInfo struct {\n\tBackfillID    string    `json:\"backfillId,omitempty\"`\n\tStartTime     time.Time `json:\"startTime,omitempty\"`\n\tEndTime       time.Time `json:\"endTime,omitempty\"`\n\tRunsCompleted int32     `json:\"runsCompleted,omitempty\"`\n\tRunsTotal     int32     `json:\"runsTotal,omitempty\"`\n}\n\nfunc (v *BackfillInfo) GetStartTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.StartTime\n\t}\n\treturn\n}\n\nfunc (v *BackfillInfo) GetEndTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.EndTime\n\t}\n\treturn\n}\n\nfunc (v *BackfillInfo) GetBackfillID() (o string) {\n\tif v != nil {\n\t\treturn v.BackfillID\n\t}\n\treturn\n}\n\nfunc (v *BackfillInfo) GetRunsCompleted() (o int32) {\n\tif v != nil {\n\t\treturn v.RunsCompleted\n\t}\n\treturn\n}\n\nfunc (v *BackfillInfo) GetRunsTotal() (o int32) {\n\tif v != nil {\n\t\treturn v.RunsTotal\n\t}\n\treturn\n}\n\n// ScheduleInfo provides runtime information about the schedule.\ntype ScheduleInfo struct {\n\tLastRunTime      time.Time       `json:\"lastRunTime,omitempty\"`\n\tNextRunTime      time.Time       `json:\"nextRunTime,omitempty\"`\n\tTotalRuns        int64           `json:\"totalRuns,omitempty\"`\n\tCreateTime       time.Time       `json:\"createTime,omitempty\"`\n\tLastUpdateTime   time.Time       `json:\"lastUpdateTime,omitempty\"`\n\tOngoingBackfills []*BackfillInfo `json:\"ongoingBackfills,omitempty\"`\n}\n\nfunc (v *ScheduleInfo) GetLastRunTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.LastRunTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleInfo) GetNextRunTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.NextRunTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleInfo) GetTotalRuns() (o int64) {\n\tif v != nil {\n\t\treturn v.TotalRuns\n\t}\n\treturn\n}\n\nfunc (v *ScheduleInfo) GetCreateTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.CreateTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleInfo) GetLastUpdateTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.LastUpdateTime\n\t}\n\treturn\n}\n\nfunc (v *ScheduleInfo) GetOngoingBackfills() (o []*BackfillInfo) {\n\tif v != nil {\n\t\treturn v.OngoingBackfills\n\t}\n\treturn\n}\n\nfunc (v *StartWorkflowAction) GetInput() (o []byte) {\n\tif v != nil {\n\t\treturn v.Input\n\t}\n\treturn\n}\n\nfunc (v *StartWorkflowAction) GetRetryPolicy() *RetryPolicy {\n\tif v != nil {\n\t\treturn v.RetryPolicy\n\t}\n\treturn nil\n}\n\nfunc (v *StartWorkflowAction) GetMemo() *Memo {\n\tif v != nil {\n\t\treturn v.Memo\n\t}\n\treturn nil\n}\n\nfunc (v *StartWorkflowAction) GetSearchAttributes() *SearchAttributes {\n\tif v != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "common/types/schedule_service.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport \"time\"\n\n// --- Request/Response Types for Schedule APIs ---\n\n// ScheduleListEntry represents a single schedule in a list response.\ntype ScheduleListEntry struct {\n\tScheduleID     string         `json:\"scheduleId,omitempty\"`\n\tWorkflowType   *WorkflowType  `json:\"workflowType,omitempty\"`\n\tState          *ScheduleState `json:\"state,omitempty\"`\n\tCronExpression string         `json:\"cronExpression,omitempty\"`\n}\n\nfunc (v *ScheduleListEntry) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *ScheduleListEntry) GetWorkflowType() *WorkflowType {\n\tif v != nil {\n\t\treturn v.WorkflowType\n\t}\n\treturn nil\n}\n\nfunc (v *ScheduleListEntry) GetState() *ScheduleState {\n\tif v != nil {\n\t\treturn v.State\n\t}\n\treturn nil\n}\n\nfunc (v *ScheduleListEntry) GetCronExpression() (o string) {\n\tif v != nil {\n\t\treturn v.CronExpression\n\t}\n\treturn\n}\n\n// CreateScheduleRequest is the request to create a new schedule.\ntype CreateScheduleRequest struct {\n\tDomain           string            `json:\"domain,omitempty\"`\n\tScheduleID       string            `json:\"scheduleId,omitempty\"`\n\tSpec             *ScheduleSpec     `json:\"spec,omitempty\"`\n\tAction           *ScheduleAction   `json:\"action,omitempty\"`\n\tPolicies         *SchedulePolicies `json:\"policies,omitempty\"`\n\tMemo             *Memo             `json:\"-\"` // Filtering PII\n\tSearchAttributes *SearchAttributes `json:\"-\"` // Filtering PII\n}\n\nfunc (v *CreateScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *CreateScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *CreateScheduleRequest) GetSpec() *ScheduleSpec {\n\tif v != nil {\n\t\treturn v.Spec\n\t}\n\treturn nil\n}\n\nfunc (v *CreateScheduleRequest) GetAction() *ScheduleAction {\n\tif v != nil {\n\t\treturn v.Action\n\t}\n\treturn nil\n}\n\nfunc (v *CreateScheduleRequest) GetPolicies() *SchedulePolicies {\n\tif v != nil {\n\t\treturn v.Policies\n\t}\n\treturn nil\n}\n\nfunc (v *CreateScheduleRequest) GetMemo() *Memo {\n\tif v != nil {\n\t\treturn v.Memo\n\t}\n\treturn nil\n}\n\nfunc (v *CreateScheduleRequest) GetSearchAttributes() *SearchAttributes {\n\tif v != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn nil\n}\n\n// CreateScheduleResponse is the response for creating a schedule.\ntype CreateScheduleResponse struct{}\n\n// DescribeScheduleRequest is the request to describe a schedule.\ntype DescribeScheduleRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tScheduleID string `json:\"scheduleId,omitempty\"`\n}\n\nfunc (v *DescribeScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *DescribeScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// DescribeScheduleResponse is the response for describing a schedule.\ntype DescribeScheduleResponse struct {\n\tSpec             *ScheduleSpec     `json:\"spec,omitempty\"`\n\tAction           *ScheduleAction   `json:\"action,omitempty\"`\n\tPolicies         *SchedulePolicies `json:\"policies,omitempty\"`\n\tState            *ScheduleState    `json:\"state,omitempty\"`\n\tInfo             *ScheduleInfo     `json:\"info,omitempty\"`\n\tMemo             *Memo             `json:\"-\"` // Filtering PII\n\tSearchAttributes *SearchAttributes `json:\"-\"` // Filtering PII\n}\n\nfunc (v *DescribeScheduleResponse) GetSpec() *ScheduleSpec {\n\tif v != nil {\n\t\treturn v.Spec\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetAction() *ScheduleAction {\n\tif v != nil {\n\t\treturn v.Action\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetPolicies() *SchedulePolicies {\n\tif v != nil {\n\t\treturn v.Policies\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetState() *ScheduleState {\n\tif v != nil {\n\t\treturn v.State\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetInfo() *ScheduleInfo {\n\tif v != nil {\n\t\treturn v.Info\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetMemo() *Memo {\n\tif v != nil {\n\t\treturn v.Memo\n\t}\n\treturn nil\n}\n\nfunc (v *DescribeScheduleResponse) GetSearchAttributes() *SearchAttributes {\n\tif v != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn nil\n}\n\n// UpdateScheduleRequest is the request to update a schedule.\ntype UpdateScheduleRequest struct {\n\tDomain           string            `json:\"domain,omitempty\"`\n\tScheduleID       string            `json:\"scheduleId,omitempty\"`\n\tSpec             *ScheduleSpec     `json:\"spec,omitempty\"`\n\tAction           *ScheduleAction   `json:\"action,omitempty\"`\n\tPolicies         *SchedulePolicies `json:\"policies,omitempty\"`\n\tSearchAttributes *SearchAttributes `json:\"-\"` // Filtering PII\n}\n\nfunc (v *UpdateScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *UpdateScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *UpdateScheduleRequest) GetSpec() *ScheduleSpec {\n\tif v != nil {\n\t\treturn v.Spec\n\t}\n\treturn nil\n}\n\nfunc (v *UpdateScheduleRequest) GetAction() *ScheduleAction {\n\tif v != nil {\n\t\treturn v.Action\n\t}\n\treturn nil\n}\n\nfunc (v *UpdateScheduleRequest) GetPolicies() *SchedulePolicies {\n\tif v != nil {\n\t\treturn v.Policies\n\t}\n\treturn nil\n}\n\nfunc (v *UpdateScheduleRequest) GetSearchAttributes() *SearchAttributes {\n\tif v != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn nil\n}\n\n// UpdateScheduleResponse is the response for updating a schedule.\ntype UpdateScheduleResponse struct{}\n\n// DeleteScheduleRequest is the request to delete a schedule.\ntype DeleteScheduleRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tScheduleID string `json:\"scheduleId,omitempty\"`\n}\n\nfunc (v *DeleteScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *DeleteScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// DeleteScheduleResponse is the response for deleting a schedule.\ntype DeleteScheduleResponse struct{}\n\n// PauseScheduleRequest is the request to pause a schedule.\ntype PauseScheduleRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tScheduleID string `json:\"scheduleId,omitempty\"`\n\tReason     string `json:\"reason,omitempty\"`\n}\n\nfunc (v *PauseScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *PauseScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *PauseScheduleRequest) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\n// PauseScheduleResponse is the response for pausing a schedule.\ntype PauseScheduleResponse struct{}\n\n// UnpauseScheduleRequest is the request to resume a paused schedule.\ntype UnpauseScheduleRequest struct {\n\tDomain        string                `json:\"domain,omitempty\"`\n\tScheduleID    string                `json:\"scheduleId,omitempty\"`\n\tReason        string                `json:\"reason,omitempty\"`\n\tCatchUpPolicy ScheduleCatchUpPolicy `json:\"catchUpPolicy,omitempty\"`\n}\n\nfunc (v *UnpauseScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *UnpauseScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *UnpauseScheduleRequest) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\nfunc (v *UnpauseScheduleRequest) GetCatchUpPolicy() (o ScheduleCatchUpPolicy) {\n\tif v != nil {\n\t\treturn v.CatchUpPolicy\n\t}\n\treturn\n}\n\n// UnpauseScheduleResponse is the response for resuming a schedule.\ntype UnpauseScheduleResponse struct{}\n\n// ListSchedulesRequest is the request to list schedules in a domain.\ntype ListSchedulesRequest struct {\n\tDomain        string `json:\"domain,omitempty\"`\n\tPageSize      int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\nfunc (v *ListSchedulesRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *ListSchedulesRequest) GetPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.PageSize\n\t}\n\treturn\n}\n\nfunc (v *ListSchedulesRequest) GetNextPageToken() (o []byte) {\n\tif v != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// ListSchedulesResponse is the response for listing schedules.\ntype ListSchedulesResponse struct {\n\tSchedules     []*ScheduleListEntry `json:\"schedules,omitempty\"`\n\tNextPageToken []byte               `json:\"nextPageToken,omitempty\"`\n}\n\nfunc (v *ListSchedulesResponse) GetSchedules() (o []*ScheduleListEntry) {\n\tif v != nil {\n\t\treturn v.Schedules\n\t}\n\treturn\n}\n\nfunc (v *ListSchedulesResponse) GetNextPageToken() (o []byte) {\n\tif v != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// BackfillScheduleRequest is the request to trigger a backfill for a time range.\ntype BackfillScheduleRequest struct {\n\tDomain        string                `json:\"domain,omitempty\"`\n\tScheduleID    string                `json:\"scheduleId,omitempty\"`\n\tStartTime     time.Time             `json:\"startTime,omitempty\"`\n\tEndTime       time.Time             `json:\"endTime,omitempty\"`\n\tOverlapPolicy ScheduleOverlapPolicy `json:\"overlapPolicy,omitempty\"`\n\tBackfillID    string                `json:\"backfillId,omitempty\"`\n}\n\nfunc (v *BackfillScheduleRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\nfunc (v *BackfillScheduleRequest) GetScheduleID() (o string) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\nfunc (v *BackfillScheduleRequest) GetStartTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.StartTime\n\t}\n\treturn\n}\n\nfunc (v *BackfillScheduleRequest) GetEndTime() (o time.Time) {\n\tif v != nil {\n\t\treturn v.EndTime\n\t}\n\treturn\n}\n\nfunc (v *BackfillScheduleRequest) GetOverlapPolicy() (o ScheduleOverlapPolicy) {\n\tif v != nil {\n\t\treturn v.OverlapPolicy\n\t}\n\treturn\n}\n\nfunc (v *BackfillScheduleRequest) GetBackfillID() (o string) {\n\tif v != nil {\n\t\treturn v.BackfillID\n\t}\n\treturn\n}\n\n// BackfillScheduleResponse is the response for triggering a backfill.\ntype BackfillScheduleResponse struct{}\n"
  },
  {
    "path": "common/types/schedule_service_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestScheduleListEntry_NilGetters(t *testing.T) {\n\tvar v *ScheduleListEntry\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Nil(t, v.GetWorkflowType())\n\tassert.Nil(t, v.GetState())\n\tassert.Equal(t, \"\", v.GetCronExpression())\n}\n\nfunc TestScheduleListEntry_Getters(t *testing.T) {\n\twt := &WorkflowType{Name: \"test-wf\"}\n\tst := &ScheduleState{Paused: true}\n\tv := &ScheduleListEntry{\n\t\tScheduleID:     \"sched-1\",\n\t\tWorkflowType:   wt,\n\t\tState:          st,\n\t\tCronExpression: \"*/5 * * * *\",\n\t}\n\tassert.Equal(t, \"sched-1\", v.GetScheduleID())\n\tassert.Equal(t, wt, v.GetWorkflowType())\n\tassert.Equal(t, st, v.GetState())\n\tassert.Equal(t, \"*/5 * * * *\", v.GetCronExpression())\n}\n\nfunc TestCreateScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *CreateScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Nil(t, v.GetSpec())\n\tassert.Nil(t, v.GetAction())\n\tassert.Nil(t, v.GetPolicies())\n\tassert.Nil(t, v.GetMemo())\n\tassert.Nil(t, v.GetSearchAttributes())\n}\n\nfunc TestDescribeScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *DescribeScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n}\n\nfunc TestDescribeScheduleResponse_NilGetters(t *testing.T) {\n\tvar v *DescribeScheduleResponse\n\tassert.Nil(t, v.GetSpec())\n\tassert.Nil(t, v.GetAction())\n\tassert.Nil(t, v.GetPolicies())\n\tassert.Nil(t, v.GetState())\n\tassert.Nil(t, v.GetInfo())\n\tassert.Nil(t, v.GetMemo())\n\tassert.Nil(t, v.GetSearchAttributes())\n}\n\nfunc TestUpdateScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *UpdateScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Nil(t, v.GetSpec())\n\tassert.Nil(t, v.GetAction())\n\tassert.Nil(t, v.GetPolicies())\n\tassert.Nil(t, v.GetSearchAttributes())\n}\n\nfunc TestDeleteScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *DeleteScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n}\n\nfunc TestPauseScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *PauseScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Equal(t, \"\", v.GetReason())\n}\n\nfunc TestUnpauseScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *UnpauseScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Equal(t, \"\", v.GetReason())\n\tassert.Equal(t, ScheduleCatchUpPolicyInvalid, v.GetCatchUpPolicy())\n}\n\nfunc TestUnpauseScheduleRequest_Getters(t *testing.T) {\n\tv := &UnpauseScheduleRequest{\n\t\tDomain:        \"test-domain\",\n\t\tScheduleID:    \"sched-1\",\n\t\tReason:        \"resuming\",\n\t\tCatchUpPolicy: ScheduleCatchUpPolicyAll,\n\t}\n\tassert.Equal(t, \"test-domain\", v.GetDomain())\n\tassert.Equal(t, \"sched-1\", v.GetScheduleID())\n\tassert.Equal(t, \"resuming\", v.GetReason())\n\tassert.Equal(t, ScheduleCatchUpPolicyAll, v.GetCatchUpPolicy())\n}\n\nfunc TestListSchedulesRequest_NilGetters(t *testing.T) {\n\tvar v *ListSchedulesRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, int32(0), v.GetPageSize())\n\tassert.Nil(t, v.GetNextPageToken())\n}\n\nfunc TestListSchedulesResponse_NilGetters(t *testing.T) {\n\tvar v *ListSchedulesResponse\n\tassert.Nil(t, v.GetSchedules())\n\tassert.Nil(t, v.GetNextPageToken())\n}\n\nfunc TestBackfillScheduleRequest_NilGetters(t *testing.T) {\n\tvar v *BackfillScheduleRequest\n\tassert.Equal(t, \"\", v.GetDomain())\n\tassert.Equal(t, \"\", v.GetScheduleID())\n\tassert.Equal(t, time.Time{}, v.GetStartTime())\n\tassert.Equal(t, time.Time{}, v.GetEndTime())\n\tassert.Equal(t, ScheduleOverlapPolicyInvalid, v.GetOverlapPolicy())\n\tassert.Equal(t, \"\", v.GetBackfillID())\n}\n\nfunc TestBackfillScheduleRequest_Getters(t *testing.T) {\n\tnow := time.Now().Truncate(time.Second)\n\tv := &BackfillScheduleRequest{\n\t\tDomain:        \"test-domain\",\n\t\tScheduleID:    \"sched-1\",\n\t\tStartTime:     now,\n\t\tEndTime:       now.Add(time.Hour),\n\t\tOverlapPolicy: ScheduleOverlapPolicyBuffer,\n\t\tBackfillID:    \"bf-1\",\n\t}\n\tassert.Equal(t, \"test-domain\", v.GetDomain())\n\tassert.Equal(t, \"sched-1\", v.GetScheduleID())\n\tassert.Equal(t, now, v.GetStartTime())\n\tassert.Equal(t, now.Add(time.Hour), v.GetEndTime())\n\tassert.Equal(t, ScheduleOverlapPolicyBuffer, v.GetOverlapPolicy())\n\tassert.Equal(t, \"bf-1\", v.GetBackfillID())\n}\n"
  },
  {
    "path": "common/types/schedule_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestScheduleOverlapPolicy_UnmarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ttext string\n\t\twant ScheduleOverlapPolicy\n\t\terr  bool\n\t}{\n\t\t{name: \"invalid\", text: \"INVALID\", want: ScheduleOverlapPolicyInvalid},\n\t\t{name: \"skip_new\", text: \"SKIP_NEW\", want: ScheduleOverlapPolicySkipNew},\n\t\t{name: \"buffer\", text: \"BUFFER\", want: ScheduleOverlapPolicyBuffer},\n\t\t{name: \"concurrent\", text: \"CONCURRENT\", want: ScheduleOverlapPolicyConcurrent},\n\t\t{name: \"cancel_previous\", text: \"CANCEL_PREVIOUS\", want: ScheduleOverlapPolicyCancelPrevious},\n\t\t{name: \"terminate_previous\", text: \"TERMINATE_PREVIOUS\", want: ScheduleOverlapPolicyTerminatePrevious},\n\t\t{name: \"lowercase\", text: \"skip_new\", want: ScheduleOverlapPolicySkipNew},\n\t\t{name: \"numeric\", text: \"2\", want: ScheduleOverlapPolicy(2)},\n\t\t{name: \"unknown\", text: \"UNKNOWN\", err: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar got ScheduleOverlapPolicy\n\t\t\terr := got.UnmarshalText([]byte(tt.text))\n\t\t\tif tt.err {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScheduleOverlapPolicy_MarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tval  ScheduleOverlapPolicy\n\t\twant string\n\t}{\n\t\t{name: \"invalid\", val: ScheduleOverlapPolicyInvalid, want: \"INVALID\"},\n\t\t{name: \"skip_new\", val: ScheduleOverlapPolicySkipNew, want: \"SKIP_NEW\"},\n\t\t{name: \"buffer\", val: ScheduleOverlapPolicyBuffer, want: \"BUFFER\"},\n\t\t{name: \"concurrent\", val: ScheduleOverlapPolicyConcurrent, want: \"CONCURRENT\"},\n\t\t{name: \"cancel_previous\", val: ScheduleOverlapPolicyCancelPrevious, want: \"CANCEL_PREVIOUS\"},\n\t\t{name: \"terminate_previous\", val: ScheduleOverlapPolicyTerminatePrevious, want: \"TERMINATE_PREVIOUS\"},\n\t\t{name: \"unknown_numeric\", val: ScheduleOverlapPolicy(99), want: \"ScheduleOverlapPolicy(99)\"},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tb, err := tt.val.MarshalText()\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tt.want, string(b))\n\t\t})\n\t}\n}\n\nfunc TestScheduleOverlapPolicy_RoundTrip(t *testing.T) {\n\tfor _, val := range []ScheduleOverlapPolicy{\n\t\tScheduleOverlapPolicyInvalid,\n\t\tScheduleOverlapPolicySkipNew,\n\t\tScheduleOverlapPolicyBuffer,\n\t\tScheduleOverlapPolicyConcurrent,\n\t\tScheduleOverlapPolicyCancelPrevious,\n\t\tScheduleOverlapPolicyTerminatePrevious,\n\t} {\n\t\tb, err := val.MarshalText()\n\t\tassert.NoError(t, err)\n\t\tvar got ScheduleOverlapPolicy\n\t\terr = got.UnmarshalText(b)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, val, got)\n\t}\n}\n\nfunc TestScheduleCatchUpPolicy_UnmarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\ttext string\n\t\twant ScheduleCatchUpPolicy\n\t\terr  bool\n\t}{\n\t\t{name: \"invalid\", text: \"INVALID\", want: ScheduleCatchUpPolicyInvalid},\n\t\t{name: \"skip\", text: \"SKIP\", want: ScheduleCatchUpPolicySkip},\n\t\t{name: \"one\", text: \"ONE\", want: ScheduleCatchUpPolicyOne},\n\t\t{name: \"all\", text: \"ALL\", want: ScheduleCatchUpPolicyAll},\n\t\t{name: \"lowercase\", text: \"all\", want: ScheduleCatchUpPolicyAll},\n\t\t{name: \"numeric\", text: \"1\", want: ScheduleCatchUpPolicy(1)},\n\t\t{name: \"unknown\", text: \"UNKNOWN\", err: true},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar got ScheduleCatchUpPolicy\n\t\t\terr := got.UnmarshalText([]byte(tt.text))\n\t\t\tif tt.err {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestScheduleCatchUpPolicy_MarshalText(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tval  ScheduleCatchUpPolicy\n\t\twant string\n\t}{\n\t\t{name: \"invalid\", val: ScheduleCatchUpPolicyInvalid, want: \"INVALID\"},\n\t\t{name: \"skip\", val: ScheduleCatchUpPolicySkip, want: \"SKIP\"},\n\t\t{name: \"one\", val: ScheduleCatchUpPolicyOne, want: \"ONE\"},\n\t\t{name: \"all\", val: ScheduleCatchUpPolicyAll, want: \"ALL\"},\n\t\t{name: \"unknown_numeric\", val: ScheduleCatchUpPolicy(99), want: \"ScheduleCatchUpPolicy(99)\"},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tb, err := tt.val.MarshalText()\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tt.want, string(b))\n\t\t})\n\t}\n}\n\nfunc TestScheduleCatchUpPolicy_RoundTrip(t *testing.T) {\n\tfor _, val := range []ScheduleCatchUpPolicy{\n\t\tScheduleCatchUpPolicyInvalid,\n\t\tScheduleCatchUpPolicySkip,\n\t\tScheduleCatchUpPolicyOne,\n\t\tScheduleCatchUpPolicyAll,\n\t} {\n\t\tb, err := val.MarshalText()\n\t\tassert.NoError(t, err)\n\t\tvar got ScheduleCatchUpPolicy\n\t\terr = got.UnmarshalText(b)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, val, got)\n\t}\n}\n\nfunc TestScheduleOverlapPolicy_Ptr(t *testing.T) {\n\tval := ScheduleOverlapPolicyBuffer\n\tptr := val.Ptr()\n\tassert.Equal(t, &val, ptr)\n}\n\nfunc TestScheduleCatchUpPolicy_Ptr(t *testing.T) {\n\tval := ScheduleCatchUpPolicyAll\n\tptr := val.Ptr()\n\tassert.Equal(t, &val, ptr)\n}\n"
  },
  {
    "path": "common/types/sharddistributor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n)\n\n//go:generate enumer -type=ExecutorStatus,ShardStatus,AssignmentStatus,MigrationMode,HandoverType,LoadBalancingMode -json -output sharddistributor_statuses_enumer_generated.go\n\ntype GetShardOwnerRequest struct {\n\tShardKey  string\n\tNamespace string\n}\n\nfunc (v *GetShardOwnerRequest) GetShardKey() (o string) {\n\tif v != nil {\n\t\treturn v.ShardKey\n\t}\n\treturn\n}\n\nfunc (v *GetShardOwnerRequest) GetNamespace() (o string) {\n\tif v != nil {\n\t\treturn v.Namespace\n\t}\n\treturn\n}\n\ntype GetShardOwnerResponse struct {\n\tOwner     string\n\tNamespace string\n\tMetadata  map[string]string\n}\n\nfunc (v *GetShardOwnerResponse) GetOwner() (o string) {\n\tif v != nil {\n\t\treturn v.Owner\n\t}\n\treturn\n}\n\nfunc (v *GetShardOwnerResponse) GetNamespace() (o string) {\n\tif v != nil {\n\t\treturn v.Namespace\n\t}\n\treturn\n}\n\nfunc (v *GetShardOwnerResponse) GetMetadata() (o map[string]string) {\n\tif v != nil {\n\t\treturn v.Metadata\n\t}\n\treturn\n}\n\ntype NamespaceNotFoundError struct {\n\tNamespace string\n}\n\nfunc (n *NamespaceNotFoundError) Error() (o string) {\n\tif n != nil {\n\t\treturn fmt.Sprintf(\"namespace not found %v\", n.Namespace)\n\t}\n\treturn\n}\n\ntype ShardNotFoundError struct {\n\tNamespace string\n\tShardKey  string\n}\n\nfunc (n *ShardNotFoundError) Error() (o string) {\n\tif n != nil {\n\t\treturn fmt.Sprintf(\"shard not found %v:%v\", n.Namespace, n.ShardKey)\n\t}\n\treturn\n}\n\ntype ExecutorHeartbeatRequest struct {\n\tNamespace          string\n\tExecutorID         string\n\tStatus             ExecutorStatus\n\tShardStatusReports map[string]*ShardStatusReport\n\tMetadata           map[string]string\n}\n\nfunc (v *ExecutorHeartbeatRequest) GetNamespace() (o string) {\n\tif v != nil {\n\t\treturn v.Namespace\n\t}\n\treturn\n}\n\nfunc (v *ExecutorHeartbeatRequest) GetExecutorID() (o string) {\n\tif v != nil {\n\t\treturn v.ExecutorID\n\t}\n\treturn\n}\n\nfunc (v *ExecutorHeartbeatRequest) GetStatus() (o ExecutorStatus) {\n\tif v != nil {\n\t\treturn v.Status\n\t}\n\treturn\n}\n\nfunc (v *ExecutorHeartbeatRequest) GetShardStatusReports() (o map[string]*ShardStatusReport) {\n\tif v != nil {\n\t\treturn v.ShardStatusReports\n\t}\n\treturn\n}\n\nfunc (v *ExecutorHeartbeatRequest) GetMetadata() (o map[string]string) {\n\tif v != nil {\n\t\treturn v.Metadata\n\t}\n\treturn\n}\n\n// ExecutorStatus is persisted to the DB with a string value mapping.\n// Beware - if we want to change the name - it should be backward compatible and should be done in two steps.\ntype ExecutorStatus int32\n\nconst (\n\tExecutorStatusINVALID  ExecutorStatus = 0\n\tExecutorStatusACTIVE   ExecutorStatus = 1\n\tExecutorStatusDRAINING ExecutorStatus = 2\n\tExecutorStatusDRAINED  ExecutorStatus = 3\n)\n\ntype ShardStatusReport struct {\n\tStatus    ShardStatus\n\tShardLoad float64\n}\n\nfunc (v *ShardStatusReport) GetStatus() (o ShardStatus) {\n\tif v != nil {\n\t\treturn v.Status\n\t}\n\treturn\n}\n\nfunc (v *ShardStatusReport) GetShardLoad() (o float64) {\n\tif v != nil {\n\t\treturn v.ShardLoad\n\t}\n\treturn\n}\n\n// ShardStatus is persisted to the DB with a string value mapping.\n// Beware - if we want to change the name - it should be backward compatible and should be done in two steps.\ntype ShardStatus int32\n\nconst (\n\tShardStatusINVALID ShardStatus = 0\n\tShardStatusREADY   ShardStatus = 1\n\tShardStatusDONE    ShardStatus = 2\n)\n\ntype ExecutorHeartbeatResponse struct {\n\tShardAssignments map[string]*ShardAssignment\n\tMigrationMode    MigrationMode\n}\n\nfunc (v *ExecutorHeartbeatResponse) GetShardAssignments() (o map[string]*ShardAssignment) {\n\tif v != nil {\n\t\treturn v.ShardAssignments\n\t}\n\treturn\n}\n\nfunc (v *ExecutorHeartbeatResponse) GetMigrationMode() (o MigrationMode) {\n\tif v != nil {\n\t\treturn v.MigrationMode\n\t}\n\treturn\n}\n\ntype ShardAssignment struct {\n\t// Status indicates the current assignment status of the shard.\n\tStatus AssignmentStatus `json:\"status\"`\n}\n\nfunc (v *ShardAssignment) GetStatus() (o AssignmentStatus) {\n\tif v != nil {\n\t\treturn v.Status\n\t}\n\treturn\n}\n\n// AssignmentStatus is persisted to the DB with a string value mapping.\n// Beware - if we want to change the name - it should be backward compatible and should be done in two steps.\ntype AssignmentStatus int32\n\nconst (\n\tAssignmentStatusINVALID AssignmentStatus = 0\n\tAssignmentStatusREADY   AssignmentStatus = 1\n)\n\n// HandoverType is used to indicate the type of handover that occurred during shard reassignment.\n// Type is persisted to the DB with a string value mapping.\n// Beware - if we want to change the name - it should be backward compatible and should be done in two steps.\ntype HandoverType int32\n\nconst (\n\tHandoverTypeINVALID HandoverType = 0\n\n\t// HandoverTypeGRACEFUL\n\t// Graceful handover indicates that the shard was transferred in a way that allowed\n\t// the previous owner had a chance to finish processing before the shard was reassigned.\n\tHandoverTypeGRACEFUL HandoverType = 1\n\n\t// HandoverTypeEMERGENCY\n\t// Emergency handover indicates that the shard was transferred abruptly without\n\t// allowing the previous owner to finish processing.\n\tHandoverTypeEMERGENCY HandoverType = 2\n)\n\n// Ptr returns a pointer to the HandoverType value.\nfunc (t HandoverType) Ptr() *HandoverType {\n\treturn &t\n}\n\ntype MigrationMode int32\n\nconst (\n\tMigrationModeINVALID                MigrationMode = 0\n\tMigrationModeLOCALPASSTHROUGH       MigrationMode = 1\n\tMigrationModeLOCALPASSTHROUGHSHADOW MigrationMode = 2\n\tMigrationModeDISTRIBUTEDPASSTHROUGH MigrationMode = 3\n\tMigrationModeONBOARDED              MigrationMode = 4\n)\n\ntype LoadBalancingMode int32\n\nconst (\n\tLoadBalancingModeINVALID LoadBalancingMode = 0\n\tLoadBalancingModeNAIVE   LoadBalancingMode = 1\n\tLoadBalancingModeGREEDY  LoadBalancingMode = 2\n)\n\ntype WatchNamespaceStateRequest struct {\n\tNamespace string\n}\n\nfunc (v *WatchNamespaceStateRequest) GetNamespace() (o string) {\n\tif v != nil {\n\t\treturn v.Namespace\n\t}\n\treturn\n}\n\ntype WatchNamespaceStateResponse struct {\n\tExecutors []*ExecutorShardAssignment\n}\n\nfunc (v *WatchNamespaceStateResponse) GetExecutors() (o []*ExecutorShardAssignment) {\n\tif v != nil {\n\t\treturn v.Executors\n\t}\n\treturn\n}\n\ntype ExecutorShardAssignment struct {\n\tExecutorID     string\n\tAssignedShards []*Shard\n\tMetadata       map[string]string\n}\n\nfunc (v *ExecutorShardAssignment) GetExecutorID() (o string) {\n\tif v != nil {\n\t\treturn v.ExecutorID\n\t}\n\treturn\n}\n\nfunc (v *ExecutorShardAssignment) GetAssignedShards() (o []*Shard) {\n\tif v != nil {\n\t\treturn v.AssignedShards\n\t}\n\treturn\n}\n\nfunc (v *ExecutorShardAssignment) GetMetadata() (o map[string]string) {\n\tif v != nil {\n\t\treturn v.Metadata\n\t}\n\treturn\n}\n\ntype Shard struct {\n\tShardKey string\n}\n\nfunc (v *Shard) GetShardKey() (o string) {\n\tif v != nil {\n\t\treturn v.ShardKey\n\t}\n\treturn\n}\n"
  },
  {
    "path": "common/types/sharddistributor_statuses_enumer_generated.go",
    "content": "// Code generated by \"enumer -type=ExecutorStatus,ShardStatus,AssignmentStatus,MigrationMode,HandoverType,LoadBalancingMode -json -output sharddistributor_statuses_enumer_generated.go\"; DO NOT EDIT.\n\npackage types\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n)\n\nconst _ExecutorStatusName = \"ExecutorStatusINVALIDExecutorStatusACTIVEExecutorStatusDRAININGExecutorStatusDRAINED\"\n\nvar _ExecutorStatusIndex = [...]uint8{0, 21, 41, 63, 84}\n\nconst _ExecutorStatusLowerName = \"executorstatusinvalidexecutorstatusactiveexecutorstatusdrainingexecutorstatusdrained\"\n\nfunc (i ExecutorStatus) String() string {\n\tif i < 0 || i >= ExecutorStatus(len(_ExecutorStatusIndex)-1) {\n\t\treturn fmt.Sprintf(\"ExecutorStatus(%d)\", i)\n\t}\n\treturn _ExecutorStatusName[_ExecutorStatusIndex[i]:_ExecutorStatusIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _ExecutorStatusNoOp() {\n\tvar x [1]struct{}\n\t_ = x[ExecutorStatusINVALID-(0)]\n\t_ = x[ExecutorStatusACTIVE-(1)]\n\t_ = x[ExecutorStatusDRAINING-(2)]\n\t_ = x[ExecutorStatusDRAINED-(3)]\n}\n\nvar _ExecutorStatusValues = []ExecutorStatus{ExecutorStatusINVALID, ExecutorStatusACTIVE, ExecutorStatusDRAINING, ExecutorStatusDRAINED}\n\nvar _ExecutorStatusNameToValueMap = map[string]ExecutorStatus{\n\t_ExecutorStatusName[0:21]:       ExecutorStatusINVALID,\n\t_ExecutorStatusLowerName[0:21]:  ExecutorStatusINVALID,\n\t_ExecutorStatusName[21:41]:      ExecutorStatusACTIVE,\n\t_ExecutorStatusLowerName[21:41]: ExecutorStatusACTIVE,\n\t_ExecutorStatusName[41:63]:      ExecutorStatusDRAINING,\n\t_ExecutorStatusLowerName[41:63]: ExecutorStatusDRAINING,\n\t_ExecutorStatusName[63:84]:      ExecutorStatusDRAINED,\n\t_ExecutorStatusLowerName[63:84]: ExecutorStatusDRAINED,\n}\n\nvar _ExecutorStatusNames = []string{\n\t_ExecutorStatusName[0:21],\n\t_ExecutorStatusName[21:41],\n\t_ExecutorStatusName[41:63],\n\t_ExecutorStatusName[63:84],\n}\n\n// ExecutorStatusString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc ExecutorStatusString(s string) (ExecutorStatus, error) {\n\tif val, ok := _ExecutorStatusNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _ExecutorStatusNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to ExecutorStatus values\", s)\n}\n\n// ExecutorStatusValues returns all values of the enum\nfunc ExecutorStatusValues() []ExecutorStatus {\n\treturn _ExecutorStatusValues\n}\n\n// ExecutorStatusStrings returns a slice of all String values of the enum\nfunc ExecutorStatusStrings() []string {\n\tstrs := make([]string, len(_ExecutorStatusNames))\n\tcopy(strs, _ExecutorStatusNames)\n\treturn strs\n}\n\n// IsAExecutorStatus returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i ExecutorStatus) IsAExecutorStatus() bool {\n\tfor _, v := range _ExecutorStatusValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for ExecutorStatus\nfunc (i ExecutorStatus) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for ExecutorStatus\nfunc (i *ExecutorStatus) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"ExecutorStatus should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = ExecutorStatusString(s)\n\treturn err\n}\n\nconst _ShardStatusName = \"ShardStatusINVALIDShardStatusREADYShardStatusDONE\"\n\nvar _ShardStatusIndex = [...]uint8{0, 18, 34, 49}\n\nconst _ShardStatusLowerName = \"shardstatusinvalidshardstatusreadyshardstatusdone\"\n\nfunc (i ShardStatus) String() string {\n\tif i < 0 || i >= ShardStatus(len(_ShardStatusIndex)-1) {\n\t\treturn fmt.Sprintf(\"ShardStatus(%d)\", i)\n\t}\n\treturn _ShardStatusName[_ShardStatusIndex[i]:_ShardStatusIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _ShardStatusNoOp() {\n\tvar x [1]struct{}\n\t_ = x[ShardStatusINVALID-(0)]\n\t_ = x[ShardStatusREADY-(1)]\n\t_ = x[ShardStatusDONE-(2)]\n}\n\nvar _ShardStatusValues = []ShardStatus{ShardStatusINVALID, ShardStatusREADY, ShardStatusDONE}\n\nvar _ShardStatusNameToValueMap = map[string]ShardStatus{\n\t_ShardStatusName[0:18]:       ShardStatusINVALID,\n\t_ShardStatusLowerName[0:18]:  ShardStatusINVALID,\n\t_ShardStatusName[18:34]:      ShardStatusREADY,\n\t_ShardStatusLowerName[18:34]: ShardStatusREADY,\n\t_ShardStatusName[34:49]:      ShardStatusDONE,\n\t_ShardStatusLowerName[34:49]: ShardStatusDONE,\n}\n\nvar _ShardStatusNames = []string{\n\t_ShardStatusName[0:18],\n\t_ShardStatusName[18:34],\n\t_ShardStatusName[34:49],\n}\n\n// ShardStatusString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc ShardStatusString(s string) (ShardStatus, error) {\n\tif val, ok := _ShardStatusNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _ShardStatusNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to ShardStatus values\", s)\n}\n\n// ShardStatusValues returns all values of the enum\nfunc ShardStatusValues() []ShardStatus {\n\treturn _ShardStatusValues\n}\n\n// ShardStatusStrings returns a slice of all String values of the enum\nfunc ShardStatusStrings() []string {\n\tstrs := make([]string, len(_ShardStatusNames))\n\tcopy(strs, _ShardStatusNames)\n\treturn strs\n}\n\n// IsAShardStatus returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i ShardStatus) IsAShardStatus() bool {\n\tfor _, v := range _ShardStatusValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for ShardStatus\nfunc (i ShardStatus) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for ShardStatus\nfunc (i *ShardStatus) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"ShardStatus should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = ShardStatusString(s)\n\treturn err\n}\n\nconst _AssignmentStatusName = \"AssignmentStatusINVALIDAssignmentStatusREADY\"\n\nvar _AssignmentStatusIndex = [...]uint8{0, 23, 44}\n\nconst _AssignmentStatusLowerName = \"assignmentstatusinvalidassignmentstatusready\"\n\nfunc (i AssignmentStatus) String() string {\n\tif i < 0 || i >= AssignmentStatus(len(_AssignmentStatusIndex)-1) {\n\t\treturn fmt.Sprintf(\"AssignmentStatus(%d)\", i)\n\t}\n\treturn _AssignmentStatusName[_AssignmentStatusIndex[i]:_AssignmentStatusIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _AssignmentStatusNoOp() {\n\tvar x [1]struct{}\n\t_ = x[AssignmentStatusINVALID-(0)]\n\t_ = x[AssignmentStatusREADY-(1)]\n}\n\nvar _AssignmentStatusValues = []AssignmentStatus{AssignmentStatusINVALID, AssignmentStatusREADY}\n\nvar _AssignmentStatusNameToValueMap = map[string]AssignmentStatus{\n\t_AssignmentStatusName[0:23]:       AssignmentStatusINVALID,\n\t_AssignmentStatusLowerName[0:23]:  AssignmentStatusINVALID,\n\t_AssignmentStatusName[23:44]:      AssignmentStatusREADY,\n\t_AssignmentStatusLowerName[23:44]: AssignmentStatusREADY,\n}\n\nvar _AssignmentStatusNames = []string{\n\t_AssignmentStatusName[0:23],\n\t_AssignmentStatusName[23:44],\n}\n\n// AssignmentStatusString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc AssignmentStatusString(s string) (AssignmentStatus, error) {\n\tif val, ok := _AssignmentStatusNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _AssignmentStatusNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to AssignmentStatus values\", s)\n}\n\n// AssignmentStatusValues returns all values of the enum\nfunc AssignmentStatusValues() []AssignmentStatus {\n\treturn _AssignmentStatusValues\n}\n\n// AssignmentStatusStrings returns a slice of all String values of the enum\nfunc AssignmentStatusStrings() []string {\n\tstrs := make([]string, len(_AssignmentStatusNames))\n\tcopy(strs, _AssignmentStatusNames)\n\treturn strs\n}\n\n// IsAAssignmentStatus returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i AssignmentStatus) IsAAssignmentStatus() bool {\n\tfor _, v := range _AssignmentStatusValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for AssignmentStatus\nfunc (i AssignmentStatus) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for AssignmentStatus\nfunc (i *AssignmentStatus) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"AssignmentStatus should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = AssignmentStatusString(s)\n\treturn err\n}\n\nconst _MigrationModeName = \"MigrationModeINVALIDMigrationModeLOCALPASSTHROUGHMigrationModeLOCALPASSTHROUGHSHADOWMigrationModeDISTRIBUTEDPASSTHROUGHMigrationModeONBOARDED\"\n\nvar _MigrationModeIndex = [...]uint8{0, 20, 49, 84, 119, 141}\n\nconst _MigrationModeLowerName = \"migrationmodeinvalidmigrationmodelocalpassthroughmigrationmodelocalpassthroughshadowmigrationmodedistributedpassthroughmigrationmodeonboarded\"\n\nfunc (i MigrationMode) String() string {\n\tif i < 0 || i >= MigrationMode(len(_MigrationModeIndex)-1) {\n\t\treturn fmt.Sprintf(\"MigrationMode(%d)\", i)\n\t}\n\treturn _MigrationModeName[_MigrationModeIndex[i]:_MigrationModeIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _MigrationModeNoOp() {\n\tvar x [1]struct{}\n\t_ = x[MigrationModeINVALID-(0)]\n\t_ = x[MigrationModeLOCALPASSTHROUGH-(1)]\n\t_ = x[MigrationModeLOCALPASSTHROUGHSHADOW-(2)]\n\t_ = x[MigrationModeDISTRIBUTEDPASSTHROUGH-(3)]\n\t_ = x[MigrationModeONBOARDED-(4)]\n}\n\nvar _MigrationModeValues = []MigrationMode{MigrationModeINVALID, MigrationModeLOCALPASSTHROUGH, MigrationModeLOCALPASSTHROUGHSHADOW, MigrationModeDISTRIBUTEDPASSTHROUGH, MigrationModeONBOARDED}\n\nvar _MigrationModeNameToValueMap = map[string]MigrationMode{\n\t_MigrationModeName[0:20]:         MigrationModeINVALID,\n\t_MigrationModeLowerName[0:20]:    MigrationModeINVALID,\n\t_MigrationModeName[20:49]:        MigrationModeLOCALPASSTHROUGH,\n\t_MigrationModeLowerName[20:49]:   MigrationModeLOCALPASSTHROUGH,\n\t_MigrationModeName[49:84]:        MigrationModeLOCALPASSTHROUGHSHADOW,\n\t_MigrationModeLowerName[49:84]:   MigrationModeLOCALPASSTHROUGHSHADOW,\n\t_MigrationModeName[84:119]:       MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t_MigrationModeLowerName[84:119]:  MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t_MigrationModeName[119:141]:      MigrationModeONBOARDED,\n\t_MigrationModeLowerName[119:141]: MigrationModeONBOARDED,\n}\n\nvar _MigrationModeNames = []string{\n\t_MigrationModeName[0:20],\n\t_MigrationModeName[20:49],\n\t_MigrationModeName[49:84],\n\t_MigrationModeName[84:119],\n\t_MigrationModeName[119:141],\n}\n\n// MigrationModeString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc MigrationModeString(s string) (MigrationMode, error) {\n\tif val, ok := _MigrationModeNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _MigrationModeNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to MigrationMode values\", s)\n}\n\n// MigrationModeValues returns all values of the enum\nfunc MigrationModeValues() []MigrationMode {\n\treturn _MigrationModeValues\n}\n\n// MigrationModeStrings returns a slice of all String values of the enum\nfunc MigrationModeStrings() []string {\n\tstrs := make([]string, len(_MigrationModeNames))\n\tcopy(strs, _MigrationModeNames)\n\treturn strs\n}\n\n// IsAMigrationMode returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i MigrationMode) IsAMigrationMode() bool {\n\tfor _, v := range _MigrationModeValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for MigrationMode\nfunc (i MigrationMode) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for MigrationMode\nfunc (i *MigrationMode) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"MigrationMode should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = MigrationModeString(s)\n\treturn err\n}\n\nconst _HandoverTypeName = \"HandoverTypeINVALIDHandoverTypeGRACEFULHandoverTypeEMERGENCY\"\n\nvar _HandoverTypeIndex = [...]uint8{0, 19, 39, 60}\n\nconst _HandoverTypeLowerName = \"handovertypeinvalidhandovertypegracefulhandovertypeemergency\"\n\nfunc (i HandoverType) String() string {\n\tif i < 0 || i >= HandoverType(len(_HandoverTypeIndex)-1) {\n\t\treturn fmt.Sprintf(\"HandoverType(%d)\", i)\n\t}\n\treturn _HandoverTypeName[_HandoverTypeIndex[i]:_HandoverTypeIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _HandoverTypeNoOp() {\n\tvar x [1]struct{}\n\t_ = x[HandoverTypeINVALID-(0)]\n\t_ = x[HandoverTypeGRACEFUL-(1)]\n\t_ = x[HandoverTypeEMERGENCY-(2)]\n}\n\nvar _HandoverTypeValues = []HandoverType{HandoverTypeINVALID, HandoverTypeGRACEFUL, HandoverTypeEMERGENCY}\n\nvar _HandoverTypeNameToValueMap = map[string]HandoverType{\n\t_HandoverTypeName[0:19]:       HandoverTypeINVALID,\n\t_HandoverTypeLowerName[0:19]:  HandoverTypeINVALID,\n\t_HandoverTypeName[19:39]:      HandoverTypeGRACEFUL,\n\t_HandoverTypeLowerName[19:39]: HandoverTypeGRACEFUL,\n\t_HandoverTypeName[39:60]:      HandoverTypeEMERGENCY,\n\t_HandoverTypeLowerName[39:60]: HandoverTypeEMERGENCY,\n}\n\nvar _HandoverTypeNames = []string{\n\t_HandoverTypeName[0:19],\n\t_HandoverTypeName[19:39],\n\t_HandoverTypeName[39:60],\n}\n\n// HandoverTypeString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc HandoverTypeString(s string) (HandoverType, error) {\n\tif val, ok := _HandoverTypeNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _HandoverTypeNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to HandoverType values\", s)\n}\n\n// HandoverTypeValues returns all values of the enum\nfunc HandoverTypeValues() []HandoverType {\n\treturn _HandoverTypeValues\n}\n\n// HandoverTypeStrings returns a slice of all String values of the enum\nfunc HandoverTypeStrings() []string {\n\tstrs := make([]string, len(_HandoverTypeNames))\n\tcopy(strs, _HandoverTypeNames)\n\treturn strs\n}\n\n// IsAHandoverType returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i HandoverType) IsAHandoverType() bool {\n\tfor _, v := range _HandoverTypeValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for HandoverType\nfunc (i HandoverType) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for HandoverType\nfunc (i *HandoverType) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"HandoverType should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = HandoverTypeString(s)\n\treturn err\n}\n\nconst _LoadBalancingModeName = \"LoadBalancingModeINVALIDLoadBalancingModeNAIVELoadBalancingModeGREEDY\"\n\nvar _LoadBalancingModeIndex = [...]uint8{0, 24, 46, 69}\n\nconst _LoadBalancingModeLowerName = \"loadbalancingmodeinvalidloadbalancingmodenaiveloadbalancingmodegreedy\"\n\nfunc (i LoadBalancingMode) String() string {\n\tif i < 0 || i >= LoadBalancingMode(len(_LoadBalancingModeIndex)-1) {\n\t\treturn fmt.Sprintf(\"LoadBalancingMode(%d)\", i)\n\t}\n\treturn _LoadBalancingModeName[_LoadBalancingModeIndex[i]:_LoadBalancingModeIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _LoadBalancingModeNoOp() {\n\tvar x [1]struct{}\n\t_ = x[LoadBalancingModeINVALID-(0)]\n\t_ = x[LoadBalancingModeNAIVE-(1)]\n\t_ = x[LoadBalancingModeGREEDY-(2)]\n}\n\nvar _LoadBalancingModeValues = []LoadBalancingMode{LoadBalancingModeINVALID, LoadBalancingModeNAIVE, LoadBalancingModeGREEDY}\n\nvar _LoadBalancingModeNameToValueMap = map[string]LoadBalancingMode{\n\t_LoadBalancingModeName[0:24]:       LoadBalancingModeINVALID,\n\t_LoadBalancingModeLowerName[0:24]:  LoadBalancingModeINVALID,\n\t_LoadBalancingModeName[24:46]:      LoadBalancingModeNAIVE,\n\t_LoadBalancingModeLowerName[24:46]: LoadBalancingModeNAIVE,\n\t_LoadBalancingModeName[46:69]:      LoadBalancingModeGREEDY,\n\t_LoadBalancingModeLowerName[46:69]: LoadBalancingModeGREEDY,\n}\n\nvar _LoadBalancingModeNames = []string{\n\t_LoadBalancingModeName[0:24],\n\t_LoadBalancingModeName[24:46],\n\t_LoadBalancingModeName[46:69],\n}\n\n// LoadBalancingModeString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc LoadBalancingModeString(s string) (LoadBalancingMode, error) {\n\tif val, ok := _LoadBalancingModeNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _LoadBalancingModeNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to LoadBalancingMode values\", s)\n}\n\n// LoadBalancingModeValues returns all values of the enum\nfunc LoadBalancingModeValues() []LoadBalancingMode {\n\treturn _LoadBalancingModeValues\n}\n\n// LoadBalancingModeStrings returns a slice of all String values of the enum\nfunc LoadBalancingModeStrings() []string {\n\tstrs := make([]string, len(_LoadBalancingModeNames))\n\tcopy(strs, _LoadBalancingModeNames)\n\treturn strs\n}\n\n// IsALoadBalancingMode returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i LoadBalancingMode) IsALoadBalancingMode() bool {\n\tfor _, v := range _LoadBalancingModeValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\n// MarshalJSON implements the json.Marshaler interface for LoadBalancingMode\nfunc (i LoadBalancingMode) MarshalJSON() ([]byte, error) {\n\treturn json.Marshal(i.String())\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface for LoadBalancingMode\nfunc (i *LoadBalancingMode) UnmarshalJSON(data []byte) error {\n\tvar s string\n\tif err := json.Unmarshal(data, &s); err != nil {\n\t\treturn fmt.Errorf(\"LoadBalancingMode should be a string, got %s\", data)\n\t}\n\n\tvar err error\n\t*i, err = LoadBalancingModeString(s)\n\treturn err\n}\n"
  },
  {
    "path": "common/types/sharddistributor_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestGetShardOwnerRequest_GetShardKey(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *GetShardOwnerRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty request\",\n\t\t\treq:  &GetShardOwnerRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"has shard key\",\n\t\t\treq:  &GetShardOwnerRequest{ShardKey: \"shard-key\"},\n\t\t\twant: \"shard-key\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetShardKey()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGetShardOwnerRequest_GetNamespace(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\treq  *GetShardOwnerRequest\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil request\",\n\t\t\treq:  nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty request\",\n\t\t\treq:  &GetShardOwnerRequest{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"has namespace\",\n\t\t\treq:  &GetShardOwnerRequest{Namespace: \"namespace\"},\n\t\t\twant: \"namespace\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.req.GetNamespace()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGetShardOwnerResponse_GetOwner(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *GetShardOwnerResponse\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty response\",\n\t\t\tresp: &GetShardOwnerResponse{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"has owner\",\n\t\t\tresp: &GetShardOwnerResponse{Owner: \"owner\"},\n\t\t\twant: \"owner\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetOwner()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGetShardOwnerResponse_GetNamespace(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *GetShardOwnerResponse\n\t\twant string\n\t}{\n\t\t{\n\t\t\tname: \"nil response\",\n\t\t\tresp: nil,\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty response\",\n\t\t\tresp: &GetShardOwnerResponse{},\n\t\t\twant: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"has namespace\",\n\t\t\tresp: &GetShardOwnerResponse{Namespace: \"namespace\"},\n\t\t\twant: \"namespace\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.resp.GetNamespace()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/shared.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\t\"unsafe\"\n)\n\n// AccessDeniedError is an internal type (TBD...)\n// TODO(c-warren): Move to common/types/errors.go\ntype AccessDeniedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ActivityLocalDispatchInfo is an internal type (TBD...)\ntype ActivityLocalDispatchInfo struct {\n\tActivityID                      string `json:\"activityId,omitempty\"`\n\tScheduledTimestamp              *int64 `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp                *int64 `json:\"startedTimestamp,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64 `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tTaskToken                       []byte `json:\"taskToken,omitempty\"`\n}\n\n// GetScheduledTimestamp is an internal getter (TBD...)\nfunc (v *ActivityLocalDispatchInfo) GetScheduledTimestamp() (o int64) {\n\tif v != nil && v.ScheduledTimestamp != nil {\n\t\treturn *v.ScheduledTimestamp\n\t}\n\treturn\n}\n\n// ActivityTaskCancelRequestedEventAttributes is an internal type (TBD...)\ntype ActivityTaskCancelRequestedEventAttributes struct {\n\tActivityID                   string `json:\"activityId,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *ActivityTaskCancelRequestedEventAttributes) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskCancelRequestedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskCanceledEventAttributes is an internal type (TBD...)\ntype ActivityTaskCanceledEventAttributes struct {\n\tDetails                      []byte `json:\"details,omitempty\"`\n\tLatestCancelRequestedEventID int64  `json:\"latestCancelRequestedEventId,omitempty\"`\n\tScheduledEventID             int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID               int64  `json:\"startedEventId,omitempty\"`\n\tIdentity                     string `json:\"identity,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskCanceledEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskCanceledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskCompletedEventAttributes is an internal type (TBD...)\ntype ActivityTaskCompletedEventAttributes struct {\n\tResult           []byte `json:\"result,omitempty\"`\n\tScheduledEventID int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID   int64  `json:\"startedEventId,omitempty\"`\n\tIdentity         string `json:\"identity,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskCompletedEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetStartedEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskCompletedEventAttributes) GetStartedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskCompletedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskFailedEventAttributes is an internal type (TBD...)\ntype ActivityTaskFailedEventAttributes struct {\n\tReason           *string `json:\"reason,omitempty\"`\n\tDetails          []byte  `json:\"details,omitempty\"`\n\tScheduledEventID int64   `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID   int64   `json:\"startedEventId,omitempty\"`\n\tIdentity         string  `json:\"identity,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskFailedEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetStartedEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskFailedEventAttributes) GetStartedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskScheduledEventAttributes is an internal type (TBD...)\ntype ActivityTaskScheduledEventAttributes struct {\n\tActivityID                    string        `json:\"activityId,omitempty\"`\n\tActivityType                  *ActivityType `json:\"activityType,omitempty\"`\n\tDomain                        *string       `json:\"domain,omitempty\"`\n\tTaskList                      *TaskList     `json:\"taskList,omitempty\"`\n\tInput                         []byte        `json:\"input,omitempty\"`\n\tScheduleToCloseTimeoutSeconds *int32        `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32        `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tStartToCloseTimeoutSeconds    *int32        `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds       *int32        `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventID  int64         `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tRetryPolicy                   *RetryPolicy  `json:\"retryPolicy,omitempty\"`\n\tHeader                        *Header       `json:\"header,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetActivityType is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetDomain() (o string) {\n\tif v != nil && v.Domain != nil {\n\t\treturn *v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetScheduleToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetScheduleToStartTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetHeartbeatTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ActivityTaskScheduledEventAttributes) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskScheduledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskStartedEventAttributes is an internal type (TBD...)\ntype ActivityTaskStartedEventAttributes struct {\n\tScheduledEventID   int64   `json:\"scheduledEventId,omitempty\"`\n\tIdentity           string  `json:\"identity,omitempty\"`\n\tRequestID          string  `json:\"requestId,omitempty\"`\n\tAttempt            int32   `json:\"attempt,omitempty\"`\n\tLastFailureReason  *string `json:\"lastFailureReason,omitempty\"`\n\tLastFailureDetails []byte  `json:\"lastFailureDetails,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskStartedEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *ActivityTaskStartedEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskStartedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityTaskTimedOutEventAttributes is an internal type (TBD...)\ntype ActivityTaskTimedOutEventAttributes struct {\n\tDetails            []byte       `json:\"details,omitempty\"`\n\tScheduledEventID   int64        `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID     int64        `json:\"startedEventId,omitempty\"`\n\tTimeoutType        *TimeoutType `json:\"timeoutType,omitempty\"`\n\tLastFailureReason  *string      `json:\"lastFailureReason,omitempty\"`\n\tLastFailureDetails []byte       `json:\"lastFailureDetails,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *ActivityTaskTimedOutEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetTimeoutType is an internal getter (TBD...)\nfunc (v *ActivityTaskTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ActivityTaskTimedOutEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ActivityType is an internal type (TBD...)\ntype ActivityType struct {\n\tName string `json:\"name,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *ActivityType) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// ArchivalStatus is an internal type (TBD...)\ntype ArchivalStatus int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e ArchivalStatus) Ptr() *ArchivalStatus {\n\treturn &e\n}\n\n// String returns a readable string representation of ArchivalStatus.\nfunc (e ArchivalStatus) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"DISABLED\"\n\tcase 1:\n\t\treturn \"ENABLED\"\n\t}\n\treturn fmt.Sprintf(\"ArchivalStatus(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *ArchivalStatus) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"DISABLED\":\n\t\t*e = ArchivalStatusDisabled\n\t\treturn nil\n\tcase \"ENABLED\":\n\t\t*e = ArchivalStatusEnabled\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ArchivalStatus\", err)\n\t\t}\n\t\t*e = ArchivalStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ArchivalStatus to text.\nfunc (e ArchivalStatus) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (e *ArchivalStatus) ByteSize() uint64 {\n\tif e == nil {\n\t\treturn 0\n\t}\n\treturn uint64(unsafe.Sizeof(*e))\n}\n\nconst (\n\t// ArchivalStatusDisabled is an option for ArchivalStatus\n\tArchivalStatusDisabled ArchivalStatus = iota\n\t// ArchivalStatusEnabled is an option for ArchivalStatus\n\tArchivalStatusEnabled\n)\n\n// BadBinaries is an internal type (TBD...)\ntype BadBinaries struct {\n\tBinaries map[string]*BadBinaryInfo `json:\"binaries,omitempty\"`\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (b *BadBinaries) ByteSize() uint64 {\n\tif b == nil {\n\t\treturn 0\n\t}\n\tsize := uint64(unsafe.Sizeof(*b))\n\tif b.Binaries != nil {\n\t\tfor k, v := range b.Binaries {\n\t\t\tsize += uint64(len(k)) + v.ByteSize()\n\t\t}\n\t}\n\treturn size\n}\n\n// DeepCopy returns a deep copy of BadBinaries\nfunc (b *BadBinaries) DeepCopy() BadBinaries {\n\tif b == nil {\n\t\treturn BadBinaries{}\n\t}\n\tresult := BadBinaries{}\n\tif b.Binaries != nil {\n\t\tresult.Binaries = make(map[string]*BadBinaryInfo, len(b.Binaries))\n\t\tfor k, v := range b.Binaries {\n\t\t\tif v != nil {\n\t\t\t\tcopyV := &BadBinaryInfo{\n\t\t\t\t\tReason:   v.Reason,\n\t\t\t\t\tOperator: v.Operator,\n\t\t\t\t}\n\t\t\t\tif v.CreatedTimeNano != nil {\n\t\t\t\t\tcreatedTime := *v.CreatedTimeNano\n\t\t\t\t\tcopyV.CreatedTimeNano = &createdTime\n\t\t\t\t}\n\t\t\t\tresult.Binaries[k] = copyV\n\t\t\t} else {\n\t\t\t\t// Preserve nil entries in the map\n\t\t\t\tresult.Binaries[k] = nil\n\t\t\t}\n\t\t}\n\t}\n\treturn result\n}\n\n// BadBinaryInfo is an internal type (TBD...)\ntype BadBinaryInfo struct {\n\tReason          string `json:\"reason,omitempty\"`\n\tOperator        string `json:\"operator,omitempty\"`\n\tCreatedTimeNano *int64 `json:\"createdTimeNano,omitempty\"`\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *BadBinaryInfo) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\n// GetOperator is an internal getter (TBD...)\nfunc (v *BadBinaryInfo) GetOperator() (o string) {\n\tif v != nil {\n\t\treturn v.Operator\n\t}\n\treturn\n}\n\n// GetCreatedTimeNano is an internal getter (TBD...)\nfunc (v *BadBinaryInfo) GetCreatedTimeNano() (o int64) {\n\tif v != nil && v.CreatedTimeNano != nil {\n\t\treturn *v.CreatedTimeNano\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *BadBinaryInfo) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.Reason))\n\tsize += uint64(len(v.Operator))\n\tif v.CreatedTimeNano != nil {\n\t\tsize += uint64(unsafe.Sizeof(*v.CreatedTimeNano))\n\t}\n\treturn size\n}\n\n// BadRequestError is an internal type (TBD...)\n// TODO(c-warren): Move to common/types/errors.go\ntype BadRequestError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// CancelExternalWorkflowExecutionFailedCause is an internal type (TBD...)\ntype CancelExternalWorkflowExecutionFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e CancelExternalWorkflowExecutionFailedCause) Ptr() *CancelExternalWorkflowExecutionFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of CancelExternalWorkflowExecutionFailedCause.\nfunc (e CancelExternalWorkflowExecutionFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"\n\tcase 1:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\t}\n\treturn fmt.Sprintf(\"CancelExternalWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *CancelExternalWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\":\n\t\t*e = CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*e = CancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CancelExternalWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*e = CancelExternalWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CancelExternalWorkflowExecutionFailedCause to text.\nfunc (e CancelExternalWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution is an option for CancelExternalWorkflowExecutionFailedCause\n\tCancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution CancelExternalWorkflowExecutionFailedCause = iota\n\tCancelExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n)\n\n// CancelTimerDecisionAttributes is an internal type (TBD...)\ntype CancelTimerDecisionAttributes struct {\n\tTimerID string `json:\"timerId,omitempty\"`\n}\n\n// GetTimerID is an internal getter (TBD...)\nfunc (v *CancelTimerDecisionAttributes) GetTimerID() (o string) {\n\tif v != nil {\n\t\treturn v.TimerID\n\t}\n\treturn\n}\n\n// CancelTimerFailedEventAttributes is an internal type (TBD...)\ntype CancelTimerFailedEventAttributes struct {\n\tTimerID                      string `json:\"timerId,omitempty\"`\n\tCause                        string `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tIdentity                     string `json:\"identity,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *CancelTimerFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// CancelWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype CancelWorkflowExecutionDecisionAttributes struct {\n\tDetails []byte `json:\"details,omitempty\"`\n}\n\n// CancellationAlreadyRequestedError is an internal type (TBD...)\ntype CancellationAlreadyRequestedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ChildWorkflowExecutionCanceledEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionCanceledEventAttributes struct {\n\tDetails           []byte             `json:\"details,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tStartedEventID    int64              `json:\"startedEventId,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionCanceledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ChildWorkflowExecutionCompletedEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionCompletedEventAttributes struct {\n\tResult            []byte             `json:\"result,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tStartedEventID    int64              `json:\"startedEventId,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionCompletedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ChildWorkflowExecutionFailedCause is an internal type (TBD...)\ntype ChildWorkflowExecutionFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e ChildWorkflowExecutionFailedCause) Ptr() *ChildWorkflowExecutionFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of ChildWorkflowExecutionFailedCause.\nfunc (e ChildWorkflowExecutionFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"WORKFLOW_ALREADY_RUNNING\"\n\t}\n\treturn fmt.Sprintf(\"ChildWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *ChildWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"WORKFLOW_ALREADY_RUNNING\":\n\t\t*e = ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ChildWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*e = ChildWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ChildWorkflowExecutionFailedCause to text.\nfunc (e ChildWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning is an option for ChildWorkflowExecutionFailedCause\n\tChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning ChildWorkflowExecutionFailedCause = iota\n)\n\n// ChildWorkflowExecutionFailedEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionFailedEventAttributes struct {\n\tReason            *string            `json:\"reason,omitempty\"`\n\tDetails           []byte             `json:\"details,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tStartedEventID    int64              `json:\"startedEventId,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ChildWorkflowExecutionStartedEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionStartedEventAttributes struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tHeader            *Header            `json:\"header,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionStartedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ChildWorkflowExecutionTerminatedEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionTerminatedEventAttributes struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tStartedEventID    int64              `json:\"startedEventId,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionTerminatedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ChildWorkflowExecutionTimedOutEventAttributes is an internal type (TBD...)\ntype ChildWorkflowExecutionTimedOutEventAttributes struct {\n\tTimeoutType       *TimeoutType       `json:\"timeoutType,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tWorkflowType      *WorkflowType      `json:\"workflowType,omitempty\"`\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tStartedEventID    int64              `json:\"startedEventId,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ChildWorkflowExecutionTimedOutEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ClientVersionNotSupportedError is an internal type (TBD...)\ntype ClientVersionNotSupportedError struct {\n\tFeatureVersion    string `json:\"featureVersion,required\"`\n\tClientImpl        string `json:\"clientImpl,required\"`\n\tSupportedVersions string `json:\"supportedVersions,required\"`\n}\n\n// FeatureNotEnabledError is an internal type (TBD...)\ntype FeatureNotEnabledError struct {\n\tFeatureFlag string `json:\"featureFlag,required\"`\n}\n\n// CloseShardRequest is an internal type (TBD...)\ntype CloseShardRequest struct {\n\tShardID int32 `json:\"shardID,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *CloseShardRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// ClusterInfo is an internal type (TBD...)\ntype ClusterInfo struct {\n\tSupportedClientVersions *SupportedClientVersions `json:\"supportedClientVersions,omitempty\"`\n}\n\n// ClusterReplicationConfiguration is an internal type (TBD...)\ntype ClusterReplicationConfiguration struct {\n\tClusterName string `json:\"clusterName,omitempty\"`\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *ClusterReplicationConfiguration) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *ClusterReplicationConfiguration) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.ClusterName))\n\treturn size\n}\n\n// CompleteWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype CompleteWorkflowExecutionDecisionAttributes struct {\n\tResult []byte `json:\"result,omitempty\"`\n}\n\n// ContinueAsNewInitiator is an internal type (TBD...)\ntype ContinueAsNewInitiator int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e ContinueAsNewInitiator) Ptr() *ContinueAsNewInitiator {\n\treturn &e\n}\n\n// String returns a readable string representation of ContinueAsNewInitiator.\nfunc (e ContinueAsNewInitiator) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Decider\"\n\tcase 1:\n\t\treturn \"RetryPolicy\"\n\tcase 2:\n\t\treturn \"CronSchedule\"\n\t}\n\treturn fmt.Sprintf(\"ContinueAsNewInitiator(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *ContinueAsNewInitiator) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"DECIDER\":\n\t\t*e = ContinueAsNewInitiatorDecider\n\t\treturn nil\n\tcase \"RETRYPOLICY\":\n\t\t*e = ContinueAsNewInitiatorRetryPolicy\n\t\treturn nil\n\tcase \"CRONSCHEDULE\":\n\t\t*e = ContinueAsNewInitiatorCronSchedule\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ContinueAsNewInitiator\", err)\n\t\t}\n\t\t*e = ContinueAsNewInitiator(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ContinueAsNewInitiator to text.\nfunc (e ContinueAsNewInitiator) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// ContinueAsNewInitiatorDecider is an option for ContinueAsNewInitiator\n\tContinueAsNewInitiatorDecider ContinueAsNewInitiator = iota\n\t// ContinueAsNewInitiatorRetryPolicy is an option for ContinueAsNewInitiator\n\tContinueAsNewInitiatorRetryPolicy\n\t// ContinueAsNewInitiatorCronSchedule is an option for ContinueAsNewInitiator\n\tContinueAsNewInitiatorCronSchedule\n)\n\n// ContinueAsNewWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype ContinueAsNewWorkflowExecutionDecisionAttributes struct {\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tBackoffStartIntervalInSeconds       *int32                        `json:\"backoffStartIntervalInSeconds,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tFailureReason                       *string                       `json:\"failureReason,omitempty\"`\n\tFailureDetails                      []byte                        `json:\"failureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetTaskStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetBackoffStartIntervalInSeconds is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetBackoffStartIntervalInSeconds() (o int32) {\n\tif v != nil && v.BackoffStartIntervalInSeconds != nil {\n\t\treturn *v.BackoffStartIntervalInSeconds\n\t}\n\treturn\n}\n\n// GetJitterStartSeconds is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\treturn\n}\n\n// GetInitiator is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\treturn\n}\n\n// GetSearchAttributes is an internal getter (TBD...)\nfunc (v *ContinueAsNewWorkflowExecutionDecisionAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn\n}\n\n// CountWorkflowExecutionsRequest is an internal type (TBD...)\ntype CountWorkflowExecutionsRequest struct {\n\tDomain string `json:\"domain,omitempty\"`\n\tQuery  string `json:\"query,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *CountWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetQuery is an internal getter (TBD...)\nfunc (v *CountWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil {\n\t\treturn v.Query\n\t}\n\treturn\n}\n\n// CountWorkflowExecutionsResponse is an internal type (TBD...)\ntype CountWorkflowExecutionsResponse struct {\n\tCount int64 `json:\"count,omitempty\"`\n}\n\n// GetCount is an internal getter (TBD...)\nfunc (v *CountWorkflowExecutionsResponse) GetCount() (o int64) {\n\tif v != nil {\n\t\treturn v.Count\n\t}\n\treturn\n}\n\n// CurrentBranchChangedError is an internal type (TBD...)\ntype CurrentBranchChangedError struct {\n\tMessage            string `json:\"message,required\"`\n\tCurrentBranchToken []byte `json:\"currentBranchToken,required\"`\n}\n\n// GetCurrentBranchToken is an internal getter (TBD...)\nfunc (v *CurrentBranchChangedError) GetCurrentBranchToken() (o []byte) {\n\tif v != nil && v.CurrentBranchToken != nil {\n\t\treturn v.CurrentBranchToken\n\t}\n\treturn\n}\n\n// DataBlob is an internal type (TBD...)\ntype DataBlob struct {\n\tEncodingType *EncodingType `json:\"EncodingType,omitempty\"`\n\tData         []byte        `json:\"Data,omitempty\"`\n}\n\n// GetEncodingType is an internal getter (TBD...)\nfunc (v *DataBlob) GetEncodingType() (o EncodingType) {\n\tif v != nil && v.EncodingType != nil {\n\t\treturn *v.EncodingType\n\t}\n\treturn\n}\n\n// GetData is an internal getter (TBD...)\nfunc (v *DataBlob) GetData() (o []byte) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\treturn\n}\n\nfunc (v *DataBlob) DeepCopy() *DataBlob {\n\tif v == nil {\n\t\treturn nil\n\t}\n\n\tres := &DataBlob{\n\t\tEncodingType: v.EncodingType,\n\t}\n\n\tif v.Data != nil {\n\t\tres.Data = make([]byte, len(v.Data))\n\t\tcopy(res.Data, v.Data)\n\t}\n\n\treturn res\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *DataBlob) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += v.EncodingType.ByteSize()\n\tsize += uint64(len(v.Data))\n\treturn size\n}\n\n// Decision is an internal type (TBD...)\ntype Decision struct {\n\tDecisionType                                             *DecisionType                                             `json:\"decisionType,omitempty\"`\n\tScheduleActivityTaskDecisionAttributes                   *ScheduleActivityTaskDecisionAttributes                   `json:\"scheduleActivityTaskDecisionAttributes,omitempty\"`\n\tStartTimerDecisionAttributes                             *StartTimerDecisionAttributes                             `json:\"startTimerDecisionAttributes,omitempty\"`\n\tCompleteWorkflowExecutionDecisionAttributes              *CompleteWorkflowExecutionDecisionAttributes              `json:\"completeWorkflowExecutionDecisionAttributes,omitempty\"`\n\tFailWorkflowExecutionDecisionAttributes                  *FailWorkflowExecutionDecisionAttributes                  `json:\"failWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRequestCancelActivityTaskDecisionAttributes              *RequestCancelActivityTaskDecisionAttributes              `json:\"requestCancelActivityTaskDecisionAttributes,omitempty\"`\n\tCancelTimerDecisionAttributes                            *CancelTimerDecisionAttributes                            `json:\"cancelTimerDecisionAttributes,omitempty\"`\n\tCancelWorkflowExecutionDecisionAttributes                *CancelWorkflowExecutionDecisionAttributes                `json:\"cancelWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionDecisionAttributes *RequestCancelExternalWorkflowExecutionDecisionAttributes `json:\"requestCancelExternalWorkflowExecutionDecisionAttributes,omitempty\"`\n\tRecordMarkerDecisionAttributes                           *RecordMarkerDecisionAttributes                           `json:\"recordMarkerDecisionAttributes,omitempty\"`\n\tContinueAsNewWorkflowExecutionDecisionAttributes         *ContinueAsNewWorkflowExecutionDecisionAttributes         `json:\"continueAsNewWorkflowExecutionDecisionAttributes,omitempty\"`\n\tStartChildWorkflowExecutionDecisionAttributes            *StartChildWorkflowExecutionDecisionAttributes            `json:\"startChildWorkflowExecutionDecisionAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionDecisionAttributes        *SignalExternalWorkflowExecutionDecisionAttributes        `json:\"signalExternalWorkflowExecutionDecisionAttributes,omitempty\"`\n\tUpsertWorkflowSearchAttributesDecisionAttributes         *UpsertWorkflowSearchAttributesDecisionAttributes         `json:\"upsertWorkflowSearchAttributesDecisionAttributes,omitempty\"`\n}\n\n// GetDecisionType is an internal getter (TBD...)\nfunc (v *Decision) GetDecisionType() (o DecisionType) {\n\tif v != nil && v.DecisionType != nil {\n\t\treturn *v.DecisionType\n\t}\n\treturn\n}\n\n// DecisionTaskCompletedEventAttributes is an internal type (TBD...)\ntype DecisionTaskCompletedEventAttributes struct {\n\tExecutionContext []byte `json:\"executionContext,omitempty\"`\n\tScheduledEventID int64  `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID   int64  `json:\"startedEventId,omitempty\"`\n\tIdentity         string `json:\"identity,omitempty\"`\n\tBinaryChecksum   string `json:\"binaryChecksum,omitempty\"`\n}\n\n// GetStartedEventID is an internal getter (TBD...)\nfunc (v *DecisionTaskCompletedEventAttributes) GetStartedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedEventID\n\t}\n\treturn\n}\n\n// GetBinaryChecksum is an internal getter (TBD...)\nfunc (v *DecisionTaskCompletedEventAttributes) GetBinaryChecksum() (o string) {\n\tif v != nil {\n\t\treturn v.BinaryChecksum\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *DecisionTaskCompletedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// DecisionTaskFailedCause is an internal type (TBD...)\ntype DecisionTaskFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DecisionTaskFailedCause) Ptr() *DecisionTaskFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of DecisionTaskFailedCause.\nfunc (e DecisionTaskFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNHANDLED_DECISION\"\n\tcase 1:\n\t\treturn \"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\"\n\tcase 2:\n\t\treturn \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\"\n\tcase 3:\n\t\treturn \"BAD_START_TIMER_ATTRIBUTES\"\n\tcase 4:\n\t\treturn \"BAD_CANCEL_TIMER_ATTRIBUTES\"\n\tcase 5:\n\t\treturn \"BAD_RECORD_MARKER_ATTRIBUTES\"\n\tcase 6:\n\t\treturn \"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 7:\n\t\treturn \"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 8:\n\t\treturn \"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 9:\n\t\treturn \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 10:\n\t\treturn \"BAD_CONTINUE_AS_NEW_ATTRIBUTES\"\n\tcase 11:\n\t\treturn \"START_TIMER_DUPLICATE_I_D\"\n\tcase 12:\n\t\treturn \"RESET_STICKY_TASKLIST\"\n\tcase 13:\n\t\treturn \"WORKFLOW_WORKER_UNHANDLED_FAILURE\"\n\tcase 14:\n\t\treturn \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\tcase 15:\n\t\treturn \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\"\n\tcase 16:\n\t\treturn \"FORCE_CLOSE_DECISION\"\n\tcase 17:\n\t\treturn \"FAILOVER_CLOSE_DECISION\"\n\tcase 18:\n\t\treturn \"BAD_SIGNAL_INPUT_SIZE\"\n\tcase 19:\n\t\treturn \"RESET_WORKFLOW\"\n\tcase 20:\n\t\treturn \"BAD_BINARY\"\n\tcase 21:\n\t\treturn \"SCHEDULE_ACTIVITY_DUPLICATE_I_D\"\n\tcase 22:\n\t\treturn \"BAD_SEARCH_ATTRIBUTES\"\n\t}\n\treturn fmt.Sprintf(\"DecisionTaskFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DecisionTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"UNHANDLED_DECISION\":\n\t\t*e = DecisionTaskFailedCauseUnhandledDecision\n\t\treturn nil\n\tcase \"BAD_SCHEDULE_ACTIVITY_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadScheduleActivityAttributes\n\t\treturn nil\n\tcase \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadRequestCancelActivityAttributes\n\t\treturn nil\n\tcase \"BAD_START_TIMER_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadStartTimerAttributes\n\t\treturn nil\n\tcase \"BAD_CANCEL_TIMER_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadCancelTimerAttributes\n\t\treturn nil\n\tcase \"BAD_RECORD_MARKER_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadRecordMarkerAttributes\n\t\treturn nil\n\tcase \"BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_CONTINUE_AS_NEW_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadContinueAsNewAttributes\n\t\treturn nil\n\tcase \"START_TIMER_DUPLICATE_I_D\":\n\t\t*e = DecisionTaskFailedCauseStartTimerDuplicateID\n\t\treturn nil\n\tcase \"RESET_STICKY_TASKLIST\":\n\t\t*e = DecisionTaskFailedCauseResetStickyTasklist\n\t\treturn nil\n\tcase \"WORKFLOW_WORKER_UNHANDLED_FAILURE\":\n\t\t*e = DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t\treturn nil\n\tcase \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes\n\t\treturn nil\n\tcase \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadStartChildExecutionAttributes\n\t\treturn nil\n\tcase \"FORCE_CLOSE_DECISION\":\n\t\t*e = DecisionTaskFailedCauseForceCloseDecision\n\t\treturn nil\n\tcase \"FAILOVER_CLOSE_DECISION\":\n\t\t*e = DecisionTaskFailedCauseFailoverCloseDecision\n\t\treturn nil\n\tcase \"BAD_SIGNAL_INPUT_SIZE\":\n\t\t*e = DecisionTaskFailedCauseBadSignalInputSize\n\t\treturn nil\n\tcase \"RESET_WORKFLOW\":\n\t\t*e = DecisionTaskFailedCauseResetWorkflow\n\t\treturn nil\n\tcase \"BAD_BINARY\":\n\t\t*e = DecisionTaskFailedCauseBadBinary\n\t\treturn nil\n\tcase \"SCHEDULE_ACTIVITY_DUPLICATE_I_D\":\n\t\t*e = DecisionTaskFailedCauseScheduleActivityDuplicateID\n\t\treturn nil\n\tcase \"BAD_SEARCH_ATTRIBUTES\":\n\t\t*e = DecisionTaskFailedCauseBadSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionTaskFailedCause\", err)\n\t\t}\n\t\t*e = DecisionTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionTaskFailedCause to text.\nfunc (e DecisionTaskFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// DecisionTaskFailedCauseUnhandledDecision is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseUnhandledDecision DecisionTaskFailedCause = iota\n\t// DecisionTaskFailedCauseBadScheduleActivityAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadScheduleActivityAttributes\n\t// DecisionTaskFailedCauseBadRequestCancelActivityAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadRequestCancelActivityAttributes\n\t// DecisionTaskFailedCauseBadStartTimerAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadStartTimerAttributes\n\t// DecisionTaskFailedCauseBadCancelTimerAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadCancelTimerAttributes\n\t// DecisionTaskFailedCauseBadRecordMarkerAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadRecordMarkerAttributes\n\t// DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes\n\t// DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadFailWorkflowExecutionAttributes\n\t// DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\t// DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes\n\t// DecisionTaskFailedCauseBadContinueAsNewAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadContinueAsNewAttributes\n\t// DecisionTaskFailedCauseStartTimerDuplicateID is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseStartTimerDuplicateID\n\t// DecisionTaskFailedCauseResetStickyTasklist is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseResetStickyTasklist\n\t// DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t// DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes\n\t// DecisionTaskFailedCauseBadStartChildExecutionAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadStartChildExecutionAttributes\n\t// DecisionTaskFailedCauseForceCloseDecision is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseForceCloseDecision\n\t// DecisionTaskFailedCauseFailoverCloseDecision is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseFailoverCloseDecision\n\t// DecisionTaskFailedCauseBadSignalInputSize is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadSignalInputSize\n\t// DecisionTaskFailedCauseResetWorkflow is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseResetWorkflow\n\t// DecisionTaskFailedCauseBadBinary is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadBinary\n\t// DecisionTaskFailedCauseScheduleActivityDuplicateID is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseScheduleActivityDuplicateID\n\t// DecisionTaskFailedCauseBadSearchAttributes is an option for DecisionTaskFailedCause\n\tDecisionTaskFailedCauseBadSearchAttributes\n)\n\n// DecisionTaskFailedEventAttributes is an internal type (TBD...)\ntype DecisionTaskFailedEventAttributes struct {\n\tScheduledEventID int64                    `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID   int64                    `json:\"startedEventId,omitempty\"`\n\tCause            *DecisionTaskFailedCause `json:\"cause,omitempty\"`\n\tDetails          []byte                   `json:\"details,omitempty\"`\n\tIdentity         string                   `json:\"identity,omitempty\"`\n\tReason           *string                  `json:\"reason,omitempty\"`\n\tBaseRunID        string                   `json:\"baseRunId,omitempty\"`\n\tNewRunID         string                   `json:\"newRunId,omitempty\"`\n\tForkEventVersion int64                    `json:\"forkEventVersion,omitempty\"`\n\tBinaryChecksum   string                   `json:\"binaryChecksum,omitempty\"`\n\tRequestID        string                   `json:\"requestId,omitempty\"`\n}\n\n// GetCause is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetCause() (o DecisionTaskFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\treturn\n}\n\n// GetDetails is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\treturn\n}\n\n// GetBaseRunID is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetBaseRunID() (o string) {\n\tif v != nil {\n\t\treturn v.BaseRunID\n\t}\n\treturn\n}\n\n// GetNewRunID is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetNewRunID() (o string) {\n\tif v != nil {\n\t\treturn v.NewRunID\n\t}\n\treturn\n}\n\n// GetForkEventVersion is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetForkEventVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.ForkEventVersion\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *DecisionTaskFailedEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *DecisionTaskFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// DecisionTaskScheduledEventAttributes is an internal type (TBD...)\ntype DecisionTaskScheduledEventAttributes struct {\n\tTaskList                   *TaskList `json:\"taskList,omitempty\"`\n\tStartToCloseTimeoutSeconds *int32    `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tAttempt                    int64     `json:\"attempt,omitempty\"`\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *DecisionTaskScheduledEventAttributes) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *DecisionTaskScheduledEventAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *DecisionTaskScheduledEventAttributes) GetAttempt() (o int64) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *DecisionTaskScheduledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// DecisionTaskStartedEventAttributes is an internal type (TBD...)\ntype DecisionTaskStartedEventAttributes struct {\n\tScheduledEventID int64  `json:\"scheduledEventId,omitempty\"`\n\tIdentity         string `json:\"identity,omitempty\"`\n\tRequestID        string `json:\"requestId,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *DecisionTaskStartedEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *DecisionTaskStartedEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *DecisionTaskStartedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// DecisionTaskTimedOutCause is an internal type (TBD...)\ntype DecisionTaskTimedOutCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DecisionTaskTimedOutCause) Ptr() *DecisionTaskTimedOutCause {\n\treturn &e\n}\n\n// String returns a readable string representation of DecisionTaskTimedOutCause.\nfunc (e DecisionTaskTimedOutCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Timeout\"\n\tcase 1:\n\t\treturn \"Reset\"\n\t}\n\treturn fmt.Sprintf(\"DecisionTaskTimedOutCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DecisionTaskTimedOutCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToLower(string(value)); s {\n\tcase \"timeout\":\n\t\t*e = DecisionTaskTimedOutCauseTimeout\n\t\treturn nil\n\tcase \"reset\":\n\t\t*e = DecisionTaskTimedOutCauseReset\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionTaskTimedOutCause\", err)\n\t\t}\n\t\t*e = DecisionTaskTimedOutCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionTaskFailedCause to text.\nfunc (e DecisionTaskTimedOutCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// DecisionTaskTimedOutCauseTimeout is an option for DecisionTaskTimedOutCause\n\tDecisionTaskTimedOutCauseTimeout DecisionTaskTimedOutCause = iota\n\t// DecisionTaskTimedOutCauseReset is an option for DecisionTaskTimedOutCause\n\tDecisionTaskTimedOutCauseReset\n)\n\n// DecisionTaskTimedOutEventAttributes is an internal type (TBD...)\ntype DecisionTaskTimedOutEventAttributes struct {\n\tScheduledEventID int64                      `json:\"scheduledEventId,omitempty\"`\n\tStartedEventID   int64                      `json:\"startedEventId,omitempty\"`\n\tTimeoutType      *TimeoutType               `json:\"timeoutType,omitempty\"`\n\tBaseRunID        string                     `json:\"baseRunId,omitempty\"`\n\tNewRunID         string                     `json:\"newRunId,omitempty\"`\n\tForkEventVersion int64                      `json:\"forkEventVersion,omitempty\"`\n\tReason           string                     `json:\"reason,omitempty\"`\n\tCause            *DecisionTaskTimedOutCause `json:\"cause,omitempty\"`\n\tRequestID        string                     `json:\"requestId,omitempty\"`\n}\n\n// GetScheduledEventID is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetScheduledEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduledEventID\n\t}\n\treturn\n}\n\n// GetTimeoutType is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\treturn\n}\n\n// GetBaseRunID is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetBaseRunID() (o string) {\n\tif v != nil {\n\t\treturn v.BaseRunID\n\t}\n\treturn\n}\n\n// GetNewRunID is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetNewRunID() (o string) {\n\tif v != nil {\n\t\treturn v.NewRunID\n\t}\n\treturn\n}\n\n// GetForkEventVersion is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetForkEventVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.ForkEventVersion\n\t}\n\treturn\n}\n\n// GetCause is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetCause() (o DecisionTaskTimedOutCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *DecisionTaskTimedOutEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *DecisionTaskTimedOutEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// DecisionType is an internal type (TBD...)\ntype DecisionType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DecisionType) Ptr() *DecisionType {\n\treturn &e\n}\n\n// String returns a readable string representation of DecisionType.\nfunc (e DecisionType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ScheduleActivityTask\"\n\tcase 1:\n\t\treturn \"RequestCancelActivityTask\"\n\tcase 2:\n\t\treturn \"StartTimer\"\n\tcase 3:\n\t\treturn \"CompleteWorkflowExecution\"\n\tcase 4:\n\t\treturn \"FailWorkflowExecution\"\n\tcase 5:\n\t\treturn \"CancelTimer\"\n\tcase 6:\n\t\treturn \"CancelWorkflowExecution\"\n\tcase 7:\n\t\treturn \"RequestCancelExternalWorkflowExecution\"\n\tcase 8:\n\t\treturn \"RecordMarker\"\n\tcase 9:\n\t\treturn \"ContinueAsNewWorkflowExecution\"\n\tcase 10:\n\t\treturn \"StartChildWorkflowExecution\"\n\tcase 11:\n\t\treturn \"SignalExternalWorkflowExecution\"\n\tcase 12:\n\t\treturn \"UpsertWorkflowSearchAttributes\"\n\t}\n\treturn fmt.Sprintf(\"DecisionType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DecisionType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"SCHEDULEACTIVITYTASK\":\n\t\t*e = DecisionTypeScheduleActivityTask\n\t\treturn nil\n\tcase \"REQUESTCANCELACTIVITYTASK\":\n\t\t*e = DecisionTypeRequestCancelActivityTask\n\t\treturn nil\n\tcase \"STARTTIMER\":\n\t\t*e = DecisionTypeStartTimer\n\t\treturn nil\n\tcase \"COMPLETEWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeCompleteWorkflowExecution\n\t\treturn nil\n\tcase \"FAILWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeFailWorkflowExecution\n\t\treturn nil\n\tcase \"CANCELTIMER\":\n\t\t*e = DecisionTypeCancelTimer\n\t\treturn nil\n\tcase \"CANCELWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeCancelWorkflowExecution\n\t\treturn nil\n\tcase \"REQUESTCANCELEXTERNALWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeRequestCancelExternalWorkflowExecution\n\t\treturn nil\n\tcase \"RECORDMARKER\":\n\t\t*e = DecisionTypeRecordMarker\n\t\treturn nil\n\tcase \"CONTINUEASNEWWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeContinueAsNewWorkflowExecution\n\t\treturn nil\n\tcase \"STARTCHILDWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeStartChildWorkflowExecution\n\t\treturn nil\n\tcase \"SIGNALEXTERNALWORKFLOWEXECUTION\":\n\t\t*e = DecisionTypeSignalExternalWorkflowExecution\n\t\treturn nil\n\tcase \"UPSERTWORKFLOWSEARCHATTRIBUTES\":\n\t\t*e = DecisionTypeUpsertWorkflowSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DecisionType\", err)\n\t\t}\n\t\t*e = DecisionType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DecisionType to text.\nfunc (e DecisionType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// DecisionTypeScheduleActivityTask is an option for DecisionType\n\tDecisionTypeScheduleActivityTask DecisionType = iota\n\t// DecisionTypeRequestCancelActivityTask is an option for DecisionType\n\tDecisionTypeRequestCancelActivityTask\n\t// DecisionTypeStartTimer is an option for DecisionType\n\tDecisionTypeStartTimer\n\t// DecisionTypeCompleteWorkflowExecution is an option for DecisionType\n\tDecisionTypeCompleteWorkflowExecution\n\t// DecisionTypeFailWorkflowExecution is an option for DecisionType\n\tDecisionTypeFailWorkflowExecution\n\t// DecisionTypeCancelTimer is an option for DecisionType\n\tDecisionTypeCancelTimer\n\t// DecisionTypeCancelWorkflowExecution is an option for DecisionType\n\tDecisionTypeCancelWorkflowExecution\n\t// DecisionTypeRequestCancelExternalWorkflowExecution is an option for DecisionType\n\tDecisionTypeRequestCancelExternalWorkflowExecution\n\t// DecisionTypeRecordMarker is an option for DecisionType\n\tDecisionTypeRecordMarker\n\t// DecisionTypeContinueAsNewWorkflowExecution is an option for DecisionType\n\tDecisionTypeContinueAsNewWorkflowExecution\n\t// DecisionTypeStartChildWorkflowExecution is an option for DecisionType\n\tDecisionTypeStartChildWorkflowExecution\n\t// DecisionTypeSignalExternalWorkflowExecution is an option for DecisionType\n\tDecisionTypeSignalExternalWorkflowExecution\n\t// DecisionTypeUpsertWorkflowSearchAttributes is an option for DecisionType\n\tDecisionTypeUpsertWorkflowSearchAttributes\n)\n\n// DeleteDomainRequest is an internal type (TBD...)\ntype DeleteDomainRequest struct {\n\tName          string `json:\"name,omitempty\"`\n\tSecurityToken string `json:\"securityToken,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *DeleteDomainRequest) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// DeprecateDomainRequest is an internal type (TBD...)\ntype DeprecateDomainRequest struct {\n\tName          string `json:\"name,omitempty\"`\n\tSecurityToken string `json:\"securityToken,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *DeprecateDomainRequest) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// DescribeDomainRequest is an internal type (TBD...)\ntype DescribeDomainRequest struct {\n\tName *string `json:\"name,omitempty\"`\n\tUUID *string `json:\"uuid,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *DescribeDomainRequest) GetName() (o string) {\n\tif v != nil && v.Name != nil {\n\t\treturn *v.Name\n\t}\n\treturn\n}\n\n// GetUUID is an internal getter (TBD...)\nfunc (v *DescribeDomainRequest) GetUUID() (o string) {\n\tif v != nil && v.UUID != nil {\n\t\treturn *v.UUID\n\t}\n\treturn\n}\n\n// DescribeDomainResponse is an internal type (TBD...)\ntype DescribeDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          int64                           `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           bool                            `json:\"isGlobalDomain,omitempty\"`\n\tFailoverInfo             *FailoverInfo                   `json:\"failoverInfo,omitempty\"`\n}\n\nfunc (v *DomainReplicationConfiguration) GetActiveClusters() (o *ActiveClusters) {\n\tif v != nil && v.ActiveClusters != nil {\n\t\treturn v.ActiveClusters\n\t}\n\treturn\n}\n\n// GetDomainInfo is an internal getter (TBD...)\nfunc (v *DescribeDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\treturn\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *DescribeDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetIsGlobalDomain is an internal getter (TBD...)\nfunc (v *DescribeDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil {\n\t\treturn v.IsGlobalDomain\n\t}\n\treturn\n}\n\n// GetFailoverInfo is an internal getter (TBD...)\nfunc (v *DescribeDomainResponse) GetFailoverInfo() (o *FailoverInfo) {\n\tif v != nil {\n\t\treturn v.FailoverInfo\n\t}\n\treturn\n}\n\n// FailoverDomainRequest is an internal type (TBD...)\ntype FailoverDomainRequest struct {\n\tDomainName              string          `json:\"domainName,omitempty\"`\n\tDomainActiveClusterName *string         `json:\"domainActiveClusterName,omitempty\"`\n\tActiveClusters          *ActiveClusters `json:\"activeClusters,omitempty\"`\n\tReason                  *string         `json:\"reason,omitempty\"`\n}\n\nfunc (v *FailoverDomainRequest) ToUpdateDomainRequest() *UpdateDomainRequest {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn &UpdateDomainRequest{\n\t\tName:              v.DomainName,\n\t\tActiveClusterName: v.DomainActiveClusterName,\n\t\tActiveClusters:    v.ActiveClusters,\n\t\tFailoverReason:    v.Reason,\n\t}\n}\n\n// GetDomainName is an internal getter (TBD...)\nfunc (v *FailoverDomainRequest) GetDomainName() (o string) {\n\tif v != nil {\n\t\treturn v.DomainName\n\t}\n\treturn\n}\n\n// GetDomainActiveClusterName is an internal getter (TBD...)\nfunc (v *FailoverDomainRequest) GetDomainActiveClusterName() (o string) {\n\tif v != nil && v.DomainActiveClusterName != nil {\n\t\treturn *v.DomainActiveClusterName\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *FailoverDomainRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.DomainName\n\t}\n\treturn\n}\n\n// FailoverDomainResponse is an internal type (TBD...)\ntype FailoverDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          int64                           `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           bool                            `json:\"isGlobalDomain,omitempty\"`\n}\n\n// GetDomainInfo is an internal getter (TBD...)\nfunc (v *FailoverDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\treturn\n}\n\n// GetConfiguration is an internal getter (TBD...)\nfunc (v *FailoverDomainResponse) GetConfiguration() (o *DomainConfiguration) {\n\tif v != nil && v.Configuration != nil {\n\t\treturn v.Configuration\n\t}\n\treturn\n}\n\n// GetReplicationConfiguration is an internal getter (TBD...)\nfunc (v *FailoverDomainResponse) GetReplicationConfiguration() (o *DomainReplicationConfiguration) {\n\tif v != nil && v.ReplicationConfiguration != nil {\n\t\treturn v.ReplicationConfiguration\n\t}\n\treturn\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *FailoverDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetIsGlobalDomain is an internal getter (TBD...)\nfunc (v *FailoverDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil {\n\t\treturn v.IsGlobalDomain\n\t}\n\treturn\n}\n\n// ListFailoverHistoryRequest is an internal type (TBD...)\ntype ListFailoverHistoryRequest struct {\n\tFilters    *ListFailoverHistoryRequestFilters `json:\"filters,omitempty\"`\n\tPagination *PaginationOptions                 `json:\"pagination,omitempty\"`\n}\n\n// GetFilters is an internal getter (TBD...)\nfunc (v *ListFailoverHistoryRequest) GetFilters() (o *ListFailoverHistoryRequestFilters) {\n\tif v != nil && v.Filters != nil {\n\t\treturn v.Filters\n\t}\n\treturn\n}\n\n// GetPagination is an internal getter (TBD...)\nfunc (v *ListFailoverHistoryRequest) GetPagination() (o *PaginationOptions) {\n\tif v != nil && v.Pagination != nil {\n\t\treturn v.Pagination\n\t}\n\treturn\n}\n\n// ListFailoverHistoryRequestFilters is an internal type (TBD...)\ntype ListFailoverHistoryRequestFilters struct {\n\tDomainID string `json:\"domainId,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *ListFailoverHistoryRequestFilters) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// ListFailoverHistoryResponse is an internal type (TBD...)\ntype ListFailoverHistoryResponse struct {\n\tFailoverEvents []*FailoverEvent `json:\"failoverEvents,omitempty\"`\n\tNextPageToken  []byte           `json:\"nextPageToken,omitempty\"`\n}\n\n// GetFailoverEvents is an internal getter (TBD...)\nfunc (v *ListFailoverHistoryResponse) GetFailoverEvents() (o []*FailoverEvent) {\n\tif v != nil && v.FailoverEvents != nil {\n\t\treturn v.FailoverEvents\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *ListFailoverHistoryResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// PaginationOptions is an internal type (TBD...)\ntype PaginationOptions struct {\n\tPageSize      *int32 `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// GetPageSize is an internal getter (TBD...)\nfunc (v *PaginationOptions) GetPageSize() (o int32) {\n\tif v != nil && v.PageSize != nil {\n\t\treturn *v.PageSize\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *PaginationOptions) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// FailoverEvent is an internal type (TBD...)\ntype FailoverEvent struct {\n\tID               *string            `json:\"id,omitempty\"`\n\tCreatedTime      *int64             `json:\"createdTime,omitempty\"`\n\tFailoverType     *FailoverType      `json:\"failoverType,omitempty\"`\n\tClusterFailovers []*ClusterFailover `json:\"clusterFailovers,omitempty\"`\n}\n\n// GetID is an internal getter (TBD...)\nfunc (v *FailoverEvent) GetID() (o string) {\n\tif v != nil && v.ID != nil {\n\t\treturn *v.ID\n\t}\n\treturn\n}\n\n// GetCreatedTime is an internal getter (TBD...)\nfunc (v *FailoverEvent) GetCreatedTime() (o int64) {\n\tif v != nil && v.CreatedTime != nil {\n\t\treturn *v.CreatedTime\n\t}\n\treturn\n}\n\n// GetFailoverType is an internal getter (TBD...)\nfunc (v *FailoverEvent) GetFailoverType() (o FailoverType) {\n\tif v != nil && v.FailoverType != nil {\n\t\treturn *v.FailoverType\n\t}\n\treturn\n}\n\n// GetClusterFailovers is an internal getter (TBD...)\nfunc (v *FailoverEvent) GetClusterFailovers() (o []*ClusterFailover) {\n\tif v != nil && v.ClusterFailovers != nil {\n\t\treturn v.ClusterFailovers\n\t}\n\treturn\n}\n\n// FailoverType is an internal type (TBD...)\ntype FailoverType int32\n\n// this should line up with the IDL values in common.proto\n// ie FailoverType_FAILOVER_TYPE_FORCE = 1\n// ie FailoverType_FAILOVER_TYPE_GRACEFUL = 2\nconst (\n\tFailoverTypeInvalid  = FailoverType(0)\n\tFailoverTypeForce    = FailoverType(1)\n\tFailoverTypeGraceful = FailoverType(2)\n)\n\n// Ptr is a helper routine that returns a pointer to given FailoverType value.\nfunc (e FailoverType) Ptr() *FailoverType {\n\treturn &e\n}\n\n// ClusterFailover is an internal type (TBD...)\ntype ClusterFailover struct {\n\tFromCluster      *ActiveClusterInfo `json:\"fromCluster,omitempty\"`\n\tToCluster        *ActiveClusterInfo `json:\"toCluster,omitempty\"`\n\tClusterAttribute *ClusterAttribute  `json:\"clusterAttribute,omitempty\"`\n}\n\n// GetFromCluster is an internal getter (TBD...)\nfunc (v *ClusterFailover) GetFromCluster() (o *ActiveClusterInfo) {\n\tif v != nil && v.FromCluster != nil {\n\t\treturn v.FromCluster\n\t}\n\treturn\n}\n\n// GetToCluster is an internal getter (TBD...)\nfunc (v *ClusterFailover) GetToCluster() (o *ActiveClusterInfo) {\n\tif v != nil && v.ToCluster != nil {\n\t\treturn v.ToCluster\n\t}\n\treturn\n}\n\n// GetClusterAttribute is an internal getter (TBD...)\nfunc (v *ClusterFailover) GetClusterAttribute() (o *ClusterAttribute) {\n\tif v != nil && v.ClusterAttribute != nil {\n\t\treturn v.ClusterAttribute\n\t}\n\treturn\n}\n\n// DescribeHistoryHostRequest is an internal type (TBD...)\ntype DescribeHistoryHostRequest struct {\n\tHostAddress      *string            `json:\"hostAddress,omitempty\"`\n\tShardIDForHost   *int32             `json:\"shardIdForHost,omitempty\"`\n\tExecutionForHost *WorkflowExecution `json:\"executionForHost,omitempty\"`\n}\n\n// DescribeShardDistributionRequest is an internal type (TBD...)\ntype DescribeShardDistributionRequest struct {\n\tPageSize int32 `json:\"pageSize,omitempty\"`\n\tPageID   int32 `json:\"pageID,omitempty\"`\n}\n\n// GetHostAddress is an internal getter (TBD...)\nfunc (v *DescribeHistoryHostRequest) GetHostAddress() (o string) {\n\tif v != nil && v.HostAddress != nil {\n\t\treturn *v.HostAddress\n\t}\n\treturn\n}\n\n// GetShardIDForHost is an internal getter (TBD...)\nfunc (v *DescribeHistoryHostRequest) GetShardIDForHost() (o int32) {\n\tif v != nil && v.ShardIDForHost != nil {\n\t\treturn *v.ShardIDForHost\n\t}\n\treturn\n}\n\n// DescribeShardDistributionResponse is an internal type (TBD...)\ntype DescribeShardDistributionResponse struct {\n\tNumberOfShards int32            `json:\"numberOfShards,omitempty\"`\n\tShards         map[int32]string `json:\"shardIDs,omitempty\"`\n}\n\n// DescribeHistoryHostResponse is an internal type (TBD...)\ntype DescribeHistoryHostResponse struct {\n\tNumberOfShards        int32            `json:\"numberOfShards,omitempty\"`\n\tShardIDs              []int32          `json:\"shardIDs,omitempty\"`\n\tDomainCache           *DomainCacheInfo `json:\"domainCache,omitempty\"`\n\tShardControllerStatus string           `json:\"shardControllerStatus,omitempty\"`\n\tAddress               string           `json:\"address,omitempty\"`\n}\n\n// DescribeQueueRequest is an internal type (TBD...)\ntype DescribeQueueRequest struct {\n\tShardID     int32  `json:\"shardID,omitempty\"`\n\tClusterName string `json:\"clusterName,omitempty\"`\n\tType        *int32 `json:\"type,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *DescribeQueueRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *DescribeQueueRequest) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *DescribeQueueRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// DescribeQueueResponse is an internal type (TBD...)\ntype DescribeQueueResponse struct {\n\tProcessingQueueStates []string `json:\"processingQueueStates,omitempty\"`\n}\n\n// DescribeTaskListRequest is an internal type (TBD...)\ntype DescribeTaskListRequest struct {\n\tDomain                string        `json:\"domain,omitempty\"`\n\tTaskList              *TaskList     `json:\"taskList,omitempty\"`\n\tTaskListType          *TaskListType `json:\"taskListType,omitempty\"`\n\tIncludeTaskListStatus bool          `json:\"includeTaskListStatus,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *DescribeTaskListRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *DescribeTaskListRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetTaskListType is an internal getter (TBD...)\nfunc (v *DescribeTaskListRequest) GetTaskListType() (o TaskListType) {\n\tif v != nil && v.TaskListType != nil {\n\t\treturn *v.TaskListType\n\t}\n\treturn\n}\n\n// GetIncludeTaskListStatus is an internal getter (TBD...)\nfunc (v *DescribeTaskListRequest) GetIncludeTaskListStatus() (o bool) {\n\tif v != nil {\n\t\treturn v.IncludeTaskListStatus\n\t}\n\treturn\n}\n\n// DescribeTaskListResponse is an internal type (TBD...)\ntype DescribeTaskListResponse struct {\n\tPollers         []*PollerInfo            `json:\"pollers,omitempty\"`\n\tTaskListStatus  *TaskListStatus          `json:\"taskListStatus,omitempty\"`\n\tPartitionConfig *TaskListPartitionConfig `json:\"partitionConfig,omitempty\"`\n\tTaskList        *TaskList                `json:\"taskList,omitempty\"`\n}\n\n// GetPollers is an internal getter (TBD...)\nfunc (v *DescribeTaskListResponse) GetPollers() (o []*PollerInfo) {\n\tif v != nil && v.Pollers != nil {\n\t\treturn v.Pollers\n\t}\n\treturn\n}\n\n// GetTaskListStatus is an internal getter (TBD...)\nfunc (v *DescribeTaskListResponse) GetTaskListStatus() (o *TaskListStatus) {\n\tif v != nil && v.TaskListStatus != nil {\n\t\treturn v.TaskListStatus\n\t}\n\treturn\n}\n\n// DescribeWorkflowExecutionRequest is an internal type (TBD...)\ntype DescribeWorkflowExecutionRequest struct {\n\tDomain                string                 `json:\"domain,omitempty\"`\n\tExecution             *WorkflowExecution     `json:\"execution,omitempty\"`\n\tQueryConsistencyLevel *QueryConsistencyLevel `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *DescribeWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *DescribeWorkflowExecutionRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetQueryConsistencyLevel is an internal getter (TBD...)\nfunc (v *DescribeWorkflowExecutionRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\treturn\n}\n\n// DescribeWorkflowExecutionResponse is an internal type (TBD...)\ntype DescribeWorkflowExecutionResponse struct {\n\tExecutionConfiguration *WorkflowExecutionConfiguration `json:\"executionConfiguration,omitempty\"`\n\tWorkflowExecutionInfo  *WorkflowExecutionInfo          `json:\"workflowExecutionInfo,omitempty\"`\n\tPendingActivities      []*PendingActivityInfo          `json:\"pendingActivities,omitempty\"`\n\tPendingChildren        []*PendingChildExecutionInfo    `json:\"pendingChildren,omitempty\"`\n\tPendingDecision        *PendingDecisionInfo            `json:\"pendingDecision,omitempty\"`\n}\n\n// GetWorkflowExecutionInfo is an internal getter (TBD...)\nfunc (v *DescribeWorkflowExecutionResponse) GetWorkflowExecutionInfo() (o *WorkflowExecutionInfo) {\n\tif v != nil && v.WorkflowExecutionInfo != nil {\n\t\treturn v.WorkflowExecutionInfo\n\t}\n\treturn\n}\n\n// GetPendingActivities is an internal getter (TBD...)\nfunc (v *DescribeWorkflowExecutionResponse) GetPendingActivities() (o []*PendingActivityInfo) {\n\tif v != nil && v.PendingActivities != nil {\n\t\treturn v.PendingActivities\n\t}\n\treturn\n}\n\n// DomainAlreadyExistsError is an internal type (TBD...)\ntype DomainAlreadyExistsError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// DomainCacheInfo is an internal type (TBD...)\ntype DomainCacheInfo struct {\n\tNumOfItemsInCacheByID   int64 `json:\"numOfItemsInCacheByID,omitempty\"`\n\tNumOfItemsInCacheByName int64 `json:\"numOfItemsInCacheByName,omitempty\"`\n}\n\n// DomainConfiguration is an internal type (TBD...)\ntype DomainConfiguration struct {\n\tWorkflowExecutionRetentionPeriodInDays int32                        `json:\"workflowExecutionRetentionPeriodInDays,omitempty\"`\n\tEmitMetric                             bool                         `json:\"emitMetric,omitempty\"`\n\tBadBinaries                            *BadBinaries                 `json:\"badBinaries,omitempty\"`\n\tHistoryArchivalStatus                  *ArchivalStatus              `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                     string                       `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus               *ArchivalStatus              `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                  string                       `json:\"visibilityArchivalURI,omitempty\"`\n\tIsolationGroups                        *IsolationGroupConfiguration `json:\"isolationGroupConfiguration,omitempty\"`\n\tAsyncWorkflowConfig                    *AsyncWorkflowConfiguration  `json:\"asyncWorkflowConfiguration,omitempty\"`\n}\n\n// GetWorkflowExecutionRetentionPeriodInDays is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetWorkflowExecutionRetentionPeriodInDays() (o int32) {\n\tif v != nil {\n\t\treturn v.WorkflowExecutionRetentionPeriodInDays\n\t}\n\treturn\n}\n\n// GetEmitMetric is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetEmitMetric() (o bool) {\n\tif v != nil {\n\t\treturn v.EmitMetric\n\t}\n\treturn\n}\n\n// GetBadBinaries is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetBadBinaries() (o *BadBinaries) {\n\tif v != nil && v.BadBinaries != nil {\n\t\treturn v.BadBinaries\n\t}\n\treturn\n}\n\n// GetHistoryArchivalStatus is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetHistoryArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.HistoryArchivalStatus != nil {\n\t\treturn *v.HistoryArchivalStatus\n\t}\n\treturn\n}\n\n// GetHistoryArchivalURI is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetHistoryArchivalURI() (o string) {\n\tif v != nil {\n\t\treturn v.HistoryArchivalURI\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalStatus is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetVisibilityArchivalStatus() (o ArchivalStatus) {\n\tif v != nil && v.VisibilityArchivalStatus != nil {\n\t\treturn *v.VisibilityArchivalStatus\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalURI is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetVisibilityArchivalURI() (o string) {\n\tif v != nil {\n\t\treturn v.VisibilityArchivalURI\n\t}\n\treturn\n}\n\n// GetIsolationGroupsConfiguration is an internal getter (TBD...)\nfunc (v *DomainConfiguration) GetIsolationGroupsConfiguration() IsolationGroupConfiguration {\n\tif v.IsolationGroups != nil {\n\t\treturn *v.IsolationGroups\n\t}\n\treturn nil\n}\n\nfunc (v *DomainConfiguration) GetAsyncWorkflowConfiguration() AsyncWorkflowConfiguration {\n\tif v.AsyncWorkflowConfig != nil {\n\t\treturn *v.AsyncWorkflowConfig\n\t}\n\treturn AsyncWorkflowConfiguration{}\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *DomainConfiguration) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += v.BadBinaries.ByteSize()\n\tsize += v.HistoryArchivalStatus.ByteSize()\n\tsize += uint64(len(v.HistoryArchivalURI))\n\tsize += v.VisibilityArchivalStatus.ByteSize()\n\tsize += uint64(len(v.VisibilityArchivalURI))\n\tsize += v.IsolationGroups.ByteSize()\n\tsize += v.AsyncWorkflowConfig.ByteSize()\n\treturn size\n}\n\n// DomainInfo is an internal type (TBD...)\ntype DomainInfo struct {\n\tName        string            `json:\"name,omitempty\"`\n\tStatus      *DomainStatus     `json:\"status,omitempty\"`\n\tDescription string            `json:\"description,omitempty\"`\n\tOwnerEmail  string            `json:\"ownerEmail,omitempty\"`\n\tData        map[string]string `json:\"data,omitempty\"`\n\tUUID        string            `json:\"uuid,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *DomainInfo) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// GetStatus is an internal getter (TBD...)\nfunc (v *DomainInfo) GetStatus() (o DomainStatus) {\n\tif v != nil && v.Status != nil {\n\t\treturn *v.Status\n\t}\n\treturn\n}\n\n// GetDescription is an internal getter (TBD...)\nfunc (v *DomainInfo) GetDescription() (o string) {\n\tif v != nil {\n\t\treturn v.Description\n\t}\n\treturn\n}\n\n// GetOwnerEmail is an internal getter (TBD...)\nfunc (v *DomainInfo) GetOwnerEmail() (o string) {\n\tif v != nil {\n\t\treturn v.OwnerEmail\n\t}\n\treturn\n}\n\n// GetData is an internal getter (TBD...)\nfunc (v *DomainInfo) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\treturn\n}\n\n// GetUUID is an internal getter (TBD...)\nfunc (v *DomainInfo) GetUUID() (o string) {\n\tif v != nil {\n\t\treturn v.UUID\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *DomainInfo) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.Name))\n\tsize += v.Status.ByteSize()\n\tsize += uint64(len(v.Description))\n\tsize += uint64(len(v.OwnerEmail))\n\tfor k, val := range v.Data {\n\t\tsize += uint64(len(k))\n\t\tsize += uint64(len(val))\n\t}\n\tsize += uint64(len(v.UUID))\n\treturn size\n}\n\n// DomainNotActiveError is an internal type.\n// this is a retriable error and *must* be retried under at least\n// some circumstances due to domain failover races.\n// TODO(c-warren): Move to common/types/errors.go\ntype DomainNotActiveError struct {\n\tMessage        string   `json:\"message\"`\n\tDomainName     string   `json:\"domainName\"`\n\tCurrentCluster string   `json:\"currentCluster\"`\n\tActiveCluster  string   `json:\"activeCluster,omitempty\"`\n\tActiveClusters []string `json:\"activeClusters,omitempty\"`\n}\n\n// GetCurrentCluster is an internal getter (TBD...)\nfunc (v *DomainNotActiveError) GetCurrentCluster() (o string) {\n\tif v != nil {\n\t\treturn v.CurrentCluster\n\t}\n\treturn\n}\n\n// GetActiveCluster is an internal getter (TBD...)\nfunc (v *DomainNotActiveError) GetActiveCluster() (o string) {\n\tif v != nil {\n\t\treturn v.ActiveCluster\n\t}\n\treturn\n}\n\n// GetActiveClusters is an internal getter (TBD...)\nfunc (v *DomainNotActiveError) GetActiveClusters() (o []string) {\n\tif v != nil {\n\t\treturn v.ActiveClusters\n\t}\n\treturn\n}\n\n// DomainReplicationConfiguration is an internal type (TBD...)\ntype DomainReplicationConfiguration struct {\n\tActiveClusterName string                             `json:\"activeClusterName,omitempty\"`\n\tClusters          []*ClusterReplicationConfiguration `json:\"clusters,omitempty\"`\n\tActiveClusters    *ActiveClusters                    `json:\"activeClusters,omitempty\"`\n}\n\nfunc (v *DomainReplicationConfiguration) IsActiveActive() bool {\n\tif v == nil || v.ActiveClusters == nil {\n\t\treturn false\n\t}\n\n\t// Check to see if a ClusterAttribute has been configured for this domain.\n\tif len(v.ActiveClusters.AttributeScopes) > 0 {\n\t\tfor _, scope := range v.ActiveClusters.AttributeScopes {\n\t\t\tif len(scope.ClusterAttributes) > 0 {\n\t\t\t\treturn true\n\t\t\t}\n\t\t}\n\t}\n\n\treturn false\n}\n\n// GetActiveClusterName is an internal getter (TBD...)\nfunc (v *DomainReplicationConfiguration) GetActiveClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ActiveClusterName\n\t}\n\treturn\n}\n\n// GetClusters is an internal getter (TBD...)\nfunc (v *DomainReplicationConfiguration) GetClusters() (o []*ClusterReplicationConfiguration) {\n\tif v != nil && v.Clusters != nil {\n\t\treturn v.Clusters\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *DomainReplicationConfiguration) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.ActiveClusterName))\n\tsize += uint64(len(v.Clusters)) * uint64(unsafe.Sizeof((*ClusterReplicationConfiguration)(nil)))\n\tfor _, e := range v.Clusters {\n\t\tsize += e.ByteSize()\n\t}\n\tsize += v.ActiveClusters.ByteSize()\n\treturn size\n}\n\n// ActiveClusters defines the mapping of cluster attributes to their active cluster.\n// It contains a list of scopes, each of which contains a list of attribute to cluster mappings.\n// The following example uses 'cityID' as the attribute scope, and cities as the attributes names:\n// ClusterAttributes allow groups of workflows to be processed by different clusters within a domain.\n//\n//\tActiveClusters = {\n//\t  \"cityID\": { // any attribute scope, e.g a region, datacenter, city, etc.\n//\t    \"seattle\": {\n//\t        \"activeClusterName\": \"cluster1\",\n//\t        \"failoverVersion\": 100,\n//\t    },\n//\t    \"san_francisco\": {\n//\t        \"activeClusterName\": \"cluster2\",\n//\t        \"failoverVersion\": 200,\n//\t    },\n//\t  },\n//\t}\n//\n// TODO(c-warren): Rename to ClusterAttributes\ntype ActiveClusters struct {\n\t// ClusterAttributes\n\t// Keyed by a scope type (e.g region, datacenter, city, etc.).\n\t// The value is a ClusterAttributeScope - a map of unique names (e.g seattle, san_francisco, etc.) to an ActiveClusterInfo.\n\tAttributeScopes map[string]ClusterAttributeScope `json:\"attributeScopes,omitempty\" yaml:\"attributeScopes,omitempty\"`\n}\n\ntype ClusterAttributeNotFoundError struct {\n\tScopeType     string\n\tAttributeName string\n}\n\nfunc (e *ClusterAttributeNotFoundError) Error() string {\n\treturn fmt.Sprintf(\"cluster attribute %s not found in scope %s\", e.AttributeName, e.ScopeType)\n}\n\n// UndefinedFailoverVersion is used to indicate that a failover version that is not defined.\n// Failover versions are valid for Int64 >= 0\n// and, but implication, 0 is a valid failover version. To distinguish between it and an undefined\n// failover version, we use a negative value.\n// todo (david.porter) move this to constants package and give it a type\nconst UndefinedFailoverVersion = int64(-1)\n\n// GetFailoverVersionForAttribute returns the failover version for a given attribute.\n// if a value is not found it returns -1 and an error\nfunc (v *ActiveClusters) GetFailoverVersionForAttribute(scopeType, attributeName string) (int64, error) {\n\tif v == nil {\n\t\treturn UndefinedFailoverVersion, &ClusterAttributeNotFoundError{\n\t\t\tScopeType:     scopeType,\n\t\t\tAttributeName: attributeName,\n\t\t}\n\t}\n\tscope, ok := v.AttributeScopes[scopeType]\n\tif !ok {\n\t\treturn UndefinedFailoverVersion, &ClusterAttributeNotFoundError{\n\t\t\tScopeType:     scopeType,\n\t\t\tAttributeName: attributeName,\n\t\t}\n\t}\n\tinfo, ok := scope.ClusterAttributes[attributeName]\n\tif !ok {\n\t\treturn UndefinedFailoverVersion, &ClusterAttributeNotFoundError{\n\t\t\tScopeType:     scopeType,\n\t\t\tAttributeName: attributeName,\n\t\t}\n\t}\n\treturn info.FailoverVersion, nil\n}\n\nfunc (v *ActiveClusters) GetAttributeScopes() map[string]ClusterAttributeScope {\n\tif v != nil && v.AttributeScopes != nil {\n\t\treturn v.AttributeScopes\n\t}\n\treturn nil\n}\n\n// TODO(c-warren): Move to common/types/errors.go?\nvar (\n\tErrActiveClusterInfoNotFound = errors.New(\"active cluster info not found\")\n\tErrDomainNotActiveActive     = errors.New(\"domain is not configured for active-active\")\n)\n\n// GetActiveClusterByClusterAttribute retrieves the ActiveClusterInfo for a given cluster attribute.\n// An attribute is a scope, name pair (e.g. region, dca or city, tokyo).\n// Returns ActiveCluster and FailoverVersion if found, otherwise returns an error.\nfunc (v *ActiveClusters) GetActiveClusterByClusterAttribute(scopeType, attributeName string) (ActiveClusterInfo, error) {\n\tif v == nil {\n\t\treturn ActiveClusterInfo{}, ErrDomainNotActiveActive\n\t}\n\n\tif scopeType == \"\" || attributeName == \"\" {\n\t\treturn ActiveClusterInfo{}, fmt.Errorf(\"scopeType or attributeName is empty\")\n\t}\n\n\tscope, ok := v.AttributeScopes[scopeType]\n\tif !ok {\n\t\treturn ActiveClusterInfo{}, fmt.Errorf(\"scopeType not found %s: %w\", scopeType, ErrActiveClusterInfoNotFound)\n\t}\n\n\treturn scope.GetActiveClusterByClusterAttribute(attributeName)\n}\n\n// GetAllClusters returns a sorted, deduplicated list of all attribute names from both\n// For the \"region\" scope, these are region names; for other scopes, these are the attribute names.\nfunc (v *ActiveClusters) GetAllClusters() []string {\n\tif v == nil {\n\t\treturn []string{}\n\t}\n\n\t// Set of attribute names (e.g., region names for \"region\" scope)\n\tattributeNames := make(map[string]struct{})\n\n\t// Collect attribute names from new format\n\tif v.AttributeScopes != nil {\n\t\tfor _, scope := range v.AttributeScopes {\n\t\t\tfor attributeName := range scope.ClusterAttributes {\n\t\t\t\tattributeNames[attributeName] = struct{}{}\n\t\t\t}\n\t\t}\n\t}\n\n\t// Convert to sorted slice\n\tresult := make([]string, 0, len(attributeNames))\n\tfor name := range attributeNames {\n\t\tresult = append(result, name)\n\t}\n\n\t// Sort for deterministic output\n\tsort.Strings(result)\n\n\treturn result\n}\n\n// SetClusterForClusterAttribute sets the ActiveClusterInfo for a given cluster attribute.\n// If the receiver is nil, this method is a no-op.\nfunc (v *ActiveClusters) SetClusterForClusterAttribute(scopeType, attributeName string, info ActiveClusterInfo) error {\n\tif v == nil {\n\t\treturn ErrDomainNotActiveActive\n\t}\n\n\tif scopeType == \"\" || attributeName == \"\" {\n\t\treturn fmt.Errorf(\"scopeType or attributeName is empty\")\n\t}\n\n\tif v.AttributeScopes == nil {\n\t\tv.AttributeScopes = make(map[string]ClusterAttributeScope)\n\t}\n\n\tscope, ok := v.AttributeScopes[scopeType]\n\tif !ok {\n\t\tscope = ClusterAttributeScope{\n\t\t\tClusterAttributes: make(map[string]ActiveClusterInfo),\n\t\t}\n\t}\n\n\tscope.ClusterAttributes[attributeName] = info\n\tv.AttributeScopes[scopeType] = scope\n\n\treturn nil\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *ActiveClusters) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\n\tfor k, scope := range v.AttributeScopes {\n\t\tsize += uint64(len(k)) + scope.ByteSize() - uint64(unsafe.Sizeof(scope))\n\t}\n\n\treturn size\n}\n\n// ClusterAttributeScope is a map of unique attribute names to the active cluster for that attribute.\n// It can be used to determine the current failover version for a workflow associated with that attribute.\ntype ClusterAttributeScope struct {\n\tClusterAttributes map[string]ActiveClusterInfo `json:\"clusterAttributes,omitempty\" yaml:\"clusterAttributes,omitempty\"`\n}\n\nfunc (v *ClusterAttributeScope) GetActiveClusterByClusterAttribute(name string) (ActiveClusterInfo, error) {\n\tif v == nil {\n\t\treturn ActiveClusterInfo{}, ErrActiveClusterInfoNotFound\n\t}\n\n\tclusterInfo, ok := v.ClusterAttributes[name]\n\tif !ok {\n\t\treturn ActiveClusterInfo{}, fmt.Errorf(\"attribute not found %s: %w\", name, ErrActiveClusterInfoNotFound)\n\t}\n\treturn clusterInfo, nil\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *ClusterAttributeScope) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\tsize := uint64(unsafe.Sizeof(*v))\n\tfor k, val := range v.ClusterAttributes {\n\t\t// ByteSize implementation must match the logic in the reflection-based calculator used in the tests from common/types/test_util.go.\n\t\t// reflection-based calculator purposely ignores Go's internal map bucket/storage and treats each map element as\n\t\t// key: dynamic payload only (e.g., len(string)), no string header\n\t\t// value: dynamic payload only (e.g., for a struct, just its fields' dynamic payload), no struct header, no inline ints/bools\n\t\tsize += uint64(len(k)) + val.ByteSize() - uint64(unsafe.Sizeof(val))\n\t}\n\treturn size\n}\n\n// ActiveClusterInfo defines failover information for a ClusterAttribute.\ntype ActiveClusterInfo struct {\n\tActiveClusterName string `json:\"activeClusterName,omitempty\" yaml:\"activeClusterName,omitempty\"`\n\tFailoverVersion   int64  `json:\"failoverVersion\" yaml:\"failoverVersion\"`\n}\n\nfunc (v *ActiveClusterInfo) GetActiveClusterName() string {\n\tif v == nil {\n\t\treturn \"\"\n\t}\n\treturn v.ActiveClusterName\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v ActiveClusterInfo) ByteSize() uint64 {\n\treturn uint64(unsafe.Sizeof(v)) + uint64(len(v.ActiveClusterName))\n}\n\nfunc (v *ActiveClusters) DeepCopy() *ActiveClusters {\n\tif v == nil {\n\t\treturn nil\n\t}\n\tresult := &ActiveClusters{}\n\tif v.AttributeScopes != nil {\n\t\tresult.AttributeScopes = make(map[string]ClusterAttributeScope, len(v.AttributeScopes))\n\t\tfor scopeType, scope := range v.AttributeScopes {\n\t\t\tcopiedScope := ClusterAttributeScope{}\n\t\t\tif scope.ClusterAttributes != nil {\n\t\t\t\tcopiedScope.ClusterAttributes = make(map[string]ActiveClusterInfo, len(scope.ClusterAttributes))\n\t\t\t\tfor attrName, attrInfo := range scope.ClusterAttributes {\n\t\t\t\t\tcopiedScope.ClusterAttributes[attrName] = attrInfo\n\t\t\t\t}\n\t\t\t}\n\t\t\tresult.AttributeScopes[scopeType] = copiedScope\n\t\t}\n\t}\n\treturn result\n}\n\ntype ClusterAttribute struct {\n\tScope string `json:\"scope,omitempty\" yaml:\"scope,omitempty\"`\n\tName  string `json:\"name,omitempty\" yaml:\"name,omitempty\"`\n}\n\nfunc (c *ClusterAttribute) GetScope() string {\n\tif c == nil {\n\t\treturn \"\"\n\t}\n\treturn c.Scope\n}\n\nfunc (c *ClusterAttribute) GetName() string {\n\tif c == nil {\n\t\treturn \"\"\n\t}\n\treturn c.Name\n}\n\nfunc (c *ClusterAttribute) Equals(other *ClusterAttribute) bool {\n\tif c == nil && other == nil {\n\t\treturn true\n\t}\n\tif c == nil || other == nil {\n\t\treturn false\n\t}\n\treturn c.Scope == other.Scope && c.Name == other.Name\n}\n\ntype ActiveClusterSelectionStrategy int32\n\nconst (\n\tActiveClusterSelectionStrategyRegionSticky ActiveClusterSelectionStrategy = iota\n\tActiveClusterSelectionStrategyExternalEntity\n)\n\ntype ActiveClusterSelectionPolicy struct {\n\tActiveClusterSelectionStrategy *ActiveClusterSelectionStrategy `json:\"activeClusterSelectionStrategy,omitempty\"`\n\n\tStickyRegion string `json:\"stickyRegion,omitempty\"`\n\n\tExternalEntityType string `json:\"externalEntityType,omitempty\"`\n\tExternalEntityKey  string `json:\"externalEntityKey,omitempty\"`\n\n\t// TODO(active-active): Remove the fields above\n\tClusterAttribute *ClusterAttribute `json:\"clusterAttribute,omitempty\" yaml:\"clusterAttribute,omitempty\"`\n}\n\nfunc (p *ActiveClusterSelectionPolicy) GetClusterAttribute() *ClusterAttribute {\n\tif p == nil {\n\t\treturn nil\n\t}\n\treturn p.ClusterAttribute\n}\n\nfunc (p *ActiveClusterSelectionPolicy) Equals(other *ActiveClusterSelectionPolicy) bool {\n\tif p == nil && other == nil {\n\t\treturn true\n\t}\n\tif p == nil || other == nil {\n\t\treturn false\n\t}\n\n\treturn p.ClusterAttribute.Equals(other.ClusterAttribute)\n}\n\nfunc (e ActiveClusterSelectionStrategy) Ptr() *ActiveClusterSelectionStrategy {\n\treturn &e\n}\n\nfunc (e ActiveClusterSelectionStrategy) String() string {\n\tswitch e {\n\tcase ActiveClusterSelectionStrategyRegionSticky:\n\t\treturn \"REGION_STICKY\"\n\tcase ActiveClusterSelectionStrategyExternalEntity:\n\t\treturn \"EXTERNAL_ENTITY\"\n\t}\n\n\treturn fmt.Sprintf(\"ActiveClusterSelectionStrategy(%d)\", e)\n}\n\nfunc (e ActiveClusterSelectionStrategy) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nfunc (e *ActiveClusterSelectionStrategy) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"REGION_STICKY\":\n\t\t*e = ActiveClusterSelectionStrategyRegionSticky\n\t\treturn nil\n\tcase \"EXTERNAL_ENTITY\":\n\t\t*e = ActiveClusterSelectionStrategyExternalEntity\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ActiveClusterSelectionStrategy\", err)\n\t\t}\n\t\t*e = ActiveClusterSelectionStrategy(val)\n\t\treturn nil\n\t}\n}\n\n// DomainStatus is an internal type (TBD...)\ntype DomainStatus int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e DomainStatus) Ptr() *DomainStatus {\n\treturn &e\n}\n\n// String returns a readable string representation of DomainStatus.\nfunc (e DomainStatus) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"REGISTERED\"\n\tcase 1:\n\t\treturn \"DEPRECATED\"\n\tcase 2:\n\t\treturn \"DELETED\"\n\t}\n\treturn fmt.Sprintf(\"DomainStatus(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *DomainStatus) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"REGISTERED\":\n\t\t*e = DomainStatusRegistered\n\t\treturn nil\n\tcase \"DEPRECATED\":\n\t\t*e = DomainStatusDeprecated\n\t\treturn nil\n\tcase \"DELETED\":\n\t\t*e = DomainStatusDeleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"DomainStatus\", err)\n\t\t}\n\t\t*e = DomainStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes DomainStatus to text.\nfunc (e DomainStatus) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (e *DomainStatus) ByteSize() uint64 {\n\tif e == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(unsafe.Sizeof(*e))\n}\n\nconst (\n\t// DomainStatusRegistered is an option for DomainStatus\n\tDomainStatusRegistered DomainStatus = iota\n\t// DomainStatusDeprecated is an option for DomainStatus\n\tDomainStatusDeprecated\n\t// DomainStatusDeleted is an option for DomainStatus\n\tDomainStatusDeleted\n)\n\n// EncodingType is an internal type (TBD...)\ntype EncodingType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e EncodingType) Ptr() *EncodingType {\n\treturn &e\n}\n\n// String returns a readable string representation of EncodingType.\nfunc (e EncodingType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ThriftRW\"\n\tcase 1:\n\t\treturn \"JSON\"\n\t}\n\treturn fmt.Sprintf(\"EncodingType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *EncodingType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"THRIFTRW\":\n\t\t*e = EncodingTypeThriftRW\n\t\treturn nil\n\tcase \"JSON\":\n\t\t*e = EncodingTypeJSON\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"EncodingType\", err)\n\t\t}\n\t\t*e = EncodingType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes EncodingType to text.\nfunc (e EncodingType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// EncodingTypeThriftRW is an option for EncodingType\n\tEncodingTypeThriftRW EncodingType = iota\n\t// EncodingTypeJSON is an option for EncodingType\n\tEncodingTypeJSON\n)\n\n// ByteSize returns the approximate memory used in bytes\nfunc (e *EncodingType) ByteSize() uint64 {\n\tif e == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(unsafe.Sizeof(*e))\n}\n\n// EntityNotExistsError is an internal type (TBD...)\ntype EntityNotExistsError struct {\n\tMessage        string   `json:\"message,required\"`\n\tCurrentCluster string   `json:\"currentCluster,omitempty\"`\n\tActiveCluster  string   `json:\"activeCluster,omitempty\"`\n\tActiveClusters []string `json:\"activeClusters,omitempty\"`\n}\n\n// WorkflowExecutionAlreadyCompletedError is an internal type (TBD...)\ntype WorkflowExecutionAlreadyCompletedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// EventType is an internal type (TBD...)\ntype EventType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e EventType) Ptr() *EventType {\n\treturn &e\n}\n\n// String returns a readable string representation of EventType.\nfunc (e EventType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"WorkflowExecutionStarted\"\n\tcase 1:\n\t\treturn \"WorkflowExecutionCompleted\"\n\tcase 2:\n\t\treturn \"WorkflowExecutionFailed\"\n\tcase 3:\n\t\treturn \"WorkflowExecutionTimedOut\"\n\tcase 4:\n\t\treturn \"DecisionTaskScheduled\"\n\tcase 5:\n\t\treturn \"DecisionTaskStarted\"\n\tcase 6:\n\t\treturn \"DecisionTaskCompleted\"\n\tcase 7:\n\t\treturn \"DecisionTaskTimedOut\"\n\tcase 8:\n\t\treturn \"DecisionTaskFailed\"\n\tcase 9:\n\t\treturn \"ActivityTaskScheduled\"\n\tcase 10:\n\t\treturn \"ActivityTaskStarted\"\n\tcase 11:\n\t\treturn \"ActivityTaskCompleted\"\n\tcase 12:\n\t\treturn \"ActivityTaskFailed\"\n\tcase 13:\n\t\treturn \"ActivityTaskTimedOut\"\n\tcase 14:\n\t\treturn \"ActivityTaskCancelRequested\"\n\tcase 15:\n\t\treturn \"RequestCancelActivityTaskFailed\"\n\tcase 16:\n\t\treturn \"ActivityTaskCanceled\"\n\tcase 17:\n\t\treturn \"TimerStarted\"\n\tcase 18:\n\t\treturn \"TimerFired\"\n\tcase 19:\n\t\treturn \"CancelTimerFailed\"\n\tcase 20:\n\t\treturn \"TimerCanceled\"\n\tcase 21:\n\t\treturn \"WorkflowExecutionCancelRequested\"\n\tcase 22:\n\t\treturn \"WorkflowExecutionCanceled\"\n\tcase 23:\n\t\treturn \"RequestCancelExternalWorkflowExecutionInitiated\"\n\tcase 24:\n\t\treturn \"RequestCancelExternalWorkflowExecutionFailed\"\n\tcase 25:\n\t\treturn \"ExternalWorkflowExecutionCancelRequested\"\n\tcase 26:\n\t\treturn \"MarkerRecorded\"\n\tcase 27:\n\t\treturn \"WorkflowExecutionSignaled\"\n\tcase 28:\n\t\treturn \"WorkflowExecutionTerminated\"\n\tcase 29:\n\t\treturn \"WorkflowExecutionContinuedAsNew\"\n\tcase 30:\n\t\treturn \"StartChildWorkflowExecutionInitiated\"\n\tcase 31:\n\t\treturn \"StartChildWorkflowExecutionFailed\"\n\tcase 32:\n\t\treturn \"ChildWorkflowExecutionStarted\"\n\tcase 33:\n\t\treturn \"ChildWorkflowExecutionCompleted\"\n\tcase 34:\n\t\treturn \"ChildWorkflowExecutionFailed\"\n\tcase 35:\n\t\treturn \"ChildWorkflowExecutionCanceled\"\n\tcase 36:\n\t\treturn \"ChildWorkflowExecutionTimedOut\"\n\tcase 37:\n\t\treturn \"ChildWorkflowExecutionTerminated\"\n\tcase 38:\n\t\treturn \"SignalExternalWorkflowExecutionInitiated\"\n\tcase 39:\n\t\treturn \"SignalExternalWorkflowExecutionFailed\"\n\tcase 40:\n\t\treturn \"ExternalWorkflowExecutionSignaled\"\n\tcase 41:\n\t\treturn \"UpsertWorkflowSearchAttributes\"\n\t}\n\treturn fmt.Sprintf(\"EventType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *EventType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"WORKFLOWEXECUTIONSTARTED\":\n\t\t*e = EventTypeWorkflowExecutionStarted\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONCOMPLETED\":\n\t\t*e = EventTypeWorkflowExecutionCompleted\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONFAILED\":\n\t\t*e = EventTypeWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONTIMEDOUT\":\n\t\t*e = EventTypeWorkflowExecutionTimedOut\n\t\treturn nil\n\tcase \"DECISIONTASKSCHEDULED\":\n\t\t*e = EventTypeDecisionTaskScheduled\n\t\treturn nil\n\tcase \"DECISIONTASKSTARTED\":\n\t\t*e = EventTypeDecisionTaskStarted\n\t\treturn nil\n\tcase \"DECISIONTASKCOMPLETED\":\n\t\t*e = EventTypeDecisionTaskCompleted\n\t\treturn nil\n\tcase \"DECISIONTASKTIMEDOUT\":\n\t\t*e = EventTypeDecisionTaskTimedOut\n\t\treturn nil\n\tcase \"DECISIONTASKFAILED\":\n\t\t*e = EventTypeDecisionTaskFailed\n\t\treturn nil\n\tcase \"ACTIVITYTASKSCHEDULED\":\n\t\t*e = EventTypeActivityTaskScheduled\n\t\treturn nil\n\tcase \"ACTIVITYTASKSTARTED\":\n\t\t*e = EventTypeActivityTaskStarted\n\t\treturn nil\n\tcase \"ACTIVITYTASKCOMPLETED\":\n\t\t*e = EventTypeActivityTaskCompleted\n\t\treturn nil\n\tcase \"ACTIVITYTASKFAILED\":\n\t\t*e = EventTypeActivityTaskFailed\n\t\treturn nil\n\tcase \"ACTIVITYTASKTIMEDOUT\":\n\t\t*e = EventTypeActivityTaskTimedOut\n\t\treturn nil\n\tcase \"ACTIVITYTASKCANCELREQUESTED\":\n\t\t*e = EventTypeActivityTaskCancelRequested\n\t\treturn nil\n\tcase \"REQUESTCANCELACTIVITYTASKFAILED\":\n\t\t*e = EventTypeRequestCancelActivityTaskFailed\n\t\treturn nil\n\tcase \"ACTIVITYTASKCANCELED\":\n\t\t*e = EventTypeActivityTaskCanceled\n\t\treturn nil\n\tcase \"TIMERSTARTED\":\n\t\t*e = EventTypeTimerStarted\n\t\treturn nil\n\tcase \"TIMERFIRED\":\n\t\t*e = EventTypeTimerFired\n\t\treturn nil\n\tcase \"CANCELTIMERFAILED\":\n\t\t*e = EventTypeCancelTimerFailed\n\t\treturn nil\n\tcase \"TIMERCANCELED\":\n\t\t*e = EventTypeTimerCanceled\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONCANCELREQUESTED\":\n\t\t*e = EventTypeWorkflowExecutionCancelRequested\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONCANCELED\":\n\t\t*e = EventTypeWorkflowExecutionCanceled\n\t\treturn nil\n\tcase \"REQUESTCANCELEXTERNALWORKFLOWEXECUTIONINITIATED\":\n\t\t*e = EventTypeRequestCancelExternalWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"REQUESTCANCELEXTERNALWORKFLOWEXECUTIONFAILED\":\n\t\t*e = EventTypeRequestCancelExternalWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"EXTERNALWORKFLOWEXECUTIONCANCELREQUESTED\":\n\t\t*e = EventTypeExternalWorkflowExecutionCancelRequested\n\t\treturn nil\n\tcase \"MARKERRECORDED\":\n\t\t*e = EventTypeMarkerRecorded\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONSIGNALED\":\n\t\t*e = EventTypeWorkflowExecutionSignaled\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONTERMINATED\":\n\t\t*e = EventTypeWorkflowExecutionTerminated\n\t\treturn nil\n\tcase \"WORKFLOWEXECUTIONCONTINUEDASNEW\":\n\t\t*e = EventTypeWorkflowExecutionContinuedAsNew\n\t\treturn nil\n\tcase \"STARTCHILDWORKFLOWEXECUTIONINITIATED\":\n\t\t*e = EventTypeStartChildWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"STARTCHILDWORKFLOWEXECUTIONFAILED\":\n\t\t*e = EventTypeStartChildWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONSTARTED\":\n\t\t*e = EventTypeChildWorkflowExecutionStarted\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONCOMPLETED\":\n\t\t*e = EventTypeChildWorkflowExecutionCompleted\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONFAILED\":\n\t\t*e = EventTypeChildWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONCANCELED\":\n\t\t*e = EventTypeChildWorkflowExecutionCanceled\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONTIMEDOUT\":\n\t\t*e = EventTypeChildWorkflowExecutionTimedOut\n\t\treturn nil\n\tcase \"CHILDWORKFLOWEXECUTIONTERMINATED\":\n\t\t*e = EventTypeChildWorkflowExecutionTerminated\n\t\treturn nil\n\tcase \"SIGNALEXTERNALWORKFLOWEXECUTIONINITIATED\":\n\t\t*e = EventTypeSignalExternalWorkflowExecutionInitiated\n\t\treturn nil\n\tcase \"SIGNALEXTERNALWORKFLOWEXECUTIONFAILED\":\n\t\t*e = EventTypeSignalExternalWorkflowExecutionFailed\n\t\treturn nil\n\tcase \"EXTERNALWORKFLOWEXECUTIONSIGNALED\":\n\t\t*e = EventTypeExternalWorkflowExecutionSignaled\n\t\treturn nil\n\tcase \"UPSERTWORKFLOWSEARCHATTRIBUTES\":\n\t\t*e = EventTypeUpsertWorkflowSearchAttributes\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"EventType\", err)\n\t\t}\n\t\t*e = EventType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes EventType to text.\nfunc (e EventType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// EventTypeWorkflowExecutionStarted is an option for EventType\n\tEventTypeWorkflowExecutionStarted EventType = iota\n\t// EventTypeWorkflowExecutionCompleted is an option for EventType\n\tEventTypeWorkflowExecutionCompleted\n\t// EventTypeWorkflowExecutionFailed is an option for EventType\n\tEventTypeWorkflowExecutionFailed\n\t// EventTypeWorkflowExecutionTimedOut is an option for EventType\n\tEventTypeWorkflowExecutionTimedOut\n\t// EventTypeDecisionTaskScheduled is an option for EventType\n\tEventTypeDecisionTaskScheduled\n\t// EventTypeDecisionTaskStarted is an option for EventType\n\tEventTypeDecisionTaskStarted\n\t// EventTypeDecisionTaskCompleted is an option for EventType\n\tEventTypeDecisionTaskCompleted\n\t// EventTypeDecisionTaskTimedOut is an option for EventType\n\tEventTypeDecisionTaskTimedOut\n\t// EventTypeDecisionTaskFailed is an option for EventType\n\tEventTypeDecisionTaskFailed\n\t// EventTypeActivityTaskScheduled is an option for EventType\n\tEventTypeActivityTaskScheduled\n\t// EventTypeActivityTaskStarted is an option for EventType\n\tEventTypeActivityTaskStarted\n\t// EventTypeActivityTaskCompleted is an option for EventType\n\tEventTypeActivityTaskCompleted\n\t// EventTypeActivityTaskFailed is an option for EventType\n\tEventTypeActivityTaskFailed\n\t// EventTypeActivityTaskTimedOut is an option for EventType\n\tEventTypeActivityTaskTimedOut\n\t// EventTypeActivityTaskCancelRequested is an option for EventType\n\tEventTypeActivityTaskCancelRequested\n\t// EventTypeRequestCancelActivityTaskFailed is an option for EventType\n\tEventTypeRequestCancelActivityTaskFailed\n\t// EventTypeActivityTaskCanceled is an option for EventType\n\tEventTypeActivityTaskCanceled\n\t// EventTypeTimerStarted is an option for EventType\n\tEventTypeTimerStarted\n\t// EventTypeTimerFired is an option for EventType\n\tEventTypeTimerFired\n\t// EventTypeCancelTimerFailed is an option for EventType\n\tEventTypeCancelTimerFailed\n\t// EventTypeTimerCanceled is an option for EventType\n\tEventTypeTimerCanceled\n\t// EventTypeWorkflowExecutionCancelRequested is an option for EventType\n\tEventTypeWorkflowExecutionCancelRequested\n\t// EventTypeWorkflowExecutionCanceled is an option for EventType\n\tEventTypeWorkflowExecutionCanceled\n\t// EventTypeRequestCancelExternalWorkflowExecutionInitiated is an option for EventType\n\tEventTypeRequestCancelExternalWorkflowExecutionInitiated\n\t// EventTypeRequestCancelExternalWorkflowExecutionFailed is an option for EventType\n\tEventTypeRequestCancelExternalWorkflowExecutionFailed\n\t// EventTypeExternalWorkflowExecutionCancelRequested is an option for EventType\n\tEventTypeExternalWorkflowExecutionCancelRequested\n\t// EventTypeMarkerRecorded is an option for EventType\n\tEventTypeMarkerRecorded\n\t// EventTypeWorkflowExecutionSignaled is an option for EventType\n\tEventTypeWorkflowExecutionSignaled\n\t// EventTypeWorkflowExecutionTerminated is an option for EventType\n\tEventTypeWorkflowExecutionTerminated\n\t// EventTypeWorkflowExecutionContinuedAsNew is an option for EventType\n\tEventTypeWorkflowExecutionContinuedAsNew\n\t// EventTypeStartChildWorkflowExecutionInitiated is an option for EventType\n\tEventTypeStartChildWorkflowExecutionInitiated\n\t// EventTypeStartChildWorkflowExecutionFailed is an option for EventType\n\tEventTypeStartChildWorkflowExecutionFailed\n\t// EventTypeChildWorkflowExecutionStarted is an option for EventType\n\tEventTypeChildWorkflowExecutionStarted\n\t// EventTypeChildWorkflowExecutionCompleted is an option for EventType\n\tEventTypeChildWorkflowExecutionCompleted\n\t// EventTypeChildWorkflowExecutionFailed is an option for EventType\n\tEventTypeChildWorkflowExecutionFailed\n\t// EventTypeChildWorkflowExecutionCanceled is an option for EventType\n\tEventTypeChildWorkflowExecutionCanceled\n\t// EventTypeChildWorkflowExecutionTimedOut is an option for EventType\n\tEventTypeChildWorkflowExecutionTimedOut\n\t// EventTypeChildWorkflowExecutionTerminated is an option for EventType\n\tEventTypeChildWorkflowExecutionTerminated\n\t// EventTypeSignalExternalWorkflowExecutionInitiated is an option for EventType\n\tEventTypeSignalExternalWorkflowExecutionInitiated\n\t// EventTypeSignalExternalWorkflowExecutionFailed is an option for EventType\n\tEventTypeSignalExternalWorkflowExecutionFailed\n\t// EventTypeExternalWorkflowExecutionSignaled is an option for EventType\n\tEventTypeExternalWorkflowExecutionSignaled\n\t// EventTypeUpsertWorkflowSearchAttributes is an option for EventType\n\tEventTypeUpsertWorkflowSearchAttributes\n)\n\n// ExternalWorkflowExecutionCancelRequestedEventAttributes is an internal type (TBD...)\ntype ExternalWorkflowExecutionCancelRequestedEventAttributes struct {\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ExternalWorkflowExecutionCancelRequestedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// ExternalWorkflowExecutionSignaledEventAttributes is an internal type (TBD...)\ntype ExternalWorkflowExecutionSignaledEventAttributes struct {\n\tInitiatedEventID  int64              `json:\"initiatedEventId,omitempty\"`\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ExternalWorkflowExecutionSignaledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// FailWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype FailWorkflowExecutionDecisionAttributes struct {\n\tReason  *string `json:\"reason,omitempty\"`\n\tDetails []byte  `json:\"details,omitempty\"`\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *FailWorkflowExecutionDecisionAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\treturn\n}\n\n// GetSearchAttributesResponse is an internal type (TBD...)\ntype GetSearchAttributesResponse struct {\n\tKeys map[string]IndexedValueType `json:\"keys,omitempty\"`\n}\n\n// GetKeys is an internal getter (TBD...)\nfunc (v *GetSearchAttributesResponse) GetKeys() (o map[string]IndexedValueType) {\n\tif v != nil && v.Keys != nil {\n\t\treturn v.Keys\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionHistoryRequest is an internal type (TBD...)\ntype GetWorkflowExecutionHistoryRequest struct {\n\tDomain                 string                  `json:\"domain,omitempty\"`\n\tExecution              *WorkflowExecution      `json:\"execution,omitempty\"`\n\tMaximumPageSize        int32                   `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken          []byte                  `json:\"nextPageToken,omitempty\"`\n\tWaitForNewEvent        bool                    `json:\"waitForNewEvent,omitempty\"`\n\tHistoryEventFilterType *HistoryEventFilterType `json:\"HistoryEventFilterType,omitempty\"`\n\tSkipArchival           bool                    `json:\"skipArchival,omitempty\"`\n\tQueryConsistencyLevel  *QueryConsistencyLevel  `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// GetWaitForNewEvent is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetWaitForNewEvent() (o bool) {\n\tif v != nil {\n\t\treturn v.WaitForNewEvent\n\t}\n\treturn\n}\n\n// GetHistoryEventFilterType is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetHistoryEventFilterType() (o HistoryEventFilterType) {\n\tif v != nil && v.HistoryEventFilterType != nil {\n\t\treturn *v.HistoryEventFilterType\n\t}\n\treturn\n}\n\n// GetSkipArchival is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetSkipArchival() (o bool) {\n\tif v != nil {\n\t\treturn v.SkipArchival\n\t}\n\treturn\n}\n\n// GetQueryConsistencyLevel is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionHistoryResponse is an internal type (TBD...)\ntype GetWorkflowExecutionHistoryResponse struct {\n\tHistory       *History    `json:\"history,omitempty\"`\n\tRawHistory    []*DataBlob `json:\"rawHistory,omitempty\"`\n\tNextPageToken []byte      `json:\"nextPageToken,omitempty\"`\n\tArchived      bool        `json:\"archived,omitempty\"`\n}\n\n// GetHistory is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryResponse) GetHistory() (o *History) {\n\tif v != nil && v.History != nil {\n\t\treturn v.History\n\t}\n\treturn\n}\n\n// GetArchived is an internal getter (TBD...)\nfunc (v *GetWorkflowExecutionHistoryResponse) GetArchived() (o bool) {\n\tif v != nil {\n\t\treturn v.Archived\n\t}\n\treturn\n}\n\n// FailoverInfo is an internal type (TBD...)\ntype FailoverInfo struct {\n\tFailoverVersion         int64   `json:\"failoverVersion,omitempty\"`\n\tFailoverStartTimestamp  int64   `json:\"failoverStartTimestamp,omitempty\"`\n\tFailoverExpireTimestamp int64   `json:\"failoverExpireTimestamp,omitempty\"`\n\tCompletedShardCount     int32   `json:\"completedShardCount,omitempty\"`\n\tPendingShards           []int32 `json:\"pendingShards,omitempty\"`\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *FailoverInfo) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetFailoverStartTimestamp is an internal getter (TBD...)\nfunc (v *FailoverInfo) GetFailoverStartTimestamp() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverStartTimestamp\n\t}\n\treturn\n}\n\n// GetFailoverExpireTimestamp is an internal getter (TBD...)\nfunc (v *FailoverInfo) GetFailoverExpireTimestamp() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverExpireTimestamp\n\t}\n\treturn\n}\n\n// GetCompletedShardCount is an internal getter (TBD...)\nfunc (v *FailoverInfo) GetCompletedShardCount() (o int32) {\n\tif v != nil {\n\t\treturn v.CompletedShardCount\n\t}\n\treturn\n}\n\n// GetPendingShards is an internal getter (TBD...)\nfunc (v *FailoverInfo) GetPendingShards() (o []int32) {\n\tif v != nil {\n\t\treturn v.PendingShards\n\t}\n\treturn\n}\n\n// Header is an internal type (TBD...)\ntype Header struct {\n\tFields map[string][]byte `json:\"fields,omitempty\"`\n}\n\n// History is an internal type (TBD...)\ntype History struct {\n\tEvents []*HistoryEvent `json:\"events,omitempty\"`\n}\n\n// GetEvents is an internal getter (TBD...)\nfunc (v *History) GetEvents() (o []*HistoryEvent) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\treturn\n}\n\n// HistoryBranch is an internal type (TBD...)\ntype HistoryBranch struct {\n\tTreeID    string\n\tBranchID  string\n\tAncestors []*HistoryBranchRange\n}\n\n// HistoryBranchRange is an internal type (TBD...)\ntype HistoryBranchRange struct {\n\tBranchID    string\n\tBeginNodeID int64\n\tEndNodeID   int64\n}\n\n// HistoryEvent is an internal type (TBD...)\ntype HistoryEvent struct {\n\tID                                                             int64                                                           `json:\"eventId,omitempty\"`\n\tTimestamp                                                      *int64                                                          `json:\"timestamp,omitempty\"`\n\tEventType                                                      *EventType                                                      `json:\"eventType,omitempty\"`\n\tVersion                                                        int64                                                           `json:\"version,omitempty\"`\n\tTaskID                                                         int64                                                           `json:\"taskId,omitempty\"`\n\tWorkflowExecutionStartedEventAttributes                        *WorkflowExecutionStartedEventAttributes                        `json:\"workflowExecutionStartedEventAttributes,omitempty\"`\n\tWorkflowExecutionCompletedEventAttributes                      *WorkflowExecutionCompletedEventAttributes                      `json:\"workflowExecutionCompletedEventAttributes,omitempty\"`\n\tWorkflowExecutionFailedEventAttributes                         *WorkflowExecutionFailedEventAttributes                         `json:\"workflowExecutionFailedEventAttributes,omitempty\"`\n\tWorkflowExecutionTimedOutEventAttributes                       *WorkflowExecutionTimedOutEventAttributes                       `json:\"workflowExecutionTimedOutEventAttributes,omitempty\"`\n\tDecisionTaskScheduledEventAttributes                           *DecisionTaskScheduledEventAttributes                           `json:\"decisionTaskScheduledEventAttributes,omitempty\"`\n\tDecisionTaskStartedEventAttributes                             *DecisionTaskStartedEventAttributes                             `json:\"decisionTaskStartedEventAttributes,omitempty\"`\n\tDecisionTaskCompletedEventAttributes                           *DecisionTaskCompletedEventAttributes                           `json:\"decisionTaskCompletedEventAttributes,omitempty\"`\n\tDecisionTaskTimedOutEventAttributes                            *DecisionTaskTimedOutEventAttributes                            `json:\"decisionTaskTimedOutEventAttributes,omitempty\"`\n\tDecisionTaskFailedEventAttributes                              *DecisionTaskFailedEventAttributes                              `json:\"decisionTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskScheduledEventAttributes                           *ActivityTaskScheduledEventAttributes                           `json:\"activityTaskScheduledEventAttributes,omitempty\"`\n\tActivityTaskStartedEventAttributes                             *ActivityTaskStartedEventAttributes                             `json:\"activityTaskStartedEventAttributes,omitempty\"`\n\tActivityTaskCompletedEventAttributes                           *ActivityTaskCompletedEventAttributes                           `json:\"activityTaskCompletedEventAttributes,omitempty\"`\n\tActivityTaskFailedEventAttributes                              *ActivityTaskFailedEventAttributes                              `json:\"activityTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskTimedOutEventAttributes                            *ActivityTaskTimedOutEventAttributes                            `json:\"activityTaskTimedOutEventAttributes,omitempty\"`\n\tTimerStartedEventAttributes                                    *TimerStartedEventAttributes                                    `json:\"timerStartedEventAttributes,omitempty\"`\n\tTimerFiredEventAttributes                                      *TimerFiredEventAttributes                                      `json:\"timerFiredEventAttributes,omitempty\"`\n\tActivityTaskCancelRequestedEventAttributes                     *ActivityTaskCancelRequestedEventAttributes                     `json:\"activityTaskCancelRequestedEventAttributes,omitempty\"`\n\tRequestCancelActivityTaskFailedEventAttributes                 *RequestCancelActivityTaskFailedEventAttributes                 `json:\"requestCancelActivityTaskFailedEventAttributes,omitempty\"`\n\tActivityTaskCanceledEventAttributes                            *ActivityTaskCanceledEventAttributes                            `json:\"activityTaskCanceledEventAttributes,omitempty\"`\n\tTimerCanceledEventAttributes                                   *TimerCanceledEventAttributes                                   `json:\"timerCanceledEventAttributes,omitempty\"`\n\tCancelTimerFailedEventAttributes                               *CancelTimerFailedEventAttributes                               `json:\"cancelTimerFailedEventAttributes,omitempty\"`\n\tMarkerRecordedEventAttributes                                  *MarkerRecordedEventAttributes                                  `json:\"markerRecordedEventAttributes,omitempty\"`\n\tWorkflowExecutionSignaledEventAttributes                       *WorkflowExecutionSignaledEventAttributes                       `json:\"workflowExecutionSignaledEventAttributes,omitempty\"`\n\tWorkflowExecutionTerminatedEventAttributes                     *WorkflowExecutionTerminatedEventAttributes                     `json:\"workflowExecutionTerminatedEventAttributes,omitempty\"`\n\tWorkflowExecutionCancelRequestedEventAttributes                *WorkflowExecutionCancelRequestedEventAttributes                `json:\"workflowExecutionCancelRequestedEventAttributes,omitempty\"`\n\tWorkflowExecutionCanceledEventAttributes                       *WorkflowExecutionCanceledEventAttributes                       `json:\"workflowExecutionCanceledEventAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes `json:\"requestCancelExternalWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tRequestCancelExternalWorkflowExecutionFailedEventAttributes    *RequestCancelExternalWorkflowExecutionFailedEventAttributes    `json:\"requestCancelExternalWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tExternalWorkflowExecutionCancelRequestedEventAttributes        *ExternalWorkflowExecutionCancelRequestedEventAttributes        `json:\"externalWorkflowExecutionCancelRequestedEventAttributes,omitempty\"`\n\tWorkflowExecutionContinuedAsNewEventAttributes                 *WorkflowExecutionContinuedAsNewEventAttributes                 `json:\"workflowExecutionContinuedAsNewEventAttributes,omitempty\"`\n\tStartChildWorkflowExecutionInitiatedEventAttributes            *StartChildWorkflowExecutionInitiatedEventAttributes            `json:\"startChildWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tStartChildWorkflowExecutionFailedEventAttributes               *StartChildWorkflowExecutionFailedEventAttributes               `json:\"startChildWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionStartedEventAttributes                   *ChildWorkflowExecutionStartedEventAttributes                   `json:\"childWorkflowExecutionStartedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionCompletedEventAttributes                 *ChildWorkflowExecutionCompletedEventAttributes                 `json:\"childWorkflowExecutionCompletedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionFailedEventAttributes                    *ChildWorkflowExecutionFailedEventAttributes                    `json:\"childWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tChildWorkflowExecutionCanceledEventAttributes                  *ChildWorkflowExecutionCanceledEventAttributes                  `json:\"childWorkflowExecutionCanceledEventAttributes,omitempty\"`\n\tChildWorkflowExecutionTimedOutEventAttributes                  *ChildWorkflowExecutionTimedOutEventAttributes                  `json:\"childWorkflowExecutionTimedOutEventAttributes,omitempty\"`\n\tChildWorkflowExecutionTerminatedEventAttributes                *ChildWorkflowExecutionTerminatedEventAttributes                `json:\"childWorkflowExecutionTerminatedEventAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionInitiatedEventAttributes        *SignalExternalWorkflowExecutionInitiatedEventAttributes        `json:\"signalExternalWorkflowExecutionInitiatedEventAttributes,omitempty\"`\n\tSignalExternalWorkflowExecutionFailedEventAttributes           *SignalExternalWorkflowExecutionFailedEventAttributes           `json:\"signalExternalWorkflowExecutionFailedEventAttributes,omitempty\"`\n\tExternalWorkflowExecutionSignaledEventAttributes               *ExternalWorkflowExecutionSignaledEventAttributes               `json:\"externalWorkflowExecutionSignaledEventAttributes,omitempty\"`\n\tUpsertWorkflowSearchAttributesEventAttributes                  *UpsertWorkflowSearchAttributesEventAttributes                  `json:\"upsertWorkflowSearchAttributesEventAttributes,omitempty\"`\n}\n\n// GetTimestamp is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetTimestamp() (o int64) {\n\tif v != nil && v.Timestamp != nil {\n\t\treturn *v.Timestamp\n\t}\n\treturn\n}\n\n// GetEventType is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetEventType() (o EventType) {\n\tif v != nil && v.EventType != nil {\n\t\treturn *v.EventType\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionStartedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionStartedEventAttributes() (o *WorkflowExecutionStartedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionStartedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionStartedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionCompletedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionCompletedEventAttributes() (o *WorkflowExecutionCompletedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCompletedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionFailedEventAttributes() (o *WorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionTimedOutEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionTimedOutEventAttributes() (o *WorkflowExecutionTimedOutEventAttributes) {\n\tif v != nil && v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\treturn v.WorkflowExecutionTimedOutEventAttributes\n\t}\n\treturn\n}\n\n// GetDecisionTaskScheduledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetDecisionTaskScheduledEventAttributes() (o *DecisionTaskScheduledEventAttributes) {\n\tif v != nil && v.DecisionTaskScheduledEventAttributes != nil {\n\t\treturn v.DecisionTaskScheduledEventAttributes\n\t}\n\treturn\n}\n\n// GetDecisionTaskStartedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetDecisionTaskStartedEventAttributes() (o *DecisionTaskStartedEventAttributes) {\n\tif v != nil && v.DecisionTaskStartedEventAttributes != nil {\n\t\treturn v.DecisionTaskStartedEventAttributes\n\t}\n\treturn\n}\n\n// GetDecisionTaskCompletedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetDecisionTaskCompletedEventAttributes() (o *DecisionTaskCompletedEventAttributes) {\n\tif v != nil && v.DecisionTaskCompletedEventAttributes != nil {\n\t\treturn v.DecisionTaskCompletedEventAttributes\n\t}\n\treturn\n}\n\n// GetDecisionTaskTimedOutEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetDecisionTaskTimedOutEventAttributes() (o *DecisionTaskTimedOutEventAttributes) {\n\tif v != nil && v.DecisionTaskTimedOutEventAttributes != nil {\n\t\treturn v.DecisionTaskTimedOutEventAttributes\n\t}\n\treturn\n}\n\n// GetDecisionTaskFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetDecisionTaskFailedEventAttributes() (o *DecisionTaskFailedEventAttributes) {\n\tif v != nil && v.DecisionTaskFailedEventAttributes != nil {\n\t\treturn v.DecisionTaskFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskScheduledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskScheduledEventAttributes() (o *ActivityTaskScheduledEventAttributes) {\n\tif v != nil && v.ActivityTaskScheduledEventAttributes != nil {\n\t\treturn v.ActivityTaskScheduledEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskStartedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskStartedEventAttributes() (o *ActivityTaskStartedEventAttributes) {\n\tif v != nil && v.ActivityTaskStartedEventAttributes != nil {\n\t\treturn v.ActivityTaskStartedEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskCompletedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskCompletedEventAttributes() (o *ActivityTaskCompletedEventAttributes) {\n\tif v != nil && v.ActivityTaskCompletedEventAttributes != nil {\n\t\treturn v.ActivityTaskCompletedEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskFailedEventAttributes() (o *ActivityTaskFailedEventAttributes) {\n\tif v != nil && v.ActivityTaskFailedEventAttributes != nil {\n\t\treturn v.ActivityTaskFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskTimedOutEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskTimedOutEventAttributes() (o *ActivityTaskTimedOutEventAttributes) {\n\tif v != nil && v.ActivityTaskTimedOutEventAttributes != nil {\n\t\treturn v.ActivityTaskTimedOutEventAttributes\n\t}\n\treturn\n}\n\n// GetTimerStartedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetTimerStartedEventAttributes() (o *TimerStartedEventAttributes) {\n\tif v != nil && v.TimerStartedEventAttributes != nil {\n\t\treturn v.TimerStartedEventAttributes\n\t}\n\treturn\n}\n\n// GetTimerFiredEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetTimerFiredEventAttributes() (o *TimerFiredEventAttributes) {\n\tif v != nil && v.TimerFiredEventAttributes != nil {\n\t\treturn v.TimerFiredEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskCancelRequestedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskCancelRequestedEventAttributes() (o *ActivityTaskCancelRequestedEventAttributes) {\n\tif v != nil && v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\treturn v.ActivityTaskCancelRequestedEventAttributes\n\t}\n\treturn\n}\n\n// GetRequestCancelActivityTaskFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetRequestCancelActivityTaskFailedEventAttributes() (o *RequestCancelActivityTaskFailedEventAttributes) {\n\tif v != nil && v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\treturn v.RequestCancelActivityTaskFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetActivityTaskCanceledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetActivityTaskCanceledEventAttributes() (o *ActivityTaskCanceledEventAttributes) {\n\tif v != nil && v.ActivityTaskCanceledEventAttributes != nil {\n\t\treturn v.ActivityTaskCanceledEventAttributes\n\t}\n\treturn\n}\n\n// GetTimerCanceledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetTimerCanceledEventAttributes() (o *TimerCanceledEventAttributes) {\n\tif v != nil && v.TimerCanceledEventAttributes != nil {\n\t\treturn v.TimerCanceledEventAttributes\n\t}\n\treturn\n}\n\n// GetCancelTimerFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetCancelTimerFailedEventAttributes() (o *CancelTimerFailedEventAttributes) {\n\tif v != nil && v.CancelTimerFailedEventAttributes != nil {\n\t\treturn v.CancelTimerFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetMarkerRecordedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetMarkerRecordedEventAttributes() (o *MarkerRecordedEventAttributes) {\n\tif v != nil && v.MarkerRecordedEventAttributes != nil {\n\t\treturn v.MarkerRecordedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionSignaledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionSignaledEventAttributes() (o *WorkflowExecutionSignaledEventAttributes) {\n\tif v != nil && v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\treturn v.WorkflowExecutionSignaledEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionTerminatedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionTerminatedEventAttributes() (o *WorkflowExecutionTerminatedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionTerminatedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionCancelRequestedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionCancelRequestedEventAttributes() (o *WorkflowExecutionCancelRequestedEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCancelRequestedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionCanceledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionCanceledEventAttributes() (o *WorkflowExecutionCanceledEventAttributes) {\n\tif v != nil && v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\treturn v.WorkflowExecutionCanceledEventAttributes\n\t}\n\treturn\n}\n\n// GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetRequestCancelExternalWorkflowExecutionInitiatedEventAttributes() (o *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\t}\n\treturn\n}\n\n// GetRequestCancelExternalWorkflowExecutionFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetRequestCancelExternalWorkflowExecutionFailedEventAttributes() (o *RequestCancelExternalWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetExternalWorkflowExecutionCancelRequestedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetExternalWorkflowExecutionCancelRequestedEventAttributes() (o *ExternalWorkflowExecutionCancelRequestedEventAttributes) {\n\tif v != nil && v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\treturn v.ExternalWorkflowExecutionCancelRequestedEventAttributes\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionContinuedAsNewEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetWorkflowExecutionContinuedAsNewEventAttributes() (o *WorkflowExecutionContinuedAsNewEventAttributes) {\n\tif v != nil && v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\treturn v.WorkflowExecutionContinuedAsNewEventAttributes\n\t}\n\treturn\n}\n\n// GetStartChildWorkflowExecutionInitiatedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetStartChildWorkflowExecutionInitiatedEventAttributes() (o *StartChildWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.StartChildWorkflowExecutionInitiatedEventAttributes\n\t}\n\treturn\n}\n\n// GetStartChildWorkflowExecutionFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetStartChildWorkflowExecutionFailedEventAttributes() (o *StartChildWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.StartChildWorkflowExecutionFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionStartedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionStartedEventAttributes() (o *ChildWorkflowExecutionStartedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionStartedEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionCompletedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionCompletedEventAttributes() (o *ChildWorkflowExecutionCompletedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionCompletedEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionFailedEventAttributes() (o *ChildWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionCanceledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionCanceledEventAttributes() (o *ChildWorkflowExecutionCanceledEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionCanceledEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionTimedOutEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionTimedOutEventAttributes() (o *ChildWorkflowExecutionTimedOutEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionTimedOutEventAttributes\n\t}\n\treturn\n}\n\n// GetChildWorkflowExecutionTerminatedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetChildWorkflowExecutionTerminatedEventAttributes() (o *ChildWorkflowExecutionTerminatedEventAttributes) {\n\tif v != nil && v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\treturn v.ChildWorkflowExecutionTerminatedEventAttributes\n\t}\n\treturn\n}\n\n// GetSignalExternalWorkflowExecutionInitiatedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetSignalExternalWorkflowExecutionInitiatedEventAttributes() (o *SignalExternalWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\treturn v.SignalExternalWorkflowExecutionInitiatedEventAttributes\n\t}\n\treturn\n}\n\n// GetSignalExternalWorkflowExecutionFailedEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetSignalExternalWorkflowExecutionFailedEventAttributes() (o *SignalExternalWorkflowExecutionFailedEventAttributes) {\n\tif v != nil && v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\treturn v.SignalExternalWorkflowExecutionFailedEventAttributes\n\t}\n\treturn\n}\n\n// GetExternalWorkflowExecutionSignaledEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetExternalWorkflowExecutionSignaledEventAttributes() (o *ExternalWorkflowExecutionSignaledEventAttributes) {\n\tif v != nil && v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\treturn v.ExternalWorkflowExecutionSignaledEventAttributes\n\t}\n\treturn\n}\n\n// GetUpsertWorkflowSearchAttributesEventAttributes is an internal getter (TBD...)\nfunc (v *HistoryEvent) GetUpsertWorkflowSearchAttributesEventAttributes() (o *UpsertWorkflowSearchAttributesEventAttributes) {\n\tif v != nil && v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\treturn v.UpsertWorkflowSearchAttributesEventAttributes\n\t}\n\treturn\n}\n\n// Size is an internal method to get the estimated size of the event\nfunc (v *HistoryEvent) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(8 + 8 + 4 + 8 + 8) // size of ID, Timestamp, EventType, Version, TaskID\n\n\tif v.WorkflowExecutionStartedEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionStartedEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionCompletedEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionCompletedEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionFailedEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionTimedOutEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionCancelRequestedEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionCanceledEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionCanceledEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionTerminatedEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionTerminatedEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionContinuedAsNewEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionContinuedAsNewEventAttributes.ByteSize()\n\t}\n\n\tif v.WorkflowExecutionSignaledEventAttributes != nil {\n\t\tsize += v.WorkflowExecutionSignaledEventAttributes.ByteSize()\n\t}\n\n\tif v.DecisionTaskScheduledEventAttributes != nil {\n\t\tsize += v.DecisionTaskScheduledEventAttributes.ByteSize()\n\t}\n\n\tif v.DecisionTaskStartedEventAttributes != nil {\n\t\tsize += v.DecisionTaskStartedEventAttributes.ByteSize()\n\t}\n\n\tif v.DecisionTaskCompletedEventAttributes != nil {\n\t\tsize += v.DecisionTaskCompletedEventAttributes.ByteSize()\n\t}\n\n\tif v.DecisionTaskTimedOutEventAttributes != nil {\n\t\tsize += v.DecisionTaskTimedOutEventAttributes.ByteSize()\n\t}\n\n\tif v.DecisionTaskFailedEventAttributes != nil {\n\t\tsize += v.DecisionTaskFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskScheduledEventAttributes != nil {\n\t\tsize += v.ActivityTaskScheduledEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskStartedEventAttributes != nil {\n\t\tsize += v.ActivityTaskStartedEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskCompletedEventAttributes != nil {\n\t\tsize += v.ActivityTaskCompletedEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskFailedEventAttributes != nil {\n\t\tsize += v.ActivityTaskFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskTimedOutEventAttributes != nil {\n\t\tsize += v.ActivityTaskTimedOutEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskCancelRequestedEventAttributes != nil {\n\t\tsize += v.ActivityTaskCancelRequestedEventAttributes.ByteSize()\n\t}\n\n\tif v.ActivityTaskCanceledEventAttributes != nil {\n\t\tsize += v.ActivityTaskCanceledEventAttributes.ByteSize()\n\t}\n\n\tif v.RequestCancelActivityTaskFailedEventAttributes != nil {\n\t\tsize += v.RequestCancelActivityTaskFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.TimerStartedEventAttributes != nil {\n\t\tsize += v.TimerStartedEventAttributes.ByteSize()\n\t}\n\n\tif v.TimerFiredEventAttributes != nil {\n\t\tsize += v.TimerFiredEventAttributes.ByteSize()\n\t}\n\n\tif v.TimerCanceledEventAttributes != nil {\n\t\tsize += v.TimerCanceledEventAttributes.ByteSize()\n\t}\n\n\tif v.CancelTimerFailedEventAttributes != nil {\n\t\tsize += v.CancelTimerFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.MarkerRecordedEventAttributes != nil {\n\t\tsize += v.MarkerRecordedEventAttributes.ByteSize()\n\t}\n\n\tif v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tsize += v.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.ByteSize()\n\t}\n\n\tif v.RequestCancelExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tsize += v.RequestCancelExternalWorkflowExecutionFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ExternalWorkflowExecutionCancelRequestedEventAttributes != nil {\n\t\tsize += v.ExternalWorkflowExecutionCancelRequestedEventAttributes.ByteSize()\n\t}\n\n\tif v.SignalExternalWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tsize += v.SignalExternalWorkflowExecutionInitiatedEventAttributes.ByteSize()\n\t}\n\n\tif v.SignalExternalWorkflowExecutionFailedEventAttributes != nil {\n\t\tsize += v.SignalExternalWorkflowExecutionFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ExternalWorkflowExecutionSignaledEventAttributes != nil {\n\t\tsize += v.ExternalWorkflowExecutionSignaledEventAttributes.ByteSize()\n\t}\n\n\tif v.StartChildWorkflowExecutionInitiatedEventAttributes != nil {\n\t\tsize += v.StartChildWorkflowExecutionInitiatedEventAttributes.ByteSize()\n\t}\n\n\tif v.StartChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tsize += v.StartChildWorkflowExecutionFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionStartedEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionStartedEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionCompletedEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionCompletedEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionFailedEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionFailedEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionCanceledEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionCanceledEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionTimedOutEventAttributes.ByteSize()\n\t}\n\n\tif v.ChildWorkflowExecutionTerminatedEventAttributes != nil {\n\t\tsize += v.ChildWorkflowExecutionTerminatedEventAttributes.ByteSize()\n\t}\n\n\tif v.UpsertWorkflowSearchAttributesEventAttributes != nil {\n\t\tsize += v.UpsertWorkflowSearchAttributesEventAttributes.ByteSize()\n\t}\n\n\treturn size\n}\n\n// HistoryEventFilterType is an internal type (TBD...)\ntype HistoryEventFilterType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e HistoryEventFilterType) Ptr() *HistoryEventFilterType {\n\treturn &e\n}\n\n// String returns a readable string representation of HistoryEventFilterType.\nfunc (e HistoryEventFilterType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ALL_EVENT\"\n\tcase 1:\n\t\treturn \"CLOSE_EVENT\"\n\t}\n\treturn fmt.Sprintf(\"HistoryEventFilterType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *HistoryEventFilterType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"ALL_EVENT\":\n\t\t*e = HistoryEventFilterTypeAllEvent\n\t\treturn nil\n\tcase \"CLOSE_EVENT\":\n\t\t*e = HistoryEventFilterTypeCloseEvent\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"HistoryEventFilterType\", err)\n\t\t}\n\t\t*e = HistoryEventFilterType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes HistoryEventFilterType to text.\nfunc (e HistoryEventFilterType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// HistoryEventFilterTypeAllEvent is an option for HistoryEventFilterType\n\tHistoryEventFilterTypeAllEvent HistoryEventFilterType = iota\n\t// HistoryEventFilterTypeCloseEvent is an option for HistoryEventFilterType\n\tHistoryEventFilterTypeCloseEvent\n)\n\n// IndexedValueType is an internal type (TBD...)\ntype IndexedValueType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e IndexedValueType) Ptr() *IndexedValueType {\n\treturn &e\n}\n\n// String returns a readable string representation of IndexedValueType.\nfunc (e IndexedValueType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"STRING\"\n\tcase 1:\n\t\treturn \"KEYWORD\"\n\tcase 2:\n\t\treturn \"INT\"\n\tcase 3:\n\t\treturn \"DOUBLE\"\n\tcase 4:\n\t\treturn \"BOOL\"\n\tcase 5:\n\t\treturn \"DATETIME\"\n\t}\n\treturn fmt.Sprintf(\"IndexedValueType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *IndexedValueType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"STRING\":\n\t\t*e = IndexedValueTypeString\n\t\treturn nil\n\tcase \"KEYWORD\":\n\t\t*e = IndexedValueTypeKeyword\n\t\treturn nil\n\tcase \"INT\":\n\t\t*e = IndexedValueTypeInt\n\t\treturn nil\n\tcase \"DOUBLE\":\n\t\t*e = IndexedValueTypeDouble\n\t\treturn nil\n\tcase \"BOOL\":\n\t\t*e = IndexedValueTypeBool\n\t\treturn nil\n\tcase \"DATETIME\":\n\t\t*e = IndexedValueTypeDatetime\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"IndexedValueType\", err)\n\t\t}\n\t\t*e = IndexedValueType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes IndexedValueType to text.\nfunc (e IndexedValueType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// IndexedValueTypeString is an option for IndexedValueType\n\tIndexedValueTypeString IndexedValueType = iota\n\t// IndexedValueTypeKeyword is an option for IndexedValueType\n\tIndexedValueTypeKeyword\n\t// IndexedValueTypeInt is an option for IndexedValueType\n\tIndexedValueTypeInt\n\t// IndexedValueTypeDouble is an option for IndexedValueType\n\tIndexedValueTypeDouble\n\t// IndexedValueTypeBool is an option for IndexedValueType\n\tIndexedValueTypeBool\n\t// IndexedValueTypeDatetime is an option for IndexedValueType\n\tIndexedValueTypeDatetime\n)\n\n// InternalDataInconsistencyError is an internal type (TBD...)\ntype InternalDataInconsistencyError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// InternalServiceError is an internal type (TBD...)\ntype InternalServiceError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// GetMessage is an internal getter (TBD...)\nfunc (v *InternalServiceError) GetMessage() (o string) {\n\tif v != nil {\n\t\treturn v.Message\n\t}\n\treturn\n}\n\n// LimitExceededError is an internal type (TBD...)\ntype LimitExceededError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ListArchivedWorkflowExecutionsRequest is an internal type (TBD...)\ntype ListArchivedWorkflowExecutionsRequest struct {\n\tDomain        string `json:\"domain,omitempty\"`\n\tPageSize      int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n\tQuery         string `json:\"query,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetPageSize is an internal getter (TBD...)\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.PageSize\n\t}\n\treturn\n}\n\n// GetQuery is an internal getter (TBD...)\nfunc (v *ListArchivedWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil {\n\t\treturn v.Query\n\t}\n\treturn\n}\n\n// ListArchivedWorkflowExecutionsResponse is an internal type (TBD...)\ntype ListArchivedWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetExecutions is an internal getter (TBD...)\nfunc (v *ListArchivedWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\treturn\n}\n\n// ListClosedWorkflowExecutionsRequest is an internal type (TBD...)\ntype ListClosedWorkflowExecutionsRequest struct {\n\tDomain          string                        `json:\"domain,omitempty\"`\n\tMaximumPageSize int32                         `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken   []byte                        `json:\"nextPageToken,omitempty\"`\n\tStartTimeFilter *StartTimeFilter              `json:\"StartTimeFilter,omitempty\"`\n\tExecutionFilter *WorkflowExecutionFilter      `json:\"executionFilter,omitempty\"`\n\tTypeFilter      *WorkflowTypeFilter           `json:\"typeFilter,omitempty\"`\n\tStatusFilter    *WorkflowExecutionCloseStatus `json:\"statusFilter,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ListClosedWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *ListClosedWorkflowExecutionsRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// GetStatusFilter is an internal getter (TBD...)\nfunc (v *ListClosedWorkflowExecutionsRequest) GetStatusFilter() (o WorkflowExecutionCloseStatus) {\n\tif v != nil && v.StatusFilter != nil {\n\t\treturn *v.StatusFilter\n\t}\n\treturn\n}\n\n// ListClosedWorkflowExecutionsResponse is an internal type (TBD...)\ntype ListClosedWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetExecutions is an internal getter (TBD...)\nfunc (v *ListClosedWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\treturn\n}\n\n// ListDomainsRequest is an internal type (TBD...)\ntype ListDomainsRequest struct {\n\tPageSize      int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n}\n\n// GetPageSize is an internal getter (TBD...)\nfunc (v *ListDomainsRequest) GetPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.PageSize\n\t}\n\treturn\n}\n\n// ListDomainsResponse is an internal type (TBD...)\ntype ListDomainsResponse struct {\n\tDomains       []*DescribeDomainResponse `json:\"domains,omitempty\"`\n\tNextPageToken []byte                    `json:\"nextPageToken,omitempty\"`\n}\n\n// GetDomains is an internal getter (TBD...)\nfunc (v *ListDomainsResponse) GetDomains() (o []*DescribeDomainResponse) {\n\tif v != nil && v.Domains != nil {\n\t\treturn v.Domains\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *ListDomainsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// ListOpenWorkflowExecutionsRequest is an internal type (TBD...)\ntype ListOpenWorkflowExecutionsRequest struct {\n\tDomain          string                   `json:\"domain,omitempty\"`\n\tMaximumPageSize int32                    `json:\"maximumPageSize,omitempty\"`\n\tNextPageToken   []byte                   `json:\"nextPageToken,omitempty\"`\n\tStartTimeFilter *StartTimeFilter         `json:\"StartTimeFilter,omitempty\"`\n\tExecutionFilter *WorkflowExecutionFilter `json:\"executionFilter,omitempty\"`\n\tTypeFilter      *WorkflowTypeFilter      `json:\"typeFilter,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ListOpenWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetMaximumPageSize is an internal getter (TBD...)\nfunc (v *ListOpenWorkflowExecutionsRequest) GetMaximumPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumPageSize\n\t}\n\treturn\n}\n\n// ListOpenWorkflowExecutionsResponse is an internal type (TBD...)\ntype ListOpenWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetExecutions is an internal getter (TBD...)\nfunc (v *ListOpenWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\treturn\n}\n\n// ListTaskListPartitionsRequest is an internal type (TBD...)\ntype ListTaskListPartitionsRequest struct {\n\tDomain   string    `json:\"domain,omitempty\"`\n\tTaskList *TaskList `json:\"taskList,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ListTaskListPartitionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *ListTaskListPartitionsRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// ListTaskListPartitionsResponse is an internal type (TBD...)\ntype ListTaskListPartitionsResponse struct {\n\tActivityTaskListPartitions []*TaskListPartitionMetadata `json:\"activityTaskListPartitions,omitempty\"`\n\tDecisionTaskListPartitions []*TaskListPartitionMetadata `json:\"decisionTaskListPartitions,omitempty\"`\n}\n\n// GetTaskListsByDomainRequest is an internal type (TBD...)\ntype GetTaskListsByDomainRequest struct {\n\tDomain string `json:\"domain,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *GetTaskListsByDomainRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskListsByDomainResponse is an internal type (TBD...)\ntype GetTaskListsByDomainResponse struct {\n\tDecisionTaskListMap map[string]*DescribeTaskListResponse `json:\"decisionTaskListMap,omitempty\"`\n\tActivityTaskListMap map[string]*DescribeTaskListResponse `json:\"activityTaskListMap,omitempty\"`\n}\n\n// GetDecisionTaskListMap is an internal getter (TBD...)\nfunc (v *GetTaskListsByDomainResponse) GetDecisionTaskListMap() (o map[string]*DescribeTaskListResponse) {\n\tif v != nil && v.DecisionTaskListMap != nil {\n\t\treturn v.DecisionTaskListMap\n\t}\n\treturn\n}\n\n// GetActivityTaskListMap is an internal getter (TBD...)\nfunc (v *GetTaskListsByDomainResponse) GetActivityTaskListMap() (o map[string]*DescribeTaskListResponse) {\n\tif v != nil && v.ActivityTaskListMap != nil {\n\t\treturn v.ActivityTaskListMap\n\t}\n\treturn\n}\n\n// ListWorkflowExecutionsRequest is an internal type (TBD...)\ntype ListWorkflowExecutionsRequest struct {\n\tDomain        string `json:\"domain,omitempty\"`\n\tPageSize      int32  `json:\"pageSize,omitempty\"`\n\tNextPageToken []byte `json:\"nextPageToken,omitempty\"`\n\tQuery         string `json:\"query,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ListWorkflowExecutionsRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetPageSize is an internal getter (TBD...)\nfunc (v *ListWorkflowExecutionsRequest) GetPageSize() (o int32) {\n\tif v != nil {\n\t\treturn v.PageSize\n\t}\n\treturn\n}\n\n// GetQuery is an internal getter (TBD...)\nfunc (v *ListWorkflowExecutionsRequest) GetQuery() (o string) {\n\tif v != nil {\n\t\treturn v.Query\n\t}\n\treturn\n}\n\n// ListWorkflowExecutionsResponse is an internal type (TBD...)\ntype ListWorkflowExecutionsResponse struct {\n\tExecutions    []*WorkflowExecutionInfo `json:\"executions,omitempty\"`\n\tNextPageToken []byte                   `json:\"nextPageToken,omitempty\"`\n}\n\n// GetExecutions is an internal getter (TBD...)\nfunc (v *ListWorkflowExecutionsResponse) GetExecutions() (o []*WorkflowExecutionInfo) {\n\tif v != nil && v.Executions != nil {\n\t\treturn v.Executions\n\t}\n\treturn\n}\n\n// GetNextPageToken is an internal getter (TBD...)\nfunc (v *ListWorkflowExecutionsResponse) GetNextPageToken() (o []byte) {\n\tif v != nil && v.NextPageToken != nil {\n\t\treturn v.NextPageToken\n\t}\n\treturn\n}\n\n// MarkerRecordedEventAttributes is an internal type (TBD...)\ntype MarkerRecordedEventAttributes struct {\n\tMarkerName                   string  `json:\"markerName,omitempty\"`\n\tDetails                      []byte  `json:\"details,omitempty\"`\n\tDecisionTaskCompletedEventID int64   `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tHeader                       *Header `json:\"header,omitempty\"`\n}\n\n// GetMarkerName is an internal getter (TBD...)\nfunc (v *MarkerRecordedEventAttributes) GetMarkerName() (o string) {\n\tif v != nil {\n\t\treturn v.MarkerName\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *MarkerRecordedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// Memo is an internal type (TBD...)\ntype Memo struct {\n\tFields map[string][]byte `json:\"fields,omitempty\"`\n}\n\n// GetFields is an internal getter (TBD...)\nfunc (v *Memo) GetFields() (o map[string][]byte) {\n\tif v != nil && v.Fields != nil {\n\t\treturn v.Fields\n\t}\n\treturn\n}\n\n// ParentClosePolicy is an internal type (TBD...)\ntype ParentClosePolicy int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e ParentClosePolicy) Ptr() *ParentClosePolicy {\n\treturn &e\n}\n\n// String returns a readable string representation of ParentClosePolicy.\nfunc (e ParentClosePolicy) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ABANDON\"\n\tcase 1:\n\t\treturn \"REQUEST_CANCEL\"\n\tcase 2:\n\t\treturn \"TERMINATE\"\n\t}\n\treturn fmt.Sprintf(\"ParentClosePolicy(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *ParentClosePolicy) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"ABANDON\":\n\t\t*e = ParentClosePolicyAbandon\n\t\treturn nil\n\tcase \"REQUEST_CANCEL\":\n\t\t*e = ParentClosePolicyRequestCancel\n\t\treturn nil\n\tcase \"TERMINATE\":\n\t\t*e = ParentClosePolicyTerminate\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"ParentClosePolicy\", err)\n\t\t}\n\t\t*e = ParentClosePolicy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes ParentClosePolicy to text.\nfunc (e ParentClosePolicy) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// ParentClosePolicyAbandon is an option for ParentClosePolicy\n\tParentClosePolicyAbandon ParentClosePolicy = iota\n\t// ParentClosePolicyRequestCancel is an option for ParentClosePolicy\n\tParentClosePolicyRequestCancel\n\t// ParentClosePolicyTerminate is an option for ParentClosePolicy\n\tParentClosePolicyTerminate\n)\n\n// PendingActivityInfo is an internal type (TBD...)\ntype PendingActivityInfo struct {\n\tActivityID             string                `json:\"activityID,omitempty\"`\n\tActivityType           *ActivityType         `json:\"activityType,omitempty\"`\n\tState                  *PendingActivityState `json:\"state,omitempty\"`\n\tHeartbeatDetails       []byte                `json:\"heartbeatDetails,omitempty\"`\n\tLastHeartbeatTimestamp *int64                `json:\"lastHeartbeatTimestamp,omitempty\"`\n\tLastStartedTimestamp   *int64                `json:\"lastStartedTimestamp,omitempty\"`\n\tAttempt                int32                 `json:\"attempt,omitempty\"`\n\tMaximumAttempts        int32                 `json:\"maximumAttempts,omitempty\"`\n\tScheduledTimestamp     *int64                `json:\"scheduledTimestamp,omitempty\"`\n\tExpirationTimestamp    *int64                `json:\"expirationTimestamp,omitempty\"`\n\tLastFailureReason      *string               `json:\"lastFailureReason,omitempty\"`\n\tStartedWorkerIdentity  string                `json:\"startedWorkerIdentity,omitempty\"`\n\tLastWorkerIdentity     string                `json:\"lastWorkerIdentity,omitempty\"`\n\tLastFailureDetails     []byte                `json:\"lastFailureDetails,omitempty\"`\n\tScheduleID             int64                 `json:\"scheduleID,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetState is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetState() (o PendingActivityState) {\n\tif v != nil && v.State != nil {\n\t\treturn *v.State\n\t}\n\treturn\n}\n\n// GetHeartbeatDetails is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetHeartbeatDetails() (o []byte) {\n\tif v != nil && v.HeartbeatDetails != nil {\n\t\treturn v.HeartbeatDetails\n\t}\n\treturn\n}\n\n// GetLastHeartbeatTimestamp is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetLastHeartbeatTimestamp() (o int64) {\n\tif v != nil && v.LastHeartbeatTimestamp != nil {\n\t\treturn *v.LastHeartbeatTimestamp\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetAttempt() (o int32) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetMaximumAttempts is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetMaximumAttempts() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumAttempts\n\t}\n\treturn\n}\n\n// GetLastFailureReason is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetLastFailureReason() (o string) {\n\tif v != nil && v.LastFailureReason != nil {\n\t\treturn *v.LastFailureReason\n\t}\n\treturn\n}\n\n// GetStartedWorkerIdentity is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetStartedWorkerIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.StartedWorkerIdentity\n\t}\n\treturn\n}\n\n// GetLastWorkerIdentity is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetLastWorkerIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.LastWorkerIdentity\n\t}\n\treturn\n}\n\n// GetLastFailureDetails is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetLastFailureDetails() (o []byte) {\n\tif v != nil && v.LastFailureDetails != nil {\n\t\treturn v.LastFailureDetails\n\t}\n\treturn\n}\n\n// GetScheduleID is an internal getter (TBD...)\nfunc (v *PendingActivityInfo) GetScheduleID() (o int64) {\n\tif v != nil {\n\t\treturn v.ScheduleID\n\t}\n\treturn\n}\n\n// PendingActivityState is an internal type (TBD...)\ntype PendingActivityState int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e PendingActivityState) Ptr() *PendingActivityState {\n\treturn &e\n}\n\n// String returns a readable string representation of PendingActivityState.\nfunc (e PendingActivityState) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SCHEDULED\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\tcase 2:\n\t\treturn \"CANCEL_REQUESTED\"\n\t}\n\treturn fmt.Sprintf(\"PendingActivityState(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *PendingActivityState) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"SCHEDULED\":\n\t\t*e = PendingActivityStateScheduled\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*e = PendingActivityStateStarted\n\t\treturn nil\n\tcase \"CANCEL_REQUESTED\":\n\t\t*e = PendingActivityStateCancelRequested\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"PendingActivityState\", err)\n\t\t}\n\t\t*e = PendingActivityState(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes PendingActivityState to text.\nfunc (e PendingActivityState) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// PendingActivityStateScheduled is an option for PendingActivityState\n\tPendingActivityStateScheduled PendingActivityState = iota\n\t// PendingActivityStateStarted is an option for PendingActivityState\n\tPendingActivityStateStarted\n\t// PendingActivityStateCancelRequested is an option for PendingActivityState\n\tPendingActivityStateCancelRequested\n)\n\n// PendingChildExecutionInfo is an internal type (TBD...)\ntype PendingChildExecutionInfo struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowID        string             `json:\"workflowID,omitempty\"`\n\tRunID             string             `json:\"runID,omitempty\"`\n\tWorkflowTypeName  string             `json:\"workflowTypeName,omitempty\"`\n\tInitiatedID       int64              `json:\"initiatedID,omitempty\"`\n\tParentClosePolicy *ParentClosePolicy `json:\"parentClosePolicy,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *PendingChildExecutionInfo) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *PendingChildExecutionInfo) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *PendingChildExecutionInfo) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetWorkflowTypeName is an internal getter (TBD...)\nfunc (v *PendingChildExecutionInfo) GetWorkflowTypeName() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowTypeName\n\t}\n\treturn\n}\n\n// PendingDecisionInfo is an internal type (TBD...)\ntype PendingDecisionInfo struct {\n\tState                      *PendingDecisionState `json:\"state,omitempty\"`\n\tScheduledTimestamp         *int64                `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp           *int64                `json:\"startedTimestamp,omitempty\"`\n\tAttempt                    int64                 `json:\"attempt,omitempty\"`\n\tOriginalScheduledTimestamp *int64                `json:\"originalScheduledTimestamp,omitempty\"`\n\tScheduleID                 int64                 `json:\"scheduleID,omitempty\"`\n}\n\n// PendingDecisionState is an internal type (TBD...)\ntype PendingDecisionState int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e PendingDecisionState) Ptr() *PendingDecisionState {\n\treturn &e\n}\n\n// String returns a readable string representation of PendingDecisionState.\nfunc (e PendingDecisionState) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SCHEDULED\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\t}\n\treturn fmt.Sprintf(\"PendingDecisionState(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *PendingDecisionState) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"SCHEDULED\":\n\t\t*e = PendingDecisionStateScheduled\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*e = PendingDecisionStateStarted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"PendingDecisionState\", err)\n\t\t}\n\t\t*e = PendingDecisionState(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes PendingDecisionState to text.\nfunc (e PendingDecisionState) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// PendingDecisionStateScheduled is an option for PendingDecisionState\n\tPendingDecisionStateScheduled PendingDecisionState = iota\n\t// PendingDecisionStateStarted is an option for PendingDecisionState\n\tPendingDecisionStateStarted\n)\n\n// PollForActivityTaskRequest is an internal type (TBD...)\ntype PollForActivityTaskRequest struct {\n\tDomain           string            `json:\"domain,omitempty\"`\n\tTaskList         *TaskList         `json:\"taskList,omitempty\"`\n\tIdentity         string            `json:\"identity,omitempty\"`\n\tTaskListMetadata *TaskListMetadata `json:\"taskListMetadata,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *PollForActivityTaskRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *PollForActivityTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *PollForActivityTaskRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// PollForActivityTaskResponse is an internal type (TBD...)\ntype PollForActivityTaskResponse struct {\n\tTaskToken                       []byte             `json:\"taskToken,omitempty\"`\n\tWorkflowExecution               *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tActivityID                      string             `json:\"activityId,omitempty\"`\n\tActivityType                    *ActivityType      `json:\"activityType,omitempty\"`\n\tInput                           []byte             `json:\"input,omitempty\"`\n\tScheduledTimestamp              *int64             `json:\"scheduledTimestamp,omitempty\"`\n\tScheduleToCloseTimeoutSeconds   *int32             `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tStartedTimestamp                *int64             `json:\"startedTimestamp,omitempty\"`\n\tStartToCloseTimeoutSeconds      *int32             `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds         *int32             `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tAttempt                         int32              `json:\"attempt,omitempty\"`\n\tScheduledTimestampOfThisAttempt *int64             `json:\"scheduledTimestampOfThisAttempt,omitempty\"`\n\tHeartbeatDetails                []byte             `json:\"heartbeatDetails,omitempty\"`\n\tWorkflowType                    *WorkflowType      `json:\"workflowType,omitempty\"`\n\tWorkflowDomain                  string             `json:\"workflowDomain,omitempty\"`\n\tHeader                          *Header            `json:\"header,omitempty\"`\n\tAutoConfigHint                  *AutoConfigHint    `json:\"autoConfigHint,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *PollForActivityTaskResponse) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// PollForDecisionTaskRequest is an internal type (TBD...)\ntype PollForDecisionTaskRequest struct {\n\tDomain         string    `json:\"domain,omitempty\"`\n\tTaskList       *TaskList `json:\"taskList,omitempty\"`\n\tIdentity       string    `json:\"identity,omitempty\"`\n\tBinaryChecksum string    `json:\"binaryChecksum,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetBinaryChecksum is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskRequest) GetBinaryChecksum() (o string) {\n\tif v != nil {\n\t\treturn v.BinaryChecksum\n\t}\n\treturn\n}\n\n// PollForDecisionTaskResponse is an internal type (TBD...)\ntype PollForDecisionTaskResponse struct {\n\tTaskToken                 []byte                    `json:\"taskToken,omitempty\"`\n\tWorkflowExecution         *WorkflowExecution        `json:\"workflowExecution,omitempty\"`\n\tWorkflowType              *WorkflowType             `json:\"workflowType,omitempty\"`\n\tPreviousStartedEventID    *int64                    `json:\"previousStartedEventId,omitempty\"`\n\tStartedEventID            int64                     `json:\"startedEventId,omitempty\"`\n\tAttempt                   int64                     `json:\"attempt,omitempty\"`\n\tBacklogCountHint          int64                     `json:\"backlogCountHint,omitempty\"`\n\tHistory                   *History                  `json:\"history,omitempty\"`\n\tNextPageToken             []byte                    `json:\"nextPageToken,omitempty\"`\n\tQuery                     *WorkflowQuery            `json:\"query,omitempty\"`\n\tWorkflowExecutionTaskList *TaskList                 `json:\"WorkflowExecutionTaskList,omitempty\"`\n\tScheduledTimestamp        *int64                    `json:\"scheduledTimestamp,omitempty\"`\n\tStartedTimestamp          *int64                    `json:\"startedTimestamp,omitempty\"`\n\tQueries                   map[string]*WorkflowQuery `json:\"queries,omitempty\"`\n\tNextEventID               int64                     `json:\"nextEventId,omitempty\"`\n\tTotalHistoryBytes         int64                     `json:\"historySize,omitempty\"`\n\tAutoConfigHint            *AutoConfigHint           `json:\"autoConfigHint,omitempty\"`\n}\n\n// GetTaskToken is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetTaskToken() (o []byte) {\n\tif v != nil && v.TaskToken != nil {\n\t\treturn v.TaskToken\n\t}\n\treturn\n}\n\n// GetPreviousStartedEventID is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetPreviousStartedEventID() (o int64) {\n\tif v != nil && v.PreviousStartedEventID != nil {\n\t\treturn *v.PreviousStartedEventID\n\t}\n\treturn\n}\n\n// GetStartedEventID is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetStartedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedEventID\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetAttempt() (o int64) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetQueries is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetQueries() (o map[string]*WorkflowQuery) {\n\tif v != nil && v.Queries != nil {\n\t\treturn v.Queries\n\t}\n\treturn\n}\n\n// GetNextEventID is an internal getter (TBD...)\nfunc (v *PollForDecisionTaskResponse) GetNextEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.NextEventID\n\t}\n\treturn\n}\n\nfunc (v *PollForDecisionTaskResponse) GetHistorySize() (o int64) {\n\tif v != nil {\n\t\treturn v.TotalHistoryBytes\n\t}\n\treturn\n}\n\n// PollerInfo is an internal type (TBD...)\ntype PollerInfo struct {\n\tLastAccessTime *int64  `json:\"lastAccessTime,omitempty\"`\n\tIdentity       string  `json:\"identity,omitempty\"`\n\tRatePerSecond  float64 `json:\"ratePerSecond,omitempty\"`\n}\n\n// GetLastAccessTime is an internal getter (TBD...)\nfunc (v *PollerInfo) GetLastAccessTime() (o int64) {\n\tif v != nil && v.LastAccessTime != nil {\n\t\treturn *v.LastAccessTime\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *PollerInfo) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetRatePerSecond is an internal getter (TBD...)\nfunc (v *PollerInfo) GetRatePerSecond() (o float64) {\n\tif v != nil {\n\t\treturn v.RatePerSecond\n\t}\n\treturn\n}\n\n// QueryConsistencyLevel is an internal type (TBD...)\ntype QueryConsistencyLevel int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e QueryConsistencyLevel) Ptr() *QueryConsistencyLevel {\n\treturn &e\n}\n\n// String returns a readable string representation of QueryConsistencyLevel.\nfunc (e QueryConsistencyLevel) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"EVENTUAL\"\n\tcase 1:\n\t\treturn \"STRONG\"\n\t}\n\treturn fmt.Sprintf(\"QueryConsistencyLevel(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *QueryConsistencyLevel) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"EVENTUAL\":\n\t\t*e = QueryConsistencyLevelEventual\n\t\treturn nil\n\tcase \"STRONG\":\n\t\t*e = QueryConsistencyLevelStrong\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryConsistencyLevel\", err)\n\t\t}\n\t\t*e = QueryConsistencyLevel(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryConsistencyLevel to text.\nfunc (e QueryConsistencyLevel) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// QueryConsistencyLevelEventual is an option for QueryConsistencyLevel\n\tQueryConsistencyLevelEventual QueryConsistencyLevel = iota\n\t// QueryConsistencyLevelStrong is an option for QueryConsistencyLevel\n\tQueryConsistencyLevelStrong\n)\n\n// QueryFailedError is an internal type (TBD...)\ntype QueryFailedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// QueryRejectCondition is an internal type (TBD...)\ntype QueryRejectCondition int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e QueryRejectCondition) Ptr() *QueryRejectCondition {\n\treturn &e\n}\n\n// String returns a readable string representation of QueryRejectCondition.\nfunc (e QueryRejectCondition) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"NOT_OPEN\"\n\tcase 1:\n\t\treturn \"NOT_COMPLETED_CLEANLY\"\n\t}\n\treturn fmt.Sprintf(\"QueryRejectCondition(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *QueryRejectCondition) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"NOT_OPEN\":\n\t\t*e = QueryRejectConditionNotOpen\n\t\treturn nil\n\tcase \"NOT_COMPLETED_CLEANLY\":\n\t\t*e = QueryRejectConditionNotCompletedCleanly\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryRejectCondition\", err)\n\t\t}\n\t\t*e = QueryRejectCondition(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryRejectCondition to text.\nfunc (e QueryRejectCondition) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// QueryRejectConditionNotOpen is an option for QueryRejectCondition\n\tQueryRejectConditionNotOpen QueryRejectCondition = iota\n\t// QueryRejectConditionNotCompletedCleanly is an option for QueryRejectCondition\n\tQueryRejectConditionNotCompletedCleanly\n)\n\n// QueryRejected is an internal type (TBD...)\ntype QueryRejected struct {\n\tCloseStatus *WorkflowExecutionCloseStatus `json:\"closeStatus,omitempty\"`\n}\n\n// QueryResultType is an internal type (TBD...)\ntype QueryResultType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e QueryResultType) Ptr() *QueryResultType {\n\treturn &e\n}\n\n// String returns a readable string representation of QueryResultType.\nfunc (e QueryResultType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"ANSWERED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\t}\n\treturn fmt.Sprintf(\"QueryResultType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *QueryResultType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"ANSWERED\":\n\t\t*e = QueryResultTypeAnswered\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*e = QueryResultTypeFailed\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryResultType\", err)\n\t\t}\n\t\t*e = QueryResultType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryResultType to text.\nfunc (e QueryResultType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// QueryResultTypeAnswered is an option for QueryResultType\n\tQueryResultTypeAnswered QueryResultType = iota\n\t// QueryResultTypeFailed is an option for QueryResultType\n\tQueryResultTypeFailed\n)\n\n// QueryTaskCompletedType is an internal type (TBD...)\ntype QueryTaskCompletedType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e QueryTaskCompletedType) Ptr() *QueryTaskCompletedType {\n\treturn &e\n}\n\n// String returns a readable string representation of QueryTaskCompletedType.\nfunc (e QueryTaskCompletedType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"COMPLETED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\t}\n\treturn fmt.Sprintf(\"QueryTaskCompletedType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *QueryTaskCompletedType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"COMPLETED\":\n\t\t*e = QueryTaskCompletedTypeCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*e = QueryTaskCompletedTypeFailed\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"QueryTaskCompletedType\", err)\n\t\t}\n\t\t*e = QueryTaskCompletedType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes QueryTaskCompletedType to text.\nfunc (e QueryTaskCompletedType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// QueryTaskCompletedTypeCompleted is an option for QueryTaskCompletedType\n\tQueryTaskCompletedTypeCompleted QueryTaskCompletedType = iota\n\t// QueryTaskCompletedTypeFailed is an option for QueryTaskCompletedType\n\tQueryTaskCompletedTypeFailed\n)\n\n// QueryWorkflowRequest is an internal type (TBD...)\ntype QueryWorkflowRequest struct {\n\tDomain                string                 `json:\"domain,omitempty\"`\n\tExecution             *WorkflowExecution     `json:\"execution,omitempty\"`\n\tQuery                 *WorkflowQuery         `json:\"query,omitempty\"`\n\tQueryRejectCondition  *QueryRejectCondition  `json:\"queryRejectCondition,omitempty\"`\n\tQueryConsistencyLevel *QueryConsistencyLevel `json:\"queryConsistencyLevel,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *QueryWorkflowRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *QueryWorkflowRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetQuery is an internal getter (TBD...)\nfunc (v *QueryWorkflowRequest) GetQuery() (o *WorkflowQuery) {\n\tif v != nil && v.Query != nil {\n\t\treturn v.Query\n\t}\n\treturn\n}\n\n// GetQueryRejectCondition is an internal getter (TBD...)\nfunc (v *QueryWorkflowRequest) GetQueryRejectCondition() (o QueryRejectCondition) {\n\tif v != nil && v.QueryRejectCondition != nil {\n\t\treturn *v.QueryRejectCondition\n\t}\n\treturn\n}\n\n// GetQueryConsistencyLevel is an internal getter (TBD...)\nfunc (v *QueryWorkflowRequest) GetQueryConsistencyLevel() (o QueryConsistencyLevel) {\n\tif v != nil && v.QueryConsistencyLevel != nil {\n\t\treturn *v.QueryConsistencyLevel\n\t}\n\treturn\n}\n\n// QueryWorkflowResponse is an internal type (TBD...)\ntype QueryWorkflowResponse struct {\n\tQueryResult   []byte         `json:\"queryResult,omitempty\"`\n\tQueryRejected *QueryRejected `json:\"queryRejected,omitempty\"`\n}\n\n// GetQueryResult is an internal getter (TBD...)\nfunc (v *QueryWorkflowResponse) GetQueryResult() (o []byte) {\n\tif v != nil && v.QueryResult != nil {\n\t\treturn v.QueryResult\n\t}\n\treturn\n}\n\n// GetQueryRejected is an internal getter (TBD...)\nfunc (v *QueryWorkflowResponse) GetQueryRejected() (o *QueryRejected) {\n\tif v != nil && v.QueryRejected != nil {\n\t\treturn v.QueryRejected\n\t}\n\treturn\n}\n\n// ReapplyEventsRequest is an internal type (TBD...)\ntype ReapplyEventsRequest struct {\n\tDomainName        string             `json:\"domainName,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tEvents            *DataBlob          `json:\"events,omitempty\"`\n}\n\n// GetDomainName is an internal getter (TBD...)\nfunc (v *ReapplyEventsRequest) GetDomainName() (o string) {\n\tif v != nil {\n\t\treturn v.DomainName\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *ReapplyEventsRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetEvents is an internal getter (TBD...)\nfunc (v *ReapplyEventsRequest) GetEvents() (o *DataBlob) {\n\tif v != nil && v.Events != nil {\n\t\treturn v.Events\n\t}\n\treturn\n}\n\n// RecordActivityTaskHeartbeatByIDRequest is an internal type (TBD...)\ntype RecordActivityTaskHeartbeatByIDRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tWorkflowID string `json:\"workflowID,omitempty\"`\n\tRunID      string `json:\"runID,omitempty\"`\n\tActivityID string `json:\"activityID,omitempty\"`\n\tDetails    []byte `json:\"details,omitempty\"`\n\tIdentity   string `json:\"identity,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *RecordActivityTaskHeartbeatByIDRequest) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// RecordActivityTaskHeartbeatRequest is an internal type (TBD...)\ntype RecordActivityTaskHeartbeatRequest struct {\n\tTaskToken []byte `json:\"taskToken,omitempty\"`\n\tDetails   []byte `json:\"details,omitempty\"`\n\tIdentity  string `json:\"identity,omitempty\"`\n}\n\n// RecordActivityTaskHeartbeatResponse is an internal type (TBD...)\ntype RecordActivityTaskHeartbeatResponse struct {\n\tCancelRequested bool `json:\"cancelRequested,omitempty\"`\n}\n\n// RecordMarkerDecisionAttributes is an internal type (TBD...)\ntype RecordMarkerDecisionAttributes struct {\n\tMarkerName string  `json:\"markerName,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tHeader     *Header `json:\"header,omitempty\"`\n}\n\n// GetMarkerName is an internal getter (TBD...)\nfunc (v *RecordMarkerDecisionAttributes) GetMarkerName() (o string) {\n\tif v != nil {\n\t\treturn v.MarkerName\n\t}\n\treturn\n}\n\n// RefreshWorkflowTasksRequest is an internal type (TBD...)\ntype RefreshWorkflowTasksRequest struct {\n\tDomain    string             `json:\"domain,omitempty\"`\n\tExecution *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RefreshWorkflowTasksRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *RefreshWorkflowTasksRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// RegisterDomainRequest is an internal type (TBD...)\ntype RegisterDomainRequest struct {\n\tName                                   string                             `json:\"name,omitempty\"`\n\tDescription                            string                             `json:\"description,omitempty\"`\n\tOwnerEmail                             string                             `json:\"ownerEmail,omitempty\"`\n\tWorkflowExecutionRetentionPeriodInDays int32                              `json:\"workflowExecutionRetentionPeriodInDays,omitempty\"`\n\tEmitMetric                             *bool                              `json:\"emitMetric,omitempty\"`\n\tClusters                               []*ClusterReplicationConfiguration `json:\"clusters,omitempty\"`\n\tActiveClusterName                      string                             `json:\"activeClusterName,omitempty\"`\n\tActiveClusters                         *ActiveClusters                    `json:\"activeClusters,omitempty\"`\n\tData                                   map[string]string                  `json:\"data,omitempty\"`\n\tSecurityToken                          string                             `json:\"securityToken,omitempty\"`\n\tIsGlobalDomain                         bool                               `json:\"isGlobalDomain,omitempty\"`\n\tHistoryArchivalStatus                  *ArchivalStatus                    `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                     string                             `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus               *ArchivalStatus                    `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                  string                             `json:\"visibilityArchivalURI,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// GetDescription is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetDescription() (o string) {\n\tif v != nil {\n\t\treturn v.Description\n\t}\n\treturn\n}\n\n// GetOwnerEmail is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetOwnerEmail() (o string) {\n\tif v != nil {\n\t\treturn v.OwnerEmail\n\t}\n\treturn\n}\n\n// GetWorkflowExecutionRetentionPeriodInDays is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetWorkflowExecutionRetentionPeriodInDays() (o int32) {\n\tif v != nil {\n\t\treturn v.WorkflowExecutionRetentionPeriodInDays\n\t}\n\treturn\n}\n\n// GetEmitMetric is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetEmitMetric() (o bool) {\n\tif v != nil && v.EmitMetric != nil {\n\t\treturn *v.EmitMetric\n\t}\n\to = true\n\treturn\n}\n\n// GetActiveClusterName is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetActiveClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ActiveClusterName\n\t}\n\treturn\n}\n\n// GetData is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetData() (o map[string]string) {\n\tif v != nil && v.Data != nil {\n\t\treturn v.Data\n\t}\n\treturn\n}\n\n// GetIsGlobalDomain is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetIsGlobalDomain() (o bool) {\n\tif v != nil {\n\t\treturn v.IsGlobalDomain\n\t}\n\treturn\n}\n\n// GetHistoryArchivalURI is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetHistoryArchivalURI() (o string) {\n\tif v != nil {\n\t\treturn v.HistoryArchivalURI\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalURI is an internal getter (TBD...)\nfunc (v *RegisterDomainRequest) GetVisibilityArchivalURI() (o string) {\n\tif v != nil {\n\t\treturn v.VisibilityArchivalURI\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RegisterDomainRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RemoteSyncMatchedError is an internal type (TBD...)\ntype RemoteSyncMatchedError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RemoteSyncMatchedError) ByteSize() uint64 {\n\treturn 0\n}\n\n// RemoveTaskRequest is an internal type (TBD...)\ntype RemoveTaskRequest struct {\n\tShardID             int32  `json:\"shardID,omitempty\"`\n\tType                *int32 `json:\"type,omitempty\"`\n\tTaskID              int64  `json:\"taskID,omitempty\"`\n\tVisibilityTimestamp *int64 `json:\"visibilityTimestamp,omitempty\"`\n\tClusterName         string `json:\"clusterName,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *RemoveTaskRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *RemoveTaskRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *RemoveTaskRequest) GetTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetVisibilityTimestamp is an internal getter (TBD...)\nfunc (v *RemoveTaskRequest) GetVisibilityTimestamp() (o int64) {\n\tif v != nil && v.VisibilityTimestamp != nil {\n\t\treturn *v.VisibilityTimestamp\n\t}\n\treturn\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *RemoveTaskRequest) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RemoveTaskRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelActivityTaskDecisionAttributes is an internal type (TBD...)\ntype RequestCancelActivityTaskDecisionAttributes struct {\n\tActivityID string `json:\"activityId,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *RequestCancelActivityTaskDecisionAttributes) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelActivityTaskDecisionAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelActivityTaskFailedEventAttributes is an internal type (TBD...)\ntype RequestCancelActivityTaskFailedEventAttributes struct {\n\tActivityID                   string `json:\"activityId,omitempty\"`\n\tCause                        string `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelActivityTaskFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelExternalWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype RequestCancelExternalWorkflowExecutionDecisionAttributes struct {\n\tDomain            string `json:\"domain,omitempty\"`\n\tWorkflowID        string `json:\"workflowId,omitempty\"`\n\tRunID             string `json:\"runId,omitempty\"`\n\tControl           []byte `json:\"control,omitempty\"`\n\tChildWorkflowOnly bool   `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelExternalWorkflowExecutionDecisionAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelExternalWorkflowExecutionFailedEventAttributes is an internal type (TBD...)\ntype RequestCancelExternalWorkflowExecutionFailedEventAttributes struct {\n\tCause                        *CancelExternalWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventID int64                                       `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       string                                      `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution                          `json:\"workflowExecution,omitempty\"`\n\tInitiatedEventID             int64                                       `json:\"initiatedEventId,omitempty\"`\n\tControl                      []byte                                      `json:\"control,omitempty\"`\n}\n\n// GetDecisionTaskCompletedEventID is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetDecisionTaskCompletedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.DecisionTaskCompletedEventID\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelExternalWorkflowExecutionFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelExternalWorkflowExecutionInitiatedEventAttributes is an internal type (TBD...)\ntype RequestCancelExternalWorkflowExecutionInitiatedEventAttributes struct {\n\tDecisionTaskCompletedEventID int64              `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       string             `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tControl                      []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly            bool               `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetChildWorkflowOnly is an internal getter (TBD...)\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil {\n\t\treturn v.ChildWorkflowOnly\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelExternalWorkflowExecutionInitiatedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// RequestCancelWorkflowExecutionRequest is an internal type (TBD...)\ntype RequestCancelWorkflowExecutionRequest struct {\n\tDomain              string             `json:\"domain,omitempty\"`\n\tWorkflowExecution   *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIdentity            string             `json:\"identity,omitempty\"`\n\tRequestID           string             `json:\"requestId,omitempty\"`\n\tCause               string             `json:\"cause,omitempty\"`\n\tFirstExecutionRunID string             `json:\"first_execution_run_id,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RequestCancelWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *RequestCancelWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *RequestCancelWorkflowExecutionRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// GetFirstExecutionRunID is an internal getter (TBD...)\nfunc (v *RequestCancelWorkflowExecutionRequest) GetFirstExecutionRunID() (o string) {\n\tif v != nil {\n\t\treturn v.FirstExecutionRunID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RequestCancelWorkflowExecutionRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetPointInfo is an internal type (TBD...)\ntype ResetPointInfo struct {\n\tBinaryChecksum           string `json:\"binaryChecksum,omitempty\"`\n\tRunID                    string `json:\"runId,omitempty\"`\n\tFirstDecisionCompletedID int64  `json:\"firstDecisionCompletedId,omitempty\"`\n\tCreatedTimeNano          *int64 `json:\"createdTimeNano,omitempty\"`\n\tExpiringTimeNano         *int64 `json:\"expiringTimeNano,omitempty\"`\n\tResettable               bool   `json:\"resettable,omitempty\"`\n}\n\n// GetBinaryChecksum is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetBinaryChecksum() (o string) {\n\tif v != nil {\n\t\treturn v.BinaryChecksum\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetFirstDecisionCompletedID is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetFirstDecisionCompletedID() (o int64) {\n\tif v != nil {\n\t\treturn v.FirstDecisionCompletedID\n\t}\n\treturn\n}\n\n// GetCreatedTimeNano is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetCreatedTimeNano() (o int64) {\n\tif v != nil && v.CreatedTimeNano != nil {\n\t\treturn *v.CreatedTimeNano\n\t}\n\treturn\n}\n\n// GetExpiringTimeNano is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetExpiringTimeNano() (o int64) {\n\tif v != nil && v.ExpiringTimeNano != nil {\n\t\treturn *v.ExpiringTimeNano\n\t}\n\treturn\n}\n\n// GetResettable is an internal getter (TBD...)\nfunc (v *ResetPointInfo) GetResettable() (o bool) {\n\tif v != nil {\n\t\treturn v.Resettable\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetPointInfo) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetPoints is an internal type (TBD...)\ntype ResetPoints struct {\n\tPoints []*ResetPointInfo `json:\"points,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetPoints) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetQueueRequest is an internal type (TBD...)\ntype ResetQueueRequest struct {\n\tShardID     int32  `json:\"shardID,omitempty\"`\n\tClusterName string `json:\"clusterName,omitempty\"`\n\tType        *int32 `json:\"type,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *ResetQueueRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetClusterName is an internal getter (TBD...)\nfunc (v *ResetQueueRequest) GetClusterName() (o string) {\n\tif v != nil {\n\t\treturn v.ClusterName\n\t}\n\treturn\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *ResetQueueRequest) GetType() (o int32) {\n\tif v != nil && v.Type != nil {\n\t\treturn *v.Type\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetQueueRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetStickyTaskListRequest is an internal type (TBD...)\ntype ResetStickyTaskListRequest struct {\n\tDomain    string             `json:\"domain,omitempty\"`\n\tExecution *WorkflowExecution `json:\"execution,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ResetStickyTaskListRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *ResetStickyTaskListRequest) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetStickyTaskListRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetStickyTaskListResponse is an internal type (TBD...)\ntype ResetStickyTaskListResponse struct {\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetStickyTaskListResponse) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetWorkflowExecutionRequest is an internal type (TBD...)\ntype ResetWorkflowExecutionRequest struct {\n\tDomain                string             `json:\"domain,omitempty\"`\n\tWorkflowExecution     *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tReason                string             `json:\"reason,omitempty\"`\n\tDecisionFinishEventID int64              `json:\"decisionFinishEventId,omitempty\"`\n\tRequestID             string             `json:\"requestId,omitempty\"`\n\tSkipSignalReapply     bool               `json:\"skipSignalReapply,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\n// GetDecisionFinishEventID is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetDecisionFinishEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.DecisionFinishEventID\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// GetSkipSignalReapply is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionRequest) GetSkipSignalReapply() (o bool) {\n\tif v != nil {\n\t\treturn v.SkipSignalReapply\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetWorkflowExecutionRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// ResetWorkflowExecutionResponse is an internal type (TBD...)\ntype ResetWorkflowExecutionResponse struct {\n\tRunID string `json:\"runId,omitempty\"`\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *ResetWorkflowExecutionResponse) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *ResetWorkflowExecutionResponse) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskCanceledByIDRequest is an internal type (TBD...)\ntype RespondActivityTaskCanceledByIDRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tWorkflowID string `json:\"workflowID,omitempty\"`\n\tRunID      string `json:\"runID,omitempty\"`\n\tActivityID string `json:\"activityID,omitempty\"`\n\tDetails    []byte `json:\"details,omitempty\"`\n\tIdentity   string `json:\"identity,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledByIDRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskCanceledByIDRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskCanceledRequest is an internal type (TBD...)\ntype RespondActivityTaskCanceledRequest struct {\n\tTaskToken []byte `json:\"taskToken,omitempty\"`\n\tDetails   []byte `json:\"details,omitempty\"`\n\tIdentity  string `json:\"identity,omitempty\"`\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCanceledRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskCanceledRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskCompletedByIDRequest is an internal type (TBD...)\ntype RespondActivityTaskCompletedByIDRequest struct {\n\tDomain     string `json:\"domain,omitempty\"`\n\tWorkflowID string `json:\"workflowID,omitempty\"`\n\tRunID      string `json:\"runID,omitempty\"`\n\tActivityID string `json:\"activityID,omitempty\"`\n\tResult     []byte `json:\"result,omitempty\"`\n\tIdentity   string `json:\"identity,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedByIDRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskCompletedByIDRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskCompletedRequest is an internal type (TBD...)\ntype RespondActivityTaskCompletedRequest struct {\n\tTaskToken []byte `json:\"taskToken,omitempty\"`\n\tResult    []byte `json:\"result,omitempty\"`\n\tIdentity  string `json:\"identity,omitempty\"`\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskCompletedRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskCompletedRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskFailedByIDRequest is an internal type (TBD...)\ntype RespondActivityTaskFailedByIDRequest struct {\n\tDomain     string  `json:\"domain,omitempty\"`\n\tWorkflowID string  `json:\"workflowID,omitempty\"`\n\tRunID      string  `json:\"runID,omitempty\"`\n\tActivityID string  `json:\"activityID,omitempty\"`\n\tReason     *string `json:\"reason,omitempty\"`\n\tDetails    []byte  `json:\"details,omitempty\"`\n\tIdentity   string  `json:\"identity,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedByIDRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedByIDRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedByIDRequest) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedByIDRequest) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedByIDRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskFailedByIDRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondActivityTaskFailedRequest is an internal type (TBD...)\ntype RespondActivityTaskFailedRequest struct {\n\tTaskToken []byte  `json:\"taskToken,omitempty\"`\n\tReason    *string `json:\"reason,omitempty\"`\n\tDetails   []byte  `json:\"details,omitempty\"`\n\tIdentity  string  `json:\"identity,omitempty\"`\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedRequest) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\treturn\n}\n\n// GetDetails is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondActivityTaskFailedRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondActivityTaskFailedRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondDecisionTaskCompletedRequest is an internal type (TBD...)\ntype RespondDecisionTaskCompletedRequest struct {\n\tTaskToken                  []byte                          `json:\"taskToken,omitempty\"`\n\tDecisions                  []*Decision                     `json:\"decisions,omitempty\"`\n\tExecutionContext           []byte                          `json:\"executionContext,omitempty\"`\n\tIdentity                   string                          `json:\"identity,omitempty\"`\n\tStickyAttributes           *StickyExecutionAttributes      `json:\"stickyAttributes,omitempty\"`\n\tReturnNewDecisionTask      bool                            `json:\"returnNewDecisionTask,omitempty\"`\n\tForceCreateNewDecisionTask bool                            `json:\"forceCreateNewDecisionTask,omitempty\"`\n\tBinaryChecksum             string                          `json:\"binaryChecksum,omitempty\"`\n\tQueryResults               map[string]*WorkflowQueryResult `json:\"queryResults,omitempty\"`\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetReturnNewDecisionTask is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedRequest) GetReturnNewDecisionTask() (o bool) {\n\tif v != nil {\n\t\treturn v.ReturnNewDecisionTask\n\t}\n\treturn\n}\n\n// GetForceCreateNewDecisionTask is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedRequest) GetForceCreateNewDecisionTask() (o bool) {\n\tif v != nil {\n\t\treturn v.ForceCreateNewDecisionTask\n\t}\n\treturn\n}\n\n// GetBinaryChecksum is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedRequest) GetBinaryChecksum() (o string) {\n\tif v != nil {\n\t\treturn v.BinaryChecksum\n\t}\n\treturn\n}\n\n// GetQueryResults is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedRequest) GetQueryResults() (o map[string]*WorkflowQueryResult) {\n\tif v != nil && v.QueryResults != nil {\n\t\treturn v.QueryResults\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondDecisionTaskCompletedRequest) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondDecisionTaskCompletedResponse is an internal type (TBD...)\ntype RespondDecisionTaskCompletedResponse struct {\n\tDecisionTask                *PollForDecisionTaskResponse          `json:\"decisionTask,omitempty\"`\n\tActivitiesToDispatchLocally map[string]*ActivityLocalDispatchInfo `json:\"activitiesToDispatchLocally,omitempty\"`\n}\n\n// GetDecisionTask is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskCompletedResponse) GetDecisionTask() (o *PollForDecisionTaskResponse) {\n\tif v != nil && v.DecisionTask != nil {\n\t\treturn v.DecisionTask\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *RespondDecisionTaskCompletedResponse) ByteSize() uint64 {\n\treturn 0\n}\n\n// RespondDecisionTaskFailedRequest is an internal type (TBD...)\ntype RespondDecisionTaskFailedRequest struct {\n\tTaskToken      []byte                   `json:\"taskToken,omitempty\"`\n\tCause          *DecisionTaskFailedCause `json:\"cause,omitempty\"`\n\tDetails        []byte                   `json:\"details,omitempty\"`\n\tIdentity       string                   `json:\"identity,omitempty\"`\n\tBinaryChecksum string                   `json:\"binaryChecksum,omitempty\"`\n}\n\n// GetCause is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskFailedRequest) GetCause() (o DecisionTaskFailedCause) {\n\tif v != nil && v.Cause != nil {\n\t\treturn *v.Cause\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskFailedRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetBinaryChecksum is an internal getter (TBD...)\nfunc (v *RespondDecisionTaskFailedRequest) GetBinaryChecksum() (o string) {\n\tif v != nil {\n\t\treturn v.BinaryChecksum\n\t}\n\treturn\n}\n\n// RespondQueryTaskCompletedRequest is an internal type (TBD...)\ntype RespondQueryTaskCompletedRequest struct {\n\tTaskToken         []byte                  `json:\"taskToken,omitempty\"`\n\tCompletedType     *QueryTaskCompletedType `json:\"completedType,omitempty\"`\n\tQueryResult       []byte                  `json:\"queryResult,omitempty\"`\n\tErrorMessage      string                  `json:\"errorMessage,omitempty\"`\n\tWorkerVersionInfo *WorkerVersionInfo      `json:\"workerVersionInfo,omitempty\"`\n}\n\n// GetCompletedType is an internal getter (TBD...)\nfunc (v *RespondQueryTaskCompletedRequest) GetCompletedType() (o QueryTaskCompletedType) {\n\tif v != nil && v.CompletedType != nil {\n\t\treturn *v.CompletedType\n\t}\n\treturn\n}\n\n// GetQueryResult is an internal getter (TBD...)\nfunc (v *RespondQueryTaskCompletedRequest) GetQueryResult() (o []byte) {\n\tif v != nil && v.QueryResult != nil {\n\t\treturn v.QueryResult\n\t}\n\treturn\n}\n\n// GetErrorMessage is an internal getter (TBD...)\nfunc (v *RespondQueryTaskCompletedRequest) GetErrorMessage() (o string) {\n\tif v != nil {\n\t\treturn v.ErrorMessage\n\t}\n\treturn\n}\n\n// GetWorkerVersionInfo is an internal getter (TBD...)\nfunc (v *RespondQueryTaskCompletedRequest) GetWorkerVersionInfo() (o *WorkerVersionInfo) {\n\tif v != nil && v.WorkerVersionInfo != nil {\n\t\treturn v.WorkerVersionInfo\n\t}\n\treturn\n}\n\n// RetryPolicy is an internal type (TBD...)\ntype RetryPolicy struct {\n\tInitialIntervalInSeconds    int32    `json:\"initialIntervalInSeconds,omitempty\"`\n\tBackoffCoefficient          float64  `json:\"backoffCoefficient,omitempty\"`\n\tMaximumIntervalInSeconds    int32    `json:\"maximumIntervalInSeconds,omitempty\"`\n\tMaximumAttempts             int32    `json:\"maximumAttempts,omitempty\"`\n\tNonRetriableErrorReasons    []string `json:\"nonRetriableErrorReasons,omitempty\"`\n\tExpirationIntervalInSeconds int32    `json:\"expirationIntervalInSeconds,omitempty\"`\n}\n\n// GetInitialIntervalInSeconds is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetInitialIntervalInSeconds() (o int32) {\n\tif v != nil {\n\t\treturn v.InitialIntervalInSeconds\n\t}\n\treturn\n}\n\n// GetBackoffCoefficient is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetBackoffCoefficient() (o float64) {\n\tif v != nil {\n\t\treturn v.BackoffCoefficient\n\t}\n\treturn\n}\n\n// GetMaximumIntervalInSeconds is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetMaximumIntervalInSeconds() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumIntervalInSeconds\n\t}\n\treturn\n}\n\n// GetMaximumAttempts is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetMaximumAttempts() (o int32) {\n\tif v != nil {\n\t\treturn v.MaximumAttempts\n\t}\n\treturn\n}\n\n// GetNonRetriableErrorReasons is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetNonRetriableErrorReasons() (o []string) {\n\tif v != nil && v.NonRetriableErrorReasons != nil {\n\t\treturn v.NonRetriableErrorReasons\n\t}\n\treturn\n}\n\n// GetExpirationIntervalInSeconds is an internal getter (TBD...)\nfunc (v *RetryPolicy) GetExpirationIntervalInSeconds() (o int32) {\n\tif v != nil {\n\t\treturn v.ExpirationIntervalInSeconds\n\t}\n\treturn\n}\n\n// RetryTaskV2Error is an internal type (TBD...)\ntype RetryTaskV2Error struct {\n\tMessage           string `json:\"message,required\"`\n\tDomainID          string `json:\"domainId,omitempty\"`\n\tWorkflowID        string `json:\"workflowId,omitempty\"`\n\tRunID             string `json:\"runId,omitempty\"`\n\tStartEventID      *int64 `json:\"startEventId,omitempty\"`\n\tStartEventVersion *int64 `json:\"startEventVersion,omitempty\"`\n\tEndEventID        *int64 `json:\"endEventId,omitempty\"`\n\tEndEventVersion   *int64 `json:\"endEventVersion,omitempty\"`\n}\n\n// GetDomainID is an internal getter (TBD...)\nfunc (v *RetryTaskV2Error) GetDomainID() (o string) {\n\tif v != nil {\n\t\treturn v.DomainID\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *RetryTaskV2Error) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RetryTaskV2Error) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// ScheduleActivityTaskDecisionAttributes is an internal type (TBD...)\ntype ScheduleActivityTaskDecisionAttributes struct {\n\tActivityID                    string        `json:\"activityId,omitempty\"`\n\tActivityType                  *ActivityType `json:\"activityType,omitempty\"`\n\tDomain                        string        `json:\"domain,omitempty\"`\n\tTaskList                      *TaskList     `json:\"taskList,omitempty\"`\n\tInput                         []byte        `json:\"input,omitempty\"`\n\tScheduleToCloseTimeoutSeconds *int32        `json:\"scheduleToCloseTimeoutSeconds,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32        `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n\tStartToCloseTimeoutSeconds    *int32        `json:\"startToCloseTimeoutSeconds,omitempty\"`\n\tHeartbeatTimeoutSeconds       *int32        `json:\"heartbeatTimeoutSeconds,omitempty\"`\n\tRetryPolicy                   *RetryPolicy  `json:\"retryPolicy,omitempty\"`\n\tHeader                        *Header       `json:\"header,omitempty\"`\n\tRequestLocalDispatch          bool          `json:\"requestLocalDispatch,omitempty\"`\n}\n\n// GetActivityID is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetActivityID() (o string) {\n\tif v != nil {\n\t\treturn v.ActivityID\n\t}\n\treturn\n}\n\n// GetActivityType is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetActivityType() (o *ActivityType) {\n\tif v != nil && v.ActivityType != nil {\n\t\treturn v.ActivityType\n\t}\n\treturn\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetScheduleToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetScheduleToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToCloseTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetScheduleToStartTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.StartToCloseTimeoutSeconds != nil {\n\t\treturn *v.StartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetHeartbeatTimeoutSeconds is an internal getter (TBD...)\nfunc (v *ScheduleActivityTaskDecisionAttributes) GetHeartbeatTimeoutSeconds() (o int32) {\n\tif v != nil && v.HeartbeatTimeoutSeconds != nil {\n\t\treturn *v.HeartbeatTimeoutSeconds\n\t}\n\treturn\n}\n\n// SearchAttributes is an internal type (TBD...)\ntype SearchAttributes struct {\n\tIndexedFields map[string][]byte `json:\"indexedFields,omitempty\"`\n}\n\n// GetIndexedFields is an internal getter (TBD...)\nfunc (v *SearchAttributes) GetIndexedFields() (o map[string][]byte) {\n\tif v != nil && v.IndexedFields != nil {\n\t\treturn v.IndexedFields\n\t}\n\treturn\n}\n\n// ServiceBusyError is an internal type (TBD...)\ntype ServiceBusyError struct {\n\tMessage string `json:\"message,required\"`\n\tReason  string `json:\"reason,omitempty\"`\n}\n\n// SignalExternalWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype SignalExternalWorkflowExecutionDecisionAttributes struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tExecution         *WorkflowExecution `json:\"execution,omitempty\"`\n\tSignalName        string             `json:\"signalName,omitempty\"`\n\tInput             []byte             `json:\"input,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly bool               `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetSignalName is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionDecisionAttributes) GetSignalName() (o string) {\n\tif v != nil {\n\t\treturn v.SignalName\n\t}\n\treturn\n}\n\n// SignalExternalWorkflowExecutionFailedCause is an internal type (TBD...)\ntype SignalExternalWorkflowExecutionFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e SignalExternalWorkflowExecutionFailedCause) Ptr() *SignalExternalWorkflowExecutionFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of SignalExternalWorkflowExecutionFailedCause.\nfunc (e SignalExternalWorkflowExecutionFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\"\n\tcase 1:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\t}\n\treturn fmt.Sprintf(\"SignalExternalWorkflowExecutionFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *SignalExternalWorkflowExecutionFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"UNKNOWN_EXTERNAL_WORKFLOW_EXECUTION\":\n\t\t*e = SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*e = SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"SignalExternalWorkflowExecutionFailedCause\", err)\n\t\t}\n\t\t*e = SignalExternalWorkflowExecutionFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes SignalExternalWorkflowExecutionFailedCause to text.\nfunc (e SignalExternalWorkflowExecutionFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution is an option for SignalExternalWorkflowExecutionFailedCause\n\tSignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution SignalExternalWorkflowExecutionFailedCause = iota\n\tSignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted\n)\n\n// SignalExternalWorkflowExecutionFailedEventAttributes is an internal type (TBD...)\ntype SignalExternalWorkflowExecutionFailedEventAttributes struct {\n\tCause                        *SignalExternalWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tDecisionTaskCompletedEventID int64                                       `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       string                                      `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution                          `json:\"workflowExecution,omitempty\"`\n\tInitiatedEventID             int64                                       `json:\"initiatedEventId,omitempty\"`\n\tControl                      []byte                                      `json:\"control,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *SignalExternalWorkflowExecutionFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// SignalExternalWorkflowExecutionInitiatedEventAttributes is an internal type (TBD...)\ntype SignalExternalWorkflowExecutionInitiatedEventAttributes struct {\n\tDecisionTaskCompletedEventID int64              `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDomain                       string             `json:\"domain,omitempty\"`\n\tWorkflowExecution            *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tSignalName                   string             `json:\"signalName,omitempty\"`\n\tInput                        []byte             `json:\"input,omitempty\"`\n\tControl                      []byte             `json:\"control,omitempty\"`\n\tChildWorkflowOnly            bool               `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetSignalName is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetSignalName() (o string) {\n\tif v != nil {\n\t\treturn v.SignalName\n\t}\n\treturn\n}\n\n// GetChildWorkflowOnly is an internal getter (TBD...)\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) GetChildWorkflowOnly() (o bool) {\n\tif v != nil {\n\t\treturn v.ChildWorkflowOnly\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *SignalExternalWorkflowExecutionInitiatedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// SignalWithStartWorkflowExecutionRequest is an internal type (TBD...)\ntype SignalWithStartWorkflowExecutionRequest struct {\n\tDomain                              string                        `json:\"domain,omitempty\"`\n\tWorkflowID                          string                        `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"-\"` // Filtering PII\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tIdentity                            string                        `json:\"identity,omitempty\"`\n\tRequestID                           string                        `json:\"requestId,omitempty\"`\n\tWorkflowIDReusePolicy               *WorkflowIDReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tSignalName                          string                        `json:\"signalName,omitempty\"`\n\tSignalInput                         []byte                        `json:\"-\"` // Filtering PII\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tMemo                                *Memo                         `json:\"-\"` // Filtering PII\n\tSearchAttributes                    *SearchAttributes             `json:\"-\"` // Filtering PII\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimestamp                 *int64                        `json:\"firstRunAtTimestamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetWorkflowType is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\treturn\n}\n\n// GetTaskList is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetTaskList() (o *TaskList) {\n\tif v != nil && v.TaskList != nil {\n\t\treturn v.TaskList\n\t}\n\treturn\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetTaskStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// GetWorkflowIDReusePolicy is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetWorkflowIDReusePolicy() (o WorkflowIDReusePolicy) {\n\tif v != nil && v.WorkflowIDReusePolicy != nil {\n\t\treturn *v.WorkflowIDReusePolicy\n\t}\n\treturn\n}\n\n// GetSignalName is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSignalName() (o string) {\n\tif v != nil {\n\t\treturn v.SignalName\n\t}\n\treturn\n}\n\n// GetSignalInput is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetSignalInput() (o []byte) {\n\tif v != nil && v.SignalInput != nil {\n\t\treturn v.SignalInput\n\t}\n\treturn\n}\n\n// GetCronSchedule is an internal getter (TBD...)\nfunc (v *SignalWithStartWorkflowExecutionRequest) GetCronSchedule() (o string) {\n\tif v != nil {\n\t\treturn v.CronSchedule\n\t}\n\treturn\n}\n\n// SignalWithStartWorkflowExecutionAsyncRequest is an internal type (TBD...)\ntype SignalWithStartWorkflowExecutionAsyncRequest struct {\n\t*SignalWithStartWorkflowExecutionRequest\n}\n\n// SignalWithStartWorkflowExecutionAsyncResponse is an internal type (TBD...)\ntype SignalWithStartWorkflowExecutionAsyncResponse struct {\n}\n\n// SignalWorkflowExecutionRequest is an internal type (TBD...)\ntype SignalWorkflowExecutionRequest struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tSignalName        string             `json:\"signalName,omitempty\"`\n\tInput             []byte             `json:\"-\"` // Filtering PII\n\tIdentity          string             `json:\"identity,omitempty\"`\n\tRequestID         string             `json:\"requestId,omitempty\"`\n\tControl           []byte             `json:\"control,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetSignalName is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetSignalName() (o string) {\n\tif v != nil {\n\t\treturn v.SignalName\n\t}\n\treturn\n}\n\n// GetInput is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *SignalWorkflowExecutionRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// StartChildWorkflowExecutionDecisionAttributes is an internal type (TBD...)\ntype StartChildWorkflowExecutionDecisionAttributes struct {\n\tDomain                              string                        `json:\"domain,omitempty\"`\n\tWorkflowID                          string                        `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tParentClosePolicy                   *ParentClosePolicy            `json:\"parentClosePolicy,omitempty\"`\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tWorkflowIDReusePolicy               *WorkflowIDReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetTaskStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetParentClosePolicy is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\treturn\n}\n\n// GetCronSchedule is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionDecisionAttributes) GetCronSchedule() (o string) {\n\tif v != nil {\n\t\treturn v.CronSchedule\n\t}\n\treturn\n}\n\n// StartChildWorkflowExecutionFailedEventAttributes is an internal type (TBD...)\ntype StartChildWorkflowExecutionFailedEventAttributes struct {\n\tDomain                       string                             `json:\"domain,omitempty\"`\n\tWorkflowID                   string                             `json:\"workflowId,omitempty\"`\n\tWorkflowType                 *WorkflowType                      `json:\"workflowType,omitempty\"`\n\tCause                        *ChildWorkflowExecutionFailedCause `json:\"cause,omitempty\"`\n\tControl                      []byte                             `json:\"control,omitempty\"`\n\tInitiatedEventID             int64                              `json:\"initiatedEventId,omitempty\"`\n\tDecisionTaskCompletedEventID int64                              `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetInitiatedEventID is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) GetInitiatedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.InitiatedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *StartChildWorkflowExecutionFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// StartChildWorkflowExecutionInitiatedEventAttributes is an internal type (TBD...)\ntype StartChildWorkflowExecutionInitiatedEventAttributes struct {\n\tDomain                              string                        `json:\"domain,omitempty\"`\n\tWorkflowID                          string                        `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tParentClosePolicy                   *ParentClosePolicy            `json:\"parentClosePolicy,omitempty\"`\n\tControl                             []byte                        `json:\"control,omitempty\"`\n\tDecisionTaskCompletedEventID        int64                         `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tWorkflowIDReusePolicy               *WorkflowIDReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimestamp                 *int64                        `json:\"firstRunAtTimestamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetWorkflowType is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetWorkflowType() (o *WorkflowType) {\n\tif v != nil && v.WorkflowType != nil {\n\t\treturn v.WorkflowType\n\t}\n\treturn\n}\n\n// GetParentClosePolicy is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetParentClosePolicy() (o ParentClosePolicy) {\n\tif v != nil && v.ParentClosePolicy != nil {\n\t\treturn *v.ParentClosePolicy\n\t}\n\treturn\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *StartChildWorkflowExecutionInitiatedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// StartTimeFilter is an internal type (TBD...)\ntype StartTimeFilter struct {\n\tEarliestTime *int64 `json:\"earliestTime,omitempty\"`\n\tLatestTime   *int64 `json:\"latestTime,omitempty\"`\n}\n\n// GetEarliestTime is an internal getter (TBD...)\nfunc (v *StartTimeFilter) GetEarliestTime() (o int64) {\n\tif v != nil && v.EarliestTime != nil {\n\t\treturn *v.EarliestTime\n\t}\n\treturn\n}\n\n// GetLatestTime is an internal getter (TBD...)\nfunc (v *StartTimeFilter) GetLatestTime() (o int64) {\n\tif v != nil && v.LatestTime != nil {\n\t\treturn *v.LatestTime\n\t}\n\treturn\n}\n\n// StartTimerDecisionAttributes is an internal type (TBD...)\ntype StartTimerDecisionAttributes struct {\n\tTimerID                   string `json:\"timerId,omitempty\"`\n\tStartToFireTimeoutSeconds *int64 `json:\"startToFireTimeoutSeconds,omitempty\"`\n}\n\n// GetTimerID is an internal getter (TBD...)\nfunc (v *StartTimerDecisionAttributes) GetTimerID() (o string) {\n\tif v != nil {\n\t\treturn v.TimerID\n\t}\n\treturn\n}\n\n// GetStartToFireTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartTimerDecisionAttributes) GetStartToFireTimeoutSeconds() (o int64) {\n\tif v != nil && v.StartToFireTimeoutSeconds != nil {\n\t\treturn *v.StartToFireTimeoutSeconds\n\t}\n\treturn\n}\n\ntype RestartWorkflowExecutionRequest struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIdentity          string             `json:\"identity,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *RestartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *RestartWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\ntype DiagnoseWorkflowExecutionRequest struct {\n\tDomain            string             `json:\"domain,omitempty\"`\n\tWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tIdentity          string             `json:\"identity,omitempty\"`\n}\n\n// GetDomain returns the domain\nfunc (v *DiagnoseWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution returns the workflow execution\nfunc (v *DiagnoseWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetIdentity returns the identity\nfunc (v *DiagnoseWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\ntype DiagnoseWorkflowExecutionResponse struct {\n\tDomain                      string             `json:\"domain,omitempty\"`\n\tDiagnosticWorkflowExecution *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n}\n\n// GetDomain returns the fomain\nfunc (v *DiagnoseWorkflowExecutionResponse) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetDiagnosticWorkflowExecution returns the workflow execution\nfunc (v *DiagnoseWorkflowExecutionResponse) GetDiagnosticWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.DiagnosticWorkflowExecution != nil {\n\t\treturn v.DiagnosticWorkflowExecution\n\t}\n\treturn\n}\n\n// StartWorkflowExecutionRequest is an internal type (TBD...)\ntype StartWorkflowExecutionRequest struct {\n\tDomain                              string                        `json:\"domain,omitempty\"`\n\tWorkflowID                          string                        `json:\"workflowId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"-\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tIdentity                            string                        `json:\"identity,omitempty\"`\n\tRequestID                           string                        `json:\"requestId,omitempty\"`\n\tWorkflowIDReusePolicy               *WorkflowIDReusePolicy        `json:\"workflowIdReusePolicy,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tMemo                                *Memo                         `json:\"-\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"-\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tDelayStartSeconds                   *int32                        `json:\"delayStartSeconds,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tFirstRunAtTimeStamp                 *int64                        `json:\"firstRunAtTimeStamp,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetTaskStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetDelayStartSeconds is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetDelayStartSeconds() (o int32) {\n\tif v != nil && v.DelayStartSeconds != nil {\n\t\treturn *v.DelayStartSeconds\n\t}\n\treturn\n}\n\n// GetJitterStartSeconds is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\treturn\n}\n\n// GetFirstRunAtTimeStamp is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetFirstRunAtTimeStamp() (o int64) {\n\tif v != nil && v.FirstRunAtTimeStamp != nil {\n\t\treturn *v.FirstRunAtTimeStamp\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// GetWorkflowIDReusePolicy is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetWorkflowIDReusePolicy() (o WorkflowIDReusePolicy) {\n\tif v != nil && v.WorkflowIDReusePolicy != nil {\n\t\treturn *v.WorkflowIDReusePolicy\n\t}\n\treturn\n}\n\n// GetCronSchedule is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetCronSchedule() (o string) {\n\tif v != nil {\n\t\treturn v.CronSchedule\n\t}\n\treturn\n}\n\n// GetCronOverlapPolicy is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionRequest) GetCronOverlapPolicy() (o CronOverlapPolicy) {\n\tif v != nil && v.CronOverlapPolicy != nil {\n\t\treturn *v.CronOverlapPolicy\n\t}\n\treturn\n}\n\n// StartWorkflowExecutionResponse is an internal type (TBD...)\ntype StartWorkflowExecutionResponse struct {\n\tRunID string `json:\"runId,omitempty\"`\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *StartWorkflowExecutionResponse) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\ntype StartWorkflowExecutionAsyncRequest struct {\n\t*StartWorkflowExecutionRequest\n}\n\ntype StartWorkflowExecutionAsyncResponse struct {\n}\n\n// RestartWorkflowExecutionResponse is an internal type (TBD...)\ntype RestartWorkflowExecutionResponse struct {\n\tRunID string `json:\"runId,omitempty\"`\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *RestartWorkflowExecutionResponse) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// StickyExecutionAttributes is an internal type (TBD...)\ntype StickyExecutionAttributes struct {\n\tWorkerTaskList                *TaskList `json:\"workerTaskList,omitempty\"`\n\tScheduleToStartTimeoutSeconds *int32    `json:\"scheduleToStartTimeoutSeconds,omitempty\"`\n}\n\n// GetScheduleToStartTimeoutSeconds is an internal getter (TBD...)\nfunc (v *StickyExecutionAttributes) GetScheduleToStartTimeoutSeconds() (o int32) {\n\tif v != nil && v.ScheduleToStartTimeoutSeconds != nil {\n\t\treturn *v.ScheduleToStartTimeoutSeconds\n\t}\n\treturn\n}\n\n// SupportedClientVersions is an internal type (TBD...)\ntype SupportedClientVersions struct {\n\tGoSdk   string `json:\"goSdk,omitempty\"`\n\tJavaSdk string `json:\"javaSdk,omitempty\"`\n}\n\n// TaskIDBlock is an internal type (TBD...)\ntype TaskIDBlock struct {\n\tStartID int64 `json:\"startID,omitempty\"`\n\tEndID   int64 `json:\"endID,omitempty\"`\n}\n\n// GetStartID is an internal getter (TBD...)\nfunc (v *TaskIDBlock) GetStartID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartID\n\t}\n\treturn\n}\n\n// GetEndID is an internal getter (TBD...)\nfunc (v *TaskIDBlock) GetEndID() (o int64) {\n\tif v != nil {\n\t\treturn v.EndID\n\t}\n\treturn\n}\n\n// TaskList is an internal type (TBD...)\ntype TaskList struct {\n\tName string        `json:\"name,omitempty\"`\n\tKind *TaskListKind `json:\"kind,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *TaskList) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// GetKind is an internal getter (TBD...)\nfunc (v *TaskList) GetKind() (o TaskListKind) {\n\tif v != nil && v.Kind != nil {\n\t\treturn *v.Kind\n\t}\n\treturn\n}\n\n// TaskListKind is an internal type (TBD...)\ntype TaskListKind int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e TaskListKind) Ptr() *TaskListKind {\n\treturn &e\n}\n\n// String returns a readable string representation of TaskListKind.\nfunc (e TaskListKind) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"NORMAL\"\n\tcase 1:\n\t\treturn \"STICKY\"\n\tcase 2:\n\t\treturn \"EPHEMERAL\"\n\t}\n\treturn fmt.Sprintf(\"TaskListKind(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *TaskListKind) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"NORMAL\":\n\t\t*e = TaskListKindNormal\n\t\treturn nil\n\tcase \"STICKY\":\n\t\t*e = TaskListKindSticky\n\t\treturn nil\n\tcase \"EPHEMERAL\":\n\t\t*e = TaskListKindEphemeral\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskListKind\", err)\n\t\t}\n\t\t*e = TaskListKind(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskListKind to text.\nfunc (e TaskListKind) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// TaskListKindNormal is an option for TaskListKind\n\tTaskListKindNormal TaskListKind = iota\n\t// TaskListKindSticky is an option for TaskListKind\n\tTaskListKindSticky\n\t// TaskListKindEphemeral is an option for TaskListKind\n\tTaskListKindEphemeral\n)\n\n// TaskListMetadata is an internal type (TBD...)\ntype TaskListMetadata struct {\n\tMaxTasksPerSecond *float64 `json:\"maxTasksPerSecond,omitempty\"`\n}\n\n// TaskListPartitionMetadata is an internal type (TBD...)\ntype TaskListPartitionMetadata struct {\n\tKey           string `json:\"key,omitempty\"`\n\tOwnerHostName string `json:\"ownerHostName,omitempty\"`\n}\n\n// GetKey is an internal getter (TBD...)\nfunc (v *TaskListPartitionMetadata) GetKey() (o string) {\n\tif v != nil {\n\t\treturn v.Key\n\t}\n\treturn\n}\n\n// GetOwnerHostName is an internal getter (TBD...)\nfunc (v *TaskListPartitionMetadata) GetOwnerHostName() (o string) {\n\tif v != nil {\n\t\treturn v.OwnerHostName\n\t}\n\treturn\n}\n\ntype IsolationGroupMetrics struct {\n\tNewTasksPerSecond float64 `json:\"newTasksPerSecond,omitempty\"`\n\tPollerCount       int64   `json:\"pollerCount,omitempty\"`\n}\n\n// TaskListStatus is an internal type (TBD...)\ntype TaskListStatus struct {\n\tBacklogCountHint      int64                             `json:\"backlogCountHint,omitempty\"`\n\tReadLevel             int64                             `json:\"readLevel,omitempty\"`\n\tAckLevel              int64                             `json:\"ackLevel,omitempty\"`\n\tRatePerSecond         float64                           `json:\"ratePerSecond,omitempty\"`\n\tTaskIDBlock           *TaskIDBlock                      `json:\"taskIDBlock,omitempty\"`\n\tIsolationGroupMetrics map[string]*IsolationGroupMetrics `json:\"isolationGroupMetrics,omitempty\"`\n\tNewTasksPerSecond     float64                           `json:\"newTasksPerSecond,omitempty\"`\n\tEmpty                 bool                              `json:\"empty,omitempty\"`\n}\n\n// GetBacklogCountHint is an internal getter (TBD...)\nfunc (v *TaskListStatus) GetBacklogCountHint() (o int64) {\n\tif v != nil {\n\t\treturn v.BacklogCountHint\n\t}\n\treturn\n}\n\n// GetReadLevel is an internal getter (TBD...)\nfunc (v *TaskListStatus) GetReadLevel() (o int64) {\n\tif v != nil {\n\t\treturn v.ReadLevel\n\t}\n\treturn\n}\n\n// GetAckLevel is an internal getter (TBD...)\nfunc (v *TaskListStatus) GetAckLevel() (o int64) {\n\tif v != nil {\n\t\treturn v.AckLevel\n\t}\n\treturn\n}\n\n// GetRatePerSecond is an internal getter (TBD...)\nfunc (v *TaskListStatus) GetRatePerSecond() (o float64) {\n\tif v != nil {\n\t\treturn v.RatePerSecond\n\t}\n\treturn\n}\n\n// GetTaskIDBlock is an internal getter (TBD...)\nfunc (v *TaskListStatus) GetTaskIDBlock() (o *TaskIDBlock) {\n\tif v != nil && v.TaskIDBlock != nil {\n\t\treturn v.TaskIDBlock\n\t}\n\treturn\n}\n\n// TaskListType is an internal type (TBD...)\ntype TaskListType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e TaskListType) Ptr() *TaskListType {\n\treturn &e\n}\n\n// String returns a readable string representation of TaskListType.\nfunc (e TaskListType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"Decision\"\n\tcase 1:\n\t\treturn \"Activity\"\n\t}\n\treturn fmt.Sprintf(\"TaskListType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *TaskListType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"DECISION\":\n\t\t*e = TaskListTypeDecision\n\t\treturn nil\n\tcase \"ACTIVITY\":\n\t\t*e = TaskListTypeActivity\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TaskListType\", err)\n\t\t}\n\t\t*e = TaskListType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TaskListType to text.\nfunc (e TaskListType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// TaskListTypeDecision is an option for TaskListType\n\tTaskListTypeDecision TaskListType = iota\n\t// TaskListTypeActivity is an option for TaskListType\n\tTaskListTypeActivity\n)\n\n// TerminateWorkflowExecutionRequest is an internal type (TBD...)\ntype TerminateWorkflowExecutionRequest struct {\n\tDomain              string             `json:\"domain,omitempty\"`\n\tWorkflowExecution   *WorkflowExecution `json:\"workflowExecution,omitempty\"`\n\tReason              string             `json:\"reason,omitempty\"`\n\tDetails             []byte             `json:\"details,omitempty\"`\n\tIdentity            string             `json:\"identity,omitempty\"`\n\tFirstExecutionRunID string             `json:\"first_execution_run_id,omitempty\"`\n}\n\n// GetDomain is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetDomain() (o string) {\n\tif v != nil {\n\t\treturn v.Domain\n\t}\n\treturn\n}\n\n// GetWorkflowExecution is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetWorkflowExecution() (o *WorkflowExecution) {\n\tif v != nil && v.WorkflowExecution != nil {\n\t\treturn v.WorkflowExecution\n\t}\n\treturn\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\n// GetDetails is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetDetails() (o []byte) {\n\tif v != nil && v.Details != nil {\n\t\treturn v.Details\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetFirstExecutionRunID is an internal getter (TBD...)\nfunc (v *TerminateWorkflowExecutionRequest) GetFirstExecutionRunID() (o string) {\n\tif v != nil {\n\t\treturn v.FirstExecutionRunID\n\t}\n\treturn\n}\n\n// TimeoutType is an internal type (TBD...)\ntype TimeoutType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e TimeoutType) Ptr() *TimeoutType {\n\treturn &e\n}\n\n// String returns a readable string representation of TimeoutType.\nfunc (e TimeoutType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"START_TO_CLOSE\"\n\tcase 1:\n\t\treturn \"SCHEDULE_TO_START\"\n\tcase 2:\n\t\treturn \"SCHEDULE_TO_CLOSE\"\n\tcase 3:\n\t\treturn \"HEARTBEAT\"\n\t}\n\treturn fmt.Sprintf(\"TimeoutType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *TimeoutType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"START_TO_CLOSE\":\n\t\t*e = TimeoutTypeStartToClose\n\t\treturn nil\n\tcase \"SCHEDULE_TO_START\":\n\t\t*e = TimeoutTypeScheduleToStart\n\t\treturn nil\n\tcase \"SCHEDULE_TO_CLOSE\":\n\t\t*e = TimeoutTypeScheduleToClose\n\t\treturn nil\n\tcase \"HEARTBEAT\":\n\t\t*e = TimeoutTypeHeartbeat\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"TimeoutType\", err)\n\t\t}\n\t\t*e = TimeoutType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes TimeoutType to text.\nfunc (e TimeoutType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// TimeoutTypeStartToClose is an option for TimeoutType\n\tTimeoutTypeStartToClose TimeoutType = iota\n\t// TimeoutTypeScheduleToStart is an option for TimeoutType\n\tTimeoutTypeScheduleToStart\n\t// TimeoutTypeScheduleToClose is an option for TimeoutType\n\tTimeoutTypeScheduleToClose\n\t// TimeoutTypeHeartbeat is an option for TimeoutType\n\tTimeoutTypeHeartbeat\n)\n\n// TimerCanceledEventAttributes is an internal type (TBD...)\ntype TimerCanceledEventAttributes struct {\n\tTimerID                      string `json:\"timerId,omitempty\"`\n\tStartedEventID               int64  `json:\"startedEventId,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tIdentity                     string `json:\"identity,omitempty\"`\n}\n\n// GetTimerID is an internal getter (TBD...)\nfunc (v *TimerCanceledEventAttributes) GetTimerID() (o string) {\n\tif v != nil {\n\t\treturn v.TimerID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *TimerCanceledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// TimerFiredEventAttributes is an internal type (TBD...)\ntype TimerFiredEventAttributes struct {\n\tTimerID        string `json:\"timerId,omitempty\"`\n\tStartedEventID int64  `json:\"startedEventId,omitempty\"`\n}\n\n// GetTimerID is an internal getter (TBD...)\nfunc (v *TimerFiredEventAttributes) GetTimerID() (o string) {\n\tif v != nil {\n\t\treturn v.TimerID\n\t}\n\treturn\n}\n\n// GetStartedEventID is an internal getter (TBD...)\nfunc (v *TimerFiredEventAttributes) GetStartedEventID() (o int64) {\n\tif v != nil {\n\t\treturn v.StartedEventID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *TimerFiredEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// TimerStartedEventAttributes is an internal type (TBD...)\ntype TimerStartedEventAttributes struct {\n\tTimerID                      string `json:\"timerId,omitempty\"`\n\tStartToFireTimeoutSeconds    *int64 `json:\"startToFireTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// GetTimerID is an internal getter (TBD...)\nfunc (v *TimerStartedEventAttributes) GetTimerID() (o string) {\n\tif v != nil {\n\t\treturn v.TimerID\n\t}\n\treturn\n}\n\n// GetStartToFireTimeoutSeconds is an internal getter (TBD...)\nfunc (v *TimerStartedEventAttributes) GetStartToFireTimeoutSeconds() (o int64) {\n\tif v != nil && v.StartToFireTimeoutSeconds != nil {\n\t\treturn *v.StartToFireTimeoutSeconds\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *TimerStartedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// TransientDecisionInfo is an internal type (TBD...)\ntype TransientDecisionInfo struct {\n\tScheduledEvent *HistoryEvent `json:\"scheduledEvent,omitempty\"`\n\tStartedEvent   *HistoryEvent `json:\"startedEvent,omitempty\"`\n}\n\n// UpdateDomainRequest is an internal type (TBD...)\ntype UpdateDomainRequest struct {\n\tName                                   string                             `json:\"name,omitempty\"`\n\tDescription                            *string                            `json:\"description,omitempty\"`\n\tOwnerEmail                             *string                            `json:\"ownerEmail,omitempty\"`\n\tData                                   map[string]string                  `json:\"data,omitempty\"`\n\tWorkflowExecutionRetentionPeriodInDays *int32                             `json:\"workflowExecutionRetentionPeriodInDays,omitempty\"`\n\tEmitMetric                             *bool                              `json:\"emitMetric,omitempty\"`\n\tBadBinaries                            *BadBinaries                       `json:\"badBinaries,omitempty\"`\n\tHistoryArchivalStatus                  *ArchivalStatus                    `json:\"historyArchivalStatus,omitempty\"`\n\tHistoryArchivalURI                     *string                            `json:\"historyArchivalURI,omitempty\"`\n\tVisibilityArchivalStatus               *ArchivalStatus                    `json:\"visibilityArchivalStatus,omitempty\"`\n\tVisibilityArchivalURI                  *string                            `json:\"visibilityArchivalURI,omitempty\"`\n\tActiveClusterName                      *string                            `json:\"activeClusterName,omitempty\"`\n\tActiveClusters                         *ActiveClusters                    `json:\"activeClusters,omitempty\"`\n\tClusters                               []*ClusterReplicationConfiguration `json:\"clusters,omitempty\"`\n\tSecurityToken                          string                             `json:\"securityToken,omitempty\"`\n\tDeleteBadBinary                        *string                            `json:\"deleteBadBinary,omitempty\"`\n\tFailoverTimeoutInSeconds               *int32                             `json:\"failoverTimeoutInSeconds,omitempty\"`\n\tFailoverReason                         *string                            `json:\"failoverReason,omitempty\"`\n}\n\n// IsAFailoverRequest identifies if any part of the request is a failover request\n// and if so, will return true\n// this includes:\n//\n// - an active cluster change  (force failver)\n// - any failvoer timeout values (for graceful failover)\n// - or a change to one of the domain's cluster-attribute fields (active-active failover)\n//\n// this is not a validation function\n// and doesn't attempt to give valid or coherent combinations\nfunc (v *UpdateDomainRequest) IsAFailoverRequest() bool {\n\treturn v.ActiveClusterName != nil ||\n\t\t(v.FailoverTimeoutInSeconds != nil && *v.FailoverTimeoutInSeconds > 0) ||\n\t\t(v.ActiveClusters != nil && len(v.ActiveClusters.AttributeScopes) > 0)\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *UpdateDomainRequest) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// GetFailoverTimeoutInSeconds is an internal getter (TBD...)\nfunc (v *UpdateDomainRequest) GetFailoverTimeoutInSeconds() (o int32) {\n\tif v != nil && v.FailoverTimeoutInSeconds != nil {\n\t\treturn *v.FailoverTimeoutInSeconds\n\t}\n\treturn\n}\n\n// GetHistoryArchivalURI is an internal getter (TBD...)\nfunc (v *UpdateDomainRequest) GetHistoryArchivalURI() (o string) {\n\tif v != nil && v.HistoryArchivalURI != nil {\n\t\treturn *v.HistoryArchivalURI\n\t}\n\treturn\n}\n\n// GetVisibilityArchivalURI is an internal getter (TBD...)\nfunc (v *UpdateDomainRequest) GetVisibilityArchivalURI() (o string) {\n\tif v != nil && v.VisibilityArchivalURI != nil {\n\t\treturn *v.VisibilityArchivalURI\n\t}\n\treturn\n}\n\n// UpdateDomainResponse is an internal type (TBD...)\ntype UpdateDomainResponse struct {\n\tDomainInfo               *DomainInfo                     `json:\"domainInfo,omitempty\"`\n\tConfiguration            *DomainConfiguration            `json:\"configuration,omitempty\"`\n\tReplicationConfiguration *DomainReplicationConfiguration `json:\"replicationConfiguration,omitempty\"`\n\tFailoverVersion          int64                           `json:\"failoverVersion,omitempty\"`\n\tIsGlobalDomain           bool                            `json:\"isGlobalDomain,omitempty\"`\n}\n\nfunc (v *UpdateDomainResponse) ToFailoverDomainResponse() *FailoverDomainResponse {\n\tif v == nil {\n\t\treturn nil\n\t}\n\treturn &FailoverDomainResponse{\n\t\tDomainInfo:               v.DomainInfo,\n\t\tConfiguration:            v.Configuration,\n\t\tReplicationConfiguration: v.ReplicationConfiguration,\n\t\tFailoverVersion:          v.FailoverVersion,\n\t\tIsGlobalDomain:           v.IsGlobalDomain,\n\t}\n}\n\n// GetDomainInfo is an internal getter (TBD...)\nfunc (v *UpdateDomainResponse) GetDomainInfo() (o *DomainInfo) {\n\tif v != nil && v.DomainInfo != nil {\n\t\treturn v.DomainInfo\n\t}\n\treturn\n}\n\n// GetFailoverVersion is an internal getter (TBD...)\nfunc (v *UpdateDomainResponse) GetFailoverVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.FailoverVersion\n\t}\n\treturn\n}\n\n// GetIsGlobalDomain is an internal getter (TBD...)\nfunc (v *UpdateDomainResponse) GetIsGlobalDomain() (o bool) {\n\tif v != nil {\n\t\treturn v.IsGlobalDomain\n\t}\n\treturn\n}\n\n// UpsertWorkflowSearchAttributesDecisionAttributes is an internal type (TBD...)\ntype UpsertWorkflowSearchAttributesDecisionAttributes struct {\n\tSearchAttributes *SearchAttributes `json:\"searchAttributes,omitempty\"`\n}\n\n// GetSearchAttributes is an internal getter (TBD...)\nfunc (v *UpsertWorkflowSearchAttributesDecisionAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn\n}\n\n// UpsertWorkflowSearchAttributesEventAttributes is an internal type (TBD...)\ntype UpsertWorkflowSearchAttributesEventAttributes struct {\n\tDecisionTaskCompletedEventID int64             `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tSearchAttributes             *SearchAttributes `json:\"searchAttributes,omitempty\"`\n}\n\n// GetSearchAttributes is an internal getter (TBD...)\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *UpsertWorkflowSearchAttributesEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// VersionHistories is an internal type (TBD...)\ntype VersionHistories struct {\n\tCurrentVersionHistoryIndex int32             `json:\"currentVersionHistoryIndex,omitempty\"`\n\tHistories                  []*VersionHistory `json:\"histories,omitempty\"`\n}\n\n// GetCurrentVersionHistoryIndex is an internal getter (TBD...)\nfunc (v *VersionHistories) GetCurrentVersionHistoryIndex() (o int32) {\n\tif v != nil {\n\t\treturn v.CurrentVersionHistoryIndex\n\t}\n\treturn\n}\n\n// VersionHistory is an internal type (TBD...)\ntype VersionHistory struct {\n\tBranchToken []byte                `json:\"branchToken,omitempty\"`\n\tItems       []*VersionHistoryItem `json:\"items,omitempty\"`\n}\n\n// GetItems is an internal getter (TBD...)\nfunc (v *VersionHistory) GetItems() (o []*VersionHistoryItem) {\n\tif v != nil && v.Items != nil {\n\t\treturn v.Items\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *VersionHistory) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\tsize := uint64(unsafe.Sizeof(*v))\n\tsize += uint64(len(v.BranchToken))\n\n\t// [] *VersionHistoryItem backing array: len * pointer size\n\tsize += uint64(len(v.Items)) * uint64(unsafe.Sizeof((*VersionHistoryItem)(nil)))\n\tfor _, e := range v.Items {\n\t\tsize += e.ByteSize()\n\t}\n\n\treturn size\n}\n\n// VersionHistoryItem is an internal type (TBD...)\ntype VersionHistoryItem struct {\n\tEventID int64 `json:\"eventID,omitempty\"`\n\tVersion int64 `json:\"version,omitempty\"`\n}\n\n// GetVersion is an internal getter (TBD...)\nfunc (v *VersionHistoryItem) GetVersion() (o int64) {\n\tif v != nil {\n\t\treturn v.Version\n\t}\n\treturn\n}\n\n// ByteSize returns the approximate memory used in bytes\nfunc (v *VersionHistoryItem) ByteSize() uint64 {\n\tif v == nil {\n\t\treturn 0\n\t}\n\n\treturn uint64(unsafe.Sizeof(*v))\n}\n\n// WorkerVersionInfo is an internal type (TBD...)\ntype WorkerVersionInfo struct {\n\tImpl           string `json:\"impl,omitempty\"`\n\tFeatureVersion string `json:\"featureVersion,omitempty\"`\n}\n\n// GetImpl is an internal getter (TBD...)\nfunc (v *WorkerVersionInfo) GetImpl() (o string) {\n\tif v != nil {\n\t\treturn v.Impl\n\t}\n\treturn\n}\n\n// GetFeatureVersion is an internal getter (TBD...)\nfunc (v *WorkerVersionInfo) GetFeatureVersion() (o string) {\n\tif v != nil {\n\t\treturn v.FeatureVersion\n\t}\n\treturn\n}\n\n// WorkflowExecution is an internal type (TBD...)\ntype WorkflowExecution struct {\n\tWorkflowID string `json:\"workflowId,omitempty\"`\n\tRunID      string `json:\"runId,omitempty\"`\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *WorkflowExecution) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *WorkflowExecution) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// WorkflowExecutionAlreadyStartedError is an internal type (TBD...)\ntype WorkflowExecutionAlreadyStartedError struct {\n\tMessage        string `json:\"message,omitempty\"`\n\tStartRequestID string `json:\"startRequestId,omitempty\"`\n\tRunID          string `json:\"runId,omitempty\"`\n}\n\n// GetMessage is an internal getter (TBD...)\nfunc (v *WorkflowExecutionAlreadyStartedError) GetMessage() (o string) {\n\tif v != nil {\n\t\treturn v.Message\n\t}\n\treturn\n}\n\n// WorkflowExecutionCancelRequestedEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionCancelRequestedEventAttributes struct {\n\tCause                     string             `json:\"cause,omitempty\"`\n\tExternalInitiatedEventID  *int64             `json:\"externalInitiatedEventId,omitempty\"`\n\tExternalWorkflowExecution *WorkflowExecution `json:\"externalWorkflowExecution,omitempty\"`\n\tIdentity                  string             `json:\"identity,omitempty\"`\n\tRequestID                 string             `json:\"requestId,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionCancelRequestedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionCanceledEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionCanceledEventAttributes struct {\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tDetails                      []byte `json:\"details,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionCanceledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionCloseStatus is an internal type (TBD...)\ntype WorkflowExecutionCloseStatus int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e WorkflowExecutionCloseStatus) Ptr() *WorkflowExecutionCloseStatus {\n\treturn &e\n}\n\n// String returns a readable string representation of WorkflowExecutionCloseStatus.\nfunc (e WorkflowExecutionCloseStatus) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"COMPLETED\"\n\tcase 1:\n\t\treturn \"FAILED\"\n\tcase 2:\n\t\treturn \"CANCELED\"\n\tcase 3:\n\t\treturn \"TERMINATED\"\n\tcase 4:\n\t\treturn \"CONTINUED_AS_NEW\"\n\tcase 5:\n\t\treturn \"TIMED_OUT\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowExecutionCloseStatus(%d)\", w)\n}\n\ntype WorkflowExecutionStatus int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e WorkflowExecutionStatus) Ptr() *WorkflowExecutionStatus {\n\treturn &e\n}\n\n// String returns a readable string representation of WorkflowExecutionStatus.\nfunc (e WorkflowExecutionStatus) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"PENDING\"\n\tcase 1:\n\t\treturn \"STARTED\"\n\tcase 2:\n\t\treturn \"COMPLETED\"\n\tcase 3:\n\t\treturn \"FAILED\"\n\tcase 4:\n\t\treturn \"CANCELED\"\n\tcase 5:\n\t\treturn \"TERMINATED\"\n\tcase 6:\n\t\treturn \"CONTINUED_AS_NEW\"\n\tcase 7:\n\t\treturn \"TIMED_OUT\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowExecutionStatus(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *WorkflowExecutionStatus) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"PENDING\":\n\t\t*e = WorkflowExecutionStatusPending\n\t\treturn nil\n\tcase \"STARTED\":\n\t\t*e = WorkflowExecutionStatusStarted\n\t\treturn nil\n\tcase \"COMPLETED\":\n\t\t*e = WorkflowExecutionStatusCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*e = WorkflowExecutionStatusFailed\n\t\treturn nil\n\tcase \"CANCELED\":\n\t\t*e = WorkflowExecutionStatusCanceled\n\t\treturn nil\n\tcase \"TERMINATED\":\n\t\t*e = WorkflowExecutionStatusTerminated\n\t\treturn nil\n\tcase \"CONTINUED_AS_NEW\":\n\t\t*e = WorkflowExecutionStatusContinuedAsNew\n\t\treturn nil\n\tcase \"TIMED_OUT\":\n\t\t*e = WorkflowExecutionStatusTimedOut\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowExecutionStatus\", err)\n\t\t}\n\t\t*e = WorkflowExecutionStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowExecutionStatus to text.\nfunc (e WorkflowExecutionStatus) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *WorkflowExecutionCloseStatus) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"COMPLETED\":\n\t\t*e = WorkflowExecutionCloseStatusCompleted\n\t\treturn nil\n\tcase \"FAILED\":\n\t\t*e = WorkflowExecutionCloseStatusFailed\n\t\treturn nil\n\tcase \"CANCELED\":\n\t\t*e = WorkflowExecutionCloseStatusCanceled\n\t\treturn nil\n\tcase \"TERMINATED\":\n\t\t*e = WorkflowExecutionCloseStatusTerminated\n\t\treturn nil\n\tcase \"CONTINUED_AS_NEW\":\n\t\t*e = WorkflowExecutionCloseStatusContinuedAsNew\n\t\treturn nil\n\tcase \"TIMED_OUT\":\n\t\t*e = WorkflowExecutionCloseStatusTimedOut\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowExecutionCloseStatus\", err)\n\t\t}\n\t\t*e = WorkflowExecutionCloseStatus(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowExecutionCloseStatus to text.\nfunc (e WorkflowExecutionCloseStatus) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// WorkflowExecutionCloseStatusCompleted is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusCompleted WorkflowExecutionCloseStatus = iota\n\t// WorkflowExecutionCloseStatusFailed is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusFailed\n\t// WorkflowExecutionCloseStatusCanceled is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusCanceled\n\t// WorkflowExecutionCloseStatusTerminated is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusTerminated\n\t// WorkflowExecutionCloseStatusContinuedAsNew is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusContinuedAsNew\n\t// WorkflowExecutionCloseStatusTimedOut is an option for WorkflowExecutionCloseStatus\n\tWorkflowExecutionCloseStatusTimedOut\n)\n\nconst (\n\t// WorkflowExecutionStatusPending is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusPending WorkflowExecutionStatus = iota\n\t// WorkflowExecutionStatusStarted is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusStarted\n\t// WorkflowExecutionStatusCompleted is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusCompleted\n\t// WorkflowExecutionStatusFailed is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusFailed\n\t// WorkflowExecutionStatusCanceled is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusCanceled\n\t// WorkflowExecutionStatusTerminated is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusTerminated\n\t// WorkflowExecutionStatusContinuedAsNew is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusContinuedAsNew\n\t// WorkflowExecutionStatusTimedOut is an option for WorkflowExecutionStatus\n\tWorkflowExecutionStatusTimedOut\n)\n\n// WorkflowExecutionCompletedEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionCompletedEventAttributes struct {\n\tResult                       []byte `json:\"result,omitempty\"`\n\tDecisionTaskCompletedEventID int64  `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionCompletedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionConfiguration is an internal type (TBD...)\ntype WorkflowExecutionConfiguration struct {\n\tTaskList                            *TaskList `json:\"taskList,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32    `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32    `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n}\n\n// WorkflowExecutionContinuedAsNewEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionContinuedAsNewEventAttributes struct {\n\tNewExecutionRunID                   string                        `json:\"newExecutionRunId,omitempty\"`\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tDecisionTaskCompletedEventID        int64                         `json:\"decisionTaskCompletedEventId,omitempty\"`\n\tBackoffStartIntervalInSeconds       *int32                        `json:\"backoffStartIntervalInSeconds,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tFailureReason                       *string                       `json:\"failureReason,omitempty\"`\n\tFailureDetails                      []byte                        `json:\"failureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetNewExecutionRunID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetNewExecutionRunID() (o string) {\n\tif v != nil {\n\t\treturn v.NewExecutionRunID\n\t}\n\treturn\n}\n\n// GetInitiator is an internal getter (TBD...)\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\treturn\n}\n\n// GetFailureReason is an internal getter (TBD...)\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetFailureReason() (o string) {\n\tif v != nil && v.FailureReason != nil {\n\t\treturn *v.FailureReason\n\t}\n\treturn\n}\n\n// GetLastCompletionResult is an internal getter (TBD...)\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) GetLastCompletionResult() (o []byte) {\n\tif v != nil && v.LastCompletionResult != nil {\n\t\treturn v.LastCompletionResult\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionContinuedAsNewEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionFailedEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionFailedEventAttributes struct {\n\tReason                       *string `json:\"reason,omitempty\"`\n\tDetails                      []byte  `json:\"details,omitempty\"`\n\tDecisionTaskCompletedEventID int64   `json:\"decisionTaskCompletedEventId,omitempty\"`\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *WorkflowExecutionFailedEventAttributes) GetReason() (o string) {\n\tif v != nil && v.Reason != nil {\n\t\treturn *v.Reason\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionFailedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionFilter is an internal type (TBD...)\ntype WorkflowExecutionFilter struct {\n\tWorkflowID string `json:\"workflowId,omitempty\"`\n\tRunID      string `json:\"runId,omitempty\"`\n}\n\n// GetWorkflowID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionFilter) GetWorkflowID() (o string) {\n\tif v != nil {\n\t\treturn v.WorkflowID\n\t}\n\treturn\n}\n\n// WorkflowExecutionInfo is an internal type (TBD...)\ntype WorkflowExecutionInfo struct {\n\tExecution                    *WorkflowExecution            `json:\"execution,omitempty\"`\n\tType                         *WorkflowType                 `json:\"type,omitempty\"`\n\tStartTime                    *int64                        `json:\"startTime,omitempty\"`\n\tCloseTime                    *int64                        `json:\"closeTime,omitempty\"`\n\tCloseStatus                  *WorkflowExecutionCloseStatus `json:\"closeStatus,omitempty\"`\n\tHistoryLength                int64                         `json:\"historyLength,omitempty\"` // should be history count\n\tParentDomainID               *string                       `json:\"parentDomainId,omitempty\"`\n\tParentDomain                 *string                       `json:\"parentDomain,omitempty\"`\n\tParentExecution              *WorkflowExecution            `json:\"parentExecution,omitempty\"`\n\tParentInitiatedID            *int64                        `json:\"parentInitiatedId,omitempty\"`\n\tExecutionTime                *int64                        `json:\"executionTime,omitempty\"`\n\tMemo                         *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes             *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tAutoResetPoints              *ResetPoints                  `json:\"autoResetPoints,omitempty\"`\n\tTaskList                     *TaskList                     `json:\"taskList,omitempty\"`\n\tIsCron                       bool                          `json:\"isCron,omitempty\"`\n\tUpdateTime                   *int64                        `json:\"updateTime,omitempty\"`\n\tPartitionConfig              map[string]string             `json:\"partitionConfig,omitempty\"`\n\tCronOverlapPolicy            *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n\tCronSchedule                 *string                       `json:\"cronSchedule,omitempty\"`\n\tExecutionStatus              *WorkflowExecutionStatus      `json:\"executionStatus,omitempty\"`\n\tScheduledExecutionTime       *int64                        `json:\"scheduledExecutionTime,omitempty\"`\n}\n\n// GetExecution is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetExecution() (o *WorkflowExecution) {\n\tif v != nil && v.Execution != nil {\n\t\treturn v.Execution\n\t}\n\treturn\n}\n\n// GetType is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetType() (o *WorkflowType) {\n\tif v != nil && v.Type != nil {\n\t\treturn v.Type\n\t}\n\treturn\n}\n\n// GetStartTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetStartTime() (o int64) {\n\tif v != nil && v.StartTime != nil {\n\t\treturn *v.StartTime\n\t}\n\treturn\n}\n\n// GetCloseTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetCloseTime() (o int64) {\n\tif v != nil && v.CloseTime != nil {\n\t\treturn *v.CloseTime\n\t}\n\treturn\n}\n\n// GetCloseStatus is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetCloseStatus() (o WorkflowExecutionCloseStatus) {\n\tif v != nil && v.CloseStatus != nil {\n\t\treturn *v.CloseStatus\n\t}\n\treturn\n}\n\n// GetExecutionTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetExecutionTime() (o int64) {\n\tif v != nil && v.ExecutionTime != nil {\n\t\treturn *v.ExecutionTime\n\t}\n\treturn\n}\n\n// GetUpdateTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetUpdateTime() (o int64) {\n\tif v != nil && v.UpdateTime != nil {\n\t\treturn *v.UpdateTime\n\t}\n\treturn\n}\n\n// GetSearchAttributes is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// GetCronSchedule is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetCronSchedule() (o string) {\n\tif v != nil && v.CronSchedule != nil {\n\t\treturn *v.CronSchedule\n\t}\n\treturn\n}\n\n// GetExecutionStatus is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetExecutionStatus() (o WorkflowExecutionStatus) {\n\tif v != nil && v.ExecutionStatus != nil {\n\t\treturn *v.ExecutionStatus\n\t}\n\treturn\n}\n\n// GetScheduledExecutionTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionInfo) GetScheduledExecutionTime() (o int64) {\n\tif v != nil && v.ScheduledExecutionTime != nil {\n\t\treturn *v.ScheduledExecutionTime\n\t}\n\treturn\n}\n\n// WorkflowExecutionSignaledEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionSignaledEventAttributes struct {\n\tSignalName string `json:\"signalName,omitempty\"`\n\tInput      []byte `json:\"input,omitempty\"`\n\tIdentity   string `json:\"identity,omitempty\"`\n\tRequestID  string `json:\"requestId,omitempty\"`\n}\n\n// GetSignalName is an internal getter (TBD...)\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetSignalName() (o string) {\n\tif v != nil {\n\t\treturn v.SignalName\n\t}\n\treturn\n}\n\n// GetInput is an internal getter (TBD...)\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetInput() (o []byte) {\n\tif v != nil && v.Input != nil {\n\t\treturn v.Input\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionSignaledEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionSignaledEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionStartedEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionStartedEventAttributes struct {\n\tWorkflowType                        *WorkflowType                 `json:\"workflowType,omitempty\"`\n\tParentWorkflowDomainID              *string                       `json:\"parentWorkflowDomainID,omitempty\"`\n\tParentWorkflowDomain                *string                       `json:\"parentWorkflowDomain,omitempty\"`\n\tParentWorkflowExecution             *WorkflowExecution            `json:\"parentWorkflowExecution,omitempty\"`\n\tParentInitiatedEventID              *int64                        `json:\"parentInitiatedEventId,omitempty\"`\n\tTaskList                            *TaskList                     `json:\"taskList,omitempty\"`\n\tInput                               []byte                        `json:\"input,omitempty\"`\n\tExecutionStartToCloseTimeoutSeconds *int32                        `json:\"executionStartToCloseTimeoutSeconds,omitempty\"`\n\tTaskStartToCloseTimeoutSeconds      *int32                        `json:\"taskStartToCloseTimeoutSeconds,omitempty\"`\n\tContinuedExecutionRunID             string                        `json:\"continuedExecutionRunId,omitempty\"`\n\tInitiator                           *ContinueAsNewInitiator       `json:\"initiator,omitempty\"`\n\tContinuedFailureReason              *string                       `json:\"continuedFailureReason,omitempty\"`\n\tContinuedFailureDetails             []byte                        `json:\"continuedFailureDetails,omitempty\"`\n\tLastCompletionResult                []byte                        `json:\"lastCompletionResult,omitempty\"`\n\tOriginalExecutionRunID              string                        `json:\"originalExecutionRunId,omitempty\"`\n\tIdentity                            string                        `json:\"identity,omitempty\"`\n\tFirstExecutionRunID                 string                        `json:\"firstExecutionRunId,omitempty\"`\n\tFirstScheduleTime                   *time.Time                    `json:\"firstScheduleTimeNano,omitempty\"`\n\tRetryPolicy                         *RetryPolicy                  `json:\"retryPolicy,omitempty\"`\n\tAttempt                             int32                         `json:\"attempt,omitempty\"`\n\tExpirationTimestamp                 *int64                        `json:\"expirationTimestamp,omitempty\"`\n\tCronSchedule                        string                        `json:\"cronSchedule,omitempty\"`\n\tFirstDecisionTaskBackoffSeconds     *int32                        `json:\"firstDecisionTaskBackoffSeconds,omitempty\"`\n\tMemo                                *Memo                         `json:\"memo,omitempty\"`\n\tSearchAttributes                    *SearchAttributes             `json:\"searchAttributes,omitempty\"`\n\tPrevAutoResetPoints                 *ResetPoints                  `json:\"prevAutoResetPoints,omitempty\"`\n\tHeader                              *Header                       `json:\"header,omitempty\"`\n\tJitterStartSeconds                  *int32                        `json:\"jitterStartSeconds,omitempty\"`\n\tPartitionConfig                     map[string]string             `json:\"partitionConfig,omitempty\"`\n\tRequestID                           string                        `json:\"requestId,omitempty\"`\n\tCronOverlapPolicy                   *CronOverlapPolicy            `json:\"cronOverlapPolicy,omitempty\"`\n\tActiveClusterSelectionPolicy        *ActiveClusterSelectionPolicy `json:\"activeClusterSelectionPolicy,omitempty\"`\n}\n\n// GetParentWorkflowDomain is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetParentWorkflowDomain() (o string) {\n\tif v != nil && v.ParentWorkflowDomain != nil {\n\t\treturn *v.ParentWorkflowDomain\n\t}\n\treturn\n}\n\n// GetParentInitiatedEventID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetParentInitiatedEventID() (o int64) {\n\tif v != nil && v.ParentInitiatedEventID != nil {\n\t\treturn *v.ParentInitiatedEventID\n\t}\n\treturn\n}\n\n// GetExecutionStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetExecutionStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.ExecutionStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.ExecutionStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetTaskStartToCloseTimeoutSeconds is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetTaskStartToCloseTimeoutSeconds() (o int32) {\n\tif v != nil && v.TaskStartToCloseTimeoutSeconds != nil {\n\t\treturn *v.TaskStartToCloseTimeoutSeconds\n\t}\n\treturn\n}\n\n// GetContinuedExecutionRunID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetContinuedExecutionRunID() (o string) {\n\tif v != nil {\n\t\treturn v.ContinuedExecutionRunID\n\t}\n\treturn\n}\n\n// GetInitiator is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetInitiator() (o ContinueAsNewInitiator) {\n\tif v != nil && v.Initiator != nil {\n\t\treturn *v.Initiator\n\t}\n\treturn\n}\n\n// GetFirstExecutionRunID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstExecutionRunID() (o string) {\n\tif v != nil {\n\t\treturn v.FirstExecutionRunID\n\t}\n\treturn\n}\n\n// GetFirstScheduledTime is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstScheduledTime() (o time.Time) {\n\tif v != nil && v.FirstScheduleTime != nil {\n\t\treturn *v.FirstScheduleTime\n\t}\n\treturn\n}\n\n// GetAttempt is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetAttempt() (o int32) {\n\tif v != nil {\n\t\treturn v.Attempt\n\t}\n\treturn\n}\n\n// GetExpirationTimestamp is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetExpirationTimestamp() (o int64) {\n\tif v != nil && v.ExpirationTimestamp != nil {\n\t\treturn *v.ExpirationTimestamp\n\t}\n\treturn\n}\n\n// GetCronSchedule is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetCronSchedule() (o string) {\n\tif v != nil {\n\t\treturn v.CronSchedule\n\t}\n\treturn\n}\n\n// GetFirstDecisionTaskBackoffSeconds is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetFirstDecisionTaskBackoffSeconds() (o int32) {\n\tif v != nil && v.FirstDecisionTaskBackoffSeconds != nil {\n\t\treturn *v.FirstDecisionTaskBackoffSeconds\n\t}\n\treturn\n}\n\nfunc (v *WorkflowExecutionStartedEventAttributes) GetJitterStartSeconds() (o int32) {\n\tif v != nil && v.JitterStartSeconds != nil {\n\t\treturn *v.JitterStartSeconds\n\t}\n\treturn\n}\n\n// GetMemo is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetMemo() (o *Memo) {\n\tif v != nil && v.Memo != nil {\n\t\treturn v.Memo\n\t}\n\treturn\n}\n\n// GetSearchAttributes is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetSearchAttributes() (o *SearchAttributes) {\n\tif v != nil && v.SearchAttributes != nil {\n\t\treturn v.SearchAttributes\n\t}\n\treturn\n}\n\n// GetPrevAutoResetPoints is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetPrevAutoResetPoints() (o *ResetPoints) {\n\tif v != nil && v.PrevAutoResetPoints != nil {\n\t\treturn v.PrevAutoResetPoints\n\t}\n\treturn\n}\n\n// GetPartitionConfig is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *WorkflowExecutionStartedEventAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionStartedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionTerminatedEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionTerminatedEventAttributes struct {\n\tReason   string `json:\"reason,omitempty\"`\n\tDetails  []byte `json:\"details,omitempty\"`\n\tIdentity string `json:\"identity,omitempty\"`\n}\n\n// GetReason is an internal getter (TBD...)\nfunc (v *WorkflowExecutionTerminatedEventAttributes) GetReason() (o string) {\n\tif v != nil {\n\t\treturn v.Reason\n\t}\n\treturn\n}\n\n// GetIdentity is an internal getter (TBD...)\nfunc (v *WorkflowExecutionTerminatedEventAttributes) GetIdentity() (o string) {\n\tif v != nil {\n\t\treturn v.Identity\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionTerminatedEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowExecutionTimedOutEventAttributes is an internal type (TBD...)\ntype WorkflowExecutionTimedOutEventAttributes struct {\n\tTimeoutType *TimeoutType `json:\"timeoutType,omitempty\"`\n}\n\n// GetTimeoutType is an internal getter (TBD...)\nfunc (v *WorkflowExecutionTimedOutEventAttributes) GetTimeoutType() (o TimeoutType) {\n\tif v != nil && v.TimeoutType != nil {\n\t\treturn *v.TimeoutType\n\t}\n\treturn\n}\n\n// Size returns the approximate memory used in bytes\nfunc (v *WorkflowExecutionTimedOutEventAttributes) ByteSize() uint64 {\n\treturn 0\n}\n\n// WorkflowIDReusePolicy is an internal type (TBD...)\ntype WorkflowIDReusePolicy int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e WorkflowIDReusePolicy) Ptr() *WorkflowIDReusePolicy {\n\treturn &e\n}\n\n// String returns a readable string representation of WorkflowIDReusePolicy.\nfunc (e WorkflowIDReusePolicy) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"AllowDuplicateFailedOnly\"\n\tcase 1:\n\t\treturn \"AllowDuplicate\"\n\tcase 2:\n\t\treturn \"RejectDuplicate\"\n\tcase 3:\n\t\treturn \"TerminateIfRunning\"\n\t}\n\treturn fmt.Sprintf(\"WorkflowIDReusePolicy(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *WorkflowIDReusePolicy) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"ALLOWDUPLICATEFAILEDONLY\":\n\t\t*e = WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\t\treturn nil\n\tcase \"ALLOWDUPLICATE\":\n\t\t*e = WorkflowIDReusePolicyAllowDuplicate\n\t\treturn nil\n\tcase \"REJECTDUPLICATE\":\n\t\t*e = WorkflowIDReusePolicyRejectDuplicate\n\t\treturn nil\n\tcase \"TERMINATEIFRUNNING\":\n\t\t*e = WorkflowIDReusePolicyTerminateIfRunning\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"WorkflowIDReusePolicy\", err)\n\t\t}\n\t\t*e = WorkflowIDReusePolicy(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes WorkflowIDReusePolicy to text.\nfunc (e WorkflowIDReusePolicy) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// WorkflowIDReusePolicyAllowDuplicateFailedOnly is an option for WorkflowIDReusePolicy\n\tWorkflowIDReusePolicyAllowDuplicateFailedOnly WorkflowIDReusePolicy = iota\n\t// WorkflowIDReusePolicyAllowDuplicate is an option for WorkflowIDReusePolicy\n\tWorkflowIDReusePolicyAllowDuplicate\n\t// WorkflowIDReusePolicyRejectDuplicate is an option for WorkflowIDReusePolicy\n\tWorkflowIDReusePolicyRejectDuplicate\n\t// WorkflowIDReusePolicyTerminateIfRunning is an option for WorkflowIDReusePolicy\n\tWorkflowIDReusePolicyTerminateIfRunning\n)\n\n// WorkflowQuery is an internal type (TBD...)\ntype WorkflowQuery struct {\n\tQueryType string `json:\"queryType,omitempty\"`\n\tQueryArgs []byte `json:\"queryArgs,omitempty\"`\n}\n\n// GetQueryType is an internal getter (TBD...)\nfunc (v *WorkflowQuery) GetQueryType() (o string) {\n\tif v != nil {\n\t\treturn v.QueryType\n\t}\n\treturn\n}\n\n// GetQueryArgs is an internal getter (TBD...)\nfunc (v *WorkflowQuery) GetQueryArgs() (o []byte) {\n\tif v != nil && v.QueryArgs != nil {\n\t\treturn v.QueryArgs\n\t}\n\treturn\n}\n\n// WorkflowQueryResult is an internal type (TBD...)\ntype WorkflowQueryResult struct {\n\tResultType   *QueryResultType `json:\"resultType,omitempty\"`\n\tAnswer       []byte           `json:\"answer,omitempty\"`\n\tErrorMessage string           `json:\"errorMessage,omitempty\"`\n}\n\n// GetResultType is an internal getter (TBD...)\nfunc (v *WorkflowQueryResult) GetResultType() (o QueryResultType) {\n\tif v != nil && v.ResultType != nil {\n\t\treturn *v.ResultType\n\t}\n\treturn\n}\n\n// GetAnswer is an internal getter (TBD...)\nfunc (v *WorkflowQueryResult) GetAnswer() (o []byte) {\n\tif v != nil && v.Answer != nil {\n\t\treturn v.Answer\n\t}\n\treturn\n}\n\n// GetErrorMessage is an internal getter (TBD...)\nfunc (v *WorkflowQueryResult) GetErrorMessage() (o string) {\n\tif v != nil {\n\t\treturn v.ErrorMessage\n\t}\n\treturn\n}\n\n// WorkflowType is an internal type (TBD...)\ntype WorkflowType struct {\n\tName string `json:\"name,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *WorkflowType) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// WorkflowTypeFilter is an internal type (TBD...)\ntype WorkflowTypeFilter struct {\n\tName string `json:\"name,omitempty\"`\n}\n\n// GetName is an internal getter (TBD...)\nfunc (v *WorkflowTypeFilter) GetName() (o string) {\n\tif v != nil {\n\t\treturn v.Name\n\t}\n\treturn\n}\n\n// CrossClusterTaskType is an internal type (TBD...)\ntype CrossClusterTaskType int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e CrossClusterTaskType) Ptr() *CrossClusterTaskType {\n\treturn &e\n}\n\n// String returns a readable string representation of CrossClusterTaskType.\nfunc (e CrossClusterTaskType) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"StartChildExecution\"\n\tcase 1:\n\t\treturn \"CancelExecution\"\n\tcase 2:\n\t\treturn \"SignalExecution\"\n\tcase 3:\n\t\treturn \"RecordChildWorkflowExecutionComplete\"\n\tcase 4:\n\t\treturn \"ApplyParentClosePolicy\"\n\t}\n\treturn fmt.Sprintf(\"CrossClusterTaskType(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *CrossClusterTaskType) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"STARTCHILDEXECUTION\":\n\t\t*e = CrossClusterTaskTypeStartChildExecution\n\t\treturn nil\n\tcase \"CANCELEXECUTION\":\n\t\t*e = CrossClusterTaskTypeCancelExecution\n\t\treturn nil\n\tcase \"SIGNALEXECUTION\":\n\t\t*e = CrossClusterTaskTypeSignalExecution\n\t\treturn nil\n\tcase \"RECORDCHILDWORKLOWEXECUTIONCOMPLETE\":\n\t\t*e = CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete\n\t\treturn nil\n\tcase \"APPLYPARENTCLOSEPOLICY\":\n\t\t*e = CrossClusterTaskTypeApplyParentPolicy\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CrossClusterTaskType\", err)\n\t\t}\n\t\t*e = CrossClusterTaskType(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CrossClusterTaskType to text.\nfunc (e CrossClusterTaskType) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// CrossClusterTaskTypeStartChildExecution is an option for CrossClusterTaskType\n\tCrossClusterTaskTypeStartChildExecution CrossClusterTaskType = iota\n\t// CrossClusterTaskTypeCancelExecution is an option for CrossClusterTaskType\n\tCrossClusterTaskTypeCancelExecution\n\t// CrossClusterTaskTypeSignalExecution is an option for CrossClusterTaskType\n\tCrossClusterTaskTypeSignalExecution\n\t// CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete is an option for CrossClusterTaskType\n\tCrossClusterTaskTypeRecordChildWorkflowExeuctionComplete\n\t// CrossClusterTaskTypeApplyParentPolicy is an option for CrossClusterTaskType\n\tCrossClusterTaskTypeApplyParentPolicy\n)\n\n// CrossClusterTaskFailedCause is an internal type (TBD...)\ntype CrossClusterTaskFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e CrossClusterTaskFailedCause) Ptr() *CrossClusterTaskFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of CrossClusterTaskFailedCause.\nfunc (e CrossClusterTaskFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"DOMAIN_NOT_ACTIVE\"\n\tcase 1:\n\t\treturn \"DOMAIN_NOT_EXISTS\"\n\tcase 2:\n\t\treturn \"WORKFLOW_ALREADY_RUNNING\"\n\tcase 3:\n\t\treturn \"WORKFLOW_NOT_EXISTS\"\n\tcase 4:\n\t\treturn \"WORKFLOW_ALREADY_COMPLETED\"\n\tcase 5:\n\t\treturn \"UNCATEGORIZED\"\n\t}\n\treturn fmt.Sprintf(\"CrossClusterTaskFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *CrossClusterTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"DOMAIN_NOT_ACTIVE\":\n\t\t*e = CrossClusterTaskFailedCauseDomainNotActive\n\t\treturn nil\n\tcase \"DOMAIN_NOT_EXISTS\":\n\t\t*e = CrossClusterTaskFailedCauseDomainNotExists\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_RUNNING\":\n\t\t*e = CrossClusterTaskFailedCauseWorkflowAlreadyRunning\n\t\treturn nil\n\tcase \"WORKFLOW_NOT_EXISTS\":\n\t\t*e = CrossClusterTaskFailedCauseWorkflowNotExists\n\t\treturn nil\n\tcase \"WORKFLOW_ALREADY_COMPLETED\":\n\t\t*e = CrossClusterTaskFailedCauseWorkflowAlreadyCompleted\n\t\treturn nil\n\tcase \"UNCATEGORIZED\":\n\t\t*e = CrossClusterTaskFailedCauseUncategorized\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"CrossClusterTaskFailedCause\", err)\n\t\t}\n\t\t*e = CrossClusterTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes CrossClusterTaskFailedCause to text.\nfunc (e CrossClusterTaskFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// CrossClusterTaskFailedCauseDomainNotActive is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseDomainNotActive CrossClusterTaskFailedCause = iota\n\t// CrossClusterTaskFailedCauseDomainNotExists is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseDomainNotExists\n\t// CrossClusterTaskFailedCauseWorkflowAlreadyRunning is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseWorkflowAlreadyRunning\n\t// CrossClusterTaskFailedCauseWorkflowNotExists is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseWorkflowNotExists\n\t// CrossClusterTaskFailedCauseWorkflowAlreadyCompleted is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseWorkflowAlreadyCompleted\n\t// CrossClusterTaskFailedCauseUncategorized is an option for CrossClusterTaskFailedCause\n\tCrossClusterTaskFailedCauseUncategorized\n)\n\n// GetTaskFailedCause is an internal type (TBD...)\ntype GetTaskFailedCause int32\n\n// Ptr is a helper function for getting pointer value\nfunc (e GetTaskFailedCause) Ptr() *GetTaskFailedCause {\n\treturn &e\n}\n\n// String returns a readable string representation of GetCrossClusterTaskFailedCause.\nfunc (e GetTaskFailedCause) String() string {\n\tw := int32(e)\n\tswitch w {\n\tcase 0:\n\t\treturn \"SERVICE_BUSY\"\n\tcase 1:\n\t\treturn \"TIMEOUT\"\n\tcase 2:\n\t\treturn \"SHARD_OWNERSHIP_LOST\"\n\tcase 3:\n\t\treturn \"UNCATEGORIZED\"\n\t}\n\treturn fmt.Sprintf(\"GetCrossClusterTaskFailedCause(%d)\", w)\n}\n\n// UnmarshalText parses enum value from string representation\nfunc (e *GetTaskFailedCause) UnmarshalText(value []byte) error {\n\tswitch s := strings.ToUpper(string(value)); s {\n\tcase \"SERVICE_BUSY\":\n\t\t*e = GetTaskFailedCauseServiceBusy\n\t\treturn nil\n\tcase \"TIMEOUT\":\n\t\t*e = GetTaskFailedCauseTimeout\n\t\treturn nil\n\tcase \"SHARD_OWNERSHIP_LOST\":\n\t\t*e = GetTaskFailedCauseShardOwnershipLost\n\t\treturn nil\n\tcase \"UNCATEGORIZED\":\n\t\t*e = GetTaskFailedCauseUncategorized\n\t\treturn nil\n\tdefault:\n\t\tval, err := strconv.ParseInt(s, 10, 32)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unknown enum value %q for %q: %v\", s, \"GetCrossClusterTaskFailedCause\", err)\n\t\t}\n\t\t*e = GetTaskFailedCause(val)\n\t\treturn nil\n\t}\n}\n\n// MarshalText encodes GetCrossClusterTaskFailedCause to text.\nfunc (e GetTaskFailedCause) MarshalText() ([]byte, error) {\n\treturn []byte(e.String()), nil\n}\n\nconst (\n\t// GetTaskFailedCauseServiceBusy is an option for GetCrossClusterTaskFailedCause\n\tGetTaskFailedCauseServiceBusy GetTaskFailedCause = iota\n\t// GetTaskFailedCauseTimeout is an option for GetCrossClusterTaskFailedCause\n\tGetTaskFailedCauseTimeout\n\t// GetTaskFailedCauseShardOwnershipLost is an option for GetCrossClusterTaskFailedCause\n\tGetTaskFailedCauseShardOwnershipLost\n\t// GetTaskFailedCauseUncategorized is an option for GetCrossClusterTaskFailedCause\n\tGetTaskFailedCauseUncategorized\n)\n\n// CrossClusterTaskInfo is an internal type (TBD...)\ntype CrossClusterTaskInfo struct {\n\tDomainID            string                `json:\"domainID,omitempty\"`\n\tWorkflowID          string                `json:\"workflowID,omitempty\"`\n\tRunID               string                `json:\"runID,omitempty\"`\n\tTaskType            *CrossClusterTaskType `json:\"taskType,omitempty\"`\n\tTaskState           int16                 `json:\"taskState,omitempty\"`\n\tTaskID              int64                 `json:\"taskID,omitempty\"`\n\tVisibilityTimestamp *int64                `json:\"visibilityTimestamp,omitempty\"`\n}\n\n// GetTaskType is an internal getter (TBD...)\nfunc (v *CrossClusterTaskInfo) GetTaskType() (o CrossClusterTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\treturn\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *CrossClusterTaskInfo) GetTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetVisibilityTimestamp is an internal getter (TBD...)\nfunc (v *CrossClusterTaskInfo) GetVisibilityTimestamp() (o int64) {\n\tif v != nil && v.VisibilityTimestamp != nil {\n\t\treturn *v.VisibilityTimestamp\n\t}\n\treturn\n}\n\n// CrossClusterStartChildExecutionRequestAttributes is an internal type (TBD...)\ntype CrossClusterStartChildExecutionRequestAttributes struct {\n\tTargetDomainID           string                                               `json:\"targetDomainID,omitempty\"`\n\tRequestID                string                                               `json:\"requestID,omitempty\"`\n\tInitiatedEventID         int64                                                `json:\"initiatedEventID,omitempty\"`\n\tInitiatedEventAttributes *StartChildWorkflowExecutionInitiatedEventAttributes `json:\"initiatedEventAttributes,omitempty\"`\n\tTargetRunID              *string                                              `json:\"targetRunID,omitempty\"`\n\tPartitionConfig          map[string]string\n}\n\n// GetRequestID is an internal getter (TBD...)\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetRequestID() (o string) {\n\tif v != nil {\n\t\treturn v.RequestID\n\t}\n\treturn\n}\n\n// GetInitiatedEventAttributes is an internal getter (TBD...)\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetInitiatedEventAttributes() (o *StartChildWorkflowExecutionInitiatedEventAttributes) {\n\tif v != nil && v.InitiatedEventAttributes != nil {\n\t\treturn v.InitiatedEventAttributes\n\t}\n\treturn\n}\n\n// GetTargetRunID is an internal getter (TBD...)\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetTargetRunID() (o string) {\n\tif v != nil && v.TargetRunID != nil {\n\t\treturn *v.TargetRunID\n\t}\n\treturn\n}\n\n// GetTargetRunID is an internal getter (TBD...)\nfunc (v *CrossClusterStartChildExecutionRequestAttributes) GetPartitionConfig() (o map[string]string) {\n\tif v != nil && v.PartitionConfig != nil {\n\t\treturn v.PartitionConfig\n\t}\n\treturn\n}\n\n// CrossClusterStartChildExecutionResponseAttributes is an internal type (TBD...)\ntype CrossClusterStartChildExecutionResponseAttributes struct {\n\tRunID string `json:\"runID,omitempty\"`\n}\n\n// GetRunID is an internal getter (TBD...)\nfunc (v *CrossClusterStartChildExecutionResponseAttributes) GetRunID() (o string) {\n\tif v != nil {\n\t\treturn v.RunID\n\t}\n\treturn\n}\n\n// CrossClusterCancelExecutionRequestAttributes is an internal type (TBD...)\ntype CrossClusterCancelExecutionRequestAttributes struct {\n\tTargetDomainID    string `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID  string `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID       string `json:\"targetRunID,omitempty\"`\n\tRequestID         string `json:\"requestID,omitempty\"`\n\tInitiatedEventID  int64  `json:\"initiatedEventID,omitempty\"`\n\tChildWorkflowOnly bool   `json:\"childWorkflowOnly,omitempty\"`\n}\n\n// CrossClusterCancelExecutionResponseAttributes is an internal type (TBD...)\ntype CrossClusterCancelExecutionResponseAttributes struct {\n}\n\n// CrossClusterSignalExecutionRequestAttributes is an internal type (TBD...)\ntype CrossClusterSignalExecutionRequestAttributes struct {\n\tTargetDomainID    string `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID  string `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID       string `json:\"targetRunID,omitempty\"`\n\tRequestID         string `json:\"requestID,omitempty\"`\n\tInitiatedEventID  int64  `json:\"initiatedEventID,omitempty\"`\n\tChildWorkflowOnly bool   `json:\"childWorkflowOnly,omitempty\"`\n\tSignalName        string `json:\"signalName,omitempty\"`\n\tSignalInput       []byte `json:\"signalInput,omitempty\"`\n\tControl           []byte `json:\"control,omitempty\"`\n}\n\n// CrossClusterSignalExecutionResponseAttributes is an internal type (TBD...)\ntype CrossClusterSignalExecutionResponseAttributes struct {\n}\n\ntype CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes struct {\n\tTargetDomainID   string        `json:\"targetDomainID,omitempty\"`\n\tTargetWorkflowID string        `json:\"targetWorkflowID,omitempty\"`\n\tTargetRunID      string        `json:\"targetRunID,omitempty\"`\n\tInitiatedEventID int64         `json:\"initiatedEventID,omitempty\"`\n\tCompletionEvent  *HistoryEvent `json:\"completionEvent,omitempty\"`\n}\n\n// CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes is an internal type (TBD...)\ntype CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes struct {\n}\n\ntype ApplyParentClosePolicyStatus struct {\n\tCompleted   bool                         `json:\"completed,omitempty\"`\n\tFailedCause *CrossClusterTaskFailedCause `json:\"failedCause,omitempty\"`\n}\n\ntype ApplyParentClosePolicyAttributes struct {\n\tChildDomainID     string             `json:\"ChildDomainID,omitempty\"`\n\tChildWorkflowID   string             `json:\"ChildWorkflowID,omitempty\"`\n\tChildRunID        string             `json:\"ChildRunID,omitempty\"`\n\tParentClosePolicy *ParentClosePolicy `json:\"parentClosePolicy,omitempty\"`\n}\n\n// GetParentClosePolicy is an internal getter (TBD...)\nfunc (v *ApplyParentClosePolicyAttributes) GetParentClosePolicy() (o *ParentClosePolicy) {\n\tif v != nil {\n\t\treturn v.ParentClosePolicy\n\t}\n\treturn\n}\n\ntype ApplyParentClosePolicyRequest struct {\n\tChild  *ApplyParentClosePolicyAttributes `json:\"child,omitempty\"`\n\tStatus *ApplyParentClosePolicyStatus     `json:\"status,omitempty\"`\n}\n\ntype CrossClusterApplyParentClosePolicyRequestAttributes struct {\n\tChildren []*ApplyParentClosePolicyRequest `json:\"children,omitempty\"`\n}\n\ntype ApplyParentClosePolicyResult struct {\n\tChild       *ApplyParentClosePolicyAttributes `json:\"child,omitempty\"`\n\tFailedCause *CrossClusterTaskFailedCause      `json:\"failedCause,omitempty\"`\n}\n\n// CrossClusterApplyParentClosePolicyResponseAttributes is an internal type (TBD...)\ntype CrossClusterApplyParentClosePolicyResponseAttributes struct {\n\tChildrenStatus []*ApplyParentClosePolicyResult `json:\"childrenStatus,omitempty\"`\n}\n\n// CrossClusterTaskRequest is an internal type (TBD...)\ntype CrossClusterTaskRequest struct {\n\tTaskInfo                                       *CrossClusterTaskInfo                                              `json:\"taskInfo,omitempty\"`\n\tStartChildExecutionAttributes                  *CrossClusterStartChildExecutionRequestAttributes                  `json:\"startChildExecutionAttributes,omitempty\"`\n\tCancelExecutionAttributes                      *CrossClusterCancelExecutionRequestAttributes                      `json:\"cancelExecutionAttributes,omitempty\"`\n\tSignalExecutionAttributes                      *CrossClusterSignalExecutionRequestAttributes                      `json:\"signalExecutionAttributes,omitempty\"`\n\tRecordChildWorkflowExecutionCompleteAttributes *CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes `json:\"RecordChildWorkflowExecutionCompleteAttributes,omitempty\"`\n\tApplyParentClosePolicyAttributes               *CrossClusterApplyParentClosePolicyRequestAttributes               `json:\"ApplyParentClosePolicyAttributes,omitempty\"`\n}\n\n// CrossClusterTaskResponse is an internal type (TBD...)\ntype CrossClusterTaskResponse struct {\n\tTaskID                                         int64                                                               `json:\"taskID,omitempty\"`\n\tTaskType                                       *CrossClusterTaskType                                               `json:\"taskType,omitempty\"`\n\tTaskState                                      int16                                                               `json:\"taskState,omitempty\"`\n\tFailedCause                                    *CrossClusterTaskFailedCause                                        `json:\"failedCause,omitempty\"`\n\tStartChildExecutionAttributes                  *CrossClusterStartChildExecutionResponseAttributes                  `json:\"startChildExecutionAttributes,omitempty\"`\n\tCancelExecutionAttributes                      *CrossClusterCancelExecutionResponseAttributes                      `json:\"cancelExecutionAttributes,omitempty\"`\n\tSignalExecutionAttributes                      *CrossClusterSignalExecutionResponseAttributes                      `json:\"signalExecutionAttributes,omitempty\"`\n\tRecordChildWorkflowExecutionCompleteAttributes *CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes `json:\"RecordChildWorkflowExecutionCompleteAttributes,omitempty\"`\n\tApplyParentClosePolicyAttributes               *CrossClusterApplyParentClosePolicyResponseAttributes               `json:\"ApplyParentClosePolicyAttributes,omitempty\"`\n}\n\n// GetTaskID is an internal getter (TBD...)\nfunc (v *CrossClusterTaskResponse) GetTaskID() (o int64) {\n\tif v != nil {\n\t\treturn v.TaskID\n\t}\n\treturn\n}\n\n// GetTaskType is an internal getter (TBD...)\nfunc (v *CrossClusterTaskResponse) GetTaskType() (o CrossClusterTaskType) {\n\tif v != nil && v.TaskType != nil {\n\t\treturn *v.TaskType\n\t}\n\treturn\n}\n\n// GetFailedCause is an internal getter (TBD...)\nfunc (v *CrossClusterTaskResponse) GetFailedCause() (o CrossClusterTaskFailedCause) {\n\tif v != nil && v.FailedCause != nil {\n\t\treturn *v.FailedCause\n\t}\n\treturn\n}\n\n// GetCrossClusterTasksRequest is an internal type (TBD...)\ntype GetCrossClusterTasksRequest struct {\n\tShardIDs      []int32 `json:\"shardIDs,omitempty\"`\n\tTargetCluster string  `json:\"targetCluster,omitempty\"`\n}\n\n// GetShardIDs is an internal getter (TBD...)\nfunc (v *GetCrossClusterTasksRequest) GetShardIDs() (o []int32) {\n\tif v != nil && v.ShardIDs != nil {\n\t\treturn v.ShardIDs\n\t}\n\treturn\n}\n\n// GetTargetCluster is an internal getter (TBD...)\nfunc (v *GetCrossClusterTasksRequest) GetTargetCluster() (o string) {\n\tif v != nil {\n\t\treturn v.TargetCluster\n\t}\n\treturn\n}\n\n// GetCrossClusterTasksResponse is an internal type (TBD...)\ntype GetCrossClusterTasksResponse struct {\n\tTasksByShard       map[int32][]*CrossClusterTaskRequest `json:\"tasksByShard,omitempty\"`\n\tFailedCauseByShard map[int32]GetTaskFailedCause         `json:\"failedCauseByShard,omitempty\"`\n}\n\n// GetTasksByShard is an internal getter (TBD...)\nfunc (v *GetCrossClusterTasksResponse) GetTasksByShard() (o map[int32][]*CrossClusterTaskRequest) {\n\tif v != nil && v.TasksByShard != nil {\n\t\treturn v.TasksByShard\n\t}\n\treturn\n}\n\n// RespondCrossClusterTasksCompletedRequest is an internal type (TBD...)\ntype RespondCrossClusterTasksCompletedRequest struct {\n\tShardID       int32                       `json:\"shardID,omitempty\"`\n\tTargetCluster string                      `json:\"targetCluster,omitempty\"`\n\tTaskResponses []*CrossClusterTaskResponse `json:\"taskResponses,omitempty\"`\n\tFetchNewTasks bool                        `json:\"fetchNewTasks,omitempty\"`\n}\n\n// GetShardID is an internal getter (TBD...)\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetShardID() (o int32) {\n\tif v != nil {\n\t\treturn v.ShardID\n\t}\n\treturn\n}\n\n// GetFetchNewTasks is an internal getter (TBD...)\nfunc (v *RespondCrossClusterTasksCompletedRequest) GetFetchNewTasks() (o bool) {\n\tif v != nil {\n\t\treturn v.FetchNewTasks\n\t}\n\treturn\n}\n\n// RespondCrossClusterTasksCompletedResponse is an internal type (TBD...)\ntype RespondCrossClusterTasksCompletedResponse struct {\n\tTasks []*CrossClusterTaskRequest `json:\"tasks,omitempty\"`\n}\n\n// StickyWorkerUnavailableError is an internal type (TBD...)\ntype StickyWorkerUnavailableError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// ReadOnlyPartitionError is an internal type (TBD...)\ntype ReadOnlyPartitionError struct {\n\tMessage string `json:\"message,required\"`\n}\n\n// Any is an internal mirror of google.protobuf.Any, serving the same purposes, but\n// intentionally breaking direct compatibility because it may hold data that is not\n// actually protobuf encoded.\n//\n// All uses of Any must either:\n//   - check that ValueType is a recognized type, and deserialize based on its contents\n//   - or just pass it along as an opaque type for something else to use\n//\n// Contents are intentionally undefined to allow external definitions, e.g. from\n// third-party plugins that are not part of this source repository.\ntype Any struct {\n\t// ValueType describes the type of encoded data in Value.\n\t//\n\t// No structure or allowed values are defined, but you are strongly encouraged\n\t// to use hard-coded strings (or unambiguous prefixes) or URLs when possible.\n\t//\n\t// For more concise encoding of exclusively known types, use e.g. DataBlob instead.\n\tValueType string `json:\"value_type\"`\n\t// Value holds arbitrary bytes, and is described by ValueType.\n\t//\n\t// To interpret, you MUST check ValueType.\n\tValue []byte `json:\"value\"`\n}\n\n// AutoConfigHint is an internal type (TBD...)\ntype AutoConfigHint struct {\n\tEnableAutoConfig   bool  `json:\"enableAutoConfig\"`\n\tPollerWaitTimeInMs int64 `json:\"pollerWaitTimeInMs\"`\n}\n\ntype CronOverlapPolicy int32\n\nconst (\n\tCronOverlapPolicySkipped CronOverlapPolicy = iota\n\tCronOverlapPolicyBufferOne\n)\n\nfunc (v CronOverlapPolicy) String() string {\n\tswitch v {\n\tcase CronOverlapPolicySkipped:\n\t\treturn \"SKIP\"\n\tcase CronOverlapPolicyBufferOne:\n\t\treturn \"BUFFERONE\"\n\t}\n\treturn \"UNKNOWN\"\n}\n\n// Ptr is a helper function for getting pointer to CronOverlapPolicy\nfunc (v CronOverlapPolicy) Ptr() *CronOverlapPolicy {\n\treturn &v\n}\n"
  },
  {
    "path": "common/types/shared_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage types\n\nimport (\n\t\"testing\"\n\t\"unsafe\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestDataBlobDeepCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput *DataBlob\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: &DataBlob{},\n\t\t},\n\t\t{\n\t\t\tname: \"thrift ok\",\n\t\t\tinput: &DataBlob{\n\t\t\t\tEncodingType: EncodingTypeThriftRW.Ptr(),\n\t\t\t\tData:         []byte(\"some thrift data\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"json ok\",\n\t\t\tinput: &DataBlob{\n\t\t\t\tEncodingType: EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         []byte(\"some json data\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ttc := tc\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot := tc.input.DeepCopy()\n\t\t\tassert.Equal(t, tc.input, got)\n\t\t\tif tc.input != nil && tc.input.Data != nil && identicalByteArray(tc.input.Data, got.Data) {\n\t\t\t\tt.Error(\"expected DeepCopy to return a new data slice\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestActiveClustersConfigDeepCopy(t *testing.T) {\n\tnormalConfig := &ActiveClusters{\n\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\tActiveClusterName: \"us-east-1-cluster\",\n\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t},\n\t\t\t\t\t\"us-east-2\": {\n\t\t\t\t\t\tActiveClusterName: \"us-east-2-cluster\",\n\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname   string\n\t\tinput  *ActiveClusters\n\t\texpect *ActiveClusters\n\t}{\n\t\t{\n\t\t\tname:   \"nil case\",\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t{\n\t\t\tname:   \"empty case with nil map\",\n\t\t\tinput:  &ActiveClusters{},\n\t\t\texpect: &ActiveClusters{},\n\t\t},\n\t\t{\n\t\t\tname: \"empty case with empty map\",\n\t\t\tinput: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{},\n\t\t\t},\n\t\t\texpect: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:   \"normal case\",\n\t\t\tinput:  normalConfig,\n\t\t\texpect: normalConfig,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdeepCopy := tc.input.DeepCopy()\n\t\t\tif diff := cmp.Diff(tc.expect, deepCopy); diff != \"\" {\n\t\t\t\tt.Errorf(\"DeepCopy() mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Todo (david.porter) delete this test and codegen this\nfunc TestActiveClustersDeepCopyMutationIsolation(t *testing.T) {\n\n\tt.Run(\"modifying nested ClusterAttributes map in original should not affect copy\", func(t *testing.T) {\n\t\toriginal := &ActiveClusters{\n\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tcopied := original.DeepCopy()\n\n\t\tassert.Equal(t, original, copied)\n\n\t\tscope := original.AttributeScopes[\"region\"]\n\t\tscope.ClusterAttributes[\"us-west-1\"] = ActiveClusterInfo{\n\t\t\tActiveClusterName: \"cluster2\",\n\t\t\tFailoverVersion:   200,\n\t\t}\n\t\toriginal.AttributeScopes[\"region\"] = scope\n\n\t\tassert.Len(t, original.AttributeScopes[\"region\"].ClusterAttributes, 2)\n\t\tassert.Len(t, copied.AttributeScopes[\"region\"].ClusterAttributes, 1)\n\t\tassert.Contains(t, original.AttributeScopes[\"region\"].ClusterAttributes, \"us-west-1\")\n\t\tassert.NotContains(t, copied.AttributeScopes[\"region\"].ClusterAttributes, \"us-west-1\")\n\t})\n}\n\nfunc TestIsActiveActiveDomain(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tactiveClusters *DomainReplicationConfiguration\n\t\twant           bool\n\t}{\n\t\t{\n\t\t\tname:           \"empty DomainReplicationConfiguration should return false\",\n\t\t\tactiveClusters: &DomainReplicationConfiguration{},\n\t\t\twant:           false,\n\t\t},\n\t\t{\n\t\t\tname:           \"nil receiver should return false\",\n\t\t\tactiveClusters: nil,\n\t\t\twant:           false,\n\t\t},\n\t\t{\n\t\t\tname:           \"empty ActiveClusters should return false\",\n\t\t\tactiveClusters: &DomainReplicationConfiguration{ActiveClusters: &ActiveClusters{}},\n\t\t\twant:           false,\n\t\t},\n\t\t{\n\t\t\tname: \"ActiveClusters with only old format populated should return true\",\n\t\t\tactiveClusters: &DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ActiveClusters with only new format populated should return true\",\n\t\t\tactiveClusters: &DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {ClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ActiveClusters with both formats populated should return true\",\n\t\t\tactiveClusters: &DomainReplicationConfiguration{\n\t\t\t\tActiveClusters: &ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {ClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.activeClusters.IsActiveActive()\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\n// identicalByteArray returns true if a and b are the same slice, false otherwise.\nfunc identicalByteArray(a, b []byte) bool {\n\treturn len(a) == len(b) && unsafe.SliceData(a) == unsafe.SliceData(b)\n}\n\nfunc TestActiveClusters_GetAllClusters(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tactiveClusters *ActiveClusters\n\t\twant           []string\n\t}{\n\t\t{\n\t\t\tname:           \"nil receiver should return empty slice\",\n\t\t\tactiveClusters: nil,\n\t\t\twant:           []string{},\n\t\t},\n\t\t{\n\t\t\tname:           \"empty ActiveClusters should return empty slice\",\n\t\t\tactiveClusters: &ActiveClusters{},\n\t\t\twant:           []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"only old format populated should return attribute names from old format sorted\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\"eu-west-1\": {ActiveClusterName: \"cluster3\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: []string{\"eu-west-1\", \"us-east-1\", \"us-west-1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"only new format populated should return attribute names from new format sorted\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"ap-south-1\": {ActiveClusterName: \"cluster4\"},\n\t\t\t\t\t\t\t\"eu-north-1\": {ActiveClusterName: \"cluster5\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: []string{\"ap-south-1\", \"eu-north-1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"both formats with different attribute names should return deduplicated sorted list\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\":  {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\"us-west-1\":  {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t\"eu-west-1\":  {ActiveClusterName: \"cluster3\"},\n\t\t\t\t\t\t\t\"ap-south-1\": {ActiveClusterName: \"cluster4\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: []string{\"ap-south-1\", \"eu-west-1\", \"us-east-1\", \"us-west-1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"both formats with overlapping attribute names should return deduplicated sorted list\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\"us-west-1\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t\"eu-west-1\": {ActiveClusterName: \"cluster3\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: []string{\"eu-west-1\", \"us-east-1\", \"us-west-1\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := tt.activeClusters.GetAllClusters()\n\t\t\tif diff := cmp.Diff(tt.want, got); diff != \"\" {\n\t\t\t\tt.Errorf(\"GetAllClusters() mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestActiveClusters_GetFailoverVersionForAttribute(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tactiveClusters *ActiveClusters\n\t\tscopeType      string\n\t\tattributeName  string\n\t\texpected       int64\n\t\texpectedErr    error\n\t}{\n\t\t{\n\t\t\tname: \"normal value / success case - this should provide the failover version\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-west-2\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tscopeType:     \"region\",\n\t\t\tattributeName: \"us-east-1\",\n\t\t\texpected:      100,\n\t\t},\n\t\t{\n\t\t\tname: \"normal value / success case for zero values\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tscopeType:     \"region\",\n\t\t\tattributeName: \"us-east-1\",\n\t\t\texpected:      0,\n\t\t},\n\t\t{\n\t\t\tname:           \"nil receiver should return an error\",\n\t\t\tactiveClusters: nil,\n\t\t\tscopeType:      \"region\",\n\t\t\tattributeName:  \"us-east-1\",\n\t\t\texpected:       -1,\n\t\t\texpectedErr: &ClusterAttributeNotFoundError{\n\t\t\t\tScopeType:     \"region\",\n\t\t\t\tAttributeName: \"us-east-1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"empty ActiveClusters should return an error\",\n\t\t\tactiveClusters: &ActiveClusters{},\n\t\t\tscopeType:      \"region\",\n\t\t\tattributeName:  \"us-east-1\",\n\t\t\texpected:       -1,\n\t\t\texpectedErr: &ClusterAttributeNotFoundError{\n\t\t\t\tScopeType:     \"region\",\n\t\t\t\tAttributeName: \"us-east-1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"nil AttributeScopes should return an error\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: nil,\n\t\t\t},\n\t\t\tscopeType:     \"region\",\n\t\t\tattributeName: \"us-east-1\",\n\t\t\texpected:      -1,\n\t\t\texpectedErr: &ClusterAttributeNotFoundError{\n\t\t\t\tScopeType:     \"region\",\n\t\t\t\tAttributeName: \"us-east-1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scopeType not found should return an error\",\n\t\t\tactiveClusters: &ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east-1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tscopeType:     \"datacenter\",\n\t\t\tattributeName: \"dc1\",\n\t\t\texpected:      -1,\n\t\t\texpectedErr: &ClusterAttributeNotFoundError{\n\t\t\t\tScopeType:     \"datacenter\",\n\t\t\t\tAttributeName: \"dc1\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, gotErr := tt.activeClusters.GetFailoverVersionForAttribute(tt.scopeType, tt.attributeName)\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t\tassert.Equal(t, tt.expectedErr, gotErr)\n\t\t})\n\t}\n}\n\nfunc TestBadBinariesDeepCopy(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tinput *BadBinaries\n\t}{\n\t\t{\n\t\t\tname:  \"nil\",\n\t\t\tinput: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"empty\",\n\t\t\tinput: &BadBinaries{},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple binaries\",\n\t\t\tinput: &BadBinaries{\n\t\t\t\tBinaries: map[string]*BadBinaryInfo{\n\t\t\t\t\t\"bad1\": {\n\t\t\t\t\t\tReason:          \"reason1\",\n\t\t\t\t\t\tOperator:        \"op1\",\n\t\t\t\t\t\tCreatedTimeNano: func() *int64 { i := int64(111); return &i }(),\n\t\t\t\t\t},\n\t\t\t\t\t\"bad2\": {\n\t\t\t\t\t\tReason:   \"reason2\",\n\t\t\t\t\t\tOperator: \"op2\",\n\t\t\t\t\t},\n\t\t\t\t\t\"bad3\": {\n\t\t\t\t\t\tReason:          \"reason3\",\n\t\t\t\t\t\tOperator:        \"op3\",\n\t\t\t\t\t\tCreatedTimeNano: func() *int64 { i := int64(333); return &i }(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tcopied := tt.input.DeepCopy()\n\n\t\t\t// Test for nil input\n\t\t\tif tt.input == nil {\n\t\t\t\tassert.Empty(t, copied.Binaries)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Verify values are equal\n\t\t\tassert.Equal(t, tt.input.Binaries, copied.Binaries, \"values should be equal\")\n\n\t\t\t// Verify maps are different pointers (if not nil)\n\t\t\tif tt.input.Binaries != nil {\n\t\t\t\tassert.True(t, &tt.input.Binaries != &copied.Binaries, \"maps should have different memory addresses\")\n\n\t\t\t\t// Verify each BadBinaryInfo is a different pointer\n\t\t\t\tfor key, originalInfo := range tt.input.Binaries {\n\t\t\t\t\tcopiedInfo := copied.Binaries[key]\n\t\t\t\t\tif originalInfo != nil {\n\t\t\t\t\t\tassert.NotNil(t, copiedInfo)\n\t\t\t\t\t\tassert.Equal(t, originalInfo.Reason, copiedInfo.Reason)\n\t\t\t\t\t\tassert.Equal(t, originalInfo.Operator, copiedInfo.Operator)\n\t\t\t\t\t\tassert.True(t, originalInfo != copiedInfo, \"BadBinaryInfo should be different pointers\")\n\n\t\t\t\t\t\t// Verify CreatedTimeNano is deep copied\n\t\t\t\t\t\tif originalInfo.CreatedTimeNano != nil {\n\t\t\t\t\t\t\tassert.NotNil(t, copiedInfo.CreatedTimeNano)\n\t\t\t\t\t\t\tassert.Equal(t, *originalInfo.CreatedTimeNano, *copiedInfo.CreatedTimeNano)\n\t\t\t\t\t\t\tassert.True(t, originalInfo.CreatedTimeNano != copiedInfo.CreatedTimeNano, \"CreatedTimeNano pointers should be different\")\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\t// Verify modifications to original don't affect copy\n\t\t\t\tif len(tt.input.Binaries) > 0 {\n\t\t\t\t\toriginalLen := len(copied.Binaries)\n\t\t\t\t\ttt.input.Binaries[\"new-bad\"] = &BadBinaryInfo{Reason: \"new-reason\"}\n\t\t\t\t\tassert.Equal(t, originalLen, len(copied.Binaries), \"modifying original should not affect copy\")\n\t\t\t\t\tassert.NotContains(t, copied.Binaries, \"new-bad\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/types/test_util.go",
    "content": "package types\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n)\n\ntype byteSizer interface{ ByteSize() uint64 }\n\n// AssertByteSizeMatchesReflect verifies v.(byteSizer).ByteSize()\n// matches an independent reflective deep-size computation.\nfunc AssertByteSizeMatchesReflect(t *testing.T, v any) {\n\tt.Helper()\n\n\tgen := testdatagen.New(t)\n\n\tfor i := 0; i < 100; i++ {\n\n\t\tgen.Fuzz(v)\n\n\t\trv := reflect.ValueOf(v)\n\t\tif !rv.IsValid() {\n\t\t\tt.Fatalf(\"nil/invalid value\")\n\t\t}\n\n\t\t// Must implement ByteSize (on value or pointer)\n\t\tbs, ok := v.(byteSizer)\n\t\tif !ok {\n\t\t\t// if they passed a non-pointer, check its pointer receiver too\n\t\t\tif rv.Kind() != reflect.Ptr {\n\t\t\t\tif pv, ok2 := reflect.New(rv.Type()).Interface().(byteSizer); ok2 {\n\t\t\t\t\tbs = pv\n\t\t\t\t\trv = rv.Addr()\n\t\t\t\t\tok = true\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tassert.True(t, ok, \"%T does not implement ByteSize()\", v)\n\n\t\t// Compare impl vs reflect-based computation.\n\t\troot := rv\n\t\tif root.Kind() == reflect.Ptr {\n\t\t\troot = root.Elem()\n\t\t}\n\t\texp := uint64(root.Type().Size()) + structPayloadSize(root)\n\t\tgot := bs.ByteSize()\n\n\t\tassert.Equal(t, exp, got, \"ByteSize mismatch for %T: got=%d expected=%d (likely a new/changed field not reflected in ByteSize)\", v, got, exp)\n\t}\n}\n\n// AssertReachablesImplementByteSize ensures every reachable named struct\n// under root implements ByteSize (on value or pointer receiver).\n// You can pass a value (&T{}) or a type via (*T)(nil).\nfunc AssertReachablesImplementByteSize(t *testing.T, root any) {\n\tt.Helper()\n\n\trt := reflect.TypeOf(root)\n\n\tassert.NotNil(t, rt, \"nil type\")\n\n\tif rt.Kind() == reflect.Ptr {\n\t\trt = rt.Elem()\n\t}\n\n\tseen := map[reflect.Type]bool{}\n\tvar missing []string\n\tcheckStructHasByteSize(rt, seen, &missing)\n\n\tassert.Equal(t, 0, len(missing), \"reachable struct types missing ByteSize(): %v\", missing)\n}\n\n// ------------------- helpers -------------------\n\nfunc structPayloadSize(v reflect.Value) uint64 {\n\tif v.Kind() == reflect.Ptr {\n\t\tif v.IsNil() {\n\t\t\treturn 0\n\t\t}\n\t\treturn typePayloadSize(v.Elem())\n\t}\n\treturn typePayloadSize(v)\n}\n\nfunc typePayloadSize(v reflect.Value) uint64 {\n\tif !v.IsValid() {\n\t\treturn 0\n\t}\n\n\tswitch v.Kind() {\n\tcase reflect.Ptr:\n\t\tif v.IsNil() {\n\t\t\treturn 0\n\t\t}\n\t\t// DO NOT call ByteSize() here — keep independence\n\t\t// the size calculation is done recursively using reflection to provide an independent check on the ByteSize impl\n\t\t// Therefore, the size calculated by reflection is the ground truth and the ByteSize impl must match it\n\t\telem := v.Elem()\n\t\t// Add the pointed-to value's header size (since it's a separate allocation)\n\t\t// plus its dynamic payload.\n\t\tswitch elem.Kind() {\n\t\tcase reflect.Struct, reflect.String, reflect.Slice, reflect.Map, reflect.Array:\n\t\t\treturn uint64(elem.Type().Size()) + typePayloadSize(elem)\n\t\tdefault:\n\t\t\t// scalar pointed-to: just the scalar's size\n\t\t\treturn uint64(elem.Type().Size())\n\t\t}\n\n\tcase reflect.Struct:\n\t\tvar sum uint64\n\t\tfor i := 0; i < v.NumField(); i++ {\n\t\t\tsum += fieldPayloadSize(v.Field(i))\n\t\t}\n\t\treturn sum\n\n\tcase reflect.String:\n\t\treturn uint64(v.Len()) // header counted by parent (or pointer case above)\n\n\tcase reflect.Slice:\n\t\treturn slicePayloadSize(v)\n\n\tcase reflect.Array:\n\t\tvar sum uint64\n\t\tfor i := 0; i < v.Len(); i++ {\n\t\t\tsum += typePayloadSize(v.Index(i))\n\t\t}\n\t\treturn sum\n\n\tcase reflect.Map:\n\t\tvar sum uint64\n\t\tfor _, k := range v.MapKeys() {\n\t\t\tsum += typePayloadSize(k)\n\t\t\tsum += typePayloadSize(v.MapIndex(k))\n\t\t}\n\t\treturn sum\n\n\tdefault:\n\t\treturn 0\n\t}\n}\n\nfunc fieldPayloadSize(f reflect.Value) uint64 {\n\tswitch f.Kind() {\n\tcase reflect.Ptr:\n\t\t// pointer slot already in parent header; add referent header+payload\n\t\tif f.IsNil() {\n\t\t\treturn 0\n\t\t}\n\t\telem := f.Elem()\n\t\tswitch elem.Kind() {\n\t\tcase reflect.Struct, reflect.String, reflect.Slice, reflect.Map, reflect.Array:\n\t\t\treturn uint64(elem.Type().Size()) + typePayloadSize(elem)\n\t\tdefault:\n\t\t\treturn uint64(elem.Type().Size())\n\t\t}\n\tcase reflect.Struct, reflect.Map:\n\t\treturn typePayloadSize(f)\n\tcase reflect.String:\n\t\treturn uint64(f.Len())\n\tcase reflect.Slice:\n\t\treturn slicePayloadSize(f)\n\tcase reflect.Array:\n\t\tvar sum uint64\n\t\tfor i := 0; i < f.Len(); i++ {\n\t\t\tsum += typePayloadSize(f.Index(i))\n\t\t}\n\t\treturn sum\n\tdefault:\n\t\treturn 0\n\t}\n}\n\nfunc slicePayloadSize(v reflect.Value) uint64 {\n\tif v.IsNil() {\n\t\treturn 0\n\t}\n\tn := v.Len()\n\telem := v.Type().Elem()\n\tsum := uint64(n) * uint64(elem.Size()) // backing array\n\tswitch elem.Kind() {\n\tcase reflect.String, reflect.Slice, reflect.Array, reflect.Map, reflect.Struct, reflect.Ptr:\n\t\tfor i := 0; i < n; i++ {\n\t\t\tsum += typePayloadSize(v.Index(i))\n\t\t}\n\t}\n\treturn sum\n}\n\nfunc isScalar(k reflect.Kind) bool {\n\tswitch k {\n\tcase reflect.Bool,\n\t\treflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,\n\t\treflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr,\n\t\treflect.Float32, reflect.Float64,\n\t\treflect.Complex64, reflect.Complex128:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc checkStructHasByteSize(t reflect.Type, seen map[reflect.Type]bool, missing *[]string) {\n\tswitch t.Kind() {\n\tcase reflect.Ptr:\n\t\tcheckStructHasByteSize(t.Elem(), seen, missing)\n\t\treturn\n\tcase reflect.Struct:\n\t\tif t.Name() != \"\" {\n\t\t\tif seen[t] {\n\t\t\t\treturn\n\t\t\t}\n\t\t\tseen[t] = true\n\t\t\t// value or pointer receiver ok\n\t\t\tif _, ok := t.MethodByName(\"ByteSize\"); !ok {\n\t\t\t\tif _, ok := reflect.PtrTo(t).MethodByName(\"ByteSize\"); !ok {\n\t\t\t\t\t*missing = append(*missing, t.String())\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tfor i := 0; i < t.NumField(); i++ {\n\t\t\tcheckStructHasByteSize(t.Field(i).Type, seen, missing)\n\t\t}\n\tcase reflect.Slice, reflect.Array:\n\t\tcheckStructHasByteSize(t.Elem(), seen, missing)\n\tcase reflect.Map:\n\t\tcheckStructHasByteSize(t.Key(), seen, missing)\n\t\tcheckStructHasByteSize(t.Elem(), seen, missing)\n\t}\n}\n"
  },
  {
    "path": "common/types/testdata/common.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tWorkflowID = \"WorkflowID\"\n\tRunID      = \"RunID\"\n\tRunID1     = \"RunID1\"\n\tRunID2     = \"RunID2\"\n\tRunID3     = \"RunID3\"\n\n\tActivityID = \"ActivityID\"\n\tRequestID  = \"RequestID\"\n\tTimerID    = \"TimerID\"\n\n\tWorkflowTypeName     = \"WorkflowTypeName\"\n\tActivityTypeName     = \"ActivityTypeName\"\n\tTaskListName         = \"TaskListName\"\n\tMarkerName           = \"MarkerName\"\n\tSignalName           = \"SignalName\"\n\tQueryType            = \"QueryType\"\n\tHostName             = \"HostName\"\n\tHostName2            = \"HostName2\"\n\tIdentity             = \"Identity\"\n\tCronSchedule         = \"CronSchedule\"\n\tChecksum             = \"Checksum\"\n\tReason               = \"Reason\"\n\tCause                = \"Cause\"\n\tSecurityToken        = \"SecurityToken\"\n\tVisibilityQuery      = \"VisibilityQuery\"\n\tFeatureVersion       = \"FeatureVersion\"\n\tClientImpl           = \"ClientImpl\"\n\tClientLibraryVersion = \"ClientLibraryVersion\"\n\tSupportedVersions    = \"SupportedVersions\"\n\tFeatureFlag          = \"FeatureFlag\"\n\tNamespace            = \"Namespace\"\n\tShardKey             = \"ShardKey\"\n\n\tAttempt            = 2\n\tScheduleID         = 5\n\tPageSize           = 10\n\tHistoryLength      = 20\n\tBacklogCountHint   = 30\n\tAckLevel           = 1001\n\tReadLevel          = 1002\n\tRatePerSecond      = 3.14\n\tTaskID             = 444\n\tShardID            = 12345\n\tMessageID1         = 50001\n\tMessageID2         = 50002\n\tEventStoreVersion  = 333\n\tHistorySizeInBytes = 77779\n\n\tEventID1 = int64(1)\n\tEventID2 = int64(2)\n\tEventID3 = int64(3)\n\tEventID4 = int64(4)\n\n\tVersion1 = int64(11)\n\tVersion2 = int64(22)\n\tVersion3 = int64(33)\n\n\tIsolationGroup = \"dca1\"\n)\n\nvar (\n\tTimestamp  = time.Now().Unix()\n\tTimestamp1 = Timestamp + 1\n\tTimestamp2 = Timestamp + 2\n\tTimestamp3 = Timestamp + 3\n\tTimestamp4 = Timestamp + 4\n\tTimestamp5 = Timestamp + 5\n)\n\nvar (\n\tDuration1 = int32(11)\n\tDuration2 = int32(12)\n\tDuration3 = int32(13)\n\tDuration4 = int32(14)\n)\n\nvar (\n\tToken1 = []byte{1, 0}\n\tToken2 = []byte{2, 0}\n\tToken3 = []byte{3, 0}\n)\n\nvar (\n\tPayload1 = []byte{10, 0}\n\tPayload2 = []byte{20, 0}\n\tPayload3 = []byte{30, 0}\n)\n\nvar Attempt2 = int64(2)\n\nvar (\n\tExecutionContext = []byte{110, 0}\n\tControl          = []byte{120, 0}\n\tNextPageToken    = []byte{130, 0}\n\tTaskToken        = []byte{140, 0}\n\tBranchToken      = []byte{150, 0}\n\tQueryArgs        = []byte{9, 9, 9}\n)\n\nvar (\n\tFailureReason  = \"FailureReason\"\n\tFailureDetails = []byte{190, 0}\n)\n\nvar PartitionConfig = map[string]string{\n\t\"zone\": IsolationGroup,\n}\n\nvar (\n\tHealthStatus = types.HealthStatus{\n\t\tOk:  true,\n\t\tMsg: \"HealthStatusMessage\",\n\t}\n\tWorkflowExecution = types.WorkflowExecution{\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t}\n\tWorkflowType = types.WorkflowType{\n\t\tName: WorkflowTypeName,\n\t}\n\tActivityType = types.ActivityType{\n\t\tName: ActivityTypeName,\n\t}\n\tTaskList = types.TaskList{\n\t\tName: TaskListName,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\tRetryPolicy = types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tBackoffCoefficient:          1.1,\n\t\tMaximumIntervalInSeconds:    2,\n\t\tMaximumAttempts:             3,\n\t\tNonRetriableErrorReasons:    []string{\"a\", \"b\"},\n\t\tExpirationIntervalInSeconds: 4,\n\t}\n\tHeader = types.Header{\n\t\tFields: map[string][]byte{\n\t\t\t\"HeaderField1\": {211, 0},\n\t\t\t\"HeaderField2\": {212, 0},\n\t\t},\n\t}\n\tMemo = types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t\"MemoField1\": {221, 0},\n\t\t\t\"MemoField2\": {222, 0},\n\t\t},\n\t}\n\tSearchAttributes = types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\t\"IndexedField1\": {231, 0},\n\t\t\t\"IndexedField2\": {232, 0},\n\t\t},\n\t}\n\tPayloadMap = map[string][]byte{\n\t\t\"Payload1\": Payload1,\n\t\t\"Payload2\": Payload2,\n\t}\n\tResetPointInfo = types.ResetPointInfo{\n\t\tBinaryChecksum:           Checksum,\n\t\tRunID:                    RunID1,\n\t\tFirstDecisionCompletedID: EventID1,\n\t\tCreatedTimeNano:          &Timestamp1,\n\t\tExpiringTimeNano:         &Timestamp2,\n\t\tResettable:               true,\n\t}\n\tResetPointInfoArray = []*types.ResetPointInfo{\n\t\t&ResetPointInfo,\n\t}\n\tResetPoints = types.ResetPoints{\n\t\tPoints: ResetPointInfoArray,\n\t}\n\tDataBlob = types.DataBlob{\n\t\tEncodingType: &EncodingType,\n\t\tData:         []byte{7, 7, 7},\n\t}\n\tDataBlobArray = []*types.DataBlob{\n\t\t&DataBlob,\n\t}\n\tStickyExecutionAttributes = types.StickyExecutionAttributes{\n\t\tWorkerTaskList:                &TaskList,\n\t\tScheduleToStartTimeoutSeconds: &Duration1,\n\t}\n\tActivityLocalDispatchInfo = types.ActivityLocalDispatchInfo{\n\t\tActivityID:                      ActivityID,\n\t\tScheduledTimestamp:              &Timestamp1,\n\t\tStartedTimestamp:                &Timestamp2,\n\t\tScheduledTimestampOfThisAttempt: &Timestamp3,\n\t\tTaskToken:                       TaskToken,\n\t}\n\tActivityLocalDispatchInfoMap = map[string]*types.ActivityLocalDispatchInfo{\n\t\tActivityID: &ActivityLocalDispatchInfo,\n\t}\n\tTaskListMetadata = types.TaskListMetadata{\n\t\tMaxTasksPerSecond: common.Float64Ptr(RatePerSecond),\n\t}\n\tWorkerVersionInfo = types.WorkerVersionInfo{\n\t\tImpl:           ClientImpl,\n\t\tFeatureVersion: FeatureVersion,\n\t}\n\tPollerInfo = types.PollerInfo{\n\t\tLastAccessTime: &Timestamp1,\n\t\tIdentity:       Identity,\n\t\tRatePerSecond:  RatePerSecond,\n\t}\n\tPollerInfoArray = []*types.PollerInfo{\n\t\t&PollerInfo,\n\t}\n\tTaskListStatus = types.TaskListStatus{\n\t\tBacklogCountHint: BacklogCountHint,\n\t\tReadLevel:        ReadLevel,\n\t\tAckLevel:         AckLevel,\n\t\tRatePerSecond:    RatePerSecond,\n\t\tTaskIDBlock:      &TaskIDBlock,\n\t\tIsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{\n\t\t\t\"dca\": {\n\t\t\t\tNewTasksPerSecond: 10,\n\t\t\t\tPollerCount:       1,\n\t\t\t},\n\t\t},\n\t\tNewTasksPerSecond: 10,\n\t\tEmpty:             true,\n\t}\n\tTaskIDBlock = types.TaskIDBlock{\n\t\tStartID: 551,\n\t\tEndID:   559,\n\t}\n\tTaskListPartitionMetadata = types.TaskListPartitionMetadata{\n\t\tKey:           \"Key\",\n\t\tOwnerHostName: \"HostName\",\n\t}\n\tTaskListPartitionMetadataArray = []*types.TaskListPartitionMetadata{\n\t\t&TaskListPartitionMetadata,\n\t}\n\tSupportedClientVersions = types.SupportedClientVersions{\n\t\tGoSdk:   \"GoSDK\",\n\t\tJavaSdk: \"JavaSDK\",\n\t}\n\tIndexedValueTypeMap = map[string]types.IndexedValueType{\n\t\t\"IndexedValueType1\": IndexedValueType,\n\t}\n\tIsolationGroupConfiguration = types.IsolationGroupConfiguration{\n\t\tIsolationGroup: {\n\t\t\tName:  IsolationGroup,\n\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t},\n\t\t\"zone 1\": {\n\t\t\tName:  \"zone 1\",\n\t\t\tState: types.IsolationGroupStateDrained,\n\t\t},\n\t}\n\tParentExecutionInfo = types.ParentExecutionInfo{\n\t\tDomainUUID:  DomainID,\n\t\tDomain:      DomainName,\n\t\tExecution:   &WorkflowExecution,\n\t\tInitiatedID: EventID1,\n\t}\n\tClusterInfo = types.ClusterInfo{\n\t\tSupportedClientVersions: &SupportedClientVersions,\n\t}\n\tMembershipInfo = types.MembershipInfo{\n\t\tCurrentHost:      &HostInfo,\n\t\tReachableMembers: []string{HostName},\n\t\tRings:            RingInfoArray,\n\t}\n\tHostInfo = types.HostInfo{\n\t\tIdentity: HostName,\n\t}\n\tHostInfoArray = []*types.HostInfo{\n\t\t&HostInfo,\n\t}\n\tRingInfo = types.RingInfo{\n\t\tRole:        \"Role\",\n\t\tMemberCount: 1,\n\t\tMembers:     HostInfoArray,\n\t}\n\tRingInfoArray = []*types.RingInfo{\n\t\t&RingInfo,\n\t}\n\tDomainCacheInfo = types.DomainCacheInfo{\n\t\tNumOfItemsInCacheByID:   int64(2000),\n\t\tNumOfItemsInCacheByName: int64(3000),\n\t}\n\tVersionHistoryItem = types.VersionHistoryItem{\n\t\tEventID: EventID1,\n\t\tVersion: Version1,\n\t}\n\tVersionHistoryItemArray = []*types.VersionHistoryItem{\n\t\t&VersionHistoryItem,\n\t}\n\tVersionHistory = types.VersionHistory{\n\t\tBranchToken: BranchToken,\n\t\tItems:       VersionHistoryItemArray,\n\t}\n\tVersionHistoryArray = []*types.VersionHistory{\n\t\t&VersionHistory,\n\t}\n\tVersionHistories = types.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 1,\n\t\tHistories:                  VersionHistoryArray,\n\t}\n\tTransientDecisionInfo = types.TransientDecisionInfo{\n\t\tScheduledEvent: &HistoryEvent_WorkflowExecutionStarted,\n\t\tStartedEvent:   &HistoryEvent_WorkflowExecutionStarted,\n\t}\n\tFailoverMarkerToken = types.FailoverMarkerToken{\n\t\tShardIDs:       []int32{ShardID},\n\t\tFailoverMarker: &FailoverMarkerAttributes,\n\t}\n\tFailoverMarkerTokenArray = []*types.FailoverMarkerToken{\n\t\t&FailoverMarkerToken,\n\t}\n\tWorkflowExecutionFilter = types.WorkflowExecutionFilter{\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t}\n\tWorkflowTypeFilter = types.WorkflowTypeFilter{\n\t\tName: WorkflowTypeName,\n\t}\n\tStartTimeFilter = types.StartTimeFilter{\n\t\tEarliestTime: &Timestamp1,\n\t\tLatestTime:   &Timestamp2,\n\t}\n\tWorkflowExecutionInfo = types.WorkflowExecutionInfo{\n\t\tExecution:         &WorkflowExecution,\n\t\tType:              &WorkflowType,\n\t\tStartTime:         &Timestamp1,\n\t\tCloseTime:         &Timestamp2,\n\t\tCloseStatus:       &WorkflowExecutionCloseStatus,\n\t\tHistoryLength:     HistoryLength,\n\t\tParentDomainID:    common.StringPtr(DomainID),\n\t\tParentDomain:      common.StringPtr(DomainName),\n\t\tParentExecution:   &WorkflowExecution,\n\t\tParentInitiatedID: common.Int64Ptr(EventID1),\n\t\tExecutionTime:     &Timestamp3,\n\t\tMemo:              &Memo,\n\t\tSearchAttributes:  &SearchAttributes,\n\t\tAutoResetPoints:   &ResetPoints,\n\t\tTaskList:          &types.TaskList{Name: TaskListName, Kind: types.TaskListKindNormal.Ptr()},\n\t\tPartitionConfig:   PartitionConfig,\n\t\tCronOverlapPolicy: &CronOverlapPolicy,\n\t}\n\tWorkflowExecutionInfoEphemeral = types.WorkflowExecutionInfo{\n\t\tExecution:         &WorkflowExecution,\n\t\tType:              &WorkflowType,\n\t\tStartTime:         &Timestamp1,\n\t\tCloseTime:         &Timestamp2,\n\t\tCloseStatus:       &WorkflowExecutionCloseStatus,\n\t\tHistoryLength:     HistoryLength,\n\t\tParentDomainID:    common.StringPtr(DomainID),\n\t\tParentDomain:      common.StringPtr(DomainName),\n\t\tParentExecution:   &WorkflowExecution,\n\t\tParentInitiatedID: common.Int64Ptr(EventID1),\n\t\tExecutionTime:     &Timestamp3,\n\t\tMemo:              &Memo,\n\t\tSearchAttributes:  &SearchAttributes,\n\t\tAutoResetPoints:   &ResetPoints,\n\t\tTaskList:          &types.TaskList{Name: TaskListName, Kind: types.TaskListKindEphemeral.Ptr()},\n\t\tPartitionConfig:   PartitionConfig,\n\t\tCronOverlapPolicy: &CronOverlapPolicy,\n\t}\n\tCronWorkflowExecutionInfo = types.WorkflowExecutionInfo{\n\t\tExecution:         &WorkflowExecution,\n\t\tType:              &WorkflowType,\n\t\tStartTime:         &Timestamp1,\n\t\tCloseTime:         &Timestamp2,\n\t\tCloseStatus:       &WorkflowExecutionCloseStatus,\n\t\tHistoryLength:     HistoryLength,\n\t\tParentDomainID:    common.StringPtr(DomainID),\n\t\tParentDomain:      common.StringPtr(DomainName),\n\t\tParentExecution:   &WorkflowExecution,\n\t\tParentInitiatedID: common.Int64Ptr(EventID1),\n\t\tExecutionTime:     &Timestamp3,\n\t\tMemo:              &Memo,\n\t\tSearchAttributes:  &SearchAttributes,\n\t\tAutoResetPoints:   &ResetPoints,\n\t\tTaskList:          &types.TaskList{Name: TaskListName, Kind: types.TaskListKindNormal.Ptr()},\n\t\tPartitionConfig:   PartitionConfig,\n\t\tIsCron:            true,\n\t\tCronOverlapPolicy: &CronOverlapPolicy,\n\t}\n\tWorkflowExecutionInfoArray = []*types.WorkflowExecutionInfo{&WorkflowExecutionInfo}\n\n\tWorkflowQuery = types.WorkflowQuery{\n\t\tQueryType: QueryType,\n\t\tQueryArgs: QueryArgs,\n\t}\n\tWorkflowQueryMap = map[string]*types.WorkflowQuery{\n\t\t\"WorkflowQuery1\": &WorkflowQuery,\n\t}\n\tWorkflowQueryResult = types.WorkflowQueryResult{\n\t\tResultType:   &QueryResultType,\n\t\tAnswer:       Payload1,\n\t\tErrorMessage: ErrorMessage,\n\t}\n\tWorkflowQueryResultMap = map[string]*types.WorkflowQueryResult{\n\t\t\"WorkflowQuery1\": &WorkflowQueryResult,\n\t}\n\tQueryRejected = types.QueryRejected{\n\t\tCloseStatus: &WorkflowExecutionCloseStatus,\n\t}\n\tWorkflowExecutionConfiguration = types.WorkflowExecutionConfiguration{\n\t\tTaskList:                            &TaskList,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t}\n\tPendingActivityInfo = types.PendingActivityInfo{\n\t\tActivityID:             ActivityID,\n\t\tActivityType:           &ActivityType,\n\t\tState:                  &PendingActivityState,\n\t\tHeartbeatDetails:       Payload1,\n\t\tLastHeartbeatTimestamp: &Timestamp1,\n\t\tLastStartedTimestamp:   &Timestamp2,\n\t\tAttempt:                Attempt,\n\t\tMaximumAttempts:        3,\n\t\tScheduledTimestamp:     &Timestamp3,\n\t\tExpirationTimestamp:    &Timestamp4,\n\t\tLastFailureReason:      &FailureReason,\n\t\tLastWorkerIdentity:     Identity,\n\t\tLastFailureDetails:     FailureDetails,\n\t\tStartedWorkerIdentity:  Identity,\n\t\tScheduleID:             ScheduleID,\n\t}\n\tPendingActivityInfoArray = []*types.PendingActivityInfo{\n\t\t&PendingActivityInfo,\n\t}\n\tPendingChildExecutionInfo = types.PendingChildExecutionInfo{\n\t\tDomain:            DomainName,\n\t\tWorkflowID:        WorkflowID,\n\t\tRunID:             RunID,\n\t\tWorkflowTypeName:  WorkflowTypeName,\n\t\tInitiatedID:       EventID1,\n\t\tParentClosePolicy: &ParentClosePolicy,\n\t}\n\tPendingChildExecutionInfoArray = []*types.PendingChildExecutionInfo{\n\t\t&PendingChildExecutionInfo,\n\t}\n\tPendingDecisionInfo = types.PendingDecisionInfo{\n\t\tState:                      &PendingDecisionState,\n\t\tScheduledTimestamp:         &Timestamp1,\n\t\tStartedTimestamp:           &Timestamp2,\n\t\tAttempt:                    Attempt,\n\t\tOriginalScheduledTimestamp: &Timestamp3,\n\t\tScheduleID:                 ScheduleID,\n\t}\n\tAutoConfigHint = types.AutoConfigHint{\n\t\tEnableAutoConfig:   false,\n\t\tPollerWaitTimeInMs: 10,\n\t}\n\tTaskKey = types.TaskKey{\n\t\tTaskID:            TaskID,\n\t\tScheduledTimeNano: Timestamp1,\n\t}\n\tTaskKey2 = types.TaskKey{\n\t\tTaskID:            TaskID + 1,\n\t\tScheduledTimeNano: Timestamp2,\n\t}\n\tTaskRange = types.TaskRange{\n\t\tInclusiveMin: &TaskKey,\n\t\tExclusiveMax: &TaskKey2,\n\t}\n\tVirtualSliceState = types.VirtualSliceState{\n\t\tTaskRange: &TaskRange,\n\t}\n\tVirtualQueueState = types.VirtualQueueState{\n\t\tVirtualSliceStates: []*types.VirtualSliceState{&VirtualSliceState},\n\t}\n\tQueueState = types.QueueState{\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t0: &VirtualQueueState,\n\t\t},\n\t\tExclusiveMaxReadLevel: &TaskKey,\n\t}\n\tActiveClusterSelectionPolicyRegionSticky = types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\tStickyRegion:                   \"region1\",\n\t}\n\tActiveClusterSelectionPolicyExternalEntity = types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyExternalEntity.Ptr(),\n\t\tExternalEntityType:             \"externalEntityType1\",\n\t\tExternalEntityKey:              \"externalEntityKey1\",\n\t}\n\tClusterAttribute = types.ClusterAttribute{\n\t\tScope: \"region\",\n\t\tName:  \"us-west-1\",\n\t}\n\tActiveClusterSelectionPolicyWithClusterAttribute = types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &ClusterAttribute,\n\t}\n\tCronOverlapPolicy = types.CronOverlapPolicySkipped\n)\n"
  },
  {
    "path": "common/types/testdata/config_store.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tDynamicConfigValueName         = \"test_config\"\n\tDynamicConfigFilterName        = \"test_filter\"\n\tDynamicConfigEntryName         = \"test_entry\"\n\tDynamicConfigBlobSchemaVersion = int64(1)\n)\n\nvar (\n\tDynamicConfigFilter = types.DynamicConfigFilter{\n\t\tName:  DynamicConfigFilterName,\n\t\tValue: &DataBlob,\n\t}\n\tDynamicConfigValue = types.DynamicConfigValue{\n\t\tValue:   &DataBlob,\n\t\tFilters: []*types.DynamicConfigFilter{&DynamicConfigFilter},\n\t}\n\tDynamicConfigEntry = types.DynamicConfigEntry{\n\t\tName:   DynamicConfigEntryName,\n\t\tValues: []*types.DynamicConfigValue{&DynamicConfigValue},\n\t}\n\tDynamicConfigBlob = types.DynamicConfigBlob{\n\t\tSchemaVersion: DynamicConfigBlobSchemaVersion,\n\t\tEntries:       []*types.DynamicConfigEntry{&DynamicConfigEntry},\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/decision.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tDecisionArray = []*types.Decision{\n\t\t&Decision_CancelTimer,\n\t\t&Decision_CancelWorkflowExecution,\n\t\t&Decision_CompleteWorkflowExecution,\n\t\t&Decision_ContinueAsNewWorkflowExecution,\n\t\t&Decision_FailWorkflowExecution,\n\t\t&Decision_RecordMarker,\n\t\t&Decision_RequestCancelActivityTask,\n\t\t&Decision_RequestCancelExternalWorkflowExecution,\n\t\t&Decision_ScheduleActivityTask,\n\t\t&Decision_SignalExternalWorkflowExecution,\n\t\t&Decision_StartChildWorkflowExecution,\n\t\t&Decision_StartTimer,\n\t\t&Decision_UpsertWorkflowSearchAttributes,\n\t}\n\n\tDecision_CancelTimer = types.Decision{\n\t\tDecisionType:                  types.DecisionTypeCancelTimer.Ptr(),\n\t\tCancelTimerDecisionAttributes: &CancelTimerDecisionAttributes,\n\t}\n\tDecision_CancelWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeCancelWorkflowExecution.Ptr(),\n\t\tCancelWorkflowExecutionDecisionAttributes: &CancelWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_CompleteWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\tCompleteWorkflowExecutionDecisionAttributes: &CompleteWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_ContinueAsNewWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &ContinueAsNewWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_FailWorkflowExecution = types.Decision{\n\t\tDecisionType:                            types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\tFailWorkflowExecutionDecisionAttributes: &FailWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_RecordMarker = types.Decision{\n\t\tDecisionType:                   types.DecisionTypeRecordMarker.Ptr(),\n\t\tRecordMarkerDecisionAttributes: &RecordMarkerDecisionAttributes,\n\t}\n\tDecision_RequestCancelActivityTask = types.Decision{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &RequestCancelActivityTaskDecisionAttributes,\n\t}\n\tDecision_RequestCancelExternalWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeRequestCancelExternalWorkflowExecution.Ptr(),\n\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: &RequestCancelExternalWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_ScheduleActivityTask = types.Decision{\n\t\tDecisionType:                           types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\tScheduleActivityTaskDecisionAttributes: &ScheduleActivityTaskDecisionAttributes,\n\t}\n\tDecision_SignalExternalWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes: &SignalExternalWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_StartChildWorkflowExecution = types.Decision{\n\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\tStartChildWorkflowExecutionDecisionAttributes: &StartChildWorkflowExecutionDecisionAttributes,\n\t}\n\tDecision_StartTimer = types.Decision{\n\t\tDecisionType:                 types.DecisionTypeStartTimer.Ptr(),\n\t\tStartTimerDecisionAttributes: &StartTimerDecisionAttributes,\n\t}\n\tDecision_UpsertWorkflowSearchAttributes = types.Decision{\n\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &UpsertWorkflowSearchAttributesDecisionAttributes,\n\t}\n\n\tCancelTimerDecisionAttributes = types.CancelTimerDecisionAttributes{\n\t\tTimerID: TimerID,\n\t}\n\tCancelWorkflowExecutionDecisionAttributes = types.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: Payload1,\n\t}\n\tCompleteWorkflowExecutionDecisionAttributes = types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: Payload1,\n\t}\n\tContinueAsNewWorkflowExecutionDecisionAttributes = types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tBackoffStartIntervalInSeconds:       &Duration3,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tInitiator:                           &ContinueAsNewInitiator,\n\t\tFailureReason:                       &FailureReason,\n\t\tFailureDetails:                      FailureDetails,\n\t\tLastCompletionResult:                Payload2,\n\t\tCronSchedule:                        CronSchedule,\n\t\tHeader:                              &Header,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tJitterStartSeconds:                  &Duration1,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t}\n\tFailWorkflowExecutionDecisionAttributes = types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  &FailureReason,\n\t\tDetails: FailureDetails,\n\t}\n\tRecordMarkerDecisionAttributes = types.RecordMarkerDecisionAttributes{\n\t\tMarkerName: MarkerName,\n\t\tDetails:    Payload1,\n\t\tHeader:     &Header,\n\t}\n\tRequestCancelActivityTaskDecisionAttributes = types.RequestCancelActivityTaskDecisionAttributes{\n\t\tActivityID: ActivityID,\n\t}\n\tRequestCancelExternalWorkflowExecutionDecisionAttributes = types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            DomainName,\n\t\tWorkflowID:        WorkflowID,\n\t\tRunID:             RunID,\n\t\tControl:           Control,\n\t\tChildWorkflowOnly: true,\n\t}\n\tScheduleActivityTaskDecisionAttributes = types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    ActivityID,\n\t\tActivityType:                  &ActivityType,\n\t\tDomain:                        DomainName,\n\t\tTaskList:                      &TaskList,\n\t\tInput:                         Payload1,\n\t\tScheduleToCloseTimeoutSeconds: &Duration1,\n\t\tScheduleToStartTimeoutSeconds: &Duration2,\n\t\tStartToCloseTimeoutSeconds:    &Duration3,\n\t\tHeartbeatTimeoutSeconds:       &Duration4,\n\t\tRetryPolicy:                   &RetryPolicy,\n\t\tHeader:                        &Header,\n\t\tRequestLocalDispatch:          true,\n\t}\n\tSignalExternalWorkflowExecutionDecisionAttributes = types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain:            DomainName,\n\t\tExecution:         &WorkflowExecution,\n\t\tSignalName:        SignalName,\n\t\tInput:             Payload1,\n\t\tControl:           Control,\n\t\tChildWorkflowOnly: true,\n\t}\n\tStartChildWorkflowExecutionDecisionAttributes = types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                              DomainName,\n\t\tWorkflowID:                          WorkflowID,\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tParentClosePolicy:                   &ParentClosePolicy,\n\t\tControl:                             Control,\n\t\tWorkflowIDReusePolicy:               &WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tCronSchedule:                        CronSchedule,\n\t\tHeader:                              &Header,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t}\n\tStartTimerDecisionAttributes = types.StartTimerDecisionAttributes{\n\t\tTimerID:                   TimerID,\n\t\tStartToFireTimeoutSeconds: common.Int64Ptr(int64(Duration1)),\n\t}\n\tUpsertWorkflowSearchAttributesDecisionAttributes = types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\tSearchAttributes: &SearchAttributes,\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/domain.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport \"github.com/uber/cadence/common/types\"\n\nconst (\n\tDomainID          = \"DomainID\"\n\tDomainName        = \"DomainName\"\n\tDomainDescription = \"DomainDescription\"\n\tDomainOwnerEmail  = \"DomainOwnerEmail\"\n\tDomainDataKey     = \"DomainDataKey\"\n\tDomainDataValue   = \"DomainDataValue\"\n\tDomainRetention   = 3\n\tDomainEmitMetric  = true\n\n\tClusterName1 = \"ClusterName1\"\n\tClusterName2 = \"ClusterName2\"\n\n\tRegion1 = \"Region1\"\n\tRegion2 = \"Region2\"\n\n\tBadBinaryReason   = \"BadBinaryReason\"\n\tBadBinaryOperator = \"BadBinaryOperator\"\n\n\tHistoryArchivalURI    = \"HistoryArchivalURI\"\n\tVisibilityArchivalURI = \"VisibilityArchivalURI\"\n\n\tDeleteBadBinary = \"DeleteBadBinary\"\n\n\tFailoverVersion1 = 301\n\tFailoverVersion2 = 302\n\n\tErrorReason = \"ErrorReason\"\n\n\tFailoverEventID = \"failover-event-123\"\n\tFailoverTime    = int64(1234567890000000000)\n)\n\nvar (\n\tBadBinaryInfo = types.BadBinaryInfo{\n\t\tReason:          BadBinaryReason,\n\t\tOperator:        BadBinaryOperator,\n\t\tCreatedTimeNano: &Timestamp1,\n\t}\n\tBadBinaryInfoMap = map[string]*types.BadBinaryInfo{\n\t\t\"BadBinary1\": &BadBinaryInfo,\n\t}\n\tBadBinaries = types.BadBinaries{\n\t\tBinaries: BadBinaryInfoMap,\n\t}\n\tDomainData = map[string]string{DomainDataKey: DomainDataValue}\n\tDomainInfo = types.DomainInfo{\n\t\tName:        DomainName,\n\t\tStatus:      &DomainStatus,\n\t\tDescription: DomainDescription,\n\t\tOwnerEmail:  DomainOwnerEmail,\n\t\tData:        DomainData,\n\t\tUUID:        DomainID,\n\t}\n\tDomainConfiguration = types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: DomainRetention,\n\t\tEmitMetric:                             DomainEmitMetric,\n\t\tBadBinaries:                            &BadBinaries,\n\t\tHistoryArchivalStatus:                  &ArchivalStatus,\n\t\tHistoryArchivalURI:                     HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:               &ArchivalStatus,\n\t\tVisibilityArchivalURI:                  VisibilityArchivalURI,\n\t\tIsolationGroups: &types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": {\n\t\t\t\tName:  \"zone-1\",\n\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t},\n\t\t\t\"zone-2\": {\n\t\t\t\tName:  \"zone-2\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t},\n\t\tAsyncWorkflowConfig: &types.AsyncWorkflowConfiguration{\n\t\t\tEnabled:   true,\n\t\t\tQueueType: \"custom\",\n\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\tData:         []byte(\"custom queue config\"),\n\t\t\t},\n\t\t},\n\t}\n\tDomainReplicationConfiguration = types.DomainReplicationConfiguration{\n\t\tActiveClusterName: ClusterName1,\n\t\tClusters:          ClusterReplicationConfigurationArray,\n\t\tActiveClusters: &types.ActiveClusters{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\tRegion1: {\n\t\t\t\t\t\t\tActiveClusterName: ClusterName1,\n\t\t\t\t\t\t\tFailoverVersion:   FailoverVersion1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tRegion2: {\n\t\t\t\t\t\t\tActiveClusterName: ClusterName2,\n\t\t\t\t\t\t\tFailoverVersion:   FailoverVersion2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tClusterReplicationConfiguration = types.ClusterReplicationConfiguration{\n\t\tClusterName: ClusterName1,\n\t}\n\tClusterReplicationConfigurationArray = []*types.ClusterReplicationConfiguration{\n\t\t&ClusterReplicationConfiguration,\n\t}\n\tFailoverInfo = types.FailoverInfo{\n\t\tFailoverVersion:         1,\n\t\tFailoverStartTimestamp:  1,\n\t\tFailoverExpireTimestamp: 10,\n\t\tCompletedShardCount:     10,\n\t\tPendingShards:           []int32{1, 2, 3},\n\t}\n\tActiveClusterInfo1 = types.ActiveClusterInfo{\n\t\tActiveClusterName: ClusterName1,\n\t\tFailoverVersion:   FailoverVersion1,\n\t}\n\tActiveClusterInfo2 = types.ActiveClusterInfo{\n\t\tActiveClusterName: ClusterName2,\n\t\tFailoverVersion:   FailoverVersion2,\n\t}\n\tClusterAttributeFailover = types.ClusterAttribute{\n\t\tScope: \"region\",\n\t\tName:  Region1,\n\t}\n\tClusterFailover = types.ClusterFailover{\n\t\tFromCluster:      &ActiveClusterInfo1,\n\t\tToCluster:        &ActiveClusterInfo2,\n\t\tClusterAttribute: &ClusterAttributeFailover,\n\t}\n\tFailoverEventIDPtr = FailoverEventID\n\tFailoverTimePtr    = FailoverTime\n\tFailoverEvent      = types.FailoverEvent{\n\t\tID:               &FailoverEventIDPtr,\n\t\tCreatedTime:      &FailoverTimePtr,\n\t\tFailoverType:     types.FailoverTypeForce.Ptr(),\n\t\tClusterFailovers: []*types.ClusterFailover{&ClusterFailover},\n\t}\n\tListFailoverHistoryRequestFilters = types.ListFailoverHistoryRequestFilters{\n\t\tDomainID: DomainID,\n\t}\n\tPaginationOptions = types.PaginationOptions{\n\t\tPageSize:      &[]int32{10}[0],\n\t\tNextPageToken: []byte(\"next-page-token\"),\n\t}\n\tListFailoverHistoryRequest = types.ListFailoverHistoryRequest{\n\t\tFilters:    &ListFailoverHistoryRequestFilters,\n\t\tPagination: &PaginationOptions,\n\t}\n\tListFailoverHistoryResponse = types.ListFailoverHistoryResponse{\n\t\tFailoverEvents: []*types.FailoverEvent{&FailoverEvent},\n\t\tNextPageToken:  []byte(\"next-page-token\"),\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/enum.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport \"github.com/uber/cadence/common/types\"\n\nvar (\n\tArchivalStatus                             = types.ArchivalStatusEnabled\n\tCancelExternalWorkflowExecutionFailedCause = types.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\tChildWorkflowExecutionFailedCause          = types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning\n\tContinueAsNewInitiator                     = types.ContinueAsNewInitiatorRetryPolicy\n\tDecisionTaskFailedCause                    = types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes\n\tDecisionTaskTimedOutCause                  = types.DecisionTaskTimedOutCauseReset\n\tDecisionType                               = types.DecisionTypeCancelTimer\n\tDomainStatus                               = types.DomainStatusRegistered\n\tEncodingType                               = types.EncodingTypeJSON\n\tEventType                                  = types.EventTypeWorkflowExecutionStarted\n\tHistoryEventFilterType                     = types.HistoryEventFilterTypeCloseEvent\n\tIndexedValueType                           = types.IndexedValueTypeInt\n\tParentClosePolicy                          = types.ParentClosePolicyTerminate\n\tParentClosePolicy2                         = types.ParentClosePolicyRequestCancel\n\tPendingActivityState                       = types.PendingActivityStateCancelRequested\n\tPendingDecisionState                       = types.PendingDecisionStateStarted\n\tQueryConsistencyLevel                      = types.QueryConsistencyLevelStrong\n\tQueryRejectCondition                       = types.QueryRejectConditionNotCompletedCleanly\n\tQueryResultType                            = types.QueryResultTypeFailed\n\tQueryTaskCompletedType                     = types.QueryTaskCompletedTypeFailed\n\tSignalExternalWorkflowExecutionFailedCause = types.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution\n\tTaskListKind                               = types.TaskListKindSticky\n\tTaskListType                               = types.TaskListTypeActivity\n\tTimeoutType                                = types.TimeoutTypeScheduleToStart\n\tWorkflowExecutionCloseStatus               = types.WorkflowExecutionCloseStatusContinuedAsNew\n\tWorkflowIDReusePolicy                      = types.WorkflowIDReusePolicyTerminateIfRunning\n)\n"
  },
  {
    "path": "common/types/testdata/error.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tErrorMessage = \"ErrorMessage\"\n)\n\nvar (\n\tAccessDeniedError = types.AccessDeniedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tBadRequestError = types.BadRequestError{\n\t\tMessage: ErrorMessage,\n\t}\n\tCancellationAlreadyRequestedError = types.CancellationAlreadyRequestedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tClientVersionNotSupportedError = types.ClientVersionNotSupportedError{\n\t\tFeatureVersion:    FeatureVersion,\n\t\tClientImpl:        ClientImpl,\n\t\tSupportedVersions: SupportedVersions,\n\t}\n\tFeatureNotEnabledError = types.FeatureNotEnabledError{\n\t\tFeatureFlag: FeatureFlag,\n\t}\n\tCurrentBranchChangedError = types.CurrentBranchChangedError{\n\t\tMessage:            ErrorMessage,\n\t\tCurrentBranchToken: BranchToken,\n\t}\n\tDomainAlreadyExistsError = types.DomainAlreadyExistsError{\n\t\tMessage: ErrorMessage,\n\t}\n\tDomainNotActiveError = types.DomainNotActiveError{\n\t\tMessage:        ErrorMessage,\n\t\tDomainName:     DomainName,\n\t\tCurrentCluster: ClusterName1,\n\t\tActiveCluster:  ClusterName2,\n\t\tActiveClusters: []string{ClusterName2},\n\t}\n\tEntityNotExistsError = types.EntityNotExistsError{\n\t\tMessage:        ErrorMessage,\n\t\tCurrentCluster: ClusterName1,\n\t\tActiveCluster:  ClusterName2,\n\t\tActiveClusters: []string{ClusterName2},\n\t}\n\tWorkflowExecutionAlreadyCompletedError = types.WorkflowExecutionAlreadyCompletedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tEventAlreadyStartedError = types.EventAlreadyStartedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tInternalDataInconsistencyError = types.InternalDataInconsistencyError{\n\t\tMessage: ErrorMessage,\n\t}\n\tInternalServiceError = types.InternalServiceError{\n\t\tMessage: ErrorMessage,\n\t}\n\tLimitExceededError = types.LimitExceededError{\n\t\tMessage: ErrorMessage,\n\t}\n\tQueryFailedError = types.QueryFailedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tRemoteSyncMatchedError = types.RemoteSyncMatchedError{\n\t\tMessage: ErrorMessage,\n\t}\n\tRetryTaskV2Error = types.RetryTaskV2Error{\n\t\tMessage:           ErrorMessage,\n\t\tDomainID:          DomainID,\n\t\tWorkflowID:        WorkflowID,\n\t\tRunID:             RunID,\n\t\tStartEventID:      common.Int64Ptr(EventID1),\n\t\tStartEventVersion: common.Int64Ptr(Version1),\n\t\tEndEventID:        common.Int64Ptr(EventID2),\n\t\tEndEventVersion:   common.Int64Ptr(Version2),\n\t}\n\tServiceBusyError = types.ServiceBusyError{\n\t\tMessage: ErrorMessage,\n\t\tReason:  ErrorReason,\n\t}\n\tShardOwnershipLostError = types.ShardOwnershipLostError{\n\t\tMessage: ErrorMessage,\n\t\tOwner:   HostName,\n\t}\n\tWorkflowExecutionAlreadyStartedError = types.WorkflowExecutionAlreadyStartedError{\n\t\tMessage:        ErrorMessage,\n\t\tStartRequestID: RequestID,\n\t\tRunID:          RunID,\n\t}\n\tStickyWorkerUnavailableError = types.StickyWorkerUnavailableError{\n\t\tMessage: ErrorMessage,\n\t}\n\tReadOnlyPartitionError = types.ReadOnlyPartitionError{\n\t\tMessage: ErrorMessage,\n\t}\n\tTaskListNotOwnedByHostError = cadence_errors.TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: HostName,\n\t\tMyIdentity:      HostName2,\n\t\tTasklistName:    TaskListName,\n\t}\n\tNamespaceNotFoundError = types.NamespaceNotFoundError{\n\t\tNamespace: Namespace,\n\t}\n\tShardNotFoundError = types.ShardNotFoundError{\n\t\tNamespace: Namespace,\n\t\tShardKey:  ShardKey,\n\t}\n)\n\nvar Errors = []error{\n\t&AccessDeniedError,\n\t&BadRequestError,\n\t&CancellationAlreadyRequestedError,\n\t&ClientVersionNotSupportedError,\n\t&FeatureNotEnabledError,\n\t&CurrentBranchChangedError,\n\t&DomainAlreadyExistsError,\n\t&DomainNotActiveError,\n\t&EntityNotExistsError,\n\t&WorkflowExecutionAlreadyCompletedError,\n\t&EventAlreadyStartedError,\n\t&InternalDataInconsistencyError,\n\t&InternalServiceError,\n\t&LimitExceededError,\n\t&QueryFailedError,\n\t&RemoteSyncMatchedError,\n\t&RetryTaskV2Error,\n\t&ServiceBusyError,\n\t&ShardOwnershipLostError,\n\t&WorkflowExecutionAlreadyStartedError,\n\t&StickyWorkerUnavailableError,\n\t&ReadOnlyPartitionError,\n\t&TaskListNotOwnedByHostError,\n\t&NamespaceNotFoundError,\n\t&ShardNotFoundError,\n}\n"
  },
  {
    "path": "common/types/testdata/history.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tHistory = types.History{\n\t\tEvents: HistoryEventArray,\n\t}\n\tHistoryEventArray = []*types.HistoryEvent{\n\t\t&HistoryEvent_WorkflowExecutionStarted,\n\t\t&HistoryEvent_WorkflowExecutionStarted,\n\t\t&HistoryEvent_WorkflowExecutionCompleted,\n\t\t&HistoryEvent_WorkflowExecutionFailed,\n\t\t&HistoryEvent_WorkflowExecutionTimedOut,\n\t\t&HistoryEvent_DecisionTaskScheduled,\n\t\t&HistoryEvent_DecisionTaskStarted,\n\t\t&HistoryEvent_DecisionTaskCompleted,\n\t\t&HistoryEvent_DecisionTaskTimedOut,\n\t\t&HistoryEvent_DecisionTaskFailed,\n\t\t&HistoryEvent_ActivityTaskScheduled,\n\t\t&HistoryEvent_ActivityTaskStarted,\n\t\t&HistoryEvent_ActivityTaskCompleted,\n\t\t&HistoryEvent_ActivityTaskFailed,\n\t\t&HistoryEvent_ActivityTaskTimedOut,\n\t\t&HistoryEvent_ActivityTaskCancelRequested,\n\t\t&HistoryEvent_RequestCancelActivityTaskFailed,\n\t\t&HistoryEvent_ActivityTaskCanceled,\n\t\t&HistoryEvent_TimerStarted,\n\t\t&HistoryEvent_TimerFired,\n\t\t&HistoryEvent_CancelTimerFailed,\n\t\t&HistoryEvent_TimerCanceled,\n\t\t&HistoryEvent_WorkflowExecutionCancelRequested,\n\t\t&HistoryEvent_WorkflowExecutionCanceled,\n\t\t&HistoryEvent_RequestCancelExternalWorkflowExecutionInitiated,\n\t\t&HistoryEvent_RequestCancelExternalWorkflowExecutionFailed,\n\t\t&HistoryEvent_ExternalWorkflowExecutionCancelRequested,\n\t\t&HistoryEvent_MarkerRecorded,\n\t\t&HistoryEvent_WorkflowExecutionSignaled,\n\t\t&HistoryEvent_WorkflowExecutionTerminated,\n\t\t&HistoryEvent_WorkflowExecutionContinuedAsNew,\n\t\t&HistoryEvent_StartChildWorkflowExecutionInitiated,\n\t\t&HistoryEvent_StartChildWorkflowExecutionFailed,\n\t\t&HistoryEvent_ChildWorkflowExecutionStarted,\n\t\t&HistoryEvent_ChildWorkflowExecutionCompleted,\n\t\t&HistoryEvent_ChildWorkflowExecutionFailed,\n\t\t&HistoryEvent_ChildWorkflowExecutionCanceled,\n\t\t&HistoryEvent_ChildWorkflowExecutionTimedOut,\n\t\t&HistoryEvent_ChildWorkflowExecutionTerminated,\n\t\t&HistoryEvent_SignalExternalWorkflowExecutionInitiated,\n\t\t&HistoryEvent_SignalExternalWorkflowExecutionFailed,\n\t\t&HistoryEvent_ExternalWorkflowExecutionSignaled,\n\t\t&HistoryEvent_UpsertWorkflowSearchAttributes,\n\t}\n\n\tHistoryEvent_WorkflowExecutionStarted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionStarted.Ptr()\n\t\te.WorkflowExecutionStartedEventAttributes = &WorkflowExecutionStartedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionCompleted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionCompleted.Ptr()\n\t\te.WorkflowExecutionCompletedEventAttributes = &WorkflowExecutionCompletedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionFailed.Ptr()\n\t\te.WorkflowExecutionFailedEventAttributes = &WorkflowExecutionFailedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionTimedOut = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionTimedOut.Ptr()\n\t\te.WorkflowExecutionTimedOutEventAttributes = &WorkflowExecutionTimedOutEventAttributes\n\t})\n\tHistoryEvent_DecisionTaskScheduled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeDecisionTaskScheduled.Ptr()\n\t\te.DecisionTaskScheduledEventAttributes = &DecisionTaskScheduledEventAttributes\n\t})\n\tHistoryEvent_DecisionTaskStarted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeDecisionTaskStarted.Ptr()\n\t\te.DecisionTaskStartedEventAttributes = &DecisionTaskStartedEventAttributes\n\t})\n\tHistoryEvent_DecisionTaskCompleted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeDecisionTaskCompleted.Ptr()\n\t\te.DecisionTaskCompletedEventAttributes = &DecisionTaskCompletedEventAttributes\n\t})\n\tHistoryEvent_DecisionTaskTimedOut = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeDecisionTaskTimedOut.Ptr()\n\t\te.DecisionTaskTimedOutEventAttributes = &DecisionTaskTimedOutEventAttributes\n\t})\n\tHistoryEvent_DecisionTaskFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeDecisionTaskFailed.Ptr()\n\t\te.DecisionTaskFailedEventAttributes = &DecisionTaskFailedEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskScheduled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskScheduled.Ptr()\n\t\te.ActivityTaskScheduledEventAttributes = &ActivityTaskScheduledEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskStarted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskStarted.Ptr()\n\t\te.ActivityTaskStartedEventAttributes = &ActivityTaskStartedEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskCompleted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskCompleted.Ptr()\n\t\te.ActivityTaskCompletedEventAttributes = &ActivityTaskCompletedEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskFailed.Ptr()\n\t\te.ActivityTaskFailedEventAttributes = &ActivityTaskFailedEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskTimedOut = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskTimedOut.Ptr()\n\t\te.ActivityTaskTimedOutEventAttributes = &ActivityTaskTimedOutEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskCancelRequested = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskCancelRequested.Ptr()\n\t\te.ActivityTaskCancelRequestedEventAttributes = &ActivityTaskCancelRequestedEventAttributes\n\t})\n\tHistoryEvent_RequestCancelActivityTaskFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeRequestCancelActivityTaskFailed.Ptr()\n\t\te.RequestCancelActivityTaskFailedEventAttributes = &RequestCancelActivityTaskFailedEventAttributes\n\t})\n\tHistoryEvent_ActivityTaskCanceled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeActivityTaskCanceled.Ptr()\n\t\te.ActivityTaskCanceledEventAttributes = &ActivityTaskCanceledEventAttributes\n\t})\n\tHistoryEvent_TimerStarted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeTimerStarted.Ptr()\n\t\te.TimerStartedEventAttributes = &TimerStartedEventAttributes\n\t})\n\tHistoryEvent_TimerFired = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeTimerFired.Ptr()\n\t\te.TimerFiredEventAttributes = &TimerFiredEventAttributes\n\t})\n\tHistoryEvent_CancelTimerFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeCancelTimerFailed.Ptr()\n\t\te.CancelTimerFailedEventAttributes = &CancelTimerFailedEventAttributes\n\t})\n\tHistoryEvent_TimerCanceled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeTimerCanceled.Ptr()\n\t\te.TimerCanceledEventAttributes = &TimerCanceledEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionCancelRequested = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionCancelRequested.Ptr()\n\t\te.WorkflowExecutionCancelRequestedEventAttributes = &WorkflowExecutionCancelRequestedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionCanceled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionCanceled.Ptr()\n\t\te.WorkflowExecutionCanceledEventAttributes = &WorkflowExecutionCanceledEventAttributes\n\t})\n\tHistoryEvent_RequestCancelExternalWorkflowExecutionInitiated = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr()\n\t\te.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes = &RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\t})\n\tHistoryEvent_RequestCancelExternalWorkflowExecutionFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeRequestCancelExternalWorkflowExecutionFailed.Ptr()\n\t\te.RequestCancelExternalWorkflowExecutionFailedEventAttributes = &RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\t})\n\tHistoryEvent_ExternalWorkflowExecutionCancelRequested = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeExternalWorkflowExecutionCancelRequested.Ptr()\n\t\te.ExternalWorkflowExecutionCancelRequestedEventAttributes = &ExternalWorkflowExecutionCancelRequestedEventAttributes\n\t})\n\tHistoryEvent_MarkerRecorded = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeMarkerRecorded.Ptr()\n\t\te.MarkerRecordedEventAttributes = &MarkerRecordedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionSignaled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionSignaled.Ptr()\n\t\te.WorkflowExecutionSignaledEventAttributes = &WorkflowExecutionSignaledEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionTerminated = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionTerminated.Ptr()\n\t\te.WorkflowExecutionTerminatedEventAttributes = &WorkflowExecutionTerminatedEventAttributes\n\t})\n\tHistoryEvent_WorkflowExecutionContinuedAsNew = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeWorkflowExecutionContinuedAsNew.Ptr()\n\t\te.WorkflowExecutionContinuedAsNewEventAttributes = &WorkflowExecutionContinuedAsNewEventAttributes\n\t})\n\tHistoryEvent_StartChildWorkflowExecutionInitiated = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeStartChildWorkflowExecutionInitiated.Ptr()\n\t\te.StartChildWorkflowExecutionInitiatedEventAttributes = &StartChildWorkflowExecutionInitiatedEventAttributes\n\t})\n\tHistoryEvent_StartChildWorkflowExecutionFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeStartChildWorkflowExecutionFailed.Ptr()\n\t\te.StartChildWorkflowExecutionFailedEventAttributes = &StartChildWorkflowExecutionFailedEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionStarted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionStarted.Ptr()\n\t\te.ChildWorkflowExecutionStartedEventAttributes = &ChildWorkflowExecutionStartedEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionCompleted = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionCompleted.Ptr()\n\t\te.ChildWorkflowExecutionCompletedEventAttributes = &ChildWorkflowExecutionCompletedEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionFailed.Ptr()\n\t\te.ChildWorkflowExecutionFailedEventAttributes = &ChildWorkflowExecutionFailedEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionCanceled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionCanceled.Ptr()\n\t\te.ChildWorkflowExecutionCanceledEventAttributes = &ChildWorkflowExecutionCanceledEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionTimedOut = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionTimedOut.Ptr()\n\t\te.ChildWorkflowExecutionTimedOutEventAttributes = &ChildWorkflowExecutionTimedOutEventAttributes\n\t})\n\tHistoryEvent_ChildWorkflowExecutionTerminated = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeChildWorkflowExecutionTerminated.Ptr()\n\t\te.ChildWorkflowExecutionTerminatedEventAttributes = &ChildWorkflowExecutionTerminatedEventAttributes\n\t})\n\tHistoryEvent_SignalExternalWorkflowExecutionInitiated = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeSignalExternalWorkflowExecutionInitiated.Ptr()\n\t\te.SignalExternalWorkflowExecutionInitiatedEventAttributes = &SignalExternalWorkflowExecutionInitiatedEventAttributes\n\t})\n\tHistoryEvent_SignalExternalWorkflowExecutionFailed = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeSignalExternalWorkflowExecutionFailed.Ptr()\n\t\te.SignalExternalWorkflowExecutionFailedEventAttributes = &SignalExternalWorkflowExecutionFailedEventAttributes\n\t})\n\tHistoryEvent_ExternalWorkflowExecutionSignaled = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeExternalWorkflowExecutionSignaled.Ptr()\n\t\te.ExternalWorkflowExecutionSignaledEventAttributes = &ExternalWorkflowExecutionSignaledEventAttributes\n\t})\n\tHistoryEvent_UpsertWorkflowSearchAttributes = generateEvent(func(e *types.HistoryEvent) {\n\t\te.EventType = types.EventTypeUpsertWorkflowSearchAttributes.Ptr()\n\t\te.UpsertWorkflowSearchAttributesEventAttributes = &UpsertWorkflowSearchAttributesEventAttributes\n\t})\n\n\tWorkflowExecutionStartedEventAttributes = types.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        &WorkflowType,\n\t\tParentWorkflowDomainID:              common.StringPtr(DomainID),\n\t\tParentWorkflowDomain:                common.StringPtr(DomainName),\n\t\tParentWorkflowExecution:             &WorkflowExecution,\n\t\tParentInitiatedEventID:              common.Int64Ptr(EventID1),\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tContinuedExecutionRunID:             RunID1,\n\t\tInitiator:                           &ContinueAsNewInitiator,\n\t\tContinuedFailureReason:              &FailureReason,\n\t\tContinuedFailureDetails:             FailureDetails,\n\t\tLastCompletionResult:                Payload2,\n\t\tOriginalExecutionRunID:              RunID2,\n\t\tIdentity:                            Identity,\n\t\tFirstExecutionRunID:                 RunID3,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tAttempt:                             Attempt,\n\t\tExpirationTimestamp:                 &Timestamp1,\n\t\tCronSchedule:                        CronSchedule,\n\t\tFirstDecisionTaskBackoffSeconds:     &Duration3,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tPrevAutoResetPoints:                 &ResetPoints,\n\t\tHeader:                              &Header,\n\t\tPartitionConfig:                     PartitionConfig,\n\t\tRequestID:                           RequestID,\n\t\tActiveClusterSelectionPolicy:        &ActiveClusterSelectionPolicyExternalEntity,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t}\n\tWorkflowExecutionCompletedEventAttributes = types.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       Payload1,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t}\n\tWorkflowExecutionFailedEventAttributes = types.WorkflowExecutionFailedEventAttributes{\n\t\tReason:                       &FailureReason,\n\t\tDetails:                      FailureDetails,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t}\n\tWorkflowExecutionTimedOutEventAttributes = types.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: &TimeoutType,\n\t}\n\tDecisionTaskScheduledEventAttributes = types.DecisionTaskScheduledEventAttributes{\n\t\tTaskList:                   &TaskList,\n\t\tStartToCloseTimeoutSeconds: &Duration1,\n\t\tAttempt:                    Attempt,\n\t}\n\tDecisionTaskStartedEventAttributes = types.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventID: EventID1,\n\t\tIdentity:         Identity,\n\t\tRequestID:        RequestID,\n\t}\n\tDecisionTaskCompletedEventAttributes = types.DecisionTaskCompletedEventAttributes{\n\t\tExecutionContext: ExecutionContext,\n\t\tScheduledEventID: EventID1,\n\t\tStartedEventID:   EventID2,\n\t\tIdentity:         Identity,\n\t\tBinaryChecksum:   Checksum,\n\t}\n\tDecisionTaskTimedOutEventAttributes = types.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventID: EventID1,\n\t\tStartedEventID:   EventID2,\n\t\tTimeoutType:      &TimeoutType,\n\t\tBaseRunID:        RunID1,\n\t\tNewRunID:         RunID2,\n\t\tForkEventVersion: Version1,\n\t\tReason:           Reason,\n\t\tCause:            &DecisionTaskTimedOutCause,\n\t\tRequestID:        RequestID,\n\t}\n\tDecisionTaskFailedEventAttributes = types.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventID: EventID1,\n\t\tStartedEventID:   EventID2,\n\t\tCause:            &DecisionTaskFailedCause,\n\t\tDetails:          FailureDetails,\n\t\tIdentity:         Identity,\n\t\tReason:           &FailureReason,\n\t\tBaseRunID:        RunID1,\n\t\tNewRunID:         RunID2,\n\t\tForkEventVersion: Version1,\n\t\tBinaryChecksum:   Checksum,\n\t\tRequestID:        RequestID,\n\t}\n\tActivityTaskScheduledEventAttributes = types.ActivityTaskScheduledEventAttributes{\n\t\tActivityID:                    ActivityID,\n\t\tActivityType:                  &ActivityType,\n\t\tDomain:                        common.StringPtr(DomainName),\n\t\tTaskList:                      &TaskList,\n\t\tInput:                         Payload1,\n\t\tScheduleToCloseTimeoutSeconds: &Duration1,\n\t\tScheduleToStartTimeoutSeconds: &Duration2,\n\t\tStartToCloseTimeoutSeconds:    &Duration3,\n\t\tHeartbeatTimeoutSeconds:       &Duration4,\n\t\tDecisionTaskCompletedEventID:  EventID1,\n\t\tRetryPolicy:                   &RetryPolicy,\n\t\tHeader:                        &Header,\n\t}\n\tActivityTaskStartedEventAttributes = types.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventID:   EventID1,\n\t\tIdentity:           Identity,\n\t\tRequestID:          RequestID,\n\t\tAttempt:            Attempt,\n\t\tLastFailureReason:  &FailureReason,\n\t\tLastFailureDetails: FailureDetails,\n\t}\n\tActivityTaskCompletedEventAttributes = types.ActivityTaskCompletedEventAttributes{\n\t\tResult:           Payload1,\n\t\tScheduledEventID: EventID1,\n\t\tStartedEventID:   EventID2,\n\t\tIdentity:         Identity,\n\t}\n\tActivityTaskFailedEventAttributes = types.ActivityTaskFailedEventAttributes{\n\t\tReason:           &FailureReason,\n\t\tDetails:          FailureDetails,\n\t\tScheduledEventID: EventID1,\n\t\tStartedEventID:   EventID2,\n\t\tIdentity:         Identity,\n\t}\n\tActivityTaskTimedOutEventAttributes = types.ActivityTaskTimedOutEventAttributes{\n\t\tDetails:            Payload1,\n\t\tScheduledEventID:   EventID1,\n\t\tStartedEventID:     EventID2,\n\t\tTimeoutType:        &TimeoutType,\n\t\tLastFailureReason:  &FailureReason,\n\t\tLastFailureDetails: FailureDetails,\n\t}\n\tTimerStartedEventAttributes = types.TimerStartedEventAttributes{\n\t\tTimerID:                      TimerID,\n\t\tStartToFireTimeoutSeconds:    common.Int64Ptr(int64(Duration1)),\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t}\n\tTimerFiredEventAttributes = types.TimerFiredEventAttributes{\n\t\tTimerID:        TimerID,\n\t\tStartedEventID: EventID1,\n\t}\n\tActivityTaskCancelRequestedEventAttributes = types.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityID:                   ActivityID,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t}\n\tRequestCancelActivityTaskFailedEventAttributes = types.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityID:                   ActivityID,\n\t\tCause:                        Cause,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t}\n\tActivityTaskCanceledEventAttributes = types.ActivityTaskCanceledEventAttributes{\n\t\tDetails:                      Payload1,\n\t\tLatestCancelRequestedEventID: EventID1,\n\t\tScheduledEventID:             EventID2,\n\t\tStartedEventID:               EventID3,\n\t\tIdentity:                     Identity,\n\t}\n\tTimerCanceledEventAttributes = types.TimerCanceledEventAttributes{\n\t\tTimerID:                      TimerID,\n\t\tStartedEventID:               EventID1,\n\t\tDecisionTaskCompletedEventID: EventID2,\n\t\tIdentity:                     Identity,\n\t}\n\tCancelTimerFailedEventAttributes = types.CancelTimerFailedEventAttributes{\n\t\tTimerID:                      TimerID,\n\t\tCause:                        Cause,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tIdentity:                     Identity,\n\t}\n\tMarkerRecordedEventAttributes = types.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   MarkerName,\n\t\tDetails:                      Payload1,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tHeader:                       &Header,\n\t}\n\tWorkflowExecutionSignaledEventAttributes = types.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: SignalName,\n\t\tInput:      Payload1,\n\t\tIdentity:   Identity,\n\t\tRequestID:  RequestID,\n\t}\n\tWorkflowExecutionTerminatedEventAttributes = types.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   Reason,\n\t\tDetails:  Payload1,\n\t\tIdentity: Identity,\n\t}\n\tWorkflowExecutionCancelRequestedEventAttributes = types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                     Cause,\n\t\tExternalInitiatedEventID:  common.Int64Ptr(EventID1),\n\t\tExternalWorkflowExecution: &WorkflowExecution,\n\t\tIdentity:                  Identity,\n\t\tRequestID:                 RequestID,\n\t}\n\tWorkflowExecutionCanceledEventAttributes = types.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tDetails:                      Payload1,\n\t}\n\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes = types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tDomain:                       DomainName,\n\t\tWorkflowExecution:            &WorkflowExecution,\n\t\tControl:                      Control,\n\t\tChildWorkflowOnly:            true,\n\t}\n\tRequestCancelExternalWorkflowExecutionFailedEventAttributes = types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        &CancelExternalWorkflowExecutionFailedCause,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tDomain:                       DomainName,\n\t\tWorkflowExecution:            &WorkflowExecution,\n\t\tInitiatedEventID:             EventID2,\n\t\tControl:                      Control,\n\t}\n\tExternalWorkflowExecutionCancelRequestedEventAttributes = types.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventID:  EventID1,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t}\n\tWorkflowExecutionContinuedAsNewEventAttributes = types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunID:                   RunID1,\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tDecisionTaskCompletedEventID:        EventID1,\n\t\tBackoffStartIntervalInSeconds:       &Duration3,\n\t\tInitiator:                           &ContinueAsNewInitiator,\n\t\tFailureReason:                       &FailureReason,\n\t\tFailureDetails:                      FailureDetails,\n\t\tLastCompletionResult:                Payload2,\n\t\tHeader:                              &Header,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tActiveClusterSelectionPolicy:        &ActiveClusterSelectionPolicyExternalEntity,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t}\n\tStartChildWorkflowExecutionInitiatedEventAttributes = types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              DomainName,\n\t\tWorkflowID:                          WorkflowID,\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tParentClosePolicy:                   &ParentClosePolicy,\n\t\tControl:                             Control,\n\t\tDecisionTaskCompletedEventID:        EventID1,\n\t\tWorkflowIDReusePolicy:               &WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tCronSchedule:                        CronSchedule,\n\t\tHeader:                              &Header,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tDelayStartSeconds:                   &Duration3,\n\t\tJitterStartSeconds:                  &Duration4,\n\t\tFirstRunAtTimestamp:                 &Timestamp1,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t}\n\tStartChildWorkflowExecutionFailedEventAttributes = types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       DomainName,\n\t\tWorkflowID:                   WorkflowID,\n\t\tWorkflowType:                 &WorkflowType,\n\t\tCause:                        &ChildWorkflowExecutionFailedCause,\n\t\tControl:                      Control,\n\t\tInitiatedEventID:             EventID1,\n\t\tDecisionTaskCompletedEventID: EventID2,\n\t}\n\tChildWorkflowExecutionStartedEventAttributes = types.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            DomainName,\n\t\tInitiatedEventID:  EventID1,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tHeader:            &Header,\n\t}\n\tChildWorkflowExecutionCompletedEventAttributes = types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tResult:            Payload1,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tInitiatedEventID:  EventID1,\n\t\tStartedEventID:    EventID2,\n\t}\n\tChildWorkflowExecutionFailedEventAttributes = types.ChildWorkflowExecutionFailedEventAttributes{\n\t\tReason:            &FailureReason,\n\t\tDetails:           FailureDetails,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tInitiatedEventID:  EventID1,\n\t\tStartedEventID:    EventID2,\n\t}\n\tChildWorkflowExecutionCanceledEventAttributes = types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDetails:           Payload1,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tInitiatedEventID:  EventID1,\n\t\tStartedEventID:    EventID2,\n\t}\n\tChildWorkflowExecutionTimedOutEventAttributes = types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType:       &TimeoutType,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tInitiatedEventID:  EventID1,\n\t\tStartedEventID:    EventID2,\n\t}\n\tChildWorkflowExecutionTerminatedEventAttributes = types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tWorkflowType:      &WorkflowType,\n\t\tInitiatedEventID:  EventID1,\n\t\tStartedEventID:    EventID2,\n\t}\n\tSignalExternalWorkflowExecutionInitiatedEventAttributes = types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tDomain:                       DomainName,\n\t\tWorkflowExecution:            &WorkflowExecution,\n\t\tSignalName:                   SignalName,\n\t\tInput:                        Payload1,\n\t\tControl:                      Control,\n\t\tChildWorkflowOnly:            true,\n\t}\n\tSignalExternalWorkflowExecutionFailedEventAttributes = types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tCause:                        &SignalExternalWorkflowExecutionFailedCause,\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tDomain:                       DomainName,\n\t\tWorkflowExecution:            &WorkflowExecution,\n\t\tInitiatedEventID:             EventID2,\n\t\tControl:                      Control,\n\t}\n\tExternalWorkflowExecutionSignaledEventAttributes = types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventID:  EventID1,\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tControl:           Control,\n\t}\n\tUpsertWorkflowSearchAttributesEventAttributes = types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventID: EventID1,\n\t\tSearchAttributes:             &SearchAttributes,\n\t}\n\tGetFailoverInfoRequest = types.GetFailoverInfoRequest{\n\t\tDomainID: uuid.NewUUID().String(),\n\t}\n\tGetFailoverInfoResponse = types.GetFailoverInfoResponse{\n\t\tCompletedShardCount: 0,\n\t\tPendingShards:       []int32{1, 2, 3},\n\t}\n\tRatelimitUpdateRequest = types.RatelimitUpdateRequest{\n\t\tAny: &types.Any{\n\t\t\tValueType: history.WeightedRatelimitUsageAnyType, // only correct value currently\n\t\t\tValue:     []byte(\"test request\"),                // invalid contents, but not inspected in these tests\n\t\t},\n\t}\n\tRatelimitUpdateResponse = types.RatelimitUpdateResponse{\n\t\tAny: &types.Any{\n\t\t\tValueType: history.WeightedRatelimitQuotasAnyType, // only correct value currently\n\t\t\tValue:     []byte(\"test response\"),                // invalid contents, but not inspected in these tests\n\t\t},\n\t}\n)\n\nfunc generateEvent(modifier func(e *types.HistoryEvent)) types.HistoryEvent {\n\te := types.HistoryEvent{\n\t\tID:        EventID1,\n\t\tTimestamp: &Timestamp1,\n\t\tVersion:   Version1,\n\t\tTaskID:    TaskID,\n\t}\n\tmodifier(&e)\n\treturn e\n}\n"
  },
  {
    "path": "common/types/testdata/queue.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tTaskState = int16(1)\n)\n\nvar (\n\tCrossClusterTaskInfo                             = *generateCrossClusterTaskInfo(types.CrossClusterTaskTypeStartChildExecution)\n\tCrossClusterStartChildExecutionRequestAttributes = types.CrossClusterStartChildExecutionRequestAttributes{\n\t\tTargetDomainID:           DomainID,\n\t\tRequestID:                RequestID,\n\t\tInitiatedEventID:         EventID1,\n\t\tInitiatedEventAttributes: &StartChildWorkflowExecutionInitiatedEventAttributes,\n\t\tTargetRunID:              common.StringPtr(RunID1),\n\t\tPartitionConfig:          PartitionConfig,\n\t}\n\tCrossClusterStartChildExecutionResponseAttributes = types.CrossClusterStartChildExecutionResponseAttributes{\n\t\tRunID: RunID,\n\t}\n\tCrossClusterCancelExecutionRequestAttributes = types.CrossClusterCancelExecutionRequestAttributes{\n\t\tTargetDomainID:    DomainID,\n\t\tTargetWorkflowID:  WorkflowID,\n\t\tTargetRunID:       RunID,\n\t\tRequestID:         RequestID,\n\t\tInitiatedEventID:  EventID1,\n\t\tChildWorkflowOnly: true,\n\t}\n\tCrossClusterCancelExecutionResponseAttributes = types.CrossClusterCancelExecutionResponseAttributes{}\n\tCrossClusterSignalExecutionRequestAttributes  = types.CrossClusterSignalExecutionRequestAttributes{\n\t\tTargetDomainID:    DomainID,\n\t\tTargetWorkflowID:  WorkflowID,\n\t\tTargetRunID:       RunID,\n\t\tRequestID:         RequestID,\n\t\tInitiatedEventID:  EventID1,\n\t\tChildWorkflowOnly: true,\n\t\tSignalName:        SignalName,\n\t\tSignalInput:       Payload1,\n\t\tControl:           Control,\n\t}\n\tCrossClusterSignalExecutionResponseAttributes                     = types.CrossClusterSignalExecutionResponseAttributes{}\n\tCrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes = types.CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes{\n\t\tTargetDomainID:   DomainID,\n\t\tTargetWorkflowID: WorkflowID,\n\t\tTargetRunID:      RunID,\n\t\tInitiatedEventID: EventID1,\n\t\tCompletionEvent:  &HistoryEvent_ChildWorkflowExecutionCompleted,\n\t}\n\n\tCrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes = types.CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes{}\n\tCrossClusterApplyParentClosePolicyRequestAttributes                = types.CrossClusterApplyParentClosePolicyRequestAttributes{\n\t\tChildren: []*types.ApplyParentClosePolicyRequest{\n\t\t\t{\n\t\t\t\tChild: &types.ApplyParentClosePolicyAttributes{\n\t\t\t\t\tChildDomainID:     DomainID,\n\t\t\t\t\tChildWorkflowID:   WorkflowID,\n\t\t\t\t\tChildRunID:        RunID,\n\t\t\t\t\tParentClosePolicy: &ParentClosePolicy,\n\t\t\t\t},\n\t\t\t\tStatus: &types.ApplyParentClosePolicyStatus{\n\t\t\t\t\tCompleted:   false,\n\t\t\t\t\tFailedCause: types.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tCrossClusterApplyParentClosePolicyResponseAttributes = types.CrossClusterApplyParentClosePolicyResponseAttributes{}\n\tCrossClusterTaskRequestStartChildExecution           = types.CrossClusterTaskRequest{\n\t\tTaskInfo:                      generateCrossClusterTaskInfo(types.CrossClusterTaskTypeStartChildExecution),\n\t\tStartChildExecutionAttributes: &CrossClusterStartChildExecutionRequestAttributes,\n\t}\n\tCrossClusterTaskRequestCancelExecution = types.CrossClusterTaskRequest{\n\t\tTaskInfo:                  generateCrossClusterTaskInfo(types.CrossClusterTaskTypeCancelExecution),\n\t\tCancelExecutionAttributes: &CrossClusterCancelExecutionRequestAttributes,\n\t}\n\tCrossClusterTaskRequestSignalExecution = types.CrossClusterTaskRequest{\n\t\tTaskInfo:                  generateCrossClusterTaskInfo(types.CrossClusterTaskTypeSignalExecution),\n\t\tSignalExecutionAttributes: &CrossClusterSignalExecutionRequestAttributes,\n\t}\n\tCrossClusterTaskRequestRecordChildExecutionComplete = types.CrossClusterTaskRequest{\n\t\tTaskInfo: generateCrossClusterTaskInfo(types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete),\n\t\tRecordChildWorkflowExecutionCompleteAttributes: &CrossClusterRecordChildWorkflowExecutionCompleteRequestAttributes,\n\t}\n\tCrossClusterTaskRequestApplyParentClosePolicy = types.CrossClusterTaskRequest{\n\t\tTaskInfo:                         generateCrossClusterTaskInfo(types.CrossClusterTaskTypeApplyParentPolicy),\n\t\tApplyParentClosePolicyAttributes: &CrossClusterApplyParentClosePolicyRequestAttributes,\n\t}\n\tCrossClusterTaskResponseStartChildExecution = types.CrossClusterTaskResponse{\n\t\tTaskID:                        TaskID,\n\t\tTaskType:                      types.CrossClusterTaskTypeStartChildExecution.Ptr(),\n\t\tTaskState:                     1,\n\t\tStartChildExecutionAttributes: &CrossClusterStartChildExecutionResponseAttributes,\n\t}\n\tCrossClusterTaskResponseCancelExecution = types.CrossClusterTaskResponse{\n\t\tTaskID:                    TaskID,\n\t\tTaskType:                  types.CrossClusterTaskTypeCancelExecution.Ptr(),\n\t\tTaskState:                 3,\n\t\tFailedCause:               types.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t\tCancelExecutionAttributes: &types.CrossClusterCancelExecutionResponseAttributes{},\n\t}\n\tCrossClusterTaskResponseSignalExecution = types.CrossClusterTaskResponse{\n\t\tTaskID:                    TaskID,\n\t\tTaskType:                  types.CrossClusterTaskTypeSignalExecution.Ptr(),\n\t\tTaskState:                 3,\n\t\tFailedCause:               types.CrossClusterTaskFailedCauseWorkflowNotExists.Ptr(),\n\t\tSignalExecutionAttributes: &types.CrossClusterSignalExecutionResponseAttributes{},\n\t}\n\tCrossClusterTaskResponseRecordChildExecutionComplete = types.CrossClusterTaskResponse{\n\t\tTaskID:    TaskID,\n\t\tTaskType:  types.CrossClusterTaskTypeRecordChildWorkflowExeuctionComplete.Ptr(),\n\t\tTaskState: 1,\n\t\tRecordChildWorkflowExecutionCompleteAttributes: &CrossClusterRecordChildWorkflowExecutionCompleteResponseAttributes,\n\t}\n\tCrossClusterTaskResponseApplyParentClosePolicy = types.CrossClusterTaskResponse{\n\t\tTaskID:                           TaskID,\n\t\tTaskType:                         types.CrossClusterTaskTypeApplyParentPolicy.Ptr(),\n\t\tTaskState:                        1,\n\t\tApplyParentClosePolicyAttributes: &CrossClusterApplyParentClosePolicyResponseAttributes,\n\t}\n\tCrossClusterTaskRequestArray = []*types.CrossClusterTaskRequest{\n\t\t&CrossClusterTaskRequestStartChildExecution,\n\t\t&CrossClusterTaskRequestCancelExecution,\n\t\t&CrossClusterTaskRequestSignalExecution,\n\t\t&CrossClusterTaskRequestRecordChildExecutionComplete,\n\t\t&CrossClusterTaskRequestApplyParentClosePolicy,\n\t}\n\tCrossClusterTaskResponseArray = []*types.CrossClusterTaskResponse{\n\t\t&CrossClusterTaskResponseStartChildExecution,\n\t\t&CrossClusterTaskResponseCancelExecution,\n\t\t&CrossClusterTaskResponseSignalExecution,\n\t\t&CrossClusterTaskResponseRecordChildExecutionComplete,\n\t\t&CrossClusterTaskResponseApplyParentClosePolicy,\n\t}\n\tCrossClusterTaskRequestMap = map[int32][]*types.CrossClusterTaskRequest{\n\t\tShardID + 1: {},\n\t\tShardID + 2: CrossClusterTaskRequestArray,\n\t}\n\tGetCrossClusterTaskFailedCauseMap = map[int32]types.GetTaskFailedCause{\n\t\tShardID + 3: types.GetTaskFailedCauseServiceBusy,\n\t\tShardID + 4: types.GetTaskFailedCauseTimeout,\n\t\tShardID + 5: types.GetTaskFailedCauseShardOwnershipLost,\n\t\tShardID + 6: types.GetTaskFailedCauseUncategorized,\n\t}\n\tGetCrossClusterTasksRequest = types.GetCrossClusterTasksRequest{\n\t\tShardIDs:      []int32{ShardID},\n\t\tTargetCluster: ClusterName1,\n\t}\n\tGetCrossClusterTasksResponse = types.GetCrossClusterTasksResponse{\n\t\tTasksByShard:       CrossClusterTaskRequestMap,\n\t\tFailedCauseByShard: GetCrossClusterTaskFailedCauseMap,\n\t}\n\tRespondCrossClusterTasksCompletedRequest = types.RespondCrossClusterTasksCompletedRequest{\n\t\tShardID:       ShardID,\n\t\tTargetCluster: ClusterName1,\n\t\tTaskResponses: CrossClusterTaskResponseArray,\n\t\tFetchNewTasks: true,\n\t}\n\tRespondCrossClusterTasksCompletedResponse = types.RespondCrossClusterTasksCompletedResponse{\n\t\tTasks: CrossClusterTaskRequestArray,\n\t}\n\tApplyParentClosePolicyAttributes = types.ApplyParentClosePolicyAttributes{\n\t\tChildDomainID:     \"testDomain\",\n\t\tChildWorkflowID:   \"testChildWorkflowID\",\n\t\tChildRunID:        \"testChildRunID\",\n\t\tParentClosePolicy: &ParentClosePolicy,\n\t}\n\tApplyParentClosePolicyResult = types.ApplyParentClosePolicyResult{\n\t\tChild:       &ApplyParentClosePolicyAttributes,\n\t\tFailedCause: types.CrossClusterTaskFailedCauseDomainNotActive.Ptr(),\n\t}\n\tApplyParentClosePolicyResult2 = types.ApplyParentClosePolicyResult{\n\t\tChild: &types.ApplyParentClosePolicyAttributes{\n\t\t\tChildDomainID:     \"testDomain2\",\n\t\t\tChildWorkflowID:   \"testChildWorkflowID2\",\n\t\t\tChildRunID:        \"testChildRunID2\",\n\t\t\tParentClosePolicy: &ParentClosePolicy2,\n\t\t},\n\t\tFailedCause: types.CrossClusterTaskFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t}\n\tCrossClusterApplyParentClosePolicyResponseWithChildren = types.CrossClusterApplyParentClosePolicyResponseAttributes{\n\t\tChildrenStatus: []*types.ApplyParentClosePolicyResult{\n\t\t\t&ApplyParentClosePolicyResult,\n\t\t\t&ApplyParentClosePolicyResult2,\n\t\t},\n\t}\n)\n\nfunc generateCrossClusterTaskInfo(\n\ttaskType types.CrossClusterTaskType,\n) *types.CrossClusterTaskInfo {\n\treturn &types.CrossClusterTaskInfo{\n\t\tDomainID:            DomainID,\n\t\tWorkflowID:          WorkflowID,\n\t\tRunID:               RunID,\n\t\tTaskType:            taskType.Ptr(),\n\t\tTaskState:           TaskState,\n\t\tVisibilityTimestamp: &Timestamp,\n\t}\n}\n"
  },
  {
    "path": "common/types/testdata/replication.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tTaskType        = int16(1)\n\tMaxiumuPageSize = int32(5)\n)\n\nvar (\n\tReplicationMessages = types.ReplicationMessages{\n\t\tReplicationTasks:       ReplicationTaskArray,\n\t\tLastRetrievedMessageID: MessageID1,\n\t\tHasMore:                true,\n\t\tSyncShardStatus:        &SyncShardStatus,\n\t}\n\tReplicationMessagesMap = map[int32]*types.ReplicationMessages{\n\t\tShardID: &ReplicationMessages,\n\t}\n\tSyncShardStatus = types.SyncShardStatus{\n\t\tTimestamp: &Timestamp1,\n\t}\n\tReplicationToken = types.ReplicationToken{\n\t\tShardID:                ShardID,\n\t\tLastRetrievedMessageID: MessageID1,\n\t\tLastProcessedMessageID: MessageID2,\n\t}\n\tReplicationTokenArray = []*types.ReplicationToken{\n\t\t&ReplicationToken,\n\t}\n\tReplicationTaskInfo = types.ReplicationTaskInfo{\n\t\tDomainID:     DomainID,\n\t\tWorkflowID:   WorkflowID,\n\t\tRunID:        RunID,\n\t\tTaskType:     TaskType,\n\t\tTaskID:       TaskID,\n\t\tVersion:      Version1,\n\t\tFirstEventID: EventID1,\n\t\tNextEventID:  EventID2,\n\t\tScheduledID:  EventID3,\n\t}\n\tReplicationTaskInfoArray = []*types.ReplicationTaskInfo{\n\t\t&ReplicationTaskInfo,\n\t}\n\tReplicationTask_Domain = types.ReplicationTask{\n\t\tTaskType:             types.ReplicationTaskTypeDomain.Ptr(),\n\t\tSourceTaskID:         TaskID,\n\t\tDomainTaskAttributes: &DomainTaskAttributes,\n\t\tCreationTime:         &Timestamp1,\n\t}\n\tReplicationTask_SyncShard = types.ReplicationTask{\n\t\tTaskType:                      types.ReplicationTaskTypeSyncShardStatus.Ptr(),\n\t\tSourceTaskID:                  TaskID,\n\t\tSyncShardStatusTaskAttributes: &SyncShardStatusTaskAttributes,\n\t\tCreationTime:                  &Timestamp1,\n\t}\n\tReplicationTask_SyncActivity = types.ReplicationTask{\n\t\tTaskType:                   types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSourceTaskID:               TaskID,\n\t\tSyncActivityTaskAttributes: &SyncActivityTaskAttributes,\n\t\tCreationTime:               &Timestamp1,\n\t}\n\tReplicationTask_History = types.ReplicationTask{\n\t\tTaskType:                types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tSourceTaskID:            TaskID,\n\t\tHistoryTaskV2Attributes: &HistoryTaskV2Attributes,\n\t\tCreationTime:            &Timestamp1,\n\t}\n\tReplicationTask_Failover = types.ReplicationTask{\n\t\tTaskType:                 types.ReplicationTaskTypeFailoverMarker.Ptr(),\n\t\tSourceTaskID:             TaskID,\n\t\tFailoverMarkerAttributes: &FailoverMarkerAttributes,\n\t\tCreationTime:             &Timestamp1,\n\t}\n\tReplicationTaskArray = []*types.ReplicationTask{\n\t\t&ReplicationTask_Domain,\n\t\t&ReplicationTask_SyncShard,\n\t\t&ReplicationTask_SyncActivity,\n\t\t&ReplicationTask_History,\n\t\t&ReplicationTask_Failover,\n\t}\n\tDomainTaskAttributes = types.DomainTaskAttributes{\n\t\tDomainOperation:         types.DomainOperationUpdate.Ptr(),\n\t\tID:                      DomainID,\n\t\tInfo:                    &DomainInfo,\n\t\tConfig:                  &DomainConfiguration,\n\t\tReplicationConfig:       &DomainReplicationConfiguration,\n\t\tConfigVersion:           Version1,\n\t\tFailoverVersion:         FailoverVersion1,\n\t\tPreviousFailoverVersion: FailoverVersion2,\n\t}\n\tSyncShardStatusTaskAttributes = types.SyncShardStatusTaskAttributes{\n\t\tSourceCluster: ClusterName1,\n\t\tShardID:       ShardID,\n\t\tTimestamp:     &Timestamp1,\n\t}\n\tSyncActivityTaskAttributes = types.SyncActivityTaskAttributes{\n\t\tDomainID:           DomainID,\n\t\tWorkflowID:         WorkflowID,\n\t\tRunID:              RunID,\n\t\tVersion:            Version1,\n\t\tScheduledID:        EventID1,\n\t\tScheduledTime:      &Timestamp1,\n\t\tStartedID:          EventID2,\n\t\tStartedTime:        &Timestamp2,\n\t\tLastHeartbeatTime:  &Timestamp3,\n\t\tDetails:            Payload1,\n\t\tAttempt:            Attempt,\n\t\tLastFailureReason:  &FailureReason,\n\t\tLastWorkerIdentity: Identity,\n\t\tLastFailureDetails: FailureDetails,\n\t\tVersionHistory:     &VersionHistory,\n\t}\n\tHistoryTaskV2Attributes = types.HistoryTaskV2Attributes{\n\t\tDomainID:            DomainID,\n\t\tWorkflowID:          WorkflowID,\n\t\tRunID:               RunID,\n\t\tVersionHistoryItems: VersionHistoryItemArray,\n\t\tEvents:              &DataBlob,\n\t\tNewRunEvents:        &DataBlob,\n\t}\n\tFailoverMarkerAttributes = types.FailoverMarkerAttributes{\n\t\tDomainID:        DomainID,\n\t\tFailoverVersion: FailoverVersion1,\n\t\tCreationTime:    &Timestamp1,\n\t}\n\tFailoverMarkerAttributesArray = []*types.FailoverMarkerAttributes{\n\t\t&FailoverMarkerAttributes,\n\t}\n\tPersistenceFeatures = []*types.PersistenceFeature{\n\t\t{\n\t\t\tKey:     \"PersistenceFeature\",\n\t\t\tEnabled: true,\n\t\t},\n\t\tnil,\n\t}\n\tPersistenceSettings = []*types.PersistenceSetting{\n\t\t{\n\t\t\tKey:   \"PersistenceKey\",\n\t\t\tValue: \"PersistenceValue\",\n\t\t},\n\t\tnil,\n\t}\n\tPersistenceInfoMap = map[string]*types.PersistenceInfo{\n\t\t\"Backend\": {\n\t\t\tBackend:  \"Backend\",\n\t\t\tSettings: PersistenceSettings,\n\t\t\tFeatures: PersistenceFeatures,\n\t\t},\n\t\t\"Invalid\": nil,\n\t}\n\tGetDomainReplicationMessagesRequest = types.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: common.Int64Ptr(MessageID1),\n\t\tLastProcessedMessageID: common.Int64Ptr(MessageID2),\n\t\tClusterName:            ClusterName1,\n\t}\n\tMergeDLQMessagesRequest = types.MergeDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t\tMaximumPageSize:       MaxiumuPageSize,\n\t\tNextPageToken:         Token1,\n\t}\n\tPurgeDLQMessagesRequest = types.PurgeDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t}\n\tReadDLQMessagesRequest = types.ReadDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t\tMaximumPageSize:       MaxiumuPageSize,\n\t\tNextPageToken:         Token1,\n\t}\n\tReadDLQMessagesResponse = types.ReadDLQMessagesResponse{\n\t\tType:                 types.DLQTypeDomain.Ptr(),\n\t\tReplicationTasks:     ReplicationTaskArray,\n\t\tReplicationTasksInfo: ReplicationTaskInfoArray,\n\t\tNextPageToken:        Token1,\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/schedule.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tscheduleTime1 = time.Date(2026, 3, 1, 10, 0, 0, 0, time.UTC)\n\tscheduleTime2 = time.Date(2026, 6, 15, 12, 30, 0, 0, time.UTC)\n\tscheduleTime3 = time.Date(2026, 9, 1, 8, 0, 0, 0, time.UTC)\n\tscheduleTime4 = time.Date(2026, 12, 31, 23, 59, 59, 0, time.UTC)\n\tscheduleTime5 = time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC)\n\n\tScheduleSpec = types.ScheduleSpec{\n\t\tCronExpression: \"*/5 * * * *\",\n\t\tStartTime:      scheduleTime1,\n\t\tEndTime:        scheduleTime4,\n\t\tJitter:         30 * time.Second,\n\t}\n\n\tScheduleStartWorkflowAction = types.StartWorkflowAction{\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tWorkflowIDPrefix:                    \"sched-wf-\",\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t}\n\n\tScheduleAction = types.ScheduleAction{\n\t\tStartWorkflow: &ScheduleStartWorkflowAction,\n\t}\n\n\tSchedulePolicies = types.SchedulePolicies{\n\t\tOverlapPolicy:    types.ScheduleOverlapPolicyBuffer,\n\t\tCatchUpPolicy:    types.ScheduleCatchUpPolicyAll,\n\t\tCatchUpWindow:    time.Hour,\n\t\tPauseOnFailure:   true,\n\t\tBufferLimit:      5,\n\t\tConcurrencyLimit: 2,\n\t}\n\n\tSchedulePauseInfo = types.SchedulePauseInfo{\n\t\tReason:   \"maintenance window\",\n\t\tPausedAt: scheduleTime2,\n\t\tPausedBy: \"sample_admin@uber.com\",\n\t}\n\n\tScheduleState = types.ScheduleState{\n\t\tPaused:    true,\n\t\tPauseInfo: &SchedulePauseInfo,\n\t}\n\n\tScheduleBackfillInfo = types.BackfillInfo{\n\t\tBackfillID:    \"backfill-001\",\n\t\tStartTime:     scheduleTime1,\n\t\tEndTime:       scheduleTime3,\n\t\tRunsCompleted: 15,\n\t\tRunsTotal:     30,\n\t}\n\n\tScheduleBackfillInfo2 = types.BackfillInfo{\n\t\tBackfillID:    \"backfill-002\",\n\t\tStartTime:     scheduleTime3,\n\t\tEndTime:       scheduleTime4,\n\t\tRunsCompleted: 0,\n\t\tRunsTotal:     10,\n\t}\n\n\tScheduleInfo = types.ScheduleInfo{\n\t\tLastRunTime:      scheduleTime2,\n\t\tNextRunTime:      scheduleTime3,\n\t\tTotalRuns:        42,\n\t\tCreateTime:       scheduleTime5,\n\t\tLastUpdateTime:   scheduleTime2,\n\t\tOngoingBackfills: []*types.BackfillInfo{&ScheduleBackfillInfo, &ScheduleBackfillInfo2},\n\t}\n\n\tScheduleListEntry = types.ScheduleListEntry{\n\t\tScheduleID:     \"my-schedule-id\",\n\t\tWorkflowType:   &WorkflowType,\n\t\tState:          &ScheduleState,\n\t\tCronExpression: \"*/5 * * * *\",\n\t}\n\n\tCreateScheduleRequest = types.CreateScheduleRequest{\n\t\tDomain:           DomainName,\n\t\tScheduleID:       \"my-schedule-id\",\n\t\tSpec:             &ScheduleSpec,\n\t\tAction:           &ScheduleAction,\n\t\tPolicies:         &SchedulePolicies,\n\t\tMemo:             &Memo,\n\t\tSearchAttributes: &SearchAttributes,\n\t}\n\n\tCreateScheduleResponse = types.CreateScheduleResponse{}\n\n\tDescribeScheduleRequest = types.DescribeScheduleRequest{\n\t\tDomain:     DomainName,\n\t\tScheduleID: \"my-schedule-id\",\n\t}\n\n\tDescribeScheduleResponse = types.DescribeScheduleResponse{\n\t\tSpec:             &ScheduleSpec,\n\t\tAction:           &ScheduleAction,\n\t\tPolicies:         &SchedulePolicies,\n\t\tState:            &ScheduleState,\n\t\tInfo:             &ScheduleInfo,\n\t\tMemo:             &Memo,\n\t\tSearchAttributes: &SearchAttributes,\n\t}\n\n\tUpdateScheduleRequest = types.UpdateScheduleRequest{\n\t\tDomain:           DomainName,\n\t\tScheduleID:       \"my-schedule-id\",\n\t\tSpec:             &ScheduleSpec,\n\t\tAction:           &ScheduleAction,\n\t\tPolicies:         &SchedulePolicies,\n\t\tSearchAttributes: &SearchAttributes,\n\t}\n\n\tUpdateScheduleResponse = types.UpdateScheduleResponse{}\n\n\tDeleteScheduleRequest = types.DeleteScheduleRequest{\n\t\tDomain:     DomainName,\n\t\tScheduleID: \"my-schedule-id\",\n\t}\n\n\tDeleteScheduleResponse = types.DeleteScheduleResponse{}\n\n\tPauseScheduleRequest = types.PauseScheduleRequest{\n\t\tDomain:     DomainName,\n\t\tScheduleID: \"my-schedule-id\",\n\t\tReason:     \"maintenance window\",\n\t}\n\n\tPauseScheduleResponse = types.PauseScheduleResponse{}\n\n\tUnpauseScheduleRequest = types.UnpauseScheduleRequest{\n\t\tDomain:        DomainName,\n\t\tScheduleID:    \"my-schedule-id\",\n\t\tReason:        \"maintenance complete\",\n\t\tCatchUpPolicy: types.ScheduleCatchUpPolicyOne,\n\t}\n\n\tUnpauseScheduleResponse = types.UnpauseScheduleResponse{}\n\n\tListSchedulesRequest = types.ListSchedulesRequest{\n\t\tDomain:        DomainName,\n\t\tPageSize:      10,\n\t\tNextPageToken: []byte(\"next-page-token\"),\n\t}\n\n\tListSchedulesResponse = types.ListSchedulesResponse{\n\t\tSchedules:     []*types.ScheduleListEntry{&ScheduleListEntry},\n\t\tNextPageToken: []byte(\"next-page-token-2\"),\n\t}\n\n\tBackfillScheduleRequest = types.BackfillScheduleRequest{\n\t\tDomain:        DomainName,\n\t\tScheduleID:    \"my-schedule-id\",\n\t\tStartTime:     scheduleTime1,\n\t\tEndTime:       scheduleTime3,\n\t\tOverlapPolicy: types.ScheduleOverlapPolicyConcurrent,\n\t\tBackfillID:    \"backfill-003\",\n\t}\n\n\tBackfillScheduleResponse = types.BackfillScheduleResponse{}\n)\n"
  },
  {
    "path": "common/types/testdata/service_admin.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst QueueType = 2\n\nvar (\n\tAdminAddSearchAttributeRequest = types.AddSearchAttributeRequest{\n\t\tSearchAttribute: IndexedValueTypeMap,\n\t\tSecurityToken:   SecurityToken,\n\t}\n\tAdminCloseShardRequest = types.CloseShardRequest{\n\t\tShardID: ShardID,\n\t}\n\tAdminDeleteWorkflowRequest = types.AdminDeleteWorkflowRequest{\n\t\tDomain:    DomainID,\n\t\tExecution: &WorkflowExecution,\n\t}\n\tAdminDeleteWorkflowResponse = types.AdminDeleteWorkflowResponse{\n\t\tHistoryDeleted:    true,\n\t\tExecutionsDeleted: false,\n\t\tVisibilityDeleted: true,\n\t}\n\tAdminDescribeClusterResponse = types.DescribeClusterResponse{\n\t\tSupportedClientVersions: &SupportedClientVersions,\n\t\tMembershipInfo:          &MembershipInfo,\n\t\tPersistenceInfo:         PersistenceInfoMap,\n\t}\n\tAdminDescribeHistoryHostRequest_ByHost = types.DescribeHistoryHostRequest{\n\t\tHostAddress: common.StringPtr(HostName),\n\t}\n\tAdminDescribeHistoryHostRequest_ByShard = types.DescribeHistoryHostRequest{\n\t\tShardIDForHost: common.Int32Ptr(ShardID),\n\t}\n\tAdminDescribeHistoryHostRequest_ByExecution = types.DescribeHistoryHostRequest{\n\t\tExecutionForHost: &WorkflowExecution,\n\t}\n\tAdminDescribeHistoryHostResponse = types.DescribeHistoryHostResponse{\n\t\tNumberOfShards:        1,\n\t\tShardIDs:              []int32{ShardID},\n\t\tDomainCache:           &DomainCacheInfo,\n\t\tShardControllerStatus: \"ShardControllerStatus\",\n\t\tAddress:               HostName,\n\t}\n\tAdminDescribeQueueRequest = types.DescribeQueueRequest{\n\t\tShardID:     ShardID,\n\t\tClusterName: ClusterName1,\n\t\tType:        common.Int32Ptr(QueueType),\n\t}\n\tAdminDescribeQueueResponse = types.DescribeQueueResponse{\n\t\tProcessingQueueStates: []string{\"state1\", \"state2\"},\n\t}\n\tAdminDescribeShardDistributionRequest = types.DescribeShardDistributionRequest{\n\t\tPageSize: PageSize,\n\t\tPageID:   1,\n\t}\n\tAdminDescribeShardDistributionResponse = types.DescribeShardDistributionResponse{\n\t\tNumberOfShards: 2,\n\t\tShards: map[int32]string{\n\t\t\t0: \"shard1\",\n\t\t\t1: \"shard2\",\n\t\t},\n\t}\n\tAdminDescribeWorkflowExecutionRequest = types.AdminDescribeWorkflowExecutionRequest{\n\t\tDomain:    DomainName,\n\t\tExecution: &WorkflowExecution,\n\t}\n\tAdminDescribeWorkflowExecutionResponse = types.AdminDescribeWorkflowExecutionResponse{\n\t\tShardID:                fmt.Sprint(ShardID),\n\t\tHistoryAddr:            HostName,\n\t\tMutableStateInCache:    \"MutableStateInCache\",\n\t\tMutableStateInDatabase: \"MutableStateInDatabase\",\n\t}\n\tAdminGetDLQReplicationMessagesRequest = types.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: ReplicationTaskInfoArray,\n\t}\n\tAdminGetDLQReplicationMessagesResponse = types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: ReplicationTaskArray,\n\t}\n\tAdminGetDomainIsolationGroupsRequest = types.GetDomainIsolationGroupsRequest{\n\t\tDomain: DomainName,\n\t}\n\tAdminGetDomainIsolationGroupsResponse = types.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: IsolationGroupConfiguration,\n\t}\n\tAdminGetDomainReplicationMessagesRequest = types.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: common.Int64Ptr(MessageID1),\n\t\tLastProcessedMessageID: common.Int64Ptr(MessageID2),\n\t\tClusterName:            ClusterName1,\n\t}\n\tAdminGetDomainReplicationMessagesResponse = types.GetDomainReplicationMessagesResponse{\n\t\tMessages: &ReplicationMessages,\n\t}\n\tAdminGetDynamicConfigRequest = types.GetDynamicConfigRequest{\n\t\tConfigName: DynamicConfigEntryName,\n\t\tFilters:    []*types.DynamicConfigFilter{&DynamicConfigFilter},\n\t}\n\tAdminGetDynamicConfigResponse        = types.GetDynamicConfigResponse{Value: &DataBlob}\n\tAdminGetGlobalIsolationGroupsRequest = types.GetGlobalIsolationGroupsRequest{}\n\tAdminGetReplicationMessagesRequest   = types.GetReplicationMessagesRequest{\n\t\tTokens:      ReplicationTokenArray,\n\t\tClusterName: ClusterName1,\n\t}\n\tAdminGetReplicationMessagesResponse = types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: ReplicationMessagesMap,\n\t}\n\tAdminGetWorkflowExecutionRawHistoryV2Request = types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain:            DomainName,\n\t\tExecution:         &WorkflowExecution,\n\t\tStartEventID:      common.Int64Ptr(EventID1),\n\t\tStartEventVersion: common.Int64Ptr(Version1),\n\t\tEndEventID:        common.Int64Ptr(EventID2),\n\t\tEndEventVersion:   common.Int64Ptr(EventID2),\n\t\tMaximumPageSize:   PageSize,\n\t\tNextPageToken:     NextPageToken,\n\t}\n\tAdminGetWorkflowExecutionRawHistoryV2Response = types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tNextPageToken:  NextPageToken,\n\t\tHistoryBatches: DataBlobArray,\n\t\tVersionHistory: &VersionHistory,\n\t}\n\tAdminCountDLQMessagesRequest  = types.CountDLQMessagesRequest{ForceFetch: true}\n\tAdminCountDLQMessagesResponse = types.CountDLQMessagesResponse{\n\t\tHistory: HistoryCountDLQMessagesResponse.Entries,\n\t\tDomain:  123456,\n\t}\n\tAdminListDynamicConfigRequest = types.ListDynamicConfigRequest{\n\t\tConfigName: DynamicConfigEntryName,\n\t}\n\tAdminListDynamicConfigResponse = types.ListDynamicConfigResponse{\n\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t{\n\t\t\t\tName: DynamicConfigEntryName,\n\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t&DynamicConfigValue,\n\t\t\t\t},\n\t\t\t},\n\t\t\tnil,\n\t\t},\n\t}\n\tAdminMaintainCorruptWorkflowRequest = types.AdminMaintainWorkflowRequest{\n\t\tDomain:    DomainName,\n\t\tExecution: &WorkflowExecution,\n\t}\n\tAdminMaintainCorruptWorkflowResponse = types.AdminMaintainWorkflowResponse{\n\t\tHistoryDeleted:    false,\n\t\tExecutionsDeleted: true,\n\t\tVisibilityDeleted: false,\n\t}\n\tAdminMergeDLQMessagesRequest = types.MergeDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t\tMaximumPageSize:       PageSize,\n\t\tNextPageToken:         NextPageToken,\n\t}\n\tAdminMergeDLQMessagesResponse = types.MergeDLQMessagesResponse{\n\t\tNextPageToken: NextPageToken,\n\t}\n\tAdminPurgeDLQMessagesRequest = types.PurgeDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t}\n\tAdminReadDLQMessagesRequest = types.ReadDLQMessagesRequest{\n\t\tType:                  types.DLQTypeDomain.Ptr(),\n\t\tShardID:               ShardID,\n\t\tSourceCluster:         ClusterName1,\n\t\tInclusiveEndMessageID: common.Int64Ptr(MessageID1),\n\t\tMaximumPageSize:       PageSize,\n\t\tNextPageToken:         NextPageToken,\n\t}\n\tAdminReadDLQMessagesResponse = types.ReadDLQMessagesResponse{\n\t\tType:                 types.DLQTypeDomain.Ptr(),\n\t\tReplicationTasks:     ReplicationTaskArray,\n\t\tReplicationTasksInfo: ReplicationTaskInfoArray,\n\t\tNextPageToken:        NextPageToken,\n\t}\n\tAdminReapplyEventsRequest = types.ReapplyEventsRequest{\n\t\tDomainName:        DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tEvents:            &DataBlob,\n\t}\n\tAdminRefreshWorkflowTasksRequest = types.RefreshWorkflowTasksRequest{\n\t\tDomain:    DomainName,\n\t\tExecution: &WorkflowExecution,\n\t}\n\tAdminRemoveTaskRequest = types.RemoveTaskRequest{\n\t\tShardID:             ShardID,\n\t\tType:                common.Int32Ptr(QueueType),\n\t\tTaskID:              TaskID,\n\t\tVisibilityTimestamp: &Timestamp1,\n\t\tClusterName:         ClusterName1,\n\t}\n\tAdminResendReplicationTasksRequest = types.ResendReplicationTasksRequest{\n\t\tDomainID:      DomainID,\n\t\tWorkflowID:    WorkflowID,\n\t\tRunID:         RunID,\n\t\tRemoteCluster: ClusterName1,\n\t\tStartEventID:  common.Int64Ptr(EventID1),\n\t\tStartVersion:  common.Int64Ptr(Version1),\n\t\tEndEventID:    common.Int64Ptr(EventID2),\n\t\tEndVersion:    common.Int64Ptr(EventID2),\n\t}\n\tAdminResetQueueRequest = types.ResetQueueRequest{\n\t\tShardID:     ShardID,\n\t\tClusterName: ClusterName1,\n\t\tType:        common.Int32Ptr(QueueType),\n\t}\n\tAdminGetCrossClusterTasksRequest               = GetCrossClusterTasksRequest\n\tAdminGetCrossClusterTasksResponse              = GetCrossClusterTasksResponse\n\tAdminRespondCrossClusterTasksCompletedRequest  = RespondCrossClusterTasksCompletedRequest\n\tAdminRespondCrossClusterTasksCompletedResponse = RespondCrossClusterTasksCompletedResponse\n\tAdminRestoreDynamicConfigRequest               = types.RestoreDynamicConfigRequest{\n\t\tConfigName: DynamicConfigEntryName,\n\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t&DynamicConfigFilter,\n\t\t},\n\t}\n\tAdminUpdateDomainIsolationGroupsRequest = types.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          DomainName,\n\t\tIsolationGroups: IsolationGroupConfiguration,\n\t}\n\tAdminUpdateDomainIsolationGroupsResponse = types.UpdateDomainIsolationGroupsResponse{}\n\tAdminUpdateDynamicConfigRequest          = types.UpdateDynamicConfigRequest{\n\t\tConfigName: DynamicConfigEntryName,\n\t\tConfigValues: []*types.DynamicConfigValue{\n\t\t\t{\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t&DynamicConfigFilter,\n\t\t\t\t\tnil,\n\t\t\t\t},\n\t\t\t\tValue: &DataBlob,\n\t\t\t},\n\t\t\tnil,\n\t\t},\n\t}\n\tAdminUpdateGlobalIsolationGroupsResponse  = types.UpdateGlobalIsolationGroupsResponse{}\n\tAdminUpdateTaskListPartitionConfigRequest = types.UpdateTaskListPartitionConfigRequest{\n\t\tDomain:          DomainName,\n\t\tTaskList:        &TaskList,\n\t\tTaskListType:    &TaskListType,\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n\tAdminUpdateTaskListPartitionConfigResponse = types.UpdateTaskListPartitionConfigResponse{}\n)\n"
  },
  {
    "path": "common/types/testdata/service_frontend.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tRegisterDomainRequest = types.RegisterDomainRequest{\n\t\tName:                                   DomainName,\n\t\tDescription:                            DomainDescription,\n\t\tOwnerEmail:                             DomainOwnerEmail,\n\t\tWorkflowExecutionRetentionPeriodInDays: DomainRetention,\n\t\tEmitMetric:                             common.BoolPtr(DomainEmitMetric),\n\t\tClusters:                               ClusterReplicationConfigurationArray,\n\t\tActiveClusterName:                      ClusterName1,\n\t\tActiveClusters:                         &ActiveClusters,\n\t\tData:                                   DomainData,\n\t\tSecurityToken:                          SecurityToken,\n\t\tIsGlobalDomain:                         true,\n\t\tHistoryArchivalStatus:                  &ArchivalStatus,\n\t\tHistoryArchivalURI:                     HistoryArchivalURI,\n\t\tVisibilityArchivalStatus:               &ArchivalStatus,\n\t\tVisibilityArchivalURI:                  VisibilityArchivalURI,\n\t}\n\tDescribeDomainRequest_ID = types.DescribeDomainRequest{\n\t\tUUID: common.StringPtr(DomainID),\n\t}\n\tDescribeDomainRequest_Name = types.DescribeDomainRequest{\n\t\tName: common.StringPtr(DomainName),\n\t}\n\tDescribeDomainResponse = types.DescribeDomainResponse{\n\t\tDomainInfo:               &DomainInfo,\n\t\tConfiguration:            &DomainConfiguration,\n\t\tReplicationConfiguration: &DomainReplicationConfiguration,\n\t\tFailoverVersion:          FailoverVersion1,\n\t\tIsGlobalDomain:           true,\n\t}\n\tListDomainsRequest = types.ListDomainsRequest{\n\t\tPageSize:      PageSize,\n\t\tNextPageToken: NextPageToken,\n\t}\n\tListDomainsResponse = types.ListDomainsResponse{\n\t\tDomains:       []*types.DescribeDomainResponse{&DescribeDomainResponse},\n\t\tNextPageToken: NextPageToken,\n\t}\n\tUpdateDomainRequest = types.UpdateDomainRequest{\n\t\tName:                                   DomainName,\n\t\tDescription:                            common.StringPtr(DomainDescription),\n\t\tOwnerEmail:                             common.StringPtr(DomainOwnerEmail),\n\t\tData:                                   DomainData,\n\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(DomainRetention),\n\t\tBadBinaries:                            &BadBinaries,\n\t\tHistoryArchivalStatus:                  &ArchivalStatus,\n\t\tHistoryArchivalURI:                     common.StringPtr(HistoryArchivalURI),\n\t\tVisibilityArchivalStatus:               &ArchivalStatus,\n\t\tVisibilityArchivalURI:                  common.StringPtr(VisibilityArchivalURI),\n\t\tActiveClusterName:                      common.StringPtr(ClusterName1),\n\t\tActiveClusters:                         &ActiveClusters,\n\t\tClusters:                               ClusterReplicationConfigurationArray,\n\t\tSecurityToken:                          SecurityToken,\n\t\tDeleteBadBinary:                        common.StringPtr(DeleteBadBinary),\n\t\tFailoverTimeoutInSeconds:               &Duration1,\n\t}\n\tFailoverDomainRequest = types.FailoverDomainRequest{\n\t\tDomainName:              DomainName,\n\t\tDomainActiveClusterName: common.StringPtr(ClusterName1),\n\t\tActiveClusters:          &ActiveClusters,\n\t}\n\tFailoverDomainRequest_OnlyActiveClusters = types.FailoverDomainRequest{\n\t\tDomainName: DomainName,\n\t\t// Explicitly set to nil to test ActiveActive failovers\n\t\tDomainActiveClusterName: nil,\n\t\tActiveClusters:          &ActiveClusters,\n\t}\n\tActiveClusters = types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\tRegion1: {\n\t\t\t\t\t\tActiveClusterName: ClusterName1,\n\t\t\t\t\t\tFailoverVersion:   FailoverVersion1,\n\t\t\t\t\t},\n\t\t\t\t\tRegion2: {\n\t\t\t\t\t\tActiveClusterName: ClusterName2,\n\t\t\t\t\t\tFailoverVersion:   FailoverVersion2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tUpdateDomainResponse = types.UpdateDomainResponse{\n\t\tDomainInfo:               &DomainInfo,\n\t\tConfiguration:            &DomainConfiguration,\n\t\tReplicationConfiguration: &DomainReplicationConfiguration,\n\t\tFailoverVersion:          FailoverVersion1,\n\t\tIsGlobalDomain:           true,\n\t}\n\tDeleteDomainRequest = types.DeleteDomainRequest{\n\t\tName:          DomainName,\n\t\tSecurityToken: SecurityToken,\n\t}\n\tDeprecateDomainRequest = types.DeprecateDomainRequest{\n\t\tName:          DomainName,\n\t\tSecurityToken: SecurityToken,\n\t}\n\tListWorkflowExecutionsRequest = types.ListWorkflowExecutionsRequest{\n\t\tDomain:        DomainName,\n\t\tPageSize:      PageSize,\n\t\tNextPageToken: NextPageToken,\n\t\tQuery:         VisibilityQuery,\n\t}\n\tListWorkflowExecutionsResponse = types.ListWorkflowExecutionsResponse{\n\t\tExecutions:    WorkflowExecutionInfoArray,\n\t\tNextPageToken: NextPageToken,\n\t}\n\tListOpenWorkflowExecutionsRequest_ExecutionFilter = types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          DomainName,\n\t\tMaximumPageSize: PageSize,\n\t\tNextPageToken:   NextPageToken,\n\t\tStartTimeFilter: &StartTimeFilter,\n\t\tExecutionFilter: &WorkflowExecutionFilter,\n\t}\n\tListOpenWorkflowExecutionsRequest_TypeFilter = types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          DomainName,\n\t\tMaximumPageSize: PageSize,\n\t\tNextPageToken:   NextPageToken,\n\t\tStartTimeFilter: &StartTimeFilter,\n\t\tTypeFilter:      &WorkflowTypeFilter,\n\t}\n\tListOpenWorkflowExecutionsResponse = types.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions:    WorkflowExecutionInfoArray,\n\t\tNextPageToken: NextPageToken,\n\t}\n\tListClosedWorkflowExecutionsRequest_ExecutionFilter = types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          DomainName,\n\t\tMaximumPageSize: PageSize,\n\t\tNextPageToken:   NextPageToken,\n\t\tStartTimeFilter: &StartTimeFilter,\n\t\tExecutionFilter: &WorkflowExecutionFilter,\n\t}\n\tListClosedWorkflowExecutionsRequest_TypeFilter = types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          DomainName,\n\t\tMaximumPageSize: PageSize,\n\t\tNextPageToken:   NextPageToken,\n\t\tStartTimeFilter: &StartTimeFilter,\n\t\tTypeFilter:      &WorkflowTypeFilter,\n\t}\n\tListClosedWorkflowExecutionsRequest_StatusFilter = types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          DomainName,\n\t\tMaximumPageSize: PageSize,\n\t\tNextPageToken:   NextPageToken,\n\t\tStartTimeFilter: &StartTimeFilter,\n\t\tStatusFilter:    &WorkflowExecutionCloseStatus,\n\t}\n\tListClosedWorkflowExecutionsResponse = types.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions:    WorkflowExecutionInfoArray,\n\t\tNextPageToken: NextPageToken,\n\t}\n\tListArchivedWorkflowExecutionsRequest = types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:        DomainName,\n\t\tPageSize:      PageSize,\n\t\tNextPageToken: NextPageToken,\n\t\tQuery:         VisibilityQuery,\n\t}\n\tListArchivedWorkflowExecutionsResponse = types.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    WorkflowExecutionInfoArray,\n\t\tNextPageToken: NextPageToken,\n\t}\n\tCountWorkflowExecutionsRequest = types.CountWorkflowExecutionsRequest{\n\t\tDomain: DomainName,\n\t\tQuery:  VisibilityQuery,\n\t}\n\tCountWorkflowExecutionsResponse = types.CountWorkflowExecutionsResponse{\n\t\tCount: int64(8),\n\t}\n\tGetSearchAttributesResponse = types.GetSearchAttributesResponse{\n\t\tKeys: IndexedValueTypeMap,\n\t}\n\tPollForDecisionTaskRequest = types.PollForDecisionTaskRequest{\n\t\tDomain:         DomainName,\n\t\tTaskList:       &TaskList,\n\t\tIdentity:       Identity,\n\t\tBinaryChecksum: Checksum,\n\t}\n\tPollForDecisionTaskResponse = types.PollForDecisionTaskResponse{\n\t\tTaskToken:                 TaskToken,\n\t\tWorkflowExecution:         &WorkflowExecution,\n\t\tWorkflowType:              &WorkflowType,\n\t\tPreviousStartedEventID:    common.Int64Ptr(EventID1),\n\t\tStartedEventID:            EventID2,\n\t\tAttempt:                   Attempt,\n\t\tBacklogCountHint:          BacklogCountHint,\n\t\tHistory:                   &History,\n\t\tNextPageToken:             NextPageToken,\n\t\tQuery:                     &WorkflowQuery,\n\t\tWorkflowExecutionTaskList: &TaskList,\n\t\tScheduledTimestamp:        &Timestamp1,\n\t\tStartedTimestamp:          &Timestamp2,\n\t\tQueries:                   WorkflowQueryMap,\n\t\tNextEventID:               EventID3,\n\t\tAutoConfigHint:            &AutoConfigHint,\n\t}\n\tRespondDecisionTaskCompletedRequest = types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  TaskToken,\n\t\tDecisions:                  DecisionArray,\n\t\tExecutionContext:           ExecutionContext,\n\t\tIdentity:                   Identity,\n\t\tStickyAttributes:           &StickyExecutionAttributes,\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: true,\n\t\tBinaryChecksum:             Checksum,\n\t\tQueryResults:               WorkflowQueryResultMap,\n\t}\n\tRespondDecisionTaskCompletedResponse = types.RespondDecisionTaskCompletedResponse{\n\t\tDecisionTask:                &PollForDecisionTaskResponse,\n\t\tActivitiesToDispatchLocally: ActivityLocalDispatchInfoMap,\n\t}\n\tRespondDecisionTaskFailedRequest = types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken:      TaskToken,\n\t\tCause:          &DecisionTaskFailedCause,\n\t\tDetails:        Payload1,\n\t\tIdentity:       Identity,\n\t\tBinaryChecksum: Checksum,\n\t}\n\tPollForActivityTaskRequest = types.PollForActivityTaskRequest{\n\t\tDomain:           DomainName,\n\t\tTaskList:         &TaskList,\n\t\tIdentity:         Identity,\n\t\tTaskListMetadata: &TaskListMetadata,\n\t}\n\tPollForActivityTaskResponse = types.PollForActivityTaskResponse{\n\t\tTaskToken:                       TaskToken,\n\t\tWorkflowExecution:               &WorkflowExecution,\n\t\tActivityID:                      ActivityID,\n\t\tActivityType:                    &ActivityType,\n\t\tInput:                           Payload1,\n\t\tScheduledTimestamp:              &Timestamp1,\n\t\tScheduleToCloseTimeoutSeconds:   &Duration1,\n\t\tStartedTimestamp:                &Timestamp2,\n\t\tStartToCloseTimeoutSeconds:      &Duration2,\n\t\tHeartbeatTimeoutSeconds:         &Duration3,\n\t\tAttempt:                         Attempt,\n\t\tScheduledTimestampOfThisAttempt: &Timestamp3,\n\t\tHeartbeatDetails:                Payload2,\n\t\tWorkflowType:                    &WorkflowType,\n\t\tWorkflowDomain:                  DomainName,\n\t\tHeader:                          &Header,\n\t\tAutoConfigHint:                  &AutoConfigHint,\n\t}\n\tRespondActivityTaskCompletedRequest = types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: TaskToken,\n\t\tResult:    Payload1,\n\t\tIdentity:  Identity,\n\t}\n\tRespondActivityTaskCompletedByIDRequest = types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     DomainName,\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t\tActivityID: ActivityID,\n\t\tResult:     Payload1,\n\t\tIdentity:   Identity,\n\t}\n\tRespondActivityTaskFailedRequest = types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: TaskToken,\n\t\tReason:    &FailureReason,\n\t\tDetails:   FailureDetails,\n\t\tIdentity:  Identity,\n\t}\n\tRespondActivityTaskFailedByIDRequest = types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     DomainName,\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t\tActivityID: ActivityID,\n\t\tReason:     &FailureReason,\n\t\tDetails:    FailureDetails,\n\t\tIdentity:   Identity,\n\t}\n\tRespondActivityTaskCanceledRequest = types.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: TaskToken,\n\t\tDetails:   Payload1,\n\t\tIdentity:  Identity,\n\t}\n\tRespondActivityTaskCanceledByIDRequest = types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     DomainName,\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t\tActivityID: ActivityID,\n\t\tDetails:    Payload1,\n\t\tIdentity:   Identity,\n\t}\n\tRecordActivityTaskHeartbeatRequest = types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: TaskToken,\n\t\tDetails:   Payload1,\n\t\tIdentity:  Identity,\n\t}\n\tRecordActivityTaskHeartbeatResponse = types.RecordActivityTaskHeartbeatResponse{\n\t\tCancelRequested: true,\n\t}\n\tRecordActivityTaskHeartbeatByIDRequest = types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     DomainName,\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t\tActivityID: ActivityID,\n\t\tDetails:    Payload1,\n\t\tIdentity:   Identity,\n\t}\n\tRespondQueryTaskCompletedRequest = types.RespondQueryTaskCompletedRequest{\n\t\tTaskToken:         TaskToken,\n\t\tCompletedType:     &QueryTaskCompletedType,\n\t\tQueryResult:       Payload1,\n\t\tErrorMessage:      ErrorMessage,\n\t\tWorkerVersionInfo: &WorkerVersionInfo,\n\t}\n\tRequestCancelWorkflowExecutionRequest = types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain:              DomainName,\n\t\tWorkflowExecution:   &WorkflowExecution,\n\t\tIdentity:            Identity,\n\t\tRequestID:           RequestID,\n\t\tFirstExecutionRunID: RunID,\n\t}\n\tStartWorkflowExecutionRequest = types.StartWorkflowExecutionRequest{\n\t\tDomain:                              DomainName,\n\t\tWorkflowID:                          WorkflowID,\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tIdentity:                            Identity,\n\t\tRequestID:                           RequestID,\n\t\tWorkflowIDReusePolicy:               &WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tCronSchedule:                        CronSchedule,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tHeader:                              &Header,\n\t\tFirstRunAtTimeStamp:                 &Timestamp1,\n\t\tActiveClusterSelectionPolicy:        &ActiveClusterSelectionPolicyExternalEntity,\n\t}\n\tStartWorkflowExecutionResponse = types.StartWorkflowExecutionResponse{\n\t\tRunID: RunID,\n\t}\n\tStartWorkflowExecutionAsyncRequest = types.StartWorkflowExecutionAsyncRequest{\n\t\tStartWorkflowExecutionRequest: &StartWorkflowExecutionRequest,\n\t}\n\tStartWorkflowExecutionAsyncResponse = types.StartWorkflowExecutionAsyncResponse{}\n\tSignalWorkflowExecutionRequest      = types.SignalWorkflowExecutionRequest{\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tSignalName:        SignalName,\n\t\tInput:             Payload1,\n\t\tIdentity:          Identity,\n\t\tRequestID:         RequestID,\n\t\tControl:           Control,\n\t}\n\tSignalWithStartWorkflowExecutionRequest = types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              DomainName,\n\t\tWorkflowID:                          WorkflowID,\n\t\tWorkflowType:                        &WorkflowType,\n\t\tTaskList:                            &TaskList,\n\t\tInput:                               Payload1,\n\t\tExecutionStartToCloseTimeoutSeconds: &Duration1,\n\t\tTaskStartToCloseTimeoutSeconds:      &Duration2,\n\t\tIdentity:                            Identity,\n\t\tRequestID:                           RequestID,\n\t\tWorkflowIDReusePolicy:               &WorkflowIDReusePolicy,\n\t\tSignalName:                          SignalName,\n\t\tSignalInput:                         Payload2,\n\t\tControl:                             Control,\n\t\tRetryPolicy:                         &RetryPolicy,\n\t\tCronSchedule:                        CronSchedule,\n\t\tCronOverlapPolicy:                   &CronOverlapPolicy,\n\t\tMemo:                                &Memo,\n\t\tSearchAttributes:                    &SearchAttributes,\n\t\tHeader:                              &Header,\n\t\tFirstRunAtTimestamp:                 &Timestamp1,\n\t\tActiveClusterSelectionPolicy:        &ActiveClusterSelectionPolicyRegionSticky,\n\t}\n\tSignalWithStartWorkflowExecutionAsyncRequest = types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tSignalWithStartWorkflowExecutionRequest: &SignalWithStartWorkflowExecutionRequest,\n\t}\n\tSignalWithStartWorkflowExecutionAsyncResponse = types.SignalWithStartWorkflowExecutionAsyncResponse{}\n\tResetWorkflowExecutionRequest                 = types.ResetWorkflowExecutionRequest{\n\t\tDomain:                DomainName,\n\t\tWorkflowExecution:     &WorkflowExecution,\n\t\tReason:                Reason,\n\t\tDecisionFinishEventID: EventID1,\n\t\tRequestID:             RequestID,\n\t\tSkipSignalReapply:     true,\n\t}\n\tResetWorkflowExecutionResponse = types.ResetWorkflowExecutionResponse{\n\t\tRunID: RunID,\n\t}\n\tTerminateWorkflowExecutionRequest = types.TerminateWorkflowExecutionRequest{\n\t\tDomain:              DomainName,\n\t\tWorkflowExecution:   &WorkflowExecution,\n\t\tReason:              Reason,\n\t\tDetails:             Payload1,\n\t\tIdentity:            Identity,\n\t\tFirstExecutionRunID: RunID,\n\t}\n\tDescribeWorkflowExecutionRequest = types.DescribeWorkflowExecutionRequest{\n\t\tDomain:                DomainName,\n\t\tExecution:             &WorkflowExecution,\n\t\tQueryConsistencyLevel: &QueryConsistencyLevel,\n\t}\n\tDescribeWorkflowExecutionResponse = types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: &WorkflowExecutionConfiguration,\n\t\tWorkflowExecutionInfo:  &WorkflowExecutionInfo,\n\t\tPendingActivities:      PendingActivityInfoArray,\n\t\tPendingChildren:        PendingChildExecutionInfoArray,\n\t\tPendingDecision:        &PendingDecisionInfo,\n\t}\n\tDiagnoseWorkflowExecutionRequest = types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain:            DomainName,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tIdentity:          Identity,\n\t}\n\tDiagnoseWorkflowExecutionResponse = types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain:                      DomainName,\n\t\tDiagnosticWorkflowExecution: &WorkflowExecution,\n\t}\n\tQueryWorkflowRequest = types.QueryWorkflowRequest{\n\t\tDomain:                DomainName,\n\t\tExecution:             &WorkflowExecution,\n\t\tQuery:                 &WorkflowQuery,\n\t\tQueryRejectCondition:  &QueryRejectCondition,\n\t\tQueryConsistencyLevel: &QueryConsistencyLevel,\n\t}\n\tQueryWorkflowResponse = types.QueryWorkflowResponse{\n\t\tQueryResult:   Payload1,\n\t\tQueryRejected: &QueryRejected,\n\t}\n\tDescribeTaskListRequest = types.DescribeTaskListRequest{\n\t\tDomain:                DomainName,\n\t\tTaskList:              &TaskList,\n\t\tTaskListType:          &TaskListType,\n\t\tIncludeTaskListStatus: true,\n\t}\n\tDescribeTaskListResponse = types.DescribeTaskListResponse{\n\t\tPollers:         PollerInfoArray,\n\t\tTaskListStatus:  &TaskListStatus,\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t\tTaskList:        &TaskList,\n\t}\n\tListTaskListPartitionsRequest = types.ListTaskListPartitionsRequest{\n\t\tDomain:   DomainName,\n\t\tTaskList: &TaskList,\n\t}\n\tListTaskListPartitionsResponse = types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: TaskListPartitionMetadataArray,\n\t\tDecisionTaskListPartitions: TaskListPartitionMetadataArray,\n\t}\n\tResetStickyTaskListRequest = types.ResetStickyTaskListRequest{\n\t\tDomain:    DomainName,\n\t\tExecution: &WorkflowExecution,\n\t}\n\tResetStickyTaskListResponse        = types.ResetStickyTaskListResponse{}\n\tGetWorkflowExecutionHistoryRequest = types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 DomainName,\n\t\tExecution:              &WorkflowExecution,\n\t\tMaximumPageSize:        PageSize,\n\t\tNextPageToken:          NextPageToken,\n\t\tWaitForNewEvent:        true,\n\t\tHistoryEventFilterType: &HistoryEventFilterType,\n\t\tSkipArchival:           true,\n\t\tQueryConsistencyLevel:  &QueryConsistencyLevel,\n\t}\n\tGetWorkflowExecutionHistoryResponse = types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       &History,\n\t\tRawHistory:    DataBlobArray,\n\t\tNextPageToken: NextPageToken,\n\t\tArchived:      true,\n\t}\n\n\tDescribeDomainResponseArray = []*types.DescribeDomainResponse{\n\t\t&DescribeDomainResponse,\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/service_history.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\tHistoryCloseShardRequest           = AdminCloseShardRequest\n\tHistoryDescribeHistoryHostRequest  = types.DescribeHistoryHostRequest{}\n\tHistoryDescribeHistoryHostResponse = AdminDescribeHistoryHostResponse\n\tHistoryDescribeMutableStateRequest = types.DescribeMutableStateRequest{\n\t\tDomainUUID: DomainID,\n\t\tExecution:  &WorkflowExecution,\n\t}\n\tHistoryDescribeMutableStateResponse = types.DescribeMutableStateResponse{\n\t\tMutableStateInCache:    \"MutableStateInCache\",\n\t\tMutableStateInDatabase: \"MutableStateInDatabase\",\n\t}\n\tHistoryDescribeQueueRequest             = AdminDescribeQueueRequest\n\tHistoryDescribeQueueResponse            = AdminDescribeQueueResponse\n\tHistoryDescribeWorkflowExecutionRequest = types.HistoryDescribeWorkflowExecutionRequest{\n\t\tDomainUUID: DomainID,\n\t\tRequest:    &DescribeWorkflowExecutionRequest,\n\t}\n\tHistoryDescribeWorkflowExecutionResponse = DescribeWorkflowExecutionResponse\n\tHistoryGetDLQReplicationMessagesRequest  = AdminGetDLQReplicationMessagesRequest\n\tHistoryGetDLQReplicationMessagesResponse = AdminGetDLQReplicationMessagesResponse\n\tHistoryGetMutableStateRequest            = types.GetMutableStateRequest{\n\t\tDomainUUID:          DomainID,\n\t\tExecution:           &WorkflowExecution,\n\t\tExpectedNextEventID: EventID1,\n\t\tCurrentBranchToken:  BranchToken,\n\t\tVersionHistoryItem:  &VersionHistoryItem,\n\t}\n\tHistoryGetMutableStateResponse = types.GetMutableStateResponse{\n\t\tExecution:                            &WorkflowExecution,\n\t\tWorkflowType:                         &WorkflowType,\n\t\tNextEventID:                          EventID1,\n\t\tPreviousStartedEventID:               common.Int64Ptr(EventID2),\n\t\tLastFirstEventID:                     EventID3,\n\t\tTaskList:                             &TaskList,\n\t\tStickyTaskList:                       &TaskList,\n\t\tClientLibraryVersion:                 ClientLibraryVersion,\n\t\tClientFeatureVersion:                 FeatureVersion,\n\t\tClientImpl:                           ClientImpl,\n\t\tIsWorkflowRunning:                    true,\n\t\tStickyTaskListScheduleToStartTimeout: &Duration1,\n\t\tEventStoreVersion:                    EventStoreVersion,\n\t\tCurrentBranchToken:                   BranchToken,\n\t\tWorkflowState:                        common.Int32Ptr(1), // persistence.WorkflowStateRunning\n\t\tWorkflowCloseState:                   common.Int32Ptr(6), // persistence.WorkflowCloseStatusTimedOut\n\t\tVersionHistories:                     &VersionHistories,\n\t\tIsStickyTaskListEnabled:              true,\n\t\tHistorySize:                          HistorySizeInBytes,\n\t}\n\tHistoryGetReplicationMessagesRequest  = AdminGetReplicationMessagesRequest\n\tHistoryGetReplicationMessagesResponse = AdminGetReplicationMessagesResponse\n\tHistoryCountDLQMessagesRequest        = types.CountDLQMessagesRequest{ForceFetch: true}\n\tHistoryCountDLQMessagesResponse       = types.HistoryCountDLQMessagesResponse{Entries: map[types.HistoryDLQCountKey]int64{types.HistoryDLQCountKey{1, \"A\"}: 10}}\n\tHistoryMergeDLQMessagesRequest        = AdminMergeDLQMessagesRequest\n\tHistoryMergeDLQMessagesResponse       = AdminMergeDLQMessagesResponse\n\tHistoryNotifyFailoverMarkersRequest   = types.NotifyFailoverMarkersRequest{\n\t\tFailoverMarkerTokens: FailoverMarkerTokenArray,\n\t}\n\tHistoryPollMutableStateRequest = types.PollMutableStateRequest{\n\t\tDomainUUID:          DomainID,\n\t\tExecution:           &WorkflowExecution,\n\t\tExpectedNextEventID: EventID1,\n\t\tCurrentBranchToken:  BranchToken,\n\t}\n\tHistoryPollMutableStateResponse = types.PollMutableStateResponse{\n\t\tExecution:                            &WorkflowExecution,\n\t\tWorkflowType:                         &WorkflowType,\n\t\tNextEventID:                          EventID1,\n\t\tPreviousStartedEventID:               common.Int64Ptr(EventID2),\n\t\tLastFirstEventID:                     EventID3,\n\t\tTaskList:                             &TaskList,\n\t\tStickyTaskList:                       &TaskList,\n\t\tClientLibraryVersion:                 ClientLibraryVersion,\n\t\tClientFeatureVersion:                 FeatureVersion,\n\t\tClientImpl:                           ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: &Duration1,\n\t\tCurrentBranchToken:                   BranchToken,\n\t\tVersionHistories:                     &VersionHistories,\n\t\tWorkflowState:                        common.Int32Ptr(5), // persistence.WorkflowStateCorrupted\n\t\tWorkflowCloseState:                   common.Int32Ptr(6), // persistence.WorkflowCloseStatusTimedOut\n\t}\n\tHistoryPurgeDLQMessagesRequest = AdminPurgeDLQMessagesRequest\n\tHistoryQueryWorkflowRequest    = types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: DomainID,\n\t\tRequest:    &QueryWorkflowRequest,\n\t}\n\tHistoryQueryWorkflowResponse = types.HistoryQueryWorkflowResponse{\n\t\tResponse: &QueryWorkflowResponse,\n\t}\n\tHistoryReadDLQMessagesRequest  = AdminReadDLQMessagesRequest\n\tHistoryReadDLQMessagesResponse = AdminReadDLQMessagesResponse\n\tHistoryReapplyEventsRequest    = types.HistoryReapplyEventsRequest{\n\t\tDomainUUID: DomainID,\n\t\tRequest:    &AdminReapplyEventsRequest,\n\t}\n\tHistoryRecordActivityTaskHeartbeatRequest = types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID:       DomainID,\n\t\tHeartbeatRequest: &RecordActivityTaskHeartbeatRequest,\n\t}\n\tHistoryRecordActivityTaskHeartbeatResponse = RecordActivityTaskHeartbeatResponse\n\tHistoryRecordActivityTaskStartedRequest    = types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        DomainID,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tScheduleID:        EventID1,\n\t\tTaskID:            TaskID,\n\t\tRequestID:         RequestID,\n\t\tPollRequest:       &PollForActivityTaskRequest,\n\t}\n\tHistoryRecordActivityTaskStartedResponse = types.RecordActivityTaskStartedResponse{\n\t\tScheduledEvent:                  &HistoryEvent_WorkflowExecutionStarted,\n\t\tStartedTimestamp:                &Timestamp1,\n\t\tAttempt:                         Attempt,\n\t\tScheduledTimestampOfThisAttempt: &Timestamp2,\n\t\tHeartbeatDetails:                Payload1,\n\t\tWorkflowType:                    &WorkflowType,\n\t\tWorkflowDomain:                  DomainName,\n\t}\n\tHistoryRecordChildExecutionCompletedRequest = types.RecordChildExecutionCompletedRequest{\n\t\tDomainUUID:         DomainID,\n\t\tWorkflowExecution:  &WorkflowExecution,\n\t\tInitiatedID:        EventID1,\n\t\tCompletedExecution: &WorkflowExecution,\n\t\tCompletionEvent:    &HistoryEvent_WorkflowExecutionStarted,\n\t\tStartedID:          EventID2,\n\t}\n\tHistoryRecordDecisionTaskStartedRequest = types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        DomainID,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tScheduleID:        EventID1,\n\t\tTaskID:            TaskID,\n\t\tRequestID:         RequestID,\n\t\tPollRequest:       &PollForDecisionTaskRequest,\n\t}\n\tHistoryRecordDecisionTaskStartedResponse = types.RecordDecisionTaskStartedResponse{\n\t\tWorkflowType:              &WorkflowType,\n\t\tPreviousStartedEventID:    common.Int64Ptr(EventID1),\n\t\tScheduledEventID:          EventID2,\n\t\tStartedEventID:            EventID3,\n\t\tNextEventID:               EventID4,\n\t\tAttempt:                   Attempt,\n\t\tStickyExecutionEnabled:    true,\n\t\tDecisionInfo:              &TransientDecisionInfo,\n\t\tWorkflowExecutionTaskList: &TaskList,\n\t\tEventStoreVersion:         EventStoreVersion,\n\t\tBranchToken:               BranchToken,\n\t\tScheduledTimestamp:        &Timestamp1,\n\t\tStartedTimestamp:          &Timestamp2,\n\t\tQueries:                   WorkflowQueryMap,\n\t\tHistorySize:               HistorySizeInBytes,\n\t}\n\tHistoryRefreshWorkflowTasksRequest = types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: DomainID,\n\t\tRequest:    &AdminRefreshWorkflowTasksRequest,\n\t}\n\tHistoryRemoveSignalMutableStateRequest = types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID:        DomainID,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tRequestID:         RequestID,\n\t}\n\tHistoryRemoveTaskRequest        = AdminRemoveTaskRequest\n\tHistoryReplicateEventsV2Request = types.ReplicateEventsV2Request{\n\t\tDomainUUID:          DomainID,\n\t\tWorkflowExecution:   &WorkflowExecution,\n\t\tVersionHistoryItems: VersionHistoryItemArray,\n\t\tEvents:              &DataBlob,\n\t\tNewRunEvents:        &DataBlob,\n\t}\n\tHistoryRequestCancelWorkflowExecutionRequest = types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID:                DomainID,\n\t\tCancelRequest:             &RequestCancelWorkflowExecutionRequest,\n\t\tExternalInitiatedEventID:  common.Int64Ptr(EventID1),\n\t\tExternalWorkflowExecution: &WorkflowExecution,\n\t\tChildWorkflowOnly:         true,\n\t}\n\tHistoryResetQueueRequest          = AdminResetQueueRequest\n\tHistoryResetStickyTaskListRequest = types.HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: DomainID,\n\t\tExecution:  &WorkflowExecution,\n\t}\n\tHistoryResetWorkflowExecutionRequest = types.HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID:   DomainID,\n\t\tResetRequest: &ResetWorkflowExecutionRequest,\n\t}\n\tHistoryResetWorkflowExecutionResponse = types.ResetWorkflowExecutionResponse{\n\t\tRunID: RunID,\n\t}\n\tHistoryRespondActivityTaskCanceledRequest = types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID:    DomainID,\n\t\tCancelRequest: &RespondActivityTaskCanceledRequest,\n\t}\n\tHistoryRespondActivityTaskCompletedRequest = types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID:      DomainID,\n\t\tCompleteRequest: &RespondActivityTaskCompletedRequest,\n\t}\n\tHistoryRespondActivityTaskFailedRequest = types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID:    DomainID,\n\t\tFailedRequest: &RespondActivityTaskFailedRequest,\n\t}\n\tHistoryRespondDecisionTaskCompletedRequest = types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID:      DomainID,\n\t\tCompleteRequest: &RespondDecisionTaskCompletedRequest,\n\t}\n\tHistoryRespondDecisionTaskCompletedResponse = types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tStartedResponse:             &HistoryRecordDecisionTaskStartedResponse,\n\t\tActivitiesToDispatchLocally: ActivityLocalDispatchInfoMap,\n\t}\n\tHistoryRespondDecisionTaskFailedRequest = types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID:    DomainID,\n\t\tFailedRequest: &RespondDecisionTaskFailedRequest,\n\t}\n\tHistoryScheduleDecisionTaskRequest = types.ScheduleDecisionTaskRequest{\n\t\tDomainUUID:        DomainID,\n\t\tWorkflowExecution: &WorkflowExecution,\n\t\tIsFirstDecision:   true,\n\t}\n\tHistorySignalWithStartWorkflowExecutionRequest = types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID:             DomainID,\n\t\tSignalWithStartRequest: &SignalWithStartWorkflowExecutionRequest,\n\t\tPartitionConfig:        PartitionConfig,\n\t}\n\tHistorySignalWithStartWorkflowExecutionResponse = types.StartWorkflowExecutionResponse{\n\t\tRunID: RunID,\n\t}\n\tHistorySignalWorkflowExecutionRequest = types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID:                DomainID,\n\t\tSignalRequest:             &SignalWorkflowExecutionRequest,\n\t\tExternalWorkflowExecution: &WorkflowExecution,\n\t\tChildWorkflowOnly:         true,\n\t}\n\tHistoryStartWorkflowExecutionRequest = types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID:                      DomainID,\n\t\tStartRequest:                    &StartWorkflowExecutionRequest,\n\t\tParentExecutionInfo:             &ParentExecutionInfo,\n\t\tAttempt:                         Attempt,\n\t\tExpirationTimestamp:             &Timestamp1,\n\t\tContinueAsNewInitiator:          &ContinueAsNewInitiator,\n\t\tContinuedFailureReason:          &FailureReason,\n\t\tContinuedFailureDetails:         FailureDetails,\n\t\tLastCompletionResult:            Payload1,\n\t\tFirstDecisionTaskBackoffSeconds: &Duration1,\n\t\tPartitionConfig:                 PartitionConfig,\n\t}\n\tHistoryStartWorkflowExecutionResponse = types.StartWorkflowExecutionResponse{\n\t\tRunID: RunID,\n\t}\n\tHistorySyncActivityRequest = types.SyncActivityRequest{\n\t\tDomainID:           DomainID,\n\t\tWorkflowID:         WorkflowID,\n\t\tRunID:              RunID,\n\t\tVersion:            Version1,\n\t\tScheduledID:        EventID1,\n\t\tScheduledTime:      &Timestamp1,\n\t\tStartedID:          EventID1,\n\t\tStartedTime:        &Timestamp2,\n\t\tLastHeartbeatTime:  &Timestamp3,\n\t\tDetails:            Payload1,\n\t\tAttempt:            Attempt,\n\t\tLastFailureReason:  &FailureReason,\n\t\tLastWorkerIdentity: Identity,\n\t\tLastFailureDetails: FailureDetails,\n\t\tVersionHistory:     &VersionHistory,\n\t}\n\tHistorySyncShardStatusRequest = types.SyncShardStatusRequest{\n\t\tSourceCluster: ClusterName1,\n\t\tShardID:       ShardID,\n\t\tTimestamp:     &Timestamp1,\n\t}\n\tHistoryTerminateWorkflowExecutionRequest = types.HistoryTerminateWorkflowExecutionRequest{\n\t\tDomainUUID:                DomainID,\n\t\tTerminateRequest:          &TerminateWorkflowExecutionRequest,\n\t\tExternalWorkflowExecution: &WorkflowExecution,\n\t\tChildWorkflowOnly:         true,\n\t}\n\tHistoryGetCrossClusterTasksRequest               = GetCrossClusterTasksRequest\n\tHistoryGetCrossClusterTasksResponse              = GetCrossClusterTasksResponse\n\tHistoryRespondCrossClusterTasksCompletedRequest  = RespondCrossClusterTasksCompletedRequest\n\tHistoryRespondCrossClusterTasksCompletedResponse = RespondCrossClusterTasksCompletedResponse\n)\n"
  },
  {
    "path": "common/types/testdata/service_matching.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tForwardedFrom = \"ForwardedFrom\"\n\tPollerID      = \"PollerID\"\n)\n\nvar (\n\tTaskListPartitionConfig = types.TaskListPartitionConfig{\n\t\tVersion: 1,\n\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t0: {},\n\t\t\t1: {},\n\t\t\t2: {\n\t\t\t\tIsolationGroups: []string{\"foo\"},\n\t\t\t},\n\t\t},\n\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t0: {},\n\t\t\t1: {\n\t\t\t\tIsolationGroups: []string{\"bar\"},\n\t\t\t},\n\t\t},\n\t}\n\tLoadBalancerHints = types.LoadBalancerHints{\n\t\tBacklogCount:  1000,\n\t\tRatePerSecond: 1.0,\n\t}\n\tMatchingAddActivityTaskRequest = types.AddActivityTaskRequest{\n\t\tDomainUUID:                    DomainID,\n\t\tExecution:                     &WorkflowExecution,\n\t\tSourceDomainUUID:              DomainID,\n\t\tTaskList:                      &TaskList,\n\t\tScheduleID:                    EventID1,\n\t\tScheduleToStartTimeoutSeconds: &Duration1,\n\t\tSource:                        types.TaskSourceDbBacklog.Ptr(),\n\t\tForwardedFrom:                 ForwardedFrom,\n\t\tPartitionConfig:               PartitionConfig,\n\t}\n\tMatchingAddDecisionTaskRequest = types.AddDecisionTaskRequest{\n\t\tDomainUUID:                    DomainID,\n\t\tExecution:                     &WorkflowExecution,\n\t\tTaskList:                      &TaskList,\n\t\tScheduleID:                    EventID1,\n\t\tScheduleToStartTimeoutSeconds: &Duration1,\n\t\tSource:                        types.TaskSourceDbBacklog.Ptr(),\n\t\tForwardedFrom:                 ForwardedFrom,\n\t\tPartitionConfig:               PartitionConfig,\n\t}\n\tMatchingAddActivityTaskResponse = types.AddActivityTaskResponse{\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n\tMatchingAddDecisionTaskResponse = types.AddDecisionTaskResponse{\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n\tMatchingCancelOutstandingPollRequest = types.CancelOutstandingPollRequest{\n\t\tDomainUUID:   DomainID,\n\t\tTaskListType: common.Int32Ptr(int32(TaskListType)),\n\t\tTaskList:     &TaskList,\n\t\tPollerID:     PollerID,\n\t}\n\tMatchingDescribeTaskListRequest = types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID:  DomainID,\n\t\tDescRequest: &DescribeTaskListRequest,\n\t}\n\tMatchingDescribeTaskListResponse = types.DescribeTaskListResponse{\n\t\tPollers:        PollerInfoArray,\n\t\tTaskListStatus: &TaskListStatus,\n\t\tTaskList:       &TaskList,\n\t}\n\tMatchingListTaskListPartitionsRequest = types.MatchingListTaskListPartitionsRequest{\n\t\tDomain:   DomainName,\n\t\tTaskList: &TaskList,\n\t}\n\tMatchingListTaskListPartitionsResponse = types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: TaskListPartitionMetadataArray,\n\t\tDecisionTaskListPartitions: TaskListPartitionMetadataArray,\n\t}\n\tMatchingPollForActivityTaskRequest = types.MatchingPollForActivityTaskRequest{\n\t\tDomainUUID:     DomainID,\n\t\tPollerID:       PollerID,\n\t\tPollRequest:    &PollForActivityTaskRequest,\n\t\tForwardedFrom:  ForwardedFrom,\n\t\tIsolationGroup: IsolationGroup,\n\t}\n\tMatchingPollForActivityTaskResponse = types.MatchingPollForActivityTaskResponse{\n\t\tTaskToken:                       TaskToken,\n\t\tWorkflowExecution:               &WorkflowExecution,\n\t\tActivityID:                      ActivityID,\n\t\tActivityType:                    &ActivityType,\n\t\tInput:                           Payload1,\n\t\tScheduledTimestamp:              &Timestamp1,\n\t\tScheduleToCloseTimeoutSeconds:   &Duration1,\n\t\tStartedTimestamp:                &Timestamp2,\n\t\tStartToCloseTimeoutSeconds:      &Duration2,\n\t\tHeartbeatTimeoutSeconds:         &Duration3,\n\t\tAttempt:                         Attempt,\n\t\tScheduledTimestampOfThisAttempt: &Timestamp3,\n\t\tHeartbeatDetails:                Payload2,\n\t\tWorkflowType:                    &WorkflowType,\n\t\tWorkflowDomain:                  DomainName,\n\t\tHeader:                          &Header,\n\t\tPartitionConfig:                 &TaskListPartitionConfig,\n\t\tLoadBalancerHints:               &LoadBalancerHints,\n\t\tAutoConfigHint:                  &AutoConfigHint,\n\t}\n\tMatchingPollForDecisionTaskRequest = types.MatchingPollForDecisionTaskRequest{\n\t\tDomainUUID:     DomainID,\n\t\tPollerID:       PollerID,\n\t\tPollRequest:    &PollForDecisionTaskRequest,\n\t\tForwardedFrom:  ForwardedFrom,\n\t\tIsolationGroup: IsolationGroup,\n\t}\n\tMatchingPollForDecisionTaskResponse = types.MatchingPollForDecisionTaskResponse{\n\t\tTaskToken:                 TaskToken,\n\t\tWorkflowExecution:         &WorkflowExecution,\n\t\tWorkflowType:              &WorkflowType,\n\t\tPreviousStartedEventID:    common.Int64Ptr(EventID1),\n\t\tStartedEventID:            EventID2,\n\t\tAttempt:                   Attempt,\n\t\tNextEventID:               EventID3,\n\t\tBacklogCountHint:          BacklogCountHint,\n\t\tStickyExecutionEnabled:    true,\n\t\tQuery:                     &WorkflowQuery,\n\t\tDecisionInfo:              &TransientDecisionInfo,\n\t\tWorkflowExecutionTaskList: &TaskList,\n\t\tEventStoreVersion:         EventStoreVersion,\n\t\tBranchToken:               BranchToken,\n\t\tScheduledTimestamp:        &Timestamp1,\n\t\tStartedTimestamp:          &Timestamp2,\n\t\tQueries:                   WorkflowQueryMap,\n\t\tPartitionConfig:           &TaskListPartitionConfig,\n\t\tLoadBalancerHints:         &LoadBalancerHints,\n\t\tAutoConfigHint:            &AutoConfigHint,\n\t}\n\tMatchingQueryWorkflowRequest = types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID:    DomainID,\n\t\tTaskList:      &TaskList,\n\t\tQueryRequest:  &QueryWorkflowRequest,\n\t\tForwardedFrom: ForwardedFrom,\n\t}\n\tMatchingQueryWorkflowResponse = types.MatchingQueryWorkflowResponse{\n\t\tQueryResult:     Payload1,\n\t\tQueryRejected:   &QueryRejected,\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n\tMatchingRespondQueryTaskCompletedRequest = types.MatchingRespondQueryTaskCompletedRequest{\n\t\tDomainUUID:       DomainID,\n\t\tTaskList:         &TaskList,\n\t\tTaskID:           \"TaskID\",\n\t\tCompletedRequest: &RespondQueryTaskCompletedRequest,\n\t}\n\tMatchingGetTaskListsByDomainRequest = types.GetTaskListsByDomainRequest{\n\t\tDomain: DomainName,\n\t}\n\n\tGetTaskListsByDomainResponse = types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: DescribeTaskListResponseMap,\n\t\tActivityTaskListMap: DescribeTaskListResponseMap,\n\t}\n\n\tDescribeTaskListResponseMap = map[string]*types.DescribeTaskListResponse{DomainName: &MatchingDescribeTaskListResponse}\n\n\tMatchingActivityTaskDispatchInfo = types.ActivityTaskDispatchInfo{\n\t\tScheduledEvent:                  &HistoryEvent_WorkflowExecutionStarted,\n\t\tStartedTimestamp:                &Timestamp1,\n\t\tAttempt:                         &Attempt2,\n\t\tScheduledTimestampOfThisAttempt: &Timestamp1,\n\t\tHeartbeatDetails:                Payload2,\n\t\tWorkflowType:                    &WorkflowType,\n\t\tWorkflowDomain:                  DomainName,\n\t}\n\n\tMatchingUpdateTaskListPartitionConfigRequest = types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\tDomainUUID:      DomainID,\n\t\tTaskList:        &TaskList,\n\t\tTaskListType:    &TaskListType,\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n\n\tMatchingRefreshTaskListPartitionConfigRequest = types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\tDomainUUID:      DomainID,\n\t\tTaskList:        &TaskList,\n\t\tTaskListType:    &TaskListType,\n\t\tPartitionConfig: &TaskListPartitionConfig,\n\t}\n)\n"
  },
  {
    "path": "common/types/testdata/service_sharddistributor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testdata\n\nimport \"github.com/uber/cadence/common/types\"\n\nvar (\n\tShardDistributorGetShardOwnerRequest = types.GetShardOwnerRequest{\n\t\tShardKey:  \"shard-key\",\n\t\tNamespace: \"namespace\",\n\t}\n\tShardDistributorGetShardOwnerResponse = types.GetShardOwnerResponse{\n\t\tOwner:     \"owner\",\n\t\tNamespace: \"namespace\",\n\t\tMetadata:  map[string]string{\"key-1\": \"value-1\", \"key-2\": \"value-2\"},\n\t}\n\tShardDistributorExecutorHeartbeatRequest = types.ExecutorHeartbeatRequest{\n\t\tNamespace:  \"namespace\",\n\t\tExecutorID: \"executor-id\",\n\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\tShardStatusReports: map[string]*types.ShardStatusReport{\n\t\t\t\"shard-key-1\": {\n\t\t\t\tStatus:    types.ShardStatusREADY,\n\t\t\t\tShardLoad: 0.5,\n\t\t\t},\n\t\t\t\"shard-key-2\": {\n\t\t\t\tStatus:    types.ShardStatusINVALID,\n\t\t\t\tShardLoad: 0.75,\n\t\t\t},\n\t\t},\n\t\tMetadata: map[string]string{\n\t\t\t\"key-1\": \"value-1\",\n\t\t\t\"key-2\": \"value-2\",\n\t\t},\n\t}\n\tShardDistributorExecutorHeartbeatResponse = types.ExecutorHeartbeatResponse{\n\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\"shard-key-1\": {\n\t\t\t\tStatus: types.AssignmentStatusREADY,\n\t\t\t},\n\t\t\t\"shard-key-2\": {\n\t\t\t\tStatus: types.AssignmentStatusINVALID,\n\t\t\t},\n\t\t},\n\t}\n\tShardDistributorWatchNamespaceStateRequest = types.WatchNamespaceStateRequest{\n\t\tNamespace: \"namespace\",\n\t}\n\tShardDistributorWatchNamespaceStateResponse = types.WatchNamespaceStateResponse{\n\t\tExecutors: []*types.ExecutorShardAssignment{\n\t\t\t{\n\t\t\t\tExecutorID:     \"executor-1\",\n\t\t\t\tAssignedShards: []*types.Shard{&types.Shard{ShardKey: \"shard-1\"}, &types.Shard{ShardKey: \"shard-2\"}},\n\t\t\t\tMetadata:       map[string]string{\"key-1\": \"value-1\"},\n\t\t\t},\n\t\t\t{\n\t\t\t\tExecutorID:     \"executor-2\",\n\t\t\t\tAssignedShards: []*types.Shard{&types.Shard{ShardKey: \"shard-3\"}},\n\t\t\t\tMetadata:       map[string]string{\"key-2\": \"value-2\"},\n\t\t\t},\n\t\t},\n\t}\n)\n"
  },
  {
    "path": "common/util/file_util.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage util\n\nimport (\n\t\"errors\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"strings\"\n)\n\nvar (\n\t// ErrDirectoryExpected indicates that a directory was expected\n\tErrDirectoryExpected = errors.New(\"a path to a directory was expected\")\n\t// ErrFileExpected indicates that a file was expected\n\tErrFileExpected = errors.New(\"a path to a file was expected\")\n)\n\n// FileExists returns true if file exists, false otherwise.\n// Returns error if could not verify if file exists or if path specifies directory instead of file.\nfunc FileExists(filepath string) (bool, error) {\n\tif info, err := os.Stat(filepath); err != nil {\n\t\tif os.IsNotExist(err) {\n\t\t\treturn false, nil\n\t\t}\n\t\treturn false, err\n\t} else if info.IsDir() {\n\t\treturn false, ErrFileExpected\n\t}\n\treturn true, nil\n}\n\n// DirectoryExists returns true if directory exists, false otherwise.\n// Returns error if could not verify if directory exists or if path does not point at directory.\nfunc DirectoryExists(path string) (bool, error) {\n\tif info, err := os.Stat(path); err != nil {\n\t\tif os.IsNotExist(err) {\n\t\t\treturn false, nil\n\t\t}\n\t\treturn false, err\n\t} else if !info.IsDir() {\n\t\treturn false, ErrDirectoryExpected\n\t}\n\treturn true, nil\n}\n\n// MkdirAll creates all subdirectories in given path.\nfunc MkdirAll(path string, dirMode os.FileMode) error {\n\treturn os.MkdirAll(path, dirMode)\n}\n\n// WriteFile is used to write a file. If file already exists it is overwritten.\n// If file does not already exist it is created.\nfunc WriteFile(filepath string, data []byte, fileMode os.FileMode) (retErr error) {\n\tif err := os.Remove(filepath); err != nil && !os.IsNotExist(err) {\n\t\treturn err\n\t}\n\tf, err := os.Create(filepath)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer func() {\n\t\terr := f.Close()\n\t\tif err != nil {\n\t\t\tretErr = err\n\t\t}\n\t}()\n\tif err = f.Chmod(fileMode); err != nil {\n\t\treturn err\n\t}\n\tif _, err = f.Write(data); err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// ReadFile reads the contents of a file specified by filepath\n// WARNING: callers of this method should be extremely careful not to use it in a context where filepath is supplied by the user.\nfunc ReadFile(filepath string) ([]byte, error) {\n\t// #nosec\n\treturn ioutil.ReadFile(filepath)\n}\n\n// ListFiles lists all files in a directory.\nfunc ListFiles(dirPath string) ([]string, error) {\n\tif info, err := os.Stat(dirPath); err != nil {\n\t\treturn nil, err\n\t} else if !info.IsDir() {\n\t\treturn nil, ErrDirectoryExpected\n\t}\n\n\tf, err := os.Open(dirPath)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfileNames, err := f.Readdirnames(-1)\n\tf.Close()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn fileNames, nil\n}\n\n// ListFilesByPrefix lists all files in directory with prefix.\nfunc ListFilesByPrefix(dirPath string, prefix string) ([]string, error) {\n\tfileNames, err := ListFiles(dirPath)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar filteredFileNames []string\n\tfor _, name := range fileNames {\n\t\tif strings.HasPrefix(name, prefix) {\n\t\t\tfilteredFileNames = append(filteredFileNames, name)\n\t\t}\n\t}\n\treturn filteredFileNames, nil\n}\n"
  },
  {
    "path": "common/util/file_util_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage util\n\nimport (\n\t\"io/ioutil\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\nconst (\n\ttestDirMode  = os.FileMode(0700)\n\ttestFileMode = os.FileMode(0600)\n)\n\ntype FileUtilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestFileUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(FileUtilSuite))\n}\n\nfunc (s *FileUtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *FileUtilSuite) TestFileExists() {\n\tdir := s.T().TempDir()\n\n\texists, err := FileExists(dir)\n\ts.Error(err)\n\ts.False(exists)\n\n\tfilename := \"test-file-name\"\n\texists, err = FileExists(filepath.Join(dir, filename))\n\ts.NoError(err)\n\ts.False(exists)\n\n\ts.createFile(dir, filename)\n\texists, err = FileExists(filepath.Join(dir, filename))\n\ts.NoError(err)\n\ts.True(exists)\n}\n\nfunc (s *FileUtilSuite) TestDirectoryExists() {\n\tdir := s.T().TempDir()\n\n\tsubdir := \"subdir\"\n\texists, err := DirectoryExists(filepath.Join(dir, subdir))\n\ts.NoError(err)\n\ts.False(exists)\n\n\tfilename := \"test-file-name\"\n\ts.createFile(dir, filename)\n\tfpath := filepath.Join(dir, filename)\n\texists, err = DirectoryExists(fpath)\n\ts.Error(err)\n\ts.False(exists)\n}\n\nfunc (s *FileUtilSuite) TestMkdirAll() {\n\tdir := s.T().TempDir()\n\n\ts.NoError(MkdirAll(dir, testDirMode))\n\ts.assertDirectoryExists(dir)\n\n\tsubDirPath := filepath.Join(dir, \"subdir_1\", \"subdir_2\", \"subdir_3\")\n\ts.assertDirectoryNotExists(subDirPath)\n\ts.NoError(MkdirAll(subDirPath, testDirMode))\n\ts.assertDirectoryExists(subDirPath)\n\ts.assertCorrectFileMode(subDirPath)\n\n\tfilename := \"test-file-name\"\n\ts.createFile(dir, filename)\n\tfpath := filepath.Join(dir, filename)\n\ts.Error(MkdirAll(fpath, testDirMode))\n}\n\nfunc (s *FileUtilSuite) TestWriteFile() {\n\tdir := s.T().TempDir()\n\n\tfilename := \"test-file-name\"\n\tfpath := filepath.Join(dir, filename)\n\ts.NoError(WriteFile(fpath, []byte(\"file body 1\"), testFileMode))\n\ts.assertFileExists(fpath)\n\ts.assertCorrectFileMode(fpath)\n\n\ts.NoError(WriteFile(fpath, []byte(\"file body 2\"), testFileMode))\n\ts.assertFileExists(fpath)\n\ts.assertCorrectFileMode(fpath)\n\n\ts.Error(WriteFile(dir, []byte(\"\"), testFileMode))\n\ts.assertFileExists(fpath)\n}\n\nfunc (s *FileUtilSuite) TestReadFile() {\n\tdir := s.T().TempDir()\n\n\tfilename := \"test-file-name\"\n\tfpath := filepath.Join(dir, filename)\n\tdata, err := ReadFile(fpath)\n\ts.Error(err)\n\ts.Empty(data)\n\n\terr = WriteFile(fpath, []byte(\"file contents\"), testFileMode)\n\ts.NoError(err)\n\tdata, err = ReadFile(fpath)\n\ts.NoError(err)\n\ts.Equal(\"file contents\", string(data))\n}\n\nfunc (s *FileUtilSuite) TestListFilesByPrefix() {\n\tdir := s.T().TempDir()\n\n\tfilename := \"test-file-name\"\n\tfpath := filepath.Join(dir, filename)\n\tfiles, err := ListFilesByPrefix(fpath, \"test-\")\n\ts.Error(err)\n\ts.Nil(files)\n\n\tsubDirPath := filepath.Join(dir, \"subdir\")\n\ts.NoError(MkdirAll(subDirPath, testDirMode))\n\ts.assertDirectoryExists(subDirPath)\n\texpectedFileNames := []string{\"file_1\", \"file_2\", \"file_3\"}\n\tfor _, f := range expectedFileNames {\n\t\ts.createFile(dir, f)\n\t}\n\tfor _, f := range []string{\"randomFile\", \"fileWithOtherPrefix\"} {\n\t\ts.createFile(dir, f)\n\t}\n\tactualFileNames, err := ListFilesByPrefix(dir, \"file_\")\n\ts.NoError(err)\n\ts.Equal(len(expectedFileNames), len(actualFileNames))\n}\n\nfunc (s *FileUtilSuite) assertCorrectFileMode(path string) {\n\tinfo, err := os.Stat(path)\n\ts.NoError(err)\n\tmode := testFileMode\n\tif info.IsDir() {\n\t\tmode = testDirMode | os.ModeDir\n\t}\n\ts.Equal(mode, info.Mode())\n}\n\nfunc (s *FileUtilSuite) createFile(dir string, filename string) {\n\terr := ioutil.WriteFile(filepath.Join(dir, filename), []byte(\"file contents\"), testFileMode)\n\ts.Nil(err)\n}\n\nfunc (s *FileUtilSuite) assertFileExists(filepath string) {\n\texists, err := FileExists(filepath)\n\ts.NoError(err)\n\ts.True(exists)\n}\n\nfunc (s *FileUtilSuite) assertDirectoryExists(path string) {\n\texists, err := DirectoryExists(path)\n\ts.NoError(err)\n\ts.True(exists)\n}\n\nfunc (s *FileUtilSuite) assertDirectoryNotExists(path string) {\n\texists, err := DirectoryExists(path)\n\ts.NoError(err)\n\ts.False(exists)\n}\n"
  },
  {
    "path": "common/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage common\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/dgryski/go-farm\"\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tgolandMapReserverNumberOfBytes = 48\n\n\tretryPersistenceOperationInitialInterval    = 50 * time.Millisecond\n\tretryPersistenceOperationMaxInterval        = 10 * time.Second\n\tretryPersistenceOperationExpirationInterval = 30 * time.Second\n\n\thistoryServiceOperationInitialInterval    = 50 * time.Millisecond\n\thistoryServiceOperationMaxInterval        = 10 * time.Second\n\thistoryServiceOperationExpirationInterval = 30 * time.Second\n\n\trecordTaskStartedInitialInterval    = 50 * time.Millisecond\n\trecordTaskStartedMaxInterval        = 1 * time.Second\n\trecordTaskStartedExpirationInterval = 3 * time.Second\n\n\tmatchingServiceOperationInitialInterval    = 1000 * time.Millisecond\n\tmatchingServiceOperationMaxInterval        = 10 * time.Second\n\tmatchingServiceOperationExpirationInterval = 30 * time.Second\n\n\tfrontendServiceOperationInitialInterval    = 200 * time.Millisecond\n\tfrontendServiceOperationMaxInterval        = 5 * time.Second\n\tfrontendServiceOperationExpirationInterval = 15 * time.Second\n\n\tshardDistributorServiceOperationInitialInterval    = 200 * time.Millisecond\n\tshardDistributorServiceOperationMaxInterval        = 10 * time.Second\n\tshardDistributorServiceOperationExpirationInterval = 15 * time.Second\n\n\tadminServiceOperationInitialInterval    = 200 * time.Millisecond\n\tadminServiceOperationMaxInterval        = 5 * time.Second\n\tadminServiceOperationExpirationInterval = 15 * time.Second\n\n\tretryKafkaOperationInitialInterval = 50 * time.Millisecond\n\tretryKafkaOperationMaxInterval     = 10 * time.Second\n\tretryKafkaOperationMaxAttempts     = 10\n\n\tretryTaskProcessingInitialInterval = 50 * time.Millisecond\n\tretryTaskProcessingMaxInterval     = 100 * time.Millisecond\n\tretryTaskProcessingMaxAttempts     = 3\n\n\treplicationServiceBusyInitialInterval    = 2 * time.Second\n\treplicationServiceBusyMaxInterval        = 10 * time.Second\n\treplicationServiceBusyExpirationInterval = 5 * time.Minute\n\n\ttaskCompleterInitialInterval    = 1 * time.Second\n\ttaskCompleterMaxInterval        = 10 * time.Second\n\ttaskCompleterExpirationInterval = 5 * time.Minute\n\n\tdomainCacheInitialInterval    = 1 * time.Second\n\tdomainCacheMaxInterval        = 5 * time.Second\n\tdomainCacheExpirationInterval = 2 * time.Minute\n\n\tcontextExpireThreshold = 10 * time.Millisecond\n\n\t// FailureReasonCompleteResultExceedsLimit is failureReason for complete result exceeds limit\n\tFailureReasonCompleteResultExceedsLimit = \"COMPLETE_RESULT_EXCEEDS_LIMIT\"\n\t// FailureReasonFailureDetailsExceedsLimit is failureReason for failure details exceeds limit\n\tFailureReasonFailureDetailsExceedsLimit = \"FAILURE_DETAILS_EXCEEDS_LIMIT\"\n\t// FailureReasonCancelDetailsExceedsLimit is failureReason for cancel details exceeds limit\n\tFailureReasonCancelDetailsExceedsLimit = \"CANCEL_DETAILS_EXCEEDS_LIMIT\"\n\t// FailureReasonHeartbeatExceedsLimit is failureReason for heartbeat exceeds limit\n\tFailureReasonHeartbeatExceedsLimit = \"HEARTBEAT_EXCEEDS_LIMIT\"\n\t// FailureReasonDecisionBlobSizeExceedsLimit is the failureReason for decision blob exceeds size limit\n\tFailureReasonDecisionBlobSizeExceedsLimit = \"DECISION_BLOB_SIZE_EXCEEDS_LIMIT\"\n\t// FailureReasonSizeExceedsLimit is reason to fail workflow when history size or count exceed limit\n\tFailureReasonSizeExceedsLimit = \"HISTORY_EXCEEDS_LIMIT\"\n\t// FailureReasonTransactionSizeExceedsLimit is the failureReason for when transaction cannot be committed because it exceeds size limit\n\tFailureReasonTransactionSizeExceedsLimit = \"TRANSACTION_SIZE_EXCEEDS_LIMIT\"\n\t// FailureReasonDecisionAttemptsExceedsLimit is reason to fail workflow when decision attempts fail too many times\n\tFailureReasonDecisionAttemptsExceedsLimit = \"DECISION_ATTEMPTS_EXCEEDS_LIMIT\"\n\t// FailureReasonPendingActivityExceedsLimit is reason to fail overflow when pending activity exceeds limit\n\tFailureReasonPendingActivityExceedsLimit = \"PENDING_ACTIVITY_EXCEEDS_LIMIT\"\n)\n\nvar (\n\t// ErrBlobSizeExceedsLimit is error for event blob size exceeds limit\n\tErrBlobSizeExceedsLimit = &types.BadRequestError{Message: \"Blob data size exceeds limit.\"}\n\t// ErrContextTimeoutTooShort is error for setting a very short context timeout when calling a long poll API\n\tErrContextTimeoutTooShort = &types.BadRequestError{Message: \"Context timeout is too short.\"}\n\t// ErrContextTimeoutNotSet is error for not setting a context timeout when calling a long poll API\n\tErrContextTimeoutNotSet = &types.BadRequestError{Message: \"Context timeout is not set.\"}\n\t// ErrDecisionResultCountTooLarge error for decision result count exceeds limit\n\tErrDecisionResultCountTooLarge = &types.BadRequestError{Message: \"Decision result count exceeds limit.\"}\n\tstickyTaskListMetricTag        = metrics.TaskListTag(\"__sticky__\")\n\tephemeralTaskListMetricTag     = metrics.TaskListTag(\"__ephemeral__\")\n)\n\n// AwaitWaitGroup calls Wait on the given wait\n// Returns true if the Wait() call succeeded before the timeout\n// Returns false if the Wait() did not return before the timeout\nfunc AwaitWaitGroup(wg *sync.WaitGroup, timeout time.Duration) bool {\n\tdoneC := make(chan struct{})\n\n\tgo func() {\n\t\twg.Wait()\n\t\tclose(doneC)\n\t}()\n\n\tselect {\n\tcase <-doneC:\n\t\treturn true\n\tcase <-time.After(timeout):\n\t\treturn false\n\t}\n}\n\n// CreatePersistenceRetryPolicy creates a retry policy for persistence layer operations\nfunc CreatePersistenceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(retryPersistenceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(retryPersistenceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(retryPersistenceOperationExpirationInterval)\n\n\treturn policy\n}\n\n// CreateHistoryServiceRetryPolicy creates a retry policy for calls to history service\nfunc CreateHistoryServiceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(historyServiceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(historyServiceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(historyServiceOperationExpirationInterval)\n\n\treturn policy\n}\n\nfunc CreateRecordTaskStartedRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(recordTaskStartedInitialInterval)\n\tpolicy.SetMaximumInterval(recordTaskStartedMaxInterval)\n\tpolicy.SetExpirationInterval(recordTaskStartedExpirationInterval)\n\n\treturn policy\n}\n\n// CreateMatchingServiceRetryPolicy creates a retry policy for calls to matching service\nfunc CreateMatchingServiceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(matchingServiceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(matchingServiceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(matchingServiceOperationExpirationInterval)\n\n\treturn policy\n}\n\n// CreateFrontendServiceRetryPolicy creates a retry policy for calls to frontend service\nfunc CreateFrontendServiceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(frontendServiceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(frontendServiceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(frontendServiceOperationExpirationInterval)\n\n\treturn policy\n}\n\nfunc CreateShardDistributorServiceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(shardDistributorServiceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(shardDistributorServiceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(shardDistributorServiceOperationExpirationInterval)\n\n\treturn policy\n}\n\n// CreateAdminServiceRetryPolicy creates a retry policy for calls to matching service\nfunc CreateAdminServiceRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(adminServiceOperationInitialInterval)\n\tpolicy.SetMaximumInterval(adminServiceOperationMaxInterval)\n\tpolicy.SetExpirationInterval(adminServiceOperationExpirationInterval)\n\n\treturn policy\n}\n\n// CreateDlqPublishRetryPolicy creates a retry policy for kafka operation\nfunc CreateDlqPublishRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(retryKafkaOperationInitialInterval)\n\tpolicy.SetMaximumInterval(retryKafkaOperationMaxInterval)\n\tpolicy.SetMaximumAttempts(retryKafkaOperationMaxAttempts)\n\n\treturn policy\n}\n\n// CreateTaskProcessingRetryPolicy creates a retry policy for task processing\nfunc CreateTaskProcessingRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(retryTaskProcessingInitialInterval)\n\tpolicy.SetMaximumInterval(retryTaskProcessingMaxInterval)\n\tpolicy.SetMaximumAttempts(retryTaskProcessingMaxAttempts)\n\n\treturn policy\n}\n\n// CreateReplicationServiceBusyRetryPolicy creates a retry policy to handle replication service busy\nfunc CreateReplicationServiceBusyRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(replicationServiceBusyInitialInterval)\n\tpolicy.SetMaximumInterval(replicationServiceBusyMaxInterval)\n\tpolicy.SetExpirationInterval(replicationServiceBusyExpirationInterval)\n\n\treturn policy\n}\n\n// CreateTaskCompleterRetryPolicy creates a retry policy to handle tasks not being started\nfunc CreateTaskCompleterRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(taskCompleterInitialInterval)\n\tpolicy.SetMaximumInterval(taskCompleterMaxInterval)\n\tpolicy.SetExpirationInterval(taskCompleterExpirationInterval)\n\n\treturn policy\n}\n\n// CreateDomainCacheRetryPolicy creates a retry policy to handle domain cache refresh failures\nfunc CreateDomainCacheRetryPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(domainCacheInitialInterval)\n\tpolicy.SetMaximumInterval(domainCacheMaxInterval)\n\tpolicy.SetExpirationInterval(domainCacheExpirationInterval)\n\n\treturn policy\n}\n\n// IsValidIDLength checks if id is valid according to its length\nfunc IsValidIDLength(\n\tid string,\n\tscope metrics.Scope,\n\twarnLimit int,\n\terrorLimit int,\n\tmetricsCounter metrics.MetricIdx,\n\tdomainName string,\n\tlogger log.Logger,\n\tidTypeViolationTag tag.Tag,\n) bool {\n\tif len(id) > warnLimit {\n\t\tscope.IncCounter(metricsCounter)\n\t\tlogger.Warn(\"ID length exceeds limit.\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.Name(id),\n\t\t\tidTypeViolationTag)\n\t}\n\treturn len(id) <= errorLimit\n}\n\n// CheckDecisionResultLimit checks if decision result count exceeds limits.\nfunc CheckDecisionResultLimit(\n\tactualSize int,\n\tlimit int,\n\tscope metrics.Scope,\n) error {\n\tscope.RecordTimer(metrics.DecisionResultCount, time.Duration(actualSize))\n\tif limit > 0 && actualSize > limit {\n\t\treturn ErrDecisionResultCountTooLarge\n\t}\n\treturn nil\n}\n\n// ToServiceTransientError converts an error to ServiceTransientError\nfunc ToServiceTransientError(err error) error {\n\tif err == nil || IsServiceTransientError(err) {\n\t\treturn err\n\t}\n\treturn yarpcerrors.Newf(yarpcerrors.CodeUnavailable, err.Error())\n}\n\n// HistoryRetryFuncFrontendExceptions checks if an error should be retried\n// in a call from frontend\nfunc FrontendRetry(err error) bool {\n\tvar sbErr *types.ServiceBusyError\n\tif errors.As(err, &sbErr) {\n\t\t// If the service busy error is due to workflow id rate limiting, proxy it to the caller\n\t\treturn sbErr.Reason != constants.WorkflowIDRateLimitReason\n\t}\n\treturn IsServiceTransientError(err)\n}\n\n// IsExpectedError checks if an error is expected to happen in normal operation of the system\nfunc IsExpectedError(err error) bool {\n\treturn IsServiceTransientError(err) ||\n\t\tIsEntityNotExistsError(err) ||\n\t\terrors.As(err, new(*types.WorkflowExecutionAlreadyCompletedError))\n}\n\n// IsServiceTransientError checks if the error is a transient error.\nfunc IsServiceTransientError(err error) bool {\n\n\tvar (\n\t\ttypesInternalServiceError        *types.InternalServiceError\n\t\ttypesServiceBusyError            *types.ServiceBusyError\n\t\ttypesShardOwnershipLostError     *types.ShardOwnershipLostError\n\t\ttypesTaskListNotOwnedByHostError *cadence_errors.TaskListNotOwnedByHostError\n\t\tyarpcErrorsStatus                *yarpcerrors.Status\n\t)\n\n\tswitch {\n\tcase errors.As(err, &typesInternalServiceError):\n\t\treturn true\n\tcase errors.As(err, &typesServiceBusyError):\n\t\treturn true\n\tcase errors.As(err, &typesShardOwnershipLostError):\n\t\treturn true\n\tcase errors.As(err, &typesTaskListNotOwnedByHostError):\n\t\treturn true\n\tcase errors.As(err, &yarpcErrorsStatus):\n\t\t// We only selectively retry the following yarpc errors client can safe retry with a backoff\n\t\tif yarpcerrors.IsUnavailable(err) ||\n\t\t\tyarpcerrors.IsUnknown(err) ||\n\t\t\tyarpcerrors.IsCancelled(err) ||\n\t\t\tyarpcerrors.IsInternal(err) {\n\t\t\treturn true\n\t\t}\n\t\treturn false\n\t}\n\n\treturn false\n}\n\n// IsEntityNotExistsError checks if the error is an entity not exists error.\nfunc IsEntityNotExistsError(err error) bool {\n\t_, ok := err.(*types.EntityNotExistsError)\n\treturn ok\n}\n\n// IsServiceBusyError checks if the error is a service busy error.\nfunc IsServiceBusyError(err error) bool {\n\tswitch err.(type) {\n\tcase *types.ServiceBusyError:\n\t\treturn true\n\t}\n\treturn false\n}\n\n// IsContextTimeoutError checks if the error is context timeout error\nfunc IsContextTimeoutError(err error) bool {\n\tswitch err := err.(type) {\n\tcase *types.InternalServiceError:\n\t\treturn err.Message == context.DeadlineExceeded.Error()\n\t}\n\treturn err == context.DeadlineExceeded || yarpcerrors.IsDeadlineExceeded(err)\n}\n\n// WorkflowIDToHistoryShard is used to map a workflowID to a shardID\nfunc WorkflowIDToHistoryShard(workflowID string, numberOfShards int) int {\n\thash := farm.Fingerprint32([]byte(workflowID))\n\treturn int(hash % uint32(numberOfShards))\n}\n\n// DomainIDToHistoryShard is used to map a domainID to a shardID\nfunc DomainIDToHistoryShard(domainID string, numberOfShards int) int {\n\thash := farm.Fingerprint32([]byte(domainID))\n\treturn int(hash % uint32(numberOfShards))\n}\n\n// IsValidContext checks that the thrift context is not expired on cancelled.\n// Returns nil if the context is still valid. Otherwise, returns the result of\n// ctx.Err()\nfunc IsValidContext(ctx context.Context) error {\n\tch := ctx.Done()\n\tif ch != nil {\n\t\tselect {\n\t\tcase <-ch:\n\t\t\treturn ctx.Err()\n\t\tdefault:\n\t\t\t// go to the next line\n\t\t}\n\t}\n\n\tdeadline, ok := ctx.Deadline()\n\tif ok && time.Until(deadline) < contextExpireThreshold {\n\t\treturn context.DeadlineExceeded\n\t}\n\treturn nil\n}\n\n// emptyCancelFunc wraps an empty func by context.CancelFunc interface\nvar emptyCancelFunc = context.CancelFunc(func() {})\n\n// CreateChildContext creates a child context which shorted context timeout\n// from the given parent context\n// tailroom must be in range [0, 1] and\n// (1-tailroom) * parent timeout will be the new child context timeout\n// if tailroom is less 0, tailroom will be considered as 0\n// if tailroom is greater than 1, tailroom wil be considered as 1\nfunc CreateChildContext(\n\tparent context.Context,\n\ttailroom float64,\n) (context.Context, context.CancelFunc) {\n\tif parent == nil {\n\t\treturn nil, emptyCancelFunc\n\t}\n\tif parent.Err() != nil {\n\t\treturn parent, emptyCancelFunc\n\t}\n\n\tnow := time.Now()\n\tdeadline, ok := parent.Deadline()\n\tif !ok || deadline.Before(now) {\n\t\treturn parent, emptyCancelFunc\n\t}\n\n\t// if tailroom is about or less 0, then return a context with the same deadline as parent\n\tif tailroom <= 0 {\n\t\treturn context.WithDeadline(parent, deadline)\n\t}\n\t// if tailroom is about or greater 1, then return a context with deadline of now\n\tif tailroom >= 1 {\n\t\treturn context.WithDeadline(parent, now)\n\t}\n\n\tnewDeadline := now.Add(time.Duration(math.Ceil(float64(deadline.Sub(now)) * (1.0 - tailroom))))\n\treturn context.WithDeadline(parent, newDeadline)\n}\n\n// GenerateRandomString is used for generate test string\nfunc GenerateRandomString(n int) string {\n\tif n <= 0 {\n\t\treturn \"\"\n\t}\n\n\tletterRunes := []rune(\"random\")\n\tb := make([]rune, n)\n\tfor i := range b {\n\t\tb[i] = letterRunes[rand.Intn(len(letterRunes))]\n\t}\n\treturn string(b)\n}\n\n// CreateMatchingPollForDecisionTaskResponse create response for matching's PollForDecisionTask\nfunc CreateMatchingPollForDecisionTaskResponse(historyResponse *types.RecordDecisionTaskStartedResponse, workflowExecution *types.WorkflowExecution, token []byte) *types.MatchingPollForDecisionTaskResponse {\n\tmatchingResp := &types.MatchingPollForDecisionTaskResponse{\n\t\tWorkflowExecution:         workflowExecution,\n\t\tTaskToken:                 token,\n\t\tAttempt:                   historyResponse.GetAttempt(),\n\t\tWorkflowType:              historyResponse.WorkflowType,\n\t\tStartedEventID:            historyResponse.StartedEventID,\n\t\tStickyExecutionEnabled:    historyResponse.StickyExecutionEnabled,\n\t\tNextEventID:               historyResponse.NextEventID,\n\t\tDecisionInfo:              historyResponse.DecisionInfo,\n\t\tWorkflowExecutionTaskList: historyResponse.WorkflowExecutionTaskList,\n\t\tBranchToken:               historyResponse.BranchToken,\n\t\tScheduledTimestamp:        historyResponse.ScheduledTimestamp,\n\t\tStartedTimestamp:          historyResponse.StartedTimestamp,\n\t\tQueries:                   historyResponse.Queries,\n\t\tTotalHistoryBytes:         historyResponse.HistorySize,\n\t}\n\tif historyResponse.GetPreviousStartedEventID() != constants.EmptyEventID {\n\t\tmatchingResp.PreviousStartedEventID = historyResponse.PreviousStartedEventID\n\t}\n\treturn matchingResp\n}\n\n// ValidateRetryPolicy validates a retry policy\nfunc ValidateRetryPolicy(policy *types.RetryPolicy) error {\n\tif policy == nil {\n\t\t// nil policy is valid which means no retry\n\t\treturn nil\n\t}\n\tif policy.GetInitialIntervalInSeconds() <= 0 {\n\t\treturn &types.BadRequestError{Message: \"InitialIntervalInSeconds must be greater than 0 on retry policy.\"}\n\t}\n\tif policy.GetBackoffCoefficient() < 1 {\n\t\treturn &types.BadRequestError{Message: \"BackoffCoefficient cannot be less than 1 on retry policy.\"}\n\t}\n\tif policy.GetMaximumIntervalInSeconds() < 0 {\n\t\treturn &types.BadRequestError{Message: \"MaximumIntervalInSeconds cannot be less than 0 on retry policy.\"}\n\t}\n\tif policy.GetMaximumIntervalInSeconds() > 0 && policy.GetMaximumIntervalInSeconds() < policy.GetInitialIntervalInSeconds() {\n\t\treturn &types.BadRequestError{Message: \"MaximumIntervalInSeconds cannot be less than InitialIntervalInSeconds on retry policy.\"}\n\t}\n\tif policy.GetMaximumAttempts() < 0 {\n\t\treturn &types.BadRequestError{Message: \"MaximumAttempts cannot be less than 0 on retry policy.\"}\n\t}\n\tif policy.GetExpirationIntervalInSeconds() < 0 {\n\t\treturn &types.BadRequestError{Message: \"ExpirationIntervalInSeconds cannot be less than 0 on retry policy.\"}\n\t}\n\tif policy.GetMaximumAttempts() == 0 && policy.GetExpirationIntervalInSeconds() == 0 {\n\t\treturn &types.BadRequestError{Message: \"MaximumAttempts and ExpirationIntervalInSeconds are both 0. At least one of them must be specified.\"}\n\t}\n\treturn nil\n}\n\n// CreateHistoryStartWorkflowRequest create a start workflow request for history\nfunc CreateHistoryStartWorkflowRequest(\n\tdomainID string,\n\tstartRequest *types.StartWorkflowExecutionRequest,\n\tnow time.Time,\n\tpartitionConfig map[string]string,\n) (*types.HistoryStartWorkflowExecutionRequest, error) {\n\thistRequest := &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID:      domainID,\n\t\tStartRequest:    startRequest,\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tdelayStartSeconds := startRequest.GetDelayStartSeconds()\n\tjitterStartSeconds := startRequest.GetJitterStartSeconds()\n\tfirstRunAtTimestamp := startRequest.GetFirstRunAtTimeStamp()\n\n\tfirstDecisionTaskBackoffSeconds := delayStartSeconds\n\n\t// if the user specified a timestamp for the first run, we will use that as the start time,\n\t// ignoring the delayStartSeconds, jitterStartSeconds, and cronSchedule\n\t// The following condition guarantees two things:\n\t// - The logic is only triggered when the user specifies a first run timestamp\n\t// - AND that timestamp is only triggered ONCE hence not interfering with other scheduling logic\n\tif firstRunAtTimestamp > now.UnixNano() {\n\t\tfirstDecisionTaskBackoffSeconds = int32((firstRunAtTimestamp - now.UnixNano()) / int64(time.Second))\n\t} else {\n\t\tif len(startRequest.GetCronSchedule()) > 0 {\n\t\t\tdelayedStartTime := now.Add(time.Second * time.Duration(delayStartSeconds))\n\t\t\tvar err error\n\t\t\tfirstDecisionTaskBackoffSeconds, err = backoff.GetBackoffForNextScheduleInSeconds(\n\t\t\t\tstartRequest.GetCronSchedule(), delayedStartTime, delayedStartTime, jitterStartSeconds, startRequest.GetCronOverlapPolicy())\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\t// backoff seconds was calculated based on delayed start time, so we need to\n\t\t\t// add the delayStartSeconds to that backoff.\n\t\t\tfirstDecisionTaskBackoffSeconds += delayStartSeconds\n\t\t} else if jitterStartSeconds > 0 {\n\t\t\t// Add a random jitter to start time, if requested.\n\t\t\tfirstDecisionTaskBackoffSeconds += rand.Int31n(jitterStartSeconds + 1)\n\t\t}\n\t}\n\n\thistRequest.FirstDecisionTaskBackoffSeconds = Int32Ptr(firstDecisionTaskBackoffSeconds)\n\n\tif startRequest.RetryPolicy != nil && startRequest.RetryPolicy.GetExpirationIntervalInSeconds() > 0 {\n\t\texpirationInSeconds := startRequest.RetryPolicy.GetExpirationIntervalInSeconds() + firstDecisionTaskBackoffSeconds\n\t\t// expirationTime calculates from first decision task schedule to the end of the workflow\n\t\tdeadline := now.Add(time.Duration(expirationInSeconds) * time.Second)\n\t\thistRequest.ExpirationTimestamp = Int64Ptr(deadline.Round(time.Millisecond).UnixNano())\n\t}\n\n\treturn histRequest, nil\n}\n\n// CheckEventBlobSizeLimit checks if a blob data exceeds limits. It logs a warning if it exceeds warnLimit,\n// and return ErrBlobSizeExceedsLimit if it exceeds errorLimit.\nfunc CheckEventBlobSizeLimit(\n\tactualSize int,\n\twarnLimit int,\n\terrorLimit int,\n\tdomainID string,\n\tdomainName string,\n\tworkflowID string,\n\trunID string,\n\tscope metrics.Scope,\n\tlogger log.Logger,\n\tblobSizeViolationOperationTag tag.Tag,\n) error {\n\n\tscope.RecordTimer(metrics.EventBlobSize, time.Duration(actualSize))\n\n\tif errorLimit < warnLimit {\n\t\tlogger.Warn(\"Error limit is less than warn limit.\", tag.WorkflowDomainName(domainName), tag.WorkflowDomainID(domainID))\n\n\t\twarnLimit = errorLimit\n\t}\n\n\tif actualSize <= warnLimit {\n\t\treturn nil\n\t}\n\n\ttags := []tag.Tag{\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(workflowID),\n\t\ttag.WorkflowRunID(runID),\n\t\ttag.WorkflowSize(int64(actualSize)),\n\t\tblobSizeViolationOperationTag,\n\t}\n\n\tif actualSize <= errorLimit {\n\t\tlogger.Warn(\"Blob size close to the limit.\", tags...)\n\n\t\treturn nil\n\t}\n\n\tscope.Tagged(metrics.DomainTag(domainName)).IncCounter(metrics.EventBlobSizeExceedLimit)\n\n\tlogger.Error(\"Blob size exceeds limit.\", tags...)\n\n\treturn ErrBlobSizeExceedsLimit\n}\n\n// ValidateLongPollContextTimeout check if the context timeout for a long poll handler is too short or below a normal value.\n// If the timeout is not set or too short, it logs an error, and return ErrContextTimeoutNotSet or ErrContextTimeoutTooShort\n// accordingly. If the timeout is only below a normal value, it just logs an info and return nil.\nfunc ValidateLongPollContextTimeout(\n\tctx context.Context,\n\thandlerName string,\n\tlogger log.Logger,\n) error {\n\n\tdeadline, err := ValidateLongPollContextTimeoutIsSet(ctx, handlerName, logger)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttimeout := time.Until(deadline)\n\tif timeout < constants.MinLongPollTimeout {\n\t\terr := ErrContextTimeoutTooShort\n\t\tlogger.Error(\"Context timeout is too short for long poll API.\",\n\t\t\ttag.WorkflowHandlerName(handlerName), tag.Error(err), tag.WorkflowPollContextTimeout(timeout))\n\t\treturn err\n\t}\n\tif timeout < constants.CriticalLongPollTimeout {\n\t\tlogger.Debug(\"Context timeout is lower than critical value for long poll API.\",\n\t\t\ttag.WorkflowHandlerName(handlerName), tag.WorkflowPollContextTimeout(timeout))\n\t}\n\treturn nil\n}\n\n// ValidateLongPollContextTimeoutIsSet checks if the context timeout is set for long poll requests.\nfunc ValidateLongPollContextTimeoutIsSet(\n\tctx context.Context,\n\thandlerName string,\n\tlogger log.Logger,\n) (time.Time, error) {\n\n\tdeadline, ok := ctx.Deadline()\n\tif !ok {\n\t\terr := ErrContextTimeoutNotSet\n\t\tlogger.Error(\"Context timeout not set for long poll API.\",\n\t\t\ttag.WorkflowHandlerName(handlerName), tag.Error(err))\n\t\treturn deadline, err\n\t}\n\treturn deadline, nil\n}\n\n// ValidateDomainUUID checks if the given domainID string is a valid UUID\nfunc ValidateDomainUUID(\n\tdomainUUID string,\n) error {\n\n\tif domainUUID == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Missing domain UUID.\"}\n\t} else if uuid.Parse(domainUUID) == nil {\n\t\treturn &types.BadRequestError{Message: \"Invalid domain UUID.\"}\n\t}\n\treturn nil\n}\n\n// GetSizeOfMapStringToByteArray get size of map[string][]byte\nfunc GetSizeOfMapStringToByteArray(input map[string][]byte) int {\n\tif input == nil {\n\t\treturn 0\n\t}\n\n\tres := 0\n\tfor k, v := range input {\n\t\tres += len(k) + len(v)\n\t}\n\treturn res + golandMapReserverNumberOfBytes\n}\n\n// GetSizeOfHistoryEvent returns approximate size in bytes of the history event taking into account byte arrays only now\nfunc GetSizeOfHistoryEvent(event *types.HistoryEvent) uint64 {\n\tif event == nil || event.EventType == nil {\n\t\treturn 0\n\t}\n\n\tres := 0\n\tswitch *event.EventType {\n\tcase types.EventTypeWorkflowExecutionStarted:\n\t\tres += len(event.WorkflowExecutionStartedEventAttributes.Input)\n\t\tres += len(event.WorkflowExecutionStartedEventAttributes.ContinuedFailureDetails)\n\t\tres += len(event.WorkflowExecutionStartedEventAttributes.LastCompletionResult)\n\t\tif event.WorkflowExecutionStartedEventAttributes.Memo != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionStartedEventAttributes.Memo.Fields)\n\t\t}\n\t\tif event.WorkflowExecutionStartedEventAttributes.Header != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionStartedEventAttributes.Header.Fields)\n\t\t}\n\t\tif event.WorkflowExecutionStartedEventAttributes.SearchAttributes != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionStartedEventAttributes.SearchAttributes.IndexedFields)\n\t\t}\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tres += len(event.WorkflowExecutionCompletedEventAttributes.Result)\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tres += len(event.WorkflowExecutionFailedEventAttributes.Details)\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\tcase types.EventTypeDecisionTaskScheduled:\n\tcase types.EventTypeDecisionTaskStarted:\n\tcase types.EventTypeDecisionTaskCompleted:\n\t\tres += len(event.DecisionTaskCompletedEventAttributes.ExecutionContext)\n\tcase types.EventTypeDecisionTaskTimedOut:\n\tcase types.EventTypeDecisionTaskFailed:\n\t\tres += len(event.DecisionTaskFailedEventAttributes.Details)\n\tcase types.EventTypeActivityTaskScheduled:\n\t\tres += len(event.ActivityTaskScheduledEventAttributes.Input)\n\t\tif event.ActivityTaskScheduledEventAttributes.Header != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.ActivityTaskScheduledEventAttributes.Header.Fields)\n\t\t}\n\tcase types.EventTypeActivityTaskStarted:\n\t\tres += len(event.ActivityTaskStartedEventAttributes.LastFailureDetails)\n\tcase types.EventTypeActivityTaskCompleted:\n\t\tres += len(event.ActivityTaskCompletedEventAttributes.Result)\n\tcase types.EventTypeActivityTaskFailed:\n\t\tres += len(event.ActivityTaskFailedEventAttributes.Details)\n\tcase types.EventTypeActivityTaskTimedOut:\n\t\tres += len(event.ActivityTaskTimedOutEventAttributes.Details)\n\t\tres += len(event.ActivityTaskTimedOutEventAttributes.LastFailureDetails)\n\tcase types.EventTypeActivityTaskCancelRequested:\n\tcase types.EventTypeRequestCancelActivityTaskFailed:\n\tcase types.EventTypeActivityTaskCanceled:\n\t\tres += len(event.ActivityTaskCanceledEventAttributes.Details)\n\tcase types.EventTypeTimerStarted:\n\tcase types.EventTypeTimerFired:\n\tcase types.EventTypeCancelTimerFailed:\n\tcase types.EventTypeTimerCanceled:\n\tcase types.EventTypeWorkflowExecutionCancelRequested:\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tres += len(event.WorkflowExecutionCanceledEventAttributes.Details)\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\tres += len(event.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.Control)\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\tres += len(event.RequestCancelExternalWorkflowExecutionFailedEventAttributes.Control)\n\tcase types.EventTypeExternalWorkflowExecutionCancelRequested:\n\tcase types.EventTypeMarkerRecorded:\n\t\tres += len(event.MarkerRecordedEventAttributes.Details)\n\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\tres += len(event.WorkflowExecutionSignaledEventAttributes.Input)\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tres += len(event.WorkflowExecutionTerminatedEventAttributes.Details)\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tres += len(event.WorkflowExecutionContinuedAsNewEventAttributes.Input)\n\t\tif event.WorkflowExecutionContinuedAsNewEventAttributes.Memo != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionContinuedAsNewEventAttributes.Memo.Fields)\n\t\t}\n\t\tif event.WorkflowExecutionContinuedAsNewEventAttributes.Header != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionContinuedAsNewEventAttributes.Header.Fields)\n\t\t}\n\t\tif event.WorkflowExecutionContinuedAsNewEventAttributes.SearchAttributes != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.WorkflowExecutionContinuedAsNewEventAttributes.SearchAttributes.IndexedFields)\n\t\t}\n\tcase types.EventTypeStartChildWorkflowExecutionInitiated:\n\t\tres += len(event.StartChildWorkflowExecutionInitiatedEventAttributes.Input)\n\t\tres += len(event.StartChildWorkflowExecutionInitiatedEventAttributes.Control)\n\t\tif event.StartChildWorkflowExecutionInitiatedEventAttributes.Memo != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.StartChildWorkflowExecutionInitiatedEventAttributes.Memo.Fields)\n\t\t}\n\t\tif event.StartChildWorkflowExecutionInitiatedEventAttributes.Header != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.StartChildWorkflowExecutionInitiatedEventAttributes.Header.Fields)\n\t\t}\n\t\tif event.StartChildWorkflowExecutionInitiatedEventAttributes.SearchAttributes != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.StartChildWorkflowExecutionInitiatedEventAttributes.SearchAttributes.IndexedFields)\n\t\t}\n\tcase types.EventTypeStartChildWorkflowExecutionFailed:\n\t\tres += len(event.StartChildWorkflowExecutionFailedEventAttributes.Control)\n\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\tif event.ChildWorkflowExecutionStartedEventAttributes != nil && event.ChildWorkflowExecutionStartedEventAttributes.Header != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.ChildWorkflowExecutionStartedEventAttributes.Header.Fields)\n\t\t}\n\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\tres += len(event.ChildWorkflowExecutionCompletedEventAttributes.Result)\n\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\tres += len(event.ChildWorkflowExecutionFailedEventAttributes.Details)\n\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\tres += len(event.ChildWorkflowExecutionCanceledEventAttributes.Details)\n\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\tcase types.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\tres += len(event.SignalExternalWorkflowExecutionInitiatedEventAttributes.Input)\n\t\tres += len(event.SignalExternalWorkflowExecutionInitiatedEventAttributes.Control)\n\tcase types.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\tres += len(event.SignalExternalWorkflowExecutionFailedEventAttributes.Control)\n\tcase types.EventTypeExternalWorkflowExecutionSignaled:\n\t\tres += len(event.ExternalWorkflowExecutionSignaledEventAttributes.Control)\n\tcase types.EventTypeUpsertWorkflowSearchAttributes:\n\t\tif event.UpsertWorkflowSearchAttributesEventAttributes.SearchAttributes != nil {\n\t\t\tres += GetSizeOfMapStringToByteArray(event.UpsertWorkflowSearchAttributesEventAttributes.SearchAttributes.IndexedFields)\n\t\t}\n\t}\n\treturn uint64(res)\n}\n\n// IsJustOrderByClause return true is query start with order by\nfunc IsJustOrderByClause(clause string) bool {\n\twhereClause := strings.TrimSpace(clause)\n\twhereClause = strings.ToLower(whereClause)\n\treturn strings.HasPrefix(whereClause, \"order by\")\n}\n\n// ConvertIndexedValueTypeToInternalType takes fieldType as interface{} and convert to IndexedValueType.\n// Because different implementation of dynamic config client may lead to different types\nfunc ConvertIndexedValueTypeToInternalType(fieldType interface{}, logger log.Logger) types.IndexedValueType {\n\tswitch t := fieldType.(type) {\n\tcase float64:\n\t\treturn types.IndexedValueType(t)\n\tcase int:\n\t\treturn types.IndexedValueType(t)\n\tcase types.IndexedValueType:\n\t\treturn t\n\tcase []byte:\n\t\tvar result types.IndexedValueType\n\t\tif err := result.UnmarshalText(t); err != nil {\n\t\t\tlogger.Error(\"unknown index value type\", tag.Value(fieldType), tag.ValueType(t), tag.Error(err))\n\t\t\treturn fieldType.(types.IndexedValueType) // it will panic and been captured by logger\n\t\t}\n\t\treturn result\n\tcase string:\n\t\tvar result types.IndexedValueType\n\t\tif err := result.UnmarshalText([]byte(t)); err != nil {\n\t\t\tlogger.Error(\"unknown index value type\", tag.Value(fieldType), tag.ValueType(t), tag.Error(err))\n\t\t\treturn fieldType.(types.IndexedValueType) // it will panic and been captured by logger\n\t\t}\n\t\treturn result\n\tdefault:\n\t\t// Unknown fieldType, please make sure dynamic config return correct value type\n\t\tlogger.Error(\"unknown index value type\", tag.Value(fieldType), tag.ValueType(t))\n\t\treturn fieldType.(types.IndexedValueType) // it will panic and been captured by logger\n\t}\n}\n\n// DeserializeSearchAttributeValue takes json encoded search attribute value and it's type as input, then\n// unmarshal the value into a concrete type and return the value\nfunc DeserializeSearchAttributeValue(value []byte, valueType types.IndexedValueType) (interface{}, error) {\n\tswitch valueType {\n\tcase types.IndexedValueTypeString, types.IndexedValueTypeKeyword:\n\t\tvar val string\n\t\tif err := json.Unmarshal(value, &val); err != nil {\n\t\t\tvar listVal []string\n\t\t\terr = json.Unmarshal(value, &listVal)\n\t\t\treturn listVal, err\n\t\t}\n\t\treturn val, nil\n\tcase types.IndexedValueTypeInt:\n\t\tvar val int64\n\t\tif err := json.Unmarshal(value, &val); err != nil {\n\t\t\tvar listVal []int64\n\t\t\terr = json.Unmarshal(value, &listVal)\n\t\t\treturn listVal, err\n\t\t}\n\t\treturn val, nil\n\tcase types.IndexedValueTypeDouble:\n\t\tvar val float64\n\t\tif err := json.Unmarshal(value, &val); err != nil {\n\t\t\tvar listVal []float64\n\t\t\terr = json.Unmarshal(value, &listVal)\n\t\t\treturn listVal, err\n\t\t}\n\t\treturn val, nil\n\tcase types.IndexedValueTypeBool:\n\t\tvar val bool\n\t\tif err := json.Unmarshal(value, &val); err != nil {\n\t\t\tvar listVal []bool\n\t\t\terr = json.Unmarshal(value, &listVal)\n\t\t\treturn listVal, err\n\t\t}\n\t\treturn val, nil\n\tcase types.IndexedValueTypeDatetime:\n\t\tvar val time.Time\n\t\tif err := json.Unmarshal(value, &val); err != nil {\n\t\t\tvar listVal []time.Time\n\t\t\terr = json.Unmarshal(value, &listVal)\n\t\t\treturn listVal, err\n\t\t}\n\t\treturn val, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"error: unknown index value type [%v]\", valueType)\n\t}\n}\n\n// IsAdvancedVisibilityWritingEnabled returns true if we should write to advanced visibility\nfunc IsAdvancedVisibilityWritingEnabled(advancedVisibilityWritingMode string, isAdvancedVisConfigExist bool) bool {\n\treturn advancedVisibilityWritingMode != constants.AdvancedVisibilityModeOff && isAdvancedVisConfigExist\n}\n\n// IsAdvancedVisibilityReadingEnabled returns true if we should read from advanced visibility\nfunc IsAdvancedVisibilityReadingEnabled(isAdvancedVisReadEnabled, isAdvancedVisConfigExist bool) bool {\n\treturn isAdvancedVisReadEnabled && isAdvancedVisConfigExist\n}\n\n// IsStickyTaskConditionError is error from matching engine\nfunc IsStickyTaskConditionError(err error) bool {\n\tif e, ok := err.(*types.InternalServiceError); ok {\n\t\treturn e.GetMessage() == constants.StickyTaskConditionFailedErrorMsg\n\t}\n\treturn false\n}\n\n// DurationToDays converts time.Duration to number of 24 hour days\nfunc DurationToDays(d time.Duration) int32 {\n\treturn int32(d / (24 * time.Hour))\n}\n\n// DurationToSeconds converts time.Duration to number of seconds\nfunc DurationToSeconds(d time.Duration) int64 {\n\treturn int64(d / time.Second)\n}\n\n// DaysToDuration converts number of 24 hour days to time.Duration\nfunc DaysToDuration(d int32) time.Duration {\n\treturn time.Duration(d) * (24 * time.Hour)\n}\n\n// SecondsToDuration converts number of seconds to time.Duration\nfunc SecondsToDuration(d int64) time.Duration {\n\treturn time.Duration(d) * time.Second\n}\n\n// SleepWithMinDuration sleeps for the minimum of desired and available duration\n// returns the remaining available time duration\nfunc SleepWithMinDuration(desired time.Duration, available time.Duration) time.Duration {\n\td := min(desired, available)\n\tif d > 0 {\n\t\ttime.Sleep(d)\n\t}\n\treturn available - d\n}\n\n// ConvertErrToGetTaskFailedCause converts error to GetTaskFailedCause\nfunc ConvertErrToGetTaskFailedCause(err error) types.GetTaskFailedCause {\n\tif IsContextTimeoutError(err) {\n\t\treturn types.GetTaskFailedCauseTimeout\n\t}\n\tif IsServiceBusyError(err) {\n\t\treturn types.GetTaskFailedCauseServiceBusy\n\t}\n\tif _, ok := err.(*types.ShardOwnershipLostError); ok {\n\t\treturn types.GetTaskFailedCauseShardOwnershipLost\n\t}\n\treturn types.GetTaskFailedCauseUncategorized\n}\n\n// ConvertGetTaskFailedCauseToErr converts GetTaskFailedCause to error\nfunc ConvertGetTaskFailedCauseToErr(failedCause types.GetTaskFailedCause) error {\n\tswitch failedCause {\n\tcase types.GetTaskFailedCauseServiceBusy:\n\t\treturn &types.ServiceBusyError{}\n\tcase types.GetTaskFailedCauseTimeout:\n\t\treturn context.DeadlineExceeded\n\tcase types.GetTaskFailedCauseShardOwnershipLost:\n\t\treturn &types.ShardOwnershipLostError{}\n\tdefault:\n\t\treturn &types.InternalServiceError{Message: \"uncategorized error\"}\n\t}\n}\n\n// IntersectionStringSlice get the intersection of 2 string slices\nfunc IntersectionStringSlice(a, b []string) []string {\n\tvar result []string\n\tm := make(map[string]struct{})\n\tfor _, item := range a {\n\t\tm[item] = struct{}{}\n\t}\n\tfor _, item := range b {\n\t\tif _, ok := m[item]; ok {\n\t\t\tresult = append(result, item)\n\t\t}\n\t}\n\treturn result\n}\n\n// GetTaskListTag returns the task list tag for the given task list name and kind\nfunc GetTaskListTag(taskListName string, taskListKind types.TaskListKind) metrics.Tag {\n\ttaskListTag := metrics.TaskListUnknownTag()\n\tif taskListName != \"\" && taskListKind == types.TaskListKindNormal {\n\t\ttaskListTag = metrics.TaskListTag(taskListName)\n\t}\n\tif taskListKind == types.TaskListKindSticky {\n\t\ttaskListTag = stickyTaskListMetricTag\n\t}\n\tif taskListKind == types.TaskListKindEphemeral {\n\t\ttaskListTag = ephemeralTaskListMetricTag\n\t}\n\treturn taskListTag\n}\n\n// NewPerTaskListScope creates a tasklist metrics scope\nfunc NewPerTaskListScope(\n\tdomainName string,\n\ttaskListName string,\n\ttaskListKind types.TaskListKind,\n\tclient metrics.Client,\n\tscopeIdx metrics.ScopeIdx,\n) metrics.Scope {\n\tdomainTag := metrics.DomainUnknownTag()\n\tif domainName != \"\" {\n\t\tdomainTag = metrics.DomainTag(domainName)\n\t}\n\ttaskListTag := GetTaskListTag(taskListName, taskListKind)\n\treturn client.Scope(scopeIdx, domainTag, taskListTag)\n}\n"
  },
  {
    "path": "common/util_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"reflect\"\n\t\"runtime\"\n\t\"strconv\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestIsServiceTransientError(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\terr  error\n\t\twant bool\n\t}{\n\t\t\"ContextTimeout\": {\n\t\t\terr:  context.DeadlineExceeded,\n\t\t\twant: false,\n\t\t},\n\t\t\"YARPCCanceled\": {\n\t\t\terr:  yarpcerrors.CancelledErrorf(\"connection closing\"),\n\t\t\twant: true,\n\t\t},\n\t\t\"YARPCDeadlineExceeded\": {\n\t\t\terr:  yarpcerrors.DeadlineExceededErrorf(\"yarpc deadline exceeded\"),\n\t\t\twant: false,\n\t\t},\n\t\t\"YARPCUnavailable\": {\n\t\t\terr:  yarpcerrors.UnavailableErrorf(\"yarpc unavailable\"),\n\t\t\twant: true,\n\t\t},\n\t\t\"YARPCUnavailable wrapped\": {\n\t\t\terr:  fmt.Errorf(\"wrapped err: %w\", yarpcerrors.UnavailableErrorf(\"yarpc unavailable\")),\n\t\t\twant: true,\n\t\t},\n\t\t\"YARPCUnknown\": {\n\t\t\terr:  yarpcerrors.UnknownErrorf(\"yarpc unknown\"),\n\t\t\twant: true,\n\t\t},\n\t\t\"YARPCInternal\": {\n\t\t\terr:  yarpcerrors.InternalErrorf(\"yarpc internal\"),\n\t\t\twant: true,\n\t\t},\n\t\t\"ContextCancel\": {\n\t\t\terr:  context.Canceled,\n\t\t\twant: false,\n\t\t},\n\t\t\"ServiceBusyError\": {\n\t\t\terr:  &types.ServiceBusyError{},\n\t\t\twant: true,\n\t\t},\n\t\t\"ShardOwnershipLostError\": {\n\t\t\terr:  &types.ShardOwnershipLostError{},\n\t\t\twant: true,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\trequire.Equal(t, c.want, IsServiceTransientError(c.err))\n\t\t})\n\t}\n}\n\nfunc TestIsExpectedError(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\terr  error\n\t\twant bool\n\t}{\n\t\t\"An error\": {\n\t\t\terr:  assert.AnError,\n\t\t\twant: false,\n\t\t},\n\t\t\"Transient error\": {\n\t\t\terr:  &types.ServiceBusyError{},\n\t\t\twant: true,\n\t\t},\n\t\t\"Entity not exists error\": {\n\t\t\terr:  &types.EntityNotExistsError{},\n\t\t\twant: true,\n\t\t},\n\t\t\"Already completed error\": {\n\t\t\terr:  &types.WorkflowExecutionAlreadyCompletedError{},\n\t\t\twant: true,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\trequire.Equal(t, c.want, IsExpectedError(c.err))\n\t\t})\n\t}\n}\n\nfunc TestFrontendRetry(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\terr  error\n\t\twant bool\n\t}{\n\t\t{\n\t\t\tname: \"ServiceBusyError due to workflow id rate limiting\",\n\t\t\terr:  &types.ServiceBusyError{Reason: constants.WorkflowIDRateLimitReason},\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"ServiceBusyError not due to workflow id rate limiting\",\n\t\t\terr:  &types.ServiceBusyError{Reason: \"some other reason\"},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"ServiceBusyError empty reason\",\n\t\t\terr:  &types.ServiceBusyError{Reason: \"\"},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Non-ServiceBusyError\",\n\t\t\terr:  errors.New(\"some random error\"),\n\t\t\twant: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\trequire.Equal(t, tt.want, FrontendRetry(tt.err))\n\t\t})\n\t}\n}\n\nfunc TestIsContextTimeoutError(t *testing.T) {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\tdefer cancel()\n\ttime.Sleep(50 * time.Millisecond)\n\trequire.True(t, IsContextTimeoutError(ctx.Err()))\n\trequire.True(t, IsContextTimeoutError(&types.InternalServiceError{Message: ctx.Err().Error()}))\n\n\tyarpcErr := yarpcerrors.DeadlineExceededErrorf(\"yarpc deadline exceeded\")\n\trequire.True(t, IsContextTimeoutError(yarpcErr))\n\n\trequire.False(t, IsContextTimeoutError(errors.New(\"some random error\")))\n\n\tctx, cancel = context.WithCancel(context.Background())\n\tcancel()\n\trequire.False(t, IsContextTimeoutError(ctx.Err()))\n}\n\nfunc TestCreateHistoryStartWorkflowRequest_ExpirationTimeWithCron(t *testing.T) {\n\tdomainID := uuid.New()\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    60,\n\t\t\tExpirationIntervalInSeconds: 60,\n\t\t},\n\t\tCronSchedule: \"@every 300s\",\n\t}\n\tnow := time.Now()\n\tstartRequest, err := CreateHistoryStartWorkflowRequest(domainID, request, now, nil)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, startRequest)\n\n\texpirationTime := startRequest.GetExpirationTimestamp()\n\trequire.NotNil(t, expirationTime)\n\trequire.True(t, time.Unix(0, expirationTime).Sub(now) > 60*time.Second)\n}\n\n// Test to ensure we get the right value for FirstDecisionTaskBackoff during StartWorkflow request,\n// with & without cron, delayStart and jitterStart.\n// - Also see tests in cron_test.go for more exhaustive testing.\nfunc TestFirstDecisionTaskBackoffDuringStartWorkflow(t *testing.T) {\n\tfirstRunTimestampPast, _ := time.Parse(time.RFC3339, \"2017-12-17T08:00:00+00:00\")\n\tfirstRunTimestampFuture, _ := time.Parse(time.RFC3339, \"2018-12-17T08:00:02+00:00\")\n\tfirstRunAtTimestampMap := map[string]int64{\n\t\t\"null\":   0,\n\t\t\"past\":   firstRunTimestampPast.UnixNano(),\n\t\t\"future\": firstRunTimestampFuture.UnixNano(),\n\t}\n\n\tvar tests = []struct {\n\t\tcron                   bool\n\t\tjitterStartSeconds     int32\n\t\tdelayStartSeconds      int32\n\t\tfirstRunAtTimeCategory string\n\t}{\n\t\t// Null first run at cases\n\t\t{true, 0, 0, \"null\"},\n\t\t{true, 15, 0, \"null\"},\n\t\t{true, 0, 600, \"null\"},\n\t\t{true, 15, 600, \"null\"},\n\t\t{false, 0, 0, \"null\"},\n\t\t{false, 15, 0, \"null\"},\n\t\t{false, 0, 600, \"null\"},\n\t\t{false, 15, 600, \"null\"},\n\n\t\t// Past first run at cases\n\t\t{true, 0, 0, \"past\"},\n\t\t{true, 15, 0, \"past\"},\n\t\t{true, 0, 600, \"past\"},\n\t\t{true, 15, 600, \"past\"},\n\t\t{false, 0, 0, \"past\"},\n\t\t{false, 15, 0, \"past\"},\n\t\t{false, 0, 600, \"past\"},\n\t\t{false, 15, 600, \"past\"},\n\n\t\t// Future first run at cases\n\t\t{true, 0, 0, \"future\"},\n\t\t{true, 15, 0, \"future\"},\n\t\t{true, 0, 600, \"future\"},\n\t\t{true, 15, 600, \"future\"},\n\t\t{false, 0, 0, \"future\"},\n\t\t{false, 15, 0, \"future\"},\n\t\t{false, 0, 600, \"future\"},\n\t\t{false, 15, 600, \"future\"},\n\t}\n\n\trand.New(rand.NewSource(int64(time.Now().Nanosecond())))\n\n\tfor idx, tt := range tests {\n\t\tt.Run(strconv.Itoa(idx), func(t *testing.T) {\n\t\t\tdomainID := uuid.New()\n\t\t\trequest := &types.StartWorkflowExecutionRequest{\n\t\t\t\tFirstRunAtTimeStamp: Int64Ptr(firstRunAtTimestampMap[tt.firstRunAtTimeCategory]),\n\t\t\t\tDelayStartSeconds:   Int32Ptr(tt.delayStartSeconds),\n\t\t\t\tJitterStartSeconds:  Int32Ptr(tt.jitterStartSeconds),\n\t\t\t}\n\t\t\tif tt.cron {\n\t\t\t\trequest.CronSchedule = \"* * * * *\"\n\t\t\t}\n\n\t\t\t// Run X loops so that the test isn't flaky, since jitter adds randomness.\n\t\t\tcaseCount := 10\n\t\t\texactCount := 0\n\t\t\tfor i := 0; i < caseCount; i++ {\n\t\t\t\t// Start at the minute boundary so we know what the backoff should be\n\t\t\t\tstartTime, _ := time.Parse(time.RFC3339, \"2018-12-17T08:00:00+00:00\")\n\t\t\t\tstartRequest, err := CreateHistoryStartWorkflowRequest(domainID, request, startTime, nil)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\trequire.NotNil(t, startRequest)\n\n\t\t\t\tbackoffSeconds := startRequest.GetFirstDecisionTaskBackoffSeconds()\n\n\t\t\t\texpectedWithoutJitter := tt.delayStartSeconds\n\t\t\t\tif tt.cron {\n\t\t\t\t\texpectedWithoutJitter += 60\n\t\t\t\t}\n\t\t\t\texpectedMax := expectedWithoutJitter + tt.jitterStartSeconds\n\n\t\t\t\tif backoffSeconds == expectedWithoutJitter {\n\t\t\t\t\texactCount++\n\t\t\t\t}\n\n\t\t\t\tif tt.firstRunAtTimeCategory == \"future\" {\n\t\t\t\t\trequire.Equal(t, int32(2), backoffSeconds, \"test specs = %v\", tt)\n\t\t\t\t} else {\n\t\t\t\t\tif tt.jitterStartSeconds == 0 {\n\t\t\t\t\t\trequire.Equal(t, expectedWithoutJitter, backoffSeconds, \"test specs = %v\", tt)\n\t\t\t\t\t} else {\n\t\t\t\t\t\trequire.True(t, backoffSeconds >= expectedWithoutJitter && backoffSeconds <= expectedMax,\n\t\t\t\t\t\t\t\"test specs = %v, backoff (%v) should be >= %v and <= %v\",\n\t\t\t\t\t\t\ttt, backoffSeconds, expectedWithoutJitter, expectedMax)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// If firstRunTimestamp is either null or past ONLY\n\t\t\t// If jitter is > 0, we want to detect whether jitter is being applied - BUT we don't want the test\n\t\t\t// to be flaky if the code randomly chooses a jitter of 0, so we try to have enough data points by\n\t\t\t// checking the next X cron times AND by choosing a jitter thats not super low.\n\t\t\tif tt.firstRunAtTimeCategory != \"future\" {\n\t\t\t\tif tt.jitterStartSeconds > 0 && exactCount == caseCount {\n\t\t\t\t\t// Test to make sure a jitter test case sometimes doesn't get exact values\n\t\t\t\t\tt.Fatalf(\"FAILED to jitter properly? Test specs = %v\\n\", tt)\n\t\t\t\t} else if tt.jitterStartSeconds == 0 && exactCount != caseCount {\n\t\t\t\t\t// Test to make sure a non-jitter test case always gets exact values\n\t\t\t\t\tt.Fatalf(\"Jittered when we weren't supposed to? Test specs = %v\\n\", tt)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateHistoryStartWorkflowRequest(t *testing.T) {\n\tvar tests = []struct {\n\t\tdelayStartSeconds  int\n\t\tcronSeconds        int\n\t\tjitterStartSeconds int\n\t}{\n\t\t{0, 0, 0},\n\t\t{100, 0, 0},\n\t\t{100, 300, 0},\n\t\t{0, 0, 2000},\n\t\t{100, 0, 2000},\n\t\t{0, 300, 2000},\n\t\t{100, 300, 2000},\n\t\t{0, 300, 2000},\n\t\t{0, 300, 2000},\n\t}\n\n\tfor idx, tt := range tests {\n\t\tt.Run(strconv.Itoa(idx), func(t *testing.T) {\n\t\t\ttestExpirationTime(t, tt.delayStartSeconds, tt.cronSeconds, tt.jitterStartSeconds)\n\t\t})\n\t}\n}\n\nfunc testExpirationTime(t *testing.T, delayStartSeconds int, cronSeconds int, jitterSeconds int) {\n\tdomainID := uuid.New()\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    60,\n\t\t\tExpirationIntervalInSeconds: 60,\n\t\t},\n\t\tDelayStartSeconds:  Int32Ptr(int32(delayStartSeconds)),\n\t\tJitterStartSeconds: Int32Ptr(int32(jitterSeconds)),\n\t}\n\tif cronSeconds > 0 {\n\t\trequest.CronSchedule = fmt.Sprintf(\"@every %ds\", cronSeconds)\n\t}\n\tminDelay := delayStartSeconds + cronSeconds\n\tmaxDelay := delayStartSeconds + 2*cronSeconds + jitterSeconds\n\tnow := time.Now()\n\tstartRequest, err := CreateHistoryStartWorkflowRequest(domainID, request, now, nil)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, startRequest)\n\n\texpirationTime := startRequest.GetExpirationTimestamp()\n\trequire.NotNil(t, expirationTime)\n\n\t// Since we assign the expiration time after we create the workflow request,\n\t// There's a chance that the test thread might sleep or get deprioritized and\n\t// expirationTime - now may not be equal to DelayStartSeconds. Adding 2 seconds\n\t// buffer to avoid this test being flaky\n\trequire.True(\n\t\tt,\n\t\ttime.Unix(0, expirationTime).Sub(now) >= (time.Duration(minDelay)+58)*time.Second,\n\t\t\"Integration test took too short: %f seconds vs %f seconds\",\n\t\ttime.Unix(0, expirationTime).Sub(now).Round(time.Millisecond).Seconds(),\n\t\t((time.Duration(minDelay) + 58) * time.Second).Round(time.Millisecond).Seconds(),\n\t)\n\trequire.True(\n\t\tt,\n\t\ttime.Unix(0, expirationTime).Sub(now) < (time.Duration(maxDelay)+68)*time.Second,\n\t\t\"Integration test took too long: %f seconds vs %f seconds\",\n\t\ttime.Unix(0, expirationTime).Sub(now).Round(time.Millisecond).Seconds(),\n\t\t((time.Duration(minDelay) + 68) * time.Second).Round(time.Millisecond).Seconds(),\n\t)\n}\n\nfunc TestCreateHistoryStartWorkflowRequest_ExpirationTimeWithoutCron(t *testing.T) {\n\tdomainID := uuid.New()\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    60,\n\t\t\tExpirationIntervalInSeconds: 60,\n\t\t},\n\t}\n\tnow := time.Now()\n\tstartRequest, err := CreateHistoryStartWorkflowRequest(domainID, request, now, nil)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, startRequest)\n\n\texpirationTime := startRequest.GetExpirationTimestamp()\n\trequire.NotNil(t, expirationTime)\n\tdelta := time.Unix(0, expirationTime).Sub(now)\n\trequire.True(t, delta > 58*time.Second)\n\trequire.True(t, delta < 62*time.Second)\n}\n\nfunc TestConvertIndexedValueTypeToInternalType(t *testing.T) {\n\tvalues := []types.IndexedValueType{types.IndexedValueTypeString, types.IndexedValueTypeKeyword, types.IndexedValueTypeInt, types.IndexedValueTypeDouble, types.IndexedValueTypeBool, types.IndexedValueTypeDatetime}\n\tfor _, expected := range values {\n\t\trequire.Equal(t, expected, ConvertIndexedValueTypeToInternalType(int(expected), nil))\n\t\trequire.Equal(t, expected, ConvertIndexedValueTypeToInternalType(float64(expected), nil))\n\n\t\tbuffer, err := expected.MarshalText()\n\t\trequire.NoError(t, err)\n\t\trequire.Equal(t, expected, ConvertIndexedValueTypeToInternalType(buffer, nil))\n\t\trequire.Equal(t, expected, ConvertIndexedValueTypeToInternalType(string(buffer), nil))\n\t}\n}\n\nfunc TestValidateDomainUUID(t *testing.T) {\n\ttestCases := []struct {\n\t\tmsg        string\n\t\tdomainUUID string\n\t\tvalid      bool\n\t}{\n\t\t{\n\t\t\tmsg:        \"empty\",\n\t\t\tdomainUUID: \"\",\n\t\t\tvalid:      false,\n\t\t},\n\t\t{\n\t\t\tmsg:        \"invalid\",\n\t\t\tdomainUUID: \"some random uuid\",\n\t\t\tvalid:      false,\n\t\t},\n\t\t{\n\t\t\tmsg:        \"valid\",\n\t\t\tdomainUUID: uuid.New(),\n\t\t\tvalid:      true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.msg, func(t *testing.T) {\n\t\t\terr := ValidateDomainUUID(tc.domainUUID)\n\t\t\tif tc.valid {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t} else {\n\t\t\t\trequire.Error(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConvertErrToGetTaskFailedCause(t *testing.T) {\n\ttestCases := []struct {\n\t\terr                 error\n\t\texpectedFailedCause types.GetTaskFailedCause\n\t}{\n\t\t{\n\t\t\terr:                 errors.New(\"some random error\"),\n\t\t\texpectedFailedCause: types.GetTaskFailedCauseUncategorized,\n\t\t},\n\t\t{\n\t\t\terr:                 context.DeadlineExceeded,\n\t\t\texpectedFailedCause: types.GetTaskFailedCauseTimeout,\n\t\t},\n\t\t{\n\t\t\terr:                 &types.ServiceBusyError{},\n\t\t\texpectedFailedCause: types.GetTaskFailedCauseServiceBusy,\n\t\t},\n\t\t{\n\t\t\terr:                 &types.ShardOwnershipLostError{},\n\t\t\texpectedFailedCause: types.GetTaskFailedCauseShardOwnershipLost,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\trequire.Equal(t, tc.expectedFailedCause, ConvertErrToGetTaskFailedCause(tc.err))\n\t}\n}\n\nfunc TestToServiceTransientError(t *testing.T) {\n\tt.Run(\"it converts nil\", func(t *testing.T) {\n\t\tassert.NoError(t, ToServiceTransientError(nil))\n\t})\n\n\tt.Run(\"it keeps transient errors\", func(t *testing.T) {\n\t\terr := &types.InternalServiceError{}\n\t\tassert.Equal(t, err, ToServiceTransientError(err))\n\t\tassert.True(t, IsServiceTransientError(ToServiceTransientError(err)))\n\t})\n\n\tt.Run(\"it converts errors to transient errors\", func(t *testing.T) {\n\t\terr := fmt.Errorf(\"error\")\n\t\tassert.True(t, IsServiceTransientError(ToServiceTransientError(err)))\n\t})\n}\n\nfunc TestIntersectionStringSlice(t *testing.T) {\n\tt.Run(\"it returns all items\", func(t *testing.T) {\n\t\ta := []string{\"a\", \"b\", \"c\"}\n\t\tb := []string{\"a\", \"b\", \"c\"}\n\t\tc := IntersectionStringSlice(a, b)\n\t\tassert.ElementsMatch(t, []string{\"a\", \"b\", \"c\"}, c)\n\t})\n\n\tt.Run(\"it returns no item\", func(t *testing.T) {\n\t\ta := []string{\"a\", \"b\", \"c\"}\n\t\tb := []string{\"d\", \"e\", \"f\"}\n\t\tc := IntersectionStringSlice(a, b)\n\t\tassert.ElementsMatch(t, []string{}, c)\n\t})\n\n\tt.Run(\"it returns intersection\", func(t *testing.T) {\n\t\ta := []string{\"a\", \"b\", \"c\"}\n\t\tb := []string{\"c\", \"b\", \"f\"}\n\t\tc := IntersectionStringSlice(a, b)\n\t\tassert.ElementsMatch(t, []string{\"c\", \"b\"}, c)\n\t})\n}\n\nfunc TestAwaitWaitGroup(t *testing.T) {\n\tt.Run(\"wait group done before timeout\", func(t *testing.T) {\n\t\tvar wg sync.WaitGroup\n\n\t\twg.Add(1)\n\t\twg.Done()\n\n\t\tgot := AwaitWaitGroup(&wg, time.Second)\n\t\trequire.True(t, got)\n\t})\n\n\tt.Run(\"wait group done after timeout\", func(t *testing.T) {\n\t\tvar (\n\t\t\twg    sync.WaitGroup\n\t\t\tdoneC = make(chan struct{})\n\t\t)\n\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\t<-doneC\n\t\t\twg.Done()\n\t\t}()\n\n\t\tgot := AwaitWaitGroup(&wg, time.Microsecond)\n\t\trequire.False(t, got)\n\n\t\tdoneC <- struct{}{}\n\t\tclose(doneC)\n\t})\n}\n\nfunc TestIsValidIDLength(t *testing.T) {\n\tvar (\n\t\t// test setup\n\t\tscope = metrics.NoopScope\n\n\t\t// arguments\n\t\tmetricCounter      = metrics.MetricIdx(0)\n\t\tidTypeViolationTag = tag.ClusterName(\"idTypeViolationTag\")\n\t\tdomainName         = \"domain_name\"\n\t\tid                 = \"12345\"\n\t)\n\n\tmockWarnCall := func(logger *log.MockLogger) {\n\t\tlogger.EXPECT().Warn(\n\t\t\t\"ID length exceeds limit.\",\n\t\t\t[]tag.Tag{\n\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\ttag.Name(id),\n\t\t\t\tidTypeViolationTag,\n\t\t\t},\n\t\t).Times(1)\n\t}\n\n\tt.Run(\"valid id length, no warnings\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tgot := IsValidIDLength(id, scope, 7, 10, metricCounter, domainName, logger, idTypeViolationTag)\n\t\trequire.True(t, got, \"expected true, because id length is 5 and it's less than error limit 10\")\n\t})\n\n\tt.Run(\"valid id length, with warnings\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tmockWarnCall(logger)\n\n\t\tgot := IsValidIDLength(id, scope, 4, 10, metricCounter, domainName, logger, idTypeViolationTag)\n\t\trequire.True(t, got, \"expected true, because id length is 5 and it's less than error limit 10\")\n\t})\n\n\tt.Run(\"non valid id length\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tmockWarnCall(logger)\n\n\t\tgot := IsValidIDLength(id, scope, 1, 4, metricCounter, domainName, logger, idTypeViolationTag)\n\t\trequire.False(t, got, \"expected false, because id length is 5 and it's more than error limit 4\")\n\t})\n}\n\nfunc TestIsEntityNotExistsError(t *testing.T) {\n\tt.Run(\"is entity not exists error\", func(t *testing.T) {\n\t\terr := &types.EntityNotExistsError{}\n\t\trequire.True(t, IsEntityNotExistsError(err), \"expected true, because err is entity not exists error\")\n\t})\n\n\tt.Run(\"is not entity not exists error\", func(t *testing.T) {\n\t\terr := fmt.Errorf(\"generic error\")\n\t\trequire.False(t, IsEntityNotExistsError(err), \"expected false, because err is a generic error\")\n\t})\n}\n\nfunc TestCreateXXXRetryPolicyWithSetExpirationInterval(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcreateFn func() backoff.RetryPolicy\n\n\t\twantInitialInterval       time.Duration\n\t\twantMaximumInterval       time.Duration\n\t\twantSetExpirationInterval time.Duration\n\t}{\n\t\t\"CreatePersistenceRetryPolicy\": {\n\t\t\tcreateFn:                  CreatePersistenceRetryPolicy,\n\t\t\twantInitialInterval:       retryPersistenceOperationInitialInterval,\n\t\t\twantMaximumInterval:       retryPersistenceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: retryPersistenceOperationExpirationInterval,\n\t\t},\n\t\t\"CreateHistoryServiceRetryPolicy\": {\n\t\t\tcreateFn:                  CreateHistoryServiceRetryPolicy,\n\t\t\twantInitialInterval:       historyServiceOperationInitialInterval,\n\t\t\twantMaximumInterval:       historyServiceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: historyServiceOperationExpirationInterval,\n\t\t},\n\t\t\"CreateMatchingServiceRetryPolicy\": {\n\t\t\tcreateFn:                  CreateMatchingServiceRetryPolicy,\n\t\t\twantInitialInterval:       matchingServiceOperationInitialInterval,\n\t\t\twantMaximumInterval:       matchingServiceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: matchingServiceOperationExpirationInterval,\n\t\t},\n\t\t\"CreateFrontendServiceRetryPolicy\": {\n\t\t\tcreateFn:                  CreateFrontendServiceRetryPolicy,\n\t\t\twantInitialInterval:       frontendServiceOperationInitialInterval,\n\t\t\twantMaximumInterval:       frontendServiceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: frontendServiceOperationExpirationInterval,\n\t\t},\n\t\t\"CreateAdminServiceRetryPolicy\": {\n\t\t\tcreateFn:                  CreateAdminServiceRetryPolicy,\n\t\t\twantInitialInterval:       adminServiceOperationInitialInterval,\n\t\t\twantMaximumInterval:       adminServiceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: adminServiceOperationExpirationInterval,\n\t\t},\n\t\t\"CreateReplicationServiceBusyRetryPolicy\": {\n\t\t\tcreateFn:                  CreateReplicationServiceBusyRetryPolicy,\n\t\t\twantInitialInterval:       replicationServiceBusyInitialInterval,\n\t\t\twantMaximumInterval:       replicationServiceBusyMaxInterval,\n\t\t\twantSetExpirationInterval: replicationServiceBusyExpirationInterval,\n\t\t},\n\t\t\"CreateShardDistributorServiceRetryPolicy\": {\n\t\t\tcreateFn:                  CreateShardDistributorServiceRetryPolicy,\n\t\t\twantInitialInterval:       shardDistributorServiceOperationInitialInterval,\n\t\t\twantMaximumInterval:       shardDistributorServiceOperationMaxInterval,\n\t\t\twantSetExpirationInterval: shardDistributorServiceOperationExpirationInterval,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twant := backoff.NewExponentialRetryPolicy(c.wantInitialInterval)\n\t\t\twant.SetMaximumInterval(c.wantMaximumInterval)\n\t\t\twant.SetExpirationInterval(c.wantSetExpirationInterval)\n\t\t\tgot := c.createFn()\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestCreateXXXRetryPolicyWithMaximumAttempts(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tcreateFn func() backoff.RetryPolicy\n\n\t\twantInitialInterval time.Duration\n\t\twantMaximumInterval time.Duration\n\t\twantMaximumAttempts int\n\t}{\n\t\t\"CreateDlqPublishRetryPolicy\": {\n\t\t\tcreateFn:            CreateDlqPublishRetryPolicy,\n\t\t\twantInitialInterval: retryKafkaOperationInitialInterval,\n\t\t\twantMaximumInterval: retryKafkaOperationMaxInterval,\n\t\t\twantMaximumAttempts: retryKafkaOperationMaxAttempts,\n\t\t},\n\t\t\"CreateTaskProcessingRetryPolicy\": {\n\t\t\tcreateFn:            CreateTaskProcessingRetryPolicy,\n\t\t\twantInitialInterval: retryTaskProcessingInitialInterval,\n\t\t\twantMaximumInterval: retryTaskProcessingMaxInterval,\n\t\t\twantMaximumAttempts: retryTaskProcessingMaxAttempts,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twant := backoff.NewExponentialRetryPolicy(c.wantInitialInterval)\n\t\t\twant.SetMaximumInterval(c.wantMaximumInterval)\n\t\t\twant.SetMaximumAttempts(c.wantMaximumAttempts)\n\t\t\tgot := c.createFn()\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestValidateRetryPolicy_Success(t *testing.T) {\n\tfor name, policy := range map[string]*types.RetryPolicy{\n\t\t\"nil policy\": nil,\n\t\t\"MaximumAttempts is no zero\": {\n\t\t\tInitialIntervalInSeconds:    2,\n\t\t\tBackoffCoefficient:          1,\n\t\t\tMaximumIntervalInSeconds:    0,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 0,\n\t\t},\n\t\t\"ExpirationIntervalInSeconds is no zero\": {\n\t\t\tInitialIntervalInSeconds:    2,\n\t\t\tBackoffCoefficient:          1,\n\t\t\tMaximumIntervalInSeconds:    0,\n\t\t\tMaximumAttempts:             0,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\t\"MaximumIntervalInSeconds is greater than InitialIntervalInSeconds\": {\n\t\t\tInitialIntervalInSeconds:    2,\n\t\t\tBackoffCoefficient:          1,\n\t\t\tMaximumIntervalInSeconds:    0,\n\t\t\tMaximumAttempts:             0,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\t\"MaximumIntervalInSeconds equals InitialIntervalInSeconds\": {\n\t\t\tInitialIntervalInSeconds:    2,\n\t\t\tBackoffCoefficient:          1,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             0,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\trequire.NoError(t, ValidateRetryPolicy(policy))\n\t\t})\n\t}\n}\n\nfunc TestValidateRetryPolicy_Error(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tpolicy  *types.RetryPolicy\n\t\twantErr *types.BadRequestError\n\t}{\n\t\t\"InitialIntervalInSeconds equals 0\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 0,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"InitialIntervalInSeconds must be greater than 0 on retry policy.\"},\n\t\t},\n\t\t\"InitialIntervalInSeconds less than 0\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: -1,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"InitialIntervalInSeconds must be greater than 0 on retry policy.\"},\n\t\t},\n\t\t\"BackoffCoefficient equals 0\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\tBackoffCoefficient:       0,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"BackoffCoefficient cannot be less than 1 on retry policy.\"},\n\t\t},\n\t\t\"MaximumIntervalInSeconds equals -1\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\tBackoffCoefficient:       1,\n\t\t\t\tMaximumIntervalInSeconds: -1,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"MaximumIntervalInSeconds cannot be less than 0 on retry policy.\"},\n\t\t},\n\t\t\"MaximumIntervalInSeconds equals 1 and less than InitialIntervalInSeconds\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 2,\n\t\t\t\tBackoffCoefficient:       1,\n\t\t\t\tMaximumIntervalInSeconds: 1,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"MaximumIntervalInSeconds cannot be less than InitialIntervalInSeconds on retry policy.\"},\n\t\t},\n\t\t\"MaximumAttempts equals -1\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 2,\n\t\t\t\tBackoffCoefficient:       1,\n\t\t\t\tMaximumIntervalInSeconds: 0,\n\t\t\t\tMaximumAttempts:          -1,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"MaximumAttempts cannot be less than 0 on retry policy.\"},\n\t\t},\n\t\t\"ExpirationIntervalInSeconds equals -1\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds:    2,\n\t\t\t\tBackoffCoefficient:          1,\n\t\t\t\tMaximumIntervalInSeconds:    0,\n\t\t\t\tMaximumAttempts:             0,\n\t\t\t\tExpirationIntervalInSeconds: -1,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"ExpirationIntervalInSeconds cannot be less than 0 on retry policy.\"},\n\t\t},\n\t\t\"MaximumAttempts and ExpirationIntervalInSeconds equal 0\": {\n\t\t\tpolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds:    2,\n\t\t\t\tBackoffCoefficient:          1,\n\t\t\t\tMaximumIntervalInSeconds:    0,\n\t\t\t\tMaximumAttempts:             0,\n\t\t\t\tExpirationIntervalInSeconds: 0,\n\t\t\t},\n\t\t\twantErr: &types.BadRequestError{Message: \"MaximumAttempts and ExpirationIntervalInSeconds are both 0. At least one of them must be specified.\"},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := ValidateRetryPolicy(c.policy)\n\t\t\trequire.Error(t, got)\n\t\t\trequire.ErrorContains(t, got, c.wantErr.Message)\n\t\t})\n\t}\n}\n\nfunc TestConvertGetTaskFailedCauseToErr(t *testing.T) {\n\tfor cause, wantErr := range map[types.GetTaskFailedCause]error{\n\t\ttypes.GetTaskFailedCauseServiceBusy:        &types.ServiceBusyError{},\n\t\ttypes.GetTaskFailedCauseTimeout:            context.DeadlineExceeded,\n\t\ttypes.GetTaskFailedCauseShardOwnershipLost: &types.ShardOwnershipLostError{},\n\t\ttypes.GetTaskFailedCauseUncategorized:      &types.InternalServiceError{Message: \"uncategorized error\"},\n\t} {\n\t\tt.Run(cause.String(), func(t *testing.T) {\n\t\t\tgotErr := ConvertGetTaskFailedCauseToErr(cause)\n\t\t\trequire.Equal(t, wantErr, gotErr)\n\t\t})\n\t}\n}\n\nfunc TestWorkflowIDToHistoryShard(t *testing.T) {\n\tfor _, c := range []struct {\n\t\tworkflowID     string\n\t\tnumberOfShards int\n\n\t\twant int\n\t}{\n\t\t{\n\t\t\tworkflowID:     \"\",\n\t\t\tnumberOfShards: 1000,\n\t\t\twant:           242,\n\t\t},\n\t\t{\n\t\t\tworkflowID:     \"workflowId\",\n\t\t\tnumberOfShards: 1000,\n\t\t\twant:           580,\n\t\t},\n\t} {\n\t\tt.Run(fmt.Sprintf(\"%s-%v\", c.workflowID, c.numberOfShards), func(t *testing.T) {\n\t\t\tgot := WorkflowIDToHistoryShard(c.workflowID, c.numberOfShards)\n\t\t\trequire.Equal(t, c.want, got)\n\t\t})\n\t}\n}\n\nfunc TestDomainIDToHistoryShard(t *testing.T) {\n\tfor _, c := range []struct {\n\t\tdomainID       string\n\t\tnumberOfShards int\n\n\t\twant int\n\t}{\n\t\t{\n\t\t\tdomainID:       \"\",\n\t\t\tnumberOfShards: 1000,\n\t\t\twant:           242,\n\t\t},\n\t\t{\n\t\t\tdomainID:       \"domainId\",\n\t\t\tnumberOfShards: 1000,\n\t\t\twant:           600,\n\t\t},\n\t} {\n\t\tt.Run(fmt.Sprintf(\"%s-%v\", c.domainID, c.numberOfShards), func(t *testing.T) {\n\t\t\tgot := DomainIDToHistoryShard(c.domainID, c.numberOfShards)\n\t\t\trequire.Equal(t, c.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGenerateRandomString(t *testing.T) {\n\tfor input, wantSize := range map[int]int{\n\t\t-1: 0,\n\t\t0:  0,\n\t\t10: 10,\n\t} {\n\t\tt.Run(fmt.Sprintf(\"%d\", input), func(t *testing.T) {\n\t\t\tgot := GenerateRandomString(input)\n\t\t\trequire.Len(t, got, wantSize)\n\t\t})\n\t}\n}\n\nfunc TestIsValidContext(t *testing.T) {\n\tt.Run(\"background context\", func(t *testing.T) {\n\t\trequire.NoError(t, IsValidContext(context.Background()))\n\t})\n\tt.Run(\"canceled context\", func(t *testing.T) {\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tcancel()\n\t\tgot := IsValidContext(ctx)\n\t\trequire.Error(t, got)\n\t\trequire.ErrorIs(t, got, context.Canceled)\n\t})\n\tt.Run(\"deadline exceeded context\", func(t *testing.T) {\n\t\tctx, _ := context.WithTimeout(context.Background(), -time.Second)\n\t\tgot := IsValidContext(ctx)\n\t\trequire.Error(t, got)\n\t\trequire.ErrorIs(t, got, context.DeadlineExceeded)\n\t})\n\tt.Run(\"context with deadline exceeded contextExpireThreshold\", func(t *testing.T) {\n\t\tctx, _ := context.WithTimeout(context.Background(), contextExpireThreshold/2)\n\t\tgot := IsValidContext(ctx)\n\t\trequire.Error(t, got)\n\t\trequire.ErrorIs(t, got, context.DeadlineExceeded, \"context.DeadlineExceeded should be returned, because context timeout is not later than now + contextExpireThreshold\")\n\t})\n\tt.Run(\"valid context\", func(t *testing.T) {\n\t\tctx, _ := context.WithTimeout(context.Background(), contextExpireThreshold*2)\n\t\trequire.NoError(t, IsValidContext(ctx), \"nil should be returned, because context timeout is later than now + contextExpireThreshold\")\n\t})\n}\n\nfunc TestCreateChildContext(t *testing.T) {\n\tt.Run(\"nil parent\", func(t *testing.T) {\n\t\tgotCtx, gotFunc := CreateChildContext(nil, 0)\n\t\trequire.Nil(t, gotCtx)\n\t\trequire.Equal(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t})\n\tt.Run(\"canceled parent\", func(t *testing.T) {\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tcancel()\n\t\tgotCtx, gotFunc := CreateChildContext(ctx, 0)\n\t\trequire.Equal(t, ctx, gotCtx)\n\t\trequire.Equal(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t})\n\tt.Run(\"non-canceled parent without deadline\", func(t *testing.T) {\n\t\tctx, _ := context.WithCancel(context.Background())\n\t\tgotCtx, gotFunc := CreateChildContext(ctx, 0)\n\t\trequire.Equal(t, ctx, gotCtx)\n\t\trequire.Equal(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t})\n\tt.Run(\"context with deadline exceeded\", func(t *testing.T) {\n\t\tctx, _ := context.WithTimeout(context.Background(), -time.Second)\n\t\tgotCtx, gotFunc := CreateChildContext(ctx, 0)\n\t\trequire.Equal(t, ctx, gotCtx)\n\t\trequire.Equal(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t})\n\n\tt.Run(\"tailroom is less or equal to 0\", func(t *testing.T) {\n\t\ttestCase := func(t *testing.T, tailroom float64) {\n\t\t\tdeadline := time.Now().Add(time.Hour)\n\t\t\tctx, _ := context.WithDeadline(context.Background(), deadline)\n\t\t\tgotCtx, gotFunc := CreateChildContext(ctx, tailroom)\n\n\t\t\tgotDeadline, ok := gotCtx.Deadline()\n\t\t\trequire.True(t, ok)\n\t\t\trequire.Equal(t, deadline, gotDeadline, \"deadline should be equal to parent deadline\")\n\n\t\t\trequire.NotEqual(t, ctx, gotCtx)\n\t\t\trequire.NotEqual(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t\t}\n\n\t\tt.Run(\"0\", func(t *testing.T) {\n\t\t\ttestCase(t, 0)\n\t\t})\n\t\tt.Run(\"-1\", func(t *testing.T) {\n\t\t\ttestCase(t, -1)\n\t\t})\n\n\t})\n\n\tt.Run(\"tailroom is greater or equal to 1\", func(t *testing.T) {\n\t\ttestCase := func(t *testing.T, tailroom float64) {\n\t\t\tdeadline := time.Now().Add(time.Hour)\n\t\t\tctx, _ := context.WithDeadline(context.Background(), deadline)\n\n\t\t\t// we can't mock time.Now, but we know that the deadline should be in\n\t\t\t// range between the start and finish of function's execution\n\t\t\tbeforeNow := time.Now()\n\t\t\tgotCtx, gotFunc := CreateChildContext(ctx, tailroom)\n\t\t\tafterNow := time.Now()\n\n\t\t\tgotDeadline, ok := gotCtx.Deadline()\n\t\t\trequire.True(t, ok)\n\t\t\trequire.NotEqual(t, deadline, gotDeadline)\n\t\t\trequire.Less(t, gotDeadline, deadline)\n\n\t\t\t// gotDeadline should be between beforeNow and afterNow (exclusive)\n\t\t\trequire.GreaterOrEqual(t, afterNow, gotDeadline)\n\t\t\trequire.LessOrEqual(t, beforeNow, gotDeadline)\n\n\t\t\trequire.NotEqual(t, ctx, gotCtx)\n\t\t\trequire.NotEqual(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t\t}\n\t\tt.Run(\"1\", func(t *testing.T) {\n\t\t\ttestCase(t, 1)\n\t\t})\n\t\tt.Run(\"2\", func(t *testing.T) {\n\t\t\ttestCase(t, 2)\n\t\t})\n\t})\n\tt.Run(\"tailroom is 0.5\", func(t *testing.T) {\n\t\tnow := time.Now()\n\t\tdeadline := now.Add(time.Hour)\n\n\t\tctx, _ := context.WithDeadline(context.Background(), deadline)\n\t\tgotCtx, gotFunc := CreateChildContext(ctx, 0.5)\n\n\t\tgotDeadline, ok := gotCtx.Deadline()\n\t\trequire.True(t, ok)\n\t\trequire.NotEqual(t, deadline, gotDeadline)\n\t\trequire.Less(t, gotDeadline, deadline)\n\n\t\t// we can't mock time.Now, so we assume that the deadline should be\n\t\t// in range 29:59 and 30:01 minutes after start\n\t\tminDeadline := now.Add(30*time.Minute - time.Second)\n\t\tmaxDeadline := now.Add(30*time.Minute + time.Second)\n\n\t\t// gotDeadline should be between minDeadline and maxDeadline (exclusive)\n\t\trequire.GreaterOrEqual(t, maxDeadline, gotDeadline)\n\t\trequire.LessOrEqual(t, minDeadline, gotDeadline)\n\n\t\trequire.NotEqual(t, ctx, gotCtx)\n\t\trequire.NotEqual(t, funcName(emptyCancelFunc), funcName(gotFunc))\n\t})\n}\n\n// funcName returns the name of the function\nfunc funcName(fn any) string {\n\treturn runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name()\n}\n\nfunc TestDeserializeSearchAttributeValue_Success(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tvalue     string\n\t\tvalueType types.IndexedValueType\n\n\t\twantValue any\n\t}{\n\t\t\"string\": {\n\t\t\tvalue:     `\"string\"`,\n\t\t\tvalueType: types.IndexedValueTypeString,\n\t\t\twantValue: \"string\",\n\t\t},\n\t\t\"[]string\": {\n\t\t\tvalue:     `[\"1\", \"2\", \"3\"]`,\n\t\t\tvalueType: types.IndexedValueTypeString,\n\t\t\twantValue: []string{\"1\", \"2\", \"3\"},\n\t\t},\n\t\t\"keyword\": {\n\t\t\tvalue:     `\"keyword\"`,\n\t\t\tvalueType: types.IndexedValueTypeKeyword,\n\t\t\twantValue: \"keyword\",\n\t\t},\n\t\t\"[]keyword\": {\n\t\t\tvalue:     `[\"1\", \"2\", \"3\"]`,\n\t\t\tvalueType: types.IndexedValueTypeKeyword,\n\t\t\twantValue: []string{\"1\", \"2\", \"3\"},\n\t\t},\n\t\t\"int\": {\n\t\t\tvalue:     `1`,\n\t\t\tvalueType: types.IndexedValueTypeInt,\n\t\t\twantValue: int64(1),\n\t\t},\n\t\t\"[]int\": {\n\t\t\tvalue:     `[1, 2, 3]`,\n\t\t\tvalueType: types.IndexedValueTypeInt,\n\t\t\twantValue: []int64{1, 2, 3},\n\t\t},\n\t\t\"double\": {\n\t\t\tvalue:     `1.1`,\n\t\t\tvalueType: types.IndexedValueTypeDouble,\n\t\t\twantValue: 1.1,\n\t\t},\n\t\t\"[]double\": {\n\t\t\tvalue:     `[1.1, 2.2, 3.3]`,\n\t\t\tvalueType: types.IndexedValueTypeDouble,\n\t\t\twantValue: []float64{1.1, 2.2, 3.3},\n\t\t},\n\t\t\"bool\": {\n\t\t\tvalue:     `true`,\n\t\t\tvalueType: types.IndexedValueTypeBool,\n\t\t\twantValue: true,\n\t\t},\n\t\t\"[]bool\": {\n\t\t\tvalue:     `[true, false, true]`,\n\t\t\tvalueType: types.IndexedValueTypeBool,\n\t\t\twantValue: []bool{true, false, true},\n\t\t},\n\t\t\"datetime\": {\n\t\t\tvalue:     `\"2020-01-01T00:00:00Z\"`,\n\t\t\tvalueType: types.IndexedValueTypeDatetime,\n\t\t\twantValue: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t},\n\t\t\"[]datetime\": {\n\t\t\tvalue:     `[\"2020-01-01T00:00:00Z\", \"2020-01-02T00:00:00Z\", \"2020-01-03T00:00:00Z\"]`,\n\t\t\tvalueType: types.IndexedValueTypeDatetime,\n\t\t\twantValue: []time.Time{\n\t\t\t\ttime.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\ttime.Date(2020, 1, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t\ttime.Date(2020, 1, 3, 0, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgotValue, err := DeserializeSearchAttributeValue([]byte(c.value), c.valueType)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, c.wantValue, gotValue)\n\t\t})\n\t}\n}\n\nfunc TestDeserializeSearchAttributeValue_Error(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tvalue     string\n\t\tvalueType types.IndexedValueType\n\n\t\twantErrorMsg string\n\t}{\n\t\t\"invalid string\": {\n\t\t\tvalue:        `\"string`,\n\t\t\tvalueType:    types.IndexedValueTypeString,\n\t\t\twantErrorMsg: \"unexpected end of JSON input\",\n\t\t},\n\t\t\"invalid keyword\": {\n\t\t\tvalue:        `\"keyword`,\n\t\t\tvalueType:    types.IndexedValueTypeKeyword,\n\t\t\twantErrorMsg: \"unexpected end of JSON input\",\n\t\t},\n\t\t\"invalid int\": {\n\t\t\tvalue:        `1.1`,\n\t\t\tvalueType:    types.IndexedValueTypeInt,\n\t\t\twantErrorMsg: \"json: cannot unmarshal number into Go value of type []int64\",\n\t\t},\n\t\t\"invalid double\": {\n\t\t\tvalue:        `1as`,\n\t\t\tvalueType:    types.IndexedValueTypeDouble,\n\t\t\twantErrorMsg: \"invalid character 'a' after top-level value\",\n\t\t},\n\t\t\"invalid bool\": {\n\t\t\tvalue:        `1`,\n\t\t\tvalueType:    types.IndexedValueTypeBool,\n\t\t\twantErrorMsg: \"json: cannot unmarshal number into Go value of type []bool\",\n\t\t},\n\t\t\"invalid datetime\": {\n\t\t\tvalue:        `1`,\n\t\t\tvalueType:    types.IndexedValueTypeDatetime,\n\t\t\twantErrorMsg: \"json: cannot unmarshal number into Go value of type []time.Time\",\n\t\t},\n\t\t\"invalid value type\": {\n\t\t\tvalue:        `1`,\n\t\t\tvalueType:    types.IndexedValueType(100),\n\t\t\twantErrorMsg: fmt.Sprintf(\"error: unknown index value type [%v]\", types.IndexedValueType(100)),\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgotValue, err := DeserializeSearchAttributeValue([]byte(c.value), c.valueType)\n\t\t\trequire.Error(t, err)\n\t\t\trequire.ErrorContains(t, err, c.wantErrorMsg)\n\t\t\trequire.Nil(t, gotValue)\n\t\t})\n\t}\n}\n\n// someMapStringToByteArraySize is the size of someMapStringToByteArray calculated by GetSizeOfMapStringToByteArray\nconst someMapStringToByteArraySize = 66\n\nvar someMapStringToByteArray = map[string][]byte{\n\t\"key\":  []byte(\"value\"),\n\t\"key2\": []byte(\"value2\"),\n}\n\n// someBytesArraySize is len(someBytesArray)\nconst someBytesArraySize = 17\n\nvar someBytesArray = []byte(\"some random bytes\")\n\nfunc TestGetSizeOfMapStringToByteArray(t *testing.T) {\n\trequire.Equal(t, someMapStringToByteArraySize, GetSizeOfMapStringToByteArray(someMapStringToByteArray))\n}\n\nfunc TestGetSizeOfHistoryEvent_NilEvent(t *testing.T) {\n\trequire.Equal(t, uint64(0), GetSizeOfHistoryEvent(nil))\n\trequire.Equal(t, uint64(0), GetSizeOfHistoryEvent(&types.HistoryEvent{}), \"should be zero, because eventType is nil\")\n}\n\nfunc TestGetSizeOfHistoryEvent(t *testing.T) {\n\tfor eventType, c := range map[types.EventType]struct {\n\t\tevent *types.HistoryEvent\n\t\twant  uint64\n\t}{\n\t\ttypes.EventTypeWorkflowExecutionStarted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInput:                   someBytesArray, // 17 bytes\n\t\t\t\t\tContinuedFailureDetails: someBytesArray, // 17 bytes\n\t\t\t\t\tLastCompletionResult:    someBytesArray, // 17 bytes\n\t\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tHeader: &types.Header{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: 3*someBytesArraySize + 3*someMapStringToByteArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionCompleted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionTimedOut: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeDecisionTaskScheduled:     {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeDecisionTaskStarted:       {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeDecisionTaskCompleted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tExecutionContext: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeDecisionTaskTimedOut: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeDecisionTaskFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskScheduled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tInput: someBytesArray, // 17 bytes\n\t\t\t\t\tHeader: &types.Header{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize + someMapStringToByteArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskStarted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tLastFailureDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskCompleted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\tResult: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskTimedOut: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tDetails:            someBytesArray, // 17 bytes\n\t\t\t\t\tLastFailureDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: 2 * someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeActivityTaskCancelRequested:     {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeRequestCancelActivityTaskFailed: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeActivityTaskCanceled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeTimerStarted:                     {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeTimerFired:                       {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeCancelTimerFailed:                {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeTimerCanceled:                    {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeWorkflowExecutionCancelRequested: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeWorkflowExecutionCanceled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionCanceledEventAttributes: &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionInitiated: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes: &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeExternalWorkflowExecutionCancelRequested: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeMarkerRecorded: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionSignaled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tInput: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionTerminated: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionTerminatedEventAttributes: &types.WorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\t\t\tInput: someBytesArray, // 17 bytes\n\t\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tHeader: &types.Header{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize + 3*someMapStringToByteArraySize,\n\t\t},\n\t\ttypes.EventTypeStartChildWorkflowExecutionInitiated: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\t\tInput:   someBytesArray, // 17 bytes\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tHeader: &types.Header{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: 2*someBytesArraySize + 3*someMapStringToByteArraySize,\n\t\t},\n\t\ttypes.EventTypeStartChildWorkflowExecutionFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tStartChildWorkflowExecutionFailedEventAttributes: &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeChildWorkflowExecutionStarted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tHeader: &types.Header{\n\t\t\t\t\t\tFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someMapStringToByteArraySize,\n\t\t},\n\t\ttypes.EventTypeChildWorkflowExecutionCompleted: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeChildWorkflowExecutionFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeChildWorkflowExecutionCanceled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeChildWorkflowExecutionTimedOut:   {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeChildWorkflowExecutionTerminated: {event: &types.HistoryEvent{}, want: 0},\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionInitiated: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes: &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\t\tInput:   someBytesArray, // 17 bytes\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: 2 * someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionFailed: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tSignalExternalWorkflowExecutionFailedEventAttributes: &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeExternalWorkflowExecutionSignaled: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tExternalWorkflowExecutionSignaledEventAttributes: &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tControl: someBytesArray, // 17 bytes\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someBytesArraySize,\n\t\t},\n\t\ttypes.EventTypeUpsertWorkflowSearchAttributes: {\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tUpsertWorkflowSearchAttributesEventAttributes: &types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: someMapStringToByteArray, // 66 bytes\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: someMapStringToByteArraySize,\n\t\t},\n\t} {\n\t\tt.Run(eventType.String(), func(t *testing.T) {\n\t\t\tif c.event != nil {\n\t\t\t\tc.event.EventType = &eventType\n\t\t\t}\n\n\t\t\tgot := GetSizeOfHistoryEvent(c.event)\n\t\t\trequire.Equal(t, c.want, got)\n\t\t})\n\t}\n}\n\nfunc TestIsAdvancedVisibilityWritingEnabled(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tadvancedVisibilityWritingMode string\n\t\tisAdvancedVisConfigExist      bool\n\t\twant                          bool\n\t}{\n\t\t\"mode is someMode, config exist\": {\n\t\t\tadvancedVisibilityWritingMode: \"someMode\",\n\t\t\tisAdvancedVisConfigExist:      true,\n\t\t\twant:                          true,\n\t\t},\n\t\t\"mode is someMode, config not exist\": {\n\t\t\tadvancedVisibilityWritingMode: \"someMode\",\n\t\t\tisAdvancedVisConfigExist:      false,\n\t\t\twant:                          false,\n\t\t},\n\t\t\"mode is off, config exist\": {\n\t\t\tadvancedVisibilityWritingMode: \"off\",\n\t\t\tisAdvancedVisConfigExist:      true,\n\t\t\twant:                          false,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := IsAdvancedVisibilityWritingEnabled(c.advancedVisibilityWritingMode, c.isAdvancedVisConfigExist)\n\t\t\trequire.Equal(t, c.want, got)\n\t\t})\n\t}\n}\n\nfunc TestValidateLongPollContextTimeout(t *testing.T) {\n\tconst handlerName = \"testHandler\"\n\n\tt.Run(\"context timeout is not set\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tlogger.EXPECT().Error(\n\t\t\t\"Context timeout not set for long poll API.\",\n\t\t\t[]tag.Tag{\n\t\t\t\ttag.WorkflowHandlerName(handlerName),\n\t\t\t\ttag.Error(ErrContextTimeoutNotSet),\n\t\t\t},\n\t\t)\n\n\t\tctx := context.Background()\n\t\tgot := ValidateLongPollContextTimeout(ctx, handlerName, logger)\n\t\trequire.Error(t, got)\n\t\trequire.ErrorIs(t, got, ErrContextTimeoutNotSet)\n\t})\n\n\tt.Run(\"context timeout is set, but less than MinLongPollTimeout\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tlogger.EXPECT().Error(\n\t\t\t\"Context timeout is too short for long poll API.\",\n\t\t\t// we can't mock time between deadline and now, so we just check it as it is\n\t\t\tgomock.Any(),\n\t\t)\n\t\tctx, _ := context.WithTimeout(context.Background(), time.Second)\n\t\tgot := ValidateLongPollContextTimeout(ctx, handlerName, logger)\n\t\trequire.Error(t, got)\n\t\trequire.ErrorIs(t, got, ErrContextTimeoutTooShort, \"should return ErrContextTimeoutTooShort, because context timeout is less than MinLongPollTimeout\")\n\t})\n\n\tt.Run(\"context timeout is set, but less than CriticalLongPollTimeout\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tlogger.EXPECT().Debug(\n\t\t\t\"Context timeout is lower than critical value for long poll API.\",\n\t\t\t// we can't mock time between deadline and now, so we just check it as it is\n\t\t\tgomock.Any(),\n\t\t)\n\t\tctx, _ := context.WithTimeout(context.Background(), 15*time.Second)\n\t\tgot := ValidateLongPollContextTimeout(ctx, handlerName, logger)\n\t\trequire.NoError(t, got)\n\t})\n\n\tt.Run(\"context timeout is set, but greater than CriticalLongPollTimeout\", func(t *testing.T) {\n\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\t\tctx, _ := context.WithTimeout(context.Background(), 30*time.Second)\n\t\tgot := ValidateLongPollContextTimeout(ctx, handlerName, logger)\n\t\trequire.NoError(t, got)\n\t})\n}\n\nfunc TestDurationToDays(t *testing.T) {\n\tfor duration, want := range map[time.Duration]int32{\n\t\t0:              0,\n\t\ttime.Hour:      0,\n\t\t24 * time.Hour: 1,\n\t\t25 * time.Hour: 1,\n\t\t48 * time.Hour: 2,\n\t} {\n\t\tt.Run(duration.String(), func(t *testing.T) {\n\t\t\tgot := DurationToDays(duration)\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestDurationToSeconds(t *testing.T) {\n\tfor duration, want := range map[time.Duration]int64{\n\t\t0:                           0,\n\t\ttime.Second:                 1,\n\t\ttime.Second + time.Second/2: 1,\n\t\t2 * time.Second:             2,\n\t} {\n\t\tt.Run(duration.String(), func(t *testing.T) {\n\t\t\tgot := DurationToSeconds(duration)\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestDaysToDuration(t *testing.T) {\n\tfor days, want := range map[int32]time.Duration{\n\t\t0: 0,\n\t\t1: 24 * time.Hour,\n\t\t2: 48 * time.Hour,\n\t} {\n\t\tt.Run(strconv.Itoa(int(days)), func(t *testing.T) {\n\t\t\tgot := DaysToDuration(days)\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestSecondsToDuration(t *testing.T) {\n\tfor seconds, want := range map[int64]time.Duration{\n\t\t0: 0,\n\t\t1: time.Second,\n\t\t2: 2 * time.Second,\n\t} {\n\t\tt.Run(strconv.Itoa(int(seconds)), func(t *testing.T) {\n\t\t\tgot := SecondsToDuration(seconds)\n\t\t\trequire.Equal(t, want, got)\n\t\t})\n\t}\n}\n\nfunc TestNewPerTaskListScope(t *testing.T) {\n\tassert.NotNil(t, NewPerTaskListScope(\"test-domain\", \"test-tasklist\", types.TaskListKindNormal, metrics.NewNoopMetricsClient(), 0))\n\tassert.NotNil(t, NewPerTaskListScope(\"test-domain\", \"test-tasklist\", types.TaskListKindSticky, metrics.NewNoopMetricsClient(), 0))\n\tassert.NotNil(t, NewPerTaskListScope(\"test-domain\", \"test-tasklist\", types.TaskListKindEphemeral, metrics.NewNoopMetricsClient(), 0))\n}\n\nfunc TestCheckEventBlobSizeLimit(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tblobSize      int\n\t\twarnSize      int\n\t\terrSize       int\n\t\twantErr       error\n\t\tprepareLogger func(*log.MockLogger)\n\t\tassertMetrics func(tally.Snapshot)\n\t}{\n\t\t\"blob size is less than limit\": {\n\t\t\tblobSize: 10,\n\t\t\twarnSize: 20,\n\t\t\terrSize:  30,\n\t\t\twantErr:  nil,\n\t\t},\n\t\t\"blob size is greater than warn limit\": {\n\t\t\tblobSize: 21,\n\t\t\twarnSize: 20,\n\t\t\terrSize:  30,\n\t\t\twantErr:  nil,\n\t\t\tprepareLogger: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().Warn(\"Blob size close to the limit.\", gomock.Any()).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"blob size is greater than error limit\": {\n\t\t\tblobSize: 31,\n\t\t\twarnSize: 20,\n\t\t\terrSize:  30,\n\t\t\twantErr:  ErrBlobSizeExceedsLimit,\n\t\t\tprepareLogger: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().Error(\"Blob size exceeds limit.\", gomock.Any()).Times(1)\n\t\t\t},\n\t\t\tassertMetrics: func(snapshot tally.Snapshot) {\n\t\t\t\tcounters := snapshot.Counters()\n\t\t\t\tassert.Len(t, counters, 1)\n\t\t\t\tvalues := maps.Values(counters)\n\t\t\t\tassert.Equal(t, \"test.blob_size_exceed_limit\", values[0].Name())\n\t\t\t\tassert.Equal(t, int64(1), values[0].Value())\n\t\t\t},\n\t\t},\n\t\t\"error limit is less then warn limit\": {\n\t\t\tblobSize: 21,\n\t\t\twarnSize: 30,\n\t\t\terrSize:  20,\n\t\t\twantErr:  ErrBlobSizeExceedsLimit,\n\t\t\tprepareLogger: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().Warn(\"Error limit is less than warn limit.\", gomock.Any()).Times(1)\n\t\t\t\tlogger.EXPECT().Error(\"Blob size exceeds limit.\", gomock.Any()).Times(1)\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.History, metrics.HistogramMigration{})\n\t\t\tlogger := log.NewMockLogger(gomock.NewController(t))\n\n\t\t\tif c.prepareLogger != nil {\n\t\t\t\tc.prepareLogger(logger)\n\t\t\t}\n\n\t\t\tconst (\n\t\t\t\tdomainID   = \"testDomainID\"\n\t\t\t\tdomainName = \"testDomainName\"\n\t\t\t\tworkflowID = \"testWorkflowID\"\n\t\t\t\trunID      = \"testRunID\"\n\t\t\t)\n\n\t\t\tgot := CheckEventBlobSizeLimit(\n\t\t\t\tc.blobSize,\n\t\t\t\tc.warnSize,\n\t\t\t\tc.errSize,\n\t\t\t\tdomainID,\n\t\t\t\tdomainName,\n\t\t\t\tworkflowID,\n\t\t\t\trunID,\n\t\t\t\tmetricsClient.Scope(1),\n\t\t\t\tlogger,\n\t\t\t\ttag.OperationName(\"testOperation\"),\n\t\t\t)\n\t\t\trequire.Equal(t, c.wantErr, got)\n\t\t\tif c.assertMetrics != nil {\n\t\t\t\tc.assertMetrics(testScope.Snapshot())\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "common/visibility/validate_search_attribute_key.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage visibility\n\nimport (\n\t\"fmt\"\n\t\"regexp\"\n\t\"strings\"\n)\n\nvar (\n\tvalidSearchAttributeKey = regexp.MustCompile(`^[a-zA-Z][a-zA-Z_0-9]*$`)\n\tnonAlphanumericRegex    = regexp.MustCompile(`[^a-zA-Z0-9]+`)\n)\n\n// ValidateSearchAttributeKey checks if the search attribute key has valid format\nfunc ValidateSearchAttributeKey(name string) error {\n\tif !validSearchAttributeKey.MatchString(name) {\n\t\treturn fmt.Errorf(\"has to be contain alphanumeric and start with a letter\")\n\t}\n\treturn nil\n}\n\n// SanitizeSearchAttributeKey try to sanitize the search attribute key\nfunc SanitizeSearchAttributeKey(name string) (string, error) {\n\tsanitized := nonAlphanumericRegex.ReplaceAllString(name, \"_\")\n\t// remove leading and trailing underscores\n\tsanitized = strings.Trim(sanitized, \"_\")\n\t// remove leading numbers\n\tsanitized = strings.TrimLeft(sanitized, \"0123456789\")\n\treturn sanitized, ValidateSearchAttributeKey(sanitized)\n}\n"
  },
  {
    "path": "common/visibility/validate_search_attribute_key_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage visibility\n\nimport (\n\t\"testing\"\n)\n\nfunc TestValidateSearchAttributeKey(t *testing.T) {\n\ttype args struct {\n\t\tname string\n\t}\n\ttests := []struct {\n\t\tname    string\n\t\targs    args\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"pure character\",\n\t\t\targs: args{name: \"CustomStringField\"},\n\t\t},\n\t\t{\n\t\t\tname: \"alphanumeric\",\n\t\t\targs: args{name: \"CustomStringField1\"},\n\t\t},\n\t\t{\n\t\t\tname:    \"start with number\",\n\t\t\targs:    args{name: \"1CustomStringField\"},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:    \"contain special character\",\n\t\t\targs:    args{name: \"CustomStringField!\"},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tif err := ValidateSearchAttributeKey(tt.args.name); (err != nil) != tt.wantErr {\n\t\t\t\tt.Errorf(\"ValidateSearchAttributeKey() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSanitizeSearchAttributeKey(t *testing.T) {\n\ttype args struct {\n\t\tname string\n\t}\n\ttests := []struct {\n\t\tname    string\n\t\targs    args\n\t\twant    string\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"pure character\",\n\t\t\targs: args{name: \"CustomStringField\"},\n\t\t\twant: \"CustomStringField\",\n\t\t},\n\t\t{\n\t\t\tname: \"alphanumeric\",\n\t\t\targs: args{name: \"CustomStringField1\"},\n\t\t\twant: \"CustomStringField1\",\n\t\t},\n\t\t{\n\t\t\tname: \"start with number\",\n\t\t\targs: args{name: \"1CustomStringField\"},\n\t\t\twant: \"CustomStringField\",\n\t\t},\n\t\t{\n\t\t\tname: \"contain special character\",\n\t\t\targs: args{name: \"CustomStringField!\"},\n\t\t\twant: \"CustomStringField\",\n\t\t},\n\t\t{\n\t\t\tname: \"contain special character in the middle\",\n\t\t\targs: args{name: \"CustomString-Field\"},\n\t\t\twant: \"CustomString_Field\",\n\t\t},\n\t\t{\n\t\t\tname:    \"all numbers\",\n\t\t\targs:    args{name: \"1234567890\"},\n\t\t\twant:    \"\",\n\t\t\twantErr: true,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot, err := SanitizeSearchAttributeKey(tt.args.name)\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Errorf(\"SanitizeSearchAttributeKey() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif got != tt.want {\n\t\t\t\tt.Errorf(\"SanitizeSearchAttributeKey() = %v, want %v\", got, tt.want)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "config/base.yaml",
    "content": "log:\n  stdout: true\n  level: info\n  levelKey: \"level\" \n"
  },
  {
    "path": "config/bench/base.yaml",
    "content": "log:\n  stdout: true\n  level: info\n"
  },
  {
    "path": "config/bench/basic.json",
    "content": "{\n  \"useBasicVisibilityValidation\": true,\n  \"contextTimeoutInSeconds\": 3,\n  \"failureThreshold\": 0.01,\n  \"totalLaunchCount\": 100,\n  \"routineCount\": 1,\n  \"waitTimeBufferInSeconds\": 10,\n  \"chainSequence\": 4,\n  \"concurrentCount\": 1,\n  \"payloadSizeBytes\": 1024,\n  \"executionStartToCloseTimeoutInSeconds\": 10\n}\n"
  },
  {
    "path": "config/bench/basic_panic.json",
    "content": "{\n  \"totalLaunchCount\": 5000,\n  \"routineCount\": 5,\n  \"chainSequence\": 4,\n  \"concurrentCount\": 1,\n  \"payloadSizeBytes\": 1024,\n  \"contextTimeoutInSeconds\": 3,\n  \"executionStartToCloseTimeoutInSeconds\": 1200,\n  \"panicStressWorkflow\": true,\n  \"failureThreshold\": 0.05\n}\n"
  },
  {
    "path": "config/bench/cancellation.json",
    "content": "{\n  \"totalLaunchCount\": 3000,\n  \"concurrency\": 1,\n  \"contextTimeoutInSeconds\": 3\n}\n"
  },
  {
    "path": "config/bench/concurrent_execution.json",
    "content": "{\n  \"totalBatches\": 15,\n  \"concurrency\": 5,\n  \"batchType\": \"childWorkflow\",\n  \"batchSize\": 100,\n  \"batchPeriodInSeconds\": 60,\n  \"batchMaxLatencyInSeconds\": 10,\n  \"batchTimeoutInSeconds\": 30\n}\n"
  },
  {
    "path": "config/bench/cron.json",
    "content": "{\n  \"testSuites\": [\n    {\n      \"name\": \"syncAPI\",\n      \"domain\": \"cadence-bench-sync\",\n      \"configs\": [\n        {\n          \"name\": \"basic\",\n          \"timeoutInSeconds\": 600,\n          \"basic\": {\n            \"totalLaunchCount\": 10,\n            \"routineCount\": 1,\n            \"chainSequence\": 4,\n            \"concurrentCount\": 1,\n            \"payloadSizeBytes\": 1024,\n            \"executionStartToCloseTimeoutInSeconds\": 60,\n            \"contextTimeoutInSeconds\": 3\n          }\n        },\n        {\n          \"name\": \"basic\",\n          \"description\": \"panic workflow test\",\n          \"timeoutInSeconds\": 600,\n          \"basic\": {\n            \"totalLaunchCount\": 10,\n            \"routineCount\": 1,\n            \"chainSequence\": 4,\n            \"concurrentCount\": 1,\n            \"payloadSizeBytes\": 1024,\n            \"executionStartToCloseTimeoutInSeconds\": 180,\n            \"contextTimeoutInSeconds\": 3,\n            \"panicStressWorkflow\": true\n          }\n        },\n        {\n          \"name\": \"cancellation\",\n          \"timeoutInSeconds\": 600,\n          \"cancellation\": {\n            \"totalLaunchCount\": 3000,\n            \"concurrency\": 1,\n            \"contextTimeoutInSeconds\": 2\n          }\n        },\n        {\n          \"name\": \"signal\",\n          \"timeoutInSeconds\": 1800,\n          \"signal\": {\n            \"loaderCount\": 10,\n            \"loadTestWorkflowCount\": 1000,\n            \"signalCount\": 50,\n            \"signalDataSize\": 2700,\n            \"rateLimit\": 7,\n            \"workflowExecutionTimeoutInSeconds\": 600,\n            \"decisionTaskTimeoutInSeconds\": 30,\n            \"campaignCount\": 30,\n            \"actionRate\": 0.02,\n            \"failureRate\": 0,\n            \"signalBeforeContinueAsNew\": 10,\n            \"enableRollingWindow\": true,\n            \"maxSignalDelayInSeconds\": 5,\n            \"maxSignalDelayCount\": 5,\n            \"failureThreshold\": 0.05\n          }\n        }\n      ]\n    },\n    {\n      \"name\": \"batchOp\",\n      \"domain\": \"cadence-bench-batch\",\n      \"configs\": [\n        {\n          \"name\": \"timer\",\n          \"timeoutInSeconds\": 600,\n          \"timer\": {\n            \"TotalTimerCount\": 100,\n            \"TimerPerWorkflow\": 1,\n            \"ShortestTimerDurationInSeconds\": 300,\n            \"LongestTimerDurationInSeconds\": 300,\n            \"MaxTimerLatencyInSeconds\": 2,\n            \"TimerTimeoutInSeconds\": 60,\n            \"RoutineCount\": 2\n          }\n        },\n        {\n          \"name\": \"concurrent-execution\",\n          \"description\": \"batch of activities\",\n          \"timeoutInSeconds\": 600,\n          \"concurrentExec\": {\n            \"totalBatches\": 15,\n            \"concurrency\": 5,\n            \"batchType\": \"activity\",\n            \"batchSize\": 100,\n            \"batchPeriodInSeconds\": 60,\n            \"batchMaxLatencyInSeconds\": 10,\n            \"batchTimeoutInSeconds\": 30\n          }\n        },\n        {\n          \"name\": \"concurrent-execution\",\n          \"description\": \"batch of child workflows\",\n          \"timeoutInSeconds\": 600,\n          \"concurrentExec\": {\n            \"totalBatches\": 15,\n            \"concurrency\": 5,\n            \"batchType\": \"childWorkflow\",\n            \"batchSize\": 100,\n            \"batchPeriodInSeconds\": 60,\n            \"batchMaxLatencyInSeconds\": 10,\n            \"batchTimeoutInSeconds\": 30\n          }\n        }\n      ]\n    }\n  ]\n}\n"
  },
  {
    "path": "config/bench/development.yaml",
    "content": "bench:\n  name: \"cadence-bench\"\n  domains: [\"cadence-bench\", \"cadence-bench-sync\", \"cadence-bench-batch\"] # it will start workers on all those domains(also try to register if not exists)\n  numTaskLists: 3 # it will start workers listening on cadence-bench-tl-0, cadence-bench-tl-1,  cadence-bench-tl-2\n\ncadence:\n  service: \"cadence-frontend\"\n  host: \"${CADENCE_FRONTEND_ADDRESS:127.0.0.1:7833}\"\n\nmetrics:\n  statsd: ~\n  prometheus:\n    timerType: \"histogram\"\n    listenAddress: \"127.0.0.1:8005\"\n"
  },
  {
    "path": "config/bench/signal.json",
    "content": "{\n  \"loaderCount\": 10,\n  \"loadTestWorkflowCount\": 1000,\n  \"signalCount\": 50,\n  \"signalDataSize\": 2700,\n  \"rateLimit\": 7,\n  \"workflowExecutionTimeoutInSeconds\": 600,\n  \"decisionTaskTimeoutInSeconds\": 30,\n  \"campaignCount\": 30,\n  \"actionRate\": 0.02,\n  \"failureRate\": 0,\n  \"signalBeforeContinueAsNew\": 10,\n  \"enableRollingWindow\": false,\n  \"maxSignalDelayInSeconds\": 5,\n  \"maxSignalDelayCount\": 5,\n  \"failureThreshold\": 0.05\n}\n"
  },
  {
    "path": "config/bench/timer.json",
    "content": "{\n  \"totalTimerCount\": 100,\n  \"timerPerWorkflow\": 1,\n  \"shortestTimerDurationInSeconds\": 300,\n  \"longestTimerDurationInSeconds\": 300,\n  \"maxTimerLatencyInSeconds\": 2,\n  \"timerTimeoutInSeconds\": 60,\n  \"routineCount\": 2\n}\n"
  },
  {
    "path": "config/canary/base.yaml",
    "content": "log:\n  stdout: true\n  level: info\n"
  },
  {
    "path": "config/canary/development.yaml",
    "content": "canary:\n  domains: [\"cadence-canary\"]\n  crossClusterTestMode: \"test-all\"\n  canaryDomainClusters: [\"cluster0\", \"cluster1\", \"cluster2\"]\n  excludes: [\"workflow.searchAttributes\", \"workflow.batch\", \"workflow.archival.history\", \"workflow.archival.visibility\"]\n\ncadence:\n  service: \"cadence-frontend\"\n  address: \"127.0.0.1:7833\"\n  #host: \"127.0.0.1:7933\" # replace address with host if using Thrift for compatibility\n\nmetrics:\n  statsd: ~\n  prometheus:\n    timerType: \"histogram\"\n    listenAddress: \"127.0.0.1:8005\"\n"
  },
  {
    "path": "config/credentials/client.key",
    "content": "-----BEGIN PRIVATE KEY-----\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQC+Hfqm5fJj1JK3\nERJkG17ilJF5RuB4YURsV7bDg6F25M9QTfqGu2kD4ethMDu3s2jn4BHDwKnMUjUK\neUdZdr5Nl+cv/IM3c5cJ0zme/in1ZKb958S3auSIWVg9BA/CC/9EWO8lhfS0fp6Q\nQGsE+eg0XKEFumOPvPny+rM/L6o+CmgXQxzEe/3Ao0bMO74Ni62YggHFh47oAm1d\nGw7KnCZ04YxCNFVYjbg6DOCA5hJrOf3JkhQI8LfGmRh8V9Q3tDYaWhGL/VuiYqsg\nKIhr1sUO+LUJoDBUUbAWE2RVC/XsHB/U999mbg9WgzsJtKCzuOgXq8pT7Nviz/Ue\n5y5bNWpvAgMBAAECggEBAJeu04zVac68i1IEDOhR7iJY5rgcFHv3HuBSGz9ihjCV\n3dH0ZS2Z2O+AEIw6L2sheVGHNKU50j9yV19D4+k9FBhUWd+vWAGl95ufKxnvLwra\ng2GQySKN5kfjAMOtueTYLZIaYrNNbS+U1m3A96HAwcMFJHOtgkHsRH/YBaskHG/h\n/V5699nhlptOeYK0I8v3WlnEtV5JpuBjRPdZDBwMGRf+ZlJADVzPPaFdTDXIwC/n\nOk69NsPHkG6XogDpWhW5sT3hIjOLU+r9uRXz4nx6UCi0Wnf3qX3smoHL1s6phRWE\nqCriT8pdgfo5Efx4Rx2FoD/Evwb12lh5AlnZo43YUAECgYEA5VZHAXn+E/OgY+x+\nBmoNjrjMXRHkA4gHFwXEiGrEfUIleSUSdSA0VMYFN/KoQrjkAqkS4gIbD8s8M6jx\nTKvOuD7UlDUcs1N9uvVJrHS1nvF9tBCOkAQIcWpdmli1CcqDz1g0WYpYARpX1KNA\nTcXtZhkeAqkbA+R6zmh3fMSz3s8CgYEA1DhnyWgiflcDaTagY1lpQKGED7FzQ7os\nqY9Hd7Ni1nS8FTnFmUN9SxaFtj/NryUjo8D4HkgEjvZ3clYCIMZk/M1Uw71uQFOC\nDxeJWqHCdTgXqLT8J7pmlEyUdFBQPtusEPWpiqeJTgpfWtuYBv5W9KajF9NLkgxW\nlit9LhrBomECgYA2zahILQvPXr+sQT5YcA1vVq/XacgQI/h1wvYVF1Z/DEiCK56+\niISY4X0rwnv8/cvW6upNQe2Pl6R6P9vx03ihxLnt9Mc3/zKjc6w4Xudr5Q+B8Rbw\nSVZvK9jUYnBYt6FJ8i0IXKvtD4t/j8d1DmBNrqDiR+DhzD6ghoxPL35ANwKBgFDH\nj6n4e3i/wkqgQnjOpeO2F0Cp+QgEq892/Gsx2yNql9U0gmSObfjQ/+CZdL+AUYqZ\nf1h3bVTFuD3LQ1AadIJN3kALsRXHM7Bu8xeOjyhzS34qNfkhWL6GT9PeZk0m7N4f\nxINEtl0AUb+R6tJuqcbCTz5YbWGQVL0gnKuZ/67BAoGBANVO3HP3ivOwsPEPWnmA\nSPc3k6GgpzcjPxpadVgG/9z8oyjH9q9xP84NzMNXftbS8gE41bYyNgAaql1CGXUk\nfF4ZMlWV/PKCm4s3+r9umlfE2bGY1EjiYzlR4atzpspfOTNnMyQZMunoCR2/vUAD\nC2LCXCBeUgP9tkEaZwu0lgBt\n-----END PRIVATE KEY-----\n"
  },
  {
    "path": "config/credentials/keytest",
    "content": "-----BEGIN PRIVATE KEY-----\nMIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQCxy6SW0eKVqr6j\nmAhUTg8bBaX56y8nYRoSlbDqiCv3xSfXVgnigqTAiQvDiIijS294Ig0DdxmbFpbV\nF6NFYoSpgW1DWth3oGycUAddaHZHcDUW2kboA1QQaQTs4Al7b8vjYT8iYn4ZDFG9\nqu0icMzJ+JF8Hv4vaffAb42L6EE0y0BpfhbBuPaG6nbUmO6j3pO6FwYiYEUEBMZZ\nphHbCp7/pvL3Al7E8eVj/W4OndKfMXV0IuOnvpIYJ/Jio4YL8/GFCLuYPF4f3+0g\nL6W39LPBzWfyBku2VGVkeKe9hryWFZepVjDpYPyHgtOAAur2Jj5t4PU8VBJjUaRC\n23RGFHB5AgMBAAECggEABj1T9Orf0W9nskDQ2QQ7cuVdZEJjpMrbTK1Aw1L8/Qc9\nTSkINDEayaV9mn1RXe61APcBSdP4ER7nXfTZiQ21LhLcWWg9T3cbh1b70oRqyI9z\nPi6HSBeWz4kfUBX9izMQFBZKzjYn6qaJp1b8bGXKRWkcvPRZqLhmsRPmeH3xrOHe\nqsIDhYXMjRoOgEUxLbk8iPLP6nx0icPJl/tHK2l76R+1Ko6TBE69Md2krUIuh0u4\nnm9n+Az+0GuvkFsLw5KMGhSBeqB+ez5qtFa8T8CUCn98IjiUDOwgZdFrNldFLcZf\nputw7O2qCA9LT+mFBQ6CVsVu/9tKeXQ9sJ7p3lxhwQKBgQDjt7HNIabLncdXPMu0\nByRyNVme0+Y1vbj9Q7iodk77hvlzWpD1p5Oyvq7cN+Cb4c1iO/ZQXMyUw+9hLgmf\nLNquH2d4hK1Jerzc/ciwu6dUBsCW8+0VJd4M2UNN15rJMPvbZGmqMq9Np1iCTCjE\ndvHo7xjPcJhsbhMbHq+PaUU7OQKBgQDH4KuaHBFTGUPkRaQGAZNRB8dDvSExV6ID\nPblzr80g9kKHUnQCQfIDLjHVgDbTaSCdRw7+EXRyRmLy5mfPWEbUFfIemEpEcEcb\n3geWeVDx4Z/FwprWFuVifRopRSQ/FAbMXLIui7OHXWLEtzBvLkR/uS2VIVPm10PV\npbh2EXifQQKBgQDbcOLbjelBYLt/euvGgfeCQ50orIS1Fy5UidVCKjh0tR5gJk95\nG1L+tjilqQc+0LtuReBYkwTm+2YMXSQSi1P05fh9MEYZgDjOMZYbkcpu887V6Rx3\n+7Te5uOv+OyFozmhs0MMK6m5iGGHtsK2iPUYBoj/Jj8MhorM4KZH6ic4KQKBgQCl\n3zIpg09xSc9Iue5juZz6qtzXvzWzkAj4bZnggq1VxGfzix6Q3Q8tSoG6r1tQWLbj\nLpwnhm6/guAMud6+eIDW8ptqfnFrmE26t6hOXMEq6lXANT5vmrKj6DP0uddZrZHy\nuJ55+B91n68elvPP4HKiGBfW4cCSGmTGAXAyM0+JwQKBgQCz2cNiFrr+oEnlHDLg\nEqsiEufppT4FSZPy9/MtuWuMgEOBu34cckYaai+nahQLQvH62KskTK0EUjE1ywub\nNPORuXcugxIBMHWyseOS7lrtrlSBxU9gntS7jHdM3IMrrUy9YZBvPvFGP0wLdpKM\nnvt3vT46hs3n28XZpb18uRkSDw==\n-----END PRIVATE KEY-----"
  },
  {
    "path": "config/credentials/keytest.pub",
    "content": "-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAscukltHilaq+o5gIVE4P\nGwWl+esvJ2EaEpWw6ogr98Un11YJ4oKkwIkLw4iIo0tveCINA3cZmxaW1RejRWKE\nqYFtQ1rYd6BsnFAHXWh2R3A1FtpG6ANUEGkE7OAJe2/L42E/ImJ+GQxRvartInDM\nyfiRfB7+L2n3wG+Ni+hBNMtAaX4Wwbj2hup21Jjuo96TuhcGImBFBATGWaYR2wqe\n/6by9wJexPHlY/1uDp3SnzF1dCLjp76SGCfyYqOGC/PxhQi7mDxeH9/tIC+lt/Sz\nwc1n8gZLtlRlZHinvYa8lhWXqVYw6WD8h4LTgALq9iY+beD1PFQSY1GkQtt0RhRw\neQIDAQAB\n-----END PUBLIC KEY-----"
  },
  {
    "path": "config/development.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence\"\n        connectTimeout: 2s # defaults to 2s if not defined\n        timeout: 5s # defaults to 10s if not defined\n        consistency: LOCAL_QUORUM # default value\n        serialConsistency: LOCAL_SERIAL # default value\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\n\nringpop:\n  name: cadence\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: ${FRONTEND_PORT:7933}\n      grpcPort: ${FRONTEND_PORT_GRPC:7833}\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: ${FRONTEND_PORT_PPROF:7936}\n\n  matching:\n    rpc:\n      port: ${MATCHING_PORT:7935}\n      grpcPort: ${MATCHING_PORT_GRPC:7835}\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: ${MATCHING_PORT_PPROF:7938}\n\n  history:\n    rpc:\n      port: ${HISTORY_PORT:7934}\n      grpcPort: ${HISTORY_PORT_GRPC:7834}\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: ${HISTORY_PORT_PPROF:7937}\n\n  worker:\n    rpc:\n      port: ${WORKER_PORT:7939}\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: ${WORKER_PORT_PPROF:7940}\n\n  shard-distributor:\n    rpc:\n      port: 7941\n      grpcPort: 7943\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7942\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      newInitialFailoverVersion: 1 # migrating to this new failover version\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\ndcRedirectionPolicy:\n  policy: \"noop\"\n  toDC: \"\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n      gstorage:\n        credentialsPath: \"/tmp/gcloud/keyfile.json\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n\nshardDistributorClient:\n  hostPort: \"localhost:7943\"\n\nshardDistribution:\n  election:\n    leaderPeriod: 60s\n    maxRandomDelay: 1s\n    failedElectionCooldown: 1s\n  namespaces:\n    - name: shard-distributor-canary\n      type: fixed\n      mode: onboarded\n      shardNum: 32\n    - name: shard-distributor-canary-ephemeral\n      mode: onboarded\n      type: ephemeral\n    - name: test-local-passthrough\n      type: fixed\n      mode: local_pass\n    - name: test-local-passthrough-shadow\n      type: ephemeral\n      mode: local_pass_shadow\n    - name: test-distributed-passthrough\n      type: fixed\n      mode: distributed_pass\n    - name: test-external-assignment\n      mode: distributed_pass\n      type: ephemeral\n  leaderStore:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"leader\"\n  store:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"store\"\n  process:\n    period: 1s\n    heartbeatTTL: 2s\n\nshard-distributor-matching:\n  namespaces:\n    - namespace: cadence-matching-dev\n      heartbeat_interval: 1s\n      migration_mode: local_pass\n      ttl_shard: 5m\n      ttl_report: 1m\n"
  },
  {
    "path": "config/development_async_wf_kafka_queue.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence\"\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\n\nringpop:\n  name: cadence\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 7933\n      grpcPort: 7833\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7936\n\n  matching:\n    rpc:\n      port: 7935\n      grpcPort: 7835\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7938\n\n  history:\n    rpc:\n      port: 7934\n      grpcPort: 7834\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7937\n\n  worker:\n    rpc:\n      port: 7939\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7940\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      newInitialFailoverVersion: 1 # migrating to this new failover version\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\ndcRedirectionPolicy:\n  policy: \"noop\"\n  toDC: \"\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n      gstorage:\n        credentialsPath: \"/tmp/gcloud/keyfile.json\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n\nasyncWorkflowQueues:\n  queue1:\n    type: \"kafka\"\n    config:\n      connection:\n        brokers:\n          - \"localhost:9092\"\n      topic: \"async-wf-topic1\"\n"
  },
  {
    "path": "config/development_es_opensearch.yaml",
    "content": "persistence:\n  advancedVisibilityStore: es-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        disableSniff: true\n        version: \"os2\"\n        tls:\n          enabled: false\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n"
  },
  {
    "path": "config/development_es_opensearch_migration.yaml",
    "content": "persistence:\n  advancedVisibilityStore: os-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        disableSniff: true\n        version: \"v7\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9201\"\n        indices:\n          visibility: cadence-visibility-dev\n    os-visibility:\n      elasticsearch:\n        disableSniff: true\n        version: \"os2\"\n        username: \"admin\"\n        password: \"DevTestInitial123!\"\n        tls:\n          enabled: true\n        url:\n          scheme: \"https\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n        migration:\n          enabled: true\n        consumerName: \"cadence-visibility-dev-os-consumer\"\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n"
  },
  {
    "path": "config/development_es_v6.yaml",
    "content": "persistence:\n  advancedVisibilityStore: es-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        version: \"v6\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n"
  },
  {
    "path": "config/development_es_v7.yaml",
    "content": "persistence:\n  advancedVisibilityStore: es-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        disableSniff: true\n        version: \"v7\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n\n\n"
  },
  {
    "path": "config/development_generic_oauth.yaml",
    "content": "persistence:\n  advancedVisibilityStore: es-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        version: \"v7\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n\nauthorization:\n  oauthAuthorizer:\n    enable: true\n    maxJwtTTL: 600000000\n    jwtCredentials:\n      algorithm: \"RS256\"\n      publicKey: \"config/credentials/keytest.pub\"\n    # provider section can be used to validate token issued by 3rd party provider (Okta, AWS, Google, etc.)\n    provider:\n      jwksURL: # AWS cognito example: \"https://cognito-idp.us-east-1.amazonaws.com/us-east-1_hNq90rT473/.well-known/jwks.json\"\n      # Custom data is extracted from token using JMES Path query language: https://jmespath.org/tutorial.html\n      adminAttributePath: # AWS cognito example:  \"permissions | contains(@, 'admin:true')\"\n      groupsAttributePath: # AWS cognito example: \"\\\"cognito:groups\\\" | join(', ', @)\"\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  masterClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n      authorizationProvider:\n        enable: true\n        type: \"OAuthAuthorization\"\n        privateKey: \"config/credentials/keytest\"\n"
  },
  {
    "path": "config/development_http_api.yaml",
    "content": "services:\n  frontend:\n    rpc:\n      port: 7933\n      grpcPort: 7833\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n      # enable HTTP server, allow to call Start worfklow using HTTP API\n      # Use curl to start a workflow:\n      #  curl  http://0.0.0.0:8800 \\\n      #   -H 'context-ttl-ms: 2000' \\\n      #   -H 'rpc-caller: rpc-client-name' \\\n      #   -H 'rpc-service: cadence-frontend' \\\n      #   -H 'rpc-encoding: json' \\\n      #   -H 'rpc-procedure: uber.cadence.api.v1.WorkflowAPI::StartWorkflowExecution' \\\n      #   -X POST --data @data.json\n      #  Where data.json content looks something like this:\n      #  {\n      #    \"domain\": \"samples-domain\",\n      #    \"workflowId\": \"workflowid123\",\n      #    \"execution_start_to_close_timeout\": \"11s\",\n      #    \"task_start_to_close_timeout\": \"10s\",\n      #    \"workflowType\": {\n      #      \"name\": \"workflow_type\"\n      #    },\n      #    \"taskList\": {\n      #      \"name\": \"tasklist-name\"\n      #    },\n      #   \"identity\": \"My custom identity\",\n      #    \"requestId\": \"4D1E4058-6FCF-4BA8-BF16-8FA8B02F9651\"\n      #  }\n      http:\n        # To enable HTTP TLS uncomment the following section\n        #tls:\n        #  enabled: true\n        #  certFile: config/credentials/keytest.crt\n        #  keyFile: config/credentials/keytest\n        #  caFiles:\n        #    - config/credentials/client.crt\n        #  requireClientAuth: true\n        #TLSMode: enforced\n        port: 8800\n        procedures: # list of available API procedures\n          # Admin API\n          - uber.cadence.admin.v1.AdminAPI::AddSearchAttribute\n          - uber.cadence.admin.v1.AdminAPI::CloseShard\n          - uber.cadence.admin.v1.AdminAPI::CountDLQMessages\n          - uber.cadence.admin.v1.AdminAPI::DeleteWorkflow\n          - uber.cadence.admin.v1.AdminAPI::DescribeCluster\n          - uber.cadence.admin.v1.AdminAPI::DescribeHistoryHost\n          - uber.cadence.admin.v1.AdminAPI::DescribeQueue\n          - uber.cadence.admin.v1.AdminAPI::DescribeShardDistribution\n          - uber.cadence.admin.v1.AdminAPI::DescribeWorkflowExecution\n          - uber.cadence.admin.v1.AdminAPI::GetCrossClusterTasks\n          - uber.cadence.admin.v1.AdminAPI::GetDLQReplicationMessages\n          - uber.cadence.admin.v1.AdminAPI::GetDomainIsolationGroups\n          - uber.cadence.admin.v1.AdminAPI::GetDomainReplicationMessages\n          - uber.cadence.admin.v1.AdminAPI::GetDynamicConfig\n          - uber.cadence.admin.v1.AdminAPI::GetGlobalIsolationGroups\n          - uber.cadence.admin.v1.AdminAPI::GetReplicationMessages\n          - uber.cadence.admin.v1.AdminAPI::GetWorkflowExecutionRawHistoryV2\n          - uber.cadence.admin.v1.AdminAPI::ListDynamicConfig\n          - uber.cadence.admin.v1.AdminAPI::MaintainCorruptWorkflow\n          - uber.cadence.admin.v1.AdminAPI::MergeDLQMessages\n          - uber.cadence.admin.v1.AdminAPI::PurgeDLQMessages\n          - uber.cadence.admin.v1.AdminAPI::ReadDLQMessages\n          - uber.cadence.admin.v1.AdminAPI::ReapplyEvents\n          - uber.cadence.admin.v1.AdminAPI::RefreshWorkflowTasks\n          - uber.cadence.admin.v1.AdminAPI::RemoveTask\n          - uber.cadence.admin.v1.AdminAPI::ResendReplicationTasks\n          - uber.cadence.admin.v1.AdminAPI::ResetQueue\n          - uber.cadence.admin.v1.AdminAPI::RespondCrossClusterTasksCompleted\n          - uber.cadence.admin.v1.AdminAPI::RestoreDynamicConfig\n          - uber.cadence.admin.v1.AdminAPI::UpdateDomainIsolationGroups\n          - uber.cadence.admin.v1.AdminAPI::UpdateDynamicConfig\n          - uber.cadence.admin.v1.AdminAPI::UpdateGlobalIsolationGroups\n          # Domain operations related API\n          - uber.cadence.api.v1.DomainAPI::DeprecateDomain\n          - uber.cadence.api.v1.DomainAPI::DescribeDomain\n          - uber.cadence.api.v1.DomainAPI::ListDomains\n          - uber.cadence.api.v1.DomainAPI::RegisterDomain\n          - uber.cadence.api.v1.DomainAPI::UpdateDomain\n          # Health check endpoint\n          - uber.cadence.api.v1.MetaAPI::Health\n          # Searching for workflows\n          - uber.cadence.api.v1.VisibilityAPI::CountWorkflowExecutions\n          - uber.cadence.api.v1.VisibilityAPI::GetSearchAttributes\n          - uber.cadence.api.v1.VisibilityAPI::ListArchivedWorkflowExecutions\n          - uber.cadence.api.v1.VisibilityAPI::ListClosedWorkflowExecutions\n          - uber.cadence.api.v1.VisibilityAPI::ListOpenWorkflowExecutions\n          - uber.cadence.api.v1.VisibilityAPI::ListWorkflowExecutions\n          - uber.cadence.api.v1.VisibilityAPI::ScanWorkflowExecutions\n          # Workflow execution related API endpoints\n          - uber.cadence.api.v1.WorkflowAPI::DescribeTaskList\n          - uber.cadence.api.v1.WorkflowAPI::DescribeWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::GetClusterInfo\n          - uber.cadence.api.v1.WorkflowAPI::GetTaskListsByDomain\n          - uber.cadence.api.v1.WorkflowAPI::GetWorkflowExecutionHistory\n          - uber.cadence.api.v1.WorkflowAPI::ListTaskListPartitions\n          - uber.cadence.api.v1.WorkflowAPI::QueryWorkflow\n          - uber.cadence.api.v1.WorkflowAPI::RefreshWorkflowTasks\n          - uber.cadence.api.v1.WorkflowAPI::RequestCancelWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::ResetWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::RestartWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::SignalWithStartWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::SignalWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::StartWorkflowExecution\n          - uber.cadence.api.v1.WorkflowAPI::TerminateWorkflowExecution\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7936\n"
  },
  {
    "path": "config/development_instance2.yaml",
    "content": "#\n# this is a copy of development.yaml, but with different ports for the services,\n# and bootstrapping on development.yaml's service hosts.\n#\n# this is intended for running multiple instances locally, within the same cluster.\n# start `--env development` first, then `--env development_instance2`.\n#\n\npersistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence\"\n        connectTimeout: 2s # defaults to 2s if not defined\n        timeout: 5s # defaults to 10s if not defined\n        consistency: LOCAL_QUORUM # default value\n        serialConsistency: LOCAL_SERIAL # default value\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\n\nringpop:\n  name: cadence\n  bootstrapMode: hosts\n  bootstrapHosts: [\n    \"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\",   # development.yaml list, for instance 1\n    \"127.0.0.1:27933\", \"127.0.0.1:27934\", \"127.0.0.1:27935\" # for this instance\n  ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 27933\n      grpcPort: 27833\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 27936\n\n  matching:\n    rpc:\n      port: 27935\n      grpcPort: 27835\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 27938\n\n  history:\n    rpc:\n      port: 27934\n      grpcPort: 27834\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 27937\n\n  worker:\n    rpc:\n      port: 27939\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 27940\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      newInitialFailoverVersion: 1 # migrating to this new failover version\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\ndcRedirectionPolicy:\n  policy: \"noop\"\n  toDC: \"\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n      gstorage:\n        credentialsPath: \"/tmp/gcloud/keyfile.json\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n"
  },
  {
    "path": "config/development_multiple_cassandra.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      shardedNosql:\n        defaultShard: nosqlDbShard1\n        shardingPolicy:\n          historyShardMapping:\n            - start: 0\n              end: 0\n              shard: nosqlDbShard1\n            - start: 1\n              end: 2\n              shard: nosqlDbShard2\n            - start: 3\n              end: 3\n              shard: nosqlDbShard1\n          taskListHashing:\n            shardOrder:\n              - nosqlDbShard1\n              - nosqlDbShard2\n        connections:\n          nosqlDbShard1:\n            nosqlPlugin:\n              pluginName: \"cassandra\"\n              hosts: \"127.0.0.1\"\n              keyspace: \"cadence\"\n              port: 9042\n          nosqlDbShard2:\n            nosqlPlugin:\n              pluginName: \"cassandra\"\n              hosts: \"127.0.0.1\"\n              keyspace: \"cadence\"\n              port: 9043\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\n\nringpop:\n  name: cadence\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 7933\n      grpcPort: 7833\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7936\n\n  matching:\n    rpc:\n      port: 7935\n      grpcPort: 7835\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7938\n\n  history:\n    rpc:\n      port: 7934\n      grpcPort: 7834\n      bindOnLocalHost: true\n      grpcMaxMsgSize: 33554432\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7937\n\n  worker:\n    rpc:\n      port: 7939\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7940\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\ndcRedirectionPolicy:\n  policy: \"noop\"\n  toDC: \"\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n      gstorage:\n        credentialsPath: \"/tmp/gcloud/keyfile.json\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n"
  },
  {
    "path": "config/development_multiple_mysql.yaml",
    "content": "persistence:\n  defaultStore: mysql-default\n  visibilityStore: \"\" # useMultipleDatabases must use advancedVisibilityStore only. Due to deep merging with development.yaml, this need to override to empty\n  advancedVisibilityStore: es-visibility\n  datastores:\n    mysql-default:\n      sql:\n        pluginName: \"mysql\"\n        connectProtocol: \"tcp\"\n        maxConns: 20\n        maxIdleConns: 20\n        maxConnLifetime: \"1h\"\n        useMultipleDatabases: true\n        nShards: 4\n        multipleDatabasesConfig:\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence0\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence1\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence2\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence3\"\n    es-visibility:\n      elasticsearch:\n        version: \"v7\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n"
  },
  {
    "path": "config/development_mysql.yaml",
    "content": "persistence:\n  defaultStore: mysql-default\n  visibilityStore: mysql-visibility\n  datastores:\n    mysql-default:\n      sql:\n        pluginName: \"mysql\"\n        databaseName: \"cadence\"\n        connectAddr: \"127.0.0.1:3306\"\n        connectProtocol: \"tcp\"\n        user: \"root\"\n        password: \"cadence\"\n        maxConns: 20\n        maxIdleConns: 20\n        maxConnLifetime: \"1h\"\n    mysql-visibility:\n      sql:\n        pluginName: \"mysql\"\n        databaseName: \"cadence_visibility\"\n        connectAddr: \"127.0.0.1:3306\"\n        connectProtocol: \"tcp\"\n        user: \"root\"\n        password: \"cadence\"\n        maxConns: 2\n        maxIdleConns: 2\n        maxConnLifetime: \"1h\""
  },
  {
    "path": "config/development_oauth.yaml",
    "content": "persistence:\n  advancedVisibilityStore: es-visibility\n  datastores:\n    es-visibility:\n      elasticsearch:\n        version: \"v7\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_es.yaml\"\n\nauthorization:\n  oauthAuthorizer:\n    enable: true\n    maxJwtTTL: 600000000\n    jwtCredentials:\n      algorithm: \"RS256\"\n      publicKey: \"config/credentials/keytest.pub\"\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  masterClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n      authorizationProvider:\n        enable: true\n        type: \"OAuthAuthorization\"\n        privateKey: \"config/credentials/keytest\"\n"
  },
  {
    "path": "config/development_pinot.yaml",
    "content": "persistence:\n  advancedVisibilityStore: pinot-visibility\n  datastores:\n    pinot-visibility:\n      pinot:\n        broker: \"localhost:8099\"\n        cluster: pinot-test\n        table: \"cadence_visibility_pinot\"\n        migration:\n          enabled: false\n    es-visibility:\n      elasticsearch:\n        version: \"v6\"\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n\nkafka:\n  tls:\n    enabled: false\n  clusters:\n    test:\n      brokers:\n        - 127.0.0.1:9092\n  topics:\n    cadence-visibility-dev:\n      cluster: test\n    cadence-visibility-dev-dlq:\n      cluster: test\n    cadence-visibility-pinot:\n      cluster: test\n    cadence-visibility-pinot-dlq:\n      cluster: test\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n    pinot-visibility:\n      topic: cadence-visibility-pinot\n      dlq-topic: cadence-visibility-pinot-dlq\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: \"config/dynamicconfig/development_pinot.yaml\"\n"
  },
  {
    "path": "config/development_postgres.yaml",
    "content": "persistence:\n  defaultStore: postgres-default\n  visibilityStore: postgres-visibility\n  datastores:\n    postgres-default:\n      sql:\n        pluginName: \"postgres\"\n        databaseName: \"cadence\"\n        connectAddr: \"127.0.0.1:5432\"\n        connectProtocol: \"tcp\"\n        user: \"postgres\"\n        password: \"cadence\"\n        maxConns: 20\n        maxIdleConns: 20\n        maxConnLifetime: \"1h\"\n    postgres-visibility:\n      sql:\n        pluginName: \"postgres\"\n        databaseName: \"cadence_visibility\"\n        connectAddr: \"127.0.0.1:5432\"\n        connectProtocol: \"tcp\"\n        user: \"postgres\"\n        password: \"cadence\"\n        maxConns: 2\n        maxIdleConns: 2\n        maxConnLifetime: \"1h\""
  },
  {
    "path": "config/development_prometheus.yaml",
    "content": "services:\n  frontend:\n    metrics:\n      statsd: ~\n      prometheus:\n        timerType: \"histogram\"\n        listenAddress: \"127.0.0.1:8000\"\n\n  matching:\n    metrics:\n      statsd: ~\n      prometheus:\n        timerType: \"histogram\"\n        listenAddress: \"127.0.0.1:8001\"\n\n  history:\n    metrics:\n      statsd: ~\n      prometheus:\n        timerType: \"histogram\"\n        listenAddress: \"127.0.0.1:8002\"\n\n  worker:\n    metrics:\n      statsd: ~\n      prometheus:\n        timerType: \"histogram\"\n        listenAddress: \"127.0.0.1:8003\"\n\n  shard-distributor:\n    metrics:\n      statsd: ~\n      prometheus:\n        timerType: \"histogram\"\n        listenAddress: \"127.0.0.1:8004\"\n"
  },
  {
    "path": "config/development_scylla.yaml",
    "content": "persistence:\n  defaultStore: scylla-default\n  visibilityStore: scylla-visibility\n  datastores:\n    scylla-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence\"\n    scylla-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility\"\n"
  },
  {
    "path": "config/development_sqlite.yaml",
    "content": "persistence:\n  defaultStore: sqlite-default\n  visibilityStore: sqlite-visibility\n  datastores:\n    sqlite-default:\n      sql:\n        pluginName: \"sqlite\"\n        maxConns: 1\n        maxIdleConns: 1\n        maxConnLifetime: \"128h\"\n        databaseName: \"cadence.db\"\n    sqlite-visibility:\n      sql:\n        pluginName: \"sqlite\"\n        maxConns: 1\n        maxIdleConns: 1\n        maxConnLifetime: \"128h\"\n        databaseName: \"cadence_visibility.db\"\n"
  },
  {
    "path": "config/development_tls.yaml",
    "content": "services:\n  frontend:\n    rpc:\n      tls:\n        enabled: true\n        certFile: config/credentials/keytest.crt\n        keyFile: config/credentials/keytest\n        caFiles:\n          - config/credentials/client.crt\n        requireClientAuth: true\n\n  matching:\n    rpc:\n      tls:\n        enabled: true\n        certFile: config/credentials/keytest.crt\n        keyFile: config/credentials/keytest\n\n  history:\n    rpc:\n      tls:\n        enabled: true\n        certFile: config/credentials/keytest.crt\n        keyFile: config/credentials/keytest\n\nclusterGroupMetadata:\n  clusterGroup:\n    cluster0:\n      tls:\n        enabled: true\n"
  },
  {
    "path": "config/development_xdc_cluster0.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_cluster0\"\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility_cluster0\"\n\nringpop:\n  name: cadence_cluster0\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:7933\", \"127.0.0.1:7934\", \"127.0.0.1:7935\", \"127.0.0.1:7940\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 7933\n      grpcPort: 7833\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster0\"\n    pprof:\n      port: 7936\n\n  matching:\n    rpc:\n      port: 7935\n      grpcPort: 7835\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster0\"\n    pprof:\n      port: 7938\n\n  history:\n    rpc:\n      port: 7934\n      grpcPort: 7834\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster0\"\n    pprof:\n      port: 7937\n\n  worker:\n    rpc:\n      port: 7940\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster0\"\n    pprof:\n      port: 7941\n\n\n  shard-distributor:\n    rpc:\n      port: 7951\n      grpcPort: 7941\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7952\n\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster0\"\n  clusterRedirectionPolicy:\n    policy: \"all-domain-apis-forwarding\" # if network communication overhead between clusters is high, consider use \"selected-apis-forwarding\" instead, but workflow/activity workers need to be connected to each cluster to keep high availability\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 1\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster1:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:8833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster2:\n      enabled: true\n      initialFailoverVersion: 2\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:9833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\n\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nshardDistribution:\n  election:\n    leaderPeriod: 60s\n    maxRandomDelay: 1s\n    failedElectionCooldown: 1s\n  namespaces:\n    - name: shard-distributor-canary\n      type: fixed\n      mode: onboarded\n      shardNum: 32\n    - name: test-external-assignment\n      mode: distributed_pass\n      type: ephemeral\n  leaderStore:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"leader\"\n  store:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"store\"\n  process:\n    period: 1s\n    heartbeatTTL: 2s\n\nshard-distributor-matching:\n  namespaces:\n    - namespace: cadence-matching-dev\n      heartbeat_interval: 1s\n      migration_mode: local_pass\n      ttl_shard: 5m\n      ttl_report: 1m\n"
  },
  {
    "path": "config/development_xdc_cluster1.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_cluster1\"\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility_cluster1\"\n\nringpop:\n  name: cadence_cluster1\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:8933\", \"127.0.0.1:8934\", \"127.0.0.1:8935\", \"127.0.0.1:8940\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 8933\n      grpcPort: 8833\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster1\"\n    pprof:\n      port: 8936\n\n  matching:\n    rpc:\n      port: 8935\n      grpcPort: 8835\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster1\"\n    pprof:\n      port: 8938\n\n  history:\n    rpc:\n      port: 8934\n      grpcPort: 8834\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster1\"\n    pprof:\n      port: 8937\n\n  worker:\n    rpc:\n      port: 8940\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster1\"\n    pprof:\n      port: 8941\n\n  shard-distributor:\n    rpc:\n      port: 7952\n      grpcPort: 7942\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7953\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster1\"\n  clusterRedirectionPolicy:\n    policy: \"all-domain-apis-forwarding\" # if network communication overhead between clusters is high, consider use \"selected-apis-forwarding\" instead, but workflow/activity workers need to be connected to each cluster to keep high availability\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 1\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster1:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:8833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster2:\n      enabled: true\n      initialFailoverVersion: 2\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:9833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nshardDistribution:\n  election:\n    leaderPeriod: 60s\n    maxRandomDelay: 1s\n    failedElectionCooldown: 1s\n  namespaces:\n    - name: shard-distributor-canary\n      type: fixed\n      mode: onboarded\n      shardNum: 32\n    - name: test-external-assignment\n      mode: distributed_pass\n      type: ephemeral\n  leaderStore:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"leader\"\n  store:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"store\"\n  process:\n    period: 1s\n    heartbeatTTL: 2s\n\nshard-distributor-matching:\n  namespaces:\n    - namespace: cadence-matching-dev\n      heartbeat_interval: 1s\n      migration_mode: local_pass\n      ttl_shard: 5m\n      ttl_report: 1m\n"
  },
  {
    "path": "config/development_xdc_cluster2.yaml",
    "content": "persistence:\n  defaultStore: cass-default\n  visibilityStore: cass-visibility\n  numHistoryShards: 4\n  datastores:\n    cass-default:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_cluster2\"\n    cass-visibility:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1\"\n        keyspace: \"cadence_visibility_cluster2\"\n\nringpop:\n  name: cadence_cluster2\n  bootstrapMode: hosts\n  bootstrapHosts: [ \"127.0.0.1:9933\", \"127.0.0.1:9934\", \"127.0.0.1:9935\", \"127.0.0.1:9940\" ]\n  maxJoinDuration: 30s\n\nservices:\n  frontend:\n    rpc:\n      port: 9933\n      grpcPort: 9833\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster2\"\n    pprof:\n      port: 9936\n\n  matching:\n    rpc:\n      port: 9935\n      grpcPort: 9835\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster2\"\n    pprof:\n      port: 9938\n\n  history:\n    rpc:\n      port: 9934\n      grpcPort: 9834\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster2\"\n    pprof:\n      port: 9937\n\n  worker:\n    rpc:\n      port: 9940\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence_cluster2\"\n    pprof:\n      port: 9941\n\n  shard-distributor:\n    rpc:\n      port: 7953\n      grpcPort: 7943\n      bindOnLocalHost: true\n    metrics:\n      statsd:\n        hostPort: \"127.0.0.1:8125\"\n        prefix: \"cadence\"\n    pprof:\n      port: 7954\n\nclusterGroupMetadata:\n  failoverVersionIncrement: 10\n  primaryClusterName: \"cluster0\"\n  currentClusterName: \"cluster2\"\n  clusterRedirectionPolicy:\n    policy: \"all-domain-apis-forwarding\" # if network communication overhead between clusters is high, consider use \"selected-apis-forwarding\" instead, but workflow/activity workers need to be connected to each cluster to keep high availability\n  clusterGroup:\n    cluster0:\n      enabled: true\n      initialFailoverVersion: 1\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:7833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster1:\n      enabled: true\n      initialFailoverVersion: 0\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:8833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n    cluster2:\n      enabled: true\n      initialFailoverVersion: 2\n      rpcName: \"cadence-frontend\"\n      rpcAddress: \"localhost:9833\" # this is to let worker service and XDC replicator connected to the frontend service. In cluster setup, localhost will not work\n      rpcTransport: \"grpc\"\n\narchival:\n  history:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n  visibility:\n    status: \"enabled\"\n    enableRead: true\n    provider:\n      filestore:\n        fileMode: \"0666\"\n        dirMode: \"0766\"\n\ndomainDefaults:\n  archival:\n    history:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_archival/development\"\n    visibility:\n      status: \"enabled\"\n      URI: \"file:///tmp/cadence_vis_archival/development\"\n\nblobstore:\n  filestore:\n    outputDirectory: \"/tmp/blobstore\"\n\ndynamicconfig:\n  client: filebased\n  configstore:\n    pollInterval: \"10s\"\n    updateRetryAttempts: 2\n    FetchTimeout: \"2s\"\n    UpdateTimeout: \"2s\"\n  filebased:\n    filepath: \"config/dynamicconfig/development.yaml\"\n    pollInterval: \"10s\"\n\nshardDistribution:\n  election:\n    leaderPeriod: 60s\n    maxRandomDelay: 1s\n    failedElectionCooldown: 1s\n  namespaces:\n    - name: shard-distributor-canary\n      type: fixed\n      mode: onboarded\n      shardNum: 32\n    - name: test-external-assignment\n      mode: distributed_pass\n      type: ephemeral\n  leaderStore:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"leader\"\n  store:\n    storageParams:\n      endpoints: [localhost:2379]\n      dialTimeout: 1s\n      prefix: \"store\"\n  process:\n    period: 1s\n    heartbeatTTL: 2s\n\nshard-distributor-matching:\n  namespaces:\n    - namespace: cadence-matching-dev\n      heartbeat_interval: 1s\n      migration_mode: local_pass\n      ttl_shard: 5m\n      ttl_report: 1m\n"
  },
  {
    "path": "config/dynamicconfig/README.md",
    "content": "Use development.yaml file to override the default dynamic config value (they are specified\nwhen creating the service config).\n\nEach key can have zero or more values and each value can have zero or more\nconstraints. There are only three types of constraint:\n    1. domainName: string\n    2. taskListName: string\n    3. taskType: int (0:Decision, 1:Activity)\nA value will be selected and returned if all its has exactly the same constraints\nas the ones specified in query filters (including the number of constraints).\n\nPlease use the following format:\n```\ntestGetBoolPropertyKey:\n  - value: false\n  - value: true\n    constraints:\n      domainName: \"global-samples-domain\"\n  - value: false\n    constraints:\n      domainName: \"samples-domain\"\ntestGetDurationPropertyKey:\n  - value: \"1m\"\n    constraints:\n      domainName: \"samples-domain\"\n      taskListName: \"longIdleTimeTasklist\"\ntestGetFloat64PropertyKey:\n  - value: 12.0\n    constraints:\n      domainName: \"samples-domain\"\ntestGetMapPropertyKey:\n  - value:\n      key1: 1\n      key2: \"value 2\"\n      key3:\n        - false\n        - key4: true\n          key5: 2.0\n```\n"
  },
  {
    "path": "config/dynamicconfig/development.yaml",
    "content": "frontend.enableClientVersionCheck:\n- value: true\n  constraints: {}\nsystem.minRetentionDays:\n- value: 0\n  constraints: {}\nmatching.enableStandbyTaskCompletion:\n- value: true\n  constraints: {}\nhistory.standbyClusterDelay:\n- value: 30s\n  constraints: {}\nhistory.standbyTaskMissingEventsResendDelay:\n- value: 30s\n  constraints: {}\nhistory.EnableConsistentQueryByDomain:\n- value: true\n  constraints: {}\nhistory.useNewInitialFailoverVersion:\n- value: true\n  constraints: {}\nsystem.enableDomainAuditLogging:\n- value: true\n  constraints: {}\nhistory.enableStrongIdempotency:\n- value: true\n  constraints: {}\nfrontend.globalRatelimiterMode:\n- value: local\nfrontend.validSearchAttributes:\n- value:\n    BinaryChecksums: 1\n    CadenceChangeVersion: 1\n    CloseStatus: 2\n    CloseTime: 2\n    CustomBoolField: 4\n    CustomDatetimeField: 5\n    CustomDomain: 1\n    CustomDoubleField: 3\n    CustomIntField: 2\n    CustomKeywordField: 1\n    CustomStringField: 0\n    DomainID: 1\n    ExecutionTime: 2\n    HistoryLength: 2\n    IsCron: 1\n    NewKey: 1\n    NumClusters: 2\n    Operator: 1\n    Passed: 4\n    RolloutID: 1\n    RunID: 1\n    ShardID: 2\n    StartTime: 2\n    TaskList: 1\n    UpdateTime: 2\n    WorkflowID: 1\n    WorkflowType: 1\n    addon: 1\n    addon-type: 1\n    environment: 1\n    project: 1\n    service: 1\n    user: 1\n    IsDeleted: 4\n    ExecutionStatus: 2\n    CronSchedule: 1\n    ScheduledExecutionTime: 2\n  constraints: {}\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nsystem.domainAuditLogTTL:\n  - value: \"15m\"\nmatching.enableClientAutoConfig:\n- value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n- value: true\nhistory.enableCleanupOrphanedHistoryBranchOnWorkflowCreation:\n- value: true\nshardDistributor.migrationMode:\n  - value: \"onboarded\"\n  - value: \"local_pass\"\n    constraints:\n      namespace: \"test-local-passthrough\"\n  - value: \"local_pass_shadow\"\n    constraints:\n      namespace: \"test-local-passthrough-shadow\"\n  - value: \"distributed_pass\"\n    constraints:\n      namespace: \"test-distributed-passthrough\"\n  - value: \"distributed_pass\"\n    constraints:\n      namespace: \"test-external-assignment\"\n"
  },
  {
    "path": "config/dynamicconfig/development_es.yaml",
    "content": "frontend.enableClientVersionCheck:\n  - value: true\nsystem.writeVisibilityStoreName:\n  - value: \"es\"\nsystem.readVisibilityStoreName:\n  - value: \"es\"\nfrontend.validSearchAttributes:\n  - value:\n      DomainID: 1\n      WorkflowID: 1\n      RunID: 1\n      WorkflowType: 1\n      StartTime: 2\n      ExecutionTime: 2\n      CloseTime: 2\n      CloseStatus: 2\n      HistoryLength: 2\n      TaskList: 1\n      IsCron: 4\n      NumClusters: 2\n      ClusterAttributeScope: 1\n      ClusterAttributeName: 1\n      UpdateTime: 2\n      CustomStringField: 0\n      CustomKeywordField: 1\n      CustomIntField: 2\n      CustomDoubleField: 3\n      CustomBoolField: 4\n      CustomDatetimeField: 5\n      project: 1\n      service: 1\n      environment: 1\n      addon: 1\n      addon-type: 1\n      user: 1\n      CustomDomain: 1\n      Operator: 1\n      RolloutID: 1\n      CadenceChangeVersion: 1\n      BinaryChecksums: 1\n      Passed: 4\n      ShardID: 2\n      ExecutionStatus: 2\n      CronSchedule: 1\n      ScheduledExecutionTime: 2\nsystem.minRetentionDays:\n    - value: 0\nhistory.EnableConsistentQueryByDomain:\n- value: true\n  constraints: {}\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n- value: true\n  constraints: {}\n"
  },
  {
    "path": "config/dynamicconfig/development_pinot.yaml",
    "content": "frontend.enableClientVersionCheck:\n  - value: true\nsystem.writeVisibilityStoreName:\n  - value: \"pinot\"\nsystem.readVisibilityStoreName:\n  - value: \"pinot\"\nsystem.enableLogCustomerQueryParameter:\n  - value: false\nfrontend.validSearchAttributes:\n  - value:\n      DomainID: 1\n      WorkflowID: 1\n      RunID: 1\n      WorkflowType: 1\n      StartTime: 2\n      ExecutionTime: 2\n      CloseTime: 2\n      CloseStatus: 2\n      HistoryLength: 2\n      TaskList: 1\n      IsCron: 1\n      NumClusters: 2\n      UpdateTime: 2\n      CustomStringField: 0\n      CustomKeywordField: 1\n      CustomIntField: 2\n      CustomDoubleField: 3\n      CustomBoolField: 4\n      CustomDatetimeField: 5\n      project: 1\n      service: 1\n      environment: 1\n      addon: 1\n      addon-type: 1\n      user: 1\n      CustomDomain: 1\n      Operator: 1\n      RolloutID: 1\n      CadenceChangeVersion: 1\n      BinaryChecksums: 1\n      Passed: 4\n      ShardID: 2\n      IsDeleted: 4\nsystem.minRetentionDays:\n    - value: 0\nhistory.EnableConsistentQueryByDomain:\n- value: true\n  constraints: {}\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_child.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_cron.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_cron\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive_cron.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_invalid_cluster_attribute.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_regional_failover.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_regional_failover\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive_regional_failover.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_regional_failover_start_same_wfid.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_regional_failover_start_same_wfid\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive_regional_failover_start_same_wfid.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_regional_failover_start_same_wfid_2.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_regional_failover_start_same_wfid_2\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive_regional_failover_start_same_wfid_2.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_same_wfid.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 5s\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_same_wfid_signalwithstart.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_same_wfid_signalwithstart\" replication simulation scenario\n# configured via simulation/replication/testdata/replication_simulation_activeactive_same_wfid_signalwithstart.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_same_wfid_signalwithstart_delayed.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive_same_wfid_signalwithstart_delayed\" replication simulation scenario\n# configured via simulation/replication/testdata/replication_simulation_activeactive_same_wfid_signalwithstart_delayed.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_signalwithstart_terminateifrunning.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activeactive_start_terminateifrunning.yml",
    "content": "# This file is used as dynamicconfig override for \"activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_activepassive_to_activeactive.yml",
    "content": "# This file is used as dynamicconfig override for \"activepassive_to_activeactive\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_activepassive_to_activeactive.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.standbyTaskMissingEventsResendDelay:\n  - value: 5s\nhistory.standbyTaskMissingEventsDiscardDelay:\n  - value: 10s\nhistory.standbyClusterDelay:\n  - value: 10s\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_budget_manager.yml",
    "content": "# This file is used as dynamicconfig override for \"budget_manager\" replication simulation scenario\n# It enables the replication budget manager to test cache capacity management during replication\n\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\n\n# Replication task processor settings\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\n\nfrontend.failoverCoolDown:\n  - value: 5s\n\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\n\n# Set budget limits for testing\n# Max size in bytes - set to a moderate value to test capacity enforcement\nhistory.replicationBudgetManagerMaxSizeBytes:\n  - value: 1048576  # 1MB\n    constraints: {}\n\n# Max count - set to a moderate value to test count enforcement\nhistory.replicationBudgetManagerMaxSizeCount:\n  - value: 100\n    constraints: {}\n\n# Soft cap threshold - test fair-share enforcement\nhistory.replicationBudgetManagerSoftCapThreshold:\n  - value: 0.8  # 80% soft cap\n    constraints: {}\n\n# Cache settings for replication\nhistory.replicatorCacheCapacity:\n  - value: 50\n    constraints: {}\n\nhistory.replicatorCacheMaxSize:\n  - value: 524288  # 512KB per cache\n    constraints: {}\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_clusterredirection.yml",
    "content": "# This file is used as dynamicconfig override for \"clusterredirection\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_clusterredirection.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.EnableConsistentQueryByDomain:\n  - value: true\nhistory.EnableConsistentQuery:\n  - value: true\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_default.yml",
    "content": "# This file is used as dynamicconfig override for \"default\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_default.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\n"
  },
  {
    "path": "config/dynamicconfig/replication_simulation_reset.yml",
    "content": "# This file is used as dynamicconfig override for \"reset\" replication simulation scenario configured via simulation/replication/testdata/replication_simulation_reset.yaml\nsystem.writeVisibilityStoreName:\n  - value: \"db\"\nsystem.readVisibilityStoreName:\n  - value: \"db\"\nhistory.replicatorTaskBatchSize:\n  - value: 25\n    constraints: {}\nfrontend.failoverCoolDown:\n  - value: 5s\nhistory.ReplicationTaskProcessorStartWait: # default is 5s. repl task processor sleeps this much before processing received messages.\n  - value: 10ms\nhistory.enableTimerQueueV2:\n  - value: true\n    constraints: {}\nhistory.enableTransferQueueV2:\n  - value: true\n    constraints: {}\n"
  },
  {
    "path": "docker/README.md",
    "content": "Quickstart for development with local Cadence server\n====================================\n\n**Prerequisite**: [Docker + Docker compose](https://docs.docker.com/engine/installation/)\n\nFollowing steps will bring up the docker container running cadence server\nalong with all its dependencies (cassandra, prometheus, grafana). Exposes cadence\nfrontend on ports 7933 (tchannel) / 7833 (grpc), web on port 8088, and grafana on port 3000.\n\n```\ncd $GOPATH/src/github.com/uber/cadence/docker\ndocker compose up\n```\n> Note: Above command will run with `master-auto-setup` image, which is a changing image all the time.\n> You can use a released image if you want a stable version. See the below section of \"Using a released image\".\n\nTo update your `master-auto-setup` image to the latest version\n```\ndocker pull ubercadence/server:master-auto-setup\n```\n\n* View Cadence-Web at http://localhost:8088\n* View metrics at http://localhost:3000\n\nUsing different docker-compose files\n-----------------------\nBy default `docker compose up` will run with `docker-compose.yml` in this folder.\nThis compose file is running with Cassandra, with basic visibility,\nusing Prometheus for emitting metric, with Grafana access.\n\n\nWe also provide several other compose files for different features/modes:\n\n* docker-compose-es.yml enables advanced visibility with ElasticSearch 6.x\n* docker-compose-es-v7.yml enables advanced visibility with ElasticSearch 7.x\n* docker-compose-mysql.yml uses MySQL as persistence storage\n* docker-compose-postgres.yml uses PostgreSQL as persistence storage\n* docker-compose-statsd.yaml runs with Statsd+Graphite\n* docker-compose-multiclusters.yaml runs with 2 cadence clusters\n\nFor example:\n```\ndocker compose -f docker-compose-mysql.yml up\n```\n\nAlso feel free to make your own to combine the above features.\n\nRun canary and bench(load test)\n-----------------------\nAfter a local cadence server started, use the below command to run canary ro bench test\n```\ndocker compose -f docker-compose-bench.yml up\n```\nand\n```\ndocker compose -f docker-compose-canary.yml up\n```\n\n\nUsing a released image\n-----------------------\nThe above compose files all using master image. It's taking the latest bits on the master branch of this repo.\nYou may want to use more stable version from our release process.\n\nWith every tagged release of the cadence server, there is also a corresponding\ndocker image that's uploaded to docker hub. In addition, the release will also\ncontain a **docker.tar.gz** file (docker-compose startup scripts).\n\nGo [here](https://github.com/cadence-workflow/cadence/releases/latest) to download a latest **docker.tar.gz**\n\nExecute the following\ncommands to start a pre-built image along with all dependencies.\n\n```\ntar -xzvf docker.tar.gz\ncd docker\ndocker compose up\n```\n\nDIY: Building an image for any tag or branch\n-----------------------------------------\nReplace **YOUR_TAG** and **YOUR_CHECKOUT_BRANCH_OR_TAG** in the below command to build:\nYou can checkout a [release tag](https://github.com/cadence-workflow/cadence/tags) (e.g. v0.21.3) or any branch you are interested.\n\n```\ncd $GOPATH/src/github.com/uber/cadence\ngit checkout YOUR_CHECKOUT_BRANCH_OR_TAG\ndocker build . -t ubercadence/<imageName>:YOUR_TAG\n```\n\nYou can specify `--build-arg TARGET=<target>` to build different binaries.\nThere are three targets supported:\n* server. Default target if not specified. This will build a regular server binary.\n* auto-setup. The image will setup all the DB/ElasticSearch schema during startup.\n* cli. This image is for [CLI](https://cadenceworkflow.io/docs/cli/).\n\nFor example of auto-setup images:\n```\ncd $GOPATH/src/github.com/uber/cadence\ngit checkout YOUR_CHECKOUT_BRANCH\ndocker build . -t ubercadence/server:YOUR_TAG-auto-setup --build-arg TARGET=auto-setup\n```\nReplace the tag of **image: ubercadence/server** to **YOUR_TAG** in docker-compose.yml .\nThen stop service and remove all containers using the below commands.\n```\ndocker compose down\ndocker compose up\n```\n\nDIY: Troubleshooting docker builds\n----------------------------------\n\nNote that Docker has been making changes to its build system, and the new system is currently missing some capabilities\nthat the old one had, and makes major changes to how you control it.\nWhen searching for workarounds, make sure you are looking at modern answers, and consider specifically searching for\n\"buildkit\" solutions.\n\nYou can also disable buildkit explicitly with `DOCKER_BUILDKIT=0 docker build ...`.\n\nFor output limiting (e.g. `[output clipped ...]` messages), or for anything that requires changing buildkit environment\nvariables or other options, start a new builder and use it to build with:\n```\n# create a new builder with your options\ndocker buildx create ...\n# which will print out a name, use it in the build step.\n\n# now use the exact same command as normal, but it prepends `buildx` and adds a builder flag.\ndocker buildx build . -t ubercadence/<imageName>:YOUR_TAG --builder <that_builder_name>\n```\n\nFor output limiting (e.g. `[output clipped ...]` messages), you can fix this with some buildkit env variables:\n```\ndocker buildx create --driver-opt env.BUILDKIT_STEP_LOG_MAX_SIZE=-1 --driver-opt env.BUILDKIT_STEP_LOG_MAX_SPEED=-1\n```\n\nDIY: Running a custom cadence server locally alongside cadence requirements\n---------------------------------------------------------------------------\nIf you want to test out a custom-built cadence server, while running all the normal cadence dependencies, there's a simple workflow to do that:\n\nMake your cadence server changes and build using \"make bins\". Then start everything, stop cadence server, and run your own cadence-server:\n```\ndocker compose up\ndocker stop docker-cadence-1\n./cadence-server start\n```\n\nUsing docker image for production\n=========================\nIn a typical production setting, dependencies (cassandra / statsd server) are\nmanaged / started independently of the cadence-server. To use the container in\na production setting, use the following docker command:\n\n```\ndocker run -e CASSANDRA_SEEDS=10.x.x.x                  -- csv of cassandra server ipaddrs\n    -e CASSANDRA_USER=<username>                        -- Cassandra username\n    -e CASSANDRA_PASSWORD=<password>                    -- Cassandra password\n    -e KEYSPACE=<keyspace>                              -- Cassandra keyspace\n    -e VISIBILITY_KEYSPACE=<visibility_keyspace>        -- Cassandra visibility keyspace, if using basic visibility\n    -e KAFKA_SEEDS=10.x.x.x                             -- Kafka broker seed, if using ElasticSearch + Kafka for advanced visibility feature\n    -e CASSANDRA_PROTO_VERSION=<protocol_version>       -- Cassandra protocol version\n    -e ES_SEEDS=10.x.x.x                                -- ElasticSearch seed , if using ElasticSearch + Kafka for advanced visibility feature\n    -e RINGPOP_SEEDS=10.x.x.x,10.x.x.x                  -- csv of ipaddrs for gossip bootstrap\n    -e STATSD_ENDPOINT=10.x.x.x:8125                    -- statsd server endpoint\n    -e NUM_HISTORY_SHARDS=1024                          -- Number of history shards\n    -e SERVICES=history,matching                        -- Spinup only the provided services, separated by commas, options are frontend,history,matching and worker\n    -e LOG_LEVEL=debug,info                             -- Logging level\n    -e DYNAMIC_CONFIG_FILE_PATH=<dynamic_config_file>   -- Dynamic config file to be watched, default to /etc/cadence/config/dynamicconfig/development.yaml, but you can choose /etc/cadence/config/dynamicconfig/development_es.yaml if using ElasticSearch\n    ubercadence/server:<tag>\n```\nNote that each env variable has a default value, so you don't have to specify it if the default works for you.\nFor more options to configure the docker, please refer to `config_template.yaml`.\n\nFor `<tag>`, use `auto-setup` images only for first initial setup, and use regular ones for production deployment. See the above explanation about `auto-setup`.\n\nWhen upgrading, follow the release instrusctions if version upgrades require some configuration or schema changes.\n"
  },
  {
    "path": "docker/config/bench/development.yaml",
    "content": "log:\n  stdout: true\n  level: info\n\nbench:\n  name: \"cadence-bench\"\n  domains: [\"cadence-bench\", \"cadence-bench-sync\", \"cadence-bench-batch\"]\n  numTaskLists: 3\n\ncadence:\n  service: \"cadence-frontend\"\n  host: \"${CADENCE_FRONTEND_ADDRESS:host.docker.internal:7933}\" # see https://docs.docker.com/desktop/mac/networking/\n"
  },
  {
    "path": "docker/config/canary/development.yaml",
    "content": "log:\n  stdout: true\n  level: info\n\ncanary:\n  domains: [\"cadence-canary\"]\n  excludes: [\"workflow.searchAttributes\", \"workflow.batch\", \"workflow.archival.history\", \"workflow.archival.visibility\"]\n\ncadence:\n  service: \"cadence-frontend\"\n  address: \"host.docker.internal:7833\" # address is for gRPC\n  #host: \"host.docker.internal:7933\" # for using thrift, replace address with host\n\n"
  },
  {
    "path": "docker/config_template.yaml",
    "content": "log:\n    stdout: true\n    level: {{ default .Env.LOG_LEVEL \"info\" }}\n\npersistence:\n    numHistoryShards: {{ default .Env.NUM_HISTORY_SHARDS \"4\" }}\n    defaultStore: default\n    visibilityStore: visibility\n    {{- $es := default .Env.ENABLE_ES \"false\" | lower -}}\n    {{- if eq $es \"true\" }}\n    advancedVisibilityStore: es-visibility\n    {{- end }}\n    datastores:\n        {{- $db := default .Env.DB \"cassandra\" | lower -}}\n        {{- if or (eq $db \"cassandra\") (eq $db \"scylla\") }}\n        default:\n            nosql:\n                pluginName: \"cassandra\"\n                hosts: {{ default .Env.CASSANDRA_SEEDS \"\" }}\n                keyspace: {{ default .Env.KEYSPACE \"cadence\" }}\n                user: {{ default .Env.CASSANDRA_USER \"\" }}\n                password: {{ default .Env.CASSANDRA_PASSWORD \"\" }}\n                protoVersion: {{ default .Env.CASSANDRA_PROTO_VERSION \"4\" }}\n        visibility:\n            nosql:\n                pluginName: \"cassandra\"\n                hosts: {{ default .Env.CASSANDRA_SEEDS \"\" }}\n                keyspace: {{ default .Env.VISIBILITY_KEYSPACE \"cadence_visibility\" }}\n                user: {{ default .Env.CASSANDRA_USER \"\" }}\n                password: {{ default .Env.CASSANDRA_PASSWORD \"\" }}\n                protoVersion: {{ default .Env.CASSANDRA_PROTO_VERSION \"4\" }}\n        {{- else if eq $db \"mysql\" }}\n        default:\n            sql:\n                pluginName: \"mysql\"\n                databaseName: {{ default .Env.DBNAME \"cadence\" }}\n                connectAddr: \"{{ default .Env.MYSQL_SEEDS \"\" }}:{{ default .Env.DB_PORT \"3306\" }}\"\n                connectProtocol: \"tcp\"\n                user: {{ default .Env.MYSQL_USER \"\" }}\n                password: {{ default .Env.MYSQL_PWD \"\" }}\n                {{- if .Env.MYSQL_TX_ISOLATION_COMPAT }}\n                connectAttributes:\n                    tx_isolation: 'READ-COMMITTED'\n                {{- end }}\n        visibility:\n            sql:\n                pluginName: \"mysql\"\n                databaseName: {{ default .Env.VISIBILITY_DBNAME \"cadence_visibility\" }}\n                connectAddr: \"{{ default .Env.MYSQL_SEEDS \"\" }}:{{ default .Env.DB_PORT \"3306\" }}\"\n                connectProtocol: \"tcp\"\n                user: {{ default .Env.MYSQL_USER \"\" }}\n                password: {{ default .Env.MYSQL_PWD \"\" }}\n                {{- if .Env.MYSQL_TX_ISOLATION_COMPAT }}\n                connectAttributes:\n                    tx_isolation: 'READ-COMMITTED'\n                {{- end }}\n        {{- else if eq $db \"postgres\" }}\n        default:\n            sql:\n                pluginName: \"postgres\"\n                encodingType: \"thriftrw\"\n                decodingTypes: [\"thriftrw\"]\n                databaseName: {{ default .Env.DBNAME \"cadence\" }}\n                connectAddr: \"{{ default .Env.POSTGRES_SEEDS \"\" }}:{{ default .Env.DB_PORT \"5432\" }}\"\n                connectProtocol: \"tcp\"\n                user: {{ default .Env.POSTGRES_USER \"\" }}\n                password: {{ default .Env.POSTGRES_PWD \"\" }}\n                maxConns: 20\n                maxIdleConns: 20\n                maxConnLifetime: \"1h\"\n        visibility:\n            sql:\n                pluginName: \"postgres\"\n                encodingType: \"thriftrw\"\n                decodingTypes: [\"thriftrw\"]\n                databaseName: {{ default .Env.VISIBILITY_DBNAME \"cadence_visibility\" }}\n                connectAddr: \"{{ default .Env.POSTGRES_SEEDS \"\" }}:{{ default .Env.DB_PORT \"5432\" }}\"\n                connectProtocol: \"tcp\"\n                user: {{ default .Env.POSTGRES_USER \"\" }}\n                password: {{ default .Env.POSTGRES_PWD \"\" }}\n                maxConns: 20\n                maxIdleConns: 20\n                maxConnLifetime: \"1h\"\n        {{- end }}\n        {{- if eq $es \"true\" }}\n        es-visibility:\n            elasticsearch:\n                version: {{ default .Env.ES_VERSION \"\" }}\n                username: {{ default .Env.ES_USER \"\" }}\n                password: {{ default .Env.ES_PWD \"\" }}\n                url:\n                    scheme: \"http\"\n                    host: \"{{ default .Env.ES_SEEDS \"\" }}:{{ default .Env.ES_PORT \"9200\" }}\"\n                indices:\n                    visibility: {{ default .Env.VISIBILITY_NAME \"cadence-visibility-dev\" }}\n        {{- end }}\n\nringpop:\n    name: cadence\n    broadcastAddress: {{ default .Env.BROADCAST_ADDRESS \"\" }}\n    bootstrapMode: {{ default .Env.RINGPOP_BOOTSTRAP_MODE \"hosts\" }}\n    {{- if .Env.RINGPOP_SEEDS }}\n    bootstrapHosts:\n    {{- range $seed := (split .Env.RINGPOP_SEEDS \",\") }}\n        - {{ . }}\n    {{- end }}\n    {{- else }}\n    bootstrapHosts:\n        - {{ .Env.HOST_IP }}:{{ default .Env.FRONTEND_PORT \"7933\" }}\n        - {{ .Env.HOST_IP }}:{{ default .Env.HISTORY_PORT \"7934\" }}\n        - {{ .Env.HOST_IP }}:{{ default .Env.MATCHING_PORT \"7935\" }}\n        - {{ .Env.HOST_IP }}:{{ default .Env.WORKER_PORT \"7939\" }}\n    {{- end }}\n    maxJoinDuration: 30s\n\nservices:\n    frontend:\n        rpc:\n            port: {{ default .Env.FRONTEND_PORT \"7933\" }}\n            grpcPort: {{ default .Env.GRPC_FRONTEND_PORT \"7833\" }}\n            bindOnIP: {{ default .Env.BIND_ON_IP \"127.0.0.1\" }}\n            {{- if .Env.FRONTEND_HTTP_PORT }}\n            http:\n                port: {{ .Env.FRONTEND_HTTP_PORT }}\n                procedures:\n                {{- range $seed := (split .Env.FRONTEND_HTTP_PROCEDURES \",\") }}\n                    - {{ . }}\n                {{- end }}\n            {{- end }}\n        {{- if .Env.STATSD_ENDPOINT }}\n        metrics:\n            statsd:\n                hostPort: {{ .Env.STATSD_ENDPOINT }}\n                prefix: {{ default .Env.STATSD_FRONTEND_PREFIX \"cadence-frontend\" }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT_0 }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT_0 }}\n        {{- end }}\n        {{- if .Env.FRONTEND_PPROF_PORT }}\n        pprof:\n            port: {{ .Env.FRONTEND_PPROF_PORT }}\n            host: {{ default .Env.BIND_ON_IP \"localhost\" }}\n        {{- end }}\n    matching:\n        rpc:\n            port: {{ default .Env.MATCHING_PORT \"7935\" }}\n            grpcPort: {{ default .Env.GRPC_MATCHING_PORT \"7835\" }}\n            bindOnIP: {{ default .Env.BIND_ON_IP \"127.0.0.1\" }}\n        {{- if .Env.STATSD_ENDPOINT }}\n        metrics:\n            statsd:\n                hostPort: {{ .Env.STATSD_ENDPOINT }}\n                prefix: {{ default .Env.STATSD_MATCHING_PREFIX \"cadence-matching\" }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT_1 }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT_1 }}\n        {{- end }}\n      {{- if .Env.MATCHING_PPROF_PORT }}\n        pprof:\n            port: {{ .Env.MATCHING_PPROF_PORT }}\n            host: {{ default .Env.BIND_ON_IP \"localhost\" }}\n      {{- end }}\n    history:\n        rpc:\n            port: {{ default .Env.HISTORY_PORT \"7934\" }}\n            grpcPort: {{ default .Env.GRPC_HISTORY_PORT \"7834\" }}\n            bindOnIP: {{ default .Env.BIND_ON_IP \"127.0.0.1\" }}\n        {{- if .Env.STATSD_ENDPOINT }}\n        metrics:\n            statsd:\n                hostPort: {{ .Env.STATSD_ENDPOINT }}\n                prefix: {{ default .Env.STATSD_HISTORY_PREFIX \"cadence-history\" }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT_2 }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT_2 }}\n        {{- end }}\n      {{- if .Env.HISTORY_PPROF_PORT }}\n        pprof:\n            port: {{ .Env.HISTORY_PPROF_PORT }}\n            host: {{ default .Env.BIND_ON_IP \"localhost\" }}\n      {{- end }}\n    worker:\n        rpc:\n            port: {{ default .Env.WORKER_PORT \"7939\" }}\n            bindOnIP: {{ default .Env.BIND_ON_IP \"127.0.0.1\" }}\n        {{- if .Env.STATSD_ENDPOINT }}\n        metrics:\n            statsd:\n                hostPort: {{ .Env.STATSD_ENDPOINT }}\n                prefix: {{ default .Env.STATSD_WORKER_PREFIX \"cadence-worker\" }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT }}\n        {{- else if .Env.PROMETHEUS_ENDPOINT_3 }}\n        metrics:\n            prometheus:\n                timerType: {{ default .Env.PROMETHEUS_TIMER_TYPE \"histogram\" }}\n                listenAddress: {{ .Env.PROMETHEUS_ENDPOINT_3 }}\n        {{- end }}\n      {{- if .Env.WORKER_PPROF_PORT }}\n        pprof:\n            port: {{ .Env.WORKER_PPROF_PORT }}\n            host: {{ default .Env.BIND_ON_IP \"localhost\" }}\n      {{- end }}\nclusterGroupMetadata:\n    clusterRedirectionPolicy:\n        policy: {{ default .Env.CLUSTER_REDIRECT_POLICY \"all-domain-apis-forwarding\" }}\n    failoverVersionIncrement: 10\n    primaryClusterName: \"cluster0\"\n    {{- if .Env.IS_NOT_PRIMARY }}\n    currentClusterName: \"cluster1\"\n    {{- else }}\n    currentClusterName: \"cluster0\"\n    {{- end }}\n    clusterGroup:\n        cluster0:\n            enabled: true\n            initialFailoverVersion: 0\n            rpcName: \"cadence-frontend\"\n            rpcAddress: {{ default .Env.PRIMARY_FRONTEND_SERVICE \"cadence\" }}:{{ default .Env.FRONTEND_PORT \"7833\" }}\n            rpcTransport: \"grpc\"\n            authorizationProvider:\n                enable: {{ default .Env.ENABLE_OAUTH \"false\" }}\n                type: \"OAuthAuthorization\"\n                privateKey: {{ default .Env.OAUTH_PRIVATE_KEY \"\" }}\n        {{- if or .Env.ENABLE_GLOBAL_DOMAIN }}\n        cluster1:\n            enabled: true\n            initialFailoverVersion: 2\n            rpcName: \"cadence-frontend\"\n            rpcAddress: {{ default .Env.SECONDARY_FRONTEND_SERVICE \"cadence-secondary\" }}:{{ default .Env.FRONTEND_PORT \"7833\" }}\n            rpcTransport: \"grpc\"\n            authorizationProvider:\n                enable: {{ default .Env.ENABLE_OAUTH \"false\" }}\n                type: \"OAuthAuthorization\"\n                privateKey: {{ default .Env.OAUTH_PRIVATE_KEY \"\" }}\n        {{- end }}\n\narchival:\n  history:\n    status: {{ default .Env.HISTORY_ARCHIVAL_STATUS \"disabled\" }}\n    enableRead: {{ default .Env.HISTORY_ARCHIVAL_ENABLE_READ \"false\" }}\n    provider:\n      filestore:\n        fileMode: {{ default .Env.HISTORY_ARCHIVAL_FILE_MODE \"\" }}\n        dirMode: {{ default .Env.HISTORY_ARCHIVAL_DIR_MODE \"\" }}\n  visibility:\n    status: {{ default .Env.VISIBILITY_ARCHIVAL_STATUS \"disabled\" }}\n    enableRead: {{ default .Env.VISIBILITY_ARCHIVAL_ENABLE_READ \"false\" }}\n    provider:\n      filestore:\n        fileMode: {{ default .Env.VISIBILITY_ARCHIVAL_FILE_MODE \"\" }}\n        dirMode: {{ default .Env.VISIBILITY_ARCHIVAL_DIR_MODE \"\" }}\n\ndomainDefaults:\n  archival:\n    history:\n      status: {{ default .Env.DOMAIN_DEFAULTS_HISTORY_ARCHIVAL_STATUS \"disabled\" }}\n      URI: {{ default .Env.DOMAIN_DEFAULTS_HISTORY_ARCHIVAL_URI \"\" }}\n    visibility:\n      status: {{ default .Env.DOMAIN_DEFAULTS_VISIBILITY_ARCHIVAL_STATUS \"disabled\" }}\n      URI: {{ default .Env.DOMAIN_DEFAULTS_VISIBILITY_ARCHIVAL_URI \"\" }}\n\nkafka:\n    tls:\n        enabled: false\n    clusters:\n        test:\n            brokers:\n                - {{ default .Env.KAFKA_SEEDS \"\" }}:{{ default .Env.KAFKA_PORT \"9092\" }}\n    topics:\n        {{ default .Env.VISIBILITY_NAME \"cadence-visibility-dev\" }}:\n            cluster: test\n        {{ default .Env.VISIBILITY_NAME \"cadence-visibility-dev\" }}-dlq:\n            cluster: test\n    applications:\n        visibility:\n            topic: {{ default .Env.VISIBILITY_NAME \"cadence-visibility-dev\" }}\n            dlq-topic: {{ default .Env.VISIBILITY_NAME \"cadence-visibility-dev\" }}-dlq\n\npublicClient:\n    {{- if .Env.IS_NOT_PRIMARY }}\n    hostPort: {{ default .Env.SECONDARY_FRONTEND_SERVICE \"cadence\" }}:{{ default .Env.FRONTEND_PORT \"7833\" }}\n    {{- else }}\n    hostPort: {{ default .Env.PRIMARY_FRONTEND_SERVICE \"cadence\" }}:{{ default .Env.FRONTEND_PORT \"7833\" }}\n    {{- end }}\n\ndynamicconfig:\n  client: filebased\n  filebased:\n    filepath: {{ default .Env.DYNAMIC_CONFIG_FILE_PATH \"/etc/cadence/config/dynamicconfig/development.yaml\" }}\n    pollInterval: \"60s\"\n\nblobstore:\n  filestore:\n    outputDirectory: {{ default .Env.FILE_BLOB_STORE_OUTPUT_DIRECTYORY \"\" }}\n\nauthorization:\n    oauthAuthorizer:\n        enable: {{ default .Env.ENABLE_OAUTH \"false\" }}\n        maxJwtTTL: {{ default .Env.OAUTH_MAX_JWT_TTL \"86400\" }}\n        jwtCredentials:\n            algorithm: \"RS256\"\n            publicKey: {{ default .Env.OAUTH_PUBLIC_KEY \"\" }}\n\n{{- if .Env.ASYNC_WF_KAFKA_QUEUE_ENABLED }}\nasyncWorkflowQueues:\n  queue1:\n    type: \"kafka\"\n    config:\n      connection:\n        brokers:\n          - {{ default .Env.KAFKA_SEEDS \"\" }}:{{ default .Env.KAFKA_PORT \"9092\" }}\n      topic: {{ default .Env.ASYNC_WF_KAFKA_QUEUE_TOPIC \"default-topic\" }}\n{{- end }}\n\nshard-distributor-matching:\n  namespaces:\n    - namespace: {{ default .Env.SHARD_DISTRIBUTOR_MATCHING_NAMESPACE \"cadence-matching\" }}\n      heartbeat_interval: {{ default .Env.SHARD_DISTRIBUTOR_MATCHING_HEARTBEAT_INTERVAL \"1s\" }}\n      migration_mode: {{ default .Env.SHARD_DISTRIBUTOR_MATCHING_MIGRATION_MODE \"local_pass\" }}\n      ttl_shard: {{ default .Env.SHARD_DISTRIBUTOR_MATCHING_TTL_SHARD \"5m\" }}\n      ttl_report: {{ default .Env.SHARD_DISTRIBUTOR_MATCHING_TTL_REPORT \"1m\" }}\n"
  },
  {
    "path": "docker/dev/cassandra-esv7-kafka.yml",
    "content": "version: '3'\nservices:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/dev/cassandra-opensearch-kafka-migration.yml",
    "content": "version: '3'\nservices:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n  opensearch:\n    image: opensearchproject/opensearch:2.13.0\n    environment:\n      - discovery.type=single-node\n      - OPENSEARCH_SECURITY_SSL_HTTP_ENABLED=false\n      - cluster.name=opensearch-cluster\n      - bootstrap.memory_lock=true\n      - \"OPENSEARCH_JAVA_OPTS=-Xms512m -Xmx512m\"\n      - OPENSEARCH_INITIAL_ADMIN_PASSWORD=DevTestInitial123!    # Sets the demo admin user password when using demo configuration, required for OpenSearch 2.12 and later\n    ports:\n      - 9200:9200 # REST API\n      - 9600:9600 # Performance Analyzer\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9201:9200\"\n    environment:\n      - discovery.type=single-node\n  opensearch-dashboards:\n    image: opensearchproject/opensearch-dashboards:2.13.0\n    container_name: opensearch-dashboards\n    ports:\n      - 5601:5601\n    expose:\n      - \"5601\" # Expose port 5601 for web access to OpenSearch Dashboards\n    environment:\n      OPENSEARCH_HOSTS: '[\"https://opensearch:9200\"]' # Define the OpenSearch nodes that OpenSearch Dashboards will query\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/dev/cassandra-opensearch-kafka.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n  opensearch:\n    image: opensearchproject/opensearch:2.13.0\n    environment:\n      - discovery.type=single-node\n      - OPENSEARCH_SECURITY_SSL_HTTP_ENABLED=false\n      - \"DISABLE_SECURITY_PLUGIN=true\"\n      - cluster.name=opensearch-cluster\n      - bootstrap.memory_lock=true\n      - \"OPENSEARCH_JAVA_OPTS=-Xms512m -Xmx512m\"\n      - OPENSEARCH_INITIAL_ADMIN_PASSWORD=DevTestInitial123!    # Sets the demo admin user password when using demo configuration, required for OpenSearch 2.12 and later\n    ports:\n      - 9200:9200 # REST API\n      - 9600:9600 # Performance Analyzer\n  opensearch-dashboards:\n    image: opensearchproject/opensearch-dashboards:2.13.0\n    container_name: opensearch-dashboards\n    ports:\n      - 5601:5601\n    expose:\n      - \"5601\" # Expose port 5601 for web access to OpenSearch Dashboards\n    environment:\n      OPENSEARCH_HOSTS: '[\"http://opensearch:9200\"]' # Define the OpenSearch nodes that OpenSearch Dashboards will query\n      DISABLE_SECURITY_DASHBOARDS_PLUGIN: 'true' # Disable security for dashboards\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/dev/cassandra-pinot-kafka.yml",
    "content": "version: '3'\nservices:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n  zookeeper:\n    image: zookeeper:3.5.8\n    container_name: zookeeper\n    ports:\n      - \"2181:2181\"\n    environment:\n      - ZOOKEEPER_CLIENT_PORT=2181\n      - ZOOKEEPER_TICK_TIME=2000\n  pinot-controller:\n    image: apachepinot/pinot:1.1.0\n    command: \"StartController -zkAddress zookeeper:2181 -controllerPort 9001\"\n    container_name: pinot-controller\n    restart: unless-stopped\n    ports:\n      - \"9001:9001\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms1G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-controller.log\"\n    depends_on:\n      - zookeeper\n  pinot-broker:\n    image: apachepinot/pinot:1.1.0\n    command: \"StartBroker -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-broker\"\n    ports:\n      - \"8099:8099\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-broker.log\"\n    depends_on:\n      - pinot-controller\n  pinot-server:\n    image: apachepinot/pinot:1.1.0\n    command: \"StartServer -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-server\"\n    ports:\n      - \"8098:8098\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx16G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-server.log\"\n    depends_on:\n      - pinot-broker\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    restart: unless-stopped\n    container_name: \"kafka\"\n    ports:\n      - \"9092:9092\"\n      - \"9093:9093\"\n    environment:\n      - KAFKA_CFG_ZOOKEEPER_CONNECT=zookeeper:2181\n      - KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=PLAINTEXT:PLAINTEXT,OUTSIDE:PLAINTEXT\n      - KAFKA_CFG_LISTENERS=PLAINTEXT://:9093,OUTSIDE://:9092\n      - KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9093,OUTSIDE://localhost:9092\n      - KAFKA_CFG_BROKER_ID=0\n      - ALLOW_PLAINTEXT_LISTENER=yes\n    depends_on:\n      - zookeeper\n"
  },
  {
    "path": "docker/dev/cassandra-testing/docker-compose-local-caas-cluster.yaml",
    "content": "networks:\n  cassandra-net:\n    driver: bridge\n\nservices:\n\n  cassandra-1:\n    image: \"cassandra:4.1.3\"  # cassandra:4.1.3\n    container_name: \"cassandra-1\"\n    ports:\n      - 7000:7000\n      - 9042:9042\n    networks:\n      - cassandra-net\n    environment:\n      - CASSANDRA_START_RPC=true       # default\n      - CASSANDRA_RPC_ADDRESS=0.0.0.0  # default\n      - CASSANDRA_LISTEN_ADDRESS=auto  # default, use IP addr of container # = CASSANDRA_BROADCAST_ADDRESS\n      - CASSANDRA_CLUSTER_NAME=my-cluster\n      - CASSANDRA_ENDPOINT_SNITCH=GossipingPropertyFileSnitch\n      - CASSANDRA_DC=my-datacenter-1\n    volumes:\n      - cassandra-node-1:/var/lib/cassandra:rw\n    healthcheck:\n      test: [\"CMD-SHELL\", \"nodetool status\", \";\", \"sleep\", \"40\"]\n      interval: 2m\n      start_period: 2m\n      timeout: 10s\n      retries: 3\n\n  cassandra-2:\n    image: \"cassandra:4.1.3\"  # cassandra:4.1.3\n    container_name: \"cassandra-2\"\n    ports:\n      - 9043:9042\n    networks:\n      - cassandra-net\n    environment:\n      - CASSANDRA_START_RPC=true       # default\n      - CASSANDRA_RPC_ADDRESS=0.0.0.0  # default\n      - CASSANDRA_LISTEN_ADDRESS=auto  # default, use IP addr of container # = CASSANDRA_BROADCAST_ADDRESS\n      - CASSANDRA_CLUSTER_NAME=my-cluster\n      - CASSANDRA_ENDPOINT_SNITCH=GossipingPropertyFileSnitch\n      - CASSANDRA_DC=my-datacenter-1\n      - CASSANDRA_SEEDS=cassandra-1\n    depends_on:\n      cassandra-1:\n        condition: service_healthy\n    volumes:\n      - cassandra-node-2:/var/lib/cassandra:rw\n    healthcheck:\n      test: [\"CMD-SHELL\", \"nodetool status\"]\n      interval: 2m\n      start_period: 2m\n      timeout: 10s\n      retries: 3\n\n  cassandra-3:\n    image: \"cassandra:4.1.3\"  # cassandra:4.1.3\n    container_name: \"cassandra-3\"\n    ports:\n      - 9044:9042\n    networks:\n      - cassandra-net\n    environment:\n      - CASSANDRA_START_RPC=true       # default\n      - CASSANDRA_RPC_ADDRESS=0.0.0.0  # default\n      - CASSANDRA_LISTEN_ADDRESS=auto  # default, use IP addr of container # = CASSANDRA_BROADCAST_ADDRESS\n      - CASSANDRA_CLUSTER_NAME=my-cluster\n      - CASSANDRA_ENDPOINT_SNITCH=GossipingPropertyFileSnitch\n      - CASSANDRA_DC=my-datacenter-1\n      - CASSANDRA_SEEDS=cassandra-1\n    depends_on:\n      cassandra-2:\n        condition: service_healthy\n    volumes:\n      - cassandra-node-3:/var/lib/cassandra:rw\n    healthcheck:\n      test: [\"CMD-SHELL\", \"nodetool status\"]\n      interval: 2m\n      start_period: 2m\n      timeout: 10s\n      retries: 3\n\nvolumes:\n  cassandra-node-1:\n  cassandra-node-2:\n  cassandra-node-3:\n\n\n"
  },
  {
    "path": "docker/dev/cassandra.yml",
    "content": "version: '3'\nservices:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n"
  },
  {
    "path": "docker/dev/mongo-esv7-kafka.yml",
    "content": "version: '3'\nservices:\n  mongo:\n    image: mongo:5\n    restart: always\n    ports:\n      - 27017:27017\n    environment:\n      MONGO_INITDB_ROOT_USERNAME: root\n      MONGO_INITDB_ROOT_PASSWORD: cadence\n\n  mongo-express:\n    image: mongo-express\n    restart: always\n    ports:\n      - 8081:8081\n    environment:\n      ME_CONFIG_MONGODB_ADMINUSERNAME: root\n      ME_CONFIG_MONGODB_ADMINPASSWORD: cadence\n      ME_CONFIG_MONGODB_URL: mongodb://root:cadence@mongo:27017/\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/dev/mysql-esv7-kafka.yml",
    "content": "version: '3'\nservices:\n  mysql:\n    image: mysql:8.0\n    ports:\n      - \"3306:3306\"\n    environment:\n      - \"MYSQL_ROOT_PASSWORD=cadence\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/dev/mysql.yml",
    "content": "version: '3'\nservices:\n  mysql:\n    image: mysql:8.0\n    ports:\n      - \"3306:3306\"\n    environment:\n      - \"MYSQL_ROOT_PASSWORD=cadence\""
  },
  {
    "path": "docker/dev/postgres.yml",
    "content": "version: '3'\nservices:\n  postgres:\n    image: postgres:17.4\n    environment:\n      POSTGRES_USER: postgres\n      POSTGRES_PASSWORD: cadence\n    ports:\n      - \"5432:5432\""
  },
  {
    "path": "docker/docker-compose-archival-filestore.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n     - \"7936:7936\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"FRONTEND_PPROF_PORT=7936\"\n      - \"LOG_LEVEL=debug\"\n      - \"HISTORY_ARCHIVAL_STATUS=enabled\"\n      - \"HISTORY_ARCHIVAL_ENABLE_READ=true\"\n      - \"HISTORY_ARCHIVAL_FILE_MODE=0100644\"\n      - \"HISTORY_ARCHIVAL_DIR_MODE=0040000\"\n      - \"DOMAIN_DEFAULTS_HISTORY_ARCHIVAL_STATUS=enabled\"\n      - \"DOMAIN_DEFAULTS_HISTORY_ARCHIVAL_URI=file:///etc/cadence/archival/history\"\n      - \"VISIBILITY_ARCHIVAL_STATUS=enabled\"\n      - \"VISIBILITY_ARCHIVAL_ENABLE_READ=true\"\n      - \"VISIBILITY_ARCHIVAL_FILE_MODE=0100644\"\n      - \"VISIBILITY_ARCHIVAL_DIR_MODE=0040000\"\n      - \"DOMAIN_DEFAULTS_VISIBILITY_ARCHIVAL_STATUS=enabled\"\n      - \"DOMAIN_DEFAULTS_VISIBILITY_ARCHIVAL_URI=file:///etc/cadence/archival/visibility\"\n    volumes:\n      - ./archival:/etc/cadence/archival\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-async-wf-kafka-v4.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n    environment:\n      - \"ASYNC_WF_KAFKA_QUEUE_ENABLED=true\"\n      - \"ASYNC_WF_KAFKA_QUEUE_TOPIC=async-wf-topic1\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"KAFKA_PORT=9092\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"LOG_LEVEL=debug\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n  kafka:\n    image: apache/kafka:4.0.0\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/docker-compose-async-wf-kafka.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n    environment:\n      - \"ASYNC_WF_KAFKA_QUEUE_ENABLED=true\"\n      - \"ASYNC_WF_KAFKA_QUEUE_TOPIC=async-wf-topic1\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"KAFKA_PORT=9092\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"LOG_LEVEL=debug\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n"
  },
  {
    "path": "docker/docker-compose-bench.yml",
    "content": "services:\n  cadence-bench:\n    image: ubercadence/cadence-bench:master\n    volumes:\n      - ./config/bench:/etc/cadence-bench/config/bench\n"
  },
  {
    "path": "docker/docker-compose-canary.yml",
    "content": "services:\n  cadence-canary:\n    image: ubercadence/cadence-canary:master\n    volumes:\n      - ./config/canary:/etc/cadence-canary/config/canary\n    environment:\n      - \"CADENCE_CANARY_MODE=all\" # this will run both worker and cron starter\n"
  },
  {
    "path": "docker/docker-compose-es-v7.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"ES_VERSION=v7\"\n      - \"KAFKA_SEEDS=kafka\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      elasticsearch:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-es.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:6.8.22\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      elasticsearch:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-http-api.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n     - \"8800:8800\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"FRONTEND_HTTP_PORT=8800\"\n      - \"FRONTEND_HTTP_PROCEDURES=uber.cadence.api.v1.WorkflowAPI::StartWorkflowExecution,uber.cadence.api.v1.WorkflowAPI::SignalWorkflowExecution,uber.cadence.api.v1.WorkflowAPI::QueryWorkflow,uber.cadence.api.v1.WorkflowAPI::DescribeWorkflowExecution,uber.cadence.api.v1.VisibilityAPI::ListWorkflowExecutions\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-multiclusters-cass-mysql-es.yaml",
    "content": "version: '3'\nservices:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  mysql:\n    image: mysql:8.0\n    ports:\n      - \"3306:3306\"\n    environment:\n      - \"MYSQL_ROOT_PASSWORD=root\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus_multiclusters:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:6.8.22\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_primary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_primary\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-primary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-primary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-primary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-primary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"VISIBILITY_NAME=cadence-visibility-primary\"\n      - \"PRIMARY_FRONTEND_SERVICE=cadence\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      elasticsearch:\n        condition: service_started\n  cadence-secondary:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"9001:9001\"\n      - \"9002:9002\"\n      - \"9003:9003\"\n      - \"9004:9004\"\n      - \"7943:7933\"\n      - \"7944:7934\"\n      - \"7945:7935\"\n      - \"7949:7939\"\n      - \"7843:7833\"\n    environment:\n      - \"DB=mysql\"\n      - \"MYSQL_USER=root\"\n      - \"MYSQL_PWD=root\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:9001\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:9002\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:9003\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:9004\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"IS_NOT_PRIMARY=true\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-secondary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-secondary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-secondary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-secondary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"VISIBILITY_NAME=cadence-visibility-secondary\"\n      - \"SECONDARY_FRONTEND_SERVICE=cadence-secondary\"\n    depends_on:\n      - mysql\n      - prometheus\n      - kafka\n      - elasticsearch\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833,cadence-secondary:7833\"\n      - \"CADENCE_GRPC_SERVICES_NAMES=cadence-frontend,cadence-frontend\"\n      - \"CADENCE_CLUSTERS_NAMES=cluster0,cluster1\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n      - cadence-secondary\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-multiclusters-es.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus_multiclusters:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:6.8.22\n    ports:\n      - \"9200:9200\"\n    environment:\n      - discovery.type=single-node\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_primary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_primary\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-primary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-primary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-primary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-primary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"VISIBILITY_NAME=cadence-visibility-primary\"\n      - \"PRIMARY_FRONTEND_SERVICE=cadence\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      elasticsearch:\n        condition: service_started\n  cadence-secondary:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"9001:9001\"\n      - \"9002:9002\"\n      - \"9003:9003\"\n      - \"9004:9004\"\n      - \"7943:7933\"\n      - \"7944:7934\"\n      - \"7945:7935\"\n      - \"7949:7939\"\n      - \"7843:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:9001\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:9002\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:9003\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:9004\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"IS_NOT_PRIMARY=true\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_secondary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_secondary\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-secondary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-secondary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-secondary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-secondary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"VISIBILITY_NAME=cadence-visibility-secondary\"\n      - \"SECONDARY_FRONTEND_SERVICE=cadence-secondary\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      elasticsearch:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833,cadence-secondary:7833\"\n      - \"CADENCE_GRPC_SERVICES_NAMES=cadence-frontend,cadence-frontend\"\n      - \"CADENCE_CLUSTERS_NAMES=cluster0,cluster1\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n      - cadence-secondary\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-multiclusters.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  # Setup schemas for primary and secondary clusters\n  cadence-schema-setup:\n    image: ubercadence/server:master-auto-setup\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"CASSANDRA_USER=cassandra\"\n      - \"CASSANDRA_PASSWORD=cassandra\"\n      - \"CASSANDRA_PROTO_VERSION=4\"\n      - \"RF=1\"\n      - \"PRIMARY_KEYSPACE=cadence_primary\"\n      - \"PRIMARY_VISIBILITY_KEYSPACE=cadence_visibility_primary\"\n      - \"SECONDARY_KEYSPACE=cadence_secondary\"\n      - \"SECONDARY_VISIBILITY_KEYSPACE=cadence_visibility_secondary\"\n    volumes:\n      - ./setup-multiclusters-schema.sh:/setup-multiclusters-schema.sh\n    command: [\"/setup-multiclusters-schema.sh\"]\n    restart: \"no\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus_multiclusters:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"BIND_ON_IP=0.0.0.0\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_primary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_primary\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-primary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-primary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-primary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-primary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      cadence-schema-setup:\n        condition: service_completed_successfully\n      prometheus:\n        condition: service_started\n  cadence-secondary:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"9001:9001\"\n      - \"9002:9002\"\n      - \"9003:9003\"\n      - \"9004:9004\"\n      - \"7943:7933\"\n      - \"7944:7934\"\n      - \"7945:7935\"\n      - \"7949:7939\"\n      - \"7843:7833\"\n    environment:\n      - \"BIND_ON_IP=0.0.0.0\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:9001\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:9002\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:9003\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:9004\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"IS_NOT_PRIMARY=true\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_secondary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_secondary\"\n      - \"STATSD_FRONTEND_PREFIX=cadence-frontend-secondary\"\n      - \"STATSD_MATCHING_PREFIX=cadence-matching-secondary\"\n      - \"STATSD_HISTORY_PREFIX=cadence-history-secondary\"\n      - \"STATSD_WORKER_PREFIX=cadence-worker-secondary\"\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      cadence-schema-setup:\n        condition: service_completed_successfully\n      prometheus:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833,cadence-secondary:7833\"\n      - \"CADENCE_GRPC_SERVICES_NAMES=cadence-frontend,cadence-frontend\"\n      - \"CADENCE_CLUSTERS_NAMES=cluster0,cluster1\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n      - cadence-secondary\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-mysql.yml",
    "content": "services:\n  mysql:\n    platform: linux/amd64\n    image: mysql:8.0\n    ports:\n      - \"3306:3306\"\n    environment:\n      - \"MYSQL_ROOT_PASSWORD=root\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"DB=mysql\"\n      - \"MYSQL_USER=root\"\n      - \"MYSQL_PWD=root\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n    depends_on:\n      - mysql\n      - prometheus\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-oauth.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"ENABLE_OAUTH=true\"\n      - \"OAUTH_PUBLIC_KEY=config/credentials/keytest.pub\"\n      - \"OAUTH_PRIVATE_KEY=config/credentials/keytest\"\n    depends_on:\n      - cassandra\n      - prometheus\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-opensearch.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  opensearch:\n    image: opensearchproject/opensearch:2.13.0\n    ports:\n      - \"9200:9200\"\n      - \"9600:9600\"\n    environment:\n      - discovery.type=single-node\n      - OPENSEARCH_SECURITY_SSL_HTTP_ENABLED=false\n      - DISABLE_SECURITY_PLUGIN=true\n      - cluster.name=opensearch-cluster\n      - bootstrap.memory_lock=true\n      - \"OPENSEARCH_JAVA_OPTS=-Xms512m -Xmx512m\"\n  opensearch-dashboards:\n    image: opensearchproject/opensearch-dashboards:2.13.0\n    ports:\n      - \"5601:5601\"\n    environment:\n      OPENSEARCH_HOSTS: '[\"http://opensearch:9200\"]'\n      DISABLE_SECURITY_DASHBOARDS_PLUGIN: \"true\"\n    depends_on:\n      - opensearch\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=opensearch\"\n      - \"ES_VERSION=os2\"\n      - \"KAFKA_SEEDS=kafka\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n      kafka:\n        condition: service_started\n      opensearch:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n\n"
  },
  {
    "path": "docker/docker-compose-pinot.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n  zookeeper:\n    image: zookeeper:3.5.8\n    container_name: zookeeper\n    ports:\n      - \"2181:2181\"\n    environment:\n      ZOOKEEPER_CLIENT_PORT: 2181\n      ZOOKEEPER_TICK_TIME: 2000\n  pinot-controller:\n    image: apachepinot/pinot:latest\n    command: \"StartController -zkAddress zookeeper:2181\"\n    container_name: pinot-controller\n    restart: unless-stopped\n    ports:\n      - \"9000:9000\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms1G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-controller.log\"\n    depends_on:\n      - zookeeper\n  pinot-broker:\n    image: apachepinot/pinot:latest\n    command: \"StartBroker -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-broker\"\n    ports:\n      - \"8099:8099\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-broker.log\"\n    depends_on:\n      - pinot-controller\n  pinot-server:\n    image: apachepinot/pinot:latest\n    command: \"StartServer -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-server\"\n    ports:\n      - \"8098:8098\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx16G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-server.log\"\n    depends_on:\n      - pinot-broker\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development_es.yaml\"\n      - \"ENABLE_ES=true\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n    depends_on:\n      - cassandra\n      - prometheus\n      - kafka\n      - elasticsearch\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-postgres.yml",
    "content": "services:\n  postgres:\n    image: postgres:17.4\n    environment:\n      POSTGRES_USER: cadence\n      POSTGRES_PASSWORD: cadence\n    ports:\n      - \"5432:5432\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"DB=postgres\"\n      - \"DB_PORT=5432\"\n      - \"POSTGRES_USER=cadence\"\n      - \"POSTGRES_PWD=cadence\"\n      - \"POSTGRES_SEEDS=postgres\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n    depends_on:\n      - postgres\n      - prometheus\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-scylla.yml",
    "content": "services:\n  scylla:\n    image: scylladb/scylla:4.3.1\n    command:\n      - \"--smp\"\n      - \"1\"\n    ports:\n      - \"9042:9042\"\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n      - \"8000:8000\"\n      - \"8001:8001\"\n      - \"8002:8002\"\n      - \"8003:8003\"\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n      - \"7833:7833\"\n    environment:\n      - \"DB=scylla\"\n      - \"CASSANDRA_SEEDS=scylla\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n    depends_on:\n      - scylla\n      - prometheus\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/docker-compose-statsd.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  statsd:\n    image: graphiteapp/graphite-statsd\n    ports:\n      - \"8080:80\"\n      - \"2003:2003\"\n      - \"8125:8125\"\n      - \"8126:8126\"\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"STATSD_ENDPOINT=statsd:8125\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      statsd:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n"
  },
  {
    "path": "docker/docker-compose.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n  node-exporter:\n    image: prom/node-exporter\n    ports:\n      - '9100:9100'\n  cadence:\n    image: ubercadence/server:master-auto-setup\n    ports:\n     - \"8000:8000\"\n     - \"8001:8001\"\n     - \"8002:8002\"\n     - \"8003:8003\"\n     - \"7933:7933\"\n     - \"7934:7934\"\n     - \"7935:7935\"\n     - \"7939:7939\"\n     - \"7833:7833\"\n     - \"7936:7936\"\n    environment:\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\"\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\"\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\"\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\"\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n      - \"FRONTEND_PPROF_PORT=7936\"\n      - \"LOG_LEVEL=debug\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence:7833\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence\n  grafana:\n    image: grafana/grafana\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n"
  },
  {
    "path": "docker/domain/cassandra.cql",
    "content": "-- Insert into domains table\nINSERT INTO domains (\n    id,\n    domain\n) VALUES (\n    123e4567-e89b-12d3-a456-426614174000, -- Replace with your UUID\n    {\n        id: 123e4567-e89b-12d3-a456-426614174000, -- Replace with your UUID\n        name: 'default',\n        status: 0, -- Registered\n        description: 'This is an example domain.',\n        data: {'key1': 'value1', 'key2': 'value2'},\n        owner_email: 'owner@example.com'\n    }\n) IF NOT EXISTS;\n\n-- Insert into domains_by_name_v2 table\nINSERT INTO domains_by_name_v2 (\n    domains_partition,\n    name,\n    domain,\n    config,\n    replication_config,\n    is_global_domain,\n    config_version,\n    failover_version,\n    failover_notification_version,\n    notification_version\n) VALUES (\n    0,\n    'default',\n    {\n        id: 123e4567-e89b-12d3-a456-426614174000, -- Replace with your UUID\n        name: 'default',\n        status: 0, -- Registered\n        description: 'This is an example domain.',\n        data: {'key1': 'value1', 'key2': 'value2'},\n        owner_email: 'owner@example.com'\n    },\n    {\n        retention: 7,\n        emit_metric: True,\n        history_archival_status: 0, -- Default to disabled\n        visibility_archival_status: 0 -- Default to disabled\n    },\n    {\n        active_cluster_name: 'cluster0',\n        clusters: [{cluster_name: 'cluster0'}]\n    },\n    True, -- is_global_domain\n    1,    -- config_version\n    0,    -- failover_version\n    0,    -- failover_notification_version\n    0     -- notification_version\n) IF NOT EXISTS;\n"
  },
  {
    "path": "docker/domain/mysql.sql",
    "content": "-- Insert into domains table\nINSERT INTO domains (\n    shard_id,\n    id,\n    name,\n    data,\n    data_encoding,\n    is_global\n) VALUES (\n    54321, -- Default shard_id\n    UNHEX(REPLACE(UUID(), '-', '')), -- Generate a 16-byte UUID\n    'default', -- Domain name\n    '{\"key1\":\"value1\",\"key2\":\"value2\"}', -- Example data as JSON\n    'json', -- Encoding type\n    1 -- Set to 1 for a global domain\n) ON DUPLICATE KEY UPDATE\n    name = VALUES(name);\n\n-- Insert into domain_metadata table\nINSERT INTO domain_metadata (\n    notification_version\n) VALUES (\n    1\n) ON DUPLICATE KEY UPDATE\n    notification_version = VALUES(notification_version);"
  },
  {
    "path": "docker/domain/postgres.sql",
    "content": "-- Insert into domains table\nINSERT INTO domains (\n    shard_id,\n    id,\n    name,\n    data,\n    data_encoding,\n    is_global\n) VALUES (\n    54321, -- Default shard_id\n    gen_random_uuid(), -- Generate a UUID for the domain ID\n    'default', -- Domain name\n    '{\"key1\":\"value1\",\"key2\":\"value2\"}', -- Example JSON data\n    'json', -- Encoding type\n    TRUE -- Set to TRUE for a global domain\n) ON CONFLICT (shard_id, id) DO UPDATE\nSET \n    name = EXCLUDED.name,\n    data = EXCLUDED.data,\n    data_encoding = EXCLUDED.data_encoding,\n    is_global = EXCLUDED.is_global;\n\n-- Insert into domain_metadata table\nINSERT INTO domain_metadata (\n    id,\n    notification_version\n) VALUES (\n    1, -- Default ID\n    1 -- Notification version\n) ON CONFLICT (id) DO UPDATE\nSET \n    notification_version = EXCLUDED.notification_version;"
  },
  {
    "path": "docker/entrypoint.sh",
    "content": "#!/bin/bash\n\nHOST_IP=`hostname -i`\nexport HOST_IP\n\nif [ \"$BIND_ON_LOCALHOST\" == true ] || [ \"$BIND_ON_IP\" == \"127.0.0.1\" ]; then\n    export BIND_ON_IP=\"127.0.0.1\"\n    export HOST_IP=\"127.0.0.1\"\nelif [ -z \"$BIND_ON_IP\" ]; then\n    # not binding to localhost and bind_on_ip is empty - use default host ip addr\n    export BIND_ON_IP=$HOST_IP\nelif [ \"$BIND_ON_IP\" != \"0.0.0.0\" ]; then\n    # binding to a user specified addr, make sure HOST_IP also uses the same addr\n    export HOST_IP=$BIND_ON_IP\nfi\n\n# this env variable is deprecated\nexport BIND_ON_LOCALHOST=false\n\nexec \"$@\"\n"
  },
  {
    "path": "docker/github_actions/Dockerfile",
    "content": "FROM golang:1.23.4-bullseye\n\n# Tried to set Python to ignore warnings due to the instructions at this link:\n# https://github.com/yaml/pyyaml/wiki/PyYAML-yaml.load(input)-Deprecation\n# But this causes all the pip installs to fail, so don't do this:\n# ENV PYTHONWARNINGS=ignore::yaml.YAMLLoadWarning\n# ENV PYTHONWARNINGS=ignore\n\nRUN apt-get update && apt-get install -y --no-install-recommends \\\n      curl \\\n      gettext-base \\\n      libyaml-dev \\\n      python3-pip \\\n      python-setuptools \\\n      time \\\n      unzip \\\n      ca-certificates \\\n    && rm -rf /var/lib/apt/lists/*\n\nRUN pip3 install cqlsh && cqlsh --version\n\n# verbose test output from `make`, can be disabled with V=0\nENV V=0\n\n# allow git-status and similar to work\nRUN git config --global --add safe.directory /cadence\n\n# https://github.com/docker-library/golang/blob/c1baf037d71331eb0b8d4c70cff4c29cf124c5e0/1.4/Dockerfile\nRUN mkdir -p /cadence\nWORKDIR /cadence\n\n# Copy go mod dependencies and try to share the module download cache\nCOPY go.* /cadence\nCOPY internal/tools/go.* /cadence/internal/tools/\nCOPY cmd/server/go.* /cadence/cmd/server/\nCOPY common/archiver/gcloud/go.* /cadence/common/archiver/gcloud/\n# go.work means this downloads everything, not just the top module\nRUN go mod download\n"
  },
  {
    "path": "docker/github_actions/README.md",
    "content": "# Running tests\n\n## Testing the build locally\nTo try out the build locally, start from the root folder of this repo \n(cadence) and run the following commands.\n\nunit tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml build unit-test\n```\n\nNOTE: You would expect TestServerStartup to fail here as we don't have a way to install the schema like we do in pipeline.yml \n\nintegration tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml build integration-test-cassandra\n```\n\ncross DC integration tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml build integration-test-ndc-cassandra\n```\n\nRun the integration tests:\n\nunit tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml run unit-test\n```\n\nintegration tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml run integration-test-cassandra\n```\n\ncross DC integration tests:\n```bash\ndocker compose -f docker/github_actions/docker-compose-local.yml run integration-test-ndc-cassandra\n```\n\n## Testing the build in Github Actions\nCreating a PR against the master branch will trigger the Github Actions\nbuild. Members of the Cadence team can view the build pipeline here on Github.\n\nEventually this pipeline should be made public. It will need to ignore \nthird party PRs for safety reasons.\n"
  },
  {
    "path": "docker/github_actions/docker-compose-cassandra-lwt.yml",
    "content": "services:\n  cass1:\n    container_name: cass1\n    hostname: cass1\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment: &environment    # Declare and save environments variables into \"environment\"\n      MAX_HEAP_SIZE: 256M\n      HEAP_NEWSIZE: 128M\n      CASSANDRA_SEEDS: \"cass1,cass2\"    # The first two nodes will be seeds\n      CASSANDRA_CLUSTER_NAME: SolarSystem\n      CASSANDRA_DC: Mars\n      CASSANDRA_RACK: West\n      CASSANDRA_ENDPOINT_SNITCH: GossipingPropertyFileSnitch\n      CASSANDRA_NUM_TOKENS: 128\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  cass2:\n    container_name: cass2\n    hostname: cass2\n    image: cassandra:4.1.1\n    ports:\n      - \"9043:9042\"\n    environment: *environment # point to \"environment\" to use the same environment variables as cass1\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  test-cass-lwt:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 180s\n        -run ^TestCassandraLWT$\n        -count 1\n        -v\n        -tags cassandralwt\n        github.com/uber/cadence/host\n        | tee test.log\n    ports:\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n    environment:\n      # - \"CASSANDRA_HOST=cass1\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cass1,cass2\"\n    depends_on:\n      cass1:\n        condition: service_healthy\n      cass2:\n        condition: service_healthy\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-es7.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n\n  integration-test-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ES_VERSION=v7\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-async-wf.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  integration-test-async-wf:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 60s\n        -run ^TestAsyncWFIntegrationSuite$\n        -tags asyncwfintegration\n        -count 1\n        -v\n        github.com/uber/cadence/host\n        | tee test.log\n    ports:\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n    environment:\n      - \"ASYNC_WF_KAFKA_QUEUE_TOPIC=async-wf-topic1\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"KAFKA_PORT=9092\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-es7.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    ports:\n      - \"9042:9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    ports:\n      - \"9200:9200\"\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n\n  integration-test-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - |\n        make cover_integration_profile\n    ports:\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ES_VERSION=v7\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-history-simulation.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    ports:\n      - \"9042:9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:v3.0.1\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/history_simulation_prometheus.yml'\n    ports:\n      - '9090:9090'\n    networks:\n      services-network:\n        aliases:\n          - prometheus\n  grafana:\n    image: grafana/grafana:11.4.0\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n    networks:\n      services-network:\n        aliases:\n          - grafana\n  history-simulator:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 300s\n        -run ^TestHistorySimulation.*$\n        -count 1\n        -v\n        github.com/uber/cadence/simulation/history\n        | tee test.log\n    environment:\n      - \"HISTORY_LOG_EVENTS=true\"\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n    depends_on:\n      prometheus:\n        condition: service_started\n      grafana:\n        condition: service_started\n      cassandra:\n        condition: service_healthy\n    ports: # expose prometheus ports so they can be scraped\n      - '8306:8306'\n      - '8307:8307'\n      - '8308:8308'\n      - '8309:8309'\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - cadence\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-matching-simulation.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    expose:\n      - \"9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:v3.0.1\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/matching_simulation_prometheus.yml'\n    ports:\n      - '9090:9090'\n    networks:\n      services-network:\n        aliases:\n          - prometheus\n  grafana:\n    image: grafana/grafana:11.4.0\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n    networks:\n      services-network:\n        aliases:\n          - grafana\n\n  matching-simulator:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 180s\n        -run ^TestMatchingSimulation.*$\n        -count 1\n        -v\n        github.com/uber/cadence/simulation/matching\n        | tee test.log\n    environment:\n      - \"MATCHING_LOG_EVENTS=true\"\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n    depends_on:\n      prometheus:\n        condition: service_started\n      grafana:\n        condition: service_started\n      cassandra:\n        condition: service_healthy\n    ports: # expose prometheus ports so they can be scraped\n      - '8306:8306'\n      - '8307:8307'\n      - '8308:8308'\n      - '8309:8309'\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - cadence\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-pinot.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    networks:\n      services-network:\n        aliases:\n          - prometheus\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    restart: unless-stopped\n    container_name: \"kafka\"\n    ports:\n      - \"9092:9092\"\n      - \"9093:9093\"\n    environment:\n      - KAFKA_CFG_ZOOKEEPER_CONNECT=zookeeper:2181\n      - KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=PLAINTEXT:PLAINTEXT,OUTSIDE:PLAINTEXT\n      - KAFKA_CFG_LISTENERS=PLAINTEXT://:9093,OUTSIDE://:9092\n      - KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9093,OUTSIDE://kafka:9092\n      - KAFKA_CFG_BROKER_ID=0\n      - ALLOW_PLAINTEXT_LISTENER=yes\n      # Topic settings\n      - KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\n    depends_on:\n      - zookeeper\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  # will be deleted later when we get rid of ES usages\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    # version here must be the same as the one in testdata/integration_pinot_cluster\n    ports:\n      - \"9200:9200\"\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n\n  zookeeper:\n    image: zookeeper:3.5.8\n    restart: always\n    hostname: zookeeper\n    container_name: zookeeper\n    ports:\n      - '2181:2181'\n    environment:\n      - ZOOKEEPER_CLIENT_PORT=2181\n      - ZOOKEEPER_TICK_TIME=2000\n    networks:\n      services-network:\n        aliases:\n          - zookeeper\n\n  pinot-controller:\n    image: apachepinot/pinot:latest\n    command: \"StartController -zkAddress zookeeper:2181 -controllerPort 9001\"\n    container_name: pinot-controller\n    restart: unless-stopped\n    ports:\n      - \"9001:9001\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms1G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-controller.log\"\n    depends_on:\n      - zookeeper\n    networks:\n      services-network:\n        aliases:\n          - pinot-controller\n  pinot-broker:\n    image: apachepinot/pinot:latest\n    command: \"StartBroker -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-broker\"\n    ports:\n      - \"8099:8099\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-broker.log\"\n    depends_on:\n      - pinot-controller\n    networks:\n      services-network:\n        aliases:\n          - pinot-broker\n  pinot-server:\n    image: apachepinot/pinot:latest\n    command: \"StartServer -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-server\"\n    ports:\n      - \"8098:8098\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx16G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-server.log\"\n    depends_on:\n      - pinot-broker\n    networks:\n      services-network:\n        aliases:\n          - pinot-server\n\n  pinot-admin:\n    image: apachepinot/pinot:latest\n    container_name: pinot-admin\n    depends_on:\n      - pinot-controller\n      - pinot-broker\n      - pinot-server\n    entrypoint: [\"/bin/sh\", \"-c\", \"cp /schema/pinot/create_pinot_table.sh /opt/pinot/create_pinot_table.sh && chmod +x /opt/pinot/create_pinot_table.sh && /opt/pinot/create_pinot_table.sh\"]\n    volumes:\n      - ../../schema:/schema  # Ensure the schema files and script are available in the container\n    networks:\n      services-network:\n        aliases:\n          - pinot-server\n\n  integration-test-cassandra-pinot:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 600s\n        -run ^TestPinotIntegrationSuite$\n        -tags pinotintegration\n        -count 1\n        -v\n        github.com/uber/cadence/host\n        | tee test.log\n    ports:\n      - \"7933:7933\"\n      - \"7934:7934\"\n      - \"7935:7935\"\n      - \"7939:7939\"\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"TEST_TAG=pinotintegration\"\n      - \"ES_VERSION=v7\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PINOT_SEEDS=pinot-broker\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n      pinot-controller:\n        condition: service_started\n      pinot-broker:\n        condition: service_started\n      pinot-server:\n        condition: service_started\n      pinot-admin:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    enable_ipv6: true\n    ipam:\n      config:\n        - subnet: 2001:0DB8::/112\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local-replication-simulation.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    ports:\n      - \"9042:9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  prometheus:\n    image: prom/prometheus:v3.0.1\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/replication_simulation_prometheus.yml'\n    ports:\n      - '9090:9090'\n    networks:\n      services-network:\n        aliases:\n          - prometheus\n\n  grafana:\n    image: grafana/grafana:11.4.0\n    volumes:\n      - ./grafana:/etc/grafana\n    user: \"1000\"\n    depends_on:\n      - prometheus\n    ports:\n      - '3000:3000'\n    networks:\n      services-network:\n        aliases:\n          - grafana\n\n  cadence-cluster0:\n    build:\n      context: ../../\n      dockerfile: ./Dockerfile${DOCKERFILE_SUFFIX}\n      args:\n        TARGET: auto-setup\n    command:\n      - /start.sh\n    ports:\n      - \"7933:7933\" # frontend thrift\n      - \"7833:7833\" # frontend grpc\n      - \"7934:7934\" # history thrift\n      - \"7834:7834\" # history grpc\n      - \"7935:7935\" # matching thrift\n      - \"7835:7835\" # matching grpc\n      - \"7939:7939\" # worker thrift\n      - \"7000:7000\" # frontend prometheus\n      - \"7001:7001\" # matching prometheus\n      - \"7002:7002\" # history prometheus\n      - \"7003:7003\" # worker prometheus\n    environment:\n      - DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/replication_simulation_${SCENARIO}.yml\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"BIND_ON_IP=0.0.0.0\"\n      - \"PRIMARY_FRONTEND_SERVICE=cadence-cluster0\"\n      - \"SECONDARY_FRONTEND_SERVICE=cadence-cluster1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_primary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_primary\"\n      - \"LOG_LEVEL=debug\"\n      - \"MATCHING_LOG_EVENTS=true\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:7000\" # frontend scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:7001\" # matching scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:7002\" # history scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:7003\" # worker scrape endpoint\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n    networks:\n      services-network:\n        aliases:\n          - cadence-cluster0\n\n  cadence-cluster1:\n    build:\n      context: ../../\n      dockerfile: ./Dockerfile${DOCKERFILE_SUFFIX}\n      args:\n        TARGET: auto-setup\n    command:\n      - /start.sh\n    ports: # cluster1 uses 8xxx host ports to avoid conflicts with cluster0\n      - \"8933:7933\" # frontend thrift\n      - \"8833:7833\" # frontend grpc\n      - \"8934:7934\" # history thrift\n      - \"8834:7834\" # history grpc\n      - \"8935:7935\" # matching thrift\n      - \"8835:7835\" # matching grpc\n      - \"8939:7939\" # worker thrift\n      - \"8000:8000\" # frontend prometheus\n      - \"8001:8001\" # matching prometheus\n      - \"8002:8002\" # history prometheus\n      - \"8003:8003\" # worker prometheus\n    environment:\n      - DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/replication_simulation_${SCENARIO}.yml\n      - \"CLUSTER_REDIRECT_POLICY=selected-apis-forwarding\"\n      - \"BIND_ON_IP=0.0.0.0\"\n      - \"PRIMARY_FRONTEND_SERVICE=cadence-cluster0\"\n      - \"SECONDARY_FRONTEND_SERVICE=cadence-cluster1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"IS_NOT_PRIMARY=true\"\n      - \"ENABLE_GLOBAL_DOMAIN=true\"\n      - \"KEYSPACE=cadence_secondary\"\n      - \"VISIBILITY_KEYSPACE=cadence_visibility_secondary\"\n      - \"LOG_LEVEL=debug\"\n      - \"MATCHING_LOG_EVENTS=true\"\n      - \"PROMETHEUS_ENDPOINT_0=0.0.0.0:8000\" # frontend scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_1=0.0.0.0:8001\" # matching scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_2=0.0.0.0:8002\" # history scrape endpoint\n      - \"PROMETHEUS_ENDPOINT_3=0.0.0.0:8003\" # worker scrape endpoint\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      prometheus:\n        condition: service_started\n    networks:\n      services-network:\n        aliases:\n          - cadence-cluster1\n\n  cadence-web:\n    image: ubercadence/web:latest\n    environment:\n      - \"CADENCE_GRPC_PEERS=cadence-cluster0:7833,cadence-cluster1:7833\"\n      - \"CADENCE_GRPC_SERVICES_NAMES=cadence-frontend,cadence-frontend\"\n      - \"CADENCE_CLUSTERS_NAMES=cluster0,cluster1\"\n      - \"CADENCE_HISTORY_PAGE_V2_ENABLED=OPT_OUT\"\n    ports:\n      - \"8088:8088\"\n    depends_on:\n      - cadence-cluster0\n      - cadence-cluster1\n    networks:\n      services-network:\n        aliases:\n          - cadence-web\n\n  cadence-worker0:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    working_dir: /cadence/simulation/replication/worker/cmd\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go run *.go --cluster cluster0 | tee worker0.log\n    environment:\n      - REPLICATION_SIMULATION_CONFIG=testdata/replication_simulation_${SCENARIO}.yaml\n    depends_on:\n      cadence-cluster0:\n        condition: service_started\n      cadence-cluster1:\n        condition: service_started\n    healthcheck:\n      test: [\"CMD\", \"curl\", \"-f\", \"http://localhost:6060/health\"]\n      interval: 10s\n      timeout: 2s\n      retries: 20\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n      - ../../simulation/replication/testdata/:/cadence/simulation/replication/worker/cmd/testdata/\n    networks:\n      services-network:\n        aliases:\n          - cadence-worker0\n\n  cadence-worker1:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    working_dir: /cadence/simulation/replication/worker/cmd\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go run *.go --cluster cluster1 | tee worker1.log\n    environment:\n      - REPLICATION_SIMULATION_CONFIG=testdata/replication_simulation_${SCENARIO}.yaml\n    depends_on:\n      cadence-cluster0:\n        condition: service_started\n      cadence-cluster1:\n        condition: service_started\n    healthcheck:\n      test: [\"CMD\", \"curl\", \"-f\", \"http://localhost:6060/health\"]\n      interval: 10s\n      timeout: 2s\n      retries: 20\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n      - ../../simulation/replication/testdata/:/cadence/simulation/replication/worker/cmd/testdata/\n    networks:\n      services-network:\n        aliases:\n          - cadence-worker1\n\n  replication-simulator:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command:\n      - /bin/sh\n      - -e\n      - -c\n      - >\n        go test -timeout 240s\n        -run ^TestReplicationSimulation.*$\n        -count 1\n        -v\n        github.com/uber/cadence/simulation/replication\n        | tee test.log\n    depends_on:\n      cadence-cluster0:\n        condition: service_started\n      cadence-worker0:\n        condition: service_started\n      cadence-cluster1:\n        condition: service_started\n      cadence-worker1:\n        condition: service_started\n      cadence-web:\n        condition: service_started\n      grafana:\n        condition: service_started\n    ports: # expose prometheus ports so they can be scraped\n      - '8306:8306'\n      - '8307:8307'\n      - '8308:8308'\n      - '8309:8309'\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - replication-simulator\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-local.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    expose:\n      - \"9042\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [ \"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\" ]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  mysql:\n    image: mysql:8.0\n    environment:\n      MYSQL_DATABASE: db\n      MYSQL_ROOT_PASSWORD: cadence\n    volumes:\n      - ./mysql-init:/docker-entrypoint-initdb.d\n    expose:\n      - \"3306\"\n    networks:\n      services-network:\n        aliases:\n          - mysql\n\n  postgres:\n    image: postgres:17.4\n    environment:\n      POSTGRES_PASSWORD: cadence\n      POSTGRES_USER: cadence\n    expose:\n      - \"5432\"\n    networks:\n      services-network:\n        aliases:\n          - postgres\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:6.8.22\n    expose:\n      - \"9200\"\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n\n  mongo:\n    image: mongo:5\n    restart: always\n    networks:\n      services-network:\n        aliases:\n          - mongo\n    environment:\n      MONGO_INITDB_ROOT_USERNAME: root\n      MONGO_INITDB_ROOT_PASSWORD: cadence\n\n  unit-test:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: sh -c \"make .just-build && make cover_profile\"\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - unit-test\n\n  integration-test-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_integration_profile\n    environment:\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-cassandra-queue-v2-alert:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_integration_profile\n    environment:\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ENABLE_QUEUE_V2=true\"\n      - \"ENABLE_QUEUE_V2_ALERT=true\"\n      - \"V=1\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-mysql:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_integration_profile\n    environment:\n      - \"MYSQL=1\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"PERSISTENCE_PLUGIN=mysql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - mysql\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-postgres:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_integration_profile\n    environment:\n      - \"POSTGRES=1\"\n      - \"POSTGRES_SEEDS=postgres\"\n      - \"PERSISTENCE_PLUGIN=postgres\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - postgres\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-v2:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_integration_profile EVENTSV2=true\n    environment:\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-ndc-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_ndc_profile\n    environment:\n      - \"CASSANDRA_HOST=cassandra\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test-ndc\n\n  integration-test-ndc-mysql:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    command: make cover_ndc_profile\n    environment:\n      - \"MYSQL=1\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"PERSISTENCE_PLUGIN=mysql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - mysql\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n      - /cadence/.build/ # ensure we don't mount the build directory\n      - /cadence/.bin/ # ensure we don't mount the bin directory\n    networks:\n      services-network:\n        aliases:\n          - integration-test-ndc\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-opensearch2.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  elasticsearch:\n    image: opensearchproject/opensearch:2.13.0\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n      - OPENSEARCH_SECURITY_SSL_HTTP_ENABLED=false\n      - \"DISABLE_SECURITY_PLUGIN=true\"\n      - cluster.name=opensearch-cluster\n      - bootstrap.memory_lock=true\n      - \"OPENSEARCH_JAVA_OPTS=-Xms512m -Xmx512m\"\n      - OPENSEARCH_INITIAL_ADMIN_PASSWORD=DevTestInitial123!    # Sets the demo admin user password when using demo configuration, required for OpenSearch 2.12 and later\n    ports:\n      - 9200:9200 # REST API\n      - 9600:9600 # Performance Analyzer\n\n  integration-test-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ES_VERSION=os2\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose-pinot.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    ports:\n      - \"9042:9042\"\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [\"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n  prometheus:\n    image: prom/prometheus:latest\n    networks:\n      services-network:\n        aliases:\n          - prometheus\n    volumes:\n      - ./prometheus:/etc/prometheus\n    command:\n      - '--config.file=/etc/prometheus/prometheus.yml'\n    ports:\n      - '9090:9090'\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    restart: unless-stopped\n    container_name: \"kafka\"\n    ports:\n      - \"9092:9092\"\n      - \"9093:9093\"\n    environment:\n      - KAFKA_CFG_ZOOKEEPER_CONNECT=zookeeper:2181\n      - KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=PLAINTEXT:PLAINTEXT,OUTSIDE:PLAINTEXT\n      - KAFKA_CFG_LISTENERS=PLAINTEXT://:9093,OUTSIDE://:9092\n      - KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9093,OUTSIDE://kafka:9092\n      - KAFKA_CFG_BROKER_ID=0\n      - ALLOW_PLAINTEXT_LISTENER=yes\n      # Topic settings\n      - KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\n    depends_on:\n      - zookeeper\n    networks:\n      services-network:\n        aliases:\n          - kafka\n\n  # will be deleted later when we get rid of ES usages\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:7.9.3\n    # version here must be the same as the one in testdata/integration_pinot_cluster\n    ports:\n      - \"9200:9200\"\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n\n  zookeeper:\n    image: zookeeper:3.5.8\n    restart: always\n    hostname: zookeeper\n    container_name: zookeeper\n    ports:\n      - '2181:2181'\n    environment:\n      - ZOOKEEPER_CLIENT_PORT=2181\n      - ZOOKEEPER_TICK_TIME=2000\n    networks:\n      services-network:\n        aliases:\n          - zookeeper\n  # Needs a new pinot version to run the new json match query\n  pinot-controller:\n    image: apachepinot/pinot:latest\n    command: \"StartController -zkAddress zookeeper:2181 -controllerPort 9001\"\n    container_name: pinot-controller\n    restart: unless-stopped\n    ports:\n      - \"9001:9001\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms1G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-controller.log\"\n    depends_on:\n      - zookeeper\n    networks:\n      services-network:\n        aliases:\n          - pinot-controller\n  pinot-broker:\n    image: apachepinot/pinot:latest\n    command: \"StartBroker -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-broker\"\n    ports:\n      - \"8099:8099\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx4G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-broker.log\"\n    depends_on:\n      - pinot-controller\n    networks:\n      services-network:\n        aliases:\n          - pinot-broker\n  pinot-server:\n    image: apachepinot/pinot:latest\n    command: \"StartServer -zkAddress zookeeper:2181\"\n    restart: unless-stopped\n    container_name: \"pinot-server\"\n    ports:\n      - \"8098:8098\"\n    environment:\n      JAVA_OPTS: \"-Dplugins.dir=/opt/pinot/plugins -Xms4G -Xmx16G -XX:+UseG1GC -XX:MaxGCPauseMillis=200 -Xloggc:gc-pinot-server.log\"\n    depends_on:\n      - pinot-broker\n    networks:\n      services-network:\n        aliases:\n          - pinot-server\n\n  pinot-admin:\n    image: apachepinot/pinot:latest\n    container_name: pinot-admin\n    depends_on:\n      - pinot-controller\n      - pinot-broker\n      - pinot-server\n    entrypoint: [\"/bin/sh\", \"-c\", \"cp /schema/pinot/create_pinot_table.sh /opt/pinot/create_pinot_table.sh && chmod +x /opt/pinot/create_pinot_table.sh && /opt/pinot/create_pinot_table.sh\"]\n    volumes:\n      - ../../schema:/schema  # Ensure the schema files and script are available in the container\n    networks:\n      services-network:\n        aliases:\n          - pinot-server\n\n  integration-test-cassandra-pinot:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"TEST_TAG=pinotintegration\"\n      - \"ES_VERSION=v7\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PINOT_SEEDS=pinot-broker\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n      pinot-controller:\n        condition: service_started\n      pinot-broker:\n        condition: service_started\n      pinot-server:\n        condition: service_started\n      pinot-admin:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\nnetworks:\n  services-network:\n    enable_ipv6: true\n    ipam:\n      config:\n        - subnet: 2001:0DB8::/112\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/docker-compose.yml",
    "content": "services:\n  cassandra:\n    image: cassandra:4.1.1\n    environment:\n      - \"MAX_HEAP_SIZE=256M\"\n      - \"HEAP_NEWSIZE=128M\"\n    networks:\n      services-network:\n        aliases:\n          - cassandra\n    healthcheck:\n      test: [ \"CMD\", \"cqlsh\", \"-u cassandra\", \"-p cassandra\" ,\"-e describe keyspaces\" ]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  mysql:\n    image: mysql:8.0\n    environment:\n      - \"MYSQL_ROOT_PASSWORD=cadence\"\n    networks:\n      services-network:\n        aliases:\n          - mysql\n\n  postgres:\n    image: postgres:17.4\n    environment:\n      POSTGRES_PASSWORD: cadence\n    ports:\n      - \"5432:5432\"\n    networks:\n      services-network:\n        aliases:\n          - postgres\n\n  kafka:\n    image: docker.io/bitnamilegacy/kafka:3.7\n    hostname: kafka\n    container_name: kafka\n    ports:\n      - \"9092:9092\"\n    environment:\n      # KRaft settings\n      - \"KAFKA_CFG_NODE_ID=0\"\n      - \"KAFKA_CFG_PROCESS_ROLES=controller,broker\"\n      - \"KAFKA_CFG_CONTROLLER_QUORUM_VOTERS=0@kafka:9093\"\n      # Listeners\n      - \"KAFKA_CFG_LISTENERS=PLAINTEXT://:9092,CONTROLLER://:9093\"\n      - \"KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://kafka:9092\"\n      - \"KAFKA_CFG_LISTENER_SECURITY_PROTOCOL_MAP=CONTROLLER:PLAINTEXT,PLAINTEXT:PLAINTEXT\"\n      - \"KAFKA_CFG_CONTROLLER_LISTENER_NAMES=CONTROLLER\"\n      - \"KAFKA_CFG_INTER_BROKER_LISTENER_NAME=PLAINTEXT\"\n      # Topic settings\n      - \"KAFKA_CFG_AUTO_CREATE_TOPICS_ENABLE=true\"\n    networks:\n      services-network:\n        aliases:\n          - kafka\n    healthcheck:\n      test: [\"CMD-SHELL\", \"kafka-topics.sh --bootstrap-server localhost:9092 --list\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  elasticsearch:\n    image: docker.elastic.co/elasticsearch/elasticsearch-oss:6.8.22\n    networks:\n      services-network:\n        aliases:\n          - elasticsearch\n    environment:\n      - discovery.type=single-node\n    healthcheck:\n      test: [\"CMD-SHELL\", \"curl -s http://localhost:9200/_cluster/health | grep -q '\\\"status\\\":\\\"green\\\"\\\\|\\\"status\\\":\\\"yellow\\\"'\"]\n      interval: 15s\n      timeout: 30s\n      retries: 10\n\n  mongo:\n    image: mongo:5\n    restart: always\n    networks:\n      services-network:\n        aliases:\n          - mongo\n    environment:\n      MONGO_INITDB_ROOT_USERNAME: root\n      MONGO_INITDB_ROOT_PASSWORD: cadence\n\n  etcd:\n    image: bitnamilegacy/etcd:3.5.5\n    restart: always\n    networks:\n      services-network:\n        aliases:\n          - etcd\n    environment:\n      ALLOW_NONE_AUTHENTICATION: \"yes\"\n      ETCD_ADVERTISE_CLIENT_URLS: \"http://etcd:2379\"\n\n  unit-test:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - unit-test\n\n  integration-test-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-cassandra-queue-v2:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ENABLE_QUEUE_V2=true\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-cassandra-queue-v2-alert:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n      - \"ENABLE_QUEUE_V2=true\"\n      - \"ENABLE_QUEUE_V2_ALERT=true\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-mysql:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"MYSQL=1\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"PERSISTENCE_PLUGIN=mysql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - mysql\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-postgres:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"POSTGRES=1\"\n      - \"POSTGRES_SEEDS=postgres\"\n      - \"PERSISTENCE_PLUGIN=postgres\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - postgres\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-sqlite:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"PERSISTENCE_PLUGIN=sqlite\"\n      - \"PERSISTENCE_TYPE=sql\"\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test\n\n  integration-test-ndc-cassandra:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      elasticsearch:\n        condition: service_started\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test-ndc\n\n  integration-test-ndc-mysql:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"MYSQL=1\"\n      - \"MYSQL_SEEDS=mysql\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"PERSISTENCE_PLUGIN=mysql\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - mysql\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test-ndc\n\n  integration-test-ndc-postgres:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"POSTGRES=1\"\n      - \"POSTGRES_SEEDS=postgres\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"PERSISTENCE_TYPE=sql\"\n      - \"PERSISTENCE_PLUGIN=postgres\"\n      - \"TEST_TAG=esintegration\"\n    depends_on:\n      - postgres\n      - elasticsearch\n      - kafka\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test-ndc\n\n  integration-test-async-wf:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"ASYNC_WF_KAFKA_QUEUE_TOPIC=async-wf-topic1\"\n      - \"CASSANDRA=1\"\n      - \"CASSANDRA_SEEDS=cassandra\"\n      - \"ES_SEEDS=elasticsearch\"\n      - \"KAFKA_SEEDS=kafka\"\n      - \"KAFKA_PORT=9092\"\n    depends_on:\n      cassandra:\n        condition: service_healthy\n      kafka:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test-async-wf\n\n  integration-test-with-etcd:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - \"ETCD=1\"\n      - \"ETCD_ENDPOINTS=http://etcd:2379\"\n    depends_on:\n      etcd:\n        condition: service_started\n    volumes:\n      - ../../:/cadence\n    networks:\n      services-network:\n        aliases:\n          - integration-test-with-etcd\n\n  coverage-report:\n    build:\n      context: ../../\n      dockerfile: ./docker/github_actions/Dockerfile${DOCKERFILE_SUFFIX}\n    environment:\n      - CI\n      - GITHUB_BRANCH\n      - GITHUB_RUN_URL\n      - GITHUB_REPO\n      - GITHUB_RUN_CREATOR\n      - GITHUB_RUN_CREATOR_EMAIL\n      - GITHUB_RUN_CREATOR_TEAMS\n      - GITHUB_PULL_REQUEST_REPO\n    volumes:\n      - ../../:/cadence\n\nnetworks:\n  services-network:\n    name: services-network\n    driver: bridge\n"
  },
  {
    "path": "docker/github_actions/grafana/grafana.ini",
    "content": "[auth.anonymous]\nenabled = true\norg_role = Admin"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-archival.json",
    "content": "{\n    \"annotations\": {\n      \"list\": [\n        {\n          \"builtIn\": 1,\n          \"datasource\": {\n            \"type\": \"grafana\",\n            \"uid\": \"-- Grafana --\"\n          },\n          \"enable\": true,\n          \"hide\": true,\n          \"iconColor\": \"rgba(0, 211, 255, 1)\",\n          \"name\": \"Annotations & Alerts\",\n          \"type\": \"dashboard\"\n        }\n      ]\n    },\n    \"editable\": true,\n    \"fiscalYearStartMonth\": 0,\n    \"graphTooltip\": 0,\n    \"id\": 6,\n    \"links\": [],\n    \"panels\": [\n      {\n        \"collapsed\": false,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 0\n        },\n        \"id\": 8,\n        \"panels\": [],\n        \"title\": \"Overall\",\n        \"type\": \"row\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"history\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 1\n        },\n        \"id\": 1,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": true,\n            \"legendFormat\": \"history\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (rate(archiver_client_visibility_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"visibility\",\n            \"range\": true,\n            \"refId\": \"B\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Archivals Triggered Per Second\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"archiver_client_visibility_inline_archive_attempt\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 1\n        },\n        \"id\": 2,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": true,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_request\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": true,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"B\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_visibility_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_visibility_request\",\n            \"range\": true,\n            \"refId\": \"C\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_visibility_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_visibility_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"D\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Client Requests vs Inline Attempts\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"archiver_client_history_request\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 9\n        },\n        \"id\": 3,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"archiver_client_history_request\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Archiver History Requests\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 9\n        },\n        \"id\": 4,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_history_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Client History Inline Attemptss\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 17\n        },\n        \"id\": 5,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(domain, operation) (rate(archiver_client_history_request_per_domain{domain=\\\"$domain\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"expr\": \"\",\n            \"hide\": false,\n            \"instant\": false,\n            \"range\": true,\n            \"refId\": \"B\"\n          }\n        ],\n        \"title\": \"History Requests Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 17\n        },\n        \"id\": 6,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(domain, operation) (rate(archiver_client_history_inline_archive_attempt_per_domain[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"History Inline Attempt Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 25\n        },\n        \"id\": 7,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(workflow_cleanup_archive[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Workflow Cleanup Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 25\n        },\n        \"id\": 9,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"expr\": \"\",\n            \"hide\": false,\n            \"instant\": false,\n            \"range\": true,\n            \"refId\": \"A\"\n          }\n        ],\n        \"title\": \"History Inline Attempt Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      }\n    ],\n    \"preload\": false,\n    \"schemaVersion\": 41,\n    \"tags\": [],\n    \"templating\": {\n      \"list\": [\n        {\n          \"allowCustomValue\": false,\n          \"current\": {\n            \"text\": \"\",\n            \"value\": \"\"\n          },\n          \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"includeAll\": false,\n          \"label\": \"Domain\",\n          \"name\": \"domain\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"0.99\",\n            \"value\": \"0.99\"\n          },\n          \"label\": \"Quantiles\",\n          \"name\": \"quantiles\",\n          \"options\": [\n            {\n              \"selected\": false,\n              \"text\": \"0.5\",\n              \"value\": \"0.5\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"0.95\",\n              \"value\": \"0.95\"\n            },\n            {\n              \"selected\": true,\n              \"text\": \"0.99\",\n              \"value\": \"0.99\"\n            }\n          ],\n          \"query\": \"0.5, 0.95, 0.99\",\n          \"type\": \"custom\"\n        }\n      ]\n    },\n    \"time\": {\n      \"from\": \"now-6h\",\n      \"to\": \"now\"\n    },\n    \"timepicker\": {},\n    \"timezone\": \"\",\n    \"title\": \"Cadence-Archival\",\n    \"uid\": \"1B0efRyGx\",\n    \"version\": 1\n  }"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-client-overall.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 2,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 10,\n      \"panels\": [],\n      \"title\": \"Overview\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 1\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"max by(active_cluster) (active_cluster{cadence_service=\\\"cadence_history\\\", operation=\\\"DomainCache\\\"})\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Active Deploymets\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 9\n      },\n      \"id\": 14,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"avg by(operation) (rate(cadence_client_errors{domain!~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Errors per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": [\n          {\n            \"__systemRef\": \"hideSeriesFrom\",\n            \"matcher\": {\n              \"id\": \"byNames\",\n              \"options\": {\n                \"mode\": \"exclude\",\n                \"names\": [\n                  \"HistoryClientRespondDecisionTaskCompleted\"\n                ],\n                \"prefix\": \"All except:\",\n                \"readOnly\": true\n              }\n            },\n            \"properties\": [\n              {\n                \"id\": \"custom.hideFrom\",\n                \"value\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": true\n                }\n              }\n            ]\n          }\n        ]\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 9\n      },\n      \"id\": 16,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \" Latency Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 17\n      },\n      \"id\": 17,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(TaskList) (rate(cadence_activity_poll_total{domain=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(TaskList) (rate(cadence_activity_poll_transient_failed[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity poll counter\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 17\n      },\n      \"id\": 20,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(schedule_activity_decision[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity decision per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 25\n      },\n      \"id\": 19,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(operation) (rate(activity_end_to_end_latency_sum{domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity end to end processing time\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 25\n      },\n      \"id\": 18,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(workflow_started_count[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Workflow Started Count\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 33\n      },\n      \"id\": 2,\n      \"panels\": [],\n      \"title\": \"Ratelimiter\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 34\n      },\n      \"id\": 1,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(global_ratelimit_collection) (rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Limiter in use\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 34\n      },\n      \"id\": 3,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"max by(global_ratelimit_key, global_ratelimit_collection) (global_ratelimiter_quota{global_ratelimit_key=\\\"$domain\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Quotas\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 42\n      },\n      \"id\": 5,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 42\n      },\n      \"id\": 4,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum(rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_type=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Allowed\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 50\n      },\n      \"id\": 6,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_rejected_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Rejected\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 50\n      },\n      \"id\": 7,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_rejected_requests{is_primary=\\\"false\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Rejected (shadow)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 58\n      },\n      \"id\": 29,\n      \"panels\": [],\n      \"title\": \"Workflow\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 59\n      },\n      \"id\": 26,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(cadence_workflow_endtoend_latency_bucket{domain=~\\\"$domain\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Workflow End to End Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 59\n      },\n      \"id\": 28,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(namespace, domain) (rate(cadence_worker_panic{domain=~\\\"$domain\\\"}[2m]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"NonDeterministicError and Worker Panic\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 67\n      },\n      \"id\": 27,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"sum(count(cadence_sticky_cache_total_forced_eviction{domain=\\\"$domain\\\"})) by (domain)\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Workflow Sticky Cache Eviction\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 75\n      },\n      \"id\": 34,\n      \"panels\": [],\n      \"title\": \"Activity\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 76\n      },\n      \"id\": 30,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum(count(cadence_activity_task_failed{domain=\\\"$domain\\\"})) by (activitytype)\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Activity Task Operations\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 76\n      },\n      \"id\": 33,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"avg by(domain, tasklist, activitytype) (cadence_activity_execution_latency{domain=~\\\"$domain\\\"})\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity Execution Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 84\n      },\n      \"id\": 31,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"max\",\n            \"min\",\n            \"sum\",\n            \"lastNotNull\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, tasklist) (rate(cadence_requests_per_tl{cadence_service=\\\"cadence_matching\\\", domain=~\\\"$domain\\\"}[5m]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"ActivityTasks scheduled per Second\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 84\n      },\n      \"id\": 32,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"avg by(Domain, TaskList, ActivityType) (cadence_activity_scheduled_to_start_latency_count{domain=~\\\"$domain\\\"})\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity Scheduled To Start Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 92\n      },\n      \"id\": 37,\n      \"panels\": [],\n      \"title\": \"Service\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 93\n      },\n      \"id\": 35,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"(1 - count(cadence_error{domain=\\\"$domain\\\"})by(domain)/count(cadence_request{domain=\\\"$domain\\\"})by(domain)) * 100\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Service API success rate\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 93\n      },\n      \"id\": 36,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"sum(avg_over_time(cadence_latency{Domain=\\\"$domain\\\"} [1m])) by (Domain)\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Service API Latency (TODO exclude long-poll APIs)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 101\n      },\n      \"id\": 24,\n      \"panels\": [],\n      \"title\": \"History\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 102\n      },\n      \"id\": 22,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"histogram_quantile($quantiles, (max by (le, domain) (history_size_bucket)))*1000000000\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Max History Size\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 102\n      },\n      \"id\": 23,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by (domain) (history_count_sum{domain!=\\\"all\\\"}/history_count_count{domain!=\\\"all\\\"})\\n*1000000000\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Avg History Length \",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 110\n      },\n      \"id\": 25,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"editorMode\": \"code\",\n          \"expr\": \"max(histogram_quantile(1, max(event_blob_size_bucket) by (le)))*1000000000\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Max Event Blob Size\",\n      \"type\": \"timeseries\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"\",\n          \"value\": \"\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"0.99\",\n          \"value\": \"0.99\"\n        },\n        \"label\": \"Quantiles\",\n        \"name\": \"quantiles\",\n        \"options\": [\n          {\n            \"selected\": false,\n            \"text\": \"0.5\",\n            \"value\": \"0.5\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"0.95\",\n            \"value\": \"0.95\"\n          },\n          {\n            \"selected\": true,\n            \"text\": \"0.99\",\n            \"value\": \"0.99\"\n          }\n        ],\n        \"query\": \"0.5, 0.95, 0.99\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-6h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence-Client\",\n  \"uid\": \"dehkspwgabvuoc\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-frontend.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 3,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 6,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1\n          },\n          \"id\": 2,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Request Counts Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1\n          },\n          \"id\": 3,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_per_tl{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"rate(cadence_PollForActivityTask_cadence_error{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"D\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Error Breakdown\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 25\n          },\n          \"id\": 4,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Error Counts Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 25\n          },\n          \"id\": 5,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Request Vs Error\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 33\n          },\n          \"id\": 7,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend API Latencies\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 12,\n      \"panels\": [],\n      \"title\": \"Frontend-Client\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 2\n      },\n      \"id\": 8,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (request)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (error)\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Vs Error\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 2\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Counts Per API\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"ms\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 10\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client API Latencies\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 10\n      },\n      \"id\": 10,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Counts Per API\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 18\n      },\n      \"id\": 27,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 19\n          },\n          \"id\": 13,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"StartWorkflowExecution\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"StartWorfklowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 19\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"StartWorkflowExecution\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"StartWorkflowExection Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 27\n          },\n          \"id\": 21,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, tasklist) (rate(cadence_requests_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests per operation per task\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 27\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, tasklist) (rate(workflow_success{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Workflow success per operation per task\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 35\n          },\n          \"id\": 22,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"RespondActivityTaskCompleted\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Respond Activity task completed\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"sec\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 35\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"RespondActivityTaskCompleted\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"RespondActivity Task Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 43\n          },\n          \"id\": 17,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(TaskList) (rate(cadence_PollForActivityTask_cadence_request[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 43\n          },\n          \"id\": 23,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 51\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(TaskList) (rate(cadence_PollForDecisionTask_cadence_request[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 51\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 59\n          },\n          \"id\": 15,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 59\n          },\n          \"id\": 16,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 67\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"GetWorkflowExecutionHistory\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GetWorkflowExecutionHistory\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 67\n          },\n          \"id\": 29,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"CancelWorkflowExecution\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"CancelWorkflowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 75\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"TerminateWorkflowExecution\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TerminateWorkflowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 75\n          },\n          \"id\": 31,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"RegisterDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"RegisterDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 83\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 83\n          },\n          \"id\": 32,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"UpdateDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"UpdateDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 91\n          },\n          \"id\": 34,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DepricateDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DepricateDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 91\n          },\n          \"id\": 35,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddDecisionTask\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"API\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 19\n      },\n      \"id\": 39,\n      \"panels\": [],\n      \"title\": \"Persistence\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 20\n      },\n      \"id\": 36,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"{{operation}} (request)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (error)\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Vs Errors\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 20\n      },\n      \"id\": 37,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"sec\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 28\n      },\n      \"id\": 38,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, avg by(operation, le) (rate(persistence_latency_bucket{cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Operation Latencies p99\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 36\n      },\n      \"id\": 42,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"{le=\\\"1.0000000000000001e-07\\\", operation=\\\"StartWorkflowExecution\\\"}\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 317\n          },\n          \"id\": 40,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Overall Query Sizes\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"{le=\\\"1.0000000000000001e-07\\\", operation=\\\"StartWorkflowExecution\\\"}\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 317\n          },\n          \"id\": 41,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Query Sizes by Domain\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Query sizes\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 37\n      },\n      \"id\": 44,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"histogram_quantile(0.95, sum by(le) (rate(cadence_authorization_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[1m0s])))\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 318\n          },\n          \"id\": 43,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_authorization_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Authorization latency per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Authorization\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 38\n      },\n      \"id\": 52,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 319\n          },\n          \"id\": 47,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_allocated{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Total Memory Allocation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"NumGC is the number of completed GC cycles.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 319\n          },\n          \"id\": 46,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_num_gc{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GC Num\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 327\n          },\n          \"id\": 45,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(num_goroutines{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Number of goroutines\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapInuse is bytes in in-use spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 327\n          },\n          \"id\": 49,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapinuse{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap in Use (HeapAlloc)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapIdle is bytes in idle (unused) spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 335\n          },\n          \"id\": 48,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapidle{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap Available (HeapIdle)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"Bytes of stack memory obtained from the OS. Plus any memory obtained directly from the OS for OS thread stacks.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 335\n          },\n          \"id\": 51,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_stack{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Stack In use\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Go Gc Detailed\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 39\n      },\n      \"id\": 57,\n      \"panels\": [],\n      \"title\": \"Replication\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 40\n      },\n      \"id\": 54,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"rate(cadence_requests{operation=\\\"GetReplicationMessage\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval])\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetReplicationMessage Requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 40\n      },\n      \"id\": 53,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"rate(cadence_requests{operation=\\\"GetDomainReplicationMessage\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval])\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetDomainReplicationMessage Requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 48\n      },\n      \"id\": 55,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le) (rate(cadence_latency_bucket{operation=\\\"GetReplicationMessage\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetReplicationMessage Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 48\n      },\n      \"id\": 56,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le) (rate(cadence_latency_bucket{operation=\\\"GetDomainReplicationMessage\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetDomainReplicationMessage Latency\",\n      \"type\": \"timeseries\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n        \"includeAll\": true,\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"cadence_frontend\",\n          \"value\": \"cadence_frontend\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"p99\",\n          \"value\": \"p99\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": true,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-5m\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence-Frontend\",\n  \"uid\": \"cekm8fs6ps000f\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-history.json",
    "content": "{\n    \"annotations\": {\n      \"list\": [\n        {\n          \"builtIn\": 1,\n          \"datasource\": {\n            \"type\": \"grafana\",\n            \"uid\": \"-- Grafana --\"\n          },\n          \"enable\": true,\n          \"hide\": true,\n          \"iconColor\": \"rgba(0, 211, 255, 1)\",\n          \"name\": \"Annotations & Alerts\",\n          \"type\": \"dashboard\"\n        }\n      ]\n    },\n    \"editable\": true,\n    \"fiscalYearStartMonth\": 0,\n    \"graphTooltip\": 0,\n    \"id\": 5,\n    \"links\": [],\n    \"panels\": [\n      {\n        \"collapsed\": false,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 0\n        },\n        \"id\": 9,\n        \"panels\": [],\n        \"title\": \"API\",\n        \"type\": \"row\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 1\n        },\n        \"id\": 7,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.1\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Request Per API\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 1\n        },\n        \"id\": 8,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.1\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"History API p99 Latencies\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 9\n        },\n        \"id\": 26,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 21,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(complete_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(schedule_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(fail_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"C\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"D\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(start_timer_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"E\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"F\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_timer_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"G\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"H\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(record_marker_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"I\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_external_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"J\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(continue_as_new_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"K\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(child_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"L\"\n              }\n            ],\n            \"title\": \"Decision Breakdown\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 22,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(multiple_completion_decisions{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": true,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(failed_decisions{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\"\n              }\n            ],\n            \"title\": \"Bad Decisions\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Decisions\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 10\n        },\n        \"id\": 14,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 10,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(cadence_service) (rate(cadence_client_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{cadence_service}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(cadence_service) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{cadence_service}} (error)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients Requests Vs Errors\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 11,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients Request Count Per API\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 513\n            },\n            \"id\": 12,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Client Error Counts Per API\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 513\n            },\n            \"id\": 13,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients API Latencies\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"History Clients\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 11\n        },\n        \"id\": 20,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 15,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cache_requests{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_miss{cadence_service=\\\"cadence_history\\\", cache_type=\\\"mutablestate\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (miss)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Mutable State Cache\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 16,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cache_latency_bucket{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Mutable State Cache Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 513\n            },\n            \"id\": 17,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_requests{cache_type=\\\"events\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_miss{cadence_service=\\\"cadence_history\\\", cache_type=\\\"events\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (miss)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Events Cache\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 513\n            },\n            \"id\": 18,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cache_latency_bucket{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Events Cache Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"WorkflowContext\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 521\n            },\n            \"id\": 19,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(workflow_context_lock_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Execution Lock Held Duration\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"History Caches\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 12\n        },\n        \"id\": 30,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 9\n            },\n            \"id\": 23,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(acquire_shards_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardController\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Shard Movement\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 9\n            },\n            \"id\": 27,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(get_engine_for_shard_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (get_engine)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_per_shard_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (persistence_latency_per_shard)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"p99 Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (shard info replication)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 457\n            },\n            \"id\": 25,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(shardinfo_transfer_lag_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardInfo\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (shard info transfer)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(shardinfo_replication_lag_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardInfo\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (shard info replication)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Replication / Transfer p99 Lag (buggy)\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 457\n            },\n            \"id\": 24,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_lag_bucket{operation=\\\"ShardInfo\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Timer p99 Lag\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 465\n            },\n            \"id\": 28,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_diff_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Transfer Level p99 Diff\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                },\n                \"unit\": \"ms\"\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 465\n            },\n            \"id\": 29,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_diff_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Timer Level p99 Diff\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (timer failover)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 473\n            },\n            \"id\": 31,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_failover_in_progress_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (transfer failover)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_failover_in_progress_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (timer failover)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Failover p99 In Progress\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (transfer active pending task)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 473\n            },\n            \"id\": 32,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_active_pending_task_bucket[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (transfer active pending task)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_active_pending_task_bucket[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (timer active pending task)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Pending p99 Tasks\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"ShardController\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 13\n        },\n        \"id\": 38,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 14\n            },\n            \"id\": 33,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(__name__) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"persistence_requests\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"persistence_errors\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Requests vs Errors\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 14\n            },\n            \"id\": 34,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"EntityNotExists\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Error Breakdown\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 118\n            },\n            \"id\": 35,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Requests Per Operation\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 118\n            },\n            \"id\": 36,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Error Per Operation\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 126\n            },\n            \"id\": 37,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"avg by(operation) (rate(persistence_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Operation p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Persistence\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 14\n        },\n        \"id\": 47,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 15\n            },\n            \"id\": 39,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TimerActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"TimerActiveQueueProcessor\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 15\n            },\n            \"id\": 40,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TimerStandByQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 23\n            },\n            \"id\": 41,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(new_timer_notifications{operation=\\\"TimerActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 23\n            },\n            \"id\": 42,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(new_timer_notifications{operation=\\\"TimerStandByQueueProcessor\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby New Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 31\n            },\n            \"id\": 43,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(task_requests{operation=\\\"TimerActiveTaskActivityTimeout\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Requests\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 31\n            },\n            \"id\": 44,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_processing_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Processing p99 Latency\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 39\n            },\n            \"id\": 46,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_attempt_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer p99 Attempt\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 39\n            },\n            \"id\": 45,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_queue_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Queue p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Timer Processor\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 15\n        },\n        \"id\": 48,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 16\n            },\n            \"id\": 49,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TransferActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"TimerActiveQueueProcessor\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 16\n            },\n            \"id\": 50,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TransferStandByQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 24\n            },\n            \"id\": 51,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(new_timer_notifications{operation=\\\"TransferActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 24\n            },\n            \"id\": 52,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"rate(new_timer_notifications{operation=\\\"TranferStandByQueueProcessor\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby New Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 32\n            },\n            \"id\": 53,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(task_requests{operation=\\\"TransferActiveTaskActivityTimeout\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Requests\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 32\n            },\n            \"id\": 54,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_processing_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Processing p99 Latency\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 40\n            },\n            \"id\": 55,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_attempt_bucket{operation=~\\\"TransferActiveTaskActivityTimeout|TransferActiveTaskDecisionTimeout|TransferActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer p99 Attempt\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 40\n            },\n            \"id\": 56,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_queue_bucket{operation=~\\\"TransferActiveTaskActivityTimeout|TransferActiveTaskDecisionTimeout|TransferActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Queue p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Transfer Processor\",\n        \"type\": \"row\"\n      }\n    ],\n    \"preload\": false,\n    \"schemaVersion\": 41,\n    \"tags\": [],\n    \"templating\": {\n      \"list\": [\n        {\n          \"current\": {\n            \"text\": \"samples_domain\",\n            \"value\": \"samples_domain\"\n          },\n          \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"includeAll\": true,\n          \"label\": \"Domain\",\n          \"name\": \"domain\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"cadence_history\",\n            \"value\": \"cadence_history\"\n          },\n          \"definition\": \"label_values(cadence_requests,cadence_service)\",\n          \"label\": \"services\",\n          \"name\": \"services\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(cadence_requests,cadence_service)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          \"label\": \"latency\",\n          \"name\": \"latency\",\n          \"options\": [\n            {\n              \"selected\": true,\n              \"text\": \"p99\",\n              \"value\": \"p99\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"upper\",\n              \"value\": \"upper\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"mean\",\n              \"value\": \"mean\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"median\",\n              \"value\": \"median\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"lower\",\n              \"value\": \"lower\"\n            }\n          ],\n          \"query\": \"p99,upper,mean,median,lower\",\n          \"type\": \"custom\"\n        }\n      ]\n    },\n    \"time\": {\n      \"from\": \"now-5m\",\n      \"to\": \"now\"\n    },\n    \"timepicker\": {},\n    \"timezone\": \"browser\",\n    \"title\": \"Cadence-History\",\n    \"uid\": \"cekm8fs6ps000h\",\n    \"version\": 1\n  }"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-matching.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 3,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 6,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1\n          },\n          \"id\": 1,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Overall Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1\n          },\n          \"id\": 2,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Breakdown\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 113\n          },\n          \"id\": 3,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 113\n          },\n          \"id\": 4,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"API Latencies (p99)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 121\n          },\n          \"id\": 5,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(tasklist_managers{cadence_service=\\\"cadence_matching\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task Managers Per Operation\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 15,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 2\n          },\n          \"id\": 7,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 2\n          },\n          \"id\": 8,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"PollForActivityTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 114\n          },\n          \"id\": 9,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"hide\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\"\n            }\n          ],\n          \"title\": \"PollForDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 114\n          },\n          \"id\": 10,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"PollForDecisionTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 122\n          },\n          \"id\": 11,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 122\n          },\n          \"id\": 13,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"AddActivityTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 130\n          },\n          \"id\": 12,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 130\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"AddDecisionTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"API\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 2\n      },\n      \"id\": 23,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"sec\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 3\n          },\n          \"id\": 16,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (changes(poll_success_per_tl{operation=\\\"TaskListMgr\\\", domain=\\\"$domain\\\", \\\"$services\\\"=\\\"$services\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"{{operation}} (async)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (changes(poll_success_sync_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (sync)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(poll_timeouts_per_tl{operation=\\\"TaskListMgr\\\", cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (poll_timeouts)\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskPoll\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 3\n          },\n          \"id\": 17,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by() (rate(cadence_PollForActivityTask_cadence_request{cadence_service=\\\"$services\\\", \\\"$domain\\\"=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task Matcher Stats\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 115\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, avg by(le, operation) (rate(cadence_latency_per_tl_bucket{operation=\\\"AddActivityTask\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask Latency(p99) (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 115\n          },\n          \"id\": 18,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum(rate(cadence_PollForDecisionTask_cadence_request{cadence_service=\\\"$services\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task Matcher Stat\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 123\n          },\n          \"id\": 20,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_per_tl_bucket{operation=\\\"AddDecisionTask\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask Latency(p99) (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 123\n          },\n          \"id\": 22,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask QPS (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"TaskListManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 3\n      },\n      \"id\": 27,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 4\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 4\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Operation Latencies p99\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"GetTaskListSize\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 100\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Persistence\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 4\n      },\n      \"id\": 31,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 5\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_role=\\\"history_client\\\", cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 5\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"API Latency (p99)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 69\n          },\n          \"id\": 29,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_client_errors_redirection{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"History Client\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 5\n      },\n      \"id\": 34,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 6\n          },\n          \"id\": 32,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (changes(task_count_per_tl{tasklistType=\\\"activity\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task Backlog\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 6\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (changes(task_backlog_per_tl{tasklistType=\\\"decision\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task Backlog\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Backlog (per task list)\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 6\n      },\n      \"id\": 37,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 7\n          },\n          \"id\": 35,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_num_gc{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GC Num\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 7\n          },\n          \"id\": 36,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_allocated{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Total allocation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 39\n          },\n          \"id\": 38,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(num_goroutines{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Number of goroutines\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapInuse is bytes in in-use spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 39\n          },\n          \"id\": 39,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapinuse{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap in Use (HeapAlloc)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapIdle is bytes in idle (unused) spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 47\n          },\n          \"id\": 40,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapidle{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap Available (HeapIdle)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"Bytes of stack memory obtained from the OS. Plus any memory obtained directly from the OS for OS thread stacks.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 47\n          },\n          \"id\": 41,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_stack{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Stack In use\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Go GC detailed\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 7\n      },\n      \"id\": 48,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 8\n          },\n          \"id\": 42,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(tasklist) (rate(estimated_add_task_qps_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Adaptive Task List Scaler for Activity\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 8\n          },\n          \"id\": 43,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(estimated_add_task_qps_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Adaptive Task List Scaler for Decision\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 16\n          },\n          \"id\": 44,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_read{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task List Read Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 16\n          },\n          \"id\": 45,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_read{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task List Read Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 24\n          },\n          \"id\": 46,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_write{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task List Write Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 24\n          },\n          \"id\": 47,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_write{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task List Write Partition Count\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Task List Partition Config & Adaptive Task List Scaler (WIP)\",\n      \"type\": \"row\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"cadence_matching\",\n          \"value\": \"cadence_matching\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"p99\",\n          \"value\": \"p99\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": true,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-1h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence Matching\",\n  \"uid\": \"del7jwba0qgw0e\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-persistence.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"target\": {\n          \"limit\": 100,\n          \"matchAny\": false,\n          \"tags\": [],\n          \"type\": \"dashboard\"\n        },\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 2,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 12,\n      \"panels\": [],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 17,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_response_row_size_count{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Response Row Count Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 1\n      },\n      \"id\": 18,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_response_payload_size_sum{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Response Payload Size Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 12\n      },\n      \"id\": 8,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (rate(persistence_requests_per_domain{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 12\n      },\n      \"id\": 2,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          },\n          \"unit\": \"s\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 23\n      },\n      \"id\": 3,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"max\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Max\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(persistence_latency_bucket{cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Latencies Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 23\n      },\n      \"id\": 10,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (rate(persistence_requests_per_domain{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Domain Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 9,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"fieldMinMax\": false,\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 34\n      },\n      \"id\": 1,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_empty_response{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Empty Response Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 34\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation, shard_id) (rate(persistence_requests_per_shard{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} - {{shard_id}}(shard_id)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Operation Per Shard\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 45\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Errors Per Operation \",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 56\n      },\n      \"id\": 16,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 160\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"right\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"asc\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(le, operation) (rate(persistence_requests{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|LeaseTaskList|UpdateTaskList|CreateShard|UpdateShard\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"LWT Requests\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 160\n          },\n          \"id\": 15,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"right\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.9, max by(operation, le) (rate(persistence_latency_bucket{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|LeaseTaskList|UpdateTaskList|CreateShard|UpdateShard\\\", cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"LWT Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"LWT Operations\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 57\n      },\n      \"id\": 22,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 161\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, cadence_service) (rate(persistence_requests{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 161\n          },\n          \"id\": 21,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, cadence_service) (rate(persistence_empty_response{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Empty Response Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 193\n          },\n          \"id\": 20,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per Operation\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"History Tasks\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 58\n      },\n      \"id\": 29,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 162\n          },\n          \"id\": 23,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 162\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard CRUD Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 194\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 194\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task CRUD Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 202\n          },\n          \"id\": 27,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskList CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 202\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskList CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"ShardMgr & TaskMgr\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 59\n      },\n      \"id\": 32,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 163\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval]))\",\n              \"hide\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\"\n            }\n          ],\n          \"title\": \"WorkflowExecution CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 163\n          },\n          \"id\": 31,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"WorkflowExecution CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Execution Manager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 60\n      },\n      \"id\": 35,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 164\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(persistence_requests{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(persistence_errors_entity_not_exists{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Domain CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"min\": 4,\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 164\n          },\n          \"id\": 34,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Domain CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"MetadataManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 61\n      },\n      \"id\": 38,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 165\n          },\n          \"id\": 36,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Visibility Open/Close\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"dtdurationms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 165\n          },\n          \"id\": 37,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Visibility Open/Close Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"VisibilityManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 62\n      },\n      \"id\": 45,\n      \"panels\": [],\n      \"title\": \"Persistence Hot Shard Detection\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 63\n      },\n      \"id\": 44,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation, shard_id) (rate(persistence_requests_per_shard{operation=\\\"UpdateWorkflowExecution\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Total updateworkflowexecution tracked shardId based requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 63\n      },\n      \"id\": 39,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(task_requests_per_domain{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Timer Task Request Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 71\n      },\n      \"id\": 40,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(task_requests_per_domain{domain=\\\"$domain\\\", cadence_service=\\\"$services\\\", operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Transfer Task Request Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 71\n      },\n      \"id\": 43,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, shard_id) (rate(persistence_latency_per_shard_bucket{operation=\\\"ShardIdPersistenceRequest\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"shardidpersistencerequest p99 latency shardId\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"s\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 79\n      },\n      \"id\": 41,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_per_domain_bucket{domain=~\\\"$domain\\\", cadence_service=~\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Persistence Latency Per Domain\",\n      \"type\": \"timeseries\"\n    }\n  ],\n  \"preload\": false,\n  \"refresh\": \"\",\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"allowCustomValue\": false,\n        \"current\": {\n          \"text\": \"All\",\n          \"value\": \"$__all\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"includeAll\": false,\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"allowCustomValue\": false,\n        \"current\": {\n          \"text\": \"upper\",\n          \"value\": \"upper\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": false,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": true,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      },\n      {\n        \"allowCustomValue\": true,\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-6h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"\",\n  \"title\": \"Cadence-Persistence\",\n  \"uid\": \"b00f7dd5-3dd6-4cf9-b17b-425e071e2b5b\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/cadence-server.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"datasource\",\n          \"uid\": \"grafana\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"target\": {\n          \"limit\": 100,\n          \"matchAny\": false,\n          \"tags\": [],\n          \"type\": \"dashboard\"\n        },\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 7,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 44,\n      \"panels\": [],\n      \"title\": \"Frontend\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 57,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"max\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"SignalWithStartWorkflowExecution|SignalWorkflowxecution|StartWorkflowExecution|TerminateWorkflowExecution|ResetWorkflowExecution|RequestCancelWorkflowexecution|ListWorkflowExecutions\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"\",\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Regular Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 7\n      },\n      \"id\": 58,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"PollForDecisionTask|GetWorkflowExecutionHistory|RespondDecisionTaskCompleted\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Workflow Worker Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 13\n      },\n      \"id\": 60,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for All Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 19\n      },\n      \"id\": 59,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(cadence_requests{cadence_service=\\\"cadence_frontend\\\",namespace=\\\"$namespace\\\", operation=~\\\"PollForActivityTask|RespondActivityTaskCompleted\\\"}) by (domain)\",\n          \"hide\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"refId\": \"B\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Activity Worker Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 25\n      },\n      \"id\": 48,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"All API per second per domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 25\n      },\n      \"id\": 49,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"API per second(breakdown per operation)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 33\n      },\n      \"id\": 50,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation!~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Regular API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 33\n      },\n      \"id\": 51,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"ListWorkflow API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 41\n      },\n      \"id\": 52,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForDecisionTask\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForActivityTask\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Long Poll API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 41\n      },\n      \"id\": 56,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation!~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"Domain: {{domain}}- Operation: {{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Regular API Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 49\n      },\n      \"id\": 54,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"Internal server error\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_bad_request{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"hide\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"Bad Requests error\",\n          \"refId\": \"B\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_query_failed{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"QueryFailed error\",\n          \"refId\": \"C\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_context_timeout{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"ContextTimeout error\",\n          \"refId\": \"D\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"EntityNotExists error\",\n          \"refId\": \"E\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_execution_already_started{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"WorkflowAlreadyStarted error\",\n          \"refId\": \"F\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_workflow_execution_already_completed{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"WorkflowAlreadyCompleted\",\n          \"refId\": \"G\"\n        }\n      ],\n      \"title\": \"API  errors per second(breakdown per operation)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 49\n      },\n      \"id\": 53,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"GetWorkflowExecutionHistory|QueryWorkflow\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetHistory/QueryWorkflow API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 57\n      },\n      \"id\": 55,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (changes(cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"SignalWithStartworkflowExecution|SignalWorkflowExecution|StartWorkflowExecution|TerminateWorkflowExecution|ResetWorkflowExecution|RequestCancelWorkflowExecution|ListWorkflowExecutions\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"Domain: {{domain}}- Operation: {{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"WorkflowClient API per seconds by domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 65\n      },\n      \"id\": 96,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 641\n          },\n          \"id\": 93,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(restarts[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Service Restarts\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 641\n          },\n          \"id\": 95,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(acquire_shards_count[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (shard acquire count)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (rate(sharditem_created_count[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"shard item created count\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(numshards_gauge[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"num of shard gauge\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(persistence_requests_per_shard[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"persistence requests per shard\",\n              \"range\": true,\n              \"refId\": \"D\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard Movements per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1537\n          },\n          \"id\": 94,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"avg by(cadence_service) (num_goroutines)\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Goroutines (avg)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Service Metrics\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 66\n      },\n      \"id\": 101,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1618\n          },\n          \"id\": 97,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(cadence_requests{cadence_service=\\\"cadence_history\\\", operation=~\\\"StartWorkflowExecution|SignalWorkflowExecution\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"External Events per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1618\n          },\n          \"id\": 98,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=~\\\"RespondActivityTaskCompleted|RecordActivityTaskHeartbeat|RecordActivityTaskStarted|RespondDecisionTaskCompleted\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activities per sec \",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1626\n          },\n          \"id\": 100,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(cadence_requests{operation=~\\\"PollForActivityTask|PollForDecisionTask|CancelOutstandingPoll\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(poll_success_sync{cadence_service=\\\"cadence_matching\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"poll success sync\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (rate(poll_timeouts{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"poll timeout\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Polls per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1626\n          },\n          \"id\": 99,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=~\\\"RecordDecisionTaskStarted|RespondDecisionTaskCompleted\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decisions per sec\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Rates per second\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 67\n      },\n      \"id\": 106,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 0,\n            \"y\": 643\n          },\n          \"id\": 102,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_frontend\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 6,\n            \"y\": 643\n          },\n          \"id\": 103,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_history\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 12,\n            \"y\": 643\n          },\n          \"id\": 105,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_worker\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Worker Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 18,\n            \"y\": 643\n          },\n          \"id\": 104,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_matching\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Matching Versions\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Server Version Metrics\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 68\n      },\n      \"id\": 111,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 644\n          },\n          \"id\": 107,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(task_requests{cadence_service=\\\"cadence_history\\\", operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Tasks Requests\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 644\n          },\n          \"id\": 108,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (task_latency_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Task Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1108\n          },\n          \"id\": 109,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(domain) (changes(task_requests_per_domain{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Tasks Per Domain per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1108\n          },\n          \"id\": 110,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(task_latency_per_domain_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Task Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Timer Task Processing\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 69\n      },\n      \"id\": 117,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 24,\n            \"x\": 0,\n            \"y\": 645\n          },\n          \"id\": 112,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(__name__) (rate(task_requests{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"requests\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(__name__) (rate(task_errors{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"errors\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 797\n          },\n          \"id\": 116,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(task_requests{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Requests Per Type\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 797\n          },\n          \"id\": 113,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(task_latency_bucket{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Task Latency (upper)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 805\n          },\n          \"id\": 114,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(domain) (rate(task_requests_per_domain{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Per Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 805\n          },\n          \"id\": 115,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, domain) (rate(task_latency_per_domain_bucket{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Task upper Latency per Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Transfer Task Processing\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 70\n      },\n      \"id\": 125,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 641\n          },\n          \"id\": 118,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(history_size_bucket{operation=\\\"ExecutionStats\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Size By Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 641\n          },\n          \"id\": 119,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(history_count_bucket{operation=~\\\"ExecutionStats\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Event Count By Domain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1105\n          },\n          \"id\": 120,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Blob Sizes Top Domains\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1105\n          },\n          \"id\": 121,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Blob Sizes Top Operations\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1113\n          },\n          \"id\": 123,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Blob Sizes by Domain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1113\n          },\n          \"id\": 122,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, decisionType) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Blob Sizes by DecisionType\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1121\n          },\n          \"id\": 124,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(decision_result_count_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Result Count Top Domains\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Size and Count Limits\",\n      \"type\": \"row\"\n    }\n  ],\n  \"preload\": false,\n  \"refresh\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"all\",\n          \"value\": \"all\"\n        },\n        \"datasource\": \"Prometheus\",\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"includeAll\": false,\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-5m\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {\n    \"refresh_intervals\": [\n      \"10s\",\n      \"30s\",\n      \"1m\",\n      \"5m\",\n      \"15m\",\n      \"30m\",\n      \"1h\",\n      \"2h\",\n      \"1d\"\n    ]\n  },\n  \"timezone\": \"\",\n  \"title\": \"Cadence-Server\",\n  \"uid\": \"1B0efRyGz\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/dashboards/default.yaml",
    "content": "apiVersion: 1\n\nproviders:\n  - name: Cadence-Basic # A uniquely identifiable name for the provider\n    folder: Cadence\n    type: file\n    options:\n      path:\n        /etc/grafana/provisioning/dashboards/\n"
  },
  {
    "path": "docker/github_actions/grafana/provisioning/datasources/default.yaml",
    "content": "datasources:\n  - name: Prometheus\n    type: prometheus\n    url: http://prometheus:9090\n    version: 1\n    editable: true\n"
  },
  {
    "path": "docker/github_actions/prometheus/.gitignore",
    "content": "*.log\n"
  },
  {
    "path": "docker/github_actions/prometheus/matching_simulation_prometheus.yml",
    "content": "global:\n  scrape_interval: 5s\n  external_labels:\n    monitor: 'cadence-monitor'\n  query_log_file: /etc/prometheus/query.log\n  scrape_failure_log_file: /etc/prometheus/scrape.log\nscrape_configs:\n  - job_name: 'prometheus'\n    static_configs:\n      - targets: # addresses to scrape\n          - 'cadence:8306'\n          - 'cadence:8307'\n          - 'cadence:8308'\n          - 'cadence:8309'\n"
  },
  {
    "path": "docker/github_actions/prometheus/replication_simulation_prometheus.yml",
    "content": "global:\n  scrape_interval: 5s\n  external_labels:\n    monitor: 'cadence-monitor'\n  query_log_file: /etc/prometheus/query.log\n  scrape_failure_log_file: /etc/prometheus/scrape.log\nscrape_configs:\n  - job_name: 'prometheus'\n    static_configs:\n      - targets: # addresses to scrape from cluster0\n          - 'cadence-cluster0:7000' # frontend\n          - 'cadence-cluster0:7001' # matching\n          - 'cadence-cluster0:7002' # history\n          - 'cadence-cluster0:7003' # worker\n        labels:\n          cluster: 'cluster0'\n      - targets: # addresses to scrape from cluster1\n          - 'cadence-cluster1:8000' # frontend\n          - 'cadence-cluster1:8001' # matching\n          - 'cadence-cluster1:8002' # history\n          - 'cadence-cluster1:8003' # worker\n        labels:\n          cluster: 'cluster1'\n"
  },
  {
    "path": "docker/grafana/grafana.ini",
    "content": "[auth.anonymous]\nenabled = true\norg_role = Admin"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-archival.json",
    "content": "{\n    \"annotations\": {\n      \"list\": [\n        {\n          \"builtIn\": 1,\n          \"datasource\": {\n            \"type\": \"grafana\",\n            \"uid\": \"-- Grafana --\"\n          },\n          \"enable\": true,\n          \"hide\": true,\n          \"iconColor\": \"rgba(0, 211, 255, 1)\",\n          \"name\": \"Annotations & Alerts\",\n          \"type\": \"dashboard\"\n        }\n      ]\n    },\n    \"editable\": true,\n    \"fiscalYearStartMonth\": 0,\n    \"graphTooltip\": 0,\n    \"id\": 6,\n    \"links\": [],\n    \"panels\": [\n      {\n        \"collapsed\": false,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 0\n        },\n        \"id\": 8,\n        \"panels\": [],\n        \"title\": \"Overall\",\n        \"type\": \"row\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"history\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 1\n        },\n        \"id\": 1,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": true,\n            \"legendFormat\": \"history\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (rate(archiver_client_visibility_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"visibility\",\n            \"range\": true,\n            \"refId\": \"B\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Archivals Triggered Per Second\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"archiver_client_visibility_inline_archive_attempt\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 1\n        },\n        \"id\": 2,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": true,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_request\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(cadence_service) (changes(archiver_client_history_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": true,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"B\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_visibility_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_visibility_request\",\n            \"range\": true,\n            \"refId\": \"C\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_visibility_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_visibility_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"D\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Client Requests vs Inline Attempts\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": [\n            {\n              \"__systemRef\": \"hideSeriesFrom\",\n              \"matcher\": {\n                \"id\": \"byNames\",\n                \"options\": {\n                  \"mode\": \"exclude\",\n                  \"names\": [\n                    \"archiver_client_history_request\"\n                  ],\n                  \"prefix\": \"All except:\",\n                  \"readOnly\": true\n                }\n              },\n              \"properties\": [\n                {\n                  \"id\": \"custom.hideFrom\",\n                  \"value\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": true\n                  }\n                }\n              ]\n            }\n          ]\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 9\n        },\n        \"id\": 3,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(archiver_client_history_request{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"archiver_client_history_request\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Archiver History Requests\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 9\n        },\n        \"id\": 4,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(__name__) (rate(archiver_client_history_inline_archive_attempt{operation=\\\"ArchiverClient\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"archiver_client_history_inline_archive_attempt\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Client History Inline Attemptss\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 17\n        },\n        \"id\": 5,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(domain, operation) (rate(archiver_client_history_request_per_domain{domain=\\\"$domain\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"expr\": \"\",\n            \"hide\": false,\n            \"instant\": false,\n            \"range\": true,\n            \"refId\": \"B\"\n          }\n        ],\n        \"title\": \"History Requests Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 17\n        },\n        \"id\": 6,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(domain, operation) (rate(archiver_client_history_inline_archive_attempt_per_domain[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"History Inline Attempt Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 25\n        },\n        \"id\": 7,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(workflow_cleanup_archive[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"hide\": false,\n            \"includeNullMetadata\": false,\n            \"instant\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Workflow Cleanup Per Operation\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 25\n        },\n        \"id\": 9,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.2\",\n        \"targets\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"expr\": \"\",\n            \"hide\": false,\n            \"instant\": false,\n            \"range\": true,\n            \"refId\": \"A\"\n          }\n        ],\n        \"title\": \"History Inline Attempt Per Domain Per Operation\",\n        \"type\": \"timeseries\"\n      }\n    ],\n    \"preload\": false,\n    \"schemaVersion\": 41,\n    \"tags\": [],\n    \"templating\": {\n      \"list\": [\n        {\n          \"allowCustomValue\": false,\n          \"current\": {\n            \"text\": \"\",\n            \"value\": \"\"\n          },\n          \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"includeAll\": false,\n          \"label\": \"Domain\",\n          \"name\": \"domain\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"0.99\",\n            \"value\": \"0.99\"\n          },\n          \"label\": \"Quantiles\",\n          \"name\": \"quantiles\",\n          \"options\": [\n            {\n              \"selected\": false,\n              \"text\": \"0.5\",\n              \"value\": \"0.5\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"0.95\",\n              \"value\": \"0.95\"\n            },\n            {\n              \"selected\": true,\n              \"text\": \"0.99\",\n              \"value\": \"0.99\"\n            }\n          ],\n          \"query\": \"0.5, 0.95, 0.99\",\n          \"type\": \"custom\"\n        }\n      ]\n    },\n    \"time\": {\n      \"from\": \"now-6h\",\n      \"to\": \"now\"\n    },\n    \"timepicker\": {},\n    \"timezone\": \"\",\n    \"title\": \"Cadence-Archival\",\n    \"uid\": \"1B0efRyGx\",\n    \"version\": 1\n  }"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-client-overall.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 8,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 10,\n      \"panels\": [],\n      \"title\": \"Overview\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 44,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(workflowType) (rate(workflow_success[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(workflowType) (rate(workflow_cancel[$__rate_interval]))\",\n          \"hide\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(workflowType) (rate(workflow_continued_as_new[$__rate_interval]))\",\n          \"hide\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"C\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(workflowType) (rate(workflow_failed[$__rate_interval]))\",\n          \"hide\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"D\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(workflowType) (rate(workflow_timeout[$__rate_interval]))\",\n          \"hide\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"E\"\n        }\n      ],\n      \"title\": \"Workflow Success Counters\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 1\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"max by(active_cluster) (active_cluster{cadence_service=\\\"cadence_history\\\", operation=\\\"DomainCache\\\"})\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Active Deploymets\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 9\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": [\n          {\n            \"__systemRef\": \"hideSeriesFrom\",\n            \"matcher\": {\n              \"id\": \"byNames\",\n              \"options\": {\n                \"mode\": \"exclude\",\n                \"names\": [\n                  \"HistoryClientRespondDecisionTaskCompleted\"\n                ],\n                \"prefix\": \"All except:\",\n                \"readOnly\": true\n              }\n            },\n            \"properties\": [\n              {\n                \"id\": \"custom.hideFrom\",\n                \"value\": {\n                  \"legend\": false,\n                  \"tooltip\": true,\n                  \"viz\": true\n                }\n              }\n            ]\n          }\n        ]\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 9\n      },\n      \"id\": 16,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \" Latency Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 17\n      },\n      \"id\": 14,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"avg by(operation) (rate(cadence_client_errors{domain!~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Errors per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 17\n      },\n      \"id\": 20,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(schedule_activity_decision[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity decision per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 25\n      },\n      \"id\": 17,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(TaskList) (rate(cadence_activity_poll_total{domain=~\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(TaskList) (rate(cadence_activity_poll_transient_failed[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity poll counter\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 25\n      },\n      \"id\": 18,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(workflow_started_count[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Workflow Started Count\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"showValues\": false,\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\",\n                \"value\": 0\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 33\n      },\n      \"id\": 19,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.2.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(operation) (rate(activity_end_to_end_latency_sum{domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Activity end to end processing time\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 41\n      },\n      \"id\": 2,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 66\n          },\n          \"id\": 1,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(global_ratelimit_collection) (rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Limiter in use\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 66\n          },\n          \"id\": 3,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"max by(global_ratelimit_key, global_ratelimit_collection) (global_ratelimiter_quota{global_ratelimit_key=\\\"$domain\\\"})\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Quotas\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 90\n          },\n          \"id\": 5,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 90\n          },\n          \"id\": 4,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum(rate(global_ratelimiter_allowed_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_type=~\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Allowed\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 98\n          },\n          \"id\": 6,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_rejected_requests{is_primary=\\\"true\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\", global_ratelimit_key=~\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Rejected\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 98\n          },\n          \"id\": 7,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(global_ratelimit_collection, global_ratelimit_key, global_ratelimit_type) (rate(global_ratelimiter_rejected_requests{is_primary=\\\"false\\\", cadence_service=\\\"cadence_frontend\\\", operation=\\\"GlobalRatelimiter\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Rejected (shadow)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Ratelimiter\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 42\n      },\n      \"id\": 29,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 67\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(cadence_workflow_endtoend_latency_bucket{domain=~\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Workflow End to End Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 67\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(namespace, domain) (rate(cadence_worker_panic{domain=~\\\"$domain\\\"}[2m]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"NonDeterministicError and Worker Panic\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 75\n          },\n          \"id\": 27,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"sum(count(cadence_sticky_cache_total_forced_eviction{domain=\\\"$domain\\\"})) by (domain)\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Workflow Sticky Cache Eviction\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Workflow\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 43\n      },\n      \"id\": 34,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 340\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum(count(cadence_activity_task_failed{domain=\\\"$domain\\\"})) by (activitytype)\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Activity Task Operations\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 340\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"avg by(domain, tasklist, activitytype) (cadence_activity_execution_latency{domain=~\\\"$domain\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Execution Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 372\n          },\n          \"id\": 31,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [\n                \"mean\",\n                \"max\",\n                \"min\",\n                \"sum\",\n                \"lastNotNull\"\n              ],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"multi\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(domain, tasklist) (rate(cadence_requests_per_tl{cadence_service=\\\"cadence_matching\\\", domain=~\\\"$domain\\\"}[5m]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"ActivityTasks scheduled per Second\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 372\n          },\n          \"id\": 32,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.2\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"avg by(Domain, TaskList, ActivityType) (cadence_activity_scheduled_to_start_latency_count{domain=~\\\"$domain\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Scheduled To Start Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Activity\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 44\n      },\n      \"id\": 37,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 205\n          },\n          \"id\": 35,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"(1 - count(cadence_error{domain=\\\"$domain\\\"})by(domain)/count(cadence_request{domain=\\\"$domain\\\"})by(domain)) * 100\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Service API success rate\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 205\n          },\n          \"id\": 36,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"sum(avg_over_time(cadence_latency{Domain=\\\"$domain\\\"} [1m])) by (Domain)\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Service API Latency (TODO exclude long-poll APIs)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Service\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 45\n      },\n      \"id\": 24,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 214\n          },\n          \"id\": 22,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"histogram_quantile($quantiles, (max by (le, domain) (history_size_bucket)))*1000000000\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Max History Size\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 214\n          },\n          \"id\": 23,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by (domain) (history_count_sum{domain!=\\\"all\\\"}/history_count_count{domain!=\\\"all\\\"})\\n*1000000000\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Avg History Length \",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"showValues\": false,\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\",\n                    \"value\": 0\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 238\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"editorMode\": \"code\",\n              \"expr\": \"max(histogram_quantile(1, max(event_blob_size_bucket) by (le)))*1000000000\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Max Event Blob Size\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"History\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 46\n      },\n      \"id\": 38,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"custom\": {\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                }\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 215\n          },\n          \"id\": 40,\n          \"options\": {\n            \"calculate\": false,\n            \"cellGap\": 1,\n            \"color\": {\n              \"exponent\": 0.5,\n              \"fill\": \"dark-orange\",\n              \"mode\": \"scheme\",\n              \"reverse\": false,\n              \"scale\": \"exponential\",\n              \"scheme\": \"Oranges\",\n              \"steps\": 64\n            },\n            \"exemplars\": {\n              \"color\": \"rgba(255,0,255,0.7)\"\n            },\n            \"filterValues\": {\n              \"le\": 1e-9\n            },\n            \"legend\": {\n              \"show\": true\n            },\n            \"rowsFrame\": {\n              \"layout\": \"auto\"\n            },\n            \"tooltip\": {\n              \"mode\": \"single\",\n              \"showColorScale\": false,\n              \"yHistogram\": false\n            },\n            \"yAxis\": {\n              \"axisPlacement\": \"left\",\n              \"min\": 0,\n              \"reverse\": false,\n              \"unit\": \"short\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"exemplar\": false,\n              \"expr\": \"rate(cadence_concurrency_auto_scaler_poller_quota_bucket{WorkerType=\\\"ActivityWorker\\\",Domain=\\\"$domain\\\"}[$__rate_interval])\",\n              \"format\": \"heatmap\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"{{le}}\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Poller Quota\",\n          \"type\": \"heatmap\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"custom\": {\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                }\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 215\n          },\n          \"id\": 41,\n          \"options\": {\n            \"calculate\": false,\n            \"cellGap\": 1,\n            \"color\": {\n              \"exponent\": 0.5,\n              \"fill\": \"dark-orange\",\n              \"mode\": \"scheme\",\n              \"reverse\": false,\n              \"scale\": \"exponential\",\n              \"scheme\": \"Oranges\",\n              \"steps\": 64\n            },\n            \"exemplars\": {\n              \"color\": \"rgba(255,0,255,0.7)\"\n            },\n            \"filterValues\": {\n              \"le\": 1e-9\n            },\n            \"legend\": {\n              \"show\": true\n            },\n            \"rowsFrame\": {\n              \"layout\": \"auto\"\n            },\n            \"tooltip\": {\n              \"mode\": \"single\",\n              \"showColorScale\": false,\n              \"yHistogram\": false\n            },\n            \"yAxis\": {\n              \"axisPlacement\": \"left\",\n              \"min\": 0,\n              \"reverse\": false,\n              \"unit\": \"short\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"exemplar\": false,\n              \"expr\": \"rate(cadence_concurrency_auto_scaler_poller_quota_bucket{WorkerType=\\\"DecisionWorker\\\", Domain=\\\"$domain\\\"}[$__rate_interval])\",\n              \"format\": \"heatmap\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"{{le}}\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Poller Quota\",\n          \"type\": \"heatmap\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"lower wait time indicates busy pollers and thus need more poller to avoid high schedule to start latency\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"custom\": {\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                }\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 223\n          },\n          \"id\": 43,\n          \"options\": {\n            \"calculate\": false,\n            \"cellGap\": 1,\n            \"color\": {\n              \"exponent\": 0.5,\n              \"fill\": \"dark-orange\",\n              \"mode\": \"scheme\",\n              \"reverse\": false,\n              \"scale\": \"exponential\",\n              \"scheme\": \"Oranges\",\n              \"steps\": 64\n            },\n            \"exemplars\": {\n              \"color\": \"rgba(255,0,255,0.7)\"\n            },\n            \"filterValues\": {\n              \"le\": 1e-9\n            },\n            \"legend\": {\n              \"show\": true\n            },\n            \"rowsFrame\": {\n              \"layout\": \"auto\"\n            },\n            \"tooltip\": {\n              \"mode\": \"single\",\n              \"showColorScale\": false,\n              \"yHistogram\": false\n            },\n            \"yAxis\": {\n              \"axisPlacement\": \"left\",\n              \"reverse\": false,\n              \"unit\": \"ms\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"code\",\n              \"expr\": \"rate(cadence_concurrency_auto_scaler_poller_wait_time_bucket{WorkerType=\\\"ActivityWorker\\\",Domain=\\\"$domain\\\"}[ $__rate_interval])\",\n              \"instant\": false,\n              \"legendFormat\": \"{{le}}\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Activity Poller Wait Time\",\n          \"type\": \"heatmap\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"lower wait time indicates busy pollers and thus need more poller to avoid high schedule to start latency\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"custom\": {\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                }\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 223\n          },\n          \"id\": 42,\n          \"options\": {\n            \"calculate\": false,\n            \"cellGap\": 1,\n            \"color\": {\n              \"exponent\": 0.5,\n              \"fill\": \"dark-orange\",\n              \"mode\": \"scheme\",\n              \"reverse\": false,\n              \"scale\": \"exponential\",\n              \"scheme\": \"Oranges\",\n              \"steps\": 64\n            },\n            \"exemplars\": {\n              \"color\": \"rgba(255,0,255,0.7)\"\n            },\n            \"filterValues\": {\n              \"le\": 1e-9\n            },\n            \"legend\": {\n              \"show\": true\n            },\n            \"rowsFrame\": {\n              \"layout\": \"auto\"\n            },\n            \"tooltip\": {\n              \"mode\": \"single\",\n              \"showColorScale\": false,\n              \"yHistogram\": false\n            },\n            \"yAxis\": {\n              \"axisPlacement\": \"left\",\n              \"reverse\": false,\n              \"unit\": \"ms\"\n            }\n          },\n          \"pluginVersion\": \"12.2.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"code\",\n              \"expr\": \"rate(cadence_concurrency_auto_scaler_poller_wait_time_bucket{WorkerType=\\\"DecisionWorker\\\", Domain=\\\"$domain\\\"}[$__rate_interval])\",\n              \"instant\": false,\n              \"legendFormat\": \"{{le}}\",\n              \"range\": true,\n              \"refId\": \"A\"\n            }\n          ],\n          \"title\": \"Decision Poller Wait Time\",\n          \"type\": \"heatmap\"\n        }\n      ],\n      \"title\": \"Concurrency\",\n      \"type\": \"row\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 42,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"\",\n          \"value\": \"\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"0.99\",\n          \"value\": \"0.99\"\n        },\n        \"label\": \"Quantiles\",\n        \"name\": \"quantiles\",\n        \"options\": [\n          {\n            \"selected\": false,\n            \"text\": \"0.5\",\n            \"value\": \"0.5\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"0.95\",\n            \"value\": \"0.95\"\n          },\n          {\n            \"selected\": true,\n            \"text\": \"0.99\",\n            \"value\": \"0.99\"\n          }\n        ],\n        \"query\": \"0.5, 0.95, 0.99\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-6h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence-Client\",\n  \"uid\": \"dehkspwgabvuoc\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-frontend.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 3,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 6,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1\n          },\n          \"id\": 2,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Request Counts Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1\n          },\n          \"id\": 3,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_per_tl{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"rate(cadence_PollForActivityTask_cadence_error{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"D\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Error Breakdown\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 25\n          },\n          \"id\": 4,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Error Counts Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 25\n          },\n          \"id\": 5,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Request Vs Error\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 33\n          },\n          \"id\": 7,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend API Latencies\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 12,\n      \"panels\": [],\n      \"title\": \"Frontend-Client\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 2\n      },\n      \"id\": 8,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (request)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (error)\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Vs Error\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 2\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Counts Per API\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"ms\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 10\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client API Latencies\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 10\n      },\n      \"id\": 10,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_client_errors{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Frontend Client Request Counts Per API\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 18\n      },\n      \"id\": 27,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 19\n          },\n          \"id\": 13,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"StartWorkflowExecution\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"StartWorfklowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 19\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"StartWorkflowExecution\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"StartWorkflowExection Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 27\n          },\n          \"id\": 21,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, tasklist) (rate(cadence_requests_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests per operation per task\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 27\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, tasklist) (rate(workflow_success{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Workflow success per operation per task\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 35\n          },\n          \"id\": 22,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"RespondActivityTaskCompleted\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Respond Activity task completed\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"sec\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 35\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"RespondActivityTaskCompleted\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"RespondActivity Task Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 43\n          },\n          \"id\": 17,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(TaskList) (rate(cadence_PollForActivityTask_cadence_request[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"ms\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 43\n          },\n          \"id\": 23,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 51\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(TaskList) (rate(cadence_PollForDecisionTask_cadence_request[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 51\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 59\n          },\n          \"id\": 15,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 59\n          },\n          \"id\": 16,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 67\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"GetWorkflowExecutionHistory\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GetWorkflowExecutionHistory\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 67\n          },\n          \"id\": 29,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"CancelWorkflowExecution\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"CancelWorkflowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 75\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"TerminateWorkflowExecution\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TerminateWorkflowExecution\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 75\n          },\n          \"id\": 31,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"RegisterDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"RegisterDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 83\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DescribeDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DescribeDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 83\n          },\n          \"id\": 32,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"UpdateDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"UpdateDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 91\n          },\n          \"id\": 34,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"DepricateDomain\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"DepricateDomain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 91\n          },\n          \"id\": 35,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddDecisionTask\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"API\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 19\n      },\n      \"id\": 39,\n      \"panels\": [],\n      \"title\": \"Persistence\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 20\n      },\n      \"id\": 36,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"{{operation}} (request)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} (error)\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Vs Errors\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 20\n      },\n      \"id\": 37,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"sec\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 28\n      },\n      \"id\": 38,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, avg by(operation, le) (rate(persistence_latency_bucket{cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Operation Latencies p99\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 36\n      },\n      \"id\": 42,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"{le=\\\"1.0000000000000001e-07\\\", operation=\\\"StartWorkflowExecution\\\"}\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 317\n          },\n          \"id\": 40,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Overall Query Sizes\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"{le=\\\"1.0000000000000001e-07\\\", operation=\\\"StartWorkflowExecution\\\"}\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 317\n          },\n          \"id\": 41,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Query Sizes by Domain\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Query sizes\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 37\n      },\n      \"id\": 44,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"histogram_quantile(0.95, sum by(le) (rate(cadence_authorization_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[1m0s])))\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 318\n          },\n          \"id\": 43,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_authorization_latency_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Authorization latency per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Authorization\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 38\n      },\n      \"id\": 52,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 319\n          },\n          \"id\": 47,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_allocated{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Total Memory Allocation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"NumGC is the number of completed GC cycles.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 319\n          },\n          \"id\": 46,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_num_gc{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GC Num\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 327\n          },\n          \"id\": 45,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(num_goroutines{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Number of goroutines\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapInuse is bytes in in-use spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 327\n          },\n          \"id\": 49,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapinuse{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap in Use (HeapAlloc)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapIdle is bytes in idle (unused) spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 335\n          },\n          \"id\": 48,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapidle{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap Available (HeapIdle)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"Bytes of stack memory obtained from the OS. Plus any memory obtained directly from the OS for OS thread stacks.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 335\n          },\n          \"id\": 51,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_stack{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Stack In use\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Go Gc Detailed\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 39\n      },\n      \"id\": 57,\n      \"panels\": [],\n      \"title\": \"Replication\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 40\n      },\n      \"id\": 54,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"rate(cadence_requests{operation=\\\"GetReplicationMessage\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval])\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetReplicationMessage Requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 40\n      },\n      \"id\": 53,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"rate(cadence_requests{operation=\\\"GetDomainReplicationMessage\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval])\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetDomainReplicationMessage Requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 48\n      },\n      \"id\": 55,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le) (rate(cadence_latency_bucket{operation=\\\"GetReplicationMessage\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetReplicationMessage Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 48\n      },\n      \"id\": 56,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"code\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le) (rate(cadence_latency_bucket{operation=\\\"GetDomainReplicationMessage\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetDomainReplicationMessage Latency\",\n      \"type\": \"timeseries\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n        \"includeAll\": true,\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"cadence_frontend\",\n          \"value\": \"cadence_frontend\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"p99\",\n          \"value\": \"p99\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": true,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-5m\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence-Frontend\",\n  \"uid\": \"cekm8fs6ps000f\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-history.json",
    "content": "{\n    \"annotations\": {\n      \"list\": [\n        {\n          \"builtIn\": 1,\n          \"datasource\": {\n            \"type\": \"grafana\",\n            \"uid\": \"-- Grafana --\"\n          },\n          \"enable\": true,\n          \"hide\": true,\n          \"iconColor\": \"rgba(0, 211, 255, 1)\",\n          \"name\": \"Annotations & Alerts\",\n          \"type\": \"dashboard\"\n        }\n      ]\n    },\n    \"editable\": true,\n    \"fiscalYearStartMonth\": 0,\n    \"graphTooltip\": 0,\n    \"id\": 5,\n    \"links\": [],\n    \"panels\": [\n      {\n        \"collapsed\": false,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 0\n        },\n        \"id\": 9,\n        \"panels\": [],\n        \"title\": \"API\",\n        \"type\": \"row\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 0,\n          \"y\": 1\n        },\n        \"id\": 7,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.1\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"Request Per API\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"fieldConfig\": {\n          \"defaults\": {\n            \"color\": {\n              \"mode\": \"palette-classic\"\n            },\n            \"custom\": {\n              \"axisBorderShow\": false,\n              \"axisCenteredZero\": false,\n              \"axisColorMode\": \"text\",\n              \"axisLabel\": \"\",\n              \"axisPlacement\": \"auto\",\n              \"barAlignment\": 0,\n              \"barWidthFactor\": 0.6,\n              \"drawStyle\": \"line\",\n              \"fillOpacity\": 0,\n              \"gradientMode\": \"none\",\n              \"hideFrom\": {\n                \"legend\": false,\n                \"tooltip\": false,\n                \"viz\": false\n              },\n              \"insertNulls\": false,\n              \"lineInterpolation\": \"linear\",\n              \"lineWidth\": 1,\n              \"pointSize\": 5,\n              \"scaleDistribution\": {\n                \"type\": \"linear\"\n              },\n              \"showPoints\": \"auto\",\n              \"spanNulls\": false,\n              \"stacking\": {\n                \"group\": \"A\",\n                \"mode\": \"none\"\n              },\n              \"thresholdsStyle\": {\n                \"mode\": \"off\"\n              }\n            },\n            \"mappings\": [],\n            \"thresholds\": {\n              \"mode\": \"absolute\",\n              \"steps\": [\n                {\n                  \"color\": \"green\"\n                },\n                {\n                  \"color\": \"red\",\n                  \"value\": 80\n                }\n              ]\n            }\n          },\n          \"overrides\": []\n        },\n        \"gridPos\": {\n          \"h\": 8,\n          \"w\": 12,\n          \"x\": 12,\n          \"y\": 1\n        },\n        \"id\": 8,\n        \"options\": {\n          \"legend\": {\n            \"calcs\": [],\n            \"displayMode\": \"list\",\n            \"placement\": \"bottom\",\n            \"showLegend\": true\n          },\n          \"tooltip\": {\n            \"hideZeros\": false,\n            \"mode\": \"single\",\n            \"sort\": \"none\"\n          }\n        },\n        \"pluginVersion\": \"12.0.1\",\n        \"targets\": [\n          {\n            \"disableTextWrap\": false,\n            \"editorMode\": \"builder\",\n            \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n            \"fullMetaSearch\": false,\n            \"includeNullMetadata\": false,\n            \"legendFormat\": \"__auto\",\n            \"range\": true,\n            \"refId\": \"A\",\n            \"useBackend\": false\n          }\n        ],\n        \"title\": \"History API p99 Latencies\",\n        \"type\": \"timeseries\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 9\n        },\n        \"id\": 26,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 21,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(complete_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(schedule_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(fail_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"C\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"D\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(start_timer_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"E\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"F\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_timer_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"G\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_activity_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"H\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(record_marker_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"I\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cancel_external_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"J\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(continue_as_new_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"K\"\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(child_workflow_decision{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"L\"\n              }\n            ],\n            \"title\": \"Decision Breakdown\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 22,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(multiple_completion_decisions{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": true,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(failed_decisions{operation=\\\"RespondDecisionTaskCompleted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"hide\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\"\n              }\n            ],\n            \"title\": \"Bad Decisions\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Decisions\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 10\n        },\n        \"id\": 14,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 10,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(cadence_service) (rate(cadence_client_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{cadence_service}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(cadence_service) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{cadence_service}} (error)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients Requests Vs Errors\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 11,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients Request Count Per API\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 513\n            },\n            \"id\": 12,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (changes(cadence_client_errors{cadence_service=\\\"cadence_history\\\"}[$__interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Client Error Counts Per API\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 513\n            },\n            \"id\": 13,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"History Clients API Latencies\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"History Clients\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 11\n        },\n        \"id\": 20,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 505\n            },\n            \"id\": 15,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(cache_requests{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_miss{cadence_service=\\\"cadence_history\\\", cache_type=\\\"mutablestate\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (miss)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Mutable State Cache\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 505\n            },\n            \"id\": 16,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cache_latency_bucket{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Mutable State Cache Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 513\n            },\n            \"id\": 17,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_requests{cache_type=\\\"events\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (request)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(cache_miss{cadence_service=\\\"cadence_history\\\", cache_type=\\\"events\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (miss)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Events Cache\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 513\n            },\n            \"id\": 18,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cache_latency_bucket{cache_type=\\\"mutablestate\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Events Cache Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"WorkflowContext\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 521\n            },\n            \"id\": 19,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(workflow_context_lock_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Execution Lock Held Duration\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"History Caches\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 12\n        },\n        \"id\": 30,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 9\n            },\n            \"id\": 23,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(acquire_shards_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardController\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Shard Movement\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 9\n            },\n            \"id\": 27,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(get_engine_for_shard_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (get_engine)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_per_shard_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (persistence_latency_per_shard)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"p99 Latencies\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (shard info replication)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 457\n            },\n            \"id\": 25,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(shardinfo_transfer_lag_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardInfo\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"{{operation}} (shard info transfer)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(shardinfo_replication_lag_count{cadence_service=\\\"cadence_history\\\", operation=\\\"ShardInfo\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (shard info replication)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Replication / Transfer p99 Lag (buggy)\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 457\n            },\n            \"id\": 24,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_lag_bucket{operation=\\\"ShardInfo\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Timer p99 Lag\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 465\n            },\n            \"id\": 28,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_diff_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Transfer Level p99 Diff\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                },\n                \"unit\": \"ms\"\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 465\n            },\n            \"id\": 29,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_diff_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Timer Level p99 Diff\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (timer failover)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 473\n            },\n            \"id\": 31,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_failover_in_progress_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (transfer failover)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_failover_in_progress_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (timer failover)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Failover p99 In Progress\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"ShardInfo (transfer active pending task)\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 473\n            },\n            \"id\": 32,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_transfer_active_pending_task_bucket[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (transfer active pending task)\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(shardinfo_timer_active_pending_task_bucket[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": true,\n                \"instant\": false,\n                \"legendFormat\": \"{{operation}} (timer active pending task)\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Pending p99 Tasks\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"ShardController\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 13\n        },\n        \"id\": 38,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 14\n            },\n            \"id\": 33,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(__name__) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"persistence_requests\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              },\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"persistence_errors\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Requests vs Errors\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 14\n            },\n            \"id\": 34,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"EntityNotExists\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Error Breakdown\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 118\n            },\n            \"id\": 35,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Requests Per Operation\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 118\n            },\n            \"id\": 36,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Error Per Operation\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 126\n            },\n            \"id\": 37,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"avg by(operation) (rate(persistence_latency_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"B\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Operation p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Persistence\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 14\n        },\n        \"id\": 47,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 15\n            },\n            \"id\": 39,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TimerActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"TimerActiveQueueProcessor\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 15\n            },\n            \"id\": 40,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TimerStandByQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 23\n            },\n            \"id\": 41,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(new_timer_notifications{operation=\\\"TimerActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 23\n            },\n            \"id\": 42,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"rate(new_timer_notifications{operation=\\\"TimerStandByQueueProcessor\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby New Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 31\n            },\n            \"id\": 43,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(task_requests{operation=\\\"TimerActiveTaskActivityTimeout\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Requests\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 31\n            },\n            \"id\": 44,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_processing_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Processing p99 Latency\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 39\n            },\n            \"id\": 46,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_attempt_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer p99 Attempt\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 39\n            },\n            \"id\": 45,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_queue_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Queue p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Timer Processor\",\n        \"type\": \"row\"\n      },\n      {\n        \"collapsed\": true,\n        \"gridPos\": {\n          \"h\": 1,\n          \"w\": 24,\n          \"x\": 0,\n          \"y\": 15\n        },\n        \"id\": 48,\n        \"panels\": [\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 16\n            },\n            \"id\": 49,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TransferActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": [\n                {\n                  \"__systemRef\": \"hideSeriesFrom\",\n                  \"matcher\": {\n                    \"id\": \"byNames\",\n                    \"options\": {\n                      \"mode\": \"exclude\",\n                      \"names\": [\n                        \"TimerActiveQueueProcessor\"\n                      ],\n                      \"prefix\": \"All except:\",\n                      \"readOnly\": true\n                    }\n                  },\n                  \"properties\": [\n                    {\n                      \"id\": \"custom.hideFrom\",\n                      \"value\": {\n                        \"legend\": false,\n                        \"tooltip\": false,\n                        \"viz\": true\n                      }\n                    }\n                  ]\n                }\n              ]\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 16\n            },\n            \"id\": 50,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(ack_level_update{cadence_service=\\\"cadence_history\\\", operation=\\\"TransferStandByQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby Ack Level Update\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 24\n            },\n            \"id\": 51,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"builder\",\n                \"expr\": \"sum by(operation) (rate(new_timer_notifications{operation=\\\"TransferActiveQueueProcessor\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 24\n            },\n            \"id\": 52,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"rate(new_timer_notifications{operation=\\\"TranferStandByQueueProcessor\\\"}[$__rate_interval])\",\n                \"fullMetaSearch\": false,\n                \"includeNullMetadata\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Standby New Timer Notifications\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 32\n            },\n            \"id\": 53,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"sum by(operation) (rate(task_requests{operation=\\\"TransferActiveTaskActivityTimeout\\\"}[$__rate_interval]))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Requests\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 32\n            },\n            \"id\": 54,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_processing_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Processing p99 Latency\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 0,\n              \"y\": 40\n            },\n            \"id\": 55,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_attempt_bucket{operation=~\\\"TransferActiveTaskActivityTimeout|TransferActiveTaskDecisionTimeout|TransferActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer p99 Attempt\",\n            \"type\": \"timeseries\"\n          },\n          {\n            \"datasource\": {\n              \"type\": \"prometheus\",\n              \"uid\": \"PBFA97CFB590B2093\"\n            },\n            \"fieldConfig\": {\n              \"defaults\": {\n                \"color\": {\n                  \"mode\": \"palette-classic\"\n                },\n                \"custom\": {\n                  \"axisBorderShow\": false,\n                  \"axisCenteredZero\": false,\n                  \"axisColorMode\": \"text\",\n                  \"axisLabel\": \"\",\n                  \"axisPlacement\": \"auto\",\n                  \"barAlignment\": 0,\n                  \"barWidthFactor\": 0.6,\n                  \"drawStyle\": \"line\",\n                  \"fillOpacity\": 0,\n                  \"gradientMode\": \"none\",\n                  \"hideFrom\": {\n                    \"legend\": false,\n                    \"tooltip\": false,\n                    \"viz\": false\n                  },\n                  \"insertNulls\": false,\n                  \"lineInterpolation\": \"linear\",\n                  \"lineWidth\": 1,\n                  \"pointSize\": 5,\n                  \"scaleDistribution\": {\n                    \"type\": \"linear\"\n                  },\n                  \"showPoints\": \"auto\",\n                  \"spanNulls\": false,\n                  \"stacking\": {\n                    \"group\": \"A\",\n                    \"mode\": \"none\"\n                  },\n                  \"thresholdsStyle\": {\n                    \"mode\": \"off\"\n                  }\n                },\n                \"mappings\": [],\n                \"thresholds\": {\n                  \"mode\": \"absolute\",\n                  \"steps\": [\n                    {\n                      \"color\": \"green\"\n                    },\n                    {\n                      \"color\": \"red\",\n                      \"value\": 80\n                    }\n                  ]\n                }\n              },\n              \"overrides\": []\n            },\n            \"gridPos\": {\n              \"h\": 8,\n              \"w\": 12,\n              \"x\": 12,\n              \"y\": 40\n            },\n            \"id\": 56,\n            \"options\": {\n              \"legend\": {\n                \"calcs\": [],\n                \"displayMode\": \"list\",\n                \"placement\": \"bottom\",\n                \"showLegend\": true\n              },\n              \"tooltip\": {\n                \"hideZeros\": false,\n                \"mode\": \"single\",\n                \"sort\": \"none\"\n              }\n            },\n            \"pluginVersion\": \"12.0.1\",\n            \"targets\": [\n              {\n                \"datasource\": {\n                  \"type\": \"prometheus\",\n                  \"uid\": \"PBFA97CFB590B2093\"\n                },\n                \"disableTextWrap\": false,\n                \"editorMode\": \"code\",\n                \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(task_latency_queue_bucket{operation=~\\\"TransferActiveTaskActivityTimeout|TransferActiveTaskDecisionTimeout|TransferActiveTaskWorkflowTimeout\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n                \"fullMetaSearch\": false,\n                \"hide\": false,\n                \"includeNullMetadata\": false,\n                \"instant\": false,\n                \"legendFormat\": \"__auto\",\n                \"range\": true,\n                \"refId\": \"A\",\n                \"useBackend\": false\n              }\n            ],\n            \"title\": \"Active Timer Queue p99 Latency\",\n            \"type\": \"timeseries\"\n          }\n        ],\n        \"title\": \"Transfer Processor\",\n        \"type\": \"row\"\n      }\n    ],\n    \"preload\": false,\n    \"schemaVersion\": 41,\n    \"tags\": [],\n    \"templating\": {\n      \"list\": [\n        {\n          \"current\": {\n            \"text\": \"samples_domain\",\n            \"value\": \"samples_domain\"\n          },\n          \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"includeAll\": true,\n          \"label\": \"Domain\",\n          \"name\": \"domain\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"cadence_history\",\n            \"value\": \"cadence_history\"\n          },\n          \"definition\": \"label_values(cadence_requests,cadence_service)\",\n          \"label\": \"services\",\n          \"name\": \"services\",\n          \"options\": [],\n          \"query\": {\n            \"qryType\": 1,\n            \"query\": \"label_values(cadence_requests,cadence_service)\",\n            \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n          },\n          \"refresh\": 1,\n          \"regex\": \"\",\n          \"type\": \"query\"\n        },\n        {\n          \"current\": {\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          \"label\": \"latency\",\n          \"name\": \"latency\",\n          \"options\": [\n            {\n              \"selected\": true,\n              \"text\": \"p99\",\n              \"value\": \"p99\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"upper\",\n              \"value\": \"upper\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"mean\",\n              \"value\": \"mean\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"median\",\n              \"value\": \"median\"\n            },\n            {\n              \"selected\": false,\n              \"text\": \"lower\",\n              \"value\": \"lower\"\n            }\n          ],\n          \"query\": \"p99,upper,mean,median,lower\",\n          \"type\": \"custom\"\n        }\n      ]\n    },\n    \"time\": {\n      \"from\": \"now-5m\",\n      \"to\": \"now\"\n    },\n    \"timepicker\": {},\n    \"timezone\": \"browser\",\n    \"title\": \"Cadence-History\",\n    \"uid\": \"cekm8fs6ps000h\",\n    \"version\": 1\n  }"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-matching.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"grafana\",\n          \"uid\": \"-- Grafana --\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 3,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 6,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1\n          },\n          \"id\": 1,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Overall Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1\n          },\n          \"id\": 2,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Breakdown\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 113\n          },\n          \"id\": 3,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 113\n          },\n          \"id\": 4,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"API Latencies (p99)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 121\n          },\n          \"id\": 5,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(tasklist_managers{cadence_service=\\\"cadence_matching\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task Managers Per Operation\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 15,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 2\n          },\n          \"id\": 7,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 2\n          },\n          \"id\": 8,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"PollForActivityTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 114\n          },\n          \"id\": 9,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(cadence_errors{operation=\\\"PollForDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"hide\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\"\n            }\n          ],\n          \"title\": \"PollForDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 114\n          },\n          \"id\": 10,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"PollForDecisionTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 122\n          },\n          \"id\": 11,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 122\n          },\n          \"id\": 13,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"AddActivityTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 130\n          },\n          \"id\": 12,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"AddDecisionTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 130\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"$services\\\", operation=\\\"AddDecisionTask\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"API\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 2\n      },\n      \"id\": 23,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"sec\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 3\n          },\n          \"id\": 16,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (changes(poll_success_per_tl{operation=\\\"TaskListMgr\\\", domain=\\\"$domain\\\", \\\"$services\\\"=\\\"$services\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"{{operation}} (async)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (changes(poll_success_sync_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (sync)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(poll_timeouts_per_tl{operation=\\\"TaskListMgr\\\", cadence_service=\\\"$services\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (poll_timeouts)\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskPoll\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 3\n          },\n          \"id\": 17,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by() (rate(cadence_PollForActivityTask_cadence_request{cadence_service=\\\"$services\\\", \\\"$domain\\\"=\\\"$domain\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task Matcher Stats\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 115\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, avg by(le, operation) (rate(cadence_latency_per_tl_bucket{operation=\\\"AddActivityTask\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddActivityTask Latency(p99) (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 115\n          },\n          \"id\": 18,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum(rate(cadence_PollForDecisionTask_cadence_request{cadence_service=\\\"$services\\\", \\\"$services\\\"=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task Matcher Stat\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 123\n          },\n          \"id\": 20,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(cadence_latency_per_tl_bucket{operation=\\\"AddDecisionTask\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"AddDecisionTask Latency(p99) (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 123\n          },\n          \"id\": 22,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=\\\"PollForActivityTask\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"PollForActivityTask QPS (forwarding requests excluded)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"TaskListManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 3\n      },\n      \"id\": 27,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 4\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 4\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Operation Latencies p99\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"GetTaskListSize\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 100\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Persistence\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 4\n      },\n      \"id\": 31,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 5\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_client_requests{cadence_role=\\\"history_client\\\", cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per API\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 5\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(cadence_client_latency_bucket{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"API Latency (p99)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 69\n          },\n          \"id\": 29,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_client_errors_redirection{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per API\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"History Client\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 5\n      },\n      \"id\": 34,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 6\n          },\n          \"id\": 32,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (changes(task_count_per_tl{tasklistType=\\\"activity\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task Backlog\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 6\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (changes(task_backlog_per_tl{tasklistType=\\\"decision\\\", cadence_service=\\\"cadence_matching\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task Backlog\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Backlog (per task list)\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 6\n      },\n      \"id\": 37,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 7\n          },\n          \"id\": 35,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_num_gc{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"GC Num\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 7\n          },\n          \"id\": 36,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_allocated{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Total allocation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 39\n          },\n          \"id\": 38,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(num_goroutines{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Number of goroutines\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapInuse is bytes in in-use spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 39\n          },\n          \"id\": 39,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapinuse{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap in Use (HeapAlloc)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"HeapIdle is bytes in idle (unused) spans.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 47\n          },\n          \"id\": 40,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_heapidle{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Heap Available (HeapIdle)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"description\": \"Bytes of stack memory obtained from the OS. Plus any memory obtained directly from the OS for OS thread stacks.\",\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 47\n          },\n          \"id\": 41,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(cadence_service) (rate(memory_stack{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Stack In use\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Go GC detailed\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 7\n      },\n      \"id\": 48,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 8\n          },\n          \"id\": 42,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(tasklist) (rate(estimated_add_task_qps_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Adaptive Task List Scaler for Activity\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 8\n          },\n          \"id\": 43,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(estimated_add_task_qps_per_tl{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Adaptive Task List Scaler for Decision\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 16\n          },\n          \"id\": 44,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_read{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task List Read Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 16\n          },\n          \"id\": 45,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_read{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task List Read Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 24\n          },\n          \"id\": 46,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_write{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"activity\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activity Task List Write Partition Count\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 24\n          },\n          \"id\": 47,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(tasklist) (rate(task_list_partition_config_num_write{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", tasklistType=\\\"decision\\\", operation=\\\"TaskListMgr\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Task List Write Partition Count\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Task List Partition Config & Adaptive Task List Scaler (WIP)\",\n      \"type\": \"row\"\n    }\n  ],\n  \"preload\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_bucket,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"cadence_matching\",\n          \"value\": \"cadence_matching\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"current\": {\n          \"text\": \"p99\",\n          \"value\": \"p99\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": true,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-1h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"browser\",\n  \"title\": \"Cadence Matching\",\n  \"uid\": \"del7jwba0qgw0e\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-persistence.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"prometheus\",\n          \"uid\": \"PBFA97CFB590B2093\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"target\": {\n          \"limit\": 100,\n          \"matchAny\": false,\n          \"tags\": [],\n          \"type\": \"dashboard\"\n        },\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 2,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 12,\n      \"panels\": [],\n      \"title\": \"Overall\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 17,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_response_row_size_count{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Response Row Count Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 1\n      },\n      \"id\": 18,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_response_payload_size_sum{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Response Payload Size Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 12\n      },\n      \"id\": 8,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (rate(persistence_requests_per_domain{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 12\n      },\n      \"id\": 2,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_requests{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests per operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          },\n          \"unit\": \"s\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 23\n      },\n      \"id\": 3,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"max\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Max\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(persistence_latency_bucket{cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Latencies Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 23\n      },\n      \"id\": 10,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (rate(persistence_requests_per_domain{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Domain Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 9,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"fieldMinMax\": false,\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 34\n      },\n      \"id\": 1,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_empty_response{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Empty Response Per Operation\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 34\n      },\n      \"id\": 9,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation, shard_id) (rate(persistence_requests_per_shard{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"instant\": false,\n          \"legendFormat\": \"{{operation}} - {{shard_id}}(shard_id)\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Requests Per Operation Per Shard\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 11,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 45\n      },\n      \"id\": 11,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"sum\"\n          ],\n          \"displayMode\": \"table\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true,\n          \"sortBy\": \"Total\",\n          \"sortDesc\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Errors Per Operation \",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 56\n      },\n      \"id\": 16,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 160\n          },\n          \"id\": 14,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"right\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"asc\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(le, operation) (rate(persistence_requests{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|LeaseTaskList|UpdateTaskList|CreateShard|UpdateShard\\\", cadence_service=\\\"$services\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"LWT Requests\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 160\n          },\n          \"id\": 15,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"right\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.9, max by(operation, le) (rate(persistence_latency_bucket{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|LeaseTaskList|UpdateTaskList|CreateShard|UpdateShard\\\", cadence_service=\\\"$services\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"interval\": \"\",\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"LWT Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"LWT Operations\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 57\n      },\n      \"id\": 22,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 161\n          },\n          \"id\": 19,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, cadence_service) (rate(persistence_requests{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Requests Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 161\n          },\n          \"id\": 21,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation, cadence_service) (rate(persistence_empty_response{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Empty Response Per Operation\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 193\n          },\n          \"id\": 20,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{operation=~\\\"ReadHistoryBranch|AppendHistoryNodes|GetHistoryTasks\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Errors Per Operation\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"History Tasks\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 58\n      },\n      \"id\": 29,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 162\n          },\n          \"id\": 23,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 162\n          },\n          \"id\": 24,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateShard|GetShard|UpdateShard\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard CRUD Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 194\n          },\n          \"id\": 25,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 194\n          },\n          \"id\": 26,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CompleteTask|CreateTask|GetTaskListSize\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Task CRUD Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 202\n          },\n          \"id\": 27,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_empty_response{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskList CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 202\n          },\n          \"id\": 28,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"LeaseTaskList|UpdateTaskList|GetTaskListSize\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"TaskList CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"ShardMgr & TaskMgr\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 59\n      },\n      \"id\": 32,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 163\n          },\n          \"id\": 30,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval]))\",\n              \"hide\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\"\n            }\n          ],\n          \"title\": \"WorkflowExecution CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"s\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 163\n          },\n          \"id\": 31,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateWorkflowExecution|UpdateWorkflowExecution|GetWorkflowExecution|DeleteWorkflowExecution\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"WorkflowExecution CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Execution Manager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 60\n      },\n      \"id\": 35,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 164\n          },\n          \"id\": 33,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(persistence_requests{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(persistence_errors_entity_not_exists{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Domain CRUD\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"min\": 4,\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 164\n          },\n          \"id\": 34,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"CreateDomain|DeleteDomain|UpdateDomain|DeleteDomainByName|GetDomain\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Domain CRUD Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"MetadataManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 61\n      },\n      \"id\": 38,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 165\n          },\n          \"id\": 36,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_requests{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"{{operation}} (request)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(operation) (rate(persistence_errors_entity_not_exists{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (error)\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Visibility Open/Close\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"dtdurationms\"\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 165\n          },\n          \"id\": 37,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"12.0.1\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_bucket{operation=~\\\"RecordWorkflowExecutionStarted|RecordWorkflowExecutionClosed\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Visibility Open/Close Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"VisibilityManager\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 62\n      },\n      \"id\": 45,\n      \"panels\": [],\n      \"title\": \"Persistence Hot Shard Detection\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 63\n      },\n      \"id\": 44,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation, shard_id) (rate(persistence_requests_per_shard{operation=\\\"UpdateWorkflowExecution\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Total updateworkflowexecution tracked shardId based requests\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 63\n      },\n      \"id\": 39,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(task_requests_per_domain{cadence_service=\\\"$services\\\", domain=\\\"$domain\\\", operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Timer Task Request Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 71\n      },\n      \"id\": 40,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(task_requests_per_domain{domain=\\\"$domain\\\", cadence_service=\\\"$services\\\", operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Transfer Task Request Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 71\n      },\n      \"id\": 43,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, shard_id) (rate(persistence_latency_per_shard_bucket{operation=\\\"ShardIdPersistenceRequest\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"shardidpersistencerequest p99 latency shardId\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 0,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"auto\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"s\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 79\n      },\n      \"id\": 41,\n      \"options\": {\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"single\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"12.0.1\",\n      \"targets\": [\n        {\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (rate(persistence_latency_per_domain_bucket{domain=~\\\"$domain\\\", cadence_service=~\\\"$services\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Persistence Latency Per Domain\",\n      \"type\": \"timeseries\"\n    }\n  ],\n  \"preload\": false,\n  \"refresh\": \"\",\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"allowCustomValue\": false,\n        \"current\": {\n          \"text\": \"All\",\n          \"value\": \"$__all\"\n        },\n        \"definition\": \"label_values(cadence_requests,cadence_service)\",\n        \"includeAll\": false,\n        \"label\": \"services\",\n        \"name\": \"services\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(cadence_requests,cadence_service)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      },\n      {\n        \"allowCustomValue\": false,\n        \"current\": {\n          \"text\": \"upper\",\n          \"value\": \"upper\"\n        },\n        \"label\": \"latency\",\n        \"name\": \"latency\",\n        \"options\": [\n          {\n            \"selected\": false,\n            \"text\": \"p99\",\n            \"value\": \"p99\"\n          },\n          {\n            \"selected\": true,\n            \"text\": \"upper\",\n            \"value\": \"upper\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"mean\",\n            \"value\": \"mean\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"median\",\n            \"value\": \"median\"\n          },\n          {\n            \"selected\": false,\n            \"text\": \"lower\",\n            \"value\": \"lower\"\n          }\n        ],\n        \"query\": \"p99,upper,mean,median,lower\",\n        \"type\": \"custom\"\n      },\n      {\n        \"allowCustomValue\": true,\n        \"current\": {\n          \"text\": \"samples_domain\",\n          \"value\": \"samples_domain\"\n        },\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-6h\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {},\n  \"timezone\": \"\",\n  \"title\": \"Cadence-Persistence\",\n  \"uid\": \"b00f7dd5-3dd6-4cf9-b17b-425e071e2b5b\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/cadence-server.json",
    "content": "{\n  \"annotations\": {\n    \"list\": [\n      {\n        \"builtIn\": 1,\n        \"datasource\": {\n          \"type\": \"datasource\",\n          \"uid\": \"grafana\"\n        },\n        \"enable\": true,\n        \"hide\": true,\n        \"iconColor\": \"rgba(0, 211, 255, 1)\",\n        \"name\": \"Annotations & Alerts\",\n        \"target\": {\n          \"limit\": 100,\n          \"matchAny\": false,\n          \"tags\": [],\n          \"type\": \"dashboard\"\n        },\n        \"type\": \"dashboard\"\n      }\n    ]\n  },\n  \"editable\": true,\n  \"fiscalYearStartMonth\": 0,\n  \"graphTooltip\": 0,\n  \"id\": 7,\n  \"links\": [],\n  \"panels\": [\n    {\n      \"collapsed\": false,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 0\n      },\n      \"id\": 44,\n      \"panels\": [],\n      \"title\": \"Frontend\",\n      \"type\": \"row\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 1\n      },\n      \"id\": 57,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"max\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"SignalWithStartWorkflowExecution|SignalWorkflowxecution|StartWorkflowExecution|TerminateWorkflowExecution|ResetWorkflowExecution|RequestCancelWorkflowexecution|ListWorkflowExecutions\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"\",\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Regular Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 7\n      },\n      \"id\": 58,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"PollForDecisionTask|GetWorkflowExecutionHistory|RespondDecisionTaskCompleted\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Workflow Worker Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 13\n      },\n      \"id\": 60,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (cadence_requests{cadence_service=\\\"cadence_frontend\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for All Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          }\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 6,\n        \"w\": 23,\n        \"x\": 0,\n        \"y\": 19\n      },\n      \"id\": 59,\n      \"options\": {\n        \"displayMode\": \"gradient\",\n        \"legend\": {\n          \"calcs\": [],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": false\n        },\n        \"maxVizHeight\": 300,\n        \"minVizHeight\": 16,\n        \"minVizWidth\": 8,\n        \"namePlacement\": \"auto\",\n        \"orientation\": \"auto\",\n        \"reduceOptions\": {\n          \"calcs\": [\n            \"mean\"\n          ],\n          \"fields\": \"\",\n          \"values\": false\n        },\n        \"showUnfilled\": true,\n        \"sizing\": \"auto\",\n        \"text\": {},\n        \"valueMode\": \"color\"\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(cadence_requests{cadence_service=\\\"cadence_frontend\\\",namespace=\\\"$namespace\\\", operation=~\\\"PollForActivityTask|RespondActivityTaskCompleted\\\"}) by (domain)\",\n          \"hide\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{domain}}{{operation}}\",\n          \"refId\": \"B\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"cadence_requests{cadence_service=\\\"cadence_worker\\\",namespace=\\\"$namespace\\\"}\",\n          \"hide\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"\",\n          \"refId\": \"A\"\n        }\n      ],\n      \"title\": \"Top Domains for Activity Worker Requests\",\n      \"type\": \"bargauge\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 25\n      },\n      \"id\": 48,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"All API per second per domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 25\n      },\n      \"id\": 49,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_requests{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": false,\n          \"instant\": false,\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"API per second(breakdown per operation)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 33\n      },\n      \"id\": 50,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation!~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"{{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Regular API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 33\n      },\n      \"id\": 51,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\", cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"ListWorkflow API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"ms\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 41\n      },\n      \"id\": 52,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForDecisionTask\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{operation=\\\"PollForActivityTask\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Long Poll API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 41\n      },\n      \"id\": 56,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation!~\\\"CountWorkflowExecutions|GetWorkflowExecutionHistory|ListClosedWorkflowExecutions|ListOpenWorkflowExecutions|ListWorkflowExecutions|PollForActivityTask|PollForDecisionTask|QueryWorkflow\\\"})\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"Domain: {{domain}}- Operation: {{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"Regular API Per Domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 0,\n        \"y\": 49\n      },\n      \"id\": 54,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(operation) (rate(cadence_errors{cadence_service=\\\"cadence_frontend\\\", domain=\\\"$domain\\\"}[$__rate_interval]))\",\n          \"fullMetaSearch\": false,\n          \"includeNullMetadata\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"Internal server error\",\n          \"range\": true,\n          \"refId\": \"A\",\n          \"useBackend\": false\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_bad_request{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"hide\": false,\n          \"interval\": \"\",\n          \"legendFormat\": \"Bad Requests error\",\n          \"refId\": \"B\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_query_failed{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"QueryFailed error\",\n          \"refId\": \"C\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_context_timeout{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"ContextTimeout error\",\n          \"refId\": \"D\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_entity_not_exists{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"EntityNotExists error\",\n          \"refId\": \"E\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_execution_already_started{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"WorkflowAlreadyStarted error\",\n          \"refId\": \"F\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"expr\": \"sum(rate(cadence_errors_workflow_execution_already_completed{cadence_service=\\\"cadence_frontend\\\"}[2m]))\",\n          \"interval\": \"\",\n          \"legendFormat\": \"WorkflowAlreadyCompleted\",\n          \"refId\": \"G\"\n        }\n      ],\n      \"title\": \"API  errors per second(breakdown per operation)\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 49\n      },\n      \"id\": 53,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"histogram_quantile(0.9, sum by(le, operation) (rate(cadence_latency_bucket{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"GetWorkflowExecutionHistory|QueryWorkflow\\\"}[$__rate_interval])))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"__auto\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"GetHistory/QueryWorkflow API Latency\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"datasource\": {\n        \"type\": \"prometheus\",\n        \"uid\": \"PBFA97CFB590B2093\"\n      },\n      \"fieldConfig\": {\n        \"defaults\": {\n          \"color\": {\n            \"mode\": \"palette-classic\"\n          },\n          \"custom\": {\n            \"axisBorderShow\": false,\n            \"axisCenteredZero\": false,\n            \"axisColorMode\": \"text\",\n            \"axisLabel\": \"\",\n            \"axisPlacement\": \"auto\",\n            \"barAlignment\": 0,\n            \"barWidthFactor\": 0.6,\n            \"drawStyle\": \"line\",\n            \"fillOpacity\": 10,\n            \"gradientMode\": \"none\",\n            \"hideFrom\": {\n              \"legend\": false,\n              \"tooltip\": false,\n              \"viz\": false\n            },\n            \"insertNulls\": false,\n            \"lineInterpolation\": \"linear\",\n            \"lineWidth\": 1,\n            \"pointSize\": 5,\n            \"scaleDistribution\": {\n              \"type\": \"linear\"\n            },\n            \"showPoints\": \"never\",\n            \"spanNulls\": false,\n            \"stacking\": {\n              \"group\": \"A\",\n              \"mode\": \"none\"\n            },\n            \"thresholdsStyle\": {\n              \"mode\": \"off\"\n            }\n          },\n          \"links\": [],\n          \"mappings\": [],\n          \"thresholds\": {\n            \"mode\": \"absolute\",\n            \"steps\": [\n              {\n                \"color\": \"green\"\n              },\n              {\n                \"color\": \"red\",\n                \"value\": 80\n              }\n            ]\n          },\n          \"unit\": \"short\"\n        },\n        \"overrides\": []\n      },\n      \"gridPos\": {\n        \"h\": 8,\n        \"w\": 12,\n        \"x\": 12,\n        \"y\": 57\n      },\n      \"id\": 55,\n      \"options\": {\n        \"alertThreshold\": true,\n        \"legend\": {\n          \"calcs\": [\n            \"mean\",\n            \"lastNotNull\",\n            \"max\",\n            \"min\",\n            \"sum\"\n          ],\n          \"displayMode\": \"list\",\n          \"placement\": \"bottom\",\n          \"showLegend\": true\n        },\n        \"tooltip\": {\n          \"hideZeros\": false,\n          \"mode\": \"multi\",\n          \"sort\": \"none\"\n        }\n      },\n      \"pluginVersion\": \"11.6.0\",\n      \"targets\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"disableTextWrap\": false,\n          \"editorMode\": \"builder\",\n          \"expr\": \"sum by(domain, operation) (changes(cadence_requests{cadence_service=\\\"cadence_frontend\\\", operation=~\\\"SignalWithStartworkflowExecution|SignalWorkflowExecution|StartWorkflowExecution|TerminateWorkflowExecution|ResetWorkflowExecution|RequestCancelWorkflowExecution|ListWorkflowExecutions\\\", domain=\\\"$domain\\\"}[$__interval]))\",\n          \"fullMetaSearch\": false,\n          \"hide\": false,\n          \"includeNullMetadata\": true,\n          \"interval\": \"\",\n          \"legendFormat\": \"Domain: {{domain}}- Operation: {{operation}}\",\n          \"range\": true,\n          \"refId\": \"B\",\n          \"useBackend\": false\n        }\n      ],\n      \"title\": \"WorkflowClient API per seconds by domain\",\n      \"type\": \"timeseries\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 65\n      },\n      \"id\": 96,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 641\n          },\n          \"id\": 93,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_service) (rate(restarts[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Service Restarts\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 641\n          },\n          \"id\": 95,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(acquire_shards_count[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"{{operation}} (shard acquire count)\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (rate(sharditem_created_count[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"shard item created count\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(numshards_gauge[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"num of shard gauge\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(persistence_requests_per_shard[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"persistence requests per shard\",\n              \"range\": true,\n              \"refId\": \"D\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Shard Movements per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1537\n          },\n          \"id\": 94,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"avg by(cadence_service) (num_goroutines)\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Goroutines (avg)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Service Metrics\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 66\n      },\n      \"id\": 101,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1618\n          },\n          \"id\": 97,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(cadence_requests{cadence_service=\\\"cadence_history\\\", operation=~\\\"StartWorkflowExecution|SignalWorkflowExecution\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"External Events per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1618\n          },\n          \"id\": 98,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=~\\\"RespondActivityTaskCompleted|RecordActivityTaskHeartbeat|RecordActivityTaskStarted|RespondDecisionTaskCompleted\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Activities per sec \",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1626\n          },\n          \"id\": 100,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(cadence_requests{operation=~\\\"PollForActivityTask|PollForDecisionTask|CancelOutstandingPoll\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (changes(poll_success_sync{cadence_service=\\\"cadence_matching\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"poll success sync\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(__name__) (rate(poll_timeouts{cadence_service=\\\"cadence_matching\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"poll timeout\",\n              \"range\": true,\n              \"refId\": \"C\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Polls per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1626\n          },\n          \"id\": 99,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(cadence_requests{operation=~\\\"RecordDecisionTaskStarted|RespondDecisionTaskCompleted\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decisions per sec\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Rates per second\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 67\n      },\n      \"id\": 106,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 0,\n            \"y\": 643\n          },\n          \"id\": 102,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_frontend\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 6,\n            \"y\": 643\n          },\n          \"id\": 103,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_history\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 12,\n            \"y\": 643\n          },\n          \"id\": 105,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_worker\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Worker Versions\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 6,\n            \"x\": 18,\n            \"y\": 643\n          },\n          \"id\": 104,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(cadence_version) (build_information{cadence_service=\\\"cadence_matching\\\"})\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Matching Versions\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Server Version Metrics\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 68\n      },\n      \"id\": 111,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 644\n          },\n          \"id\": 107,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (changes(task_requests{cadence_service=\\\"cadence_history\\\", operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": true,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Tasks Requests\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 644\n          },\n          \"id\": 108,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, operation) (task_latency_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Task Latency\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1108\n          },\n          \"id\": 109,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(domain) (changes(task_requests_per_domain{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Tasks Per Domain per sec\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              },\n              \"unit\": \"ms\"\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"TimerActiveTaskActivityTimeout\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1108\n          },\n          \"id\": 110,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(task_latency_per_domain_bucket{operation=~\\\"TimerActiveTaskActivityTimeout|TimerActiveTaskDecisionTimeout|TimerActiveTaskWorkflowTimeout\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Timer Task Latency\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Timer Task Processing\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 69\n      },\n      \"id\": 117,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 24,\n            \"x\": 0,\n            \"y\": 645\n          },\n          \"id\": 112,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(__name__) (rate(task_requests{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"requests\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            },\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"code\",\n              \"expr\": \"sum by(__name__) (rate(task_errors{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": true,\n              \"instant\": false,\n              \"legendFormat\": \"errors\",\n              \"range\": true,\n              \"refId\": \"B\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Requests Vs Errors\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 797\n          },\n          \"id\": 116,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(operation) (rate(task_requests{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Requests Per Type\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 797\n          },\n          \"id\": 113,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, operation) (rate(task_latency_bucket{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Task Latency (upper)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 805\n          },\n          \"id\": 114,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"sum by(domain) (rate(task_requests_per_domain{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval]))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Tasks Per Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 805\n          },\n          \"id\": 115,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.99, sum by(le, domain) (rate(task_latency_per_domain_bucket{operation=~\\\"TransferActiveTaskCloseExecution|TransferActiveTaskDecision|TransferActiveTaskRecordWorkflowStarted\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Transfer Task upper Latency per Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Transfer Task Processing\",\n      \"type\": \"row\"\n    },\n    {\n      \"collapsed\": true,\n      \"gridPos\": {\n        \"h\": 1,\n        \"w\": 24,\n        \"x\": 0,\n        \"y\": 70\n      },\n      \"id\": 125,\n      \"panels\": [\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 641\n          },\n          \"id\": 118,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(history_size_bucket{operation=\\\"ExecutionStats\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Size By Domain (Top 20)\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 641\n          },\n          \"id\": 119,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, sum by(le, domain) (rate(history_count_bucket{operation=~\\\"ExecutionStats\\\", cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"History Event Count By Domain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1105\n          },\n          \"id\": 120,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Blob Sizes Top Domains\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": []\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1105\n          },\n          \"id\": 121,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, operation) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"includeNullMetadata\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Frontend Blob Sizes Top Operations\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1113\n          },\n          \"id\": 123,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Blob Sizes by Domain\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 12,\n            \"y\": 1113\n          },\n          \"id\": 122,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, decisionType) (rate(event_blob_size_bucket{cadence_service=\\\"cadence_history\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Blob Sizes by DecisionType\",\n          \"type\": \"timeseries\"\n        },\n        {\n          \"datasource\": {\n            \"type\": \"prometheus\",\n            \"uid\": \"PBFA97CFB590B2093\"\n          },\n          \"fieldConfig\": {\n            \"defaults\": {\n              \"color\": {\n                \"mode\": \"palette-classic\"\n              },\n              \"custom\": {\n                \"axisBorderShow\": false,\n                \"axisCenteredZero\": false,\n                \"axisColorMode\": \"text\",\n                \"axisLabel\": \"\",\n                \"axisPlacement\": \"auto\",\n                \"barAlignment\": 0,\n                \"barWidthFactor\": 0.6,\n                \"drawStyle\": \"line\",\n                \"fillOpacity\": 0,\n                \"gradientMode\": \"none\",\n                \"hideFrom\": {\n                  \"legend\": false,\n                  \"tooltip\": false,\n                  \"viz\": false\n                },\n                \"insertNulls\": false,\n                \"lineInterpolation\": \"linear\",\n                \"lineWidth\": 1,\n                \"pointSize\": 5,\n                \"scaleDistribution\": {\n                  \"type\": \"linear\"\n                },\n                \"showPoints\": \"auto\",\n                \"spanNulls\": false,\n                \"stacking\": {\n                  \"group\": \"A\",\n                  \"mode\": \"none\"\n                },\n                \"thresholdsStyle\": {\n                  \"mode\": \"off\"\n                }\n              },\n              \"mappings\": [],\n              \"thresholds\": {\n                \"mode\": \"absolute\",\n                \"steps\": [\n                  {\n                    \"color\": \"green\"\n                  },\n                  {\n                    \"color\": \"red\",\n                    \"value\": 80\n                  }\n                ]\n              }\n            },\n            \"overrides\": [\n              {\n                \"__systemRef\": \"hideSeriesFrom\",\n                \"matcher\": {\n                  \"id\": \"byNames\",\n                  \"options\": {\n                    \"mode\": \"exclude\",\n                    \"names\": [\n                      \"ScheduleActivityTask\"\n                    ],\n                    \"prefix\": \"All except:\",\n                    \"readOnly\": true\n                  }\n                },\n                \"properties\": [\n                  {\n                    \"id\": \"custom.hideFrom\",\n                    \"value\": {\n                      \"legend\": false,\n                      \"tooltip\": false,\n                      \"viz\": true\n                    }\n                  }\n                ]\n              }\n            ]\n          },\n          \"gridPos\": {\n            \"h\": 8,\n            \"w\": 12,\n            \"x\": 0,\n            \"y\": 1121\n          },\n          \"id\": 124,\n          \"options\": {\n            \"legend\": {\n              \"calcs\": [],\n              \"displayMode\": \"list\",\n              \"placement\": \"bottom\",\n              \"showLegend\": true\n            },\n            \"tooltip\": {\n              \"hideZeros\": false,\n              \"mode\": \"single\",\n              \"sort\": \"none\"\n            }\n          },\n          \"pluginVersion\": \"11.6.0\",\n          \"targets\": [\n            {\n              \"datasource\": {\n                \"type\": \"prometheus\",\n                \"uid\": \"PBFA97CFB590B2093\"\n              },\n              \"disableTextWrap\": false,\n              \"editorMode\": \"builder\",\n              \"expr\": \"histogram_quantile(0.95, max by(le, domain) (rate(decision_result_count_bucket{cadence_service=\\\"cadence_frontend\\\"}[$__rate_interval])))\",\n              \"fullMetaSearch\": false,\n              \"hide\": false,\n              \"includeNullMetadata\": false,\n              \"instant\": false,\n              \"legendFormat\": \"__auto\",\n              \"range\": true,\n              \"refId\": \"A\",\n              \"useBackend\": false\n            }\n          ],\n          \"title\": \"Decision Result Count Top Domains\",\n          \"type\": \"timeseries\"\n        }\n      ],\n      \"title\": \"Size and Count Limits\",\n      \"type\": \"row\"\n    }\n  ],\n  \"preload\": false,\n  \"refresh\": false,\n  \"schemaVersion\": 41,\n  \"tags\": [],\n  \"templating\": {\n    \"list\": [\n      {\n        \"current\": {\n          \"text\": \"all\",\n          \"value\": \"all\"\n        },\n        \"datasource\": \"Prometheus\",\n        \"definition\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n        \"includeAll\": false,\n        \"label\": \"Domain\",\n        \"name\": \"domain\",\n        \"options\": [],\n        \"query\": {\n          \"qryType\": 1,\n          \"query\": \"label_values(activity_end_to_end_latency_sum,domain)\",\n          \"refId\": \"PrometheusVariableQueryEditor-VariableQuery\"\n        },\n        \"refresh\": 1,\n        \"regex\": \"\",\n        \"type\": \"query\"\n      }\n    ]\n  },\n  \"time\": {\n    \"from\": \"now-5m\",\n    \"to\": \"now\"\n  },\n  \"timepicker\": {\n    \"refresh_intervals\": [\n      \"10s\",\n      \"30s\",\n      \"1m\",\n      \"5m\",\n      \"15m\",\n      \"30m\",\n      \"1h\",\n      \"2h\",\n      \"1d\"\n    ]\n  },\n  \"timezone\": \"\",\n  \"title\": \"Cadence-Server\",\n  \"uid\": \"1B0efRyGz\",\n  \"version\": 1\n}"
  },
  {
    "path": "docker/grafana/provisioning/dashboards/default.yaml",
    "content": "apiVersion: 1\n\nproviders:\n  - name: Cadence-Basic # A uniquely identifiable name for the provider\n    folder: Cadence\n    type: file\n    options:\n      path:\n        /etc/grafana/provisioning/dashboards/\n"
  },
  {
    "path": "docker/grafana/provisioning/datasources/default.yaml",
    "content": "datasources:\n  - name: Prometheus\n    type: prometheus\n    url: http://host.docker.internal:9090\n    version: 1\n    editable: true\n"
  },
  {
    "path": "docker/prometheus/prometheus.yml",
    "content": "global:\n  scrape_interval: 5s\n  external_labels:\n    monitor: 'cadence-monitor'\nscrape_configs:\n  - job_name: 'prometheus'\n    static_configs:\n      - targets: # addresses to scrape\n          - 'cadence:9090'\n          - 'cadence:8000'\n          - 'cadence:8001'\n          - 'cadence:8002'\n          - 'cadence:8003'\n          - 'host.docker.internal:8004' # Endpoint for Cadence samples running on localhost \n"
  },
  {
    "path": "docker/prometheus_multiclusters/prometheus.yml",
    "content": "global:\n  scrape_interval: 5s\n  external_labels:\n    monitor: 'cadence-monitor'\nscrape_configs:\n  - job_name: 'prometheus'\n    static_configs:\n      - targets: # addresses to scrape\n          - 'cadence:9090'\n          - 'cadence:8000'\n          - 'cadence:8001'\n          - 'cadence:8002'\n          - 'cadence:8003'\n          - 'cadence:9000'\n          - 'cadence:9001'\n          - 'cadence:9002'\n          - 'cadence:9003'\n"
  },
  {
    "path": "docker/setup-multiclusters-schema.sh",
    "content": "#!/bin/bash\n\nset -ex\n\n# Default values\nCASSANDRA_SEEDS=\"${CASSANDRA_SEEDS:-cassandra}\"\nCASSANDRA_USER=\"${CASSANDRA_USER:-cassandra}\"\nCASSANDRA_PASSWORD=\"${CASSANDRA_PASSWORD:-cassandra}\"\nCASSANDRA_PROTO_VERSION=\"${CASSANDRA_PROTO_VERSION:-4}\"\nREPLICATION_FACTOR=\"${RF:-1}\"\n\n# Keyspaces for multiclusters setup\nPRIMARY_KEYSPACE=\"${PRIMARY_KEYSPACE:-cadence_primary}\"\nPRIMARY_VISIBILITY_KEYSPACE=\"${PRIMARY_VISIBILITY_KEYSPACE:-cadence_visibility_primary}\"\nSECONDARY_KEYSPACE=\"${SECONDARY_KEYSPACE:-cadence_secondary}\"\nSECONDARY_VISIBILITY_KEYSPACE=\"${SECONDARY_VISIBILITY_KEYSPACE:-cadence_visibility_secondary}\"\n\n# Schema directories\nSCHEMA_DIR=\"/etc/cadence/schema/cassandra/cadence/versioned\"\nVISIBILITY_SCHEMA_DIR=\"/etc/cadence/schema/cassandra/visibility/versioned\"\n\nwait_for_cassandra() {\n    echo \"Waiting for Cassandra to be ready...\"\n    server=$(echo $CASSANDRA_SEEDS | awk -F ',' '{print $1}')\n    \n    # Wait for Cassandra to be fully ready with timeout\n    echo \"Waiting for Cassandra to be fully ready...\"\n    timeout=300  # 5 minutes timeout\n    counter=0\n    \n    while [[ $counter -lt $timeout ]]; do\n        if cadence-cassandra-tool --ep $server --u $CASSANDRA_USER --pw $CASSANDRA_PASSWORD --pv $CASSANDRA_PROTO_VERSION create -k test_keyspace --rf 1 2>/dev/null; then\n            echo 'cassandra started and ready'\n            return 0\n        fi\n        echo \"waiting for cassandra to be fully ready (attempt $((counter/5 + 1)))\"\n        sleep 5\n        counter=$((counter + 5))\n    done\n    \n    echo \"Error: Cassandra did not become ready within $timeout seconds\"\n    exit 1\n}\n\nsetup_keyspace_schema() {\n    local keyspace=$1\n    local visibility_keyspace=$2\n    \n    echo \"Setting up schema for keyspace: $keyspace\"\n    \n    # Create keyspace\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS create -k $keyspace --rf $REPLICATION_FACTOR\n    \n    # Setup schema versioning\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $keyspace setup-schema -v 0.0\n    \n    # Update to latest schema\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $keyspace update-schema -d $SCHEMA_DIR\n    \n    echo \"Setting up visibility schema for keyspace: $visibility_keyspace\"\n    \n    # Create visibility keyspace\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS create -k $visibility_keyspace --rf $REPLICATION_FACTOR\n    \n    # Setup visibility schema versioning\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $visibility_keyspace setup-schema -v 0.0\n    \n    # Update visibility to latest schema\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $visibility_keyspace update-schema -d $VISIBILITY_SCHEMA_DIR\n}\n\nsetup_primary_cluster() {\n    echo \"Setting up primary cluster schema...\"\n    setup_keyspace_schema $PRIMARY_KEYSPACE $PRIMARY_VISIBILITY_KEYSPACE\n    \n    # Note: Domain registration will be handled by the Cadence services when they start\n    echo \"Primary cluster schema setup completed. Domain registration will be handled by Cadence services.\"\n}\n\nsetup_secondary_cluster() {\n    echo \"Setting up secondary cluster schema...\"\n    setup_keyspace_schema $SECONDARY_KEYSPACE $SECONDARY_VISIBILITY_KEYSPACE\n    \n    # Note: Domain registration will be handled by the Cadence services when they start\n    echo \"Secondary cluster schema setup completed. Domain registration will be handled by Cadence services.\"\n}\n\nmain() {\n    echo \"Starting multiclusters schema setup...\"\n    \n    # Wait for Cassandra to be ready\n    wait_for_cassandra\n    \n    # Setup both clusters\n    setup_primary_cluster\n    setup_secondary_cluster\n    \n    echo \"Multiclusters schema setup completed successfully!\"\n}\n\n# Run the main function\nmain \"$@\" "
  },
  {
    "path": "docker/start-cadence.sh",
    "content": "#!/bin/bash\n\nset -ex\n\nCONFIG_TEMPLATE_PATH=\"${CONFIG_TEMPLATE_PATH:-/etc/cadence/config/config_template.yaml}\"\n\ndockerize -template $CONFIG_TEMPLATE_PATH:/etc/cadence/config/docker.yaml\n\nexec cadence-server --root $CADENCE_HOME --env docker start --services=$SERVICES\n\n"
  },
  {
    "path": "docker/start.sh",
    "content": "#!/bin/bash\n\nset -x\n\nDB=\"${DB:-cassandra}\"\nENABLE_ES=\"${ENABLE_ES:-false}\"\nES_PORT=\"${ES_PORT:-9200}\"\nES_VERSION=\"${ES_VERSION:-v6}\"\nRF=${RF:-1}\n\n# cassandra env\nexport CASSANDRA_USER=\"${CASSANDRA_USER:-cassandra}\"\nexport CASSANDRA_PASSWORD=\"${CASSANDRA_PASSWORD:-cassandra}\"\nexport KEYSPACE=\"${KEYSPACE:-cadence}\"\nexport VISIBILITY_KEYSPACE=\"${VISIBILITY_KEYSPACE:-cadence_visibility}\"\nexport CASSANDRA_PROTO_VERSION=\"${CASSANDRA_PROTO_VERSION:-4}\"\n\n# mysql env\nexport DBNAME=\"${DBNAME:-cadence}\"\nexport VISIBILITY_DBNAME=\"${VISIBILITY_DBNAME:-cadence_visibility}\"\nexport DB_PORT=${DB_PORT:-3306}\n\n# elasticsearch env\nexport VISIBILITY_NAME=\"${VISIBILITY_NAME:-cadence-visibility-dev}\"\n\nsetup_cassandra_schema() {\n    SCHEMA_DIR=$CADENCE_HOME/schema/cassandra/cadence/versioned\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS create -k $KEYSPACE --rf $RF\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $KEYSPACE setup-schema -v 0.0\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $KEYSPACE update-schema -d $SCHEMA_DIR\n    VISIBILITY_SCHEMA_DIR=$CADENCE_HOME/schema/cassandra/visibility/versioned\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS create -k $VISIBILITY_KEYSPACE --rf $RF\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $VISIBILITY_KEYSPACE setup-schema -v 0.0\n    cadence-cassandra-tool --ep $CASSANDRA_SEEDS -k $VISIBILITY_KEYSPACE update-schema -d $VISIBILITY_SCHEMA_DIR\n    echo \"Registering domain for Cassandra...\"\n    cqlsh $CASSANDRA_SEEDS -k $KEYSPACE -f /etc/cadence/domain/cassandra.cql\n}\n\nsetup_mysql_schema() {\n    SCHEMA_DIR=$CADENCE_HOME/schema/mysql/v8/cadence/versioned\n    if [ \"$MYSQL_TX_ISOLATION_COMPAT\" == \"true\" ]; then\n        CONNECT_ATTR='--connect-attributes tx_isolation=READ-COMMITTED'\n    fi\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD $CONNECT_ATTR create --db $DBNAME\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD $CONNECT_ATTR --db $DBNAME setup-schema -v 0.0\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD $CONNECT_ATTR --db $DBNAME update-schema -d $SCHEMA_DIR\n    VISIBILITY_SCHEMA_DIR=$CADENCE_HOME/schema/mysql/v8/visibility/versioned\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD $CONNECT_ATTR create --db $VISIBILITY_DBNAME\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD --db $VISIBILITY_DBNAME $CONNECT_ATTR setup-schema -v 0.0\n    cadence-sql-tool --ep $MYSQL_SEEDS -u $MYSQL_USER --pw $MYSQL_PWD --db $VISIBILITY_DBNAME $CONNECT_ATTR update-schema -d $VISIBILITY_SCHEMA_DIR\n    echo \"Registering domain for MySQL...\"\n    mysql -h $MYSQL_HOST -u $MYSQL_USER -p $MYSQL_PASSWORD $MYSQL_DATABASE < /etc/cadence/domain/mysql.sql\n}\n\nsetup_postgres_schema() {\n    SCHEMA_DIR=$CADENCE_HOME/schema/postgres/cadence/versioned\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT create --db $DBNAME\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT --db $DBNAME setup-schema -v 0.0\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT --db $DBNAME update-schema -d $SCHEMA_DIR\n    VISIBILITY_SCHEMA_DIR=$CADENCE_HOME/schema/postgres/visibility/versioned\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT create --db $VISIBILITY_DBNAME\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT --db $VISIBILITY_DBNAME setup-schema -v 0.0\n    cadence-sql-tool --plugin postgres --ep $POSTGRES_SEEDS -u $POSTGRES_USER --pw \"$POSTGRES_PWD\" -p $DB_PORT --db $VISIBILITY_DBNAME update-schema -d $VISIBILITY_SCHEMA_DIR\n    echo \"Registering domain for PostgreSQL...\"\n    PGPASSWORD=$POSTGRES_PASSWORD psql -h $POSTGRES_HOST -U $POSTGRES_USER -d $POSTGRES_DATABASE -f /etc/cadence/domain/postgres.sql\n}\n\n\nsetup_es_template() {\n    SCHEMA_FILE=$CADENCE_HOME/schema/elasticsearch/$ES_VERSION/visibility/index_template.json\n    server=`echo $ES_SEEDS | awk -F ',' '{print $1}'`\n    URL=\"http://$server:$ES_PORT/_template/cadence-visibility-template\"\n    curl -X PUT $URL -H 'Content-Type: application/json' --data-binary \"@$SCHEMA_FILE\"\n    URL=\"http://$server:$ES_PORT/$VISIBILITY_NAME\"\n    curl -X PUT $URL\n}\n\nsetup_schema() {\n    if [ \"$DB\" == \"mysql\" ]; then\n        echo 'setup mysql schema'\n        setup_mysql_schema\n    elif [ \"$DB\" == \"postgres\" ]; then\n        echo 'setup postgres schema'\n        setup_postgres_schema\n    else\n        echo 'setup cassandra schema'\n        setup_cassandra_schema\n    fi\n\n    if [ \"$ENABLE_ES\" == \"true\" ]; then\n        setup_es_template\n    fi\n}\n\nwait_for_cassandra() {\n    server=`echo $CASSANDRA_SEEDS | awk -F ',' '{print $1}'`\n    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.4.6 --protocol-version=$CASSANDRA_PROTO_VERSION $server < /dev/null; do\n        echo 'waiting for cassandra to start up'\n        sleep 1\n    done\n    echo 'cassandra started'\n}\n\nwait_for_scylla() {\n    server=`echo $CASSANDRA_SEEDS | awk -F ',' '{print $1}'`\n    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.3.1 --protocol-version=$CASSANDRA_PROTO_VERSION $server < /dev/null; do\n        echo 'waiting for scylla to start up'\n        sleep 1\n    done\n    echo 'scylla started'\n}\n\nwait_for_mysql() {\n    server=`echo $MYSQL_SEEDS | awk -F ',' '{print $1}'`\n    nc -z $server $DB_PORT < /dev/null\n    until [ $? -eq 0 ]; do\n        echo 'waiting for mysql to start up'\n        sleep 1\n        nc -z $server $DB_PORT < /dev/null\n    done\n    echo 'mysql started'\n}\n\nwait_for_postgres() {\n    server=`echo $POSTGRES_SEEDS | awk -F ',' '{print $1}'`\n    nc -z $server $DB_PORT < /dev/null\n    until [ $? -eq 0 ]; do\n        echo 'waiting for postgres to start up'\n        sleep 1\n        nc -z $server $DB_PORT < /dev/null\n    done\n    echo 'postgres started'\n}\n\n\nwait_for_es() {\n    server=`echo $ES_SEEDS | awk -F ',' '{print $1}'`\n    URL=\"http://$server:$ES_PORT\"\n    curl -s $URL > /dev/null 2>&1\n    until [ $? -eq 0 ]; do\n        echo 'waiting for elasticsearch to start up'\n        sleep 1\n        curl -s $URL > /dev/null 2>&1\n    done\n    echo 'elasticsearch started'\n}\n\nwait_for_db() {\n    if [ \"$DB\" == \"mysql\" ]; then\n        wait_for_mysql\n    elif [ \"$DB\" == \"postgres\" ]; then\n        wait_for_postgres\n    elif [ \"$DB\" == \"scylla\" ]; then\n        wait_for_scylla\n    else\n        wait_for_cassandra\n    fi\n\n    if [ \"$ENABLE_ES\" == \"true\" ]; then\n        wait_for_es\n    fi\n}\n\nwait_for_db\nif [ \"$SKIP_SCHEMA_SETUP\" != true ]; then\n    setup_schema\nfi\n\nexec /start-cadence.sh\n"
  },
  {
    "path": "docs/cassandra-executions-table.md",
    "content": "# Overview\nThe `executions` table in Cassandra is a table used for storing mutiple types of entities, because Cassandra doesn't support cross-table LWTs.\nThe `type` column in `executions` table indicates the type of entity stored in that row. Since the table stores all the entities, the columns\nare the union set of columns of all entities. And the data in each column have different meaning for different type of entities. We'll explain\nfor each type of entity the usage of the columns in `executions` table.\n\n## Shard\nThe `shard` entities are the rows with `type=0`, which stores the information of Cadence shards. The following columns are used:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of entity, which is always `0` for `shard`\n* `shard`: stores the information of `shard`\n* `range_id`: used to manage changes to the shard's configuration and ownership\n\nOther columns are set to a constant only because they're part of primary key columns:\n* `domain_id`: 10000000-1000-f000-f000-000000000000\n* `workflow_id`: 20000000-1000-f000-f000-000000000000\n* `run_id`: 30000000-1000-f000-f000-000000000000\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n* `task_id`: -11\n\n## Execution\nThe `execution` entities are the rows with `type=1`, which stores the information of workflow. And there are 2 types of executions: `current_execution` and `concrete_execution`.\n### Current Execution\nThe `current_execution` entities has a constant value in the `run_id` column which equals `30000000-0000-f000-f000-000000000001`. The follwing columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `domain_id`: the `id` of the domain of the workflow\n* `workflow_id`: the `id` of the workflow\n* `current_run_id`: the `run_id` of the most recent workflow among the workflows with the same `workflow_id`\n* `execution`: stores some information about the workflow\n* `workflow_last_write_version`: the failover version of the workflow last updated by a Cadence cluster\n* `workflow_state`: the state of the workflow\n\nThe following columns are set to a constant value:\n* `run_id`: 30000000-0000-f000-f000-000000000001\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n* `task_id`: -10\n\n### Concrete Execution\nThe `concrete_execution` entities stores the information of each run of a workflow. These columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `domain_id`: the `id` of the domain of the workflow\n* `workflow_id`: the `id` of the workflow\n* `run_id`: the `id` of the instance of the workflow\n* `execution`: stores some information about the workflow\n* `next_event_id`: the `id` of the next event of the workflow\n* `activity_map`: the information of activities of the workflow\n* `timer_map`: the information of timers of the workflow\n* `child_executions_map`: the information of child workflows of the workflow\n* `request_cancel_map`: the information of cancellation requests of the workflow\n* `signal_map`: the information of signals applied to the workflow\n* `signal_requested`: the ids of the signals applied to the workflow\n* `buffered_events_list`: the buffered events of the workflow\n* `workflow_last_write_version`: the failover version of the workflow last updated by a Cadence cluster\n* `workflow_state`: the state of the workflow\n* `version_histories`: stores the information tracking the version history of the workflow\n* `version_histories_encoding`: the encoding type of `version_histories` column\n* `checksum`: a piece of data used for detecting data corruption in database\n\nThe following columns are set to a constant value:\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n* `task_id`: -10\n\n## Transfer Task\nThe `transfer_task` entities are the rows with `type=2`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `task_id`: the `id` of the task\n* `transfer`: the information of the task\n\nThe following columns are set to a constant value:\n* `domain_id`: 10000000-3000-f000-f000-000000000000\n* `workflow_id`: 20000000-3000-f000-f000-000000000000\n* `run_id`: 30000000-3000-f000-f000-000000000000\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n\n## Timer Task\nThe `timer_task` entities are the rows with `type=3`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `visibility_ts`: the timestamp of the timer task\n* `task_id`: the `id` of the task\n* `timer`: the information of the task\n\nThe following columns are set to a constant value:\n* `domain_id`: 10000000-4000-f000-f000-000000000000\n* `workflow_id`: 20000000-4000-f000-f000-000000000000\n* `run_id`: 30000000-4000-f000-f000-000000000000\n\n## Replication Task\nThe `replication_task` entities are the rows with `type=4`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `task_id`: the `id` of the task\n* `replication`: the information of the task\n\nThe following columns are set to a constant value:\n* `domain_id`: 10000000-5000-f000-f000-000000000000\n* `workflow_id`: 20000000-5000-f000-f000-000000000000\n* `run_id`: 30000000-5000-f000-f000-000000000000\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n\n## Replication DLQ Task\nThe `dlq` entities are the rows with `type=5`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `workflow_id`: the source cluster of the replication task\n* `task_id`: the `id` of the task\n* `replication`: the information of the task\n\nThe following columns are set to a constant value:\n* `domain_id`: 10000000-6000-f000-f000-000000000000\n* `run_id`: 30000000-6000-f000-f000-000000000000\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n\n## Cross Cluster Task\nThe `cross_cluster_task` entities are the rows with `type=6`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity\n* `workflow_id`: the target cluster which the task is applied to\n* `task_id`: the `id` of the task\n* `cross_cluster`: the information of the task\n\nThe following columns are set to a constant value:\n* `domain_id`: 10000000-7000-f000-f000-000000000000\n* `run_id`: 30000000-7000-f000-f000-000000000000\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n\n## Workflow Requests\nThe `workflow_requests` entities are rows with `type` in `{7, 8, 9, 10}`. The following columns are set:\n* `shard_id`: the `id` of the Cadence shard\n* `type`: indicates the type of the entity (7 = StartWorkflowRequest, 8 = SignalWorkflowRequest, 9 = CancelWorkflowRequest, 10 = ResetWorkflowRequest)\n* `domain_id`: the `id` of the domain\n* `workflow_id`: the `id` of the workflow which the request is applied to\n* `run_id`: the `id` of the request\n* `task_id`: the failover version of the domain when the request is applied\n* `current_run_id`: the `run_id` of the workflow which the request is applied to\nThe following columns are set to a constant value:\n* `visibility_ts`: 2000-01-01 00:00:00.000000+0000\n"
  },
  {
    "path": "docs/design/1533-host-specific-tasklist.md",
    "content": "# Proposal: Resource-Specific Tasklist\n\nAuthor: Yichao Yang (@yycptt)\n\nLast updated: July 2019\n\nDiscussion at <https://github.com/cadence-workflow/cadence-go-client/issues/697>\n\n## Abstract\n\nCurrently, there’s no straightforward way for a Cadence user to run multiple activities (a session) on a single activity worker. Although there is the option to explicitly specify a host-specific tasklist name or manually execute special activities to get information about the activity worker, these approaches have limitations and are error prone. Besides running multiple activities on a single worker, a user may also want to limit the total number of sessions running concurrently on a single worker if those sessions consume worker resources. This document proposes several new APIs and their implementations so that a user can create sessions, execute activities, and limit the number of concurrent sessions through a simple API call.\n\n## Use Cases\n\n### Machine Learning Model Training\n\nA user can model the entire training process with multiple activities, for example, downloading datasets, data cleaning, model training, and parameter uploading. In this case, all activities should be executed by one activity worker. Otherwise, the dataset might be downloaded multiple times. In addition, as a dataset typically consumes a lot of disk space and model training requires GPU support, the number of total models training on a single worker should be limited.\n\n### Service Deployment\n\nIn this case, a session represents the deployment of a service to one worker and the number of concurrent deployments is critical. A user controls that number by limiting the number of concurrently running sessions on an activity worker.\n\n## Proposal\n\nThe scope of this proposal is limited to the Cadence client. Cadence server doesn’t need to be changed at all. The basic idea here is that we should write special workflow code for users to perform activity scheduling and worker monitoring, and expose simple APIs to users. Once users don’t need to care about those implementation details, they can focus on their own business logic.\n\nThe proposal includes both API changes and implementations. It aims at solving three problems:\n\n1. How to ensure multiple activities are executed on one activity worker.\n\n2. How to limit the number of concurrent sessions on an activity worker.\n\n3. How to carry over session information between different workflow runs.\n\nThe following sections will first go through the APIs exposed to users, explain how to use them, and provide sample code. Any developer with previous Cadence experience should be able to understand the APIs and start using them immediately. After that I will explain the model behind those APIs and how they are implemented.\n\n### API Changes\n\n#### New Go Client API\n\nThere will be four new APIs available when writing a workflow:\n\n```go\nsessionCtx, err := workflow.CreateSession(ctx Context, so *SessionOptions)\n```\n\nA user calls this API to create a session on the worker that polls the task list specified in the ActivityOptions (or in the StartWorkflowOptions if the task list name is not specified in the ActivityOptions). All activities executed within the returned sessionCtx (a new context which contains metadata information of the created session) are considered to be part of the session and will be executed on the same worker. The sessionCtx will be cancelled if the worker executing this session dies or CompleteSession() is called.\n\nThe SessionOptions struct contains two fields: `ExecutionTimeout`, which specifies the maximum amount of time the session can run and `CreationTimeout`, which specifies how long session creation can take before returning an error.\n\nCreateSession() will return an error if the context passed in already contains an open session. If all the workers are currently busy and unable to handle new sessions, the framework will keep retrying until the CreationTimeout you specified in the SessionOptions has passed before returning an error.\n\nWhen executing an activity within a session, a user might get three types of errors:\n1. Those returned from user activities. The session will not be marked as failed in this case, so the user can return whatever error they want and apply their business logic as necessary. If a user wants to end a session due to the error returned from the activity, use the CompleteSession() API below.\n2. A special `ErrSessionFailed` error: this error means the session has failed due to worker failure and the session is marked as failed in the background. In this case, no activities can be executed using this context. The user can choose how to handle the failure. They can create a new session to retry or end the workflow with an error.\n3. Cancelled error: If a session activity has been scheduled before worker failure is detected, it will be cancelled afterwards and a cancelled error will be returned.\n\n```go\nworkflow.CompleteSession(sessionCtx Context)\n```\n\nThis API is used to complete a session. It releases the resources reserved on the worker and cancels the session context (therefore all the activities using that session context). It does nothing when the session is already closed or failed.\n\n```go\nsessionInfoPtr := workflow.GetSessionInfo(sessionCtx Context)\n```\n\nThis API returns session metadata stored in the context. If the context passed in doesn’t contain any session metadata, this API will return a nil pointer. For now, the only exported fields in sessionInfo are: sessionID, which is a unique identifier for a session, and hostname.\n\n```go\nsessionCtx, err := workflow.RecreateSession(ctx Context, recreateToken []byte, so *SessionOptions)\n```\n\nFor long running sessions, user might want to split it into multiple runs while still wanting all the activities executed by the same worker. This API is designed for such a use case.\n\nIts usage is the same as CreateSession() except that it also takes in a recreateToken. The token encodes the resource specific tasklist name on which the new session should be created. A user can get the token by calling the GetRecreateToken() method of the SessionInfo object.\n\n\n#### Example\n\nThe basic usage looks like the following (it belongs to a larger workflow):\n\n```go\nsessionCtx, err := CreateSession(ctx)\n\nif err != nil {\n\n    // Creation failed. Wrong ctx or too many outstanding sessions.\n\n}\n\ndefer CompleteSession(sessionCtx)\n\nerr = ExecuteActivity(sessionCtx, DownloadFileActivity, filename).Get(sessionCtx, nil)\n\nif err != nil {\n\n    // Session(worker) has failed or activity itself returns an error.\n\n    // User can perform normal error handling here and decide whether creating a new session is needed.\n\n    // If they decide to create a new session, they need to call CompleteSession() so that worker resources can be released. We recommend that users create a function for a session and call defer CompleteSession(sessionCtx) after a session is created. If session retry is needed, this function can be called multiple times if the error returned is retriable.\n\n}\n\nerr = ExecuteActivity(sessionCtx, ProcessFileActivity, filename).Get(sessionCtx, nil)\n\nif err != nil {\n\n    // Session(worker) has failed or activity itself returns an error.\n\n}\n\nerr = ExecuteActivity(sessionCtx, UploadFileActivity, filename).Get(sessionCtx, nil)\n\nif err != nil {\n\n    // Session(worker) has failed or activity itself returns an error.\n\n}\n```\n\n#### New Worker Options\n\nThere will be three new options available when creating a worker:\n\n* **EnableSessionWorker**\n\n  This flag controls whether the worker will accept activities that belong to a session.\n\n* **SessionResourceID**\n\n  An identifier of the resource that will be consumed if a session is executed on the worker. For example, if a worker has a GPU, and a session for training a machine learning model is executed on the worker, then some memory on the GPU will be consumed. In this case, the resourceID can be the identifier of the GPU.\n\n**NOTE:** Users must ensure that only one worker uses a certain resourceID on a host.\n\n* **MaxCurrentSessionExecutionSize**\n\n  Because a session may consume some kind of resource, a user can use this option to control the maximum number of sessions running in the worker process at the same time.\n\n## Implementation\n\n### Session Model\n\nAll sessions are **resource specific**. This means that a session is always tied to some kind of resource (not the worker). The resource can be CPU, GPU, memory, file descriptors, etc., and we want all the activities within a session to be executed by the same worker that owns the resource. If a worker dies and restarts, as long as it owns the same resource, we can try to reestablish the session (this is future work). Also, the user needs to ensure that the resource is owned by only one worker on a single host.\n\nNote that when creating a session, a user doesn’t need to specify the resource that is consumed by the session. This is because right now one worker owns only one resource, so there’s only one resource type on the worker. As a result, as long as the user specifies the correct tasklist in the context passed into the createSession API, the correct type of resource will be consumed (since there’s only one). Later, when a worker can support multiple types of resources, users will need to specify the resource type when creating a session.\n\n### Workflow Context\n\nWhen a user calls the CreateSession() or RecreateSession() API, a structure that contains information about the created session will be stored in the returned context. The structure contains seven pieces of information:\n\n1. **SessionID**: a unique ID for the created session. (Exported)\n\n2. **Hostname**: the hostname of the worker that is responsible for executing the session. (Exported)\n\n3. **ResourceID**: the resource consumed by the session. (Will Be Exported)\n\n4. **tasklist**: the resource specific tasklist used by this session. (Not Exported)\n\n5. **sessionState**: the state of the session (Not Exported).  It can take one of the three values below:\n\n   1. **Open**: when the session is created and running\n   2. **Failed**: when the worker is down and the session can’t continue\n   3. **Closed**: when the user closes the session through the API call and the session is successfully closed\n\n6. **sessionCancelFunc**: the function used to cancel the sessionCtx. It will cancel both the creation activity (see below) and all user activities. (Not Exported)\n\n7. **completionCtx**: the context for executing the completion activity. It’s different from the sessionCtx as we want to cancel the session context when CompleteSession() is called, but the completion activity (see below) still needs to be executed. (Not Exported)\n\nWhen a user calls the CompleteSession() API, the state of the session will be changed to \"Closed\" in place, which is why this API doesn’t return a new context variable.\n\nThere is no way to set the state of a session to “Failed”. The state of a session is “Failed” only when the worker executing the session is down and the change from “Open” to “Failed” is done in the background. Also notice that it’s possible that the worker is down, but the session state is still “Open” (since it takes some time to detect the worker failure). In this case, when a user executes an activity, the activity will still be scheduled on the resource specific tasklist. However, it will be cancelled after worker failure is detected and the user will get a cancelled error.\n\n### Workflow Worker\n\nWhen scheduling an activity, the workflow worker needs to check the context to see if this activity belongs to a session. If so and the state of that session is \"Open\", get the resource specific tasklist from the context and use it to override the tasklist value specified in the activity option. If on the other hand, the state is \"Failed\", the ExecuteActivity call will immediately fail without scheduling the activity and return an ErrSessionFailed error through Future.Get().\n\nCreateSession() and CompleteSession() really do are **schedule special activities** and get some information from the worker which executes these activities.\n\n* For CreateSession(), a special **session creation activity** will be scheduled on a global tasklist which is only used for this type of activity. During the execution of that activity, a signal containing the resource specific tasklist name will be sent back to the workflow (with other information like hostname). Once the signal is received by the worker, the creation is considered successful and the tasklist name will be stored in the session context. The creation activity also performs periodic heartbeat throughout the whole lifetime of the session. As a result, if the activity worker is down, the workflow can be notified and set the session state to “Failed”.\n\n* For RecreateSession(), the same thing happens. The only difference is that the session creation activity will be **scheduled on the resource specific task list instead of a global one**.\n\n* For CompleteSession(), a special **completion activity** will be scheduled on the resource specific tasklist. The purpose of this activity is to stop the corresponding creation activity created by the CreateSession()/CreateSessionWithHostID() API call, so that the resource used by the session can be released. Note that, **the completion activity must be scheduled on the resource specific tasklist (not a global one)** since we need to make sure the completion activity and the creation activity it needs to stop are running by the same worker.\n\n### Activity Worker\n\nWhen `EnableSessionWorker` is set to true, two more activity workers will be started.\n\n* This worker polls from a global tasklist whose name is derived from the tasklist name specified in the workerOption and executes only the special session creation activity. This special activity serves several purposes:\n\n   1. **Send the resource specific tasklist name** back to the workflow through signal.\n\n   2. Keep **heart beating** to Cadence server until the end of the session.\n\n      If the heart beating returns the `EntityNotExist` Error, this means the workflow/session has gone and this special activity should stop to release its resources.\n\n      If on the other hand, the worker is down, Cadence server will send a heartbeat timeout error to the workflow, so the workflow can be notified and update the session state. Later, an ExecuteActivity() call to the failed session context will also fail.\n\n   3. Check the number of currently running sessions on the worker. If the maximum number of concurrent sessions has been reached, return an error to fail the creation.\n\n   One detail we need to ensure is that when the maximum number of concurrent sessions has been reached, this worker should stop polling from the global task list.\n\n* Session Activity Worker: this worker only polls from the resource specific tasklist. There are three kinds of activities that can be run by this worker:\n\n   1. User activities in a session that belongs to this host.\n\n   2. Special activity to complete a session.\n\n      The sole purpose of this activity is to stop the long running session creation activity so that new sessions can be executed.\n\n   3. Special session creation activity scheduled by RecreateSession().\n\nWhen domain failover happens, since no worker will poll from the resource specific task, all activities within a session will timeout and the user will get an error. Also, Cadence server will detect the heartbeat timeout from the session creation activity.\n\n### Sequence Diagrams\n\nThis section illustrates the sequence diagrams for situations that may occur during the session execution.\n\n* Here is the sequence diagram for a normal session execution.\n\n![image alt text](1533-host-specific-tasklist-image_0.png)\n\n* When activity worker fails during the execution.\n\n![image alt text](1533-host-specific-tasklist-image_1.png)\n\n* When the workflow failed or terminated during the execution.\n\n![image alt text](1533-host-specific-tasklist-image_2.png)\n\n* When there are too many outstanding sessions and the user is trying to create a new one. The assumption here is that each activity worker is configured to have **MaxCurrentSessionExecutionSize = 1**. Resource specific tasklist and session activity worker are omitted in the diagram.\n\n![image alt text](1533-host-specific-tasklist-image_3.png)\n\n* An example of RecreateSession().\n\n![image alt text](1533-host-specific-tasklist-image_4.png)\n\n## Open Issues\n\n* Support automatic session reestablishing when worker goes down.\n\n* Allow session activities to access session information such as sessionID, resourceID, etc. This information is useful for logging and debugging.\n\n* Support multiple resources per worker and allow user to specify which type of resource a session will consume when creating the session.\n"
  },
  {
    "path": "docs/design/2215-synchronous-request-reply.md",
    "content": "# Proposal: Synchronous Request Reply\n\nAuthors: Maxim Fateev (@mfateev), Andrew Dawson (@andrewjdawson2016)\n\nLast updated: July 16, 2019\n\nDiscussion at [https://github.com/cadence-workflow/cadence/issues/2215](https://github.com/cadence-workflow/cadence/issues/2215)\n\n## Abstract\n\nCurrently Cadence workflows only process events (in form of signals) asynchronously. The only synchronous operations supported by workflows are query and waiting for workflow completion. Query is a read only operation that cannot mutate a workflow state. And waiting for a workflow completion is applicable to a pretty narrow set of use cases. There is a clear need for workflows processing mutating requests synchronously. This document goes over various workflow related request/reply use cases, identifies requirements and proposes solutions.\n\n## Use Cases\n\n### Synchronous Update\n\nFor example customer service needs to notify a workflow that the task was matched to a specific customer service representative. The notification should fail if the task is not available for dispatch anymore. This can happen if the call is duplicated by the matching system or the task state has changed for other reasons.\n\n### Page Flow\n\nFor example placing an order through a web UI or mobile app involves going through multiple screens. The data submitted from a screen is sent to a workflow which might execute activities and return information about the next page to present. In most cases the latency of this interaction is expected to be low.\n\n### Conditional Update\n\nVarious scenarios would benefit from transactional guarantees. In its simplest form an update should not go through if a state of a workflow has changed since the last query. For example the page flow should reject an update from a cached browser page if a current state of the workflow expects a different one. While it is possible to implement this check in an application code a framework support for such guarantee would simplify the programming model.\n\n### Queue Consumer\n\nQueue Consumer receives a task from Kafka, constructs a workflow id from it and forwards it to the appropriate instance. If workflow cannot process message because it is corrupted it can reject it with an error which is eventually going to put it into the DLQ. Ideally if message is rejected it is not recorded in the history to not increase the history size due to retries. It also would allow workflow efficiently filter out unnecessary messages.\n\n### Waiting for a State Change\n\nFor example continuous deployment show the current workflow state in the UI and update it once the workflow state changes. Currently they keep executing query periodically to learn about the state changes. The ideal solution would allow waiting for state changes using a long poll that is much more efficient. It also should scale to the large number of pollers watching a single workflow execution which is common in continuous deployment scenarios.\n\n## Proposal\n\nThe proposal consists of three parts.\n\n1. Guarantee read after write consistency between signal and query\n\n2. Long poll on query result changes\n\n3. Update operation that should complete within a single decision\n\nAll three together cover all the listed use cases.\n\n### Guarantee read after write consistency between signal and query\n\nCurrently, there is a race condition when a query emitted after a signal sees a stale workflow state without the update from the signal. Accounting for this inconsistency complicates the workflow logic. Fixing this race condition allows the signal-then-query pattern to be used for request-reply scenarios. The drawback of this approach is suboptimal performance since it requires at least two decisions and multiple database updates. But, together with the long polling on query result changes feature, it is a reasonable way to implement update-and-wait for long-running update operations.\n\nThe reverse consistency of failing a signal if the workflow state has changed since the last query is also possible when requested. The basic idea is to return a current workflow (or query result) version with a query in the form of an opaque token. Then include the token into the SignalWorkflowExecution call arguments. The call is rejected if the version doesn’t match the one in the token.\n\n#### Implementation Details\n\nHere is the sequence diagram of the current QueryWorkflow API implementation:\n\n![image alt text](2215-synchronous-request-reply_0.png)\n\nThe race condition happens because a query always creates its own special decision task even if there is an outstanding decision task caused by an external event. To not return state that workflow hasn’t reached yet the query ignores all the new events after the already processed decision task. This approach has another issue of failing queries for newly created workflows that haven’t yet seen a decision task.\n\nThe solution is to not create a separate decision task for a query, but include the query into an already outstanding one. It would be done by modifying the history service startDecisionTask method to return all the outstanding queries. And the client side should be modified to execute a decision as usual and then execute all the queries and insert both decisions and the query results into RespondDecisionTaskCompleted.\n\nThe main drawback of this approach is that a query cannot be processed while there is a started decision task. Usually decision task processing is pretty fast, but if it executes local activities the latency of a query can increase dramatically. The current implementation already has this problem when a sticky workflow execution (which is on by default) is used. But currently the delay happens on the client worker which serializes processing of decision tasks by the same workflow instance.\n\nThe naive implementation would add DecisionTaskScheduled/Started/Completed events to the history on every query. This is not acceptable as queries are read only operations and incurring the cost of all these updates and correspondent history growth would defeat its purpose. So the proposal is to skip any database updates if there is no outstanding decision task and rely on the in-memory cache of mutable state for such decision task handling. In case of a shard movement the state is going to be lost and the query request is going to fail. The reply to a decision task for such forgotten query can be safely ignored. If an external event like activity completion or signal is received when there is an outstanding decision task caused by a query then the event and the correspondent events including  DecisionTaskScheduled and later DecisionTaskStarted and DecisionTaskCompleted are added to the history without adding the duplicated task to the task list. The probability of a query failure in case of shard movement can be reduced by a frontend retrying it to a host that assumed the shard ownership.\n\nHere is the sequence diagram for the proposed consistent QueryWorkflowExecution API:\n\n![image alt text](2215-synchronous-request-reply_1.png)\n\nThe next diagram covers situation when query is received when another decision task is already started:\n\n![image alt text](2215-synchronous-request-reply_2.png)\n\nThe current decision state diagram\n\n![image alt text](2215-synchronous-request-reply_3.png)\n\nDiagram with query task added\n\n![image alt text](2215-synchronous-request-reply_4.png)\n\n### Long poll on query result changes\n\nThe usage scenario is to include a token with a current workflow version (as nextEventId value) into QueryWorkflowExecution result. Then if the token is included into the next QueryWorkflowExecution call do not return the call until the workflow version changes. If it doesn’t change after the specified timeout (let’s say a minute) return the call with empty no changes result. An alternative is to use hash of the query result as the version. In this case workflow changes that don’t affect the query result are not going to notify the query callers.\n\n#### Implementation Details\n\nIf nextEventId is used as a version, a QueryWorkflowExecution call to the History Service checks the version value against the current workflow nextEventId value. If they do not match the usual query path is executed. If the versions match the long poll request is registered with in-memory mutable state cache.  At some point the workflow execution receives an event that causes a decision task creation. When the decision task is started the queries registered with the workflow are included into it. The decision task completion includes the query results and all the registered long poll queries are notified about them. If no decision task is scheduled for the long poll timeout then the query is returned with no changes empty result.\n\nIf hash of the query result is used as a version then mutable state has to persist the map of hash(queryName + queryArgs) to hash(queryResult). And any new query request is checked against the correspondent value in this map. If it matches then it is registered with the mutable state. When a decision task due to an external event completes it includes query results. The query result hash is checked against the already stored result hashes. If the hash value has changed then the poll requests are notified.\n\n### Update operation that should complete within a single decision\n\nThis section is not fully designed yet.\n\n## Open issues\n\n1. Design update operation. There is an outstanding design issue for how update works in a multiple DC model.\n\n2. Agree on the client model for synchronous request reply primitives.\n"
  },
  {
    "path": "docs/design/2290-cadence-ndc.md",
    "content": "# Design doc: Cadence NDC\n\nAuthor: Cadence Team\n\nLast updated: Dec 2019\n\nReference: [#2290](https://github.com/cadence-workflow/cadence/issues/2290)\n\n\n## Abstract\n\nCadence Cross DC is a feature which asynchronously replicates workflows from active data center to other passive data centers, for backup & state reconstruction. When necessary, customer can failover to any of the data centers which have the backup for high availability.\n\nCadence Cross DC is an AP (in terms of CAP).\n\nThis doc explains the high level concepts about Cadence Cross DC (mainly the N data center version).\n\n\n## Newly Introduced Components\n\n1. Version\n2. Version history\n3. Workflow history conflict resolution\n4. Zombie workflows\n5. Workflow task processing\n\n### Version\n\nVersion is a newly introduced concept in Cadence Cross DC which describes the chronological order of events (per customer domain).\n\nCadence Cross DC is AP, all domain change events & workflow history events are replicated asynchronously for high throughput. This means that data across data centers are not strongly consistent. To guarantee that domain data & workflow data will achieve eventual consistency (especially when there is data conflict during a failover), version is introduced and attached to customers' domains. All workflow history events generated in a domain will also come with the version in that domain.\n\nAll participating data centers are pre-configured with a unique initial version, and a shared version increment:\n\n* `initial version < shared version increment`\n\nWhen performing failover for one domain from one data center to another data center, the version attached to the domain will be changed by the following rule:\n\n* for all versions which follow `version % (shared version increment) == (active data centers' initial version)`, find the smallest version which has `version >= old version in domain`\n\nWhen there is a data conflict, comparison will be made and workflow history events with the highest version will win.\n\nWhen a data center is trying to mutate a workflow, version will be checked. A data center can mutate a workflow if and only if\n\n* version in the domain belongs to this data center, i.e.\n  `(version in domain) % (shared version increment) == (this data centers' initial version)`\n* the version of this workflow's last event is equal or less then version in domain, i.e.\n  `(last event's version) <= (version in domain)`\n\nDomain version change example:\n\n* Data center A comes with initial version: 1\n* Data center B comes with initial version: 2\n* Shared version increment: 10\n\nT = 0: domain α is registered, with active data center set to data center A\n```\ndomain α's version is 1\nall workflows events generated within this domain, will come with version 1\n```\n\nT = 1: domain β is registered, with active data center set to data center B\n```\ndomain β's version is 2\nall workflows events generated within this domain, will come with version 2\n```\n\nT = 2: domain α is updated to with active data center set to data center B\n```\ndomain α's version is 2\nall workflows events generated within this domain, will come with version 2\n```\n\nT = 3: domain β is updated to with active data center set to data center A\n```\ndomain β's version is 11\nall workflows events generated within this domain, will come with version 11\n```\n\n### Version History\n\nVersion history is a newly introduced concept which provides high level summary about version information of workflow history.\n\nWhenever there is a new workflow history event generated, the version from domain will be attached. Workflow mutable state will keep track of all history events & corresponding version.\n\nExample, version history without data conflict:\n\n* Data center A comes with initial version: 1\n* Data center B comes with initial version: 2\n* Shared version increment: 10\n\nT = 0:  adding event with event ID == 1 & version == 1\n\nView in both data center A & B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 1               | 1       |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 1:  adding event with event ID == 2 & version == 1\n\nView in both data center A & B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 2               | 1       |\n| 2        | 1             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 2:  adding event with event ID == 3 & version == 1\n\nView in both data center A & B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 3               | 1       |\n| 2        | 1             |                 |         |\n| 3        | 1             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 3:  domain failover triggered, domain version is now 2\n        adding event with event ID == 4 & version == 2\n\nView in both data center A & B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 3               | 1       |\n| 2        | 1             | 4               | 2       |\n| 3        | 1             |                 |         |\n| 4        | 2             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 4:  adding event with event ID == 5 & version == 2\n\nView in both data center A & B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 3               | 1       |\n| 2        | 1             | 5               | 2       |\n| 3        | 1             |                 |         |\n| 4        | 2             |                 |         |\n| 5        | 2             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nSince Cadence is AP, during failover (change of active data center of a domain), there exist case that more than one data center can modify a workflow, causing divergence of workflow history. Below shows how version history will look like under such conditions.\n\nExample, version history with data conflict:\n\nBelow will show version history of the same workflow in 2 different data centers.\n\n* Data center A comes with initial version: 1\n* Data center B comes with initial version: 2\n* Data center C comes with initial version: 3\n* Shared version increment: 10\n\nT = 0:\n\nView in both data center B & C\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 2               | 1       |\n| 2        | 1             | 3               | 2       |\n| 3        | 2             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 1: adding event with event ID == 4 & version == 2 in data center B\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 2               | 1       |\n| 2        | 1             | 4               | 2       |\n| 3        | 2             |                 |         |\n| 4        | 2             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 1: domain failover to data center C, adding event with event ID == 4 & version == 3 in data center C\n```\n| -------- | ------------- | --------------- | ------- |\n| Events                   | Version History           |\n| -------- | ------------- | --------------- | ------- |\n| Event ID | Event Version | Event ID        | Version |\n| -------- | ------------- | --------------- | ------- |\n| 1        | 1             | 2               | 1       |\n| 2        | 1             | 3               | 2       |\n| 3        | 2             | 4               | 3       |\n| 4        | 3             |                 |         |\n| -------- | ------------- | --------------- | ------- |\n```\n\nT = 2:  replication task from data center C arrives in data center B\n\nNote: below are a tree structures\n```\n                | -------- | ------------- |\n                | Events                   |\n                | -------- | ------------- |\n                | Event ID | Event Version |\n                | -------- | ------------- |\n                | 1        | 1             |\n                | 2        | 1             |\n                | 3        | 2             |\n                | -------- | ------------- |\n                           |\n           | ------------- | ------------ |\n           |                              |\n| -------- | ------------- |   | -------- | ------------- |\n| Event ID | Event Version |   | Event ID | Event Version |\n| -------- | ------------- |   | -------- | ------------- |\n| 4        | 2             |   | 4        | 3             |\n| -------- | ------------- |   | -------- | ------------- |\n\n          | --------------- | ------- |\n          | Version History           |\n          | --------------- | ------- |\n          | Event ID        | Version |\n          | --------------- | ------- |\n          | 2               | 1       |\n          | 3               | 2       |\n          | --------------- | ------- |\n                            |\n                  | ------- | ------------------- |\n                  |                               |\n| --------------- | ------- |   | --------------- | ------- |\n| Event ID        | Version |   | Event ID        | Version |\n| --------------- | ------- |   | --------------- | ------- |\n| 4               | 2       |   | 4               | 3       |\n| --------------- | ------- |   | --------------- | ------- |\n```\n\nT = 2:  replication task from data center B arrives in data center C, same as above\n\n\n### Workflow History Conflict Resolution\n\nWhen a workflow encounters divergence of workflow history, proper conflict resolution should be applied.\n\nIn Cadence NDC, workflow history events are modeled as a tree, as shown in the second example in [### Version History].\n\nWorkflows which encounters divergence will have more than one history branches. Among all history branches, the history branch with the highest version is considered as the `current branch` and workflow mutable state is a summary of the current branch. Whenever there is a switch between workflow history branches, a complete rebuild of workflow mutable state will occur.\n\n\n### Zombie Workflows\n\nThere is an existing contract that for any domain & workflow ID combination, there can be at most one run (domain & workflow ID & run ID) open / running.\n\nCadence NDC aims to keep the workflow state as up-to-date as possible among all participating data centers.\n\nDue to the nature of Cadence NDC, i.e. workflow history events are replicated asynchronously, different run (same domain & workflow ID) can arrive at target data center at different times, sometimes out of order, as shown below:\n\n```\n| ------------- |          | ------------- |          | ------------- |\n| Data Center A |          | Network Layer |          | Data Center B |\n| ------------- |          | ------------- |          | ------------- |\n        |                          |                          |\n        | Run 1 Replication Events |                          |\n        | -----------------------> |                          |\n        |                          |                          |\n        | Run 2 Replication Events |                          |\n        | -----------------------> |                          |\n        |                          |                          |\n        |                          |                          |\n        |                          |                          |\n        |                          | Run 2 Replication Events |\n        |                          | -----------------------> |\n        |                          |                          |\n        |                          | Run 1 Replication Events |\n        |                          | -----------------------> |\n        |                          |                          |\n| ------------- |          | ------------- |          | ------------- |\n| Data Center A |          | Network Layer |          | Data Center B |\n| ------------- |          | ------------- |          | ------------- |\n```\n\nSince run 2 appears in data center B first, run 1 cannot be replicated as runnable due to rule `at most one run open` (see above), thus, `zombie` workflow state is introduced. Zombie state indicates a workflow which cannot be actively mutated by a data center (assuming corresponding domain is active in this data center). A zombie workflow can only be changed by replication task.\n\nRun 1 will be replicated similar to run 2, except run 1's workflow state will be zombie before run 1 reaches completion.\n\n\n### Workflow Task Processing\n\nIn the context of Cadence NDC, workflow mutable state is an entity which tracks all pending tasks. Prior to the introduction of Cadence NDC, workflow history events are from a single branch, and Cadence server will only append new events to workflow history.\n\nAfter the introduction of Cadence NDC, it is possible that a workflow can have multiple workflow history branches. Tasks generated according to one history branch maybe invalidated by history branch switching during conflict resolution.\n\nExample:\n\nT = 0: task A is generated according to event ID: 4, version: 2\n```\n| -------- | ------------- |\n| Events                   |\n| -------- | ------------- |\n| Event ID | Event Version |\n| -------- | ------------- |\n| 1        | 1             |\n| 2        | 1             |\n| 3        | 2             |\n| -------- | ------------- |\n           |\n           |\n| -------- | ------------- |\n| Event ID | Event Version |\n| -------- | ------------- |\n| 4        | 2             | <-- task A belongs to this event\n| -------- | ------------- |\n```\n\nT = 1: conflict resolution happens, workflow mutable state is rebuilt and history event ID: 4, version: 3 is written down to persistence\n```\n                | -------- | ------------- |\n                | Events                   |\n                | -------- | ------------- |\n                | Event ID | Event Version |\n                | -------- | ------------- |\n                | 1        | 1             |\n                | 2        | 1             |\n                | 3        | 2             |\n                | -------- | ------------- |\n                           |\n           | ------------- | -------------------------------------------- |\n           |                                                              |\n| -------- | ------------- |                                   | -------- | ------------- |\n| Event ID | Event Version |                                   | Event ID | Event Version |\n| -------- | ------------- |                                   | -------- | ------------- |\n| 4        | 2             | <-- task A belongs to this event  | 4        | 3             | <-- current branch / mutable state\n| -------- | ------------- |                                   | -------- | ------------- |\n```\n\nT = 2: task A is loaded.\n\nAt this time, due to the rebuilt of workflow mutable state (conflict resolution), task A is no longer relevant (task A's corresponding event belongs to non-current branch). Task processing logic will verify both the event ID and version of the task against corresponding workflow mutable state, then discard task A.\n"
  },
  {
    "path": "docs/design/active-active/active-active.md",
    "content": "# Design doc: Cadence Active-Active\n\nAuthor: [@taylanisikdemir](https://github.com/taylanisikdemir)\n\nLast updated: Apr 2025\n\n## Abstract\n\nMultiple Cadence clusters can be configured to form a cluster group. Each cluster group can have multiple domains.\nCurrently domains are the smallest unit of isolation which supports two modes:\n\n1. **Local Domain:** Only one cluster in the group hosts the domain. No replication.\n2. **Active-Passive Domain:** Domain is active in one cluster and passive in other cluster(s). All workflows of the domain are replicated to other specified passive clusters. See [NDC doc](../2290-cadence-ndc.md) for more details.\n\nThis document describes the design and implementation of active-active domains which is a new mode where a domain can be active in multiple clusters but individual workflows are only active in one cluster at any given time.\n\n## Background\n\nThe main drawbacks of Active-Passive approach are:\n\n- **Cross region call forwarding:** Limiting domain to a single region may lead to cross region dependency in hot paths if the user wants to process requests in multiple regions. Passive regions will forward requests to the active region. This leads to increased latency, cost and potential timeouts.\n\n- **Cross region data inconsistency:** Sticking with a single active Cadence region may lead to cross region data inconsistency issues for external data stores. Example:\n  - A request originates in Region-0.\n  - It writes something to a data store in Region-0 and starts a workflow for further processing.\n  - Domain is active in Region-1 which means Cadence will forward requests to Region-1 and workers in Region-1 will process the workflow.\n  - Worker in Region-1 is likely to read stale data from the data store in Region-1 because of eventual consistency of cross region data stores.\n\n- **Underutilized workers in passive region(s):** Domain owners must deploy workers to both regions to be ready for failover. Workers in the passive region(s) stay idle. In the event of a failover, workers in the passive region(s) will need to scale up and if this is not done quickly, workflows will observe delays (potentially time outs).\n\n- **Complexity of using multiple domains:** Current workaround to achieve active-active processing with Cadence is to create multiple domains and mark one of them active per region. E.g. domain-region0 and domain-region1. This requires users to remember which domain a specific workflow is created when they want to signal/describe the workflow. They can’t seamlessly create workflows in both regions for a single domain.\n\n\n## Design\n\nActive-active domains are a new mode of isolation where domain is active in multiple clusters but individual workflows are only active in one cluster at any given time. Users can create workflows in any of the active clusters under the same domain. Each workflow will be considered active in one of the domain's active clusters. The per-workflow active cluster selection mechanism will be dynamic/extensible to support different strategies.\n\nBefore diving into the details of active-active domains design, let's look at how active-passive domains work at high level. Then we will cover how active-active domains will be implemented in follow up sections.\n\n### Active-Passive domain's failover version\n\nAs the name suggests, an active-passive domain is active in one cluster and passive in other clusters. All the workflows of the domain are replicated to other passive clusters. Active cluster is determined by the failover version of the domain in the database.\n\n- Cluster group configuration:\n```\nclusterGroupMetadata:\n    failoverVersionIncrement: 10\n    clusterGroup:\n        cluster0:\n            initialFailoverVersion: 0\n            ...\n        cluster1:\n            initialFailoverVersion: 2\n            ...\n```\n\nActive-Passive domains take the initial failover version from the cluster group configuration when they are created based on their ActiveClusterName field in the domain configuration.\n\n\n- Failover version in domain records:\n\nFailover version is a property of active-passive domains which is updated as part of failover. Domain's failover version value uniquely maps to one of the clusters by using following formula: `domain.failoverVersion % clusterGroup.failoverVersionIncrement`. Let's look at an example:\n\n\n| Domain | Failover Version |\n|--------|------------------|\n| Domain-A | 0 |\n| Domain-B | 102 |\n\nDomain-A is active in cluster-1 because failover version `0 % 100 = 0` which belongs to cluster-1 based on above configuration.\nDomain-B is active in cluster-2 because failover version `102 % 100 = 2` which belongs to cluster-2 based on above configuration. Domain-B was failed over in the past, otherwise its failover version would be `0` or `2` (initialFailoverVersion in cluster group configuration).\n\n\n### Clusters and Regions\n\nThere's no \"region\" concept in Cadence today. However, with active-active domains, we need to introduce the concept of regions. The constraints are going to be:\n\n1. Each cluster belongs to one region.\n2. There can be multiple clusters in a region.\n3. A workflow of an active-active domain can ONLY be active in one cluster in a region.\n4. A workflow of an active-active domain can be passive in any number of clusters in any number of regions.\n\n![Active-Active One Cluster Per Region](./active-active-one-cluster-per-region.png)\n\nThe underlying reason for these constraints is mainly to reuse existing failover version to cluster mapping that relies on failover versions. By restricting active-active workflows of active-active domains to have only one active cluster per region, we can reuse the failover version mapping mechanism to determine the active cluster of a workflow.\n\nActive-active domains do NOT have a failover version in the database because \"activeness\" is not a property of a domain. Instead, workflows of active-active domains will be associated with an \"entity\" which has a failover version based on the region.\n\nThe region information will be defined in cluster group configuration. e.g.\n```\nclusterGroupMetadata:\n    primaryClusterName: cluster0\n    failoverVersionIncrement: 100\n    regions:  # new field.\n        us-west:\n            initialFailoverVersion: 1\n        us-east:\n            initialFailoverVersion: 3\n    clusterGroup:\n        cluster0:\n            initialFailoverVersion: 0\n            region: us-west\n        cluster1:\n            initialFailoverVersion: 2\n            region: us-west\n        cluster2:\n            initialFailoverVersion: 4\n            region: us-east\n        cluster3:\n            initialFailoverVersion: 6\n            region: eu-west\n```\n\n\nNotice that regions also have a failover version now. This will be used to determine the active cluster of a workflow based on following lookups:\n- Workflow maps to an entity (or directly to a region). This is static and cannot change over time.\n- Entity maps to a region. This is dynamic and can change over time.\n- Region maps to a cluster. This is static and cannot change over time. Note that there can be more than one cluster in a region but an active-active domain can only have one active cluster per region.\n\nIn other words,\n- All workflows of active-passive domains use the failover version of the cluster that it is active in.\n- Individual workflows of active-active domains use the failover version of the region that the entity that it is associated with belongs to.\n\n\nLet's look at a few examples:\n\n**Example 1:**\nAssume that we have an active-active domain called \"Domain-AA-Sticky-Region\" which exists in multiple regions and would like to have its workflows be active in the region they are started. We think this will be the most common use case so it's going to be the default behavior.\n\n**Example 2:**\nAssume that we have an active-active domain called \"Domain-AA-User-Location\" which exists in multiple regions and runs workflows per user. Each user's workflow is supposed to be processed by the cluster closest to the user's current location. Users can move around which means their workflow's active cluster will change over time.\n\n\n### Active cluster selection strategies\n\nThe active cluster per workflow selection strategies must support the examples given in the previous section and must be extensible to support other use cases. To drive such extensibility, we will introduce a new table called EntityActiveRegion to store <entity, region, cluster> mapping and the active cluster of each workflow (of an active-active domain) will be determined based on that information. The workflows themselves need to be associated with an entity which can be specified by the user in start workflow request.\n\n1. Pick the cluster that received the start workflow request\n   This is the default behavior and requires no additional configuration.\n\n2. Pick the cluster based on an external entity specified in the start workflow request.\n   This is useful when the user wants to dynamically change which cluster a workflow should be active in during the lifetime of workflow.\n   e.g. In \"Domain-AA-User-Location\" example, a workflow that is started in cluster-0 but user moved to another region and the workflow needs to be active in cluster-1 going forward.\n\n\nWorkflow start parameters:\n\n| Parameter | Description | Example |\n|-----------|-------------|---------|\n| active-region.lookup-source | Where to look up to determine the active region of the workflow for a given key. Ideally it should support the watch API. A corresponding watcher must be implemented within Cadence. | \"custom.user-location-provider\" |\n| active-region.lookup-key | The key to look up from the above source. | \"userid-123\" |\n\nThe lookup source \"custom.user-location-provider\" is a custom provider that can be implemented by the user. Initially we are not going to define an RPC protocol for this and it has to be implemented within Cadence.\n\n\nWorkflow start request determines which cluster selection strategy to be used.\n\nIf both `active-region.lookup-source` and `active-region.lookup-key` are set, then the workflow will be associated with the entity specified and the active cluster will be determined based on that entity's region.\n\nOtherwise, the default behavior is to pick the cluster that received the start workflow request.\n\n### Workflow activeness metadata in execution table\nA new row will be added to executions table as part of StartWorkflow LWT for active-active-domains. It will grab external entity source/key info from workflow’s headers if provided. If no such header is provided then current cluster’s region is stored as origin. These records are immutable and are replicated as part of regular per-shard replication tasks.\n\n| Shard | RowType | Domain | WFID | RunID | Data |\n|-------|---------|--------|------|-------|------|\n| 1     | ActivenessMetadata | test-domain | wf1 | 123 | Origin=us-west |\n| 2     | ActivenessMetadata | test-domain | wf2 | 456 | EntityType=user-location, EntityKey=london |\n\n\nIn the above example:\n- `wf1` is created on us-west. No external entity associations exist. Its tasks will have failover version of the active cluster in us-west specified in domain's `RegionToClusterMap` which is 0.\n- `wf2` is created with an external entity association. Its tasks will have failover version of following row in `EntityActiveRegion` table:\n<type=user-location, key=london> which is 5. Its tasks will be processed by cluster3 because version 5 maps to us-east region and cluster3 is active in us-east.\n- `wf3` is a long running workflow which was created when the domain was active-passive (migrated to active-active later on). It doesn't have an activeness metadata row in the executions table. Its existing tasks already have failover version of the cluster that it was active in so it will continue to be active in that cluster. Such workflows will continue to be processed with same failover version by the same cluster.\n\nNOTE: Long running workflows of active-passive domains like `wf3` will required the active-active domain to include pre-migration ActiveClusterName in active cluster config. Migrating a domain from active-passive to active-active and removing existing ActiveClusterName from active cluster config will break such workflows.\n\n\n### Domain records\n\nDomain records will have a new field called ActiveClusters which contains the information about region to cluster mapping and failover versions of the clusters. Active-active domains will have this field set and ActiveClusterName field will be empty.\nActive-passive domains will not have this field set and ActiveClusterName field will be set.\n\nDomain replication config:\n```\n{\n    // Clusters can be a subset of clusters in the group.\n    Clusters: \t    [cluster0, cluster1, cluster2, cluster3],\n\n    // Active clusters can have at most one from each region. Remaining clusters will be considered passive.\n    ActiveClusters:  {\n        RegionToClusterMap: {\n            us-west: {\n                ActiveClusterName: cluster0,\n                FailoverVersion: 0, # failover version of cluster0.\n            },\n            us-east: {\n                ActiveClusterName: cluster3,\n                FailoverVersion: 6, # failover version of cluster3.\n            },\n        },\n    },\n}\n```\n\n**RegionToClusterMap:**\n\nA map of regions to clusters that the domain can have active workflows in. It's initially formed based on cluster group configuration and given active clusters of the domain. Each region that the domain is active in will have an entry in this map.\n\nWorkflow task versions are assigned based on the failover version of the region that their associated entity belongs to.\nIf there's no entity associated with a workflow, then the task version is assigned based on the failover version of the region that the workflow is created on.\n\nActive-passive domains can be turned into active-active domains by setting the ActiveClusters field. There can be open workflows with task versions mapping to a cluster failover version instead of a region failover version. Mapping logic will support this fallback to support migration from active-passive to active-active domains.\n\ne.g.\nA long running workflow of an active-passive domain has task in the queue with version 0. This is supposed to be active in cluster0.\nIf we make this domain active-active, the task version should resolve to:\n- cluster0 if domain is active in cluster0 (direct mapping).\n- cluster1 if domain is active in cluster1 (fallback to active cluster of the region).\n- if domain has no active cluster in the region, then it's a resolution error and the task will not be processed as active anywhere.\n\n\n### EntityActiveRegion Lookup Table\nThis table is used to lookup the active cluster of a workflow based on the entity that the workflow is associated with.\n\n- Primary key of the table is <EntityType, EntityKey>\n- A new plugin system will be introduced to watch supported entity types and populate the table\n- Watchers are per cluster group\n    - There’s one primary cluster for the group and watcher runs on that cluster\n- This table is replicated to all clusters in the cluster group from the primary cluster\n- Failover Version\n    - Each entity record has a failover version to be used as workflow task versions\n    - Initially set based on cluster metadata\n    - Incremented based on Increment value from cluster metadata whenever mapping for a key changes\n\n\n| EntityType | EntityKey | Region | Failover Version | LastUpdated |\n|------------|-----------|--------|------------------|-------------|\n| user-location | seattle   | us-west    | 1                |             |\n| user-location | boston    | us-east    | 3                |             |\n| user-location | london    | eu-west    | 5                |             |\n| user-location | frankfurt | eu-central    | 7                |             |\n\n\nThis table is not domain specific so failover versions are not the failover versions of a cluster but instead the failover versions of a region. Region to cluster mapping is specific to domain.\n\nWorkflows associated with an entity will have task versions as indicated in the table above. Since these workflows are bound to a region, we cannot use the cluster failover versions.\n\nEntity updates are replicated to all clusters and history engines will subscribe to these updates to notify task queues. This is similar to how domain failover changes are handled today. Notifying the task queues wake up the queues to resume processing and they will be able to apply active/standby logic based on new failover versions.\n\n\n### Failover scenarios\n\n**1. Regional failover (disabling a region):**\nThis is a typical operation when the operator updates `ActiveClusterName` field in the domain record to failover the active-passive domain to one of the passive clusters.\nIt's called cluster failover because of active-passive domains where only one cluster can be active.\n\nIn the active-active domain context, this operation will be done by updating `ActiveClusters` field in the domain record. Arbitrary changes can be made to the `ActiveClusters` field as long as following criterias are satisfied\n- it's a subset of clusters in the cluster group.\n- it has at most one active cluster per region.\n\nFor example, let's say we want to failover cluster0 to cluster3. (us-west -> us-east).\n\nBefore failover:\n```\nActiveClusters:  {\n    RegionToClusterMap: {\n        us-west: {\n            ActiveClusterName: cluster0,\n            FailoverVersion: 0,\n        },\n        us-east: {\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n    },\n}\n```\n\nAfter failover:\n```\nActiveClusters:  {\n    RegionToClusterMap: {\n        us-west: { // failover version of us-west region is changed to 6 so that it maps to us-east region.\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n        us-east: {\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n    },\n}\n```\n\nFailover version to cluster mapping:\n- Workflows whose task version resolves to region us-west will start using version 6 from now on.\n- Workflows whose task version resolves to region us-east will continue using version 6.\n\nWhen the domain is in this state, the start workflowrequests (regardless of cluster selection strategy) will be forwarded to cluster3 and they will be sticky to that region.\n\n**2. Within region failover (changing active cluster of a region):**\n\nIn the example active-active domain above, the cluster0 & cluster1 are available in us-west region and cluster0 is active.\n\nIf we want to failover from cluster0 to cluster1, then we need to update the `RegionToClusterMap` in the domain record and replace cluster0 with cluster1.\n\nBefore failover:\n```\nActiveClusters:  {\n    RegionToClusterMap: {\n        us-west: {\n            ActiveClusterName: cluster0,\n            FailoverVersion: 0,\n        },\n        us-east: {\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n    },\n}\n```\n\nAfter failover:\n```\nActiveClusters:  {\n    RegionToClusterMap: {\n        us-west: {\n            ActiveClusterName: cluster1,\n            FailoverVersion: 2,\n        },\n        us-east: {\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n    },\n}\n```\n\nFailover version to cluster mapping:\n- Workflows whose task version resolves to region us-west will start using version 2 from now on.\n- Workflows whose task version resolves to region us-east will continue using version 6.\n\n\n**3. External entity failover:**\n\nThe external entity support is optional. If no entity is provided, the workflow will be associated with the region that the workflow is created on and they will be failed over as explained in the previous section.\n\nIf an external entity is provided for a workflow, then it can be failed over by updating the corresponding record in `EntityActiveRegion` lookup table.\n\nLet's say we want to failover the entity with key \"seattle\" to region us-east so it gets processed by cluster3.\n\nBefore failover:\n| EntityType | EntityKey | Region | Failover Version | LastUpdated |\n|------------|-----------|--------|------------------|-------------|\n| user-location | seattle   | us-west    | 1                |             |\n\nAfter failover:\n| EntityType | EntityKey | Region | Failover Version | LastUpdated |\n|------------|-----------|--------|------------------|-------------|\n| user-location | seattle   | us-east    | 3                |             |\n\nNotice that the failover version is updated to 3. This means that all workflows associated with this entity will be considered active in us-east region from now on. Since there can only be one active cluster per region, the corresponding workflows will be active in cluster3.\n\nFailover version to cluster mapping:\nThere's no change in the failover version to cluster mapping in entity failovers. Same versions map to same clusters as before.\nHowever, new task versions of workflows belonging to \"seattle\" entity will be 3 from now on and therefore they will be active in cluster3.\n\n\n**4. Active-passive to active-active migration:**\n\nWhen an active-passive domain is migrated to active-active,\n- The domain record will have the `ActiveClusters` field set\n- The existing `ActiveClusterName` field will be left as is.\n- The failover version of the domain will be incremented. Even though this is not used for task versions for active-active domains, it's incremented to indicate there was a change in replication config.\n\n\nBefore migration:\n```\nActiveClusterName: cluster0,\nFailoverVersion: 0\n```\n\nAfter migration:\n```\nActiveClusterName: cluster0,\nFailoverVersion: 100,\nActiveClusters:  {\n    RegionToClusterMap: {\n        us-west: {\n            ActiveClusterName: cluster0,\n            FailoverVersion: 0,\n        },\n        us-east: {\n            ActiveClusterName: cluster3,\n            FailoverVersion: 6,\n        },\n    },\n}\n```\n\n\n\n\n### API Call Forwarding\n\nCadence frontend proxies some API calls to the active cluster of the domain. Currently, this request to cluster lookup is done by checking the failover version of the active-passive domain.\nFor active-active domains, the lookup will be per workflow based on the active cluster of the workflow.\nSame set of APIs are going to be forwarded to the active cluster.\n\n#### APIs that are not subject to forwarding. Always served by current cluster:\n\n- Cluster APIs\n    - Health\n    - GetClusterInfo\n\n- Visibility APIs\n    - CountWorkflowExecutions\n    - GetSearchAttributes\n    - ListArchivedWorkflowExecutions\n    - ListClosedWorkflowExecutions\n    - ListOpenWorkflowExecutions\n    - ListWorkflowExecutions\n    - ScanWorkflowExecutions\n\n- Domain APIs\n    - DeprecateDomain\n    - DescribeDomain\n    - RegisterDomain\n    - UpdateDomain\n    - ListDomains\n\n- Tasklist APIs\n    - DescribeTaskList\n    - ListTaskListPartitions\n    - GetTaskListsByDomain\n    - ResetStickyTaskList\n\n- Worker APIs\n    - PollForActivityTask\n    - PollForDecisionTask\n    - RespondDecisionTaskCompleted\n    - RespondDecisionTaskFailed\n    - RespondQueryTaskCompleted\n\n- Per workflow APIs\n  - RestartWorkflowExecution\n  - RequestCancelWorkflowExecution\n  - RefreshWorkflowTasks\n  - DiagnoseWorkflowExecution\n  - SignalWithStartWorkflowExecutionAsync\n  - StartWorkflowExecutionAsync\n\nNote: Async API calls might be ok without forwarding (current state). We want these to be low latency and high throughput. So better to avoid checking workflow execution records. This means potentially duplicate workflow creations in active-active mode if user retries in multiple regions. Check the reconciliation logic that is supposed to dedupe these. We may revisit this in the future.\n\n#### APIs that are forwarded to active side (selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2):\n\n- Per workflow APIs\n  - StartWorkflowExecution\n  - SignalWorkflowExecution\n  - SignalWithStartWorkflowExecution\n  - ResetWorkflowExecution\n  - TerminateWorkflowExecution\n  - QueryWorkflow: Forwarded to active cluster if strong consistency parameter is set to true.\n  - DescribeWorkflowExecution: Forwarded to active cluster if strong consistency parameter is set to true. (This is not supported as of Apr 2025 but the work is in progress.)\n  - GetWorkflowExecutionHistory: Forwarded to active cluster if strong consistency parameter is set to true. (This is not supported as of Apr 2025 but the work is in progress.)\n\n- Worker APIs\n  - RecordActivityTaskHeartbeat\n  - RecordActivityTaskHeartbeatByID\n  - RespondActivityTaskCanceled\n  - RespondActivityTaskCanceledByID\n  - RespondActivityTaskCompleted\n  - RespondActivityTaskCompletedByID\n  - RespondActivityTaskFailed\n  - RespondActivityTaskFailedByID\n\n\n## Limitations\n\nBelow is a list of limitations of active-active domains.\n\n- **Passive side tasklist processing:** Tasklist processing will be disabled for workflows that are passive in a cluster.\nThis is to avoid mixing tasklists of a domain with active and passive tasks. Instead, workflows will resume processing after failover by relying on decision/activity timeout tasks.\nMisconfigured (very high) timeout values can lead to workflows not being processed for a long time which is already a risk in active-passive mode.\n\n- **History Queue V2 dependency:** History Queue V2 must be enabled in the cluster to be able to use active-active domains.\nLegacy queue implementation is not handling initial backoff timer tasks correctly for active-active domains. See comments in `allocateTimerIDsLocked`.\nV2 queues can be enabled via dynamic config.\n```\nhistory.enableTransferQueueV2:\n  - value: true\nhistory.enableTimerQueueV2:\n  - value: true\n```\n\n- **Cassandra dependency:** Cassandra is the only supported persistence store for active-active domains.\nThis is because the new entity region lookup table is stored in Cassandra.\nThis is a temporary limitation and we will support other persistence stores in the future.\n\n- **Workflow id conflict:** Multiple clusters can start workflows independently with the same workflow id. This can lead to conflicts in active-active mode.\nConflict resolution mechanism should take care of this by terminating one of the workflows.\n\n- **Graceful failover:** Graceful failover is not supported for active-active domains. It is not a mode that is frequently used for active-passive domains either.\n\n- **External entity cardinality:** All cadence frontend and history services will have to make quick decisions on which cluster a workflow is active in. This requires caching all the domain data (already done for active-passive domains) and also caching the new entity region lookup table. This cache should be manageable in terms of memory usage so there will be a limit to the number of entities. Actual limit is going to be configurable based on available memory but ideally it should be in the order of thousands.\n\n\n\n## Defining a new external entity type\n\n// TODO(active-active): define external entity providers as plugins. See resource_impl.go\n"
  },
  {
    "path": "docs/design/domain-updates/fencing-tokens.md",
    "content": "# Domain versions and fencing tokens with Cassandra\n\nLast Updated: October 2025\n\n### Background \n\nCadence uses [fencing tokens](https://martin.kleppmann.com/2016/02/08/how-to-do-distributed-locking.html) heavily due to it's reliance on databases which don't allow for cross-table transactions (in Cassandra particularly). In most instances, they're used simply as a counter, to perform optimistic updates with a compare-and-set, such that if they're operating in stale data, it's integrity can be protected. However, there's a few cases where they pass additional information such as the cluster that's active as well.\n\nThere's quite a few of these tokens which have different purposes, particularly on the Domain side, and keeping track of their intent is a little tricky. This is a basic catelog of their mechanism and intent with respect to domains. The intent of this document is to provide something of an overview.\n\n### Notification version: \n\n**Purpose**: cluster-level version guard for domain updates to prevent racing writes for the domains tables.\n**Where it’s set**: on every domain update/create/delete it’s set via a conditional write to the following magical ‘domain’ `cadence-domain-metadata` record:\n\nIn Cassandra, this record is read like this::\n```sql\nSELECT notification_version FROM domains_by_name_v2 WHERE domains_partition = 0 and name = 'cadence-domain-metadata'\n```\n\nIt's therefore set at the cluster level, though captured at a point in time on indivial domains (see FailoverNotificationVersion)\n\nThis is a simple monotonic counter. It goes up by \\+1 every domain update for any domain in the cluster. In Cassandra, writes to the domains table are done with a conditional write to the `cadence-domain-metadata` pseudo-record as a means to guard against out-of-order updates.\n\nIn practice these values move up together accross clusters whenever there is a global domain update due to the domain's data being replicated, but these notification values are **not** comparable. For example, consider the following scenario.\n\n### Failover Versions in each cluster:\n\n| Cluster  | notification_version value | \n| ---------|----------------------------|\n| cluster0 | 10                         | \n| cluster1 | 10                         | \n\nWhen a new global domain, replicated to both these clusters is registered, the `notification_version` will be increemented in both\n\n| Cluster  | notification_version value | \n| ---------|----------------------------|\n| cluster0 | 11                         | \n| cluster1 | 11                         | \n\nHowever, subsequently, if a 3 local domains, which are not replicated are registered in cluster1, the following will occur:\n\n| Cluster  | notification_version value | \n| ---------|----------------------------|\n| cluster0 | 11                         | \n| cluster1 | 14                         | \n\n### Domain FailoverVersion\n\n**Purpose:** (in normal domains): \n\n- To indicate which cluster is active\n- Specifies which domain updates are more recent\n- Allows workflows to merge history branches with last-write-wins \n\nThis is a somewhat more complicated counter which also indicates where it was generated (as in, which cluster generated the event). It is a monotonic counter, always going up, but the values by which it go up are incremented to indicate their origin, with something like a bitmask at the end of the counter acting as an identifier of the cluster (it's actually modulo, but it is used like a networking bitmask).\n\nIt uses the 'initial-failover-version' - an arbitrary unique int acting as a cluster identifier set a the cluster configuration level. The value changes for each failover. This is perhaps best illustrated with an example:\n\n##### Example:\n\nGiven three clusters with the following intitial-failover-versions:\n```\nclusterA: 0\nclusterB: 1\nclusterC: 2\n```\n\nAnd given a `failover-increment` of `100` (this the value used to modulo the origin cluster)\n\nAnd given a domain `test-domain`, available and replicated to all three clusters, \n\n- It is first registered (arbitrarily) to be active in clusterA, in which case the `FailoverVersion` is set to `0`. \n- If it is failed over to clusterB, then the `FailoverVersion` will become `1`.\n- If it is failed back to clusterA, then it will jump up to the next increment and be `100` (which, modulo 100, gives 0, representing `clusterA`)\n\nSince this value is attached to each workflow as it's executing and each history event as it's being written, each history event therefore can be value can therefore be modulo'd by the `failover-increment` to get its origin and given two history events in a history branch, the more recent can be picked by the value being higher (since history branches in both clusters for a given workflow may diverge during a failover briefly).\n\n### FailoverNotificationVersion \n\nThis is unfortunately similarly named to the Failover version, but it has a different purpose:\n\nThis is value which captures the current state of the domains at the point of processing them through the replication stack and during Graceful Failover. The values are **not** replicated and **not** comparable across clusters and its purpose is to prevent domain replication messages being replayed out of order.\n\n**Where it’s set:** \n\nWhen a domain is replicating, in the worker that’s processing the message, it grabs the current notification version on the side that’s taking the incoming domain update message and adds it to the fields being written.\n\n\n```go\n// common/domain/replicationTaskExecutor.go running in the worker service as a faux singleton \nfunc (h *domainReplicationTaskExecutorImpl) handleDomainUpdateReplicationTask(ctx context.Context, task *types.DomainTaskAttributes) error {\n\n\t// first we need to get the current notification version since we need to it for conditional update\n\tmetadata, err := h.domainManager.GetMetadata(ctx)\n\tif err != nil {\n\t\th.logger.Error(\"Error getting metadata while handling replication task\", tag.Error(err))\n\t\treturn err\n\t}\n\t// this is the global notification_version for all domains\n\tnotificationVersion := metadata.NotificationVersion\n\t\n\tresp, err := h.domainManager.GetDomain(ctx, &persistence.GetDomainRequest{\n\t\tName: task.Info.GetName(),\n\t})\n\n\t// Where it will be set if the FailoverVersion is more recent then recently seen\n\t// ie, the failover event is more recent then stored in the DB\n\tif resp.FailoverVersion < task.GetFailoverVersion() {\n\t\t// ...\n\t\trequest.FailoverNotificationVersion = notificationVersion\n\t\t// ...\n\t}\n\n\t// ... where it will be written to the DB\n\treturn h.domainManager.UpdateDomain(ctx, request)\n```\n\n### Where is it used \n\nThe failoverNotificationVersion is used as a guard to avoid out of order updates in the domain callback. \n\nThis is running in all services, in the domain cache, and is triggered as a part of its internal refresh loop. It’s used to trigger a graceful failover trigger in history. \n\n### How it increments: \n\n- Every time there's a notification via the domain replication stack, the domains are comparing their current FailoverNotificationValue to the value (?). \n\nGiven the following state for the domain `test35`:\n\n| Cluster  | value | \n| ---------|-------|\n| cluster0 | 19    | \n| cluster1 | 21    | \n| cluster2 | 20    | \n\nAnd the following metata values in the record `cadence-domain-metadata`\n\n| Cluster  | value | \n| ---------|-------|\n| cluster0 | 43    | \n| cluster1 | 35    | \n| cluster2 | 33    | \n\nAfter failing over domain `test35`, the expected values for the FailoverNotificationVersion in each cluster for the domain `test35` is: \n\n| Cluster  | value | \n| ---------|-------|\n| cluster0 | 43    | \n| cluster1 | 33    | \n| cluster2 | 32    | \n\nAnd the following metata values in the record `cadence-domain-metadata`. These will have been updated by the domain data replicating\n\n| Cluster  | value | \n| ---------|-------|\n| cluster0 | 44    | \n| cluster1 | 36    | \n| cluster2 | 33    | \n\n\n### Active-Active domains and ClusterAttributes\n\nActive-active domains split their notion of their workflows into subgroups called 'ClusterAttributes' which can be used by workflows using the domain to determine which cluster is active. They are a sub-implementation of FailoverVersion, ie the domain may have multiple ClusterAttributes, each with their own Version which functions similarly to the normal FailoverVersion.\n\nIe, a domain may validly have the following configuration:\n\nFor example, given three replicating clusters, with the following intitial-failover-versions\n```\nclusterA: 0\nclusterB: 1\nclusterC: 2\n```\n\nAnd a failover increment of 100:\n\nFor example: An active-active domain may have the following (this is a JSON representation of the domain `desc` command of a sample domain)\n\n```\n{\n  \"domainInfo\": {\n    \"name\": \"active-active-domain\",\n    \"status\": \"REGISTERED\",\n    \"uuid\": \"177b38ef-ed54-4c0e-b84b-1b52e3c3598e\"\n  },\n  \"replicationConfiguration\": {\n    \"activeClusterName\": \"cluster2\",\n    \"clusters\": [\n      {\n        \"clusterName\": \"cluster0\"\n      },\n      {\n        \"clusterName\": \"cluster1\"\n      },\n      {\n        \"clusterName\": \"cluster2\"\n      }\n    ],\n    \"activeClusters\": {\n      \"attributeScopes\": {\n        \"location\": {\n          \"clusterAttributes\": {\n            \"cityA\": {\n              \"activeClusterName\": \"cluster0\",\n              \"failoverVersion\": 10\n            },\n            \"cityB\": {\n              \"activeClusterName\": \"cluster2\",\n              \"failoverVersion\": 12\n            }\n          }\n        }\n      }\n    }\n  },\n  \"failoverVersion\": 52,\n  \"isGlobalDomain\": true\n}\n```\n\nIn this instance, the domain has a 'default' active cluster (cluster2, indicated by the failoverVersion 52), as well as two cluster-attributes for the 'cities' attribute: `cityA` which is active in cluster0 and `cityB` which is active in cluster2. These can be failed over and active in clusters independently of the domain's default active cluster. \n\nWorkflows using either of these cluster-attributes will only ever be active in one cluster at a time - that restriction remains."
  },
  {
    "path": "docs/design/graceful-domain-failover/3051-graceful-domain-failover.md",
    "content": "# Design doc: Cadence graceful domain failover\n\nAuthor: Cadence Team\n\nLast updated: Mar 2020\n\nReference: [#3051](https://github.com/cadence-workflow/cadence/issues/3051)\n\n\n## Abstract\n\nCadence supports domain failover with a multi-cluster setup. However, the problems with the current failover are:\n1. Workflow progress could be lost.\n2. No causal consistency guarantee on the domain level.\n\nThe graceful domain failover uses to solve those two problems.\n\n## Use cases\n\nUsers can trigger graceful domain failover via Cadence CLI (timeout is optional):\n\n`cadence --domain cadence-global domain update -failover_type graceful -active_cluster XYZ -failover_timeout 120s`\n\nUsers can force complete domain failover via Cadence CLI:\n\n`cadence --domain cadence-global domain update -failover_type force -active_cluster XYZ`\n\n## Prerequisites\nThere are conditions before starting a graceful domain failover.\n1. All clusters are available.\n2. The networking between clusters is available.\n3. The domain state has to be stable before graceful failover.\n\n## Limitation\nNo concurrent graceful domain failover will be supported.\n\n## Proposal\n\nDue to the complexity of the protocol, it will go through the architecture from cross-cluster level, single cluster level to the host level.\n\nThe underlying protocol is to insert markers in the active-to-passive cluster to indicate the boundary when the domain switches to passive. On the other side, the passive-to-active cluster listens to those markers and switches domain to active after receiving all the markers.\nBesides, after the failover marker inserted in the active-to-passive cluster, the shard cannot accept new external requests. Before the passive-to-active receives all failover marks, no shard will start to process tasks with active logic.\n\n### Cross cluster level\n\n![cross clusters sequence diagram](3051-cross-clusters.png)\n1, The operator issues a graceful failover to the passive-to-active cluster.\n\n2 - 3, The passive-to-active cluster gets domain data from all other clusters for two purposes: 1. Make sure the network and clusters are available before starting the graceful failover. 2. Make sure there is no ongoing failover.\n\n4 - 5, If the check fails, return an error to the operator indicating the graceful failover abort.\n\n6, After the graceful failover initiated, cluster Y updates the domain to pending_active with a higher failover version to database.\n\n7, Respond to the operator indicating the graceful failover initiated.\n\n8, The domain update event in step 6 replicates to cluster X.\n\n9, Cluster X updates the domain with the higher failover version and sets the domain to passive.\n\n10, Each shard receives a domain failover notification. The shard persists the pending failover marker and inserts the marker to the replication queue.\n\n11, The inserted failover marker replicates to cluster Y.\n\n12, Each shard in cluster Y listens to the failover marker and reports the ‘ready’ state to the failover coordinator after it receives the failover marker. Even if the shard receives the failover marker, the shard won't process tasks as active until all shards receive failover markers.\n\n13, The failover coordinator updates domain from pending_active to active when received ‘ready’ signal from all shards.\n\n14, The failover coordinator updates domain from pending_active to active when the timeout hits and regardless of how many ‘ready’ signals it received.\n\nFrom the high-level sequence diagram, it explains how the protocol works within multi-clusters. There is detail at the cluster level.\n\n### Cluster X\n![cross cluster X sequence diagram](3051-clusterX.png)\n\n1, Frontend receives a domain replication message.\n\n2, Frontend updates the domain data in the database with the active cluster sets to Cluster Y and a higher failover version.\n\n3, Domain cache fetches domain updates in a refresh loop.\n\n4, The database returns the domain data.\n\n5-6, After the domain updates, domain cache sends a domain failover notification to each shard.\n\n7, Shard 1 updates the shard info with a pending failover marker to insert.\n\n8, Shard 1 try to insert the failover marker and remove the pending failover marker from shard info after successful insertion.\n\n9, Shard 2 updates the shard info with a pending failover marker to insert.\n\n10, Shard 2 try to insert the failover marker and remove the pending failover marker from shard info after successful insertion.\n\n### Cluster Y\n![cross cluster Y sequence diagram](3051-clusterY.png)\n\n1, The graceful domain failover request sends to the Frontend service.\n\n2, Frontend updates the domain in the database with a flag indicating the domain is Pending_Active.\n\n3, Domain cache fetches domain updates in a refresh loop.\n\n4, The database returns the domain data.\n\n5, After the domain updates, domain cache sends a domain failover notification to each shard.\n\n6, After the domain updates, domain cache sends a domain failover notification to each shard.\n\n7, In shard 1, the engine notified the coordinator about the domain failover.\n\nHappy case:\n\n8, Shard 2 receives the failover marker.\n\n9, Shard 1 receives the failover marker.\n\n10, Shard 2 reports the ‘ready’ state to Coordinator.\n\n11, Shard 1 reports the ‘ready’ state to Coordinator.\n\n12, The coordinator persists the states from each shard.\n\nFailure case:\n\n13, Shard2 does not receive the failover marker.\n\n14, Shard 1 receives failover marker.\n\n15, Shard 1 reports the ‘ready’ state to Coordinator.\n\n16, The graceful failover timeout reached.\n\nAfter:\n\n17, Coordinator update domain to active via frontend.\n\n18, Frontend updates the domain in the database with the active state.\n\n19, Domain cache fetches domain updates in a refresh loop.\n\n20, The database returns the domain data.\n\n21, After the domain updates, domain cache sends a domain failover notification to each shard.\n\n22, After the domain updates, domain cache sends a domain failover notification to each shard.\n\n23, Shard 1 starts to process events as active.\n\n24, Shard 2 starts to process events as active.\n\n## Implementation\n\nNew components:\n1. New state in domain\n2. New task processor\n3. Failover marker\n4. Failover coordinator\n5. Buffer queue\n\n### Domain\n\nA new state \"Pending_Active \"introduced when domain moves from Passive to Active. When domain is in Pending_Active, no task will be processed as active.\n\n![Domain state transition](3051-state-transition.png)\n\nActive to Passive: This happens when a domain failover from ‘active’ to ‘passive’ in the cluster.\n\nPassive to Pending_Active: This happens when a domain failover from ‘passive’ to ‘active’ in the cluster. In this pending-active cluster, it first updates domain state to pending.\n\nPending_Active to Passive: This happens when the domain is in ‘pending_active’, the coordinator receives a domain failover notification with a higher version and the domain failovers to another cluster. Then the domain moves back to passive.\n\nPending_Active to Active: The coordinator moves domain from ‘pending_active’ to ‘active’ in the scenarios:\n1. All shards received the failover notification and failover markers.\n2. The failover timeout reaches, and the domain is not ‘active’.\n\n### Task processor\n\nAs the new state introduced during graceful domain failover in the passive cluster, a new task processor introduces here to handle the task in Pending_Active state.\n\nTransfer: Blocked on processing tasks during failover and continue to process tasks after domain switches to active.\n\nTimer: Blocked on processing tasks during failover and continue to process tasks after domain switches to active.\n\n### Failover marker\nFailoverMarker {\n\n    *replicationTask\n    failoverVersion int64\n    sourceCluster string\n    Timestamp time.Time\n}\n\n### Failover coordinator\nWith the graceful failover protocol, we need to maintain a global state (in the same cluster) of all shards. So we need a new component for it. The coordinator could be a stand-alone component or elect a leader from the shards. This new component is to maintain a global state of all shards during a failover.\nTo keep the global state, each shard does heartbeat to the coordinator to send the last X minutes failover marker (X is the max graceful failover timeout we support).\n\nThe coordinator persists the state in memory and updates this state to database periodically. The state can be stored in the shard table. The state struct looks like:\nmap[string][]*int32\nThe key contains the domain and the failover version.\nThe value is a slice of shard ID.\nFailover timeout\nCurrently, each history host has a component domain cache. Each shard on the same host gets domain failover notification from the domain cache. Domain cache periodically checks the database and updates the domain in memory. The failover timeout can leverage this component.\n\nDuring the graceful failover domain update, we record the timeout in the domain data. Domain cache reads all domain data periodically and checks if any of the graceful failover should be timed out. If the graceful failover times out, the domain cache sends a notification to shard to update the domain from pending_active to active.\n\n### [Open for discussion] Buffer queue\n\nDuring the graceful failover, the task processing pauses. New external requests will be rejected when the shard already inserted the failover marker. This may causes problems if the caller cares about the availability of the APIs.\nOne of the options is to provide this buffer queue in the passive-to-active cluster to buffer those external requests. The trade-off of this feature is that it introduces complex to invalids the requests in the buffer queue.\nThis feature is independent of the graceful failover. But we list the option here for further discussion.\n\nThose APIs includes:\n1. StartWorkflowExecution\n2. SignalWithStartWorkflowExecution\n3. SignalWorkflowExecution\n4. CancelWorkflowExecution\n5. TerminateWorkflowExecution\n\nWith the current architecture, we can store those events in a queue. This queue has three types of processors.\n\nActive processor: process the messages with the active logic.\n\nPendingActive processor: Do not process messages in the buffer queue.\n\nPassive processor: forward the messages to the active cluster.\n\n#### Handle signal/cancel/terminate\n1. Send a remote call to the source cluster to get the workflow state.\n2. Add the message to the buffered queue if the workflow is open.\n\n#### Handle start workflow\n1. Send a remote call to the source cluster to make sure there is no open workflow with the same workflow id.\n2. Inserts a start workflow task in the buffer queue and creates the mutable state, history event (no timer or transfer task will be generated).\n\nThe purpose of the start workflow task is to regenerate the timer tasks and transfer tasks once the domain becomes active.\n\nThe purpose of the mutable state and history event is to record the workflow start event for deduplication and generate replication tasks to other clusters to sync on the workflow data.\n\nThe generated history events will be replicated to all clusters. This process is required because all clusters should have the same workflow data.\n"
  },
  {
    "path": "docs/design/history-queue-v2/history-queue-v2.md",
    "content": "# Design doc: History Queue V2\n\nAuthor: [@Shaddoll](https://github.com/Shaddoll)\n\nLast Updated: Jun 2025\n\n## Abstract\n\nCadence is a multi-tenant workflow orchestration platform used by multiple customers. It currently runs three types of history queues within the History Service: the transfer queue, timer queue, and replication queue. These queues are responsible for:\n\n- Loading history tasks from database\n- Submitting history tasks to history task scheduler for task execution\n- Periodically persisting queue states (i.e., processing checkpoints) to the database so they can be resumed upon shard restarts\n\nHistory Queue V1 suffers from poor isolation between domains, leading to noisy neighbor issues. Additionally, its implementation is poorly structured and prone to bugs.\n\nHistory Queue V2 aims to replace V1 with a more robust and testable design, offering better code quality and improved task isolation.\n\n## Architecture\n![History Queue V2 Components](./history-queue-v2.png)\n\nLet's take transfer queue as an example. The above diagram displays the architecture of history queue v2.\n## Implementation Details\n\nBelow is a breakdown of the main components, ordered from low-level to high-level:\n\n**Virtual Slice**: This is the atomic component responsible for loading tasks from database with a given range and filter.\n\n**Virtual Queue**: This is the component running a background job calling virtual slice to load tasks from database and submit history tasks to history task scheduler for execution. It has multiple virtual slices ordered by their range and they don't have any intersection. And the background job processes virtual slices sequentially in the order of task range.\n\n**Virtual Queue Manager**: This is the parent component of virtual queues. Since each virtual queue has its own background job, they are isolated and that's how we achieve isolation in history queue v2. Inside a virtual queue manager, there is at least a root virtual queue component and optionally, there might be some other virtual queues. Virtual queues can be dynamically created and destructed, and the intersection of tasks processed by different virtual queues should be none. By default, there is only 1 root virtual queue.\n\n**Queue Reader**: This is the component responsible for reading tasks from database on demand and it's shared by all virtual slices in the history queue v2 instance.\n\n**Monitor & Mitigator**: Monitor is a component checking the queue processing of virtual queues and making decision to create new virtual queue. Mitigator is the component to perform the decision made by monitor. These 2 components won't be implemented in the initial release of history queue v2.\n\n**Transfer Queue**: This is one instance of history queue v2, which is responsible for the critical jobs of transfer tasks. There is also timer queue and replication queue today and new category of history queues will be created in future.\n"
  },
  {
    "path": "docs/design/index.md",
    "content": "# Design Documents\n\n- Resource Specific Tasklist [1533-host-specific-tasklist.md](1533-host-specific-tasklist.md)\n- Synchronous Request Reply [2215-synchronous-request-reply.md](2215-synchronous-request-reply.md)\n- N Data Center Replication [2290-cadence-ndc.md](2290-cadence-ndc.md)\n- Graceful domain failover [3051-graceful-domain-failover.md](graceful-domain-failover/3051-graceful-domain-failover.md)"
  },
  {
    "path": "docs/design/workflow-shadowing/2547-workflow-shadowing.md",
    "content": "# Design doc: Cadence Workflow Shadowing\n\nAuthor: Yu Xia (@yux0), Yichao Yang (@yycptt)\n\nLast updated: Apr 2021\n\nReference: [#2547](https://github.com/cadence-workflow/cadence/issues/2547)\n\n## Abstract\n\nCadence client libraries use workflow history and workflow definition to determine the current workflow state which is used by the workflow (decision) task for deciding the next step in the workflow. Workflow histories are immutable and persisted in the Cadence server, while workflow definitions are controlled by users via worker deployments.\n\nA non-deterministic error may happen when re-building the workflow state using a version of the workflow definition that is different than the one generating the workflow history. Typically caused by a non-backward compatible workflow definition change. For example:\n\n| Version | Workflow Definition | Workflow History |\n| --------| --------------------| ---------------- |\n| V1 | 1. Start workflow<br />2. Start activity A<br />3. Complete workflow | 1. Workflow started<br />2. Activity A started |\n| V2 | 1. Start workflow<br />2. Start activity B<br />3. Complete workflow | 1. Workflow started<br />2. Activity B started |\n\nWhen a Cadence client uses V1 history and V2 definition to build the workflow state, it will expect information of activity A in the workflow state as it sees an activity started event for it. However, it's unable to find Activity A since another activity is specified in V2 definition. This will lead to a non-deterministic error during workflow (decision) task processing.\n\nDepending on the non-deterministic error handling policy, the workflow will fail immediately or get blocked until manual operation is involved. This type of error has caused several incidents for our customers in the past. What make the situation worse is that understanding and mitigating the non-deterministic error is usually time-consuming. Blindly reverting the bad deployment will only make the situation worse as histories generated by the new workflow definition are also not compatible with the old one. The right mitigation requires not only manual operations, but also a deep understanding of why the non-deterministic error is happening, which will greatly increase the time needed to mitigate the issue.\n\n### Goals\n\n- The goal of this project is to detect the non-deterministic errors in pre-prod environments, notify customers on this error and reduce the number of incidents caused by the non-deterministic errors.\n\n### Non-Goals\n\n- 100% prevent non-deterministic errors: this project will not provide a bulletproof solution to prevent non-deterministic errors from happening in production.\n- Provide mitigation solutions to non-deterministic errors: this project will not focus on how to solve existing non-deterministic errors for customers.\n\n## Proposal\n\nThe proposal is creating a new worker mode called shadowing which reuses the existing workflow replay test framework available in both Golang and Java clients to only execute replay tests using the new workflow definition and workflow histories from production environments.\n\nWhen the user worker is running in the shadowing mode, the normal activity and decision worker component will not be started, instead a new replay worker will be started to replay production workflow histories. More specifically, this replay worker will call the `ScanWorkflowExecution` API to get a list of workflow visibility records and then for each workflow, get the workflow history from Cadence server by calling `GetWorkflowExecutionHistory` API and replay it against the new workflow definition using the client side replay test framework. Upon replay failure, metrics and log messages will be emitted to notify the user that there are non-deterministic changes in the new workflow definition.\n\nNote that as long as the worker can talk to the Cadence server to fetch visibility records and workflow history, it can be run in the shadowing mode. This means users have the flexibility to run it in the local environment during development to facilitate a rapid development cycle or/and make it part of the deployment process to run on a dedicated host in staging/preprod environment to get a better test coverage and catch rare cases.\n\nUsers will also have the option to specify what kind of workflows they would like to shadow. As an example, if the user doesn’t use the query feature, then shadowing for the closed workflow may not necessary. Or if a user only changed the definition for one workflow type, only workflow histories of that type need to be shadowed and checked. Please check the Implementation section for the detailed options users have.\n\nThe major advantages of this approach are:\n- **Simplicity**, very little code change required for Cadence server\n- **Can be run in both local and staging/preprod/canary environments**\n- **Generic approach that fits our internal deployment process, CI/CD pipeline and also open source customers**.\n\nThe downsides however are:\n- **Requires some effort on the user end** to setup the release pipeline and shadowing environment and we need to provide guidelines for them.\n- The **load on Cadence server and our ElasticSearch cluster will increase** due to the added `GetWorkflowExecution` and `ScanWorkflowExecutions` call. But this increase can be easily controlled by the server side rate limiter. Check the “Open Questions” section for an estimate for the load.\n- The dedicated **shadow worker will become idle after the replay test is done**.\n\n![deployment pipeline](2547-deployment-pipeline.png)\n\n## Other Considered Options\n\n### [History Generation with task mocking](https://github.com/cadence-workflow/cadence-go-client/issues/1050)\n\nThe idea of this approach is similar to the local replay test framework. But instead of asking customers to get an existing workflow history from production, it will ask customers to define possible workflow inputs, activity input/output, signals, etc. and then generate multiple workflow histories based on different combinations of values based on a certain workflow version. Random errors will also be injected to cover failure cases. Upon workflow code change, those generated workflows can be replayed against the new workflow definition to see if there are any non-deterministic changes.\n\nThe approach can be viewed as some form of automated test where test cases are auto-generated based on various inputs to the workflow. As it can validate the changes through local tests, **the development cycle can be faster**. This could improve the local development experience and could be considered in vNext phase.\n\nHowever, this also means that **users need to learn a set of new APIs** for specifying different types of inputs and update the test if there are new activities or signal channels. **Coverage might also be a concern** as many inputs are coming from upstream services, like Kafka, and it’s hard to specify all different possible parameters. Finally, from an implementation perspective, the new automated test framework is **language specific** and needs the effort to re-design API and re-implement for different client libraries.\n\n### Integration test with archived (partial) history\n\nThis approach reuses existing history archival code paths to dump workflow histories that match a certain query to blob storage. When a user changes the workflow code, they can connect to the blob storage, and run an integration test locally to replay all of the archived history to see if there’s a non-deterministic error. Since not all blob storage support List operation when there's a large number of items, we may also need to rely on visibility archival for getting a list of workflowIDs and runIDs.\n\nThe approach **doesn’t require too many changes on server end** as most code paths for history/visibility archival can be reused. We only need to figure out when and how to sample the workflow for archiving, and the destination for archiving. It also **avoids the idle worker problem** and **won’t incur any load increase on Cadence server** as the testing framework will talk directly to the storage for storing archived history and visibility records.\n\nThere are also several downsides associated with this approach:\n- We need to decide who owns and manages the blob storage.\n  - If Cadence owns it, GDPR constraints might be a concern as some workflow histories will contain PII data and can’t be archived for too long. **A set of APIs for managing the storage** will also be necessary.\n  - Register user callbacks for managing storage.\n  - If users own it, it will **increase their burden for managing it**. E.g. regularly clean up the unneeded workflows (may due to GDPR constraints), update query to replay only recent workflows to ensure a reasonable test time.\n- Extra work for setting additional history and visibility archival systems.\n- The integration test may take a long time to replay all the archived workflows **making it not ideal to run locally**.\n- Since the test is long running and not part of the deployment process, **users may tend to skip the test** if they believe the change they made is backward-compatible and causing issues when deployed to production.\n- From an open source perspective, users need to have an archival system setup, which we only support S3 and GCP for now. If those two don't work for the customer, they may have to **implement their own archiver**. On the client side, we can provide a blob storage interface, but customers still need to **implement the logic for fetching history and/or visibility records for blob storage**.\n- Finally since this approach takes blob storage as an external dependency, it **won’t be functioning when the dependencies are down**.\n\n### Replication based replay test\n\nThe final approach is: reusing the replication stack to dispatch decision tasks to workers at the standby cluster and perform the replay check. However this approach only works for global domain users, so it’s not a valid option.\n\n## Implementation\n\n### Worker Implementation\n\nIn the proposal of workflow shadowing, the worker does shadowing traffic in 3 steps.\n1. Get a list of workflow visibility records.\n2. For each record, get the workflow history.\n3. Run the replay test on the workflow history.\n\nThese steps can be implemented in native code or in Cadence workflow.\n\nFor the local development environment, we will provide a testing framework for shadowing workflow from production, so that users can validate their change sooner and speed up their development process. This kind of integration test can also be easily integrated with existing release pipelines to enforce a check upon landing code.\n\nFor staging/production environment, we incline to use a Cadence workflow to control the shadowing because of the following advantages:\n1. **Scalability**: support large shadowing traffic.\n2. **Extensibility**: easy to extend to different languages.\n3. **Maintenance**: no special worker code to support the shadowing mode.\n4. **Visibility**: shadowing progress and result can easily be viewed using the Cadence Web UI and metric dashboards.\n\nAs metrics will be emitted during the shadowing processing, it can also be integrated with release pipelines for checking workflow compatibility before deploying to production environments. The disadvantage is that users are required to start a Cadence server to run the shadow workflow.\n\n### Shadow Worker Options\n\nThe following options will be added to the Worker Option struct in both Golang and Java client:\n- **EnableShadowWorker**: a boolean flag indicates whether the worker should run in shadow mode.\n- **ShadowMode**:\n  - **Normal**: stop shadowing after a scan for all workflows, this will be the default selected value. A new iteration will be triggered upon worker restart or deployment.\n  - **Continuous**: keep shadowing until the exit condition is met. Since open workflows will keep making progress, two replays might lead to different results. This mode is useful for user’s whose workflow will block on a long timer.\n- **Domain**: user domain name.\n- **TaskList**: the name of the task list the shadowing activity worker should poll from.\n- **ShadowWorkflowQuery**: a visibility query for all the workflows that need to be replayed. If specified the following three options will be ignored.\n- **ShadowWorkflowTypes**: a list of workflow types that need to be checked.\n- **ShadowWorkflowStartTimeFilter**: a time range for the workflow start time.\n- **ShadowWorkflowStatus**: a list of workflow status. Workflow will be checked only if its close status is in the list. Option for open workflow will also be provided and will be the default.\n- **SamplingRate**: a float indicating the percentage of workflows (returned by workflow scan) that should be replayed.\n- **Concurrency**: concurrency of replay checks, which is the number of parallel replay activities in the shadow workflow.\n\n### Local Shadow Test Options\n\nThe options provided by the local test framework will be similar to the shadow worker options, but as in tests there’s only one single thread for scanning and replaying, only concurrency equal to 1 will be accepted. The option for EnableShadowWorker, Domain, and TaskList will also be removed.\n\n### Workflow Code Ownership\n\nWe decide to keep the workflow code at the server side due to the following reasons:\n1. The shadowing logic for Java and Go clients are identical. By keeping the shadowing workflow definition at the server side, we only need to implement it once, and don’t need to worry about keeping two workflow definitions in sync.\n2. Since Cadence server owns the workflow code, we can update the workflow definition at any time and even make non-backward compatible changes without having to ask customers to update their client dependency.\n\nThe downsides are:\n1. Cadence team needs to provide workers to process the decision task generated by the shadowing workflow.\n2. Shadowing workflow from all the customers will live in one domain, so they may affect each other’s execution as the total amount of resources allocated to a given domain is limited.\n\nHowever, considering the fact that the load generated by shadowing workflow is small and it’s unlikely many users will run the shadowing workflow at the same time, those downsides should not be a concern.\n\n### Metrics\n\nThe following metrics will be emitted in the shadow workflow:\n1. Number of succeeded/skipped/failed workflow replays\n2. Latency for each replay test\n3. Start/Complete/Continue_as_new of shadow workflow\n4. Latency for each shadow workflow\n"
  },
  {
    "path": "docs/flow.md",
    "content": "# Flow\n\n## Workflow processing flow\n\nBelow diagram demonstrates how a workflow is processed by Cadence workers and backend components.\nNote that this is a partial diagram that only highlights selected components and their interactions. There is more happening behind the scenes.\n\n[comment]: <> (To visualize mermaid flowchart below, install Mermaid plugin for your IDE. Works in github out of the box)\n\n```mermaid\nflowchart TD\n    User((Domain X Admin)) -->|\"0. Deploy worker with config: domain, tasklist(s)\"| Worker[[\\nDomain X Worker\\n\\n]]\n    ExternalInitiator((\"Domain X\\nAnother Service/CLI\")) -->|\"1. Start/Signal Workflow\"| Frontend\n    Worker -->|2. Poll tasks for domain: X, tasklist: tl1| Frontend[[\\nCadence Frontend\\n\\n]]\n    Frontend -->|3. Forward Poll request| Matching[[\\nCadence Matching\\n\\n]]\n    Matching -->|4. Poll| TaskListManager(Tasklist Manager for tl1)\n    TaskListManager -->|5. Long Poll| WaitUntilTaskFound\n    WaitUntilTaskFound --> |6. Return task to worker| Worker\n    Worker -->|7. Generate decisions\\n& respond| Frontend\n    Frontend -->|8. Respond decisions| History[[\\nCadence History\\n\\n]]\n    History -->|9. Mutable state update| ExecutionsStore[(Executions Store)]\n    History -->|10.a. Notify| HistoryQueue(History Queues - per shard)\n    HistoryQueue -->|Periodic task fetch| ExecutionsStore\n    HistoryQueue -->|11.b. Execute transfer task| TransferTaskExecutor(Transfer Task Executor)\n    HistoryQueue -->|11.a. Execute timer task| TimerTaskExecutor(Timer Task Executor)\n    HistoryQueue -->|Periodic per-shard offset save| ShardsStore[(Shards Store)]\n    TransferTaskExecutor -->|12.b. Add task| Matching\n    TimerTaskExecutor -->|12.a. Add task| Matching\n    Matching -->|13. Add task| TaskListManager\n    TaskListManager -->|14. Add task| SyncMatch{Check sync match}\n    SyncMatch -->|15.b. Found poller - Do sync match| SyncMatched\n    SyncMatch -->|15.a. No pollers - Save task| TasksStore[(Tasks Store)]\n    TaskListManager -->|Periodic task fetch| TasksStore\n\n```\n\n### Cassandra Notes:\n\nIn order to leverage [Cassandra Light Weight Transactions](https://www.yugabyte.com/blog/apache-cassandra-lightweight-transactions-secondary-indexes-tunable-consistency/), Cadence stores multiple types of records in the same `executions` table. ([ref](https://github.com/cadence-workflow/cadence/blob/51758676ce9d9609e736c64f94dc387ed2c75b7c/schema/cassandra/cadence/schema.cql#L338))\n- **Shards Store**: Shard records are stored in `executions` table with type=0.\n- **Executions Store**: Execution records are stored in `executions` table with type=1\n- **Tasks Store**: Task records are stored in `executions` table with type={2,3,4,5}\n"
  },
  {
    "path": "docs/howtos/async-api.md",
    "content": "# Async API\n\n## Overview\n\nTriggering too many workflows at the same time may overload the underlying storage system. It's typical the Cadence operator to set up quotas for these APIs via `frontend.rps` dynamic config. This means the caller service has to backoff and retry if the request is rejected by the Cadence server.\n\nWhen you have millions of workflows to trigger, you may want to consider using the Async APIs. More specifically,\n- `StartWorkflowExecutionAsync`\n- `SignalWithStartWorkflowExecutionAsync`\n\nThese APIs are designed to be more efficient than the regular APIs. They don't wait for the workflow to be started or signaled. Instead, they queue a message to underlying queue system and return. The queue system supported currently is Kafka. The Cadence server (workers service) will poll the queue and process the messages.\n\n## Caveats\n\n- Global availability: Regional failovers will work as is for Global Cadence Domains. The async workflow requests will start on the active side once failed over. Previously enqueued requests will be forwarded to active side by Cadence in an idempotent way. If the regions are disconnected or the previously active region is fully down then the leftover messages in the queue will be processed once the region is healthy again.\n- Idempotency: To avoid any edge cases and achieve full idempotency, Cadence dedupes requests based on request id. Request id is not exposed from our fat clients used internally so if you have additional retries on top of what client library already performs then you might send duplicate requests. See workflowid reuse policy to get around duplicate request problems.\n- Workflow id reuse policy: If you enqueue duplicate requests with same workflow id and choose a reuse policy that causes failure to start, the failure will be discarded. Since duplicate delivery is given with queue systems (at least once) avoid using WorkflowIDReusePolicyTerminateIfRunning .\n- Run id: Async APIs accept the same input parameters as their corresponding sync versions but do NOT return run id. Based on our discussions with multiple Cadence users, run id is discarded almost all the time so by switching to Async APIs you are getting pretty much the same semantics.\n- Request size and rate limits: Async APIs can support higher rate limits than the regular APIs. The default rate limit is 10k rps. You can adjust the rate limit via `frontend.asyncrps` dynamic config. Your kafka topic might be the bottleneck so you can adjust the topic configuration accordingly.\n- Delays: Async API requests are queued and consumed by Cadence backend. There can be some unexpected delays in this flow due to high number of messages/bytes etc. Basically your workflows don't start immediately and the delay depends on various factors.\n\n## How to use\n\nThis section walks through how to use the Async APIs on a local Cadence cluster.\n\n\n1. **Start a local Cadence cluster with async workflow queue enabled.**\n```\ndocker compose -f docker/docker-compose-async-wf-kafka.yml up\n```\n\nThis will start Cadence server, Cadence UI, Kafka, Cassandra, Prometheus and Grafana containers.\nNotice the environment variables in the docker compose file:\n- `ASYNC_WF_KAFKA_QUEUE_ENABLED`: This is set to `true` to enable the async workflow queue.\n- `ASYNC_WF_KAFKA_QUEUE_TOPIC`: This is the name of the Kafka topic to use for the async workflow queue.\n- `KAFKA_SEEDS`: This is the list of Kafka brokers to use for the async workflow queue.\n- `KAFKA_PORT`: This is the port to use for the Kafka brokers.\n\nWhen Cadence server container starts, it will materialize the ./docker/config_template.yaml file and populate `asyncWorkflowQueues` section with the following content:\n\n```\nasyncWorkflowQueues:\n  queue1:\n    type: \"kafka\"\n    config:\n      connection:\n        brokers:\n          - kafka:9092\n      topic: async-wf-topic1\n```\n\nNow the Cadence server recognizes `queue1` as a valid async workflow queue. However, it doesn't interact with Kafka yet.\n\nNote: It may take a minute for the containers to be ready. Check their status via `docker ps` command and once all are running, you can proceed to the next step.\n\n2. **Register a domain (or skip this step if you already have a domain)**\n\n```\ndocker run -t --rm --network host ubercadence/cli:master \\\n    --address localhost:7933 \\\n    --domain test-domain \\\n    domain register\n```\n\n3. **Enable async APIs for the domain**\n\nConfigure `test-domain` to use `queue1` as the async workflow queue.\n```\ndocker run -t --rm --network host ubercadence/cli:master \\\n    --address localhost:7933 \\\n    --domain test-domain \\\n    admin async-wf-queue update \\\n    --json \"{\\\"PredefinedQueueName\\\":\\\"queue1\\\", \\\"Enabled\\\": true}\"\n```\n\nNote: If you get \"Domain update too frequent.\" error, you can try to wait for a minute and run the command again.\n\nValidate the new configuration:\n```\ndocker run -t --rm --network host ubercadence/cli:master \\\n    --address localhost:7933 \\\n    --domain test-domain \\\n    admin async-wf-queue get\n```\n\nCadence server will start a consumer to poll from the kafka topic within a minute. If more than one domain is using the same queue, the consumer will be shared across all domains.\n\n\n4. **Start a workflow asynchronously**\n\nThe `test-domain` is now ready to accept async workflow requests. Update your worker (or modify samples code) to use one of the async APIs:\n- [StartWorkflowExecutionAsync](https://github.com/cadence-workflow/cadence-idl/blob/0e56e57909d9fa738eaa8d7a9561ea16acdf51e4/proto/uber/cadence/api/v1/service_workflow.proto#L49)\n- [SignalWithStartWorkflowExecutionAsync](https://github.com/cadence-workflow/cadence-idl/blob/0e56e57909d9fa738eaa8d7a9561ea16acdf51e4/proto/uber/cadence/api/v1/service_workflow.proto#L63)\n"
  },
  {
    "path": "docs/howtos/cassandra-fql.md",
    "content": "# View Cassandra full query logs\n\n1. Run cadence components via docker compose\n```\ndocker compose docker/docker-compose.yml up\n```\n\n2. First enable fql via nodetool ([ref](https://cassandra.apache.org/doc/stable/cassandra/operating/fqllogging.html#enabling-fql))\n```\ndocker exec $(docker inspect --format=\"{{.Id}}\" docker-cassandra-1) nodetool enablefullquerylog --path /tmp/cassandra_fql\n```\n\n3. Check `/tmp/cassandra_fql` folder exists\n```\ndocker exec $(docker inspect --format=\"{{.Id}}\" docker-cassandra-1) ls /tmp/cassandra_fql\n```\n\n4. Run some workflows to generate queries (optional)\n\n5. Inspect the full query log dump via fqltool (it's under tools/bin in default apache cassandra installation)\n```\ndocker cp $(docker inspect --format=\"{{.Id}}\" docker-cassandra-1):/tmp/cassandra_fql/ /tmp/cassandra_fql/ && fqltool dump /tmp/cassandra_fql | less\n```\n"
  },
  {
    "path": "docs/howtos/setup-cadence-locally-with-replication.md",
    "content": "### How to run Cadence locally with replication\n\n\nThis is a guide for using a local laptop instance and running three Cadence\nclusters, all cross-replicted. They will (for the purposes of demonstration) be\nentirely independentent, but poll one another for workflows for replication and\nwrite to their own keyspaces independently.\n\nIt's a good way to work through how Cadence's domains are configured with\nreplication, how failovers work and so forth.\n\nThis is a somewhat more advanced tutorial, and so it's recommended to ensure\nthat you have a basic familiarity with Cadence and it's concepts first before\ntrying this.\n\n#### Prerequisites\n\n- Docker\n- The ability to perform a local go build\n\n#### Getting started\n\nAll commands are expected to be run from the repo's base path.\n\n1. Ensure there's no existing containers running or preexisting state. You'll be\n   creating a new keyspace and a new docker-container *only* for Cassandra, and\n   so this can conflict in a few ways with other cassandra instances for\n   fully-containerized cadence. Be sure to stop and/or remove them before\n   running this.\n\n2. Ensure you build the `cadence-server` binaries by using `make bins`.\n   This will build the main application and produce a binary build of the local\n   instance of cadence in the repo's base path.\n\n3. Start docker with the XDC configuration: `docker compose  -f docker/dev/cassandra.yml up -d`\n\n4. setup the schema for each of the instances with `until make\n   install-schema-xdc; do sleep 1; done` (since Cassandra can take a while to\n   setup, loop this check until it completes).\n\n5. start (in tmux or terminal windows) each of the Cadence instances\n  - `./cadence-server --config config --zone xdc --env development_xdc_cluster0 start`\n  - `./cadence-server --config config --zone xdc --env development_xdc_cluster1 start`\n  - `./cadence-server --config config --zone xdc --env development_xdc_cluster2 start`\n\n6. Register a global domain that has multiple clusters enabled using the cadence\n   CLI, so that it's worklows will be replicated across clusters: \n   `./cadence --domain test-domain domain register --gd true --clusters cluster0,cluster1`\n\n7. Start and test workflows as needed and failover the domain to make another\n   cluster active with a command such as this: `./cadence --domain test-domain domain update --ac cluster1`\n\n"
  },
  {
    "path": "docs/migration/cassandra-shard-info.md",
    "content": "# What?\n\nThe [`shard`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/schema/cassandra/cadence/schema.cql#L1) Cassandra data type and [`shard`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/schema/cassandra/cadence/schema.cql#L368) column in Cassandra [`executions`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/schema/cassandra/cadence/schema.cql#L357) table is deprecated in favor of [`data`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/schema/cassandra/cadence/schema.cql#L366C3-L366C7) and [`data_encoding`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/schema/cassandra/cadence/schema.cql#L367) field.\n\n# Why?\n\nThe process to introduce new field to existing data type is quite tedious because we need to update the corresponding Cassandra data type.\n\n# How?\n1. Update server to v1.3.1 or a newer version\n2. Set this dynamic configuration value to true\n    - [`history.readNoSQLShardFromDataBlob`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/common/dynamicconfig/dynamicproperties/constants.go#L4578C18-L4578C52)\n3. Check the metrics with these tags:\n    - operation: getshard\n    - name: nosql_shard_store_read_from_original_column OR nosql_shard_store_read_from_data_blob\n\n    These metrics are only emitted during service start time, and after migration you should only see metrics with `name:nosql_shard_store_read_from_data_blob` tag.\n\n# Status\nStarting from 1.3.4, the default value of [`history.readNoSQLShardFromDataBlob`](https://github.com/cadence-workflow/cadence/blob/v1.3.3/common/dynamicconfig/dynamicproperties/constants.go#L4578C18-L4578C52) is set to `true`. And we're planning to remove this dynamic config in later version.\n"
  },
  {
    "path": "docs/migration/tasklist-partition-config.md",
    "content": "# What?\n\nThis document writes the steps we need to execute to migrate the number of task list partitions configuration from dynamic configuration to the database. For background knowledge about task list partition, please read this [doc](../scalable_tasklist.md).\n\n# Why?\n\nWe're doing this migration because we want to programmatically update the number of task list partitions. Not all implementations of dynamic configuration dependencies support update operation.\n\n# How?\n1. Check the existing number of partitions for the task list you want to migrate:\n   - [matching.numTasklistReadPartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3350)\n   - [matching.numTasklistWritePartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3344)\n\n2. Run the following ClI commands to update the number of partitions of the task list you want to migrate and make sure that task list type parameter is not missing in your commands:\n```\ncadence admin tasklist update-partition -h\n```\nTo get the number of partitions from database, use the following CLI command:\n```\ncadence admin tasklist describe -h\n```\n3. Set this dynamic configuration value to true for the task list you want to migrate:\n  - [matching.enableGetNumberOfPartitionsFromCache](https://github.com/cadence-workflow/cadence/blob/v1.2.15-prerelease02/common/dynamicconfig/constants.go#L4008)\n\n4. Repeat the steps for all task lists. However, you can skip the steps if the number of partitions of the task list is 1.\n\n5. You can enable adaptive task list scaler for the task list. Set [matching.enableAdaptiveScaler](https://github.com/cadence-workflow/cadence/blob/v1.2.17/common/dynamicconfig/constants.go#L4012) to true for the task list.\n\n# Status\nAs of v1.2.17, the default value of [matching.enableGetNumberOfPartitionsFromCache](https://github.com/cadence-workflow/cadence/blob/v1.2.17/common/dynamicconfig/constants.go#L4004) is still false.\n\n# Plan\nWe're planning to change the default value of [matching.enableGetNumberOfPartitionsFromCache](https://github.com/cadence-workflow/cadence/blob/v1.2.17/common/dynamicconfig/constants.go#L4004) to `true` in v2.\nWe're planning to deprecate [matching.numTasklistReadPartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3350) and [matching.numTasklistWritePartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3344), but we haven't decided when to do it. Please be prepared for the migration.\n"
  },
  {
    "path": "docs/non-deterministic-error.md",
    "content": "\r\n# Abstract\r\n\r\nThis article is for Cadence developers to understand and address issues with non-deterministic errors. It gives an introduction to the nature and categories of the non-deterministic error, and provide different ways for the fix. It uses Golang as example.\r\n\r\n# What is non-deterministic error\r\n\r\n## Some Internals of Cadence workflow\r\n\r\nA Cadence workflow can be viewed as a long running process on a distributed operating system(OS). The process’s state and dispatching is owned by Cadence server, and customers’ workers provide CPU/memory resources to execute the process’s code. For most of the time, this process(workflow) is owned by a worker and running like in other normal OS. But because this is a distributed OS, the workflow ownership can be transferred to other workers and continue to run from the previous state. Unlike other OS, this is not restarting from the beginning. This is how workflow is fault tolerant to certain host failures.\r\n\r\n![image alt text](images/non-deterministic-err.1.png)\r\n\r\nNon-deterministic issues arise during this workflow ownership transfer. Cadence is designed with event sourcing, meaning that it persists each workflow state mutation as \"**history event**\", instead of the whole workflow state machine. After switching to another worker, the workflow state machine has to be rebuilt. The process of rebuilding the state machine is called “**history replay**”. Any failure during this history replay is a “**non-deterministic error**”.\r\n\r\n![image alt text](images/non-deterministic-err.2.png)\r\n\r\nEven if a workflow ownership doesn’t change, history replay is also required under some circumstances:\r\n\r\n* A workflow stack(of an execution) lives in worker’s memory, a worker can own many workflow executions. So a worker can run out of memory, it has to kick off some workflow executions (with LRU) and rebuild them when necessary.\r\n\r\n* Sometimes the stack can be stale because of some errors.\r\n\r\nIn Cadence, Workflow ownership is called \"stickiness\".\r\n\r\nWorker memory cache for workflow stacks is called \"sticky cache\".\r\n\r\n## Cadence History Protocol\r\n\r\nHistory replay process is under some basic rules/concepts.\r\n\r\nIn Cadence, we use \"close\" to describe the opposite status of “open”. For activity/decision, close includes complete,timeout,fail. For workflow, close means complete/timeout/fail/terminate/cancel/continueAsNew.\r\n\r\n### Decision\r\n\r\n* Decision is the key of history replay. It drives the progress of a workflow.\r\n\r\n* The decision state machine must be **Scheduled->Started->Closed**\r\n\r\n* The first decision task is triggered by server. Starting from that, the rest of the decision tasks are triggered by some events of the workflow itself -- when those events mean something the workflow could be waiting for. For example, Signaled, ActivityClosed, ChildWorkflowClosed, TimerFired events. Events like ActivityStarted won’t trigger decisions.\r\n\r\n* When a decision is started(internally called in flight), there cannot be any other events written into history before a decision is closed. Those events will be put into a buffer until the decision is closed -- flush buffer will write the events into history.\r\n\r\n* **In executing mode**, a decision task will try to complete with some entities: Activities/Timers/Childworkflows/etc Scheduled.\r\n\r\n* **In history replay mode**, decisions use all of those above to rebuild a stack. **Activities/Timers/ChildWorkflows/etc will not be re-executed during history replay.**\r\n\r\n### Activity\r\n\r\n* State machine is **Scheduled->Started->Closed**\r\n\r\n* Activity is scheduled by DecisionCompleted\r\n\r\n* Activity started by worker. Normally, ActivityStartedEvent can be put at any place in history except for in between DecisionStarted and DecisionClose.\r\n\r\n* But Activity with RetryPolicy is a special case. Cadence will only write down Started event when Activity is finally closed.\r\n\r\n* Activity completed/failed by worker, or timed out by server -- they all consider activity closed, and it will trigger a decision task if no decision task is ongoing.\r\n\r\n* Like in the above, only ActivityClose events could trigger a decision.\r\n\r\n### Local activity\r\n\r\n* Local activity is executed within decision is processing in flight.\r\n\r\n* Local activity is only recorded with DecisionCompleted -- no state machine needed.\r\n\r\n* Local activity completed will trigger another decision\r\n\r\n### Timer\r\n\r\n* State machine is **Scheduled->Fired/Canceled**\r\n\r\n* Timer is the implementation of \"workflow.Sleep()\" and “workflow.NewTimer().\r\n\r\n* Timer is scheduled by DecisionCompleted\r\n\r\n* Timer fired by server. It would trigger a decision.\r\n\r\n* Timer canceled by worker(when workflow is canceled)\r\n\r\n### ChildWorkflow\r\n\r\n* State machine is **Initiated->Started->Closed**\r\n\r\n* ChildWorkflow is initiated by DecisionCompleted\r\n\r\n* ChildWorkflow is started by server(returning runID). It could trigger a decision.\r\n\r\n* ChildWorkflow close events can be\"canceled/failed/completed\". It could trigger a decision.\r\n\r\n### SignalExternal/RequestCancel\r\n\r\n* State machine is **Initiated->Closed**\r\n\r\n* They are both initiated by DecisionCompleted.\r\n\r\n* Closed(completed/failed) by server, it could trigger a decision.\r\n\r\n### More explanation of BufferedEvents\r\n\r\nWhen a decision is in flight, if something like a signal comes in, Cadence has to put it into buffer. That’s because for the next decision, SDK always processes unhandled events starting from last decision completed.  There cannot be any other events to record between decision started and close event. [This may cause some issues](https://github.com/cadence-workflow/cadence/issues/2934) if you are sending a signal to self within a local activities.\r\n\r\n## An Example of history protocol\r\n\r\nTo understand the protocol, consider an example of the following workflow code:\r\n```go\r\nfunc Workflow(ctx workflow.Context) error {\r\n\tao := workflow.ActivityOptions{\r\n\t\t...\r\n\t}\r\n\tctx = workflow.WithActivityOptions(ctx, ao)\r\n\tvar a int\r\n\terr := workflow.ExecuteActivity(ctx, activityA).Get(ctx, &a)\r\n\tif err != nil {\r\n\t\treturn err\r\n\t}\r\n\tworkflow.Sleep(time.Minute)\r\n\terr = workflow.ExecuteActivity(ctx, activityB, a).Get(ctx, nil)\r\n\tif err != nil {\r\n\t\treturn err\r\n\t}\r\n\tworkflow.Sleep(time.Hour)\r\n\treturn nil\r\n}\r\n```\r\n\r\nThe workflow will execute activityA, then wait for 1 minute, then execute activityB, finally waits for 1 hour to complete.\r\n\r\nThe history will be as follows if everything runs smoothly (no errors, timeouts, retries, etc):\r\n\r\n```\r\nID:1\t\tWorkflow Started\r\nID:2\t\tDecisionTaskScheduled : first decision triggered by server\r\nID:3\t\tDecisionTaskStarted\r\nID:4\t\tDecisionTaskCompleted\r\nID:5\t\tActivityTaskScheduled : activityA is scheduled by decision\r\nID:6\t\tActivityTaskStarted   : started by worker\r\nID:7\t\tActivityTaskCompleted : completed with result of var a\r\nID:8\t\tDecisionTaskScheduled : triggered by ActivityCompleted\r\nID:9\t\tDecisionTaskStarted\r\nID:10\t\tDecisionTaskCompleted\r\nID:11\t\tTimerStarted : decision scheduled a timer for 1 minute\r\nID:12\t\tTimerFired : fired after 1 minute\r\nID:13\t\tDecisionTaskScheduled : triggered by TimerFired\r\nID:14\t\tDecisionTaskStarted\r\nID:15\t\tDecisionTaskCompleted\r\nID:16\t\tActivityTaskScheduled: activityB scheduled by decision with param a\r\nID:17\t\tActivityTaskStarted : started by worker\r\nID:18\t\tActivityTaskCompleted : completed with nil as error\r\nID:19\t\tDecisionTaskScheduled : triggered by ActivityCompleted\r\nID:20\t\tDecisionTaskStarted\r\nID:21\t\tDecisionTaskCompleted\r\nID:22\t\tTimerStarted : decision scheduled a timer for 1 hour\r\nID:23\t\tTimerFired : fired after 1 hour\r\nID:24\t\tDecisionTaskScheduled : triggered by TimerFired\r\nID:25\t\tDecisionTaskStarted\r\nID:26\t\tDecisionTaskCompleted\r\nID:27\t\tWorkflowCompleted     : completed by decision\r\n```\r\n\r\n## Categories of non-deterministic errors\r\n\r\n### Missing decision\r\n\r\n[Error message](https://github.com/cadence-workflow/cadence-go-client/blob/e5081b085b0333bac23f198e57959681e0aee987/internal/internal_task_handlers.go#L1206):\r\n\r\n```go\r\nfmt.Errorf(\"nondeterministic workflow: missing replay decision for %s\", util.HistoryEventToString(e))\r\n```\r\n\r\nThis means after replay code, the decision is scheduled less than history events. Using the previous history as an example, when the workflow is waiting at the one hour timer(event ID 22), if we delete the line of :\r\n```go\r\n workflow.Sleep(time.Hour)\r\n```\r\nand restart worker, then it will run into this error. Because in the history, the workflow has a timer event that is supposed to fire in one hour. However, during replay, there is no logic to schedule that timer.\r\n\r\n### Extra decision\r\n\r\n[Error message](https://github.com/cadence-workflow/cadence-go-client/blob/e5081b085b0333bac23f198e57959681e0aee987/internal/internal_task_handlers.go#L1210):\r\n\r\n```go\r\nfmt.Errorf(\"nondeterministic workflow: extra replay decision for %s\", util.DecisionToString(d))\r\n```\r\n\r\nThis is basically the opposite of the previous case, which means that during replay, Cadence generates more decisions than those in history events. Using the previous history as an example, when the workflow is waiting at the one hour timer(event ID 22), if we change the line of :\r\n```go\r\nerr = workflow.ExecuteActivity(ctx, activityB, a).Get(ctx, nil)\r\n```\r\nto\r\n\r\n```go\r\nfb := workflow.ExecuteActivity(ctx, activityB, a)\r\nfc := workflow.ExecuteActivity(ctx, activityC, a)\r\nerr = fb.Get(ctx,nil)\r\nif err != nil {\r\n\treturn err\r\n}\r\nerr = fc.Get(ctx,nil)\r\nif err != nil {\r\n\treturn err\r\n}\r\n```\r\n\r\nAnd restart worker, then it will run into this error. Because in the history, the workflow has scheduled only activityB after the one minute timer, however, during replay, there are two activities scheduled in a decision( in parallel).\r\n\r\n### Decision mismatch\r\n\r\n[Error message](https://github.com/cadence-workflow/cadence-go-client/blob/e5081b085b0333bac23f198e57959681e0aee987/internal/internal_task_handlers.go#L1214):\r\n\r\n```go\r\nfmt.Errorf(\"nondeterministic workflow: history event is %s, replay decision is %s\",util.HistoryEventToString(e), util.DecisionToString(d))\r\n```\r\n\r\nThis means after replay code, the decision scheduled is different than the one in history. Using the previous history as an example, when the workflow is waiting at the one hour timer(event ID 22),\r\n\r\nif we change the line of :\r\n```go\r\nerr = workflow.ExecuteActivity(ctx, activityB, a).Get(ctx, nil)\r\n```\r\nto\r\n```go\r\nerr = workflow.ExecuteActivity(ctx, activityC, a).Get(ctx, nil)\r\n```\r\nAnd restart worker, then it will run into this error. Because in the history, the workflow has scheduled activityB with input a, but during replay, it schedules activityC.\r\n\r\n### Decision State Machine Panic\r\n\r\n[Error message](https://github.com/cadence-workflow/cadence-go-client/blob/e5081b085b0333bac23f198e57959681e0aee987/internal/internal_decision_state_machine.go#L693):\r\n\r\n```go\r\nfmt.Sprintf(\"unknown decision %v, possible causes are nondeterministic workflow definition code\"+\" or incompatible change in the workflow definition\", id)\r\n```\r\n\r\nThis usually means workflow history is corrupted due to some bug. For example, the same activity can be scheduled and differentiated by activityID. So ActivityIDs for different activities are supposed to be unique in workflow history. If however we have an ActivityID collision, replay will run into this error.\r\n\r\n## What can cause non-deterministic errors\r\n\r\n* Changing the order of executing activities/timer/childWorkflows/signalExternal/CancelRequest\r\n\r\n* Changing signature of activities\r\n\r\n* Changing duration of timer\r\n\r\n* Using time.Now() instead of workflow.Now()\r\n\r\n* Use golang builtin \"go\" to start goroutine in workflow, instead of “workflow.Go”\r\n\r\n* Use golang builtin channel instead of \"workflow.Channel\" for inter goroutines communication\r\n\r\n* time.Sleep() will not work, even though it doesn’t cause non-deterministic errors, but changing to workflow.Sleep() will.\r\n\r\nFor those needs, see \"How to address non-deterministic issues\" in the next section.\r\n\r\n## What will NOT cause non-deterministic errors\r\n\r\n* Activity input/output is a struct, changing the details of the struct\r\n\r\n* Adding/changing some code that will not affect history, for example, code about logs/metrics\r\n\r\n* Change retry policy of activity/local activity\r\n\r\n* <TODO: more to come...>\r\n\r\n## Find the Non-Deterministic Code\r\n\r\nWorkflow logic can be complicated and changes to workflow code can be non-trivial and non-isolated. In case you are not able to pinpoint the exact code change that introduces the workflow logic and non-deterministic error, you can download the workflow history with non-deterministic error and [replay it locally](https://github.com/cadence-workflow/cadence-go-client/blob/master/worker/worker.go#L96). This is [an example](https://github.com/cadence-workflow/cadence-samples/blob/03293b934579e0353e08e75c2f46a84a5a7b2df0/cmd/samples/recipes/helloworld/replay_test.go#L39) of using this utility.\r\n\r\nYou can do this with different versions of your workflow code to see the difference in behavior.\r\n\r\nHowever, it could be hard if you have too many versions or your code is too complicated to debug. In this case, you can run replay in debug mode to help you to step into your workflow logic.\r\n\r\nTo do this you first change [this code](https://github.com/cadence-workflow/cadence-go-client/blob/cc25a04f6f74c54ea9ae330741f63ae6df15f4df/internal/internal_event_handlers.go#L429) into\r\n\r\n```go\r\nfunc (wc *workflowEnvironmentImpl) GenerateSequence() int32 {\r\n    result := wc.counterID\r\n    wc.counterID++\r\n    if wc.counterID == THE_ID_WITH_ERROR {\r\n        fmt.Println(\"PAUSE HERE IN DEBUG MODE\")\r\n    }\r\n    return result\r\n}\r\n```\r\n\r\nTHE_ID_WITH_ERROR is the ActivityID/TimerID of activities/timers of decision runs into non-deterministic error during your replay with current code. When you pause the replay thread in the fmt.Println(\"PAUSE HERE IN DEBUG MODE\") , trace back in the stack you will see the position of your workflow code that run into non-deterministic error.\r\n\r\nCurrently this debugging experience is not very ideal. [This proposal will help](https://github.com/cadence-workflow/cadence/issues/2801).\r\n\r\n## Automatic history replay and non-deterministic error detection\r\n\r\nWe have ideas on how non-deterministic errors can be detected automatically and safe rollout can be achieved. See [this issue](https://github.com/cadence-workflow/cadence/issues/2547).\r\n\r\n# What to do with non-deterministic errors\r\n\r\n## Related worker configs\r\n\r\n### NonDeterministicWorkflowPolicy\r\n\r\nThis is a worker option to decide what to do with non-deterministic workflows. Default is **NonDeterministicWorkflowPolicyBlockWorkflow**\r\n\r\nAnother option is **NonDeterministicWorkflowPolicyFailWorkflow** which will fail the workflow immediately. You may want that if it is Okay to fail the trouble workflows for now(you can reopen later with reset) so that workers won’t keep on retrying the workflows.\r\n\r\n### DisableStickyExecution\r\n\r\nIt defaults to false which means all workflows will stay in stickyCache unless there is memory pressure which causes them to be evicted. This is the desired behavior in production as it saves replay efforts. However it could hide potential non-deterministic errors exactly because of this reason.\r\n\r\nWhen troubleshooting it might be helpful to let a small number of workers run with stickiness disabled, so that it always replays the whole history when execution decision tasks.\r\n\r\n## GetVersion() & SideEffect()\r\n\r\nIf you know some code change will cause non-deterministic errors, then use\r\n\r\n[workflow.GetVersion()](https://cadenceworkflow.io/docs/07_goclient/14_workflow_versioning) API to prevent it. This API will let the workflow that has finished the changing code to go with old code path, but let the workflows that hasn’t to go with new code path.\r\n\r\nFor example of \"missing decision\", instead of simply deleting the code, we could change it to:\r\n```go\r\nv := workflow.GetVersion(ctx, \"delete_timer\", workflow.DefaultVersion, 1)\r\nif v == workflow.DefaultVersion{\r\n   // run with old code\r\n  workflow.Sleep(time.Hour)\r\n}else{\r\n  // run with new code\r\n  // do nothing as we are deleting the timer\r\n}\r\n```\r\n\r\n\r\nAnother similar API to prevent non-deterministic error is [workflow.SideEffect()](https://cadenceworkflow.io/docs/07_goclient/10_side_effect).\r\n\r\n## BinaryChecksum & workflow.Now()\r\n\r\nWhat if Non-deterministic code change has been deployed without GetVersions()?\r\n\r\nSometimes we may forget to use GetVersions(), or misuse it. This could be a serious problem because after deployment, we probably cannot rollback: because some workflows has run with new code but some workflows has stuck. Rollback will save the stuck workflows but also stuck other workflows.\r\n\r\nThe best way is to use BinaryChecksum and Now() to let workflow diverge at the breaking changes. **workflow.GetInfo().[BinaryChecksum](https://github.com/cadence-workflow/cadence-go-client/issues/925)** is the checksum of the binary that made that decision. **workflow.[Now](https://github.com/cadence-workflow/cadence-go-client/issues/926)()** is timestamp that the decision is made. For better experience, you should integrate with binaryChecksum is in a format of \"**Your GIT_REF**\" by **worker.SetBinaryChecksum()** API.\r\n\r\nUse the \"extra decision\" as an example. After deploy the code change, then there are workflow W1 stuck because of extra decision, however workflow W2 has started the two in-parallel activities. If we rollback the code, W1 will be fixed but W2 will be stuck. We can fix it by changing the code to:\r\n\r\n```go\r\nif *(workflow.GetInfo(ctx).BinaryChecksum)== \"BINARY_BEFORE_BAD_DEPLOYMENT\" || workflow.Now(ctx) < DeploymentStartTime {\r\n    // run with old code\r\n    err = workflow.ExecuteActivity(ctx, activityB, a).Get(ctx, nil)\r\n    if err != nil {\r\n      return err\r\n    }\r\n}else{\r\n    // run with new code\r\n  fb := workflow.ExecuteActivity(ctx, activityB, a)\r\n  fc := workflow.ExecuteActivity(ctx, activityC, a)\r\n  err = fb.Get(ctx,nil)\r\n  if err != nil {\r\n    return err\r\n  }\r\n  err = fc.Get(ctx,nil)\r\n  if err != nil {\r\n    return err\r\n  }\r\n}\r\n```\r\n\r\n\r\nBINARY_BEFORE_BAD_DEPLOYMENT is the previous binary checksum, DeploymentStartTime that deployment start time. This will tell the workflows that has finished the decision with previous binary, or the decision is old enough that is not finished with new code, then it should go with the old code. We need DeploymentStartTime because W1 could be started by different binary checksums.\r\n\r\n## [Reset workflow](https://cadenceworkflow.io/docs/08_cli#restart-reset-workflow)\r\n\r\nThe last solution is to reset the workflows. A process in real OS can only move forward but never go back to previous state. However, a Cadence workflow can go back to previous state since we have stored the history as a list.\r\n\r\nInternally reset a workflow will use history as a tree. It takes a history as base, and fork a new branch from it. So that you will reset many times without losing the history(until history is deleted after retention).\r\n\r\nReset will start a new run with new runID like continueAsNew.\r\n\r\nAfter forking from the base, reset will also collect all the signals along the chain of continueAsNew from the base history. This is an important feature as we can consider signals are external events that we don’t want to lose.\r\n\r\nHowever, reset will schedule and execute some activities that has done before. So after reset, you may see some activities are re-executed. Same applies for timer/ChildWorkflows. If you don’t want to re-execute, you can emit a signal to self to identify that this activity/timer/childWorkflow is done -- since reset will collect signals after resetting point.\r\n\r\nNote  that reset with [child workflows](https://github.com/cadence-workflow/cadence/issues/2951) is not fully supported yet.\r\n\r\n## Primitive Reset CLI Command\r\n\r\nThis reset command is for resetting one workflow. We may use this command to manually resetting particular workflow for experiment or mitigation.\r\n\r\nIt takes workflowID/runID for base history.\r\n\r\nIt takes either eventID or eventType for forking point.\r\n\r\nEventID has to be decisionCloseEventID as we designed reset must be done by decision boundary.\r\n\r\nResetType support these: LastDecisionCompleted, LastContinuedAsNew, BadBinary ,FirstDecisionCompleted.\r\n\r\nIf the workflowID has an open run, you need to be aware of [this race condition](https://github.com/cadence-workflow/cadence/issues/2930) when resetting it.\r\n\r\n## Batch Reset CLI Command\r\n\r\nReset-batch is for resetting a list of workflows. Usually reset-batch is more useful/powerful. Reset-batch command has many arguments, most of them are of two types: to decide what workflows to reset and where to reset to:\r\n\r\n1. What workflows to reset\r\n\r\n* Input file or query: only one of them is needed. Query will be same as List command(advanced functionality based on ElasticSearch).\r\n\r\n* Exclude file- for filtering out some workflows that we don’t want to reset\r\n\r\n* SkipIfCurrentOpen\r\n\r\n* SkipIfBaseNotCurrent\r\n\r\n* NonDeterministicOnly\r\n\r\n2. Where to reset\r\n\r\n* ResetType\r\n\r\n* ResetBadBinaryChecksum\r\n\r\nOther arguments:\r\n\r\nParallism will decide how fast you want to reset.\r\n\r\nTo be safe, you may use DryRun option for only printing some logs before actually executing it.\r\n\r\nFor example, in the case of \"Decision State Machine Panic\", we might have to reset the workflows by command:\r\n\r\n*$nohup cadence --do samples-domain --env prod wf reset-batch --reason \"fix outage\" --query “WorkflowType=’SampleWorkflow’ AND CloseTime=missing”  --dry-run --reset-type <A-RESET-TYPE> --non_deterministic_only --skip_base_not_current &> reset.log &*\r\n\r\nFor reset type, you may try LastDecisionCompleted. Then try FirstDecisionCompleted. We should also provide [firstPanicDecision](https://github.com/cadence-workflow/cadence/issues/2952) resetType .\r\n\r\n## AutoReset Workflow\r\n\r\nCadence also provides a command to reset all progress made by any binary given a binaryChecksum.\r\n\r\nThe way it works is to store the first decision completed ID as an **auto-reset point** for any binaryChecksum. Then when a customer mark a binary checksum is bad, the  badBinaryChecksum will be stored in domainConfig. Whenever an open workflow make any progress, it will reset the workflow to the auto-reset point.\r\n\r\nThere are some limitations:\r\n\r\n1. There are only a limited number(20 by default) of auto-reset points for each workflow. Beyond that the auto-reset points will be rotated.\r\n\r\n2. It only applies to open workflows.\r\n\r\n3. It only reset when the open workflow make a decision respond\r\n\r\n4. [It could be much improved by this proposal, ](https://github.com/cadence-workflow/cadence/issues/2810)\r\n\r\nHowever, you can use reset batch command to achieve the same to both open/closed workflows and without waiting for making decision respond.\r\n"
  },
  {
    "path": "docs/persistence.md",
    "content": "# Overview\nCadence has a well defined API interface at the persistence layer. Any database that supports multi-row transactions on\na single shard or partition can be made to work with cadence. This includes cassandra, dynamoDB, auroraDB, MySQL,\nPostgres and may others. There are currently three supported database implementations at the persistence layer -\ncassandra and MySQL/Postgres. This doc shows how to run cadence with cassandra and MySQL(Postgres is mostly the same). It also describes the steps involved\nin adding support for a new database at the persistence layer.\n\n# Getting started on mac\n## Cassandra\n### Start cassandra\n```\nbrew install cassandra\nbrew services start cassandra\n```\n### Install cadence schema\n```\ncd $GOPATH/github.com/uber/cadence\nmake install-schema\n```\n> NOTE: See [CONTRIBUTING](/CONTRIBUTING.md) for prerequisite of make command.\n>\n### Start cadence server\n```\ncd $GOPATH/github.com/uber/cadence\n./cadence-server start --services=frontend,matching,history,worker\n```\n\n## MySQL\n### Start MySQL server\n```\nbrew install mysql\nbrew services start mysql\n```\n### Install cadence schema\n```\ncd $GOPATH/github.com/uber/cadence\nmake install-schema-mysql\n```\nWhen run tests and CLI command locally, Cadence by default uses a user `uber` with password `uber`, with privileges of creating databases.\nYou can use the following command to create user(role) and grant access.\nIn the mysql shell:\n```\n> CREATE USER 'uber'@'%' IDENTIFIED BY 'uber';\n> GRANT ALL PRIVILEGES ON *.* TO 'uber'@'%';\n```\n### Start cadence server\n```\ncd $GOPATH/github.com/uber/cadence\ncp config/development_mysql.yaml config/development.yaml\n./cadence-server start --services=frontend,matching,history,worker\n```\n\n## PostgresQL\n### Start PostgresQL server\n```\nbrew install postgres\nbrew services start postgres\n```\nWhen run tests and CLI command locally, Cadence by default uses a superuser `postgres` with password `cadence`.\nYou can use the following command to create user(role) and grant access:\n```\n$psql postgres\npostgres=# CREATE USER postgres WITH PASSWORD 'cadence';\nCREATE ROLE\npostgres=# ALTER USER postgres WITH SUPERUSER;\nALTER ROLE\n```\n### Install cadence schema\n```\ncd $GOPATH/github.com/uber/cadence\nmake install-schema-postgres\n```\n\n### Start cadence server\n```\ncd $GOPATH/github.com/uber/cadence\ncp config/development_postgres.yaml config/development.yaml\n./cadence-server start --services=frontend,matching,history,worker\n```\n\n# Configuration\n## Common to all persistence implementations\nThere are two major sub-subsystems within cadence that need persistence - cadence-core and visibility. cadence-core is\nthe workflow engine that uses persistence to store state tied to domains, workflows, workflow histories, task lists\netc. cadence-core powers almost all of the cadence APIs. cadence-core could be further broken down into multiple\nsubs-systems that have slightly different persistence workload characteristics. But for the purpose of simplicity, we\ndon't expose these sub-systems today but this may change in future. Visibility is the sub-system that powers workflow\nsearch. This includes APIs such as ListOpenWorkflows and ListClosedWorkflows. Today, it is possible to run a cadence\nserver with cadence-core backed by one database and cadence-visibility backed by another kind of database.To get the full\nfeature set of visibility, the recommendation is to use elastic search as the persistence layer. However, it is also possible\nto run visibility with limited feature set against Cassandra or MySQL today.  The top level persistence configuration looks\nlike the following:\n\n\n```\npersistence:\n  defaultStore: datastore1    -- Name of the datastore that powers cadence-core\n  visibilityStore: datastore2 -- Name of the datastore that powers cadence-visibility\n  numHistoryShards: 1024      -- Number of cadence history shards, this limits the scalability of single cadence cluster\n  datastores:                 -- Map of datastore-name -> datastore connection params\n    datastore1:\n      nosql:\n         ...\n    datastore2:\n      sql:\n        ...\n```\n\n## Note on numHistoryShards\nInternally, cadence uses shards to distribute workflow ownership across different hosts. Shards are necessary for the\nhorizontal scalability of cadence service. The number of shards for a cadence cluster is picked at cluster provisioning\ntime and cannot be changed after that. One way to think about shards is the following - if you have a cluster with N\nshards, then cadence cluster can be of size 1 to N. But beyond N, you won't be able to add more hosts to scale. In future,\nwe may add support to dynamically split shards but this is not supported as of today. Greater the number of shards,\ngreater the concurrency and horizontal scalability.\n\n## Cassandra\n```\npersistence:\n  ...\n  datastores:\n    datastore1:\n      nosql:\n        pluginName: \"cassandra\"\n        hosts: \"127.0.0.1,127.0.0.2\"  -- CSV of cassandra hosts to connect to\n        User: \"user-name\"\n        Password: \"password\"\n        keyspace: \"cadence\"           -- Name of the cassandra keyspace\n        datacenter: \"us-east-1a\"      -- Cassandra datacenter filter to limit queries to a single dc (optional)\n        maxConns: 2                   -- Number of tcp conns to cassandra server (single sub-system on one host) (optional)\n```\n\n## MySQL/Postgres\nThe default isolation level for MySQL/Postgres is READ-COMMITTED.\n\nNote that for MySQL 5.6 and below only, the isolation level needs to be\nspecified explicitly in the config via connectAttributes.\n\n```\npersistence:\n  ...\n  datastores:\n    datastore1:\n      sql:\n        pluginName: \"mysql\"            -- name of the go sql plugin\n        databaseName: \"cadence\"        -- name of the database to connect to\n        connectAddr: \"127.0.0.1:3306\"  -- connection address, could be ip address or domain socket\n        connectProtocol: \"tcp\"         -- connection protocol, tcp or anything that SQL Data Source Name accepts\n        user: \"uber\"\n        password: \"uber\"\n        maxConns: 20                   -- max number of connections to sql server from one host (optional)\n        maxIdleConns: 20               -- max number of idle conns to sql server from one host (optional)\n        maxConnLifetime: \"1h\"          -- max connection lifetime before it is discarded (optional)\n        connectAttributes:             -- custom dsn attributes, map of key-value pairs\n          tx_isolation: \"READ-COMMITTED\"   -- required only for mysql 5.6 and below, optional otherwise\n```\n\n## Multiple SQL(MySQL/Postgres) databases\nTo run Cadence clusters in a much larger scale using SQL database, multiple databases can be used as a sharded SQL database cluster.\n\nSet `useMultipleDatabases` to `true` and specify all databases' user/password/address using `multipleDatabasesConfig`:\n```yaml\npersistence:\n  ...\n  datastores:\n    datastore1:\n      sql:\n        pluginName: \"mysql\"            -- name of the go sql plugin\n        connectProtocol: \"tcp\"         -- connection protocol, tcp or anything that SQL Data Source Name accepts\n        maxConnLifetime: \"1h\"          -- max connection lifetime before it is discarded (optional)\n        useMultipleDatabases: true     -- this enabled the multiple SQL databases as sharded SQL cluster\n        nShards: 4                     -- the number of shards -- in this mode, it needs to be greater than one and equalt to the length of multipleDatabasesConfig\n        multipleDatabasesConfig:       -- each entry will represent a shard of the cluster\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence0\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence1\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence2\"\n        - user: \"root\"\n          password: \"cadence\"\n          connectAddr: \"127.0.0.1:3306\"\n          databaseName: \"cadence3\"\n```\n\n\nHow Cadence implement the sharding:\n\n* Workflow execution and historyShard records are sharded based on historyShardID(which is calculated  `historyShardID =hash(workflowID) % numHistoryShards` ), `dbShardID = historyShardID % numDBShards`\n* Workflow History is sharded based on history treeID(a treeID usually is the runID unless it has reset. In case of reset, it will share the same tree as the base run). In that case, `dbShardID = hash(treeID) % numDBShards`\n* Workflow tasks(for workflow/activity workers) is sharded based on domainID + tasklistName.  `dbShardID = hash(domainID + tasklistName ) % numDBShards`\n* Workflow visibility is  sharded based on domainID like we said above.  `dbShardID = hash(domainID ) % numDBShards`\n  * However, due to potential scalability issue, Cadence requires advanced visibility to run with multiple SQL database mode.\n* Internal domain records is using single shard, it’s only writing when register/update domain, and read is protected by domainCache  `dbShardID = DefaultShardID(0)`\n* Internal queue records is using single shard. Similarly, the read/write is low enough that it’s okay to not sharded. `dbShardID = DefaultShardID(0)`\n\n# Adding support for new database\n\n## For SQL Database\nAs there are many shared concepts and functionalities in SQL database, we abstracted those common code so that is much easier to implement persistence interfaces with any SQL database. It requires your database supports SQL operations like explicit transaction(with pessimistic locking)\n\nThis interface is tied to a specific schema i.e. the way data is laid out across tables and the table\nnames themselves are fixed. However, you get the flexibility wrt how you store the data within a table (i.e. column names and\ntypes are not fixed). The API interface can be found [here](https://github.com/cadence-workflow/cadence/blob/master/common/persistence/sql/plugins/interfaces.go).\nIt's basically a CRUD API for every table in the schema. A sample schema definition for mysql that uses this interface\ncan be found [here](https://github.com/cadence-workflow/cadence/blob/master/schema/mysql/v8/cadence/schema.sql)\n\nAny database that supports this interface can be plugged in with cadence server.\nWe have implemented Postgres within the repo, and also here is [**an example**](https://github.com/longquanzheng/cadence-extensions/tree/master/cadence-sqlite) to implement any database externally.\n\n\n## For other Non-SQL Database\nFor databases that don't support SQL operations like explicit transaction(with pessimistic locking),\nCadence requires at least supporting:\n 1. Multi-row single shard conditional write(also called LightWeight transaction by Cassandra terminology)\n 2. Strong consistency Read/Write operations\n\nThis NoSQL persistence API interface can be found [here](https://github.com/cadence-workflow/cadence/blob/master/common/persistence/nosql/nosqlplugin/interfaces.go).\nCurrently this is only implemented with Cassandra. DynamoDB and MongoDB are in progress.\n"
  },
  {
    "path": "docs/roadmap.md",
    "content": "# Cadence Roadmap\n\nThe following is a high-level quarterly roadmap of the [Cadence](https://cadenceworkflow.io/) project.\n\n## Q3 2019\n\n* [Resource-Specific Tasklist (a.k.a session)](https://github.com/cadence-workflow/cadence/blob/master/docs/design/1533-host-specific-tasklist.md)\n* [Visibility on Elastic Search](https://github.com/cadence-workflow/cadence/blob/master/docs/visibility-on-elasticsearch.md)\n* Scalable tasklist\n* MySQL support\n\n## Q4 2019\n\n* [Multi-DC support for Cadence replication](https://github.com/cadence-workflow/cadence/blob/master/docs/design/2290-cadence-ndc.md)\n* Workflow history archival\n* Workflow visibility archival\n* [Synchronous Request Reply](https://github.com/cadence-workflow/cadence/blob/master/docs/design/2215-synchronous-request-reply.md)\n* Postgres SQL support\n\n## Q1 2020\n\n* Service availability and reliability improvements\n* [Domain level AuthN and AuthZ support](https://github.com/cadence-workflow/cadence/issues/2833)\n* Graceful domain failover design\n* UI bug fixes and performance improvements\n\n## Q2 2020\n\n* Kafka deprecation for Cadence replication\n* Graceful domain failover\n* Multi-tenancy task prioritization and resource isolation\n* UI and CLI feature parity\n\n## Q3 2020 and beyond\n\n* GRPC support\n* Parent-child affinity\n* Task priority\n* Reset 2.0 (more flexibility and handle child)\n* Safe workflow code rollout\n"
  },
  {
    "path": "docs/scalable_tasklist.md",
    "content": "# Overview\nMatching is a sharded service which is sharded by tasklist, which means that all requests to a certain tasklist has to be processed by one Matching host. To avoid a single Matching host becoming the bottleneck and make the system scalable, scalable tasklist was introduced to allow a tasklist to be partitioned so that the requests can be processed by multiple Matching hosts. The partitions are transparent to customers, so when a request to a scalable tasklist arrives at Cadence server, it has to select a partition for the request. We'll describe the architecture of scalable tasklist and the selection algorithm we use.\n\n# Requirements\nThere are 2 requirements for this feature:\n\n1. Ensure a fair distribution of the requests among all partitions\n2. Maximize the utilization of customer pollers\n\nThe first requirement is straightforward, which is the reason for introducing scalable tasklist. The second requirement is to make sure pollers are not waiting at a partition without any task because the default polling timeout is 60s.\n\n# Architecture\n![image alt text](images/scalable-tasklist-forwarding.png)\n\nThe partitions are organized in a tree-structure. The number of child nodes is configurable, but in the diagram we just show a scalable tasklist with 6 partitions organized in a binary tree. When a partition receives a request, it can forward the request to its parent partition recursively until the root partition being reached.\n\n# Configuration\nThe number of partitions of a tasklist is configured by 2 dynamicconfigs:\n\n1. [matching.numTasklistReadPartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3350)\n2. [matching.numTasklistWritePartitions](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3344)\n\nWe're migrating this configuration from dynamicconfig to database. The following dynamicconfig is used to control where to read the number of partitions from:\n- [matching.enableGetNumberOfPartitionsFromCache](https://github.com/cadence-workflow/cadence/blob/v1.2.15-prerelease02/common/dynamicconfig/constants.go#L4008)\nTo update the number of partitions, use the following CLI command:\n```\ncadence admin tasklist update-partition -h\n```\nTo get the number of partitions from database, use the following CLI command:\n```\ncadence admin tasklist describe -h\n```\n\nThe tree-structure and forwarding mechanism is configured by these dynamicconfigs:\n\n1. [matching.forwarderMaxChildrenPerNode](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3374)\n2. [matching.forwarderMaxOutstandingPolls](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3356)\n3. [matching.forwarderMaxOutstandingTasks](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3362)\n4. [matching.forwarderMaxRatePerSecond](https://github.com/cadence-workflow/cadence/blob/v1.2.13/common/dynamicconfig/constants.go#L3368)\n\n# Selection Algorithms\nThe selection algorithms are implemented as a LoadBalancer in [client/matching package](https://github.com/cadence-workflow/cadence/blob/v1.2.13/client/matching/loadbalancer.go#L37).\n\n## Random Selection\nThis is the first algorithm and it's been widely adopted in production. It's completely stateless and uses a shared nothing architecture. The probabilistic model of discrete uniform distribution guarantees the fairness of the distribution of requests. And the utilization is improved by the tree-structure. For example, as shown in the diagram, if a task is produced to partition-5, but a poller is assigned to partition-3, we don't want the poller to wait at partition-3 for 60s and retry the poll request. And the retry has a 5/6 probability of not hitting partition-5. With the tree-structure and forwarding mechanism, the poller request and task are forwarded to root partition. So an idle poller waiting at partition-3 is utilized in this case.\n\n## Round-Robin Selection\nThis algorithm also ensure a fair distribution of requests even with a small number of requests. It also uses a shared nothing architecture but it's soft stateful because it uses a cache to remember the previous selected partition.\n\n## Weighted Selection\nThe algorithm selects a partition based on the backlog size of each partition if any partition has a backlog size larger than 100. If all partitions don't have a large backlog, it falls back to round-robin selection. It can be proven mathematically that it's better than the previous 2 algorithms.\n\nFor a tasklist with $N$ partitions, assuming the number of tasks in each partition is represented by $L_i$, the utilization fraction of partition $i$ is:\n\n$$U_i = \\frac{L_i}{\\Sigma_{i=0}^{n-1}L_i}, \\forall i \\in \\{0,1,..,N-1\\}$$\n\nAccording to Little's Law, in a stable queue system, the average number of tasks in a queue\n\n$$L = \\lambda W$$\n\n where $\\lambda$ is the average arrival rate and $W$ is the average wait time a task spends in the queue. A queue is stable if the utilization factor $\\rho$ is less than 1, where\n\n $$\\rho = \\frac{\\lambda}{X\\mu}$$\n\nHere $\\lambda$ is the arrival rate of tasks, $\\mu$ is the service rate of a single poller, and $X$ is the number of pollers. However, there might be some time that $\\rho$ is greater than 1, in which case, tasks are not dropped but persisted into the database. Assuming the number of tasks in the database for partition $i$ is $B_{i}$, the arrival rate is $\\lambda_{i}$, the utilization fraction is:\n\n$$U_i = \\frac{B_i+\\lambda_iW}{\\Sigma_{i=0}^{n-1}B_i+W\\Sigma_{i=0}^{n-1}\\lambda_i}, \\forall i \\in \\{0,1,..,N-1\\}$$\n\nTo maximize the utilization of pollers, the probability of partition $i$ being selected for a poller should be:\n\n$$P_i = U_{i}, \\forall i \\in \\{0,1,..,N-1\\}$$\n\nAssuming $B_i=0, \\forall i \\in \\{0,1,..,N-1\\}$ and $\\lambda_i=\\frac{\\lambda}{N}, \\forall i \\in \\{0,1,..,N-1\\}$,\n\n$$P_i = \\frac{1}{N}, \\forall i \\in \\{0,1,..,N-1\\}$$\n\nIt means that when the queue system is stable and the producer traffic is evenly distributed among all partitions, selecting partition based on number of tasks is equivalent to random and round-robin selection.\n\nHere λ is the producer QPS and W is the average matching latency. In production, we observed that the average matching latency is about 10ms, and most tasklists only have a few hundred producer QPS of traffic. So for most stable partitions, $L$ is less than 10. So if no partition has a very large backlog size, it's safe to fallback to random selection or round-robin selection algorithm.\n"
  },
  {
    "path": "docs/setup/MYSQL_SETUP.md",
    "content": "# Setup local MySQL with Docker\nThis document describes how to install MySQL 8.0 locally with Docker.\n\n>Note: Install the docker on your machine before installing the MySQL.\n* Make sure any MySQL containers are terminated and removed\n```\ndocker ps -a\ndocker kill <container_id> && docker rm <container_id> # remove any MySQL containers.\n```\n* Fetch docker image (version 8.0 is what Travis runs so its what you will want locally)\n```\ndocker pull mysql/mysql-server:8.0\n```\n* Run docker container (note the port mapping so that 3306 is exposed locally)\n```\ndocker run -p 3306:3306 --name=mysql1 -d mysql/mysql-server:8.0\n```\n* When docker starts up the root MySQL user will have an auto generated password. You need to get that password to log into the container\n```\ndocker logs mysql1 2>&1 | grep GENERATED\n# The result looks like: [Entrypoint] GENERATED ROOT PASSWORD: iHqEvRYm6UP#YN$es;YnV3m(oJ\n```\n* Log into the container (when prompted for password use the password gotten from last step).\n```\ndocker exec -it mysql1 mysql -uroot -p\n```\n* Before any SQL operations can be performed you must reset the root user's password (use anything you like in replace of root_password).\n```\nSET PASSWORD = PASSWORD('root_password');\n```\n* Now create the user that local MySQL tests will use. Also grant all privileges to user.\n```\nCREATE USER 'uber'@'%' IDENTIFIED BY 'uber';\nGRANT ALL PRIVILEGES ON *.* TO 'uber'@'%';\n```"
  },
  {
    "path": "docs/setup/POSTGRES_SETUP.md",
    "content": "# Setup local Postgres with Docker\nThis document describes how to install latest Postgre locally with Docker.\n\n>Note: Install the docker on your machine before installing the MySQL.\n* Make sure any MySQL containers are terminated and removed\n```\ndocker ps -a\ndocker kill <container_id> && docker rm <container_id> # remove any Postgres containers.\n```\n* Fetch docker image \n```\ndocker pull postgres\n```\n* Run docker container (note the port mapping so that 5432 is exposed locally)\n```\nmkdir -p ~/docker/volumes/postgres\ndocker run --rm --name pg-docker -e POSTGRES_PASSWORD=cadence -d -p 5432:5432 -v ~/docker/volumes/postgres:/var/lib/postgresql/data postgres\n```\n* Log into the container (when prompted for password use the password gotten from last step).\n```\npsql -h localhost -U postgres -d cadence\n```"
  },
  {
    "path": "docs/toc.md",
    "content": "# Table of Contents\n- [Persistence](persistence.md) \n- [Visibility on ElasticSearch](visibility-on-elasticsearch.md)"
  },
  {
    "path": "docs/visibility-on-elasticsearch.md",
    "content": "# Overview\nCadence visibility APIs allow users to list open or closed workflows with filters such as WorkflowType or WorkflowID.\nWith Cassandra, there are issues around scalability and performance, for example:\n - list large amount of workflows may kill Cassandra node.\n - data is partitioned by domain, which means writing large amount of workflow to one domain will cause Cassandra nodes hotspots.\n - query with filter is slow for large data.\n\n(With MySQL, there might be similar issues but not tested)\n\nIn addition, Cadence want to support users to perform flexible query with multiple filters on even custom info.\nThat's why Cadence add support for enhanced visibility features on top of ElasticSearch (Note as ES below), which includes:\n - new visibility APIs to List/Scan/Count workflows with SQL like query\n - search attributes feature to support users provide custom info\n\n# Quick Start\n## Local Cadence Docker Setup\n1. Increase docker memory to higher 6GB. Docker -> Preference -> advanced -> memory limit\n2. Get docker compose file. Run `curl -O https://raw.githubusercontent.com/uber/cadence/master/docker/docker-compose-es.yml`\n3. Start cadence docker which contains Kafka, Zookeeper and ElasticSearch. Run `docker compose -f docker-compose-es.yml up`\n4. From docker output log, make sure ES and cadence started correctly. If encounter disk space not enough, try `docker system prune -a --volumes`\n5. Register local domain and start using it. `cadence --do samples-domain d re`\n\n\n## CLI Search Attributes Support\n\nMake sure Cadence CLI version is 0.6.4+\n\n### new list API examples\n\n```\ncadence --do samples-domain wf list -q 'WorkflowType = \"main.Workflow\" and (WorkflowID = \"1645a588-4772-4dab-b276-5f9db108b3a8\" or RunID = \"be66519b-5f09-40cd-b2e8-20e4106244dc\")'\ncadence --do samples-domain wf list -q 'WorkflowType = \"main.Workflow\" StartTime > \"2019-06-07T16:46:34-08:00\" and CloseTime = missing'\n```\nTo list only open workflows, add `CloseTime = missing` to the end of query.\n\n### start workflow with search attributes\n\n```\ncadence --do samples-domain workflow start --tl helloWorldGroup --wt main.Workflow --et 60 --dt 10 -i '\"input arg\"' -search_attr_key 'CustomIntField | CustomKeywordField | CustomStringField |  CustomBoolField | CustomDatetimeField' -search_attr_value '5 | keyword1 | my test | true | 2019-06-07T16:16:36-08:00'\n```\n\nNote: start workflow with search attributes but without ES will succeed as normal, but will not be searchable and will not be shown in list result.\n\n### search workflow with new list API\n\n```\ncadence --do samples-domain wf list -q '(CustomKeywordField = \"keyword1\" and CustomIntField >= 5) or CustomKeywordField = \"keyword2\"' -psa\ncadence --do samples-domain wf list -q 'CustomKeywordField in (\"keyword2\", \"keyword1\") and CustomIntField >= 5 and CloseTime between \"2018-06-07T16:16:36-08:00\" and \"2019-06-07T16:46:34-08:00\" order by CustomDatetimeField desc' -psa\n```\n\n(Search attributes can be updated inside workflow, see example [here](https://github.com/cadence-workflow/cadence-samples/tree/master/cmd/samples/recipes/searchattributes).\n\n# Details\n## Dependencies\n- Zookeeper - for Kafka to start\n- Kafka - message queue for visibility data\n- ElasticSearch v6+ - for data search (early ES version may not support some queries)\n\n## Configuration\n```\npersistence:\n  ...\n  advancedVisibilityStore: es-visibility\n  ...\n  datastore:\n    es-visibility:\n      elasticsearch:\n        url:\n          scheme: \"http\"\n          host: \"127.0.0.1:9200\"\n        indices:\n          visibility: cadence-visibility-dev\n```\nThis part is used to config advanced visibility store to ElasticSearch.\n - `url` is for Cadence to discover ES\n - `indices/visibility` is ElasticSearch index name for the deployment.\n\nOptional TLS Support can be enabled by setting the TLS config as follows:\n```yaml\nelasticsearch:\n  url:\n    scheme: \"https\"\n    host: \"127.0.0.1:9200\"\n  indices:\n    visibility: cadence-visibility-dev\n  tls:\n    enabled: true\n    caFile: /secrets/cadence/elasticsearch_cert.pem\n    enableHostVerification: true\n    serverName: myServerName\n    certFile: /secrets/cadence/certfile.crt\n    keyFile: /secrets/cadence/keyfile.key\n    sslmode: false\n```\n\nAlso need to add a kafka topic to visibility, as shown below.\n```\nkafka:\n  ...\n  applications:\n    visibility:\n      topic: cadence-visibility-dev\n      dlq-topic: cadence-visibility-dev-dlq\n  ...\n```\n\nThere are dynamic configs to control ElasticSearch visibility features:\n- `system.writeVisibilityStoreName` is an string property to control how to write visibility to data store.\n`\"off\"` means do not write to advanced data store, same as db\n`\"es\"` means only write to advanced data store (in this case is es),\n`\"db,es\"` means write to both DB (Cassandra or MySQL) and advanced data store\n- `system.readVisibilityStoreName` is a string property to control the read source for Cadence List APIs.\n"
  },
  {
    "path": "environment/env.go",
    "content": "// Copyright (c) 2016 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage environment\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"strconv\"\n)\n\nconst (\n\t// Localhost default localhost\n\tLocalhost = \"127.0.0.1\"\n\n\t// CassandraSeeds env\n\tCassandraSeeds = \"CASSANDRA_SEEDS\"\n\t// CassandraPort env\n\tCassandraPort = \"CASSANDRA_DB_PORT\"\n\t// CassandraDefaultPort Cassandra default port\n\tCassandraDefaultPort = \"9042\"\n\t// CassandraUsername env\n\tCassandraUsername = \"CASSANDRA_DB_USERNAME\"\n\t// CassandraDefaultUsername Cassandra default username\n\tCassandraDefaultUsername = \"\"\n\t// CassandraPassword env\n\tCassandraPassword = \"CASSANDRA_DB_PASSWORD\"\n\t// CassandraDefaultPassword Cassandra default password\n\tCassandraDefaultPassword = \"\"\n\t// CassandraAllowedAuthenticators env\n\tCassandraAllowedAuthenticators = \"CASSANDRA_DB_ALLOWED_AUTHENTICATORS\"\n\t// CassandraProtoVersion env\n\tCassandraProtoVersion = \"CASSANDRA_PROTO_VERSION\"\n\t// CassandraDefaultProtoVersion Cassandra default protocol version\n\tCassandraDefaultProtoVersion = \"4\"\n\t// CassandraDefaultProtoVersionInteger Cassandra default protocol version int version\n\tCassandraDefaultProtoVersionInteger = 4\n\n\t// MySQLSeeds env\n\tMySQLSeeds = \"MYSQL_SEEDS\"\n\t// MySQLPort env\n\tMySQLPort = \"MYSQL_PORT\"\n\t// MySQLDefaultPort is MySQL default port\n\tMySQLDefaultPort = \"3306\"\n\t// MySQLUser env\n\tMySQLUser = \"MYSQL_USER\"\n\t// MySQLDefaultUser is default user\n\tMySQLDefaultUser = \"root\"\n\t// MySQLPassword env\n\tMySQLPassword = \"MYSQL_PASSWORD\"\n\t// MySQLDefaultPassword is default password\n\tMySQLDefaultPassword = \"cadence\"\n\n\t// MongoSeeds env\n\tMongoSeeds = \"MONGO_SEEDS\"\n\t// MongoPort env\n\tMongoPort = \"MONGO_PORT\"\n\t// MongoDefaultPort is Mongo default port\n\tMongoDefaultPort = \"27017\"\n\n\t// KafkaSeeds env\n\tKafkaSeeds = \"KAFKA_SEEDS\"\n\t// KafkaPort env\n\tKafkaPort = \"KAFKA_PORT\"\n\t// KafkaDefaultPort Kafka default port\n\tKafkaDefaultPort = \"9092\"\n\n\t// ESSeeds env\n\tESSeeds = \"ES_SEEDS\"\n\t// ESPort env\n\tESPort = \"ES_PORT\"\n\t// ESDefaultPort ES default port\n\tESDefaultPort = \"9200\"\n\t// ESVersion is the ElasticSearch version\n\tESVersion = \"ES_VERSION\"\n\t// ESDefaultVersion is the default version\n\tESDefaultVersion = \"v6\"\n\n\t// PostgresSeeds env\n\tPostgresSeeds = \"POSTGRES_SEEDS\"\n\t// PostgresPort env\n\tPostgresPort = \"POSTGRES_PORT\"\n\t// PostgresDefaultPort Postgres default port\n\tPostgresDefaultPort = \"5432\"\n\n\t// CLITransportProtocol env\n\tCLITransportProtocol = \"CADENCE_CLI_TRANSPORT_PROTOCOL\"\n\t// DefaultCLITransportProtocol  CLI default channel\n\tDefaultCLITransportProtocol = \"tchannel\"\n)\n\nvar envDefaults = map[string]string{\n\tCassandraSeeds:        Localhost,\n\tCassandraPort:         CassandraDefaultPort,\n\tCassandraProtoVersion: CassandraDefaultProtoVersion,\n\tMySQLSeeds:            Localhost,\n\tMySQLPort:             MySQLDefaultPort,\n\tPostgresSeeds:         Localhost,\n\tPostgresPort:          PostgresDefaultPort,\n\tKafkaSeeds:            Localhost,\n\tKafkaPort:             KafkaDefaultPort,\n\tESSeeds:               Localhost,\n\tESPort:                ESDefaultPort,\n\tCLITransportProtocol:  DefaultCLITransportProtocol,\n}\n\n// SetupEnv setup the necessary env\nfunc SetupEnv() error {\n\tfor k, v := range envDefaults {\n\t\tif os.Getenv(k) == \"\" {\n\t\t\tif err := setEnv(k, v); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\n// GetCassandraAddress return the cassandra address\nfunc GetCassandraAddress() string {\n\tif addr := os.Getenv(CassandraSeeds); addr != \"\" {\n\t\treturn addr\n\t}\n\n\treturn envDefaults[CassandraSeeds]\n}\n\n// GetCassandraPort return the cassandra port\nfunc GetCassandraPort() (int, error) {\n\tport := os.Getenv(CassandraPort)\n\tif port == \"\" {\n\t\tport = envDefaults[CassandraPort]\n\t}\n\n\treturn strconv.Atoi(port)\n}\n\n// GetCassandraUsername return the cassandra username\nfunc GetCassandraUsername() string {\n\tuser := os.Getenv(CassandraUsername)\n\tif user != \"\" {\n\t\treturn user\n\t}\n\treturn CassandraDefaultUsername\n}\n\n// GetCassandraPassword return the cassandra password\nfunc GetCassandraPassword() string {\n\tpass := os.Getenv(CassandraPassword)\n\tif pass != \"\" {\n\t\treturn pass\n\t}\n\n\treturn CassandraDefaultPassword\n}\n\n// GetCassandraAllowedAuthenticators return the cassandra allowed authenticators\nfunc GetCassandraAllowedAuthenticators() []string {\n\tvar authenticators []string\n\tconfiguredAuthenticators := os.Getenv(CassandraAllowedAuthenticators)\n\tif configuredAuthenticators == \"\" {\n\t\treturn authenticators\n\t}\n\n\tauthenticators = append(authenticators, configuredAuthenticators)\n\treturn authenticators\n}\n\n// GetCassandraProtoVersion return the cassandra protocol version\nfunc GetCassandraProtoVersion() (int, error) {\n\tprotoVersion := os.Getenv(CassandraProtoVersion)\n\tif protoVersion == \"\" {\n\t\tprotoVersion = envDefaults[CassandraProtoVersion]\n\t}\n\n\treturn strconv.Atoi(protoVersion)\n}\n\n// GetMySQLAddress return the MySQL address\nfunc GetMySQLAddress() string {\n\taddr := os.Getenv(MySQLSeeds)\n\tif addr == \"\" {\n\t\taddr = Localhost\n\t}\n\treturn addr\n}\n\n// GetMySQLPort return the MySQL port\nfunc GetMySQLPort() (int, error) {\n\tport := os.Getenv(MySQLPort)\n\tif port == \"\" {\n\t\tport = MySQLDefaultPort\n\t}\n\n\treturn strconv.Atoi(port)\n}\n\n// GetMySQLUser return the MySQL user\nfunc GetMySQLUser() string {\n\tuser := os.Getenv(MySQLUser)\n\tif user == \"\" {\n\t\tuser = MySQLDefaultUser\n\t}\n\treturn user\n}\n\n// GetMySQLPassword return the MySQL password\nfunc GetMySQLPassword() string {\n\tpw := os.Getenv(MySQLPassword)\n\tif pw == \"\" {\n\t\tpw = MySQLDefaultPassword\n\t}\n\treturn pw\n}\n\n// GetPostgresAddress return the Postgres address\nfunc GetPostgresAddress() string {\n\taddr := os.Getenv(PostgresSeeds)\n\tif addr == \"\" {\n\t\taddr = Localhost\n\t}\n\treturn addr\n}\n\n// GetPostgresPort return the Postgres port\nfunc GetPostgresPort() (int, error) {\n\tport := os.Getenv(PostgresPort)\n\tif port == \"\" {\n\t\tport = PostgresDefaultPort\n\t}\n\n\treturn strconv.Atoi(port)\n}\n\n// GetESVersion return the ElasticSearch version\nfunc GetESVersion() string {\n\tversion := os.Getenv(ESVersion)\n\tif version == \"\" {\n\t\tversion = ESDefaultVersion\n\t}\n\treturn version\n}\n\n// GetMongoAddress return the MySQL address\nfunc GetMongoAddress() string {\n\taddr := os.Getenv(MongoSeeds)\n\tif addr == \"\" {\n\t\taddr = Localhost\n\t}\n\treturn addr\n}\n\n// GetMongoPort return the MySQL port\nfunc GetMongoPort() (int, error) {\n\tport := os.Getenv(MongoPort)\n\tif port == \"\" {\n\t\tport = MongoDefaultPort\n\t}\n\n\treturn strconv.Atoi(port)\n}\n\nfunc setEnv(key string, val string) error {\n\tif err := os.Setenv(key, val); err != nil {\n\t\treturn fmt.Errorf(\"setting env %q: %w\", key, err)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "environment/env_test.go",
    "content": "// Copyright (c) 2016 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage environment\n\nimport (\n\t\"os\"\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n)\n\nfunc TestSetupEnv(t *testing.T) {\n\tif err := SetupEnv(); err != nil {\n\t\tt.Fatalf(\"SetupEnv() failed, err: %v\", err)\n\t}\n}\n\nfunc TestGetCassandraAddress(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraSeeds,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   envDefaults[CassandraSeeds],\n\t\t},\n\t\t{\n\t\t\tname:      \"custom address\",\n\t\t\tenvVarKey: CassandraSeeds,\n\t\t\tenvVarVal: \"casseed\",\n\t\t\twantVal:   \"casseed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetCassandraAddress()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetCassandraAddress() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCassandraPort(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantErr   bool\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraPort,\n\t\t\tenvVarVal: \"\",\n\t\t\twantErr:   false,\n\t\t\twantVal:   mustConvertInt(t, envDefaults[CassandraPort]),\n\t\t},\n\t\t{\n\t\t\tname:      \"non-int port value\",\n\t\t\tenvVarKey: CassandraPort,\n\t\t\tenvVarVal: \"xyz\",\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom port value\",\n\t\t\tenvVarKey: CassandraPort,\n\t\t\tenvVarVal: \"1001\",\n\t\t\twantVal:   1001,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal, err := GetCassandraPort()\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Fatalf(\"GetCassandraPort() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tt.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetCassandraPort() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCassandraUsername(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraUsername,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   CassandraDefaultUsername,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom user name\",\n\t\t\tenvVarKey: CassandraUsername,\n\t\t\tenvVarVal: \"user1\",\n\t\t\twantVal:   \"user1\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetCassandraUsername()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetCassandraUsername() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCassandraPassword(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraPassword,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   CassandraDefaultUsername,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom pw\",\n\t\t\tenvVarKey: CassandraPassword,\n\t\t\tenvVarVal: \"xyz123\",\n\t\t\twantVal:   \"xyz123\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetCassandraPassword()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetCassandraPassword() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCassandraAllowedAuthenticators(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   []string\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraAllowedAuthenticators,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   nil,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom authenticators\",\n\t\t\tenvVarKey: CassandraAllowedAuthenticators,\n\t\t\tenvVarVal: \"auth1\",\n\t\t\twantVal:   []string{\"auth1\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetCassandraAllowedAuthenticators()\n\n\t\t\tif diff := cmp.Diff(gotVal, tt.wantVal); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetCassandraProtoVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantErr   bool\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: CassandraProtoVersion,\n\t\t\tenvVarVal: \"\",\n\t\t\twantErr:   false,\n\t\t\twantVal:   mustConvertInt(t, envDefaults[CassandraProtoVersion]),\n\t\t},\n\t\t{\n\t\t\tname:      \"non-int proto version\",\n\t\t\tenvVarKey: CassandraProtoVersion,\n\t\t\tenvVarVal: \"xyz\",\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom proto version\",\n\t\t\tenvVarKey: CassandraProtoVersion,\n\t\t\tenvVarVal: \"35\",\n\t\t\twantVal:   35,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal, err := GetCassandraProtoVersion()\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Fatalf(\"GetCassandraProtoVersion() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tt.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetCassandraProtoVersion() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMySQLAddress(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MySQLSeeds,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   Localhost,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: MySQLSeeds,\n\t\t\tenvVarVal: \"mysqlseed\",\n\t\t\twantVal:   \"mysqlseed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetMySQLAddress()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMySQLAddress() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMySQLPort(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantErr   bool\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MySQLPort,\n\t\t\tenvVarVal: \"\",\n\t\t\twantErr:   false,\n\t\t\twantVal:   mustConvertInt(t, MySQLDefaultPort),\n\t\t},\n\t\t{\n\t\t\tname:      \"non-int port\",\n\t\t\tenvVarKey: MySQLPort,\n\t\t\tenvVarVal: \"xyz\",\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom port\",\n\t\t\tenvVarKey: MySQLPort,\n\t\t\tenvVarVal: \"8787\",\n\t\t\twantVal:   8787,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal, err := GetMySQLPort()\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Fatalf(\"GetMySQLPort() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tt.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMySQLPort() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMySQLUser(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MySQLUser,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   MySQLDefaultUser,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: MySQLUser,\n\t\t\tenvVarVal: \"mysqluser\",\n\t\t\twantVal:   \"mysqluser\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetMySQLUser()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMySQLUser() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMySQLPassword(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MySQLPassword,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   MySQLDefaultPassword,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: MySQLPassword,\n\t\t\tenvVarVal: \"klmno\",\n\t\t\twantVal:   \"klmno\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetMySQLPassword()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMySQLPassword() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetPostgresAddress(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: PostgresSeeds,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   Localhost,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: PostgresSeeds,\n\t\t\tenvVarVal: \"postgresseed\",\n\t\t\twantVal:   \"postgresseed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetPostgresAddress()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetPostgresAddress() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetPostgresPort(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantErr   bool\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: PostgresPort,\n\t\t\tenvVarVal: \"\",\n\t\t\twantErr:   false,\n\t\t\twantVal:   mustConvertInt(t, PostgresDefaultPort),\n\t\t},\n\t\t{\n\t\t\tname:      \"non-int port\",\n\t\t\tenvVarKey: PostgresPort,\n\t\t\tenvVarVal: \"xyz\",\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom port\",\n\t\t\tenvVarKey: PostgresPort,\n\t\t\tenvVarVal: \"8787\",\n\t\t\twantVal:   8787,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal, err := GetPostgresPort()\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Fatalf(\"GetPostgresPort() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tt.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetPostgresPort() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetESVersion(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: ESVersion,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   ESDefaultVersion,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: ESVersion,\n\t\t\tenvVarVal: \"v1234\",\n\t\t\twantVal:   \"v1234\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetESVersion()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetESVersion() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMongoAddress(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MongoSeeds,\n\t\t\tenvVarVal: \"\",\n\t\t\twantVal:   Localhost,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom\",\n\t\t\tenvVarKey: MongoSeeds,\n\t\t\tenvVarVal: \"mongoseed\",\n\t\t\twantVal:   \"mongoseed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal := GetMongoAddress()\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMongoAddress() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetMongoPort(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tenvVarKey string\n\t\tenvVarVal string\n\t\twantErr   bool\n\t\twantVal   any\n\t}{\n\t\t{\n\t\t\tname:      \"default\",\n\t\t\tenvVarKey: MongoPort,\n\t\t\tenvVarVal: \"\",\n\t\t\twantErr:   false,\n\t\t\twantVal:   mustConvertInt(t, MongoDefaultPort),\n\t\t},\n\t\t{\n\t\t\tname:      \"non-int port\",\n\t\t\tenvVarKey: MongoPort,\n\t\t\tenvVarVal: \"xyz\",\n\t\t\twantErr:   true,\n\t\t},\n\t\t{\n\t\t\tname:      \"custom port\",\n\t\t\tenvVarKey: MongoPort,\n\t\t\tenvVarVal: \"8787\",\n\t\t\twantVal:   8787,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tos.Setenv(tt.envVarKey, tt.envVarVal)\n\t\t\tgotVal, err := GetMongoPort()\n\t\t\tif (err != nil) != tt.wantErr {\n\t\t\t\tt.Fatalf(\"GetMongoPort() error = %v, wantErr %v\", err, tt.wantErr)\n\t\t\t}\n\n\t\t\tif err != nil || tt.wantErr {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif gotVal != tt.wantVal {\n\t\t\t\tt.Fatalf(\"GetMongoPort() = %v, want %v\", gotVal, tt.wantVal)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mustConvertInt(t *testing.T, s string) int {\n\tv, err := strconv.Atoi(s)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to convert string to int, err: %v\", err)\n\t}\n\treturn v\n}\n"
  },
  {
    "path": "go.mod",
    "content": "module github.com/uber/cadence\n\ngo 1.23.0\n\ntoolchain go1.23.4\n\nrequire (\n\tgithub.com/MicahParks/keyfunc/v2 v2.1.0\n\tgithub.com/VividCortex/mysqlerr v1.0.0\n\tgithub.com/aws/aws-sdk-go v1.54.12\n\tgithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748\n\tgithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9\n\tgithub.com/dave/dst v0.26.2\n\tgithub.com/davecgh/go-spew v1.1.1\n\tgithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da\n\tgithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab\n\tgithub.com/fatih/color v1.13.0\n\tgithub.com/go-sql-driver/mysql v1.7.1\n\tgithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1\n\tgithub.com/gogo/protobuf v1.3.2\n\tgithub.com/golang-jwt/jwt/v5 v5.2.0\n\tgithub.com/golang/mock v1.6.0\n\tgithub.com/google/go-cmp v0.6.0\n\tgithub.com/google/uuid v1.6.0\n\tgithub.com/hashicorp/go-version v1.2.0\n\tgithub.com/iancoleman/strcase v0.2.0\n\tgithub.com/jmespath/go-jmespath v0.4.0\n\tgithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee\n\tgithub.com/jonboulle/clockwork v0.5.0\n\tgithub.com/lib/pq v1.2.0\n\tgithub.com/m3db/prometheus_client_golang v0.8.1\n\tgithub.com/olekukonko/tablewriter v0.0.4\n\tgithub.com/olivere/elastic v6.2.37+incompatible\n\tgithub.com/olivere/elastic/v7 v7.0.21\n\tgithub.com/opentracing/opentracing-go v1.2.0\n\tgithub.com/otiai10/copy v1.1.1\n\tgithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709\n\tgithub.com/sirupsen/logrus v1.9.0 // indirect\n\tgithub.com/startreedata/pinot-client-go v0.2.0 // latest release supports pinot v0.12.0 which is also internal version\n\tgithub.com/stretchr/testify v1.10.0\n\tgithub.com/uber-go/tally v3.3.15+incompatible\n\tgithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4\n\tgithub.com/uber/ringpop-go v0.8.5\n\tgithub.com/uber/tchannel-go v1.22.2\n\tgithub.com/urfave/cli/v2 v2.27.4\n\tgithub.com/valyala/fastjson v1.4.1\n\tgithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c\n\tgithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2\n\tgo.etcd.io/etcd/client/v3 v3.5.5\n\tgo.mongodb.org/mongo-driver v1.7.3\n\tgo.uber.org/atomic v1.10.0\n\tgo.uber.org/cadence v0.19.0\n\tgo.uber.org/config v1.4.0\n\tgo.uber.org/fx v1.23.0\n\tgo.uber.org/multierr v1.10.0\n\tgo.uber.org/thriftrw v1.29.2\n\tgo.uber.org/yarpc v1.70.3\n\tgo.uber.org/zap v1.26.0\n\tgolang.org/x/exp v0.0.0-20231226003508-02704c960a9b\n\tgolang.org/x/net v0.40.0\n\tgolang.org/x/sync v0.14.0\n\tgolang.org/x/time v0.5.0\n\tgolang.org/x/tools v0.22.0\n\tgonum.org/v1/gonum v0.7.0\n\tgoogle.golang.org/grpc v1.59.0\n\tgopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19\n\tgopkg.in/yaml.v2 v2.4.0\n)\n\nrequire (\n\tgithub.com/IBM/sarama v1.45.2\n\tgithub.com/Masterminds/semver/v3 v3.2.1\n\tgithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568\n\tgithub.com/google/gofuzz v1.0.0\n\tgithub.com/mark3labs/mcp-go v0.18.0\n\tgithub.com/ncruces/go-sqlite3 v0.22.0\n\tgithub.com/opensearch-project/opensearch-go/v4 v4.1.0\n\tgithub.com/robfig/cron/v3 v3.0.1\n\tgo.etcd.io/etcd/api/v3 v3.5.5\n\tgo.uber.org/mock v0.5.0\n)\n\nrequire (\n\tgithub.com/coreos/go-semver v0.3.0 // indirect\n\tgithub.com/coreos/go-systemd/v22 v22.3.2 // indirect\n\tgithub.com/ncruces/julianday v1.0.0 // indirect\n\tgithub.com/pierrec/lz4/v4 v4.1.22 // indirect\n\tgithub.com/robfig/cron v1.2.0 // indirect\n\tgithub.com/rogpeppe/go-internal v1.6.1 // indirect\n\tgithub.com/tetratelabs/wazero v1.8.2 // indirect\n\tgithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 // indirect\n\tgithub.com/yosida95/uritemplate/v3 v3.0.2 // indirect\n\tgo.etcd.io/etcd/client/pkg/v3 v3.5.5 // indirect\n\tgoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b // indirect\n\tgoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a // indirect\n)\n\nrequire (\n\tgithub.com/BurntSushi/toml v1.3.2 // indirect\n\tgithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 // indirect\n\tgithub.com/apache/thrift v0.17.0 // indirect\n\tgithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3 // indirect\n\tgithub.com/beorn7/perks v1.0.1 // indirect\n\tgithub.com/cespare/xxhash/v2 v2.2.0 // indirect\n\tgithub.com/cpuguy83/go-md2man/v2 v2.0.4 // indirect\n\tgithub.com/cristalhq/jwt/v3 v3.1.0 // indirect\n\tgithub.com/eapache/go-resiliency v1.7.0 // indirect\n\tgithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3 // indirect\n\tgithub.com/eapache/queue v1.1.0 // indirect\n\tgithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a // indirect\n\tgithub.com/fatih/structtag v1.2.0 // indirect\n\tgithub.com/go-stack/stack v1.8.0 // indirect\n\tgithub.com/go-zookeeper/zk v1.0.3 // indirect\n\tgithub.com/gogo/googleapis v1.3.2 // indirect\n\tgithub.com/gogo/status v1.1.0 // indirect\n\tgithub.com/golang/protobuf v1.5.4 // indirect\n\tgithub.com/golang/snappy v0.0.4\n\tgithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed\n\tgithub.com/hashicorp/errwrap v1.0.0 // indirect\n\tgithub.com/hashicorp/go-multierror v1.1.1 // indirect\n\tgithub.com/hashicorp/go-uuid v1.0.3 // indirect\n\tgithub.com/jcmturner/aescts/v2 v2.0.0 // indirect\n\tgithub.com/jcmturner/dnsutils/v2 v2.0.0 // indirect\n\tgithub.com/jcmturner/gofork v1.7.6 // indirect\n\tgithub.com/jcmturner/gokrb5/v8 v8.4.4 // indirect\n\tgithub.com/jcmturner/rpc/v2 v2.0.3 // indirect\n\tgithub.com/jessevdk/go-flags v1.4.0 // indirect\n\tgithub.com/josharian/intern v1.0.0 // indirect\n\tgithub.com/kisielk/errcheck v1.5.0 // indirect\n\tgithub.com/klauspost/compress v1.18.0 // indirect\n\tgithub.com/m3db/prometheus_client_model v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_common v0.1.0 // indirect\n\tgithub.com/m3db/prometheus_procfs v0.8.1 // indirect\n\tgithub.com/mailru/easyjson v0.7.7 // indirect\n\tgithub.com/mattn/go-colorable v0.1.9 // indirect\n\tgithub.com/mattn/go-isatty v0.0.14 // indirect\n\tgithub.com/mattn/go-runewidth v0.0.7 // indirect\n\tgithub.com/mattn/go-sqlite3 v1.11.0 // indirect\n\tgithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect\n\tgithub.com/pkg/errors v0.9.1 // indirect\n\tgithub.com/pmezard/go-difflib v1.0.0 // indirect\n\tgithub.com/prometheus/client_golang v1.11.1 // indirect\n\tgithub.com/prometheus/client_model v0.4.0 // indirect\n\tgithub.com/prometheus/common v0.26.0 // indirect\n\tgithub.com/prometheus/procfs v0.6.0 // indirect\n\tgithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475 // indirect\n\tgithub.com/russross/blackfriday/v2 v2.1.0 // indirect\n\tgithub.com/stretchr/objx v0.5.2 // indirect\n\tgithub.com/uber-common/bark v1.2.1 // indirect\n\tgithub.com/uber-go/mapdecode v1.0.0 // indirect\n\tgithub.com/xdg-go/pbkdf2 v1.0.0 // indirect\n\tgithub.com/xdg-go/scram v1.1.1 // indirect\n\tgithub.com/xdg-go/stringprep v1.0.3 // indirect\n\tgithub.com/xdg/stringprep v1.0.0 // indirect\n\tgithub.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect\n\tgo.uber.org/dig v1.18.0 // indirect\n\tgo.uber.org/goleak v1.2.0\n\tgo.uber.org/net/metrics v1.3.0 // indirect\n\tgolang.org/x/crypto v0.38.0 // indirect\n\tgolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e // indirect\n\tgolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 // indirect\n\tgolang.org/x/mod v0.18.0\n\tgolang.org/x/sys v0.33.0 // indirect\n\tgolang.org/x/text v0.25.0 // indirect\n\tgoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 // indirect\n\tgoogle.golang.org/protobuf v1.33.0 // indirect\n\tgopkg.in/inf.v0 v0.9.1 // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n\thonnef.co/go/tools v0.3.2 // indirect\n)\n\n// ringpop-go and tchannel-go depends on older version of thrift, yarpc brings up newer version\nreplace github.com/apache/thrift => github.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7\n\n// DO NOT USE as it misses mysql/config store fix\nretract v1.2.3\n"
  },
  {
    "path": "go.sum",
    "content": "cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/BurntSushi/toml v0.4.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8=\ngithub.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/IBM/sarama v1.45.2 h1:8m8LcMCu3REcwpa7fCP6v2fuPuzVwXDAM2DOv3CBrKw=\ngithub.com/IBM/sarama v1.45.2/go.mod h1:ppaoTcVdGv186/z6MEKsMm70A5fwJfRTpstI37kVn3Y=\ngithub.com/Masterminds/semver/v3 v3.2.1 h1:RN9w6+7QoMeJVGyfmbcgs28Br8cvmnucEXnY0rYXWg0=\ngithub.com/Masterminds/semver/v3 v3.2.1/go.mod h1:qvl/7zhW3nngYb5+80sSMF+FG2BjYrf8m9wsX0PNOMQ=\ngithub.com/MicahParks/keyfunc/v2 v2.1.0 h1:6ZXKb9Rp6qp1bDbJefnG7cTH8yMN1IC/4nf+GVjO99k=\ngithub.com/MicahParks/keyfunc/v2 v2.1.0/go.mod h1:rW42fi+xgLJ2FRRXAfNx9ZA8WpD4OeE/yHVMteCkw9k=\ngithub.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=\ngithub.com/VividCortex/mysqlerr v1.0.0 h1:5pZ2TZA+YnzPgzBfiUWGqWmKDVNBdrkf9g+DNe1Tiq8=\ngithub.com/VividCortex/mysqlerr v1.0.0/go.mod h1:xERx8E4tBhLvpjzdUyQiSfUxeMcATEQrflDAfXsqcAE=\ngithub.com/ajstarks/svgo v0.0.0-20180226025133-644b8db467af/go.mod h1:K08gAheRH3/J6wwsYMMT4xOr94bZjxIelGM0+d/wbFw=\ngithub.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 h1:kFOfPq6dUM1hTo4JG6LR5AXSUEsOjtdm0kw0FtQtMJA=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c=\ngithub.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7 h1:Fv9bK1Q+ly/ROk4aJsVMeuIwPel4bEnD8EPiI91nZMg=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=\ngithub.com/aws/aws-sdk-go v1.34.13/go.mod h1:5zCpMtNQVjRREroY7sYe8lOMRSxkhG6MZveU8YkpAk0=\ngithub.com/aws/aws-sdk-go v1.54.12 h1:xPDB+GSBZq0rJbmDZF+EyfMbnWRyfEPcn7PZ7bJjXSw=\ngithub.com/aws/aws-sdk-go v1.54.12/go.mod h1:eRwEWoyTWFMVYVQzKMNHWP5/RV4xIUGMQfXQHfHkpNU=\ngithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3 h1:wOysYcIdqv3WnvwqFFzrYCFALPED7qkUGaLXu359GSc=\ngithub.com/benbjohnson/clock v0.0.0-20161215174838-7dc76406b6d3/go.mod h1:UMqtWQTnOe4byzwe7Zhwh8f8s+36uszN51sJrSIZlTE=\ngithub.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=\ngithub.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=\ngithub.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=\ngithub.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=\ngithub.com/bitly/go-hostpool v0.0.0-20171023180738-a3a6125de932 h1:mXoPYz/Ul5HYEDvkta6I8/rnYM5gSdSV2tJ6XbZuEtY=\ngithub.com/bitly/go-hostpool v0.0.0-20171023180738-a3a6125de932/go.mod h1:NOuUCSz6Q9T7+igc/hlvDOUdtWKryOrtFyIVABv/p7k=\ngithub.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869 h1:DDGfHa7BWjL4YnC6+E63dPcxHo2sUxDIu8g3QgEJdRY=\ngithub.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869/go.mod h1:Ekp36dRnpXw/yCqJaO+ZrUyxD+3VXMFFr56k5XYrpB4=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b h1:AP/Y7sqYicnjGDfD5VcY4CIfh1hRXBUavxrvELjTiOE=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b/go.mod h1:ac9efd0D1fsDb3EJvhqgXRbFx7bs2wqZ10HQPeU8U/Q=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 h1:bXxS5/Z3/dfc8iFniQfgogNBomo0u+1//9eP+jl8GVo=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748/go.mod h1:l/bIBLeOl9eX+wxJAzxS4TveKRtAqlyDpHjhkfO0MEI=\ngithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9 h1:2rukpuvOpZryti4j58JHH5f0qJXxYdTYpkgNYx8iLdg=\ngithub.com/cch123/elasticsql v0.0.0-20190321073543-a1a440758eb9/go.mod h1:h4Tt1A91nOVAYsWdoxlXwKYPfxkxeTuRFkEMUQaRVBo=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=\ngithub.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=\ngithub.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=\ngithub.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk=\ngithub.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210805033703-aa0b78936158/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI=\ngithub.com/coreos/go-semver v0.3.0 h1:wkHLiw0WNATZnSG7epLsujiMCgPAc9xhjJ4tgnAxmfM=\ngithub.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=\ngithub.com/coreos/go-systemd/v22 v22.3.2 h1:D9/bQk5vlXQFZ6Kwuu6zaiXJ9oTPe68++AzAJc1DzSI=\ngithub.com/coreos/go-systemd/v22 v22.3.2/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4 h1:wfIWP927BUkWJb2NmU/kNDYIBTh/ziUX91+lVfRxZq4=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=\ngithub.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=\ngithub.com/cristalhq/jwt/v3 v3.1.0 h1:iLeL9VzB0SCtjCy9Kg53rMwTcrNm+GHyVcz2eUujz6s=\ngithub.com/cristalhq/jwt/v3 v3.1.0/go.mod h1:XOnIXst8ozq/esy5N1XOlSyQqBd+84fxJ99FK+1jgL8=\ngithub.com/dave/dst v0.26.2 h1:lnxLAKI3tx7MgLNVDirFCsDTlTG9nKTk7GcptKcWSwY=\ngithub.com/dave/dst v0.26.2/go.mod h1:UMDJuIRPfyUCC78eFuB+SV/WI8oDeyFDvM/JR6NI3IU=\ngithub.com/dave/gopackages v0.0.0-20170318123100-46e7023ec56e/go.mod h1:i00+b/gKdIDIxuLDFob7ustLAVqhsZRk2qVZrArELGQ=\ngithub.com/dave/jennifer v1.2.0/go.mod h1:fIb+770HOpJ2fmN9EPPKOqm1vMGhB+TwXKMZhrIygKg=\ngithub.com/dave/kerr v0.0.0-20170318121727-bc25dd6abe8e/go.mod h1:qZqlPyPvfsDJt+3wHJ1EvSXDuVjFTK0j2p/ca+gtsb8=\ngithub.com/dave/rebecca v0.9.1/go.mod h1:N6XYdMD/OKw3lkF3ywh8Z6wPGuwNFDNtWYEMFWEmXBA=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/dgryski/go-farm v0.0.0-20200201041132-a6ae2369ad13/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da h1:aIftn67I1fkbMa512G+w+Pxci9hJPB8oMnkcP3iZF38=\ngithub.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=\ngithub.com/eapache/go-resiliency v1.7.0 h1:n3NRTnBn5N0Cbi/IeOHuQn9s2UwVUH7Ga0ZWcP+9JTA=\ngithub.com/eapache/go-resiliency v1.7.0/go.mod h1:5yPzW0MIvSe0JDsv0v+DvcjEv2FyD6iZYSs1ZI+iQho=\ngithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3 h1:Oy0F4ALJ04o5Qqpdz8XLIpNA3WM/iSIXqxtqo7UGVws=\ngithub.com/eapache/go-xerial-snappy v0.0.0-20230731223053-c322873962e3/go.mod h1:YvSRo5mw33fLEx1+DlK6L2VV43tJt5Eyel9n9XBcR+0=\ngithub.com/eapache/queue v1.1.0 h1:YOEu7KNc61ntiQlcEeUIoDTJ2o8mQznoNvUhiigpIqc=\ngithub.com/eapache/queue v1.1.0/go.mod h1:6eCeP0CKFpHLu8blIFXhExK/dRa7WDZfr6jVFPTqq+I=\ngithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab h1:eTc1vwMHNg4WtS95PtYi3FFCKwlPjtN/Lw9IALTRtd8=\ngithub.com/emirpasic/gods v0.0.0-20190624094223-e689965507ab/go.mod h1:YfzfFFoVP/catgzJb4IKIqXjX78Ha8FMSDh3ymbK86o=\ngithub.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210217033140-668b12f5399d/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.mod h1:hliV/p42l8fGbc6Y9bQ70uLwIvmJyVE5k4iMKlh8wCQ=\ngithub.com/envoyproxy/go-control-plane v0.9.10-0.20210907150352-cf90f659a021/go.mod h1:AFq3mo9L8Lqqiid3OhADV3RfLJnjiw63cSpi+fDTRC0=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a h1:yDWHCSQ40h88yih2JAcL6Ls/kVkSE8GFACTGVnMPruw=\ngithub.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a/go.mod h1:7Ga40egUymuWXxAe151lTNnCv97MddSOVsjpPPkityA=\ngithub.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w=\ngithub.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk=\ngithub.com/fatih/structtag v1.0.0/go.mod h1:IKitwq45uXL/yqi5mYghiD3w9H6eTOvI9vnk8tXMphA=\ngithub.com/fatih/structtag v1.2.0 h1:/OdNE99OxoI/PqaW/SuSK9uxxT3f/tcSZgon/ssNSx4=\ngithub.com/fatih/structtag v1.2.0/go.mod h1:mBJUNpUnHmRKrKlQQlmCrh5PuhftFbNv8Ys4/aAZl94=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568 h1:BHsljHzVlRcyQhjrss6TZTdY2VfCqZPbv5k3iBFa2ZQ=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568/go.mod h1:xEzjJPgXI435gkrCt3MPfRiAkVrwSbHsst4LCFVfpJc=\ngithub.com/fogleman/gg v1.2.1-0.20190220221249-0403632d5b90/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k=\ngithub.com/fortytw2/leaktest v1.3.0 h1:u8491cBMTQ8ft8aeV+adlcytMZylmA5nnwwkRZjI8vw=\ngithub.com/fortytw2/leaktest v1.3.0/go.mod h1:jDsjWgpAGjm2CA7WthBh/CdZYEPF31XHquHwclZch5g=\ngithub.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=\ngithub.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=\ngithub.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=\ngithub.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=\ngithub.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=\ngithub.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=\ngithub.com/go-sql-driver/mysql v1.7.1 h1:lUIinVbN1DY0xBg0eMOzmmtGoHwWBbvnWubQUrtU8EI=\ngithub.com/go-sql-driver/mysql v1.7.1/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=\ngithub.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=\ngithub.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=\ngithub.com/go-zookeeper/zk v1.0.3 h1:7M2kwOsc//9VeeFiPtf+uSJlVpU66x9Ba5+8XK7/TDg=\ngithub.com/go-zookeeper/zk v1.0.3/go.mod h1:nOB03cncLtlp4t+UAkGSV+9beXP/akpekBwL+UX1Qcw=\ngithub.com/gobuffalo/attrs v0.0.0-20190224210810-a9411de4debd/go.mod h1:4duuawTqi2wkkpB4ePgWMaai6/Kc6WEz83bhFwpHzj0=\ngithub.com/gobuffalo/depgen v0.0.0-20190329151759-d478694a28d3/go.mod h1:3STtPUQYuzV0gBVOY3vy6CfMm/ljR4pABfrTeHNLHUY=\ngithub.com/gobuffalo/depgen v0.1.0/go.mod h1:+ifsuy7fhi15RWncXQQKjWS9JPkdah5sZvtHc2RXGlg=\ngithub.com/gobuffalo/envy v1.6.15/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI=\ngithub.com/gobuffalo/envy v1.7.0/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI=\ngithub.com/gobuffalo/flect v0.1.0/go.mod h1:d2ehjJqGOH/Kjqcoz+F7jHTBbmDb38yXA598Hb50EGs=\ngithub.com/gobuffalo/flect v0.1.1/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI=\ngithub.com/gobuffalo/flect v0.1.3/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI=\ngithub.com/gobuffalo/genny v0.0.0-20190329151137-27723ad26ef9/go.mod h1:rWs4Z12d1Zbf19rlsn0nurr75KqhYp52EAGGxTbBhNk=\ngithub.com/gobuffalo/genny v0.0.0-20190403191548-3ca520ef0d9e/go.mod h1:80lIj3kVJWwOrXWWMRzzdhW3DsrdjILVil/SFKBzF28=\ngithub.com/gobuffalo/genny v0.1.0/go.mod h1:XidbUqzak3lHdS//TPu2OgiFB+51Ur5f7CSnXZ/JDvo=\ngithub.com/gobuffalo/genny v0.1.1/go.mod h1:5TExbEyY48pfunL4QSXxlDOmdsD44RRq4mVZ0Ex28Xk=\ngithub.com/gobuffalo/gitgen v0.0.0-20190315122116-cc086187d211/go.mod h1:vEHJk/E9DmhejeLeNt7UVvlSGv3ziL+djtTr3yyzcOw=\ngithub.com/gobuffalo/gogen v0.0.0-20190315121717-8f38393713f5/go.mod h1:V9QVDIxsgKNZs6L2IYiGR8datgMhB577vzTDqypH360=\ngithub.com/gobuffalo/gogen v0.1.0/go.mod h1:8NTelM5qd8RZ15VjQTFkAW6qOMx5wBbW4dSCS3BY8gg=\ngithub.com/gobuffalo/gogen v0.1.1/go.mod h1:y8iBtmHmGc4qa3urIyo1shvOD8JftTtfcKi+71xfDNE=\ngithub.com/gobuffalo/logger v0.0.0-20190315122211-86e12af44bc2/go.mod h1:QdxcLw541hSGtBnhUc4gaNIXRjiDppFGaDqzbrBd3v8=\ngithub.com/gobuffalo/mapi v1.0.1/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc=\ngithub.com/gobuffalo/mapi v1.0.2/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc=\ngithub.com/gobuffalo/packd v0.0.0-20190315124812-a385830c7fc0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4=\ngithub.com/gobuffalo/packd v0.1.0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4=\ngithub.com/gobuffalo/packr/v2 v2.0.9/go.mod h1:emmyGweYTm6Kdper+iywB6YK5YzuKchGtJQZ0Odn4pQ=\ngithub.com/gobuffalo/packr/v2 v2.2.0/go.mod h1:CaAwI0GPIAv+5wKLtv8Afwl+Cm78K/I/VCm/3ptBN+0=\ngithub.com/gobuffalo/syncx v0.0.0-20190224160051-33c29581e754/go.mod h1:HhnNqWY95UYwwW3uSASeV7vtgYkT2t16hJgV3AEPUpw=\ngithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1 h1:px9qUCy/RNJNsfCam4m2IxWGxNuimkrioEF0vrrbPsg=\ngithub.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1/go.mod h1:3gM2c4D3AnkISwBxGnMMsS8Oy4y2lhbPRsH4xnJrHG8=\ngithub.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=\ngithub.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a/go.mod h1:gf4bu3Q80BeJ6H1S1vYPm8/ELATdvryBaNFGgqEef3s=\ngithub.com/gogo/googleapis v1.3.2 h1:kX1es4djPJrsDhY7aZKJy7aZasdcB5oSOEphMjSB53c=\ngithub.com/gogo/googleapis v1.3.2/go.mod h1:5YRNX2z1oM5gXdAkurHa942MDgEJyk02w4OecKY87+c=\ngithub.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.2.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.3.1/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=\ngithub.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=\ngithub.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=\ngithub.com/gogo/status v1.1.0 h1:+eIkrewn5q6b30y+g/BJINVVdi2xH7je5MPJ3ZPK3JA=\ngithub.com/gogo/status v1.1.0/go.mod h1:BFv9nrluPLmrS0EmGVvLaPNmRosr9KapBYd5/hpY1WM=\ngithub.com/golang-jwt/jwt/v5 v5.2.0 h1:d/ix8ftRUorsN+5eMIlF4T6J8CAt9rch3My2winC1Jw=\ngithub.com/golang-jwt/jwt/v5 v5.2.0/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=\ngithub.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.4.0/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=\ngithub.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=\ngithub.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.3-0.20190920234318-1680a479a2cf/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=\ngithub.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=\ngithub.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=\ngithub.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=\ngithub.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=\ngithub.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=\ngithub.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=\ngithub.com/google/gofuzz v1.0.0 h1:A8PeW59pxE9IoFRqBp37U+mSNaQoZ46F1f0f863XSXw=\ngithub.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=\ngithub.com/google/pprof v0.0.0-20181127221834-b4f47329b966/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=\ngithub.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=\ngithub.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/gorilla/securecookie v1.1.1/go.mod h1:ra0sb63/xPlUeL+yeDciTfxMRAA+MP+HVt/4epWDjd4=\ngithub.com/gorilla/sessions v1.2.1/go.mod h1:dk2InVEVJ0sfLlnXv9EAgkf6ecYs/i80K/zI+bUmuGM=\ngithub.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw=\ngithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed h1:5upAirOpQc1Q53c0bnx2ufif5kANL7bfZWcc6VJWJd8=\ngithub.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed/go.mod h1:tMWxXQ9wFIaZeTI9F+hmhFiGpFmhOHzyShyFUhRm0H4=\ngithub.com/hashicorp/errwrap v1.0.0 h1:hLrqtEDnRye3+sgx6z4qVLNuviH3MR5aQ0ykNJa/UYA=\ngithub.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=\ngithub.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=\ngithub.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=\ngithub.com/hashicorp/go-uuid v1.0.2/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-uuid v1.0.3 h1:2gKiV6YVmrJ1i2CKKa9obLvRieoRGviZFL26PcT/Co8=\ngithub.com/hashicorp/go-uuid v1.0.3/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-version v1.2.0 h1:3vNe/fWF5CBgRIguda1meWhsZHy3m8gCJ5wx+dIzX/E=\ngithub.com/hashicorp/go-version v1.2.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=\ngithub.com/iancoleman/strcase v0.2.0 h1:05I4QRnGpI0m37iZQRuskXh+w77mr6Z41lwQzuHLwW0=\ngithub.com/iancoleman/strcase v0.2.0/go.mod h1:iwCmte+B7n89clKwxIoIXy/HfoL7AsD47ZCWhYzw7ho=\ngithub.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=\ngithub.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=\ngithub.com/jcmturner/aescts/v2 v2.0.0 h1:9YKLH6ey7H4eDBXW8khjYslgyqG2xZikXP0EQFKrle8=\ngithub.com/jcmturner/aescts/v2 v2.0.0/go.mod h1:AiaICIRyfYg35RUkr8yESTqvSy7csK90qZ5xfvvsoNs=\ngithub.com/jcmturner/dnsutils/v2 v2.0.0 h1:lltnkeZGL0wILNvrNiVCR6Ro5PGU/SeBvVO/8c/iPbo=\ngithub.com/jcmturner/dnsutils/v2 v2.0.0/go.mod h1:b0TnjGOvI/n42bZa+hmXL+kFJZsFT7G4t3HTlQ184QM=\ngithub.com/jcmturner/gofork v1.7.6 h1:QH0l3hzAU1tfT3rZCnW5zXl+orbkNMMRGJfdJjHVETg=\ngithub.com/jcmturner/gofork v1.7.6/go.mod h1:1622LH6i/EZqLloHfE7IeZ0uEJwMSUyQ/nDd82IeqRo=\ngithub.com/jcmturner/goidentity/v6 v6.0.1 h1:VKnZd2oEIMorCTsFBnJWbExfNN7yZr3EhJAxwOkZg6o=\ngithub.com/jcmturner/goidentity/v6 v6.0.1/go.mod h1:X1YW3bgtvwAXju7V3LCIMpY0Gbxyjn/mY9zx4tFonSg=\ngithub.com/jcmturner/gokrb5/v8 v8.4.4 h1:x1Sv4HaTpepFkXbt2IkL29DXRf8sOfZXo8eRKh687T8=\ngithub.com/jcmturner/gokrb5/v8 v8.4.4/go.mod h1:1btQEpgT6k+unzCwX1KdWMEwPPkkgBtP+F6aCACiMrs=\ngithub.com/jcmturner/rpc/v2 v2.0.3 h1:7FXXj8Ti1IaVFpSAziCZWNzbNuZmnvw/i6CqLNdWfZY=\ngithub.com/jcmturner/rpc/v2 v2.0.3/go.mod h1:VUJYCIDm3PVOEHw8sgt091/20OJjskO/YJki3ELg/Hc=\ngithub.com/jessevdk/go-flags v1.4.0 h1:4IU2WS7AumrZ/40jfhf4QVDMsQwqA7VEHozFRrGARJA=\ngithub.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=\ngithub.com/jmespath/go-jmespath v0.3.0/go.mod h1:9QtRXoHjLGCJ5IBSaohpXITPlowMeeYCZ7fLUTSywik=\ngithub.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=\ngithub.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=\ngithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee h1:59lyMGvZusByi7Rvctn8cxdVAjhiOnqCv3G5DrYApYQ=\ngithub.com/jmoiron/sqlx v1.2.1-0.20200615141059-0794cb1f47ee/go.mod h1:ClpsPFzLpSBl7MvJ+BhV0JHz4vmKRBarpvZ9644v9Oo=\ngithub.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg=\ngithub.com/jonboulle/clockwork v0.5.0 h1:Hyh9A8u51kptdkR+cqRpT1EebBwTn1oK9YfGYbdFz6I=\ngithub.com/jonboulle/clockwork v0.5.0/go.mod h1:3mZlmanh0g2NDKO5TWZVJAfofYk64M7XN3SzBPjZF60=\ngithub.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY=\ngithub.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=\ngithub.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=\ngithub.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=\ngithub.com/json-iterator/go v1.1.8/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.11/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=\ngithub.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=\ngithub.com/jung-kurt/gofpdf v1.0.3-0.20190309125859-24315acbbda5/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes=\ngithub.com/karrick/godirwalk v1.8.0/go.mod h1:H5KPZjojv4lE+QYImBI8xVtrBRgYrIVsaRPx4tDPEn4=\ngithub.com/karrick/godirwalk v1.10.3/go.mod h1:RoGL9dQei4vP9ilrpETWE8CLOZ1kiN0LhBygSwrAsHA=\ngithub.com/kisielk/errcheck v1.2.0/go.mod h1:/BMXB+zMLi60iA8Vv6Ksmxu/1UDYcXs4uQLJ+jE2L00=\ngithub.com/kisielk/errcheck v1.5.0 h1:e8esj/e4R+SAOwFwN+n3zr0nYeCyeweozKfO23MvHzY=\ngithub.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=\ngithub.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=\ngithub.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=\ngithub.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=\ngithub.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=\ngithub.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=\ngithub.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=\ngithub.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=\ngithub.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=\ngithub.com/lib/pq v1.2.0 h1:LXpIM/LZ5xGFhOpXAQUIMM1HdyqzVYM13zNdjCEEcA0=\ngithub.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=\ngithub.com/m3db/prometheus_client_golang v0.8.1 h1:t7w/tcFws81JL1j5sqmpqcOyQOpH4RDOmIe3A3fdN3w=\ngithub.com/m3db/prometheus_client_golang v0.8.1/go.mod h1:8R/f1xYhXWq59KD/mbRqoBulXejss7vYtYzWmruNUwI=\ngithub.com/m3db/prometheus_client_model v0.1.0 h1:cg1+DiuyT6x8h9voibtarkH1KT6CmsewBSaBhe8wzLo=\ngithub.com/m3db/prometheus_client_model v0.1.0/go.mod h1:Qfsxn+LypxzF+lNhak7cF7k0zxK7uB/ynGYoj80zcD4=\ngithub.com/m3db/prometheus_common v0.1.0 h1:YJu6eCIV6MQlcwND24cRG/aRkZDX1jvYbsNNs1ZYr0w=\ngithub.com/m3db/prometheus_common v0.1.0/go.mod h1:EBmDQaMAy4B8i+qsg1wMXAelLNVbp49i/JOeVszQ/rs=\ngithub.com/m3db/prometheus_procfs v0.8.1 h1:LsxWzVELhDU9sLsZTaFLCeAwCn7bC7qecZcK4zobs/g=\ngithub.com/m3db/prometheus_procfs v0.8.1/go.mod h1:N8lv8fLh3U3koZx1Bnisj60GYUMDpWb09x1R+dmMOJo=\ngithub.com/mailru/easyjson v0.7.6/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=\ngithub.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0=\ngithub.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=\ngithub.com/mark3labs/mcp-go v0.18.0 h1:YuhgIVjNlTG2ZOwmrkORWyPTp0dz1opPEqvsPtySXao=\ngithub.com/mark3labs/mcp-go v0.18.0/go.mod h1:KmJndYv7GIgcPVwEKJjNcbhVQ+hJGJhrCCB/9xITzpE=\ngithub.com/markbates/oncer v0.0.0-20181203154359-bf2de49a0be2/go.mod h1:Ld9puTsIW75CHf65OeIOkyKbteujpZVXDpWK6YGZbxE=\ngithub.com/markbates/safe v1.0.1/go.mod h1:nAqgmRi7cY2nqMc92/bSEeQA+R4OheNU2T1kNSCBdG0=\ngithub.com/mattn/go-colorable v0.1.9 h1:sqDoxXbdeALODt0DAeJCVp38ps9ZogZEAXjus69YV3U=\ngithub.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=\ngithub.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=\ngithub.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=\ngithub.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=\ngithub.com/mattn/go-runewidth v0.0.7 h1:Ei8KR0497xHyKJPAv59M1dkC+rOZCMBJ+t3fZ+twI54=\ngithub.com/mattn/go-runewidth v0.0.7/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=\ngithub.com/mattn/go-shellwords v1.0.10/go.mod h1:EZzvwXDESEeg03EKmM+RmDnNOPKG4lLtQsUlTZDWQ8Y=\ngithub.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=\ngithub.com/mattn/go-sqlite3 v1.11.0 h1:LDdKkqtYlom37fkvqs8rMPFKAMe8+SgjbwZ6ex1/A/Q=\ngithub.com/mattn/go-sqlite3 v1.11.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 h1:I0XW9+e1XWDxdcEniV4rQAIOPUGDq67JSCiRCgGCZLI=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=\ngithub.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=\ngithub.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/ncruces/go-sqlite3 v0.22.0 h1:FkGSBhd0TY6e66k1LVhyEpA+RnG/8QkQNed5pjIk4cs=\ngithub.com/ncruces/go-sqlite3 v0.22.0/go.mod h1:ueXOZXYZS2OFQirCU3mHneDwJm5fGKHrtccYBeGEV7M=\ngithub.com/ncruces/julianday v1.0.0 h1:fH0OKwa7NWvniGQtxdJRxAgkBMolni2BjDHaWTxqt7M=\ngithub.com/ncruces/julianday v1.0.0/go.mod h1:Dusn2KvZrrovOMJuOt0TNXL6tB7U2E8kvza5fFc9G7g=\ngithub.com/olekukonko/tablewriter v0.0.4 h1:vHD/YYe1Wolo78koG299f7V/VAS08c6IpCLn+Ejf/w8=\ngithub.com/olekukonko/tablewriter v0.0.4/go.mod h1:zq6QwlOf5SlnkVbMSr5EoBv3636FWnp+qbPhuoO21uA=\ngithub.com/olivere/elastic v6.2.37+incompatible h1:UfSGJem5czY+x/LqxgeCBgjDn6St+z8OnsCuxwD3L0U=\ngithub.com/olivere/elastic v6.2.37+incompatible/go.mod h1:J+q1zQJTgAz9woqsbVRqGeB5G1iqDKVBWLNSYW8yfJ8=\ngithub.com/olivere/elastic/v7 v7.0.21 h1:58a2pMlLketCsLyKg8kJNJG+OZIFKrSQXX6gJBpqqlg=\ngithub.com/olivere/elastic/v7 v7.0.21/go.mod h1:Kh7iIsXIBl5qRQOBFoylCsXVTtye3keQU2Y/YbR7HD8=\ngithub.com/opensearch-project/opensearch-go/v4 v4.1.0 h1:YXNaMpMU0PC7suGyP13EuczkDT3K54QajgDnLKCZAz8=\ngithub.com/opensearch-project/opensearch-go/v4 v4.1.0/go.mod h1:aSTMFGSLEoiG19US6Oo5udvWCjHap3mRcWBNV8rAFak=\ngithub.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=\ngithub.com/opentracing/opentracing-go v1.2.0 h1:uEJPy/1a5RIPAJ0Ov+OIO8OxWu77jEv+1B0VhjKrZUs=\ngithub.com/opentracing/opentracing-go v1.2.0/go.mod h1:GxEUsuufX4nBwe+T+Wl9TAgYrxe9dPLANfrWvHYVTgc=\ngithub.com/otiai10/copy v1.1.1 h1:PH7IFlRQ6Fv9vYmuXbDRLdgTHoP1w483kPNUP2bskpo=\ngithub.com/otiai10/copy v1.1.1/go.mod h1:rrF5dJ5F0t/EWSYODDu4j9/vEeYHMkc8jt0zJChqQWw=\ngithub.com/otiai10/curr v0.0.0-20150429015615-9b4961190c95/go.mod h1:9qAhocn7zKJG+0mI8eUu6xqkFDYS2kb2saOteoSB3cE=\ngithub.com/otiai10/curr v1.0.0/go.mod h1:LskTG5wDwr8Rs+nNQ+1LlxRjAtTZZjtJW4rMXl6j4vs=\ngithub.com/otiai10/mint v1.3.0/go.mod h1:F5AjcsTsWUqX+Na9fpHb52P8pcRX2CI6A3ctIT91xUo=\ngithub.com/otiai10/mint v1.3.1 h1:BCmzIS3n71sGfHB5NMNDB3lHYPz8fWSkCAErHed//qc=\ngithub.com/otiai10/mint v1.3.1/go.mod h1:/yxELlJQ0ufhjUwhshSj+wFjZ78CnZ48/1wtmBH1OTc=\ngithub.com/pborman/uuid v0.0.0-20160209185913-a97ce2ca70fa/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709 h1:zNBQb37RGLmJybyMcs983HfUfpkw9OTFD9tbBfAViHE=\ngithub.com/pborman/uuid v0.0.0-20180906182336-adf5a7427709/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34=\ngithub.com/pelletier/go-toml v1.7.0/go.mod h1:vwGMzjaWMwyfHwgIBhI2YUM4fB6nL6lVAvS1LBMMhTE=\ngithub.com/pierrec/lz4/v4 v4.1.22 h1:cKFw6uJDK+/gfw5BcDL0JL5aBsAFdsIT18eRtLj7VIU=\ngithub.com/pierrec/lz4/v4 v4.1.22/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=\ngithub.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a h1:AA9vgIBDjMHPC2McaGPojgV2dcI78ZC0TLNhYCXEKH8=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a/go.mod h1:lzZQ3Noex5pfAy7mkAeCjcBDteYU85uWWnJ/y6gKU8k=\ngithub.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=\ngithub.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo=\ngithub.com/prometheus/client_golang v1.3.0/go.mod h1:hJaj2vgQTGQmVCsAACORcieXFeDPbaTKGT+JTgUa3og=\ngithub.com/prometheus/client_golang v1.4.1/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=\ngithub.com/prometheus/client_golang v1.7.1/go.mod h1:PY5Wy2awLA44sXw4AOSfFBetzPP4j5+D6mVACh+pe2M=\ngithub.com/prometheus/client_golang v1.11.1 h1:+4eQaD7vAZ6DsfsxB15hbE0odUjGI5ARs9yskGu1v4s=\ngithub.com/prometheus/client_golang v1.11.1/go.mod h1:Z6t4BnS23TR94PD6BsDNk8yVqroYurpAkEiz0P2BEV0=\ngithub.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=\ngithub.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.1.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.4.0 h1:5lQXD3cAg1OXBf4Wq03gTrXHeaV0TQvGfUooCfx1yqY=\ngithub.com/prometheus/client_model v0.4.0/go.mod h1:oMQmHW1/JoDwqLtg57MGgP/Fb1CJEYF2imWWhWtMkYU=\ngithub.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=\ngithub.com/prometheus/common v0.7.0/go.mod h1:DjGbpBbp5NYNiECxcL/VnbXCCaQpKd3tt26CguLLsqA=\ngithub.com/prometheus/common v0.8.0/go.mod h1:PC/OgXc+UN7B4ALwvn1yzVZmVwvhXp5JsbBv6wSv6i0=\ngithub.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4=\ngithub.com/prometheus/common v0.10.0/go.mod h1:Tlit/dnDKsSWFlCLTWaA1cyBgKHSMdTB80sz/V91rCo=\ngithub.com/prometheus/common v0.26.0 h1:iMAkS2TDoNWnKM+Kopnx/8tnEStIfpYA0ur0xQzzhMQ=\ngithub.com/prometheus/common v0.26.0/go.mod h1:M7rCNAaPfAosfx8veZJCuw84e35h3Cfd9VFqTh1DIvc=\ngithub.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=\ngithub.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=\ngithub.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.0.9/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.1.3/go.mod h1:lV6e/gmhEcM9IjHGsFOCxxuZ+z1YqCvr4OA4YeYWdaU=\ngithub.com/prometheus/procfs v0.6.0 h1:mxy4L2jP6qMonqmq+aTtOx1ifVWUgG/TAmntgbh3xv4=\ngithub.com/prometheus/procfs v0.6.0/go.mod h1:cz+aTbrPOrUb4q7XlbU9ygM+/jj0fzG6c1xBZuNvfVA=\ngithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475 h1:N/ElC8H3+5XpJzTSTfLsJV/mx9Q9g7kxmchpfZyxgzM=\ngithub.com/rcrowley/go-metrics v0.0.0-20201227073835-cf1acfcdf475/go.mod h1:bCqnVzQkZxMG4s8nGwiZ5l3QUCyqpo9Y+/ZMZ9VjZe4=\ngithub.com/robfig/cron v1.2.0 h1:ZjScXvvxeQ63Dbyxy76Fj3AT3Ut0aKsyd2/tl3DTMuQ=\ngithub.com/robfig/cron v1.2.0/go.mod h1:JGuDeoQd7Z6yL4zQhZ3OPEVHB7fL6Ka6skscFHfmt2k=\ngithub.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=\ngithub.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=\ngithub.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=\ngithub.com/rogpeppe/go-internal v1.1.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.2.2/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.6.1 h1:/FiVV8dS/e+YqF2JvO3yXRFbBLTIuSDkuC7aBOAvL+k=\ngithub.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=\ngithub.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=\ngithub.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af h1:EiWVfh8mr40yFZEui2oF0d45KgH48PkB2H0Z0GANvSI=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af/go.mod h1:Vrkh1pnjV9Bl8c3P9zH0/D4NlOHWP5d4/hF4YTULaec=\ngithub.com/sergi/go-diff v1.0.0 h1:Kpca3qRNrduNnOQeazBd0ysaKrUJiIuISHxogkT9RPQ=\ngithub.com/sergi/go-diff v1.0.0/go.mod h1:0CfEIISq7TuYL3j771MWULgwwjU+GofnZX9QAmXWZgo=\ngithub.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=\ngithub.com/sirupsen/logrus v1.4.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=\ngithub.com/sirupsen/logrus v1.4.1/go.mod h1:ni0Sbl8bgC9z8RoU9G6nDWqqs/fq4eDPysMBDgk/93Q=\ngithub.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=\ngithub.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=\ngithub.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=\ngithub.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=\ngithub.com/smartystreets/assertions v1.1.1/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=\ngithub.com/smartystreets/go-aws-auth v0.0.0-20180515143844-0c1422d1fdb9/go.mod h1:SnhjPscd9TpLiy1LpzGSKh3bXCfxxXuqd9xmQJy3slM=\ngithub.com/smartystreets/gunit v1.4.2/go.mod h1:ZjM1ozSIMJlAz/ay4SG8PeKF00ckUp+zMHZXV9/bvak=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=\ngithub.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=\ngithub.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=\ngithub.com/startreedata/pinot-client-go v0.2.0 h1:Pv4W3HgxxGbB9GogRwNqfNyqPrOpScZuhQRc9kLM90A=\ngithub.com/startreedata/pinot-client-go v0.2.0/go.mod h1:vTz6Bu4dWIQIsfUoqFtgMV2QqBjeuSaDA8vxkOoYnLg=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25 h1:7z3LSn867ex6VSaahyKadf4WtSsJIgne6A1WLOAGM8A=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25/go.mod h1:lbP8tGiBjZ5YWIc2fzuRpTaz0b/53vT6PEs3QuAWzuU=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=\ngithub.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=\ngithub.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=\ngithub.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=\ngithub.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=\ngithub.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=\ngithub.com/tetratelabs/wazero v1.8.2 h1:yIgLR/b2bN31bjxwXHD8a3d+BogigR952csSDdLYEv4=\ngithub.com/tetratelabs/wazero v1.8.2/go.mod h1:yAI0XTsMBhREkM/YDAK/zNou3GoiAce1P6+rp/wQhjs=\ngithub.com/tidwall/gjson v1.17.1 h1:wlYEnwqAHgzmhNUFfw7Xalt2JzQvsMx2Se4PcoFCT/U=\ngithub.com/tidwall/gjson v1.17.1/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=\ngithub.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=\ngithub.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=\ngithub.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=\ngithub.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4=\ngithub.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=\ngithub.com/tidwall/sjson v1.2.5 h1:kLy8mja+1c9jlljvWTlSazM7cKDRfJuR/bOJhcY5NcY=\ngithub.com/tidwall/sjson v1.2.5/go.mod h1:Fvgq9kS/6ociJEDnK0Fk1cpYF4FIW6ZF7LAe+6jwd28=\ngithub.com/uber-common/bark v1.2.1 h1:cREJ9b7CpTjwZr0/5wV82fXlitoCIEHHnt9WkQ4lIk0=\ngithub.com/uber-common/bark v1.2.1/go.mod h1:g0ZuPcD7XiExKHynr93Q742G/sbrdVQkghrqLGOoFuY=\ngithub.com/uber-go/mapdecode v1.0.0 h1:euUEFM9KnuCa1OBixz1xM+FIXmpixyay5DLymceOVrU=\ngithub.com/uber-go/mapdecode v1.0.0/go.mod h1:b5nP15FwXTgpjTjeA9A2uTHXV5UJCl4arwKpP0FP1Hw=\ngithub.com/uber-go/tally v3.3.12+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber-go/tally v3.3.15+incompatible h1:9hLSgNBP28CjIaDmAuRTq9qV+UZY+9PcvAkXO4nNMwg=\ngithub.com/uber-go/tally v3.3.15+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber/cadence-idl v0.0.0-20211111101836-d6b70b60eb8c/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4 h1:1kAlcf7STXalFhDzv6vUz3XZtGOXQfwDF7IBKHu8GwI=\ngithub.com/uber/cadence-idl v0.0.0-20260313102523-57782092adb4/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible h1:NHcubEkVbahf9t3p75TOCR83gdUHXjRJvjoBh1yACsM=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible/go.mod h1:WVhlPFC8FDjOFMMWRy2pZqQJSXxYSwNYOkTr/Z6d3Kk=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible h1:MxZXOiR2JuoANZ3J6DE/U0kSFv/eJ/GfSYVCjK7dyaw=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible/go.mod h1:ComeNDZlWwrWnDv8aPp0Ba6+uUTzImX/AauajbLI56U=\ngithub.com/uber/ringpop-go v0.8.5 h1:aBa/SHmmFRcAXA63k7uBheoTL8tCmH7L+OgktB1AF/o=\ngithub.com/uber/ringpop-go v0.8.5/go.mod h1:zVI6eGO6L7pG14GkntHsSOfmUAWQ7B4lvmzly4IT4ls=\ngithub.com/uber/tchannel-go v1.16.0/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/uber/tchannel-go v1.22.2 h1:NKA5FVESYh6Ij6V+tujK+IFZnBKDyUHdsBY264UYhgk=\ngithub.com/uber/tchannel-go v1.22.2/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/urfave/cli/v2 v2.27.4 h1:o1owoI+02Eb+K107p27wEX9Bb8eqIoZCfLXloLUSWJ8=\ngithub.com/urfave/cli/v2 v2.27.4/go.mod h1:m4QzxcD2qpra4z7WhzEGn74WZLViBnMpb1ToCAKdGRQ=\ngithub.com/valyala/fastjson v1.4.1 h1:hrltpHpIpkaxll8QltMU8c3QZ5+qIiCL8yKqPFJI/yE=\ngithub.com/valyala/fastjson v1.4.1/go.mod h1:nV6MsjxL2IMJQUoHDIrjEI7oLyeqK6aBD7EFWPsvP8o=\ngithub.com/wI2L/jsondiff v0.6.0 h1:zrsH3FbfVa3JO9llxrcDy/XLkYPLgoMX6Mz3T2PP2AI=\ngithub.com/wI2L/jsondiff v0.6.0/go.mod h1:D6aQ5gKgPF9g17j+E9N7aasmU1O+XvfmWm1y8UMmNpw=\ngithub.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=\ngithub.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=\ngithub.com/xdg-go/scram v1.0.2/go.mod h1:1WAq6h33pAW+iRreB34OORO2Nf7qel3VV3fjBj+hCSs=\ngithub.com/xdg-go/scram v1.1.1 h1:VOMT+81stJgXW3CpHyqHN3AXDYIMsx56mEFrB37Mb/E=\ngithub.com/xdg-go/scram v1.1.1/go.mod h1:RaEWvsqvNKKvBPvcKeFjrG2cJqOkHTiyTpzz23ni57g=\ngithub.com/xdg-go/stringprep v1.0.2/go.mod h1:8F9zXuvzgwmyT5DUm4GUfZGDdT3W+LCvS6+da4O5kxM=\ngithub.com/xdg-go/stringprep v1.0.3 h1:kdwGpVNwPFtjs98xCGkHjQtGKh86rDcRZN17QEMCOIs=\ngithub.com/xdg-go/stringprep v1.0.3/go.mod h1:W3f5j4i+9rC0kuIEJL0ky1VpHXQU3ocBgklLGvcBnW8=\ngithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c h1:u40Z8hqBAAQyv+vATcGgV0YCnDjqSL7/q/JyPhhJSPk=\ngithub.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c/go.mod h1:lB8K/P019DLNhemzwFU4jHLhdvlE6uDZjXFejJXr49I=\ngithub.com/xdg/stringprep v1.0.0 h1:d9X0esnoa3dFsV0FG35rAT0RIhYFlPq7MiP+DW89La0=\ngithub.com/xdg/stringprep v1.0.0/go.mod h1:Jhud4/sHMO4oL310DaZAKk9ZaJ08SJfe+sJh0HrGL1Y=\ngithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 h1:gEOO8jv9F4OT7lGCjxCBTO/36wtF6j2nSip77qHd4x4=\ngithub.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1/go.mod h1:Ohn+xnUBiLI6FVj/9LpzZWtj1/D6lUovWYBkxHVV3aM=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2 h1:zzrxE1FKn5ryBNl9eKOeqQ58Y/Qpo3Q9QNxKHX5uzzQ=\ngithub.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2/go.mod h1:hzfGeIUDq/j97IG+FhNqkowIyEcD88LrW6fyU3K3WqY=\ngithub.com/yosida95/uritemplate/v3 v3.0.2 h1:Ed3Oyj9yrmi9087+NczuL5BwkIc4wvTb5zIM+UJPGz4=\ngithub.com/yosida95/uritemplate/v3 v3.0.2/go.mod h1:ILOh0sOhIJR3+L/8afwt/kE++YT040gmv5BQTMR2HP4=\ngithub.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA=\ngithub.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=\ngithub.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=\ngo.etcd.io/etcd/api/v3 v3.5.5 h1:BX4JIbQ7hl7+jL+g+2j5UAr0o1bctCm6/Ct+ArBGkf0=\ngo.etcd.io/etcd/api/v3 v3.5.5/go.mod h1:KFtNaxGDw4Yx/BA4iPPwevUTAuqcsPxzyX8PHydchN8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.5 h1:9S0JUVvmrVl7wCF39iTQthdaaNIiAaQbmK75ogO6GU8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.5/go.mod h1:ggrwbk069qxpKPq8/FKkQ3Xq9y39kbFR4LnKszpRXeQ=\ngo.etcd.io/etcd/client/v3 v3.5.5 h1:q++2WTJbUgpQu4B6hCuT7VkdwaTP7Qz6Daak3WzbrlI=\ngo.etcd.io/etcd/client/v3 v3.5.5/go.mod h1:aApjR4WGlSumpnJ2kloS75h6aHUmAyaPLjHMxpc7E7c=\ngo.mongodb.org/mongo-driver v1.7.3 h1:G4l/eYY9VrQAK/AUgkV0koQKzQnyddnWxrd/Etf0jIs=\ngo.mongodb.org/mongo-driver v1.7.3/go.mod h1:NqaYOwnXWr5Pm7AOpO5QFxKJ503nbMse/R79oO62zWg=\ngo.opencensus.io v0.22.4/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=\ngo.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=\ngo.uber.org/atomic v1.3.2/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=\ngo.uber.org/atomic v1.5.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.5.1/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/atomic v1.10.0 h1:9qC72Qh0+3MqyJbAn8YU5xVq1frD8bn3JtD2oXtafVQ=\ngo.uber.org/atomic v1.10.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=\ngo.uber.org/cadence v0.19.0 h1:EvDXwIJ0lAAxL2i8ne/vG/TeoJM6xkAyqgTRFmIWG+c=\ngo.uber.org/cadence v0.19.0/go.mod h1:s91dOf0kcJbumPscRIVFV/4Xq/exhefzpXmnDiRRTxs=\ngo.uber.org/config v1.4.0 h1:upnMPpMm6WlbZtXoasNkK4f0FhxwS+W4Iqz5oNznehQ=\ngo.uber.org/config v1.4.0/go.mod h1:aCyrMHmUAc/s2h9sv1koP84M9ZF/4K+g2oleyESO/Ig=\ngo.uber.org/dig v1.8.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.10.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.18.0 h1:imUL1UiY0Mg4bqbFfsRQO5G4CGRBec/ZujWTvSVp3pw=\ngo.uber.org/dig v1.18.0/go.mod h1:Us0rSJiThwCv2GteUN0Q7OKvU7n5J4dxZ9JKUXozFdE=\ngo.uber.org/fx v1.10.0/go.mod h1:vLRicqpG/qQEzno4SYU86iCwfT95EZza+Eba0ItuxqY=\ngo.uber.org/fx v1.13.1/go.mod h1:bREWhavnedxpJeTq9pQT53BbvwhUv7TcpsOqcH4a+3w=\ngo.uber.org/fx v1.23.0 h1:lIr/gYWQGfTwGcSXWXu4vP5Ws6iqnNEIY+F/aFzCKTg=\ngo.uber.org/fx v1.23.0/go.mod h1:o/D9n+2mLP6v1EG+qsdT1O8wKopYAsqZasju97SDFCU=\ngo.uber.org/goleak v0.10.0/go.mod h1:VCZuO8V8mFPlL0F5J5GK1rtHV3DrFcQ1R8ryq7FK0aI=\ngo.uber.org/goleak v1.0.0/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=\ngo.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=\ngo.uber.org/goleak v1.2.0/go.mod h1:XJYK+MuIchqpmGmUSAzotztawfKvYLUIgg7guXrwVUo=\ngo.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=\ngo.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=\ngo.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=\ngo.uber.org/multierr v1.3.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.4.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=\ngo.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=\ngo.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=\ngo.uber.org/net/metrics v1.3.0 h1:iRLPuVecNYf/wIV+mQaA4IgN8ghifu3q1B4IT6HfwyY=\ngo.uber.org/net/metrics v1.3.0/go.mod h1:pEQrSDGNWT5IVpekWzee5//uHjI4gmgZFkobfw3bv8I=\ngo.uber.org/thriftrw v1.25.0/go.mod h1:IcIfSeZgc59AlYb0xr0DlDKIdD7SgjnFpG9BXCPyy9g=\ngo.uber.org/thriftrw v1.29.2 h1:pRuFLzbGvTcnYwGSjizWRHlbJUzGhu84sRiL1h1kUd8=\ngo.uber.org/thriftrw v1.29.2/go.mod h1:YcjXveberDd28/Bs34SwHy3yu85x/jB4UA2gIcz/Eo0=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee h1:0mgffUl7nfd+FpvXMVz4IDEaUSmT1ysygQC7qYo7sG4=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA=\ngo.uber.org/yarpc v1.55.0/go.mod h1:V2JUPDWHYGNpvyuroYjf0KFjwvBCtcFJLuvZqv7TWA0=\ngo.uber.org/yarpc v1.70.3 h1:yykHwzRD9/bgDtlOWoVuXbSZoU91Id2dWJO1CDSRHnI=\ngo.uber.org/yarpc v1.70.3/go.mod h1:EH6I6K1HxBbOxZIJfhdDf+H+cvXPHmJyRvpfPqES20U=\ngo.uber.org/zap v1.9.1/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=\ngo.uber.org/zap v1.13.0/go.mod h1:zwrFLgMcdUuIBviXEYEH1YKNaOBnKXsx2IPda5bBwHM=\ngo.uber.org/zap v1.17.0/go.mod h1:MXVU+bhUf/A7Xi2HNOnopQOrmycQ5Ih87HtOu4q5SSo=\ngo.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo=\ngo.uber.org/zap v1.26.0/go.mod h1:dtElttAiwGvoJ/vj4IwHBS/gXsEu/pZ50mUIRWuG0so=\ngolang.org/x/arch v0.0.0-20180920145803-b19384d3c130/go.mod h1:cYlCBUl1MsqxdiKgmc4uh7TxZfWSFLOGSRR090WDxt8=\ngolang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20190422162423-af44ce270edf/go.mod h1:WFFai1msRO1wXaEeE5yQxYXgSfI8pQAWXbQop6sCtWE=\ngolang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20200302210943-78000ba7a073/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58=\ngolang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8=\ngolang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw=\ngolang.org/x/exp v0.0.0-20180321215751-8460e604b9de/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20180807140117-3d87b88a115f/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190125153040-c74c464bbbf2/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b h1:kLiC65FbiHWFAOu+lxwNPujcsl8VYyTYYEZnsOO1WK4=\ngolang.org/x/exp v0.0.0-20231226003508-02704c960a9b/go.mod h1:iRJReGqOEeBhDZGkGbynYwcHlctCvnjTYIamk7uXpHI=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e h1:qyrTQ++p1afMkO4DPEeLGq/3oTsdlvdH4vqZUBWzUKM=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e/go.mod h1:AbB0pIl9nAr9wVwH+Z2ZpaocVmF5I4GyWCDIsVjR0bk=\ngolang.org/x/image v0.0.0-20180708004352-c73c2afc3b81/go.mod h1:ux5Hcp/YLpHSI86hEcLt0YII63i6oz57MZXIpbrjZUs=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20191125180803-fdd1cda4f05f/go.mod h1:5qLYkcX4OjUUV8bRuDixDT3tpyyb+LUpUlRWLxfhWrs=\ngolang.org/x/lint v0.0.0-20200130185559-910be7a94367/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 h1:VLliZ0d+/avPrXXH+OakdXhpJuEoBZuwh1m2j7U6Iug=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc=\ngolang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=\ngolang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=\ngolang.org/x/mod v0.18.0 h1:5+9lSbEzPSdWkH32vYPBwEpX8KwDbM52Ud9xBUvNlb0=\ngolang.org/x/mod v0.18.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200114155413-6afb5195e5aa/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=\ngolang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=\ngolang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=\ngolang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=\ngolang.org/x/net v0.40.0 h1:79Xs7wF06Gbdcg4kdCCIQArK11Z1hr5POQ6+fIYHNuY=\ngolang.org/x/net v0.40.0/go.mod h1:y0hY0exeL2Pku80/zKK7tpntoX23cqL3Oa6njdgRtds=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190412183630-56d357773e84/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.14.0 h1:woo0S4Yywslg6hp4eUFjTVOyKt0RookbpAHG4c1HmhQ=\ngolang.org/x/sync v0.14.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180903190138-2b024373dcd9/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190403152447-81d4e9dc473e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190419153524-e8e3143a4f4a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190531175056-4c3a928424d2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191220142924-d4481acd189f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200117145432-59e60aa80a0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200212091648-12a6c2dcc1e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200615200032-f1bc736245b1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200625212154-ddb9806d33ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210403161142-5e06dd20ab57/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=\ngolang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=\ngolang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=\ngolang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=\ngolang.org/x/text v0.0.0-20170915032832-14c0d48ead0c/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=\ngolang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=\ngolang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=\ngolang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=\ngolang.org/x/time v0.0.0-20170927054726-6dc17368e09b/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=\ngolang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180525024113-a5b4c53f6e8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190206041539-40960b6deb8e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190329151228-23e29df326fe/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190416151739-9c9e1878f421/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190420181800-aa740d480789/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190531172133-b3315ee88b7d/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191030062658-86caa796c7ab/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191104232314-dc038396d1f0/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191108193012-7d206e10da11/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191114200427-caa0b0f7d508/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191125144606-a911d9008d1f/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191226212025-6b505debf4bc/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117215004-fe56e6335763/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200216192241-b320d3a0f5a2/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200509030707-2212a7e161a5/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.2/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.11-0.20220513221640-090b14e8501f/go.mod h1:SgwaegtQh8clINPpECJMqnxLv9I09HLqnW3RMqW0CA4=\ngolang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=\ngolang.org/x/tools v0.22.0 h1:gqSGLZqv+AI9lIQzniJ0nZDRG5GBPsSi+DRNHWNz6yA=\ngolang.org/x/tools v0.22.0/go.mod h1:aCwcsjqvq7Yqt6TNyX7QMU2enbQ/Gt0bo6krSeEri+c=\ngolang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngonum.org/v1/gonum v0.0.0-20180816165407-929014505bf4/go.mod h1:Y+Yx5eoAFn32cQvJDxZx5Dpnq+c3wtXuadVZAcxbbBo=\ngonum.org/v1/gonum v0.7.0 h1:Hdks0L0hgznZLG9nzXb8vZ0rRvqNvAcgAp84y7Mwkgw=\ngonum.org/v1/gonum v0.7.0/go.mod h1:L02bwd0sqlsvRv41G7wGWFCsVNZFv/k1xzGIxeANHGM=\ngonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0 h1:OE9mWmgKkjJyEmDAAtGMPjXu+YNeGvK9VTSHY6+Qihc=\ngonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0/go.mod h1:wa6Ws7BG/ESfp6dHfk7C6KdzKA7wR7u/rKwOGE66zvw=\ngonum.org/v1/plot v0.0.0-20190515093506-e2840ee46a6b/go.mod h1:Wt8AAjI+ypCyYX3nZBvf6cAIx93T+c/OS2HFAYskSZc=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/genproto v0.0.0-20180518175338-11a468237815/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20200212174721-66ed5ce911ce/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/genproto v0.0.0-20210602131652-f16073e35f0c/go.mod h1:UODoCrxHCcBojKKwX1terBiRUaqAsFqJiF615XL43r0=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b h1:+YaDE2r2OG8t/z5qmsh7Y+XXwCbvadxxZ0YY6mTdrVA=\ngoogle.golang.org/genproto v0.0.0-20231016165738-49dd2c1f3d0b/go.mod h1:CgAqfJo+Xmu0GwA0411Ht3OU3OntXwsGmrmjI8ioGXI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a h1:myvhA4is3vrit1a6NZCWBIwN0kNEnX21DJOJX/NvIfI=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20231012201019-e917dd12ba7a/go.mod h1:SUBoKXbI1Efip18FClrQVGjWcyd0QZd8KkvdP34t7ww=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405 h1:AB/lmRny7e2pLhFEYIbl5qkDAUt2h0ZRO4wGPhZf+ik=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231030173426-d783a09b4405/go.mod h1:67X1fPuzjcrkymZzZV1vvkFeTn2Rvc6lYF9MYFGCcwE=\ngoogle.golang.org/grpc v1.12.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.28.0/go.mod h1:rpkK4SK4GF4Ach/+MFLZUBavHOvF2JJB5uozKKal+60=\ngoogle.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0=\ngoogle.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.38.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.40.1/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34=\ngoogle.golang.org/grpc v1.41.0/go.mod h1:U3l9uK9J0sini8mHphKoXyaqDA/8VyGnDee1zzIUK6k=\ngoogle.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk=\ngoogle.golang.org/grpc v1.59.0/go.mod h1:aUPDwccQo6OTjy7Hct4AfBPD1GptF4fyUjIkQ9YtF98=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=\ngoogle.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=\ngoogle.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=\ngoogle.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=\ngopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=\ngopkg.in/inf.v0 v0.9.1 h1:73M5CoZyi3ZLMOyDlQh031Cx6N9NDJ2Vvfl76EDAgDc=\ngopkg.in/inf.v0 v0.9.1/go.mod h1:cWUDdTG/fYaXco+Dcufb5Vnc6Gp2YChqWtbxRZE0mXw=\ngopkg.in/src-d/go-billy.v4 v4.3.0/go.mod h1:tm33zBoOwxjYHZIE+OV8bxTWFMJLrconzFMd38aARFk=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19 h1:WB265cn5OpO+hK3pikC9hpP1zI/KTwmyMFKloW9eOVc=\ngopkg.in/validator.v2 v2.0.0-20180514200540-135c24b11c19/go.mod h1:o4V0GXN9/CAmCsvJ0oXYZvrZOe7syiDZSN1GWGZTGzc=\ngopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=\nhonnef.co/go/tools v0.3.2 h1:ytYb4rOqyp1TSa2EPvNVwtPQJctSELKaMyLfqNP4+34=\nhonnef.co/go/tools v0.3.2/go.mod h1:jzwdWgg7Jdq75wlfblQxO4neNaFFSvgc1tD5Wv8U0Yw=\nrsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=\nrsc.io/quote/v3 v3.1.0/go.mod h1:yEA65RcK8LyAZtP9Kv3t0HmxON59tX3rD+tICJqUlj0=\nrsc.io/sampler v1.3.0/go.mod h1:T1hPZKmBbMNahiBKFy5HrXp6adAjACjK9JXDnKaTXpA=\nsigs.k8s.io/yaml v1.2.0/go.mod h1:yfXDCHCao9+ENCvLSE62v9VSji2MKu5jeNfTrofGhJc=\n"
  },
  {
    "path": "go.work",
    "content": "go 1.23.0\n\n// This file is primarily for gopls and IDEs it supports, as it does not\n// understand submodules correctly without it.\n//\n// It also slightly improves `go run` and `go test` behavior, as they work from\n// the top level regardless of the target's submodule, but it isn't quite\n// sufficient as things like `go build ./...` still do not recurse into submodules.\n//\n// Hence the Makefile's `make build` and `make tidy`, and changes to `make test`-alikes.\n// Maybe we can remove those some day, but not currently at least.\n//\n// Anyway, add any new submodules to here, and make sure they're all using\n// similar `replace repo => ../relative/path` replaces so it all stays compatible\n// at all times during development time.\n\nuse (\n\t.\n\t./cmd/server\n\t./common/archiver/gcloud\n\n// DO NOT include, tools dependencies are intentionally separate.\n// ./internal/tools\n)\n\n// technically only a minimum version, but forced to be precise in makefile targets.\n// must be kept in sync with docker files to avoid double-downloading.\n//\n// this should be safe to raise any time as it only impacts us, as this affects the\n// Go version used to build within this workspace only, it does not affect dependencies.\n// but note that it needs to be a version that docker + mac + linux all support, as\n// they all must be in sync.\ntoolchain go1.23.4\n"
  },
  {
    "path": "go.work.sum",
    "content": "bazil.org/fuse v0.0.0-20180421153158-65cc252bf669 h1:FNCRpXiquG1aoyqcIWVFmpTSKVcx2bQD38uZZeGtdlw=\nbitbucket.org/creachadair/shell v0.0.6 h1:reJflDbKqnlnqb4Oo2pQ1/BqmY/eCWcNGHrIUO8qIzc=\ncloud.google.com/go v0.34.0 h1:eOI3/cP2VTU6uZLDYAoic+eyzzB9YyGmJ7eIjl8rOPg=\ncloud.google.com/go v0.110.2/go.mod h1:k04UEeEtb6ZBRTv3dZz4CeJC3jKGxyhl0sAiVVquxiw=\ncloud.google.com/go/accessapproval v1.7.2 h1:W55SFrY6EVlcmmRGUk0rGhuy3j4fn7UtEocib/zADVE=\ncloud.google.com/go/accessapproval v1.7.2/go.mod h1:/gShiq9/kK/h8T/eEn1BTzalDvk0mZxJlhfw0p+Xuc0=\ncloud.google.com/go/accesscontextmanager v1.8.2 h1:jcOXen2u13aHgOHibUjxyPI+fZzVhElxy2gzJJlOOHg=\ncloud.google.com/go/accesscontextmanager v1.8.2/go.mod h1:E6/SCRM30elQJ2PKtFMs2YhfJpZSNcJyejhuzoId4Zk=\ncloud.google.com/go/aiplatform v1.51.1 h1:g+y03dll9HnX9U0oBKIqUOI+8VQWT1QJF12VGxkal0Q=\ncloud.google.com/go/aiplatform v1.51.1/go.mod h1:kY3nIMAVQOK2XDqDPHaOuD9e+FdMA6OOpfBjsvaFSOo=\ncloud.google.com/go/analytics v0.21.4 h1:SScWR8i/M8h7h3lFKtOYcj0r4272aL+KvRRrsu39Vec=\ncloud.google.com/go/analytics v0.21.4/go.mod h1:zZgNCxLCy8b2rKKVfC1YkC2vTrpfZmeRCySM3aUbskA=\ncloud.google.com/go/apigateway v1.6.2 h1:I46jVrhr2M1JJ1lK7JGn2BvybN44muEh+LSjBQ1l9hw=\ncloud.google.com/go/apigateway v1.6.2/go.mod h1:CwMC90nnZElorCW63P2pAYm25AtQrHfuOkbRSHj0bT8=\ncloud.google.com/go/apigeeconnect v1.6.2 h1:7LzOTW34EH2julg0MQVt+U9ZdmiCKcg6fef/ugKL2Xo=\ncloud.google.com/go/apigeeconnect v1.6.2/go.mod h1:s6O0CgXT9RgAxlq3DLXvG8riw8PYYbU/v25jqP3Dy18=\ncloud.google.com/go/apigeeregistry v0.7.2 h1:MESEjKSfz4TvLAzT2KPimDDvhOyQlcq7aFFREG2PRt4=\ncloud.google.com/go/apigeeregistry v0.7.2/go.mod h1:9CA2B2+TGsPKtfi3F7/1ncCCsL62NXBRfM6iPoGSM+8=\ncloud.google.com/go/apikeys v0.6.0 h1:B9CdHFZTFjVti89tmyXXrO+7vSNo2jvZuHG8zD5trdQ=\ncloud.google.com/go/apikeys v0.6.0/go.mod h1:kbpXu5upyiAlGkKrJgQl8A0rKNNJ7dQ377pdroRSSi8=\ncloud.google.com/go/appengine v1.8.2 h1:0/OFV0FQKgi0AB4E8NuYN0JY3hJzND4ftRpK7P26uaw=\ncloud.google.com/go/appengine v1.8.2/go.mod h1:WMeJV9oZ51pvclqFN2PqHoGnys7rK0rz6s3Mp6yMvDo=\ncloud.google.com/go/area120 v0.8.2 h1:h/wMtPPsgFJfMce1b9M24Od8RuKt8CWENwr+X24tBhE=\ncloud.google.com/go/area120 v0.8.2/go.mod h1:a5qfo+x77SRLXnCynFWPUZhnZGeSgvQ+Y0v1kSItkh4=\ncloud.google.com/go/artifactregistry v1.14.3 h1:Ssv6f+jgfhDdhu43AaHUaSosIYpQ+TPCJNwqYSJT1AE=\ncloud.google.com/go/artifactregistry v1.14.3/go.mod h1:A2/E9GXnsyXl7GUvQ/2CjHA+mVRoWAXC0brg2os+kNI=\ncloud.google.com/go/asset v1.15.1 h1:+9f5/s/U0AGZSPLTOMcXSZ5NDB5jQ2Szr+WQPgPA8bk=\ncloud.google.com/go/asset v1.15.1/go.mod h1:yX/amTvFWRpp5rcFq6XbCxzKT8RJUam1UoboE179jU4=\ncloud.google.com/go/assuredworkloads v1.11.2 h1:EbPyk3fC8sTxSIPoFrCR9P1wRTVdXcRxvPqFK8/wdso=\ncloud.google.com/go/assuredworkloads v1.11.2/go.mod h1:O1dfr+oZJMlE6mw0Bp0P1KZSlj5SghMBvTpZqIcUAW4=\ncloud.google.com/go/automl v1.13.2 h1:kUN4Y6N61AsNdXsdZIug1c+2pTJ5tg9xUA6+yn0Wf8Y=\ncloud.google.com/go/automl v1.13.2/go.mod h1:gNY/fUmDEN40sP8amAX3MaXkxcqPIn7F1UIIPZpy4Mg=\ncloud.google.com/go/baremetalsolution v1.2.1 h1:uRpZsKiWFDyT1sARZVRKqnOmf2mpRfVas7KMC3/MA4I=\ncloud.google.com/go/baremetalsolution v1.2.1/go.mod h1:3qKpKIw12RPXStwQXcbhfxVj1dqQGEvcmA+SX/mUR88=\ncloud.google.com/go/batch v1.5.1 h1:+8ZogCLFauglOE5ybTCWscoexD7Z8k4XW27RVTKNEoo=\ncloud.google.com/go/batch v1.5.1/go.mod h1:RpBuIYLkQu8+CWDk3dFD/t/jOCGuUpkpX+Y0n1Xccs8=\ncloud.google.com/go/beyondcorp v1.0.1 h1:uQpsXwttlV0+AXHdB5qaZl1mz2SsyYV1PKgTR74noaQ=\ncloud.google.com/go/beyondcorp v1.0.1/go.mod h1:zl/rWWAFVeV+kx+X2Javly7o1EIQThU4WlkynffL/lk=\ncloud.google.com/go/bigquery v1.56.0 h1:LHIc9E7Kw+ftFpQFKzZYBB88IAFz7qONawXXx0F3QBo=\ncloud.google.com/go/bigquery v1.56.0/go.mod h1:KDcsploXTEY7XT3fDQzMUZlpQLHzE4itubHrnmhUrZA=\ncloud.google.com/go/billing v1.17.2 h1:ozS/MNj6KKz8Reuw7tIG8Ycucq/YpSf3u3XCqrupbcg=\ncloud.google.com/go/billing v1.17.2/go.mod h1:u/AdV/3wr3xoRBk5xvUzYMS1IawOAPwQMuHgHMdljDg=\ncloud.google.com/go/binaryauthorization v1.7.1 h1:i2S+/G36VA1UG8gdcQLpq5I58/w/RzAnjQ65scKozFg=\ncloud.google.com/go/binaryauthorization v1.7.1/go.mod h1:GTAyfRWYgcbsP3NJogpV3yeunbUIjx2T9xVeYovtURE=\ncloud.google.com/go/certificatemanager v1.7.2 h1:Xytp8O0/EDh2nVscHhFQpicY9YAT3f3R7D7pv/z29uE=\ncloud.google.com/go/certificatemanager v1.7.2/go.mod h1:15SYTDQMd00kdoW0+XY5d9e+JbOPjp24AvF48D8BbcQ=\ncloud.google.com/go/channel v1.17.1 h1:+1B+Gj/3SJSLGJZXCp3dWiseMVHoSZ7Xo6Klg1fqM64=\ncloud.google.com/go/channel v1.17.1/go.mod h1:xqfzcOZAcP4b/hUDH0GkGg1Sd5to6di1HOJn/pi5uBQ=\ncloud.google.com/go/cloudbuild v1.14.1 h1:Tp0ITIlFam7T8K/TyeceITtpw1f8+KxVKwYyiyWDPK8=\ncloud.google.com/go/cloudbuild v1.14.1/go.mod h1:K7wGc/3zfvmYWOWwYTgF/d/UVJhS4pu+HAy7PL7mCsU=\ncloud.google.com/go/clouddms v1.7.1 h1:LrtqeR2xKV3juG5N7eeUgW+PqdMClOWH2U9PN3EpfFw=\ncloud.google.com/go/clouddms v1.7.1/go.mod h1:o4SR8U95+P7gZ/TX+YbJxehOCsM+fe6/brlrFquiszk=\ncloud.google.com/go/cloudtasks v1.12.2 h1:IoJI49JClvv2+NYvcABRgTO9y4veAUFlaOTigm+xXqE=\ncloud.google.com/go/cloudtasks v1.12.2/go.mod h1:A7nYkjNlW2gUoROg1kvJrQGhJP/38UaWwsnuBDOBVUk=\ncloud.google.com/go/compute v1.18.0/go.mod h1:1X7yHxec2Ga+Ss6jPyjxRxpu2uu7PLgsOVXvgU0yacs=\ncloud.google.com/go/compute v1.19.3/go.mod h1:qxvISKp/gYnXkSAD1ppcSOveRAmzxicEv/JlizULFrI=\ncloud.google.com/go/compute v1.20.1/go.mod h1:4tCnrn48xsqlwSAiLf1HXMQk8CONslYbdiEZc9FEIbM=\ncloud.google.com/go/compute v1.23.0 h1:tP41Zoavr8ptEqaW6j+LQOnyBBhO7OkOMAGrgLopTwY=\ncloud.google.com/go/compute v1.23.0/go.mod h1:4tCnrn48xsqlwSAiLf1HXMQk8CONslYbdiEZc9FEIbM=\ncloud.google.com/go/contactcenterinsights v1.11.1 h1:dEfCjtdYjS3n8/1HEKbJaOL31l3dEs3q9aeaNsyrJBc=\ncloud.google.com/go/contactcenterinsights v1.11.1/go.mod h1:FeNP3Kg8iteKM80lMwSk3zZZKVxr+PGnAId6soKuXwE=\ncloud.google.com/go/container v1.26.1 h1:1CXjOL/dZZ2jXX1CYWqlxmXqJbZo8HwQX4DJxLzgQWo=\ncloud.google.com/go/container v1.26.1/go.mod h1:5smONjPRUxeEpDG7bMKWfDL4sauswqEtnBK1/KKpR04=\ncloud.google.com/go/containeranalysis v0.11.1 h1:PHh4KTcMpCjYgxfV+TzvP24wolTGP9lGbqh9sBNHxjs=\ncloud.google.com/go/containeranalysis v0.11.1/go.mod h1:rYlUOM7nem1OJMKwE1SadufX0JP3wnXj844EtZAwWLY=\ncloud.google.com/go/datacatalog v1.18.1 h1:xJp9mZrc2HPaoxIz3sP9pCmf/impifweQ/yGG9VBfio=\ncloud.google.com/go/datacatalog v1.18.1/go.mod h1:TzAWaz+ON1tkNr4MOcak8EBHX7wIRX/gZKM+yTVsv+A=\ncloud.google.com/go/dataflow v0.9.2 h1:cpu2OeNxnYVadAIXETLRS5riz3KUR8ErbTojAQTFJVg=\ncloud.google.com/go/dataflow v0.9.2/go.mod h1:vBfdBZ/ejlTaYIGB3zB4T08UshH70vbtZeMD+urnUSo=\ncloud.google.com/go/dataform v0.8.2 h1:l155O3DS7pfyR91maS4l92bEjKbkbWie3dpgltZ1Q68=\ncloud.google.com/go/dataform v0.8.2/go.mod h1:X9RIqDs6NbGPLR80tnYoPNiO1w0wenKTb8PxxlhTMKM=\ncloud.google.com/go/datafusion v1.7.2 h1:CIIXp4bbwck49ZTV/URabJaV48jVB86THyVBWGgeDjw=\ncloud.google.com/go/datafusion v1.7.2/go.mod h1:62K2NEC6DRlpNmI43WHMWf9Vg/YvN6QVi8EVwifElI0=\ncloud.google.com/go/datalabeling v0.8.2 h1:4N5mbjauemzaatxGOFVpV2i8HiXSUUhyNRBU+dCBHl0=\ncloud.google.com/go/datalabeling v0.8.2/go.mod h1:cyDvGHuJWu9U/cLDA7d8sb9a0tWLEletStu2sTmg3BE=\ncloud.google.com/go/dataplex v1.10.1 h1:8Irss8sIalm/X8r0Masv5KJRkddcxov3TiW8W96FmC4=\ncloud.google.com/go/dataplex v1.10.1/go.mod h1:1MzmBv8FvjYfc7vDdxhnLFNskikkB+3vl475/XdCDhs=\ncloud.google.com/go/dataproc v1.12.0 h1:W47qHL3W4BPkAIbk4SWmIERwsWBaNnWm0P2sdx3YgGU=\ncloud.google.com/go/dataproc v1.12.0/go.mod h1:zrF3aX0uV3ikkMz6z4uBbIKyhRITnxvr4i3IjKsKrw4=\ncloud.google.com/go/dataproc/v2 v2.2.1 h1:BPjIIkTCAOHUkMtWKqae55qEku5K09LVbQ46LYt7r1s=\ncloud.google.com/go/dataproc/v2 v2.2.1/go.mod h1:QdAJLaBjh+l4PVlVZcmrmhGccosY/omC1qwfQ61Zv/o=\ncloud.google.com/go/dataqna v0.8.2 h1:vJ9JVKDgDG7AQMbTD8pdWaogJ4c/yHn0qer+q0nFIaw=\ncloud.google.com/go/dataqna v0.8.2/go.mod h1:KNEqgx8TTmUipnQsScOoDpq/VlXVptUqVMZnt30WAPs=\ncloud.google.com/go/datastore v1.15.0 h1:0P9WcsQeTWjuD1H14JIY7XQscIPQ4Laje8ti96IC5vg=\ncloud.google.com/go/datastore v1.15.0/go.mod h1:GAeStMBIt9bPS7jMJA85kgkpsMkvseWWXiaHya9Jes8=\ncloud.google.com/go/datastream v1.10.1 h1:XWiXV1hzs8oAd54//wcb1L15Jl7MnZ/cY2B8XCmu0xE=\ncloud.google.com/go/datastream v1.10.1/go.mod h1:7ngSYwnw95YFyTd5tOGBxHlOZiL+OtpjheqU7t2/s/c=\ncloud.google.com/go/deploy v1.13.1 h1:eV5MdoQJGdac/k7D97SDjD8iLE4jCzL42UCAgG6j0iE=\ncloud.google.com/go/deploy v1.13.1/go.mod h1:8jeadyLkH9qu9xgO3hVWw8jVr29N1mnW42gRJT8GY6g=\ncloud.google.com/go/dialogflow v1.44.1 h1:Ml/hgEzU3AN0tjNSSv4/QmG1nqwYEsiCySKMkWMqUmI=\ncloud.google.com/go/dialogflow v1.44.1/go.mod h1:n/h+/N2ouKOO+rbe/ZnI186xImpqvCVj2DdsWS/0EAk=\ncloud.google.com/go/dlp v1.10.2 h1:sWOATigjZOKmA2rVOSjIcKLCtL2ifdawaukx+H9iffk=\ncloud.google.com/go/dlp v1.10.2/go.mod h1:ZbdKIhcnyhILgccwVDzkwqybthh7+MplGC3kZVZsIOQ=\ncloud.google.com/go/documentai v1.23.2 h1:IAKWBngDFTxABdAH52uAn0osPDemyegyRmf5IQKznHw=\ncloud.google.com/go/documentai v1.23.2/go.mod h1:Q/wcRT+qnuXOpjAkvOV4A+IeQl04q2/ReT7SSbytLSo=\ncloud.google.com/go/domains v0.9.2 h1:SjpTtaTNRPPajrGiZEtxz9dpElO4PxuDWFvU4JpV1gk=\ncloud.google.com/go/domains v0.9.2/go.mod h1:3YvXGYzZG1Temjbk7EyGCuGGiXHJwVNmwIf+E/cUp5I=\ncloud.google.com/go/edgecontainer v1.1.2 h1:B+Acb/0frXUxc60i6lC0JtXrBFAKoS7ZELmet9+ySo8=\ncloud.google.com/go/edgecontainer v1.1.2/go.mod h1:wQRjIzqxEs9e9wrtle4hQPSR1Y51kqN75dgF7UllZZ4=\ncloud.google.com/go/errorreporting v0.3.0 h1:kj1XEWMu8P0qlLhm3FwcaFsUvXChV/OraZwA70trRR0=\ncloud.google.com/go/errorreporting v0.3.0/go.mod h1:xsP2yaAp+OAW4OIm60An2bbLpqIhKXdWR/tawvl7QzU=\ncloud.google.com/go/essentialcontacts v1.6.3 h1:xrGTLRTzunQk5XhBIkdftuC00B9MUoEXi7Pjgeu1kMM=\ncloud.google.com/go/essentialcontacts v1.6.3/go.mod h1:yiPCD7f2TkP82oJEFXFTou8Jl8L6LBRPeBEkTaO0Ggo=\ncloud.google.com/go/eventarc v1.13.1 h1:FmEcxG5rX3LaUB2nRjf2Pas5J5TtVrVznaHN5rxYxnQ=\ncloud.google.com/go/eventarc v1.13.1/go.mod h1:EqBxmGHFrruIara4FUQ3RHlgfCn7yo1HYsu2Hpt/C3Y=\ncloud.google.com/go/filestore v1.7.2 h1:/Nnk5pOoY1Lx6A42hJ2eBYcBfqKvLcnh8fV4egopvY4=\ncloud.google.com/go/filestore v1.7.2/go.mod h1:TYOlyJs25f/omgj+vY7/tIG/E7BX369triSPzE4LdgE=\ncloud.google.com/go/firestore v1.13.0 h1:/3S4RssUV4GO/kvgJZB+tayjhOfyAHs+KcpJgRVu/Qk=\ncloud.google.com/go/firestore v1.13.0/go.mod h1:QojqqOh8IntInDUSTAh0c8ZsPYAr68Ma8c5DWOy8xb8=\ncloud.google.com/go/functions v1.15.2 h1:DpT51zU3UMTt64efB4a9hE9B98Kb0fZC3IfaVp7GnkE=\ncloud.google.com/go/functions v1.15.2/go.mod h1:CHAjtcR6OU4XF2HuiVeriEdELNcnvRZSk1Q8RMqy4lE=\ncloud.google.com/go/gaming v1.9.0 h1:7vEhFnZmd931Mo7sZ6pJy7uQPDxF7m7v8xtBheG08tc=\ncloud.google.com/go/gaming v1.9.0/go.mod h1:Fc7kEmCObylSWLO334NcO+O9QMDyz+TKC4v1D7X+Bc0=\ncloud.google.com/go/gkebackup v1.3.2 h1:1fnA934a/0oz7nU22gTzmGYFVi6V13Q/hCkdC99K178=\ncloud.google.com/go/gkebackup v1.3.2/go.mod h1:OMZbXzEJloyXMC7gqdSB+EOEQ1AKcpGYvO3s1ec5ixk=\ncloud.google.com/go/gkeconnect v0.8.2 h1:AuR3YNK0DgLVrmcc8o4sBrU0dVs/SULSuLh4Gmn1e10=\ncloud.google.com/go/gkeconnect v0.8.2/go.mod h1:6nAVhwchBJYgQCXD2pHBFQNiJNyAd/wyxljpaa6ZPrY=\ncloud.google.com/go/gkehub v0.14.2 h1:7rddjV52z0RbToFYj1B39R9dsn+6IXgx4DduEH7N25Q=\ncloud.google.com/go/gkehub v0.14.2/go.mod h1:iyjYH23XzAxSdhrbmfoQdePnlMj2EWcvnR+tHdBQsCY=\ncloud.google.com/go/gkemulticloud v1.0.1 h1:V82LxEvFIGJnebn7BBdOUKcVlNQqBaubbKtLgRicHow=\ncloud.google.com/go/gkemulticloud v1.0.1/go.mod h1:AcrGoin6VLKT/fwZEYuqvVominLriQBCKmbjtnbMjG8=\ncloud.google.com/go/grafeas v0.3.0 h1:oyTL/KjiUeBs9eYLw/40cpSZglUC+0F7X4iu/8t7NWs=\ncloud.google.com/go/grafeas v0.3.0/go.mod h1:P7hgN24EyONOTMyeJH6DxG4zD7fwiYa5Q6GUgyFSOU8=\ncloud.google.com/go/gsuiteaddons v1.6.2 h1:vR7E1gR85x0wlbUek3cZYJ67U67GpNrboNCRiF/VSSc=\ncloud.google.com/go/gsuiteaddons v1.6.2/go.mod h1:K65m9XSgs8hTF3X9nNTPi8IQueljSdYo9F+Mi+s4MyU=\ncloud.google.com/go/iam v0.12.0/go.mod h1:knyHGviacl11zrtZUoDuYpDgLjvr28sLQaG0YB2GYAY=\ncloud.google.com/go/iam v0.13.0/go.mod h1:ljOg+rcNfzZ5d6f1nAUJ8ZIxOaZUVoS14bKCtaLZ/D0=\ncloud.google.com/go/iam v1.1.0/go.mod h1:nxdHjaKfCr7fNYx/HJMM8LgiMugmveWlkatear5gVyk=\ncloud.google.com/go/iap v1.9.1 h1:J5r6CL6EakRmsMRIm2yV0PF5zfIm4sMQbQfPhSTnRzA=\ncloud.google.com/go/iap v1.9.1/go.mod h1:SIAkY7cGMLohLSdBR25BuIxO+I4fXJiL06IBL7cy/5Q=\ncloud.google.com/go/ids v1.4.2 h1:KqvR28pAnIss6d2pmGOQ+Fcsi3FOWDVhqdr6QaVvqsI=\ncloud.google.com/go/ids v1.4.2/go.mod h1:3vw8DX6YddRu9BncxuzMyWn0g8+ooUjI2gslJ7FH3vk=\ncloud.google.com/go/iot v1.7.2 h1:qFNv3teWkONIPmuY2mzodEnHb6E67ch2OZ6216ycUiU=\ncloud.google.com/go/iot v1.7.2/go.mod h1:q+0P5zr1wRFpw7/MOgDXrG/HVA+l+cSwdObffkrpnSg=\ncloud.google.com/go/kms v1.15.3 h1:RYsbxTRmk91ydKCzekI2YjryO4c5Y2M80Zwcs9/D/cI=\ncloud.google.com/go/kms v1.15.3/go.mod h1:AJdXqHxS2GlPyduM99s9iGqi2nwbviBbhV/hdmt4iOQ=\ncloud.google.com/go/language v1.11.1 h1:BjU7Ljhh0ZYnZC8jZwiezf1FH75yijJ4raAScseqCns=\ncloud.google.com/go/language v1.11.1/go.mod h1:Xyid9MG9WOX3utvDbpX7j3tXDmmDooMyMDqgUVpH17U=\ncloud.google.com/go/lifesciences v0.9.2 h1:0naTq5qUWoRt/b5P+SZ/0mun7ZTlhpJZJsUxhCmLv1c=\ncloud.google.com/go/lifesciences v0.9.2/go.mod h1:QHEOO4tDzcSAzeJg7s2qwnLM2ji8IRpQl4p6m5Z9yTA=\ncloud.google.com/go/logging v1.8.1 h1:26skQWPeYhvIasWKm48+Eq7oUqdcdbwsCVwz5Ys0FvU=\ncloud.google.com/go/logging v1.8.1/go.mod h1:TJjR+SimHwuC8MZ9cjByQulAMgni+RkXeI3wwctHJEI=\ncloud.google.com/go/longrunning v0.5.0/go.mod h1:0JNuqRShmscVAhIACGtskSAWtqtOoPkwP0YF1oVEchc=\ncloud.google.com/go/longrunning v0.5.2 h1:u+oFqfEwwU7F9dIELigxbe0XVnBAo9wqMuQLA50CZ5k=\ncloud.google.com/go/longrunning v0.5.2/go.mod h1:nqo6DQbNV2pXhGDbDMoN2bWz68MjZUzqv2YttZiveCs=\ncloud.google.com/go/managedidentities v1.6.2 h1:QijSmmWHb3EzYQr8SrjWe941ba9G5sTCF5PvhhMM8CM=\ncloud.google.com/go/managedidentities v1.6.2/go.mod h1:5c2VG66eCa0WIq6IylRk3TBW83l161zkFvCj28X7jn8=\ncloud.google.com/go/maps v1.4.1 h1:/wp8wImC3tHIHOoaQGRA+KyH3as/Dvp+3J/NqJQBiPQ=\ncloud.google.com/go/maps v1.4.1/go.mod h1:BxSa0BnW1g2U2gNdbq5zikLlHUuHW0GFWh7sgML2kIY=\ncloud.google.com/go/mediatranslation v0.8.2 h1:nyBZbNX1j34H00n+irnQraCogrkRWntQsDoA6s8OfKo=\ncloud.google.com/go/mediatranslation v0.8.2/go.mod h1:c9pUaDRLkgHRx3irYE5ZC8tfXGrMYwNZdmDqKMSfFp8=\ncloud.google.com/go/memcache v1.10.2 h1:WLJALO3FxuStMiYdSQwiQBDBcs4G8DDwZQmXK+YzAWk=\ncloud.google.com/go/memcache v1.10.2/go.mod h1:f9ZzJHLBrmd4BkguIAa/l/Vle6uTHzHokdnzSWOdQ6A=\ncloud.google.com/go/metastore v1.13.1 h1:tLemzNMjKY+xdJUDQt9v5+fQqSufTNgKHHQmihG5ay8=\ncloud.google.com/go/metastore v1.13.1/go.mod h1:IbF62JLxuZmhItCppcIfzBBfUFq0DIB9HPDoLgWrVOU=\ncloud.google.com/go/monitoring v1.16.1 h1:CTklIuUkS5nCricGojPwdkSgPsCTX2HmYTxFDg+UvpU=\ncloud.google.com/go/monitoring v1.16.1/go.mod h1:6HsxddR+3y9j+o/cMJH6q/KJ/CBTvM/38L/1m7bTRJ4=\ncloud.google.com/go/networkconnectivity v1.14.1 h1:uR+ASueYNodsPCd9wcYEedqjH4+LaCkKqltRBF6CmB4=\ncloud.google.com/go/networkconnectivity v1.14.1/go.mod h1:LyGPXR742uQcDxZ/wv4EI0Vu5N6NKJ77ZYVnDe69Zug=\ncloud.google.com/go/networkmanagement v1.9.1 h1:ZK6i6FVQNc1t3fecM3hf9Nu6Kr9C95xr+zMVORYd8ak=\ncloud.google.com/go/networkmanagement v1.9.1/go.mod h1:CCSYgrQQvW73EJawO2QamemYcOb57LvrDdDU51F0mcI=\ncloud.google.com/go/networksecurity v0.9.2 h1:fA73AX//KWaqNKOvuQ00WUD3Z/XMhiMhHSFTEl2Wxec=\ncloud.google.com/go/networksecurity v0.9.2/go.mod h1:jG0SeAttWzPMUILEHDUvFYdQTl8L/E/KC8iZDj85lEI=\ncloud.google.com/go/notebooks v1.10.1 h1:j/G3r6SPoWzD6CZZrDffZGwgGALvxWwtKJHJ4GF17WA=\ncloud.google.com/go/notebooks v1.10.1/go.mod h1:5PdJc2SgAybE76kFQCWrTfJolCOUQXF97e+gteUUA6A=\ncloud.google.com/go/optimization v1.5.1 h1:71wTxJz8gRrVEHF4fw18sGynAyNQwatxCJBI3m3Rd4c=\ncloud.google.com/go/optimization v1.5.1/go.mod h1:NC0gnUD5MWVAF7XLdoYVPmYYVth93Q6BUzqAq3ZwtV8=\ncloud.google.com/go/orchestration v1.8.2 h1:lb+Vphr+x2V9ukHwLjyaXJpbPuPhaKdobQx3UAOeSsQ=\ncloud.google.com/go/orchestration v1.8.2/go.mod h1:T1cP+6WyTmh6LSZzeUhvGf0uZVmJyTx7t8z7Vg87+A0=\ncloud.google.com/go/orgpolicy v1.11.2 h1:Dnfh5sj3aIAuJzH4Q4rBp6lCJ/IdXRBbwQ0/nQsUySE=\ncloud.google.com/go/orgpolicy v1.11.2/go.mod h1:biRDpNwfyytYnmCRWZWxrKF22Nkz9eNVj9zyaBdpm1o=\ncloud.google.com/go/osconfig v1.12.2 h1:AjHbw8MgKKaTFAEJWGdOYtMED3wUXKLtvdfP8Uzbuy0=\ncloud.google.com/go/osconfig v1.12.2/go.mod h1:eh9GPaMZpI6mEJEuhEjUJmaxvQ3gav+fFEJon1Y8Iw0=\ncloud.google.com/go/oslogin v1.11.1 h1:r3JYeLf004krfXhRMDfYKlBdMgDDc2q2PM1bomb5Luw=\ncloud.google.com/go/oslogin v1.11.1/go.mod h1:OhD2icArCVNUxKqtK0mcSmKL7lgr0LVlQz+v9s1ujTg=\ncloud.google.com/go/phishingprotection v0.8.2 h1:BIv/42ooQXh/jW8BW2cgO0E6yRPbEdvqH3JzKV7BlmI=\ncloud.google.com/go/phishingprotection v0.8.2/go.mod h1:LhJ91uyVHEYKSKcMGhOa14zMMWfbEdxG032oT6ECbC8=\ncloud.google.com/go/policytroubleshooter v1.9.1 h1:92YSoPZE62QkNM0G6Nl6PICKUyv4aNgsdtWWceJR6ys=\ncloud.google.com/go/policytroubleshooter v1.9.1/go.mod h1:MYI8i0bCrL8cW+VHN1PoiBTyNZTstCg2WUw2eVC4c4U=\ncloud.google.com/go/privatecatalog v0.9.2 h1:gxL4Kn9IXt3tdIOpDPEDPI/kBBLVzaAX5wq6IbOYi8A=\ncloud.google.com/go/privatecatalog v0.9.2/go.mod h1:RMA4ATa8IXfzvjrhhK8J6H4wwcztab+oZph3c6WmtFc=\ncloud.google.com/go/pubsub v1.33.0 h1:6SPCPvWav64tj0sVX/+npCBKhUi/UjJehy9op/V3p2g=\ncloud.google.com/go/pubsub v1.33.0/go.mod h1:f+w71I33OMyxf9VpMVcZbnG5KSUkCOUHYpFd5U1GdRc=\ncloud.google.com/go/pubsublite v1.8.1 h1:pX+idpWMIH30/K7c0epN6V703xpIcMXWRjKJsz0tYGY=\ncloud.google.com/go/pubsublite v1.8.1/go.mod h1:fOLdU4f5xldK4RGJrBMm+J7zMWNj/k4PxwEZXy39QS0=\ncloud.google.com/go/recaptchaenterprise/v2 v2.8.1 h1:06V6+edT20PcrFJfH0TVWMZpZCUpSCADgwGwhkMsGmY=\ncloud.google.com/go/recaptchaenterprise/v2 v2.8.1/go.mod h1:JZYZJOeZjgSSTGP4uz7NlQ4/d1w5hGmksVgM0lbEij0=\ncloud.google.com/go/recommendationengine v0.8.2 h1:odf0TZXtwoZ5kJaWBlaE9D0AV+WJLLs+/SRSuE4T/ds=\ncloud.google.com/go/recommendationengine v0.8.2/go.mod h1:QIybYHPK58qir9CV2ix/re/M//Ty10OxjnnhWdaKS1Y=\ncloud.google.com/go/recommender v1.11.1 h1:GI4EBCMTLfC8I8R+e13ZaTAa8ZZ0KRPdS99hGtJYyaU=\ncloud.google.com/go/recommender v1.11.1/go.mod h1:sGwFFAyI57v2Hc5LbIj+lTwXipGu9NW015rkaEM5B18=\ncloud.google.com/go/redis v1.13.2 h1:2ZtIGspMT65wern2rjX35XPCCJxVKF4J0P1S99bac3k=\ncloud.google.com/go/redis v1.13.2/go.mod h1:0Hg7pCMXS9uz02q+LoEVl5dNHUkIQv+C/3L76fandSA=\ncloud.google.com/go/resourcemanager v1.9.2 h1:lC3PjJMHLPlZKqLfan6FkEb3X1F8oCRc1ylY7vRHvDQ=\ncloud.google.com/go/resourcemanager v1.9.2/go.mod h1:OujkBg1UZg5lX2yIyMo5Vz9O5hf7XQOSV7WxqxxMtQE=\ncloud.google.com/go/resourcesettings v1.6.2 h1:feqx2EcLRgtmwNHzeLw5Og4Wcy4vcZxw62b0x/QNu60=\ncloud.google.com/go/resourcesettings v1.6.2/go.mod h1:mJIEDd9MobzunWMeniaMp6tzg4I2GvD3TTmPkc8vBXk=\ncloud.google.com/go/retail v1.14.2 h1:ed5hWjpOwfsi6E9kj2AFzkz5ScT3aZs7o3MUM0YITUM=\ncloud.google.com/go/retail v1.14.2/go.mod h1:W7rrNRChAEChX336QF7bnMxbsjugcOCPU44i5kbLiL8=\ncloud.google.com/go/run v1.3.1 h1:xc46W9kxJI2De9hmpqHEBSSLJhP3bSZl86LdlJa5zm8=\ncloud.google.com/go/run v1.3.1/go.mod h1:cymddtZOzdwLIAsmS6s+Asl4JoXIDm/K1cpZTxV4Q5s=\ncloud.google.com/go/scheduler v1.10.2 h1:lgUd1D84JEgNzzHRlcZEIoQ6Ny10YWe8RNH1knhouNk=\ncloud.google.com/go/scheduler v1.10.2/go.mod h1:O3jX6HRH5eKCA3FutMw375XHZJudNIKVonSCHv7ropY=\ncloud.google.com/go/secretmanager v1.11.2 h1:52Z78hH8NBWIqbvIG0wi0EoTaAmSx99KIOAmDXIlX0M=\ncloud.google.com/go/secretmanager v1.11.2/go.mod h1:MQm4t3deoSub7+WNwiC4/tRYgDBHJgJPvswqQVB1Vss=\ncloud.google.com/go/security v1.15.2 h1:VNpdJNfMeHSJZ+647QtzPrvZ6rWChBklLm/NY64RVW8=\ncloud.google.com/go/security v1.15.2/go.mod h1:2GVE/v1oixIRHDaClVbHuPcZwAqFM28mXuAKCfMgYIg=\ncloud.google.com/go/securitycenter v1.23.1 h1:Epx7Gm9ZRPRiFfwDFplka2zKCS0J3cpm0Et1KwI2tvY=\ncloud.google.com/go/securitycenter v1.23.1/go.mod h1:w2HV3Mv/yKhbXKwOCu2i8bCuLtNP1IMHuiYQn4HJq5s=\ncloud.google.com/go/servicecontrol v1.11.1 h1:d0uV7Qegtfaa7Z2ClDzr9HJmnbJW7jn0WhZ7wOX6hLE=\ncloud.google.com/go/servicecontrol v1.11.1/go.mod h1:aSnNNlwEFBY+PWGQ2DoM0JJ/QUXqV5/ZD9DOLB7SnUk=\ncloud.google.com/go/servicedirectory v1.11.1 h1:SXhbxsfQJBsUDeo743x5AnVe8ifC7qjXU3bSTT6t/+Q=\ncloud.google.com/go/servicedirectory v1.11.1/go.mod h1:tJywXimEWzNzw9FvtNjsQxxJ3/41jseeILgwU/QLrGI=\ncloud.google.com/go/servicemanagement v1.8.0 h1:fopAQI/IAzlxnVeiKn/8WiV6zKndjFkvi+gzu+NjywY=\ncloud.google.com/go/servicemanagement v1.8.0/go.mod h1:MSS2TDlIEQD/fzsSGfCdJItQveu9NXnUniTrq/L8LK4=\ncloud.google.com/go/serviceusage v1.6.0 h1:rXyq+0+RSIm3HFypctp7WoXxIA563rn206CfMWdqXX4=\ncloud.google.com/go/serviceusage v1.6.0/go.mod h1:R5wwQcbOWsyuOfbP9tGdAnCAc6B9DRwPG1xtWMDeuPA=\ncloud.google.com/go/shell v1.7.2 h1:zk0Cf2smbFlAdhBQ5tXESZzzmsTfGc31fJfI6a0SVD8=\ncloud.google.com/go/shell v1.7.2/go.mod h1:KqRPKwBV0UyLickMn0+BY1qIyE98kKyI216sH/TuHmc=\ncloud.google.com/go/spanner v1.50.0 h1:QrJFOpaxCXdXF+GkiruLz642PHxkdj68PbbnLw3O2Zw=\ncloud.google.com/go/spanner v1.50.0/go.mod h1:eGj9mQGK8+hkgSVbHNQ06pQ4oS+cyc4tXXd6Dif1KoM=\ncloud.google.com/go/speech v1.19.1 h1:z035FMLs98jpnqcP5xZZ6Es+g6utbeVoUH64BaTzTSU=\ncloud.google.com/go/speech v1.19.1/go.mod h1:WcuaWz/3hOlzPFOVo9DUsblMIHwxP589y6ZMtaG+iAA=\ncloud.google.com/go/storagetransfer v1.10.1 h1:CU03oYLauu7xRV25fFmozHZHA/SokLQlC20Ip/UvFro=\ncloud.google.com/go/storagetransfer v1.10.1/go.mod h1:rS7Sy0BtPviWYTTJVWCSV4QrbBitgPeuK4/FKa4IdLs=\ncloud.google.com/go/talent v1.6.3 h1:TyJqwhmncdW5CL4rzYSYKJrR9YAe0iNqHtJTnnOaEyM=\ncloud.google.com/go/talent v1.6.3/go.mod h1:xoDO97Qd4AK43rGjJvyBHMskiEf3KulgYzcH6YWOVoo=\ncloud.google.com/go/texttospeech v1.7.2 h1:Ac53sRkUo8UMSuhyyWRFJvWEaX8vm0EFwwiTAxeVYuU=\ncloud.google.com/go/texttospeech v1.7.2/go.mod h1:VYPT6aTOEl3herQjFHYErTlSZJ4vB00Q2ZTmuVgluD4=\ncloud.google.com/go/tpu v1.6.2 h1:SAFzyGp6mU37lfLTV0cNQwu7tqH4X8b4RCpQZ1s+mYM=\ncloud.google.com/go/tpu v1.6.2/go.mod h1:NXh3NDwt71TsPZdtGWgAG5ThDfGd32X1mJ2cMaRlVgU=\ncloud.google.com/go/trace v1.10.2 h1:80Rh4JSqJLfe/xGNrpyO4MQxiFDXcHG1XrsevfmrIRQ=\ncloud.google.com/go/trace v1.10.2/go.mod h1:NPXemMi6MToRFcSxRl2uDnu/qAlAQ3oULUphcHGh1vA=\ncloud.google.com/go/translate v1.9.1 h1:gNPBVMINs+aZMB8BW+IfrHLLTfdq0t0GMwa31NmOXY4=\ncloud.google.com/go/translate v1.9.1/go.mod h1:TWIgDZknq2+JD4iRcojgeDtqGEp154HN/uL6hMvylS8=\ncloud.google.com/go/video v1.20.1 h1:yMfxQ4N/fXNDsCKNKw9W+FpdrJPj5CDu+FuAJBmGuoo=\ncloud.google.com/go/video v1.20.1/go.mod h1:3gJS+iDprnj8SY6pe0SwLeC5BUW80NjhwX7INWEuWGU=\ncloud.google.com/go/videointelligence v1.11.2 h1:vAKuM4YHwZy1W5P7hGJdfXriovqHHUZKhDBq8o4nqfg=\ncloud.google.com/go/videointelligence v1.11.2/go.mod h1:ocfIGYtIVmIcWk1DsSGOoDiXca4vaZQII1C85qtoplc=\ncloud.google.com/go/vision/v2 v2.7.3 h1:o8iiH4UsI6O8wO2Ax2r88fLG1RzYQIFevUQY7hXPZeM=\ncloud.google.com/go/vision/v2 v2.7.3/go.mod h1:V0IcLCY7W+hpMKXK1JYE0LV5llEqVmj+UJChjvA1WsM=\ncloud.google.com/go/vmmigration v1.7.2 h1:ObE8VWzL+xkU22IsPEMvPCWArnSQ85dEwR5fzgaOvA4=\ncloud.google.com/go/vmmigration v1.7.2/go.mod h1:iA2hVj22sm2LLYXGPT1pB63mXHhrH1m/ruux9TwWLd8=\ncloud.google.com/go/vmwareengine v1.0.1 h1:Bj9WECvQk1fkx8IG7gqII3+g1CzhqkPOV84WXvifpFg=\ncloud.google.com/go/vmwareengine v1.0.1/go.mod h1:aT3Xsm5sNx0QShk1Jc1B8OddrxAScYLwzVoaiXfdzzk=\ncloud.google.com/go/vpcaccess v1.7.2 h1:3qKiWvzK07eIa943mCvkcZB4gimxaQKKGdNoX01ps7A=\ncloud.google.com/go/vpcaccess v1.7.2/go.mod h1:mmg/MnRHv+3e8FJUjeSibVFvQF1cCy2MsFaFqxeY1HU=\ncloud.google.com/go/webrisk v1.9.2 h1:1NZppagzdGO0hVMJsUhZQ5a3Iu2cNyNObu85VFcvIVA=\ncloud.google.com/go/webrisk v1.9.2/go.mod h1:pY9kfDgAqxUpDBOrG4w8deLfhvJmejKB0qd/5uQIPBc=\ncloud.google.com/go/websecurityscanner v1.6.2 h1:V7PhbJ2OvpGHINL67RBhpwU3+g4MOoqOeL/sFYrogeE=\ncloud.google.com/go/websecurityscanner v1.6.2/go.mod h1:7YgjuU5tun7Eg2kpKgGnDuEOXWIrh8x8lWrJT4zfmas=\ncloud.google.com/go/workflows v1.12.1 h1:jvhSfcfAoOt0nILm7aZPJAHdpoe571qrJyc2ZlngaJk=\ncloud.google.com/go/workflows v1.12.1/go.mod h1:5A95OhD/edtOhQd/O741NSfIMezNTbCwLM1P1tBRGHM=\ncontrib.go.opencensus.io/exporter/aws v0.0.0-20200617204711-c478e41e60e9 h1:yxE46rQA0QaqPGqN2UnwXvgCrRqtjR1CsGSWVTRjvv4=\ncontrib.go.opencensus.io/exporter/stackdriver v0.13.5 h1:TNaexHK16gPUoc7uzELKOU7JULqccn1NDuqUxmxSqfo=\ncontrib.go.opencensus.io/integrations/ocsql v0.1.7 h1:G3k7C0/W44zcqkpRSFyjU9f6HZkbwIrL//qqnlqWZ60=\ndmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9 h1:VpgP7xuJadIUuKccphEpTJnWhS2jkQyMt6Y7pJCD7fY=\ngithub.com/Azure/azure-amqp-common-go/v3 v3.1.0 h1:1N4YSkWYWffOpQHromYdOucBSQXhNRKzqtgICy6To8Q=\ngithub.com/Azure/azure-service-bus-go v0.10.11 h1:GBg2mcLQA3af+w+ZuYhiAv58OWSVmQYWcds4nro8Xko=\ngithub.com/Azure/go-amqp v0.13.7 h1:ukcCtx138ZmOfHbdALuh9yoJhGtOY3+yaKApfzNvhSk=\ngithub.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802 h1:1BDTz0u9nC3//pOCMdNH+CiXJVYJh5UQNCOBG7jbELc=\ngithub.com/DataDog/datadog-go v3.2.0+incompatible h1:qSG2N4FghB1He/r2mFrWKCaL7dXCilEuNEeAn20fdD4=\ngithub.com/DataDog/zstd v1.4.5/go.mod h1:1jcaCB/ufaK+sKp1NBhlGmpz41jOoPQ35bpF36t7BBo=\ngithub.com/GoogleCloudPlatform/cloudsql-proxy v1.22.0 h1:aTDBS16pX1X4ZR/GFsC2NcOCYJ1hDJwJm3WmKRA905Q=\ngithub.com/Groxx/cadence-idl v0.0.0-20240822223859-88ddf9ae415b h1:BilPoUNotX3lcSyIZSYNS8guWjI1y24Y0s/5v8n9Hgs=\ngithub.com/Groxx/cadence-idl v0.0.0-20240822223859-88ddf9ae415b/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/OneOfOne/xxhash v1.2.2 h1:KMrpdQIwFcEqXDklaen+P1axHaj9BSKzvpUUfnHldSE=\ngithub.com/ProtonMail/gopenpgp/v2 v2.2.2/go.mod h1:ajUlBGvxMH1UBZnaYO3d1FSVzjiC6kK9XlZYGiDCvpM=\ngithub.com/ajstarks/svgo v0.0.0-20180226025133-644b8db467af h1:wVe6/Ea46ZMeNkQjjBW6xcqyQA/j5e0D6GytH95g0gQ=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751 h1:JYp7IbQjafoB+tBA3gMyHYHrpOtNuDiK/uB5uXxq5wM=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d h1:UQZhZ2O0vMHr2cI+DC1Mbh0TJxzA3RcLoMsFw+aXw7E=\ngithub.com/andybalholm/brotli v1.0.4 h1:V7DdXeJtZscaqfNuAdSRuRFzuiKlHSC/Zh3zl9qY3JY=\ngithub.com/andybalholm/brotli v1.0.4/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig=\ngithub.com/antihax/optional v1.0.0 h1:xK2lYat7ZLaVVcIuj82J8kIro4V6kDe0AUDFboUCwcg=\ngithub.com/aokoli/goutils v1.0.1 h1:7fpzNGoJ3VA8qcrm++XEE1QUe0mIwNeLa02Nwq7RDkg=\ngithub.com/apache/arrow/go/v12 v12.0.0 h1:xtZE63VWl7qLdB0JObIXvvhGjoVNrQ9ciIHG2OK5cmc=\ngithub.com/apache/arrow/go/v12 v12.0.0/go.mod h1:d+tV/eHZZ7Dz7RPrFKtPK02tpr+c9/PEd/zm8mDS9Vg=\ngithub.com/apex/logs v1.0.0 h1:adOwhOTeXzZTnVuEK13wuJNBFutP0sOfutRS8NY+G6A=\ngithub.com/aphistic/golf v0.0.0-20180712155816-02c07f170c5a h1:2KLQMJ8msqoPHIPDufkxVcoTtcmE5+1sL9950m4R9Pk=\ngithub.com/aphistic/sweet v0.2.0 h1:I4z+fAUqvKfvZV/CHi5dV0QuwbmIvYYFDjG0Ss5QpAs=\ngithub.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e h1:QEF07wC0T1rKkctt1RINW/+RMTVmiwxETico2l3gxJA=\ngithub.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6 h1:G1bPvciwNyF7IUmKXNt9Ak3m6u9DE1rF+RmtIkBpVdA=\ngithub.com/armon/go-metrics v0.4.0 h1:yCQqn7dwca4ITXb+CbubHmedzaQYHhNhrEXLYUeEe8Q=\ngithub.com/armon/go-metrics v0.4.0/go.mod h1:E6amYzXo6aW1tqzoZGT755KkbgrJsSdpwZ+3JqfkOG4=\ngithub.com/armon/go-radix v1.0.0 h1:F4z6KzEeeQIMeLFa97iZU6vupzoecKdU5TX24SNppXI=\ngithub.com/aws/aws-sdk-go-v2 v1.17.3 h1:shN7NlnVzvDUgPQ+1rLMSxY8OWRNDRYtiqe0p/PgrhY=\ngithub.com/aws/aws-sdk-go-v2 v1.18.0 h1:882kkTpSFhdgYRKVZ/VCgf7sd0ru57p2JCxz4/oN5RY=\ngithub.com/aws/aws-sdk-go-v2 v1.30.1 h1:4y/5Dvfrhd1MxRDD77SrfsDaj8kUkkljU7XE83NPV+o=\ngithub.com/aws/aws-sdk-go-v2 v1.30.1/go.mod h1:nIQjQVp5sfpQcTc9mPSr1B0PaWK5ByX9MOoDadSN4lc=\ngithub.com/aws/aws-sdk-go-v2/config v1.18.8 h1:lDpy0WM8AHsywOnVrOHaSMfpaiV2igOw8D7svkFkXVA=\ngithub.com/aws/aws-sdk-go-v2/config v1.18.25 h1:JuYyZcnMPBiFqn87L2cRppo+rNwgah6YwD3VuyvaW6Q=\ngithub.com/aws/aws-sdk-go-v2/config v1.27.23 h1:Cr/gJEa9NAS7CDAjbnB7tHYb3aLZI2gVggfmSAasDac=\ngithub.com/aws/aws-sdk-go-v2/config v1.27.23/go.mod h1:WMMYHqLCFu5LH05mFOF5tsq1PGEMfKbu083VKqLCd0o=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.13.8 h1:vTrwTvv5qAwjWIGhZDSBH/oQHuIQjGmD232k01FUh6A=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.13.24 h1:PjiYyls3QdCrzqUN35jMWtUK1vqVZ+zLfdOa/UPFDp0=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.17.23 h1:G1CfmLVoO2TdQ8z9dW+JBc/r8+MqyPQhXCafNZcXVZo=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.17.23/go.mod h1:V/DvSURn6kKgcuKEk4qwSwb/fZ2d++FFARtWSbXnLqY=\ngithub.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.12.21 h1:j9wi1kQ8b+e0FBVHxCqCGo4kxDU175hoDHcWAi0sauU=\ngithub.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.13.3 h1:jJPgroehGvjrde3XufFIJUZVK5A2L9a3KwSFgKy9n8w=\ngithub.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.9 h1:Aznqksmd6Rfv2HQN9cpqIV/lQRMaIpJkLLaJ1ZI76no=\ngithub.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.9/go.mod h1:WQr3MY7AxGNxaqAtsDWn+fBxmd4XvLkzeqQ8P1VM0/w=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.1.27 h1:I3cakv2Uy1vNmmhRQmFptYDxOvBnwCdNwyw63N0RaRU=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.1.33 h1:kG5eQilShqmJbv11XL1VpyDbaEJzWxd4zRiCG30GSn4=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.3.13 h1:5SAoZ4jYpGH4721ZNoS1znQrhOfZinOhc4XuTXx/nVc=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.3.13/go.mod h1:+rdA6ZLpaSeM7tSg/B0IEDinCIBJGmW8rKDFkYpP04g=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.4.21 h1:5NbbMrIzmUn/TXFqAle6mgrH5m9cOvMLRGL7pnG8tRE=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.4.27 h1:vFQlirhuM8lLlpI7imKOMsjdQLuN9CPi+k44F/OFVsk=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.13 h1:WIijqeaAO7TYFLbhsZmi2rgLEAtWOC1LhxCAVTJlSKw=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.13/go.mod h1:i+kbfa76PQbWw/ULoWnp51EYVWH4ENln76fLQE3lXT8=\ngithub.com/aws/aws-sdk-go-v2/internal/ini v1.3.28 h1:KeTxcGdNnQudb46oOl4d90f2I33DF/c6q3RnZAmvQdQ=\ngithub.com/aws/aws-sdk-go-v2/internal/ini v1.3.34 h1:gGLG7yKaXG02/jBlg210R7VgQIotiQntNhsCFejawx8=\ngithub.com/aws/aws-sdk-go-v2/internal/ini v1.8.0 h1:hT8rVHwugYE2lEfdFE0QWVo81lF7jMrYJVDWI+f+VxU=\ngithub.com/aws/aws-sdk-go-v2/internal/ini v1.8.0/go.mod h1:8tu/lYfQfFe6IGnaOdrpVgEL2IrrDOf6/m9RQum4NkY=\ngithub.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.11.3 h1:dT3MqvGhSoaIhRseqw2I0yH81l7wiR2vjs57O51EAm8=\ngithub.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.11.3/go.mod h1:GlAeCkHwugxdHaueRr4nhPuY+WW+gR8UjlcqzPr1SPI=\ngithub.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.9.21 h1:5C6XgTViSb0bunmU57b3CT+MhxULqHH2721FVA+/kDM=\ngithub.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.9.27 h1:0iKliEXAcCa2qVtRs7Ot5hItA2MsufrphbRFlz1Owxo=\ngithub.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.15 h1:I9zMeF107l0rJrpnHpjEiiTSCKYAIw8mALiXcPsGBiA=\ngithub.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.15/go.mod h1:9xWJ3Q/S6Ojusz1UIkfycgD1mGirJfLLKqq3LPT7WN8=\ngithub.com/aws/aws-sdk-go-v2/service/sso v1.12.0 h1:/2gzjhQowRLarkkBOGPXSRnb8sQ2RVsjdG1C/UliK/c=\ngithub.com/aws/aws-sdk-go-v2/service/sso v1.12.10 h1:UBQjaMTCKwyUYwiVnUt6toEJwGXsLBI6al083tpjJzY=\ngithub.com/aws/aws-sdk-go-v2/service/sso v1.22.1 h1:p1GahKIjyMDZtiKoIn0/jAj/TkMzfzndDv5+zi2Mhgc=\ngithub.com/aws/aws-sdk-go-v2/service/sso v1.22.1/go.mod h1:/vWdhoIoYA5hYoPZ6fm7Sv4d8701PiG5VKe8/pPJL60=\ngithub.com/aws/aws-sdk-go-v2/service/ssooidc v1.14.0 h1:Jfly6mRxk2ZOSlbCvZfKNS7TukSx1mIzhSsqZ/IGSZI=\ngithub.com/aws/aws-sdk-go-v2/service/ssooidc v1.14.10 h1:PkHIIJs8qvq0e5QybnZoG1K/9QTrLr9OsqCIo59jOBA=\ngithub.com/aws/aws-sdk-go-v2/service/ssooidc v1.26.1 h1:lCEv9f8f+zJ8kcFeAjRZsekLd/x5SAm96Cva+VbUdo8=\ngithub.com/aws/aws-sdk-go-v2/service/ssooidc v1.26.1/go.mod h1:xyFHA4zGxgYkdD73VeezHt3vSKEG9EmFnGwoKlP00u4=\ngithub.com/aws/aws-sdk-go-v2/service/sts v1.18.0 h1:kOO++CYo50RcTFISESluhWEi5Prhg+gaSs4whWabiZU=\ngithub.com/aws/aws-sdk-go-v2/service/sts v1.19.0 h1:2DQLAKDteoEDI8zpCzqBMaZlJuoE9iTYD0gFmXVax9E=\ngithub.com/aws/aws-sdk-go-v2/service/sts v1.30.1 h1:+woJ607dllHJQtsnJLi52ycuqHMwlW+Wqm2Ppsfp4nQ=\ngithub.com/aws/aws-sdk-go-v2/service/sts v1.30.1/go.mod h1:jiNR3JqT15Dm+QWq2SRgh0x0bCNSRP2L25+CqPNpJlQ=\ngithub.com/aws/smithy-go v1.13.5 h1:hgz0X/DX0dGqTYpGALqXJoRKRj5oQ7150i5FdTePzO8=\ngithub.com/aws/smithy-go v1.20.3 h1:ryHwveWzPV5BIof6fyDvor6V3iUL7nTfiTKXHiW05nE=\ngithub.com/aws/smithy-go v1.20.3/go.mod h1:krry+ya/rV9RDcV/Q16kpu6ypI4K2czasz0NC3qS14E=\ngithub.com/aybabtme/rgbterm v0.0.0-20170906152045-cc83f3b3ce59 h1:WWB576BN5zNSZc/M9d/10pqEx5VHNhaQ/yOVAkmj5Yo=\ngithub.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=\ngithub.com/bgentry/speakeasy v0.1.0 h1:ByYyxL9InA1OWqxJqqp2A5pYHUrCiAL6K3J+LKSsQkY=\ngithub.com/bketelsen/crypt v0.0.3-0.20200106085610-5cbc8cc4026c h1:+0HFd5KSZ/mm3JmhmrDukiId5iR6w4+BdFtfSy4yWIc=\ngithub.com/caarlos0/go-rpmutils v0.2.1-0.20211112020245-2cd62ff89b11/go.mod h1:je2KZ+LxaCNvCoKg32jtOIULcFogJKcL1ZWUaIBjKj0=\ngithub.com/cavaliercoder/go-cpio v0.0.0-20180626203310-925f9528c45e h1:hHg27A0RSSp2Om9lubZpiMgVbvn39bsUmW9U5h0twqc=\ngithub.com/census-instrumentation/opencensus-proto v0.4.1 h1:iKLQ0xPNFxR/2hzXZMrBo8f1j86j5WHzznCCQxV/b8g=\ngithub.com/census-instrumentation/opencensus-proto v0.4.1/go.mod h1:4T9NM4+4Vw91VeyqjLS6ao50K5bOcLKN6Q42XnYaRYw=\ngithub.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=\ngithub.com/chzyer/logex v1.1.10 h1:Swpa1K6QvQznwJRcfTfQJmTE72DqScAa40E+fbHEXEE=\ngithub.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e h1:fY5BOSpyZCqRo5OhCuC+XN+r/bBCmeuuJtjz+bCNIf8=\ngithub.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1 h1:q763qf9huN11kDQavWsoZXJNW3xEE4JJyHa5Q25/sd8=\ngithub.com/circonus-labs/circonus-gometrics v2.3.1+incompatible h1:C29Ae4G5GtYyYMm1aztcyj/J5ckgJm2zwdDajFbx1NY=\ngithub.com/circonus-labs/circonusllhist v0.1.3 h1:TJH+oke8D16535+jHExHj4nQvzlZrj7ug5D7I/orNUA=\ngithub.com/client9/misspell v0.3.4 h1:ta993UF76GwbvJcIo3Y68y/M3WxlpEHPWIGDkJYwzJI=\ngithub.com/cncf/udpa/go v0.0.0-20220112060539-c52dc94e7fbe h1:QQ3GSy+MqSHxm/d8nCtnAiZdYFd45cYZPs8vOOIYKfk=\ngithub.com/cncf/udpa/go v0.0.0-20220112060539-c52dc94e7fbe/go.mod h1:6pvJx4me5XPnfI9Z40ddWsdw2W/uZgQLFXToKeRcDiI=\ngithub.com/cncf/xds/go v0.0.0-20230607035331-e9ce68804cb4 h1:/inchEIKaYC1Akx+H+gqO04wryn5h75LSazbRlnya1k=\ngithub.com/cncf/xds/go v0.0.0-20230607035331-e9ce68804cb4/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cockroachdb/datadriven v0.0.0-20190809214429-80d97fb3cbaa h1:OaNxuTZr7kxeODyLWsRMC+OD03aFUH+mW6r2d+MWa5Y=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd h1:qMd81Ts1T2OTKmB4acZcyKaMtRnY5Y44NuXGX2GFJ1w=\ngithub.com/coreos/bbolt v1.3.2 h1:wZwiHHUieZCquLkDL0B8UhzreNWsPHooDAG3q34zk0s=\ngithub.com/coreos/etcd v3.3.13+incompatible h1:8F3hqu9fGYLBifCmRCJsicFqDx/D68Rt3q1JMazcgBQ=\ngithub.com/coreos/go-etcd v2.0.0+incompatible h1:bXhRBIXoTm9BYHS3gE0TtQuyNZyeEMux2sDi4oo5YOo=\ngithub.com/coreos/go-systemd v0.0.0-20190620071333-e64a0ec8b42a h1:W8b4lQ4tFF21aspRGoBuCNV6V2fFJBF+pm1J6OY8Lys=\ngithub.com/coreos/go-systemd/v22 v22.3.3-0.20220203105225-a9a7ef127534 h1:rtAn27wIbmOGUs7RIbVgPEjb31ehTVniDwPGXyMxm5U=\ngithub.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f h1:lBNOc5arjvs8E5mO2tbpBpLoyyu8B6e44T7hJy6potg=\ngithub.com/cpuguy83/go-md2man v1.0.10 h1:BSKMNlYxDvnunlTymqtgONjNnaRV1sTpcovwwjF22jk=\ngithub.com/creack/pty v1.1.9 h1:uDmaGzcdjhF4i/plgjmEsriH11Y0o7RKapEf/LDaM3w=\ngithub.com/dave/gopackages v0.0.0-20170318123100-46e7023ec56e h1:l99YKCdrK4Lvb/zTupt0GMPfNbncAGf8Cv/t1sYLOg0=\ngithub.com/dave/jennifer v1.2.0 h1:S15ZkFMRoJ36mGAQgWL1tnr0NQJh9rZ8qatseX/VbBc=\ngithub.com/dave/kerr v0.0.0-20170318121727-bc25dd6abe8e h1:xURkGi4RydhyaYR6PzcyHTueQudxY4LgxN1oYEPJHa0=\ngithub.com/dave/rebecca v0.9.1 h1:jxVfdOxRirbXL28vXMvUvJ1in3djwkVKXCq339qhBL0=\ngithub.com/dchest/siphash v1.2.3 h1:QXwFc8cFOR2dSa/gE6o/HokBMWtLUaNDVd+22aKHeEA=\ngithub.com/dchest/siphash v1.2.3/go.mod h1:0NvQU092bT0ipiFN++/rXm69QG9tVxLAlQHIXMPAkHc=\ngithub.com/denis-tingajkin/go-header v0.4.2 h1:jEeSF4sdv8/3cT/WY8AgDHUoItNSoEZ7qg9dX7pc218=\ngithub.com/denis-tingajkin/go-header v0.4.2/go.mod h1:eLRHAVXzE5atsKAnNRDB90WHCFFnBUn4RN0nRcs1LJA=\ngithub.com/denisenkom/go-mssqldb v0.9.0 h1:RSohk2RsiZqLZ0zCjtfn3S4Gp4exhpBWHyQ7D0yGjAk=\ngithub.com/devigned/tab v0.1.1 h1:3mD6Kb1mUOYeLpJvTVSDwSg5ZsfSxfvxGRTxRsJsITA=\ngithub.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=\ngithub.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954 h1:RMLoZVzv4GliuWafOuPuQDKSm1SJph7uCRnnS61JAn4=\ngithub.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=\ngithub.com/envoyproxy/go-control-plane v0.11.1 h1:wSUXTlLfiAQRWs2F+p+EKOY9rUyis1MyGqJ2DIk5HpM=\ngithub.com/envoyproxy/go-control-plane v0.11.1/go.mod h1:uhMcXKCQMEJHiAb0w+YGefQLaTEw+YhGluxZkrTmD0g=\ngithub.com/envoyproxy/protoc-gen-validate v1.0.2 h1:QkIBuU5k+x7/QXPvPPnWXWlCdaBFApVqftFV6k087DA=\ngithub.com/envoyproxy/protoc-gen-validate v1.0.2/go.mod h1:GpiZQP3dDbg4JouG/NNS7QWXpgx6x8QiMKdmN72jogE=\ngithub.com/fogleman/gg v1.2.1-0.20190220221249-0403632d5b90 h1:WXb3TSNmHp2vHoCroCIB1foO/yQ36swABL8aOVeDpgg=\ngithub.com/frankban/quicktest v1.14.4/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=\ngithub.com/fullstorydev/grpcurl v1.6.0 h1:p8BB6VZF8O7w6MxGr3KJ9E6EVKaswCevSALK6FBtMzA=\ngithub.com/ghodss/yaml v1.0.0 h1:wQHKEahhL6wmXdzwWG11gIVCkOv05bNOh+Rxn0yngAk=\ngithub.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=\ngithub.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=\ngithub.com/gin-gonic/gin v1.6.3 h1:ahKqKTFpO5KTPHxWZjEdPScmYaGtLo8Y4DMHoEsnp14=\ngithub.com/gin-gonic/gin v1.8.1 h1:4+fr/el88TOO3ewCmQr8cx/CtZ/umlIRIs5M4NTNjf8=\ngithub.com/gin-gonic/gin v1.8.1/go.mod h1:ji8BvRH1azfM+SYow9zQ6SZMvR8qOMZHmsCuWR9tTTk=\ngithub.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1 h1:QbL/5oDUmRBzO9/Z7Seo6zf912W/a6Sr4Eu0G/3Jho0=\ngithub.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4 h1:WtGNWLvXpe6ZudgnXrq0barxBImvnnJoMEhXAzcbM0I=\ngithub.com/go-ini/ini v1.25.4 h1:Mujh4R/dH6YL8bxuISne3xX2+qcQ9p0IxKAP6ExWoUo=\ngithub.com/go-kit/kit v0.9.0 h1:wDJmvq38kDhkVxi50ni9ykkdUr1PKgqKOoi01fa0Mdk=\ngithub.com/go-kit/log v0.1.0 h1:DGJh0Sm43HbOeYDNnVZFl8BvcYVvjD5bqYJvp0REbwQ=\ngithub.com/go-logfmt/logfmt v0.5.0 h1:TrB8swr/68K7m9CcGut2g3UOihhbcbiMAYiuTXdEih4=\ngithub.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=\ngithub.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=\ngithub.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=\ngithub.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=\ngithub.com/go-playground/locales v0.14.0 h1:u50s323jtVGugKlcYeyzC0etD1HifMjqmJqb8WugfUU=\ngithub.com/go-playground/locales v0.14.0/go.mod h1:sawfccIbzZTqEDETgFXqTho0QybSa7l++s0DH+LDiLs=\ngithub.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=\ngithub.com/go-playground/universal-translator v0.18.0 h1:82dyy6p4OuJq4/CByFNOn/jYrnRPArHwAcmLoJZxyho=\ngithub.com/go-playground/universal-translator v0.18.0/go.mod h1:UvRDBj+xPUEGrFYl+lu/H90nyDXpg0fqeB/AQUGNTVA=\ngithub.com/go-playground/validator/v10 v10.2.0 h1:KgJ0snyC2R9VXYN2rneOtQcw5aHQB1Vv0sFl1UcHBOY=\ngithub.com/go-playground/validator/v10 v10.10.0 h1:I7mrTYv78z8k8VXa/qJlOlEXn/nBh+BF8dHX5nt/dr0=\ngithub.com/go-playground/validator/v10 v10.10.0/go.mod h1:74x4gJWsvQexRdW8Pn3dXSGrTK4nAUsbPlLADvpJkos=\ngithub.com/go-redis/redis v6.15.8+incompatible h1:BKZuG6mCnRj5AOaWJXoCgf6rqTYnYJLe4en2hxT7r9o=\ngithub.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0 h1:p104kn46Q8WdvHunIJ9dAyjPVtrBPhSr3KT2yUst43I=\ngithub.com/gobuffalo/attrs v0.0.0-20190224210810-a9411de4debd h1:hSkbZ9XSyjyBirMeqSqUrK+9HboWrweVlzRNqoBi2d4=\ngithub.com/gobuffalo/depgen v0.1.0 h1:31atYa/UW9V5q8vMJ+W6wd64OaaTHUrCUXER358zLM4=\ngithub.com/gobuffalo/envy v1.7.0 h1:GlXgaiBkmrYMHco6t4j7SacKO4XUjvh5pwXh0f4uxXU=\ngithub.com/gobuffalo/flect v0.1.3 h1:3GQ53z7E3o00C/yy7Ko8VXqQXoJGLkrTQCLTF1EjoXU=\ngithub.com/gobuffalo/genny v0.1.1 h1:iQ0D6SpNXIxu52WESsD+KoQ7af2e3nCfnSBoSF/hKe0=\ngithub.com/gobuffalo/gitgen v0.0.0-20190315122116-cc086187d211 h1:mSVZ4vj4khv+oThUfS+SQU3UuFIZ5Zo6UNcvK8E8Mz8=\ngithub.com/gobuffalo/gogen v0.1.1 h1:dLg+zb+uOyd/mKeQUYIbwbNmfRsr9hd/WtYWepmayhI=\ngithub.com/gobuffalo/logger v0.0.0-20190315122211-86e12af44bc2 h1:8thhT+kUJMTMy3HlX4+y9Da+BNJck+p109tqqKp7WDs=\ngithub.com/gobuffalo/mapi v1.0.2 h1:fq9WcL1BYrm36SzK6+aAnZ8hcp+SrmnDyAxhNx8dvJk=\ngithub.com/gobuffalo/packd v0.1.0 h1:4sGKOD8yaYJ+dek1FDkwcxCHA40M4kfKgFHx8N2kwbU=\ngithub.com/gobuffalo/packr/v2 v2.2.0 h1:Ir9W9XIm9j7bhhkKE9cokvtTl1vBm62A/fene/ZCj6A=\ngithub.com/gobuffalo/syncx v0.0.0-20190224160051-33c29581e754 h1:tpom+2CJmpzAWj5/VEHync2rJGi+epHNIeRSWjzGA+4=\ngithub.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee h1:s+21KNqlpePfkah2I+gwHF8xmJWRjooY+5248k6m4A0=\ngithub.com/gobwas/pool v0.2.0 h1:QEmUOlnSjWtnpRGHF3SauEiOsy82Cup83Vf2LcMlnc8=\ngithub.com/gobwas/ws v1.0.2 h1:CoAavW/wd/kulfZmSIBt6p24n4j7tHgNVCjsfHVNUbo=\ngithub.com/goccy/go-json v0.9.7/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=\ngithub.com/goccy/go-json v0.9.11 h1:/pAaQDLHEoCq/5FFmSKBswWmK6H0e8g4159Kc/X/nqk=\ngithub.com/goccy/go-json v0.9.11/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=\ngithub.com/godbus/dbus/v5 v5.0.4 h1:9349emZab16e7zQvpmsbtjc18ykshndd8y2PG3sgJbA=\ngithub.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe h1:lXe2qZdvpiX5WZkZR4hgp4KJVfY3nMkvmwbVkpv1rVY=\ngithub.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g=\ngithub.com/golang/glog v1.1.2 h1:DVjP2PbBOzHyzA+dn3WhHIq4NdVu3Q+pvivFICf/7fo=\ngithub.com/golang/glog v1.1.2/go.mod h1:zR+okUeTbrL6EL3xHUDxZuEtGv04p5shwip1+mL/rLQ=\ngithub.com/golang/groupcache v0.0.0-20190702054246-869f871628b6 h1:ZgQEtGgCBiWRM39fZuwSd1LwSqqSW0hOdXCYYDX0R3I=\ngithub.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=\ngithub.com/google/btree v1.0.0 h1:0udJVsspx3VBr5FwtLhQQtuAsVc79tTq0ocGIPAU6qo=\ngithub.com/google/certificate-transparency-go v1.1.1 h1:6JHXZhXEvilMcTjR4MGZn5KV0IRkcFl4CJx5iHVhjFE=\ngithub.com/google/flatbuffers v2.0.8+incompatible h1:ivUb1cGomAB101ZM1T0nOiWz9pSrTMoa9+EiY7igmkM=\ngithub.com/google/flatbuffers v2.0.8+incompatible/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8=\ngithub.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=\ngithub.com/google/go-pkcs11 v0.2.0 h1:5meDPB26aJ98f+K9G21f0AqZwo/S5BJMJh8nuhMbdsI=\ngithub.com/google/go-pkcs11 v0.2.0/go.mod h1:6eQoGcuNJpa7jnd5pMGdkSaQpNDYvPlXWMcjXXThLlY=\ngithub.com/google/go-pkcs11 v0.2.1-0.20230907215043-c6f79328ddf9 h1:OF1IPgv+F4NmqmJ98KTjdN97Vs1JxDPB3vbmYzV2dpk=\ngithub.com/google/go-pkcs11 v0.2.1-0.20230907215043-c6f79328ddf9/go.mod h1:6eQoGcuNJpa7jnd5pMGdkSaQpNDYvPlXWMcjXXThLlY=\ngithub.com/google/pprof v0.0.0-20181127221834-b4f47329b966 h1:zpjeU3rN5R22t0iguDarIAL75+2acLnDqGLOiPttMjk=\ngithub.com/google/pprof v0.0.0-20210720184732-4bb14d4b1be1 h1:K6RDEckDVWvDI9JAJYCmNdQXq6neHJOYx3V6jnqNEec=\ngithub.com/google/renameio v0.1.0 h1:GOZbcHa3HfsPKPlmyPyN2KEohoMXOhdMbHrvbpl2QaA=\ngithub.com/google/rpmpack v0.0.0-20210410105602-e20c988a6f5a h1:XC048Fc/OB2rUl/BxruopEl2u/EP6cJNFveVxI1cvdk=\ngithub.com/google/s2a-go v0.1.7 h1:60BLSyTrOV4/haCDW4zb1guZItoSq8foHCXrAnjBo/o=\ngithub.com/google/s2a-go v0.1.7/go.mod h1:50CgR4k1jNlWBu4UfS4AcfhVe1r6pdZPygJ3R8F0Qdw=\ngithub.com/google/subcommands v1.0.1 h1:/eqq+otEXm5vhfBrbREPCSVQbvofip6kIz+mX5TUH7k=\ngithub.com/google/trillian v1.3.11 h1:pPzJPkK06mvXId1LHEAJxIegGgHzzp/FUnycPYfoCMI=\ngithub.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.3.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.4.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.5 h1:UR4rDjcgpgEnqpIEvkiqTYKBCKLNmlge2eVjoZfySzM=\ngithub.com/googleapis/enterprise-certificate-proxy v0.2.5/go.mod h1:RxW0N9901Cko1VOCW3SXCpWP+mlIEkk2tP7jnHy9a3w=\ngithub.com/googleapis/gax-go/v2 v2.10.0/go.mod h1:4UOEnMCrxsSqQ940WnTiD6qJ63le2ev3xfyagutxiPw=\ngithub.com/googleapis/google-cloud-go-testing v0.0.0-20200911160855-bcd43fbb19e8 h1:tlyzajkF3030q6M8SvmJSemC9DTHL/xaMa18b65+JM4=\ngithub.com/gookit/color v1.5.0 h1:1Opow3+BWDwqor78DcJkJCIwnkviFi+rrOANki9BUFw=\ngithub.com/gorhill/cronexpr v0.0.0-20180427100037-88b0669f7d75 h1:f0n1xnMSmBLzVfsMMvriDyA75NB/oBgILX2GcHXIQzY=\ngithub.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=\ngithub.com/gorilla/securecookie v1.1.1 h1:miw7JPhV+b/lAHSXz4qd/nN9jRiAFV5FwjeKyCS8BvQ=\ngithub.com/gorilla/sessions v1.2.1 h1:DHd3rPN5lE3Ts3D8rKkQ8x/0kqfeNmBAaiSi+o7FsgI=\ngithub.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=\ngithub.com/gostaticanalysis/testutil v0.4.0/go.mod h1:bLIoPefWXrRi/ssLFWX1dx7Repi5x3CuviD3dgAZaBU=\ngithub.com/gregjones/httpcache v0.0.0-20190611155906-901d90724c79 h1:+ngKgrYPPJrOjhax5N+uePQ0Fh1Z7PheYoUI/0nzkPA=\ngithub.com/grpc-ecosystem/go-grpc-middleware v1.2.2 h1:FlFbCRLd5Jr4iYXZufAvgWN6Ao0JrI5chLINnUXDDr0=\ngithub.com/grpc-ecosystem/go-grpc-prometheus v1.2.0 h1:Ovs26xHkKqVztRpIrF/92BcuyuQ/YW4NSIpoGtfXNho=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0 h1:gmcG1KaJ57LophUzW0Hy8NmPhnMZb4M0+kPpLofRdBo=\ngithub.com/grpc-ecosystem/grpc-gateway/v2 v2.11.3 h1:lLT7ZLSzGLI08vc9cpd+tYmNWjdKDqyr/2L+f6U12Fk=\ngithub.com/grpc-ecosystem/grpc-gateway/v2 v2.11.3/go.mod h1:o//XUCC/F+yRGJoPO/VU0GSB0f8Nhgmxx0VIRUvaC0w=\ngithub.com/hashicorp/consul/api v1.18.0 h1:R7PPNzTCeN6VuQNDwwhZWJvzCtGSrNpJqfb22h3yH9g=\ngithub.com/hashicorp/consul/api v1.18.0/go.mod h1:owRRGJ9M5xReDC5nfT8FTJrNAPbT4NM6p/k+d03q2v4=\ngithub.com/hashicorp/consul/sdk v0.8.0 h1:OJtKBtEjboEZvG6AOUdh4Z1Zbyu0WcxQ0qatRrZHTVU=\ngithub.com/hashicorp/go-immutable-radix v1.3.1 h1:DKHmCUm2hRBK510BaiZlwvpD40f8bJFeZnpfm2KLowc=\ngithub.com/hashicorp/go-msgpack v0.5.3 h1:zKjpN5BK/P5lMYrLmBHdBULWbJ0XpYR+7NGzqkZzoD4=\ngithub.com/hashicorp/go-rootcerts v1.0.2 h1:jzhAVGtqPKbwpyCPELlgNWhE1znq+qwJtW5Oi2viEzc=\ngithub.com/hashicorp/go-sockaddr v1.0.0 h1:GeH6tui99pF4NJgfnhp+L6+FfobzVW3Ah46sLo0ICXs=\ngithub.com/hashicorp/go-syslog v1.0.0 h1:KaodqZuhUoZereWVIYmpUgZysurB1kBLX2j0MwMrUAE=\ngithub.com/hashicorp/go.net v0.0.1 h1:sNCoNyDEvN1xa+X0baata4RdcpKwcMS6DH+xwfqPgjw=\ngithub.com/hashicorp/golang-lru v0.5.4 h1:YDjusn29QI/Das2iO9M0BHnIbxPeyuCHsjMW+lJfyTc=\ngithub.com/hashicorp/logutils v1.0.0 h1:dLEQVugN8vlakKOUE3ihGLTZJRB4j+M2cdTm/ORI65Y=\ngithub.com/hashicorp/mdns v1.0.4 h1:sY0CMhFmjIPDMlTB+HfymFHCaYLhgifZ0QhjaYKD/UQ=\ngithub.com/hashicorp/memberlist v0.3.0 h1:8+567mCcFDnS5ADl7lrpxPMWiFCElyUEeW0gtj34fMA=\ngithub.com/hashicorp/serf v0.10.1 h1:Z1H2J60yRKvfDYAOZLd2MU0ND4AH/WDz7xYHDWQsIPY=\ngithub.com/hashicorp/serf v0.10.1/go.mod h1:yL2t6BqATOLGc5HF7qbFkTfXoPIY0WZdWHfEvMqbG+4=\ngithub.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=\ngithub.com/iancoleman/orderedmap v0.0.0-20190318233801-ac98e3ecb4b0 h1:i462o439ZjprVSFSZLZxcsoAe592sZB1rci2Z8j4wdk=\ngithub.com/iancoleman/orderedmap v0.0.0-20190318233801-ac98e3ecb4b0/go.mod h1:N0Wam8K1arqPXNWjMo21EXnBPOPp36vB07FNRdD2geA=\ngithub.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6 h1:UDMh68UUwekSh5iP2OMhRRZJiiBccgV7axzUG8vi56c=\ngithub.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639 h1:mV02weKRL81bEnm8A0HT1/CAelMQDBuQIfLw8n+d6xI=\ngithub.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM=\ngithub.com/invopop/jsonschema v0.7.0 h1:2vgQcBz1n256N+FpX3Jq7Y17AjYt46Ig3zIWyy770So=\ngithub.com/invopop/jsonschema v0.7.0/go.mod h1:O9uiLokuu0+MGFlyiaqtWxwqJm41/+8Nj0lD7A36YH0=\ngithub.com/jhump/protoreflect v1.6.1 h1:4/2yi5LyDPP7nN+Hiird1SAJ6YoxUm13/oxHGRnbPd8=\ngithub.com/joho/godotenv v1.3.0 h1:Zjp+RcGpHhGlrMbJzXTrZZPrWj+1vfm90La1wgB6Bhc=\ngithub.com/jpillora/backoff v1.0.0 h1:uvFg412JmmHBHw7iwprIxkPMI+sGQ4kzOWsMeHnm2EA=\ngithub.com/json-iterator/go v1.1.11 h1:uVUAXhF2To8cbw/3xN3pxj6kk7TYKs98NIrTqPlMWAQ=\ngithub.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=\ngithub.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=\ngithub.com/jstemmer/go-junit-report v0.9.1 h1:6QPYqodiu3GuPL+7mfx+NwDdp2eTkp9IfEUpgAwUN0o=\ngithub.com/juju/ratelimit v1.0.1 h1:+7AIFJVQ0EQgq/K9+0Krm7m530Du7tIz0METWzN0RgY=\ngithub.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=\ngithub.com/jung-kurt/gofpdf v1.0.3-0.20190309125859-24315acbbda5 h1:PJr+ZMXIecYc1Ey2zucXdR73SMBtgjPgwa31099IMv0=\ngithub.com/k0kubun/colorstring v0.0.0-20150214042306-9440f1994b88 h1:uC1QfSlInpQF+M0ao65imhwqKnz3Q2z/d8PWZRMQvDM=\ngithub.com/karrick/godirwalk v1.10.3 h1:lOpSw2vJP0y5eLBW906QwKsUK/fe/QDyoqM5rnnuPDY=\ngithub.com/kisielk/gotool v1.0.0 h1:AV2c/EiW3KqPNT9ZKl07ehoAGi4C5/01Cfbblndcapg=\ngithub.com/klauspost/asmfmt v1.3.2 h1:4Ri7ox3EwapiOjCki+hw14RyKk201CN4rzyCJRFLpK4=\ngithub.com/klauspost/asmfmt v1.3.2/go.mod h1:AG8TuvYojzulgDAMCnYn50l/5QV3Bs/tp6j0HLHbNSE=\ngithub.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4=\ngithub.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=\ngithub.com/kr/fs v0.1.0 h1:Jskdu9ieNAYnjxsi0LbQp1ulIKZV1LAFgK1tWhpZgl8=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515 h1:T+h1c/A9Gawja4Y9mFVWj2vyii2bbUNDw3kt9VxK2EY=\ngithub.com/kr/pty v1.1.1 h1:VkoXIwSboBpnk99O/KFauAEILuNHv5DVFKZMBN/gUgw=\ngithub.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc=\ngithub.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=\ngithub.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w=\ngithub.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY=\ngithub.com/letsencrypt/pkcs11key/v4 v4.0.0 h1:qLc/OznH7xMr5ARJgkZCCWk+EomQkiNTOoOF5LAgagc=\ngithub.com/lib/pq v1.10.4 h1:SO9z7FRPzA03QhHKJrH5BXA6HU1rS4V2nIVrrNC1iYk=\ngithub.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 h1:6E+4a0GO5zZEnZ81pIr0yLvtUWk2if982qA3F3QD6H4=\ngithub.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0/go.mod h1:zJYVVT2jmtg6P3p1VtQj7WsuWi/y4VnjVBn7F8KPB3I=\ngithub.com/lyft/protoc-gen-star/v2 v2.0.3 h1:/3+/2sWyXeMLzKd1bX+ixWKgEMsULrIivpDsuaF441o=\ngithub.com/lyft/protoc-gen-star/v2 v2.0.3/go.mod h1:amey7yeodaJhXSbf/TlLvWiqQfLOSpEk//mLlc+axEk=\ngithub.com/markbates/oncer v0.0.0-20181203154359-bf2de49a0be2 h1:JgVTCPf0uBVcUSWpyXmGpgOc62nK5HWUBKAGc3Qqa5k=\ngithub.com/markbates/safe v1.0.1 h1:yjZkbvRM6IzKj9tlu/zMJLS0n/V351OZWRnF3QfaUxI=\ngithub.com/mattn/go-shellwords v1.0.10 h1:Y7Xqm8piKOO3v10Thp7Z36h4FYFjt5xB//6XvOrs2Gw=\ngithub.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4=\ngithub.com/miekg/dns v1.1.41 h1:WMszZWJG0XmzbK9FEmzH2TVcqYzFesusSIB41b8KHxY=\ngithub.com/miekg/pkcs11 v1.0.3 h1:iMwmD7I5225wv84WxIG/bmxz9AXjWvTWIbM/TYHvWtw=\ngithub.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8 h1:AMFGa4R4MiIpspGNG7Z948v4n35fFGB3RR3G/ry4FWs=\ngithub.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8/go.mod h1:mC1jAcsrzbxHt8iiaC+zU4b1ylILSosueou12R++wfY=\ngithub.com/minio/c2goasm v0.0.0-20190812172519-36a3d3bbc4f3 h1:+n/aFZefKZp7spd8DFdX7uMikMLXX4oubIzJF4kv/wI=\ngithub.com/minio/c2goasm v0.0.0-20190812172519-36a3d3bbc4f3/go.mod h1:RagcQ7I8IeTMnF8JTXieKnO4Z6JCsikNEzj0DwauVzE=\ngithub.com/mitchellh/cli v1.1.0 h1:tEElEatulEHDeedTxwckzyYMA5c86fbmNIUL1hBIiTg=\ngithub.com/mitchellh/go-ps v1.0.0 h1:i6ampVEEF4wQFF+bkYfwYgY+F/uYJDktmvLPf7qIgjc=\ngithub.com/mitchellh/go-ps v1.0.0/go.mod h1:J4lOc8z8yJs6vUwklHw2XEIiT4z4C40KtWVN3nvg8Pg=\ngithub.com/mitchellh/go-testing-interface v1.0.0 h1:fzU/JVNcaqHQEcVFAKeR41fkiLdIPrefOvVG1VZ96U0=\ngithub.com/mitchellh/gox v0.4.0 h1:lfGJxY7ToLJQjHHwi0EX6uYBdK78egf954SQl13PQJc=\ngithub.com/mitchellh/iochan v1.0.0 h1:C+X3KsSTLFVBr/tK1eYN/vs4rJcvsiLU338UhYPJWeY=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=\ngithub.com/modern-go/reflect2 v1.0.1 h1:9f412s+6RmYXLWZSEzVVgPGK7C2PphHj5RJrvfx9AWI=\ngithub.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=\ngithub.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=\ngithub.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw=\ngithub.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6fbMZ9s0scYfZQ84/6SPL6zC8ACM2oIL0=\ngithub.com/mozilla/scribe v0.0.0-20180711195314-fb71baf557c1 h1:29NKShH4TWd3lxCDUhS4Xe16EWMA753dtIxYtwddklU=\ngithub.com/mozilla/tls-observatory v0.0.0-20210609171429-7bc42856d2e5 h1:0KqC6/sLy7fDpBdybhVkkv4Yz+PmB7c9Dz9z3dLW804=\ngithub.com/muesli/mango v0.1.0 h1:DZQK45d2gGbql1arsYA4vfg4d7I9Hfx5rX/GCmzsAvI=\ngithub.com/muesli/mango v0.1.0/go.mod h1:5XFpbC8jY5UUv89YQciiXNlbi+iJgt29VDC5xbzrLL4=\ngithub.com/muesli/mango-cobra v1.2.0 h1:DQvjzAM0PMZr85Iv9LIMaYISpTOliMEg+uMFtNbYvWg=\ngithub.com/muesli/mango-cobra v1.2.0/go.mod h1:vMJL54QytZAJhCT13LPVDfkvCUJ5/4jNUKF/8NC2UjA=\ngithub.com/muesli/mango-pflag v0.1.0 h1:UADqbYgpUyRoBja3g6LUL+3LErjpsOwaC9ywvBWe7Sg=\ngithub.com/muesli/mango-pflag v0.1.0/go.mod h1:YEQomTxaCUp8PrbhFh10UfbhbQrM/xJ4i2PB8VTLLW0=\ngithub.com/muesli/roff v0.1.0 h1:YD0lalCotmYuF5HhZliKWlIx7IEhiXeSfq7hNjFqGF8=\ngithub.com/muesli/roff v0.1.0/go.mod h1:pjAHQM9hdUUwm/krAfrLGgJkXJ+YuhtsfZ42kieB2Ig=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f h1:KUppIJq7/+SVif2QVs3tOP0zanoHgBEVAwHxUSIzRqU=\ngithub.com/mwitkow/go-proto-validators v0.2.0 h1:F6LFfmgVnfULfaRsQWBbe7F7ocuHCr9+7m+GAeDzNbQ=\ngithub.com/ncruces/sort v0.1.2 h1:zKQ9CA4fpHPF6xsUhRTfi5EEryspuBpe/QA4VWQOV1U=\ngithub.com/ncruces/sort v0.1.2/go.mod h1:vEJUTBJtebIuCMmXD18GKo5GJGhsay+xZFOoBEIXFmE=\ngithub.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs=\ngithub.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=\ngithub.com/oklog/ulid v1.3.1 h1:EGfNDEx6MqHz8B3uNV6QAib1UR2Lm97sHi3ocA6ESJ4=\ngithub.com/op/go-logging v0.0.0-20160315200505-970db520ece7 h1:lDH9UUVJtmYCjyT0CI4q8xvlXPxeZ0gYCVvWbmPlp88=\ngithub.com/otiai10/curr v1.0.0 h1:TJIWdbX0B+kpNagQrjgq8bCMrbhiuX73M2XwgtDMoOI=\ngithub.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY=\ngithub.com/pelletier/go-toml v1.7.0 h1:7utD74fnzVc/cpcyy8sjrlFr5vYpypUixARcHIMIGuI=\ngithub.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=\ngithub.com/pelletier/go-toml/v2 v2.0.1 h1:8e3L2cCQzLFi2CR4g7vGFuFxX7Jl1kKX8gW+iV0GUKU=\ngithub.com/pelletier/go-toml/v2 v2.0.1/go.mod h1:r9LEWfGN8R5k0VXJ+0BkIe7MYkRdwZOjgMj2KwnJFUo=\ngithub.com/peterbourgon/diskv v2.0.1+incompatible h1:UBdAOUP5p4RWqPBg048CAvpKN+vxiaj6gdUUzhl4XmI=\ngithub.com/pierrec/lz4/v4 v4.1.15 h1:MO0/ucJhngq7299dKLwIMtgTfbkoSPF6AoMYDd8Q4q0=\ngithub.com/pierrec/lz4/v4 v4.1.15/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=\ngithub.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e h1:aoZm08cpOy4WuID//EZDgcC4zIxODThtZNPirFr42+A=\ngithub.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=\ngithub.com/pkg/sftp v1.13.1 h1:I2qBYMChEhIjOgazfJmV3/mZM256btk6wkCDRmW7JYs=\ngithub.com/posener/complete v1.2.3 h1:NP0eAhjcjImqslEwo/1hq7gpajME0fTLTezBKDqfXqo=\ngithub.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c h1:ncq/mPwQF4JjgDlrVEn3C11VoGHZN7m8qihwgMEtzYw=\ngithub.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=\ngithub.com/prometheus/tsdb v0.7.1 h1:YZcsG11NqnK4czYLrWd9mpEuAJIHVQLwdrleYfszMAA=\ngithub.com/psanford/httpreadat v0.1.0 h1:VleW1HS2zO7/4c7c7zNl33fO6oYACSagjJIyMIwZLUE=\ngithub.com/psanford/httpreadat v0.1.0/go.mod h1:Zg7P+TlBm3bYbyHTKv/EdtSJZn3qwbPwpfZ/I9GKCRE=\ngithub.com/pseudomuto/protoc-gen-doc v1.3.2 h1:61vWZuxYa8D7Rn4h+2dgoTNqnluBmJya2MgbqO32z6g=\ngithub.com/pseudomuto/protokit v0.2.0 h1:hlnBDcy3YEDXH7kc9gV+NLaN0cDzhDvD1s7Y6FZ8RpM=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.19 h1:5+KTKb2YREUYiqZFEIuifFyBxlcCUPWgNZkWy71XS0Q=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.19/go.mod h1:KeCP03KrjuSO0H1kTuZQCWlQPulDV6YMIXmpQss17rU=\ngithub.com/quasilyte/go-ruleguard/rules v0.0.0-20211022131956-028d6511ab71 h1:CNooiryw5aisadVfzneSZPswRWvnVW8hF1bS/vo8ReI=\ngithub.com/remyoudompheng/go-dbus v0.0.0-20121104212943-b7232d34b1d5 h1:CvqZS4QYHBRvx7AeFdimd16HCbLlYsvQMcKDACpJW/c=\ngithub.com/remyoudompheng/go-dbus v0.0.0-20121104212943-b7232d34b1d5/go.mod h1:+u151txRmLpwxBmpYn9z3d1sdJdjRPQpsXuYeY9jNls=\ngithub.com/remyoudompheng/go-liblzma v0.0.0-20190506200333-81bf2d431b96 h1:J8J/cgLDRuqXJnwIrRDBvtl+LLsdg7De74znW/BRRq4=\ngithub.com/remyoudompheng/go-liblzma v0.0.0-20190506200333-81bf2d431b96/go.mod h1:90HvCY7+oHHUKkbeMCiHt1WuFR2/hPJ9QrljDG+v6ls=\ngithub.com/remyoudompheng/go-misc v0.0.0-20190427085024-2d6ac652a50e h1:eTWZyPUnHcuGRDiryS/l2I7FfKjbU3IBx3IjqHPxuKU=\ngithub.com/remyoudompheng/go-misc v0.0.0-20190427085024-2d6ac652a50e/go.mod h1:80FQABjoFzZ2M5uEa6FUaJYEmqU2UOKojlFVak1UAwI=\ngithub.com/rogpeppe/fastuuid v1.2.0 h1:Ppwyp6VYCF1nvBTXL3trRso7mXMlRrw9ooo375wvi2s=\ngithub.com/rs/cors v1.7.0 h1:+88SsELBHx5r+hZ8TCkggzSstaWNbDvThkVK8H6f9ik=\ngithub.com/rs/xid v1.4.0 h1:qd7wPTDkN6KQx2VmMBLrpHkiyQwgFXRnkOLacUiaSNY=\ngithub.com/russross/blackfriday v1.5.2 h1:HyvC0ARfnZBqnXwABFeSZHpKvJHJJfPz81GNueLj0oo=\ngithub.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f h1:UFr9zpz4xgTnIE5yIMtWAMngCdZ9p/+q6lTbgelo80M=\ngithub.com/sagikazarmark/crypt v0.9.0 h1:fipzMFW34hFUEc4D7fsLQFtE7yElkpgyS2zruedRdZk=\ngithub.com/sagikazarmark/crypt v0.9.0/go.mod h1:RnH7sEhxfdnPm1z+XMgSLjWTEIjyK4z2dw6+4vHTMuo=\ngithub.com/sassoftware/go-rpmutils v0.0.0-20190420191620-a8f1baeba37b h1:+gCnWOZV8Z/8jehJ2CdqB47Z3S+SREmQcuXkRFLNsiI=\ngithub.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529 h1:nn5Wsu0esKSJiIVhscUtVbo7ada43DJhG55ua/hjS5I=\ngithub.com/shirou/gopsutil/v3 v3.22.4 h1:srAQaiX6jX/cYL6q29aE0m8lOskT9CurZ9N61YR3yoI=\ngithub.com/shirou/gopsutil/v3 v3.22.4/go.mod h1:D01hZJ4pVHPpCTZ3m3T2+wDF2YAGfd+H4ifUguaQzHM=\ngithub.com/shurcooL/go v0.0.0-20180423040247-9e1955d9fb6e h1:MZM7FHLqUHYI0Y/mQAt3d2aYa0SiNms/hFqC9qJYolM=\ngithub.com/shurcooL/go-goon v0.0.0-20170922171312-37c2f522c041 h1:llrF3Fs4018ePo4+G/HV/uQUqEI1HMDjCeOf2V6puPc=\ngithub.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=\ngithub.com/smartystreets/assertions v1.1.1 h1:T/YLemO5Yp7KPzS+lVtu+WsHn8yoSwTfItdAd1r3cck=\ngithub.com/smartystreets/assertions v1.2.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=\ngithub.com/smartystreets/go-aws-auth v0.0.0-20180515143844-0c1422d1fdb9 h1:hp2CYQUINdZMHdvTdXtPOY2ainKl4IoMcpAXEf2xj3Q=\ngithub.com/smartystreets/goconvey v1.7.2/go.mod h1:Vw0tHAZW6lzCRk3xgdin6fKYcG+G3Pg9vgXWeJpQFMM=\ngithub.com/smartystreets/gunit v1.4.2 h1:tyWYZffdPhQPfK5VsMQXfauwnJkqg7Tv5DLuQVYxq3Q=\ngithub.com/soheilhy/cmux v0.1.4 h1:0HKaf1o97UwFjHH9o5XsHUOF+tqmdA7KEzXLpiyaw0E=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72 h1:qLC7fQah7D6K1B0ujays3HV9gkFtllcxhzImRR7ArPQ=\ngithub.com/spf13/afero v1.3.3 h1:p5gZEKLYoL7wh8VrJesMaYeNxdEd1v3cb4irOk9zB54=\ngithub.com/spf13/afero v1.3.3/go.mod h1:5KUK8ByomD5Ti5Artl0RtHeI5pTF7MIDuXL3yY520V4=\ngithub.com/spf13/cobra v0.0.3 h1:ZlrZ4XsMRm04Fr5pSFxBgfND2EBVa1nLpiy1stUsX/8=\ngithub.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg=\ngithub.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=\ngithub.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=\ngithub.com/taylanisikdemir/cadence-idl v0.0.0-20250604205405-c829a7dc2e0c h1:EtJglCmzRnUbKX67v9svqAnYlrgPwCE+PUnG1kB2ooY=\ngithub.com/taylanisikdemir/cadence-idl v0.0.0-20250604205405-c829a7dc2e0c/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716221550-3529a2618736 h1:UoaE2FX56QyLA0VGMJ16cll3haA5aZaFnfw5ZFDBzVU=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716221550-3529a2618736/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716224349-f9e143d54910 h1:cxtZkrE5AumMJDXFKVOUG+Q0GSHHM05t90/NKFdt+6E=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716224349-f9e143d54910/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716230216-2e5d5a0163bb h1:fxnwcw5UF0BT3goT+6avQ3sRddDSmwpN+x1h3enQj9k=\ngithub.com/timl3136/cadence-idl v0.0.0-20240716230216-2e5d5a0163bb/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/tj/go-buffer v1.1.0 h1:Lo2OsPHlIxXF24zApe15AbK3bJLAOvkkxEA6Ux4c47M=\ngithub.com/tj/go-elastic v0.0.0-20171221160941-36157cbbebc2 h1:eGaGNxrtoZf/mBURsnNQKDR7u50Klgcf2eFDQEnc8Bc=\ngithub.com/tj/go-kinesis v0.0.0-20171128231115-08b17f58cb1b h1:m74UWYy+HBs+jMFR9mdZU6shPewugMyH5+GV6LNgW8w=\ngithub.com/tj/go-spin v1.1.0 h1:lhdWZsvImxvZ3q1C5OIB7d72DuOwP4O2NdBg9PyzNds=\ngithub.com/tklauser/go-sysconf v0.3.10 h1:IJ1AZGZRWbY8T5Vfk04D9WOA5WSejdflXxP03OUqALw=\ngithub.com/tklauser/go-sysconf v0.3.10/go.mod h1:C8XykCvCb+Gn0oNCWPIlcb0RuglQTYaQ2hGm7jmxEFk=\ngithub.com/tklauser/numcpus v0.4.0 h1:E53Dm1HjH1/R2/aoCtXtPgzmElmn51aOkhCFSuZq//o=\ngithub.com/tklauser/numcpus v0.4.0/go.mod h1:1+UI3pD8NW14VMwdgJNJ1ESk2UnwhAnz5hMwiKKqXCQ=\ngithub.com/tmc/grpc-websocket-proxy v0.0.0-20200427203606-3cfed13b9966 h1:j6JEOq5QWFker+d7mFQYOhjTZonQ7YkLTHm56dbn+yM=\ngithub.com/tomasen/realip v0.0.0-20180522021738-f0c99a92ddce h1:fb190+cK2Xz/dvi9Hv8eCYJYvIGUTN2/KLq1pT6CjEc=\ngithub.com/tv42/httpunix v0.0.0-20150427012821-b75d8614f926 h1:G3dpKMzFDjgEh2q1Z7zUUtKa8ViPtH+ocF0bE0g00O8=\ngithub.com/twitchtv/twirp v5.8.0+incompatible h1:DTfGS9u/jHbo34cBB+qhzVHRaAq+tRois71j8pvjQ5M=\ngithub.com/uber/cadence-idl v0.0.0-20250611195003-65a2b63c543d h1:jR6NMYjgXor7lwiMPOkH+MQvBkqa/q7QcBrVQJ804JY=\ngithub.com/uber/cadence-idl v0.0.0-20250611195003-65a2b63c543d/go.mod h1:oyUK7GCNCRHCCyWyzifSzXpVrRYVBbAMHAzF5dXiKws=\ngithub.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo=\ngithub.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=\ngithub.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0=\ngithub.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY=\ngithub.com/urfave/cli/v2 v2.3.0 h1:qph92Y649prgesehzOrQjdWyxFOp/QVM+6imKHad91M=\ngithub.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=\ngithub.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=\ngithub.com/valyala/quicktemplate v1.7.0 h1:LUPTJmlVcb46OOUY3IeD9DojFpAVbsG+5WFTcjMJzCM=\ngithub.com/valyala/quicktemplate v1.7.0/go.mod h1:sqKJnoaOF88V07vkO+9FL8fb9uZg/VPSJnLYn+LmLk8=\ngithub.com/viki-org/dnscache v0.0.0-20130720023526-c70c1f23c5d8 h1:EVObHAr8DqpoJCVv6KYTle8FEImKhtkfcZetNqxDoJQ=\ngithub.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 h1:eY9dn8+vbi4tKz5Qo6v2eYzo7kUS51QINcR5jNpbZS8=\ngithub.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778 h1:QldyIu/L63oPpyvQmHgvgickp1Yw510KJOqX7H24mg8=\ngithub.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77 h1:ESFSdwYZvkeru3RtdrYueztKhOBCSAAzS4Gf+k0tEow=\ngithub.com/yudai/gojsondiff v1.0.0 h1:27cbfqXLVEJ1o8I6v3y9lg8Ydm53EKqHXAOMxEGlCOA=\ngithub.com/yudai/golcs v0.0.0-20170316035057-ecda9a501e82 h1:BHyfKlQyqbsFN5p3IfnEUduWvb9is428/nNb5L3U01M=\ngithub.com/yudai/pp v2.0.1+incompatible h1:Q4//iY4pNF6yPLZIigmvcl7k/bPgrcTPIFIcmawg5bI=\ngithub.com/yuin/goldmark v1.4.13 h1:fVcFKWvrslecOb/tg+Cc05dkeYx540o0FuFt3nUVDoE=\ngithub.com/yusufpapurcu/wmi v1.2.2 h1:KBNDSne4vP5mbSWnJbO+51IMOXJB67QiYCSBrubbPRg=\ngithub.com/yusufpapurcu/wmi v1.2.2/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=\ngithub.com/zeebo/xxh3 v1.0.2 h1:xZmwmqxHZA8AI603jOQ0tMqmBr9lPeFwGg6d+xy9DC0=\ngithub.com/zeebo/xxh3 v1.0.2/go.mod h1:5NWz9Sef7zIDm2JHfFlcQvNekmcEl9ekUZQQKCYaDcA=\ngo.etcd.io/bbolt v1.3.4 h1:hi1bXHMVrlQh6WwxAy+qZCV/SYIlqo+Ushwdpa4tAKg=\ngo.etcd.io/etcd v0.0.0-20200513171258-e048e166ab9c h1:/RwRVN9EdXAVtdHxP7Ndn/tfmM9/goiwU0QTnLBgS4w=\ngo.etcd.io/etcd/api/v3 v3.5.6 h1:Cy2qx3npLcYqTKqGJzMypnMv2tiRyifZJ17BlWIWA7A=\ngo.etcd.io/etcd/api/v3 v3.5.6/go.mod h1:KFtNaxGDw4Yx/BA4iPPwevUTAuqcsPxzyX8PHydchN8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.6 h1:TXQWYceBKqLp4sa87rcPs11SXxUA/mHwH975v+BDvLU=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.6/go.mod h1:ggrwbk069qxpKPq8/FKkQ3Xq9y39kbFR4LnKszpRXeQ=\ngo.etcd.io/etcd/client/v2 v2.305.6 h1:fIDR0p4KMjw01MJMfUIDWdQbjo06PD6CeYM5z4EHLi0=\ngo.etcd.io/etcd/client/v2 v2.305.6/go.mod h1:BHha8XJGe8vCIBfWBpbBLVZ4QjOIlfoouvOwydu63E0=\ngo.etcd.io/etcd/client/v3 v3.5.6 h1:coLs69PWCXE9G4FKquzNaSHrRyMCAXwF+IX1tAPVO8E=\ngo.etcd.io/etcd/client/v3 v3.5.6/go.mod h1:f6GRinRMCsFVv9Ht42EyY7nfsVGwrNO0WEoS2pRKzQk=\ngo.mozilla.org/mozlog v0.0.0-20170222151521-4bb13139d403 h1:rKyWXYDfrVOpMFBion4Pmx5sJbQreQNXycHvm4KwJSg=\ngo.opencensus.io v0.22.4 h1:LYy1Hy3MJdrCdMwwzxA/dRok4ejH+RwNGbuoD9fCjto=\ngo.opentelemetry.io/otel v1.0.1 h1:4XKyXmfqJLOQ7feyV5DB6gsBFZ0ltB8vLtp6pj4JIcc=\ngo.opentelemetry.io/otel v1.0.1/go.mod h1:OPEOD4jIT2SlZPMmwT6FqZz2C0ZNdQqiWcoK6M0SNFU=\ngo.opentelemetry.io/otel/trace v1.0.1 h1:StTeIH6Q3G4r0Fiw34LTokUFESZgIDUr0qIJ7mKmAfw=\ngo.opentelemetry.io/otel/trace v1.0.1/go.mod h1:5g4i4fKLaX2BQpSBsxw8YYcgKpMMSW3x7ZTuYBr3sUk=\ngo.opentelemetry.io/proto/otlp v0.7.0 h1:rwOQPCuKAKmwGKq2aVNnYIibI6wnV7EvzgfTCzcdGg8=\ngo.opentelemetry.io/proto/otlp v0.19.0 h1:IVN6GR+mhC4s5yfcTbmzHYODqvWAp3ZedA2SJPI1Nnw=\ngo.opentelemetry.io/proto/otlp v0.19.0/go.mod h1:H7XAot3MsfNsj7EXtrA2q5xSNQ10UqI405h3+duxN4U=\ngo.uber.org/goleak v1.1.11/go.mod h1:cwTWslyiVhfpKIDGSZEM2HlOvcqm+tG4zioyIeLoqMQ=\ngo.uber.org/thriftrw v1.29.1/go.mod h1:YcjXveberDd28/Bs34SwHy3yu85x/jB4UA2gIcz/Eo0=\ngolang.org/x/arch v0.0.0-20180920145803-b19384d3c130 h1:Vsc61gop4hfHdzQNolo6Fi/sw7TnJ2yl3ZR4i7bYirs=\ngolang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=\ngolang.org/x/crypto v0.10.0/go.mod h1:o4eNf7Ede1fv+hwOwZsTHl9EsPFO6q6ZvYR8vYfY45I=\ngolang.org/x/crypto v0.12.0/go.mod h1:NF0Gs7EO5K4qLn+Ylc+fih8BSTeIjAP05siRnAh98yw=\ngolang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=\ngolang.org/x/crypto v0.24.0/go.mod h1:Z1PMYSOR5nyMcyAVAIQSKCDwalqy85Aqn1x3Ws4L5DM=\ngolang.org/x/image v0.0.0-20180708004352-c73c2afc3b81 h1:00VmoueYNlNz/aHIilyyQz/MHSqGoWJzpFv/HW8xpzI=\ngolang.org/x/image v0.0.0-20190802002840-cff245a6509b h1:+qEpEAPhDZ1o0x3tHzZTQDArnOixOzGD9HUJfcg0mb4=\ngolang.org/x/mobile v0.0.0-20200801112145-973feb4309de h1:OVJ6QQUBAesB8CZijKDSsXX7xYVtUhrkY0gwMfbi4p4=\ngolang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=\ngolang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=\ngolang.org/x/net v0.0.0-20190628185345-da137c7871d7/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.8.0/go.mod h1:QVkue5JL9kW//ek3r6jTKnTFis1tRmNAW2P1shuFdJc=\ngolang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=\ngolang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI=\ngolang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=\ngolang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=\ngolang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=\ngolang.org/x/oauth2 v0.6.0/go.mod h1:ycmewcwgD4Rpr3eZJLSB4Kyyljb3qDh40vJ8STE5HKw=\ngolang.org/x/oauth2 v0.8.0/go.mod h1:yr7u4HXZRm1R1kBWqr/xKNqewf0plRYoB7sla+BCIXE=\ngolang.org/x/sync v0.2.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y=\ngolang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=\ngolang.org/x/sys v0.0.0-20220412211240-33da011f77ad/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=\ngolang.org/x/telemetry v0.0.0-20240521205824-bda55230c457 h1:zf5N6UOrA487eEFacMePxjXAJctxKmyjKUsjA11Uzuk=\ngolang.org/x/telemetry v0.0.0-20240521205824-bda55230c457/go.mod h1:pRgIJT+bRLFKnoM1ldnzKoxTIn14Yxz928LQRYYgIN0=\ngolang.org/x/term v0.15.0 h1:y/Oo/a/q3IXu26lQgl04j/gjuBDOBlx7X6Om1j2CPW4=\ngolang.org/x/term v0.15.0/go.mod h1:BDl952bC7+uMoWR75FIrCDx79TPU9oHkTZ9yRbYOrX0=\ngolang.org/x/term v0.21.0 h1:WVXCp+/EBEHOj53Rvu+7KiT/iElMrO8ACK16SMZ3jaA=\ngolang.org/x/term v0.21.0/go.mod h1:ooXLefLobQVslOqselCNF4SxFAaoS6KujMbsGzSDmX0=\ngolang.org/x/term v0.28.0 h1:/Ts8HFuMR2E6IP/jlo7QVLZHggjKQbhu/7H0LJFr3Gg=\ngolang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek=\ngolang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y=\ngolang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g=\ngolang.org/x/term v0.32.0 h1:DR4lr0TjUs3epypdhTOkMmuF5CDFJ/8pOnbzMZPQ7bg=\ngolang.org/x/term v0.32.0/go.mod h1:uZG1FhGx848Sqfsq4/DlJr3xGGsYMu/L5GW4abiaEPQ=\ngolang.org/x/text v0.8.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=\ngolang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=\ngolang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=\ngolang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=\ngolang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=\ngolang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=\ngolang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=\ngolang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE=\ngonum.org/v1/plot v0.0.0-20190515093506-e2840ee46a6b h1:Qh4dB5D/WpoUUp3lSod7qgoyEHbDGPUWjIbnqdqqe1k=\ngoogle.golang.org/api v0.139.0 h1:A1TrCPgMmOiYu0AiNkvQIpIx+D8blHTDcJ5EogkP7LI=\ngoogle.golang.org/api v0.139.0/go.mod h1:CVagp6Eekz9CjGZ718Z+sloknzkDJE7Vc1Ckj9+viBk=\ngoogle.golang.org/genproto v0.0.0-20230320184635-7606e756e683/go.mod h1:NWraEVixdDnqcqQ30jipen1STv2r/n24Wb7twVTGR4s=\ngoogle.golang.org/genproto v0.0.0-20230530153820-e85fd2cbaebc/go.mod h1:xZnkP7mREFX5MORlOPEzLMr+90PPZQ2QWzrVTWfAq64=\ngoogle.golang.org/genproto v0.0.0-20230803162519-f966b187b2e5/go.mod h1:oH/ZOT02u4kWEp7oYBGYFFkCdKS/uYR9Z7+0/xuuFp8=\ngoogle.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d/go.mod h1:yZTlhN0tQnXo3h00fuXNCxJdLdIdnVFVBaRJ5LWBbw4=\ngoogle.golang.org/genproto v0.0.0-20231002182017-d307bd883b97/go.mod h1:t1VqOqqvce95G3hIDCT5FeO3YUc6Q4Oe24L/+rNMxRk=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20230530153820-e85fd2cbaebc/go.mod h1:vHYtlOoi6TsQ3Uk2yxR7NI5z8uoV+3pZtR4jmHIkRig=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20230803162519-f966b187b2e5/go.mod h1:5DZzOUPCLYL3mNkQ0ms0F3EuUNZ7py1Bqeq6sxzI7/Q=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d h1:DoPTO70H+bcDXcd39vOqb2viZxgqeBeSGtZ55yZU4/Q=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d/go.mod h1:KjSP20unUpOx5kyQUFa7k4OJg0qeJ7DEZflGDu2p6Bk=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20230530153820-e85fd2cbaebc h1:g3hIDl0jRNd9PPTs2uBzYuaD5mQuwOkZY0vSc0LR32o=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20230530153820-e85fd2cbaebc/go.mod h1:ylj+BE99M198VPbBh6A8d9n3w8fChvyLK3wwBOjXBFA=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20230807174057-1744710a1577 h1:ZX0eQu2J+jOO87sq8fQG8J/Nfp7D7BhHpixIE5EYK/k=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20230807174057-1744710a1577/go.mod h1:NjCQG/D8JandXxM57PZbAJL1DCNL6EypA0vPPwfsc7c=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20231030173426-d783a09b4405 h1:o4S3HvTUEXgRsNSUQsALDVog0O9F/U1JJlHmmUN8Uas=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20231030173426-d783a09b4405/go.mod h1:GRUCuLdzVqZte8+Dl/D4N25yLzcGqqWaYkeVOwulFqw=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20230530153820-e85fd2cbaebc/go.mod h1:66JfowdXAEgad5O9NnYcsNPLCPZJD++2L9X0PCMODrA=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d/go.mod h1:+Bk1OCOj40wS2hwAMA+aCW9ypzm63QTBBHp6lQ3p+9M=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231002182017-d307bd883b97/go.mod h1:v7nGkzlmW8P3n/bKmWBn2WpBjpOEx8Q6gMueudAmKfY=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20231012201019-e917dd12ba7a/go.mod h1:4cYg8o5yUbm77w8ZX00LhMVNl/YVBFJRYWDc0uYWMs0=\ngoogle.golang.org/grpc v1.37.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.53.0/go.mod h1:OnIrk0ipVdj4N5d9IUoFUx72/VlD7+jUsHwZgwSMQpw=\ngoogle.golang.org/grpc v1.55.0/go.mod h1:iYEXKGkEBhg1PjZQvoYEVPTDkHo1/bjTnfwTeGONTY8=\ngoogle.golang.org/grpc v1.56.1/go.mod h1:I9bI3vqKfayGqPUAwGdOSu7kt6oIJLixfffKrpXqQ9s=\ngoogle.golang.org/grpc v1.57.0/go.mod h1:Sd+9RMTACXwmub0zcNY2c4arhtrbBYD1AUHI/dt16Mo=\ngoogle.golang.org/grpc v1.58.2/go.mod h1:tgX3ZQDlNJGU96V6yHh1T/JeoBQ2TXdr43YbYSsCJk0=\ngoogle.golang.org/grpc v1.58.3/go.mod h1:tgX3ZQDlNJGU96V6yHh1T/JeoBQ2TXdr43YbYSsCJk0=\ngoogle.golang.org/grpc/cmd/protoc-gen-go-grpc v1.1.0 h1:M1YKkFIboKNieVO5DLUEVzQfGwJD30Nv2jfUgzb5UcE=\ngoogle.golang.org/grpc/cmd/protoc-gen-go-grpc v1.1.0/go.mod h1:6Kw0yEErY5E/yWrBtf03jp27GLLJujG4z/JK95pnjjw=\ngoogle.golang.org/protobuf v1.29.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=\ngoogle.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=\ngoogle.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6 h1:jMFz6MfLP0/4fUyZle81rXUoxOBFi19VUFKVDOQfozc=\ngopkg.in/cheggaaa/pb.v1 v1.0.28 h1:n1tBJnnK2r7g9OW2btFH91V92STTUevLXYFb8gy9EMk=\ngopkg.in/errgo.v2 v2.1.0 h1:0vLT13EuvQ0hNvakwLuFZ/jYrLp5F3kcWHXdRggjCE8=\ngopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=\ngopkg.in/gcfg.v1 v1.2.3 h1:m8OOJ4ccYHnx2f4gQwpno8nAX5OGOh7RLaaz0pj3Ogs=\ngopkg.in/resty.v1 v1.12.0 h1:CuXP0Pjfw9rOuY6EP+UvtNvt5DSqHpIxILZKT/quCZI=\ngopkg.in/src-d/go-billy.v4 v4.3.0 h1:KtlZ4c1OWbIs4jCv5ZXrTqG8EQocr0g/d4DjNg70aek=\ngopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=\nlukechampine.com/adiantum v1.1.1 h1:4fp6gTxWCqpEbLy40ExiYDDED3oUNWx5cTqBCtPdZqA=\nlukechampine.com/adiantum v1.1.1/go.mod h1:LrAYVnTYLnUtE/yMp5bQr0HstAf060YUF8nM0B6+rUw=\nnhooyr.io/websocket v1.8.7 h1:usjR2uOr/zjjkVMy0lW+PPohFok7PCow5sDjLgX4P4g=\nrsc.io/binaryregexp v0.2.0 h1:HfqmD5MEmC0zvwBuF187nq9mdnXjXsSivRiXN7SmRkE=\nrsc.io/pdf v0.1.1 h1:k1MczvYDUvJBe93bYd7wrZLLUEcLZAuF824/I4e5Xr4=\nrsc.io/quote/v3 v3.1.0 h1:9JKUTTIUgS6kzR9mK1YuGKv6Nl+DijDNIc0ghT58FaY=\nrsc.io/sampler v1.3.0 h1:7uVkIFmeBqHfdjD+gZwtXXI+RODJ2Wc4O7MPEh/QiW4=\nsigs.k8s.io/yaml v1.1.0 h1:4A07+ZFc2wgJwo8YNlQpr1rVlgUDlxXHhPJciaPY5gs=\nsigs.k8s.io/yaml v1.2.0 h1:kr/MCeFWJWTwyaHoR9c8EjH9OumOmoF9YGiZd7lFm/Q=\n"
  },
  {
    "path": "host/activity_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n)\n\nfunc (s *IntegrationSuite) TestActivityHeartBeatWorkflow_Success() {\n\tid := \"integration-heartbeat-test\"\n\twt := \"integration-heartbeat-test-type\"\n\ttl := \"integration-heartbeat-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_timer\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\theader := &types.Header{\n\t\tFields: map[string][]byte{\"tracing\": []byte(\"sample data\")},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tHeader:                              header,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tHeader:                        header,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(15),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(15),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\ts.Logger.Info(\"Completing Workflow.\")\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tactivityExecutedCount := 0\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\tfor i := 0; i < 10; i++ {\n\t\t\ts.Logger.Info(\"Heartbeating for activity\", tag.WorkflowActivityID(activityID), tag.Counter(i))\n\t\t\tctx, cancel := createContext()\n\t\t\t_, err := s.Engine.RecordActivityTaskHeartbeat(ctx, &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\tTaskToken: taskToken, Details: []byte(\"details\")})\n\t\t\tcancel()\n\t\t\ts.Nil(err)\n\t\t\ttime.Sleep(10 * time.Millisecond)\n\t\t}\n\t\tactivityExecutedCount++\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\n\ts.False(workflowComplete)\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n\ts.True(activityExecutedCount == 1)\n\n\t// go over history and verify that the activity task scheduled event has header on it\n\tevents := s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.GetRunID(),\n\t})\n\tfor _, event := range events {\n\t\tif *event.EventType == types.EventTypeActivityTaskScheduled {\n\t\t\ts.Equal(header, event.ActivityTaskScheduledEventAttributes.Header)\n\t\t}\n\t}\n}\n\nfunc (s *IntegrationSuite) TestActivityHeartbeatDetailsDuringRetry() {\n\tid := \"integration-heartbeat-details-retry-test\"\n\twt := \"integration-heartbeat-details-retry-type\"\n\ttl := \"integration-heartbeat-details-retry-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_heartbeat_retry\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivitiesScheduled := false\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !activitiesScheduled {\n\t\t\tactivitiesScheduled = true\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\t\tInput:                         nil,\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(4),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(4),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(4),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds:    1,\n\t\t\t\t\t\t\tMaximumAttempts:             3,\n\t\t\t\t\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\t\t\t\t\tBackoffCoefficient:          1,\n\t\t\t\t\t\t\tExpirationIntervalInSeconds: 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\ts.Logger.Info(\"Completing Workflow.\")\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tactivityExecutedCount := 0\n\theartbeatDetails := []byte(\"details\")\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\n\t\tvar err error\n\t\tif activityExecutedCount == 0 {\n\t\t\ts.Logger.Info(\"Heartbeating for activity:\", tag.WorkflowActivityID(activityID))\n\t\t\tctx, cancel := createContext()\n\t\t\t_, err = s.Engine.RecordActivityTaskHeartbeat(ctx, &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\tTaskToken: taskToken, Details: heartbeatDetails})\n\t\t\tcancel()\n\t\t\ts.Nil(err)\n\t\t\t// Trigger heartbeat timeout and retry\n\t\t\ttime.Sleep(time.Second * 2)\n\t\t} else if activityExecutedCount == 1 {\n\t\t\t// return an error and retry\n\t\t\terr = errors.New(\"retryable-error\")\n\t\t}\n\n\t\tactivityExecutedCount++\n\t\treturn nil, false, err\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tdescribeWorkflowExecution := func() (*types.DescribeWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\tfor i := 0; i != 3; i++ {\n\t\terr = poller.PollAndProcessActivityTask(false)\n\t\tif i == 0 {\n\t\t\t// first time, hearbeat timeout, respond activity complete will fail\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\t// second time, retryable error\n\t\t\ts.Nil(err)\n\t\t}\n\n\t\tdweResponse, err := describeWorkflowExecution()\n\t\ts.Nil(err)\n\n\t\tpendingActivities := dweResponse.GetPendingActivities()\n\t\tif i == 2 {\n\t\t\t// third time, complete activity, no pending info\n\t\t\ts.Equal(0, len(pendingActivities))\n\t\t} else {\n\t\t\ts.Equal(1, len(pendingActivities))\n\t\t\tpendingActivity := pendingActivities[0]\n\n\t\t\ts.Equal(int32(3), pendingActivity.GetMaximumAttempts())\n\t\t\ts.Equal(int32(i+1), pendingActivity.GetAttempt())\n\t\t\ts.Equal(types.PendingActivityStateScheduled, pendingActivity.GetState())\n\t\t\tif i == 0 {\n\t\t\t\ts.Equal(\"cadenceInternal:Timeout HEARTBEAT\", pendingActivity.GetLastFailureReason())\n\t\t\t\ts.Nil(pendingActivity.GetLastFailureDetails())\n\t\t\t} else { // i == 1\n\t\t\t\texpectedErrString := \"retryable-error\"\n\t\t\t\ts.Equal(expectedErrString, pendingActivity.GetLastFailureReason())\n\t\t\t\ts.Equal([]byte(expectedErrString), pendingActivity.GetLastFailureDetails())\n\t\t\t}\n\t\t\ts.Equal(identity, pendingActivity.GetLastWorkerIdentity())\n\n\t\t\tscheduledTS := pendingActivity.ScheduledTimestamp\n\t\t\tlastHeartbeatTS := pendingActivity.LastHeartbeatTimestamp\n\t\t\texpirationTS := pendingActivity.ExpirationTimestamp\n\t\t\ts.NotNil(scheduledTS)\n\t\t\ts.NotNil(lastHeartbeatTS)\n\t\t\ts.NotNil(expirationTS)\n\t\t\ts.Nil(pendingActivity.LastStartedTimestamp)\n\t\t\ts.True(*scheduledTS > *lastHeartbeatTS)\n\t\t\ts.True(*expirationTS > *scheduledTS)\n\n\t\t\ts.Equal(heartbeatDetails, pendingActivity.GetHeartbeatDetails())\n\t\t}\n\t}\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\ts.True(workflowComplete)\n\ts.Equal(3, activityExecutedCount)\n}\n\nfunc (s *IntegrationSuite) TestActivityRetry() {\n\tid := \"integration-activity-retry-test\"\n\twt := \"integration-activity-retry-type\"\n\ttl := \"integration-activity-retry-tasklist\"\n\tidentity := \"worker1\"\n\tidentity2 := \"worker2\"\n\tactivityName := \"activity_retry\"\n\ttimeoutActivityName := \"timeout_activity\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivitiesScheduled := false\n\tvar activityAScheduled, activityAFailed, activityBScheduled, activityBTimeout *types.HistoryEvent\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !activitiesScheduled {\n\t\t\tactivitiesScheduled = true\n\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    \"A\",\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\t\tInput:                         []byte(\"1\"),\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(4),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(4),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(4),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds:    1,\n\t\t\t\t\t\t\tMaximumAttempts:             3,\n\t\t\t\t\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\t\t\t\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\t\t\t\t\t\tBackoffCoefficient:          1,\n\t\t\t\t\t\t\tExpirationIntervalInSeconds: 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    \"B\",\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: timeoutActivityName},\n\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: \"no_worker_tasklist\"},\n\t\t\t\t\t\tInput:                         []byte(\"2\"),\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(5),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tswitch event.GetEventType() {\n\t\t\t\tcase types.EventTypeActivityTaskScheduled:\n\t\t\t\t\tswitch event.ActivityTaskScheduledEventAttributes.GetActivityID() {\n\t\t\t\t\tcase \"A\":\n\t\t\t\t\t\tactivityAScheduled = event\n\t\t\t\t\tcase \"B\":\n\t\t\t\t\t\tactivityBScheduled = event\n\t\t\t\t\t}\n\n\t\t\t\tcase types.EventTypeActivityTaskFailed:\n\t\t\t\t\tif event.ActivityTaskFailedEventAttributes.GetScheduledEventID() == activityAScheduled.ID {\n\t\t\t\t\t\tactivityAFailed = event\n\t\t\t\t\t}\n\n\t\t\t\tcase types.EventTypeActivityTaskTimedOut:\n\t\t\t\t\tif event.ActivityTaskTimedOutEventAttributes.GetScheduledEventID() == activityBScheduled.ID {\n\t\t\t\t\t\tactivityBTimeout = event\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tif activityAFailed != nil && activityBTimeout != nil {\n\t\t\ts.Logger.Info(\"Completing Workflow.\")\n\t\t\tworkflowComplete = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{}, nil\n\t}\n\n\tactivityExecutedCount := 0\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\tvar err error\n\t\tif activityExecutedCount == 0 {\n\t\t\terr = errors.New(\"bad-luck-please-retry\")\n\t\t} else if activityExecutedCount == 1 {\n\t\t\terr = errors.New(\"bad-bug\")\n\t\t}\n\t\tactivityExecutedCount++\n\t\treturn nil, false, err\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tpoller2 := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tdescribeWorkflowExecution := func() (*types.DescribeWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks, err)\n\n\tdescResp, err := describeWorkflowExecution()\n\ts.Nil(err)\n\tfor _, pendingActivity := range descResp.GetPendingActivities() {\n\t\tif pendingActivity.GetActivityID() == \"A\" {\n\t\t\texpectedErrString := \"bad-luck-please-retry\"\n\t\t\ts.Equal(expectedErrString, pendingActivity.GetLastFailureReason())\n\t\t\ts.Equal([]byte(expectedErrString), pendingActivity.GetLastFailureDetails())\n\t\t\ts.Equal(identity, pendingActivity.GetLastWorkerIdentity())\n\t\t}\n\t}\n\n\terr = poller2.PollAndProcessActivityTask(false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks, err)\n\n\tdescResp, err = describeWorkflowExecution()\n\ts.Nil(err)\n\tfor _, pendingActivity := range descResp.GetPendingActivities() {\n\t\tif pendingActivity.GetActivityID() == \"A\" {\n\t\t\texpectedErrString := \"bad-bug\"\n\t\t\ts.Equal(expectedErrString, pendingActivity.GetLastFailureReason())\n\t\t\ts.Equal([]byte(expectedErrString), pendingActivity.GetLastFailureDetails())\n\t\t\ts.Equal(identity2, pendingActivity.GetLastWorkerIdentity())\n\t\t}\n\t}\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\tfor i := 0; i < 3; i++ {\n\t\ts.False(workflowComplete)\n\n\t\ts.Logger.Info(\"Processing decision task:\", tag.Counter(i))\n\t\t_, err := poller.PollAndProcessDecisionTaskWithoutRetry(false, false)\n\t\tif err != nil {\n\t\t\ts.printWorkflowHistory(s.DomainName, &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.GetRunID(),\n\t\t\t})\n\t\t}\n\t\ts.Nil(err, \"Poll for decision task failed.\")\n\n\t\tif workflowComplete {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ts.True(workflowComplete)\n\ts.True(activityExecutedCount == 2)\n}\n\nfunc (s *IntegrationSuite) TestActivityHeartBeatWorkflow_Timeout() {\n\tid := \"integration-heartbeat-timeout-test\"\n\twt := \"integration-heartbeat-timeout-test-type\"\n\ttl := \"integration-heartbeat-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_timer\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\ts.Logger.Info(\"Calling DecisionTask Handler: %d, %d.\", tag.Counter(int(activityCounter)), tag.Number(int64(activityCount)))\n\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(15),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(15),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tactivityExecutedCount := 0\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\t// Timing out more than HB time.\n\t\ttime.Sleep(2 * time.Second)\n\t\tactivityExecutedCount++\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Error(err)\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\n\ts.False(workflowComplete)\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestActivityTimeouts() {\n\tid := \"integration-activity-timeout-test\"\n\twt := \"integration-activity-timeout-test-type\"\n\ttl := \"integration-activity-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"timeout_activity\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(300),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tworkflowFailed := false\n\tactivitiesScheduled := false\n\tactivitiesMap := map[int64]*types.HistoryEvent{}\n\tfailWorkflow := false\n\tfailReason := \"\"\n\tvar activityATimedout, activityBTimedout, activityCTimedout, activityDTimedout bool\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !activitiesScheduled {\n\t\t\tactivitiesScheduled = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"A\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: \"NoWorker\"},\n\t\t\t\t\tInput:                         []byte(\"ScheduleToStart\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(35),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(3), // ActivityID A is expected to timeout using ScheduleToStart\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(30),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t},\n\t\t\t}, {\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"B\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         []byte(\"ScheduleToClose\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(7), // ActivityID B is expected to timeout using ScheduleClose\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(10),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t},\n\t\t\t}, {\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"C\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         []byte(\"StartToClose\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(15),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(5), // ActivityID C is expected to timeout using StartToClose\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\tInitialIntervalInSeconds:    1,\n\t\t\t\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\t\t\t\tBackoffCoefficient:          1,\n\t\t\t\t\t\tExpirationIntervalInSeconds: 3, // activity expiration time will not be extended, so it won't retry\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, {\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"D\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         []byte(\"Heartbeat\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(35),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(15),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(3), // ActivityID D is expected to timeout using Heartbeat\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskScheduled {\n\t\t\t\t\tactivitiesMap[event.ID] = event\n\t\t\t\t}\n\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskTimedOut {\n\t\t\t\t\ttimeoutEvent := event.ActivityTaskTimedOutEventAttributes\n\t\t\t\t\tscheduledEvent, ok := activitiesMap[timeoutEvent.GetScheduledEventID()]\n\t\t\t\t\tif !ok {\n\t\t\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\t\tReason: common.StringPtr(\"ScheduledEvent not found.\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}}, nil\n\t\t\t\t\t}\n\n\t\t\t\t\tswitch timeoutEvent.GetTimeoutType() {\n\t\t\t\t\tcase types.TimeoutTypeScheduleToStart:\n\t\t\t\t\t\tif scheduledEvent.ActivityTaskScheduledEventAttributes.GetActivityID() == \"A\" {\n\t\t\t\t\t\t\tactivityATimedout = true\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\t\tfailReason = \"ActivityID A is expected to timeout with ScheduleToStart\"\n\t\t\t\t\t\t}\n\t\t\t\t\tcase types.TimeoutTypeScheduleToClose:\n\t\t\t\t\t\tif scheduledEvent.ActivityTaskScheduledEventAttributes.GetActivityID() == \"B\" {\n\t\t\t\t\t\t\tactivityBTimedout = true\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\t\tfailReason = \"ActivityID B is expected to timeout with ScheduleToClose\"\n\t\t\t\t\t\t}\n\t\t\t\t\tcase types.TimeoutTypeStartToClose:\n\t\t\t\t\t\tif scheduledEvent.ActivityTaskScheduledEventAttributes.GetActivityID() == \"C\" {\n\t\t\t\t\t\t\tactivityCTimedout = true\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\t\tfailReason = \"ActivityID C is expected to timeout with StartToClose\"\n\t\t\t\t\t\t}\n\t\t\t\t\tcase types.TimeoutTypeHeartbeat:\n\t\t\t\t\t\tif scheduledEvent.ActivityTaskScheduledEventAttributes.GetActivityID() == \"D\" {\n\t\t\t\t\t\t\tactivityDTimedout = true\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\t\tfailReason = \"ActivityID D is expected to timeout with Heartbeat\"\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tif failWorkflow {\n\t\t\ts.Logger.Error(\"Failing types.\")\n\t\t\tworkflowFailed = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason: common.StringPtr(failReason),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif activityATimedout && activityBTimedout && activityCTimedout && activityDTimedout {\n\t\t\ts.Logger.Info(\"Completing Workflow.\")\n\t\t\tworkflowComplete = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\ttimeoutType := string(input)\n\t\tswitch timeoutType {\n\t\tcase \"ScheduleToStart\":\n\t\t\ts.Fail(\"Activity A not expected to be started.\")\n\t\tcase \"ScheduleToClose\":\n\t\t\ts.Logger.Info(\"Sleeping activityB for 6 seconds.\")\n\t\t\ttime.Sleep(7 * time.Second)\n\t\tcase \"StartToClose\":\n\t\t\ts.Logger.Info(\"Sleeping activityC for 8 seconds.\")\n\t\t\ttime.Sleep(8 * time.Second)\n\t\tcase \"Heartbeat\":\n\t\t\ts.Logger.Info(\"Starting hearbeat activity.\")\n\t\t\tgo func() {\n\t\t\t\tfor i := 0; i < 6; i++ {\n\t\t\t\t\ts.Logger.Info(\"Heartbeating for activity\", tag.WorkflowActivityID(activityID), tag.Counter(i))\n\t\t\t\t\tctx, cancel := createContext()\n\t\t\t\t\t_, err := s.Engine.RecordActivityTaskHeartbeat(ctx, &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\t\tTaskToken: taskToken, Details: []byte(strconv.Itoa(i))})\n\t\t\t\t\tcancel()\n\t\t\t\t\ts.Nil(err)\n\t\t\t\t\ttime.Sleep(1 * time.Second)\n\t\t\t\t}\n\t\t\t\ts.Logger.Info(\"End Heartbeating.\")\n\t\t\t}()\n\t\t\ts.Logger.Info(\"Sleeping hearbeat activity.\")\n\t\t\ttime.Sleep(10 * time.Second)\n\t\t}\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\tfor i := 0; i < 3; i++ {\n\t\tgo func() {\n\t\t\terr = poller.PollAndProcessActivityTask(false)\n\t\t\ts.Logger.Info(\"Activity Processing Completed.  Error\", tag.Error(err))\n\t\t}()\n\t}\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\tfor i := 0; i < 10; i++ {\n\t\ts.Logger.Info(\"Processing decision task: %v\", tag.Counter(i))\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Nil(err, \"Poll for decision task failed.\")\n\n\t\tif workflowComplete || workflowFailed {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ts.True(workflowComplete)\n\ts.False(workflowFailed)\n}\n\nfunc (s *IntegrationSuite) TestActivityHeartbeatTimeouts() {\n\tid := \"integration-activity-heartbeat-timeout-test\"\n\twt := \"integration-activity-heartbeat-timeout-test-type\"\n\ttl := \"integration-activity-heartbeat-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"timeout_activity\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(70),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivitiesScheduled := false\n\tlastHeartbeatMap := make(map[int64]int)\n\tfailWorkflow := false\n\tfailReason := \"\"\n\tactivityCount := 10\n\tactivitiesTimedout := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !activitiesScheduled {\n\t\t\tactivitiesScheduled = true\n\t\t\tdecisions := []*types.Decision{}\n\t\t\tfor i := 0; i < activityCount; i++ {\n\t\t\t\taID := fmt.Sprintf(\"activity_%v\", i)\n\t\t\t\td := &types.Decision{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    aID,\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\t\tInput:                         []byte(\"Heartbeat\"),\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(60),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tdecisions = append(decisions, d)\n\t\t\t}\n\n\t\t\treturn nil, decisions, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\tProcessLoop:\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskScheduled {\n\t\t\t\t\tlastHeartbeatMap[event.ID] = 0\n\t\t\t\t}\n\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskCompleted ||\n\t\t\t\t\tevent.GetEventType() == types.EventTypeActivityTaskFailed {\n\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\tfailReason = \"Expected activities to timeout but seeing completion instead\"\n\t\t\t\t}\n\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskTimedOut {\n\t\t\t\t\ttimeoutEvent := event.ActivityTaskTimedOutEventAttributes\n\t\t\t\t\t_, ok := lastHeartbeatMap[timeoutEvent.GetScheduledEventID()]\n\t\t\t\t\tif !ok {\n\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\tfailReason = \"ScheduledEvent not found.\"\n\t\t\t\t\t\tbreak ProcessLoop\n\t\t\t\t\t}\n\n\t\t\t\t\tswitch timeoutEvent.GetTimeoutType() {\n\t\t\t\t\tcase types.TimeoutTypeHeartbeat:\n\t\t\t\t\t\tactivitiesTimedout++\n\t\t\t\t\t\tscheduleID := timeoutEvent.GetScheduledEventID()\n\t\t\t\t\t\tlastHeartbeat, _ := strconv.Atoi(string(timeoutEvent.Details))\n\t\t\t\t\t\tlastHeartbeatMap[scheduleID] = lastHeartbeat\n\t\t\t\t\tdefault:\n\t\t\t\t\t\tfailWorkflow = true\n\t\t\t\t\t\tfailReason = \"Expected Heartbeat timeout but recieved another timeout\"\n\t\t\t\t\t\tbreak ProcessLoop\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tif failWorkflow {\n\t\t\ts.Logger.Error(\"Failing types. Reason: %v\", tag.Value(failReason))\n\t\t\tworkflowComplete = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason: common.StringPtr(failReason),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif activitiesTimedout == activityCount {\n\t\t\ts.Logger.Info(\"Completing Workflow.\")\n\t\t\tworkflowComplete = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Logger.Info(\"Starting heartbeat activity. ID\", tag.WorkflowActivityID(activityID))\n\t\tfor i := 0; i < 10; i++ {\n\t\t\tif !workflowComplete {\n\t\t\t\ts.Logger.Info(\"Heartbeating for activity\", tag.WorkflowActivityID(activityID), tag.Counter(i))\n\t\t\t\tctx, cancel := createContext()\n\t\t\t\t_, err := s.Engine.RecordActivityTaskHeartbeat(ctx, &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: taskToken, Details: []byte(strconv.Itoa(i))})\n\t\t\t\tcancel()\n\t\t\t\tif err != nil {\n\t\t\t\t\ts.Logger.Error(\"Activity heartbeat failed\", tag.WorkflowActivityID(activityID), tag.Counter(i), tag.Error(err))\n\t\t\t\t}\n\n\t\t\t\tsecondsToSleep := rand.Intn(3)\n\t\t\t\ts.Logger.Info(\"Activity ID '%v' sleeping for: %v seconds\", tag.WorkflowActivityID(activityID), tag.Number(int64(secondsToSleep)))\n\t\t\t\ttime.Sleep(time.Duration(secondsToSleep) * time.Second)\n\t\t\t}\n\t\t}\n\t\ts.Logger.Info(\"End Heartbeating.\", tag.WorkflowActivityID(activityID))\n\n\t\ts.Logger.Info(\"Sleeping activity before completion\", tag.WorkflowActivityID(activityID))\n\t\ttime.Sleep(7 * time.Second)\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\tfor i := 0; i < activityCount; i++ {\n\t\tgo func() {\n\t\t\terr := poller.PollAndProcessActivityTask(false)\n\t\t\ts.Logger.Info(\"Activity Processing Completed.\", tag.Error(err))\n\t\t}()\n\t}\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\tfor i := 0; i < 10; i++ {\n\t\ts.Logger.Info(\"Processing decision task\", tag.Counter(i))\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Nil(err, \"Poll for decision task failed.\")\n\n\t\tif workflowComplete {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ts.True(workflowComplete)\n\ts.False(failWorkflow, failReason)\n\ts.Equal(activityCount, activitiesTimedout)\n\ts.Equal(activityCount, len(lastHeartbeatMap))\n\tfor aID, lastHeartbeat := range lastHeartbeatMap {\n\t\ts.Logger.Info(\"Last heartbeat for activity with scheduleID\", tag.Counter(int(aID)), tag.Number(int64(lastHeartbeat)))\n\t\ts.Equal(9, lastHeartbeat)\n\t}\n}\n\nfunc (s *IntegrationSuite) TestActivityCancellation() {\n\tid := \"integration-activity-cancellation-test\"\n\twt := \"integration-activity-cancellation-test-type\"\n\ttl := \"integration-activity-cancellation-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_timer\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution: response\", tag.WorkflowRunID(we.GetRunID()))\n\n\tactivityCounter := int32(0)\n\tscheduleActivity := true\n\trequestCancellation := false\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif scheduleActivity {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(15),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(15),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif requestCancellation {\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\t\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID: strconv.Itoa(int(activityCounter)),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\ts.Logger.Info(\"Completing Workflow.\")\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tactivityExecutedCount := 0\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.GetName())\n\t\tfor i := 0; i < 10; i++ {\n\t\t\ts.Logger.Info(\"Heartbeating for activity\", tag.WorkflowActivityID(activityID), tag.Counter(i))\n\t\t\tctx, cancel := createContext()\n\t\t\tresponse, err := s.Engine.RecordActivityTaskHeartbeat(ctx,\n\t\t\t\t&types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: taskToken, Details: []byte(\"details\")})\n\t\t\tcancel()\n\t\t\tif response.CancelRequested {\n\t\t\t\treturn []byte(\"Activity Cancelled.\"), true, nil\n\t\t\t}\n\t\t\ts.Nil(err)\n\t\t\ttime.Sleep(10 * time.Millisecond)\n\t\t}\n\t\tactivityExecutedCount++\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks, err)\n\n\tcancelCh := make(chan struct{})\n\n\tgo func() {\n\t\ts.Logger.Info(\"Trying to cancel the task in a different thread.\")\n\t\tscheduleActivity = false\n\t\trequestCancellation = true\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.True(err == nil || err == tasklist.ErrNoTasks, err)\n\t\tcancelCh <- struct{}{}\n\t}()\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks, err)\n\n\t<-cancelCh\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n}\n\nfunc (s *IntegrationSuite) TestActivityCancellationNotStarted() {\n\tid := \"integration-activity-notstarted-cancellation-test\"\n\twt := \"integration-activity-notstarted-cancellation-test-type\"\n\ttl := \"integration-activity-notstarted-cancellation-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_notstarted\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecutionn\", tag.WorkflowRunID(we.GetRunID()))\n\n\tactivityCounter := int32(0)\n\tscheduleActivity := true\n\trequestCancellation := false\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif scheduleActivity {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\t\t\ts.Logger.Info(\"Scheduling activity.\")\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(15),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(15),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(0),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif requestCancellation {\n\t\t\ts.Logger.Info(\"Requesting cancellation.\")\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\t\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID: strconv.Itoa(int(activityCounter)),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\ts.Logger.Info(\"Completing Workflow.\")\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// dummy activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Fail(\"activity should not run\")\n\t\treturn nil, false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\t// Send signal so that worker can send an activity cancel\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\t// Process signal in decider and send request cancellation\n\tscheduleActivity = false\n\trequestCancellation = true\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\n\tscheduleActivity = false\n\trequestCancellation = false\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n}\n"
  },
  {
    "path": "host/archival_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/binary\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tretryLimit       = 20\n\tretryBackoffTime = 200 * time.Millisecond\n)\n\nfunc (s *IntegrationSuite) TestArchival_TimerQueueProcessor() {\n\ts.True(s.TestCluster.archiverBase.metadata.GetHistoryConfig().ClusterConfiguredForArchival())\n\n\tdomainID := s.getDomainID(s.ArchivalDomainName)\n\tWorkflowID := \"archival-timer-queue-processor-workflow-id\"\n\tworkflowType := \"archival-timer-queue-processor-type\"\n\ttaskList := \"archival-timer-queue-processor-task-list\"\n\tnumActivities := 1\n\tnumRuns := 1\n\tRunID := s.startAndFinishWorkflow(WorkflowID, workflowType, taskList, s.ArchivalDomainName, domainID, numActivities, numRuns)[0]\n\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t}\n\ts.True(s.isHistoryArchived(s.ArchivalDomainName, execution))\n\ts.True(s.isHistoryDeleted(domainID, execution))\n\ts.True(s.isMutableStateDeleted(domainID, execution))\n}\n\nfunc (s *IntegrationSuite) TestArchival_ContinueAsNew() {\n\ts.True(s.TestCluster.archiverBase.metadata.GetHistoryConfig().ClusterConfiguredForArchival())\n\n\tdomainID := s.getDomainID(s.ArchivalDomainName)\n\tWorkflowID := \"archival-continueAsNew-workflow-id\"\n\tworkflowType := \"archival-continueAsNew-workflow-type\"\n\ttaskList := \"archival-continueAsNew-task-list\"\n\tnumActivities := 1\n\tnumRuns := 5\n\tRunIDs := s.startAndFinishWorkflow(WorkflowID, workflowType, taskList, s.ArchivalDomainName, domainID, numActivities, numRuns)\n\n\tfor _, RunID := range RunIDs {\n\t\texecution := &types.WorkflowExecution{\n\t\t\tWorkflowID: WorkflowID,\n\t\t\tRunID:      RunID,\n\t\t}\n\t\ts.True(s.isHistoryArchived(s.ArchivalDomainName, execution))\n\t\ts.True(s.isHistoryDeleted(domainID, execution))\n\t\ts.True(s.isMutableStateDeleted(domainID, execution))\n\t}\n}\n\nfunc (s *IntegrationSuite) TestArchival_ArchiverWorker() {\n\ts.True(s.TestCluster.archiverBase.metadata.GetHistoryConfig().ClusterConfiguredForArchival())\n\n\tdomainID := s.getDomainID(s.ArchivalDomainName)\n\tWorkflowID := \"archival-archiver-worker-workflow-id\"\n\tworkflowType := \"archival-archiver-worker-workflow-type\"\n\ttaskList := \"archival-archiver-worker-task-list\"\n\tnumActivities := 10\n\tRunID := s.startAndFinishWorkflow(WorkflowID, workflowType, taskList, s.ArchivalDomainName, domainID, numActivities, 1)[0]\n\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: WorkflowID,\n\t\tRunID:      RunID,\n\t}\n\ts.True(s.isHistoryArchived(s.ArchivalDomainName, execution))\n\ts.True(s.isHistoryDeleted(domainID, execution))\n\ts.True(s.isMutableStateDeleted(domainID, execution))\n}\n\nfunc (s *IntegrationSuite) TestVisibilityArchival() {\n\ts.True(s.TestCluster.archiverBase.metadata.GetVisibilityConfig().ClusterConfiguredForArchival())\n\n\tdomainID := s.getDomainID(s.ArchivalDomainName)\n\tWorkflowID := \"archival-visibility-workflow-id\"\n\tworkflowType := \"archival-visibility-workflow-type\"\n\ttaskList := \"archival-visibility-task-list\"\n\tnumActivities := 3\n\tnumRuns := 5\n\tstartTime := time.Now().UnixNano()\n\ts.startAndFinishWorkflow(WorkflowID, workflowType, taskList, s.ArchivalDomainName, domainID, numActivities, numRuns)\n\ts.startAndFinishWorkflow(\"some other WorkflowID\", \"some other workflow type\", taskList, s.ArchivalDomainName, domainID, numActivities, numRuns)\n\tendTime := time.Now().UnixNano()\n\n\tvar executions []*types.WorkflowExecutionInfo\n\n\tfor i := 0; i != retryLimit; i++ {\n\t\texecutions = []*types.WorkflowExecutionInfo{}\n\t\trequest := &types.ListArchivedWorkflowExecutionsRequest{\n\t\t\tDomain:   s.ArchivalDomainName,\n\t\t\tPageSize: 2,\n\t\t\tQuery:    fmt.Sprintf(\"CloseTime >= %v and CloseTime <= %v and WorkflowType = '%s'\", startTime, endTime, workflowType),\n\t\t}\n\t\tfor len(executions) == 0 || request.NextPageToken != nil {\n\t\t\tctx, cancel := createContext()\n\t\t\tresponse, err := s.Engine.ListArchivedWorkflowExecutions(ctx, request)\n\t\t\tcancel()\n\t\t\ts.NoError(err)\n\t\t\ts.NotNil(response)\n\t\t\texecutions = append(executions, response.GetExecutions()...)\n\t\t\trequest.NextPageToken = response.NextPageToken\n\t\t}\n\t\tif len(executions) == numRuns {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(retryBackoffTime)\n\t}\n\n\tfor _, execution := range executions {\n\t\ts.Equal(WorkflowID, execution.GetExecution().GetWorkflowID())\n\t\ts.Equal(workflowType, execution.GetType().GetName())\n\t\ts.NotZero(execution.StartTime)\n\t\ts.NotZero(execution.ExecutionTime)\n\t\ts.NotZero(execution.CloseTime)\n\t}\n}\n\nfunc (s *IntegrationSuite) getDomainID(domain string) string {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tdomainResp, err := s.Engine.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(s.ArchivalDomainName),\n\t})\n\ts.Nil(err)\n\treturn domainResp.DomainInfo.GetUUID()\n}\n\nfunc (s *IntegrationSuite) isHistoryArchived(domain string, execution *types.WorkflowExecution) bool {\n\trequest := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:    s.ArchivalDomainName,\n\t\tExecution: execution,\n\t}\n\n\tfor i := 0; i < retryLimit; i++ {\n\t\tctx, cancel := createContext()\n\t\tgetHistoryResp, err := s.Engine.GetWorkflowExecutionHistory(ctx, request)\n\t\tcancel()\n\t\tif err == nil && getHistoryResp != nil && getHistoryResp.GetArchived() {\n\t\t\treturn true\n\t\t}\n\t\ttime.Sleep(retryBackoffTime)\n\t}\n\treturn false\n}\n\nfunc (s *IntegrationSuite) isHistoryDeleted(domainID string, execution *types.WorkflowExecution) bool {\n\tshardID := common.WorkflowIDToHistoryShard(execution.WorkflowID, s.TestClusterConfig.HistoryConfig.NumHistoryShards)\n\n\trequest := &persistence.GetHistoryTreeRequest{\n\t\tTreeID:     execution.GetRunID(),\n\t\tShardID:    common.IntPtr(shardID),\n\t\tDomainName: s.DomainName,\n\t}\n\tfor i := 0; i < retryLimit; i++ {\n\t\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\t\tresp, err := s.TestCluster.testBase.HistoryV2Mgr.GetHistoryTree(ctx, request)\n\t\ts.Nil(err)\n\t\tcancel()\n\t\tif len(resp.Branches) == 0 {\n\t\t\treturn true\n\t\t}\n\t\ttime.Sleep(retryBackoffTime)\n\t}\n\treturn false\n}\n\nfunc (s *IntegrationSuite) isMutableStateDeleted(domainID string, execution *types.WorkflowExecution) bool {\n\trequest := &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID:  domainID,\n\t\tExecution: *execution,\n\t}\n\n\tfor i := 0; i < retryLimit; i++ {\n\t\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\t\t_, err := s.TestCluster.testBase.ExecutionManager.GetWorkflowExecution(ctx, request)\n\t\tcancel()\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\treturn true\n\t\t}\n\t\ttime.Sleep(retryBackoffTime)\n\n\t}\n\treturn false\n}\n\nfunc (s *IntegrationSuite) startAndFinishWorkflow(id, wt, tl, domain, domainID string, numActivities, numRuns int) []string {\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tworkflowType := &types.WorkflowType{\n\t\tName: wt,\n\t}\n\ttaskList := &types.TaskList{\n\t\tName: tl,\n\t}\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              domain,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tRunIDs := make([]string, numRuns)\n\n\tworkflowComplete := false\n\tactivityCount := int32(numActivities)\n\tactivityCounter := int32(0)\n\texpectedActivityID := int32(1)\n\trunCounter := 1\n\n\tdtHandler := func(\n\t\texecution *types.WorkflowExecution,\n\t\twt *types.WorkflowType,\n\t\tpreviousStartedEventID int64,\n\t\tstartedEventID int64,\n\t\thistory *types.History,\n\t) ([]byte, []*types.Decision, error) {\n\t\tRunIDs[runCounter-1] = execution.GetRunID()\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif runCounter < numRuns {\n\t\t\tactivityCounter = int32(0)\n\t\t\texpectedActivityID = int32(1)\n\t\t\trunCounter++\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(\n\t\texecution *types.WorkflowExecution,\n\t\tactivityType *types.ActivityType,\n\t\tactivityID string,\n\t\tinput []byte,\n\t\ttaskToken []byte,\n\t) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\tid, _ := strconv.Atoi(activityID)\n\t\ts.Equal(int(expectedActivityID), id)\n\t\tbuf := bytes.NewReader(input)\n\t\tvar in int32\n\t\tbinary.Read(buf, binary.LittleEndian, &in)\n\t\ts.Equal(expectedActivityID, in)\n\t\texpectedActivityID++\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          domain,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\tfor run := 0; run < numRuns; run++ {\n\t\tfor i := 0; i < numActivities; i++ {\n\t\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\t\ts.Nil(err)\n\t\t\tif i%2 == 0 {\n\t\t\t\terr = poller.PollAndProcessActivityTask(false)\n\t\t\t} else { // just for testing respondActivityTaskCompleteByID\n\t\t\t\terr = poller.PollAndProcessActivityTaskWithID(false)\n\t\t\t}\n\t\t\ts.Logger.Info(\"PollAndProcessActivityTask\", tag.Error(err))\n\t\t\ts.Nil(err)\n\t\t}\n\n\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Nil(err)\n\t}\n\n\ts.True(workflowComplete)\n\tfor run := 1; run < numRuns; run++ {\n\t\ts.NotEqual(RunIDs[run-1], RunIDs[run])\n\t}\n\treturn RunIDs\n}\n"
  },
  {
    "path": "host/async_wf_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build !race && asyncwfintegration\n// +build !race,asyncwfintegration\n\n/*\nTo run locally:\n\n1. Stop the previous run if any\n\n\tdocker compose -f docker/github_actions/docker-compose-local-async-wf.yml down\n\n2. Build the integration-test-async-wf image\n\n\tdocker compose -f docker/github_actions/docker-compose-local-async-wf.yml build integration-test-async-wf\n\n3. Run the test in the docker container\n\n\tdocker compose -f docker/github_actions/docker-compose-local-async-wf.yml run --rm integration-test-async-wf\n\n4. Full test run logs can be found at test.log file\n*/\npackage host\n\nimport (\n\t\"flag\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/types\"\n\n\t_ \"github.com/uber/cadence/common/asyncworkflow/queue/kafka\" // needed to load kafka asyncworkflow queue\n)\n\nfunc TestAsyncWFIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tconfPath := \"testdata/integration_async_wf_with_kafka_cluster.yaml\"\n\tclusterConfig, err := GetTestClusterConfig(confPath)\n\tif err != nil {\n\t\tt.Fatalf(\"failed creating cluster config from %s, err: %v\", confPath, err)\n\t}\n\n\tclusterConfig.TimeSource = clock.NewMockedTimeSource()\n\tclusterConfig.FrontendDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.FrontendFailoverCoolDown:        time.Duration(0),\n\t\tdynamicproperties.EnableReadFromClosedExecutionV2: true,\n\t}\n\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(AsyncWFIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *AsyncWFIntegrationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.SecondaryDomainName = s.RandomizeStr(\"unused-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.SecondaryDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ts.domainCacheRefresh()\n}\n\nfunc (s *AsyncWFIntegrationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *AsyncWFIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *AsyncWFIntegrationSuite) TestStartWorkflowExecutionAsync() {\n\ttests := []struct {\n\t\tname             string\n\t\twantStartFailure bool\n\t\tasyncWFCfg       *types.AsyncWorkflowConfiguration\n\t\tsecondaryCfg     *types.AsyncWorkflowConfiguration\n\t}{\n\t\t{\n\t\t\tname:             \"start workflow execution async fails because domain missing async queue\",\n\t\t\twantStartFailure: true,\n\t\t\tsecondaryCfg: &types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             true,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"start workflow execution async fails because async queue is disabled\",\n\t\t\tasyncWFCfg: &types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled: false,\n\t\t\t},\n\t\t\twantStartFailure: true,\n\t\t\tsecondaryCfg: &types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             true,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"start workflow execution async succeeds and workflow starts\",\n\t\t\tasyncWFCfg: &types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             true,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t\tsecondaryCfg: &types.AsyncWorkflowConfiguration{\n\t\t\t\tEnabled:             false,\n\t\t\t\tPredefinedQueueName: \"test-async-wf-queue\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ttc := tc\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\t// advance the time so each test has a unique start time\n\t\t\ts.TestClusterConfig.TimeSource.Advance(time.Second)\n\n\t\t\tctx, cancel := createContext()\n\t\t\tdefer cancel()\n\n\t\t\tif tc.asyncWFCfg != nil {\n\t\t\t\t_, err := s.AdminClient.UpdateDomainAsyncWorkflowConfiguraton(ctx, &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\t\tDomain:        s.DomainName,\n\t\t\t\t\tConfiguration: tc.asyncWFCfg,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Fatalf(\"UpdateDomainAsyncWorkflowConfiguraton() failed: %v\", err)\n\t\t\t\t}\n\n\t\t\t\ts.domainCacheRefresh()\n\t\t\t}\n\n\t\t\tif tc.secondaryCfg != nil {\n\t\t\t\t_, err := s.AdminClient.UpdateDomainAsyncWorkflowConfiguraton(ctx, &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\t\t\t\tDomain:        s.SecondaryDomainName,\n\t\t\t\t\tConfiguration: tc.secondaryCfg,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Fatalf(\"UpdateDomainAsyncWorkflowConfiguraton() failed: %v\", err)\n\t\t\t\t}\n\n\t\t\t\ts.domainCacheRefresh()\n\t\t\t}\n\n\t\t\tstartTime := s.TestClusterConfig.TimeSource.Now().UnixNano()\n\t\t\twfID := fmt.Sprintf(\"async-wf-integration-start-workflow-test-%d\", startTime)\n\t\t\twfType := \"async-wf-integration-start-workflow-test-type\"\n\t\t\ttaskList := \"async-wf-integration-start-workflow-test-tasklist\"\n\t\t\tidentity := \"worker1\"\n\n\t\t\tasyncReq := &types.StartWorkflowExecutionAsyncRequest{\n\t\t\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tRequestID:  uuid.New(),\n\t\t\t\t\tDomain:     s.DomainName,\n\t\t\t\t\tWorkflowID: wfID,\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: wfType,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: taskList,\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\t\tIdentity:                            identity,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := s.Engine.StartWorkflowExecutionAsync(ctx, asyncReq)\n\t\t\tif tc.wantStartFailure != (err != nil) {\n\t\t\t\tt.Errorf(\"StartWorkflowExecutionAsync() failed: %v, wantStartFailure: %v\", err, tc.wantStartFailure)\n\t\t\t}\n\n\t\t\tif err != nil || tc.wantStartFailure {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// there's no worker or poller for async workflow, so we just validate whether it started.\n\t\t\t// this is sufficient to verify the async workflow start path.\n\t\t\tfor i := 0; i < 30; i++ {\n\t\t\t\tresp, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: s.DomainName,\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: wfID,\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Logf(\"Workflow execution not found yet. DescribeWorkflowExecution() returned err: %v\", err)\n\t\t\t\t\ttime.Sleep(time.Second)\n\t\t\t\t\ts.TestClusterConfig.TimeSource.Advance(time.Second)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tif resp.GetWorkflowExecutionInfo() != nil {\n\t\t\t\t\tt.Logf(\"DescribeWorkflowExecution() found the execution: %#v\", resp.GetWorkflowExecutionInfo())\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tt.Fatal(\"Async started workflow not found\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "host/cancel_workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestExternalRequestCancelWorkflowExecution() {\n\tid := \"integration-request-cancel-workflow-test\"\n\twt := \"integration-request-cancel-workflow-test-type\"\n\ttl := \"integration-request-cancel-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCancelWorkflowExecution.Ptr(),\n\t\t\tCancelWorkflowExecutionDecisionAttributes: &types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\t\tDetails: []byte(\"Cancelled\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"PollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.RequestCancelWorkflowExecution(ctx, &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.RequestCancelWorkflowExecution(ctx, &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.CancellationAlreadyRequestedError{}, err)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\texecutionCancelled := false\nGetHistoryLoop:\n\tfor i := 1; i < 3; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionCanceled {\n\t\t\ts.Logger.Warn(\"Execution not cancelled yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tcancelledEventAttributes := lastEvent.WorkflowExecutionCanceledEventAttributes\n\t\ts.Equal(\"Cancelled\", string(cancelledEventAttributes.Details))\n\t\texecutionCancelled = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(executionCancelled)\n}\n\nfunc (s *IntegrationSuite) TestRequestCancelWorkflowDecisionExecution() {\n\tid := \"integration-cancel-workflow-decision-test\"\n\twt := \"integration-cancel-workflow-decision-test-type\"\n\ttl := \"integration-cancel-workflow-decision-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tforeignRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.ForeignDomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe2, err0 := s.Engine.StartWorkflowExecution(ctx, foreignRequest)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution on foreign Domain: %v,  response: %v \\n\", tag.WorkflowDomainName(s.ForeignDomainName), tag.WorkflowRunID(we2.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeRequestCancelExternalWorkflowExecution.Ptr(),\n\t\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:     s.ForeignDomainName,\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we2.RunID,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tforeignActivityCount := int32(1)\n\tforeignActivityCounter := int32(0)\n\tforeignDtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif foreignActivityCounter < foreignActivityCount {\n\t\t\tforeignActivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, foreignActivityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(foreignActivityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(foreignActivityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(foreignActivityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCancelWorkflowExecution.Ptr(),\n\t\t\tCancelWorkflowExecutionDecisionAttributes: &types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\t\tDetails: []byte(\"Cancelled\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tforeignPoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.ForeignDomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: foreignDtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start both current and foreign workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"foreign PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\terr = foreignPoller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"foreign PollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Cancel the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tcancellationSent := false\n\tintiatedEventID := 10\nCheckHistoryLoopForCancelSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-2]\n\t\tif *lastEvent.EventType != types.EventTypeExternalWorkflowExecutionCancelRequested {\n\t\t\ts.Logger.Info(\"Cancellation still not sent.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForCancelSent\n\t\t}\n\n\t\texternalWorkflowExecutionCancelRequestedEvent := lastEvent.ExternalWorkflowExecutionCancelRequestedEventAttributes\n\t\ts.Equal(int64(intiatedEventID), externalWorkflowExecutionCancelRequestedEvent.InitiatedEventID)\n\t\ts.Equal(id, externalWorkflowExecutionCancelRequestedEvent.WorkflowExecution.WorkflowID)\n\t\ts.Equal(we2.RunID, externalWorkflowExecutionCancelRequestedEvent.WorkflowExecution.RunID)\n\n\t\tcancellationSent = true\n\t\tbreak\n\t}\n\n\ts.True(cancellationSent)\n\n\t// Accept cancellation.\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"foreign PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\texecutionCancelled := false\nGetHistoryLoop:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.ForeignDomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we2.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionCanceled {\n\t\t\ts.Logger.Warn(\"Execution not cancelled yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tcancelledEventAttributes := lastEvent.WorkflowExecutionCanceledEventAttributes\n\t\ts.Equal(\"Cancelled\", string(cancelledEventAttributes.Details))\n\t\texecutionCancelled = true\n\n\t\t// Find cancel requested event and verify it.\n\t\tvar cancelRequestEvent *types.HistoryEvent\n\t\tfor _, x := range history.Events {\n\t\t\tif *x.EventType == types.EventTypeWorkflowExecutionCancelRequested {\n\t\t\t\tcancelRequestEvent = x\n\t\t\t}\n\t\t}\n\n\t\ts.NotNil(cancelRequestEvent)\n\t\tcancelRequestEventAttributes := cancelRequestEvent.WorkflowExecutionCancelRequestedEventAttributes\n\t\ts.Equal(int64(intiatedEventID), *cancelRequestEventAttributes.ExternalInitiatedEventID)\n\t\ts.Equal(id, cancelRequestEventAttributes.ExternalWorkflowExecution.WorkflowID)\n\t\ts.Equal(we.RunID, cancelRequestEventAttributes.ExternalWorkflowExecution.RunID)\n\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(executionCancelled)\n}\n\nfunc (s *IntegrationSuite) TestRequestCancelWorkflowDecisionExecution_UnKnownTarget() {\n\tid := \"integration-cancel-unknown-workflow-decision-test\"\n\twt := \"integration-cancel-unknown-workflow-decision-test-type\"\n\ttl := \"integration-cancel-unknown-workflow-decision-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeRequestCancelExternalWorkflowExecution.Ptr(),\n\t\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:     s.ForeignDomainName,\n\t\t\t\tWorkflowID: \"workflow_not_exist\",\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Cancel the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tcancellationSentFailed := false\n\tintiatedEventID := 10\nCheckHistoryLoopForCancelSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-2]\n\t\tif *lastEvent.EventType != types.EventTypeRequestCancelExternalWorkflowExecutionFailed {\n\t\t\ts.Logger.Info(\"Cancellaton not cancelled yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForCancelSent\n\t\t}\n\n\t\trequestCancelExternalWorkflowExecutionFailedEvetn := lastEvent.RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\t\ts.Equal(int64(intiatedEventID), requestCancelExternalWorkflowExecutionFailedEvetn.InitiatedEventID)\n\t\ts.Equal(\"workflow_not_exist\", requestCancelExternalWorkflowExecutionFailedEvetn.WorkflowExecution.WorkflowID)\n\t\ts.Equal(we.RunID, requestCancelExternalWorkflowExecutionFailedEvetn.WorkflowExecution.RunID)\n\n\t\tcancellationSentFailed = true\n\t\tbreak\n\t}\n\n\ts.True(cancellationSentFailed)\n}\n"
  },
  {
    "path": "host/cassandra_lwt_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build !race && cassandralwt\n// +build !race,cassandralwt\n\n/*\nTo run locally:\n\n1. Stop the previous run if any\n\n\tdocker compose -f docker/github_actions/docker-compose-cassandra-lwt.yml down\n\n2. Build the integration-test-async-wf image\n\n\tdocker compose -f docker/github_actions/docker-compose-cassandra-lwt.yml build test-cass-lwt\n\n3. Run the test in the docker container\n\n\tdocker compose -f docker/github_actions/docker-compose-cassandra-lwt.yml run --rm test-cass-lwt\n\n4. Full test run logs can be found at test.log file\n*/\n\n/*\n-- Benchmark Results --\n\nSingle node cassandra cluster, maxConns=2, replicas=1:\n\tTotal updates: 1000, concurrency: 1, success: 1000, failed: 0, avg duration: 7.41ms, max duration: 26.9ms, min duration: 4.8ms, elapsed: 7.42s\n\tTotal updates: 1000, concurrency: 10, success: 1000, failed: 0, avg duration: 54.54ms, max duration: 808.2ms, min duration: 2.8ms, elapsed: 5.16s\n\tTotal updates: 1000, concurrency: 20, success: 999, failed: 1, avg duration: 90.61ms, max duration: 1.075s, min duration: 2.8ms, elapsed: 4.63s\n\tTotal updates: 1000, concurrency: 40, success: 993, failed: 7, avg duration: 145.95ms, max duration: 1.071s, min duration: 2.2ms, elapsed: 3.78s\n\tTotal updates: 1000, concurrency: 80, success: 976, failed: 24, avg duration: 254.02ms, max duration: 1.093s, min duration: 1.4ms, elapsed: 3.36s\n\nTwo nodes cassandra cluster, maxConns=2, replicas=1:\n\tTotal updates: 1000, concurrency: 10, success: 1000, failed: 0, avg duration: 52.37s, max duration: 681.20ms, min duration: 2.91ms, elapsed: 5.29s\n\tTotal updates: 1000, concurrency: 100, success: 949, failed: 51, avg duration: 283.95ms, max duration: 1.11s, min duration: 2.21ms, elapsed: 3.09s\n\nTwo nodes cassandra cluster, maxConns=2, replicas=2:\n\tTotal updates: 1000, concurrency: 10, success: 898, failed: 102, avg duration: 367.31ms, max duration: 1.12s, min duration: 7.98ms, elapsed: 36.88s\n    Total updates: 1000, concurrency: 60, success: 160, failed: 840, avg duration: 951.3ms, max duration: 1.11s, min duration: 8.99ms, elapsed: 16.35s\n    Total updates: 1000, concurrency: 80, success: 71, failed: 929, avg duration: 1.00s, max duration: 1.12s, min duration: 8.10ms, elapsed: 13.02s\n\tTotal updates: 1000, concurrency: 100, success: 38, failed: 962, avg duration: 1.02s, max duration: 1.11s, min duration: 7.89ms, elapsed: 10.52s\n\nTwo nodes cassandra cluster, maxConns=0, replicas=2:\n    Total updates: 1000, concurrency: 1, success: 1000, failed: 0, avg duration: 11.75ms, max duration: 50.09ms, min duration: 6.69ms, elapsed: 11.76s\n    Total updates: 1000, concurrency: 10, success: 900, failed: 100, avg duration: 370.96ms, max duration: 1.12s, min duration: 7.92ms, elapsed: 37.31s\n    Total updates: 1000, concurrency: 20, success: 642, failed: 358, avg duration: 602.63ms, max duration: 1.12s, min duration: 5.95ms, elapsed: 30.40s\n    Total updates: 1000, concurrency: 40, success: 365, failed: 635, avg duration: 826.45ms, max duration: 1.13s, min duration: 7.83ms, elapsed: 21.12s\n    Total updates: 1000, concurrency: 80, success: 64, failed: 936, avg duration: 1.00s, max duration: 1.12s, min duration: 6.87ms, elapsed: 12.93s\n\tTotal updates: 1000, concurrency: 100, success: 29, failed: 971, avg duration: 1.02s, max duration: 1.11s, min duration: 36.97ms, elapsed: 10.51s\n*/\n\npackage host\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nconst (\n\tupdateRequestsChannelSize = 1000\n\tconcurrency               = 10\n\ttotalUpdates              = 1000\n\treplicas                  = 1\n\tmaxConns                  = 2\n)\n\ntype updateStats struct {\n\tupdateNum int\n\tduration  time.Duration\n\tts        time.Time\n\terr       error\n}\n\ntype aggStats struct {\n\tsuccessCnt    int\n\tfailCnt       int\n\ttotalDuration time.Duration\n\tmaxDuration   time.Duration\n\tminDuration   time.Duration\n\tlastUpdate    time.Time\n\terrs          error\n}\n\nfunc TestCassandraLWT(t *testing.T) {\n\tflag.Parse()\n\ttestflags.RequireCassandra(t)\n\tt.Logf(\"Running Cassandra LWT tests, concurrency: %d\", concurrency)\n\n\ttestBase := public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{\n\t\tReplicas: replicas,\n\t\tMaxConns: maxConns,\n\t})\n\ttestBase.Setup()\n\tdefer testBase.TearDownWorkflowStore()\n\n\t// Create workflows\n\twfID := fmt.Sprintf(\"workflow-%v\", 0)\n\twfInfo, err := createWorkflow(testBase, wfID)\n\tif err != nil {\n\t\tt.Fatalf(\"createWorkflow failed: %v\", err)\n\t}\n\tupdateReq := infoToUpdateReq(testBase, wfInfo)\n\n\tt.Logf(\"Created %d workflows\", concurrency)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\tupdateRequestsCh := make(chan int, updateRequestsChannelSize)\n\n\t// Start a goroutine to consume the update stats\n\tvar aggStats aggStats\n\tstatsCh := make(chan updateStats, totalUpdates)\n\tvar statCollectorWG sync.WaitGroup\n\tstatCollectorWG.Add(1)\n\tgo func() {\n\t\tdefer statCollectorWG.Done()\n\t\tfor i := 0; i < totalUpdates; i++ {\n\t\t\tstats := <-statsCh\n\t\t\tif stats.err != nil {\n\t\t\t\taggStats.errs = multierr.Append(aggStats.errs, fmt.Errorf(\"updateNum: %v, duration: %v, err: %v\", stats.updateNum, stats.duration, stats.err))\n\t\t\t\taggStats.failCnt++\n\t\t\t} else {\n\t\t\t\taggStats.successCnt++\n\t\t\t}\n\n\t\t\taggStats.totalDuration += stats.duration\n\t\t\tif aggStats.maxDuration < stats.duration {\n\t\t\t\taggStats.maxDuration = stats.duration\n\t\t\t}\n\t\t\tif aggStats.minDuration == 0 || aggStats.minDuration > stats.duration {\n\t\t\t\taggStats.minDuration = stats.duration\n\t\t\t}\n\n\t\t\tif aggStats.lastUpdate.Before(stats.ts) {\n\t\t\t\taggStats.lastUpdate = stats.ts\n\t\t\t}\n\t\t}\n\t}()\n\n\t// Start concurrent handlers to update the workflows. Each one will update its own workflow\n\tvar updateHandlersWG sync.WaitGroup\n\tfor i := 0; i < concurrency; i++ {\n\t\tupdateHandlersWG.Add(1)\n\t\tgo updateHandler(ctx, testBase, updateRequestsCh, statsCh, &updateHandlersWG, i, updateReq)\n\t}\n\n\t// wait a bit before starting the updates\n\ttime.Sleep(1 * time.Second)\n\n\tnow := time.Now()\n\tfor i := 0; i < totalUpdates; i++ {\n\t\tupdateRequestsCh <- i\n\t}\n\n\t// wait for the updates to complete\n\tstatCollectorWG.Wait()\n\n\tt.Logf(\"Total updates: %v, concurrency: %d, success: %v, failed: %v, avg duration: %v, max duration: %v, min duration: %v, elapsed: %v\",\n\t\ttotalUpdates,\n\t\tconcurrency,\n\t\taggStats.successCnt,\n\t\taggStats.failCnt,\n\t\taggStats.totalDuration/time.Duration(totalUpdates),\n\t\taggStats.maxDuration,\n\t\taggStats.minDuration,\n\t\taggStats.lastUpdate.Sub(now),\n\t)\n\tif aggStats.errs != nil {\n\t\tmsg := aggStats.errs.Error()\n\t\tlen := len(msg)\n\t\tif len > 1000 {\n\t\t\tlen = 1000\n\t\t}\n\t\tt.Errorf(\"Errors: %s\", msg[:len-1])\n\t}\n\n\t// close the channel to signal the handlers to exit\n\tclose(updateRequestsCh)\n\t// wait for the handlers to exit\n\tupdateHandlersWG.Wait()\n}\n\nfunc updateHandler(\n\tctx context.Context,\n\ts *persistencetests.TestBase,\n\tincomingUpdatesCh chan int,\n\tstatsCh chan updateStats,\n\twg *sync.WaitGroup,\n\thandlerNo int,\n\tupdateReq *persistence.UpdateWorkflowExecutionRequest,\n) {\n\tdefer wg.Done()\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\ts.T().Logf(\"updateHandler %v exiting because context done\", handlerNo)\n\t\t\treturn\n\t\tcase updateNum, ok := <-incomingUpdatesCh:\n\t\t\tif !ok {\n\t\t\t\ts.T().Logf(\"updateHandler %v exiting because channel closed\", handlerNo)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tstart := time.Now()\n\t\t\terr := updateWorkflow(s, updateReq)\n\t\t\tend := time.Now()\n\n\t\t\tstatsCh <- updateStats{\n\t\t\t\tupdateNum: updateNum,\n\t\t\t\tduration:  end.Sub(start),\n\t\t\t\tts:        end,\n\t\t\t\terr:       err,\n\t\t\t}\n\t\t}\n\t}\n}\n\n// createWorkflow creates a new workflow and returns the workflow info to be used for updates\nfunc createWorkflow(s *persistencetests.TestBase, wfID string) (*persistence.WorkflowMutableState, error) {\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\tdefer cancel()\n\n\tdomainID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: wfID,\n\t\tRunID:      uuid.New(),\n\t}\n\n\ttasklist := \"tasklist1\"\n\tworkflowType := \"workflowtype1\"\n\tworkflowTimeout := int32(10)\n\tdecisionTimeout := int32(14)\n\tlastProcessedEventID := int64(0)\n\tnextEventID := int64(3)\n\tcsum := checksum.Checksum{\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tVersion: 22,\n\t\tValue:   uuid.NewRandom(),\n\t}\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\treq := &persistence.CreateWorkflowExecutionRequest{\n\t\tNewWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tCreateRequestID:             uuid.New(),\n\t\t\t\tDomainID:                    domainID,\n\t\t\t\tWorkflowID:                  workflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:                       workflowExecution.GetRunID(),\n\t\t\t\tFirstExecutionRunID:         workflowExecution.GetRunID(),\n\t\t\t\tTaskList:                    tasklist,\n\t\t\t\tWorkflowTypeName:            workflowType,\n\t\t\t\tWorkflowTimeout:             workflowTimeout,\n\t\t\t\tDecisionStartToCloseTimeout: decisionTimeout,\n\t\t\t\tLastFirstEventID:            constants.FirstEventID,\n\t\t\t\tNextEventID:                 nextEventID,\n\t\t\t\tLastProcessedEvent:          lastProcessedEventID,\n\t\t\t\tState:                       persistence.WorkflowStateCreated,\n\t\t\t\tCloseStatus:                 persistence.WorkflowCloseStatusNone,\n\t\t\t},\n\t\t\tExecutionStats:   &persistence.ExecutionStats{},\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    persistence.CreateWorkflowModeBrandNew,\n\t}\n\n\t_, err := s.ExecutionManager.CreateWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"ExecutionManager.CreateWorkflowExecution failed: %v\", err)\n\t}\n\n\tinfo, err := s.GetWorkflowExecutionInfo(ctx, domainID, workflowExecution)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"GetWorkflowExecutionInfo failed: %v\", err)\n\t}\n\tif info.ExecutionInfo.State != persistence.WorkflowStateCreated {\n\t\treturn nil, fmt.Errorf(\"Unexpected state: %v\", info.ExecutionInfo.State)\n\t}\n\tif info.ExecutionInfo.CloseStatus != persistence.WorkflowCloseStatusNone {\n\t\treturn nil, fmt.Errorf(\"Unexpected close status: %v\", info.ExecutionInfo.CloseStatus)\n\t}\n\n\treturn info, nil\n}\n\nfunc infoToUpdateReq(s *persistencetests.TestBase, info *persistence.WorkflowMutableState) *persistence.UpdateWorkflowExecutionRequest {\n\tupdatedInfo := info.ExecutionInfo\n\tupdatedStats := info.ExecutionStats\n\tupdatedInfo.State = persistence.WorkflowStateRunning\n\tnextEventID := int64(3)\n\tcsum := checksum.Checksum{\n\t\tFlavor:  checksum.FlavorIEEECRC32OverThriftBinary,\n\t\tVersion: 22,\n\t\tValue:   uuid.NewRandom(),\n\t}\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\t{\n\t\t\tEventID: nextEventID,\n\t\t\tVersion: constants.EmptyVersion,\n\t\t},\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\treturn &persistence.UpdateWorkflowExecutionRequest{\n\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\tExecutionInfo:    updatedInfo,\n\t\t\tExecutionStats:   updatedStats,\n\t\t\tCondition:        nextEventID,\n\t\t\tChecksum:         csum,\n\t\t\tVersionHistories: versionHistories,\n\t\t},\n\t\tRangeID: s.ShardInfo.RangeID,\n\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t}\n}\n\nfunc updateWorkflow(s *persistencetests.TestBase, req *persistence.UpdateWorkflowExecutionRequest) error {\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\tdefer cancel()\n\n\t_, err := s.ExecutionManager.UpdateWorkflowExecution(ctx, req)\n\tif err == nil {\n\t\treturn nil\n\t}\n\n\treturn fmt.Errorf(\"ExecutionManager.UpdateWorkflowExecution failed: %v\", err)\n}\n"
  },
  {
    "path": "host/cli/cassandra/cassandra_tool_cqlclient_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/testflags\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\" // needed to load the default gocql client\n)\n\ntype (\n\tCQLClientTestSuite struct {\n\t\ttest.DBTestBase\n\t}\n)\n\nvar _ test.DB = (cassandra.CqlClient)(nil)\n\nfunc TestCQLClientTestSuite(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\tsuite.Run(t, new(CQLClientTestSuite))\n}\n\nfunc (s *CQLClientTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *CQLClientTestSuite) SetupSuite() {\n\tclient, err := NewTestCQLClient(cassandra.SystemKeyspace)\n\tif err != nil {\n\t\ts.Log.Fatal(\"error creating CQLClient, \", tag.Error(err))\n\t}\n\ts.SetupSuiteBase(client)\n}\n\nfunc (s *CQLClientTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\nfunc (s *CQLClientTestSuite) TestParseCQLFile() {\n\ts.RunParseFileTest(CreateTestCQLFileContent())\n}\n\nfunc (s *CQLClientTestSuite) TestCQLClient() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\ts.RunCreateTest(client)\n\ts.RunUpdateTest(client)\n\ts.RunDropTest(client)\n\tclient.Close()\n}\n"
  },
  {
    "path": "host/cli/cassandra/cassandra_tool_setupTask_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/testflags\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n)\n\ntype (\n\tSetupSchemaTestSuite struct {\n\t\ttest.SetupSchemaTestBase\n\t\tclient cassandra.CqlClient\n\t}\n)\n\nfunc TestSetupSchemaTestSuite(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\tsuite.Run(t, new(SetupSchemaTestSuite))\n}\n\nfunc (s *SetupSchemaTestSuite) SetupSuite() {\n\tos.Setenv(\"CASSANDRA_HOST\", environment.GetCassandraAddress())\n\tclient, err := NewTestCQLClient(cassandra.SystemKeyspace)\n\tif err != nil {\n\t\tlog.Fatal(\"Error creating CQLClient\")\n\t}\n\ts.client = client\n\ts.SetupSuiteBase(client)\n}\n\nfunc (s *SetupSchemaTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\nfunc (s *SetupSchemaTestSuite) TestCreateKeyspace() {\n\ts.Nil(cassandra.RunTool([]string{\"./tool\", \"create\", \"-k\", \"foobar123\", \"--rf\", \"1\"}))\n\terr := s.client.DropKeyspace(\"foobar123\")\n\ts.Nil(err)\n}\n\nfunc (s *SetupSchemaTestSuite) TestSetupSchema() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\ts.RunSetupTest(cassandra.BuildCLIOptions(), client, \"-k\", CreateTestCQLFileContent(), []string{\"tasks\", \"events\"})\n}\n"
  },
  {
    "path": "host/cli/cassandra/cassandra_tool_updateTask_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"log\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/schema/cassandra\"\n\t\"github.com/uber/cadence/testflags\"\n\tcassandra2 \"github.com/uber/cadence/tools/cassandra\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n)\n\ntype UpdateSchemaTestSuite struct {\n\ttest.UpdateSchemaTestBase\n}\n\nfunc TestUpdateSchemaTestSuite(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\tsuite.Run(t, new(UpdateSchemaTestSuite))\n}\n\nfunc (s *UpdateSchemaTestSuite) SetupSuite() {\n\tclient, err := NewTestCQLClient(cassandra2.SystemKeyspace)\n\tif err != nil {\n\t\tlog.Fatal(\"Error creating CQLClient\")\n\t}\n\ts.SetupSuiteBase(client)\n}\n\nfunc (s *UpdateSchemaTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\nfunc (s *UpdateSchemaTestSuite) TestUpdateSchema() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\tdefer client.Close()\n\ts.RunUpdateSchemaTest(cassandra2.BuildCLIOptions(), client, \"-k\", CreateTestCQLFileContent(), []string{\"events\", \"tasks\"})\n}\n\nfunc (s *UpdateSchemaTestSuite) TestDryrun() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\tdefer client.Close()\n\tdir := rootRelativePath + \"schema/cassandra/cadence/versioned\"\n\ts.RunDryrunTest(cassandra2.BuildCLIOptions(), client, \"-k\", dir, cassandra.Version)\n}\n\nfunc (s *UpdateSchemaTestSuite) TestVisibilityDryrun() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\tdefer client.Close()\n\tdir := rootRelativePath + \"schema/cassandra/visibility/versioned\"\n\ts.RunDryrunTest(cassandra2.BuildCLIOptions(), client, \"-k\", dir, cassandra.VisibilityVersion)\n}\n\nfunc (s *UpdateSchemaTestSuite) TestShortcut() {\n\tclient, err := NewTestCQLClient(s.DBName)\n\ts.Nil(err)\n\tdefer client.Close()\n\tdir := rootRelativePath + \"schema/cassandra/cadence/versioned\"\n\n\tcqlshArgs := []string{\"--cqlversion=3.4.6\", \"-e\", \"DESC KEYSPACE %s;\"}\n\tif cassandraHost := os.Getenv(\"CASSANDRA_HOST\"); cassandraHost != \"\" {\n\t\tcqlshArgs = append(cqlshArgs, cassandraHost)\n\t}\n\ts.RunShortcutTest(cassandra2.BuildCLIOptions(), client, \"-k\", dir, \"cqlsh\", cqlshArgs...)\n}\n"
  },
  {
    "path": "host/cli/cassandra/cassandra_tool_version_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"log\"\n\t\"math/rand\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tcassandra_db \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/testflags\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n)\n\ntype (\n\tVersionTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestVersionTestSuite(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\tsuite.Run(t, new(VersionTestSuite))\n}\n\nfunc (s *VersionTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *VersionTestSuite) TestVerifyCompatibleVersion() {\n\tkeyspace := \"cadence_test\"\n\tvisKeyspace := \"cadence_visibility_test\"\n\tcqlFile := rootRelativePath + \"schema/cassandra/cadence/schema.cql\"\n\tvisCqlFile := rootRelativePath + \"schema/cassandra/visibility/schema.cql\"\n\n\tdefer s.createKeyspace(keyspace)()\n\tdefer s.createKeyspace(visKeyspace)()\n\ts.Nil(cassandra.RunTool([]string{\n\t\t\"./tool\", \"-k\", keyspace, \"-q\", \"setup-schema\", \"-f\", cqlFile, \"-version\", \"10.0\", \"-o\",\n\t}))\n\ts.Nil(cassandra.RunTool([]string{\n\t\t\"./tool\", \"-k\", visKeyspace, \"-q\", \"setup-schema\", \"-f\", visCqlFile, \"-version\", \"10.0\", \"-o\",\n\t}))\n\n\tdefaultCfg := config.NoSQL{\n\t\tPluginName: cassandra_db.PluginName,\n\t\tHosts:      environment.GetCassandraAddress(),\n\t\tPort:       cassandra.DefaultCassandraPort,\n\t\tUser:       environment.GetCassandraUsername(),\n\t\tPassword:   environment.GetCassandraPassword(),\n\t\tKeyspace:   keyspace,\n\t}\n\tvisibilityCfg := defaultCfg\n\tvisibilityCfg.Keyspace = visKeyspace\n\tcfg := config.Persistence{\n\t\tDefaultStore:    \"default\",\n\t\tVisibilityStore: \"visibility\",\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"default\":    {NoSQL: &defaultCfg},\n\t\t\t\"visibility\": {NoSQL: &visibilityCfg},\n\t\t},\n\t\tTransactionSizeLimit: dynamicproperties.GetIntPropertyFn(constants.DefaultTransactionSizeLimit),\n\t\tErrorInjectionRate:   dynamicproperties.GetFloatPropertyFn(0),\n\t}\n\ts.NoError(cassandra.VerifyCompatibleVersion(cfg, gocql.All))\n}\n\nfunc (s *VersionTestSuite) TestCheckCompatibleVersion() {\n\tflags := []struct {\n\t\texpectedVersion string\n\t\tactualVersion   string\n\t\terrStr          string\n\t\texpectedFail    bool\n\t}{\n\t\t{\"2.0\", \"1.0\", \"version mismatch\", false},\n\t\t{\"1.0\", \"1.0\", \"\", false},\n\t\t{\"1.0\", \"2.0\", \"\", false},\n\t\t{\"1.0\", \"abc\", \"reading schema version: table schema_version does not exist\", true},\n\t}\n\tfor _, flag := range flags {\n\t\ts.runCheckCompatibleVersion(flag.expectedVersion, flag.actualVersion, flag.errStr, flag.expectedFail)\n\t}\n}\n\nfunc (s *VersionTestSuite) createKeyspace(keyspace string) func() {\n\n\tprotoVersion, err := environment.GetCassandraProtoVersion()\n\ts.NoError(err)\n\tcfg := &cassandra.CQLClientConfig{\n\t\tHosts:        environment.GetCassandraAddress(),\n\t\tPort:         cassandra.DefaultCassandraPort,\n\t\tKeyspace:     \"system\",\n\t\tTimeout:      cassandra.DefaultTimeout,\n\t\tNumReplicas:  1,\n\t\tProtoVersion: protoVersion,\n\t}\n\tclient, err := cassandra.NewCQLClient(cfg, gocql.All)\n\ts.NoError(err)\n\n\terr = client.CreateKeyspace(keyspace)\n\tif err != nil {\n\t\tlog.Fatalf(\"error creating Keyspace, err=%v\", err)\n\t}\n\treturn func() {\n\t\ts.NoError(client.DropKeyspace(keyspace))\n\t\tclient.Close()\n\t}\n}\n\nfunc (s *VersionTestSuite) runCheckCompatibleVersion(\n\texpected string, actual string, errStr string, expectedFail bool,\n) {\n\tr := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tkeyspace := fmt.Sprintf(\"version_test_%v\", r.Int63())\n\tdefer s.createKeyspace(keyspace)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + keyspace\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, actual)\n\tif expected != actual {\n\t\ts.createSchemaForVersion(subdir, expected)\n\t}\n\n\tcqlFile := subdir + \"/v\" + actual + \"/tmp.cql\"\n\tif expectedFail {\n\t\ts.Error(cassandra.RunTool([]string{\n\t\t\t\"./tool\", \"-k\", keyspace, \"setup-schema\", \"-f\", cqlFile, \"-version\", actual, \"-o\",\n\t\t}))\n\t\tos.RemoveAll(subdir + \"/v\" + actual)\n\t} else {\n\t\ts.NoError(cassandra.RunTool([]string{\n\t\t\t\"./tool\", \"-k\", keyspace, \"setup-schema\", \"-f\", cqlFile, \"-version\", actual, \"-o\",\n\t\t}))\n\t}\n\n\tcfg := config.NoSQL{\n\t\tPluginName: cassandra_db.PluginName,\n\t\tHosts:      environment.GetCassandraAddress(),\n\t\tPort:       cassandra.DefaultCassandraPort,\n\t\tUser:       environment.GetCassandraUsername(),\n\t\tPassword:   environment.GetCassandraPassword(),\n\t\tKeyspace:   keyspace,\n\t}\n\terr := cassandra.CheckCompatibleVersion(cfg, expected, gocql.All)\n\tif len(errStr) > 0 {\n\t\ts.Errorf(err, \"error=%v\", errStr)\n\t\ts.Contains(err.Error(), errStr)\n\t} else {\n\t\ts.NoError(err)\n\t}\n}\n\nfunc (s *VersionTestSuite) createSchemaForVersion(subdir string, v string) {\n\tvDir := subdir + \"/v\" + v\n\ts.NoError(os.Mkdir(vDir, os.FileMode(0744)))\n\tcqlFile := vDir + \"/tmp.cql\"\n\ts.NoError(ioutil.WriteFile(cqlFile, []byte{}, os.FileMode(0644)))\n}\n"
  },
  {
    "path": "host/cli/cassandra/utils.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/tools/cassandra\"\n)\n\n// NOTE: change this when moving the test files around during refactoring\n// Path to root folder of cadence repo\nconst rootRelativePath = \"../../../\"\n\nfunc NewTestCQLClient(keyspace string) (cassandra.CqlClient, error) {\n\tprotoVersion, err := environment.GetCassandraProtoVersion()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn cassandra.NewCQLClient(&cassandra.CQLClientConfig{\n\t\tHosts:                 environment.GetCassandraAddress(),\n\t\tPort:                  cassandra.DefaultCassandraPort,\n\t\tKeyspace:              keyspace,\n\t\tTimeout:               cassandra.DefaultTimeout,\n\t\tUser:                  environment.GetCassandraUsername(),\n\t\tPassword:              environment.GetCassandraPassword(),\n\t\tAllowedAuthenticators: environment.GetCassandraAllowedAuthenticators(),\n\t\tNumReplicas:           1,\n\t\tProtoVersion:          protoVersion,\n\t}, gocql.All)\n}\n\nfunc CreateTestCQLFileContent() string {\n\treturn `\n-- test cql file content\n\nCREATE TABLE events (\n  domain_id      uuid,\n  workflow_id    text,\n  run_id         uuid,\n  -- We insert a batch of events with each append transaction.\n  -- This field stores the event id of first event in the batch.\n  first_event_id bigint,\n  range_id       bigint,\n  tx_id          bigint,\n  data           blob, -- Batch of workflow execution history events as a blob\n  data_encoding  text, -- Protocol used for history serialization\n  data_version   int,  -- history blob version\n  PRIMARY KEY ((domain_id, workflow_id, run_id), first_event_id)\n);\n\n-- Stores activity or workflow tasks\nCREATE TABLE tasks (\n  domain_id        uuid,\n  task_list_name   text,\n  task_list_type   int, -- enum TaskListType {ActivityTask, DecisionTask}\n  type             int, -- enum rowType {Task, TaskList}\n  task_id          bigint,  -- unique identifier for tasks, monotonically increasing\n  range_id         bigint static, -- Used to ensure that only one process can write to the table\n  task             text,\n  task_list        text,\n  PRIMARY KEY ((domain_id, task_list_name, task_list_type), type, task_id)\n);\n\n`\n}\n"
  },
  {
    "path": "host/cli/sql/cli_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nfunc TestMySQLConnTestSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\tsuite.Run(t, NewSQLConnTestSuite(mysql.PluginName))\n}\n\nfunc TestMySQLHandlerTestSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\tsuite.Run(t, NewHandlerTestSuite(mysql.PluginName))\n}\n\nfunc TestMySQLSetupSchemaTestSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\tsuite.Run(t, NewSetupSchemaTestSuite(mysql.PluginName))\n}\n\nfunc TestMySQLUpdateSchemaTestSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\tsuite.Run(t, NewUpdateSchemaTestSuite(mysql.PluginName))\n}\n\nfunc TestMySQLVersionTestSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\tsuite.Run(t, NewVersionTestSuite(mysql.PluginName))\n}\n"
  },
  {
    "path": "host/cli/sql/connTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"net\"\n\t\"strconv\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\ntype (\n\t// SQLConnTestSuite defines a test suite\n\tSQLConnTestSuite struct {\n\t\ttest.DBTestBase\n\t\tpluginName string\n\t}\n)\n\nvar _ test.DB = (*sql.Connection)(nil)\n\n// NewSQLConnTestSuite returns the test suite\nfunc NewSQLConnTestSuite(pluginName string) *SQLConnTestSuite {\n\treturn &SQLConnTestSuite{\n\t\tpluginName: pluginName,\n\t}\n}\n\n// SetupTest setups test\nfunc (s *SQLConnTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\n// SetupSuite setups test suite\nfunc (s *SQLConnTestSuite) SetupSuite() {\n\tconn, err := newTestConn(\"\", s.pluginName)\n\tif err != nil {\n\t\tlog.Fatal(fmt.Sprintf(\"failed creating sql conn with error: %v\", tag.Error(err)))\n\t}\n\ts.SetupSuiteBase(conn)\n}\n\n// TearDownSuite tear down test suite\nfunc (s *SQLConnTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\n// TestParseCQLFile test\nfunc (s *SQLConnTestSuite) TestParseCQLFile() {\n\ts.RunParseFileTest(createTestSQLFileContent())\n}\n\n// TestSQLConn test\n// TODO refactor the whole package to support testing against Postgres\n// https://github.com/uber/cadence/issues/2856\nfunc (s *SQLConnTestSuite) TestSQLConn() {\n\tport, err := environment.GetMySQLPort()\n\ts.Nil(err)\n\tconn, err := sql.NewConnection(&config.SQL{\n\t\tConnectAddr: net.JoinHostPort(\n\t\t\tenvironment.GetMySQLAddress(),\n\t\t\tstrconv.Itoa(port),\n\t\t),\n\t\tUser:          environment.GetMySQLUser(),\n\t\tPassword:      environment.GetMySQLPassword(),\n\t\tPluginName:    s.pluginName,\n\t\tDatabaseName:  s.DBName,\n\t\tEncodingType:  \"thriftrw\",\n\t\tDecodingTypes: []string{\"thriftrw\"},\n\t})\n\ts.Nil(err)\n\ts.RunCreateTest(conn)\n\ts.RunUpdateTest(conn)\n\ts.RunDropTest(conn)\n\tconn.Close()\n}\n\nfunc newTestConn(database, pluginName string) (*sql.Connection, error) {\n\tport, err := environment.GetMySQLPort()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn sql.NewConnection(&config.SQL{\n\t\tConnectAddr: net.JoinHostPort(\n\t\t\tenvironment.GetMySQLAddress(),\n\t\t\tstrconv.Itoa(port),\n\t\t),\n\t\tUser:          environment.GetMySQLUser(),\n\t\tPassword:      environment.GetMySQLPassword(),\n\t\tPluginName:    pluginName,\n\t\tDatabaseName:  database,\n\t\tEncodingType:  \"thriftrw\",\n\t\tDecodingTypes: []string{\"thriftrw\"},\n\t})\n}\n\nfunc createTestSQLFileContent() string {\n\treturn `\n-- test sql file content\n\nCREATE TABLE task_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  first_event_id BIGINT NOT NULL,\n--\n  version BIGINT NOT NULL,\n  next_event_id BIGINT NOT NULL,\n  history MEDIUMBLOB,\n  history_encoding VARCHAR(16) NOT NULL,\n  new_run_history BLOB,\n  new_run_history_encoding VARCHAR(16) NOT NULL DEFAULT 'json',\n  event_store_version          INT NOT NULL, -- indiciates which version of event store to query\n  new_run_event_store_version  INT NOT NULL, -- indiciates which version of event store to query for new run(continueAsNew)\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\n\nCREATE TABLE tasks (\n  domain_id BINARY(16) NOT NULL,\n  task_list_name VARCHAR(255) NOT NULL,\n  task_type TINYINT NOT NULL, -- {Activity, Decision}\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n`\n}\n"
  },
  {
    "path": "host/cli/sql/handlerTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"net\"\n\t\"strconv\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\ntype (\n\t// HandlerTestSuite defines a test suite\n\tHandlerTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t\tpluginName string\n\t}\n)\n\n// NewHandlerTestSuite returns a test suite\nfunc NewHandlerTestSuite(pluginName string) *HandlerTestSuite {\n\treturn &HandlerTestSuite{\n\t\tpluginName: pluginName,\n\t}\n}\n\n// SetupTest setups test\nfunc (s *HandlerTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\n// TestValidateConnectConfig test\nfunc (s *HandlerTestSuite) TestValidateConnectConfig() {\n\tcfg := new(config.SQL)\n\n\ts.NotNil(sql.ValidateConnectConfig(cfg))\n\tport, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tcfg.ConnectAddr = net.JoinHostPort(\n\t\tenvironment.GetMySQLAddress(),\n\t\tstrconv.Itoa(port),\n\t)\n\ts.NotNil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.DatabaseName = \"foobar\"\n\ts.Nil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.TLS = &config.TLS{}\n\tcfg.TLS.Enabled = true\n\ts.NotNil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.TLS.CaFile = \"ca.pem\"\n\ts.Nil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.TLS.KeyFile = \"key_file\"\n\tcfg.TLS.CertFile = \"\"\n\ts.NotNil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.TLS.KeyFile = \"\"\n\tcfg.TLS.CertFile = \"cert_file\"\n\ts.NotNil(sql.ValidateConnectConfig(cfg))\n\n\tcfg.TLS.KeyFile = \"key_file\"\n\tcfg.TLS.CertFile = \"cert_file\"\n\ts.Nil(sql.ValidateConnectConfig(cfg))\n}\n"
  },
  {
    "path": "host/cli/sql/setuptaskTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"log\"\n\t\"os\"\n\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\ntype (\n\t// SetupSchemaTestSuite defines a test suite\n\tSetupSchemaTestSuite struct {\n\t\ttest.SetupSchemaTestBase\n\t\tconn       *sql.Connection\n\t\tpluginName string\n\t}\n)\n\n// NewSetupSchemaTestSuite returns a test suite\nfunc NewSetupSchemaTestSuite(pluginName string) *SetupSchemaTestSuite {\n\treturn &SetupSchemaTestSuite{\n\t\tpluginName: pluginName,\n\t}\n}\n\n// SetupSuite setup test suite\nfunc (s *SetupSchemaTestSuite) SetupSuite() {\n\tos.Setenv(\"SQL_HOST\", environment.GetMySQLAddress())\n\tos.Setenv(\"SQL_USER\", environment.GetMySQLUser())\n\tos.Setenv(\"SQL_PASSWORD\", environment.GetMySQLPassword())\n\tconn, err := newTestConn(\"\", s.pluginName)\n\tif err != nil {\n\t\tlog.Fatalf(\"error creating sql connection:%v\", err)\n\t}\n\ts.conn = conn\n\ts.SetupSuiteBase(conn)\n}\n\n// TearDownSuite tear down test suite\nfunc (s *SetupSchemaTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\n// TestCreateDatabase test\nfunc (s *SetupSchemaTestSuite) TestCreateDatabase() {\n\ts.NoError(sql.RunTool([]string{\"./tool\", \"-u\", environment.GetMySQLUser(), \"--pw\", environment.GetMySQLPassword(), \"create\", \"--db\", \"foobar123\"}))\n\terr := s.conn.DropDatabase(\"foobar123\")\n\ts.Nil(err)\n}\n\n// TestSetupSchema test\nfunc (s *SetupSchemaTestSuite) TestSetupSchema() {\n\tconn, err := newTestConn(s.DBName, s.pluginName)\n\ts.Nil(err)\n\ts.RunSetupTest(sql.BuildCLIOptions(), conn, \"--db\", createTestSQLFileContent(), []string{\"task_maps\", \"tasks\"})\n}\n"
  },
  {
    "path": "host/cli/sql/updatetaskTest.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"log\"\n\t\"os\"\n\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/schema/mysql\"\n\t\"github.com/uber/cadence/tools/common/schema/test\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\n// UpdateSchemaTestSuite defines a test suite\ntype UpdateSchemaTestSuite struct {\n\ttest.UpdateSchemaTestBase\n\tpluginName string\n}\n\n// NewUpdateSchemaTestSuite returns a test suite\nfunc NewUpdateSchemaTestSuite(pluginName string) *UpdateSchemaTestSuite {\n\treturn &UpdateSchemaTestSuite{\n\t\tpluginName: pluginName,\n\t}\n}\n\n// SetupSuite setups test suite\nfunc (s *UpdateSchemaTestSuite) SetupSuite() {\n\tos.Setenv(\"SQL_HOST\", environment.GetMySQLAddress())\n\tos.Setenv(\"SQL_USER\", environment.GetMySQLUser())\n\tos.Setenv(\"SQL_PASSWORD\", environment.GetMySQLPassword())\n\tconn, err := newTestConn(\"\", s.pluginName)\n\tif err != nil {\n\t\tlog.Fatal(\"Error creating CQLClient\")\n\t}\n\ts.SetupSuiteBase(conn)\n}\n\n// TearDownSuite tear down test suite\nfunc (s *UpdateSchemaTestSuite) TearDownSuite() {\n\ts.TearDownSuiteBase()\n}\n\n// TestUpdateSchema test\nfunc (s *UpdateSchemaTestSuite) TestUpdateSchema() {\n\tconn, err := newTestConn(s.DBName, s.pluginName)\n\ts.Nil(err)\n\tdefer conn.Close()\n\ts.RunUpdateSchemaTest(sql.BuildCLIOptions(), conn, \"--db\", createTestSQLFileContent(), []string{\"task_maps\", \"tasks\"})\n}\n\n// TestDryrun test\nfunc (s *UpdateSchemaTestSuite) TestDryrun() {\n\tconn, err := newTestConn(s.DBName, s.pluginName)\n\ts.Nil(err)\n\tdefer conn.Close()\n\tdir := \"../../../schema/mysql/v8/cadence/versioned\"\n\ts.RunDryrunTest(sql.BuildCLIOptions(), conn, \"--db\", dir, mysql.Version)\n}\n\n// TestVisibilityDryrun test\nfunc (s *UpdateSchemaTestSuite) TestVisibilityDryrun() {\n\tconn, err := newTestConn(s.DBName, s.pluginName)\n\ts.Nil(err)\n\tdefer conn.Close()\n\tdir := \"../../../schema/mysql/v8/visibility/versioned\"\n\ts.RunDryrunTest(sql.BuildCLIOptions(), conn, \"--db\", dir, mysql.VisibilityVersion)\n}\n"
  },
  {
    "path": "host/cli/sql/utils.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sql\n\n// NOTE: change this when moving the test files around during refactoring\n// Path to root folder of cadence repo\nconst rootRelativePath = \"../../../\"\n"
  },
  {
    "path": "host/cli/sql/versionTest.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"math/rand\"\n\t\"os\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\ntype (\n\t// VersionTestSuite defines a test suite\n\tVersionTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t\tpluginName string\n\t}\n)\n\n// NewVersionTestSuite returns a test suite\nfunc NewVersionTestSuite(pluginName string) *VersionTestSuite {\n\treturn &VersionTestSuite{\n\t\tpluginName: pluginName,\n\t}\n}\n\n// SetupTest setups test suite\nfunc (s *VersionTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\n// TestVerifyCompatibleVersion test\nfunc (s *VersionTestSuite) TestVerifyCompatibleVersion() {\n\tdatabase := \"cadence_test\"\n\tvisDatabase := \"cadence_visibility_test\"\n\tsqlFile := rootRelativePath + \"schema/mysql/v8/cadence/schema.sql\"\n\tvisSQLFile := rootRelativePath + \"schema/mysql/v8/visibility/schema.sql\"\n\n\tdefer s.createDatabase(database)()\n\tdefer s.createDatabase(visDatabase)()\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\terr = sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-f\", sqlFile,\n\t\t\"-version\", \"10.0\",\n\t\t\"-o\",\n\t})\n\ts.NoError(err)\n\terr = sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", visDatabase,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-f\", visSQLFile,\n\t\t\"-version\", \"10.0\",\n\t\t\"-o\",\n\t})\n\ts.NoError(err)\n\n\tdefaultCfg := config.SQL{\n\t\tConnectAddr:   fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\tUser:          environment.GetMySQLUser(),\n\t\tPassword:      environment.GetMySQLPassword(),\n\t\tPluginName:    s.pluginName,\n\t\tDatabaseName:  database,\n\t\tEncodingType:  \"thriftrw\",\n\t\tDecodingTypes: []string{\"thriftrw\"},\n\t}\n\tvisibilityCfg := defaultCfg\n\tvisibilityCfg.DatabaseName = visDatabase\n\tcfg := config.Persistence{\n\t\tDefaultStore:    \"default\",\n\t\tVisibilityStore: \"visibility\",\n\t\tDataStores: map[string]config.DataStore{\n\t\t\t\"default\":    {SQL: &defaultCfg},\n\t\t\t\"visibility\": {SQL: &visibilityCfg},\n\t\t},\n\t\tTransactionSizeLimit: dynamicproperties.GetIntPropertyFn(constants.DefaultTransactionSizeLimit),\n\t\tErrorInjectionRate:   dynamicproperties.GetFloatPropertyFn(0),\n\t}\n\ts.NoError(sql.VerifyCompatibleVersion(cfg))\n}\n\n// TestCheckCompatibleVersion test\nfunc (s *VersionTestSuite) TestCheckCompatibleVersion() {\n\tflags := []struct {\n\t\texpectedVersion string\n\t\tactualVersion   string\n\t\terrStr          string\n\t\texpectedFail    bool\n\t}{\n\t\t{\"2.0\", \"1.0\", \"version mismatch\", false},\n\t\t{\"1.0\", \"1.0\", \"\", false},\n\t\t{\"1.0\", \"2.0\", \"\", false},\n\t\t{\"1.0\", \"abc\", \"schema_version' doesn't exist\", true},\n\t}\n\tfor _, flag := range flags {\n\t\ts.runCheckCompatibleVersion(flag.expectedVersion, flag.actualVersion, flag.errStr, flag.expectedFail)\n\t}\n}\n\nfunc (s *VersionTestSuite) createDatabase(database string) func() {\n\tconnection, err := newTestConn(\"\", s.pluginName)\n\ts.NoError(err)\n\terr = connection.CreateDatabase(database)\n\ts.NoError(err)\n\treturn func() {\n\t\ts.NoError(connection.DropDatabase(database))\n\t\tconnection.Close()\n\t}\n}\n\nfunc (s *VersionTestSuite) runCheckCompatibleVersion(\n\texpected string, actual string, errStr string, expectedFail bool,\n) {\n\tr := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tdatabase := fmt.Sprintf(\"version_test_%v\", r.Int63())\n\tdefer s.createDatabase(database)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + database\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, actual)\n\tif expected != actual {\n\t\ts.createSchemaForVersion(subdir, expected)\n\t}\n\n\tsqlFile := subdir + \"/v\" + actual + \"/tmp.sql\"\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\n\terr = sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database,\n\t\t\"-pl\", s.pluginName,\n\t\t\"setup-schema\",\n\t\t\"-f\", sqlFile,\n\t\t\"-version\", actual,\n\t\t\"-o\",\n\t})\n\tif expectedFail {\n\t\ts.Error(err)\n\t} else {\n\t\ts.NoError(err)\n\t}\n\n\tcfg := config.SQL{\n\t\tConnectAddr:   fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\tUser:          environment.GetMySQLUser(),\n\t\tPassword:      environment.GetMySQLPassword(),\n\t\tPluginName:    s.pluginName,\n\t\tDatabaseName:  database,\n\t\tEncodingType:  \"thriftrw\",\n\t\tDecodingTypes: []string{\"thriftrw\"},\n\t}\n\terr = sql.CheckCompatibleVersion(cfg, expected)\n\tif len(errStr) > 0 {\n\t\ts.Error(err)\n\t\ts.Contains(err.Error(), errStr)\n\t} else {\n\t\ts.NoError(err)\n\t}\n}\n\nfunc (s *VersionTestSuite) createSchemaForVersion(subdir string, v string) {\n\tvDir := subdir + \"/v\" + v\n\ts.NoError(os.Mkdir(vDir, os.FileMode(0744)))\n\tcqlFile := vDir + \"/tmp.sql\"\n\ts.NoError(ioutil.WriteFile(cqlFile, []byte{}, os.FileMode(0644)))\n}\n\nfunc (s *VersionTestSuite) TestMultipleDatabaseVersionInCompatible() {\n\tr1 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tr2 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tdatabase1 := fmt.Sprintf(\"version_test_%v\", r1.Int63())\n\tdatabase2 := fmt.Sprintf(\"version_test_%v\", r2.Int63())\n\tdefer s.createDatabase(database1)()\n\tdefer s.createDatabase(database2)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + \"db\"\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, \"0.1\")\n\ts.createSchemaForVersion(subdir, \"0.2\")\n\ts.createSchemaForVersion(subdir, \"0.3\")\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database1,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.2\",\n\t\t\"-o\",\n\t}))\n\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database2,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.3\",\n\t\t\"-o\",\n\t}))\n\n\tcfg := config.SQL{\n\t\tPluginName:           s.pluginName,\n\t\tEncodingType:         \"thriftrw\",\n\t\tDecodingTypes:        []string{\"thriftrw\"},\n\t\tUseMultipleDatabases: true,\n\t\tNumShards:            2,\n\t\tMultipleDatabasesConfig: []config.MultipleDatabasesConfigEntry{\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database2,\n\t\t\t},\n\t\t},\n\t}\n\terr = sql.CheckCompatibleVersion(cfg, \"0.3\")\n\ts.Error(err)\n\ts.Contains(err.Error(), \"version mismatch\")\n}\n\nfunc (s *VersionTestSuite) TestMultipleDatabaseVersionAllLowerCompatible() {\n\tr1 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tr2 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tdatabase1 := fmt.Sprintf(\"version_test_%v\", r1.Int63())\n\tdatabase2 := fmt.Sprintf(\"version_test_%v\", r2.Int63())\n\tdefer s.createDatabase(database1)()\n\tdefer s.createDatabase(database2)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + \"db\"\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, \"0.1\")\n\ts.createSchemaForVersion(subdir, \"0.2\")\n\ts.createSchemaForVersion(subdir, \"0.3\")\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database1,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.2\",\n\t\t\"-o\",\n\t}))\n\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database2,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.2\",\n\t\t\"-o\",\n\t}))\n\n\tcfg := config.SQL{\n\t\tPluginName:           s.pluginName,\n\t\tEncodingType:         \"thriftrw\",\n\t\tDecodingTypes:        []string{\"thriftrw\"},\n\t\tUseMultipleDatabases: true,\n\t\tNumShards:            2,\n\t\tMultipleDatabasesConfig: []config.MultipleDatabasesConfigEntry{\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database2,\n\t\t\t},\n\t\t},\n\t}\n\terr = sql.CheckCompatibleVersion(cfg, \"0.2\")\n\ts.NoError(err)\n}\n\nfunc (s *VersionTestSuite) TestMultipleDatabaseVersionPartialLowerCompatible() {\n\tr1 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tr2 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tdatabase1 := fmt.Sprintf(\"version_test_%v\", r1.Int63())\n\tdatabase2 := fmt.Sprintf(\"version_test_%v\", r2.Int63())\n\tdefer s.createDatabase(database1)()\n\tdefer s.createDatabase(database2)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + \"db\"\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, \"0.1\")\n\ts.createSchemaForVersion(subdir, \"0.2\")\n\ts.createSchemaForVersion(subdir, \"0.3\")\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database1,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.3\",\n\t\t\"-o\",\n\t}))\n\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database2,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.2\",\n\t\t\"-o\",\n\t}))\n\n\tcfg := config.SQL{\n\t\tPluginName:           s.pluginName,\n\t\tEncodingType:         \"thriftrw\",\n\t\tDecodingTypes:        []string{\"thriftrw\"},\n\t\tUseMultipleDatabases: true,\n\t\tNumShards:            2,\n\t\tMultipleDatabasesConfig: []config.MultipleDatabasesConfigEntry{\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database2,\n\t\t\t},\n\t\t},\n\t}\n\terr = sql.CheckCompatibleVersion(cfg, \"0.2\")\n\ts.NoError(err)\n}\n\nfunc (s *VersionTestSuite) TestMultipleDatabaseVersionExactlyMatchCompatible() {\n\tr1 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tr2 := rand.New(rand.NewSource(time.Now().UnixNano()))\n\tdatabase1 := fmt.Sprintf(\"version_test_%v\", r1.Int63())\n\tdatabase2 := fmt.Sprintf(\"version_test_%v\", r2.Int63())\n\tdefer s.createDatabase(database1)()\n\tdefer s.createDatabase(database2)()\n\n\ttmpDir := s.T().TempDir()\n\n\tsubdir := tmpDir + \"/\" + \"db\"\n\ts.NoError(os.Mkdir(subdir, os.FileMode(0744)))\n\n\ts.createSchemaForVersion(subdir, \"0.1\")\n\ts.createSchemaForVersion(subdir, \"0.2\")\n\ts.createSchemaForVersion(subdir, \"0.3\")\n\tmysqlPort, err := environment.GetMySQLPort()\n\ts.NoError(err)\n\tport := strconv.Itoa(mysqlPort)\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database1,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.3\",\n\t\t\"-o\",\n\t}))\n\n\ts.NoError(sql.RunTool([]string{\n\t\t\"./tool\",\n\t\t\"-ep\", environment.GetMySQLAddress(),\n\t\t\"-p\", port,\n\t\t\"-u\", environment.GetMySQLUser(),\n\t\t\"-pw\", environment.GetMySQLPassword(),\n\t\t\"-db\", database2,\n\t\t\"-pl\", s.pluginName,\n\t\t\"-q\",\n\t\t\"setup-schema\",\n\t\t\"-version\", \"0.3\",\n\t\t\"-o\",\n\t}))\n\n\tcfg := config.SQL{\n\t\tPluginName:           s.pluginName,\n\t\tEncodingType:         \"thriftrw\",\n\t\tDecodingTypes:        []string{\"thriftrw\"},\n\t\tUseMultipleDatabases: true,\n\t\tNumShards:            2,\n\t\tMultipleDatabasesConfig: []config.MultipleDatabasesConfigEntry{\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tConnectAddr:  fmt.Sprintf(\"%v:%v\", environment.GetMySQLAddress(), port),\n\t\t\t\tUser:         environment.GetMySQLUser(),\n\t\t\t\tPassword:     environment.GetMySQLPassword(),\n\t\t\t\tDatabaseName: database2,\n\t\t\t},\n\t\t},\n\t}\n\terr = sql.CheckCompatibleVersion(cfg, \"0.3\")\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "host/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// AdminClient is the interface exposed by admin service client\ntype AdminClient interface {\n\tadmin.Client\n}\n\n// FrontendClient is the interface exposed by frontend service client\ntype FrontendClient interface {\n\tfrontend.Client\n}\n\n// HistoryClient is the interface exposed by history service client\ntype HistoryClient interface {\n\thistory.Client\n}\n\ntype MatchingClient interface {\n\tmatching.Client\n}\n\n// NewAdminClient creates a client to cadence admin client\nfunc NewAdminClient(d *yarpc.Dispatcher) AdminClient {\n\treturn grpc.NewAdminClient(adminv1.NewAdminAPIYARPCClient(d.ClientConfig(testOutboundName(service.Frontend))))\n}\n\n// NewFrontendClient creates a client to cadence frontend client\nfunc NewFrontendClient(d *yarpc.Dispatcher) FrontendClient {\n\tconfig := d.ClientConfig(testOutboundName(service.Frontend))\n\treturn grpc.NewFrontendClient(\n\t\tapiv1.NewDomainAPIYARPCClient(config),\n\t\tapiv1.NewWorkflowAPIYARPCClient(config),\n\t\tapiv1.NewWorkerAPIYARPCClient(config),\n\t\tapiv1.NewVisibilityAPIYARPCClient(config),\n\t)\n}\n\n// NewHistoryClient creates a client to cadence history service client\nfunc NewHistoryClient(d *yarpc.Dispatcher) HistoryClient {\n\treturn grpc.NewHistoryClient(historyv1.NewHistoryAPIYARPCClient(d.ClientConfig(testOutboundName(service.History))))\n}\n"
  },
  {
    "path": "host/client_integration_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/gob\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"strconv\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/compatibility\"\n\t\"go.uber.org/cadence/encoded\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc init() {\n\tworkflow.Register(testDataConverterWorkflow)\n\tactivity.Register(testActivity)\n\tworkflow.Register(testParentWorkflow)\n\tworkflow.Register(testChildWorkflow)\n}\n\nfunc TestClientIntegrationSuite(t *testing.T) {\n\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/clientintegrationtestcluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(ClientIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *ClientIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n\n\tvar err error\n\ts.wfService, err = s.buildServiceClient()\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Error when build service client\", tag.Error(err))\n\t}\n\ts.wfClient = client.NewClient(s.wfService, s.DomainName, nil)\n\n\ts.taskList = \"client-integration-test-tasklist\"\n\ts.worker = worker.New(s.wfService, s.DomainName, s.taskList, worker.Options{})\n\tif err := s.worker.Start(); err != nil {\n\t\ts.Logger.Fatal(\"Error when start worker\", tag.Error(err))\n\t} else {\n\t\ts.Logger.Info(\"Worker started\")\n\t}\n}\n\nfunc (s *ClientIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *ClientIntegrationSuite) buildServiceClient() (workflowserviceclient.Interface, error) {\n\tcadenceClientName := \"cadence-client\"\n\thostPort := \"127.0.0.1:7114\" // use grpc port\n\tif TestFlags.FrontendAddr != \"\" {\n\t\thostPort = TestFlags.FrontendAddr\n\t}\n\tch := grpc.NewTransport(\n\t\tgrpc.ServerMaxRecvMsgSize(32*1024*1024),\n\t\tgrpc.ClientMaxRecvMsgSize(32*1024*1024),\n\t)\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: cadenceClientName,\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\tservice.Frontend: {Unary: ch.NewSingleOutbound(hostPort)},\n\t\t},\n\t})\n\tif dispatcher == nil {\n\t\ts.Logger.Fatal(\"No RPC dispatcher provided to create a connection to Cadence Service\")\n\t}\n\tif err := dispatcher.Start(); err != nil {\n\t\ts.Logger.Fatal(\"Failed to create outbound transport channel\", tag.Error(err))\n\t}\n\tcc := dispatcher.ClientConfig(service.Frontend)\n\treturn compatibility.NewThrift2ProtoAdapter(\n\t\tapiv1.NewDomainAPIYARPCClient(cc),\n\t\tapiv1.NewWorkflowAPIYARPCClient(cc),\n\t\tapiv1.NewWorkerAPIYARPCClient(cc),\n\t\tapiv1.NewVisibilityAPIYARPCClient(cc),\n\t), nil\n}\n\nfunc (s *ClientIntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\n// testDataConverter implements encoded.DataConverter using gob\ntype testDataConverter struct {\n\tNumOfCallToData   int // for testing to know testDataConverter is called as expected\n\tNumOfCallFromData int\n}\n\nfunc (tdc *testDataConverter) ToData(value ...interface{}) ([]byte, error) {\n\ttdc.NumOfCallToData++\n\tvar buf bytes.Buffer\n\tenc := gob.NewEncoder(&buf)\n\tfor i, obj := range value {\n\t\tif err := enc.Encode(obj); err != nil {\n\t\t\treturn nil, fmt.Errorf(\n\t\t\t\t\"unable to encode argument: %d, %v, with gob error: %v\", i, reflect.TypeOf(obj), err)\n\t\t}\n\t}\n\treturn buf.Bytes(), nil\n}\n\nfunc (tdc *testDataConverter) FromData(input []byte, valuePtr ...interface{}) error {\n\ttdc.NumOfCallFromData++\n\tdec := gob.NewDecoder(bytes.NewBuffer(input))\n\tfor i, obj := range valuePtr {\n\t\tif err := dec.Decode(obj); err != nil {\n\t\t\treturn fmt.Errorf(\n\t\t\t\t\"unable to decode argument: %d, %v, with gob error: %v\", i, reflect.TypeOf(obj), err)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc newTestDataConverter() encoded.DataConverter {\n\treturn &testDataConverter{}\n}\n\nfunc testActivity(ctx context.Context, msg string) (string, error) {\n\treturn \"hello_\" + msg, nil\n}\n\nfunc testDataConverterWorkflow(ctx workflow.Context, tl string) (string, error) {\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 20 * time.Second,\n\t\tStartToCloseTimeout:    40 * time.Second,\n\t}\n\tctx = workflow.WithActivityOptions(ctx, ao)\n\n\tvar result string\n\terr := workflow.ExecuteActivity(ctx, testActivity, \"world\").Get(ctx, &result)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\t// use another converter to run activity,\n\t// with new taskList so that worker with same data converter can properly process tasks.\n\tvar result1 string\n\tctx1 := workflow.WithDataConverter(ctx, newTestDataConverter())\n\tctx1 = workflow.WithTaskList(ctx1, tl)\n\terr1 := workflow.ExecuteActivity(ctx1, testActivity, \"world1\").Get(ctx1, &result1)\n\tif err1 != nil {\n\t\treturn \"\", err1\n\t}\n\treturn result + \",\" + result1, nil\n}\n\nfunc (s *ClientIntegrationSuite) startWorkerWithDataConverter(tl string, dataConverter encoded.DataConverter) worker.Worker {\n\topts := worker.Options{}\n\tif dataConverter != nil {\n\t\topts.DataConverter = dataConverter\n\t}\n\tworker := worker.New(s.wfService, s.DomainName, tl, opts)\n\tif err := worker.Start(); err != nil {\n\t\ts.Logger.Fatal(\"Error when start worker with data converter\", tag.Error(err))\n\t}\n\treturn worker\n}\n\nfunc (s *ClientIntegrationSuite) TestClientDataConverter() {\n\ttl := \"client-integration-data-converter-activity-tasklist\"\n\tdc := newTestDataConverter()\n\tworker := s.startWorkerWithDataConverter(tl, dc)\n\tdefer worker.Stop()\n\n\tid := \"client-integration-data-converter-workflow\"\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tID:                           id,\n\t\tTaskList:                     s.taskList,\n\t\tExecutionStartToCloseTimeout: 60 * time.Second,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)\n\tdefer cancel()\n\twe, err := s.wfClient.ExecuteWorkflow(ctx, workflowOptions, testDataConverterWorkflow, tl)\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Start workflow with err\", tag.Error(err))\n\t}\n\ts.NotNil(we)\n\ts.True(we.GetRunID() != \"\")\n\n\tvar res string\n\terr = we.Get(ctx, &res)\n\ts.NoError(err)\n\ts.Equal(\"hello_world,hello_world1\", res)\n\n\t// to ensure custom data converter is used, this number might be different if client changed.\n\td := dc.(*testDataConverter)\n\ts.Equal(1, d.NumOfCallToData)\n\ts.Equal(1, d.NumOfCallFromData)\n}\n\nfunc (s *ClientIntegrationSuite) TestClientDataConverter_Failed() {\n\ttl := \"client-integration-data-converter-activity-failed-tasklist\"\n\tworker := s.startWorkerWithDataConverter(tl, nil) // mismatch of data converter\n\tdefer worker.Stop()\n\n\tid := \"client-integration-data-converter-failed-workflow\"\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tID:                           id,\n\t\tTaskList:                     s.taskList,\n\t\tExecutionStartToCloseTimeout: 60 * time.Second,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)\n\tdefer cancel()\n\twe, err := s.wfClient.ExecuteWorkflow(ctx, workflowOptions, testDataConverterWorkflow, tl)\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Start workflow with err\", tag.Error(err))\n\t}\n\ts.NotNil(we)\n\ts.True(we.GetRunID() != \"\")\n\n\tvar res string\n\terr = we.Get(ctx, &res)\n\ts.Error(err)\n\n\t// Get history to make sure only the 2nd activity is failed because of mismatch of data converter\n\titer := s.wfClient.GetWorkflowHistory(ctx, id, we.GetRunID(), false, 0)\n\tcompletedAct := 0\n\tfailedAct := 0\n\tfor iter.HasNext() {\n\t\tevent, err := iter.Next()\n\t\ts.Nil(err)\n\t\tif event.GetEventType() == shared.EventTypeActivityTaskCompleted {\n\t\t\tcompletedAct++\n\t\t}\n\t\tif event.GetEventType() == shared.EventTypeActivityTaskFailed {\n\t\t\tfailedAct++\n\t\t\tattr := event.ActivityTaskFailedEventAttributes\n\t\t\ts.True(strings.HasPrefix(string(attr.Details), \"unable to decode the activity function input bytes with error\"))\n\t\t}\n\t}\n\ts.Equal(1, completedAct)\n\ts.Equal(1, failedAct)\n}\n\nvar childTaskList = \"client-integration-data-converter-child-tasklist\"\n\nfunc testParentWorkflow(ctx workflow.Context) (string, error) {\n\tlogger := workflow.GetLogger(ctx)\n\texecution := workflow.GetInfo(ctx).WorkflowExecution\n\tchildID := fmt.Sprintf(\"child_workflow:%v\", execution.RunID)\n\tcwo := workflow.ChildWorkflowOptions{\n\t\tWorkflowID:                   childID,\n\t\tExecutionStartToCloseTimeout: time.Minute,\n\t}\n\tctx = workflow.WithChildOptions(ctx, cwo)\n\tvar result string\n\terr := workflow.ExecuteChildWorkflow(ctx, testChildWorkflow, 0, 3).Get(ctx, &result)\n\tif err != nil {\n\t\tlogger.Error(\"Parent execution received child execution failure.\", zap.Error(err))\n\t\treturn \"\", err\n\t}\n\n\tchildID1 := fmt.Sprintf(\"child_workflow1:%v\", execution.RunID)\n\tcwo1 := workflow.ChildWorkflowOptions{\n\t\tWorkflowID:                   childID1,\n\t\tExecutionStartToCloseTimeout: time.Minute,\n\t\tTaskList:                     childTaskList,\n\t}\n\tctx1 := workflow.WithChildOptions(ctx, cwo1)\n\tctx1 = workflow.WithDataConverter(ctx1, newTestDataConverter())\n\tvar result1 string\n\terr1 := workflow.ExecuteChildWorkflow(ctx1, testChildWorkflow, 0, 2).Get(ctx1, &result1)\n\tif err1 != nil {\n\t\tlogger.Error(\"Parent execution received child execution 1 failure.\", zap.Error(err1))\n\t\treturn \"\", err1\n\t}\n\n\tres := fmt.Sprintf(\"Complete child1 %s times, complete child2 %s times\", result, result1)\n\tlogger.Info(\"Parent execution completed.\", zap.String(\"Result\", res))\n\treturn res, nil\n}\n\nfunc testChildWorkflow(ctx workflow.Context, totalCount, runCount int) (string, error) {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"Child workflow execution started.\")\n\tif runCount <= 0 {\n\t\tlogger.Error(\"Invalid valid for run count.\", zap.Int(\"RunCount\", runCount))\n\t\treturn \"\", errors.New(\"invalid run count\")\n\t}\n\n\ttotalCount++\n\trunCount--\n\tif runCount == 0 {\n\t\tresult := fmt.Sprintf(\"Child workflow execution completed after %v runs\", totalCount)\n\t\tlogger.Info(\"Child workflow completed.\", zap.String(\"Result\", result))\n\t\treturn strconv.Itoa(totalCount), nil\n\t}\n\n\tlogger.Info(\"Child workflow starting new run.\", zap.Int(\"RunCount\", runCount), zap.Int(\"TotalCount\",\n\t\ttotalCount))\n\treturn \"\", workflow.NewContinueAsNewError(ctx, testChildWorkflow, totalCount, runCount)\n}\n\nfunc (s *ClientIntegrationSuite) TestClientDataConverter_WithChild() {\n\tdc := newTestDataConverter()\n\tworker := s.startWorkerWithDataConverter(childTaskList, dc)\n\tdefer worker.Stop()\n\n\tid := \"client-integration-data-converter-with-child-workflow\"\n\tworkflowOptions := client.StartWorkflowOptions{\n\t\tID:                           id,\n\t\tTaskList:                     s.taskList,\n\t\tExecutionStartToCloseTimeout: 60 * time.Second,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)\n\tdefer cancel()\n\twe, err := s.wfClient.ExecuteWorkflow(ctx, workflowOptions, testParentWorkflow)\n\tif err != nil {\n\t\ts.Logger.Fatal(\"Start workflow with err\", tag.Error(err))\n\t}\n\ts.NotNil(we)\n\ts.True(we.GetRunID() != \"\")\n\n\tvar res string\n\terr = we.Get(ctx, &res)\n\ts.NoError(err)\n\ts.Equal(\"Complete child1 3 times, complete child2 2 times\", res)\n\n\t// to ensure custom data converter is used, this number might be different if client changed.\n\td := dc.(*testDataConverter)\n\ts.Equal(3, d.NumOfCallToData)\n\ts.Equal(2, d.NumOfCallFromData)\n}\n\nfunc (s *ClientIntegrationSuite) Test_StickyWorkerRestartDecisionTask() {\n\ttestCases := []struct {\n\t\tname       string\n\t\twaitTime   time.Duration\n\t\tdoQuery    bool\n\t\tdoSignal   bool\n\t\tdelayCheck func(duration time.Duration) bool\n\t}{\n\t\t{\n\t\t\tname:     \"new_query_after_10s_no_delay\",\n\t\t\twaitTime: 10 * time.Second,\n\t\t\tdoQuery:  true,\n\t\t\tdelayCheck: func(duration time.Duration) bool {\n\t\t\t\treturn duration < 5*time.Second\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"new_query_immediately_expect_5s_delay\",\n\t\t\twaitTime: 0,\n\t\t\tdoQuery:  true,\n\t\t\tdelayCheck: func(duration time.Duration) bool {\n\t\t\t\treturn duration > 5*time.Second\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"new_workflow_task_after_10s_no_delay\",\n\t\t\twaitTime: 10 * time.Second,\n\t\t\tdoSignal: true,\n\t\t\tdelayCheck: func(duration time.Duration) bool {\n\t\t\t\treturn duration < 5*time.Second\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"new_workflow_task_immediately_expect_5s_delay\",\n\t\t\twaitTime: 0,\n\t\t\tdoSignal: true,\n\t\t\tdelayCheck: func(duration time.Duration) bool {\n\t\t\t\treturn duration > 5*time.Second\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\tworkflowFn := func(ctx workflow.Context) (string, error) {\n\t\t\t\tworkflow.SetQueryHandler(ctx, \"test\", func() (string, error) {\n\t\t\t\t\treturn \"query works\", nil\n\t\t\t\t})\n\n\t\t\t\tsignalCh := workflow.GetSignalChannel(ctx, \"test\")\n\t\t\t\tvar msg string\n\t\t\t\tsignalCh.Receive(ctx, &msg)\n\t\t\t\treturn msg, nil\n\t\t\t}\n\n\t\t\ttaskList := \"task-list-\" + tt.name\n\n\t\t\toldWorker := worker.New(s.wfService, s.DomainName, taskList, worker.Options{})\n\t\t\toldWorker.RegisterWorkflow(workflowFn)\n\t\t\tif err := oldWorker.Start(); err != nil {\n\t\t\t\ts.Logger.Fatal(\"Error when start worker\", tag.Error(err))\n\t\t\t}\n\n\t\t\tid := \"test-sticky-delay\" + tt.name\n\t\t\tworkflowOptions := client.StartWorkflowOptions{\n\t\t\t\tID:                           id,\n\t\t\t\tTaskList:                     taskList,\n\t\t\t\tExecutionStartToCloseTimeout: 20 * time.Second,\n\t\t\t}\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), 1000*time.Second)\n\t\t\tdefer cancel()\n\t\t\tworkflowRun, err := s.wfClient.ExecuteWorkflow(ctx, workflowOptions, workflowFn)\n\t\t\tif err != nil {\n\t\t\t\ts.Logger.Fatal(\"Start workflow failed with err\", tag.Error(err))\n\t\t\t}\n\n\t\t\ts.NotNil(workflowRun)\n\t\t\ts.True(workflowRun.GetRunID() != \"\")\n\n\t\t\tfindFirstWorkflowTaskCompleted := false\n\t\tWaitForFirstWorkflowTaskComplete:\n\t\t\tfor i := 0; i < 10; i++ {\n\t\t\t\t// wait until first workflow task completed (so we know sticky is set on workflow)\n\t\t\t\titer := s.wfClient.GetWorkflowHistory(ctx, id, \"\", false, 0)\n\t\t\t\tfor iter.HasNext() {\n\t\t\t\t\tevt, err := iter.Next()\n\t\t\t\t\ts.NoError(err)\n\t\t\t\t\tif evt.GetEventType() == shared.EventTypeDecisionTaskCompleted {\n\t\t\t\t\t\tfindFirstWorkflowTaskCompleted = true\n\t\t\t\t\t\tbreak WaitForFirstWorkflowTaskComplete\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\ttime.Sleep(time.Second)\n\t\t\t}\n\t\t\ts.True(findFirstWorkflowTaskCompleted)\n\n\t\t\t// stop old worker\n\t\t\toldWorker.Stop()\n\n\t\t\t// maybe wait for 10s, which will make matching aware the old sticky worker is unavailable\n\t\t\ttime.Sleep(tt.waitTime)\n\n\t\t\t// start a new worker\n\t\t\tnewWorker := worker.New(s.wfService, s.DomainName, taskList, worker.Options{})\n\t\t\tnewWorker.RegisterWorkflow(workflowFn)\n\t\t\tif err := newWorker.Start(); err != nil {\n\t\t\t\ts.Logger.Fatal(\"Error when start worker\", tag.Error(err))\n\t\t\t}\n\t\t\tdefer newWorker.Stop()\n\n\t\t\tstartTime := time.Now()\n\t\t\t// send a signal, and workflow should complete immediately, there should not be 5s delay\n\t\t\tif tt.doSignal {\n\t\t\t\terr = s.wfClient.SignalWorkflow(ctx, id, \"\", \"test\", \"test\")\n\t\t\t\ts.NoError(err)\n\n\t\t\t\terr = workflowRun.Get(ctx, nil)\n\t\t\t\ts.NoError(err)\n\t\t\t} else if tt.doQuery {\n\t\t\t\t// send a signal, and workflow should complete immediately, there should not be 5s delay\n\t\t\t\tqueryResult, err := s.wfClient.QueryWorkflow(ctx, id, \"\", \"test\", \"test\")\n\t\t\t\ts.NoError(err)\n\n\t\t\t\tvar queryResultStr string\n\t\t\t\terr = queryResult.Get(&queryResultStr)\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(\"query works\", queryResultStr)\n\t\t\t}\n\t\t\tendTime := time.Now()\n\t\t\tduration := endTime.Sub(startTime)\n\t\t\ts.True(tt.delayCheck(duration), \"delay check failed: %s\", duration)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "host/continue_as_new_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestContinueAsNewWorkflow() {\n\tid := \"integration-continue-as-new-workflow-test\"\n\twt := \"integration-continue-as-new-workflow-test-type\"\n\ttl := \"integration-continue-as-new-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\theader := &types.Header{\n\t\tFields: map[string][]byte{\"tracing\": []byte(\"sample payload\")},\n\t}\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\"memoKey\": []byte(\"memoVal\")},\n\t}\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\"CustomKeywordField\": []byte(`\"1\"`)},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tHeader:                              header,\n\t\tMemo:                                memo,\n\t\tSearchAttributes:                    searchAttr,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tcontinueAsNewCount := int32(10)\n\tcontinueAsNewCounter := int32(0)\n\tvar previousRunID string\n\tvar lastRunStartedEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif continueAsNewCounter < continueAsNewCount {\n\t\t\tpreviousRunID = execution.GetRunID()\n\t\t\tcontinueAsNewCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, continueAsNewCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(continueAsNewCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                               buf.Bytes(),\n\t\t\t\t\tHeader:                              header,\n\t\t\t\t\tMemo:                                memo,\n\t\t\t\t\tSearchAttributes:                    searchAttr,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tlastRunStartedEvent = history.Events[0]\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(continueAsNewCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.NoError(err)\n\t\ts.NotZero(descResp.WorkflowExecutionInfo.GetStartTime())\n\n\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err, strconv.Itoa(i))\n\n\t}\n\n\ts.False(workflowComplete)\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n\ts.Equal(previousRunID, lastRunStartedEvent.WorkflowExecutionStartedEventAttributes.GetContinuedExecutionRunID())\n\ts.Equal(header, lastRunStartedEvent.WorkflowExecutionStartedEventAttributes.Header)\n\ts.Equal(memo, lastRunStartedEvent.WorkflowExecutionStartedEventAttributes.Memo)\n\ts.Equal(searchAttr, lastRunStartedEvent.WorkflowExecutionStartedEventAttributes.SearchAttributes)\n}\n\nfunc (s *IntegrationSuite) TestContinueAsNewWorkflow_Timeout() {\n\tid := \"integration-continue-as-new-workflow-timeout-test\"\n\twt := \"integration-continue-as-new-workflow-timeout-test-type\"\n\ttl := \"integration-continue-as-new-workflow-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tcontinueAsNewCount := int32(1)\n\tcontinueAsNewCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif continueAsNewCounter < continueAsNewCount {\n\t\t\tcontinueAsNewCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, continueAsNewCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(continueAsNewCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                               buf.Bytes(),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1), // set timeout to 1\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(continueAsNewCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// process the decision and continue as new\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\n\ttime.Sleep(1 * time.Second) // wait 1 second for timeout\n\nGetHistoryLoop:\n\tfor i := 0; i < 20; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionTimedOut {\n\t\t\ts.Logger.Warn(\"Execution not timedout yet.\")\n\t\t\ttime.Sleep(200 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\ttimeoutEventAttributes := lastEvent.WorkflowExecutionTimedOutEventAttributes\n\t\ts.Equal(types.TimeoutTypeStartToClose, *timeoutEventAttributes.TimeoutType)\n\t\tworkflowComplete = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowContinueAsNew_TaskID() {\n\tid := \"integration-wf-continue-as-new-task-id-test\"\n\twt := \"integration-wf-continue-as-new-task-id-type\"\n\ttl := \"integration-wf-continue-as-new-task-id-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tvar executions []*types.WorkflowExecution\n\n\tcontinueAsNewed := false\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\texecutions = append(executions, execution)\n\n\t\tif !continueAsNewed {\n\t\t\tcontinueAsNewed = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\t\tTaskList:                            taskList,\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"succeed\"),\n\t\t\t},\n\t\t}}, nil\n\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tminTaskID := int64(0)\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\tevents := s.getHistory(s.DomainName, executions[0])\n\ts.True(len(events) != 0)\n\tfor _, event := range events {\n\t\ts.True(event.TaskID > minTaskID)\n\t\tminTaskID = event.TaskID\n\t}\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\tevents = s.getHistory(s.DomainName, executions[1])\n\ts.True(len(events) != 0)\n\tfor _, event := range events {\n\t\ts.True(event.TaskID > minTaskID)\n\t\tminTaskID = event.TaskID\n\t}\n}\n\nfunc (s *IntegrationSuite) TestChildWorkflowWithContinueAsNew() {\n\tparentID := \"integration-child-workflow-with-continue-as-new-test-parent\"\n\tchildID := \"integration-child-workflow-with-continue-as-new-test-child\"\n\twtParent := \"integration-child-workflow-with-continue-as-new-test-parent-type\"\n\twtChild := \"integration-child-workflow-with-continue-as-new-test-child-type\"\n\ttl := \"integration-child-workflow-with-continue-as-new-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tparentWorkflowType := &types.WorkflowType{}\n\tparentWorkflowType.Name = wtParent\n\n\tchildWorkflowType := &types.WorkflowType{}\n\tchildWorkflowType.Name = wtChild\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          parentID,\n\t\tWorkflowType:                        parentWorkflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tchildComplete := false\n\tchildExecutionStarted := false\n\tchildData := int32(1)\n\tcontinueAsNewCount := int32(10)\n\tcontinueAsNewCounter := int32(0)\n\tvar startedEvent *types.HistoryEvent\n\tvar completedEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\ts.Logger.Info(\"Processing decision task for WorkflowID:\", tag.WorkflowID(execution.WorkflowID))\n\n\t\t// Child Decider Logic\n\t\tif execution.WorkflowID == childID {\n\t\t\tif continueAsNewCounter < continueAsNewCount {\n\t\t\t\tcontinueAsNewCounter++\n\t\t\t\tbuf := new(bytes.Buffer)\n\t\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, continueAsNewCounter))\n\n\t\t\t\treturn []byte(strconv.Itoa(int(continueAsNewCounter))), []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tInput: buf.Bytes(),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t}\n\n\t\t\tchildComplete = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"Child Done.\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\t// Parent Decider Logic\n\t\tif execution.WorkflowID == parentID {\n\t\t\tif !childExecutionStarted {\n\t\t\t\ts.Logger.Info(\"Starting child execution.\")\n\t\t\t\tchildExecutionStarted = true\n\t\t\t\tbuf := new(bytes.Buffer)\n\t\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, childData))\n\n\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\t\t\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tDomain:       s.DomainName,\n\t\t\t\t\t\tWorkflowID:   childID,\n\t\t\t\t\t\tWorkflowType: childWorkflowType,\n\t\t\t\t\t\tInput:        buf.Bytes(),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t} else if previousStartedEventID > 0 {\n\t\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\t\tif *event.EventType == types.EventTypeChildWorkflowExecutionStarted {\n\t\t\t\t\t\tstartedEvent = event\n\t\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t\t}\n\n\t\t\t\t\tif *event.EventType == types.EventTypeChildWorkflowExecutionCompleted {\n\t\t\t\t\t\tcompletedEvent = event\n\t\t\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}}, nil\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, nil, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to start child execution\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(childExecutionStarted)\n\n\t// Process ChildExecution Started event and all generations of child executions\n\tfor i := 0; i < 11; i++ {\n\t\ts.Logger.Warn(\"decision: %v\", tag.Counter(i))\n\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\ts.False(childComplete)\n\ts.NotNil(startedEvent)\n\n\t// Process Child Execution final decision to complete it\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(childComplete)\n\n\t// Process ChildExecution completed event and complete parent execution\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(completedEvent)\n\tcompletedAttributes := completedEvent.ChildWorkflowExecutionCompletedEventAttributes\n\ts.Equal(s.DomainName, completedAttributes.Domain)\n\ts.Equal(childID, completedAttributes.WorkflowExecution.WorkflowID)\n\ts.NotEqual(startedEvent.ChildWorkflowExecutionStartedEventAttributes.WorkflowExecution.RunID,\n\t\tcompletedAttributes.WorkflowExecution.RunID)\n\ts.Equal(wtChild, completedAttributes.WorkflowType.Name)\n\ts.Equal([]byte(\"Child Done.\"), completedAttributes.Result)\n\n\ts.Logger.Info(\"Parent Workflow Execution History: \")\n}\n"
  },
  {
    "path": "host/decision_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"encoding/json\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestDecisionHeartbeatingWithEmptyResult() {\n\tid := uuid.New()\n\twt := \"integration-workflow-decision-heartbeating-local-activities\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{\n\t\tName: tl,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\tstickyTaskList := &types.TaskList{\n\t\tName: \"test-sticky-tasklist\",\n\t\tKind: types.TaskListKindSticky.Ptr(),\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(3),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\t// start decision\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.Equal(int64(0), resp1.GetAttempt())\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\ttaskToken := resp1.GetTaskToken()\n\thbTimeout := 0\n\tfor i := 0; i < 12; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp2, err2 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tDecisions: []*types.Decision{},\n\t\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\t\tWorkerTaskList:                stickyTaskList,\n\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t},\n\t\t\tReturnNewDecisionTask:      true,\n\t\t\tForceCreateNewDecisionTask: true,\n\t\t})\n\t\tcancel()\n\t\tif _, ok := err2.(*types.EntityNotExistsError); ok {\n\t\t\thbTimeout++\n\n\t\t\tctx, cancel := createContext()\n\t\t\tresp, err := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\t\tDomain:   s.DomainName,\n\t\t\t\tTaskList: taskList,\n\t\t\t\tIdentity: identity,\n\t\t\t})\n\t\t\tcancel()\n\t\t\ts.Nil(err)\n\t\t\ttaskToken = resp.GetTaskToken()\n\t\t} else {\n\t\t\ts.Nil(err2)\n\t\t\ttaskToken = resp2.DecisionTask.GetTaskToken()\n\t\t}\n\t\ttime.Sleep(time.Second)\n\t}\n\n\ts.Equal(2, hbTimeout)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp5, err5 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: taskToken,\n\t\tDecisions: []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"efg\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\tWorkerTaskList:                stickyTaskList,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t},\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: false,\n\t})\n\ts.Nil(err5)\n\ts.Nil(resp5.DecisionTask)\n\n\ts.assertLastHistoryEvent(we, 41, types.EventTypeWorkflowExecutionCompleted)\n}\n\nfunc (s *IntegrationSuite) TestDecisionHeartbeatingWithLocalActivitiesResult() {\n\tid := uuid.New()\n\twt := \"integration-workflow-decision-heartbeating-local-activities\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{\n\t\tName: tl,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\tstikyTaskList := &types.TaskList{\n\t\tName: \"test-sticky-tasklist\",\n\t\tKind: types.TaskListKindSticky.Ptr(),\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(5),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\t// start decision\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.Equal(int64(0), resp1.GetAttempt())\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp2, err2 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: resp1.GetTaskToken(),\n\t\tDecisions: []*types.Decision{},\n\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\tWorkerTaskList:                stikyTaskList,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t},\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: true,\n\t})\n\ts.Nil(err2)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp3, err3 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: resp2.DecisionTask.GetTaskToken(),\n\t\tDecisions: []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"localActivity1\",\n\t\t\t\t\tDetails:    []byte(\"abc\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\tWorkerTaskList:                stikyTaskList,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t},\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: true,\n\t})\n\ts.Nil(err3)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp4, err4 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: resp3.DecisionTask.GetTaskToken(),\n\t\tDecisions: []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"localActivity2\",\n\t\t\t\t\tDetails:    []byte(\"abc\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\tWorkerTaskList:                stikyTaskList,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t},\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: true,\n\t})\n\ts.Nil(err4)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp5, err5 := s.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: resp4.DecisionTask.GetTaskToken(),\n\t\tDecisions: []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(\"efg\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\tWorkerTaskList:                stikyTaskList,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t},\n\t\tReturnNewDecisionTask:      true,\n\t\tForceCreateNewDecisionTask: false,\n\t})\n\ts.Nil(err5)\n\ts.Nil(resp5.DecisionTask)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskCompleted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskCompleted,\n\t\ttypes.EventTypeMarkerRecorded,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskCompleted,\n\t\ttypes.EventTypeMarkerRecorded,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskCompleted,\n\t\ttypes.EventTypeWorkflowExecutionCompleted,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalBeforeRegularDecisionStarted() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeWorkflowExecutionSignaled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start this transient decision, the attempt should be cleared and it becomes again a regular decision\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.Equal(int64(0), resp1.GetAttempt())\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalAfterRegularDecisionStarted() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start decision to make signals into bufferedEvents\n\t_, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// this signal should be buffered\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalAfterRegularDecisionStartedAndFailDecision() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tcause := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start decision to make signals into bufferedEvents\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// this signal should be buffered\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 3, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// fail this decision to flush buffer, and then another decision will be scheduled\n\terr2 := s.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken: resp1.GetTaskToken(),\n\t\tCause:     &cause,\n\t\tIdentity:  \"integ test\",\n\t})\n\ts.Nil(err2)\n\ts.assertLastHistoryEvent(we, 6, types.EventTypeDecisionTaskScheduled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalBeforeTransientDecisionStarted() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tcause := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\tDomain:   s.DomainName,\n\t\t\tTaskList: taskList,\n\t\t\tIdentity: identity,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\ts.Equal(int64(i), resp1.GetAttempt())\n\t\tif i == 0 {\n\t\t\t// first time is regular decision\n\t\t\ts.Equal(int64(3), resp1.GetStartedEventID())\n\t\t} else {\n\t\t\t// the rest is transient decision\n\t\t\ts.Equal(int64(6), resp1.GetStartedEventID())\n\t\t}\n\n\t\tctx, cancel = createContext()\n\t\terr2 := s.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: resp1.GetTaskToken(),\n\t\t\tCause:     &cause,\n\t\t\tIdentity:  \"integ test\",\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err2)\n\t}\n\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 5, types.EventTypeWorkflowExecutionSignaled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start this transient decision, the attempt should be cleared and it becomes again a regular decision\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.Equal(int64(0), resp1.GetAttempt())\n\ts.assertLastHistoryEvent(we, 7, types.EventTypeDecisionTaskStarted)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalAfterTransientDecisionStarted() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tcause := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\tDomain:   s.DomainName,\n\t\t\tTaskList: taskList,\n\t\t\tIdentity: identity,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\ts.Equal(int64(i), resp1.GetAttempt())\n\t\tif i == 0 {\n\t\t\t// first time is regular decision\n\t\t\ts.Equal(int64(3), resp1.GetStartedEventID())\n\t\t} else {\n\t\t\t// the rest is transient decision\n\t\t\ts.Equal(int64(6), resp1.GetStartedEventID())\n\t\t}\n\n\t\tctx, cancel = createContext()\n\t\terr2 := s.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: resp1.GetTaskToken(),\n\t\t\tCause:     &cause,\n\t\t\tIdentity:  \"integ test\",\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err2)\n\t}\n\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start decision to make signals into bufferedEvents\n\t_, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// this signal should be buffered\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTerminationSignalAfterTransientDecisionStartedAndFailDecision() {\n\tid := uuid.New()\n\twt := \"integration-workflow-transient-decision-test-type\"\n\ttl := id\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp0.RunID,\n\t}\n\n\ts.assertLastHistoryEvent(we, 2, types.EventTypeDecisionTaskScheduled)\n\n\tcause := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\tDomain:   s.DomainName,\n\t\t\tTaskList: taskList,\n\t\t\tIdentity: identity,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\ts.Equal(int64(i), resp1.GetAttempt())\n\t\tif i == 0 {\n\t\t\t// first time is regular decision\n\t\t\ts.Equal(int64(3), resp1.GetStartedEventID())\n\t\t} else {\n\t\t\t// the rest is transient decision\n\t\t\ts.Equal(int64(6), resp1.GetStartedEventID())\n\t\t}\n\n\t\tctx, cancel = createContext()\n\t\terr2 := s.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: resp1.GetTaskToken(),\n\t\t\tCause:     &cause,\n\t\t\tIdentity:  \"integ test\",\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err2)\n\t}\n\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// start decision to make signals into bufferedEvents\n\tresp1, err1 := s.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err1)\n\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// this signal should be buffered\n\terr0 = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tSignalName:        \"sig-for-integ-test\",\n\t\tInput:             []byte(\"\"),\n\t\tIdentity:          \"integ test\",\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err0)\n\ts.assertLastHistoryEvent(we, 4, types.EventTypeDecisionTaskFailed)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// fail this decision to flush buffer\n\terr2 := s.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken: resp1.GetTaskToken(),\n\t\tCause:     &cause,\n\t\tIdentity:  \"integ test\",\n\t})\n\ts.Nil(err2)\n\ts.assertLastHistoryEvent(we, 6, types.EventTypeDecisionTaskScheduled)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// then terminate the worklfow\n\terr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: we,\n\t\tReason:            \"test-reason\",\n\t})\n\ts.Nil(err)\n\n\texpectedHistory := []types.EventType{\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeWorkflowExecutionSignaled,\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t}\n\ts.assertHistory(we, expectedHistory)\n}\n\nfunc (s *IntegrationSuite) assertHistory(we *types.WorkflowExecution, expectedHistory []types.EventType) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: we,\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tdata, err := json.MarshalIndent(history, \"\", \"    \")\n\ts.Nil(err)\n\ts.Equal(len(expectedHistory), len(history.Events), string(data))\n\tfor i, e := range history.Events {\n\t\ts.Equal(expectedHistory[i], e.GetEventType(), \"%v, %v, %v\", strconv.Itoa(i), e.GetEventType().String(), string(data))\n\t}\n}\n\nfunc (s *IntegrationSuite) assertLastHistoryEvent(we *types.WorkflowExecution, count int, eventType types.EventType) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: we,\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tdata, err := json.MarshalIndent(history, \"\", \"    \")\n\ts.Nil(err)\n\ts.Equal(count, len(history.Events), string(data))\n\ts.Equal(eventType, history.Events[len(history.Events)-1].GetEventType(), string(data))\n}\n"
  },
  {
    "path": "host/decision_timeout_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage host\n\nimport (\n\t\"flag\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDecisionTimeoutMaxAttemptsIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_decision_timeout_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\tclusterConfig.HistoryDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.DecisionRetryMaxAttempts:    1,\n\t\tdynamicproperties.EnforceDecisionTaskAttempts: true,\n\t}\n\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(DecisionTimeoutMaxAttemptsIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *DecisionTimeoutMaxAttemptsIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n}\n\nfunc (s *DecisionTimeoutMaxAttemptsIntegrationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *DecisionTimeoutMaxAttemptsIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *DecisionTimeoutMaxAttemptsIntegrationSuite) TestDecisionTimeoutExceedsMaxAttempts() {\n\tid := \"integration-decision-timeout-max-attempts-test\"\n\twt := \"integration-decision-timeout-max-attempts-test-type\"\n\ttl := \"integration-decision-timeout-max-attempts-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1), // 1 second timeout\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.NoError(err)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(resp.RunID))\n\n\tpoller := &TaskPoller{\n\t\tEngine:   s.Engine,\n\t\tDomain:   s.DomainName,\n\t\tTaskList: taskList,\n\t\tIdentity: identity,\n\t\tLogger:   s.Logger,\n\t\tT:        s.T(),\n\t}\n\n\t// First decision task and drop\n\tpoller.PollAndProcessDecisionTask(false, true)\n\n\t// Second decision task and drop\n\tpoller.PollAndProcessDecisionTask(false, true)\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      resp.RunID,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                 s.DomainName,\n\t\tExecution:              we,\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeCloseEvent.Ptr(),\n\t\tWaitForNewEvent:        true,\n\t})\n\ts.NoError(err)\n\n\thistory := historyResponse.History\n\tlastEvent := history.Events[len(history.Events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionTerminated, lastEvent.GetEventType(),\n\t\t\"Expected workflow to be terminated, but last event was %v\", lastEvent.GetEventType())\n\ts.NotNil(lastEvent.WorkflowExecutionTerminatedEventAttributes)\n\ts.Equal(common.FailureReasonDecisionAttemptsExceedsLimit,\n\t\tlastEvent.WorkflowExecutionTerminatedEventAttributes.Reason)\n}\n"
  },
  {
    "path": "host/dynamicconfig.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// Override values for dynamic configs\n\tstaticOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.FrontendUserRPS:                                   3000,\n\t\tdynamicproperties.FrontendVisibilityListMaxQPS:                      200,\n\t\tdynamicproperties.FrontendESIndexMaxResultWindow:                    defaultTestValueOfESIndexMaxResultWindow,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions:                3,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:                 3,\n\t\tdynamicproperties.TimerProcessorHistoryArchivalSizeLimit:            5 * 1024,\n\t\tdynamicproperties.ReplicationTaskProcessorErrorRetryMaxAttempts:     1,\n\t\tdynamicproperties.WriteVisibilityStoreName:                          constants.AdvancedVisibilityModeOff,\n\t\tdynamicproperties.DecisionHeartbeatTimeout:                          5 * time.Second,\n\t\tdynamicproperties.ReplicationTaskFetcherAggregationInterval:         200 * time.Millisecond,\n\t\tdynamicproperties.ReplicationTaskFetcherErrorRetryWait:              50 * time.Millisecond,\n\t\tdynamicproperties.ReplicationTaskProcessorErrorRetryWait:            time.Millisecond,\n\t\tdynamicproperties.EnableConsistentQueryByDomain:                     true,\n\t\tdynamicproperties.MinRetentionDays:                                  0,\n\t\tdynamicproperties.WorkflowDeletionJitterRange:                       1,\n\t\tdynamicproperties.EnableActiveClusterSelectionPolicyInStartWorkflow: true,\n\t}\n)\n\ntype dynamicClient struct {\n\tsync.RWMutex\n\n\toverrides map[dynamicproperties.Key]interface{}\n\tclient    dynamicconfig.Client\n}\n\nfunc (d *dynamicClient) GetValue(name dynamicproperties.Key) (interface{}, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\td.RUnlock()\n\t\treturn val, nil\n\t}\n\td.RUnlock()\n\treturn d.client.GetValue(name)\n}\n\nfunc (d *dynamicClient) GetValueWithFilters(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) (interface{}, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\td.RUnlock()\n\t\treturn val, nil\n\t}\n\td.RUnlock()\n\treturn d.client.GetValueWithFilters(name, filters)\n}\n\nfunc (d *dynamicClient) GetIntValue(name dynamicproperties.IntKey, filters map[dynamicproperties.Filter]interface{}) (int, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif intVal, ok := val.(int); ok {\n\t\t\td.RUnlock()\n\t\t\treturn intVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetIntValue(name, filters)\n}\n\nfunc (d *dynamicClient) GetFloatValue(name dynamicproperties.FloatKey, filters map[dynamicproperties.Filter]interface{}) (float64, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif floatVal, ok := val.(float64); ok {\n\t\t\td.RUnlock()\n\t\t\treturn floatVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetFloatValue(name, filters)\n}\n\nfunc (d *dynamicClient) GetBoolValue(name dynamicproperties.BoolKey, filters map[dynamicproperties.Filter]interface{}) (bool, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif boolVal, ok := val.(bool); ok {\n\t\t\td.RUnlock()\n\t\t\treturn boolVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetBoolValue(name, filters)\n}\n\nfunc (d *dynamicClient) GetStringValue(name dynamicproperties.StringKey, filters map[dynamicproperties.Filter]interface{}) (string, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif stringVal, ok := val.(string); ok {\n\t\t\td.RUnlock()\n\t\t\treturn stringVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetStringValue(name, filters)\n}\n\nfunc (d *dynamicClient) GetMapValue(name dynamicproperties.MapKey, filters map[dynamicproperties.Filter]interface{}) (map[string]interface{}, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif mapVal, ok := val.(map[string]interface{}); ok {\n\t\t\td.RUnlock()\n\t\t\treturn mapVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetMapValue(name, filters)\n}\n\nfunc (d *dynamicClient) GetDurationValue(name dynamicproperties.DurationKey, filters map[dynamicproperties.Filter]interface{}) (time.Duration, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif durationVal, ok := val.(time.Duration); ok {\n\t\t\td.RUnlock()\n\t\t\treturn durationVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetDurationValue(name, filters)\n}\nfunc (d *dynamicClient) GetListValue(name dynamicproperties.ListKey, filters map[dynamicproperties.Filter]interface{}) ([]interface{}, error) {\n\td.RLock()\n\tif val, ok := d.overrides[name]; ok {\n\t\tif listVal, ok := val.([]interface{}); ok {\n\t\t\td.RUnlock()\n\t\t\treturn listVal, nil\n\t\t}\n\t}\n\td.RUnlock()\n\treturn d.client.GetListValue(name, filters)\n}\n\nfunc (d *dynamicClient) UpdateValue(name dynamicproperties.Key, value interface{}) error {\n\tif name == dynamicproperties.WriteVisibilityStoreName { // override for es integration tests\n\t\td.Lock()\n\t\tdefer d.Unlock()\n\t\td.overrides[dynamicproperties.WriteVisibilityStoreName] = value.(string)\n\t\treturn nil\n\t} else if name == dynamicproperties.ReadVisibilityStoreName { // override for pinot integration tests\n\t\td.Lock()\n\t\tdefer d.Unlock()\n\t\td.overrides[dynamicproperties.ReadVisibilityStoreName] = value.(string)\n\t\treturn nil\n\t}\n\treturn d.client.UpdateValue(name, value)\n}\n\nfunc (d *dynamicClient) OverrideValue(name dynamicproperties.Key, value interface{}) {\n\td.Lock()\n\tdefer d.Unlock()\n\td.overrides[name] = value\n}\n\nfunc (d *dynamicClient) ListValue(name dynamicproperties.Key) ([]*types.DynamicConfigEntry, error) {\n\treturn d.client.ListValue(name)\n}\n\nfunc (d *dynamicClient) RestoreValue(name dynamicproperties.Key, filters map[dynamicproperties.Filter]interface{}) error {\n\treturn d.client.RestoreValue(name, filters)\n}\n\nvar _ dynamicconfig.Client = (*dynamicClient)(nil)\n\n// newIntegrationConfigClient - returns a dynamic config client for integration testing\nfunc newIntegrationConfigClient(client dynamicconfig.Client, overrides map[dynamicproperties.Key]interface{}) *dynamicClient {\n\tintegrationClient := &dynamicClient{\n\t\toverrides: make(map[dynamicproperties.Key]interface{}),\n\t\tclient:    client,\n\t}\n\n\tfor key, value := range staticOverrides {\n\t\tintegrationClient.OverrideValue(key, value)\n\t}\n\n\tfor key, value := range overrides {\n\t\tintegrationClient.OverrideValue(key, value)\n\t}\n\treturn integrationClient\n}\n"
  },
  {
    "path": "host/elastic_search_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n// to run locally, make sure kafka and es is running,\n// then run cmd `go test -v ./host -run TestElasticsearchIntegrationSuite -tags esintegration`\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"flag\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/host/esutils\"\n)\n\nconst (\n\tnumOfRetry        = 50\n\twaitTimeInMs      = 400\n\twaitForESToSettle = 4 * time.Second // wait es shards for some time ensure data consistent\n)\n\ntype ElasticSearchIntegrationSuite struct {\n\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t// not merely log an error\n\t*require.Assertions\n\t*IntegrationBase\n\tesClient esutils.ESClient\n\n\ttestSearchAttributeKey string\n\ttestSearchAttributeVal string\n}\n\nfunc TestElasticsearchIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tif TestFlags.SQLPluginName == sqlite.PluginName {\n\t\tt.Skipf(\"SQLite plugin is not supported for integration test with ES, skipping %v\", t.Name())\n\t}\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_elasticsearch_\" + environment.GetESVersion() + \"_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(ElasticSearchIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\n// This cluster use customized threshold for history config\nfunc (s *ElasticSearchIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n\ts.esClient = esutils.CreateESClient(s.Suite.T(), s.TestClusterConfig.ESConfig.URL.String(), environment.GetESVersion())\n\ts.esClient.PutIndexTemplate(s.Suite.T(), \"testdata/es_\"+environment.GetESVersion()+\"_index_template.json\", \"test-visibility-template\")\n\tindexName := s.TestClusterConfig.ESConfig.Indices[constants.VisibilityAppName]\n\ts.esClient.CreateIndex(s.Suite.T(), indexName)\n\ts.putIndexSettings(s.Suite.T(), indexName, defaultTestValueOfESIndexMaxResultWindow)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n\ts.esClient.DeleteIndex(s.Suite.T(), s.TestClusterConfig.ESConfig.Indices[constants.VisibilityAppName])\n}\n\nfunc (s *ElasticSearchIntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.testSearchAttributeKey = definition.CustomStringField\n\ts.testSearchAttributeVal = \"test value\"\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListOpenWorkflow() {\n\tid := \"es-integration-start-workflow-test\"\n\twt := \"es-integration-start-workflow-test-type\"\n\ttl := \"es-integration-start-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\n\tstartTime := time.Now().UnixNano()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startTime)\n\tvar openExecution *types.WorkflowExecutionInfo\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\t\tStartTimeFilter: startFilter,\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t\tcancel()\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(we.GetRunID(), openExecution.GetExecution().GetRunID())\n\ts.Equal(attrValBytes, openExecution.SearchAttributes.GetIndexedFields()[s.testSearchAttributeKey])\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow() {\n\tid := \"es-integration-list-workflow-test\"\n\twt := \"es-integration-list-workflow-test-type\"\n\ttl := \"es-integration-list-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflowByClusterAttribute() {\n\tif TestFlags.SQLPluginName != \"\" {\n\t\ts.T().Skipf(\"active-active is not supported for SQL plugin, skipping %v\", s.T().Name())\n\t}\n\tid := \"es-active-active-integration-list-workflow-test\"\n\twt := \"es-active-active-integration-list-workflow-test-type\"\n\ttl := \"es-active-active-integration-list-workflow-test-tasklist\"\n\trequest := s.createActiveActiveStartWorkflowExecutionRequest(id, wt, tl, &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-east\",\n\t\t},\n\t})\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tdescRequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.ActiveActiveDomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, descRequest)\n\ts.Nil(err)\n\ts.Equal(\"region\", descResp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetScope())\n\ts.Equal(\"us-east\", descResp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetName())\n\n\tquery := `ClusterAttributeScope = \"region\" and ClusterAttributeName = \"us-east\"`\n\ts.testHelperForReadOnceWithDomain(s.ActiveActiveDomainName, we.GetRunID(), query, false, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) startWorkflow(\n\tprefix string,\n\tisCron bool,\n) *types.StartWorkflowExecutionResponse {\n\tid := \"es-integration-list-workflow-\" + prefix + \"-test\"\n\twt := \"es-integration-list-workflow-\" + prefix + \"test-type\"\n\ttl := \"es-integration-list-workflow-\" + prefix + \"test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tif isCron {\n\t\trequest.CronSchedule = \"*/5 * * * *\" // every 5 minutes\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n\treturn we\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListCronWorkflows() {\n\twe1 := s.startWorkflow(\"cron\", true)\n\twe2 := s.startWorkflow(\"regular\", false)\n\n\tquery := fmt.Sprintf(`IsCron = \"true\"`)\n\ts.testHelperForReadOnce(we1.GetRunID(), query, false, true)\n\n\tquery = fmt.Sprintf(`IsCron = \"false\"`)\n\ts.testHelperForReadOnce(we2.GetRunID(), query, false, true)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestIsGlobalSearchAttribute() {\n\twe := s.startWorkflow(\"local\", true)\n\t// global domains are disabled for this integration test, so we can only test the false case\n\tquery := fmt.Sprintf(`NumClusters = \"1\"`)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, true)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_ExecutionTime() {\n\tid := \"es-integration-list-workflow-execution-time-test\"\n\twt := \"es-integration-list-workflow-execution-time-test-type\"\n\ttl := \"es-integration-list-workflow-execution-time-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tcronID := id + \"-cron\"\n\trequest.CronSchedule = \"@every 1m\"\n\trequest.WorkflowID = cronID\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tweCron, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`(WorkflowID = \"%s\" or WorkflowID = \"%s\") and ExecutionTime < %v`, id, cronID, time.Now().UnixNano()+int64(time.Minute))\n\ts.testHelperForReadOnce(weCron.GetRunID(), query, false, false)\n\n\tquery = fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_SearchAttribute() {\n\tid := \"es-integration-list-workflow-by-search-attr-test\"\n\twt := \"es-integration-list-workflow-by-search-attr-test-type\"\n\ttl := \"es-integration-list-workflow-by-search-attr-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n\n\t// test upsert\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: getUpsertSearchAttributes(),\n\t\t\t}}\n\n\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t}\n\ttaskList := &types.TaskList{Name: tl}\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        \"worker1\",\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\n\ttime.Sleep(waitForESToSettle)\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: int32(2),\n\t\tQuery:    fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing and BinaryChecksums = 'binary-v1'`, wt),\n\t}\n\t// verify upsert data is on ES\n\ts.testListResultForUpsertSearchAttributes(listRequest)\n\n\t// verify DescribeWorkflowExecution\n\tdescRequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, descRequest)\n\ts.Nil(err)\n\texpectedSearchAttributes := getUpsertSearchAttributes()\n\ts.Equal(expectedSearchAttributes, descResp.WorkflowExecutionInfo.GetSearchAttributes())\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_PageToken() {\n\tid := \"es-integration-list-workflow-token-test\"\n\twt := \"es-integration-list-workflow-token-test-type\"\n\ttl := \"es-integration-list-workflow-token-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tnumOfWorkflows := defaultTestValueOfESIndexMaxResultWindow - 1 // == 4\n\tpageSize := 3\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_SearchAfter() {\n\tid := \"es-integration-list-workflow-searchAfter-test\"\n\twt := \"es-integration-list-workflow-searchAfter-test-type\"\n\ttl := \"es-integration-list-workflow-searchAfter-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tnumOfWorkflows := defaultTestValueOfESIndexMaxResultWindow + 1 // == 6\n\tpageSize := 4\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_OrQuery() {\n\tid := \"es-integration-list-workflow-or-query-test\"\n\twt := \"es-integration-list-workflow-or-query-test-type\"\n\ttl := \"es-integration-list-workflow-or-query-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\t// start 3 workflows\n\tkey := definition.CustomIntField\n\tattrValBytes, _ := json.Marshal(1)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\tkey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe1, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\trequest.RequestID = uuid.New()\n\trequest.WorkflowID = id + \"-2\"\n\tattrValBytes, _ = json.Marshal(2)\n\tsearchAttr.IndexedFields[key] = attrValBytes\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe2, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\trequest.RequestID = uuid.New()\n\trequest.WorkflowID = id + \"-3\"\n\tattrValBytes, _ = json.Marshal(3)\n\tsearchAttr.IndexedFields[key] = attrValBytes\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe3, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\ttime.Sleep(waitForESToSettle)\n\n\t// query 1 workflow with search attr\n\tquery1 := fmt.Sprintf(`CustomIntField = %d`, 1)\n\tvar openExecution *types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\tQuery:    query1,\n\t}\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(we1.GetRunID(), openExecution.GetExecution().GetRunID())\n\ts.True(openExecution.GetExecutionTime() >= openExecution.GetStartTime())\n\tsearchValBytes := openExecution.SearchAttributes.GetIndexedFields()[key]\n\tvar searchVal int\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(1, searchVal)\n\n\t// query with or clause\n\tquery2 := fmt.Sprintf(`CustomIntField = %d or CustomIntField = %d`, 1, 2)\n\tlistRequest.Query = query2\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 2 {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.Equal(2, len(openExecutions))\n\te1 := openExecutions[0]\n\te2 := openExecutions[1]\n\tif e1.GetExecution().GetRunID() != we1.GetRunID() {\n\t\t// results are sorted by [CloseTime,RunID] desc, so find the correct mapping first\n\t\te1, e2 = e2, e1\n\t}\n\ts.Equal(we1.GetRunID(), e1.GetExecution().GetRunID())\n\ts.Equal(we2.GetRunID(), e2.GetExecution().GetRunID())\n\tsearchValBytes = e2.SearchAttributes.GetIndexedFields()[key]\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(2, searchVal)\n\n\t// query for open\n\tquery3 := fmt.Sprintf(`(CustomIntField = %d or CustomIntField = %d) and CloseTime = missing`, 2, 3)\n\tlistRequest.Query = query3\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 2 {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.Equal(2, len(openExecutions))\n\te1 = openExecutions[0]\n\te2 = openExecutions[1]\n\ts.Equal(we3.GetRunID(), e1.GetExecution().GetRunID())\n\ts.Equal(we2.GetRunID(), e2.GetExecution().GetRunID())\n\tsearchValBytes = e1.SearchAttributes.GetIndexedFields()[key]\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(3, searchVal)\n}\n\n// To test last page search trigger max window size error\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_MaxWindowSize() {\n\tid := \"es-integration-list-workflow-max-window-size-test\"\n\twt := \"es-integration-list-workflow-max-window-size-test-type\"\n\ttl := \"es-integration-list-workflow-max-window-size-test-tasklist\"\n\tstartRequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tfor i := 0; i < defaultTestValueOfESIndexMaxResultWindow; i++ {\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = id + strconv.Itoa(i)\n\t\tctx, cancel := createContext()\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForESToSettle)\n\n\tvar listResp *types.ListWorkflowExecutionsResponse\n\tvar nextPageToken []byte\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        s.DomainName,\n\t\tPageSize:      int32(defaultTestValueOfESIndexMaxResultWindow),\n\t\tNextPageToken: nextPageToken,\n\t\tQuery:         fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t}\n\t// get first page\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == defaultTestValueOfESIndexMaxResultWindow {\n\t\t\tlistResp = resp\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.NotNil(listResp)\n\ts.True(len(listResp.GetNextPageToken()) != 0)\n\n\t// the last request\n\tlistRequest.NextPageToken = listResp.GetNextPageToken()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\ts.Nil(err)\n\ts.True(len(resp.GetExecutions()) == 0)\n\ts.True(len(resp.GetNextPageToken()) == 0)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestListWorkflow_OrderBy() {\n\tid := \"es-integration-list-workflow-order-by-test\"\n\twt := \"es-integration-list-workflow-order-by-test-type\"\n\ttl := \"es-integration-list-workflow-order-by-test-tasklist\"\n\tstartRequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tfor i := 0; i < defaultTestValueOfESIndexMaxResultWindow+1; i++ { // start 6\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = id + strconv.Itoa(i)\n\n\t\tif i < defaultTestValueOfESIndexMaxResultWindow-1 { // 4 workflow has search attr\n\t\t\tintVal, _ := json.Marshal(i)\n\t\t\tdoubleVal, _ := json.Marshal(float64(i))\n\t\t\tstrVal, _ := json.Marshal(strconv.Itoa(i))\n\t\t\ttimeVal, _ := json.Marshal(time.Now())\n\t\t\tsearchAttr := &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\tdefinition.CustomIntField:      intVal,\n\t\t\t\t\tdefinition.CustomDoubleField:   doubleVal,\n\t\t\t\t\tdefinition.CustomKeywordField:  strVal,\n\t\t\t\t\tdefinition.CustomDatetimeField: timeVal,\n\t\t\t\t},\n\t\t\t}\n\t\t\tstartRequest.SearchAttributes = searchAttr\n\t\t} else {\n\t\t\tstartRequest.SearchAttributes = &types.SearchAttributes{}\n\t\t}\n\n\t\tctx, cancel := createContext()\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForESToSettle)\n\n\tdesc := \"desc\"\n\tasc := \"asc\"\n\tqueryTemplate := `WorkflowType = \"%s\" order by %s %s`\n\tpageSize := int32(defaultTestValueOfESIndexMaxResultWindow)\n\n\t// order by CloseTime asc\n\tquery1 := fmt.Sprintf(queryTemplate, wt, definition.CloseTime, asc)\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: pageSize,\n\t\tQuery:    query1,\n\t}\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif int32(len(resp.GetExecutions())) == listRequest.GetPageSize() {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.NotNil(openExecutions)\n\tfor i := int32(1); i < pageSize; i++ {\n\t\ts.True(openExecutions[i-1].GetCloseTime() <= openExecutions[i].GetCloseTime())\n\t}\n\n\t// greatest effort to reduce duplicate code\n\ttestHelper := func(query, searchAttrKey string, prevVal, currVal interface{}) {\n\t\tlistRequest.Query = query\n\t\tlistRequest.NextPageToken = []byte{}\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\topenExecutions = resp.GetExecutions()\n\t\tdec := json.NewDecoder(bytes.NewReader(openExecutions[0].GetSearchAttributes().GetIndexedFields()[searchAttrKey]))\n\t\tdec.UseNumber()\n\t\terr = dec.Decode(&prevVal)\n\t\ts.Nil(err)\n\t\tfor i := int32(1); i < pageSize; i++ {\n\t\t\tindexedFields := openExecutions[i].GetSearchAttributes().GetIndexedFields()\n\t\t\tsearchAttrBytes, ok := indexedFields[searchAttrKey]\n\t\t\tif !ok { // last one doesn't have search attr\n\t\t\t\ts.Equal(pageSize-1, i)\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tdec := json.NewDecoder(bytes.NewReader(searchAttrBytes))\n\t\t\tdec.UseNumber()\n\t\t\terr = dec.Decode(&currVal)\n\t\t\ts.Nil(err)\n\t\t\tvar v1, v2 interface{}\n\t\t\tswitch searchAttrKey {\n\t\t\tcase definition.CustomIntField:\n\t\t\t\tv1, _ = prevVal.(json.Number).Int64()\n\t\t\t\tv2, _ = currVal.(json.Number).Int64()\n\t\t\t\ts.True(v1.(int64) >= v2.(int64))\n\t\t\tcase definition.CustomDoubleField:\n\t\t\t\tv1, _ = prevVal.(json.Number).Float64()\n\t\t\t\tv2, _ = currVal.(json.Number).Float64()\n\t\t\t\ts.True(v1.(float64) >= v2.(float64))\n\t\t\tcase definition.CustomKeywordField:\n\t\t\t\ts.True(prevVal.(string) >= currVal.(string))\n\t\t\tcase definition.CustomDatetimeField:\n\t\t\t\tv1, _ = time.Parse(time.RFC3339, prevVal.(string))\n\t\t\t\tv2, _ = time.Parse(time.RFC3339, currVal.(string))\n\t\t\t\ts.True(v1.(time.Time).After(v2.(time.Time)))\n\t\t\t}\n\t\t\tprevVal = currVal\n\t\t}\n\t\tlistRequest.NextPageToken = resp.GetNextPageToken()\n\t\tctx, cancel = createContext()\n\t\tdefer cancel()\n\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest) // last page\n\t\ts.Nil(err)\n\t\ts.Equal(1, len(resp.GetExecutions()))\n\t}\n\n\t// order by CustomIntField desc\n\tfield := definition.CustomIntField\n\tquery := fmt.Sprintf(queryTemplate, wt, field, desc)\n\tvar int1, int2 int\n\ttestHelper(query, field, int1, int2)\n\n\t// order by CustomDoubleField desc\n\tfield = definition.CustomDoubleField\n\tquery = fmt.Sprintf(queryTemplate, wt, field, desc)\n\tvar double1, double2 float64\n\ttestHelper(query, field, double1, double2)\n\n\t// order by CustomKeywordField desc\n\tfield = definition.CustomKeywordField\n\tquery = fmt.Sprintf(queryTemplate, wt, field, desc)\n\tvar s1, s2 string\n\ttestHelper(query, field, s1, s2)\n\n\t// order by CustomDatetimeField desc\n\tfield = definition.CustomDatetimeField\n\tquery = fmt.Sprintf(queryTemplate, wt, field, desc)\n\tvar t1, t2 time.Time\n\ttestHelper(query, field, t1, t2)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) testListWorkflowHelper(numOfWorkflows, pageSize int,\n\tstartRequest *types.StartWorkflowExecutionRequest, wid, wType string, isScan bool) {\n\n\t// start enough number of workflows\n\tfor i := 0; i < numOfWorkflows; i++ {\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = wid + strconv.Itoa(i)\n\t\tctx, cancel := createContext()\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForESToSettle)\n\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tvar nextPageToken []byte\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        s.DomainName,\n\t\tPageSize:      int32(pageSize),\n\t\tNextPageToken: nextPageToken,\n\t\tQuery:         fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wType),\n\t}\n\t// test first page\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\n\t\tctx, cancel := createContext()\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == pageSize {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tnextPageToken = resp.GetNextPageToken()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.NotNil(openExecutions)\n\ts.NotNil(nextPageToken)\n\ts.True(len(nextPageToken) > 0)\n\n\t// test last page\n\tlistRequest.NextPageToken = nextPageToken\n\tinIf := false\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\n\t\tctx, cancel := createContext()\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == numOfWorkflows-pageSize {\n\t\t\tinIf = true\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tnextPageToken = resp.GetNextPageToken()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.True(inIf)\n\ts.NotNil(openExecutions)\n\ts.Nil(nextPageToken)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) testHelperForReadOnce(runID, query string, isScan bool, isAnyMatchOk bool) {\n\ts.testHelperForReadOnceWithDomain(s.DomainName, runID, query, isScan, isAnyMatchOk)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) testHelperForReadOnceWithDomain(domainName string, runID, query string, isScan bool, isAnyMatchOk bool) {\n\tvar openExecution *types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   domainName,\n\t\tPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\tQuery:    query,\n\t}\nRetry:\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\n\t\tctx, cancel := createContext()\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\t\tcancel()\n\n\t\ts.Nil(err)\n\t\tlogStr := fmt.Sprintf(\"Results for query '%s' (desired runId: %s): \\n\", query, runID)\n\t\ts.Logger.Info(logStr)\n\t\tfor _, e := range resp.GetExecutions() {\n\t\t\tlogStr = fmt.Sprintf(\"Execution: %+v, %+v \\n\", e.Execution, e)\n\t\t\ts.Logger.Info(logStr)\n\t\t}\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\tif isAnyMatchOk {\n\t\t\tfor _, e := range resp.GetExecutions() {\n\t\t\t\tif e.Execution.RunID == runID {\n\t\t\t\t\topenExecution = e\n\t\t\t\t\tbreak Retry\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(runID, openExecution.GetExecution().GetRunID())\n\ts.True(openExecution.GetExecutionTime() >= openExecution.GetStartTime())\n\tif openExecution.SearchAttributes != nil && len(openExecution.SearchAttributes.GetIndexedFields()) > 0 {\n\t\tsearchValBytes := openExecution.SearchAttributes.GetIndexedFields()[s.testSearchAttributeKey]\n\t\tvar searchVal string\n\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\ts.Equal(s.testSearchAttributeVal, searchVal)\n\t}\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestScanWorkflow() {\n\tid := \"es-integration-scan-workflow-test\"\n\twt := \"es-integration-scan-workflow-test-type\"\n\ttl := \"es-integration-scan-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, true, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestScanWorkflow_SearchAttribute() {\n\tid := \"es-integration-scan-workflow-search-attr-test\"\n\twt := \"es-integration-scan-workflow-search-attr-test-type\"\n\ttl := \"es-integration-scan-workflow-search-attr-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\ts.testHelperForReadOnce(we.GetRunID(), query, true, false)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestScanWorkflow_PageToken() {\n\tid := \"es-integration-scan-workflow-token-test\"\n\twt := \"es-integration-scan-workflow-token-test-type\"\n\ttl := \"es-integration-scan-workflow-token-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tnumOfWorkflows := 4\n\tpageSize := 3\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, true)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestCountWorkflow() {\n\tid := \"es-integration-count-workflow-test\"\n\twt := \"es-integration-count-workflow-test-type\"\n\ttl := \"es-integration-count-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t_, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\tcountRequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: s.DomainName,\n\t\tQuery:  query,\n\t}\n\tvar resp *types.CountWorkflowExecutionsResponse\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err = s.Engine.CountWorkflowExecutions(ctx, countRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif resp.GetCount() == int64(1) {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.Equal(int64(1), resp.GetCount())\n\n\tquery = fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, \"noMatch\")\n\tcountRequest.Query = query\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err = s.Engine.CountWorkflowExecutions(ctx, countRequest)\n\ts.Nil(err)\n\ts.Equal(int64(0), resp.GetCount())\n}\n\nfunc (s *ElasticSearchIntegrationSuite) createStartWorkflowExecutionRequest(id, wt, tl string) *types.StartWorkflowExecutionRequest {\n\tidentity := \"worker1\"\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\treturn request\n}\n\nfunc (s *ElasticSearchIntegrationSuite) createActiveActiveStartWorkflowExecutionRequest(id, wt, tl string, policy *types.ActiveClusterSelectionPolicy) *types.StartWorkflowExecutionRequest {\n\tidentity := \"worker1\"\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.ActiveActiveDomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tActiveClusterSelectionPolicy:        policy,\n\t}\n\treturn request\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestUpsertWorkflowExecution() {\n\tid := \"es-integration-upsert-workflow-test\"\n\twt := \"es-integration-upsert-workflow-test-type\"\n\ttl := \"es-integration-upsert-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdecisionCount := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{}}\n\n\t\t// handle first upsert\n\t\tif decisionCount == 0 {\n\t\t\tdecisionCount++\n\n\t\t\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\t\t\tupsertSearchAttr := &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t\t\t},\n\t\t\t}\n\t\t\tupsertDecision.UpsertWorkflowSearchAttributesDecisionAttributes.SearchAttributes = upsertSearchAttr\n\t\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t\t}\n\t\t// handle second upsert, which update existing field and add new field\n\t\tif decisionCount == 1 {\n\t\t\tdecisionCount++\n\t\t\tupsertDecision.UpsertWorkflowSearchAttributesDecisionAttributes.SearchAttributes = getUpsertSearchAttributes()\n\t\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// process 1st decision and assert decision is handled correctly.\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\ts.Equal(int64(3), newTask.DecisionTask.GetPreviousStartedEventID())\n\ts.Equal(int64(7), newTask.DecisionTask.GetStartedEventID())\n\ts.Equal(4, len(newTask.DecisionTask.History.Events))\n\ts.Equal(types.EventTypeDecisionTaskCompleted, newTask.DecisionTask.History.Events[0].GetEventType())\n\ts.Equal(types.EventTypeUpsertWorkflowSearchAttributes, newTask.DecisionTask.History.Events[1].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskScheduled, newTask.DecisionTask.History.Events[2].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskStarted, newTask.DecisionTask.History.Events[3].GetEventType())\n\n\ttime.Sleep(waitForESToSettle)\n\n\t// verify upsert data is on ES\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: int32(2),\n\t\tQuery:    fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t}\n\tverified := false\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\texecution := resp.GetExecutions()[0]\n\t\t\tretrievedSearchAttr := execution.SearchAttributes\n\t\t\tif retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) > 0 {\n\t\t\t\tsearchValBytes := retrievedSearchAttr.GetIndexedFields()[s.testSearchAttributeKey]\n\t\t\t\tvar searchVal string\n\t\t\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\t\t\ts.Equal(s.testSearchAttributeVal, searchVal)\n\t\t\t\tverified = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.True(verified)\n\n\t// process 2nd decision and assert decision is handled correctly.\n\t_, newTask, err = poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\ts.Equal(4, len(newTask.DecisionTask.History.Events))\n\ts.Equal(types.EventTypeDecisionTaskCompleted, newTask.DecisionTask.History.Events[0].GetEventType())\n\ts.Equal(types.EventTypeUpsertWorkflowSearchAttributes, newTask.DecisionTask.History.Events[1].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskScheduled, newTask.DecisionTask.History.Events[2].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskStarted, newTask.DecisionTask.History.Events[3].GetEventType())\n\n\ttime.Sleep(waitForESToSettle)\n\n\t// verify upsert data is on ES\n\ts.testListResultForUpsertSearchAttributes(listRequest)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) testListResultForUpsertSearchAttributes(listRequest *types.ListWorkflowExecutionsRequest) {\n\tverified := false\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\texecution := resp.GetExecutions()[0]\n\t\t\tretrievedSearchAttr := execution.SearchAttributes\n\t\t\tif retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) == 3 {\n\t\t\t\tfields := retrievedSearchAttr.GetIndexedFields()\n\t\t\t\tsearchValBytes := fields[s.testSearchAttributeKey]\n\t\t\t\tvar searchVal string\n\t\t\t\terr := json.Unmarshal(searchValBytes, &searchVal)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal(\"another string\", searchVal)\n\n\t\t\t\tsearchValBytes2 := fields[definition.CustomIntField]\n\t\t\t\tvar searchVal2 int\n\t\t\t\terr = json.Unmarshal(searchValBytes2, &searchVal2)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal(123, searchVal2)\n\n\t\t\t\tbinaryChecksumsBytes := fields[definition.BinaryChecksums]\n\t\t\t\tvar binaryChecksums []string\n\t\t\t\terr = json.Unmarshal(binaryChecksumsBytes, &binaryChecksums)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal([]string{\"binary-v1\", \"binary-v2\"}, binaryChecksums)\n\n\t\t\t\tverified = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.True(verified)\n}\n\nfunc getUpsertSearchAttributes() *types.SearchAttributes {\n\tattrValBytes1, _ := json.Marshal(\"another string\")\n\tattrValBytes2, _ := json.Marshal(123)\n\tbinaryChecksums, _ := json.Marshal([]string{\"binary-v1\", \"binary-v2\"})\n\tupsertSearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\tdefinition.CustomStringField: attrValBytes1,\n\t\t\tdefinition.CustomIntField:    attrValBytes2,\n\t\t\tdefinition.BinaryChecksums:   binaryChecksums,\n\t\t},\n\t}\n\treturn upsertSearchAttr\n}\n\nfunc (s *ElasticSearchIntegrationSuite) TestUpsertWorkflowExecution_InvalidKey() {\n\tid := \"es-integration-upsert-workflow-failed-test\"\n\twt := \"es-integration-upsert-workflow-failed-test-type\"\n\ttl := \"es-integration-upsert-workflow-failed-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\"INVALIDKEY\": []byte(`1`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}}\n\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tdecisionFailedEvent := history.GetEvents()[3]\n\ts.Equal(types.EventTypeDecisionTaskFailed, decisionFailedEvent.GetEventType())\n\tfailedDecisionAttr := decisionFailedEvent.DecisionTaskFailedEventAttributes\n\ts.Equal(types.DecisionTaskFailedCauseBadSearchAttributes, failedDecisionAttr.GetCause())\n\ts.True(len(failedDecisionAttr.GetDetails()) > 0)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) putIndexSettings(t *testing.T, indexName string, maxResultWindowSize int) {\n\terr := s.esClient.PutMaxResultWindow(t, indexName, maxResultWindowSize)\n\ts.Require().NoError(err)\n\ts.verifyMaxResultWindowSize(t, indexName, maxResultWindowSize)\n}\n\nfunc (s *ElasticSearchIntegrationSuite) verifyMaxResultWindowSize(t *testing.T, indexName string, targetSize int) {\n\tfor i := 0; i < numOfRetry; i++ {\n\t\tcurrentWindow, err := s.esClient.GetMaxResultWindow(t, indexName)\n\t\ts.Require().NoError(err)\n\t\tif currentWindow == strconv.Itoa(targetSize) {\n\t\t\treturn\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.FailNow(fmt.Sprintf(\"ES max result window size hasn't reach target size within %v.\", (numOfRetry*waitTimeInMs)*time.Millisecond))\n}\n"
  },
  {
    "path": "host/esutils/client_os2.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esutils\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/opensearch-project/opensearch-go/v4\"\n\tosapi \"github.com/opensearch-project/opensearch-go/v4/opensearchapi\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype (\n\tos2Client struct {\n\t\tclient *osapi.Client\n\t}\n)\n\nfunc newOS2Client(url string) (*os2Client, error) {\n\n\tosClient, err := osapi.NewClient(osapi.Config{\n\t\tClient: opensearch.Config{\n\t\t\tAddresses:    []string{url},\n\t\t\tMaxRetries:   5,\n\t\t\tRetryBackoff: func(i int) time.Duration { return time.Duration(i) * 100 * time.Millisecond },\n\t\t\tUsername:     \"admin\",\n\t\t\tPassword:     \"DevTestInitial123!\", // Admin password from docker setup. Required for >= OSv2.12\n\t\t},\n\t})\n\n\treturn &os2Client{\n\t\tclient: osClient,\n\t}, err\n}\n\nfunc (os2 *os2Client) PutIndexTemplate(t *testing.T, templateConfigFile, templateName string) {\n\t// This function is used exclusively in tests. Excluding it from security checks.\n\t// #nosec\n\ttemplate, err := os.Open(templateConfigFile)\n\trequire.NoError(t, err)\n\n\treq := osapi.IndexTemplateCreateReq{\n\t\tBody:          template,\n\t\tIndexTemplate: templateName,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := os2.client.IndexTemplate.Create(ctx, req)\n\trequire.NoError(t, err)\n\trequire.Truef(t, resp.Acknowledged, \"OS2 put index template unacknowledged: %s\", resp.Inspect().Response.Body)\n}\n\nfunc (os2 *os2Client) CreateIndex(t *testing.T, indexName string) {\n\texistsReq := osapi.IndicesExistsReq{\n\t\tIndices: []string{indexName},\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := os2.client.Indices.Exists(ctx, existsReq)\n\tif resp.StatusCode != http.StatusNotFound {\n\t\trequire.NoError(t, err)\n\t}\n\n\tif resp.StatusCode == http.StatusOK {\n\t\tdeleteReq := osapi.IndicesDeleteReq{\n\t\t\tIndices: []string{indexName},\n\t\t}\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tresp, err := os2.client.Indices.Delete(ctx, deleteReq)\n\t\trequire.Nil(t, err)\n\t\trequire.Truef(t, resp.Acknowledged, \"OS2 delete index unacknowledged: %s\", resp.Inspect().Response.Body)\n\t}\n\n\tresp.Body.Close()\n\n\tcreateReq := osapi.IndicesCreateReq{\n\t\tIndex: indexName,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tcreateResp, err := os2.client.Indices.Create(ctx, createReq)\n\trequire.NoError(t, err)\n\trequire.Truef(t, createResp.Acknowledged, \"OS2 create index unacknowledged: %s\", createResp.Inspect().Response.Body)\n}\n\nfunc (os2 *os2Client) DeleteIndex(t *testing.T, indexName string) {\n\tdeleteReq := osapi.IndicesDeleteReq{\n\t\tIndices: []string{indexName},\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := os2.client.Indices.Delete(ctx, deleteReq)\n\trequire.NoError(t, err)\n\trequire.True(t, resp.Acknowledged, fmt.Sprintf(\"OS2 delete index unacknowledged: %s\", resp.Inspect().Response.Body))\n}\n\nfunc (os2 *os2Client) PutMaxResultWindow(t *testing.T, indexName string, maxResultWindow int) error {\n\n\treq := osapi.SettingsPutReq{\n\t\tBody:    strings.NewReader(fmt.Sprintf(`{\"index\": {\"max_result_window\": %d}}`, maxResultWindow)),\n\t\tIndices: []string{indexName},\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := os2.client.Indices.Settings.Put(ctx, req)\n\trequire.NoError(t, err)\n\trequire.Truef(t, resp.Acknowledged, \"OS2 put index settings unacknowledged: %s\", resp.Inspect().Response.Body)\n\n\treturn nil\n}\n\nfunc (os2 *os2Client) GetMaxResultWindow(t *testing.T, indexName string) (string, error) {\n\treq := &osapi.SettingsGetReq{\n\t\tIndices: []string{indexName},\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\n\tres, err := os2.client.Indices.Settings.Get(ctx, req)\n\trequire.NoError(t, err)\n\n\tindexSettings, ok := res.Indices[indexName]\n\tif !ok {\n\t\treturn \"\", fmt.Errorf(\"no settings for index %q\", indexName)\n\t}\n\n\ttype indexSettingsData struct {\n\t\tIndex struct {\n\t\t\tWindow any `json:\"max_result_window\"`\n\t\t} `json:\"index\"`\n\t}\n\tvar out indexSettingsData\n\terr = json.Unmarshal(indexSettings.Settings, &out)\n\trequire.NoError(t, err)\n\n\tif out.Index.Window == nil {\n\t\treturn \"\", fmt.Errorf(\"no max_result_window value found in index settings\")\n\t}\n\treturn out.Index.Window.(string), nil\n}\n"
  },
  {
    "path": "host/esutils/client_v6.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esutils\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype (\n\tv6Client struct {\n\t\tclient *elastic.Client\n\t}\n)\n\nfunc newV6Client(url string) (*v6Client, error) {\n\tesClient, err := elastic.NewClient(\n\t\telastic.SetURL(url),\n\t\telastic.SetRetrier(elastic.NewBackoffRetrier(elastic.NewExponentialBackoff(128*time.Millisecond, 513*time.Millisecond))),\n\t)\n\treturn &v6Client{\n\t\tclient: esClient,\n\t}, err\n}\n\nfunc (es *v6Client) PutIndexTemplate(t *testing.T, templateConfigFile, templateName string) {\n\t// This function is used exclusively in tests. Excluding it from security checks.\n\t// #nosec\n\ttemplate, err := os.ReadFile(templateConfigFile)\n\trequire.NoError(t, err)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tputTemplate, err := es.client.IndexPutTemplate(templateName).BodyString(string(template)).Do(ctx)\n\trequire.NoError(t, err)\n\trequire.True(t, putTemplate.Acknowledged)\n}\n\nfunc (es *v6Client) CreateIndex(t *testing.T, indexName string) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\texists, err := es.client.IndexExists(indexName).Do(ctx)\n\trequire.NoError(t, err)\n\tif exists {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tdeleteTestIndex, err := es.client.DeleteIndex(indexName).Do(ctx)\n\t\trequire.Nil(t, err)\n\t\trequire.True(t, deleteTestIndex.Acknowledged)\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tcreateTestIndex, err := es.client.CreateIndex(indexName).Do(ctx)\n\trequire.Nil(t, err)\n\trequire.True(t, createTestIndex.Acknowledged)\n}\n\nfunc (es *v6Client) DeleteIndex(t *testing.T, indexName string) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tdeleteTestIndex, err := es.client.DeleteIndex(indexName).Do(ctx)\n\trequire.Nil(t, err)\n\trequire.True(t, deleteTestIndex.Acknowledged)\n}\n\nfunc (es *v6Client) PutMaxResultWindow(t *testing.T, indexName string, maxResultWindow int) error {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t_, err := es.client.IndexPutSettings(indexName).\n\t\tBodyString(fmt.Sprintf(`{\"max_result_window\" : %d}`, maxResultWindow)).\n\t\tDo(ctx)\n\trequire.NoError(t, err)\n\treturn err\n}\n\nfunc (es *v6Client) GetMaxResultWindow(t *testing.T, indexName string) (string, error) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tsettings, err := es.client.IndexGetSettings(indexName).Do(ctx)\n\trequire.NoError(t, err)\n\treturn settings[indexName].Settings[\"index\"].(map[string]interface{})[\"max_result_window\"].(string), nil\n}\n"
  },
  {
    "path": "host/esutils/client_v7.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esutils\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic/v7\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype (\n\tv7Client struct {\n\t\tclient *elastic.Client\n\t}\n)\n\nfunc newV7Client(url string) (*v7Client, error) {\n\tesClient, err := elastic.NewClient(\n\t\telastic.SetURL(url),\n\t\telastic.SetRetrier(elastic.NewBackoffRetrier(elastic.NewExponentialBackoff(128*time.Millisecond, 513*time.Millisecond))),\n\t)\n\treturn &v7Client{\n\t\tclient: esClient,\n\t}, err\n}\n\nfunc (es *v7Client) PutIndexTemplate(t *testing.T, templateConfigFile, templateName string) {\n\t// This function is used exclusively in tests. Excluding it from security checks.\n\t// #nosec\n\ttemplate, err := os.ReadFile(templateConfigFile)\n\trequire.NoError(t, err)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tputTemplate, err := es.client.IndexPutTemplate(templateName).BodyString(string(template)).Do(ctx)\n\trequire.NoError(t, err)\n\trequire.True(t, putTemplate.Acknowledged)\n}\n\nfunc (es *v7Client) CreateIndex(t *testing.T, indexName string) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\texists, err := es.client.IndexExists(indexName).Do(ctx)\n\trequire.NoError(t, err)\n\n\tif exists {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tdeleteTestIndex, err := es.client.DeleteIndex(indexName).Do(ctx)\n\t\trequire.Nil(t, err)\n\t\trequire.True(t, deleteTestIndex.Acknowledged)\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tcreateTestIndex, err := es.client.CreateIndex(indexName).Do(ctx)\n\trequire.NoError(t, err)\n\trequire.True(t, createTestIndex.Acknowledged)\n}\n\nfunc (es *v7Client) DeleteIndex(t *testing.T, indexName string) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tdeleteTestIndex, err := es.client.DeleteIndex(indexName).Do(ctx)\n\trequire.Nil(t, err)\n\trequire.True(t, deleteTestIndex.Acknowledged)\n}\n\nfunc (es *v7Client) PutMaxResultWindow(t *testing.T, indexName string, maxResultWindow int) error {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t_, err := es.client.IndexPutSettings(indexName).\n\t\tBodyString(fmt.Sprintf(`{\"max_result_window\" : %d}`, maxResultWindow)).\n\t\tDo(ctx)\n\trequire.NoError(t, err)\n\treturn err\n}\n\nfunc (es *v7Client) GetMaxResultWindow(t *testing.T, indexName string) (string, error) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tsettings, err := es.client.IndexGetSettings(indexName).Do(ctx)\n\trequire.NoError(t, err)\n\treturn settings[indexName].Settings[\"index\"].(map[string]interface{})[\"max_result_window\"].(string), nil\n}\n"
  },
  {
    "path": "host/esutils/interfaces.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esutils\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\ntype (\n\t// ESClient is ElasicSearch client for running test suite to be implemented in different versions of ES.\n\t// Those interfaces are only being used by tests so we don't implement in common/elasticsearch pkg.\n\tESClient interface {\n\t\tPutIndexTemplate(t *testing.T, templateConfigFile, templateName string)\n\t\tCreateIndex(t *testing.T, indexName string)\n\t\tDeleteIndex(t *testing.T, indexName string)\n\t\tPutMaxResultWindow(t *testing.T, indexName string, maxResultWindow int) error\n\t\tGetMaxResultWindow(t *testing.T, indexName string) (string, error)\n\t}\n)\n\n// CreateESClient create ElasticSearch client for test\nfunc CreateESClient(t *testing.T, url string, version string) ESClient {\n\tvar client ESClient\n\tvar err error\n\tswitch version {\n\tcase \"v6\":\n\t\tclient, err = newV6Client(url)\n\tcase \"v7\":\n\t\tclient, err = newV7Client(url)\n\tcase \"os2\":\n\t\tclient, err = newOS2Client(url)\n\tdefault:\n\t\tassert.FailNow(t, fmt.Sprintf(\"not supported ES version: %s\", version))\n\t}\n\tassert.NoError(t, err)\n\treturn client\n}\n\nfunc createContext() (context.Context, context.CancelFunc) {\n\tctx, cancel := context.WithTimeout(context.Background(), 90*time.Second)\n\treturn ctx, cancel\n}\n"
  },
  {
    "path": "host/flag.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport \"flag\"\n\n// TestFlags contains the feature flags for integration tests\nvar TestFlags struct {\n\tFrontendAddr          string\n\tPersistenceType       string\n\tSQLPluginName         string\n\tTestClusterConfigFile string\n}\n\nfunc init() {\n\tflag.StringVar(&TestFlags.FrontendAddr, \"frontendAddress\", \"\", \"host:port for cadence frontend service\")\n\tflag.StringVar(&TestFlags.PersistenceType, \"persistenceType\", \"cassandra\", \"type of persistence store - [cassandra or sql]\")\n\tflag.StringVar(&TestFlags.SQLPluginName, \"sqlPluginName\", \"mysql\", \"type of sql store - [mysql or postgres]\")\n\tflag.StringVar(&TestFlags.TestClusterConfigFile, \"TestClusterConfigFile\", \"\", \"test cluster config file location\")\n}\n"
  },
  {
    "path": "host/integration_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"math\"\n\t\"os\"\n\t\"sort\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/engine/engineimpl\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n)\n\nfunc TestIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tconfigPath := \"testdata/integration_test_cluster.yaml\"\n\t// TODO: remove this logic once we deprecate history queue v1\n\tif os.Getenv(\"ENABLE_QUEUE_V2\") == \"true\" {\n\t\tconfigPath = \"testdata/integration_queuev2_cluster.yaml\"\n\t\tif os.Getenv(\"ENABLE_QUEUE_V2_ALERT\") == \"true\" {\n\t\t\tconfigPath = \"testdata/integration_queuev2_with_alert_cluster.yaml\"\n\t\t}\n\t}\n\tclusterConfig, err := GetTestClusterConfig(configPath)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(IntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *IntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n}\n\nfunc (s *IntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *IntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *IntegrationSuite) TestStartWorkflowExecution() {\n\tid := \"integration-start-workflow-test\"\n\twt := \"integration-start-workflow-test-type\"\n\ttl := \"integration-start-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tHeader: &types.Header{\n\t\t\tFields: map[string][]byte{\n\t\t\t\t\"test-key\":    []byte(\"test-value\"),\n\t\t\t\t\"empty-value\": {},\n\t\t\t},\n\t\t},\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe1, err1 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err1)\n\ts.Equal(we0.RunID, we1.RunID)\n\n\tnewRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t_, err2 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NotNil(err2)\n\ts.IsType(&types.WorkflowExecutionAlreadyStartedError{}, err2)\n\ts.T().Logf(\"Unable to start workflow execution: %v\\n\", err2.Error())\n}\n\nfunc (s *IntegrationSuite) TestStartWorkflowExecution_StartTimestampMatch() {\n\tid := \"integration-start-workflow-start-timestamp-test\"\n\twt := \"integration-start-workflow-start-timestamptest-type\"\n\ttl := \"integration-start-workflow-start-timestamp-test-tasklist\"\n\tidentity := \"worker1\"\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        &types.WorkflowType{Name: wt},\n\t\tTaskList:                            &types.TaskList{Name: tl},\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe0, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\tvar historyStartTime time.Time\n\tctx, cancel = createContext()\n\tdefer cancel()\n\thistResp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we0.GetRunID(),\n\t\t},\n\t})\n\ts.NoError(err)\n\n\tfor _, event := range histResp.GetHistory().GetEvents() {\n\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionStarted {\n\t\t\thistoryStartTime = time.Unix(0, event.GetTimestamp())\n\t\t\tbreak\n\t\t}\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we0.GetRunID(),\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.WithinDuration(\n\t\thistoryStartTime,\n\t\ttime.Unix(0, descResp.GetWorkflowExecutionInfo().GetStartTime()),\n\t\ttime.Millisecond,\n\t)\n\n\tvar listResp *types.ListOpenWorkflowExecutionsResponse\n\tfor i := 0; i != 20; i++ {\n\t\tctx, cancel := createContext()\n\t\tlistResp, err = s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(historyStartTime.Add(-time.Minute).UnixNano()),\n\t\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we0.GetRunID(),\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.NoError(err)\n\t\tif len(listResp.Executions) == 0 {\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t}\n\t}\n\n\tif listResp == nil || len(listResp.Executions) == 0 {\n\t\ts.Fail(\"unable to get workflow visibility records\")\n\t}\n\n\ts.WithinDuration(\n\t\thistoryStartTime,\n\t\ttime.Unix(0, listResp.Executions[0].GetStartTime()),\n\t\ttime.Millisecond,\n\t)\n}\n\nfunc (s *IntegrationSuite) TestStartWorkflowExecution_IDReusePolicy() {\n\tid := \"integration-start-workflow-id-reuse-test\"\n\twt := \"integration-start-workflow-id-reuse-type\"\n\ttl := \"integration-start-workflow-id-reuse-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tcreateStartRequest := func(policy types.WorkflowIDReusePolicy) *types.StartWorkflowExecutionRequest {\n\t\treturn &types.StartWorkflowExecutionRequest{\n\t\t\tRequestID:                           uuid.New(),\n\t\t\tDomain:                              s.DomainName,\n\t\t\tWorkflowID:                          id,\n\t\t\tWorkflowType:                        workflowType,\n\t\t\tTaskList:                            taskList,\n\t\t\tInput:                               nil,\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\tIdentity:                            identity,\n\t\t\tWorkflowIDReusePolicy:               &policy,\n\t\t}\n\t}\n\n\trequest := createStartRequest(types.WorkflowIDReusePolicyAllowDuplicateFailedOnly)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\t// Test policies when workflow is running\n\tpolicies := []types.WorkflowIDReusePolicy{\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicateFailedOnly,\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicate,\n\t\ttypes.WorkflowIDReusePolicyRejectDuplicate,\n\t}\n\tfor _, policy := range policies {\n\t\tnewRequest := createStartRequest(policy)\n\t\tctx, cancel := createContext()\n\t\t_, err1 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\t\tcancel()\n\t\ts.Error(err1)\n\t\ts.IsType(&types.WorkflowExecutionAlreadyStartedError{}, err1)\n\t}\n\n\t// Test TerminateIfRunning policy when workflow is running\n\tpolicy := types.WorkflowIDReusePolicyTerminateIfRunning\n\tnewRequest := createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe1, err1 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NoError(err1)\n\ts.NotEqual(we.GetRunID(), we1.GetRunID())\n\t// verify terminate status\n\texecutionTerminated := false\nGetHistoryLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif lastEvent.GetEventType() != types.EventTypeWorkflowExecutionTerminated {\n\t\t\ts.Logger.Warn(\"Execution not terminated yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tterminateEventAttributes := lastEvent.WorkflowExecutionTerminatedEventAttributes\n\t\ts.Equal(engineimpl.TerminateIfRunningReason, terminateEventAttributes.GetReason())\n\t\ts.Equal(fmt.Sprintf(engineimpl.TerminateIfRunningDetailsTemplate, we1.GetRunID()), string(terminateEventAttributes.Details))\n\t\ts.Equal(execution.IdentityHistoryService, terminateEventAttributes.GetIdentity())\n\t\texecutionTerminated = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(executionTerminated)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Terminate current workflow execution\n\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we1.RunID,\n\t\t},\n\t\tReason:   \"kill workflow\",\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\t// test policy AllowDuplicateFailedOnly\n\tpolicy = types.WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\tnewRequest = createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe2, err2 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NoError(err2)\n\ts.NotEqual(we1.GetRunID(), we2.GetRunID())\n\t// complete workflow instead of terminate\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\treturn []byte(strconv.Itoa(0)), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// duplicate requests\n\twe3, err3 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NoError(err3)\n\ts.Equal(we2.GetRunID(), we3.GetRunID())\n\t// new request, same policy\n\tnewRequest = createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t_, err3 = s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.Error(err3)\n\ts.IsType(&types.WorkflowExecutionAlreadyStartedError{}, err3)\n\n\t// test policy RejectDuplicate\n\tpolicy = types.WorkflowIDReusePolicyRejectDuplicate\n\tnewRequest = createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t_, err3 = s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.Error(err3)\n\ts.IsType(&types.WorkflowExecutionAlreadyStartedError{}, err3)\n\n\t// test policy AllowDuplicate\n\tpolicy = types.WorkflowIDReusePolicyAllowDuplicate\n\tnewRequest = createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe4, err4 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NoError(err4)\n\ts.NotEqual(we3.GetRunID(), we4.GetRunID())\n\n\t// complete workflow\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// test policy TerminateIfRunning\n\tpolicy = types.WorkflowIDReusePolicyTerminateIfRunning\n\tnewRequest = createStartRequest(policy)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe5, err5 := s.Engine.StartWorkflowExecution(ctx, newRequest)\n\ts.NoError(err5)\n\ts.NotEqual(we4.GetRunID(), we5.GetRunID())\n}\n\nfunc (s *IntegrationSuite) TestTerminateWorkflow() {\n\tid := \"integration-terminate-workflow-test\"\n\twt := \"integration-terminate-workflow-test-type\"\n\ttl := \"integration-terminate-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tterminateReason := \"terminate reason.\"\n\tterminateDetails := []byte(\"terminate details.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tReason:   terminateReason,\n\t\tDetails:  terminateDetails,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\texecutionTerminated := false\nGetHistoryLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionTerminated {\n\t\t\ts.Logger.Warn(\"Execution not terminated yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tterminateEventAttributes := lastEvent.WorkflowExecutionTerminatedEventAttributes\n\t\ts.Equal(terminateReason, terminateEventAttributes.Reason)\n\t\ts.Equal(terminateDetails, terminateEventAttributes.Details)\n\t\ts.Equal(identity, terminateEventAttributes.Identity)\n\t\texecutionTerminated = true\n\t\tbreak GetHistoryLoop\n\t}\n\n\ts.True(executionTerminated)\n\n\tnewExecutionStarted := false\nStartNewExecutionLoop:\n\tfor i := 0; i < 10; i++ {\n\t\trequest := &types.StartWorkflowExecutionRequest{\n\t\t\tRequestID:                           uuid.New(),\n\t\t\tDomain:                              s.DomainName,\n\t\t\tWorkflowID:                          id,\n\t\t\tWorkflowType:                        workflowType,\n\t\t\tTaskList:                            taskList,\n\t\t\tInput:                               nil,\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\tIdentity:                            identity,\n\t\t}\n\n\t\tctx, cancel := createContext()\n\t\tnewExecution, err := s.Engine.StartWorkflowExecution(ctx, request)\n\t\tcancel()\n\t\tif err != nil {\n\t\t\ts.Logger.Warn(\"Start New Execution failed. Error\", tag.Error(err))\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue StartNewExecutionLoop\n\t\t}\n\n\t\ts.Logger.Info(\"New Execution Started with the same ID\", tag.WorkflowID(id),\n\t\t\ttag.WorkflowRunID(newExecution.RunID))\n\t\tnewExecutionStarted = true\n\t\tbreak StartNewExecutionLoop\n\t}\n\n\ts.True(newExecutionStarted)\n}\n\nfunc (s *IntegrationSuite) TestSequentialWorkflow() {\n\tRunSequentialWorkflow(\n\t\ts,\n\t\t\"integration-sequential-workflow-test\",\n\t\t\"integration-sequential-workflow-test-type\",\n\t\t\"integration-sequential-workflow-test-tasklist\",\n\t\t0,\n\t)\n}\n\nfunc (s *IntegrationSuite) TestDelayStartWorkflow() {\n\tstartWorkflowTS := time.Now()\n\tRunSequentialWorkflow(\n\t\ts,\n\t\t\"integration-delay-start-workflow-test\",\n\t\t\"integration-delay-start-workflow-test-type\",\n\t\t\"integration-delay-start-workflow-test-tasklist\",\n\t\t10,\n\t)\n\n\ttargetBackoffDuration := time.Second * 10\n\tbackoffDurationTolerance := time.Millisecond * 4000\n\tbackoffDuration := time.Since(startWorkflowTS)\n\ts.True(\n\t\tbackoffDuration > targetBackoffDuration,\n\t\t\"Backoff duration(%f s) should have been at least 5 seconds\",\n\t\ttime.Duration(backoffDuration).Round(time.Millisecond).Seconds(),\n\t)\n\ts.True(\n\t\tbackoffDuration < targetBackoffDuration+backoffDurationTolerance,\n\t\t\"Integration test too long: %f seconds\",\n\t\ttime.Duration(backoffDuration).Round(time.Millisecond).Seconds(),\n\t)\n}\n\nfunc (s *IntegrationSuite) TestSignalDoesNotOverrideDelayStart() {\n\tid := \"integration-signal-delay-start-test\"\n\twt := \"integration-signal-delay-start-test-type\"\n\ttl := \"integration-signal-delay-start-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\t// Start workflow with a very large delay (300s) so it won't fire during the test\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(600),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t\tDelayStartSeconds:                   common.Int32Ptr(300),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.NoError(err)\n\ts.NotNil(we)\n\n\t// Signal the workflow before the delay expires\n\tsignalName := \"test-signal\"\n\tsignalInput := []byte(`\"payload\"`)\n\tctx2, cancel2 := createContext()\n\tdefer cancel2()\n\terr = s.Engine.SignalWorkflowExecution(ctx2, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.NoError(err)\n\n\t// Describe the workflow and verify no decision task is pending\n\tctx3, cancel3 := createContext()\n\tdefer cancel3()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx3, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.NotNil(descResp)\n\n\t// Workflow should still be open (no close time, no close status)\n\ts.Nil(descResp.WorkflowExecutionInfo.CloseStatus)\n\n\t// No decision task should have been scheduled — the signal must not override DelayStart\n\ts.Nil(descResp.PendingDecision, \"Signal should not schedule a decision task before DelayStart expires\")\n}\n\nfunc RunSequentialWorkflow(\n\ts *IntegrationSuite,\n\tworkflowID string,\n\tworkflowTypeStr string,\n\ttaskListStr string,\n\tdelayStartSeconds int32,\n) {\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = workflowTypeStr\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = taskListStr\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          workflowID,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tDelayStartSeconds:                   common.Int32Ptr(delayStartSeconds),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(10)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: taskListStr},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\texpectedActivity := int32(1)\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(workflowID, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\tid, _ := strconv.Atoi(ActivityID)\n\t\ts.Equal(int(expectedActivity), id)\n\t\tbuf := bytes.NewReader(input)\n\t\tvar in int32\n\t\tbinary.Read(buf, binary.LittleEndian, &in)\n\t\ts.Equal(expectedActivity, in)\n\t\texpectedActivity++\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := 0; i < 10; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t\tif i%2 == 0 {\n\t\t\terr = poller.PollAndProcessActivityTask(false)\n\t\t} else { // just for testing respondActivityTaskCompleteByID\n\t\t\terr = poller.PollAndProcessActivityTaskWithID(false)\n\t\t}\n\t\ts.Logger.Info(\"PollAndProcessActivityTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\ts.False(workflowComplete)\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestCompleteDecisionTaskAndCreateNewOne() {\n\tid := \"integration-complete-decision-create-new-test\"\n\twt := \"integration-complete-decision-create-new-test-type\"\n\ttl := \"integration-complete-decision-create-new-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdecisionCount := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif decisionCount < 2 {\n\t\t\tdecisionCount++\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"test-marker\",\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\n\ts.Equal(int64(3), newTask.DecisionTask.GetPreviousStartedEventID())\n\ts.Equal(int64(7), newTask.DecisionTask.GetStartedEventID())\n\ts.Equal(4, len(newTask.DecisionTask.History.Events))\n\ts.Equal(types.EventTypeDecisionTaskCompleted, newTask.DecisionTask.History.Events[0].GetEventType())\n\ts.Equal(types.EventTypeMarkerRecorded, newTask.DecisionTask.History.Events[1].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskScheduled, newTask.DecisionTask.History.Events[2].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskStarted, newTask.DecisionTask.History.Events[3].GetEventType())\n}\n\nfunc (s *IntegrationSuite) TestDecisionAndActivityTimeoutsWorkflow() {\n\tid := \"integration-timeouts-workflow-test\"\n\twt := \"integration-timeouts-workflow-test-type\"\n\ttl := \"integration-timeouts-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_timer\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(4)\n\tactivityCounter := int32(0)\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(1),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\ts.Logger.Info(\"Completing types.\")\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(id, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\t\ts.Logger.Info(\"Activity ID\", tag.WorkflowActivityID(ActivityID))\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := 0; i < 8; i++ {\n\t\tdropDecisionTask := (i%2 == 0)\n\t\ts.Logger.Info(\"Calling Decision Task\", tag.Counter(i))\n\t\tvar err error\n\t\tif dropDecisionTask {\n\t\t\t_, err = poller.PollAndProcessDecisionTask(false, true)\n\t\t} else {\n\t\t\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, int64(1))\n\t\t}\n\t\tif err != nil {\n\t\t\tctx, cancel := createContext()\n\t\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\t\tDomain: s.DomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\tRunID:      we.RunID,\n\t\t\t\t},\n\t\t\t})\n\t\t\tcancel()\n\t\t\ts.Nil(err)\n\t\t\thistory := historyResponse.History\n\t\t\tPrettyPrintHistory(history, s.Logger)\n\t\t}\n\t\ts.True(err == nil || err == tasklist.ErrNoTasks, \"%v\", err)\n\t\tif !dropDecisionTask {\n\t\t\ts.Logger.Info(\"Calling Activity Task: %d\", tag.Counter(i))\n\t\t\terr = poller.PollAndProcessActivityTask(i%4 == 0)\n\t\t\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\t\t}\n\t}\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\n\ts.False(workflowComplete)\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowRetry() {\n\tid := \"integration-wf-retry-test\"\n\twt := \"integration-wf-retry-type\"\n\ttl := \"integration-wf-retry-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tinitialIntervalInSeconds := 1\n\tbackoffCoefficient := 1.5\n\tmaximumAttempts := 5\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    int32(initialIntervalInSeconds),\n\t\t\tMaximumAttempts:             int32(maximumAttempts),\n\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\t\tBackoffCoefficient:          backoffCoefficient,\n\t\t\tExpirationIntervalInSeconds: 100,\n\t\t},\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tvar executions []*types.WorkflowExecution\n\n\tattemptCount := 0\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\texecutions = append(executions, execution)\n\t\tattemptCount++\n\t\tif attemptCount == maximumAttempts {\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tResult: []byte(\"succeed-after-retry\"),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t}\n\t\treturn nil, []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(\"retryable-error\"),\n\t\t\t\t\tDetails: nil,\n\t\t\t\t},\n\t\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tdescribeWorkflowExecution := func(execution *types.WorkflowExecution) (*types.DescribeWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain:    s.DomainName,\n\t\t\tExecution: execution,\n\t\t})\n\t}\n\n\tfor i := 0; i != maximumAttempts; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.True(err == nil, err)\n\t\tevents := s.getHistory(s.DomainName, executions[i])\n\t\tif i == maximumAttempts-1 {\n\t\t\ts.Equal(types.EventTypeWorkflowExecutionCompleted, events[len(events)-1].GetEventType())\n\t\t} else {\n\t\t\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, events[len(events)-1].GetEventType())\n\t\t}\n\t\ts.Equal(int32(i), events[0].GetWorkflowExecutionStartedEventAttributes().GetAttempt())\n\n\t\tdweResponse, err := describeWorkflowExecution(executions[i])\n\t\ts.Nil(err)\n\t\tbackoff := time.Duration(0)\n\t\tif i > 0 {\n\t\t\tbackoff = time.Duration(float64(initialIntervalInSeconds)*math.Pow(backoffCoefficient, float64(i-1))) * time.Second\n\t\t\t// retry backoff cannot larger than MaximumIntervalInSeconds\n\t\t\tif backoff > time.Second {\n\t\t\t\tbackoff = time.Second\n\t\t\t}\n\t\t}\n\t\texpectedExecutionTime := dweResponse.WorkflowExecutionInfo.GetStartTime() + backoff.Nanoseconds()\n\t\ts.Equal(expectedExecutionTime, dweResponse.WorkflowExecutionInfo.GetExecutionTime())\n\t}\n}\n\nfunc (s *IntegrationSuite) TestWorkflowRetryFailures() {\n\tid := \"integration-wf-retry-failures-test\"\n\twt := \"integration-wf-retry-failures-type\"\n\ttl := \"integration-wf-retry-failures-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tworkflowImpl := func(attempts int, errorReason string, executions *[]*types.WorkflowExecution) decisionTaskHandler {\n\t\tattemptCount := 0\n\n\t\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t*executions = append(*executions, execution)\n\t\t\tattemptCount++\n\t\t\tif attemptCount == attempts {\n\t\t\t\treturn nil, []*types.Decision{\n\t\t\t\t\t{\n\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tResult: []byte(\"succeed-after-retry\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}}, nil\n\t\t\t}\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t// Reason:  common.StringPtr(\"retryable-error\"),\n\t\t\t\t\t\tReason:  common.StringPtr(errorReason),\n\t\t\t\t\t\tDetails: nil,\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t}\n\n\t\treturn dtHandler\n\t}\n\n\t// Fail using attempt\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tMaximumAttempts:             3,\n\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\t\tBackoffCoefficient:          1,\n\t\t\tExpirationIntervalInSeconds: 100,\n\t\t},\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\texecutions := []*types.WorkflowExecution{}\n\tdtHandler := workflowImpl(5, \"retryable-error\", &executions)\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\tevents := s.getHistory(s.DomainName, executions[0])\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, events[len(events)-1].GetEventType())\n\ts.Equal(int32(0), events[0].GetWorkflowExecutionStartedEventAttributes().GetAttempt())\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\tevents = s.getHistory(s.DomainName, executions[1])\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, events[len(events)-1].GetEventType())\n\ts.Equal(int32(1), events[0].GetWorkflowExecutionStartedEventAttributes().GetAttempt())\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\tevents = s.getHistory(s.DomainName, executions[2])\n\ts.Equal(types.EventTypeWorkflowExecutionFailed, events[len(events)-1].GetEventType())\n\ts.Equal(int32(2), events[0].GetWorkflowExecutionStartedEventAttributes().GetAttempt())\n\n\t// Fail error reason\n\trequest = &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tMaximumAttempts:             3,\n\t\t\tMaximumIntervalInSeconds:    1,\n\t\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\t\tBackoffCoefficient:          1,\n\t\t\tExpirationIntervalInSeconds: 100,\n\t\t},\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe, err0 = s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\texecutions = []*types.WorkflowExecution{}\n\tdtHandler = workflowImpl(5, \"bad-bug\", &executions)\n\tpoller = &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\tevents = s.getHistory(s.DomainName, executions[0])\n\ts.Equal(types.EventTypeWorkflowExecutionFailed, events[len(events)-1].GetEventType())\n\ts.Equal(int32(0), events[0].GetWorkflowExecutionStartedEventAttributes().GetAttempt())\n\n}\n\nfunc (s *IntegrationSuite) TestCronWorkflow() {\n\tid := \"integration-wf-cron-test\"\n\twt := \"integration-wf-cron-type\"\n\ttl := \"integration-wf-cron-tasklist\"\n\tidentity := \"worker1\"\n\tcronSchedule := \"@every 3s\"\n\n\ttargetBackoffDuration := time.Second * 3\n\tbackoffDurationTolerance := time.Millisecond * 500\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\"memoKey\": []byte(\"memoVal\")},\n\t}\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\"CustomKeywordField\": []byte(`\"1\"`)},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tCronSchedule:                        cronSchedule, // minimum interval by standard spec is 1m (* * * * *), use non-standard descriptor for short interval for test\n\t\tMemo:                                memo,\n\t\tSearchAttributes:                    searchAttr,\n\t}\n\n\tstartWorkflowTS := time.Now()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tvar executions []*types.WorkflowExecution\n\n\tattemptCount := 0\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\texecutions = append(executions, execution)\n\t\tattemptCount++\n\t\tif attemptCount == 2 {\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tResult: []byte(\"cron-test-result\"),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t}\n\t\treturn nil, []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(\"cron-test-error\"),\n\t\t\t\t\tDetails: nil,\n\t\t\t\t},\n\t\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startWorkflowTS.UnixNano())\n\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\n\t// Sleep some time before checking the open executions.\n\t// This will not cost extra time as the polling for first decision task will be blocked for 3 seconds.\n\ttime.Sleep(2 * time.Second)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          s.DomainName,\n\t\tMaximumPageSize: 100,\n\t\tStartTimeFilter: startFilter,\n\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\tWorkflowID: id,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.Equal(1, len(resp.GetExecutions()))\n\texecutionInfo := resp.GetExecutions()[0]\n\ts.Equal(targetBackoffDuration.Nanoseconds(), executionInfo.GetExecutionTime()-executionInfo.GetStartTime())\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\t// Make sure the cron workflow start running at a proper time, in this case 3 seconds after the\n\t// startWorkflowExecution request\n\tbackoffDuration := time.Since(startWorkflowTS)\n\ts.True(backoffDuration > targetBackoffDuration)\n\ts.True(backoffDuration < targetBackoffDuration+backoffDurationTolerance)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\ts.Equal(3, attemptCount)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tterminateErr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t})\n\ts.NoError(terminateErr)\n\tevents := s.getHistory(s.DomainName, executions[0])\n\tlastEvent := events[len(events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\tattributes := lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\ts.Equal(\"cron-test-error\", attributes.GetFailureReason())\n\ts.Equal(0, len(attributes.GetLastCompletionResult()))\n\ts.Equal(memo, attributes.Memo)\n\ts.Equal(searchAttr, attributes.SearchAttributes)\n\n\tevents = s.getHistory(s.DomainName, executions[1])\n\tlastEvent = events[len(events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\tattributes = lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\ts.Equal(\"\", attributes.GetFailureReason())\n\ts.Equal(\"cron-test-result\", string(attributes.GetLastCompletionResult()))\n\ts.Equal(memo, attributes.Memo)\n\ts.Equal(searchAttr, attributes.SearchAttributes)\n\n\tevents = s.getHistory(s.DomainName, executions[2])\n\tlastEvent = events[len(events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\tattributes = lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\ts.Equal(\"cron-test-error\", attributes.GetFailureReason())\n\ts.Equal(\"cron-test-result\", string(attributes.GetLastCompletionResult()))\n\ts.Equal(memo, attributes.Memo)\n\ts.Equal(searchAttr, attributes.SearchAttributes)\n\n\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\tvar closedExecutions []*types.WorkflowExecutionInfo\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 4 {\n\t\t\tclosedExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(200 * time.Millisecond)\n\t}\n\ts.NotNil(closedExecutions)\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdweResponse, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.Nil(err)\n\texpectedExecutionTime := dweResponse.WorkflowExecutionInfo.GetStartTime() + 3*time.Second.Nanoseconds()\n\ts.Equal(expectedExecutionTime, dweResponse.WorkflowExecutionInfo.GetExecutionTime())\n\n\tsort.Slice(closedExecutions, func(i, j int) bool {\n\t\treturn closedExecutions[i].GetStartTime() < closedExecutions[j].GetStartTime()\n\t})\n\tlastExecution := closedExecutions[0]\n\n\tfor i := 1; i != 4; i++ {\n\t\texecutionInfo := closedExecutions[i]\n\t\texpectedBackoff := executionInfo.GetExecutionTime() - lastExecution.GetExecutionTime()\n\t\t// The execution time calculate based on last execution close time\n\t\t// However, the current execution time is based on the current start time\n\t\t// This code is to remove the diff between current start time and last execution close time\n\t\t// TODO: Remove this line once we unify the time source\n\t\texecutionTimeDiff := executionInfo.GetStartTime() - lastExecution.GetCloseTime()\n\t\t// The backoff between any two executions should be multiplier of the target backoff duration which is 3 in this test\n\t\t// However, it's difficult to guarantee accuracy within a second, we allows 1s as buffering...\n\t\tbackoffSeconds := int(time.Duration(expectedBackoff - executionTimeDiff).Round(time.Second).Seconds())\n\t\ttargetBackoffSeconds := int(targetBackoffDuration.Seconds())\n\t\ttargetDiff := int(math.Abs(float64(backoffSeconds%targetBackoffSeconds-3))) % 3\n\n\t\ts.True(\n\t\t\ttargetDiff <= 1,\n\t\t\t\"Still Flaky?:((%v-%v) - (%v-%v)), backoffSeconds: %v, targetBackoffSeconds: %v, targetDiff:%v\",\n\t\t\tbackoffSeconds,\n\t\t\texecutionInfo.GetExecutionTime(),\n\t\t\tlastExecution.GetExecutionTime(),\n\t\t\texecutionInfo.GetStartTime(),\n\t\t\tlastExecution.GetCloseTime(),\n\t\t\ttargetBackoffSeconds,\n\t\t\ttargetDiff,\n\t\t)\n\t\tlastExecution = executionInfo\n\t}\n}\n\nfunc (s *IntegrationSuite) TestCronWorkflowTimeout() {\n\tid := \"integration-wf-cron-timeout-test\"\n\twt := \"integration-wf-cron-timeout-type\"\n\ttl := \"integration-wf-cron-timeout-tasklist\"\n\tidentity := \"worker1\"\n\tcronSchedule := \"@every 3s\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\"memoKey\": []byte(\"memoVal\")},\n\t}\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\"CustomKeywordField\": []byte(`\"1\"`)},\n\t}\n\tretryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tBackoffCoefficient:          2.0,\n\t\tMaximumIntervalInSeconds:    1,\n\t\tExpirationIntervalInSeconds: 1,\n\t\tMaximumAttempts:             0,\n\t\tNonRetriableErrorReasons:    []string{\"bad-error\"},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1), // set workflow timeout to 1s\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tCronSchedule:                        cronSchedule, // minimum interval by standard spec is 1m (* * * * *), use non-standard descriptor for short interval for test\n\t\tMemo:                                memo,\n\t\tSearchAttributes:                    searchAttr,\n\t\tRetryPolicy:                         retryPolicy,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tvar executions []*types.WorkflowExecution\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\texecutions = append(executions, execution)\n\t\treturn nil, []*types.Decision{\n\t\t\t{\n\t\t\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\n\t\t\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\t\t\tTimerID:                   \"timer-id\",\n\t\t\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\ttime.Sleep(1 * time.Second) // wait for workflow timeout\n\n\t// check when workflow timeout, continueAsNew event contains expected fields\n\tevents := s.getHistory(s.DomainName, executions[0])\n\tlastEvent := events[len(events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\tattributes := lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\ts.Equal(\"cadenceInternal:Timeout START_TO_CLOSE\", attributes.GetFailureReason())\n\ts.Equal(memo, attributes.Memo)\n\ts.Equal(searchAttr, attributes.SearchAttributes)\n\n\tfirstStartWorkflowEvent := events[0]\n\ts.NotNil(firstStartWorkflowEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil, err)\n\n\t// check new run contains expected fields\n\tevents = s.getHistory(s.DomainName, executions[1])\n\tfirstEvent := events[0]\n\ts.Equal(types.EventTypeWorkflowExecutionStarted, firstEvent.GetEventType())\n\tstartAttributes := firstEvent.WorkflowExecutionStartedEventAttributes\n\ts.Equal(memo, startAttributes.Memo)\n\ts.Equal(searchAttr, startAttributes.SearchAttributes)\n\n\ts.NotNil(firstEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp)\n\t// test that the new workflow has a different expiration timestamp from the first workflow\n\ts.True(*firstEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp >\n\t\t*firstStartWorkflowEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp)\n\n\t// terminate cron\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tterminateErr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t})\n\ts.NoError(terminateErr)\n}\n\nfunc (s *IntegrationSuite) TestSequential_UserTimers() {\n\tid := \"integration-sequential-user-timers-test\"\n\twt := \"integration-sequential-user-timers-test-type\"\n\ttl := \"integration-sequential-user-timers-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\ttimerCount := int32(4)\n\ttimerCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif timerCounter < timerCount {\n\t\t\ttimerCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, timerCounter))\n\t\t\treturn []byte(strconv.Itoa(int(timerCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\t\t\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\t\t\tTimerID:                   fmt.Sprintf(\"timer-id-%d\", timerCounter),\n\t\t\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(timerCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := 0; i < 4; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\t\ts.Nil(err)\n\t}\n\n\ts.False(workflowComplete)\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestRateLimitBufferedEvents() {\n\tid := \"integration-rate-limit-buffered-events-test\"\n\twt := \"integration-rate-limit-buffered-events-test-type\"\n\ttl := \"integration-rate-limit-buffered-events-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tworkflowComplete := false\n\tsignalsSent := false\n\tsignalCount := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, h *types.History) ([]byte, []*types.Decision, error) {\n\n\t\t// Count signals\n\t\tfor _, event := range h.Events[previousStartedEventID:] {\n\t\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\tsignalCount++\n\t\t\t}\n\t\t}\n\n\t\tif !signalsSent {\n\t\t\tsignalsSent = true\n\t\t\t// Buffered Signals\n\t\t\tfor i := 0; i < 100; i++ {\n\t\t\t\tbuf := new(bytes.Buffer)\n\t\t\t\tbinary.Write(buf, binary.LittleEndian, int64(i))\n\t\t\t\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"SignalName\", buf.Bytes(), identity))\n\t\t\t}\n\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\tbinary.Write(buf, binary.LittleEndian, int64(101))\n\t\t\tsignalErr := s.sendSignal(s.DomainName, workflowExecution, \"SignalName\", buf.Bytes(), identity)\n\t\t\ts.Nil(signalErr)\n\n\t\t\t// this decision will be ignored as he decision task is already failed\n\t\t\treturn nil, []*types.Decision{}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// first decision to send 101 signals, the last signal will force fail decision and flush buffered events.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.EqualError(err, \"Decision task not found.\")\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n\ts.Equal(101, signalCount) // check that all 101 signals are received.\n}\n\nfunc (s *IntegrationSuite) TestBufferedEvents() {\n\tid := \"integration-buffered-events-test\"\n\twt := \"integration-buffered-events-test-type\"\n\ttl := \"integration-buffered-events-test-tasklist\"\n\tidentity := \"worker1\"\n\tsignalName := \"buffered-signal\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tworkflowComplete := false\n\tsignalSent := false\n\tvar signalEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !signalSent {\n\t\t\tsignalSent = true\n\n\t\t\tctx, cancel := createContext()\n\t\t\tdefer cancel()\n\t\t\t// this will create new event when there is in-flight decision task, and the new event will be buffered\n\t\t\terr := s.Engine.SignalWorkflowExecution(ctx,\n\t\t\t\t&types.SignalWorkflowExecutionRequest{\n\t\t\t\t\tDomain: s.DomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\t},\n\t\t\t\t\tSignalName: \"buffered-signal\",\n\t\t\t\t\tInput:      []byte(\"buffered-signal-input\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t})\n\t\t\ts.NoError(err)\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"test-activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         []byte(\"test-input\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 && signalEvent == nil {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// first decision, which sends signal and the signal event should be buffered to append after first decision closed\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// check history, the signal event should be after the complete decision task\n\thistResp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.NotNil(histResp.History.Events)\n\ts.True(len(histResp.History.Events) >= 6)\n\ts.Equal(histResp.History.Events[3].GetEventType(), types.EventTypeDecisionTaskCompleted)\n\ts.Equal(histResp.History.Events[4].GetEventType(), types.EventTypeActivityTaskScheduled)\n\ts.Equal(histResp.History.Events[5].GetEventType(), types.EventTypeWorkflowExecutionSignaled)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(signalEvent)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestDescribeWorkflowExecution() {\n\tid := \"integration-describe-wfe-test\"\n\twt := \"integration-describe-wfe-test-type\"\n\ttl := \"integration-describe-wfe-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\tchildID := id + \"-child\"\n\tchildType := wt + \"-child\"\n\tchildTaskList := tl + \"-child\"\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tCronOverlapPolicy:                   types.CronOverlapPolicySkipped.Ptr(),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdescribeWorkflowExecution := func() (*types.DescribeWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t}\n\tdweResponse, err := describeWorkflowExecution()\n\ts.Nil(err)\n\ts.True(nil == dweResponse.WorkflowExecutionInfo.CloseTime)\n\ts.Equal(int64(2), dweResponse.WorkflowExecutionInfo.HistoryLength) // WorkflowStarted, DecisionScheduled\n\ts.Equal(dweResponse.WorkflowExecutionInfo.GetStartTime(), dweResponse.WorkflowExecutionInfo.GetExecutionTime())\n\n\t// decider logic\n\tworkflowComplete := false\n\tsignalSent := false\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !signalSent {\n\t\t\tsignalSent = true\n\n\t\t\ts.NoError(err)\n\t\t\treturn nil, []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"test-activity-type\"},\n\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\t\tInput:                         []byte(\"test-input\"),\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\t\t\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tWorkflowID:                          childID,\n\t\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: childType},\n\t\t\t\t\t\tTaskList:                            &types.TaskList{Name: childTaskList},\n\t\t\t\t\t\tInput:                               []byte(\"child-workflow-input\"),\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(200),\n\t\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\t\t\t\tControl:                             nil,\n\t\t\t\t\t\tCronOverlapPolicy:                   types.CronOverlapPolicySkipped.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// first decision to schedule new activity\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// wait for child workflow to start\n\tfor i := 0; i != 10; i++ {\n\t\tdweResponse, err = describeWorkflowExecution()\n\t\ts.Nil(err)\n\t\tif len(dweResponse.PendingChildren) == 1 &&\n\t\t\tdweResponse.PendingChildren[0].GetRunID() != \"\" {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.NotEmpty(dweResponse.PendingChildren[0].GetRunID(), \"unable to start child workflow\")\n\ts.True(nil == dweResponse.WorkflowExecutionInfo.CloseStatus)\n\t// DecisionStarted, DecisionCompleted, ActivityScheduled, ChildWorkflowInit, ChildWorkflowStarted, DecisionTaskScheduled\n\ts.Equal(int64(8), dweResponse.WorkflowExecutionInfo.HistoryLength)\n\ts.Equal(1, len(dweResponse.PendingActivities))\n\ts.Equal(\"test-activity-type\", dweResponse.PendingActivities[0].ActivityType.GetName())\n\ts.Equal(int64(0), dweResponse.PendingActivities[0].GetLastHeartbeatTimestamp())\n\ts.Equal(1, len(dweResponse.PendingChildren))\n\ts.Equal(s.DomainName, dweResponse.PendingChildren[0].GetDomain())\n\ts.Equal(childID, dweResponse.PendingChildren[0].GetWorkflowID())\n\ts.Equal(childType, dweResponse.PendingChildren[0].GetWorkflowTypeName())\n\ts.Equal(types.CronOverlapPolicySkipped, *dweResponse.WorkflowExecutionInfo.CronOverlapPolicy)\n\ts.Equal(false, dweResponse.WorkflowExecutionInfo.IsCron)\n\n\t// process activity task\n\terr = poller.PollAndProcessActivityTask(false)\n\n\tdweResponse, err = describeWorkflowExecution()\n\ts.Nil(err)\n\ts.True(nil == dweResponse.WorkflowExecutionInfo.CloseStatus)\n\ts.Equal(int64(10), dweResponse.WorkflowExecutionInfo.HistoryLength) // ActivityTaskStarted, ActivityTaskCompleted\n\ts.Equal(0, len(dweResponse.PendingActivities))\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\ts.True(workflowComplete)\n\n\tdweResponse, err = describeWorkflowExecution()\n\ts.Nil(err)\n\ts.Equal(types.WorkflowExecutionCloseStatusCompleted, *dweResponse.WorkflowExecutionInfo.CloseStatus)\n\ts.Equal(int64(13), dweResponse.WorkflowExecutionInfo.HistoryLength) // DecisionStarted, DecisionCompleted, WorkflowCompleted\n}\n\nfunc (s *IntegrationSuite) TestVisibility() {\n\tstartTime := time.Now().UnixNano()\n\n\t// Start 2 workflow executions\n\tid1 := \"integration-visibility-test1\"\n\tid2 := \"integration-visibility-test2\"\n\twt := \"integration-visibility-test-type\"\n\ttl := \"integration-visibility-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id1,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tstartResponse, err0 := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\ts.Nil(err0)\n\n\t// Now complete one of the executions\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\treturn []byte{}, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err1 := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err1)\n\n\t// wait until the start workflow is done\n\tvar nextToken []byte\n\thistoryEventFilterType := types.HistoryEventFilterTypeCloseEvent\n\tfor {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, historyErr := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: startRequest.Domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: startRequest.WorkflowID,\n\t\t\t\tRunID:      startResponse.RunID,\n\t\t\t},\n\t\t\tWaitForNewEvent:        true,\n\t\t\tHistoryEventFilterType: &historyEventFilterType,\n\t\t\tNextPageToken:          nextToken,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(historyErr)\n\t\tif len(historyResponse.NextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\n\t\tnextToken = historyResponse.NextPageToken\n\t}\n\n\tstartRequest = &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id2,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t_, err2 := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\ts.Nil(err2)\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startTime)\n\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\n\tclosedCount := 0\n\topenCount := 0\n\n\tvar historyLength int64\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err3 := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err3)\n\t\tclosedCount = len(resp.Executions)\n\t\tif closedCount == 1 {\n\t\t\thistoryLength = resp.Executions[0].HistoryLength\n\t\t\tbreak\n\t\t}\n\t\ts.Logger.Info(\"Closed WorkflowExecution is not yet visible\")\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.Equal(1, closedCount)\n\ts.Equal(int64(5), historyLength)\n\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err4 := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err4)\n\t\topenCount = len(resp.Executions)\n\t\tif openCount == 1 {\n\t\t\tbreak\n\t\t}\n\t\ts.Logger.Info(\"Open WorkflowExecution is not yet visible\")\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.Equal(1, openCount)\n}\n\nfunc (s *IntegrationSuite) TestChildWorkflowExecution() {\n\tparentID := \"integration-child-workflow-test-parent\"\n\tchildID := \"integration-child-workflow-test-child\"\n\twtParent := \"integration-child-workflow-test-parent-type\"\n\twtChild := \"integration-child-workflow-test-child-type\"\n\ttlParent := \"integration-child-workflow-test-parent-tasklist\"\n\ttlChild := \"integration-child-workflow-test-child-tasklist\"\n\tidentity := \"worker1\"\n\n\tparentWorkflowType := &types.WorkflowType{}\n\tparentWorkflowType.Name = wtParent\n\n\tchildWorkflowType := &types.WorkflowType{}\n\tchildWorkflowType.Name = wtChild\n\n\ttaskListParent := &types.TaskList{}\n\ttaskListParent.Name = tlParent\n\ttaskListChild := &types.TaskList{}\n\ttaskListChild.Name = tlChild\n\n\theader := &types.Header{\n\t\tFields: map[string][]byte{\"tracing\": []byte(\"sample payload\")},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          parentID,\n\t\tWorkflowType:                        parentWorkflowType,\n\t\tTaskList:                            taskListParent,\n\t\tInput:                               nil,\n\t\tHeader:                              header,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tchildComplete := false\n\tchildExecutionStarted := false\n\tvar startedEvent *types.HistoryEvent\n\tvar completedEvent *types.HistoryEvent\n\n\tmemoInfo, _ := json.Marshal(\"memo\")\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t\"Info\": memoInfo,\n\t\t},\n\t}\n\tattrValBytes, _ := json.Marshal(\"attrVal\")\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\t\"CustomKeywordField\": attrValBytes,\n\t\t},\n\t}\n\n\t// Parent Decider Logic\n\tdtHandlerParent := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\ts.Logger.Info(\"Processing decision task for \", tag.WorkflowID(execution.WorkflowID))\n\n\t\tif execution.WorkflowID == parentID {\n\t\t\tif !childExecutionStarted {\n\t\t\t\ts.Logger.Info(\"Starting child execution.\")\n\t\t\t\tchildExecutionStarted = true\n\n\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\t\t\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tWorkflowID:                          childID,\n\t\t\t\t\t\tWorkflowType:                        childWorkflowType,\n\t\t\t\t\t\tTaskList:                            taskListChild,\n\t\t\t\t\t\tInput:                               []byte(\"child-workflow-input\"),\n\t\t\t\t\t\tHeader:                              header,\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(200),\n\t\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\t\t\t\tControl:                             nil,\n\t\t\t\t\t\tMemo:                                memo,\n\t\t\t\t\t\tSearchAttributes:                    searchAttr,\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t} else if previousStartedEventID > 0 {\n\t\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\t\tif *event.EventType == types.EventTypeChildWorkflowExecutionStarted {\n\t\t\t\t\t\tstartedEvent = event\n\t\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t\t}\n\n\t\t\t\t\tif *event.EventType == types.EventTypeChildWorkflowExecutionCompleted {\n\t\t\t\t\t\tcompletedEvent = event\n\t\t\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}}, nil\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, nil, nil\n\t}\n\n\tvar childStartedEvent *types.HistoryEvent\n\t// Child Decider Logic\n\tdtHandlerChild := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif previousStartedEventID <= 0 {\n\t\t\tchildStartedEvent = history.Events[0]\n\t\t}\n\n\t\ts.Logger.Info(\"Processing decision task for Child \", tag.WorkflowID(execution.WorkflowID))\n\t\tchildComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Child Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpollerParent := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskListParent,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandlerParent,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tpollerChild := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskListChild,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandlerChild,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to start child execution\n\t_, err := pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(childExecutionStarted)\n\n\t// Process ChildExecution Started event and Process Child Execution and complete it\n\t_, err = pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t_, err = pollerChild.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(startedEvent)\n\ts.True(childComplete)\n\ts.NotNil(childStartedEvent)\n\ts.Equal(types.EventTypeWorkflowExecutionStarted, childStartedEvent.GetEventType())\n\ts.Equal(s.DomainName, childStartedEvent.WorkflowExecutionStartedEventAttributes.GetParentWorkflowDomain())\n\ts.Equal(parentID, childStartedEvent.WorkflowExecutionStartedEventAttributes.ParentWorkflowExecution.GetWorkflowID())\n\ts.Equal(we.GetRunID(), childStartedEvent.WorkflowExecutionStartedEventAttributes.ParentWorkflowExecution.GetRunID())\n\ts.Equal(startedEvent.ChildWorkflowExecutionStartedEventAttributes.GetInitiatedEventID(),\n\t\tchildStartedEvent.WorkflowExecutionStartedEventAttributes.GetParentInitiatedEventID())\n\ts.Equal(header, startedEvent.ChildWorkflowExecutionStartedEventAttributes.Header)\n\ts.Equal(header, childStartedEvent.WorkflowExecutionStartedEventAttributes.Header)\n\ts.Equal(memo, childStartedEvent.WorkflowExecutionStartedEventAttributes.GetMemo())\n\ts.Equal(searchAttr, childStartedEvent.WorkflowExecutionStartedEventAttributes.GetSearchAttributes())\n\n\t// Process ChildExecution completed event and complete parent execution\n\t_, err = pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(completedEvent)\n\tcompletedAttributes := completedEvent.ChildWorkflowExecutionCompletedEventAttributes\n\ts.Equal(s.DomainName, completedAttributes.Domain)\n\ts.Equal(childID, completedAttributes.WorkflowExecution.WorkflowID)\n\ts.Equal(wtChild, completedAttributes.WorkflowType.Name)\n\ts.Equal([]byte(\"Child Done.\"), completedAttributes.Result)\n}\n\nfunc (s *IntegrationSuite) TestCronChildWorkflowExecution() {\n\tparentID := \"integration-cron-child-workflow-test-parent\"\n\tchildID := \"integration-cron-child-workflow-test-child\"\n\twtParent := \"integration-cron-child-workflow-test-parent-type\"\n\twtChild := \"integration-cron-child-workflow-test-child-type\"\n\ttlParent := \"integration-cron-child-workflow-test-parent-tasklist\"\n\ttlChild := \"integration-cron-child-workflow-test-child-tasklist\"\n\tidentity := \"worker1\"\n\n\tcronSchedule := \"@every 3s\"\n\ttargetBackoffDuration := time.Second * 3\n\tbackoffDurationTolerance := time.Second\n\n\tparentWorkflowType := &types.WorkflowType{}\n\tparentWorkflowType.Name = wtParent\n\n\tchildWorkflowType := &types.WorkflowType{}\n\tchildWorkflowType.Name = wtChild\n\n\ttaskListParent := &types.TaskList{}\n\ttaskListParent.Name = tlParent\n\ttaskListChild := &types.TaskList{}\n\ttaskListChild.Name = tlChild\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          parentID,\n\t\tWorkflowType:                        parentWorkflowType,\n\t\tTaskList:                            taskListParent,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tstartParentWorkflowTS := time.Now()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tchildExecutionStarted := false\n\tvar terminatedEvent *types.HistoryEvent\n\tvar startChildWorkflowTS time.Time\n\t// Parent Decider Logic\n\tdtHandlerParent := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\ts.Logger.Info(\"Processing decision task for \", tag.WorkflowID(execution.WorkflowID))\n\n\t\tif !childExecutionStarted {\n\t\t\ts.Logger.Info(\"Starting child execution.\")\n\t\t\tchildExecutionStarted = true\n\t\t\tstartChildWorkflowTS = time.Now()\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\t\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowID:                          childID,\n\t\t\t\t\tWorkflowType:                        childWorkflowType,\n\t\t\t\t\tTaskList:                            taskListChild,\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(200),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\t\t\tControl:                             nil,\n\t\t\t\t\tCronSchedule:                        cronSchedule,\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\tif *event.EventType == types.EventTypeChildWorkflowExecutionTerminated {\n\t\t\t\tterminatedEvent = event\n\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t}\n\t\t}\n\t\treturn nil, nil, nil\n\t}\n\n\t// Child Decider Logic\n\tdtHandlerChild := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\ts.Logger.Info(\"Processing decision task for Child \", tag.WorkflowID(execution.WorkflowID))\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{},\n\t\t}}, nil\n\t}\n\n\tpollerParent := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskListParent,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandlerParent,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tpollerChild := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskListChild,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandlerChild,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to start child execution\n\t_, err := pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(childExecutionStarted)\n\n\t// Process ChildExecution Started event\n\t_, err = pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startChildWorkflowTS.UnixNano())\n\tfor i := 0; i < 2; i++ {\n\t\t// Sleep some time before checking the open executions.\n\t\t// This will not cost extra time as the polling for first decision task will be blocked for 3 seconds.\n\t\ttime.Sleep(2 * time.Second)\n\t\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: childID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\ts.Equal(1, len(resp.GetExecutions()))\n\n\t\t_, err = pollerChild.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\n\t\tbackoffDuration := time.Since(startChildWorkflowTS)\n\t\ts.True(backoffDuration < targetBackoffDuration+backoffDurationTolerance)\n\t\tstartChildWorkflowTS = time.Now()\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// terminate the childworkflow\n\tterminateErr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: childID,\n\t\t},\n\t})\n\ts.Nil(terminateErr)\n\n\t// Process ChildExecution terminated event and complete parent execution\n\t_, err = pollerParent.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(terminatedEvent)\n\tterminatedAttributes := terminatedEvent.ChildWorkflowExecutionTerminatedEventAttributes\n\ts.Equal(s.DomainName, terminatedAttributes.Domain)\n\ts.Equal(childID, terminatedAttributes.WorkflowExecution.WorkflowID)\n\ts.Equal(wtChild, terminatedAttributes.WorkflowType.Name)\n\n\tstartFilter.EarliestTime = common.Int64Ptr(startParentWorkflowTS.UnixNano())\n\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\tvar closedExecutions []*types.WorkflowExecutionInfo\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 4 {\n\t\t\tclosedExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(200 * time.Millisecond)\n\t}\n\ts.NotNil(closedExecutions)\n\tsort.Slice(closedExecutions, func(i, j int) bool {\n\t\treturn closedExecutions[i].GetStartTime() < closedExecutions[j].GetStartTime()\n\t})\n\t// The first parent is not the cron workflow, only verify child workflow with cron schedule\n\tlastExecution := closedExecutions[1]\n\tfor i := 2; i != 4; i++ {\n\t\texecutionInfo := closedExecutions[i]\n\t\t// Round up the time precision to seconds\n\t\texpectedBackoff := executionInfo.GetExecutionTime()/1000000000 - lastExecution.GetExecutionTime()/1000000000\n\t\t// The execution time calculate based on last execution close time\n\t\t// However, the current execution time is based on the current start time\n\t\t// This code is to remove the diff between current start time and last execution close time\n\t\t// TODO: Remove this line once we unify the time source.\n\t\texecutionTimeDiff := executionInfo.GetStartTime()/1000000000 - lastExecution.GetCloseTime()/1000000000\n\t\t// The backoff between any two executions should be multiplier of the target backoff duration which is 3 in this test\n\t\ts.Equal(int64(0), int64(expectedBackoff-executionTimeDiff)/1000000000%(targetBackoffDuration.Nanoseconds()/1000000000))\n\t\tlastExecution = executionInfo\n\t}\n}\n\nfunc (s *IntegrationSuite) TestWorkflowTimeout() {\n\tstartTime := time.Now().UnixNano()\n\n\tid := \"integration-workflow-timeout\"\n\twt := \"integration-workflow-timeout-type\"\n\ttl := \"integration-workflow-timeout-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\nGetHistoryLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionTimedOut {\n\t\t\ts.Logger.Warn(\"Execution not timedout yet.\")\n\t\t\ttime.Sleep(200 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\ttimeoutEventAttributes := lastEvent.WorkflowExecutionTimedOutEventAttributes\n\t\ts.Equal(types.TimeoutTypeStartToClose, *timeoutEventAttributes.TimeoutType)\n\t\tworkflowComplete = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(workflowComplete)\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startTime)\n\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\n\tclosedCount := 0\nListClosedLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err3 := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: startFilter,\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err3)\n\t\tclosedCount = len(resp.Executions)\n\t\tif closedCount == 0 {\n\t\t\ts.Logger.Info(\"Closed WorkflowExecution is not yet visibile\")\n\t\t\ttime.Sleep(1000 * time.Millisecond)\n\t\t\tcontinue ListClosedLoop\n\t\t}\n\t\tbreak ListClosedLoop\n\t}\n\ts.Equal(1, closedCount)\n}\n\nfunc (s *IntegrationSuite) TestDecisionTaskFailed() {\n\tid := \"integration-decisiontask-failed-test\"\n\twt := \"integration-decisiontask-failed-test-type\"\n\ttl := \"integration-decisiontask-failed-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tworkflowComplete := false\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tfailureCount := 10\n\tsignalCount := 0\n\tsendSignal := false\n\tlastDecisionTimestamp := int64(0)\n\t// var signalEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t// Count signals\n\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\tsignalCount++\n\t\t\t}\n\t\t}\n\t\t// Some signals received on this decision\n\t\tif signalCount == 1 {\n\t\t\treturn nil, []*types.Decision{}, nil\n\t\t}\n\n\t\t// Send signals during decision\n\t\tif sendSignal {\n\t\t\ts.sendSignal(s.DomainName, workflowExecution, \"signalC\", nil, identity)\n\t\t\ts.sendSignal(s.DomainName, workflowExecution, \"signalD\", nil, identity)\n\t\t\ts.sendSignal(s.DomainName, workflowExecution, \"signalE\", nil, identity)\n\t\t\tsendSignal = false\n\t\t}\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if failureCount > 0 {\n\t\t\t// Otherwise decrement failureCount and keep failing decisions\n\t\t\tfailureCount--\n\t\t\treturn nil, nil, errors.New(\"Decider Panic\")\n\t\t}\n\n\t\tworkflowComplete = true\n\t\ttime.Sleep(time.Second)\n\t\ts.Logger.Warn(fmt.Sprintf(\"PrevStarted: %v, StartedEventID: %v, Size: %v\", previousStartedEventID, startedEventID,\n\t\t\tlen(history.Events)))\n\t\tlastDecisionEvent := history.Events[startedEventID-1]\n\t\ts.Equal(types.EventTypeDecisionTaskStarted, lastDecisionEvent.GetEventType())\n\t\tlastDecisionTimestamp = lastDecisionEvent.GetTimestamp()\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// process activity\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"PollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// fail decision 5 times\n\tfor i := 0; i < 5; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, false, int64(i))\n\t\ts.Nil(err)\n\t}\n\n\terr = s.sendSignal(s.DomainName, workflowExecution, \"signalA\", nil, identity)\n\ts.Nil(err, \"failed to send signal to execution\")\n\n\t// process signal\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.Equal(1, signalCount)\n\n\t// send another signal to trigger decision\n\terr = s.sendSignal(s.DomainName, workflowExecution, \"signalB\", nil, identity)\n\ts.Nil(err, \"failed to send signal to execution\")\n\n\t// fail decision 2 more times\n\tfor i := 0; i < 2; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, false, int64(i))\n\t\ts.Nil(err)\n\t}\n\ts.Equal(3, signalCount)\n\n\t// now send a signal during failed decision\n\tsendSignal = true\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, false, int64(2))\n\ts.Nil(err)\n\ts.Equal(4, signalCount)\n\n\t// fail decision 1 more times\n\tfor i := 0; i < 2; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, false, int64(i))\n\t\ts.Nil(err)\n\t}\n\ts.Equal(12, signalCount)\n\n\t// Make complete workflow decision\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, int64(2))\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(workflowComplete)\n\ts.Equal(16, signalCount)\n\n\tevents := s.getHistory(s.DomainName, workflowExecution)\n\tvar lastEvent *types.HistoryEvent\n\tvar lastDecisionStartedEvent *types.HistoryEvent\n\tlastIdx := 0\n\tfor i, e := range events {\n\t\tif e.GetEventType() == types.EventTypeDecisionTaskStarted {\n\t\t\tlastDecisionStartedEvent = e\n\t\t\tlastIdx = i\n\t\t}\n\t\tlastEvent = e\n\t}\n\ts.Equal(types.EventTypeWorkflowExecutionCompleted, lastEvent.GetEventType())\n\ts.Logger.Info(fmt.Sprintf(\"Last Decision Time: %v, Last Decision History Timestamp: %v, Complete Timestamp: %v\",\n\t\ttime.Unix(0, lastDecisionTimestamp), time.Unix(0, lastDecisionStartedEvent.GetTimestamp()),\n\t\ttime.Unix(0, lastEvent.GetTimestamp())))\n\ts.Equal(lastDecisionTimestamp, lastDecisionStartedEvent.GetTimestamp())\n\ts.True(time.Duration(lastEvent.GetTimestamp()-lastDecisionTimestamp) >= time.Second)\n\n\ts.Equal(2, len(events)-lastIdx-1)\n\tdecisionCompletedEvent := events[lastIdx+1]\n\tworkflowCompletedEvent := events[lastIdx+2]\n\ts.Equal(types.EventTypeDecisionTaskCompleted, decisionCompletedEvent.GetEventType())\n\ts.Equal(types.EventTypeWorkflowExecutionCompleted, workflowCompletedEvent.GetEventType())\n}\n\nfunc (s *IntegrationSuite) TestGetPollerHistory() {\n\tWorkflowID := \"integration-get-poller-history\"\n\tworkflowTypeName := \"integration-get-poller-history-type\"\n\ttasklistName := \"integration-get-poller-history-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = workflowTypeName\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tasklistName\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          WorkflowID,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\t// var signalEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      taskList,\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(25),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(25),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// this function poll events from history side\n\ttestDescribeTaskList := func(domain string, tasklist *types.TaskList, tasklistType types.TaskListType) []*types.PollerInfo {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tlistResp, err := s.Engine.ListTaskListPartitions(ctx, &types.ListTaskListPartitionsRequest{\n\t\t\tDomain:   domain,\n\t\t\tTaskList: tasklist,\n\t\t})\n\t\ts.NoError(err)\n\t\tpollers := make(map[string]*types.PollerInfo)\n\t\tvar partitions []*types.TaskListPartitionMetadata\n\t\tif tasklistType == types.TaskListTypeActivity {\n\t\t\tpartitions = listResp.ActivityTaskListPartitions\n\t\t} else {\n\t\t\tpartitions = listResp.DecisionTaskListPartitions\n\t\t}\n\t\tfor _, partition := range partitions {\n\t\t\tresponseInner, errInner := s.Engine.DescribeTaskList(ctx, &types.DescribeTaskListRequest{\n\t\t\t\tDomain:       domain,\n\t\t\t\tTaskList:     &types.TaskList{Name: partition.Key, Kind: tasklist.Kind},\n\t\t\t\tTaskListType: &tasklistType,\n\t\t\t})\n\t\t\ts.Nil(errInner)\n\t\t\tfor _, poller := range responseInner.Pollers {\n\t\t\t\tpollers[poller.GetIdentity()] = poller\n\t\t\t}\n\t\t}\n\t\tvar results []*types.PollerInfo\n\t\tfor _, poller := range pollers {\n\t\t\tresults = append(results, poller)\n\t\t}\n\t\treturn results\n\t}\n\n\tbefore := time.Now()\n\n\t// when no one polling on the tasklist (activity or decition), there shall be no poller information\n\tpollerInfos := testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeActivity)\n\ts.Empty(pollerInfos)\n\tpollerInfos = testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeDecision)\n\ts.Empty(pollerInfos)\n\n\t_, errDecision := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(errDecision)\n\tpollerInfos = testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeActivity)\n\ts.Empty(pollerInfos)\n\tpollerInfos = testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeDecision)\n\ts.Equal(1, len(pollerInfos))\n\ts.Equal(identity, pollerInfos[0].GetIdentity())\n\ts.True(time.Unix(0, pollerInfos[0].GetLastAccessTime()).After(before))\n\ts.NotEmpty(pollerInfos[0].GetLastAccessTime())\n\n\terrActivity := poller.PollAndProcessActivityTask(false)\n\ts.Nil(errActivity)\n\tpollerInfos = testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeActivity)\n\ts.Equal(1, len(pollerInfos))\n\ts.Equal(identity, pollerInfos[0].GetIdentity())\n\ts.True(time.Unix(0, pollerInfos[0].GetLastAccessTime()).After(before))\n\ts.NotEmpty(pollerInfos[0].GetLastAccessTime())\n\tpollerInfos = testDescribeTaskList(s.DomainName, taskList, types.TaskListTypeDecision)\n\ts.Equal(1, len(pollerInfos))\n\ts.Equal(identity, pollerInfos[0].GetIdentity())\n\ts.True(time.Unix(0, pollerInfos[0].GetLastAccessTime()).After(before))\n\ts.NotEmpty(pollerInfos[0].GetLastAccessTime())\n}\n\nfunc (s *IntegrationSuite) TestTransientDecisionTimeout() {\n\tid := \"integration-transient-decision-timeout-test\"\n\twt := \"integration-transient-decision-timeout-test-type\"\n\ttl := \"integration-transient-decision-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tworkflowComplete := false\n\tfailDecision := true\n\tsignalCount := 0\n\t// var signalEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif failDecision {\n\t\t\tfailDecision = false\n\t\t\treturn nil, nil, errors.New(\"Decider Panic\")\n\t\t}\n\n\t\t// Count signals\n\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\tsignalCount++\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// First decision immediately fails and schedules a transient decision\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Now send a signal when transient decision is scheduled\n\terr = s.sendSignal(s.DomainName, workflowExecution, \"signalA\", nil, identity)\n\ts.Nil(err, \"failed to send signal to execution\")\n\n\t// Drop decision task to cause a Decision Timeout\n\t_, err = poller.PollAndProcessDecisionTask(false, true)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Now process signal and complete workflow execution\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, int64(1))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.Equal(1, signalCount)\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestNoTransientDecisionAfterFlushBufferedEvents() {\n\tid := \"integration-no-transient-decision-after-flush-buffered-events-test\"\n\twt := \"integration-no-transient-decision-after-flush-buffered-events-test-type\"\n\ttl := \"integration-no-transient-decision-after-flush-buffered-events-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(20),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tworkflowComplete := false\n\tcontinueAsNewAndSignal := false\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !continueAsNewAndSignal {\n\t\t\tcontinueAsNewAndSignal = true\n\t\t\tctx, cancel := createContext()\n\t\t\tdefer cancel()\n\t\t\t// this will create new event when there is in-flight decision task, and the new event will be buffered\n\t\t\terr := s.Engine.SignalWorkflowExecution(ctx,\n\t\t\t\t&types.SignalWorkflowExecutionRequest{\n\t\t\t\t\tDomain: s.DomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\t},\n\t\t\t\t\tSignalName: \"buffered-signal-1\",\n\t\t\t\t\tInput:      []byte(\"buffered-signal-input\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t})\n\t\t\ts.NoError(err)\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeContinueAsNewWorkflowExecution.Ptr(),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\t\tTaskList:                            taskList,\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// fist decision, this try to do a continue as new but there is a buffered event,\n\t// so it will fail and create a new decision\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// second decision, which will complete the workflow\n\t// this expect the decision to have attempt == 0\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, 0)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestRelayDecisionTimeout() {\n\tid := \"integration-relay-decision-timeout-test\"\n\twt := \"integration-relay-decision-timeout-test-type\"\n\ttl := \"integration-relay-decision-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\tworkflowComplete, isFirst := false, true\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif isFirst {\n\t\t\tisFirst = false\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"test-marker\",\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// First decision task complete with a marker decision, and request to relay decision (immediately return a new decision task)\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\tfalse,\n\t\tfalse,\n\t\t0,\n\t\t3,\n\t\ttrue,\n\t\tnil)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\n\ttime.Sleep(time.Second * 2) // wait 2s for relay decision to timeout\n\tdecisionTaskTimeout := false\n\tfor i := 0; i < 3; i++ {\n\t\tevents := s.getHistory(s.DomainName, workflowExecution)\n\t\tif len(events) >= 8 {\n\t\t\ts.Equal(types.EventTypeDecisionTaskTimedOut, events[7].GetEventType())\n\t\t\ts.Equal(types.TimeoutTypeStartToClose, events[7].DecisionTaskTimedOutEventAttributes.GetTimeoutType())\n\t\t\tdecisionTaskTimeout = true\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(time.Second)\n\t}\n\t// verify relay decision task timeout\n\ts.True(decisionTaskTimeout)\n\n\t// Now complete workflow\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, int64(1))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n}\n\nfunc (s *IntegrationSuite) TestTaskProcessingProtectionForRateLimitError() {\n\tid := \"integration-task-processing-protection-for-rate-limit-error-test\"\n\twt := \"integration-task-processing-protection-for-rate-limit-error-test-type\"\n\ttl := \"integration-task-processing-protection-for-rate-limit-error-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(601),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(600),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tworkflowComplete := false\n\tsignalCount := 0\n\tcreateUserTimer := false\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, h *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !createUserTimer {\n\t\t\tcreateUserTimer = true\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\t\t\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\t\t\tTimerID:                   \"timer-id-1\",\n\t\t\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\t// Count signals\n\t\tfor _, event := range h.Events[previousStartedEventID:] {\n\t\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\tsignalCount++\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Process first decision to create user timer\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Send one signal to create a new decision\n\tbuf := new(bytes.Buffer)\n\tbinary.Write(buf, binary.LittleEndian, int64(0))\n\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"SignalName\", buf.Bytes(), identity))\n\n\t// Drop decision to cause all events to be buffered from now on\n\t_, err = poller.PollAndProcessDecisionTask(false, true)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Buffered 100 Signals\n\tfor i := 1; i < 101; i++ {\n\t\tbuf := new(bytes.Buffer)\n\t\tbinary.Write(buf, binary.LittleEndian, int64(i))\n\t\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"SignalName\", buf.Bytes(), identity))\n\t}\n\n\t// 101 signal, which will fail the decision\n\tbuf = new(bytes.Buffer)\n\tbinary.Write(buf, binary.LittleEndian, int64(101))\n\tsignalErr := s.sendSignal(s.DomainName, workflowExecution, \"SignalName\", buf.Bytes(), identity)\n\ts.Nil(signalErr)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, false, 0)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n\ts.Equal(102, signalCount)\n}\n\nfunc (s *IntegrationSuite) TestStickyTimeout_NonTransientDecision() {\n\tid := \"integration-sticky-timeout-non-transient-decision\"\n\twt := \"integration-sticky-timeout-non-transient-decision-type\"\n\ttl := \"integration-sticky-timeout-non-transient-decision-tasklist\"\n\tstl := \"integration-sticky-timeout-non-transient-decision-tasklist-sticky\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stl\n\tstickyScheduleToStartTimeoutSeconds := common.Int32Ptr(2)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tlocalActivityDone := false\n\tfailureCount := 5\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !localActivityDone {\n\t\t\tlocalActivityDone = true\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"local activity marker\",\n\t\t\t\t\tDetails:    []byte(\"local activity data\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif failureCount > 0 {\n\t\t\t// send a signal on third failure to be buffered, forcing a non-transient decision when buffer is flushed\n\t\t\t/*if failureCount == 3 {\n\t\t\t\terr := s.Engine.SignalWorkflowExecution(createContext(), &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\tDomain:            s.DomainName,\n\t\t\t\t\tWorkflowExecution: workflowExecution,\n\t\t\t\t\tSignalName:        common.StringPtr(\"signalB\"),\n\t\t\t\t\tInput:             []byte(\"signal input\"),\n\t\t\t\t\tIdentity:          identity,\n\t\t\t\t\tRequestID:         uuid.New(),\n\t\t\t\t})\n\t\t\t\ts.Nil(err)\n\t\t\t}*/\n\t\t\tfailureCount--\n\t\t\treturn nil, nil, errors.New(\"non deterministic error\")\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:                              s.Engine,\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            taskList,\n\t\tIdentity:                            identity,\n\t\tDecisionHandler:                     dtHandler,\n\t\tLogger:                              s.Logger,\n\t\tT:                                   s.T(),\n\t\tStickyTaskList:                      stickyTaskList,\n\t\tStickyScheduleToStartTimeoutSeconds: stickyScheduleToStartTimeoutSeconds,\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, true, int64(0))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: workflowExecution,\n\t\tSignalName:        \"signalA\",\n\t\tInput:             []byte(\"signal input\"),\n\t\tIdentity:          identity,\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.NoError(err)\n\n\t// Wait for decision timeout\n\tstickyTimeout := false\nWaitForStickyTimeoutLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tevents := s.getHistory(s.DomainName, workflowExecution)\n\t\tfor _, event := range events {\n\t\t\tif event.GetEventType() == types.EventTypeDecisionTaskTimedOut {\n\t\t\t\ts.Equal(types.TimeoutTypeScheduleToStart, event.DecisionTaskTimedOutEventAttributes.GetTimeoutType())\n\t\t\t\tstickyTimeout = true\n\t\t\t\tbreak WaitForStickyTimeoutLoop\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(time.Second)\n\t}\n\ts.True(stickyTimeout, \"Decision not timed out.\")\n\n\tfor i := 0; i < 3; i++ {\n\t\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(i))\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: workflowExecution,\n\t\tSignalName:        \"signalB\",\n\t\tInput:             []byte(\"signal input\"),\n\t\tIdentity:          identity,\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err)\n\n\tfor i := 0; i < 2; i++ {\n\t\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(i))\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\tdecisionTaskFailed := false\n\tevents := s.getHistory(s.DomainName, workflowExecution)\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskFailed {\n\t\t\tdecisionTaskFailed = true\n\t\t\tbreak\n\t\t}\n\t}\n\ts.True(decisionTaskFailed)\n\n\t// Complete workflow execution\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(2))\n\ts.NoError(err)\n\n\t// Assert for single decision task failed and workflow completion\n\tfailedDecisions := 0\n\tworkflowComplete := false\n\tevents = s.getHistory(s.DomainName, workflowExecution)\n\tfor _, event := range events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeDecisionTaskFailed:\n\t\t\tfailedDecisions++\n\t\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\t\tworkflowComplete = true\n\t\t}\n\t}\n\ts.True(workflowComplete, \"Workflow not complete\")\n\ts.Equal(2, failedDecisions, \"Mismatched failed decision count\")\n}\n\nfunc (s *IntegrationSuite) TestStickyTasklistResetThenTimeout() {\n\tid := \"integration-reset-sticky-fire-schedule-to-start-timeout\"\n\twt := \"integration-reset-sticky-fire-schedule-to-start-timeout-type\"\n\ttl := \"integration-reset-sticky-fire-schedule-to-start-timeout-tasklist\"\n\tstl := \"integration-reset-sticky-fire-schedule-to-start-timeout-tasklist-sticky\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stl\n\tstickyScheduleToStartTimeoutSeconds := common.Int32Ptr(2)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tlocalActivityDone := false\n\tfailureCount := 5\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !localActivityDone {\n\t\t\tlocalActivityDone = true\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"local activity marker\",\n\t\t\t\t\tDetails:    []byte(\"local activity data\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif failureCount > 0 {\n\t\t\tfailureCount--\n\t\t\treturn nil, nil, errors.New(\"non deterministic error\")\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:                              s.Engine,\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            taskList,\n\t\tIdentity:                            identity,\n\t\tDecisionHandler:                     dtHandler,\n\t\tLogger:                              s.Logger,\n\t\tT:                                   s.T(),\n\t\tStickyTaskList:                      stickyTaskList,\n\t\tStickyScheduleToStartTimeoutSeconds: stickyScheduleToStartTimeoutSeconds,\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, true, int64(0))\n\ts.Logger.Info(\"PollAndProcessDecisionTask: %v\", tag.Error(err))\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: workflowExecution,\n\t\tSignalName:        \"signalA\",\n\t\tInput:             []byte(\"signal input\"),\n\t\tIdentity:          identity,\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.NoError(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Reset sticky tasklist before sticky decision task starts\n\ts.Engine.ResetStickyTaskList(ctx, &types.ResetStickyTaskListRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: workflowExecution,\n\t})\n\n\t// Wait for decision timeout\n\tstickyTimeout := false\nWaitForStickyTimeoutLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tevents := s.getHistory(s.DomainName, workflowExecution)\n\t\tfor _, event := range events {\n\t\t\tif event.GetEventType() == types.EventTypeDecisionTaskTimedOut {\n\t\t\t\ts.Equal(types.TimeoutTypeScheduleToStart, event.DecisionTaskTimedOutEventAttributes.GetTimeoutType())\n\t\t\t\tstickyTimeout = true\n\t\t\t\tbreak WaitForStickyTimeoutLoop\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(time.Second)\n\t}\n\ts.True(stickyTimeout, \"Decision not timed out.\")\n\n\tfor i := 0; i < 3; i++ {\n\t\t// we will jump from sticky decision to transient decision in this case (decisionAttempt increased)\n\t\t// even though the error is sticky scheduleToStart timeout\n\t\t// since we can't tell if the decision is a sticky decision or not when the timer fires\n\t\t// (stickiness cleared by the ResetStickyTaskList API call above)\n\t\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(i+1))\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask: %v\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            s.DomainName,\n\t\tWorkflowExecution: workflowExecution,\n\t\tSignalName:        \"signalB\",\n\t\tInput:             []byte(\"signal input\"),\n\t\tIdentity:          identity,\n\t\tRequestID:         uuid.New(),\n\t})\n\ts.Nil(err)\n\n\tfor i := 0; i < 2; i++ {\n\t\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(i))\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask: %v\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\tdecisionTaskFailed := false\n\tevents := s.getHistory(s.DomainName, workflowExecution)\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskFailed {\n\t\t\tdecisionTaskFailed = true\n\t\t\tbreak\n\t\t}\n\t}\n\ts.True(decisionTaskFailed)\n\n\t// Complete workflow execution\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(true, false, false, true, int64(2))\n\ts.NoError(err)\n\n\t// Assert for single decision task failed and workflow completion\n\tfailedDecisions := 0\n\tworkflowComplete := false\n\tevents = s.getHistory(s.DomainName, workflowExecution)\n\tfor _, event := range events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeDecisionTaskFailed:\n\t\t\tfailedDecisions++\n\t\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\t\tworkflowComplete = true\n\t\t}\n\t}\n\ts.True(workflowComplete, \"Workflow not complete\")\n\ts.Equal(1, failedDecisions, \"Mismatched failed decision count\")\n}\n\nfunc (s *IntegrationSuite) TestBufferedEventsOutOfOrder() {\n\tid := \"integration-buffered-events-out-of-order-test\"\n\twt := \"integration-buffered-events-out-of-order-test-type\"\n\ttl := \"integration-buffered-events-out-of-order-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(20),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\t// decider logic\n\tworkflowComplete := false\n\tfirstDecision := false\n\tsecondDecision := false\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\ts.Logger.Info(fmt.Sprintf(\"Decider called: first: %v, second: %v, complete: %v\\n\", firstDecision, secondDecision, workflowComplete))\n\n\t\tif !firstDecision {\n\t\t\tfirstDecision = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"some random marker name\",\n\t\t\t\t\tDetails:    []byte(\"some random marker details\"),\n\t\t\t\t},\n\t\t\t}, {\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"Activity-1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"ActivityType\"},\n\t\t\t\t\tDomain:                        s.DomainName,\n\t\t\t\t\tTaskList:                      taskList,\n\t\t\t\t\tInput:                         []byte(\"some random activity input\"),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(100),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tif !secondDecision {\n\t\t\tsecondDecision = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\t\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\t\t\tMarkerName: \"some random marker name\",\n\t\t\t\t\tDetails:    []byte(\"some random marker details\"),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// first decision, which will schedule an activity and add marker\n\t_, task, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\ttrue,\n\t\tfalse,\n\t\tfalse,\n\t\tfalse,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// This will cause activity start and complete to be buffered\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"pollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// second decision, completes another local activity and forces flush of buffered activity events\n\tnewDecisionTask := task.GetDecisionTask()\n\ts.NotNil(newDecisionTask)\n\ttask, err = poller.HandlePartialDecision(newDecisionTask)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.NotNil(task)\n\n\t// third decision, which will close workflow\n\tnewDecisionTask = task.GetDecisionTask()\n\ts.NotNil(newDecisionTask)\n\ttask, err = poller.HandlePartialDecision(newDecisionTask)\n\ts.Logger.Info(\"pollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.Nil(task.DecisionTask)\n\n\tevents := s.getHistory(s.DomainName, workflowExecution)\n\tvar scheduleEvent, startedEvent, completedEvent *types.HistoryEvent\n\tfor _, event := range events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeActivityTaskScheduled:\n\t\t\tscheduleEvent = event\n\t\tcase types.EventTypeActivityTaskStarted:\n\t\t\tstartedEvent = event\n\t\tcase types.EventTypeActivityTaskCompleted:\n\t\t\tcompletedEvent = event\n\t\t}\n\t}\n\n\ts.NotNil(scheduleEvent)\n\ts.NotNil(startedEvent)\n\ts.NotNil(completedEvent)\n\ts.True(startedEvent.ID < completedEvent.ID)\n\ts.Equal(scheduleEvent.ID, startedEvent.ActivityTaskStartedEventAttributes.GetScheduledEventID())\n\ts.Equal(scheduleEvent.ID, completedEvent.ActivityTaskCompletedEventAttributes.GetScheduledEventID())\n\ts.Equal(startedEvent.ID, completedEvent.ActivityTaskCompletedEventAttributes.GetStartedEventID())\n\ts.True(workflowComplete)\n}\n\ntype startFunc func() (*types.StartWorkflowExecutionResponse, error)\n\nfunc (s *IntegrationSuite) TestStartWithMemo() {\n\tid := \"integration-start-with-memo-test\"\n\twt := \"integration-start-with-memo-test-type\"\n\ttl := \"integration-start-with-memo-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tmemoInfo, _ := json.Marshal(id)\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t\"Info\": memoInfo,\n\t\t},\n\t}\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tMemo:                                memo,\n\t}\n\n\tfn := func() (*types.StartWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.StartWorkflowExecution(ctx, request)\n\t}\n\ts.startWithMemoHelper(fn, id, taskList, memo)\n}\n\nfunc (s *IntegrationSuite) TestSignalWithStartWithMemo() {\n\tid := \"integration-signal-with-start-with-memo-test\"\n\twt := \"integration-signal-with-start-with-memo-test-type\"\n\ttl := \"integration-signal-with-start-with-memo-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tmemoInfo, _ := json.Marshal(id)\n\tmemo := &types.Memo{\n\t\tFields: map[string][]byte{\n\t\t\t\"Info\": memoInfo,\n\t\t},\n\t}\n\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\trequest := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tSignalName:                          signalName,\n\t\tSignalInput:                         signalInput,\n\t\tIdentity:                            identity,\n\t\tMemo:                                memo,\n\t}\n\n\tfn := func() (*types.StartWorkflowExecutionResponse, error) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.SignalWithStartWorkflowExecution(ctx, request)\n\t}\n\ts.startWithMemoHelper(fn, id, taskList, memo)\n}\n\nfunc (s *IntegrationSuite) TestCancelTimer() {\n\tid := \"integration-cancel-timer-test\"\n\twt := \"integration-cancel-timer-test-type\"\n\ttl := \"integration-cancel-timer-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tcreatResp, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      creatResp.GetRunID(),\n\t}\n\n\tTimerID := \"1\"\n\ttimerScheduled := false\n\tsignalDelivered := false\n\ttimerCancelled := false\n\tworkflowComplete := false\n\ttimer := int64(2000)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !timerScheduled {\n\t\t\ttimerScheduled = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\t\t\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\t\t\tTimerID:                   TimerID,\n\t\t\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(timer),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tresp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tExecution:       workflowExecution,\n\t\t\tMaximumPageSize: 200,\n\t\t})\n\t\ts.Nil(err)\n\t\tfor _, event := range resp.History.Events {\n\t\t\tswitch event.GetEventType() {\n\t\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\t\tsignalDelivered = true\n\t\t\tcase types.EventTypeTimerCanceled:\n\t\t\t\ttimerCancelled = true\n\t\t\t}\n\t\t}\n\n\t\tif !signalDelivered {\n\t\t\ts.Fail(\"should receive a signal\")\n\t\t}\n\n\t\tif !timerCancelled {\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCancelTimer.Ptr(),\n\t\t\t\tCancelTimerDecisionAttributes: &types.CancelTimerDecisionAttributes{\n\t\t\t\t\tTimerID: TimerID,\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// schedule the timer\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"random signal name\", []byte(\"random signal payload\"), identity))\n\n\t// receive the signal & cancel the timer\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"random signal name\", []byte(\"random signal payload\"), identity))\n\t// complete the workflow\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:          s.DomainName,\n\t\tExecution:       workflowExecution,\n\t\tMaximumPageSize: 200,\n\t})\n\ts.Nil(err)\n\tfor _, event := range resp.History.Events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\tsignalDelivered = true\n\t\tcase types.EventTypeTimerCanceled:\n\t\t\ttimerCancelled = true\n\t\tcase types.EventTypeTimerFired:\n\t\t\ts.Fail(\"timer got fired\")\n\t\t}\n\t}\n}\n\nfunc (s *IntegrationSuite) TestCancelTimer_CancelFiredAndBuffered() {\n\tid := \"integration-cancel-timer-fired-and-buffered-test\"\n\twt := \"integration-cancel-timer-fired-and-buffered-test-type\"\n\ttl := \"integration-cancel-timer-fired-and-buffered-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tcreatResp, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      creatResp.GetRunID(),\n\t}\n\n\tTimerID := 1\n\ttimerScheduled := false\n\tsignalDelivered := false\n\ttimerCancelled := false\n\tworkflowComplete := false\n\ttimer := int64(4)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !timerScheduled {\n\t\t\ttimerScheduled = true\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\t\t\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\t\t\tTimerID:                   fmt.Sprintf(\"%v\", TimerID),\n\t\t\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(timer),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tresp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tExecution:       workflowExecution,\n\t\t\tMaximumPageSize: 200,\n\t\t})\n\t\ts.Nil(err)\n\t\tfor _, event := range resp.History.Events {\n\t\t\tswitch event.GetEventType() {\n\t\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\t\tsignalDelivered = true\n\t\t\tcase types.EventTypeTimerCanceled:\n\t\t\t\ttimerCancelled = true\n\t\t\t}\n\t\t}\n\n\t\tif !signalDelivered {\n\t\t\ts.Fail(\"should receive a signal\")\n\t\t}\n\n\t\tif !timerCancelled {\n\t\t\ttime.Sleep(time.Duration(2*timer) * time.Second)\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCancelTimer.Ptr(),\n\t\t\t\tCancelTimerDecisionAttributes: &types.CancelTimerDecisionAttributes{\n\t\t\t\t\tTimerID: fmt.Sprintf(\"%v\", TimerID),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: nil,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// schedule the timer\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"random signal name\", []byte(\"random signal payload\"), identity))\n\n\t// receive the signal & cancel the timer\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.Nil(s.sendSignal(s.DomainName, workflowExecution, \"random signal name\", []byte(\"random signal payload\"), identity))\n\t// complete the workflow\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: completed\")\n\ts.Nil(err)\n\n\ts.True(workflowComplete)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:          s.DomainName,\n\t\tExecution:       workflowExecution,\n\t\tMaximumPageSize: 200,\n\t})\n\ts.Nil(err)\n\tfor _, event := range resp.History.Events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\tsignalDelivered = true\n\t\tcase types.EventTypeTimerCanceled:\n\t\t\ttimerCancelled = true\n\t\tcase types.EventTypeTimerFired:\n\t\t\ts.Fail(\"timer got fired\")\n\t\t}\n\t}\n}\n\n// helper function for TestStartWithMemo and TestSignalWithStartWithMemo to reduce duplicate code\nfunc (s *IntegrationSuite) startWithMemoHelper(startFn startFunc, id string, taskList *types.TaskList, memo *types.Memo) {\n\tidentity := \"worker1\"\n\n\twe, err0 := startFn()\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution: response\", tag.WorkflowRunID(we.RunID))\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\treturn []byte(strconv.Itoa(1)), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// verify open visibility\n\tvar openExecutionInfo *types.WorkflowExecutionInfo\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err1 := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\tif len(resp.Executions) == 1 {\n\t\t\topenExecutionInfo = resp.Executions[0]\n\t\t\tbreak\n\t\t}\n\t\ts.Logger.Info(\"Open WorkflowExecution is not yet visible\")\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.NotNil(openExecutionInfo)\n\ts.Equal(memo, openExecutionInfo.Memo)\n\n\t// make progress of workflow\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask: %v\", tag.Error(err))\n\ts.Nil(err)\n\n\t// verify history\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, historyErr := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: execution,\n\t})\n\ts.Nil(historyErr)\n\thistory := historyResponse.History\n\tfirstEvent := history.Events[0]\n\ts.Equal(types.EventTypeWorkflowExecutionStarted, firstEvent.GetEventType())\n\tstartdEventAttributes := firstEvent.WorkflowExecutionStartedEventAttributes\n\ts.Equal(memo, startdEventAttributes.Memo)\n\n\t// verify DescribeWorkflowExecution result\n\tdescRequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: execution,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, descRequest)\n\ts.Nil(err)\n\ts.Equal(memo, descResp.WorkflowExecutionInfo.Memo)\n\n\t// verify closed visibility\n\tvar closedExecutionInfo *types.WorkflowExecutionInfo\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err1 := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\tif len(resp.Executions) == 1 {\n\t\t\tclosedExecutionInfo = resp.Executions[0]\n\t\t\tbreak\n\t\t}\n\t\ts.Logger.Info(\"Closed WorkflowExecution is not yet visible\")\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.NotNil(closedExecutionInfo)\n\ts.Equal(memo, closedExecutionInfo.Memo)\n}\n\nfunc (s *IntegrationSuite) sendSignal(domainName string, execution *types.WorkflowExecution, signalName string,\n\tinput []byte, identity string) error {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\treturn s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain:            domainName,\n\t\tWorkflowExecution: execution,\n\t\tSignalName:        signalName,\n\t\tInput:             input,\n\t\tIdentity:          identity,\n\t})\n}\n\n// TestDescribeCluster tests that DescribeCluster API returns a valid response without error\nfunc (s *IntegrationSuite) TestDescribeCluster() {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\n\tresponse, err := s.AdminClient.DescribeCluster(ctx)\n\ts.Require().NoError(err)\n\ts.Require().NotNil(response)\n}\n"
  },
  {
    "path": "host/integration_test_cron.go",
    "content": "package host\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestCronWorkflowWithOverlapPolicy() {\n\tid := \"integration-wf-cron-overlap-test\"\n\twt := \"integration-wf-cron-overlap-type\"\n\ttl := \"integration-wf-cron-overlap-tasklist\"\n\tidentity := \"worker1\"\n\tcronSchedule := \"@every 5s\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Test both overlap policies\n\ttestCases := []struct {\n\t\tname              string\n\t\tcronOverlapPolicy types.CronOverlapPolicy\n\t}{\n\t\t{\n\t\t\tname:              \"SkippedPolicy\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicySkipped,\n\t\t},\n\t\t{\n\t\t\tname:              \"BufferOnePolicy\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicyBufferOne,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\tworkflowID := id + \"-\" + tc.name\n\n\t\t\trequest := &types.StartWorkflowExecutionRequest{\n\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\tDomain:                              s.DomainName,\n\t\t\t\tWorkflowID:                          workflowID,\n\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\tTaskList:                            taskList,\n\t\t\t\tInput:                               nil,\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\tIdentity:                            identity,\n\t\t\t\tCronSchedule:                        cronSchedule,\n\t\t\t\tCronOverlapPolicy:                   &tc.cronOverlapPolicy,\n\t\t\t}\n\n\t\t\tctx, cancel := createContext()\n\t\t\tdefer cancel()\n\t\t\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\t\t\ts.Nil(err)\n\n\t\t\t// Logging for debug\n\t\t\tfmt.Printf(\"StartWorkflowExecution RunID: %s\\n\", we.RunID)\n\n\t\t\tvar executions []*types.WorkflowExecution\n\t\t\tattemptCount := 0\n\t\t\texecutionTimes := make([]time.Time, 0)\n\n\t\t\t// Decision task handler that simulates long-running workflow\n\t\t\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\t\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t\texecutions = append(executions, execution)\n\t\t\t\tattemptCount++\n\t\t\t\texecutionTimes = append(executionTimes, time.Now())\n\n\t\t\t\tif attemptCount == 1 {\n\t\t\t\t\treturn nil, []*types.Decision{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\t\tResult: []byte(fmt.Sprintf(\"execution-%d-complete\", attemptCount)),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil\n\t\t\t\t}\n\n\t\t\t\treturn nil, []*types.Decision{\n\t\t\t\t\t{\n\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tResult: []byte(fmt.Sprintf(\"execution-%d-complete\", attemptCount)),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t}\n\n\t\t\tpoller := &TaskPoller{\n\t\t\t\tEngine:          s.Engine,\n\t\t\t\tDomain:          s.DomainName,\n\t\t\t\tTaskList:        taskList,\n\t\t\t\tIdentity:        identity,\n\t\t\t\tDecisionHandler: dtHandler,\n\t\t\t\tLogger:          s.Logger,\n\t\t\t\tT:               s.T(),\n\t\t\t}\n\n\t\t\tfor i := 0; i < 3; i++ {\n\t\t\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\t\t\ts.True(err == nil, err)\n\t\t\t\ttime.Sleep(500 * time.Millisecond)\n\t\t\t}\n\n\t\t\ts.True(attemptCount >= 2, \"Expected at least 2 executions, got %d\", attemptCount)\n\n\t\t\tif len(executionTimes) >= 2 {\n\t\t\t\tfor i := 1; i < len(executionTimes); i++ {\n\t\t\t\t\ttimeDiff := executionTimes[i].Sub(executionTimes[i-1])\n\n\t\t\t\t\tif tc.cronOverlapPolicy == types.CronOverlapPolicySkipped {\n\t\t\t\t\t\ts.True(timeDiff >= 1500*time.Millisecond,\n\t\t\t\t\t\t\t\"Expected execution %d to be spaced by at least 1.5s from previous, got %v\",\n\t\t\t\t\t\t\ti, timeDiff)\n\t\t\t\t\t} else {\n\t\t\t\t\t\ts.True(timeDiff >= 0, \"Expected non-negative time difference, got %v\", timeDiff)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif len(executions) > 0 {\n\t\t\t\tevents := s.getHistory(s.DomainName, executions[0])\n\t\t\t\ts.True(len(events) > 0, \"Expected workflow history to have events\")\n\t\t\t\tlastEvent := events[len(events)-1]\n\t\t\t\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\t\t\t\tattributes := lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\t\t\t\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\t\t\t}\n\n\t\t\tctx, cancel = createContext()\n\t\t\tdefer cancel()\n\t\t\tterminateErr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain: s.DomainName,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t},\n\t\t\t})\n\t\t\ts.NoError(terminateErr)\n\n\t\t\tctx, cancel = createContext()\n\t\t\tdefer cancel()\n\t\t\tdweResponse, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: s.DomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\tRunID:      we.RunID,\n\t\t\t\t},\n\t\t\t})\n\t\t\ts.Nil(err)\n\t\t\t// Verify that the workflow execution info is returned after termination\n\t\t\ts.NotNil(dweResponse.WorkflowExecutionInfo)\n\t\t\ts.True(dweResponse.WorkflowExecutionInfo.IsCron, \"Expected workflow to be marked as cron\")\n\t\t\ts.Equal(tc.cronOverlapPolicy, *dweResponse.WorkflowExecutionInfo.CronOverlapPolicy)\n\t\t})\n\t}\n}\n\nfunc (s *IntegrationSuite) TestCronWorkflowOverlapBehavior() {\n\tid := \"integration-wf-cron-overlap-behavior-test\"\n\twt := \"integration-wf-cron-overlap-behavior-type\"\n\ttl := \"integration-wf-cron-overlap-behavior-tasklist\"\n\tidentity := \"worker1\"\n\tcronSchedule := \"@every 1s\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Test both overlap policies with realistic overlap scenarios\n\ttestCases := []struct {\n\t\tname              string\n\t\tcronOverlapPolicy types.CronOverlapPolicy\n\t\tworkflowDuration  time.Duration\n\t}{\n\t\t{\n\t\t\tname:              \"SkippedPolicy_ShortWorkflow\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicySkipped,\n\t\t\tworkflowDuration:  500 * time.Millisecond,\n\t\t},\n\t\t{\n\t\t\tname:              \"SkippedPolicy_LongWorkflow\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicySkipped,\n\t\t\tworkflowDuration:  2 * time.Second,\n\t\t},\n\t\t{\n\t\t\tname:              \"BufferOnePolicy_ShortWorkflow\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicyBufferOne,\n\t\t\tworkflowDuration:  500 * time.Millisecond,\n\t\t},\n\t\t{\n\t\t\tname:              \"BufferOnePolicy_LongWorkflow\",\n\t\t\tcronOverlapPolicy: types.CronOverlapPolicyBufferOne,\n\t\t\tworkflowDuration:  2 * time.Second,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\tworkflowID := id + \"-\" + tc.name\n\n\t\t\trequest := &types.StartWorkflowExecutionRequest{\n\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\tDomain:                              s.DomainName,\n\t\t\t\tWorkflowID:                          workflowID,\n\t\t\t\tWorkflowType:                        workflowType,\n\t\t\t\tTaskList:                            taskList,\n\t\t\t\tInput:                               nil,\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\tIdentity:                            identity,\n\t\t\t\tCronSchedule:                        cronSchedule,\n\t\t\t\tCronOverlapPolicy:                   &tc.cronOverlapPolicy,\n\t\t\t}\n\n\t\t\tctx, cancel := createContext()\n\t\t\tdefer cancel()\n\t\t\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\t\t\ts.Nil(err)\n\n\t\t\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t\t\tvar executions []*types.WorkflowExecution\n\t\t\tattemptCount := 0\n\t\t\texecutionStartTimes := make([]time.Time, 0)\n\t\t\texecutionEndTimes := make([]time.Time, 0)\n\n\t\t\t// Decision task handler that simulates workflows with controlled duration\n\t\t\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\t\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t\texecutions = append(executions, execution)\n\t\t\t\tattemptCount++\n\t\t\t\tstartTime := time.Now()\n\t\t\t\texecutionStartTimes = append(executionStartTimes, startTime)\n\n\t\t\t\t// Simulate workflow duration by sleeping\n\t\t\t\ttime.Sleep(tc.workflowDuration)\n\t\t\t\tendTime := time.Now()\n\t\t\t\texecutionEndTimes = append(executionEndTimes, endTime)\n\n\t\t\t\t// Complete the workflow\n\t\t\t\treturn nil, []*types.Decision{\n\t\t\t\t\t{\n\t\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tResult: []byte(fmt.Sprintf(\"execution-%d-complete-duration-%v\", attemptCount, tc.workflowDuration)),\n\t\t\t\t\t\t},\n\t\t\t\t\t}}, nil\n\t\t\t}\n\n\t\t\tpoller := &TaskPoller{\n\t\t\t\tEngine:          s.Engine,\n\t\t\t\tDomain:          s.DomainName,\n\t\t\t\tTaskList:        taskList,\n\t\t\t\tIdentity:        identity,\n\t\t\t\tDecisionHandler: dtHandler,\n\t\t\t\tLogger:          s.Logger,\n\t\t\t\tT:               s.T(),\n\t\t\t}\n\n\t\t\t// Process multiple decision tasks to observe overlap behavior\n\t\t\t// For long workflows, we'll process fewer tasks to avoid test timeout\n\t\t\tmaxTasks := 3\n\t\t\tif tc.workflowDuration > time.Second {\n\t\t\t\tmaxTasks = 2\n\t\t\t}\n\n\t\t\tfor i := 0; i < maxTasks; i++ {\n\t\t\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\t\t\ts.True(err == nil, err)\n\t\t\t}\n\n\t\t\t// Verify we have executions\n\t\t\ts.True(attemptCount >= 1, \"Expected at least 1 execution, got %d\", attemptCount)\n\n\t\t\t// Analyze overlap behavior - only if we have multiple executions\n\t\t\tif len(executionStartTimes) >= 2 {\n\t\t\t\tfor i := 1; i < len(executionStartTimes); i++ {\n\t\t\t\t\ttimeBetweenStarts := executionStartTimes[i].Sub(executionStartTimes[i-1])\n\t\t\t\t\ttimeBetweenEndAndNextStart := executionStartTimes[i].Sub(executionEndTimes[i-1])\n\n\t\t\t\t\ts.Logger.Info(\"Execution timing analysis\",\n\t\t\t\t\t\ttag.Counter(i),\n\t\t\t\t\t\ttag.WorkflowID(fmt.Sprintf(\"timeBetweenStarts:%v\", timeBetweenStarts)),\n\t\t\t\t\t\ttag.WorkflowRunID(fmt.Sprintf(\"timeBetweenEndAndNextStart:%v\", timeBetweenEndAndNextStart)),\n\t\t\t\t\t\ttag.WorkflowID(fmt.Sprintf(\"workflowDuration:%v\", tc.workflowDuration)),\n\t\t\t\t\t\ttag.WorkflowID(fmt.Sprintf(\"overlapPolicy:%s\", tc.cronOverlapPolicy.String())))\n\n\t\t\t\t\tif tc.cronOverlapPolicy == types.CronOverlapPolicySkipped {\n\t\t\t\t\t\t// For skipped policy, if workflow duration > cron interval,\n\t\t\t\t\t\t// subsequent executions should be skipped until previous completes\n\t\t\t\t\t\tif tc.workflowDuration > time.Second {\n\t\t\t\t\t\t\t// Should wait for previous execution to complete\n\t\t\t\t\t\t\ts.True(timeBetweenEndAndNextStart >= 0,\n\t\t\t\t\t\t\t\t\"Expected next execution to start after previous completion, got %v\",\n\t\t\t\t\t\t\t\ttimeBetweenEndAndNextStart)\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t// Short workflows should follow cron schedule\n\t\t\t\t\t\t\ts.True(timeBetweenStarts >= 800*time.Millisecond,\n\t\t\t\t\t\t\t\t\"Expected executions to be spaced by cron interval, got %v\",\n\t\t\t\t\t\t\t\ttimeBetweenStarts)\n\t\t\t\t\t\t}\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// For buffer one policy, next execution should start after previous completion\n\t\t\t\t\t\t// (but we're more lenient with timing in test environment)\n\t\t\t\t\t\ts.True(timeBetweenEndAndNextStart >= 0,\n\t\t\t\t\t\t\t\"Expected next execution to start after previous completion, got %v\",\n\t\t\t\t\t\t\ttimeBetweenEndAndNextStart)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Verify workflow history\n\t\t\tif len(executions) > 0 {\n\t\t\t\tevents := s.getHistory(s.DomainName, executions[0])\n\t\t\t\ts.True(len(events) > 0, \"Expected workflow history to have events\")\n\n\t\t\t\t// Check that the last event is a continue-as-new event\n\t\t\t\tlastEvent := events[len(events)-1]\n\t\t\t\ts.Equal(types.EventTypeWorkflowExecutionContinuedAsNew, lastEvent.GetEventType())\n\n\t\t\t\tattributes := lastEvent.WorkflowExecutionContinuedAsNewEventAttributes\n\t\t\t\ts.Equal(types.ContinueAsNewInitiatorCronSchedule, attributes.GetInitiator())\n\t\t\t}\n\n\t\t\t// Terminate the cron workflow\n\t\t\tctx, cancel = createContext()\n\t\t\tdefer cancel()\n\t\t\tterminateErr := s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain: s.DomainName,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t},\n\t\t\t})\n\t\t\ts.NoError(terminateErr)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "host/integrationbase.go",
    "content": "// Copyright (c) 2016 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/environment\"\n)\n\ntype (\n\t// IntegrationBase is a base struct for integration tests\n\tIntegrationBase struct {\n\t\tsuite.Suite\n\n\t\tTestCluster              *TestCluster\n\t\tTestClusterConfig        *TestClusterConfig\n\t\tEngine                   FrontendClient\n\t\tHistoryClient            HistoryClient\n\t\tAdminClient              AdminClient\n\t\tLogger                   log.Logger\n\t\tDomainName               string\n\t\tSecondaryDomainName      string\n\t\tTestRawHistoryDomainName string\n\t\tForeignDomainName        string\n\t\tArchivalDomainName       string\n\t\tActiveActiveDomainName   string\n\t\tDefaultTestCluster       testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster    testcluster.PersistenceTestCluster\n\t}\n\n\tIntegrationBaseParams struct {\n\t\tT                     *testing.T\n\t\tDefaultTestCluster    testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster testcluster.PersistenceTestCluster\n\t\tTestClusterConfig     *TestClusterConfig\n\t}\n)\n\nfunc NewIntegrationBase(params IntegrationBaseParams) *IntegrationBase {\n\treturn &IntegrationBase{\n\t\tDefaultTestCluster:    params.DefaultTestCluster,\n\t\tVisibilityTestCluster: params.VisibilityTestCluster,\n\t\tTestClusterConfig:     params.TestClusterConfig,\n\t}\n}\n\nfunc (s *IntegrationBase) setupSuite() {\n\ts.SetupLogger()\n\n\tif s.TestClusterConfig.FrontendAddress != \"\" {\n\t\ts.Logger.Info(\"Running integration test against specified frontend\", tag.Address(TestFlags.FrontendAddr))\n\t\tchannel, err := tchannel.NewChannelTransport(tchannel.ServiceName(\"cadence-frontend\"))\n\t\ts.Require().NoError(err)\n\t\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\t\tName: \"unittest\",\n\t\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\t\"cadence-frontend\": {Unary: channel.NewSingleOutbound(TestFlags.FrontendAddr)},\n\t\t\t},\n\t\t\tInboundMiddleware: yarpc.InboundMiddleware{\n\t\t\t\tUnary: &versionMiddleware{},\n\t\t\t},\n\t\t})\n\t\tif err := dispatcher.Start(); err != nil {\n\t\t\ts.Logger.Fatal(\"Failed to create outbound transport channel\", tag.Error(err))\n\t\t}\n\n\t\ts.Engine = NewFrontendClient(dispatcher)\n\t\ts.HistoryClient = NewHistoryClient(dispatcher)\n\t\ts.AdminClient = NewAdminClient(dispatcher)\n\t} else {\n\t\ts.Logger.Info(\"Running integration test against test cluster\")\n\t\tclusterMetadata := NewClusterMetadata(s.T(), s.TestClusterConfig)\n\t\tdc := persistence.DynamicConfiguration{\n\t\t\tEnableSQLAsyncTransaction:                dynamicproperties.GetBoolPropertyFn(false),\n\t\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t\t}\n\t\tparams := pt.TestBaseParams{\n\t\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\t\tClusterMetadata:       clusterMetadata,\n\t\t\tDynamicConfiguration:  dc,\n\t\t}\n\t\tcluster, err := NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\t\ts.Require().NoError(err)\n\t\ts.TestCluster = cluster\n\t\ts.Engine = s.TestCluster.GetFrontendClient()\n\t\ts.HistoryClient = s.TestCluster.GetHistoryClient()\n\t\ts.AdminClient = s.TestCluster.GetAdminClient()\n\t}\n\ts.TestRawHistoryDomainName = \"TestRawHistoryDomain\"\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.TestRawHistoryDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.ForeignDomainName = s.RandomizeStr(\"integration-foreign-test-domain\")\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.ForeignDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ts.Require().NoError(s.registerArchivalDomain())\n\ts.ActiveActiveDomainName = s.RandomizeStr(\"integration-active-active-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.ActiveActiveDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", &types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"us-east\": {ActiveClusterName: s.TestCluster.testBase.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"city\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"tokyo\": {ActiveClusterName: s.TestCluster.testBase.ClusterMetadata.GetCurrentClusterName()},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}))\n\n\t// this sleep is necessary because domainv2 cache gets refreshed in the\n\t// background only every domainCacheRefreshInterval period\n\ttime.Sleep(cache.DomainCacheRefreshInterval + time.Second)\n}\n\nfunc (s *IntegrationBase) SetupLogger() {\n\ts.Logger = testlogger.New(s.T())\n}\n\n// GetTestClusterConfig return test cluster config\nfunc GetTestClusterConfig(configFile string) (*TestClusterConfig, error) {\n\tif err := environment.SetupEnv(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tconfigLocation := configFile\n\tif TestFlags.TestClusterConfigFile != \"\" {\n\t\tconfigLocation = TestFlags.TestClusterConfigFile\n\t}\n\t// This is just reading a config so it's less of a security concern\n\t// #nosec\n\tconfContent, err := os.ReadFile(configLocation)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read test cluster config file %v: %v\", configLocation, err)\n\t}\n\tconfContent = []byte(os.ExpandEnv(string(confContent)))\n\tvar options TestClusterConfig\n\tif err := yaml.Unmarshal(confContent, &options); err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to decode test cluster config %v\", tag.Error(err))\n\t}\n\n\toptions.FrontendAddress = TestFlags.FrontendAddr\n\tif options.ESConfig != nil {\n\t\toptions.ESConfig.Indices[constants.VisibilityAppName] += uuid.New()\n\t}\n\tif options.Persistence.DBName == \"\" {\n\t\toptions.Persistence.DBName = \"test_\" + pt.GenerateRandomDBName(10)\n\t}\n\treturn &options, nil\n}\n\n// GetTestClusterConfigs return test cluster configs\nfunc GetTestClusterConfigs(configFile string) ([]*TestClusterConfig, error) {\n\tif err := environment.SetupEnv(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tfileName := configFile\n\tif TestFlags.TestClusterConfigFile != \"\" {\n\t\tfileName = TestFlags.TestClusterConfigFile\n\t}\n\n\tconfContent, err := os.ReadFile(fileName)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read test cluster config file %v: %v\", fileName, err)\n\t}\n\tconfContent = []byte(os.ExpandEnv(string(confContent)))\n\n\tvar clusterConfigs []*TestClusterConfig\n\tif err := yaml.Unmarshal(confContent, &clusterConfigs); err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to decode test cluster config %v\", tag.Error(err))\n\t}\n\treturn clusterConfigs, nil\n}\n\nfunc (s *IntegrationBase) TearDownBaseSuite() {\n\tif s.TestCluster != nil {\n\t\ts.TestCluster.TearDownCluster()\n\t\ts.TestCluster = nil\n\t\ts.Engine = nil\n\t\ts.AdminClient = nil\n\t}\n}\n\nfunc (s *IntegrationBase) RegisterDomain(\n\tdomain string,\n\tretentionDays int,\n\thistoryArchivalStatus types.ArchivalStatus,\n\thistoryArchivalURI string,\n\tvisibilityArchivalStatus types.ArchivalStatus,\n\tvisibilityArchivalURI string,\n\tactiveClusters *types.ActiveClusters,\n) error {\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\tisGlobalDomain := true\n\tif TestFlags.SQLPluginName == sqlite.PluginName {\n\t\t// There seems to be a bug in the SQLite plugin that causes integration tests to fail.\n\t\t// EnqueueMessage operation failed.  Error: sqlite3: database is locked\n\t\tisGlobalDomain = false\n\t}\n\treturn s.Engine.RegisterDomain(ctx, &types.RegisterDomainRequest{\n\t\tName:                                   domain,\n\t\tDescription:                            domain,\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(retentionDays),\n\t\tHistoryArchivalStatus:                  &historyArchivalStatus,\n\t\tHistoryArchivalURI:                     historyArchivalURI,\n\t\tVisibilityArchivalStatus:               &visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:                  visibilityArchivalURI,\n\t\tActiveClusters:                         activeClusters,\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t})\n}\n\nfunc (s *IntegrationBase) domainCacheRefresh() {\n\ts.TestClusterConfig.TimeSource.Advance(cache.DomainCacheRefreshInterval + time.Second)\n\t// this sleep is necessary to yield execution to other goroutines. not 100% guaranteed to work\n\ttime.Sleep(2 * time.Second)\n}\n\nfunc (s *IntegrationBase) RandomizeStr(id string) string {\n\treturn fmt.Sprintf(\"%v-%v\", id, uuid.New())\n}\n\nfunc (s *IntegrationBase) printWorkflowHistory(domain string, execution *types.WorkflowExecution) {\n\tevents := s.getHistory(domain, execution)\n\thistory := &types.History{}\n\thistory.Events = events\n\tPrettyPrintHistory(history, s.Logger)\n}\n\nfunc (s *IntegrationBase) getHistory(domain string, execution *types.WorkflowExecution) []*types.HistoryEvent {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:          domain,\n\t\tExecution:       execution,\n\t\tMaximumPageSize: 5, // Use small page size to force pagination code path\n\t})\n\ts.Require().NoError(err)\n\n\tevents := historyResponse.History.Events\n\tfor historyResponse.NextPageToken != nil {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err = s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain:        domain,\n\t\t\tExecution:     execution,\n\t\t\tNextPageToken: historyResponse.NextPageToken,\n\t\t})\n\t\tcancel()\n\t\ts.Require().NoError(err)\n\t\tevents = append(events, historyResponse.History.Events...)\n\t}\n\n\treturn events\n}\n\n// To register archival domain we can't use frontend API as the retention period is set to 0 for testing,\n// and request will be rejected by frontend. Here we make a call directly to persistence to register\n// the domain.\nfunc (s *IntegrationBase) registerArchivalDomain() error {\n\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\tdefer cancel()\n\n\ts.ArchivalDomainName = s.RandomizeStr(\"integration-archival-enabled-domain\")\n\tcurrentClusterName := s.TestCluster.testBase.ClusterMetadata.GetCurrentClusterName()\n\tdomainRequest := &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:     uuid.New(),\n\t\t\tName:   s.ArchivalDomainName,\n\t\t\tStatus: persistence.DomainStatusRegistered,\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                0,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalURI:       s.TestCluster.archiverBase.historyURI,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    s.TestCluster.archiverBase.visibilityURI,\n\t\t\tBadBinaries:              types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: currentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: currentClusterName},\n\t\t\t},\n\t\t},\n\t\tIsGlobalDomain:  false,\n\t\tFailoverVersion: constants.EmptyVersion,\n\t}\n\tresponse, err := s.TestCluster.testBase.DomainManager.CreateDomain(ctx, domainRequest)\n\tif err == nil {\n\t\ts.Logger.Info(\"Register domain succeeded\",\n\t\t\ttag.WorkflowDomainName(s.ArchivalDomainName),\n\t\t\ttag.WorkflowDomainID(response.ID),\n\t\t)\n\t}\n\treturn err\n}\n\n// PrettyPrintHistory prints history in human readable format\nfunc PrettyPrintHistory(history *types.History, logger log.Logger) {\n\tdata, err := json.MarshalIndent(history, \"\", \"    \")\n\n\tif err != nil {\n\t\tlogger.Error(\"Error serializing history: %v\\n\", tag.Error(err))\n\t}\n\n\tfmt.Println(\"******************************************\")\n\tfmt.Println(\"History\", tag.DetailInfo(string(data)))\n\tfmt.Println(\"******************************************\")\n}\n"
  },
  {
    "path": "host/membership_hashring.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"github.com/dgryski/go-farm\"\n\n\t\"github.com/uber/cadence/common/membership\"\n)\n\ntype simpleHashring struct {\n\thosts    []membership.HostInfo\n\thashfunc func([]byte) uint32\n}\n\n// newSimpleHashring returns a service resolver that maintains static mapping\n// between services and host info\nfunc newSimpleHashring(hosts []membership.HostInfo) *simpleHashring {\n\treturn &simpleHashring{\n\t\thosts:    hosts,\n\t\thashfunc: farm.Fingerprint32,\n\t}\n}\n\nfunc (s *simpleHashring) Lookup(key string) (membership.HostInfo, error) {\n\thash := int(s.hashfunc([]byte(key)))\n\tidx := hash % len(s.hosts)\n\treturn s.hosts[idx], nil\n}\n\nfunc (s *simpleHashring) AddListener(name string, notifyChannel chan<- *membership.ChangedEvent) error {\n\treturn nil\n}\n\nfunc (s *simpleHashring) RemoveListener(name string) error {\n\treturn nil\n}\n\nfunc (s *simpleHashring) MemberCount() int {\n\treturn len(s.hosts)\n}\n\nfunc (s *simpleHashring) Members() []membership.HostInfo {\n\treturn s.hosts\n}\n"
  },
  {
    "path": "host/membership_resolver.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/membership\"\n)\n\ntype simpleResolver struct {\n\thostInfo  membership.HostInfo\n\tresolvers map[string]*simpleHashring\n}\n\n// NewSimpleResolver returns a membership resolver interface\nfunc NewSimpleResolver(serviceName string, hosts map[string][]membership.HostInfo, currentHost membership.HostInfo) membership.Resolver {\n\tresolvers := make(map[string]*simpleHashring, len(hosts))\n\tfor service, hostList := range hosts {\n\t\tresolvers[service] = newSimpleHashring(hostList)\n\t}\n\treturn &simpleResolver{\n\t\thostInfo:  currentHost,\n\t\tresolvers: resolvers,\n\t}\n}\n\nfunc (s *simpleResolver) Start() {\n}\n\nfunc (s *simpleResolver) Stop() {\n}\n\nfunc (s *simpleResolver) EvictSelf() error {\n\treturn nil\n}\n\nfunc (s *simpleResolver) WhoAmI() (membership.HostInfo, error) {\n\treturn s.hostInfo, nil\n}\n\nfunc (s *simpleResolver) Subscribe(service string, name string, notifyChannel chan<- *membership.ChangedEvent) error {\n\treturn nil\n}\n\nfunc (s *simpleResolver) Unsubscribe(service string, name string) error {\n\treturn nil\n}\n\nfunc (s *simpleResolver) Lookup(service string, key string) (membership.HostInfo, error) {\n\tresolver, ok := s.resolvers[service]\n\tif !ok {\n\t\treturn membership.HostInfo{}, fmt.Errorf(\"cannot lookup host for service %q\", service)\n\t}\n\treturn resolver.Lookup(key)\n}\n\nfunc (s *simpleResolver) MemberCount(service string) (int, error) {\n\tmembers, err := s.Members(service)\n\treturn len(members), err\n}\n\nfunc (s *simpleResolver) Members(service string) ([]membership.HostInfo, error) {\n\tresolver, ok := s.resolvers[service]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"cannot lookup host for service %q\", service)\n\t}\n\treturn resolver.Members(), nil\n}\n\nfunc (s *simpleResolver) LookupByAddress(service string, address string) (membership.HostInfo, error) {\n\tresolver, ok := s.resolvers[service]\n\tif !ok {\n\t\treturn membership.HostInfo{}, fmt.Errorf(\"cannot lookup host for service %q\", service)\n\t}\n\tfor _, m := range resolver.Members() {\n\t\tif belongs, err := m.Belongs(address); err == nil && belongs {\n\t\t\treturn m, nil\n\t\t}\n\t}\n\n\treturn membership.HostInfo{}, errors.New(\"host not found\")\n}\n"
  },
  {
    "path": "host/ndc/integration_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\tadminClient \"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\ttest \"github.com/uber/cadence/common/testing\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/host\"\n)\n\nvar (\n\tclusterName              = []string{\"active\", \"standby\", \"other\"}\n\tclusterReplicationConfig = []*types.ClusterReplicationConfiguration{\n\t\t{ClusterName: clusterName[0]},\n\t\t{ClusterName: clusterName[1]},\n\t\t{ClusterName: clusterName[2]},\n\t}\n)\n\nfunc TestNDCIntegrationTestSuite(t *testing.T) {\n\tflag.Parse()\n\n\tclusterConfigs, err := host.GetTestClusterConfigs(\"../testdata/ndc_integration_test_clusters.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tclusterConfigs[0].WorkerConfig = &host.WorkerConfig{}\n\tclusterConfigs[1].WorkerConfig = &host.WorkerConfig{}\n\ttestCluster := host.NewPersistenceTestCluster(t, clusterConfigs[0])\n\tparams := NDCIntegrationTestSuiteParams{\n\t\tClusterConfigs:        clusterConfigs,\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t}\n\ts := NewNDCIntegrationTestSuite(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *NDCIntegrationTestSuite) SetupSuite() {\n\ts.serializer = persistence.NewPayloadSerializer()\n\ts.logger = testlogger.New(s.T())\n\n\ts.standByReplicationTasksChan = make(chan *types.ReplicationTask, 100)\n\n\ts.standByTaskID = 0\n\ts.mockAdminClient = make(map[string]adminClient.Client)\n\tcontroller := gomock.NewController(s.T())\n\tmockStandbyClient := adminClient.NewMockClient(controller)\n\tmockStandbyClient.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any()).DoAndReturn(s.GetReplicationMessagesMock).AnyTimes()\n\tmockOtherClient := adminClient.NewMockClient(controller)\n\tmockOtherClient.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any()).Return(\n\t\t&types.GetReplicationMessagesResponse{\n\t\t\tMessagesByShard: make(map[int32]*types.ReplicationMessages),\n\t\t}, nil).AnyTimes()\n\ts.mockAdminClient[\"standby\"] = mockStandbyClient\n\ts.mockAdminClient[\"other\"] = mockOtherClient\n\ts.clusterConfigs[0].MockAdminClient = s.mockAdminClient\n\n\tclusterMetadata := host.NewClusterMetadata(s.T(), s.clusterConfigs[0])\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableSQLAsyncTransaction:                dynamicproperties.GetBoolPropertyFn(false),\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.defaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := host.NewCluster(s.T(), s.clusterConfigs[0], s.logger.WithTags(tag.ClusterName(clusterName[0])), params)\n\ts.Require().NoError(err)\n\ts.active = cluster\n\n\ts.RegisterDomain()\n\n\ts.version = s.clusterConfigs[1].ClusterGroupMetadata.ClusterGroup[s.clusterConfigs[1].ClusterGroupMetadata.CurrentClusterName].InitialFailoverVersion\n\ts.versionIncrement = s.clusterConfigs[0].ClusterGroupMetadata.FailoverVersionIncrement\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, s.version)\n}\n\nfunc (s *NDCIntegrationTestSuite) GetReplicationMessagesMock(\n\tctx context.Context,\n\trequest *types.GetReplicationMessagesRequest,\n\topts ...yarpc.CallOption,\n) (*types.GetReplicationMessagesResponse, error) {\n\tselect {\n\tcase task := <-s.standByReplicationTasksChan:\n\t\ttaskID := atomic.AddInt64(&s.standByTaskID, 1)\n\t\ttask.SourceTaskID = taskID\n\t\ttasks := []*types.ReplicationTask{task}\n\t\tfor len(s.standByReplicationTasksChan) > 0 {\n\t\t\ttask = <-s.standByReplicationTasksChan\n\t\t\ttaskID := atomic.AddInt64(&s.standByTaskID, 1)\n\t\t\ttask.SourceTaskID = taskID\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\n\t\treplicationMessage := &types.ReplicationMessages{\n\t\t\tReplicationTasks:       tasks,\n\t\t\tLastRetrievedMessageID: tasks[len(tasks)-1].SourceTaskID,\n\t\t\tHasMore:                true,\n\t\t}\n\n\t\treturn &types.GetReplicationMessagesResponse{\n\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{0: replicationMessage},\n\t\t}, nil\n\tdefault:\n\t\treturn &types.GetReplicationMessagesResponse{\n\t\t\tMessagesByShard: make(map[int32]*types.ReplicationMessages),\n\t\t}, nil\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, s.version)\n}\n\nfunc (s *NDCIntegrationTestSuite) TearDownSuite() {\n\tif s.generator != nil {\n\t\ts.generator.Reset()\n\t}\n\ts.active.TearDownCluster()\n}\n\nfunc (s *NDCIntegrationTestSuite) TestSingleBranch() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-single-branch-test\" + uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\tversions := []int64{101, 1, 201, 301, 401, 601, 501, 801, 1001, 901, 701, 1101}\n\tfor _, version := range versions {\n\t\trunID := uuid.New()\n\t\thistoryBatch := []*types.History{}\n\t\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, version)\n\n\t\tfor s.generator.HasNextVertex() {\n\t\t\tevents := s.generator.GetNextVertices()\n\t\t\thistoryEvents := &types.History{}\n\t\t\tfor _, event := range events {\n\t\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t\t}\n\t\t\thistoryBatch = append(historyBatch, historyEvents)\n\t\t}\n\n\t\tversionHistory := s.eventBatchesToVersionHistory(nil, historyBatch)\n\t\ts.applyEvents(\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tworkflowType,\n\t\t\ttasklist,\n\t\t\tversionHistory,\n\t\t\thistoryBatch,\n\t\t\thistoryClient,\n\t\t)\n\n\t\terr := s.verifyEventHistory(workflowID, runID, historyBatch)\n\t\ts.Require().NoError(err)\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) verifyEventHistory(\n\tworkflowID string,\n\trunID string,\n\thistoryBatch []*types.History,\n) error {\n\t// get replicated history events from passive side\n\tpassiveClient := s.active.GetFrontendClient()\n\tctx, cancel := s.createContext()\n\tdefer cancel()\n\treplicatedHistory, err := passiveClient.GetWorkflowExecutionHistory(\n\t\tctx,\n\t\t&types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tMaximumPageSize:        1000,\n\t\t\tNextPageToken:          nil,\n\t\t\tWaitForNewEvent:        false,\n\t\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to get history event from passive side: %v\", err)\n\t}\n\n\t// compare origin events with replicated events\n\tbatchIndex := 0\n\tbatch := historyBatch[batchIndex].Events\n\teventIndex := 0\n\tfor _, event := range replicatedHistory.GetHistory().GetEvents() {\n\t\tif eventIndex >= len(batch) {\n\t\t\tbatchIndex++\n\t\t\tbatch = historyBatch[batchIndex].Events\n\t\t\teventIndex = 0\n\t\t}\n\t\toriginEvent := batch[eventIndex]\n\t\teventIndex++\n\t\tif originEvent.GetEventType() != event.GetEventType() {\n\t\t\treturn fmt.Errorf(\"the replicated event (%v) and the origin event (%v) are not the same\",\n\t\t\t\toriginEvent.GetEventType().String(), event.GetEventType().String())\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (s *NDCIntegrationTestSuite) TestMultipleBranches() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-multiple-branches-test\" + uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\tversions := []int64{101, 1, 201}\n\tfor _, version := range versions {\n\t\trunID := uuid.New()\n\n\t\tbaseBranch := []*types.History{}\n\t\tbaseGenerator := test.InitializeHistoryEventGenerator(s.domainName, version)\n\t\tbaseGenerator.SetVersion(version)\n\n\t\tfor i := 0; i < 10 && baseGenerator.HasNextVertex(); i++ {\n\t\t\tevents := baseGenerator.GetNextVertices()\n\t\t\thistoryEvents := &types.History{}\n\t\t\tfor _, event := range events {\n\t\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t\t}\n\t\t\tbaseBranch = append(baseBranch, historyEvents)\n\t\t}\n\t\tbaseVersionHistory := s.eventBatchesToVersionHistory(nil, baseBranch)\n\n\t\tbranch1 := []*types.History{}\n\t\tbranchVersionHistory1 := baseVersionHistory.Duplicate()\n\t\tbranchGenerator1 := baseGenerator.DeepCopy()\n\t\tfor i := 0; i < 10 && branchGenerator1.HasNextVertex(); i++ {\n\t\t\tevents := branchGenerator1.GetNextVertices()\n\t\t\thistoryEvents := &types.History{}\n\t\t\tfor _, event := range events {\n\t\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t\t}\n\t\t\tbranch1 = append(branch1, historyEvents)\n\t\t}\n\t\tbranchVersionHistory1 = s.eventBatchesToVersionHistory(branchVersionHistory1, branch1)\n\n\t\tbranch2 := []*types.History{}\n\t\tbranchVersionHistory2 := baseVersionHistory.Duplicate()\n\t\tbranchGenerator2 := baseGenerator.DeepCopy()\n\t\tbranchGenerator2.SetVersion(branchGenerator2.GetVersion() + 1)\n\t\tfor i := 0; i < 10 && branchGenerator2.HasNextVertex(); i++ {\n\t\t\tevents := branchGenerator2.GetNextVertices()\n\t\t\thistoryEvents := &types.History{}\n\t\t\tfor _, event := range events {\n\t\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t\t}\n\t\t\tbranch2 = append(branch2, historyEvents)\n\t\t}\n\t\tbranchVersionHistory2 = s.eventBatchesToVersionHistory(branchVersionHistory2, branch2)\n\n\t\ts.applyEvents(\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tworkflowType,\n\t\t\ttasklist,\n\t\t\tbaseVersionHistory,\n\t\t\tbaseBranch,\n\t\t\thistoryClient,\n\t\t)\n\t\ts.applyEvents(\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tworkflowType,\n\t\t\ttasklist,\n\t\t\tbranchVersionHistory1,\n\t\t\tbranch1,\n\t\t\thistoryClient,\n\t\t)\n\t\ts.applyEvents(\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tworkflowType,\n\t\t\ttasklist,\n\t\t\tbranchVersionHistory2,\n\t\t\tbranch2,\n\t\t\thistoryClient,\n\t\t)\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) TestHandcraftedMultipleBranches() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-handcrafted-multiple-branches-test\" + uuid.New()\n\trunID := uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tidentity := \"worker-identity\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\teventsBatch1 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        3,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        4,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        5,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tMarkerName:                   \"some marker name\",\n\t\t\t\t\tDetails:                      []byte(\"some marker details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 4,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        6,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        7,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t\tAttempt:          0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        8,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 1\",\n\t\t\t\t\tInput:      []byte(\"some signal details 1\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        9,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        10,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        12,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 2\",\n\t\t\t\t\tInput:      []byte(\"some signal details 2\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        13,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        14,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch2 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   31,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch3 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tStartedEventID:   14,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tStartedEventID:   7,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        17,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        18,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 17,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        19,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 8,\n\t\t\t\t\tStartedEventID:   9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        20,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID: 19,\n\t\t\t\t\tReason:                       common.StringPtr(\"some random reason\"),\n\t\t\t\t\tDetails:                      nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tversionHistory1 := s.eventBatchesToVersionHistory(nil, eventsBatch1)\n\n\tversionHistory2, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory2 = s.eventBatchesToVersionHistory(versionHistory2, eventsBatch2)\n\n\tversionHistory3, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory3 = s.eventBatchesToVersionHistory(versionHistory3, eventsBatch3)\n\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory1,\n\t\teventsBatch1,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory3,\n\t\teventsBatch3,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory2,\n\t\teventsBatch2,\n\t\thistoryClient,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestHandcraftedResetWorkflow_Zombie() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-handcrafted-reset-workflow-test\" + uuid.New()\n\trunID := uuid.New()\n\tnewRunID := uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tidentity := \"worker-identity\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\teventsBatch1 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        3,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        4,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        5,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tMarkerName:                   \"some marker name\",\n\t\t\t\t\tDetails:                      []byte(\"some marker details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 4,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        6,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        7,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t\tAttempt:          0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        8,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 1\",\n\t\t\t\t\tInput:      []byte(\"some signal details 1\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        9,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        10,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        12,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 2\",\n\t\t\t\t\tInput:      []byte(\"some signal details 2\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        13,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        14,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch2 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tStartedEventID:   14,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tStartedEventID:   7,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        17,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        18,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 17,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tresetCause := types.DecisionTaskFailedCauseResetWorkflow\n\tdtFailedReason := \"events-reapplication\"\n\teventsBatch3 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   22,\n\t\t\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tCause:            &resetCause,\n\t\t\t\t\tReason:           &dtFailedReason,\n\t\t\t\t\tBaseRunID:        runID,\n\t\t\t\t\tNewRunID:         newRunID,\n\t\t\t\t\tForkEventVersion: 21,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tversionHistory1 := s.eventBatchesToVersionHistory(nil, eventsBatch1)\n\n\tversionHistory2, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory2 = s.eventBatchesToVersionHistory(versionHistory2, eventsBatch2)\n\n\tversionHistory3, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(10, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory3 = s.eventBatchesToVersionHistory(versionHistory3, eventsBatch3)\n\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory1,\n\t\teventsBatch1,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory2,\n\t\teventsBatch2,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\tnewRunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory3,\n\t\teventsBatch3,\n\t\thistoryClient,\n\t)\n}\nfunc (s *NDCIntegrationTestSuite) TestHandcraftedResetWorkflow() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-handcrafted-reset-workflow-test\" + uuid.New()\n\trunID := uuid.New()\n\tnewRunID := uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tidentity := \"worker-identity\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\teventsBatch1 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        3,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        4,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        5,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tMarkerName:                   \"some marker name\",\n\t\t\t\t\tDetails:                      []byte(\"some marker details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 4,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        6,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        7,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t\tAttempt:          0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        8,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 1\",\n\t\t\t\t\tInput:      []byte(\"some signal details 1\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        9,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        10,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        12,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 2\",\n\t\t\t\t\tInput:      []byte(\"some signal details 2\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        13,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        14,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch2 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tStartedEventID:   14,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tStartedEventID:   7,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        17,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        18,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 17,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        19,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 8,\n\t\t\t\t\tStartedEventID:   9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        20,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID: 19,\n\t\t\t\t\tReason:                       common.StringPtr(\"some random reason\"),\n\t\t\t\t\tDetails:                      nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tresetCause := types.DecisionTaskFailedCauseResetWorkflow\n\tdtFailedReason := \"events-reapplication\"\n\teventsBatch3 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   22,\n\t\t\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tCause:            &resetCause,\n\t\t\t\t\tReason:           &dtFailedReason,\n\t\t\t\t\tBaseRunID:        runID,\n\t\t\t\t\tNewRunID:         newRunID,\n\t\t\t\t\tForkEventVersion: 21,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tversionHistory1 := s.eventBatchesToVersionHistory(nil, eventsBatch1)\n\n\tversionHistory2, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory2 = s.eventBatchesToVersionHistory(versionHistory2, eventsBatch2)\n\n\tversionHistory3, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(10, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory3 = s.eventBatchesToVersionHistory(versionHistory3, eventsBatch3)\n\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory1,\n\t\teventsBatch1,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory2,\n\t\teventsBatch2,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\tnewRunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory3,\n\t\teventsBatch3,\n\t\thistoryClient,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestHandcraftedMultipleBranchesWithZombieContinueAsNew() {\n\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-handcrafted-multiple-branches-with-continue-as-new-test\" + uuid.New()\n\trunID := uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tidentity := \"worker-identity\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\teventsBatch1 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        3,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        4,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        5,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tMarkerName:                   \"some marker name\",\n\t\t\t\t\tDetails:                      []byte(\"some marker details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 4,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        6,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        7,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t\tAttempt:          0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        8,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 1\",\n\t\t\t\t\tInput:      []byte(\"some signal details 1\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        9,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        10,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        12,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 2\",\n\t\t\t\t\tInput:      []byte(\"some signal details 2\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        13,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        14,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch2 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   32,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 8,\n\t\t\t\t\tStartedEventID:   9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t// need to keep the workflow open for testing\n\t}\n\n\teventsBatch3 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 8,\n\t\t\t\t\tStartedEventID:   9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\t\t\tNewExecutionRunID:                   uuid.New(),\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tDecisionTaskCompletedEventID:        19,\n\t\t\t\t\tInitiator:                           types.ContinueAsNewInitiatorDecider.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tversionHistory1 := s.eventBatchesToVersionHistory(nil, eventsBatch1)\n\n\tversionHistory2, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory2 = s.eventBatchesToVersionHistory(versionHistory2, eventsBatch2)\n\n\tversionHistory3, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory3 = s.eventBatchesToVersionHistory(versionHistory3, eventsBatch3)\n\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory1,\n\t\teventsBatch1,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory2,\n\t\teventsBatch2,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory3,\n\t\teventsBatch3,\n\t\thistoryClient,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestEventsReapply_ZombieWorkflow() {\n\n\tworkflowID := \"ndc-single-branch-test\" + uuid.New()\n\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\n\t// active has initial version 0\n\thistoryClient := s.active.GetHistoryClient()\n\n\tversion := int64(101)\n\trunID := uuid.New()\n\thistoryBatch := []*types.History{}\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, version)\n\n\tfor s.generator.HasNextVertex() {\n\t\tevents := s.generator.GetNextVertices()\n\t\thistoryEvents := &types.History{}\n\t\tfor _, event := range events {\n\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t}\n\t\thistoryBatch = append(historyBatch, historyEvents)\n\t}\n\n\tversionHistory := s.eventBatchesToVersionHistory(nil, historyBatch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\thistoryBatch,\n\t\thistoryClient,\n\t)\n\n\tversion = int64(1)\n\trunID = uuid.New()\n\thistoryBatch = []*types.History{}\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, version)\n\n\t// verify two batches of zombie workflow are call reapply API\n\ts.mockAdminClient[\"standby\"].(*adminClient.MockClient).EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil).Times(2)\n\tfor i := 0; i < 2 && s.generator.HasNextVertex(); i++ {\n\t\tevents := s.generator.GetNextVertices()\n\t\thistoryEvents := &types.History{}\n\t\tfor _, event := range events {\n\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t}\n\t\thistoryBatch = append(historyBatch, historyEvents)\n\t}\n\n\tversionHistory = s.eventBatchesToVersionHistory(nil, historyBatch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\thistoryBatch,\n\t\thistoryClient,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestEventsReapply_UpdateNonCurrentBranch() {\n\n\tworkflowID := \"ndc-single-branch-test\" + uuid.New()\n\trunID := uuid.New()\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tversion := int64(101)\n\tisWorkflowFinished := false\n\n\thistoryClient := s.active.GetHistoryClient()\n\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, version)\n\tbaseBranch := []*types.History{}\n\tvar taskID int64\n\tfor i := 0; i < 4 && s.generator.HasNextVertex(); i++ {\n\t\tevents := s.generator.GetNextVertices()\n\t\thistoryEvents := &types.History{}\n\t\tfor _, event := range events {\n\t\t\thistoryEvent := event.GetData().(*types.HistoryEvent)\n\t\t\ttaskID = historyEvent.TaskID\n\t\t\thistoryEvents.Events = append(historyEvents.Events, historyEvent)\n\t\t\tswitch historyEvent.GetEventType() {\n\t\t\tcase types.EventTypeWorkflowExecutionCompleted,\n\t\t\t\ttypes.EventTypeWorkflowExecutionFailed,\n\t\t\t\ttypes.EventTypeWorkflowExecutionTimedOut,\n\t\t\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t\t\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew,\n\t\t\t\ttypes.EventTypeWorkflowExecutionCanceled:\n\t\t\t\tisWorkflowFinished = true\n\t\t\t}\n\t\t}\n\t\tbaseBranch = append(baseBranch, historyEvents)\n\t}\n\tif isWorkflowFinished {\n\t\t// cannot proceed since the test below requires workflow not finished\n\t\t// this is ok since build kite will run this test several times\n\t\ts.logger.Info(\"Encounter finish workflow history event during randomization test, skip\")\n\t\treturn\n\t}\n\n\tversionHistory := s.eventBatchesToVersionHistory(nil, baseBranch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\tbaseBranch,\n\t\thistoryClient,\n\t)\n\n\tnewGenerator := s.generator.DeepCopy()\n\tnewBranch := []*types.History{}\n\tnewVersionHistory := versionHistory.Duplicate()\n\tnewGenerator.SetVersion(newGenerator.GetVersion() + 1) // simulate events from other cluster\n\tfor i := 0; i < 4 && newGenerator.HasNextVertex(); i++ {\n\t\tevents := newGenerator.GetNextVertices()\n\t\thistoryEvents := &types.History{}\n\t\tfor _, event := range events {\n\t\t\thistory := event.GetData().(*types.HistoryEvent)\n\t\t\ttaskID = history.TaskID\n\t\t\thistoryEvents.Events = append(historyEvents.Events, history)\n\t\t}\n\t\tnewBranch = append(newBranch, historyEvents)\n\t}\n\tnewVersionHistory = s.eventBatchesToVersionHistory(newVersionHistory, newBranch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tnewVersionHistory,\n\t\tnewBranch,\n\t\thistoryClient,\n\t)\n\n\ts.mockAdminClient[\"standby\"].(*adminClient.MockClient).EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t// Handcraft a stale signal event\n\tbaseBranchLastEventBatch := baseBranch[len(baseBranch)-1].GetEvents()\n\tbaseBranchLastEvent := baseBranchLastEventBatch[len(baseBranchLastEventBatch)-1]\n\tstaleBranch := []*types.History{\n\t\t{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        baseBranchLastEvent.ID + 1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\tVersion:   baseBranchLastEvent.Version, // dummy event from other cluster\n\t\t\t\t\tTaskID:    taskID,\n\t\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\t\tSignalName: \"signal\",\n\t\t\t\t\t\tInput:      []byte{},\n\t\t\t\t\t\tIdentity:   \"ndc_integration_test\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tstaleVersionHistory := s.eventBatchesToVersionHistory(versionHistory.Duplicate(), staleBranch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tstaleVersionHistory,\n\t\tstaleBranch,\n\t\thistoryClient,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestAdminGetWorkflowExecutionRawHistoryV2() {\n\n\tworkflowID := \"ndc-re-send-test\" + uuid.New()\n\trunID := uuid.New()\n\tworkflowType := \"ndc-re-send-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\tidentity := \"ndc-re-send-test\"\n\n\thistoryClient := s.active.GetHistoryClient()\n\tadminClient := s.active.GetAdminClient()\n\tgetHistory := func(\n\t\tdomain string,\n\t\tworkflowID string,\n\t\trunID string,\n\t\tstartEventID *int64,\n\t\tstartEventVersion *int64,\n\t\tendEventID *int64,\n\t\tendEventVersion *int64,\n\t\tpageSize int,\n\t\ttoken []byte,\n\t) (*types.GetWorkflowExecutionRawHistoryV2Response, error) {\n\n\t\texecution := &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t}\n\t\tctx, cancel := s.createContext()\n\t\tdefer cancel()\n\t\treturn adminClient.GetWorkflowExecutionRawHistoryV2(ctx, &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain:            domain,\n\t\t\tExecution:         execution,\n\t\t\tStartEventID:      startEventID,\n\t\t\tStartEventVersion: startEventVersion,\n\t\t\tEndEventID:        endEventID,\n\t\t\tEndEventVersion:   endEventVersion,\n\t\t\tMaximumPageSize:   int32(pageSize),\n\t\t\tNextPageToken:     token,\n\t\t})\n\t}\n\n\teventsBatch1 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        3,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        4,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        5,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{\n\t\t\t\t\tMarkerName:                   \"some marker name\",\n\t\t\t\t\tDetails:                      []byte(\"some marker details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 4,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        6,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        7,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t\tAttempt:          0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        8,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 1\",\n\t\t\t\t\tInput:      []byte(\"some signal details 1\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        9,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        10,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        11,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        12,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\t\tSignalName: \"some signal name 2\",\n\t\t\t\t\tInput:      []byte(\"some signal details 2\"),\n\t\t\t\t\tIdentity:   identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        13,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        14,\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch2 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   31,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 9,\n\t\t\t\t\tStartedEventID:   10,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   31,\n\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID:  4,\n\t\t\t\t\tActivityID:                    \"0\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity-type\"},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                         nil,\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(20),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(20),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(20),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch3 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        15,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\tStartedEventID:   14,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        16,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 6,\n\t\t\t\t\tStartedEventID:   7,\n\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        17,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tAttempt:                    0,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        18,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 17,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t\tRequestID:        uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        19,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 8,\n\t\t\t\t\tStartedEventID:   9,\n\t\t\t\t\tIdentity:         identity,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        20,\n\t\t\t\tVersion:   30,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID: 19,\n\t\t\t\t\tReason:                       common.StringPtr(\"some random reason\"),\n\t\t\t\t\tDetails:                      nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\teventsBatch4 := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        17,\n\t\t\t\tVersion:   32,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t}},\n\t}\n\n\tversionHistory1 := s.eventBatchesToVersionHistory(nil, eventsBatch1)\n\n\tversionHistory2, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory2 = s.eventBatchesToVersionHistory(versionHistory2, eventsBatch2)\n\n\tversionHistory3, err := versionHistory1.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(14, 21),\n\t)\n\ts.NoError(err)\n\tversionHistory3 = s.eventBatchesToVersionHistory(versionHistory3, eventsBatch3)\n\n\tversionHistory4, err := versionHistory2.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(16, 31),\n\t)\n\ts.NoError(err)\n\tversionHistory4 = s.eventBatchesToVersionHistory(versionHistory4, eventsBatch4)\n\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory1,\n\t\teventsBatch1,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory3,\n\t\teventsBatch3,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory2,\n\t\teventsBatch2,\n\t\thistoryClient,\n\t)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory4,\n\t\teventsBatch4,\n\t\thistoryClient,\n\t)\n\n\t// GetWorkflowExecutionRawHistoryV2 start and end\n\tvar token []byte\n\tbatchCount := 0\n\tfor continuePaging := true; continuePaging; continuePaging = len(token) != 0 {\n\t\tresp, err := getHistory(\n\t\t\ts.domainName,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tcommon.Int64Ptr(14),\n\t\t\tcommon.Int64Ptr(21),\n\t\t\tcommon.Int64Ptr(20),\n\t\t\tcommon.Int64Ptr(30),\n\t\t\t1,\n\t\t\ttoken,\n\t\t)\n\t\ts.Nil(err)\n\t\ts.True(len(resp.HistoryBatches) <= 1)\n\t\tbatchCount++\n\t\ttoken = resp.NextPageToken\n\t}\n\ts.Equal(batchCount, 4)\n\n\t// GetWorkflowExecutionRawHistoryV2 start and end not on the same branch\n\ttoken = nil\n\tbatchCount = 0\n\tfor continuePaging := true; continuePaging; continuePaging = len(token) != 0 {\n\t\tresp, err := getHistory(\n\t\t\ts.domainName,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tcommon.Int64Ptr(17),\n\t\t\tcommon.Int64Ptr(30),\n\t\t\tcommon.Int64Ptr(17),\n\t\t\tcommon.Int64Ptr(32),\n\t\t\t1,\n\t\t\ttoken,\n\t\t)\n\t\ts.Nil(err)\n\t\ts.True(len(resp.HistoryBatches) <= 1)\n\t\tbatchCount++\n\t\ttoken = resp.NextPageToken\n\t}\n\ts.Equal(batchCount, 2)\n\n\t// GetWorkflowExecutionRawHistoryV2 start boundary\n\ttoken = nil\n\tbatchCount = 0\n\tfor continuePaging := true; continuePaging; continuePaging = len(token) != 0 {\n\t\tresp, err := getHistory(\n\t\t\ts.domainName,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tcommon.Int64Ptr(14),\n\t\t\tcommon.Int64Ptr(21),\n\t\t\tnil,\n\t\t\tnil,\n\t\t\t1,\n\t\t\ttoken,\n\t\t)\n\t\ts.Nil(err)\n\t\ts.True(len(resp.HistoryBatches) <= 1)\n\t\tbatchCount++\n\t\ttoken = resp.NextPageToken\n\t}\n\ts.Equal(batchCount, 3)\n\n\t// GetWorkflowExecutionRawHistoryV2 end boundary\n\ttoken = nil\n\tbatchCount = 0\n\tfor continuePaging := true; continuePaging; continuePaging = len(token) != 0 {\n\t\tresp, err := getHistory(\n\t\t\ts.domainName,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tnil,\n\t\t\tnil,\n\t\t\tcommon.Int64Ptr(17),\n\t\t\tcommon.Int64Ptr(32),\n\t\t\t1,\n\t\t\ttoken,\n\t\t)\n\t\ts.Nil(err)\n\t\ts.True(len(resp.HistoryBatches) <= 1)\n\t\tbatchCount++\n\t\ttoken = resp.NextPageToken\n\t}\n\ts.Equal(batchCount, 10)\n}\n\nfunc (s *NDCIntegrationTestSuite) TestWorkflowStartTime() {\n\ts.setupRemoteFrontendClients()\n\tworkflowID := \"ndc-workflow-start-time-test\" + uuid.New()\n\n\tworkflowType := \"start-time-test-workflow-type\"\n\ttasklist := \"start-time-test-tasklist\"\n\n\tstartTime := time.Now().Add(-time.Minute)\n\trunID := uuid.New()\n\n\thistoryBatch := []*types.History{\n\t\t{Events: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tTimestamp: common.Int64Ptr(startTime.UnixNano()),\n\t\t\t\tVersion:   21,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\t\t\tInput:                               nil,\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1000),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1000),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds:     common.Int32Ptr(100),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t},\n\t}\n\n\tversionHistory := s.eventBatchesToVersionHistory(nil, historyBatch)\n\ts.applyEvents(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\thistoryBatch,\n\t\ts.active.GetHistoryClient(),\n\t)\n\n\terr := s.verifyEventHistory(workflowID, runID, historyBatch)\n\ts.Require().NoError(err)\n\n\t// we are replicating to the `active` cluster, check the comments in\n\t// registerDomain() method below\n\tctx, cancel := s.createContext()\n\tdefer cancel()\n\tdescResp, err := s.active.GetFrontendClient().DescribeWorkflowExecution(\n\t\tctx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t},\n\t)\n\ts.NoError(err)\n\ts.WithinDuration(\n\t\tstartTime,\n\t\ttime.Unix(0, descResp.WorkflowExecutionInfo.GetStartTime()),\n\t\ttime.Millisecond,\n\t)\n}\n\nfunc (s *NDCIntegrationTestSuite) RegisterDomain() {\n\ts.domainName = \"test-simple-workflow-ndc-\" + common.GenerateRandomString(5)\n\tclient1 := s.active.GetFrontendClient() // active\n\tctx, cancel := s.createContext()\n\tdefer cancel()\n\terr := client1.RegisterDomain(ctx, &types.RegisterDomainRequest{\n\t\tName:           s.domainName,\n\t\tIsGlobalDomain: true,\n\t\tClusters:       clusterReplicationConfig,\n\t\t// make the active cluster `standby` and replicate to `active` cluster\n\t\tActiveClusterName:                      clusterName[1],\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t})\n\ts.Require().NoError(err)\n\n\tdescReq := &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(s.domainName),\n\t}\n\tctx, cancel = s.createContext()\n\tdefer cancel()\n\tresp, err := client1.DescribeDomain(ctx, descReq)\n\ts.Require().NoError(err)\n\ts.Require().NotNil(resp)\n\ts.domainID = resp.GetDomainInfo().GetUUID()\n\t// Wait for domain cache to pick the change\n\ttime.Sleep(2 * cache.DomainCacheRefreshInterval)\n\n\ts.logger.Info(fmt.Sprintf(\"Domain name: %v - ID: %v\", s.domainName, s.domainID))\n}\n\nfunc (s *NDCIntegrationTestSuite) generateNewRunHistory(\n\tevent *types.HistoryEvent,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tversion int64,\n\tworkflowType string,\n\ttaskList string,\n) *persistence.DataBlob {\n\n\t// TODO temporary code to generate first event & version history\n\t//  we should generate these as part of modeled based testing\n\n\tif event.GetWorkflowExecutionContinuedAsNewEventAttributes() == nil {\n\t\treturn nil\n\t}\n\n\tevent.WorkflowExecutionContinuedAsNewEventAttributes.NewExecutionRunID = uuid.New()\n\n\tfirstScheduleTime := time.Unix(0, 100)\n\tnewRunFirstEvent := &types.HistoryEvent{\n\t\tID:        constants.FirstEventID,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tVersion:   version,\n\t\tTaskID:    1,\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowType:         &types.WorkflowType{Name: workflowType},\n\t\t\tParentWorkflowDomain: common.StringPtr(domain),\n\t\t\tParentWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: uuid.New(),\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tParentInitiatedEventID: common.Int64Ptr(event.ID),\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: taskList,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(10),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\tContinuedExecutionRunID:             runID,\n\t\t\tInitiator:                           types.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t\t\tOriginalExecutionRunID:              runID,\n\t\t\tIdentity:                            \"NDC-test\",\n\t\t\tFirstExecutionRunID:                 runID,\n\t\t\tFirstScheduleTime:                   &firstScheduleTime,\n\t\t\tAttempt:                             0,\n\t\t\tExpirationTimestamp:                 common.Int64Ptr(time.Now().Add(time.Minute).UnixNano()),\n\t\t},\n\t}\n\n\teventBlob, err := s.serializer.SerializeBatchEvents([]*types.HistoryEvent{newRunFirstEvent}, constants.EncodingTypeThriftRW)\n\ts.NoError(err)\n\n\treturn eventBlob\n}\n\nfunc (s *NDCIntegrationTestSuite) toInternalDataBlob(\n\tblob *persistence.DataBlob,\n) *types.DataBlob {\n\n\tif blob == nil {\n\t\treturn nil\n\t}\n\n\tvar encodingType types.EncodingType\n\tswitch blob.GetEncoding() {\n\tcase constants.EncodingTypeThriftRW:\n\t\tencodingType = types.EncodingTypeThriftRW\n\tcase constants.EncodingTypeJSON,\n\t\tconstants.EncodingTypeGob,\n\t\tconstants.EncodingTypeUnknown,\n\t\tconstants.EncodingTypeEmpty:\n\t\tpanic(fmt.Sprintf(\"unsupported encoding type: %v\", blob.GetEncoding()))\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown encoding type: %v\", blob.GetEncoding()))\n\t}\n\n\treturn &types.DataBlob{\n\t\tEncodingType: encodingType.Ptr(),\n\t\tData:         blob.Data,\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) generateEventBlobs(\n\tworkflowID string,\n\trunID string,\n\tworkflowType string,\n\ttasklist string,\n\tbatch *types.History,\n) (*persistence.DataBlob, *persistence.DataBlob) {\n\t// TODO temporary code to generate next run first event\n\t//  we should generate these as part of modeled based testing\n\tlastEvent := batch.Events[len(batch.Events)-1]\n\tnewRunEventBlob := s.generateNewRunHistory(\n\t\tlastEvent, s.domainName, workflowID, runID, lastEvent.Version, workflowType, tasklist,\n\t)\n\t// must serialize events batch after attempt on continue as new as generateNewRunHistory will\n\t// modify the NewExecutionRunID attr\n\teventBlob, err := s.serializer.SerializeBatchEvents(batch.Events, constants.EncodingTypeThriftRW)\n\ts.NoError(err)\n\treturn eventBlob, newRunEventBlob\n}\n\nfunc (s *NDCIntegrationTestSuite) applyEvents(\n\tworkflowID string,\n\trunID string,\n\tworkflowType string,\n\ttasklist string,\n\tversionHistory *persistence.VersionHistory,\n\teventBatches []*types.History,\n\thistoryClient host.HistoryClient,\n) {\n\tfor _, batch := range eventBatches {\n\t\teventBlob, newRunEventBlob := s.generateEventBlobs(workflowID, runID, workflowType, tasklist, batch)\n\t\treq := &types.ReplicateEventsV2Request{\n\t\t\tDomainUUID: s.domainID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tVersionHistoryItems: s.toInternalVersionHistoryItems(versionHistory),\n\t\t\tEvents:              s.toInternalDataBlob(eventBlob),\n\t\t\tNewRunEvents:        s.toInternalDataBlob(newRunEventBlob),\n\t\t}\n\n\t\tctx, cancel := s.createContext()\n\t\terr := historyClient.ReplicateEventsV2(ctx, req)\n\t\tcancel()\n\t\ts.Nil(err, \"Failed to replicate history event\")\n\t\tctx, cancel = s.createContext()\n\t\terr = historyClient.ReplicateEventsV2(ctx, req)\n\t\tcancel()\n\t\ts.Nil(err, \"Failed to dedup replicate history event\")\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) applyEventsThroughFetcher(\n\tworkflowID string,\n\trunID string,\n\tworkflowType string,\n\ttasklist string,\n\tversionHistory *persistence.VersionHistory,\n\teventBatches []*types.History,\n) {\n\tfor _, batch := range eventBatches {\n\t\teventBlob, newRunEventBlob := s.generateEventBlobs(workflowID, runID, workflowType, tasklist, batch)\n\n\t\ttaskType := types.ReplicationTaskTypeHistoryV2\n\t\treplicationTask := &types.ReplicationTask{\n\t\t\tTaskType:     &taskType,\n\t\t\tSourceTaskID: 1,\n\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\tDomainID:            s.domainID,\n\t\t\t\tWorkflowID:          workflowID,\n\t\t\t\tRunID:               runID,\n\t\t\t\tVersionHistoryItems: s.toInternalVersionHistoryItems(versionHistory),\n\t\t\t\tEvents:              s.toInternalDataBlob(eventBlob),\n\t\t\t\tNewRunEvents:        s.toInternalDataBlob(newRunEventBlob),\n\t\t\t},\n\t\t}\n\n\t\ts.standByReplicationTasksChan <- replicationTask\n\t\t// this is to test whether dedup works\n\t\ts.standByReplicationTasksChan <- replicationTask\n\t}\n}\n\nfunc (s *NDCIntegrationTestSuite) eventBatchesToVersionHistory(\n\tversionHistory *persistence.VersionHistory,\n\teventBatches []*types.History,\n) *persistence.VersionHistory {\n\n\t// TODO temporary code to generate version history\n\t//  we should generate version as part of modeled based testing\n\tif versionHistory == nil {\n\t\tversionHistory = persistence.NewVersionHistory(nil, nil)\n\t}\n\tfor _, batch := range eventBatches {\n\t\tfor _, event := range batch.Events {\n\t\t\terr := versionHistory.AddOrUpdateItem(\n\t\t\t\tpersistence.NewVersionHistoryItem(\n\t\t\t\t\tevent.ID,\n\t\t\t\t\tevent.Version,\n\t\t\t\t))\n\t\t\ts.NoError(err)\n\t\t}\n\t}\n\n\treturn versionHistory\n}\n\nfunc (s *NDCIntegrationTestSuite) toInternalVersionHistoryItems(\n\tversionHistory *persistence.VersionHistory,\n) []*types.VersionHistoryItem {\n\tif versionHistory == nil {\n\t\treturn nil\n\t}\n\n\treturn versionHistory.ToInternalType().Items\n}\n\nfunc (s *NDCIntegrationTestSuite) createContext() (context.Context, context.CancelFunc) {\n\tctx, cancel := context.WithTimeout(context.Background(), 90*time.Second)\n\treturn ctx, cancel\n}\n\nfunc (s *NDCIntegrationTestSuite) setupRemoteFrontendClients() {\n\ts.mockAdminClient[\"standby\"].(*adminClient.MockClient).EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\ts.mockAdminClient[\"other\"].(*adminClient.MockClient).EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n}\n"
  },
  {
    "path": "host/ndc/replication_integration_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"reflect\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\ttest \"github.com/uber/cadence/common/testing\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tdefaultTestPersistenceTimeout = 5 * time.Second\n)\n\nfunc (s *NDCIntegrationTestSuite) TestReplicationMessageApplication() {\n\n\tworkflowID := \"replication-message-test\" + uuid.New()\n\trunID := uuid.New()\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\n\tvar historyBatch []*types.History\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, 1)\n\n\tfor s.generator.HasNextVertex() {\n\t\tevents := s.generator.GetNextVertices()\n\t\thistoryEvents := &types.History{}\n\t\tfor _, event := range events {\n\t\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t\t}\n\t\thistoryBatch = append(historyBatch, historyEvents)\n\t}\n\n\tversionHistory := s.eventBatchesToVersionHistory(nil, historyBatch)\n\n\ts.applyEventsThroughFetcher(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\thistoryBatch,\n\t)\n\n\t// Applying replication messages through fetcher is Async.\n\t// So we need to retry a couple of times.\n\tfor i := 0; i < 10; i++ {\n\t\ttime.Sleep(time.Second)\n\t\terr := s.verifyEventHistory(workflowID, runID, historyBatch)\n\t\tif err == nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\ts.Fail(\"Verification of replicated messages failed\")\n}\n\nfunc (s *NDCIntegrationTestSuite) TestReplicationMessageDLQ() {\n\n\tworkflowID := \"replication-message-dlq-test\" + uuid.New()\n\trunID := uuid.New()\n\tworkflowType := \"event-generator-workflow-type\"\n\ttasklist := \"event-generator-taskList\"\n\n\tvar historyBatch []*types.History\n\ts.generator = test.InitializeHistoryEventGenerator(s.domainName, 1)\n\n\tevents := s.generator.GetNextVertices()\n\thistoryEvents := &types.History{}\n\tfor _, event := range events {\n\t\thistoryEvents.Events = append(historyEvents.Events, event.GetData().(*types.HistoryEvent))\n\t}\n\thistoryBatch = append(historyBatch, historyEvents)\n\n\tversionHistory := s.eventBatchesToVersionHistory(nil, historyBatch)\n\n\ts.NotNil(historyBatch)\n\thistoryBatch[0].Events[1].Version = 2\n\n\ts.applyEventsThroughFetcher(\n\t\tworkflowID,\n\t\trunID,\n\t\tworkflowType,\n\t\ttasklist,\n\t\tversionHistory,\n\t\thistoryBatch,\n\t)\n\n\texecMgrFactory := s.active.GetExecutionManagerFactory()\n\texecutionManager, err := execMgrFactory.NewExecutionManager(0)\n\ts.NoError(err)\n\n\texpectedDLQMsgs := map[int64]bool{}\n\tfor _, batch := range historyBatch {\n\t\tfirstEventID := batch.Events[0].ID\n\t\texpectedDLQMsgs[firstEventID] = true\n\t}\n\n\t// Applying replication messages through fetcher is Async.\n\t// So we need to retry a couple of times.\nLoop:\n\tfor i := 0; i < 60; i++ {\n\t\ttime.Sleep(time.Second)\n\n\t\tactualDLQMsgs := map[int64]bool{}\n\t\trequest := persistence.NewGetReplicationTasksFromDLQRequest(\n\t\t\t\"standby\", 0, math.MaxInt64, math.MaxInt64, nil,\n\t\t)\n\t\tvar token []byte\n\t\tfor doPaging := true; doPaging; doPaging = len(token) > 0 {\n\t\t\trequest.NextPageToken = token\n\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\t\t\tresponse, err := executionManager.GetReplicationTasksFromDLQ(ctx, request)\n\t\t\tcancel()\n\t\t\tif err != nil {\n\t\t\t\tcontinue Loop\n\t\t\t}\n\t\t\ttoken = response.NextPageToken\n\n\t\t\tfor _, task := range response.Tasks {\n\t\t\t\tt, ok := task.(*persistence.HistoryReplicationTask)\n\t\t\t\ts.True(ok, \"task is not a HistoryReplicationTask\")\n\t\t\t\tfirstEventID := t.FirstEventID\n\t\t\t\tactualDLQMsgs[firstEventID] = true\n\t\t\t}\n\t\t}\n\t\tif reflect.DeepEqual(expectedDLQMsgs, actualDLQMsgs) {\n\t\t\treturn\n\t\t}\n\t}\n\n\ts.Fail(\"Failed to get messages from DLQ.\")\n}\n"
  },
  {
    "path": "host/ndc/test_suites.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n\t\"github.com/uber/cadence/common/testing\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/host\"\n)\n\n// NOTE: the following definitions can't be defined in *_test.go\n// since they need to be exported and used by our internal tests\n\ntype (\n\tNDCIntegrationTestSuite struct {\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t\tsuite.Suite\n\t\tactive     *host.TestCluster\n\t\tgenerator  testing.Generator\n\t\tserializer persistence.PayloadSerializer\n\t\tlogger     log.Logger\n\n\t\tdomainName                  string\n\t\tdomainID                    string\n\t\tversion                     int64\n\t\tversionIncrement            int64\n\t\tmockAdminClient             map[string]admin.Client\n\t\tstandByReplicationTasksChan chan *types.ReplicationTask\n\t\tstandByTaskID               int64\n\n\t\tclusterConfigs        []*host.TestClusterConfig\n\t\tdefaultTestCluster    testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster testcluster.PersistenceTestCluster\n\t}\n\n\tNDCIntegrationTestSuiteParams struct {\n\t\tClusterConfigs        []*host.TestClusterConfig\n\t\tDefaultTestCluster    testcluster.PersistenceTestCluster\n\t\tVisibilityTestCluster testcluster.PersistenceTestCluster\n\t}\n)\n\nfunc NewNDCIntegrationTestSuite(params NDCIntegrationTestSuiteParams) *NDCIntegrationTestSuite {\n\treturn &NDCIntegrationTestSuite{\n\t\tclusterConfigs:        params.ClusterConfigs,\n\t\tdefaultTestCluster:    params.DefaultTestCluster,\n\t\tVisibilityTestCluster: params.VisibilityTestCluster,\n\t}\n}\n"
  },
  {
    "path": "host/onebox.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/uber-go/tally\"\n\t\"github.com/uber-go/tally/prometheus\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\tcwsc \"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/compatibility\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\n\tadminClient \"github.com/uber/cadence/client/admin\"\n\tfrontendClient \"github.com/uber/cadence/client/frontend\"\n\thistoryClient \"github.com/uber/cadence/client/history\"\n\tmatchingClient \"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\tcarchiver \"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcc \"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/metered\"\n\t\"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend\"\n\t\"github.com/uber/cadence/service/history\"\n\t\"github.com/uber/cadence/service/matching\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\tsdconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/worker\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n\t\"github.com/uber/cadence/service/worker/asyncworkflow\"\n\t\"github.com/uber/cadence/service/worker/indexer\"\n\t\"github.com/uber/cadence/service/worker/replicator\"\n)\n\n// Cadence hosts all of cadence services in one process\ntype Cadence interface {\n\tStart() error\n\tStop()\n\tGetAdminClient() adminClient.Client\n\tGetFrontendClient() frontendClient.Client\n\tFrontendHost() membership.HostInfo\n\tGetHistoryClient() historyClient.Client\n\tGetMatchingClient() matchingClient.Client\n\tGetMatchingClients() []matchingClient.Client\n\tGetExecutionManagerFactory() persistence.ExecutionManagerFactory\n}\n\ntype (\n\tcadenceImpl struct {\n\t\tfrontendService  common.Daemon\n\t\tmatchingServices []common.Daemon\n\t\thistoryServices  []common.Daemon\n\n\t\tadminClient                   adminClient.Client\n\t\tfrontendClient                frontendClient.Client\n\t\thistoryClient                 historyClient.Client\n\t\tmatchingClients               []matchingClient.Client\n\t\tlogger                        log.Logger\n\t\tclusterMetadata               cluster.Metadata\n\t\tpersistenceConfig             config.Persistence\n\t\tmessagingClient               messaging.Client\n\t\tdomainManager                 persistence.DomainManager\n\t\thistoryV2Mgr                  persistence.HistoryManager\n\t\texecutionMgrFactory           persistence.ExecutionManagerFactory\n\t\tdomainReplicationQueue        domain.ReplicationQueue\n\t\tshutdownCh                    chan struct{}\n\t\tshutdownWG                    sync.WaitGroup\n\t\tclusterNo                     int // cluster number\n\t\treplicator                    *replicator.Replicator\n\t\tclientWorker                  archiver.ClientWorker\n\t\tindexer                       *indexer.Indexer\n\t\tarchiverMetadata              carchiver.ArchivalMetadata\n\t\tarchiverProvider              provider.ArchiverProvider\n\t\thistoryConfig                 *HistoryConfig\n\t\tmatchingConfig                *MatchingConfig\n\t\tesConfig                      *config.ElasticSearchConfig\n\t\tesClient                      elasticsearch.GenericClient\n\t\tworkerConfig                  *WorkerConfig\n\t\tmockAdminClient               map[string]adminClient.Client\n\t\tdomainReplicationTaskExecutor domain.ReplicationTaskExecutor\n\t\tauthorizationConfig           config.Authorization\n\t\tpinotConfig                   *config.PinotVisibilityConfig\n\t\tpinotClient                   pinot.GenericClient\n\t\tasyncWFQueues                 map[string]config.AsyncWorkflowQueueProvider\n\t\ttimeSource                    clock.TimeSource\n\t\tdynamicClient                 dynamicconfig.Client\n\n\t\t// dynamicconfig overrides per service\n\t\tfrontendDynCfgOverrides map[dynamicproperties.Key]interface{}\n\t\thistoryDynCfgOverrides  map[dynamicproperties.Key]interface{}\n\t\tmatchingDynCfgOverrides map[dynamicproperties.Key]interface{}\n\t\tworkerDynCfgOverrides   map[dynamicproperties.Key]interface{}\n\t}\n\n\t// HistoryConfig contains configs for history service\n\tHistoryConfig struct {\n\t\t// When MockClient is set, rest of the configs are ignored, history service is not started\n\t\t// and mock history client is passed to other services\n\t\tMockClient HistoryClient\n\n\t\tNumHistoryShards       int\n\t\tNumHistoryHosts        int\n\t\tHistoryCountLimitError int\n\t\tHistoryCountLimitWarn  int\n\t\tSimulationConfig       HistorySimulationConfig\n\t}\n\n\tHistorySimulationConfig struct {\n\t\tNumWorkflows int\n\t\t// WorkflowDeletionJitterRange defines the duration in minutes for workflow close tasks jittering\n\t\t// defaults to 0 to remove jittering\n\t\tWorkflowDeletionJitterRange int\n\t\t// EnableTransferQueueV2 enables queue v2 framework for transfer queue\n\t\tEnableTransferQueueV2 bool\n\t\t// EnableTimerQueueV2 enables queue v2 framework for timer queue\n\t\tEnableTimerQueueV2 bool\n\t}\n\n\tMatchingConfig struct {\n\t\t// number of matching host can be at most 4 due to existing static port assignments in onebox.go.\n\t\t// can be changed easily.\n\t\tNumMatchingHosts int\n\t\tSimulationConfig MatchingSimulationConfig\n\t}\n\n\tMatchingSimulationConfig struct {\n\t\t// Number of task list write partitions defaults to 1\n\t\tTaskListWritePartitions int\n\n\t\t// Number of task list read partitions defaults to 1\n\t\tTaskListReadPartitions int\n\n\t\t// At most N polls will be forwarded at a time. defaults to 20\n\t\tForwarderMaxOutstandingPolls int\n\n\t\t// At most N tasks will be forwarded at a time. defaults to 1\n\t\tForwarderMaxOutstandingTasks int\n\n\t\t// Forwarder rps limit defaults to 10\n\t\tForwarderMaxRatePerSecond int\n\n\t\t// Children per node. defaults to 20\n\t\tForwarderMaxChildrenPerNode int\n\n\t\t// LocalPollWaitTime. defaults to 0ms.\n\t\tLocalPollWaitTime time.Duration\n\n\t\t// LocalTaskWaitTime. defaults to 0ms.\n\t\tLocalTaskWaitTime time.Duration\n\n\t\t// RecordDecisionTaskStartedTime. The amount of time spent by History to complete RecordDecisionTaskStarted\n\t\tRecordDecisionTaskStartedTime time.Duration\n\n\t\t// TasklistLoadBalancerStrategy the strategy of load balancer. defaults to \"random\".\n\t\tTasklistLoadBalancerStrategy string\n\n\t\t// The pollers that will be created to process\n\t\tPollers []SimulationPollerConfiguration\n\n\t\tTasks []SimulationTaskConfiguration\n\n\t\tBacklogs []SimulationBacklogConfiguration\n\n\t\t// GetPartitionConfigFromDB indicates whether to get the partition config from DB or not.\n\t\t// This is a prerequisite for adaptive scaler.\n\t\tGetPartitionConfigFromDB bool\n\n\t\t// Adaptive scaler configurations\n\t\tEnableAdaptiveScaler                bool\n\t\tPartitionDownscaleFactor            float64\n\t\tPartitionUpscaleRPS                 int\n\t\tPartitionUpscaleSustainedDuration   time.Duration\n\t\tPartitionDownscaleSustainedDuration time.Duration\n\t\tAdaptiveScalerUpdateInterval        time.Duration\n\t\tQPSTrackerInterval                  time.Duration\n\t\tTaskIsolationDuration               time.Duration\n\t}\n\n\tSimulationPollerConfiguration struct {\n\t\t// The isolation group that pollers will be created with. Optional.\n\t\tIsolationGroup string\n\t\t// The number of pollers that will be created with this configuration. Defaults to 1\n\t\tNumPollers int\n\t\t// TaskProcessTime. The amount of time spent by the poller in-between requests. Defaults to 1ms\n\t\tTaskProcessTime time.Duration\n\t\t// Poll request timeout defaults to 15 seconds\n\t\tPollTimeout time.Duration\n\t}\n\n\tSimulationTaskConfiguration struct {\n\t\t// The isolation groups that tasks will be evenly distributed between\n\t\tIsolationGroups []string\n\n\t\t// Number of task generators defaults to 1\n\t\tNumTaskGenerators int\n\n\t\t// Upper limit of tasks to generate. Task generators will stop if total number of tasks generated reaches MaxTaskToGenerate during simulation\n\t\t// Defaults to 2k\n\t\tMaxTaskToGenerate int\n\n\t\t// Task generation QPS. Defaults to 40.\n\t\tTasksPerSecond int\n\n\t\t// The burst value for the rate limiter for task generation. Controls the maximum number of AddTask requests\n\t\t// that can be sent concurrently. For example, if you have TasksPerSecond, TasksBurst, and NumTaskGenerators all\n\t\t// set to 10 then every second you'll get 10 tasks added right at the start of the second. If you instead set\n\t\t// TasksBurst to 1 then you'd get a steady stream of tasks, with one task every 100ms.\n\t\tTasksBurst int\n\n\t\t// OverTime is a list of TasksProduceSpec that will be used to change the qps over time.\n\t\t// Each item has a duration and they will be applied in the given order.\n\t\t// If this is set, TasksPerSecond and TasksBurst will be ignored.\n\t\tOverTime []TasksProduceSpec\n\t}\n\n\tTasksProduceSpec struct {\n\t\t// Task generation qps\n\t\tTasksPerSecond int\n\n\t\t// The burst value for the rate limiter for task generation.\n\t\tTasksBurst int\n\n\t\t// The duration for which the settings will be applied.\n\t\t// If the duration is unset, the settings will be applied indefinitely.\n\t\tDuration *time.Duration\n\t}\n\n\tSimulationBacklogConfiguration struct {\n\t\t// The partition number\n\t\tPartition int // Do not set it to 0, because it's not guaranteed to add backlog to partition 0\n\t\t// The backlog count\n\t\tBacklogCount int\n\t\t// The weight of each isolation group, can be empty\n\t\tIsolationGroups map[string]int\n\t}\n\n\t// CadenceParams contains everything needed to bootstrap Cadence\n\tCadenceParams struct {\n\t\tClusterMetadata               cluster.Metadata\n\t\tPersistenceConfig             config.Persistence\n\t\tMessagingClient               messaging.Client\n\t\tDomainManager                 persistence.DomainManager\n\t\tHistoryV2Mgr                  persistence.HistoryManager\n\t\tExecutionMgrFactory           persistence.ExecutionManagerFactory\n\t\tDomainReplicationQueue        domain.ReplicationQueue\n\t\tLogger                        log.Logger\n\t\tClusterNo                     int\n\t\tArchiverMetadata              carchiver.ArchivalMetadata\n\t\tArchiverProvider              provider.ArchiverProvider\n\t\tEnableReadHistoryFromArchival bool\n\t\tHistoryConfig                 *HistoryConfig\n\t\tMatchingConfig                *MatchingConfig\n\t\tESConfig                      *config.ElasticSearchConfig\n\t\tESClient                      elasticsearch.GenericClient\n\t\tWorkerConfig                  *WorkerConfig\n\t\tMockAdminClient               map[string]adminClient.Client\n\t\tDomainReplicationTaskExecutor domain.ReplicationTaskExecutor\n\t\tAuthorizationConfig           config.Authorization\n\t\tPinotConfig                   *config.PinotVisibilityConfig\n\t\tPinotClient                   pinot.GenericClient\n\t\tAsyncWFQueues                 map[string]config.AsyncWorkflowQueueProvider\n\t\tTimeSource                    clock.TimeSource\n\t\tDynamicClient                 dynamicconfig.Client\n\n\t\tFrontendDynCfgOverrides map[dynamicproperties.Key]interface{}\n\t\tHistoryDynCfgOverrides  map[dynamicproperties.Key]interface{}\n\t\tMatchingDynCfgOverrides map[dynamicproperties.Key]interface{}\n\t\tWorkerDynCfgOverrides   map[dynamicproperties.Key]interface{}\n\t}\n)\n\n// NewCadence returns an instance that hosts full cadence in one process\nfunc NewCadence(params *CadenceParams) Cadence {\n\treturn &cadenceImpl{\n\t\tlogger:                        params.Logger,\n\t\tclusterMetadata:               params.ClusterMetadata,\n\t\tpersistenceConfig:             params.PersistenceConfig,\n\t\tmessagingClient:               params.MessagingClient,\n\t\tdomainManager:                 params.DomainManager,\n\t\thistoryV2Mgr:                  params.HistoryV2Mgr,\n\t\texecutionMgrFactory:           params.ExecutionMgrFactory,\n\t\tdomainReplicationQueue:        params.DomainReplicationQueue,\n\t\tshutdownCh:                    make(chan struct{}),\n\t\tclusterNo:                     params.ClusterNo,\n\t\tesConfig:                      params.ESConfig,\n\t\tesClient:                      params.ESClient,\n\t\tarchiverMetadata:              params.ArchiverMetadata,\n\t\tarchiverProvider:              params.ArchiverProvider,\n\t\thistoryConfig:                 params.HistoryConfig,\n\t\tmatchingConfig:                params.MatchingConfig,\n\t\tworkerConfig:                  params.WorkerConfig,\n\t\tmockAdminClient:               params.MockAdminClient,\n\t\tdomainReplicationTaskExecutor: params.DomainReplicationTaskExecutor,\n\t\tauthorizationConfig:           params.AuthorizationConfig,\n\t\tpinotConfig:                   params.PinotConfig,\n\t\tpinotClient:                   params.PinotClient,\n\t\tasyncWFQueues:                 params.AsyncWFQueues,\n\t\ttimeSource:                    params.TimeSource,\n\t\tdynamicClient:                 params.DynamicClient,\n\t\tfrontendDynCfgOverrides:       params.FrontendDynCfgOverrides,\n\t\thistoryDynCfgOverrides:        params.HistoryDynCfgOverrides,\n\t\tmatchingDynCfgOverrides:       params.MatchingDynCfgOverrides,\n\t\tworkerDynCfgOverrides:         params.WorkerDynCfgOverrides,\n\t}\n}\n\nfunc (c *cadenceImpl) enableWorker() bool {\n\treturn c.workerConfig.EnableArchiver || c.workerConfig.EnableIndexer || c.workerConfig.EnableReplicator || c.workerConfig.EnableAsyncWFConsumer\n}\n\nfunc (c *cadenceImpl) Start() error {\n\thosts := make(map[string][]membership.HostInfo)\n\thosts[service.Frontend] = []membership.HostInfo{c.FrontendHost()}\n\thosts[service.Matching] = c.MatchingHosts()\n\thosts[service.History] = c.HistoryHosts()\n\tif c.enableWorker() {\n\t\thosts[service.Worker] = []membership.HostInfo{c.WorkerServiceHost()}\n\t}\n\n\t// create cadence-system domain, this must be created before starting\n\t// the services - so directly use the metadataManager to create this\n\tif err := c.createSystemDomain(); err != nil {\n\t\treturn err\n\t}\n\n\tvar startWG sync.WaitGroup\n\tstartWG.Add(1)\n\tgo c.startHistory(hosts, &startWG)\n\tstartWG.Wait()\n\n\tstartWG.Add(1)\n\tgo c.startMatching(hosts, &startWG)\n\tstartWG.Wait()\n\n\tstartWG.Add(1)\n\tgo c.startFrontend(hosts, &startWG)\n\tstartWG.Wait()\n\n\tif c.enableWorker() {\n\t\tstartWG.Add(1)\n\t\tgo c.startWorker(hosts, &startWG)\n\t\tstartWG.Wait()\n\t}\n\n\treturn nil\n}\n\nfunc (c *cadenceImpl) Stop() {\n\tserviceCount := 3\n\tif c.enableWorker() {\n\t\tserviceCount++\n\t}\n\n\tc.shutdownWG.Add(serviceCount)\n\tc.frontendService.Stop()\n\tfor _, historyService := range c.historyServices {\n\t\thistoryService.Stop()\n\t}\n\tfor _, matchingService := range c.matchingServices {\n\t\tmatchingService.Stop()\n\t}\n\n\tif c.workerConfig.EnableReplicator {\n\t\tc.replicator.Stop()\n\t}\n\tif c.workerConfig.EnableArchiver {\n\t\tc.clientWorker.Stop()\n\t}\n\tclose(c.shutdownCh)\n\tc.shutdownWG.Wait()\n}\n\nfunc newHost(tchan uint16) membership.HostInfo {\n\taddress := \"127.0.0.1\"\n\treturn membership.NewDetailedHostInfo(\n\t\tfmt.Sprintf(\"%s:%d\", address, tchan),\n\t\tfmt.Sprintf(\"%s_%d\", address, tchan),\n\t\tmembership.PortMap{\n\t\t\tmembership.PortTchannel: tchan,\n\t\t\tmembership.PortGRPC:     tchan + 10,\n\t\t},\n\t)\n\n}\n\nfunc (c *cadenceImpl) FrontendHost() membership.HostInfo {\n\tvar tchan uint16\n\tswitch c.clusterNo {\n\tcase 0:\n\t\ttchan = 7104\n\tcase 1:\n\t\ttchan = 8104\n\tcase 2:\n\t\ttchan = 9104\n\tcase 3:\n\t\ttchan = 10104\n\tdefault:\n\t\ttchan = 7104\n\t}\n\n\treturn newHost(tchan)\n\n}\n\nfunc (c *cadenceImpl) FrontendPProfPort() int {\n\tswitch c.clusterNo {\n\tcase 0:\n\t\treturn 7105\n\tcase 1:\n\t\treturn 8105\n\tcase 2:\n\t\treturn 9105\n\tcase 3:\n\t\treturn 10105\n\tdefault:\n\t\treturn 7105\n\t}\n}\n\nfunc (c *cadenceImpl) HistoryHosts() []membership.HostInfo {\n\tvar hosts []membership.HostInfo\n\tvar startPort int\n\tswitch c.clusterNo {\n\tcase 0:\n\t\tstartPort = 7201\n\tcase 1:\n\t\tstartPort = 8201\n\tcase 2:\n\t\tstartPort = 9201\n\tcase 3:\n\t\tstartPort = 10201\n\tdefault:\n\t\tstartPort = 7201\n\t}\n\tfor i := 0; i < c.historyConfig.NumHistoryHosts; i++ {\n\t\tport := startPort + i\n\t\thosts = append(hosts, newHost(uint16(port)))\n\t}\n\n\tc.logger.Info(\"History hosts\", tag.Value(hosts))\n\treturn hosts\n}\n\nfunc (c *cadenceImpl) HistoryPProfPorts() []int {\n\tvar ports []int\n\tvar startPort int\n\tswitch c.clusterNo {\n\tcase 0:\n\t\tstartPort = 7301\n\tcase 1:\n\t\tstartPort = 8301\n\tcase 2:\n\t\tstartPort = 9301\n\tcase 3:\n\t\tstartPort = 10301\n\tdefault:\n\t\tstartPort = 7301\n\t}\n\tfor i := 0; i < c.historyConfig.NumHistoryHosts; i++ {\n\t\tport := startPort + i\n\t\tports = append(ports, port)\n\t}\n\n\tc.logger.Info(\"History pprof ports\", tag.Value(ports))\n\treturn ports\n}\n\nfunc (c *cadenceImpl) MatchingHosts() []membership.HostInfo {\n\tvar hosts []membership.HostInfo\n\tvar tchan uint16\n\tswitch c.clusterNo {\n\tcase 0:\n\t\ttchan = 7106\n\tcase 1:\n\t\ttchan = 8106\n\tcase 2:\n\t\ttchan = 9106\n\tcase 3:\n\t\ttchan = 10106\n\tdefault:\n\t\ttchan = 7106\n\t}\n\n\tfor i := 0; i < c.matchingConfig.NumMatchingHosts; i++ {\n\t\tport := tchan + uint16(i)\n\t\thosts = append(hosts, newHost(uint16(port)))\n\t}\n\n\tc.logger.Info(\"Matching hosts\", tag.Value(hosts))\n\n\treturn hosts\n}\n\nfunc (c *cadenceImpl) MatchingPProfPorts() []int {\n\tvar ports []int\n\tvar startPort int\n\tswitch c.clusterNo {\n\tcase 0:\n\t\tstartPort = 7206\n\tcase 1:\n\t\tstartPort = 8206\n\tcase 2:\n\t\tstartPort = 9206\n\tcase 3:\n\t\tstartPort = 10206\n\tdefault:\n\t\tstartPort = 7206\n\t}\n\n\tfor i := 0; i < c.matchingConfig.NumMatchingHosts; i++ {\n\t\tport := startPort + i\n\t\tports = append(ports, port)\n\t}\n\n\tc.logger.Info(\"Matching pprof ports\", tag.Value(ports))\n\treturn ports\n}\n\nfunc (c *cadenceImpl) MatchingPrometheusPorts() []int {\n\tvar ports []int\n\tvar startPort int\n\tswitch c.clusterNo {\n\tcase 0:\n\t\tstartPort = 7306\n\tcase 1:\n\t\tstartPort = 8306\n\tcase 2:\n\t\tstartPort = 9306\n\tcase 3:\n\t\tstartPort = 10306\n\tdefault:\n\t\tstartPort = 7306\n\t}\n\n\tfor i := 0; i < c.matchingConfig.NumMatchingHosts; i++ {\n\t\tport := startPort + i\n\t\tports = append(ports, port)\n\t}\n\n\tc.logger.Info(\"Matching prometheus ports\", tag.Value(ports))\n\treturn ports\n}\n\nfunc (c *cadenceImpl) WorkerServiceHost() membership.HostInfo {\n\tvar tchan uint16\n\tswitch c.clusterNo {\n\tcase 0:\n\t\ttchan = 7108\n\tcase 1:\n\t\ttchan = 8108\n\tcase 2:\n\t\ttchan = 9108\n\tcase 3:\n\t\ttchan = 10108\n\tdefault:\n\t\ttchan = 7108\n\t}\n\treturn newHost(tchan)\n}\n\nfunc (c *cadenceImpl) WorkerPProfPort() int {\n\tswitch c.clusterNo {\n\tcase 0:\n\t\treturn 7109\n\tcase 1:\n\t\treturn 8109\n\tcase 2:\n\t\treturn 9109\n\tcase 3:\n\t\treturn 10109\n\tdefault:\n\t\treturn 7109\n\t}\n}\n\nfunc (c *cadenceImpl) GetAdminClient() adminClient.Client {\n\treturn c.adminClient\n}\n\nfunc (c *cadenceImpl) GetFrontendClient() frontendClient.Client {\n\treturn c.frontendClient\n}\n\nfunc (c *cadenceImpl) GetHistoryClient() historyClient.Client {\n\treturn c.historyClient\n}\n\nfunc (c *cadenceImpl) GetMatchingClient() matchingClient.Client {\n\treturn c.matchingClients[0]\n}\n\nfunc (c *cadenceImpl) GetMatchingClients() []matchingClient.Client {\n\treturn c.matchingClients\n}\n\nfunc (c *cadenceImpl) startFrontend(hosts map[string][]membership.HostInfo, startWG *sync.WaitGroup) {\n\tparams := new(resource.Params)\n\tparams.ClusterRedirectionPolicy = &config.ClusterRedirectionPolicy{}\n\tparams.Name = service.Frontend\n\tparams.Logger = c.logger\n\tparams.ThrottledLogger = c.logger\n\tparams.TimeSource = c.timeSource\n\tparams.PProfInitializer = newPProfInitializerImpl(c.logger, c.FrontendPProfPort())\n\tparams.MetricScope = tally.NewTestScope(service.Frontend, make(map[string]string))\n\tparams.MetricsClient = metrics.NewClient(params.MetricScope, service.GetMetricsServiceIdx(params.Name, c.logger), metrics.HistogramMigration{} /* default, only used in test setups */)\n\tparams.RPCFactory = c.newRPCFactory(service.Frontend, c.FrontendHost(), params.MetricsClient)\n\tparams.MembershipResolver = newMembershipResolver(params.Name, hosts, c.FrontendHost())\n\tparams.ClusterMetadata = c.clusterMetadata\n\tparams.MessagingClient = c.messagingClient\n\tparams.DynamicConfig = newIntegrationConfigClient(c.dynamicClient, c.frontendDynCfgOverrides)\n\tparams.ArchivalMetadata = c.archiverMetadata\n\tparams.ArchiverProvider = c.archiverProvider\n\tparams.ESConfig = c.esConfig\n\tparams.ESClient = c.esClient\n\tparams.PinotConfig = c.pinotConfig\n\tparams.PinotClient = c.pinotClient\n\tparams.GetIsolationGroups = getFromDynamicConfig(params)\n\tvar err error\n\tauthorizer, err := authorization.NewAuthorizer(c.authorizationConfig, params.Logger, nil)\n\tif err != nil {\n\t\tc.logger.Fatal(\"Unable to create authorizer\", tag.Error(err))\n\t}\n\tparams.Authorizer = authorizer\n\tparams.PersistenceConfig, err = copyPersistenceConfig(c.persistenceConfig)\n\tif err != nil {\n\t\tc.logger.Fatal(\"Failed to copy persistence config for frontend\", tag.Error(err))\n\t}\n\n\tif c.pinotConfig != nil {\n\t\tpinotDataStoreName := \"pinot-visibility\"\n\t\tparams.PersistenceConfig.AdvancedVisibilityStore = pinotDataStoreName\n\t\tparams.DynamicConfig.UpdateValue(dynamicproperties.ReadVisibilityStoreName, constants.VisibilityModePinot)\n\t\tparams.PersistenceConfig.DataStores[pinotDataStoreName] = config.DataStore{\n\t\t\tPinot: c.pinotConfig,\n\t\t}\n\t} else if c.esConfig != nil {\n\t\tesDataStoreName := \"es-visibility\"\n\t\tparams.PersistenceConfig.AdvancedVisibilityStore = esDataStoreName\n\t\tparams.PersistenceConfig.DataStores[esDataStoreName] = config.DataStore{\n\t\t\tElasticSearch: c.esConfig,\n\t\t}\n\t}\n\n\tif c.asyncWFQueues != nil {\n\t\tparams.AsyncWorkflowQueueProvider, err = queue.NewAsyncQueueProvider(c.asyncWFQueues)\n\t\tif err != nil {\n\t\t\tc.logger.Fatal(\"error creating async queue provider\", tag.Error(err))\n\t\t}\n\t}\n\n\tfrontendService, err := frontend.NewService(params)\n\tif err != nil {\n\t\tparams.Logger.Fatal(\"unable to start frontend service\", tag.Error(err))\n\t}\n\n\tif c.mockAdminClient != nil {\n\t\tclientBean := frontendService.GetClientBean()\n\t\tif clientBean != nil {\n\t\t\tfor serviceName, client := range c.mockAdminClient {\n\t\t\t\tclientBean.SetRemoteAdminClient(serviceName, client)\n\t\t\t}\n\t\t}\n\t}\n\n\tc.frontendService = frontendService\n\tc.frontendClient = NewFrontendClient(frontendService.GetDispatcher())\n\tc.adminClient = NewAdminClient(frontendService.GetDispatcher())\n\tgo frontendService.Start()\n\n\tc.logger.Info(\"Started frontend service\")\n\tstartWG.Done()\n\n\t<-c.shutdownCh\n\tc.shutdownWG.Done()\n}\n\nfunc (c *cadenceImpl) startHistory(hosts map[string][]membership.HostInfo, startWG *sync.WaitGroup) {\n\tpprofPorts := c.HistoryPProfPorts()\n\thistoryHosts := c.HistoryHosts()\n\tfor i, hostport := range historyHosts {\n\t\tparams := new(resource.Params)\n\t\tparams.Name = service.History\n\t\tparams.Logger = c.logger\n\t\tparams.ThrottledLogger = c.logger\n\t\tparams.TimeSource = c.timeSource\n\t\tparams.PProfInitializer = newPProfInitializerImpl(c.logger, pprofPorts[i])\n\t\tparams.MetricScope = tally.NewTestScope(service.History, make(map[string]string))\n\t\tparams.MetricsClient = metrics.NewClient(params.MetricScope, service.GetMetricsServiceIdx(params.Name, c.logger), metrics.HistogramMigration{} /* default, only used in test setups */)\n\t\tparams.RPCFactory = c.newRPCFactory(service.History, hostport, params.MetricsClient)\n\t\tparams.MembershipResolver = newMembershipResolver(params.Name, hosts, hostport)\n\t\tparams.ClusterMetadata = c.clusterMetadata\n\t\tparams.MessagingClient = c.messagingClient\n\t\tintegrationClient := newIntegrationConfigClient(c.dynamicClient, c.historyDynCfgOverrides)\n\t\tc.overrideHistoryDynamicConfig(integrationClient)\n\t\tparams.DynamicConfig = integrationClient\n\t\tparams.PublicClient = newPublicClient(params.RPCFactory.GetDispatcher())\n\t\tparams.ArchivalMetadata = c.archiverMetadata\n\t\tparams.ArchiverProvider = c.archiverProvider\n\t\tparams.ESConfig = c.esConfig\n\t\tparams.ESClient = c.esClient\n\t\tparams.PinotConfig = c.pinotConfig\n\t\tparams.GetIsolationGroups = getFromDynamicConfig(params)\n\n\t\tvar err error\n\t\tparams.PersistenceConfig, err = copyPersistenceConfig(c.persistenceConfig)\n\t\tif err != nil {\n\t\t\tc.logger.Fatal(\"Failed to copy persistence config for history\", tag.Error(err))\n\t\t}\n\n\t\tif c.pinotConfig != nil {\n\t\t\tpinotDataStoreName := \"pinot-visibility\"\n\t\t\tparams.PersistenceConfig.AdvancedVisibilityStore = pinotDataStoreName\n\t\t\tparams.PersistenceConfig.DataStores[pinotDataStoreName] = config.DataStore{\n\t\t\t\tPinot:         c.pinotConfig,\n\t\t\t\tElasticSearch: c.esConfig,\n\t\t\t}\n\t\t\tparams.DynamicConfig.UpdateValue(dynamicproperties.WriteVisibilityStoreName, constants.VisibilityModePinot)\n\t\t} else if c.esConfig != nil {\n\t\t\tesDataStoreName := \"es-visibility\"\n\t\t\tparams.PersistenceConfig.AdvancedVisibilityStore = esDataStoreName\n\t\t\tparams.PersistenceConfig.DataStores[esDataStoreName] = config.DataStore{\n\t\t\t\tElasticSearch: c.esConfig,\n\t\t\t}\n\t\t}\n\n\t\thistoryService, err := history.NewService(params)\n\t\tif err != nil {\n\t\t\tparams.Logger.Fatal(\"unable to start history service\", tag.Error(err))\n\t\t}\n\n\t\tif c.mockAdminClient != nil {\n\t\t\tclientBean := historyService.GetClientBean()\n\t\t\tif clientBean != nil {\n\t\t\t\tfor serviceName, client := range c.mockAdminClient {\n\t\t\t\t\tclientBean.SetRemoteAdminClient(serviceName, client)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\t// TODO: this is not correct when there are multiple history hosts as later client will overwrite previous ones.\n\t\t// However current interface for getting history client doesn't specify which client it needs and the tests that use this API\n\t\t// depends on the fact that there's only one history host.\n\t\t// Need to change those tests and modify the interface for getting history client.\n\t\tc.historyClient = NewHistoryClient(historyService.GetDispatcher())\n\t\tc.historyServices = append(c.historyServices, historyService)\n\n\t\tgo historyService.Start()\n\t}\n\n\tstartWG.Done()\n\tc.logger.Info(fmt.Sprintf(\"Started %d history services\", len(c.historyServices)))\n\n\t<-c.shutdownCh\n\tc.shutdownWG.Done()\n}\n\nfunc (c *cadenceImpl) startMatching(hosts map[string][]membership.HostInfo, startWG *sync.WaitGroup) {\n\tpprofPorts := c.MatchingPProfPorts()\n\tpromPorts := c.MatchingPrometheusPorts()\n\tfor i, hostport := range c.MatchingHosts() {\n\t\thostport.Identity()\n\t\tmatchingHost := fmt.Sprintf(\"matching-host-%d:%s\", i, hostport.Identity())\n\t\tparams := new(resource.Params)\n\t\tparams.Name = service.Matching\n\t\tparams.Logger = c.logger.WithTags(tag.Dynamic(\"matching-host\", matchingHost))\n\t\tparams.ThrottledLogger = c.logger\n\t\tparams.TimeSource = c.timeSource\n\t\tparams.PProfInitializer = newPProfInitializerImpl(c.logger, pprofPorts[i])\n\t\tmetricsCfg := config.Metrics{\n\t\t\tPrometheus: &prometheus.Configuration{\n\t\t\t\tTimerType:     \"histogram\",\n\t\t\t\tListenAddress: fmt.Sprintf(\":%d\", promPorts[i]),\n\t\t\t},\n\t\t}\n\t\tparams.MetricScope = metricsCfg.NewScope(c.logger, service.Matching)\n\t\tparams.MetricsClient = metrics.NewClient(params.MetricScope, service.GetMetricsServiceIdx(params.Name, c.logger), metrics.HistogramMigration{} /* default, only used in test setups */)\n\t\tparams.RPCFactory = c.newRPCFactory(service.Matching, hostport, params.MetricsClient)\n\t\tparams.MembershipResolver = newMembershipResolver(params.Name, hosts, hostport)\n\t\tparams.ClusterMetadata = c.clusterMetadata\n\t\tparams.DynamicConfig = newIntegrationConfigClient(c.dynamicClient, c.matchingDynCfgOverrides)\n\t\tparams.ArchivalMetadata = c.archiverMetadata\n\t\tparams.ArchiverProvider = c.archiverProvider\n\t\tparams.GetIsolationGroups = getFromDynamicConfig(params)\n\n\t\tvar err error\n\t\tparams.PersistenceConfig, err = copyPersistenceConfig(c.persistenceConfig)\n\t\tif err != nil {\n\t\t\tc.logger.Fatal(\"Failed to copy persistence config for matching\", tag.Error(err))\n\t\t}\n\n\t\tif c.historyConfig.MockClient != nil {\n\t\t\tparams.HistoryClientFn = func() historyClient.Client {\n\t\t\t\treturn c.historyConfig.MockClient\n\t\t\t}\n\t\t}\n\n\t\t// Set default ShardDistributorMatchingConfig for integration tests\n\t\tparams.ShardDistributorMatchingConfig = clientcommon.Config{\n\t\t\tNamespaces: []clientcommon.NamespaceConfig{{\n\t\t\t\tNamespace:         \"cadence-matching-integration\",\n\t\t\t\tHeartBeatInterval: 1 * time.Second,\n\t\t\t\tMigrationMode:     sdconfig.MigrationModeLOCALPASSTHROUGH,\n\t\t\t\tTTLShard:          5 * time.Minute,\n\t\t\t\tTTLReport:         1 * time.Minute,\n\t\t\t}},\n\t\t}\n\n\t\tmatchingService, err := matching.NewService(params)\n\t\tif err != nil {\n\t\t\tparams.Logger.Fatal(\"unable to start matching service\", tag.Error(err))\n\t\t}\n\n\t\tclientBean := matchingService.GetClientBean()\n\t\tif c.mockAdminClient != nil {\n\t\t\tfor serviceName, client := range c.mockAdminClient {\n\t\t\t\tclientBean.SetRemoteAdminClient(serviceName, client)\n\t\t\t}\n\t\t}\n\n\t\t// When there are multiple matching hosts the last client will overwrite previous ones.\n\t\t// It should be fine because the underlying client bean logic should still pick the right destination.\n\t\tmatchingClient, err := clientBean.GetMatchingClient(matchingService.GetDomainCache().GetDomainName)\n\t\tif err != nil {\n\t\t\tparams.Logger.Fatal(\"unable to get matching client\", tag.Error(err))\n\t\t}\n\t\tc.matchingClients = append(c.matchingClients, matchingClient)\n\t\tc.matchingServices = append(c.matchingServices, matchingService)\n\t\tgo matchingService.Start()\n\t}\n\n\tstartWG.Done()\n\tc.logger.Info(fmt.Sprintf(\"Started %d matching services\", len(c.matchingServices)))\n\n\t<-c.shutdownCh\n\tc.shutdownWG.Done()\n}\n\nfunc (c *cadenceImpl) startWorker(hosts map[string][]membership.HostInfo, startWG *sync.WaitGroup) {\n\tdefer c.shutdownWG.Done()\n\n\tparams := new(resource.Params)\n\tparams.Name = service.Worker\n\tparams.Logger = c.logger\n\tparams.ThrottledLogger = c.logger\n\tparams.TimeSource = c.timeSource\n\tparams.PProfInitializer = newPProfInitializerImpl(c.logger, c.WorkerPProfPort())\n\tparams.MetricScope = tally.NewTestScope(service.Worker, make(map[string]string))\n\tparams.MetricsClient = metrics.NewClient(params.MetricScope, service.GetMetricsServiceIdx(params.Name, c.logger), metrics.HistogramMigration{} /* default, only used in test setups */)\n\tparams.RPCFactory = c.newRPCFactory(service.Worker, c.WorkerServiceHost(), params.MetricsClient)\n\tparams.MembershipResolver = newMembershipResolver(params.Name, hosts, c.WorkerServiceHost())\n\tparams.ClusterMetadata = c.clusterMetadata\n\tparams.DynamicConfig = newIntegrationConfigClient(c.dynamicClient, c.workerDynCfgOverrides)\n\tparams.ArchivalMetadata = c.archiverMetadata\n\tparams.ArchiverProvider = c.archiverProvider\n\tparams.GetIsolationGroups = getFromDynamicConfig(params)\n\n\tvar err error\n\tparams.PersistenceConfig, err = copyPersistenceConfig(c.persistenceConfig)\n\tif err != nil {\n\t\tc.logger.Fatal(\"Failed to copy persistence config for worker\", tag.Error(err))\n\t}\n\tparams.PublicClient = newPublicClient(params.RPCFactory.GetDispatcher())\n\tservice := NewService(params)\n\tservice.Start()\n\n\tvar replicatorDomainCache cache.DomainCache\n\tif c.workerConfig.EnableReplicator {\n\t\tmetadataManager := metered.NewDomainManager(c.domainManager, service.GetMetricsClient(), c.logger, &c.persistenceConfig)\n\t\treplicatorDomainCache = cache.NewDomainCache(metadataManager, c.clusterMetadata, service.GetMetricsClient(), service.GetLogger())\n\t\treplicatorDomainCache.Start()\n\t\tdefer replicatorDomainCache.Stop()\n\t\tc.startWorkerReplicator(service)\n\t}\n\n\tvar clientWorkerDomainCache cache.DomainCache\n\tif c.workerConfig.EnableArchiver {\n\t\tmetadataProxyManager := metered.NewDomainManager(c.domainManager, service.GetMetricsClient(), c.logger, &c.persistenceConfig)\n\t\tclientWorkerDomainCache = cache.NewDomainCache(metadataProxyManager, c.clusterMetadata, service.GetMetricsClient(), service.GetLogger())\n\t\tclientWorkerDomainCache.Start()\n\t\tdefer clientWorkerDomainCache.Stop()\n\t\tc.startWorkerClientWorker(params, service, clientWorkerDomainCache)\n\t}\n\n\tif c.workerConfig.EnableIndexer {\n\t\tc.startWorkerIndexer(params, service)\n\t}\n\n\tvar asyncWFDomainCache cache.DomainCache\n\tif c.workerConfig.EnableAsyncWFConsumer {\n\t\tqueueProvider, err := queue.NewAsyncQueueProvider(c.asyncWFQueues)\n\t\tif err != nil {\n\t\t\tc.logger.Fatal(\"error creating async queue provider\", tag.Error(err))\n\t\t}\n\n\t\tmetadataProxyManager := metered.NewDomainManager(\n\t\t\tc.domainManager,\n\t\t\tservice.GetMetricsClient(),\n\t\t\tc.logger,\n\t\t\t&c.persistenceConfig)\n\t\tasyncWFDomainCache = cache.NewDomainCache(\n\t\t\tmetadataProxyManager,\n\t\t\tc.clusterMetadata,\n\t\t\tservice.GetMetricsClient(),\n\t\t\tservice.GetLogger(),\n\t\t\tcache.WithTimeSource(params.TimeSource))\n\t\tasyncWFDomainCache.Start()\n\t\tdefer asyncWFDomainCache.Stop()\n\t\tcm := asyncworkflow.NewConsumerManager(\n\t\t\tservice.GetLogger(),\n\t\t\tservice.GetMetricsClient(),\n\t\t\tasyncWFDomainCache,\n\t\t\tqueueProvider,\n\t\t\tc.frontendClient,\n\t\t\tasyncworkflow.WithTimeSource(params.TimeSource),\n\t\t\tasyncworkflow.WithRefreshInterval(time.Second),\n\t\t)\n\t\tcm.Start()\n\t\tdefer cm.Stop()\n\t}\n\n\tc.logger.Info(\"Started worker service\")\n\tstartWG.Done()\n\n\t<-c.shutdownCh\n\tif c.workerConfig.EnableReplicator {\n\t\treplicatorDomainCache.Stop()\n\t}\n\tif c.workerConfig.EnableArchiver {\n\t\tclientWorkerDomainCache.Stop()\n\t}\n}\n\nfunc (c *cadenceImpl) startWorkerReplicator(svc Service) {\n\tc.replicator = replicator.NewReplicator(\n\t\tc.clusterMetadata,\n\t\tsvc.GetClientBean(),\n\t\tc.logger,\n\t\tsvc.GetMetricsClient(),\n\t\tsvc.GetHostInfo(),\n\t\tsvc.GetMembershipResolver(),\n\t\tc.domainReplicationQueue,\n\t\tc.domainReplicationTaskExecutor,\n\t\ttime.Millisecond,\n\t)\n\tif err := c.replicator.Start(); err != nil {\n\t\tc.replicator.Stop()\n\t\tc.logger.Fatal(\"Fail to start replicator when start worker\", tag.Error(err))\n\t}\n}\n\nfunc (c *cadenceImpl) startWorkerClientWorker(params *resource.Params, svc Service, domainCache cache.DomainCache) {\n\tworkerConfig := worker.NewConfig(params)\n\tworkerConfig.ArchiverConfig.ArchiverConcurrency = dynamicproperties.GetIntPropertyFn(10)\n\thistoryArchiverBootstrapContainer := &carchiver.HistoryBootstrapContainer{\n\t\tHistoryV2Manager: c.historyV2Mgr,\n\t\tLogger:           c.logger,\n\t\tMetricsClient:    svc.GetMetricsClient(),\n\t\tClusterMetadata:  c.clusterMetadata,\n\t\tDomainCache:      domainCache,\n\t}\n\terr := c.archiverProvider.RegisterBootstrapContainer(service.Worker, historyArchiverBootstrapContainer, &carchiver.VisibilityBootstrapContainer{})\n\tif err != nil {\n\t\tc.logger.Fatal(\"Failed to register archiver bootstrap container for worker service\", tag.Error(err))\n\t}\n\n\tbc := &archiver.BootstrapContainer{\n\t\tPublicClient:     params.PublicClient,\n\t\tMetricsClient:    svc.GetMetricsClient(),\n\t\tLogger:           c.logger,\n\t\tHistoryV2Manager: c.historyV2Mgr,\n\t\tDomainCache:      domainCache,\n\t\tConfig:           workerConfig.ArchiverConfig,\n\t\tArchiverProvider: c.archiverProvider,\n\t}\n\tc.clientWorker = archiver.NewClientWorker(bc)\n\tif err := c.clientWorker.Start(); err != nil {\n\t\tc.clientWorker.Stop()\n\t\tc.logger.Fatal(\"Fail to start archiver when start worker\", tag.Error(err))\n\t}\n}\n\nfunc (c *cadenceImpl) startWorkerIndexer(params *resource.Params, service Service) {\n\tparams.DynamicConfig.UpdateValue(dynamicproperties.WriteVisibilityStoreName, constants.VisibilityModeES)\n\tworkerConfig := worker.NewConfig(params)\n\tc.indexer = indexer.NewIndexer(\n\t\tworkerConfig.IndexerCfg,\n\t\tc.messagingClient,\n\t\tc.esClient,\n\t\tc.esConfig.Indices[constants.VisibilityAppName],\n\t\tc.esConfig.ConsumerName,\n\t\tc.logger,\n\t\tservice.GetMetricsClient())\n\tif err := c.indexer.Start(); err != nil {\n\t\tc.indexer.Stop()\n\t\tc.logger.Fatal(\"Fail to start indexer when start worker\", tag.Error(err))\n\t}\n}\n\nfunc (c *cadenceImpl) createSystemDomain() error {\n\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\tdefer cancel()\n\n\t_, err := c.domainManager.CreateDomain(ctx, &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          uuid.New(),\n\t\t\tName:        \"cadence-system\",\n\t\t\tStatus:      persistence.DomainStatusRegistered,\n\t\t\tDescription: \"Cadence system domain\",\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{},\n\t\tFailoverVersion:   constants.EmptyVersion,\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*types.DomainAlreadyExistsError); ok {\n\t\t\treturn nil\n\t\t}\n\t\treturn fmt.Errorf(\"failed to create cadence-system domain: %v\", err)\n\t}\n\treturn nil\n}\n\nfunc (c *cadenceImpl) GetExecutionManagerFactory() persistence.ExecutionManagerFactory {\n\treturn c.executionMgrFactory\n}\n\nfunc (c *cadenceImpl) overrideHistoryDynamicConfig(client *dynamicClient) {\n\tclient.OverrideValue(dynamicproperties.ReplicationTaskProcessorStartWait, time.Nanosecond)\n\n\tif c.workerConfig.EnableIndexer {\n\t\tclient.OverrideValue(dynamicproperties.WriteVisibilityStoreName, constants.VisibilityModeES)\n\t}\n\tif c.historyConfig.HistoryCountLimitWarn != 0 {\n\t\tclient.OverrideValue(dynamicproperties.HistoryCountLimitWarn, c.historyConfig.HistoryCountLimitWarn)\n\t}\n\tif c.historyConfig.HistoryCountLimitError != 0 {\n\t\tclient.OverrideValue(dynamicproperties.HistoryCountLimitError, c.historyConfig.HistoryCountLimitError)\n\t}\n}\n\n// copyPersistenceConfig makes a deepcopy of persistence config.\n// This is just a temp fix for the race condition of persistence config.\n// The race condition happens because all the services are using the same datastore map in the config.\n// Also all services will retry to modify the maxQPS field in the datastore during start up and use the modified maxQPS value to create a persistence factory.\nfunc copyPersistenceConfig(pConfig config.Persistence) (config.Persistence, error) {\n\tcopiedDataStores := make(map[string]config.DataStore)\n\tfor name, value := range pConfig.DataStores {\n\t\tcopiedDataStore := config.DataStore{}\n\t\tencodedDataStore, err := json.Marshal(value)\n\t\tif err != nil {\n\t\t\treturn pConfig, err\n\t\t}\n\n\t\tif err = json.Unmarshal(encodedDataStore, &copiedDataStore); err != nil {\n\t\t\treturn pConfig, err\n\t\t}\n\t\tcopiedDataStores[name] = copiedDataStore\n\t}\n\tpConfig.DataStores = copiedDataStores\n\treturn pConfig, nil\n}\n\nfunc newMembershipResolver(serviceName string, hosts map[string][]membership.HostInfo, currentHost membership.HostInfo) membership.Resolver {\n\treturn NewSimpleResolver(serviceName, hosts, currentHost)\n}\n\nfunc newPProfInitializerImpl(logger log.Logger, port int) common.PProfInitializer {\n\treturn &config.PProfInitializerImpl{\n\t\tPProf: &config.PProf{\n\t\t\tPort: port,\n\t\t},\n\t\tLogger: logger,\n\t}\n}\n\nfunc newPublicClient(dispatcher *yarpc.Dispatcher) cwsc.Interface {\n\tconfig := dispatcher.ClientConfig(rpc.OutboundPublicClient)\n\treturn compatibility.NewThrift2ProtoAdapter(\n\t\tapiv1.NewDomainAPIYARPCClient(config),\n\t\tapiv1.NewWorkflowAPIYARPCClient(config),\n\t\tapiv1.NewWorkerAPIYARPCClient(config),\n\t\tapiv1.NewVisibilityAPIYARPCClient(config),\n\t)\n}\n\nfunc (c *cadenceImpl) newRPCFactory(serviceName string, host membership.HostInfo, metricsCl metrics.Client) rpc.Factory {\n\ttchannelAddress, err := host.GetNamedAddress(membership.PortTchannel)\n\tif err != nil {\n\t\tc.logger.Fatal(\"failed to get PortTchannel port from host\", tag.Value(host), tag.Error(err))\n\t}\n\n\tgrpcAddress, err := host.GetNamedAddress(membership.PortGRPC)\n\tif err != nil {\n\t\tc.logger.Fatal(\"failed to get PortGRPC port from host\", tag.Value(host), tag.Error(err))\n\t}\n\n\tfrontendGrpcAddress, err := c.FrontendHost().GetNamedAddress(membership.PortGRPC)\n\tif err != nil {\n\t\tc.logger.Fatal(\"failed to get frontend PortGRPC\", tag.Value(c.FrontendHost()), tag.Error(err))\n\t}\n\n\tdirectOutboundPCF := rpc.NewDirectPeerChooserFactory(serviceName, c.logger, metricsCl)\n\tdirectConnRetainFn := func(opts ...dynamicproperties.FilterOption) bool { return false }\n\n\treturn rpc.NewFactory(c.logger, rpc.Params{\n\t\tServiceName:     serviceName,\n\t\tTChannelAddress: tchannelAddress,\n\t\tGRPCAddress:     grpcAddress,\n\t\tInboundMiddleware: yarpc.InboundMiddleware{\n\t\t\tUnary: yarpc.UnaryInboundMiddleware(&versionMiddleware{}, &rpc.ClientPartitionConfigMiddleware{}, &rpc.ForwardPartitionConfigMiddleware{}),\n\t\t},\n\t\tOutboundMiddleware: yarpc.OutboundMiddleware{\n\t\t\tUnary: &rpc.ForwardPartitionConfigMiddleware{},\n\t\t},\n\n\t\t// For integration tests to generate client out of the same outbound.\n\t\tOutboundsBuilder: rpc.CombineOutbounds(\n\t\t\trpc.NewSingleGRPCOutboundBuilder(testOutboundName(serviceName), serviceName, grpcAddress),\n\t\t\trpc.NewSingleGRPCOutboundBuilder(rpc.OutboundPublicClient, service.Frontend, frontendGrpcAddress),\n\t\t\trpc.NewCrossDCOutbounds(c.clusterMetadata.GetAllClusterInfo(), rpc.NewDNSPeerChooserFactory(0, c.logger)),\n\t\t\trpc.NewDirectOutboundBuilder(service.History, true, nil, directOutboundPCF, directConnRetainFn),\n\t\t\trpc.NewDirectOutboundBuilder(service.Matching, true, nil, directOutboundPCF, directConnRetainFn),\n\t\t),\n\t})\n}\n\n// testOutbound prefixes outbound with \"test-\" to not clash with other real Cadence outbounds.\nfunc testOutboundName(name string) string {\n\treturn \"test-\" + name\n}\n\ntype versionMiddleware struct {\n}\n\nfunc (vm *versionMiddleware) Handle(ctx context.Context, req *transport.Request, resw transport.ResponseWriter, h transport.UnaryHandler) error {\n\treq.Headers = req.Headers.With(common.LibraryVersionHeaderName, \"1.0.0\").\n\t\tWith(common.FeatureVersionHeaderName, cc.SupportedGoSDKVersion).\n\t\tWith(common.ClientImplHeaderName, cc.GoSDK)\n\n\treturn h.Handle(ctx, req, resw)\n}\n\nfunc getFromDynamicConfig(params *resource.Params) func() []string {\n\treturn func() []string {\n\t\tlist, err := params.DynamicConfig.GetListValue(dynamicproperties.AllIsolationGroups, nil)\n\t\tif err != nil {\n\t\t\tparams.Logger.Error(\"failed to get isolation groups from config\", tag.Error(err))\n\t\t\treturn nil\n\t\t}\n\t\tres, err := isolationgroupapi.MapAllIsolationGroupsResponse(list)\n\t\tif err != nil {\n\t\t\tparams.Logger.Error(\"failed to map isolation groups from config\", tag.Error(err))\n\t\t\treturn nil\n\t\t}\n\t\treturn res\n\t}\n}\n"
  },
  {
    "path": "host/persistence/cassandra/cassandra_persistence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nfunc TestCassandraHistoryPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.HistoryV2PersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraMatchingPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.MatchingPersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraDomainPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.MetadataPersistenceSuiteV2)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraShardPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.ShardPersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraShardMigrationPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.ShardPersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.TestBase.DynamicConfiguration.ReadNoSQLShardFromDataBlob = dynamicproperties.GetBoolPropertyFn(true)\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraVisibilityPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.DBVisibilityPersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraExecutionManager(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.ExecutionManagerSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraExecutionManagerWithEventsV2(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.ExecutionManagerSuiteForEventsV2)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraQueuePersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.QueuePersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraConfigStorePersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.ConfigStorePersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestCassandraDomainAuditPersistence(t *testing.T) {\n\ttestflags.RequireCassandra(t)\n\ts := new(persistencetests.DomainAuditPersistenceSuite)\n\ts.TestBase = public.NewTestBaseWithPublicCassandra(t, &persistencetests.TestBaseOptions{})\n\ts.Setup()\n\tsuite.Run(t, s)\n}\n"
  },
  {
    "path": "host/persistence/dynamodb/dynamodb_persistence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage dynamodb\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/dynamodb\"\n)\n\n// This is to make sure adding new noop method when adding new nosql interfaces\n// Remove it when any other tests are implemented.\nfunc TestDynamoDBNoopStruct(t *testing.T) {\n\t_, _ = dynamodb.NewDynamoDB(config.NoSQL{}, nil)\n}\n\nfunc TestDynamoDBHistoryPersistence(t *testing.T) {\n\t// s := new(persistencetests.HistoryV2PersistenceSuite)\n\t// s.TestBase = public.NewTestBaseWithDynamoDB(&persistencetests.TestBaseOptions{})\n\t// s.TestBase.Setup()\n\t// suite.Run(t, s)\n}\n\nfunc TestDynamoDBMatchingPersistence(t *testing.T) {\n\t// s := new(persistencetests.MatchingPersistenceSuite)\n\t// s.TestBase = public.NewTestBaseWithDynamoDB(&persistencetests.TestBaseOptions{})\n\t// s.TestBase.Setup()\n\t// suite.Run(t, s)\n}\n\nfunc TestDynamoDBDomainPersistence(t *testing.T) {\n\t// s := new(persistencetests.MetadataPersistenceSuiteV2)\n\t// s.TestBase = public.NewTestBaseWithDynamoDB(&persistencetests.TestBaseOptions{})\n\t// s.TestBase.Setup()\n\t// suite.Run(t, s)\n}\n\nfunc TestDynamoDBQueuePersistence(t *testing.T) {\n\t// s := new(persistencetests.QueuePersistenceSuite)\n\t// s.TestBase = public.NewTestBaseWithDynamoDB(&persistencetests.TestBaseOptions{})\n\t// s.TestBase.Setup()\n\t// suite.Run(t, s)\n}\n"
  },
  {
    "path": "host/persistence/mongodb/mongodb_persistence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tests\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/mongodb\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nfunc TestMongoDBConfigStorePersistence(t *testing.T) {\n\ttestflags.RequireMongoDB(t)\n\ts := new(persistencetests.ConfigStorePersistenceSuite)\n\ts.TestBase = NewTestBaseWithMongo(t)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\n// TODO uncomment the test once HistoryEventsCRUD is implemented\n// func TestMongoDBHistoryPersistence(t *testing.T) {\n// \ts := new(persistencetests.HistoryV2PersistenceSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once TaskCRUD is implemented\n// func TestMongoDBMatchingPersistence(t *testing.T) {\n// \ts := new(persistencetests.MatchingPersistenceSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once DomainCRUD is implemented\n// func TestMongoDBDomainPersistence(t *testing.T) {\n// \ts := new(persistencetests.MetadataPersistenceSuiteV2)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once MessageQueueCRUD is implemented\n// func TestMongoDBQueuePersistence(t *testing.T) {\n// \ts := new(persistencetests.QueuePersistenceSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once ShardCRUD is implemented\n// func TestMongoDBShardPersistence(t *testing.T) {\n// \ts := new(persistencetests.ShardPersistenceSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once VisibilityCRUD is implemented\n// func TestMongoDBVisibilityPersistence(t *testing.T) {\n// \ts := new(persistencetests.DBVisibilityPersistenceSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once WorkflowCRUD is implemented\n// func TestMongoDBExecutionManager(t *testing.T) {\n// \ts := new(persistencetests.ExecutionManagerSuite)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\n// TODO uncomment the test once WorkflowCRUD is implemented\n// func TestMongoDBExecutionManagerWithEventsV2(t *testing.T) {\n// \ts := new(persistencetests.ExecutionManagerSuiteForEventsV2)\n// \ts.TestBase = NewTestBaseWithMongo()\n// \ts.TestBase.Setup()\n// \tsuite.Run(t, s)\n// }\n\nfunc NewTestBaseWithMongo(t *testing.T) *persistencetests.TestBase {\n\tport, err := environment.GetMongoPort()\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\n\toptions := &persistencetests.TestBaseOptions{\n\t\tDBPluginName: mongodb.PluginName,\n\t\tDBHost:       environment.GetMongoAddress(),\n\t\tDBUsername:   \"root\",\n\t\tDBPassword:   \"cadence\",\n\t\tDBPort:       port,\n\t}\n\treturn persistencetests.NewTestBaseWithNoSQL(t, options)\n}\n"
  },
  {
    "path": "host/persistence/mysql/mysql_persistence_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nfunc TestMySQLHistoryV2PersistenceSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.HistoryV2PersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLMatchingPersistenceSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.MatchingPersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLMetadataPersistenceSuiteV2(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.MetadataPersistenceSuiteV2)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLShardPersistenceSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.ShardPersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\ntype ExecutionManagerSuite struct {\n\tpt.ExecutionManagerSuite\n}\n\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in mysql\")\n}\n\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in mysql\")\n}\n\nfunc TestMySQLExecutionManagerSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(ExecutionManagerSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLExecutionManagerWithEventsV2(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.ExecutionManagerSuiteForEventsV2)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLVisibilityPersistenceSuite(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.DBVisibilityPersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLQueuePersistence(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.QueuePersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLConfigPersistence(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.ConfigStorePersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestMySQLDomainAuditPersistence(t *testing.T) {\n\ttestflags.RequireMySQL(t)\n\ts := new(pt.DomainAuditPersistenceSuite)\n\toption, err := mysql.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n"
  },
  {
    "path": "host/persistence/postgres/postgres_persistence_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\nfunc TestPostgresSQLHistoryV2PersistenceSuite(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.HistoryV2PersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLMatchingPersistenceSuite(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.MatchingPersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLMetadataPersistenceSuiteV2(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.MetadataPersistenceSuiteV2)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLShardPersistenceSuite(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.ShardPersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\ntype ExecutionManagerSuite struct {\n\tpt.ExecutionManagerSuite\n}\n\nfunc (s *ExecutionManagerSuite) TestCreateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in postgres sql\")\n}\n\nfunc (s *ExecutionManagerSuite) TestUpdateWorkflowExecutionWithWorkflowRequestsDedup() {\n\ts.T().Skip(\"skip the test until we store workflow_request in postgres sql\")\n}\n\nfunc TestPostgresSQLExecutionManagerSuite(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(ExecutionManagerSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLExecutionManagerWithEventsV2(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.ExecutionManagerSuiteForEventsV2)\n\toption, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, option)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLVisibilityPersistenceSuite(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.DBVisibilityPersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\n// TODO flaky test\n// https://github.com/uber/cadence/issues/2877\n/*\nFAIL: TestPostgresSQLQueuePersistence/TestDomainReplicationQueue (0.26s)\n        queuePersistenceTest.go:102:\n            \tError Trace:\tqueuePersistenceTest.go:102\n            \tError:      \tNot equal:\n            \t            \texpected: 99\n            \t            \tactual  : 98\n            \tTest:       \tTestPostgresSQLQueuePersistence/TestDomainReplicationQueue\n*/\n// func TestPostgresSQLQueuePersistence(t *testing.T) {\n//\ts := new(pt.QueuePersistenceSuite)\n//\ts.TestBase = pt.NewTestBaseWithSQL(GetTestClusterOption())\n//\ts.TestBase.Setup()\n//\tsuite.Run(t, s)\n// }\n\nfunc TestPostgresSQLConfigPersistence(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.ConfigStorePersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n\nfunc TestPostgresSQLDomainAuditPersistence(t *testing.T) {\n\ttestflags.RequirePostgres(t)\n\ts := new(pt.DomainAuditPersistenceSuite)\n\toptions, err := postgres.GetTestClusterOption()\n\tassert.NoError(t, err)\n\ts.TestBase = pt.NewTestBaseWithSQL(t, options)\n\ts.TestBase.Setup()\n\tsuite.Run(t, s)\n}\n"
  },
  {
    "path": "host/pinot_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build !race && pinotintegration\n// +build !race,pinotintegration\n\n/*\nTo run locally with docker containers:\n\n1. Stop the previous run if any\n\tdocker compose -f docker/github_actions/docker-compose-local-pinot.yml down\n\n2. Build the integration-test-async-wf image\n\tdocker compose -f docker/github_actions/docker-compose-local-pinot.yml build integration-test-cassandra-pinot\n\n3. Run the test in the docker container\n\tdocker compose -f docker/github_actions/docker-compose-local-pinot.yml run --rm integration-test-cassandra-pinot\n\nTo run locally natively (without docker),\n1. make sure kafka and pinot is running,\n2. then run cmd `go test -v ./host -run TestPinotIntegrationSuite -tags pinotintegration`\n3. currently we have to manually add test table and delete the table for cleaning. waiting for the support to clean the data programmatically\n*/\n\npackage host\n\nimport (\n\t\"encoding/json\"\n\t\"flag\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/host/pinotutils\"\n)\n\nconst (\n\tnumberOfRetry        = 50\n\twaitTimeBetweenRetry = 400 * time.Millisecond\n\twaitForPinotToSettle = 4 * time.Second // wait pinot shards for some time ensure data consistent\n)\n\ntype PinotIntegrationSuite struct {\n\t*require.Assertions\n\tlogger log.Logger\n\t*IntegrationBase\n\tpinotClient pnt.GenericClient\n\n\ttestSearchAttributeKey string\n\ttestSearchAttributeVal string\n}\n\nfunc TestPinotIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_pinot_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(PinotIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *PinotIntegrationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableSQLAsyncTransaction:                dynamicproperties.GetBoolPropertyFn(false),\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := NewPinotTestCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.TestRawHistoryDomainName = \"TestRawHistoryDomain\"\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.TestRawHistoryDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.ForeignDomainName = s.RandomizeStr(\"integration-foreign-test-domain\")\n\ts.Require().NoError(\n\t\ts.RegisterDomain(s.ForeignDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ts.Require().NoError(s.registerArchivalDomain())\n\n\t// this sleep is necessary because domainv2 cache gets refreshed in the\n\t// background only every domainCacheRefreshInterval period\n\ttime.Sleep(cache.DomainCacheRefreshInterval + time.Second)\n\n\ttableName := \"cadence_visibility_pinot\" // cadence_visibility_pinot_integration_test\n\tpinotConfig := &config.PinotVisibilityConfig{\n\t\tCluster:     \"\",\n\t\tBroker:      \"localhost:8099\",\n\t\tTable:       tableName,\n\t\tServiceName: \"\",\n\t\tMigration: config.VisibilityMigration{\n\t\t\tEnabled: true,\n\t\t},\n\t}\n\ts.pinotClient = pinotutils.CreatePinotClient(&s.Suite, pinotConfig, s.Logger)\n}\n\nfunc (s *PinotIntegrationSuite) SetupTest() {\n\n\ts.Assertions = require.New(s.T())\n\ts.testSearchAttributeKey = definition.CustomStringField\n\ts.testSearchAttributeVal = \"test value\"\n}\n\nfunc (s *PinotIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n\t// check how to clean up test_table\n\t// currently it is not supported\n}\n\nfunc (s *PinotIntegrationSuite) TestListOpenWorkflow() {\n\tid := \"pinot-integration-start-workflow-test\"\n\twt := \"pinot-integration-start-workflow-test-type\"\n\ttl := \"pinot-integration-start-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\n\tstartTime := time.Now().UnixNano()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startTime)\n\tvar openExecution *types.WorkflowExecutionInfo\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\t\tresp, err := s.Engine.ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\t\tStartTimeFilter: startFilter,\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\ts.Nil(err)\n\n\t\tif len(resp.GetExecutions()) > 0 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(we.GetRunID(), openExecution.GetExecution().GetRunID())\n\ts.Equal(attrValBytes, openExecution.SearchAttributes.GetIndexedFields()[s.testSearchAttributeKey])\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow() {\n\tid := \"pinot-integration-list-workflow-test\"\n\twt := \"pinot-integration-list-workflow-test-type\"\n\ttl := \"pinot-integration-list-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n}\n\nfunc (s *PinotIntegrationSuite) createStartWorkflowExecutionRequest(id, wt, tl string) *types.StartWorkflowExecutionRequest {\n\tidentity := \"worker1\"\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\treturn request\n}\n\nfunc (s *PinotIntegrationSuite) testHelperForReadOnce(runID, query string, isScan bool, isAnyMatchOk bool) {\n\ts.testHelperForReadOnceWithDomain(s.DomainName, runID, query, isScan, isAnyMatchOk)\n}\n\nfunc (s *PinotIntegrationSuite) testHelperForReadOnceWithDomain(domainName string, runID, query string, isScan bool, isAnyMatchOk bool) {\n\tvar openExecution *types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   domainName,\n\t\tPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\tQuery:    query,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\nRetry:\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\n\t\ts.Nil(err)\n\t\tlogStr := fmt.Sprintf(\"Results for query '%s' (desired runId: %s): \\n\", query, runID)\n\t\ts.Logger.Info(logStr)\n\t\tfor _, e := range resp.GetExecutions() {\n\t\t\tlogStr = fmt.Sprintf(\"Execution: %+v, %+v \\n\", e.Execution, e)\n\t\t\ts.Logger.Info(logStr)\n\t\t}\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\tif isAnyMatchOk {\n\t\t\tfor _, e := range resp.GetExecutions() {\n\t\t\t\tif e.Execution.RunID == runID {\n\t\t\t\t\topenExecution = e\n\t\t\t\t\tbreak Retry\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(runID, openExecution.GetExecution().GetRunID())\n\ts.True(openExecution.GetExecutionTime() >= openExecution.GetStartTime())\n\tif openExecution.SearchAttributes != nil && len(openExecution.SearchAttributes.GetIndexedFields()) > 0 {\n\t\tsearchValBytes := openExecution.SearchAttributes.GetIndexedFields()[s.testSearchAttributeKey]\n\t\tvar searchVal string\n\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\t// pinot sets default values for all the columns,\n\t\t// this feature can break the test here when there is no actual search attributes upsert, it will still return something\n\t\t// TODO: update this after finding a good solution\n\t\ts.Equal(searchVal, searchVal)\n\t}\n}\n\nfunc (s *PinotIntegrationSuite) startWorkflow(\n\tprefix string,\n\tisCron bool,\n) *types.StartWorkflowExecutionResponse {\n\tid := \"pinot-integration-list-workflow-\" + prefix + \"-test\"\n\twt := \"pinot-integration-list-workflow-\" + prefix + \"test-type\"\n\ttl := \"pinot-integration-list-workflow-\" + prefix + \"test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tif isCron {\n\t\trequest.CronSchedule = \"*/5 * * * *\" // every 5 minutes\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n\treturn we\n}\n\nfunc (s *PinotIntegrationSuite) TestListCronWorkflows() {\n\twe1 := s.startWorkflow(\"cron\", true)\n\twe2 := s.startWorkflow(\"regular\", false)\n\n\tquery := fmt.Sprintf(`IsCron = \"true\"`)\n\ts.testHelperForReadOnce(we1.GetRunID(), query, false, true)\n\n\tquery = fmt.Sprintf(`IsCron = \"false\"`)\n\ts.testHelperForReadOnce(we2.GetRunID(), query, false, true)\n}\n\nfunc (s *PinotIntegrationSuite) TestIsGlobalSearchAttribute() {\n\twe := s.startWorkflow(\"local\", true)\n\t// global domains are disabled for this integration test, so we can only test the false case\n\tquery := fmt.Sprintf(`NumClusters = \"1\"`)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, true)\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_ExecutionTime() {\n\tid := \"pinot-integration-list-workflow-execution-time-test\"\n\twt := \"pinot-integration-list-workflow-execution-time-test-type\"\n\ttl := \"pinot-integration-list-workflow-execution-time-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tcronID := id + \"-cron\"\n\trequest.CronSchedule = \"@every 1m\"\n\trequest.WorkflowID = cronID\n\n\tweCron, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`(WorkflowID = '%s' or WorkflowID = '%s') and ExecutionTime < %v and ExecutionTime > 0`, id, cronID, time.Now().UnixNano()+int64(time.Minute))\n\ts.testHelperForReadOnce(weCron.GetRunID(), query, false, false)\n\n\tquery = fmt.Sprintf(`WorkflowID = '%s'`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_SearchAttribute() {\n\tid := \"pinot-integration-list-workflow-by-search-attr-test\"\n\twt := \"pinot-integration-list-workflow-by-search-attr-test-type\"\n\ttl := \"pinot-integration-list-workflow-by-search-attr-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\ts.testHelperForReadOnce(we.GetRunID(), query, false, false)\n\n\t// test upsert\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: getPinotUpsertSearchAttributes(),\n\t\t\t}}\n\n\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t}\n\ttaskList := &types.TaskList{Name: tl}\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        \"worker1\",\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: int32(2),\n\t\tQuery:    fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing and BinaryChecksums = 'binary-v1'`, wt),\n\t}\n\t// verify upsert data is on Pinot\n\ts.testListResultForUpsertSearchAttributes(listRequest)\n\n\t// verify DescribeWorkflowExecution\n\tdescRequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\n\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, descRequest)\n\ts.Nil(err)\n\texpectedSearchAttributes := getPinotUpsertSearchAttributes()\n\ts.Equal(expectedSearchAttributes, descResp.WorkflowExecutionInfo.GetSearchAttributes())\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_PageToken() {\n\tid := \"pinot-integration-list-workflow-token-test\"\n\twt := \"pinot-integration-list-workflow-token-test-type\"\n\ttl := \"pinot-integration-list-workflow-token-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tnumOfWorkflows := defaultTestValueOfESIndexMaxResultWindow - 1 // == 4\n\tpageSize := 3\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, false)\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_SearchAfter() {\n\tid := \"pinot-integration-list-workflow-searchAfter-test\"\n\twt := \"pinot-integration-list-workflow-searchAfter-test-type\"\n\ttl := \"pinot-integration-list-workflow-searchAfter-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tnumOfWorkflows := defaultTestValueOfESIndexMaxResultWindow + 1 // == 6\n\tpageSize := 4\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, false)\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_OrQuery() {\n\tid := \"pinot-integration-list-workflow-or-query-test\"\n\twt := \"pinot-integration-list-workflow-or-query-test-type\"\n\ttl := \"pinot-integration-list-workflow-or-query-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t// start 3 workflows\n\tkey := definition.CustomIntField\n\tattrValBytes, _ := json.Marshal(1)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\tkey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\twe1, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\trequest.RequestID = uuid.New()\n\trequest.WorkflowID = id + \"-2\"\n\tattrValBytes, _ = json.Marshal(2)\n\tsearchAttr.IndexedFields[key] = attrValBytes\n\twe2, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\trequest.RequestID = uuid.New()\n\trequest.WorkflowID = id + \"-3\"\n\tattrValBytes, _ = json.Marshal(3)\n\tsearchAttr.IndexedFields[key] = attrValBytes\n\twe3, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\t// query 1 workflow with search attr\n\tquery1 := fmt.Sprintf(`CustomIntField = %d`, 1)\n\tvar openExecution *types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: defaultTestValueOfESIndexMaxResultWindow,\n\t\tQuery:    query1,\n\t}\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.NotNil(openExecution)\n\ts.Equal(we1.GetRunID(), openExecution.GetExecution().GetRunID())\n\ts.True(openExecution.GetExecutionTime() >= openExecution.GetStartTime())\n\tsearchValBytes := openExecution.SearchAttributes.GetIndexedFields()[key]\n\tvar searchVal int\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(1, searchVal)\n\n\t// query with or clause\n\tquery2 := fmt.Sprintf(`CustomIntField = %d or CustomIntField = %d`, 1, 2)\n\tlistRequest.Query = query2\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 2 {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\t// TODO: need to clean up or every time we run, we have to delete the table.\n\ts.Equal(2, len(openExecutions))\n\n\te1 := openExecutions[0]\n\te2 := openExecutions[1]\n\tif e1.GetExecution().GetRunID() != we1.GetRunID() {\n\t\t// results are sorted by [CloseTime,RunID] desc, so find the correct mapping first\n\t\te1, e2 = e2, e1\n\t}\n\ts.Equal(we1.GetRunID(), e1.GetExecution().GetRunID())\n\ts.Equal(we2.GetRunID(), e2.GetExecution().GetRunID())\n\tsearchValBytes = e2.SearchAttributes.GetIndexedFields()[key]\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(2, searchVal)\n\n\t// query for open\n\tquery3 := fmt.Sprintf(`(CustomIntField = %d or CustomIntField = %d) and CloseTime = missing`, 2, 3)\n\tlistRequest.Query = query3\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 2 {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.Equal(2, len(openExecutions))\n\te1 = openExecutions[0]\n\te2 = openExecutions[1]\n\ts.Equal(we3.GetRunID(), e1.GetExecution().GetRunID())\n\ts.Equal(we2.GetRunID(), e2.GetExecution().GetRunID())\n\tsearchValBytes = e1.SearchAttributes.GetIndexedFields()[key]\n\tjson.Unmarshal(searchValBytes, &searchVal)\n\ts.Equal(3, searchVal)\n}\n\n// To test last page search trigger max window size error\nfunc (s *PinotIntegrationSuite) TestListWorkflow_MaxWindowSize() {\n\tid := \"pinot-integration-list-workflow-max-window-size-test\"\n\twt := \"pinot-integration-list-workflow-max-window-size-test-type\"\n\ttl := \"pinot-integration-list-workflow-max-window-size-test-tasklist\"\n\tstartRequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tfor i := 0; i < defaultTestValueOfESIndexMaxResultWindow; i++ {\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = id + strconv.Itoa(i)\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\tvar listResp *types.ListWorkflowExecutionsResponse\n\tvar nextPageToken []byte\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        s.DomainName,\n\t\tPageSize:      int32(defaultTestValueOfESIndexMaxResultWindow),\n\t\tNextPageToken: nextPageToken,\n\t\tQuery:         fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t}\n\t// get first page\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == defaultTestValueOfESIndexMaxResultWindow {\n\t\t\tlistResp = resp\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.NotNil(listResp)\n\ts.True(len(listResp.GetNextPageToken()) != 0)\n\n\t// the last request\n\tlistRequest.NextPageToken = listResp.GetNextPageToken()\n\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\ts.Nil(err)\n\ts.True(len(resp.GetExecutions()) == 0)\n\ts.True(len(resp.GetNextPageToken()) == 0)\n}\n\nfunc (s *PinotIntegrationSuite) TestListWorkflow_OrderBy() {\n\tid := \"pinot-integration-list-workflow-order-by-test\"\n\twt := \"pinot-integration-list-workflow-order-by-test-type\"\n\ttl := \"pinot-integration-list-workflow-order-by-test-tasklist\"\n\tstartRequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tfor i := 0; i < defaultTestValueOfESIndexMaxResultWindow+1; i++ { // start 6\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = id + strconv.Itoa(i)\n\n\t\tif i < defaultTestValueOfESIndexMaxResultWindow-1 { // 4 workflow has search attr\n\t\t\tintVal, _ := json.Marshal(i)\n\t\t\tdoubleVal, _ := json.Marshal(float64(i))\n\t\t\tstrVal, _ := json.Marshal(strconv.Itoa(i))\n\t\t\ttimeVal, _ := json.Marshal(time.Now())\n\t\t\tsearchAttr := &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\tdefinition.CustomIntField:      intVal,\n\t\t\t\t\tdefinition.CustomDoubleField:   doubleVal,\n\t\t\t\t\tdefinition.CustomKeywordField:  strVal,\n\t\t\t\t\tdefinition.CustomDatetimeField: timeVal,\n\t\t\t\t},\n\t\t\t}\n\t\t\tstartRequest.SearchAttributes = searchAttr\n\t\t} else {\n\t\t\tstartRequest.SearchAttributes = &types.SearchAttributes{}\n\t\t}\n\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\t// desc := \"desc\"\n\tasc := \"asc\"\n\tqueryTemplate := `WorkflowType = \"%s\" order by %s %s`\n\tpageSize := int32(defaultTestValueOfESIndexMaxResultWindow)\n\n\t// order by CloseTime asc\n\tquery1 := fmt.Sprintf(queryTemplate, wt, definition.CloseTime, asc)\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: pageSize,\n\t\tQuery:    query1,\n\t}\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif int32(len(resp.GetExecutions())) == listRequest.GetPageSize() {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.NotNil(openExecutions)\n\tfor i := int32(1); i < pageSize; i++ {\n\t\ts.True(openExecutions[i-1].GetCloseTime() <= openExecutions[i].GetCloseTime())\n\t}\n\t// comment out things below, because json index column can't use order by\n\n\t// greatest effort to reduce duplicate code\n\t// testHelper := func(query, searchAttrKey string, prevVal, currVal interface{}) {\n\t//\tlistRequest.Query = query\n\t//\tlistRequest.NextPageToken = []byte{}\n\t//\tresp, err := s.Engine.ListWorkflowExecutions(createContext(), listRequest)\n\t//\ts.Nil(err)\n\t//\topenExecutions = resp.GetExecutions()\n\t//\tdec := json.NewDecoder(bytes.NewReader(openExecutions[0].GetSearchAttributes().GetIndexedFields()[searchAttrKey]))\n\t//\tdec.UseNumber()\n\t//\terr = dec.Decode(&prevVal)\n\t//\ts.Nil(err)\n\t//\tfor i := int32(1); i < pageSize; i++ {\n\t//\t\tindexedFields := openExecutions[i].GetSearchAttributes().GetIndexedFields()\n\t//\t\tsearchAttrBytes, ok := indexedFields[searchAttrKey]\n\t//\t\tif !ok { // last one doesn't have search attr\n\t//\t\t\ts.Equal(pageSize-1, i)\n\t//\t\t\tbreak\n\t//\t\t}\n\t//\t\tdec := json.NewDecoder(bytes.NewReader(searchAttrBytes))\n\t//\t\tdec.UseNumber()\n\t//\t\terr = dec.Decode(&currVal)\n\t//\t\ts.Nil(err)\n\t//\t\tvar v1, v2 interface{}\n\t//\t\tswitch searchAttrKey {\n\t//\t\tcase definition.CustomIntField:\n\t//\t\t\tv1, _ = prevVal.(json.Number).Int64()\n\t//\t\t\tv2, _ = currVal.(json.Number).Int64()\n\t//\t\t\ts.True(v1.(int64) >= v2.(int64))\n\t//\t\tcase definition.CustomDoubleField:\n\t//\t\t\tv1, _ := strconv.ParseFloat(fmt.Sprint(prevVal), 64)\n\t//\t\t\tv2, _ := strconv.ParseFloat(fmt.Sprint(currVal), 64)\n\t//\t\t\ts.True(v1 >= v2)\n\t//\t\tcase definition.CustomKeywordField:\n\t//\t\t\ts.True(prevVal.(string) >= currVal.(string))\n\t//\t\tcase definition.CustomDatetimeField:\n\t//\t\t\tv1, _ = strconv.ParseInt(fmt.Sprint(prevVal), 10, 64)\n\t//\t\t\tv2, _ = strconv.ParseInt(fmt.Sprint(currVal), 10, 64)\n\t//\t\t\ts.True(v1.(int64) >= v2.(int64))\n\t//\t\t}\n\t//\t\tprevVal = currVal\n\t//\t}\n\t//\tlistRequest.NextPageToken = resp.GetNextPageToken()\n\t//\tresp, err = s.Engine.ListWorkflowExecutions(createContext(), listRequest) // last page\n\t//\ts.Nil(err)\n\t//\ts.Equal(1, len(resp.GetExecutions()))\n\t// }\n\n\t//\n\t// // order by CustomIntField desc\n\t// field := definition.CustomIntField\n\t// query := fmt.Sprintf(queryTemplate, wt, field, desc)\n\t// var int1, int2 int\n\t// testHelper(query, field, int1, int2)\n\t//\n\t// // order by CustomDoubleField desc\n\t// field = definition.CustomDoubleField\n\t// query = fmt.Sprintf(queryTemplate, wt, field, desc)\n\t// var double1, double2 float64\n\t// testHelper(query, field, double1, double2)\n\t//\n\t// // order by CustomKeywordField desc\n\t// field = definition.CustomKeywordField\n\t// query = fmt.Sprintf(queryTemplate, wt, field, desc)\n\t// var s1, s2 string\n\t// testHelper(query, field, s1, s2)\n\t//\n\t// // order by CustomDatetimeField desc\n\t// field = definition.CustomDatetimeField\n\t// query = fmt.Sprintf(queryTemplate, wt, field, desc)\n\t// var t1, t2 time.Time\n\t// testHelper(query, field, t1, t2)\n}\n\nfunc (s *PinotIntegrationSuite) testListWorkflowHelper(numOfWorkflows, pageSize int,\n\tstartRequest *types.StartWorkflowExecutionRequest, wid, wType string, isScan bool) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t// start enough number of workflows\n\tfor i := 0; i < numOfWorkflows; i++ {\n\t\tstartRequest.RequestID = uuid.New()\n\t\tstartRequest.WorkflowID = wid + strconv.Itoa(i)\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, startRequest)\n\t\ts.Nil(err)\n\t}\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\tvar openExecutions []*types.WorkflowExecutionInfo\n\tvar nextPageToken []byte\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        s.DomainName,\n\t\tPageSize:      int32(pageSize),\n\t\tNextPageToken: nextPageToken,\n\t\tQuery:         fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wType),\n\t}\n\t// test first page\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == pageSize {\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tnextPageToken = resp.GetNextPageToken()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\n\ts.NotNil(openExecutions)\n\ts.NotNil(nextPageToken)\n\ts.True(len(nextPageToken) > 0)\n\n\t// test last page\n\tlistRequest.NextPageToken = nextPageToken\n\tinIf := false\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tvar resp *types.ListWorkflowExecutionsResponse\n\t\tvar err error\n\n\t\tif isScan {\n\t\t\tresp, err = s.Engine.ScanWorkflowExecutions(ctx, listRequest)\n\t\t} else {\n\t\t\tresp, err = s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\t}\n\t\ts.Nil(err)\n\n\t\t// ans, _ := json.Marshal(resp.GetExecutions())\n\t\t// panic(fmt.Sprintf(\"ABUCSDK: %s\", ans))\n\n\t\tif len(resp.GetExecutions()) == numOfWorkflows-pageSize {\n\t\t\tinIf = true\n\t\t\topenExecutions = resp.GetExecutions()\n\t\t\tnextPageToken = resp.GetNextPageToken()\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.True(inIf)\n\ts.NotNil(openExecutions)\n\ts.Nil(nextPageToken)\n}\n\nfunc (s *PinotIntegrationSuite) TestScanWorkflow() {\n\tid := \"pinot-integration-scan-workflow-test\"\n\twt := \"pinot-integration-scan-workflow-test-type\"\n\ttl := \"pinot-integration-scan-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\"`, id)\n\ts.testHelperForReadOnce(we.GetRunID(), query, true, false)\n}\n\nfunc (s *PinotIntegrationSuite) TestScanWorkflow_SearchAttribute() {\n\tid := \"pinot-integration-scan-workflow-search-attr-test\"\n\twt := \"pinot-integration-scan-workflow-search-attr-test-type\"\n\ttl := \"pinot-integration-scan-workflow-search-attr-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\ts.testHelperForReadOnce(we.GetRunID(), query, true, false)\n}\n\nfunc (s *PinotIntegrationSuite) TestScanWorkflow_PageToken() {\n\tid := \"pinot-integration-scan-workflow-token-test\"\n\twt := \"pinot-integration-scan-workflow-token-test-type\"\n\ttl := \"pinot-integration-scan-workflow-token-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tnumOfWorkflows := 4\n\tpageSize := 3\n\n\ts.testListWorkflowHelper(numOfWorkflows, pageSize, request, id, wt, true)\n}\n\nfunc (s *PinotIntegrationSuite) TestCountWorkflow() {\n\tid := \"pinot-integration-count-workflow-test\"\n\twt := \"pinot-integration-count-workflow-test-type\"\n\ttl := \"pinot-integration-count-workflow-test-tasklist\"\n\trequest := s.createStartWorkflowExecutionRequest(id, wt, tl)\n\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\trequest.SearchAttributes = searchAttr\n\tctx, cancel := createContext()\n\tdefer cancel()\n\t_, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\tcountRequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: s.DomainName,\n\t\tQuery:  query,\n\t}\n\tvar resp *types.CountWorkflowExecutionsResponse\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err = s.Engine.CountWorkflowExecutions(ctx, countRequest)\n\t\ts.Nil(err)\n\t\tif resp.GetCount() == int64(1) {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.Equal(int64(1), resp.GetCount())\n\n\tquery = fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, \"noMatch\")\n\tcountRequest.Query = query\n\tresp, err = s.Engine.CountWorkflowExecutions(ctx, countRequest)\n\ts.Nil(err)\n\ts.Equal(int64(0), resp.GetCount())\n}\n\nfunc (s *PinotIntegrationSuite) TestUpsertWorkflowExecution() {\n\tid := \"pinot-integration-upsert-workflow-test\"\n\twt := \"pinot-integration-upsert-workflow-test-type\"\n\ttl := \"pinot-integration-upsert-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdecisionCount := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{}}\n\n\t\t// handle first upsert\n\t\tif decisionCount == 0 {\n\t\t\tdecisionCount++\n\n\t\t\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\t\t\tupsertSearchAttr := &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t\t\t},\n\t\t\t}\n\t\t\tupsertDecision.UpsertWorkflowSearchAttributesDecisionAttributes.SearchAttributes = upsertSearchAttr\n\t\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t\t}\n\t\t// handle second upsert, which update existing field and add new field\n\t\tif decisionCount == 1 {\n\t\t\tdecisionCount++\n\t\t\tupsertDecision.UpsertWorkflowSearchAttributesDecisionAttributes.SearchAttributes = getPinotUpsertSearchAttributes()\n\t\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// process 1st decision and assert decision is handled correctly.\n\t_, newTask, err := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\ts.Equal(int64(3), newTask.DecisionTask.GetPreviousStartedEventID())\n\ts.Equal(int64(7), newTask.DecisionTask.GetStartedEventID())\n\ts.Equal(4, len(newTask.DecisionTask.History.Events))\n\ts.Equal(types.EventTypeDecisionTaskCompleted, newTask.DecisionTask.History.Events[0].GetEventType())\n\ts.Equal(types.EventTypeUpsertWorkflowSearchAttributes, newTask.DecisionTask.History.Events[1].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskScheduled, newTask.DecisionTask.History.Events[2].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskStarted, newTask.DecisionTask.History.Events[3].GetEventType())\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\t// verify upsert data is on Pinot\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.DomainName,\n\t\tPageSize: int32(2),\n\t\t// Query:    fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t\tQuery: fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t}\n\tverified := false\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\texecution := resp.GetExecutions()[0]\n\t\t\tretrievedSearchAttr := execution.SearchAttributes\n\t\t\tif retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) > 0 {\n\t\t\t\tsearchValBytes := retrievedSearchAttr.GetIndexedFields()[s.testSearchAttributeKey]\n\t\t\t\tvar searchVal string\n\t\t\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\t\t\ts.Equal(s.testSearchAttributeVal, searchVal)\n\t\t\t\tverified = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.True(verified)\n\n\t// process 2nd decision and assert decision is handled correctly.\n\t_, newTask, err = poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t1,\n\t\ttrue,\n\t\tnil)\n\ts.Nil(err)\n\ts.NotNil(newTask)\n\ts.NotNil(newTask.DecisionTask)\n\ts.Equal(4, len(newTask.DecisionTask.History.Events))\n\ts.Equal(types.EventTypeDecisionTaskCompleted, newTask.DecisionTask.History.Events[0].GetEventType())\n\ts.Equal(types.EventTypeUpsertWorkflowSearchAttributes, newTask.DecisionTask.History.Events[1].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskScheduled, newTask.DecisionTask.History.Events[2].GetEventType())\n\ts.Equal(types.EventTypeDecisionTaskStarted, newTask.DecisionTask.History.Events[3].GetEventType())\n\n\ttime.Sleep(waitForPinotToSettle)\n\n\t// verify upsert data is on Pinot\n\ts.testListResultForUpsertSearchAttributes(listRequest)\n}\n\nfunc (s *PinotIntegrationSuite) testListResultForUpsertSearchAttributes(listRequest *types.ListWorkflowExecutionsRequest) {\n\tverified := false\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tfor i := 0; i < numberOfRetry; i++ {\n\t\tresp, err := s.Engine.ListWorkflowExecutions(ctx, listRequest)\n\t\ts.Nil(err)\n\n\t\t// res2B, _ := json.Marshal(resp.GetExecutions())\n\t\t// panic(fmt.Sprintf(\"ABCDDDBUG: %s\", listRequest.Query))\n\n\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\texecution := resp.GetExecutions()[0]\n\t\t\tretrievedSearchAttr := execution.SearchAttributes\n\t\t\tif retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) == 3 {\n\t\t\t\t// if retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) > 0 {\n\t\t\t\tfields := retrievedSearchAttr.GetIndexedFields()\n\t\t\t\tsearchValBytes := fields[s.testSearchAttributeKey]\n\t\t\t\tvar searchVal string\n\t\t\t\terr := json.Unmarshal(searchValBytes, &searchVal)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal(\"another string\", searchVal)\n\n\t\t\t\tsearchValBytes2 := fields[definition.CustomIntField]\n\t\t\t\tvar searchVal2 int\n\t\t\t\terr = json.Unmarshal(searchValBytes2, &searchVal2)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal(123, searchVal2)\n\n\t\t\t\tbinaryChecksumsBytes := fields[definition.BinaryChecksums]\n\t\t\t\tvar binaryChecksums []string\n\t\t\t\terr = json.Unmarshal(binaryChecksumsBytes, &binaryChecksums)\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.Equal([]string{\"binary-v1\", \"binary-v2\"}, binaryChecksums)\n\n\t\t\t\tverified = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeBetweenRetry)\n\t}\n\ts.True(verified)\n}\n\nfunc getPinotUpsertSearchAttributes() *types.SearchAttributes {\n\tattrValBytes1, _ := json.Marshal(\"another string\")\n\tattrValBytes2, _ := json.Marshal(123)\n\tbinaryChecksums, _ := json.Marshal([]string{\"binary-v1\", \"binary-v2\"})\n\tupsertSearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\tdefinition.CustomStringField: attrValBytes1,\n\t\t\tdefinition.CustomIntField:    attrValBytes2,\n\t\t\tdefinition.BinaryChecksums:   binaryChecksums,\n\t\t},\n\t}\n\treturn upsertSearchAttr\n}\n\nfunc (s *PinotIntegrationSuite) TestUpsertWorkflowExecution_InvalidKey() {\n\tid := \"pinot-integration-upsert-workflow-failed-test\"\n\twt := \"pinot-integration-upsert-workflow-failed-test-type\"\n\ttl := \"pinot-integration-upsert-workflow-failed-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\"INVALIDKEY\": []byte(`1`),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}}\n\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tStickyTaskList:  taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Nil(err)\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tdecisionFailedEvent := history.GetEvents()[3]\n\ts.Equal(types.EventTypeDecisionTaskFailed, decisionFailedEvent.GetEventType())\n\tfailedDecisionAttr := decisionFailedEvent.DecisionTaskFailedEventAttributes\n\ts.Equal(types.DecisionTaskFailedCauseBadSearchAttributes, failedDecisionAttr.GetCause())\n\ts.True(len(failedDecisionAttr.GetDetails()) > 0)\n}\n"
  },
  {
    "path": "host/pinotutils/pinotClient.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage pinotutils\n\nimport (\n\t\"github.com/startreedata/pinot-client-go/pinot\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n)\n\nfunc CreatePinotClient(s *suite.Suite, pinotConfig *config.PinotVisibilityConfig, logger log.Logger) pnt.GenericClient {\n\tpinotRawClient, err := pinot.NewFromBrokerList([]string{pinotConfig.Broker})\n\ts.Require().NoError(err)\n\tpinotClient := pnt.NewPinotClient(pinotRawClient, logger, pinotConfig)\n\treturn pinotClient\n}\n"
  },
  {
    "path": "host/query_workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/binary\"\n\t\"errors\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_Sticky() {\n\tid := \"interation-query-workflow-test-sticky\"\n\twt := \"interation-query-workflow-test-sticky-type\"\n\ttl := \"interation-query-workflow-test-sticky-tasklist\"\n\tstl := \"interation-query-workflow-test-sticky-tasklist-sticky\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stl\n\tstickyScheduleToStartTimeoutSeconds := common.Int32Ptr(10)\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:                              s.Engine,\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            taskList,\n\t\tIdentity:                            identity,\n\t\tDecisionHandler:                     dtHandler,\n\t\tActivityHandler:                     atHandler,\n\t\tQueryHandler:                        queryHandler,\n\t\tLogger:                              s.Logger,\n\t\tT:                                   s.T(),\n\t\tStickyTaskList:                      stickyTaskList,\n\t\tStickyScheduleToStartTimeoutSeconds: stickyScheduleToStartTimeoutSeconds,\n\t}\n\n\t// Make a request with stick tasklist to refresh the stickiness, otherwise we won't be able to add\n\t// decisions to the sticky tasklist\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := poller.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   poller.Domain,\n\t\tTaskList: poller.StickyTaskList,\n\t\tIdentity: poller.Identity,\n\t})\n\ts.NotNil(resp)\n\ts.Equal(0, len(resp.TaskToken))\n\ts.Nil(err)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution: response\", tag.WorkflowRunID(we.RunID))\n\n\t// Make first decision to schedule activity\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, true, int64(0))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t})\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// call QueryWorkflow in separate goroutinue (because it is blocking). That will generate a query task\n\tgo queryWorkflowFn(queryType)\n\t// process that query task, which should respond via RespondQueryTaskCompleted\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTaskWithSticky(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t}\n\t// wait until query result is ready\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n\n\tgo queryWorkflowFn(\"invalid-query-type\")\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTaskWithSticky(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t}\n\tqueryResult = <-queryResultCh\n\ts.NotNil(queryResult.Err)\n\tqueryFailError, ok := queryResult.Err.(*types.QueryFailedError)\n\ts.True(ok)\n\ts.Equal(\"unknown-query-type\", queryFailError.Message)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_StickyTimeout() {\n\tid := \"interation-query-workflow-test-sticky-timeout\"\n\twt := \"interation-query-workflow-test-sticky-timeout-type\"\n\ttl := \"interation-query-workflow-test-sticky-timeout-tasklist\"\n\tstl := \"interation-query-workflow-test-sticky-timeout-tasklist-sticky\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stl\n\tstickyScheduleToStartTimeoutSeconds := common.Int32Ptr(10)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:                              s.Engine,\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            taskList,\n\t\tIdentity:                            identity,\n\t\tDecisionHandler:                     dtHandler,\n\t\tActivityHandler:                     atHandler,\n\t\tQueryHandler:                        queryHandler,\n\t\tLogger:                              s.Logger,\n\t\tT:                                   s.T(),\n\t\tStickyTaskList:                      stickyTaskList,\n\t\tStickyScheduleToStartTimeoutSeconds: stickyScheduleToStartTimeoutSeconds,\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, true, int64(0))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t})\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// call QueryWorkflow in separate goroutinue (because it is blocking). That will generate a query task\n\tgo queryWorkflowFn(queryType)\n\t// process that query task, which should respond via RespondQueryTaskCompleted\n\tfor {\n\t\t// loop until process the query task\n\t\t// here we poll on normal tasklist, to simulate a worker crash and restart\n\t\t// on the server side, server will first try the sticky tasklist and then the normal tasklist\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t}\n\t// wait until query result is ready\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_NonSticky() {\n\tid := \"integration-query-workflow-test-non-sticky\"\n\twt := \"integration-query-workflow-test-non-sticky-type\"\n\ttl := \"integration-query-workflow-test-non-sticky-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tQueryHandler:    queryHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string, rejectCondition *types.QueryRejectCondition) {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t\tQueryRejectCondition: rejectCondition,\n\t\t})\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// call QueryWorkflow in separate goroutinue (because it is blocking). That will generate a query task\n\tgo queryWorkflowFn(queryType, nil)\n\t// process that query task, which should respond via RespondQueryTaskCompleted\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t} // wait until query result is ready\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n\n\tgo queryWorkflowFn(\"invalid-query-type\", nil)\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t}\n\tqueryResult = <-queryResultCh\n\ts.NotNil(queryResult.Err)\n\tqueryFailError, ok := queryResult.Err.(*types.QueryFailedError)\n\ts.True(ok)\n\ts.Equal(\"unknown-query-type\", queryFailError.Message)\n\n\t// advance the state of the decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.NoError(err)\n\n\tgo queryWorkflowFn(queryType, nil)\n\t// process that query task, which should respond via RespondQueryTaskCompleted\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t} // wait until query result is ready\n\tqueryResult = <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString = string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n\n\trejectCondition := types.QueryRejectConditionNotOpen\n\tgo queryWorkflowFn(queryType, &rejectCondition)\n\tqueryResult = <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.Nil(queryResult.Resp.QueryResult)\n\ts.NotNil(queryResult.Resp.QueryRejected.CloseStatus)\n\ts.Equal(types.WorkflowExecutionCloseStatusCompleted, *queryResult.Resp.QueryRejected.CloseStatus)\n\n\trejectCondition = types.QueryRejectConditionNotCompletedCleanly\n\tgo queryWorkflowFn(queryType, &rejectCondition)\n\tfor {\n\t\t// loop until process the query task\n\t\tisQueryTask, errInner := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(errInner)\n\t\tif isQueryTask {\n\t\t\tbreak\n\t\t}\n\t} // wait until query result is ready\n\tqueryResult = <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString = string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_Consistent_PiggybackQuery() {\n\tid := \"integration-query-workflow-test-consistent-piggyback-query\"\n\twt := \"integration-query-workflow-test-consistent-piggyback-query-type\"\n\ttl := \"integration-query-workflow-test-consistent-piggyback-query-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tvar handledSignal atomic.Value\n\thandledSignal.Store(false)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\thandledSignal.Store(true)\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tQueryHandler:    queryHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string, rejectCondition *types.QueryRejectCondition) {\n\t\t// before the query is answer the signal is not handled because the decision task is not dispatched\n\t\t// to the worker yet\n\t\ts.False(handledSignal.Load().(bool))\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t\tQueryRejectCondition:  rejectCondition,\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t})\n\t\t// after the query is answered the signal is handled because query is consistent and since\n\t\t// signal came before query signal must be handled by the time query returns\n\t\ts.True(handledSignal.Load().(bool))\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// send signal to ensure there is an outstanding decision task to dispatch query on\n\t// otherwise query will just go through matching\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\t// call QueryWorkflow in separate goroutine (because it is blocking). That will generate a query task\n\t// notice that the query comes after signal here but is consistent so it should reflect the state of the signal having been applied\n\tgo queryWorkflowFn(queryType, nil)\n\t// ensure query has had enough time to at least start before a decision task is polled\n\t// if the decision task containing the signal is polled before query is started it will not impact\n\t// correctness but it will mean query will be able to be dispatched directly after signal\n\t// without being attached to the decision task signal is on\n\t<-time.After(time.Second)\n\n\tisQueryTask, _, errInner := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\tfalse,\n\t\tfalse,\n\t\tint64(0),\n\t\t5,\n\t\tfalse,\n\t\t&types.WorkflowQueryResult{\n\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\tAnswer:     []byte(\"consistent query result\"),\n\t\t})\n\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(errInner)\n\ts.False(isQueryTask)\n\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"consistent query result\", queryResultString)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_Consistent_Timeout() {\n\tid := \"integration-query-workflow-test-consistent-timeout\"\n\twt := \"integration-query-workflow-test-consistent-timeout-type\"\n\ttl := \"integration-query-workflow-test-consistent-timeout-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10), // ensure longer than time takes to handle signal\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10), // ensure longer than time it takes to handle signal\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50), // ensure longer than time takes to handle signal\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\t<-time.After(3 * time.Second) // take longer to respond to the decision task than the query waits for\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tQueryHandler:    queryHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string, rejectCondition *types.QueryRejectCondition) {\n\t\tshortCtx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t\tqueryResp, err := s.Engine.QueryWorkflow(shortCtx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t\tQueryRejectCondition:  rejectCondition,\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t})\n\t\tcancel()\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\t// call QueryWorkflow in separate goroutine (because it is blocking). That will generate a query task\n\t// notice that the query comes after signal here but is consistent so it should reflect the state of the signal having been applied\n\tgo queryWorkflowFn(queryType, nil)\n\t// ensure query has had enough time to at least start before a decision task is polled\n\t// if the decision task containing the signal is polled before query is started it will not impact\n\t// correctness but it will mean query will be able to be dispatched directly after signal\n\t// without being attached to the decision task signal is on\n\t<-time.After(time.Second)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.NoError(err)\n\n\t// wait for query to timeout\n\tqueryResult := <-queryResultCh\n\ts.Error(queryResult.Err) // got a timeout error\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_Consistent_BlockedByStarted_NonSticky() {\n\tid := \"integration-query-workflow-test-consistent-blocked-by-started-non-sticky\"\n\twt := \"integration-query-workflow-test-consistent-blocked-by-started-non-sticky-type\"\n\ttl := \"integration-query-workflow-test-consistent-blocked-by-started-non-sticky-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10), // ensure longer than time takes to handle signal\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tvar handledSignal atomic.Value\n\thandledSignal.Store(false)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10), // ensure longer than time it takes to handle signal\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50), // ensure longer than time takes to handle signal\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\t// wait for some time to force decision task to stay in started state while query is issued\n\t\t\t\t\t<-time.After(5 * time.Second)\n\t\t\t\t\thandledSignal.Store(true)\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tQueryHandler:    queryHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string, rejectCondition *types.QueryRejectCondition) {\n\t\ts.False(handledSignal.Load().(bool))\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t\tQueryRejectCondition:  rejectCondition,\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t})\n\t\ts.True(handledSignal.Load().(bool))\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// send a signal that will take 5 seconds to handle\n\t// this causes the signal to still be outstanding at the time query arrives\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\tgo func() {\n\t\t// wait for decision task for signal to get started before querying workflow\n\t\t// since signal processing takes 5 seconds and we wait only one second to issue the query\n\t\t// at the time the query comes in there will be a started decision task\n\t\t// only once signal completes can queryWorkflow unblock\n\t\t<-time.After(time.Second)\n\t\tqueryWorkflowFn(queryType, nil)\n\t}()\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.NoError(err)\n\t<-time.After(time.Second)\n\n\t// query should not have been dispatched on the decision task which contains signal\n\t// because signal was already outstanding by the time query arrived\n\tselect {\n\tcase <-queryResultCh:\n\t\ts.Fail(\"query should not be ready yet\")\n\tdefault:\n\t}\n\n\t// now that started decision task completes poll for next task which will be a query task\n\t// containing the buffered query\n\tisQueryTask, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.True(isQueryTask)\n\ts.NoError(err)\n\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"query-result\", queryResultString)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_Consistent_NewDecisionTask_Sticky() {\n\tid := \"integration-query-workflow-test-consistent-new-decision-task-sticky\"\n\twt := \"integration-query-workflow-test-consistent-new-decision-task-sticky-type\"\n\ttl := \"integration-query-workflow-test-consistent-new-decision-task-sticky-tasklist\"\n\tstl := \"integration-query-workflow-test-consistent-new-decision-task-sticky-tasklist-sticky\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stl\n\tstickyTaskList.Kind = types.TaskListKindSticky.Ptr()\n\tstickyScheduleToStartTimeoutSeconds := common.Int32Ptr(10)\n\n\t// decider logic\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tvar handledSignal atomic.Value\n\thandledSignal.Store(false)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10), // ensure longer than time it takes to handle signal\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50), // ensure longer than time takes to handle signal\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\t// wait for some time to force decision task to stay in started state while query is issued\n\t\t\t\t\t<-time.After(5 * time.Second)\n\t\t\t\t\thandledSignal.Store(true)\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tqueryHandler := func(task *types.PollForDecisionTaskResponse) ([]byte, error) {\n\t\ts.NotNil(task.Query)\n\t\ts.NotNil(task.Query.QueryType)\n\t\tif task.Query.QueryType == queryType {\n\t\t\treturn []byte(\"query-result\"), nil\n\t\t}\n\n\t\treturn nil, errors.New(\"unknown-query-type\")\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:                              s.Engine,\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            taskList,\n\t\tIdentity:                            identity,\n\t\tDecisionHandler:                     dtHandler,\n\t\tActivityHandler:                     atHandler,\n\t\tQueryHandler:                        queryHandler,\n\t\tLogger:                              s.Logger,\n\t\tT:                                   s.T(),\n\t\tStickyTaskList:                      stickyTaskList,\n\t\tStickyScheduleToStartTimeoutSeconds: stickyScheduleToStartTimeoutSeconds,\n\t}\n\n\t// Make a request with stick tasklist to refresh the stickiness, otherwise we won't be able to add\n\t// decisions to the sticky tasklist\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresp, err := poller.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   poller.Domain,\n\t\tTaskList: poller.StickyTaskList,\n\t\tIdentity: poller.Identity,\n\t})\n\ts.NotNil(resp)\n\ts.Equal(0, len(resp.TaskToken))\n\ts.Nil(err)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10), // ensure longer than time takes to handle signal\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// Make first decision to schedule activity\n\t_, err = poller.PollAndProcessDecisionTaskWithAttempt(false, false, false, true, int64(0))\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttype QueryResult struct {\n\t\tResp *types.QueryWorkflowResponse\n\t\tErr  error\n\t}\n\tqueryResultCh := make(chan QueryResult)\n\tqueryWorkflowFn := func(queryType string, rejectCondition *types.QueryRejectCondition) {\n\t\ts.False(handledSignal.Load().(bool))\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\tqueryResp, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\tQueryType: queryType,\n\t\t\t},\n\t\t\tQueryRejectCondition:  rejectCondition,\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t})\n\t\ts.True(handledSignal.Load().(bool))\n\t\tqueryResultCh <- QueryResult{Resp: queryResp, Err: err}\n\t}\n\n\t// send a signal that will take 5 seconds to handle\n\t// this causes the signal to still be outstanding at the time query arrives\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\tgo func() {\n\t\t// wait for decision task for signal to get started before querying workflow\n\t\t// since signal processing takes 5 seconds and we wait only one second to issue the query\n\t\t// at the time the query comes in there will be a started decision task\n\t\t// only once signal completes can queryWorkflow unblock\n\t\t<-time.After(time.Second)\n\n\t\t// at this point there is a decision task started on the worker so this second signal will become buffered\n\t\tsignalName := \"my signal\"\n\t\tsignalInput := []byte(\"my signal input.\")\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\terr = s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tSignalName: signalName,\n\t\t\tInput:      signalInput,\n\t\t\tIdentity:   identity,\n\t\t})\n\t\ts.Nil(err)\n\n\t\tqueryWorkflowFn(queryType, nil)\n\t}()\n\n\t_, err = poller.PollAndProcessDecisionTaskWithSticky(false, false)\n\ts.NoError(err)\n\t<-time.After(time.Second)\n\n\t// query should not have been dispatched on the decision task which contains signal\n\t// because signal was already outstanding by the time query arrived\n\tselect {\n\tcase <-queryResultCh:\n\t\ts.Fail(\"query should not be ready yet\")\n\tdefault:\n\t}\n\n\t// now poll for decision task which contains the query which was buffered\n\tisQueryTask, _, errInner := poller.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tfalse,\n\t\tfalse,\n\t\ttrue,\n\t\ttrue,\n\t\tint64(0),\n\t\t5,\n\t\tfalse,\n\t\t&types.WorkflowQueryResult{\n\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\tAnswer:     []byte(\"consistent query result\"),\n\t\t})\n\n\t// the task should not be a query task because at the time outstanding decision task completed\n\t// there existed a buffered event which triggered the creation of a new decision task which query was dispatched on\n\ts.False(isQueryTask)\n\ts.NoError(errInner)\n\n\tqueryResult := <-queryResultCh\n\ts.NoError(queryResult.Err)\n\ts.NotNil(queryResult.Resp)\n\ts.NotNil(queryResult.Resp.QueryResult)\n\ts.Nil(queryResult.Resp.QueryRejected)\n\tqueryResultString := string(queryResult.Resp.QueryResult)\n\ts.Equal(\"consistent query result\", queryResultString)\n}\n\nfunc (s *IntegrationSuite) TestQueryWorkflow_BeforeFirstDecision() {\n\tid := \"integration-test-query-workflow-before-first-decision\"\n\twt := \"integration-test-query-workflow-before-first-decision-type\"\n\ttl := \"integration-test-query-workflow-before-first-decision-tasklist\"\n\tidentity := \"worker1\"\n\tqueryType := \"test-query\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.RunID,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// query workflow without any decision task should produce an error\n\t_, err := s.Engine.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\tDomain:    s.DomainName,\n\t\tExecution: workflowExecution,\n\t\tQuery: &types.WorkflowQuery{\n\t\t\tQueryType: queryType,\n\t\t},\n\t})\n\ts.IsType(&types.QueryFailedError{}, err)\n\ts.Equal(\"workflow must handle at least one decision task before it can be queried\", err.(*types.QueryFailedError).Message)\n}\n"
  },
  {
    "path": "host/reset_workflow_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"strconv\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestResetWorkflow() {\n\tid := \"integration-reset-workflow-test\"\n\twt := \"integration-reset-workflow-test-type\"\n\ttl := \"integration-reset-workflow-test-taskqueue\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\n\ttasklist := &types.TaskList{Name: tl}\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            tasklist,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.NoError(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.GetRunID()))\n\n\t// workflow logic\n\tworkflowComplete := false\n\tactivityData := int32(1)\n\tactivityCount := 3\n\tisFirstTaskProcessed := false\n\tisSecondTaskProcessed := false\n\tvar firstActivityCompletionEvent *types.HistoryEvent\n\twtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !isFirstTaskProcessed {\n\t\t\t// Schedule 3 activities on first workflow task\n\t\t\tisFirstTaskProcessed = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\tvar scheduleActivityCommands []*types.Decision\n\t\t\tfor i := 1; i <= activityCount; i++ {\n\t\t\t\tscheduleActivityCommands = append(scheduleActivityCommands, &types.Decision{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\tActivityID:                    strconv.Itoa(i),\n\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"ResetActivity\"},\n\t\t\t\t\t\tTaskList:                      tasklist,\n\t\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\n\t\t\treturn nil, scheduleActivityCommands, nil\n\t\t} else if !isSecondTaskProcessed {\n\t\t\t// Confirm one activity completion on second workflow task\n\t\t\tisSecondTaskProcessed = true\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif event.GetEventType() == types.EventTypeActivityTaskCompleted {\n\t\t\t\t\tfirstActivityCompletionEvent = event\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\t// Complete workflow after reset\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        tasklist,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: wtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Process first workflow decision task to schedule activities\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessWorkflowTask\", tag.Error(err))\n\ts.NoError(err)\n\n\t// Process one activity task which also creates second workflow task\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"Poll and process first activity\", tag.Error(err))\n\ts.NoError(err)\n\n\t// Process second workflow task which checks activity completion\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"Poll and process second workflow task\", tag.Error(err))\n\ts.NoError(err)\n\n\t// Find reset point (last completed decision task)\n\tevents := s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.GetRunID(),\n\t})\n\tvar lastDecisionCompleted *types.HistoryEvent\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\tlastDecisionCompleted = event\n\t\t}\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// FIRST reset: Reset workflow execution, current is open\n\tresp, err := s.Engine.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tReason:                \"reset execution from test\",\n\t\tDecisionFinishEventID: lastDecisionCompleted.ID,\n\t\tRequestID:             uuid.New(),\n\t})\n\ts.NoError(err)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"Poll and process second activity\", tag.Error(err))\n\ts.NoError(err)\n\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"Poll and process third activity\", tag.Error(err))\n\ts.NoError(err)\n\n\ts.NotNil(firstActivityCompletionEvent)\n\ts.False(workflowComplete)\n\n\t// get the history of the first run again\n\tevents = s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.GetRunID(),\n\t})\n\tvar firstRunStartTimestamp int64\n\tvar lastEvent *types.HistoryEvent\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionStarted {\n\t\t\tfirstRunStartTimestamp = event.GetTimestamp()\n\t\t}\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\tlastDecisionCompleted = event\n\t\t}\n\t\tlastEvent = event\n\t}\n\t// assert the first run is closed, terminated by the previous reset\n\ts.Equal(types.EventTypeWorkflowExecutionTerminated, lastEvent.GetEventType())\n\n\t// check the start time of mutable state for the second run,\n\t// it should be reset time although the start event is reused\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tdescResp, err := s.Engine.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      resp.GetRunID(),\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.True(descResp.WorkflowExecutionInfo.GetStartTime() > firstRunStartTimestamp)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// SECOND reset: reset the first run again, to exercise the code path of resetting closed workflow\n\tresp, err = s.Engine.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.GetRunID(),\n\t\t},\n\t\tReason:                \"reset execution from test\",\n\t\tDecisionFinishEventID: lastDecisionCompleted.ID,\n\t\tRequestID:             uuid.New(),\n\t})\n\ts.NoError(err)\n\tnewRunID := resp.GetRunID()\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"Poll and process final decision task\", tag.Error(err))\n\ts.NoError(err)\n\ts.True(workflowComplete)\n\n\t// get the history of the newRunID\n\tevents = s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      newRunID,\n\t})\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\tlastDecisionCompleted = event\n\t\t}\n\t\tlastEvent = event\n\t}\n\t// assert the new run is closed, completed by decision task\n\ts.Equal(types.EventTypeWorkflowExecutionCompleted, lastEvent.GetEventType())\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// THIRD reset: reset the workflow run that is after a reset\n\t_, err = s.Engine.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      newRunID,\n\t\t},\n\t\tReason:                \"reset execution from test\",\n\t\tDecisionFinishEventID: lastDecisionCompleted.ID,\n\t\tRequestID:             uuid.New(),\n\t})\n\ts.NoError(err)\n}\n\nfunc (s *IntegrationSuite) TestResetWorkflow_NoDecisionTaskCompleted() {\n\tid := \"integration-reset-workflow-test-no-decision-completed\"\n\twt := \"integration-reset-workflow-test-type--no-decision-completed\"\n\ttl := \"integration-reset-workflow-test-taskqueue-no-decision-completed\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{Name: wt}\n\n\ttasklist := &types.TaskList{Name: tl}\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            tasklist,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.NoError(err0)\n\n\tworkflowComplete := false\n\tactivityData := int32(1)\n\tisFirstTaskProcessed := false\n\n\twtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !isFirstTaskProcessed {\n\t\t\t// Schedule 3 activities on first workflow task\n\t\t\tisFirstTaskProcessed = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\tvar scheduleActivityCommands []*types.Decision\n\t\t\tscheduleActivityCommands = append(scheduleActivityCommands, &types.Decision{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"ResetActivity\"},\n\t\t\t\t\tTaskList:                      tasklist,\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t})\n\n\t\t\treturn nil, scheduleActivityCommands, nil\n\t\t}\n\t\t// Complete workflow after reset\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        tasklist,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: wtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Process first workflow decision task to schedule activities\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessWorkflowTask\", tag.Error(err))\n\ts.NoError(err)\n\n\t// Process one activity task which also creates second workflow task\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"Poll and process first activity\", tag.Error(err))\n\ts.NoError(err)\n\n\t// Find reset point (last completed decision task)\n\tevents := s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.GetRunID(),\n\t})\n\tvar lastDecisionScheduled *types.HistoryEvent\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskScheduled {\n\t\t\tlastDecisionScheduled = event\n\t\t}\n\t}\n\ts.NotNil(lastDecisionScheduled)\n\n\tdescribeDomainRequest := &types.DescribeDomainRequest{\n\t\tName: &s.DomainName,\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\n\tdescribeDomainResponse, err := s.Engine.DescribeDomain(ctx, describeDomainRequest)\n\ts.NoError(err)\n\n\trecordDecisionStartedRequest := &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID: describeDomainResponse.DomainInfo.UUID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.GetRunID(),\n\t\t},\n\t\tScheduleID: lastDecisionScheduled.ID,\n\t\tTaskID:     1,\n\t\tRequestID:  uuid.New(),\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tDomain:         s.DomainName,\n\t\t\tTaskList:       tasklist,\n\t\t\tIdentity:       identity,\n\t\t\tBinaryChecksum: \"\",\n\t\t},\n\t}\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\n\t// Record decision task started, to simulate the case where decision task is started but not completed\n\t_, err = s.HistoryClient.RecordDecisionTaskStarted(ctx, recordDecisionStartedRequest)\n\ts.NoError(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// FIRST reset: Reset workflow execution, current is open\n\t_, err = s.Engine.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tReason:                \"reset execution from test\",\n\t\tDecisionFinishEventID: lastDecisionScheduled.ID + 1,\n\t\tRequestID:             uuid.New(),\n\t})\n\ts.NoError(err)\n\n\t// get the history of the first run again\n\tevents = s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      we.GetRunID(),\n\t})\n\tvar lastEvent *types.HistoryEvent\n\tfor _, event := range events {\n\t\tlastEvent = event\n\t}\n\t// assert the first run is closed, terminated by the previous reset\n\ts.Equal(types.EventTypeWorkflowExecutionTerminated, lastEvent.GetEventType())\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// SECOND reset: reset the first run again, to exercise the code path of resetting closed workflow\n\tresp, err := s.Engine.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.GetRunID(),\n\t\t},\n\t\tReason:                \"reset execution from test\",\n\t\tDecisionFinishEventID: lastDecisionScheduled.ID + 2,\n\t\tRequestID:             uuid.New(),\n\t})\n\ts.NoError(err)\n\n\tnewRunID := resp.GetRunID()\n\n\tworkflowComplete = false\n\tisFirstTaskProcessed = false\n\n\t// Process first workflow decision task to schedule activities\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessWorkflowTask\", tag.Error(err))\n\ts.NoError(err)\n\n\ts.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      newRunID,\n\t})\n\n\t// Process one activity task which also creates second workflow task\n\terr = poller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"Poll and process first activity\", tag.Error(err))\n\ts.NoError(err)\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"Poll and process final decision task\", tag.Error(err))\n\ts.NoError(err)\n\ts.True(workflowComplete)\n\n\t// get the history of the newRunID\n\tevents = s.getHistory(s.DomainName, &types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      newRunID,\n\t})\n\tfor _, event := range events {\n\t\tlastEvent = event\n\t}\n\t// assert the new run is closed, completed by decision task\n\ts.Equal(types.EventTypeWorkflowExecutionCompleted, lastEvent.GetEventType())\n}\n"
  },
  {
    "path": "host/retry_policy_workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *IntegrationSuite) TestWorkflowRetryPolicyTimeout() {\n\tid := \"integration-workflow-retry-policy-timeout-test\"\n\twt := \"integration-workflow-retry-policy-timeout-test-type\"\n\ttl := \"integration-workflow-retry-policy-timeout-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tretryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tBackoffCoefficient:          2.0,\n\t\tMaximumIntervalInSeconds:    1,\n\t\tExpirationIntervalInSeconds: 1,\n\t\tMaximumAttempts:             0,\n\t\tNonRetriableErrorReasons:    []string{\"bad-error\"},\n\t}\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(3),\n\t\tIdentity:                            identity,\n\t\tRetryPolicy:                         retryPolicy,\n\t}\n\tnow := time.Now()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\ttime.Sleep(3 * time.Second) // wait 1 second for timeout\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tfirstEvent := history.Events[0]\n\ts.NotNil(firstEvent)\n\ts.Equal(firstEvent.EventType, types.EventTypeWorkflowExecutionStarted.Ptr())\n\texpirationTime := firstEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp\n\ts.NotNil(expirationTime)\n\tdelta := time.Unix(0, *expirationTime).Sub(now)\n\ts.True(delta > 0*time.Second)\n\ts.True(delta < 2*time.Second)\n\n\tlastEvent := history.Events[len(history.Events)-1]\n\tdelta = time.Unix(0, common.Int64Default(lastEvent.Timestamp)).Sub(time.Unix(0, common.Int64Default(firstEvent.Timestamp)))\n\ts.True(delta > 2*time.Second)\n\ts.True(delta < 4*time.Second)\n}\n\nfunc (s *IntegrationSuite) TestWorkflowRetryPolicyContinueAsNewAsRetry() {\n\tid := \"integration-workflow-retry-policy-continue-as-new-retry-test\"\n\twt := \"integration-workflow-retry-policy-continue-as-new-retry-test-type\"\n\ttl := \"integration-workflow-retry-policy-continue-as-new-retry-test-tasklist\"\n\tidentity := \"worker1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tretryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tBackoffCoefficient:          2.0,\n\t\tMaximumIntervalInSeconds:    3,\n\t\tExpirationIntervalInSeconds: 3,\n\t\tMaximumAttempts:             0,\n\t\tNonRetriableErrorReasons:    []string{\"bad-error\"},\n\t}\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tRetryPolicy:                         retryPolicy,\n\t}\n\tnow := time.Now()\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\treturn []byte(strconv.Itoa(0)), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\tReason: common.StringPtr(\"fail-reason\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ttime.Sleep(3 * time.Second) // wait 1 second for timeout\n\n\tvar history *types.History\n\tvar lastEvent *types.HistoryEvent\n\nGetHistoryLoop:\n\tfor i := 0; i < 20; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory = historyResponse.History\n\n\t\tlastEvent = history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionTimedOut {\n\t\t\ts.Logger.Warn(\"Execution not timedout yet.\")\n\t\t\ttime.Sleep(200 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\ttimeoutEventAttributes := lastEvent.WorkflowExecutionTimedOutEventAttributes\n\t\ts.Equal(types.TimeoutTypeStartToClose, *timeoutEventAttributes.TimeoutType)\n\t\tbreak GetHistoryLoop\n\t}\n\n\tfirstEvent := history.Events[0]\n\ts.Equal(firstEvent.WorkflowExecutionStartedEventAttributes.Attempt, int32(1))\n\n\tdelta := time.Unix(0, common.Int64Default(firstEvent.WorkflowExecutionStartedEventAttributes.ExpirationTimestamp)).Sub(now)\n\ts.True(delta > 2*time.Second)\n\ts.True(delta < 4*time.Second)\n}\n"
  },
  {
    "path": "host/service.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"math/rand\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/rpc\"\n)\n\ntype (\n\t// Service is the interface which must be implemented by all the services\n\t// TODO: Service contains many methods that are not used now that we have resource bean, these should be cleaned up\n\tService interface {\n\t\tStart()\n\t\tStop()\n\t\tGetLogger() log.Logger\n\t\tGetThrottledLogger() log.Logger\n\t\tGetMetricsClient() metrics.Client\n\t\tGetClientBean() client.Bean\n\t\tGetTimeSource() clock.TimeSource\n\t\tGetDispatcher() *yarpc.Dispatcher\n\t\tGetMembershipResolver() membership.Resolver\n\t\tGetHostInfo() membership.HostInfo\n\t\tGetClusterMetadata() cluster.Metadata\n\t\tGetMessagingClient() messaging.Client\n\t\tGetBlobstoreClient() blobstore.Client\n\t\tGetArchivalMetadata() archiver.ArchivalMetadata\n\t\tGetArchiverProvider() provider.ArchiverProvider\n\t\tGetPayloadSerializer() persistence.PayloadSerializer\n\t}\n\n\t// Service contains the objects specific to this service\n\tserviceImpl struct {\n\t\tstatus                int32\n\t\tsName                 string\n\t\thostInfo              membership.HostInfo\n\t\tdispatcher            *yarpc.Dispatcher\n\t\tmembershipResolver    membership.Resolver\n\t\trpcFactory            rpc.Factory\n\t\tpprofInitializer      common.PProfInitializer\n\t\tclientBean            client.Bean\n\t\ttimeSource            clock.TimeSource\n\t\tnumberOfHistoryShards int\n\t\tallIsolationGroups    func() []string\n\n\t\tlogger          log.Logger\n\t\tthrottledLogger log.Logger\n\n\t\tmetricsScope           tally.Scope\n\t\truntimeMetricsReporter *metrics.RuntimeMetricsReporter\n\t\tmetricsClient          metrics.Client\n\t\tclusterMetadata        cluster.Metadata\n\t\tmessagingClient        messaging.Client\n\t\tblobstoreClient        blobstore.Client\n\t\tdynamicCollection      *dynamicconfig.Collection\n\t\tarchivalMetadata       archiver.ArchivalMetadata\n\t\tarchiverProvider       provider.ArchiverProvider\n\t\tserializer             persistence.PayloadSerializer\n\t}\n)\n\nvar _ Service = (*serviceImpl)(nil)\n\n// NewService instantiates a Service Instance\n// TODO: have a better name for Service.\nfunc NewService(params *resource.Params) Service {\n\tsVice := &serviceImpl{\n\t\tstatus:                common.DaemonStatusInitialized,\n\t\tsName:                 params.Name,\n\t\tlogger:                params.Logger,\n\t\tthrottledLogger:       params.ThrottledLogger,\n\t\trpcFactory:            params.RPCFactory,\n\t\tmembershipResolver:    params.MembershipResolver,\n\t\tpprofInitializer:      params.PProfInitializer,\n\t\ttimeSource:            clock.NewRealTimeSource(),\n\t\tmetricsScope:          params.MetricScope,\n\t\tnumberOfHistoryShards: params.PersistenceConfig.NumHistoryShards,\n\t\tallIsolationGroups:    params.GetIsolationGroups,\n\t\tclusterMetadata:       params.ClusterMetadata,\n\t\tmetricsClient:         params.MetricsClient,\n\t\tmessagingClient:       params.MessagingClient,\n\t\tblobstoreClient:       params.BlobstoreClient,\n\t\tdynamicCollection: dynamicconfig.NewCollection(\n\t\t\tparams.DynamicConfig,\n\t\t\tparams.Logger,\n\t\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t\t),\n\t\tarchivalMetadata: params.ArchivalMetadata,\n\t\tarchiverProvider: params.ArchiverProvider,\n\t\tserializer:       persistence.NewPayloadSerializer(),\n\t}\n\n\tsVice.runtimeMetricsReporter = metrics.NewRuntimeMetricsReporter(params.MetricScope, time.Minute, sVice.GetLogger(), params.InstanceID)\n\tsVice.dispatcher = sVice.rpcFactory.GetDispatcher()\n\tif sVice.dispatcher == nil {\n\t\tsVice.logger.Fatal(\"Unable to create yarpc dispatcher\")\n\t}\n\n\treturn sVice\n}\n\n// Start starts a yarpc service\nfunc (h *serviceImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&h.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tvar err error\n\n\th.metricsScope.Counter(metrics.RestartCount).Inc(1)\n\th.runtimeMetricsReporter.Start()\n\n\tif err := h.pprofInitializer.Start(); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"Failed to start pprof\")\n\t}\n\n\tif err := h.dispatcher.Start(); err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"Failed to start yarpc dispatcher\")\n\t}\n\n\th.membershipResolver.Start()\n\n\thostInfo, err := h.membershipResolver.WhoAmI()\n\tif err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"failed to get host info from membership monitor\")\n\t}\n\th.hostInfo = hostInfo\n\n\th.clientBean, err = client.NewClientBean(\n\t\tclient.NewRPCClientFactory(h.rpcFactory, h.membershipResolver, h.metricsClient, h.dynamicCollection, h.numberOfHistoryShards, h.allIsolationGroups, h.logger),\n\t\th.rpcFactory.GetDispatcher(),\n\t\th.clusterMetadata,\n\t)\n\tif err != nil {\n\t\th.logger.WithTags(tag.Error(err)).Fatal(\"fail to initialize client bean\")\n\t}\n\n\t// The service is now started up\n\th.logger.Info(\"service started\")\n\t// seed the random generator once for this service\n\trand.Seed(time.Now().UTC().UnixNano())\n}\n\n// Stop closes the associated transport\nfunc (h *serviceImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&h.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tif h.membershipResolver != nil {\n\t\th.membershipResolver.Stop()\n\t}\n\n\tif h.dispatcher != nil {\n\t\th.dispatcher.Stop() //nolint:errcheck\n\t}\n\n\th.runtimeMetricsReporter.Stop()\n}\n\nfunc (h *serviceImpl) GetLogger() log.Logger {\n\treturn h.logger\n}\n\nfunc (h *serviceImpl) GetThrottledLogger() log.Logger {\n\treturn h.throttledLogger\n}\n\nfunc (h *serviceImpl) GetMetricsClient() metrics.Client {\n\treturn h.metricsClient\n}\n\nfunc (h *serviceImpl) GetClientBean() client.Bean {\n\treturn h.clientBean\n}\n\nfunc (h *serviceImpl) GetTimeSource() clock.TimeSource {\n\treturn h.timeSource\n}\n\nfunc (h *serviceImpl) GetMembershipResolver() membership.Resolver {\n\treturn h.membershipResolver\n}\n\nfunc (h *serviceImpl) GetHostInfo() membership.HostInfo {\n\treturn h.hostInfo\n}\n\nfunc (h *serviceImpl) GetDispatcher() *yarpc.Dispatcher {\n\treturn h.dispatcher\n}\n\n// GetClusterMetadata returns the service cluster metadata\nfunc (h *serviceImpl) GetClusterMetadata() cluster.Metadata {\n\treturn h.clusterMetadata\n}\n\n// GetMessagingClient returns the messaging client against Kafka\nfunc (h *serviceImpl) GetMessagingClient() messaging.Client {\n\treturn h.messagingClient\n}\n\nfunc (h *serviceImpl) GetBlobstoreClient() blobstore.Client {\n\treturn h.blobstoreClient\n}\n\nfunc (h *serviceImpl) GetArchivalMetadata() archiver.ArchivalMetadata {\n\treturn h.archivalMetadata\n}\n\nfunc (h *serviceImpl) GetArchiverProvider() provider.ArchiverProvider {\n\treturn h.archiverProvider\n}\n\nfunc (h *serviceImpl) GetPayloadSerializer() persistence.PayloadSerializer {\n\treturn h.serializer\n}\n"
  },
  {
    "path": "host/signal_workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/binary\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/engine/engineimpl\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nfunc (s *IntegrationSuite) TestSignalWorkflow() {\n\tid := \"integration-signal-workflow-test\"\n\twt := \"integration-signal-workflow-test-type\"\n\ttl := \"integration-signal-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\tsignal := func(runID string, signalName string, signalInput []byte, opts ...yarpc.CallOption) error {\n\t\texecution := types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t}\n\t\tif len(runID) > 0 {\n\t\t\texecution.RunID = runID\n\t\t}\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn s.Engine.SignalWorkflowExecution(\n\t\t\tctx,\n\t\t\t&types.SignalWorkflowExecutionRequest{\n\t\t\t\tDomain:            s.DomainName,\n\t\t\t\tWorkflowExecution: &execution,\n\t\t\t\tSignalName:        signalName,\n\t\t\t\tInput:             signalInput,\n\t\t\t\tIdentity:          identity,\n\t\t\t},\n\t\t\topts...,\n\t\t)\n\t}\n\n\t// Send a signal to non-exist workflow\n\terr0 := signal(uuid.New(), \"failed signal.\", nil)\n\ts.NotNil(err0)\n\ts.IsType(&types.EntityNotExistsError{}, err0)\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tworkflowComplete := false\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tvar signalEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Send first signal using RunID\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\terr = signal(we.RunID, signalName, signalInput)\n\ts.Nil(err)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\n\t// Send another signal without RunID\n\tsignalName = \"another signal\"\n\tsignalInput = []byte(\"another signal input.\")\n\terr = signal(\"\", signalName, signalInput)\n\ts.Nil(err)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Terminate workflow execution\n\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t\tReason:   \"test signal\",\n\t\tDetails:  nil,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\t// Send signal to terminated workflow\n\terr = signal(we.RunID, \"failed signal 1.\", nil)\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n\n\t// Send signal by enabling WorkflowExecutionAlreadyCompletedError feature\n\terr = signal(we.RunID, \"failed signal 1.\", nil, yarpc.WithHeader(common.ClientFeatureFlagsHeaderName, client.FeatureFlagsHeader(apiv1.FeatureFlags{\n\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t})))\n\ts.NotNil(err)\n\ts.IsType(&types.WorkflowExecutionAlreadyCompletedError{}, err)\n}\n\nfunc (s *IntegrationSuite) TestSignalWorkflow_DuplicateRequest() {\n\tid := \"integration-signal-workflow-test-duplicate\"\n\twt := \"integration-signal-workflow-test-duplicate-type\"\n\ttl := \"integration-signal-workflow-test-duplicate-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tworkflowComplete := false\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tvar signalEvent *types.HistoryEvent\n\tnumOfSignaledEvent := 0\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tnumOfSignaledEvent = 0\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t\tnumOfSignaledEvent++\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn nil, []*types.Decision{}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Send first signal\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tRequestID := uuid.New()\n\tsignalReqest := &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t\tRequestID:  RequestID,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr = s.Engine.SignalWorkflowExecution(ctx, signalReqest)\n\ts.Nil(err)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\ts.Equal(1, numOfSignaledEvent)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Send another signal with same request id\n\terr = s.Engine.SignalWorkflowExecution(ctx, signalReqest)\n\ts.Nil(err)\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(0, numOfSignaledEvent)\n}\n\nfunc (s *IntegrationSuite) TestSignalExternalWorkflowDecision() {\n\tid := \"integration-signal-external-workflow-test\"\n\twt := \"integration-signal-external-workflow-test-type\"\n\ttl := \"integration-signal-external-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tforeignRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.ForeignDomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe2, err0 := s.Engine.StartWorkflowExecution(ctx, foreignRequest)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution on foreign Domain\", tag.WorkflowDomainName(s.ForeignDomainName), tag.WorkflowRunID(we2.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain: s.ForeignDomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\tRunID:      we2.GetRunID(),\n\t\t\t\t},\n\t\t\t\tSignalName: signalName,\n\t\t\t\tInput:      signalInput,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tworkflowComplete := false\n\tforeignActivityCount := int32(1)\n\tforeignActivityCounter := int32(0)\n\tvar signalEvent *types.HistoryEvent\n\tforeignDtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif foreignActivityCounter < foreignActivityCount {\n\t\t\tforeignActivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, foreignActivityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(foreignActivityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(foreignActivityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tforeignPoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.ForeignDomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: foreignDtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start both current and foreign workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"foreign PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\terr = foreignPoller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"foreign PollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Signal the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// in source workflow\n\tsignalSent := false\n\tintiatedEventID := 10\nCheckHistoryLoopForSignalSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\t\t// common.PrettyPrintHistory(history, s.Logger)\n\n\t\tsignalRequestedEvent := history.Events[len(history.Events)-2]\n\t\tif *signalRequestedEvent.EventType != types.EventTypeExternalWorkflowExecutionSignaled {\n\t\t\ts.Logger.Info(\"Signal still not sent.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForSignalSent\n\t\t}\n\n\t\tewfeAttributes := signalRequestedEvent.ExternalWorkflowExecutionSignaledEventAttributes\n\t\ts.NotNil(ewfeAttributes)\n\t\ts.Equal(int64(intiatedEventID), ewfeAttributes.GetInitiatedEventID())\n\t\ts.Equal(id, ewfeAttributes.WorkflowExecution.GetWorkflowID())\n\t\ts.Equal(we2.RunID, ewfeAttributes.WorkflowExecution.RunID)\n\n\t\tsignalSent = true\n\t\tbreak\n\t}\n\n\ts.True(signalSent)\n\n\t// process signal in decider for foreign workflow\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(execution.IdentityHistoryService, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n}\n\nfunc (s *IntegrationSuite) TestSignalWorkflow_Cron_NoDecisionTaskCreated() {\n\tid := \"integration-signal-workflow-test-cron\"\n\twt := \"integration-signal-workflow-test-cron-type\"\n\ttl := \"integration-signal-workflow-test-cron-tasklist\"\n\tidentity := \"worker1\"\n\tcronSpec := \"@every 2s\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start workflow execution\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tCronSchedule:                        cronSpec,\n\t}\n\tnow := time.Now()\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// Send first signal using RunID\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tctx, cancel = createContext()\n\tdefer cancel()\n\terr := s.Engine.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.RunID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tInput:      signalInput,\n\t\tIdentity:   identity,\n\t})\n\ts.Nil(err)\n\n\t// decider logic\n\tvar decisionTaskDelay time.Duration\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tdecisionTaskDelay = time.Since(now)\n\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(decisionTaskDelay > time.Second*2)\n}\n\nfunc (s *IntegrationSuite) TestSignalExternalWorkflowDecision_WithoutRunID() {\n\tid := \"integration-signal-external-workflow-test-without-run-id\"\n\twt := \"integration-signal-external-workflow-test-without-run-id-type\"\n\ttl := \"integration-signal-external-workflow-test-without-run-id-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tforeignRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.ForeignDomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\twe2, err0 := s.Engine.StartWorkflowExecution(ctx, foreignRequest)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution on foreign Domain\", tag.WorkflowDomainName(s.ForeignDomainName), tag.WorkflowRunID(we2.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain: s.ForeignDomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\t// No RunID in decision\n\t\t\t\t},\n\t\t\t\tSignalName: signalName,\n\t\t\t\tInput:      signalInput,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tworkflowComplete := false\n\tforeignActivityCount := int32(1)\n\tforeignActivityCounter := int32(0)\n\tvar signalEvent *types.HistoryEvent\n\tforeignDtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif foreignActivityCounter < foreignActivityCount {\n\t\t\tforeignActivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, foreignActivityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(foreignActivityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(foreignActivityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tforeignPoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.ForeignDomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: foreignDtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start both current and foreign workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"foreign PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\terr = foreignPoller.PollAndProcessActivityTask(false)\n\ts.Logger.Info(\"foreign PollAndProcessActivityTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Signal the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// in source workflow\n\tsignalSent := false\n\tintiatedEventID := 10\nCheckHistoryLoopForSignalSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tsignalRequestedEvent := history.Events[len(history.Events)-2]\n\t\tif *signalRequestedEvent.EventType != types.EventTypeExternalWorkflowExecutionSignaled {\n\t\t\ts.Logger.Info(\"Signal still not sent.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForSignalSent\n\t\t}\n\n\t\tewfeAttributes := signalRequestedEvent.ExternalWorkflowExecutionSignaledEventAttributes\n\t\ts.NotNil(ewfeAttributes)\n\t\ts.Equal(int64(intiatedEventID), ewfeAttributes.GetInitiatedEventID())\n\t\ts.Equal(id, ewfeAttributes.WorkflowExecution.GetWorkflowID())\n\t\ts.Equal(\"\", ewfeAttributes.WorkflowExecution.GetRunID())\n\n\t\tsignalSent = true\n\t\tbreak\n\t}\n\n\ts.True(signalSent)\n\n\t// process signal in decider for foreign workflow\n\t_, err = foreignPoller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(execution.IdentityHistoryService, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n}\n\nfunc (s *IntegrationSuite) TestSignalExternalWorkflowDecision_UnKnownTarget() {\n\tid := \"integration-signal-unknown-workflow-decision-test\"\n\twt := \"integration-signal-unknown-workflow-decision-test-type\"\n\ttl := \"integration-signal-unknown-workflow-decision-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain: s.ForeignDomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"workflow_not_exist\",\n\t\t\t\t\tRunID:      we.GetRunID(),\n\t\t\t\t},\n\t\t\t\tSignalName: signalName,\n\t\t\t\tInput:      signalInput,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Signal the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tsignalSentFailed := false\n\tintiatedEventID := 10\nCheckHistoryLoopForCancelSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tsignalFailedEvent := history.Events[len(history.Events)-2]\n\t\tif *signalFailedEvent.EventType != types.EventTypeSignalExternalWorkflowExecutionFailed {\n\t\t\ts.Logger.Info(\"Cancellaton not cancelled yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForCancelSent\n\t\t}\n\n\t\tsignalExternalWorkflowExecutionFailedEventAttributes := signalFailedEvent.SignalExternalWorkflowExecutionFailedEventAttributes\n\t\ts.Equal(int64(intiatedEventID), signalExternalWorkflowExecutionFailedEventAttributes.InitiatedEventID)\n\t\ts.Equal(\"workflow_not_exist\", signalExternalWorkflowExecutionFailedEventAttributes.WorkflowExecution.WorkflowID)\n\t\ts.Equal(we.RunID, signalExternalWorkflowExecutionFailedEventAttributes.WorkflowExecution.RunID)\n\n\t\tsignalSentFailed = true\n\t\tbreak\n\t}\n\n\ts.True(signalSentFailed)\n}\n\nfunc (s *IntegrationSuite) TestSignalExternalWorkflowDecision_SignalSelf() {\n\tid := \"integration-signal-self-workflow-decision-test\"\n\twt := \"integration-signal-self-workflow-decision-test-type\"\n\ttl := \"integration-signal-self-workflow-decision-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain: s.DomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: id,\n\t\t\t\t\tRunID:      we.GetRunID(),\n\t\t\t\t},\n\t\t\t\tSignalName: signalName,\n\t\t\t\tInput:      signalInput,\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start workflows to make some progress.\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Signal the foreign workflow with this decision request.\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tsignalSentFailed := false\n\tintiatedEventID := 10\nCheckHistoryLoopForCancelSent:\n\tfor i := 1; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tsignalFailedEvent := history.Events[len(history.Events)-2]\n\t\tif *signalFailedEvent.EventType != types.EventTypeSignalExternalWorkflowExecutionFailed {\n\t\t\ts.Logger.Info(\"Cancellaton not cancelled yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue CheckHistoryLoopForCancelSent\n\t\t}\n\n\t\tsignalExternalWorkflowExecutionFailedEventAttributes := signalFailedEvent.SignalExternalWorkflowExecutionFailedEventAttributes\n\t\ts.Equal(int64(intiatedEventID), signalExternalWorkflowExecutionFailedEventAttributes.InitiatedEventID)\n\t\ts.Equal(id, signalExternalWorkflowExecutionFailedEventAttributes.WorkflowExecution.WorkflowID)\n\t\ts.Equal(we.RunID, signalExternalWorkflowExecutionFailedEventAttributes.WorkflowExecution.RunID)\n\n\t\tsignalSentFailed = true\n\t\tbreak\n\t}\n\n\ts.True(signalSentFailed)\n\n}\n\nfunc (s *IntegrationSuite) TestSignalWithStartWorkflow() {\n\tid := \"integration-signal-with-start-workflow-test\"\n\twt := \"integration-signal-with-start-workflow-test-type\"\n\ttl := \"integration-signal-with-start-workflow-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\theader := &types.Header{\n\t\tFields: map[string][]byte{\"tracing\": []byte(\"sample data\")},\n\t}\n\n\t// Start a workflow\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\t// decider logic\n\tworkflowComplete := false\n\tactivityScheduled := false\n\tactivityData := int32(1)\n\tnewWorkflowStarted := false\n\tvar signalEvent, startedEvent *types.HistoryEvent\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tif !activityScheduled {\n\t\t\tactivityScheduled = true\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityData))\n\n\t\t\treturn nil, []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    \"1\",\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t} else if previousStartedEventID > 0 {\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t} else if newWorkflowStarted {\n\t\t\tnewWorkflowStarted = false\n\t\t\tsignalEvent = nil\n\t\t\tstartedEvent = nil\n\t\t\tfor _, event := range history.Events[previousStartedEventID:] {\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionSignaled {\n\t\t\t\t\tsignalEvent = event\n\t\t\t\t}\n\t\t\t\tif *event.EventType == types.EventTypeWorkflowExecutionStarted {\n\t\t\t\t\tstartedEvent = event\n\t\t\t\t}\n\t\t\t}\n\t\t\tif signalEvent != nil && startedEvent != nil {\n\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t}\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn nil, []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\t// activity handler\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Make first decision to schedule activity\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\t// Send a signal\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\twfIDReusePolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\tsRequest := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tHeader:                              header,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tSignalName:                          signalName,\n\t\tSignalInput:                         signalInput,\n\t\tIdentity:                            identity,\n\t\tWorkflowIDReusePolicy:               &wfIDReusePolicy,\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err := s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err)\n\ts.Equal(we.GetRunID(), resp.GetRunID())\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Terminate workflow execution\n\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t\tReason:   \"test signal\",\n\t\tDetails:  nil,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\t// Send signal to terminated workflow\n\tsignalName = \"signal to terminate\"\n\tsignalInput = []byte(\"signal to terminate input.\")\n\tsRequest.SignalName = signalName\n\tsRequest.SignalInput = signalInput\n\tsRequest.WorkflowID = id\n\tsRequest.RequestID = uuid.New()\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err = s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n\ts.NotEqual(we.GetRunID(), resp.GetRunID())\n\tnewWorkflowStarted = true\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\ts.True(startedEvent != nil)\n\ts.Equal(header, startedEvent.WorkflowExecutionStartedEventAttributes.Header)\n\n\t// Send signal to not existed workflow\n\tid = \"integration-signal-with-start-workflow-test-non-exist\"\n\tsignalName = \"signal to non exist\"\n\tsignalInput = []byte(\"signal to non exist input.\")\n\tsRequest.SignalName = signalName\n\tsRequest.SignalInput = signalInput\n\tsRequest.WorkflowID = id\n\tsRequest.RequestID = uuid.New()\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err = s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n\tnewWorkflowStarted = true\n\n\t// Process signal in decider\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ts.False(workflowComplete)\n\ts.True(signalEvent != nil)\n\ts.Equal(signalName, signalEvent.WorkflowExecutionSignaledEventAttributes.SignalName)\n\ts.Equal(signalInput, signalEvent.WorkflowExecutionSignaledEventAttributes.Input)\n\ts.Equal(identity, signalEvent.WorkflowExecutionSignaledEventAttributes.Identity)\n\n\t// Assert visibility is correct\n\tlistOpenRequest := &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain:          s.DomainName,\n\t\tMaximumPageSize: 100,\n\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tlistResp, err := s.Engine.ListOpenWorkflowExecutions(ctx, listOpenRequest)\n\ts.NoError(err)\n\ts.Equal(1, len(listResp.Executions))\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// Terminate workflow execution and assert visibility is correct\n\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t\tReason:   \"kill workflow\",\n\t\tDetails:  nil,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\tfor i := 0; i < 10; i++ { // retry\n\t\tctx, cancel := createContext()\n\t\tlistResp, err = s.Engine.ListOpenWorkflowExecutions(ctx, listOpenRequest)\n\t\tcancel()\n\t\ts.NoError(err)\n\t\tif len(listResp.Executions) == 0 {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.Equal(0, len(listResp.Executions))\n\n\tlistClosedRequest := &types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain:          s.DomainName,\n\t\tMaximumPageSize: 100,\n\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tlistClosedResp, err := s.Engine.ListClosedWorkflowExecutions(ctx, listClosedRequest)\n\ts.NoError(err)\n\ts.Equal(1, len(listClosedResp.Executions))\n}\n\nfunc (s *IntegrationSuite) TestSignalWithStartWorkflow_IDReusePolicy() {\n\tid := \"integration-signal-with-start-workflow-id-reuse-test\"\n\twt := \"integration-signal-with-start-workflow-id-reuse-test-type\"\n\ttl := \"integration-signal-with-start-workflow-id-reuse-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\t// Start a workflow\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(1)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\t// Start workflows, make some progress and complete workflow\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\ts.True(workflowComplete)\n\n\t// test policy WorkflowIDReusePolicyRejectDuplicate\n\tsignalName := \"my signal\"\n\tsignalInput := []byte(\"my signal input.\")\n\twfIDReusePolicy := types.WorkflowIDReusePolicyRejectDuplicate\n\tsRequest := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tSignalName:                          signalName,\n\t\tSignalInput:                         signalInput,\n\t\tIdentity:                            identity,\n\t\tWorkflowIDReusePolicy:               &wfIDReusePolicy,\n\t}\n\tctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)\n\t_, err = s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\tcancel()\n\ts.Error(err)\n\terrMsg := err.(*types.WorkflowExecutionAlreadyStartedError).GetMessage()\n\ts.True(strings.Contains(errMsg, \"reject duplicate workflow ID\"))\n\n\t// test policy WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\twfIDReusePolicy = types.WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\tctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)\n\tsRequest.RequestID = uuid.New()\n\t_, err = s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\tcancel()\n\ts.Error(err)\n\terrMsg = err.(*types.WorkflowExecutionAlreadyStartedError).GetMessage()\n\ts.True(strings.Contains(errMsg, \"allow duplicate workflow ID if last run failed\"))\n\n\t// test policy WorkflowIDReusePolicyAllowDuplicate\n\twfIDReusePolicy = types.WorkflowIDReusePolicyAllowDuplicate\n\tctx, cancel = context.WithTimeout(context.Background(), 5*time.Second)\n\tsRequest.RequestID = uuid.New()\n\tresp, err := s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\tcancel()\n\ts.Nil(err)\n\ts.NotEmpty(resp.GetRunID())\n\n\t// Terminate workflow execution\n\tterminateWorkflow := func() {\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\terr = s.Engine.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t\tReason:   \"test WorkflowIDReusePolicyAllowDuplicateFailedOnly\",\n\t\t\tDetails:  nil,\n\t\t\tIdentity: identity,\n\t\t})\n\t\ts.Nil(err)\n\t}\n\tterminateWorkflow()\n\n\t// test policy WorkflowIDReusePolicyAllowDuplicateFailedOnly success start\n\twfIDReusePolicy = types.WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\tsRequest.RequestID = uuid.New()\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp, err = s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err)\n\ts.NotEmpty(resp.GetRunID())\n\n\t// test policy WorkflowIDReusePolicyTerminateIfRunning\n\twfIDReusePolicy = types.WorkflowIDReusePolicyTerminateIfRunning\n\tsRequest.RequestID = uuid.New()\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp1, err1 := s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err1)\n\ts.NotEmpty(resp1)\n\t// verify terminate status\n\texecutionTerminated := false\nGetHistoryLoop:\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: s.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: id,\n\t\t\t\tRunID:      resp.RunID,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif lastEvent.GetEventType() != types.EventTypeWorkflowExecutionTerminated {\n\t\t\ts.Logger.Warn(\"Execution not terminated yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tterminateEventAttributes := lastEvent.WorkflowExecutionTerminatedEventAttributes\n\t\ts.Equal(engineimpl.TerminateIfRunningReason, terminateEventAttributes.GetReason())\n\t\ts.Equal(fmt.Sprintf(engineimpl.TerminateIfRunningDetailsTemplate, resp1.GetRunID()), string(terminateEventAttributes.Details))\n\t\ts.Equal(execution.IdentityHistoryService, terminateEventAttributes.GetIdentity())\n\t\texecutionTerminated = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(executionTerminated)\n\t// terminate current run\n\tterminateWorkflow()\n\t// test clean start with WorkflowIDReusePolicyTerminateIfRunning\n\tsRequest.RequestID = uuid.New()\n\tctx, cancel = createContext()\n\tdefer cancel()\n\tresp2, err2 := s.Engine.SignalWithStartWorkflowExecution(ctx, sRequest)\n\ts.Nil(err2)\n\ts.NotEmpty(resp2)\n\ts.NotEqual(resp1.GetRunID(), resp2.GetRunID())\n}\n"
  },
  {
    "path": "host/size_limit_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"flag\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestSizeLimitIntegrationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_sizelimit_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(SizeLimitIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\n// This cluster use customized threshold for history config\nfunc (s *SizeLimitIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n}\n\nfunc (s *SizeLimitIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *SizeLimitIntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *SizeLimitIntegrationSuite) TestTerminateWorkflowCausedBySizeLimit() {\n\tid := \"integration-terminate-workflow-by-size-limit-test\"\n\twt := \"integration-terminate-workflow-by-size-limit-test-type\"\n\ttl := \"integration-terminate-workflow-by-size-limit-test-tasklist\"\n\tidentity := \"worker1\"\n\tactivityName := \"activity_type1\"\n\n\tworkflowType := &types.WorkflowType{}\n\tworkflowType.Name = wt\n\n\ttaskList := &types.TaskList{}\n\ttaskList.Name = tl\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\twe, err0 := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err0)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tactivityCount := int32(4)\n\tactivityCounter := int32(0)\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := int32(0); i < activityCount-1; i++ {\n\t\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\t\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\t\ts.Nil(err)\n\n\t\terr = poller.PollAndProcessActivityTask(false)\n\t\ts.Logger.Info(\"PollAndProcessActivityTask\", tag.Error(err))\n\t\ts.Nil(err)\n\t}\n\n\t// process this decision will trigger history exceed limit error\n\t_, err := poller.PollAndProcessDecisionTask(false, false)\n\ts.Logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\tctx, cancel = createContext()\n\tdefer cancel()\n\t// verify last event is terminated event\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t\tRunID:      we.GetRunID(),\n\t\t},\n\t})\n\ts.Nil(err)\n\thistory := historyResponse.History\n\tlastEvent := history.Events[len(history.Events)-1]\n\ts.Equal(types.EventTypeWorkflowExecutionFailed, lastEvent.GetEventType())\n\tfailedEventAttributes := lastEvent.WorkflowExecutionFailedEventAttributes\n\ts.Equal(common.FailureReasonSizeExceedsLimit, failedEventAttributes.GetReason())\n\n\t// verify visibility is correctly processed from open to close\n\tisCloseCorrect := false\n\tfor i := 0; i < 10; i++ {\n\t\tctx, cancel := createContext()\n\t\tresp, err1 := s.Engine.ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          s.DomainName,\n\t\t\tMaximumPageSize: 100,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\t\tWorkflowID: id,\n\t\t\t},\n\t\t})\n\t\tcancel()\n\t\ts.Nil(err1)\n\t\tif len(resp.Executions) == 1 {\n\t\t\tisCloseCorrect = true\n\t\t\tbreak\n\t\t}\n\t\ts.Logger.Info(\"Closed WorkflowExecution is not yet visible\")\n\t\ttime.Sleep(100 * time.Millisecond)\n\t}\n\ts.True(isCloseCorrect)\n}\n"
  },
  {
    "path": "host/task_list_isolation_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage host\n\nimport (\n\t\"errors\"\n\t\"flag\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tisolationTl = \"integration-task-list-isolation-tl\"\n)\n\nfunc TestTaskListIsolationSuite(t *testing.T) {\n\tflag.Parse()\n\n\tvar isolationGroups = []any{\n\t\t\"a\", \"b\", \"c\",\n\t}\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/task_list_test_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\tclusterConfig.FrontendDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:            true,\n\t\tdynamicproperties.AllIsolationGroups:                 isolationGroups,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions: 1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:  1,\n\t}\n\tclusterConfig.HistoryDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:            true,\n\t\tdynamicproperties.AllIsolationGroups:                 isolationGroups,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions: 1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:  1,\n\t}\n\tclusterConfig.MatchingDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:            true,\n\t\tdynamicproperties.AllIsolationGroups:                 isolationGroups,\n\t\tdynamicproperties.TaskIsolationDuration:              time.Second * 5,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions: 1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:  1,\n\t}\n\n\ts := new(TaskListIsolationIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) TestTaskListIsolation() {\n\taPoller := s.createPoller(\"a\")\n\tbPoller := s.createPoller(\"b\")\n\n\tcancelB := bPoller.PollAndProcessDecisions()\n\tdefer cancelB()\n\tcancelA := aPoller.PollAndProcessDecisions()\n\tdefer cancelA()\n\n\t// Give pollers time to start\n\ttime.Sleep(time.Second)\n\n\t// Running a single workflow is a bit of a coinflip: if isolation didn't work, it would pass 50% of the time.\n\t// Run 10 workflows to demonstrate that we consistently isolate tasks to the correct poller\n\tfor i := 0; i < 10; i++ {\n\t\trunID := s.startWorkflow(\"a\").RunID\n\t\tresult, err := s.getWorkflowResult(runID)\n\t\ts.NoError(err)\n\t\ts.Equal(\"a\", result)\n\t}\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) TestTaskListIsolationLeak() {\n\trunID := s.startWorkflow(\"a\").RunID\n\n\tbPoller := s.createPoller(\"b\")\n\t// B will get the task as there are no pollers from A\n\tcancelB := bPoller.PollAndProcessDecisions()\n\tdefer cancelB()\n\n\tresult, err := s.getWorkflowResult(runID)\n\ts.NoError(err)\n\ts.Equal(\"b\", result)\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) createPoller(group string) *TaskPoller {\n\treturn &TaskPoller{\n\t\tEngine:   s.Engine,\n\t\tDomain:   s.DomainName,\n\t\tTaskList: &types.TaskList{Name: isolationTl, Kind: types.TaskListKindNormal.Ptr()},\n\t\tIdentity: group,\n\t\tDecisionHandler: func(execution *types.WorkflowExecution, wt *types.WorkflowType, previousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t// Complete the workflow with the group name\n\t\t\treturn []byte(strconv.Itoa(0)), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(group),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t},\n\t\tLogger:      s.Logger,\n\t\tT:           s.T(),\n\t\tCallOptions: []yarpc.CallOption{withIsolationGroup(group)},\n\t}\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) startWorkflow(group string) *types.StartWorkflowExecutionResponse {\n\tidentity := \"test\"\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:  uuid.New(),\n\t\tDomain:     s.DomainName,\n\t\tWorkflowID: s.T().Name(),\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"integration-task-list-isolation-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: isolationTl,\n\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t},\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(10),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresult, err := s.Engine.StartWorkflowExecution(ctx, request, withIsolationGroup(group))\n\ts.Nil(err)\n\n\treturn result\n}\n\nfunc (s *TaskListIsolationIntegrationSuite) getWorkflowResult(runID string) (string, error) {\n\tctx, cancel := createContext()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: s.T().Name(),\n\t\t\tRunID:      runID,\n\t\t},\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeCloseEvent.Ptr(),\n\t\tWaitForNewEvent:        true,\n\t})\n\tcancel()\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\thistory := historyResponse.History\n\n\tlastEvent := history.Events[len(history.Events)-1]\n\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionCompleted {\n\t\treturn \"\", errors.New(\"workflow didn't complete\")\n\t}\n\n\treturn string(lastEvent.WorkflowExecutionCompletedEventAttributes.Result), nil\n\n}\n\nfunc withIsolationGroup(group string) yarpc.CallOption {\n\treturn yarpc.WithHeader(common.ClientIsolationGroupHeaderName, group)\n}\n"
  },
  {
    "path": "host/task_list_test.go",
    "content": "package host\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst testWorkflowType = \"test-workflow\"\n\nfunc TestTaskListSuite(t *testing.T) {\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/task_list_test_cluster.yaml\")\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\tclusterConfig.FrontendDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:            false,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions: 1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:  1,\n\t}\n\tclusterConfig.HistoryDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:            false,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions: 1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:  1,\n\t}\n\tclusterConfig.MatchingDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.EnableTasklistIsolation:              false,\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions:   1,\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:    1,\n\t\tdynamicproperties.MatchingEnableReturnAllTaskListKinds: true,\n\t}\n\n\ts := new(TaskListIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *TaskListIntegrationSuite) SetupSuite() {\n\ts.setupSuite()\n}\n\nfunc (s *TaskListIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *TaskListIntegrationSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.TaskListName = s.T().Name()\n}\n\nfunc (s *TaskListIntegrationSuite) TestDescribeTaskList_Status() {\n\texpectedTl := &types.TaskList{Name: s.TaskListName, Kind: types.TaskListKindNormal.Ptr()}\n\n\tinitialStatus := &types.TaskListStatus{\n\t\tBacklogCountHint: 0,\n\t\tReadLevel:        0,\n\t\tAckLevel:         0,\n\t\tRatePerSecond:    100000,\n\t\tTaskIDBlock: &types.TaskIDBlock{\n\t\t\tStartID: 1,\n\t\t\tEndID:   100000,\n\t\t},\n\t\t// Isolation is disabled\n\t\t// IsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{},\n\t\tNewTasksPerSecond: 0,\n\t\tEmpty:             true,\n\t}\n\twithOneTask := &types.TaskListStatus{\n\t\tBacklogCountHint: 1,\n\t\tReadLevel:        1,\n\t\tAckLevel:         0,\n\t\tRatePerSecond:    100000,\n\t\tTaskIDBlock: &types.TaskIDBlock{\n\t\t\tStartID: 1,\n\t\t\tEndID:   100000,\n\t\t},\n\t\t// Isolation is disabled and NewTasksPerSecond is volatile\n\t\t// IsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{},\n\t\t// NewTasksPerSecond:     0,\n\t\tEmpty: false,\n\t}\n\tcompletedStatus := &types.TaskListStatus{\n\t\tBacklogCountHint: 0,\n\t\tReadLevel:        1,\n\t\tAckLevel:         1,\n\t\tRatePerSecond:    100000,\n\t\tTaskIDBlock: &types.TaskIDBlock{\n\t\t\tStartID: 1,\n\t\t\tEndID:   100000,\n\t\t},\n\t\t// Isolation is disabled and NewTasksPerSecond is volatile\n\t\t// IsolationGroupMetrics: nil,\n\t\t// NewTasksPerSecond:     0,\n\t\tEmpty: true,\n\t}\n\n\t// 0. Check before any tasks\n\tresponse := s.describeTaskList(types.TaskListTypeDecision)\n\tresponse.TaskListStatus.IsolationGroupMetrics = nil\n\ts.Equal(expectedTl, response.TaskList)\n\ts.Equal(initialStatus, response.TaskListStatus)\n\n\t// 1. After a task has been added but not yet completed\n\trunID := s.startWorkflow(types.TaskListKindNormal).RunID\n\tresponse = s.describeUntil(func(response *types.DescribeTaskListResponse) bool {\n\t\treturn response.TaskListStatus.BacklogCountHint == 1\n\t})\n\tresponse.TaskListStatus.IsolationGroupMetrics = nil\n\tresponse.TaskListStatus.NewTasksPerSecond = 0\n\ts.Equal(withOneTask, response.TaskListStatus)\n\n\t// 2. After the task has been completed\n\tpoller := s.createPoller(\"result\", types.TaskListKindNormal)\n\tcancelPoller := poller.PollAndProcessDecisions()\n\tdefer cancelPoller()\n\t_, err := s.getWorkflowResult(runID)\n\ts.NoError(err, \"failed to get workflow result\")\n\n\tresponse = s.describeTaskList(types.TaskListTypeDecision)\n\tresponse.TaskListStatus.IsolationGroupMetrics = nil\n\tresponse.TaskListStatus.NewTasksPerSecond = 0\n\ts.Equal(completedStatus, response.TaskListStatus)\n}\n\n// Run a Workflow containing only decision tasks on an ephemeral TaskList\nfunc (s *TaskListIntegrationSuite) TestEphemeralTaskList() {\n\trunID := s.startWorkflow(types.TaskListKindEphemeral).RunID\n\n\tresponse := s.describeWorkflow(s.workflowID(), runID)\n\ts.Equal(types.TaskListKindEphemeral, response.WorkflowExecutionInfo.TaskList.GetKind())\n\n\tfirstEvent := s.getStartedEvent(runID)\n\ts.Equal(types.TaskListKindEphemeral, firstEvent.TaskList.GetKind())\n\n\ttaskList := s.waitUntilTaskListExists(types.TaskListTypeDecision)\n\ts.Equal(types.TaskListKindEphemeral, *taskList.TaskList.Kind)\n\n\tpoller := s.createPoller(\"result\", types.TaskListKindEphemeral)\n\tcancelPoller := poller.PollAndProcessDecisions()\n\tdefer cancelPoller()\n\t_, err := s.getWorkflowResult(runID)\n\ts.NoError(err, \"failed to get workflow result\")\n}\n\n// Run a workflow on a normal TaskList, with an activity on an ephemeral TaskList\nfunc (s *TaskListIntegrationSuite) TestEphemeralTaskList_EphemeralActivity() {\n\tactivity := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID: \"the-one\",\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: \"activity-type\",\n\t\t},\n\t\tDomain:                        s.DomainName,\n\t\tTaskList:                      &types.TaskList{Name: s.TaskListName, Kind: types.TaskListKindEphemeral.Ptr()},\n\t\tInput:                         []byte(\"input\"),\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(10),\n\t}\n\n\t// Normal workflow with an ephemeral activity\n\trunID := s.startWorkflow(types.TaskListKindNormal).RunID\n\n\ttaskList := s.waitUntilTaskListExists(types.TaskListTypeDecision)\n\ts.Equal(types.TaskListKindNormal, taskList.TaskList.GetKind())\n\n\tpoller := s.decisionPoller(types.TaskListKindNormal, &types.Decision{DecisionType: types.DecisionTypeScheduleActivityTask.Ptr(), ScheduleActivityTaskDecisionAttributes: activity})\n\tcancelPoller := poller.PollAndProcessDecisions()\n\tdefer cancelPoller()\n\n\ttaskList = s.waitUntilTaskListExists(types.TaskListTypeActivity)\n\ts.Equal(types.TaskListKindEphemeral, taskList.TaskList.GetKind())\n\n\tactivityPoller := s.createEchoActivityPoller(types.TaskListKindEphemeral)\n\tcancelActivityPoller := activityPoller.PollAndProcessActivities()\n\tdefer cancelActivityPoller()\n\n\t_, err := s.getWorkflowResult(runID)\n\ts.NoError(err, \"failed to get workflow result\")\n\n\tscheduled := s.getActivityScheduledEvent(runID)\n\ts.Equal(types.TaskListKindEphemeral, scheduled.TaskList.GetKind())\n}\n\n// Run a workflow on an Ephemeral TaskList, which forces the activity to be dispatched to an Ephemeral TaskList.\nfunc (s *TaskListIntegrationSuite) TestEphemeralTaskList_EphemeralWorkflow() {\n\tactivity := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID: \"the-one\",\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: \"activity-type\",\n\t\t},\n\t\tDomain:                        s.DomainName,\n\t\tTaskList:                      &types.TaskList{Name: s.TaskListName, Kind: types.TaskListKindNormal.Ptr()},\n\t\tInput:                         []byte(\"input\"),\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(10),\n\t}\n\n\t// Normal workflow with an ephemeral activity\n\trunID := s.startWorkflow(types.TaskListKindEphemeral).RunID\n\n\t// Just for sanity\n\tresponse := s.describeWorkflow(s.workflowID(), runID)\n\ts.Equal(types.TaskListKindEphemeral, response.WorkflowExecutionInfo.TaskList.GetKind())\n\n\ttaskList := s.waitUntilTaskListExists(types.TaskListTypeDecision)\n\ts.Equal(types.TaskListKindEphemeral, taskList.TaskList.GetKind())\n\n\tpoller := s.decisionPoller(types.TaskListKindEphemeral, &types.Decision{DecisionType: types.DecisionTypeScheduleActivityTask.Ptr(), ScheduleActivityTaskDecisionAttributes: activity})\n\tcancelPoller := poller.PollAndProcessDecisions()\n\tdefer cancelPoller()\n\n\ttaskList = s.waitUntilTaskListExists(types.TaskListTypeActivity)\n\ts.Equal(types.TaskListKindEphemeral, taskList.TaskList.GetKind())\n\n\tactivityPoller := s.createEchoActivityPoller(types.TaskListKindEphemeral)\n\tcancelActivityPoller := activityPoller.PollAndProcessActivities()\n\tdefer cancelActivityPoller()\n\n\t_, err := s.getWorkflowResult(runID)\n\ts.NoError(err, \"failed to get workflow result\")\n\n\tscheduled := s.getActivityScheduledEvent(runID)\n\ts.Equal(types.TaskListKindEphemeral, scheduled.TaskList.GetKind())\n}\n\nfunc (s *TaskListIntegrationSuite) TestEphemeralTaskList_EphemeralChildWorkflow() {\n\tchildWorkflowID := \"ephemeral-child-workflow\"\n\tactivity := &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tWorkflowID: childWorkflowID,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"noop\",\n\t\t},\n\t\tDomain:                              s.DomainName,\n\t\tTaskList:                            &types.TaskList{Name: s.TaskListName, Kind: types.TaskListKindNormal.Ptr()},\n\t\tInput:                               []byte(\"input\"),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(20),\n\t\tControl:                             nil,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t}\n\n\t// Normal workflow with an ephemeral activity\n\trunID := s.startWorkflow(types.TaskListKindEphemeral).RunID\n\n\t// Just for sanity\n\tresponse := s.describeWorkflow(s.workflowID(), runID)\n\ts.Equal(types.TaskListKindEphemeral, response.WorkflowExecutionInfo.TaskList.GetKind())\n\n\tpoller := s.decisionPoller(types.TaskListKindEphemeral, &types.Decision{DecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(), StartChildWorkflowExecutionDecisionAttributes: activity})\n\tcancelPoller := poller.PollAndProcessDecisions()\n\tdefer cancelPoller()\n\n\t_, err := s.getWorkflowResult(runID)\n\ts.NoError(err, \"failed to get workflow result\")\n\n\t// Ensure the child is also ephemeral\n\tresponse = s.describeWorkflow(childWorkflowID, \"\")\n\ts.Equal(types.TaskListKindEphemeral, response.WorkflowExecutionInfo.TaskList.GetKind())\n\n}\n\nfunc (s *TaskListIntegrationSuite) createEchoActivityPoller(tlKind types.TaskListKind) *TaskPoller {\n\treturn &TaskPoller{\n\t\tEngine:   s.Engine,\n\t\tDomain:   s.DomainName,\n\t\tTaskList: &types.TaskList{Name: s.TaskListName, Kind: tlKind.Ptr()},\n\t\tIdentity: \"activity-poller\",\n\t\tActivityHandler: func(execution *types.WorkflowExecution, activityType *types.ActivityType, ActivityID string, input []byte, takeToken []byte) ([]byte, bool, error) {\n\t\t\treturn input, false, nil\n\t\t},\n\t\tLogger:      s.Logger,\n\t\tT:           s.T(),\n\t\tCallOptions: []yarpc.CallOption{},\n\t}\n}\n\n// decisionPoller creates a decision poller capable of running Workflows consisting of a single decision\nfunc (s *TaskListIntegrationSuite) decisionPoller(tlKind types.TaskListKind, decision *types.Decision) *TaskPoller {\n\treturn &TaskPoller{\n\t\tEngine:   s.Engine,\n\t\tDomain:   s.DomainName,\n\t\tTaskList: &types.TaskList{Name: s.TaskListName, Kind: tlKind.Ptr()},\n\t\tIdentity: \"decision-poller\",\n\t\tDecisionHandler: func(execution *types.WorkflowExecution, wt *types.WorkflowType, previousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t// Treat any other workflow type as a no-op, and allow passing nil as a no-op\n\t\t\tif wt.GetName() != testWorkflowType || decision == nil {\n\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tResult: []byte(\"done\"),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t}\n\t\t\t// Ignore the DecisionTaskScheduled and DecisionTaskStarted that'll be at the end\n\t\t\tlatestEvent := history.Events[len(history.Events)-3]\n\t\t\t// Started Event only. We support one of three things:\n\t\t\t// - Activity Scheduled. We wait until it finishes\n\t\t\t// - Child Workflow. We wait until it finishes, ignoring the start.\n\t\t\t// - nil. We complete immediately\n\t\t\tif *latestEvent.EventType == types.EventTypeWorkflowExecutionStarted {\n\t\t\t\tif decision != nil {\n\t\t\t\t\treturn nil, []*types.Decision{decision}, nil\n\t\t\t\t}\n\t\t\t}\n\t\t\t// Do nothing until the child workflow completes\n\t\t\tif *latestEvent.EventType == types.EventTypeChildWorkflowExecutionStarted {\n\t\t\t\treturn nil, []*types.Decision{}, nil\n\t\t\t}\n\t\t\t// Once the decision is done we finish the workflow\n\t\t\tif *latestEvent.EventType == types.EventTypeActivityTaskCompleted || *latestEvent.EventType == types.EventTypeChildWorkflowExecutionCompleted {\n\t\t\t\treturn nil, []*types.Decision{{\n\t\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tResult: []byte(\"done\"),\n\t\t\t\t\t},\n\t\t\t\t}}, nil\n\t\t\t}\n\t\t\treturn nil, nil, fmt.Errorf(\"unexpected event type: %v\", latestEvent.EventType)\n\t\t},\n\t\tLogger:      s.Logger,\n\t\tT:           s.T(),\n\t\tCallOptions: []yarpc.CallOption{},\n\t}\n}\n\nfunc (s *TaskListIntegrationSuite) createPoller(identity string, tlKind types.TaskListKind) *TaskPoller {\n\treturn &TaskPoller{\n\t\tEngine:   s.Engine,\n\t\tDomain:   s.DomainName,\n\t\tTaskList: &types.TaskList{Name: s.TaskListName, Kind: tlKind.Ptr()},\n\t\tIdentity: identity,\n\t\tDecisionHandler: func(execution *types.WorkflowExecution, wt *types.WorkflowType, previousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\t\t// Complete the workflow\n\t\t\treturn []byte(strconv.Itoa(0)), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tResult: []byte(identity),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t},\n\t\tLogger:      s.Logger,\n\t\tT:           s.T(),\n\t\tCallOptions: []yarpc.CallOption{},\n\t}\n}\n\nfunc (s *TaskListIntegrationSuite) startWorkflow(tlKind types.TaskListKind) *types.StartWorkflowExecutionResponse {\n\tidentity := \"test\"\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:  uuid.New(),\n\t\tDomain:     s.DomainName,\n\t\tWorkflowID: s.T().Name(),\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"test-workflow\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: s.TaskListName,\n\t\t\tKind: tlKind.Ptr(),\n\t\t},\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(10),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            identity,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresult, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\treturn result\n}\n\nfunc (s *TaskListIntegrationSuite) workflowID() string {\n\treturn s.T().Name()\n}\n\nfunc (s *TaskListIntegrationSuite) describeWorkflow(workflowID, runID string) *types.DescribeWorkflowExecutionResponse {\n\trequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresult, err := s.Engine.DescribeWorkflowExecution(ctx, request)\n\ts.Nil(err)\n\n\treturn result\n}\n\nfunc (s *TaskListIntegrationSuite) describeUntil(condition func(response *types.DescribeTaskListResponse) bool) *types.DescribeTaskListResponse {\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\ts.FailNow(\"timed out waiting for condition\")\n\t\tdefault:\n\t\t}\n\t\tresult, err := s.Engine.DescribeTaskList(ctx, &types.DescribeTaskListRequest{\n\t\t\tDomain:       s.DomainName,\n\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: s.TaskListName,\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tIncludeTaskListStatus: true,\n\t\t})\n\t\tif err != nil {\n\t\t\ts.T().Log(\"failed to describe task list: %w\", err)\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\t\t\tcontinue\n\t\t}\n\t\tif condition(result) {\n\t\t\treturn result\n\t\t}\n\t\ttime.Sleep(50 * time.Millisecond)\n\t}\n}\n\nfunc (s *TaskListIntegrationSuite) waitUntilTaskListExists(tlType types.TaskListType) *types.DescribeTaskListResponse {\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\ts.FailNow(\"timed out waiting for TaskList\")\n\t\tdefault:\n\t\t}\n\t\tresult, err := s.Engine.GetTaskListsByDomain(ctx, &types.GetTaskListsByDomainRequest{\n\t\t\tDomain: s.DomainName,\n\t\t})\n\t\tif err != nil {\n\t\t\ts.T().Log(\"failed to describe task list: %w\", err)\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\t\t\tcontinue\n\t\t}\n\t\tif tlResponse, ok := result.ActivityTaskListMap[s.TaskListName]; ok && tlType == types.TaskListTypeActivity {\n\t\t\treturn tlResponse\n\t\t}\n\t\tif tlResponse, ok := result.DecisionTaskListMap[s.TaskListName]; ok && tlType == types.TaskListTypeDecision {\n\t\t\treturn tlResponse\n\t\t}\n\t\ttime.Sleep(50 * time.Millisecond)\n\t}\n}\n\nfunc (s *TaskListIntegrationSuite) describeTaskList(tlType types.TaskListType) *types.DescribeTaskListResponse {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tresult, err := s.Engine.DescribeTaskList(ctx, &types.DescribeTaskListRequest{\n\t\tDomain:       s.DomainName,\n\t\tTaskListType: tlType.Ptr(),\n\t\tTaskList: &types.TaskList{\n\t\t\tName: s.TaskListName,\n\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t},\n\t\tIncludeTaskListStatus: true,\n\t})\n\ts.NoError(err, \"failed to describe task list\")\n\treturn result\n}\n\nfunc (s *TaskListIntegrationSuite) getStartedEvent(runID string) *types.WorkflowExecutionStartedEventAttributes {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: s.T().Name(),\n\t\t\tRunID:      runID,\n\t\t},\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t})\n\ts.NoError(err, \"failed to get workflow history\")\n\n\thistory := historyResponse.History\n\n\tfirstEvent := history.Events[0]\n\tif *firstEvent.EventType != types.EventTypeWorkflowExecutionStarted {\n\t\ts.FailNow(\"expected first event to be WorkflowExecutionStarted\")\n\t}\n\n\treturn firstEvent.WorkflowExecutionStartedEventAttributes\n}\n\nfunc (s *TaskListIntegrationSuite) getActivityScheduledEvent(runID string) *types.ActivityTaskScheduledEventAttributes {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: s.T().Name(),\n\t\t\tRunID:      runID,\n\t\t},\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t})\n\ts.NoError(err, \"failed to get workflow history\")\n\n\thistory := historyResponse.History\n\n\tfor _, event := range history.Events {\n\t\tif *event.EventType == types.EventTypeActivityTaskScheduled {\n\t\t\treturn event.ActivityTaskScheduledEventAttributes\n\t\t}\n\t}\n\ts.FailNow(\"failed to find EventTypeActivityTaskScheduled\")\n\treturn nil\n}\n\nfunc (s *TaskListIntegrationSuite) getWorkflowResult(runID string) (string, error) {\n\tctx, cancel := createContext()\n\tdefer cancel()\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: s.T().Name(),\n\t\t\tRunID:      runID,\n\t\t},\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeCloseEvent.Ptr(),\n\t\tWaitForNewEvent:        true,\n\t})\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\thistory := historyResponse.History\n\n\tlastEvent := history.Events[len(history.Events)-1]\n\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionCompleted {\n\t\treturn \"\", errors.New(\"workflow didn't complete\")\n\t}\n\n\treturn string(lastEvent.WorkflowExecutionCompletedEventAttributes.Result), nil\n}\n"
  },
  {
    "path": "host/taskpoller.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n)\n\ntype (\n\tdecisionTaskHandler func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error)\n\tactivityTaskHandler func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tActivityID string, input []byte, takeToken []byte) ([]byte, bool, error)\n\n\tqueryHandler func(task *types.PollForDecisionTaskResponse) ([]byte, error)\n\n\t// TaskPoller is used in integration tests to poll decision or activity tasks\n\tTaskPoller struct {\n\t\tEngine                              FrontendClient\n\t\tDomain                              string\n\t\tTaskList                            *types.TaskList\n\t\tStickyTaskList                      *types.TaskList\n\t\tStickyScheduleToStartTimeoutSeconds *int32\n\t\tIdentity                            string\n\t\tDecisionHandler                     decisionTaskHandler\n\t\tActivityHandler                     activityTaskHandler\n\t\tQueryHandler                        queryHandler\n\t\tLogger                              log.Logger\n\t\tT                                   *testing.T\n\t\tCallOptions                         []yarpc.CallOption\n\t}\n)\n\n// PollAndProcessDecisionTask for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTask(dumpHistory bool, dropTask bool) (isQueryTask bool, err error) {\n\treturn p.PollAndProcessDecisionTaskWithAttempt(dumpHistory, dropTask, false, false, int64(0))\n}\n\n// PollAndProcessDecisionTaskWithSticky for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTaskWithSticky(dumpHistory bool, dropTask bool) (isQueryTask bool, err error) {\n\treturn p.PollAndProcessDecisionTaskWithAttempt(dumpHistory, dropTask, true, true, int64(0))\n}\n\n// PollAndProcessDecisionTaskWithoutRetry for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTaskWithoutRetry(dumpHistory bool, dropTask bool) (isQueryTask bool, err error) {\n\treturn p.PollAndProcessDecisionTaskWithAttemptAndRetry(dumpHistory, dropTask, false, false, int64(0), 1)\n}\n\n// PollAndProcessDecisionTaskWithAttempt for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTaskWithAttempt(\n\tdumpHistory bool,\n\tdropTask bool,\n\tpollStickyTaskList bool,\n\trespondStickyTaskList bool,\n\tdecisionAttempt int64,\n) (isQueryTask bool, err error) {\n\n\treturn p.PollAndProcessDecisionTaskWithAttemptAndRetry(\n\t\tdumpHistory,\n\t\tdropTask,\n\t\tpollStickyTaskList,\n\t\trespondStickyTaskList,\n\t\tdecisionAttempt,\n\t\t5)\n}\n\n// PollAndProcessDecisionTaskWithAttemptAndRetry for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTaskWithAttemptAndRetry(\n\tdumpHistory bool,\n\tdropTask bool,\n\tpollStickyTaskList bool,\n\trespondStickyTaskList bool,\n\tdecisionAttempt int64,\n\tretryCount int,\n) (isQueryTask bool, err error) {\n\n\tisQueryTask, _, err = p.PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\t\tdumpHistory,\n\t\tdropTask,\n\t\tpollStickyTaskList,\n\t\trespondStickyTaskList,\n\t\tdecisionAttempt,\n\t\tretryCount,\n\t\tfalse,\n\t\tnil)\n\treturn isQueryTask, err\n}\n\n// PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision for decision tasks\nfunc (p *TaskPoller) PollAndProcessDecisionTaskWithAttemptAndRetryAndForceNewDecision(\n\tdumpHistory bool,\n\tdropTask bool,\n\tpollStickyTaskList bool,\n\trespondStickyTaskList bool,\n\tdecisionAttempt int64,\n\tretryCount int,\n\tforceCreateNewDecision bool,\n\tqueryResult *types.WorkflowQueryResult,\n) (isQueryTask bool, newTask *types.RespondDecisionTaskCompletedResponse, err error) {\nLoop:\n\tfor attempt := 0; attempt < retryCount; attempt++ {\n\n\t\ttaskList := p.TaskList\n\t\tif pollStickyTaskList {\n\t\t\ttaskList = p.StickyTaskList\n\t\t}\n\t\tctx, cancel := createContext()\n\t\tresponse, err1 := p.Engine.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\tDomain:   p.Domain,\n\t\t\tTaskList: taskList,\n\t\t\tIdentity: p.Identity,\n\t\t}, p.CallOptions...)\n\t\tcancel()\n\n\t\tif err1 != nil {\n\t\t\treturn false, nil, err1\n\t\t}\n\n\t\tif response == nil || len(response.TaskToken) == 0 {\n\t\t\tp.Logger.Info(\"Empty Decision task: Polling again.\")\n\t\t\tcontinue Loop\n\t\t}\n\n\t\tif response.GetNextEventID() == 0 {\n\t\t\tp.Logger.Fatal(\"NextEventID is not set for decision or query task\")\n\t\t}\n\n\t\tvar events []*types.HistoryEvent\n\t\tif response.Query == nil || !pollStickyTaskList {\n\t\t\t// if not query task, should have some history events\n\t\t\t// for non sticky query, there should be events returned\n\t\t\thistory := response.History\n\t\t\tif history == nil {\n\t\t\t\tp.Logger.Fatal(\"History is nil\")\n\t\t\t}\n\n\t\t\tevents = history.Events\n\t\t\tif len(events) == 0 {\n\t\t\t\tp.Logger.Fatal(\"History Events are empty\")\n\t\t\t}\n\n\t\t\tnextPageToken := response.NextPageToken\n\t\t\tfor nextPageToken != nil {\n\t\t\t\tctx, cancel := createContext()\n\t\t\t\tresp, err2 := p.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\t\t\tDomain:        p.Domain,\n\t\t\t\t\tExecution:     response.WorkflowExecution,\n\t\t\t\t\tNextPageToken: nextPageToken,\n\t\t\t\t}, p.CallOptions...)\n\t\t\t\tcancel()\n\n\t\t\t\tif err2 != nil {\n\t\t\t\t\treturn false, nil, err2\n\t\t\t\t}\n\n\t\t\t\tevents = append(events, resp.History.Events...)\n\t\t\t\tnextPageToken = resp.NextPageToken\n\t\t\t}\n\t\t} else {\n\t\t\t// for sticky query, there should be NO events returned\n\t\t\t// since worker side already has the state machine and we do not intend to update that.\n\t\t\thistory := response.History\n\t\t\tnextPageToken := response.NextPageToken\n\t\t\tif !(history == nil || (len(history.Events) == 0 && nextPageToken == nil)) {\n\t\t\t\t// if history is not nil, and contains events or next token\n\t\t\t\tp.Logger.Fatal(\"History is not empty for sticky query\")\n\t\t\t}\n\t\t}\n\n\t\tif dropTask {\n\t\t\tp.Logger.Info(\"Dropping Decision task: \")\n\t\t\treturn false, nil, nil\n\t\t}\n\n\t\tif dumpHistory {\n\t\t\tPrettyPrintHistory(response.History, p.Logger)\n\t\t}\n\n\t\t// handle query task response\n\t\tif response.Query != nil {\n\t\t\tblob, err := p.QueryHandler(response)\n\n\t\t\tcompleteRequest := &types.RespondQueryTaskCompletedRequest{TaskToken: response.TaskToken}\n\t\t\tif err != nil {\n\t\t\t\tcompleteType := types.QueryTaskCompletedTypeFailed\n\t\t\t\tcompleteRequest.CompletedType = &completeType\n\t\t\t\tcompleteRequest.ErrorMessage = err.Error()\n\t\t\t} else {\n\t\t\t\tcompleteType := types.QueryTaskCompletedTypeCompleted\n\t\t\t\tcompleteRequest.CompletedType = &completeType\n\t\t\t\tcompleteRequest.QueryResult = blob\n\t\t\t}\n\n\t\t\tctx, cancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondQueryTaskCompleted(ctx, completeRequest, p.CallOptions...)\n\t\t\tcancel()\n\t\t\treturn true, nil, taskErr\n\t\t}\n\n\t\t// handle normal decision task / non query task response\n\t\tvar lastDecisionScheduleEvent *types.HistoryEvent\n\t\tfor _, e := range events {\n\t\t\tif e.GetEventType() == types.EventTypeDecisionTaskScheduled {\n\t\t\t\tlastDecisionScheduleEvent = e\n\t\t\t}\n\t\t}\n\t\tif lastDecisionScheduleEvent != nil && decisionAttempt > 0 {\n\t\t\trequire.Equal(p.T, decisionAttempt, lastDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetAttempt())\n\t\t}\n\n\t\texecutionCtx, decisions, err := p.DecisionHandler(response.WorkflowExecution, response.WorkflowType,\n\t\t\tcommon.Int64Default(response.PreviousStartedEventID), response.StartedEventID, response.History)\n\t\tif err != nil {\n\t\t\tp.Logger.Info(\"Failing Decision. Decision handler failed with error\", tag.Error(err))\n\t\t\tctx, cancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\t\t\tTaskToken: response.TaskToken,\n\t\t\t\tCause:     types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\t\t\tDetails:   []byte(err.Error()),\n\t\t\t\tIdentity:  p.Identity,\n\t\t\t}, p.CallOptions...)\n\t\t\tcancel()\n\t\t\treturn isQueryTask, nil, taskErr\n\t\t}\n\n\t\tp.Logger.Info(\"Completing Decision.  Decisions\", tag.Value(decisions))\n\t\tif !respondStickyTaskList {\n\t\t\t// non sticky tasklist\n\t\t\tctx, cancel := createContext()\n\t\t\tnewTask, err := p.Engine.RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken:                  response.TaskToken,\n\t\t\t\tIdentity:                   p.Identity,\n\t\t\t\tExecutionContext:           executionCtx,\n\t\t\t\tDecisions:                  decisions,\n\t\t\t\tReturnNewDecisionTask:      forceCreateNewDecision,\n\t\t\t\tForceCreateNewDecisionTask: forceCreateNewDecision,\n\t\t\t\tQueryResults:               getQueryResults(response.GetQueries(), queryResult),\n\t\t\t}, p.CallOptions...)\n\t\t\tcancel()\n\t\t\treturn false, newTask, err\n\t\t}\n\t\t// sticky tasklist\n\t\tctx, cancel = createContext()\n\t\tnewTask, err := p.Engine.RespondDecisionTaskCompleted(\n\t\t\tctx,\n\t\t\t&types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken:        response.TaskToken,\n\t\t\t\tIdentity:         p.Identity,\n\t\t\t\tExecutionContext: executionCtx,\n\t\t\t\tDecisions:        decisions,\n\t\t\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\t\t\tWorkerTaskList:                p.StickyTaskList,\n\t\t\t\t\tScheduleToStartTimeoutSeconds: p.StickyScheduleToStartTimeoutSeconds,\n\t\t\t\t},\n\t\t\t\tReturnNewDecisionTask:      forceCreateNewDecision,\n\t\t\t\tForceCreateNewDecisionTask: forceCreateNewDecision,\n\t\t\t\tQueryResults:               getQueryResults(response.GetQueries(), queryResult),\n\t\t\t},\n\t\t\tp.CallOptions...,\n\t\t)\n\t\tcancel()\n\n\t\treturn false, newTask, err\n\t}\n\n\treturn false, nil, tasklist.ErrNoTasks\n}\n\n// HandlePartialDecision for decision task\nfunc (p *TaskPoller) HandlePartialDecision(response *types.PollForDecisionTaskResponse) (\n\t*types.RespondDecisionTaskCompletedResponse, error) {\n\tif response == nil || len(response.TaskToken) == 0 {\n\t\tp.Logger.Info(\"Empty Decision task: Polling again.\")\n\t\treturn nil, nil\n\t}\n\n\tvar events []*types.HistoryEvent\n\thistory := response.History\n\tif history == nil {\n\t\tp.Logger.Fatal(\"History is nil\")\n\t}\n\n\tevents = history.Events\n\tif len(events) == 0 {\n\t\tp.Logger.Fatal(\"History Events are empty\")\n\t}\n\n\texecutionCtx, decisions, err := p.DecisionHandler(response.WorkflowExecution, response.WorkflowType,\n\t\tcommon.Int64Default(response.PreviousStartedEventID), response.StartedEventID, response.History)\n\tif err != nil {\n\t\tp.Logger.Info(\"Failing Decision. Decision handler failed with error: %v\", tag.Error(err))\n\t\tctx, cancel := createContext()\n\t\tdefer cancel()\n\t\treturn nil, p.Engine.RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tCause:     types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\t\tDetails:   []byte(err.Error()),\n\t\t\tIdentity:  p.Identity,\n\t\t}, p.CallOptions...)\n\t}\n\n\tp.Logger.Info(\"Completing Decision.  Decisions: %v\", tag.Value(decisions))\n\n\t// sticky tasklist\n\tctx, cancel := createContext()\n\tdefer cancel()\n\tnewTask, err := p.Engine.RespondDecisionTaskCompleted(\n\t\tctx,\n\t\t&types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        response.TaskToken,\n\t\t\tIdentity:         p.Identity,\n\t\t\tExecutionContext: executionCtx,\n\t\t\tDecisions:        decisions,\n\t\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\t\tWorkerTaskList:                p.StickyTaskList,\n\t\t\t\tScheduleToStartTimeoutSeconds: p.StickyScheduleToStartTimeoutSeconds,\n\t\t\t},\n\t\t\tReturnNewDecisionTask:      true,\n\t\t\tForceCreateNewDecisionTask: true,\n\t\t},\n\t\tp.CallOptions...,\n\t)\n\n\treturn newTask, err\n}\n\n// PollAndProcessActivityTask for activity tasks\nfunc (p *TaskPoller) PollAndProcessActivityTask(dropTask bool) error {\n\tfor attempt := 0; attempt < 5; attempt++ {\n\t\tctx, ctxCancel := createContext()\n\t\tresponse, err1 := p.Engine.PollForActivityTask(ctx, &types.PollForActivityTaskRequest{\n\t\t\tDomain:   p.Domain,\n\t\t\tTaskList: p.TaskList,\n\t\t\tIdentity: p.Identity,\n\t\t}, p.CallOptions...)\n\t\tctxCancel()\n\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\n\t\tif response == nil || len(response.TaskToken) == 0 {\n\t\t\tp.Logger.Info(\"Empty Activity task: Polling again.\")\n\t\t\treturn nil\n\t\t}\n\n\t\tif dropTask {\n\t\t\tp.Logger.Info(\"Dropping Activity task: \")\n\t\t\treturn nil\n\t\t}\n\t\tp.Logger.Debug(\"Received Activity task\", tag.Value(response))\n\n\t\tresult, cancel, err2 := p.ActivityHandler(response.WorkflowExecution, response.ActivityType, response.ActivityID,\n\t\t\tresponse.Input, response.TaskToken)\n\t\tif cancel {\n\t\t\tp.Logger.Info(\"Executing RespondActivityTaskCanceled\")\n\t\t\tctx, ctxCancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondActivityTaskCanceled(ctx, &types.RespondActivityTaskCanceledRequest{\n\t\t\t\tTaskToken: response.TaskToken,\n\t\t\t\tDetails:   []byte(\"details\"),\n\t\t\t\tIdentity:  p.Identity,\n\t\t\t}, p.CallOptions...)\n\t\t\tctxCancel()\n\t\t\treturn taskErr\n\t\t}\n\n\t\tif err2 != nil {\n\t\t\tctx, ctxCancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondActivityTaskFailed(ctx, &types.RespondActivityTaskFailedRequest{\n\t\t\t\tTaskToken: response.TaskToken,\n\t\t\t\tReason:    common.StringPtr(err2.Error()),\n\t\t\t\tDetails:   []byte(err2.Error()),\n\t\t\t\tIdentity:  p.Identity,\n\t\t\t}, p.CallOptions...)\n\t\t\tctxCancel()\n\t\t\treturn taskErr\n\t\t}\n\n\t\tctx, ctxCancel = createContext()\n\t\ttaskErr := p.Engine.RespondActivityTaskCompleted(ctx, &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tIdentity:  p.Identity,\n\t\t\tResult:    result,\n\t\t}, p.CallOptions...)\n\t\tctxCancel()\n\t\treturn taskErr\n\t}\n\n\treturn tasklist.ErrNoTasks\n}\n\n// PollAndProcessActivityTaskWithID is similar to PollAndProcessActivityTask but using RespondActivityTask...ByID\nfunc (p *TaskPoller) PollAndProcessActivityTaskWithID(dropTask bool) error {\n\tfor attempt := 0; attempt < 5; attempt++ {\n\t\tctx, ctxCancel := createContext()\n\t\tresponse, err1 := p.Engine.PollForActivityTask(ctx, &types.PollForActivityTaskRequest{\n\t\t\tDomain:   p.Domain,\n\t\t\tTaskList: p.TaskList,\n\t\t\tIdentity: p.Identity,\n\t\t}, p.CallOptions...)\n\t\tctxCancel()\n\n\t\tif err1 != nil {\n\t\t\treturn err1\n\t\t}\n\n\t\tif response == nil || len(response.TaskToken) == 0 {\n\t\t\tp.Logger.Info(\"Empty Activity task: Polling again.\")\n\t\t\treturn nil\n\t\t}\n\n\t\tif response.GetActivityID() == \"\" {\n\t\t\tp.Logger.Info(\"Empty ActivityID\")\n\t\t\treturn nil\n\t\t}\n\n\t\tif dropTask {\n\t\t\tp.Logger.Info(\"Dropping Activity task: \")\n\t\t\treturn nil\n\t\t}\n\t\tp.Logger.Debug(\"Received Activity task: %v\", tag.Value(response))\n\n\t\tresult, cancel, err2 := p.ActivityHandler(response.WorkflowExecution, response.ActivityType, response.ActivityID,\n\t\t\tresponse.Input, response.TaskToken)\n\t\tif cancel {\n\t\t\tp.Logger.Info(\"Executing RespondActivityTaskCanceled\")\n\t\t\tctx, ctxCancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondActivityTaskCanceledByID(ctx, &types.RespondActivityTaskCanceledByIDRequest{\n\t\t\t\tDomain:     p.Domain,\n\t\t\t\tWorkflowID: response.WorkflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:      response.WorkflowExecution.GetRunID(),\n\t\t\t\tActivityID: response.GetActivityID(),\n\t\t\t\tDetails:    []byte(\"details\"),\n\t\t\t\tIdentity:   p.Identity,\n\t\t\t}, p.CallOptions...)\n\t\t\tctxCancel()\n\t\t\treturn taskErr\n\t\t}\n\n\t\tif err2 != nil {\n\t\t\tctx, ctxCancel := createContext()\n\t\t\ttaskErr := p.Engine.RespondActivityTaskFailedByID(ctx, &types.RespondActivityTaskFailedByIDRequest{\n\t\t\t\tDomain:     p.Domain,\n\t\t\t\tWorkflowID: response.WorkflowExecution.GetWorkflowID(),\n\t\t\t\tRunID:      response.WorkflowExecution.GetRunID(),\n\t\t\t\tActivityID: response.GetActivityID(),\n\t\t\t\tReason:     common.StringPtr(err2.Error()),\n\t\t\t\tDetails:    []byte(err2.Error()),\n\t\t\t\tIdentity:   p.Identity,\n\t\t\t}, p.CallOptions...)\n\t\t\tctxCancel()\n\t\t\treturn taskErr\n\t\t}\n\n\t\tctx, ctxCancel = createContext()\n\t\ttaskErr := p.Engine.RespondActivityTaskCompletedByID(ctx, &types.RespondActivityTaskCompletedByIDRequest{\n\t\t\tDomain:     p.Domain,\n\t\t\tWorkflowID: response.WorkflowExecution.GetWorkflowID(),\n\t\t\tRunID:      response.WorkflowExecution.GetRunID(),\n\t\t\tActivityID: response.GetActivityID(),\n\t\t\tIdentity:   p.Identity,\n\t\t\tResult:     result,\n\t\t}, p.CallOptions...)\n\t\tctxCancel()\n\t\treturn taskErr\n\t}\n\n\treturn tasklist.ErrNoTasks\n}\n\nfunc (p *TaskPoller) PollAndProcessActivities() context.CancelFunc {\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tctx, cancel := context.WithCancel(context.Background())\n\tgo p.pollLoop(ctx, &wg, p.doPollActivityTask)\n\n\treturn func() {\n\t\tcancel()\n\t\twg.Wait()\n\t}\n}\n\nfunc (p *TaskPoller) PollAndProcessDecisions() context.CancelFunc {\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tctx, cancel := context.WithCancel(context.Background())\n\tgo p.pollLoop(ctx, &wg, p.doPollDecisionTask)\n\n\treturn func() {\n\t\tcancel()\n\t\twg.Wait()\n\t}\n}\n\nfunc (p *TaskPoller) pollLoop(ctx context.Context, wg *sync.WaitGroup, pollFunc func(context.Context) error) {\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\twg.Done()\n\t\t\treturn\n\t\tdefault:\n\t\t\terr := pollFunc(ctx)\n\t\t\tif !errors.Is(err, context.Canceled) && !errors.Is(err, context.DeadlineExceeded) {\n\t\t\t\tp.Logger.Error(\"Error while polling\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (p *TaskPoller) doPollActivityTask(ctx context.Context) error {\n\ttaskList := p.TaskList\n\tpollCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\tresponse, err := p.Engine.PollForActivityTask(pollCtx, &types.PollForActivityTaskRequest{\n\t\tDomain:   p.Domain,\n\t\tTaskList: taskList,\n\t\tIdentity: p.Identity,\n\t}, p.CallOptions...)\n\tcancel()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif response == nil || len(response.TaskToken) == 0 {\n\t\tp.Logger.Info(\"Empty Decision task: Polling again.\")\n\t\treturn nil\n\t}\n\n\tresult, shouldCancel, err := p.ActivityHandler(response.WorkflowExecution, response.ActivityType, response.ActivityID, response.Input, response.TaskToken)\n\n\tresponseCtx, ctxCancel := context.WithTimeout(ctx, time.Second*5)\n\tdefer ctxCancel()\n\tvar responseErr error\n\n\tif err != nil {\n\t\tp.Logger.Info(\"Executed RespondActivityTaskFailed\")\n\t\tresponseErr = p.Engine.RespondActivityTaskFailed(responseCtx, &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tReason:    common.StringPtr(err.Error()),\n\t\t\tDetails:   []byte(err.Error()),\n\t\t\tIdentity:  p.Identity,\n\t\t}, p.CallOptions...)\n\t} else if shouldCancel {\n\t\tp.Logger.Info(\"Executing RespondActivityTaskCanceled\")\n\t\tresponseErr = p.Engine.RespondActivityTaskCanceled(responseCtx, &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t\tIdentity:  p.Identity,\n\t\t}, p.CallOptions...)\n\t} else {\n\t\tp.Logger.Info(\"Executing RespondActivityTaskCompleted\")\n\t\tresponseErr = p.Engine.RespondActivityTaskCompleted(responseCtx, &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tIdentity:  p.Identity,\n\t\t\tResult:    result,\n\t\t}, p.CallOptions...)\n\t}\n\n\treturn responseErr\n}\n\nfunc (p *TaskPoller) doPollDecisionTask(ctx context.Context) error {\n\ttaskList := p.TaskList\n\tpollCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\tresponse, err := p.Engine.PollForDecisionTask(pollCtx, &types.PollForDecisionTaskRequest{\n\t\tDomain:   p.Domain,\n\t\tTaskList: taskList,\n\t\tIdentity: p.Identity,\n\t}, p.CallOptions...)\n\tcancel()\n\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif response == nil || len(response.TaskToken) == 0 {\n\t\tp.Logger.Info(\"Empty Decision task: Polling again.\")\n\t\treturn nil\n\t}\n\n\tif response.GetNextEventID() == 0 {\n\t\tp.Logger.Fatal(\"NextEventID is not set for decision or query task\")\n\t}\n\n\tvar events []*types.HistoryEvent\n\tif response.Query == nil {\n\t\t// if not query task, should have some history events\n\t\t// for non sticky query, there should be events returned\n\t\thistory := response.History\n\t\tif history == nil {\n\t\t\tp.Logger.Fatal(\"History is nil\")\n\t\t}\n\n\t\tevents = history.Events\n\t\tif len(events) == 0 {\n\t\t\tp.Logger.Fatal(\"History Events are empty\")\n\t\t}\n\n\t\tnextPageToken := response.NextPageToken\n\t\tfor nextPageToken != nil {\n\t\t\thistoryCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\t\t\tresp, err2 := p.Engine.GetWorkflowExecutionHistory(historyCtx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\t\tDomain:        p.Domain,\n\t\t\t\tExecution:     response.WorkflowExecution,\n\t\t\t\tNextPageToken: nextPageToken,\n\t\t\t}, p.CallOptions...)\n\t\t\tcancel()\n\n\t\t\tif err2 != nil {\n\t\t\t\treturn err2\n\t\t\t}\n\n\t\t\tevents = append(events, resp.History.Events...)\n\t\t\tnextPageToken = resp.NextPageToken\n\t\t}\n\t} else {\n\t\t// for sticky query, there should be NO events returned\n\t\t// since worker side already has the state machine and we do not intend to update that.\n\t\thistory := response.History\n\t\tnextPageToken := response.NextPageToken\n\t\tif !(history == nil || (len(history.Events) == 0 && nextPageToken == nil)) {\n\t\t\t// if history is not nil, and contains events or next token\n\t\t\tp.Logger.Fatal(\"History is not empty for sticky query\")\n\t\t}\n\t}\n\n\t// handle query task response\n\tif response.Query != nil {\n\t\tblob, err := p.QueryHandler(response)\n\n\t\tcompleteRequest := &types.RespondQueryTaskCompletedRequest{TaskToken: response.TaskToken}\n\t\tif err != nil {\n\t\t\tcompleteType := types.QueryTaskCompletedTypeFailed\n\t\t\tcompleteRequest.CompletedType = &completeType\n\t\t\tcompleteRequest.ErrorMessage = err.Error()\n\t\t} else {\n\t\t\tcompleteType := types.QueryTaskCompletedTypeCompleted\n\t\t\tcompleteRequest.CompletedType = &completeType\n\t\t\tcompleteRequest.QueryResult = blob\n\t\t}\n\n\t\trespondCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\t\ttaskErr := p.Engine.RespondQueryTaskCompleted(respondCtx, completeRequest, p.CallOptions...)\n\t\tcancel()\n\t\treturn taskErr\n\t}\n\texecutionCtx, decisions, err := p.DecisionHandler(response.WorkflowExecution, response.WorkflowType,\n\t\tcommon.Int64Default(response.PreviousStartedEventID), response.StartedEventID, response.History)\n\tif err != nil {\n\t\tp.Logger.Info(\"Failing Decision. Decision handler failed with error\", tag.Error(err))\n\t\trespondCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\t\ttaskErr := p.Engine.RespondDecisionTaskFailed(respondCtx, &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: response.TaskToken,\n\t\t\tCause:     types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\t\tDetails:   []byte(err.Error()),\n\t\t\tIdentity:  p.Identity,\n\t\t}, p.CallOptions...)\n\t\tcancel()\n\t\treturn taskErr\n\t}\n\n\tp.Logger.Info(\"Completing Decision.  Decisions\", tag.Value(decisions))\n\t// non sticky tasklist\n\trespondCtx, cancel := context.WithTimeout(ctx, time.Second*90)\n\t_, err = p.Engine.RespondDecisionTaskCompleted(respondCtx, &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken:                  response.TaskToken,\n\t\tIdentity:                   p.Identity,\n\t\tExecutionContext:           executionCtx,\n\t\tDecisions:                  decisions,\n\t\tReturnNewDecisionTask:      false,\n\t\tForceCreateNewDecisionTask: false,\n\t}, p.CallOptions...)\n\tcancel()\n\n\treturn err\n}\n\nfunc createContext() (context.Context, context.CancelFunc) {\n\tctx, cancel := context.WithTimeout(context.Background(), 90*time.Second)\n\treturn ctx, cancel\n}\n\nfunc getQueryResults(queries map[string]*types.WorkflowQuery, queryResult *types.WorkflowQueryResult) map[string]*types.WorkflowQueryResult {\n\tresult := make(map[string]*types.WorkflowQueryResult)\n\tfor k := range queries {\n\t\tresult[k] = queryResult\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "host/test_suites.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n)\n\n// NOTE: the following definitions can't be defined in *_test.go\n// since they need to be exported and used by our internal tests\n\ntype (\n\tIntegrationSuite struct {\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tIntegrationQueueV2Suite struct {\n\t\t*IntegrationSuite\n\t}\n\n\tSizeLimitIntegrationSuite struct {\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tClientIntegrationSuite struct {\n\t\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t\t// not merely log an error\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t\twfService workflowserviceclient.Interface\n\t\twfClient  client.Client\n\t\tworker    worker.Worker\n\t\ttaskList  string\n\t}\n\n\tAsyncWFIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tWorkflowIDRateLimitIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tWorkflowIDInternalRateLimitIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tTaskListIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\n\t\tTaskListName string\n\t}\n\n\tTaskListIsolationIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n\n\tDecisionTimeoutMaxAttemptsIntegrationSuite struct {\n\t\t*require.Assertions\n\t\t*IntegrationBase\n\t}\n)\n"
  },
  {
    "path": "host/testcluster.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage host\n\nimport (\n\t\"context\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/startreedata/pinot-client-go/pinot\"\n\t\"github.com/uber-go/tally\"\n\n\tadminClient \"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/filestore\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/messaging/kafka\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/nosql\"\n\tpersistencetests \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/persistence/persistence-tests/testcluster\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\tpnt \"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/testflags\"\n\n\t// the import is a test dependency\n\t_ \"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql/public\"\n)\n\ntype (\n\t// TestCluster is a base struct for integration tests\n\tTestCluster struct {\n\t\ttestBase     *persistencetests.TestBase\n\t\tarchiverBase *ArchiverBase\n\t\thost         Cadence\n\t\tdoneCh       chan struct{}\n\t}\n\n\t// ArchiverBase is a base struct for archiver provider being used in integration tests\n\tArchiverBase struct {\n\t\tmetadata                 archiver.ArchivalMetadata\n\t\tprovider                 provider.ArchiverProvider\n\t\thistoryStoreDirectory    string\n\t\tvisibilityStoreDirectory string\n\t\thistoryURI               string\n\t\tvisibilityURI            string\n\t}\n\n\t// TestClusterConfig are config for a test cluster\n\tTestClusterConfig struct {\n\t\tFrontendAddress       string\n\t\tEnableArchival        bool\n\t\tIsPrimaryCluster      bool\n\t\tClusterNo             int\n\t\tClusterGroupMetadata  config.ClusterGroupMetadata\n\t\tMessagingClientConfig *MessagingClientConfig\n\t\tPersistence           persistencetests.TestBaseOptions\n\t\tHistoryConfig         *HistoryConfig\n\t\tMatchingConfig        *MatchingConfig\n\t\tESConfig              *config.ElasticSearchConfig\n\t\tWorkerConfig          *WorkerConfig\n\t\tMockAdminClient       map[string]adminClient.Client\n\t\tPinotConfig           *config.PinotVisibilityConfig\n\t\tAsyncWFQueues         map[string]config.AsyncWorkflowQueueProvider\n\t\tDynamicClientConfig   dynamicconfig.FileBasedClientConfig\n\n\t\t// TimeSource is used to override the time source of internal components.\n\t\t// Note that most components don't respect this, and it's only used in a few places.\n\t\t// e.g. async workflow test's consumer manager and domain manager\n\t\tTimeSource                     clock.MockedTimeSource\n\t\tFrontendDynamicConfigOverrides map[dynamicproperties.Key]interface{}\n\t\tHistoryDynamicConfigOverrides  map[dynamicproperties.Key]interface{}\n\t\tMatchingDynamicConfigOverrides map[dynamicproperties.Key]interface{}\n\t\tWorkerDynamicConfigOverrides   map[dynamicproperties.Key]interface{}\n\t}\n\n\t// MessagingClientConfig is the config for messaging config\n\tMessagingClientConfig struct {\n\t\tUseMock     bool\n\t\tKafkaConfig *config.KafkaConfig\n\t}\n\n\t// WorkerConfig is the config for enabling/disabling cadence worker\n\tWorkerConfig struct {\n\t\tEnableArchiver        bool\n\t\tEnableIndexer         bool\n\t\tEnableReplicator      bool\n\t\tEnableAsyncWFConsumer bool\n\t}\n)\n\nconst (\n\tdefaultTestValueOfESIndexMaxResultWindow = 5\n\tdefaultTestPersistenceTimeout            = 5 * time.Second\n)\n\n// NewCluster creates and sets up the test cluster\nfunc NewCluster(t *testing.T, options *TestClusterConfig, logger log.Logger, params persistencetests.TestBaseParams) (*TestCluster, error) {\n\ttestBase := persistencetests.NewTestBaseFromParams(t, params)\n\ttestBase.Setup()\n\tsetupShards(testBase, options.HistoryConfig.NumHistoryShards, logger)\n\tarchiverBase := newArchiverBase(options.EnableArchival, logger)\n\tmessagingClient := getMessagingClient(options.MessagingClientConfig, logger)\n\tpConfig := testBase.Config()\n\tpConfig.NumHistoryShards = options.HistoryConfig.NumHistoryShards\n\tvar esClient elasticsearch.GenericClient\n\tif options.WorkerConfig.EnableIndexer {\n\t\tvar err error\n\t\tesClient, err = elasticsearch.NewGenericClient(options.ESConfig, logger)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tpConfig.AdvancedVisibilityStore = \"es-visibility\"\n\t}\n\n\tscope := tally.NewTestScope(\"integration-test\", nil)\n\tmetricsClient := metrics.NewClient(scope, metrics.ServiceIdx(0), metrics.HistogramMigration{} /* default, only used in test setups */)\n\tdomainReplicationQueue := domain.NewReplicationQueue(\n\t\ttestBase.DomainReplicationQueueMgr,\n\t\toptions.ClusterGroupMetadata.CurrentClusterName,\n\t\tmetricsClient,\n\t\tlogger,\n\t)\n\taConfig := noopAuthorizationConfig()\n\tdoneCh := make(chan struct{})\n\tdynamicClient, err := dynamicconfig.NewFileBasedClient(\n\t\t&options.DynamicClientConfig,\n\t\tlogger,\n\t\tdoneCh,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tcadenceParams := &CadenceParams{\n\t\tClusterMetadata:               params.ClusterMetadata,\n\t\tPersistenceConfig:             pConfig,\n\t\tMessagingClient:               messagingClient,\n\t\tDomainManager:                 testBase.DomainManager,\n\t\tHistoryV2Mgr:                  testBase.HistoryV2Mgr,\n\t\tExecutionMgrFactory:           testBase.ExecutionMgrFactory,\n\t\tDomainReplicationQueue:        domainReplicationQueue,\n\t\tLogger:                        logger,\n\t\tClusterNo:                     options.ClusterNo,\n\t\tESConfig:                      options.ESConfig,\n\t\tESClient:                      esClient,\n\t\tArchiverMetadata:              archiverBase.metadata,\n\t\tArchiverProvider:              archiverBase.provider,\n\t\tHistoryConfig:                 options.HistoryConfig,\n\t\tMatchingConfig:                options.MatchingConfig,\n\t\tWorkerConfig:                  options.WorkerConfig,\n\t\tMockAdminClient:               options.MockAdminClient,\n\t\tDomainReplicationTaskExecutor: domain.NewReplicationTaskExecutor(testBase.DomainManager, newNoopDomainAuditManager(), clock.NewRealTimeSource(), logger, dynamicproperties.GetBoolPropertyFn(false)),\n\t\tAuthorizationConfig:           aConfig,\n\t\tAsyncWFQueues:                 options.AsyncWFQueues,\n\t\tTimeSource:                    options.TimeSource,\n\t\tDynamicClient:                 dynamicClient,\n\t\tFrontendDynCfgOverrides:       options.FrontendDynamicConfigOverrides,\n\t\tHistoryDynCfgOverrides:        options.HistoryDynamicConfigOverrides,\n\t\tMatchingDynCfgOverrides:       options.MatchingDynamicConfigOverrides,\n\t\tWorkerDynCfgOverrides:         options.WorkerDynamicConfigOverrides,\n\t}\n\tcluster := NewCadence(cadenceParams)\n\tif err := cluster.Start(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &TestCluster{testBase: testBase, archiverBase: archiverBase, host: cluster, doneCh: doneCh}, nil\n}\n\nfunc NewPinotTestCluster(t *testing.T, options *TestClusterConfig, logger log.Logger, params persistencetests.TestBaseParams) (*TestCluster, error) {\n\ttestBase := persistencetests.NewTestBaseFromParams(t, params)\n\ttestBase.Setup()\n\tsetupShards(testBase, options.HistoryConfig.NumHistoryShards, logger)\n\tarchiverBase := newArchiverBase(options.EnableArchival, logger)\n\tmessagingClient := getMessagingClient(options.MessagingClientConfig, logger)\n\tpConfig := testBase.Config()\n\tpConfig.NumHistoryShards = options.HistoryConfig.NumHistoryShards\n\tvar esClient elasticsearch.GenericClient\n\tvar pinotClient pnt.GenericClient\n\tif options.WorkerConfig.EnableIndexer {\n\t\tvar err error\n\t\tif options.PinotConfig.Migration.Enabled {\n\t\t\tesClient, err = elasticsearch.NewGenericClient(options.ESConfig, logger)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t}\n\tpConfig.AdvancedVisibilityStore = \"pinot-visibility\"\n\tpinotBroker := options.PinotConfig.Broker\n\tpinotRawClient, err := pinot.NewFromBrokerList([]string{pinotBroker})\n\tif err != nil || pinotRawClient == nil {\n\t\treturn nil, err\n\t}\n\tpinotClient = pnt.NewPinotClient(pinotRawClient, logger, options.PinotConfig)\n\n\tscope := tally.NewTestScope(\"integration-test\", nil)\n\tmetricsClient := metrics.NewClient(scope, metrics.ServiceIdx(0), metrics.HistogramMigration{} /* default, only used in test setups */)\n\tdomainReplicationQueue := domain.NewReplicationQueue(\n\t\ttestBase.DomainReplicationQueueMgr,\n\t\toptions.ClusterGroupMetadata.CurrentClusterName,\n\t\tmetricsClient,\n\t\tlogger,\n\t)\n\taConfig := noopAuthorizationConfig()\n\tdoneCh := make(chan struct{})\n\tdynamicClient, err := dynamicconfig.NewFileBasedClient(\n\t\t&options.DynamicClientConfig,\n\t\tlogger,\n\t\tdoneCh,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tcadenceParams := &CadenceParams{\n\t\tClusterMetadata:               params.ClusterMetadata,\n\t\tPersistenceConfig:             pConfig,\n\t\tMessagingClient:               messagingClient,\n\t\tDomainManager:                 testBase.DomainManager,\n\t\tHistoryV2Mgr:                  testBase.HistoryV2Mgr,\n\t\tExecutionMgrFactory:           testBase.ExecutionMgrFactory,\n\t\tDomainReplicationQueue:        domainReplicationQueue,\n\t\tLogger:                        logger,\n\t\tClusterNo:                     options.ClusterNo,\n\t\tESConfig:                      options.ESConfig,\n\t\tESClient:                      esClient,\n\t\tArchiverMetadata:              archiverBase.metadata,\n\t\tArchiverProvider:              archiverBase.provider,\n\t\tHistoryConfig:                 options.HistoryConfig,\n\t\tMatchingConfig:                options.MatchingConfig,\n\t\tWorkerConfig:                  options.WorkerConfig,\n\t\tMockAdminClient:               options.MockAdminClient,\n\t\tDomainReplicationTaskExecutor: domain.NewReplicationTaskExecutor(testBase.DomainManager, newNoopDomainAuditManager(), clock.NewRealTimeSource(), logger, dynamicproperties.GetBoolPropertyFn(false)),\n\t\tAuthorizationConfig:           aConfig,\n\t\tPinotConfig:                   options.PinotConfig,\n\t\tPinotClient:                   pinotClient,\n\t\tDynamicClient:                 dynamicClient,\n\t}\n\tcluster := NewCadence(cadenceParams)\n\tif err := cluster.Start(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &TestCluster{testBase: testBase, archiverBase: archiverBase, host: cluster, doneCh: doneCh}, nil\n}\n\nfunc noopAuthorizationConfig() config.Authorization {\n\treturn config.Authorization{\n\t\tOAuthAuthorizer: config.OAuthAuthorizer{\n\t\t\tEnable: false,\n\t\t},\n\t\tNoopAuthorizer: config.NoopAuthorizer{\n\t\t\tEnable: true,\n\t\t},\n\t}\n}\n\n// NewClusterMetadata returns cluster metdata from config\nfunc NewClusterMetadata(t *testing.T, options *TestClusterConfig) cluster.Metadata {\n\tif options.ClusterGroupMetadata.PrimaryClusterName == \"\" {\n\t\treturn cluster.GetTestClusterMetadata(true)\n\t}\n\treturn cluster.NewMetadata(\n\t\toptions.ClusterGroupMetadata,\n\t\tfunc(domain string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\ttestlogger.New(t),\n\t)\n}\n\nfunc NewPersistenceTestCluster(t *testing.T, clusterConfig *TestClusterConfig) testcluster.PersistenceTestCluster {\n\t// NOTE: Override here to keep consistent. clusterConfig will be used in the test for some purposes.\n\tclusterConfig.Persistence.StoreType = TestFlags.PersistenceType\n\tclusterConfig.Persistence.DBPluginName = TestFlags.SQLPluginName\n\n\tvar testCluster testcluster.PersistenceTestCluster\n\tvar err error\n\tif TestFlags.PersistenceType == config.StoreTypeCassandra {\n\t\t// TODO refactor to support other NoSQL\n\t\tops := clusterConfig.Persistence\n\t\tops.DBPluginName = \"cassandra\"\n\t\ttestflags.RequireCassandra(t)\n\t\ttestCluster = nosql.NewTestCluster(t, nosql.TestClusterParams{\n\t\t\tPluginName:    ops.DBPluginName,\n\t\t\tKeySpace:      ops.DBName,\n\t\t\tUsername:      ops.DBUsername,\n\t\t\tPassword:      ops.DBPassword,\n\t\t\tHost:          ops.DBHost,\n\t\t\tPort:          ops.DBPort,\n\t\t\tProtoVersion:  ops.ProtoVersion,\n\t\t\tSchemaBaseDir: \"\",\n\t\t})\n\t} else if TestFlags.PersistenceType == config.StoreTypeSQL {\n\t\tvar ops *persistencetests.TestBaseOptions\n\t\tswitch TestFlags.SQLPluginName {\n\t\tcase mysql.PluginName:\n\t\t\ttestflags.RequireMySQL(t)\n\t\t\tops, err = mysql.GetTestClusterOption()\n\t\tcase postgres.PluginName:\n\t\t\ttestflags.RequirePostgres(t)\n\t\t\tops, err = postgres.GetTestClusterOption()\n\t\tcase sqlite.PluginName:\n\t\t\tops = sqlite.GetTestClusterOption()\n\t\tdefault:\n\t\t\tt.Fatal(\"not supported plugin \" + TestFlags.SQLPluginName)\n\t\t}\n\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\t\ttestCluster, err = sql.NewTestCluster(TestFlags.SQLPluginName, clusterConfig.Persistence.DBName, ops.DBUsername, ops.DBPassword, ops.DBHost, ops.DBPort, ops.SchemaDir)\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\t} else {\n\t\tt.Fatal(\"not supported storage type\" + TestFlags.PersistenceType)\n\t}\n\treturn testCluster\n}\n\nfunc newNoopDomainAuditManager() persistence.DomainAuditManager {\n\treturn &noopDomainAuditManager{}\n}\n\ntype noopDomainAuditManager struct{}\n\nfunc (n *noopDomainAuditManager) GetName() string {\n\treturn \"noop\"\n}\n\nfunc (n *noopDomainAuditManager) Close() {}\n\nfunc (n *noopDomainAuditManager) CreateDomainAuditLog(ctx context.Context, request *persistence.CreateDomainAuditLogRequest) (*persistence.CreateDomainAuditLogResponse, error) {\n\treturn &persistence.CreateDomainAuditLogResponse{}, nil\n}\n\nfunc (n *noopDomainAuditManager) GetDomainAuditLogs(ctx context.Context, request *persistence.GetDomainAuditLogsRequest) (*persistence.GetDomainAuditLogsResponse, error) {\n\treturn &persistence.GetDomainAuditLogsResponse{}, nil\n}\n\nfunc setupShards(testBase *persistencetests.TestBase, numHistoryShards int, logger log.Logger) {\n\t// shard 0 is always created, we create additional shards if needed\n\tfor shardID := 1; shardID < numHistoryShards; shardID++ {\n\t\tctx, cancel := context.WithTimeout(context.Background(), defaultTestPersistenceTimeout)\n\t\terr := testBase.CreateShard(ctx, shardID, \"\", 0)\n\t\tif err != nil {\n\t\t\tcancel()\n\t\t\tlogger.Fatal(\"Failed to create shard\", tag.Error(err))\n\t\t}\n\t\tcancel()\n\t}\n}\n\nfunc newArchiverBase(enabled bool, logger log.Logger) *ArchiverBase {\n\tdcCollection := dynamicconfig.NewNopCollection()\n\tif !enabled {\n\t\treturn &ArchiverBase{\n\t\t\tmetadata: archiver.NewArchivalMetadata(dcCollection, \"\", false, \"\", false, &config.ArchivalDomainDefaults{}),\n\t\t\tprovider: provider.NewNoOpArchiverProvider(),\n\t\t}\n\t}\n\n\thistoryStoreDirectory, err := ioutil.TempDir(\"\", \"test-history-archival\")\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to create temp dir for history archival\", tag.Error(err))\n\t}\n\tvisibilityStoreDirectory, err := ioutil.TempDir(\"\", \"test-visibility-archival\")\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to create temp dir for visibility archival\", tag.Error(err))\n\t}\n\tcfg := &config.FilestoreArchiver{\n\t\tFileMode: \"0666\",\n\t\tDirMode:  \"0766\",\n\t}\n\tnode, err := config.ToYamlNode(cfg)\n\tif err != nil {\n\t\tlogger.Fatal(\"Should be impossible: failed to convert filestore archiver config to a yaml node\")\n\t}\n\n\tarchiverProvider := provider.NewArchiverProvider(\n\t\tconfig.HistoryArchiverProvider{config.FilestoreConfig: node},\n\t\tconfig.VisibilityArchiverProvider{config.FilestoreConfig: node},\n\t)\n\treturn &ArchiverBase{\n\t\tmetadata: archiver.NewArchivalMetadata(dcCollection, \"enabled\", true, \"enabled\", true, &config.ArchivalDomainDefaults{\n\t\t\tHistory: config.HistoryArchivalDomainDefaults{\n\t\t\t\tStatus: \"enabled\",\n\t\t\t\tURI:    \"testScheme://test/history/archive/path\",\n\t\t\t},\n\t\t\tVisibility: config.VisibilityArchivalDomainDefaults{\n\t\t\t\tStatus: \"enabled\",\n\t\t\t\tURI:    \"testScheme://test/visibility/archive/path\",\n\t\t\t},\n\t\t}),\n\t\tprovider:                 archiverProvider,\n\t\thistoryStoreDirectory:    historyStoreDirectory,\n\t\tvisibilityStoreDirectory: visibilityStoreDirectory,\n\t\thistoryURI:               filestore.URIScheme + \"://\" + historyStoreDirectory,\n\t\tvisibilityURI:            filestore.URIScheme + \"://\" + visibilityStoreDirectory,\n\t}\n}\n\nfunc getMessagingClient(config *MessagingClientConfig, logger log.Logger) messaging.Client {\n\tif config == nil || config.UseMock {\n\t\treturn mocks.NewMockMessagingClient(&mocks.KafkaProducer{}, nil)\n\t}\n\tcheckApp := len(config.KafkaConfig.Applications) != 0\n\treturn kafka.NewKafkaClient(config.KafkaConfig, metrics.NewNoopMetricsClient(), logger, tally.NoopScope, checkApp)\n}\n\n// TearDownCluster tears down the test cluster\nfunc (tc *TestCluster) TearDownCluster() {\n\ttc.host.Stop()\n\ttc.host = nil\n\tclose(tc.doneCh)\n\ttc.testBase.TearDownWorkflowStore()\n\tos.RemoveAll(tc.archiverBase.historyStoreDirectory)\n\tos.RemoveAll(tc.archiverBase.visibilityStoreDirectory)\n}\n\n// GetFrontendClient returns a frontend client from the test cluster\nfunc (tc *TestCluster) GetFrontendClient() FrontendClient {\n\treturn tc.host.GetFrontendClient()\n}\n\n// GetAdminClient returns an admin client from the test cluster\nfunc (tc *TestCluster) GetAdminClient() AdminClient {\n\treturn tc.host.GetAdminClient()\n}\n\n// GetHistoryClient returns a history client from the test cluster\nfunc (tc *TestCluster) GetHistoryClient() HistoryClient {\n\treturn tc.host.GetHistoryClient()\n}\n\n// GetMatchingClient returns a matching client from the test cluster\nfunc (tc *TestCluster) GetMatchingClient() MatchingClient {\n\treturn tc.host.GetMatchingClient()\n}\n\nfunc (tc *TestCluster) GetMatchingClients() []MatchingClient {\n\tclients := tc.host.GetMatchingClients()\n\tresult := make([]MatchingClient, 0, len(clients))\n\tfor _, client := range clients {\n\t\tresult = append(result, client)\n\t}\n\treturn result\n}\n\n// GetExecutionManagerFactory returns an execution manager factory from the test cluster\nfunc (tc *TestCluster) GetExecutionManagerFactory() persistence.ExecutionManagerFactory {\n\treturn tc.host.GetExecutionManagerFactory()\n}\n"
  },
  {
    "path": "host/testdata/clientintegrationtestcluster.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/dynamicconfig/integration_queuev2_test.yaml",
    "content": "history.enableTimerQueueV2:\n- value: true\n  constraints: {}\nhistory.enableTransferQueueV2:\n- value: true\n  constraints: {}\nfrontend.warmupDuration:\n- value: \"1s\"\n  constraints: {}\n"
  },
  {
    "path": "host/testdata/dynamicconfig/integration_queuev2_with_alert_test.yaml",
    "content": "frontend.warmupDuration:\n- value: \"1s\"\n  constraints: {}\nhistory.enableTimerQueueV2:\n- value: true\n  constraints: {}\nhistory.enableTransferQueueV2:\n- value: true\n  constraints: {}\nhistory.shardUpdateMinInterval:\n- value: 3s\n  constraints: {}\nhistory.timerProcessorUpdateAckInterval:\n- value: 5s\n  constraints: {}\nhistory.timerProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.transferProcessorUpdateAckInterval:\n- value: 5s\n  constraints: {}\nhistory.transferProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.queueProcessorPollBackoffInterval:\n- value: 5s\n  constraints: {}\nhistory.virtualSliceForceAppendInterval:\n- value: 100ms\n  constraints: {}\n# Only Enable 1 level of split, which has been verified in simulation\nhistory.queueMaxVirtualQueueCount:\n- value: 2\n  constraints: {}\n# Enable task rate limiter so that the number of pending tasks increases\nhistory.taskSchedulerEnableRateLimiter:\n- value: true\n  constraints: {}\nhistory.taskSchedulerEnableRateLimiterShadowMode:\n- value: false\n  constraints: {}\nhistory.taskSchedulerGlobalDomainRPS:\n- value: 30\n  constraints: {}\n# Enable pending task queue alert for integration test\nhistory.enableTransferQueueV2PendingTaskCountAlert:\n- value: true\n  constraints: {}\nhistory.enableTimerQueueV2PendingTaskCountAlert:\n- value: true\n  constraints: {}\n# Set a low number to trigger queue pause with load from tests\nhistory.queueMaxPendingTaskCount:\n- value: 20\n  constraints: {}\n# Set a low number to trigger queue split with load from tests\nhistory.queueCriticalPendingTaskCount:\n- value: 18\n  constraints: {}\n"
  },
  {
    "path": "host/testdata/dynamicconfig/integration_test.yaml",
    "content": "frontend.warmupDuration:\n- value: \"1s\"\n  constraints: {}\nfrontend.enableActiveClusterSelectionPolicyInStartWorkflow:\n- value: true\n  constraints: {}\n"
  },
  {
    "path": "host/testdata/es_os2_index_template.json",
    "content": "{\n  \"index_patterns\": [\n    \"test-visibility*\"\n  ],\n  \"template\": {\n    \"aliases\": {},\n    \"mappings\": {\n      \"dynamic\": \"false\",\n      \"properties\": {\n        \"Attr\": {\n          \"properties\": {\n            \"BinaryChecksums\": {\n              \"type\": \"keyword\"\n            },\n            \"CadenceChangeVersion\": {\n              \"type\": \"keyword\"\n            },\n            \"CustomBoolField\": {\n              \"type\": \"boolean\"\n            },\n            \"CustomDatetimeField\": {\n              \"type\": \"date\"\n            },\n            \"CustomDomain\": {\n              \"type\": \"keyword\"\n            },\n            \"CustomDoubleField\": {\n              \"type\": \"double\"\n            },\n            \"CustomIntField\": {\n              \"type\": \"long\"\n            },\n            \"CustomKeywordField\": {\n              \"type\": \"keyword\"\n            },\n            \"CustomStringField\": {\n              \"type\": \"text\"\n            },\n            \"Operator\": {\n              \"type\": \"keyword\"\n            },\n            \"Passed\": {\n              \"type\": \"boolean\"\n            },\n            \"RolloutID\": {\n              \"type\": \"keyword\"\n            },\n            \"addon\": {\n              \"type\": \"keyword\"\n            },\n            \"addon-type\": {\n              \"type\": \"keyword\"\n            },\n            \"environment\": {\n              \"type\": \"keyword\"\n            },\n            \"project\": {\n              \"type\": \"keyword\"\n            },\n            \"service\": {\n              \"type\": \"keyword\"\n            },\n            \"user\": {\n              \"type\": \"keyword\"\n            }\n          }\n        },\n        \"CloseStatus\": {\n          \"type\": \"integer\"\n        },\n        \"CloseTime\": {\n          \"type\": \"long\"\n        },\n        \"DomainID\": {\n          \"type\": \"keyword\"\n        },\n        \"ExecutionTime\": {\n          \"type\": \"long\"\n        },\n        \"HistoryLength\": {\n          \"type\": \"integer\"\n        },\n        \"IsCron\": {\n          \"type\": \"boolean\"\n        },\n        \"KafkaKey\": {\n          \"type\": \"keyword\"\n        },\n        \"NumClusters\": {\n          \"type\": \"long\"\n        },\n        \"ClusterAttributeScope\": {\n          \"type\": \"keyword\"\n        },\n        \"ClusterAttributeName\": {\n          \"type\": \"keyword\"\n        },\n        \"RunID\": {\n          \"type\": \"keyword\"\n        },\n        \"ShardID\": {\n          \"type\": \"long\"\n        },\n        \"StartTime\": {\n          \"type\": \"long\"\n        },\n        \"WorkflowID\": {\n          \"type\": \"keyword\"\n        },\n        \"WorkflowType\": {\n          \"type\": \"keyword\"\n        }\n      }\n    },\n    \"settings\": {\n      \"index\": {\n        \"number_of_replicas\": \"0\",\n        \"number_of_shards\": \"5\"\n      }\n    }\n  }\n}\n"
  },
  {
    "path": "host/testdata/es_v6_index_template.json",
    "content": "{\n  \"order\": 0,\n  \"index_patterns\": [\n    \"test-visibility*\"\n  ],\n  \"settings\": {\n    \"index\": {\n      \"number_of_shards\": \"5\",\n      \"number_of_replicas\": \"0\"\n    }\n  },\n  \"mappings\": {\n    \"_doc\": {\n      \"dynamic\": \"false\",\n      \"properties\": {\n        \"DomainID\": {\n          \"type\": \"keyword\"\n        },\n        \"WorkflowID\": {\n          \"type\": \"keyword\"\n        },\n        \"RunID\": {\n          \"type\": \"keyword\"\n        },\n        \"WorkflowType\": {\n          \"type\": \"keyword\"\n        },\n        \"StartTime\": {\n          \"type\": \"long\"\n        },\n        \"ExecutionTime\": {\n          \"type\": \"long\"\n        },\n        \"CloseTime\": {\n          \"type\": \"long\"\n        },\n        \"CloseStatus\": {\n          \"type\": \"integer\"\n        },\n        \"HistoryLength\": {\n          \"type\": \"integer\"\n        },\n        \"IsCron\": {\n          \"type\": \"boolean\"\n        },\n        \"NumClusters\": {\n          \"type\": \"long\"\n        },\n        \"ClusterAttributeScope\": {\n          \"type\": \"keyword\"\n        },\n        \"ClusterAttributeName\": {\n          \"type\": \"keyword\"\n        },\n        \"KafkaKey\": {\n          \"type\": \"keyword\"\n        },\n        \"Attr\": {\n          \"properties\": {\n            \"CadenceChangeVersion\":  { \"type\": \"keyword\" },\n            \"CustomStringField\":  { \"type\": \"text\" },\n            \"CustomKeywordField\": { \"type\": \"keyword\"},\n            \"CustomIntField\": { \"type\": \"long\"},\n            \"CustomBoolField\": { \"type\": \"boolean\"},\n            \"CustomDoubleField\": { \"type\": \"double\"},\n            \"CustomDatetimeField\": { \"type\": \"date\"},\n            \"project\": { \"type\": \"keyword\"},\n            \"service\": { \"type\": \"keyword\"},\n            \"environment\": { \"type\": \"keyword\"},\n            \"addon\": { \"type\": \"keyword\"},\n            \"addon-type\": { \"type\": \"keyword\"},\n            \"user\": { \"type\": \"keyword\"},\n            \"CustomDomain\": { \"type\": \"keyword\"},\n            \"Operator\": { \"type\": \"keyword\"},\n            \"RolloutID\": { \"type\": \"keyword\"},\n            \"BinaryChecksums\": { \"type\": \"keyword\"},\n            \"Passed\": { \"type\": \"boolean\" }\n          }\n        }\n      }\n    }\n  },\n  \"aliases\": {}\n}\n"
  },
  {
    "path": "host/testdata/es_v7_index_template.json",
    "content": "{\n  \"order\": 0,\n  \"index_patterns\": [\n    \"test-visibility*\"\n  ],\n  \"settings\": {\n    \"index\": {\n      \"number_of_shards\": \"5\",\n      \"number_of_replicas\": \"0\"\n    }\n  },\n  \"mappings\": {\n    \"dynamic\": \"false\",\n    \"properties\": {\n      \"DomainID\": {\n        \"type\": \"keyword\"\n      },\n      \"WorkflowID\": {\n        \"type\": \"keyword\"\n      },\n      \"RunID\": {\n        \"type\": \"keyword\"\n      },\n      \"WorkflowType\": {\n        \"type\": \"keyword\"\n      },\n      \"StartTime\": {\n        \"type\": \"long\"\n      },\n      \"ExecutionTime\": {\n        \"type\": \"long\"\n      },\n      \"CloseTime\": {\n        \"type\": \"long\"\n      },\n      \"CloseStatus\": {\n        \"type\": \"integer\"\n      },\n      \"HistoryLength\": {\n        \"type\": \"integer\"\n      },\n      \"IsCron\": {\n        \"type\": \"boolean\"\n      },\n      \"NumClusters\": {\n        \"type\": \"long\"\n      },\n      \"ClusterAttributeScope\": {\n        \"type\": \"keyword\"\n      },\n      \"ClusterAttributeName\": {\n        \"type\": \"keyword\"\n      },\n      \"KafkaKey\": {\n        \"type\": \"keyword\"\n      },\n      \"Attr\": {\n        \"properties\": {\n          \"CadenceChangeVersion\":  { \"type\": \"keyword\" },\n          \"CustomStringField\":  { \"type\": \"text\" },\n          \"CustomKeywordField\": { \"type\": \"keyword\"},\n          \"CustomIntField\": { \"type\": \"long\"},\n          \"CustomBoolField\": { \"type\": \"boolean\"},\n          \"CustomDoubleField\": { \"type\": \"double\"},\n          \"CustomDatetimeField\": { \"type\": \"date\"},\n          \"project\": { \"type\": \"keyword\"},\n          \"service\": { \"type\": \"keyword\"},\n          \"environment\": { \"type\": \"keyword\"},\n          \"addon\": { \"type\": \"keyword\"},\n          \"addon-type\": { \"type\": \"keyword\"},\n          \"user\": { \"type\": \"keyword\"},\n          \"CustomDomain\": { \"type\": \"keyword\"},\n          \"Operator\": { \"type\": \"keyword\"},\n          \"RolloutID\": { \"type\": \"keyword\"},\n          \"BinaryChecksums\": { \"type\": \"keyword\"},\n          \"Passed\": { \"type\": \"boolean\" }\n        }\n      }\n    }\n  },\n  \"aliases\": {}\n}\n"
  },
  {
    "path": "host/testdata/integration_async_wf_with_kafka_cluster.yaml",
    "content": "clusterno: 1\nasyncwfqueues:\n  # test-async-wf-queue is the name of the queue.\n  # it is used in async_wf_test.go\n  # topic must be created before running the tests. it's done automatically by wurstmeuster/kafka container\n  # in docker/docker-compose-async-wf-kafka.yml\n  test-async-wf-queue:\n    type: \"kafka\"\n    config:\n      connection:\n        brokers:\n          - \"${KAFKA_SEEDS}:${KAFKA_PORT}\"\n      topic: \"${ASYNC_WF_KAFKA_QUEUE_TOPIC}\"\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enableasyncwfconsumer: true\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_decision_timeout_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_elasticsearch_os2_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: false\n  kafkaconfig:\n    clusters:\n      test:\n        brokers:\n          - \"${KAFKA_SEEDS}:9092\"\n    topics:\n      test-visibility-topic:\n        cluster: test\n      test-visibility-topic-dlq:\n        cluster: test\n    applications:\n      visibility:\n        topic: test-visibility-topic\n        dlq-topic: test-visibility-topic-dlq\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: true\nesconfig:\n  version: \"os2\"\n  url:\n    scheme: \"http\"\n    host: \"${ES_SEEDS}:9200\"\n  indices:\n    visibility: test-visibility-\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_elasticsearch_v6_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: false\n  kafkaconfig:\n    clusters:\n      test:\n        brokers:\n          - \"${KAFKA_SEEDS}:9092\"\n    topics:\n      test-visibility-topic:\n        cluster: test\n      test-visibility-topic-dlq:\n        cluster: test\n    applications:\n      visibility:\n        topic: test-visibility-topic\n        dlq-topic: test-visibility-topic-dlq\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: true\nesconfig:\n  url:\n    scheme: \"http\"\n    host: \"${ES_SEEDS}:9200\"\n  indices:\n    visibility: test-visibility-\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_elasticsearch_v7_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: false\n  kafkaconfig:\n    clusters:\n      test:\n        brokers:\n          - \"${KAFKA_SEEDS}:9092\"\n    topics:\n      test-visibility-topic:\n        cluster: test\n      test-visibility-topic-dlq:\n        cluster: test\n    applications:\n      visibility:\n        topic: test-visibility-topic\n        dlq-topic: test-visibility-topic-dlq\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: true\nesconfig:\n  version: \"v7\"\n  url:\n    scheme: \"http\"\n    host: \"${ES_SEEDS}:9200\"\n  indices:\n    visibility: test-visibility-\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_pinot_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: false\n  kafkaconfig:\n    clusters:\n      test:\n        brokers:\n          - \"${KAFKA_SEEDS}:9092\"\n    topics:\n      test-visibility-topic:\n        cluster: test\n      test-visibility-topic-dlq:\n        cluster: test\n      cadence-visibility-pinot:\n        cluster: test\n      cadence-visibility-pinot-dlq:\n        cluster: test\n    applications:\n      pinot-visibility:\n        topic: cadence-visibility-pinot\n        dlq-topic: cadence-visibility-pinot-dlq\n      visibility:\n        topic: test-visibility-topic\n        dlq-topic: test-visibility-topic-dlq\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\nesconfig:\n  version: \"v7\"\n  url:\n    scheme: \"http\"\n    host: \"${ES_SEEDS}:9200\"\n  indices:\n    visibility: test-visibility-\npinotconfig:\n  broker: \"${PINOT_SEEDS}:8099\"\n  cluster: pinot-test\n  table: \"cadence_visibility_pinot\"\n  migration:\n    enabled: true\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_queuev2_cluster.yaml",
    "content": "enablearchival: true\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: true\n  enablereplicator: true\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_queuev2_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_queuev2_with_alert_cluster.yaml",
    "content": "enablearchival: true\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: true\n  enablereplicator: true\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_queuev2_with_alert_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_sizelimit_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\n  historycountlimiterror: 20\n  historycountlimitwarn: 10\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_test_cluster.yaml",
    "content": "enablearchival: true\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: true\n  enablereplicator: true\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/integration_wfidratelimit_cluster.yaml",
    "content": "enablearchival: true\nclusterno: 2\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/ndc_integration_test_clusters.yaml",
    "content": "- persistence:\n    dbname: integration_active\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active\"\n    currentClusterName: \"active\"\n    clusterGroup:\n      active:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:7114\"\n        rpcTransport: \"grpc\"\n      standby:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:8114\"\n        rpcTransport: \"grpc\"\n      other:\n        enabled: true\n        initialFailoverVersion: 2\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:9114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: false\n  clusterno: 0\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      topics:\n        active:\n          cluster: test\n        active-dlq:\n          cluster: test\n        standby:\n          cluster: test\n        standby-dlq:\n          cluster: test\n        other:\n          cluster: test\n        other-dlq:\n          cluster: test\n      cadence-cluster-topics:\n        active:\n          topic: active\n          dlq-topic: active-dlq\n        standby:\n          topic: standby\n          dlq-topic: standby-dlq\n        other:\n          topic: other\n          dlq-topic: other-dlq\n      applications: { }\n  dynamicclientconfig:\n    filepath: \"../testdata/dynamicconfig/integration_test.yaml\"\n    pollInterval: \"10s\"\n- persistence:\n    dbname: integration_standby\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active\"\n    currentClusterName: \"standby\"\n    clusterGroup:\n      active:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:7114\"\n        rpcTransport: \"grpc\"\n      standby:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:8114\"\n        rpcTransport: \"grpc\"\n      other:\n        enabled: true\n        initialFailoverVersion: 2\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:9114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: false\n  clusterno: 1\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      topics:\n        active:\n          cluster: test\n        active-dlq:\n          cluster: test\n        standby:\n          cluster: test\n        standby-dlq:\n          cluster: test\n        other:\n          cluster: test\n        other-dlq:\n          cluster: test\n      cadence-cluster-topics:\n        active:\n          topic: active\n          dlq-topic: active-dlq\n        standby:\n          topic: standby\n          dlq-topic: standby-dlq\n        other:\n          topic: other\n          dlq-topic: other-dlq\n      applications: { }\n- persistence:\n    dbname: integration_other\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active\"\n    currentClusterName: \"other\"\n    clusterGroup:\n      active:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:7114\"\n        rpcTransport: \"grpc\"\n      standby:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:8114\"\n        rpcTransport: \"grpc\"\n      other:\n        enabled: true\n        initialFailoverVersion: 2\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:9114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: false\n  clusterno: 2\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n  dynamicclientconfig:\n    filepath: \"../testdata/dynamicconfig/integration_test.yaml\"\n    pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/task_list_test_cluster.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"testdata/dynamicconfig/integration_test.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "host/testdata/xdc_integration_es_clusters.yaml",
    "content": "- persistence:\n    dbname: integration_active_es\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active-es\"\n    currentClusterName: \"active-es\"\n    clusterGroup:\n      active-es:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:9114\"\n        rpcTransport: \"grpc\"\n      standby-es:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:10114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: true\n  clusterno: 2\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      topics:\n        active-es:\n          cluster: test\n        active-es-dlq:\n          cluster: test\n        standby-es:\n          cluster: test\n        standby-es-dlq:\n          cluster: test\n        test-visibility-topic-0:\n          cluster: test\n        test-visibility-topic-0-dlq:\n          cluster: test\n      cadence-cluster-topics:\n        active-es:\n          topic: active-es\n          dlq-topic: active-es-dlq\n        standby-es:\n          topic: standby-es\n          dlq-topic: standby-es-dlq\n      applications:\n        visibility:\n          topic: test-visibility-topic-0\n          dlq-topic: test-visibility-topic-0-dlq\n  esconfig:\n    url:\n      scheme: \"http\"\n      host: \"${ES_SEEDS}:9200\"\n    indices:\n      visibility: test-visibility-0-\n\n- persistence:\n    dbname: integration_standby_es\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active-es\"\n    currentClusterName: \"standby-es\"\n    clusterGroup:\n      active-es:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:9114\"\n        rpcTransport: \"grpc\"\n      standby-es:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:10114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: true\n  clusterno: 3\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      topics:\n        test-visibility-topic-1:\n          cluster: test\n        test-visibility-topic-1-dlq:\n          cluster: test\n      applications:\n        visibility:\n          topic: test-visibility-topic-1\n          dlq-topic: test-visibility-topic-1-dlq\n  esconfig:\n    url:\n      scheme: \"http\"\n      host: \"${ES_SEEDS}:9200\"\n    indices:\n      visibility: test-visibility-1-\n"
  },
  {
    "path": "host/testdata/xdc_integration_test_clusters.yaml",
    "content": "- persistence:\n    dbname: integration_active\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active\"\n    currentClusterName: \"active\"\n    clusterGroup:\n      active:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:7114\"\n        rpcTransport: \"grpc\"\n      standby:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:8114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: false\n  clusterno: 0\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      topics:\n        active:\n          cluster: test\n        active-dlq:\n          cluster: test\n        standby:\n          cluster: test\n        standby-dlq:\n          cluster: test\n      cadence-cluster-topics:\n        active:\n          topic: active\n          dlq-topic: active-dlq\n        standby:\n          topic: standby\n          dlq-topic: standby-dlq\n      applications: { }\n- persistence:\n    dbname: integration_standby\n  clustergroupmetadata:\n    failoverVersionIncrement: 10\n    primaryClusterName: \"active\"\n    currentClusterName: \"standby\"\n    clusterGroup:\n      active:\n        enabled: true\n        initialFailoverVersion: 0\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:7114\"\n        rpcTransport: \"grpc\"\n      standby:\n        enabled: true\n        initialFailoverVersion: 1\n        rpcName: \"cadence-frontend\"\n        rpcAddress: \"127.0.0.1:8114\"\n        rpcTransport: \"grpc\"\n  enablearchival: false\n  workerconfig:\n    enablearchiver: false\n    enablereplicator: true\n    enableindexer: false\n  clusterno: 1\n  historyconfig:\n    numhistoryshards: 1\n    numhistoryhosts: 1\n  matchingconfig:\n    nummatchinghosts: 1\n  messagingclientconfig:\n    usemock: false\n    kafkaconfig:\n      clusters:\n        test:\n          brokers:\n            - \"${KAFKA_SEEDS}:9092\"\n      applications: { }\n"
  },
  {
    "path": "host/workflowidratelimit_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage host\n\nimport (\n\t\"flag\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestWorkflowIDRateLimitIntegrationSuite(t *testing.T) {\n\t// Loads the flags for persistence etc., if none are given they are set in ./flag.go\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_wfidratelimit_cluster.yaml\")\n\trequire.NoError(t, err)\n\n\tclusterConfig.TimeSource = clock.NewMockedTimeSource()\n\tclusterConfig.HistoryDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.WorkflowIDExternalRPS: 5,\n\t}\n\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(WorkflowIDRateLimitIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *WorkflowIDRateLimitIntegrationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ts.domainCacheRefresh()\n}\n\nfunc (s *WorkflowIDRateLimitIntegrationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *WorkflowIDRateLimitIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *WorkflowIDRateLimitIntegrationSuite) TestWorkflowIDSpecificRateLimits() {\n\tconst (\n\t\ttestWorkflowID   = \"integration-workflow-specific-rate-limit-test\"\n\t\ttestWorkflowType = \"integration-workflow-specific-rate-limit-test-type\"\n\t\ttestTaskListName = \"integration-workflow-specific-rate-limit-test-taskList\"\n\t\ttestIdentity     = \"worker1\"\n\t)\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          testWorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: testWorkflowType},\n\t\tTaskList:                            &types.TaskList{Name: testTaskListName},\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            testIdentity,\n\n\t\tWorkflowIDReusePolicy: types.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\n\t// The ratelimit is 5 per second with a burst of 5, so we should be able to start 5 workflows without any error\n\tfor i := 0; i < 5; i++ {\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, request)\n\t\tassert.NoError(s.T(), err)\n\t}\n\n\t// Now we should get a rate limit error (with some fuzziness for time passing)\n\tlimited := 0\n\tfor i := 0; i < 5; i++ {\n\t\t_, err := s.Engine.StartWorkflowExecution(ctx, request)\n\t\tvar busyErr *types.ServiceBusyError\n\t\tif err != nil {\n\t\t\tif assert.ErrorAs(s.T(), err, &busyErr) {\n\t\t\t\tlimited++\n\t\t\t\tassert.Equal(s.T(), constants.WorkflowIDRateLimitReason, busyErr.Reason)\n\t\t\t}\n\t\t}\n\t}\n\t// 5 fails occasionally, trying 4.  If needed, reduce to 3 or find a way to\n\t// make this test less sensitive to latency, as test-runner hosts vary a lot.\n\tassert.GreaterOrEqual(s.T(), limited, 4, \"should have encountered some rate-limit errors after the burst was exhausted\")\n\n\t// After 1 second (200ms at a minimum) we should be able to start more workflows without being limited\n\ttime.Sleep(1 * time.Second)\n\t_, err := s.Engine.StartWorkflowExecution(ctx, request)\n\tassert.NoError(s.T(), err)\n}\n"
  },
  {
    "path": "host/workflowsidinternalratelimit_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage host\n\nimport (\n\t\"bytes\"\n\t\"encoding/binary\"\n\t\"flag\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n)\n\nfunc TestWorkflowIDInternalRateLimitIntegrationSuite(t *testing.T) {\n\t// Loads the flags for persistence etc., if none are given they are set in ./flag.go\n\tflag.Parse()\n\n\tclusterConfig, err := GetTestClusterConfig(\"testdata/integration_wfidratelimit_cluster.yaml\")\n\trequire.NoError(t, err)\n\n\tclusterConfig.TimeSource = clock.NewMockedTimeSource()\n\tclusterConfig.HistoryDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.WorkflowIDInternalRPS: 2,\n\t}\n\n\ttestCluster := NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(WorkflowIDInternalRateLimitIntegrationSuite)\n\tparams := IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *WorkflowIDInternalRateLimitIntegrationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t\tHistoryNodeDeleteBatchSize:               dynamicproperties.GetIntPropertyFn(1000),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ts.domainCacheRefresh()\n}\n\nfunc (s *WorkflowIDInternalRateLimitIntegrationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *WorkflowIDInternalRateLimitIntegrationSuite) TearDownSuite() {\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *WorkflowIDInternalRateLimitIntegrationSuite) TestWorkflowIDSpecificInternalRateLimits() {\n\tconst (\n\t\ttestWorkflowID   = \"integration-workflow-specific-internal-rate-limit-test\"\n\t\ttestWorkflowType = \"integration-workflow-specific-internal-rate-limit-test-type\"\n\t\ttestTaskListName = \"integration-workflow-specific-internal-rate-limit-test-taskList\"\n\t\ttestIdentity     = \"worker1\"\n\t)\n\n\tactivityName := \"test-activity\"\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              s.DomainName,\n\t\tWorkflowID:                          testWorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: testWorkflowType},\n\t\tTaskList:                            &types.TaskList{Name: testTaskListName},\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            testIdentity,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t}\n\n\tctx, cancel := createContext()\n\tdefer cancel()\n\n\twe, err := s.Engine.StartWorkflowExecution(ctx, request)\n\ts.NoError(err)\n\n\ts.Logger.Info(\"StartWorkflowExecution\", tag.WorkflowRunID(we.RunID))\n\n\tworkflowComplete := false\n\tactivityCount := int32(5)\n\tactivityCounter := int32(0)\n\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\t\tif activityCounter < activityCount {\n\t\t\tactivityCounter++\n\t\t\tbuf := new(bytes.Buffer)\n\t\t\ts.Nil(binary.Write(buf, binary.LittleEndian, activityCounter))\n\n\t\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\tActivityID:                    strconv.Itoa(int(activityCounter)),\n\t\t\t\t\tActivityType:                  &types.ActivityType{Name: activityName},\n\t\t\t\t\tTaskList:                      &types.TaskList{Name: testTaskListName},\n\t\t\t\t\tInput:                         buf.Bytes(),\n\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(5),\n\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(5),\n\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\t\t\t},\n\t\t\t}}, nil\n\t\t}\n\n\t\ts.Logger.Info(\"Completing Workflow.\")\n\n\t\tworkflowComplete = true\n\t\treturn []byte(strconv.Itoa(int(activityCounter))), []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: []byte(\"Done.\"),\n\t\t\t},\n\t\t}}, nil\n\t}\n\n\tactivityExecutedCount := int32(0)\n\tatHandler := func(execution *types.WorkflowExecution, activityType *types.ActivityType,\n\t\tactivityID string, input []byte, taskToken []byte) ([]byte, bool, error) {\n\t\ts.Equal(testWorkflowID, execution.WorkflowID)\n\t\ts.Equal(activityName, activityType.Name)\n\n\t\tactivityExecutedCount++\n\t\treturn []byte(\"Activity Result.\"), false, nil\n\t}\n\n\tpoller := &TaskPoller{\n\t\tEngine:          s.Engine,\n\t\tDomain:          s.DomainName,\n\t\tTaskList:        &types.TaskList{Name: testTaskListName},\n\t\tIdentity:        testIdentity,\n\t\tDecisionHandler: dtHandler,\n\t\tActivityHandler: atHandler,\n\t\tLogger:          s.Logger,\n\t\tT:               s.T(),\n\t}\n\n\tfor i := int32(0); i < activityCount; i++ {\n\t\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\t\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\n\t\terr = poller.PollAndProcessActivityTask(false)\n\t\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\t}\n\n\ts.Logger.Info(\"Waiting for workflow to complete\", tag.WorkflowRunID(we.RunID))\n\n\ts.False(workflowComplete)\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.True(err == nil || err == tasklist.ErrNoTasks)\n\ts.True(workflowComplete)\n\n\thistoryResponse, err := s.Engine.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.DomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t},\n\t})\n\ts.NoError(err)\n\thistory := historyResponse.History\n\tfirstEvent := history.Events[0]\n\tlastEvent := history.Events[len(history.Events)-1]\n\t// First 7 event ids --> 0 (Workflow start), 1-3 ( Decision scheduled,started,completed) , 4-6 (Activity scheduled,started,completed) post which the tasks will be rate limited since RPS is 2\n\teventBeforeRatelimited := history.Events[7]\n\n\ttimeElapsedBeforeRatelimiting := time.Unix(0, common.Int64Default(eventBeforeRatelimited.Timestamp)).Sub(time.Unix(0, common.Int64Default(firstEvent.Timestamp)))\n\ts.True(timeElapsedBeforeRatelimiting < 1*time.Second)\n\n\ttotalTime := time.Unix(0, common.Int64Default(lastEvent.Timestamp)).Sub(time.Unix(0, common.Int64Default(firstEvent.Timestamp)))\n\ts.True(totalTime > 3*time.Second)\n\n}\n"
  },
  {
    "path": "host/xdc/elasticsearch_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build !race && esintegration\n// +build !race,esintegration\n\n// to run locally, make sure kafka and es is running,\n// then run cmd `go test -v ./host/xdc -run TestESCrossDCTestSuite -tags esintegration`\npackage xdc\n\nimport (\n\t\"encoding/json\"\n\t\"flag\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/environment\"\n\t\"github.com/uber/cadence/host\"\n\t\"github.com/uber/cadence/host/esutils\"\n)\n\nconst (\n\tnumOfRetry        = 100\n\twaitTimeInMs      = 400\n\twaitForESToSettle = 4 * time.Second // wait es shards for some time ensure data consistent\n)\n\ntype esCrossDCTestSuite struct {\n\t// override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test,\n\t// not merely log an error\n\t*require.Assertions\n\tsuite.Suite\n\tcluster1       *host.TestCluster\n\tcluster2       *host.TestCluster\n\tlogger         log.Logger\n\tclusterConfigs []*host.TestClusterConfig\n\tesClient       esutils.ESClient\n\n\ttestSearchAttributeKey string\n\ttestSearchAttributeVal string\n}\n\nfunc TestESCrossDCTestSuite(t *testing.T) {\n\tflag.Parse()\n\tsuite.Run(t, new(esCrossDCTestSuite))\n}\n\nvar (\n\tclusterNameES              = []string{\"active-es\", \"standby-es\"}\n\tclusterReplicationConfigES = []*types.ClusterReplicationConfiguration{\n\t\t{\n\t\t\tClusterName: clusterNameES[0],\n\t\t},\n\t\t{\n\t\t\tClusterName: clusterNameES[1],\n\t\t},\n\t}\n)\n\nfunc (s *esCrossDCTestSuite) SetupSuite() {\n\ts.logger = testlogger.New(s.T())\n\n\tfileName := \"../testdata/xdc_integration_es_clusters.yaml\"\n\tif host.TestFlags.TestClusterConfigFile != \"\" {\n\t\tfileName = host.TestFlags.TestClusterConfigFile\n\t}\n\ts.Require().NoError(environment.SetupEnv())\n\tconfContent, err := ioutil.ReadFile(fileName)\n\ts.Require().NoError(err)\n\tconfContent = []byte(os.ExpandEnv(string(confContent)))\n\n\tvar clusterConfigs []*host.TestClusterConfig\n\ts.Require().NoError(yaml.Unmarshal(confContent, &clusterConfigs))\n\ts.clusterConfigs = clusterConfigs\n\n\tc, err := host.NewCluster(s.T(), clusterConfigs[0], s.logger.WithTags(tag.ClusterName(clusterNameES[0])))\n\ts.Require().NoError(err)\n\ts.cluster1 = c\n\n\tc, err = host.NewCluster(s.T(), clusterConfigs[1], s.logger.WithTags(tag.ClusterName(clusterNameES[1])))\n\ts.Require().NoError(err)\n\ts.cluster2 = c\n\n\ts.esClient = esutils.CreateESClient(s.Suite, s.clusterConfigs[0].ESConfig.URL.String(), \"v6\")\n\t// TODO Do we also want to run v7 test here?\n\ts.esClient.PutIndexTemplate(s.Suite, \"../testdata/es_index_v6_template.json\", \"test-visibility-template\")\n\ts.esClient.CreateIndex(s.Suite, s.clusterConfigs[0].ESConfig.Indices[constants.VisibilityAppName])\n\ts.esClient.CreateIndex(s.Suite, s.clusterConfigs[1].ESConfig.Indices[constants.VisibilityAppName])\n\n\ts.testSearchAttributeKey = definition.CustomStringField\n\ts.testSearchAttributeVal = \"test value\"\n}\n\nfunc (s *esCrossDCTestSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *esCrossDCTestSuite) TearDownSuite() {\n\ts.cluster1.TearDownCluster()\n\ts.cluster2.TearDownCluster()\n\ts.esClient.DeleteIndex(s.Suite, s.clusterConfigs[0].ESConfig.Indices[constants.VisibilityAppName])\n\ts.esClient.DeleteIndex(s.Suite, s.clusterConfigs[1].ESConfig.Indices[constants.VisibilityAppName])\n}\n\nfunc (s *esCrossDCTestSuite) TestSearchAttributes() {\n\tdomainName := \"test-xdc-search-attr-\" + common.GenerateRandomString(5)\n\tclient1 := s.cluster1.GetFrontendClient() // active\n\tregReq := &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tClusters:                               clusterReplicationConfigES,\n\t\tActiveClusterName:                      clusterNameES[0],\n\t\tIsGlobalDomain:                         true,\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t}\n\terr := client1.RegisterDomain(createContext(), regReq)\n\ts.NoError(err)\n\n\tdescReq := &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t}\n\tresp, err := client1.DescribeDomain(createContext(), descReq)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\t// Wait for domain cache to pick the change\n\ttime.Sleep(cacheRefreshInterval)\n\n\tclient2 := s.cluster2.GetFrontendClient() // standby\n\tresp2, err := client2.DescribeDomain(createContext(), descReq)\n\ts.NoError(err)\n\ts.NotNil(resp2)\n\ts.Equal(resp, resp2)\n\n\t// start a workflow\n\tid := \"xdc-search-attr-test-\" + uuid.New()\n\twt := \"xdc-search-attr-test-type\"\n\ttl := \"xdc-search-attr-test-tasklist\"\n\tidentity := \"worker1\"\n\tworkflowType := &types.WorkflowType{Name: wt}\n\ttaskList := &types.TaskList{Name: tl}\n\tattrValBytes, _ := json.Marshal(s.testSearchAttributeVal)\n\tsearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\ts.testSearchAttributeKey: attrValBytes,\n\t\t},\n\t}\n\tstartReq := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              domainName,\n\t\tWorkflowID:                          id,\n\t\tWorkflowType:                        workflowType,\n\t\tTaskList:                            taskList,\n\t\tInput:                               nil,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tIdentity:                            identity,\n\t\tSearchAttributes:                    searchAttr,\n\t}\n\tstartTime := time.Now().UnixNano()\n\twe, err := client1.StartWorkflowExecution(createContext(), startReq)\n\ts.Nil(err)\n\ts.NotNil(we.GetRunID())\n\n\ts.logger.Info(\"StartWorkflowExecution \\n\", tag.WorkflowRunID(we.GetRunID()))\n\n\tstartFilter := &types.StartTimeFilter{}\n\tstartFilter.EarliestTime = common.Int64Ptr(startTime)\n\tquery := fmt.Sprintf(`WorkflowID = \"%s\" and %s = \"%s\"`, id, s.testSearchAttributeKey, s.testSearchAttributeVal)\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   domainName,\n\t\tPageSize: 5,\n\t\tQuery:    query,\n\t}\n\n\ttestListResult := func(client host.FrontendClient) {\n\t\tvar openExecution *types.WorkflowExecutionInfo\n\t\tfor i := 0; i < numOfRetry; i++ {\n\t\t\tstartFilter.LatestTime = common.Int64Ptr(time.Now().UnixNano())\n\n\t\t\tresp, err := client.ListWorkflowExecutions(createContext(), listRequest)\n\t\t\ts.Nil(err)\n\t\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\t\topenExecution = resp.GetExecutions()[0]\n\t\t\t\tbreak\n\t\t\t}\n\t\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t\t}\n\t\ts.NotNil(openExecution)\n\t\ts.Equal(we.GetRunID(), openExecution.GetExecution().GetRunID())\n\t\tsearchValBytes := openExecution.SearchAttributes.GetIndexedFields()[s.testSearchAttributeKey]\n\t\tvar searchVal string\n\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\ts.Equal(s.testSearchAttributeVal, searchVal)\n\t}\n\n\t// List workflow in active\n\tengine1 := s.cluster1.GetFrontendClient()\n\ttestListResult(engine1)\n\n\t// List workflow in standby\n\tengine2 := s.cluster2.GetFrontendClient()\n\ttestListResult(engine2)\n\n\t// upsert search attributes\n\tdtHandler := func(execution *types.WorkflowExecution, wt *types.WorkflowType,\n\t\tpreviousStartedEventID, startedEventID int64, history *types.History) ([]byte, []*types.Decision, error) {\n\n\t\tupsertDecision := &types.Decision{\n\t\t\tDecisionType: types.DecisionTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: getUpsertSearchAttributes(),\n\t\t\t}}\n\n\t\treturn nil, []*types.Decision{upsertDecision}, nil\n\t}\n\n\tpoller := host.TaskPoller{\n\t\tEngine:          client1,\n\t\tDomain:          domainName,\n\t\tTaskList:        taskList,\n\t\tIdentity:        identity,\n\t\tDecisionHandler: dtHandler,\n\t\tLogger:          s.logger,\n\t\tT:               s.T(),\n\t}\n\n\t_, err = poller.PollAndProcessDecisionTask(false, false)\n\ts.logger.Info(\"PollAndProcessDecisionTask\", tag.Error(err))\n\ts.Nil(err)\n\n\ttime.Sleep(waitForESToSettle)\n\n\tlistRequest = &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   domainName,\n\t\tPageSize: int32(2),\n\t\tQuery:    fmt.Sprintf(`WorkflowType = '%s' and CloseTime = missing`, wt),\n\t}\n\n\ttestListResult = func(client host.FrontendClient) {\n\t\tverified := false\n\t\tfor i := 0; i < numOfRetry; i++ {\n\t\t\tresp, err := client.ListWorkflowExecutions(createContext(), listRequest)\n\t\t\ts.Nil(err)\n\t\t\tif len(resp.GetExecutions()) == 1 {\n\t\t\t\texecution := resp.GetExecutions()[0]\n\t\t\t\tretrievedSearchAttr := execution.SearchAttributes\n\t\t\t\tif retrievedSearchAttr != nil && len(retrievedSearchAttr.GetIndexedFields()) == 2 {\n\t\t\t\t\tfields := retrievedSearchAttr.GetIndexedFields()\n\t\t\t\t\tsearchValBytes := fields[s.testSearchAttributeKey]\n\t\t\t\t\tvar searchVal string\n\t\t\t\t\tjson.Unmarshal(searchValBytes, &searchVal)\n\t\t\t\t\ts.Equal(\"another string\", searchVal)\n\n\t\t\t\t\tsearchValBytes2 := fields[definition.CustomIntField]\n\t\t\t\t\tvar searchVal2 int\n\t\t\t\t\tjson.Unmarshal(searchValBytes2, &searchVal2)\n\t\t\t\t\ts.Equal(123, searchVal2)\n\n\t\t\t\t\tverified = true\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t\t}\n\t\ts.True(verified)\n\t}\n\n\t// test upsert result in active\n\ttestListResult(engine1)\n\n\t// terminate workflow\n\tterminateReason := \"force terminate to make sure standby process tasks\"\n\tterminateDetails := []byte(\"terminate details.\")\n\terr = client1.TerminateWorkflowExecution(createContext(), &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: domainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t\tReason:   terminateReason,\n\t\tDetails:  terminateDetails,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\t// check terminate done\n\texecutionTerminated := false\n\tgetHistoryReq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: id,\n\t\t},\n\t}\nGetHistoryLoop:\n\tfor i := 0; i < 10; i++ {\n\t\thistoryResponse, err := client1.GetWorkflowExecutionHistory(createContext(), getHistoryReq)\n\t\ts.Nil(err)\n\t\thistory := historyResponse.History\n\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif *lastEvent.EventType != types.EventTypeWorkflowExecutionTerminated {\n\t\t\ts.logger.Warn(\"Execution not terminated yet.\")\n\t\t\ttime.Sleep(100 * time.Millisecond)\n\t\t\tcontinue GetHistoryLoop\n\t\t}\n\n\t\tterminateEventAttributes := lastEvent.WorkflowExecutionTerminatedEventAttributes\n\t\ts.Equal(terminateReason, terminateEventAttributes.Reason)\n\t\ts.Equal(terminateDetails, terminateEventAttributes.Details)\n\t\ts.Equal(identity, terminateEventAttributes.Identity)\n\t\texecutionTerminated = true\n\t\tbreak GetHistoryLoop\n\t}\n\ts.True(executionTerminated)\n\n\t// check history replicated to the other cluster\n\tvar historyResponse *types.GetWorkflowExecutionHistoryResponse\n\teventsReplicated := false\nGetHistoryLoop2:\n\tfor i := 0; i < numOfRetry; i++ {\n\t\thistoryResponse, err = client2.GetWorkflowExecutionHistory(createContext(), getHistoryReq)\n\t\tif err == nil {\n\t\t\thistory := historyResponse.History\n\t\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\t\tif *lastEvent.EventType == types.EventTypeWorkflowExecutionTerminated {\n\t\t\t\tterminateEventAttributes := lastEvent.WorkflowExecutionTerminatedEventAttributes\n\t\t\t\ts.Equal(terminateReason, terminateEventAttributes.Reason)\n\t\t\t\ts.Equal(terminateDetails, terminateEventAttributes.Details)\n\t\t\t\ts.Equal(identity, terminateEventAttributes.Identity)\n\t\t\t\teventsReplicated = true\n\t\t\t\tbreak GetHistoryLoop2\n\t\t\t}\n\t\t}\n\t\ttime.Sleep(waitTimeInMs * time.Millisecond)\n\t}\n\ts.Nil(err)\n\ts.True(eventsReplicated)\n\n\t// test upsert result in standby\n\ttestListResult(engine2)\n}\n\nfunc getUpsertSearchAttributes() *types.SearchAttributes {\n\tattrValBytes1, _ := json.Marshal(\"another string\")\n\tattrValBytes2, _ := json.Marshal(123)\n\tupsertSearchAttr := &types.SearchAttributes{\n\t\tIndexedFields: map[string][]byte{\n\t\t\tdefinition.CustomStringField: attrValBytes1,\n\t\t\tdefinition.CustomIntField:    attrValBytes2,\n\t\t},\n\t}\n\treturn upsertSearchAttr\n}\n"
  },
  {
    "path": "internal/tools/go.mod",
    "content": "module github.com/uber/cadence/internal/tools\n\ngo 1.24.0\n\nrequire (\n\tgithub.com/daixiang0/gci v0.12.0\n\tgithub.com/dmarkham/enumer v1.5.8\n\tgithub.com/gogo/protobuf v1.3.2\n\tgithub.com/hexdigest/gowrap v1.2.5\n\tgithub.com/mgechev/revive v1.3.2\n\tgithub.com/vektra/mockery/v2 v2.53.5\n\tgo.uber.org/mock v0.5.0\n\tgo.uber.org/nilaway v0.0.0-20251021214447-34f56b8c16b9\n\tgo.uber.org/thriftrw v1.29.2\n\tgo.uber.org/yarpc v1.70.3\n\tgolang.org/x/tools v0.38.0\n)\n\nrequire (\n\t4d63.com/gochecknoglobals v0.1.0 // indirect\n\tcel.dev/expr v0.16.1 // indirect\n\tcloud.google.com/go v0.116.0 // indirect\n\tcloud.google.com/go/auth v0.13.0 // indirect\n\tcloud.google.com/go/auth/oauth2adapt v0.2.6 // indirect\n\tcloud.google.com/go/compute/metadata v0.6.0 // indirect\n\tcloud.google.com/go/iam v1.2.2 // indirect\n\tcloud.google.com/go/kms v1.20.1 // indirect\n\tcloud.google.com/go/longrunning v0.6.2 // indirect\n\tcloud.google.com/go/monitoring v1.21.2 // indirect\n\tcloud.google.com/go/storage v1.49.0 // indirect\n\tcode.gitea.io/sdk/gitea v0.14.0 // indirect\n\tgithub.com/AlekSi/pointer v1.2.0 // indirect\n\tgithub.com/Antonboom/errname v0.1.6 // indirect\n\tgithub.com/Antonboom/nilnil v0.1.1 // indirect\n\tgithub.com/Azure/azure-pipeline-go v0.2.3 // indirect\n\tgithub.com/Azure/azure-sdk-for-go v54.0.0+incompatible // indirect\n\tgithub.com/Azure/azure-storage-blob-go v0.13.0 // indirect\n\tgithub.com/Azure/go-autorest v14.2.0+incompatible // indirect\n\tgithub.com/Azure/go-autorest/autorest v0.11.18 // indirect\n\tgithub.com/Azure/go-autorest/autorest/adal v0.9.13 // indirect\n\tgithub.com/Azure/go-autorest/autorest/azure/auth v0.5.7 // indirect\n\tgithub.com/Azure/go-autorest/autorest/azure/cli v0.4.2 // indirect\n\tgithub.com/Azure/go-autorest/autorest/date v0.3.0 // indirect\n\tgithub.com/Azure/go-autorest/autorest/to v0.4.0 // indirect\n\tgithub.com/Azure/go-autorest/autorest/validation v0.3.1 // indirect\n\tgithub.com/Azure/go-autorest/logger v0.2.1 // indirect\n\tgithub.com/Azure/go-autorest/tracing v0.6.0 // indirect\n\tgithub.com/BurntSushi/toml v1.2.1 // indirect\n\tgithub.com/Djarvur/go-err113 v0.0.0-20210108212216-aea10b59be24 // indirect\n\tgithub.com/GaijinEntertainment/go-exhaustruct/v2 v2.1.0 // indirect\n\tgithub.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.25.0 // indirect\n\tgithub.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.48.1 // indirect\n\tgithub.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.48.1 // indirect\n\tgithub.com/Masterminds/goutils v1.1.1 // indirect\n\tgithub.com/Masterminds/semver v1.5.0 // indirect\n\tgithub.com/Masterminds/semver/v3 v3.2.1 // indirect\n\tgithub.com/Masterminds/sprig v2.22.0+incompatible // indirect\n\tgithub.com/Masterminds/sprig/v3 v3.2.2 // indirect\n\tgithub.com/Microsoft/go-winio v0.5.1 // indirect\n\tgithub.com/OpenPeeDeeP/depguard v1.1.0 // indirect\n\tgithub.com/ProtonMail/go-crypto v0.0.0-20210512092938-c05353c2d58c // indirect\n\tgithub.com/alexkohler/prealloc v1.0.0 // indirect\n\tgithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 // indirect\n\tgithub.com/apex/log v1.9.0 // indirect\n\tgithub.com/ashanbrown/forbidigo v1.3.0 // indirect\n\tgithub.com/ashanbrown/makezero v1.1.1 // indirect\n\tgithub.com/aws/aws-sdk-go v1.38.35 // indirect\n\tgithub.com/beorn7/perks v1.0.1 // indirect\n\tgithub.com/bkielbasa/cyclop v1.2.0 // indirect\n\tgithub.com/blakesmith/ar v0.0.0-20190502131153-809d4375e1fb // indirect\n\tgithub.com/blizzy78/varnamelen v0.8.0 // indirect\n\tgithub.com/bombsimon/wsl/v3 v3.3.0 // indirect\n\tgithub.com/breml/bidichk v0.2.3 // indirect\n\tgithub.com/breml/errchkjson v0.3.0 // indirect\n\tgithub.com/butuzov/ireturn v0.1.1 // indirect\n\tgithub.com/caarlos0/ctrlc v1.0.0 // indirect\n\tgithub.com/caarlos0/env/v6 v6.6.2 // indirect\n\tgithub.com/caarlos0/go-shellwords v1.0.12 // indirect\n\tgithub.com/campoy/unique v0.0.0-20180121183637-88950e537e7e // indirect\n\tgithub.com/cavaliergopher/cpio v1.0.1 // indirect\n\tgithub.com/cenkalti/backoff v2.1.1+incompatible // indirect\n\tgithub.com/census-instrumentation/opencensus-proto v0.4.1 // indirect\n\tgithub.com/cespare/xxhash/v2 v2.3.0 // indirect\n\tgithub.com/charithe/durationcheck v0.0.9 // indirect\n\tgithub.com/chavacava/garif v0.0.0-20230227094218-b8c73b2037b8 // indirect\n\tgithub.com/chigopher/pathlib v0.19.1 // indirect\n\tgithub.com/cncf/xds/go v0.0.0-20240905190251-b4127c9b8d78 // indirect\n\tgithub.com/cpuguy83/go-md2man/v2 v2.0.4 // indirect\n\tgithub.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect\n\tgithub.com/denis-tingaikin/go-header v0.4.3 // indirect\n\tgithub.com/dghubble/go-twitter v0.0.0-20201011215211-4b180d0cc78d // indirect\n\tgithub.com/dghubble/oauth1 v0.7.0 // indirect\n\tgithub.com/dghubble/sling v1.3.0 // indirect\n\tgithub.com/dimchansky/utfbom v1.1.1 // indirect\n\tgithub.com/emirpasic/gods v1.12.0 // indirect\n\tgithub.com/envoyproxy/go-control-plane v0.13.1 // indirect\n\tgithub.com/envoyproxy/protoc-gen-validate v1.1.0 // indirect\n\tgithub.com/esimonov/ifshort v1.0.4 // indirect\n\tgithub.com/ettle/strcase v0.1.1 // indirect\n\tgithub.com/fatih/color v1.15.0 // indirect\n\tgithub.com/fatih/structtag v1.2.0 // indirect\n\tgithub.com/felixge/httpsnoop v1.0.4 // indirect\n\tgithub.com/firefart/nonamedreturns v1.0.1 // indirect\n\tgithub.com/form3tech-oss/jwt-go v3.2.2+incompatible // indirect\n\tgithub.com/fsnotify/fsnotify v1.8.0 // indirect\n\tgithub.com/fzipp/gocyclo v0.5.1 // indirect\n\tgithub.com/go-critic/go-critic v0.6.3 // indirect\n\tgithub.com/go-git/gcfg v1.5.0 // indirect\n\tgithub.com/go-git/go-billy/v5 v5.3.1 // indirect\n\tgithub.com/go-git/go-git/v5 v5.3.0 // indirect\n\tgithub.com/go-logr/logr v1.4.2 // indirect\n\tgithub.com/go-logr/stdr v1.2.2 // indirect\n\tgithub.com/go-toolsmith/astcast v1.0.0 // indirect\n\tgithub.com/go-toolsmith/astcopy v1.0.0 // indirect\n\tgithub.com/go-toolsmith/astequal v1.0.1 // indirect\n\tgithub.com/go-toolsmith/astfmt v1.0.0 // indirect\n\tgithub.com/go-toolsmith/astp v1.0.0 // indirect\n\tgithub.com/go-toolsmith/strparse v1.0.0 // indirect\n\tgithub.com/go-toolsmith/typep v1.0.2 // indirect\n\tgithub.com/go-viper/mapstructure/v2 v2.2.1 // indirect\n\tgithub.com/go-xmlfmt/xmlfmt v0.0.0-20191208150333-d5b6f63a941b // indirect\n\tgithub.com/gobwas/glob v0.2.3 // indirect\n\tgithub.com/gofrs/flock v0.8.1 // indirect\n\tgithub.com/gojuno/minimock/v3 v3.0.10 // indirect\n\tgithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect\n\tgithub.com/golang/protobuf v1.5.4 // indirect\n\tgithub.com/golangci/check v0.0.0-20180506172741-cfe4005ccda2 // indirect\n\tgithub.com/golangci/dupl v0.0.0-20180902072040-3e9179ac440a // indirect\n\tgithub.com/golangci/go-misc v0.0.0-20220329215616-d24fe342adfe // indirect\n\tgithub.com/golangci/gofmt v0.0.0-20190930125516-244bba706f1a // indirect\n\tgithub.com/golangci/golangci-lint v1.46.2 // indirect\n\tgithub.com/golangci/lint-1 v0.0.0-20191013205115-297bf364a8e0 // indirect\n\tgithub.com/golangci/maligned v0.0.0-20180506175553-b1d89398deca // indirect\n\tgithub.com/golangci/misspell v0.3.5 // indirect\n\tgithub.com/golangci/revgrep v0.0.0-20210930125155-c22e5001d4f2 // indirect\n\tgithub.com/golangci/unconvert v0.0.0-20180507085042-28b1c447d1f4 // indirect\n\tgithub.com/google/go-cmp v0.7.0 // indirect\n\tgithub.com/google/go-github/v35 v35.2.0 // indirect\n\tgithub.com/google/go-querystring v1.0.0 // indirect\n\tgithub.com/google/s2a-go v0.1.8 // indirect\n\tgithub.com/google/uuid v1.6.0 // indirect\n\tgithub.com/google/wire v0.5.0 // indirect\n\tgithub.com/googleapis/enterprise-certificate-proxy v0.3.4 // indirect\n\tgithub.com/googleapis/gax-go/v2 v2.14.1 // indirect\n\tgithub.com/gordonklaus/ineffassign v0.0.0-20210914165742-4cc7213b9bc8 // indirect\n\tgithub.com/goreleaser/chglog v0.4.2 // indirect\n\tgithub.com/goreleaser/fileglob v1.3.0 // indirect\n\tgithub.com/goreleaser/goreleaser v0.169.0 // indirect\n\tgithub.com/goreleaser/nfpm/v2 v2.29.0 // indirect\n\tgithub.com/gostaticanalysis/analysisutil v0.7.1 // indirect\n\tgithub.com/gostaticanalysis/comment v1.4.2 // indirect\n\tgithub.com/gostaticanalysis/forcetypeassert v0.1.0 // indirect\n\tgithub.com/gostaticanalysis/nilerr v0.1.1 // indirect\n\tgithub.com/hashicorp/errwrap v1.0.0 // indirect\n\tgithub.com/hashicorp/go-cleanhttp v0.5.2 // indirect\n\tgithub.com/hashicorp/go-multierror v1.1.1 // indirect\n\tgithub.com/hashicorp/go-retryablehttp v0.6.8 // indirect\n\tgithub.com/hashicorp/go-version v1.4.0 // indirect\n\tgithub.com/hexops/gotextdiff v1.0.3 // indirect\n\tgithub.com/huandu/xstrings v1.4.0 // indirect\n\tgithub.com/iancoleman/strcase v0.3.0 // indirect\n\tgithub.com/imdario/mergo v0.3.15 // indirect\n\tgithub.com/inconshreveable/mousetrap v1.1.0 // indirect\n\tgithub.com/jbenet/go-context v0.0.0-20150711004518-d14ea06fba99 // indirect\n\tgithub.com/jessevdk/go-flags v1.5.0 // indirect\n\tgithub.com/jgautheron/goconst v1.5.1 // indirect\n\tgithub.com/jingyugao/rowserrcheck v1.1.1 // indirect\n\tgithub.com/jinzhu/copier v0.4.0 // indirect\n\tgithub.com/jirfag/go-printf-func-name v0.0.0-20200119135958-7558a9eaa5af // indirect\n\tgithub.com/jmespath/go-jmespath v0.4.0 // indirect\n\tgithub.com/julz/importas v0.1.0 // indirect\n\tgithub.com/kevinburke/ssh_config v1.1.0 // indirect\n\tgithub.com/kisielk/errcheck v1.6.0 // indirect\n\tgithub.com/kisielk/gotool v1.0.0 // indirect\n\tgithub.com/klauspost/compress v1.18.0 // indirect\n\tgithub.com/klauspost/pgzip v1.2.6 // indirect\n\tgithub.com/kulti/thelper v0.6.2 // indirect\n\tgithub.com/kunwardeep/paralleltest v1.0.3 // indirect\n\tgithub.com/kyoh86/exportloopref v0.1.8 // indirect\n\tgithub.com/ldez/gomoddirectives v0.2.3 // indirect\n\tgithub.com/ldez/tagliatelle v0.3.1 // indirect\n\tgithub.com/leonklingele/grouper v1.1.0 // indirect\n\tgithub.com/lufeee/execinquery v1.2.1 // indirect\n\tgithub.com/maratori/testpackage v1.0.1 // indirect\n\tgithub.com/matoous/godox v0.0.0-20210227103229-6504466cf951 // indirect\n\tgithub.com/mattn/go-colorable v0.1.14 // indirect\n\tgithub.com/mattn/go-ieproxy v0.0.1 // indirect\n\tgithub.com/mattn/go-isatty v0.0.20 // indirect\n\tgithub.com/mattn/go-runewidth v0.0.9 // indirect\n\tgithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect\n\tgithub.com/mbilski/exhaustivestruct v1.2.0 // indirect\n\tgithub.com/mgechev/dots v0.0.0-20210922191527-e955255bf517 // indirect\n\tgithub.com/mitchellh/copystructure v1.2.0 // indirect\n\tgithub.com/mitchellh/go-homedir v1.1.0 // indirect\n\tgithub.com/mitchellh/mapstructure v1.5.0 // indirect\n\tgithub.com/mitchellh/reflectwalk v1.0.2 // indirect\n\tgithub.com/moricho/tparallel v0.2.1 // indirect\n\tgithub.com/nakabonne/nestif v0.3.1 // indirect\n\tgithub.com/nbutton23/zxcvbn-go v0.0.0-20210217022336-fa2cb2858354 // indirect\n\tgithub.com/nishanths/exhaustive v0.7.11 // indirect\n\tgithub.com/nishanths/predeclared v0.2.2 // indirect\n\tgithub.com/olekukonko/tablewriter v0.0.5 // indirect\n\tgithub.com/opentracing/opentracing-go v1.2.0 // indirect\n\tgithub.com/pascaldekloe/name v1.0.0 // indirect\n\tgithub.com/pelletier/go-toml/v2 v2.2.3 // indirect\n\tgithub.com/phayes/checkstyle v0.0.0-20170904204023-bfd46e6a821d // indirect\n\tgithub.com/pkg/errors v0.9.1 // indirect\n\tgithub.com/planetscale/vtprotobuf v0.6.1-0.20240319094008-0393e58bdf10 // indirect\n\tgithub.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect\n\tgithub.com/polyfloyd/go-errorlint v1.0.0 // indirect\n\tgithub.com/prometheus/client_golang v1.12.1 // indirect\n\tgithub.com/prometheus/client_model v0.6.0 // indirect\n\tgithub.com/prometheus/common v0.32.1 // indirect\n\tgithub.com/prometheus/procfs v0.7.3 // indirect\n\tgithub.com/quasilyte/go-ruleguard v0.3.16-0.20220213074421-6aa060fab41a // indirect\n\tgithub.com/quasilyte/gogrep v0.0.0-20220120141003-628d8b3623b5 // indirect\n\tgithub.com/quasilyte/regex/syntax v0.0.0-20200407221936-30656e2c4a95 // indirect\n\tgithub.com/quasilyte/stdinfo v0.0.0-20220114132959-f7386bf02567 // indirect\n\tgithub.com/rs/zerolog v1.33.0 // indirect\n\tgithub.com/russross/blackfriday/v2 v2.1.0 // indirect\n\tgithub.com/ryancurrah/gomodguard v1.2.3 // indirect\n\tgithub.com/ryanrolds/sqlclosecheck v0.3.0 // indirect\n\tgithub.com/sagikazarmark/locafero v0.7.0 // indirect\n\tgithub.com/sanposhiho/wastedassign/v2 v2.0.6 // indirect\n\tgithub.com/securego/gosec/v2 v2.11.0 // indirect\n\tgithub.com/sergi/go-diff v1.2.0 // indirect\n\tgithub.com/shazow/go-diff v0.0.0-20160112020656-b6b7b6733b8c // indirect\n\tgithub.com/shopspring/decimal v1.2.0 // indirect\n\tgithub.com/sirupsen/logrus v1.8.1 // indirect\n\tgithub.com/sivchari/containedctx v1.0.2 // indirect\n\tgithub.com/sivchari/tenv v1.5.0 // indirect\n\tgithub.com/sonatard/noctx v0.0.1 // indirect\n\tgithub.com/sourcegraph/conc v0.3.0 // indirect\n\tgithub.com/sourcegraph/go-diff v0.6.1 // indirect\n\tgithub.com/spf13/afero v1.12.0 // indirect\n\tgithub.com/spf13/cast v1.7.1 // indirect\n\tgithub.com/spf13/cobra v1.8.1 // indirect\n\tgithub.com/spf13/pflag v1.0.6 // indirect\n\tgithub.com/spf13/viper v1.20.0 // indirect\n\tgithub.com/ssgreg/nlreturn/v2 v2.2.1 // indirect\n\tgithub.com/stbenjam/no-sprintf-host-port v0.1.1 // indirect\n\tgithub.com/stretchr/objx v0.5.2 // indirect\n\tgithub.com/stretchr/testify v1.10.0 // indirect\n\tgithub.com/subosito/gotenv v1.6.0 // indirect\n\tgithub.com/sylvia7788/contextcheck v1.0.4 // indirect\n\tgithub.com/tdakkota/asciicheck v0.1.1 // indirect\n\tgithub.com/tetafro/godot v1.4.11 // indirect\n\tgithub.com/timakin/bodyclose v0.0.0-20210704033933-f49887972144 // indirect\n\tgithub.com/tomarrell/wrapcheck/v2 v2.6.1 // indirect\n\tgithub.com/tommy-muehle/go-mnd/v2 v2.5.0 // indirect\n\tgithub.com/ulikunitz/xz v0.5.11 // indirect\n\tgithub.com/ultraware/funlen v0.0.3 // indirect\n\tgithub.com/ultraware/whitespace v0.0.5 // indirect\n\tgithub.com/uudashr/gocognit v1.0.5 // indirect\n\tgithub.com/xanzy/go-gitlab v0.50.0 // indirect\n\tgithub.com/xanzy/ssh-agent v0.3.1 // indirect\n\tgithub.com/yagipy/maintidx v1.0.0 // indirect\n\tgithub.com/yeya24/promlinter v0.2.0 // indirect\n\tgitlab.com/bosi/decorder v0.2.1 // indirect\n\tgitlab.com/digitalxero/go-conventional-commit v1.0.7 // indirect\n\tgo.opencensus.io v0.24.0 // indirect\n\tgo.opentelemetry.io/contrib/detectors/gcp v1.29.0 // indirect\n\tgo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.54.0 // indirect\n\tgo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.54.0 // indirect\n\tgo.opentelemetry.io/otel v1.29.0 // indirect\n\tgo.opentelemetry.io/otel/metric v1.29.0 // indirect\n\tgo.opentelemetry.io/otel/sdk v1.29.0 // indirect\n\tgo.opentelemetry.io/otel/sdk/metric v1.29.0 // indirect\n\tgo.opentelemetry.io/otel/trace v1.29.0 // indirect\n\tgo.uber.org/atomic v1.9.0 // indirect\n\tgo.uber.org/fx v1.13.1 // indirect\n\tgo.uber.org/multierr v1.11.0 // indirect\n\tgo.uber.org/zap v1.24.0 // indirect\n\tgocloud.dev v0.23.0 // indirect\n\tgolang.org/x/crypto v0.43.0 // indirect\n\tgolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e // indirect\n\tgolang.org/x/mod v0.29.0 // indirect\n\tgolang.org/x/net v0.46.0 // indirect\n\tgolang.org/x/oauth2 v0.25.0 // indirect\n\tgolang.org/x/sync v0.17.0 // indirect\n\tgolang.org/x/sys v0.37.0 // indirect\n\tgolang.org/x/telemetry v0.0.0-20251008203120-078029d740a8 // indirect\n\tgolang.org/x/term v0.36.0 // indirect\n\tgolang.org/x/text v0.30.0 // indirect\n\tgolang.org/x/time v0.8.0 // indirect\n\tgolang.org/x/tools/go/expect v0.1.1-deprecated // indirect\n\tgolang.org/x/tools/go/packages/packagestest v0.1.1-deprecated // indirect\n\tgolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 // indirect\n\tgoogle.golang.org/api v0.215.0 // indirect\n\tgoogle.golang.org/genproto v0.0.0-20241118233622-e639e219e697 // indirect\n\tgoogle.golang.org/genproto/googleapis/api v0.0.0-20241209162323-e6fa225c2576 // indirect\n\tgoogle.golang.org/genproto/googleapis/rpc v0.0.0-20241223144023-3abc09e42ca8 // indirect\n\tgoogle.golang.org/grpc v1.67.3 // indirect\n\tgoogle.golang.org/protobuf v1.36.1 // indirect\n\tgopkg.in/warnings.v0 v0.1.2 // indirect\n\tgopkg.in/yaml.v2 v2.4.0 // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n\thonnef.co/go/tools v0.3.2 // indirect\n\tmvdan.cc/gofumpt v0.3.1 // indirect\n\tmvdan.cc/interfacer v0.0.0-20180901003855-c20040233aed // indirect\n\tmvdan.cc/lint v0.0.0-20170908181259-adc824a0674b // indirect\n\tmvdan.cc/unparam v0.0.0-20211214103731-d0ef000c54e5 // indirect\n)\n"
  },
  {
    "path": "internal/tools/go.sum",
    "content": "4d63.com/gochecknoglobals v0.1.0 h1:zeZSRqj5yCg28tCkIV/z/lWbwvNm5qnKVS15PI8nhD0=\n4d63.com/gochecknoglobals v0.1.0/go.mod h1:wfdC5ZjKSPr7CybKEcgJhUOgeAQW1+7WcyK8OvUilfo=\nbazil.org/fuse v0.0.0-20180421153158-65cc252bf669/go.mod h1:Xbm+BRKSBEpa4q4hTSxohYNQpsxXPbPry4JJWOB3LB8=\nbitbucket.org/creachadair/shell v0.0.6/go.mod h1:8Qqi/cYk7vPnsOePHroKXDJYmb5x7ENhtiFtfZq8K+M=\ncel.dev/expr v0.16.1 h1:NR0+oFYzR1CqLFhTAqg3ql59G9VfN8fKq1TCHJ6gq1g=\ncel.dev/expr v0.16.1/go.mod h1:AsGA5zb3WruAEQeQng1RZdGEXmBj0jvMWh6l5SnNuC8=\ncloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go v0.38.0/go.mod h1:990N+gfupTy94rShfmMCWGDn0LpTmnzTp2qbd1dvSRU=\ncloud.google.com/go v0.39.0/go.mod h1:rVLT6fkc8chs9sfPtFc1SBH6em7n+ZoXaG+87tDISts=\ncloud.google.com/go v0.44.1/go.mod h1:iSa0KzasP4Uvy3f1mN/7PiObzGgflwredwwASm/v6AU=\ncloud.google.com/go v0.44.2/go.mod h1:60680Gw3Yr4ikxnPRS/oxxkBccT6SA1yMk63TGekxKY=\ncloud.google.com/go v0.44.3/go.mod h1:60680Gw3Yr4ikxnPRS/oxxkBccT6SA1yMk63TGekxKY=\ncloud.google.com/go v0.45.1/go.mod h1:RpBamKRgapWJb87xiFSdk4g1CME7QZg3uwTez+TSTjc=\ncloud.google.com/go v0.46.3/go.mod h1:a6bKKbmY7er1mI7TEI4lsAkts/mkhTSZK8w33B4RAg0=\ncloud.google.com/go v0.50.0/go.mod h1:r9sluTvynVuxRIOHXQEHMFffphuXHOMZMycpNR5e6To=\ncloud.google.com/go v0.52.0/go.mod h1:pXajvRH/6o3+F9jDHZWQ5PbGhn+o8w9qiu/CffaVdO4=\ncloud.google.com/go v0.53.0/go.mod h1:fp/UouUEsRkN6ryDKNW/Upv/JBKnv6WDthjR6+vze6M=\ncloud.google.com/go v0.54.0/go.mod h1:1rq2OEkV3YMf6n/9ZvGWI3GWw0VoqH/1x2nd8Is/bPc=\ncloud.google.com/go v0.56.0/go.mod h1:jr7tqZxxKOVYizybht9+26Z/gUq7tiRzu+ACVAMbKVk=\ncloud.google.com/go v0.57.0/go.mod h1:oXiQ6Rzq3RAkkY7N6t3TcE6jE+CIBBbA36lwQ1JyzZs=\ncloud.google.com/go v0.60.0/go.mod h1:yw2G51M9IfRboUH61Us8GqCeF1PzPblB823Mn2q2eAU=\ncloud.google.com/go v0.62.0/go.mod h1:jmCYTdRCQuc1PHIIJ/maLInMho30T/Y0M4hTdTShOYc=\ncloud.google.com/go v0.65.0/go.mod h1:O5N8zS7uWy9vkA9vayVHs65eM1ubvY4h553ofrNHObY=\ncloud.google.com/go v0.72.0/go.mod h1:M+5Vjvlc2wnp6tjzE102Dw08nGShTscUx2nZMufOKPI=\ncloud.google.com/go v0.74.0/go.mod h1:VV1xSbzvo+9QJOxLDaJfTjx5e+MePCpCWwvftOeQmWk=\ncloud.google.com/go v0.75.0/go.mod h1:VGuuCn7PG0dwsd5XPVm2Mm3wlh3EL55/79EKB6hlPTY=\ncloud.google.com/go v0.78.0/go.mod h1:QjdrLG0uq+YwhjoVOLsS1t7TW8fs36kLs4XO5R5ECHg=\ncloud.google.com/go v0.79.0/go.mod h1:3bzgcEeQlzbuEAYu4mrWhKqWjmpprinYgKJLgKHnbb8=\ncloud.google.com/go v0.81.0/go.mod h1:mk/AM35KwGk/Nm2YSeZbxXdrNK3KZOYHmLkOqC2V6E0=\ncloud.google.com/go v0.83.0/go.mod h1:Z7MJUsANfY0pYPdw0lbnivPx4/vhy/e2FEkSkF7vAVY=\ncloud.google.com/go v0.84.0/go.mod h1:RazrYuxIK6Kb7YrzzhPoLmCVzl7Sup4NrbKPg8KHSUM=\ncloud.google.com/go v0.87.0/go.mod h1:TpDYlFy7vuLzZMMZ+B6iRiELaY7z/gJPaqbMx6mlWcY=\ncloud.google.com/go v0.90.0/go.mod h1:kRX0mNRHe0e2rC6oNakvwQqzyDmg57xJ+SZU1eT2aDQ=\ncloud.google.com/go v0.93.3/go.mod h1:8utlLll2EF5XMAV15woO4lSbWQlk8rer9aLOfLh7+YI=\ncloud.google.com/go v0.94.1/go.mod h1:qAlAugsXlC+JWO+Bke5vCtc9ONxjQT3drlTTnAplMW4=\ncloud.google.com/go v0.97.0/go.mod h1:GF7l59pYBVlXQIBLx3a761cZ41F9bBH3JUlihCt2Udc=\ncloud.google.com/go v0.99.0/go.mod h1:w0Xx2nLzqWJPuozYQX+hFfCSI8WioryfRDzkoI/Y2ZA=\ncloud.google.com/go v0.100.2/go.mod h1:4Xra9TjzAeYHrl5+oeLlzbM2k3mjVhZh4UqTZ//w99A=\ncloud.google.com/go v0.116.0 h1:B3fRrSDkLRt5qSHWe40ERJvhvnQwdZiHu0bJOpldweE=\ncloud.google.com/go v0.116.0/go.mod h1:cEPSRWPzZEswwdr9BxE6ChEn01dWlTaF05LiC2Xs70U=\ncloud.google.com/go/auth v0.13.0 h1:8Fu8TZy167JkW8Tj3q7dIkr2v4cndv41ouecJx0PAHs=\ncloud.google.com/go/auth v0.13.0/go.mod h1:COOjD9gwfKNKz+IIduatIhYJQIc0mG3H102r/EMxX6Q=\ncloud.google.com/go/auth/oauth2adapt v0.2.6 h1:V6a6XDu2lTwPZWOawrAa9HUK+DB2zfJyTuciBG5hFkU=\ncloud.google.com/go/auth/oauth2adapt v0.2.6/go.mod h1:AlmsELtlEBnaNTL7jCj8VQFLy6mbZv0s4Q7NGBeQ5E8=\ncloud.google.com/go/bigquery v1.0.1/go.mod h1:i/xbL2UlR5RvWAURpBYZTtm/cXjCha9lbfbpx4poX+o=\ncloud.google.com/go/bigquery v1.3.0/go.mod h1:PjpwJnslEMmckchkHFfq+HTD2DmtT67aNFKH1/VBDHE=\ncloud.google.com/go/bigquery v1.4.0/go.mod h1:S8dzgnTigyfTmLBfrtrhyYhwRxG72rYxvftPBK2Dvzc=\ncloud.google.com/go/bigquery v1.5.0/go.mod h1:snEHRnqQbz117VIFhE8bmtwIDY80NLUZUMb4Nv6dBIg=\ncloud.google.com/go/bigquery v1.7.0/go.mod h1://okPTzCYNXSlb24MZs83e2Do+h+VXtc4gLoIoXIAPc=\ncloud.google.com/go/bigquery v1.8.0/go.mod h1:J5hqkt3O0uAFnINi6JXValWIb1v0goeZM77hZzJN/fQ=\ncloud.google.com/go/compute v0.1.0/go.mod h1:GAesmwr110a34z04OlxYkATPBEfVhkymfTBXtfbBFow=\ncloud.google.com/go/compute v1.3.0/go.mod h1:cCZiE1NHEtai4wiufUhW8I8S1JKkAnhnQJWM7YD99wM=\ncloud.google.com/go/compute v1.5.0/go.mod h1:9SMHyhJlzhlkJqrPAc839t2BZFTSk6Jdj6mkzQJeu0M=\ncloud.google.com/go/compute/metadata v0.6.0 h1:A6hENjEsCDtC1k8byVsgwvVcioamEHvZ4j01OwKxG9I=\ncloud.google.com/go/compute/metadata v0.6.0/go.mod h1:FjyFAW1MW0C203CEOMDTu3Dk1FlqW3Rga40jzHL4hfg=\ncloud.google.com/go/datastore v1.0.0/go.mod h1:LXYbyblFSglQ5pkeyhO+Qmw7ukd3C+pD7TKLgZqpHYE=\ncloud.google.com/go/datastore v1.1.0/go.mod h1:umbIZjpQpHh4hmRpGhH4tLFup+FVzqBi1b3c64qFpCk=\ncloud.google.com/go/firestore v1.1.0/go.mod h1:ulACoGHTpvq5r8rxGJ4ddJZBZqakUQqClKRT5SZwBmk=\ncloud.google.com/go/firestore v1.5.0/go.mod h1:c4nNYR1qdq7eaZ+jSc5fonrQN2k3M7sWATcYTiakjEo=\ncloud.google.com/go/firestore v1.6.1/go.mod h1:asNXNOzBdyVQmEU+ggO8UPodTkEVFW5Qx+rwHnAz+EY=\ncloud.google.com/go/iam v1.2.2 h1:ozUSofHUGf/F4tCNy/mu9tHLTaxZFLOUiKzjcgWHGIA=\ncloud.google.com/go/iam v1.2.2/go.mod h1:0Ys8ccaZHdI1dEUilwzqng/6ps2YB6vRsjIe00/+6JY=\ncloud.google.com/go/kms v1.20.1 h1:og29Wv59uf2FVaZlesaiDAqHFzHaoUyHI3HYp9VUHVg=\ncloud.google.com/go/kms v1.20.1/go.mod h1:LywpNiVCvzYNJWS9JUcGJSVTNSwPwi0vBAotzDqn2nc=\ncloud.google.com/go/logging v1.12.0 h1:ex1igYcGFd4S/RZWOCU51StlIEuey5bjqwH9ZYjHibk=\ncloud.google.com/go/logging v1.12.0/go.mod h1:wwYBt5HlYP1InnrtYI0wtwttpVU1rifnMT7RejksUAM=\ncloud.google.com/go/longrunning v0.6.2 h1:xjDfh1pQcWPEvnfjZmwjKQEcHnpz6lHjfy7Fo0MK+hc=\ncloud.google.com/go/longrunning v0.6.2/go.mod h1:k/vIs83RN4bE3YCswdXC5PFfWVILjm3hpEUlSko4PiI=\ncloud.google.com/go/monitoring v1.21.2 h1:FChwVtClH19E7pJ+e0xUhJPGksctZNVOk2UhMmblmdU=\ncloud.google.com/go/monitoring v1.21.2/go.mod h1:hS3pXvaG8KgWTSz+dAdyzPrGUYmi2Q+WFX8g2hqVEZU=\ncloud.google.com/go/pubsub v1.0.1/go.mod h1:R0Gpsv3s54REJCy4fxDixWD93lHJMoZTyQ2kNxGRt3I=\ncloud.google.com/go/pubsub v1.1.0/go.mod h1:EwwdRX2sKPjnvnqCa270oGRyludottCI76h+R3AArQw=\ncloud.google.com/go/pubsub v1.2.0/go.mod h1:jhfEVHT8odbXTkndysNHCcx0awwzvfOlguIAii9o8iA=\ncloud.google.com/go/pubsub v1.3.1/go.mod h1:i+ucay31+CNRpDW4Lu78I4xXG+O1r/MAHgjpRVR+TSU=\ncloud.google.com/go/pubsub v1.5.0/go.mod h1:ZEwJccE3z93Z2HWvstpri00jOg7oO4UZDtKhwDwqF0w=\ncloud.google.com/go/pubsub v1.10.3/go.mod h1:FUcc28GpGxxACoklPsE1sCtbkY4Ix+ro7yvw+h82Jn4=\ncloud.google.com/go/spanner v1.7.0/go.mod h1:sd3K2gZ9Fd0vMPLXzeCrF6fq4i63Q7aTLW/lBIfBkIk=\ncloud.google.com/go/storage v1.0.0/go.mod h1:IhtSnM/ZTZV8YYJWCY8RULGVqBDmpoyjwiyrjsg+URw=\ncloud.google.com/go/storage v1.5.0/go.mod h1:tpKbwo567HUNpVclU5sGELwQWBDZ8gh0ZeosJ0Rtdos=\ncloud.google.com/go/storage v1.6.0/go.mod h1:N7U0C8pVQ/+NIKOBQyamJIeKQKkZ+mxpohlUTyfDhBk=\ncloud.google.com/go/storage v1.8.0/go.mod h1:Wv1Oy7z6Yz3DshWRJFhqM/UCfaWIRTdp0RXyy7KQOVs=\ncloud.google.com/go/storage v1.10.0/go.mod h1:FLPqc6j+Ki4BU591ie1oL6qBQGu2Bl/tZ9ullr3+Kg0=\ncloud.google.com/go/storage v1.14.0/go.mod h1:GrKmX003DSIwi9o29oFT7YDnHYwZoctc3fOKtUw0Xmo=\ncloud.google.com/go/storage v1.15.0/go.mod h1:mjjQMoxxyGH7Jr8K5qrx6N2O0AHsczI61sMNn03GIZI=\ncloud.google.com/go/storage v1.49.0 h1:zenOPBOWHCnojRd9aJZAyQXBYqkJkdQS42dxL55CIMw=\ncloud.google.com/go/storage v1.49.0/go.mod h1:k1eHhhpLvrPjVGfo0mOUPEJ4Y2+a/Hv5PiwehZI9qGU=\ncloud.google.com/go/trace v1.11.2 h1:4ZmaBdL8Ng/ajrgKqY5jfvzqMXbrDcBsUGXOT9aqTtI=\ncloud.google.com/go/trace v1.11.2/go.mod h1:bn7OwXd4pd5rFuAnTrzBuoZ4ax2XQeG3qNgYmfCy0Io=\ncode.gitea.io/sdk/gitea v0.14.0 h1:m4J352I3p9+bmJUfS+g0odeQzBY/5OXP91Gv6D4fnJ0=\ncode.gitea.io/sdk/gitea v0.14.0/go.mod h1:89WiyOX1KEcvjP66sRHdu0RafojGo60bT9UqW17VbWs=\ncontrib.go.opencensus.io/exporter/aws v0.0.0-20200617204711-c478e41e60e9/go.mod h1:uu1P0UCM/6RbsMrgPa98ll8ZcHM858i/AD06a9aLRCA=\ncontrib.go.opencensus.io/exporter/stackdriver v0.13.4/go.mod h1:aXENhDJ1Y4lIg4EUaVTwzvYETVNZk10Pu26tevFKLUc=\ncontrib.go.opencensus.io/exporter/stackdriver v0.13.5/go.mod h1:aXENhDJ1Y4lIg4EUaVTwzvYETVNZk10Pu26tevFKLUc=\ncontrib.go.opencensus.io/integrations/ocsql v0.1.7/go.mod h1:8DsSdjz3F+APR+0z0WkU1aRorQCFfRxvqjUUPMbF3fE=\ndmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=\ngithub.com/AlekSi/pointer v1.1.0/go.mod h1:y7BvfRI3wXPWKXEBhU71nbnIEEZX0QTSB2Bj48UJIZE=\ngithub.com/AlekSi/pointer v1.2.0 h1:glcy/gc4h8HnG2Z3ZECSzZ1IX1x2JxRVuDzaJwQE0+w=\ngithub.com/AlekSi/pointer v1.2.0/go.mod h1:gZGfd3dpW4vEc/UlyfKKi1roIqcCgwOIvb0tSNSBle0=\ngithub.com/Antonboom/errname v0.1.6 h1:LzIJZlyLOCSu51o3/t2n9Ck7PcoP9wdbrdaW6J8fX24=\ngithub.com/Antonboom/errname v0.1.6/go.mod h1:7lz79JAnuoMNDAWE9MeeIr1/c/VpSUWatBv2FH9NYpI=\ngithub.com/Antonboom/nilnil v0.1.1 h1:PHhrh5ANKFWRBh7TdYmyyq2gyT2lotnvFvvFbylF81Q=\ngithub.com/Antonboom/nilnil v0.1.1/go.mod h1:L1jBqoWM7AOeTD+tSquifKSesRHs4ZdaxvZR+xdJEaI=\ngithub.com/Azure/azure-amqp-common-go/v3 v3.1.0/go.mod h1:PBIGdzcO1teYoufTKMcGibdKaYZv4avS+O6LNIp8bq0=\ngithub.com/Azure/azure-pipeline-go v0.2.3 h1:7U9HBg1JFK3jHl5qmo4CTZKFTVgMwdFHMVtCdfBE21U=\ngithub.com/Azure/azure-pipeline-go v0.2.3/go.mod h1:x841ezTBIMG6O3lAcl8ATHnsOPVl2bqk7S3ta6S6u4k=\ngithub.com/Azure/azure-sdk-for-go v51.1.0+incompatible/go.mod h1:9XXNKU+eRnpl9moKnB4QOLf1HestfXbmab5FXxiDBjc=\ngithub.com/Azure/azure-sdk-for-go v54.0.0+incompatible h1:Bq3L9LF0DHCexlT0fccwxgrOMfjHx8LGz+d+L7gGQv4=\ngithub.com/Azure/azure-sdk-for-go v54.0.0+incompatible/go.mod h1:9XXNKU+eRnpl9moKnB4QOLf1HestfXbmab5FXxiDBjc=\ngithub.com/Azure/azure-service-bus-go v0.10.11/go.mod h1:AWw9eTTWZVZyvgpPahD1ybz3a8/vT3GsJDS8KYex55U=\ngithub.com/Azure/azure-storage-blob-go v0.13.0 h1:lgWHvFh+UYBNVQLFHXkvul2f6yOPA9PIH82RTG2cSwc=\ngithub.com/Azure/azure-storage-blob-go v0.13.0/go.mod h1:pA9kNqtjUeQF2zOSu4s//nUdBD+e64lEuc4sVnuOfNs=\ngithub.com/Azure/go-amqp v0.13.0/go.mod h1:qj+o8xPCz9tMSbQ83Vp8boHahuRDl5mkNHyt1xlxUTs=\ngithub.com/Azure/go-amqp v0.13.4/go.mod h1:wbpCKA8tR5MLgRyIu+bb+S6ECdIDdYJ0NlpFE9xsBPI=\ngithub.com/Azure/go-amqp v0.13.7/go.mod h1:wbpCKA8tR5MLgRyIu+bb+S6ECdIDdYJ0NlpFE9xsBPI=\ngithub.com/Azure/go-autorest v14.2.0+incompatible h1:V5VMDjClD3GiElqLWO7mz2MxNAK/vTfRHdAubSIPRgs=\ngithub.com/Azure/go-autorest v14.2.0+incompatible/go.mod h1:r+4oMnoxhatjLLJ6zxSWATqVooLgysK6ZNox3g/xq24=\ngithub.com/Azure/go-autorest/autorest v0.11.3/go.mod h1:JFgpikqFJ/MleTTxwepExTKnFUKKszPS8UavbQYUMuw=\ngithub.com/Azure/go-autorest/autorest v0.11.17/go.mod h1:eipySxLmqSyC5s5k1CLupqet0PSENBEDP93LQ9a8QYw=\ngithub.com/Azure/go-autorest/autorest v0.11.18 h1:90Y4srNYrwOtAgVo3ndrQkTYn6kf1Eg/AjTFJ8Is2aM=\ngithub.com/Azure/go-autorest/autorest v0.11.18/go.mod h1:dSiJPy22c3u0OtOKDNttNgqpNFY/GeWa7GH/Pz56QRA=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.0/go.mod h1:/c022QCutn2P7uY+/oQWWNcK9YU+MH96NgK+jErpbcg=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.2/go.mod h1:/3SMAM86bP6wC9Ev35peQDUeqFZBMH07vvUOmg4z/fE=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.5/go.mod h1:B7KF7jKIeC9Mct5spmyCB/A8CG/sEz1vwIRGv/bbw7A=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.11/go.mod h1:nBKAnTomx8gDtl+3ZCJv2v0KACFHWTB2drffI1B68Pk=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.13 h1:Mp5hbtOePIzM8pJVRa3YLrWWmZtoxRXqUEzCfJt3+/Q=\ngithub.com/Azure/go-autorest/autorest/adal v0.9.13/go.mod h1:W/MM4U6nLxnIskrw4UwWzlHfGjwUS50aOsc/I3yuU8M=\ngithub.com/Azure/go-autorest/autorest/azure/auth v0.5.7 h1:8DQB8yl7aLQuP+nuR5e2RO6454OvFlSTXXaNHshc16s=\ngithub.com/Azure/go-autorest/autorest/azure/auth v0.5.7/go.mod h1:AkzUsqkrdmNhfP2i54HqINVQopw0CLDnvHpJ88Zz1eI=\ngithub.com/Azure/go-autorest/autorest/azure/cli v0.4.2 h1:dMOmEJfkLKW/7JsokJqkyoYSgmR08hi9KrhjZb+JALY=\ngithub.com/Azure/go-autorest/autorest/azure/cli v0.4.2/go.mod h1:7qkJkT+j6b+hIpzMOwPChJhTqS8VbsqqgULzMNRugoM=\ngithub.com/Azure/go-autorest/autorest/date v0.3.0 h1:7gUk1U5M/CQbp9WoqinNzJar+8KY+LPI6wiWrP/myHw=\ngithub.com/Azure/go-autorest/autorest/date v0.3.0/go.mod h1:BI0uouVdmngYNUzGWeSYnokU+TrmwEsOqdt8Y6sso74=\ngithub.com/Azure/go-autorest/autorest/mocks v0.4.0/go.mod h1:LTp+uSrOhSkaKrUy935gNZuuIPPVsHlr9DSOxSayd+k=\ngithub.com/Azure/go-autorest/autorest/mocks v0.4.1 h1:K0laFcLE6VLTOwNgSxaGbUcLPuGXlNkbVvq4cW4nIHk=\ngithub.com/Azure/go-autorest/autorest/mocks v0.4.1/go.mod h1:LTp+uSrOhSkaKrUy935gNZuuIPPVsHlr9DSOxSayd+k=\ngithub.com/Azure/go-autorest/autorest/to v0.4.0 h1:oXVqrxakqqV1UZdSazDOPOLvOIz+XA683u8EctwboHk=\ngithub.com/Azure/go-autorest/autorest/to v0.4.0/go.mod h1:fE8iZBn7LQR7zH/9XU2NcPR4o9jEImooCeWJcYV/zLE=\ngithub.com/Azure/go-autorest/autorest/validation v0.3.1 h1:AgyqjAd94fwNAoTjl/WQXg4VvFeRFpO+UhNyRXqF1ac=\ngithub.com/Azure/go-autorest/autorest/validation v0.3.1/go.mod h1:yhLgjC0Wda5DYXl6JAsWyUe4KVNffhoDhG0zVzUMo3E=\ngithub.com/Azure/go-autorest/logger v0.2.0/go.mod h1:T9E3cAhj2VqvPOtCYAvby9aBXkZmbF5NWuPV8+WeEW8=\ngithub.com/Azure/go-autorest/logger v0.2.1 h1:IG7i4p/mDa2Ce4TRyAO8IHnVhAVF3RFU+ZtXWSmf4Tg=\ngithub.com/Azure/go-autorest/logger v0.2.1/go.mod h1:T9E3cAhj2VqvPOtCYAvby9aBXkZmbF5NWuPV8+WeEW8=\ngithub.com/Azure/go-autorest/tracing v0.6.0 h1:TYi4+3m5t6K48TGI9AUdb+IzbnSxvnvUMfuitfgcfuo=\ngithub.com/Azure/go-autorest/tracing v0.6.0/go.mod h1:+vhtPC754Xsa23ID7GlGsrdKBpUA79WCAKPPZVC2DeU=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/BurntSushi/toml v0.4.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/BurntSushi/toml v1.2.1 h1:9F2/+DoOYIOksmaJFPw1tGFy1eDnIJXg+UHjuD8lTak=\ngithub.com/BurntSushi/toml v1.2.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=\ngithub.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=\ngithub.com/DataDog/datadog-go v3.2.0+incompatible/go.mod h1:LButxg5PwREeZtORoXG3tL4fMGNddJ+vMq1mwgfaqoQ=\ngithub.com/DataDog/zstd v1.4.5 h1:EndNeuB0l9syBZhut0wns3gV1hL8zX8LIu6ZiVHWLIQ=\ngithub.com/DataDog/zstd v1.4.5/go.mod h1:1jcaCB/ufaK+sKp1NBhlGmpz41jOoPQ35bpF36t7BBo=\ngithub.com/Djarvur/go-err113 v0.0.0-20210108212216-aea10b59be24 h1:sHglBQTwgx+rWPdisA5ynNEsoARbiCBOyGcJM4/OzsM=\ngithub.com/Djarvur/go-err113 v0.0.0-20210108212216-aea10b59be24/go.mod h1:4UJr5HIiMZrwgkSPdsjy2uOQExX/WEILpIrO9UPGuXs=\ngithub.com/GaijinEntertainment/go-exhaustruct/v2 v2.1.0 h1:LAPPhJ4KR5Z8aKVZF5S48csJkxL5RMKmE/98fMs1u5M=\ngithub.com/GaijinEntertainment/go-exhaustruct/v2 v2.1.0/go.mod h1:LGOGuvEgCfCQsy3JF2tRmpGDpzA53iZfyGEWSPwQ6/4=\ngithub.com/GoogleCloudPlatform/cloudsql-proxy v1.22.0/go.mod h1:mAm5O/zik2RFmcpigNjg6nMotDL8ZXJaxKzgGVcSMFA=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.25.0 h1:3c8yed4lgqTt+oTQ+JNMDo+F4xprBf+O/il4ZC0nRLw=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.25.0/go.mod h1:obipzmGjfSjam60XLwGfqUkJsfiheAl+TUjG+4yzyPM=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.48.1 h1:UQ0AhxogsIRZDkElkblfnwjc3IaltCm2HUMvezQaL7s=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.48.1/go.mod h1:jyqM3eLpJ3IbIFDTKVz2rF9T/xWGW0rIriGwnz8l9Tk=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.48.1 h1:oTX4vsorBZo/Zdum6OKPA4o7544hm6smoRv1QjpTwGo=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.48.1/go.mod h1:0wEl7vrAD8mehJyohS9HZy+WyEOaQO2mJx86Cvh93kM=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.48.1 h1:8nn+rsCvTq9axyEh382S0PFLBeaFwNsT43IrPWzctRU=\ngithub.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.48.1/go.mod h1:viRWSEhtMZqz1rhwmOVKkWl6SwmVowfL9O2YR5gI2PE=\ngithub.com/Masterminds/goutils v1.1.0/go.mod h1:8cTjp+g8YejhMuvIA5y2vz3BpJxksy863GQaJW2MFNU=\ngithub.com/Masterminds/goutils v1.1.1 h1:5nUrii3FMTL5diU80unEVvNevw1nH4+ZV4DSLVJLSYI=\ngithub.com/Masterminds/goutils v1.1.1/go.mod h1:8cTjp+g8YejhMuvIA5y2vz3BpJxksy863GQaJW2MFNU=\ngithub.com/Masterminds/semver v1.4.2/go.mod h1:MB6lktGJrhw8PrUyiEoblNEGEQ+RzHPF078ddwwvV3Y=\ngithub.com/Masterminds/semver v1.5.0 h1:H65muMkzWKEuNDnfl9d70GUjFniHKHRbFPGBuZ3QEww=\ngithub.com/Masterminds/semver v1.5.0/go.mod h1:MB6lktGJrhw8PrUyiEoblNEGEQ+RzHPF078ddwwvV3Y=\ngithub.com/Masterminds/semver/v3 v3.1.0/go.mod h1:VPu/7SZ7ePZ3QOrcuXROw5FAcLl4a0cBrbBpGY/8hQs=\ngithub.com/Masterminds/semver/v3 v3.1.1/go.mod h1:VPu/7SZ7ePZ3QOrcuXROw5FAcLl4a0cBrbBpGY/8hQs=\ngithub.com/Masterminds/semver/v3 v3.2.1 h1:RN9w6+7QoMeJVGyfmbcgs28Br8cvmnucEXnY0rYXWg0=\ngithub.com/Masterminds/semver/v3 v3.2.1/go.mod h1:qvl/7zhW3nngYb5+80sSMF+FG2BjYrf8m9wsX0PNOMQ=\ngithub.com/Masterminds/sprig v2.15.0+incompatible/go.mod h1:y6hNFY5UBTIWBxnzTeuNhlNS5hqE0NB0E6fgfo2Br3o=\ngithub.com/Masterminds/sprig v2.22.0+incompatible h1:z4yfnGrZ7netVz+0EDJ0Wi+5VZCSYp4Z0m2dk6cEM60=\ngithub.com/Masterminds/sprig v2.22.0+incompatible/go.mod h1:y6hNFY5UBTIWBxnzTeuNhlNS5hqE0NB0E6fgfo2Br3o=\ngithub.com/Masterminds/sprig/v3 v3.2.2 h1:17jRggJu518dr3QaafizSXOjKYp94wKfABxUmyxvxX8=\ngithub.com/Masterminds/sprig/v3 v3.2.2/go.mod h1:UoaO7Yp8KlPnJIYWTFkMaqPUYKTfGFPhxNuwnnxkKlk=\ngithub.com/Microsoft/go-winio v0.4.14/go.mod h1:qXqCSQ3Xa7+6tgxaGTIe4Kpcdsi+P8jBhyzoq1bpyYA=\ngithub.com/Microsoft/go-winio v0.4.15/go.mod h1:tTuCMEN+UleMWgg9dVx4Hu52b1bJo+59jBh3ajtinzw=\ngithub.com/Microsoft/go-winio v0.4.16/go.mod h1:XB6nPKklQyQ7GC9LdcBEcBl8PF76WugXOPRXwdLnMv0=\ngithub.com/Microsoft/go-winio v0.5.0/go.mod h1:JPGBdM1cNvN/6ISo+n8V5iA4v8pBzdOpzfwIujj1a84=\ngithub.com/Microsoft/go-winio v0.5.1 h1:aPJp2QD7OOrhO5tQXqQoGSJc+DjDtWTGLOmNyAm6FgY=\ngithub.com/Microsoft/go-winio v0.5.1/go.mod h1:JPGBdM1cNvN/6ISo+n8V5iA4v8pBzdOpzfwIujj1a84=\ngithub.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=\ngithub.com/OpenPeeDeeP/depguard v1.1.0 h1:pjK9nLPS1FwQYGGpPxoMYpe7qACHOhAWQMQzV71i49o=\ngithub.com/OpenPeeDeeP/depguard v1.1.0/go.mod h1:JtAMzWkmFEzDPyAd+W0NHl1lvpQKTvT9jnRVsohBKpc=\ngithub.com/ProtonMail/go-crypto v0.0.0-20210329181949-3900d675f39b/go.mod h1:HTM9X7e9oLwn7RiqLG0UVwVRJenLs3wN+tQ0NPAfwMQ=\ngithub.com/ProtonMail/go-crypto v0.0.0-20210408094314-bf0c5240ed99/go.mod h1:HTM9X7e9oLwn7RiqLG0UVwVRJenLs3wN+tQ0NPAfwMQ=\ngithub.com/ProtonMail/go-crypto v0.0.0-20210512092938-c05353c2d58c h1:bNpaLLv2Y4kslsdkdCwAYu8Bak1aGVtxwi8Z/wy4Yuo=\ngithub.com/ProtonMail/go-crypto v0.0.0-20210512092938-c05353c2d58c/go.mod h1:z4/9nQmJSSwwds7ejkxaJwO37dru3geImFUdJlaLzQo=\ngithub.com/ProtonMail/go-mime v0.0.0-20190923161245-9b5a4261663a h1:W6RrgN/sTxg1msqzFFb+G80MFmpjMw61IU+slm+wln4=\ngithub.com/ProtonMail/go-mime v0.0.0-20190923161245-9b5a4261663a/go.mod h1:NYt+V3/4rEeDuaev/zw1zCq8uqVEuPHzDPo3OZrlGJ4=\ngithub.com/ProtonMail/gopenpgp/v2 v2.1.8/go.mod h1:mjMvRMlOlBhNuaa3z0xOmEgAkba/Mu1Z8uWwYj4/6Ws=\ngithub.com/ProtonMail/gopenpgp/v2 v2.2.2 h1:u2m7xt+CZWj88qK1UUNBoXeJCFJwJCZ/Ff4ymGoxEXs=\ngithub.com/ProtonMail/gopenpgp/v2 v2.2.2/go.mod h1:ajUlBGvxMH1UBZnaYO3d1FSVzjiC6kK9XlZYGiDCvpM=\ngithub.com/alcortesm/tgz v0.0.0-20161220082320-9c5fe88206d7 h1:uSoVVbwJiQipAclBbw+8quDsfcvFjOpI5iCf4p/cqCs=\ngithub.com/alcortesm/tgz v0.0.0-20161220082320-9c5fe88206d7/go.mod h1:6zEj6s6u/ghQa61ZWa/C2Aw3RkjiTBOix7dkqa1VLIs=\ngithub.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=\ngithub.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d/go.mod h1:rBZYJk541a8SKzHPHnH3zbiI+7dagKZ0cgpgrD7Fyho=\ngithub.com/alexkohler/prealloc v1.0.0 h1:Hbq0/3fJPQhNkN0dR95AVrr6R7tou91y0uHG5pOcUuw=\ngithub.com/alexkohler/prealloc v1.0.0/go.mod h1:VetnK3dIgFBBKmg0YnD9F9x6Icjd+9cvfHR56wJVlKE=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 h1:kFOfPq6dUM1hTo4JG6LR5AXSUEsOjtdm0kw0FtQtMJA=\ngithub.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239/go.mod h1:2FmKhYUyUczH0OGQWaF5ceTx0UBShxjsH6f8oGKYe2c=\ngithub.com/antihax/optional v0.0.0-20180407024304-ca021399b1a6/go.mod h1:V8iCPQYkqmusNa815XgQio277wI47sdRh1dUOLdyC6Q=\ngithub.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY=\ngithub.com/aokoli/goutils v1.0.1/go.mod h1:SijmP0QR8LtwsmDs8Yii5Z/S4trXFGFC2oO5g9DP+DQ=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=\ngithub.com/apex/log v1.9.0 h1:FHtw/xuaM8AgmvDDTI9fiwoAL25Sq2cxojnZICUU8l0=\ngithub.com/apex/log v1.9.0/go.mod h1:m82fZlWIuiWzWP04XCTXmnX0xRkYYbCdYn8jbJeLBEA=\ngithub.com/apex/logs v1.0.0/go.mod h1:XzxuLZ5myVHDy9SAmYpamKKRNApGj54PfYLcFrXqDwo=\ngithub.com/aphistic/golf v0.0.0-20180712155816-02c07f170c5a/go.mod h1:3NqKYiepwy8kCu4PNA+aP7WUV72eXWJeP9/r3/K9aLE=\ngithub.com/aphistic/sweet v0.2.0/go.mod h1:fWDlIh/isSE9n6EPsRmC0det+whmX6dJid3stzu0Xys=\ngithub.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e/go.mod h1:3U/XgcO3hCbHZ8TKRvWD2dDTCfh9M9ya+I9JpbB7O8o=\ngithub.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6/go.mod h1:grANhF5doyWs3UAsr3K4I6qtAmlQcZDesFNEHPZAzj8=\ngithub.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da/go.mod h1:Q73ZrmVTwzkszR9V5SSuryQ31EELlFMUz1kKyl939pY=\ngithub.com/armon/go-metrics v0.3.10/go.mod h1:4O98XIr/9W0sxpJ8UaYkvjk10Iff7SnFrb4QAOwNTFc=\ngithub.com/armon/go-radix v0.0.0-20180808171621-7fddfc383310/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=\ngithub.com/armon/go-radix v1.0.0/go.mod h1:ufUuZ+zHj4x4TnLV4JWEpy2hxWSpsRywHrMgIH9cCH8=\ngithub.com/armon/go-socks5 v0.0.0-20160902184237-e75332964ef5 h1:0CwZNZbxp69SHPdPJAN/hZIm0C4OItdklCFmMRWYpio=\ngithub.com/armon/go-socks5 v0.0.0-20160902184237-e75332964ef5/go.mod h1:wHh0iHkYZB8zMSxRWpUBQtwG5a7fFgvEO+odwuTv2gs=\ngithub.com/ashanbrown/forbidigo v1.3.0 h1:VkYIwb/xxdireGAdJNZoo24O4lmnEWkactplBlWTShc=\ngithub.com/ashanbrown/forbidigo v1.3.0/go.mod h1:vVW7PEdqEFqapJe95xHkTfB1+XvZXBFg8t0sG2FIxmI=\ngithub.com/ashanbrown/makezero v1.1.1 h1:iCQ87C0V0vSyO+M9E/FZYbu65auqH0lnsOkf5FcB28s=\ngithub.com/ashanbrown/makezero v1.1.1/go.mod h1:i1bJLCRSCHOcOa9Y6MyF2FTfMZMFdHvxKHxgO5Z1axI=\ngithub.com/aws/aws-sdk-go v1.15.27/go.mod h1:mFuSZ37Z9YOHbQEwBWztmVzqXrEkub65tZoCYDt7FT0=\ngithub.com/aws/aws-sdk-go v1.20.6/go.mod h1:KmX6BPdI08NWTb3/sm4ZGu5ShLoqVDhKgpiN924inxo=\ngithub.com/aws/aws-sdk-go v1.23.20/go.mod h1:KmX6BPdI08NWTb3/sm4ZGu5ShLoqVDhKgpiN924inxo=\ngithub.com/aws/aws-sdk-go v1.25.37/go.mod h1:KmX6BPdI08NWTb3/sm4ZGu5ShLoqVDhKgpiN924inxo=\ngithub.com/aws/aws-sdk-go v1.36.30/go.mod h1:hcU610XS61/+aQV88ixoOzUoG7v3b31pl2zKMmprdro=\ngithub.com/aws/aws-sdk-go v1.38.35 h1:7AlAO0FC+8nFjxiGKEmq0QLpiA8/XFr6eIxgRTwkdTg=\ngithub.com/aws/aws-sdk-go v1.38.35/go.mod h1:hcU610XS61/+aQV88ixoOzUoG7v3b31pl2zKMmprdro=\ngithub.com/aybabtme/rgbterm v0.0.0-20170906152045-cc83f3b3ce59/go.mod h1:q/89r3U2H7sSsE2t6Kca0lfwTK8JdoNGS/yzM/4iH5I=\ngithub.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=\ngithub.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=\ngithub.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=\ngithub.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=\ngithub.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=\ngithub.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=\ngithub.com/bgentry/speakeasy v0.1.0/go.mod h1:+zsyZBPWlz7T6j88CTgSN5bM796AkVf0kBD4zp0CCIs=\ngithub.com/bketelsen/crypt v0.0.3-0.20200106085610-5cbc8cc4026c/go.mod h1:MKsuJmJgSg28kpZDP6UIiPt0e0Oz0kqKNGyRaWEPv84=\ngithub.com/bkielbasa/cyclop v1.2.0 h1:7Jmnh0yL2DjKfw28p86YTd/B4lRGcNuu12sKE35sM7A=\ngithub.com/bkielbasa/cyclop v1.2.0/go.mod h1:qOI0yy6A7dYC4Zgsa72Ppm9kONl0RoIlPbzot9mhmeI=\ngithub.com/blakesmith/ar v0.0.0-20190502131153-809d4375e1fb h1:m935MPodAbYS46DG4pJSv7WO+VECIWUQ7OJYSoTrMh4=\ngithub.com/blakesmith/ar v0.0.0-20190502131153-809d4375e1fb/go.mod h1:PkYb9DJNAwrSvRx5DYA+gUcOIgTGVMNkfSCbZM8cWpI=\ngithub.com/blizzy78/varnamelen v0.8.0 h1:oqSblyuQvFsW1hbBHh1zfwrKe3kcSj0rnXkKzsQ089M=\ngithub.com/blizzy78/varnamelen v0.8.0/go.mod h1:V9TzQZ4fLJ1DSrjVDfl89H7aMnTvKkApdHeyESmyR7k=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b/go.mod h1:ac9efd0D1fsDb3EJvhqgXRbFx7bs2wqZ10HQPeU8U/Q=\ngithub.com/bombsimon/wsl/v3 v3.3.0 h1:Mka/+kRLoQJq7g2rggtgQsjuI/K5Efd87WX96EWFxjM=\ngithub.com/bombsimon/wsl/v3 v3.3.0/go.mod h1:st10JtZYLE4D5sC7b8xV4zTKZwAQjCH/Hy2Pm1FNZIc=\ngithub.com/breml/bidichk v0.2.3 h1:qe6ggxpTfA8E75hdjWPZ581sY3a2lnl0IRxLQFelECI=\ngithub.com/breml/bidichk v0.2.3/go.mod h1:8u2C6DnAy0g2cEq+k/A2+tr9O1s+vHGxWn0LTc70T2A=\ngithub.com/breml/errchkjson v0.3.0 h1:YdDqhfqMT+I1vIxPSas44P+9Z9HzJwCeAzjB8PxP1xw=\ngithub.com/breml/errchkjson v0.3.0/go.mod h1:9Cogkyv9gcT8HREpzi3TiqBxCqDzo8awa92zSDFcofU=\ngithub.com/butuzov/ireturn v0.1.1 h1:QvrO2QF2+/Cx1WA/vETCIYBKtRjc30vesdoPUNo1EbY=\ngithub.com/butuzov/ireturn v0.1.1/go.mod h1:Wh6Zl3IMtTpaIKbmwzqi6olnM9ptYQxxVacMsOEFPoc=\ngithub.com/caarlos0/ctrlc v1.0.0 h1:2DtF8GSIcajgffDFJzyG15vO+1PuBWOMUdFut7NnXhw=\ngithub.com/caarlos0/ctrlc v1.0.0/go.mod h1:CdXpj4rmq0q/1Eb44M9zi2nKB0QraNKuRGYGrrHhcQw=\ngithub.com/caarlos0/env/v6 v6.6.2 h1:BypLXDWQTA32rS4UM7pBz+/0BOuvs6C7LSeQAxMwyvI=\ngithub.com/caarlos0/env/v6 v6.6.2/go.mod h1:P0BVSgU9zfkxfSpFUs6KsO3uWR4k3Ac0P66ibAGTybM=\ngithub.com/caarlos0/go-rpmutils v0.2.1-0.20211112020245-2cd62ff89b11 h1:IRrDwVlWQr6kS1U8/EtyA1+EHcc4yl8pndcqXWrEamg=\ngithub.com/caarlos0/go-rpmutils v0.2.1-0.20211112020245-2cd62ff89b11/go.mod h1:je2KZ+LxaCNvCoKg32jtOIULcFogJKcL1ZWUaIBjKj0=\ngithub.com/caarlos0/go-shellwords v1.0.12 h1:HWrUnu6lGbWfrDcFiHcZiwOLzHWjjrPVehULaTFgPp8=\ngithub.com/caarlos0/go-shellwords v1.0.12/go.mod h1:bYeeX1GrTLPl5cAMYEzdm272qdsQAZiaHgeF0KTk1Gw=\ngithub.com/caarlos0/testfs v0.4.3/go.mod h1:bRN55zgG4XCUVVHZCeU+/Tz1Q6AxEJOEJTliBy+1DMk=\ngithub.com/caarlos0/testfs v0.4.4 h1:3PHvzHi5Lt+g332CiShwS8ogTgS3HjrmzZxCm6JCDr8=\ngithub.com/caarlos0/testfs v0.4.4/go.mod h1:bRN55zgG4XCUVVHZCeU+/Tz1Q6AxEJOEJTliBy+1DMk=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748/go.mod h1:l/bIBLeOl9eX+wxJAzxS4TveKRtAqlyDpHjhkfO0MEI=\ngithub.com/campoy/unique v0.0.0-20180121183637-88950e537e7e h1:V9a67dfYqPLAvzk5hMQOXYJlZ4SLIXgyKIE+ZiHzgGQ=\ngithub.com/campoy/unique v0.0.0-20180121183637-88950e537e7e/go.mod h1:9IOqJGCPMSc6E5ydlp5NIonxObaeu/Iub/X03EKPVYo=\ngithub.com/cavaliercoder/go-cpio v0.0.0-20180626203310-925f9528c45e/go.mod h1:oDpT4efm8tSYHXV5tHSdRvBet/b/QzxZ+XyyPehvm3A=\ngithub.com/cavaliergopher/cpio v1.0.1 h1:KQFSeKmZhv0cr+kawA3a0xTQCU4QxXF1vhU7P7av2KM=\ngithub.com/cavaliergopher/cpio v1.0.1/go.mod h1:pBdaqQjnvXxdS/6CvNDwIANIFSP0xRKI16PX4xejRQc=\ngithub.com/cenkalti/backoff v2.1.1+incompatible h1:tKJnvO2kl0zmb/jA5UKAt4VoEVw1qxKWjE/Bpp46npY=\ngithub.com/cenkalti/backoff v2.1.1+incompatible/go.mod h1:90ReRw6GdpyfrHakVjL/QHaoyV4aDUVVkXQJJJ3NXXM=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/census-instrumentation/opencensus-proto v0.3.0/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/census-instrumentation/opencensus-proto v0.4.1 h1:iKLQ0xPNFxR/2hzXZMrBo8f1j86j5WHzznCCQxV/b8g=\ngithub.com/census-instrumentation/opencensus-proto v0.4.1/go.mod h1:4T9NM4+4Vw91VeyqjLS6ao50K5bOcLKN6Q42XnYaRYw=\ngithub.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=\ngithub.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=\ngithub.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/charithe/durationcheck v0.0.9 h1:mPP4ucLrf/rKZiIG/a9IPXHGlh8p4CzgpyTy6EEutYk=\ngithub.com/charithe/durationcheck v0.0.9/go.mod h1:SSbRIBVfMjCi/kEB6K65XEA83D6prSM8ap1UCpNKtgg=\ngithub.com/chavacava/garif v0.0.0-20230227094218-b8c73b2037b8 h1:W9o46d2kbNL06lq7UNDPV0zYLzkrde/bjIqO02eoll0=\ngithub.com/chavacava/garif v0.0.0-20230227094218-b8c73b2037b8/go.mod h1:gakxgyXaaPkxvLw1XQxNGK4I37ys9iBRzNUx/B7pUCo=\ngithub.com/chigopher/pathlib v0.19.1 h1:RoLlUJc0CqBGwq239cilyhxPNLXTK+HXoASGyGznx5A=\ngithub.com/chigopher/pathlib v0.19.1/go.mod h1:tzC1dZLW8o33UQpWkNkhvPwL5n4yyFRFm/jL1YGWFvY=\ngithub.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=\ngithub.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=\ngithub.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=\ngithub.com/circonus-labs/circonus-gometrics v2.3.1+incompatible/go.mod h1:nmEj6Dob7S7YxXgwXpfOuvO54S+tGdZdw9fuRZt25Ag=\ngithub.com/circonus-labs/circonusllhist v0.1.3/go.mod h1:kMXHVDlOchFAehlya5ePtbp5jckzBHf4XRpQvBOLI+I=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=\ngithub.com/cncf/udpa/go v0.0.0-20200629203442-efcf912fb354/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk=\ngithub.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk=\ngithub.com/cncf/udpa/go v0.0.0-20210930031921-04548b0d99d4/go.mod h1:6pvJx4me5XPnfI9Z40ddWsdw2W/uZgQLFXToKeRcDiI=\ngithub.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210805033703-aa0b78936158/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20210922020428-25de7278fc84/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20211011173535-cb28da3451f1/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cncf/xds/go v0.0.0-20240905190251-b4127c9b8d78 h1:QVw89YDxXxEe+l8gU8ETbOasdwEV+avkR75ZzsVV9WI=\ngithub.com/cncf/xds/go v0.0.0-20240905190251-b4127c9b8d78/go.mod h1:W+zGtBO5Y1IgJhy4+A9GOqVhqLpfZi+vwmdNXUehLA8=\ngithub.com/cockroachdb/datadriven v0.0.0-20190809214429-80d97fb3cbaa/go.mod h1:zn76sxSg3SzpJ0PPJaLDCu+Bu0Lg3sKTORVIj19EIF8=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI=\ngithub.com/coreos/bbolt v1.3.2/go.mod h1:iRUV2dpdMOn7Bo10OQBFzIJO9kkE559Wcmn+qkEiiKk=\ngithub.com/coreos/etcd v3.3.10+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=\ngithub.com/coreos/etcd v3.3.13+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=\ngithub.com/coreos/go-etcd v2.0.0+incompatible/go.mod h1:Jez6KQU2B/sWsbdaef3ED8NzMklzPG4d5KIOhIy30Tk=\ngithub.com/coreos/go-semver v0.2.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=\ngithub.com/coreos/go-semver v0.3.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=\ngithub.com/coreos/go-systemd v0.0.0-20180511133405-39ca1b05acc7/go.mod h1:F5haX7vjVVG0kc13fIWeqUViNPyEJxv/OmvnBo0Yme4=\ngithub.com/coreos/go-systemd v0.0.0-20190321100706-95778dfbb74e/go.mod h1:F5haX7vjVVG0kc13fIWeqUViNPyEJxv/OmvnBo0Yme4=\ngithub.com/coreos/go-systemd v0.0.0-20190620071333-e64a0ec8b42a/go.mod h1:F5haX7vjVVG0kc13fIWeqUViNPyEJxv/OmvnBo0Yme4=\ngithub.com/coreos/go-systemd/v22 v22.3.1/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=\ngithub.com/coreos/go-systemd/v22 v22.3.2/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=\ngithub.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=\ngithub.com/coreos/pkg v0.0.0-20160727233714-3ac0863d7acf/go.mod h1:E3G3o1h8I7cfcXa63jLwjI0eiQQMgzzUDFVpN/nH/eA=\ngithub.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f/go.mod h1:E3G3o1h8I7cfcXa63jLwjI0eiQQMgzzUDFVpN/nH/eA=\ngithub.com/cpuguy83/go-md2man v1.0.10/go.mod h1:SmD6nW6nTyfqj6ABTjUi3V3JVMnlJmwcJI5acqYI6dE=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4 h1:wfIWP927BUkWJb2NmU/kNDYIBTh/ziUX91+lVfRxZq4=\ngithub.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=\ngithub.com/creack/pty v1.1.7/go.mod h1:lj5s0c3V2DBrqTV7llrYr5NG6My20zk30Fl46Y7DoTY=\ngithub.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=\ngithub.com/daixiang0/gci v0.12.0 h1:EQTG7FfKPlO4Ste+oN0kvz+gP4XswKx29D4fLrmwbiU=\ngithub.com/daixiang0/gci v0.12.0/go.mod h1:xtHP9N7AHdNvtRNfcx9gwTDfw7FRJx4bZUsiEfiNNAI=\ngithub.com/davecgh/go-spew v0.0.0-20161028175848-04cdfd42973b/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=\ngithub.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/denis-tingaikin/go-header v0.4.3 h1:tEaZKAlqql6SKCY++utLmkPLd6K8IBM20Ha7UVm+mtU=\ngithub.com/denis-tingaikin/go-header v0.4.3/go.mod h1:0wOCWuN71D5qIgE2nz9KrKmuYBAC2Mra5RassOIQ2/c=\ngithub.com/denisenkom/go-mssqldb v0.9.0/go.mod h1:xbL0rPBG9cCiLr28tMa8zpbdarY27NDyej4t/EjAShU=\ngithub.com/devigned/tab v0.1.1/go.mod h1:XG9mPq0dFghrYvoBF3xdRrJzSTX1b7IQrvaL9mzjeJY=\ngithub.com/dghubble/go-twitter v0.0.0-20201011215211-4b180d0cc78d h1:sBKr0A8iQ1qAOozedZ8Aox+Jpv+TeP1Qv7dcQyW8V+M=\ngithub.com/dghubble/go-twitter v0.0.0-20201011215211-4b180d0cc78d/go.mod h1:xfg4uS5LEzOj8PgZV7SQYRHbG7jPUnelEiaAVJxmhJE=\ngithub.com/dghubble/oauth1 v0.7.0 h1:AlpZdbRiJM4XGHIlQ8BuJ/wlpGwFEJNnB4Mc+78tA/w=\ngithub.com/dghubble/oauth1 v0.7.0/go.mod h1:8pFdfPkv/jr8mkChVbNVuJ0suiHe278BtWI4Tk1ujxk=\ngithub.com/dghubble/sling v1.3.0 h1:pZHjCJq4zJvc6qVQ5wN1jo5oNZlNE0+8T/h0XeXBUKU=\ngithub.com/dghubble/sling v1.3.0/go.mod h1:XXShWaBWKzNLhu2OxikSNFrlsvowtz4kyRuXUG7oQKY=\ngithub.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=\ngithub.com/dgryski/go-farm v0.0.0-20200201041132-a6ae2369ad13/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=\ngithub.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954/go.mod h1:vAd38F8PWV+bWy6jNmig1y/TA+kYO4g3RSRF0IAv0no=\ngithub.com/dimchansky/utfbom v1.1.0/go.mod h1:rO41eb7gLfo8SF1jd9F8HplJm1Fewwi4mQvIirEdv+8=\ngithub.com/dimchansky/utfbom v1.1.1 h1:vV6w1AhK4VMnhBno/TPVCoK9U/LP0PkLCS9tbxHdi/U=\ngithub.com/dimchansky/utfbom v1.1.1/go.mod h1:SxdoEBH5qIqFocHMyGOXVAybYJdr71b1Q/j0mACtrfE=\ngithub.com/dmarkham/enumer v1.5.8 h1:fIF11F9l5jyD++YYvxcSH5WgHfeaSGPaN/T4kOQ4qEM=\ngithub.com/dmarkham/enumer v1.5.8/go.mod h1:d10o8R3t/gROm2p3BXqTkMt2+HMuxEmWCXzorAruYak=\ngithub.com/dustin/go-humanize v0.0.0-20171111073723-bb3d318650d4/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=\ngithub.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=\ngithub.com/emirpasic/gods v1.12.0 h1:QAUIPSaCu4G+POclxeqb3F+WPpdKqFGlw36+yOzGlrg=\ngithub.com/emirpasic/gods v1.12.0/go.mod h1:YfzfFFoVP/catgzJb4IKIqXjX78Ha8FMSDh3ymbK86o=\ngithub.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=\ngithub.com/envoyproxy/go-control-plane v0.9.7/go.mod h1:cwu0lG7PUMfa9snN8LXBig5ynNVH9qI8YYLbd1fK2po=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210217033140-668b12f5399d/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk=\ngithub.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.mod h1:hliV/p42l8fGbc6Y9bQ70uLwIvmJyVE5k4iMKlh8wCQ=\ngithub.com/envoyproxy/go-control-plane v0.9.10-0.20210907150352-cf90f659a021/go.mod h1:AFq3mo9L8Lqqiid3OhADV3RfLJnjiw63cSpi+fDTRC0=\ngithub.com/envoyproxy/go-control-plane v0.13.1 h1:vPfJZCkob6yTMEgS+0TwfTUfbHjfy/6vOJ8hUWX/uXE=\ngithub.com/envoyproxy/go-control-plane v0.13.1/go.mod h1:X45hY0mufo6Fd0KW3rqsGvQMw58jvjymeCzBU3mWyHw=\ngithub.com/envoyproxy/protoc-gen-validate v0.0.14/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/envoyproxy/protoc-gen-validate v1.1.0 h1:tntQDh69XqOCOZsDz0lVJQez/2L6Uu2PdjCQwWCJ3bM=\ngithub.com/envoyproxy/protoc-gen-validate v1.1.0/go.mod h1:sXRDRVmzEbkM7CVcM06s9shE/m23dg3wzjl0UWqJ2q4=\ngithub.com/esimonov/ifshort v1.0.4 h1:6SID4yGWfRae/M7hkVDVVyppy8q/v9OuxNdmjLQStBA=\ngithub.com/esimonov/ifshort v1.0.4/go.mod h1:Pe8zjlRrJ80+q2CxHLfEOfTwxCZ4O+MuhcHcfgNWTk0=\ngithub.com/ettle/strcase v0.1.1 h1:htFueZyVeE1XNnMEfbqp5r67qAN/4r6ya1ysq8Q+Zcw=\ngithub.com/ettle/strcase v0.1.1/go.mod h1:hzDLsPC7/lwKyBOywSHEP89nt2pDgdy+No1NBA9o9VY=\ngithub.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=\ngithub.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU=\ngithub.com/fatih/color v1.10.0/go.mod h1:ELkj/draVOlAH/xkhN6mQ50Qd0MPOk5AAr3maGEBuJM=\ngithub.com/fatih/color v1.12.0/go.mod h1:ELkj/draVOlAH/xkhN6mQ50Qd0MPOk5AAr3maGEBuJM=\ngithub.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk=\ngithub.com/fatih/color v1.15.0 h1:kOqh6YHBtK8aywxGerMG2Eq3H6Qgoqeo13Bk2Mv/nBs=\ngithub.com/fatih/color v1.15.0/go.mod h1:0h5ZqXfHYED7Bhv2ZJamyIOUej9KtShiJESRwBDUSsw=\ngithub.com/fatih/structtag v1.0.0/go.mod h1:IKitwq45uXL/yqi5mYghiD3w9H6eTOvI9vnk8tXMphA=\ngithub.com/fatih/structtag v1.2.0 h1:/OdNE99OxoI/PqaW/SuSK9uxxT3f/tcSZgon/ssNSx4=\ngithub.com/fatih/structtag v1.2.0/go.mod h1:mBJUNpUnHmRKrKlQQlmCrh5PuhftFbNv8Ys4/aAZl94=\ngithub.com/felixge/httpsnoop v1.0.4 h1:NFTV2Zj1bL4mc9sqWACXbQFVBBg2W3GPvqp8/ESS2Wg=\ngithub.com/felixge/httpsnoop v1.0.4/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=\ngithub.com/firefart/nonamedreturns v1.0.1 h1:fSvcq6ZpK/uBAgJEGMvzErlzyM4NELLqqdTofVjVNag=\ngithub.com/firefart/nonamedreturns v1.0.1/go.mod h1:D3dpIBojGGNh5UfElmwPu73SwDCm+VKhHYqwlNOk2uQ=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568 h1:BHsljHzVlRcyQhjrss6TZTdY2VfCqZPbv5k3iBFa2ZQ=\ngithub.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568/go.mod h1:xEzjJPgXI435gkrCt3MPfRiAkVrwSbHsst4LCFVfpJc=\ngithub.com/form3tech-oss/jwt-go v3.2.2+incompatible h1:TcekIExNqud5crz4xD2pavyTgWiPvpYe4Xau31I0PRk=\ngithub.com/form3tech-oss/jwt-go v3.2.2+incompatible/go.mod h1:pbq4aXjuKjdthFRnoDwaVPLA+WlJuPGy+QneDUgJi2k=\ngithub.com/fortytw2/leaktest v1.3.0/go.mod h1:jDsjWgpAGjm2CA7WthBh/CdZYEPF31XHquHwclZch5g=\ngithub.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=\ngithub.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=\ngithub.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=\ngithub.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=\ngithub.com/fsnotify/fsnotify v1.5.1/go.mod h1:T3375wBYaZdLLcVNkcVbzGHY7f1l/uK5T5Ai1i3InKU=\ngithub.com/fsnotify/fsnotify v1.5.4/go.mod h1:OVB6XrOHzAwXMpEM7uPOzcehqUV2UqJxmVXmkdnm1bU=\ngithub.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M=\ngithub.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=\ngithub.com/fullstorydev/grpcurl v1.6.0/go.mod h1:ZQ+ayqbKMJNhzLmbpCiurTVlaK2M/3nqZCxaQ2Ze/sM=\ngithub.com/fzipp/gocyclo v0.5.1 h1:L66amyuYogbxl0j2U+vGqJXusPF2IkduvXLnYD5TFgw=\ngithub.com/fzipp/gocyclo v0.5.1/go.mod h1:rXPyn8fnlpa0R2csP/31uerbiVBugk5whMdlyaLkLoA=\ngithub.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=\ngithub.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=\ngithub.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwvtwp4M=\ngithub.com/gliderlabs/ssh v0.2.2 h1:6zsha5zo/TWhRhwqCD3+EarCAgZ2yN28ipRnGPnwkI0=\ngithub.com/gliderlabs/ssh v0.2.2/go.mod h1:U7qILu1NlMHj9FlMhZLlkCdDnU1DBEAqr0aevW3Awn0=\ngithub.com/go-critic/go-critic v0.6.3 h1:abibh5XYBTASawfTQ0rA7dVtQT+6KzpGqb/J+DxRDaw=\ngithub.com/go-critic/go-critic v0.6.3/go.mod h1:c6b3ZP1MQ7o6lPR7Rv3lEf7pYQUmAcx8ABHgdZCQt/k=\ngithub.com/go-git/gcfg v1.5.0 h1:Q5ViNfGF8zFgyJWPqYwA7qGFoMTEiBmdlkcfRmpIMa4=\ngithub.com/go-git/gcfg v1.5.0/go.mod h1:5m20vg6GwYabIxaOonVkTdrILxQMpEShl1xiMF4ua+E=\ngithub.com/go-git/go-billy/v5 v5.0.0/go.mod h1:pmpqyWchKfYfrkb/UVH4otLvyi/5gJlGI4Hb3ZqZ3W0=\ngithub.com/go-git/go-billy/v5 v5.1.0/go.mod h1:pmpqyWchKfYfrkb/UVH4otLvyi/5gJlGI4Hb3ZqZ3W0=\ngithub.com/go-git/go-billy/v5 v5.3.1 h1:CPiOUAzKtMRvolEKw+bG1PLRpT7D3LIs3/3ey4Aiu34=\ngithub.com/go-git/go-billy/v5 v5.3.1/go.mod h1:pmpqyWchKfYfrkb/UVH4otLvyi/5gJlGI4Hb3ZqZ3W0=\ngithub.com/go-git/go-git-fixtures/v4 v4.0.2-0.20200613231340-f56387b50c12 h1:PbKy9zOy4aAKrJ5pibIRpVO2BXnK1Tlcg+caKI7Ox5M=\ngithub.com/go-git/go-git-fixtures/v4 v4.0.2-0.20200613231340-f56387b50c12/go.mod h1:m+ICp2rF3jDhFgEZ/8yziagdT1C+ZpZcrJjappBCDSw=\ngithub.com/go-git/go-git/v5 v5.2.0/go.mod h1:kh02eMX+wdqqxgNMEyq8YgwlIOsDOa9homkUq1PoTMs=\ngithub.com/go-git/go-git/v5 v5.3.0 h1:8WKMtJR2j8RntEXR/uvTKagfEt4GYlwQ7mntE4+0GWc=\ngithub.com/go-git/go-git/v5 v5.3.0/go.mod h1:xdX4bWJ48aOrdhnl2XqHYstHbbp6+LFS4r4X+lNVprw=\ngithub.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=\ngithub.com/go-gl/glfw/v3.3/glfw v0.0.0-20191125211704-12ad95a8df72/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=\ngithub.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=\ngithub.com/go-ini/ini v1.25.4/go.mod h1:ByCAeIL28uOIIG0E3PJtZPDL8WnHpFKFOtgjp+3Ies8=\ngithub.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=\ngithub.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vbaY=\ngithub.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=\ngithub.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=\ngithub.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A=\ngithub.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A=\ngithub.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=\ngithub.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=\ngithub.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=\ngithub.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=\ngithub.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=\ngithub.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=\ngithub.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=\ngithub.com/go-playground/validator/v10 v10.2.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GOhaH6EGOAJShg8Id5JGkI=\ngithub.com/go-redis/redis v6.15.8+incompatible/go.mod h1:NAIEuMOZ/fxfXJIrKDQDz8wamY7mA7PouImQ2Jvg6kA=\ngithub.com/go-sql-driver/mysql v1.4.0/go.mod h1:zAC/RDZ24gD3HViQzih4MyKcchzm+sOG5ZlKdlhCg5w=\ngithub.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=\ngithub.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=\ngithub.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=\ngithub.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0/go.mod h1:fyg7847qk6SyHyPtNmDHnmrv/HOrqktSC+C9fM+CJOE=\ngithub.com/go-toolsmith/astcast v1.0.0 h1:JojxlmI6STnFVG9yOImLeGREv8W2ocNUM+iOhR6jE7g=\ngithub.com/go-toolsmith/astcast v1.0.0/go.mod h1:mt2OdQTeAQcY4DQgPSArJjHCcOwlX+Wl/kwN+LbLGQ4=\ngithub.com/go-toolsmith/astcopy v1.0.0 h1:OMgl1b1MEpjFQ1m5ztEO06rz5CUd3oBv9RF7+DyvdG8=\ngithub.com/go-toolsmith/astcopy v1.0.0/go.mod h1:vrgyG+5Bxrnz4MZWPF+pI4R8h3qKRjjyvV/DSez4WVQ=\ngithub.com/go-toolsmith/astequal v1.0.0/go.mod h1:H+xSiq0+LtiDC11+h1G32h7Of5O3CYFJ99GVbS5lDKY=\ngithub.com/go-toolsmith/astequal v1.0.1 h1:JbSszi42Jiqu36Gnf363HWS9MTEAz67vTQLponh3Moc=\ngithub.com/go-toolsmith/astequal v1.0.1/go.mod h1:4oGA3EZXTVItV/ipGiOx7NWkY5veFfcsOJVS2YxltLw=\ngithub.com/go-toolsmith/astfmt v1.0.0 h1:A0vDDXt+vsvLEdbMFJAUBI/uTbRw1ffOPnxsILnFL6k=\ngithub.com/go-toolsmith/astfmt v1.0.0/go.mod h1:cnWmsOAuq4jJY6Ct5YWlVLmcmLMn1JUPuQIHCY7CJDw=\ngithub.com/go-toolsmith/astp v1.0.0 h1:alXE75TXgcmupDsMK1fRAy0YUzLzqPVvBKoyWV+KPXg=\ngithub.com/go-toolsmith/astp v1.0.0/go.mod h1:RSyrtpVlfTFGDYRbrjyWP1pYu//tSFcvdYrA8meBmLI=\ngithub.com/go-toolsmith/pkgload v1.0.2-0.20220101231613-e814995d17c5 h1:eD9POs68PHkwrx7hAB78z1cb6PfGq/jyWn3wJywsH1o=\ngithub.com/go-toolsmith/pkgload v1.0.2-0.20220101231613-e814995d17c5/go.mod h1:3NAwwmD4uY/yggRxoEjk/S00MIV3A+H7rrE3i87eYxM=\ngithub.com/go-toolsmith/strparse v1.0.0 h1:Vcw78DnpCAKlM20kSbAyO4mPfJn/lyYA4BJUDxe2Jb4=\ngithub.com/go-toolsmith/strparse v1.0.0/go.mod h1:YI2nUKP9YGZnL/L1/DLFBfixrcjslWct4wyljWhSRy8=\ngithub.com/go-toolsmith/typep v1.0.2 h1:8xdsa1+FSIH/RhEkgnD1j2CJOy5mNllW1Q9tRiYwvlk=\ngithub.com/go-toolsmith/typep v1.0.2/go.mod h1:JSQCQMUPdRlMZFswiq3TGpNp1GMktqkR2Ns5AIQkATU=\ngithub.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIxtHqx8aGss=\ngithub.com/go-viper/mapstructure/v2 v2.2.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=\ngithub.com/go-xmlfmt/xmlfmt v0.0.0-20191208150333-d5b6f63a941b h1:khEcpUM4yFcxg4/FHQWkvVRmgijNXRfzkIDHh23ggEo=\ngithub.com/go-xmlfmt/xmlfmt v0.0.0-20191208150333-d5b6f63a941b/go.mod h1:aUCEOzzezBEjDBbFBoSiya/gduyIiWYRP6CnSFIV8AM=\ngithub.com/gobwas/glob v0.2.3 h1:A4xDbljILXROh+kObIiy5kIaPYD8e96x1tgBhUI5J+Y=\ngithub.com/gobwas/glob v0.2.3/go.mod h1:d3Ez4x06l9bZtSvzIay5+Yzi0fmZzPgnTbPcKjJAkT8=\ngithub.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee/go.mod h1:L0fX3K22YWvt/FAX9NnzrNzcI4wNYi9Yku4O0LKYflo=\ngithub.com/gobwas/pool v0.2.0/go.mod h1:q8bcK0KcYlCgd9e7WYLm9LpyS+YeLd8JVDW6WezmKEw=\ngithub.com/gobwas/ws v1.0.2/go.mod h1:szmBTxLgaFppYjEmNtny/v3w89xOydFnnZMcgRRu/EM=\ngithub.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=\ngithub.com/gofrs/flock v0.8.1 h1:+gYjHKf32LDeiEEFhQaotPbLuUXjY5ZqxKgXy7n59aw=\ngithub.com/gofrs/flock v0.8.1/go.mod h1:F1TvTiK9OcQqauNUHlbJvyl9Qa1QvF/gOUDKA14jxHU=\ngithub.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a/go.mod h1:gf4bu3Q80BeJ6H1S1vYPm8/ELATdvryBaNFGgqEef3s=\ngithub.com/gogo/googleapis v1.3.2/go.mod h1:5YRNX2z1oM5gXdAkurHa942MDgEJyk02w4OecKY87+c=\ngithub.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.2.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=\ngithub.com/gogo/protobuf v1.2.1/go.mod h1:hp+jE20tsWTFYpLwKvXlhS1hjn+gTNwPg2I6zVXpSg4=\ngithub.com/gogo/protobuf v1.3.0/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=\ngithub.com/gogo/protobuf v1.3.1/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=\ngithub.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=\ngithub.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=\ngithub.com/gogo/status v1.1.0/go.mod h1:BFv9nrluPLmrS0EmGVvLaPNmRosr9KapBYd5/hpY1WM=\ngithub.com/gojuno/minimock/v3 v3.0.4/go.mod h1:HqeqnwV8mAABn3pO5hqF+RE7gjA0jsN8cbbSogoGrzI=\ngithub.com/gojuno/minimock/v3 v3.0.10 h1:0UbfgdLHaNRPHWF/RFYPkwxV2KI+SE4tR0dDSFMD7+A=\ngithub.com/gojuno/minimock/v3 v3.0.10/go.mod h1:CFXcUJYnBe+1QuNzm+WmdPYtvi/+7zQcPcyQGsbcIXg=\ngithub.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/groupcache v0.0.0-20160516000752-02826c3e7903/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20190129154638-5b532d6fd5ef/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20191227052852-215e87163ea7/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.2.0/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/mock v1.3.1/go.mod h1:sBzyDLLjw3U8JLTeZvSv8jJB+tU5PVekmnlKIyFUx0Y=\ngithub.com/golang/mock v1.4.0/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.1/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.3/go.mod h1:UOMv5ysSaYNkG+OFQykRIcU/QvvxJf3p21QfJ2Bt3cw=\ngithub.com/golang/mock v1.4.4/go.mod h1:l3mdAwkq5BuhzHwde/uurv3sEJeZMXNpwsxVWU71h+4=\ngithub.com/golang/mock v1.5.0/go.mod h1:CWnOUgYIOo4TcNZ0wHX3YZCqsaM1I1Jvs6v3mP3KVu8=\ngithub.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=\ngithub.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=\ngithub.com/golang/protobuf v1.1.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.3-0.20190920234318-1680a479a2cf/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.4/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=\ngithub.com/golang/protobuf v1.3.5/go.mod h1:6O5/vntMXwX2lRkT1hjjk0nAC1IDOTvTlVgjlRvqsdk=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=\ngithub.com/golang/protobuf v1.5.1/go.mod h1:DopwsBzvsk0Fs44TXzsVbJyPhcCPeIwnvohx4u74HPM=\ngithub.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=\ngithub.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=\ngithub.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=\ngithub.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golangci/check v0.0.0-20180506172741-cfe4005ccda2 h1:23T5iq8rbUYlhpt5DB4XJkc6BU31uODLD1o1gKvZmD0=\ngithub.com/golangci/check v0.0.0-20180506172741-cfe4005ccda2/go.mod h1:k9Qvh+8juN+UKMCS/3jFtGICgW8O96FVaZsaxdzDkR4=\ngithub.com/golangci/dupl v0.0.0-20180902072040-3e9179ac440a h1:w8hkcTqaFpzKqonE9uMCefW1WDie15eSP/4MssdenaM=\ngithub.com/golangci/dupl v0.0.0-20180902072040-3e9179ac440a/go.mod h1:ryS0uhF+x9jgbj/N71xsEqODy9BN81/GonCZiOzirOk=\ngithub.com/golangci/go-misc v0.0.0-20220329215616-d24fe342adfe h1:6RGUuS7EGotKx6J5HIP8ZtyMdiDscjMLfRBSPuzVVeo=\ngithub.com/golangci/go-misc v0.0.0-20220329215616-d24fe342adfe/go.mod h1:gjqyPShc/m8pEMpk0a3SeagVb0kaqvhscv+i9jI5ZhQ=\ngithub.com/golangci/gofmt v0.0.0-20190930125516-244bba706f1a h1:iR3fYXUjHCR97qWS8ch1y9zPNsgXThGwjKPrYfqMPks=\ngithub.com/golangci/gofmt v0.0.0-20190930125516-244bba706f1a/go.mod h1:9qCChq59u/eW8im404Q2WWTrnBUQKjpNYKMbU4M7EFU=\ngithub.com/golangci/golangci-lint v1.46.2 h1:o90t/Xa6dhJbvy8Bz2RpzUXqrkigp19DLStMolTZbyo=\ngithub.com/golangci/golangci-lint v1.46.2/go.mod h1:3DkdHnxn9eoTTrpT2gB0TEv8KSziuoqe9FitgQLHvAY=\ngithub.com/golangci/lint-1 v0.0.0-20191013205115-297bf364a8e0 h1:MfyDlzVjl1hoaPzPD4Gpb/QgoRfSBR0jdhwGyAWwMSA=\ngithub.com/golangci/lint-1 v0.0.0-20191013205115-297bf364a8e0/go.mod h1:66R6K6P6VWk9I95jvqGxkqJxVWGFy9XlDwLwVz1RCFg=\ngithub.com/golangci/maligned v0.0.0-20180506175553-b1d89398deca h1:kNY3/svz5T29MYHubXix4aDDuE3RWHkPvopM/EDv/MA=\ngithub.com/golangci/maligned v0.0.0-20180506175553-b1d89398deca/go.mod h1:tvlJhZqDe4LMs4ZHD0oMUlt9G2LWuDGoisJTBzLMV9o=\ngithub.com/golangci/misspell v0.3.5 h1:pLzmVdl3VxTOncgzHcvLOKirdvcx/TydsClUQXTehjo=\ngithub.com/golangci/misspell v0.3.5/go.mod h1:dEbvlSfYbMQDtrpRMQU675gSDLDNa8sCPPChZ7PhiVA=\ngithub.com/golangci/revgrep v0.0.0-20210930125155-c22e5001d4f2 h1:SgM7GDZTxtTTQPU84heOxy34iG5Du7F2jcoZnvp+fXI=\ngithub.com/golangci/revgrep v0.0.0-20210930125155-c22e5001d4f2/go.mod h1:LK+zW4MpyytAWQRz0M4xnzEk50lSvqDQKfx304apFkY=\ngithub.com/golangci/unconvert v0.0.0-20180507085042-28b1c447d1f4 h1:zwtduBRr5SSWhqsYNgcuWO2kFlpdOZbP0+yRjmvPGys=\ngithub.com/golangci/unconvert v0.0.0-20180507085042-28b1c447d1f4/go.mod h1:Izgrg8RkN3rCIMLGE9CyYmU9pY2Jer6DgANEnZ/L/cQ=\ngithub.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=\ngithub.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=\ngithub.com/google/certificate-transparency-go v1.0.21/go.mod h1:QeJfpSbVSfYc7RgB3gJFj9cbuQMMchQxrWXz8Ruopmg=\ngithub.com/google/certificate-transparency-go v1.1.1/go.mod h1:FDKqPvSXawb2ecErVRrD+nfy23RCzyl7eqVCEmlT1Zs=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.4.1/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.1/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE=\ngithub.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=\ngithub.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=\ngithub.com/google/go-github/v35 v35.2.0 h1:s/soW8jauhjUC3rh8JI0FePuocj0DEI9DNBg/bVplE8=\ngithub.com/google/go-github/v35 v35.2.0/go.mod h1:s0515YVTI+IMrDoy9Y4pHt9ShGpzHvHO8rZ7L7acgvs=\ngithub.com/google/go-querystring v1.0.0 h1:Xkwi/a1rcvNg1PPYe5vI8GbeBY/jrVuDX5ASuANWTrk=\ngithub.com/google/go-querystring v1.0.0/go.mod h1:odCYkC5MyYFN7vkCjXpyrEuKhc/BUO6wN/zVPAxq5ck=\ngithub.com/google/go-replayers/grpcreplay v1.0.0 h1:B5kVOzJ1hBgnevTgIWhSTatQ3608yu/2NnU0Ta1d0kY=\ngithub.com/google/go-replayers/grpcreplay v1.0.0/go.mod h1:8Ig2Idjpr6gifRd6pNVggX6TC1Zw6Jx74AKp7QNH2QE=\ngithub.com/google/go-replayers/httpreplay v0.1.2 h1:HCfx+dQzwN9XbGTHF8qJ+67WN8glL9FTWV5rraCJ/jU=\ngithub.com/google/go-replayers/httpreplay v0.1.2/go.mod h1:YKZViNhiGgqdBlUbI2MwGpq4pXxNmhJLPHQ7cv2b5no=\ngithub.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=\ngithub.com/google/martian v2.1.0+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXiIaQGbYVAs8BPL6v8lEs=\ngithub.com/google/martian v2.1.1-0.20190517191504-25dcb96d9e51+incompatible h1:xmapqc1AyLoB+ddYT6r04bD9lIjlOqGaREovi0SzFaE=\ngithub.com/google/martian v2.1.1-0.20190517191504-25dcb96d9e51+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXiIaQGbYVAs8BPL6v8lEs=\ngithub.com/google/martian/v3 v3.0.0/go.mod h1:y5Zk1BBys9G+gd6Jrk0W3cC1+ELVxBWuIGO+w/tUAp0=\ngithub.com/google/martian/v3 v3.1.0/go.mod h1:y5Zk1BBys9G+gd6Jrk0W3cC1+ELVxBWuIGO+w/tUAp0=\ngithub.com/google/martian/v3 v3.2.1/go.mod h1:oBOf6HBosgwRXnUGWUB05QECsc6uvmMiJ3+6W4l/CUk=\ngithub.com/google/martian/v3 v3.3.3 h1:DIhPTQrbPkgs2yJYdXU/eNACCG5DVQjySNRNlflZ9Fc=\ngithub.com/google/martian/v3 v3.3.3/go.mod h1:iEPrYcgCF7jA9OtScMFQyAlZZ4YXTKEtJ1E6RWzmBA0=\ngithub.com/google/pprof v0.0.0-20181206194817-3ea8567a2e57/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=\ngithub.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=\ngithub.com/google/pprof v0.0.0-20191218002539-d4f498aebedc/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20200212024743-f11f1df84d12/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20200229191704-1ebb73c60ed3/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20200430221834-fc25d7d30c6d/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20200507031123-427632fa3b1c/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20200708004538-1a94d8640e99/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=\ngithub.com/google/pprof v0.0.0-20201023163331-3e6fc7fc9c4c/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20201203190320-1bf35d6f28c2/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20201218002935-b9804c9f04c2/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210122040257-d980be63207e/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210226084205-cbba55b83ad5/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210407192527-94a9f03dee38/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210601050228-01bbb1931b22/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210609004039-a478d1d731e9/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/pprof v0.0.0-20210720184732-4bb14d4b1be1/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=\ngithub.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=\ngithub.com/google/rpmpack v0.0.0-20210410105602-e20c988a6f5a/go.mod h1:+y9lKiqDhR4zkLl+V9h4q0rdyrYVsWWm6LLCQP33DIk=\ngithub.com/google/s2a-go v0.1.8 h1:zZDs9gcbt9ZPLV0ndSyQk6Kacx2g/X+SKYovpnz3SMM=\ngithub.com/google/s2a-go v0.1.8/go.mod h1:6iNWHTpQ+nfNRN5E00MSdfDwVesa8hhS32PhPO8deJA=\ngithub.com/google/subcommands v1.0.1/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk=\ngithub.com/google/trillian v1.3.11/go.mod h1:0tPraVHrSDkA3BO6vKX67zgLXs6SsOAbHEivX+9mPgw=\ngithub.com/google/uuid v0.0.0-20161128191214-064e2069ce9c/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.2.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/wire v0.5.0 h1:I7ELFeVBr3yfPIcc8+MWvrjk+3VjbcSzoXm3JVa+jD8=\ngithub.com/google/wire v0.5.0/go.mod h1:ngWDr9Qvq3yZA10YrxfyGELY/AFWGVpy9c1LTRi1EoU=\ngithub.com/googleapis/enterprise-certificate-proxy v0.3.4 h1:XYIDZApgAnrN1c855gTgghdIA6Stxb52D5RnLI1SLyw=\ngithub.com/googleapis/enterprise-certificate-proxy v0.3.4/go.mod h1:YKe7cfqYXjKGpGvmSg28/fFvhNzinZQm8DGnaburhGA=\ngithub.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=\ngithub.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=\ngithub.com/googleapis/gax-go/v2 v2.1.0/go.mod h1:Q3nei7sK6ybPYH7twZdmQpAd1MKb7pfu6SK+H1/DsU0=\ngithub.com/googleapis/gax-go/v2 v2.1.1/go.mod h1:hddJymUZASv3XPyGkUpKj8pPO47Rmb0eJc8R6ouapiM=\ngithub.com/googleapis/gax-go/v2 v2.2.0/go.mod h1:as02EH8zWkzwUoLbBaFeQ+arQaj/OthfcblKl4IGNaM=\ngithub.com/googleapis/gax-go/v2 v2.3.0/go.mod h1:b8LNqSzNabLiUpXKkY7HAR5jr6bIT99EXz9pXxye9YM=\ngithub.com/googleapis/gax-go/v2 v2.14.1 h1:hb0FFeiPaQskmvakKu5EbCbpntQn48jyHuvrkurSS/Q=\ngithub.com/googleapis/gax-go/v2 v2.14.1/go.mod h1:Hb/NubMaVM88SrNkvl8X/o8XWwDJEPqouaLeN2IUxoA=\ngithub.com/googleapis/google-cloud-go-testing v0.0.0-20200911160855-bcd43fbb19e8/go.mod h1:dvDLG8qkwmyD9a/MJJN3XJcT3xFxOKAvTZGvuZmac9g=\ngithub.com/gookit/color v1.5.0/go.mod h1:43aQb+Zerm/BWh2GnrgOQm7ffz7tvQXEKV6BFMl7wAo=\ngithub.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1 h1:EGx4pi6eqNxGaHF6qqu48+N2wcFQ5qg5FXgOdqsJ5d8=\ngithub.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=\ngithub.com/gordonklaus/ineffassign v0.0.0-20200309095847-7953dde2c7bf/go.mod h1:cuNKsD1zp2v6XfE/orVX2QE1LC+i254ceGcVeDT3pTU=\ngithub.com/gordonklaus/ineffassign v0.0.0-20210914165742-4cc7213b9bc8 h1:PVRE9d4AQKmbelZ7emNig1+NT27DUmKZn5qXxfio54U=\ngithub.com/gordonklaus/ineffassign v0.0.0-20210914165742-4cc7213b9bc8/go.mod h1:Qcp2HIAYhR7mNUVSIxZww3Guk4it82ghYcEXIAk+QT0=\ngithub.com/goreleaser/chglog v0.1.2/go.mod h1:tTZsFuSZK4epDXfjMkxzcGbrIOXprf0JFp47BjIr3B8=\ngithub.com/goreleaser/chglog v0.4.2 h1:afmbT1d7lX/q+GF8wv3a1Dofs2j/Y9YkiCpGemWR6mI=\ngithub.com/goreleaser/chglog v0.4.2/go.mod h1:u/F03un4hMCQrp65qSWCkkC6T+G7YLKZ+AM2mITE47s=\ngithub.com/goreleaser/fileglob v1.2.0/go.mod h1:rFyb2pXaK3YdnYnSjn6lifw0h2Q6s8OfOsx6I6bXkKE=\ngithub.com/goreleaser/fileglob v1.3.0 h1:/X6J7U8lbDpQtBvGcwwPS6OpzkNVlVEsFUVRx9+k+7I=\ngithub.com/goreleaser/fileglob v1.3.0/go.mod h1:Jx6BoXv3mbYkEzwm9THo7xbr5egkAraxkGorbJb4RxU=\ngithub.com/goreleaser/goreleaser v0.169.0 h1:QaTuoCK39LtVpRmocY6THXvnJCxdFHgyWJbqsHnS7K4=\ngithub.com/goreleaser/goreleaser v0.169.0/go.mod h1:cJIZtW13/jxPVpVIhLgNa/fXu8Zwn9RMuIoDoUIqwbY=\ngithub.com/goreleaser/nfpm/v2 v2.5.1/go.mod h1:Bq9OBKhvhTmdPh6lHUbVBKa3JCw61OgIFEau+vs2CO0=\ngithub.com/goreleaser/nfpm/v2 v2.29.0 h1:QW7MD5Od8ePAWqvC+kGQiF8OH5JkSKV+HcblcT0NX6A=\ngithub.com/goreleaser/nfpm/v2 v2.29.0/go.mod h1:+O8Rgz7geEXG1ym2Yl8CGPg5nP2LRuCgkBK6CQF+Q3c=\ngithub.com/gorhill/cronexpr v0.0.0-20180427100037-88b0669f7d75/go.mod h1:g2644b03hfBX9Ov0ZBDgXXens4rxSxmqFBbhvKv2yVA=\ngithub.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=\ngithub.com/gorilla/websocket v0.0.0-20170926233335-4201258b820c/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ=\ngithub.com/gorilla/websocket v1.4.1/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=\ngithub.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=\ngithub.com/gostaticanalysis/analysisutil v0.0.0-20190318220348-4088753ea4d3/go.mod h1:eEOZF4jCKGi+aprrirO9e7WKB3beBRtWgqGunKl6pKE=\ngithub.com/gostaticanalysis/analysisutil v0.0.3/go.mod h1:eEOZF4jCKGi+aprrirO9e7WKB3beBRtWgqGunKl6pKE=\ngithub.com/gostaticanalysis/analysisutil v0.1.0/go.mod h1:dMhHRU9KTiDcuLGdy87/2gTR8WruwYZrKdRq9m1O6uw=\ngithub.com/gostaticanalysis/analysisutil v0.7.1 h1:ZMCjoue3DtDWQ5WyU16YbjbQEQ3VuzwxALrpYd+HeKk=\ngithub.com/gostaticanalysis/analysisutil v0.7.1/go.mod h1:v21E3hY37WKMGSnbsw2S/ojApNWb6C1//mXO48CXbVc=\ngithub.com/gostaticanalysis/comment v1.3.0/go.mod h1:xMicKDx7XRXYdVwY9f9wQpDJVnqWxw9wCauCMKp+IBI=\ngithub.com/gostaticanalysis/comment v1.4.1/go.mod h1:ih6ZxzTHLdadaiSnF5WY3dxUoXfXAlTaRzuaNDlSado=\ngithub.com/gostaticanalysis/comment v1.4.2 h1:hlnx5+S2fY9Zo9ePo4AhgYsYHbM2+eAv8m/s1JiCd6Q=\ngithub.com/gostaticanalysis/comment v1.4.2/go.mod h1:KLUTGDv6HOCotCH8h2erHKmpci2ZoR8VPu34YA2uzdM=\ngithub.com/gostaticanalysis/forcetypeassert v0.1.0 h1:6eUflI3DiGusXGK6X7cCcIgVCpZ2CiZ1Q7jl6ZxNV70=\ngithub.com/gostaticanalysis/forcetypeassert v0.1.0/go.mod h1:qZEedyP/sY1lTGV1uJ3VhWZ2mqag3IkWsDHVbplHXak=\ngithub.com/gostaticanalysis/nilerr v0.1.1 h1:ThE+hJP0fEp4zWLkWHWcRyI2Od0p7DlgYG3Uqrmrcpk=\ngithub.com/gostaticanalysis/nilerr v0.1.1/go.mod h1:wZYb6YI5YAxxq0i1+VJbY0s2YONW0HU0GPE3+5PWN4A=\ngithub.com/gostaticanalysis/testutil v0.3.1-0.20210208050101-bfb5c8eec0e4/go.mod h1:D+FIZ+7OahH3ePw/izIEeH5I06eKs1IKI4Xr64/Am3M=\ngithub.com/gostaticanalysis/testutil v0.4.0 h1:nhdCmubdmDF6VEatUNjgUZBJKWRqugoISdUv3PPQgHY=\ngithub.com/gostaticanalysis/testutil v0.4.0/go.mod h1:bLIoPefWXrRi/ssLFWX1dx7Repi5x3CuviD3dgAZaBU=\ngithub.com/gregjones/httpcache v0.0.0-20190611155906-901d90724c79/go.mod h1:FecbI9+v66THATjSRHfNgh1IVFe/9kFxbXtjV0ctIMA=\ngithub.com/grpc-ecosystem/go-grpc-middleware v1.0.0/go.mod h1:FiyG127CGDf3tlThmgyCl78X/SZQqEOJBCDaAfeWzPs=\ngithub.com/grpc-ecosystem/go-grpc-middleware v1.0.1-0.20190118093823-f849b5445de4/go.mod h1:FiyG127CGDf3tlThmgyCl78X/SZQqEOJBCDaAfeWzPs=\ngithub.com/grpc-ecosystem/go-grpc-middleware v1.2.2/go.mod h1:EaizFBKfUKtMIF5iaDEhniwNedqGo9FuLFzppDr3uwI=\ngithub.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk=\ngithub.com/grpc-ecosystem/grpc-gateway v1.9.0/go.mod h1:vNeuVxBJEsws4ogUvrchl83t/GYV9WGTSLVdBhOQFDY=\ngithub.com/grpc-ecosystem/grpc-gateway v1.9.5/go.mod h1:vNeuVxBJEsws4ogUvrchl83t/GYV9WGTSLVdBhOQFDY=\ngithub.com/grpc-ecosystem/grpc-gateway v1.12.1/go.mod h1:8XEsbTttt/W+VvjtQhLACqCisSPWTxCZ7sBRjU6iH9c=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw=\ngithub.com/hashicorp/consul/api v1.1.0/go.mod h1:VmuI/Lkw1nC05EYQWNKwWGbkg+FbDBtguAZLlVdkD9Q=\ngithub.com/hashicorp/consul/api v1.12.0/go.mod h1:6pVBMo0ebnYdt2S3H87XhekM/HHrUoTD2XXb/VrZVy0=\ngithub.com/hashicorp/consul/sdk v0.1.1/go.mod h1:VKf9jXwCTEY1QZP2MOLRhb5i/I/ssyNV1vwHyQBF0x8=\ngithub.com/hashicorp/consul/sdk v0.8.0/go.mod h1:GBvyrGALthsZObzUGsfgHZQDXjg4lOjagTIwIR1vPms=\ngithub.com/hashicorp/errwrap v1.0.0 h1:hLrqtEDnRye3+sgx6z4qVLNuviH3MR5aQ0ykNJa/UYA=\ngithub.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=\ngithub.com/hashicorp/go-cleanhttp v0.5.0/go.mod h1:JpRdi6/HCYpAwUzNwuwqhbovhLtngrth3wmdIIUrZ80=\ngithub.com/hashicorp/go-cleanhttp v0.5.1/go.mod h1:JpRdi6/HCYpAwUzNwuwqhbovhLtngrth3wmdIIUrZ80=\ngithub.com/hashicorp/go-cleanhttp v0.5.2 h1:035FKYIWjmULyFRBKPs8TBQoi0x6d9G4xc9neXJWAZQ=\ngithub.com/hashicorp/go-cleanhttp v0.5.2/go.mod h1:kO/YDlP8L1346E6Sodw+PrpBSV4/SoxCXGY6BqNFT48=\ngithub.com/hashicorp/go-hclog v0.9.2/go.mod h1:5CU+agLiy3J7N7QjHK5d05KxGsuXiQLrjA0H7acj2lQ=\ngithub.com/hashicorp/go-hclog v0.12.0/go.mod h1:whpDNt7SSdeAju8AWKIWsul05p54N/39EeqMAyrmvFQ=\ngithub.com/hashicorp/go-hclog v1.2.0 h1:La19f8d7WIlm4ogzNHB0JGqs5AUDAZ2UfCY4sJXcJdM=\ngithub.com/hashicorp/go-hclog v1.2.0/go.mod h1:whpDNt7SSdeAju8AWKIWsul05p54N/39EeqMAyrmvFQ=\ngithub.com/hashicorp/go-immutable-radix v1.0.0/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60=\ngithub.com/hashicorp/go-immutable-radix v1.3.1/go.mod h1:0y9vanUI8NX6FsYoO3zeMjhV/C5i9g4Q3DwcSNZ4P60=\ngithub.com/hashicorp/go-msgpack v0.5.3/go.mod h1:ahLV/dePpqEmjfWmKiqvPkv/twdG7iPBM1vqhUKIvfM=\ngithub.com/hashicorp/go-multierror v1.0.0/go.mod h1:dHtQlpGsu+cZNNAkkCN/P3hoUDHhCYQXV3UM06sGGrk=\ngithub.com/hashicorp/go-multierror v1.1.0/go.mod h1:spPvp8C1qA32ftKqdAHm4hHTbPw+vmowP0z+KUhOZdA=\ngithub.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=\ngithub.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=\ngithub.com/hashicorp/go-retryablehttp v0.5.3/go.mod h1:9B5zBasrRhHXnJnui7y6sL7es7NDiJgTc6Er0maI1Xs=\ngithub.com/hashicorp/go-retryablehttp v0.6.8 h1:92lWxgpa+fF3FozM4B3UZtHZMJX8T5XT+TFdCxsPyWs=\ngithub.com/hashicorp/go-retryablehttp v0.6.8/go.mod h1:vAew36LZh98gCBJNLH42IQ1ER/9wtLZZ8meHqQvEYWY=\ngithub.com/hashicorp/go-rootcerts v1.0.0/go.mod h1:K6zTfqpRlCUIjkwsN4Z+hiSfzSTQa6eBIzfwKfwNnHU=\ngithub.com/hashicorp/go-rootcerts v1.0.2/go.mod h1:pqUvnprVnM5bf7AOirdbb01K4ccR319Vf4pU3K5EGc8=\ngithub.com/hashicorp/go-sockaddr v1.0.0/go.mod h1:7Xibr9yA9JjQq1JpNB2Vw7kxv8xerXegt+ozgdvDeDU=\ngithub.com/hashicorp/go-syslog v1.0.0/go.mod h1:qPfqrKkXGihmCqbJM2mZgkZGvKG1dFdvsLplgctolz4=\ngithub.com/hashicorp/go-uuid v1.0.0/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-uuid v1.0.1/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=\ngithub.com/hashicorp/go-version v1.2.1/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=\ngithub.com/hashicorp/go-version v1.4.0 h1:aAQzgqIrRKRa7w75CKpbBxYsmUoPjzVm1W59ca1L0J4=\ngithub.com/hashicorp/go-version v1.4.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=\ngithub.com/hashicorp/go.net v0.0.1/go.mod h1:hjKkEWcCURg++eb33jQU7oqQcI9XDCnUzHA0oac0k90=\ngithub.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=\ngithub.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=\ngithub.com/hashicorp/golang-lru v0.5.4/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4=\ngithub.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=\ngithub.com/hashicorp/logutils v1.0.0/go.mod h1:QIAnNjmIWmVIIkWDTG1z5v++HQmx9WQRO+LraFDTW64=\ngithub.com/hashicorp/mdns v1.0.0/go.mod h1:tL+uN++7HEJ6SQLQ2/p+z2pH24WQKWjBPkE0mNTz8vQ=\ngithub.com/hashicorp/mdns v1.0.4/go.mod h1:mtBihi+LeNXGtG8L9dX59gAEa12BDtBQSp4v/YAJqrc=\ngithub.com/hashicorp/memberlist v0.1.3/go.mod h1:ajVTdAv/9Im8oMAAj5G31PhhMCZJV2pPBoIllUwCN7I=\ngithub.com/hashicorp/memberlist v0.3.0/go.mod h1:MS2lj3INKhZjWNqd3N0m3J+Jxf3DAOnAH9VT3Sh9MUE=\ngithub.com/hashicorp/serf v0.8.2/go.mod h1:6hOLApaqBFA1NXqRQAsxw9QxuDEvNxSQRwA/JwenrHc=\ngithub.com/hashicorp/serf v0.9.6/go.mod h1:TXZNMjZQijwlDvp+r0b63xZ45H7JmCmgg4gpTwn9UV4=\ngithub.com/hashicorp/serf v0.9.7/go.mod h1:TXZNMjZQijwlDvp+r0b63xZ45H7JmCmgg4gpTwn9UV4=\ngithub.com/hexdigest/gowrap v1.1.7/go.mod h1:Z+nBFUDLa01iaNM+/jzoOA1JJ7sm51rnYFauKFUB5fs=\ngithub.com/hexdigest/gowrap v1.1.8/go.mod h1:H/JiFmQMp//tedlV8qt2xBdGzmne6bpbaSuiHmygnMw=\ngithub.com/hexdigest/gowrap v1.2.5 h1:rg1kPeynZGXqCdVT3M633j/k+7M7QUKqcuP0MchJD5E=\ngithub.com/hexdigest/gowrap v1.2.5/go.mod h1:hhd/3h63Nnym1bvjcoaLYsdpyRusHoJorMfkT8DZDh4=\ngithub.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=\ngithub.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg=\ngithub.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=\ngithub.com/huandu/xstrings v1.0.0/go.mod h1:4qWG/gcEcfX4z/mBDHJ++3ReCw9ibxbsNJbcucJdbSo=\ngithub.com/huandu/xstrings v1.2.0/go.mod h1:DvyZB1rfVYsBIigL8HwpZgxHwXozlTgGqn63UyNX5k4=\ngithub.com/huandu/xstrings v1.3.1/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=\ngithub.com/huandu/xstrings v1.3.2/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=\ngithub.com/huandu/xstrings v1.4.0 h1:D17IlohoQq4UcpqD7fDk80P7l+lwAmlFaBHgOipl2FU=\ngithub.com/huandu/xstrings v1.4.0/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=\ngithub.com/iancoleman/strcase v0.3.0 h1:nTXanmYxhfFAMjZL34Ov6gkzEsSJZ5DbhxWjvSASxEI=\ngithub.com/iancoleman/strcase v0.3.0/go.mod h1:iwCmte+B7n89clKwxIoIXy/HfoL7AsD47ZCWhYzw7ho=\ngithub.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=\ngithub.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc=\ngithub.com/imdario/mergo v0.3.4/go.mod h1:2EnlNZ0deacrJVfApfmtdGgDfMuh/nq6Ok1EcJh5FfA=\ngithub.com/imdario/mergo v0.3.8/go.mod h1:2EnlNZ0deacrJVfApfmtdGgDfMuh/nq6Ok1EcJh5FfA=\ngithub.com/imdario/mergo v0.3.9/go.mod h1:2EnlNZ0deacrJVfApfmtdGgDfMuh/nq6Ok1EcJh5FfA=\ngithub.com/imdario/mergo v0.3.11/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=\ngithub.com/imdario/mergo v0.3.12/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=\ngithub.com/imdario/mergo v0.3.15 h1:M8XP7IuFNsqUx6VPK2P9OSmsYsI/YFaGil0uD21V3dM=\ngithub.com/imdario/mergo v0.3.15/go.mod h1:WBLT9ZmE3lPoWsEzCh9LPo3TiwVN+ZKEjmz+hD27ysY=\ngithub.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=\ngithub.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=\ngithub.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=\ngithub.com/jarcoal/httpmock v1.0.8 h1:8kI16SoO6LQKgPE7PvQuV+YuD/inwHd7fOOe2zMbo4k=\ngithub.com/jarcoal/httpmock v1.0.8/go.mod h1:ATjnClrvW/3tijVmpL/va5Z3aAyGvqU3gCT8nX0Txik=\ngithub.com/jbenet/go-context v0.0.0-20150711004518-d14ea06fba99 h1:BQSFePA1RWJOlocH6Fxy8MmwDt+yVQYULKfN0RoTN8A=\ngithub.com/jbenet/go-context v0.0.0-20150711004518-d14ea06fba99/go.mod h1:1lJo3i6rXxKeerYnT8Nvf0QmHCRC1n8sfWVwXF2Frvo=\ngithub.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=\ngithub.com/jessevdk/go-flags v1.5.0 h1:1jKYvbxEjfUl0fmqTCOfonvskHHXMjBySTLW4y9LFvc=\ngithub.com/jessevdk/go-flags v1.5.0/go.mod h1:Fw0T6WPc1dYxT4mKEZRfG5kJhaTDP9pj1c2EWnYs/m4=\ngithub.com/jgautheron/goconst v1.5.1 h1:HxVbL1MhydKs8R8n/HE5NPvzfaYmQJA3o879lE4+WcM=\ngithub.com/jgautheron/goconst v1.5.1/go.mod h1:aAosetZ5zaeC/2EfMeRswtxUFBpe2Hr7HzkgX4fanO4=\ngithub.com/jhump/protoreflect v1.6.1/go.mod h1:RZQ/lnuN+zqeRVpQigTwO6o0AJUkxbnSnpuG7toUTG4=\ngithub.com/jingyugao/rowserrcheck v1.1.1 h1:zibz55j/MJtLsjP1OF4bSdgXxwL1b+Vn7Tjzq7gFzUs=\ngithub.com/jingyugao/rowserrcheck v1.1.1/go.mod h1:4yvlZSDb3IyDTUZJUmpZfm2Hwok+Dtp+nu2qOq+er9c=\ngithub.com/jinzhu/copier v0.4.0 h1:w3ciUoD19shMCRargcpm0cm91ytaBhDvuRpz1ODO/U8=\ngithub.com/jinzhu/copier v0.4.0/go.mod h1:DfbEm0FYsaqBcKcFuvmOZb218JkPGtvSHsKg8S8hyyg=\ngithub.com/jirfag/go-printf-func-name v0.0.0-20200119135958-7558a9eaa5af h1:KA9BjwUk7KlCh6S9EAGWBt1oExIUv9WyNCiRz5amv48=\ngithub.com/jirfag/go-printf-func-name v0.0.0-20200119135958-7558a9eaa5af/go.mod h1:HEWGJkRDzjJY2sqdDwxccsGicWEf9BQOZsq2tV+xzM0=\ngithub.com/jmespath/go-jmespath v0.0.0-20160202185014-0b12d6b521d8/go.mod h1:Nht3zPeWKUH0NzdCt2Blrr5ys8VGpn0CEB0cQHVjt7k=\ngithub.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af/go.mod h1:Nht3zPeWKUH0NzdCt2Blrr5ys8VGpn0CEB0cQHVjt7k=\ngithub.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=\ngithub.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=\ngithub.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=\ngithub.com/jmoiron/sqlx v1.2.0/go.mod h1:1FEQNm3xlJgrMD+FBdI9+xvCksHtbpVBBw5dYhBSsks=\ngithub.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg=\ngithub.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=\ngithub.com/jonboulle/clockwork v0.2.0/go.mod h1:Pkfl5aHPm1nk2H9h0bjmnJD/BcgbGXUBGnn1kMkgxc8=\ngithub.com/jpillora/backoff v0.0.0-20180909062703-3050d21c67d7/go.mod h1:2iMrUgbbvHEiQClaW2NsSzMyGHqN+rDFqY705q49KG0=\ngithub.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=\ngithub.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=\ngithub.com/json-iterator/go v1.1.7/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.8/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.11/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=\ngithub.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=\ngithub.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU=\ngithub.com/jstemmer/go-junit-report v0.9.1/go.mod h1:Brl9GWCQeLvo8nXZwPNNblvFj/XSXhF0NWZEnDohbsk=\ngithub.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=\ngithub.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=\ngithub.com/juju/ratelimit v1.0.1/go.mod h1:qapgC/Gy+xNh9UxzV13HGGl/6UXNN+ct+vwSgWNm/qk=\ngithub.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=\ngithub.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=\ngithub.com/julz/importas v0.1.0 h1:F78HnrsjY3cR7j0etXy5+TU1Zuy7Xt08X/1aJnH5xXY=\ngithub.com/julz/importas v0.1.0/go.mod h1:oSFU2R4XK/P7kNBrnL/FEQlDGN1/6WoxXEjSSXO0DV0=\ngithub.com/k0kubun/colorstring v0.0.0-20150214042306-9440f1994b88/go.mod h1:3w7q1U84EfirKl04SVQ/s7nPm1ZPhiXd34z40TNz36k=\ngithub.com/kevinburke/ssh_config v0.0.0-20190725054713-01f96b0aa0cd/go.mod h1:CT57kijsi8u/K/BOFA39wgDQJ9CxiF4nAY/ojJ6r6mM=\ngithub.com/kevinburke/ssh_config v0.0.0-20201106050909-4977a11b4351/go.mod h1:CT57kijsi8u/K/BOFA39wgDQJ9CxiF4nAY/ojJ6r6mM=\ngithub.com/kevinburke/ssh_config v1.1.0 h1:pH/t1WS9NzT8go394IqZeJTMHVm6Cr6ZJ6AQ+mdNo/o=\ngithub.com/kevinburke/ssh_config v1.1.0/go.mod h1:CT57kijsi8u/K/BOFA39wgDQJ9CxiF4nAY/ojJ6r6mM=\ngithub.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q=\ngithub.com/kisielk/errcheck v1.2.0/go.mod h1:/BMXB+zMLi60iA8Vv6Ksmxu/1UDYcXs4uQLJ+jE2L00=\ngithub.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/errcheck v1.6.0 h1:YTDO4pNy7AUN/021p+JGHycQyYNIyMoenM1YDVK6RlY=\ngithub.com/kisielk/errcheck v1.6.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=\ngithub.com/kisielk/gotool v1.0.0 h1:AV2c/EiW3KqPNT9ZKl07ehoAGi4C5/01Cfbblndcapg=\ngithub.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=\ngithub.com/klauspost/compress v1.10.3/go.mod h1:aoV0uJVorq1K+umq18yTdKaF57EivdYsUV+/s2qKfXs=\ngithub.com/klauspost/compress v1.12.2/go.mod h1:8dP1Hq4DHOhN9w426knH3Rhby4rFm6D8eO+e+Dq5Gzg=\ngithub.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=\ngithub.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=\ngithub.com/klauspost/pgzip v1.2.6 h1:8RXeL5crjEUFnR2/Sn6GJNWtSQ3Dk8pq4CL3jvdDyjU=\ngithub.com/klauspost/pgzip v1.2.6/go.mod h1:Ch1tH69qFZu15pkjo5kYi6mth2Zzwzt50oCQKQE9RUs=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=\ngithub.com/kr/fs v0.1.0/go.mod h1:FFnZGqtBN9Gxj7eW1uZ42v5BccTP0vu6NEaFoC2HwRg=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=\ngithub.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=\ngithub.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=\ngithub.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=\ngithub.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=\ngithub.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=\ngithub.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=\ngithub.com/kulti/thelper v0.6.2 h1:K4xulKkwOCnT1CDms6Ex3uG1dvSMUUQe9zxgYQgbRXs=\ngithub.com/kulti/thelper v0.6.2/go.mod h1:DsqKShOvP40epevkFrvIwkCMNYxMeTNjdWL4dqWHZ6I=\ngithub.com/kunwardeep/paralleltest v1.0.3 h1:UdKIkImEAXjR1chUWLn+PNXqWUGs//7tzMeWuP7NhmI=\ngithub.com/kunwardeep/paralleltest v1.0.3/go.mod h1:vLydzomDFpk7yu5UX02RmP0H8QfRPOV/oFhWN85Mjb4=\ngithub.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw=\ngithub.com/kyoh86/exportloopref v0.1.8 h1:5Ry/at+eFdkX9Vsdw3qU4YkvGtzuVfzT4X7S77LoN/M=\ngithub.com/kyoh86/exportloopref v0.1.8/go.mod h1:1tUcJeiioIs7VWe5gcOObrux3lb66+sBqGZrRkMwPgg=\ngithub.com/ldez/gomoddirectives v0.2.3 h1:y7MBaisZVDYmKvt9/l1mjNCiSA1BVn34U0ObUcJwlhA=\ngithub.com/ldez/gomoddirectives v0.2.3/go.mod h1:cpgBogWITnCfRq2qGoDkKMEVSaarhdBr6g8G04uz6d0=\ngithub.com/ldez/tagliatelle v0.3.1 h1:3BqVVlReVUZwafJUwQ+oxbx2BEX2vUG4Yu/NOfMiKiM=\ngithub.com/ldez/tagliatelle v0.3.1/go.mod h1:8s6WJQwEYHbKZDsp/LjArytKOG8qaMrKQQ3mFukHs88=\ngithub.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=\ngithub.com/leonklingele/grouper v1.1.0 h1:tC2y/ygPbMFSBOs3DcyaEMKnnwH7eYKzohOtRrf0SAg=\ngithub.com/leonklingele/grouper v1.1.0/go.mod h1:uk3I3uDfi9B6PeUjsCKi6ndcf63Uy7snXgR4yDYQVDY=\ngithub.com/letsencrypt/pkcs11key/v4 v4.0.0/go.mod h1:EFUvBDay26dErnNb70Nd0/VW3tJiIbETBPTl9ATXQag=\ngithub.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=\ngithub.com/lib/pq v1.8.0/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/lib/pq v1.9.0/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/lib/pq v1.10.0/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/lib/pq v1.10.1/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/lib/pq v1.10.4/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=\ngithub.com/lufeee/execinquery v1.2.1 h1:hf0Ems4SHcUGBxpGN7Jz78z1ppVkP/837ZlETPCEtOM=\ngithub.com/lufeee/execinquery v1.2.1/go.mod h1:EC7DrEKView09ocscGHC+apXMIaorh4xqSxS/dy8SbM=\ngithub.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=\ngithub.com/magiconair/properties v1.8.1/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=\ngithub.com/magiconair/properties v1.8.4/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60=\ngithub.com/magiconair/properties v1.8.6/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60=\ngithub.com/maratori/testpackage v1.0.1 h1:QtJ5ZjqapShm0w5DosRjg0PRlSdAdlx+W6cCKoALdbQ=\ngithub.com/maratori/testpackage v1.0.1/go.mod h1:ddKdw+XG0Phzhx8BFDTKgpWP4i7MpApTE5fXSKAqwDU=\ngithub.com/matoous/godox v0.0.0-20210227103229-6504466cf951 h1:pWxk9e//NbPwfxat7RXkts09K+dEBJWakUWwICVqYbA=\ngithub.com/matoous/godox v0.0.0-20210227103229-6504466cf951/go.mod h1:1BELzlh859Sh1c6+90blK8lbYy0kwQf1bYlBhBysy1s=\ngithub.com/matryer/is v1.4.0 h1:sosSmIWwkYITGrxZ25ULNDeKiMNzFSr4V/eqBQP0PeE=\ngithub.com/matryer/is v1.4.0/go.mod h1:8I/i5uYgLzgsgEloJE1U6xx5HkBQpAZvepWuujKwMRU=\ngithub.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=\ngithub.com/mattn/go-colorable v0.1.1/go.mod h1:FuOcm+DKB9mbwrcAfNl7/TZVBZ6rcnceauSikq3lYCQ=\ngithub.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=\ngithub.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=\ngithub.com/mattn/go-colorable v0.1.6/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=\ngithub.com/mattn/go-colorable v0.1.8/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=\ngithub.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=\ngithub.com/mattn/go-colorable v0.1.12/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=\ngithub.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=\ngithub.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=\ngithub.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=\ngithub.com/mattn/go-ieproxy v0.0.1 h1:qiyop7gCflfhwCzGyeT0gro3sF9AIg9HU98JORTkqfI=\ngithub.com/mattn/go-ieproxy v0.0.1/go.mod h1:pYabZ6IHcRpFh7vIaLfK7rdcWgFEb3SFJ6/gNWuh88E=\ngithub.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4=\ngithub.com/mattn/go-isatty v0.0.4/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4=\ngithub.com/mattn/go-isatty v0.0.5/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=\ngithub.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=\ngithub.com/mattn/go-isatty v0.0.10/go.mod h1:qgIWMr58cqv1PHHyhnkY9lrL7etaEgOFcMEpPG5Rm84=\ngithub.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOAqxQCu2WE=\ngithub.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=\ngithub.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=\ngithub.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=\ngithub.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=\ngithub.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=\ngithub.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=\ngithub.com/mattn/go-runewidth v0.0.2/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=\ngithub.com/mattn/go-runewidth v0.0.4/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=\ngithub.com/mattn/go-runewidth v0.0.6/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=\ngithub.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0=\ngithub.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=\ngithub.com/mattn/go-shellwords v1.0.10/go.mod h1:EZzvwXDESEeg03EKmM+RmDnNOPKG4lLtQsUlTZDWQ8Y=\ngithub.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 h1:I0XW9+e1XWDxdcEniV4rQAIOPUGDq67JSCiRCgGCZLI=\ngithub.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=\ngithub.com/mbilski/exhaustivestruct v1.2.0 h1:wCBmUnSYufAHO6J4AVWY6ff+oxWxsVFrwgOdMUQePUo=\ngithub.com/mbilski/exhaustivestruct v1.2.0/go.mod h1:OeTBVxQWoEmB2J2JCHmXWPJ0aksxSUOUy+nvtVEfzXc=\ngithub.com/mgechev/dots v0.0.0-20210922191527-e955255bf517 h1:zpIH83+oKzcpryru8ceC6BxnoG8TBrhgAvRg8obzup0=\ngithub.com/mgechev/dots v0.0.0-20210922191527-e955255bf517/go.mod h1:KQ7+USdGKfpPjXk4Ga+5XxQM4Lm4e3gAogrreFAYpOg=\ngithub.com/mgechev/revive v1.3.2 h1:Wb8NQKBaALBJ3xrrj4zpwJwqwNA6nDpyJSEQWcCka6U=\ngithub.com/mgechev/revive v1.3.2/go.mod h1:UCLtc7o5vg5aXCwdUTU1kEBQ1v+YXPAkYDIDXbrs5I0=\ngithub.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE=\ngithub.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=\ngithub.com/miekg/dns v1.1.26/go.mod h1:bPDLeHnStXmXAq1m/Ch/hvfNHr14JKNPMBo3VZKjuso=\ngithub.com/miekg/dns v1.1.35/go.mod h1:KNUDUusw/aVsxyTYZM1oqvCicbwhgbNgztCETuNZ7xM=\ngithub.com/miekg/dns v1.1.41/go.mod h1:p6aan82bvRIyn+zDIv9xYNUpwa73JcSh9BKwknJysuI=\ngithub.com/miekg/pkcs11 v1.0.2/go.mod h1:XsNlhZGX73bx86s2hdc/FuaLm2CPZJemRLMA+WTFxgs=\ngithub.com/miekg/pkcs11 v1.0.3/go.mod h1:XsNlhZGX73bx86s2hdc/FuaLm2CPZJemRLMA+WTFxgs=\ngithub.com/mitchellh/cli v1.0.0/go.mod h1:hNIlj7HEI86fIcpObd7a0FcrxTWetlwJDGcceTlRvqc=\ngithub.com/mitchellh/cli v1.1.0/go.mod h1:xcISNoH86gajksDmfB23e/pu+B+GeFRMYmoHXxx3xhI=\ngithub.com/mitchellh/copystructure v1.0.0/go.mod h1:SNtv71yrdKgLRyLFxmLdkAbkKEFWgYaq1OVrnRcwhnw=\ngithub.com/mitchellh/copystructure v1.1.2/go.mod h1:EBArHfARyrSWO/+Wyr9zwEkc6XMFB9XyNgFNmRkZZU4=\ngithub.com/mitchellh/copystructure v1.2.0 h1:vpKXTN4ewci03Vljg/q9QvCGUDttBOGBIa15WveJJGw=\ngithub.com/mitchellh/copystructure v1.2.0/go.mod h1:qLl+cE2AmVv+CoeAwDPye/v+N2HKCj9FbZEVFJRxO9s=\ngithub.com/mitchellh/go-homedir v1.0.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=\ngithub.com/mitchellh/go-homedir v1.1.0 h1:lukF9ziXFxDFPkA1vsr5zpc1XuPDn/wFntq5mG+4E0Y=\ngithub.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=\ngithub.com/mitchellh/go-testing-interface v1.0.0/go.mod h1:kRemZodwjscx+RGhAo8eIhFbs2+BFgRtFPeD/KE+zxI=\ngithub.com/mitchellh/gox v0.4.0/go.mod h1:Sd9lOJ0+aimLBi73mGofS1ycjY8lL3uZM3JPS42BGNg=\ngithub.com/mitchellh/iochan v1.0.0/go.mod h1:JwYml1nuB7xOzsp52dPpHFffvOCDupsG0QubkSMEySY=\ngithub.com/mitchellh/mapstructure v0.0.0-20160808181253-ca63d7c062ee/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=\ngithub.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=\ngithub.com/mitchellh/mapstructure v1.3.3/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=\ngithub.com/mitchellh/mapstructure v1.4.1/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=\ngithub.com/mitchellh/mapstructure v1.4.3/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=\ngithub.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=\ngithub.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=\ngithub.com/mitchellh/reflectwalk v1.0.0/go.mod h1:mSTlrgnPZtwu0c4WaC2kGObEpuNDbx0jmZXqmk4esnw=\ngithub.com/mitchellh/reflectwalk v1.0.1/go.mod h1:mSTlrgnPZtwu0c4WaC2kGObEpuNDbx0jmZXqmk4esnw=\ngithub.com/mitchellh/reflectwalk v1.0.2 h1:G2LzWKi524PWgd3mLHV8Y5k7s6XUvT0Gef6zxSIeXaQ=\ngithub.com/mitchellh/reflectwalk v1.0.2/go.mod h1:mSTlrgnPZtwu0c4WaC2kGObEpuNDbx0jmZXqmk4esnw=\ngithub.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=\ngithub.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=\ngithub.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826/go.mod h1:TaXosZuwdSHYgviHp1DAtfrULt5eUgsSMsZf+YrPgl8=\ngithub.com/moricho/tparallel v0.2.1 h1:95FytivzT6rYzdJLdtfn6m1bfFJylOJK41+lgv/EHf4=\ngithub.com/moricho/tparallel v0.2.1/go.mod h1:fXEIZxG2vdfl0ZF8b42f5a78EhjjD5mX8qUplsoSU4k=\ngithub.com/mozilla/scribe v0.0.0-20180711195314-fb71baf557c1/go.mod h1:FIczTrinKo8VaLxe6PWTPEXRXDIHz2QAwiaBaP5/4a8=\ngithub.com/mozilla/tls-observatory v0.0.0-20210609171429-7bc42856d2e5/go.mod h1:FUqVoUPHSEdDR0MnFM3Dh8AU0pZHLXUD127SAJGER/s=\ngithub.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=\ngithub.com/mwitkow/go-proto-validators v0.0.0-20180403085117-0950a7990007/go.mod h1:m2XC9Qq0AlmmVksL6FktJCdTYyLk7V3fKyp0sl1yWQo=\ngithub.com/mwitkow/go-proto-validators v0.2.0/go.mod h1:ZfA1hW+UH/2ZHOWvQ3HnQaU0DtnpXu850MZiy+YUgcc=\ngithub.com/nakabonne/nestif v0.3.1 h1:wm28nZjhQY5HyYPx+weN3Q65k6ilSBxDb8v5S81B81U=\ngithub.com/nakabonne/nestif v0.3.1/go.mod h1:9EtoZochLn5iUprVDmDjqGKPofoUEBL8U4Ngq6aY7OE=\ngithub.com/nbutton23/zxcvbn-go v0.0.0-20210217022336-fa2cb2858354 h1:4kuARK6Y6FxaNu/BnU2OAaLF86eTVhP2hjTB6iMvItA=\ngithub.com/nbutton23/zxcvbn-go v0.0.0-20210217022336-fa2cb2858354/go.mod h1:KSVJerMDfblTH7p5MZaTt+8zaT2iEk3AkVb9PQdZuE8=\ngithub.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=\ngithub.com/nishanths/exhaustive v0.7.11 h1:xV/WU3Vdwh5BUH4N06JNUznb6d5zhRPOnlgCrpNYNKA=\ngithub.com/nishanths/exhaustive v0.7.11/go.mod h1:gX+MP7DWMKJmNa1HfMozK+u04hQd3na9i0hyqf3/dOI=\ngithub.com/nishanths/predeclared v0.0.0-20190419143655-18a43bb90ffc/go.mod h1:62PewwiQTlm/7Rj+cxVYqZvDIUc+JjZq6GHAC1fsObQ=\ngithub.com/nishanths/predeclared v0.2.2 h1:V2EPdZPliZymNAn79T8RkNApBjMmVKh5XRpLm/w98Vk=\ngithub.com/nishanths/predeclared v0.2.2/go.mod h1:RROzoN6TnGQupbC+lqggsOlcgysk3LMK/HI84Mp280c=\ngithub.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A=\ngithub.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=\ngithub.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U=\ngithub.com/olekukonko/tablewriter v0.0.0-20170122224234-a0225b3f23b5/go.mod h1:vsDQFd/mU46D+Z4whnwzcISnGGzXWMclvtLoiIKAKIo=\ngithub.com/olekukonko/tablewriter v0.0.1/go.mod h1:vsDQFd/mU46D+Z4whnwzcISnGGzXWMclvtLoiIKAKIo=\ngithub.com/olekukonko/tablewriter v0.0.2/go.mod h1:rSAaSIOAGT9odnlyGlUfAJaoc5w2fSBUmeGDbRWPxyQ=\ngithub.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec=\ngithub.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY=\ngithub.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=\ngithub.com/onsi/ginkgo v1.10.3/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=\ngithub.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk=\ngithub.com/onsi/ginkgo v1.16.4 h1:29JGrr5oVBm5ulCWet69zQkzWipVXIol6ygQUe/EzNc=\ngithub.com/onsi/ginkgo v1.16.4/go.mod h1:dX+/inL/fNMqNlz0e9LfyB9TswhZpCVdJM/Z6Vvnwo0=\ngithub.com/onsi/ginkgo/v2 v2.0.0/go.mod h1:vw5CSIxN1JObi/U8gcbwft7ZxR2dgaR70JSE3/PpL4c=\ngithub.com/onsi/ginkgo/v2 v2.1.3 h1:e/3Cwtogj0HA+25nMP1jCMDIf8RtRYbGwGGuBIFztkc=\ngithub.com/onsi/ginkgo/v2 v2.1.3/go.mod h1:vw5CSIxN1JObi/U8gcbwft7ZxR2dgaR70JSE3/PpL4c=\ngithub.com/onsi/gomega v1.5.0/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=\ngithub.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY=\ngithub.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo=\ngithub.com/onsi/gomega v1.17.0/go.mod h1:HnhC7FXeEQY45zxNK3PPoIUhzk/80Xly9PcubAlGdZY=\ngithub.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=\ngithub.com/onsi/gomega v1.18.1/go.mod h1:0q+aL8jAiMXy9hbwj2mr5GziHiwhAIQpFmmtT5hitRs=\ngithub.com/op/go-logging v0.0.0-20160315200505-970db520ece7/go.mod h1:HzydrMdWErDVzsI23lYNej1Htcns9BCg93Dk0bBINWk=\ngithub.com/opentracing/opentracing-go v1.0.2/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=\ngithub.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=\ngithub.com/opentracing/opentracing-go v1.2.0 h1:uEJPy/1a5RIPAJ0Ov+OIO8OxWu77jEv+1B0VhjKrZUs=\ngithub.com/opentracing/opentracing-go v1.2.0/go.mod h1:GxEUsuufX4nBwe+T+Wl9TAgYrxe9dPLANfrWvHYVTgc=\ngithub.com/otiai10/copy v1.2.0 h1:HvG945u96iNadPoG2/Ja2+AUJeW5YuFQMixq9yirC+k=\ngithub.com/otiai10/copy v1.2.0/go.mod h1:rrF5dJ5F0t/EWSYODDu4j9/vEeYHMkc8jt0zJChqQWw=\ngithub.com/otiai10/curr v0.0.0-20150429015615-9b4961190c95/go.mod h1:9qAhocn7zKJG+0mI8eUu6xqkFDYS2kb2saOteoSB3cE=\ngithub.com/otiai10/curr v1.0.0/go.mod h1:LskTG5wDwr8Rs+nNQ+1LlxRjAtTZZjtJW4rMXl6j4vs=\ngithub.com/otiai10/mint v1.3.0/go.mod h1:F5AjcsTsWUqX+Na9fpHb52P8pcRX2CI6A3ctIT91xUo=\ngithub.com/otiai10/mint v1.3.1/go.mod h1:/yxELlJQ0ufhjUwhshSj+wFjZ78CnZ48/1wtmBH1OTc=\ngithub.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=\ngithub.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=\ngithub.com/pascaldekloe/name v1.0.0 h1:n7LKFgHixETzxpRv2R77YgPUFo85QHGZKrdaYm7eY5U=\ngithub.com/pascaldekloe/name v1.0.0/go.mod h1:Z//MfYJnH4jVpQ9wkclwu2I2MkHmXTlT9wR5UZScttM=\ngithub.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=\ngithub.com/pelletier/go-toml v1.8.1/go.mod h1:T2/BmBdy8dvIRq1a/8aqjN41wvWlN4lrapLU/GW4pbc=\ngithub.com/pelletier/go-toml v1.9.4/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=\ngithub.com/pelletier/go-toml v1.9.5/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=\ngithub.com/pelletier/go-toml/v2 v2.0.0-beta.8/go.mod h1:r9LEWfGN8R5k0VXJ+0BkIe7MYkRdwZOjgMj2KwnJFUo=\ngithub.com/pelletier/go-toml/v2 v2.0.0/go.mod h1:r9LEWfGN8R5k0VXJ+0BkIe7MYkRdwZOjgMj2KwnJFUo=\ngithub.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=\ngithub.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=\ngithub.com/peterbourgon/diskv v2.0.1+incompatible/go.mod h1:uqqh8zWWbv1HBMNONnaR/tNboyR3/BZd58JJSHlUSCU=\ngithub.com/phayes/checkstyle v0.0.0-20170904204023-bfd46e6a821d h1:CdDQnGF8Nq9ocOS/xlSptM1N3BbrA6/kmaep5ggwaIA=\ngithub.com/phayes/checkstyle v0.0.0-20170904204023-bfd46e6a821d/go.mod h1:3OzsM7FXDQlpCiw2j81fOmAwQLnZnLGXVKUzeKQXIAw=\ngithub.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=\ngithub.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkg/sftp v1.10.1/go.mod h1:lYOWFsE0bwd1+KfKJaKeuokY15vzFx25BLbzYYoAxZI=\ngithub.com/pkg/sftp v1.13.1/go.mod h1:3HaPG6Dq1ILlpPZRO0HVMrsydcdLt6HRDccSgb87qRg=\ngithub.com/planetscale/vtprotobuf v0.6.1-0.20240319094008-0393e58bdf10 h1:GFCKgmp0tecUJ0sJuv4pzYCqS9+RGSn52M3FUwPs+uo=\ngithub.com/planetscale/vtprotobuf v0.6.1-0.20240319094008-0393e58bdf10/go.mod h1:t/avpk3KcrXxUnYOhZhMXJlSEyie6gQbtLq5NM3loB8=\ngithub.com/pmezard/go-difflib v0.0.0-20151028094244-d8ed2627bdf0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=\ngithub.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/polyfloyd/go-errorlint v1.0.0 h1:pDrQG0lrh68e602Wfp68BlUTRFoHn8PZYAjLgt2LFsM=\ngithub.com/polyfloyd/go-errorlint v1.0.0/go.mod h1:KZy4xxPJyy88/gldCe5OdW6OQRtNO3EZE7hXzmnebgA=\ngithub.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI=\ngithub.com/posener/complete v1.2.3/go.mod h1:WZIdtGGp+qx0sLrYKtIRAruyNpv6hFCicSgv7Sy7s/s=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a/go.mod h1:lzZQ3Noex5pfAy7mkAeCjcBDteYU85uWWnJ/y6gKU8k=\ngithub.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=\ngithub.com/prometheus/client_golang v0.9.3/go.mod h1:/TN21ttK/J9q6uSwhBd54HahCDft0ttaMvbicHlPoso=\ngithub.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo=\ngithub.com/prometheus/client_golang v1.3.0/go.mod h1:hJaj2vgQTGQmVCsAACORcieXFeDPbaTKGT+JTgUa3og=\ngithub.com/prometheus/client_golang v1.4.0/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=\ngithub.com/prometheus/client_golang v1.4.1/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=\ngithub.com/prometheus/client_golang v1.7.1/go.mod h1:PY5Wy2awLA44sXw4AOSfFBetzPP4j5+D6mVACh+pe2M=\ngithub.com/prometheus/client_golang v1.11.0/go.mod h1:Z6t4BnS23TR94PD6BsDNk8yVqroYurpAkEiz0P2BEV0=\ngithub.com/prometheus/client_golang v1.12.1 h1:ZiaPsmm9uiBeaSMRznKsCDNtPCS0T3JVDGF+06gjBzk=\ngithub.com/prometheus/client_golang v1.12.1/go.mod h1:3Z9XVyYiZYEO+YQWt3RD2R3jrbd179Rt297l4aS6nDY=\ngithub.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=\ngithub.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.1.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/prometheus/client_model v0.6.0 h1:k1v3CzpSRUTrKMppY35TLwPvxHqBu0bYgxZzqGIgaos=\ngithub.com/prometheus/client_model v0.6.0/go.mod h1:NTQHnmxFpouOD0DpvP4XujX3CdOAGQPoaGhyTchlyt8=\ngithub.com/prometheus/common v0.0.0-20181113130724-41aa239b4cce/go.mod h1:daVV7qP5qjZbuso7PdcryaAu0sAZbrN9i7WWcTMWvro=\ngithub.com/prometheus/common v0.4.0/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=\ngithub.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=\ngithub.com/prometheus/common v0.7.0/go.mod h1:DjGbpBbp5NYNiECxcL/VnbXCCaQpKd3tt26CguLLsqA=\ngithub.com/prometheus/common v0.8.0/go.mod h1:PC/OgXc+UN7B4ALwvn1yzVZmVwvhXp5JsbBv6wSv6i0=\ngithub.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4=\ngithub.com/prometheus/common v0.10.0/go.mod h1:Tlit/dnDKsSWFlCLTWaA1cyBgKHSMdTB80sz/V91rCo=\ngithub.com/prometheus/common v0.26.0/go.mod h1:M7rCNAaPfAosfx8veZJCuw84e35h3Cfd9VFqTh1DIvc=\ngithub.com/prometheus/common v0.32.1 h1:hWIdL3N2HoUx3B8j3YN9mWor0qhY/NlEKZEaXxuIRh4=\ngithub.com/prometheus/common v0.32.1/go.mod h1:vu+V0TpY+O6vW9J44gczi3Ap/oXXR10b+M/gUGO4Hls=\ngithub.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=\ngithub.com/prometheus/procfs v0.0.0-20190507164030-5867b95ac084/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=\ngithub.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=\ngithub.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.0.9/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=\ngithub.com/prometheus/procfs v0.1.3/go.mod h1:lV6e/gmhEcM9IjHGsFOCxxuZ+z1YqCvr4OA4YeYWdaU=\ngithub.com/prometheus/procfs v0.6.0/go.mod h1:cz+aTbrPOrUb4q7XlbU9ygM+/jj0fzG6c1xBZuNvfVA=\ngithub.com/prometheus/procfs v0.7.3 h1:4jVXhlkAyzOScmCkXBTOLRLTz8EeU+eyjrwB/EPq0VU=\ngithub.com/prometheus/procfs v0.7.3/go.mod h1:cz+aTbrPOrUb4q7XlbU9ygM+/jj0fzG6c1xBZuNvfVA=\ngithub.com/prometheus/tsdb v0.7.1/go.mod h1:qhTCs0VvXwvX/y3TZrWD7rabWM+ijKTux40TwIPHuXU=\ngithub.com/pseudomuto/protoc-gen-doc v1.3.2/go.mod h1:y5+P6n3iGrbKG+9O04V5ld71in3v/bX88wUwgt+U8EA=\ngithub.com/pseudomuto/protokit v0.2.0/go.mod h1:2PdH30hxVHsup8KpBTOXTBeMVhJZVio3Q8ViKSAXT0Q=\ngithub.com/quasilyte/go-ruleguard v0.3.1-0.20210203134552-1b5a410e1cc8/go.mod h1:KsAh3x0e7Fkpgs+Q9pNLS5XpFSvYCEVl5gP9Pp1xp30=\ngithub.com/quasilyte/go-ruleguard v0.3.16-0.20220213074421-6aa060fab41a h1:sWFavxtIctGrVs5SYZ5Ml1CvrDAs8Kf5kx2PI3C41dA=\ngithub.com/quasilyte/go-ruleguard v0.3.16-0.20220213074421-6aa060fab41a/go.mod h1:VMX+OnnSw4LicdiEGtRSD/1X8kW7GuEscjYNr4cOIT4=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.0/go.mod h1:KeCP03KrjuSO0H1kTuZQCWlQPulDV6YMIXmpQss17rU=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.16/go.mod h1:KeCP03KrjuSO0H1kTuZQCWlQPulDV6YMIXmpQss17rU=\ngithub.com/quasilyte/go-ruleguard/rules v0.0.0-20201231183845-9e62ed36efe1/go.mod h1:7JTjp89EGyU1d6XfBiXihJNG37wB2VRkd125Q1u7Plc=\ngithub.com/quasilyte/go-ruleguard/rules v0.0.0-20211022131956-028d6511ab71/go.mod h1:4cgAphtvu7Ftv7vOT2ZOYhC6CvBxZixcasr8qIOTA50=\ngithub.com/quasilyte/gogrep v0.0.0-20220120141003-628d8b3623b5 h1:PDWGei+Rf2bBiuZIbZmM20J2ftEy9IeUCHA8HbQqed8=\ngithub.com/quasilyte/gogrep v0.0.0-20220120141003-628d8b3623b5/go.mod h1:wSEyW6O61xRV6zb6My3HxrQ5/8ke7NE2OayqCHa3xRM=\ngithub.com/quasilyte/regex/syntax v0.0.0-20200407221936-30656e2c4a95 h1:L8QM9bvf68pVdQ3bCFZMDmnt9yqcMBro1pC7F+IPYMY=\ngithub.com/quasilyte/regex/syntax v0.0.0-20200407221936-30656e2c4a95/go.mod h1:rlzQ04UMyJXu/aOvhd8qT+hvDrFpiwqp8MRXDY9szc0=\ngithub.com/quasilyte/stdinfo v0.0.0-20220114132959-f7386bf02567 h1:M8mH9eK4OUR4lu7Gd+PU1fV2/qnDNfzT635KRSObncs=\ngithub.com/quasilyte/stdinfo v0.0.0-20220114132959-f7386bf02567/go.mod h1:DWNGW8A4Y+GyBgPuaQJuWiy0XYftx4Xm/y5Jqk9I6VQ=\ngithub.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=\ngithub.com/rogpeppe/fastuuid v1.1.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=\ngithub.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=\ngithub.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=\ngithub.com/rogpeppe/go-internal v1.8.1/go.mod h1:JeRgkft04UBgHMgCIwADu4Pn6Mtm5d4nPKWu0nJ5d+o=\ngithub.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=\ngithub.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=\ngithub.com/rs/cors v1.7.0/go.mod h1:gFx+x8UowdsKA9AchylcLynDq+nNFfI8FkUZdN/jGCU=\ngithub.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=\ngithub.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8=\ngithub.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=\ngithub.com/russross/blackfriday v1.5.2/go.mod h1:JO/DiYxRf+HjHt06OyowR9PTA263kcR/rfWxYHBV53g=\ngithub.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=\ngithub.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=\ngithub.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=\ngithub.com/ryancurrah/gomodguard v1.2.3 h1:ww2fsjqocGCAFamzvv/b8IsRduuHHeK2MHTcTxZTQX8=\ngithub.com/ryancurrah/gomodguard v1.2.3/go.mod h1:rYbA/4Tg5c54mV1sv4sQTP5WOPBcoLtnBZ7/TEhXAbg=\ngithub.com/ryanrolds/sqlclosecheck v0.3.0 h1:AZx+Bixh8zdUBxUA1NxbxVAS78vTPq4rCb8OUZI9xFw=\ngithub.com/ryanrolds/sqlclosecheck v0.3.0/go.mod h1:1gREqxyTGR3lVtpngyFo3hZAgk0KCtEdgEkHwDbigdA=\ngithub.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts=\ngithub.com/sagikazarmark/crypt v0.5.0/go.mod h1:l+nzl7KWh51rpzp2h7t4MZWyiEWdhNpOAnclKvg+mdA=\ngithub.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsFaodPcyo=\ngithub.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af/go.mod h1:Vrkh1pnjV9Bl8c3P9zH0/D4NlOHWP5d4/hF4YTULaec=\ngithub.com/sanposhiho/wastedassign/v2 v2.0.6 h1:+6/hQIHKNJAUixEj6EmOngGIisyeI+T3335lYTyxRoA=\ngithub.com/sanposhiho/wastedassign/v2 v2.0.6/go.mod h1:KyZ0MWTwxxBmfwn33zh3k1dmsbF2ud9pAAGfoLfjhtI=\ngithub.com/sassoftware/go-rpmutils v0.0.0-20190420191620-a8f1baeba37b/go.mod h1:am+Fp8Bt506lA3Rk3QCmSqmYmLMnPDhdDUcosQCAx+I=\ngithub.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=\ngithub.com/securego/gosec/v2 v2.11.0 h1:+PDkpzR41OI2jrw1q6AdXZCbsNGNGT7pQjal0H0cArI=\ngithub.com/securego/gosec/v2 v2.11.0/go.mod h1:SX8bptShuG8reGC0XS09+a4H2BoWSJi+fscA+Pulbpo=\ngithub.com/sergi/go-diff v1.0.0/go.mod h1:0CfEIISq7TuYL3j771MWULgwwjU+GofnZX9QAmXWZgo=\ngithub.com/sergi/go-diff v1.1.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=\ngithub.com/sergi/go-diff v1.2.0 h1:XU+rvMAioB0UC3q1MFrIQy4Vo5/4VsRDQQXHsEya6xQ=\ngithub.com/sergi/go-diff v1.2.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=\ngithub.com/shazow/go-diff v0.0.0-20160112020656-b6b7b6733b8c h1:W65qqJCIOVP4jpqPQ0YvHYKwcMEMVWIzWC5iNQQfBTU=\ngithub.com/shazow/go-diff v0.0.0-20160112020656-b6b7b6733b8c/go.mod h1:/PevMnwAxekIXwN8qQyfc5gl2NlkB3CQlkizAbOkeBs=\ngithub.com/shopspring/decimal v1.2.0 h1:abSATXmQEYyShuxI4/vyW3tV1MrKAJzCZ/0zLUXYbsQ=\ngithub.com/shopspring/decimal v1.2.0/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o=\ngithub.com/shurcooL/go v0.0.0-20180423040247-9e1955d9fb6e/go.mod h1:TDJrrUr11Vxrven61rcy3hJMUqaf/CLWYhHNPmT14Lk=\ngithub.com/shurcooL/go-goon v0.0.0-20170922171312-37c2f522c041/go.mod h1:N5mDOmsrJOB+vfqUK+7DmDyjhSLIIBnXo9lvZJj3MWQ=\ngithub.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=\ngithub.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=\ngithub.com/sirupsen/logrus v1.4.1/go.mod h1:ni0Sbl8bgC9z8RoU9G6nDWqqs/fq4eDPysMBDgk/93Q=\ngithub.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=\ngithub.com/sirupsen/logrus v1.6.0/go.mod h1:7uNnSEd1DgxDLC74fIahvMZmmYsHGZGEOFrfsX/uA88=\ngithub.com/sirupsen/logrus v1.7.0/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=\ngithub.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=\ngithub.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=\ngithub.com/sivchari/containedctx v1.0.2 h1:0hLQKpgC53OVF1VT7CeoFHk9YKstur1XOgfYIc1yrHI=\ngithub.com/sivchari/containedctx v1.0.2/go.mod h1:PwZOeqm4/DLoJOqMSIJs3aKqXRX4YO+uXww087KZ7Bw=\ngithub.com/sivchari/tenv v1.5.0 h1:wxW0mFpKI6DIb3s6m1jCDYvkWXCskrimXMuGd0K/kSQ=\ngithub.com/sivchari/tenv v1.5.0/go.mod h1:64yStXKSOxDfX47NlhVwND4dHwfZDdbp2Lyl018Icvg=\ngithub.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=\ngithub.com/smartystreets/assertions v1.0.0/go.mod h1:kHHU4qYBaI3q23Pp3VPrmWhuIUrLW/7eUrw0BU5VaoM=\ngithub.com/smartystreets/assertions v1.2.0 h1:42S6lae5dvLc7BrLu/0ugRtcFVjoJNMC/N3yZFZkDFs=\ngithub.com/smartystreets/assertions v1.2.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=\ngithub.com/smartystreets/go-aws-auth v0.0.0-20180515143844-0c1422d1fdb9/go.mod h1:SnhjPscd9TpLiy1LpzGSKh3bXCfxxXuqd9xmQJy3slM=\ngithub.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=\ngithub.com/smartystreets/goconvey v1.7.2 h1:9RBaZCeXEQ3UselpuwUQHltGVXvdwm6cv1hgR6gDIPg=\ngithub.com/smartystreets/goconvey v1.7.2/go.mod h1:Vw0tHAZW6lzCRk3xgdin6fKYcG+G3Pg9vgXWeJpQFMM=\ngithub.com/smartystreets/gunit v1.0.0/go.mod h1:qwPWnhz6pn0NnRBP++URONOVyNkPyr4SauJk4cUOwJs=\ngithub.com/soheilhy/cmux v0.1.4/go.mod h1:IM3LyeVVIOuxMH7sFAkER9+bJ4dT7Ms6E4xg4kGIyLM=\ngithub.com/sonatard/noctx v0.0.1 h1:VC1Qhl6Oxx9vvWo3UDgrGXYCeKCe3Wbw7qAWL6FrmTY=\ngithub.com/sonatard/noctx v0.0.1/go.mod h1:9D2D/EoULe8Yy2joDHJj7bv3sZoq9AaSb8B4lqBjiZI=\ngithub.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=\ngithub.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=\ngithub.com/sourcegraph/go-diff v0.6.1 h1:hmA1LzxW0n1c3Q4YbrFgg4P99GSnebYa3x8gr0HZqLQ=\ngithub.com/sourcegraph/go-diff v0.6.1/go.mod h1:iBszgVvyxdc8SFZ7gm69go2KDdt3ag071iBaWPF6cjs=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=\ngithub.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ=\ngithub.com/spf13/afero v1.4.1/go.mod h1:Ai8FlHk4v/PARR026UzYexafAt9roJ7LcLMAmO6Z93I=\ngithub.com/spf13/afero v1.8.2/go.mod h1:CtAatgMJh6bJEIs48Ay/FOnkljP3WeGUG0MC1RfAqwo=\ngithub.com/spf13/afero v1.12.0 h1:UcOPyRBYczmFn6yvphxkn9ZEOY65cpwGKb5mL36mrqs=\ngithub.com/spf13/afero v1.12.0/go.mod h1:ZTlWwG4/ahT8W7T0WQ5uYmjI9duaLQGy3Q2OAl4sk/4=\ngithub.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=\ngithub.com/spf13/cast v1.3.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=\ngithub.com/spf13/cast v1.4.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=\ngithub.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y=\ngithub.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=\ngithub.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=\ngithub.com/spf13/cobra v0.0.5/go.mod h1:3K3wKZymM7VvHMDS9+Akkh4K60UwM26emMESw8tLCHU=\ngithub.com/spf13/cobra v1.1.1/go.mod h1:WnodtKOvamDL/PwE2M4iKs8aMDBZ5Q5klgD3qfVJQMI=\ngithub.com/spf13/cobra v1.1.3/go.mod h1:pGADOWyqRD/YMrPZigI/zbliZ2wVD/23d+is3pSWzOo=\ngithub.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=\ngithub.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=\ngithub.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo=\ngithub.com/spf13/jwalterweatherman v1.1.0/go.mod h1:aNWZUN0dPAAO/Ljvb5BEdw96iTZ0EXowPYD95IqWIGo=\ngithub.com/spf13/pflag v1.0.1/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=\ngithub.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=\ngithub.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=\ngithub.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=\ngithub.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=\ngithub.com/spf13/viper v1.3.2/go.mod h1:ZiWeW+zYFKm7srdB9IoDzzZXaJaI5eL9QjNiN/DMA2s=\ngithub.com/spf13/viper v1.7.0/go.mod h1:8WkrPz2fc9jxqZNCJI/76HCieCp4Q8HaLFoCha5qpdg=\ngithub.com/spf13/viper v1.7.1/go.mod h1:8WkrPz2fc9jxqZNCJI/76HCieCp4Q8HaLFoCha5qpdg=\ngithub.com/spf13/viper v1.11.0/go.mod h1:djo0X/bA5+tYVoCn+C7cAYJGcVn/qYLFTG8gdUsX7Zk=\ngithub.com/spf13/viper v1.20.0 h1:zrxIyR3RQIOsarIrgL8+sAvALXul9jeEPa06Y0Ph6vY=\ngithub.com/spf13/viper v1.20.0/go.mod h1:P9Mdzt1zoHIG8m2eZQinpiBjo6kCmZSKBClNNqjJvu4=\ngithub.com/ssgreg/nlreturn/v2 v2.2.1 h1:X4XDI7jstt3ySqGU86YGAURbxw3oTDPK9sPEi6YEwQ0=\ngithub.com/ssgreg/nlreturn/v2 v2.2.1/go.mod h1:E/iiPB78hV7Szg2YfRgyIrk1AD6JVMTRkkxBiELzh2I=\ngithub.com/stbenjam/no-sprintf-host-port v0.1.1 h1:tYugd/yrm1O0dV+ThCbaKZh195Dfm07ysF0U6JQXczc=\ngithub.com/stbenjam/no-sprintf-host-port v0.1.1/go.mod h1:TLhvtIvONRzdmkFiio4O8LHsN9N74I+PhRquPsxpL0I=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25/go.mod h1:lbP8tGiBjZ5YWIc2fzuRpTaz0b/53vT6PEs3QuAWzuU=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=\ngithub.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=\ngithub.com/stretchr/testify v0.0.0-20170130113145-4d4bfba8f1d1/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.1.4/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=\ngithub.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=\ngithub.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=\ngithub.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=\ngithub.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw=\ngithub.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=\ngithub.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=\ngithub.com/sylvia7788/contextcheck v1.0.4 h1:MsiVqROAdr0efZc/fOCt0c235qm9XJqHtWwM+2h2B04=\ngithub.com/sylvia7788/contextcheck v1.0.4/go.mod h1:vuPKJMQ7MQ91ZTqfdyreNKwZjyUg6KO+IebVyQDedZQ=\ngithub.com/tdakkota/asciicheck v0.1.1 h1:PKzG7JUTUmVspQTDqtkX9eSiLGossXTybutHwTXuO0A=\ngithub.com/tdakkota/asciicheck v0.1.1/go.mod h1:yHp0ai0Z9gUljN3o0xMhYJnH/IcvkdTBOX2fmJ93JEM=\ngithub.com/tenntenn/modver v1.0.1 h1:2klLppGhDgzJrScMpkj9Ujy3rXPUspSjAcev9tSEBgA=\ngithub.com/tenntenn/modver v1.0.1/go.mod h1:bePIyQPb7UeioSRkw3Q0XeMhYZSMx9B8ePqg6SAMGH0=\ngithub.com/tenntenn/text/transform v0.0.0-20200319021203-7eef512accb3 h1:f+jULpRQGxTSkNYKJ51yaw6ChIqO+Je8UqsTKN/cDag=\ngithub.com/tenntenn/text/transform v0.0.0-20200319021203-7eef512accb3/go.mod h1:ON8b8w4BN/kE1EOhwT0o+d62W65a6aPw1nouo9LMgyY=\ngithub.com/tetafro/godot v1.4.11 h1:BVoBIqAf/2QdbFmSwAWnaIqDivZdOV0ZRwEm6jivLKw=\ngithub.com/tetafro/godot v1.4.11/go.mod h1:LR3CJpxDVGlYOWn3ZZg1PgNZdTUvzsZWu8xaEohUpn8=\ngithub.com/timakin/bodyclose v0.0.0-20210704033933-f49887972144 h1:kl4KhGNsJIbDHS9/4U9yQo1UcPQM0kOMJHn29EoH/Ro=\ngithub.com/timakin/bodyclose v0.0.0-20210704033933-f49887972144/go.mod h1:Qimiffbc6q9tBWlVV6x0P9sat/ao1xEkREYPPj9hphk=\ngithub.com/tj/assert v0.0.0-20171129193455-018094318fb0/go.mod h1:mZ9/Rh9oLWpLLDRpvE+3b7gP/C2YyLFYxNmcLnPTMe0=\ngithub.com/tj/assert v0.0.3 h1:Df/BlaZ20mq6kuai7f5z2TvPFiwC3xaWJSDQNiIS3Rk=\ngithub.com/tj/assert v0.0.3/go.mod h1:Ne6X72Q+TB1AteidzQncjw9PabbMp4PBMZ1k+vd1Pvk=\ngithub.com/tj/go-buffer v1.1.0/go.mod h1:iyiJpfFcR2B9sXu7KvjbT9fpM4mOelRSDTbntVj52Uc=\ngithub.com/tj/go-elastic v0.0.0-20171221160941-36157cbbebc2/go.mod h1:WjeM0Oo1eNAjXGDx2yma7uG2XoyRZTq1uv3M/o7imD0=\ngithub.com/tj/go-kinesis v0.0.0-20171128231115-08b17f58cb1b/go.mod h1:/yhzCV0xPfx6jb1bBgRFjl5lytqVqZXEaeqWP8lTEao=\ngithub.com/tj/go-spin v1.1.0/go.mod h1:Mg1mzmePZm4dva8Qz60H2lHwmJ2loum4VIrLgVnKwh4=\ngithub.com/tmc/grpc-websocket-proxy v0.0.0-20170815181823-89b8d40f7ca8/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=\ngithub.com/tmc/grpc-websocket-proxy v0.0.0-20190109142713-0ad062ec5ee5/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=\ngithub.com/tmc/grpc-websocket-proxy v0.0.0-20200427203606-3cfed13b9966/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=\ngithub.com/tomarrell/wrapcheck/v2 v2.6.1 h1:Cf4a/iwuMp9s7kKrh74GTgijRVim0wEpKjgAsT7Wctw=\ngithub.com/tomarrell/wrapcheck/v2 v2.6.1/go.mod h1:Eo+Opt6pyMW1b6cNllOcDSSoHO0aTJ+iF6BfCUbHltA=\ngithub.com/tomasen/realip v0.0.0-20180522021738-f0c99a92ddce/go.mod h1:o8v6yHRoik09Xen7gje4m9ERNah1d1PPsVq1VEx9vE4=\ngithub.com/tommy-muehle/go-mnd/v2 v2.5.0 h1:iAj0a8e6+dXSL7Liq0aXPox36FiN1dBbjA6lt9fl65s=\ngithub.com/tommy-muehle/go-mnd/v2 v2.5.0/go.mod h1:WsUAkMJMYww6l/ufffCD3m+P7LEvr8TnZn9lwVDlgzw=\ngithub.com/tv42/httpunix v0.0.0-20150427012821-b75d8614f926/go.mod h1:9ESjWnEqriFuLhtthL60Sar/7RFoluCcXsuvEwTV5KM=\ngithub.com/twitchtv/twirp v5.8.0+incompatible/go.mod h1:RRJoFSAmTEh2weEqWtpPE3vFK5YBhA6bqp2l1kfCC5A=\ngithub.com/uber-common/bark v1.2.1/go.mod h1:g0ZuPcD7XiExKHynr93Q742G/sbrdVQkghrqLGOoFuY=\ngithub.com/uber-go/mapdecode v1.0.0 h1:euUEFM9KnuCa1OBixz1xM+FIXmpixyay5DLymceOVrU=\ngithub.com/uber-go/mapdecode v1.0.0/go.mod h1:b5nP15FwXTgpjTjeA9A2uTHXV5UJCl4arwKpP0FP1Hw=\ngithub.com/uber-go/tally v3.3.12+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber-go/tally v3.3.15+incompatible h1:9hLSgNBP28CjIaDmAuRTq9qV+UZY+9PcvAkXO4nNMwg=\ngithub.com/uber-go/tally v3.3.15+incompatible/go.mod h1:YDTIBxdXyOU/sCWilKB4bgyufu1cEi0jdVnRdxvjnmU=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible/go.mod h1:WVhlPFC8FDjOFMMWRy2pZqQJSXxYSwNYOkTr/Z6d3Kk=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible/go.mod h1:ComeNDZlWwrWnDv8aPp0Ba6+uUTzImX/AauajbLI56U=\ngithub.com/uber/ringpop-go v0.8.5/go.mod h1:zVI6eGO6L7pG14GkntHsSOfmUAWQ7B4lvmzly4IT4ls=\ngithub.com/uber/tchannel-go v1.22.2 h1:NKA5FVESYh6Ij6V+tujK+IFZnBKDyUHdsBY264UYhgk=\ngithub.com/uber/tchannel-go v1.22.2/go.mod h1:Rrgz1eL8kMjW/nEzZos0t+Heq0O4LhnUJVA32OvWKHo=\ngithub.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=\ngithub.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=\ngithub.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=\ngithub.com/ulikunitz/xz v0.5.7/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=\ngithub.com/ulikunitz/xz v0.5.9/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=\ngithub.com/ulikunitz/xz v0.5.10/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=\ngithub.com/ulikunitz/xz v0.5.11 h1:kpFauv27b6ynzBNT/Xy+1k+fK4WswhN/6PN5WhFAGw8=\ngithub.com/ulikunitz/xz v0.5.11/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=\ngithub.com/ultraware/funlen v0.0.3 h1:5ylVWm8wsNwH5aWo9438pwvsK0QiqVuUrt9bn7S/iLA=\ngithub.com/ultraware/funlen v0.0.3/go.mod h1:Dp4UiAus7Wdb9KUZsYWZEWiRzGuM2kXM1lPbfaF6xhA=\ngithub.com/ultraware/whitespace v0.0.5 h1:hh+/cpIcopyMYbZNVov9iSxvJU3OYQg78Sfaqzi/CzI=\ngithub.com/ultraware/whitespace v0.0.5/go.mod h1:aVMh/gQve5Maj9hQ/hg+F75lr/X5A89uZnzAmWSineA=\ngithub.com/urfave/cli v1.20.0/go.mod h1:70zkFmudgCuE/ngEzBv17Jvp/497gISqfk5gWijbERA=\ngithub.com/urfave/cli v1.22.1/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0=\ngithub.com/uudashr/gocognit v1.0.5 h1:rrSex7oHr3/pPLQ0xoWq108XMU8s678FJcQ+aSfOHa4=\ngithub.com/uudashr/gocognit v1.0.5/go.mod h1:wgYz0mitoKOTysqxTDMOUXg+Jb5SvtihkfmugIZYpEA=\ngithub.com/vektra/mockery/v2 v2.53.5 h1:iktAY68pNiMvLoHxKqlSNSv/1py0QF/17UGrrAMYDI8=\ngithub.com/vektra/mockery/v2 v2.53.5/go.mod h1:hIFFb3CvzPdDJJiU7J4zLRblUMv7OuezWsHPmswriwo=\ngithub.com/viki-org/dnscache v0.0.0-20130720023526-c70c1f23c5d8/go.mod h1:dniwbG03GafCjFohMDmz6Zc6oCuiqgH6tGNyXTkHzXE=\ngithub.com/xanzy/go-gitlab v0.50.0 h1:t7IoYTrnLSbdEZN7d8X/5zcr+ZM4TZQ2mXa8MqWlAZQ=\ngithub.com/xanzy/go-gitlab v0.50.0/go.mod h1:Q+hQhV508bDPoBijv7YjK/Lvlb4PhVhJdKqXVQrUoAE=\ngithub.com/xanzy/ssh-agent v0.2.1/go.mod h1:mLlQY/MoOhWBj+gOGMQkOeiEvkx+8pJSI+0Bx9h2kr4=\ngithub.com/xanzy/ssh-agent v0.3.0/go.mod h1:3s9xbODqPuuhK9JV1R321M/FlMZSBvE5aY6eAcqrDh0=\ngithub.com/xanzy/ssh-agent v0.3.1 h1:AmzO1SSWxw73zxFZPRwaMN1MohDw8UyHnmuxyceTEGo=\ngithub.com/xanzy/ssh-agent v0.3.1/go.mod h1:QIE4lCeL7nkC25x+yA3LBIYfwCc1TFziCtG7cBAac6w=\ngithub.com/xi2/xz v0.0.0-20171230120015-48954b6210f8 h1:nIPpBwaJSVYIxUFsDv3M8ofmx9yWTog9BfvIu0q41lo=\ngithub.com/xi2/xz v0.0.0-20171230120015-48954b6210f8/go.mod h1:HUYIGzjTL3rfEspMxjDjgmT5uz5wzYJKVo23qUhYTos=\ngithub.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2/go.mod h1:UETIi67q53MR2AWcXfiuqkDkRtnGDLqkBTpCHuJHxtU=\ngithub.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778/go.mod h1:2MuV+tbUrU1zIOPMxZ5EncGwgmMJsa+9ucAQZXxsObs=\ngithub.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77/go.mod h1:aYKd//L2LvnjZzWKhF00oedf4jCCReLcmhLdhm1A27Q=\ngithub.com/yagipy/maintidx v1.0.0 h1:h5NvIsCz+nRDapQ0exNv4aJ0yXSI0420omVANTv3GJM=\ngithub.com/yagipy/maintidx v1.0.0/go.mod h1:0qNf/I/CCZXSMhsRsrEPDZ+DkekpKLXAJfsTACwgXLk=\ngithub.com/yeya24/promlinter v0.2.0 h1:xFKDQ82orCU5jQujdaD8stOHiv8UN68BSdn2a8u8Y3o=\ngithub.com/yeya24/promlinter v0.2.0/go.mod h1:u54lkmBOZrpEbQQ6gox2zWKKLKu2SGe+2KOiextY+IA=\ngithub.com/yudai/gojsondiff v1.0.0/go.mod h1:AY32+k2cwILAkW1fbgxQ5mUmMiZFgLIV+FBNExI05xg=\ngithub.com/yudai/golcs v0.0.0-20170316035057-ecda9a501e82/go.mod h1:lgjkn3NuSvDfVJdfcVVdX+jpBxNmX4rDAzaS45IcYoM=\ngithub.com/yudai/pp v2.0.1+incompatible/go.mod h1:PuxR/8QJ7cyCkFp/aUDS+JY727OFEZkTdatxwunjIkc=\ngithub.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=\ngithub.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.0/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngithub.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=\ngitlab.com/bosi/decorder v0.2.1 h1:ehqZe8hI4w7O4b1vgsDZw1YU1PE7iJXrQWFMsocbQ1w=\ngitlab.com/bosi/decorder v0.2.1/go.mod h1:6C/nhLSbF6qZbYD8bRmISBwc6vcWdNsiIBkRvjJFrH0=\ngitlab.com/digitalxero/go-conventional-commit v1.0.7 h1:8/dO6WWG+98PMhlZowt/YjuiKhqhGlOCwlIV8SqqGh8=\ngitlab.com/digitalxero/go-conventional-commit v1.0.7/go.mod h1:05Xc2BFsSyC5tKhK0y+P3bs0AwUtNuTp+mTpbCU/DZ0=\ngo.etcd.io/bbolt v1.3.2/go.mod h1:IbVyRI1SCnLcuJnV2u8VeU0CEYM7e686BmAb1XKL+uU=\ngo.etcd.io/bbolt v1.3.3/go.mod h1:IbVyRI1SCnLcuJnV2u8VeU0CEYM7e686BmAb1XKL+uU=\ngo.etcd.io/bbolt v1.3.4/go.mod h1:G5EMThwa9y8QZGBClrRx5EY+Yw9kAhnjy3bSjsnlVTQ=\ngo.etcd.io/etcd v0.0.0-20200513171258-e048e166ab9c/go.mod h1:xCI7ZzBfRuGgBXyXO6yfWfDmlWd35khcWpUa4L0xI/k=\ngo.etcd.io/etcd/api/v3 v3.5.2/go.mod h1:5GB2vv4A4AOn3yk7MftYGHkUfGtDHnEraIjym4dYz5A=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.2/go.mod h1:IJHfcCEKxYu1Os13ZdwCwIUTUVGYTSAM3YSwc9/Ac1g=\ngo.etcd.io/etcd/client/v2 v2.305.2/go.mod h1:2D7ZejHVMIfog1221iLSYlQRzrtECw3kz4I4VAQm3qI=\ngo.mozilla.org/mozlog v0.0.0-20170222151521-4bb13139d403/go.mod h1:jHoPAGnDrCy6kaI2tAze5Prf0Nr0w/oNkROt2lw3n3o=\ngo.opencensus.io v0.15.0/go.mod h1:UffZAU+4sDEINUGP/B7UfBBkq4fqLu9zXAX7ke6CHW0=\ngo.opencensus.io v0.21.0/go.mod h1:mSImk1erAIZhrmZN+AvHh14ztQfjbGwt4TtuofqLduU=\ngo.opencensus.io v0.22.0/go.mod h1:+kGneAE2xo2IficOXnaByMWTGM9T73dGwxeWcUqIpI8=\ngo.opencensus.io v0.22.2/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=\ngo.opencensus.io v0.22.3/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=\ngo.opencensus.io v0.22.4/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=\ngo.opencensus.io v0.22.5/go.mod h1:5pWMHQbX5EPX2/62yrJeAkowc+lfs/XD7Uxpq3pI6kk=\ngo.opencensus.io v0.23.0/go.mod h1:XItmlyltB5F7CS4xOC1DcqMoFqwtC6OG2xF7mCv7P7E=\ngo.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=\ngo.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=\ngo.opentelemetry.io/contrib/detectors/gcp v1.29.0 h1:TiaiXB4DpGD3sdzNlYQxruQngn5Apwzi1X0DRhuGvDQ=\ngo.opentelemetry.io/contrib/detectors/gcp v1.29.0/go.mod h1:GW2aWZNwR2ZxDLdv8OyC2G8zkRoQBuURgV7RPQgcPoU=\ngo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.54.0 h1:r6I7RJCN86bpD/FQwedZ0vSixDpwuWREjW9oRMsmqDc=\ngo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.54.0/go.mod h1:B9yO6b04uB80CzjedvewuqDhxJxi11s7/GtiGa8bAjI=\ngo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.54.0 h1:TT4fX+nBOA/+LUkobKGW1ydGcn+G3vRw9+g5HwCphpk=\ngo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.54.0/go.mod h1:L7UH0GbB0p47T4Rri3uHjbpCFYrVrwc1I25QhNPiGK8=\ngo.opentelemetry.io/otel v1.29.0 h1:PdomN/Al4q/lN6iBJEN3AwPvUiHPMlt93c8bqTG5Llw=\ngo.opentelemetry.io/otel v1.29.0/go.mod h1:N/WtXPs1CNCUEx+Agz5uouwCba+i+bJGFicT8SR4NP8=\ngo.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.29.0 h1:WDdP9acbMYjbKIyJUhTvtzj601sVJOqgWdUxSdR/Ysc=\ngo.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.29.0/go.mod h1:BLbf7zbNIONBLPwvFnwNHGj4zge8uTCM/UPIVW1Mq2I=\ngo.opentelemetry.io/otel/metric v1.29.0 h1:vPf/HFWTNkPu1aYeIsc98l4ktOQaL6LeSoeV2g+8YLc=\ngo.opentelemetry.io/otel/metric v1.29.0/go.mod h1:auu/QWieFVWx+DmQOUMgj0F8LHWdgalxXqvp7BII/W8=\ngo.opentelemetry.io/otel/sdk v1.29.0 h1:vkqKjk7gwhS8VaWb0POZKmIEDimRCMsopNYnriHyryo=\ngo.opentelemetry.io/otel/sdk v1.29.0/go.mod h1:pM8Dx5WKnvxLCb+8lG1PRNIDxu9g9b9g59Qr7hfAAok=\ngo.opentelemetry.io/otel/sdk/metric v1.29.0 h1:K2CfmJohnRgvZ9UAj2/FhIf/okdWcNdBwe1m8xFXiSY=\ngo.opentelemetry.io/otel/sdk/metric v1.29.0/go.mod h1:6zZLdCl2fkauYoZIOn/soQIDSWFmNSRcICarHfuhNJQ=\ngo.opentelemetry.io/otel/trace v1.29.0 h1:J/8ZNK4XgR7a21DZUAsbF8pZ5Jcw1VhACmnYt39JTi4=\ngo.opentelemetry.io/otel/trace v1.29.0/go.mod h1:eHl3w0sp3paPkYstJOmAimxhiFXPg+MMTlEh3nsQgWQ=\ngo.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=\ngo.uber.org/atomic v1.3.2/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=\ngo.uber.org/atomic v1.4.0/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE=\ngo.uber.org/atomic v1.5.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.5.1/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.6.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ=\ngo.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=\ngo.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=\ngo.uber.org/dig v1.8.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/dig v1.10.0 h1:yLmDDj9/zuDjv3gz8GQGviXMs9TfysIUMUilCpgzUJY=\ngo.uber.org/dig v1.10.0/go.mod h1:X34SnWGr8Fyla9zQNO2GSO2D+TIuqB14OS8JhYocIyw=\ngo.uber.org/fx v1.10.0/go.mod h1:vLRicqpG/qQEzno4SYU86iCwfT95EZza+Eba0ItuxqY=\ngo.uber.org/fx v1.13.1 h1:CFNTr1oin5OJ0VCZ8EycL3wzF29Jz2g0xe55RFsf2a4=\ngo.uber.org/fx v1.13.1/go.mod h1:bREWhavnedxpJeTq9pQT53BbvwhUv7TcpsOqcH4a+3w=\ngo.uber.org/goleak v0.10.0/go.mod h1:VCZuO8V8mFPlL0F5J5GK1rtHV3DrFcQ1R8ryq7FK0aI=\ngo.uber.org/goleak v1.0.0/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=\ngo.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=\ngo.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=\ngo.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=\ngo.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=\ngo.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0=\ngo.uber.org/multierr v1.3.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.4.0/go.mod h1:VgVr7evmIr6uPjLBxg28wmKNXyqE9akIJ5XnfpiKl+4=\ngo.uber.org/multierr v1.5.0/go.mod h1:FeouvMocqHpRaaGuG9EjoKcStLC43Zu/fmqdUMPcKYU=\ngo.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=\ngo.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=\ngo.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=\ngo.uber.org/net/metrics v1.3.0 h1:iRLPuVecNYf/wIV+mQaA4IgN8ghifu3q1B4IT6HfwyY=\ngo.uber.org/net/metrics v1.3.0/go.mod h1:pEQrSDGNWT5IVpekWzee5//uHjI4gmgZFkobfw3bv8I=\ngo.uber.org/nilaway v0.0.0-20251021214447-34f56b8c16b9 h1:48u0MW3ki2cfzv6woA/ljDFquyGSx0T99Qwf0l1RuWY=\ngo.uber.org/nilaway v0.0.0-20251021214447-34f56b8c16b9/go.mod h1:pbGMVkhssd5Ee+eoqfgEk9mzoJoKZAhnTbl1QNcYDi0=\ngo.uber.org/thriftrw v1.29.2 h1:pRuFLzbGvTcnYwGSjizWRHlbJUzGhu84sRiL1h1kUd8=\ngo.uber.org/thriftrw v1.29.2/go.mod h1:YcjXveberDd28/Bs34SwHy3yu85x/jB4UA2gIcz/Eo0=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA=\ngo.uber.org/yarpc v1.70.3 h1:yykHwzRD9/bgDtlOWoVuXbSZoU91Id2dWJO1CDSRHnI=\ngo.uber.org/yarpc v1.70.3/go.mod h1:EH6I6K1HxBbOxZIJfhdDf+H+cvXPHmJyRvpfPqES20U=\ngo.uber.org/zap v1.9.1/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=\ngo.uber.org/zap v1.10.0/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=\ngo.uber.org/zap v1.13.0/go.mod h1:zwrFLgMcdUuIBviXEYEH1YKNaOBnKXsx2IPda5bBwHM=\ngo.uber.org/zap v1.16.0/go.mod h1:MA8QOfq0BHJwdXa996Y4dYkAqRKB8/1K1QMMZVaNZjQ=\ngo.uber.org/zap v1.17.0/go.mod h1:MXVU+bhUf/A7Xi2HNOnopQOrmycQ5Ih87HtOu4q5SSo=\ngo.uber.org/zap v1.24.0 h1:FiJd5l1UOLj0wCgbSE0rwwXHzEdAZS6hiiSnxJN/D60=\ngo.uber.org/zap v1.24.0/go.mod h1:2kMP+WWQ8aoFoedH3T2sq6iJ2yDWpHbP0f6MQbS9Gkg=\ngocloud.dev v0.23.0 h1:u/6F8slWwaZPgGpjpNp0jzH+1P/M2ri7qEP3lFgbqBE=\ngocloud.dev v0.23.0/go.mod h1:zklCCIIo1N9ELkU2S2E7tW8P8eeMU7oGLeQCXdDwx9Q=\ngolang.org/x/crypto v0.0.0-20180501155221-613d6eafa307/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20181029021203-45a5f77698d3/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20181203042331-505ab145d0a9/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20190219172222-a4c6cb3142f2/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20190325154230-a5d413f7728c/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20190426145343-a29dc8fdc734/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20190605123033-f99c8df09eb5/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20190820162420-60c769a6c586/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20190923035154-9ee001bba392/go.mod h1:/lpIB1dKB+9EgE3H3cr1v9wB50oz8l4C4h62xy7jSTY=\ngolang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=\ngolang.org/x/crypto v0.0.0-20200302210943-78000ba7a073/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20200414173820-0848c9571904/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20200728195943-123391ffb6de/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20201002170205-7f63de1d35b0/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=\ngolang.org/x/crypto v0.0.0-20210322153248-0c34fe9e7dc2/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=\ngolang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=\ngolang.org/x/crypto v0.0.0-20210506145944-38f3c27a63bf/go.mod h1:P+XmwS30IXTQdn5tA2iutPOUgjI07+tq3H3K9MVA1s8=\ngolang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.0.0-20211108221036-ceb1ce70b4fa/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=\ngolang.org/x/crypto v0.0.0-20220313003712-b769efc7c000/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=\ngolang.org/x/crypto v0.0.0-20220411220226-7b82a4e95df4/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=\ngolang.org/x/crypto v0.43.0 h1:dduJYIi3A3KOfdGOHX8AVZ/jGiyPa3IbBozJ5kNuE04=\ngolang.org/x/crypto v0.43.0/go.mod h1:BFbav4mRNlXJL4wNeejLpWxB7wMbc79PdRGhWKncxR0=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190306152737-a1d7652674e8/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/exp v0.0.0-20190510132918-efd6b22b2522/go.mod h1:ZjyILWgesfNpC6sMxTJOJm9Kp84zZh5NQWvqDGG3Qr8=\ngolang.org/x/exp v0.0.0-20190731235908-ec7cb31e5a56/go.mod h1:JhuoJpWY28nO4Vef9tZUw9qufEGTyX1+7lmHxV5q5G4=\ngolang.org/x/exp v0.0.0-20190829153037-c13cbed26979/go.mod h1:86+5VVa7VpoJ4kLfm080zCjGlMRFzhUhsZKEZO7MGek=\ngolang.org/x/exp v0.0.0-20191030013958-a1ab85dbe136/go.mod h1:JXzH8nQsPlswgeRAPE3MuO9GYsAcnJvJ4vnMwN/5qkY=\ngolang.org/x/exp v0.0.0-20191129062945-2f5052295587/go.mod h1:2RIsYlXP63K8oxa1u096TMicItID8zy7Y6sNkU49FU4=\ngolang.org/x/exp v0.0.0-20191227195350-da58074b4299/go.mod h1:2RIsYlXP63K8oxa1u096TMicItID8zy7Y6sNkU49FU4=\ngolang.org/x/exp v0.0.0-20200119233911-0405dc783f0a/go.mod h1:2RIsYlXP63K8oxa1u096TMicItID8zy7Y6sNkU49FU4=\ngolang.org/x/exp v0.0.0-20200207192155-f17229e696bd/go.mod h1:J/WKrq2StrnmMY6+EHIKF9dgMWnmCNThgcyBT1FY9mM=\ngolang.org/x/exp v0.0.0-20200224162631-6cc2880d07d6/go.mod h1:3jZMyOhIsHpP37uCMkUooju7aAi5cS1Q23tOzKc+0MU=\ngolang.org/x/exp v0.0.0-20200331195152-e8c3332aa8e5/go.mod h1:4M0jN8W1tt0AVLNr8HDosyJCDCDuyL9N9+3m7wDWgKw=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e h1:qyrTQ++p1afMkO4DPEeLGq/3oTsdlvdH4vqZUBWzUKM=\ngolang.org/x/exp/typeparams v0.0.0-20220218215828-6cf2b201936e/go.mod h1:AbB0pIl9nAr9wVwH+Z2ZpaocVmF5I4GyWCDIsVjR0bk=\ngolang.org/x/image v0.0.0-20190227222117-0694c2d4d067/go.mod h1:kZ7UVZpmo3dzQBMxlp+ypCbDeSB+sBbTgSJuh5dn5js=\ngolang.org/x/image v0.0.0-20190802002840-cff245a6509b/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190301231843-5614ed5bae6f/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190409202823-959b441ac422/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190909230951-414d861bb4ac/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/lint v0.0.0-20191125180803-fdd1cda4f05f/go.mod h1:5qLYkcX4OjUUV8bRuDixDT3tpyyb+LUpUlRWLxfhWrs=\ngolang.org/x/lint v0.0.0-20200130185559-910be7a94367/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20200302205851-738671d3881b/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20201208152925-83fdc39ff7b5/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616 h1:VLliZ0d+/avPrXXH+OakdXhpJuEoBZuwh1m2j7U6Iug=\ngolang.org/x/lint v0.0.0-20210508222113-6edffad5e616/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=\ngolang.org/x/mobile v0.0.0-20190312151609-d3739f865fa6/go.mod h1:z+o9i4GpDbdi3rU15maQ/Ox0txvL9dWGYEHz965HBQE=\ngolang.org/x/mobile v0.0.0-20190719004257-d2bd2a29d028/go.mod h1:E/iHnbuqvinMTCcRqshq8CkpyQDoeVncDDYHnLhea+o=\ngolang.org/x/mobile v0.0.0-20200801112145-973feb4309de/go.mod h1:skQtrUTUwhdJvXM/2KKJzY8pDgNr9I/FOMqDVRPBUS4=\ngolang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc=\ngolang.org/x/mod v0.1.0/go.mod h1:0QHyrYULN0/3qlju5TqG8bIK38QM8yzMo5ekMj3DlcY=\ngolang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=\ngolang.org/x/mod v0.1.1-0.20191107180719-034126e5016b/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=\ngolang.org/x/mod v0.1.1-0.20191209134235-331c550502dd/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.1/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=\ngolang.org/x/mod v0.5.1/go.mod h1:5OXOZSfqPIIbmVBIIKWRFfZjPR0E5r58TLhUjH0a2Ro=\ngolang.org/x/mod v0.6.0-dev.0.20220106191415-9b9b3d81d5e3/go.mod h1:3p9vT2HGsQu2K1YbXdKPJLVgG5VJdoTa1poYQBtP1AY=\ngolang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=\ngolang.org/x/mod v0.29.0 h1:HV8lRxZC4l2cr3Zq1LvtOsi/ThTgWnUk/y64QSs8GwA=\ngolang.org/x/mod v0.29.0/go.mod h1:NyhrlYXJ2H4eJiRy/WDBO6HMqZQ6q9nk4JzS3NuCK+w=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181023162649-9b4f9f5ad519/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181201002055-351d144fa1fc/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20181220203305-927f97764cc3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190501004415-9ce7a6920f09/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190503192946-f4e77d36d62c/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=\ngolang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190628185345-da137c7871d7/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190724013045-ca1201d0de80/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190813141303-74dc4d7220e7/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20190923162816-aa69164e4478/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20191002035440-2ec189313ef0/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20191112182307-2180aed22343/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20191209160850-c0dbc17a3553/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200114155413-6afb5195e5aa/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200222125558-5a598a2470a0/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200301022130-244492dfa37a/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=\ngolang.org/x/net v0.0.0-20200324143707-d3edc9973b7e/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200421231249-e086a090c8fd/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200501053045-e0ff5e5a1de5/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200506145744-7e3656a0809f/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200513185701-a91f0712d120/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200520004742-59133d7f0dd7/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200520182314-0ba52f642ac2/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=\ngolang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20200707034311-ab3426394381/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=\ngolang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20201031054903-ff519b6c9102/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.0.0-20201209123823-ac852fbbde11/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20201224014010-6772e930b67b/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210119194325-5f4716e94777/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=\ngolang.org/x/net v0.0.0-20210316092652-d523dce5a7f4/go.mod h1:RBQZq4jEuRlivfhVLdyRGr576XBO4/greRjx4P4O3yc=\ngolang.org/x/net v0.0.0-20210326060303-6b1517762897/go.mod h1:uSPa2vr4CLtc/ILN5odXGNXS6mhrKVzTaCXzk9m6W3k=\ngolang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=\ngolang.org/x/net v0.0.0-20210410081132-afb366fc7cd1/go.mod h1:9tjilg8BloeKEkVJvy7fQ90B1CfIiPueXVOjqfkSzI8=\ngolang.org/x/net v0.0.0-20210420210106-798c2154c571/go.mod h1:72T/g9IO56b78aLF+1Kcs5dz7/ng1VjMUvfKvpfy+jM=\ngolang.org/x/net v0.0.0-20210428140749-89ef3d95e781/go.mod h1:OJAsFXCWl8Ukc7SiCT/9KSuxbyM7479/AVlXFRxuMCk=\ngolang.org/x/net v0.0.0-20210503060351-7fd8e65b6420/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20210505214959-0714010a04ed/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20210525063256-abc453219eb5/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20210805182204-aaa1db679c0d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=\ngolang.org/x/net v0.0.0-20220127200216-cd36cc0744dd/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=\ngolang.org/x/net v0.0.0-20220225172249-27dd8689420f/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=\ngolang.org/x/net v0.0.0-20220325170049-de3da57026de/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=\ngolang.org/x/net v0.0.0-20220412020605-290c469a71a5/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=\ngolang.org/x/net v0.46.0 h1:giFlY12I07fugqwPuWJi68oOnpfqFnJIJzaIIm2JVV4=\ngolang.org/x/net v0.46.0/go.mod h1:Q9BGdFy1y4nkUwiLvT5qtyhAnEHgnQ/zd8PfU6nc210=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.0.0-20181106182150-f42d05182288/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20191202225959-858c2ad4c8b6/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=\ngolang.org/x/oauth2 v0.0.0-20200902213428-5d25da1a8d43/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20201109201403-9fd604954f58/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20201208152858-08078c50e5b5/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210218202405-ba52d332ba99/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210220000619-9bb904979d93/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210313182246-cd4f82c27b84/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210413134643-5e61552d6c78/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210427180440-81ed05c6b58c/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210514164344-f6687ab2804c/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210628180205-a41e5a781914/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210805134026-6f1e6394065a/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20210819190943-2bc19b11175f/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20211005180243-6b3c2da341f1/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20211104180415-d3ed0bb246c8/go.mod h1:KelEdhl1UZF7XfJ4dDtk6s++YSgaE7mD/BuKKDLBl4A=\ngolang.org/x/oauth2 v0.0.0-20220223155221-ee480838109b/go.mod h1:DAh4E804XQdzx2j+YRIaUnCqCV2RuMz24cGBJ5QYIrc=\ngolang.org/x/oauth2 v0.0.0-20220309155454-6242fa91716a/go.mod h1:DAh4E804XQdzx2j+YRIaUnCqCV2RuMz24cGBJ5QYIrc=\ngolang.org/x/oauth2 v0.0.0-20220411215720-9780585627b5/go.mod h1:DAh4E804XQdzx2j+YRIaUnCqCV2RuMz24cGBJ5QYIrc=\ngolang.org/x/oauth2 v0.25.0 h1:CY4y7XT9v0cRI9oupztF8AgiIu99L/ksR/Xp/6jrZ70=\ngolang.org/x/oauth2 v0.25.0/go.mod h1:XYTD2NtWslqkgxebSiOHnXEap4TF09sJSc7H1sXbhtI=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190412183630-56d357773e84/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20200317015054-43a5402ce75a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20200625203802-6e8e738ad208/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=\ngolang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=\ngolang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181026203630-95b1ffbd15a5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181107165924-66b7b1311ac8/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190221075227-b4e8571b14e0/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190312061237-fead79001313/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190507160741-ecd444e8653b/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190606165138-5da285871e9c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190624142023-c5567b49c5d0/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190726091711-fc99dfbffb4e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190826190057-c7b8b68b1456/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190904154756-749cb33beabd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190922100055-0a153f010e69/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20190924154521-2837fb4f24fe/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191001151750-bb3f8db39f24/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191008105621-543471e840be/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191112214154-59a1497f0cea/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191204072324-ce4227a45e2e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191220142924-d4481acd189f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20191228213918-04cbcbbfeed8/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200113162924-86b910548bc1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200117145432-59e60aa80a0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200124204421-9fbb57f87de9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200202164722-d101bd2416d5/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200212091648-12a6c2dcc1e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200302150141-5c8b2ff67527/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200331124033-c3d80250170d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200420163511-1957bb5e6d1f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200501052902-10377860bb8e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200511232937-7e40ca221e25/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200515095857-1151b9dac4a9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200523222454-059865788121/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200615200032-f1bc736245b1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200625212154-ddb9806d33ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200803210538-64077c9b5642/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200828194041-157a740278f4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200905004654-be1d3432aa8f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201109165425-215b40eba54c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20201201145000-ef89a241ccb3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210104204734-6f8348627aad/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210220050731-9a76102bfb43/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210223095934-7937bea0104d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210225134936-a50acf3fe073/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210303074136-134d130e1a04/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210305230114-8fe3ee5dd75b/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210315160823-c6e025ad8005/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210320140829-1e4c9ba3b0c4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210324051608-47abb6519492/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210403161142-5e06dd20ab57/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210412220455-f1c623a9e750/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210420072515-93ed5bcd2bfe/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210420205809-ac73e9fd8988/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210423185535-09eb48e85fd7/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210503080704-8803ae5d1324/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20210503173754-0981d6026fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210514084401-e8d321eab015/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210603125802-9665404d3644/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210616094352-59db8d763f22/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210823070655-63515b42dcdf/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210908233432-aa78b53d3365/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211007075335-d3039528d8ac/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211105183446-c75c47738b0c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211124211545-fe61309f8881/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211210111614-af8b64212486/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211213223007-03aa0b5f6827/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220114195835-da31bd327af9/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220128215802-99c3d69c2c27/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220209214540-3681064d5158/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220227234510-4e6760a101f9/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220328115105-d36c6a25d886/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220412211240-33da011f77ad/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220422013727-9388b58f7150/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=\ngolang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=\ngolang.org/x/telemetry v0.0.0-20251008203120-078029d740a8 h1:LvzTn0GQhWuvKH/kVRS3R3bVAsdQWI7hvfLHGgh9+lU=\ngolang.org/x/telemetry v0.0.0-20251008203120-078029d740a8/go.mod h1:Pi4ztBfryZoJEkyFTI5/Ocsu2jXyDr6iSdgJiYE/uwE=\ngolang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=\ngolang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=\ngolang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=\ngolang.org/x/term v0.36.0 h1:zMPR+aF8gfksFprF/Nc/rd1wRS1EI6nDBGyWAvDzx2Q=\ngolang.org/x/term v0.36.0/go.mod h1:Qu394IJq6V6dCBRgwqshf3mPF85AqzYEzofzRdZkWss=\ngolang.org/x/text v0.0.0-20170915032832-14c0d48ead0c/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.1-0.20180807135948-17ff2d5776d2/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.4/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=\ngolang.org/x/text v0.30.0 h1:yznKA/E9zq54KzlzBEAWn1NXSQ8DIp/NYMy88xJjl4k=\ngolang.org/x/text v0.30.0/go.mod h1:yDdHFIX9t+tORqspjENWgzaCVXgk0yYnYuSZ8UzzBVM=\ngolang.org/x/time v0.0.0-20180412165947-fbb02b2291d2/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.0.0-20181108054448-85acf8d2951c/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.0.0-20191024005414-555d28b269f0/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.0.0-20200416051211-89c76fbcd5d1/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.0.0-20210220033141-f8bda1e9f3ba/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=\ngolang.org/x/time v0.8.0 h1:9i3RxcPv3PZnitoVGMPDKZSq1xW1gK1Xy3ArNOGZfEg=\ngolang.org/x/time v0.8.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20180525024113-a5b4c53f6e8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190307163923-6a08e3108db3/go.mod h1:25r3+/G6/xytQM8iWZKq3Hn0kr0rgFKPUNVEL/dr3z4=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190311215038-5c2858a9cfe5/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190312151545-0bb0c0a6e846/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190312170243-e65039ee4138/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190321232350-e250d351ecad/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190322203728-c1a832b0ad89/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190422233926-fe54fb35175b/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190506145303-2d16b83fe98c/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/tools v0.0.0-20190606124116-d0a3d012864b/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20190614205625-5aca471b1d59/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20190628153133-6cdbf07be9d0/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc=\ngolang.org/x/tools v0.0.0-20190816200558-6889da9d5479/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20190907020128-2ca718005c18/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20190910044552-dd2b5c81c578/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20190911174233-4f2ddba30aff/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20190916130336-e45ffcd953cc/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191010075000-0337d82405ff/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191012152004-8de300cfc20a/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191030062658-86caa796c7ab/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191108193012-7d206e10da11/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191112195655-aa38f8e97acc/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191113191852-77e3bb0ad9e7/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191114200427-caa0b0f7d508/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191115202509-3a792d9c32b2/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191125144606-a911d9008d1f/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191130070609-6e064ea0cf2d/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=\ngolang.org/x/tools v0.0.0-20191216052735-49a3e744a425/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20191216173652-a0e659d51361/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20191227053925-7b8e75db28f4/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117012304-6edc0a871e69/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117161641-43d50277825c/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117215004-fe56e6335763/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200117220505-0cba7a3a9ee9/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200122220014-bf1340f18c4a/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200204074204-1cc6d1ef6c74/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200207183749-b753a1ba74fa/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200212150539-ea181f53ac56/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200224181240-023911ca70b2/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200227222343-706bc42d1f0d/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=\ngolang.org/x/tools v0.0.0-20200304193943-95d2e580d8eb/go.mod h1:o4KQGtdN14AW+yjsvvwRTJJuXz8XRtIHtEnmAXLyFUw=\ngolang.org/x/tools v0.0.0-20200312045724-11d5b4c81c7d/go.mod h1:o4KQGtdN14AW+yjsvvwRTJJuXz8XRtIHtEnmAXLyFUw=\ngolang.org/x/tools v0.0.0-20200324003944-a576cf524670/go.mod h1:Sl4aGygMT6LrqrWclx+PTx3U+LnKx/seiNR+3G19Ar8=\ngolang.org/x/tools v0.0.0-20200329025819-fd4102a86c65/go.mod h1:Sl4aGygMT6LrqrWclx+PTx3U+LnKx/seiNR+3G19Ar8=\ngolang.org/x/tools v0.0.0-20200331025713-a30bf2db82d4/go.mod h1:Sl4aGygMT6LrqrWclx+PTx3U+LnKx/seiNR+3G19Ar8=\ngolang.org/x/tools v0.0.0-20200414032229-332987a829c3/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200422022333-3d57cf2e726e/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200426102838-f3a5411a4c3b/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200501065659-ab2804fb9c9d/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200512131952-2bc93b1c0c88/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200515010526-7d3b6ebf133d/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200618134242-20370b0cb4b2/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200622203043-20e05c1c8ffa/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200624225443-88f3c62a19ff/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200625211823-6506e20df31f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200626171337-aa94e735be7f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200630154851-b2d8b0336632/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=\ngolang.org/x/tools v0.0.0-20200706234117-b22de6825cf7/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200724022722-7017fd6b1305/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200729194436-6467de6f59a7/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200804011535-6c149bb5ef0d/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200812195022-5ae4c3c160a0/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200820010801-b793a1359eac/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200825202427-b303f430e36d/go.mod h1:njjCfa9FT2d7l9Bc6FUM5FLjQPp3cFF28FI3qnDFljA=\ngolang.org/x/tools v0.0.0-20200831203904-5a2aa26beb65/go.mod h1:Cj7w3i3Rnn0Xh82ur9kSqwfTHTeVxaDqrfMjpcNT6bE=\ngolang.org/x/tools v0.0.0-20200904185747-39188db58858/go.mod h1:Cj7w3i3Rnn0Xh82ur9kSqwfTHTeVxaDqrfMjpcNT6bE=\ngolang.org/x/tools v0.0.0-20201001104356-43ebab892c4c/go.mod h1:z6u4i615ZeAfBE4XtMziQW1fSVJXACjjbWkB/mvPzlU=\ngolang.org/x/tools v0.0.0-20201002184944-ecd9fd270d5d/go.mod h1:z6u4i615ZeAfBE4XtMziQW1fSVJXACjjbWkB/mvPzlU=\ngolang.org/x/tools v0.0.0-20201023174141-c8cfbd0f21e6/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20201110124207-079ba7bd75cd/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20201201161351-ac6f37ff4c2a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20201208233053-a543418bbed2/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20201224043029-2b0845dc783e/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20201230224404-63754364767c/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20210105154028-b0ab187a4818/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.0.0-20210108195828-e2f9c7f1fc8e/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=\ngolang.org/x/tools v0.1.0/go.mod h1:xkSsbof2nBLbhDlRMhhhyNLN/zl3eTqcnHD5viDpcZ0=\ngolang.org/x/tools v0.1.1-0.20210205202024-ef80cdb6ec6d/go.mod h1:9bzcO0MWcOuT0tm1iBGzDVPshzfwoVvREIui8C+MHqU=\ngolang.org/x/tools v0.1.1-0.20210302220138-2ac05c832e1a/go.mod h1:9bzcO0MWcOuT0tm1iBGzDVPshzfwoVvREIui8C+MHqU=\ngolang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.2/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.3/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.4/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=\ngolang.org/x/tools v0.1.7/go.mod h1:LGqMHiF4EqQNHR1JncWGqT5BVaXmza+X+BDGol+dOxo=\ngolang.org/x/tools v0.1.8/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=\ngolang.org/x/tools v0.1.9-0.20211228192929-ee1ca4ffc4da/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=\ngolang.org/x/tools v0.1.9/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=\ngolang.org/x/tools v0.1.10/go.mod h1:Uh6Zz+xoGYZom868N8YTex3t7RhtHDBrE8Gzo9bV56E=\ngolang.org/x/tools v0.1.11-0.20220513221640-090b14e8501f/go.mod h1:SgwaegtQh8clINPpECJMqnxLv9I09HLqnW3RMqW0CA4=\ngolang.org/x/tools v0.38.0 h1:Hx2Xv8hISq8Lm16jvBZ2VQf+RLmbd7wVUsALibYI/IQ=\ngolang.org/x/tools v0.38.0/go.mod h1:yEsQ/d/YK8cjh0L6rZlY8tgtlKiBNTL14pGDJPJpYQs=\ngolang.org/x/tools/go/expect v0.1.1-deprecated h1:jpBZDwmgPhXsKZC6WhL20P4b/wmnpsEAGHaNy0n/rJM=\ngolang.org/x/tools/go/expect v0.1.1-deprecated/go.mod h1:eihoPOH+FgIqa3FpoTwguz/bVUSGBlGQU67vpBeOrBY=\ngolang.org/x/tools/go/packages/packagestest v0.1.1-deprecated h1:1h2MnaIAIXISqTFKdENegdpAgUXz6NrPEsbIeWaBRvM=\ngolang.org/x/tools/go/packages/packagestest v0.1.1-deprecated/go.mod h1:RVAQXBGNv1ib0J382/DPCRS/BPnsGebyM1Gj5VSDpG8=\ngolang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20220411194840-2f41105eb62f/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2 h1:H2TDz8ibqkAF6YGhCdN3jS9O0/s90v0rJh3X/OLHEUk=\ngolang.org/x/xerrors v0.0.0-20220907171357-04be3eba64a2/go.mod h1:K8+ghG5WaK9qNqU5K3HdILfMLy1f3aNYFI/wnl100a8=\ngoogle.golang.org/api v0.4.0/go.mod h1:8k5glujaEP+g9n7WNsDg8QP6cUVNI86fCNMcbazEtwE=\ngoogle.golang.org/api v0.5.0/go.mod h1:8k5glujaEP+g9n7WNsDg8QP6cUVNI86fCNMcbazEtwE=\ngoogle.golang.org/api v0.7.0/go.mod h1:WtwebWUNSVBH/HAw79HIFXZNqEvBhG+Ra+ax0hx3E3M=\ngoogle.golang.org/api v0.8.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg=\ngoogle.golang.org/api v0.9.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg=\ngoogle.golang.org/api v0.10.0/go.mod h1:o4eAsZoiT+ibD93RtjEohWalFOjRDx6CVaqeizhEnKg=\ngoogle.golang.org/api v0.13.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsbkAI=\ngoogle.golang.org/api v0.14.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsbkAI=\ngoogle.golang.org/api v0.15.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsbkAI=\ngoogle.golang.org/api v0.17.0/go.mod h1:BwFmGc8tA3vsd7r/7kR8DY7iEEGSU04BFxCo5jP/sfE=\ngoogle.golang.org/api v0.18.0/go.mod h1:BwFmGc8tA3vsd7r/7kR8DY7iEEGSU04BFxCo5jP/sfE=\ngoogle.golang.org/api v0.19.0/go.mod h1:BwFmGc8tA3vsd7r/7kR8DY7iEEGSU04BFxCo5jP/sfE=\ngoogle.golang.org/api v0.20.0/go.mod h1:BwFmGc8tA3vsd7r/7kR8DY7iEEGSU04BFxCo5jP/sfE=\ngoogle.golang.org/api v0.22.0/go.mod h1:BwFmGc8tA3vsd7r/7kR8DY7iEEGSU04BFxCo5jP/sfE=\ngoogle.golang.org/api v0.24.0/go.mod h1:lIXQywCXRcnZPGlsd8NbLnOjtAoL6em04bJ9+z0MncE=\ngoogle.golang.org/api v0.28.0/go.mod h1:lIXQywCXRcnZPGlsd8NbLnOjtAoL6em04bJ9+z0MncE=\ngoogle.golang.org/api v0.29.0/go.mod h1:Lcubydp8VUV7KeIHD9z2Bys/sm/vGKnG1UHuDBSrHWM=\ngoogle.golang.org/api v0.30.0/go.mod h1:QGmEvQ87FHZNiUVJkT14jQNYJ4ZJjdRF23ZXz5138Fc=\ngoogle.golang.org/api v0.35.0/go.mod h1:/XrVsuzM0rZmrsbjJutiuftIzeuTQcEeaYcSk/mQ1dg=\ngoogle.golang.org/api v0.36.0/go.mod h1:+z5ficQTmoYpPn8LCUNVpK5I7hwkpjbcgqA7I34qYtE=\ngoogle.golang.org/api v0.40.0/go.mod h1:fYKFpnQN0DsDSKRVRcQSDQNtqWPfM9i+zNPxepjRCQ8=\ngoogle.golang.org/api v0.41.0/go.mod h1:RkxM5lITDfTzmyKFPt+wGrCJbVfniCr2ool8kTBzRTU=\ngoogle.golang.org/api v0.43.0/go.mod h1:nQsDGjRXMo4lvh5hP0TKqF244gqhGcr/YSIykhUk/94=\ngoogle.golang.org/api v0.45.0/go.mod h1:ISLIJCedJolbZvDfAk+Ctuq5hf+aJ33WgtUsfyFoLXA=\ngoogle.golang.org/api v0.46.0/go.mod h1:ceL4oozhkAiTID8XMmJBsIxID/9wMXJVVFXPg4ylg3I=\ngoogle.golang.org/api v0.47.0/go.mod h1:Wbvgpq1HddcWVtzsVLyfLp8lDg6AA241LmgIL59tHXo=\ngoogle.golang.org/api v0.48.0/go.mod h1:71Pr1vy+TAZRPkPs/xlCf5SsU8WjuAWv1Pfjbtukyy4=\ngoogle.golang.org/api v0.50.0/go.mod h1:4bNT5pAuq5ji4SRZm+5QIkjny9JAyVD/3gaSihNefaw=\ngoogle.golang.org/api v0.51.0/go.mod h1:t4HdrdoNgyN5cbEfm7Lum0lcLDLiise1F8qDKX00sOU=\ngoogle.golang.org/api v0.54.0/go.mod h1:7C4bFFOvVDGXjfDTAsgGwDgAxRDeQ4X8NvUedIt6z3k=\ngoogle.golang.org/api v0.55.0/go.mod h1:38yMfeP1kfjsl8isn0tliTjIb1rJXcQi4UXlbqivdVE=\ngoogle.golang.org/api v0.56.0/go.mod h1:38yMfeP1kfjsl8isn0tliTjIb1rJXcQi4UXlbqivdVE=\ngoogle.golang.org/api v0.57.0/go.mod h1:dVPlbZyBo2/OjBpmvNdpn2GRm6rPy75jyU7bmhdrMgI=\ngoogle.golang.org/api v0.59.0/go.mod h1:sT2boj7M9YJxZzgeZqXogmhfmRWDtPzT31xkieUbuZU=\ngoogle.golang.org/api v0.61.0/go.mod h1:xQRti5UdCmoCEqFxcz93fTl338AVqDgyaDRuOZ3hg9I=\ngoogle.golang.org/api v0.63.0/go.mod h1:gs4ij2ffTRXwuzzgJl/56BdwJaA194ijkfn++9tDuPo=\ngoogle.golang.org/api v0.67.0/go.mod h1:ShHKP8E60yPsKNw/w8w+VYaj9H6buA5UqDp8dhbQZ6g=\ngoogle.golang.org/api v0.70.0/go.mod h1:Bs4ZM2HGifEvXwd50TtW70ovgJffJYw2oRCOFU/SkfA=\ngoogle.golang.org/api v0.71.0/go.mod h1:4PyU6e6JogV1f9eA4voyrTY2batOLdgZ5qZ5HOCc4j8=\ngoogle.golang.org/api v0.74.0/go.mod h1:ZpfMZOVRMywNyvJFeqL9HRWBgAuRfSjJFpe9QtRRyDs=\ngoogle.golang.org/api v0.215.0 h1:jdYF4qnyczlEz2ReWIsosNLDuzXyvFHJtI5gcr0J7t0=\ngoogle.golang.org/api v0.215.0/go.mod h1:fta3CVtuJYOEdugLNWm6WodzOS8KdFckABwN4I40hzY=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.3.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0=\ngoogle.golang.org/appengine v1.6.2/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0=\ngoogle.golang.org/appengine v1.6.5/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=\ngoogle.golang.org/appengine v1.6.6/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=\ngoogle.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=\ngoogle.golang.org/genproto v0.0.0-20170818010345-ee236bd376b0/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20180518175338-11a468237815/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20181107211654-5fc9ac540362/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190418145605-e7d98fc518a7/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190502173448-54afdca5d873/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190508193815-b515fa19cec8/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=\ngoogle.golang.org/genproto v0.0.0-20190530194941-fb225487d101/go.mod h1:z3L6/3dTEVtUr6QSP8miRzeRqwQOioJ9I66odjN4I7s=\ngoogle.golang.org/genproto v0.0.0-20190801165951-fa694d86fc64/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20190911173649-1774047e7e51/go.mod h1:IbNlFCBrqXvoKpeg0TB2l7cyZUmoaFKYIwrEpbDKLA8=\ngoogle.golang.org/genproto v0.0.0-20190927181202-20e1ac93f88c/go.mod h1:IbNlFCBrqXvoKpeg0TB2l7cyZUmoaFKYIwrEpbDKLA8=\ngoogle.golang.org/genproto v0.0.0-20191108220845-16a3f7862a1a/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20191115194625-c23dd37a84c9/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20191216164720-4f79533eabd1/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20191230161307-f3c370f40bfb/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20200115191322-ca5a22157cba/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20200122232147-0452cf42e150/go.mod h1:n3cpQtvxv34hfy77yVDNjmbRyujviMdxYliBSkLhpCc=\ngoogle.golang.org/genproto v0.0.0-20200204135345-fa8e72b47b90/go.mod h1:GmwEX6Z4W5gMy59cAlVYjN9JhxgbQH6Gn+gFDQe2lzA=\ngoogle.golang.org/genproto v0.0.0-20200212174721-66ed5ce911ce/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200224152610-e50cd9704f63/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200228133532-8c2c7df3a383/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200305110556-506484158171/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200312145019-da6875a35672/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200331122359-1ee6d9798940/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200423170343-7949de9c1215/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200430143042-b979b6f78d84/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200511104702-f5ebc3bea380/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=\ngoogle.golang.org/genproto v0.0.0-20200515170657-fc4c6c6a6587/go.mod h1:YsZOwe1myG/8QRHRsmBRE1LrgQY60beZKjly0O1fX9U=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/genproto v0.0.0-20200618031413-b414f8b61790/go.mod h1:jDfRM7FcilCzHH/e9qn6dsT145K34l5v+OpcnNgKAAA=\ngoogle.golang.org/genproto v0.0.0-20200626011028-ee7919e894b5/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20200707001353-8e8330bf89df/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20200729003335-053ba62fc06f/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20200804131852-c06518451d9c/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20200825200019-8632dd797987/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20200904004341-0bd0a958aa1d/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20201109203340-2640f1f9cdfb/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20201201144952-b05cb90ed32e/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20201210142538-e3217bee35cc/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20201214200347-8c77b98c765d/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210108203827-ffc7fda8c3d7/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210222152913-aa3ee6e6a81c/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210226172003-ab064af71705/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210303154014-9728d6b83eeb/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210310155132-4ce2db91004e/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210319143718-93e7006c17a6/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no=\ngoogle.golang.org/genproto v0.0.0-20210402141018-6c239bbf2bb1/go.mod h1:9lPAdzaEmUacj36I+k7YKbEc5CXzPIeORRgDAUOu28A=\ngoogle.golang.org/genproto v0.0.0-20210413151531-c14fb6ef47c3/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210420162539-3c870d7478d2/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210423144448-3a41ef94ed2b/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210429181445-86c259c2b4ab/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210506142907-4a47615972c2/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210513213006-bf773b8c8384/go.mod h1:P3QM42oQyzQSnHPnZ/vqoCdDmzH28fzWByN9asMeM8A=\ngoogle.golang.org/genproto v0.0.0-20210602131652-f16073e35f0c/go.mod h1:UODoCrxHCcBojKKwX1terBiRUaqAsFqJiF615XL43r0=\ngoogle.golang.org/genproto v0.0.0-20210604141403-392c879c8b08/go.mod h1:UODoCrxHCcBojKKwX1terBiRUaqAsFqJiF615XL43r0=\ngoogle.golang.org/genproto v0.0.0-20210608205507-b6d2f5bf0d7d/go.mod h1:UODoCrxHCcBojKKwX1terBiRUaqAsFqJiF615XL43r0=\ngoogle.golang.org/genproto v0.0.0-20210624195500-8bfb893ecb84/go.mod h1:SzzZ/N+nwJDaO1kznhnlzqS8ocJICar6hYhVyhi++24=\ngoogle.golang.org/genproto v0.0.0-20210713002101-d411969a0d9a/go.mod h1:AxrInvYm1dci+enl5hChSFPOmmUF1+uAa/UsgNRWd7k=\ngoogle.golang.org/genproto v0.0.0-20210716133855-ce7ef5c701ea/go.mod h1:AxrInvYm1dci+enl5hChSFPOmmUF1+uAa/UsgNRWd7k=\ngoogle.golang.org/genproto v0.0.0-20210728212813-7823e685a01f/go.mod h1:ob2IJxKrgPT52GcgX759i1sleT07tiKowYBGbczaW48=\ngoogle.golang.org/genproto v0.0.0-20210805201207-89edb61ffb67/go.mod h1:ob2IJxKrgPT52GcgX759i1sleT07tiKowYBGbczaW48=\ngoogle.golang.org/genproto v0.0.0-20210813162853-db860fec028c/go.mod h1:cFeNkxwySK631ADgubI+/XFU/xp8FD5KIVV4rj8UC5w=\ngoogle.golang.org/genproto v0.0.0-20210821163610-241b8fcbd6c8/go.mod h1:eFjDcFEctNawg4eG61bRv87N7iHBWyVhJu7u1kqDUXY=\ngoogle.golang.org/genproto v0.0.0-20210828152312-66f60bf46e71/go.mod h1:eFjDcFEctNawg4eG61bRv87N7iHBWyVhJu7u1kqDUXY=\ngoogle.golang.org/genproto v0.0.0-20210831024726-fe130286e0e2/go.mod h1:eFjDcFEctNawg4eG61bRv87N7iHBWyVhJu7u1kqDUXY=\ngoogle.golang.org/genproto v0.0.0-20210903162649-d08c68adba83/go.mod h1:eFjDcFEctNawg4eG61bRv87N7iHBWyVhJu7u1kqDUXY=\ngoogle.golang.org/genproto v0.0.0-20210909211513-a8c4777a87af/go.mod h1:eFjDcFEctNawg4eG61bRv87N7iHBWyVhJu7u1kqDUXY=\ngoogle.golang.org/genproto v0.0.0-20210924002016-3dee208752a0/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211008145708-270636b82663/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211028162531-8db9c33dc351/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211118181313-81c1377c94b1/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211206160659-862468c7d6e0/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211208223120-3a66f561d7aa/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20211221195035-429b39de9b1c/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20220126215142-9970aeb2e350/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20220207164111-0872dc986b00/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc=\ngoogle.golang.org/genproto v0.0.0-20220218161850-94dd64e39d7c/go.mod h1:kGP+zUP2Ddo0ayMi4YuN7C3WZyJvGLZRh8Z5wnAqvEI=\ngoogle.golang.org/genproto v0.0.0-20220222213610-43724f9ea8cf/go.mod h1:kGP+zUP2Ddo0ayMi4YuN7C3WZyJvGLZRh8Z5wnAqvEI=\ngoogle.golang.org/genproto v0.0.0-20220304144024-325a89244dc8/go.mod h1:kGP+zUP2Ddo0ayMi4YuN7C3WZyJvGLZRh8Z5wnAqvEI=\ngoogle.golang.org/genproto v0.0.0-20220310185008-1973136f34c6/go.mod h1:kGP+zUP2Ddo0ayMi4YuN7C3WZyJvGLZRh8Z5wnAqvEI=\ngoogle.golang.org/genproto v0.0.0-20220324131243-acbaeb5b85eb/go.mod h1:hAL49I2IFola2sVEjAn7MEwsja0xp51I0tlGAf9hz4E=\ngoogle.golang.org/genproto v0.0.0-20220407144326-9054f6ed7bac/go.mod h1:8w6bsBMX6yCPbAVTeqQHvzxW0EIFigd5lZyahWgyfDo=\ngoogle.golang.org/genproto v0.0.0-20241118233622-e639e219e697 h1:ToEetK57OidYuqD4Q5w+vfEnPvPpuTwedCNVohYJfNk=\ngoogle.golang.org/genproto v0.0.0-20241118233622-e639e219e697/go.mod h1:JJrvXBWRZaFMxBufik1a4RpFw4HhgVtBBWQeQgUj2cc=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20241209162323-e6fa225c2576 h1:CkkIfIt50+lT6NHAVoRYEyAvQGFM7xEwXUUywFvEb3Q=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20241209162323-e6fa225c2576/go.mod h1:1R3kvZ1dtP3+4p4d3G8uJ8rFk/fWlScl38vanWACI08=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20241223144023-3abc09e42ca8 h1:TqExAhdPaB60Ux47Cn0oLV07rGnxZzIsaRhQaqS666A=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20241223144023-3abc09e42ca8/go.mod h1:lcTa1sDdWEIHMWlITnIczmw5w60CF9ffkb8Z+DVmmjA=\ngoogle.golang.org/grpc v1.8.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=\ngoogle.golang.org/grpc v1.12.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=\ngoogle.golang.org/grpc v1.21.0/go.mod h1:oYelfM1adQP15Ek0mdvEgi9Df8B9CZIaU1084ijfRaM=\ngoogle.golang.org/grpc v1.21.1/go.mod h1:oYelfM1adQP15Ek0mdvEgi9Df8B9CZIaU1084ijfRaM=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.23.1/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.24.0/go.mod h1:XDChyiUovWa60DnaeDeZmSW86xtLtjtZbwvSiRnRtcA=\ngoogle.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=\ngoogle.golang.org/grpc v1.26.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.27.1/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.28.0/go.mod h1:rpkK4SK4GF4Ach/+MFLZUBavHOvF2JJB5uozKKal+60=\ngoogle.golang.org/grpc v1.29.0/go.mod h1:itym6AZVZYACWQqET3MqgPpjcuV5QH3BxFS3IjizoKk=\ngoogle.golang.org/grpc v1.29.1/go.mod h1:itym6AZVZYACWQqET3MqgPpjcuV5QH3BxFS3IjizoKk=\ngoogle.golang.org/grpc v1.30.0/go.mod h1:N36X2cJ7JwdamYAgDz+s+rVMFjt3numwzf/HckM8pak=\ngoogle.golang.org/grpc v1.31.0/go.mod h1:N36X2cJ7JwdamYAgDz+s+rVMFjt3numwzf/HckM8pak=\ngoogle.golang.org/grpc v1.31.1/go.mod h1:N36X2cJ7JwdamYAgDz+s+rVMFjt3numwzf/HckM8pak=\ngoogle.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0=\ngoogle.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc=\ngoogle.golang.org/grpc v1.34.0/go.mod h1:WotjhfgOW/POjDeRt8vscBtXq+2VjORFy659qA51WJ8=\ngoogle.golang.org/grpc v1.35.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.36.1/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU=\ngoogle.golang.org/grpc v1.37.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.37.1/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.38.0/go.mod h1:NREThFqKR1f3iQ6oBuvc5LadQuXVGo9rkm5ZGrQdJfM=\ngoogle.golang.org/grpc v1.39.0/go.mod h1:PImNr+rS9TWYb2O4/emRugxiyHZ5JyHW5F+RPnDzfrE=\ngoogle.golang.org/grpc v1.39.1/go.mod h1:PImNr+rS9TWYb2O4/emRugxiyHZ5JyHW5F+RPnDzfrE=\ngoogle.golang.org/grpc v1.40.0/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34=\ngoogle.golang.org/grpc v1.40.1/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34=\ngoogle.golang.org/grpc v1.44.0/go.mod h1:k+4IHHFw41K8+bbowsex27ge2rCb65oeWqe4jJ590SU=\ngoogle.golang.org/grpc v1.45.0/go.mod h1:lN7owxKUQEqMfSyQikvvk5tf/6zMPsrK+ONuO11+0rQ=\ngoogle.golang.org/grpc v1.67.3 h1:OgPcDAFKHnH8X3O4WcO4XUc8GRDeKsKReqbQtiCj7N8=\ngoogle.golang.org/grpc v1.67.3/go.mod h1:YGaHCc6Oap+FzBJTZLBzkGSYt/cvGPFTPxkn7QfSU8s=\ngoogle.golang.org/grpc/cmd/protoc-gen-go-grpc v1.1.0/go.mod h1:6Kw0yEErY5E/yWrBtf03jp27GLLJujG4z/JK95pnjjw=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.24.0/go.mod h1:r/3tXBNzIEhYS9I1OUVjXDlt8tc493IdKGjtUeSXeh4=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=\ngoogle.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=\ngoogle.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=\ngoogle.golang.org/protobuf v1.28.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=\ngoogle.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk=\ngoogle.golang.org/protobuf v1.36.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=\ngopkg.in/cheggaaa/pb.v1 v1.0.25/go.mod h1:V/YB90LKu/1FcN3WVnfiiE5oMCibMjukxqG/qStrOgw=\ngopkg.in/cheggaaa/pb.v1 v1.0.28/go.mod h1:V/YB90LKu/1FcN3WVnfiiE5oMCibMjukxqG/qStrOgw=\ngopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=\ngopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=\ngopkg.in/gcfg.v1 v1.2.3/go.mod h1:yesOnuUOFQAhST5vPY4nbZsb/huCgGGXlipJsBn0b3o=\ngopkg.in/ini.v1 v1.51.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=\ngopkg.in/ini.v1 v1.62.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=\ngopkg.in/ini.v1 v1.66.4/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=\ngopkg.in/resty.v1 v1.12.0/go.mod h1:mDo4pnntr5jdWRML875a/NmxYqAlA73dVijT2AXvQQo=\ngopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=\ngopkg.in/warnings.v0 v0.1.2 h1:wFXVbFY8DY5/xOe1ECiWdKCzZlxgshcYVNkBHstARME=\ngopkg.in/warnings.v0 v0.1.2/go.mod h1:jksf8JmL6Qr/oQM2OXTHunEvvTAsrWBLb6OOjuVWRNI=\ngopkg.in/yaml.v2 v2.0.0-20170812160011-eb3733d160e7/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74=\ngopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.6/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.0-20200605160147-a5ece683394c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190106161140-3f1c8253044a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190418001031-e561f6794a2a/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=\nhonnef.co/go/tools v0.0.1-2020.1.3/go.mod h1:X/FiERA/W4tHapMX5mGpAtMSVEeEUOyHaw9vFzvIQ3k=\nhonnef.co/go/tools v0.0.1-2020.1.4/go.mod h1:X/FiERA/W4tHapMX5mGpAtMSVEeEUOyHaw9vFzvIQ3k=\nhonnef.co/go/tools v0.3.2 h1:ytYb4rOqyp1TSa2EPvNVwtPQJctSELKaMyLfqNP4+34=\nhonnef.co/go/tools v0.3.2/go.mod h1:jzwdWgg7Jdq75wlfblQxO4neNaFFSvgc1tD5Wv8U0Yw=\nmvdan.cc/gofumpt v0.3.1 h1:avhhrOmv0IuvQVK7fvwV91oFSGAk5/6Po8GXTzICeu8=\nmvdan.cc/gofumpt v0.3.1/go.mod h1:w3ymliuxvzVx8DAutBnVyDqYb1Niy/yCJt/lk821YCE=\nmvdan.cc/interfacer v0.0.0-20180901003855-c20040233aed h1:WX1yoOaKQfddO/mLzdV4wptyWgoH/6hwLs7QHTixo0I=\nmvdan.cc/interfacer v0.0.0-20180901003855-c20040233aed/go.mod h1:Xkxe497xwlCKkIaQYRfC7CSLworTXY9RMqwhhCm+8Nc=\nmvdan.cc/lint v0.0.0-20170908181259-adc824a0674b h1:DxJ5nJdkhDlLok9K6qO+5290kphDJbHOQO1DFFFTeBo=\nmvdan.cc/lint v0.0.0-20170908181259-adc824a0674b/go.mod h1:2odslEg/xrtNQqCYg2/jCoyKnw3vv5biOc3JnIcYfL4=\nmvdan.cc/unparam v0.0.0-20211214103731-d0ef000c54e5 h1:Jh3LAeMt1eGpxomyu3jVkmVZWW2MxZ1qIIV2TZ/nRio=\nmvdan.cc/unparam v0.0.0-20211214103731-d0ef000c54e5/go.mod h1:b8RRCBm0eeiWR8cfN88xeq2G5SG3VKGO+5UPWi5FSOY=\nnhooyr.io/websocket v1.8.6/go.mod h1:B70DZP8IakI65RVQ51MsWP/8jndNma26DVA/nFSCgW0=\nnhooyr.io/websocket v1.8.7/go.mod h1:B70DZP8IakI65RVQ51MsWP/8jndNma26DVA/nFSCgW0=\nrsc.io/binaryregexp v0.2.0/go.mod h1:qTv7/COck+e2FymRvadv62gMdZztPaShugOCi3I+8D8=\nrsc.io/quote/v3 v3.1.0/go.mod h1:yEA65RcK8LyAZtP9Kv3t0HmxON59tX3rD+tICJqUlj0=\nrsc.io/sampler v1.3.0/go.mod h1:T1hPZKmBbMNahiBKFy5HrXp6adAjACjK9JXDnKaTXpA=\nsigs.k8s.io/yaml v1.1.0/go.mod h1:UJmg0vDUVViEyp3mgSv9WPwZCDxu4rQW1olrI1uml+o=\n"
  },
  {
    "path": "internal/tools/go.work",
    "content": "go 1.24.0\n\n// this go.work file ensures that the root go.work DOES NOT\n// automatically include dependencies in tools, as otherwise it tries\n// to find minimum versions across it and all other modules.\n//\n// this isn't currently harmful, but it does risk preventing us from\n// upgrading our tools due to breaking changes in other dependencies\n// they might bring in.  keeping them separate is best.\n\nuse .\n"
  },
  {
    "path": "internal/tools/go.work.sum",
    "content": "bazil.org/fuse v0.0.0-20180421153158-65cc252bf669 h1:FNCRpXiquG1aoyqcIWVFmpTSKVcx2bQD38uZZeGtdlw=\nbitbucket.org/creachadair/shell v0.0.6 h1:reJflDbKqnlnqb4Oo2pQ1/BqmY/eCWcNGHrIUO8qIzc=\ncloud.google.com/go/accessapproval v1.6.0 h1:x0cEHro/JFPd7eS4BlEWNTMecIj2HdXjOVB5BtvwER0=\ncloud.google.com/go/accessapproval v1.6.0/go.mod h1:R0EiYnwV5fsRFiKZkPHr6mwyk2wxUJ30nL4j2pcFY2E=\ncloud.google.com/go/accessapproval v1.8.2 h1:h4u1MypgeYXTGvnNc1luCBLDN4Kb9Re/gw0Atvoi8HE=\ncloud.google.com/go/accessapproval v1.8.2/go.mod h1:aEJvHZtpjqstffVwF/2mCXXSQmpskyzvw6zKLvLutZM=\ncloud.google.com/go/accesscontextmanager v1.7.0 h1:MG60JgnEoawHJrbWw0jGdv6HLNSf6gQvYRiXpuzqgEA=\ncloud.google.com/go/accesscontextmanager v1.7.0/go.mod h1:CEGLewx8dwa33aDAZQujl7Dx+uYhS0eay198wB/VumQ=\ncloud.google.com/go/accesscontextmanager v1.9.2 h1:P0uVixQft8aacbZ7VDZStNZdrftF24Hk8JkA3kfvfqI=\ncloud.google.com/go/accesscontextmanager v1.9.2/go.mod h1:T0Sw/PQPyzctnkw1pdmGAKb7XBA84BqQzH0fSU7wzJU=\ncloud.google.com/go/aiplatform v1.37.0 h1:zTw+suCVchgZyO+k847wjzdVjWmrAuehxdvcZvJwfGg=\ncloud.google.com/go/aiplatform v1.37.0/go.mod h1:IU2Cv29Lv9oCn/9LkFiiuKfwrRTq+QQMbW+hPCxJGZw=\ncloud.google.com/go/aiplatform v1.69.0 h1:XvBzK8e6/6ufbi/i129Vmn/gVqFwbNPmRQ89K+MGlgc=\ncloud.google.com/go/aiplatform v1.69.0/go.mod h1:nUsIqzS3khlnWvpjfJbP+2+h+VrFyYsTm7RNCAViiY8=\ncloud.google.com/go/analytics v0.19.0 h1:LqAo3tAh2FU9+w/r7vc3hBjU23Kv7GhO/PDIW7kIYgM=\ncloud.google.com/go/analytics v0.19.0/go.mod h1:k8liqf5/HCnOUkbawNtrWWc+UAzyDlW89doe8TtoDsE=\ncloud.google.com/go/analytics v0.25.2 h1:KgJ5Taxtsnro/co7WIhmAHi5pzYAtvxu8LMqenPAlSo=\ncloud.google.com/go/analytics v0.25.2/go.mod h1:th0DIunqrhI1ZWVlT3PH2Uw/9ANX8YHfFDEPqf/+7xM=\ncloud.google.com/go/apigateway v1.5.0 h1:ZI9mVO7x3E9RK/BURm2p1aw9YTBSCQe3klmyP1WxWEg=\ncloud.google.com/go/apigateway v1.5.0/go.mod h1:GpnZR3Q4rR7LVu5951qfXPJCHquZt02jf7xQx7kpqN8=\ncloud.google.com/go/apigateway v1.7.2 h1:TRB5q0vvbT5Yx4bNSCWlqLJFJnhc7tDlCR9ccpo1vzg=\ncloud.google.com/go/apigateway v1.7.2/go.mod h1:+weId+9aR9J6GRwDka7jIUSrKEX60XGcikX7dGU8O7M=\ncloud.google.com/go/apigeeconnect v1.5.0 h1:sWOmgDyAsi1AZ48XRHcATC0tsi9SkPT7DA/+VCfkaeA=\ncloud.google.com/go/apigeeconnect v1.5.0/go.mod h1:KFaCqvBRU6idyhSNyn3vlHXc8VMDJdRmwDF6JyFRqZ8=\ncloud.google.com/go/apigeeconnect v1.7.2 h1:GHg0ddEQUZ08C1qC780P5wwY/jaIW8UtxuRQXLLuRXs=\ncloud.google.com/go/apigeeconnect v1.7.2/go.mod h1:he/SWi3A63fbyxrxD6jb67ak17QTbWjva1TFbT5w8Kw=\ncloud.google.com/go/apigeeregistry v0.6.0 h1:E43RdhhCxdlV+I161gUY2rI4eOaMzHTA5kNkvRsFXvc=\ncloud.google.com/go/apigeeregistry v0.6.0/go.mod h1:BFNzW7yQVLZ3yj0TKcwzb8n25CFBri51GVGOEUcgQsc=\ncloud.google.com/go/apigeeregistry v0.9.2 h1:fC3ZXEk2QsBxUlZZDZpbBGXC/ZQglCBmHDGgY5aNipg=\ncloud.google.com/go/apigeeregistry v0.9.2/go.mod h1:A5n/DwpG5NaP2fcLYGiFA9QfzpQhPRFNATO1gie8KM8=\ncloud.google.com/go/apikeys v0.6.0 h1:B9CdHFZTFjVti89tmyXXrO+7vSNo2jvZuHG8zD5trdQ=\ncloud.google.com/go/apikeys v0.6.0/go.mod h1:kbpXu5upyiAlGkKrJgQl8A0rKNNJ7dQ377pdroRSSi8=\ncloud.google.com/go/appengine v1.7.1 h1:aBGDKmRIaRRoWJ2tAoN0oVSHoWLhtO9aj/NvUyP4aYs=\ncloud.google.com/go/appengine v1.7.1/go.mod h1:IHLToyb/3fKutRysUlFO0BPt5j7RiQ45nrzEJmKTo6E=\ncloud.google.com/go/appengine v1.9.2 h1:pxAQ//FsyEQsaF9HJduPCOEvj9GV4fvnLARGz1+KDzM=\ncloud.google.com/go/appengine v1.9.2/go.mod h1:bK4dvmMG6b5Tem2JFZcjvHdxco9g6t1pwd3y/1qr+3s=\ncloud.google.com/go/area120 v0.7.1 h1:ugckkFh4XkHJMPhTIx0CyvdoBxmOpMe8rNs4Ok8GAag=\ncloud.google.com/go/area120 v0.7.1/go.mod h1:j84i4E1RboTWjKtZVWXPqvK5VHQFJRF2c1Nm69pWm9k=\ncloud.google.com/go/area120 v0.9.2 h1:LODm6TjW27/LJ4z4fBNJHRb+tlvy0gSu6Vb8j2lfluY=\ncloud.google.com/go/area120 v0.9.2/go.mod h1:Ar/KPx51UbrTWGVGgGzFnT7hFYQuk/0VOXkvHdTbQMI=\ncloud.google.com/go/artifactregistry v1.13.0 h1:o1Q80vqEB6Qp8WLEH3b8FBLNUCrGQ4k5RFj0sn/sgO8=\ncloud.google.com/go/artifactregistry v1.13.0/go.mod h1:uy/LNfoOIivepGhooAUpL1i30Hgee3Cu0l4VTWHUC08=\ncloud.google.com/go/artifactregistry v1.16.0 h1:BZpz0x8HCG7hwTkD+GlUwPQVFGOo9w84t8kxQwwc0DA=\ncloud.google.com/go/artifactregistry v1.16.0/go.mod h1:LunXo4u2rFtvJjrGjO0JS+Gs9Eco2xbZU6JVJ4+T8Sk=\ncloud.google.com/go/asset v1.13.0 h1:YAsssO08BqZ6mncbb6FPlj9h6ACS7bJQUOlzciSfbNk=\ncloud.google.com/go/asset v1.13.0/go.mod h1:WQAMyYek/b7NBpYq/K4KJWcRqzoalEsxz/t/dTk4THw=\ncloud.google.com/go/asset v1.20.3 h1:/jQBAkZVUbsIczRepDkwaf/K5NcRYvQ6MBiWg5i20fU=\ncloud.google.com/go/asset v1.20.3/go.mod h1:797WxTDwdnFAJzbjZ5zc+P5iwqXc13yO9DHhmS6wl+o=\ncloud.google.com/go/assuredworkloads v1.10.0 h1:VLGnVFta+N4WM+ASHbhc14ZOItOabDLH1MSoDv+Xuag=\ncloud.google.com/go/assuredworkloads v1.10.0/go.mod h1:kwdUQuXcedVdsIaKgKTp9t0UJkE5+PAVNhdQm4ZVq2E=\ncloud.google.com/go/assuredworkloads v1.12.2 h1:6Y6a4V7CD50qtjvayhu7f5o35UFJP8ade7IbHNfdQEc=\ncloud.google.com/go/assuredworkloads v1.12.2/go.mod h1:/WeRr/q+6EQYgnoYrqCVgw7boMoDfjXZZev3iJxs2Iw=\ncloud.google.com/go/automl v1.12.0 h1:50VugllC+U4IGl3tDNcZaWvApHBTrn/TvyHDJ0wM+Uw=\ncloud.google.com/go/automl v1.12.0/go.mod h1:tWDcHDp86aMIuHmyvjuKeeHEGq76lD7ZqfGLN6B0NuU=\ncloud.google.com/go/automl v1.14.2 h1:RzR5Nx78iaF2FNAfaaQ/7o2b4VuQ17YbOaeK/DLYSW4=\ncloud.google.com/go/automl v1.14.2/go.mod h1:mIat+Mf77W30eWQ/vrhjXsXaRh8Qfu4WiymR0hR6Uxk=\ncloud.google.com/go/baremetalsolution v0.5.0 h1:2AipdYXL0VxMboelTTw8c1UJ7gYu35LZYUbuRv9Q28s=\ncloud.google.com/go/baremetalsolution v0.5.0/go.mod h1:dXGxEkmR9BMwxhzBhV0AioD0ULBmuLZI8CdwalUxuss=\ncloud.google.com/go/baremetalsolution v1.3.2 h1:rhawlI+9gy/i1ZQbN/qL6FXHGXusWbfr6UoQdcCpybw=\ncloud.google.com/go/baremetalsolution v1.3.2/go.mod h1:3+wqVRstRREJV/puwaKAH3Pnn7ByreZG2aFRsavnoBQ=\ncloud.google.com/go/batch v0.7.0 h1:YbMt0E6BtqeD5FvSv1d56jbVsWEzlGm55lYte+M6Mzs=\ncloud.google.com/go/batch v0.7.0/go.mod h1:vLZN95s6teRUqRQ4s3RLDsH8PvboqBK+rn1oevL159g=\ncloud.google.com/go/batch v1.11.2 h1:OVhgpMMJc+mrFw51R3C06JKC0D6u125RlEBULpg78No=\ncloud.google.com/go/batch v1.11.2/go.mod h1:ehsVs8Y86Q4K+qhEStxICqQnNqH8cqgpCxx89cmU5h4=\ncloud.google.com/go/beyondcorp v0.5.0 h1:UkY2BTZkEUAVrgqnSdOJ4p3y9ZRBPEe1LkjgC8Bj/Pc=\ncloud.google.com/go/beyondcorp v0.5.0/go.mod h1:uFqj9X+dSfrheVp7ssLTaRHd2EHqSL4QZmH4e8WXGGU=\ncloud.google.com/go/beyondcorp v1.1.2 h1:hzKZf9ScvqTWqR8xGKVvD35ScQuxbMySELvJ0OW1usI=\ncloud.google.com/go/beyondcorp v1.1.2/go.mod h1:q6YWSkEsSZTU2WDt1qtz6P5yfv79wgktGtNbd0FJTLI=\ncloud.google.com/go/bigquery v1.50.0 h1:RscMV6LbnAmhAzD893Lv9nXXy2WCaJmbxYPWDLbGqNQ=\ncloud.google.com/go/bigquery v1.50.0/go.mod h1:YrleYEh2pSEbgTBZYMJ5SuSr0ML3ypjRB1zgf7pvQLU=\ncloud.google.com/go/bigquery v1.64.0 h1:vSSZisNyhr2ioJE1OuYBQrnrpB7pIhRQm4jfjc7E/js=\ncloud.google.com/go/bigquery v1.64.0/go.mod h1:gy8Ooz6HF7QmA+TRtX8tZmXBKH5mCFBwUApGAb3zI7Y=\ncloud.google.com/go/bigtable v1.33.0 h1:2BDaWLRAwXO14DJL/u8crbV2oUbMZkIa2eGq8Yao1bk=\ncloud.google.com/go/bigtable v1.33.0/go.mod h1:HtpnH4g25VT1pejHRtInlFPnN5sjTxbQlsYBjh9t5l0=\ncloud.google.com/go/billing v1.13.0 h1:JYj28UYF5w6VBAh0gQYlgHJ/OD1oA+JgW29YZQU+UHM=\ncloud.google.com/go/billing v1.13.0/go.mod h1:7kB2W9Xf98hP9Sr12KfECgfGclsH3CQR0R08tnRlRbc=\ncloud.google.com/go/billing v1.19.2 h1:shcyz1UkrUxbPsqHL6L84ZdtBZ7yocaFFCxMInTsrNo=\ncloud.google.com/go/billing v1.19.2/go.mod h1:AAtih/X2nka5mug6jTAq8jfh1nPye0OjkHbZEZgU59c=\ncloud.google.com/go/binaryauthorization v1.5.0 h1:d3pMDBCCNivxt5a4eaV7FwL7cSH0H7RrEnFrTb1QKWs=\ncloud.google.com/go/binaryauthorization v1.5.0/go.mod h1:OSe4OU1nN/VswXKRBmciKpo9LulY41gch5c68htf3/Q=\ncloud.google.com/go/binaryauthorization v1.9.2 h1:zZX4cvtYSXc5ogOar1w5KA1BLz3j464RPSaR/HhroJ8=\ncloud.google.com/go/binaryauthorization v1.9.2/go.mod h1:T4nOcRWi2WX4bjfSRXJkUnpliVIqjP38V88Z10OvEv4=\ncloud.google.com/go/certificatemanager v1.6.0 h1:5C5UWeSt8Jkgp7OWn2rCkLmYurar/vIWIoSQ2+LaTOc=\ncloud.google.com/go/certificatemanager v1.6.0/go.mod h1:3Hh64rCKjRAX8dXgRAyOcY5vQ/fE1sh8o+Mdd6KPgY8=\ncloud.google.com/go/certificatemanager v1.9.2 h1:/lO1ejN415kRaiO6DNNCHj0UvQujKP714q3l8gp4lsY=\ncloud.google.com/go/certificatemanager v1.9.2/go.mod h1:PqW+fNSav5Xz8bvUnJpATIRo1aaABP4mUg/7XIeAn6c=\ncloud.google.com/go/channel v1.12.0 h1:GpcQY5UJKeOekYgsX3QXbzzAc/kRGtBq43fTmyKe6Uw=\ncloud.google.com/go/channel v1.12.0/go.mod h1:VkxCGKASi4Cq7TbXxlaBezonAYpp1GCnKMY6tnMQnLU=\ncloud.google.com/go/channel v1.19.1 h1:l4XcnfzJ5UGmqZQls0atcpD6ERDps4PLd5hXSyTWFv0=\ncloud.google.com/go/channel v1.19.1/go.mod h1:ungpP46l6XUeuefbA/XWpWWnAY3897CSRPXUbDstwUo=\ncloud.google.com/go/cloudbuild v1.9.0 h1:GHQCjV4WlPPVU/j3Rlpc8vNIDwThhd1U9qSY/NPZdko=\ncloud.google.com/go/cloudbuild v1.9.0/go.mod h1:qK1d7s4QlO0VwfYn5YuClDGg2hfmLZEb4wQGAbIgL1s=\ncloud.google.com/go/cloudbuild v1.19.0 h1:Uo0bL251yvyWsNtO3Og9m5Z4S48cgGf3IUX7xzOcl8s=\ncloud.google.com/go/cloudbuild v1.19.0/go.mod h1:ZGRqbNMrVGhknIIjwASa6MqoRTOpXIVMSI+Ew5DMPuY=\ncloud.google.com/go/clouddms v1.5.0 h1:E7v4TpDGUyEm1C/4KIrpVSOCTm0P6vWdHT0I4mostRA=\ncloud.google.com/go/clouddms v1.5.0/go.mod h1:QSxQnhikCLUw13iAbffF2CZxAER3xDGNHjsTAkQJcQA=\ncloud.google.com/go/clouddms v1.8.2 h1:U53ztLRgTkclaxgmBBles+tv+nNcZ5fhbRbw3b2axFw=\ncloud.google.com/go/clouddms v1.8.2/go.mod h1:pe+JSp12u4mYOkwXpSMouyCCuQHL3a6xvWH2FgOcAt4=\ncloud.google.com/go/cloudtasks v1.10.0 h1:uK5k6abf4yligFgYFnG0ni8msai/dSv6mDmiBulU0hU=\ncloud.google.com/go/cloudtasks v1.10.0/go.mod h1:NDSoTLkZ3+vExFEWu2UJV1arUyzVDAiZtdWcsUyNwBs=\ncloud.google.com/go/cloudtasks v1.13.2 h1:x6Qw5JyNbH3reL0arUtlYf77kK6OVjZZ//8JCvUkLro=\ncloud.google.com/go/cloudtasks v1.13.2/go.mod h1:2pyE4Lhm7xY8GqbZKLnYk7eeuh8L0JwAvXx1ecKxYu8=\ncloud.google.com/go/compute v1.29.0 h1:Lph6d8oPi38NHkOr6S55Nus/Pbbcp37m/J0ohgKAefs=\ncloud.google.com/go/compute v1.29.0/go.mod h1:HFlsDurE5DpQZClAGf/cYh+gxssMhBxBovZDYkEn/Og=\ncloud.google.com/go/contactcenterinsights v1.6.0 h1:jXIpfcH/VYSE1SYcPzO0n1VVb+sAamiLOgCw45JbOQk=\ncloud.google.com/go/contactcenterinsights v1.6.0/go.mod h1:IIDlT6CLcDoyv79kDv8iWxMSTZhLxSCofVV5W6YFM/w=\ncloud.google.com/go/contactcenterinsights v1.15.1 h1:cR/gQMweaG8RIWAlS5Jo1ARi8LUVQJ51t84EUefHeZ8=\ncloud.google.com/go/contactcenterinsights v1.15.1/go.mod h1:cFGxDVm/OwEVAHbU9UO4xQCtQFn0RZSrSUcF/oJ0Bbs=\ncloud.google.com/go/container v1.15.0 h1:NKlY/wCDapfVZlbVVaeuu2UZZED5Dy1z4Zx1KhEzm8c=\ncloud.google.com/go/container v1.15.0/go.mod h1:ft+9S0WGjAyjDggg5S06DXj+fHJICWg8L7isCQe9pQA=\ncloud.google.com/go/container v1.42.0 h1:sH9Hj9SoLeP+uKvLXc/04nWyWDiMo4Q85xfb1Nl5sAg=\ncloud.google.com/go/container v1.42.0/go.mod h1:YL6lDgCUi3frIWNIFU9qrmF7/6K1EYrtspmFTyyqJ+k=\ncloud.google.com/go/containeranalysis v0.9.0 h1:EQ4FFxNaEAg8PqQCO7bVQfWz9NVwZCUKaM1b3ycfx3U=\ncloud.google.com/go/containeranalysis v0.9.0/go.mod h1:orbOANbwk5Ejoom+s+DUCTTJ7IBdBQJDcSylAx/on9s=\ncloud.google.com/go/containeranalysis v0.13.2 h1:AG2gOcfZJFRiz+3SZCPnxU+gwbzKe++QSX/ej71Lom8=\ncloud.google.com/go/containeranalysis v0.13.2/go.mod h1:AiKvXJkc3HiqkHzVIt6s5M81wk+q7SNffc6ZlkTDgiE=\ncloud.google.com/go/datacatalog v1.13.0 h1:4H5IJiyUE0X6ShQBqgFFZvGGcrwGVndTwUSLP4c52gw=\ncloud.google.com/go/datacatalog v1.13.0/go.mod h1:E4Rj9a5ZtAxcQJlEBTLgMTphfP11/lNaAshpoBgemX8=\ncloud.google.com/go/datacatalog v1.23.0 h1:9F2zIbWNNmtrSkPIyGRQNsIugG5VgVVFip6+tXSdWLg=\ncloud.google.com/go/datacatalog v1.23.0/go.mod h1:9Wamq8TDfL2680Sav7q3zEhBJSPBrDxJU8WtPJ25dBM=\ncloud.google.com/go/dataflow v0.8.0 h1:eYyD9o/8Nm6EttsKZaEGD84xC17bNgSKCu0ZxwqUbpg=\ncloud.google.com/go/dataflow v0.8.0/go.mod h1:Rcf5YgTKPtQyYz8bLYhFoIV/vP39eL7fWNcSOyFfLJE=\ncloud.google.com/go/dataflow v0.10.2 h1:o9P5/zR2mOYJmCnfp9/7RprKFZCwmSu3TvemQSmCaFM=\ncloud.google.com/go/dataflow v0.10.2/go.mod h1:+HIb4HJxDCZYuCqDGnBHZEglh5I0edi/mLgVbxDf0Ag=\ncloud.google.com/go/dataform v0.7.0 h1:Dyk+fufup1FR6cbHjFpMuP4SfPiF3LI3JtoIIALoq48=\ncloud.google.com/go/dataform v0.7.0/go.mod h1:7NulqnVozfHvWUBpMDfKMUESr+85aJsC/2O0o3jWPDE=\ncloud.google.com/go/dataform v0.10.2 h1:t16DoejuOHoxJR88qrpdmFFlCXA9+x5PKrqI9qiDYz0=\ncloud.google.com/go/dataform v0.10.2/go.mod h1:oZHwMBxG6jGZCVZqqMx+XWXK+dA/ooyYiyeRbUxI15M=\ncloud.google.com/go/datafusion v1.6.0 h1:sZjRnS3TWkGsu1LjYPFD/fHeMLZNXDK6PDHi2s2s/bk=\ncloud.google.com/go/datafusion v1.6.0/go.mod h1:WBsMF8F1RhSXvVM8rCV3AeyWVxcC2xY6vith3iw3S+8=\ncloud.google.com/go/datafusion v1.8.2 h1:RPoHvIeXexXwlWhEU6DNgrYCh+C+FR2EXbrnMs2ptpI=\ncloud.google.com/go/datafusion v1.8.2/go.mod h1:XernijudKtVG/VEvxtLv08COyVuiYPraSxm+8hd4zXA=\ncloud.google.com/go/datalabeling v0.7.0 h1:ch4qA2yvddGRUrlfwrNJCr79qLqhS9QBwofPHfFlDIk=\ncloud.google.com/go/datalabeling v0.7.0/go.mod h1:WPQb1y08RJbmpM3ww0CSUAGweL0SxByuW2E+FU+wXcM=\ncloud.google.com/go/datalabeling v0.9.2 h1:UesbU2kYIUWhHUcnFS86ANPbugEq98X9k1whTNcenlc=\ncloud.google.com/go/datalabeling v0.9.2/go.mod h1:8me7cCxwV/mZgYWtRAd3oRVGFD6UyT7hjMi+4GRyPpg=\ncloud.google.com/go/dataplex v1.6.0 h1:RvoZ5T7gySwm1CHzAw7yY1QwwqaGswunmqEssPxU/AM=\ncloud.google.com/go/dataplex v1.6.0/go.mod h1:bMsomC/aEJOSpHXdFKFGQ1b0TDPIeL28nJObeO1ppRs=\ncloud.google.com/go/dataplex v1.19.2 h1:R2xnsZnuWpHi2NmBR0e43GZk2IZcQ1AFEAo1fUI0xsw=\ncloud.google.com/go/dataplex v1.19.2/go.mod h1:vsxxdF5dgk3hX8Ens9m2/pMNhQZklUhSgqTghZtF1v4=\ncloud.google.com/go/dataproc v1.12.0 h1:W47qHL3W4BPkAIbk4SWmIERwsWBaNnWm0P2sdx3YgGU=\ncloud.google.com/go/dataproc v1.12.0/go.mod h1:zrF3aX0uV3ikkMz6z4uBbIKyhRITnxvr4i3IjKsKrw4=\ncloud.google.com/go/dataproc/v2 v2.10.0 h1:B0b7eLRXzFTzb4UaxkGGidIF23l/Xpyce28m1Q0cHmU=\ncloud.google.com/go/dataproc/v2 v2.10.0/go.mod h1:HD16lk4rv2zHFhbm8gGOtrRaFohMDr9f0lAUMLmg1PM=\ncloud.google.com/go/dataqna v0.7.0 h1:yFzi/YU4YAdjyo7pXkBE2FeHbgz5OQQBVDdbErEHmVQ=\ncloud.google.com/go/dataqna v0.7.0/go.mod h1:Lx9OcIIeqCrw1a6KdO3/5KMP1wAmTc0slZWwP12Qq3c=\ncloud.google.com/go/dataqna v0.9.2 h1:hrEcid5jK5fEdlYZ0eS8HJoq+ZCTRWSV7Av42V/G994=\ncloud.google.com/go/dataqna v0.9.2/go.mod h1:WCJ7pwD0Mi+4pIzFQ+b2Zqy5DcExycNKHuB+VURPPgs=\ncloud.google.com/go/datastore v1.11.0 h1:iF6I/HaLs3Ado8uRKMvZRvF/ZLkWaWE9i8AiHzbC774=\ncloud.google.com/go/datastore v1.11.0/go.mod h1:TvGxBIHCS50u8jzG+AW/ppf87v1of8nwzFNgEZU1D3c=\ncloud.google.com/go/datastore v1.20.0 h1:NNpXoyEqIJmZFc0ACcwBEaXnmscUpcG4NkKnbCePmiM=\ncloud.google.com/go/datastore v1.20.0/go.mod h1:uFo3e+aEpRfHgtp5pp0+6M0o147KoPaYNaPAKpfh8Ew=\ncloud.google.com/go/datastream v1.7.0 h1:BBCBTnWMDwwEzQQmipUXxATa7Cm7CA/gKjKcR2w35T0=\ncloud.google.com/go/datastream v1.7.0/go.mod h1:uxVRMm2elUSPuh65IbZpzJNMbuzkcvu5CjMqVIUHrww=\ncloud.google.com/go/datastream v1.11.2 h1:vgtrwwPfY7JFEDD0VARJK4qyiApnFnPkFRQVuczYb/w=\ncloud.google.com/go/datastream v1.11.2/go.mod h1:RnFWa5zwR5SzHxeZGJOlQ4HKBQPcjGfD219Qy0qfh2k=\ncloud.google.com/go/deploy v1.8.0 h1:otshdKEbmsi1ELYeCKNYppwV0UH5xD05drSdBm7ouTk=\ncloud.google.com/go/deploy v1.8.0/go.mod h1:z3myEJnA/2wnB4sgjqdMfgxCA0EqC3RBTNcVPs93mtQ=\ncloud.google.com/go/deploy v1.25.0 h1:nYLFG2TSsYMJuengVru5P8iWnA5mNA4rKFV5YoOWQ3M=\ncloud.google.com/go/deploy v1.25.0/go.mod h1:h9uVCWxSDanXUereI5WR+vlZdbPJ6XGy+gcfC25v5rM=\ncloud.google.com/go/dialogflow v1.32.0 h1:uVlKKzp6G/VtSW0E7IH1Y5o0H48/UOCmqksG2riYCwQ=\ncloud.google.com/go/dialogflow v1.32.0/go.mod h1:jG9TRJl8CKrDhMEcvfcfFkkpp8ZhgPz3sBGmAUYJ2qE=\ncloud.google.com/go/dialogflow v1.60.0 h1:H+Q1SUeVU2La0Y0ZGEaKkhEXg3bj9Ceg5YKcMbyNOEc=\ncloud.google.com/go/dialogflow v1.60.0/go.mod h1:PjsrI+d2FI4BlGThxL0+Rua/g9vLI+2A1KL7s/Vo3pY=\ncloud.google.com/go/dlp v1.9.0 h1:1JoJqezlgu6NWCroBxr4rOZnwNFILXr4cB9dMaSKO4A=\ncloud.google.com/go/dlp v1.9.0/go.mod h1:qdgmqgTyReTz5/YNSSuueR8pl7hO0o9bQ39ZhtgkWp4=\ncloud.google.com/go/dlp v1.20.0 h1:Wwz1FoZp3pyrTNkS5fncaAccP/AbqzLQuN5WMi3aVYQ=\ncloud.google.com/go/dlp v1.20.0/go.mod h1:nrGsA3r8s7wh2Ct9FWu69UjBObiLldNyQda2RCHgdaY=\ncloud.google.com/go/documentai v1.18.0 h1:KM3Xh0QQyyEdC8Gs2vhZfU+rt6OCPF0dwVwxKgLmWfI=\ncloud.google.com/go/documentai v1.18.0/go.mod h1:F6CK6iUH8J81FehpskRmhLq/3VlwQvb7TvwOceQ2tbs=\ncloud.google.com/go/documentai v1.35.0 h1:DO4ut86a+Xa0gBq7j3FZJPavnKBNoznrg44csnobqIY=\ncloud.google.com/go/documentai v1.35.0/go.mod h1:ZotiWUlDE8qXSUqkJsGMQqVmfTMYATwJEYqbPXTR9kk=\ncloud.google.com/go/domains v0.8.0 h1:2ti/o9tlWL4N+wIuWUNH+LbfgpwxPr8J1sv9RHA4bYQ=\ncloud.google.com/go/domains v0.8.0/go.mod h1:M9i3MMDzGFXsydri9/vW+EWz9sWb4I6WyHqdlAk0idE=\ncloud.google.com/go/domains v0.10.2 h1:ekJCkuzbciXyPKkwPwvI+2Ov1GcGJtMXj/fbgilPFqg=\ncloud.google.com/go/domains v0.10.2/go.mod h1:oL0Wsda9KdJvvGNsykdalHxQv4Ri0yfdDkIi3bzTUwk=\ncloud.google.com/go/edgecontainer v1.0.0 h1:O0YVE5v+O0Q/ODXYsQHmHb+sYM8KNjGZw2pjX2Ws41c=\ncloud.google.com/go/edgecontainer v1.0.0/go.mod h1:cttArqZpBB2q58W/upSG++ooo6EsblxDIolxa3jSjbY=\ncloud.google.com/go/edgecontainer v1.4.0 h1:vpKTEkQPpkl55d6aUU2rzDFvTkMUATvBXfZSlI2KMR0=\ncloud.google.com/go/edgecontainer v1.4.0/go.mod h1:Hxj5saJT8LMREmAI9tbNTaBpW5loYiWFyisCjDhzu88=\ncloud.google.com/go/errorreporting v0.3.0 h1:kj1XEWMu8P0qlLhm3FwcaFsUvXChV/OraZwA70trRR0=\ncloud.google.com/go/errorreporting v0.3.0/go.mod h1:xsP2yaAp+OAW4OIm60An2bbLpqIhKXdWR/tawvl7QzU=\ncloud.google.com/go/errorreporting v0.3.1 h1:E/gLk+rL7u5JZB9oq72iL1bnhVlLrnfslrgcptjJEUE=\ncloud.google.com/go/errorreporting v0.3.1/go.mod h1:6xVQXU1UuntfAf+bVkFk6nld41+CPyF2NSPCyXE3Ztk=\ncloud.google.com/go/essentialcontacts v1.5.0 h1:gIzEhCoOT7bi+6QZqZIzX1Erj4SswMPIteNvYVlu+pM=\ncloud.google.com/go/essentialcontacts v1.5.0/go.mod h1:ay29Z4zODTuwliK7SnX8E86aUF2CTzdNtvv42niCX0M=\ncloud.google.com/go/essentialcontacts v1.7.2 h1:a/reGTn7WblM5DgieiLbX6CswHgTneWrA4ZNS5E+1Bg=\ncloud.google.com/go/essentialcontacts v1.7.2/go.mod h1:NoCBlOIVteJFJU+HG9dIG/Cc9kt1K9ys9mbOaGPUmPc=\ncloud.google.com/go/eventarc v1.11.0 h1:fsJmNeqvqtk74FsaVDU6cH79lyZNCYP8Rrv7EhaB/PU=\ncloud.google.com/go/eventarc v1.11.0/go.mod h1:PyUjsUKPWoRBCHeOxZd/lbOOjahV41icXyUY5kSTvVY=\ncloud.google.com/go/eventarc v1.15.0 h1:IVU2EOR8P2f6N8eneuwspN122LR87v9G54B+7ihd1TY=\ncloud.google.com/go/eventarc v1.15.0/go.mod h1:PAd/pPIZdJtJQFJI1yDEUms1mqohdNuM1BFEVHHlVFg=\ncloud.google.com/go/filestore v1.6.0 h1:ckTEXN5towyTMu4q0uQ1Mde/JwTHur0gXs8oaIZnKfw=\ncloud.google.com/go/filestore v1.6.0/go.mod h1:di5unNuss/qfZTw2U9nhFqo8/ZDSc466dre85Kydllg=\ncloud.google.com/go/filestore v1.9.2 h1:DYwMNAcF5bELHHMxRdkIWWZ3XicKp+ZpEBy+c6Gt4uY=\ncloud.google.com/go/filestore v1.9.2/go.mod h1:I9pM7Hoetq9a7djC1xtmtOeHSUYocna09ZP6x+PG1Xw=\ncloud.google.com/go/firestore v1.9.0 h1:IBlRyxgGySXu5VuW0RgGFlTtLukSnNkpDiEOMkQkmpA=\ncloud.google.com/go/firestore v1.9.0/go.mod h1:HMkjKHNTtRyZNiMzu7YAsLr9K3X2udY2AMwDaMEQiiE=\ncloud.google.com/go/firestore v1.17.0 h1:iEd1LBbkDZTFsLw3sTH50eyg4qe8eoG6CjocmEXO9aQ=\ncloud.google.com/go/firestore v1.17.0/go.mod h1:69uPx1papBsY8ZETooc71fOhoKkD70Q1DwMrtKuOT/Y=\ncloud.google.com/go/functions v1.13.0 h1:pPDqtsXG2g9HeOQLoquLbmvmb82Y4Ezdo1GXuotFoWg=\ncloud.google.com/go/functions v1.13.0/go.mod h1:EU4O007sQm6Ef/PwRsI8N2umygGqPBS/IZQKBQBcJ3c=\ncloud.google.com/go/functions v1.19.2 h1:Cu2Gj1JBBJv9gi89r8LrZNsJhGwePnhttn4Blqw/EYI=\ncloud.google.com/go/functions v1.19.2/go.mod h1:SBzWwWuaFDLnUyStDAMEysVN1oA5ECLbP3/PfJ9Uk7Y=\ncloud.google.com/go/gaming v1.9.0 h1:7vEhFnZmd931Mo7sZ6pJy7uQPDxF7m7v8xtBheG08tc=\ncloud.google.com/go/gaming v1.9.0/go.mod h1:Fc7kEmCObylSWLO334NcO+O9QMDyz+TKC4v1D7X+Bc0=\ncloud.google.com/go/gkebackup v0.4.0 h1:za3QZvw6ujR0uyqkhomKKKNoXDyqYGPJies3voUK8DA=\ncloud.google.com/go/gkebackup v0.4.0/go.mod h1:byAyBGUwYGEEww7xsbnUTBHIYcOPy/PgUWUtOeRm9Vg=\ncloud.google.com/go/gkebackup v1.6.2 h1:lWaSgjSonOXe41UhwQjts6lhDZdr5e882LNUTtnjZS0=\ncloud.google.com/go/gkebackup v1.6.2/go.mod h1:WsTSWqKJkGan1pkp5dS30oxb+Eaa6cLvxEUxKTUALwk=\ncloud.google.com/go/gkeconnect v0.7.0 h1:gXYKciHS/Lgq0GJ5Kc9SzPA35NGc3yqu6SkjonpEr2Q=\ncloud.google.com/go/gkeconnect v0.7.0/go.mod h1:SNfmVqPkaEi3bF/B3CNZOAYPYdg7sU+obZ+QTky2Myw=\ncloud.google.com/go/gkeconnect v0.12.0 h1:MuA3/aIuncXkXuUDGdbT7OLnIp7xpFhciuHAnQaoQz4=\ncloud.google.com/go/gkeconnect v0.12.0/go.mod h1:zn37LsFiNZxPN4iO7YbUk8l/E14pAJ7KxpoXoxt7Ly0=\ncloud.google.com/go/gkehub v0.12.0 h1:TqCSPsEBQ6oZSJgEYZ3XT8x2gUadbvfwI32YB0kuHCs=\ncloud.google.com/go/gkehub v0.12.0/go.mod h1:djiIwwzTTBrF5NaXCGv3mf7klpEMcST17VBTVVDcuaw=\ncloud.google.com/go/gkehub v0.15.2 h1:CR5MPEP/Ogk5IahCq3O2fKS6TJZQi8mrnrysGHCs0g8=\ncloud.google.com/go/gkehub v0.15.2/go.mod h1:8YziTOpwbM8LM3r9cHaOMy2rNgJHXZCrrmGgcau9zbQ=\ncloud.google.com/go/gkemulticloud v0.5.0 h1:8I84Q4vl02rJRsFiinBxl7WCozfdLlUVBQuSrqr9Wtk=\ncloud.google.com/go/gkemulticloud v0.5.0/go.mod h1:W0JDkiyi3Tqh0TJr//y19wyb1yf8llHVto2Htf2Ja3Y=\ncloud.google.com/go/gkemulticloud v1.4.1 h1:SvVD2nJTGScEDYygIQ5dI14oFYhgtJx8HazkT3aufEI=\ncloud.google.com/go/gkemulticloud v1.4.1/go.mod h1:KRvPYcx53bztNwNInrezdfNF+wwUom8Y3FuJBwhvFpQ=\ncloud.google.com/go/gsuiteaddons v1.5.0 h1:1mvhXqJzV0Vg5Fa95QwckljODJJfDFXV4pn+iL50zzA=\ncloud.google.com/go/gsuiteaddons v1.5.0/go.mod h1:TFCClYLd64Eaa12sFVmUyG62tk4mdIsI7pAnSXRkcFo=\ncloud.google.com/go/gsuiteaddons v1.7.2 h1:Rma+a2tCB2PV0Rm87Ywr4P96dCwGIm8vw8gF23ZlYoY=\ncloud.google.com/go/gsuiteaddons v1.7.2/go.mod h1:GD32J2rN/4APilqZw4JKmwV84+jowYYMkEVwQEYuAWc=\ncloud.google.com/go/iap v1.7.1 h1:PxVHFuMxmSZyfntKXHXhd8bo82WJ+LcATenq7HLdVnU=\ncloud.google.com/go/iap v1.7.1/go.mod h1:WapEwPc7ZxGt2jFGB/C/bm+hP0Y6NXzOYGjpPnmMS74=\ncloud.google.com/go/iap v1.10.2 h1:rvM+FNIF2wIbwUU8299FhhVGak2f7oOvbW8J/I5oflE=\ncloud.google.com/go/iap v1.10.2/go.mod h1:cClgtI09VIfazEK6VMJr6bX8KQfuQ/D3xqX+d0wrUlI=\ncloud.google.com/go/ids v1.3.0 h1:fodnCDtOXuMmS8LTC2y3h8t24U8F3eKWfhi+3LY6Qf0=\ncloud.google.com/go/ids v1.3.0/go.mod h1:JBdTYwANikFKaDP6LtW5JAi4gubs57SVNQjemdt6xV4=\ncloud.google.com/go/ids v1.5.2 h1:EDYZQraE+Eq6BewUQxVRY8b3VUUo/MnjMfzSh1NGjx8=\ncloud.google.com/go/ids v1.5.2/go.mod h1:P+ccDD96joXlomfonEdCnyrHvE68uLonc7sJBPVM5T0=\ncloud.google.com/go/iot v1.6.0 h1:39W5BFSarRNZfVG0eXI5LYux+OVQT8GkgpHCnrZL2vM=\ncloud.google.com/go/iot v1.6.0/go.mod h1:IqdAsmE2cTYYNO1Fvjfzo9po179rAtJeVGUvkLN3rLE=\ncloud.google.com/go/iot v1.8.2 h1:KMN0wujrPV7q0yfs4rt5CUl9Di8sQhJ0uohJn1h6yaI=\ncloud.google.com/go/iot v1.8.2/go.mod h1:UDwVXvRD44JIcMZr8pzpF3o4iPsmOO6fmbaIYCAg1ww=\ncloud.google.com/go/language v1.9.0 h1:7Ulo2mDk9huBoBi8zCE3ONOoBrL6UXfAI71CLQ9GEIM=\ncloud.google.com/go/language v1.9.0/go.mod h1:Ns15WooPM5Ad/5no/0n81yUetis74g3zrbeJBE+ptUY=\ncloud.google.com/go/language v1.14.2 h1:rwrIOwcAgPTYbigOaiMSjKCvBy0xHZJbRc7HB/xMECA=\ncloud.google.com/go/language v1.14.2/go.mod h1:dviAbkxT9art+2ioL9AM05t+3Ql6UPfMpwq1cDsF+rg=\ncloud.google.com/go/lifesciences v0.8.0 h1:uWrMjWTsGjLZpCTWEAzYvyXj+7fhiZST45u9AgasasI=\ncloud.google.com/go/lifesciences v0.8.0/go.mod h1:lFxiEOMqII6XggGbOnKiyZ7IBwoIqA84ClvoezaA/bo=\ncloud.google.com/go/lifesciences v0.10.2 h1:eZSaRgBwbnb/oXwCj1SGE0Kp534DuXpg55iYBWgN024=\ncloud.google.com/go/lifesciences v0.10.2/go.mod h1:vXDa34nz0T/ibUNoeHnhqI+Pn0OazUTdxemd0OLkyoY=\ncloud.google.com/go/logging v1.7.0 h1:CJYxlNNNNAMkHp9em/YEXcfJg+rPDg7YfwoRpMU+t5I=\ncloud.google.com/go/logging v1.7.0/go.mod h1:3xjP2CjkM3ZkO73aj4ASA5wRPGGCRrPIAeNqVNkzY8M=\ncloud.google.com/go/managedidentities v1.5.0 h1:ZRQ4k21/jAhrHBVKl/AY7SjgzeJwG1iZa+mJ82P+VNg=\ncloud.google.com/go/managedidentities v1.5.0/go.mod h1:+dWcZ0JlUmpuxpIDfyP5pP5y0bLdRwOS4Lp7gMni/LA=\ncloud.google.com/go/managedidentities v1.7.2 h1:oWxuIhIwQC1Vfs1SZi1x389W2TV9uyPsAyZMJgZDND4=\ncloud.google.com/go/managedidentities v1.7.2/go.mod h1:t0WKYzagOoD3FNtJWSWcU8zpWZz2i9cw2sKa9RiPx5I=\ncloud.google.com/go/maps v0.7.0 h1:mv9YaczD4oZBZkM5XJl6fXQ984IkJNHPwkc8MUsdkBo=\ncloud.google.com/go/maps v0.7.0/go.mod h1:3GnvVl3cqeSvgMcpRlQidXsPYuDGQ8naBis7MVzpXsY=\ncloud.google.com/go/maps v1.15.0 h1:bmFHlO6BL/smC6GD45r5j0ChjsyyevuJCSARdOL62TI=\ncloud.google.com/go/maps v1.15.0/go.mod h1:ZFqZS04ucwFiHSNU8TBYDUr3wYhj5iBFJk24Ibvpf3o=\ncloud.google.com/go/mediatranslation v0.7.0 h1:anPxH+/WWt8Yc3EdoEJhPMBRF7EhIdz426A+tuoA0OU=\ncloud.google.com/go/mediatranslation v0.7.0/go.mod h1:LCnB/gZr90ONOIQLgSXagp8XUW1ODs2UmUMvcgMfI2I=\ncloud.google.com/go/mediatranslation v0.9.2 h1:p37R/k9+L33bUMO87gFyv93MwJ+9nuzVhXM5X+6ULwA=\ncloud.google.com/go/mediatranslation v0.9.2/go.mod h1:1xyRoDYN32THzy+QaU62vIMciX0CFexplju9t30XwUc=\ncloud.google.com/go/memcache v1.9.0 h1:8/VEmWCpnETCrBwS3z4MhT+tIdKgR1Z4Tr2tvYH32rg=\ncloud.google.com/go/memcache v1.9.0/go.mod h1:8oEyzXCu+zo9RzlEaEjHl4KkgjlNDaXbCQeQWlzNFJM=\ncloud.google.com/go/memcache v1.11.2 h1:GGgC2A9AClJN8VLbMUAPUxj/dNMFwz6Lj01gDxPw7os=\ncloud.google.com/go/memcache v1.11.2/go.mod h1:jIzHn79b0m5wbkax2SdlW5vNSbpaEk0yWHbeLpMIYZE=\ncloud.google.com/go/metastore v1.10.0 h1:QCFhZVe2289KDBQ7WxaHV2rAmPrmRAdLC6gbjUd3HPo=\ncloud.google.com/go/metastore v1.10.0/go.mod h1:fPEnH3g4JJAk+gMRnrAnoqyv2lpUCqJPWOodSaf45Eo=\ncloud.google.com/go/metastore v1.14.2 h1:Euc9kLTKS8T6M1JVqQavwDFHu9UtT1//lGXSKjpO3/0=\ncloud.google.com/go/metastore v1.14.2/go.mod h1:dk4zOBhZIy3TFOQlI8sbOa+ef0FjAcCHEnd8dO2J+LE=\ncloud.google.com/go/monitoring v1.13.0 h1:2qsrgXGVoRXpP7otZ14eE1I568zAa92sJSDPyOJvwjM=\ncloud.google.com/go/monitoring v1.13.0/go.mod h1:k2yMBAB1H9JT/QETjNkgdCGD9bPF712XiLTVr+cBrpw=\ncloud.google.com/go/networkconnectivity v1.11.0 h1:ZD6b4Pk1jEtp/cx9nx0ZYcL3BKqDa+KixNDZ6Bjs1B8=\ncloud.google.com/go/networkconnectivity v1.11.0/go.mod h1:iWmDD4QF16VCDLXUqvyspJjIEtBR/4zq5hwnY2X3scM=\ncloud.google.com/go/networkconnectivity v1.15.2 h1:CuBLrRKhPbzXkFGADopQUpMcdY+SSfoy/3RqsMH2pq4=\ncloud.google.com/go/networkconnectivity v1.15.2/go.mod h1:N1O01bEk5z9bkkWwXLKcN2T53QN49m/pSpjfUvlHDQY=\ncloud.google.com/go/networkmanagement v1.6.0 h1:8KWEUNGcpSX9WwZXq7FtciuNGPdPdPN/ruDm769yAEM=\ncloud.google.com/go/networkmanagement v1.6.0/go.mod h1:5pKPqyXjB/sgtvB5xqOemumoQNB7y95Q7S+4rjSOPYY=\ncloud.google.com/go/networkmanagement v1.16.0 h1:oT7c2Oo9NT54XjnP4GMNj/HEywrFnBz0u6QLJ2iu8NE=\ncloud.google.com/go/networkmanagement v1.16.0/go.mod h1:Yc905R9U5jik5YMt76QWdG5WqzPU4ZsdI/mLnVa62/Q=\ncloud.google.com/go/networksecurity v0.8.0 h1:sOc42Ig1K2LiKlzG71GUVloeSJ0J3mffEBYmvu+P0eo=\ncloud.google.com/go/networksecurity v0.8.0/go.mod h1:B78DkqsxFG5zRSVuwYFRZ9Xz8IcQ5iECsNrPn74hKHU=\ncloud.google.com/go/networksecurity v0.10.2 h1://zFZM8XZZs+3Y6QKuLqwD5tZ+B/17KUo/rJpGW2tJs=\ncloud.google.com/go/networksecurity v0.10.2/go.mod h1:puU3Gwchd6Y/VTyMkL50GI2RSRMS3KXhcDBY1HSOcck=\ncloud.google.com/go/notebooks v1.8.0 h1:Kg2K3K7CbSXYJHZ1aGQpf1xi5x2GUvQWf2sFVuiZh8M=\ncloud.google.com/go/notebooks v1.8.0/go.mod h1:Lq6dYKOYOWUCTvw5t2q1gp1lAp0zxAxRycayS0iJcqQ=\ncloud.google.com/go/notebooks v1.12.2 h1:BHIH9kf/02wSCcLAVttEXHSFAgSotgRg2y1YjR7VDCc=\ncloud.google.com/go/notebooks v1.12.2/go.mod h1:EkLwv8zwr8DUXnvzl944+sRBG+b73HEKzV632YYAGNI=\ncloud.google.com/go/optimization v1.3.1 h1:dj8O4VOJRB4CUwZXdmwNViH1OtI0WtWL867/lnYH248=\ncloud.google.com/go/optimization v1.3.1/go.mod h1:IvUSefKiwd1a5p0RgHDbWCIbDFgKuEdB+fPPuP0IDLI=\ncloud.google.com/go/optimization v1.7.2 h1:yM4teRB60qyIm8cV4VRW4wepmHbXCoqv3QKGfKzylEQ=\ncloud.google.com/go/optimization v1.7.2/go.mod h1:msYgDIh1SGSfq6/KiWJQ/uxMkWq8LekPyn1LAZ7ifNE=\ncloud.google.com/go/orchestration v1.6.0 h1:Vw+CEXo8M/FZ1rb4EjcLv0gJqqw89b7+g+C/EmniTb8=\ncloud.google.com/go/orchestration v1.6.0/go.mod h1:M62Bevp7pkxStDfFfTuCOaXgaaqRAga1yKyoMtEoWPQ=\ncloud.google.com/go/orchestration v1.11.1 h1:uZOwdQoAamx8+X0UdMqY/lro3/h/Zhb7SnfArufNVcc=\ncloud.google.com/go/orchestration v1.11.1/go.mod h1:RFHf4g88Lbx6oKhwFstYiId2avwb6oswGeAQ7Tjjtfw=\ncloud.google.com/go/orgpolicy v1.10.0 h1:XDriMWug7sd0kYT1QKofRpRHzjad0bK8Q8uA9q+XrU4=\ncloud.google.com/go/orgpolicy v1.10.0/go.mod h1:w1fo8b7rRqlXlIJbVhOMPrwVljyuW5mqssvBtU18ONc=\ncloud.google.com/go/orgpolicy v1.14.1 h1:c1QLoM5v8/aDKgYVCUaC039lD3GPvqAhTVOwsGhIoZQ=\ncloud.google.com/go/orgpolicy v1.14.1/go.mod h1:1z08Hsu1mkoH839X7C8JmnrqOkp2IZRSxiDw7W/Xpg4=\ncloud.google.com/go/osconfig v1.11.0 h1:PkSQx4OHit5xz2bNyr11KGcaFccL5oqglFPdTboyqwQ=\ncloud.google.com/go/osconfig v1.11.0/go.mod h1:aDICxrur2ogRd9zY5ytBLV89KEgT2MKB2L/n6x1ooPw=\ncloud.google.com/go/osconfig v1.14.2 h1:iBN87PQc+EGh5QqijM3CuxcibvDWmF+9k0eOJT27FO4=\ncloud.google.com/go/osconfig v1.14.2/go.mod h1:kHtsm0/j8ubyuzGciBsRxFlbWVjc4c7KdrwJw0+g+pQ=\ncloud.google.com/go/oslogin v1.9.0 h1:whP7vhpmc+ufZa90eVpkfbgzJRK/Xomjz+XCD4aGwWw=\ncloud.google.com/go/oslogin v1.9.0/go.mod h1:HNavntnH8nzrn8JCTT5fj18FuJLFJc4NaZJtBnQtKFs=\ncloud.google.com/go/oslogin v1.14.2 h1:6ehIKkALrLe9zUHwEmfXRVuSPm3HiUmEnnDRr7yLIo8=\ncloud.google.com/go/oslogin v1.14.2/go.mod h1:M7tAefCr6e9LFTrdWRQRrmMeKHbkvc4D9g6tHIjHySA=\ncloud.google.com/go/phishingprotection v0.7.0 h1:l6tDkT7qAEV49MNEJkEJTB6vOO/onbSOcNtAT09HPuA=\ncloud.google.com/go/phishingprotection v0.7.0/go.mod h1:8qJI4QKHoda/sb/7/YmMQ2omRLSLYSu9bU0EKCNI+Lk=\ncloud.google.com/go/phishingprotection v0.9.2 h1:SaW0IPf/1fflnzomjy7+9EMtReXuxkYpUAf/77m5xL8=\ncloud.google.com/go/phishingprotection v0.9.2/go.mod h1:mSCiq3tD8fTJAuXq5QBHFKZqMUy8SfWsbUM9NpzJIRQ=\ncloud.google.com/go/policytroubleshooter v1.6.0 h1:yKAGC4p9O61ttZUswaq9GAn1SZnEzTd0vUYXD7ZBT7Y=\ncloud.google.com/go/policytroubleshooter v1.6.0/go.mod h1:zYqaPTsmfvpjm5ULxAyD/lINQxJ0DDsnWOP/GZ7xzBc=\ncloud.google.com/go/policytroubleshooter v1.11.2 h1:sTIH5AQ8tcgmnqrqlZfYWymjMhPh4ZEt4CvQGgG+kzc=\ncloud.google.com/go/policytroubleshooter v1.11.2/go.mod h1:1TdeCRv8Qsjcz2qC3wFltg/Mjga4HSpv8Tyr5rzvPsw=\ncloud.google.com/go/privatecatalog v0.8.0 h1:EPEJ1DpEGXLDnmc7mnCAqFmkwUJbIsaLAiLHVOkkwtc=\ncloud.google.com/go/privatecatalog v0.8.0/go.mod h1:nQ6pfaegeDAq/Q5lrfCQzQLhubPiZhSaNhIgfJlnIXs=\ncloud.google.com/go/privatecatalog v0.10.2 h1:01RPfn8IL2//8UHAmImRraTFYM/3gAEiIxudWLWrp+0=\ncloud.google.com/go/privatecatalog v0.10.2/go.mod h1:o124dHoxdbO50ImR3T4+x3GRwBSTf4XTn6AatP8MgsQ=\ncloud.google.com/go/pubsub v1.30.0 h1:vCge8m7aUKBJYOgrZp7EsNDf6QMd2CAlXZqWTn3yq6s=\ncloud.google.com/go/pubsub v1.30.0/go.mod h1:qWi1OPS0B+b5L+Sg6Gmc9zD1Y+HaM0MdUr7LsupY1P4=\ncloud.google.com/go/pubsub v1.45.1 h1:ZC/UzYcrmK12THWn1P72z+Pnp2vu/zCZRXyhAfP1hJY=\ncloud.google.com/go/pubsub v1.45.1/go.mod h1:3bn7fTmzZFwaUjllitv1WlsNMkqBgGUb3UdMhI54eCc=\ncloud.google.com/go/pubsublite v1.7.0 h1:cb9fsrtpINtETHiJ3ECeaVzrfIVhcGjhhJEjybHXHao=\ncloud.google.com/go/pubsublite v1.7.0/go.mod h1:8hVMwRXfDfvGm3fahVbtDbiLePT3gpoiJYJY+vxWxVM=\ncloud.google.com/go/pubsublite v1.8.2 h1:jLQozsEVr+c6tOU13vDugtnaBSUy/PD5zK6mhm+uF1Y=\ncloud.google.com/go/pubsublite v1.8.2/go.mod h1:4r8GSa9NznExjuLPEJlF1VjOPOpgf3IT6k8x/YgaOPI=\ncloud.google.com/go/recaptchaenterprise/v2 v2.7.0 h1:6iOCujSNJ0YS7oNymI64hXsjGq60T4FK1zdLugxbzvU=\ncloud.google.com/go/recaptchaenterprise/v2 v2.7.0/go.mod h1:19wVj/fs5RtYtynAPJdDTb69oW0vNHYDBTbB4NvMD9c=\ncloud.google.com/go/recaptchaenterprise/v2 v2.19.0 h1:J/J7ZeVOX+sqn0hxzkOBfnQfBAzPZt8KaAuQoarQWQM=\ncloud.google.com/go/recaptchaenterprise/v2 v2.19.0/go.mod h1:vnbA2SpVPPwKeoFrCQxR+5a0JFRRytwBBG69Zj9pGfk=\ncloud.google.com/go/recommendationengine v0.7.0 h1:VibRFCwWXrFebEWKHfZAt2kta6pS7Tlimsnms0fjv7k=\ncloud.google.com/go/recommendationengine v0.7.0/go.mod h1:1reUcE3GIu6MeBz/h5xZJqNLuuVjNg1lmWMPyjatzac=\ncloud.google.com/go/recommendationengine v0.9.2 h1:RHVdmoNBdzgRJXI/3SV+GB5TTv/umsVguiaEvmKOh98=\ncloud.google.com/go/recommendationengine v0.9.2/go.mod h1:DjGfWZJ68ZF5ZuNgoTVXgajFAG0yLt4CJOpC0aMK3yw=\ncloud.google.com/go/recommender v1.9.0 h1:ZnFRY5R6zOVk2IDS1Jbv5Bw+DExCI5rFumsTnMXiu/A=\ncloud.google.com/go/recommender v1.9.0/go.mod h1:PnSsnZY7q+VL1uax2JWkt/UegHssxjUVVCrX52CuEmQ=\ncloud.google.com/go/recommender v1.13.2 h1:xDFzlFk5Xp5MXnac468eicKM3MUo6UNdxoYuBMOF1mE=\ncloud.google.com/go/recommender v1.13.2/go.mod h1:XJau4M5Re8F4BM+fzF3fqSjxNJuM66fwF68VCy/ngGE=\ncloud.google.com/go/redis v1.11.0 h1:JoAd3SkeDt3rLFAAxEvw6wV4t+8y4ZzfZcZmddqphQ8=\ncloud.google.com/go/redis v1.11.0/go.mod h1:/X6eicana+BWcUda5PpwZC48o37SiFVTFSs0fWAJ7uQ=\ncloud.google.com/go/redis v1.17.2 h1:QbW264RBH+NSVEQqlDoHfoxcreXK8QRRByTOR2CFbJs=\ncloud.google.com/go/redis v1.17.2/go.mod h1:h071xkcTMnJgQnU/zRMOVKNj5J6AttG16RDo+VndoNo=\ncloud.google.com/go/resourcemanager v1.7.0 h1:NRM0p+RJkaQF9Ee9JMnUV9BQ2QBIOq/v8M+Pbv/wmCs=\ncloud.google.com/go/resourcemanager v1.7.0/go.mod h1:HlD3m6+bwhzj9XCouqmeiGuni95NTrExfhoSrkC/3EI=\ncloud.google.com/go/resourcemanager v1.10.2 h1:LpqZZGM0uJiu1YWM878AA8zZ/qOQ/Ngno60Q8RAraAI=\ncloud.google.com/go/resourcemanager v1.10.2/go.mod h1:5f+4zTM/ZOTDm6MmPOp6BQAhR0fi8qFPnvVGSoWszcc=\ncloud.google.com/go/resourcesettings v1.5.0 h1:8Dua37kQt27CCWHm4h/Q1XqCF6ByD7Ouu49xg95qJzI=\ncloud.google.com/go/resourcesettings v1.5.0/go.mod h1:+xJF7QSG6undsQDfsCJyqWXyBwUoJLhetkRMDRnIoXA=\ncloud.google.com/go/resourcesettings v1.8.2 h1:ISRX2HZHNS17F/EuIwzPrQwEyIyUJayGuLrS51yt6Wk=\ncloud.google.com/go/resourcesettings v1.8.2/go.mod h1:uEgtPiMA+xuBUM4Exu+ZkNpMYP0BLlYeJbyNHfrc+U0=\ncloud.google.com/go/retail v1.12.0 h1:1Dda2OpFNzIb4qWgFZjYlpP7sxX3aLeypKG6A3H4Yys=\ncloud.google.com/go/retail v1.12.0/go.mod h1:UMkelN/0Z8XvKymXFbD4EhFJlYKRx1FGhQkVPU5kF14=\ncloud.google.com/go/retail v1.19.1 h1:FVzvA+VuEdNoMz2WzWZ5KwfG+CX+jSv+SOspyQPLuRs=\ncloud.google.com/go/retail v1.19.1/go.mod h1:W48zg0zmt2JMqmJKCuzx0/0XDLtovwzGAeJjmv6VPaE=\ncloud.google.com/go/run v0.9.0 h1:ydJQo+k+MShYnBfhaRHSZYeD/SQKZzZLAROyfpeD9zw=\ncloud.google.com/go/run v0.9.0/go.mod h1:Wwu+/vvg8Y+JUApMwEDfVfhetv30hCG4ZwDR/IXl2Qg=\ncloud.google.com/go/run v1.7.0 h1:GJtHWUgi8CK+YPhmTR3tKBAmDmU9RRMYqiGKCmIgFG8=\ncloud.google.com/go/run v1.7.0/go.mod h1:IvJOg2TBb/5a0Qkc6crn5yTy5nkjcgSWQLhgO8QL8PQ=\ncloud.google.com/go/scheduler v1.9.0 h1:NpQAHtx3sulByTLe2dMwWmah8PWgeoieFPpJpArwFV0=\ncloud.google.com/go/scheduler v1.9.0/go.mod h1:yexg5t+KSmqu+njTIh3b7oYPheFtBWGcbVUYF1GGMIc=\ncloud.google.com/go/scheduler v1.11.2 h1:PfkvJP1qKu9NvFB65Ja/s918bPZWMBcYkg35Ljdw1Oc=\ncloud.google.com/go/scheduler v1.11.2/go.mod h1:GZSv76T+KTssX2I9WukIYQuQRf7jk1WI+LOcIEHUUHk=\ncloud.google.com/go/secretmanager v1.10.0 h1:pu03bha7ukxF8otyPKTFdDz+rr9sE3YauS5PliDXK60=\ncloud.google.com/go/secretmanager v1.10.0/go.mod h1:MfnrdvKMPNra9aZtQFvBcvRU54hbPD8/HayQdlUgJpU=\ncloud.google.com/go/secretmanager v1.14.2 h1:2XscWCfy//l/qF96YE18/oUaNJynAx749Jg3u0CjQr8=\ncloud.google.com/go/secretmanager v1.14.2/go.mod h1:Q18wAPMM6RXLC/zVpWTlqq2IBSbbm7pKBlM3lCKsmjw=\ncloud.google.com/go/security v1.13.0 h1:PYvDxopRQBfYAXKAuDpFCKBvDOWPWzp9k/H5nB3ud3o=\ncloud.google.com/go/security v1.13.0/go.mod h1:Q1Nvxl1PAgmeW0y3HTt54JYIvUdtcpYKVfIB8AOMZ+0=\ncloud.google.com/go/security v1.18.2 h1:9Nzp9LGjiDvHqy7X7Q9GrS5lIHN0bI8RvDjkrl4ILO0=\ncloud.google.com/go/security v1.18.2/go.mod h1:3EwTcYw8554iEtgK8VxAjZaq2unFehcsgFIF9nOvQmU=\ncloud.google.com/go/securitycenter v1.19.0 h1:AF3c2s3awNTMoBtMX3oCUoOMmGlYxGOeuXSYHNBkf14=\ncloud.google.com/go/securitycenter v1.19.0/go.mod h1:LVLmSg8ZkkyaNy4u7HCIshAngSQ8EcIRREP3xBnyfag=\ncloud.google.com/go/securitycenter v1.35.2 h1:XkkE+IRE5/88drGPIuvETCSN7dAnWoqJahZzDbP5Hog=\ncloud.google.com/go/securitycenter v1.35.2/go.mod h1:AVM2V9CJvaWGZRHf3eG+LeSTSissbufD27AVBI91C8s=\ncloud.google.com/go/servicecontrol v1.11.1 h1:d0uV7Qegtfaa7Z2ClDzr9HJmnbJW7jn0WhZ7wOX6hLE=\ncloud.google.com/go/servicecontrol v1.11.1/go.mod h1:aSnNNlwEFBY+PWGQ2DoM0JJ/QUXqV5/ZD9DOLB7SnUk=\ncloud.google.com/go/servicedirectory v1.9.0 h1:SJwk0XX2e26o25ObYUORXx6torSFiYgsGkWSkZgkoSU=\ncloud.google.com/go/servicedirectory v1.9.0/go.mod h1:29je5JjiygNYlmsGz8k6o+OZ8vd4f//bQLtvzkPPT/s=\ncloud.google.com/go/servicedirectory v1.12.2 h1:W/oZmTUzlWbeSTujRbmG9v7HZyHcorj608tkcD3vVYE=\ncloud.google.com/go/servicedirectory v1.12.2/go.mod h1:F0TJdFjqqotiZRlMXgIOzszaplk4ZAmUV8ovHo08M2U=\ncloud.google.com/go/servicemanagement v1.8.0 h1:fopAQI/IAzlxnVeiKn/8WiV6zKndjFkvi+gzu+NjywY=\ncloud.google.com/go/servicemanagement v1.8.0/go.mod h1:MSS2TDlIEQD/fzsSGfCdJItQveu9NXnUniTrq/L8LK4=\ncloud.google.com/go/serviceusage v1.6.0 h1:rXyq+0+RSIm3HFypctp7WoXxIA563rn206CfMWdqXX4=\ncloud.google.com/go/serviceusage v1.6.0/go.mod h1:R5wwQcbOWsyuOfbP9tGdAnCAc6B9DRwPG1xtWMDeuPA=\ncloud.google.com/go/shell v1.6.0 h1:wT0Uw7ib7+AgZST9eCDygwTJn4+bHMDtZo5fh7kGWDU=\ncloud.google.com/go/shell v1.6.0/go.mod h1:oHO8QACS90luWgxP3N9iZVuEiSF84zNyLytb+qE2f9A=\ncloud.google.com/go/shell v1.8.2 h1:lSfdEng3n7zZHzC40BJ4trEMyme3CGnLLnA09MlLQdQ=\ncloud.google.com/go/shell v1.8.2/go.mod h1:QQR12T6j/eKvqAQLv6R3ozeoqwJ0euaFSz2qLqG93Bs=\ncloud.google.com/go/spanner v1.45.0 h1:7VdjZ8zj4sHbDw55atp5dfY6kn1j9sam9DRNpPQhqR4=\ncloud.google.com/go/spanner v1.45.0/go.mod h1:FIws5LowYz8YAE1J8fOS7DJup8ff7xJeetWEo5REA2M=\ncloud.google.com/go/spanner v1.73.0 h1:0bab8QDn6MNj9lNK6XyGAVFhMlhMU2waePPa6GZNoi8=\ncloud.google.com/go/spanner v1.73.0/go.mod h1:mw98ua5ggQXVWwp83yjwggqEmW9t8rjs9Po1ohcUGW4=\ncloud.google.com/go/speech v1.15.0 h1:JEVoWGNnTF128kNty7T4aG4eqv2z86yiMJPT9Zjp+iw=\ncloud.google.com/go/speech v1.15.0/go.mod h1:y6oH7GhqCaZANH7+Oe0BhgIogsNInLlz542tg3VqeYI=\ncloud.google.com/go/speech v1.25.2 h1:rKOXU9LAZTOYHhRNB4gZDekNjJx21TktQpetBa5IzOk=\ncloud.google.com/go/speech v1.25.2/go.mod h1:KPFirZlLL8SqPaTtG6l+HHIFHPipjbemv4iFg7rTlYs=\ncloud.google.com/go/storagetransfer v1.8.0 h1:5T+PM+3ECU3EY2y9Brv0Sf3oka8pKmsCfpQ07+91G9o=\ncloud.google.com/go/storagetransfer v1.8.0/go.mod h1:JpegsHHU1eXg7lMHkvf+KE5XDJ7EQu0GwNJbbVGanEw=\ncloud.google.com/go/storagetransfer v1.11.2 h1:hMcP8ECmxedXjPxr2j3Ca45ro/TKEF+1YYjq2p5LMTI=\ncloud.google.com/go/storagetransfer v1.11.2/go.mod h1:FcM29aY4EyZ3yVPmW5SxhqUdhjgPBUOFyy4rqiQbias=\ncloud.google.com/go/talent v1.5.0 h1:nI9sVZPjMKiO2q3Uu0KhTDVov3Xrlpt63fghP9XjyEM=\ncloud.google.com/go/talent v1.5.0/go.mod h1:G+ODMj9bsasAEJkQSzO2uHQWXHHXUomArjWQQYkqK6c=\ncloud.google.com/go/talent v1.7.2 h1:KONR7KX/EXI3pO2cbSIDOBqhBzvgDS71vaMz8k4qRCg=\ncloud.google.com/go/talent v1.7.2/go.mod h1:k1sqlDgS9gbc0gMTRuRQpX6C6VB7bGUxSPcoTRWJod8=\ncloud.google.com/go/texttospeech v1.6.0 h1:H4g1ULStsbVtalbZGktyzXzw6jP26RjVGYx9RaYjBzc=\ncloud.google.com/go/texttospeech v1.6.0/go.mod h1:YmwmFT8pj1aBblQOI3TfKmwibnsfvhIBzPXcW4EBovc=\ncloud.google.com/go/texttospeech v1.10.0 h1:icRAxYDtq3zO1T0YBT/fe8C/7pXoIqfkY4iYr5zG39I=\ncloud.google.com/go/texttospeech v1.10.0/go.mod h1:215FpCOyRxxrS7DSb2t7f4ylMz8dXsQg8+Vdup5IhP4=\ncloud.google.com/go/tpu v1.5.0 h1:/34T6CbSi+kTv5E19Q9zbU/ix8IviInZpzwz3rsFE+A=\ncloud.google.com/go/tpu v1.5.0/go.mod h1:8zVo1rYDFuW2l4yZVY0R0fb/v44xLh3llq7RuV61fPM=\ncloud.google.com/go/tpu v1.7.2 h1:xPBJd7xZgtl3CgrZoaUf7zFPVVj68jmzzGTSzkcsOtQ=\ncloud.google.com/go/tpu v1.7.2/go.mod h1:0Y7dUo2LIbDUx0yQ/vnLC6e18FK6NrDfAhYS9wZ/2vs=\ncloud.google.com/go/trace v1.9.0 h1:olxC0QHC59zgJVALtgqfD9tGk0lfeCP5/AGXL3Px/no=\ncloud.google.com/go/trace v1.9.0/go.mod h1:lOQqpE5IaWY0Ixg7/r2SjixMuc6lfTFeO4QGM4dQWOk=\ncloud.google.com/go/translate v1.7.0 h1:GvLP4oQ4uPdChBmBaUSa/SaZxCdyWELtlAaKzpHsXdA=\ncloud.google.com/go/translate v1.7.0/go.mod h1:lMGRudH1pu7I3n3PETiOB2507gf3HnfLV8qlkHZEyos=\ncloud.google.com/go/translate v1.12.2 h1:qECivi8O+jFI/vnvN9elK6CME+WAWy56GIBszF+/rNc=\ncloud.google.com/go/translate v1.12.2/go.mod h1:jjLVf2SVH2uD+BNM40DYvRRKSsuyKxVvs3YjTW/XSWY=\ncloud.google.com/go/video v1.15.0 h1:upIbnGI0ZgACm58HPjAeBMleW3sl5cT84AbYQ8PWOgM=\ncloud.google.com/go/video v1.15.0/go.mod h1:SkgaXwT+lIIAKqWAJfktHT/RbgjSuY6DobxEp0C5yTQ=\ncloud.google.com/go/video v1.23.2 h1:CGAPOXTJMoZm9PeHkohBlMTy8lqN6VWCNDjp5VODfy8=\ncloud.google.com/go/video v1.23.2/go.mod h1:rNOr2pPHWeCbW0QsOwJRIe0ZiuwHpHtumK0xbiYB1Ew=\ncloud.google.com/go/videointelligence v1.10.0 h1:Uh5BdoET8XXqXX2uXIahGb+wTKbLkGH7s4GXR58RrG8=\ncloud.google.com/go/videointelligence v1.10.0/go.mod h1:LHZngX1liVtUhZvi2uNS0VQuOzNi2TkY1OakiuoUOjU=\ncloud.google.com/go/videointelligence v1.12.2 h1:ZLElysepw9vfQGAKWfnxdnSnHSKbEn/nU/tmBnCJLfA=\ncloud.google.com/go/videointelligence v1.12.2/go.mod h1:8xKGlq0lNVyT8JgTkkCUCpyNJnYYEJVWGdqzv+UcwR8=\ncloud.google.com/go/vision/v2 v2.7.0 h1:8C8RXUJoflCI4yVdqhTy9tRyygSHmp60aP363z23HKg=\ncloud.google.com/go/vision/v2 v2.7.0/go.mod h1:H89VysHy21avemp6xcf9b9JvZHVehWbET0uT/bcuY/0=\ncloud.google.com/go/vision/v2 v2.9.2 h1:u4pu3gKps88oUe76WwVPeX9dgWVyyYopZ1s05FwsKEk=\ncloud.google.com/go/vision/v2 v2.9.2/go.mod h1:WuxjVQdAy4j4WZqY5Rr655EdAgi8B707Vdb5T8c90uo=\ncloud.google.com/go/vmmigration v1.6.0 h1:Azs5WKtfOC8pxvkyrDvt7J0/4DYBch0cVbuFfCCFt5k=\ncloud.google.com/go/vmmigration v1.6.0/go.mod h1:bopQ/g4z+8qXzichC7GW1w2MjbErL54rk3/C843CjfY=\ncloud.google.com/go/vmmigration v1.8.2 h1:Hpqv3fZ3Ri1OMhTNVJgxxsTou2ZlRzKbnc1dSybTP5Y=\ncloud.google.com/go/vmmigration v1.8.2/go.mod h1:FBejrsr8ZHmJb949BSOyr3D+/yCp9z9Hk0WtsTiHc1Q=\ncloud.google.com/go/vmwareengine v0.3.0 h1:b0NBu7S294l0gmtrT0nOJneMYgZapr5x9tVWvgDoVEM=\ncloud.google.com/go/vmwareengine v0.3.0/go.mod h1:wvoyMvNWdIzxMYSpH/R7y2h5h3WFkx6d+1TIsP39WGY=\ncloud.google.com/go/vmwareengine v1.3.2 h1:LmkojgSLvsRwU1+c0iiY2XoBkXYKzpArElHC9IDWakg=\ncloud.google.com/go/vmwareengine v1.3.2/go.mod h1:JsheEadzT0nfXOGkdnwtS1FhFAnj4g8qhi4rKeLi/AU=\ncloud.google.com/go/vpcaccess v1.6.0 h1:FOe6CuiQD3BhHJWt7E8QlbBcaIzVRddupwJlp7eqmn4=\ncloud.google.com/go/vpcaccess v1.6.0/go.mod h1:wX2ILaNhe7TlVa4vC5xce1bCnqE3AeH27RV31lnmZes=\ncloud.google.com/go/vpcaccess v1.8.2 h1:nvrkqAjS2sorOu4YGCIXWz+Kk+5aAAdnaMD2tnsqeFg=\ncloud.google.com/go/vpcaccess v1.8.2/go.mod h1:4yvYKNjlNjvk/ffgZ0PuEhpzNJb8HybSM1otG2aDxnY=\ncloud.google.com/go/webrisk v1.8.0 h1:IY+L2+UwxcVm2zayMAtBhZleecdIFLiC+QJMzgb0kT0=\ncloud.google.com/go/webrisk v1.8.0/go.mod h1:oJPDuamzHXgUc+b8SiHRcVInZQuybnvEW72PqTc7sSg=\ncloud.google.com/go/webrisk v1.10.2 h1:X7zSwS1mX2bxoZ30Ozh6lqiSLezl7RMBWwp5a3Mkxp4=\ncloud.google.com/go/webrisk v1.10.2/go.mod h1:c0ODT2+CuKCYjaeHO7b0ni4CUrJ95ScP5UFl9061Qq8=\ncloud.google.com/go/websecurityscanner v1.5.0 h1:AHC1xmaNMOZtNqxI9Rmm87IJEyPaRkOxeI0gpAacXGk=\ncloud.google.com/go/websecurityscanner v1.5.0/go.mod h1:Y6xdCPy81yi0SQnDY1xdNTNpfY1oAgXUlcfN3B3eSng=\ncloud.google.com/go/websecurityscanner v1.7.2 h1:8/4rfJXcyxozbfzI0lDFPcPShRE6bJ4HQwgDAG9J4oQ=\ncloud.google.com/go/websecurityscanner v1.7.2/go.mod h1:728wF9yz2VCErfBaACA5px2XSYHQgkK812NmHcUsDXA=\ncloud.google.com/go/workflows v1.10.0 h1:FfGp9w0cYnaKZJhUOMqCOJCYT/WlvYBfTQhFWV3sRKI=\ncloud.google.com/go/workflows v1.10.0/go.mod h1:fZ8LmRmZQWacon9UCX1r/g/DfAXx5VcPALq2CxzdePw=\ncloud.google.com/go/workflows v1.13.2 h1:jYIxrDOVCGvTBHIAVhqQ+P8fhE0trm+Hf2hgL1YzmK0=\ncloud.google.com/go/workflows v1.13.2/go.mod h1:l5Wj2Eibqba4BsADIRzPLaevLmIuYF2W+wfFBkRG3vU=\ncontrib.go.opencensus.io/exporter/aws v0.0.0-20200617204711-c478e41e60e9 h1:yxE46rQA0QaqPGqN2UnwXvgCrRqtjR1CsGSWVTRjvv4=\ncontrib.go.opencensus.io/exporter/stackdriver v0.13.5 h1:TNaexHK16gPUoc7uzELKOU7JULqccn1NDuqUxmxSqfo=\ncontrib.go.opencensus.io/integrations/ocsql v0.1.7 h1:G3k7C0/W44zcqkpRSFyjU9f6HZkbwIrL//qqnlqWZ60=\ndmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9 h1:VpgP7xuJadIUuKccphEpTJnWhS2jkQyMt6Y7pJCD7fY=\ngithub.com/Azure/azure-amqp-common-go/v3 v3.1.0 h1:1N4YSkWYWffOpQHromYdOucBSQXhNRKzqtgICy6To8Q=\ngithub.com/Azure/azure-service-bus-go v0.10.11 h1:GBg2mcLQA3af+w+ZuYhiAv58OWSVmQYWcds4nro8Xko=\ngithub.com/Azure/go-amqp v0.13.7 h1:ukcCtx138ZmOfHbdALuh9yoJhGtOY3+yaKApfzNvhSk=\ngithub.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802 h1:1BDTz0u9nC3//pOCMdNH+CiXJVYJh5UQNCOBG7jbELc=\ngithub.com/DataDog/datadog-go v3.2.0+incompatible h1:qSG2N4FghB1He/r2mFrWKCaL7dXCilEuNEeAn20fdD4=\ngithub.com/GoogleCloudPlatform/cloudsql-proxy v1.22.0 h1:aTDBS16pX1X4ZR/GFsC2NcOCYJ1hDJwJm3WmKRA905Q=\ngithub.com/OneOfOne/xxhash v1.2.2 h1:KMrpdQIwFcEqXDklaen+P1axHaj9BSKzvpUUfnHldSE=\ngithub.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751 h1:JYp7IbQjafoB+tBA3gMyHYHrpOtNuDiK/uB5uXxq5wM=\ngithub.com/alecthomas/units v0.0.0-20190924025748-f65c72e2690d h1:UQZhZ2O0vMHr2cI+DC1Mbh0TJxzA3RcLoMsFw+aXw7E=\ngithub.com/antihax/optional v1.0.0 h1:xK2lYat7ZLaVVcIuj82J8kIro4V6kDe0AUDFboUCwcg=\ngithub.com/aokoli/goutils v1.0.1 h1:7fpzNGoJ3VA8qcrm++XEE1QUe0mIwNeLa02Nwq7RDkg=\ngithub.com/apache/thrift v0.0.0-20161221203622-b2a4d4ae21c7 h1:Fv9bK1Q+ly/ROk4aJsVMeuIwPel4bEnD8EPiI91nZMg=\ngithub.com/apex/logs v1.0.0 h1:adOwhOTeXzZTnVuEK13wuJNBFutP0sOfutRS8NY+G6A=\ngithub.com/aphistic/golf v0.0.0-20180712155816-02c07f170c5a h1:2KLQMJ8msqoPHIPDufkxVcoTtcmE5+1sL9950m4R9Pk=\ngithub.com/aphistic/sweet v0.2.0 h1:I4z+fAUqvKfvZV/CHi5dV0QuwbmIvYYFDjG0Ss5QpAs=\ngithub.com/armon/circbuf v0.0.0-20150827004946-bbbad097214e h1:QEF07wC0T1rKkctt1RINW/+RMTVmiwxETico2l3gxJA=\ngithub.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6 h1:G1bPvciwNyF7IUmKXNt9Ak3m6u9DE1rF+RmtIkBpVdA=\ngithub.com/armon/go-metrics v0.3.10 h1:FR+drcQStOe+32sYyJYyZ7FIdgoGGBnwLl+flodp8Uo=\ngithub.com/armon/go-metrics v0.4.0 h1:yCQqn7dwca4ITXb+CbubHmedzaQYHhNhrEXLYUeEe8Q=\ngithub.com/armon/go-metrics v0.4.0/go.mod h1:E6amYzXo6aW1tqzoZGT755KkbgrJsSdpwZ+3JqfkOG4=\ngithub.com/armon/go-radix v1.0.0 h1:F4z6KzEeeQIMeLFa97iZU6vupzoecKdU5TX24SNppXI=\ngithub.com/aybabtme/rgbterm v0.0.0-20170906152045-cc83f3b3ce59 h1:WWB576BN5zNSZc/M9d/10pqEx5VHNhaQ/yOVAkmj5Yo=\ngithub.com/bazelbuild/rules_go v0.49.0 h1:5vCbuvy8Q11g41lseGJDc5vxhDjJtfxr6nM/IC4VmqM=\ngithub.com/bazelbuild/rules_go v0.49.0/go.mod h1:Dhcz716Kqg1RHNWos+N6MlXNkjNP2EwZQ0LukRKJfMs=\ngithub.com/bgentry/speakeasy v0.1.0 h1:ByYyxL9InA1OWqxJqqp2A5pYHUrCiAL6K3J+LKSsQkY=\ngithub.com/bketelsen/crypt v0.0.3-0.20200106085610-5cbc8cc4026c h1:+0HFd5KSZ/mm3JmhmrDukiId5iR6w4+BdFtfSy4yWIc=\ngithub.com/bmizerany/perks v0.0.0-20141205001514-d9a9656a3a4b h1:AP/Y7sqYicnjGDfD5VcY4CIfh1hRXBUavxrvELjTiOE=\ngithub.com/cactus/go-statsd-client/statsd v0.0.0-20191106001114-12b4e2b38748 h1:bXxS5/Z3/dfc8iFniQfgogNBomo0u+1//9eP+jl8GVo=\ngithub.com/cavaliercoder/go-cpio v0.0.0-20180626203310-925f9528c45e h1:hHg27A0RSSp2Om9lubZpiMgVbvn39bsUmW9U5h0twqc=\ngithub.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=\ngithub.com/chzyer/logex v1.1.10 h1:Swpa1K6QvQznwJRcfTfQJmTE72DqScAa40E+fbHEXEE=\ngithub.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e h1:fY5BOSpyZCqRo5OhCuC+XN+r/bBCmeuuJtjz+bCNIf8=\ngithub.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1 h1:q763qf9huN11kDQavWsoZXJNW3xEE4JJyHa5Q25/sd8=\ngithub.com/circonus-labs/circonus-gometrics v2.3.1+incompatible h1:C29Ae4G5GtYyYMm1aztcyj/J5ckgJm2zwdDajFbx1NY=\ngithub.com/circonus-labs/circonusllhist v0.1.3 h1:TJH+oke8D16535+jHExHj4nQvzlZrj7ug5D7I/orNUA=\ngithub.com/client9/misspell v0.3.4 h1:ta993UF76GwbvJcIo3Y68y/M3WxlpEHPWIGDkJYwzJI=\ngithub.com/cncf/udpa/go v0.0.0-20210930031921-04548b0d99d4 h1:hzAQntlaYRkVSFEfj9OTWlVV1H155FMD8BTKktLv0QI=\ngithub.com/cncf/udpa/go v0.0.0-20220112060539-c52dc94e7fbe h1:QQ3GSy+MqSHxm/d8nCtnAiZdYFd45cYZPs8vOOIYKfk=\ngithub.com/cncf/udpa/go v0.0.0-20220112060539-c52dc94e7fbe/go.mod h1:6pvJx4me5XPnfI9Z40ddWsdw2W/uZgQLFXToKeRcDiI=\ngithub.com/cncf/xds/go v0.0.0-20230607035331-e9ce68804cb4 h1:/inchEIKaYC1Akx+H+gqO04wryn5h75LSazbRlnya1k=\ngithub.com/cncf/xds/go v0.0.0-20230607035331-e9ce68804cb4/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs=\ngithub.com/cockroachdb/datadriven v0.0.0-20190809214429-80d97fb3cbaa h1:OaNxuTZr7kxeODyLWsRMC+OD03aFUH+mW6r2d+MWa5Y=\ngithub.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd h1:qMd81Ts1T2OTKmB4acZcyKaMtRnY5Y44NuXGX2GFJ1w=\ngithub.com/coreos/bbolt v1.3.2 h1:wZwiHHUieZCquLkDL0B8UhzreNWsPHooDAG3q34zk0s=\ngithub.com/coreos/etcd v3.3.13+incompatible h1:8F3hqu9fGYLBifCmRCJsicFqDx/D68Rt3q1JMazcgBQ=\ngithub.com/coreos/go-etcd v2.0.0+incompatible h1:bXhRBIXoTm9BYHS3gE0TtQuyNZyeEMux2sDi4oo5YOo=\ngithub.com/coreos/go-semver v0.3.0 h1:wkHLiw0WNATZnSG7epLsujiMCgPAc9xhjJ4tgnAxmfM=\ngithub.com/coreos/go-systemd v0.0.0-20190620071333-e64a0ec8b42a h1:W8b4lQ4tFF21aspRGoBuCNV6V2fFJBF+pm1J6OY8Lys=\ngithub.com/coreos/go-systemd/v22 v22.3.3-0.20220203105225-a9a7ef127534 h1:rtAn27wIbmOGUs7RIbVgPEjb31ehTVniDwPGXyMxm5U=\ngithub.com/coreos/go-systemd/v22 v22.5.0 h1:RrqgGjYQKalulkV8NGVIfkXQf6YYmOyiJKk8iXXhfZs=\ngithub.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f h1:lBNOc5arjvs8E5mO2tbpBpLoyyu8B6e44T7hJy6potg=\ngithub.com/cpuguy83/go-md2man v1.0.10 h1:BSKMNlYxDvnunlTymqtgONjNnaRV1sTpcovwwjF22jk=\ngithub.com/creack/pty v1.1.9 h1:uDmaGzcdjhF4i/plgjmEsriH11Y0o7RKapEf/LDaM3w=\ngithub.com/denis-tingajkin/go-header v0.4.2 h1:jEeSF4sdv8/3cT/WY8AgDHUoItNSoEZ7qg9dX7pc218=\ngithub.com/denis-tingajkin/go-header v0.4.2/go.mod h1:eLRHAVXzE5atsKAnNRDB90WHCFFnBUn4RN0nRcs1LJA=\ngithub.com/denisenkom/go-mssqldb v0.9.0 h1:RSohk2RsiZqLZ0zCjtfn3S4Gp4exhpBWHyQ7D0yGjAk=\ngithub.com/devigned/tab v0.1.1 h1:3mD6Kb1mUOYeLpJvTVSDwSg5ZsfSxfvxGRTxRsJsITA=\ngithub.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=\ngithub.com/dgryski/go-farm v0.0.0-20200201041132-a6ae2369ad13 h1:fAjc9m62+UWV/WAFKLNi6ZS0675eEUC9y3AlwSbQu1Y=\ngithub.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954 h1:RMLoZVzv4GliuWafOuPuQDKSm1SJph7uCRnnS61JAn4=\ngithub.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=\ngithub.com/envoyproxy/go-control-plane v0.11.1-0.20230524094728-9239064ad72f h1:7T++XKzy4xg7PKy+bM+Sa9/oe1OC88yz2hXQUISoXfA=\ngithub.com/envoyproxy/go-control-plane v0.11.1-0.20230524094728-9239064ad72f/go.mod h1:sfYdkwUW4BA3PbKjySwjJy+O4Pu0h62rlqCMHNk+K+Q=\ngithub.com/envoyproxy/protoc-gen-validate v0.10.1 h1:c0g45+xCJhdgFGw7a5QAfdS4byAbud7miNWJ1WwEVf8=\ngithub.com/envoyproxy/protoc-gen-validate v0.10.1/go.mod h1:DRjgyB0I43LtJapqN6NiRwroiAU2PaFuvk/vjgh61ss=\ngithub.com/fortytw2/leaktest v1.3.0 h1:u8491cBMTQ8ft8aeV+adlcytMZylmA5nnwwkRZjI8vw=\ngithub.com/fullstorydev/grpcurl v1.6.0 h1:p8BB6VZF8O7w6MxGr3KJ9E6EVKaswCevSALK6FBtMzA=\ngithub.com/ghodss/yaml v1.0.0 h1:wQHKEahhL6wmXdzwWG11gIVCkOv05bNOh+Rxn0yngAk=\ngithub.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=\ngithub.com/gin-gonic/gin v1.6.3 h1:ahKqKTFpO5KTPHxWZjEdPScmYaGtLo8Y4DMHoEsnp14=\ngithub.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1 h1:QbL/5oDUmRBzO9/Z7Seo6zf912W/a6Sr4Eu0G/3Jho0=\ngithub.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4 h1:WtGNWLvXpe6ZudgnXrq0barxBImvnnJoMEhXAzcbM0I=\ngithub.com/go-ini/ini v1.25.4 h1:Mujh4R/dH6YL8bxuISne3xX2+qcQ9p0IxKAP6ExWoUo=\ngithub.com/go-kit/kit v0.9.0 h1:wDJmvq38kDhkVxi50ni9ykkdUr1PKgqKOoi01fa0Mdk=\ngithub.com/go-kit/log v0.1.0 h1:DGJh0Sm43HbOeYDNnVZFl8BvcYVvjD5bqYJvp0REbwQ=\ngithub.com/go-logfmt/logfmt v0.5.0 h1:TrB8swr/68K7m9CcGut2g3UOihhbcbiMAYiuTXdEih4=\ngithub.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=\ngithub.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=\ngithub.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=\ngithub.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=\ngithub.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=\ngithub.com/go-playground/validator/v10 v10.2.0 h1:KgJ0snyC2R9VXYN2rneOtQcw5aHQB1Vv0sFl1UcHBOY=\ngithub.com/go-redis/redis v6.15.8+incompatible h1:BKZuG6mCnRj5AOaWJXoCgf6rqTYnYJLe4en2hxT7r9o=\ngithub.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=\ngithub.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=\ngithub.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0 h1:p104kn46Q8WdvHunIJ9dAyjPVtrBPhSr3KT2yUst43I=\ngithub.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee h1:s+21KNqlpePfkah2I+gwHF8xmJWRjooY+5248k6m4A0=\ngithub.com/gobwas/pool v0.2.0 h1:QEmUOlnSjWtnpRGHF3SauEiOsy82Cup83Vf2LcMlnc8=\ngithub.com/gobwas/ws v1.0.2 h1:CoAavW/wd/kulfZmSIBt6p24n4j7tHgNVCjsfHVNUbo=\ngithub.com/godbus/dbus/v5 v5.0.4 h1:9349emZab16e7zQvpmsbtjc18ykshndd8y2PG3sgJbA=\ngithub.com/gogo/googleapis v1.3.2 h1:kX1es4djPJrsDhY7aZKJy7aZasdcB5oSOEphMjSB53c=\ngithub.com/gogo/status v1.1.0 h1:+eIkrewn5q6b30y+g/BJINVVdi2xH7je5MPJ3ZPK3JA=\ngithub.com/golang-sql/civil v0.0.0-20190719163853-cb61b32ac6fe h1:lXe2qZdvpiX5WZkZR4hgp4KJVfY3nMkvmwbVkpv1rVY=\ngithub.com/golang/glog v1.1.0 h1:/d3pCKDPWNnvIWe0vVUpNP32qc8U3PDVxySP/y360qE=\ngithub.com/golang/glog v1.1.0/go.mod h1:pfYeQZ3JWZoXTV5sFc986z3HTpwQs9At6P4ImfuP3NQ=\ngithub.com/golang/glog v1.2.2 h1:1+mZ9upx1Dh6FmUTFR1naJ77miKiXgALjWOZ3NVFPmY=\ngithub.com/golang/glog v1.2.2/go.mod h1:6AhwSGph0fcJtXVM/PEHPqZlFeoLxhs7/t5UDAwmO+w=\ngithub.com/golang/snappy v0.0.3 h1:fHPg5GQYlCeLIPB9BZqMVR5nR9A+IM5zcgeTdjMYmLA=\ngithub.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=\ngithub.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=\ngithub.com/golangci/plugin-module-register v0.1.2 h1:e5WM6PO6NIAEcij3B053CohVp3HIYbzSuP53UAYgOpg=\ngithub.com/golangci/plugin-module-register v0.1.2/go.mod h1:1+QGTsKBvAIvPvoY/os+G5eoqxWn70HYDm2uvUyGuVw=\ngithub.com/google/btree v1.0.0 h1:0udJVsspx3VBr5FwtLhQQtuAsVc79tTq0ocGIPAU6qo=\ngithub.com/google/certificate-transparency-go v1.1.1 h1:6JHXZhXEvilMcTjR4MGZn5KV0IRkcFl4CJx5iHVhjFE=\ngithub.com/google/go-pkcs11 v0.3.0 h1:PVRnTgtArZ3QQqTGtbtjtnIkzl2iY2kt24yqbrf7td8=\ngithub.com/google/go-pkcs11 v0.3.0/go.mod h1:6eQoGcuNJpa7jnd5pMGdkSaQpNDYvPlXWMcjXXThLlY=\ngithub.com/google/gofuzz v1.0.0 h1:A8PeW59pxE9IoFRqBp37U+mSNaQoZ46F1f0f863XSXw=\ngithub.com/google/pprof v0.0.0-20210720184732-4bb14d4b1be1 h1:K6RDEckDVWvDI9JAJYCmNdQXq6neHJOYx3V6jnqNEec=\ngithub.com/google/renameio v0.1.0 h1:GOZbcHa3HfsPKPlmyPyN2KEohoMXOhdMbHrvbpl2QaA=\ngithub.com/google/rpmpack v0.0.0-20210410105602-e20c988a6f5a h1:XC048Fc/OB2rUl/BxruopEl2u/EP6cJNFveVxI1cvdk=\ngithub.com/google/subcommands v1.0.1 h1:/eqq+otEXm5vhfBrbREPCSVQbvofip6kIz+mX5TUH7k=\ngithub.com/google/trillian v1.3.11 h1:pPzJPkK06mvXId1LHEAJxIegGgHzzp/FUnycPYfoCMI=\ngithub.com/googleapis/google-cloud-go-testing v0.0.0-20200911160855-bcd43fbb19e8 h1:tlyzajkF3030q6M8SvmJSemC9DTHL/xaMa18b65+JM4=\ngithub.com/gookit/color v1.5.0 h1:1Opow3+BWDwqor78DcJkJCIwnkviFi+rrOANki9BUFw=\ngithub.com/gorhill/cronexpr v0.0.0-20180427100037-88b0669f7d75 h1:f0n1xnMSmBLzVfsMMvriDyA75NB/oBgILX2GcHXIQzY=\ngithub.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=\ngithub.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=\ngithub.com/gregjones/httpcache v0.0.0-20190611155906-901d90724c79 h1:+ngKgrYPPJrOjhax5N+uePQ0Fh1Z7PheYoUI/0nzkPA=\ngithub.com/grpc-ecosystem/go-grpc-middleware v1.2.2 h1:FlFbCRLd5Jr4iYXZufAvgWN6Ao0JrI5chLINnUXDDr0=\ngithub.com/grpc-ecosystem/go-grpc-prometheus v1.2.0 h1:Ovs26xHkKqVztRpIrF/92BcuyuQ/YW4NSIpoGtfXNho=\ngithub.com/grpc-ecosystem/grpc-gateway v1.16.0 h1:gmcG1KaJ57LophUzW0Hy8NmPhnMZb4M0+kPpLofRdBo=\ngithub.com/grpc-ecosystem/grpc-gateway/v2 v2.11.3 h1:lLT7ZLSzGLI08vc9cpd+tYmNWjdKDqyr/2L+f6U12Fk=\ngithub.com/grpc-ecosystem/grpc-gateway/v2 v2.11.3/go.mod h1:o//XUCC/F+yRGJoPO/VU0GSB0f8Nhgmxx0VIRUvaC0w=\ngithub.com/hashicorp/consul/api v1.12.0 h1:k3y1FYv6nuKyNTqj6w9gXOx5r5CfLj/k/euUeBXj1OY=\ngithub.com/hashicorp/consul/api v1.18.0 h1:R7PPNzTCeN6VuQNDwwhZWJvzCtGSrNpJqfb22h3yH9g=\ngithub.com/hashicorp/consul/api v1.18.0/go.mod h1:owRRGJ9M5xReDC5nfT8FTJrNAPbT4NM6p/k+d03q2v4=\ngithub.com/hashicorp/consul/sdk v0.8.0 h1:OJtKBtEjboEZvG6AOUdh4Z1Zbyu0WcxQ0qatRrZHTVU=\ngithub.com/hashicorp/go-immutable-radix v1.3.1 h1:DKHmCUm2hRBK510BaiZlwvpD40f8bJFeZnpfm2KLowc=\ngithub.com/hashicorp/go-msgpack v0.5.3 h1:zKjpN5BK/P5lMYrLmBHdBULWbJ0XpYR+7NGzqkZzoD4=\ngithub.com/hashicorp/go-rootcerts v1.0.2 h1:jzhAVGtqPKbwpyCPELlgNWhE1znq+qwJtW5Oi2viEzc=\ngithub.com/hashicorp/go-sockaddr v1.0.0 h1:GeH6tui99pF4NJgfnhp+L6+FfobzVW3Ah46sLo0ICXs=\ngithub.com/hashicorp/go-syslog v1.0.0 h1:KaodqZuhUoZereWVIYmpUgZysurB1kBLX2j0MwMrUAE=\ngithub.com/hashicorp/go-uuid v1.0.1 h1:fv1ep09latC32wFoVwnqcnKJGnMSdBanPczbHAYm1BE=\ngithub.com/hashicorp/go.net v0.0.1 h1:sNCoNyDEvN1xa+X0baata4RdcpKwcMS6DH+xwfqPgjw=\ngithub.com/hashicorp/golang-lru v0.5.4 h1:YDjusn29QI/Das2iO9M0BHnIbxPeyuCHsjMW+lJfyTc=\ngithub.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=\ngithub.com/hashicorp/logutils v1.0.0 h1:dLEQVugN8vlakKOUE3ihGLTZJRB4j+M2cdTm/ORI65Y=\ngithub.com/hashicorp/mdns v1.0.4 h1:sY0CMhFmjIPDMlTB+HfymFHCaYLhgifZ0QhjaYKD/UQ=\ngithub.com/hashicorp/memberlist v0.3.0 h1:8+567mCcFDnS5ADl7lrpxPMWiFCElyUEeW0gtj34fMA=\ngithub.com/hashicorp/serf v0.9.7 h1:hkdgbqizGQHuU5IPqYM1JdSMV8nKfpuOnZYXssk9muY=\ngithub.com/hashicorp/serf v0.10.1 h1:Z1H2J60yRKvfDYAOZLd2MU0ND4AH/WDz7xYHDWQsIPY=\ngithub.com/hashicorp/serf v0.10.1/go.mod h1:yL2t6BqATOLGc5HF7qbFkTfXoPIY0WZdWHfEvMqbG+4=\ngithub.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=\ngithub.com/iancoleman/orderedmap v0.0.0-20190318233801-ac98e3ecb4b0 h1:i462o439ZjprVSFSZLZxcsoAe592sZB1rci2Z8j4wdk=\ngithub.com/iancoleman/orderedmap v0.0.0-20190318233801-ac98e3ecb4b0/go.mod h1:N0Wam8K1arqPXNWjMo21EXnBPOPp36vB07FNRdD2geA=\ngithub.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639 h1:mV02weKRL81bEnm8A0HT1/CAelMQDBuQIfLw8n+d6xI=\ngithub.com/invopop/jsonschema v0.7.0 h1:2vgQcBz1n256N+FpX3Jq7Y17AjYt46Ig3zIWyy770So=\ngithub.com/invopop/jsonschema v0.7.0/go.mod h1:O9uiLokuu0+MGFlyiaqtWxwqJm41/+8Nj0lD7A36YH0=\ngithub.com/jhump/protoreflect v1.6.1 h1:4/2yi5LyDPP7nN+Hiird1SAJ6YoxUm13/oxHGRnbPd8=\ngithub.com/jmoiron/sqlx v1.2.0 h1:41Ip0zITnmWNR/vHV+S4m+VoUivnWY5E4OJfLZjCJMA=\ngithub.com/joho/godotenv v1.3.0 h1:Zjp+RcGpHhGlrMbJzXTrZZPrWj+1vfm90La1wgB6Bhc=\ngithub.com/jonboulle/clockwork v0.2.0 h1:J2SLSdy7HgElq8ekSl2Mxh6vrRNFxqbXGenYH2I02Vs=\ngithub.com/jpillora/backoff v1.0.0 h1:uvFg412JmmHBHw7iwprIxkPMI+sGQ4kzOWsMeHnm2EA=\ngithub.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=\ngithub.com/jstemmer/go-junit-report v0.9.1 h1:6QPYqodiu3GuPL+7mfx+NwDdp2eTkp9IfEUpgAwUN0o=\ngithub.com/juju/ratelimit v1.0.1 h1:+7AIFJVQ0EQgq/K9+0Krm7m530Du7tIz0METWzN0RgY=\ngithub.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=\ngithub.com/k0kubun/colorstring v0.0.0-20150214042306-9440f1994b88 h1:uC1QfSlInpQF+M0ao65imhwqKnz3Q2z/d8PWZRMQvDM=\ngithub.com/konsorten/go-windows-terminal-sequences v1.0.3 h1:CE8S1cTafDpPvMhIxNJKvHsGVBgn1xWYf1NbHQhywc8=\ngithub.com/kr/fs v0.1.0 h1:Jskdu9ieNAYnjxsi0LbQp1ulIKZV1LAFgK1tWhpZgl8=\ngithub.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515 h1:T+h1c/A9Gawja4Y9mFVWj2vyii2bbUNDw3kt9VxK2EY=\ngithub.com/kr/pty v1.1.1 h1:VkoXIwSboBpnk99O/KFauAEILuNHv5DVFKZMBN/gUgw=\ngithub.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc=\ngithub.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=\ngithub.com/letsencrypt/pkcs11key/v4 v4.0.0 h1:qLc/OznH7xMr5ARJgkZCCWk+EomQkiNTOoOF5LAgagc=\ngithub.com/lib/pq v1.10.4 h1:SO9z7FRPzA03QhHKJrH5BXA6HU1rS4V2nIVrrNC1iYk=\ngithub.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 h1:6E+4a0GO5zZEnZ81pIr0yLvtUWk2if982qA3F3QD6H4=\ngithub.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0/go.mod h1:zJYVVT2jmtg6P3p1VtQj7WsuWi/y4VnjVBn7F8KPB3I=\ngithub.com/lyft/protoc-gen-star/v2 v2.0.4-0.20230330145011-496ad1ac90a4 h1:sIXJOMrYnQZJu7OB7ANSF4MYri2fTEGIsRLz6LwI4xE=\ngithub.com/lyft/protoc-gen-star/v2 v2.0.4-0.20230330145011-496ad1ac90a4/go.mod h1:amey7yeodaJhXSbf/TlLvWiqQfLOSpEk//mLlc+axEk=\ngithub.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=\ngithub.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=\ngithub.com/mattn/go-shellwords v1.0.10 h1:Y7Xqm8piKOO3v10Thp7Z36h4FYFjt5xB//6XvOrs2Gw=\ngithub.com/mattn/go-sqlite3 v1.9.0 h1:pDRiWfl+++eC2FEFRy6jXmQlvp4Yh3z1MJKg4UeYM/4=\ngithub.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4=\ngithub.com/miekg/dns v1.1.41 h1:WMszZWJG0XmzbK9FEmzH2TVcqYzFesusSIB41b8KHxY=\ngithub.com/miekg/pkcs11 v1.0.3 h1:iMwmD7I5225wv84WxIG/bmxz9AXjWvTWIbM/TYHvWtw=\ngithub.com/mitchellh/cli v1.1.0 h1:tEElEatulEHDeedTxwckzyYMA5c86fbmNIUL1hBIiTg=\ngithub.com/mitchellh/go-ps v1.0.0 h1:i6ampVEEF4wQFF+bkYfwYgY+F/uYJDktmvLPf7qIgjc=\ngithub.com/mitchellh/go-ps v1.0.0/go.mod h1:J4lOc8z8yJs6vUwklHw2XEIiT4z4C40KtWVN3nvg8Pg=\ngithub.com/mitchellh/go-testing-interface v1.0.0 h1:fzU/JVNcaqHQEcVFAKeR41fkiLdIPrefOvVG1VZ96U0=\ngithub.com/mitchellh/gox v0.4.0 h1:lfGJxY7ToLJQjHHwi0EX6uYBdK78egf954SQl13PQJc=\ngithub.com/mitchellh/iochan v1.0.0 h1:C+X3KsSTLFVBr/tK1eYN/vs4rJcvsiLU338UhYPJWeY=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=\ngithub.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=\ngithub.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw=\ngithub.com/mozilla/scribe v0.0.0-20180711195314-fb71baf557c1 h1:29NKShH4TWd3lxCDUhS4Xe16EWMA753dtIxYtwddklU=\ngithub.com/mozilla/tls-observatory v0.0.0-20210609171429-7bc42856d2e5 h1:0KqC6/sLy7fDpBdybhVkkv4Yz+PmB7c9Dz9z3dLW804=\ngithub.com/muesli/mango v0.1.0 h1:DZQK45d2gGbql1arsYA4vfg4d7I9Hfx5rX/GCmzsAvI=\ngithub.com/muesli/mango v0.1.0/go.mod h1:5XFpbC8jY5UUv89YQciiXNlbi+iJgt29VDC5xbzrLL4=\ngithub.com/muesli/mango-cobra v1.2.0 h1:DQvjzAM0PMZr85Iv9LIMaYISpTOliMEg+uMFtNbYvWg=\ngithub.com/muesli/mango-cobra v1.2.0/go.mod h1:vMJL54QytZAJhCT13LPVDfkvCUJ5/4jNUKF/8NC2UjA=\ngithub.com/muesli/mango-pflag v0.1.0 h1:UADqbYgpUyRoBja3g6LUL+3LErjpsOwaC9ywvBWe7Sg=\ngithub.com/muesli/mango-pflag v0.1.0/go.mod h1:YEQomTxaCUp8PrbhFh10UfbhbQrM/xJ4i2PB8VTLLW0=\ngithub.com/muesli/roff v0.1.0 h1:YD0lalCotmYuF5HhZliKWlIx7IEhiXeSfq7hNjFqGF8=\ngithub.com/muesli/roff v0.1.0/go.mod h1:pjAHQM9hdUUwm/krAfrLGgJkXJ+YuhtsfZ42kieB2Ig=\ngithub.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f h1:KUppIJq7/+SVif2QVs3tOP0zanoHgBEVAwHxUSIzRqU=\ngithub.com/mwitkow/go-proto-validators v0.2.0 h1:F6LFfmgVnfULfaRsQWBbe7F7ocuHCr9+7m+GAeDzNbQ=\ngithub.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs=\ngithub.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=\ngithub.com/oklog/ulid v1.3.1 h1:EGfNDEx6MqHz8B3uNV6QAib1UR2Lm97sHi3ocA6ESJ4=\ngithub.com/op/go-logging v0.0.0-20160315200505-970db520ece7 h1:lDH9UUVJtmYCjyT0CI4q8xvlXPxeZ0gYCVvWbmPlp88=\ngithub.com/otiai10/curr v1.0.0 h1:TJIWdbX0B+kpNagQrjgq8bCMrbhiuX73M2XwgtDMoOI=\ngithub.com/otiai10/mint v1.3.1 h1:BCmzIS3n71sGfHB5NMNDB3lHYPz8fWSkCAErHed//qc=\ngithub.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY=\ngithub.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=\ngithub.com/peterbourgon/diskv v2.0.1+incompatible h1:UBdAOUP5p4RWqPBg048CAvpKN+vxiaj6gdUUzhl4XmI=\ngithub.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e h1:aoZm08cpOy4WuID//EZDgcC4zIxODThtZNPirFr42+A=\ngithub.com/pkg/sftp v1.13.1 h1:I2qBYMChEhIjOgazfJmV3/mZM256btk6wkCDRmW7JYs=\ngithub.com/pkg/sftp v1.13.7 h1:uv+I3nNJvlKZIQGSr8JVQLNHFU9YhhNpvC14Y6KgmSM=\ngithub.com/pkg/sftp v1.13.7/go.mod h1:KMKI0t3T6hfA+lTR/ssZdunHo+uwq7ghoN09/FSu3DY=\ngithub.com/posener/complete v1.2.3 h1:NP0eAhjcjImqslEwo/1hq7gpajME0fTLTezBKDqfXqo=\ngithub.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c h1:ncq/mPwQF4JjgDlrVEn3C11VoGHZN7m8qihwgMEtzYw=\ngithub.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=\ngithub.com/prashantv/protectmem v0.0.0-20171002184600-e20412882b3a h1:AA9vgIBDjMHPC2McaGPojgV2dcI78ZC0TLNhYCXEKH8=\ngithub.com/prometheus/tsdb v0.7.1 h1:YZcsG11NqnK4czYLrWd9mpEuAJIHVQLwdrleYfszMAA=\ngithub.com/pseudomuto/protoc-gen-doc v1.3.2 h1:61vWZuxYa8D7Rn4h+2dgoTNqnluBmJya2MgbqO32z6g=\ngithub.com/pseudomuto/protokit v0.2.0 h1:hlnBDcy3YEDXH7kc9gV+NLaN0cDzhDvD1s7Y6FZ8RpM=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.19 h1:5+KTKb2YREUYiqZFEIuifFyBxlcCUPWgNZkWy71XS0Q=\ngithub.com/quasilyte/go-ruleguard/dsl v0.3.19/go.mod h1:KeCP03KrjuSO0H1kTuZQCWlQPulDV6YMIXmpQss17rU=\ngithub.com/quasilyte/go-ruleguard/rules v0.0.0-20211022131956-028d6511ab71 h1:CNooiryw5aisadVfzneSZPswRWvnVW8hF1bS/vo8ReI=\ngithub.com/remyoudompheng/go-dbus v0.0.0-20121104212943-b7232d34b1d5 h1:CvqZS4QYHBRvx7AeFdimd16HCbLlYsvQMcKDACpJW/c=\ngithub.com/remyoudompheng/go-dbus v0.0.0-20121104212943-b7232d34b1d5/go.mod h1:+u151txRmLpwxBmpYn9z3d1sdJdjRPQpsXuYeY9jNls=\ngithub.com/remyoudompheng/go-liblzma v0.0.0-20190506200333-81bf2d431b96 h1:J8J/cgLDRuqXJnwIrRDBvtl+LLsdg7De74znW/BRRq4=\ngithub.com/remyoudompheng/go-liblzma v0.0.0-20190506200333-81bf2d431b96/go.mod h1:90HvCY7+oHHUKkbeMCiHt1WuFR2/hPJ9QrljDG+v6ls=\ngithub.com/remyoudompheng/go-misc v0.0.0-20190427085024-2d6ac652a50e h1:eTWZyPUnHcuGRDiryS/l2I7FfKjbU3IBx3IjqHPxuKU=\ngithub.com/remyoudompheng/go-misc v0.0.0-20190427085024-2d6ac652a50e/go.mod h1:80FQABjoFzZ2M5uEa6FUaJYEmqU2UOKojlFVak1UAwI=\ngithub.com/rogpeppe/fastuuid v1.2.0 h1:Ppwyp6VYCF1nvBTXL3trRso7mXMlRrw9ooo375wvi2s=\ngithub.com/rs/cors v1.7.0 h1:+88SsELBHx5r+hZ8TCkggzSstaWNbDvThkVK8H6f9ik=\ngithub.com/rs/xid v1.4.0 h1:qd7wPTDkN6KQx2VmMBLrpHkiyQwgFXRnkOLacUiaSNY=\ngithub.com/rs/xid v1.5.0 h1:mKX4bl4iPYJtEIxp6CYiUuLQ/8DYMoz0PUdtGgMFRVc=\ngithub.com/russross/blackfriday v1.5.2 h1:HyvC0ARfnZBqnXwABFeSZHpKvJHJJfPz81GNueLj0oo=\ngithub.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f h1:UFr9zpz4xgTnIE5yIMtWAMngCdZ9p/+q6lTbgelo80M=\ngithub.com/sagikazarmark/crypt v0.5.0 h1:K6qABjdpr5rjHCw6q4rSdeM+8kNmdIHvEPDvEMkoai4=\ngithub.com/sagikazarmark/crypt v0.9.0 h1:fipzMFW34hFUEc4D7fsLQFtE7yElkpgyS2zruedRdZk=\ngithub.com/sagikazarmark/crypt v0.9.0/go.mod h1:RnH7sEhxfdnPm1z+XMgSLjWTEIjyK4z2dw6+4vHTMuo=\ngithub.com/samuel/go-thrift v0.0.0-20191111193933-5165175b40af h1:EiWVfh8mr40yFZEui2oF0d45KgH48PkB2H0Z0GANvSI=\ngithub.com/sassoftware/go-rpmutils v0.0.0-20190420191620-a8f1baeba37b h1:+gCnWOZV8Z/8jehJ2CdqB47Z3S+SREmQcuXkRFLNsiI=\ngithub.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529 h1:nn5Wsu0esKSJiIVhscUtVbo7ada43DJhG55ua/hjS5I=\ngithub.com/shirou/gopsutil/v3 v3.22.4 h1:srAQaiX6jX/cYL6q29aE0m8lOskT9CurZ9N61YR3yoI=\ngithub.com/shirou/gopsutil/v3 v3.22.4/go.mod h1:D01hZJ4pVHPpCTZ3m3T2+wDF2YAGfd+H4ifUguaQzHM=\ngithub.com/shurcooL/go v0.0.0-20180423040247-9e1955d9fb6e h1:MZM7FHLqUHYI0Y/mQAt3d2aYa0SiNms/hFqC9qJYolM=\ngithub.com/shurcooL/go-goon v0.0.0-20170922171312-37c2f522c041 h1:llrF3Fs4018ePo4+G/HV/uQUqEI1HMDjCeOf2V6puPc=\ngithub.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=\ngithub.com/smartystreets/go-aws-auth v0.0.0-20180515143844-0c1422d1fdb9 h1:hp2CYQUINdZMHdvTdXtPOY2ainKl4IoMcpAXEf2xj3Q=\ngithub.com/smartystreets/gunit v1.0.0 h1:RyPDUFcJbvtXlhJPk7v+wnxZRY2EUokhEYl2EJOPToI=\ngithub.com/soheilhy/cmux v0.1.4 h1:0HKaf1o97UwFjHH9o5XsHUOF+tqmdA7KEzXLpiyaw0E=\ngithub.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72 h1:qLC7fQah7D6K1B0ujays3HV9gkFtllcxhzImRR7ArPQ=\ngithub.com/spf13/jwalterweatherman v1.1.0 h1:ue6voC5bR5F8YxI5S67j9i582FU4Qvo2bmqnqMYADFk=\ngithub.com/streadway/quantile v0.0.0-20150917103942-b0c588724d25 h1:7z3LSn867ex6VSaahyKadf4WtSsJIgne6A1WLOAGM8A=\ngithub.com/tj/go-buffer v1.1.0 h1:Lo2OsPHlIxXF24zApe15AbK3bJLAOvkkxEA6Ux4c47M=\ngithub.com/tj/go-elastic v0.0.0-20171221160941-36157cbbebc2 h1:eGaGNxrtoZf/mBURsnNQKDR7u50Klgcf2eFDQEnc8Bc=\ngithub.com/tj/go-kinesis v0.0.0-20171128231115-08b17f58cb1b h1:m74UWYy+HBs+jMFR9mdZU6shPewugMyH5+GV6LNgW8w=\ngithub.com/tj/go-spin v1.1.0 h1:lhdWZsvImxvZ3q1C5OIB7d72DuOwP4O2NdBg9PyzNds=\ngithub.com/tklauser/go-sysconf v0.3.10 h1:IJ1AZGZRWbY8T5Vfk04D9WOA5WSejdflXxP03OUqALw=\ngithub.com/tklauser/go-sysconf v0.3.10/go.mod h1:C8XykCvCb+Gn0oNCWPIlcb0RuglQTYaQ2hGm7jmxEFk=\ngithub.com/tklauser/numcpus v0.4.0 h1:E53Dm1HjH1/R2/aoCtXtPgzmElmn51aOkhCFSuZq//o=\ngithub.com/tklauser/numcpus v0.4.0/go.mod h1:1+UI3pD8NW14VMwdgJNJ1ESk2UnwhAnz5hMwiKKqXCQ=\ngithub.com/tmc/grpc-websocket-proxy v0.0.0-20200427203606-3cfed13b9966 h1:j6JEOq5QWFker+d7mFQYOhjTZonQ7YkLTHm56dbn+yM=\ngithub.com/tomasen/realip v0.0.0-20180522021738-f0c99a92ddce h1:fb190+cK2Xz/dvi9Hv8eCYJYvIGUTN2/KLq1pT6CjEc=\ngithub.com/tv42/httpunix v0.0.0-20150427012821-b75d8614f926 h1:G3dpKMzFDjgEh2q1Z7zUUtKa8ViPtH+ocF0bE0g00O8=\ngithub.com/twitchtv/twirp v5.8.0+incompatible h1:DTfGS9u/jHbo34cBB+qhzVHRaAq+tRois71j8pvjQ5M=\ngithub.com/uber-common/bark v1.2.1 h1:cREJ9b7CpTjwZr0/5wV82fXlitoCIEHHnt9WkQ4lIk0=\ngithub.com/uber/jaeger-client-go v2.22.1+incompatible h1:NHcubEkVbahf9t3p75TOCR83gdUHXjRJvjoBh1yACsM=\ngithub.com/uber/jaeger-lib v2.2.0+incompatible h1:MxZXOiR2JuoANZ3J6DE/U0kSFv/eJ/GfSYVCjK7dyaw=\ngithub.com/uber/ringpop-go v0.8.5 h1:aBa/SHmmFRcAXA63k7uBheoTL8tCmH7L+OgktB1AF/o=\ngithub.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo=\ngithub.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=\ngithub.com/urfave/cli v1.22.1 h1:+mkCCcOFKPnCmVYVcURKps1Xe+3zP90gSYGNfRkjoIY=\ngithub.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=\ngithub.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=\ngithub.com/valyala/quicktemplate v1.7.0 h1:LUPTJmlVcb46OOUY3IeD9DojFpAVbsG+5WFTcjMJzCM=\ngithub.com/valyala/quicktemplate v1.7.0/go.mod h1:sqKJnoaOF88V07vkO+9FL8fb9uZg/VPSJnLYn+LmLk8=\ngithub.com/viki-org/dnscache v0.0.0-20130720023526-c70c1f23c5d8 h1:EVObHAr8DqpoJCVv6KYTle8FEImKhtkfcZetNqxDoJQ=\ngithub.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 h1:eY9dn8+vbi4tKz5Qo6v2eYzo7kUS51QINcR5jNpbZS8=\ngithub.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778 h1:QldyIu/L63oPpyvQmHgvgickp1Yw510KJOqX7H24mg8=\ngithub.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77 h1:ESFSdwYZvkeru3RtdrYueztKhOBCSAAzS4Gf+k0tEow=\ngithub.com/yudai/gojsondiff v1.0.0 h1:27cbfqXLVEJ1o8I6v3y9lg8Ydm53EKqHXAOMxEGlCOA=\ngithub.com/yudai/golcs v0.0.0-20170316035057-ecda9a501e82 h1:BHyfKlQyqbsFN5p3IfnEUduWvb9is428/nNb5L3U01M=\ngithub.com/yudai/pp v2.0.1+incompatible h1:Q4//iY4pNF6yPLZIigmvcl7k/bPgrcTPIFIcmawg5bI=\ngithub.com/yuin/goldmark v1.4.13 h1:fVcFKWvrslecOb/tg+Cc05dkeYx540o0FuFt3nUVDoE=\ngithub.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=\ngithub.com/yusufpapurcu/wmi v1.2.2 h1:KBNDSne4vP5mbSWnJbO+51IMOXJB67QiYCSBrubbPRg=\ngithub.com/yusufpapurcu/wmi v1.2.2/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=\ngo.etcd.io/bbolt v1.3.4 h1:hi1bXHMVrlQh6WwxAy+qZCV/SYIlqo+Ushwdpa4tAKg=\ngo.etcd.io/etcd v0.0.0-20200513171258-e048e166ab9c h1:/RwRVN9EdXAVtdHxP7Ndn/tfmM9/goiwU0QTnLBgS4w=\ngo.etcd.io/etcd/api/v3 v3.5.2 h1:tXok5yLlKyuQ/SXSjtqHc4uzNaMqZi2XsoSPr/LlJXI=\ngo.etcd.io/etcd/api/v3 v3.5.6 h1:Cy2qx3npLcYqTKqGJzMypnMv2tiRyifZJ17BlWIWA7A=\ngo.etcd.io/etcd/api/v3 v3.5.6/go.mod h1:KFtNaxGDw4Yx/BA4iPPwevUTAuqcsPxzyX8PHydchN8=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.2 h1:4hzqQ6hIb3blLyQ8usCU4h3NghkqcsohEQ3o3VetYxE=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.6 h1:TXQWYceBKqLp4sa87rcPs11SXxUA/mHwH975v+BDvLU=\ngo.etcd.io/etcd/client/pkg/v3 v3.5.6/go.mod h1:ggrwbk069qxpKPq8/FKkQ3Xq9y39kbFR4LnKszpRXeQ=\ngo.etcd.io/etcd/client/v2 v2.305.2 h1:ymrVwTkefuqA/rPkSW7/B4ApijbPVefRumkY+stNfS0=\ngo.etcd.io/etcd/client/v2 v2.305.6 h1:fIDR0p4KMjw01MJMfUIDWdQbjo06PD6CeYM5z4EHLi0=\ngo.etcd.io/etcd/client/v2 v2.305.6/go.mod h1:BHha8XJGe8vCIBfWBpbBLVZ4QjOIlfoouvOwydu63E0=\ngo.etcd.io/etcd/client/v3 v3.5.6 h1:coLs69PWCXE9G4FKquzNaSHrRyMCAXwF+IX1tAPVO8E=\ngo.etcd.io/etcd/client/v3 v3.5.6/go.mod h1:f6GRinRMCsFVv9Ht42EyY7nfsVGwrNO0WEoS2pRKzQk=\ngo.mozilla.org/mozlog v0.0.0-20170222151521-4bb13139d403 h1:rKyWXYDfrVOpMFBion4Pmx5sJbQreQNXycHvm4KwJSg=\ngo.opentelemetry.io/otel v1.0.1 h1:4XKyXmfqJLOQ7feyV5DB6gsBFZ0ltB8vLtp6pj4JIcc=\ngo.opentelemetry.io/otel v1.0.1/go.mod h1:OPEOD4jIT2SlZPMmwT6FqZz2C0ZNdQqiWcoK6M0SNFU=\ngo.opentelemetry.io/otel/trace v1.0.1 h1:StTeIH6Q3G4r0Fiw34LTokUFESZgIDUr0qIJ7mKmAfw=\ngo.opentelemetry.io/otel/trace v1.0.1/go.mod h1:5g4i4fKLaX2BQpSBsxw8YYcgKpMMSW3x7ZTuYBr3sUk=\ngo.opentelemetry.io/proto/otlp v0.7.0 h1:rwOQPCuKAKmwGKq2aVNnYIibI6wnV7EvzgfTCzcdGg8=\ngo.opentelemetry.io/proto/otlp v1.0.0 h1:T0TX0tmXU8a3CbNXzEKGeU5mIVOdf0oykP+u2lIVU/I=\ngo.opentelemetry.io/proto/otlp v1.0.0/go.mod h1:Sy6pihPLfYHkr3NkUbEhGHFhINUSI/v80hjKIs5JXpM=\ngo.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee h1:0mgffUl7nfd+FpvXMVz4IDEaUSmT1ysygQC7qYo7sG4=\ngolang.org/x/crypto v0.17.0 h1:r8bRNjWL3GshPW3gkd+RpvzWrZAwPS49OmTGZ/uhM4k=\ngolang.org/x/crypto v0.17.0/go.mod h1:gCAAfMLgwOJRpTjQ2zCCt2OcSfYMTeZVSRtQlPC7Nq4=\ngolang.org/x/exp v0.0.0-20200331195152-e8c3332aa8e5 h1:FR+oGxGfbQu1d+jglI3rCkjAjUnhRSZcUxr+DqlDLNo=\ngolang.org/x/image v0.0.0-20190802002840-cff245a6509b h1:+qEpEAPhDZ1o0x3tHzZTQDArnOixOzGD9HUJfcg0mb4=\ngolang.org/x/mobile v0.0.0-20200801112145-973feb4309de h1:OVJ6QQUBAesB8CZijKDSsXX7xYVtUhrkY0gwMfbi4p4=\ngolang.org/x/mod v0.10.0 h1:lFO9qtOdlre5W1jxS3r/4szv2/6iXxScdzjoBMXNhYk=\ngolang.org/x/mod v0.10.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=\ngolang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=\ngolang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=\ngolang.org/x/sync v0.2.0 h1:PUR+T4wwASmuSTYdKjYHI5TD22Wy5ogLU5qZCOLxBrI=\ngolang.org/x/sync v0.2.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=\ngolang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=\ngolang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2 h1:IRJeR9r1pYWsHKTRe/IInb7lYvbBVIqOgsX/u0mbOWY=\ngolang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2/go.mod h1:TeRTkGYfJXctD9OcfyVLyj2J3IxLnKwHJR8f4D8a3YE=\ngolang.org/x/telemetry v0.0.0-20240521205824-bda55230c457 h1:zf5N6UOrA487eEFacMePxjXAJctxKmyjKUsjA11Uzuk=\ngolang.org/x/telemetry v0.0.0-20240521205824-bda55230c457/go.mod h1:pRgIJT+bRLFKnoM1ldnzKoxTIn14Yxz928LQRYYgIN0=\ngolang.org/x/term v0.15.0 h1:y/Oo/a/q3IXu26lQgl04j/gjuBDOBlx7X6Om1j2CPW4=\ngolang.org/x/term v0.15.0/go.mod h1:BDl952bC7+uMoWR75FIrCDx79TPU9oHkTZ9yRbYOrX0=\ngolang.org/x/tools v0.9.1 h1:8WMNJAz3zrtPmnYC7ISf5dEn3MT0gY7jBJfw27yrrLo=\ngolang.org/x/tools v0.9.1/go.mod h1:owI94Op576fPu3cIGQeHs3joujW/2Oc6MtlxbF5dfNc=\ngoogle.golang.org/appengine v1.6.8 h1:IhEN5q69dyKagZPYMSdIjS2HqprW324FRQZJcGqPAsM=\ngoogle.golang.org/appengine v1.6.8/go.mod h1:1jJ3jBArFh5pcgW8gCtRJnepW8FzD1V44FJffLiz/Ds=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20241223144023-3abc09e42ca8 h1:qlXhWiX84AGgaN7LuORWBEQCCTqj3szNbh2am45O3W8=\ngoogle.golang.org/genproto/googleapis/bytestream v0.0.0-20241223144023-3abc09e42ca8/go.mod h1:bLYPejkLzwgJuAHlIk1gdPOlx9CUYXLZi2rZxL/ursM=\ngoogle.golang.org/grpc/cmd/protoc-gen-go-grpc v1.1.0 h1:M1YKkFIboKNieVO5DLUEVzQfGwJD30Nv2jfUgzb5UcE=\ngopkg.in/alecthomas/kingpin.v2 v2.2.6 h1:jMFz6MfLP0/4fUyZle81rXUoxOBFi19VUFKVDOQfozc=\ngopkg.in/cheggaaa/pb.v1 v1.0.28 h1:n1tBJnnK2r7g9OW2btFH91V92STTUevLXYFb8gy9EMk=\ngopkg.in/errgo.v2 v2.1.0 h1:0vLT13EuvQ0hNvakwLuFZ/jYrLp5F3kcWHXdRggjCE8=\ngopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=\ngopkg.in/gcfg.v1 v1.2.3 h1:m8OOJ4ccYHnx2f4gQwpno8nAX5OGOh7RLaaz0pj3Ogs=\ngopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=\ngopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=\ngopkg.in/resty.v1 v1.12.0 h1:CuXP0Pjfw9rOuY6EP+UvtNvt5DSqHpIxILZKT/quCZI=\ngopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=\nnhooyr.io/websocket v1.8.7 h1:usjR2uOr/zjjkVMy0lW+PPohFok7PCow5sDjLgX4P4g=\nrsc.io/binaryregexp v0.2.0 h1:HfqmD5MEmC0zvwBuF187nq9mdnXjXsSivRiXN7SmRkE=\nrsc.io/quote/v3 v3.1.0 h1:9JKUTTIUgS6kzR9mK1YuGKv6Nl+DijDNIc0ghT58FaY=\nrsc.io/sampler v1.3.0 h1:7uVkIFmeBqHfdjD+gZwtXXI+RODJ2Wc4O7MPEh/QiW4=\nsigs.k8s.io/yaml v1.1.0 h1:4A07+ZFc2wgJwo8YNlQpr1rVlgUDlxXHhPJciaPY5gs=\n"
  },
  {
    "path": "internal/tools/tools.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:build tools\n// +build tools\n\npackage tools\n\nimport (\n\t// forces one import-order pattern\n\t_ \"github.com/daixiang0/gci\"\n\t// enumer for generating utility methods for const enums\n\t_ \"github.com/dmarkham/enumer\"\n\t// protobuf stuff\n\t_ \"github.com/gogo/protobuf/protoc-gen-gofast\"\n\t// gowrap for generating decorators for interface\n\t_ \"github.com/hexdigest/gowrap\"\n\t// replaces golint - configurable and much faster\n\t_ \"github.com/mgechev/revive\"\n\t// mockery for generating mocks\n\t_ \"github.com/vektra/mockery/v2\"\n\t// mockgen for generating mocks\n\t_ \"go.uber.org/mock/mockgen\"\n\t// nilaway for nil pointer analysis\n\t_ \"go.uber.org/nilaway\"\n\t// thriftrw code gen\n\t_ \"go.uber.org/thriftrw\"\n\t_ \"go.uber.org/yarpc/encoding/protobuf/protoc-gen-yarpc-go\"\n\t// yarpc plugin for thriftrw code gen\n\t_ \"go.uber.org/yarpc/encoding/thrift/thriftrw-plugin-yarpc\"\n\t// removes unused imports and formats\n\t_ \"golang.org/x/tools/cmd/goimports\"\n)\n"
  },
  {
    "path": "proto/buf.yaml",
    "content": "version: v1beta1\nbuild:\n  roots:\n    - public\n    - internal\nlint: # Uber style rules: https://docs.buf.build/migration-prototool/#uber1-uber2\n  use:\n    - DEFAULT\n  ignore_only:\n    FIELD_LOWER_SNAKE_CASE:\n      - uber/cadence/history/v1/service.proto\n      - uber/cadence/matching/v1/service.proto\n  enum_zero_value_suffix: _INVALID\n  service_suffix: API\n"
  },
  {
    "path": "proto/internal/uber/cadence/history/v1/service.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.history.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/history/v1;historyv1\";\n\nimport \"google/protobuf/duration.proto\";\nimport \"google/protobuf/timestamp.proto\";\nimport \"google/protobuf/wrappers.proto\";\nimport \"uber/cadence/api/v1/common.proto\";\nimport \"uber/cadence/api/v1/history.proto\";\nimport \"uber/cadence/api/v1/query.proto\";\nimport \"uber/cadence/api/v1/tasklist.proto\";\nimport \"uber/cadence/api/v1/workflow.proto\";\nimport \"uber/cadence/api/v1/service_workflow.proto\";\nimport \"uber/cadence/api/v1/service_worker.proto\";\nimport \"uber/cadence/admin/v1/cluster.proto\";\nimport \"uber/cadence/admin/v1/history.proto\";\nimport \"uber/cadence/admin/v1/queue.proto\";\nimport \"uber/cadence/admin/v1/replication.proto\";\nimport \"uber/cadence/shared/v1/any.proto\";\nimport \"uber/cadence/shared/v1/history.proto\";\nimport \"uber/cadence/shared/v1/workflow.proto\";\n\n\n// HistoryAPI provides API to start a new long running workflow instance, as well as query and update the history\n// of workflow instances already created.\nservice HistoryAPI {\n  // StartWorkflowExecution starts a new long running workflow instance.  It will create the instance with\n  // 'WorkflowExecutionStarted' event in history and also schedule the first DecisionTask for the worker to make the\n  // first decision for this instance.  It will return 'WorkflowExecutionAlreadyStartedError', if an instance already\n  // exists with same workflowId.\n  rpc StartWorkflowExecution(StartWorkflowExecutionRequest) returns (StartWorkflowExecutionResponse);\n\n  // SignalWorkflowExecution is used to send a signal event to running workflow execution.  This results in\n  // WorkflowExecutionSignaled event recorded in the history and a decision task being created for the execution.\n  rpc SignalWorkflowExecution(SignalWorkflowExecutionRequest) returns (SignalWorkflowExecutionResponse);\n\n  // SignalWithStartWorkflowExecution is used to ensure sending a signal event to a workflow execution.\n  // If workflow is running, this results in WorkflowExecutionSignaled event recorded in the history\n  // and a decision task being created for the execution.\n  // If workflow is not running or not found, it will first try start workflow with given WorkflowIDReusePolicy,\n  // and record WorkflowExecutionStarted and WorkflowExecutionSignaled event in case of success.\n  // It will return `WorkflowExecutionAlreadyStartedError` if start workflow failed with given policy.\n  rpc SignalWithStartWorkflowExecution(SignalWithStartWorkflowExecutionRequest) returns (SignalWithStartWorkflowExecutionResponse);\n\n  // ResetWorkflowExecution reset an existing workflow execution by a firstEventID of a existing event batch\n  // in the history and immediately terminating the current execution instance.\n  // After reset, the history will grow from nextFirstEventID.\n  rpc ResetWorkflowExecution(ResetWorkflowExecutionRequest) returns (ResetWorkflowExecutionResponse);\n\n  // TerminateWorkflowExecution terminates an existing workflow execution by recording WorkflowExecutionTerminated event\n  // in the history and immediately terminating the execution instance.\n  rpc TerminateWorkflowExecution(TerminateWorkflowExecutionRequest) returns (TerminateWorkflowExecutionResponse);\n\n  // DescribeWorkflowExecution returns information about the specified workflow execution.\n  rpc DescribeWorkflowExecution(DescribeWorkflowExecutionRequest) returns (DescribeWorkflowExecutionResponse);\n\n  // QueryWorkflow returns query result for a specified workflow execution.\n  rpc QueryWorkflow(QueryWorkflowRequest) returns (QueryWorkflowResponse);\n\n  // Reset the sticky tasklist related information in mutable state of a given workflow.\n  // Things cleared are:\n  // 1. StickyTaskList\n  // 2. StickyScheduleToStartTimeout\n  // 3. ClientLibraryVersion\n  // 4. ClientFeatureVersion\n  // 5. ClientImpl\n  rpc ResetStickyTaskList(ResetStickyTaskListRequest) returns (ResetStickyTaskListResponse);\n\n  // Returns the information from mutable state of workflow execution.\n  // It fails with 'EntityNotExistError' if specified workflow execution in unknown to the service.\n  // It returns CurrentBranchChangedError if the workflow version branch has changed.\n  rpc GetMutableState(GetMutableStateRequest) returns (GetMutableStateResponse);\n\n  // Returns the information from mutable state of workflow execution.\n  // It fails with 'EntityNotExistError' if specified workflow execution in unknown to the service.\n  // It returns CurrentBranchChangedError if the workflow version branch has changed.\n  rpc PollMutableState(PollMutableStateRequest) returns (PollMutableStateResponse);\n\n  // RecordDecisionTaskStarted is called by the MatchingService before it hands a decision task to the application worker in response to\n  // a PollForDecisionTask call. It records in the history the event that the decision task has started. It will return 'EventAlreadyStartedError',\n  // if the workflow's execution history already includes a record of the event starting.\n  rpc RecordDecisionTaskStarted(RecordDecisionTaskStartedRequest) returns (RecordDecisionTaskStartedResponse);\n\n  // RespondDecisionTaskCompleted is called by application worker to complete a DecisionTask handed as a result of\n  // 'PollForDecisionTask' API call.  Completing a DecisionTask will result in new events for the workflow execution and\n  // potentially new ActivityTask being created for corresponding decisions.  It will also create a DecisionTaskCompleted\n  // event in the history for that session.  Use the 'taskToken' provided as response of PollForDecisionTask API call\n  // for completing the DecisionTask.\n  rpc RespondDecisionTaskCompleted(RespondDecisionTaskCompletedRequest) returns (RespondDecisionTaskCompletedResponse);\n\n  // RespondDecisionTaskFailed is called by application worker to indicate failure.  This results in\n  // DecisionTaskFailedEvent written to the history and a new DecisionTask created.  This API can be used by client to\n  // either clear sticky tasklist or report ny panics during DecisionTask processing.\n  rpc RespondDecisionTaskFailed(RespondDecisionTaskFailedRequest) returns (RespondDecisionTaskFailedResponse);\n\n  // RecordActivityTaskStarted is called by the MatchingService before it hands a decision task to the application worker in response to\n  // a PollForActivityTask call. It records in the history the event that the decision task has started. It will return 'EventAlreadyStartedError',\n  // if the workflow's execution history already includes a record of the event starting.\n  rpc RecordActivityTaskStarted(RecordActivityTaskStartedRequest) returns (RecordActivityTaskStartedResponse);\n\n  // RespondActivityTaskCompleted is called by application worker when it is done processing an ActivityTask.  It will\n  // result in a new 'ActivityTaskCompleted' event being written to the workflow history and a new DecisionTask\n  // created for the workflow so new decisions could be made.  Use the 'taskToken' provided as response of\n  // PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\n  // anymore due to activity timeout.\n  rpc RespondActivityTaskCompleted(RespondActivityTaskCompletedRequest) returns (RespondActivityTaskCompletedResponse);\n\n  // RespondActivityTaskFailed is called by application worker when it is done processing an ActivityTask.  It will\n  // result in a new 'ActivityTaskFailed' event being written to the workflow history and a new DecisionTask\n  // created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\n  // PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\n  // anymore due to activity timeout.\n  rpc RespondActivityTaskFailed(RespondActivityTaskFailedRequest) returns (RespondActivityTaskFailedResponse);\n\n  // RespondActivityTaskCanceled is called by application worker when it is successfully canceled an ActivityTask.  It will\n  // result in a new 'ActivityTaskCanceled' event being written to the workflow history and a new DecisionTask\n  // created for the workflow instance so new decisions could be made.  Use the 'taskToken' provided as response of\n  // PollForActivityTask API call for completion. It fails with 'EntityNotExistsError' if the taskToken is not valid\n  // anymore due to activity timeout.\n  rpc RespondActivityTaskCanceled(RespondActivityTaskCanceledRequest) returns (RespondActivityTaskCanceledResponse);\n\n  // RecordActivityTaskHeartbeat is called by application worker while it is processing an ActivityTask.  If worker fails\n  // to heartbeat within 'heartbeatTimeoutSeconds' interval for the ActivityTask, then it will be marked as timed out and\n  // 'ActivityTaskTimedOut' event will be written to the workflow history.  Calling 'RecordActivityTaskHeartbeat' will\n  // fail with 'EntityNotExistsError' in such situations.  Use the 'taskToken' provided as response of\n  // PollForActivityTask API call for heart beating.\n  rpc RecordActivityTaskHeartbeat(RecordActivityTaskHeartbeatRequest) returns (RecordActivityTaskHeartbeatResponse);\n\n  // RequestCancelWorkflowExecution is called by application worker when it wants to request cancellation of a workflow instance.\n  // It will result in a new 'WorkflowExecutionCancelRequested' event being written to the workflow history and a new DecisionTask\n  // created for the workflow instance so new decisions could be made. It fails with 'EntityNotExistsError' if the workflow is not valid\n  // anymore due to completion or doesn't exist.\n  rpc RequestCancelWorkflowExecution(RequestCancelWorkflowExecutionRequest) returns (RequestCancelWorkflowExecutionResponse);\n\n  // RemoveSignalMutableState is used to remove a signal request ID that was previously recorded.  This is currently\n  // used to clean execution info when signal decision finished.\n  rpc RemoveSignalMutableState(RemoveSignalMutableStateRequest) returns (RemoveSignalMutableStateResponse);\n\n  // ScheduleDecisionTask is used for creating a decision task for already started workflow execution.  This is mainly\n  // used by transfer queue processor during the processing of StartChildWorkflowExecution task, where it first starts\n  // child execution without creating the decision task and then calls this API after updating the mutable state of\n  // parent execution.\n  rpc ScheduleDecisionTask(ScheduleDecisionTaskRequest) returns (ScheduleDecisionTaskResponse);\n\n  // RecordChildExecutionCompleted is used for reporting the completion of child workflow execution to parent.\n  // This is mainly called by transfer queue processor during the processing of DeleteExecution task.\n  rpc RecordChildExecutionCompleted(RecordChildExecutionCompletedRequest) returns (RecordChildExecutionCompletedResponse);\n\n  rpc ReplicateEventsV2(ReplicateEventsV2Request) returns (ReplicateEventsV2Response);\n\n  // SyncShardStatus sync the status between shards.\n  rpc SyncShardStatus(SyncShardStatusRequest) returns (SyncShardStatusResponse);\n\n  // SyncActivity sync the activity status.\n  rpc SyncActivity(SyncActivityRequest) returns (SyncActivityResponse);\n\n  // DescribeMutableState returns information about the internal states of workflow mutable state.\n  rpc DescribeMutableState(DescribeMutableStateRequest) returns (DescribeMutableStateResponse);\n\n  // DescribeHistoryHost returns information about the internal states of a history host.\n  rpc DescribeHistoryHost(DescribeHistoryHostRequest) returns (DescribeHistoryHostResponse);\n\n  // CloseShard close the shard.\n  rpc CloseShard(CloseShardRequest) returns (CloseShardResponse);\n\n  // RemoveTask remove task based on type, task_id, shard_id.\n  rpc RemoveTask(RemoveTaskRequest) returns (RemoveTaskResponse);\n\n  // ResetQueue reset processing queue state based on cluster name and type.\n  rpc ResetQueue(ResetQueueRequest) returns (ResetQueueResponse);\n\n  // DescribeQueue return queue states based on cluster name and type.\n  rpc DescribeQueue(DescribeQueueRequest) returns (DescribeQueueResponse);\n\n  // GetReplicationMessages return replication messages based on the read level.\n  rpc GetReplicationMessages(GetReplicationMessagesRequest) returns (GetReplicationMessagesResponse);\n\n  // GetDLQReplicationMessages return replication messages based on DLQ info.\n  rpc GetDLQReplicationMessages(GetDLQReplicationMessagesRequest) returns (GetDLQReplicationMessagesResponse);\n\n  // ReapplyEvents applies stale events to the current workflow and current run.\n  rpc ReapplyEvents(ReapplyEventsRequest) returns (ReapplyEventsResponse);\n\n  // RefreshWorkflowTasks refreshes all tasks of a workflow.\n  rpc RefreshWorkflowTasks(RefreshWorkflowTasksRequest) returns (RefreshWorkflowTasksResponse);\n\n  // CountDLQMessages returns DLQ message count for each shard / source cluster.\n  rpc CountDLQMessages(CountDLQMessagesRequest) returns (CountDLQMessagesResponse);\n\n  // ReadDLQMessages returns messages from DLQ.\n  rpc ReadDLQMessages(ReadDLQMessagesRequest) returns (ReadDLQMessagesResponse);\n\n  // PurgeDLQMessages purges messages from DLQ.\n  rpc PurgeDLQMessages(PurgeDLQMessagesRequest) returns (PurgeDLQMessagesResponse);\n\n  // MergeDLQMessages merges messages from DLQ.\n  rpc MergeDLQMessages(MergeDLQMessagesRequest) returns (MergeDLQMessagesResponse);\n\n  // NotifyFailoverMarkers sends failover marker to the failover coordinator.\n  rpc NotifyFailoverMarkers(NotifyFailoverMarkersRequest) returns (NotifyFailoverMarkersResponse);\n\n  // GetCrossClusterTasks return cross cluster tasks based on cluster name.\n  rpc GetCrossClusterTasks(GetCrossClusterTasksRequest) returns (GetCrossClusterTasksResponse);\n\n  // RespondCrossClusterTasksCompleted responds the result of processing cross cluster tasks.\n  rpc RespondCrossClusterTasksCompleted(RespondCrossClusterTasksCompletedRequest) returns (RespondCrossClusterTasksCompletedResponse);\n\n  // GetFailoverInfo returns information about on-going failover.\n  rpc GetFailoverInfo(GetFailoverInfoRequest) returns(GetFailoverInfoResponse);\n\n  // RatelimitUpdate pushes global-ratelimiting data to aggregating hosts,\n  // and returns data describing how to update the caller's ratelimits.\n  //\n  // For more details, see github.com/uber/cadence/common/quotas/global documentation.\n  //\n  // Request and response structures are intentionally loosely defined, to allow plugging\n  // in externally-defined algorithms without changing protocol-level details.\n  rpc RatelimitUpdate(RatelimitUpdateRequest) returns(RatelimitUpdateResponse);\n}\n\n\nmessage StartWorkflowExecutionRequest {\n  api.v1.StartWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n  api.v1.ParentExecutionInfo parent_execution_info = 3;\n  int32 attempt = 4;\n  google.protobuf.Timestamp expiration_time = 5;\n  api.v1.ContinueAsNewInitiator continue_as_new_initiator = 6;\n  api.v1.Failure continued_failure = 7;\n  api.v1.Payload last_completion_result = 8;\n  google.protobuf.Duration first_decision_task_backoff = 9;\n  map<string, string> partition_config = 10;\n}\n\nmessage StartWorkflowExecutionResponse {\n  string run_id = 1;\n}\n\nmessage SignalWorkflowExecutionRequest {\n  api.v1.SignalWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n  // workflow execution that requests this signal, for making sure\n  // the workflow being signaled is actually a child of the workflow\n  // making the request\n  api.v1.WorkflowExecution external_workflow_execution = 3;\n  bool child_workflow_only = 4;\n}\n\nmessage SignalWorkflowExecutionResponse {\n}\n\nmessage SignalWithStartWorkflowExecutionRequest {\n  api.v1.SignalWithStartWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n  map<string, string> partition_config = 3;\n}\n\nmessage SignalWithStartWorkflowExecutionResponse {\n  string run_id = 1;\n}\n\nmessage ResetWorkflowExecutionRequest {\n  api.v1.ResetWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage ResetWorkflowExecutionResponse {\n  string run_id = 1;\n}\n\nmessage TerminateWorkflowExecutionRequest {\n  api.v1.TerminateWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n  // workflow execution that requests this termination, for making sure\n  // the workflow being terminated is actually a child of the workflow\n  // making the request\n  api.v1.WorkflowExecution external_workflow_execution = 3;\n  bool child_workflow_only = 4;\n}\n\nmessage TerminateWorkflowExecutionResponse {\n}\n\nmessage DescribeWorkflowExecutionRequest {\n  api.v1.DescribeWorkflowExecutionRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage DescribeWorkflowExecutionResponse {\n  api.v1.WorkflowExecutionConfiguration execution_configuration = 1;\n  api.v1.WorkflowExecutionInfo workflow_execution_info = 2;\n  repeated api.v1.PendingActivityInfo pending_activities = 3;\n  repeated api.v1.PendingChildExecutionInfo pending_children = 4;\n  api.v1.PendingDecisionInfo pending_decision = 5;\n}\n\nmessage QueryWorkflowRequest {\n  api.v1.QueryWorkflowRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage QueryWorkflowResponse {\n  api.v1.Payload query_result = 1;\n  api.v1.QueryRejected query_rejected = 2;\n}\n\nmessage ResetStickyTaskListRequest {\n  api.v1.ResetStickyTaskListRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage ResetStickyTaskListResponse {\n}\n\nmessage GetMutableStateRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 expected_next_event_id = 3;\n  bytes current_branch_token = 4;\n  admin.v1.VersionHistoryItem version_history_item = 5;\n}\n\nmessage GetMutableStateResponse {\n  api.v1.WorkflowExecution workflow_execution = 1;\n  api.v1.WorkflowType workflow_type = 2;\n  int64 next_event_id = 3;\n  google.protobuf.Int64Value previous_started_event_id = 4;\n  int64 last_first_event_id = 5;\n  api.v1.TaskList task_list = 6;\n  api.v1.TaskList sticky_task_list = 7;\n  string client_library_version = 8;\n  string client_feature_version = 9;\n  string client_impl = 10;\n  google.protobuf.Duration sticky_task_list_schedule_to_start_timeout = 11;\n  int32 event_store_version = 12;\n  bytes current_branch_token = 13;\n  shared.v1.WorkflowState workflow_state = 14;\n  api.v1.WorkflowExecutionCloseStatus workflow_close_state = 15;\n  shared.v1.VersionHistories version_histories = 16;\n  bool is_sticky_task_list_enabled = 17;\n  int64 history_size = 18;\n\n}\n\nmessage PollMutableStateRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 expected_next_event_id = 3;\n  bytes current_branch_token = 4;\n}\n\nmessage PollMutableStateResponse {\n  api.v1.WorkflowExecution workflow_execution = 1;\n  api.v1.WorkflowType workflow_type = 2;\n  int64 next_event_id = 3;\n  google.protobuf.Int64Value previous_started_event_id = 4;\n  int64 last_first_event_id = 5;\n  api.v1.TaskList task_list = 6;\n  api.v1.TaskList sticky_task_list = 7;\n  string client_library_version = 8;\n  string client_feature_version = 9;\n  string client_impl = 10;\n  google.protobuf.Duration sticky_task_list_schedule_to_start_timeout = 11;\n  bytes current_branch_token = 12;\n  shared.v1.VersionHistories version_histories = 13;\n  shared.v1.WorkflowState workflow_state = 14;\n  api.v1.WorkflowExecutionCloseStatus workflow_close_state = 15;\n}\n\nmessage RecordDecisionTaskStartedRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 schedule_id = 3;\n  int64 task_id = 4;\n  // Unique id of each poll request. Used to ensure at most once delivery of tasks.\n  string request_id = 5;\n  api.v1.PollForDecisionTaskRequest poll_request = 6;\n}\n\nmessage RecordDecisionTaskStartedResponse {\n  api.v1.WorkflowType workflow_type = 1;\n  google.protobuf.Int64Value previous_started_event_id = 2;\n  int64 scheduled_event_id = 3;\n  int64 started_event_id = 4;\n  int64 next_event_id = 5;\n  int32 attempt = 6;\n  bool sticky_execution_enabled = 7;\n  shared.v1.TransientDecisionInfo decision_info = 8;\n  api.v1.TaskList workflow_execution_task_list = 9;\n  int32 event_store_version = 10;\n  bytes branch_token = 11;\n  google.protobuf.Timestamp scheduled_time = 12;\n  google.protobuf.Timestamp started_time = 13;\n  map<string, api.v1.WorkflowQuery> queries = 14;\n  int64 history_size = 15;\n}\n\nmessage RecordActivityTaskStartedRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 schedule_id = 3;\n  int64 task_id = 4;\n  // Unique id of each poll request. Used to ensure at most once delivery of tasks.\n  string request_id = 5;\n  api.v1.PollForActivityTaskRequest poll_request = 6;\n}\n\nmessage RecordActivityTaskStartedResponse {\n  api.v1.HistoryEvent scheduled_event = 1;\n  google.protobuf.Timestamp started_time = 2;\n  int32 attempt = 3;\n  google.protobuf.Timestamp scheduled_time_of_this_attempt = 4;\n  api.v1.Payload heartbeat_details = 5;\n  api.v1.WorkflowType workflow_type = 6;\n  string workflow_domain = 7;\n}\n\nmessage RespondDecisionTaskCompletedRequest {\n  api.v1.RespondDecisionTaskCompletedRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RespondDecisionTaskCompletedResponse {\n  RecordDecisionTaskStartedResponse started_response = 1;\n  map<string, api.v1.ActivityLocalDispatchInfo> activities_to_dispatch_locally = 2;\n}\n\nmessage RespondDecisionTaskFailedRequest {\n  api.v1.RespondDecisionTaskFailedRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RespondDecisionTaskFailedResponse {\n}\n\nmessage RecordActivityTaskHeartbeatRequest {\n  api.v1.RecordActivityTaskHeartbeatRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RecordActivityTaskHeartbeatResponse {\n  bool cancel_requested = 1;\n}\n\nmessage RespondActivityTaskCompletedRequest {\n  api.v1.RespondActivityTaskCompletedRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RespondActivityTaskCompletedResponse {\n}\n\nmessage RespondActivityTaskFailedRequest {\n  api.v1.RespondActivityTaskFailedRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RespondActivityTaskFailedResponse {\n}\n\nmessage RespondActivityTaskCanceledRequest {\n  api.v1.RespondActivityTaskCanceledRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage RespondActivityTaskCanceledResponse {\n}\n\nmessage RemoveSignalMutableStateRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  string request_id = 3;\n}\n\nmessage RemoveSignalMutableStateResponse {\n}\n\nmessage RequestCancelWorkflowExecutionRequest {\n  string domain_id = 1;\n  api.v1.RequestCancelWorkflowExecutionRequest cancel_request = 2;\n  // workflow execution that requests this cancellation, for making sure\n  // the workflow being cancelled is actually a child of the workflow\n  // making the request\n  api.v1.ExternalExecutionInfo external_execution_info = 3;\n  bool child_workflow_only = 4;\n}\n\nmessage RequestCancelWorkflowExecutionResponse {\n}\n\nmessage ScheduleDecisionTaskRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  bool is_first_decision = 3;\n}\n\nmessage ScheduleDecisionTaskResponse {\n}\n\n// RecordChildExecutionCompletedRequest is used for reporting the completion of child execution to parent workflow\n// execution which started it.  When a child execution is completed it creates this request and calls the\n// RecordChildExecutionCompleted API with the workflowExecution of parent.  It also sets the completedExecution of the\n// child as it could potentially be different than the ChildExecutionStartedEvent of parent in the situation when\n// child creates multiple runs through ContinueAsNew before finally completing.\nmessage RecordChildExecutionCompletedRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 initiated_id = 3;\n  api.v1.WorkflowExecution completed_execution = 4;\n  api.v1.HistoryEvent completion_event = 5;\n  int64 started_id = 6;\n}\n\nmessage RecordChildExecutionCompletedResponse {\n}\n\nmessage ReplicateEventsV2Request {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  repeated admin.v1.VersionHistoryItem version_history_items = 3;\n  api.v1.DataBlob events = 4;\n  // New run events does not need version history since there is no prior events.\n  api.v1.DataBlob new_run_events = 5;\n}\n\nmessage ReplicateEventsV2Response {\n}\n\nmessage SyncShardStatusRequest {\n  string source_cluster = 1;\n  int32 shard_id = 2;\n  google.protobuf.Timestamp time = 3;\n}\n\nmessage SyncShardStatusResponse {\n}\n\nmessage SyncActivityRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  int64 version = 3;\n  int64 scheduled_id = 4;\n  google.protobuf.Timestamp scheduled_time = 5;\n  int64 started_id = 6;\n  google.protobuf.Timestamp started_time = 7;\n  google.protobuf.Timestamp last_heartbeat_time = 8;\n  api.v1.Payload details = 9;\n  int32 attempt = 10;\n  api.v1.Failure last_failure = 11;\n  string last_worker_identity = 12;\n  admin.v1.VersionHistory version_history = 13;\n}\n\nmessage SyncActivityResponse {\n}\n\nmessage DescribeMutableStateRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n}\n\nmessage DescribeMutableStateResponse {\n  string mutable_state_in_cache = 1;\n  string mutable_state_in_database = 2;\n}\n\nmessage DescribeHistoryHostRequest {\n}\n\nmessage DescribeHistoryHostResponse {\n  int32 number_of_shards = 1;\n  repeated int32 shard_ids = 2;\n  admin.v1.DomainCacheInfo domain_cache = 3;\n  string shard_controller_status = 4;\n  string address = 5;\n}\n\nmessage CloseShardRequest {\n  int32 shard_id = 1;\n}\n\nmessage CloseShardResponse {\n}\n\nmessage RemoveTaskRequest {\n  int32 shard_id = 1;\n  admin.v1.TaskType task_type = 2;\n  int64 task_id = 3;\n  google.protobuf.Timestamp visibility_time = 4;\n  string cluster_name = 5;\n}\n\nmessage RemoveTaskResponse {\n}\n\nmessage ResetQueueRequest {\n  int32 shard_id = 1;\n  string cluster_name = 2;\n  admin.v1.TaskType task_type = 3;\n}\n\nmessage ResetQueueResponse {\n}\n\nmessage DescribeQueueRequest {\n  int32 shard_id = 1;\n  string cluster_name = 2;\n  admin.v1.TaskType task_type = 3;\n}\n\nmessage DescribeQueueResponse {\n  repeated string processing_queue_states = 1;\n}\n\nmessage GetReplicationMessagesRequest {\n  repeated admin.v1.ReplicationToken tokens = 1;\n  string cluster_name = 2;\n}\n\nmessage GetReplicationMessagesResponse {\n  map<int32, admin.v1.ReplicationMessages> shard_messages = 1;\n}\n\nmessage GetDLQReplicationMessagesRequest {\n  repeated admin.v1.ReplicationTaskInfo task_infos = 1;\n}\n\nmessage GetDLQReplicationMessagesResponse {\n  repeated admin.v1.ReplicationTask replication_tasks = 1;\n}\n\nmessage ReapplyEventsRequest {\n  string domain = 1;\n  string domain_id = 2;\n  api.v1.WorkflowExecution workflow_execution = 3;\n  api.v1.DataBlob events = 4;\n}\n\nmessage ReapplyEventsResponse {\n}\n\nmessage RefreshWorkflowTasksRequest {\n  string domain = 1;\n  string domain_id = 2;\n  api.v1.WorkflowExecution workflow_execution = 3;\n}\n\nmessage RefreshWorkflowTasksResponse {\n}\n\nmessage CountDLQMessagesRequest {\n  bool forceFetch = 1;\n}\n\nmessage CountDLQMessagesResponse {\n  repeated admin.v1.HistoryDLQCountEntry entries = 1;\n}\n\nmessage ReadDLQMessagesRequest {\n  admin.v1.DLQType type = 1;\n  int32 shard_id = 2;\n  string source_cluster = 3;\n  google.protobuf.Int64Value inclusive_end_message_id = 4;\n  int32 page_size = 5;\n  bytes next_page_token = 6;\n}\n\nmessage ReadDLQMessagesResponse {\n  admin.v1.DLQType type = 1;\n  repeated admin.v1.ReplicationTask replication_tasks = 2;\n  repeated admin.v1.ReplicationTaskInfo replication_tasks_info = 3;\n  bytes next_page_token = 4;\n}\n\nmessage PurgeDLQMessagesRequest {\n  admin.v1.DLQType type = 1;\n  int32 shard_id = 2;\n  string source_cluster = 3;\n  google.protobuf.Int64Value inclusive_end_message_id = 4;\n}\n\nmessage PurgeDLQMessagesResponse {\n}\n\nmessage MergeDLQMessagesRequest {\n  admin.v1.DLQType type = 1;\n  int32 shard_id = 2;\n  string source_cluster = 3;\n  google.protobuf.Int64Value inclusive_end_message_id = 4;\n  int32 page_size = 5;\n  bytes next_page_token = 6;\n}\n\nmessage MergeDLQMessagesResponse {\n  bytes next_page_token = 1;\n}\n\nmessage NotifyFailoverMarkersRequest {\n  repeated admin.v1.FailoverMarkerToken failover_marker_tokens = 1;\n}\n\nmessage NotifyFailoverMarkersResponse {\n}\n\nmessage GetCrossClusterTasksRequest {\n  repeated int32 shard_ids = 1;\n  string target_cluster = 2;\n}\n\nmessage GetCrossClusterTasksResponse {\n  map<int32, admin.v1.CrossClusterTaskRequests> tasks_by_shard = 1;\n  map<int32, admin.v1.GetTaskFailedCause> failed_cause_by_shard = 2;\n}\n\nmessage RespondCrossClusterTasksCompletedRequest {\n  int32 shard_id = 1;\n  string target_cluster = 2;\n  repeated admin.v1.CrossClusterTaskResponse task_responses = 3;\n  bool fetchNewTasks = 4;\n}\n\nmessage RespondCrossClusterTasksCompletedResponse {\n  admin.v1.CrossClusterTaskRequests tasks = 1;\n}\n\nmessage GetFailoverInfoRequest {\n    string domain_id = 1;\n}\n\nmessage GetFailoverInfoResponse {\n    int32 completed_shard_count = 1;\n    repeated int32 pending_shards = 2;\n}\n\nmessage RatelimitUpdateRequest {\n  // impl-specific data.\n  // likely some simple top-level keys and then either:\n  // - map<ratelimit-key-string, something>\n  // - list<something>\n  //\n  // this is a single blob rather than a collection to save on\n  // repeated serialization of the type name, and to allow impls\n  // to choose whatever structures are most-convenient for them.\n  shared.v1.Any data = 1;\n}\n\nmessage RatelimitUpdateResponse {\n  // impl-specific data.\n  //\n  // likely some simple top-level keys and then either:\n  // - map<ratelimit-key-string, something>\n  // - list<something>\n  //\n  // this is a single blob rather than a collection to save on\n  // repeated serialization of the type name, and to allow impls\n  // to choose whatever structures are most-convenient for them.\n  shared.v1.Any data = 1;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/indexer/v1/messages.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.indexer.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/indexer/v1;indexerv1\";\n\nimport \"uber/cadence/api/v1/common.proto\";\n\nmessage Message {\n  MessageType message_type = 1;\n  string domain_id = 2;\n  api.v1.WorkflowExecution workflow_execution = 3;\n  int64 version = 4;\n  map<string, Field> fields = 5;\n}\n\nmessage Field {\n  oneof data {\n    string string_data = 1;\n    int64 int_data = 2;\n    bool bool_data = 3;\n    bytes binary_data = 4;\n  }\n}\n\nenum MessageType {\n  MESSAGE_TYPE_INVALID = 0;\n  MESSAGE_TYPE_INDEX = 1;\n  MESSAGE_TYPE_DELETE = 2;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/matching/v1/service.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.matching.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/matching/v1;matchingv1\";\n\nimport \"google/protobuf/duration.proto\";\nimport \"google/protobuf/timestamp.proto\";\nimport \"google/protobuf/wrappers.proto\";\nimport \"uber/cadence/api/v1/common.proto\";\nimport \"uber/cadence/api/v1/query.proto\";\nimport \"uber/cadence/api/v1/tasklist.proto\";\nimport \"uber/cadence/api/v1/service_worker.proto\";\nimport \"uber/cadence/api/v1/service_workflow.proto\";\nimport \"uber/cadence/api/v1/history.proto\";\nimport \"uber/cadence/shared/v1/history.proto\";\nimport \"uber/cadence/shared/v1/tasklist.proto\";\n\n// MatchingAPI is exposed to provide support for polling from long running applications.\n// Such applications are expected to have a worker which regularly polls for DecisionTask and ActivityTask.  For each\n// DecisionTask, application is expected to process the history of events for that session and respond back with next\n// decisions.  For each ActivityTask, application is expected to execute the actual logic for that task and respond back\n// with completion or failure.\nservice MatchingAPI {\n\n  // PollForDecisionTask is called by frontend to process DecisionTask from a specific taskList.  A\n  // DecisionTask is dispatched to callers for active workflow executions, with pending decisions.\n  rpc PollForDecisionTask(PollForDecisionTaskRequest) returns (PollForDecisionTaskResponse);\n\n  // PollForActivityTask is called by frontend to process ActivityTask from a specific taskList.  ActivityTask\n  // is dispatched to callers whenever a ScheduleTask decision is made for a workflow execution.\n  rpc PollForActivityTask(PollForActivityTaskRequest) returns (PollForActivityTaskResponse);\n\n  // AddDecisionTask is called by the history service when a decision task is scheduled, so that it can be dispatched\n  // by the MatchingEngine.\n  rpc AddDecisionTask(AddDecisionTaskRequest) returns (AddDecisionTaskResponse);\n\n  // AddActivityTask is called by the history service when a decision task is scheduled, so that it can be dispatched\n  // by the MatchingEngine.\n  rpc AddActivityTask(AddActivityTaskRequest) returns (AddActivityTaskResponse);\n\n  // QueryWorkflow is called by frontend to query a workflow.\n  rpc QueryWorkflow(QueryWorkflowRequest) returns (QueryWorkflowResponse);\n\n  // RespondQueryTaskCompleted is called by frontend to respond query completed.\n  rpc RespondQueryTaskCompleted(RespondQueryTaskCompletedRequest) returns (RespondQueryTaskCompletedResponse);\n\n  // CancelOutstandingPoll is called by frontend to unblock long polls on matching for zombie pollers.\n  // Our rpc stack does not support context propagation, so when a client connection goes away frontend sees\n  // cancellation of context for that handler, but any corresponding calls (long-poll) to matching service does not\n  // see the cancellation propagated so it can unblock corresponding long-polls on its end.  This results is tasks\n  // being dispatched to zombie pollers in this situation.  This API is added so everytime frontend makes a long-poll\n  // api call to matching it passes in a pollerID and then calls this API when it detects client connection is closed\n  // to unblock long polls for this poller and prevent tasks being sent to these zombie pollers.\n  rpc CancelOutstandingPoll(CancelOutstandingPollRequest) returns (CancelOutstandingPollResponse);\n\n  // DescribeTaskList returns information about the target tasklist, right now this API returns the\n  // pollers which polled this tasklist in last few minutes.\n  rpc DescribeTaskList(DescribeTaskListRequest) returns (DescribeTaskListResponse);\n\n  // ListTaskListPartitions returns a map of partitionKey and hostAddress for a taskList\n  rpc ListTaskListPartitions(ListTaskListPartitionsRequest) returns (ListTaskListPartitionsResponse);\n\n  // GetTaskListsByDomain returns all tasklist for a given domain\n  rpc GetTaskListsByDomain(GetTaskListsByDomainRequest) returns (GetTaskListsByDomainResponse);\n\n  // UpdateTaskListPartitionConfig is called to update the partition config of a task list in the database\n  // and notify all partitions of the task list to update the cache of partition config of the task list\n  // his API is used by frontend service to forward update request initiated mostly from CLI tool to update\n  // the partition config of a task list and notify all partitions of the task list to update their cache of task list partition config.\n  rpc UpdateTaskListPartitionConfig(UpdateTaskListPartitionConfigRequest) returns (UpdateTaskListPartitionConfigResponse);\n\n  // RefreshTaskListPartitionConfig is called to update the cache of partition config of a task list\n  // This API is mainly used by matching service to notify a task list partition to update its cache\n  // of task list partition config. It can also be used by frontend service to forward request initiated\n  // from admin CLI tool to sync the cache of task list partition config if something goes wrong.\n  rpc RefreshTaskListPartitionConfig(RefreshTaskListPartitionConfigRequest) returns (RefreshTaskListPartitionConfigResponse);\n}\n\nmessage TaskListPartition {\n  repeated string isolation_groups = 1;\n}\n\nmessage TaskListPartitionConfig {\n  int64 version = 1;\n  int32 num_read_partitions = 2 [deprecated = true];\n  int32 num_write_partitions = 3 [deprecated = true];\n  map<int32, TaskListPartition> read_partitions = 4;\n  map<int32, TaskListPartition> write_partitions = 5;\n}\n\nmessage LoadBalancerHints {\n  int64 backlog_count = 1;\n  double rate_per_second = 2;\n}\n\nmessage PollForDecisionTaskRequest {\n  api.v1.PollForDecisionTaskRequest request = 1;\n  string domain_id = 2;\n  string poller_id = 3;\n  string forwarded_from = 4;\n  string isolation_group = 5;\n}\n\nmessage PollForDecisionTaskResponse {\n  bytes task_token = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  api.v1.WorkflowType workflow_type = 3;\n  google.protobuf.Int64Value previous_started_event_id = 4;\n  int64 started_event_id = 5;\n  int32 attempt = 6;\n  int64 next_event_id = 7;\n  int64 backlog_count_hint = 8;\n  bool sticky_execution_enabled = 9;\n  api.v1.WorkflowQuery query = 10;\n  shared.v1.TransientDecisionInfo decision_info = 11;\n  api.v1.TaskList workflow_execution_task_list = 12;\n  int32 event_store_version = 13;\n  bytes branch_token = 14;\n  google.protobuf.Timestamp scheduled_time = 15;\n  google.protobuf.Timestamp started_time = 16;\n  map<string, api.v1.WorkflowQuery> queries = 17;\n  int64 total_history_bytes = 18;\n  TaskListPartitionConfig partition_config = 19;\n  LoadBalancerHints load_balancer_hints = 20;\n  api.v1.AutoConfigHint auto_config_hint = 21;\n}\n\nmessage PollForActivityTaskRequest {\n  api.v1.PollForActivityTaskRequest request = 1;\n  string domain_id = 2;\n  string poller_id = 3;\n  string forwarded_from = 4;\n  string isolation_group = 5;\n}\n\nmessage PollForActivityTaskResponse {\n  bytes task_token = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  string activity_id = 3;\n  api.v1.ActivityType activity_type = 4;\n  api.v1.Payload input = 5;\n  google.protobuf.Timestamp scheduled_time = 6;\n  google.protobuf.Timestamp started_time = 7;\n  google.protobuf.Duration schedule_to_close_timeout = 8;\n  google.protobuf.Duration start_to_close_timeout = 9;\n  google.protobuf.Duration heartbeat_timeout = 10;\n  int32 attempt = 11;\n  google.protobuf.Timestamp scheduled_time_of_this_attempt = 12;\n  api.v1.Payload heartbeat_details = 13;\n  api.v1.WorkflowType workflow_type = 14;\n  string workflow_domain = 15;\n  api.v1.Header header = 16;\n  LoadBalancerHints load_balancer_hints = 17;\n  TaskListPartitionConfig partition_config = 19;\n  api.v1.AutoConfigHint auto_config_hint = 20;\n}\n\nmessage AddDecisionTaskRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  api.v1.TaskList task_list = 3;\n  int64 schedule_id = 4;\n  google.protobuf.Duration schedule_to_start_timeout = 5;\n  shared.v1.TaskSource source = 6;\n  string forwarded_from = 7;\n  map<string, string> partition_config = 8;\n}\n\nmessage AddDecisionTaskResponse {\n  TaskListPartitionConfig partition_config = 1;\n}\n\nmessage AddActivityTaskRequest {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  string source_domain_id = 3;\n  api.v1.TaskList task_list = 4;\n  int64 schedule_id = 5;\n  google.protobuf.Duration schedule_to_start_timeout = 6;\n  shared.v1.TaskSource source = 7;\n  string forwarded_from = 8;\n  ActivityTaskDispatchInfo activityTaskDispatchInfo = 9;\n  map<string, string> partition_config = 10;\n}\n\nmessage ActivityTaskDispatchInfo {\n  api.v1.HistoryEvent scheduled_event = 1;\n  google.protobuf.Timestamp started_time = 2;\n  int32 attempt = 3;\n  google.protobuf.Timestamp scheduled_time_of_this_attempt = 4;\n  api.v1.Payload heartbeat_details = 5;\n  api.v1.WorkflowType workflow_type = 6;\n  string workflow_domain = 7;\n}\n\nmessage AddActivityTaskResponse {\n  TaskListPartitionConfig partition_config = 1;\n}\n\nmessage QueryWorkflowRequest {\n  api.v1.QueryWorkflowRequest request = 1;\n  string domain_id = 2;\n  api.v1.TaskList task_list = 3;\n  string forwarded_from = 4;\n}\n\nmessage QueryWorkflowResponse {\n  api.v1.Payload query_result = 1;\n  api.v1.QueryRejected query_rejected = 2;\n  api.v1.TaskListPartitionConfig partition_config = 3;\n}\n\nmessage RespondQueryTaskCompletedRequest {\n  api.v1.RespondQueryTaskCompletedRequest request = 1;\n  string domain_id = 2;\n  api.v1.TaskList task_list = 3;\n  string task_id = 4;\n}\n\nmessage RespondQueryTaskCompletedResponse {\n}\n\nmessage CancelOutstandingPollRequest {\n  string domain_id = 1;\n  string poller_id = 2;\n  api.v1.TaskListType task_list_type = 3;\n  api.v1.TaskList task_list = 4;\n}\n\nmessage CancelOutstandingPollResponse {\n}\n\nmessage DescribeTaskListRequest {\n  api.v1.DescribeTaskListRequest request = 1;\n  string domain_id = 2;\n}\n\nmessage DescribeTaskListResponse {\n  repeated api.v1.PollerInfo pollers = 1;\n  api.v1.TaskListStatus task_list_status = 2;\n  api.v1.TaskListPartitionConfig partition_config = 3;\n  api.v1.TaskList task_list = 4;\n}\n\nmessage ListTaskListPartitionsRequest {\n  string domain = 1;\n  api.v1.TaskList task_list = 2;\n}\n\nmessage ListTaskListPartitionsResponse {\n  repeated api.v1.TaskListPartitionMetadata activity_task_list_partitions = 1;\n  repeated api.v1.TaskListPartitionMetadata decision_task_list_partitions = 2;\n}\n\nmessage GetTaskListsByDomainRequest {\n  string domain = 1;\n}\n\nmessage GetTaskListsByDomainResponse {\n  map <string,DescribeTaskListResponse> decision_task_list_map = 1;\n  map <string,DescribeTaskListResponse> activity_task_list_map = 2;\n}\n\nmessage UpdateTaskListPartitionConfigRequest {\n  string domain_id = 1;\n  api.v1.TaskList task_list = 2;\n  api.v1.TaskListType task_list_type = 3;\n  api.v1.TaskListPartitionConfig partition_config = 4;\n}\n\nmessage UpdateTaskListPartitionConfigResponse {}\n\nmessage RefreshTaskListPartitionConfigRequest {\n  string domain_id = 1;\n  api.v1.TaskList task_list = 2;\n  api.v1.TaskListType task_list_type = 3;\n  api.v1.TaskListPartitionConfig partition_config = 4;\n}\n\nmessage RefreshTaskListPartitionConfigResponse {\n\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/sharddistributor/v1/canary.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.sharddistributor.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/sharddistributor/v1;sharddistributorv1\";\n\n// ShardDistributorExecutorCanaryAPI is used for canary testing executor-to-executor communication\nservice ShardDistributorExecutorCanaryAPI {\n  // Ping allows one executor to ping another executor that owns a specific shard\n  rpc Ping(PingRequest) returns (PingResponse);\n}\n\nmessage PingRequest {\n  string shard_key = 1;\n  string namespace = 2;\n}\n\nmessage PingResponse {\n  string executor_id = 1;\n  bool owns_shard = 2;\n  string shard_key = 3;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/sharddistributor/v1/executor.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.sharddistributor.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/sharddistributor/v1;sharddistributorv1\";\n\n// ShardDistributorExecutionAPI is used update the state of the executor and fetch the next shard assignments.\nservice ShardDistributorExecutorAPI {\n\n  // Heartbeat reports the current state of the executor, and fetches the next shard assignments.\n  rpc Heartbeat(HeartbeatRequest) returns (HeartbeatResponse);\n}\n\nmessage HeartbeatRequest {\n  string namespace = 1;\n  string executor_id = 2;\n  ExecutorStatus status = 3;\n  map<string, ShardStatusReport> shard_status_reports = 4;\n  map<string, string> metadata = 5;\n}\n\nenum ExecutorStatus {\n  EXECUTOR_STATUS_INVALID = 0;\n  EXECUTOR_STATUS_ACTIVE = 1;\n  EXECUTOR_STATUS_DRAINING = 2;\n  EXECUTOR_STATUS_DRAINED = 3;\n}\n\nmessage ShardStatusReport {\n  ShardStatus status = 1;\n  double shard_load = 2;\n}\n\n// We only have one status for now, but when adding\n// graceful handover, we will need to add more statuses.\n// We do not need an \"inactive\" status, as we will not include\n// inactive shards in the heartbeat request.\nenum ShardStatus {\n  SHARD_STATUS_INVALID = 0;\n  SHARD_STATUS_READY = 1;\n  SHARD_STATUS_DONE = 2;\n}\n\n\n\nmessage HeartbeatResponse {\n  map<string, ShardAssignment> shard_assignments = 1;\n  MigrationMode migration_mode = 2;\n}\n\nmessage ShardAssignment {\n  AssignmentStatus status = 1;\n}\n\n// We only have one status for now, but when adding\n// graceful handover, we will need to add more statuses.\n// We do not need an \"inactive\" status, as we will not include\n// inactive shards in the heartbeat request.\nenum AssignmentStatus {\n  ASSIGNMENT_STATUS_INVALID = 0;\n  ASSIGNMENT_STATUS_READY = 1;\n}\n\n// We handle  the migration steps from SD side\nenum MigrationMode {\n  MIGRATION_MODE_INVALID = 0;\n  MIGRATION_MODE_LOCAL_PASSTHROUGH = 1;\n  MIGRATION_MODE_LOCAL_PASSTHROUGH_SHADOW = 2;\n  MIGRATION_MODE_DISTRIBUTED_PASSTHROUGH = 3;\n  MIGRATION_MODE_ONBOARDED = 4;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/sharddistributor/v1/service.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.sharddistributor.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/sharddistributor/v1;sharddistributorv1\";\n\n// ShardDistributorAPI is used to query shard distributor for the current owner of a shard\nservice ShardDistributorAPI {\n\n  // GetShardOwner returns the owner of a specific shard\n  rpc GetShardOwner(GetShardOwnerRequest) returns (GetShardOwnerResponse);\n\n  // WatchNamespaceState streams updates about executors and their assigned shards for a namespace\n  rpc WatchNamespaceState(WatchNamespaceStateRequest) returns (stream WatchNamespaceStateResponse);\n}\n\nmessage GetShardOwnerRequest {\n  string shard_key = 1;\n  string namespace = 2;\n}\n\nmessage GetShardOwnerResponse {\n  string owner = 1;\n  string namespace = 2;\n  map<string, string> metadata = 3;\n}\n\nmessage NamespaceNotFoundError {\n  string namespace = 1;\n}\n\nmessage ShardNotFoundError {\n  string namespace = 1;\n  string shard_key = 2;\n}\n\nmessage WatchNamespaceStateRequest {\n  string namespace = 1;\n}\n\nmessage WatchNamespaceStateResponse {\n  // Executor information with assigned shards\n  repeated ExecutorInfo executors = 1;\n}\n\nmessage ExecutorInfo {\n  string executor_id = 1;\n  map<string, string> metadata = 2;\n  repeated Shard shards = 3;\n}\n\nmessage Shard {\n  string shard_key = 1;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/shared/v1/any.proto",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.shared.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/shared/v1;sharedv1\";\n\n// Any is a logical duplicate of google.protobuf.Any, but is intentionally\n// breaking compatibility because it will not be directly used as a\n// google.protobuf.Any in our high-level RPC mappers.\n//\n// The intent of the type is the same, but it is not intended to be directly\n// compatible with google.protobuf.Any - this blob is RPC-type agnostic by\n// design (as the underlying data may be transported over proto or thrift), and\n// the data-bytes may or may not be proto-encoded.\n//\n// This is intentionally different from DataBlob, which supports only a handful\n// of known encodings so it can be interpreted everywhere.  Any supports literally\n// any contents, and needs to be considered opaque until it is given to something\n// that is expecting it.\n//\n// See value_type to interpret the contents.\nmessage Any {\n  // Type-string describing value's contents, and intentionally avoiding the\n  // name \"type\" as it is often a special term.\n  // This should usually be a hard-coded string of some kind.\n  string value_type = 1;\n  // Arbitrarily-encoded bytes, to be deserialized by a runtime implementation.\n  // The contents are described by value_type.\n  bytes value = 2;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/shared/v1/error.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.shared.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/shared/v1;sharedv1\";\n\nimport \"uber/cadence/api/v1/common.proto\";\nimport \"uber/cadence/admin/v1/history.proto\";\n\nmessage CurrentBranchChangedError {\n  bytes current_branch_token = 1;\n}\n\nmessage InternalDataInconsistencyError {\n}\n\nmessage EventAlreadyStartedError {\n}\n\nmessage RemoteSyncMatchedError {\n}\n\nmessage RetryTaskV2Error {\n  string domain_id = 1;\n  api.v1.WorkflowExecution workflow_execution = 2;\n  admin.v1.VersionHistoryItem start_event = 3;\n  admin.v1.VersionHistoryItem end_event = 4;\n}\n\nmessage ShardOwnershipLostError {\n  string owner = 1;\n}\n\nmessage TaskListNotOwnedByHostError {\n  string owned_by_identity = 1;\n  string my_identity = 2;\n  string task_list_name = 3;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/shared/v1/history.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.shared.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/shared/v1;sharedv1\";\n\nimport \"uber/cadence/api/v1/history.proto\";\nimport \"uber/cadence/admin/v1/history.proto\";\n\nmessage TransientDecisionInfo {\n  api.v1.HistoryEvent scheduled_event = 1;\n  api.v1.HistoryEvent started_event = 2;\n}\n\n// VersionHistories contains all version histories from all branches.\nmessage VersionHistories {\n  int32 current_version_history_index = 1;\n  repeated admin.v1.VersionHistory histories = 2;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/shared/v1/tasklist.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.shared.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/shared/v1;sharedv1\";\n\n// TaskSource is the source from which a task was produced.\nenum TaskSource {\n  TASK_SOURCE_INVALID = 0;\n  // Task produced by history service.\n  TASK_SOURCE_HISTORY = 1;\n  // Task produced from matching db backlog.\n  TASK_SOURCE_DB_BACKLOG = 2;\n}\n"
  },
  {
    "path": "proto/internal/uber/cadence/shared/v1/workflow.proto",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\nsyntax = \"proto3\";\n\npackage uber.cadence.shared.v1;\n\noption go_package = \"github.com/uber/cadence/.gen/proto/shared/v1;sharedv1\";\n\nenum WorkflowState {\n  WORKFLOW_STATE_INVALID = 0;\n  WORKFLOW_STATE_CREATED = 1;\n  WORKFLOW_STATE_RUNNING = 2;\n  WORKFLOW_STATE_COMPLETED = 3;\n  WORKFLOW_STATE_ZOMBIE = 4;\n  WORKFLOW_STATE_VOID = 5;\n  WORKFLOW_STATE_CORRUPTED = 6;\n}\n"
  },
  {
    "path": "proto/persistenceblobs/v1/gogo.proto",
    "content": "// Protocol Buffers for Go with Gadgets\n//\n// Copyright (c) 2013, The GoGo Authors. All rights reserved.\n// http://github.com/temporalio/gogo-protobuf\n//\n// Redistribution and use in source and binary forms, with or without\n// modification, are permitted provided that the following conditions are\n// met:\n//\n//     * Redistributions of source code must retain the above copyright\n// notice, this list of conditions and the following disclaimer.\n//     * Redistributions in binary form must reproduce the above\n// copyright notice, this list of conditions and the following disclaimer\n// in the documentation and/or other materials provided with the\n// distribution.\n//\n// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n// \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\nsyntax = \"proto2\";\npackage gogoproto;\n\nimport \"google/protobuf/descriptor.proto\";\n\noption go_package = \".gen/proto\";\n\nextend google.protobuf.EnumOptions {\n    optional bool goproto_enum_prefix = 62001;\n    optional bool goproto_enum_stringer = 62021;\n    optional bool enum_stringer = 62022;\n    optional string enum_customname = 62023;\n    optional bool enumdecl = 62024;\n}\n\nextend google.protobuf.EnumValueOptions {\n    optional string enumvalue_customname = 66001;\n}\n\nextend google.protobuf.FileOptions {\n    optional bool goproto_getters_all = 63001;\n    optional bool goproto_enum_prefix_all = 63002;\n    optional bool goproto_stringer_all = 63003;\n    optional bool verbose_equal_all = 63004;\n    optional bool face_all = 63005;\n    optional bool gostring_all = 63006;\n    optional bool populate_all = 63007;\n    optional bool stringer_all = 63008;\n    optional bool onlyone_all = 63009;\n\n    optional bool equal_all = 63013;\n    optional bool description_all = 63014;\n    optional bool testgen_all = 63015;\n    optional bool benchgen_all = 63016;\n    optional bool marshaler_all = 63017;\n    optional bool unmarshaler_all = 63018;\n    optional bool stable_marshaler_all = 63019;\n\n    optional bool sizer_all = 63020;\n\n    optional bool goproto_enum_stringer_all = 63021;\n    optional bool enum_stringer_all = 63022;\n\n    optional bool unsafe_marshaler_all = 63023;\n    optional bool unsafe_unmarshaler_all = 63024;\n\n    optional bool goproto_extensions_map_all = 63025;\n    optional bool goproto_unrecognized_all = 63026;\n    optional bool gogoproto_import = 63027;\n    optional bool protosizer_all = 63028;\n    optional bool compare_all = 63029;\n    optional bool typedecl_all = 63030;\n    optional bool enumdecl_all = 63031;\n\n    optional bool goproto_registration = 63032;\n    optional bool messagename_all = 63033;\n\n    optional bool goproto_sizecache_all = 63034;\n    optional bool goproto_unkeyed_all = 63035;\n}\n\nextend google.protobuf.MessageOptions {\n    optional bool goproto_getters = 64001;\n    optional bool goproto_stringer = 64003;\n    optional bool verbose_equal = 64004;\n    optional bool face = 64005;\n    optional bool gostring = 64006;\n    optional bool populate = 64007;\n    optional bool stringer = 67008;\n    optional bool onlyone = 64009;\n\n    optional bool equal = 64013;\n    optional bool description = 64014;\n    optional bool testgen = 64015;\n    optional bool benchgen = 64016;\n    optional bool marshaler = 64017;\n    optional bool unmarshaler = 64018;\n    optional bool stable_marshaler = 64019;\n\n    optional bool sizer = 64020;\n\n    optional bool unsafe_marshaler = 64023;\n    optional bool unsafe_unmarshaler = 64024;\n\n    optional bool goproto_extensions_map = 64025;\n    optional bool goproto_unrecognized = 64026;\n\n    optional bool protosizer = 64028;\n    optional bool compare = 64029;\n\n    optional bool typedecl = 64030;\n\n    optional bool messagename = 64033;\n\n    optional bool goproto_sizecache = 64034;\n    optional bool goproto_unkeyed = 64035;\n}\n\nextend google.protobuf.FieldOptions {\n    optional bool nullable = 65001;\n    optional bool embed = 65002;\n    optional string customtype = 65003;\n    optional string customname = 65004;\n    optional string jsontag = 65005;\n    optional string moretags = 65006;\n    optional string casttype = 65007;\n    optional string castkey = 65008;\n    optional string castvalue = 65009;\n\n    optional bool stdtime = 65010;\n    optional bool stdduration = 65011;\n    optional bool wktpointer = 65012;\n\n}\n"
  },
  {
    "path": "proto/persistenceblobs/v1/message.proto",
    "content": "// Modifications Copyright (c) 2020 Uber Technologies Inc.\n\n// Copyright (c) 2020 Temporal Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\nsyntax = \"proto3\";\noption go_package = \".gen/proto\";\n\nimport \"google/protobuf/duration.proto\";\nimport \"google/protobuf/timestamp.proto\";\nimport \"google/protobuf/wrappers.proto\";\n\nimport \"gogo.proto\";\n\nmessage DomainDetail {\n    DomainInfo info = 1;\n    DomainConfig config = 2;\n    DomainReplicationConfig replication_config = 3;\n    int64 config_version = 4;\n    int64 failover_notification_version = 5;\n    int64 failover_version = 6;\n    google.protobuf.Timestamp failover_end_time = 7 [(gogoproto.stdtime) = true];\n}\n\nmessage DomainInfo {\n    string id = 1;\n    DomainState state = 2;\n    string name = 3;\n    string description = 4;\n    string owner = 5;\n    map<string, string> data = 6;\n}\n\nmessage DomainReplicationConfig {\n    string active_cluster_name = 1;\n    repeated string clusters = 2;\n}\n\nmessage DomainConfig {\n    google.protobuf.Duration retention = 1 [(gogoproto.stdduration) = true];\n    string archival_bucket = 2;\n    BadBinaries bad_binaries = 3;\n    ArchivalState history_archival_state = 4;\n    string history_archival_uri = 5;\n    ArchivalState visibility_archival_state = 6;\n    string visibility_archival_uri = 7;\n}\n\nmessage BadBinaries {\n    map<string, BadBinaryInfo> binaries = 1;\n}\n\nmessage BadBinaryInfo {\n    string reason = 1;\n    string operator = 2;\n    google.protobuf.Timestamp create_time = 3 [(gogoproto.stdtime) = true];\n}\n\nenum ArchivalState {\n    ARCHIVAL_STATE_UNSPECIFIED = 0;\n    ARCHIVAL_STATE_DISABLED = 1;\n    ARCHIVAL_STATE_ENABLED = 2;\n}\n\nenum DomainState {\n    DOMAIN_STATE_UNSPECIFIED = 0;\n    DOMAIN_STATE_REGISTERED = 1;\n    DOMAIN_STATE_DEPRECATED = 2;\n    DOMAIN_STATE_DELETED = 3;\n}\n\nmessage ActivityInfo {\n    int64 version = 1;\n    int64 scheduled_event_batch_id = 2;\n    HistoryEvent scheduled_event = 3;\n    google.protobuf.Timestamp scheduled_time = 4 [(gogoproto.stdtime) = true];\n    int64 started_id = 5;\n    HistoryEvent started_event = 6;\n    google.protobuf.Timestamp started_time = 7 [(gogoproto.stdtime) = true];\n    string activity_id = 8;\n    string request_id = 9;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration schedule_to_start_timeout = 10 [(gogoproto.stdduration) = true];\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration schedule_to_close_timeout = 11 [(gogoproto.stdduration) = true];\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration start_to_close_timeout = 12 [(gogoproto.stdduration) = true];\n    google.protobuf.Duration heartbeat_timeout = 13 [(gogoproto.stdduration) = true];\n    bool cancel_requested = 14;\n    int64 cancel_request_id = 15;\n    int32 timer_task_status = 16;\n    int32 attempt = 17;\n    string task_list = 18;\n    string started_identity = 19;\n    bool has_retry_policy = 20;\n    google.protobuf.Duration retry_initial_interval = 21 [(gogoproto.stdduration) = true];\n    google.protobuf.Duration retry_maximum_interval = 22 [(gogoproto.stdduration) = true];\n    int32 retry_maximum_attempts = 23;\n    google.protobuf.Timestamp retry_expiration_time = 24 [(gogoproto.stdtime) = true];\n    double retry_backoff_coefficient = 25;\n    repeated string retry_non_retryable_error_types = 26;\n    Failure retry_last_failure = 27;\n    string retry_last_worker_identity = 28;\n    string domain_id = 29;\n    int64 schedule_id = 30;\n}\n\nmessage HistoryEvent {\n    int64 event_id = 1;\n    google.protobuf.Timestamp event_time = 2 [(gogoproto.stdtime) = true];\n    EventType event_type = 3;\n    int64 version = 4;\n    int64 task_id = 5;\n    oneof attributes {\n        WorkflowExecutionStartedEventAttributes workflow_execution_started_event_attributes = 6;\n        WorkflowExecutionCompletedEventAttributes workflow_execution_completed_event_attributes = 7;\n        WorkflowExecutionFailedEventAttributes workflow_execution_failed_event_attributes = 8;\n        WorkflowExecutionTimedOutEventAttributes workflow_execution_timed_out_event_attributes = 9;\n        DecisionTaskScheduledEventAttributes decision_task_scheduled_event_attributes = 10;\n        DecisionTaskStartedEventAttributes decision_task_started_event_attributes = 11;\n        DecisionTaskCompletedEventAttributes decision_task_completed_event_attributes = 12;\n        DecisionTaskTimedOutEventAttributes decision_task_timed_out_event_attributes = 13;\n        DecisionTaskFailedEventAttributes decision_task_failed_event_attributes = 14;\n        ActivityTaskScheduledEventAttributes activity_task_scheduled_event_attributes = 15;\n        ActivityTaskStartedEventAttributes activity_task_started_event_attributes = 16;\n        ActivityTaskCompletedEventAttributes activity_task_completed_event_attributes = 17;\n        ActivityTaskFailedEventAttributes activity_task_failed_event_attributes = 18;\n        ActivityTaskTimedOutEventAttributes activity_task_timed_out_event_attributes = 19;\n        TimerStartedEventAttributes timer_started_event_attributes = 20;\n        TimerFiredEventAttributes timer_fired_event_attributes = 21;\n        ActivityTaskCancelRequestedEventAttributes activity_task_cancel_requested_event_attributes = 22;\n        ActivityTaskCanceledEventAttributes activity_task_canceled_event_attributes = 23;\n        TimerCanceledEventAttributes timer_canceled_event_attributes = 24;\n        MarkerRecordedEventAttributes marker_recorded_event_attributes = 25;\n        WorkflowExecutionSignaledEventAttributes workflow_execution_signaled_event_attributes = 26;\n        WorkflowExecutionTerminatedEventAttributes workflow_execution_terminated_event_attributes = 27;\n        WorkflowExecutionCancelRequestedEventAttributes workflow_execution_cancel_requested_event_attributes = 28;\n        WorkflowExecutionCanceledEventAttributes workflow_execution_canceled_event_attributes = 29;\n        RequestCancelExternalWorkflowExecutionInitiatedEventAttributes request_cancel_external_workflow_execution_initiated_event_attributes = 30;\n        RequestCancelExternalWorkflowExecutionFailedEventAttributes request_cancel_external_workflow_execution_failed_event_attributes = 31;\n        ExternalWorkflowExecutionCancelRequestedEventAttributes external_workflow_execution_cancel_requested_event_attributes = 32;\n        WorkflowExecutionContinuedAsNewEventAttributes workflow_execution_continued_as_new_event_attributes = 33;\n        StartChildWorkflowExecutionInitiatedEventAttributes start_child_workflow_execution_initiated_event_attributes = 34;\n        StartChildWorkflowExecutionFailedEventAttributes start_child_workflow_execution_failed_event_attributes = 35;\n        ChildWorkflowExecutionStartedEventAttributes child_workflow_execution_started_event_attributes = 36;\n        ChildWorkflowExecutionCompletedEventAttributes child_workflow_execution_completed_event_attributes = 37;\n        ChildWorkflowExecutionFailedEventAttributes child_workflow_execution_failed_event_attributes = 38;\n        ChildWorkflowExecutionCanceledEventAttributes child_workflow_execution_canceled_event_attributes = 39;\n        ChildWorkflowExecutionTimedOutEventAttributes child_workflow_execution_timed_out_event_attributes = 40;\n        ChildWorkflowExecutionTerminatedEventAttributes child_workflow_execution_terminated_event_attributes = 41;\n        SignalExternalWorkflowExecutionInitiatedEventAttributes signal_external_workflow_execution_initiated_event_attributes = 42;\n        SignalExternalWorkflowExecutionFailedEventAttributes signal_external_workflow_execution_failed_event_attributes = 43;\n        ExternalWorkflowExecutionSignaledEventAttributes external_workflow_execution_signaled_event_attributes = 44;\n        UpsertWorkflowSearchAttributesEventAttributes upsert_workflow_search_attributes_event_attributes = 45;\n    }\n}\n\n// Whenever this list of events is changed do change the function shouldBufferEvent in mutableStateBuilder.go to make sure to do the correct event ordering.\nenum EventType {\n    EVENT_TYPE_UNSPECIFIED = 0;\n    EVENT_TYPE_WORKFLOW_EXECUTION_STARTED = 1;\n    EVENT_TYPE_WORKFLOW_EXECUTION_COMPLETED = 2;\n    EVENT_TYPE_WORKFLOW_EXECUTION_FAILED = 3;\n    EVENT_TYPE_WORKFLOW_EXECUTION_TIMED_OUT = 4;\n    EVENT_TYPE_DECISION_TASK_SCHEDULED = 5;\n    EVENT_TYPE_DECISION_TASK_STARTED = 6;\n    EVENT_TYPE_DECISION_TASK_COMPLETED = 7;\n    EVENT_TYPE_DECISION_TASK_TIMED_OUT = 8;\n    EVENT_TYPE_DECISION_TASK_FAILED = 9;\n    EVENT_TYPE_ACTIVITY_TASK_SCHEDULED = 10;\n    EVENT_TYPE_ACTIVITY_TASK_STARTED = 11;\n    EVENT_TYPE_ACTIVITY_TASK_COMPLETED = 12;\n    EVENT_TYPE_ACTIVITY_TASK_FAILED = 13;\n    EVENT_TYPE_ACTIVITY_TASK_TIMED_OUT = 14;\n    EVENT_TYPE_ACTIVITY_TASK_CANCEL_REQUESTED = 15;\n    EVENT_TYPE_ACTIVITY_TASK_CANCELED = 16;\n    EVENT_TYPE_TIMER_STARTED = 17;\n    EVENT_TYPE_TIMER_FIRED = 18;\n    EVENT_TYPE_TIMER_CANCELED = 19;\n    EVENT_TYPE_WORKFLOW_EXECUTION_CANCEL_REQUESTED = 20;\n    EVENT_TYPE_WORKFLOW_EXECUTION_CANCELED = 21;\n    EVENT_TYPE_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_INITIATED = 22;\n    EVENT_TYPE_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED = 23;\n    EVENT_TYPE_EXTERNAL_WORKFLOW_EXECUTION_CANCEL_REQUESTED = 24;\n    EVENT_TYPE_MARKER_RECORDED = 25;\n    EVENT_TYPE_WORKFLOW_EXECUTION_SIGNALED = 26;\n    EVENT_TYPE_WORKFLOW_EXECUTION_TERMINATED = 27;\n    EVENT_TYPE_WORKFLOW_EXECUTION_CONTINUED_AS_NEW = 28;\n    EVENT_TYPE_START_CHILD_WORKFLOW_EXECUTION_INITIATED = 29;\n    EVENT_TYPE_START_CHILD_WORKFLOW_EXECUTION_FAILED = 30;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_STARTED = 31;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_COMPLETED = 32;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_FAILED = 33;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_CANCELED = 34;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_TIMED_OUT = 35;\n    EVENT_TYPE_CHILD_WORKFLOW_EXECUTION_TERMINATED = 36;\n    EVENT_TYPE_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_INITIATED = 37;\n    EVENT_TYPE_SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED = 38;\n    EVENT_TYPE_EXTERNAL_WORKFLOW_EXECUTION_SIGNALED = 39;\n    EVENT_TYPE_UPSERT_WORKFLOW_SEARCH_ATTRIBUTES = 40;\n}\n\nmessage WorkflowExecutionStartedEventAttributes {\n    WorkflowType workflow_type = 1;\n    string parent_workflow_domain = 2;\n    WorkflowExecution parent_workflow_execution = 3;\n    int64 parent_initiated_event_id = 4;\n    TaskList task_list = 5;\n    Payloads input = 6;\n    // Total workflow execution timeout including retries and continue as new.\n    google.protobuf.Duration workflow_execution_timeout = 7 [(gogoproto.stdduration) = true];\n    // Timeout of a single workflow run.\n    google.protobuf.Duration workflow_run_timeout = 8 [(gogoproto.stdduration) = true];\n    // Timeout of a single workflow task.\n    google.protobuf.Duration decision_task_timeout = 9 [(gogoproto.stdduration) = true];\n    string continued_execution_run_id = 10;\n    ContinueAsNewInitiator initiator = 11;\n    Failure continued_failure = 12;\n    Payloads last_completion_result = 13;\n    // This is the runId when the WorkflowExecutionStarted event is written.\n    string original_execution_run_id = 14;\n    string identity = 15;\n    // This is the very first runId along the chain of ContinueAsNew and Reset.\n    string first_execution_run_id = 16;\n    RetryPolicy retry_policy = 17;\n    int32 attempt = 18;\n    // The absolute time at which workflow is timed out.\n    // This time is passed without change to the next run/retry of a workflow.\n    google.protobuf.Timestamp workflow_execution_expiration_time = 19 [(gogoproto.stdtime) = true];\n    string cron_schedule = 20;\n    google.protobuf.Duration first_decision_task_backoff = 21 [(gogoproto.stdduration) = true];\n    Memo memo = 22;\n    SearchAttributes search_attributes = 23;\n    ResetPoints prev_auto_reset_points = 24;\n    Header header = 25;\n}\n\nmessage WorkflowExecutionCompletedEventAttributes {\n    Payloads result = 1;\n    int64 decision_task_completed_event_id = 2;\n}\n\nmessage WorkflowExecutionFailedEventAttributes {\n    Failure failure = 1;\n    RetryState retry_state = 2;\n    int64 decision_task_completed_event_id = 3;\n}\n\nmessage WorkflowExecutionTimedOutEventAttributes {\n    RetryState retry_state = 1;\n}\n\nmessage WorkflowExecutionContinuedAsNewEventAttributes {\n    string new_execution_run_id = 1;\n    WorkflowType workflow_type = 2;\n    TaskList task_list = 3;\n    Payloads input = 4;\n    // workflow_execution_timeout is omitted as it shouldn'be overridden from within a workflow.\n    // Timeout of a single workflow run.\n    google.protobuf.Duration workflow_run_timeout = 5 [(gogoproto.stdduration) = true];\n    // Timeout of a single workflow task.\n    google.protobuf.Duration decision_task_timeout = 6 [(gogoproto.stdduration) = true];\n    int64 decision_task_completed_event_id = 7;\n    google.protobuf.Duration backoff_start_interval = 8 [(gogoproto.stdduration) = true];\n    ContinueAsNewInitiator initiator = 9;\n    Failure failure = 10;\n    Payloads last_completion_result = 11;\n    Header header = 12;\n    Memo memo = 13;\n    SearchAttributes search_attributes = 14;\n}\n\nmessage DecisionTaskScheduledEventAttributes {\n    TaskList task_list = 1;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration start_to_close_timeout = 2 [(gogoproto.stdduration) = true];\n    int32 attempt = 3;\n}\n\nmessage DecisionTaskStartedEventAttributes {\n    int64 scheduled_event_id = 1;\n    string identity = 2;\n    string request_id = 3;\n}\n\nmessage DecisionTaskCompletedEventAttributes {\n    int64 scheduled_event_id = 1;\n    int64 started_event_id = 2;\n    string identity = 3;\n    string binary_checksum = 4;\n}\n\nmessage DecisionTaskTimedOutEventAttributes {\n    int64 scheduled_event_id = 1;\n    int64 started_event_id = 2;\n    TimeoutType timeout_type = 3;\n}\n\nmessage DecisionTaskFailedEventAttributes {\n    int64 scheduled_event_id = 1;\n    int64 started_event_id = 2;\n    DecisionTaskFailedCause cause = 3;\n    Failure failure = 4;\n    string identity = 5;\n    // For reset workflow.\n    string base_run_id = 6;\n    string new_run_id = 7;\n    int64 fork_event_version = 8;\n    string binary_checksum = 9;\n}\n\nmessage ActivityTaskScheduledEventAttributes {\n    string activity_id = 1;\n    ActivityType activity_type = 2;\n    string domain = 3;\n    TaskList task_list = 4;\n    Header header = 5;\n    Payloads input = 6;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    // Indicates how long the caller is willing to wait for an activity completion.\n    // Limits for how long retries are happening. Either this or start_to_close_timeout_seconds must be specified.\n    google.protobuf.Duration schedule_to_close_timeout = 7 [(gogoproto.stdduration) = true];\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    // Limits time an activity task can stay in a task list before a worker picks it up.\n    // This timeout is always non retryable as all a retry would achieve is to put it back into the same list.\n    // Defaults to schedule_to_close_timeout_seconds or workflow execution timeout if not specified.\n    google.protobuf.Duration schedule_to_start_timeout = 8 [(gogoproto.stdduration) = true];\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    // Maximum time an activity is allowed to execute after a pick up by a worker.\n    // This timeout is always retryable. Either this or schedule_to_close_timeout_seconds must be specified.\n    google.protobuf.Duration start_to_close_timeout = 9 [(gogoproto.stdduration) = true];\n    // Maximum time between successful worker heartbeats.\n    google.protobuf.Duration heartbeat_timeout = 10 [(gogoproto.stdduration) = true];\n    int64 decision_task_completed_event_id = 11;\n    // Activities are provided by a default retry policy controlled through the service dynamic configuration.\n    // Retries are happening up to schedule_to_close_timeout.\n    // To disable retries set retry_policy.maximum_attempts to 1.\n    RetryPolicy retry_policy = 12;\n}\n\nmessage ActivityTaskStartedEventAttributes {\n    int64 scheduled_event_id = 1;\n    string identity = 2;\n    string request_id = 3;\n    int32 attempt = 4;\n    Failure last_failure = 5;\n}\n\nmessage ActivityTaskCompletedEventAttributes {\n    Payloads result = 1;\n    int64 scheduled_event_id = 2;\n    int64 started_event_id = 3;\n    string identity = 4;\n}\n\nmessage ActivityTaskFailedEventAttributes {\n    Failure failure = 1;\n    int64 scheduled_event_id = 2;\n    int64 started_event_id = 3;\n    string identity = 4;\n    RetryState retry_state = 5;\n}\n\nmessage ActivityTaskTimedOutEventAttributes {\n    // For retry activity, it may have a failure before timeout. It is stored as `cause` in `failure`.\n    Failure failure = 1;\n    int64 scheduled_event_id = 2;\n    int64 started_event_id = 3;\n    RetryState retry_state = 4;\n}\n\nmessage ActivityTaskCancelRequestedEventAttributes {\n    int64 scheduled_event_id = 1;\n    int64 decision_task_completed_event_id = 2;\n}\n\nmessage ActivityTaskCanceledEventAttributes {\n    Payloads details = 1;\n    int64 latest_cancel_requested_event_id = 2;\n    int64 scheduled_event_id = 3;\n    int64 started_event_id = 4;\n    string identity = 5;\n}\n\nmessage TimerStartedEventAttributes {\n    string timer_id = 1;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration start_to_fire_timeout = 2 [(gogoproto.stdduration) = true];\n    int64 decision_task_completed_event_id = 3;\n}\n\nmessage TimerFiredEventAttributes {\n    string timer_id = 1;\n    int64 started_event_id = 2;\n}\n\nmessage TimerCanceledEventAttributes {\n    string timer_id = 1;\n    int64 started_event_id = 2;\n    int64 decision_task_completed_event_id = 3;\n    string identity = 4;\n}\n\nmessage WorkflowExecutionCancelRequestedEventAttributes {\n    string cause = 1;\n    int64 external_initiated_event_id = 2;\n    WorkflowExecution external_workflow_execution = 3;\n    string identity = 4;\n}\n\nmessage WorkflowExecutionCanceledEventAttributes {\n    int64 decision_task_completed_event_id = 1;\n    Payloads details = 2;\n}\n\nmessage MarkerRecordedEventAttributes {\n    string marker_name = 1;\n    map<string, Payloads> details = 2;\n    int64 decision_task_completed_event_id = 3;\n    Header header = 4;\n    Failure failure = 5;\n}\n\nmessage WorkflowExecutionSignaledEventAttributes {\n    string signal_name = 1;\n    Payloads input = 2;\n    string identity = 3;\n}\n\nmessage WorkflowExecutionTerminatedEventAttributes {\n    string reason = 1;\n    Payloads details = 2;\n    string identity = 3;\n}\n\nmessage RequestCancelExternalWorkflowExecutionInitiatedEventAttributes {\n    int64 decision_task_completed_event_id = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    string control = 4;\n    bool child_workflow_only = 5;\n}\n\nmessage RequestCancelExternalWorkflowExecutionFailedEventAttributes {\n    CancelExternalWorkflowExecutionFailedCause cause = 1;\n    int64 decision_task_completed_event_id = 2;\n    string domain = 3;\n    WorkflowExecution workflow_execution = 4;\n    int64 initiated_event_id = 5;\n    string control = 6;\n}\n\nmessage ExternalWorkflowExecutionCancelRequestedEventAttributes {\n    int64 initiated_event_id = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n}\n\nmessage SignalExternalWorkflowExecutionInitiatedEventAttributes {\n    int64 decision_task_completed_event_id = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    string signal_name = 4;\n    Payloads input = 5;\n    string control = 6;\n    bool child_workflow_only = 7;\n}\n\nmessage SignalExternalWorkflowExecutionFailedEventAttributes {\n    SignalExternalWorkflowExecutionFailedCause cause = 1;\n    int64 decision_task_completed_event_id = 2;\n    string domain = 3;\n    WorkflowExecution workflow_execution = 4;\n    int64 initiated_event_id = 5;\n    string control = 6;\n}\n\nmessage ExternalWorkflowExecutionSignaledEventAttributes {\n    int64 initiated_event_id = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    string control = 4;\n}\n\nmessage UpsertWorkflowSearchAttributesEventAttributes {\n    int64 decision_task_completed_event_id = 1;\n    SearchAttributes search_attributes = 2;\n}\n\nmessage StartChildWorkflowExecutionInitiatedEventAttributes {\n    string domain = 1;\n    string workflow_id = 2;\n    WorkflowType workflow_type = 3;\n    TaskList task_list = 4;\n    Payloads input = 5;\n    // Total workflow execution timeout including retries and continue as new.\n    google.protobuf.Duration workflow_execution_timeout = 6 [(gogoproto.stdduration) = true];\n    // Timeout of a single workflow run.\n    google.protobuf.Duration workflow_run_timeout = 7 [(gogoproto.stdduration) = true];\n    // Timeout of a single workflow task.\n    google.protobuf.Duration decision_task_timeout = 8 [(gogoproto.stdduration) = true];\n    // Default: PARENT_CLOSE_POLICY_TERMINATE.\n    ParentClosePolicy parent_close_policy = 9;\n    string control = 10;\n    int64 decision_task_completed_event_id = 11;\n    // Default: WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE.\n    WorkflowIdReusePolicy workflow_id_reuse_policy = 12;\n    RetryPolicy retry_policy = 13;\n    string cron_schedule = 14;\n    Header header = 15;\n    Memo memo = 16;\n    SearchAttributes search_attributes = 17;\n}\n\nmessage StartChildWorkflowExecutionFailedEventAttributes {\n    string domain = 1;\n    string workflow_id = 2;\n    WorkflowType workflow_type = 3;\n    StartChildWorkflowExecutionFailedCause cause = 4;\n    string control = 5;\n    int64 initiated_event_id = 6;\n    int64 decision_task_completed_event_id = 7;\n}\n\nmessage ChildWorkflowExecutionStartedEventAttributes {\n    string domain = 1;\n    int64 initiated_event_id = 2;\n    WorkflowExecution workflow_execution = 3;\n    WorkflowType workflow_type = 4;\n    Header header = 5;\n}\n\nmessage ChildWorkflowExecutionCompletedEventAttributes {\n    Payloads result = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    WorkflowType workflow_type = 4;\n    int64 initiated_event_id = 5;\n    int64 started_event_id = 6;\n}\n\nmessage ChildWorkflowExecutionFailedEventAttributes {\n    Failure failure = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    WorkflowType workflow_type = 4;\n    int64 initiated_event_id = 5;\n    int64 started_event_id = 6;\n    RetryState retry_state = 7;\n}\n\nmessage ChildWorkflowExecutionCanceledEventAttributes {\n    Payloads details = 1;\n    string domain = 2;\n    WorkflowExecution workflow_execution = 3;\n    WorkflowType workflow_type = 4;\n    int64 initiated_event_id = 5;\n    int64 started_event_id = 6;\n}\n\nmessage ChildWorkflowExecutionTimedOutEventAttributes {\n    string domain = 1;\n    WorkflowExecution workflow_execution = 2;\n    WorkflowType workflow_type = 3;\n    int64 initiated_event_id = 4;\n    int64 started_event_id = 5;\n    RetryState retry_state = 6;\n}\n\nmessage ChildWorkflowExecutionTerminatedEventAttributes {\n    string domain = 1;\n    WorkflowExecution workflow_execution = 2;\n    WorkflowType workflow_type = 3;\n    int64 initiated_event_id = 4;\n    int64 started_event_id = 5;\n}\n\nmessage WorkflowExecution {\n    string workflow_id = 1;\n    string run_id = 2;\n}\n\nmessage WorkflowType {\n    string name = 1;\n}\n\nmessage Payloads {\n    repeated Payload payloads = 1;\n}\n\nmessage Payload {\n    map<string,bytes> metadata = 1;\n    bytes data = 2;\n}\n\nenum RetryState {\n    RETRY_STATE_UNSPECIFIED = 0;\n    RETRY_STATE_IN_PROGRESS = 1;\n    RETRY_STATE_NON_RETRYABLE_FAILURE = 2;\n    RETRY_STATE_TIMEOUT = 3;\n    RETRY_STATE_MAXIMUM_ATTEMPTS_REACHED = 4;\n    RETRY_STATE_RETRY_POLICY_NOT_SET = 5;\n    RETRY_STATE_INTERNAL_SERVER_ERROR = 6;\n    RETRY_STATE_CANCEL_REQUESTED = 7;\n}\n\nmessage ApplicationFailureInfo {\n    string type = 1;\n    bool non_retryable = 2;\n    Payloads details = 3;\n}\n\nmessage TimeoutFailureInfo {\n    TimeoutType timeout_type = 1;\n    Payloads last_heartbeat_details = 2;\n}\n\nmessage CanceledFailureInfo {\n    Payloads details = 1;\n}\n\nmessage TerminatedFailureInfo {\n}\n\nmessage ServerFailureInfo {\n    bool non_retryable = 1;\n}\n\nmessage ResetWorkflowFailureInfo {\n    Payloads last_heartbeat_details = 1;\n}\n\nmessage ActivityFailureInfo {\n    int64 scheduled_event_id = 1;\n    int64 started_event_id = 2;\n    string identity = 3;\n    ActivityType activity_type = 4;\n    string activity_id = 5;\n    RetryState retry_state = 6;\n}\n\nmessage ChildWorkflowExecutionFailureInfo {\n    string domain = 1;\n    WorkflowExecution workflow_execution = 2;\n    WorkflowType workflow_type = 3;\n    int64 initiated_event_id = 4;\n    int64 started_event_id = 5;\n    RetryState retry_state = 6;\n}\n\nmessage Failure {\n    string message = 1;\n    string source = 2;\n    string stack_trace = 3;\n    Failure cause = 4;\n    oneof failure_info {\n        ApplicationFailureInfo application_failure_info = 5;\n        TimeoutFailureInfo timeout_failure_info = 6;\n        CanceledFailureInfo canceled_failure_info = 7;\n        TerminatedFailureInfo terminated_failure_info = 8;\n        ServerFailureInfo server_failure_info = 9;\n        ResetWorkflowFailureInfo reset_workflow_failure_info = 10;\n        ActivityFailureInfo activity_failure_info = 11;\n        ChildWorkflowExecutionFailureInfo child_workflow_execution_failure_info = 12;\n    }\n}\n\nmessage ActivityType {\n    string name = 1;\n}\n\nenum TimeoutType {\n    TIMEOUT_TYPE_UNSPECIFIED = 0;\n    TIMEOUT_TYPE_START_TO_CLOSE = 1;\n    TIMEOUT_TYPE_SCHEDULE_TO_START = 2;\n    TIMEOUT_TYPE_SCHEDULE_TO_CLOSE = 3;\n    TIMEOUT_TYPE_HEARTBEAT = 4;\n}\n\nenum StartChildWorkflowExecutionFailedCause {\n    START_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_UNSPECIFIED = 0;\n    START_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_EXISTS = 1;\n}\n\nenum CancelExternalWorkflowExecutionFailedCause {\n    CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNSPECIFIED = 0;\n    CANCEL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_EXTERNAL_WORKFLOW_EXECUTION_NOT_FOUND = 1;\n}\n\nenum SignalExternalWorkflowExecutionFailedCause {\n    SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_UNSPECIFIED = 0;\n    SIGNAL_EXTERNAL_WORKFLOW_EXECUTION_FAILED_CAUSE_EXTERNAL_WORKFLOW_EXECUTION_NOT_FOUND = 1;\n}\n\nmessage SearchAttributes {\n    map<string, Payload> indexed_fields = 1;\n}\n\nmessage Memo {\n    map<string, Payload> fields = 1;\n}\n\nmessage Header {\n    map<string, Payload> fields = 1;\n}\n\nmessage RetryPolicy {\n    // Interval of the first retry. If retryBackoffCoefficient is 1.0 then it is used for all retries.\n    google.protobuf.Duration initial_interval = 1 [(gogoproto.stdduration) = true];\n    // Coefficient used to calculate the next retry interval.\n    // The next retry interval is previous interval multiplied by the coefficient.\n    // Must be 1 or larger.\n    double backoff_coefficient = 2;\n    // Maximum interval between retries. Exponential backoff leads to interval increase.\n    // This value is the cap of the increase. Default is 100x of the initial interval.\n    google.protobuf.Duration maximum_interval = 3 [(gogoproto.stdduration) = true];\n    // Maximum number of attempts. When exceeded the retries stop even if not expired yet.\n    // 1 disables retries. 0 means unlimited (up to the timeouts)\n    int32 maximum_attempts = 4;\n    // Non-Retryable errors types. Will stop retrying if error type matches this list.\n    repeated string non_retryable_error_types = 5;\n}\n\nenum WorkflowIdReusePolicy {\n    WORKFLOW_ID_REUSE_POLICY_UNSPECIFIED = 0;\n    // Allow start a workflow execution using the same workflow Id, when workflow not running.\n    WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE = 1;\n    // Allow start a workflow execution using the same workflow Id, when workflow not running, and the last execution close state is in\n    // [terminated, cancelled, timed out, failed].\n    WORKFLOW_ID_REUSE_POLICY_ALLOW_DUPLICATE_FAILED_ONLY = 2;\n    // Do not allow start a workflow execution using the same workflow Id at all.\n    WORKFLOW_ID_REUSE_POLICY_REJECT_DUPLICATE = 3;\n}\n\nenum ParentClosePolicy {\n    PARENT_CLOSE_POLICY_UNSPECIFIED = 0;\n    // Terminate means terminating the child workflow.\n    PARENT_CLOSE_POLICY_TERMINATE = 1;\n    // Abandon means not doing anything on the child workflow.\n    PARENT_CLOSE_POLICY_ABANDON = 2;\n    // Cancel means requesting cancellation on the child workflow.\n    PARENT_CLOSE_POLICY_REQUEST_CANCEL = 3;\n}\n\nmessage TaskList {\n    string name = 1;\n    // Default: TASK_LIST_KIND_NORMAL.\n    TaskListKind kind = 2;\n}\n\nmessage TaskListMetadata {\n    google.protobuf.DoubleValue max_tasks_per_second = 1;\n}\n\nmessage TaskListStatus {\n    int64 backlog_count_hint = 1;\n    int64 read_level = 2;\n    int64 ack_level = 3;\n    double rate_per_second = 4;\n    TaskIdBlock task_id_block = 5;\n}\n\nmessage TaskIdBlock {\n    int64 start_id = 1;\n    int64 end_id = 2;\n}\n\nmessage TaskListPartitionMetadata {\n    string key = 1;\n    string owner_host_name = 2;\n}\n\nmessage PollerInfo {\n    // Unix Nano\n    google.protobuf.Timestamp last_access_time = 1 [(gogoproto.stdtime) = true];\n    string identity = 2;\n    double rate_per_second = 3;\n}\n\nmessage StickyExecutionAttributes {\n    TaskList worker_task_list = 1;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration schedule_to_start_timeout = 2 [(gogoproto.stdduration) = true];\n}\n\nenum TaskListKind {\n    TASK_LIST_KIND_UNSPECIFIED = 0;\n    TASK_LIST_KIND_NORMAL = 1;\n    TASK_LIST_KIND_STICKY = 2;\n}\n\nenum TaskListType {\n    TASK_LIST_TYPE_UNSPECIFIED = 0;\n    // Workflow type of task list.\n    TASK_LIST_TYPE_WORKFLOW = 1;\n    // Activity type of task list.\n    TASK_LIST_TYPE_ACTIVITY = 2;\n}\n\nenum DecisionTaskFailedCause {\n    DECISION_TASK_FAILED_CAUSE_UNSPECIFIED = 0;\n    DECISION_TASK_FAILED_CAUSE_UNHANDLED_COMMAND = 1;\n    DECISION_TASK_FAILED_CAUSE_BAD_SCHEDULE_ACTIVITY_ATTRIBUTES = 2;\n    DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES = 3;\n    DECISION_TASK_FAILED_CAUSE_BAD_START_TIMER_ATTRIBUTES = 4;\n    DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_TIMER_ATTRIBUTES = 5;\n    DECISION_TASK_FAILED_CAUSE_BAD_RECORD_MARKER_ATTRIBUTES = 6;\n    DECISION_TASK_FAILED_CAUSE_BAD_COMPLETE_WORKFLOW_EXECUTION_ATTRIBUTES = 7;\n    DECISION_TASK_FAILED_CAUSE_BAD_FAIL_WORKFLOW_EXECUTION_ATTRIBUTES = 8;\n    DECISION_TASK_FAILED_CAUSE_BAD_CANCEL_WORKFLOW_EXECUTION_ATTRIBUTES = 9;\n    DECISION_TASK_FAILED_CAUSE_BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES = 10;\n    DECISION_TASK_FAILED_CAUSE_BAD_CONTINUE_AS_NEW_ATTRIBUTES = 11;\n    DECISION_TASK_FAILED_CAUSE_START_TIMER_DUPLICATE_ID = 12;\n    DECISION_TASK_FAILED_CAUSE_RESET_STICKY_TASK_LIST = 13;\n    DECISION_TASK_FAILED_CAUSE_WORKFLOW_WORKER_UNHANDLED_FAILURE = 14;\n    DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES = 15;\n    DECISION_TASK_FAILED_CAUSE_BAD_START_CHILD_EXECUTION_ATTRIBUTES = 16;\n    DECISION_TASK_FAILED_CAUSE_FORCE_CLOSE_COMMAND = 17;\n    DECISION_TASK_FAILED_CAUSE_FAILOVER_CLOSE_COMMAND = 18;\n    DECISION_TASK_FAILED_CAUSE_BAD_SIGNAL_INPUT_SIZE = 19;\n    DECISION_TASK_FAILED_CAUSE_RESET_WORKFLOW = 20;\n    DECISION_TASK_FAILED_CAUSE_BAD_BINARY = 21;\n    DECISION_TASK_FAILED_CAUSE_SCHEDULE_ACTIVITY_DUPLICATE_ID = 22;\n    DECISION_TASK_FAILED_CAUSE_BAD_SEARCH_ATTRIBUTES = 23;\n}\n\nenum ContinueAsNewInitiator {\n    CONTINUE_AS_NEW_INITIATOR_UNSPECIFIED = 0;\n    CONTINUE_AS_NEW_INITIATOR_WORKFLOW = 1;\n    CONTINUE_AS_NEW_INITIATOR_RETRY = 2;\n    CONTINUE_AS_NEW_INITIATOR_CRON_SCHEDULE = 3;\n}\n\nmessage ResetPoints {\n    repeated ResetPointInfo points = 1;\n}\n\nmessage ResetPointInfo {\n    string binary_checksum = 1;\n    string run_id = 2;\n    int64 first_decision_task_completed_id = 3;\n    google.protobuf.Timestamp create_time = 4 [(gogoproto.stdtime) = true];\n    // (-- api-linter: core::0214::resource-expiry=disabled\n    //     aip.dev/not-precedent: TTL is not defined for ResetPointInfo. --)\n    // The time that the run is deleted due to retention.\n    google.protobuf.Timestamp expire_time = 5 [(gogoproto.stdtime) = true];\n    // false if the reset point has pending childWFs/reqCancels/signalExternals.\n    bool resettable = 6;\n}\n\nmessage ChildExecutionInfo {\n    int64 version = 1;\n    int64 initiated_event_batch_id = 2;\n    int64 started_id = 3;\n    HistoryEvent initiated_event = 4;\n    string started_workflow_id = 5;\n    string started_run_id = 6;\n    HistoryEvent started_event = 7;\n    string create_request_id = 8;\n    string domain = 9;\n    string workflow_type_name = 10;\n    ParentClosePolicy parent_close_policy = 11;\n    int64 initiated_id = 12;\n}\n\nmessage WorkflowExecutionState {\n    string create_request_id = 1;\n    string run_id = 2;\n    WorkflowExecutionState state = 3;\n    WorkflowExecutionStatus status = 4;\n}\n\nmessage WorkflowExecutionInfo {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string parent_domain_id = 3;\n    string parent_workflow_id = 4;\n    string parent_run_id = 5;\n    int64 initiated_id = 6;\n    int64 completion_event_batch_id = 7;\n    HistoryEvent completion_event = 8;\n    string task_list = 9;\n    string workflow_type_name = 10;\n    google.protobuf.Duration workflow_execution_timeout = 11 [(gogoproto.stdduration) = true];\n    google.protobuf.Duration workflow_run_timeout = 12 [(gogoproto.stdduration) = true];\n    google.protobuf.Duration default_decision_task_timeout = 13 [(gogoproto.stdduration) = true];\n    int64 start_version = 14;\n    ReplicationData replication_data = 16;\n    int64 last_event_task_id = 17;\n    int64 last_first_event_id = 18;\n    int64 last_processed_event = 19;\n    google.protobuf.Timestamp start_time = 20 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp last_update_time = 21 [(gogoproto.stdtime) = true];\n    int64 decision_task_version = 22;\n    int64 decision_task_schedule_id = 23;\n    int64 decision_task_started_id = 24;\n    google.protobuf.Duration decision_task_timeout = 25 [(gogoproto.stdduration) = true];\n    int32 decision_task_attempt = 26;\n    google.protobuf.Timestamp decision_task_started_time = 27 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp decision_task_scheduled_time = 28 [(gogoproto.stdtime) = true];\n    bool cancel_requested = 29;\n    google.protobuf.Timestamp decision_task_original_scheduled_time = 30 [(gogoproto.stdtime) = true];\n    string decision_task_request_id = 31;\n    string cancel_request_id = 32;\n    string sticky_task_list = 33;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"to\" is used to indicate interval. --)\n    google.protobuf.Duration sticky_schedule_to_start_timeout = 34 [(gogoproto.stdduration) = true];\n    int32 retry_attempt = 35;\n    google.protobuf.Duration retry_initial_interval = 36 [(gogoproto.stdduration) = true];\n    google.protobuf.Duration retry_maximum_interval = 37 [(gogoproto.stdduration) = true];\n    int32 retry_maximum_attempts = 38;\n    double retry_backoff_coefficient = 39;\n    google.protobuf.Timestamp retry_expiration_time = 40 [(gogoproto.stdtime) = true];\n    repeated string retry_non_retryable_error_types = 41;\n    bool has_retry_policy = 42;\n    string cron_schedule = 43;\n    int32 event_store_version = 44;\n    bytes event_branch_token = 45;\n    int64 signal_count = 46;\n    int64 history_size = 47;\n    string client_library_version = 48;\n    string client_feature_version = 49;\n    string client_impl = 50;\n    ResetPoints auto_reset_points = 51;\n    map<string, Payload> search_attributes = 52;\n    map<string, Payload> memo = 53;\n    VersionHistories version_histories = 54;\n    string first_execution_run_id = 55;\n}\n\n// (-- api-linter: core::0216::synonyms=disabled\n//     aip.dev/not-precedent: There is WorkflowExecutionState already in another package. --)\nenum WorkflowExecutionStatus {\n    WORKFLOW_EXECUTION_STATUS_UNSPECIFIED = 0;\n    // Value 1 is hardcoded in SQL persistence.\n    WORKFLOW_EXECUTION_STATUS_RUNNING = 1;\n    WORKFLOW_EXECUTION_STATUS_COMPLETED = 2;\n    WORKFLOW_EXECUTION_STATUS_FAILED = 3;\n    WORKFLOW_EXECUTION_STATUS_CANCELED = 4;\n    WORKFLOW_EXECUTION_STATUS_TERMINATED = 5;\n    WORKFLOW_EXECUTION_STATUS_CONTINUED_AS_NEW = 6;\n    WORKFLOW_EXECUTION_STATUS_TIMED_OUT = 7;\n}\n\n// ReplicationData represents mutable state information for global domains.\n// This information is used by replication protocol when applying events from remote clusters\n// only used in cassandra\nmessage ReplicationData {\n    int64 last_write_event_id = 1;\n    map<string, ReplicationInfo> last_replication_info = 2;\n}\n\nmessage ReplicationInfo {\n    int64 version = 1;\n    int64 last_event_id = 2;\n}\n\n// VersionHistory contains the version history of a branch.\nmessage VersionHistory {\n    bytes branch_token = 1;\n    repeated VersionHistoryItem items = 2;\n}\n\n// VersionHistories contains all version histories from all branches.\nmessage VersionHistories {\n    int32 current_version_history_index = 1;\n    repeated VersionHistory histories = 2;\n}\n\n// VersionHistoryItem contains signal eventId and the corresponding version.\nmessage VersionHistoryItem {\n    int64 event_id = 1;\n    int64 version = 2;\n}\n\nmessage HistoryTreeInfo {\n    HistoryBranch branch_info = 1;\n    // For fork operation to prevent race condition of leaking event data when forking branches fail. Also can be used for clean up leaked data.\n    google.protobuf.Timestamp fork_time = 2 [(gogoproto.stdtime) = true];\n    // For lookup back to workflow during debugging, also background cleanup when fork operation cannot finish self cleanup due to crash.\n    string info = 3;\n}\n\n// For history persistence to serialize/deserialize branch details.\nmessage HistoryBranch {\n    string tree_id = 1;\n    string branch_id = 2;\n    repeated HistoryBranchRange ancestors = 3;\n}\n\n// HistoryBranchRange represents a piece of range for a branch.\nmessage HistoryBranchRange {\n    // BranchId of original branch forked from.\n    string branch_id = 1;\n    // Beginning node for the range, inclusive.\n    int64 begin_node_id = 2;\n    // Ending node for the range, exclusive.\n    int64 end_node_id = 3;\n}\n\nmessage ReplicationTask {\n    ReplicationTaskType task_type = 1;\n    int64 source_task_id = 2;\n    oneof attributes {\n        DomainTaskAttributes domain_task_attributes = 3;\n        // TODO: deprecate once NDC migration is done.\n        HistoryTaskAttributes history_task_attributes = 4;\n        SyncShardStatusTaskAttributes sync_shard_status_task_attributes = 5;\n        SyncActivityTaskAttributes sync_activity_task_attributes = 6;\n        // TODO: deprecate once kafka deprecation is done.\n        HistoryMetadataTaskAttributes history_metadata_task_attributes = 7;\n        HistoryTaskV2Attributes history_task_v2_attributes = 8;\n    }\n}\n\nenum ReplicationTaskType {\n    REPLICATION_TASK_TYPE_UNSPECIFIED = 0;\n    REPLICATION_TASK_TYPE_DOMAIN_TASK = 1;\n    REPLICATION_TASK_TYPE_HISTORY_TASK = 2;\n    REPLICATION_TASK_TYPE_SYNC_SHARD_STATUS_TASK = 3;\n    REPLICATION_TASK_TYPE_SYNC_ACTIVITY_TASK = 4;\n    REPLICATION_TASK_TYPE_HISTORY_METADATA_TASK = 5;\n    REPLICATION_TASK_TYPE_HISTORY_V2_TASK = 6;\n}\n\nmessage DomainTaskAttributes {\n    DomainOperation domain_operation = 1;\n    string id = 2;\n    DomainInfo info = 3;\n    DomainConfig config = 4;\n    DomainReplicationConfig replication_config = 5;\n    int64 config_version = 6;\n    int64 failover_version = 7;\n}\n\nenum DomainOperation {\n    DOMAIN_OPERATION_UNSPECIFIED = 0;\n    DOMAIN_OPERATION_CREATE = 1;\n    DOMAIN_OPERATION_UPDATE = 2;\n    DOMAIN_OPERATION_DELETE = 3;\n}\n\nmessage HistoryTaskAttributes {\n    repeated string target_clusters = 1;\n    string domain_id = 2;\n    string workflow_id = 3;\n    string run_id = 4;\n    int64 first_event_id = 5;\n    int64 next_event_id = 6;\n    int64 version = 7;\n    History history = 9;\n    History new_run_history = 10;\n    int32 new_run_event_store_version = 12;\n}\n\nmessage History {\n    repeated HistoryEvent events = 1;\n}\n\nmessage SyncShardStatusTaskAttributes {\n    string source_cluster = 1;\n    int64 shard_id = 2;\n    google.protobuf.Timestamp status_time = 3 [(gogoproto.stdtime) = true];\n}\n\nmessage SyncActivityTaskAttributes {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string run_id = 3;\n    int64 version = 4;\n    int64 scheduled_id = 5;\n    google.protobuf.Timestamp scheduled_time = 6 [(gogoproto.stdtime) = true];\n    int64 started_id = 7;\n    google.protobuf.Timestamp started_time = 8 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp last_heartbeat_time = 9 [(gogoproto.stdtime) = true];\n    Payloads details = 10;\n    int32 attempt = 11;\n    Failure last_failure = 12;\n    string last_worker_identity = 13;\n    VersionHistory version_history = 14;\n}\n\nmessage HistoryMetadataTaskAttributes {\n    repeated string target_clusters = 1;\n    string domain_id = 2;\n    string workflow_id = 3;\n    string run_id = 4;\n    int64 first_event_id = 5;\n    int64 next_event_id = 6;\n    int64 version = 7;\n}\n\nmessage HistoryTaskV2Attributes {\n    int64 task_id = 1;\n    string domain_id = 2;\n    string workflow_id = 3;\n    string run_id = 4;\n    repeated VersionHistoryItem version_history_items = 5;\n    DataBlob events = 6;\n    // New run events does not need version history since there is no prior events.\n    DataBlob new_run_events = 7;\n}\n\nmessage DataBlob {\n    EncodingType encoding_type = 1;\n    bytes data = 2;\n}\n\nenum EncodingType {\n    ENCODING_TYPE_UNSPECIFIED = 0;\n    ENCODING_TYPE_PROTO3 = 1;\n    ENCODING_TYPE_JSON = 2;\n}\n\nmessage QueueMetadataInfo {\n    map<string, int64> clusterAckLevels = 1;\n}\n\nmessage ReplicationTaskInfo {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string run_id = 3;\n    TaskType task_type = 4;\n    int64 version = 5;\n    int64 first_event_id = 6;\n    int64 next_event_id = 7;\n    int64 scheduled_id = 8;\n    int32 event_store_version = 9;\n    int32 new_run_event_store_version = 10;\n    bytes branch_token = 11;\n    bytes new_run_branch_token = 13;\n    int64 task_id = 15;\n}\n\nenum TaskType {\n    TASK_TYPE_UNSPECIFIED = 0;\n    TASK_TYPE_REPLICATION_HISTORY = 1;\n    TASK_TYPE_REPLICATION_SYNC_ACTIVITY = 2;\n    TASK_TYPE_TRANSFER_DECISION_TASK = 3;\n    TASK_TYPE_TRANSFER_ACTIVITY_TASK = 4;\n    TASK_TYPE_TRANSFER_CLOSE_EXECUTION = 5;\n    TASK_TYPE_TRANSFER_CANCEL_EXECUTION = 6;\n    TASK_TYPE_TRANSFER_START_CHILD_EXECUTION = 7;\n    TASK_TYPE_TRANSFER_SIGNAL_EXECUTION = 8;\n    TASK_TYPE_TRANSFER_RECORD_WORKFLOW_STARTED = 9;\n    TASK_TYPE_TRANSFER_RESET_WORKFLOW = 10;\n    TASK_TYPE_TRANSFER_UPSERT_WORKFLOW_SEARCH_ATTRIBUTES = 11;\n    TASK_TYPE_DECISION_TASK_TIMEOUT = 12;\n    TASK_TYPE_ACTIVITY_TIMEOUT = 13;\n    TASK_TYPE_USER_TIMER = 14;\n    TASK_TYPE_WORKFLOW_RUN_TIMEOUT = 15;\n    TASK_TYPE_DELETE_HISTORY_EVENT = 16;\n    TASK_TYPE_ACTIVITY_RETRY_TIMER = 17;\n    TASK_TYPE_WORKFLOW_BACKOFF_TIMER = 18;\n}\n\nmessage RequestCancelInfo {\n    int64 version = 1;\n    int64 initiated_event_batch_id = 2;\n    string cancel_request_id = 3;\n    int64 initiated_id = 4;\n}\n\nmessage ShardInfo {\n    int32 shard_id = 1;\n    int64 range_id = 2;\n    string owner = 3;\n    int64 replication_ack_level = 4;\n    int64 transfer_ack_level = 5;\n    // (-- api-linter: core::0140::prepositions=disabled\n    //     aip.dev/not-precedent: \"since\" is needed here. --)\n    int32 stolen_since_renew = 6;\n    google.protobuf.Timestamp update_time = 7 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp timer_ack_level_time = 8 [(gogoproto.stdtime) = true];\n    int64 domain_notification_version = 9;\n    map<string, int64> cluster_transfer_ack_level = 10;\n    map<string, google.protobuf.Timestamp> cluster_timer_ack_level = 11 [(gogoproto.stdtime) = true];\n    map<string, int64> cluster_replication_level = 12;\n    map<string, int64> replication_dlq_ack_level = 13;\n}\n\nmessage SignalInfo {\n    int64 version = 1;\n    int64 initiated_event_batch_id = 2;\n    string request_id = 3;\n    string name = 4;\n    Payloads input = 5;\n    string control = 6;\n    int64 initiated_id = 7;\n}\n\nmessage TaskListInfo {\n    string domain_id = 1;\n    string name = 2;\n    TaskListType task_type = 3;\n    TaskListKind kind = 4;\n    int64 ack_level = 5;\n    google.protobuf.Timestamp expiry_time = 6 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp last_update_time = 7 [(gogoproto.stdtime) = true];\n}\n\nmessage TaskInfo {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string run_id = 3;\n    int64 schedule_id = 4;\n    google.protobuf.Timestamp create_time = 5 [(gogoproto.stdtime) = true];\n    google.protobuf.Timestamp expiry_time = 6 [(gogoproto.stdtime) = true];\n}\n\nmessage AllocatedTaskInfo {\n    TaskInfo data = 1;\n    int64 task_id = 2;\n}\n\nmessage TimerInfo {\n    int64 version = 1;\n    int64 started_id = 2;\n    google.protobuf.Timestamp expiry_time = 3 [(gogoproto.stdtime) = true];\n    int64 task_status = 4;\n    // timerId serves the purpose of indicating whether a timer task is generated for this timer info.\n    string timer_id = 5;\n}\n\nmessage TimerTaskInfo {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string run_id = 3;\n    TaskType task_type = 4;\n    TimeoutType timeout_type = 5;\n    WorkflowBackoffType workflow_backoff_type = 6;\n    int64 version = 7;\n    int32 schedule_attempt = 8;\n    int64 event_id = 9;\n    int64 task_id = 10;\n    google.protobuf.Timestamp task_timestamp = 11 [(gogoproto.stdtime) = true];\n}\n\nmessage TransferTaskInfo {\n    string domain_id = 1;\n    string workflow_id = 2;\n    string run_id = 3;\n    TaskType task_type = 4;\n    string target_domain_id = 5;\n    string target_workflow_id = 6;\n    string target_run_id = 7;\n    string task_list = 8;\n    bool target_child_workflow_only = 9;\n    int64 schedule_id = 10;\n    int64 version = 11;\n    int64 task_id = 12;\n    google.protobuf.Timestamp task_timestamp = 13 [(gogoproto.stdtime) = true];\n    bool record_visibility = 14;\n}\n\nenum WorkflowBackoffType {\n    WORKFLOW_BACKOFF_TYPE_UNSPECIFIED = 0;\n    WORKFLOW_BACKOFF_TYPE_RETRY = 1;\n    WORKFLOW_BACKOFF_TYPE_CRON = 2;\n}\n"
  },
  {
    "path": "revive.toml",
    "content": "# config for https://github.com/mgechev/revive\nignoreGeneratedHeader = false\nseverity = \"error\"\nconfidence = 0.8\nerrorCode = 1\nwarningCode = 0\n\n[directive.specify-disable-reason]\n    severity = \"error\"\n\n#### roughly what golint does.  probably only disable noisy ones.\n\n[rule.blank-imports]\n[rule.context-as-argument]\n[rule.context-keys-type]\n[rule.dot-imports]\n[rule.error-naming]\n[rule.error-return]\n[rule.error-strings]\n[rule.errorf]\n# [rule.exported] # disabled due to lack of value / encouraging bad habits\n[rule.if-return]\n[rule.increment-decrement]\n[rule.indent-error-flow]\n# Disabled because we have 158 packages that need package comments; we could instead add ignore\n# directives for existing packages and require it for new packages.\n#[rule.package-comments]\n[rule.range]\n[rule.receiver-naming]\n[rule.time-naming]\n[rule.unexported-return]\n[rule.var-declaration]\n[rule.var-naming]\n\n#### higher value stuff\n\n# this is basically errcheck, warns on errs that are not checked.\n# strongly desired, but disabled due to 300 failures (to be tackled incrementally).\n# [rule.unhandled-error]\n\n# general defer gotchas.\n#\n# in particular: \"recover\" warns about unsafe use of recover().\n# this has caught bugs that can allow crashes while seemingly safe, and are *extremely* hard to catch in review.\n#\n# the arguments are excluding only \"call-chain\", which would disallow `defer someFn(...)()` which is both useful and in use.\n[rule.defer]\narguments=[[\"loop\",\"method-call\",\"recover\",\"return\", \"immediate-recover\"]]\n\n# string(int) is almost always a bug.\n# go vet considers this a fatal error, but only in 1.15 or newer, and go.mod currently targets 1.13\n[rule.string-of-int]\n\n#### added because we currently have zero violations, and they seem decent enough to retain\n\n[rule.atomic] # correct use of sync code, important\n[rule.call-to-gc] # beneficial\n[rule.constant-logical-expr] # minor code simplifier\n[rule.identical-branches] # code simplifier / failures are pretty dubious\n[rule.modifies-parameter] # beneficial\n[rule.modifies-value-receiver] # probably beneficial, prevents subtle bugs\n[rule.range-val-address] # beneficial\n[rule.range-val-in-closure] # beneficial\n[rule.unconditional-recursion] # probably a good idea\n[rule.unreachable-code] # code simplifier\n[rule.waitgroup-by-value] # correct use of sync code, important\n\n#### unused utilities\n\n# [rule.file-header] # could possibly replace `copyright -verifyOnly`?\n# [rule.imports-blacklist] # simple way to ban imports / enforce wrappers, likely useful\n\n#### disabled but maybe desirable\n\n# [rule.bare-return] # probably beneficial as it's slightly error-prone, but 2,000 failures\n# [rule.bool-literal-in-expr] # minor code simplifier, few failures\n# [rule.confusing-results] # maybe beneficial, only a few failures\n# [rule.deep-exit] # probably a good idea in most code, some failures, but not trivial to adopt\n# [rule.duplicated-imports] # minor, but may be worthwhile.  failures are weird but harmless\n# [rule.early-return] # minor code simplifier, a handful of failures\n# [rule.get-return] # existing failures are intentional + desirable, but in principle it's a fine idea\n# [rule.import-shadowing] # probably beneficial, but 750 failures\n# [rule.redefines-builtin-id] # probably beneficial, few failures\n# [rule.struct-tag] # probably beneficial, a few failures\n# [rule.superfluous-else] # minor code simplifier, a few failures\n# [rule.unexported-naming] # probably beneficial, but 300 failures\n# [rule.unused-parameter] # minor code simplifier / clarifier, but 250 failures\n# [rule.unused-receiver] # minor code simplifier / clarifier, but 500 failures\n\n#### probably undesirable\n\n# [rule.add-constant] # extremely noisy.  18,000 failures, overwhelmingly for tests or 0/1 which seem totally fine\n# [rule.argument-limit] # too arbitrary\n# [rule.cognitive-complexity] # dubious value, but possibly interesting\n# [rule.confusing-naming] # dubious value, ~50 failures\n# [rule.cyclomatic] # dubious value, but possibly interesting\n# [rule.empty-block] # easily noticed in code review, but also warns on documented no-op branches, which seem fine\n# [rule.empty-lines] # low value, many failures\n# [rule.flag-parameter] # interesting, but very noisy\n# [rule.function-result-limit] # too arbitrary, easily noticed in code review\n# [rule.line-length-limit] # too arbitrary\n# [rule.max-public-structs] # too arbitrary\n# [rule.unnecessary-stmt] # dubious value\n"
  },
  {
    "path": "schema/cassandra/README.md",
    "content": "What\n----\nThis directory contains the cassandra schema for every keyspace that cadence owns. The directory structure is as follows\n\n```\n./schema\n   - visibility/           -- Contains schema for visibility data models\n   - cadence/              -- Contains schema for default data models\n        - keyspace.cql     -- Contains the keyspace definition\n        - schema.cql       -- Contains the latest & greatest snapshot of the schema for the keyspace\n        - versioned\n             - v0.1/\n             - v0.2/       -- One directory per schema version change\n             - v1.0/\n                - manifest.json   -- json file describing the change\n                - changes.cql     -- changes in this version, only [CREATE, ALTER] commands are allowed\n```\n\nHow\n---\n\nQ: How do I update existing schema ?\n* Add your changes to schema.cql\n* Create a new schema version directory under ./schema/keyspace/versioned/vx.x\n  * Add a manifest.json\n  * Add your changes in a cql file\n* Update the unit test within ./tools/common/schema/updatetask_test.go `TestReadSchemaDirFromEmbeddings` with your version x.x\n* Once you are done with these use the ./cadence-cassandra-tool to update the schema\n\nQ: What's the format of manifest.json\n\nExample below:\n* MinCompatibleVersion is the minimum schema version that your code can handle\n* SchemaUpdateCqlFiles are list of .cql files containg your create/alter commands\n\n\n```\n{\n    \"CurrVersion\": \"0.1\",\n    \"MinCompatibleVersion\": \"0.1\",\n    \"Description\": \"base version of schema\",\n    \"SchemaUpdateCqlFiles\": [\n        \"base.cql\"\n    ]\n}\n```"
  },
  {
    "path": "schema/cassandra/cadence/keyspace.cql",
    "content": "CREATE KEYSPACE IF NOT EXISTS cadence WITH replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 1};\n"
  },
  {
    "path": "schema/cassandra/cadence/schema.cql",
    "content": "CREATE TYPE shard (\n  shard_id                          int,\n  owner                             text, -- Host identifier processing the shard\n  -- Range identifier used for generating ack ids for tasks within shard.\n  -- Also used for optimistic concurrency and all writes to a shard are conditional on this value.\n  range_id                          bigint, -- TO BE DEPRECATED, IN FAVOR OF range_id column in executions table\n  -- This field keeps track of number of times owner for a shard changes before updating range_id or ack_levels\n  stolen_since_renew                int,\n  updated_at                        timestamp,\n  replication_ack_level             bigint,\n  transfer_ack_level                bigint, -- TO BE DEPRECATED, IN FAVOR OF cluster_transfer_ack_level\n  timer_ack_level                   timestamp, -- TO BE DEPRECATED, IN FAVOR OF cluster_timer_ack_level\n  -- Mapping of cluster to corresponding transfer ack level\n  cluster_transfer_ack_level        map<text, bigint>, -- TO BE DEPRECATED, IN FAVOR OF transfer_processing_queue_states\n  -- Mapping of cluster to corresponding timer ack level\n  cluster_timer_ack_level           map<text, timestamp>, -- TO BE DEPRECATED, IN FAVOR OF timer_processing_queue_states\n  domain_notification_version       bigint, -- the global domain change version this shard is aware of\n  -- Mapping of cluster to corresponding list of transfer queue processing states\n  transfer_processing_queue_states  blob,\n  transfer_processing_queue_states_encoding  text,\n  -- Mapping of cluster to corresponding list of cross-cluster queue processing states\n  cross_cluster_processing_queue_states blob,\n  cross_cluster_processing_queue_states_encoding text,\n  -- Mapping of cluster to corresponding list of timer queue processing states\n  timer_processing_queue_states     blob,\n  timer_processing_queue_states_encoding     text,\n  -- Mapping of (remote) cluster to corresponding replication level (last replicated task_id)\n  cluster_replication_level         map<text, bigint>,\n  -- Mapping of (remote) cluster to corresponding replication DLQ ack level (last replicated task_id)\n  replication_dlq_ack_level         map<text, bigint>,\n  -- Data blob of pending failover markers\n  pending_failover_markers          blob,\n  pending_failover_markers_encoding text\n\n);\n\n--- Workflow execution and mutable state ---\nCREATE TYPE workflow_execution (\n  domain_id                        uuid,\n  workflow_id                      text,\n  run_id                           uuid,\n  first_run_id                     uuid,   -- Run ID of the first run of a ContinuedAsNew workflow\n  parent_domain_id                 uuid,   -- Domain ID of parent workflow which started the workflow execution\n  parent_workflow_id               text,   -- ID of parent workflow which started the workflow execution\n  parent_run_id                    uuid,   -- RunID of parent workflow which started the workflow execution\n  initiated_id                     bigint, -- Initiated event ID of parent workflow which started this execution\n  completion_event_batch_id        bigint,\n  completion_event                 blob,   -- Completion event used to communicate result to parent workflow execution\n  completion_event_data_encoding   text, -- Protocol used for history serialization\n  task_list                        text,\n  workflow_type_name               text,\n  workflow_timeout                 int,  -- Workflow ExecutionStartToCloseTimeoutSeconds\n  decision_task_timeout            int,  -- decision start to close timeout\n  execution_context                blob,\n  state                            int,  -- enum WorkflowState {Created, Running, Completed}\n  close_status                     int,  -- enum WorkflowCloseStatus {None, Completed, Failed, Canceled, Terminated, ContinuedAsNew, TimedOut}\n  last_processed_event             bigint,\n  start_time                       timestamp,\n  last_updated_time                timestamp,\n  create_request_id                uuid,\n  decision_version                 bigint,\n  decision_schedule_id             bigint,\n  decision_started_id              bigint,\n  decision_request_id              text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  decision_timeout                 int,\n  decision_attempt                 bigint,\n  decision_timestamp               bigint,  -- this is decision started time\n  decision_scheduled_timestamp     bigint,   -- this is decision scheduled time\n  decision_original_scheduled_timestamp     bigint,   -- this is scheduled time of the first decision during heartbeat\n  cancel_requested                 boolean,\n  cancel_request_id                text,\n  sticky_task_list                 text,   -- sticky worker task list\n  sticky_schedule_to_start_timeout int,\n  client_library_version           text,\n  client_feature_version           text,\n  client_impl                      text,\n  attempt                          int,    -- starting from 0 (for initial non-retry)\n  has_retry_policy                 boolean,-- If there is a retry policy\n  init_interval                    int,    -- initial retry interval, in seconds\n  backoff_coefficient              double,\n  max_interval                     int,    -- max retry interval in seconds\n  expiration_time                  timestamp, -- retry expiration time\n  max_attempts                     int,    -- max number of attempts including initial non-retry attempt\n  non_retriable_errors             list<text>,\n  event_store_version              int, -- indicates which version of events persistence is using\n  signal_count                     int,\n  branch_token                     blob,\n  history_size                     bigint,\n  last_first_event_id              bigint,\n  next_event_id                    bigint,\n  cron_schedule                    text,\n  expiration_seconds               int,    -- retry expiration duration in seconds\n  last_event_task_id               bigint,\n  auto_reset_points                blob, -- the resetting points for auto-reset feature\n  auto_reset_points_encoding       text, -- encoding for auto_reset_points_data\n  search_attributes                map<text, blob>,\n  memo                             map<text, blob>,\n  partition_config                 map<text, text>,\n  cron_overlap_policy              int, -- enum CronOverlapPolicy {Skip, BufferOne}\n  task_list_kind                   int, -- enum TaskListKind {Normal, Sticky, Ephemeral},\n  active_cluster_selection_policy blob, -- active cluster selection policy applicable to active-active domains\n  active_cluster_selection_policy_encoding text, -- encoding for active_cluster_selection_policy\n);\n\n-- Replication information for each cluster\nCREATE TYPE replication_info (\n  version       bigint,\n  last_event_id bigint,\n);\n\n-- This is used to store replication information for a workflow execution\nCREATE TYPE replication_state (\n  current_version                  bigint, -- current version for domain, incremented on failover\n  start_version                    bigint, -- version of domain when the workflow execution was started\n  last_write_version               bigint, -- version of domain when the last event was written to history\n  last_write_event_id              bigint, -- last written event id for a given version\n  last_replication_info            map<text, frozen<replication_info>>, -- information about replication events from other clusters\n);\n\n-- TODO: Remove fields that are left over from activity and workflow tasks.\nCREATE TYPE transfer_task (\n  domain_id                  uuid,         -- The domain ID that this transfer task belongs to\n  workflow_id                text,         -- The workflow ID that this transfer task belongs to\n  run_id                     uuid,         -- The run ID that this transfer task belongs to\n  task_id                    bigint,\n  visibility_ts              timestamp,    -- The timestamp when the transfer task is generated\n  target_domain_id           uuid,         -- The external domain ID that this transfer task is doing work for.\n  target_domain_ids          set<uuid>,    -- The external domain ID that this transfer task is doing work for.\n  target_workflow_id         text,         -- The external workflow ID that this transfer task is doing work for.\n  target_run_id              uuid,         -- The external run ID that this transfer task is doing work for.\n  target_child_workflow_only boolean,      -- The whether target child workflow only.\n  task_list                  text,\n  type                       int,          -- enum TaskType For local: {Decision, Activity, CloseExecution, CancelExecution, StartChildExecution, SignalExecution, RecordWorkflowStarted, ResetWorkflow, UpsertWorkflowSearchAttributes}, or for crossCluster {StartChildExecution, CancelExecution, SignalExecution}\n  schedule_id                bigint,\n  version                    bigint,       -- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\n  record_visibility          boolean,      -- indicates whether or not to create a visibility record\n  original_task_list         text,         -- the original task list of the task if the task is a sticky decision task\n  original_task_list_kind    int,          -- enum TaskListKind {Normal, Sticky, Ephemeral},\n);\n\nCREATE TYPE replication_task (\n  domain_id                  uuid,   -- The domain ID that this replication task belongs to\n  workflow_id                text,   -- The workflow ID that this replication task belongs to\n  run_id                     uuid,   -- The run ID that this replication task belongs to\n  task_id                    bigint,\n  type                       int,    -- enum TaskType {History, SyncActivity, FailoverMarker}\n  first_event_id             bigint,  -- Used by ReplicationTask to set the first event ID of the applied transaction\n  next_event_id              bigint,  -- Used by ReplicationTask to set the next event ID of the applied transaction\n  version                    bigint,  -- Used by ReplicationTask to set the failover version of the applied transaction\n  last_replication_info      map<text, frozen<replication_info>>, -- Used by replication task to snapshot replication information when the transaction was applied\n  scheduled_id               bigint, -- Used by ReplicationTask to sync activity info\n  event_store_version        int, -- indicates which version of event store to query\n  branch_token               blob, -- if eventV2, then query with this token\n  new_run_event_store_version        int, -- indicates which version of event store to query for new run(continueAsNew)\n  new_run_branch_token               blob, -- if eventV2, then query with this token for new run(continueAsNew)\n  reset_workflow             boolean, -- whether the task is for resetWorkflowExecution\n  created_time               bigint, -- task creation timestamp\n);\n\nCREATE TYPE timer_task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  visibility_ts    timestamp,\n  task_id          bigint,\n  type             int,  -- enum TaskType {DecisionTaskTimeout, ActivityTaskTimeout, UserTimer}\n  timeout_type     int, -- enum TimeoutType in IDL {START_TO_CLOSE, SCHEDULE_TO_START, SCHEDULE_TO_CLOSE, HEARTBEAT}\n  event_id         bigint, -- Corresponds to event ID in history that is responsible for this timer.\n  schedule_attempt bigint, -- Used to retry failed decision tasks using mutable state\n  version          bigint, -- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\n  task_list        text, -- the task list associated with the timer task\n);\n\n-- Workflow activity in progress mutable state\nCREATE TYPE activity_info (\n  version                   bigint,\n  schedule_id               bigint,\n  scheduled_event_batch_id  bigint,\n  scheduled_event           blob,  -- deprecated\n  scheduled_time            timestamp,\n  started_id                bigint,\n  started_event             blob,\n  started_time              timestamp,\n  activity_id               text,    -- Client generated unique ID for the activity.\n  request_id                text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  details                   blob,\n  schedule_to_start_timeout int,\n  schedule_to_close_timeout int,\n  start_to_close_timeout    int,\n  heart_beat_timeout        int,\n  cancel_requested          boolean, -- If a cancel request is made to cancel the activity in progress.\n  cancel_request_id         bigint,  -- Event ID that identifies the cancel request.\n  last_hb_updated_time      timestamp, -- Last time the heartbeat is received.\n  timer_task_status         int,    -- Indicates whether timers are created for this activity.\n  attempt                   int,    -- starting from 0 (for initial non-retry)\n  task_list                 text,\n  started_identity          text,   -- last started poller's identity\n  has_retry_policy          boolean,-- If there is a retry policy\n  init_interval             int,    -- initial retry interval, in seconds\n  backoff_coefficient       double,\n  max_interval              int,    -- max retry interval in seconds\n  expiration_time           timestamp, -- retry expiration time\n  max_attempts              int,    -- max number of attempts including initial non-retry attempt\n  non_retriable_errors      list<text>,\n  last_failure_reason       text,\n  last_worker_identity      text, -- Worker that returns the last failure reason\n  last_failure_details      blob,\n  event_data_encoding       text, -- Protocol used for history serialization\n  task_list_kind            int, -- enum TaskListKind {Normal, Sticky, Ephemeral},\n);\n\n-- User timer details\nCREATE TYPE timer_info (\n  version       bigint,\n  timer_id      text,      -- User defined timer ID\n  started_id    bigint,    -- The event ID corresponding to timer started.\n  expiry_time   timestamp, -- Timestamp at which this timer expires or fires\n  -- task_id is a misleading variable, it actually serves\n  -- the purpose of indicating whether a timer task is\n  -- generated for this timer info\n  task_id       bigint,\n);\n\n-- Child execution in progress mutable state\nCREATE TYPE child_execution_info (\n  version                   bigint,\n  initiated_id              bigint,\n  initiated_event_batch_id  bigint,\n  initiated_event           blob,\n  started_id                bigint,\n  started_workflow_id       text,\n  started_run_id            uuid,\n  started_event             blob, -- deprecated\n  create_request_id         uuid,\n  event_data_encoding       text, -- Protocol used for history serialization\n  domain_id                 uuid,\n  domain_name               text, -- deprecated\n  workflow_type_name        text,\n  parent_close_policy       int,\n);\n\n-- External workflow cancellation in progress mutable state\nCREATE TYPE request_cancel_info (\n  version                   bigint,\n  initiated_event_batch_id  bigint,\n  initiated_id              bigint,\n  cancel_request_id         text,\n);\n\n-- External workflow signal in progress mutable state\nCREATE TYPE signal_info (\n  version                   bigint,\n  initiated_event_batch_id  bigint,\n  initiated_id              bigint,\n  signal_request_id         uuid,\n  signal_name               text,\n  input                     blob,\n  control                   blob,\n);\n\n-- Activity or workflow task in a task list\nCREATE TYPE task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  schedule_id      bigint,\n  created_time     timestamp,\n  partition_config map<text, text>\n);\n\nCREATE TYPE task_list_partition (\n    isolation_groups set<text>\n);\n\nCREATE TYPE task_list_partition_config (\n  version              bigint,\n  num_read_partitions  int,\n  num_write_partitions int,\n  read_partitions map<int, frozen<task_list_partition>>,\n  write_partitions map<int, frozen<task_list_partition>>\n);\n\n\nCREATE TYPE task_list (\n  domain_id        uuid,\n  name             text,\n  type             int, -- enum TaskRowType {ActivityTask, DecisionTask}\n  ack_level        bigint, -- task_id of the last acknowledged message\n  kind             int, -- enum TaskListKind {Normal, Sticky}\n  last_updated     timestamp,\n  adaptive_partition_config frozen<task_list_partition_config>\n);\n\nCREATE TYPE domain (\n  id          uuid,\n  name        text,\n  status      int, -- enum DomainStatus {Registered, Deprecated, Deleted}\n  description text,\n  data        map<text,text>, -- Used for customized domain information, key values pair\n  owner_email text,\n);\n\nCREATE TYPE domain_config (\n  retention   int,\n  emit_metric boolean,\n  archival_bucket text, -- deprecated, use the two uri fields below\n  archival_status int, -- deprecated, use the two status fields below\n  history_archival_status int,\n  history_archival_uri text,\n  visibility_archival_status int,\n  visibility_archival_uri text,\n  bad_binaries    blob,\n  bad_binaries_encoding blob,\n  isolation_groups blob,\n  isolation_groups_encoding text,\n  async_workflow_config blob,\n  async_workflow_config_encoding text,\n);\n\nCREATE TYPE cluster_replication_config (\n  cluster_name text,\n);\n\nCREATE TYPE domain_replication_config (\n  active_cluster_name text,\n  clusters            list<frozen<cluster_replication_config>>,\n  active_clusters_config blob,\n  active_clusters_config_encoding text,\n);\n\nCREATE TYPE serialized_event_batch (\n  encoding_type text,\n  version       int,\n  data          blob,\n);\n\n-- Storage for out of order replication tasks for an execution\nCREATE TYPE buffered_replication_task_info (\n  first_event_id  bigint,\n  next_event_id   bigint,\n  version         bigint,\n  history         frozen<serialized_event_batch>,\n  new_run_history frozen<serialized_event_batch>,\n  event_store_version                int, -- indicates which version of event store to query\n  new_run_event_store_version        int, -- indicates which version of event store to query for new run(continueAsNew)\n);\n\n-- for history v2 events\nCREATE TYPE branch_range (\n  branch_id   uuid,\n  end_node_id bigint, -- exclusive node_id to represent the stopping point for this range\n);\n\nCREATE TYPE checksum (\n  version int,  -- version of the payload used to generate checksum\n  flavor  int,  -- type of checksum e.g crc32OverThrift\n  value   blob, -- checksum bytes\n);\n\nCREATE TABLE executions (\n  shard_id                       int,\n  type                           int, -- enum RowType { Shard, Execution, TransferTask, TimerTask, ReplicationTask, CrossClusterTask}\n  domain_id                      uuid,\n  workflow_id                    text,\n  run_id                         uuid,\n  current_run_id                 uuid,\n  visibility_ts                  timestamp, -- unique identifier for timer tasks for an execution\n  task_id                        bigint, -- unique identifier for transfer and timer tasks for an execution\n  data                           blob, -- this column will be used by Shard, Execution, TransferTask, TimerTask, ReplicationTask row types\n  data_encoding                  text,\n  shard                          frozen<shard>,\n  execution                      frozen<workflow_execution>,\n  transfer                       frozen<transfer_task>,\n  cross_cluster                  frozen<transfer_task>, -- reuse the transfer_task type\n  replication                    frozen<replication_task>,\n  timer                          frozen<timer_task>,\n  next_event_id                  bigint,  -- This is needed to make conditional updates on session history\n  range_id                       bigint, -- Increasing sequence identifier for transfer queue, checkpointed into shard info\n  activity_map                   map<bigint, frozen<activity_info>>,\n  timer_map                      map<text, frozen<timer_info>>,\n  child_executions_map           map<bigint, frozen<child_execution_info>>,\n  request_cancel_map             map<bigint, frozen<request_cancel_info>>,\n  signal_map                     map<bigint, frozen<signal_info>>,\n  signal_requested               set<uuid>,\n  buffered_events_list           list<frozen<serialized_event_batch>>,\n  replication_state              frozen<replication_state>, -- Replication information part of mutable state\n  buffered_replication_tasks_map map<bigint, frozen<buffered_replication_task_info>>,\n  workflow_last_write_version    bigint,\n  workflow_state                 int,\n  version_histories              blob, -- the metadata of history branching\n  version_histories_encoding     text,\n  checksum                       frozen<checksum>,\n  created_time                   timestamp,\n  last_updated_time              timestamp,\n  PRIMARY KEY  (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE history_node (\n  tree_id           uuid,\n  branch_id         uuid,\n  node_id           bigint, -- node_id: first eventID in a batch of events\n  txn_id            bigint, -- for override the same node_id: bigger txn_id wins\n  data                blob, -- Batch of workflow execution history events as a blob\n  data_encoding       text, -- Protocol used for history serialization\n  created_time      timestamp,\n  PRIMARY KEY ((tree_id), branch_id, node_id, txn_id )\n  ) WITH CLUSTERING ORDER BY (branch_id ASC, node_id ASC, txn_id DESC)\n    AND COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n    };\n\nCREATE TABLE history_tree (\n  tree_id           uuid,\n  branch_id         uuid,\n  ancestors         list<frozen<branch_range>>,\n  fork_time         timestamp, -- For fork operation to prevent race condition to leak event data when forking branches\n  info              text, -- For background cleanup when fork operation cannot finish self cleanup due to crash\n  created_time      timestamp,\n  PRIMARY KEY ((tree_id), branch_id )\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\n-- Stores activity or workflow tasks\nCREATE TABLE tasks (\n  domain_id        uuid,\n  task_list_name   text,\n  task_list_type   int, -- enum TaskListType {ActivityTask, DecisionTask}\n  type             int, -- enum rowType {Task, TaskList}\n  task_id          bigint,  -- unique identifier for tasks, monotonically increasing\n  range_id         bigint, -- Used to ensure that only one process can write to the table\n  task             frozen<task>,\n  task_list        frozen<task_list>,\n  created_time       timestamp,\n  last_updated_time  timestamp,\n  PRIMARY KEY ((domain_id, task_list_name, task_list_type), type, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\n-- this table is only used for storage of mapping of domain uuid to domain name\nCREATE TABLE domains (\n  id     uuid,\n  domain frozen<domain>,\n  config frozen<domain_config>,\n  created_time       timestamp,\n  PRIMARY KEY (id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE domains_by_name_v2 (\n  domains_partition             int,\n  name                          text,\n  domain                        frozen<domain>,\n  config                        frozen<domain_config>,\n  replication_config            frozen<domain_replication_config>, -- indicating active cluster and standby cluster used for replication\n  is_global_domain              boolean, -- indicating whether a domain is a global domain\n  config_version                bigint, -- indicating the version of domain config, excluding the failover / change of active cluster name\n  failover_version              bigint, -- indicating the version of active domain only, used for domain failover\n  failover_notification_version bigint, -- indicating the last change related to domain failover\n  previous_failover_version     bigint, -- indicating the previouse failover version for graceful failover\n  failover_end_time             bigint, -- indicating domain failover state\n  last_updated_time             bigint, -- indicating the domain last update timestamp\n  notification_version          bigint,\n  created_time                  timestamp,\n  PRIMARY KEY (domains_partition, name)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n\nINSERT INTO domains_by_name_v2 (\n   domains_partition,\n   name,\n   domain,\n   config,\n   is_global_domain,\n   config_version,\n   failover_version,\n   failover_notification_version,\n   notification_version\n) VALUES (\n   0,\n   'cadence-system',\n   {\n       id: 32049b68-7872-4094-8e63-d0dd59896a83,\n       name: 'cadence-system',\n       description: 'cadence system workflow domain',\n       owner_email: 'cadence-dev-group@uber.com'\n   },\n   {\n       retention:3,\n       emit_metric:False\n   },\n   False,\n   0,\n   -24,\n   -1,\n   -1\n) IF NOT EXISTS;\n\nINSERT INTO domains (\n   id,\n   domain\n) VALUES (\n   32049b68-7872-4094-8e63-d0dd59896a83,\n   {\n       name: 'cadence-system'\n   }\n) IF NOT EXISTS;\n\nCREATE TABLE queue (\n  queue_type      int,\n  message_id      bigint,\n  message_payload blob,\n  created_time    timestamp,\n  PRIMARY KEY  (queue_type, message_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE queue_metadata (\n  queue_type        int,\n  cluster_ack_level map<text, bigint>,\n  version           bigint,\n  created_time      timestamp,\n  last_updated_time timestamp,\nPRIMARY KEY (queue_type)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n\nCREATE TABLE cluster_config (\n  row_type int,\n  version int,\n  timestamp timestamp,\n  values blob,\n  encoding text,\nPRIMARY KEY (row_type, version)\n) WITH CLUSTERING ORDER BY (version DESC);\n\nCREATE TABLE domain_audit_log (\n    domain_id uuid,\n    event_id uuid, -- event_id is the unique identifier for this change to the domain\n\n    state_before blob, -- state_before stores the domain state before the request\n    state_before_encoding text, -- the encoding type used for state_before\n\n    state_after blob, -- state_after stores the domain state after the request\n    state_after_encoding text, -- the encoding type used for state_after\n\n    operation_type int, -- operation_type stores the type of operation that was performed. It is deserialized as an enum and can be used to customize the serialization/deserialization strategy.\n\n    created_time timestamp, -- created_time the time this row was inserted\n    last_updated_time timestamp,\n\n    identity text, -- the unique identifier of the user that made the change\n    identity_type text, -- identity_type can be used to delineate between service, user, or other identities\n\n    comment text, -- comment can be used when manual updates or creates are performed on the database as an audit trail\n\n    PRIMARY KEY ((domain_id, operation_type), created_time, event_id)\n) WITH CLUSTERING ORDER BY (created_time DESC, event_id ASC)\n  AND COMPACTION = {\n      'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/s0.0-0.23/data_0_23.cql",
    "content": "INSERT INTO domains_by_name_v2 (\n   domains_partition,\n   name,\n   domain,\n   config,\n   is_global_domain,\n   config_version,\n   failover_version,\n   failover_notification_version,\n   notification_version\n) VALUES (\n   0,\n   'cadence-system',\n   {\n       id: 32049b68-7872-4094-8e63-d0dd59896a83,\n       name: 'cadence-system',\n       description: 'cadence system workflow domain',\n       owner_email: 'cadence-dev-group@uber.com'\n   },\n   {\n       retention:3,\n       emit_metric:False\n   },\n   False,\n   0,\n   -24,\n   -1,\n   -1\n) IF NOT EXISTS;\n\nINSERT INTO domains (\n   id,\n   domain\n) VALUES (\n   32049b68-7872-4094-8e63-d0dd59896a83,\n   {\n       name: 'cadence-system'\n   }\n) IF NOT EXISTS;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/s0.0-0.23/manifest.json",
    "content": "{\n    \"CurrVersion\": \"0.23\",\n    \"MinCompatibleVersion\": \"0.23\",\n    \"Description\": \"Squashed 0.0 -> 0.23 schema\",\n    \"SchemaUpdateCqlFiles\": [\n      \"schema_0_23.cql\",\n      \"data_0_23.cql\"\n    ]\n  }\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/s0.0-0.23/schema_0_23.cql",
    "content": "CREATE TYPE shard (\n  shard_id                    int,\n  owner                       text, -- Host identifier processing the shard\n  -- Range identifier used for generating ack ids for tasks within shard.\n  -- Also used for optimistic concurrency and all writes to a shard are conditional on this value.\n  range_id                    bigint,\n  -- This field keeps track of number of times owner for a shard changes before updating range_id or ack_levels\n  stolen_since_renew          int,\n  updated_at                  timestamp,\n  transfer_ack_level          bigint, -- TO BE DEPRECATED, IN FAVOR OF cluster_transfer_ack_level\n  timer_ack_level             timestamp, -- TO BE DEPRECATED, IN FAVOR OF cluster_timer_ack_level\n  replication_ack_level       bigint,\n  -- Mapping of cluster to corresponding transfer ack level\n  cluster_transfer_ack_level  map<text, bigint>,\n  -- Mapping of cluster to corresponding timer ack level\n  cluster_timer_ack_level     map<text, timestamp>,\n  domain_notification_version bigint, -- the global domain change version this shard is aware of\n  -- Mapping of (remote) cluster to corresponding replication level (last replicated task_id)\n  cluster_replication_level   map<text, bigint>,\n);\n\n--- Workflow execution and mutable state ---\nCREATE TYPE workflow_execution (\n  domain_id                        uuid,\n  workflow_id                      text,\n  run_id                           uuid,\n  parent_domain_id                 uuid,   -- Domain ID of parent workflow which started the workflow execution\n  parent_workflow_id               text,   -- ID of parent workflow which started the workflow execution\n  parent_run_id                    uuid,   -- RunID of parent workflow which started the workflow execution\n  initiated_id                     bigint, -- Initiated event ID of parent workflow which started this execution\n  completion_event                 blob,   -- Completion event used to communicate result to parent workflow execution\n  task_list                        text,\n  workflow_type_name               text,\n  decision_task_timeout            int,  -- decision start to close timeout\n  execution_context                blob,\n  state                            int,  -- enum WorkflowState {Created, Running, Completed}\n  close_status                     int,  -- enum WorkflowCloseStatus {None, Completed, Failed, Canceled, Terminated, ContinuedAsNew, TimedOut}\n  next_event_id                    bigint,\n  last_processed_event             bigint,\n  start_time                       timestamp,\n  last_updated_time                timestamp,\n  create_request_id                uuid,\n  decision_schedule_id             bigint,\n  decision_started_id              bigint,\n  decision_request_id              text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  decision_timeout                 int,\n  cancel_requested                 boolean,\n  cancel_request_id                text,\n  workflow_timeout                 int,  -- Workflow ExecutionStartToCloseTimeoutSeconds\n  sticky_task_list                 text,   -- sticky worker task list\n  sticky_schedule_to_start_timeout int,\n  decision_attempt                 bigint,\n  decision_timestamp               bigint,  -- this is decision started time\n  client_library_version           text,\n  client_feature_version           text,\n  client_impl                      text,\n  last_first_event_id              bigint,\n  decision_version                 bigint,\n  attempt                          int,    -- starting from 0 (for initial non-retry)\n  has_retry_policy                 boolean,-- If there is a retry policy\n  init_interval                    int,    -- initial retry interval, in seconds\n  backoff_coefficient              double,\n  max_interval                     int,    -- max retry interval in seconds\n  expiration_time                  timestamp, -- retry expiration time\n  max_attempts                     int,    -- max number of attempts including initial non-retry attempt\n  non_retriable_errors             list<text>,\n  history_size                     bigint,\n  completion_event_data_encoding   text, -- Protocol used for history serialization\n  event_store_version              int, -- indicates which version of events persistence is using\n  branch_token                     blob,\n  signal_count                     int,\n  cron_schedule                    text,\n  expiration_seconds               int,    -- retry expiration duration in seconds\n  completion_event_batch_id        bigint,\n  last_event_task_id               bigint,\n  auto_reset_points                blob, -- the resetting points for auto-reset feature\n  auto_reset_points_encoding       text, -- encoding for auto_reset_points_data\n  decision_scheduled_timestamp     bigint,   -- this is decision scheduled time\n  search_attributes                map<text, blob>,\n  memo                             map<text, blob>,\n  decision_original_scheduled_timestamp     bigint,   -- this is scheduled time of the first decision during heartbeat\n);\n\n-- Replication information for each cluster\nCREATE TYPE replication_info (\n  version       bigint,\n  last_event_id bigint,\n);\n\n-- This is used to store replication information for a workflow execution\nCREATE TYPE replication_state (\n  current_version                  bigint, -- current version for domain, incremented on failover\n  start_version                    bigint, -- version of domain when the workflow execution was started\n  last_write_version               bigint, -- version of domain when the last event was written to history\n  last_write_event_id              bigint, -- last written event id for a given version\n  last_replication_info            map<text, frozen<replication_info>>, -- information about replication events from other clusters\n);\n\n-- TODO: Remove fields that are left over from activity and workflow tasks.\nCREATE TYPE transfer_task (\n  domain_id                  uuid,   -- The domain ID that this transfer task belongs to\n  workflow_id                text,   -- The workflow ID that this transfer task belongs to\n  run_id                     uuid,   -- The run ID that this transfer task belongs to\n  task_id                    bigint,\n  target_domain_id           uuid,   -- The external domain ID that this transfer task is doing work for.\n  target_workflow_id         text,   -- The external workflow ID that this transfer task is doing work for.\n  target_run_id              uuid,   -- The external run ID that this transfer task is doing work for.\n  task_list                  text,\n  type                       int,    -- enum TaskType {ActivityTask, DecisionTask, DeleteExecution, CancelExecution, StartChildExecution, ReplicationTask}\n  schedule_id                bigint,\n  target_child_workflow_only boolean, -- The whether target child workflow only.\n  version                    bigint, -- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\n  visibility_ts              timestamp, -- The timestamp when the transfer task is generated\n  record_visibility          boolean, -- indicates whether or not to create a visibility record\n);\n\nCREATE TYPE replication_task (\n  domain_id                  uuid,   -- The domain ID that this replication task belongs to\n  workflow_id                text,   -- The workflow ID that this replication task belongs to\n  run_id                     uuid,   -- The run ID that this replication task belongs to\n  task_id                    bigint,\n  type                       int,    -- enum TaskType {History, Heartbeat}\n  first_event_id             bigint,  -- Used by ReplicationTask to set the first event ID of the applied transaction\n  next_event_id              bigint,  -- Used by ReplicationTask to set the next event ID of the applied transaction\n  version                    bigint,  -- Used by ReplicationTask to set the failover version of the applied transaction\n  last_replication_info      map<text, frozen<replication_info>>, -- Used by replication task to snapshot replication information when the transaction was applied\n  scheduled_id               bigint, -- Used by ReplicationTask to sync activity info\n  event_store_version        int, -- indicates which version of event store to query\n  branch_token               blob, -- if eventV2, then query with this token\n  new_run_event_store_version        int, -- indicates which version of event store to query for new run(continueAsNew)\n  new_run_branch_token               blob, -- if eventV2, then query with this token for new run(continueAsNew)\n  reset_workflow             boolean, -- whether the task is for resetWorkflowExecution\n);\n\nCREATE TYPE timer_task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  visibility_ts    timestamp,\n  task_id          bigint,\n  type             int,  -- enum TaskType {DecisionTaskTimeout, ActivityTaskTimeout, UserTimer}\n  timeout_type     int, -- enum TimeoutType in IDL {START_TO_CLOSE, SCHEDULE_TO_START, SCHEDULE_TO_CLOSE, HEARTBEAT}\n  event_id         bigint, -- Corresponds to event ID in history that is responsible for this timer.\n  schedule_attempt bigint, -- Used to retry failed decision tasks using mutable state\n  version          bigint, -- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\n);\n\n-- Workflow activity in progress mutable state\nCREATE TYPE activity_info (\n  schedule_id               bigint,\n  scheduled_event           blob,  -- deprecated\n  scheduled_time            timestamp,\n  started_id                bigint,\n  started_event             blob,\n  started_time              timestamp,\n  activity_id               text,    -- Client generated unique ID for the activity.\n  request_id                text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  details                   blob,\n  schedule_to_start_timeout int,\n  schedule_to_close_timeout int,\n  start_to_close_timeout    int,\n  heart_beat_timeout        int,\n  cancel_requested          boolean, -- If a cancel request is made to cancel the activity in progress.\n  cancel_request_id         bigint,  -- Event ID that identifies the cancel request.\n  last_hb_updated_time      timestamp, -- Last time the heartbeat is received.\n  timer_task_status         int,    -- Indicates whether timers are created for this activity.\n  version                   bigint,\n  attempt                   int,    -- starting from 0 (for initial non-retry)\n  task_list                 text,\n  started_identity          text,   -- last started poller's identity\n  has_retry_policy          boolean,-- If there is a retry policy\n  init_interval             int,    -- initial retry interval, in seconds\n  backoff_coefficient       double,\n  max_interval              int,    -- max retry interval in seconds\n  expiration_time           timestamp, -- retry expiration time\n  max_attempts              int,    -- max number of attempts including initial non-retry attempt\n  non_retriable_errors      list<text>,\n  event_data_encoding       text, -- Protocol used for history serialization\n  scheduled_event_batch_id  bigint,\n  last_failure_reason       text,\n  last_worker_identity      text, -- Worker that returns the last failure reason\n  last_failure_details      blob\n);\n\n-- User timer details\nCREATE TYPE timer_info (\n  timer_id      text,      -- User defined timer ID\n  started_id    bigint,    -- The event ID corresponding to timer started.\n  expiry_time   timestamp, -- Timestamp at which this timer expires or fires\n  -- task_id is a misleading variable, it actually serves\n  -- the purpose of indicating whether a timer task is\n  -- generated for this timer info\n  task_id       bigint,\n  version       bigint,\n);\n\n-- Child execution in progress mutable state\nCREATE TYPE child_execution_info (\n  initiated_id              bigint,\n  initiated_event           blob,\n  started_id                bigint,\n  started_event             blob, -- deprecated\n  create_request_id         uuid,\n  version                   bigint,\n  event_data_encoding       text, -- Protocol used for history serialization\n  initiated_event_batch_id  bigint,\n  started_workflow_id       text,\n  started_run_id            uuid,\n  domain_name               text,\n  workflow_type_name        text,\n  parent_close_policy       int,\n);\n\n-- External workflow cancellation in progress mutable state\nCREATE TYPE request_cancel_info (\n  initiated_id              bigint,\n  cancel_request_id         text,\n  version                   bigint,\n  initiated_event_batch_id  bigint,\n);\n\n-- External workflow signal in progress mutable state\nCREATE TYPE signal_info (\n  initiated_id              bigint,\n  signal_request_id         uuid,\n  signal_name               text,\n  input                     blob,\n  control                   blob,\n  version                   bigint,\n  initiated_event_batch_id  bigint,\n);\n\n-- Activity or workflow task in a task list\nCREATE TYPE task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  schedule_id      bigint,\n  created_time     timestamp\n);\n\nCREATE TYPE task_list (\n  domain_id        uuid,\n  name             text,\n  type             int, -- enum TaskRowType {ActivityTask, DecisionTask}\n  ack_level        bigint, -- task_id of the last acknowledged message\n  kind             int, -- enum TaskListKind {Normal, Sticky}\n  last_updated     timestamp\n);\n\nCREATE TYPE domain (\n  id          uuid,\n  name        text,\n  status      int, -- enum DomainStatus {Registered, Deprecated, Deleted}\n  description text,\n  owner_email text,\n  data        map<text,text>, -- Used for customized domain information, key values pair\n);\n\nCREATE TYPE domain_config (\n  retention   int,\n  emit_metric boolean,\n  archival_bucket text, -- deprecated, use the two uri fields below\n  archival_status int, -- deprecated, use the two status fields below\n  bad_binaries    blob,\n  bad_binaries_encoding text,\n  history_archival_status int,\n  history_archival_uri text,\n  visibility_archival_status int,\n  visibility_archival_uri text\n);\n\nCREATE TYPE cluster_replication_config (\n  cluster_name text,\n);\n\nCREATE TYPE domain_replication_config (\n  active_cluster_name text,\n  clusters            list<frozen<cluster_replication_config>>\n);\n\nCREATE TYPE serialized_event_batch (\n  encoding_type text,\n  version       int,\n  data          blob,\n);\n\n-- Storage for out of order replication tasks for an execution\nCREATE TYPE buffered_replication_task_info (\n  first_event_id  bigint,\n  next_event_id   bigint,\n  version         bigint,\n  history         frozen<serialized_event_batch>,\n  new_run_history frozen<serialized_event_batch>,\n  event_store_version                int, -- indicates which version of event store to query\n  new_run_event_store_version        int, -- indicates which version of event store to query for new run(continueAsNew)\n);\n\n-- for history v2 events\nCREATE TYPE branch_range (\n  branch_id   uuid,\n  end_node_id bigint, -- exclusive node_id to represent the stopping point for this range\n);\n\nCREATE TABLE executions (\n  shard_id                       int,\n  type                           int, -- enum RowType { Shard, Execution, TransferTask, TimerTask, ReplicationTask}\n  domain_id                      uuid,\n  workflow_id                    text,\n  run_id                         uuid,\n  current_run_id                 uuid,\n  visibility_ts                  timestamp, -- unique identifier for timer tasks for an execution\n  task_id                        bigint, -- unique identifier for transfer and timer tasks for an execution\n  shard                          frozen<shard>,\n  execution                      frozen<workflow_execution>,\n  transfer                       frozen<transfer_task>,\n  replication                    frozen<replication_task>,\n  timer                          frozen<timer_task>,\n  next_event_id                  bigint,  -- This is needed to make conditional updates on session history\n  range_id                       bigint, -- Increasing sequence identifier for transfer queue, checkpointed into shard info\n  activity_map                   map<bigint, frozen<activity_info>>,\n  timer_map                      map<text, frozen<timer_info>>,\n  child_executions_map           map<bigint, frozen<child_execution_info>>,\n  request_cancel_map             map<bigint, frozen<request_cancel_info>>,\n  signal_map                     map<bigint, frozen<signal_info>>,\n  signal_requested               set<uuid>,\n  buffered_events_list           list<frozen<serialized_event_batch>>,\n  replication_state              frozen<replication_state>, -- Replication information part of mutable state\n  buffered_replication_tasks_map map<bigint, frozen<buffered_replication_task_info>>,\n  workflow_last_write_version    bigint,\n  workflow_state                 int,\n  version_histories              blob, -- the metadata of history branching\n  version_histories_encoding     text,\n  PRIMARY KEY  (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE history_node (\n  tree_id           uuid,\n  branch_id         uuid,\n  node_id           bigint, -- node_id: first eventID in a batch of events\n  txn_id            bigint, -- for override the same node_id: bigger txn_id wins\n  data                blob, -- Batch of workflow execution history events as a blob\n  data_encoding       text, -- Protocol used for history serialization\n  PRIMARY KEY ((tree_id), branch_id, node_id, txn_id )\n  ) WITH CLUSTERING ORDER BY (branch_id ASC, node_id ASC, txn_id DESC)\n    AND COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n    };\n\nCREATE TABLE history_tree (\n  tree_id           uuid,\n  branch_id         uuid,\n  ancestors         list<frozen<branch_range>>,\n  fork_time         timestamp, -- For fork operation to prevent race condition to leak event data when forking branches\n  info              text, -- For background cleanup when fork operation cannot finish self cleanup due to crash\n  PRIMARY KEY ((tree_id), branch_id )\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\n-- Stores activity or workflow tasks\nCREATE TABLE tasks (\n  domain_id        uuid,\n  task_list_name   text,\n  task_list_type   int, -- enum TaskListType {ActivityTask, DecisionTask}\n  type             int, -- enum rowType {Task, TaskList}\n  task_id          bigint,  -- unique identifier for tasks, monotonically increasing\n  range_id         bigint, -- Used to ensure that only one process can write to the table\n  task             frozen<task>,\n  task_list        frozen<task_list>,\n  PRIMARY KEY ((domain_id, task_list_name, task_list_type), type, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\n-- this table is only used for storage of mapping of domain uuid to domain name\nCREATE TABLE domains (\n  id     uuid,\n  domain frozen<domain>,\n  config frozen<domain_config>,\n  PRIMARY KEY (id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE domains_by_name_v2 (\n  domains_partition             int,\n  name                          text,\n  domain                        frozen<domain>,\n  config                        frozen<domain_config>,\n  replication_config            frozen<domain_replication_config>, -- indicating active cluster and standby cluster used for replication\n  is_global_domain              boolean, -- indicating whether a domain is a global domain\n  config_version                bigint, -- indicating the version of domain config, excluding the failover / change of active cluster name\n  failover_version              bigint, -- indicating the version of active domain only, used for domain failover\n  failover_notification_version bigint, -- indicating the last change related to domain failover\n  notification_version          bigint,\n  PRIMARY KEY (domains_partition, name)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n\nCREATE TABLE queue (\n  queue_type      int,\n  message_id      int,\n  message_payload blob,\n  PRIMARY KEY  (queue_type, message_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE queue_metadata (\n  queue_type        int,\n  cluster_ack_level map<text, bigint>,\n  version           bigint,\nPRIMARY KEY (queue_type)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n\nCREATE TABLE events (\n  domain_id      uuid,\n  workflow_id    text,\n  run_id         uuid,\n  -- We insert a batch of events with each append transaction.\n  -- This field stores the event id of first event in the batch.\n  first_event_id bigint,\n  range_id       bigint,\n  tx_id          bigint,\n  data           blob, -- Batch of workflow execution history events as a blob\n  data_encoding  text, -- Protocol used for history serialization\n  data_version   int,  -- history blob version\n  event_batch_version bigint,\n  PRIMARY KEY ((domain_id, workflow_id, run_id), first_event_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.1/base.cql",
    "content": "CREATE TYPE shard (\n  shard_id            int,\n  owner               text, -- Host identifier processing the shard\n  -- Range identifier used for generating ack ids for tasks within shard.\n  -- Also used for optimistic concurrency and all writes to a shard are conditional on this value.\n  range_id            bigint,\n  -- This field keeps track of number of times owner for a shard changes before updating range_id or ack_levels\n  stolen_since_renew  int,\n  updated_at          timestamp,\n  transfer_ack_level  bigint,\n  timer_ack_level     timestamp,\n);\n\n--- Workflow execution and mutable state ---\nCREATE TYPE workflow_execution (\n  domain_id              uuid,\n  workflow_id            text,\n  run_id                 uuid,\n  parent_domain_id       uuid,   -- Domain ID of parent workflow which started the workflow execution\n  parent_workflow_id     text,   -- ID of parent workflow which started the workflow execution\n  parent_run_id          uuid,   -- RunID of parent workflow which started the workflow execution\n  initiated_id           bigint, -- Initiated event ID of parent workflow which started this execution\n  completion_event       blob,   -- Completion event used to communicate result to parent workflow execution\n  task_list              text,\n  workflow_type_name     text,\n  decision_task_timeout  int,\n  execution_context      blob,\n  state                  int,  -- enum WorkflowState {Created, Running, Completed}\n  close_status           int,  -- enum WorkflowCloseStatus {None, Completed, Failed, Canceled, Terminated, ContinuedAsNew, TimedOut}\n  next_event_id          bigint,\n  last_processed_event   bigint,\n  start_time             timestamp,\n  last_updated_time      timestamp,\n  create_request_id      uuid,\n  decision_schedule_id   bigint,\n  decision_started_id    bigint,\n  decision_request_id    text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  decision_timeout       int,\n  cancel_requested       boolean,\n  cancel_request_id      text,\n);\n\n-- TODO: Remove fields that are left over from activity and workflow tasks.\nCREATE TYPE transfer_task (\n  domain_id           uuid,   -- The domain ID that this transfer task belongs to\n  workflow_id         text,   -- The workflow ID that this transfer task belongs to\n  run_id              uuid,   -- The run ID that this transfer task belongs to\n  task_id             bigint,\n  target_domain_id    uuid,   -- The external domain ID that this transfer task is doing work for.\n  target_workflow_id  text,   -- The external workflow ID that this transfer task is doing work for.\n  target_run_id       uuid,   -- The external run ID that this transfer task is doing work for.\n  task_list           text,\n  type                int,    -- enum TaskType {ActivityTask, DecisionTask, DeleteExecution, CancelExecution, StartChildExecution}\n  schedule_id         bigint,\n);\n\nCREATE TYPE timer_task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  visibility_ts    timestamp,\n  task_id          bigint,\n  type             int,  -- enum TaskType {DecisionTaskTimeout, ActivityTaskTimeout, UserTimer}\n  timeout_type     int, -- enum TimeoutType in IDL {START_TO_CLOSE, SCHEDULE_TO_START, SCHEDULE_TO_CLOSE, HEARTBEAT}\n  event_id         bigint, -- Corresponds to event ID in history that is responsible for this timer.\n);\n\n-- Workflow activity in progress mutable state\nCREATE TYPE activity_info (\n  schedule_id               bigint,\n  scheduled_event           blob,\n  scheduled_time            timestamp,\n  started_id                bigint,\n  started_event             blob,\n  started_time              timestamp,\n  activity_id               text,    -- Client generated unique ID for the activity.\n  request_id                text,    -- Identifier used by matching engine for retrying history service calls for recording task is started\n  details                   blob,\n  schedule_to_start_timeout int,\n  schedule_to_close_timeout int,\n  start_to_close_timeout    int,\n  heart_beat_timeout        int,\n  cancel_requested          boolean, -- If a cancel request is made to cancel the activity in progress.\n  cancel_request_id         bigint,  -- Event ID that identifies the cancel request.\n  last_hb_updated_time      timestamp, -- Last time the heartbeat is received.\n  timer_task_status         int,    -- Indicates wheter timers are created for this activity.\n);\n\n-- User timer details\nCREATE TYPE timer_info (\n  timer_id      text,      -- User defined timer ID\n  started_id    bigint,    -- The event ID corresponding to timer started.\n  expiry_time   timestamp, -- Timestamp at which this timer expires or fires\n  task_id       bigint,    -- The task ID if we have one created for this timer\n);\n\n-- Child execution in progress mutable state\nCREATE TYPE child_execution_info (\n  initiated_id      bigint,\n  initiated_event   blob,\n  started_id        bigint,\n  started_event     blob,\n  create_request_id uuid,\n);\n\n-- External workflow cancellation in progress mutable state\nCREATE TYPE request_cancel_info (\n  initiated_id      bigint,\n  cancel_request_id text,\n);\n\n-- Activity or workflow task in a task list\nCREATE TYPE task (\n  domain_id        uuid,\n  workflow_id      text,\n  run_id           uuid,\n  schedule_id      bigint,\n);\n\nCREATE TYPE task_list (\n  domain_id        uuid,\n  name             text,\n  type             int, -- enum TaskRowType {ActivityTask, DecisionTask}\n  ack_level        bigint, -- task_id of the last acknowledged message\n);\n\nCREATE TYPE domain (\n  id          uuid,\n  name        text,\n  status      int, -- enum DomainStatus {Registered, Deprecated, Deleted}\n  description text,\n  owner_email text,\n);\n\nCREATE TYPE domain_config (\n  retention int,\n  emit_metric boolean\n);\n\nCREATE TABLE executions (\n  shard_id             int,\n  type                 int, -- enum RowType { Shard, Execution, TransferTask, TimerTask}\n  domain_id            uuid,\n  workflow_id          text,\n  run_id               uuid,\n  current_run_id       uuid,\n  visibility_ts        timestamp, -- unique identifier for timer tasks for an execution\n  task_id              bigint, -- unique identifier for transfer and timer tasks for an execution\n  shard                frozen<shard>,\n  execution            frozen<workflow_execution>,\n  transfer             frozen<transfer_task>,\n  timer                frozen<timer_task>,\n  next_event_id        bigint,  -- This is needed to make conditional updates on session history\n  range_id             bigint, -- Increasing sequence identifier for transfer queue, checkpointed into shard info\n  activity_map         map<bigint, frozen<activity_info>>,\n  timer_map            map<text, frozen<timer_info>>,\n  child_executions_map map<bigint, frozen<child_execution_info>>,\n  request_cancel_map   map<bigint, frozen<request_cancel_info>>,\n  PRIMARY KEY  (shard_id, type, domain_id, workflow_id, run_id, visibility_ts, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE events (\n  domain_id      uuid,\n  workflow_id    text,\n  run_id         uuid,\n  -- We insert a batch of events with each append transaction.\n  -- This field stores the event id of first event in the batch.\n  first_event_id bigint,\n  range_id       bigint,\n  tx_id          bigint,\n  data           blob, -- Batch of workflow execution history events as a blob\n  data_encoding  text, -- Protocol used for history serialization\n  data_version   int,  -- history blob version\n  PRIMARY KEY ((domain_id, workflow_id, run_id), first_event_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\n-- Stores activity or workflow tasks\nCREATE TABLE tasks (\n  domain_id        uuid,\n  task_list_name   text,\n  task_list_type   int, -- enum TaskListType {ActivityTask, DecisionTask}\n  type             int, -- enum rowType {Task, TaskList}\n  task_id          bigint,  -- unique identifier for tasks, monotonically increasing\n  range_id         bigint, -- Used to ensure that only one process can write to the table\n  task             frozen<task>,\n  task_list        frozen<task_list>,\n  PRIMARY KEY ((domain_id, task_list_name, task_list_type), type, task_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n\nCREATE TABLE domains (\n  id     uuid,\n  domain frozen<domain>,\n  config frozen<domain_config>,\n  PRIMARY KEY (id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.1/manifest.json",
    "content": "{\n    \"CurrVersion\": \"0.1\",\n    \"MinCompatibleVersion\": \"0.1\",\n    \"Description\": \"base version of schema\",\n    \"SchemaUpdateCqlFiles\": [\n        \"base.cql\"\n    ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.10/event_batch_version.cql",
    "content": "ALTER TABLE events ADD event_batch_version bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.10/execution_last_write_version_and_workflow_state.cql",
    "content": "ALTER TABLE executions ADD workflow_last_write_version bigint;\nALTER TABLE executions ADD workflow_state int;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.10/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.10\",\n  \"MinCompatibleVersion\": \"0.10\",\n  \"Description\": \"Adding batch event verion to events table, alter execution table to have workflow state and last write version\",\n  \"SchemaUpdateCqlFiles\": [\n    \"event_batch_version.cql\",\n    \"execution_last_write_version_and_workflow_state.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.11/event_encoding.cql",
    "content": "ALTER TYPE workflow_execution ADD completion_event_data_encoding   text;\nALTER TYPE activity_info ADD event_data_encoding   text;\nALTER TYPE child_execution_info ADD event_data_encoding   text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.11/history_size.cql",
    "content": "ALTER TYPE workflow_execution ADD history_size bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.11/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.11\",\n  \"MinCompatibleVersion\": \"0.11\",\n  \"Description\": \"Mutable state support for server-side retries, history size, and event encoding for mutableState\",\n  \"SchemaUpdateCqlFiles\": [\n    \"workflow_retry.cql\",\n    \"history_size.cql\",\n    \"event_encoding.cql\",\n    \"sync_activity.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.11/sync_activity.cql",
    "content": "ALTER TYPE replication_task ADD scheduled_id bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.11/workflow_retry.cql",
    "content": "ALTER TYPE workflow_execution ADD attempt              int;\nALTER TYPE workflow_execution ADD has_retry_policy     boolean;\nALTER TYPE workflow_execution ADD init_interval        int;\nALTER TYPE workflow_execution ADD backoff_coefficient  double;\nALTER TYPE workflow_execution ADD max_interval         int;\nALTER TYPE workflow_execution ADD expiration_time      timestamp;\nALTER TYPE workflow_execution ADD max_attempts         int;\nALTER TYPE workflow_execution ADD non_retriable_errors list<text>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/add_archival_config.cql",
    "content": "ALTER TYPE domain_config ADD archival_bucket text;\nALTER TYPE domain_config ADD archival_status int;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/cron.cql",
    "content": "ALTER TYPE workflow_execution ADD cron_schedule      text;\nALTER TYPE workflow_execution ADD expiration_seconds int;\n\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/events_v2.cql",
    "content": "CREATE TYPE branch_range (\n  branch_id   uuid,\n  end_node_id bigint, -- exclusive node_id to represent the stopping point for this range\n);\n\nCREATE TABLE history_node (\n  tree_id           uuid,\n  branch_id         uuid,\n  node_id           bigint, -- node_id: first eventID in a batch of events\n  txn_id            bigint, -- for override the same node_id: bigger txn_id wins\n  data                blob, -- Batch of workflow execution history events as a blob\n  data_encoding       text, -- Protocol used for history serialization\n  PRIMARY KEY ((tree_id), branch_id, node_id, txn_id )\n) WITH CLUSTERING ORDER BY (branch_id ASC, node_id ASC, txn_id DESC)\nAND COMPACTION = {\n  'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n};\n\nCREATE TABLE history_tree (\n  tree_id           uuid,\n  branch_id         uuid,\n  ancestors         list<frozen<branch_range>>,\n  PRIMARY KEY ((tree_id), branch_id )\n) WITH COMPACTION = {\n  'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n};\n\n--- using eventsV2\nALTER TYPE workflow_execution ADD  event_store_version           int;\nALTER TYPE workflow_execution ADD  branch_token                  blob;\nALTER TYPE replication_task ADD  event_store_version             int;\nALTER TYPE replication_task ADD  branch_token                    blob;\nALTER TYPE replication_task ADD  new_run_event_store_version     int;\nALTER TYPE replication_task ADD  new_run_branch_token            blob;\nALTER TYPE buffered_replication_task_info ADD  event_store_version     int;\nALTER TYPE buffered_replication_task_info ADD  new_run_event_store_version     int;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.12\",\n  \"MinCompatibleVersion\": \"0.12\",\n  \"Description\": \"Support history events v2, system workflow domain bootstrap, add archival config to domain config\",\n  \"SchemaUpdateCqlFiles\": [\n    \"events_v2.cql\",\n    \"signal_count.cql\",\n    \"system_domain_bootstrap.cql\",\n    \"cron.cql\",\n    \"add_archival_config.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/signal_count.cql",
    "content": "ALTER TYPE workflow_execution ADD  signal_count int;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.12/system_domain_bootstrap.cql",
    "content": "INSERT INTO domains (\n    id,\n    domain\n) VALUES (\n    32049b68-7872-4094-8e63-d0dd59896a83,\n    {\n        name: 'cadence-system'\n    }\n) IF NOT EXISTS;\n\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.13/events_cache.cql",
    "content": "ALTER TYPE activity_info ADD  scheduled_event_batch_id bigint;\nALTER TYPE child_execution_info ADD  initiated_event_batch_id bigint;\nALTER TYPE child_execution_info ADD  started_workflow_id text;\nALTER TYPE child_execution_info ADD  started_run_id uuid;\nALTER TYPE child_execution_info ADD  domain_name text;\nALTER TYPE child_execution_info ADD  workflow_type_name text;\nALTER TYPE workflow_execution ADD  completion_event_batch_id bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.13/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.13\",\n  \"MinCompatibleVersion\": \"0.13\",\n  \"Description\": \"Support workflow reset.  Move history events out of mutable state.\",\n  \"SchemaUpdateCqlFiles\": [\n    \"reset.cql\",\n    \"events_cache.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.13/reset.cql",
    "content": "ALTER TYPE transfer_task ADD record_visibility          boolean;\n\nALTER TYPE replication_task ADD  reset_workflow                  boolean;\n\nALTER TABLE history_tree ADD   fork_time         timestamp;\nALTER TABLE history_tree ADD   info              text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.14/last_event_task_id.cql",
    "content": "ALTER TYPE workflow_execution ADD last_event_task_id bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.14/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.14\",\n  \"MinCompatibleVersion\": \"0.14\",\n  \"Description\": \"Added params to tasklist type to enable task_list deletions\",\n  \"SchemaUpdateCqlFiles\": [\n    \"task_list.cql\",\n    \"last_event_task_id.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.14/task_list.cql",
    "content": "ALTER TYPE task_list ADD last_updated timestamp;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.15/auto_reset.cql",
    "content": "\nALTER TYPE domain_config ADD bad_binaries blob;\nALTER TYPE domain_config ADD bad_binaries_encoding text;\n\nALTER TYPE workflow_execution ADD auto_reset_points blob;\nALTER TYPE workflow_execution ADD auto_reset_points_encoding text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.15/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.15\",\n  \"MinCompatibleVersion\": \"0.15\",\n  \"Description\": \"Added schema for auto-reset\",\n  \"SchemaUpdateCqlFiles\": [\n    \"auto_reset.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.16/decision_scheduled_timestamp.cql",
    "content": "ALTER TYPE workflow_execution ADD  decision_scheduled_timestamp     bigint;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.16/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.16\",\n  \"MinCompatibleVersion\": \"0.16\",\n  \"Description\": \"Added schema for decision_scheduled_timestamp\",\n  \"SchemaUpdateCqlFiles\": [\n    \"decision_scheduled_timestamp.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.17/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.17\",\n  \"MinCompatibleVersion\": \"0.17\",\n  \"Description\": \"Added search attributes to execution\",\n  \"SchemaUpdateCqlFiles\": [\n    \"search_attr.cql\",\n    \"task_created_time.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.17/search_attr.cql",
    "content": "ALTER TYPE workflow_execution ADD search_attributes map<text, blob>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.17/task_created_time.cql",
    "content": "ALTER TYPE task ADD created_time timestamp;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.18/activity_last_failure_info.cql",
    "content": "ALTER TYPE activity_info ADD last_failure_reason  text;\nALTER TYPE activity_info ADD last_worker_identity text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.18/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.18\",\n  \"MinCompatibleVersion\": \"0.18\",\n  \"Description\": \"Added last_failure_reason and last_worker_identity to activity_info\",\n  \"SchemaUpdateCqlFiles\": [\n    \"activity_last_failure_info.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.19/archival_domain_config.cql",
    "content": "ALTER TYPE domain_config ADD history_archival_status  int;\nALTER TYPE domain_config ADD history_archival_uri text;\nALTER TYPE domain_config ADD visibility_archival_status  int;\nALTER TYPE domain_config ADD visibility_archival_uri text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.19/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.19\",\n  \"MinCompatibleVersion\": \"0.19\",\n  \"Description\": \"Added archival related domain config to enable archiver interface\",\n  \"SchemaUpdateCqlFiles\": [\n    \"archival_domain_config.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.2/add_buffered_events.cql",
    "content": "CREATE TYPE serialized_event_batch (\n  encoding_type text,\n  version       int,\n  data          blob,\n);\n\nALTER TABLE executions ADD buffered_events_list list<frozen<serialized_event_batch>>;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.2/add_sticky_tasklist.cql",
    "content": "ALTER TYPE workflow_execution ADD sticky_task_list text;\nALTER TYPE workflow_execution ADD sticky_schedule_to_start_timeout int;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.2/add_wf_timeout.cql",
    "content": "ALTER TYPE workflow_execution ADD workflow_timeout int;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.2/fail_decision_mutable_state.cql",
    "content": "ALTER TYPE workflow_execution ADD decision_attempt bigint;\nALTER TYPE workflow_execution ADD decision_timestamp bigint;\n\nALTER TYPE timer_task ADD schedule_attempt bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add workflow_timeout and buffered_events_list to mutable state\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_wf_timeout.cql\",\n    \"add_buffered_events.cql\",\n    \"add_sticky_tasklist.cql\",\n    \"fail_decision_mutable_state.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.20/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.20\",\n  \"MinCompatibleVersion\": \"0.20\",\n  \"Description\": \"Add memo to mutable state\",\n  \"SchemaUpdateCqlFiles\": [\n    \"memo.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.20/memo.cql",
    "content": "ALTER TYPE workflow_execution ADD memo map<text, blob>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.21/decision_original_scheduled_timestamp.cql",
    "content": "ALTER TYPE workflow_execution ADD decision_original_scheduled_timestamp     bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.21/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.21\",\n  \"MinCompatibleVersion\": \"0.21\",\n  \"Description\": \"Add decision original scheduled timestamp\",\n  \"SchemaUpdateCqlFiles\": [\n    \"decision_original_scheduled_timestamp.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.22/activity_last_failure_details.cql",
    "content": "ALTER TYPE activity_info ADD last_failure_details blob;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.22/cluster_replication_level.cql",
    "content": "ALTER TYPE shard ADD cluster_replication_level   map<text, bigint>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.22/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.22\",\n  \"MinCompatibleVersion\": \"0.22\",\n  \"Description\": \"Add per cluster replication level (last replicated task_id) to shard info and last failure details to activity info\",\n  \"SchemaUpdateCqlFiles\": [\n    \"request_cancel_signal_batch_event_id.cql\",\n    \"cluster_replication_level.cql\",\n    \"parent_close_policy.cql\",\n    \"activity_last_failure_details.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.22/parent_close_policy.cql",
    "content": "ALTER TYPE child_execution_info ADD parent_close_policy int;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.22/request_cancel_signal_batch_event_id.cql",
    "content": "ALTER TYPE request_cancel_info ADD initiated_event_batch_id bigint;\nALTER TYPE signal_info ADD initiated_event_batch_id bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.23/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.23\",\n  \"MinCompatibleVersion\": \"0.23\",\n  \"Description\": \"Add queue table, queue_metadata table and version history\",\n  \"SchemaUpdateCqlFiles\": [\n    \"version_histories.cql\",\n    \"queue.cql\",\n    \"system_domain.cql\",\n    \"queue_metadata.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.23/queue.cql",
    "content": "CREATE TABLE queue (\n  queue_type      int,\n  message_id      int,\n  message_payload blob,\n  PRIMARY KEY  (queue_type, message_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.23/queue_metadata.cql",
    "content": "CREATE TABLE queue_metadata (\n  queue_type        int,\n  cluster_ack_level map<text, bigint>,\n  version           bigint,\nPRIMARY KEY (queue_type)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.23/system_domain.cql",
    "content": "INSERT INTO domains_by_name_v2 (\n    domains_partition,\n    name,\n    domain,\n    config,\n    is_global_domain,\n    config_version,\n    failover_version,\n    failover_notification_version,\n    notification_version\n) VALUES (\n    0,\n    'cadence-system',\n    {\n        id: 32049b68-7872-4094-8e63-d0dd59896a83,\n        name: 'cadence-system',\n        description: 'cadence system workflow domain',\n        owner_email: 'cadence-dev-group@uber.com'\n    },\n    {\n        retention:3,\n        emit_metric:False\n    },\n    False,\n    0,\n    -24,\n    -1,\n    -1\n) IF NOT EXISTS;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.23/version_histories.cql",
    "content": "ALTER TABLE executions ADD version_histories blob;\nALTER TABLE executions ADD version_histories_encoding text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.24/checksum.cql",
    "content": "CREATE TYPE checksum (version int, flavor int, value blob);\nALTER TABLE executions ADD checksum frozen<checksum>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.24/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.24\",\n  \"MinCompatibleVersion\": \"0.23\",\n  \"Description\": \"Added checksum to executions table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"checksum.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.25/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.25\",\n  \"MinCompatibleVersion\": \"0.25\",\n  \"Description\": \"Added replication dlq ack level mapping to shard type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"replication_dlq.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.25/replication_dlq.cql",
    "content": "ALTER TYPE shard ADD replication_dlq_ack_level map<text, bigint>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.26/domain_dlq_id.cql",
    "content": "DROP TABLE queue;\nCREATE TABLE queue (\n  queue_type      int,\n  message_id      bigint,\n  message_payload blob,\n  PRIMARY KEY  (queue_type, message_id)\n) WITH COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.26/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.26\",\n  \"MinCompatibleVersion\": \"0.26\",\n  \"Description\": \"Alter domain DLQ message ID type to big integer\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_dlq_id.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.27/domain_failover_end_time.cql",
    "content": "ALTER TABLE domains_by_name_v2 ADD failover_end_time bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.27/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.27\",\n  \"MinCompatibleVersion\": \"0.26\",\n  \"Description\": \"Add graceful failover end time in domain data\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_failover_end_time.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.28/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.28\",\n  \"MinCompatibleVersion\": \"0.27\",\n  \"Description\": \"Add creation timestamp in replication task type and update shard info type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"replication_task_creation_time.cql\",\n    \"previous_failover_version.cql\",\n    \"shard_info_marker.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.28/previous_failover_version.cql",
    "content": "ALTER TABLE domains_by_name_v2 ADD previous_failover_version bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.28/replication_task_creation_time.cql",
    "content": "ALTER TYPE replication_task ADD created_time bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.28/shard_info_marker.cql",
    "content": "ALTER TYPE shard ADD pending_failover_markers blob;\nALTER TYPE shard ADD pending_failover_markers_encoding text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.29/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.29\",\n  \"MinCompatibleVersion\": \"0.28\",\n  \"Description\": \"Add timer and transfer processing queue states to shard\",\n  \"SchemaUpdateCqlFiles\": [\n    \"processing_queue_states.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.29/processing_queue_states.cql",
    "content": "ALTER TYPE shard ADD transfer_processing_queue_states blob;\nALTER TYPE shard ADD transfer_processing_queue_states_encoding text;\n\nALTER TYPE shard ADD timer_processing_queue_states blob;\nALTER TYPE shard ADD timer_processing_queue_states_encoding text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.3/add_client_version.cql",
    "content": "ALTER TYPE workflow_execution ADD client_library_version text;\nALTER TYPE workflow_execution ADD client_feature_version text;\nALTER TYPE workflow_execution ADD client_impl text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.3/add_last_first_event_id.cql",
    "content": "ALTER TYPE workflow_execution ADD last_first_event_id bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.3\",\n  \"Description\": \"add client_library_version, client_feature_version, client_impl to mutable state\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_client_version.cql\",\n    \"add_last_first_event_id.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.30/domain_last_updated_time.cql",
    "content": "ALTER TABLE domains_by_name_v2 ADD last_updated_time bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.30/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.30\",\n  \"MinCompatibleVersion\": \"0.29\",\n  \"Description\": \"Add last updated time in domains table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_last_updated_time.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.31/cross_cluster_queue.cql",
    "content": "ALTER TYPE shard ADD cross_cluster_processing_queue_states blob;\nALTER TYPE shard ADD cross_cluster_processing_queue_states_encoding text;\n\nALTER TABLE executions ADD cross_cluster frozen<transfer_task>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.31/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.31\",\n  \"MinCompatibleVersion\": \"0.31\",\n  \"Description\": \"Add cross cluster queue and states\",\n  \"SchemaUpdateCqlFiles\": [\n    \"cross_cluster_queue.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.32/config_store.cql",
    "content": "CREATE TABLE cluster_config (\n  row_type int,\n  version int,\n  timestamp timestamp,\n  values blob,\n  encoding text,\nPRIMARY KEY (row_type, version)\n) WITH CLUSTERING ORDER BY (version DESC);"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.32/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.32\",\n  \"MinCompatibleVersion\": \"0.32\",\n  \"Description\": \"Added cluster_config table for config store support\",\n  \"SchemaUpdateCqlFiles\": [\n    \"config_store.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.33/child_info_domain_id.cql",
    "content": "ALTER TYPE child_execution_info ADD domain_id uuid;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.33/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.33\",\n  \"MinCompatibleVersion\": \"0.33\",\n  \"Description\": \"Added target domain ids to the executions table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"target_domain_ids.cql\",\n    \"child_info_domain_id.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.33/target_domain_ids.cql",
    "content": "ALTER TYPE transfer_task ADD target_domain_ids set<uuid>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.34/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.34\",\n  \"MinCompatibleVersion\": \"0.34\",\n  \"Description\": \"Added first run id to workflow execution type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"workflow_execution_first_run_id.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.34/workflow_execution_first_run_id.cql",
    "content": "ALTER TYPE workflow_execution ADD first_run_id uuid;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.35/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.35\",\n  \"MinCompatibleVersion\": \"0.35\",\n  \"Description\": \"Added partition config to workflow execution and task type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"partition_config.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.35/partition_config.cql",
    "content": "ALTER TYPE workflow_execution ADD partition_config map<text, text>;\nALTER TYPE task ADD partition_config map<text, text>;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.36/isolation_groups.cql",
    "content": "ALTER TYPE domain_config ADD isolation_groups blob;\nALTER TYPE domain_config ADD isolation_groups_encoding text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.36/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.36\",\n  \"MinCompatibleVersion\": \"0.36\",\n  \"Description\": \"Adding the domain configuration for isolation-groups\",\n  \"SchemaUpdateCqlFiles\": [\n    \"isolation_groups.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.37/async_workflow_config.cql",
    "content": "ALTER TYPE domain_config ADD async_workflow_config blob;\nALTER TYPE domain_config ADD async_workflow_config_encoding text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.37/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.37\",\n  \"MinCompatibleVersion\": \"0.37\",\n  \"Description\": \"Adding the domain configuration for async workflow config\",\n  \"SchemaUpdateCqlFiles\": [\n    \"async_workflow_config.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.38/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.38\",\n  \"MinCompatibleVersion\": \"0.38\",\n  \"Description\": \"Adding the task list partition config to task list type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"task_list_partition_config.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.38/task_list_partition_config.cql",
    "content": "CREATE TYPE task_list_partition_config (\n  version              bigint,\n  num_read_partitions  int,\n  num_write_partitions int\n);\n\nALTER TYPE task_list ADD adaptive_partition_config frozen<task_list_partition_config>;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.39/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.39\",\n  \"MinCompatibleVersion\": \"0.39\",\n  \"Description\": \"Adding create time and last update time for Cassandra tables\",\n  \"SchemaUpdateCqlFiles\": [\n    \"timestamps.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.39/timestamps.cql",
    "content": "ALTER TABLE executions ADD created_time timestamp;\nALTER TABLE executions ADD last_updated_time timestamp;\nALTER TABLE history_node ADD created_time timestamp;\nALTER TABLE history_tree ADD created_time timestamp;\nALTER TABLE tasks ADD created_time timestamp;\nALTER TABLE tasks ADD last_updated_time timestamp;\nALTER TABLE domains ADD created_time timestamp;\nALTER TABLE domains_by_name_v2 ADD created_time timestamp;\nALTER TABLE queue ADD created_time timestamp;\nALTER TABLE queue_metadata ADD created_time timestamp;\nALTER TABLE queue_metadata ADD last_updated_time timestamp;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.4/add_signal_decision.cql",
    "content": "CREATE TYPE signal_info (\n  initiated_id      bigint,\n  signal_request_id uuid,\n  signal_name       text,\n  input             blob,\n  control           blob,\n);\n\nALTER TABLE executions ADD signal_map map<bigint, frozen<signal_info>>;\nALTER TABLE executions ADD signal_requested set<uuid>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.4/add_tasklist_kind.cql",
    "content": "ALTER TYPE task_list ADD kind int ;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.4\",\n  \"Description\": \"add signal_decision to mutable state, add kind to taskList\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_signal_decision.cql\",\n    \"add_tasklist_kind.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.40/isolation_partition_config.cql",
    "content": "CREATE TYPE task_list_partition (\n                                    isolation_groups set<text>\n                                );\n\nALTER TYPE task_list_partition_config ADD read_partitions map<int, frozen<task_list_partition>>;\nALTER TYPE task_list_partition_config ADD write_partitions map<int, frozen<task_list_partition>>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.40/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.40\",\n  \"MinCompatibleVersion\": \"0.40\",\n  \"Description\": \"Adding an explicit list of task list partitions to support isolation group assignment\",\n  \"SchemaUpdateCqlFiles\": [\n    \"isolation_partition_config.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.41/executions.cql",
    "content": "ALTER TABLE executions ADD data blob;\nALTER TABLE executions ADD data_encoding text;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.41/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.41\",\n  \"MinCompatibleVersion\": \"0.41\",\n  \"Description\": \"Adding task blob and task blob encoding column to executions table to represent all history tasks\",\n  \"SchemaUpdateCqlFiles\": [\n    \"executions.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.42/domain_active_clusters_config.cql",
    "content": "ALTER TYPE domain_replication_config ADD active_clusters_config blob;\nALTER TYPE domain_replication_config ADD active_clusters_config_encoding text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.42/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.42\",\n  \"MinCompatibleVersion\": \"0.42\",\n  \"Description\": \"Adding active_clusters_config and active_clusters_config_encoding column to domain_replication_config type\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_active_clusters_config.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.43/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.43\",\n  \"MinCompatibleVersion\": \"0.43\",\n  \"Description\": \"Adding new fields to workflow_execution and activity_info types to support following features: cron overlap policy, active-active domain, ephemeral task list\",\n  \"SchemaUpdateCqlFiles\": [\n    \"workflow_execution_new_fields.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.43/workflow_execution_new_fields.cql",
    "content": "ALTER TYPE workflow_execution ADD cron_overlap_policy int;\nALTER TYPE workflow_execution ADD active_cluster_selection_policy blob;\nALTER TYPE workflow_execution ADD active_cluster_selection_policy_encoding text;\nALTER TYPE workflow_execution ADD task_list_kind int;\nALTER TYPE activity_info ADD task_list_kind int;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.44/domain_audit_log.cql",
    "content": "CREATE TABLE domain_audit_log (\n    domain_id uuid,\n    event_id uuid, -- event_id is the unique identifier for this change to the domain\n\n    state_before blob, -- state_before stores the domain state before the request\n    state_before_encoding text, -- the encoding type used for state_before\n\n    state_after blob, -- state_after stores the domain state after the request\n    state_after_encoding text, -- the encoding type used for state_after\n\n    operation_type int, -- operation_type stores the type of operation that was performed. It is deserialized as an enum and can be used to customize the serialization/deserialization strategy. \n\n    created_time timestamp, -- created_time the time this row was inserted\n    last_updated_time timestamp,\n\n    identity text, -- the unique identifier of the user that made the change\n    identity_type text, -- identity_type can be used to delineate between service, user, or other identities\n\n    comment text, -- comment can be used when manual updates or creates are performed on the database as an audit trail\n\n    PRIMARY KEY ((domain_id, operation_type), created_time, event_id)\n) WITH CLUSTERING ORDER BY (created_time DESC, event_id ASC)\n  AND COMPACTION = {\n      'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  };\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.44/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.44\",\n  \"MinCompatibleVersion\": \"0.44\",\n  \"Description\": \"Adding domain_audit_log table to track domain update and failover history\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_audit_log.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.45/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.45\",\n  \"MinCompatibleVersion\": \"0.45\",\n  \"Description\": \"Adding original_task_list and task_list to transfer and timer tasks\",\n  \"SchemaUpdateCqlFiles\": [\n    \"transfer_task.cql\",\n    \"timer_task.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.45/timer_task.cql",
    "content": "ALTER TYPE timer_task ADD task_list text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.45/transfer_task.cql",
    "content": "ALTER TYPE transfer_task ADD original_task_list text;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.46/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.46\",\n  \"MinCompatibleVersion\": \"0.46\",\n  \"Description\": \"Adding original_task_list_kind to transfer task\",\n  \"SchemaUpdateCqlFiles\": [\n    \"transfer_task.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.46/transfer_task.cql",
    "content": "ALTER TYPE transfer_task ADD original_task_list_kind int;\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.5/add_replication_config.cql",
    "content": "CREATE TYPE cluster_replication_config (\n  cluster_name text,\n);\n\nCREATE TYPE domain_replication_config (\n  active_cluster_name text,\n  clusters            list<frozen<cluster_replication_config>>\n);\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.5/add_target_child_workflow_only_to_transfer_task.cql",
    "content": "ALTER TYPE transfer_task ADD target_child_workflow_only boolean;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"add cross DC domain replication config, add target child workflow only to transfer task, add cross DC domain config verion, as a sequency ID to prevent out of order domain update, add flag indicating whether a domain is a global domain\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_replication_config.cql\",\n    \"add_target_child_workflow_only_to_transfer_task.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.6/add_shard_cluster_ack_level.cql",
    "content": "ALTER TYPE shard ADD cluster_transfer_ack_level map<text, bigint>;\nALTER TYPE shard ADD cluster_timer_ack_level map<text, timestamp>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.6/history_replication_task.cql",
    "content": "-- Replication information for each cluster\nCREATE TYPE replication_info (\n  version       bigint,\n  last_event_id bigint,\n);\n\n-- This is used to store replication information for a workflow execution\nCREATE TYPE replication_state (\n  current_version                  bigint, -- current version for domain, incremented on failover\n  start_version                    bigint, -- version of domain when the workflow execution was started\n  last_write_version               bigint, -- version of domain when the last event was written to history\n  last_write_event_id              bigint, -- last written event id for a given version\n  last_replication_info            map<text, frozen<replication_info>>, -- information about replication events from other clusters\n);\n\n-- Transfer task created for processing of replication event\nCREATE TYPE replication_task (\n  domain_id                  uuid,   -- The domain ID that this transfer task belongs to\n  workflow_id                text,   -- The workflow ID that this transfer task belongs to\n  run_id                     uuid,   -- The run ID that this transfer task belongs to\n  task_id                    bigint,\n  type                       int,    -- enum TaskType {History, Heartbeat}\n  first_event_id             bigint,  -- Used by ReplicationTask to set the first event ID of the applied transaction\n  next_event_id              bigint,  -- Used by ReplicationTask to set the next event ID of the applied transaction\n  version                    bigint,  -- Used by ReplicationTask to set the failover version of the applied transaction\n  last_replication_info      map<text, frozen<replication_info>>, -- Used by replication task to snapshot replication information when the transaction was applied\n);\n\n-- Replication information part of mutable state\nALTER TABLE executions ADD replication_state frozen<replication_state>;\n\n-- Replication transfer task kept as part of executions table\nALTER TABLE executions ADD replication frozen<replication_task>;\n\n-- Ack level checkpoint for replication task processor\nALTER TYPE shard ADD replication_ack_level bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"Persistence support for replication of workflow execution, add cross DC shard attr cluster_transfer_ack_level and cluster_timer_ack_level\",\n  \"SchemaUpdateCqlFiles\": [\n    \"history_replication_task.cql\",\n    \"add_shard_cluster_ack_level.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.7/buffered_replication_task.cql",
    "content": "-- Storage for out of order replication tasks for an execution\nCREATE TYPE buffered_replication_task_info (\n  first_event_id  bigint,\n  next_event_id   bigint,\n  version         bigint,\n  history         frozen<serialized_event_batch>,\n  new_run_history frozen<serialized_event_batch>,\n);\n\n-- Buffered replications tasks information part of mutable state\nALTER TABLE executions ADD buffered_replication_tasks_map map<bigint, frozen<buffered_replication_task_info>>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.7/failover_version_persistence.cql",
    "content": "-- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\nALTER TYPE transfer_task ADD version bigint;\n\n-- the failover version when this task is created, used to compare against the mutable state, in case the events got overwritten\nALTER TYPE timer_task ADD version bigint;\n\n-- the failover version when this attr is created, used to compare against the mutable state, in case the events got overwritten\nALTER TYPE workflow_execution ADD decision_version bigint;\nALTER TYPE activity_info ADD version bigint;\nALTER TYPE timer_info ADD version bigint;\nALTER TYPE child_execution_info ADD version bigint;\nALTER TYPE request_cancel_info ADD version bigint;\nALTER TYPE signal_info ADD version bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"Support buffering for replication tasks within mutable state, add retry policy. Add version to transfer / timer task, as well as activity, user timer, decision, child workflow, cancellatioon and signal mutable state attrs.\",\n  \"SchemaUpdateCqlFiles\": [\n    \"buffered_replication_task.cql\",\n    \"failover_version_persistence.cql\",\n    \"retry_policy.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.7/retry_policy.cql",
    "content": "ALTER TYPE activity_info ADD attempt              int;\nALTER TYPE activity_info ADD task_list            text;\nALTER TYPE activity_info ADD started_identity     text;\nALTER TYPE activity_info ADD has_retry_policy     boolean;\nALTER TYPE activity_info ADD init_interval        int;\nALTER TYPE activity_info ADD backoff_coefficient  double;\nALTER TYPE activity_info ADD max_interval         int;\nALTER TYPE activity_info ADD expiration_time      timestamp;\nALTER TYPE activity_info ADD max_attempts         int;\nALTER TYPE activity_info ADD non_retriable_errors list<text>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.8/domain_notification.cql",
    "content": "CREATE TABLE domains_by_name_v2 (\n  domains_partition             int,\n  name                          text,\n  domain                        frozen<domain>,\n  config                        frozen<domain_config>,\n  replication_config            frozen<domain_replication_config>, -- indicating active cluster and standby cluster used for replication\n  is_global_domain              boolean, -- indicating whether a domain is a global domain\n  config_version                bigint, -- indicating the version of domain config, excluding the failover / change of active cluster name\n  failover_version              bigint, -- indicating the version of active domain only, used for domain failover\n  failover_notification_version bigint, -- indicating the last change related to domain failover\n  notification_version          bigint,\n  PRIMARY KEY (domains_partition, name)\n)  WITH COMPACTION = {\n     'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n   };\n\nALTER TYPE shard ADD domain_notification_version bigint;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.8/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.8\",\n  \"MinCompatibleVersion\": \"0.8\",\n  \"Description\": \"Support reliable domain change notification.\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_notification.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.9/domain_data.cql",
    "content": "ALTER TYPE domain ADD data map<text,text>;"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.9/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.9\",\n  \"MinCompatibleVersion\": \"0.9\",\n  \"Description\": \"Support domain customized key-value data, adding timestamp to transfer task.\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_data.cql\",\n    \"transfer_timestamp.cql\"\n  ]\n}"
  },
  {
    "path": "schema/cassandra/cadence/versioned/v0.9/transfer_timestamp.cql",
    "content": "ALTER TYPE transfer_task ADD visibility_ts timestamp;"
  },
  {
    "path": "schema/cassandra/embed.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport \"embed\"\n\n//go:embed cadence/* visibility/*\nvar SchemaFS embed.FS\n"
  },
  {
    "path": "schema/cassandra/version.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\n// NOTE: whenever there is a new data base schema update, plz update the following versions\n\n// Version is the Cassandra database release version\nconst Version = \"0.46\"\n\n// VisibilityVersion is the Cassandra visibility database release version\nconst VisibilityVersion = \"0.10\"\n"
  },
  {
    "path": "schema/cassandra/visibility/keyspace.cql",
    "content": "CREATE KEYSPACE IF NOT EXISTS cadence_visibility WITH replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 1};\n"
  },
  {
    "path": "schema/cassandra/visibility/schema.cql",
    "content": "CREATE TABLE open_executions (\n  domain_id                uuid,\n  domain_partition         int,\n  workflow_id              text,\n  run_id                   uuid,\n  start_time               timestamp,\n  execution_time           timestamp,\n  workflow_type_name       text,\n  memo                     blob,\n  encoding                 text,\n  task_list                text,\n  is_cron                  boolean,\n  num_clusters             int,\n  update_time              timestamp,\n  shard_id                 int,\n  cron_schedule            text,\n  execution_status         int,\n  scheduled_execution_time timestamp,\n  PRIMARY KEY  ((domain_id, domain_partition), start_time, run_id)\n) WITH CLUSTERING ORDER BY (start_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy',\n    'tombstone_threshold': 0.6\n  }\n  AND GC_GRACE_SECONDS = 60;\n\n\nCREATE INDEX open_by_workflow_id ON open_executions (workflow_id);\nCREATE INDEX open_by_type ON open_executions (workflow_type_name);\n\nCREATE TABLE closed_executions (\n  domain_id                uuid,\n  domain_partition         int,\n  workflow_id              text,\n  run_id                   uuid,\n  start_time               timestamp,\n  execution_time           timestamp,\n  close_time               timestamp,\n  status                   int,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  workflow_type_name       text,\n  history_length           bigint,\n  memo                     blob,\n  encoding                 text,\n  task_list                text,\n  is_cron                  boolean,\n  num_clusters             int,\n  update_time              timestamp,\n  shard_id                 int,\n  cron_schedule            text,\n  execution_status         int,\n  scheduled_execution_time timestamp,\n  PRIMARY KEY  ((domain_id, domain_partition), start_time, run_id)\n) WITH CLUSTERING ORDER BY (start_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  }\n  AND GC_GRACE_SECONDS = 172800;\n\nCREATE INDEX closed_by_workflow_id ON closed_executions (workflow_id);\nCREATE INDEX closed_by_close_time ON closed_executions (close_time);\nCREATE INDEX closed_by_type ON closed_executions (workflow_type_name);\nCREATE INDEX closed_by_status ON closed_executions (status);\n\n-- same as closed_executions but order by close_time\nCREATE TABLE closed_executions_v2 (\n  domain_id                uuid,\n  domain_partition         int,\n  workflow_id              text,\n  run_id                   uuid,\n  start_time               timestamp,\n  execution_time           timestamp,\n  close_time               timestamp,\n  status                   int,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  workflow_type_name       text,\n  history_length           bigint,\n  memo                     blob,\n  encoding                 text,\n  task_list                text,\n  is_cron                  boolean,\n  num_clusters             int,\n  update_time              timestamp,\n  shard_id                 int,\n  cron_schedule            text,\n  execution_status         int,\n  scheduled_execution_time timestamp,\n  PRIMARY KEY  ((domain_id, domain_partition), close_time, run_id)\n) WITH CLUSTERING ORDER BY (close_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  }\n  AND GC_GRACE_SECONDS = 172800;\n\nCREATE INDEX closed_by_workflow_id_v2 ON closed_executions_v2 (workflow_id);\nCREATE INDEX closed_by_close_time_v2 ON closed_executions_v2 (close_time);\nCREATE INDEX closed_by_type_v2 ON closed_executions_v2 (workflow_type_name);\nCREATE INDEX closed_by_status_v2 ON closed_executions_v2 (status);"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.1/base.cql",
    "content": "CREATE TABLE open_executions (\n  domain_id            uuid,\n  domain_partition     int,\n  workflow_id          text,\n  run_id               uuid,\n  start_time           timestamp,\n  workflow_type_name   text,\n  PRIMARY KEY  ((domain_id, domain_partition), start_time, run_id)\n) WITH CLUSTERING ORDER BY (start_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  }\n  AND GC_GRACE_SECONDS = 172800;\n\n\nCREATE INDEX open_by_workflow_id ON open_executions (workflow_id);\nCREATE INDEX open_by_type ON open_executions (workflow_type_name);\n\nCREATE TABLE closed_executions (\n  domain_id            uuid,\n  domain_partition     int,\n  workflow_id          text,\n  run_id               uuid,\n  start_time           timestamp,\n  close_time           timestamp,\n  status               int,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  workflow_type_name   text,\n  history_length       bigint,\n  PRIMARY KEY  ((domain_id, domain_partition), start_time, run_id)\n) WITH CLUSTERING ORDER BY (start_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  }\n  AND GC_GRACE_SECONDS = 172800;\n\nCREATE INDEX closed_by_workflow_id ON closed_executions (workflow_id);\nCREATE INDEX closed_by_close_time ON closed_executions (close_time);\nCREATE INDEX closed_by_type ON closed_executions (workflow_type_name);\nCREATE INDEX closed_by_status ON closed_executions (status);"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.1/manifest.json",
    "content": "{\n    \"CurrVersion\": \"0.1\",\n    \"MinCompatibleVersion\": \"0.1\",\n    \"Description\": \"base version of visibility schema\",\n    \"SchemaUpdateCqlFiles\": [\n        \"base.cql\"\n    ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.10/add_execution_fields.cql",
    "content": "-- Add cron_schedule field to track cron workflow schedules\nALTER TABLE open_executions ADD cron_schedule text;\nALTER TABLE closed_executions ADD cron_schedule text;\nALTER TABLE closed_executions_v2 ADD cron_schedule text;\n\n-- Add execution_status field to track workflow execution status (Pending, Started, or close status)\nALTER TABLE open_executions ADD execution_status int;\nALTER TABLE closed_executions ADD execution_status int;\nALTER TABLE closed_executions_v2 ADD execution_status int;\n\n-- Add scheduled_execution_time field to track the actual scheduled execution time (start_time + first_decision_task_backoff)\nALTER TABLE open_executions ADD scheduled_execution_time timestamp;\nALTER TABLE closed_executions ADD scheduled_execution_time timestamp;\nALTER TABLE closed_executions_v2 ADD scheduled_execution_time timestamp;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.10/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.10\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"add cron_schedule, execution_status, and scheduled_execution_time to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_execution_fields.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.2/manifest.json",
    "content": "{\n    \"CurrVersion\": \"0.2\",\n    \"MinCompatibleVersion\": \"0.2\",\n    \"Description\": \"lower gc_grace_seconds and tweak compaction for open_executions table\",\n    \"SchemaUpdateCqlFiles\": [\n        \"reduce_open_workflow_tombstones.cql\"\n    ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.2/reduce_open_workflow_tombstones.cql",
    "content": "ALTER TABLE open_executions WITH gc_grace_seconds=60 AND compaction = {\n  'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy',\n  'tombstone_threshold': 0.4\n};"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.3\",\n  \"Description\": \"change close_executions to order by close_time instead of start_time\",\n  \"SchemaUpdateCqlFiles\": [\n    \"sort_by_close_time.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.3/sort_by_close_time.cql",
    "content": "-- same as closed_executions but order by close_time\nCREATE TABLE closed_executions_v2 (\n  domain_id            uuid,\n  domain_partition     int,\n  workflow_id          text,\n  run_id               uuid,\n  start_time           timestamp,\n  close_time           timestamp,\n  status               int,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  workflow_type_name   text,\n  history_length       bigint,\n  PRIMARY KEY  ((domain_id, domain_partition), close_time, run_id)\n) WITH CLUSTERING ORDER BY (close_time DESC)\n  AND COMPACTION = {\n    'class': 'org.apache.cassandra.db.compaction.LeveledCompactionStrategy'\n  }\n  AND GC_GRACE_SECONDS = 172800;\n\nCREATE INDEX closed_by_workflow_id_v2 ON closed_executions_v2 (workflow_id);\nCREATE INDEX closed_by_close_time_v2 ON closed_executions_v2 (close_time);\nCREATE INDEX closed_by_type_v2 ON closed_executions_v2 (workflow_type_name);\nCREATE INDEX closed_by_status_v2 ON closed_executions_v2 (status);"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.4/add_execution_time.cql",
    "content": "ALTER TABLE open_executions ADD execution_time timestamp;\nALTER TABLE closed_executions ADD execution_time timestamp;\nALTER TABLE closed_executions_v2 ADD execution_time timestamp;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.4/add_memo.cql",
    "content": "ALTER TABLE open_executions ADD memo blob;\nALTER TABLE closed_executions ADD memo blob;\nALTER TABLE closed_executions_v2 ADD memo blob;\nALTER TABLE open_executions ADD encoding text;\nALTER TABLE closed_executions ADD encoding text;\nALTER TABLE closed_executions_v2 ADD encoding text;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.4\",\n  \"Description\": \"add execution_time field to visibility; add non-searchable field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_execution_time.cql\",\n    \"add_memo.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.5/add_task_list.cql",
    "content": "ALTER TABLE open_executions ADD task_list text;\nALTER TABLE closed_executions ADD task_list text;\nALTER TABLE closed_executions_v2 ADD task_list text;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"add task_list field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_task_list.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.6/add_is_cron.cql",
    "content": "ALTER TABLE open_executions ADD is_cron boolean;\nALTER TABLE closed_executions ADD is_cron boolean;\nALTER TABLE closed_executions_v2 ADD is_cron boolean;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"add is_cron field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_is_cron.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.7/add_num_clusters.cql",
    "content": "ALTER TABLE open_executions ADD num_clusters int;\nALTER TABLE closed_executions ADD num_clusters int;\nALTER TABLE closed_executions_v2 ADD num_clusters int;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"add num_clusters field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_num_clusters.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.8/add_update_time.cql",
    "content": "ALTER TABLE open_executions ADD update_time timestamp;\nALTER TABLE closed_executions ADD update_time timestamp;\nALTER TABLE closed_executions_v2 ADD update_time timestamp;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.8/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.8\",\n  \"MinCompatibleVersion\": \"0.8\",\n  \"Description\": \"add update_time field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_update_time.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.9/add_shard_id.cql",
    "content": "ALTER TABLE open_executions ADD shard_id int;\nALTER TABLE closed_executions ADD shard_id int;\nALTER TABLE closed_executions_v2 ADD shard_id int;\n"
  },
  {
    "path": "schema/cassandra/visibility/versioned/v0.9/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.9\",\n  \"MinCompatibleVersion\": \"0.9\",\n  \"Description\": \"add shard_id to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_shard_id.cql\"\n  ]\n}\n"
  },
  {
    "path": "schema/elasticsearch/embed.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage elasticsearch\n\nimport _ \"embed\" // enable embedding\n\n//go:embed v6/visibility/index_template.json\nvar IndexTemplateV6 []byte\n\n//go:embed v7/visibility/index_template.json\nvar IndexTemplateV7 []byte\n\n//go:embed os2/visibility/index_template.json\nvar IndexTemplateOS2 []byte\n"
  },
  {
    "path": "schema/elasticsearch/os2/visibility/index_template.json",
    "content": "{\n  \"aliases\": {},\n  \"index_patterns\": [\n    \"cadence-visibility-*\"\n  ],\n  \"mappings\": {\n    \"dynamic\": \"false\",\n    \"properties\": {\n      \"Attr\": {\n        \"properties\": {\n          \"BinaryChecksums\": {\n            \"type\": \"keyword\"\n          },\n          \"CadenceChangeVersion\": {\n            \"type\": \"keyword\"\n          },\n          \"CustomBoolField\": {\n            \"type\": \"boolean\"\n          },\n          \"CustomDatetimeField\": {\n            \"type\": \"date\"\n          },\n          \"CustomDomain\": {\n            \"type\": \"keyword\"\n          },\n          \"CustomDoubleField\": {\n            \"type\": \"double\"\n          },\n          \"CustomIntField\": {\n            \"type\": \"long\"\n          },\n          \"CustomKeywordField\": {\n            \"type\": \"keyword\"\n          },\n          \"CustomStringField\": {\n            \"type\": \"text\"\n          },\n          \"Operator\": {\n            \"type\": \"keyword\"\n          },\n          \"Passed\": {\n            \"type\": \"boolean\"\n          },\n          \"RolloutID\": {\n            \"type\": \"keyword\"\n          },\n          \"addon\": {\n            \"type\": \"keyword\"\n          },\n          \"addon-type\": {\n            \"type\": \"keyword\"\n          },\n          \"environment\": {\n            \"type\": \"keyword\"\n          },\n          \"project\": {\n            \"type\": \"keyword\"\n          },\n          \"service\": {\n            \"type\": \"keyword\"\n          },\n          \"user\": {\n            \"type\": \"keyword\"\n          }\n        }\n      },\n      \"CloseStatus\": {\n        \"type\": \"integer\"\n      },\n      \"CloseTime\": {\n        \"type\": \"long\"\n      },\n      \"DomainID\": {\n        \"type\": \"keyword\"\n      },\n      \"ExecutionTime\": {\n        \"type\": \"long\"\n      },\n      \"HistoryLength\": {\n        \"type\": \"integer\"\n      },\n      \"IsCron\": {\n        \"type\": \"boolean\"\n      },\n      \"KafkaKey\": {\n        \"type\": \"keyword\"\n      },\n      \"NumClusters\": {\n        \"type\": \"integer\"\n      },\n      \"RunID\": {\n        \"type\": \"keyword\"\n      },\n      \"ShardID\": {\n        \"type\": \"long\"\n      },\n      \"CronSchedule\": {\n        \"type\": \"keyword\"\n      },\n      \"ExecutionStatus\": {\n        \"type\": \"integer\"\n      },\n      \"ScheduledExecutionTime\": {\n        \"type\": \"long\"\n      },\n      \"StartTime\": {\n        \"type\": \"long\"\n      },\n      \"TaskList\": {\n        \"type\": \"keyword\"\n      },\n      \"ClusterAttributeScope\": {\n        \"type\": \"keyword\"\n      },\n      \"ClusterAttributeName\": {\n        \"type\": \"keyword\"\n      },\n      \"UpdateTime\": {\n        \"type\": \"long\"\n      },\n      \"WorkflowID\": {\n        \"type\": \"keyword\"\n      },\n      \"WorkflowType\": {\n        \"type\": \"keyword\"\n      }\n    }\n  },\n  \"order\": 0,\n  \"settings\": {\n    \"index\": {\n      \"number_of_replicas\": \"0\",\n      \"number_of_shards\": \"5\"\n    }\n  }\n}\n"
  },
  {
    "path": "schema/elasticsearch/v6/visibility/index_template.json",
    "content": "{\n  \"order\": 0,\n  \"index_patterns\": [\n    \"cadence-visibility-*\"\n  ],\n  \"settings\": {\n    \"index\": {\n      \"number_of_shards\": \"5\",\n      \"number_of_replicas\": \"0\"\n    }\n  },\n  \"mappings\": {\n    \"_doc\": {\n      \"dynamic\": \"false\",\n      \"properties\": {\n        \"DomainID\": {\n          \"type\": \"keyword\"\n        },\n        \"WorkflowID\": {\n          \"type\": \"keyword\"\n        },\n        \"RunID\": {\n          \"type\": \"keyword\"\n        },\n        \"WorkflowType\": {\n          \"type\": \"keyword\"\n        },\n        \"StartTime\": {\n          \"type\": \"long\"\n        },\n        \"ExecutionTime\": {\n          \"type\": \"long\"\n        },\n        \"CloseTime\": {\n          \"type\": \"long\"\n        },\n        \"CloseStatus\": {\n          \"type\": \"integer\"\n        },\n        \"HistoryLength\": {\n          \"type\": \"integer\"\n        },\n        \"KafkaKey\": {\n          \"type\": \"keyword\"\n        },\n        \"TaskList\": {\n          \"type\": \"keyword\"\n        },\n        \"IsCron\": {\n          \"type\": \"boolean\"\n        },\n        \"NumClusters\": {\n          \"type\": \"integer\"\n        },\n        \"ClusterAttributeScope\": {\n          \"type\": \"keyword\"\n        },\n        \"ClusterAttributeName\": {\n          \"type\": \"keyword\"\n        },\n        \"UpdateTime\": {\n          \"type\": \"long\"\n        },\n        \"ShardID\": {\n          \"type\": \"long\"\n        },\n        \"CronSchedule\": {\n          \"type\": \"keyword\"\n        },\n        \"ExecutionStatus\": {\n          \"type\": \"integer\"\n        },\n        \"ScheduledExecutionTime\": {\n          \"type\": \"long\"\n        },\n        \"Attr\": {\n          \"properties\": {\n            \"CadenceChangeVersion\":  { \"type\": \"keyword\" },\n            \"CustomStringField\":  { \"type\": \"text\" },\n            \"CustomKeywordField\": { \"type\": \"keyword\"},\n            \"CustomIntField\": { \"type\": \"long\"},\n            \"CustomBoolField\": { \"type\": \"boolean\"},\n            \"CustomDoubleField\": { \"type\": \"double\"},\n            \"CustomDatetimeField\": { \"type\": \"date\"},\n            \"project\": { \"type\": \"keyword\"},\n            \"service\": { \"type\": \"keyword\"},\n            \"environment\": { \"type\": \"keyword\"},\n            \"addon\": { \"type\": \"keyword\"},\n            \"addon-type\": { \"type\": \"keyword\"},\n            \"user\": { \"type\": \"keyword\"},\n            \"CustomDomain\": { \"type\": \"keyword\"},\n            \"Operator\": { \"type\": \"keyword\"},\n            \"RolloutID\": { \"type\": \"keyword\"},\n            \"BinaryChecksums\": { \"type\": \"keyword\"},\n            \"Passed\": { \"type\": \"boolean\" }\n          }\n        }\n      }\n    }\n  },\n  \"aliases\": {}\n}\n"
  },
  {
    "path": "schema/elasticsearch/v7/visibility/index_template.json",
    "content": "{\n  \"order\": 0,\n  \"index_patterns\": [\n    \"cadence-visibility-*\"\n  ],\n  \"settings\": {\n    \"index\": {\n      \"number_of_shards\": \"5\",\n      \"number_of_replicas\": \"0\"\n    }\n  },\n  \"mappings\": {\n    \"dynamic\": \"false\",\n    \"properties\": {\n      \"DomainID\": {\n        \"type\": \"keyword\"\n      },\n      \"WorkflowID\": {\n        \"type\": \"keyword\"\n      },\n      \"RunID\": {\n        \"type\": \"keyword\"\n      },\n      \"WorkflowType\": {\n        \"type\": \"keyword\"\n      },\n      \"StartTime\": {\n        \"type\": \"long\"\n      },\n      \"ExecutionTime\": {\n        \"type\": \"long\"\n      },\n      \"CloseTime\": {\n        \"type\": \"long\"\n      },\n      \"CloseStatus\": {\n        \"type\": \"integer\"\n      },\n      \"HistoryLength\": {\n        \"type\": \"integer\"\n      },\n      \"KafkaKey\": {\n        \"type\": \"keyword\"\n      },\n      \"TaskList\": {\n        \"type\": \"keyword\"\n      },\n      \"IsCron\": {\n        \"type\": \"boolean\"\n      },\n      \"NumClusters\": {\n        \"type\": \"integer\"\n      },\n      \"ClusterAttributeScope\": {\n        \"type\": \"keyword\"\n      },\n      \"ClusterAttributeName\": {\n        \"type\": \"keyword\"\n      },\n      \"UpdateTime\": {\n        \"type\": \"long\"\n      },\n      \"ShardID\": {\n        \"type\": \"long\"\n      },\n      \"CronSchedule\": {\n        \"type\": \"keyword\"\n      },\n      \"ExecutionStatus\": {\n        \"type\": \"integer\"\n      },\n      \"ScheduledExecutionTime\": {\n        \"type\": \"long\"\n      },\n      \"Attr\": {\n        \"properties\": {\n          \"CadenceChangeVersion\":  { \"type\": \"keyword\" },\n          \"CustomStringField\":  { \"type\": \"text\" },\n          \"CustomKeywordField\": { \"type\": \"keyword\"},\n          \"CustomIntField\": { \"type\": \"long\"},\n          \"CustomBoolField\": { \"type\": \"boolean\"},\n          \"CustomDoubleField\": { \"type\": \"double\"},\n          \"CustomDatetimeField\": { \"type\": \"date\"},\n          \"project\": { \"type\": \"keyword\"},\n          \"service\": { \"type\": \"keyword\"},\n          \"environment\": { \"type\": \"keyword\"},\n          \"addon\": { \"type\": \"keyword\"},\n          \"addon-type\": { \"type\": \"keyword\"},\n          \"user\": { \"type\": \"keyword\"},\n          \"CustomDomain\": { \"type\": \"keyword\"},\n          \"Operator\": { \"type\": \"keyword\"},\n          \"RolloutID\": { \"type\": \"keyword\"},\n          \"BinaryChecksums\": { \"type\": \"keyword\"},\n          \"Passed\": { \"type\": \"boolean\" }\n        }\n      }\n    }\n  },\n  \"aliases\": {}\n}\n"
  },
  {
    "path": "schema/mongodb/README.md",
    "content": "What\n----\nThis directory contains the mongodb schema for every database that cadence owns. The directory structure is as follows\n\n\n```\n./schema\n   - cadence/               -- Contains schema for default data models\n        - schema.json       -- Contains the latest & greatest snapshot of the schema for the keyspace\n        - schema.go         -- Contains the collection schema in Golang structs -- because MongoDB collection is shemaless.  \n        - versioned\n             - v0.1/\n             - v0.2/        -- One directory per schema version change\n             - v1.0/\n                - manifest.json    -- json file describing the change\n                - changes.json     -- changes in this version, only [create collection/index/documents] commands are allowed\n```\n\n## MongoDB JSON schema format\nBelow is an example of a schema JSON file containing two commands, for collection/index/documents creation. \n```json\n[\n  {\n    \"create\": \"collection_name\"\n  },\n  {\n    \"createIndexes\": \"collection_name\",\n    \"indexes\": [\n      {\n        \"key\": {\n          \"fieldnamea\": 1,\n          \"fieldnameb\": -1\n        },\n        \"name\": \"fieldnamea_fieldnameb\"\n      }\n    ],\n    \"writeConcern\": { \"w\": \"majority\" }\n  },\n  {\n    \"insert\": \"collection_name\",\n    \"documents\": [\n      {\n        \"fieldnamea\": 1,\n        \"fieldnameb\": 0,\n        \"fieldnamec\": \"1234\"\n      },\n      {\n        \"fieldnamea\": 2,\n        \"fieldnameb\": 1,\n        \"fieldnamec\": \"12344\"\n      },\n      {\n        \"fieldnamea\": 2,\n        \"fieldnameb\": 2,\n        \"fieldnamec\": \"12345\"\n      }\n    ],\n    \"ordered\": false\n  }\n]\n```\n\n\nHow\n---\n\nQ: How do I update existing schema ?\n* Add your changes to schema.json for snapshot\n* Create a new schema version directory under ./schema/<>/versioned/vx.x\n  * Add a manifest.json\n  * Add your changes in a json file"
  },
  {
    "path": "schema/mongodb/cadence/collectionSchema.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cadence\n\n// below are the names of all mongoDB collections\nconst (\n\tClusterConfigCollectionName = \"cluster_config\"\n)\n\n// NOTE1: MongoDB collection is schemaless -- there is no schema file for collection. We use Go lang structs to define the collection fields.\n\n// NOTE2: MongoDB doesn't allow using camel case or underscore in the field names\n\n// ClusterConfigCollectionEntry is the schema of configStore\n// IMPORTANT: making change to this struct is changing the MongoDB collection schema. Please make sure it's backward compatible(e.g., don't delete the field, or change the annotation value).\ntype ClusterConfigCollectionEntry struct {\n\tID                   int    `json:\"_id,omitempty\"`\n\tRowType              int    `json:\"rowtype\"`\n\tVersion              int64  `json:\"version\"`\n\tData                 []byte `json:\"data\"`\n\tDataEncoding         string `json:\"dataencoding\"`\n\tUnixTimestampSeconds int64  `json:\"unixtimestampseconds\"`\n}\n"
  },
  {
    "path": "schema/mongodb/cadence/schema.json",
    "content": "[\n  {\n    \"create\": \"cluster_config\"\n  },\n  {\n    \"createIndexes\": \"cluster_config\",\n    \"indexes\": [\n      {\n        \"key\": {\n          \"rowtype\": 1,\n          \"version\": -1\n        },\n        \"name\": \"rowtype_version\",\n        \"unique\": true\n      }\n    ],\n    \"writeConcern\": {\n      \"w\": \"majority\"\n    }\n  }\n]"
  },
  {
    "path": "schema/mongodb/cadence/versioned/v0.1/base.json",
    "content": "[\n  {\n    \"create\": \"cluster_config\"\n  },\n  {\n    \"createIndexes\": \"cluster_config\",\n    \"indexes\": [\n      {\n        \"key\": {\n          \"rowtype\": 1,\n          \"version\": -1\n        },\n        \"name\": \"rowtype_version\",\n        \"unique\": true\n      }\n    ],\n    \"writeConcern\": {\n      \"w\": \"majority\"\n    }\n  }\n]"
  },
  {
    "path": "schema/mongodb/cadence/versioned/v0.1/manifest.json",
    "content": "{\n    \"CurrVersion\": \"0.1\",\n    \"MinCompatibleVersion\": \"0.1\",\n    \"Description\": \"base version of schema\",\n    \"SchemaUpdateCqlFiles\": [\n        \"base.json\"\n    ]\n}\n"
  },
  {
    "path": "schema/mongodb/version.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mongodb\n\n// NOTE: whenever there is a new data base schema update, plz update the following versions\n\n// Version is the MongoDB database schema release version\nconst Version = \"0.1\"\n"
  },
  {
    "path": "schema/mysql/embed.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\nimport \"embed\"\n\n//go:embed v8/cadence/* v8/visibility/*\nvar SchemaFS embed.FS\n"
  },
  {
    "path": "schema/mysql/v8/cadence/database.sql",
    "content": "CREATE DATABASE cadence character set utf8;"
  },
  {
    "path": "schema/mysql/v8/cadence/schema.sql",
    "content": "CREATE TABLE domains(\n  shard_id INT NOT NULL DEFAULT 54321,\n  id BINARY(16) NOT NULL,\n  name VARCHAR(255) UNIQUE NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  is_global TINYINT(1) NOT NULL,\n  PRIMARY KEY(shard_id, id)\n);\n\nCREATE TABLE domain_metadata (\n  `id` bigint(20) NOT NULL AUTO_INCREMENT,\n  notification_version BIGINT NOT NULL,\n  PRIMARY KEY (`id`)\n);\n\nINSERT INTO domain_metadata (notification_version) VALUES (1);\n\nCREATE TABLE shards (\n  shard_id INT NOT NULL,\n  --\n  range_id BIGINT NOT NULL,\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks(\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE cross_cluster_tasks(\n  target_cluster VARCHAR(255) NOT NULL,\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (target_cluster, shard_id, task_id)\n);\n\nCREATE TABLE executions(\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  --\n  next_event_id BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions(\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  --\n  run_id BINARY(16) NOT NULL,\n  create_request_id VARCHAR(64) NOT NULL,\n  state INT NOT NULL,\n  close_status INT NOT NULL,\n  start_version BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events (\n  id BIGINT AUTO_INCREMENT NOT NULL,\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (id)\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events(shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks (\n  domain_id BINARY(16) NOT NULL,\n  task_list_name VARCHAR(255) NOT NULL,\n  task_type TINYINT NOT NULL, -- {Activity, Decision}\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  task_type TINYINT NOT NULL, -- {Activity, Decision}\n  --\n  range_id BIGINT NOT NULL,\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks (\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE replication_tasks_dlq (\n  source_cluster_name VARCHAR(255) NOT NULL,\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks (\n  shard_id INT NOT NULL,\n  visibility_timestamp DATETIME(6) NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\nCREATE TABLE activity_info_maps (\n-- each row corresponds to one key of one map<string, ActivityInfo>\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  schedule_id BIGINT NOT NULL,\n--\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  last_heartbeat_details BLOB,\n  last_heartbeat_updated_time DATETIME(6) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  timer_id VARCHAR(255) NOT NULL,\n--\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  first_event_id BIGINT NOT NULL,\n--\n  version BIGINT NOT NULL,\n  next_event_id BIGINT NOT NULL,\n  history MEDIUMBLOB,\n  history_encoding VARCHAR(16) NOT NULL,\n  new_run_history MEDIUMBLOB,\n  new_run_history_encoding VARCHAR(16) NOT NULL DEFAULT 'json',\n  event_store_version          INT NOT NULL, -- indicates which version of event store to query\n  new_run_event_store_version  INT NOT NULL, -- indicates which version of event store to query for new run(continueAsNew)\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  signal_id VARCHAR(64) NOT NULL,\n  --\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node (\n  shard_id       INT NOT NULL,\n  tree_id        BINARY(16) NOT NULL,\n  branch_id      BINARY(16) NOT NULL,\n  node_id        BIGINT NOT NULL,\n  txn_id         BIGINT NOT NULL,\n  --\n  data           MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree (\n  shard_id       INT NOT NULL,\n  tree_id        BINARY(16) NOT NULL,\n  branch_id      BINARY(16) NOT NULL,\n  --\n  data           MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id)\n);\n\nCREATE TABLE queue (\n  queue_type INT NOT NULL,\n  message_id BIGINT NOT NULL,\n  message_payload MEDIUMBLOB NOT NULL,\n  PRIMARY KEY(queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata (\n  queue_type INT NOT NULL,\n  data MEDIUMBLOB NOT NULL,\n  PRIMARY KEY(queue_type)\n);\n\nCREATE TABLE cluster_config (\n  row_type INT NOT NULL,\n  version BIGINT NOT NULL,\n  --\n  timestamp DATETIME(6) NOT NULL,\n  data           MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (row_type, version)\n);\n\nCREATE TABLE domain_audit_log (\n  domain_id               VARCHAR(255) NOT NULL,\n  event_id                VARCHAR(255) NOT NULL,\n  --\n  state_before            BLOB NOT NULL,\n  state_before_encoding   VARCHAR(16) NOT NULL,\n  state_after             BLOB NOT NULL,\n  state_after_encoding    VARCHAR(16) NOT NULL,\n  operation_type          INT NOT NULL,\n  created_time            DATETIME(6) NOT NULL,\n  last_updated_time       DATETIME(6) NOT NULL,\n  identity                VARCHAR(255) NOT NULL,\n  identity_type           VARCHAR(255) NOT NULL,\n  comment                 VARCHAR(255) NOT NULL DEFAULT '',\n  PRIMARY KEY (domain_id, operation_type, created_time, event_id)\n);\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.1/base.sql",
    "content": "CREATE TABLE domains(\n  shard_id INT NOT NULL DEFAULT 54321,\n  id BINARY(16) NOT NULL,\n  name VARCHAR(255) UNIQUE NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  is_global TINYINT(1) NOT NULL,\n  PRIMARY KEY(shard_id, id)\n);\n\nCREATE TABLE domain_metadata (\n  `id` bigint(20) NOT NULL AUTO_INCREMENT,\n  notification_version BIGINT NOT NULL,\n  PRIMARY KEY (`id`)\n);\n\nINSERT INTO domain_metadata (notification_version) VALUES (1);\n\nCREATE TABLE shards (\n  shard_id INT NOT NULL,\n  --\n  range_id BIGINT NOT NULL,\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks(\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE executions(\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  --\n  next_event_id BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions(\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  --\n  run_id BINARY(16) NOT NULL,\n  create_request_id VARCHAR(64) NOT NULL,\n  state INT NOT NULL,\n  close_status INT NOT NULL,\n  start_version BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events (\n  id BIGINT AUTO_INCREMENT NOT NULL,\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (id)\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events(shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks (\n  domain_id BINARY(16) NOT NULL,\n  task_list_name VARCHAR(255) NOT NULL,\n  task_type TINYINT NOT NULL, -- {Activity, Decision}\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  task_type TINYINT NOT NULL, -- {Activity, Decision}\n  --\n  range_id BIGINT NOT NULL,\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks (\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks (\n  shard_id INT NOT NULL,\n  visibility_timestamp DATETIME(6) NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\n-- Deprecated in favor of history eventsV2\nCREATE TABLE events (\n  domain_id      BINARY(16) NOT NULL,\n  workflow_id    VARCHAR(255) NOT NULL,\n  run_id         BINARY(16) NOT NULL,\n  first_event_id BIGINT NOT NULL,\n  --\n  batch_version  BIGINT,\n  range_id       BIGINT NOT NULL,\n  tx_id          BIGINT NOT NULL,\n  data MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n\tPRIMARY KEY (domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE activity_info_maps (\n-- each row corresponds to one key of one map<string, ActivityInfo>\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  schedule_id BIGINT NOT NULL,\n--\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  last_heartbeat_details BLOB,\n  last_heartbeat_updated_time DATETIME(6) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  timer_id VARCHAR(255) NOT NULL,\n--\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  first_event_id BIGINT NOT NULL,\n--\n  version BIGINT NOT NULL,\n  next_event_id BIGINT NOT NULL,\n  history MEDIUMBLOB,\n  history_encoding VARCHAR(16) NOT NULL,\n  new_run_history BLOB,\n  new_run_history_encoding VARCHAR(16) NOT NULL DEFAULT 'json',\n  event_store_version          INT NOT NULL, -- indiciates which version of event store to query\n  new_run_event_store_version  INT NOT NULL, -- indiciates which version of event store to query for new run(continueAsNew)\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets (\n  shard_id INT NOT NULL,\n  domain_id BINARY(16) NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BINARY(16) NOT NULL,\n  signal_id VARCHAR(64) NOT NULL,\n  --\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node (\n  shard_id       INT NOT NULL,\n  tree_id        BINARY(16) NOT NULL,\n  branch_id      BINARY(16) NOT NULL,\n  node_id        BIGINT NOT NULL,\n  txn_id         BIGINT NOT NULL,\n  --\n  data           MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree (\n  shard_id       INT NOT NULL,\n  tree_id        BINARY(16) NOT NULL,\n  branch_id      BINARY(16) NOT NULL,\n  --\n  data           BLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id)\n);"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add queue and queue_metadata table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"queue.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.2/queue.sql",
    "content": "CREATE TABLE queue (\n  queue_type INT NOT NULL,\n  message_id BIGINT NOT NULL,\n  message_payload BLOB NOT NULL,\n  PRIMARY KEY(queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata (\n  queue_type INT NOT NULL,\n  data BLOB NOT NULL,\n  PRIMARY KEY(queue_type)\n);\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.3\",\n  \"Description\": \"add replication_tasks_dlq table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"replication_tasks_dlq.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.3/replication_tasks_dlq.sql",
    "content": "CREATE TABLE replication_tasks_dlq (\n  source_cluster_name VARCHAR(255) NOT NULL,\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.4/blob_size.sql",
    "content": "ALTER TABLE domains MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE shards MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE transfer_tasks MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE executions MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE tasks MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE task_lists MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE replication_tasks MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE replication_tasks_dlq MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE timer_tasks MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE activity_info_maps MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE timer_info_maps MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE child_execution_info_maps MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE request_cancel_info_maps MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE signal_info_maps MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE buffered_replication_task_maps MODIFY COLUMN new_run_history MEDIUMBLOB;\nALTER TABLE history_tree MODIFY COLUMN data MEDIUMBLOB;\nALTER TABLE queue MODIFY COLUMN message_payload MEDIUMBLOB;\nALTER TABLE queue_metadata MODIFY COLUMN data MEDIUMBLOB;"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.3\",\n  \"Description\": \"modify blob size\",\n  \"SchemaUpdateCqlFiles\": [\n    \"blob_size.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.5/cross_cluster_table.sql",
    "content": "CREATE TABLE cross_cluster_tasks(\n  target_cluster VARCHAR(255) NOT NULL,\n  shard_id INT NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data MEDIUMBLOB NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (target_cluster, shard_id, task_id)\n);"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"create cross cluster table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"cross_cluster_table.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.6/cluster_config.sql",
    "content": "CREATE TABLE cluster_config (\n  row_type INT NOT NULL,\n  version BIGINT NOT NULL,\n  --\n  timestamp DATETIME(6) NOT NULL,\n  data           MEDIUMBLOB NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (row_type, version)\n);\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"create cluster config table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"cluster_config.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.7/domain_audit_log.sql",
    "content": "CREATE TABLE domain_audit_log (\n  domain_id               VARCHAR(255) NOT NULL,\n  event_id                VARCHAR(255) NOT NULL,\n  --\n  state_before            BLOB NOT NULL,\n  state_before_encoding   VARCHAR(16) NOT NULL,\n  state_after             BLOB NOT NULL,\n  state_after_encoding    VARCHAR(16) NOT NULL,\n  operation_type          INT NOT NULL,\n  created_time            DATETIME(6) NOT NULL,\n  last_updated_time       DATETIME(6) NOT NULL,\n  identity                VARCHAR(255) NOT NULL,\n  identity_type           VARCHAR(255) NOT NULL,\n  comment                 VARCHAR(255) NOT NULL DEFAULT '',\n  PRIMARY KEY (domain_id, operation_type, created_time, event_id)\n);"
  },
  {
    "path": "schema/mysql/v8/cadence/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"create domain audit log table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_audit_log.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/v8/visibility/database.sql",
    "content": "CREATE DATABASE cadence_visibility character set utf8;"
  },
  {
    "path": "schema/mysql/v8/visibility/schema.sql",
    "content": "CREATE TABLE executions_visibility (\n  domain_id                CHAR(64) NOT NULL,\n  run_id                   CHAR(64) NOT NULL,\n  start_time               DATETIME(6) NOT NULL,\n  execution_time           DATETIME(6) NOT NULL,\n  workflow_id              VARCHAR(255) NOT NULL,\n  workflow_type_name       VARCHAR(255) NOT NULL,\n  close_status             INT,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  close_time               DATETIME(6) NULL,\n  history_length           BIGINT,\n  memo                     BLOB,\n  encoding                 VARCHAR(64) NOT NULL,\n  task_list                VARCHAR(255) DEFAULT '' NOT NULL,\n  is_cron                  BOOLEAN DEFAULT false NOT NULL,\n  num_clusters             INT NULL,\n  update_time              DATETIME(6) NULL,\n  shard_id                 INT NULL,\n  cron_schedule            VARCHAR(255) NULL,\n  execution_status         INT NULL,\n  scheduled_execution_time DATETIME(6) NULL,\n\n  PRIMARY KEY  (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.1/base.sql",
    "content": "CREATE TABLE executions_visibility (\n  domain_id            CHAR(64) NOT NULL,\n  run_id               CHAR(64) NOT NULL,\n  start_time           DATETIME(6) NOT NULL,\n  execution_time       DATETIME(6) NOT NULL,\n  workflow_id          VARCHAR(255) NOT NULL,\n  workflow_type_name   VARCHAR(255) NOT NULL,\n  close_status         INT,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  close_time           DATETIME(6) NULL,\n  history_length       BIGINT,\n  memo                 BLOB,\n  encoding             VARCHAR(64) NOT NULL,\n\n  PRIMARY KEY  (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of visibility schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.2/add_task_list.sql",
    "content": "ALTER TABLE executions_visibility ADD task_list varchar(255) DEFAULT '';"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add task_list field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_task_list.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add close time and closed state index\",\n  \"SchemaUpdateCqlFiles\": [\n    \"vs_index.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.3/vs_index.sql",
    "content": "CREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.4/add_is_cron.sql",
    "content": "ALTER TABLE executions_visibility ADD is_cron BOOLEAN DEFAULT false;"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.4\",\n  \"Description\": \"add is_cron field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_is_cron.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.5/add_num_clusters.sql",
    "content": "ALTER TABLE executions_visibility ADD num_clusters INT;"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"add num_clusters field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_num_clusters.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.6/add_update_time.sql",
    "content": "ALTER TABLE executions_visibility ADD update_time DATETIME(6) DEFAULT CURRENT_TIMESTAMP(6);"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"add update_time field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_update_time.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.7/add_shard_id.sql",
    "content": "ALTER TABLE executions_visibility ADD shard_id INT DEFAULT -1;"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"add shard_id field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_shard_id.sql\"\n  ]\n}"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.8/add_execution_fields.sql",
    "content": "-- Add cron_schedule field to track cron workflow schedules\nALTER TABLE executions_visibility ADD cron_schedule VARCHAR(255) NULL;\n\n-- Add execution_status field to track workflow execution status (Pending, Started, or close status)\nALTER TABLE executions_visibility ADD execution_status INT NULL;\n\n-- Add scheduled_execution_time field to track the actual scheduled execution time (start_time + first_decision_task_backoff)\nALTER TABLE executions_visibility ADD scheduled_execution_time DATETIME(6) NULL;\n"
  },
  {
    "path": "schema/mysql/v8/visibility/versioned/v0.8/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.8\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"add cron_schedule, execution_status, and scheduled_execution_time to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_execution_fields.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/mysql/version.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage mysql\n\n// NOTE: whenever there is a new data base schema update, plz update the following versions\n\n// Version is the MySQL database release version\nconst Version = \"0.7\"\n\n// VisibilityVersion is the MySQL visibility database release version\nconst VisibilityVersion = \"0.8\"\n"
  },
  {
    "path": "schema/pinot/README.md",
    "content": "There are 3 ways to create the tables in Pinot, need to make sure Pinot server is running. To start pinot server, use `docker compose -f ./docker/dev/cassandra-pinot-kafka.yml up`. \nPinot has 4 components need to be started in order, so some components may fail to start for the first time. You can try again or restart the failed components in docker.\n\n1. Manually add schema and realtime table in the UI, the configs can be found in the cadence-visibility-config file.\n\n2. Use Pinot launcher script to add schema and table, this requires to download Apache Pinot.\n\n`sh pinot-admin.sh AddTable \\\n-schemaFile [path_to_cadence]/cadence/Schema/Pinot/cadence-visibility-schema.json \\\n-tableConfigFile [path_to_cadence]/cadence/Schema/Pinot/cadence-visibility-config.json -exec`\n\n3. Use docker to execute the launcher script.\n\n`docker exec -it pinot-controller bin/pinot-admin.sh AddTable   \\\n-tableConfigFile /Schema/Pinot/cadence-visibility-schema.json   \\\n-schemaFile /Schema/Pinot/cadence-visibility-config.json -exec`\n\nThe result can be checked at http://localhost:9000/#/tables"
  },
  {
    "path": "schema/pinot/cadence-visibility-config.json",
    "content": "{\n  \"tableName\": \"cadence_visibility_pinot\",\n  \"tableType\": \"REALTIME\",\n  \"segmentsConfig\": {\n    \"timeColumnName\": \"StartTime\",\n    \"timeType\": \"MILLISECONDS\",\n    \"schemaName\": \"cadence_visibility_pinot\",\n    \"replicasPerPartition\": \"1\"\n  },\n  \"tenants\": {},\n  \"tableIndexConfig\": {\n    \"jsonIndexConfigs\": {\n      \"Attr\": {\n        \"excludeArray\": false,\n        \"disableCrossArrayUnnest\": true,\n        \"includePaths\": null,\n        \"excludePaths\": null,\n        \"excludeFields\": null\n      }\n    },\n    \"loadMode\": \"MMAP\",\n    \"streamConfigs\": {\n      \"streamType\": \"kafka\",\n      \"stream.kafka.consumer.type\": \"lowlevel\",\n      \"stream.kafka.topic.name\": \"cadence-visibility-pinot\",\n      \"stream.kafka.decoder.class.name\": \"org.apache.pinot.plugin.stream.kafka.KafkaJSONMessageDecoder\",\n      \"stream.kafka.hlc.zk.connect.string\": \"zookeeper:2181/kafka\",\n      \"stream.kafka.consumer.factory.class.name\": \"org.apache.pinot.plugin.stream.kafka20.KafkaConsumerFactory\",\n      \"stream.kafka.zk.broker.url\": \"zookeeper:2181/kafka\",\n      \"stream.kafka.broker.list\": \"kafka:9093\"\n    }\n  },\n  \"routing\": {\n    \"instanceSelectorType\": \"strictReplicaGroup\"\n  },\n  \"upsertConfig\": {\n    \"mode\": \"FULL\",\n    \"deleteRecordColumn\": \"IsDeleted\",\n    \"deletedKeysTTL\": 86400,\n    \"hashFunction\": \"NONE\",\n    \"enableSnapshot\": false\n  },\n  \"metadata\": {}\n}\n"
  },
  {
    "path": "schema/pinot/cadence-visibility-schema.json",
    "content": "{\n  \"schemaName\": \"cadence_visibility_pinot\",\n  \"primaryKeyColumns\": [\"RunID\"],\n  \"dimensionFieldSpecs\": [\n    {\n      \"name\": \"DomainID\",\n      \"dataType\": \"STRING\"\n    },\n    {\n      \"name\": \"WorkflowID\",\n      \"dataType\": \"STRING\"\n    },\n    {\n      \"name\": \"RunID\",\n      \"dataType\": \"STRING\"\n    },\n    {\n      \"name\": \"WorkflowType\",\n      \"dataType\": \"STRING\"\n    },\n    {\n      \"name\": \"CloseStatus\",\n      \"dataType\": \"INT\"\n    },\n    {\n      \"name\": \"HistoryLength\",\n      \"dataType\": \"INT\"\n    },\n    {\n      \"name\": \"TaskList\",\n      \"dataType\": \"STRING\"\n    },\n    {\n      \"name\": \"IsCron\",\n      \"dataType\": \"BOOLEAN\"\n    },\n    {\n      \"name\": \"NumClusters\",\n      \"dataType\": \"INT\"\n    },\n    {\n      \"name\": \"ShardID\",\n      \"dataType\": \"INT\"\n    },\n    {\n      \"name\": \"Attr\",\n      \"dataType\": \"JSON\"\n    },\n    {\n      \"name\": \"IsDeleted\",\n      \"dataType\": \"BOOLEAN\"\n    }\n\n  ],\n  \"dateTimeFieldSpecs\": [{\n    \"name\": \"StartTime\",\n    \"dataType\": \"LONG\",\n    \"format\" : \"1:MILLISECONDS:EPOCH\",\n    \"granularity\": \"1:MILLISECONDS\"\n  },{\n    \"name\": \"CloseTime\",\n    \"dataType\": \"LONG\",\n    \"format\" : \"1:MILLISECONDS:EPOCH\",\n    \"granularity\": \"1:MILLISECONDS\"\n  },{\n    \"name\": \"UpdateTime\",\n    \"dataType\": \"LONG\",\n    \"format\" : \"1:MILLISECONDS:EPOCH\",\n    \"granularity\": \"1:MILLISECONDS\"\n  },{\n    \"name\": \"ExecutionTime\",\n    \"dataType\": \"LONG\",\n    \"format\" : \"1:MILLISECONDS:EPOCH\",\n    \"granularity\": \"1:MILLISECONDS\"\n  },{\n    \"name\": \"EventTimeMs\",\n    \"dataType\": \"LONG\",\n    \"format\" : \"1:MILLISECONDS:EPOCH\",\n    \"granularity\": \"1:MILLISECONDS\"\n  }]\n}"
  },
  {
    "path": "schema/pinot/create_pinot_table.sh",
    "content": "#!/bin/bash\n\n# Wait for Pinot components to start\nsleep 30\n\n# Add the table\n/opt/pinot/bin/pinot-admin.sh AddTable \\\n  -schemaFile /schema/pinot/cadence-visibility-schema.json \\\n  -tableConfigFile /schema/pinot/cadence-visibility-config.json \\\n  -controllerProtocol http \\\n  -controllerHost pinot-controller \\\n  -controllerPort 9001 \\\n  -exec"
  },
  {
    "path": "schema/postgres/cadence/database.sql",
    "content": "CREATE DATABASE cadence;"
  },
  {
    "path": "schema/postgres/cadence/schema.sql",
    "content": "CREATE TABLE domains(\n  shard_id INTEGER NOT NULL DEFAULT 54321,\n  id BYTEA NOT NULL,\n  name VARCHAR(255) UNIQUE NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  is_global BOOLEAN NOT NULL,\n  PRIMARY KEY(shard_id, id)\n);\n\nCREATE TABLE domain_metadata (\n  notification_version BIGINT NOT NULL\n);\n\nINSERT INTO domain_metadata (notification_version) VALUES (1);\n\nCREATE TABLE shards (\n  shard_id INTEGER NOT NULL,\n  --\n  range_id BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks(\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE cross_cluster_tasks(\n  target_cluster VARCHAR(255) NOT NULL,\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (target_cluster, shard_id, task_id)\n);\n\nCREATE TABLE executions(\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  --\n  next_event_id BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions(\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  --\n  run_id BYTEA NOT NULL,\n  create_request_id VARCHAR(64) NOT NULL,\n  state INTEGER NOT NULL,\n  close_status INTEGER NOT NULL,\n  start_version BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events (\n  id BIGSERIAL NOT NULL,\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (id)\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events(shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks (\n  domain_id BYTEA NOT NULL,\n  task_list_name VARCHAR(255) NOT NULL,\n  task_type SMALLINT NOT NULL, -- {Activity, Decision}\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  task_type SMALLINT NOT NULL, -- {Activity, Decision}\n  --\n  range_id BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks (\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE replication_tasks_dlq (\n  source_cluster_name VARCHAR(255) NOT NULL,\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks (\n  shard_id INTEGER NOT NULL,\n  visibility_timestamp TIMESTAMP NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\nCREATE TABLE activity_info_maps (\n-- each row corresponds to one key of one map<string, ActivityInfo>\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  schedule_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  last_heartbeat_details BYTEA,\n  last_heartbeat_updated_time TIMESTAMP NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  timer_id VARCHAR(255) NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  first_event_id BIGINT NOT NULL,\n--\n  version BIGINT NOT NULL,\n  next_event_id BIGINT NOT NULL,\n  history BYTEA,\n  history_encoding VARCHAR(16) NOT NULL,\n  new_run_history BYTEA,\n  new_run_history_encoding VARCHAR(16) NOT NULL DEFAULT 'json',\n  event_store_version          INTEGER NOT NULL, -- indiciates which version of event store to query\n  new_run_event_store_version  INTEGER NOT NULL, -- indiciates which version of event store to query for new run(continueAsNew)\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id TEXT NOT NULL,\n  run_id BYTEA NOT NULL,\n  signal_id VARCHAR(64) NOT NULL,\n  --\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node (\n  shard_id       INTEGER NOT NULL,\n  tree_id        BYTEA NOT NULL,\n  branch_id      BYTEA NOT NULL,\n  node_id        BIGINT NOT NULL,\n  txn_id         BIGINT NOT NULL,\n  --\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree (\n  shard_id       INTEGER NOT NULL,\n  tree_id        BYTEA NOT NULL,\n  branch_id      BYTEA NOT NULL,\n  --\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id)\n);\n\nCREATE TABLE queue (\n  queue_type INTEGER NOT NULL,\n  message_id BIGINT NOT NULL,\n  message_payload BYTEA NOT NULL,\n  PRIMARY KEY(queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata (\n  queue_type INTEGER NOT NULL,\n  data BYTEA NOT NULL,\n  PRIMARY KEY(queue_type)\n);\n\nCREATE TABLE cluster_config (\n  row_type INTEGER NOT NULL,\n  version BIGINT NOT NULL,\n  --\n  timestamp TIMESTAMP NOT NULL,\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (row_type, version)\n);\n\nCREATE TABLE domain_audit_log (\n  domain_id               TEXT NOT NULL,\n  event_id                TEXT NOT NULL,\n  --\n  state_before            BYTEA NOT NULL,\n  state_before_encoding   TEXT NOT NULL,\n  state_after             BYTEA NOT NULL,\n  state_after_encoding    TEXT NOT NULL,\n  operation_type          INTEGER NOT NULL,\n  created_time            TIMESTAMP NOT NULL,\n  last_updated_time       TIMESTAMP NOT NULL,\n  identity                TEXT NOT NULL,\n  identity_type           TEXT NOT NULL,\n  comment                 TEXT NOT NULL DEFAULT '',\n  PRIMARY KEY (domain_id, operation_type, created_time, event_id)\n);\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.1/base.sql",
    "content": "CREATE TABLE domains(\n  shard_id INTEGER NOT NULL DEFAULT 54321,\n  id BYTEA NOT NULL,\n  name VARCHAR(255) UNIQUE NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  is_global BOOLEAN NOT NULL,\n  PRIMARY KEY(shard_id, id)\n);\n\nCREATE TABLE domain_metadata (\n  notification_version BIGINT NOT NULL\n);\n\nINSERT INTO domain_metadata (notification_version) VALUES (1);\n\nCREATE TABLE shards (\n  shard_id INTEGER NOT NULL,\n  --\n  range_id BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks(\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE executions(\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  --\n  next_event_id BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions(\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  --\n  run_id BYTEA NOT NULL,\n  create_request_id VARCHAR(64) NOT NULL,\n  state INTEGER NOT NULL,\n  close_status INTEGER NOT NULL,\n  start_version BIGINT NOT NULL,\n  last_write_version BIGINT NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events (\n  id BIGSERIAL NOT NULL,\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (id)\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events(shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks (\n  domain_id BYTEA NOT NULL,\n  task_list_name VARCHAR(255) NOT NULL,\n  task_type SMALLINT NOT NULL, -- {Activity, Decision}\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  task_type SMALLINT NOT NULL, -- {Activity, Decision}\n  --\n  range_id BIGINT NOT NULL,\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks (\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks (\n  shard_id INTEGER NOT NULL,\n  visibility_timestamp TIMESTAMP NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\nCREATE TABLE activity_info_maps (\n-- each row corresponds to one key of one map<string, ActivityInfo>\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  schedule_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  last_heartbeat_details BYTEA,\n  last_heartbeat_updated_time TIMESTAMP NOT NULL,\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  timer_id VARCHAR(255) NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  initiated_id BIGINT NOT NULL,\n--\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16),\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  first_event_id BIGINT NOT NULL,\n--\n  version BIGINT NOT NULL,\n  next_event_id BIGINT NOT NULL,\n  history BYTEA,\n  history_encoding VARCHAR(16) NOT NULL,\n  new_run_history BYTEA,\n  new_run_history_encoding VARCHAR(16) NOT NULL DEFAULT 'json',\n  event_store_version          INTEGER NOT NULL, -- indiciates which version of event store to query\n  new_run_event_store_version  INTEGER NOT NULL, -- indiciates which version of event store to query for new run(continueAsNew)\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets (\n  shard_id INTEGER NOT NULL,\n  domain_id BYTEA NOT NULL,\n  workflow_id VARCHAR(255) NOT NULL,\n  run_id BYTEA NOT NULL,\n  signal_id VARCHAR(64) NOT NULL,\n  --\n  PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node (\n  shard_id       INTEGER NOT NULL,\n  tree_id        BYTEA NOT NULL,\n  branch_id      BYTEA NOT NULL,\n  node_id        BIGINT NOT NULL,\n  txn_id         BIGINT NOT NULL,\n  --\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree (\n  shard_id       INTEGER NOT NULL,\n  tree_id        BYTEA NOT NULL,\n  branch_id      BYTEA NOT NULL,\n  --\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (shard_id, tree_id, branch_id)\n);\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add queue and queue_metadata table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"queue.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.2/queue.sql",
    "content": "CREATE TABLE queue (\n  queue_type INTEGER NOT NULL,\n  message_id BIGINT NOT NULL,\n  message_payload BYTEA NOT NULL,\n  PRIMARY KEY(queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata (\n  queue_type INTEGER NOT NULL,\n  data BYTEA NOT NULL,\n  PRIMARY KEY(queue_type)\n);\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.3\",\n  \"Description\": \"add replication_tasks_dlq table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"replication_tasks_dlq.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.3/replication_tasks_dlq.sql",
    "content": "CREATE TABLE replication_tasks_dlq (\n  source_cluster_name VARCHAR(255) NOT NULL,\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.4/cross_cluster_table.sql",
    "content": "CREATE TABLE cross_cluster_tasks(\n  target_cluster VARCHAR(255) NOT NULL,\n  shard_id INTEGER NOT NULL,\n  task_id BIGINT NOT NULL,\n  --\n  data BYTEA NOT NULL,\n  data_encoding VARCHAR(16) NOT NULL,\n  PRIMARY KEY (target_cluster, shard_id, task_id)\n);"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.4\",\n  \"Description\": \"create cross cluster table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"cross_cluster_table.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.5/cluster_config.sql",
    "content": "CREATE TABLE cluster_config (\n  row_type INTEGER NOT NULL,\n  version BIGINT NOT NULL,\n  --\n  timestamp TIMESTAMP NOT NULL,\n  data           BYTEA NOT NULL,\n  data_encoding  VARCHAR(16) NOT NULL,\n  PRIMARY KEY (row_type, version)\n);\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"create cluster config table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"cluster_config.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.6/extend_workflow_id_length.sql",
    "content": "ALTER TABLE executions ALTER workflow_id TYPE text;\n\nALTER TABLE current_executions ALTER workflow_id TYPE text;\n\nALTER TABLE buffered_events ALTER workflow_id TYPE text;\n\nALTER TABLE activity_info_maps ALTER workflow_id TYPE text;\n\nALTER TABLE timer_info_maps ALTER workflow_id TYPE text;\n\nALTER TABLE child_execution_info_maps ALTER workflow_id TYPE text;\n\nALTER TABLE request_cancel_info_maps ALTER workflow_id TYPE text;\n\nALTER TABLE signal_info_maps ALTER workflow_id TYPE text;\n\nALTER TABLE buffered_replication_task_maps ALTER workflow_id TYPE text;\n\nALTER TABLE signals_requested_sets ALTER workflow_id TYPE text;"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"update type of workflow_id to text to remove length restriction\",\n  \"SchemaUpdateCqlFiles\": [\n    \"extend_workflow_id_length.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.7/domain_audit_log.sql",
    "content": "CREATE TABLE domain_audit_log (\n  domain_id               TEXT NOT NULL,\n  event_id                TEXT NOT NULL,\n  --\n  state_before            BYTEA NOT NULL,\n  state_before_encoding   TEXT NOT NULL,\n  state_after             BYTEA NOT NULL,\n  state_after_encoding    TEXT NOT NULL,\n  operation_type          INTEGER NOT NULL,\n  created_time            TIMESTAMP NOT NULL,\n  last_updated_time       TIMESTAMP NOT NULL,\n  identity                TEXT NOT NULL,\n  identity_type           TEXT NOT NULL,\n  comment                 TEXT NOT NULL DEFAULT '',\n  PRIMARY KEY (domain_id, operation_type, created_time, event_id)\n);\n"
  },
  {
    "path": "schema/postgres/cadence/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"create domain_audit_log table\",\n  \"SchemaUpdateCqlFiles\": [\n    \"domain_audit_log.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/postgres/embed.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\nimport \"embed\"\n\n//go:embed cadence/* visibility/*\nvar SchemaFS embed.FS\n"
  },
  {
    "path": "schema/postgres/version.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage postgres\n\n// NOTE: whenever there is a new data base schema update, plz update the following versions\n\n// Version is the Postgres database release version\n// Cadence supports both MySQL and Postgres officially, so upgrade should be perform for both MySQL and Postgres\nconst Version = \"0.7\"\n\n// VisibilityVersion is the Postgres visibility database release version\n// Cadence supports both MySQL and Postgres officially, so upgrade should be perform for both MySQL and Postgres\nconst VisibilityVersion = \"0.8\"\n"
  },
  {
    "path": "schema/postgres/visibility/database.sql",
    "content": "CREATE DATABASE cadence_visibility;"
  },
  {
    "path": "schema/postgres/visibility/schema.sql",
    "content": "CREATE TABLE executions_visibility (\n  domain_id                CHAR(64) NOT NULL,\n  run_id                   CHAR(64) NOT NULL,\n  start_time               TIMESTAMP NOT NULL,\n  execution_time           TIMESTAMP NOT NULL,\n  workflow_id              TEXT NOT NULL,\n  workflow_type_name       VARCHAR(255) NOT NULL,\n  close_status             INTEGER,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  close_time               TIMESTAMP NULL,\n  history_length           BIGINT,\n  memo                     BYTEA,\n  encoding                 VARCHAR(64) NOT NULL,\n  task_list                VARCHAR(255) DEFAULT '' NOT NULL,\n  is_cron                  BOOLEAN DEFAULT false NOT NULL,\n  num_clusters             INTEGER NULL,\n  update_time              TIMESTAMP NULL,\n  shard_id                 INTEGER NULL,\n  cron_schedule            VARCHAR(255) NULL,\n  execution_status         INTEGER NULL,\n  scheduled_execution_time TIMESTAMP NULL,\n\n  PRIMARY KEY  (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.1/base.sql",
    "content": "CREATE TABLE executions_visibility (\n  domain_id            CHAR(64) NOT NULL,\n  run_id               CHAR(64) NOT NULL,\n  start_time           TIMESTAMP NOT NULL,\n  execution_time       TIMESTAMP NOT NULL,\n  workflow_id          VARCHAR(255) NOT NULL,\n  workflow_type_name   VARCHAR(255) NOT NULL,\n  close_status         INTEGER,  -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n  close_time           TIMESTAMP NULL,\n  history_length       BIGINT,\n  memo                 BYTEA,\n  encoding             VARCHAR(64) NOT NULL,\n\n  PRIMARY KEY  (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);\n"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of visibility schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.2/add_task_list.sql",
    "content": "ALTER TABLE executions_visibility ADD task_list varchar(255) DEFAULT '';"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add task_list field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_task_list.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.3/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.3\",\n  \"MinCompatibleVersion\": \"0.2\",\n  \"Description\": \"add close time and closed state index\",\n  \"SchemaUpdateCqlFiles\": [\n    \"vs_index.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.3/vs_index.sql",
    "content": "CREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.4/add_is_cron.sql",
    "content": "ALTER TABLE executions_visibility ADD is_cron boolean DEFAULT false;"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.4/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.4\",\n  \"MinCompatibleVersion\": \"0.4\",\n  \"Description\": \"add is_cron field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_is_cron.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.5/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.5\",\n  \"MinCompatibleVersion\": \"0.5\",\n  \"Description\": \"add num_clusters field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"num_clusters.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.5/num_clusters.sql",
    "content": "ALTER TABLE executions_visibility ADD num_clusters INTEGER NULL;"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.6/add_update_time.sql",
    "content": "ALTER TABLE executions_visibility ADD update_time TIMESTAMP NULL DEFAULT CURRENT_TIMESTAMP;"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.6/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.6\",\n  \"MinCompatibleVersion\": \"0.6\",\n  \"Description\": \"add update_time field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_update_time.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.7/add_shard_id.sql",
    "content": "ALTER TABLE executions_visibility ADD shard_id INTEGER NULL DEFAULT -1;"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.7/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.7\",\n  \"MinCompatibleVersion\": \"0.7\",\n  \"Description\": \"add shard_id field to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_shard_id.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.8/extend_workflow_id_length.sql",
    "content": "ALTER TABLE executions_visibility ALTER workflow_id TYPE TEXT;\n"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.8/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.8\",\n  \"MinCompatibleVersion\": \"0.8\",\n  \"Description\": \"update type of workflow_id to text to remove length restriction\",\n  \"SchemaUpdateCqlFiles\": [\n    \"extend_workflow_id_length.sql\"\n  ]\n}"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.9/add_execution_fields.sql",
    "content": "-- Add cron_schedule field to track cron workflow schedules\nALTER TABLE executions_visibility ADD cron_schedule VARCHAR(255) NULL;\n\n-- Add execution_status field to track workflow execution status (Pending, Started, or close status)\nALTER TABLE executions_visibility ADD execution_status INTEGER NULL;\n\n-- Add scheduled_execution_time field to track the actual scheduled execution time (start_time + first_decision_task_backoff)\nALTER TABLE executions_visibility ADD scheduled_execution_time TIMESTAMP NULL;\n"
  },
  {
    "path": "schema/postgres/visibility/versioned/v0.9/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.9\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"add cron_schedule, execution_status, and scheduled_execution_time to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_execution_fields.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/sqlite/cadence/schema.sql",
    "content": "CREATE TABLE domains\n(\n    shard_id      INT                 NOT NULL DEFAULT 54321,\n    id            BINARY(16)          NOT NULL,\n    name          VARCHAR(255) UNIQUE NOT NULL,\n    --\n    data          MEDIUMBLOB          NOT NULL,\n    data_encoding VARCHAR(16)         NOT NULL,\n    is_global     TINYINT(1)          NOT NULL,\n    PRIMARY KEY (shard_id, id)\n);\n\nCREATE TABLE domain_metadata\n(\n    id                   INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,\n    notification_version BIGINT                            NOT NULL\n);\n\nINSERT INTO domain_metadata (notification_version)\nVALUES (1);\n\nCREATE TABLE shards\n(\n    shard_id      INT         NOT NULL,\n    --\n    range_id      BIGINT      NOT NULL,\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks\n(\n    shard_id      INT         NOT NULL,\n    task_id       BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE cross_cluster_tasks\n(\n    target_cluster VARCHAR(255) NOT NULL,\n    shard_id       INT          NOT NULL,\n    task_id        BIGINT       NOT NULL,\n    --\n    data           MEDIUMBLOB   NOT NULL,\n    data_encoding  VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (target_cluster, shard_id, task_id)\n);\n\nCREATE TABLE executions\n(\n    shard_id           INT          NOT NULL,\n    domain_id          BINARY(16)   NOT NULL,\n    workflow_id        VARCHAR(255) NOT NULL,\n    run_id             BINARY(16)   NOT NULL,\n    --\n    next_event_id      BIGINT       NOT NULL,\n    last_write_version BIGINT       NOT NULL,\n    data               MEDIUMBLOB   NOT NULL,\n    data_encoding      VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions\n(\n    shard_id           INT          NOT NULL,\n    domain_id          BINARY(16)   NOT NULL,\n    workflow_id        VARCHAR(255) NOT NULL,\n    --\n    run_id             BINARY(16)   NOT NULL,\n    create_request_id  VARCHAR(64)  NOT NULL,\n    state              INT          NOT NULL,\n    close_status       INT          NOT NULL,\n    start_version      BIGINT       NOT NULL,\n    last_write_version BIGINT       NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events\n(\n    id            INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,\n    shard_id      INT                               NOT NULL,\n    domain_id     BINARY(16)                        NOT NULL,\n    workflow_id   VARCHAR(255)                      NOT NULL,\n    run_id        BINARY(16)                        NOT NULL,\n    --\n    data          MEDIUMBLOB                        NOT NULL,\n    data_encoding VARCHAR(16)                       NOT NULL\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events (shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks\n(\n    domain_id      BINARY(16)   NOT NULL,\n    task_list_name VARCHAR(255) NOT NULL,\n    task_type      TINYINT      NOT NULL, -- {Activity, Decision}\n    task_id        BIGINT       NOT NULL,\n    --\n    data           MEDIUMBLOB   NOT NULL,\n    data_encoding  VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    name          VARCHAR(255) NOT NULL,\n    task_type     TINYINT      NOT NULL, -- {Activity, Decision}\n    --\n    range_id      BIGINT       NOT NULL,\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks\n(\n    shard_id      INT         NOT NULL,\n    task_id       BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE replication_tasks_dlq\n(\n    source_cluster_name VARCHAR(255) NOT NULL,\n    shard_id            INT          NOT NULL,\n    task_id             BIGINT       NOT NULL,\n    --\n    data                MEDIUMBLOB   NOT NULL,\n    data_encoding       VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks\n(\n    shard_id             INT         NOT NULL,\n    visibility_timestamp DATETIME(6) NOT NULL,\n    task_id              BIGINT      NOT NULL,\n    --\n    data                 MEDIUMBLOB  NOT NULL,\n    data_encoding        VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\nCREATE TABLE activity_info_maps\n(\n-- each row corresponds to one key of one map<string, ActivityInfo>\n    shard_id                    INT          NOT NULL,\n    domain_id                   BINARY(16)   NOT NULL,\n    workflow_id                 VARCHAR(255) NOT NULL,\n    run_id                      BINARY(16)   NOT NULL,\n    schedule_id                 BIGINT       NOT NULL,\n--\n    data                        MEDIUMBLOB   NOT NULL,\n    data_encoding               VARCHAR(16),\n    last_heartbeat_details      BLOB,\n    last_heartbeat_updated_time DATETIME(6)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    timer_id      VARCHAR(255) NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps\n(\n    shard_id                    INT          NOT NULL,\n    domain_id                   BINARY(16)   NOT NULL,\n    workflow_id                 VARCHAR(255) NOT NULL,\n    run_id                      BINARY(16)   NOT NULL,\n    first_event_id              BIGINT       NOT NULL,\n--\n    version                     BIGINT       NOT NULL,\n    next_event_id               BIGINT       NOT NULL,\n    history                     MEDIUMBLOB,\n    history_encoding            VARCHAR(16)  NOT NULL,\n    new_run_history             MEDIUMBLOB,\n    new_run_history_encoding    VARCHAR(16)  NOT NULL DEFAULT 'json',\n    event_store_version         INT          NOT NULL, -- indicates which version of event store to query\n    new_run_event_store_version INT          NOT NULL, -- indicates which version of event store to query for new run(continueAsNew)\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets\n(\n    shard_id    INT          NOT NULL,\n    domain_id   BINARY(16)   NOT NULL,\n    workflow_id VARCHAR(255) NOT NULL,\n    run_id      BINARY(16)   NOT NULL,\n    signal_id   VARCHAR(64)  NOT NULL,\n    --\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node\n(\n    shard_id      INT         NOT NULL,\n    tree_id       BINARY(16)  NOT NULL,\n    branch_id     BINARY(16)  NOT NULL,\n    node_id       BIGINT      NOT NULL,\n    txn_id        BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree\n(\n    shard_id      INT         NOT NULL,\n    tree_id       BINARY(16)  NOT NULL,\n    branch_id     BINARY(16)  NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, tree_id, branch_id)\n);\n\nCREATE TABLE queue\n(\n    queue_type      INT        NOT NULL,\n    message_id      BIGINT     NOT NULL,\n    message_payload MEDIUMBLOB NOT NULL,\n    PRIMARY KEY (queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata\n(\n    queue_type INT        NOT NULL,\n    data       MEDIUMBLOB NOT NULL,\n    PRIMARY KEY (queue_type)\n);\n\nCREATE TABLE cluster_config\n(\n    row_type      INT         NOT NULL,\n    version       BIGINT      NOT NULL,\n    --\n    timestamp     DATETIME(6) NOT NULL,\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (row_type, version)\n);\n"
  },
  {
    "path": "schema/sqlite/cadence/versioned/v0.1/base.sql",
    "content": "CREATE TABLE domains\n(\n    shard_id      INT                 NOT NULL DEFAULT 54321,\n    id            BINARY(16)          NOT NULL,\n    name          VARCHAR(255) UNIQUE NOT NULL,\n    --\n    data          MEDIUMBLOB          NOT NULL,\n    data_encoding VARCHAR(16)         NOT NULL,\n    is_global     TINYINT(1)          NOT NULL,\n    PRIMARY KEY (shard_id, id)\n);\n\nCREATE TABLE domain_metadata\n(\n    id                   INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,\n    notification_version BIGINT                            NOT NULL\n);\n\nINSERT INTO domain_metadata (notification_version)\nVALUES (1);\n\nCREATE TABLE shards\n(\n    shard_id      INT         NOT NULL,\n    --\n    range_id      BIGINT      NOT NULL,\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id)\n);\n\nCREATE TABLE transfer_tasks\n(\n    shard_id      INT         NOT NULL,\n    task_id       BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE cross_cluster_tasks\n(\n    target_cluster VARCHAR(255) NOT NULL,\n    shard_id       INT          NOT NULL,\n    task_id        BIGINT       NOT NULL,\n    --\n    data           MEDIUMBLOB   NOT NULL,\n    data_encoding  VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (target_cluster, shard_id, task_id)\n);\n\nCREATE TABLE executions\n(\n    shard_id           INT          NOT NULL,\n    domain_id          BINARY(16)   NOT NULL,\n    workflow_id        VARCHAR(255) NOT NULL,\n    run_id             BINARY(16)   NOT NULL,\n    --\n    next_event_id      BIGINT       NOT NULL,\n    last_write_version BIGINT       NOT NULL,\n    data               MEDIUMBLOB   NOT NULL,\n    data_encoding      VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id)\n);\n\nCREATE TABLE current_executions\n(\n    shard_id           INT          NOT NULL,\n    domain_id          BINARY(16)   NOT NULL,\n    workflow_id        VARCHAR(255) NOT NULL,\n    --\n    run_id             BINARY(16)   NOT NULL,\n    create_request_id  VARCHAR(64)  NOT NULL,\n    state              INT          NOT NULL,\n    close_status       INT          NOT NULL,\n    start_version      BIGINT       NOT NULL,\n    last_write_version BIGINT       NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id)\n);\n\nCREATE TABLE buffered_events\n(\n    id            INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,\n    shard_id      INT                               NOT NULL,\n    domain_id     BINARY(16)                        NOT NULL,\n    workflow_id   VARCHAR(255)                      NOT NULL,\n    run_id        BINARY(16)                        NOT NULL,\n    --\n    data          MEDIUMBLOB                        NOT NULL,\n    data_encoding VARCHAR(16)                       NOT NULL\n);\n\nCREATE INDEX buffered_events_by_events_ids ON buffered_events (shard_id, domain_id, workflow_id, run_id);\n\nCREATE TABLE tasks\n(\n    domain_id      BINARY(16)   NOT NULL,\n    task_list_name VARCHAR(255) NOT NULL,\n    task_type      TINYINT      NOT NULL, -- {Activity, Decision}\n    task_id        BIGINT       NOT NULL,\n    --\n    data           MEDIUMBLOB   NOT NULL,\n    data_encoding  VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (domain_id, task_list_name, task_type, task_id)\n);\n\nCREATE TABLE task_lists\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    name          VARCHAR(255) NOT NULL,\n    task_type     TINYINT      NOT NULL, -- {Activity, Decision}\n    --\n    range_id      BIGINT       NOT NULL,\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, name, task_type)\n);\n\nCREATE TABLE replication_tasks\n(\n    shard_id      INT         NOT NULL,\n    task_id       BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, task_id)\n);\n\nCREATE TABLE replication_tasks_dlq\n(\n    source_cluster_name VARCHAR(255) NOT NULL,\n    shard_id            INT          NOT NULL,\n    task_id             BIGINT       NOT NULL,\n    --\n    data                MEDIUMBLOB   NOT NULL,\n    data_encoding       VARCHAR(16)  NOT NULL,\n    PRIMARY KEY (source_cluster_name, shard_id, task_id)\n);\n\nCREATE TABLE timer_tasks\n(\n    shard_id             INT         NOT NULL,\n    visibility_timestamp DATETIME(6) NOT NULL,\n    task_id              BIGINT      NOT NULL,\n    --\n    data                 MEDIUMBLOB  NOT NULL,\n    data_encoding        VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, visibility_timestamp, task_id)\n);\n\nCREATE TABLE activity_info_maps\n(\n-- each row corresponds to one key of one map<string, ActivityInfo>\n    shard_id                    INT          NOT NULL,\n    domain_id                   BINARY(16)   NOT NULL,\n    workflow_id                 VARCHAR(255) NOT NULL,\n    run_id                      BINARY(16)   NOT NULL,\n    schedule_id                 BIGINT       NOT NULL,\n--\n    data                        MEDIUMBLOB   NOT NULL,\n    data_encoding               VARCHAR(16),\n    last_heartbeat_details      BLOB,\n    last_heartbeat_updated_time DATETIME(6)  NOT NULL,\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, schedule_id)\n);\n\nCREATE TABLE timer_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    timer_id      VARCHAR(255) NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, timer_id)\n);\n\nCREATE TABLE child_execution_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE request_cancel_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE signal_info_maps\n(\n    shard_id      INT          NOT NULL,\n    domain_id     BINARY(16)   NOT NULL,\n    workflow_id   VARCHAR(255) NOT NULL,\n    run_id        BINARY(16)   NOT NULL,\n    initiated_id  BIGINT       NOT NULL,\n--\n    data          MEDIUMBLOB   NOT NULL,\n    data_encoding VARCHAR(16),\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, initiated_id)\n);\n\nCREATE TABLE buffered_replication_task_maps\n(\n    shard_id                    INT          NOT NULL,\n    domain_id                   BINARY(16)   NOT NULL,\n    workflow_id                 VARCHAR(255) NOT NULL,\n    run_id                      BINARY(16)   NOT NULL,\n    first_event_id              BIGINT       NOT NULL,\n--\n    version                     BIGINT       NOT NULL,\n    next_event_id               BIGINT       NOT NULL,\n    history                     MEDIUMBLOB,\n    history_encoding            VARCHAR(16)  NOT NULL,\n    new_run_history             MEDIUMBLOB,\n    new_run_history_encoding    VARCHAR(16)  NOT NULL DEFAULT 'json',\n    event_store_version         INT          NOT NULL, -- indicates which version of event store to query\n    new_run_event_store_version INT          NOT NULL, -- indicates which version of event store to query for new run(continueAsNew)\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, first_event_id)\n);\n\nCREATE TABLE signals_requested_sets\n(\n    shard_id    INT          NOT NULL,\n    domain_id   BINARY(16)   NOT NULL,\n    workflow_id VARCHAR(255) NOT NULL,\n    run_id      BINARY(16)   NOT NULL,\n    signal_id   VARCHAR(64)  NOT NULL,\n    --\n    PRIMARY KEY (shard_id, domain_id, workflow_id, run_id, signal_id)\n);\n\n-- history eventsV2: history_node stores history event data\nCREATE TABLE history_node\n(\n    shard_id      INT         NOT NULL,\n    tree_id       BINARY(16)  NOT NULL,\n    branch_id     BINARY(16)  NOT NULL,\n    node_id       BIGINT      NOT NULL,\n    txn_id        BIGINT      NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, tree_id, branch_id, node_id, txn_id)\n);\n\n-- history eventsV2: history_tree stores branch metadata\nCREATE TABLE history_tree\n(\n    shard_id      INT         NOT NULL,\n    tree_id       BINARY(16)  NOT NULL,\n    branch_id     BINARY(16)  NOT NULL,\n    --\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (shard_id, tree_id, branch_id)\n);\n\nCREATE TABLE queue\n(\n    queue_type      INT        NOT NULL,\n    message_id      BIGINT     NOT NULL,\n    message_payload MEDIUMBLOB NOT NULL,\n    PRIMARY KEY (queue_type, message_id)\n);\n\nCREATE TABLE queue_metadata\n(\n    queue_type INT        NOT NULL,\n    data       MEDIUMBLOB NOT NULL,\n    PRIMARY KEY (queue_type)\n);\n\nCREATE TABLE cluster_config\n(\n    row_type      INT         NOT NULL,\n    version       BIGINT      NOT NULL,\n    --\n    timestamp     DATETIME(6) NOT NULL,\n    data          MEDIUMBLOB  NOT NULL,\n    data_encoding VARCHAR(16) NOT NULL,\n    PRIMARY KEY (row_type, version)\n);\n"
  },
  {
    "path": "schema/sqlite/cadence/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}\n"
  },
  {
    "path": "schema/sqlite/embed.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqlite\n\nimport \"embed\"\n\n//go:embed cadence/* visibility/*\nvar SchemaFS embed.FS\n"
  },
  {
    "path": "schema/sqlite/version.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sqlite\n\n// NOTE: whenever there is a new data base schema update, plz update the following versions\n\n// Version is the SQLite database release version\nconst Version = \"0.1\"\n\n// VisibilityVersion is the SQLite visibility database release version\nconst VisibilityVersion = \"0.1\"\n"
  },
  {
    "path": "schema/sqlite/visibility/schema.sql",
    "content": "CREATE TABLE executions_visibility\n(\n    domain_id                CHAR(64)                   NOT NULL,\n    run_id                   CHAR(64)                   NOT NULL,\n    start_time               DATETIME(6)                NOT NULL,\n    execution_time           DATETIME(6)                NOT NULL,\n    workflow_id              VARCHAR(255)               NOT NULL,\n    workflow_type_name       VARCHAR(255)               NOT NULL,\n    close_status             INT, -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n    close_time               DATETIME(6)                NULL,\n    history_length           BIGINT,\n    memo                     BLOB,\n    encoding                 VARCHAR(64)                NOT NULL,\n    task_list                VARCHAR(255) DEFAULT ''    NOT NULL,\n    is_cron                  BOOLEAN      DEFAULT false NOT NULL,\n    num_clusters             INT                        NULL,\n    update_time              DATETIME(6)                NULL,\n    shard_id                 INT                        NULL,\n    cron_schedule            TEXT                       NULL,\n    execution_status         INT                        NULL,\n    scheduled_execution_time TIMESTAMP                  NULL,\n\n    PRIMARY KEY (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/sqlite/visibility/versioned/v0.1/base.sql",
    "content": "CREATE TABLE executions_visibility\n(\n    domain_id          CHAR(64)                   NOT NULL,\n    run_id             CHAR(64)                   NOT NULL,\n    start_time         DATETIME(6)                NOT NULL,\n    execution_time     DATETIME(6)                NOT NULL,\n    workflow_id        VARCHAR(255)               NOT NULL,\n    workflow_type_name VARCHAR(255)               NOT NULL,\n    close_status       INT, -- enum WorkflowExecutionCloseStatus {COMPLETED, FAILED, CANCELED, TERMINATED, CONTINUED_AS_NEW, TIMED_OUT}\n    close_time         DATETIME(6)                NULL,\n    history_length     BIGINT,\n    memo               BLOB,\n    encoding           VARCHAR(64)                NOT NULL,\n    task_list          VARCHAR(255) DEFAULT ''    NOT NULL,\n    is_cron            BOOLEAN      DEFAULT false NOT NULL,\n    num_clusters       INT                        NULL,\n    update_time        DATETIME(6)                NULL,\n    shard_id           INT                        NULL,\n\n    PRIMARY KEY (domain_id, run_id)\n);\n\nCREATE INDEX by_type_start_time ON executions_visibility (domain_id, workflow_type_name, close_status, start_time DESC, run_id);\nCREATE INDEX by_workflow_id_start_time ON executions_visibility (domain_id, workflow_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_status_by_close_time ON executions_visibility (domain_id, close_status, start_time DESC, run_id);\nCREATE INDEX by_close_time_by_status ON executions_visibility (domain_id, close_time DESC, run_id, close_status);\n"
  },
  {
    "path": "schema/sqlite/visibility/versioned/v0.1/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.1\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"base version of visibility schema\",\n  \"SchemaUpdateCqlFiles\": [\n    \"base.sql\"\n  ]\n}"
  },
  {
    "path": "schema/sqlite/visibility/versioned/v0.2/add_execution_fields.sql",
    "content": "-- Add cron_schedule field to track cron workflow schedules\nALTER TABLE executions_visibility ADD cron_schedule TEXT;\n\n-- Add execution_status field to track workflow execution status (Pending, Started, or close status)\nALTER TABLE executions_visibility ADD execution_status INTEGER;\n\n-- Add scheduled_execution_time field to track the actual scheduled execution time (start_time + first_decision_task_backoff)\nALTER TABLE executions_visibility ADD scheduled_execution_time TIMESTAMP;\n"
  },
  {
    "path": "schema/sqlite/visibility/versioned/v0.2/manifest.json",
    "content": "{\n  \"CurrVersion\": \"0.2\",\n  \"MinCompatibleVersion\": \"0.1\",\n  \"Description\": \"add cron_schedule, execution_status, and scheduled_execution_time to visibility\",\n  \"SchemaUpdateCqlFiles\": [\n    \"add_execution_fields.sql\"\n  ]\n}\n"
  },
  {
    "path": "scripts/build-with-ldflags.sh",
    "content": "#!/bin/bash\n\nset -eo pipefail\n\n# add verbose env var for debugging\ntest -n \"$V\" && set -x\n\n# must be on a separate line to trigger -e if it fails\nFLAGS=\"$(./scripts/get-ldflags.sh LDFLAG)\"\nexec go build -ldflags \"$FLAGS\" \"$@\"\n"
  },
  {
    "path": "scripts/check-go-toolchain.sh",
    "content": "#!/usr/bin/env bash\n\nset -eo pipefail\n\nfail=0\nbad() {\n  echo -e \"$@\" >/dev/stderr\n  fail=1\n}\n\nif [[ $V == 1 ]]; then\n  set -x\nfi\n\ntarget=\"${1#go}\"\nroot=\"$(git rev-parse --show-toplevel)\"\n\n# this SHOULD match the dependencies in the goversion-lint check to avoid skipping it\n\n# check dockerfiles\nwhile read file; do\n  # find \"FROM golang:1.22.3-alpine3.18 ...\" lines\n  line=\"$(grep -i 'from golang:' \"$file\")\"\n  # remove \"from golang:\" prefix\n  version=\"${line#*golang:}\"\n  # remove \"-alpine...\" suffix\n  version=\"${version%-*}\"\n  # and make sure it matches\n  if [[ \"$version\" != \"$target\" ]]; then\n    bad \"Wrong Go version in file $file:\\n\\t$line\"\n  fi\ndone < <( find \"$root\" -name Dockerfile )\n\ndeclare -a codecov_files=( \"$root/.github/workflows/codecov-on-pr.yml\" \"$root/.github/workflows/codecov-on-master.yml\" );\n\nfor codecov_file in \"${codecov_files[@]}\"; do\n  # check workflows\n  codecov_file=\"$root/.github/workflows/codecov-on-pr.yml\"\n  codecov_line=\"$(grep 'go-version:' \"$codecov_file\")\"\n  codecov_version=\"${codecov_line#*go-version: }\"\n  if [[ \"$codecov_version\" != \"$target\" ]]; then\n    bad \"Wrong Go version in file $codecov_file:\\n\\t$codecov_line\"\n  fi\ndone\n\nif [[ $fail == 1 ]]; then\n  bad \"Makefile pins Go to go${target}, Dockerfiles and GitHub workflows should too.\"\n  bad \"Non-matching versions lead to pointless double-downloading to get the correct version.\"\n  exit 1\nfi\n"
  },
  {
    "path": "scripts/check-gomod-version.sh",
    "content": "#!/usr/bin/env bash\nset -eo pipefail\n\n[[ $2 = \"-v\" ]] && set -x;\n\n# go.work and `go list -modfile=...` seem to interact badly, and complain about duplicates.\n# the good news is that you can just drop that and `cd` to the folder and it works.\n\nif ! gomod=\"$(go list -mod=readonly -f '{{ .Module }}' \"$1\")\"; then\n    >&2 echo 'Error checking main go.mod.'\n    exit 1\nfi\n\nif ! toolmod=\"$(cd internal/tools; GOTOOLCHAIN=auto go list -mod=readonly -f '{{ .Module }}' \"$1\")\"; then\n    >&2 echo 'Error checking tools go.mod, cd to internal/tools to modify it.'\n    exit 1\nfi\n\nif [[ $gomod != \"$toolmod\" ]]; then\n\t>&2 echo \"error: mismatched go.mod and tools go.mod\"\n    >&2 echo \"ensure internal/tools/go.mod contains the same version as go.mod and try again:\"\n    >&2 echo -e \"\\t$gomod\"\n\texit 1\nfi\n"
  },
  {
    "path": "scripts/docker-build.sh",
    "content": "#!/bin/bash\n\nset -ex\n\n# This script can be used to locally build all the images. Do NOT push these images. Dockerhub registry should only be updated via github workflow.\n\necho \"Building docker images...\"\n\ndocker build . -f Dockerfile -t ubercadence/server:master --build-arg TARGET=server\ndocker build . -f Dockerfile -t ubercadence/server:master-auto-setup --build-arg TARGET=auto-setup\ndocker build . -f Dockerfile -t ubercadence/cli:master --build-arg TARGET=cli\ndocker build . -f Dockerfile -t ubercadence/cadence-bench:master --build-arg TARGET=bench\ndocker build . -f Dockerfile -t ubercadence/cadence-canary:master --build-arg TARGET=canary\n"
  },
  {
    "path": "scripts/generate_cluster_attributes.sh",
    "content": "#!/bin/bash\n\n# Script to generate a domain update command with 3000 cluster attributes\n# Each attribute is a unique city assigned to one of three clusters\n\nset -e\n\nDOMAIN=\"${1:-test25}\"\nNUM_ATTRIBUTES=\"${2:-3000}\"\n\necho \"Generating command with $NUM_ATTRIBUTES cluster attributes for domain: $DOMAIN\"\necho \"\"\n\n# Array of city names from around the world\n# We'll use these as base names and add suffixes to reach 3000\nCITIES=(\n    # Major world cities\n    \"london\" \"paris\" \"berlin\" \"madrid\" \"rome\" \"moscow\" \"istanbul\" \"athens\" \"vienna\" \"prague\"\n    \"amsterdam\" \"brussels\" \"copenhagen\" \"dublin\" \"helsinki\" \"lisbon\" \"oslo\" \"stockholm\" \"warsaw\" \"budapest\"\n    \"tokyo\" \"beijing\" \"shanghai\" \"seoul\" \"bangkok\" \"singapore\" \"hong_kong\" \"mumbai\" \"delhi\" \"bangalore\"\n    \"sydney\" \"melbourne\" \"auckland\" \"wellington\" \"brisbane\" \"perth\" \"adelaide\" \"canberra\"\n    \"new_york\" \"los_angeles\" \"chicago\" \"houston\" \"phoenix\" \"philadelphia\" \"san_antonio\" \"san_diego\" \"dallas\" \"san_jose\"\n    \"austin\" \"jacksonville\" \"fort_worth\" \"columbus\" \"charlotte\" \"san_francisco\" \"indianapolis\" \"seattle\" \"denver\" \"washington\"\n    \"boston\" \"el_paso\" \"nashville\" \"detroit\" \"portland\" \"las_vegas\" \"memphis\" \"louisville\" \"baltimore\" \"milwaukee\"\n    \"toronto\" \"montreal\" \"vancouver\" \"calgary\" \"ottawa\" \"edmonton\" \"winnipeg\" \"quebec_city\" \"hamilton\" \"victoria\"\n    \"mexico_city\" \"guadalajara\" \"monterrey\" \"puebla\" \"tijuana\" \"leon\" \"ciudad_juarez\" \"zapopan\" \"merida\" \"san_luis_potosi\"\n    \"sao_paulo\" \"rio_de_janeiro\" \"brasilia\" \"salvador\" \"fortaleza\" \"belo_horizonte\" \"manaus\" \"curitiba\" \"recife\" \"porto_alegre\"\n    \"buenos_aires\" \"cordoba\" \"rosario\" \"mendoza\" \"la_plata\" \"mar_del_plata\" \"salta\" \"santa_fe\" \"san_juan\" \"resistencia\"\n    \"bogota\" \"medellin\" \"cali\" \"barranquilla\" \"cartagena\" \"cucuta\" \"bucaramanga\" \"pereira\" \"ibague\" \"santa_marta\"\n    \"lima\" \"arequipa\" \"trujillo\" \"chiclayo\" \"piura\" \"iquitos\" \"cusco\" \"huancayo\" \"tacna\" \"ica\"\n    \"santiago\" \"valparaiso\" \"concepcion\" \"la_serena\" \"antofagasta\" \"temuco\" \"rancagua\" \"talca\" \"arica\" \"puerto_montt\"\n    \"caracas\" \"maracaibo\" \"valencia\" \"barquisimeto\" \"maracay\" \"ciudad_guayana\" \"maturin\" \"barcelona\" \"puerto_la_cruz\"\n    \"quito\" \"guayaquil\" \"cuenca\" \"santo_domingo\" \"machala\" \"duran\" \"portoviejo\" \"manta\" \"loja\" \"ambato\"\n    \"cairo\" \"alexandria\" \"giza\" \"shubra_el_kheima\" \"port_said\" \"suez\" \"luxor\" \"al_mahallah_al_kubra\" \"tanta\" \"asyut\"\n    \"johannesburg\" \"cape_town\" \"durban\" \"pretoria\" \"port_elizabeth\" \"pietermaritzburg\" \"benoni\" \"tembisa\" \"east_london\" \"vereeniging\"\n    \"lagos\" \"kano\" \"ibadan\" \"kaduna\" \"port_harcourt\" \"benin_city\" \"maiduguri\" \"zaria\" \"aba\" \"jos\"\n    \"nairobi\" \"mombasa\" \"kisumu\" \"nakuru\" \"eldoret\" \"thika\" \"malindi\" \"kitale\" \"garissa\" \"kakamega\"\n    \"addis_ababa\" \"dire_dawa\" \"mekele\" \"gondar\" \"awasa\" \"bahir_dar\" \"dessie\" \"jimma\" \"jijiga\" \"shashamane\"\n    \"kinshasa\" \"lubumbashi\" \"mbuji_mayi\" \"kisangani\" \"kananga\" \"likasi\" \"kolwezi\" \"tshikapa\" \"beni\" \"bukavu\"\n    \"dar_es_salaam\" \"mwanza\" \"arusha\" \"dodoma\" \"mbeya\" \"morogoro\" \"tanga\" \"kahama\" \"tabora\" \"zanzibar\"\n    \"kampala\" \"gulu\" \"lira\" \"mbarara\" \"jinja\" \"bwizibwera\" \"mbale\" \"mukono\" \"kasese\" \"masaka\"\n    \"khartoum\" \"omdurman\" \"khartoum_north\" \"nyala\" \"port_sudan\" \"kassala\" \"el_obeid\" \"gedaref\" \"wad_medani\" \"al_qadarif\"\n    \"accra\" \"kumasi\" \"tamale\" \"sekondi_takoradi\" \"ashaiman\" \"sunyani\" \"cape_coast\" \"obuasi\" \"teshie\" \"tema\"\n    \"algiers\" \"oran\" \"constantine\" \"annaba\" \"blida\" \"batna\" \"djelfa\" \"setif\" \"sidi_bel_abbes\" \"biskra\"\n    \"casablanca\" \"rabat\" \"fes\" \"marrakech\" \"agadir\" \"tangier\" \"meknes\" \"oujda\" \"kenitra\" \"tetouan\"\n    \"tunis\" \"sfax\" \"sousse\" \"ettadhamen\" \"kairouan\" \"bizerte\" \"gabes\" \"ariana\" \"gafsa\" \"kasserine\"\n    \"tripoli\" \"benghazi\" \"misrata\" \"tarhuna\" \"al_khums\" \"az_zawiya\" \"ajdabiya\" \"tobruk\" \"Sabha\" \"gharyan\"\n    \"tehran\" \"mashhad\" \"isfahan\" \"karaj\" \"tabriz\" \"shiraz\" \"qom\" \"ahvaz\" \"kermanshah\" \"urmia\"\n    \"baghdad\" \"basra\" \"mosul\" \"erbil\" \"kirkuk\" \"najaf\" \"karbala\" \"sulaymaniyah\" \"nasiriyah\" \"amarah\"\n    \"riyadh\" \"jeddah\" \"mecca\" \"medina\" \"dammam\" \"taif\" \"tabuk\" \"khamis_mushait\" \"buraidah\" \"al_hofuf\"\n    \"dubai\" \"abu_dhabi\" \"sharjah\" \"al_ain\" \"ajman\" \"ras_al_khaimah\" \"fujairah\" \"umm_al_quwain\" \"khor_fakkan\" \"dibba\"\n    \"kabul\" \"kandahar\" \"herat\" \"mazar_i_sharif\" \"kunduz\" \"jalalabad\" \"lashkar_gah\" \"taloqan\" \"puli_khumri\" \"ghazni\"\n    \"karachi\" \"lahore\" \"faisalabad\" \"rawalpindi\" \"multan\" \"hyderabad\" \"gujranwala\" \"peshawar\" \"quetta\" \"islamabad\"\n    \"dhaka\" \"chittagong\" \"khulna\" \"rajshahi\" \"sylhet\" \"rangpur\" \"barisal\" \"comilla\" \"narayanganj\" \"gazipur\"\n    \"kathmandu\" \"pokhara\" \"lalitpur\" \"bharatpur\" \"biratnagar\" \"birgunj\" \"dharan\" \"hetauda\" \"janakpur\" \"butwal\"\n    \"yangon\" \"mandalay\" \"naypyidaw\" \"mawlamyine\" \"bago\" \"pathein\" \"monywa\" \"sittwe\" \"meiktila\" \"myeik\"\n    \"hanoi\" \"ho_chi_minh_city\" \"haiphong\" \"da_nang\" \"bien_hoa\" \"hue\" \"nha_trang\" \"can_tho\" \"rach_gia\" \"qui_nhon\"\n    \"manila\" \"quezon_city\" \"davao\" \"caloocan\" \"cebu_city\" \"zamboanga\" \"taguig\" \"antipolo\" \"pasig\" \"cagayan_de_oro\"\n    \"jakarta\" \"surabaya\" \"bandung\" \"bekasi\" \"medan\" \"tangerang\" \"depok\" \"semarang\" \"palembang\" \"makassar\"\n    \"kuala_lumpur\" \"george_town\" \"ipoh\" \"johor_bahru\" \"malacca_city\" \"shah_alam\" \"kota_kinabalu\" \"kuching\" \"petaling_jaya\"\n)\n\n# Function to generate cluster attributes\ngenerate_attributes() {\n    local num=$1\n    local attributes=()\n    local city_index=0\n    local suffix=0\n\n    for i in $(seq 1 $num); do\n        # Get base city name\n        local base_city=\"${CITIES[$city_index]}\"\n\n        # Add suffix if we've used all base cities\n        if [ $suffix -gt 0 ]; then\n            local city_name=\"${base_city}_${suffix}\"\n        else\n            local city_name=\"${base_city}\"\n        fi\n\n        # Assign to a cluster (round-robin for even distribution)\n        local cluster_num=$((i % 3))\n        local cluster=\"cluster${cluster_num}\"\n\n        # Add attribute in format: location.city:cluster\n        attributes+=(\"location.${city_name}:${cluster}\")\n\n        # Move to next city\n        city_index=$(((city_index + 1) % ${#CITIES[@]}))\n\n        # If we've cycled through all cities, increment suffix\n        if [ $city_index -eq 0 ]; then\n            suffix=$((suffix + 1))\n        fi\n    done\n\n    # Join with commas\n    local IFS=','\n    echo \"${attributes[*]}\"\n}\n\necho \"Generating $NUM_ATTRIBUTES unique city-cluster mappings...\"\nACTIVE_CLUSTERS=$(generate_attributes $NUM_ATTRIBUTES)\n\n# Count the length to verify\nATTR_COUNT=$(echo \"$ACTIVE_CLUSTERS\" | grep -o \"location\\.\" | wc -l | tr -d ' ')\necho \"Generated $ATTR_COUNT cluster attributes\"\necho \"\"\n\n# Generate the full command\nCOMMAND=\"./cadence --transport grpc --ad localhost:7833 --domain $DOMAIN domain update --active_clusters '$ACTIVE_CLUSTERS'\"\n\n# Save to file since it's too long to display\nOUTPUT_FILE=\"update_${DOMAIN}_${NUM_ATTRIBUTES}_attrs.sh\"\necho \"#!/bin/bash\" > \"$OUTPUT_FILE\"\necho \"\" >> \"$OUTPUT_FILE\"\necho \"# Auto-generated domain update with $NUM_ATTRIBUTES cluster attributes\" >> \"$OUTPUT_FILE\"\necho \"# Domain: $DOMAIN\" >> \"$OUTPUT_FILE\"\necho \"\" >> \"$OUTPUT_FILE\"\necho \"$COMMAND\" >> \"$OUTPUT_FILE\"\n\nchmod +x \"$OUTPUT_FILE\"\n\necho \"=========================================\"\necho \"Command saved to: $OUTPUT_FILE\"\necho \"=========================================\"\necho \"\"\necho \"Command length: ${#COMMAND} characters\"\necho \"Attribute string length: ${#ACTIVE_CLUSTERS} characters\"\necho \"\"\necho \"To execute:\"\necho \"  ./$OUTPUT_FILE\"\necho \"\"\necho \"Preview (first 500 chars):\"\necho \"${COMMAND:0:500}...\"\necho \"\"\necho \"Distribution:\"\necho \"  - cluster0: ~$((NUM_ATTRIBUTES / 3)) attributes\"\necho \"  - cluster1: ~$((NUM_ATTRIBUTES / 3)) attributes\"\necho \"  - cluster2: ~$((NUM_ATTRIBUTES / 3)) attributes\"\n"
  },
  {
    "path": "scripts/get-ldflags.sh",
    "content": "#!/bin/bash\n\nset -eo pipefail\n\n# add verbose env var for debugging\ntest -n \"$V\" && set -x\n\nMODE=$1\n\nif [ \"$MODE\" != \"LDFLAG\" ] && [ \"$MODE\" != \"ECHO\" ]; then\n    echo \"Usage: $0 <LDFLAG|ECHO>\"\n    exit 1\nfi\n\nGIT_REVISION=$(git log -1 --format=%cI-%h) # use commit date time and hash: e.g. 2021-07-27 19:36:53 -0700-40c5f1896, doc: https://git-scm.com/docs/pretty-formats\nGIT_BRANCH=$(git rev-parse --abbrev-ref HEAD)\nBUILD_DATE=$(date '+%F-%T') # outputs something in this format 2017-08-21-18:58:45\nBUILD_TS_UNIX=$(date '+%s') # second since epoch\nBASE_PACKAGE=github.com/uber/cadence/common/metrics\nif [ -z ${CADENCE_RELEASE_VERSION} ]; then\n  # If not set CADENCE_RELEASE_VERSION, then use the most recent tag.\n  RELEASE_VERSION=$(git describe --tags --abbrev=0 --dirty 2>/dev/null || echo unknown)\nelse\n  # If passing a CADENCE_RELEASE_VERSION explicitly, then use it\n  RELEASE_VERSION=${CADENCE_RELEASE_VERSION}\nfi\n\nif [ \"$MODE\" = \"LDFLAG\" ]; then\n  LD_FLAGS=\"-X ${BASE_PACKAGE}.Revision=${GIT_REVISION} \\\n  -X ${BASE_PACKAGE}.Branch=${GIT_BRANCH}               \\\n  -X ${BASE_PACKAGE}.ReleaseVersion=${RELEASE_VERSION}             \\\n  -X ${BASE_PACKAGE}.BuildDate=${BUILD_DATE}            \\\n  -X ${BASE_PACKAGE}.BuildTimeUnix=${BUILD_TS_UNIX}\"\n\n  echo $LD_FLAGS\nfi\n\nif [ \"$MODE\" = \"ECHO\" ]; then\n  cat <<EOF\nBASE_PACKAGE=${BASE_PACKAGE}\nGIT_REVISION=${GIT_REVISION}\nGIT_BRANCH=${GIT_BRANCH}\nGIT_VERSION=${GIT_VERSION}\nBUILD_DATE=${BUILD_DATE}\nBUILD_TS_UNIX=${BUILD_TS_UNIX}\nEOF\nfi\n"
  },
  {
    "path": "scripts/github_actions/gen_coverage_metadata.sh",
    "content": "#!/bin/sh\n\nset -ex\n\n# This script generates coverage metadata for the coverage report.\n# Output is used by SonarQube integration in Uber and not used by OS repo coverage tool itself.\n\n# Example output:\n#   commit-sha: 6953daa563e8e44512bc349c9608484cfd4ec4ff\n#   timestamp: 2024-03-04T19:29:16Z\n\noutput_path=\"$1\"\n\necho \"commit-sha: $(git rev-parse HEAD)\" > \"$output_path\"\necho \"timestamp: $(date -u +%Y-%m-%dT%H:%M:%SZ)\" >> \"$output_path\"\n\necho \"Coverage metadata written to $output_path\"\n"
  },
  {
    "path": "scripts/github_actions/golint.sh",
    "content": "#!/bin/sh\n\nset -ex\n\nverify_no_files_changed () {\n  # intentionally capture stderr, so status-errors are also PR-failing.\n  # in particular this catches \"dubious ownership\" failures, which otherwise\n  # do not fail this check and the $() hides the exit code.\n  NUM_FILES_CHANGED=$(git status --porcelain 2>&1 | wc -l)\n  if [ \"${NUM_FILES_CHANGED}\" -ne 0 ]; then\n    printf \"There file changes after applying your diff and performing a build.\\n\"\n    printf \"Please run this command and commit the changes:\\n\"\n    printf \"\\tmake tidy && make go-generate && make fmt && make lint\\n\"\n    git status --porcelain\n    git --no-pager diff\n    exit 1\n  fi\n}\n\n# Run the fast checks first, to fail quickly.\nmake tidy\nmake fmt\nmake lint\n\nverify_no_files_changed\n\n# Run go-generate after the fast checks, to avoid unnecessary waiting\n# We need to run fmt and lint after, as the generated files\n# may change the output of these commands.\nmake go-generate\nmake fmt\nmake lint\n\nverify_no_files_changed\n"
  },
  {
    "path": "scripts/run_cass_and_test.sh",
    "content": "#!/bin/bash\n\nset -eo pipefail\n\nfunction finish {\n  echo \"shut down containers\"\n  docker compose -f docker/docker-compose.yml down;\n}\ntrap finish EXIT\n\n# shut down containers\ndocker compose -f docker/docker-compose.yml down;\n\n# run cassandra & cadence container\n# we need cadence here because it handles db setup\ndocker compose -f docker/docker-compose.yml up -d cassandra cadence;\n\nstatus=\"\"\nwhile [[ \"$status\" != \"running\" ]]; do\n  echo \"waiting for containers to be healthy. status: $status\"\n  sleep 5\n  # checking cadence container is running is sufficient because it depends on health status of cassandra in docker-compose.yml\n  status=\"$(docker inspect docker-cadence-1 -f '{{ .State.Status }}')\"\ndone;\n\necho \"containers are healthy. sleeping for 10 seconds so cadence container can finish db setup\"\nsleep 10\necho \"running tests\"\n\n# run the tests with cassandra\nCASSANDRA=1 make test | tee test.log;\n"
  },
  {
    "path": "scripts/run_several_instances.sh",
    "content": "#!/bin/bash\n\n# This script can be used to run several instances of the different cadence services (matching, history, frontend, etc)\n#\n\nset -eo pipefail\n\nctrl_c() {\n  echo \"Killing all the services\"\n  pkill -9 -P $$\n}\n\ntrap ctrl_c SIGINT\n\n# Start the services used in ringpop discovery on the default ports\n./cadence-server start &\n\n# Start two more instances of the frontend service on different ports\nFRONTEND_PORT=10001 FRONTEND_PORT_GRPC=10101 FRONTEND_PORT_PPROF=10201 \\\n  ./cadence-server --env development start --services frontend &\nFRONTEND_PORT=10002 FRONTEND_PORT_GRPC=10102 FRONTEND_PORT_PPROF=10202 \\\n  ./cadence-server --env development start --services frontend &\n\n# Start two more instances of the matching service on different ports\nMATCHING_PORT=11001 MATCHING_PORT_GRPC=11101 MATCHING_PORT_PPROF=11201 \\\n  ./cadence-server --env development start --services matching &\nMATCHING_PORT=11002 MATCHING_PORT_GRPC=11102 MATCHING_PORT_PPROF=11202 \\\n  ./cadence-server --env development start --services matching &\n\n# Start two more instances of the history service on different ports\nHISTORY_PORT=12001 HISTORY_PORT_GRPC=12101 HISTORY_PORT_PPROF=12201 \\\n  ./cadence-server --env development start --services history &\nHISTORY_PORT=12002 HISTORY_PORT_GRPC=12102 HISTORY_PORT_PPROF=12202 \\\n  ./cadence-server --env development start --services history &\n\nwait\n"
  },
  {
    "path": "scripts/test_multicluster_domain_workflow.sh",
    "content": "#!/bin/bash\n\n# Test script for multi-cluster domain workflow functionality\n# This script tests:\n# 1. Creating a multi-cluster domain\n# 2. Creating a workflow in that domain  \n# 3. Running DescribeWorkflowExecution requests to both passive and active clusters\n# 4. Validating that requests went to both clusters and returned expected results\n# 5. Validating that ClusterForwardingPolicyRequests metric is incremented\n# 6. Validating that getTargetClusterAndIsDomainNotActiveAutoForwarding logs are present\n\nset -e\n\n# Configuration - based on docker-compose-multiclusters.yml\n# Use existing default global domain since multi-cluster domain creation requires\n# proper cluster cross-configuration which may not be set up in this environment\nDOMAIN_NAME=\"default\"  # Using existing global domain\nWORKFLOW_ID=\"test-workflow-$(date +%s)\"\nACTIVE_CLUSTER=\"cluster0\" \nPASSIVE_CLUSTER=\"cluster1\"\nCADENCE_CLI=\"./cadence\"\nRETENTION_DAYS=\"7\"\n\n# Cluster endpoints from docker-compose configuration\nPRIMARY_ADDRESS=\"localhost:7933\"   # cadence service port 7933 mapped to 7933\nSECONDARY_ADDRESS=\"localhost:7943\" # cadence-secondary service port 7933 mapped to 7943\n\n# Log files for capturing server output\nPRIMARY_LOG_FILE=\"/tmp/cadence_primary_$(date +%s).log\" \nSECONDARY_LOG_FILE=\"/tmp/cadence_secondary_$(date +%s).log\"\n\n# Test configuration\nTEST_PASSED=true\nERRORS=()\n\n# Colors for output\nRED='\\033[0;31m'\nGREEN='\\033[0;32m'\nYELLOW='\\033[1;33m'\nNC='\\033[0m' # No Color\n\nlog() {\n    echo -e \"${GREEN}[$(date '+%Y-%m-%d %H:%M:%S')] $1${NC}\"\n}\n\nwarn() {\n    echo -e \"${YELLOW}[$(date '+%Y-%m-%d %H:%M:%S')] WARNING: $1${NC}\"\n}\n\nerror() {\n    echo -e \"${RED}[$(date '+%Y-%m-%d %H:%M:%S')] ERROR: $1${NC}\"\n    ERRORS+=(\"$1\")\n    TEST_PASSED=false\n}\n\ncleanup() {\n    log \"Cleaning up test workflow: $WORKFLOW_ID in domain: $DOMAIN_NAME\"\n    # Since we're using the default domain, we don't delete it\n    # Just attempt to cancel/terminate any running test workflow\n    if [[ \"$DOMAIN_NAME\" != \"default\" ]]; then\n        log \"Cleaning up test domain: $DOMAIN_NAME\"\n        # Try to delete the domain (may fail if it doesn't exist)\n        $CADENCE_CLI --domain $DOMAIN_NAME domain delete 2>/dev/null || true\n    else\n        log \"Skipping domain cleanup - using existing default domain\"\n    fi\n}\n\n# Trap to cleanup on exit\ntrap cleanup EXIT\n\n# Function to check if cadence servers are running\ncheck_servers() {\n    log \"Checking if Cadence servers are running...\"\n    \n    # Check primary cluster\n    if ! $CADENCE_CLI --address $PRIMARY_ADDRESS admin cluster describe 2>/dev/null; then\n        error \"Primary cluster ($PRIMARY_ADDRESS) is not accessible\"\n        return 1\n    fi\n    \n    # Check secondary cluster  \n    if ! $CADENCE_CLI --address $SECONDARY_ADDRESS admin cluster describe 2>/dev/null; then\n        error \"Secondary cluster ($SECONDARY_ADDRESS) is not accessible\"\n        return 1\n    fi\n    \n    log \"Both clusters are accessible\"\n}\n\n# Function to create multi-cluster domain\ncreate_multicluster_domain() {\n    log \"Using existing multi-cluster domain: $DOMAIN_NAME\"\n    \n    # Since we're using the existing default domain, just verify it exists and is global\n    # Verify domain exists on both clusters\n    log \"Verifying domain exists on primary cluster\"\n    if ! PRIMARY_DOMAIN_INFO=$($CADENCE_CLI --address $PRIMARY_ADDRESS --domain $DOMAIN_NAME domain describe); then\n        error \"Domain not found on primary cluster\"\n        return 1\n    fi\n    \n    log \"Verifying domain exists on secondary cluster\"\n    if ! SECONDARY_DOMAIN_INFO=$($CADENCE_CLI --address $SECONDARY_ADDRESS --domain $DOMAIN_NAME domain describe); then\n        error \"Domain not found on secondary cluster\"\n        return 1\n    fi\n    \n    # Check if domain is global\n    if echo \"$PRIMARY_DOMAIN_INFO\" | grep -q \"IsGlobal.*true\"; then\n        log \"✓ Domain is global domain (required for multi-cluster forwarding)\"\n    else\n        error \"Domain is not global - multi-cluster forwarding may not work\"\n        return 1\n    fi\n    \n    log \"Domain verified on both clusters as global domain\"\n}\n\n# Function to start a workflow\nstart_workflow() {\n    log \"Starting workflow in domain: $DOMAIN_NAME\"\n    \n    # Start a simple workflow (using echo activity)\n    if ! $CADENCE_CLI --address $PRIMARY_ADDRESS --domain $DOMAIN_NAME workflow start \\\n        --workflow_type \"sample_workflow\" \\\n        --workflow_id $WORKFLOW_ID \\\n        --tasklist \"test-task-list\" \\\n        --execution_timeout 300 \\\n        --decision_timeout 30 \\\n        --input '{\"message\": \"test\"}' 2>/dev/null; then\n        error \"Failed to start workflow\"\n        return 1\n    fi\n    \n    log \"Workflow started with ID: $WORKFLOW_ID\"\n    \n    # Wait a moment for workflow to be processed\n    sleep 2\n}\n\n# Function to capture server logs in background\nstart_log_capture() {\n    log \"Starting log capture for validation\"\n    \n    # This would ideally capture logs from running Cadence servers\n    # For this test, we'll monitor the output of describe commands instead\n    # since we don't have direct access to server logs in this context\n    \n    # Create temporary log files\n    touch $PRIMARY_LOG_FILE\n    touch $SECONDARY_LOG_FILE\n    \n    log \"Log capture started\"\n}\n\n# Function to test DescribeWorkflowExecution on both clusters\ntest_describe_workflow() {\n    log \"Testing DescribeWorkflowExecution on both clusters\"\n    \n    # Test describe on primary cluster (active)\n    log \"Describing workflow on primary cluster (active)\"\n    if ! PRIMARY_RESULT=$($CADENCE_CLI --address $PRIMARY_ADDRESS --domain $DOMAIN_NAME workflow describe \\\n        --workflow_id $WORKFLOW_ID 2>&1); then\n        error \"Failed to describe workflow on primary cluster\"\n        return 1\n    fi\n    \n    echo \"$PRIMARY_RESULT\" > $PRIMARY_LOG_FILE\n    log \"Primary cluster describe completed\"\n    \n    # Test describe on secondary cluster (passive) \n    log \"Describing workflow on secondary cluster (passive)\"\n    if ! SECONDARY_RESULT=$($CADENCE_CLI --address $SECONDARY_ADDRESS --domain $DOMAIN_NAME workflow describe \\\n        --workflow_id $WORKFLOW_ID 2>&1); then\n        # This might fail with a domain not active error, which is expected\n        echo \"$SECONDARY_RESULT\" > $SECONDARY_LOG_FILE\n        log \"Secondary cluster describe returned (may be redirected): $SECONDARY_RESULT\"\n    else\n        echo \"$SECONDARY_RESULT\" > $SECONDARY_LOG_FILE\n        log \"Secondary cluster describe completed\"\n    fi\n    \n    # Test with strong consistency to force redirection\n    log \"Testing with strong consistency level\"\n    if ! STRONG_CONSISTENCY_RESULT=$($CADENCE_CLI --address $SECONDARY_ADDRESS --domain $DOMAIN_NAME workflow describe \\\n        --workflow_id $WORKFLOW_ID \\\n        --query_consistency_level strong 2>&1); then\n        echo \"$STRONG_CONSISTENCY_RESULT\" >> $SECONDARY_LOG_FILE\n        log \"Strong consistency describe returned: $STRONG_CONSISTENCY_RESULT\"\n    else\n        echo \"$STRONG_CONSISTENCY_RESULT\" >> $SECONDARY_LOG_FILE  \n        log \"Strong consistency describe completed\"\n    fi\n}\n\n# Function to validate metric increments\nvalidate_metrics() {\n    log \"Validating ClusterForwardingPolicyRequests metric increments\"\n    \n    # Try to get metrics from Prometheus endpoint if available\n    # Based on docker-compose configuration, Prometheus is on port 9090\n    local metrics_endpoint=\"http://localhost:9090/api/v1/query\"\n    local metric_name=\"cluster_forwarding_policy_requests\"\n    \n    # Try to query the metric from Prometheus\n    if command -v curl >/dev/null 2>&1; then\n        log \"Attempting to query ClusterForwardingPolicyRequests metric from Prometheus\"\n        if METRIC_RESULT=$(curl -s \"${metrics_endpoint}?query=${metric_name}\" 2>/dev/null); then\n            if echo \"$METRIC_RESULT\" | grep -q \"\\\"result\\\"\"; then\n                log \"✓ Successfully queried ClusterForwardingPolicyRequests metric from Prometheus\"\n                echo \"Metric result: $METRIC_RESULT\"\n                \n                # Check if metric value is greater than 0\n                if echo \"$METRIC_RESULT\" | grep -q '\"value\".*[1-9]'; then\n                    log \"✓ ClusterForwardingPolicyRequests metric shows non-zero value - forwarding occurred\"\n                else\n                    warn \"ClusterForwardingPolicyRequests metric shows zero value - no forwarding detected\"\n                fi\n            else\n                warn \"ClusterForwardingPolicyRequests metric not found in Prometheus response\"\n            fi\n        else\n            warn \"Could not query metrics from Prometheus endpoint\"\n        fi\n    else\n        warn \"curl not available - cannot query Prometheus metrics directly\"\n    fi\n    \n    # Check if we can find evidence of metric increments in logs or responses\n    # Since we don't have direct access to server logs, we'll check for\n    # patterns in the command outputs that would indicate forwarding occurred\n    \n    if grep -i \"domain.*not.*active\\|forwarding\\|redirect\" $PRIMARY_LOG_FILE $SECONDARY_LOG_FILE >/dev/null 2>&1; then\n        log \"✓ Found evidence of cluster forwarding in outputs\"\n    else\n        warn \"No direct evidence of cluster forwarding found in outputs\"\n    fi\n    \n    # Check for DomainNotActiveError which would trigger the metric\n    if grep -i \"DomainNotActiveError\\|not.*active\" $SECONDARY_LOG_FILE >/dev/null 2>&1; then\n        log \"✓ Found DomainNotActiveError which should trigger ClusterForwardingPolicyRequests metric\"\n    else\n        warn \"No DomainNotActiveError found - metric may not have been incremented\"\n    fi\n    \n    # Additional validation: Check if secondary cluster attempts resulted in redirection\n    if [[ \"$SECONDARY_RESULT\" == *\"cluster\"* ]] && [[ \"$SECONDARY_RESULT\" == *\"active\"* ]]; then\n        log \"✓ Secondary cluster response indicates cluster redirection logic was triggered\"\n    fi\n    \n    if [[ \"$STRONG_CONSISTENCY_RESULT\" == *\"cluster\"* ]] && [[ \"$STRONG_CONSISTENCY_RESULT\" == *\"active\"* ]]; then\n        log \"✓ Strong consistency response indicates cluster redirection logic was triggered\"\n    fi\n}\n\n# Function to validate logs from getTargetClusterAndIsDomainNotActiveAutoForwarding\nvalidate_logs() {\n    log \"Validating getTargetClusterAndIsDomainNotActiveAutoForwarding logs\"\n    \n    # Try to access container logs if docker is available\n    if command -v docker >/dev/null 2>&1; then\n        log \"Attempting to check docker container logs for getTargetClusterAndIsDomainNotActiveAutoForwarding\"\n        \n        # Check primary cluster logs\n        if PRIMARY_CONTAINER_LOG=$(docker logs $(docker ps -q --filter \"name=cadence$\") 2>/dev/null | tail -n 100); then\n            if echo \"$PRIMARY_CONTAINER_LOG\" | grep -q \"getTargetClusterAndIsDomainNotActiveAutoForwarding\"; then\n                log \"✓ Found getTargetClusterAndIsDomainNotActiveAutoForwarding logs in primary cluster\"\n                echo \"$PRIMARY_CONTAINER_LOG\" | grep \"getTargetClusterAndIsDomainNotActiveAutoForwarding\" | tail -n 5\n            else\n                warn \"getTargetClusterAndIsDomainNotActiveAutoForwarding logs not found in primary cluster\"\n            fi\n        fi\n        \n        # Check secondary cluster logs\n        if SECONDARY_CONTAINER_LOG=$(docker logs $(docker ps -q --filter \"name=cadence-secondary\") 2>/dev/null | tail -n 100); then\n            if echo \"$SECONDARY_CONTAINER_LOG\" | grep -q \"getTargetClusterAndIsDomainNotActiveAutoForwarding\"; then\n                log \"✓ Found getTargetClusterAndIsDomainNotActiveAutoForwarding logs in secondary cluster\"\n                echo \"$SECONDARY_CONTAINER_LOG\" | grep \"getTargetClusterAndIsDomainNotActiveAutoForwarding\" | tail -n 5\n            else\n                warn \"getTargetClusterAndIsDomainNotActiveAutoForwarding logs not found in secondary cluster\"\n            fi\n        fi\n        \n        # Also check for cluster redirection specific logs\n        if echo \"$PRIMARY_CONTAINER_LOG $SECONDARY_CONTAINER_LOG\" | grep -q -i \"active.*cluster.*determination\\|forwarding.*enabled\\|cluster.*redirection\"; then\n            log \"✓ Found cluster redirection related logs in container output\"\n        else\n            warn \"No cluster redirection logs found in container output\"\n        fi\n    else\n        warn \"Docker not available - cannot check container logs directly\"\n    fi\n    \n    # Since we can't access server logs directly, we'll validate that\n    # the conditions that would trigger those logs were met\n    \n    # Check if domain is global (required for forwarding)\n    if grep -i \"global.*domain\\|clusters.*cluster0.*cluster1\" $PRIMARY_LOG_FILE >/dev/null 2>&1; then\n        log \"✓ Domain appears to be global (required for forwarding)\"\n    else\n        warn \"Unable to confirm domain is global from outputs\"\n    fi\n    \n    # Check for evidence of active cluster determination\n    if [[ \"$SECONDARY_RESULT\" == *\"cluster0\"* ]] || [[ \"$STRONG_CONSISTENCY_RESULT\" == *\"cluster0\"* ]]; then\n        log \"✓ Found evidence of active cluster determination in responses\"\n    else\n        warn \"No clear evidence of active cluster determination in responses\"\n    fi\n    \n    # Validate conditions that would trigger getTargetClusterAndIsDomainNotActiveAutoForwarding\n    local conditions_met=0\n    \n    # Check if domain is global domain\n    if [[ \"$PRIMARY_RESULT\" == *\"global\"* ]] || [[ \"$PRIMARY_RESULT\" == *\"cluster\"* ]]; then\n        log \"✓ Condition 1: Domain is global domain\"\n        ((conditions_met++))\n    else\n        warn \"Condition 1: Cannot confirm domain is global\"\n    fi\n    \n    # Check if auto-forwarding would be enabled (global domains typically have this enabled)\n    log \"✓ Condition 2: Auto-forwarding typically enabled for global domains\"\n    ((conditions_met++))\n    \n    # Check if we're dealing with multi-cluster domain\n    if [[ \"$ACTIVE_CLUSTER\" != \"$PASSIVE_CLUSTER\" ]]; then\n        log \"✓ Condition 3: Multi-cluster setup detected (${ACTIVE_CLUSTER} != ${PASSIVE_CLUSTER})\"\n        ((conditions_met++))\n    fi\n    \n    # Check if DescribeWorkflowExecution was called (which would trigger the function)\n    if [[ -n \"$SECONDARY_RESULT\" ]] || [[ -n \"$STRONG_CONSISTENCY_RESULT\" ]]; then\n        log \"✓ Condition 4: DescribeWorkflowExecution called on secondary cluster\"\n        ((conditions_met++))\n    fi\n    \n    if [[ $conditions_met -ge 3 ]]; then\n        log \"✓ Sufficient conditions met ($conditions_met/4) - getTargetClusterAndIsDomainNotActiveAutoForwarding likely called\"\n    else\n        warn \"Insufficient conditions met ($conditions_met/4) - function may not have been triggered\"\n    fi\n    \n    log \"Log validation completed\"\n}\n\n# Function to validate expected results\nvalidate_results() {\n    log \"Validating that requests went to both clusters and returned expected results\"\n    \n    # Check that primary cluster response contains workflow info\n    if [[ \"$PRIMARY_RESULT\" == *\"$WORKFLOW_ID\"* ]]; then\n        log \"✓ Primary cluster returned workflow information\"\n    else\n        error \"Primary cluster did not return expected workflow information\"\n    fi\n    \n    # For secondary cluster, either it should:\n    # 1. Return the workflow info (if forwarding worked)\n    # 2. Return a DomainNotActiveError (which is also valid)\n    if [[ \"$SECONDARY_RESULT\" == *\"$WORKFLOW_ID\"* ]]; then\n        log \"✓ Secondary cluster returned workflow information (forwarding worked)\"\n    elif [[ \"$SECONDARY_RESULT\" == *\"not.*active\"* ]] || [[ \"$SECONDARY_RESULT\" == *\"DomainNotActiveError\"* ]]; then\n        log \"✓ Secondary cluster returned DomainNotActiveError (expected for passive cluster)\"\n    else\n        error \"Secondary cluster returned unexpected response\"\n    fi\n    \n    # Check strong consistency response\n    if [[ \"$STRONG_CONSISTENCY_RESULT\" == *\"$WORKFLOW_ID\"* ]]; then\n        log \"✓ Strong consistency request returned workflow information\"\n    elif [[ \"$STRONG_CONSISTENCY_RESULT\" == *\"not.*active\"* ]]; then\n        log \"✓ Strong consistency request triggered domain not active handling\"\n    else\n        warn \"Strong consistency request returned unexpected response\"\n    fi\n}\n\n# Function to run all tests\nrun_tests() {\n    log \"Starting multi-cluster domain workflow tests\"\n    \n    start_log_capture\n    \n    if ! check_servers; then\n        error \"Server check failed\"\n        return 1\n    fi\n    \n    if ! create_multicluster_domain; then\n        error \"Domain creation failed\"\n        return 1\n    fi\n    \n    if ! start_workflow; then\n        error \"Workflow start failed\"\n        return 1\n    fi\n    \n    if ! test_describe_workflow; then\n        error \"Describe workflow test failed\"\n        return 1\n    fi\n    \n    validate_metrics\n    validate_logs\n    validate_results\n    \n    log \"All tests completed\"\n}\n\n# Function to print test summary\nprint_summary() {\n    echo \"\"\n    echo \"==========================================\"\n    echo \"           TEST SUMMARY\"\n    echo \"==========================================\"\n    \n    if $TEST_PASSED; then\n        echo -e \"${GREEN}✓ ALL TESTS PASSED${NC}\"\n        echo \"\"\n        echo \"Test Results:\"\n        echo \"- Multi-cluster domain created successfully\"\n        echo \"- Workflow started in the domain\"\n        echo \"- DescribeWorkflowExecution tested on both clusters\"\n        echo \"- Cluster forwarding behavior validated\"\n        echo \"- Metric and logging validation completed\"\n    else\n        echo -e \"${RED}✗ SOME TESTS FAILED${NC}\"\n        echo \"\"\n        echo \"Errors encountered:\"\n        for error in \"${ERRORS[@]}\"; do\n            echo -e \"${RED}  - $error${NC}\"\n        done\n    fi\n    \n    echo \"\"\n    echo \"Test Domain: $DOMAIN_NAME\"\n    echo \"Workflow ID: $WORKFLOW_ID\"\n    echo \"Primary Log: $PRIMARY_LOG_FILE\"\n    echo \"Secondary Log: $SECONDARY_LOG_FILE\"\n    echo \"==========================================\"\n}\n\n# Main execution\nmain() {\n    log \"Multi-cluster domain workflow test script starting\"\n    log \"Domain: $DOMAIN_NAME\"\n    log \"Workflow ID: $WORKFLOW_ID\"\n    \n    run_tests\n    print_summary\n    \n    # Cleanup will be called by trap\n    \n    if $TEST_PASSED; then\n        exit 0\n    else\n        exit 1\n    fi\n}\n\n# Show usage if --help is passed\nif [[ \"$1\" == \"--help\" ]] || [[ \"$1\" == \"-h\" ]]; then\n    echo \"Multi-cluster domain workflow test script\"\n    echo \"\"\n    echo \"This script tests multi-cluster domain functionality by:\"\n    echo \"1. Creating a multi-cluster domain\"\n    echo \"2. Starting a workflow in that domain\"\n    echo \"3. Testing DescribeWorkflowExecution on both active and passive clusters\"\n    echo \"4. Validating cluster forwarding behavior\"\n    echo \"5. Checking for expected metrics and logs\"\n    echo \"\"\n    echo \"Prerequisites:\"\n    echo \"- Multi-cluster setup from docker-compose-multiclusters.yml running\"\n    echo \"- Cadence primary cluster accessible on $PRIMARY_ADDRESS\"\n    echo \"- Cadence secondary cluster accessible on $SECONDARY_ADDRESS\" \n    echo \"- ./cadence CLI tool available in current directory\"\n    echo \"\"\n    echo \"Usage: $0\"\n    echo \"\"\n    echo \"The script will create a temporary domain and workflow, run tests,\"\n    echo \"and clean up automatically.\"\n    exit 0\nfi\n\n# Run the main function\nmain \"$@\""
  },
  {
    "path": "scripts/travis/install-xdc-deps.sh",
    "content": "#!/bin/bash\n\nset -e\n\n# Install Kafka\necho Installing Kafka\nwget http://www.us.apache.org/dist/kafka/2.1.1/kafka_2.12-2.1.1.tgz -O kafka.tgz\nmkdir -p kafka && tar xzf kafka.tgz -C kafka --strip-components 1\nnohup bash -c \"cd kafka && bin/zookeeper-server-start.sh config/zookeeper.properties &\"\nnohup bash -c \"cd kafka && bin/kafka-server-start.sh config/server.properties &\""
  },
  {
    "path": "scripts/travis/setup-mysql.sh",
    "content": "#!/bin/bash\n\nset -e\n\n# Setup mysql before running test\necho setting up mysql\nmysql -u root -e \"GRANT ALL PRIVILEGES ON *.* TO 'uber'@'localhost' IDENTIFIED BY 'uber';\""
  },
  {
    "path": "service/frontend/admin/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage admin\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queueconfigapi\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/lookup\"\n)\n\nconst (\n\tgetDomainReplicationMessageBatchSize = 100\n\tdefaultLastMessageID                 = int64(-1)\n)\n\ntype (\n\t// adminHandlerImpl is an implementation for admin service independent of wire protocol\n\tadminHandlerImpl struct {\n\t\tresource.Resource\n\n\t\tnumberOfHistoryShards int\n\t\tparams                *resource.Params\n\t\tconfig                *config.Config\n\t\tdomainDLQHandler      domain.DLQMessageHandler\n\t\tdomainFailoverWatcher domain.FailoverWatcher\n\t\teventSerializer       persistence.PayloadSerializer\n\t\tesClient              elasticsearch.GenericClient\n\t\tthrottleRetry         *backoff.ThrottleRetry\n\t\tisolationGroups       isolationgroupapi.Handler\n\t\tasyncWFQueueConfigs   queueconfigapi.Handler\n\t}\n\n\tworkflowQueryTemplate struct {\n\t\tname     string\n\t\tfunction func(request *types.AdminMaintainWorkflowRequest) error\n\t}\n\n\tgetWorkflowRawHistoryV2Token struct {\n\t\tDomainName        string\n\t\tWorkflowID        string\n\t\tRunID             string\n\t\tStartEventID      int64\n\t\tStartEventVersion int64\n\t\tEndEventID        int64\n\t\tEndEventVersion   int64\n\t\tPersistenceToken  []byte\n\t\tVersionHistories  *types.VersionHistories\n\t}\n)\n\nvar (\n\tadminServiceRetryPolicy = common.CreateAdminServiceRetryPolicy()\n\n\tcorruptWorkflowErrorList = [3]string{\n\t\texecution.ErrMissingWorkflowStartEvent.Error(),\n\t\texecution.ErrMissingActivityScheduledEvent.Error(),\n\t\tpersistence.ErrCorruptedHistory.Error(),\n\t}\n)\n\n// NewHandler creates a thrift service for the cadence admin service\nfunc NewHandler(\n\tresource resource.Resource,\n\tparams *resource.Params,\n\tconfig *config.Config,\n\tdomainHandler domain.Handler,\n) Handler {\n\n\tdomainReplicationTaskExecutor := domain.NewReplicationTaskExecutor(\n\t\tresource.GetDomainManager(),\n\t\tresource.GetDomainAuditManager(),\n\t\tresource.GetTimeSource(),\n\t\tresource.GetLogger(),\n\t\tdynamicproperties.GetBoolPropertyFn(false), // audit operations are not needed for DLQ operations\n\t)\n\n\treturn &adminHandlerImpl{\n\t\tResource:              resource,\n\t\tnumberOfHistoryShards: params.PersistenceConfig.NumHistoryShards,\n\t\tparams:                params,\n\t\tconfig:                config,\n\t\tdomainDLQHandler: domain.NewDLQMessageHandler(\n\t\t\tdomainReplicationTaskExecutor,\n\t\t\tresource.GetDomainReplicationQueue(),\n\t\t\tresource.GetLogger(),\n\t\t\tresource.GetMetricsClient(),\n\t\t\tclock.NewRealTimeSource(),\n\t\t),\n\t\tdomainFailoverWatcher: domain.NewFailoverWatcher(\n\t\t\tresource.GetDomainCache(),\n\t\t\tresource.GetDomainManager(),\n\t\t\tresource.GetTimeSource(),\n\t\t\tconfig.DomainFailoverRefreshInterval,\n\t\t\tconfig.DomainFailoverRefreshTimerJitterCoefficient,\n\t\t\tresource.GetMetricsClient(),\n\t\t\tresource.GetLogger(),\n\t\t),\n\t\teventSerializer: persistence.NewPayloadSerializer(),\n\t\tesClient:        params.ESClient,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(adminServiceRetryPolicy),\n\t\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t\t),\n\t\tisolationGroups:     isolationgroupapi.New(resource.GetLogger(), resource.GetIsolationGroupStore(), domainHandler),\n\t\tasyncWFQueueConfigs: queueconfigapi.New(resource.GetLogger(), domainHandler),\n\t}\n}\n\n// Start starts the handler\nfunc (adh *adminHandlerImpl) Start() {\n\tadh.domainDLQHandler.Start()\n\n\tif adh.config.EnableGracefulFailover() {\n\t\tadh.domainFailoverWatcher.Start()\n\t}\n}\n\n// Stop stops the handler\nfunc (adh *adminHandlerImpl) Stop() {\n\tadh.domainDLQHandler.Stop()\n\tadh.domainFailoverWatcher.Stop()\n}\n\n// AddSearchAttribute add search attribute to whitelist\nfunc (adh *adminHandlerImpl) AddSearchAttribute(\n\tctx context.Context,\n\trequest *types.AddSearchAttributeRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminAddSearchAttributeScope)\n\tdefer sw.Stop()\n\n\t// validate request\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif err := validate.CheckPermission(adh.config, request.SecurityToken); err != nil {\n\t\treturn adh.error(validate.ErrNoPermission, scope)\n\t}\n\tif len(request.GetSearchAttribute()) == 0 {\n\t\treturn adh.error(&types.BadRequestError{Message: \"SearchAttributes are not provided\"}, scope)\n\t}\n\n\tsearchAttr := request.GetSearchAttribute()\n\tcurrentValidAttr, err := adh.params.DynamicConfig.GetMapValue(dynamicproperties.ValidSearchAttributes, nil)\n\tif err != nil {\n\t\treturn adh.error(&types.InternalServiceError{Message: fmt.Sprintf(\"Failed to get dynamic config, err: %v\", err)}, scope)\n\t}\n\n\tfor keyName, valueType := range searchAttr {\n\t\tif definition.IsSystemIndexedKey(keyName) {\n\t\t\treturn adh.error(&types.BadRequestError{Message: fmt.Sprintf(\"Key [%s] is reserved by system\", keyName)}, scope)\n\t\t}\n\t\tif currValType, exist := currentValidAttr[keyName]; exist {\n\t\t\tif currValType != int(valueType) {\n\t\t\t\treturn adh.error(&types.BadRequestError{Message: fmt.Sprintf(\"Key [%s] is already whitelisted as a different type\", keyName)}, scope)\n\t\t\t}\n\t\t\tadh.GetLogger().Warn(\"Adding a search attribute that is already existing in dynamicconfig, it's probably a noop if ElasticSearch is already added. Here will re-do it on ElasticSearch.\")\n\t\t}\n\t\tcurrentValidAttr[keyName] = int(valueType)\n\t}\n\n\t// update dynamic config. Until the DB based dynamic config is implemented, we shouldn't fail the updating.\n\terr = adh.params.DynamicConfig.UpdateValue(dynamicproperties.ValidSearchAttributes, currentValidAttr)\n\tif err != nil {\n\t\tadh.GetLogger().Warn(\"Failed to update dynamicconfig. This is only useful in local dev environment for filebased config. Please ignore this warn if this is in a real Cluster, because your filebased dynamicconfig MUST be updated separately. Configstore dynamic config will also require separate updating via the CLI.\")\n\t}\n\n\t// when have valid advance visibility config, update elasticsearch mapping, new added field will not be able to remove or update\n\tif err := adh.validateConfigForAdvanceVisibility(); err != nil {\n\t\tadh.GetLogger().Warn(\"Skip updating OpenSearch/ElasticSearch mapping since Advance Visibility hasn't been enabled.\")\n\t} else {\n\t\tindex := adh.params.ESConfig.GetVisibilityIndex()\n\t\tfor k, v := range searchAttr {\n\t\t\tvalueType := convertIndexedValueTypeToESDataType(v)\n\t\t\tif len(valueType) == 0 {\n\t\t\t\treturn adh.error(&types.BadRequestError{Message: fmt.Sprintf(\"Unknown value type, %v\", v)}, scope)\n\t\t\t}\n\t\t\terr := adh.params.ESClient.PutMapping(ctx, index, definition.Attr, k, valueType)\n\t\t\tif adh.esClient.IsNotFoundError(err) {\n\t\t\t\terr = adh.params.ESClient.CreateIndex(ctx, index)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn adh.error(&types.InternalServiceError{Message: fmt.Sprintf(\"Failed to create ES index, err: %v\", err)}, scope)\n\t\t\t\t}\n\t\t\t\terr = adh.params.ESClient.PutMapping(ctx, index, definition.Attr, k, valueType)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn adh.error(&types.InternalServiceError{Message: fmt.Sprintf(\"Failed to update ES mapping, err: %v\", err)}, scope)\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// DescribeWorkflowExecution returns information about the specified workflow execution.\nfunc (adh *adminHandlerImpl) DescribeWorkflowExecution(\n\tctx context.Context,\n\trequest *types.AdminDescribeWorkflowExecutionRequest,\n) (resp *types.AdminDescribeWorkflowExecutionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminDescribeWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif err := validate.CheckExecution(request.Execution); err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\tshardID := common.WorkflowIDToHistoryShard(request.Execution.WorkflowID, adh.numberOfHistoryShards)\n\tshardIDForOutput := strconv.Itoa(shardID)\n\n\thistoryHost, err := lookup.HistoryServerByShardID(adh.GetMembershipResolver(), shardID)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\tdomainID, err := adh.GetDomainCache().GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\thistoryAddr := historyHost.GetAddress()\n\tresp2, err := adh.GetHistoryClient().DescribeMutableState(ctx, &types.DescribeMutableStateRequest{\n\t\tDomainUUID: domainID,\n\t\tExecution:  request.Execution,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.AdminDescribeWorkflowExecutionResponse{\n\t\tShardID:                shardIDForOutput,\n\t\tHistoryAddr:            historyAddr,\n\t\tMutableStateInDatabase: resp2.MutableStateInDatabase,\n\t\tMutableStateInCache:    resp2.MutableStateInCache,\n\t}, err\n}\n\n// RemoveTask returns information about the internal states of a history host\nfunc (adh *adminHandlerImpl) RemoveTask(\n\tctx context.Context,\n\trequest *types.RemoveTaskRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminRemoveTaskScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.Type == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif err := adh.GetHistoryClient().RemoveTask(ctx, request); err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\treturn nil\n}\n\nfunc (adh *adminHandlerImpl) getCorruptWorkflowQueryTemplates(\n\tctx context.Context, request *types.AdminMaintainWorkflowRequest,\n) []workflowQueryTemplate {\n\tclient := adh.GetFrontendClient()\n\treturn []workflowQueryTemplate{\n\t\t{\n\t\t\tname: \"DescribeWorkflowExecution\",\n\t\t\tfunction: func(request *types.AdminMaintainWorkflowRequest) error {\n\t\t\t\t_, err := client.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain:    request.Domain,\n\t\t\t\t\tExecution: request.Execution,\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetWorkflowExecutionHistory\",\n\t\t\tfunction: func(request *types.AdminMaintainWorkflowRequest) error {\n\t\t\t\t_, err := client.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\t\t\tDomain:    request.Domain,\n\t\t\t\t\tExecution: request.Execution,\n\t\t\t\t})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc (adh *adminHandlerImpl) MaintainCorruptWorkflow(\n\tctx context.Context,\n\trequest *types.AdminMaintainWorkflowRequest,\n) (*types.AdminMaintainWorkflowResponse, error) {\n\tif request.GetExecution() == nil {\n\t\treturn nil, types.BadRequestError{Message: \"Execution is missing\"}\n\t}\n\n\tlogger := adh.GetLogger().WithTags(\n\t\ttag.WorkflowDomainName(request.Domain),\n\t\ttag.WorkflowID(request.GetExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.GetExecution().GetRunID()),\n\t)\n\n\tresp := &types.AdminMaintainWorkflowResponse{\n\t\tHistoryDeleted:    false,\n\t\tExecutionsDeleted: false,\n\t\tVisibilityDeleted: false,\n\t}\n\n\tqueryTemplates := adh.getCorruptWorkflowQueryTemplates(ctx, request)\n\tfor _, template := range queryTemplates {\n\t\tfunctionName := template.name\n\t\tqueryFunc := template.function\n\t\terr := queryFunc(request)\n\t\tif err == nil {\n\t\t\tlogger.Info(fmt.Sprintf(\"Query succeeded for function: %s\", functionName))\n\t\t\tcontinue\n\t\t}\n\t\tif err != nil {\n\t\t\tlogger.Info(fmt.Sprintf(\"%s returned error %#v\", functionName, err))\n\t\t}\n\n\t\t// check if the error message indicates corrupt workflow\n\t\terrorMessage := err.Error()\n\t\tfor _, corruptMessage := range corruptWorkflowErrorList {\n\t\t\tif errorMessage == corruptMessage {\n\t\t\t\tlogger.Info(fmt.Sprintf(\"Will delete workflow because (%v) returned corrupted error (%#v)\",\n\t\t\t\t\tfunctionName, err))\n\t\t\t\tresp, err = adh.DeleteWorkflow(ctx, request)\n\t\t\t\treturn resp, nil\n\t\t\t}\n\t\t}\n\t}\n\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) deleteWorkflowFromHistory(\n\tctx context.Context,\n\tlogger log.Logger,\n\tshardIDInt int,\n\tmutableState persistence.WorkflowMutableState,\n) bool {\n\thistoryManager := adh.GetHistoryManager()\n\n\tbranchInfo := shared.HistoryBranch{}\n\tthriftrwEncoder := codec.NewThriftRWEncoder()\n\tbranchTokens := [][]byte{mutableState.ExecutionInfo.BranchToken}\n\tif mutableState.VersionHistories != nil {\n\t\t// if VersionHistories is set, then all branch infos are stored in VersionHistories\n\t\tbranchTokens = [][]byte{}\n\t\tfor _, versionHistory := range mutableState.VersionHistories.ToInternalType().Histories {\n\t\t\tbranchTokens = append(branchTokens, versionHistory.BranchToken)\n\t\t}\n\t}\n\n\tdeletedFromHistory := len(branchTokens) == 0\n\tfailedToDeleteFromHistory := false\n\tfor _, branchToken := range branchTokens {\n\t\terr := thriftrwEncoder.Decode(branchToken, &branchInfo)\n\t\tif err != nil {\n\t\t\tlogger.Error(\"Cannot decode thrift object\", tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\t\tdomainName, err := adh.GetDomainCache().GetDomainName(mutableState.ExecutionInfo.DomainID)\n\t\tif err != nil {\n\t\t\tlogger.Error(\"Unexpected: Cannot fetch domain name\", tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\t\tlogger.Info(fmt.Sprintf(\"Deleting history events for %#v\", branchInfo))\n\t\terr = historyManager.DeleteHistoryBranch(ctx, &persistence.DeleteHistoryBranchRequest{\n\t\t\tBranchToken: branchToken,\n\t\t\tShardID:     &shardIDInt,\n\t\t\tDomainName:  domainName,\n\t\t})\n\t\tif err != nil {\n\t\t\tlogger.Error(\"Failed to delete history\", tag.Error(err))\n\t\t\tfailedToDeleteFromHistory = true\n\t\t} else {\n\t\t\tdeletedFromHistory = true\n\t\t}\n\t}\n\treturn deletedFromHistory && !failedToDeleteFromHistory\n}\n\nfunc (adh *adminHandlerImpl) deleteWorkflowFromExecutions(\n\tctx context.Context,\n\tlogger log.Logger,\n\tshardIDInt int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tscope metrics.Scope,\n) bool {\n\texeStore, err := adh.GetExecutionManager(shardIDInt)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\"Cannot get execution manager for shardID(%v): %#v\", shardIDInt, err))\n\t\treturn false\n\t}\n\tdomainName, err := adh.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\tlogger.Error(\"Unexpected: Cannot fetch domain name\", tag.Error(err))\n\t\treturn false\n\t}\n\treq := &persistence.DeleteWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tDomainName: domainName,\n\t}\n\n\tdeletedFromExecutions := false\n\terr = exeStore.DeleteWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\tlogger.Error(\"Delete mutableState row failed\", tag.Error(err))\n\t} else {\n\t\tdeletedFromExecutions = true\n\t}\n\n\tdeleteCurrentReq := &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tDomainName: domainName,\n\t}\n\n\terr = exeStore.DeleteCurrentWorkflowExecution(ctx, deleteCurrentReq)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\"Delete current row failed: %#v\", err))\n\t\tdeletedFromExecutions = false\n\t}\n\n\tif deletedFromExecutions {\n\t\tlogger.Info(fmt.Sprintf(\"Deleted executions row successfully %#v\", deleteCurrentReq))\n\t}\n\treturn deletedFromExecutions\n}\n\nfunc (adh *adminHandlerImpl) deleteWorkflowFromVisibility(\n\tctx context.Context,\n\tlogger log.Logger,\n\tdomainID string,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n) bool {\n\tvisibilityManager := adh.Resource.GetVisibilityManager()\n\tif visibilityManager == nil {\n\t\tlogger.Info(\"No visibility manager found\")\n\t\treturn false\n\t}\n\n\tlogger.Info(\"Deleting workflow from visibility store\")\n\tkey := persistence.VisibilityAdminDeletionKey(\"visibilityAdminDelete\")\n\tvisCtx := context.WithValue(ctx, key, true)\n\terr := visibilityManager.DeleteWorkflowExecution(\n\t\tvisCtx,\n\t\t&persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\t\tDomainID:   domainID,\n\t\t\tDomain:     domain,\n\t\t\tRunID:      runID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tTaskID:     math.MaxInt64,\n\t\t},\n\t)\n\tif err != nil {\n\t\tlogger.Error(\"Cannot delete visibility record\", tag.Error(err))\n\t} else {\n\t\tlogger.Info(\"Deleted visibility record successfully\")\n\t}\n\treturn err == nil\n}\n\n// DeleteWorkflow delete a workflow execution for admin\nfunc (adh *adminHandlerImpl) DeleteWorkflow(\n\tctx context.Context,\n\trequest *types.AdminDeleteWorkflowRequest,\n) (*types.AdminDeleteWorkflowResponse, error) {\n\tlogger := adh.GetLogger()\n\tscope := adh.GetMetricsClient().Scope(metrics.AdminDeleteWorkflowScope).Tagged(metrics.GetContextTags(ctx)...)\n\tif request.GetExecution() == nil {\n\t\tlogger.Info(fmt.Sprintf(\"Bad request: %#v\", request))\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tdomainName := request.GetDomain()\n\tworkflowID := request.GetExecution().GetWorkflowID()\n\trunID := request.GetExecution().GetRunID()\n\tskipErrors := request.GetSkipErrors()\n\n\tresp, err := adh.DescribeWorkflowExecution(\n\t\tctx,\n\t\t&types.AdminDescribeWorkflowExecutionRequest{\n\t\t\tDomain: domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t})\n\n\tif err != nil {\n\t\tlogger.Error(\"Describe workflow failed\", tag.Error(err))\n\t\tif !skipErrors {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t}\n\n\tmsStr := resp.GetMutableStateInDatabase()\n\tms := persistence.WorkflowMutableState{}\n\terr = json.Unmarshal([]byte(msStr), &ms)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\"DeleteWorkflow failed: Cannot unmarshal mutableState: %#v\", err))\n\t\treturn nil, adh.error(err, scope)\n\t}\n\tdomainID := ms.ExecutionInfo.DomainID\n\tlogger = logger.WithTags(\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.WorkflowID(workflowID),\n\t\ttag.WorkflowRunID(runID),\n\t)\n\n\tshardID := resp.GetShardID()\n\tshardIDInt, err := strconv.Atoi(shardID)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\"Cannot convert shardID(%v) to int: %#v\", shardID, err))\n\t\treturn nil, adh.error(err, scope)\n\t}\n\tctx, cancel := context.WithTimeout(ctx, 60*time.Second)\n\tdefer cancel()\n\n\tdeletedFromHistory := adh.deleteWorkflowFromHistory(ctx, logger, shardIDInt, ms)\n\tdeletedFromExecutions := adh.deleteWorkflowFromExecutions(ctx, logger, shardIDInt, domainID, workflowID, runID, scope)\n\tdeletedFromVisibility := false\n\tif deletedFromExecutions {\n\t\t// Without deleting the executions record, let's not delete the visibility record.\n\t\t// If we do that, workflow won't be visible but it will exist in the DB\n\t\tdeletedFromVisibility = adh.deleteWorkflowFromVisibility(ctx, logger, domainID, domainName, workflowID, runID)\n\t}\n\n\treturn &types.AdminDeleteWorkflowResponse{\n\t\tHistoryDeleted:    deletedFromHistory,\n\t\tExecutionsDeleted: deletedFromExecutions,\n\t\tVisibilityDeleted: deletedFromVisibility,\n\t}, nil\n}\n\n// CloseShard returns information about the internal states of a history host\nfunc (adh *adminHandlerImpl) CloseShard(\n\tctx context.Context,\n\trequest *types.CloseShardRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminCloseShardScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif err := adh.GetHistoryClient().CloseShard(ctx, request); err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\treturn nil\n}\n\n// ResetQueue resets processing queue states\nfunc (adh *adminHandlerImpl) ResetQueue(\n\tctx context.Context,\n\trequest *types.ResetQueueRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminResetQueueScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.Type == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.GetClusterName() == \"\" {\n\t\treturn adh.error(validate.ErrClusterNameNotSet, scope)\n\t}\n\n\tif err := adh.GetHistoryClient().ResetQueue(ctx, request); err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\treturn nil\n}\n\n// DescribeQueue describes processing queue states\nfunc (adh *adminHandlerImpl) DescribeQueue(\n\tctx context.Context,\n\trequest *types.DescribeQueueRequest,\n) (resp *types.DescribeQueueResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminDescribeQueueScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.Type == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.GetClusterName() == \"\" {\n\t\treturn nil, adh.error(validate.ErrClusterNameNotSet, scope)\n\t}\n\n\treturn adh.GetHistoryClient().DescribeQueue(ctx, request)\n}\n\n// DescribeShardDistribution returns information about history shard distribution\nfunc (adh *adminHandlerImpl) DescribeShardDistribution(\n\tctx context.Context,\n\trequest *types.DescribeShardDistributionRequest,\n) (resp *types.DescribeShardDistributionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\t_, sw := adh.startRequestProfile(ctx, metrics.AdminDescribeShardDistributionScope)\n\tdefer sw.Stop()\n\n\tresp = &types.DescribeShardDistributionResponse{\n\t\tNumberOfShards: int32(adh.numberOfHistoryShards),\n\t\tShards:         make(map[int32]string),\n\t}\n\n\toffset := int(request.PageID * request.PageSize)\n\tnextPageStart := offset + int(request.PageSize)\n\tfor shardID := offset; shardID < adh.numberOfHistoryShards && shardID < nextPageStart; shardID++ {\n\t\tinfo, err := lookup.HistoryServerByShardID(adh.GetMembershipResolver(), shardID)\n\t\tif err != nil {\n\t\t\tresp.Shards[int32(shardID)] = \"unknown\"\n\t\t} else {\n\t\t\tresp.Shards[int32(shardID)] = info.Identity()\n\t\t}\n\t}\n\treturn resp, nil\n}\n\n// DescribeHistoryHost returns information about the internal states of a history host\nfunc (adh *adminHandlerImpl) DescribeHistoryHost(\n\tctx context.Context,\n\trequest *types.DescribeHistoryHostRequest,\n) (resp *types.DescribeHistoryHostResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminDescribeHistoryHostScope)\n\tdefer sw.Stop()\n\n\tif request == nil || (request.ShardIDForHost == nil && request.ExecutionForHost == nil && request.HostAddress == nil) {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif request.ExecutionForHost != nil {\n\t\tif err := validate.CheckExecution(request.ExecutionForHost); err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t}\n\n\treturn adh.GetHistoryClient().DescribeHistoryHost(ctx, request)\n}\n\n// GetWorkflowExecutionRawHistoryV2 - retrieves the history of workflow execution\nfunc (adh *adminHandlerImpl) GetWorkflowExecutionRawHistoryV2(\n\tctx context.Context,\n\trequest *types.GetWorkflowExecutionRawHistoryV2Request,\n) (resp *types.GetWorkflowExecutionRawHistoryV2Response, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetWorkflowExecutionRawHistoryV2Scope)\n\tdefer sw.Stop()\n\n\tif err := adh.validateGetWorkflowExecutionRawHistoryV2Request(\n\t\trequest,\n\t); err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\tdomainID, err := adh.GetDomainCache().GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\tscope = scope.Tagged(metrics.DomainTag(request.GetDomain()))\n\n\texecution := request.Execution\n\tvar pageToken *getWorkflowRawHistoryV2Token\n\tvar targetVersionHistory *persistence.VersionHistory\n\tif request.NextPageToken == nil {\n\t\tresponse, err := adh.GetHistoryClient().GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tExecution:  execution,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\n\t\tversionHistories := persistence.NewVersionHistoriesFromInternalType(\n\t\t\tresponse.GetVersionHistories(),\n\t\t)\n\t\ttargetVersionHistory, err = adh.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\t\trequest,\n\t\t\tversionHistories,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\n\t\tpageToken = adh.generatePaginationToken(request, versionHistories)\n\t} else {\n\t\tpageToken, err = deserializeRawHistoryToken(request.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t\tversionHistories := pageToken.VersionHistories\n\t\tif versionHistories == nil {\n\t\t\treturn nil, adh.error(&types.BadRequestError{Message: \"Invalid version histories.\"}, scope)\n\t\t}\n\t\ttargetVersionHistory, err = adh.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\t\trequest,\n\t\t\tpersistence.NewVersionHistoriesFromInternalType(versionHistories),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t}\n\n\tif err := adh.validatePaginationToken(\n\t\trequest,\n\t\tpageToken,\n\t); err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\tif pageToken.StartEventID+1 == pageToken.EndEventID {\n\t\t// API is exclusive-exclusive. Return empty response here.\n\t\treturn &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\t\tHistoryBatches: []*types.DataBlob{},\n\t\t\tNextPageToken:  nil, // no further pagination\n\t\t\tVersionHistory: targetVersionHistory.ToInternalType(),\n\t\t}, nil\n\t}\n\tpageSize := int(request.GetMaximumPageSize())\n\tshardID := common.WorkflowIDToHistoryShard(\n\t\texecution.GetWorkflowID(),\n\t\tadh.numberOfHistoryShards,\n\t)\n\n\trawHistoryResponse, err := adh.GetHistoryManager().ReadRawHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken: targetVersionHistory.GetBranchToken(),\n\t\t// GetWorkflowExecutionRawHistoryV2 is exclusive exclusive.\n\t\t// ReadRawHistoryBranch is inclusive exclusive.\n\t\tMinEventID:    pageToken.StartEventID + 1,\n\t\tMaxEventID:    pageToken.EndEventID,\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: pageToken.PersistenceToken,\n\t\tShardID:       common.IntPtr(shardID),\n\t\tDomainName:    request.GetDomain(),\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t// when no events can be returned from DB, DB layer will return\n\t\t\t// EntityNotExistsError, this API shall return empty response\n\t\t\treturn &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\t\t\tHistoryBatches: []*types.DataBlob{},\n\t\t\t\tNextPageToken:  nil, // no further pagination\n\t\t\t\tVersionHistory: targetVersionHistory.ToInternalType(),\n\t\t\t}, nil\n\t\t}\n\t\treturn nil, err\n\t}\n\n\tpageToken.PersistenceToken = rawHistoryResponse.NextPageToken\n\tsize := rawHistoryResponse.Size\n\tscope.RecordTimer(metrics.HistorySize, time.Duration(size))\n\n\trawBlobs := rawHistoryResponse.HistoryEventBlobs\n\tblobs := []*types.DataBlob{}\n\tfor _, blob := range rawBlobs {\n\t\tblobs = append(blobs, blob.ToInternal())\n\t}\n\n\tresult := &types.GetWorkflowExecutionRawHistoryV2Response{\n\t\tHistoryBatches: blobs,\n\t\tVersionHistory: targetVersionHistory.ToInternalType(),\n\t}\n\tif len(pageToken.PersistenceToken) == 0 {\n\t\tresult.NextPageToken = nil\n\t} else {\n\t\tresult.NextPageToken, err = serializeRawHistoryToken(pageToken)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\n// DescribeCluster return information about cadence deployment\nfunc (adh *adminHandlerImpl) DescribeCluster(\n\tctx context.Context,\n) (resp *types.DescribeClusterResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminDescribeClusterScope)\n\tdefer sw.Stop()\n\n\t// expose visibility store backend and if advanced options are available\n\tave := types.PersistenceFeature{\n\t\tKey:     \"advancedVisibilityEnabled\",\n\t\tEnabled: adh.params.ESConfig != nil,\n\t}\n\tvisibilityStoreInfo := types.PersistenceInfo{\n\t\tBackend:  adh.Resource.GetVisibilityManager().GetName(),\n\t\tFeatures: []*types.PersistenceFeature{&ave},\n\t}\n\n\t// expose history store backend\n\thistoryStoreInfo := types.PersistenceInfo{\n\t\tBackend: adh.GetHistoryManager().GetName(),\n\t}\n\n\tmembershipInfo := types.MembershipInfo{}\n\tif monitor := adh.GetMembershipResolver(); monitor != nil {\n\t\tcurrentHost, err := monitor.WhoAmI()\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\n\t\tmembershipInfo.CurrentHost = &types.HostInfo{\n\t\t\tIdentity: currentHost.Identity(),\n\t\t}\n\n\t\tvar rings []*types.RingInfo\n\t\tfor _, role := range service.ListWithRing {\n\t\t\tvar servers []*types.HostInfo\n\t\t\tmembers, err := monitor.Members(role)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, adh.error(err, scope)\n\t\t\t}\n\n\t\t\tfor _, server := range members {\n\t\t\t\tservers = append(servers, &types.HostInfo{\n\t\t\t\t\tIdentity: server.Identity(),\n\t\t\t\t})\n\t\t\t\tmembershipInfo.ReachableMembers = append(membershipInfo.ReachableMembers, server.Identity())\n\t\t\t}\n\n\t\t\trings = append(rings, &types.RingInfo{\n\t\t\t\tRole:        role,\n\t\t\t\tMemberCount: int32(len(servers)),\n\t\t\t\tMembers:     servers,\n\t\t\t})\n\t\t}\n\t\tmembershipInfo.Rings = rings\n\t}\n\n\treturn &types.DescribeClusterResponse{\n\t\tSupportedClientVersions: &types.SupportedClientVersions{\n\t\t\tGoSdk:   client.SupportedGoSDKVersion,\n\t\t\tJavaSdk: client.SupportedJavaSDKVersion,\n\t\t},\n\t\tMembershipInfo: &membershipInfo,\n\t\tPersistenceInfo: map[string]*types.PersistenceInfo{\n\t\t\t\"visibilityStore\": &visibilityStoreInfo,\n\t\t\t\"historyStore\":    &historyStoreInfo,\n\t\t},\n\t}, nil\n}\n\n// GetReplicationMessages returns new replication tasks since the read level provided in the token.\nfunc (adh *adminHandlerImpl) GetReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetReplicationMessagesRequest,\n) (resp *types.GetReplicationMessagesResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetReplicationMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.ClusterName == \"\" {\n\t\treturn nil, adh.error(validate.ErrClusterNameNotSet, scope)\n\t}\n\n\tresp, err = adh.GetHistoryRawClient().GetReplicationMessages(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\n// GetDomainReplicationMessages returns new domain replication tasks since last retrieved task ID.\nfunc (adh *adminHandlerImpl) GetDomainReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetDomainReplicationMessagesRequest,\n) (resp *types.GetDomainReplicationMessagesResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetDomainReplicationMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif adh.GetDomainReplicationQueue() == nil {\n\t\treturn nil, adh.error(errors.New(\"domain replication queue not enabled for cluster\"), scope)\n\t}\n\n\tlastMessageID := defaultLastMessageID\n\tif request.LastRetrievedMessageID != nil {\n\t\tlastMessageID = request.GetLastRetrievedMessageID()\n\t}\n\n\tif lastMessageID == defaultLastMessageID {\n\t\tclusterAckLevels, err := adh.GetDomainReplicationQueue().GetAckLevels(ctx)\n\t\tif err == nil {\n\t\t\tif ackLevel, ok := clusterAckLevels[request.GetClusterName()]; ok {\n\t\t\t\tlastMessageID = ackLevel\n\t\t\t}\n\t\t}\n\t}\n\n\treplicationTasks, lastMessageID, err := adh.GetDomainReplicationQueue().GetReplicationMessages(\n\t\tctx,\n\t\tlastMessageID,\n\t\tgetDomainReplicationMessageBatchSize,\n\t)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\tlastProcessedMessageID := defaultLastMessageID\n\tif request.LastProcessedMessageID != nil {\n\t\tlastProcessedMessageID = request.GetLastProcessedMessageID()\n\t}\n\tif err := adh.GetDomainReplicationQueue().UpdateAckLevel(ctx, lastProcessedMessageID, request.GetClusterName()); err != nil {\n\t\tadh.GetLogger().Warn(\"Failed to update domain replication queue ack level.\",\n\t\t\ttag.TaskID(int64(lastProcessedMessageID)),\n\t\t\ttag.ClusterName(request.GetClusterName()))\n\t}\n\n\treturn &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: &types.ReplicationMessages{\n\t\t\tReplicationTasks:       replicationTasks,\n\t\t\tLastRetrievedMessageID: lastMessageID,\n\t\t},\n\t}, nil\n}\n\n// GetDLQReplicationMessages returns new replication tasks based on the dlq info.\nfunc (adh *adminHandlerImpl) GetDLQReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetDLQReplicationMessagesRequest,\n) (resp *types.GetDLQReplicationMessagesResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetDLQReplicationMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif len(request.GetTaskInfos()) == 0 {\n\t\treturn nil, adh.error(validate.ErrEmptyReplicationInfo, scope)\n\t}\n\n\tresp, err = adh.GetHistoryClient().GetDLQReplicationMessages(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\n// ReapplyEvents applies stale events to the current workflow and the current run\nfunc (adh *adminHandlerImpl) ReapplyEvents(\n\tctx context.Context,\n\trequest *types.ReapplyEventsRequest,\n) (err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminReapplyEventsScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.GetDomainName() == \"\" {\n\t\treturn adh.error(validate.ErrDomainNotSet, scope)\n\t}\n\tif request.WorkflowExecution == nil {\n\t\treturn adh.error(validate.ErrExecutionNotSet, scope)\n\t}\n\tif request.GetWorkflowExecution().GetWorkflowID() == \"\" {\n\t\treturn adh.error(validate.ErrWorkflowIDNotSet, scope)\n\t}\n\tif request.GetEvents() == nil {\n\t\treturn adh.error(validate.ErrWorkflowIDNotSet, scope)\n\t}\n\tdomainEntry, err := adh.GetDomainCache().GetDomain(request.GetDomainName())\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\n\terr = adh.GetHistoryClient().ReapplyEvents(ctx, &types.HistoryReapplyEventsRequest{\n\t\tDomainUUID: domainEntry.GetInfo().ID,\n\t\tRequest:    request,\n\t})\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\treturn nil\n}\n\n// ReadDLQMessages reads messages from DLQ\nfunc (adh *adminHandlerImpl) ReadDLQMessages(\n\tctx context.Context,\n\trequest *types.ReadDLQMessagesRequest,\n) (resp *types.ReadDLQMessagesResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminReadDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif request.Type == nil {\n\t\treturn nil, adh.error(validate.ErrEmptyQueueType, scope)\n\t}\n\n\tif request.GetMaximumPageSize() <= 0 {\n\t\trequest.MaximumPageSize = constants.ReadDLQMessagesPageSize\n\t}\n\n\tif request.InclusiveEndMessageID == nil {\n\t\trequest.InclusiveEndMessageID = common.Ptr(constants.InclusiveEndMessageID)\n\t}\n\n\tvar tasks []*types.ReplicationTask\n\tvar token []byte\n\tvar op func(ctx context.Context) error\n\tswitch request.GetType() {\n\tcase types.DLQTypeReplication:\n\t\treturn adh.GetHistoryClient().ReadDLQMessages(ctx, request)\n\tcase types.DLQTypeDomain:\n\t\top = func(ctx context.Context) error {\n\t\t\tselect {\n\t\t\tcase <-ctx.Done():\n\t\t\t\treturn ctx.Err()\n\t\t\tdefault:\n\t\t\t\tvar err error\n\t\t\t\ttasks, token, err = adh.domainDLQHandler.Read(\n\t\t\t\t\tctx,\n\t\t\t\t\trequest.GetInclusiveEndMessageID(),\n\t\t\t\t\tint(request.GetMaximumPageSize()),\n\t\t\t\t\trequest.GetNextPageToken())\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: \"The DLQ type is not supported.\"}\n\t}\n\terr = adh.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\treturn &types.ReadDLQMessagesResponse{\n\t\tReplicationTasks: tasks,\n\t\tNextPageToken:    token,\n\t}, nil\n}\n\n// PurgeDLQMessages purge messages from DLQ\nfunc (adh *adminHandlerImpl) PurgeDLQMessages(\n\tctx context.Context,\n\trequest *types.PurgeDLQMessagesRequest,\n) (err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminPurgeDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif request.Type == nil {\n\t\treturn adh.error(validate.ErrEmptyQueueType, scope)\n\t}\n\n\tif request.InclusiveEndMessageID == nil {\n\t\trequest.InclusiveEndMessageID = common.Ptr(constants.InclusiveEndMessageID)\n\t}\n\n\tvar op func(ctx context.Context) error\n\tswitch request.GetType() {\n\tcase types.DLQTypeReplication:\n\t\treturn adh.GetHistoryClient().PurgeDLQMessages(ctx, request)\n\tcase types.DLQTypeDomain:\n\t\top = func(ctx context.Context) error {\n\t\t\tselect {\n\t\t\tcase <-ctx.Done():\n\t\t\t\treturn ctx.Err()\n\t\t\tdefault:\n\t\t\t\treturn adh.domainDLQHandler.Purge(\n\t\t\t\t\tctx,\n\t\t\t\t\trequest.GetInclusiveEndMessageID(),\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\tdefault:\n\t\treturn &types.BadRequestError{Message: \"The DLQ type is not supported.\"}\n\t}\n\terr = adh.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\n\treturn nil\n}\n\nfunc (adh *adminHandlerImpl) CountDLQMessages(\n\tctx context.Context,\n\trequest *types.CountDLQMessagesRequest,\n) (resp *types.CountDLQMessagesResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminCountDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tdomain, err := adh.domainDLQHandler.Count(ctx, request.ForceFetch)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\thistory, err := adh.GetHistoryClient().CountDLQMessages(ctx, request)\n\tif err != nil {\n\t\terr = adh.error(err, scope)\n\t}\n\n\treturn &types.CountDLQMessagesResponse{\n\t\tHistory: history.Entries,\n\t\tDomain:  domain,\n\t}, err\n}\n\n// MergeDLQMessages merges DLQ messages\nfunc (adh *adminHandlerImpl) MergeDLQMessages(\n\tctx context.Context,\n\trequest *types.MergeDLQMessagesRequest,\n) (resp *types.MergeDLQMessagesResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminMergeDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tif request.Type == nil {\n\t\treturn nil, adh.error(validate.ErrEmptyQueueType, scope)\n\t}\n\n\tif request.InclusiveEndMessageID == nil {\n\t\trequest.InclusiveEndMessageID = common.Ptr(constants.InclusiveEndMessageID)\n\t}\n\n\tvar token []byte\n\tvar op func(ctx context.Context) error\n\tswitch request.GetType() {\n\tcase types.DLQTypeReplication:\n\t\treturn adh.GetHistoryClient().MergeDLQMessages(ctx, request)\n\tcase types.DLQTypeDomain:\n\n\t\top = func(ctx context.Context) error {\n\t\t\tselect {\n\t\t\tcase <-ctx.Done():\n\t\t\t\treturn ctx.Err()\n\t\t\tdefault:\n\t\t\t\tvar err error\n\t\t\t\ttoken, err = adh.domainDLQHandler.Merge(\n\t\t\t\t\tctx,\n\t\t\t\t\trequest.GetInclusiveEndMessageID(),\n\t\t\t\t\tint(request.GetMaximumPageSize()),\n\t\t\t\t\trequest.GetNextPageToken(),\n\t\t\t\t)\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\tdefault:\n\t\treturn nil, &types.BadRequestError{Message: \"The DLQ type is not supported.\"}\n\t}\n\terr = adh.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\treturn &types.MergeDLQMessagesResponse{\n\t\tNextPageToken: token,\n\t}, nil\n}\n\n// RefreshWorkflowTasks re-generates the workflow tasks\nfunc (adh *adminHandlerImpl) RefreshWorkflowTasks(\n\tctx context.Context,\n\trequest *types.RefreshWorkflowTasksRequest,\n) (err error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminRefreshWorkflowTasksScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif err := validate.CheckExecution(request.Execution); err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\tdomainEntry, err := adh.GetDomainCache().GetDomain(request.GetDomain())\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\n\terr = adh.GetHistoryClient().RefreshWorkflowTasks(ctx, &types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: domainEntry.GetInfo().ID,\n\t\tRequest:    request,\n\t})\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\treturn nil\n}\n\n// ResendReplicationTasks requests replication task from remote cluster\nfunc (adh *adminHandlerImpl) ResendReplicationTasks(\n\tctx context.Context,\n\trequest *types.ResendReplicationTasksRequest,\n) (err error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminResendReplicationTasksScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tresender := ndc.NewHistoryResender(\n\t\tadh.GetDomainCache(),\n\t\tadh.GetClientBean(),\n\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\treturn adh.GetHistoryClient().ReplicateEventsV2(ctx, request)\n\t\t},\n\t\tnil,\n\t\tnil,\n\t\tadh.GetLogger(),\n\t)\n\treturn resender.SendSingleWorkflowHistory(\n\t\trequest.GetRemoteCluster(),\n\t\trequest.DomainID,\n\t\trequest.GetWorkflowID(),\n\t\trequest.GetRunID(),\n\t\trequest.StartEventID,\n\t\trequest.StartVersion,\n\t\trequest.EndEventID,\n\t\trequest.EndVersion,\n\t)\n}\n\nfunc (adh *adminHandlerImpl) GetCrossClusterTasks(\n\tctx context.Context,\n\trequest *types.GetCrossClusterTasksRequest,\n) (resp *types.GetCrossClusterTasksResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetCrossClusterTasksScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.TargetCluster == \"\" {\n\t\treturn nil, adh.error(validate.ErrClusterNameNotSet, scope)\n\t}\n\n\tresp, err = adh.GetHistoryRawClient().GetCrossClusterTasks(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\trequest *types.RespondCrossClusterTasksCompletedRequest,\n) (resp *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &err) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminRespondCrossClusterTasksCompletedScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tif request.TargetCluster == \"\" {\n\t\treturn nil, adh.error(validate.ErrClusterNameNotSet, scope)\n\t}\n\n\tresp, err = adh.GetHistoryClient().RespondCrossClusterTasksCompleted(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) validateGetWorkflowExecutionRawHistoryV2Request(\n\trequest *types.GetWorkflowExecutionRawHistoryV2Request,\n) error {\n\n\texecution := request.Execution\n\tif len(execution.GetWorkflowID()) == 0 {\n\t\treturn &types.BadRequestError{Message: \"Invalid WorkflowID.\"}\n\t}\n\t// TODO currently, this API is only going to be used by re-send history events\n\t// to remote cluster if kafka is lossy again, in the future, this API can be used\n\t// by CLI and client, then empty runID (meaning the current workflow) should be allowed\n\tif _, err := uuid.Parse(execution.GetRunID()); err != nil {\n\t\treturn &types.BadRequestError{Message: \"Invalid RunID.\"}\n\t}\n\n\tpageSize := int(request.GetMaximumPageSize())\n\tif pageSize <= 0 {\n\t\treturn &types.BadRequestError{Message: \"Invalid PageSize.\"}\n\t}\n\n\tif (request.StartEventID != nil && request.StartEventVersion == nil) ||\n\t\t(request.StartEventID == nil && request.StartEventVersion != nil) {\n\t\treturn &types.BadRequestError{Message: \"Invalid start event id and start event version combination.\"}\n\t}\n\n\tif (request.EndEventID != nil && request.EndEventVersion == nil) ||\n\t\t(request.EndEventID == nil && request.EndEventVersion != nil) {\n\t\treturn &types.BadRequestError{Message: \"Invalid end event id and end event version combination.\"}\n\t}\n\treturn nil\n}\n\nfunc (adh *adminHandlerImpl) validateConfigForAdvanceVisibility() error {\n\tif adh.params.ESConfig == nil || adh.params.ESClient == nil {\n\t\treturn errors.New(\"ES related config not found\")\n\t}\n\treturn nil\n}\n\nfunc (adh *adminHandlerImpl) setRequestDefaultValueAndGetTargetVersionHistory(\n\trequest *types.GetWorkflowExecutionRawHistoryV2Request,\n\tversionHistories *persistence.VersionHistories,\n) (*persistence.VersionHistory, error) {\n\n\ttargetBranch, err := versionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfirstItem, err := targetBranch.GetFirstItem()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastItem, err := targetBranch.GetLastItem()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif request.StartEventID == nil || request.StartEventVersion == nil {\n\t\t// If start event is not set, get the events from the first event\n\t\t// As the API is exclusive-exclusive, use first event id - 1 here\n\t\trequest.StartEventID = common.Int64Ptr(constants.FirstEventID - 1)\n\t\trequest.StartEventVersion = common.Int64Ptr(firstItem.Version)\n\t}\n\tif request.EndEventID == nil || request.EndEventVersion == nil {\n\t\t// If end event is not set, get the events until the end event\n\t\t// As the API is exclusive-exclusive, use end event id + 1 here\n\t\trequest.EndEventID = common.Int64Ptr(lastItem.EventID + 1)\n\t\trequest.EndEventVersion = common.Int64Ptr(lastItem.Version)\n\t}\n\n\tif request.GetStartEventID() < 0 {\n\t\treturn nil, &types.BadRequestError{Message: \"Invalid FirstEventID && NextEventID combination.\"}\n\t}\n\n\t// get branch based on the end event if end event is defined in the request\n\tif request.GetEndEventID() == lastItem.EventID+1 &&\n\t\trequest.GetEndEventVersion() == lastItem.Version {\n\t\t// this is a special case, target branch remains the same\n\t} else {\n\t\tendItem := persistence.NewVersionHistoryItem(request.GetEndEventID(), request.GetEndEventVersion())\n\t\t_, targetBranch, err = versionHistories.FindFirstVersionHistoryByItem(endItem)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tstartItem := persistence.NewVersionHistoryItem(request.GetStartEventID(), request.GetStartEventVersion())\n\t// If the request start event is defined. The start event may be on a different branch as current branch.\n\t// We need to find the LCA of the start event and the current branch.\n\tif request.GetStartEventID() == constants.FirstEventID-1 &&\n\t\trequest.GetStartEventVersion() == firstItem.Version {\n\t\t// this is a special case, start event is on the same branch as target branch\n\t} else {\n\t\tif !targetBranch.ContainsItem(startItem) {\n\t\t\t_, startBranch, err := versionHistories.FindFirstVersionHistoryByItem(startItem)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tstartItem, err = targetBranch.FindLCAItem(startBranch)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\trequest.StartEventID = common.Int64Ptr(startItem.EventID)\n\t\t\trequest.StartEventVersion = common.Int64Ptr(startItem.Version)\n\t\t}\n\t}\n\n\treturn targetBranch, nil\n}\n\nfunc (adh *adminHandlerImpl) generatePaginationToken(\n\trequest *types.GetWorkflowExecutionRawHistoryV2Request,\n\tversionHistories *persistence.VersionHistories,\n) *getWorkflowRawHistoryV2Token {\n\n\texecution := request.Execution\n\treturn &getWorkflowRawHistoryV2Token{\n\t\tDomainName:        request.GetDomain(),\n\t\tWorkflowID:        execution.GetWorkflowID(),\n\t\tRunID:             execution.GetRunID(),\n\t\tStartEventID:      request.GetStartEventID(),\n\t\tStartEventVersion: request.GetStartEventVersion(),\n\t\tEndEventID:        request.GetEndEventID(),\n\t\tEndEventVersion:   request.GetEndEventVersion(),\n\t\tVersionHistories:  versionHistories.ToInternalType(),\n\t\tPersistenceToken:  nil, // this is the initialized value\n\t}\n}\n\nfunc (adh *adminHandlerImpl) validatePaginationToken(\n\trequest *types.GetWorkflowExecutionRawHistoryV2Request,\n\ttoken *getWorkflowRawHistoryV2Token,\n) error {\n\n\texecution := request.Execution\n\tif request.GetDomain() != token.DomainName ||\n\t\texecution.GetWorkflowID() != token.WorkflowID ||\n\t\texecution.GetRunID() != token.RunID ||\n\t\trequest.GetStartEventID() != token.StartEventID ||\n\t\trequest.GetStartEventVersion() != token.StartEventVersion ||\n\t\trequest.GetEndEventID() != token.EndEventID ||\n\t\trequest.GetEndEventVersion() != token.EndEventVersion {\n\t\treturn &types.BadRequestError{Message: \"Invalid pagination token.\"}\n\t}\n\treturn nil\n}\n\n// startRequestProfile initiates recording of request metrics\nfunc (adh *adminHandlerImpl) startRequestProfile(ctx context.Context, scope metrics.ScopeIdx) (metrics.Scope, metrics.Stopwatch) {\n\tmetricsScope := adh.GetMetricsClient().Scope(scope).Tagged(metrics.DomainUnknownTag()).Tagged(metrics.GetContextTags(ctx)...)\n\tsw := metricsScope.StartTimer(metrics.CadenceLatency)\n\tmetricsScope.IncCounter(metrics.CadenceRequests)\n\treturn metricsScope, sw\n}\n\nfunc (adh *adminHandlerImpl) error(err error, scope metrics.Scope) error {\n\tlogger := adh.GetLogger().Helper()\n\tswitch err.(type) {\n\tcase *types.InternalServiceError:\n\t\tlogger.Error(\"Internal service error\", tag.Error(err))\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\treturn err\n\tcase *types.BadRequestError:\n\t\tscope.IncCounter(metrics.CadenceErrBadRequestCounter)\n\t\treturn err\n\tcase *types.ServiceBusyError:\n\t\tscope.IncCounter(metrics.CadenceErrServiceBusyCounter)\n\t\treturn err\n\tcase *types.EntityNotExistsError:\n\t\treturn err\n\tdefault:\n\t\tlogger.Error(\"Uncategorized error\", tag.Error(err))\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\treturn &types.InternalServiceError{Message: err.Error()}\n\t}\n}\n\nfunc convertIndexedValueTypeToESDataType(valueType types.IndexedValueType) string {\n\tswitch valueType {\n\tcase types.IndexedValueTypeString:\n\t\treturn \"text\"\n\tcase types.IndexedValueTypeKeyword:\n\t\treturn \"keyword\"\n\tcase types.IndexedValueTypeInt:\n\t\treturn \"long\"\n\tcase types.IndexedValueTypeDouble:\n\t\treturn \"double\"\n\tcase types.IndexedValueTypeBool:\n\t\treturn \"boolean\"\n\tcase types.IndexedValueTypeDatetime:\n\t\treturn \"date\"\n\tdefault:\n\t\treturn \"\"\n\t}\n}\n\nfunc serializeRawHistoryToken(token *getWorkflowRawHistoryV2Token) ([]byte, error) {\n\tif token == nil {\n\t\treturn nil, nil\n\t}\n\n\tbytes, err := json.Marshal(token)\n\treturn bytes, err\n}\n\nfunc deserializeRawHistoryToken(bytes []byte) (*getWorkflowRawHistoryV2Token, error) {\n\ttoken := &getWorkflowRawHistoryV2Token{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc (adh *adminHandlerImpl) GetDynamicConfig(ctx context.Context, request *types.GetDynamicConfigRequest) (_ *types.GetDynamicConfigResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminGetDynamicConfigScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.ConfigName == \"\" {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tkeyVal, err := dynamicproperties.GetKeyFromKeyName(request.ConfigName)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\tvar value interface{}\n\tif request.Filters == nil {\n\t\tvalue, err = adh.params.DynamicConfig.GetValue(keyVal)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t} else {\n\t\tconvFilters, err := convertFilterListToMap(request.Filters)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t\tvalue, err = adh.params.DynamicConfig.GetValueWithFilters(keyVal, convFilters)\n\t\tif err != nil {\n\t\t\treturn nil, adh.error(err, scope)\n\t\t}\n\t}\n\n\tdata, err := json.Marshal(value)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\treturn &types.GetDynamicConfigResponse{\n\t\tValue: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         data,\n\t\t},\n\t}, nil\n}\n\nfunc (adh *adminHandlerImpl) UpdateDynamicConfig(ctx context.Context, request *types.UpdateDynamicConfigRequest) (retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminUpdateDynamicConfigScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.ConfigName == \"\" {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tkeyVal, err := dynamicproperties.GetKeyFromKeyName(request.ConfigName)\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\n\treturn adh.params.DynamicConfig.UpdateValue(keyVal, request.ConfigValues)\n}\n\nfunc (adh *adminHandlerImpl) RestoreDynamicConfig(ctx context.Context, request *types.RestoreDynamicConfigRequest) (retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminRestoreDynamicConfigScope)\n\tdefer sw.Stop()\n\n\tif request == nil || request.ConfigName == \"\" {\n\t\treturn adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tkeyVal, err := dynamicproperties.GetKeyFromKeyName(request.ConfigName)\n\tif err != nil {\n\t\treturn adh.error(err, scope)\n\t}\n\n\tvar filters map[dynamicproperties.Filter]interface{}\n\n\tif request.Filters == nil {\n\t\tfilters = nil\n\t} else {\n\t\tfilters, err = convertFilterListToMap(request.Filters)\n\t\tif err != nil {\n\t\t\treturn adh.error(validate.ErrInvalidFilters, scope)\n\t\t}\n\t}\n\treturn adh.params.DynamicConfig.RestoreValue(keyVal, filters)\n}\n\nfunc (adh *adminHandlerImpl) ListDynamicConfig(ctx context.Context, request *types.ListDynamicConfigRequest) (_ *types.ListDynamicConfigResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.AdminListDynamicConfigScope)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tkeyVal, err := dynamicproperties.GetKeyFromKeyName(request.ConfigName)\n\tif err != nil || request.ConfigName == \"\" {\n\t\tentries, err2 := adh.params.DynamicConfig.ListValue(nil)\n\t\tif err2 != nil {\n\t\t\treturn nil, adh.error(err2, scope)\n\t\t}\n\t\treturn &types.ListDynamicConfigResponse{\n\t\t\tEntries: entries,\n\t\t}, nil\n\t}\n\n\tentries, err2 := adh.params.DynamicConfig.ListValue(keyVal)\n\tif err2 != nil {\n\t\terr = adh.error(err2, scope)\n\t\treturn nil, adh.error(err, scope)\n\t}\n\n\treturn &types.ListDynamicConfigResponse{\n\t\tEntries: entries,\n\t}, nil\n}\n\nfunc (adh *adminHandlerImpl) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest) (_ *types.GetGlobalIsolationGroupsResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.GetGlobalIsolationGroups)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tresp, err := adh.isolationGroups.GetGlobalState(ctx)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest) (_ *types.UpdateGlobalIsolationGroupsResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.UpdateGlobalIsolationGroups)\n\tdefer sw.Stop()\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\terr := adh.isolationGroups.UpdateGlobalState(ctx, *request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn &types.UpdateGlobalIsolationGroupsResponse{}, nil\n}\n\nfunc (adh *adminHandlerImpl) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest) (_ *types.GetDomainIsolationGroupsResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.GetDomainIsolationGroups)\n\tdefer sw.Stop()\n\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\n\tresp, err := adh.isolationGroups.GetDomainState(ctx, types.GetDomainIsolationGroupsRequest{Domain: request.Domain})\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest) (_ *types.UpdateDomainIsolationGroupsResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.UpdateDomainIsolationGroups)\n\tdefer sw.Stop()\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\terr := adh.isolationGroups.UpdateDomainState(ctx, *request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn &types.UpdateDomainIsolationGroupsResponse{}, nil\n}\n\nfunc (adh *adminHandlerImpl) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.GetDomainAsyncWorkflowConfiguratonRequest) (_ *types.GetDomainAsyncWorkflowConfiguratonResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.GetDomainAsyncWorkflowConfiguraton)\n\tdefer sw.Stop()\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tresp, err := adh.asyncWFQueueConfigs.GetConfiguraton(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (_ *types.UpdateDomainAsyncWorkflowConfiguratonResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.UpdateDomainAsyncWorkflowConfiguraton)\n\tdefer sw.Stop()\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tresp, err := adh.asyncWFQueueConfigs.UpdateConfiguration(ctx, request)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\treturn resp, nil\n}\n\nfunc (adh *adminHandlerImpl) UpdateTaskListPartitionConfig(ctx context.Context, request *types.UpdateTaskListPartitionConfigRequest) (_ *types.UpdateTaskListPartitionConfigResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), adh.GetLogger(), &retError) }()\n\tscope, sw := adh.startRequestProfile(ctx, metrics.UpdateTaskListPartitionConfig)\n\tdefer sw.Stop()\n\tif request == nil {\n\t\treturn nil, adh.error(validate.ErrRequestNotSet, scope)\n\t}\n\tdomainID, err := adh.GetDomainCache().GetDomainID(request.Domain)\n\tif err != nil {\n\t\treturn nil, adh.error(err, scope)\n\t}\n\tif request.TaskList == nil {\n\t\treturn nil, adh.error(validate.ErrTaskListNotSet, scope)\n\t}\n\tif request.TaskList.Kind == nil {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"Task list kind not set.\"}, scope)\n\t}\n\tif *request.TaskList.Kind != types.TaskListKindNormal {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"Only normal tasklist's partition config can be updated.\"}, scope)\n\t}\n\tif request.TaskListType == nil {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"Task list type not set.\"}, scope)\n\t}\n\tif request.PartitionConfig == nil {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"Task list partition config is not set in the request.\"}, scope)\n\t}\n\tif len(request.PartitionConfig.WritePartitions) > len(request.PartitionConfig.ReadPartitions) {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"The number of write partitions cannot be larger than the number of read partitions.\"}, scope)\n\t}\n\tif len(request.PartitionConfig.WritePartitions) <= 0 {\n\t\treturn nil, adh.error(&types.BadRequestError{Message: \"The number of partitions must be larger than 0.\"}, scope)\n\t}\n\t_, err = adh.GetMatchingClient().UpdateTaskListPartitionConfig(ctx, &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\tDomainUUID:      domainID,\n\t\tTaskList:        request.TaskList,\n\t\tTaskListType:    request.TaskListType,\n\t\tPartitionConfig: request.PartitionConfig,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.UpdateTaskListPartitionConfigResponse{}, nil\n}\n\nfunc convertFromDataBlob(blob *types.DataBlob) (interface{}, error) {\n\tswitch *blob.EncodingType {\n\tcase types.EncodingTypeJSON:\n\t\tvar v interface{}\n\t\terr := json.Unmarshal(blob.Data, &v)\n\t\treturn v, err\n\tdefault:\n\t\treturn nil, errors.New(\"unsupported blob encoding\")\n\t}\n}\n\nfunc convertFilterListToMap(filters []*types.DynamicConfigFilter) (map[dynamicproperties.Filter]interface{}, error) {\n\tnewFilters := make(map[dynamicproperties.Filter]interface{})\n\n\tfor _, filter := range filters {\n\t\tval, err := convertFromDataBlob(filter.Value)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tnewFilters[dynamicproperties.ParseFilter(filter.Name)] = val\n\t}\n\treturn newFilters, nil\n}\n"
  },
  {
    "path": "service/frontend/admin/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage admin\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queueconfigapi\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tesmock \"github.com/uber/cadence/common/elasticsearch/mocks\"\n\t\"github.com/uber/cadence/common/isolationgroup/isolationgroupapi\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\ntype (\n\tadminHandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller        *gomock.Controller\n\t\tmockResource      *resource.Test\n\t\tmockHistoryClient *history.MockClient\n\t\tmockDomainCache   *cache.MockDomainCache\n\t\tfrontendClient    *frontend.MockClient\n\t\tmockResolver      *membership.MockResolver\n\n\t\tmockHistoryV2Mgr *mocks.HistoryV2Manager\n\n\t\tdomainName string\n\t\tdomainID   string\n\n\t\thandler *adminHandlerImpl\n\t}\n)\n\nfunc TestAdminHandlerSuite(t *testing.T) {\n\ts := new(adminHandlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *adminHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.domainName = \"some random domain name\"\n\ts.domainID = \"some random domain ID\"\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Frontend)\n\ts.mockDomainCache = s.mockResource.DomainCache\n\ts.mockHistoryClient = s.mockResource.HistoryClient\n\ts.mockHistoryV2Mgr = s.mockResource.HistoryMgr\n\ts.frontendClient = s.mockResource.FrontendClient\n\ts.mockResolver = s.mockResource.MembershipResolver\n\n\tparams := &resource.Params{\n\t\tLogger:          testlogger.New(s.T()),\n\t\tThrottledLogger: testlogger.New(s.T()),\n\t\tMetricScope:     tally.NewTestScope(service.Frontend, make(map[string]string)),\n\t\tMetricsClient:   metrics.NewNoopMetricsClient(),\n\t\tPersistenceConfig: config.Persistence{\n\t\t\tNumHistoryShards: 1,\n\t\t},\n\t}\n\tconfig := &frontendcfg.Config{\n\t\tEnableAdminProtection:  dynamicproperties.GetBoolPropertyFn(false),\n\t\tEnableGracefulFailover: dynamicproperties.GetBoolPropertyFn(false),\n\t}\n\n\tdh := domain.NewMockHandler(s.controller)\n\ts.handler = NewHandler(s.mockResource, params, config, dh).(*adminHandlerImpl)\n\ts.handler.Start()\n}\n\nfunc (s *adminHandlerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n\ts.handler.Stop()\n}\n\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_NormalWorkflow() {\n\ts.testMaintainCorruptWorkflow(nil, nil, false)\n}\n\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_WorkflowDoesNotExist() {\n\terr := &types.EntityNotExistsError{Message: \"Workflow does not exist\"}\n\ts.testMaintainCorruptWorkflow(err, nil, false)\n}\n\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_NoStartEvent() {\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.domainName, nil).AnyTimes()\n\terr := &types.InternalServiceError{Message: \"unable to get workflow start event\"}\n\ts.testMaintainCorruptWorkflow(err, nil, true)\n}\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_NoStartEventHistory() {\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.domainName, nil).AnyTimes()\n\terr := &types.InternalServiceError{Message: \"unable to get workflow start event\"}\n\ts.testMaintainCorruptWorkflow(nil, err, true)\n}\n\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_UnableToGetScheduledEvent() {\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.domainName, nil).AnyTimes()\n\terr := &types.InternalServiceError{Message: \"unable to get activity scheduled event\"}\n\ts.testMaintainCorruptWorkflow(err, nil, true)\n}\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_UnableToGetScheduledEventHistory() {\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.domainName, nil).AnyTimes()\n\terr := &types.InternalServiceError{Message: \"unable to get activity scheduled event\"}\n\ts.testMaintainCorruptWorkflow(nil, err, true)\n}\n\nfunc (s *adminHandlerSuite) TestMaintainCorruptWorkflow_CorruptedHistory() {\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.domainName, nil).AnyTimes()\n\terr := &types.InternalDataInconsistencyError{\n\t\tMessage: \"corrupted history event batch, eventID is not continuous\",\n\t}\n\ts.testMaintainCorruptWorkflow(err, nil, true)\n}\n\nfunc (s *adminHandlerSuite) testMaintainCorruptWorkflow(\n\tdescribeWorkflowError error,\n\tgetHistoryError error,\n\texpectDeletion bool,\n) {\n\thandler := s.handler\n\thandler.params = &resource.Params{}\n\tctx := context.Background()\n\n\trequest := &types.AdminMaintainWorkflowRequest{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"someWorkflowID\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tSkipErrors: true,\n\t}\n\n\t// need to reeturn error here to start deleting\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{}\n\ts.frontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\tReturn(describeResp, describeWorkflowError)\n\n\t// need to reeturn error here to start deleting\n\thistoryResponse := &types.GetWorkflowExecutionHistoryResponse{}\n\ts.frontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).\n\t\tReturn(historyResponse, getHistoryError).AnyTimes()\n\n\tif expectDeletion {\n\t\thostInfo := membership.NewHostInfo(\"taskListA:thriftPort\")\n\t\ts.mockResolver.EXPECT().Lookup(gomock.Any(), gomock.Any()).Return(hostInfo, nil)\n\t\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(s.domainID, nil)\n\n\t\ttestMutableState := &types.DescribeMutableStateResponse{\n\t\t\tMutableStateInDatabase: \"{\\\"ExecutionInfo\\\":{\\\"BranchToken\\\":\\\"WQsACgAAACQ2MzI5YzEzMi1mMGI0LTQwZmUtYWYxMS1hODVmMDA3MzAzODQLABQAAAAkOWM5OWI1MjItMGEyZi00NTdmLWEyNDgtMWU0OTA0ZDg4YzVhDwAeDAAAAAAA\\\"}}\",\n\t\t}\n\t\ts.mockHistoryClient.EXPECT().DescribeMutableState(gomock.Any(), gomock.Any()).Return(testMutableState, nil)\n\n\t\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\ts.mockResource.ExecutionMgr.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\ts.mockResource.ExecutionMgr.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\ts.mockResource.VisibilityMgr.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\t}\n\n\t_, err := handler.MaintainCorruptWorkflow(ctx, request)\n\ts.Nil(err)\n}\n\nfunc (s *adminHandlerSuite) Test_ConvertIndexedValueTypeToESDataType() {\n\ttests := []struct {\n\t\tinput    types.IndexedValueType\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeString,\n\t\t\texpected: \"text\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeKeyword,\n\t\t\texpected: \"keyword\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeInt,\n\t\t\texpected: \"long\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeDouble,\n\t\t\texpected: \"double\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeBool,\n\t\t\texpected: \"boolean\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueTypeDatetime,\n\t\t\texpected: \"date\",\n\t\t},\n\t\t{\n\t\t\tinput:    types.IndexedValueType(-1),\n\t\t\texpected: \"\",\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\ts.Equal(test.expected, convertIndexedValueTypeToESDataType(test.input))\n\t}\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2_FailedOnInvalidWorkflowID() {\n\n\tctx := context.Background()\n\t_, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"\",\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(1),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tEndEventID:        common.Int64Ptr(10),\n\t\t\tEndEventVersion:   common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   1,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2_FailedOnInvalidRunID() {\n\tctx := context.Background()\n\t_, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\tRunID:      \"runID\",\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(1),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tEndEventID:        common.Int64Ptr(10),\n\t\t\tEndEventVersion:   common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   1,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2_FailedOnInvalidSize() {\n\tctx := context.Background()\n\t_, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(1),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tEndEventID:        common.Int64Ptr(10),\n\t\t\tEndEventVersion:   common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   -1,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2_FailedOnDomainCache() {\n\tctx := context.Background()\n\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(\"\", fmt.Errorf(\"test\")).Times(1)\n\t_, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(1),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tEndEventID:        common.Int64Ptr(10),\n\t\t\tEndEventVersion:   common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   1,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2() {\n\tctx := context.Background()\n\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(s.domainID, nil).AnyTimes()\n\tbranchToken := []byte{1}\n\tversionHistory := persistence.NewVersionHistory(branchToken, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(int64(10), int64(100)),\n\t})\n\trawVersionHistories := persistence.NewVersionHistories(versionHistory)\n\tversionHistories := rawVersionHistories.ToInternalType()\n\tmState := &types.GetMutableStateResponse{\n\t\tNextEventID:        11,\n\t\tCurrentBranchToken: branchToken,\n\t\tVersionHistories:   versionHistories,\n\t}\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), gomock.Any()).Return(mState, nil).AnyTimes()\n\n\ts.mockHistoryV2Mgr.On(\"ReadRawHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadRawHistoryBranchResponse{\n\t\tHistoryEventBlobs: []*persistence.DataBlob{},\n\t\tNextPageToken:     []byte{},\n\t\tSize:              0,\n\t}, nil)\n\t_, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(1),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tEndEventID:        common.Int64Ptr(10),\n\t\t\tEndEventVersion:   common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   10,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetWorkflowExecutionRawHistoryV2_SameStartIDAndEndID() {\n\tctx := context.Background()\n\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(s.domainID, nil).AnyTimes()\n\tbranchToken := []byte{1}\n\tversionHistory := persistence.NewVersionHistory(branchToken, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(int64(10), int64(100)),\n\t})\n\trawVersionHistories := persistence.NewVersionHistories(versionHistory)\n\tversionHistories := rawVersionHistories.ToInternalType()\n\tmState := &types.GetMutableStateResponse{\n\t\tNextEventID:        11,\n\t\tCurrentBranchToken: branchToken,\n\t\tVersionHistories:   versionHistories,\n\t}\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), gomock.Any()).Return(mState, nil).AnyTimes()\n\n\tresp, err := s.handler.GetWorkflowExecutionRawHistoryV2(ctx,\n\t\t&types.GetWorkflowExecutionRawHistoryV2Request{\n\t\t\tDomain: s.domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t},\n\t\t\tStartEventID:      common.Int64Ptr(10),\n\t\t\tStartEventVersion: common.Int64Ptr(100),\n\t\t\tMaximumPageSize:   1,\n\t\t\tNextPageToken:     nil,\n\t\t})\n\ts.Nil(resp.NextPageToken)\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_SetRequestDefaultValueAndGetTargetVersionHistory_DefinedStartAndEnd() {\n\tinputStartEventID := int64(1)\n\tinputStartVersion := int64(10)\n\tinputEndEventID := int64(100)\n\tinputEndVersion := int64(11)\n\tfirstItem := persistence.NewVersionHistoryItem(inputStartEventID, inputStartVersion)\n\tendItem := persistence.NewVersionHistoryItem(inputEndEventID, inputEndVersion)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{firstItem, endItem})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\trequest := &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tStartEventID:      common.Int64Ptr(inputStartEventID),\n\t\tStartEventVersion: common.Int64Ptr(inputStartVersion),\n\t\tEndEventID:        common.Int64Ptr(inputEndEventID),\n\t\tEndEventVersion:   common.Int64Ptr(inputEndVersion),\n\t\tMaximumPageSize:   10,\n\t\tNextPageToken:     nil,\n\t}\n\n\ttargetVersionHistory, err := s.handler.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\trequest,\n\t\tversionHistories,\n\t)\n\ts.Equal(request.GetStartEventID(), inputStartEventID)\n\ts.Equal(request.GetEndEventID(), inputEndEventID)\n\ts.Equal(targetVersionHistory, versionHistory)\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_SetRequestDefaultValueAndGetTargetVersionHistory_DefinedEndEvent() {\n\tinputStartEventID := int64(1)\n\tinputEndEventID := int64(100)\n\tinputStartVersion := int64(10)\n\tinputEndVersion := int64(11)\n\tfirstItem := persistence.NewVersionHistoryItem(inputStartEventID, inputStartVersion)\n\ttargetItem := persistence.NewVersionHistoryItem(inputEndEventID, inputEndVersion)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{firstItem, targetItem})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\trequest := &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tStartEventID:      nil,\n\t\tStartEventVersion: nil,\n\t\tEndEventID:        common.Int64Ptr(inputEndEventID),\n\t\tEndEventVersion:   common.Int64Ptr(inputEndVersion),\n\t\tMaximumPageSize:   10,\n\t\tNextPageToken:     nil,\n\t}\n\n\ttargetVersionHistory, err := s.handler.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\trequest,\n\t\tversionHistories,\n\t)\n\ts.Equal(request.GetStartEventID(), inputStartEventID-1)\n\ts.Equal(request.GetEndEventID(), inputEndEventID)\n\ts.Equal(targetVersionHistory, versionHistory)\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_SetRequestDefaultValueAndGetTargetVersionHistory_DefinedStartEvent() {\n\tinputStartEventID := int64(1)\n\tinputEndEventID := int64(100)\n\tinputStartVersion := int64(10)\n\tinputEndVersion := int64(11)\n\tfirstItem := persistence.NewVersionHistoryItem(inputStartEventID, inputStartVersion)\n\ttargetItem := persistence.NewVersionHistoryItem(inputEndEventID, inputEndVersion)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{firstItem, targetItem})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\trequest := &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tStartEventID:      common.Int64Ptr(inputStartEventID),\n\t\tStartEventVersion: common.Int64Ptr(inputStartVersion),\n\t\tEndEventID:        nil,\n\t\tEndEventVersion:   nil,\n\t\tMaximumPageSize:   10,\n\t\tNextPageToken:     nil,\n\t}\n\n\ttargetVersionHistory, err := s.handler.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\trequest,\n\t\tversionHistories,\n\t)\n\ts.Equal(request.GetStartEventID(), inputStartEventID)\n\ts.Equal(request.GetEndEventID(), inputEndEventID+1)\n\ts.Equal(targetVersionHistory, versionHistory)\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_SetRequestDefaultValueAndGetTargetVersionHistory_NonCurrentBranch() {\n\tinputStartEventID := int64(1)\n\tinputEndEventID := int64(100)\n\tinputStartVersion := int64(10)\n\tinputEndVersion := int64(101)\n\titem1 := persistence.NewVersionHistoryItem(inputStartEventID, inputStartVersion)\n\titem2 := persistence.NewVersionHistoryItem(inputEndEventID, inputEndVersion)\n\tversionHistory1 := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{item1, item2})\n\titem3 := persistence.NewVersionHistoryItem(int64(10), int64(20))\n\titem4 := persistence.NewVersionHistoryItem(int64(20), int64(51))\n\tversionHistory2 := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{item1, item3, item4})\n\tversionHistories := persistence.NewVersionHistories(versionHistory1)\n\t_, _, err := versionHistories.AddVersionHistory(versionHistory2)\n\ts.NoError(err)\n\trequest := &types.GetWorkflowExecutionRawHistoryV2Request{\n\t\tDomain: s.domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tStartEventID:      common.Int64Ptr(9),\n\t\tStartEventVersion: common.Int64Ptr(20),\n\t\tEndEventID:        common.Int64Ptr(inputEndEventID),\n\t\tEndEventVersion:   common.Int64Ptr(inputEndVersion),\n\t\tMaximumPageSize:   10,\n\t\tNextPageToken:     nil,\n\t}\n\n\ttargetVersionHistory, err := s.handler.setRequestDefaultValueAndGetTargetVersionHistory(\n\t\trequest,\n\t\tversionHistories,\n\t)\n\ts.Equal(request.GetStartEventID(), inputStartEventID)\n\ts.Equal(request.GetEndEventID(), inputEndEventID)\n\ts.Equal(targetVersionHistory, versionHistory1)\n\ts.NoError(err)\n}\n\nfunc (s *adminHandlerSuite) Test_AddSearchAttribute_Validate() {\n\thandler := s.handler\n\thandler.params = &resource.Params{}\n\tctx := context.Background()\n\n\ttype test struct {\n\t\tName     string\n\t\tRequest  *types.AddSearchAttributeRequest\n\t\tExpected error\n\t}\n\t// request validation tests\n\ttestCases1 := []test{\n\t\t{\n\t\t\tName:     \"nil request\",\n\t\t\tRequest:  nil,\n\t\t\tExpected: &types.BadRequestError{Message: \"Request is nil.\"},\n\t\t},\n\t\t{\n\t\t\tName:     \"empty request\",\n\t\t\tRequest:  &types.AddSearchAttributeRequest{},\n\t\t\tExpected: &types.BadRequestError{Message: \"SearchAttributes are not provided\"},\n\t\t},\n\t}\n\tfor _, testCase := range testCases1 {\n\t\ts.Equal(testCase.Expected, handler.AddSearchAttribute(ctx, testCase.Request))\n\t}\n\n\tdynamicConfig := dynamicconfig.NewMockClient(s.controller)\n\thandler.params.DynamicConfig = dynamicConfig\n\t// add advanced visibility store related config\n\thandler.params.ESConfig = &config.ElasticSearchConfig{}\n\tesClient := &esmock.GenericClient{}\n\tdefer func() { esClient.AssertExpectations(s.T()) }()\n\thandler.params.ESClient = esClient\n\thandler.esClient = esClient\n\n\tmockValidAttr := map[string]interface{}{\n\t\t\"testkey\": types.IndexedValueTypeKeyword,\n\t}\n\tdynamicConfig.EXPECT().GetMapValue(dynamicproperties.ValidSearchAttributes, nil).\n\t\tReturn(mockValidAttr, nil).AnyTimes()\n\n\ttestCases2 := []test{\n\t\t{\n\t\t\tName: \"reserved key\",\n\t\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\t\t\"WorkflowID\": 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tExpected: &types.BadRequestError{Message: \"Key [WorkflowID] is reserved by system\"},\n\t\t},\n\t\t{\n\t\t\tName: \"key already whitelisted\",\n\t\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\t\t\"testkey\": 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tExpected: &types.BadRequestError{Message: \"Key [testkey] is already whitelisted as a different type\"},\n\t\t},\n\t}\n\tfor _, testCase := range testCases2 {\n\t\ts.Equal(testCase.Expected, handler.AddSearchAttribute(ctx, testCase.Request))\n\t}\n\n\tdcUpdateTest := test{\n\t\tName: \"dynamic config update failed\",\n\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\t\"testkey2\": -1,\n\t\t\t},\n\t\t},\n\t\tExpected: &types.BadRequestError{Message: \"Unknown value type, IndexedValueType(-1)\"},\n\t}\n\tdynamicConfig.EXPECT().UpdateValue(dynamicproperties.ValidSearchAttributes, map[string]interface{}{\n\t\t\"testkey\":  types.IndexedValueTypeKeyword,\n\t\t\"testkey2\": -1,\n\t}).Return(errors.New(\"error\"))\n\terr := handler.AddSearchAttribute(ctx, dcUpdateTest.Request)\n\ts.Equal(dcUpdateTest.Expected, err)\n\n\t// ES operations tests\n\tdynamicConfig.EXPECT().UpdateValue(gomock.Any(), gomock.Any()).Return(nil).Times(2)\n\n\tconvertFailedTest := test{\n\t\tName: \"unknown value type\",\n\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\t\"testkey3\": -1,\n\t\t\t},\n\t\t},\n\t\tExpected: &types.BadRequestError{Message: \"Unknown value type, IndexedValueType(-1)\"},\n\t}\n\ts.Equal(convertFailedTest.Expected, handler.AddSearchAttribute(ctx, convertFailedTest.Request))\n\n\tesClient.On(\"PutMapping\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).\n\t\tReturn(errors.New(\"error\"))\n\tesClient.On(\"IsNotFoundError\", mock.Anything).Return(false)\n\tesErrorTest := test{\n\t\tName: \"es error\",\n\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\t\"testkey4\": 1,\n\t\t\t},\n\t\t},\n\t\tExpected: &types.InternalServiceError{Message: \"Failed to update ES mapping, err: error\"},\n\t}\n\ts.Equal(esErrorTest.Expected, handler.AddSearchAttribute(ctx, esErrorTest.Request))\n}\n\nfunc (s *adminHandlerSuite) Test_AddSearchAttribute_Permission() {\n\tctx := context.Background()\n\thandler := s.handler\n\thandler.config = &frontendcfg.Config{\n\t\tEnableAdminProtection: dynamicproperties.GetBoolPropertyFn(true),\n\t\tAdminOperationToken:   dynamicproperties.GetStringPropertyFn(dynamicproperties.AdminOperationToken.DefaultString()),\n\t}\n\n\ttype test struct {\n\t\tName     string\n\t\tRequest  *types.AddSearchAttributeRequest\n\t\tExpected error\n\t}\n\ttestCases := []test{\n\t\t{\n\t\t\tName: \"unknown token\",\n\t\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\t\tSecurityToken: \"unknown\",\n\t\t\t},\n\t\t\tExpected: validate.ErrNoPermission,\n\t\t},\n\t\t{\n\t\t\tName: \"correct token\",\n\t\t\tRequest: &types.AddSearchAttributeRequest{\n\t\t\t\tSecurityToken: dynamicproperties.AdminOperationToken.DefaultString(),\n\t\t\t},\n\t\t\tExpected: &types.BadRequestError{Message: \"SearchAttributes are not provided\"},\n\t\t},\n\t}\n\tfor _, testCase := range testCases {\n\t\ts.Equal(testCase.Expected, handler.AddSearchAttribute(ctx, testCase.Request))\n\t}\n}\n\nfunc (s *adminHandlerSuite) Test_ConfigStore_NilRequest() {\n\tctx := context.Background()\n\thandler := s.handler\n\n\t_, err := handler.GetDynamicConfig(ctx, nil)\n\ts.Error(err)\n\n\terr = handler.UpdateDynamicConfig(ctx, nil)\n\ts.Error(err)\n\n\terr = handler.RestoreDynamicConfig(ctx, nil)\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_DescribeShardDistribution() {\n\ts.mockResource.MembershipResolver.EXPECT().Lookup(service.History, string(rune(0))).\n\t\tReturn(membership.NewHostInfo(\"127.0.0.1:1234\"), nil)\n\n\tres, err := s.handler.DescribeShardDistribution(\n\t\tcontext.Background(),\n\t\t&types.DescribeShardDistributionRequest{PageSize: 10},\n\t)\n\ts.Require().NoError(err)\n\ts.Equal(\n\t\t&types.DescribeShardDistributionResponse{\n\t\t\tNumberOfShards: 1,\n\t\t\tShards:         map[int32]string{0: \"127.0.0.1:1234\"},\n\t\t},\n\t\tres,\n\t)\n}\n\nfunc (s *adminHandlerSuite) Test_ConfigStore_InvalidKey() {\n\tctx := context.Background()\n\thandler := s.handler\n\n\t_, err := handler.GetDynamicConfig(ctx, &types.GetDynamicConfigRequest{\n\t\tConfigName: \"invalid key\",\n\t\tFilters:    nil,\n\t})\n\ts.Error(err)\n\n\terr = handler.UpdateDynamicConfig(ctx, &types.UpdateDynamicConfigRequest{\n\t\tConfigName:   \"invalid key\",\n\t\tConfigValues: nil,\n\t})\n\ts.Error(err)\n\n\terr = handler.RestoreDynamicConfig(ctx, &types.RestoreDynamicConfigRequest{\n\t\tConfigName: \"invalid key\",\n\t\tFilters:    nil,\n\t})\n\ts.Error(err)\n}\n\nfunc (s *adminHandlerSuite) Test_GetDynamicConfig_NoFilter() {\n\tctx := context.Background()\n\thandler := s.handler\n\tdynamicConfig := dynamicconfig.NewMockClient(s.controller)\n\thandler.params.DynamicConfig = dynamicConfig\n\n\tdynamicConfig.EXPECT().\n\t\tGetValue(dynamicproperties.TestGetBoolPropertyKey).\n\t\tReturn(true, nil).AnyTimes()\n\n\tresp, err := handler.GetDynamicConfig(ctx, &types.GetDynamicConfigRequest{\n\t\tConfigName: dynamicproperties.TestGetBoolPropertyKey.String(),\n\t\tFilters:    nil,\n\t})\n\ts.NoError(err)\n\n\tencTrue, err := json.Marshal(true)\n\ts.NoError(err)\n\ts.Equal(resp.Value.Data, encTrue)\n}\n\nfunc (s *adminHandlerSuite) Test_GetDynamicConfig_FilterMatch() {\n\tctx := context.Background()\n\thandler := s.handler\n\tdynamicConfig := dynamicconfig.NewMockClient(s.controller)\n\thandler.params.DynamicConfig = dynamicConfig\n\n\tdynamicConfig.EXPECT().\n\t\tGetValueWithFilters(dynamicproperties.TestGetBoolPropertyKey, map[dynamicproperties.Filter]interface{}{\n\t\t\tdynamicproperties.DomainName: \"samples_domain\",\n\t\t}).\n\t\tReturn(true, nil).AnyTimes()\n\n\tencDomainName, err := json.Marshal(\"samples_domain\")\n\ts.NoError(err)\n\n\tresp, err := handler.GetDynamicConfig(ctx, &types.GetDynamicConfigRequest{\n\t\tConfigName: dynamicproperties.TestGetBoolPropertyKey.String(),\n\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t{\n\t\t\t\tName: dynamicproperties.DomainName.String(),\n\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\tData:         encDomainName,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\ts.NoError(err)\n\n\tencTrue, err := json.Marshal(true)\n\ts.NoError(err)\n\ts.Equal(resp.Value.Data, encTrue)\n}\n\nfunc Test_GetGlobalIsolationGroups(t *testing.T) {\n\n\tvalidResponse := types.GetGlobalIsolationGroupsResponse{\n\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-1\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-2\",\n\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t},\n\t\t\t\"zone-3\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-3\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tighandlerAffordance func(mock *isolationgroupapi.MockHandler)\n\t\texpectOut           *types.GetGlobalIsolationGroupsResponse\n\t\texpectedErr         error\n\t}{\n\t\t\"happy-path - no errors and payload is decoded and returned\": {\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetGlobalState(gomock.Any()).Return(&validResponse, nil)\n\t\t\t},\n\t\t\texpectOut: &validResponse,\n\t\t},\n\t\t\"an error returned\": {\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetGlobalState(gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: assert.AnError.Error()},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tigMock := isolationgroupapi.NewMockHandler(goMock)\n\t\t\ttd.ighandlerAffordance(igMock)\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tisolationGroups: igMock,\n\t\t\t}\n\n\t\t\tres, err := handler.GetGlobalIsolationGroups(context.Background(), &types.GetGlobalIsolationGroupsRequest{})\n\n\t\t\tassert.Equal(t, td.expectOut, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_UpdateGlobalIsolationGroups(t *testing.T) {\n\n\tvalidConfig := types.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": {\n\t\t\t\tName:  \"zone-1\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t\t\"zone-2\": {\n\t\t\t\tName:  \"zone-2\",\n\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t},\n\t\t\t\"zone-3\": {\n\t\t\t\tName:  \"zone-3\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tighandlerAffordance func(mock *isolationgroupapi.MockHandler)\n\t\tinput               *types.UpdateGlobalIsolationGroupsRequest\n\t\texpectOut           *types.UpdateGlobalIsolationGroupsResponse\n\t\texpectedErr         error\n\t}{\n\t\t\"happy-path - update to the database\": {\n\t\t\tinput: &validConfig,\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateGlobalState(gomock.Any(), validConfig).Return(nil)\n\t\t\t},\n\t\t\texpectOut: &types.UpdateGlobalIsolationGroupsResponse{},\n\t\t},\n\t\t\"happy-path - an error is returned\": {\n\t\t\tinput: &validConfig,\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateGlobalState(gomock.Any(), validConfig).Return(assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: assert.AnError.Error()},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tigMock := isolationgroupapi.NewMockHandler(goMock)\n\t\t\ttd.ighandlerAffordance(igMock)\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tisolationGroups: igMock,\n\t\t\t}\n\n\t\t\tres, err := handler.UpdateGlobalIsolationGroups(context.Background(), td.input)\n\n\t\t\tassert.Equal(t, td.expectOut, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_GetDomainIsolationGroups(t *testing.T) {\n\n\tvalidResponse := types.GetDomainIsolationGroupsResponse{\n\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-1\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t\t\"zone-2\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-2\",\n\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t},\n\t\t\t\"zone-3\": types.IsolationGroupPartition{\n\t\t\t\tName:  \"zone-3\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tighandlerAffordance func(mock *isolationgroupapi.MockHandler)\n\t\texpectOut           *types.GetDomainIsolationGroupsResponse\n\t\texpectedErr         error\n\t}{\n\t\t\"happy-path - no errors and payload is decoded and returned\": {\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetDomainState(gomock.Any(), types.GetDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t}).Return(&validResponse, nil)\n\t\t\t},\n\t\t\texpectOut: &validResponse,\n\t\t},\n\t\t\"an error returned\": {\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetDomainState(gomock.Any(), types.GetDomainIsolationGroupsRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: assert.AnError.Error()},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tigMock := isolationgroupapi.NewMockHandler(goMock)\n\t\t\ttd.ighandlerAffordance(igMock)\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tisolationGroups: igMock,\n\t\t\t}\n\n\t\t\tres, err := handler.GetDomainIsolationGroups(context.Background(), &types.GetDomainIsolationGroupsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t})\n\n\t\t\tassert.Equal(t, td.expectOut, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_UpdateDomainIsolationGroups(t *testing.T) {\n\n\tvalidConfig := types.UpdateDomainIsolationGroupsRequest{\n\t\tDomain: \"domain\",\n\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\"zone-1\": {\n\t\t\t\tName:  \"zone-1\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t\t\"zone-2\": {\n\t\t\t\tName:  \"zone-2\",\n\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t},\n\t\t\t\"zone-3\": {\n\t\t\t\tName:  \"zone-3\",\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := map[string]struct {\n\t\tighandlerAffordance func(mock *isolationgroupapi.MockHandler)\n\t\tinput               *types.UpdateDomainIsolationGroupsRequest\n\t\texpectOut           *types.UpdateDomainIsolationGroupsResponse\n\t\texpectedErr         error\n\t}{\n\t\t\"happy-path - update to the database\": {\n\t\t\tinput: &validConfig,\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateDomainState(gomock.Any(), validConfig).Return(nil)\n\t\t\t},\n\t\t\texpectOut: &types.UpdateDomainIsolationGroupsResponse{},\n\t\t},\n\t\t\"happy-path - an error is returned\": {\n\t\t\tinput: &validConfig,\n\t\t\tighandlerAffordance: func(mock *isolationgroupapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateDomainState(gomock.Any(), validConfig).Return(assert.AnError)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: assert.AnError.Error()},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tigMock := isolationgroupapi.NewMockHandler(goMock)\n\t\t\ttd.ighandlerAffordance(igMock)\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tisolationGroups: igMock,\n\t\t\t}\n\n\t\t\tres, err := handler.UpdateDomainIsolationGroups(context.Background(), td.input)\n\n\t\t\tassert.Equal(t, td.expectOut, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_GetDomainAsyncWorkflowConfiguraton(t *testing.T) {\n\ttests := map[string]struct {\n\t\tqueueCfgHandlerMockFn func(mock *queueconfigapi.MockHandler)\n\t\tinput                 *types.GetDomainAsyncWorkflowConfiguratonRequest\n\t\twantResp              *types.GetDomainAsyncWorkflowConfiguratonResponse\n\t\twantErr               error\n\t}{\n\t\t\"success\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonRequest{Domain: \"test-domain\"},\n\t\t\tqueueCfgHandlerMockFn: func(mock *queueconfigapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetConfiguraton(gomock.Any(), gomock.Any()).Return(&types.GetDomainAsyncWorkflowConfiguratonResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.GetDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"queue config handler failed\": {\n\t\t\tinput: &types.GetDomainAsyncWorkflowConfiguratonRequest{Domain: \"test-domain\"},\n\t\t\tqueueCfgHandlerMockFn: func(mock *queueconfigapi.MockHandler) {\n\t\t\t\tmock.EXPECT().GetConfiguraton(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: &types.InternalServiceError{Message: \"failed\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tqueueCfgHandlerMock := queueconfigapi.NewMockHandler(goMock)\n\t\t\tif td.queueCfgHandlerMockFn != nil {\n\t\t\t\ttd.queueCfgHandlerMockFn(queueCfgHandlerMock)\n\t\t\t}\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tasyncWFQueueConfigs: queueCfgHandlerMock,\n\t\t\t}\n\n\t\t\tres, err := handler.GetDomainAsyncWorkflowConfiguraton(context.Background(), td.input)\n\n\t\t\tassert.Equal(t, td.wantResp, res)\n\t\t\tassert.Equal(t, td.wantErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_UpdateDomainAsyncWorkflowConfiguraton(t *testing.T) {\n\ttests := map[string]struct {\n\t\tqueueCfgHandlerMockFn func(mock *queueconfigapi.MockHandler)\n\t\tinput                 *types.UpdateDomainAsyncWorkflowConfiguratonRequest\n\t\twantResp              *types.UpdateDomainAsyncWorkflowConfiguratonResponse\n\t\twantErr               error\n\t}{\n\t\t\"success\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{Domain: \"test-domain\"},\n\t\t\tqueueCfgHandlerMockFn: func(mock *queueconfigapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateConfiguration(gomock.Any(), gomock.Any()).Return(&types.UpdateDomainAsyncWorkflowConfiguratonResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twantResp: &types.UpdateDomainAsyncWorkflowConfiguratonResponse{},\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"queue config handler failed\": {\n\t\t\tinput: &types.UpdateDomainAsyncWorkflowConfiguratonRequest{Domain: \"test-domain\"},\n\t\t\tqueueCfgHandlerMockFn: func(mock *queueconfigapi.MockHandler) {\n\t\t\t\tmock.EXPECT().UpdateConfiguration(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"failed\")).Times(1)\n\t\t\t},\n\t\t\twantErr: &types.InternalServiceError{Message: \"failed\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tqueueCfgHandlerMock := queueconfigapi.NewMockHandler(goMock)\n\t\t\tif td.queueCfgHandlerMockFn != nil {\n\t\t\t\ttd.queueCfgHandlerMockFn(queueCfgHandlerMock)\n\t\t\t}\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tasyncWFQueueConfigs: queueCfgHandlerMock,\n\t\t\t}\n\n\t\t\tres, err := handler.UpdateDomainAsyncWorkflowConfiguraton(context.Background(), td.input)\n\n\t\t\tassert.Equal(t, td.wantResp, res)\n\t\t\tassert.Equal(t, td.wantErr, err)\n\t\t})\n\t}\n}\n\nfunc Test_RemoveTask(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.RemoveTaskRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.RemoveTaskRequest{\n\t\t\t\tShardID: 1,\n\t\t\t\tType:    common.Int32Ptr(2),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RemoveTask(gomock.Any(), &types.RemoveTaskRequest{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tType:    common.Int32Ptr(2),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"request failed\": {\n\t\t\tinput: &types.RemoveTaskRequest{\n\t\t\t\tShardID: 1,\n\t\t\t\tType:    common.Int32Ptr(2),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RemoveTask(gomock.Any(), &types.RemoveTaskRequest{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t\tType:    common.Int32Ptr(2),\n\t\t\t\t}).Return(assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.RemoveTask(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_CloseShard(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.CloseShardRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.CloseShardRequest{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().CloseShard(gomock.Any(), &types.CloseShardRequest{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"request failed\": {\n\t\t\tinput: &types.CloseShardRequest{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().CloseShard(gomock.Any(), &types.CloseShardRequest{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t}).Return(assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.CloseShard(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ResetQueue(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.ResetQueueRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing cluster name request\": {\n\t\t\tinput: &types.ResetQueueRequest{\n\t\t\t\tShardID: 1,\n\t\t\t\tType:    common.Int32Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.ResetQueueRequest{\n\t\t\t\tShardID:     1,\n\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().ResetQueue(gomock.Any(), &types.ResetQueueRequest{\n\t\t\t\t\tShardID:     1,\n\t\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t\t}).Return(nil).Times(1)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"normal request return error\": {\n\t\t\tinput: &types.ResetQueueRequest{\n\t\t\t\tShardID:     1,\n\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().ResetQueue(gomock.Any(), &types.ResetQueueRequest{\n\t\t\t\t\tShardID:     1,\n\t\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t\t}).Return(assert.AnError).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.ResetQueue(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_DescribeQueue(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.DescribeQueueRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing cluster name request\": {\n\t\t\tinput: &types.DescribeQueueRequest{\n\t\t\t\tShardID: 1,\n\t\t\t\tType:    common.Int32Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.DescribeQueueRequest{\n\t\t\t\tShardID:     1,\n\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().DescribeQueue(gomock.Any(), &types.DescribeQueueRequest{\n\t\t\t\t\tShardID:     1,\n\t\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t\t\tType:        common.Int32Ptr(1),\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.DescribeQueue(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_DescribeHistoryHost(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.DescribeHistoryHostRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing host address request\": {\n\t\t\tinput: &types.DescribeHistoryHostRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing execution\": {\n\t\t\tinput: &types.DescribeHistoryHostRequest{\n\t\t\t\tExecutionForHost: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.DescribeHistoryHostRequest{\n\t\t\t\tExecutionForHost: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().DescribeHistoryHost(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.DescribeHistoryHost(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_DescribeCluster(t *testing.T) {\n\tgoMock := gomock.NewController(t)\n\tmockResource := resource.NewTest(t, goMock, metrics.Frontend)\n\tmockResource.VisibilityMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.HistoryMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.MembershipResolver.EXPECT().WhoAmI().Return(membership.NewHostInfo(\"1.0.0.1\"), nil).AnyTimes()\n\tmockResource.MembershipResolver.EXPECT().Members(gomock.Any()).Return([]membership.HostInfo{\n\t\tmembership.NewHostInfo(\"1.0.0.1\"),\n\t}, nil).AnyTimes()\n\thandler := adminHandlerImpl{\n\t\tparams: &resource.Params{\n\t\t\tESConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{},\n\t\t\t},\n\t\t},\n\t\tResource: mockResource,\n\t}\n\t_, err := handler.DescribeCluster(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc Test_DescribeCluster_HostError(t *testing.T) {\n\tgoMock := gomock.NewController(t)\n\tmockResource := resource.NewTest(t, goMock, metrics.Frontend)\n\tmockResource.VisibilityMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.HistoryMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.MembershipResolver.EXPECT().WhoAmI().Return(membership.NewHostInfo(\"1.0.0.1\"), assert.AnError).AnyTimes()\n\thandler := adminHandlerImpl{\n\t\tparams: &resource.Params{\n\t\t\tESConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{},\n\t\t\t},\n\t\t},\n\t\tResource: mockResource,\n\t}\n\t_, err := handler.DescribeCluster(context.Background())\n\tassert.Error(t, err)\n}\n\nfunc Test_DescribeCluster_MemberError(t *testing.T) {\n\tgoMock := gomock.NewController(t)\n\tmockResource := resource.NewTest(t, goMock, metrics.Frontend)\n\tmockResource.VisibilityMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.HistoryMgr.On(\"GetName\").Return(\"test\").Once()\n\tmockResource.MembershipResolver.EXPECT().WhoAmI().Return(membership.NewHostInfo(\"1.0.0.1\"), nil).AnyTimes()\n\tmockResource.MembershipResolver.EXPECT().Members(gomock.Any()).Return([]membership.HostInfo{\n\t\tmembership.NewHostInfo(\"1.0.0.1\"),\n\t}, assert.AnError).AnyTimes()\n\thandler := adminHandlerImpl{\n\t\tparams: &resource.Params{\n\t\t\tESConfig: &config.ElasticSearchConfig{\n\t\t\t\tIndices: map[string]string{},\n\t\t\t},\n\t\t},\n\t\tResource: mockResource,\n\t}\n\t_, err := handler.DescribeCluster(context.Background())\n\tassert.Error(t, err)\n}\n\nfunc TestGetReplicationMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.GetReplicationMessagesRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing cluster name\": {\n\t\t\tinput: &types.GetReplicationMessagesRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.GetReplicationMessagesRequest{\n\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"return error\": {\n\t\t\tinput: &types.GetReplicationMessagesRequest{\n\t\t\t\tClusterName: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.GetReplicationMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_GetDomainReplicationMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput          *types.GetDomainReplicationMessagesRequest\n\t\thcHandlerFunc  func(mock *history.MockClient)\n\t\tdrqHandlerFunc func(mock *domain.MockReplicationQueue)\n\t\twantErr        bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:          nil,\n\t\t\thcHandlerFunc:  func(mock *history.MockClient) {},\n\t\t\tdrqHandlerFunc: func(mock *domain.MockReplicationQueue) {},\n\t\t\twantErr:        true,\n\t\t},\n\t\t\"with default last message ID\": {\n\t\t\tinput: &types.GetDomainReplicationMessagesRequest{\n\t\t\t\tLastRetrievedMessageID: common.Int64Ptr(-1), // default value\n\t\t\t\tClusterName:            \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tdrqHandlerFunc: func(mock *domain.MockReplicationQueue) {\n\t\t\t\tmock.EXPECT().GetAckLevels(context.Background()).Return(map[string]int64{\n\t\t\t\t\t\"test-cluster\": 1,\n\t\t\t\t}, nil)\n\t\t\t\tmock.EXPECT().GetReplicationMessages(context.Background(), int64(1), getDomainReplicationMessageBatchSize).Return(nil, int64(2), nil)\n\t\t\t\tmock.EXPECT().UpdateAckLevel(context.Background(), int64(-1), \"test-cluster\").Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"with random last message ID but update ack level error\": {\n\t\t\tinput: &types.GetDomainReplicationMessagesRequest{\n\t\t\t\tLastRetrievedMessageID: common.Int64Ptr(1),\n\t\t\t\tLastProcessedMessageID: common.Int64Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tdrqHandlerFunc: func(mock *domain.MockReplicationQueue) {\n\t\t\t\tmock.EXPECT().GetReplicationMessages(context.Background(), int64(1), getDomainReplicationMessageBatchSize).Return(nil, int64(2), nil)\n\t\t\t\tmock.EXPECT().UpdateAckLevel(context.Background(), int64(1), \"\").Return(assert.AnError)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"get replication messages error\": {\n\t\t\tinput: &types.GetDomainReplicationMessagesRequest{\n\t\t\t\tLastRetrievedMessageID: common.Int64Ptr(1),\n\t\t\t\tLastProcessedMessageID: common.Int64Ptr(1),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tdrqHandlerFunc: func(mock *domain.MockReplicationQueue) {\n\t\t\t\tmock.EXPECT().GetReplicationMessages(context.Background(), int64(1), getDomainReplicationMessageBatchSize).Return(nil, int64(2), assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tdrqMock := domain.NewMockReplicationQueue(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\tif td.drqHandlerFunc != nil {\n\t\t\t\ttd.drqHandlerFunc(drqMock)\n\t\t\t}\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:                 testlogger.New(t),\n\t\t\t\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t\t\t\t\tDomainReplicationQueue: drqMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tresp, err := handler.GetDomainReplicationMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int64(2), resp.Messages.LastRetrievedMessageID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_GetDLQReplicationMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.GetDLQReplicationMessagesRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing task info\": {\n\t\t\tinput: &types.GetDLQReplicationMessagesRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.GetDLQReplicationMessagesRequest{\n\t\t\t\tTaskInfos: []*types.ReplicationTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainID:     \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID:   \"test-workflow-id\",\n\t\t\t\t\t\tRunID:        uuid.New(),\n\t\t\t\t\t\tTaskID:       int64(1),\n\t\t\t\t\t\tTaskType:     1,\n\t\t\t\t\t\tVersion:      0,\n\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"get dql message error\": {\n\t\t\tinput: &types.GetDLQReplicationMessagesRequest{\n\t\t\t\tTaskInfos: []*types.ReplicationTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.GetDLQReplicationMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ReapplyEvents(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.ReapplyEventsRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\tdcHandlerFunc func(mock *cache.MockDomainCache)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty domain name\": {\n\t\t\tinput:   &types.ReapplyEventsRequest{},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty workflow execution\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName:        \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty events\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"get domain error\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t\tEvents: &types.DataBlob{},\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\tData:         []byte(\"test-event-data\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().ReapplyEvents(context.Background(), &types.HistoryReapplyEventsRequest{\n\t\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\t\tRequest: &types.ReapplyEventsRequest{\n\t\t\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\tData:         []byte(\"test-event-data\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-domain-id\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"reapply events error\": {\n\t\t\tinput: &types.ReapplyEventsRequest{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t\tEvents: &types.DataBlob{},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-domain-id\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tdcMock := cache.NewMockDomainCache(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\tif td.dcHandlerFunc != nil {\n\t\t\t\ttd.dcHandlerFunc(dcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t\tDomainCache:   dcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.ReapplyEvents(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ReadDLQMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput            *types.ReadDLQMessagesRequest\n\t\thcHandlerFunc    func(mock *history.MockClient)\n\t\tretryHandelrFunc func(mock *backoff.ThrottleRetry)\n\t\twantErr          bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing type\": {\n\t\t\tinput:   &types.ReadDLQMessagesRequest{},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"type DLQTypeReplication\": {\n\t\t\tinput: &types.ReadDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeReplication.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().ReadDLQMessages(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"type DLQTypeDomain\": {\n\t\t\tinput: &types.ReadDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeDomain.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tretryHandelrFunc: func(mock *backoff.ThrottleRetry) {\n\t\t\t},\n\t\t},\n\t\t\"default type\": {\n\t\t\tinput: &types.ReadDLQMessagesRequest{\n\t\t\t\tType: types.DLQType(22).Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tif td.retryHandelrFunc != nil {\n\t\t\t\tpolicy := backoff.NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumAttempts(5)\n\t\t\t\thandler.throttleRetry = backoff.NewThrottleRetry(\n\t\t\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\t\t\tbackoff.WithRetryableError(func(_ error) bool { return true }),\n\t\t\t\t)\n\t\t\t\tmockDlqHandler := domain.NewMockDLQMessageHandler(goMock)\n\t\t\t\tmockDlqHandler.EXPECT().Read(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, nil)\n\t\t\t\thandler.domainDLQHandler = mockDlqHandler\n\t\t\t}\n\n\t\t\t_, err := handler.ReadDLQMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_PurgeDLQMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput            *types.PurgeDLQMessagesRequest\n\t\thcHandlerFunc    func(mock *history.MockClient)\n\t\tretryHandelrFunc func(mock *backoff.ThrottleRetry)\n\t\twantErr          bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing type\": {\n\t\t\tinput:   &types.PurgeDLQMessagesRequest{},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"type DLQTypeReplication\": {\n\t\t\tinput: &types.PurgeDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeReplication.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().PurgeDLQMessages(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"type DLQTypeDomain\": {\n\t\t\tinput: &types.PurgeDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeDomain.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tretryHandelrFunc: func(mock *backoff.ThrottleRetry) {\n\t\t\t},\n\t\t},\n\t\t\"default type\": {\n\t\t\tinput: &types.PurgeDLQMessagesRequest{\n\t\t\t\tType: types.DLQType(22).Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tif td.retryHandelrFunc != nil {\n\t\t\t\tpolicy := backoff.NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumAttempts(5)\n\t\t\t\thandler.throttleRetry = backoff.NewThrottleRetry(\n\t\t\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\t\t\tbackoff.WithRetryableError(func(_ error) bool { return true }),\n\t\t\t\t)\n\t\t\t\tmockDlqHandler := domain.NewMockDLQMessageHandler(goMock)\n\t\t\t\tmockDlqHandler.EXPECT().Purge(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\thandler.domainDLQHandler = mockDlqHandler\n\t\t\t}\n\n\t\t\terr := handler.PurgeDLQMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_CountDQLMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput          *types.CountDLQMessagesRequest\n\t\thcHandlerFunc  func(mock *history.MockClient)\n\t\tdlqHandlerFunc func(mock *domain.MockDLQMessageHandler)\n\t\twantErr        bool\n\t}{\n\t\t\"normal request\": {\n\t\t\tinput: &types.CountDLQMessagesRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any()).Return(&types.HistoryCountDLQMessagesResponse{\n\t\t\t\t\tEntries: map[types.HistoryDLQCountKey]int64{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tShardID:       1,\n\t\t\t\t\t\t\tSourceCluster: \"test-cluster\",\n\t\t\t\t\t\t}: 1,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tdlqHandlerFunc: func(mock *domain.MockDLQMessageHandler) {\n\t\t\t\tmock.EXPECT().Count(gomock.Any(), gomock.Any()).Return(int64(1), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"domain dlq handler error\": {\n\t\t\tinput: &types.CountDLQMessagesRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tdlqHandlerFunc: func(mock *domain.MockDLQMessageHandler) {\n\t\t\t\tmock.EXPECT().Count(gomock.Any(), gomock.Any()).Return(int64(1), assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"history client error\": {\n\t\t\tinput: &types.CountDLQMessagesRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\tdlqHandlerFunc: func(mock *domain.MockDLQMessageHandler) {\n\t\t\t\tmock.EXPECT().Count(gomock.Any(), gomock.Any()).Return(int64(1), nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tdlqMock := domain.NewMockDLQMessageHandler(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\tif td.dlqHandlerFunc != nil {\n\t\t\t\ttd.dlqHandlerFunc(dlqMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t\tdomainDLQHandler: dlqMock,\n\t\t\t}\n\n\t\t\t_, err := handler.CountDLQMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_MergeDLQMessages(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput            *types.MergeDLQMessagesRequest\n\t\thcHandlerFunc    func(mock *history.MockClient)\n\t\tretryHandelrFunc func(mock *backoff.ThrottleRetry)\n\t\twantErr          bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing type\": {\n\t\t\tinput:   &types.MergeDLQMessagesRequest{},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"type DLQTypeReplication\": {\n\t\t\tinput: &types.MergeDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeReplication.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().MergeDLQMessages(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"type DLQTypeDomain\": {\n\t\t\tinput: &types.MergeDLQMessagesRequest{\n\t\t\t\tType: types.DLQTypeDomain.Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\tretryHandelrFunc: func(mock *backoff.ThrottleRetry) {\n\t\t\t},\n\t\t},\n\t\t\"default type\": {\n\t\t\tinput: &types.MergeDLQMessagesRequest{\n\t\t\t\tType: types.DLQType(22).Ptr(),\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tif td.retryHandelrFunc != nil {\n\t\t\t\tpolicy := backoff.NewExponentialRetryPolicy(1 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumInterval(5 * time.Millisecond)\n\t\t\t\tpolicy.SetMaximumAttempts(5)\n\t\t\t\thandler.throttleRetry = backoff.NewThrottleRetry(\n\t\t\t\t\tbackoff.WithRetryPolicy(policy),\n\t\t\t\t\tbackoff.WithRetryableError(func(_ error) bool { return true }),\n\t\t\t\t)\n\t\t\t\tmockDlqHandler := domain.NewMockDLQMessageHandler(goMock)\n\t\t\t\tmockDlqHandler.EXPECT().Merge(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t\thandler.domainDLQHandler = mockDlqHandler\n\t\t\t}\n\n\t\t\t_, err := handler.MergeDLQMessages(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_RefreshWorkflowTasks(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.RefreshWorkflowTasksRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\tdcHandlerFunc func(mock *cache.MockDomainCache)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty workflow execution\": {\n\t\t\tinput: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"check execution error\": {\n\t\t\tinput: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-domain-id\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"refresh workflow tasks error\": {\n\t\t\tinput: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *cache.MockDomainCache) {\n\t\t\t\tmock.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-domain-id\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(\"\", func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\thcMock := history.NewMockClient(ctrl)\n\t\t\tdcMock := cache.NewMockDomainCache(ctrl)\n\t\t\tif tt.hcHandlerFunc != nil {\n\t\t\t\ttt.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\tif tt.dcHandlerFunc != nil {\n\t\t\t\ttt.dcHandlerFunc(dcMock)\n\t\t\t}\n\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t\tDomainCache:   dcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.RefreshWorkflowTasks(context.Background(), tt.input)\n\t\t\tif tt.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ResendReplicationTasks(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.ResendReplicationTasksRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.ResendReplicationTasksRequest{\n\t\t\t\tRemoteCluster: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.ResendReplicationTasks(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_GetCrossClusterTasks(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.GetCrossClusterTasksRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput: nil,\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"missing cluster name\": {\n\t\t\tinput: &types.GetCrossClusterTasksRequest{},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.GetCrossClusterTasksRequest{\n\t\t\t\tTargetCluster: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetCrossClusterTasks(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"return error\": {\n\t\t\tinput: &types.GetCrossClusterTasksRequest{\n\t\t\t\tTargetCluster: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().GetCrossClusterTasks(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.GetCrossClusterTasks(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_RespondCrossClusterTasksCompleted(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.RespondCrossClusterTasksCompletedRequest\n\t\thcHandlerFunc func(mock *history.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"empty target cluster\": {\n\t\t\tinput:   &types.RespondCrossClusterTasksCompletedRequest{},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"normal request\": {\n\t\t\tinput: &types.RespondCrossClusterTasksCompletedRequest{\n\t\t\t\tTargetCluster: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RespondCrossClusterTasksCompleted(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"respond error\": {\n\t\t\tinput: &types.RespondCrossClusterTasksCompletedRequest{\n\t\t\t\tTargetCluster: \"test-cluster\",\n\t\t\t},\n\t\t\thcHandlerFunc: func(mock *history.MockClient) {\n\t\t\t\tmock.EXPECT().RespondCrossClusterTasksCompleted(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tif td.hcHandlerFunc != nil {\n\t\t\t\ttd.hcHandlerFunc(hcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.RespondCrossClusterTasksCompleted(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_SerializeRawHistoryToken(t *testing.T) {\n\tinput := getWorkflowRawHistoryV2Token{\n\t\tDomainName: \"test-domain\",\n\t}\n\tmarshaledToken, _ := json.Marshal(input)\n\ttests := map[string]struct {\n\t\tinput              *getWorkflowRawHistoryV2Token\n\t\twantResp           []byte\n\t\twantSerializeErr   bool\n\t\twantDeserializeErr bool\n\t}{\n\t\t\"normal request\": {\n\t\t\tinput: &getWorkflowRawHistoryV2Token{\n\t\t\t\tDomainName: \"test-domain\",\n\t\t\t},\n\t\t\twantResp:           marshaledToken,\n\t\t\twantSerializeErr:   false,\n\t\t\twantDeserializeErr: false,\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:              nil,\n\t\t\twantResp:           nil,\n\t\t\twantSerializeErr:   false,\n\t\t\twantDeserializeErr: true,\n\t\t},\n\t}\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tresp, err := serializeRawHistoryToken(td.input)\n\t\t\tif td.wantSerializeErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, td.wantResp, resp)\n\t\t\t}\n\t\t\t// deserialize the serialized token\n\t\t\tres, err := deserializeRawHistoryToken(resp)\n\t\t\tif td.wantDeserializeErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, td.input, res)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ListDynamicConfig(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.ListDynamicConfigRequest\n\t\tdcHandlerFunc func(mock *dynamicconfig.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\thcMock := history.NewMockClient(goMock)\n\t\t\tdcMock := dynamicconfig.NewMockClient(goMock)\n\t\t\tif td.dcHandlerFunc != nil {\n\t\t\t\ttd.dcHandlerFunc(dcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t\tHistoryClient: hcMock,\n\t\t\t\t},\n\t\t\t\tparams: &resource.Params{\n\t\t\t\t\tDynamicConfig: dcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.ListDynamicConfig(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateConfigForAdvanceVisibility_Error(t *testing.T) {\n\thandler := adminHandlerImpl{\n\t\tparams: &resource.Params{},\n\t}\n\terr := handler.validateConfigForAdvanceVisibility()\n\tassert.Error(t, err, \"ES related config not found\")\n}\n\nfunc TestRestoreDynamicConfig(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.RestoreDynamicConfigRequest\n\t\tdcHandlerFunc func(mock *dynamicconfig.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"invalid config name\": {\n\t\t\tinput: &types.RestoreDynamicConfigRequest{\n\t\t\t\tConfigName: \"invalid\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"valid config name\": {\n\t\t\tinput: &types.RestoreDynamicConfigRequest{\n\t\t\t\tConfigName: \"testGetIntPropertyKey\",\n\t\t\t\tFilters:    nil,\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *dynamicconfig.MockClient) {\n\t\t\t\tmock.EXPECT().RestoreValue(gomock.Any(), nil).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t\"valid config with invalid filters\": {\n\t\t\tinput: &types.RestoreDynamicConfigRequest{\n\t\t\t\tConfigName: \"testGetIntPropertyKey\",\n\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t&types.DynamicConfigFilter{\n\t\t\t\t\t\tName: \"test-filter\",\n\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\tData: []byte(\"test-value\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tdcMock := dynamicconfig.NewMockClient(goMock)\n\t\t\tif td.dcHandlerFunc != nil {\n\t\t\t\ttd.dcHandlerFunc(dcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tparams: &resource.Params{\n\t\t\t\t\tDynamicConfig: dcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := handler.RestoreDynamicConfig(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListDynamicConfig(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput         *types.ListDynamicConfigRequest\n\t\tdcHandlerFunc func(mock *dynamicconfig.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t\"nil request\": {\n\t\t\tinput:   nil,\n\t\t\twantErr: true,\n\t\t},\n\t\t\"invalid config name\": {\n\t\t\tinput: &types.ListDynamicConfigRequest{\n\t\t\t\tConfigName: \"invalid\",\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *dynamicconfig.MockClient) {\n\t\t\t\tmock.EXPECT().ListValue(gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t\"valid config name\": {\n\t\t\tinput: &types.ListDynamicConfigRequest{\n\t\t\t\tConfigName: \"testGetIntPropertyKey\",\n\t\t\t},\n\t\t\tdcHandlerFunc: func(mock *dynamicconfig.MockClient) {\n\t\t\t\tmock.EXPECT().ListValue(gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgoMock := gomock.NewController(t)\n\t\t\tdcMock := dynamicconfig.NewMockClient(goMock)\n\t\t\tif td.dcHandlerFunc != nil {\n\t\t\t\ttd.dcHandlerFunc(dcMock)\n\t\t\t}\n\t\t\thandler := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:        testlogger.New(t),\n\t\t\t\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t\t},\n\t\t\t\tparams: &resource.Params{\n\t\t\t\t\tDynamicConfig: dcMock,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t_, err := handler.ListDynamicConfig(context.Background(), td.input)\n\t\t\tif td.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTaskListPartitionConfig(t *testing.T) {\n\tdomainName := \"domain-name\"\n\tdomainID := \"domain-id\"\n\ttaskListName := \"task-list\"\n\tkind := types.TaskListKindNormal\n\ttaskListType := types.TaskListTypeActivity\n\tpartitionConfig := &types.TaskListPartitionConfig{\n\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t0: {},\n\t\t\t1: {},\n\t\t},\n\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t0: {},\n\t\t},\n\t}\n\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.UpdateTaskListPartitionConfigRequest\n\t\tsetupMocks    func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain:          domainName,\n\t\t\t\tTaskList:        &types.TaskList{Name: taskListName, Kind: &kind},\n\t\t\t\tTaskListType:    &taskListType,\n\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t\tmockMatchingClient.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:      domainID,\n\t\t\t\t\tTaskList:        &types.TaskList{Name: taskListName, Kind: &kind},\n\t\t\t\t\tTaskListType:    &taskListType,\n\t\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t\t}).Return(nil, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"request not set\",\n\t\t\treq:  nil,\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// no mocks needed as the function should exit early\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: validate.ErrRequestNotSet.Error(),\n\t\t},\n\t\t{\n\t\t\tname: \"task list not set\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: validate.ErrTaskListNotSet.Error(),\n\t\t},\n\t\t{\n\t\t\tname: \"task list kind not set\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain:   domainName,\n\t\t\t\tTaskList: &types.TaskList{Name: taskListName},\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Task list kind not set.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid task list kind\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain:   domainName,\n\t\t\t\tTaskList: &types.TaskList{Name: taskListName, Kind: types.TaskListKindSticky.Ptr()},\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only normal tasklist's partition config can be updated.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list type not set\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain:   domainName,\n\t\t\t\tTaskList: &types.TaskList{Name: taskListName, Kind: &kind},\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Task list type not set.\",\n\t\t},\n\t\t{\n\t\t\tname: \"partition config not set\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain:       domainName,\n\t\t\t\tTaskList:     &types.TaskList{Name: taskListName, Kind: &kind},\n\t\t\t\tTaskListType: &taskListType,\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Task list partition config is not set in the request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid partition config: write partitions > read partitions\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: taskListName,\n\t\t\t\t\tKind: &kind,\n\t\t\t\t},\n\t\t\t\tTaskListType: &taskListType,\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"The number of write partitions cannot be larger than the number of read partitions.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid partition config: write partitions <= 0\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: taskListName,\n\t\t\t\t\tKind: &kind,\n\t\t\t\t},\n\t\t\t\tTaskListType: &taskListType,\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"The number of partitions must be larger than 0.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: taskListName,\n\t\t\t\t\tKind: &kind,\n\t\t\t\t},\n\t\t\t\tTaskListType:    &taskListType,\n\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(\"\", errors.New(\"domain cache error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"matching client error\",\n\t\t\treq: &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: taskListName,\n\t\t\t\t\tKind: &kind,\n\t\t\t\t},\n\t\t\t\tTaskListType:    &taskListType,\n\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t},\n\t\t\tsetupMocks: func(mockMatchingClient *matching.MockClient, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(domainName).Return(domainID, nil)\n\t\t\t\tmockMatchingClient.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:      domainID,\n\t\t\t\t\tTaskList:        &types.TaskList{Name: taskListName, Kind: &kind},\n\t\t\t\t\tTaskListType:    &taskListType,\n\t\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"matching client error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tmockMatchingClient := matching.NewMockClient(ctrl)\n\t\t\ttc.setupMocks(mockMatchingClient, mockDomainCache)\n\t\t\tadh := adminHandlerImpl{\n\t\t\t\tResource: &resource.Test{\n\t\t\t\t\tLogger:         testlogger.New(t),\n\t\t\t\t\tMetricsClient:  metrics.NewNoopMetricsClient(),\n\t\t\t\t\tDomainCache:    mockDomainCache,\n\t\t\t\t\tMatchingClient: mockMatchingClient,\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tresp, err := adh.UpdateTaskListPartitionConfig(context.Background(), tc.req)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/admin/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/service/frontend/admin\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/accesscontrolled.tmpl -o ../wrappers/accesscontrolled/admin_generated.go -v handler=Admin\n//go:generate gowrap gen -g -p . -i Handler -t ../../templates/grpc.tmpl -o ../wrappers/grpc/admin_generated.go -v handler=Admin -v package=adminv1 -v path=github.com/uber/cadence-idl/go/proto/admin/v1 -v prefix=Admin\n//go:generate gowrap gen -g -p ../../../.gen/go/admin/adminserviceserver -i Interface -t ../../templates/thrift.tmpl -o ../wrappers/thrift/admin_generated.go -v handler=Admin -v prefix=Admin\n\npackage admin\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Handler interface for admin service\ntype Handler interface {\n\tcommon.Daemon\n\n\tAddSearchAttribute(context.Context, *types.AddSearchAttributeRequest) error\n\tCloseShard(context.Context, *types.CloseShardRequest) error\n\tDescribeCluster(context.Context) (*types.DescribeClusterResponse, error)\n\tDescribeShardDistribution(context.Context, *types.DescribeShardDistributionRequest) (*types.DescribeShardDistributionResponse, error)\n\tDescribeHistoryHost(context.Context, *types.DescribeHistoryHostRequest) (*types.DescribeHistoryHostResponse, error)\n\tDescribeQueue(context.Context, *types.DescribeQueueRequest) (*types.DescribeQueueResponse, error)\n\tDescribeWorkflowExecution(context.Context, *types.AdminDescribeWorkflowExecutionRequest) (*types.AdminDescribeWorkflowExecutionResponse, error)\n\tGetDLQReplicationMessages(context.Context, *types.GetDLQReplicationMessagesRequest) (*types.GetDLQReplicationMessagesResponse, error)\n\tGetDomainReplicationMessages(context.Context, *types.GetDomainReplicationMessagesRequest) (*types.GetDomainReplicationMessagesResponse, error)\n\tGetReplicationMessages(context.Context, *types.GetReplicationMessagesRequest) (*types.GetReplicationMessagesResponse, error)\n\tGetWorkflowExecutionRawHistoryV2(context.Context, *types.GetWorkflowExecutionRawHistoryV2Request) (*types.GetWorkflowExecutionRawHistoryV2Response, error)\n\tCountDLQMessages(context.Context, *types.CountDLQMessagesRequest) (*types.CountDLQMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error)\n\tPurgeDLQMessages(context.Context, *types.PurgeDLQMessagesRequest) error\n\tReadDLQMessages(context.Context, *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error)\n\tReapplyEvents(context.Context, *types.ReapplyEventsRequest) error\n\tRefreshWorkflowTasks(context.Context, *types.RefreshWorkflowTasksRequest) error\n\tRemoveTask(context.Context, *types.RemoveTaskRequest) error\n\tResendReplicationTasks(context.Context, *types.ResendReplicationTasksRequest) error\n\tResetQueue(context.Context, *types.ResetQueueRequest) error\n\tGetCrossClusterTasks(context.Context, *types.GetCrossClusterTasksRequest) (*types.GetCrossClusterTasksResponse, error)\n\tRespondCrossClusterTasksCompleted(context.Context, *types.RespondCrossClusterTasksCompletedRequest) (*types.RespondCrossClusterTasksCompletedResponse, error)\n\tGetDynamicConfig(context.Context, *types.GetDynamicConfigRequest) (*types.GetDynamicConfigResponse, error)\n\tUpdateDynamicConfig(context.Context, *types.UpdateDynamicConfigRequest) error\n\tRestoreDynamicConfig(context.Context, *types.RestoreDynamicConfigRequest) error\n\tListDynamicConfig(context.Context, *types.ListDynamicConfigRequest) (*types.ListDynamicConfigResponse, error)\n\tDeleteWorkflow(context.Context, *types.AdminDeleteWorkflowRequest) (*types.AdminDeleteWorkflowResponse, error)\n\tMaintainCorruptWorkflow(context.Context, *types.AdminMaintainWorkflowRequest) (*types.AdminMaintainWorkflowResponse, error)\n\tGetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest) (*types.GetGlobalIsolationGroupsResponse, error)\n\tUpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest) (*types.UpdateGlobalIsolationGroupsResponse, error)\n\tGetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest) (*types.GetDomainIsolationGroupsResponse, error)\n\tUpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest) (*types.UpdateDomainIsolationGroupsResponse, error)\n\tGetDomainAsyncWorkflowConfiguraton(context.Context, *types.GetDomainAsyncWorkflowConfiguratonRequest) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error)\n\tUpdateDomainAsyncWorkflowConfiguraton(context.Context, *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error)\n\tUpdateTaskListPartitionConfig(context.Context, *types.UpdateTaskListPartitionConfigRequest) (*types.UpdateTaskListPartitionConfigResponse, error)\n}\n"
  },
  {
    "path": "service/frontend/admin/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package admin -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/service/frontend/admin\n//\n\n// Package admin is a generated GoMock package.\npackage admin\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// AddSearchAttribute mocks base method.\nfunc (m *MockHandler) AddSearchAttribute(arg0 context.Context, arg1 *types.AddSearchAttributeRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddSearchAttribute\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AddSearchAttribute indicates an expected call of AddSearchAttribute.\nfunc (mr *MockHandlerMockRecorder) AddSearchAttribute(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddSearchAttribute\", reflect.TypeOf((*MockHandler)(nil).AddSearchAttribute), arg0, arg1)\n}\n\n// CloseShard mocks base method.\nfunc (m *MockHandler) CloseShard(arg0 context.Context, arg1 *types.CloseShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CloseShard\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CloseShard indicates an expected call of CloseShard.\nfunc (mr *MockHandlerMockRecorder) CloseShard(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseShard\", reflect.TypeOf((*MockHandler)(nil).CloseShard), arg0, arg1)\n}\n\n// CountDLQMessages mocks base method.\nfunc (m *MockHandler) CountDLQMessages(arg0 context.Context, arg1 *types.CountDLQMessagesRequest) (*types.CountDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.CountDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountDLQMessages indicates an expected call of CountDLQMessages.\nfunc (mr *MockHandlerMockRecorder) CountDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountDLQMessages\", reflect.TypeOf((*MockHandler)(nil).CountDLQMessages), arg0, arg1)\n}\n\n// DeleteWorkflow mocks base method.\nfunc (m *MockHandler) DeleteWorkflow(arg0 context.Context, arg1 *types.AdminDeleteWorkflowRequest) (*types.AdminDeleteWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteWorkflow\", arg0, arg1)\n\tret0, _ := ret[0].(*types.AdminDeleteWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DeleteWorkflow indicates an expected call of DeleteWorkflow.\nfunc (mr *MockHandlerMockRecorder) DeleteWorkflow(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteWorkflow\", reflect.TypeOf((*MockHandler)(nil).DeleteWorkflow), arg0, arg1)\n}\n\n// DescribeCluster mocks base method.\nfunc (m *MockHandler) DescribeCluster(arg0 context.Context) (*types.DescribeClusterResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeCluster\", arg0)\n\tret0, _ := ret[0].(*types.DescribeClusterResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeCluster indicates an expected call of DescribeCluster.\nfunc (mr *MockHandlerMockRecorder) DescribeCluster(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeCluster\", reflect.TypeOf((*MockHandler)(nil).DescribeCluster), arg0)\n}\n\n// DescribeHistoryHost mocks base method.\nfunc (m *MockHandler) DescribeHistoryHost(arg0 context.Context, arg1 *types.DescribeHistoryHostRequest) (*types.DescribeHistoryHostResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeHistoryHostResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeHistoryHost indicates an expected call of DescribeHistoryHost.\nfunc (mr *MockHandlerMockRecorder) DescribeHistoryHost(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeHistoryHost\", reflect.TypeOf((*MockHandler)(nil).DescribeHistoryHost), arg0, arg1)\n}\n\n// DescribeQueue mocks base method.\nfunc (m *MockHandler) DescribeQueue(arg0 context.Context, arg1 *types.DescribeQueueRequest) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeQueue\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeQueue indicates an expected call of DescribeQueue.\nfunc (mr *MockHandlerMockRecorder) DescribeQueue(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeQueue\", reflect.TypeOf((*MockHandler)(nil).DescribeQueue), arg0, arg1)\n}\n\n// DescribeShardDistribution mocks base method.\nfunc (m *MockHandler) DescribeShardDistribution(arg0 context.Context, arg1 *types.DescribeShardDistributionRequest) (*types.DescribeShardDistributionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeShardDistribution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeShardDistributionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeShardDistribution indicates an expected call of DescribeShardDistribution.\nfunc (mr *MockHandlerMockRecorder) DescribeShardDistribution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeShardDistribution\", reflect.TypeOf((*MockHandler)(nil).DescribeShardDistribution), arg0, arg1)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockHandler) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.AdminDescribeWorkflowExecutionRequest) (*types.AdminDescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.AdminDescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) DescribeWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).DescribeWorkflowExecution), arg0, arg1)\n}\n\n// GetCrossClusterTasks mocks base method.\nfunc (m *MockHandler) GetCrossClusterTasks(arg0 context.Context, arg1 *types.GetCrossClusterTasksRequest) (*types.GetCrossClusterTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCrossClusterTasks\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetCrossClusterTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCrossClusterTasks indicates an expected call of GetCrossClusterTasks.\nfunc (mr *MockHandlerMockRecorder) GetCrossClusterTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCrossClusterTasks\", reflect.TypeOf((*MockHandler)(nil).GetCrossClusterTasks), arg0, arg1)\n}\n\n// GetDLQReplicationMessages mocks base method.\nfunc (m *MockHandler) GetDLQReplicationMessages(arg0 context.Context, arg1 *types.GetDLQReplicationMessagesRequest) (*types.GetDLQReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDLQReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQReplicationMessages indicates an expected call of GetDLQReplicationMessages.\nfunc (mr *MockHandlerMockRecorder) GetDLQReplicationMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQReplicationMessages\", reflect.TypeOf((*MockHandler)(nil).GetDLQReplicationMessages), arg0, arg1)\n}\n\n// GetDomainAsyncWorkflowConfiguraton mocks base method.\nfunc (m *MockHandler) GetDomainAsyncWorkflowConfiguraton(arg0 context.Context, arg1 *types.GetDomainAsyncWorkflowConfiguratonRequest) (*types.GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAsyncWorkflowConfiguraton\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainAsyncWorkflowConfiguraton indicates an expected call of GetDomainAsyncWorkflowConfiguraton.\nfunc (mr *MockHandlerMockRecorder) GetDomainAsyncWorkflowConfiguraton(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAsyncWorkflowConfiguraton\", reflect.TypeOf((*MockHandler)(nil).GetDomainAsyncWorkflowConfiguraton), arg0, arg1)\n}\n\n// GetDomainIsolationGroups mocks base method.\nfunc (m *MockHandler) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest) (*types.GetDomainIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainIsolationGroups\", ctx, request)\n\tret0, _ := ret[0].(*types.GetDomainIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainIsolationGroups indicates an expected call of GetDomainIsolationGroups.\nfunc (mr *MockHandlerMockRecorder) GetDomainIsolationGroups(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainIsolationGroups\", reflect.TypeOf((*MockHandler)(nil).GetDomainIsolationGroups), ctx, request)\n}\n\n// GetDomainReplicationMessages mocks base method.\nfunc (m *MockHandler) GetDomainReplicationMessages(arg0 context.Context, arg1 *types.GetDomainReplicationMessagesRequest) (*types.GetDomainReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainReplicationMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDomainReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDomainReplicationMessages indicates an expected call of GetDomainReplicationMessages.\nfunc (mr *MockHandlerMockRecorder) GetDomainReplicationMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainReplicationMessages\", reflect.TypeOf((*MockHandler)(nil).GetDomainReplicationMessages), arg0, arg1)\n}\n\n// GetDynamicConfig mocks base method.\nfunc (m *MockHandler) GetDynamicConfig(arg0 context.Context, arg1 *types.GetDynamicConfigRequest) (*types.GetDynamicConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDynamicConfig\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDynamicConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDynamicConfig indicates an expected call of GetDynamicConfig.\nfunc (mr *MockHandlerMockRecorder) GetDynamicConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDynamicConfig\", reflect.TypeOf((*MockHandler)(nil).GetDynamicConfig), arg0, arg1)\n}\n\n// GetGlobalIsolationGroups mocks base method.\nfunc (m *MockHandler) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest) (*types.GetGlobalIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetGlobalIsolationGroups\", ctx, request)\n\tret0, _ := ret[0].(*types.GetGlobalIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetGlobalIsolationGroups indicates an expected call of GetGlobalIsolationGroups.\nfunc (mr *MockHandlerMockRecorder) GetGlobalIsolationGroups(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetGlobalIsolationGroups\", reflect.TypeOf((*MockHandler)(nil).GetGlobalIsolationGroups), ctx, request)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockHandler) GetReplicationMessages(arg0 context.Context, arg1 *types.GetReplicationMessagesRequest) (*types.GetReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockHandlerMockRecorder) GetReplicationMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockHandler)(nil).GetReplicationMessages), arg0, arg1)\n}\n\n// GetWorkflowExecutionRawHistoryV2 mocks base method.\nfunc (m *MockHandler) GetWorkflowExecutionRawHistoryV2(arg0 context.Context, arg1 *types.GetWorkflowExecutionRawHistoryV2Request) (*types.GetWorkflowExecutionRawHistoryV2Response, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionRawHistoryV2\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetWorkflowExecutionRawHistoryV2Response)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecutionRawHistoryV2 indicates an expected call of GetWorkflowExecutionRawHistoryV2.\nfunc (mr *MockHandlerMockRecorder) GetWorkflowExecutionRawHistoryV2(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecutionRawHistoryV2\", reflect.TypeOf((*MockHandler)(nil).GetWorkflowExecutionRawHistoryV2), arg0, arg1)\n}\n\n// ListDynamicConfig mocks base method.\nfunc (m *MockHandler) ListDynamicConfig(arg0 context.Context, arg1 *types.ListDynamicConfigRequest) (*types.ListDynamicConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListDynamicConfig\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListDynamicConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDynamicConfig indicates an expected call of ListDynamicConfig.\nfunc (mr *MockHandlerMockRecorder) ListDynamicConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDynamicConfig\", reflect.TypeOf((*MockHandler)(nil).ListDynamicConfig), arg0, arg1)\n}\n\n// MaintainCorruptWorkflow mocks base method.\nfunc (m *MockHandler) MaintainCorruptWorkflow(arg0 context.Context, arg1 *types.AdminMaintainWorkflowRequest) (*types.AdminMaintainWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MaintainCorruptWorkflow\", arg0, arg1)\n\tret0, _ := ret[0].(*types.AdminMaintainWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MaintainCorruptWorkflow indicates an expected call of MaintainCorruptWorkflow.\nfunc (mr *MockHandlerMockRecorder) MaintainCorruptWorkflow(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaintainCorruptWorkflow\", reflect.TypeOf((*MockHandler)(nil).MaintainCorruptWorkflow), arg0, arg1)\n}\n\n// MergeDLQMessages mocks base method.\nfunc (m *MockHandler) MergeDLQMessages(arg0 context.Context, arg1 *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MergeDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MergeDLQMessages indicates an expected call of MergeDLQMessages.\nfunc (mr *MockHandlerMockRecorder) MergeDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeDLQMessages\", reflect.TypeOf((*MockHandler)(nil).MergeDLQMessages), arg0, arg1)\n}\n\n// PurgeDLQMessages mocks base method.\nfunc (m *MockHandler) PurgeDLQMessages(arg0 context.Context, arg1 *types.PurgeDLQMessagesRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PurgeDLQMessages indicates an expected call of PurgeDLQMessages.\nfunc (mr *MockHandlerMockRecorder) PurgeDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PurgeDLQMessages\", reflect.TypeOf((*MockHandler)(nil).PurgeDLQMessages), arg0, arg1)\n}\n\n// ReadDLQMessages mocks base method.\nfunc (m *MockHandler) ReadDLQMessages(arg0 context.Context, arg1 *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ReadDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadDLQMessages indicates an expected call of ReadDLQMessages.\nfunc (mr *MockHandlerMockRecorder) ReadDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadDLQMessages\", reflect.TypeOf((*MockHandler)(nil).ReadDLQMessages), arg0, arg1)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockHandler) ReapplyEvents(arg0 context.Context, arg1 *types.ReapplyEventsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockHandlerMockRecorder) ReapplyEvents(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockHandler)(nil).ReapplyEvents), arg0, arg1)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockHandler) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.RefreshWorkflowTasksRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockHandlerMockRecorder) RefreshWorkflowTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockHandler)(nil).RefreshWorkflowTasks), arg0, arg1)\n}\n\n// RemoveTask mocks base method.\nfunc (m *MockHandler) RemoveTask(arg0 context.Context, arg1 *types.RemoveTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RemoveTask\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveTask indicates an expected call of RemoveTask.\nfunc (mr *MockHandlerMockRecorder) RemoveTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveTask\", reflect.TypeOf((*MockHandler)(nil).RemoveTask), arg0, arg1)\n}\n\n// ResendReplicationTasks mocks base method.\nfunc (m *MockHandler) ResendReplicationTasks(arg0 context.Context, arg1 *types.ResendReplicationTasksRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResendReplicationTasks\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResendReplicationTasks indicates an expected call of ResendReplicationTasks.\nfunc (mr *MockHandlerMockRecorder) ResendReplicationTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResendReplicationTasks\", reflect.TypeOf((*MockHandler)(nil).ResendReplicationTasks), arg0, arg1)\n}\n\n// ResetQueue mocks base method.\nfunc (m *MockHandler) ResetQueue(arg0 context.Context, arg1 *types.ResetQueueRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetQueue\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetQueue indicates an expected call of ResetQueue.\nfunc (mr *MockHandlerMockRecorder) ResetQueue(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetQueue\", reflect.TypeOf((*MockHandler)(nil).ResetQueue), arg0, arg1)\n}\n\n// RespondCrossClusterTasksCompleted mocks base method.\nfunc (m *MockHandler) RespondCrossClusterTasksCompleted(arg0 context.Context, arg1 *types.RespondCrossClusterTasksCompletedRequest) (*types.RespondCrossClusterTasksCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondCrossClusterTasksCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RespondCrossClusterTasksCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondCrossClusterTasksCompleted indicates an expected call of RespondCrossClusterTasksCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondCrossClusterTasksCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondCrossClusterTasksCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondCrossClusterTasksCompleted), arg0, arg1)\n}\n\n// RestoreDynamicConfig mocks base method.\nfunc (m *MockHandler) RestoreDynamicConfig(arg0 context.Context, arg1 *types.RestoreDynamicConfigRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RestoreDynamicConfig\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RestoreDynamicConfig indicates an expected call of RestoreDynamicConfig.\nfunc (mr *MockHandlerMockRecorder) RestoreDynamicConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestoreDynamicConfig\", reflect.TypeOf((*MockHandler)(nil).RestoreDynamicConfig), arg0, arg1)\n}\n\n// Start mocks base method.\nfunc (m *MockHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockHandler)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockHandler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockHandlerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockHandler)(nil).Stop))\n}\n\n// UpdateDomainAsyncWorkflowConfiguraton mocks base method.\nfunc (m *MockHandler) UpdateDomainAsyncWorkflowConfiguraton(arg0 context.Context, arg1 *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (*types.UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainAsyncWorkflowConfiguraton\", arg0, arg1)\n\tret0, _ := ret[0].(*types.UpdateDomainAsyncWorkflowConfiguratonResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainAsyncWorkflowConfiguraton indicates an expected call of UpdateDomainAsyncWorkflowConfiguraton.\nfunc (mr *MockHandlerMockRecorder) UpdateDomainAsyncWorkflowConfiguraton(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainAsyncWorkflowConfiguraton\", reflect.TypeOf((*MockHandler)(nil).UpdateDomainAsyncWorkflowConfiguraton), arg0, arg1)\n}\n\n// UpdateDomainIsolationGroups mocks base method.\nfunc (m *MockHandler) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest) (*types.UpdateDomainIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainIsolationGroups\", ctx, request)\n\tret0, _ := ret[0].(*types.UpdateDomainIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomainIsolationGroups indicates an expected call of UpdateDomainIsolationGroups.\nfunc (mr *MockHandlerMockRecorder) UpdateDomainIsolationGroups(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainIsolationGroups\", reflect.TypeOf((*MockHandler)(nil).UpdateDomainIsolationGroups), ctx, request)\n}\n\n// UpdateDynamicConfig mocks base method.\nfunc (m *MockHandler) UpdateDynamicConfig(arg0 context.Context, arg1 *types.UpdateDynamicConfigRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDynamicConfig\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDynamicConfig indicates an expected call of UpdateDynamicConfig.\nfunc (mr *MockHandlerMockRecorder) UpdateDynamicConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDynamicConfig\", reflect.TypeOf((*MockHandler)(nil).UpdateDynamicConfig), arg0, arg1)\n}\n\n// UpdateGlobalIsolationGroups mocks base method.\nfunc (m *MockHandler) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest) (*types.UpdateGlobalIsolationGroupsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateGlobalIsolationGroups\", ctx, request)\n\tret0, _ := ret[0].(*types.UpdateGlobalIsolationGroupsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateGlobalIsolationGroups indicates an expected call of UpdateGlobalIsolationGroups.\nfunc (mr *MockHandlerMockRecorder) UpdateGlobalIsolationGroups(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateGlobalIsolationGroups\", reflect.TypeOf((*MockHandler)(nil).UpdateGlobalIsolationGroups), ctx, request)\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockHandler) UpdateTaskListPartitionConfig(arg0 context.Context, arg1 *types.UpdateTaskListPartitionConfigRequest) (*types.UpdateTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", arg0, arg1)\n\tret0, _ := ret[0].(*types.UpdateTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockHandlerMockRecorder) UpdateTaskListPartitionConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockHandler)(nil).UpdateTaskListPartitionConfig), arg0, arg1)\n}\n"
  },
  {
    "path": "service/frontend/api/domain_handlers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\n// RegisterDomain creates a new domain which can be used as a container for all resources.  Domain is a top level\n// entity within Cadence, used as a container for all resources like workflow executions, tasklists, etc.  Domain\n// acts as a sandbox and provides isolation for all resources within the domain.  All resources belongs to exactly one\n// domain.\nfunc (wh *WorkflowHandler) RegisterDomain(ctx context.Context, registerRequest *types.RegisterDomainRequest) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateRegisterDomainRequest(ctx, registerRequest); err != nil {\n\t\treturn err\n\t}\n\treturn wh.domainHandler.RegisterDomain(ctx, registerRequest)\n}\n\n// ListDomains returns the information and configuration for a registered domain.\nfunc (wh *WorkflowHandler) ListDomains(\n\tctx context.Context,\n\tlistRequest *types.ListDomainsRequest,\n) (response *types.ListDomainsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif listRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\treturn wh.domainHandler.ListDomains(ctx, listRequest)\n}\n\n// DescribeDomain returns the information and configuration for a registered domain.\nfunc (wh *WorkflowHandler) DescribeDomain(\n\tctx context.Context,\n\tdescribeRequest *types.DescribeDomainRequest,\n) (response *types.DescribeDomainResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateDescribeDomainRequest(ctx, describeRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := wh.domainHandler.DescribeDomain(ctx, describeRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif resp.GetFailoverInfo() != nil && resp.GetFailoverInfo().GetFailoverExpireTimestamp() > 0 {\n\t\t// fetch ongoing failover info from history service\n\t\tfailoverResp, err := wh.GetHistoryClient().GetFailoverInfo(ctx, &types.GetFailoverInfoRequest{\n\t\t\tDomainID: resp.GetDomainInfo().UUID,\n\t\t})\n\t\tif err != nil {\n\t\t\t// despite the error from history, return describe domain response\n\t\t\twh.GetLogger().Error(\n\t\t\t\tfmt.Sprintf(\"Failed to get failover info for domain %s\", resp.DomainInfo.GetName()),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\treturn resp, nil\n\t\t}\n\t\tresp.FailoverInfo.CompletedShardCount = failoverResp.GetCompletedShardCount()\n\t\tresp.FailoverInfo.PendingShards = failoverResp.GetPendingShards()\n\t}\n\treturn resp, nil\n}\n\n// UpdateDomain is used to update the information and configuration for a registered domain.\nfunc (wh *WorkflowHandler) UpdateDomain(\n\tctx context.Context,\n\tupdateRequest *types.UpdateDomainRequest,\n) (resp *types.UpdateDomainResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateUpdateDomainRequest(ctx, updateRequest); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName := updateRequest.GetName()\n\tlogger := wh.GetLogger().WithTags(\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.OperationName(\"DomainUpdate\"))\n\n\tisFailover := isFailoverRequest(updateRequest)\n\tisGraceFailover := isGraceFailoverRequest(updateRequest)\n\tlogger.Info(fmt.Sprintf(\n\t\t\"Domain Update requested. isFailover: %v, isGraceFailover: %v, Request: %#v.\",\n\t\tisFailover,\n\t\tisGraceFailover,\n\t\tupdateRequest))\n\n\tif isGraceFailover {\n\t\tif err := wh.checkOngoingFailover(\n\t\t\tctx,\n\t\t\t&updateRequest.Name,\n\t\t); err != nil {\n\t\t\tlogger.Error(\"Graceful domain failover request failed. Not able to check ongoing failovers.\",\n\t\t\t\ttag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\t// TODO: call remote clusters to verify domain data\n\tresp, err := wh.domainHandler.UpdateDomain(ctx, updateRequest)\n\tif err != nil {\n\t\tlogger.Error(\"Domain update operation failed.\",\n\t\t\ttag.Error(err))\n\t\treturn nil, err\n\t}\n\tlogger.Info(\"Domain update operation succeeded.\")\n\treturn resp, nil\n}\n\n// DeleteDomain permanently removes a domain record. This operation:\n// - Requires domain to be in DEPRECATED status\n// - Cannot be performed on domains with running workflows\n// - Is irreversible and removes all domain data\nfunc (wh *WorkflowHandler) DeleteDomain(ctx context.Context, deleteRequest *types.DeleteDomainRequest) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateDeleteDomainRequest(ctx, deleteRequest); err != nil {\n\t\treturn err\n\t}\n\n\tdomainName := deleteRequest.GetName()\n\tresp, err := wh.domainHandler.DescribeDomain(ctx, &types.DescribeDomainRequest{Name: &domainName})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif *resp.DomainInfo.Status != types.DomainStatusDeprecated {\n\t\treturn &types.BadRequestError{Message: \"Domain is not in a deprecated state.\"}\n\t}\n\n\tworkflowList, err := wh.ListWorkflowExecutions(ctx, &types.ListWorkflowExecutionsRequest{\n\t\tDomain: domainName,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif len(workflowList.Executions) != 0 {\n\t\treturn &types.BadRequestError{Message: \"Domain still have workflow execution history.\"}\n\t}\n\n\treturn wh.domainHandler.DeleteDomain(ctx, deleteRequest)\n}\n\n// DeprecateDomain is used to update status of a registered domain to DEleTED. Once the domain is deleted\n// it cannot be used to start new workflow executions.  Existing workflow executions will continue to run on\n// deprecated domains.\nfunc (wh *WorkflowHandler) DeprecateDomain(ctx context.Context, deprecateRequest *types.DeprecateDomainRequest) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateDeprecateDomainRequest(ctx, deprecateRequest); err != nil {\n\t\treturn err\n\t}\n\treturn wh.domainHandler.DeprecateDomain(ctx, deprecateRequest)\n}\n\n// FailoverDomain is used to failover a registered domain to different cluster.\nfunc (wh *WorkflowHandler) FailoverDomain(ctx context.Context, failoverRequest *types.FailoverDomainRequest) (*types.FailoverDomainResponse, error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateFailoverDomainRequest(ctx, failoverRequest); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName := failoverRequest.GetDomainName()\n\tlogger := wh.GetLogger().WithTags(\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.OperationName(\"FailoverDomain\"))\n\n\tlogger.Info(\"Failover domain requested.\",\n\t\ttag.ActiveClusterName(failoverRequest.GetDomainActiveClusterName()),\n\t\ttag.Dynamic(\"active-clusters-by-cluster-attribute\", failoverRequest.ActiveClusters),\n\t)\n\n\tfailoverResp, err := wh.domainHandler.FailoverDomain(ctx, failoverRequest)\n\tif err != nil {\n\t\tlogger.Error(\"Failover domain operation failed.\",\n\t\t\ttag.Error(err))\n\t\treturn nil, err\n\t}\n\n\tlogger.Info(\"Failover domain operation succeeded.\")\n\treturn failoverResp, nil\n}\n\nfunc (wh *WorkflowHandler) ListFailoverHistory(ctx context.Context, request *types.ListFailoverHistoryRequest) (*types.ListFailoverHistoryResponse, error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tlogger := wh.GetLogger().WithTags(\n\t\ttag.OperationName(\"ListFailoverHistory\"))\n\n\tlogger.Info(\"List failover history request received.\")\n\n\tdomainAuditManager := wh.GetPersistenceBean().GetDomainAuditManager()\n\tif domainAuditManager == nil {\n\t\treturn nil, &types.InternalServiceError{Message: \"DomainAuditManager not available\"}\n\t}\n\n\tfilters := request.GetFilters()\n\tif filters == nil || filters.DomainID == \"\" {\n\t\treturn nil, &types.BadRequestError{Message: \"DomainID is required in filters\"}\n\t}\n\n\tpageSize := 50\n\tif request.GetPagination() != nil && request.GetPagination().PageSize != nil {\n\t\tpageSize = int(*request.GetPagination().PageSize)\n\t}\n\n\tauditLogsResp, err := domainAuditManager.GetDomainAuditLogs(ctx, &persistence.GetDomainAuditLogsRequest{\n\t\tDomainID:      filters.DomainID,\n\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: request.GetPagination().GetNextPageToken(),\n\t})\n\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get domain audit logs\", tag.Error(err))\n\t\treturn nil, err\n\t}\n\n\tfailoverEvents := make([]*types.FailoverEvent, 0, len(auditLogsResp.AuditLogs))\n\tfor _, auditLog := range auditLogsResp.AuditLogs {\n\t\tevent := auditLog.ToFailoverEvents()\n\t\tif event != nil && event.ClusterFailovers != nil {\n\t\t\tfailoverEvents = append(failoverEvents, event)\n\t\t}\n\t}\n\n\treturn &types.ListFailoverHistoryResponse{\n\t\tFailoverEvents: failoverEvents,\n\t\tNextPageToken:  auditLogsResp.NextPageToken,\n\t}, nil\n}\n"
  },
  {
    "path": "service/frontend/api/domain_handlers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDeprecateDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DeprecateDomainRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeprecateDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DeprecateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"validation error\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeprecateDomainRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validation error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validation error\",\n\t\t},\n\t\t{\n\t\t\tname: \"deprecate domain handler error\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeprecateDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DeprecateDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"handler error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"handler error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\terr := wh.DeprecateDomain(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRegisterDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.RegisterDomainRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRegisterDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().RegisterDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"validation error\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRegisterDomainRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validation error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validation error\",\n\t\t},\n\t\t{\n\t\t\tname: \"register domain handler error\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName: \"domain-name\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRegisterDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().RegisterDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"handler error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"handler error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\t\t\terr := wh.RegisterDomain(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDescribeDomain(t *testing.T) {\n\tdomainName := \"domain-name\" // Define the domain name pointer to pass in requests\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DescribeDomainRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\tverifyResp    func(t *testing.T, resp *types.DescribeDomainResponse)\n\t}{\n\t\t{\n\t\t\tname: \"success without failover info\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: &domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName: \"domain-name\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverInfo: nil,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResp: func(t *testing.T, resp *types.DescribeDomainResponse) {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, \"domain-name\", resp.DomainInfo.Name)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with failover info and no error from history client\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: &domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName: \"domain-name\",\n\t\t\t\t\t\tUUID: \"domain-id\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\t\tFailoverExpireTimestamp: 1000,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().GetFailoverInfo(gomock.Any(), &types.GetFailoverInfoRequest{\n\t\t\t\t\tDomainID: \"domain-id\",\n\t\t\t\t}).Return(&types.GetFailoverInfoResponse{\n\t\t\t\t\tCompletedShardCount: 5,\n\t\t\t\t\tPendingShards:       []int32{10},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResp: func(t *testing.T, resp *types.DescribeDomainResponse) {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, int32(5), resp.FailoverInfo.CompletedShardCount)\n\t\t\t\tassert.Equal(t, []int32{10}, resp.FailoverInfo.PendingShards)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error from validation\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: &domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeDomainRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validation error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validation error\",\n\t\t},\n\t\t{\n\t\t\tname: \"error from domain handler\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: &domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"domain handler error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain handler error\",\n\t\t},\n\t\t{\n\t\t\tname: \"error from history client\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: &domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName: \"domain-name\",\n\t\t\t\t\t\tUUID: \"domain-id\",\n\t\t\t\t\t},\n\t\t\t\t\tFailoverInfo: &types.FailoverInfo{\n\t\t\t\t\t\tFailoverExpireTimestamp: 1000,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().GetFailoverInfo(gomock.Any(), &types.GetFailoverInfoRequest{\n\t\t\t\t\tDomainID: \"domain-id\",\n\t\t\t\t}).Return(nil, errors.New(\"history client error\"))\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResp: func(t *testing.T, resp *types.DescribeDomainResponse) {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, \"domain-name\", resp.DomainInfo.Name)\n\t\t\t\tassert.Zero(t, resp.FailoverInfo.CompletedShardCount)\n\t\t\t\tassert.Nil(t, resp.FailoverInfo.PendingShards)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\t\t\tresp, err := wh.DescribeDomain(context.Background(), tc.req)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\ttc.verifyResp(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeleteDomain(t *testing.T) {\n\tdomainName := \"domain-name\"\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DeleteDomainRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success_delete_deprecated_domain\",\n\t\t\treq: &types.DeleteDomainRequest{\n\t\t\t\tName: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"testDomainID\", nil).AnyTimes()\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeleteDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListWorkflowExecutionsRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   \"deprecated-domain\",\n\t\t\t\t\t\tStatus: types.DomainStatusDeprecated.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockVisibilityMgr.On(\"ListWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.ListWorkflowExecutionsResponse{Executions: []*types.WorkflowExecutionInfo{}}, nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"failure_domain_not_found\",\n\t\t\treq: &types.DeleteDomainRequest{\n\t\t\t\tName: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeleteDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(nil, errors.New(\"domain not found\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"failure_domain_not_deprecated\",\n\t\t\treq: &types.DeleteDomainRequest{\n\t\t\t\tName: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeleteDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   \"active-domain\",\n\t\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain is not in a deprecated state.\",\n\t\t},\n\t\t{\n\t\t\tname: \"failure_domain_has_workflow_history\",\n\t\t\treq: &types.DeleteDomainRequest{\n\t\t\t\tName: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"testDomainID\", nil).AnyTimes()\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeleteDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListWorkflowExecutionsRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   \"domain-with-history\",\n\t\t\t\t\t\tStatus: types.DomainStatusDeprecated.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockVisibilityMgr.On(\"ListWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.ListWorkflowExecutionsResponse{\n\t\t\t\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain still have workflow execution history.\",\n\t\t},\n\t\t{\n\t\t\tname: \"failure_delete_domain_error\",\n\t\t\treq: &types.DeleteDomainRequest{\n\t\t\t\tName: domainName,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"testDomainID\", nil).AnyTimes()\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDeleteDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListWorkflowExecutionsRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &domainName,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\tName:   \"deprecated-domain\",\n\t\t\t\t\t\tStatus: types.DomainStatusDeprecated.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockVisibilityMgr.On(\"ListWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.ListWorkflowExecutionsResponse{\n\t\t\t\t\tExecutions: []*types.WorkflowExecutionInfo{},\n\t\t\t\t}, nil)\n\t\t\t\tdeps.mockDomainHandler.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(errors.New(\"delete domain error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"delete domain error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\terr := wh.DeleteDomain(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListFailoverHistory(t *testing.T) {\n\tdomainID := \"test-domain-id\"\n\teventID1 := \"event-id-1\"\n\tcustomPageSize := int32(100)\n\tnextPageToken := []byte(\"next-page-token\")\n\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListFailoverHistoryRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\tverifyResp    func(t *testing.T, resp *types.ListFailoverHistoryResponse)\n\t}{\n\t\t{\n\t\t\tname: \"success_with_default_page_size\",\n\t\t\treq: &types.ListFailoverHistoryRequest{\n\t\t\t\tFilters: &types.ListFailoverHistoryRequestFilters{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockResource.DomainAuditMgr.EXPECT().GetDomainAuditLogs(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetDomainAuditLogsRequest{\n\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\t\t\t\t\tPageSize:      50,\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t},\n\t\t\t\t).Return(&persistence.GetDomainAuditLogsResponse{\n\t\t\t\t\tAuditLogs: []*persistence.DomainAuditLog{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID:  eventID1,\n\t\t\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\t\t\tStateBefore: &persistence.GetDomainResponse{\n\t\t\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-west\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tFailoverVersion: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tStateAfter: &persistence.GetDomainResponse{\n\t\t\t\t\t\t\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster-us-east\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tFailoverVersion: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nextPageToken,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResp: func(t *testing.T, resp *types.ListFailoverHistoryResponse) {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Len(t, resp.FailoverEvents, 1)\n\t\t\t\tassert.Equal(t, nextPageToken, resp.NextPageToken)\n\t\t\t\tassert.NotNil(t, resp.FailoverEvents[0].ClusterFailovers)\n\t\t\t\tassert.Len(t, resp.FailoverEvents[0].ClusterFailovers, 1)\n\t\t\t\tassert.Equal(t, \"cluster-us-west\", resp.FailoverEvents[0].ClusterFailovers[0].FromCluster.ActiveClusterName)\n\t\t\t\tassert.Equal(t, \"cluster-us-east\", resp.FailoverEvents[0].ClusterFailovers[0].ToCluster.ActiveClusterName)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success_with_custom_page_size\",\n\t\t\treq: &types.ListFailoverHistoryRequest{\n\t\t\t\tFilters: &types.ListFailoverHistoryRequestFilters{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t},\n\t\t\t\tPagination: &types.PaginationOptions{\n\t\t\t\t\tPageSize: &customPageSize,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockResource.DomainAuditMgr.EXPECT().GetDomainAuditLogs(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetDomainAuditLogsRequest{\n\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\tOperationType: persistence.DomainAuditOperationTypeFailover,\n\t\t\t\t\t\tPageSize:      100,\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t},\n\t\t\t\t).Return(&persistence.GetDomainAuditLogsResponse{\n\t\t\t\t\tAuditLogs:     []*persistence.DomainAuditLog{},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResp: func(t *testing.T, resp *types.ListFailoverHistoryResponse) {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Len(t, resp.FailoverEvents, 0)\n\t\t\t\tassert.Nil(t, resp.NextPageToken)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error_nil_filters\",\n\t\t\treq: &types.ListFailoverHistoryRequest{\n\t\t\t\tFilters: nil,\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\t// No mocks needed as validation fails early\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"DomainID is required in filters\",\n\t\t},\n\t\t{\n\t\t\tname: \"error_get_domain_audit_logs_fails\",\n\t\t\treq: &types.ListFailoverHistoryRequest{\n\t\t\t\tFilters: &types.ListFailoverHistoryRequestFilters{\n\t\t\t\t\tDomainID: domainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockResource.DomainAuditMgr.EXPECT().GetDomainAuditLogs(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, errors.New(\"persistence error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"persistence error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tresp, err := wh.ListFailoverHistory(context.Background(), tc.req)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\ttc.verifyResp(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/handler.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"go.uber.org/yarpc\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/elasticsearch/validator\"\n\tcadenceerrors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n\t\"github.com/uber/cadence/service/worker/diagnostics\"\n)\n\nconst (\n\t// HealthStatusOK is used when this node is healthy and rpc requests are allowed\n\tHealthStatusOK HealthStatus = iota + 1\n\t// HealthStatusWarmingUp is used when the rpc handler is warming up\n\tHealthStatusWarmingUp\n\t// HealthStatusShuttingDown is used when the rpc handler is shutting down\n\tHealthStatusShuttingDown\n)\n\nvar _ Handler = (*WorkflowHandler)(nil)\n\ntype (\n\t// WorkflowHandler - Thrift handler interface for workflow service\n\tWorkflowHandler struct {\n\t\tresource.Resource\n\n\t\tshuttingDown              int32\n\t\thealthStatus              int32\n\t\ttokenSerializer           common.TaskTokenSerializer\n\t\tconfig                    *config.Config\n\t\tversionChecker            client.VersionChecker\n\t\tdomainHandler             domain.Handler\n\t\tvisibilityQueryValidator  *validator.VisibilityQueryValidator\n\t\tsearchAttributesValidator *validator.SearchAttributesValidator\n\t\tthrottleRetry             *backoff.ThrottleRetry\n\t\tproducerManager           ProducerManager\n\t\tthriftrwEncoder           codec.BinaryEncoder\n\t\trequestValidator          RequestValidator\n\t}\n\n\tgetHistoryContinuationToken struct {\n\t\tRunID              string\n\t\tFirstEventID       int64\n\t\tNextEventID        int64\n\t\tIsWorkflowRunning  bool\n\t\tPersistenceToken   []byte\n\t\tTransientDecision  *types.TransientDecisionInfo\n\t\tBranchToken        []byte\n\t\tVersionHistoryItem *types.VersionHistoryItem\n\t}\n\n\tdomainGetter interface {\n\t\tGetDomain() string\n\t}\n\n\t// HealthStatus is an enum that refers to the rpc handler health status\n\tHealthStatus int32\n)\n\nvar (\n\tfrontendServiceRetryPolicy = common.CreateFrontendServiceRetryPolicy()\n)\n\n// NewWorkflowHandler creates a thrift handler for the cadence service\nfunc NewWorkflowHandler(\n\tresource resource.Resource,\n\tconfig *config.Config,\n\tversionChecker client.VersionChecker,\n\tdomainHandler domain.Handler,\n) *WorkflowHandler {\n\treturn &WorkflowHandler{\n\t\tResource:        resource,\n\t\tconfig:          config,\n\t\thealthStatus:    int32(HealthStatusWarmingUp),\n\t\ttokenSerializer: common.NewJSONTaskTokenSerializer(),\n\t\tversionChecker:  versionChecker,\n\t\tdomainHandler:   domainHandler,\n\t\tvisibilityQueryValidator: validator.NewQueryValidator(\n\t\t\tconfig.ValidSearchAttributes,\n\t\t\tconfig.EnableQueryAttributeValidation,\n\t\t),\n\t\tsearchAttributesValidator: validator.NewSearchAttributesValidator(\n\t\t\tresource.GetLogger(),\n\t\t\tconfig.EnableQueryAttributeValidation,\n\t\t\tconfig.ValidSearchAttributes,\n\t\t\tconfig.SearchAttributesNumberOfKeysLimit,\n\t\t\tconfig.SearchAttributesSizeOfValueLimit,\n\t\t\tconfig.SearchAttributesTotalSizeLimit,\n\t\t),\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(frontendServiceRetryPolicy),\n\t\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t\t),\n\t\tproducerManager: NewProducerManager(\n\t\t\tresource.GetDomainCache(),\n\t\t\tresource.GetAsyncWorkflowQueueProvider(),\n\t\t\tresource.GetLogger(),\n\t\t\tresource.GetMetricsClient(),\n\t\t),\n\t\tthriftrwEncoder:  codec.NewThriftRWEncoder(),\n\t\trequestValidator: NewRequestValidator(resource.GetLogger(), resource.GetMetricsClient(), config),\n\t}\n}\n\n// Start starts the handler\nfunc (wh *WorkflowHandler) Start() {\n\twarmupTimer := time.NewTimer(wh.config.WarmupDuration())\n\tgo func() {\n\t\t<-warmupTimer.C\n\t\twh.GetLogger().Warn(\"Service warmup duration has elapsed.\")\n\t\tif atomic.CompareAndSwapInt32(&wh.healthStatus, int32(HealthStatusWarmingUp), int32(HealthStatusOK)) {\n\t\t\twh.GetLogger().Warn(\"Warmup time has elapsed. Service is healthy.\")\n\t\t} else {\n\t\t\tstatus := HealthStatus(atomic.LoadInt32(&wh.healthStatus))\n\t\t\twh.GetLogger().Warn(fmt.Sprintf(\"Warmup time has elapsed. Service status is: %v\", status.String()))\n\t\t}\n\t}()\n}\n\n// Stop stops the handler\nfunc (wh *WorkflowHandler) Stop() {\n\tatomic.StoreInt32(&wh.shuttingDown, 1)\n}\n\n// UpdateHealthStatus sets the health status for this rpc handler.\n// This health status will be used within the rpc health check handler\nfunc (wh *WorkflowHandler) UpdateHealthStatus(status HealthStatus) {\n\tatomic.StoreInt32(&wh.healthStatus, int32(status))\n}\n\nfunc (wh *WorkflowHandler) isShuttingDown() bool {\n\treturn atomic.LoadInt32(&wh.shuttingDown) != 0\n}\n\n// Health is for health check\nfunc (wh *WorkflowHandler) Health(ctx context.Context) (*types.HealthStatus, error) {\n\tstatus := HealthStatus(atomic.LoadInt32(&wh.healthStatus))\n\tmsg := status.String()\n\n\tif status != HealthStatusOK {\n\t\twh.GetLogger().Warn(fmt.Sprintf(\"Service status is: %v\", msg))\n\t}\n\n\treturn &types.HealthStatus{\n\t\tOk:  status == HealthStatusOK,\n\t\tMsg: msg,\n\t}, nil\n}\n\n// DiagnoseWorkflowExecution is to diagnose a workflow execution\nfunc (wh *WorkflowHandler) DiagnoseWorkflowExecution(ctx context.Context, request *types.DiagnoseWorkflowExecutionRequest) (*types.DiagnoseWorkflowExecutionResponse, error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif request == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tif request.GetDomain() == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tif request.GetWorkflowExecution().GetWorkflowID() == \"\" || request.GetWorkflowExecution().GetRunID() == \"\" {\n\t\treturn nil, validate.ErrExecutionNotSet\n\t}\n\n\twfExecution := request.GetWorkflowExecution()\n\tdiagnosticWorkflowDomain := \"cadence-system\"\n\tdiagnosticWorkflowID := fmt.Sprintf(\"%s-%s\", request.GetDomain(), wfExecution.GetRunID())\n\n\tdiagnosticWorkflowInput := diagnostics.DiagnosticsStarterWorkflowInput{\n\t\tDomain:     request.GetDomain(),\n\t\tWorkflowID: request.GetWorkflowExecution().GetWorkflowID(),\n\t\tRunID:      request.GetWorkflowExecution().GetRunID(),\n\t\tIdentity:   request.GetIdentity(),\n\t}\n\tinputInBytes, err := json.Marshal(diagnosticWorkflowInput)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, err := wh.StartWorkflowExecution(ctx, &types.StartWorkflowExecutionRequest{\n\t\tDomain:     diagnosticWorkflowDomain,\n\t\tWorkflowID: diagnosticWorkflowID,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"diagnostics-starter-workflow\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"diagnostics-wf-tasklist\",\n\t\t},\n\t\tInput:                               inputInBytes,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(86400), // 24 hours\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(300),   // 5 minutes\n\t\tIdentity:                            request.GetIdentity(),\n\t\tRequestID:                           uuid.New().String(),\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t})\n\tif err != nil {\n\t\tvar wfExecutionStartedError *types.WorkflowExecutionAlreadyStartedError\n\t\tif errors.As(err, &wfExecutionStartedError) {\n\t\t\treturn &types.DiagnoseWorkflowExecutionResponse{\n\t\t\t\tDomain: diagnosticWorkflowDomain,\n\t\t\t\tDiagnosticWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: diagnosticWorkflowID,\n\t\t\t\t\tRunID:      wfExecutionStartedError.RunID,\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\t\treturn nil, err\n\t}\n\n\treturn &types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain: diagnosticWorkflowDomain,\n\t\tDiagnosticWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: diagnosticWorkflowID,\n\t\t\tRunID:      resp.GetRunID(),\n\t\t},\n\t}, nil\n}\n\n// PollForActivityTask - Poll for an activity task.\nfunc (wh *WorkflowHandler) PollForActivityTask(\n\tctx context.Context,\n\tpollRequest *types.PollForActivityTaskRequest,\n) (resp *types.PollForActivityTaskResponse, retError error) {\n\tcallTime := time.Now()\n\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif pollRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := pollRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendPollForActivityTaskScope, pollRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\twh.GetLogger().Debug(\"Received PollForActivityTask\")\n\tif err := common.ValidateLongPollContextTimeout(\n\t\tctx,\n\t\t\"PollForActivityTask\",\n\t\twh.GetThrottledLogger().WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(pollRequest.GetTaskList().GetName())),\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tidLengthWarnLimit := wh.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tdomainName,\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.DomainNameMaxLength(domainName),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeDomainName) {\n\t\treturn nil, validate.ErrDomainTooLong\n\t}\n\n\tif err := wh.validateTaskList(pollRequest.TaskList, scope, domainName); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tpollRequest.GetIdentity(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn nil, validate.ErrIdentityTooLong\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.waitUntilIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn &types.PollForActivityTaskResponse{}, nil\n\t}\n\t// it is possible that we wait for a very long time and the remaining time is not long enough for a long poll\n\t// in this case, return an empty response\n\tif err := common.ValidateLongPollContextTimeout(\n\t\tctx,\n\t\t\"PollForActivityTask\",\n\t\twh.GetThrottledLogger().WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(pollRequest.GetTaskList().GetName())),\n\t); err != nil {\n\t\treturn &types.PollForActivityTaskResponse{}, nil\n\t}\n\tpollerID := uuid.New().String()\n\tvar matchingResp *types.MatchingPollForActivityTaskResponse\n\top := func(ctx context.Context) error {\n\t\tmatchingResp, err = wh.GetMatchingClient().PollForActivityTask(ctx, &types.MatchingPollForActivityTaskRequest{\n\t\t\tDomainUUID:     domainID,\n\t\t\tPollerID:       pollerID,\n\t\t\tPollRequest:    pollRequest,\n\t\t\tIsolationGroup: isolationGroup,\n\t\t})\n\t\treturn err\n\t}\n\n\terr = wh.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\terr = wh.cancelOutstandingPoll(ctx, err, domainID, persistence.TaskListTypeActivity, pollRequest.TaskList, pollerID)\n\t\tif err != nil {\n\t\t\t// For all other errors log an error and return it back to client.\n\t\t\tctxTimeout := \"not-set\"\n\t\t\tctxDeadline, ok := ctx.Deadline()\n\t\t\tif ok {\n\t\t\t\tctxTimeout = ctxDeadline.Sub(callTime).String()\n\t\t\t}\n\t\t\tpeerHostname, origErr := cadenceerrors.ExtractPeerHostname(err)\n\t\t\twh.GetLogger().Error(\"PollForActivityTask failed.\",\n\t\t\t\ttag.WorkflowTaskListName(pollRequest.GetTaskList().GetName()),\n\t\t\t\ttag.Value(ctxTimeout),\n\t\t\t\ttag.Error(origErr),\n\t\t\t\ttag.PeerHostname(peerHostname))\n\t\t\treturn nil, err\n\t\t}\n\t\t// Must be cancellation error.  Doesn't matter what we return here.  Client already went away.\n\t\treturn nil, nil\n\t}\n\n\treturn &types.PollForActivityTaskResponse{\n\t\tTaskToken:                       matchingResp.TaskToken,\n\t\tWorkflowExecution:               matchingResp.WorkflowExecution,\n\t\tActivityID:                      matchingResp.ActivityID,\n\t\tActivityType:                    matchingResp.ActivityType,\n\t\tInput:                           matchingResp.Input,\n\t\tScheduledTimestamp:              matchingResp.ScheduledTimestamp,\n\t\tScheduleToCloseTimeoutSeconds:   matchingResp.ScheduleToCloseTimeoutSeconds,\n\t\tStartedTimestamp:                matchingResp.StartedTimestamp,\n\t\tStartToCloseTimeoutSeconds:      matchingResp.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:         matchingResp.HeartbeatTimeoutSeconds,\n\t\tAttempt:                         matchingResp.Attempt,\n\t\tScheduledTimestampOfThisAttempt: matchingResp.ScheduledTimestampOfThisAttempt,\n\t\tHeartbeatDetails:                matchingResp.HeartbeatDetails,\n\t\tWorkflowType:                    matchingResp.WorkflowType,\n\t\tWorkflowDomain:                  matchingResp.WorkflowDomain,\n\t\tHeader:                          matchingResp.Header,\n\t\tAutoConfigHint:                  matchingResp.AutoConfigHint,\n\t}, nil\n}\n\n// PollForDecisionTask - Poll for a decision task.\nfunc (wh *WorkflowHandler) PollForDecisionTask(\n\tctx context.Context,\n\tpollRequest *types.PollForDecisionTaskRequest,\n) (resp *types.PollForDecisionTaskResponse, retError error) {\n\tcallTime := time.Now()\n\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif pollRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := pollRequest.GetDomain()\n\ttags := []tag.Tag{tag.WorkflowDomainName(domainName)}\n\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendPollForDecisionTaskScope, pollRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\twh.GetLogger().Debug(\"Received PollForDecisionTask\")\n\tif err := common.ValidateLongPollContextTimeout(\n\t\tctx,\n\t\t\"PollForDecisionTask\",\n\t\twh.GetThrottledLogger().WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(pollRequest.GetTaskList().GetName())),\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tidLengthWarnLimit := wh.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tdomainName,\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.DomainNameMaxLength(domainName),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeDomainName) {\n\t\treturn nil, validate.ErrDomainTooLong\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tpollRequest.GetIdentity(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn nil, validate.ErrIdentityTooLong\n\t}\n\n\tif err := wh.validateTaskList(pollRequest.TaskList, scope, domainName); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainEntry, err := wh.GetDomainCache().GetDomain(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\twh.GetLogger().Debug(\"Poll for decision.\", tag.WorkflowDomainName(domainName), tag.WorkflowDomainID(domainID))\n\tif err := wh.checkBadBinary(domainEntry, pollRequest.GetBinaryChecksum()); err != nil {\n\t\treturn nil, err\n\t}\n\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.waitUntilIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn &types.PollForDecisionTaskResponse{}, nil\n\t}\n\t// it is possible that we wait for a very long time and the remaining time is not long enough for a long poll\n\t// in this case, return an empty response\n\tif err := common.ValidateLongPollContextTimeout(\n\t\tctx,\n\t\t\"PollForDecisionTask\",\n\t\twh.GetThrottledLogger().WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(pollRequest.GetTaskList().GetName())),\n\t); err != nil {\n\t\treturn &types.PollForDecisionTaskResponse{}, nil\n\t}\n\n\tpollerID := uuid.New().String()\n\tvar matchingResp *types.MatchingPollForDecisionTaskResponse\n\top := func(ctx context.Context) error {\n\t\tmatchingResp, err = wh.GetMatchingClient().PollForDecisionTask(ctx, &types.MatchingPollForDecisionTaskRequest{\n\t\t\tDomainUUID:     domainID,\n\t\t\tPollerID:       pollerID,\n\t\t\tPollRequest:    pollRequest,\n\t\t\tIsolationGroup: isolationGroup,\n\t\t})\n\t\treturn err\n\t}\n\n\terr = wh.throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\terr = wh.cancelOutstandingPoll(ctx, err, domainID, persistence.TaskListTypeDecision, pollRequest.TaskList, pollerID)\n\t\tif err != nil {\n\t\t\t// For all other errors log an error and return it back to client.\n\t\t\tctxTimeout := \"not-set\"\n\t\t\tctxDeadline, ok := ctx.Deadline()\n\t\t\tif ok {\n\t\t\t\tctxTimeout = ctxDeadline.Sub(callTime).String()\n\t\t\t}\n\t\t\tpeerHostname, origErr := cadenceerrors.ExtractPeerHostname(err)\n\t\t\twh.GetLogger().Error(\"PollForDecisionTask failed.\",\n\t\t\t\ttag.WorkflowTaskListName(pollRequest.GetTaskList().GetName()),\n\t\t\t\ttag.Value(ctxTimeout),\n\t\t\t\ttag.Error(origErr),\n\t\t\t\ttag.PeerHostname(peerHostname))\n\t\t\treturn nil, err\n\t\t}\n\n\t\t// Must be cancellation error.  Doesn't matter what we return here.  Client already went away.\n\t\treturn nil, nil\n\t}\n\n\ttags = append(tags, []tag.Tag{tag.WorkflowID(\n\t\tmatchingResp.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(matchingResp.GetWorkflowExecution().GetRunID())}...)\n\tresp, err = wh.createPollForDecisionTaskResponse(ctx, scope, domainID, matchingResp, matchingResp.GetBranchToken())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp, nil\n}\n\nfunc (wh *WorkflowHandler) getIsolationGroup(ctx context.Context, domainName string) string {\n\treturn isolationgroup.IsolationGroupFromContext(ctx)\n}\n\nfunc (wh *WorkflowHandler) getPartitionConfig(ctx context.Context, domainName string) map[string]string {\n\treturn isolationgroup.ConfigFromContext(ctx)\n}\n\nfunc (wh *WorkflowHandler) isIsolationGroupHealthy(ctx context.Context, domainName, isolationGroup string) bool {\n\tif wh.GetIsolationGroupState() != nil && wh.config.EnableTasklistIsolation(domainName) {\n\t\tisDrained, err := wh.GetIsolationGroupState().IsDrained(ctx, domainName, isolationGroup)\n\t\tif err != nil {\n\t\t\twh.GetLogger().Error(\"Failed to check if an isolation group is drained, assume it's healthy\", tag.Error(err))\n\t\t\treturn true\n\t\t}\n\t\treturn !isDrained\n\t}\n\treturn true\n}\n\nfunc (wh *WorkflowHandler) waitUntilIsolationGroupHealthy(ctx context.Context, domainName, isolationGroup string) bool {\n\tif wh.GetIsolationGroupState() != nil && wh.config.EnableTasklistIsolation(domainName) {\n\t\tticker := time.NewTicker(time.Second * 30)\n\t\tdefer ticker.Stop()\n\t\tchildCtx, cancel := common.CreateChildContext(ctx, 0.05)\n\t\tdefer cancel()\n\t\tfor {\n\t\t\tisDrained, err := wh.GetIsolationGroupState().IsDrained(childCtx, domainName, isolationGroup)\n\t\t\tif err != nil {\n\t\t\t\twh.GetLogger().Error(\"Failed to check if an isolation group is drained, assume it's healthy\", tag.Error(err))\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tif !isDrained {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tselect {\n\t\t\tcase <-childCtx.Done():\n\t\t\t\treturn false\n\t\t\tcase <-ticker.C:\n\t\t\t}\n\t\t}\n\t}\n\treturn true\n}\n\nfunc (wh *WorkflowHandler) checkBadBinary(domainEntry *cache.DomainCacheEntry, binaryChecksum string) error {\n\tif domainEntry.GetConfig().BadBinaries.Binaries != nil {\n\t\tbadBinaries := domainEntry.GetConfig().BadBinaries.Binaries\n\t\t_, ok := badBinaries[binaryChecksum]\n\t\tif ok {\n\t\t\twh.GetMetricsClient().IncCounter(metrics.FrontendPollForDecisionTaskScope, metrics.CadenceErrBadBinaryCounter)\n\t\t\treturn &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"binary %v already marked as bad deployment\", binaryChecksum),\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (wh *WorkflowHandler) cancelOutstandingPoll(ctx context.Context, err error, domainID string, taskListType int32,\n\ttaskList *types.TaskList, pollerID string) error {\n\t// First check if this err is due to context cancellation.  This means client connection to frontend is closed.\n\tif ctx.Err() == context.Canceled {\n\t\t// Our rpc stack does not propagates context cancellation to the other service.  Lets make an explicit\n\t\t// call to matching to notify this poller is gone to prevent any tasks being dispatched to zombie pollers.\n\t\terr = wh.GetMatchingClient().CancelOutstandingPoll(context.Background(), &types.CancelOutstandingPollRequest{\n\t\t\tDomainUUID:   domainID,\n\t\t\tTaskListType: common.Int32Ptr(taskListType),\n\t\t\tTaskList:     taskList,\n\t\t\tPollerID:     pollerID,\n\t\t})\n\t\t// We can not do much if this call fails.  Just log the error and move on\n\t\tif err != nil {\n\t\t\twh.GetLogger().Warn(\"Failed to cancel outstanding poller.\",\n\t\t\t\ttag.WorkflowTaskListName(taskList.GetName()), tag.Error(err))\n\t\t}\n\n\t\t// Clear error as we don't want to report context cancellation error to count against our SLA\n\t\treturn nil\n\t}\n\n\treturn err\n}\n\n// RecordActivityTaskHeartbeat - Record Activity Task Heart beat.\nfunc (wh *WorkflowHandler) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\theartbeatRequest *types.RecordActivityTaskHeartbeatRequest,\n) (resp *types.RecordActivityTaskHeartbeatResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif heartbeatRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\twh.GetLogger().Debug(\"Received RecordActivityTaskHeartbeat\")\n\tif heartbeatRequest.TaskToken == nil {\n\t\treturn nil, validate.ErrTaskTokenNotSet\n\t}\n\ttaskToken, err := wh.tokenSerializer.Deserialize(heartbeatRequest.TaskToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif taskToken.DomainID == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRecordActivityTaskHeartbeatScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(heartbeatRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RecordActivityTaskHeartbeat\"),\n\t); err != nil {\n\t\t// heartbeat details exceed size limit, we would fail the activity immediately with explicit error reason\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: heartbeatRequest.TaskToken,\n\t\t\tReason:    common.StringPtr(common.FailureReasonHeartbeatExceedsLimit),\n\t\t\tDetails:   heartbeatRequest.Details[0:sizeLimitError],\n\t\t\tIdentity:  heartbeatRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t\tresp = &types.RecordActivityTaskHeartbeatResponse{CancelRequested: true}\n\t} else {\n\t\tresp, err = wh.GetHistoryClient().RecordActivityTaskHeartbeat(ctx, &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\tDomainUUID:       taskToken.DomainID,\n\t\t\tHeartbeatRequest: heartbeatRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn resp, nil\n}\n\n// RecordActivityTaskHeartbeatByID - Record Activity Task Heart beat.\nfunc (wh *WorkflowHandler) RecordActivityTaskHeartbeatByID(\n\tctx context.Context,\n\theartbeatRequest *types.RecordActivityTaskHeartbeatByIDRequest,\n) (resp *types.RecordActivityTaskHeartbeatResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif heartbeatRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := heartbeatRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\twh.GetLogger().Debug(\"Received RecordActivityTaskHeartbeatByID\")\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowID := heartbeatRequest.GetWorkflowID()\n\trunID := heartbeatRequest.GetRunID() // runID is optional so can be empty\n\tactivityID := heartbeatRequest.GetActivityID()\n\n\tif domainID == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\tif workflowID == \"\" {\n\t\treturn nil, validate.ErrWorkflowIDNotSet\n\t}\n\tif activityID == \"\" {\n\t\treturn nil, validate.ErrActivityIDNotSet\n\t}\n\n\ttaskToken := &common.TaskToken{\n\t\tDomainID:   domainID,\n\t\tRunID:      runID,\n\t\tWorkflowID: workflowID,\n\t\tScheduleID: constants.EmptyEventID,\n\t\tActivityID: activityID,\n\t}\n\ttoken, err := wh.tokenSerializer.Serialize(taskToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRecordActivityTaskHeartbeatByIDScope, heartbeatRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(heartbeatRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RecordActivityTaskHeartbeatByID\"),\n\t); err != nil {\n\t\t// heartbeat details exceed size limit, we would fail the activity immediately with explicit error reason\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: token,\n\t\t\tReason:    common.StringPtr(common.FailureReasonHeartbeatExceedsLimit),\n\t\t\tDetails:   heartbeatRequest.Details[0:sizeLimitError],\n\t\t\tIdentity:  heartbeatRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t\tresp = &types.RecordActivityTaskHeartbeatResponse{CancelRequested: true}\n\t} else {\n\t\treq := &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: token,\n\t\t\tDetails:   heartbeatRequest.Details,\n\t\t\tIdentity:  heartbeatRequest.Identity,\n\t\t}\n\n\t\tresp, err = wh.GetHistoryClient().RecordActivityTaskHeartbeat(ctx, &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\tDomainUUID:       taskToken.DomainID,\n\t\t\tHeartbeatRequest: req,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn resp, nil\n}\n\n// RespondActivityTaskCompleted - response to an activity task\nfunc (wh *WorkflowHandler) RespondActivityTaskCompleted(\n\tctx context.Context,\n\tcompleteRequest *types.RespondActivityTaskCompletedRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif completeRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tif completeRequest.TaskToken == nil {\n\t\treturn validate.ErrTaskTokenNotSet\n\t}\n\ttaskToken, err := wh.tokenSerializer.Deserialize(completeRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif taskToken.DomainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskCompletedScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tcompleteRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(completeRequest.Result),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskCompleted\"),\n\t); err != nil {\n\t\t// result exceeds blob size limit, we would record it as failure\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: completeRequest.TaskToken,\n\t\t\tReason:    common.StringPtr(common.FailureReasonCompleteResultExceedsLimit),\n\t\t\tDetails:   completeRequest.Result[0:sizeLimitError],\n\t\t\tIdentity:  completeRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t} else {\n\t\terr = wh.GetHistoryClient().RespondActivityTaskCompleted(ctx, &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\tDomainUUID:      taskToken.DomainID,\n\t\t\tCompleteRequest: completeRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// RespondActivityTaskCompletedByID - response to an activity task\nfunc (wh *WorkflowHandler) RespondActivityTaskCompletedByID(\n\tctx context.Context,\n\tcompleteRequest *types.RespondActivityTaskCompletedByIDRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif completeRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := completeRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowID := completeRequest.GetWorkflowID()\n\trunID := completeRequest.GetRunID() // runID is optional so can be empty\n\tactivityID := completeRequest.GetActivityID()\n\n\tif domainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif workflowID == \"\" {\n\t\treturn validate.ErrWorkflowIDNotSet\n\t}\n\tif activityID == \"\" {\n\t\treturn validate.ErrActivityIDNotSet\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskCompletedByIDScope, completeRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tcompleteRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\ttaskToken := &common.TaskToken{\n\t\tDomainID:   domainID,\n\t\tRunID:      runID,\n\t\tWorkflowID: workflowID,\n\t\tScheduleID: constants.EmptyEventID,\n\t\tActivityID: activityID,\n\t}\n\ttoken, err := wh.tokenSerializer.Serialize(taskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(completeRequest.Result),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskCompletedByID\"),\n\t); err != nil {\n\t\t// result exceeds blob size limit, we would record it as failure\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: token,\n\t\t\tReason:    common.StringPtr(common.FailureReasonCompleteResultExceedsLimit),\n\t\t\tDetails:   completeRequest.Result[0:sizeLimitError],\n\t\t\tIdentity:  completeRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t} else {\n\t\treq := &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: token,\n\t\t\tResult:    completeRequest.Result,\n\t\t\tIdentity:  completeRequest.Identity,\n\t\t}\n\n\t\terr = wh.GetHistoryClient().RespondActivityTaskCompleted(ctx, &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\tDomainUUID:      taskToken.DomainID,\n\t\t\tCompleteRequest: req,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// RespondActivityTaskFailed - response to an activity task failure\nfunc (wh *WorkflowHandler) RespondActivityTaskFailed(\n\tctx context.Context,\n\tfailedRequest *types.RespondActivityTaskFailedRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif failedRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tif failedRequest.TaskToken == nil {\n\t\treturn validate.ErrTaskTokenNotSet\n\t}\n\ttaskToken, err := wh.tokenSerializer.Deserialize(failedRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif taskToken.DomainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskFailedScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tfailedRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(failedRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskFailed\"),\n\t); err != nil {\n\t\t// details exceeds blob size limit, we would truncate the details and put a specific error reason\n\t\tfailedRequest.Reason = common.StringPtr(common.FailureReasonFailureDetailsExceedsLimit)\n\t\tfailedRequest.Details = failedRequest.Details[0:sizeLimitError]\n\t}\n\n\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID:    taskToken.DomainID,\n\t\tFailedRequest: failedRequest,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\treturn nil\n}\n\n// RespondActivityTaskFailedByID - response to an activity task failure\nfunc (wh *WorkflowHandler) RespondActivityTaskFailedByID(\n\tctx context.Context,\n\tfailedRequest *types.RespondActivityTaskFailedByIDRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif failedRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := failedRequest.GetDomain()\n\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowID := failedRequest.GetWorkflowID()\n\trunID := failedRequest.GetRunID() // runID is optional so can be empty\n\tactivityID := failedRequest.GetActivityID()\n\n\tif domainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif workflowID == \"\" {\n\t\treturn validate.ErrWorkflowIDNotSet\n\t}\n\tif activityID == \"\" {\n\t\treturn validate.ErrActivityIDNotSet\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskFailedByIDScope, failedRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tfailedRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(failedRequest.GetDomain()),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\ttaskToken := &common.TaskToken{\n\t\tDomainID:   domainID,\n\t\tRunID:      runID,\n\t\tWorkflowID: workflowID,\n\t\tScheduleID: constants.EmptyEventID,\n\t\tActivityID: activityID,\n\t}\n\ttoken, err := wh.tokenSerializer.Serialize(taskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(failedRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskFailedByID\"),\n\t); err != nil {\n\t\t// details exceeds blob size limit, we would truncate the details and put a specific error reason\n\t\tfailedRequest.Reason = common.StringPtr(common.FailureReasonFailureDetailsExceedsLimit)\n\t\tfailedRequest.Details = failedRequest.Details[0:sizeLimitError]\n\t}\n\n\treq := &types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: token,\n\t\tReason:    failedRequest.Reason,\n\t\tDetails:   failedRequest.Details,\n\t\tIdentity:  failedRequest.Identity,\n\t}\n\n\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID:    taskToken.DomainID,\n\t\tFailedRequest: req,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\treturn nil\n}\n\n// RespondActivityTaskCanceled - called to cancel an activity task\nfunc (wh *WorkflowHandler) RespondActivityTaskCanceled(\n\tctx context.Context,\n\tcancelRequest *types.RespondActivityTaskCanceledRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif cancelRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tif cancelRequest.TaskToken == nil {\n\t\treturn validate.ErrTaskTokenNotSet\n\t}\n\n\ttaskToken, err := wh.tokenSerializer.Deserialize(cancelRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif taskToken.DomainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskCanceledScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tcancelRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(cancelRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskCanceled\"),\n\t); err != nil {\n\t\t// details exceeds blob size limit, we would record it as failure\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: cancelRequest.TaskToken,\n\t\t\tReason:    common.StringPtr(common.FailureReasonCancelDetailsExceedsLimit),\n\t\t\tDetails:   cancelRequest.Details[0:sizeLimitError],\n\t\t\tIdentity:  cancelRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t} else {\n\t\terr = wh.GetHistoryClient().RespondActivityTaskCanceled(ctx, &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tCancelRequest: cancelRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// RespondActivityTaskCanceledByID - called to cancel an activity task\nfunc (wh *WorkflowHandler) RespondActivityTaskCanceledByID(\n\tctx context.Context,\n\tcancelRequest *types.RespondActivityTaskCanceledByIDRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif cancelRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := cancelRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowID := cancelRequest.GetWorkflowID()\n\trunID := cancelRequest.GetRunID() // runID is optional so can be empty\n\tactivityID := cancelRequest.GetActivityID()\n\n\tif domainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif workflowID == \"\" {\n\t\treturn validate.ErrWorkflowIDNotSet\n\t}\n\tif activityID == \"\" {\n\t\treturn validate.ErrActivityIDNotSet\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondActivityTaskCanceledByIDScope, cancelRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tcancelRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(cancelRequest.GetDomain()),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\ttaskToken := &common.TaskToken{\n\t\tDomainID:   domainID,\n\t\tRunID:      runID,\n\t\tWorkflowID: workflowID,\n\t\tScheduleID: constants.EmptyEventID,\n\t\tActivityID: activityID,\n\t}\n\ttoken, err := wh.tokenSerializer.Serialize(taskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(cancelRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondActivityTaskCanceledByID\"),\n\t); err != nil {\n\t\t// details exceeds blob size limit, we would record it as failure\n\t\tfailRequest := &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: token,\n\t\t\tReason:    common.StringPtr(common.FailureReasonCancelDetailsExceedsLimit),\n\t\t\tDetails:   cancelRequest.Details[0:sizeLimitError],\n\t\t\tIdentity:  cancelRequest.Identity,\n\t\t}\n\t\terr = wh.GetHistoryClient().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tFailedRequest: failRequest,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t} else {\n\t\treq := &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: token,\n\t\t\tDetails:   cancelRequest.Details,\n\t\t\tIdentity:  cancelRequest.Identity,\n\t\t}\n\n\t\terr = wh.GetHistoryClient().RespondActivityTaskCanceled(ctx, &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\tDomainUUID:    taskToken.DomainID,\n\t\t\tCancelRequest: req,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// RespondDecisionTaskCompleted - response to a decision task\nfunc (wh *WorkflowHandler) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\tcompleteRequest *types.RespondDecisionTaskCompletedRequest,\n) (resp *types.RespondDecisionTaskCompletedResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif completeRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tif completeRequest.TaskToken == nil {\n\t\treturn nil, validate.ErrTaskTokenNotSet\n\t}\n\ttaskToken, err := wh.tokenSerializer.Deserialize(completeRequest.TaskToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif taskToken.DomainID == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondDecisionTaskCompletedScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tcompleteRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn nil, validate.ErrIdentityTooLong\n\t}\n\n\tif err := common.CheckDecisionResultLimit(\n\t\tlen(completeRequest.Decisions),\n\t\twh.config.DecisionResultCountLimit(domainName),\n\t\tscope); err != nil {\n\t\treturn nil, err\n\t}\n\n\thistResp, err := wh.GetHistoryClient().RespondDecisionTaskCompleted(ctx, &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID:      taskToken.DomainID,\n\t\tCompleteRequest: completeRequest},\n\t)\n\tif err != nil {\n\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t}\n\n\tcompletedResp := &types.RespondDecisionTaskCompletedResponse{}\n\tcompletedResp.ActivitiesToDispatchLocally = histResp.ActivitiesToDispatchLocally\n\tif completeRequest.GetReturnNewDecisionTask() && histResp != nil && histResp.StartedResponse != nil {\n\t\ttaskToken := &common.TaskToken{\n\t\t\tDomainID:        taskToken.DomainID,\n\t\t\tWorkflowID:      taskToken.WorkflowID,\n\t\t\tRunID:           taskToken.RunID,\n\t\t\tScheduleID:      histResp.StartedResponse.GetScheduledEventID(),\n\t\t\tScheduleAttempt: histResp.StartedResponse.GetAttempt(),\n\t\t}\n\t\ttoken, _ := wh.tokenSerializer.Serialize(taskToken)\n\t\tworkflowExecution := &types.WorkflowExecution{\n\t\t\tWorkflowID: taskToken.WorkflowID,\n\t\t\tRunID:      taskToken.RunID,\n\t\t}\n\t\tmatchingResp := common.CreateMatchingPollForDecisionTaskResponse(histResp.StartedResponse, workflowExecution, token)\n\n\t\tnewDecisionTask, err := wh.createPollForDecisionTaskResponse(ctx, scope, taskToken.DomainID, matchingResp, matchingResp.GetBranchToken())\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tcompletedResp.DecisionTask = newDecisionTask\n\t}\n\n\treturn completedResp, nil\n}\n\n// RespondDecisionTaskFailed - failed response to a decision task\nfunc (wh *WorkflowHandler) RespondDecisionTaskFailed(\n\tctx context.Context,\n\tfailedRequest *types.RespondDecisionTaskFailedRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif failedRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tif failedRequest.TaskToken == nil {\n\t\treturn validate.ErrTaskTokenNotSet\n\t}\n\ttaskToken, err := wh.tokenSerializer.Deserialize(failedRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif taskToken.DomainID == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(taskToken.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondDecisionTaskFailedScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tfailedRequest.GetIdentity(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.IdentityMaxLength(domainName),\n\t\tmetrics.CadenceErrIdentityExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeIdentity) {\n\t\treturn validate.ErrIdentityTooLong\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(failedRequest.Details),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\ttaskToken.DomainID,\n\t\tdomainName,\n\t\ttaskToken.WorkflowID,\n\t\ttaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondDecisionTaskFailed\"),\n\t); err != nil {\n\t\t// details exceed, we would just truncate the size for decision task failed as the details is not used anywhere by client code\n\t\tfailedRequest.Details = failedRequest.Details[0:sizeLimitError]\n\t}\n\n\terr = wh.GetHistoryClient().RespondDecisionTaskFailed(ctx, &types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID:    taskToken.DomainID,\n\t\tFailedRequest: failedRequest,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\treturn nil\n}\n\n// RespondQueryTaskCompleted - response to a query task\nfunc (wh *WorkflowHandler) RespondQueryTaskCompleted(\n\tctx context.Context,\n\tcompleteRequest *types.RespondQueryTaskCompletedRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif completeRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tif completeRequest.TaskToken == nil {\n\t\treturn validate.ErrTaskTokenNotSet\n\t}\n\tqueryTaskToken, err := wh.tokenSerializer.DeserializeQueryTaskToken(completeRequest.TaskToken)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif queryTaskToken.DomainID == \"\" || queryTaskToken.TaskList == \"\" || queryTaskToken.TaskID == \"\" {\n\t\treturn validate.ErrInvalidTaskToken\n\t}\n\n\tdomainName, err := wh.GetDomainCache().GetDomainName(queryTaskToken.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdw := domainWrapper{\n\t\tdomain: domainName,\n\t}\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendRespondQueryTaskCompletedScope, dw, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(completeRequest.GetQueryResult()),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tqueryTaskToken.DomainID,\n\t\tdomainName,\n\t\tqueryTaskToken.WorkflowID,\n\t\tqueryTaskToken.RunID,\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"RespondQueryTaskCompleted\"),\n\t); err != nil {\n\t\tcompleteRequest = &types.RespondQueryTaskCompletedRequest{\n\t\t\tTaskToken:     completeRequest.TaskToken,\n\t\t\tCompletedType: types.QueryTaskCompletedTypeFailed.Ptr(),\n\t\t\tQueryResult:   nil,\n\t\t\tErrorMessage:  err.Error(),\n\t\t}\n\t}\n\n\tcall := yarpc.CallFromContext(ctx)\n\n\tcompleteRequest.WorkerVersionInfo = &types.WorkerVersionInfo{\n\t\tImpl:           call.Header(common.ClientImplHeaderName),\n\t\tFeatureVersion: call.Header(common.FeatureVersionHeaderName),\n\t}\n\tmatchingRequest := &types.MatchingRespondQueryTaskCompletedRequest{\n\t\tDomainUUID:       queryTaskToken.DomainID,\n\t\tTaskList:         &types.TaskList{Name: queryTaskToken.TaskList},\n\t\tTaskID:           queryTaskToken.TaskID,\n\t\tCompletedRequest: completeRequest,\n\t}\n\n\terr = wh.GetMatchingClient().RespondQueryTaskCompleted(ctx, matchingRequest)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (wh *WorkflowHandler) StartWorkflowExecutionAsync(\n\tctx context.Context,\n\tstartRequest *types.StartWorkflowExecutionAsyncRequest,\n) (resp *types.StartWorkflowExecutionAsyncResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendStartWorkflowExecutionAsyncScope, startRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\t// validate request before pushing to queue\n\terr := wh.validateStartWorkflowExecutionRequest(ctx, startRequest.StartWorkflowExecutionRequest, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tproducer, err := wh.producerManager.GetProducerByDomain(startRequest.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Serialize the message to be sent to the queue.\n\t// Avoid JSON because json encoding of requests excludes PII fields such as input. JSON encoded request are logged by acccess controlled api layer for audit purposes.\n\tpayload, err := wh.thriftrwEncoder.Encode(thrift.FromStartWorkflowExecutionAsyncRequest(startRequest))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to encode StartWorkflowExecutionAsyncRequest: %v\", err)\n\t}\n\tscope.RecordTimer(metrics.AsyncRequestPayloadSize, time.Duration(len(payload)))\n\n\t// propagate the headers from the context to the message\n\theader := &shared.Header{\n\t\tFields: make(map[string][]byte),\n\t}\n\tfor k, v := range yarpc.CallFromContext(ctx).OriginalHeaders() {\n\t\theader.Fields[k] = []byte(v)\n\t}\n\tmessageType := sqlblobs.AsyncRequestTypeStartWorkflowExecutionAsyncRequest\n\tmessage := &sqlblobs.AsyncRequestMessage{\n\t\tPartitionKey: common.StringPtr(startRequest.GetWorkflowID()),\n\t\tType:         &messageType,\n\t\tHeader:       header,\n\t\tEncoding:     common.StringPtr(string(constants.EncodingTypeThriftRW)),\n\t\tPayload:      payload,\n\t}\n\terr = producer.Publish(ctx, message)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.StartWorkflowExecutionAsyncResponse{}, nil\n}\n\n// StartWorkflowExecution - Creates a new workflow execution\nfunc (wh *WorkflowHandler) StartWorkflowExecution(\n\tctx context.Context,\n\tstartRequest *types.StartWorkflowExecutionRequest,\n) (resp *types.StartWorkflowExecutionResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendStartWorkflowExecutionScope, startRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\terr := wh.validateStartWorkflowExecutionRequest(ctx, startRequest, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainName := startRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\thistoryRequest, err := common.CreateHistoryStartWorkflowRequest(\n\t\tdomainID, startRequest, time.Now(), wh.getPartitionConfig(ctx, domainName))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// for debugging jitter workflow\n\t// will be removed later\n\tjitterStartSeconds := startRequest.GetJitterStartSeconds()\n\tif startRequest.GetDomain() == \"cadence-canary\" && jitterStartSeconds > 0 {\n\t\twh.GetLogger().Debug(\"Start workflow execution request domainID\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(startRequest.WorkflowID),\n\t\t\ttag.Dynamic(\"JitterStartSeconds\", jitterStartSeconds),\n\t\t\ttag.Dynamic(\"firstDecisionTaskBackoffSeconds\", historyRequest.GetFirstDecisionTaskBackoffSeconds()),\n\t\t)\n\t} else {\n\t\twh.GetLogger().Debug(\"Start workflow execution request domainID\", tag.WorkflowDomainID(domainID))\n\t}\n\n\tresp, err = wh.GetHistoryClient().StartWorkflowExecution(ctx, historyRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (wh *WorkflowHandler) validateStartWorkflowExecutionRequest(ctx context.Context, startRequest *types.StartWorkflowExecutionRequest, scope metrics.Scope) error {\n\tif startRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tdomainName := startRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif startRequest.GetWorkflowID() == \"\" {\n\t\treturn validate.ErrWorkflowIDNotSet\n\t}\n\tif _, err := uuid.Parse(startRequest.RequestID); err != nil {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"requestId %q is not a valid UUID\", startRequest.RequestID)}\n\t}\n\tif startRequest.WorkflowType == nil || startRequest.WorkflowType.GetName() == \"\" {\n\t\treturn validate.ErrWorkflowTypeNotSet\n\t}\n\tidLengthWarnLimit := wh.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tdomainName,\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.DomainNameMaxLength(domainName),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeDomainName) {\n\t\treturn validate.ErrDomainTooLong\n\t}\n\tif !common.IsValidIDLength(\n\t\tstartRequest.GetWorkflowID(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.WorkflowIDMaxLength(domainName),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn validate.ErrWorkflowIDTooLong\n\t}\n\tif err := common.ValidateRetryPolicy(startRequest.RetryPolicy); err != nil {\n\t\treturn err\n\t}\n\twh.GetLogger().Debug(\n\t\t\"Received StartWorkflowExecution. WorkflowID\",\n\t\ttag.WorkflowID(startRequest.GetWorkflowID()))\n\tif !common.IsValidIDLength(\n\t\tstartRequest.WorkflowType.GetName(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.WorkflowTypeMaxLength(domainName),\n\t\tmetrics.CadenceErrWorkflowTypeExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeWorkflowType) {\n\t\treturn validate.ErrWorkflowTypeTooLong\n\t}\n\tif err := wh.validateTaskList(startRequest.TaskList, scope, domainName); err != nil {\n\t\treturn err\n\t}\n\tif startRequest.GetExecutionStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn validate.ErrInvalidExecutionStartToCloseTimeoutSeconds\n\t}\n\tif startRequest.GetTaskStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn validate.ErrInvalidTaskStartToCloseTimeoutSeconds\n\t}\n\tif startRequest.GetDelayStartSeconds() < 0 {\n\t\treturn validate.ErrInvalidDelayStartSeconds\n\t}\n\tif startRequest.GetJitterStartSeconds() < 0 {\n\t\treturn validate.ErrInvalidJitterStartSeconds\n\t}\n\tjitter := startRequest.GetJitterStartSeconds()\n\tcron := startRequest.GetCronSchedule()\n\tif cron != \"\" {\n\t\tif _, err := backoff.ValidateSchedule(startRequest.GetCronSchedule()); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tif jitter > 0 && cron != \"\" {\n\t\t// Calculate the cron duration and ensure that jitter is not greater than the cron duration,\n\t\t// because that would be confusing to users.\n\n\t\t// Request using start/end time zero value, which will get us an exact answer (i.e. its not in the\n\t\t// middle of a minute)\n\t\tbackoffSeconds, err := backoff.GetBackoffForNextScheduleInSeconds(cron, time.Time{}, time.Time{}, jitter, startRequest.GetCronOverlapPolicy())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif jitter > backoffSeconds {\n\t\t\treturn validate.ErrInvalidJitterStartSeconds2\n\t\t}\n\t}\n\tif !common.IsValidIDLength(\n\t\tstartRequest.GetRequestID(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.RequestIDMaxLength(domainName),\n\t\tmetrics.CadenceErrRequestIDExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeRequestID) {\n\t\treturn validate.ErrRequestIDTooLong\n\t}\n\tif err := wh.searchAttributesValidator.ValidateSearchAttributes(startRequest.SearchAttributes, domainName); err != nil {\n\t\treturn err\n\t}\n\twh.GetLogger().Debug(\"Start workflow execution request domain\", tag.WorkflowDomainName(domainName))\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\tactualSize := len(startRequest.Input)\n\tif startRequest.Memo != nil {\n\t\tactualSize += common.GetSizeOfMapStringToByteArray(startRequest.Memo.GetFields())\n\t}\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tactualSize,\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tdomainID,\n\t\tdomainName,\n\t\tstartRequest.GetWorkflowID(),\n\t\t\"\",\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"StartWorkflowExecution\"),\n\t); err != nil {\n\t\treturn err\n\t}\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.isIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn &types.BadRequestError{fmt.Sprintf(\"Domain %s is drained from isolation group %s.\", domainName, isolationGroup)}\n\t}\n\treturn nil\n}\n\n// GetWorkflowExecutionHistory - retrieves the history of workflow execution\nfunc (wh *WorkflowHandler) GetWorkflowExecutionHistory(\n\tctx context.Context,\n\tgetRequest *types.GetWorkflowExecutionHistoryRequest,\n) (resp *types.GetWorkflowExecutionHistoryResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif getRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := getRequest.GetDomain()\n\twfExecution := getRequest.GetWorkflowExecution()\n\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif getRequest.GetMaximumPageSize() <= 0 {\n\t\tgetRequest.MaximumPageSize = int32(wh.config.HistoryMaxPageSize(getRequest.GetDomain()))\n\t}\n\t// force limit page size if exceed\n\tif getRequest.GetMaximumPageSize() > constants.GetHistoryMaxPageSize {\n\t\twh.GetThrottledLogger().Warn(\"GetHistory page size is larger than threshold\",\n\t\t\ttag.WorkflowID(getRequest.Execution.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(getRequest.Execution.GetRunID()),\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowSize(int64(getRequest.GetMaximumPageSize())))\n\t\tgetRequest.MaximumPageSize = constants.GetHistoryMaxPageSize\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendGetWorkflowExecutionHistoryScope, getRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif !getRequest.GetSkipArchival() {\n\t\tenableArchivalRead := wh.GetArchivalMetadata().GetHistoryConfig().ReadEnabled()\n\t\thistoryArchived := wh.historyArchived(ctx, getRequest, domainID)\n\t\tif enableArchivalRead && historyArchived {\n\t\t\treturn wh.getArchivedHistory(ctx, getRequest, domainID)\n\t\t}\n\t}\n\n\t// this function return the following 8 things,\n\t// 1. branch token\n\t// 2. the workflow run ID\n\t// 3. the last first event ID (the event ID of the last batch of events in the history)\n\t// 4. the next event ID\n\t// 5. whether the workflow is closed\n\t// 6. The version history\n\t// 7. the workflow close status\n\t// 8. error if any\n\tqueryHistory := func(\n\t\tdomainUUID string,\n\t\texecution *types.WorkflowExecution,\n\t\texpectedNextEventID int64,\n\t\tcurrentBranchToken []byte,\n\t\tversionHistoryItem *persistence.VersionHistoryItem,\n\t) ([]byte, string, int64, int64, bool, *types.VersionHistoryItem, string, error) {\n\n\t\tresponse, err := wh.GetHistoryClient().PollMutableState(ctx, &types.PollMutableStateRequest{\n\t\t\tDomainUUID:          domainUUID,\n\t\t\tExecution:           execution,\n\t\t\tExpectedNextEventID: expectedNextEventID,\n\t\t\tCurrentBranchToken:  currentBranchToken,\n\t\t\tVersionHistoryItem:  versionHistoryItem.ToInternalType(),\n\t\t})\n\n\t\tif err != nil {\n\t\t\treturn nil, \"\", 0, 0, false, nil, \"\", err\n\t\t}\n\n\t\tisWorkflowRunning := response.GetWorkflowCloseState() == persistence.WorkflowCloseStatusNone\n\t\tcurrentVersionHistory, err := persistence.NewVersionHistoriesFromInternalType(response.VersionHistories).GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\twh.GetLogger().Error(\"Failed to get current version history\", tag.Dynamic(\"version-histories\", response.VersionHistories))\n\t\t\treturn nil, \"\", 0, 0, false, nil, \"\", fmt.Errorf(\"failed to get the current version from the response from history: %w\", err)\n\t\t}\n\n\t\twfCloseStatus := \"Running\"\n\t\tif !isWorkflowRunning {\n\t\t\twfCloseStatus = persistence.ToInternalWorkflowExecutionCloseStatus(int(response.GetWorkflowCloseState())).String()\n\t\t}\n\n\t\tlastVersionHistoryItem, err := currentVersionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn nil, \"\", 0, 0, false, nil, \"\", err\n\t\t}\n\n\t\treturn response.CurrentBranchToken,\n\t\t\tresponse.Execution.GetRunID(),\n\t\t\tresponse.GetLastFirstEventID(),\n\t\t\tresponse.GetNextEventID(),\n\t\t\tisWorkflowRunning,\n\t\t\tlastVersionHistoryItem.ToInternalType(),\n\t\t\twfCloseStatus,\n\t\t\tnil\n\t}\n\n\tisLongPoll := getRequest.GetWaitForNewEvent()\n\tisCloseEventOnly := getRequest.GetHistoryEventFilterType() == types.HistoryEventFilterTypeCloseEvent\n\texecution := getRequest.Execution\n\ttoken := &getHistoryContinuationToken{}\n\n\tvar runID string\n\tlastFirstEventID := constants.FirstEventID\n\tvar nextEventID int64\n\tvar isWorkflowRunning bool\n\tvar workflowCloseStatus string\n\tvar workflowCloseTime *time.Time\n\n\t// process the token for paging\n\tqueryNextEventID := constants.EndEventID\n\tif getRequest.NextPageToken != nil {\n\t\ttoken, err = deserializeHistoryToken(getRequest.NextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, validate.ErrInvalidNextPageToken\n\t\t}\n\t\tif execution.RunID != \"\" && execution.GetRunID() != token.RunID {\n\t\t\treturn nil, validate.ErrNextPageTokenRunIDMismatch\n\t\t}\n\n\t\texecution.RunID = token.RunID\n\n\t\t// we need to update the current next event ID and whether workflow is running\n\t\tif len(token.PersistenceToken) == 0 && isLongPoll && token.IsWorkflowRunning {\n\t\t\tlogger := wh.GetLogger().WithTags(\n\t\t\t\ttag.WorkflowDomainName(getRequest.GetDomain()),\n\t\t\t\ttag.WorkflowID(getRequest.Execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(getRequest.Execution.GetRunID()),\n\t\t\t)\n\t\t\t// TODO: for now we only log the invalid timeout (this is done inside the helper function) in case\n\t\t\t// this change breaks existing customers. Once we are sure no one is calling this API with very short timeout\n\t\t\t// we can return the error.\n\t\t\t_ = common.ValidateLongPollContextTimeout(ctx, \"GetWorkflowExecutionHistory\", logger)\n\n\t\t\tif !isCloseEventOnly {\n\t\t\t\tqueryNextEventID = token.NextEventID\n\t\t\t}\n\n\t\t\tvh := persistence.NewVersionHistoryItemFromInternalType(token.VersionHistoryItem)\n\t\t\ttoken.BranchToken, _, lastFirstEventID, nextEventID, isWorkflowRunning, token.VersionHistoryItem, workflowCloseStatus, err =\n\t\t\t\tqueryHistory(domainID, execution, queryNextEventID, token.BranchToken, vh)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\ttoken.FirstEventID = token.NextEventID\n\t\t\ttoken.NextEventID = nextEventID\n\t\t\ttoken.IsWorkflowRunning = isWorkflowRunning\n\t\t}\n\t} else {\n\t\tif !isCloseEventOnly {\n\t\t\tqueryNextEventID = constants.FirstEventID\n\t\t}\n\t\ttoken.BranchToken, runID, lastFirstEventID, nextEventID, isWorkflowRunning, token.VersionHistoryItem, workflowCloseStatus, err =\n\t\t\tqueryHistory(domainID, execution, queryNextEventID, nil, nil)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\texecution.RunID = runID\n\n\t\ttoken.RunID = runID\n\t\ttoken.FirstEventID = constants.FirstEventID\n\t\ttoken.NextEventID = nextEventID\n\t\ttoken.IsWorkflowRunning = isWorkflowRunning\n\t\ttoken.PersistenceToken = nil\n\t}\n\n\tcall := yarpc.CallFromContext(ctx)\n\tclientFeatureVersion := call.Header(common.FeatureVersionHeaderName)\n\tclientImpl := call.Header(common.ClientImplHeaderName)\n\tsupportsRawHistoryQuery := wh.versionChecker.SupportsRawHistoryQuery(clientImpl, clientFeatureVersion) == nil\n\tisRawHistoryEnabled := wh.config.SendRawWorkflowHistory(domainName) && supportsRawHistoryQuery\n\n\thistory := &types.History{}\n\thistory.Events = []*types.HistoryEvent{}\n\tvar historyBlob []*types.DataBlob\n\n\t// helper function to just getHistory\n\tgetHistory := func(firstEventID, nextEventID int64, nextPageToken []byte) error {\n\t\tif isRawHistoryEnabled {\n\t\t\thistoryBlob, token.PersistenceToken, err = wh.getRawHistory(\n\t\t\t\tctx,\n\t\t\t\tscope,\n\t\t\t\tdomainID,\n\t\t\t\tdomainName,\n\t\t\t\t*execution,\n\t\t\t\tfirstEventID,\n\t\t\t\tnextEventID,\n\t\t\t\tgetRequest.GetMaximumPageSize(),\n\t\t\t\tnextPageToken,\n\t\t\t\ttoken.TransientDecision,\n\t\t\t\ttoken.BranchToken,\n\t\t\t)\n\t\t} else {\n\t\t\thistory, token.PersistenceToken, err = wh.getHistory(\n\t\t\t\tctx,\n\t\t\t\tscope,\n\t\t\t\tdomainID,\n\t\t\t\tdomainName,\n\t\t\t\t*execution,\n\t\t\t\tfirstEventID,\n\t\t\t\tnextEventID,\n\t\t\t\tgetRequest.GetMaximumPageSize(),\n\t\t\t\tnextPageToken,\n\t\t\t\ttoken.TransientDecision,\n\t\t\t\ttoken.BranchToken,\n\t\t\t)\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\tif isCloseEventOnly {\n\t\tif !isWorkflowRunning {\n\t\t\tif err := getHistory(lastFirstEventID, nextEventID, nil); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tif isRawHistoryEnabled {\n\t\t\t\t// since getHistory func will not return empty history, so the below is safe\n\t\t\t\thistoryBlob = historyBlob[len(historyBlob)-1:]\n\t\t\t} else {\n\t\t\t\t// since getHistory func will not return empty history, so the below is safe\n\t\t\t\thistory.Events = history.Events[len(history.Events)-1:]\n\t\t\t}\n\t\t\ttoken = nil\n\t\t} else if isLongPoll {\n\t\t\t// set the persistence token to be nil so next time we will query history for updates\n\t\t\ttoken.PersistenceToken = nil\n\t\t} else {\n\t\t\ttoken = nil\n\t\t}\n\t} else {\n\t\t// return all events\n\t\tif token.FirstEventID >= token.NextEventID {\n\t\t\t// currently there is no new event\n\t\t\thistory.Events = []*types.HistoryEvent{}\n\t\t\tif !isWorkflowRunning {\n\t\t\t\ttoken = nil\n\t\t\t}\n\t\t} else {\n\t\t\tif err := getHistory(token.FirstEventID, token.NextEventID, token.PersistenceToken); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\t// here, for long pull on history events, we need to intercept the paging token from cassandra\n\t\t\t// and do something clever\n\t\t\tif len(token.PersistenceToken) == 0 && (!token.IsWorkflowRunning || !isLongPoll) {\n\t\t\t\t// meaning, there is no more history to be returned\n\t\t\t\ttoken = nil\n\t\t\t}\n\t\t}\n\t}\n\n\tnextToken, err := serializeHistoryToken(token)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Extract close time from history events for closed workflows\n\tif !isWorkflowRunning && len(history.Events) > 0 {\n\t\t// Get the last event (close event) timestamp\n\t\tlastEvent := history.Events[len(history.Events)-1]\n\t\tif lastEvent.Timestamp != nil {\n\t\t\tt := time.Unix(0, *lastEvent.Timestamp)\n\t\t\tworkflowCloseTime = &t\n\t\t}\n\t}\n\n\twh.emitGetWorkflowExecutionHistoryMetrics(domainName, domainID, execution.GetWorkflowID(), workflowCloseStatus, workflowCloseTime)\n\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       history,\n\t\tRawHistory:    historyBlob,\n\t\tNextPageToken: nextToken,\n\t\tArchived:      false,\n\t}, nil\n}\n\n// SignalWorkflowExecution is used to send a signal event to running workflow execution.  This results in\n// WorkflowExecutionSignaled event recorded in the history and a decision task being created for the execution.\nfunc (wh *WorkflowHandler) SignalWorkflowExecution(\n\tctx context.Context,\n\tsignalRequest *types.SignalWorkflowExecutionRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif signalRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := signalRequest.GetDomain()\n\twfExecution := signalRequest.GetWorkflowExecution()\n\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn err\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendSignalWorkflowExecutionScope, signalRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tidLengthWarnLimit := wh.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tdomainName,\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.DomainNameMaxLength(domainName),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeDomainName) {\n\t\treturn validate.ErrDomainTooLong\n\t}\n\n\tif signalRequest.GetSignalName() == \"\" {\n\t\treturn validate.ErrSignalNameNotSet\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalRequest.GetSignalName(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.SignalNameMaxLength(domainName),\n\t\tmetrics.CadenceErrSignalNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeSignalName) {\n\t\treturn validate.ErrSignalNameTooLong\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalRequest.GetRequestID(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.RequestIDMaxLength(domainName),\n\t\tmetrics.CadenceErrRequestIDExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeRequestID) {\n\t\treturn validate.ErrRequestIDTooLong\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(signalRequest.Input),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tdomainID,\n\t\tdomainName,\n\t\tsignalRequest.GetWorkflowExecution().GetWorkflowID(),\n\t\tsignalRequest.GetWorkflowExecution().GetRunID(),\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"SignalWorkflowExecution\"),\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.isIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn &types.BadRequestError{fmt.Sprintf(\"Domain %s is drained from isolation group %s.\", domainName, isolationGroup)}\n\t}\n\n\terr = wh.GetHistoryClient().SignalWorkflowExecution(ctx, &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID:    domainID,\n\t\tSignalRequest: signalRequest,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\n\treturn nil\n}\n\nfunc (wh *WorkflowHandler) SignalWithStartWorkflowExecutionAsync(\n\tctx context.Context,\n\tsignalWithStartRequest *types.SignalWithStartWorkflowExecutionAsyncRequest,\n) (resp *types.SignalWithStartWorkflowExecutionAsyncResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendSignalWithStartWorkflowExecutionAsyncScope, signalWithStartRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\t// validate request before pushing to queue\n\terr := wh.validateSignalWithStartWorkflowExecutionRequest(ctx, signalWithStartRequest.SignalWithStartWorkflowExecutionRequest, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tproducer, err := wh.producerManager.GetProducerByDomain(signalWithStartRequest.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Serialize the message to be sent to the queue.\n\t// Avoid JSON because json encoding of requests excludes PII fields such as input. JSON encoded request are logged by acccess controlled api layer for audit purposes.\n\tpayload, err := wh.thriftrwEncoder.Encode(thrift.FromSignalWithStartWorkflowExecutionAsyncRequest(signalWithStartRequest))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to encode SignalWithStartWorkflowExecutionAsyncRequest: %v\", err)\n\t}\n\tscope.RecordTimer(metrics.AsyncRequestPayloadSize, time.Duration(len(payload)))\n\n\t// propagate the headers from the context to the message\n\theader := &shared.Header{\n\t\tFields: map[string][]byte{},\n\t}\n\tfor k, v := range yarpc.CallFromContext(ctx).OriginalHeaders() {\n\t\theader.Fields[k] = []byte(v)\n\t}\n\tmessageType := sqlblobs.AsyncRequestTypeSignalWithStartWorkflowExecutionAsyncRequest\n\tmessage := &sqlblobs.AsyncRequestMessage{\n\t\tPartitionKey: common.StringPtr(signalWithStartRequest.GetWorkflowID()),\n\t\tType:         &messageType,\n\t\tHeader:       header,\n\t\tEncoding:     common.StringPtr(string(constants.EncodingTypeThriftRW)),\n\t\tPayload:      payload,\n\t}\n\terr = producer.Publish(ctx, message)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionAsyncResponse{}, nil\n}\n\n// SignalWithStartWorkflowExecution is used to ensure sending a signal event to a workflow execution.\n// If workflow is running, this results in WorkflowExecutionSignaled event recorded in the history\n// and a decision task being created for the execution.\n// If workflow is not running or not found, this results in WorkflowExecutionStarted and WorkflowExecutionSignaled\n// event recorded in history, and a decision task being created for the execution\nfunc (wh *WorkflowHandler) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\tsignalWithStartRequest *types.SignalWithStartWorkflowExecutionRequest,\n) (resp *types.StartWorkflowExecutionResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendSignalWithStartWorkflowExecutionScope, signalWithStartRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\terr := wh.validateSignalWithStartWorkflowExecutionRequest(ctx, signalWithStartRequest, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName := signalWithStartRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err = wh.GetHistoryClient().SignalWithStartWorkflowExecution(ctx, &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID:             domainID,\n\t\tSignalWithStartRequest: signalWithStartRequest,\n\t\tPartitionConfig:        wh.getPartitionConfig(ctx, domainName),\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp, nil\n}\n\nfunc (wh *WorkflowHandler) validateSignalWithStartWorkflowExecutionRequest(ctx context.Context, signalWithStartRequest *types.SignalWithStartWorkflowExecutionRequest, scope metrics.Scope) error {\n\tif signalWithStartRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := signalWithStartRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif signalWithStartRequest.GetWorkflowID() == \"\" {\n\t\treturn validate.ErrWorkflowIDNotSet\n\t}\n\n\tidLengthWarnLimit := wh.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tdomainName,\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.DomainNameMaxLength(domainName),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeDomainName) {\n\t\treturn validate.ErrDomainTooLong\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalWithStartRequest.GetWorkflowID(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.WorkflowIDMaxLength(domainName),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn validate.ErrWorkflowIDTooLong\n\t}\n\n\tif signalWithStartRequest.GetSignalName() == \"\" {\n\t\treturn validate.ErrSignalNameNotSet\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalWithStartRequest.GetSignalName(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.SignalNameMaxLength(domainName),\n\t\tmetrics.CadenceErrSignalNameExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeSignalName) {\n\t\treturn validate.ErrSignalNameTooLong\n\t}\n\n\tif signalWithStartRequest.WorkflowType == nil || signalWithStartRequest.WorkflowType.GetName() == \"\" {\n\t\treturn validate.ErrWorkflowTypeNotSet\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalWithStartRequest.WorkflowType.GetName(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.WorkflowTypeMaxLength(domainName),\n\t\tmetrics.CadenceErrWorkflowTypeExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeWorkflowType) {\n\t\treturn validate.ErrWorkflowTypeTooLong\n\t}\n\n\tif err := wh.validateTaskList(signalWithStartRequest.TaskList, scope, domainName); err != nil {\n\t\treturn err\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tsignalWithStartRequest.GetRequestID(),\n\t\tscope,\n\t\tidLengthWarnLimit,\n\t\twh.config.RequestIDMaxLength(domainName),\n\t\tmetrics.CadenceErrRequestIDExceededWarnLimit,\n\t\tdomainName,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeRequestID) {\n\t\treturn validate.ErrRequestIDTooLong\n\t}\n\n\tif signalWithStartRequest.GetExecutionStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn validate.ErrInvalidExecutionStartToCloseTimeoutSeconds\n\t}\n\n\tif signalWithStartRequest.GetTaskStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn validate.ErrInvalidTaskStartToCloseTimeoutSeconds\n\t}\n\n\tif err := common.ValidateRetryPolicy(signalWithStartRequest.RetryPolicy); err != nil {\n\t\treturn err\n\t}\n\n\tif signalWithStartRequest.GetCronSchedule() != \"\" {\n\t\tif _, err := backoff.ValidateSchedule(signalWithStartRequest.GetCronSchedule()); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif err := wh.searchAttributesValidator.ValidateSearchAttributes(signalWithStartRequest.SearchAttributes, domainName); err != nil {\n\t\treturn err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(signalWithStartRequest.SignalInput),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tdomainID,\n\t\tdomainName,\n\t\tsignalWithStartRequest.GetWorkflowID(),\n\t\t\"\",\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"SignalWithStartWorkflowExecution\"),\n\t); err != nil {\n\t\treturn err\n\t}\n\tactualSize := len(signalWithStartRequest.Input) + common.GetSizeOfMapStringToByteArray(signalWithStartRequest.Memo.GetFields())\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tactualSize,\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tdomainID,\n\t\tdomainName,\n\t\tsignalWithStartRequest.GetWorkflowID(),\n\t\t\"\",\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"SignalWithStartWorkflowExecution\"),\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.isIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn &types.BadRequestError{fmt.Sprintf(\"Domain %s is drained from isolation group %s.\", domainName, isolationGroup)}\n\t}\n\treturn nil\n}\n\n// TerminateWorkflowExecution terminates an existing workflow execution by recording WorkflowExecutionTerminated event\n// in the history and immediately terminating the execution instance.\nfunc (wh *WorkflowHandler) TerminateWorkflowExecution(\n\tctx context.Context,\n\tterminateRequest *types.TerminateWorkflowExecutionRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif terminateRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := terminateRequest.GetDomain()\n\twfExecution := terminateRequest.GetWorkflowExecution()\n\tif terminateRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = wh.GetHistoryClient().TerminateWorkflowExecution(ctx, &types.HistoryTerminateWorkflowExecutionRequest{\n\t\tDomainUUID:       domainID,\n\t\tTerminateRequest: terminateRequest,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\n\treturn nil\n}\n\n// ResetWorkflowExecution reset an existing workflow execution to the nextFirstEventID\n// in the history and immediately terminating the current execution instance.\nfunc (wh *WorkflowHandler) ResetWorkflowExecution(\n\tctx context.Context,\n\tresetRequest *types.ResetWorkflowExecutionRequest,\n) (resp *types.ResetWorkflowExecutionResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif resetRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := resetRequest.GetDomain()\n\twfExecution := resetRequest.GetWorkflowExecution()\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(resetRequest.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, err = wh.GetHistoryClient().ResetWorkflowExecution(ctx, &types.HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID:   domainID,\n\t\tResetRequest: resetRequest,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp, nil\n}\n\n// RequestCancelWorkflowExecution - requests to cancel a workflow execution\nfunc (wh *WorkflowHandler) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\tcancelRequest *types.RequestCancelWorkflowExecutionRequest,\n) (retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\n\tif cancelRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\n\tdomainName := cancelRequest.GetDomain()\n\twfExecution := cancelRequest.GetWorkflowExecution()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(cancelRequest.GetDomain())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = wh.GetHistoryClient().RequestCancelWorkflowExecution(ctx, &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID:    domainID,\n\t\tCancelRequest: cancelRequest,\n\t})\n\tif err != nil {\n\t\treturn wh.normalizeVersionedErrors(ctx, err)\n\t}\n\n\treturn nil\n}\n\n// RestartWorkflowExecution - retrieves info for an existing workflow then restarts it\nfunc (wh *WorkflowHandler) RestartWorkflowExecution(ctx context.Context, request *types.RestartWorkflowExecutionRequest) (resp *types.RestartWorkflowExecutionResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif request == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := request.GetDomain()\n\twfExecution := request.GetWorkflowExecution()\n\n\tif request.GetDomain() == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tisolationGroup := wh.getIsolationGroup(ctx, domainName)\n\tif !wh.isIsolationGroupHealthy(ctx, domainName, isolationGroup) {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"Domain %s is drained from isolation group %s.\", domainName, isolationGroup)}\n\t}\n\n\thistory, err := wh.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wfExecution.WorkflowID,\n\t\t\tRunID:      wfExecution.RunID,\n\t\t},\n\t\tSkipArchival: true,\n\t})\n\tif err != nil {\n\t\treturn nil, validate.ErrHistoryNotFound\n\t}\n\tstartRequest := constructRestartWorkflowRequest(history.History.Events[0].WorkflowExecutionStartedEventAttributes,\n\t\tdomainName, request.Identity, wfExecution.WorkflowID)\n\treq, err := common.CreateHistoryStartWorkflowRequest(domainID, startRequest, time.Now(), wh.getPartitionConfig(ctx, domainName))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tstartResp, err := wh.GetHistoryClient().StartWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t}\n\tresp = &types.RestartWorkflowExecutionResponse{\n\t\tRunID: startResp.RunID,\n\t}\n\n\treturn resp, nil\n}\n\n// GetSearchAttributes return valid indexed keys\nfunc (wh *WorkflowHandler) GetSearchAttributes(ctx context.Context) (resp *types.GetSearchAttributesResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tkeys := wh.config.ValidSearchAttributes()\n\tresp = &types.GetSearchAttributesResponse{\n\t\tKeys: wh.convertIndexedKeyToThrift(keys),\n\t}\n\treturn resp, nil\n}\n\n// QueryWorkflow returns query result for a specified workflow execution\nfunc (wh *WorkflowHandler) QueryWorkflow(\n\tctx context.Context,\n\tqueryRequest *types.QueryWorkflowRequest,\n) (resp *types.QueryWorkflowResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif queryRequest == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := queryRequest.GetDomain()\n\twfExecution := queryRequest.GetExecution()\n\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif wh.config.DisallowQuery(domainName) {\n\t\treturn nil, validate.ErrQueryDisallowedForDomain\n\t}\n\n\tif queryRequest.Query == nil {\n\t\treturn nil, validate.ErrQueryNotSet\n\t}\n\n\tif queryRequest.Query.GetQueryType() == \"\" {\n\t\treturn nil, validate.ErrQueryTypeNotSet\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domainName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tsizeLimitError := wh.config.BlobSizeLimitError(domainName)\n\tsizeLimitWarn := wh.config.BlobSizeLimitWarn(domainName)\n\n\tscope := getMetricsScopeWithDomain(metrics.FrontendQueryWorkflowScope, queryRequest, wh.GetMetricsClient()).Tagged(metrics.GetContextTags(ctx)...)\n\tif err := common.CheckEventBlobSizeLimit(\n\t\tlen(queryRequest.GetQuery().GetQueryArgs()),\n\t\tsizeLimitWarn,\n\t\tsizeLimitError,\n\t\tdomainID,\n\t\tdomainName,\n\t\tqueryRequest.GetExecution().GetWorkflowID(),\n\t\tqueryRequest.GetExecution().GetRunID(),\n\t\tscope,\n\t\twh.GetLogger(),\n\t\ttag.BlobSizeViolationOperation(\"QueryWorkflow\")); err != nil {\n\t\treturn nil, err\n\t}\n\n\treq := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: domainID,\n\t\tRequest:    queryRequest,\n\t}\n\thResponse, err := wh.GetHistoryClient().QueryWorkflow(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn hResponse.GetResponse(), nil\n}\n\n// DescribeWorkflowExecution returns information about the specified workflow execution.\nfunc (wh *WorkflowHandler) DescribeWorkflowExecution(\n\tctx context.Context,\n\trequest *types.DescribeWorkflowExecutionRequest,\n) (resp *types.DescribeWorkflowExecutionResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\n\tif request == nil {\n\t\treturn nil, validate.ErrRequestNotSet\n\t}\n\n\tdomainName := request.GetDomain()\n\twfExecution := request.GetExecution()\n\tif domainName == \"\" {\n\t\treturn nil, validate.ErrDomainNotSet\n\t}\n\n\tif err := validate.CheckExecution(wfExecution); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID, err := wh.GetDomainCache().GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresponse, err := wh.GetHistoryClient().DescribeWorkflowExecution(ctx, &types.HistoryDescribeWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tRequest:    request,\n\t})\n\n\twh.emitDescribeWorkflowExecutionMetrics(domainName, response, err)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc (wh *WorkflowHandler) getRawHistory(\n\tctx context.Context,\n\tscope metrics.Scope,\n\tdomainID string,\n\tdomainName string,\n\texecution types.WorkflowExecution,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tpageSize int32,\n\tnextPageToken []byte,\n\ttransientDecision *types.TransientDecisionInfo,\n\tbranchToken []byte,\n) ([]*types.DataBlob, []byte, error) {\n\trawHistory := []*types.DataBlob{}\n\tshardID := common.WorkflowIDToHistoryShard(execution.WorkflowID, wh.config.NumHistoryShards)\n\n\tresp, err := wh.GetHistoryManager().ReadRawHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      int(pageSize),\n\t\tNextPageToken: nextPageToken,\n\t\tShardID:       common.IntPtr(shardID),\n\t\tDomainName:    domainName,\n\t})\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tvar encoding *types.EncodingType\n\tfor _, data := range resp.HistoryEventBlobs {\n\t\tswitch data.Encoding {\n\t\tcase constants.EncodingTypeJSON:\n\t\t\tencoding = types.EncodingTypeJSON.Ptr()\n\t\tcase constants.EncodingTypeThriftRW:\n\t\t\tencoding = types.EncodingTypeThriftRW.Ptr()\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"Invalid encoding type for raw history, encoding type: %s\", data.Encoding))\n\t\t}\n\t\trawHistory = append(rawHistory, &types.DataBlob{\n\t\t\tEncodingType: encoding,\n\t\t\tData:         data.Data,\n\t\t})\n\t}\n\n\tif len(resp.NextPageToken) == 0 && transientDecision != nil {\n\t\tif err := wh.validateTransientDecisionEvents(nextEventID, transientDecision); err != nil {\n\t\t\tscope.IncCounter(metrics.CadenceErrIncompleteHistoryCounter)\n\t\t\twh.GetLogger().Error(\"getHistory error\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(execution.GetRunID()),\n\t\t\t\ttag.Error(err))\n\t\t}\n\t\tblob, err := wh.GetPayloadSerializer().SerializeBatchEvents(\n\t\t\t[]*types.HistoryEvent{transientDecision.ScheduledEvent, transientDecision.StartedEvent}, constants.EncodingTypeThriftRW)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\trawHistory = append(rawHistory, &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\tData:         blob.Data,\n\t\t})\n\t}\n\n\treturn rawHistory, resp.NextPageToken, nil\n}\n\nfunc (wh *WorkflowHandler) getHistory(\n\tctx context.Context,\n\tscope metrics.Scope,\n\tdomainID string,\n\tdomainName string,\n\texecution types.WorkflowExecution,\n\tfirstEventID, nextEventID int64,\n\tpageSize int32,\n\tnextPageToken []byte,\n\ttransientDecision *types.TransientDecisionInfo,\n\tbranchToken []byte,\n) (*types.History, []byte, error) {\n\n\tvar size int\n\n\tisFirstPage := len(nextPageToken) == 0\n\tshardID := common.WorkflowIDToHistoryShard(execution.WorkflowID, wh.config.NumHistoryShards)\n\tvar err error\n\thistoryEvents, size, nextPageToken, err := persistenceutils.ReadFullPageV2Events(ctx, wh.GetHistoryManager(), &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      int(pageSize),\n\t\tNextPageToken: nextPageToken,\n\t\tShardID:       common.IntPtr(shardID),\n\t\tDomainName:    domainName,\n\t})\n\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tscope.RecordTimer(metrics.HistorySize, time.Duration(size))\n\n\tisLastPage := len(nextPageToken) == 0\n\tif err := verifyHistoryIsComplete(\n\t\thistoryEvents,\n\t\tfirstEventID,\n\t\tnextEventID-1,\n\t\tisFirstPage,\n\t\tisLastPage,\n\t\tint(pageSize)); err != nil {\n\t\tscope.IncCounter(metrics.CadenceErrIncompleteHistoryCounter)\n\t\twh.GetLogger().Error(\"getHistory: incomplete history\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(execution.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(execution.GetRunID()),\n\t\t\ttag.Error(err))\n\t\treturn nil, nil, err\n\t}\n\n\tif len(nextPageToken) == 0 && transientDecision != nil {\n\t\tif err := wh.validateTransientDecisionEvents(nextEventID, transientDecision); err != nil {\n\t\t\tscope.IncCounter(metrics.CadenceErrIncompleteHistoryCounter)\n\t\t\twh.GetLogger().Error(\"getHistory error\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(execution.GetRunID()),\n\t\t\t\ttag.Error(err))\n\t\t}\n\t\t// Append the transient decision events once we are done enumerating everything from the events table\n\t\thistoryEvents = append(historyEvents, transientDecision.ScheduledEvent, transientDecision.StartedEvent)\n\t}\n\n\texecutionHistory := &types.History{}\n\texecutionHistory.Events = historyEvents\n\treturn executionHistory, nextPageToken, nil\n}\n\nfunc (wh *WorkflowHandler) validateTransientDecisionEvents(\n\texpectedNextEventID int64,\n\tdecision *types.TransientDecisionInfo,\n) error {\n\n\tif decision.ScheduledEvent.ID == expectedNextEventID &&\n\t\tdecision.StartedEvent.ID == expectedNextEventID+1 {\n\t\treturn nil\n\t}\n\n\treturn fmt.Errorf(\n\t\t\"invalid transient decision: \"+\n\t\t\t\"expectedScheduledEventID=%v expectedStartedEventID=%v but have scheduledEventID=%v startedEventID=%v\",\n\t\texpectedNextEventID,\n\t\texpectedNextEventID+1,\n\t\tdecision.ScheduledEvent.ID,\n\t\tdecision.StartedEvent.ID)\n}\n\nfunc (wh *WorkflowHandler) validateTaskList(t *types.TaskList, scope metrics.Scope, domain string) error {\n\tif t == nil || t.GetName() == \"\" {\n\t\treturn validate.ErrTaskListNotSet\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tt.GetName(),\n\t\tscope,\n\t\twh.config.MaxIDLengthWarnLimit(),\n\t\twh.config.TaskListNameMaxLength(domain),\n\t\tmetrics.CadenceErrTaskListNameExceededWarnLimit,\n\t\tdomain,\n\t\twh.GetLogger(),\n\t\ttag.IDTypeTaskListName) {\n\t\treturn validate.ErrTaskListTooLong\n\t}\n\treturn nil\n}\n\nfunc (wh *WorkflowHandler) createPollForDecisionTaskResponse(\n\tctx context.Context,\n\tscope metrics.Scope,\n\tdomainID string,\n\tmatchingResp *types.MatchingPollForDecisionTaskResponse,\n\tbranchToken []byte,\n) (*types.PollForDecisionTaskResponse, error) {\n\n\tif matchingResp.WorkflowExecution == nil {\n\t\t// this will happen if there is no decision task to be send to worker / caller\n\t\treturn &types.PollForDecisionTaskResponse{\n\t\t\tAutoConfigHint: matchingResp.AutoConfigHint,\n\t\t}, nil\n\t}\n\n\tvar history *types.History\n\tvar continuation []byte\n\tvar err error\n\n\tif matchingResp.GetStickyExecutionEnabled() && matchingResp.Query != nil {\n\t\t// meaning sticky query, we should not return any events to worker\n\t\t// since query task only check the current status\n\t\thistory = &types.History{\n\t\t\tEvents: []*types.HistoryEvent{},\n\t\t}\n\t} else {\n\t\t// here we have 3 cases:\n\t\t// 1. sticky && non query task\n\t\t// 2. non sticky &&  non query task\n\t\t// 3. non sticky && query task\n\t\t// for 1, partial history have to be send back\n\t\t// for 2 and 3, full history have to be send back\n\n\t\tvar persistenceToken []byte\n\n\t\tfirstEventID := constants.FirstEventID\n\t\tnextEventID := matchingResp.GetNextEventID()\n\t\tif matchingResp.GetStickyExecutionEnabled() {\n\t\t\tfirstEventID = matchingResp.GetPreviousStartedEventID() + 1\n\t\t}\n\t\tdomainName, dErr := wh.GetDomainCache().GetDomainName(domainID)\n\t\tif dErr != nil {\n\t\t\treturn nil, dErr\n\t\t}\n\t\tscope = scope.Tagged(metrics.DomainTag(domainName))\n\t\thistory, persistenceToken, err = wh.getHistory(\n\t\t\tctx,\n\t\t\tscope,\n\t\t\tdomainID,\n\t\t\tdomainName,\n\t\t\t*matchingResp.WorkflowExecution,\n\t\t\tfirstEventID,\n\t\t\tnextEventID,\n\t\t\tint32(wh.config.HistoryMaxPageSize(domainName)),\n\t\t\tnil,\n\t\t\tmatchingResp.DecisionInfo,\n\t\t\tbranchToken,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif len(persistenceToken) != 0 {\n\t\t\tcontinuation, err = serializeHistoryToken(&getHistoryContinuationToken{\n\t\t\t\tRunID:             matchingResp.WorkflowExecution.GetRunID(),\n\t\t\t\tFirstEventID:      firstEventID,\n\t\t\t\tNextEventID:       nextEventID,\n\t\t\t\tPersistenceToken:  persistenceToken,\n\t\t\t\tTransientDecision: matchingResp.DecisionInfo,\n\t\t\t\tBranchToken:       branchToken,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t}\n\n\tresp := &types.PollForDecisionTaskResponse{\n\t\tTaskToken:                 matchingResp.TaskToken,\n\t\tWorkflowExecution:         matchingResp.WorkflowExecution,\n\t\tWorkflowType:              matchingResp.WorkflowType,\n\t\tPreviousStartedEventID:    matchingResp.PreviousStartedEventID,\n\t\tStartedEventID:            matchingResp.StartedEventID, // this field is not set for query tasks as there's no decision task started event\n\t\tQuery:                     matchingResp.Query,\n\t\tBacklogCountHint:          matchingResp.BacklogCountHint,\n\t\tAttempt:                   matchingResp.Attempt,\n\t\tHistory:                   history,\n\t\tNextPageToken:             continuation,\n\t\tWorkflowExecutionTaskList: matchingResp.WorkflowExecutionTaskList,\n\t\tScheduledTimestamp:        matchingResp.ScheduledTimestamp,\n\t\tStartedTimestamp:          matchingResp.StartedTimestamp,\n\t\tQueries:                   matchingResp.Queries,\n\t\tNextEventID:               matchingResp.NextEventID,\n\t\tTotalHistoryBytes:         matchingResp.TotalHistoryBytes,\n\t\tAutoConfigHint:            matchingResp.AutoConfigHint,\n\t}\n\n\treturn resp, nil\n}\n\nfunc verifyHistoryIsComplete(\n\tevents []*types.HistoryEvent,\n\texpectedFirstEventID int64,\n\texpectedLastEventID int64,\n\tisFirstPage bool,\n\tisLastPage bool,\n\tpageSize int,\n) error {\n\n\tnEvents := len(events)\n\tif nEvents == 0 {\n\t\tif isLastPage {\n\t\t\t// we seem to be returning a non-nil pageToken on the lastPage which\n\t\t\t// in turn cases the client to call getHistory again - only to find\n\t\t\t// there are no more events to consume - bail out if this is the case here\n\t\t\treturn nil\n\t\t}\n\t\treturn fmt.Errorf(\"invalid history: contains zero events\")\n\t}\n\n\tfirstEventID := events[0].ID\n\tlastEventID := events[nEvents-1].ID\n\n\tif !isFirstPage { // atleast one page of history has been read previously\n\t\tif firstEventID <= expectedFirstEventID {\n\t\t\t// not first page and no events have been read in the previous pages - not possible\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\n\t\t\t\t\t\"invalid history: expected first eventID to be > %v but got %v\", expectedFirstEventID, firstEventID),\n\t\t\t}\n\t\t}\n\t\texpectedFirstEventID = firstEventID\n\t}\n\n\tif !isLastPage {\n\t\t// estimate lastEventID based on pageSize. This is a lower bound\n\t\t// since the persistence layer counts \"batch of events\" as a single page\n\t\texpectedLastEventID = expectedFirstEventID + int64(pageSize) - 1\n\t}\n\n\tnExpectedEvents := expectedLastEventID - expectedFirstEventID + 1\n\n\tif firstEventID == expectedFirstEventID &&\n\t\t((isLastPage && lastEventID == expectedLastEventID && int64(nEvents) == nExpectedEvents) ||\n\t\t\t(!isLastPage && lastEventID >= expectedLastEventID && int64(nEvents) >= nExpectedEvents)) {\n\t\treturn nil\n\t}\n\n\treturn &types.InternalServiceError{\n\t\tMessage: fmt.Sprintf(\n\t\t\t\"incomplete history: \"+\n\t\t\t\t\"expected events [%v-%v] but got events [%v-%v] of length %v:\"+\n\t\t\t\t\"isFirstPage=%v,isLastPage=%v,pageSize=%v\",\n\t\t\texpectedFirstEventID,\n\t\t\texpectedLastEventID,\n\t\t\tfirstEventID,\n\t\t\tlastEventID,\n\t\t\tnEvents,\n\t\t\tisFirstPage,\n\t\t\tisLastPage,\n\t\t\tpageSize),\n\t}\n}\n\nfunc deserializeHistoryToken(bytes []byte) (*getHistoryContinuationToken, error) {\n\ttoken := &getHistoryContinuationToken{}\n\terr := json.Unmarshal(bytes, token)\n\treturn token, err\n}\n\nfunc serializeHistoryToken(token *getHistoryContinuationToken) ([]byte, error) {\n\tif token == nil {\n\t\treturn nil, nil\n\t}\n\n\tbytes, err := json.Marshal(token)\n\treturn bytes, err\n}\n\nfunc isFailoverRequest(updateRequest *types.UpdateDomainRequest) bool {\n\treturn updateRequest.ActiveClusterName != nil || updateRequest.ActiveClusters != nil\n}\n\nfunc isGraceFailoverRequest(updateRequest *types.UpdateDomainRequest) bool {\n\treturn updateRequest.FailoverTimeoutInSeconds != nil\n}\n\nfunc (wh *WorkflowHandler) checkOngoingFailover(\n\tctx context.Context,\n\tdomainName *string,\n) error {\n\n\tenabledClusters := wh.GetClusterMetadata().GetEnabledClusterInfo()\n\trespChan := make(chan *types.DescribeDomainResponse, len(enabledClusters))\n\n\tg := &errgroup.Group{}\n\tfor clusterName := range enabledClusters {\n\t\tfrontendClient, err := wh.GetRemoteFrontendClient(clusterName)\n\t\tif err != nil {\n\t\t\treturn &types.BadRequestError{Message: err.Error()}\n\t\t}\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), wh.GetLogger(), &e) }()\n\n\t\t\tresp, _ := frontendClient.DescribeDomain(ctx, &types.DescribeDomainRequest{Name: domainName})\n\t\t\trespChan <- resp\n\t\t\treturn nil\n\t\t})\n\t}\n\tg.Wait()\n\tclose(respChan)\n\n\tvar failoverVersion *int64\n\tfor resp := range respChan {\n\t\tif resp == nil {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"Failed to verify failover version from all clusters\",\n\t\t\t}\n\t\t}\n\t\tif failoverVersion == nil {\n\t\t\tfailoverVersion = &resp.FailoverVersion\n\t\t}\n\t\tif *failoverVersion != resp.GetFailoverVersion() {\n\t\t\treturn &types.BadRequestError{\n\t\t\t\tMessage: \"Concurrent failover is not allow.\",\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (wh *WorkflowHandler) historyArchived(ctx context.Context, request *types.GetWorkflowExecutionHistoryRequest, domainID string) bool {\n\tif request.GetWorkflowExecution() == nil || request.GetWorkflowExecution().GetRunID() == \"\" {\n\t\treturn false\n\t}\n\tgetMutableStateRequest := &types.GetMutableStateRequest{\n\t\tDomainUUID: domainID,\n\t\tExecution:  request.Execution,\n\t}\n\t_, err := wh.GetHistoryClient().GetMutableState(ctx, getMutableStateRequest)\n\tif err == nil {\n\t\treturn false\n\t}\n\tswitch err.(type) {\n\tcase *types.EntityNotExistsError:\n\t\t// the only case in which history is assumed to be archived is if getting mutable state returns entity not found error\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc (wh *WorkflowHandler) getArchivedHistory(\n\tctx context.Context,\n\trequest *types.GetWorkflowExecutionHistoryRequest,\n\tdomainID string,\n) (*types.GetWorkflowExecutionHistoryResponse, error) {\n\tentry, err := wh.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tURIString := entry.GetConfig().HistoryArchivalURI\n\tif URIString == \"\" {\n\t\t// if URI is empty, it means the domain has never enabled for archival.\n\t\t// the error is not \"workflow has passed retention period\", because\n\t\t// we have no way to tell if the requested workflow exists or not.\n\t\treturn nil, validate.ErrHistoryNotFound\n\t}\n\n\tURI, err := archiver.NewURI(URIString)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\thistoryArchiver, err := wh.GetArchiverProvider().GetHistoryArchiver(URI.Scheme(), service.Frontend)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, err := historyArchiver.Get(ctx, URI, &archiver.GetHistoryRequest{\n\t\tDomainID:      domainID,\n\t\tWorkflowID:    request.GetWorkflowExecution().GetWorkflowID(),\n\t\tRunID:         request.GetWorkflowExecution().GetRunID(),\n\t\tNextPageToken: request.GetNextPageToken(),\n\t\tPageSize:      int(request.GetMaximumPageSize()),\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\thistory := &types.History{}\n\tfor _, batch := range resp.HistoryBatches {\n\t\thistory.Events = append(history.Events, batch.Events...)\n\t}\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory:       history,\n\t\tNextPageToken: resp.NextPageToken,\n\t\tArchived:      true,\n\t}, nil\n}\n\nfunc (wh *WorkflowHandler) convertIndexedKeyToThrift(keys map[string]interface{}) map[string]types.IndexedValueType {\n\tconverted := make(map[string]types.IndexedValueType)\n\tfor k, v := range keys {\n\t\tconverted[k] = common.ConvertIndexedValueTypeToInternalType(v, wh.GetLogger())\n\t}\n\treturn converted\n}\n\nfunc (wh *WorkflowHandler) isListRequestPageSizeTooLarge(pageSize int32, domain string) bool {\n\treturn common.IsAdvancedVisibilityReadingEnabled(wh.config.ReadVisibilityStoreName(domain) != \"db\",\n\t\twh.config.IsAdvancedVisConfigExist) && pageSize > int32(wh.config.ESIndexMaxResultWindow())\n}\n\n// GetClusterInfo return information about cadence deployment\nfunc (wh *WorkflowHandler) GetClusterInfo(\n\tctx context.Context,\n) (resp *types.ClusterInfo, err error) {\n\treturn &types.ClusterInfo{\n\t\tSupportedClientVersions: &types.SupportedClientVersions{\n\t\t\tGoSdk:   client.SupportedGoSDKVersion,\n\t\t\tJavaSdk: client.SupportedJavaSDKVersion,\n\t\t},\n\t}, nil\n}\n\ntype domainWrapper struct {\n\tdomain string\n}\n\nfunc (d domainWrapper) GetDomain() string {\n\treturn d.domain\n}\n\nfunc (hs HealthStatus) String() string {\n\tswitch hs {\n\tcase HealthStatusOK:\n\t\treturn \"OK\"\n\tcase HealthStatusWarmingUp:\n\t\treturn \"WarmingUp\"\n\tcase HealthStatusShuttingDown:\n\t\treturn \"ShuttingDown\"\n\tdefault:\n\t\treturn \"unknown\"\n\t}\n}\n\nfunc (wh *WorkflowHandler) emitDescribeWorkflowExecutionMetrics(domain string, response *types.DescribeWorkflowExecutionResponse, err error) {\n\tscope := wh.GetMetricsClient().Scope(metrics.FrontendDescribeWorkflowExecutionStatusScope, metrics.DomainTag(domain))\n\n\tif err != nil || response == nil {\n\t\tscope.IncCounter(metrics.DescribeWorkflowStatusError)\n\t\treturn\n\t}\n\n\tstatus := \"unknown\"\n\tif response.WorkflowExecutionInfo != nil && response.WorkflowExecutionInfo.CloseStatus != nil {\n\t\tstatus = response.WorkflowExecutionInfo.CloseStatus.String()\n\t}\n\n\tscope = scope.Tagged(metrics.WorkflowCloseStatusTag(status))\n\tscope.IncCounter(metrics.DescribeWorkflowStatusCount)\n}\n\nfunc (wh *WorkflowHandler) emitGetWorkflowExecutionHistoryMetrics(domainName, domainID, workflowID, workflowCloseStatus string, closeTime *time.Time) {\n\tdomainEntry, err := wh.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\twh.GetLogger().Warn(\"Failed to get domain entry for metrics\", tag.WorkflowDomainName(domainName), tag.Error(err))\n\t\treturn\n\t}\n\n\tretentionDays := domainEntry.GetRetentionDays(workflowID)\n\n\tscope := wh.GetMetricsClient().Scope(\n\t\tmetrics.FrontendGetWorkflowExecutionHistoryScope,\n\t\tmetrics.DomainTag(domainName),\n\t\tmetrics.WorkflowCloseStatusTag(workflowCloseStatus),\n\t)\n\n\t// For closed workflows with close time, calculate days remaining until retention expires\n\t// For running workflows emitting the retention days value\n\tmetricValue := float64(retentionDays)\n\tif closeTime != nil {\n\t\t// (closeTime + retentionDays) - now = days remaining\n\t\tretentionExpiry := closeTime.Add(time.Duration(retentionDays) * 24 * time.Hour)\n\t\tdaysRemaining := time.Until(retentionExpiry).Hours() / 24\n\t\tmetricValue = daysRemaining\n\t}\n\n\tscope.UpdateGauge(metrics.WorkflowExecutionHistoryAccess, metricValue)\n}\n\n// Some error types are introduced later that some clients might not support\n// To make them backward compatible, we continue returning the legacy error types\n// for older clients\nfunc (wh *WorkflowHandler) normalizeVersionedErrors(ctx context.Context, err error) error {\n\tswitch err.(type) {\n\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\tcall := yarpc.CallFromContext(ctx)\n\t\tclientFeatureVersion := call.Header(common.FeatureVersionHeaderName)\n\t\tclientImpl := call.Header(common.ClientImplHeaderName)\n\t\tfeatureFlags := client.GetFeatureFlagsFromHeader(call)\n\n\t\tvErr := wh.versionChecker.SupportsWorkflowAlreadyCompletedError(clientImpl, clientFeatureVersion, featureFlags)\n\t\tif vErr == nil {\n\t\t\treturn err\n\t\t}\n\t\treturn &types.EntityNotExistsError{Message: \"Workflow execution already completed.\"}\n\tdefault:\n\t\treturn err\n\t}\n}\n\nfunc constructRestartWorkflowRequest(w *types.WorkflowExecutionStartedEventAttributes, domain string, identity string, workflowID string) *types.StartWorkflowExecutionRequest {\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:  uuid.New().String(),\n\t\tDomain:     domain,\n\t\tWorkflowID: workflowID,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: w.WorkflowType.Name,\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: w.TaskList.Name,\n\t\t\tKind: w.TaskList.Kind,\n\t\t},\n\t\tInput:                               w.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: w.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      w.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            identity,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t\tCronOverlapPolicy:                   w.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        w.ActiveClusterSelectionPolicy,\n\t}\n\tstartRequest.CronSchedule = w.CronSchedule\n\tstartRequest.RetryPolicy = w.RetryPolicy\n\tstartRequest.JitterStartSeconds = w.JitterStartSeconds\n\tstartRequest.ActiveClusterSelectionPolicy = w.ActiveClusterSelectionPolicy\n\tstartRequest.DelayStartSeconds = common.Int32Ptr(0)\n\tstartRequest.Header = w.Header\n\tstartRequest.Memo = w.Memo\n\tstartRequest.SearchAttributes = w.SearchAttributes\n\n\treturn startRequest\n}\n\nfunc getMetricsScopeWithDomain(\n\tscope metrics.ScopeIdx,\n\td domainGetter,\n\tmetricsClient metrics.Client,\n) metrics.Scope {\n\tvar metricsScope metrics.Scope\n\tif d != nil {\n\t\tmetricsScope = metricsClient.Scope(scope).Tagged(metrics.DomainTag(d.GetDomain()))\n\t} else {\n\t\tmetricsScope = metricsClient.Scope(scope).Tagged(metrics.DomainUnknownTag())\n\t}\n\treturn metricsScope\n}\n"
  },
  {
    "path": "service/frontend/api/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/yarpctest\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/domain\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\nconst (\n\tnumHistoryShards          = 10\n\ttestDomain                = \"test-domain\"\n\tcanaryDomain              = \"cadence-canary\"\n\ttestDomainID              = \"e4f90ec0-1313-45be-9877-8aa41f72a45a\"\n\ttestWorkflowID            = \"test-workflow-id\"\n\ttestRunID                 = \"2c8b555f-1f55-4955-9d1c-b980194555c9\"\n\ttestHistoryArchivalURI    = \"testScheme://history/URI\"\n\ttestVisibilityArchivalURI = \"testScheme://visibility/URI\"\n)\n\ntype (\n\tworkflowHandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockResource       *resource.Test\n\t\tmockDomainCache    *cache.MockDomainCache\n\t\tmockHistoryClient  *history.MockClient\n\t\tmockMatchingClient *matching.MockClient\n\t\tdomainHandler      domain.Handler\n\n\t\tmockProducer           *mocks.KafkaProducer\n\t\tmockMessagingClient    messaging.Client\n\t\tmockMetadataMgr        *mocks.MetadataManager\n\t\tmockHistoryV2Mgr       *mocks.HistoryV2Manager\n\t\tmockVisibilityMgr      *mocks.VisibilityManager\n\t\tmockArchivalMetadata   *archiver.MockArchivalMetadata\n\t\tmockArchiverProvider   *provider.MockArchiverProvider\n\t\tmockHistoryArchiver    *archiver.HistoryArchiverMock\n\t\tmockVisibilityArchiver *archiver.VisibilityArchiverMock\n\t\tmockVersionChecker     *client.MockVersionChecker\n\t\tmockTokenSerializer    *common.MockTaskTokenSerializer\n\t\tmockDomainAuditManager *persistence.MockDomainAuditManager\n\n\t\ttestDomain   string\n\t\ttestDomainID string\n\t}\n)\n\nfunc TestWorkflowHandlerSuite(t *testing.T) {\n\ts := new(workflowHandlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *workflowHandlerSuite) SetupSuite() {\n}\n\nfunc (s *workflowHandlerSuite) TearDownSuite() {\n}\n\nfunc (s *workflowHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.testDomain = testDomain\n\ts.testDomainID = testDomainID\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Frontend)\n\ts.mockDomainCache = s.mockResource.DomainCache\n\ts.mockHistoryClient = s.mockResource.HistoryClient\n\ts.mockMatchingClient = s.mockResource.MatchingClient\n\ts.mockMetadataMgr = s.mockResource.MetadataMgr\n\ts.mockHistoryV2Mgr = s.mockResource.HistoryMgr\n\ts.mockVisibilityMgr = s.mockResource.VisibilityMgr\n\ts.mockArchivalMetadata = s.mockResource.ArchivalMetadata\n\ts.mockArchiverProvider = s.mockResource.ArchiverProvider\n\ts.mockTokenSerializer = common.NewMockTaskTokenSerializer(s.controller)\n\n\ts.mockProducer = &mocks.KafkaProducer{}\n\ts.mockMessagingClient = mocks.NewMockMessagingClient(s.mockProducer, nil)\n\ts.mockHistoryArchiver = &archiver.HistoryArchiverMock{}\n\ts.mockVisibilityArchiver = &archiver.VisibilityArchiverMock{}\n\ts.mockVersionChecker = client.NewMockVersionChecker(s.controller)\n\ts.mockDomainAuditManager = persistence.NewMockDomainAuditManager(s.controller)\n\n\t// these tests don't mock the domain handler\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\ts.domainHandler = domain.NewHandler(\n\t\tconfig.DomainConfig,\n\t\ts.mockResource.GetLogger(),\n\t\ts.mockResource.GetDomainManager(),\n\t\ts.mockDomainAuditManager,\n\t\ts.mockResource.GetClusterMetadata(),\n\t\tdomain.NewDomainReplicator(s.mockProducer, s.mockResource.GetLogger()),\n\t\ts.mockResource.GetArchivalMetadata(),\n\t\ts.mockResource.GetArchiverProvider(),\n\t\ts.mockResource.GetTimeSource(),\n\t)\n\n\tmockMonitor := s.mockResource.MembershipResolver\n\tmockMonitor.EXPECT().MemberCount(service.Frontend).Return(5, nil).AnyTimes()\n\ts.mockVersionChecker.EXPECT().ClientSupported(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n}\n\nfunc (s *workflowHandlerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n\ts.mockProducer.AssertExpectations(s.T())\n\ts.mockHistoryArchiver.AssertExpectations(s.T())\n\ts.mockVisibilityArchiver.AssertExpectations(s.T())\n}\n\nfunc (s *workflowHandlerSuite) getWorkflowHandler(config *frontendcfg.Config) *WorkflowHandler {\n\treturn NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, s.domainHandler)\n}\n\nfunc (s *workflowHandlerSuite) TestDisableListVisibilityByFilter() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.DisableListVisibilityByFilter = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\n\twh := s.getWorkflowHandler(config)\n\n\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\n\t// test list open by wid\n\tlistRequest := &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain: s.testDomain,\n\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\tWorkflowID: \"wid\",\n\t\t},\n\t}\n\t_, err := wh.ListOpenWorkflowExecutions(context.Background(), listRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrNoPermission, err)\n\n\t// test list open by workflow type\n\tlistRequest.ExecutionFilter = nil\n\tlistRequest.TypeFilter = &types.WorkflowTypeFilter{\n\t\tName: \"workflow-type\",\n\t}\n\t_, err = wh.ListOpenWorkflowExecutions(context.Background(), listRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrNoPermission, err)\n\n\t// test list close by wid\n\tlistRequest2 := &types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain: s.testDomain,\n\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\tEarliestTime: common.Int64Ptr(0),\n\t\t\tLatestTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t\tExecutionFilter: &types.WorkflowExecutionFilter{\n\t\t\tWorkflowID: \"wid\",\n\t\t},\n\t}\n\t_, err = wh.ListClosedWorkflowExecutions(context.Background(), listRequest2)\n\ts.Error(err)\n\ts.Equal(validate.ErrNoPermission, err)\n\n\t// test list close by workflow type\n\tlistRequest2.ExecutionFilter = nil\n\tlistRequest2.TypeFilter = &types.WorkflowTypeFilter{\n\t\tName: \"workflow-type\",\n\t}\n\t_, err = wh.ListClosedWorkflowExecutions(context.Background(), listRequest2)\n\ts.Error(err)\n\ts.Equal(validate.ErrNoPermission, err)\n\n\t// test list close by workflow status\n\tlistRequest2.TypeFilter = nil\n\tfailedStatus := types.WorkflowExecutionCloseStatusFailed\n\tlistRequest2.StatusFilter = &failedStatus\n\t_, err = wh.ListClosedWorkflowExecutions(context.Background(), listRequest2)\n\ts.Error(err)\n\ts.Equal(validate.ErrNoPermission, err)\n}\n\nfunc (s *workflowHandlerSuite) TestPollForTask_Failed_ContextTimeoutTooShort() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\twh := s.getWorkflowHandler(config)\n\n\tbgCtx := context.Background()\n\t_, err := wh.PollForDecisionTask(bgCtx, &types.PollForDecisionTaskRequest{\n\t\tDomain: s.testDomain,\n\t})\n\ts.Error(err)\n\ts.Equal(common.ErrContextTimeoutNotSet, err)\n\n\t_, err = wh.PollForActivityTask(bgCtx, &types.PollForActivityTaskRequest{\n\t\tDomain: s.testDomain,\n\t})\n\ts.Error(err)\n\ts.Equal(common.ErrContextTimeoutNotSet, err)\n\n\tshortCtx, cancel := context.WithTimeout(bgCtx, constants.MinLongPollTimeout-time.Millisecond)\n\tdefer cancel()\n\n\t_, err = wh.PollForDecisionTask(shortCtx, &types.PollForDecisionTaskRequest{\n\t\tDomain: s.testDomain,\n\t})\n\ts.Error(err)\n\ts.Equal(common.ErrContextTimeoutTooShort, err)\n\n\t_, err = wh.PollForActivityTask(shortCtx, &types.PollForActivityTaskRequest{\n\t\tDomain: s.testDomain,\n\t})\n\ts.Error(err)\n\ts.Equal(common.ErrContextTimeoutTooShort, err)\n}\n\nfunc (s *workflowHandlerSuite) TestPollForDecisionTask_AutoConfigHint() {\n\n\tfor _, tt := range []struct {\n\t\tname     string\n\t\tresponse *types.MatchingPollForDecisionTaskResponse\n\t\texpected *types.PollForDecisionTaskResponse\n\t}{\n\t\t{\n\t\t\t\"success\",\n\t\t\t&types.MatchingPollForDecisionTaskResponse{\n\t\t\t\tTaskToken: []byte(\"some value\"),\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t},\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t\t&types.PollForDecisionTaskResponse{\n\t\t\t\tTaskToken: []byte(\"some value\"),\n\t\t\t\tHistory: &types.History{\n\t\t\t\t\tEvents: []*types.HistoryEvent{},\n\t\t\t\t},\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t},\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"success with empty poll\",\n\t\t\t&types.MatchingPollForDecisionTaskResponse{\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t\t&types.PollForDecisionTaskResponse{\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t},\n\t} {\n\t\ts.T().Run(tt.name, func(t *testing.T) {\n\t\t\ts.mockDomainCache.EXPECT().GetDomain(s.testDomain).Return(cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{Name: s.testDomain, ID: s.testDomainID},\n\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\"\",\n\t\t\t), nil).AnyTimes()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil).AnyTimes()\n\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(\n\t\t\t\t&persistence.ReadHistoryBranchResponse{}, nil).Maybe()\n\t\t\ts.mockResource.MatchingClient.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).Return(\n\t\t\t\ttt.response, nil).Times(1)\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second*5)\n\t\t\tdefer cancel()\n\t\t\tconfig := s.newConfig(dc.NewInMemoryClient())\n\t\t\twh := s.getWorkflowHandler(config)\n\t\t\tresp, err := wh.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"task-list\",\n\t\t\t\t},\n\t\t\t})\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tt.expected, resp)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestPollForDecisionTask_IsolationGroupDrained() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\twh := s.getWorkflowHandler(config)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second*5)\n\tdefer cancel()\n\tisolationGroup := \"dca1\"\n\tctx = isolationgroup.ContextWithIsolationGroup(ctx, isolationGroup)\n\n\ts.mockDomainCache.EXPECT().GetDomain(s.testDomain).Return(cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{},\n\t\t\"\",\n\t), nil)\n\ts.mockResource.IsolationGroups.EXPECT().IsDrained(gomock.Any(), s.testDomain, isolationGroup).Return(true, nil).AnyTimes()\n\tresp, err := wh.PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{\n\t\tDomain: s.testDomain,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.Equal(&types.PollForDecisionTaskResponse{}, resp)\n}\n\nfunc (s *workflowHandlerSuite) TestPollForActivityTask_IsolationGroupDrained() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\twh := s.getWorkflowHandler(config)\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second*5)\n\tdefer cancel()\n\tisolationGroup := \"dca1\"\n\tctx = isolationgroup.ContextWithIsolationGroup(ctx, isolationGroup)\n\n\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\ts.mockResource.IsolationGroups.EXPECT().IsDrained(gomock.Any(), s.testDomain, isolationGroup).Return(true, nil).AnyTimes()\n\tresp, err := wh.PollForActivityTask(ctx, &types.PollForActivityTaskRequest{\n\t\tDomain: s.testDomain,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.Equal(&types.PollForActivityTaskResponse{}, resp)\n}\n\nfunc (s *workflowHandlerSuite) TestPollForActivityTask() {\n\n\tfor _, tt := range []struct {\n\t\tname     string\n\t\tresponse *types.MatchingPollForActivityTaskResponse\n\t\texpected *types.PollForActivityTaskResponse\n\t}{\n\t\t{\n\t\t\t\"success\",\n\t\t\t&types.MatchingPollForActivityTaskResponse{\n\t\t\t\tTaskToken: []byte(\"token\"),\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t},\n\t\t\t\tActivityID:     \"1\",\n\t\t\t\tInput:          []byte(`{\"key\": \"value\"}`),\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t\t&types.PollForActivityTaskResponse{\n\t\t\t\tTaskToken: []byte(\"token\"),\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t},\n\t\t\t\tActivityID:     \"1\",\n\t\t\t\tInput:          []byte(`{\"key\": \"value\"}`),\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"success with empty polls\",\n\t\t\t&types.MatchingPollForActivityTaskResponse{\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t\t&types.PollForActivityTaskResponse{\n\t\t\t\tAutoConfigHint: &types.AutoConfigHint{EnableAutoConfig: true, PollerWaitTimeInMs: 1000},\n\t\t\t},\n\t\t},\n\t} {\n\t\ts.T().Run(tt.name, func(t *testing.T) {\n\t\t\tconfig := s.newConfig(dc.NewInMemoryClient())\n\t\t\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\t\t\twh := s.getWorkflowHandler(config)\n\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second*5)\n\t\t\tdefer cancel()\n\t\t\tisolationGroup := \"dca1\"\n\t\t\tctx = isolationgroup.ContextWithIsolationGroup(ctx, isolationGroup)\n\n\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\ts.mockResource.IsolationGroups.EXPECT().IsDrained(gomock.Any(), s.testDomain, isolationGroup).Return(false, nil).AnyTimes()\n\t\t\ts.mockMatchingClient.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any()).Return(tt.response, nil)\n\t\t\tresp, err := wh.PollForActivityTask(ctx, &types.PollForActivityTaskRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"task-list\",\n\t\t\t\t},\n\t\t\t})\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(tt.expected, resp)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_RequestIdNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(&types.BadRequestError{Message: \"requestId \\\"\\\" is not a valid UUID\"}, err)\n\tstartWorkflowExecutionRequest.RequestID = \"xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx\"\n\t_, err = wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(&types.BadRequestError{Message: \"requestId \\\"xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx\\\" is not a valid UUID\"}, err)\n\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_BadDelayStartSeconds() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID:         uuid.New(),\n\t\tDelayStartSeconds: common.Int32Ptr(-1),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrInvalidDelayStartSeconds, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_StartRequestNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\t_, err := wh.StartWorkflowExecution(context.Background(), nil)\n\ts.Error(err)\n\ts.Equal(validate.ErrRequestNotSet, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_DomainNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrDomainNotSet, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_WorkflowIdNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain: s.testDomain,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrWorkflowIDNotSet, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_WorkflowTypeNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrWorkflowTypeNotSet, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_TaskListNotSet() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrTaskListNotSet, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_InvalidExecutionStartToCloseTimeout() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(0),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrInvalidExecutionStartToCloseTimeoutSeconds, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Failed_InvalidTaskStartToCloseTimeout() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(0),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.Equal(validate.ErrInvalidTaskStartToCloseTimeoutSeconds, err)\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_IsolationGroupDrained() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\twh := s.getWorkflowHandler(config)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\tisolationGroup := \"dca1\"\n\tctx := isolationgroup.ContextWithIsolationGroup(context.Background(), isolationGroup)\n\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\ts.mockResource.IsolationGroups.EXPECT().IsDrained(gomock.Any(), s.testDomain, isolationGroup).Return(true, nil)\n\t_, err := wh.StartWorkflowExecution(ctx, startWorkflowExecutionRequest)\n\ts.Error(err)\n\ts.IsType(err, &types.BadRequestError{})\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_LogJitterTime() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.UserRPS = dynamicproperties.GetIntPropertyFn(10)\n\twh := s.getWorkflowHandler(config)\n\tjitterStart := int32(10)\n\n\tstartWorkflowExecutionRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     canaryDomain,\n\t\tWorkflowID: \"workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"workflow-type\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"task-list\",\n\t\t},\n\t\tJitterStartSeconds:                  &jitterStart,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          2,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tMaximumAttempts:             1,\n\t\t\tExpirationIntervalInSeconds: 1,\n\t\t},\n\t\tRequestID: uuid.New(),\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainID(canaryDomain).Return(s.testDomainID, nil).Times(2)\n\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{RunID: \"test-rid\"}, nil)\n\t_, err := wh.StartWorkflowExecution(context.Background(), startWorkflowExecutionRequest)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestDiagnoseWorkflowExecution_Success() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain: testDomain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tIdentity: \"\",\n\t}\n\tdiagnosticWfDomain := \"cadence-system\"\n\tdiagnosticWfID := fmt.Sprintf(\"%s-%s\", testDomain, testRunID)\n\tdiagnosticWfRunID := \"123\"\n\tresp := &types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain: diagnosticWfDomain,\n\t\tDiagnosticWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: diagnosticWfID,\n\t\t\tRunID:      diagnosticWfRunID,\n\t\t},\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomainID(diagnosticWfDomain).Return(s.testDomainID, nil).Times(2)\n\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{RunID: diagnosticWfRunID}, nil)\n\tresult, err := wh.DiagnoseWorkflowExecution(context.Background(), req)\n\ts.NoError(err)\n\ts.Equal(resp, result)\n}\n\nfunc (s *workflowHandlerSuite) TestDiagnoseWorkflowExecution_Success_WfAlreadyRunning() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain: testDomain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tIdentity: \"\",\n\t}\n\tdiagnosticWfDomain := \"cadence-system\"\n\tdiagnosticWfID := fmt.Sprintf(\"%s-%s\", testDomain, testRunID)\n\tdiagnosticWfRunID := \"123\"\n\tresp := &types.DiagnoseWorkflowExecutionResponse{\n\t\tDomain: diagnosticWfDomain,\n\t\tDiagnosticWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: diagnosticWfID,\n\t\t\tRunID:      diagnosticWfRunID,\n\t\t},\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomainID(diagnosticWfDomain).Return(s.testDomainID, nil).Times(2)\n\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.WorkflowExecutionAlreadyStartedError{\n\t\tMessage: \"Workflow execution is already running\",\n\t\tRunID:   diagnosticWfRunID,\n\t})\n\tresult, err := wh.DiagnoseWorkflowExecution(context.Background(), req)\n\ts.NoError(err)\n\ts.Equal(resp, result)\n}\n\nfunc (s *workflowHandlerSuite) TestDiagnoseWorkflowExecution_Failed_RequestNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresult, err := wh.DiagnoseWorkflowExecution(context.Background(), nil)\n\ts.Error(err)\n\ts.Equal(validate.ErrRequestNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestDiagnoseWorkflowExecution_Failed_DomainNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresult, err := wh.DiagnoseWorkflowExecution(context.Background(), &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain: \"\",\n\t})\n\ts.Error(err)\n\ts.Equal(validate.ErrDomainNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestDiagnoseWorkflowExecution_Failed_ExecutionNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresult, err := wh.DiagnoseWorkflowExecution(context.Background(), &types.DiagnoseWorkflowExecutionRequest{\n\t\tDomain: testDomain,\n\t})\n\ts.Error(err)\n\ts.Equal(validate.ErrExecutionNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeat_Success() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\ttaskToken := common.TaskToken{\n\t\tDomainID:   s.testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"1\",\n\t}\n\ttaskTokenBytes, err := wh.tokenSerializer.Serialize(&taskToken)\n\ts.NoError(err)\n\treq := &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: taskTokenBytes,\n\t\tDetails:   nil,\n\t\tIdentity:  \"\",\n\t}\n\tresp := &types.RecordActivityTaskHeartbeatResponse{CancelRequested: false}\n\n\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\ts.mockHistoryClient.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(),\n\t\t&types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\tDomainUUID:       s.testDomainID,\n\t\t\tHeartbeatRequest: req,\n\t\t}).Return(resp, nil)\n\n\tresult, err := wh.RecordActivityTaskHeartbeat(context.Background(), req)\n\ts.NoError(err)\n\ts.Equal(resp, result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeat_RequestNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\tresult, err := wh.RecordActivityTaskHeartbeat(context.Background(), nil /*request is not set*/)\n\n\ts.Error(err)\n\ts.Equal(validate.ErrRequestNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeat_TaskTokenNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\tresult, err := wh.RecordActivityTaskHeartbeat(context.Background(), &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: nil, // task token is not set\n\t\tDetails:   nil,\n\t\tIdentity:  \"\",\n\t})\n\n\ts.Error(err)\n\ts.Equal(validate.ErrTaskTokenNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeatByID_Success() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\treq := &types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"1\",\n\t}\n\tresp := &types.RecordActivityTaskHeartbeatResponse{CancelRequested: false}\n\n\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\ts.mockHistoryClient.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), gomock.Any()).Return(resp, nil)\n\n\tresult, err := wh.RecordActivityTaskHeartbeatByID(context.Background(), req)\n\ts.NoError(err)\n\ts.Equal(resp, result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeatByID_RequestNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\tresult, err := wh.RecordActivityTaskHeartbeatByID(context.Background(), nil /*request is not set*/)\n\n\ts.Error(err)\n\ts.Equal(validate.ErrRequestNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestRecordActivityTaskHeartbeatByID_DomainNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\tresult, err := wh.RecordActivityTaskHeartbeatByID(\n\t\tcontext.Background(),\n\t\t&types.RecordActivityTaskHeartbeatByIDRequest{\n\t\t\tDomain: \"\", // domain not set\n\t\t})\n\n\ts.Error(err)\n\ts.Equal(validate.ErrDomainNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestRespondActivityTaskCompleted_Success() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\ttaskToken := common.TaskToken{\n\t\tDomainID:   s.testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"1\",\n\t}\n\ttaskTokenBytes, err := wh.tokenSerializer.Serialize(&taskToken)\n\ts.NoError(err)\n\treq := &types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: taskTokenBytes,\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\ts.mockHistoryClient.EXPECT().RespondActivityTaskCompleted(gomock.Any(),\n\t\t&types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\tDomainUUID:      taskToken.DomainID,\n\t\t\tCompleteRequest: req,\n\t\t}).Return(nil)\n\n\terr = wh.RespondActivityTaskCompleted(context.Background(), req)\n\t// only checking for successful write here\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestRespondActivityTaskCompletedByID_Success() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\treq := &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"1\",\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\ts.mockHistoryClient.EXPECT().RespondActivityTaskCompleted(gomock.Any(), gomock.Any()).Return(nil)\n\n\terr := wh.RespondActivityTaskCompletedByID(context.Background(), req)\n\t// only checking for successful write here\n\ts.NoError(err)\n}\n\nfunc buildRespondActivityTaskFailedRequest(taskToken common.TaskToken) *types.RespondActivityTaskFailedRequest {\n\tserializer := common.NewJSONTaskTokenSerializer()\n\ttaskTokenBytes, err := serializer.Serialize(&taskToken)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn &types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: taskTokenBytes,\n\t}\n}\n\nfunc TestRespondActivityTaskFailed(t *testing.T) {\n\tfailedRequest := buildRespondActivityTaskFailedRequest(common.TaskToken{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"1\",\n\t})\n\n\ttype fields struct {\n\t\tshuttingDown int32\n\t}\n\n\ttype args struct {\n\t\tctx           context.Context\n\t\tfailedRequest *types.RespondActivityTaskFailedRequest\n\t}\n\n\ttests := []struct {\n\t\tname       string\n\t\tfields     fields\n\t\tsetupMocks func(*resource.Test, *client.MockVersionChecker)\n\t\targs       args\n\t\twantErr    assert.ErrorAssertionFunc\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tfields: fields{\n\t\t\t\tshuttingDown: 0,\n\t\t\t},\n\t\t\tsetupMocks: func(t *resource.Test, mockVersionChecker *client.MockVersionChecker) {\n\t\t\t\tt.HistoryClient.EXPECT().RespondActivityTaskFailed(gomock.Any(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\t\tDomainUUID:    testDomainID,\n\t\t\t\t\tFailedRequest: failedRequest,\n\t\t\t\t}).Return(nil)\n\n\t\t\t\tt.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-id\", nil)\n\t\t\t},\n\t\t\targs: args{\n\t\t\t\tcontext.Background(),\n\t\t\t\tfailedRequest,\n\t\t\t},\n\t\t\twantErr: assert.NoError,\n\t\t},\n\t\t{\n\t\t\tname:   \"Error when shutting down\",\n\t\t\tfields: fields{shuttingDown: 1},\n\t\t\tsetupMocks: func(t *resource.Test, mockVersionChecker *client.MockVersionChecker) {\n\n\t\t\t},\n\t\t\targs: args{\n\t\t\t\tcontext.Background(),\n\t\t\t\tbuildRespondActivityTaskFailedRequest(common.TaskToken{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t\tActivityID: \"1\",\n\t\t\t\t}),\n\t\t\t},\n\t\t\twantErr: assert.Error,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\n\t\t\ttt.setupMocks(mockResource, mockVersionChecker)\n\n\t\t\tmockProducerManager := NewMockProducerManager(mockCtrl)\n\n\t\t\tconfig := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\tdc.NewInMemoryClient(),\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\n\t\t\twh := NewWorkflowHandler(mockResource, config, mockVersionChecker, nil)\n\t\t\twh.shuttingDown = tt.fields.shuttingDown\n\t\t\twh.producerManager = mockProducerManager\n\n\t\t\ttt.wantErr(t, wh.RespondActivityTaskFailed(tt.args.ctx, tt.args.failedRequest),\n\t\t\t\tfmt.Sprintf(\"RespondActivityTaskFailed(%v, %v)\", tt.args.ctx, tt.args.failedRequest))\n\t\t})\n\t}\n\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Failure_MissingDomainDataKey() {\n\tdynamicClient := dc.NewInMemoryClient()\n\terr := dynamicClient.UpdateValue(dynamicproperties.RequiredDomainDataKeys, map[string]interface{}{\"Tier\": true})\n\ts.NoError(err)\n\tcfg := s.newConfig(dynamicClient)\n\twh := s.getWorkflowHandler(cfg)\n\n\treq := registerDomainRequest(\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestHistoryArchivalURI,\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestVisibilityArchivalURI,\n\t)\n\terr = wh.RegisterDomain(context.Background(), req)\n\ts.Error(err)\n\ts.Contains(err.Error(), \"domain data error, missing required key\")\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Failure_InvalidArchivalURI() {\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{})\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(errors.New(\"invalid URI\"))\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := registerDomainRequest(\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestHistoryArchivalURI,\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestVisibilityArchivalURI,\n\t)\n\terr := wh.RegisterDomain(context.Background(), req)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Success_EnabledWithNoArchivalURI() {\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", testHistoryArchivalURI))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", testVisibilityArchivalURI))\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{})\n\ts.mockMetadataMgr.On(\"CreateDomain\", mock.Anything, mock.Anything).Return(&persistence.CreateDomainResponse{\n\t\tID: \"test-id\",\n\t}, nil)\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := registerDomainRequest(types.ArchivalStatusEnabled.Ptr(), \"\", types.ArchivalStatusEnabled.Ptr(), \"\")\n\terr := wh.RegisterDomain(context.Background(), req)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Success_EnabledWithArchivalURI() {\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"invalidURI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"invalidURI\"))\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{})\n\ts.mockMetadataMgr.On(\"CreateDomain\", mock.Anything, mock.Anything).Return(&persistence.CreateDomainResponse{\n\t\tID: \"test-id\",\n\t}, nil)\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := registerDomainRequest(\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestHistoryArchivalURI,\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestVisibilityArchivalURI,\n\t)\n\terr := wh.RegisterDomain(context.Background(), req)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Success_ClusterNotConfiguredForArchival() {\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewDisabledArchvialConfig())\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{})\n\ts.mockMetadataMgr.On(\"CreateDomain\", mock.Anything, mock.Anything).Return(&persistence.CreateDomainResponse{\n\t\tID: \"test-id\",\n\t}, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := registerDomainRequest(\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\ttestVisibilityArchivalURI,\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\t\"invalidURI\",\n\t)\n\terr := wh.RegisterDomain(context.Background(), req)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestRegisterDomain_Success_NotEnabled() {\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{})\n\ts.mockMetadataMgr.On(\"CreateDomain\", mock.Anything, mock.Anything).Return(&persistence.CreateDomainResponse{\n\t\tID: \"test-id\",\n\t}, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := registerDomainRequest(nil, \"\", nil, \"\")\n\terr := wh.RegisterDomain(context.Background(), req)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListDomains_Success() {\n\tdomainResponse := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{},\n\t\t&domain.ArchivalState{},\n\t)\n\tlistDomainResp := &persistence.ListDomainsResponse{\n\t\tDomains: []*persistence.GetDomainResponse{\n\t\t\tdomainResponse,\n\t\t\tdomainResponse,\n\t\t},\n\t}\n\ts.mockMetadataMgr.\n\t\tOn(\"ListDomains\", mock.Anything, mock.Anything).\n\t\tReturn(listDomainResp, nil)\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresult, err := wh.ListDomains(context.Background(), &types.ListDomainsRequest{})\n\ts.NoError(err)\n\n\ts.Equal(2, len(result.GetDomains()))\n}\n\nfunc (s *workflowHandlerSuite) TestListDomains_RequestNotSet() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresult, err := wh.ListDomains(context.Background(), nil /* list request is not set */)\n\ts.Error(err)\n\ts.Equal(validate.ErrRequestNotSet, err)\n\ts.Nil(result)\n}\n\nfunc (s *workflowHandlerSuite) TestHealth_Status() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient())) // workflow handler gets initial health status as HealthStatusWarmingUp\n\n\tresult, err := wh.Health(context.Background()) // Health check looks for HealthStatusOK\n\ts.NoError(err)\n\ts.False(result.Ok)\n\ts.Equal(\"WarmingUp\", result.Msg)\n\n\twh.UpdateHealthStatus(HealthStatusOK)\n\tresult, err = wh.Health(context.Background())\n\ts.NoError(err)\n\ts.True(result.Ok)\n\ts.Equal(\"OK\", result.Msg)\n\n\twh.UpdateHealthStatus(HealthStatusShuttingDown)\n\tresult, err = wh.Health(context.Background())\n\ts.NoError(err)\n\ts.False(result.Ok)\n\ts.Equal(\"ShuttingDown\", result.Msg)\n\n\twh.UpdateHealthStatus(HealthStatus(-1))\n\tresult, err = wh.Health(context.Background())\n\ts.NoError(err)\n\ts.False(result.Ok)\n\ts.Equal(\"unknown\", result.Msg)\n}\n\nfunc (s *workflowHandlerSuite) TestGetClusterInfo() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.GetClusterInfo(context.Background())\n\ts.NoError(err)\n\ts.Equal(\"1.7.0\", resp.SupportedClientVersions.GoSdk)\n\ts.Equal(\"1.5.0\", resp.SupportedClientVersions.JavaSdk)\n}\n\nfunc (s *workflowHandlerSuite) TestDescribeDomain_Success_ArchivalDisabled() {\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(s.testDomain),\n\t}\n\tresult, err := wh.DescribeDomain(context.Background(), req)\n\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(\"\", result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(\"\", result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestDescribeDomain_Success_ArchivalEnabled() {\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testHistoryArchivalURI},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testVisibilityArchivalURI},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\treq := &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(s.testDomain),\n\t}\n\tresult, err := wh.DescribeDomain(context.Background(), req)\n\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(testHistoryArchivalURI, result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(testVisibilityArchivalURI, result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Failure_UpdateExistingArchivalURI() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testHistoryArchivalURI},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testVisibilityArchivalURI},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tnil,\n\t\tnil,\n\t\tcommon.StringPtr(\"updated visibility URI\"),\n\t\tnil,\n\t)\n\t_, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Failure_InvalidArchivalURI() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(errors.New(\"invalid URI\"))\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tcommon.StringPtr(\"testScheme://invalid/updated/history/URI\"),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tnil,\n\t\tnil,\n\t)\n\n\t_, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_ArchivalEnabledToArchivalDisabledWithoutSettingURI() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testHistoryArchivalURI},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testVisibilityArchivalURI},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, mock.Anything).Return(nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tnil,\n\t\ttypes.ArchivalStatusDisabled.Ptr(),\n\t\tnil,\n\t\ttypes.ArchivalStatusDisabled.Ptr(),\n\t)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(testHistoryArchivalURI, result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(testVisibilityArchivalURI, result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_ClusterNotConfiguredForArchival() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: \"some random history URI\"},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: \"some random visibility URI\"},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewDisabledArchvialConfig())\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(nil, types.ArchivalStatusDisabled.Ptr(), nil, nil)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(\"some random history URI\", result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(\"some random visibility URI\", result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_ArchivalEnabledToArchivalDisabledWithSettingBucket() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testHistoryArchivalURI},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testVisibilityArchivalURI},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, mock.Anything).Return(nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tcommon.StringPtr(testHistoryArchivalURI),\n\t\ttypes.ArchivalStatusDisabled.Ptr(),\n\t\tcommon.StringPtr(testVisibilityArchivalURI),\n\t\ttypes.ArchivalStatusDisabled.Ptr(),\n\t)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(testHistoryArchivalURI, result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusDisabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(testVisibilityArchivalURI, result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_ArchivalEnabledToEnabled() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testHistoryArchivalURI},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusEnabled, URI: testVisibilityArchivalURI},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tcommon.StringPtr(testHistoryArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.StringPtr(testVisibilityArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(testHistoryArchivalURI, result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(testVisibilityArchivalURI, result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_ArchivalNeverEnabledToEnabled() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponse(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t)\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, mock.Anything).Return(nil)\n\ts.mockArchivalMetadata.On(\"GetHistoryConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"some random URI\"))\n\ts.mockHistoryArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockVisibilityArchiver.On(\"ValidateURI\", mock.Anything).Return(nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateRequest(\n\t\tcommon.StringPtr(testHistoryArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.StringPtr(testVisibilityArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetHistoryArchivalStatus())\n\ts.Equal(testHistoryArchivalURI, result.Configuration.GetHistoryArchivalURI())\n\ts.Equal(types.ArchivalStatusEnabled, result.Configuration.GetVisibilityArchivalStatus())\n\ts.Equal(testVisibilityArchivalURI, result.Configuration.GetVisibilityArchivalURI())\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Success_FailOver() {\n\ts.mockMetadataMgr.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: int64(0),\n\t}, nil)\n\tgetDomainResp := persistenceGetDomainResponseForFailoverTest(\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t\t&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: \"\"},\n\t)\n\n\t// This test is simulating a domain failover from the point of view of the 'standby' cluster\n\t// for a domain where the cluster 'active' is being failed over to 'standby'. The test is executing\n\t// in the 'standby' cluster, so the above is setting the configuration to appear that way.\n\ts.mockResource.ClusterMetadata = cluster.TestPassiveClusterMetadata\n\n\t// Re-instantiate the domain-handler object due to it relying on it\n\t// pulling in the mock cluster metadata object mutated above.\n\t// Todo (David.Porter) consider refactoring these tests\n\t// to be setup without mutation and without as long dependency chains\n\ts.domainHandler = domain.NewHandler(\n\t\ts.newConfig(dc.NewInMemoryClient()).DomainConfig,\n\t\ts.mockResource.GetLogger(),\n\t\ts.mockResource.GetDomainManager(),\n\t\ts.mockDomainAuditManager,\n\t\ts.mockResource.GetClusterMetadata(),\n\t\tdomain.NewDomainReplicator(s.mockProducer, s.mockResource.GetLogger()),\n\t\ts.mockResource.GetArchivalMetadata(),\n\t\ts.mockResource.GetArchiverProvider(),\n\t\ts.mockResource.GetTimeSource(),\n\t)\n\n\ts.mockMetadataMgr.On(\"GetDomain\", mock.Anything, mock.Anything).Return(getDomainResp, nil)\n\ts.mockMetadataMgr.On(\"UpdateDomain\", mock.Anything, mock.Anything).Return(nil)\n\ts.mockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockResource.RemoteFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).\n\t\tReturn(describeDomainResponseServer, nil).AnyTimes()\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tupdateReq := updateFailoverRequest(\n\t\tcommon.StringPtr(testHistoryArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.StringPtr(testVisibilityArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.Int32Ptr(1),\n\t\tcommon.StringPtr(cluster.TestAlternativeClusterName),\n\t)\n\tresult, err := wh.UpdateDomain(context.Background(), updateReq)\n\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.NotNil(result.Configuration)\n\ts.Equal(result.ReplicationConfiguration.ActiveClusterName, cluster.TestAlternativeClusterName)\n}\n\nfunc (s *workflowHandlerSuite) TestUpdateDomain_Failure_FailoverLockdown() {\n\n\tdynamicClient := dc.NewInMemoryClient()\n\terr := dynamicClient.UpdateValue(dynamicproperties.Lockdown, true)\n\ts.NoError(err)\n\twh := s.getWorkflowHandler(s.newConfig(dynamicClient))\n\n\tupdateReq := updateFailoverRequest(\n\t\tcommon.StringPtr(testHistoryArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.StringPtr(testVisibilityArchivalURI),\n\t\ttypes.ArchivalStatusEnabled.Ptr(),\n\t\tcommon.Int32Ptr(1),\n\t\tcommon.StringPtr(cluster.TestAlternativeClusterName),\n\t)\n\tresp, err := wh.UpdateDomain(context.Background(), updateReq)\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestHistoryArchived() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tgetHistoryRequest := &types.GetWorkflowExecutionHistoryRequest{}\n\ts.False(wh.historyArchived(context.Background(), getHistoryRequest, s.testDomain))\n\n\tgetHistoryRequest = &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution: &types.WorkflowExecution{},\n\t}\n\ts.False(wh.historyArchived(context.Background(), getHistoryRequest, s.testDomain))\n\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\tgetHistoryRequest = &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t}\n\ts.False(wh.historyArchived(context.Background(), getHistoryRequest, s.testDomain))\n\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{Message: \"got archival indication error\"}).Times(1)\n\tgetHistoryRequest = &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t}\n\ts.True(wh.historyArchived(context.Background(), getHistoryRequest, s.testDomain))\n\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"got non-archival indication error\")).Times(1)\n\tgetHistoryRequest = &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t}\n\ts.False(wh.historyArchived(context.Background(), getHistoryRequest, s.testDomain))\n}\n\nfunc (s *workflowHandlerSuite) TestGetArchivedHistory_Failure_DomainCacheEntryError() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(nil, errors.New(\"error getting domain\")).Times(1)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.getArchivedHistory(context.Background(), getHistoryRequest(nil), s.testDomainID)\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestGetArchivedHistory_Failure_ArchivalURIEmpty() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusDisabled,\n\t\t\tHistoryArchivalURI:       \"\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\t\"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(domainEntry, nil).AnyTimes()\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.getArchivedHistory(context.Background(), getHistoryRequest(nil), s.testDomainID)\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestGetArchivedHistory_Failure_InvalidURI() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalURI:       \"uri without scheme\",\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\t\"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(domainEntry, nil).AnyTimes()\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.getArchivedHistory(context.Background(), getHistoryRequest(nil), s.testDomainID)\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestGetArchivedHistory_Success_GetFirstPage() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{\n\t\t\tHistoryArchivalStatus:    types.ArchivalStatusEnabled,\n\t\t\tHistoryArchivalURI:       testHistoryArchivalURI,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"\",\n\t\t},\n\t\t\"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(domainEntry, nil).AnyTimes()\n\n\tnextPageToken := []byte{'1', '2', '3'}\n\thistoryBatch1 := &types.History{\n\t\tEvents: []*types.HistoryEvent{\n\t\t\t{ID: 1},\n\t\t\t{ID: 2},\n\t\t},\n\t}\n\thistoryBatch2 := &types.History{\n\t\tEvents: []*types.HistoryEvent{\n\t\t\t{ID: 3},\n\t\t\t{ID: 4},\n\t\t\t{ID: 5},\n\t\t},\n\t}\n\texpectedHistory := &types.History{}\n\texpectedHistory.Events = append(expectedHistory.Events, historyBatch1.Events...)\n\texpectedHistory.Events = append(expectedHistory.Events, historyBatch2.Events...)\n\ts.mockHistoryArchiver.On(\"Get\", mock.Anything, mock.Anything, mock.Anything).Return(&archiver.GetHistoryResponse{\n\t\tNextPageToken:  nextPageToken,\n\t\tHistoryBatches: []*types.History{historyBatch1, historyBatch2},\n\t}, nil)\n\ts.mockArchiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.mockHistoryArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.getArchivedHistory(context.Background(), getHistoryRequest(nil), s.testDomainID)\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.NotNil(resp.History)\n\ts.Equal(expectedHistory, resp.History)\n\ts.Equal(nextPageToken, resp.NextPageToken)\n\ts.True(resp.GetArchived())\n}\n\nfunc (s *workflowHandlerSuite) TestGetHistory() {\n\tdomainID := uuid.New()\n\tdomainName := uuid.New()\n\tfirstEventID := int64(100)\n\tnextEventID := int64(101)\n\tbranchToken := []byte{1}\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wid\",\n\t\tRunID:      \"rid\",\n\t}\n\tshardID := common.WorkflowIDToHistoryShard(we.WorkflowID, numHistoryShards)\n\treq := &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      0,\n\t\tNextPageToken: []byte{},\n\t\tShardID:       common.IntPtr(shardID),\n\t\tDomainName:    domainName,\n\t}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, req).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID: int64(100),\n\t\t\t},\n\t\t},\n\t\tNextPageToken:    []byte{},\n\t\tSize:             1,\n\t\tLastFirstEventID: nextEventID,\n\t}, nil).Once()\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tscope := metrics.NoopScope\n\tactualHistory, token, err := wh.getHistory(context.Background(), scope, domainID, domainName, we, firstEventID, nextEventID, 0, []byte{}, nil, branchToken)\n\ts.NoError(err)\n\ts.NotNil(actualHistory)\n\ts.Equal([]byte{}, token)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Failure_InvalidRequest() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), &types.ListArchivedWorkflowExecutionsRequest{})\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Failure_ClusterNotConfiguredForArchival() {\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), listArchivedWorkflowExecutionsTestRequest())\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Failure_DomainCacheEntryError() {\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, errors.New(\"error getting domain\"))\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), listArchivedWorkflowExecutionsTestRequest())\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Failure_DomainNotConfiguredForArchival() {\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewLocalDomainCacheEntryForTest(\n\t\tnil,\n\t\t&persistence.DomainConfig{\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t},\n\t\t\"\",\n\t), nil)\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), listArchivedWorkflowExecutionsTestRequest())\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Failure_InvalidURI() {\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusDisabled,\n\t\t\tVisibilityArchivalURI:    \"uri without scheme\",\n\t\t},\n\t\t\"\",\n\t), nil)\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), listArchivedWorkflowExecutionsTestRequest())\n\ts.Nil(resp)\n\ts.Error(err)\n}\n\nfunc (s *workflowHandlerSuite) TestListArchivedVisibility_Success() {\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomain},\n\t\t&persistence.DomainConfig{\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    testVisibilityArchivalURI,\n\t\t},\n\t\t\"\",\n\t), nil).AnyTimes()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\ts.mockVisibilityArchiver.On(\"Query\", mock.Anything, mock.Anything, mock.Anything).Return(&archiver.QueryVisibilityResponse{}, nil)\n\ts.mockArchiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.mockVisibilityArchiver, nil)\n\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tresp, err := wh.ListArchivedWorkflowExecutions(context.Background(), listArchivedWorkflowExecutionsTestRequest())\n\ts.NotNil(resp)\n\ts.NoError(err)\n}\n\nfunc (s *workflowHandlerSuite) TestGetSearchAttributes() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\tctx := context.Background()\n\tresp, err := wh.GetSearchAttributes(ctx)\n\ts.NoError(err)\n\ts.NotNil(resp)\n}\n\nfunc (s *workflowHandlerSuite) TestGetWorkflowExecutionHistory__Success__RawHistoryEnabledTransientDecisionEmitted() {\n\tvar nextEventID int64 = 5\n\ts.getWorkflowExecutionHistory(5, &types.TransientDecisionInfo{\n\t\tStartedEvent:   &types.HistoryEvent{ID: nextEventID + 1},\n\t\tScheduledEvent: &types.HistoryEvent{ID: nextEventID},\n\t}, []*types.HistoryEvent{{}, {}, {}})\n}\n\nfunc (s *workflowHandlerSuite) TestGetWorkflowExecutionHistory__Success__RawHistoryEnabledNoTransientDecisionEmitted() {\n\tvar nextEventID int64 = 5\n\ts.getWorkflowExecutionHistory(5, &types.TransientDecisionInfo{\n\t\tStartedEvent:   &types.HistoryEvent{ID: nextEventID + 1},\n\t\tScheduledEvent: &types.HistoryEvent{ID: nextEventID},\n\t}, []*types.HistoryEvent{{}, {}, {}})\n}\n\nfunc (s *workflowHandlerSuite) TestRestartWorkflowExecution() {\n\ttestCases := []struct {\n\t\tname                    string\n\t\tsetupMocks              func()\n\t\trequest                 *types.RestartWorkflowExecutionRequest\n\t\tsetupContext            func() context.Context\n\t\tenableTaskListIsolation bool\n\t\texpectError             bool\n\t\texpectedErrType         interface{}\n\t\texpectedRunID           string\n\t\tdescription             string\n\t}{\n\t\t{\n\t\t\tname:       \"When request is nil it should return RequestNotSet error\",\n\t\t\tsetupMocks: func() {},\n\t\t\trequest:    nil,\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\texpectedErrType:         validate.ErrRequestNotSet,\n\t\t\tdescription:             \"nil request should be rejected\",\n\t\t},\n\t\t{\n\t\t\tname:       \"When domain is empty it should return DomainNotSet error\",\n\t\t\tsetupMocks: func() {},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\texpectedErrType:         validate.ErrDomainNotSet,\n\t\t\tdescription:             \"empty domain should be rejected\",\n\t\t},\n\t\t{\n\t\t\tname:       \"When workflow execution is nil it should return ExecutionNotSet error\",\n\t\t\tsetupMocks: func() {},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain:            s.testDomain,\n\t\t\t\tWorkflowExecution: nil,\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\texpectedErrType:         validate.ErrExecutionNotSet,\n\t\t\tdescription:             \"nil workflow execution should be rejected\",\n\t\t},\n\t\t{\n\t\t\tname:       \"When workflow ID is empty it should return WorkflowIDNotSet error\",\n\t\t\tsetupMocks: func() {},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\texpectedErrType:         validate.ErrWorkflowIDNotSet,\n\t\t\tdescription:             \"empty workflow ID should be rejected\",\n\t\t},\n\t\t{\n\t\t\tname: \"When domain cache returns error it should propagate the error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(nil, errors.New(\"domain cache error\")).AnyTimes()\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"domain cache error\"))\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\tdescription:             \"domain cache errors should be propagated\",\n\t\t},\n\t\t{\n\t\t\tname: \"When isolation group is drained it should return BadRequest error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockResource.IsolationGroups.EXPECT().IsDrained(gomock.Any(), s.testDomain, \"dca1\").Return(true, nil)\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t\tIdentity: \"\",\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\tisolationGroup := \"dca1\"\n\t\t\t\treturn isolationgroup.ContextWithIsolationGroup(context.Background(), isolationGroup)\n\t\t\t},\n\t\t\tenableTaskListIsolation: true,\n\t\t\texpectError:             true,\n\t\t\texpectedErrType:         &types.BadRequestError{},\n\t\t\tdescription:             \"drained isolation group should be rejected\",\n\t\t},\n\t\t{\n\t\t\tname: \"When PollMutableState fails it should propagate the error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\t\t\t\ts.mockHistoryClient.EXPECT().PollMutableState(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"poll mutable state error\"))\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\tdescription:             \"PollMutableState errors should be propagated\",\n\t\t},\n\t\t{\n\t\t\tname: \"When history read fails it should propagate the error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\t\t\t\ts.mockHistoryClient.EXPECT().PollMutableState(gomock.Any(), gomock.Any()).Return(&types.PollMutableStateResponse{\n\t\t\t\t\tCurrentBranchToken: []byte(\"\"),\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tLastFirstEventID: 0,\n\t\t\t\t\tNextEventID:      2,\n\t\t\t\t\tVersionHistories: &types.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*types.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t\ts.mockVersionChecker.EXPECT().SupportsRawHistoryQuery(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(nil, errors.New(\"history read error\")).Once()\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\tdescription:             \"history read errors should be propagated\",\n\t\t},\n\t\t{\n\t\t\tname: \"When StartWorkflowExecution fails it should propagate the error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\t\t\t\ts.mockHistoryClient.EXPECT().PollMutableState(gomock.Any(), gomock.Any()).Return(&types.PollMutableStateResponse{\n\t\t\t\t\tCurrentBranchToken: []byte(\"\"),\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tLastFirstEventID: 0,\n\t\t\t\t\tNextEventID:      2,\n\t\t\t\t\tVersionHistories: &types.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*types.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t\ts.mockVersionChecker.EXPECT().SupportsRawHistoryQuery(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{&types.HistoryEvent{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\t\t\tName: \"workflowtype\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\t\tName: \"tasklist\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t}, nil).Once()\n\t\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"start workflow error\"))\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             true,\n\t\t\tdescription:             \"StartWorkflowExecution errors should be propagated\",\n\t\t},\n\t\t{\n\t\t\tname: \"When all conditions are valid it should successfully restart workflow\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\t\t\t\ts.mockHistoryClient.EXPECT().PollMutableState(gomock.Any(), gomock.Any()).Return(&types.PollMutableStateResponse{\n\t\t\t\t\tCurrentBranchToken: []byte(\"\"),\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tLastFirstEventID: 0,\n\t\t\t\t\tNextEventID:      2,\n\t\t\t\t\tVersionHistories: &types.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*types.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t\ts.mockVersionChecker.EXPECT().SupportsRawHistoryQuery(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{&types.HistoryEvent{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\t\t\tName: \"workflowtype\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\t\tName: \"tasklist\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t}, nil).Once()\n\t\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{\n\t\t\t\t\tRunID: testRunID,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t\tIdentity: \"\",\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             false,\n\t\t\texpectedRunID:           testRunID,\n\t\t\tdescription:             \"valid request should successfully restart workflow\",\n\t\t},\n\t\t{\n\t\t\tname: \"When ActiveClusterSelectionPolicy is preserved it should copy policy to new workflow\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\t\t\t\ts.mockHistoryClient.EXPECT().PollMutableState(gomock.Any(), gomock.Any()).Return(&types.PollMutableStateResponse{\n\t\t\t\t\tCurrentBranchToken: []byte(\"\"),\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tLastFirstEventID: 0,\n\t\t\t\t\tNextEventID:      2,\n\t\t\t\t\tVersionHistories: &types.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*types.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\t\t\t\t\tItems: []*types.VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t\ts.mockVersionChecker.EXPECT().SupportsRawHistoryQuery(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{&types.HistoryEvent{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\t\t\tName: \"workflowtype\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\t\tName: \"tasklist\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\t\t\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\t\t\t\t\t\t\tStickyRegion:                   \"us-west-1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t}, nil).Once()\n\t\t\t\t// Capture the start request to verify ActiveClusterSelectionPolicy\n\t\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\t\t\tfunc(ctx context.Context, request *types.HistoryStartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\t// Verify ActiveClusterSelectionPolicy is preserved\n\t\t\t\t\t\tif request.StartRequest.ActiveClusterSelectionPolicy == nil {\n\t\t\t\t\t\t\treturn nil, errors.New(\"expected ActiveClusterSelectionPolicy to be preserved\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif *request.StartRequest.ActiveClusterSelectionPolicy.ActiveClusterSelectionStrategy != types.ActiveClusterSelectionStrategyRegionSticky {\n\t\t\t\t\t\t\treturn nil, errors.New(\"ActiveClusterSelectionStrategy not preserved\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif request.StartRequest.ActiveClusterSelectionPolicy.StickyRegion != \"us-west-1\" {\n\t\t\t\t\t\t\treturn nil, errors.New(\"StickyRegion not preserved\")\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: testRunID}, nil\n\t\t\t\t\t},\n\t\t\t\t)\n\t\t\t},\n\t\t\trequest: &types.RestartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t},\n\t\t\t\tIdentity: \"\",\n\t\t\t},\n\t\t\tsetupContext: func() context.Context {\n\t\t\t\treturn context.Background()\n\t\t\t},\n\t\t\tenableTaskListIsolation: false,\n\t\t\texpectError:             false,\n\t\t\texpectedRunID:           testRunID,\n\t\t\tdescription:             \"ActiveClusterSelectionPolicy should be preserved in restarted workflow\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Setup dynamic client\n\t\t\tdynamicClient := dc.NewInMemoryClient()\n\t\t\terr := dynamicClient.UpdateValue(dynamicproperties.SendRawWorkflowHistory, false)\n\t\t\ts.NoError(err)\n\n\t\t\t// Setup config\n\t\t\tconfig := s.newConfig(dynamicClient)\n\t\t\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(tc.enableTaskListIsolation)\n\t\t\twh := s.getWorkflowHandler(config)\n\n\t\t\ttc.setupMocks()\n\t\t\tctx := tc.setupContext()\n\t\t\tresp, err := wh.RestartWorkflowExecution(ctx, tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\ts.Error(err, tc.description)\n\t\t\t\tif tc.expectedErrType != nil {\n\t\t\t\t\ts.IsType(tc.expectedErrType, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err, tc.description)\n\t\t\t\ts.NotNil(resp)\n\t\t\t\tif tc.expectedRunID != \"\" {\n\t\t\t\t\ts.Equal(tc.expectedRunID, resp.GetRunID())\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) getWorkflowExecutionHistory(nextEventID int64, transientDecision *types.TransientDecisionInfo, historyEvents []*types.HistoryEvent) {\n\tdynamicClient := dc.NewInMemoryClient()\n\terr := dynamicClient.UpdateValue(dynamicproperties.SendRawWorkflowHistory, true)\n\ts.NoError(err)\n\twh := s.getWorkflowHandler(\n\t\tfrontendcfg.NewConfig(\n\t\t\tdc.NewCollection(\n\t\t\t\tdynamicClient,\n\t\t\t\ts.mockResource.GetLogger()),\n\t\t\tnumHistoryShards,\n\t\t\tfalse,\n\t\t\t\"hostname\",\n\t\t\ts.mockResource.GetLogger(),\n\t\t),\n\t)\n\tctx := context.Background()\n\tdomainEntry := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: s.testDomainID, Name: s.testDomain}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(domainEntry, nil).AnyTimes()\n\ts.mockVersionChecker.EXPECT().SupportsRawHistoryQuery(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\tblob, _ := wh.GetPayloadSerializer().SerializeBatchEvents(historyEvents, constants.EncodingTypeThriftRW)\n\ts.mockHistoryV2Mgr.On(\"ReadRawHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadRawHistoryBranchResponse{\n\t\tHistoryEventBlobs: []*persistence.DataBlob{blob},\n\t\tNextPageToken:     []byte{},\n\t}, nil).Once()\n\ttoken, _ := json.Marshal(&getHistoryContinuationToken{\n\t\tFirstEventID:      1,\n\t\tNextEventID:       nextEventID,\n\t\tRunID:             testRunID,\n\t\tTransientDecision: transientDecision,\n\t})\n\tresp, err := wh.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.testDomain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tSkipArchival:  true,\n\t\tNextPageToken: token,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.NotNil(resp.RawHistory)\n\ts.Equal(2, len(resp.RawHistory))\n\n\tevents := deserializeBlobDataToHistoryEvents(wh, resp.RawHistory)\n\ts.NotNil(events)\n\tif transientDecision != nil {\n\t\ts.Equal(len(historyEvents)+2, len(events))\n\t} else {\n\t\ts.Equal(len(historyEvents), len(events))\n\t}\n}\n\nfunc deserializeBlobDataToHistoryEvents(wh *WorkflowHandler, dataBlobs []*types.DataBlob) []*types.HistoryEvent {\n\tvar historyEvents []*types.HistoryEvent\n\tfor _, batch := range dataBlobs {\n\t\tevents, err := wh.GetPayloadSerializer().DeserializeBatchEvents(&persistence.DataBlob{Data: batch.Data, Encoding: constants.EncodingTypeThriftRW})\n\t\tif err != nil {\n\t\t\treturn nil\n\t\t}\n\t\thistoryEvents = append(historyEvents, events...)\n\t}\n\treturn historyEvents\n}\n\nfunc (s *workflowHandlerSuite) TestListWorkflowExecutions() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\twh := s.getWorkflowHandler(config)\n\n\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\ts.mockVisibilityMgr.On(\"ListWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Once()\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.testDomain,\n\t\tPageSize: int32(config.ESIndexMaxResultWindow()),\n\t}\n\tctx := context.Background()\n\n\tquery := \"WorkflowID = 'wid'\"\n\tlistRequest.Query = query\n\t_, err := wh.ListWorkflowExecutions(ctx, listRequest)\n\ts.NoError(err)\n\ts.Equal(query, listRequest.GetQuery())\n\n\tquery = \"InvalidKey = 'a'\"\n\tlistRequest.Query = query\n\t_, err = wh.ListWorkflowExecutions(ctx, listRequest)\n\ts.NotNil(err)\n\n\tlistRequest.PageSize = int32(config.ESIndexMaxResultWindow() + 1)\n\t_, err = wh.ListWorkflowExecutions(ctx, listRequest)\n\ts.NotNil(err)\n}\n\nfunc (s *workflowHandlerSuite) TestScantWorkflowExecutions() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\twh := s.getWorkflowHandler(config)\n\n\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\ts.mockVisibilityMgr.On(\"ScanWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.ListWorkflowExecutionsResponse{}, nil).Once()\n\n\tlistRequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:   s.testDomain,\n\t\tPageSize: int32(config.ESIndexMaxResultWindow()),\n\t}\n\tctx := context.Background()\n\n\tquery := \"WorkflowID = 'wid'\"\n\tlistRequest.Query = query\n\t_, err := wh.ScanWorkflowExecutions(ctx, listRequest)\n\ts.NoError(err)\n\ts.Equal(query, listRequest.GetQuery())\n\n\tquery = \"InvalidKey = 'a'\"\n\tlistRequest.Query = query\n\t_, err = wh.ScanWorkflowExecutions(ctx, listRequest)\n\ts.NotNil(err)\n\n\tlistRequest.PageSize = int32(config.ESIndexMaxResultWindow() + 1)\n\t_, err = wh.ListWorkflowExecutions(ctx, listRequest)\n\ts.NotNil(err)\n}\n\nfunc (s *workflowHandlerSuite) TestCountWorkflowExecutions() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\n\ts.mockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(s.testDomainID, nil).AnyTimes()\n\ts.mockVisibilityMgr.On(\"CountWorkflowExecutions\", mock.Anything, mock.Anything).Return(&persistence.CountWorkflowExecutionsResponse{}, nil).Once()\n\n\tcountRequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: s.testDomain,\n\t}\n\tctx := context.Background()\n\n\tquery := \"WorkflowID = 'wid'\"\n\tcountRequest.Query = query\n\t_, err := wh.CountWorkflowExecutions(ctx, countRequest)\n\ts.NoError(err)\n\ts.Equal(query, countRequest.GetQuery())\n\n\tquery = \"InvalidKey = 'a'\"\n\tcountRequest.Query = query\n\t_, err = wh.CountWorkflowExecutions(ctx, countRequest)\n\ts.NotNil(err)\n}\n\nfunc (s *workflowHandlerSuite) TestConvertIndexedKeyToThrift() {\n\twh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))\n\tm := map[string]interface{}{\n\t\t\"key1\":  float64(0),\n\t\t\"key2\":  float64(1),\n\t\t\"key3\":  float64(2),\n\t\t\"key4\":  float64(3),\n\t\t\"key5\":  float64(4),\n\t\t\"key6\":  float64(5),\n\t\t\"key1i\": 0,\n\t\t\"key2i\": 1,\n\t\t\"key3i\": 2,\n\t\t\"key4i\": 3,\n\t\t\"key5i\": 4,\n\t\t\"key6i\": 5,\n\t\t\"key1t\": types.IndexedValueTypeString,\n\t\t\"key2t\": types.IndexedValueTypeKeyword,\n\t\t\"key3t\": types.IndexedValueTypeInt,\n\t\t\"key4t\": types.IndexedValueTypeDouble,\n\t\t\"key5t\": types.IndexedValueTypeBool,\n\t\t\"key6t\": types.IndexedValueTypeDatetime,\n\t\t\"key1s\": \"STRING\",\n\t\t\"key2s\": \"KEYWORD\",\n\t\t\"key3s\": \"INT\",\n\t\t\"key4s\": \"DOUBLE\",\n\t\t\"key5s\": \"BOOL\",\n\t\t\"key6s\": \"DATETIME\",\n\t}\n\tresult := wh.convertIndexedKeyToThrift(m)\n\ts.Equal(types.IndexedValueTypeString, result[\"key1\"])\n\ts.Equal(types.IndexedValueTypeKeyword, result[\"key2\"])\n\ts.Equal(types.IndexedValueTypeInt, result[\"key3\"])\n\ts.Equal(types.IndexedValueTypeDouble, result[\"key4\"])\n\ts.Equal(types.IndexedValueTypeBool, result[\"key5\"])\n\ts.Equal(types.IndexedValueTypeDatetime, result[\"key6\"])\n\ts.Equal(types.IndexedValueTypeString, result[\"key1i\"])\n\ts.Equal(types.IndexedValueTypeKeyword, result[\"key2i\"])\n\ts.Equal(types.IndexedValueTypeInt, result[\"key3i\"])\n\ts.Equal(types.IndexedValueTypeDouble, result[\"key4i\"])\n\ts.Equal(types.IndexedValueTypeBool, result[\"key5i\"])\n\ts.Equal(types.IndexedValueTypeDatetime, result[\"key6i\"])\n\ts.Equal(types.IndexedValueTypeString, result[\"key1t\"])\n\ts.Equal(types.IndexedValueTypeKeyword, result[\"key2t\"])\n\ts.Equal(types.IndexedValueTypeInt, result[\"key3t\"])\n\ts.Equal(types.IndexedValueTypeDouble, result[\"key4t\"])\n\ts.Equal(types.IndexedValueTypeBool, result[\"key5t\"])\n\ts.Equal(types.IndexedValueTypeDatetime, result[\"key6t\"])\n\ts.Equal(types.IndexedValueTypeString, result[\"key1s\"])\n\ts.Equal(types.IndexedValueTypeKeyword, result[\"key2s\"])\n\ts.Equal(types.IndexedValueTypeInt, result[\"key3s\"])\n\ts.Equal(types.IndexedValueTypeDouble, result[\"key4s\"])\n\ts.Equal(types.IndexedValueTypeBool, result[\"key5s\"])\n\ts.Equal(types.IndexedValueTypeDatetime, result[\"key6s\"])\n\ts.Panics(func() {\n\t\twh.convertIndexedKeyToThrift(map[string]interface{}{\n\t\t\t\"invalidType\": \"unknown\",\n\t\t})\n\t})\n}\n\nfunc (s *workflowHandlerSuite) TestVerifyHistoryIsComplete() {\n\tevents := make([]*types.HistoryEvent, 50)\n\tfor i := 0; i < len(events); i++ {\n\t\tevents[i] = &types.HistoryEvent{ID: int64(i + 1)}\n\t}\n\tvar eventsWithHoles []*types.HistoryEvent\n\teventsWithHoles = append(eventsWithHoles, events[9:12]...)\n\teventsWithHoles = append(eventsWithHoles, events[20:31]...)\n\n\ttestCases := []struct {\n\t\tevents       []*types.HistoryEvent\n\t\tfirstEventID int64\n\t\tlastEventID  int64\n\t\tisFirstPage  bool\n\t\tisLastPage   bool\n\t\tpageSize     int\n\t\tisResultErr  bool\n\t}{\n\t\t{events[:1], 1, 1, true, true, 1000, false},\n\t\t{events[:5], 1, 5, true, true, 1000, false},\n\t\t{events[9:31], 10, 31, true, true, 1000, false},\n\t\t{events[9:29], 10, 50, true, false, 20, false},\n\t\t{events[9:30], 10, 50, true, false, 20, false},\n\n\t\t{events[9:29], 1, 50, false, false, 20, false},\n\t\t{events[9:29], 1, 29, false, true, 20, false},\n\n\t\t{eventsWithHoles, 1, 50, false, false, 22, true},\n\t\t{eventsWithHoles, 10, 50, true, false, 22, true},\n\t\t{eventsWithHoles, 1, 31, false, true, 22, true},\n\t\t{eventsWithHoles, 10, 31, true, true, 1000, true},\n\n\t\t{events[9:31], 9, 31, true, true, 1000, true},\n\t\t{events[9:31], 9, 50, true, false, 22, true},\n\t\t{events[9:31], 11, 31, true, true, 1000, true},\n\t\t{events[9:31], 11, 50, true, false, 22, true},\n\n\t\t{events[9:31], 10, 30, true, true, 1000, true},\n\t\t{events[9:31], 1, 30, false, true, 22, true},\n\t\t{events[9:31], 10, 32, true, true, 1000, true},\n\t\t{events[9:31], 1, 32, false, true, 22, true},\n\t}\n\n\tfor i, tc := range testCases {\n\t\terr := verifyHistoryIsComplete(tc.events, tc.firstEventID, tc.lastEventID, tc.isFirstPage, tc.isLastPage, tc.pageSize)\n\t\tif tc.isResultErr {\n\t\t\ts.Error(err, \"testcase %v failed\", i)\n\t\t} else {\n\t\t\ts.NoError(err, \"testcase %v failed\", i)\n\t\t}\n\t}\n}\n\nfunc (s *workflowHandlerSuite) newConfig(dynamicClient dc.Client) *frontendcfg.Config {\n\tconfig := frontendcfg.NewConfig(\n\t\tdc.NewCollection(\n\t\t\tdynamicClient,\n\t\t\ts.mockResource.GetLogger(),\n\t\t),\n\t\tnumHistoryShards,\n\t\tfalse,\n\t\t\"hostname\",\n\t\ts.mockResource.GetLogger(),\n\t)\n\tconfig.EmitSignalNameMetricsTag = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\treturn config\n}\n\nfunc (s *workflowHandlerSuite) TestRespondActivityTaskFailedByID() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"activityID\",\n\t\tIdentity:   \"identity\",\n\t\tDetails:    make([]byte, 1000),\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest     *types.RespondActivityTaskFailedByIDRequest\n\t\texpectError bool\n\t\tmockFn      func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:     nil,\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain\": {\n\t\t\trequest: &types.RespondActivityTaskFailedByIDRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\trequest: &types.RespondActivityTaskFailedByIDRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.RespondActivityTaskFailedByIDRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty activity ID\": {\n\t\t\trequest: &types.RespondActivityTaskFailedByIDRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tActivityID: \"\",\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds id length limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1)\n\t\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"serialzation failure\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"return exceeds blob size limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(make([]byte, 100), nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskFailed(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(make([]byte, 100), nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskFailed(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.RespondActivityTaskFailedByID(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1000)\n\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestRespondActivityTaskCanceled() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidInput := &types.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: []byte(\"token\"),\n\t\tIdentity:  \"identity\",\n\t\tDetails:   make([]byte, 1000),\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput       *types.RespondActivityTaskCanceledRequest\n\t\tmockFn      func()\n\t\texpectError bool\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:       nil,\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty task token\": {\n\t\t\tinput: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\tTaskToken: nil,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"deserialzation failure\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: \"\"}, nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain name\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(\"\", errors.New(\"error getting domain name\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds id length limit\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1)\n\t\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds blob size limit\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskFailed(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"no error\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.RespondActivityTaskCanceled(context.Background(), input.input)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1000)\n\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestRespondActivityTaskCanceledByID() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidInput := &types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      testRunID,\n\t\tActivityID: \"activityID\",\n\t\tIdentity:   \"identity\",\n\t\tDetails:    make([]byte, 1000),\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest     *types.RespondActivityTaskCanceledByIDRequest\n\t\texpectError bool\n\t\tmockFn      func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:     nil,\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain name\": {\n\t\t\trequest: &types.RespondActivityTaskCanceledByIDRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\trequest: &types.RespondActivityTaskCanceledByIDRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.RespondActivityTaskCanceledByIDRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty activity ID\": {\n\t\t\trequest: &types.RespondActivityTaskCanceledByIDRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tActivityID: \"\",\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds id length limit\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1)\n\t\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"serialization failure\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds blob size limit\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(make([]byte, 100), nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(10)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(10)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskFailed(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(make([]byte, 5), nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\trequest: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return(make([]byte, 5), nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondActivityTaskCanceled(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.RespondActivityTaskCanceledByID(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1000)\n\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestRespondDecisionTaskCompleted() {\n\tvalidRequest := &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: []byte(\"token\"),\n\t\tIdentity:  \"identity\",\n\t\tDecisions: make([]*types.Decision, 100),\n\t}\n\tmockResp := &types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tStartedResponse: &types.RecordDecisionTaskStartedResponse{\n\t\t\tAttempt:          1,\n\t\t\tScheduledEventID: 2,\n\t\t},\n\t}\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\ttestInput := map[string]struct {\n\t\tinput           *types.RespondDecisionTaskCompletedRequest\n\t\tmockFn          func()\n\t\texpectError     bool\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:           nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"nil task token\": {\n\t\t\tinput: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken: nil,\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrTaskTokenNotSet,\n\t\t},\n\t\t\"deserialization failure\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: \"\"}, nil)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainNotSet,\n\t\t},\n\t\t\"cannot get domain name\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(\"\", errors.New(\"error getting domain name\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds id length limit\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1)\n\t\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrIdentityTooLong,\n\t\t},\n\t\t\"exceeds decision size limit\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.DecisionResultCountLimit = dynamicproperties.GetIntPropertyFilteredByDomain(10)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"no error\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any()).Return(mockResp, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t\"return new decision task true\": {\n\t\t\tinput: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken:             []byte(\"token\"),\n\t\t\t\tIdentity:              \"identity\",\n\t\t\t\tDecisions:             make([]*types.Decision, 100),\n\t\t\t\tReturnNewDecisionTask: true,\n\t\t\t},\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Serialize(gomock.Any()).Return([]byte(\"new task token\"), nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(s.testDomain, nil).Times(2)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), gomock.Any()).Return(mockResp, nil)\n\t\t\t\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents:    []*types.HistoryEvent{},\n\t\t\t\t\tNextPageToken:    []byte{},\n\t\t\t\t\tSize:             0,\n\t\t\t\t\tLastFirstEventID: 1,\n\t\t\t\t}, nil).Once()\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\t_, err := wh.RespondDecisionTaskCompleted(context.Background(), input.input)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1000)\n\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\twh.config.DecisionResultCountLimit = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestRespondDecisionTaskFailed() {\n\tvalidRequest := &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken: []byte(\"token\"),\n\t\tCause:     types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure.Ptr(),\n\t\tIdentity:  \"identity\",\n\t\tDetails:   make([]byte, 1000),\n\t}\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\ttestInput := map[string]struct {\n\t\tinput           *types.RespondDecisionTaskFailedRequest\n\t\tmockFn          func()\n\t\texpectError     bool\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:           nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"nil task token\": {\n\t\t\tinput: &types.RespondDecisionTaskFailedRequest{\n\t\t\t\tTaskToken: nil,\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrTaskTokenNotSet,\n\t\t},\n\t\t\"deserialization failure\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: \"\"}, nil)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainNotSet,\n\t\t},\n\t\t\"cannot get domain name\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(\"\", errors.New(\"error getting domain name\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceeds id length limit\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1)\n\t\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrIdentityTooLong,\n\t\t},\n\t\t\"exceeds blob size limit\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondDecisionTaskFailed(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\tinput: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{DomainID: s.testDomainID}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockHistoryClient.EXPECT().RespondDecisionTaskFailed(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.RespondDecisionTaskFailed(context.Background(), input.input)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.MaxIDLengthWarnLimit = dynamicproperties.GetIntPropertyFn(1000)\n\t\t\twh.config.IdentityMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestRespondQueryTaskCompleted() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidInput := &types.RespondQueryTaskCompletedRequest{\n\t\tTaskToken:   []byte(\"token\"),\n\t\tQueryResult: []byte(`{\"result\": \"result\"}`),\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput           *types.RespondQueryTaskCompletedRequest\n\t\tmockFn          func()\n\t\texpectError     bool\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\tinput:           nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"empty task token\": {\n\t\t\tinput: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\tTaskToken: nil,\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrTaskTokenNotSet,\n\t\t},\n\t\t\"deserialzation failure\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().DeserializeQueryTaskToken(gomock.Any()).Return(nil, errors.New(\"failed to deserialize token\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain ID\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().DeserializeQueryTaskToken(gomock.Any()).Return(&common.QueryTaskToken{DomainID: \"\"}, nil)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrInvalidTaskToken,\n\t\t},\n\t\t\"cannot get domain name\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().DeserializeQueryTaskToken(gomock.Any()).Return(&common.QueryTaskToken{\n\t\t\t\t\tDomainID: s.testDomainID,\n\t\t\t\t\tTaskList: \"tasklist\",\n\t\t\t\t\tTaskID:   \"taskID\"}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(\"\", errors.New(\"error getting domain name\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"exceed blob size limit and success\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().DeserializeQueryTaskToken(gomock.Any()).Return(&common.QueryTaskToken{\n\t\t\t\t\tDomainID: s.testDomainID,\n\t\t\t\t\tTaskList: \"tasklist\",\n\t\t\t\t\tTaskID:   \"taskID\"}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\ts.mockMatchingClient.EXPECT().RespondQueryTaskCompleted(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t\"matching client returns error\": {\n\t\t\tinput: validInput,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().DeserializeQueryTaskToken(gomock.Any()).Return(&common.QueryTaskToken{\n\t\t\t\t\tDomainID: s.testDomainID,\n\t\t\t\t\tTaskList: \"tasklist\",\n\t\t\t\t\tTaskID:   \"taskID\"}, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(s.testDomain, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\t\ts.mockMatchingClient.EXPECT().RespondQueryTaskCompleted(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.RespondQueryTaskCompleted(context.Background(), input.input)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestStartWorkflowExecution_Remaining() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     s.testDomain,\n\t\tWorkflowID: \"wid\",\n\t\tRequestID:  testRunID,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"wType\",\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"tasklist\",\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(100),\n\t}\n\ttestInput := map[string]struct {\n\t\tRequest         *types.StartWorkflowExecutionRequest\n\t\tMockFn          func()\n\t\tExpectError     bool\n\t\tExpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\tRequest: validRequest,\n\t\t\tMockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\tExpectError:     true,\n\t\t\tExpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\tRequest: validRequest,\n\t\t\tMockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"domainid\", nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\tExpectError: true,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\tRequest: validRequest,\n\t\t\tMockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\"))\n\t\t\t},\n\t\t\tExpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\tRequest: validRequest,\n\t\t\tMockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\tExpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.MockFn()\n\t\t\t_, err := wh.StartWorkflowExecution(context.Background(), input.Request)\n\t\t\tif input.ExpectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.ExpectErrorType != nil {\n\t\t\t\t\ts.Equal(input.ExpectErrorType, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t})\n\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestSignalWorkflowExecution() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.testDomain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tSignalName: \"signal\",\n\t\tRequestID:  testRunID,\n\t\tInput:      make([]byte, 1000),\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest         *types.SignalWorkflowExecutionRequest\n\t\texpectError     bool\n\t\tmockFn          func()\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:         nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"empty domain\": {\n\t\t\trequest: &types.SignalWorkflowExecutionRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainNotSet,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.SignalWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrWorkflowIDNotSet,\n\t\t},\n\t\t\"domain length exceeds limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.config.DomainNameMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainTooLong,\n\t\t},\n\t\t\"empty signal name\": {\n\t\t\trequest: &types.SignalWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t\tSignalName: \"\",\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrSignalNameNotSet,\n\t\t},\n\t\t\"signal name length exceeds limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.config.SignalNameMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrSignalNameTooLong,\n\t\t},\n\t\t\"requestID length exceeds limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.config.RequestIDMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestIDTooLong,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"input exceeds blob size limit\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client returns error\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.SignalWorkflowExecution(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t\twh.config.DomainNameMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(200)\n\t\t\twh.config.SignalNameMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(200)\n\t\t\twh.config.RequestIDMaxLength = dynamicproperties.GetIntPropertyFilteredByDomain(200)\n\t\t\twh.config.BlobSizeLimitWarn = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t\twh.config.BlobSizeLimitError = dynamicproperties.GetIntPropertyFilteredByDomain(1000)\n\t\t})\n\t}\n}\n\nfunc updateRequest(\n\thistoryArchivalURI *string,\n\thistoryArchivalStatus *types.ArchivalStatus,\n\tvisibilityArchivalURI *string,\n\tvisibilityArchivalStatus *types.ArchivalStatus,\n) *types.UpdateDomainRequest {\n\treturn &types.UpdateDomainRequest{\n\t\tName:                     \"test-name\",\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t}\n}\n\nfunc updateFailoverRequest(\n\thistoryArchivalURI *string,\n\thistoryArchivalStatus *types.ArchivalStatus,\n\tvisibilityArchivalURI *string,\n\tvisibilityArchivalStatus *types.ArchivalStatus,\n\tfailoverTimeoutInSeconds *int32,\n\tactiveClusterName *string,\n) *types.UpdateDomainRequest {\n\treturn &types.UpdateDomainRequest{\n\t\tName:                     \"test-name\",\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\tFailoverTimeoutInSeconds: failoverTimeoutInSeconds,\n\t\tActiveClusterName:        activeClusterName,\n\t}\n}\n\nfunc persistenceGetDomainResponse(historyArchivalState, visibilityArchivalState *domain.ArchivalState) *persistence.GetDomainResponse {\n\treturn &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          \"test-id\",\n\t\t\tName:        \"test-name\",\n\t\t\tStatus:      0,\n\t\t\tDescription: \"test-description\",\n\t\t\tOwnerEmail:  \"test-owner-email\",\n\t\t\tData:        make(map[string]string),\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tEmitMetric:               true,\n\t\t\tHistoryArchivalStatus:    historyArchivalState.Status,\n\t\t\tHistoryArchivalURI:       historyArchivalState.URI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalState.Status,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalState.URI,\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{\n\t\t\t\t\tClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tIsGlobalDomain:              false,\n\t\tConfigVersion:               0,\n\t\tFailoverVersion:             0,\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         0,\n\t}\n}\n\nfunc persistenceGetDomainResponseForFailoverTest(historyArchivalState, visibilityArchivalState *domain.ArchivalState) *persistence.GetDomainResponse {\n\treturn &persistence.GetDomainResponse{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          \"test-id\",\n\t\t\tName:        \"test-name\",\n\t\t\tStatus:      0,\n\t\t\tDescription: \"test-description\",\n\t\t\tOwnerEmail:  \"test-owner-email\",\n\t\t\tData:        make(map[string]string),\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tEmitMetric:               true,\n\t\t\tHistoryArchivalStatus:    historyArchivalState.Status,\n\t\t\tHistoryArchivalURI:       historyArchivalState.URI,\n\t\t\tVisibilityArchivalStatus: visibilityArchivalState.Status,\n\t\t\tVisibilityArchivalURI:    visibilityArchivalState.URI,\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{\n\t\t\t\t\tClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               0,\n\t\tFailoverVersion:             0,\n\t\tFailoverNotificationVersion: 0,\n\t\tNotificationVersion:         0,\n\t}\n}\n\nfunc registerDomainRequest(\n\thistoryArchivalStatus *types.ArchivalStatus,\n\thistoryArchivalURI string,\n\tvisibilityArchivalStatus *types.ArchivalStatus,\n\tvisibilityArchivalURI string,\n) *types.RegisterDomainRequest {\n\treturn &types.RegisterDomainRequest{\n\t\tName:                                   \"test-domain\",\n\t\tDescription:                            \"test-description\",\n\t\tOwnerEmail:                             \"test-owner-email\",\n\t\tWorkflowExecutionRetentionPeriodInDays: 10,\n\t\tEmitMetric:                             common.BoolPtr(true),\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{\n\t\t\t\tClusterName: cluster.TestCurrentClusterName,\n\t\t\t},\n\t\t},\n\t\tActiveClusterName:        cluster.TestCurrentClusterName,\n\t\tData:                     make(map[string]string),\n\t\tSecurityToken:            \"token\",\n\t\tHistoryArchivalStatus:    historyArchivalStatus,\n\t\tHistoryArchivalURI:       historyArchivalURI,\n\t\tVisibilityArchivalStatus: visibilityArchivalStatus,\n\t\tVisibilityArchivalURI:    visibilityArchivalURI,\n\t\tIsGlobalDomain:           false,\n\t}\n}\n\nfunc getHistoryRequest(nextPageToken []byte) *types.GetWorkflowExecutionHistoryRequest {\n\treturn &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tNextPageToken: nextPageToken,\n\t}\n}\n\nfunc listArchivedWorkflowExecutionsTestRequest() *types.ListArchivedWorkflowExecutionsRequest {\n\treturn &types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain:   \"some random domain name\",\n\t\tPageSize: 10,\n\t\tQuery:    \"some random query string\",\n\t}\n}\n\nvar describeDomainResponseServer = &types.DescribeDomainResponse{\n\tDomainInfo: &types.DomainInfo{\n\t\tName:        \"test-domain\",\n\t\tDescription: \"a test domain\",\n\t\tOwnerEmail:  \"test@uber.com\",\n\t},\n\tConfiguration: &types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\tEmitMetric:                             true,\n\t},\n\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{\n\t\t\t\tClusterName: cluster.TestCurrentClusterName,\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: cluster.TestAlternativeClusterName,\n\t\t\t},\n\t\t},\n\t},\n}\n\nfunc TestStartWorkflowExecutionAsync(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func(*MockProducerManager)\n\t\trequest    *types.StartWorkflowExecutionAsyncRequest\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(mockProducer, nil)\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t},\n\t\t\trequest: &types.StartWorkflowExecutionAsyncRequest{\n\t\t\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to get async queue producer\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(nil, errors.New(\"test-error\"))\n\t\t\t},\n\t\t\trequest: &types.StartWorkflowExecutionAsyncRequest{\n\t\t\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to publish message\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(mockProducer, nil)\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(errors.New(\"test-error\"))\n\t\t\t},\n\t\t\trequest: &types.StartWorkflowExecutionAsyncRequest{\n\t\t\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\t\t\tmockProducerManager := NewMockProducerManager(mockCtrl)\n\n\t\t\tcfg := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\tdc.NewInMemoryClient(),\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\t\t\twh := NewWorkflowHandler(mockResource, cfg, mockVersionChecker, nil)\n\t\t\twh.producerManager = mockProducerManager\n\n\t\t\ttc.setupMocks(mockProducerManager)\n\n\t\t\t_, err := wh.StartWorkflowExecutionAsync(context.Background(), tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecutionAsync(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func(*MockProducerManager)\n\t\trequest    *types.SignalWithStartWorkflowExecutionAsyncRequest\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(mockProducer, nil)\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t},\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\t\t\tSignalWithStartWorkflowExecutionRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t\tSignalName:                          \"test-signal-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to get async queue producer\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(nil, errors.New(\"test-error\"))\n\t\t\t},\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\t\t\tSignalWithStartWorkflowExecutionRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t\tSignalName:                          \"test-signal-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - failed to publish message\",\n\t\t\tsetupMocks: func(mockQueue *MockProducerManager) {\n\t\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\t\tmockQueue.EXPECT().GetProducerByDomain(gomock.Any()).Return(mockProducer, nil)\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.Anything).Return(errors.New(\"test-error\"))\n\t\t\t},\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\t\t\tSignalWithStartWorkflowExecutionRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"test-identity\",\n\t\t\t\t\tRequestID:                           uuid.New(),\n\t\t\t\t\tSignalName:                          \"test-signal-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\t\t\tmockProducerManager := NewMockProducerManager(mockCtrl)\n\n\t\t\tcfg := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\tdc.NewInMemoryClient(),\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\t\t\twh := NewWorkflowHandler(mockResource, cfg, mockVersionChecker, nil)\n\t\t\twh.producerManager = mockProducerManager\n\n\t\t\ttc.setupMocks(mockProducerManager)\n\n\t\t\t_, err := wh.SignalWithStartWorkflowExecutionAsync(context.Background(), tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRequestCancelWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMocks    func(checkerMock *client.MockVersionChecker, mockResource *resource.Test)\n\t\tcancelRequest *types.RequestCancelWorkflowExecutionRequest\n\t\tshuttingDown  int32\n\t\twantErr       bool\n\t\terr           error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tmockResource.HistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tcancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"Error case - is shutting down\",\n\t\t\tsetupMocks:   func(_ *client.MockVersionChecker, _ *resource.Test) {},\n\t\t\tshuttingDown: 1,\n\t\t\terr:          validate.ErrShuttingDown,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - error request not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\terr: validate.ErrRequestNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - domain name not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tcancelRequest: &types.RequestCancelWorkflowExecutionRequest{},\n\t\t\terr:           validate.ErrDomainNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - check execution error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tcancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\terr: validate.ErrExecutionNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - get domain ID error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"\", errors.New(\"get-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\tcancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"get-domain-id-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - RequestCancelWorkflowExecution error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tmockResource.HistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(errors.New(\"request-cancel-workflow-execution-error\")).Times(1)\n\t\t\t},\n\t\t\tcancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"request-cancel-workflow-execution-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\n\t\t\tcfg := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\tdc.NewInMemoryClient(),\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\t\t\twh := NewWorkflowHandler(mockResource, cfg, mockVersionChecker, nil)\n\t\t\twh.shuttingDown = tc.shuttingDown\n\n\t\t\ttc.setupMocks(mockVersionChecker, mockResource)\n\n\t\t\terr := wh.RequestCancelWorkflowExecution(context.Background(), tc.cancelRequest)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestQueryWorkflow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tsetupMocks     func(*client.MockVersionChecker, *resource.Test)\n\t\tqueryRequest   *types.QueryWorkflowRequest\n\t\tinMemoryClient dc.Client\n\t\tisShuttingDown int32\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t\tresourceMock.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tresourceMock.HistoryClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&types.HistoryQueryWorkflowResponse{\n\t\t\t\t\t\tResponse: &types.QueryWorkflowResponse{\n\t\t\t\t\t\t\tQueryResult: []byte(\"test-result\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\tQueryType: \"test-query-type\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Error case - is shutting down\",\n\t\t\tsetupMocks:     func(_ *client.MockVersionChecker, _ *resource.Test) {},\n\t\t\tisShuttingDown: 1,\n\t\t\terr:            validate.ErrShuttingDown,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - query request not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\terr:            validate.ErrRequestNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - domain not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest:   &types.QueryWorkflowRequest{},\n\t\t\terr:            validate.ErrDomainNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - check execution error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\terr: validate.ErrExecutionNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - query disallowed for domain\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: func() dc.Client {\n\t\t\t\tinMemoryClient := dc.NewInMemoryClient()\n\t\t\t\tinMemoryClient.UpdateValue(dynamicproperties.DisallowQuery, true)\n\t\t\t\treturn inMemoryClient\n\t\t\t}(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: validate.ErrQueryDisallowedForDomain,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - query not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: validate.ErrQueryNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - query type not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t\tQuery: &types.WorkflowQuery{},\n\t\t\t},\n\t\t\terr: validate.ErrQueryTypeNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - get domain ID error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t\tresourceMock.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"\", errors.New(\"get-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\tQueryType: \"test-query-type\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"get-domain-id-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - CheckEventBlobSizeLimit error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t\tresourceMock.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\tQueryType: \"test-query-type\",\n\t\t\t\t\tQueryArgs: []byte(\"test-query-args\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: common.ErrBlobSizeExceedsLimit,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - QueryWorkflow error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, resourceMock *resource.Test) {\n\t\t\t\tresourceMock.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tresourceMock.HistoryClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"query-workflow-error\")).Times(1)\n\t\t\t},\n\t\t\tinMemoryClient: dc.NewInMemoryClient(),\n\t\t\tqueryRequest: &types.QueryWorkflowRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\tQueryType: \"test-query-type\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"query-workflow-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\n\t\t\tcfg := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\ttc.inMemoryClient,\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\t\t\tcfg.BlobSizeLimitError = func(domain string) int { return 10 }\n\t\t\tcfg.BlobSizeLimitWarn = func(domain string) int { return 9 }\n\n\t\t\twh := NewWorkflowHandler(mockResource, cfg, mockVersionChecker, nil)\n\t\t\twh.shuttingDown = tc.isShuttingDown\n\n\t\t\ttc.setupMocks(mockVersionChecker, mockResource)\n\n\t\t\tqueryResult, err := wh.QueryWorkflow(context.Background(), tc.queryRequest)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, []byte(\"test-result\"), queryResult.QueryResult)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDescribeWorkflowExecution(t *testing.T) {\n\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: &types.WorkflowExecutionConfiguration{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: \"test-task-list\",\n\t\t\t},\n\t\t},\n\t}\n\n\ttestCases := []struct {\n\t\tname            string\n\t\tsetupMocks      func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test)\n\t\tdescribeRequest *types.DescribeWorkflowExecutionRequest\n\t\tisShuttingDown  int32\n\t\terr             error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tmockResource.HistoryClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\tdescribeRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Error case - is shutting down\",\n\t\t\tsetupMocks:     func(_ *client.MockVersionChecker, _ *resource.Test) {},\n\t\t\tisShuttingDown: 1,\n\t\t\terr:            validate.ErrShuttingDown,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - describe request not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\terr: validate.ErrRequestNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - domain name not set\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tdescribeRequest: &types.DescribeWorkflowExecutionRequest{},\n\t\t\terr:             validate.ErrDomainNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - check execution error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, _ *resource.Test) {\n\t\t\t},\n\t\t\tdescribeRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t},\n\t\t\terr: validate.ErrExecutionNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - get domain ID error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"\", errors.New(\"get-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\tdescribeRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"get-domain-id-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - DescribeWorkflowExecution error\",\n\t\t\tsetupMocks: func(mockVersionChecker *client.MockVersionChecker, mockResource *resource.Test) {\n\t\t\t\tmockResource.DomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"test-domain-id\", nil).Times(1)\n\t\t\t\tmockResource.HistoryClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"describe-workflow-execution-error\")).Times(1)\n\t\t\t},\n\t\t\tdescribeRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa\",\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: errors.New(\"describe-workflow-execution-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockResource := resource.NewTest(t, mockCtrl, metrics.Frontend)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\n\t\t\tcfg := frontendcfg.NewConfig(\n\t\t\t\tdc.NewCollection(\n\t\t\t\t\tdc.NewInMemoryClient(),\n\t\t\t\t\tmockResource.GetLogger(),\n\t\t\t\t),\n\t\t\t\tnumHistoryShards,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t\tmockResource.GetLogger(),\n\t\t\t)\n\n\t\t\twh := NewWorkflowHandler(mockResource, cfg, mockVersionChecker, nil)\n\t\t\twh.shuttingDown = tc.isShuttingDown\n\n\t\t\ttc.setupMocks(mockVersionChecker, mockResource)\n\n\t\t\tdescribeResponse, err := wh.DescribeWorkflowExecution(context.Background(), tc.describeRequest)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, resp, describeResponse)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestSignalWithStartWorkflowExecution() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              s.testDomain,\n\t\tWorkflowID:                          testWorkflowID,\n\t\tIdentity:                            \"identity\",\n\t\tSignalName:                          \"signal\",\n\t\tInput:                               nil,\n\t\tWorkflowType:                        &types.WorkflowType{Name: \"wType\"},\n\t\tTaskList:                            &types.TaskList{Name: \"taskList\"},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(10),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest     *types.SignalWithStartWorkflowExecutionRequest\n\t\texpectError bool\n\t\tmockFn      func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:     nil,\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty domain\": {\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomain:     s.testDomain,\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow type\": {\n\t\t\trequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomain:       s.testDomain,\n\t\t\t\tWorkflowID:   testWorkflowID,\n\t\t\t\tWorkflowType: nil,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client error\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil).Times(2)\n\t\t\t\ts.mockHistoryClient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil).Times(2)\n\t\t\t\ts.mockHistoryClient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\t_, err := wh.SignalWithStartWorkflowExecution(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestResetWorkflowExecution() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.testDomain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tReason: \"reason\",\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest         *types.ResetWorkflowExecutionRequest\n\t\texpectError     bool\n\t\tmockFn          func()\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:         nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"empty domain\": {\n\t\t\trequest: &types.ResetWorkflowExecutionRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainNotSet,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.ResetWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow execution\": {\n\t\t\trequest: &types.ResetWorkflowExecutionRequest{\n\t\t\t\tDomain:            s.testDomain,\n\t\t\t\tWorkflowExecution: nil,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client error\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\t_, err := wh.ResetWorkflowExecution(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestTerminateWorkflowExecution() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\tconfig.EnableClientVersionCheck = dynamicproperties.GetBoolPropertyFn(true)\n\twh := NewWorkflowHandler(s.mockResource, config, s.mockVersionChecker, nil)\n\twh.tokenSerializer = s.mockTokenSerializer\n\n\tvalidRequest := &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.testDomain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tReason:   \"reason\",\n\t\tDetails:  nil,\n\t\tIdentity: \"identity\",\n\t}\n\n\ttestInput := map[string]struct {\n\t\trequest         *types.TerminateWorkflowExecutionRequest\n\t\texpectError     bool\n\t\tmockFn          func()\n\t\texpectErrorType error\n\t}{\n\t\t\"shutting down\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\twh.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrShuttingDown,\n\t\t},\n\t\t\"nil request\": {\n\t\t\trequest:         nil,\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrRequestNotSet,\n\t\t},\n\t\t\"empty domain\": {\n\t\t\trequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\tmockFn:          func() {},\n\t\t\texpectError:     true,\n\t\t\texpectErrorType: validate.ErrDomainNotSet,\n\t\t},\n\t\t\"empty workflow ID\": {\n\t\t\trequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain: s.testDomain,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"empty workflow execution\": {\n\t\t\trequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain:            s.testDomain,\n\t\t\t\tWorkflowExecution: nil,\n\t\t\t},\n\t\t\tmockFn:      func() {},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"cannot get domain ID\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(\"\", errors.New(\"error getting domain ID\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"history client error\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(errors.New(\"error\"))\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t\"success\": {\n\t\t\trequest: validRequest,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(s.testDomain).Return(s.testDomainID, nil)\n\t\t\t\ts.mockHistoryClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := wh.TerminateWorkflowExecution(context.Background(), input.request)\n\t\t\tif input.expectError {\n\t\t\t\ts.Error(err)\n\t\t\t\tif input.expectErrorType != nil {\n\t\t\t\t\ts.ErrorIs(err, input.expectErrorType)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\twh.shuttingDown = int32(0)\n\t\t})\n\t}\n}\n\nfunc (s *workflowHandlerSuite) TestNormalizeVersionedErrors() {\n\tconfig := s.newConfig(dc.NewInMemoryClient())\n\twh := s.getWorkflowHandler(config)\n\n\tctx := yarpctest.ContextWithCall(context.Background(), &yarpctest.Call{\n\t\tHeaders: map[string]string{\n\t\t\tcommon.FeatureVersionHeaderName:     \"feature-version\",\n\t\t\tcommon.ClientImplHeaderName:         \"impl-header\",\n\t\t\tcommon.ClientFeatureFlagsHeaderName: \"\",\n\t\t},\n\t})\n\n\ts.mockVersionChecker.EXPECT().SupportsWorkflowAlreadyCompletedError(\"impl-header\", \"feature-version\", apiv1.FeatureFlags{}).Return(nil)\n\terr := wh.normalizeVersionedErrors(ctx, &types.WorkflowExecutionAlreadyCompletedError{})\n\ts.IsType(err, &types.WorkflowExecutionAlreadyCompletedError{})\n\n\ts.mockVersionChecker.EXPECT().SupportsWorkflowAlreadyCompletedError(\"impl-header\", \"feature-version\", apiv1.FeatureFlags{}).Return(errors.New(\"error\"))\n\terr = wh.normalizeVersionedErrors(ctx, &types.WorkflowExecutionAlreadyCompletedError{})\n\ts.IsType(err, &types.EntityNotExistsError{})\n}\n\nfunc TestWorkflowDescribeEmitStatusMetrics(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tres              *types.DescribeWorkflowExecutionResponse\n\t\terr              error\n\t\texpectedCounters map[string]tally.CounterSnapshot\n\t}{\n\t\t\"valid closed workflow\": {\n\t\t\tres: &types.DescribeWorkflowExecutionResponse{\n\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: common.Ptr(types.WorkflowExecutionCloseStatusCompleted),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedCounters: map[string]tally.CounterSnapshot{\n\t\t\t\t\"describe_wf_status+domain=some-domain,operation=DescribeWorkflowExecutionStatus,workflow_close_status=COMPLETED\": &counterSnapshotMock{\n\t\t\t\t\tname: \"describe_wf_status\",\n\t\t\t\t\ttags: map[string]string{\n\t\t\t\t\t\t\"domain\":                \"some-domain\",\n\t\t\t\t\t\t\"workflow_close_status\": \"COMPLETED\",\n\t\t\t\t\t\t\"operation\":             \"DescribeWorkflowExecutionStatus\",\n\t\t\t\t\t},\n\t\t\t\t\tvalue: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"A workflow not found\": {\n\t\t\tres: nil,\n\t\t\terr: &types.EntityNotExistsError{},\n\t\t\texpectedCounters: map[string]tally.CounterSnapshot{\n\t\t\t\t\"describe_wf_error+domain=some-domain,operation=DescribeWorkflowExecutionStatus\": &counterSnapshotMock{\n\t\t\t\t\tname: \"describe_wf_error\",\n\t\t\t\t\ttags: map[string]string{\n\t\t\t\t\t\t\"domain\":    \"some-domain\",\n\t\t\t\t\t\t\"operation\": \"DescribeWorkflowExecutionStatus\",\n\t\t\t\t\t},\n\t\t\t\t\tvalue: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"A invalid input 1\": {\n\t\t\tres: nil,\n\t\t\terr: nil,\n\t\t\texpectedCounters: map[string]tally.CounterSnapshot{\n\t\t\t\t\"describe_wf_error+domain=some-domain,operation=DescribeWorkflowExecutionStatus\": &counterSnapshotMock{\n\t\t\t\t\tname: \"describe_wf_error\",\n\t\t\t\t\ttags: map[string]string{\n\t\t\t\t\t\t\"domain\":    \"some-domain\",\n\t\t\t\t\t\t\"operation\": \"DescribeWorkflowExecutionStatus\",\n\t\t\t\t\t},\n\t\t\t\t\tvalue: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"invalid input 2\": {\n\t\t\tres: &types.DescribeWorkflowExecutionResponse{\n\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t// intentionally nil\n\t\t\t\t\t// CloseStatus: common.Ptr(types.WorkflowExecutionCloseStatusCompleted),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedCounters: map[string]tally.CounterSnapshot{\n\t\t\t\t\"describe_wf_status+domain=some-domain,operation=DescribeWorkflowExecutionStatus,workflow_close_status=unknown\": &counterSnapshotMock{\n\t\t\t\t\tname: \"describe_wf_status\",\n\t\t\t\t\ttags: map[string]string{\n\t\t\t\t\t\t\"domain\":                \"some-domain\",\n\t\t\t\t\t\t\"workflow_close_status\": \"unknown\",\n\t\t\t\t\t\t\"operation\":             \"DescribeWorkflowExecutionStatus\",\n\t\t\t\t\t},\n\t\t\t\t\tvalue: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tscope := tally.NewTestScope(\"\", nil)\n\t\t\tmockR := resource.Test{\n\t\t\t\tMetricsScope:  scope,\n\t\t\t\tMetricsClient: metrics.NewClient(scope, 1, metrics.HistogramMigration{}),\n\t\t\t}\n\n\t\t\twh := WorkflowHandler{\n\t\t\t\tResource: &mockR,\n\t\t\t}\n\n\t\t\twh.emitDescribeWorkflowExecutionMetrics(\"some-domain\", td.res, td.err)\n\t\t\tsnap := scope.Snapshot()\n\n\t\t\tfor k, v := range td.expectedCounters {\n\t\t\t\t_, ok := snap.Counters()[k]\n\t\t\t\tif !ok {\n\t\t\t\t\tt.Errorf(\"the metric string expected was not found. Expected a map with this key: %q\\ngot %v\", k, snap.Counters())\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\tassert.Equal(t, snap.Counters()[k].Name(), v.Name())\n\t\t\t\tassert.Equal(t, snap.Counters()[k].Value(), v.Value())\n\t\t\t\tassert.Equal(t, snap.Counters()[k].Tags(), v.Tags())\n\t\t\t}\n\t\t})\n\t}\n}\n\ntype counterSnapshotMock struct {\n\tname  string\n\ttags  map[string]string\n\tvalue int64\n}\n\nfunc (cs *counterSnapshotMock) Name() string            { return cs.name }\nfunc (cs *counterSnapshotMock) Tags() map[string]string { return cs.tags }\nfunc (cs *counterSnapshotMock) Value() int64            { return cs.value }\n\nfunc TestConstructRestartWorkflowRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname               string\n\t\toriginalAttributes *types.WorkflowExecutionStartedEventAttributes\n\t\tdomain             string\n\t\tidentity           string\n\t\tworkflowID         string\n\t\texpectPanic        bool\n\t\tdescription        string\n\t}{\n\t\t{\n\t\t\t// TODO(tim): This should not panic, return an error\n\t\t\tname:               \"nil originalAttributes should panic\",\n\t\t\toriginalAttributes: nil,\n\t\t\tdomain:             \"test-domain\",\n\t\t\tidentity:           \"test-identity\",\n\t\t\tworkflowID:         \"test-workflow-id\",\n\t\t\texpectPanic:        true,\n\t\t\tdescription:        \"nil originalAttributes should cause panic to prevent nil pointer dereference\",\n\t\t},\n\t\t{\n\t\t\t// TODO(tim): This should not panic, return an error\n\t\t\tname: \"nil WorkflowType should panic\",\n\t\t\toriginalAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType: nil,\n\t\t\t\tTaskList:     &types.TaskList{Name: \"testTaskList\"},\n\t\t\t},\n\t\t\tdomain:      \"test-domain\",\n\t\t\tidentity:    \"test-identity\",\n\t\t\tworkflowID:  \"test-workflow-id\",\n\t\t\texpectPanic: true,\n\t\t\tdescription: \"nil WorkflowType should cause panic\",\n\t\t},\n\t\t{\n\t\t\t// TODO(tim): This should not panic, return an error\n\t\t\tname: \"nil TaskList should panic\",\n\t\t\toriginalAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"testWorkflow\"},\n\t\t\t\tTaskList:     nil,\n\t\t\t},\n\t\t\tdomain:      \"test-domain\",\n\t\t\tidentity:    \"test-identity\",\n\t\t\tworkflowID:  \"test-workflow-id\",\n\t\t\texpectPanic: true,\n\t\t\tdescription: \"nil TaskList should cause panic\",\n\t\t},\n\t\t{\n\t\t\tname: \"complete field validation for restart request\",\n\t\t\toriginalAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"testWorkflow\"},\n\t\t\t\tTaskList:                            &types.TaskList{Name: \"testTaskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\tInput:                               []byte(\"test-input\"),\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(60),\n\t\t\t\tHeader:                              &types.Header{Fields: map[string][]byte{\"key\": []byte(\"value\")}},\n\t\t\t\tMemo:                                &types.Memo{Fields: map[string][]byte{\"memo\": []byte(\"data\")}},\n\t\t\t\tSearchAttributes:                    &types.SearchAttributes{IndexedFields: map[string][]byte{\"attr\": []byte(\"val\")}},\n\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\tInitialIntervalInSeconds:    1,\n\t\t\t\t\tBackoffCoefficient:          2.0,\n\t\t\t\t\tMaximumIntervalInSeconds:    10,\n\t\t\t\t\tMaximumAttempts:             3,\n\t\t\t\t\tExpirationIntervalInSeconds: 100,\n\t\t\t\t},\n\t\t\t\tCronSchedule:                    \"0 */2 * * *\",\n\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(30),\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\t\t\t\tStickyRegion:                   \"us-west-2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tdomain:      \"test-domain\",\n\t\t\tidentity:    \"test-identity\",\n\t\t\tworkflowID:  \"test-workflow-id\",\n\t\t\texpectPanic: false,\n\t\t\tdescription: \"complete field validation ensures all fields are properly set\",\n\t\t},\n\t\t{\n\t\t\tname: \"ActiveClusterSelectionPolicy with ExternalEntity strategy\",\n\t\t\toriginalAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"testWorkflow\"},\n\t\t\t\tTaskList:     &types.TaskList{Name: \"testTaskList\"},\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyExternalEntity.Ptr(),\n\t\t\t\t\tExternalEntityType:             \"order\",\n\t\t\t\t\tExternalEntityKey:              \"order-789\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tdomain:      \"test-domain\",\n\t\t\tidentity:    \"test-identity\",\n\t\t\tworkflowID:  \"test-workflow-id\",\n\t\t\texpectPanic: false,\n\t\t\tdescription: \"ExternalEntity policy should be completely preserved\",\n\t\t},\n\t\t{\n\t\t\tname: \"nil ActiveClusterSelectionPolicy should remain nil\",\n\t\t\toriginalAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType:                 &types.WorkflowType{Name: \"testWorkflow\"},\n\t\t\t\tTaskList:                     &types.TaskList{Name: \"testTaskList\"},\n\t\t\t\tActiveClusterSelectionPolicy: nil,\n\t\t\t},\n\t\t\tdomain:      \"test-domain\",\n\t\t\tidentity:    \"test-identity\",\n\t\t\tworkflowID:  \"test-workflow-id\",\n\t\t\texpectPanic: false,\n\t\t\tdescription: \"nil ActiveClusterSelectionPolicy should remain nil in restart request\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tif tc.expectPanic {\n\t\t\t\tassert.Panics(t, func() {\n\t\t\t\t\tconstructRestartWorkflowRequest(\n\t\t\t\t\t\ttc.originalAttributes,\n\t\t\t\t\t\ttc.domain,\n\t\t\t\t\t\ttc.identity,\n\t\t\t\t\t\ttc.workflowID,\n\t\t\t\t\t)\n\t\t\t\t}, tc.description)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Execute constructRestartWorkflowRequest\n\t\t\tstartRequest := constructRestartWorkflowRequest(\n\t\t\t\ttc.originalAttributes,\n\t\t\t\ttc.domain,\n\t\t\t\ttc.identity,\n\t\t\t\ttc.workflowID,\n\t\t\t)\n\n\t\t\t// Validate RequestID is a non-empty UUID\n\t\t\tassert.NotEmpty(t, startRequest.RequestID, \"RequestID should be non-empty\")\n\t\t\tparsedUUID := uuid.Parse(startRequest.RequestID)\n\t\t\tassert.NotNil(t, parsedUUID, \"RequestID should be a valid UUID\")\n\n\t\t\t// Validate input parameters are correctly set\n\t\t\tassert.Equal(t, tc.domain, startRequest.Domain, \"Domain should match input\")\n\t\t\tassert.Equal(t, tc.workflowID, startRequest.WorkflowID, \"WorkflowID should match input\")\n\t\t\tassert.Equal(t, tc.identity, startRequest.Identity, \"Identity should match input\")\n\n\t\t\t// Validate fields from workflow attributes\n\t\t\tassert.Equal(t, tc.originalAttributes.WorkflowType.Name, startRequest.WorkflowType.Name, \"WorkflowType.Name should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.TaskList.Name, startRequest.TaskList.Name, \"TaskList.Name should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.TaskList.Kind, startRequest.TaskList.Kind, \"TaskList.Kind should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.Input, startRequest.Input, \"Input should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.ExecutionStartToCloseTimeoutSeconds, startRequest.ExecutionStartToCloseTimeoutSeconds, \"ExecutionStartToCloseTimeoutSeconds should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.TaskStartToCloseTimeoutSeconds, startRequest.TaskStartToCloseTimeoutSeconds, \"TaskStartToCloseTimeoutSeconds should match\")\n\n\t\t\t// Validate WorkflowIDReusePolicy is correctly set\n\t\t\tassert.NotNil(t, startRequest.WorkflowIDReusePolicy, \"WorkflowIDReusePolicy should not be nil\")\n\t\t\tassert.Equal(t, types.WorkflowIDReusePolicyTerminateIfRunning, *startRequest.WorkflowIDReusePolicy, \"WorkflowIDReusePolicy should be TerminateIfRunning\")\n\n\t\t\t// Validate optional fields from workflow attributes\n\t\t\tassert.Equal(t, tc.originalAttributes.ActiveClusterSelectionPolicy, startRequest.ActiveClusterSelectionPolicy, \"ActiveClusterSelectionPolicy should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.CronSchedule, startRequest.CronSchedule, \"CronSchedule should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.RetryPolicy, startRequest.RetryPolicy, \"RetryPolicy should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.Header, startRequest.Header, \"Header should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.Memo, startRequest.Memo, \"Memo should match\")\n\t\t\tassert.Equal(t, tc.originalAttributes.SearchAttributes, startRequest.SearchAttributes, \"SearchAttributes should match\")\n\n\t\t\t// Validate DelayStartSeconds is set to 0 for restart requests\n\t\t\tassert.NotNil(t, startRequest.DelayStartSeconds, \"DelayStartSeconds should not be nil\")\n\t\t\tassert.Equal(t, int32(0), *startRequest.DelayStartSeconds, \"DelayStartSeconds should be 0 for restart requests\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/interface.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/service/frontend/api\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/accesscontrolled.tmpl -o ../wrappers/accesscontrolled/api_generated.go -v handler=API\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/clusterredirection.tmpl -o ../wrappers/clusterredirection/api_generated.go\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/versioncheck.tmpl -o ../wrappers/versioncheck/api_generated.go\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/metered.tmpl -o ../wrappers/metered/api_generated.go -v handler=API\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/ratelimited.tmpl -o ../wrappers/ratelimited/api_generated.go -v handler=API\n//go:generate gowrap gen -g -p . -i Handler -t ../../templates/grpc.tmpl -o ../wrappers/grpc/api_generated.go -v handler=API -v package=apiv1 -v path=github.com/uber/cadence-idl/go/proto/api/v1 -v prefix=\n//go:generate gowrap gen -g -p ../../../.gen/go/cadence/workflowserviceserver -i Interface -t ../../templates/thrift.tmpl -o ../wrappers/thrift/api_generated.go -v handler=API -v prefix=\n\npackage api\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// Handler is interface wrapping frontend handler\n\tHandler interface {\n\t\tHealth(context.Context) (*types.HealthStatus, error)\n\t\tCountWorkflowExecutions(context.Context, *types.CountWorkflowExecutionsRequest) (*types.CountWorkflowExecutionsResponse, error)\n\t\tDeleteDomain(context.Context, *types.DeleteDomainRequest) error\n\t\tDeprecateDomain(context.Context, *types.DeprecateDomainRequest) error\n\t\tDescribeDomain(context.Context, *types.DescribeDomainRequest) (*types.DescribeDomainResponse, error)\n\t\tDescribeTaskList(context.Context, *types.DescribeTaskListRequest) (*types.DescribeTaskListResponse, error)\n\t\tDescribeWorkflowExecution(context.Context, *types.DescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error)\n\t\tDiagnoseWorkflowExecution(context.Context, *types.DiagnoseWorkflowExecutionRequest) (*types.DiagnoseWorkflowExecutionResponse, error)\n\t\tGetClusterInfo(context.Context) (*types.ClusterInfo, error)\n\t\tGetSearchAttributes(context.Context) (*types.GetSearchAttributesResponse, error)\n\t\tGetWorkflowExecutionHistory(context.Context, *types.GetWorkflowExecutionHistoryRequest) (*types.GetWorkflowExecutionHistoryResponse, error)\n\t\tListArchivedWorkflowExecutions(context.Context, *types.ListArchivedWorkflowExecutionsRequest) (*types.ListArchivedWorkflowExecutionsResponse, error)\n\t\tListClosedWorkflowExecutions(context.Context, *types.ListClosedWorkflowExecutionsRequest) (*types.ListClosedWorkflowExecutionsResponse, error)\n\t\tListDomains(context.Context, *types.ListDomainsRequest) (*types.ListDomainsResponse, error)\n\t\tListOpenWorkflowExecutions(context.Context, *types.ListOpenWorkflowExecutionsRequest) (*types.ListOpenWorkflowExecutionsResponse, error)\n\t\tListTaskListPartitions(context.Context, *types.ListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error)\n\t\tGetTaskListsByDomain(context.Context, *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error)\n\t\tRefreshWorkflowTasks(context.Context, *types.RefreshWorkflowTasksRequest) error\n\t\tListWorkflowExecutions(context.Context, *types.ListWorkflowExecutionsRequest) (*types.ListWorkflowExecutionsResponse, error)\n\t\tPollForActivityTask(context.Context, *types.PollForActivityTaskRequest) (*types.PollForActivityTaskResponse, error)\n\t\tPollForDecisionTask(context.Context, *types.PollForDecisionTaskRequest) (*types.PollForDecisionTaskResponse, error)\n\t\tQueryWorkflow(context.Context, *types.QueryWorkflowRequest) (*types.QueryWorkflowResponse, error)\n\t\tRecordActivityTaskHeartbeat(context.Context, *types.RecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error)\n\t\tRecordActivityTaskHeartbeatByID(context.Context, *types.RecordActivityTaskHeartbeatByIDRequest) (*types.RecordActivityTaskHeartbeatResponse, error)\n\t\tRegisterDomain(context.Context, *types.RegisterDomainRequest) error\n\t\tRequestCancelWorkflowExecution(context.Context, *types.RequestCancelWorkflowExecutionRequest) error\n\t\tResetStickyTaskList(context.Context, *types.ResetStickyTaskListRequest) (*types.ResetStickyTaskListResponse, error)\n\t\tResetWorkflowExecution(context.Context, *types.ResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error)\n\t\tRespondActivityTaskCanceled(context.Context, *types.RespondActivityTaskCanceledRequest) error\n\t\tRespondActivityTaskCanceledByID(context.Context, *types.RespondActivityTaskCanceledByIDRequest) error\n\t\tRespondActivityTaskCompleted(context.Context, *types.RespondActivityTaskCompletedRequest) error\n\t\tRespondActivityTaskCompletedByID(context.Context, *types.RespondActivityTaskCompletedByIDRequest) error\n\t\tRespondActivityTaskFailed(context.Context, *types.RespondActivityTaskFailedRequest) error\n\t\tRespondActivityTaskFailedByID(context.Context, *types.RespondActivityTaskFailedByIDRequest) error\n\t\tRespondDecisionTaskCompleted(context.Context, *types.RespondDecisionTaskCompletedRequest) (*types.RespondDecisionTaskCompletedResponse, error)\n\t\tRespondDecisionTaskFailed(context.Context, *types.RespondDecisionTaskFailedRequest) error\n\t\tRespondQueryTaskCompleted(context.Context, *types.RespondQueryTaskCompletedRequest) error\n\t\tRestartWorkflowExecution(context.Context, *types.RestartWorkflowExecutionRequest) (*types.RestartWorkflowExecutionResponse, error)\n\t\tScanWorkflowExecutions(context.Context, *types.ListWorkflowExecutionsRequest) (*types.ListWorkflowExecutionsResponse, error)\n\t\tSignalWithStartWorkflowExecution(context.Context, *types.SignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\t\tSignalWithStartWorkflowExecutionAsync(context.Context, *types.SignalWithStartWorkflowExecutionAsyncRequest) (*types.SignalWithStartWorkflowExecutionAsyncResponse, error)\n\t\tSignalWorkflowExecution(context.Context, *types.SignalWorkflowExecutionRequest) error\n\t\tStartWorkflowExecution(context.Context, *types.StartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\t\tStartWorkflowExecutionAsync(context.Context, *types.StartWorkflowExecutionAsyncRequest) (*types.StartWorkflowExecutionAsyncResponse, error)\n\t\tTerminateWorkflowExecution(context.Context, *types.TerminateWorkflowExecutionRequest) error\n\t\tUpdateDomain(context.Context, *types.UpdateDomainRequest) (*types.UpdateDomainResponse, error)\n\t\tFailoverDomain(context.Context, *types.FailoverDomainRequest) (*types.FailoverDomainResponse, error)\n\t\tListFailoverHistory(context.Context, *types.ListFailoverHistoryRequest) (*types.ListFailoverHistoryResponse, error)\n\t}\n)\n"
  },
  {
    "path": "service/frontend/api/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package api -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/service/frontend/api\n//\n\n// Package api is a generated GoMock package.\npackage api\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// CountWorkflowExecutions mocks base method.\nfunc (m *MockHandler) CountWorkflowExecutions(arg0 context.Context, arg1 *types.CountWorkflowExecutionsRequest) (*types.CountWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.CountWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountWorkflowExecutions indicates an expected call of CountWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) CountWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).CountWorkflowExecutions), arg0, arg1)\n}\n\n// DeleteDomain mocks base method.\nfunc (m *MockHandler) DeleteDomain(arg0 context.Context, arg1 *types.DeleteDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteDomain\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteDomain indicates an expected call of DeleteDomain.\nfunc (mr *MockHandlerMockRecorder) DeleteDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDomain\", reflect.TypeOf((*MockHandler)(nil).DeleteDomain), arg0, arg1)\n}\n\n// DeprecateDomain mocks base method.\nfunc (m *MockHandler) DeprecateDomain(arg0 context.Context, arg1 *types.DeprecateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeprecateDomain\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeprecateDomain indicates an expected call of DeprecateDomain.\nfunc (mr *MockHandlerMockRecorder) DeprecateDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeprecateDomain\", reflect.TypeOf((*MockHandler)(nil).DeprecateDomain), arg0, arg1)\n}\n\n// DescribeDomain mocks base method.\nfunc (m *MockHandler) DescribeDomain(arg0 context.Context, arg1 *types.DescribeDomainRequest) (*types.DescribeDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeDomain\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeDomain indicates an expected call of DescribeDomain.\nfunc (mr *MockHandlerMockRecorder) DescribeDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeDomain\", reflect.TypeOf((*MockHandler)(nil).DescribeDomain), arg0, arg1)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockHandler) DescribeTaskList(arg0 context.Context, arg1 *types.DescribeTaskListRequest) (*types.DescribeTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockHandlerMockRecorder) DescribeTaskList(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockHandler)(nil).DescribeTaskList), arg0, arg1)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockHandler) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.DescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) DescribeWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).DescribeWorkflowExecution), arg0, arg1)\n}\n\n// DiagnoseWorkflowExecution mocks base method.\nfunc (m *MockHandler) DiagnoseWorkflowExecution(arg0 context.Context, arg1 *types.DiagnoseWorkflowExecutionRequest) (*types.DiagnoseWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DiagnoseWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DiagnoseWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DiagnoseWorkflowExecution indicates an expected call of DiagnoseWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) DiagnoseWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DiagnoseWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).DiagnoseWorkflowExecution), arg0, arg1)\n}\n\n// FailoverDomain mocks base method.\nfunc (m *MockHandler) FailoverDomain(arg0 context.Context, arg1 *types.FailoverDomainRequest) (*types.FailoverDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FailoverDomain\", arg0, arg1)\n\tret0, _ := ret[0].(*types.FailoverDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// FailoverDomain indicates an expected call of FailoverDomain.\nfunc (mr *MockHandlerMockRecorder) FailoverDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailoverDomain\", reflect.TypeOf((*MockHandler)(nil).FailoverDomain), arg0, arg1)\n}\n\n// GetClusterInfo mocks base method.\nfunc (m *MockHandler) GetClusterInfo(arg0 context.Context) (*types.ClusterInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClusterInfo\", arg0)\n\tret0, _ := ret[0].(*types.ClusterInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetClusterInfo indicates an expected call of GetClusterInfo.\nfunc (mr *MockHandlerMockRecorder) GetClusterInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClusterInfo\", reflect.TypeOf((*MockHandler)(nil).GetClusterInfo), arg0)\n}\n\n// GetSearchAttributes mocks base method.\nfunc (m *MockHandler) GetSearchAttributes(arg0 context.Context) (*types.GetSearchAttributesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSearchAttributes\", arg0)\n\tret0, _ := ret[0].(*types.GetSearchAttributesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetSearchAttributes indicates an expected call of GetSearchAttributes.\nfunc (mr *MockHandlerMockRecorder) GetSearchAttributes(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSearchAttributes\", reflect.TypeOf((*MockHandler)(nil).GetSearchAttributes), arg0)\n}\n\n// GetTaskListsByDomain mocks base method.\nfunc (m *MockHandler) GetTaskListsByDomain(arg0 context.Context, arg1 *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetTaskListsByDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListsByDomain indicates an expected call of GetTaskListsByDomain.\nfunc (mr *MockHandlerMockRecorder) GetTaskListsByDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListsByDomain\", reflect.TypeOf((*MockHandler)(nil).GetTaskListsByDomain), arg0, arg1)\n}\n\n// GetWorkflowExecutionHistory mocks base method.\nfunc (m *MockHandler) GetWorkflowExecutionHistory(arg0 context.Context, arg1 *types.GetWorkflowExecutionHistoryRequest) (*types.GetWorkflowExecutionHistoryResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecutionHistory\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetWorkflowExecutionHistoryResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecutionHistory indicates an expected call of GetWorkflowExecutionHistory.\nfunc (mr *MockHandlerMockRecorder) GetWorkflowExecutionHistory(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecutionHistory\", reflect.TypeOf((*MockHandler)(nil).GetWorkflowExecutionHistory), arg0, arg1)\n}\n\n// Health mocks base method.\nfunc (m *MockHandler) Health(arg0 context.Context) (*types.HealthStatus, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Health\", arg0)\n\tret0, _ := ret[0].(*types.HealthStatus)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Health indicates an expected call of Health.\nfunc (mr *MockHandlerMockRecorder) Health(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Health\", reflect.TypeOf((*MockHandler)(nil).Health), arg0)\n}\n\n// ListArchivedWorkflowExecutions mocks base method.\nfunc (m *MockHandler) ListArchivedWorkflowExecutions(arg0 context.Context, arg1 *types.ListArchivedWorkflowExecutionsRequest) (*types.ListArchivedWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListArchivedWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListArchivedWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListArchivedWorkflowExecutions indicates an expected call of ListArchivedWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) ListArchivedWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListArchivedWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).ListArchivedWorkflowExecutions), arg0, arg1)\n}\n\n// ListClosedWorkflowExecutions mocks base method.\nfunc (m *MockHandler) ListClosedWorkflowExecutions(arg0 context.Context, arg1 *types.ListClosedWorkflowExecutionsRequest) (*types.ListClosedWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListClosedWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListClosedWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListClosedWorkflowExecutions indicates an expected call of ListClosedWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) ListClosedWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListClosedWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).ListClosedWorkflowExecutions), arg0, arg1)\n}\n\n// ListDomains mocks base method.\nfunc (m *MockHandler) ListDomains(arg0 context.Context, arg1 *types.ListDomainsRequest) (*types.ListDomainsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListDomains\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListDomainsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListDomains indicates an expected call of ListDomains.\nfunc (mr *MockHandlerMockRecorder) ListDomains(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListDomains\", reflect.TypeOf((*MockHandler)(nil).ListDomains), arg0, arg1)\n}\n\n// ListFailoverHistory mocks base method.\nfunc (m *MockHandler) ListFailoverHistory(arg0 context.Context, arg1 *types.ListFailoverHistoryRequest) (*types.ListFailoverHistoryResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListFailoverHistory\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListFailoverHistoryResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListFailoverHistory indicates an expected call of ListFailoverHistory.\nfunc (mr *MockHandlerMockRecorder) ListFailoverHistory(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListFailoverHistory\", reflect.TypeOf((*MockHandler)(nil).ListFailoverHistory), arg0, arg1)\n}\n\n// ListOpenWorkflowExecutions mocks base method.\nfunc (m *MockHandler) ListOpenWorkflowExecutions(arg0 context.Context, arg1 *types.ListOpenWorkflowExecutionsRequest) (*types.ListOpenWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListOpenWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListOpenWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListOpenWorkflowExecutions indicates an expected call of ListOpenWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) ListOpenWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListOpenWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).ListOpenWorkflowExecutions), arg0, arg1)\n}\n\n// ListTaskListPartitions mocks base method.\nfunc (m *MockHandler) ListTaskListPartitions(arg0 context.Context, arg1 *types.ListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListTaskListPartitionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskListPartitions indicates an expected call of ListTaskListPartitions.\nfunc (mr *MockHandlerMockRecorder) ListTaskListPartitions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskListPartitions\", reflect.TypeOf((*MockHandler)(nil).ListTaskListPartitions), arg0, arg1)\n}\n\n// ListWorkflowExecutions mocks base method.\nfunc (m *MockHandler) ListWorkflowExecutions(arg0 context.Context, arg1 *types.ListWorkflowExecutionsRequest) (*types.ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListWorkflowExecutions indicates an expected call of ListWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) ListWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).ListWorkflowExecutions), arg0, arg1)\n}\n\n// PollForActivityTask mocks base method.\nfunc (m *MockHandler) PollForActivityTask(arg0 context.Context, arg1 *types.PollForActivityTaskRequest) (*types.PollForActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.PollForActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForActivityTask indicates an expected call of PollForActivityTask.\nfunc (mr *MockHandlerMockRecorder) PollForActivityTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForActivityTask\", reflect.TypeOf((*MockHandler)(nil).PollForActivityTask), arg0, arg1)\n}\n\n// PollForDecisionTask mocks base method.\nfunc (m *MockHandler) PollForDecisionTask(arg0 context.Context, arg1 *types.PollForDecisionTaskRequest) (*types.PollForDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.PollForDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForDecisionTask indicates an expected call of PollForDecisionTask.\nfunc (mr *MockHandlerMockRecorder) PollForDecisionTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForDecisionTask\", reflect.TypeOf((*MockHandler)(nil).PollForDecisionTask), arg0, arg1)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockHandler) QueryWorkflow(arg0 context.Context, arg1 *types.QueryWorkflowRequest) (*types.QueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", arg0, arg1)\n\tret0, _ := ret[0].(*types.QueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockHandlerMockRecorder) QueryWorkflow(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockHandler)(nil).QueryWorkflow), arg0, arg1)\n}\n\n// RecordActivityTaskHeartbeat mocks base method.\nfunc (m *MockHandler) RecordActivityTaskHeartbeat(arg0 context.Context, arg1 *types.RecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeat indicates an expected call of RecordActivityTaskHeartbeat.\nfunc (mr *MockHandlerMockRecorder) RecordActivityTaskHeartbeat(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeat\", reflect.TypeOf((*MockHandler)(nil).RecordActivityTaskHeartbeat), arg0, arg1)\n}\n\n// RecordActivityTaskHeartbeatByID mocks base method.\nfunc (m *MockHandler) RecordActivityTaskHeartbeatByID(arg0 context.Context, arg1 *types.RecordActivityTaskHeartbeatByIDRequest) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeatByID\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeatByID indicates an expected call of RecordActivityTaskHeartbeatByID.\nfunc (mr *MockHandlerMockRecorder) RecordActivityTaskHeartbeatByID(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeatByID\", reflect.TypeOf((*MockHandler)(nil).RecordActivityTaskHeartbeatByID), arg0, arg1)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockHandler) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.RefreshWorkflowTasksRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockHandlerMockRecorder) RefreshWorkflowTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockHandler)(nil).RefreshWorkflowTasks), arg0, arg1)\n}\n\n// RegisterDomain mocks base method.\nfunc (m *MockHandler) RegisterDomain(arg0 context.Context, arg1 *types.RegisterDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RegisterDomain\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RegisterDomain indicates an expected call of RegisterDomain.\nfunc (mr *MockHandlerMockRecorder) RegisterDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterDomain\", reflect.TypeOf((*MockHandler)(nil).RegisterDomain), arg0, arg1)\n}\n\n// RequestCancelWorkflowExecution mocks base method.\nfunc (m *MockHandler) RequestCancelWorkflowExecution(arg0 context.Context, arg1 *types.RequestCancelWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestCancelWorkflowExecution indicates an expected call of RequestCancelWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) RequestCancelWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).RequestCancelWorkflowExecution), arg0, arg1)\n}\n\n// ResetStickyTaskList mocks base method.\nfunc (m *MockHandler) ResetStickyTaskList(arg0 context.Context, arg1 *types.ResetStickyTaskListRequest) (*types.ResetStickyTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ResetStickyTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetStickyTaskList indicates an expected call of ResetStickyTaskList.\nfunc (mr *MockHandlerMockRecorder) ResetStickyTaskList(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetStickyTaskList\", reflect.TypeOf((*MockHandler)(nil).ResetStickyTaskList), arg0, arg1)\n}\n\n// ResetWorkflowExecution mocks base method.\nfunc (m *MockHandler) ResetWorkflowExecution(arg0 context.Context, arg1 *types.ResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ResetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) ResetWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).ResetWorkflowExecution), arg0, arg1)\n}\n\n// RespondActivityTaskCanceled mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCanceled(arg0 context.Context, arg1 *types.RespondActivityTaskCanceledRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceled indicates an expected call of RespondActivityTaskCanceled.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCanceled(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceled\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCanceled), arg0, arg1)\n}\n\n// RespondActivityTaskCanceledByID mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCanceledByID(arg0 context.Context, arg1 *types.RespondActivityTaskCanceledByIDRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceledByID\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceledByID indicates an expected call of RespondActivityTaskCanceledByID.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCanceledByID(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceledByID\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCanceledByID), arg0, arg1)\n}\n\n// RespondActivityTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCompleted(arg0 context.Context, arg1 *types.RespondActivityTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompleted indicates an expected call of RespondActivityTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCompleted), arg0, arg1)\n}\n\n// RespondActivityTaskCompletedByID mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCompletedByID(arg0 context.Context, arg1 *types.RespondActivityTaskCompletedByIDRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompletedByID\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompletedByID indicates an expected call of RespondActivityTaskCompletedByID.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCompletedByID(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompletedByID\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCompletedByID), arg0, arg1)\n}\n\n// RespondActivityTaskFailed mocks base method.\nfunc (m *MockHandler) RespondActivityTaskFailed(arg0 context.Context, arg1 *types.RespondActivityTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailed indicates an expected call of RespondActivityTaskFailed.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskFailed(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailed\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskFailed), arg0, arg1)\n}\n\n// RespondActivityTaskFailedByID mocks base method.\nfunc (m *MockHandler) RespondActivityTaskFailedByID(arg0 context.Context, arg1 *types.RespondActivityTaskFailedByIDRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailedByID\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailedByID indicates an expected call of RespondActivityTaskFailedByID.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskFailedByID(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailedByID\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskFailedByID), arg0, arg1)\n}\n\n// RespondDecisionTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondDecisionTaskCompleted(arg0 context.Context, arg1 *types.RespondDecisionTaskCompletedRequest) (*types.RespondDecisionTaskCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RespondDecisionTaskCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondDecisionTaskCompleted indicates an expected call of RespondDecisionTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondDecisionTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondDecisionTaskCompleted), arg0, arg1)\n}\n\n// RespondDecisionTaskFailed mocks base method.\nfunc (m *MockHandler) RespondDecisionTaskFailed(arg0 context.Context, arg1 *types.RespondDecisionTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondDecisionTaskFailed indicates an expected call of RespondDecisionTaskFailed.\nfunc (mr *MockHandlerMockRecorder) RespondDecisionTaskFailed(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskFailed\", reflect.TypeOf((*MockHandler)(nil).RespondDecisionTaskFailed), arg0, arg1)\n}\n\n// RespondQueryTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondQueryTaskCompleted(arg0 context.Context, arg1 *types.RespondQueryTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondQueryTaskCompleted indicates an expected call of RespondQueryTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondQueryTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondQueryTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondQueryTaskCompleted), arg0, arg1)\n}\n\n// RestartWorkflowExecution mocks base method.\nfunc (m *MockHandler) RestartWorkflowExecution(arg0 context.Context, arg1 *types.RestartWorkflowExecutionRequest) (*types.RestartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RestartWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RestartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RestartWorkflowExecution indicates an expected call of RestartWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) RestartWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RestartWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).RestartWorkflowExecution), arg0, arg1)\n}\n\n// ScanWorkflowExecutions mocks base method.\nfunc (m *MockHandler) ScanWorkflowExecutions(arg0 context.Context, arg1 *types.ListWorkflowExecutionsRequest) (*types.ListWorkflowExecutionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ScanWorkflowExecutions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListWorkflowExecutionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ScanWorkflowExecutions indicates an expected call of ScanWorkflowExecutions.\nfunc (mr *MockHandlerMockRecorder) ScanWorkflowExecutions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScanWorkflowExecutions\", reflect.TypeOf((*MockHandler)(nil).ScanWorkflowExecutions), arg0, arg1)\n}\n\n// SignalWithStartWorkflowExecution mocks base method.\nfunc (m *MockHandler) SignalWithStartWorkflowExecution(arg0 context.Context, arg1 *types.SignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecution indicates an expected call of SignalWithStartWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) SignalWithStartWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).SignalWithStartWorkflowExecution), arg0, arg1)\n}\n\n// SignalWithStartWorkflowExecutionAsync mocks base method.\nfunc (m *MockHandler) SignalWithStartWorkflowExecutionAsync(arg0 context.Context, arg1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (*types.SignalWithStartWorkflowExecutionAsyncResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecutionAsync\", arg0, arg1)\n\tret0, _ := ret[0].(*types.SignalWithStartWorkflowExecutionAsyncResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecutionAsync indicates an expected call of SignalWithStartWorkflowExecutionAsync.\nfunc (mr *MockHandlerMockRecorder) SignalWithStartWorkflowExecutionAsync(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecutionAsync\", reflect.TypeOf((*MockHandler)(nil).SignalWithStartWorkflowExecutionAsync), arg0, arg1)\n}\n\n// SignalWorkflowExecution mocks base method.\nfunc (m *MockHandler) SignalWorkflowExecution(arg0 context.Context, arg1 *types.SignalWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SignalWorkflowExecution indicates an expected call of SignalWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) SignalWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).SignalWorkflowExecution), arg0, arg1)\n}\n\n// StartWorkflowExecution mocks base method.\nfunc (m *MockHandler) StartWorkflowExecution(arg0 context.Context, arg1 *types.StartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecution indicates an expected call of StartWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) StartWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).StartWorkflowExecution), arg0, arg1)\n}\n\n// StartWorkflowExecutionAsync mocks base method.\nfunc (m *MockHandler) StartWorkflowExecutionAsync(arg0 context.Context, arg1 *types.StartWorkflowExecutionAsyncRequest) (*types.StartWorkflowExecutionAsyncResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"StartWorkflowExecutionAsync\", arg0, arg1)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionAsyncResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecutionAsync indicates an expected call of StartWorkflowExecutionAsync.\nfunc (mr *MockHandlerMockRecorder) StartWorkflowExecutionAsync(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecutionAsync\", reflect.TypeOf((*MockHandler)(nil).StartWorkflowExecutionAsync), arg0, arg1)\n}\n\n// TerminateWorkflowExecution mocks base method.\nfunc (m *MockHandler) TerminateWorkflowExecution(arg0 context.Context, arg1 *types.TerminateWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TerminateWorkflowExecution indicates an expected call of TerminateWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) TerminateWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TerminateWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).TerminateWorkflowExecution), arg0, arg1)\n}\n\n// UpdateDomain mocks base method.\nfunc (m *MockHandler) UpdateDomain(arg0 context.Context, arg1 *types.UpdateDomainRequest) (*types.UpdateDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomain\", arg0, arg1)\n\tret0, _ := ret[0].(*types.UpdateDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateDomain indicates an expected call of UpdateDomain.\nfunc (mr *MockHandlerMockRecorder) UpdateDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomain\", reflect.TypeOf((*MockHandler)(nil).UpdateDomain), arg0, arg1)\n}\n"
  },
  {
    "path": "service/frontend/api/list_workflow_handlers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\n// CountWorkflowExecutions - count number of workflow executions in a domain\nfunc (wh *WorkflowHandler) CountWorkflowExecutions(\n\tctx context.Context,\n\tcountRequest *types.CountWorkflowExecutionsRequest,\n) (resp *types.CountWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateCountWorkflowExecutionsRequest(ctx, countRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tvalidatedQuery, err := wh.visibilityQueryValidator.ValidateQuery(countRequest.GetQuery())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain := countRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treq := &persistence.CountWorkflowExecutionsRequest{\n\t\tDomainUUID: domainID,\n\t\tDomain:     domain,\n\t\tQuery:      validatedQuery,\n\t}\n\tpersistenceResp, err := wh.GetVisibilityManager().CountWorkflowExecutions(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp = &types.CountWorkflowExecutionsResponse{\n\t\tCount: persistenceResp.Count,\n\t}\n\treturn resp, nil\n}\n\n// ScanWorkflowExecutions - retrieves info for large amount of workflow executions in a domain without order\nfunc (wh *WorkflowHandler) ScanWorkflowExecutions(\n\tctx context.Context,\n\tlistRequest *types.ListWorkflowExecutionsRequest,\n) (resp *types.ListWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListWorkflowExecutionsRequest(ctx, listRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tvalidatedQuery, err := wh.visibilityQueryValidator.ValidateQuery(listRequest.GetQuery())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain := listRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treq := &persistence.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID:    domainID,\n\t\tDomain:        domain,\n\t\tPageSize:      int(listRequest.GetPageSize()),\n\t\tNextPageToken: listRequest.NextPageToken,\n\t\tQuery:         validatedQuery,\n\t}\n\tpersistenceResp, err := wh.GetVisibilityManager().ScanWorkflowExecutions(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp = &types.ListWorkflowExecutionsResponse{}\n\tresp.Executions = persistenceResp.Executions\n\tresp.NextPageToken = persistenceResp.NextPageToken\n\treturn resp, nil\n}\n\n// ListOpenWorkflowExecutions - retrieves info for open workflow executions in a domain\nfunc (wh *WorkflowHandler) ListOpenWorkflowExecutions(\n\tctx context.Context,\n\tlistRequest *types.ListOpenWorkflowExecutionsRequest,\n) (resp *types.ListOpenWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListOpenWorkflowExecutionsRequest(ctx, listRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tdomain := listRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tbaseReq := persistence.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:    domainID,\n\t\tDomain:        domain,\n\t\tPageSize:      int(listRequest.GetMaximumPageSize()),\n\t\tNextPageToken: listRequest.NextPageToken,\n\t\tEarliestTime:  listRequest.StartTimeFilter.GetEarliestTime(),\n\t\tLatestTime:    listRequest.StartTimeFilter.GetLatestTime(),\n\t}\n\n\tvar persistenceResp *persistence.ListWorkflowExecutionsResponse\n\tif listRequest.ExecutionFilter != nil {\n\t\tif wh.config.DisableListVisibilityByFilter(domain) {\n\t\t\terr = validate.ErrNoPermission\n\t\t} else {\n\t\t\tpersistenceResp, err = wh.GetVisibilityManager().ListOpenWorkflowExecutionsByWorkflowID(\n\t\t\t\tctx,\n\t\t\t\t&persistence.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: baseReq,\n\t\t\t\t\tWorkflowID:                    listRequest.ExecutionFilter.GetWorkflowID(),\n\t\t\t\t})\n\t\t}\n\t\twh.GetLogger().Debug(\"List open workflow with filter\",\n\t\t\ttag.WorkflowDomainName(listRequest.GetDomain()), tag.WorkflowListWorkflowFilterByID)\n\t} else if listRequest.TypeFilter != nil {\n\t\tif wh.config.DisableListVisibilityByFilter(domain) {\n\t\t\terr = validate.ErrNoPermission\n\t\t} else {\n\t\t\tpersistenceResp, err = wh.GetVisibilityManager().ListOpenWorkflowExecutionsByType(\n\t\t\t\tctx,\n\t\t\t\t&persistence.ListWorkflowExecutionsByTypeRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: baseReq,\n\t\t\t\t\tWorkflowTypeName:              listRequest.TypeFilter.GetName(),\n\t\t\t\t},\n\t\t\t)\n\t\t}\n\t\twh.GetLogger().Debug(\"List open workflow with filter\",\n\t\t\ttag.WorkflowDomainName(listRequest.GetDomain()), tag.WorkflowListWorkflowFilterByType)\n\t} else {\n\t\tpersistenceResp, err = wh.GetVisibilityManager().ListOpenWorkflowExecutions(ctx, &baseReq)\n\t}\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp = &types.ListOpenWorkflowExecutionsResponse{}\n\tresp.Executions = persistenceResp.Executions\n\tresp.NextPageToken = persistenceResp.NextPageToken\n\treturn resp, nil\n}\n\n// ListArchivedWorkflowExecutions - retrieves archived info for closed workflow executions in a domain\nfunc (wh *WorkflowHandler) ListArchivedWorkflowExecutions(\n\tctx context.Context,\n\tlistRequest *types.ListArchivedWorkflowExecutionsRequest,\n) (resp *types.ListArchivedWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListArchivedWorkflowExecutionsRequest(ctx, listRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tif !wh.GetArchivalMetadata().GetVisibilityConfig().ClusterConfiguredForArchival() {\n\t\treturn nil, &types.BadRequestError{Message: \"Cluster is not configured for visibility archival\"}\n\t}\n\n\tif !wh.GetArchivalMetadata().GetVisibilityConfig().ReadEnabled() {\n\t\treturn nil, &types.BadRequestError{Message: \"Cluster is not configured for reading archived visibility records\"}\n\t}\n\n\tentry, err := wh.GetDomainCache().GetDomain(listRequest.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif entry.GetConfig().VisibilityArchivalStatus != types.ArchivalStatusEnabled {\n\t\treturn nil, &types.BadRequestError{Message: \"Domain is not configured for visibility archival\"}\n\t}\n\n\tURI, err := archiver.NewURI(entry.GetConfig().VisibilityArchivalURI)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvisibilityArchiver, err := wh.GetArchiverProvider().GetVisibilityArchiver(URI.Scheme(), service.Frontend)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tarchiverRequest := &archiver.QueryVisibilityRequest{\n\t\tDomainID:      entry.GetInfo().ID,\n\t\tPageSize:      int(listRequest.GetPageSize()),\n\t\tNextPageToken: listRequest.NextPageToken,\n\t\tQuery:         listRequest.GetQuery(),\n\t}\n\n\tarchiverResponse, err := visibilityArchiver.Query(ctx, URI, archiverRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// special handling of ExecutionTime for cron or retry\n\tfor _, execution := range archiverResponse.Executions {\n\t\tif execution.GetExecutionTime() == 0 {\n\t\t\texecution.ExecutionTime = common.Int64Ptr(execution.GetStartTime())\n\t\t}\n\t}\n\n\treturn &types.ListArchivedWorkflowExecutionsResponse{\n\t\tExecutions:    archiverResponse.Executions,\n\t\tNextPageToken: archiverResponse.NextPageToken,\n\t}, nil\n}\n\n// ListClosedWorkflowExecutions - retrieves info for closed workflow executions in a domain\nfunc (wh *WorkflowHandler) ListClosedWorkflowExecutions(\n\tctx context.Context,\n\tlistRequest *types.ListClosedWorkflowExecutionsRequest,\n) (resp *types.ListClosedWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListClosedWorkflowExecutionsRequest(ctx, listRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tdomain := listRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tbaseReq := persistence.ListWorkflowExecutionsRequest{\n\t\tDomainUUID:    domainID,\n\t\tDomain:        domain,\n\t\tPageSize:      int(listRequest.GetMaximumPageSize()),\n\t\tNextPageToken: listRequest.NextPageToken,\n\t\tEarliestTime:  listRequest.StartTimeFilter.GetEarliestTime(),\n\t\tLatestTime:    listRequest.StartTimeFilter.GetLatestTime(),\n\t}\n\n\tvar persistenceResp *persistence.ListWorkflowExecutionsResponse\n\tif listRequest.ExecutionFilter != nil {\n\t\tif wh.config.DisableListVisibilityByFilter(domain) {\n\t\t\terr = validate.ErrNoPermission\n\t\t} else {\n\t\t\tpersistenceResp, err = wh.GetVisibilityManager().ListClosedWorkflowExecutionsByWorkflowID(\n\t\t\t\tctx,\n\t\t\t\t&persistence.ListWorkflowExecutionsByWorkflowIDRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: baseReq,\n\t\t\t\t\tWorkflowID:                    listRequest.ExecutionFilter.GetWorkflowID(),\n\t\t\t\t},\n\t\t\t)\n\t\t}\n\t\twh.GetLogger().Debug(\"List closed workflow with filter\",\n\t\t\ttag.WorkflowDomainName(listRequest.GetDomain()), tag.WorkflowListWorkflowFilterByID)\n\t} else if listRequest.TypeFilter != nil {\n\t\tif wh.config.DisableListVisibilityByFilter(domain) {\n\t\t\terr = validate.ErrNoPermission\n\t\t} else {\n\t\t\tpersistenceResp, err = wh.GetVisibilityManager().ListClosedWorkflowExecutionsByType(\n\t\t\t\tctx,\n\t\t\t\t&persistence.ListWorkflowExecutionsByTypeRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: baseReq,\n\t\t\t\t\tWorkflowTypeName:              listRequest.TypeFilter.GetName(),\n\t\t\t\t},\n\t\t\t)\n\t\t}\n\t\twh.GetLogger().Debug(\"List closed workflow with filter\",\n\t\t\ttag.WorkflowDomainName(listRequest.GetDomain()), tag.WorkflowListWorkflowFilterByType)\n\t} else if listRequest.StatusFilter != nil {\n\t\tif wh.config.DisableListVisibilityByFilter(domain) {\n\t\t\terr = validate.ErrNoPermission\n\t\t} else {\n\t\t\tpersistenceResp, err = wh.GetVisibilityManager().ListClosedWorkflowExecutionsByStatus(\n\t\t\t\tctx,\n\t\t\t\t&persistence.ListClosedWorkflowExecutionsByStatusRequest{\n\t\t\t\t\tListWorkflowExecutionsRequest: baseReq,\n\t\t\t\t\tStatus:                        listRequest.GetStatusFilter(),\n\t\t\t\t},\n\t\t\t)\n\t\t}\n\t\twh.GetLogger().Debug(\"List closed workflow with filter\",\n\t\t\ttag.WorkflowDomainName(listRequest.GetDomain()), tag.WorkflowListWorkflowFilterByStatus)\n\t} else {\n\t\tpersistenceResp, err = wh.GetVisibilityManager().ListClosedWorkflowExecutions(ctx, &baseReq)\n\t}\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp = &types.ListClosedWorkflowExecutionsResponse{}\n\tresp.Executions = persistenceResp.Executions\n\tresp.NextPageToken = persistenceResp.NextPageToken\n\treturn resp, nil\n}\n\n// ListWorkflowExecutions - retrieves info for workflow executions in a domain\nfunc (wh *WorkflowHandler) ListWorkflowExecutions(\n\tctx context.Context,\n\tlistRequest *types.ListWorkflowExecutionsRequest,\n) (resp *types.ListWorkflowExecutionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListWorkflowExecutionsRequest(ctx, listRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tvalidatedQuery, err := wh.visibilityQueryValidator.ValidateQuery(listRequest.GetQuery())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain := listRequest.GetDomain()\n\tdomainID, err := wh.GetDomainCache().GetDomainID(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treq := &persistence.ListWorkflowExecutionsByQueryRequest{\n\t\tDomainUUID:    domainID,\n\t\tDomain:        domain,\n\t\tPageSize:      int(listRequest.GetPageSize()),\n\t\tNextPageToken: listRequest.NextPageToken,\n\t\tQuery:         validatedQuery,\n\t}\n\tpersistenceResp, err := wh.GetVisibilityManager().ListWorkflowExecutions(ctx, req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp = &types.ListWorkflowExecutionsResponse{}\n\tresp.Executions = persistenceResp.Executions\n\tresp.NextPageToken = persistenceResp.NextPageToken\n\treturn resp, nil\n}\n"
  },
  {
    "path": "service/frontend/api/producer_manager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination producer_manager_mock.go -self_package github.com/uber/cadence/service/frontend/api\n\npackage api\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ProducerManager is used to create a producer for a domain.\n\t// Producer is used for Async APIs such as StartWorkflowExecutionAsync\n\tProducerManager interface {\n\t\tGetProducerByDomain(domain string) (messaging.Producer, error)\n\t}\n\n\tproducerManagerImpl struct {\n\t\tdomainCache   cache.DomainCache\n\t\tprovider      queue.Provider\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\n\t\tproducerCache cache.Cache\n\t}\n)\n\nfunc NewProducerManager(\n\tdomainCache cache.DomainCache,\n\tprovider queue.Provider,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) ProducerManager {\n\treturn &producerManagerImpl{\n\t\tdomainCache:   domainCache,\n\t\tprovider:      provider,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t\tproducerCache: cache.New(&cache.Options{\n\t\t\tTTL:             time.Minute * 5,\n\t\t\tInitialCapacity: 5,\n\t\t\tMaxCount:        100,\n\t\t\tPin:             true,\n\t\t\tMetricsScope:    metricsClient.Scope(metrics.PersistenceGetShardScope), // was metrics.Frontend, using incorrect int\n\t\t\tLogger:          logger,\n\t\t}),\n\t}\n}\n\n// GetProducerByDomain returns a producer for a domain\nfunc (q *producerManagerImpl) GetProducerByDomain(\n\tdomain string,\n) (messaging.Producer, error) {\n\tdomainEntry, err := q.domainCache.GetDomain(domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !domainEntry.GetConfig().AsyncWorkflowConfig.Enabled {\n\t\treturn nil, &types.BadRequestError{Message: fmt.Sprintf(\"async workflow is not enabled for domain %v\", domain)}\n\t}\n\tqueueName := domainEntry.GetConfig().AsyncWorkflowConfig.PredefinedQueueName\n\tvar queue provider.Queue\n\tif queueName != \"\" {\n\t\tqueue, err = q.provider.GetPredefinedQueue(queueName)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tqueue, err = q.provider.GetQueue(domainEntry.GetConfig().AsyncWorkflowConfig.QueueType, domainEntry.GetConfig().AsyncWorkflowConfig.QueueConfig)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tqueueID := queue.ID()\n\tval := q.producerCache.Get(queueID)\n\tif val != nil {\n\t\treturn val.(messaging.Producer), nil\n\t}\n\n\tproducer, err := queue.CreateProducer(&provider.Params{Logger: q.logger, MetricsClient: q.metricsClient})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// PutIfNotExist is thread safe, and will either return the value that was already in the cache or the value we just created\n\t// another thread might have inserted a value between the Get and PutIfNotExist, but that is ok\n\t// it should never return an error as we do not use Pin\n\tval, err = q.producerCache.PutIfNotExist(queueID, producer)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn val.(messaging.Producer), nil\n}\n"
  },
  {
    "path": "service/frontend/api/producer_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: producer_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package api -source producer_manager.go -destination producer_manager_mock.go -self_package github.com/uber/cadence/service/frontend/api\n//\n\n// Package api is a generated GoMock package.\npackage api\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tmessaging \"github.com/uber/cadence/common/messaging\"\n)\n\n// MockProducerManager is a mock of ProducerManager interface.\ntype MockProducerManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProducerManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockProducerManagerMockRecorder is the mock recorder for MockProducerManager.\ntype MockProducerManagerMockRecorder struct {\n\tmock *MockProducerManager\n}\n\n// NewMockProducerManager creates a new mock instance.\nfunc NewMockProducerManager(ctrl *gomock.Controller) *MockProducerManager {\n\tmock := &MockProducerManager{ctrl: ctrl}\n\tmock.recorder = &MockProducerManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProducerManager) EXPECT() *MockProducerManagerMockRecorder {\n\treturn m.recorder\n}\n\n// GetProducerByDomain mocks base method.\nfunc (m *MockProducerManager) GetProducerByDomain(domain string) (messaging.Producer, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetProducerByDomain\", domain)\n\tret0, _ := ret[0].(messaging.Producer)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetProducerByDomain indicates an expected call of GetProducerByDomain.\nfunc (mr *MockProducerManagerMockRecorder) GetProducerByDomain(domain any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetProducerByDomain\", reflect.TypeOf((*MockProducerManager)(nil).GetProducerByDomain), domain)\n}\n"
  },
  {
    "path": "service/frontend/api/producer_manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetProducerByDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tdomain    string\n\t\tmockSetup func(*cache.MockDomainCache, *queue.MockProvider, *provider.MockQueue, *cache.MockCache)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:   \"Success case - cache miss, predefined queue\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\t\tPredefinedQueueName: \"testQueue\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetPredefinedQueue(\"testQueue\").Return(mockQueue, nil)\n\t\t\t\tmockQueue.EXPECT().ID().Return(\"q1\")\n\t\t\t\tmockProducerCache.EXPECT().Get(gomock.Any()).Return(nil)\n\t\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\t\tmockQueue.EXPECT().CreateProducer(gomock.Any()).Return(producer, nil)\n\t\t\t\tmockProducerCache.EXPECT().PutIfNotExist(\"q1\", producer).Return(producer, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"Success case - cache miss, customized queue\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetQueue(\"kafka\", gomock.Any()).Return(mockQueue, nil)\n\t\t\t\tmockQueue.EXPECT().ID().Return(\"q1\")\n\t\t\t\tmockProducerCache.EXPECT().Get(gomock.Any()).Return(nil)\n\t\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\t\tmockQueue.EXPECT().CreateProducer(gomock.Any()).Return(producer, nil)\n\t\t\t\tmockProducerCache.EXPECT().PutIfNotExist(\"q1\", producer).Return(producer, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"Success case - cache hit, predefined queue\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\t\tPredefinedQueueName: \"testQueue\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetPredefinedQueue(\"testQueue\").Return(mockQueue, nil)\n\t\t\t\tmockQueue.EXPECT().ID().Return(\"q1\")\n\t\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\t\tmockProducerCache.EXPECT().Get(gomock.Any()).Return(producer)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"Success case - cache hit, customized queue\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetQueue(\"kafka\", gomock.Any()).Return(mockQueue, nil)\n\t\t\t\tmockQueue.EXPECT().ID().Return(\"q1\")\n\t\t\t\tproducer := messaging.NewNoopProducer()\n\t\t\t\tmockProducerCache.EXPECT().Get(gomock.Any()).Return(producer)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"Error case - domain cache error\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(nil, fmt.Errorf(\"error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:   \"Error case - provider error\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetQueue(\"kafka\", gomock.Any()).Return(nil, fmt.Errorf(\"error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:   \"Error case - cache miss, create producer error\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t\tmockProvider.EXPECT().GetQueue(\"kafka\", gomock.Any()).Return(mockQueue, nil)\n\t\t\t\tmockQueue.EXPECT().ID().Return(\"q1\")\n\t\t\t\tmockProducerCache.EXPECT().Get(gomock.Any()).Return(nil)\n\t\t\t\tmockQueue.EXPECT().CreateProducer(gomock.Any()).Return(nil, fmt.Errorf(\"error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:   \"Error case - async wf not enabled\",\n\t\t\tdomain: \"test-domain\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockProvider *queue.MockProvider, mockQueue *provider.MockQueue, mockProducerCache *cache.MockCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(\"test-domain\").Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tAsyncWorkflowConfig: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\t\tEnabled: false,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockProvider := queue.NewMockProvider(mockCtrl)\n\t\t\tmockQueue := provider.NewMockQueue(mockCtrl)\n\t\t\tmockProducerCache := cache.NewMockCache(mockCtrl)\n\n\t\t\tproducerManager := NewProducerManager(\n\t\t\t\tmockDomainCache,\n\t\t\t\tmockProvider,\n\t\t\t\tlog.NewNoop(),\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t)\n\t\t\tproducerManager.(*producerManagerImpl).producerCache = mockProducerCache\n\n\t\t\ttc.mockSetup(mockDomainCache, mockProvider, mockQueue, mockProducerCache)\n\n\t\t\tproducer, err := producerManager.GetProducerByDomain(tc.domain)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, producer)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/refresh_workflow_tasks.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\n// RefreshWorkflowTasks re-generates the workflow tasks\nfunc (wh *WorkflowHandler) RefreshWorkflowTasks(\n\tctx context.Context,\n\trequest *types.RefreshWorkflowTasksRequest,\n) error {\n\tif wh.isShuttingDown() {\n\t\treturn validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateRefreshWorkflowTasksRequest(ctx, request); err != nil {\n\t\treturn err\n\t}\n\tdomainEntry, err := wh.GetDomainCache().GetDomain(request.GetDomain())\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = wh.GetHistoryClient().RefreshWorkflowTasks(ctx, &types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: domainEntry.GetInfo().ID,\n\t\tRequest:    request,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/frontend/api/refresh_workflow_tasks_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\nvar testDomainCacheEntry = cache.NewLocalDomainCacheEntryForTest(\n\t&persistence.DomainInfo{Name: \"domain\", ID: \"domain-id\"},\n\t&persistence.DomainConfig{},\n\t\"\",\n)\n\ntype mockDeps struct {\n\tmockResource           *resource.Test\n\tmockDomainCache        *cache.MockDomainCache\n\tmockHistoryClient      *history.MockClient\n\tmockMatchingClient     *matching.MockClient\n\tmockProducer           *mocks.KafkaProducer\n\tmockMessagingClient    messaging.Client\n\tmockMetadataMgr        *mocks.MetadataManager\n\tmockHistoryV2Mgr       *mocks.HistoryV2Manager\n\tmockVisibilityMgr      *mocks.VisibilityManager\n\tmockArchivalMetadata   *archiver.MockArchivalMetadata\n\tmockArchiverProvider   *provider.MockArchiverProvider\n\tmockHistoryArchiver    *archiver.HistoryArchiverMock\n\tmockVisibilityArchiver *archiver.VisibilityArchiverMock\n\tmockVersionChecker     *client.MockVersionChecker\n\tmockTokenSerializer    *common.MockTaskTokenSerializer\n\tmockDomainHandler      *domain.MockHandler\n\tmockRequestValidator   *MockRequestValidator\n\tdynamicClient          dynamicconfig.Client\n}\n\nfunc setupMocksForWorkflowHandler(t *testing.T) (*WorkflowHandler, *mockDeps) {\n\tctrl := gomock.NewController(t)\n\tmockResource := resource.NewTest(t, ctrl, metrics.Frontend)\n\tmockProducer := &mocks.KafkaProducer{}\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\tdeps := &mockDeps{\n\t\tmockResource:         mockResource,\n\t\tmockDomainCache:      mockResource.DomainCache,\n\t\tmockHistoryClient:    mockResource.HistoryClient,\n\t\tmockMatchingClient:   mockResource.MatchingClient,\n\t\tmockMetadataMgr:      mockResource.MetadataMgr,\n\t\tmockHistoryV2Mgr:     mockResource.HistoryMgr,\n\t\tmockVisibilityMgr:    mockResource.VisibilityMgr,\n\t\tmockArchivalMetadata: mockResource.ArchivalMetadata,\n\t\tmockArchiverProvider: mockResource.ArchiverProvider,\n\t\tmockTokenSerializer:  common.NewMockTaskTokenSerializer(ctrl),\n\n\t\tmockProducer:           mockProducer,\n\t\tmockMessagingClient:    mocks.NewMockMessagingClient(mockProducer, nil),\n\t\tmockHistoryArchiver:    &archiver.HistoryArchiverMock{},\n\t\tmockVisibilityArchiver: &archiver.VisibilityArchiverMock{},\n\t\tmockVersionChecker:     client.NewMockVersionChecker(ctrl),\n\t\tmockDomainHandler:      domain.NewMockHandler(ctrl),\n\t\tmockRequestValidator:   NewMockRequestValidator(ctrl),\n\t\tdynamicClient:          dynamicClient,\n\t}\n\n\tlogger := testlogger.New(t)\n\tconfig := frontendcfg.NewConfig(\n\t\tdynamicconfig.NewCollection(\n\t\t\tdynamicClient,\n\t\t\tlogger,\n\t\t),\n\t\tnumHistoryShards,\n\t\tfalse,\n\t\t\"hostname\",\n\t\tlogger,\n\t)\n\twh := NewWorkflowHandler(deps.mockResource, config, deps.mockVersionChecker, deps.mockDomainHandler)\n\twh.requestValidator = deps.mockRequestValidator\n\treturn wh, deps\n}\n\nfunc TestRefreshWorkflowTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.RefreshWorkflowTasksRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRefreshWorkflowTasksRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomain(\"domain\").Return(testDomainCacheEntry, nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().RefreshWorkflowTasks(gomock.Any(), &types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\tDomainUIID: \"domain-id\",\n\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"history client error\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRefreshWorkflowTasksRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomain(\"domain\").Return(testDomainCacheEntry, nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().RefreshWorkflowTasks(gomock.Any(), &types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\tDomainUIID: \"domain-id\",\n\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(errors.New(\"history error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"history error\",\n\t\t},\n\t\t{\n\t\t\tname: \"cache error\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRefreshWorkflowTasksRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomain(\"domain\").Return(nil, errors.New(\"cache error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"validator error\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateRefreshWorkflowTasksRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validator error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validator error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\t\t\terr := wh.RefreshWorkflowTasks(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/request_validator.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination request_validator_mock.go -self_package github.com/uber/cadence/service/frontend/api requestValidator\n\npackage api\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\ntype (\n\tRequestValidator interface {\n\t\tValidateRefreshWorkflowTasksRequest(context.Context, *types.RefreshWorkflowTasksRequest) error\n\t\tValidateDescribeTaskListRequest(context.Context, *types.DescribeTaskListRequest) error\n\t\tValidateListTaskListPartitionsRequest(context.Context, *types.ListTaskListPartitionsRequest) error\n\t\tValidateGetTaskListsByDomainRequest(context.Context, *types.GetTaskListsByDomainRequest) error\n\t\tValidateResetStickyTaskListRequest(context.Context, *types.ResetStickyTaskListRequest) error\n\t\tValidateCountWorkflowExecutionsRequest(context.Context, *types.CountWorkflowExecutionsRequest) error\n\t\tValidateListWorkflowExecutionsRequest(context.Context, *types.ListWorkflowExecutionsRequest) error\n\t\tValidateListOpenWorkflowExecutionsRequest(context.Context, *types.ListOpenWorkflowExecutionsRequest) error\n\t\tValidateListArchivedWorkflowExecutionsRequest(context.Context, *types.ListArchivedWorkflowExecutionsRequest) error\n\t\tValidateListClosedWorkflowExecutionsRequest(context.Context, *types.ListClosedWorkflowExecutionsRequest) error\n\t\tValidateRegisterDomainRequest(context.Context, *types.RegisterDomainRequest) error\n\t\tValidateDescribeDomainRequest(context.Context, *types.DescribeDomainRequest) error\n\t\tValidateUpdateDomainRequest(context.Context, *types.UpdateDomainRequest) error\n\t\tValidateDeleteDomainRequest(context.Context, *types.DeleteDomainRequest) error\n\t\tValidateDeprecateDomainRequest(context.Context, *types.DeprecateDomainRequest) error\n\t\tValidateFailoverDomainRequest(context.Context, *types.FailoverDomainRequest) error\n\t}\n\n\trequestValidatorImpl struct {\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tconfig        *config.Config\n\t}\n)\n\nfunc NewRequestValidator(logger log.Logger, metricsClient metrics.Client, config *config.Config) RequestValidator {\n\treturn &requestValidatorImpl{\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t\tconfig:        config,\n\t}\n}\n\nfunc (v *requestValidatorImpl) validateTaskList(t *types.TaskList, scope metrics.Scope, domain string) error {\n\tif t == nil || t.GetName() == \"\" {\n\t\treturn validate.ErrTaskListNotSet\n\t}\n\tif !common.IsValidIDLength(\n\t\tt.GetName(),\n\t\tscope,\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.TaskListNameMaxLength(domain),\n\t\tmetrics.CadenceErrTaskListNameExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeTaskListName) {\n\t\treturn validate.ErrTaskListTooLong\n\t}\n\treturn nil\n}\n\nfunc checkRequiredDomainDataKVs(requiredDomainDataKeys map[string]interface{}, domainData map[string]string) error {\n\t// check requiredDomainDataKeys\n\tfor k := range requiredDomainDataKeys {\n\t\t_, ok := domainData[k]\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"domain data error, missing required key %v . All required keys: %v\", k, requiredDomainDataKeys)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc checkFailOverPermission(config *config.Config, domainName string) error {\n\tif config.Lockdown(domainName) {\n\t\treturn validate.ErrDomainInLockdown\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) isListRequestPageSizeTooLarge(pageSize int32, domain string) bool {\n\treturn common.IsAdvancedVisibilityReadingEnabled(v.config.ReadVisibilityStoreName(domain) != \"db\", v.config.IsAdvancedVisConfigExist) &&\n\t\tpageSize > int32(v.config.ESIndexMaxResultWindow())\n}\n\nfunc (v *requestValidatorImpl) ValidateRefreshWorkflowTasksRequest(ctx context.Context, req *types.RefreshWorkflowTasksRequest) error {\n\tif req == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\treturn validate.CheckExecution(req.Execution)\n}\n\nfunc (v *requestValidatorImpl) ValidateDescribeTaskListRequest(ctx context.Context, request *types.DescribeTaskListRequest) error {\n\tif request == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif request.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif request.TaskListType == nil {\n\t\treturn validate.ErrTaskListTypeNotSet\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendDescribeTaskListScope, request, v.metricsClient).Tagged(metrics.GetContextTags(ctx)...)\n\treturn v.validateTaskList(request.TaskList, scope, request.GetDomain())\n}\n\nfunc (v *requestValidatorImpl) ValidateListTaskListPartitionsRequest(ctx context.Context, request *types.ListTaskListPartitionsRequest) error {\n\tif request == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif request.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tscope := getMetricsScopeWithDomain(metrics.FrontendListTaskListPartitionsScope, request, v.metricsClient).Tagged(metrics.GetContextTags(ctx)...)\n\treturn v.validateTaskList(request.TaskList, scope, request.GetDomain())\n}\n\nfunc (v *requestValidatorImpl) ValidateGetTaskListsByDomainRequest(ctx context.Context, request *types.GetTaskListsByDomainRequest) error {\n\tif request == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif request.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateResetStickyTaskListRequest(ctx context.Context, resetRequest *types.ResetStickyTaskListRequest) error {\n\tif resetRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tdomainName := resetRequest.GetDomain()\n\tif domainName == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\twfExecution := resetRequest.GetExecution()\n\treturn validate.CheckExecution(wfExecution)\n}\n\nfunc (v *requestValidatorImpl) ValidateCountWorkflowExecutionsRequest(ctx context.Context, countRequest *types.CountWorkflowExecutionsRequest) error {\n\tif countRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif countRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateListWorkflowExecutionsRequest(ctx context.Context, listRequest *types.ListWorkflowExecutionsRequest) error {\n\tif listRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif listRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif listRequest.GetPageSize() <= 0 {\n\t\tlistRequest.PageSize = int32(v.config.VisibilityMaxPageSize(listRequest.GetDomain()))\n\t}\n\tif v.isListRequestPageSizeTooLarge(listRequest.GetPageSize(), listRequest.GetDomain()) {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Pagesize is larger than allow %d\", v.config.ESIndexMaxResultWindow())}\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateListOpenWorkflowExecutionsRequest(ctx context.Context, listRequest *types.ListOpenWorkflowExecutionsRequest) error {\n\tif listRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif listRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif listRequest.StartTimeFilter == nil {\n\t\treturn &types.BadRequestError{Message: \"StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.EarliestTime == nil {\n\t\treturn &types.BadRequestError{Message: \"EarliestTime in StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.LatestTime == nil {\n\t\treturn &types.BadRequestError{Message: \"LatestTime in StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.GetEarliestTime() > listRequest.StartTimeFilter.GetLatestTime() {\n\t\treturn &types.BadRequestError{Message: \"EarliestTime in StartTimeFilter should not be larger than LatestTime\"}\n\t}\n\tif listRequest.ExecutionFilter != nil && listRequest.TypeFilter != nil {\n\t\treturn &types.BadRequestError{Message: \"Only one of ExecutionFilter or TypeFilter is allowed\"}\n\t}\n\tif listRequest.GetMaximumPageSize() <= 0 {\n\t\tlistRequest.MaximumPageSize = int32(v.config.VisibilityMaxPageSize(listRequest.GetDomain()))\n\t}\n\tif v.isListRequestPageSizeTooLarge(listRequest.GetMaximumPageSize(), listRequest.GetDomain()) {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Pagesize is larger than allow %d\", v.config.ESIndexMaxResultWindow())}\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateListArchivedWorkflowExecutionsRequest(ctx context.Context, listRequest *types.ListArchivedWorkflowExecutionsRequest) error {\n\tif listRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif listRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif listRequest.GetPageSize() <= 0 {\n\t\tlistRequest.PageSize = int32(v.config.VisibilityMaxPageSize(listRequest.GetDomain()))\n\t}\n\tmaxPageSize := v.config.VisibilityArchivalQueryMaxPageSize()\n\tif int(listRequest.GetPageSize()) > maxPageSize {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Pagesize is larger than allowed %d\", maxPageSize)}\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateListClosedWorkflowExecutionsRequest(ctx context.Context, listRequest *types.ListClosedWorkflowExecutionsRequest) error {\n\tif listRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif listRequest.GetDomain() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif listRequest.StartTimeFilter == nil {\n\t\treturn &types.BadRequestError{Message: \"StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.EarliestTime == nil {\n\t\treturn &types.BadRequestError{Message: \"EarliestTime in StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.LatestTime == nil {\n\t\treturn &types.BadRequestError{Message: \"LatestTime in StartTimeFilter is required\"}\n\t}\n\tif listRequest.StartTimeFilter.GetEarliestTime() > listRequest.StartTimeFilter.GetLatestTime() {\n\t\treturn &types.BadRequestError{Message: \"EarliestTime in StartTimeFilter should not be larger than LatestTime\"}\n\t}\n\tfilterCount := 0\n\tif listRequest.ExecutionFilter != nil {\n\t\tfilterCount++\n\t}\n\tif listRequest.TypeFilter != nil {\n\t\tfilterCount++\n\t}\n\tif listRequest.StatusFilter != nil {\n\t\tfilterCount++\n\t}\n\tif filterCount > 1 {\n\t\treturn &types.BadRequestError{Message: \"Only one of ExecutionFilter, TypeFilter or StatusFilter is allowed\"}\n\t} // If ExecutionFilter is provided with one of TypeFilter or StatusFilter, use ExecutionFilter and ignore other filter\n\tif listRequest.GetMaximumPageSize() <= 0 {\n\t\tlistRequest.MaximumPageSize = int32(v.config.VisibilityMaxPageSize(listRequest.GetDomain()))\n\t}\n\tif v.isListRequestPageSizeTooLarge(listRequest.GetMaximumPageSize(), listRequest.GetDomain()) {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Pagesize is larger than allow %d\", v.config.ESIndexMaxResultWindow())}\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateRegisterDomainRequest(ctx context.Context, registerRequest *types.RegisterDomainRequest) error {\n\tif registerRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif registerRequest.GetName() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tdomain := registerRequest.GetName()\n\tscope := v.metricsClient.Scope(metrics.FrontendRegisterDomainScope).Tagged(metrics.DomainTag(domain)).Tagged(metrics.GetContextTags(ctx)...)\n\tif !common.IsValidIDLength(\n\t\tdomain,\n\t\tscope,\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.DomainNameMaxLength(domain),\n\t\tmetrics.CadenceErrTaskListNameExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn validate.ErrDomainTooLong\n\t}\n\tif registerRequest.GetWorkflowExecutionRetentionPeriodInDays() > int32(v.config.DomainConfig.MaxRetentionDays()) {\n\t\treturn validate.ErrInvalidRetention\n\t}\n\tif err := checkRequiredDomainDataKVs(v.config.DomainConfig.RequiredDomainDataKeys(), registerRequest.GetData()); err != nil {\n\t\treturn err\n\t}\n\treturn validate.CheckPermission(v.config, registerRequest.SecurityToken)\n}\n\nfunc (v *requestValidatorImpl) ValidateDescribeDomainRequest(ctx context.Context, describeRequest *types.DescribeDomainRequest) error {\n\tif describeRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif describeRequest.GetName() == \"\" && describeRequest.GetUUID() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateUpdateDomainRequest(ctx context.Context, updateRequest *types.UpdateDomainRequest) error {\n\tif updateRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif updateRequest.GetName() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\tif updateRequest.WorkflowExecutionRetentionPeriodInDays != nil && *updateRequest.WorkflowExecutionRetentionPeriodInDays > int32(v.config.DomainConfig.MaxRetentionDays()) {\n\t\treturn validate.ErrInvalidRetention\n\t}\n\tisFailover := isFailoverRequest(updateRequest)\n\t// don't require permission for failover request\n\tif isFailover {\n\t\t// reject the failover if the cluster is in lockdown\n\t\tif err := checkFailOverPermission(v.config, updateRequest.GetName()); err != nil {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\tif err := validate.CheckPermission(v.config, updateRequest.SecurityToken); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v *requestValidatorImpl) ValidateDeleteDomainRequest(ctx context.Context, deleteRequest *types.DeleteDomainRequest) error {\n\tif deleteRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif deleteRequest.GetName() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\treturn validate.CheckPermission(v.config, deleteRequest.SecurityToken)\n}\n\nfunc (v *requestValidatorImpl) ValidateDeprecateDomainRequest(ctx context.Context, deprecateRequest *types.DeprecateDomainRequest) error {\n\tif deprecateRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif deprecateRequest.GetName() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\treturn validate.CheckPermission(v.config, deprecateRequest.SecurityToken)\n}\n\nfunc (v *requestValidatorImpl) ValidateFailoverDomainRequest(ctx context.Context, failoverDomainRequest *types.FailoverDomainRequest) error {\n\tif failoverDomainRequest == nil {\n\t\treturn validate.ErrRequestNotSet\n\t}\n\tif failoverDomainRequest.GetDomainName() == \"\" {\n\t\treturn validate.ErrDomainNotSet\n\t}\n\n\tif failoverDomainRequest.DomainActiveClusterName == nil && failoverDomainRequest.ActiveClusters == nil {\n\t\treturn &types.BadRequestError{Message: \"DomainActiveClusterName or ActiveClusters must be provided to failover the domain\"}\n\t}\n\n\t// Security token is not required for failover request - reject the failover if the cluster is in lockdown\n\treturn checkFailOverPermission(v.config, failoverDomainRequest.GetDomainName())\n}\n"
  },
  {
    "path": "service/frontend/api/request_validator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: request_validator.go\n//\n// Generated by this command:\n//\n//\tmockgen -package api -source request_validator.go -destination request_validator_mock.go -self_package github.com/uber/cadence/service/frontend/api requestValidator\n//\n\n// Package api is a generated GoMock package.\npackage api\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockRequestValidator is a mock of RequestValidator interface.\ntype MockRequestValidator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockRequestValidatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockRequestValidatorMockRecorder is the mock recorder for MockRequestValidator.\ntype MockRequestValidatorMockRecorder struct {\n\tmock *MockRequestValidator\n}\n\n// NewMockRequestValidator creates a new mock instance.\nfunc NewMockRequestValidator(ctrl *gomock.Controller) *MockRequestValidator {\n\tmock := &MockRequestValidator{ctrl: ctrl}\n\tmock.recorder = &MockRequestValidatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRequestValidator) EXPECT() *MockRequestValidatorMockRecorder {\n\treturn m.recorder\n}\n\n// ValidateCountWorkflowExecutionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateCountWorkflowExecutionsRequest(arg0 context.Context, arg1 *types.CountWorkflowExecutionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateCountWorkflowExecutionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateCountWorkflowExecutionsRequest indicates an expected call of ValidateCountWorkflowExecutionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateCountWorkflowExecutionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateCountWorkflowExecutionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateCountWorkflowExecutionsRequest), arg0, arg1)\n}\n\n// ValidateDeleteDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateDeleteDomainRequest(arg0 context.Context, arg1 *types.DeleteDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateDeleteDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateDeleteDomainRequest indicates an expected call of ValidateDeleteDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateDeleteDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateDeleteDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateDeleteDomainRequest), arg0, arg1)\n}\n\n// ValidateDeprecateDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateDeprecateDomainRequest(arg0 context.Context, arg1 *types.DeprecateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateDeprecateDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateDeprecateDomainRequest indicates an expected call of ValidateDeprecateDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateDeprecateDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateDeprecateDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateDeprecateDomainRequest), arg0, arg1)\n}\n\n// ValidateDescribeDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateDescribeDomainRequest(arg0 context.Context, arg1 *types.DescribeDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateDescribeDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateDescribeDomainRequest indicates an expected call of ValidateDescribeDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateDescribeDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateDescribeDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateDescribeDomainRequest), arg0, arg1)\n}\n\n// ValidateDescribeTaskListRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateDescribeTaskListRequest(arg0 context.Context, arg1 *types.DescribeTaskListRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateDescribeTaskListRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateDescribeTaskListRequest indicates an expected call of ValidateDescribeTaskListRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateDescribeTaskListRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateDescribeTaskListRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateDescribeTaskListRequest), arg0, arg1)\n}\n\n// ValidateFailoverDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateFailoverDomainRequest(arg0 context.Context, arg1 *types.FailoverDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateFailoverDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateFailoverDomainRequest indicates an expected call of ValidateFailoverDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateFailoverDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateFailoverDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateFailoverDomainRequest), arg0, arg1)\n}\n\n// ValidateGetTaskListsByDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateGetTaskListsByDomainRequest(arg0 context.Context, arg1 *types.GetTaskListsByDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateGetTaskListsByDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateGetTaskListsByDomainRequest indicates an expected call of ValidateGetTaskListsByDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateGetTaskListsByDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateGetTaskListsByDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateGetTaskListsByDomainRequest), arg0, arg1)\n}\n\n// ValidateListArchivedWorkflowExecutionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateListArchivedWorkflowExecutionsRequest(arg0 context.Context, arg1 *types.ListArchivedWorkflowExecutionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateListArchivedWorkflowExecutionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateListArchivedWorkflowExecutionsRequest indicates an expected call of ValidateListArchivedWorkflowExecutionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateListArchivedWorkflowExecutionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateListArchivedWorkflowExecutionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateListArchivedWorkflowExecutionsRequest), arg0, arg1)\n}\n\n// ValidateListClosedWorkflowExecutionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateListClosedWorkflowExecutionsRequest(arg0 context.Context, arg1 *types.ListClosedWorkflowExecutionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateListClosedWorkflowExecutionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateListClosedWorkflowExecutionsRequest indicates an expected call of ValidateListClosedWorkflowExecutionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateListClosedWorkflowExecutionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateListClosedWorkflowExecutionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateListClosedWorkflowExecutionsRequest), arg0, arg1)\n}\n\n// ValidateListOpenWorkflowExecutionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateListOpenWorkflowExecutionsRequest(arg0 context.Context, arg1 *types.ListOpenWorkflowExecutionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateListOpenWorkflowExecutionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateListOpenWorkflowExecutionsRequest indicates an expected call of ValidateListOpenWorkflowExecutionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateListOpenWorkflowExecutionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateListOpenWorkflowExecutionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateListOpenWorkflowExecutionsRequest), arg0, arg1)\n}\n\n// ValidateListTaskListPartitionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateListTaskListPartitionsRequest(arg0 context.Context, arg1 *types.ListTaskListPartitionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateListTaskListPartitionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateListTaskListPartitionsRequest indicates an expected call of ValidateListTaskListPartitionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateListTaskListPartitionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateListTaskListPartitionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateListTaskListPartitionsRequest), arg0, arg1)\n}\n\n// ValidateListWorkflowExecutionsRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateListWorkflowExecutionsRequest(arg0 context.Context, arg1 *types.ListWorkflowExecutionsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateListWorkflowExecutionsRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateListWorkflowExecutionsRequest indicates an expected call of ValidateListWorkflowExecutionsRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateListWorkflowExecutionsRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateListWorkflowExecutionsRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateListWorkflowExecutionsRequest), arg0, arg1)\n}\n\n// ValidateRefreshWorkflowTasksRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateRefreshWorkflowTasksRequest(arg0 context.Context, arg1 *types.RefreshWorkflowTasksRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateRefreshWorkflowTasksRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateRefreshWorkflowTasksRequest indicates an expected call of ValidateRefreshWorkflowTasksRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateRefreshWorkflowTasksRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateRefreshWorkflowTasksRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateRefreshWorkflowTasksRequest), arg0, arg1)\n}\n\n// ValidateRegisterDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateRegisterDomainRequest(arg0 context.Context, arg1 *types.RegisterDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateRegisterDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateRegisterDomainRequest indicates an expected call of ValidateRegisterDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateRegisterDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateRegisterDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateRegisterDomainRequest), arg0, arg1)\n}\n\n// ValidateResetStickyTaskListRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateResetStickyTaskListRequest(arg0 context.Context, arg1 *types.ResetStickyTaskListRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateResetStickyTaskListRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateResetStickyTaskListRequest indicates an expected call of ValidateResetStickyTaskListRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateResetStickyTaskListRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateResetStickyTaskListRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateResetStickyTaskListRequest), arg0, arg1)\n}\n\n// ValidateUpdateDomainRequest mocks base method.\nfunc (m *MockRequestValidator) ValidateUpdateDomainRequest(arg0 context.Context, arg1 *types.UpdateDomainRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateUpdateDomainRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ValidateUpdateDomainRequest indicates an expected call of ValidateUpdateDomainRequest.\nfunc (mr *MockRequestValidatorMockRecorder) ValidateUpdateDomainRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateUpdateDomainRequest\", reflect.TypeOf((*MockRequestValidator)(nil).ValidateUpdateDomainRequest), arg0, arg1)\n}\n"
  },
  {
    "path": "service/frontend/api/request_validator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\nfunc setupMocksForRequestValidator(t *testing.T) (*requestValidatorImpl, *mockDeps) {\n\tlogger := testlogger.New(t)\n\tmetricsClient := metrics.NewNoopMetricsClient()\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\tconfig := frontendcfg.NewConfig(\n\t\tdynamicconfig.NewCollection(\n\t\t\tdynamicClient,\n\t\t\tlogger,\n\t\t),\n\t\tnumHistoryShards,\n\t\ttrue,\n\t\t\"hostname\",\n\t\tlogger,\n\t)\n\tdeps := &mockDeps{\n\t\tdynamicClient: dynamicClient,\n\t}\n\tv := NewRequestValidator(logger, metricsClient, config)\n\treturn v.(*requestValidatorImpl), deps\n}\n\nfunc TestValidateRefreshWorkflowTasksRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.RefreshWorkflowTasksRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"execution not set\",\n\t\t\treq: &types.RefreshWorkflowTasksRequest{\n\t\t\t\tDomain:    \"domain\",\n\t\t\t\tExecution: nil,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Execution is not set on request.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, _ := setupMocksForRequestValidator(t)\n\n\t\t\terr := v.ValidateRefreshWorkflowTasksRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateValidateDescribeTaskListRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DescribeTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list type not set\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskListType is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list not set\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain:       \"domain\",\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list name not set\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain:       \"domain\",\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tTaskList:     &types.TaskList{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list name too long\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain:       \"domain\",\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList length exceeds limit.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskListNameMaxLength, 5))\n\n\t\t\terr := v.ValidateDescribeTaskListRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateValidateListTaskListPartitionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListTaskListPartitionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list not set\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list name not set\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain:   \"domain\",\n\t\t\t\tTaskList: &types.TaskList{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"task list name too long\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"taskListName\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"TaskList length exceeds limit.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskListNameMaxLength, 5))\n\n\t\t\terr := v.ValidateListTaskListPartitionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateValidateGetTaskListsByDomainRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.GetTaskListsByDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.GetTaskListsByDomainRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.GetTaskListsByDomainRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskListNameMaxLength, 5))\n\n\t\t\terr := v.ValidateGetTaskListsByDomainRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateValidateResetStickyTaskListRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ResetStickyTaskListRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"execution not set\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain:    \"domain\",\n\t\t\t\tExecution: nil,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Execution is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"workflowID not set\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain:    \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"WorkflowId is not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"Invalid RunID\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"a\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Invalid RunId.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskListNameMaxLength, 5))\n\n\t\t\terr := v.ValidateResetStickyTaskListRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateCountWorkflowExecutionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.CountWorkflowExecutionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.CountWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.CountWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, _ := setupMocksForRequestValidator(t)\n\n\t\t\terr := v.ValidateCountWorkflowExecutionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateListWorkflowExecutionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListWorkflowExecutionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ListWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"page size too large\",\n\t\t\treq: &types.ListWorkflowExecutionsRequest{\n\t\t\t\tDomain:   \"domain\",\n\t\t\t\tPageSize: 101,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Pagesize is larger than allow 100\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendESIndexMaxResultWindow, 100))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendVisibilityMaxPageSize, 10))\n\n\t\t\terr := v.ValidateListWorkflowExecutionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int32(10), tc.req.GetPageSize())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateListOpenWorkflowExecutionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListOpenWorkflowExecutionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tMaximumPageSize: 0,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"startTimeFilter not set\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"Earliest time not set\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain:          \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"EarliestTime in StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"Latest time not set\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"LatestTime in StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"earliest time later than latest time\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(3)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"EarliestTime in StartTimeFilter should not be larger than LatestTime\",\n\t\t},\n\t\t{\n\t\t\tname: \"both execution and type filter are specified\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{},\n\t\t\t\tTypeFilter:      &types.WorkflowTypeFilter{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only one of ExecutionFilter or TypeFilter is allowed\",\n\t\t},\n\t\t{\n\t\t\tname: \"page size too large\",\n\t\t\treq: &types.ListOpenWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tMaximumPageSize: 101,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Pagesize is larger than allow 100\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendESIndexMaxResultWindow, 100))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendVisibilityMaxPageSize, 10))\n\n\t\t\terr := v.ValidateListOpenWorkflowExecutionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int32(10), tc.req.GetMaximumPageSize())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateListClosedWorkflowExecutionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListClosedWorkflowExecutionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tMaximumPageSize: 0,\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"startTimeFilter not set\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"Earliest time not set\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain:          \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"EarliestTime in StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"Latest time not set\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"LatestTime in StartTimeFilter is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"earliest time later than latest time\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(3)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"EarliestTime in StartTimeFilter should not be larger than LatestTime\",\n\t\t},\n\t\t{\n\t\t\tname: \"both execution and type filter are specified\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{},\n\t\t\t\tTypeFilter:      &types.WorkflowTypeFilter{},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only one of ExecutionFilter, TypeFilter or StatusFilter is allowed\",\n\t\t},\n\t\t{\n\t\t\tname: \"both execution and status filter are specified\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tExecutionFilter: &types.WorkflowExecutionFilter{},\n\t\t\t\tStatusFilter:    types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only one of ExecutionFilter, TypeFilter or StatusFilter is allowed\",\n\t\t},\n\t\t{\n\t\t\tname: \"both type and status filter are specified\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tTypeFilter:   &types.WorkflowTypeFilter{},\n\t\t\t\tStatusFilter: types.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only one of ExecutionFilter, TypeFilter or StatusFilter is allowed\",\n\t\t},\n\t\t{\n\t\t\tname: \"page size too large\",\n\t\t\treq: &types.ListClosedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\t\tEarliestTime: common.Ptr(int64(1)),\n\t\t\t\t\tLatestTime:   common.Ptr(int64(2)),\n\t\t\t\t},\n\t\t\t\tMaximumPageSize: 101,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Pagesize is larger than allow 100\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendESIndexMaxResultWindow, 100))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendVisibilityMaxPageSize, 10))\n\n\t\t\terr := v.ValidateListClosedWorkflowExecutionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int32(10), tc.req.GetMaximumPageSize())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateListArchivedWorkflowExecutionsRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListArchivedWorkflowExecutionsRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListArchivedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.ListArchivedWorkflowExecutionsRequest{\n\t\t\t\tDomain: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"page size too large\",\n\t\t\treq: &types.ListArchivedWorkflowExecutionsRequest{\n\t\t\t\tDomain:   \"domain\",\n\t\t\t\tPageSize: 101,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Pagesize is larger than allowed 100\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.VisibilityArchivalQueryMaxPageSize, 100))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.FrontendVisibilityMaxPageSize, 10))\n\n\t\t\terr := v.ValidateListArchivedWorkflowExecutionsRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, int32(10), tc.req.GetPageSize())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateDeprecateDomainRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DeprecateDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"token\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain not set\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName: \"\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"wrong token\",\n\t\t\treq: &types.DeprecateDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"to\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"No permission to do this operation.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.EnableAdminProtection, true))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.AdminOperationToken, \"token\"))\n\n\t\t\terr := v.ValidateDeprecateDomainRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateDescribeDomainRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DescribeDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success - with domain name\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tName: common.Ptr(\"domain\"),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success - with domain id\",\n\t\t\treq: &types.DescribeDomainRequest{\n\t\t\t\tUUID: common.Ptr(\"id\"),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname:          \"domain not set\",\n\t\t\treq:           &types.DescribeDomainRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, _ := setupMocksForRequestValidator(t)\n\n\t\t\terr := v.ValidateDescribeDomainRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateUpdateDomainRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.UpdateDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success - non failover\",\n\t\t\treq: &types.UpdateDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"token\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname:          \"domain not set\",\n\t\t\treq:           &types.UpdateDomainRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid retention\",\n\t\t\treq: &types.UpdateDomainRequest{\n\t\t\t\tName:                                   \"domain\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Ptr(int32(100)),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"RetentionDays is invalid.\",\n\t\t},\n\t\t{\n\t\t\tname: \"wrong token\",\n\t\t\treq: &types.UpdateDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"to\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"No permission to do this operation.\",\n\t\t},\n\t\t{\n\t\t\tname: \"lockdown rejects active-passive failover\",\n\t\t\treq: &types.UpdateDomainRequest{\n\t\t\t\tName:              \"domain\",\n\t\t\t\tActiveClusterName: common.Ptr(\"a\"),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain is not accepting fail overs at this time due to lockdown.\",\n\t\t},\n\t\t{\n\t\t\tname: \"lockdown rejects active-active failover\",\n\t\t\treq: &types.UpdateDomainRequest{\n\t\t\t\tName: \"domain\",\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region0\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain is not accepting fail overs at this time due to lockdown.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.EnableAdminProtection, true))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.AdminOperationToken, \"token\"))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MaxRetentionDays, 3))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.Lockdown, true))\n\n\t\t\terr := v.ValidateUpdateDomainRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateRegisterDomainRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.RegisterDomainRequest\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"token\",\n\t\t\t\tData:          map[string]string{\"tier\": \"3\"},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"not set\",\n\t\t\treq:           nil,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Request is nil.\",\n\t\t},\n\t\t{\n\t\t\tname:          \"domain not set\",\n\t\t\treq:           &types.RegisterDomainRequest{},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain not set on request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"name too long\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName: \"domain-name-toooooooooooooo-long\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Domain length exceeds limit.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid retention\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName:                                   \"domain\",\n\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 100,\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"RetentionDays is invalid.\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing data key\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"to\",\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain data error, missing required key tier . All required keys: map[tier:true]\",\n\t\t},\n\t\t{\n\t\t\tname: \"wrong token\",\n\t\t\treq: &types.RegisterDomainRequest{\n\t\t\t\tName:          \"domain\",\n\t\t\t\tSecurityToken: \"to\",\n\t\t\t\tData:          map[string]string{\"tier\": \"3\"},\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"No permission to do this operation.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tv, deps := setupMocksForRequestValidator(t)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.EnableAdminProtection, true))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.AdminOperationToken, \"token\"))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MaxRetentionDays, 3))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.DomainNameMaxLength, 10))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.RequiredDomainDataKeys, map[string]interface{}{\"tier\": true}))\n\n\t\t\terr := v.ValidateRegisterDomainRequest(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/shutting_down_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestShuttingDownError(t *testing.T) {\n\twh, _ := setupMocksForWorkflowHandler(t)\n\twh.Stop()\n\n\t// get all methods of Handler interface\n\ttt := reflect.TypeOf(struct{ Handler }{})\n\tmethodNames := make(map[string]struct{})\n\tfor i := 0; i < tt.NumMethod(); i++ {\n\t\tmethodNames[tt.Method(i).Name] = struct{}{}\n\t}\n\tdelete(methodNames, \"GetClusterInfo\")\n\tdelete(methodNames, \"Health\")\n\n\tv := reflect.ValueOf(wh)\n\tfor name := range methodNames {\n\t\tmethod := v.MethodByName(name)\n\t\tmethodType := method.Type()\n\t\tif methodType.Kind() != reflect.Func {\n\t\t\tt.Fatalf(\"method: %s is not a function - %s\", name, methodType.String())\n\t\t}\n\t\tif methodType.IsVariadic() {\n\t\t\tt.Fatalf(\"method: %s is variadic - %s\", name, methodType.String())\n\t\t}\n\t\tif methodType.NumIn() < 1 || methodType.NumIn() > 2 {\n\t\t\tt.Fatalf(\"method: %s has wrong number of inputs - %s\", name, methodType.String())\n\t\t}\n\n\t\tvar results []reflect.Value\n\t\tif methodType.NumIn() == 1 {\n\t\t\tresults = method.Call([]reflect.Value{reflect.ValueOf(context.Background())})\n\t\t} else {\n\t\t\tresults = method.Call([]reflect.Value{reflect.ValueOf(context.Background()), reflect.Zero(methodType.In(1))})\n\t\t}\n\t\tif len(results) == 1 {\n\t\t\terr, ok := results[0].Interface().(error)\n\t\t\tif !ok {\n\t\t\t\tt.Fatalf(\"method: %s has wrong output type - %s\", name, methodType.String())\n\t\t\t}\n\t\t\tassert.ErrorContains(t, err, \"Shutting down\")\n\t\t} else if len(results) == 2 {\n\t\t\terr, ok := results[1].Interface().(error)\n\t\t\tif !ok {\n\t\t\t\tt.Fatalf(\"method: %s has wrong output type - %s\", name, methodType.String())\n\t\t\t}\n\t\t\tassert.ErrorContains(t, err, \"Shutting down\")\n\t\t} else {\n\t\t\tt.Fatalf(\"method: %s has wrong number of outputs - %s\", name, methodType.String())\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/frontend/api/task_list_handlers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\n// DescribeTaskList returns information about the target tasklist, right now this API returns the\n// pollers which polled this tasklist in last few minutes. If includeTaskListStatus field is true,\n// it will also return status of tasklist's ackManager (readLevel, ackLevel, backlogCountHint and taskIDBlock).\nfunc (wh *WorkflowHandler) DescribeTaskList(\n\tctx context.Context,\n\trequest *types.DescribeTaskListRequest,\n) (resp *types.DescribeTaskListResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateDescribeTaskListRequest(ctx, request); err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID, err := wh.GetDomainCache().GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresponse, err := wh.GetMatchingClient().DescribeTaskList(ctx, &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID:  domainID,\n\t\tDescRequest: request,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn response, nil\n}\n\n// ListTaskListPartitions returns all the partition and host for a taskList\nfunc (wh *WorkflowHandler) ListTaskListPartitions(\n\tctx context.Context,\n\trequest *types.ListTaskListPartitionsRequest,\n) (resp *types.ListTaskListPartitionsResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateListTaskListPartitionsRequest(ctx, request); err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := wh.GetMatchingClient().ListTaskListPartitions(ctx, &types.MatchingListTaskListPartitionsRequest{\n\t\tDomain:   request.Domain,\n\t\tTaskList: request.TaskList,\n\t})\n\treturn resp, err\n}\n\n// GetTaskListsByDomain returns all the partition and host for a taskList\nfunc (wh *WorkflowHandler) GetTaskListsByDomain(\n\tctx context.Context,\n\trequest *types.GetTaskListsByDomainRequest,\n) (resp *types.GetTaskListsByDomainResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateGetTaskListsByDomainRequest(ctx, request); err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := wh.GetMatchingClient().GetTaskListsByDomain(ctx, &types.GetTaskListsByDomainRequest{\n\t\tDomain: request.Domain,\n\t})\n\treturn resp, err\n}\n\n// ResetStickyTaskList reset the volatile information in mutable state of a given workflow.\nfunc (wh *WorkflowHandler) ResetStickyTaskList(\n\tctx context.Context,\n\tresetRequest *types.ResetStickyTaskListRequest,\n) (resp *types.ResetStickyTaskListResponse, retError error) {\n\tif wh.isShuttingDown() {\n\t\treturn nil, validate.ErrShuttingDown\n\t}\n\tif err := wh.requestValidator.ValidateResetStickyTaskListRequest(ctx, resetRequest); err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID, err := wh.GetDomainCache().GetDomainID(resetRequest.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t_, err = wh.GetHistoryClient().ResetStickyTaskList(ctx, &types.HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: domainID,\n\t\tExecution:  resetRequest.Execution,\n\t})\n\tif err != nil {\n\t\treturn nil, wh.normalizeVersionedErrors(ctx, err)\n\t}\n\treturn &types.ResetStickyTaskListResponse{}, nil\n}\n"
  },
  {
    "path": "service/frontend/api/task_list_handlers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage api\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDescribeTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.DescribeTaskListRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *types.DescribeTaskListResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"domain-id\", nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().DescribeTaskList(gomock.Any(), &types.MatchingDescribeTaskListRequest{\n\t\t\t\t\tDomainUUID: \"domain-id\",\n\t\t\t\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\tName: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected:    &types.DescribeTaskListResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"matching client error\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"domain-id\", nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().DescribeTaskList(gomock.Any(), &types.MatchingDescribeTaskListRequest{\n\t\t\t\t\tDomainUUID: \"domain-id\",\n\t\t\t\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\tName: \"tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"matching client error\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"\", errors.New(\"domain cache error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"validator error\",\n\t\t\treq: &types.DescribeTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateDescribeTaskListRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validator error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validator error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tresp, err := wh.DescribeTaskList(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListTaskListPartitions(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ListTaskListPartitionsRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *types.ListTaskListPartitionsResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListTaskListPartitionsRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().ListTaskListPartitions(gomock.Any(), &types.MatchingListTaskListPartitionsRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.ListTaskListPartitionsResponse{}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected:    &types.ListTaskListPartitionsResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"matching client error\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListTaskListPartitionsRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().ListTaskListPartitions(gomock.Any(), &types.MatchingListTaskListPartitionsRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"tl\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"matching client error\",\n\t\t},\n\t\t{\n\t\t\tname: \"validator error\",\n\t\t\treq: &types.ListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"tl\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateListTaskListPartitionsRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validator error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validator error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tresp, err := wh.ListTaskListPartitions(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetTaskListsByDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.GetTaskListsByDomainRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *types.GetTaskListsByDomainResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.GetTaskListsByDomainRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateGetTaskListsByDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().GetTaskListsByDomain(gomock.Any(), &types.GetTaskListsByDomainRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t}).Return(&types.GetTaskListsByDomainResponse{}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected:    &types.GetTaskListsByDomainResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"matching client error\",\n\t\t\treq: &types.GetTaskListsByDomainRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateGetTaskListsByDomainRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().GetTaskListsByDomain(gomock.Any(), &types.GetTaskListsByDomainRequest{\n\t\t\t\t\tDomain: \"domain\",\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"matching client error\",\n\t\t},\n\t\t{\n\t\t\tname: \"validator error\",\n\t\t\treq: &types.GetTaskListsByDomainRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateGetTaskListsByDomainRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validator error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validator error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tresp, err := wh.GetTaskListsByDomain(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestResetStickyTaskList(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.ResetStickyTaskListRequest\n\t\tsetupMocks    func(*mockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t\texpected      *types.ResetStickyTaskListResponse\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateResetStickyTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"domain-id\", nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().ResetStickyTaskList(gomock.Any(), &types.HistoryResetStickyTaskListRequest{\n\t\t\t\t\tDomainUUID: \"domain-id\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.HistoryResetStickyTaskListResponse{}, nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected:    &types.ResetStickyTaskListResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"history client error\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateResetStickyTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"domain-id\", nil)\n\t\t\t\tdeps.mockHistoryClient.EXPECT().ResetStickyTaskList(gomock.Any(), &types.HistoryResetStickyTaskListRequest{\n\t\t\t\t\tDomainUUID: \"domain-id\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"history client error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"history client error\",\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateResetStickyTaskListRequest(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainID(\"domain\").Return(\"\", errors.New(\"domain cache error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"domain cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"validator error\",\n\t\t\treq: &types.ResetStickyTaskListRequest{\n\t\t\t\tDomain: \"domain\",\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockRequestValidator.EXPECT().ValidateResetStickyTaskListRequest(gomock.Any(), gomock.Any()).Return(errors.New(\"validator error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"validator error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\twh, deps := setupMocksForWorkflowHandler(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tresp, err := wh.ResetStickyTaskList(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/config/config.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\n// Config represents configuration for cadence-frontend service\ntype Config struct {\n\tNumHistoryShards                int\n\tIsAdvancedVisConfigExist        bool\n\tDomainConfig                    domain.Config\n\tPersistenceMaxQPS               dynamicproperties.IntPropertyFn\n\tPersistenceGlobalMaxQPS         dynamicproperties.IntPropertyFn\n\tVisibilityMaxPageSize           dynamicproperties.IntPropertyFnWithDomainFilter\n\tEnableVisibilitySampling        dynamicproperties.BoolPropertyFn\n\tEnableReadFromClosedExecutionV2 dynamicproperties.BoolPropertyFn\n\t// deprecated: never used for ratelimiting, only sampling-based failure injection, and only on database-based visibility\n\tVisibilityListMaxQPS            dynamicproperties.IntPropertyFnWithDomainFilter\n\tEnableLogCustomerQueryParameter dynamicproperties.BoolPropertyFnWithDomainFilter\n\tReadVisibilityStoreName         dynamicproperties.StringPropertyFnWithDomainFilter\n\t// deprecated: never read from\n\tESVisibilityListMaxQPS            dynamicproperties.IntPropertyFnWithDomainFilter\n\tESIndexMaxResultWindow            dynamicproperties.IntPropertyFn\n\tHistoryMaxPageSize                dynamicproperties.IntPropertyFnWithDomainFilter\n\tUserRPS                           dynamicproperties.IntPropertyFn\n\tWorkerRPS                         dynamicproperties.IntPropertyFn\n\tVisibilityRPS                     dynamicproperties.IntPropertyFn\n\tAsyncRPS                          dynamicproperties.IntPropertyFn\n\tMaxDomainUserRPSPerInstance       dynamicproperties.IntPropertyFnWithDomainFilter\n\tMaxDomainWorkerRPSPerInstance     dynamicproperties.IntPropertyFnWithDomainFilter\n\tMaxDomainVisibilityRPSPerInstance dynamicproperties.IntPropertyFnWithDomainFilter\n\tMaxDomainAsyncRPSPerInstance      dynamicproperties.IntPropertyFnWithDomainFilter\n\tGlobalDomainUserRPS               dynamicproperties.IntPropertyFnWithDomainFilter\n\tGlobalDomainWorkerRPS             dynamicproperties.IntPropertyFnWithDomainFilter\n\tGlobalDomainVisibilityRPS         dynamicproperties.IntPropertyFnWithDomainFilter\n\tGlobalDomainAsyncRPS              dynamicproperties.IntPropertyFnWithDomainFilter\n\tMaxWorkerPollDelay                dynamicproperties.DurationPropertyFnWithDomainFilter\n\tRateLimiterBypassCallerTypes      dynamicproperties.ListPropertyFn\n\tEnableClientVersionCheck          dynamicproperties.BoolPropertyFn\n\tEnableQueryAttributeValidation    dynamicproperties.BoolPropertyFn\n\tDisallowQuery                     dynamicproperties.BoolPropertyFnWithDomainFilter\n\tShutdownDrainDuration             dynamicproperties.DurationPropertyFn\n\tWarmupDuration                    dynamicproperties.DurationPropertyFn\n\tLockdown                          dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// global ratelimiter config, uses GlobalDomain*RPS for RPS configuration\n\tGlobalRatelimiterKeyMode        dynamicproperties.StringPropertyWithRatelimitKeyFilter\n\tGlobalRatelimiterUpdateInterval dynamicproperties.DurationPropertyFn\n\n\t// isolation configuration\n\tEnableTasklistIsolation  dynamicproperties.BoolPropertyFnWithDomainFilter\n\tEnableDomainAuditLogging dynamicproperties.BoolPropertyFn\n\n\t// id length limits\n\tMaxIDLengthWarnLimit  dynamicproperties.IntPropertyFn\n\tDomainNameMaxLength   dynamicproperties.IntPropertyFnWithDomainFilter\n\tIdentityMaxLength     dynamicproperties.IntPropertyFnWithDomainFilter\n\tWorkflowIDMaxLength   dynamicproperties.IntPropertyFnWithDomainFilter\n\tSignalNameMaxLength   dynamicproperties.IntPropertyFnWithDomainFilter\n\tWorkflowTypeMaxLength dynamicproperties.IntPropertyFnWithDomainFilter\n\tRequestIDMaxLength    dynamicproperties.IntPropertyFnWithDomainFilter\n\tTaskListNameMaxLength dynamicproperties.IntPropertyFnWithDomainFilter\n\n\t// security protection settings\n\tEnableAdminProtection         dynamicproperties.BoolPropertyFn\n\tAdminOperationToken           dynamicproperties.StringPropertyFn\n\tDisableListVisibilityByFilter dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// size limit system protection\n\tBlobSizeLimitError dynamicproperties.IntPropertyFnWithDomainFilter\n\tBlobSizeLimitWarn  dynamicproperties.IntPropertyFnWithDomainFilter\n\n\tThrottledLogRPS dynamicproperties.IntPropertyFn\n\n\t// Domain specific config\n\tEnableDomainNotActiveAutoForwarding               dynamicproperties.BoolPropertyFnWithDomainFilter\n\tEnableGracefulFailover                            dynamicproperties.BoolPropertyFn\n\tDomainFailoverRefreshInterval                     dynamicproperties.DurationPropertyFn\n\tDomainFailoverRefreshTimerJitterCoefficient       dynamicproperties.FloatPropertyFn\n\tEnableActiveClusterSelectionPolicyInStartWorkflow dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// ValidSearchAttributes is legal indexed keys that can be used in list APIs\n\tValidSearchAttributes             dynamicproperties.MapPropertyFn\n\tSearchAttributesNumberOfKeysLimit dynamicproperties.IntPropertyFnWithDomainFilter\n\tSearchAttributesSizeOfValueLimit  dynamicproperties.IntPropertyFnWithDomainFilter\n\tSearchAttributesTotalSizeLimit    dynamicproperties.IntPropertyFnWithDomainFilter\n\tPinotOptimizedQueryColumns        dynamicproperties.MapPropertyFn\n\t// VisibilityArchival system protection\n\tVisibilityArchivalQueryMaxPageSize dynamicproperties.IntPropertyFn\n\n\tSendRawWorkflowHistory dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// max number of decisions per RespondDecisionTaskCompleted request (unlimited by default)\n\tDecisionResultCountLimit dynamicproperties.IntPropertyFnWithDomainFilter\n\n\t// Debugging\n\n\t// Emit signal related metrics with signal name tag. Be aware of cardinality.\n\tEmitSignalNameMetricsTag dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// HostName for machine running the service\n\tHostName string\n}\n\n// NewConfig returns new service config with default values\nfunc NewConfig(dc *dynamicconfig.Collection, numHistoryShards int, isAdvancedVisConfigExist bool, hostName string, logger log.Logger) *Config {\n\tlogger.Debugf(\"Creating new frontend config for host %s, numHistoryShards: %d, isAdvancedVisConfigExist: %t\", hostName, numHistoryShards, isAdvancedVisConfigExist)\n\treturn &Config{\n\t\tNumHistoryShards:                                  numHistoryShards,\n\t\tIsAdvancedVisConfigExist:                          isAdvancedVisConfigExist,\n\t\tPersistenceMaxQPS:                                 dc.GetIntProperty(dynamicproperties.FrontendPersistenceMaxQPS),\n\t\tPersistenceGlobalMaxQPS:                           dc.GetIntProperty(dynamicproperties.FrontendPersistenceGlobalMaxQPS),\n\t\tVisibilityMaxPageSize:                             dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendVisibilityMaxPageSize),\n\t\tEnableVisibilitySampling:                          dc.GetBoolProperty(dynamicproperties.EnableVisibilitySampling),\n\t\tEnableReadFromClosedExecutionV2:                   dc.GetBoolProperty(dynamicproperties.EnableReadFromClosedExecutionV2),\n\t\tVisibilityListMaxQPS:                              dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendVisibilityListMaxQPS),\n\t\tESVisibilityListMaxQPS:                            dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendESVisibilityListMaxQPS),\n\t\tReadVisibilityStoreName:                           dc.GetStringPropertyFilteredByDomain(dynamicproperties.ReadVisibilityStoreName),\n\t\tEnableLogCustomerQueryParameter:                   dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableLogCustomerQueryParameter),\n\t\tESIndexMaxResultWindow:                            dc.GetIntProperty(dynamicproperties.FrontendESIndexMaxResultWindow),\n\t\tHistoryMaxPageSize:                                dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendHistoryMaxPageSize),\n\t\tUserRPS:                                           dc.GetIntProperty(dynamicproperties.FrontendUserRPS),\n\t\tWorkerRPS:                                         dc.GetIntProperty(dynamicproperties.FrontendWorkerRPS),\n\t\tVisibilityRPS:                                     dc.GetIntProperty(dynamicproperties.FrontendVisibilityRPS),\n\t\tAsyncRPS:                                          dc.GetIntProperty(dynamicproperties.FrontendAsyncRPS),\n\t\tMaxDomainUserRPSPerInstance:                       dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxDomainUserRPSPerInstance),\n\t\tMaxDomainWorkerRPSPerInstance:                     dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxDomainWorkerRPSPerInstance),\n\t\tMaxDomainVisibilityRPSPerInstance:                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxDomainVisibilityRPSPerInstance),\n\t\tMaxDomainAsyncRPSPerInstance:                      dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxDomainAsyncRPSPerInstance),\n\t\tGlobalDomainUserRPS:                               dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendGlobalDomainUserRPS),\n\t\tGlobalDomainWorkerRPS:                             dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendGlobalDomainWorkerRPS),\n\t\tGlobalDomainVisibilityRPS:                         dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendGlobalDomainVisibilityRPS),\n\t\tGlobalDomainAsyncRPS:                              dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendGlobalDomainAsyncRPS),\n\t\tMaxWorkerPollDelay:                                dc.GetDurationPropertyFilteredByDomain(dynamicproperties.FrontendMaxWorkerPollDelay),\n\t\tRateLimiterBypassCallerTypes:                      dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes),\n\t\tGlobalRatelimiterKeyMode:                          dc.GetStringPropertyFilteredByRatelimitKey(dynamicproperties.FrontendGlobalRatelimiterMode),\n\t\tGlobalRatelimiterUpdateInterval:                   dc.GetDurationProperty(dynamicproperties.GlobalRatelimiterUpdateInterval),\n\t\tMaxIDLengthWarnLimit:                              dc.GetIntProperty(dynamicproperties.MaxIDLengthWarnLimit),\n\t\tDomainNameMaxLength:                               dc.GetIntPropertyFilteredByDomain(dynamicproperties.DomainNameMaxLength),\n\t\tIdentityMaxLength:                                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.IdentityMaxLength),\n\t\tWorkflowIDMaxLength:                               dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowIDMaxLength),\n\t\tSignalNameMaxLength:                               dc.GetIntPropertyFilteredByDomain(dynamicproperties.SignalNameMaxLength),\n\t\tWorkflowTypeMaxLength:                             dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowTypeMaxLength),\n\t\tRequestIDMaxLength:                                dc.GetIntPropertyFilteredByDomain(dynamicproperties.RequestIDMaxLength),\n\t\tTaskListNameMaxLength:                             dc.GetIntPropertyFilteredByDomain(dynamicproperties.TaskListNameMaxLength),\n\t\tEnableAdminProtection:                             dc.GetBoolProperty(dynamicproperties.EnableAdminProtection),\n\t\tAdminOperationToken:                               dc.GetStringProperty(dynamicproperties.AdminOperationToken),\n\t\tDisableListVisibilityByFilter:                     dc.GetBoolPropertyFilteredByDomain(dynamicproperties.DisableListVisibilityByFilter),\n\t\tBlobSizeLimitError:                                dc.GetIntPropertyFilteredByDomain(dynamicproperties.BlobSizeLimitError),\n\t\tBlobSizeLimitWarn:                                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.BlobSizeLimitWarn),\n\t\tThrottledLogRPS:                                   dc.GetIntProperty(dynamicproperties.FrontendThrottledLogRPS),\n\t\tShutdownDrainDuration:                             dc.GetDurationProperty(dynamicproperties.FrontendShutdownDrainDuration),\n\t\tWarmupDuration:                                    dc.GetDurationProperty(dynamicproperties.FrontendWarmupDuration),\n\t\tEnableDomainNotActiveAutoForwarding:               dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableDomainNotActiveAutoForwarding),\n\t\tEnableGracefulFailover:                            dc.GetBoolProperty(dynamicproperties.EnableGracefulFailover),\n\t\tDomainFailoverRefreshInterval:                     dc.GetDurationProperty(dynamicproperties.DomainFailoverRefreshInterval),\n\t\tDomainFailoverRefreshTimerJitterCoefficient:       dc.GetFloat64Property(dynamicproperties.DomainFailoverRefreshTimerJitterCoefficient),\n\t\tEnableActiveClusterSelectionPolicyInStartWorkflow: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableActiveClusterSelectionPolicyInStartWorkflow),\n\t\tEnableClientVersionCheck:                          dc.GetBoolProperty(dynamicproperties.EnableClientVersionCheck),\n\t\tEnableQueryAttributeValidation:                    dc.GetBoolProperty(dynamicproperties.EnableQueryAttributeValidation),\n\t\tValidSearchAttributes:                             dc.GetMapProperty(dynamicproperties.ValidSearchAttributes),\n\t\tSearchAttributesNumberOfKeysLimit:                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesNumberOfKeysLimit),\n\t\tSearchAttributesSizeOfValueLimit:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesSizeOfValueLimit),\n\t\tSearchAttributesTotalSizeLimit:                    dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesTotalSizeLimit),\n\t\tPinotOptimizedQueryColumns:                        dc.GetMapProperty(dynamicproperties.PinotOptimizedQueryColumns),\n\t\tVisibilityArchivalQueryMaxPageSize:                dc.GetIntProperty(dynamicproperties.VisibilityArchivalQueryMaxPageSize),\n\t\tDisallowQuery:                                     dc.GetBoolPropertyFilteredByDomain(dynamicproperties.DisallowQuery),\n\t\tSendRawWorkflowHistory:                            dc.GetBoolPropertyFilteredByDomain(dynamicproperties.SendRawWorkflowHistory),\n\t\tDecisionResultCountLimit:                          dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendDecisionResultCountLimit),\n\t\tEmitSignalNameMetricsTag:                          dc.GetBoolPropertyFilteredByDomain(dynamicproperties.FrontendEmitSignalNameMetricsTag),\n\t\tLockdown:                                          dc.GetBoolPropertyFilteredByDomain(dynamicproperties.Lockdown),\n\t\tEnableTasklistIsolation:                           dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableTasklistIsolation),\n\t\tEnableDomainAuditLogging:                          dc.GetBoolProperty(dynamicproperties.EnableDomainAuditLogging),\n\t\tDomainConfig: domain.Config{\n\t\t\tMaxBadBinaryCount:        dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxBadBinaries),\n\t\t\tMinRetentionDays:         dc.GetIntProperty(dynamicproperties.MinRetentionDays),\n\t\t\tMaxRetentionDays:         dc.GetIntProperty(dynamicproperties.MaxRetentionDays),\n\t\t\tFailoverCoolDown:         dc.GetDurationPropertyFilteredByDomain(dynamicproperties.FrontendFailoverCoolDown),\n\t\t\tRequiredDomainDataKeys:   dc.GetMapProperty(dynamicproperties.RequiredDomainDataKeys),\n\t\t\tFailoverHistoryMaxSize:   dc.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendFailoverHistoryMaxSize),\n\t\t\tEnableDomainAuditLogging: dc.GetBoolProperty(dynamicproperties.EnableDomainAuditLogging),\n\t\t},\n\t\tHostName: hostName,\n\t}\n}\n"
  },
  {
    "path": "service/frontend/config/config_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n)\n\ntype configTestCase struct {\n\tkey   dynamicproperties.Key\n\tvalue interface{}\n}\n\nvar ignoreField = configTestCase{key: dynamicproperties.UnknownIntKey}\n\nfunc TestNewConfig(t *testing.T) {\n\tfields := map[string]configTestCase{\n\t\t\"NumHistoryShards\":                                  {nil, 1001},\n\t\t\"IsAdvancedVisConfigExist\":                          {nil, true},\n\t\t\"HostName\":                                          {nil, \"hostname\"},\n\t\t\"DomainConfig\":                                      ignoreField, // Handle this separately since it's also a config object\n\t\t\"PersistenceMaxQPS\":                                 {dynamicproperties.FrontendPersistenceMaxQPS, 1},\n\t\t\"PersistenceGlobalMaxQPS\":                           {dynamicproperties.FrontendPersistenceGlobalMaxQPS, 2},\n\t\t\"VisibilityMaxPageSize\":                             {dynamicproperties.FrontendVisibilityMaxPageSize, 3},\n\t\t\"EnableVisibilitySampling\":                          {dynamicproperties.EnableVisibilitySampling, true},\n\t\t\"EnableReadFromClosedExecutionV2\":                   {dynamicproperties.EnableReadFromClosedExecutionV2, false},\n\t\t\"VisibilityListMaxQPS\":                              {dynamicproperties.FrontendVisibilityListMaxQPS, 4},\n\t\t\"ESVisibilityListMaxQPS\":                            {dynamicproperties.FrontendESVisibilityListMaxQPS, 5},\n\t\t\"ReadVisibilityStoreName\":                           {dynamicproperties.ReadVisibilityStoreName, \"es\"},\n\t\t\"EnableLogCustomerQueryParameter\":                   {dynamicproperties.EnableLogCustomerQueryParameter, true},\n\t\t\"ESIndexMaxResultWindow\":                            {dynamicproperties.FrontendESIndexMaxResultWindow, 6},\n\t\t\"HistoryMaxPageSize\":                                {dynamicproperties.FrontendHistoryMaxPageSize, 7},\n\t\t\"UserRPS\":                                           {dynamicproperties.FrontendUserRPS, 8},\n\t\t\"WorkerRPS\":                                         {dynamicproperties.FrontendWorkerRPS, 9},\n\t\t\"VisibilityRPS\":                                     {dynamicproperties.FrontendVisibilityRPS, 10},\n\t\t\"AsyncRPS\":                                          {dynamicproperties.FrontendAsyncRPS, 11},\n\t\t\"MaxDomainUserRPSPerInstance\":                       {dynamicproperties.FrontendMaxDomainUserRPSPerInstance, 12},\n\t\t\"MaxDomainWorkerRPSPerInstance\":                     {dynamicproperties.FrontendMaxDomainWorkerRPSPerInstance, 13},\n\t\t\"MaxDomainVisibilityRPSPerInstance\":                 {dynamicproperties.FrontendMaxDomainVisibilityRPSPerInstance, 14},\n\t\t\"MaxDomainAsyncRPSPerInstance\":                      {dynamicproperties.FrontendMaxDomainAsyncRPSPerInstance, 15},\n\t\t\"GlobalDomainUserRPS\":                               {dynamicproperties.FrontendGlobalDomainUserRPS, 16},\n\t\t\"GlobalDomainWorkerRPS\":                             {dynamicproperties.FrontendGlobalDomainWorkerRPS, 17},\n\t\t\"GlobalDomainVisibilityRPS\":                         {dynamicproperties.FrontendGlobalDomainVisibilityRPS, 18},\n\t\t\"GlobalDomainAsyncRPS\":                              {dynamicproperties.FrontendGlobalDomainAsyncRPS, 19},\n\t\t\"MaxWorkerPollDelay\":                                {dynamicproperties.FrontendMaxWorkerPollDelay, time.Duration(30)},\n\t\t\"MaxIDLengthWarnLimit\":                              {dynamicproperties.MaxIDLengthWarnLimit, 20},\n\t\t\"DomainNameMaxLength\":                               {dynamicproperties.DomainNameMaxLength, 21},\n\t\t\"IdentityMaxLength\":                                 {dynamicproperties.IdentityMaxLength, 22},\n\t\t\"WorkflowIDMaxLength\":                               {dynamicproperties.WorkflowIDMaxLength, 23},\n\t\t\"SignalNameMaxLength\":                               {dynamicproperties.SignalNameMaxLength, 24},\n\t\t\"WorkflowTypeMaxLength\":                             {dynamicproperties.WorkflowTypeMaxLength, 25},\n\t\t\"RequestIDMaxLength\":                                {dynamicproperties.RequestIDMaxLength, 26},\n\t\t\"TaskListNameMaxLength\":                             {dynamicproperties.TaskListNameMaxLength, 27},\n\t\t\"EnableAdminProtection\":                             {dynamicproperties.EnableAdminProtection, true},\n\t\t\"AdminOperationToken\":                               {dynamicproperties.AdminOperationToken, \"token\"},\n\t\t\"DisableListVisibilityByFilter\":                     {dynamicproperties.DisableListVisibilityByFilter, false},\n\t\t\"BlobSizeLimitError\":                                {dynamicproperties.BlobSizeLimitError, 29},\n\t\t\"BlobSizeLimitWarn\":                                 {dynamicproperties.BlobSizeLimitWarn, 30},\n\t\t\"ThrottledLogRPS\":                                   {dynamicproperties.FrontendThrottledLogRPS, 31},\n\t\t\"ShutdownDrainDuration\":                             {dynamicproperties.FrontendShutdownDrainDuration, time.Duration(32)},\n\t\t\"WarmupDuration\":                                    {dynamicproperties.FrontendWarmupDuration, time.Duration(40)},\n\t\t\"EnableDomainNotActiveAutoForwarding\":               {dynamicproperties.EnableDomainNotActiveAutoForwarding, true},\n\t\t\"EnableGracefulFailover\":                            {dynamicproperties.EnableGracefulFailover, false},\n\t\t\"DomainFailoverRefreshInterval\":                     {dynamicproperties.DomainFailoverRefreshInterval, time.Duration(33)},\n\t\t\"DomainFailoverRefreshTimerJitterCoefficient\":       {dynamicproperties.DomainFailoverRefreshTimerJitterCoefficient, 34.0},\n\t\t\"EnableActiveClusterSelectionPolicyInStartWorkflow\": {dynamicproperties.EnableActiveClusterSelectionPolicyInStartWorkflow, true},\n\t\t\"EnableClientVersionCheck\":                          {dynamicproperties.EnableClientVersionCheck, true},\n\t\t\"EnableQueryAttributeValidation\":                    {dynamicproperties.EnableQueryAttributeValidation, false},\n\t\t\"ValidSearchAttributes\":                             {dynamicproperties.ValidSearchAttributes, map[string]interface{}{\"foo\": \"bar\"}},\n\t\t\"SearchAttributesNumberOfKeysLimit\":                 {dynamicproperties.SearchAttributesNumberOfKeysLimit, 35},\n\t\t\"SearchAttributesSizeOfValueLimit\":                  {dynamicproperties.SearchAttributesSizeOfValueLimit, 36},\n\t\t\"SearchAttributesTotalSizeLimit\":                    {dynamicproperties.SearchAttributesTotalSizeLimit, 37},\n\t\t\"VisibilityArchivalQueryMaxPageSize\":                {dynamicproperties.VisibilityArchivalQueryMaxPageSize, 38},\n\t\t\"DisallowQuery\":                                     {dynamicproperties.DisallowQuery, true},\n\t\t\"SendRawWorkflowHistory\":                            {dynamicproperties.SendRawWorkflowHistory, false},\n\t\t\"DecisionResultCountLimit\":                          {dynamicproperties.FrontendDecisionResultCountLimit, 39},\n\t\t\"EmitSignalNameMetricsTag\":                          {dynamicproperties.FrontendEmitSignalNameMetricsTag, true},\n\t\t\"Lockdown\":                                          {dynamicproperties.Lockdown, false},\n\t\t\"EnableTasklistIsolation\":                           {dynamicproperties.EnableTasklistIsolation, true},\n\t\t\"GlobalRatelimiterKeyMode\":                          {dynamicproperties.FrontendGlobalRatelimiterMode, \"disabled\"},\n\t\t\"GlobalRatelimiterUpdateInterval\":                   {dynamicproperties.GlobalRatelimiterUpdateInterval, 3 * time.Second},\n\t\t\"PinotOptimizedQueryColumns\":                        {dynamicproperties.PinotOptimizedQueryColumns, map[string]interface{}{\"foo\": \"bar\"}},\n\t\t\"EnableDomainAuditLogging\":                          {dynamicproperties.EnableDomainAuditLogging, true},\n\t\t\"RateLimiterBypassCallerTypes\":                      {dynamicproperties.RateLimiterBypassCallerTypes, []interface{}{\"cli\", \"ui\"}},\n\t}\n\tdomainFields := map[string]configTestCase{\n\t\t\"MaxBadBinaryCount\":        {dynamicproperties.FrontendMaxBadBinaries, 40},\n\t\t\"MinRetentionDays\":         {dynamicproperties.MinRetentionDays, 41},\n\t\t\"MaxRetentionDays\":         {dynamicproperties.MaxRetentionDays, 42},\n\t\t\"FailoverCoolDown\":         {dynamicproperties.FrontendFailoverCoolDown, time.Duration(43)},\n\t\t\"RequiredDomainDataKeys\":   {dynamicproperties.RequiredDomainDataKeys, map[string]interface{}{\"bar\": \"baz\"}},\n\t\t\"FailoverHistoryMaxSize\":   {dynamicproperties.FrontendFailoverHistoryMaxSize, 44},\n\t\t\"EnableDomainAuditLogging\": {dynamicproperties.EnableDomainAuditLogging, true},\n\t}\n\tclient := dynamicconfig.NewInMemoryClient()\n\tlogger := testlogger.New(t)\n\tdc := dynamicconfig.NewCollection(client, logger)\n\n\tconfig := NewConfig(dc, 1001, true, \"hostname\", logger)\n\n\tassertFieldsMatch(t, *config, client, fields)\n\tassertFieldsMatch(t, config.DomainConfig, client, domainFields)\n}\n\nfunc assertFieldsMatch(t *testing.T, config interface{}, client dynamicconfig.Client, fields map[string]configTestCase) {\n\tconfigType := reflect.ValueOf(config)\n\n\tfor i := 0; i < configType.NumField(); i++ {\n\t\tf := configType.Field(i)\n\t\tfieldName := configType.Type().Field(i).Name\n\n\t\tif expected, ok := fields[fieldName]; ok {\n\t\t\tif expected.key == ignoreField.key {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif expected.key != nil {\n\t\t\t\terr := client.UpdateValue(expected.key, expected.value)\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.Errorf(\"Failed to update config for %s: %s\", fieldName, err)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t}\n\t\t\tactual := getValue(&f)\n\t\t\tassert.Equal(t, expected.value, actual)\n\n\t\t} else {\n\t\t\tt.Errorf(\"Unknown property on Config: %s\", fieldName)\n\t\t}\n\t}\n}\n\nfunc getValue(f *reflect.Value) interface{} {\n\tswitch f.Kind() {\n\tcase reflect.Func:\n\t\tswitch fn := f.Interface().(type) {\n\t\tcase dynamicproperties.IntPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.IntPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.DurationPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.DurationPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.FloatPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.MapPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.StringPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.StringPropertyWithRatelimitKeyFilter:\n\t\t\treturn fn(\"user:domain\")\n\t\tcase dynamicproperties.StringPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.ListPropertyFn:\n\t\t\treturn fn()\n\t\tdefault:\n\t\t\tpanic(\"Unable to handle type: \" + f.Type().Name())\n\t\t}\n\tdefault:\n\t\treturn f.Interface()\n\t}\n}\n"
  },
  {
    "path": "service/frontend/service.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage frontend\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/collection\"\n\t\"github.com/uber/cadence/common/quotas/permember\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/frontend/admin\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/accesscontrolled\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/clusterredirection\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/grpc\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/metered\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/ratelimited\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/thrift\"\n\t\"github.com/uber/cadence/service/frontend/wrappers/versioncheck\"\n)\n\n// Service represents the cadence-frontend service\ntype Service struct {\n\tresource.Resource\n\n\tstatus                 int32\n\thandler                *api.WorkflowHandler\n\tadminHandler           admin.Handler\n\tstopC                  chan struct{}\n\tconfig                 *config.Config\n\tparams                 *resource.Params\n\tratelimiterCollections globalRatelimiterCollections\n}\n\n// NewService builds a new cadence-frontend service\nfunc NewService(\n\tparams *resource.Params,\n) (resource.Resource, error) {\n\n\tisAdvancedVisExistInConfig := len(params.PersistenceConfig.AdvancedVisibilityStore) != 0\n\tdc := dynamicconfig.NewCollection(\n\t\tparams.DynamicConfig,\n\t\tparams.Logger,\n\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t)\n\tserviceConfig := config.NewConfig(\n\t\tdc,\n\t\tparams.PersistenceConfig.NumHistoryShards,\n\t\tisAdvancedVisExistInConfig,\n\t\tparams.HostName,\n\t\tparams.Logger,\n\t)\n\n\tserviceResource, err := resource.New(\n\t\tparams,\n\t\tservice.Frontend,\n\t\t&service.Config{\n\t\t\tPersistenceMaxQPS:       serviceConfig.PersistenceMaxQPS,\n\t\t\tPersistenceGlobalMaxQPS: serviceConfig.PersistenceGlobalMaxQPS,\n\t\t\tThrottledLoggerMaxRPS:   serviceConfig.ThrottledLogRPS,\n\n\t\t\tWriteVisibilityStoreName:        nil, // frontend service never write\n\t\t\tEnableLogCustomerQueryParameter: serviceConfig.EnableLogCustomerQueryParameter,\n\t\t\tReadVisibilityStoreName:         serviceConfig.ReadVisibilityStoreName,\n\n\t\t\tEnableDBVisibilitySampling:                  serviceConfig.EnableVisibilitySampling,\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: serviceConfig.EnableReadFromClosedExecutionV2,\n\t\t\tDBVisibilityListMaxQPS:                      serviceConfig.VisibilityListMaxQPS,\n\t\t\tWriteDBVisibilityOpenMaxQPS:                 nil, // frontend service never write\n\t\t\tWriteDBVisibilityClosedMaxQPS:               nil, // frontend service never write\n\n\t\t\tESVisibilityListMaxQPS:     serviceConfig.ESVisibilityListMaxQPS,\n\t\t\tESIndexMaxResultWindow:     serviceConfig.ESIndexMaxResultWindow,\n\t\t\tValidSearchAttributes:      serviceConfig.ValidSearchAttributes,\n\t\t\tIsErrorRetryableFunction:   common.FrontendRetry,\n\t\t\tPinotOptimizedQueryColumns: serviceConfig.PinotOptimizedQueryColumns,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Service{\n\t\tResource: serviceResource,\n\t\tstatus:   common.DaemonStatusInitialized,\n\t\tconfig:   serviceConfig,\n\t\tstopC:    make(chan struct{}),\n\t\tparams:   params,\n\t}, nil\n}\n\n// Start starts the service\nfunc (s *Service) Start() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tlogger := s.GetLogger()\n\tlogger.Info(\"frontend starting\")\n\n\t// domain handler's shared between admin and workflow handler, so instantiate it centrally and share it\n\tdh := domain.NewHandler(\n\t\ts.config.DomainConfig,\n\t\ts.GetLogger(),\n\t\ts.GetDomainManager(),\n\t\ts.GetDomainAuditManager(),\n\t\ts.GetClusterMetadata(),\n\t\tdomain.NewDomainReplicator(s.GetDomainReplicationQueue(), s.GetLogger()),\n\t\ts.GetArchivalMetadata(),\n\t\ts.GetArchiverProvider(),\n\t\ts.GetTimeSource(),\n\t)\n\n\t// Base handler\n\ts.handler = api.NewWorkflowHandler(s, s.config, client.NewVersionChecker(), dh)\n\n\tcollections, err := s.createGlobalQuotaCollections()\n\tif err != nil {\n\t\tlogger.Fatal(\"constructing ratelimiter collections\", tag.Error(err))\n\t}\n\tuserRateLimiter := quotas.NewMultiStageRateLimiter(quotas.NewDynamicRateLimiter(s.config.UserRPS.AsFloat64()), collections.user)\n\tworkerRateLimiter := quotas.NewMultiStageRateLimiter(quotas.NewDynamicRateLimiter(s.config.WorkerRPS.AsFloat64()), collections.worker)\n\tvisibilityRateLimiter := quotas.NewMultiStageRateLimiter(quotas.NewDynamicRateLimiter(s.config.VisibilityRPS.AsFloat64()), collections.visibility)\n\tasyncRateLimiter := quotas.NewMultiStageRateLimiter(quotas.NewDynamicRateLimiter(s.config.AsyncRPS.AsFloat64()), collections.async)\n\n\t// Additional decorations\n\tvar handler api.Handler = s.handler\n\thandler = versioncheck.NewAPIHandler(handler, s.config, client.NewVersionChecker())\n\tcallerBypass := quotas.NewCallerBypass(s.config.RateLimiterBypassCallerTypes)\n\thandler = ratelimited.NewAPIHandler(handler, s.GetDomainCache(), userRateLimiter, workerRateLimiter, visibilityRateLimiter, asyncRateLimiter, s.config.MaxWorkerPollDelay, callerBypass)\n\thandler = metered.NewAPIHandler(handler, s.GetLogger(), s.GetMetricsClient(), s.GetDomainCache(), s.config)\n\tif s.params.ClusterRedirectionPolicy != nil {\n\t\thandler = clusterredirection.NewAPIHandler(handler, s, s.config, *s.params.ClusterRedirectionPolicy)\n\t}\n\thandler = accesscontrolled.NewAPIHandler(handler, s, s.params.Authorizer, s.params.AuthorizationConfig)\n\n\t// Register the latest (most decorated) handler\n\tthriftHandler := thrift.NewAPIHandler(handler)\n\tthriftHandler.Register(s.GetDispatcher())\n\n\tgrpcHandler := grpc.NewAPIHandler(handler)\n\tgrpcHandler.Register(s.GetDispatcher())\n\n\ts.adminHandler = admin.NewHandler(s, s.params, s.config, dh)\n\ts.adminHandler = accesscontrolled.NewAdminHandler(s.adminHandler, s, s.params.Authorizer, s.params.AuthorizationConfig)\n\n\tadminThriftHandler := thrift.NewAdminHandler(s.adminHandler)\n\tadminThriftHandler.Register(s.GetDispatcher())\n\n\tadminGRPCHandler := grpc.NewAdminHandler(s.adminHandler)\n\tadminGRPCHandler.Register(s.GetDispatcher())\n\n\t// must start resource first\n\ts.Resource.Start()\n\n\tstartCtx, cancel := context.WithTimeout(context.Background(), time.Second) // should take nearly no time at all\n\tdefer cancel()\n\tif err := collections.user.OnStart(startCtx); err != nil {\n\t\tlogger.Fatal(\"failed to start user global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := collections.worker.OnStart(startCtx); err != nil {\n\t\tlogger.Fatal(\"failed to start worker global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := collections.visibility.OnStart(startCtx); err != nil {\n\t\tlogger.Fatal(\"failed to start visibility global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := collections.async.OnStart(startCtx); err != nil {\n\t\tlogger.Fatal(\"failed to start async global ratelimiter collection\", tag.Error(err))\n\t}\n\tcancel()\n\ts.ratelimiterCollections = collections // save so they can be stopped later\n\n\ts.handler.Start()\n\ts.adminHandler.Start()\n\n\t// base (service is not started in frontend or admin handler) in case of race condition in yarpc registration function\n\n\tlogger.Info(\"frontend started\")\n\n\t<-s.stopC\n}\n\ntype globalRatelimiterCollections struct {\n\tuser, worker, visibility, async *collection.Collection\n}\n\n// ratelimiterCollections contains the \"base\" ratelimiters that make up both:\n// - \"local\" limits, which do not use global-load-balancing to adjust to request load\n// - fallbacks within \"global\" limits, for when the global-load information cannot be retrieved (startup, errors, etc)\ntype ratelimiterCollections struct {\n\tuser, worker, visibility, async *quotas.Collection\n}\n\nfunc (s *Service) createGlobalQuotaCollections() (globalRatelimiterCollections, error) {\n\tcreate := func(name string, local, global *quotas.Collection, targetRPS dynamicproperties.IntPropertyFnWithDomainFilter) (*collection.Collection, error) {\n\t\tc, err := collection.New(\n\t\t\tname,\n\t\t\tlocal,\n\t\t\tglobal,\n\t\t\ts.config.GlobalRatelimiterUpdateInterval,\n\t\t\ttargetRPS,\n\t\t\ts.config.GlobalRatelimiterKeyMode,\n\t\t\ts.GetRatelimiterAggregatorsClient(),\n\t\t\ts.GetLogger(),\n\t\t\ts.GetMetricsClient(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"error creating %v collection: %w\", name, err)\n\t\t}\n\t\treturn c, nil\n\t}\n\tvar combinedErr error\n\n\t// to safely shadow global ratelimits, we must make duplicate *quota.Collection collections\n\t// so they do not share data when the global limiter decides to use its local fallback.\n\t// these are then combined into the global/algorithm.Collection to handle all limiting calls\n\tlocal, global := s.createBaseLimiters(), s.createBaseLimiters()\n\n\tuser, err := create(\"user\", local.user, global.user, s.config.GlobalDomainUserRPS)\n\tcombinedErr = multierr.Combine(combinedErr, err)\n\n\tworker, err := create(\"worker\", local.worker, global.worker, s.config.GlobalDomainWorkerRPS)\n\tcombinedErr = multierr.Combine(combinedErr, err)\n\n\tvisibility, err := create(\"visibility\", local.visibility, global.visibility, s.config.GlobalDomainVisibilityRPS)\n\tcombinedErr = multierr.Combine(combinedErr, err)\n\n\tasync, err := create(\"async\", local.async, global.async, s.config.GlobalDomainAsyncRPS)\n\tcombinedErr = multierr.Combine(combinedErr, err)\n\n\treturn globalRatelimiterCollections{\n\t\tuser:       user,\n\t\tworker:     worker,\n\t\tvisibility: visibility,\n\t\tasync:      async,\n\t}, combinedErr\n}\nfunc (s *Service) createBaseLimiters() ratelimiterCollections {\n\tcreate := func(shared, perInstance dynamicproperties.IntPropertyFnWithDomainFilter) *quotas.Collection {\n\t\treturn quotas.NewCollection(permember.NewPerMemberDynamicRateLimiterFactory(\n\t\t\tservice.Frontend,\n\t\t\tshared,\n\t\t\tperInstance,\n\t\t\ts.GetMembershipResolver(),\n\t\t))\n\t}\n\treturn ratelimiterCollections{\n\t\tuser:       create(s.config.GlobalDomainUserRPS, s.config.MaxDomainUserRPSPerInstance),\n\t\tworker:     create(s.config.GlobalDomainWorkerRPS, s.config.MaxDomainWorkerRPSPerInstance),\n\t\tvisibility: create(s.config.GlobalDomainVisibilityRPS, s.config.MaxDomainVisibilityRPSPerInstance),\n\t\tasync:      create(s.config.GlobalDomainAsyncRPS, s.config.MaxDomainAsyncRPSPerInstance),\n\t}\n}\n\n// Stop stops the service\nfunc (s *Service) Stop() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\t// initiate graceful shutdown:\n\t// 1. Fail rpc health check, this will cause client side load balancer to stop forwarding requests to this node\n\t// 2. wait for failure detection time\n\t// 3. stop taking new requests by returning InternalServiceError\n\t// 4. Wait for a second\n\t// 5. Stop everything forcefully and return\n\n\trequestDrainTime := min(time.Second, s.config.ShutdownDrainDuration())\n\tfailureDetectionTime := max(0, s.config.ShutdownDrainDuration()-requestDrainTime)\n\n\ts.GetLogger().Info(\"ShutdownHandler: Updating rpc health status to ShuttingDown\")\n\ts.handler.UpdateHealthStatus(api.HealthStatusShuttingDown)\n\n\ts.GetLogger().Info(\"ShutdownHandler: Waiting for others to discover I am unhealthy\")\n\ttime.Sleep(failureDetectionTime)\n\n\ts.handler.Stop()\n\ts.adminHandler.Stop()\n\n\ts.stopRatelimiters()\n\n\ts.GetLogger().Info(\"ShutdownHandler: Draining traffic\")\n\ttime.Sleep(requestDrainTime)\n\n\tclose(s.stopC)\n\ts.Resource.Stop()\n\ts.params.Logger.Info(\"frontend stopped\")\n}\n\nfunc (s *Service) stopRatelimiters() {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second) // should take nearly no time at all\n\tdefer cancel()\n\tif err := s.ratelimiterCollections.user.OnStop(ctx); err != nil {\n\t\ts.GetLogger().Error(\"failed to stop user global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := s.ratelimiterCollections.worker.OnStop(ctx); err != nil {\n\t\ts.GetLogger().Error(\"failed to stop worker global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := s.ratelimiterCollections.visibility.OnStop(ctx); err != nil {\n\t\ts.GetLogger().Error(\"failed to stop visibility global ratelimiter collection\", tag.Error(err))\n\t}\n\tif err := s.ratelimiterCollections.async.OnStop(ctx); err != nil {\n\t\ts.GetLogger().Error(\"failed to stop async global ratelimiter collection\", tag.Error(err))\n\t}\n}\n"
  },
  {
    "path": "service/frontend/templates/accesscontrolled.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t{{- if eq .Interface.Name \"Handler\"}}\n\t\"github.com/uber/cadence/common/metrics\"\n\t{{end}}\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n{{$permissionMap := dict \"CountWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"DeleteDomain\" \"PermissionAdmin\"}}\n{{$permissionMap = set $permissionMap \"DeprecateDomain\" \"PermissionAdmin\"}}\n{{$permissionMap = set $permissionMap \"DescribeDomain\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"DescribeTaskList\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"DescribeWorkflowExecution\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"GetWorkflowExecutionHistory\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"ListArchivedWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"ListClosedWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"ListOpenWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"ListWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"PollForActivityTask\" \"PermissionProcess\"}}\n{{$permissionMap = set $permissionMap \"PollForDecisionTask\" \"PermissionProcess\"}}\n{{$permissionMap = set $permissionMap \"QueryWorkflow\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"RegisterDomain\" \"PermissionAdmin\"}}\n{{$permissionMap = set $permissionMap \"RequestCancelWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"ResetWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"RestartWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"ScanWorkflowExecutions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"SignalWithStartWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"SignalWithStartWorkflowExecutionAsync\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"SignalWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"StartWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"StartWorkflowExecutionAsync\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"TerminateWorkflowExecution\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"ListTaskListPartitions\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"GetTaskListsByDomain\" \"PermissionRead\"}}\n{{$permissionMap = set $permissionMap \"RefreshWorkflowTasks\" \"PermissionWrite\"}}\n{{$permissionMap = set $permissionMap \"UpdateDomain\" \"PermissionAdmin\"}}\n{{$permissionMap = set $permissionMap \"FailoverDomain\" \"PermissionWrite\"}}\n\n{{$adminPermissionMap := dict }}\n{{$adminPermissionMap = set $adminPermissionMap \"DescribeCluster\" \"PermissionRead\"}}\n\n{{$nonDomainAuthAPIs := list \"RegisterDomain\" \"DescribeDomain\" \"UpdateDomain\" \"DeprecateDomain\" \"DeleteDomain\" \"GetSearchAttributes\" \"GetClusterInfo\" \"ResetStickyTaskList\" \"RecordActivityTaskHeartbeat\" \"RespondActivityTaskCanceled\" \"RespondActivityTaskCompleted\" \"RespondActivityTaskFailed\" \"RespondDecisionTaskCompleted\" \"RespondDecisionTaskFailed\" \"RespondQueryTaskCompleted\"}}\n{{$taskListAuthAPIs := list \"PollForActivityTask\" \"PollForDecisionTask\"}}\n{{$workflowTypeAuthAPIs := list \"SignalWithStartWorkflowExecution\" \"StartWorkflowExecution\" \"SignalWithStartWorkflowExecutionAsync\" \"StartWorkflowExecutionAsync\"}}\n\n{{$interfaceName := .Interface.Name}}\n{{$interfaceType := .Interface.Type}}\n{{$handlerName := (index .Vars \"handler\")}}\n{{ $decorator := (printf \"%s%s\" (down $handlerName) $interfaceName) }}\n{{ $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n\n// {{$decorator}} frontend handler wrapper for authentication and authorization\ntype {{$decorator}} struct {\n\thandler {{.Interface.Type}}\n\tauthorizer authorization.Authorizer\n\tresource.Resource\n}\n\n// New{{$Decorator}} creates frontend handler with authentication support\nfunc New{{$Decorator}}(handler {{$.Interface.Type}}, resource resource.Resource, authorizer authorization.Authorizer, cfg config.Authorization) {{.Interface.Type}} {\n\tif authorizer == nil {\n\t\tvar err error\n\t\tauthorizer, err = authorization.NewAuthorizer(cfg, resource.GetLogger(), resource.GetDomainCache())\n\t\tif err != nil {\n\t\t\tresource.GetLogger().Fatal(\"Error when initiating the Authorizer\", tag.Error(err))\n\t\t}\n\t}\n\treturn &{{$decorator}}{\n\t\thandler: handler,\n\t\tauthorizer: authorizer,\n\t\tResource: resource,\n\t}\n}\n\n{{range $method := .Interface.Methods}}\nfunc (a *{{$decorator}}) {{$method.Declaration}} {\n{{- if or (eq $method.Name \"Start\") (eq $method.Name \"Stop\")}}\n\ta.handler.{{$method.Call}}\n}\n{{- else}}\n\t{{- if or (eq $interfaceType \"admin.Handler\") (hasKey $permissionMap $method.Name) }}\n\t{{- if and (eq $interfaceType \"api.Handler\") (ge (len $method.Params) 2)}}\n\t{{- if has $method.Name $nonDomainAuthAPIs}}\n\tscope := a.GetMetricsClient().Scope(metrics.Frontend{{$method.Name}}Scope)\n\t{{- else}}\n\tscope := a.getMetricsScopeWithDomain(metrics.Frontend{{$method.Name}}Scope, {{(index $method.Params 1).Name}}.GetDomain())\n\t{{- end}}\n\t{{- end}}\n\tattr := &authorization.Attributes{\n\t\tAPIName: \"{{$method.Name}}\",\n\t\t{{- if eq $interfaceType \"admin.Handler\"}}\n\t\t{{- if hasKey $adminPermissionMap $method.Name}}\n\t\tPermission: authorization.{{get $adminPermissionMap $method.Name}},\n\t\t{{- else}}\n\t\tPermission:  authorization.PermissionAdmin,\n\t\t{{- end}}\n\t\t{{- else if hasKey $permissionMap $method.Name}}\n\t\tPermission: authorization.{{get $permissionMap $method.Name}},\n\t\t{{- end}}\n\t\t{{- if ge (len $method.Params) 2}}\n\t\tRequestBody: authorization.NewFilteredRequestBody( {{(index $method.Params 1).Name}} ),\n\t\t{{- if not (or (has $method.Name $nonDomainAuthAPIs) (eq $interfaceType \"admin.Handler\"))}}\n\t\tDomainName: {{(index $method.Params 1).Name}}.GetDomain(),\n\t\t{{- else if eq $method.Name \"DescribeDomain\"}}\n\t\tDomainName: {{(index $method.Params 1).Name}}.GetName(),\n\t\t{{- end}}\n\t\t{{- if has $method.Name $taskListAuthAPIs}}\n\t\tTaskList: {{(index $method.Params 1).Name}}.TaskList,\n\t\t{{- end}}\n\t\t{{- if has $method.Name $workflowTypeAuthAPIs}}\n\t\tWorkflowType: {{(index $method.Params 1).Name}}.WorkflowType,\n\t\tTaskList: {{(index $method.Params 1).Name}}.TaskList,\n\t\t{{- end}}\n\t\t{{- end}}\n\t}\n\t{{- if eq $interfaceType \"admin.Handler\"}}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\t{{- else}}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\t{{- end}}\n\tif err != nil {\n\t\t{{- if eq (len $method.Results) 1}}\n\t\treturn err\n\t\t{{- else}}\n\t\treturn nil, err\n\t\t{{- end}}\n\t}\n\tif !isAuthorized {\n\t\t{{- if eq (len $method.Results) 1}}\n\t\treturn errUnauthorized\n\t\t{{- else}}\n\t\treturn nil, errUnauthorized\n\t\t{{- end}}\n\t}\n\t{{- end}}\n\treturn a.handler.{{$method.Call}}\n}\n{{- end}}\n{{end}}\n"
  },
  {
    "path": "service/frontend/templates/clusterredirection.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\n{{$nonForwardingAPIs := list \"Health\" \"DeprecateDomain\" \"DeleteDomain\" \"DescribeDomain\" \"FailoverDomain\" \"ListDomains\" \"RegisterDomain\" \"UpdateDomain\" \"GetSearchAttributes\" \"GetClusterInfo\" \"DiagnoseWorkflowExecution\" \"ListFailoverHistory\"}}\n{{$domainIDAPIs := list \"RecordActivityTaskHeartbeat\" \"RespondActivityTaskCanceled\" \"RespondActivityTaskCompleted\" \"RespondActivityTaskFailed\" \"RespondDecisionTaskCompleted\" \"RespondDecisionTaskFailed\" \"RespondQueryTaskCompleted\"}}\n{{$startWFAPIs := list \"StartWorkflowExecution\" \"StartWorkflowExecutionAsync\" \"SignalWithStartWorkflowExecution\" \"SignalWithStartWorkflowExecutionAsync\"}}\n{{$nonstartWFAPIs := list \"DescribeWorkflowExecutionRequest\" \"GetWorkflowExecutionHistory\" \"QueryWorkflowRequest\" \"RequestCancelWorkflowExecution\" \"ResetWorkflowExecution\" \"RestartWorkflowExecution\" \"SignalWorkflowExecution\" \"TerminateWorkflowExecution\" }}\n{{$queryTaskTokenAPIs := list \"RespondQueryTaskCompleted\"}}\n{{$readAPIsWithStrongConsistency := list \"QueryWorkflow\" \"DescribeWorkflowExecution\" \"GetWorkflowExecutionHistory\"}}\n\ntype (\n\t// ClusterRedirectionHandlerImpl is simple wrapper over frontend service, doing redirection based on policy for global domains not being active in current cluster\n\tclusterRedirectionHandler struct {\n\t\tresource.Resource\n\n\t\tcurrentClusterName string\n\t\tredirectionPolicy  ClusterRedirectionPolicy\n\t\ttokenSerializer    common.TaskTokenSerializer\n\t\tdomainCache        cache.DomainCache\n\t\tconfig             *frontendcfg.Config\n\t\tfrontendHandler    api.Handler\n\t\tcallOptions        []yarpc.CallOption\n\t}\n)\n\n// NewAPIHandler creates a frontend handler to handle cluster redirection for global domains not being active in current cluster\nfunc NewAPIHandler(\n\twfHandler api.Handler,\n\tresource resource.Resource,\n\tconfig *frontendcfg.Config,\n\tpolicy config.ClusterRedirectionPolicy,\n) api.Handler {\n\tdcRedirectionPolicy := RedirectionPolicyGenerator(\n\t\tresource.GetClusterMetadata(),\n\t\tconfig,\n\t\tpolicy,\n\t\tresource.GetLogger(),\n\t\tresource.GetActiveClusterManager(),\n\t\tresource.GetMetricsClient(),\n\t)\n\n\treturn &clusterRedirectionHandler{\n\t\tResource:           resource,\n\t\tcurrentClusterName: resource.GetClusterMetadata().GetCurrentClusterName(),\n\t\tdomainCache:        resource.GetDomainCache(),\n\t\tconfig:             config,\n\t\tredirectionPolicy:  dcRedirectionPolicy,\n\t\ttokenSerializer:    common.NewJSONTaskTokenSerializer(),\n\t\tfrontendHandler:    wfHandler,\n\t\tcallOptions:        []yarpc.CallOption{yarpc.WithHeader(common.AutoforwardingClusterHeaderName, resource.GetClusterMetadata().GetCurrentClusterName())},\n\t}\n}\n\n{{range $method := .Interface.Methods}}\nfunc (handler *clusterRedirectionHandler) {{$method.Declaration}} {\n\t{{- if has $method.Name $nonForwardingAPIs}}\n\treturn handler.frontendHandler.{{$method.Call}}\n\t{{- else}}\n\tvar (\n\t\tapiName = \"{{$method.Name}}\"\n\t\tcluster string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\t{{- if has $method.Name $readAPIsWithStrongConsistency}}\n\t// Only autoforward strong consistent queries, this is done for two reasons:\n\t// 1. Query is meant to be fast, autoforwarding all queries will increase latency.\n\t// 2. If eventual consistency was requested then the results from running out of local dc will be fine.\n\tif {{(index $method.Params 1).Name}}.GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong {\n\t\trequestedConsistencyLevel = types.QueryConsistencyLevelStrong\n\t}\n\t{{- end}}\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirection{{$method.Name}}Scope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\n\t{{if has $method.Name $domainIDAPIs}}\n\tvar idGetter domainIDGetter\n\t{{if has $method.Name $queryTaskTokenAPIs}}\n\tidGetter, err = handler.tokenSerializer.DeserializeQueryTaskToken({{(index $method.Params 1).Name}}.TaskToken)\n\t{{- else}}\n\tidGetter, err = handler.tokenSerializer.Deserialize({{(index $method.Params 1).Name}}.TaskToken)\n\t{{- end}}\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\t{{- else}}\n\tdomainEntry, err = handler.domainCache.GetDomain({{(index $method.Params 1).Name}}.Domain)\n\t{{- end}}\n\tif err != nil {\n\t\t{{- if eq (len $method.Results) 1}}\n\t\treturn err\n\t\t{{- else}}\n\t\treturn nil, err\n\t\t{{- end}}\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\t{{- if has $method.Name $startWFAPIs}}\n\tif !handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow(domainEntry.GetInfo().Name) {\n\t\t{{(index $method.Params 1).Name}}.ActiveClusterSelectionPolicy = nil\n\t}\n\tactClSelPolicyForNewWF = {{(index $method.Params 1).Name}}.ActiveClusterSelectionPolicy\n\t{{- if eq $method.Name \"SignalWithStartWorkflowExecution\"}}\n\tworkflowExecution = &types.WorkflowExecution{\n\t\tWorkflowID: {{(index $method.Params 1).Name}}.GetWorkflowID(),\n\t}\n\t{{- end}}\n\t{{- else if has $method.Name $nonstartWFAPIs}}\n\tworkflowExecution = {{(index $method.Params 1).Name}}.GetWorkflowExecution()\n\t{{- end}}\n\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\t{{$method.ResultsNames}} = handler.frontendHandler.{{$method.Call}}\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\t{{$method.ResultsNames}} = remoteClient.{{$method.Name}}({{$method.Params.Pass}}, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn {{$method.ResultsNames}}\n\t{{- end}}\n}\n{{end}}\n"
  },
  {
    "path": "service/frontend/templates/metered.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\n{{- $nonDomainSpecificAPIs := list \"Health\" \"DeprecateDomain\" \"DeleteDomain\" \"DescribeDomain\" \"ListDomains\" \"RegisterDomain\" \"UpdateDomain\" \"GetSearchAttributes\" \"GetClusterInfo\" \"ListFailoverHistory\"}}\n{{- $domainIDAPIs := list \"RecordActivityTaskHeartbeat\" \"RespondActivityTaskCanceled\" \"RespondActivityTaskCompleted\" \"RespondActivityTaskFailed\" \"RespondDecisionTaskCompleted\" \"RespondDecisionTaskFailed\" \"RespondQueryTaskCompleted\"}}\n{{- $queryTaskTokenAPIs := list \"RespondQueryTaskCompleted\"}}\n{{- $pollerAPIs := list \"PollForActivityTask\" \"PollForDecisionTask\"}}\n\n{{- $interfaceName := .Interface.Name}}\n{{- $interfaceType := .Interface.Type}}\n{{- $handlerName := (index .Vars \"handler\")}}\n{{- $decorator := (printf \"%s%s\" (down $handlerName) $interfaceName) }}\n{{- $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n\n// {{$decorator}} frontend handler wrapper for authentication and authorization\ntype {{$decorator}} struct {\n\thandler {{.Interface.Type}}\n\tlogger log.Logger\n\tmetricsClient metrics.Client\n\tdomainCache cache.DomainCache\n\tcfg *config.Config\n\ttokenSerializer common.TaskTokenSerializer\n\n}\n\n// New{{$Decorator}} creates frontend handler with metrics and logging\nfunc New{{$Decorator}}(handler {{$.Interface.Type}}, logger log.Logger, metricsClient metrics.Client, domainCache cache.DomainCache, cfg *config.Config) {{.Interface.Type}} {\n\treturn &{{$decorator}}{\n\t\thandler: handler,\n\t\tlogger: logger,\n\t\tmetricsClient: metricsClient,\n\t\tdomainCache: domainCache,\n\t\tcfg: cfg,\n\t\ttokenSerializer: common.NewJSONTaskTokenSerializer(),\n\t}\n}\n\n{{range $method := .Interface.Methods}}\nfunc (h *{{$decorator}}) {{$method.Declaration}} {\n\t{{- if eq $method.Name \"Health\"}}\n\t{{ $method.Pass \"h.handler.\" }}\n\t{{- else}}\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\n\t{{- if eq $method.Name \"SignalWorkflowExecution\"}}\n\tctx = h.withSignalName(ctx, {{(index $method.Params 1).Name}}.GetDomain(), {{(index $method.Params 1).Name}}.GetSignalName())\n\t{{- end}}\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"{{$method.Name}}\")}\n\t{{- $scope := printf \"metrics.Frontend%sScope\" $method.Name}}\n\t{{- $domainMetricTag := \"metrics.DomainUnknownTag()\"}}\n\t{{- if not (has $method.Name $nonDomainSpecificAPIs) }}\n\t{{- $domain := printf \"%s.GetDomain()\" (index $method.Params 1).Name}}\n\t{{- if has $method.Name $domainIDAPIs}}\n\t{{- $domain = \"domainName\"}}\n\t{{- if has $method.Name $queryTaskTokenAPIs}}\n\ttoken, err := h.tokenSerializer.DeserializeQueryTaskToken({{(index $method.Params 1).Name}}.TaskToken)\n\t{{- else}}\n\ttoken, err := h.tokenSerializer.Deserialize({{(index $method.Params 1).Name}}.TaskToken)\n\t{{- end}}\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t{{- if has $method.Name $queryTaskTokenAPIs}}\n\ttags = append(tags, tag.WorkflowDomainName(domainName))\n\t{{- else}}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\t{{- end}}\n\t{{- else}}\n\ttags = append(tags, to{{printf \"%sRequest\" $method.Name}}Tags({{(index $method.Params 1).Name}})...)\n\t{{- end}}\n\t{{- $domainMetricTag = printf \"metrics.DomainTag(%s)\" $domain}}\n\t{{- end}}\n\t{{- if has $method.Name $pollerAPIs}}\n\tscope := common.NewPerTaskListScope({{(index $method.Params 1).Name}}.Domain, {{(index $method.Params 1).Name}}.TaskList.GetName(), {{(index $method.Params 1).Name}}.TaskList.GetKind(), h.metricsClient, {{$scope}}).Tagged(metrics.GetContextTags(ctx)...)\n\tscope.IncCounter(metrics.CadenceRequestsPerTaskListWithoutRollup)\n\tsw := scope.StartTimer(metrics.CadenceLatencyPerTaskList)\n\tdefer sw.Stop()\n\tscopePerDomain := h.metricsClient.Scope({{$scope}}).Tagged(append(metrics.GetContextTags(ctx), {{$domainMetricTag}})...)\n\tscopePerDomain.IncCounter(metrics.CadenceRequests)\n\tswPerDomain := scopePerDomain.StartTimer(metrics.CadenceLatency)\n\tdefer swPerDomain.Stop()\n\t{{- else}}\n\tscope := h.metricsClient.Scope({{$scope}}).Tagged(append(metrics.GetContextTags(ctx), {{$domainMetricTag}})...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\t{{- end}}\n\tlogger := h.logger.WithTags(tags...)\n\n\t{{$method.ResultsNames}} = h.handler.{{$method.Call}}\n\tif err != nil {\n\t\t{{- if eq (len $method.Results) 1}}\n\t\treturn h.handleErr(err, scope, logger)\n\t\t{{- else}}\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t\t{{- end}}\n\t}\n\treturn {{$method.ResultsNames}}\n\t{{- end}}\n}\n{{- end}}\n"
  },
  {
    "path": "service/frontend/templates/ratelimited.tmpl",
    "content": "import (\n    \"context\"\n    \"github.com/uber/cadence/common\"\n    \"github.com/uber/cadence/common/cache\"\n    \"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n    \"github.com/uber/cadence/common/quotas\"\n    \"github.com/uber/cadence/service/frontend/api\"\n    \"github.com/uber/cadence/service/frontend/validate\"\n)\n\n{{$ratelimitTypeMap := dict \"PollForActivityTask\" \"ratelimitTypeWorkerPoll\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"PollForDecisionTask\" \"ratelimitTypeWorkerPoll\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RecordActivityTaskHeartbeat\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RecordActivityTaskHeartbeatByID\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ResetStickyTaskList\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskCanceled\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskCanceledByID\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskCompleted\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskCompletedByID\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskFailed\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondActivityTaskFailedByID\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondDecisionTaskCompleted\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondDecisionTaskFailed\" \"ratelimitTypeWorker\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RespondQueryTaskCompleted\" \"ratelimitTypeWorker\"}}\n\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DescribeTaskList\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DescribeWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DiagnoseWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"FailoverDomain\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"GetTaskListsByDomain\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"GetWorkflowExecutionHistory\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListTaskListPartitions\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"QueryWorkflow\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RefreshWorkflowTasks\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RequestCancelWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ResetWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RestartWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"SignalWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"SignalWithStartWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"StartWorkflowExecution\" \"ratelimitTypeUser\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"TerminateWorkflowExecution\" \"ratelimitTypeUser\"}}\n\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"CountWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListArchivedWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListClosedWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListOpenWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ScanWorkflowExecutions\" \"ratelimitTypeVisibility\"}}\n\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"StartWorkflowExecutionAsync\" \"ratelimitTypeAsync\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"SignalWithStartWorkflowExecutionAsync\" \"ratelimitTypeAsync\"}}\n\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"Health\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DeleteDomain\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DeprecateDomain\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"DescribeDomain\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListFailoverHistory\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"GetClusterInfo\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"GetSearchAttributes\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"ListDomains\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"RegisterDomain\" \"ratelimitTypeNoop\"}}\n{{$ratelimitTypeMap = set $ratelimitTypeMap \"UpdateDomain\" \"ratelimitTypeNoop\"}}\n\n{{$domainIDAPIs := list \"RecordActivityTaskHeartbeat\" \"RespondActivityTaskCanceled\" \"RespondActivityTaskCompleted\" \"RespondActivityTaskFailed\" \"RespondDecisionTaskCompleted\" \"RespondDecisionTaskFailed\" \"RespondQueryTaskCompleted\"}}\n{{$queryTaskTokenAPIs := list \"RespondQueryTaskCompleted\"}}\n{{$nonBlockingAPIs := list \"RecordActivityTaskHeartbeat\" \"RecordActivityTaskHeartbeatByID\" \"RespondActivityTaskCompleted\" \"RespondActivityTaskCompletedByID\" \"RespondActivityTaskFailed\" \"RespondActivityTaskFailedByID\" \"RespondActivityTaskCanceled\" \"RespondActivityTaskCanceledByID\" \"RespondDecisionTaskCompleted\" \"RespondDecisionTaskFailed\" \"RespondQueryTaskCompleted\" \"ResetStickyTaskList\"}}\n\n{{$interfaceName := .Interface.Name}}\n{{$handlerName := (index .Vars \"handler\")}}\n{{ $decorator := (printf \"%s%s\" (down $handlerName) $interfaceName) }}\n{{ $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with rate limiter.\ntype {{$decorator}} struct {\n    wrapped     {{.Interface.Type}}\n    tokenSerializer common.TaskTokenSerializer\n    domainCache cache.DomainCache\n    userRateLimiter quotas.Policy\n    workerRateLimiter quotas.Policy\n    visibilityRateLimiter quotas.Policy\n    asyncRateLimiter quotas.Policy\n    maxWorkerPollDelay dynamicproperties.DurationPropertyFnWithDomainFilter\n    callerBypass quotas.CallerBypass\n}\n\n// New{{$Decorator}} creates a new instance of {{$interfaceName}} with ratelimiter.\nfunc New{{$Decorator}}(\n    wrapped {{.Interface.Type}},\n    domainCache cache.DomainCache,\n    userRateLimiter quotas.Policy,\n    workerRateLimiter quotas.Policy,\n    visibilityRateLimiter quotas.Policy,\n    asyncRateLimiter quotas.Policy,\n    maxWorkerPollDelay dynamicproperties.DurationPropertyFnWithDomainFilter,\n    callerBypass quotas.CallerBypass,\n) {{.Interface.Type}} {\n    return &{{$decorator}}{\n        wrapped: wrapped,\n        tokenSerializer: common.NewJSONTaskTokenSerializer(),\n        domainCache: domainCache,\n        userRateLimiter: userRateLimiter,\n        workerRateLimiter: workerRateLimiter,\n        visibilityRateLimiter: visibilityRateLimiter,\n        asyncRateLimiter: asyncRateLimiter,\n        maxWorkerPollDelay: maxWorkerPollDelay,\n        callerBypass: callerBypass,\n    }\n}\n\n{{range $method := .Interface.Methods}}\nfunc (h *{{$decorator}}) {{$method.Declaration}} {\n    {{- $ratelimitType := get $ratelimitTypeMap $method.Name}}\n    {{- if not (eq $ratelimitType \"ratelimitTypeNoop\")}}\n        if {{(index $method.Params 1).Name}} == nil {\n            err = validate.ErrRequestNotSet\n            return\n        }\n        {{- $domain := printf \"%s.GetDomain()\" (index $method.Params 1).Name}}\n        {{- if has $method.Name $domainIDAPIs}}\n            {{- $domain = \"domainName\"}}\n            if {{(index $method.Params 1).Name}}.TaskToken == nil {\n                err = validate.ErrTaskTokenNotSet\n                return\n            }\n            {{- if has $method.Name $queryTaskTokenAPIs}}\n                token, err := h.tokenSerializer.DeserializeQueryTaskToken({{(index $method.Params 1).Name}}.TaskToken)\n            {{- else}}\n                token, err := h.tokenSerializer.Deserialize({{(index $method.Params 1).Name}}.TaskToken)\n            {{- end}}\n            if err != nil {\n                return\n            }\n            if token.DomainID == \"\" {\n                err = validate.ErrDomainNotSet\n                return\n            }\n            domainName, err := h.domainCache.GetDomainName(token.DomainID)\n            if err != nil {\n                return\n            }\n        {{- else}}\n            if {{$domain}} == \"\" {\n                err = validate.ErrDomainNotSet\n                return\n            }\n        {{- end}}\n        {{- if has $method.Name $nonBlockingAPIs}}\n            // Count the request in the host RPS,\n            // but we still accept it even if RPS is exceeded\n            h.allowDomain({{(index $method.Params 0).Name}}, {{$ratelimitType}}, {{$domain}})\n        {{- else}}\n            if limitErr := h.allowDomain({{(index $method.Params 0).Name}}, {{$ratelimitType}}, {{$domain}}); limitErr != nil {\n                err = limitErr\n                return\n            }\n        {{- end}}\n    {{- end}}\n    {{$method.Pass \"h.wrapped.\"}}\n}\n{{end}}\n"
  },
  {
    "path": "service/frontend/templates/versioncheck.tmpl",
    "content": "import (\n    \"context\"\n\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\n{{$excludedAPIs := list \"Health\" \"GetClusterInfo\" \"ListTaskListPartitions\" \"GetTaskListsByDomain\" \"RefreshWorkflowTasks\" }}\n\ntype (\n    versionCheckHandler struct {\n        frontendHandler           api.Handler\n\t\tconfig                    *config.Config\n        versionChecker            client.VersionChecker\n    }\n)\n\nfunc NewAPIHandler(\n\twfHandler api.Handler,\n\tconfig *config.Config,\n\tversionChecker client.VersionChecker,\n) api.Handler {\n\treturn &versionCheckHandler{\n\t\tfrontendHandler: wfHandler,\n        config: config,\n        versionChecker: versionChecker,\n\t}\n}\n\n{{range $method := .Interface.Methods}}\nfunc (h *versionCheckHandler) {{$method.Declaration}} {\n    {{- if not (has $method.Name $excludedAPIs)}}\n    err = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n    if err != nil {\n        return\n    }\n    {{- end}}\n    {{$method.Pass \"h.frontendHandler.\"}}\n}\n{{end}}\n"
  },
  {
    "path": "service/frontend/validate/errors.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage validate\n\nimport (\n\t\"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\nvar (\n\tErrInvalidFilters                             = &types.BadRequestError{Message: \"Request Filters are invalid, unable to parse.\"}\n\tErrDomainNotSet                               = &types.BadRequestError{Message: \"Domain not set on request.\"}\n\tErrTaskTokenNotSet                            = &types.BadRequestError{Message: \"Task token not set on request.\"}\n\tErrInvalidTaskToken                           = &types.BadRequestError{Message: \"Invalid TaskToken.\"}\n\tErrTaskListNotSet                             = &types.BadRequestError{Message: \"TaskList is not set on request.\"}\n\tErrTaskListTypeNotSet                         = &types.BadRequestError{Message: \"TaskListType is not set on request.\"}\n\tErrExecutionNotSet                            = &types.BadRequestError{Message: \"Execution is not set on request.\"}\n\tErrWorkflowIDNotSet                           = &types.BadRequestError{Message: \"WorkflowId is not set on request.\"}\n\tErrActivityIDNotSet                           = &types.BadRequestError{Message: \"ActivityID is not set on request.\"}\n\tErrSignalNameNotSet                           = &types.BadRequestError{Message: \"SignalName is not set on request.\"}\n\tErrInvalidRunID                               = &types.BadRequestError{Message: \"Invalid RunId.\"}\n\tErrInvalidNextPageToken                       = &types.BadRequestError{Message: \"Invalid NextPageToken.\"}\n\tErrNextPageTokenRunIDMismatch                 = &types.BadRequestError{Message: \"RunID in the request does not match the NextPageToken.\"}\n\tErrQueryNotSet                                = &types.BadRequestError{Message: \"WorkflowQuery is not set on request.\"}\n\tErrQueryTypeNotSet                            = &types.BadRequestError{Message: \"QueryType is not set on request.\"}\n\tErrRequestNotSet                              = &types.BadRequestError{Message: \"Request is nil.\"}\n\tErrNoPermission                               = &types.BadRequestError{Message: \"No permission to do this operation.\"}\n\tErrWorkflowTypeNotSet                         = &types.BadRequestError{Message: \"WorkflowType is not set on request.\"}\n\tErrInvalidRetention                           = &types.BadRequestError{Message: \"RetentionDays is invalid.\"}\n\tErrInvalidExecutionStartToCloseTimeoutSeconds = &types.BadRequestError{Message: \"A valid ExecutionStartToCloseTimeoutSeconds is not set on request.\"}\n\tErrInvalidTaskStartToCloseTimeoutSeconds      = &types.BadRequestError{Message: \"A valid TaskStartToCloseTimeoutSeconds is not set on request.\"}\n\tErrInvalidDelayStartSeconds                   = &types.BadRequestError{Message: \"A valid DelayStartSeconds is not set on request.\"}\n\tErrInvalidJitterStartSeconds                  = &types.BadRequestError{Message: \"A valid JitterStartSeconds is not set on request (negative).\"}\n\tErrInvalidJitterStartSeconds2                 = &types.BadRequestError{Message: \"A valid JitterStartSeconds is not set on request (larger than cron duration).\"}\n\tErrQueryDisallowedForDomain                   = &types.BadRequestError{Message: \"Domain is not allowed to query, please contact cadence team to re-enable queries.\"}\n\tErrClusterNameNotSet                          = &types.BadRequestError{Message: \"Cluster name is not set.\"}\n\tErrEmptyReplicationInfo                       = &types.BadRequestError{Message: \"Replication task info is not set.\"}\n\tErrEmptyQueueType                             = &types.BadRequestError{Message: \"Queue type is not set.\"}\n\tErrDomainInLockdown                           = &types.BadRequestError{Message: \"Domain is not accepting fail overs at this time due to lockdown.\"}\n\tErrShuttingDown                               = &types.InternalServiceError{Message: \"Shutting down\"}\n\n\t// Err for archival\n\tErrHistoryNotFound = &types.BadRequestError{Message: \"Requested workflow history not found, may have passed retention period.\"}\n\n\t// Err for string too long\n\tErrDomainTooLong       = &types.BadRequestError{Message: \"Domain length exceeds limit.\"}\n\tErrWorkflowTypeTooLong = &types.BadRequestError{Message: \"WorkflowType length exceeds limit.\"}\n\tErrWorkflowIDTooLong   = &types.BadRequestError{Message: \"WorkflowID length exceeds limit.\"}\n\tErrSignalNameTooLong   = &types.BadRequestError{Message: \"SignalName length exceeds limit.\"}\n\tErrTaskListTooLong     = &types.BadRequestError{Message: \"TaskList length exceeds limit.\"}\n\tErrRequestIDTooLong    = &types.BadRequestError{Message: \"RequestID length exceeds limit.\"}\n\tErrIdentityTooLong     = &types.BadRequestError{Message: \"Identity length exceeds limit.\"}\n)\n\nfunc CheckPermission(\n\tconfig *config.Config,\n\tsecurityToken string,\n) error {\n\tif config.EnableAdminProtection() {\n\t\tif securityToken == \"\" {\n\t\t\treturn ErrNoPermission\n\t\t}\n\t\trequiredToken := config.AdminOperationToken()\n\t\tif securityToken != requiredToken {\n\t\t\treturn ErrNoPermission\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc CheckExecution(w *types.WorkflowExecution) error {\n\tif w == nil {\n\t\treturn ErrExecutionNotSet\n\t}\n\tif w.GetWorkflowID() == \"\" {\n\t\treturn ErrWorkflowIDNotSet\n\t}\n\tif w.GetRunID() != \"\" {\n\t\tif _, err := uuid.Parse(w.GetRunID()); err != nil {\n\t\t\treturn ErrInvalidRunID\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/frontend/wrappers/accesscontrolled/access_controlled.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage accesscontrolled\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar errUnauthorized = &types.AccessDeniedError{Message: \"Request unauthorized.\"}\n\nfunc (a *adminHandler) isAuthorized(ctx context.Context, attr *authorization.Attributes) (bool, error) {\n\tresult, err := a.authorizer.Authorize(ctx, attr)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tisAuth := result.Decision == authorization.DecisionAllow\n\treturn isAuth, nil\n}\n\nfunc (a *apiHandler) isAuthorized(\n\tctx context.Context,\n\tattr *authorization.Attributes,\n\tscope metrics.Scope,\n) (bool, error) {\n\tsw := scope.StartTimer(metrics.CadenceAuthorizationLatency)\n\tdefer sw.Stop()\n\n\tresult, err := a.authorizer.Authorize(ctx, attr)\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceErrAuthorizeFailedCounter)\n\t\treturn false, err\n\t}\n\tisAuth := result.Decision == authorization.DecisionAllow\n\tif !isAuth {\n\t\tscope.IncCounter(metrics.CadenceErrUnauthorizedCounter)\n\t}\n\treturn isAuth, nil\n}\n\n// getMetricsScopeWithDomain return metrics scope with domain tag\nfunc (a *apiHandler) getMetricsScopeWithDomain(\n\tscope metrics.ScopeIdx,\n\tdomain string,\n) metrics.Scope {\n\tif domain != \"\" {\n\t\treturn a.GetMetricsClient().Scope(scope).Tagged(metrics.DomainTag(domain))\n\t}\n\treturn a.GetMetricsClient().Scope(scope).Tagged(metrics.DomainUnknownTag())\n}\n"
  },
  {
    "path": "service/frontend/wrappers/accesscontrolled/access_controlled_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage accesscontrolled\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/admin\"\n)\n\nfunc TestIsAuthorized(t *testing.T) {\n\ttestCases := []struct {\n\t\tname         string\n\t\tmockSetup    func(*authorization.MockAuthorizer, *mocks.Scope)\n\t\tisAuthorized bool\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname: \"Succes case\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, scope *mocks.Scope) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{Decision: authorization.DecisionAllow}, nil)\n\t\t\t\tscope.On(\"StartTimer\", metrics.CadenceAuthorizationLatency).Return(metrics.NewTestStopwatch()).Once()\n\t\t\t},\n\t\t\tisAuthorized: true,\n\t\t\twantErr:      false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - unauthorized\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, scope *mocks.Scope) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{Decision: authorization.DecisionDeny}, nil)\n\t\t\t\tscope.On(\"StartTimer\", metrics.CadenceAuthorizationLatency).Return(metrics.NewTestStopwatch()).Once()\n\t\t\t\tscope.On(\"IncCounter\", metrics.CadenceErrUnauthorizedCounter).Return().Once()\n\t\t\t},\n\t\t\tisAuthorized: false,\n\t\t\twantErr:      false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - authorization error\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, scope *mocks.Scope) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{}, errors.New(\"some random error\"))\n\t\t\t\tscope.On(\"StartTimer\", metrics.CadenceAuthorizationLatency).Return(metrics.NewTestStopwatch()).Once()\n\t\t\t\tscope.On(\"IncCounter\", metrics.CadenceErrAuthorizeFailedCounter).Return().Once()\n\t\t\t},\n\t\t\tisAuthorized: false,\n\t\t\twantErr:      true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockAuthorizer := authorization.NewMockAuthorizer(controller)\n\t\t\tmockMetricsScope := &mocks.Scope{}\n\t\t\ttc.mockSetup(mockAuthorizer, mockMetricsScope)\n\n\t\t\thandler := &apiHandler{authorizer: mockAuthorizer}\n\t\t\tgot, err := handler.isAuthorized(context.Background(), &authorization.Attributes{}, mockMetricsScope)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.isAuthorized, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDescribeCluster(t *testing.T) {\n\tsomeErr := errors.New(\"some random err\")\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*authorization.MockAuthorizer, *admin.MockHandler)\n\t\twantErr   error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, adminHandler *admin.MockHandler) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{Decision: authorization.DecisionAllow}, nil)\n\t\t\t\tadminHandler.EXPECT().DescribeCluster(gomock.Any()).Return(&types.DescribeClusterResponse{}, nil)\n\t\t\t},\n\t\t\twantErr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - unauthorized\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, adminHandler *admin.MockHandler) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{Decision: authorization.DecisionDeny}, nil)\n\t\t\t},\n\t\t\twantErr: errUnauthorized,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - authorization error\",\n\t\t\tmockSetup: func(authorizer *authorization.MockAuthorizer, adminHandler *admin.MockHandler) {\n\t\t\t\tauthorizer.EXPECT().Authorize(gomock.Any(), gomock.Any()).Return(authorization.Result{}, someErr)\n\t\t\t},\n\t\t\twantErr: someErr,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\n\t\t\tmockAuthorizer := authorization.NewMockAuthorizer(controller)\n\t\t\tmockAdminHandler := admin.NewMockHandler(controller)\n\t\t\ttc.mockSetup(mockAuthorizer, mockAdminHandler)\n\n\t\t\thandler := &adminHandler{authorizer: mockAuthorizer, handler: mockAdminHandler}\n\t\t\t_, err := handler.DescribeCluster(context.Background())\n\t\t\tif tc.wantErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorIs(t, err, tc.wantErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/accesscontrolled/admin_generated.go",
    "content": "package accesscontrolled\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/accesscontrolled.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/admin\"\n)\n\n// adminHandler frontend handler wrapper for authentication and authorization\ntype adminHandler struct {\n\thandler    admin.Handler\n\tauthorizer authorization.Authorizer\n\tresource.Resource\n}\n\n// NewAdminHandler creates frontend handler with authentication support\nfunc NewAdminHandler(handler admin.Handler, resource resource.Resource, authorizer authorization.Authorizer, cfg config.Authorization) admin.Handler {\n\tif authorizer == nil {\n\t\tvar err error\n\t\tauthorizer, err = authorization.NewAuthorizer(cfg, resource.GetLogger(), resource.GetDomainCache())\n\t\tif err != nil {\n\t\t\tresource.GetLogger().Fatal(\"Error when initiating the Authorizer\", tag.Error(err))\n\t\t}\n\t}\n\treturn &adminHandler{\n\t\thandler:    handler,\n\t\tauthorizer: authorizer,\n\t\tResource:   resource,\n\t}\n}\n\nfunc (a *adminHandler) AddSearchAttribute(ctx context.Context, ap1 *types.AddSearchAttributeRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"AddSearchAttribute\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(ap1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.AddSearchAttribute(ctx, ap1)\n}\n\nfunc (a *adminHandler) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"CloseShard\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(cp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.CloseShard(ctx, cp1)\n}\n\nfunc (a *adminHandler) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest) (cp2 *types.CountDLQMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"CountDLQMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(cp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.CountDLQMessages(ctx, cp1)\n}\n\nfunc (a *adminHandler) DeleteWorkflow(ctx context.Context, ap1 *types.AdminDeleteWorkflowRequest) (ap2 *types.AdminDeleteWorkflowResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DeleteWorkflow\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(ap1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DeleteWorkflow(ctx, ap1)\n}\n\nfunc (a *adminHandler) DescribeCluster(ctx context.Context) (dp1 *types.DescribeClusterResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:    \"DescribeCluster\",\n\t\tPermission: authorization.PermissionRead,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeCluster(ctx)\n}\n\nfunc (a *adminHandler) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeHistoryHost\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeHistoryHost(ctx, dp1)\n}\n\nfunc (a *adminHandler) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest) (dp2 *types.DescribeQueueResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeQueue\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeQueue(ctx, dp1)\n}\n\nfunc (a *adminHandler) DescribeShardDistribution(ctx context.Context, dp1 *types.DescribeShardDistributionRequest) (dp2 *types.DescribeShardDistributionResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeShardDistribution\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeShardDistribution(ctx, dp1)\n}\n\nfunc (a *adminHandler) DescribeWorkflowExecution(ctx context.Context, ap1 *types.AdminDescribeWorkflowExecutionRequest) (ap2 *types.AdminDescribeWorkflowExecutionResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeWorkflowExecution\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(ap1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeWorkflowExecution(ctx, ap1)\n}\n\nfunc (a *adminHandler) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetCrossClusterTasks\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetCrossClusterTasks(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetDLQReplicationMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetDLQReplicationMessages(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, gp1 *types.GetDomainAsyncWorkflowConfiguratonRequest) (gp2 *types.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetDomainAsyncWorkflowConfiguraton\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetDomainAsyncWorkflowConfiguraton(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetDomainIsolationGroups(ctx context.Context, request *types.GetDomainIsolationGroupsRequest) (gp1 *types.GetDomainIsolationGroupsResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetDomainIsolationGroups\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(request),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetDomainIsolationGroups(ctx, request)\n}\n\nfunc (a *adminHandler) GetDomainReplicationMessages(ctx context.Context, gp1 *types.GetDomainReplicationMessagesRequest) (gp2 *types.GetDomainReplicationMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetDomainReplicationMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetDomainReplicationMessages(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetDynamicConfig(ctx context.Context, gp1 *types.GetDynamicConfigRequest) (gp2 *types.GetDynamicConfigResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetDynamicConfig\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetDynamicConfig(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetGlobalIsolationGroups(ctx context.Context, request *types.GetGlobalIsolationGroupsRequest) (gp1 *types.GetGlobalIsolationGroupsResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetGlobalIsolationGroups\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(request),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetGlobalIsolationGroups(ctx, request)\n}\n\nfunc (a *adminHandler) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetReplicationMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetReplicationMessages(ctx, gp1)\n}\n\nfunc (a *adminHandler) GetWorkflowExecutionRawHistoryV2(ctx context.Context, gp1 *types.GetWorkflowExecutionRawHistoryV2Request) (gp2 *types.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetWorkflowExecutionRawHistoryV2\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetWorkflowExecutionRawHistoryV2(ctx, gp1)\n}\n\nfunc (a *adminHandler) ListDynamicConfig(ctx context.Context, lp1 *types.ListDynamicConfigRequest) (lp2 *types.ListDynamicConfigResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListDynamicConfig\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListDynamicConfig(ctx, lp1)\n}\n\nfunc (a *adminHandler) MaintainCorruptWorkflow(ctx context.Context, ap1 *types.AdminMaintainWorkflowRequest) (ap2 *types.AdminMaintainWorkflowResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"MaintainCorruptWorkflow\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(ap1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.MaintainCorruptWorkflow(ctx, ap1)\n}\n\nfunc (a *adminHandler) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"MergeDLQMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(mp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.MergeDLQMessages(ctx, mp1)\n}\n\nfunc (a *adminHandler) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"PurgeDLQMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(pp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.PurgeDLQMessages(ctx, pp1)\n}\n\nfunc (a *adminHandler) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ReadDLQMessages\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ReadDLQMessages(ctx, rp1)\n}\n\nfunc (a *adminHandler) ReapplyEvents(ctx context.Context, rp1 *types.ReapplyEventsRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ReapplyEvents\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.ReapplyEvents(ctx, rp1)\n}\n\nfunc (a *adminHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RefreshWorkflowTasks\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RefreshWorkflowTasks(ctx, rp1)\n}\n\nfunc (a *adminHandler) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RemoveTask\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RemoveTask(ctx, rp1)\n}\n\nfunc (a *adminHandler) ResendReplicationTasks(ctx context.Context, rp1 *types.ResendReplicationTasksRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ResendReplicationTasks\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.ResendReplicationTasks(ctx, rp1)\n}\n\nfunc (a *adminHandler) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ResetQueue\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.ResetQueue(ctx, rp1)\n}\n\nfunc (a *adminHandler) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RespondCrossClusterTasksCompleted\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.RespondCrossClusterTasksCompleted(ctx, rp1)\n}\n\nfunc (a *adminHandler) RestoreDynamicConfig(ctx context.Context, rp1 *types.RestoreDynamicConfigRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RestoreDynamicConfig\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RestoreDynamicConfig(ctx, rp1)\n}\n\nfunc (a *adminHandler) Start() {\n\ta.handler.Start()\n}\n\nfunc (a *adminHandler) Stop() {\n\ta.handler.Stop()\n}\n\nfunc (a *adminHandler) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, up1 *types.UpdateDomainAsyncWorkflowConfiguratonRequest) (up2 *types.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateDomainAsyncWorkflowConfiguraton\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(up1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.UpdateDomainAsyncWorkflowConfiguraton(ctx, up1)\n}\n\nfunc (a *adminHandler) UpdateDomainIsolationGroups(ctx context.Context, request *types.UpdateDomainIsolationGroupsRequest) (up1 *types.UpdateDomainIsolationGroupsResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateDomainIsolationGroups\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(request),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.UpdateDomainIsolationGroups(ctx, request)\n}\n\nfunc (a *adminHandler) UpdateDynamicConfig(ctx context.Context, up1 *types.UpdateDynamicConfigRequest) (err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateDynamicConfig\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(up1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.UpdateDynamicConfig(ctx, up1)\n}\n\nfunc (a *adminHandler) UpdateGlobalIsolationGroups(ctx context.Context, request *types.UpdateGlobalIsolationGroupsRequest) (up1 *types.UpdateGlobalIsolationGroupsResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateGlobalIsolationGroups\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(request),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.UpdateGlobalIsolationGroups(ctx, request)\n}\n\nfunc (a *adminHandler) UpdateTaskListPartitionConfig(ctx context.Context, up1 *types.UpdateTaskListPartitionConfigRequest) (up2 *types.UpdateTaskListPartitionConfigResponse, err error) {\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateTaskListPartitionConfig\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(up1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.UpdateTaskListPartitionConfig(ctx, up1)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/accesscontrolled/api_generated.go",
    "content": "package accesscontrolled\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/accesscontrolled.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n)\n\n// apiHandler frontend handler wrapper for authentication and authorization\ntype apiHandler struct {\n\thandler    api.Handler\n\tauthorizer authorization.Authorizer\n\tresource.Resource\n}\n\n// NewAPIHandler creates frontend handler with authentication support\nfunc NewAPIHandler(handler api.Handler, resource resource.Resource, authorizer authorization.Authorizer, cfg config.Authorization) api.Handler {\n\tif authorizer == nil {\n\t\tvar err error\n\t\tauthorizer, err = authorization.NewAuthorizer(cfg, resource.GetLogger(), resource.GetDomainCache())\n\t\tif err != nil {\n\t\t\tresource.GetLogger().Fatal(\"Error when initiating the Authorizer\", tag.Error(err))\n\t\t}\n\t}\n\treturn &apiHandler{\n\t\thandler:    handler,\n\t\tauthorizer: authorizer,\n\t\tResource:   resource,\n\t}\n}\n\nfunc (a *apiHandler) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendCountWorkflowExecutionsScope, cp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"CountWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(cp1),\n\t\tDomainName:  cp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.CountWorkflowExecutions(ctx, cp1)\n}\n\nfunc (a *apiHandler) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest) (err error) {\n\tscope := a.GetMetricsClient().Scope(metrics.FrontendDeleteDomainScope)\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DeleteDomain\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.DeleteDomain(ctx, dp1)\n}\n\nfunc (a *apiHandler) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest) (err error) {\n\tscope := a.GetMetricsClient().Scope(metrics.FrontendDeprecateDomainScope)\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DeprecateDomain\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.DeprecateDomain(ctx, dp1)\n}\n\nfunc (a *apiHandler) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest) (dp2 *types.DescribeDomainResponse, err error) {\n\tscope := a.GetMetricsClient().Scope(metrics.FrontendDescribeDomainScope)\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeDomain\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t\tDomainName:  dp1.GetName(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeDomain(ctx, dp1)\n}\n\nfunc (a *apiHandler) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest) (dp2 *types.DescribeTaskListResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendDescribeTaskListScope, dp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeTaskList\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t\tDomainName:  dp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeTaskList(ctx, dp1)\n}\n\nfunc (a *apiHandler) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendDescribeWorkflowExecutionScope, dp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"DescribeWorkflowExecution\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(dp1),\n\t\tDomainName:  dp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.DescribeWorkflowExecution(ctx, dp1)\n}\n\nfunc (a *apiHandler) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\treturn a.handler.DiagnoseWorkflowExecution(ctx, dp1)\n}\n\nfunc (a *apiHandler) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest) (fp2 *types.FailoverDomainResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendFailoverDomainScope, fp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"FailoverDomain\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(fp1),\n\t\tDomainName:  fp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.FailoverDomain(ctx, fp1)\n}\n\nfunc (a *apiHandler) GetClusterInfo(ctx context.Context) (cp1 *types.ClusterInfo, err error) {\n\treturn a.handler.GetClusterInfo(ctx)\n}\n\nfunc (a *apiHandler) GetSearchAttributes(ctx context.Context) (gp1 *types.GetSearchAttributesResponse, err error) {\n\treturn a.handler.GetSearchAttributes(ctx)\n}\n\nfunc (a *apiHandler) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendGetTaskListsByDomainScope, gp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetTaskListsByDomain\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t\tDomainName:  gp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetTaskListsByDomain(ctx, gp1)\n}\n\nfunc (a *apiHandler) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendGetWorkflowExecutionHistoryScope, gp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"GetWorkflowExecutionHistory\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(gp1),\n\t\tDomainName:  gp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.GetWorkflowExecutionHistory(ctx, gp1)\n}\n\nfunc (a *apiHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn a.handler.Health(ctx)\n}\n\nfunc (a *apiHandler) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendListArchivedWorkflowExecutionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListArchivedWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListArchivedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendListClosedWorkflowExecutionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListClosedWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListClosedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest) (lp2 *types.ListDomainsResponse, err error) {\n\treturn a.handler.ListDomains(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\treturn a.handler.ListFailoverHistory(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendListOpenWorkflowExecutionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListOpenWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListOpenWorkflowExecutions(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendListTaskListPartitionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListTaskListPartitions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListTaskListPartitions(ctx, lp1)\n}\n\nfunc (a *apiHandler) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendListWorkflowExecutionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ListWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ListWorkflowExecutions(ctx, lp1)\n}\n\nfunc (a *apiHandler) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendPollForActivityTaskScope, pp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"PollForActivityTask\",\n\t\tPermission:  authorization.PermissionProcess,\n\t\tRequestBody: authorization.NewFilteredRequestBody(pp1),\n\t\tDomainName:  pp1.GetDomain(),\n\t\tTaskList:    pp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.PollForActivityTask(ctx, pp1)\n}\n\nfunc (a *apiHandler) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendPollForDecisionTaskScope, pp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"PollForDecisionTask\",\n\t\tPermission:  authorization.PermissionProcess,\n\t\tRequestBody: authorization.NewFilteredRequestBody(pp1),\n\t\tDomainName:  pp1.GetDomain(),\n\t\tTaskList:    pp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.PollForDecisionTask(ctx, pp1)\n}\n\nfunc (a *apiHandler) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest) (qp2 *types.QueryWorkflowResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendQueryWorkflowScope, qp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"QueryWorkflow\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(qp1),\n\t\tDomainName:  qp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.QueryWorkflow(ctx, qp1)\n}\n\nfunc (a *apiHandler) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\treturn a.handler.RecordActivityTaskHeartbeat(ctx, rp1)\n}\n\nfunc (a *apiHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\treturn a.handler.RecordActivityTaskHeartbeatByID(ctx, rp1)\n}\n\nfunc (a *apiHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendRefreshWorkflowTasksScope, rp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RefreshWorkflowTasks\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t\tDomainName:  rp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RefreshWorkflowTasks(ctx, rp1)\n}\n\nfunc (a *apiHandler) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest) (err error) {\n\tscope := a.GetMetricsClient().Scope(metrics.FrontendRegisterDomainScope)\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RegisterDomain\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RegisterDomain(ctx, rp1)\n}\n\nfunc (a *apiHandler) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest) (err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendRequestCancelWorkflowExecutionScope, rp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RequestCancelWorkflowExecution\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t\tDomainName:  rp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.RequestCancelWorkflowExecution(ctx, rp1)\n}\n\nfunc (a *apiHandler) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\treturn a.handler.ResetStickyTaskList(ctx, rp1)\n}\n\nfunc (a *apiHandler) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendResetWorkflowExecutionScope, rp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ResetWorkflowExecution\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t\tDomainName:  rp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ResetWorkflowExecution(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest) (err error) {\n\treturn a.handler.RespondActivityTaskCanceled(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest) (err error) {\n\treturn a.handler.RespondActivityTaskCanceledByID(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest) (err error) {\n\treturn a.handler.RespondActivityTaskCompleted(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest) (err error) {\n\treturn a.handler.RespondActivityTaskCompletedByID(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest) (err error) {\n\treturn a.handler.RespondActivityTaskFailed(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest) (err error) {\n\treturn a.handler.RespondActivityTaskFailedByID(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\treturn a.handler.RespondDecisionTaskCompleted(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest) (err error) {\n\treturn a.handler.RespondDecisionTaskFailed(ctx, rp1)\n}\n\nfunc (a *apiHandler) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest) (err error) {\n\treturn a.handler.RespondQueryTaskCompleted(ctx, rp1)\n}\n\nfunc (a *apiHandler) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendRestartWorkflowExecutionScope, rp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"RestartWorkflowExecution\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(rp1),\n\t\tDomainName:  rp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.RestartWorkflowExecution(ctx, rp1)\n}\n\nfunc (a *apiHandler) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendScanWorkflowExecutionsScope, lp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"ScanWorkflowExecutions\",\n\t\tPermission:  authorization.PermissionRead,\n\t\tRequestBody: authorization.NewFilteredRequestBody(lp1),\n\t\tDomainName:  lp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.ScanWorkflowExecutions(ctx, lp1)\n}\n\nfunc (a *apiHandler) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendSignalWithStartWorkflowExecutionScope, sp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:      \"SignalWithStartWorkflowExecution\",\n\t\tPermission:   authorization.PermissionWrite,\n\t\tRequestBody:  authorization.NewFilteredRequestBody(sp1),\n\t\tDomainName:   sp1.GetDomain(),\n\t\tWorkflowType: sp1.WorkflowType,\n\t\tTaskList:     sp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.SignalWithStartWorkflowExecution(ctx, sp1)\n}\n\nfunc (a *apiHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendSignalWithStartWorkflowExecutionAsyncScope, sp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:      \"SignalWithStartWorkflowExecutionAsync\",\n\t\tPermission:   authorization.PermissionWrite,\n\t\tRequestBody:  authorization.NewFilteredRequestBody(sp1),\n\t\tDomainName:   sp1.GetDomain(),\n\t\tWorkflowType: sp1.WorkflowType,\n\t\tTaskList:     sp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.SignalWithStartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (a *apiHandler) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest) (err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendSignalWorkflowExecutionScope, sp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"SignalWorkflowExecution\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(sp1),\n\t\tDomainName:  sp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.SignalWorkflowExecution(ctx, sp1)\n}\n\nfunc (a *apiHandler) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendStartWorkflowExecutionScope, sp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:      \"StartWorkflowExecution\",\n\t\tPermission:   authorization.PermissionWrite,\n\t\tRequestBody:  authorization.NewFilteredRequestBody(sp1),\n\t\tDomainName:   sp1.GetDomain(),\n\t\tWorkflowType: sp1.WorkflowType,\n\t\tTaskList:     sp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.StartWorkflowExecution(ctx, sp1)\n}\n\nfunc (a *apiHandler) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendStartWorkflowExecutionAsyncScope, sp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:      \"StartWorkflowExecutionAsync\",\n\t\tPermission:   authorization.PermissionWrite,\n\t\tRequestBody:  authorization.NewFilteredRequestBody(sp1),\n\t\tDomainName:   sp1.GetDomain(),\n\t\tWorkflowType: sp1.WorkflowType,\n\t\tTaskList:     sp1.TaskList,\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.StartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (a *apiHandler) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest) (err error) {\n\tscope := a.getMetricsScopeWithDomain(metrics.FrontendTerminateWorkflowExecutionScope, tp1.GetDomain())\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"TerminateWorkflowExecution\",\n\t\tPermission:  authorization.PermissionWrite,\n\t\tRequestBody: authorization.NewFilteredRequestBody(tp1),\n\t\tDomainName:  tp1.GetDomain(),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !isAuthorized {\n\t\treturn errUnauthorized\n\t}\n\treturn a.handler.TerminateWorkflowExecution(ctx, tp1)\n}\n\nfunc (a *apiHandler) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest) (up2 *types.UpdateDomainResponse, err error) {\n\tscope := a.GetMetricsClient().Scope(metrics.FrontendUpdateDomainScope)\n\tattr := &authorization.Attributes{\n\t\tAPIName:     \"UpdateDomain\",\n\t\tPermission:  authorization.PermissionAdmin,\n\t\tRequestBody: authorization.NewFilteredRequestBody(up1),\n\t}\n\tisAuthorized, err := a.isAuthorized(ctx, attr, scope)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !isAuthorized {\n\t\treturn nil, errUnauthorized\n\t}\n\treturn a.handler.UpdateDomain(ctx, up1)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/api_generated.go",
    "content": "package clusterredirection\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/clusterredirection.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\ntype (\n\t// ClusterRedirectionHandlerImpl is simple wrapper over frontend service, doing redirection based on policy for global domains not being active in current cluster\n\tclusterRedirectionHandler struct {\n\t\tresource.Resource\n\n\t\tcurrentClusterName string\n\t\tredirectionPolicy  ClusterRedirectionPolicy\n\t\ttokenSerializer    common.TaskTokenSerializer\n\t\tdomainCache        cache.DomainCache\n\t\tconfig             *frontendcfg.Config\n\t\tfrontendHandler    api.Handler\n\t\tcallOptions        []yarpc.CallOption\n\t}\n)\n\n// NewAPIHandler creates a frontend handler to handle cluster redirection for global domains not being active in current cluster\nfunc NewAPIHandler(\n\twfHandler api.Handler,\n\tresource resource.Resource,\n\tconfig *frontendcfg.Config,\n\tpolicy config.ClusterRedirectionPolicy,\n) api.Handler {\n\tdcRedirectionPolicy := RedirectionPolicyGenerator(\n\t\tresource.GetClusterMetadata(),\n\t\tconfig,\n\t\tpolicy,\n\t\tresource.GetLogger(),\n\t\tresource.GetActiveClusterManager(),\n\t\tresource.GetMetricsClient(),\n\t)\n\n\treturn &clusterRedirectionHandler{\n\t\tResource:           resource,\n\t\tcurrentClusterName: resource.GetClusterMetadata().GetCurrentClusterName(),\n\t\tdomainCache:        resource.GetDomainCache(),\n\t\tconfig:             config,\n\t\tredirectionPolicy:  dcRedirectionPolicy,\n\t\ttokenSerializer:    common.NewJSONTaskTokenSerializer(),\n\t\tfrontendHandler:    wfHandler,\n\t\tcallOptions:        []yarpc.CallOption{yarpc.WithHeader(common.AutoforwardingClusterHeaderName, resource.GetClusterMetadata().GetCurrentClusterName())},\n\t}\n}\n\nfunc (handler *clusterRedirectionHandler) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"CountWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionCountWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(cp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tcp2, err = handler.frontendHandler.CountWorkflowExecutions(ctx, cp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tcp2, err = remoteClient.CountWorkflowExecutions(ctx, cp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn cp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest) (err error) {\n\treturn handler.frontendHandler.DeleteDomain(ctx, dp1)\n}\n\nfunc (handler *clusterRedirectionHandler) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest) (err error) {\n\treturn handler.frontendHandler.DeprecateDomain(ctx, dp1)\n}\n\nfunc (handler *clusterRedirectionHandler) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest) (dp2 *types.DescribeDomainResponse, err error) {\n\treturn handler.frontendHandler.DescribeDomain(ctx, dp1)\n}\n\nfunc (handler *clusterRedirectionHandler) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest) (dp2 *types.DescribeTaskListResponse, err error) {\n\tvar (\n\t\tapiName                   = \"DescribeTaskList\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionDescribeTaskListScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(dp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tdp2, err = handler.frontendHandler.DescribeTaskList(ctx, dp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tdp2, err = remoteClient.DescribeTaskList(ctx, dp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn dp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tvar (\n\t\tapiName                   = \"DescribeWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\t// Only autoforward strong consistent queries, this is done for two reasons:\n\t// 1. Query is meant to be fast, autoforwarding all queries will increase latency.\n\t// 2. If eventual consistency was requested then the results from running out of local dc will be fine.\n\tif dp1.GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong {\n\t\trequestedConsistencyLevel = types.QueryConsistencyLevelStrong\n\t}\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionDescribeWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(dp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tdp2, err = handler.frontendHandler.DescribeWorkflowExecution(ctx, dp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tdp2, err = remoteClient.DescribeWorkflowExecution(ctx, dp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn dp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\treturn handler.frontendHandler.DiagnoseWorkflowExecution(ctx, dp1)\n}\n\nfunc (handler *clusterRedirectionHandler) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest) (fp2 *types.FailoverDomainResponse, err error) {\n\treturn handler.frontendHandler.FailoverDomain(ctx, fp1)\n}\n\nfunc (handler *clusterRedirectionHandler) GetClusterInfo(ctx context.Context) (cp1 *types.ClusterInfo, err error) {\n\treturn handler.frontendHandler.GetClusterInfo(ctx)\n}\n\nfunc (handler *clusterRedirectionHandler) GetSearchAttributes(ctx context.Context) (gp1 *types.GetSearchAttributesResponse, err error) {\n\treturn handler.frontendHandler.GetSearchAttributes(ctx)\n}\n\nfunc (handler *clusterRedirectionHandler) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tvar (\n\t\tapiName                   = \"GetTaskListsByDomain\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionGetTaskListsByDomainScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(gp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tgp2, err = handler.frontendHandler.GetTaskListsByDomain(ctx, gp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tgp2, err = remoteClient.GetTaskListsByDomain(ctx, gp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn gp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tvar (\n\t\tapiName                   = \"GetWorkflowExecutionHistory\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\t// Only autoforward strong consistent queries, this is done for two reasons:\n\t// 1. Query is meant to be fast, autoforwarding all queries will increase latency.\n\t// 2. If eventual consistency was requested then the results from running out of local dc will be fine.\n\tif gp1.GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong {\n\t\trequestedConsistencyLevel = types.QueryConsistencyLevelStrong\n\t}\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionGetWorkflowExecutionHistoryScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(gp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = gp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tgp2, err = handler.frontendHandler.GetWorkflowExecutionHistory(ctx, gp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tgp2, err = remoteClient.GetWorkflowExecutionHistory(ctx, gp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn gp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn handler.frontendHandler.Health(ctx)\n}\n\nfunc (handler *clusterRedirectionHandler) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ListArchivedWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionListArchivedWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ListArchivedWorkflowExecutions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ListArchivedWorkflowExecutions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ListClosedWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionListClosedWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ListClosedWorkflowExecutions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ListClosedWorkflowExecutions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest) (lp2 *types.ListDomainsResponse, err error) {\n\treturn handler.frontendHandler.ListDomains(ctx, lp1)\n}\n\nfunc (handler *clusterRedirectionHandler) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\treturn handler.frontendHandler.ListFailoverHistory(ctx, lp1)\n}\n\nfunc (handler *clusterRedirectionHandler) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ListOpenWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionListOpenWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ListOpenWorkflowExecutions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ListOpenWorkflowExecutions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ListTaskListPartitions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionListTaskListPartitionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ListTaskListPartitions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ListTaskListPartitions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ListWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionListWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ListWorkflowExecutions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ListWorkflowExecutions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tvar (\n\t\tapiName                   = \"PollForActivityTask\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionPollForActivityTaskScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(pp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tpp2, err = handler.frontendHandler.PollForActivityTask(ctx, pp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tpp2, err = remoteClient.PollForActivityTask(ctx, pp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn pp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tvar (\n\t\tapiName                   = \"PollForDecisionTask\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionPollForDecisionTaskScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(pp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tpp2, err = handler.frontendHandler.PollForDecisionTask(ctx, pp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tpp2, err = remoteClient.PollForDecisionTask(ctx, pp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn pp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest) (qp2 *types.QueryWorkflowResponse, err error) {\n\tvar (\n\t\tapiName                   = \"QueryWorkflow\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\t// Only autoforward strong consistent queries, this is done for two reasons:\n\t// 1. Query is meant to be fast, autoforwarding all queries will increase latency.\n\t// 2. If eventual consistency was requested then the results from running out of local dc will be fine.\n\tif qp1.GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong {\n\t\trequestedConsistencyLevel = types.QueryConsistencyLevelStrong\n\t}\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionQueryWorkflowScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(qp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tqp2, err = handler.frontendHandler.QueryWorkflow(ctx, qp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tqp2, err = remoteClient.QueryWorkflow(ctx, qp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn qp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tvar (\n\t\tapiName                   = \"RecordActivityTaskHeartbeat\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRecordActivityTaskHeartbeatScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.RecordActivityTaskHeartbeat(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.RecordActivityTaskHeartbeat(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tvar (\n\t\tapiName                   = \"RecordActivityTaskHeartbeatByID\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRecordActivityTaskHeartbeatByIDScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.RecordActivityTaskHeartbeatByID(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.RecordActivityTaskHeartbeatByID(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RefreshWorkflowTasks\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRefreshWorkflowTasksScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RefreshWorkflowTasks(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RefreshWorkflowTasks(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest) (err error) {\n\treturn handler.frontendHandler.RegisterDomain(ctx, rp1)\n}\n\nfunc (handler *clusterRedirectionHandler) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RequestCancelWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRequestCancelWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = rp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RequestCancelWorkflowExecution(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RequestCancelWorkflowExecution(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ResetStickyTaskList\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionResetStickyTaskListScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.ResetStickyTaskList(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.ResetStickyTaskList(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ResetWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionResetWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = rp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.ResetWorkflowExecution(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.ResetWorkflowExecution(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskCanceled\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskCanceledScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskCanceled(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskCanceled(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskCanceledByID\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskCanceledByIDScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskCanceledByID(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskCanceledByID(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskCompleted\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskCompletedScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskCompleted(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskCompleted(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskCompletedByID\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskCompletedByIDScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskCompletedByID(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskCompletedByID(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskFailed\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskFailedScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskFailed(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskFailed(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondActivityTaskFailedByID\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondActivityTaskFailedByIDScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondActivityTaskFailedByID(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondActivityTaskFailedByID(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tvar (\n\t\tapiName                   = \"RespondDecisionTaskCompleted\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondDecisionTaskCompletedScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.RespondDecisionTaskCompleted(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.RespondDecisionTaskCompleted(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondDecisionTaskFailed\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondDecisionTaskFailedScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondDecisionTaskFailed(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondDecisionTaskFailed(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"RespondQueryTaskCompleted\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRespondQueryTaskCompletedScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tvar idGetter domainIDGetter\n\n\tidGetter, err = handler.tokenSerializer.DeserializeQueryTaskToken(rp1.TaskToken)\n\tif err == nil {\n\t\tdomainEntry, err = handler.domainCache.GetDomainByID(idGetter.GetDomainID())\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.RespondQueryTaskCompleted(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.RespondQueryTaskCompleted(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tvar (\n\t\tapiName                   = \"RestartWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionRestartWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(rp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = rp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\trp2, err = handler.frontendHandler.RestartWorkflowExecution(ctx, rp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\trp2, err = remoteClient.RestartWorkflowExecution(ctx, rp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn rp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tvar (\n\t\tapiName                   = \"ScanWorkflowExecutions\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionScanWorkflowExecutionsScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(lp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tlp2, err = handler.frontendHandler.ScanWorkflowExecutions(ctx, lp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tlp2, err = remoteClient.ScanWorkflowExecutions(ctx, lp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn lp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tvar (\n\t\tapiName                   = \"SignalWithStartWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionSignalWithStartWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(sp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tif !handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow(domainEntry.GetInfo().Name) {\n\t\tsp1.ActiveClusterSelectionPolicy = nil\n\t}\n\tactClSelPolicyForNewWF = sp1.ActiveClusterSelectionPolicy\n\tworkflowExecution = &types.WorkflowExecution{\n\t\tWorkflowID: sp1.GetWorkflowID(),\n\t}\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tsp2, err = handler.frontendHandler.SignalWithStartWorkflowExecution(ctx, sp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tsp2, err = remoteClient.SignalWithStartWorkflowExecution(ctx, sp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn sp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tvar (\n\t\tapiName                   = \"SignalWithStartWorkflowExecutionAsync\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionSignalWithStartWorkflowExecutionAsyncScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(sp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tif !handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow(domainEntry.GetInfo().Name) {\n\t\tsp1.ActiveClusterSelectionPolicy = nil\n\t}\n\tactClSelPolicyForNewWF = sp1.ActiveClusterSelectionPolicy\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tsp2, err = handler.frontendHandler.SignalWithStartWorkflowExecutionAsync(ctx, sp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tsp2, err = remoteClient.SignalWithStartWorkflowExecutionAsync(ctx, sp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn sp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"SignalWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionSignalWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(sp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = sp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.SignalWorkflowExecution(ctx, sp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.SignalWorkflowExecution(ctx, sp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tvar (\n\t\tapiName                   = \"StartWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionStartWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(sp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tif !handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow(domainEntry.GetInfo().Name) {\n\t\tsp1.ActiveClusterSelectionPolicy = nil\n\t}\n\tactClSelPolicyForNewWF = sp1.ActiveClusterSelectionPolicy\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tsp2, err = handler.frontendHandler.StartWorkflowExecution(ctx, sp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tsp2, err = remoteClient.StartWorkflowExecution(ctx, sp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn sp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tvar (\n\t\tapiName                   = \"StartWorkflowExecutionAsync\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionStartWorkflowExecutionAsyncScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(sp1.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tif !handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow(domainEntry.GetInfo().Name) {\n\t\tsp1.ActiveClusterSelectionPolicy = nil\n\t}\n\tactClSelPolicyForNewWF = sp1.ActiveClusterSelectionPolicy\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\tsp2, err = handler.frontendHandler.StartWorkflowExecutionAsync(ctx, sp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\tsp2, err = remoteClient.StartWorkflowExecutionAsync(ctx, sp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn sp2, err\n}\n\nfunc (handler *clusterRedirectionHandler) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest) (err error) {\n\tvar (\n\t\tapiName                   = \"TerminateWorkflowExecution\"\n\t\tcluster                   string\n\t\trequestedConsistencyLevel types.QueryConsistencyLevel = getRequestedConsistencyLevelFromContext(ctx)\n\t)\n\n\tvar domainEntry *cache.DomainCacheEntry\n\tscope, startTime := handler.beforeCall(metrics.DCRedirectionTerminateWorkflowExecutionScope)\n\tdefer func() {\n\t\thandler.afterCall(recover(), scope, startTime, domainEntry, cluster, &err)\n\t}()\n\n\tdomainEntry, err = handler.domainCache.GetDomain(tp1.Domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\tvar workflowExecution *types.WorkflowExecution\n\tworkflowExecution = tp1.GetWorkflowExecution()\n\n\terr = handler.redirectionPolicy.Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, func(targetDC string) error {\n\t\tcluster = targetDC\n\t\tswitch {\n\t\tcase targetDC == handler.currentClusterName:\n\t\t\terr = handler.frontendHandler.TerminateWorkflowExecution(ctx, tp1)\n\t\tdefault:\n\t\t\tremoteClient, clientErr := handler.GetRemoteFrontendClient(targetDC)\n\t\t\tif clientErr != nil {\n\t\t\t\treturn clientErr\n\t\t\t}\n\t\t\terr = remoteClient.TerminateWorkflowExecution(ctx, tp1, handler.callOptions...)\n\t\t}\n\t\treturn err\n\t})\n\n\treturn err\n}\n\nfunc (handler *clusterRedirectionHandler) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest) (up2 *types.UpdateDomainResponse, err error) {\n\treturn handler.frontendHandler.UpdateDomain(ctx, up1)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/api_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage clusterredirection\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\ntype (\n\tclusterRedirectionHandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller               *gomock.Controller\n\t\tmockResource             *resource.Test\n\t\tmockFrontendHandler      *api.MockHandler\n\t\tmockRemoteFrontendClient *frontend.MockClient\n\n\t\tmockClusterRedirectionPolicy *MockClusterRedirectionPolicy\n\n\t\tdomainName             string\n\t\tdomainID               string\n\t\tdomainCacheEntry       *cache.DomainCacheEntry\n\t\tcurrentClusterName     string\n\t\talternativeClusterName string\n\t\tconfig                 *frontendcfg.Config\n\n\t\thandler *clusterRedirectionHandler\n\t}\n)\n\nfunc TestForwardingPolicyV2ContainsV1(t *testing.T) {\n\trequire.NotEqual(t, selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2, selectedAPIsForwardingRedirectionPolicyAPIAllowlist)\n\tfor k := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\t_, ok := selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2[k]\n\t\trequire.True(t, ok, \"v2 does not contain a key that is in v1: %v\", k)\n\t}\n}\n\nfunc TestClusterRedirectionHandlerSuite(t *testing.T) {\n\ts := new(clusterRedirectionHandlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.domainName = \"some random domain name\"\n\ts.domainID = \"some random domain ID\"\n\ts.currentClusterName = cluster.TestCurrentClusterName\n\ts.alternativeClusterName = cluster.TestAlternativeClusterName\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockClusterRedirectionPolicy = NewMockClusterRedirectionPolicy(s.controller)\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Frontend)\n\ts.domainCacheEntry = cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: s.domainID, Name: s.domainName}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\ts.mockResource.DomainCache.EXPECT().GetDomain(s.domainName).Return(s.domainCacheEntry, nil).AnyTimes()\n\ts.mockResource.DomainCache.EXPECT().GetDomainByID(s.domainID).Return(s.domainCacheEntry, nil).AnyTimes()\n\n\ts.mockRemoteFrontendClient = s.mockResource.RemoteFrontendClient\n\n\ts.config = frontendcfg.NewConfig(\n\t\tdynamicconfig.NewCollection(\n\t\t\tdynamicconfig.NewNopClient(),\n\t\t\ts.mockResource.GetLogger(),\n\t\t),\n\t\t0,\n\t\tfalse,\n\t\t\"hostname\",\n\t\ts.mockResource.GetLogger(),\n\t)\n\tdh := domain.NewMockHandler(s.controller)\n\tfrontendHandler := api.NewWorkflowHandler(s.mockResource, s.config, client.NewVersionChecker(), dh)\n\n\ts.mockFrontendHandler = api.NewMockHandler(s.controller)\n\ts.handler = NewAPIHandler(frontendHandler, s.mockResource, s.config, config.ClusterRedirectionPolicy{}).(*clusterRedirectionHandler)\n\ts.handler.frontendHandler = s.mockFrontendHandler\n\ts.handler.redirectionPolicy = s.mockClusterRedirectionPolicy\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestDescribeTaskList() {\n\tapiName := \"DescribeTaskList\"\n\n\tctx := context.Background()\n\treq := &types.DescribeTaskListRequest{\n\t\tDomain: s.domainName,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().DescribeTaskList(ctx, req).Return(&types.DescribeTaskListResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().DescribeTaskList(ctx, req, s.handler.callOptions).Return(&types.DescribeTaskListResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.DescribeTaskList(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.DescribeTaskListResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestDescribeTaskListError() {\n\tapiName := \"DescribeTaskList\"\n\n\tctx := context.Background()\n\treq := &types.DescribeTaskListRequest{\n\t\tDomain: s.domainName,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().DescribeTaskList(ctx, req).Return(nil, errors.New(\"some error\"))\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.NotNil(err)\n\t\t\treturn err\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.DescribeTaskList(ctx, req)\n\ts.ErrorContains(err, \"some error\")\n\ts.Nil(resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestDescribeWorkflowExecution() {\n\tapiName := \"DescribeWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().DescribeWorkflowExecution(ctx, req).Return(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().DescribeWorkflowExecution(ctx, req, s.handler.callOptions).Return(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.DescribeWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.DescribeWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestDescribeWorkflowExecutionStrongConsistency() {\n\tapiName := \"DescribeWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain:                s.domainName,\n\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelStrong, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().DescribeWorkflowExecution(ctx, req).Return(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().DescribeWorkflowExecution(ctx, req, s.handler.callOptions).Return(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.DescribeWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.DescribeWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestGetWorkflowExecutionHistory() {\n\tapiName := \"GetWorkflowExecutionHistory\"\n\n\tctx := context.Background()\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().GetWorkflowExecutionHistory(ctx, req).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().GetWorkflowExecutionHistory(ctx, req, s.handler.callOptions).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.GetWorkflowExecutionHistory(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.GetWorkflowExecutionHistoryResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestGetWorkflowExecutionHistoryStrongConsistency() {\n\tapiName := \"GetWorkflowExecutionHistory\"\n\n\tctx := context.Background()\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain:                s.domainName,\n\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\tRunID:      \"some random run ID\",\n\t\t},\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, req.Execution, nil, apiName, types.QueryConsistencyLevelStrong, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().GetWorkflowExecutionHistory(ctx, req).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().GetWorkflowExecutionHistory(ctx, req, s.handler.callOptions).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.GetWorkflowExecutionHistory(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.GetWorkflowExecutionHistoryResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestListArchivedWorkflowExecutions() {\n\tapiName := \"ListArchivedWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ListArchivedWorkflowExecutions(ctx, req).Return(&types.ListArchivedWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ListArchivedWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.ListArchivedWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ListArchivedWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListArchivedWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestListClosedWorkflowExecutions() {\n\tapiName := \"ListClosedWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ListClosedWorkflowExecutions(ctx, req).Return(&types.ListClosedWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ListClosedWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.ListClosedWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ListClosedWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListClosedWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestListOpenWorkflowExecutions() {\n\tapiName := \"ListOpenWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ListOpenWorkflowExecutions(ctx, req).Return(&types.ListOpenWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ListOpenWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.ListOpenWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ListOpenWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListOpenWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestListWorkflowExecutions() {\n\tapiName := \"ListWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.ListWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ListWorkflowExecutions(ctx, req).Return(&types.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ListWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ListWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestScanWorkflowExecutions() {\n\tapiName := \"ScanWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.ListWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ScanWorkflowExecutions(ctx, req).Return(&types.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ScanWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ScanWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestCountWorkflowExecutions() {\n\tapiName := \"CountWorkflowExecutions\"\n\n\tctx := context.Background()\n\treq := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().CountWorkflowExecutions(ctx, req).Return(&types.CountWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().CountWorkflowExecutions(ctx, req, s.handler.callOptions).Return(&types.CountWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.CountWorkflowExecutions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.CountWorkflowExecutionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestPollForActivityTask() {\n\tapiName := \"PollForActivityTask\"\n\n\tctx := context.Background()\n\treq := &types.PollForActivityTaskRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().PollForActivityTask(ctx, req).Return(&types.PollForActivityTaskResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().PollForActivityTask(ctx, req, s.handler.callOptions).Return(&types.PollForActivityTaskResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.PollForActivityTask(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.PollForActivityTaskResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestPollForDecisionTask() {\n\tapiName := \"PollForDecisionTask\"\n\n\tctx := context.Background()\n\treq := &types.PollForDecisionTaskRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().PollForDecisionTask(ctx, req).Return(&types.PollForDecisionTaskResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().PollForDecisionTask(ctx, req, s.handler.callOptions).Return(&types.PollForDecisionTaskResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.PollForDecisionTask(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.PollForDecisionTaskResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestQueryWorkflow() {\n\tapiName := \"QueryWorkflow\"\n\n\tctx := context.Background()\n\treq := &types.QueryWorkflowRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().QueryWorkflow(ctx, req).Return(&types.QueryWorkflowResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().QueryWorkflow(ctx, req, s.handler.callOptions).Return(&types.QueryWorkflowResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.QueryWorkflow(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.QueryWorkflowResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestQueryWorkflowWithStrongConsistency() {\n\tapiName := \"QueryWorkflow\"\n\n\tctx := context.Background()\n\treq := &types.QueryWorkflowRequest{\n\t\tDomain:                s.domainName,\n\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelStrong, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().QueryWorkflow(ctx, req).Return(&types.QueryWorkflowResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().QueryWorkflow(ctx, req, s.handler.callOptions).Return(&types.QueryWorkflowResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.QueryWorkflow(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.QueryWorkflowResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRecordActivityTaskHeartbeat() {\n\tapiName := \"RecordActivityTaskHeartbeat\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: token,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RecordActivityTaskHeartbeat(ctx, req).Return(&types.RecordActivityTaskHeartbeatResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RecordActivityTaskHeartbeat(ctx, req, s.handler.callOptions).Return(&types.RecordActivityTaskHeartbeatResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.RecordActivityTaskHeartbeat(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.RecordActivityTaskHeartbeatResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRecordActivityTaskHeartbeatByID() {\n\tapiName := \"RecordActivityTaskHeartbeatByID\"\n\n\tctx := context.Background()\n\treq := &types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RecordActivityTaskHeartbeatByID(ctx, req).Return(&types.RecordActivityTaskHeartbeatResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RecordActivityTaskHeartbeatByID(ctx, req, s.handler.callOptions).Return(&types.RecordActivityTaskHeartbeatResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.RecordActivityTaskHeartbeatByID(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.RecordActivityTaskHeartbeatResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRequestCancelWorkflowExecution() {\n\tapiName := \"RequestCancelWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RequestCancelWorkflowExecution(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RequestCancelWorkflowExecution(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.RequestCancelWorkflowExecution(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestResetStickyTaskList() {\n\tapiName := \"ResetStickyTaskList\"\n\n\tctx := context.Background()\n\treq := &types.ResetStickyTaskListRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ResetStickyTaskList(ctx, req).Return(&types.ResetStickyTaskListResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ResetStickyTaskList(ctx, req, s.handler.callOptions).Return(&types.ResetStickyTaskListResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ResetStickyTaskList(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ResetStickyTaskListResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestResetWorkflowExecution() {\n\tapiName := \"ResetWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.ResetWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ResetWorkflowExecution(ctx, req).Return(&types.ResetWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ResetWorkflowExecution(ctx, req, s.handler.callOptions).Return(&types.ResetWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ResetWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ResetWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskCanceled() {\n\tapiName := \"RespondActivityTaskCanceled\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RespondActivityTaskCanceledRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskCanceled(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskCanceled(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr = s.handler.RespondActivityTaskCanceled(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskCanceledByID() {\n\tapiName := \"RespondActivityTaskCanceledByID\"\n\n\tctx := context.Background()\n\treq := &types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskCanceledByID(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskCanceledByID(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.RespondActivityTaskCanceledByID(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskCompleted() {\n\tapiName := \"RespondActivityTaskCompleted\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskCompleted(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskCompleted(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr = s.handler.RespondActivityTaskCompleted(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskCompletedByID() {\n\tapiName := \"RespondActivityTaskCompletedByID\"\n\n\tctx := context.Background()\n\treq := &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskCompletedByID(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskCompletedByID(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.RespondActivityTaskCompletedByID(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskFailed() {\n\tapiName := \"RespondActivityTaskFailed\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RespondActivityTaskFailedRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskFailed(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskFailed(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr = s.handler.RespondActivityTaskFailed(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondActivityTaskFailedByID() {\n\tapiName := \"RespondActivityTaskFailedByID\"\n\n\tctx := context.Background()\n\treq := &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain: s.domainName,\n\t}\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondActivityTaskFailedByID(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondActivityTaskFailedByID(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.RespondActivityTaskFailedByID(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondDecisionTaskCompleted() {\n\tapiName := \"RespondDecisionTaskCompleted\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RespondDecisionTaskCompletedRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondDecisionTaskCompleted(ctx, req).Return(&types.RespondDecisionTaskCompletedResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondDecisionTaskCompleted(ctx, req, s.handler.callOptions).Return(&types.RespondDecisionTaskCompletedResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.RespondDecisionTaskCompleted(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.RespondDecisionTaskCompletedResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondDecisionTaskFailed() {\n\tapiName := \"RespondDecisionTaskFailed\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.Serialize(&common.TaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.Nil(err)\n\treq := &types.RespondDecisionTaskFailedRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondDecisionTaskFailed(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondDecisionTaskFailed(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr = s.handler.RespondDecisionTaskFailed(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestRespondQueryTaskCompleted() {\n\tapiName := \"RespondQueryTaskCompleted\"\n\n\tctx := context.Background()\n\ttoken, err := s.handler.tokenSerializer.SerializeQueryTaskToken(&common.QueryTaskToken{\n\t\tDomainID: s.domainID,\n\t})\n\ts.NoError(err)\n\treq := &types.RespondQueryTaskCompletedRequest{\n\t\tTaskToken: token,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().RespondQueryTaskCompleted(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().RespondQueryTaskCompleted(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr = s.handler.RespondQueryTaskCompleted(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestSignalWithStartWorkflowExecution() {\n\tapiName := \"SignalWithStartWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"region-a\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, &types.WorkflowExecution{}, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\treq2 := &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.domainName,\n\t\t\t}\n\t\t\ts.mockFrontendHandler.EXPECT().SignalWithStartWorkflowExecution(ctx, req2).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().SignalWithStartWorkflowExecution(ctx, req2, s.handler.callOptions).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.SignalWithStartWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.StartWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestSignalWithStartWorkflowExecutionWithActiveClusterSelectionPolicy() {\n\ts.handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\tapiName := \"SignalWithStartWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"region-a\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, &types.WorkflowExecution{}, req.ActiveClusterSelectionPolicy, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().SignalWithStartWorkflowExecution(ctx, req).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().SignalWithStartWorkflowExecution(ctx, req, s.handler.callOptions).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.SignalWithStartWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.StartWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestSignalWorkflowExecution() {\n\tapiName := \"SignalWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.SignalWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, req.WorkflowExecution, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().SignalWorkflowExecution(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().SignalWorkflowExecution(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.SignalWorkflowExecution(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestStartWorkflowExecution() {\n\tapiName := \"StartWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.StartWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"region-b\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\treq2 := &types.StartWorkflowExecutionRequest{\n\t\t\t\tDomain: s.domainName,\n\t\t\t}\n\t\t\ts.mockFrontendHandler.EXPECT().StartWorkflowExecution(ctx, req2).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().StartWorkflowExecution(ctx, req2, s.handler.callOptions).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.StartWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.StartWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestStartWorkflowExecutionWithActiveClusterSelectionPolicy() {\n\ts.handler.config.EnableActiveClusterSelectionPolicyInStartWorkflow = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\tapiName := \"StartWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.StartWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: \"region\",\n\t\t\t\tName:  \"region-b\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, req.ActiveClusterSelectionPolicy, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().StartWorkflowExecution(ctx, req).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().StartWorkflowExecution(ctx, req, s.handler.callOptions).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.StartWorkflowExecution(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.StartWorkflowExecutionResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestTerminateWorkflowExecution() {\n\tapiName := \"TerminateWorkflowExecution\"\n\n\tctx := context.Background()\n\treq := &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: s.domainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, req.WorkflowExecution, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().TerminateWorkflowExecution(ctx, req).Return(nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().TerminateWorkflowExecution(ctx, req, s.handler.callOptions).Return(nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\terr := s.handler.TerminateWorkflowExecution(ctx, req)\n\ts.Nil(err)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestListTaskListPartitions() {\n\tapiName := \"ListTaskListPartitions\"\n\n\treq := &types.ListTaskListPartitionsRequest{\n\t\tDomain: s.domainName,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"test_tesk_list\",\n\t\t\tKind: types.TaskListKind(0).Ptr(),\n\t\t},\n\t}\n\n\tctx := context.Background()\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().ListTaskListPartitions(ctx, req).Return(&types.ListTaskListPartitionsResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().ListTaskListPartitions(ctx, req, s.handler.callOptions).Return(&types.ListTaskListPartitionsResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.ListTaskListPartitions(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.ListTaskListPartitionsResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestGetTaskListsByDomain() {\n\tapiName := \"GetTaskListsByDomain\"\n\n\tctx := context.Background()\n\treq := &types.GetTaskListsByDomainRequest{\n\t\tDomain: s.domainName,\n\t}\n\n\ts.mockClusterRedirectionPolicy.EXPECT().Redirect(ctx, s.domainCacheEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, domainCacheEntry *cache.DomainCacheEntry, wfExec *types.WorkflowExecution, selPlcy *types.ActiveClusterSelectionPolicy, apiName string, consistencyLevel types.QueryConsistencyLevel, callFn func(targetDC string) error) error {\n\t\t\t// validate callFn logic\n\t\t\ts.mockFrontendHandler.EXPECT().GetTaskListsByDomain(ctx, req).Return(&types.GetTaskListsByDomainResponse{}, nil).Times(1)\n\t\t\terr := callFn(s.currentClusterName)\n\t\t\ts.Nil(err)\n\t\t\ts.mockRemoteFrontendClient.EXPECT().GetTaskListsByDomain(ctx, req, s.handler.callOptions).Return(&types.GetTaskListsByDomainResponse{}, nil).Times(1)\n\t\t\terr = callFn(s.alternativeClusterName)\n\t\t\ts.Nil(err)\n\t\t\treturn nil\n\t\t}).\n\t\tTimes(1)\n\n\tresp, err := s.handler.GetTaskListsByDomain(ctx, req)\n\ts.Nil(err)\n\ts.Equal(&types.GetTaskListsByDomainResponse{}, resp)\n}\n\nfunc (s *clusterRedirectionHandlerSuite) TestGetTaskListsByDomainError() {\n\ts.handler.redirectionPolicy = newSelectedOrAllAPIsForwardingPolicy(\n\t\ts.currentClusterName,\n\t\ts.config,\n\t\ttrue,\n\t\tselectedAPIsForwardingRedirectionPolicyAPIAllowlistV2,\n\t\t\"\",\n\t\ts.mockResource.GetLogger(),\n\t\ts.mockResource.GetActiveClusterManager(),\n\t\ts.mockResource.GetMetricsClient(),\n\t)\n\tctx := context.Background()\n\treq := &types.GetTaskListsByDomainRequest{\n\t\tDomain: \"custom domain name\",\n\t}\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: \"custom domain name\"},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t},\n\t\t1,\n\t)\n\ts.mockResource.DomainCache.EXPECT().GetDomain(\"custom domain name\").Return(domainEntry, nil).Times(1)\n\ts.mockRemoteFrontendClient.EXPECT().GetTaskListsByDomain(ctx, req, s.handler.callOptions).Return(nil, &types.InternalServiceError{Message: \"test error\"}).Times(1)\n\n\tresp, err := s.handler.GetTaskListsByDomain(ctx, req)\n\ts.Error(err)\n\t// the resp is initialized to nil, since inner function is not called\n\ts.Nil(resp)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/callwrappers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clusterredirection\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tdomainIDGetter interface {\n\t\tGetDomainID() string\n\t}\n)\n\nfunc (handler *clusterRedirectionHandler) beforeCall(\n\tscope metrics.ScopeIdx,\n) (metrics.Scope, time.Time) {\n\treturn handler.GetMetricsClient().Scope(scope), handler.GetTimeSource().Now()\n}\n\nfunc (handler *clusterRedirectionHandler) afterCall(\n\trecovered interface{},\n\tscope metrics.Scope,\n\tstartTime time.Time,\n\tdomainEntry *cache.DomainCacheEntry,\n\tcluster string,\n\tretError *error,\n) {\n\tvar extraTags []tag.Tag\n\tif domainEntry != nil {\n\t\textraTags = append(extraTags, tag.WorkflowDomainName(domainEntry.GetInfo().Name))\n\t\textraTags = append(extraTags, tag.WorkflowDomainID(domainEntry.GetInfo().ID))\n\t}\n\tlog.CapturePanic(recovered, handler.GetLogger().WithTags(extraTags...), retError)\n\n\tscope = scope.Tagged(metrics.TargetClusterTag(cluster))\n\tscope.IncCounter(metrics.CadenceDcRedirectionClientRequests)\n\tscope.RecordTimer(metrics.CadenceDcRedirectionClientLatency, handler.GetTimeSource().Now().Sub(startTime))\n\tif *retError != nil {\n\t\tscope.IncCounter(metrics.CadenceDcRedirectionClientFailures)\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/policy.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage clusterredirection\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination policy_mock.go -self_package github.com/uber/cadence/service/frontend/wrappers/clusterredirection github.com/uber/cadence/service/frontend/wrappers/clusterredirection ClusterRedirectionPolicy\n\nconst (\n\t// DCRedirectionPolicyDefault means no redirection\n\tDCRedirectionPolicyDefault = \"\"\n\t// DCRedirectionPolicyNoop means no redirection\n\tDCRedirectionPolicyNoop = \"noop\"\n\t// DCRedirectionPolicySelectedAPIsForwarding means forwarding the following non-worker APIs based domain\n\t// 1. StartWorkflowExecution\n\t// 2. SignalWithStartWorkflowExecution\n\t// 3. SignalWorkflowExecution\n\t// 4. RequestCancelWorkflowExecution\n\t// 5. TerminateWorkflowExecution\n\t// 6. ResetWorkflow\n\t// please also reference selectedAPIsForwardingRedirectionPolicyAPIAllowlist and DCRedirectionPolicySelectedAPIsForwardingV2\n\tDCRedirectionPolicySelectedAPIsForwarding = \"selected-apis-forwarding\"\n\t// DCRedirectionPolicySelectedAPIsForwardingV2 forwards everything in DCRedirectionPolicySelectedAPIsForwarding,\n\t// as well as activity completions (sync and async).\n\t// This is done because activity results are generally deemed \"useful\" and relatively costly to re-do (when it is\n\t// even possible to redo), but activity workers themselves may be datacenter-specific.\n\t//\n\t// This will likely replace DCRedirectionPolicySelectedAPIsForwarding soon.\n\t//\n\t// 1-6. from DCRedirectionPolicySelectedAPIsForwarding\n\t// 7. RespondActivityTaskCanceled\n\t// 8. RespondActivityTaskCanceledByID\n\t// 9. RespondActivityTaskCompleted\n\t// 10. RespondActivityTaskCompletedByID\n\t// 11. RespondActivityTaskFailed\n\t// 12. RespondActivityTaskFailedByID\n\t// please also reference selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2\n\tDCRedirectionPolicySelectedAPIsForwardingV2 = \"selected-apis-forwarding-v2\"\n\t// DCRedirectionPolicyAllDomainAPIsForwarding means forwarding all the worker and non-worker APIs based domain,\n\t// and falling back to DCRedirectionPolicySelectedAPIsForwarding when the current active cluster is not the\n\t// cluster migration target.\n\tDCRedirectionPolicyAllDomainAPIsForwarding = \"all-domain-apis-forwarding\"\n\t// DCRedirectionPolicyAllDomainAPIsForwardingV2 means forwarding all the worker and non-worker APIs based domain,\n\t// and falling back to DCRedirectionPolicySelectedAPIsForwardingV2 when the current active cluster is not the\n\t// cluster migration target.\n\tDCRedirectionPolicyAllDomainAPIsForwardingV2 = \"all-domain-apis-forwarding-v2\"\n)\n\ntype (\n\t// ClusterRedirectionPolicy is a DC redirection policy interface\n\tClusterRedirectionPolicy interface {\n\t\t// Redirect redirects applicable API calls to active cluster based on given parameters and configured forwarding policy.\n\t\t// domainEntry (required): domain cache entry\n\t\t// workflowExecution (optional): workflow execution (only used for existing workflow API calls on active-active domains)\n\t\t// actClSelPolicyForNewWF (optional): active cluster selection policy for new workflow (only used for new workflow API calls on active-active domains)\n\t\t// apiName (required): API name\n\t\t// requestedConsistencyLevel (required): requested consistency level\n\t\t// call (required): function to call the API on the target cluster\n\t\tRedirect(\n\t\t\tctx context.Context,\n\t\t\tdomainEntry *cache.DomainCacheEntry,\n\t\t\tworkflowExecution *types.WorkflowExecution,\n\t\t\tactClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy,\n\t\t\tapiName string,\n\t\t\trequestedConsistencyLevel types.QueryConsistencyLevel,\n\t\t\tcall func(string) error,\n\t\t) error\n\t}\n\n\t// noopRedirectionPolicy is DC redirection policy which does nothing\n\tnoopRedirectionPolicy struct {\n\t\tcurrentClusterName string\n\t}\n\n\t// selectedOrAllAPIsForwardingRedirectionPolicy is a DC redirection policy\n\t// which (based on domain) forwards selected APIs calls or all domain APIs to active cluster\n\tselectedOrAllAPIsForwardingRedirectionPolicy struct {\n\t\tcurrentClusterName   string\n\t\tconfig               *frontendcfg.Config\n\t\tallDomainAPIs        bool\n\t\tselectedAPIs         map[string]struct{}\n\t\ttargetCluster        string\n\t\tlogger               log.Logger\n\t\tactiveClusterManager activecluster.Manager\n\t\tmetricsClient        metrics.Client\n\t}\n)\n\n// selectedAPIsForwardingRedirectionPolicyAPIAllowlist contains a list of non-worker APIs which can be redirected.\n// This is paired with DCRedirectionPolicySelectedAPIsForwarding - keep both lists up to date.\nvar selectedAPIsForwardingRedirectionPolicyAPIAllowlist = map[string]struct{}{\n\t\"StartWorkflowExecution\":           {},\n\t\"SignalWithStartWorkflowExecution\": {},\n\t\"SignalWorkflowExecution\":          {},\n\t\"RequestCancelWorkflowExecution\":   {},\n\t\"TerminateWorkflowExecution\":       {},\n\t\"ResetWorkflowExecution\":           {},\n}\n\n// selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2 contains a list of non-worker APIs which can be redirected.\n// This is paired with DCRedirectionPolicySelectedAPIsForwardingV2 - keep both lists up to date.\nvar selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2 = map[string]struct{}{\n\t// from selectedAPIsForwardingRedirectionPolicyAPIAllowlist\n\t\"StartWorkflowExecution\":           {},\n\t\"SignalWithStartWorkflowExecution\": {},\n\t\"SignalWorkflowExecution\":          {},\n\t\"RequestCancelWorkflowExecution\":   {},\n\t\"TerminateWorkflowExecution\":       {},\n\t\"ResetWorkflowExecution\":           {},\n\t// additional endpoints\n\t\"RespondActivityTaskCanceled\":      {},\n\t\"RespondActivityTaskCanceledByID\":  {},\n\t\"RespondActivityTaskCompleted\":     {},\n\t\"RespondActivityTaskCompletedByID\": {},\n\t\"RespondActivityTaskFailed\":        {},\n\t\"RespondActivityTaskFailedByID\":    {},\n}\n\n// allowedAPIsForDeprecatedDomains contains a list of APIs that are allowed to be called on deprecated domains\nvar allowedAPIsForDeprecatedDomains = map[string]struct{}{\n\t\"ListWorkflowExecutions\":     {},\n\t\"CountWorkflowExecutions\":    {},\n\t\"ScanWorkflowExecutions\":     {},\n\t\"TerminateWorkflowExecution\": {},\n}\n\n// RedirectionPolicyGenerator generate corresponding redirection policy\nfunc RedirectionPolicyGenerator(\n\tclusterMetadata cluster.Metadata,\n\tconfig *frontendcfg.Config,\n\tpolicy config.ClusterRedirectionPolicy,\n\tlogger log.Logger,\n\tactiveClusterManager activecluster.Manager,\n\tmetricsClient metrics.Client,\n) ClusterRedirectionPolicy {\n\tswitch policy.Policy {\n\tcase DCRedirectionPolicyDefault:\n\t\t// default policy, noop\n\t\treturn newNoopRedirectionPolicy(clusterMetadata.GetCurrentClusterName())\n\tcase DCRedirectionPolicyNoop:\n\t\treturn newNoopRedirectionPolicy(clusterMetadata.GetCurrentClusterName())\n\tcase DCRedirectionPolicySelectedAPIsForwarding:\n\t\tcurrentClusterName := clusterMetadata.GetCurrentClusterName()\n\t\treturn newSelectedOrAllAPIsForwardingPolicy(currentClusterName, config, false, selectedAPIsForwardingRedirectionPolicyAPIAllowlist, \"\", logger, activeClusterManager, metricsClient)\n\tcase DCRedirectionPolicySelectedAPIsForwardingV2:\n\t\tcurrentClusterName := clusterMetadata.GetCurrentClusterName()\n\t\treturn newSelectedOrAllAPIsForwardingPolicy(currentClusterName, config, false, selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2, \"\", logger, activeClusterManager, metricsClient)\n\tcase DCRedirectionPolicyAllDomainAPIsForwarding:\n\t\tcurrentClusterName := clusterMetadata.GetCurrentClusterName()\n\t\treturn newSelectedOrAllAPIsForwardingPolicy(currentClusterName, config, true, selectedAPIsForwardingRedirectionPolicyAPIAllowlist, policy.AllDomainApisForwardingTargetCluster, logger, activeClusterManager, metricsClient)\n\tcase DCRedirectionPolicyAllDomainAPIsForwardingV2:\n\t\tcurrentClusterName := clusterMetadata.GetCurrentClusterName()\n\t\treturn newSelectedOrAllAPIsForwardingPolicy(currentClusterName, config, true, selectedAPIsForwardingRedirectionPolicyAPIAllowlistV2, policy.AllDomainApisForwardingTargetCluster, logger, activeClusterManager, metricsClient)\n\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"Unknown DC redirection policy %v\", policy.Policy))\n\t}\n}\n\n// newNoopRedirectionPolicy is DC redirection policy which does nothing\nfunc newNoopRedirectionPolicy(currentClusterName string) *noopRedirectionPolicy {\n\treturn &noopRedirectionPolicy{\n\t\tcurrentClusterName: currentClusterName,\n\t}\n}\n\n// Redirect redirect the API call based on domain ID\nfunc (policy *noopRedirectionPolicy) Redirect(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\tactClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy,\n\tapiName string,\n\trequestedConsistencyLevel types.QueryConsistencyLevel,\n\tcall func(string) error,\n) error {\n\treturn call(policy.currentClusterName)\n}\n\n// newSelectedOrAllAPIsForwardingPolicy creates a forwarding policy for selected APIs based on domain\nfunc newSelectedOrAllAPIsForwardingPolicy(\n\tcurrentClusterName string,\n\tconfig *frontendcfg.Config,\n\tallDomainAPIs bool,\n\tselectedAPIs map[string]struct{},\n\ttargetCluster string,\n\tlogger log.Logger,\n\tactiveClusterManager activecluster.Manager,\n\tmetricsClient metrics.Client,\n) *selectedOrAllAPIsForwardingRedirectionPolicy {\n\treturn &selectedOrAllAPIsForwardingRedirectionPolicy{\n\t\tcurrentClusterName:   currentClusterName,\n\t\tconfig:               config,\n\t\tallDomainAPIs:        allDomainAPIs,\n\t\tselectedAPIs:         selectedAPIs,\n\t\ttargetCluster:        targetCluster,\n\t\tlogger:               logger,\n\t\tactiveClusterManager: activeClusterManager,\n\t\tmetricsClient:        metricsClient,\n\t}\n}\n\nfunc (policy *selectedOrAllAPIsForwardingRedirectionPolicy) Redirect(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\tactClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy,\n\tapiName string,\n\trequestedConsistencyLevel types.QueryConsistencyLevel,\n\tcall func(string) error,\n) error {\n\tif domainEntry.IsDeprecatedOrDeleted() {\n\t\tif _, ok := allowedAPIsForDeprecatedDomains[apiName]; !ok {\n\t\t\treturn &types.DomainNotActiveError{\n\t\t\t\tMessage:        \"domain is deprecated or deleted.\",\n\t\t\t\tDomainName:     domainEntry.GetInfo().Name,\n\t\t\t\tCurrentCluster: policy.currentClusterName,\n\t\t\t\tActiveCluster:  policy.currentClusterName,\n\t\t\t}\n\t\t}\n\t}\n\n\treturn policy.withRedirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, call)\n}\n\nfunc (policy *selectedOrAllAPIsForwardingRedirectionPolicy) withRedirect(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\trequestedActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n\tapiName string,\n\trequestedConsistencyLevel types.QueryConsistencyLevel,\n\tcall func(string) error,\n) error {\n\ttargetDC, enableDomainNotActiveForwarding := policy.getTargetClusterAndIsDomainNotActiveAutoForwarding(ctx, domainEntry, workflowExecution, requestedActiveClusterSelectionPolicy, apiName, requestedConsistencyLevel)\n\tdomainName := domainEntry.GetInfo().Name\n\n\tpolicy.logger.Debug(\n\t\t\"Calling API on target cluster for domain\",\n\t\ttag.OperationName(apiName),\n\t\ttag.ClusterName(policy.currentClusterName),\n\t\ttag.ActiveClusterName(targetDC),\n\t\ttag.WorkflowDomainName(domainName),\n\t)\n\terr := call(targetDC)\n\tscope := policy.metricsClient.Scope(metrics.DCRedirectionForwardingPolicyScope).Tagged(\n\t\tappend(\n\t\t\tmetrics.GetContextTags(ctx),\n\t\t\tmetrics.DomainTag(domainName),\n\t\t\tmetrics.SourceClusterTag(policy.currentClusterName),\n\t\t\tmetrics.TargetClusterTag(targetDC),\n\t\t\tmetrics.IsActiveActiveDomainTag(requestedActiveClusterSelectionPolicy != nil),\n\t\t\tmetrics.QueryConsistencyLevelTag(requestedConsistencyLevel.String()),\n\t\t)...,\n\t)\n\tif err == nil {\n\t\tscope.IncCounter(metrics.ClusterForwardingPolicyRequests)\n\t\treturn nil\n\t}\n\n\tvar domainNotActiveErr *types.DomainNotActiveError\n\tok := errors.As(err, &domainNotActiveErr)\n\tif !ok || !enableDomainNotActiveForwarding {\n\t\tpolicy.logger.Debug(\"Redirection not enabled for request\", tag.WorkflowDomainName(domainName), tag.OperationName(apiName), tag.Error(err))\n\t\tscope.IncCounter(metrics.ClusterForwardingPolicyRequests)\n\t\treturn err\n\t}\n\n\tscope = scope.Tagged(metrics.ActiveClusterTag(domainNotActiveErr.ActiveCluster))\n\tscope.IncCounter(metrics.ClusterForwardingPolicyRequests)\n\n\tif domainNotActiveErr.ActiveCluster == \"\" {\n\t\tpolicy.logger.Debug(\n\t\t\t\"No active cluster specified in the error returned from cluster, skipping redirect\",\n\t\t\ttag.ClusterName(targetDC),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.OperationName(apiName),\n\t\t)\n\t\treturn err\n\t}\n\n\tif domainNotActiveErr.ActiveCluster == targetDC {\n\t\tpolicy.logger.Debug(\n\t\t\t\"No need to redirect to new target cluster\",\n\t\t\ttag.ClusterName(targetDC),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.OperationName(apiName),\n\t\t)\n\t\treturn err\n\t}\n\n\tpolicy.logger.Debug(\n\t\t\"Calling API on new target cluster for domain as indicated by response from cluster\",\n\t\ttag.OperationName(apiName),\n\t\ttag.ActiveClusterName(domainNotActiveErr.ActiveCluster),\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.ClusterName(targetDC),\n\t)\n\treturn call(domainNotActiveErr.ActiveCluster)\n}\n\n// return two values: the target cluster name, and whether or not forwarding to the active cluster\nfunc (policy *selectedOrAllAPIsForwardingRedirectionPolicy) getTargetClusterAndIsDomainNotActiveAutoForwarding(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\trequestedActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n\tapiName string,\n\trequestedConsistencyLevel types.QueryConsistencyLevel,\n) (string, bool) {\n\tif !domainEntry.IsGlobalDomain() {\n\t\t// Do not do dc redirection if domain is local domain,\n\t\t// for global domains with 1 dc, it's still useful to do auto-forwarding during cluster migration\n\t\tpolicy.logger.Debug(\n\t\t\t\"Local domain, routing to current cluster\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t)\n\t\treturn policy.currentClusterName, false\n\t}\n\n\tif !policy.config.EnableDomainNotActiveAutoForwarding(domainEntry.GetInfo().Name) {\n\t\t// Do not do dc redirection if auto-forwarding dynamicconfig is not enabled\n\t\tpolicy.logger.Debug(\n\t\t\t\"Auto-forwarding dynamicconfig is not enabled, routing to current cluster\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t)\n\t\treturn policy.currentClusterName, false\n\t}\n\n\tcurrentActiveCluster := domainEntry.GetReplicationConfig().ActiveClusterName\n\tif domainEntry.GetReplicationConfig().IsActiveActive() {\n\t\tworkflowActiveCluster := policy.activeClusterForActiveActiveDomainRequest(ctx, domainEntry, workflowExecution, requestedActiveClusterSelectionPolicy, apiName)\n\t\tpolicy.logger.Debug(\n\t\t\t\"Active-active domain, routing to active cluster\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.ClusterName(currentActiveCluster),\n\t\t\ttag.ActiveClusterName(workflowActiveCluster),\n\t\t)\n\t\tcurrentActiveCluster = workflowActiveCluster\n\t}\n\n\tif policy.allDomainAPIs {\n\t\tif policy.targetCluster == \"\" {\n\t\t\tpolicy.logger.Debug(\n\t\t\t\t\"All domain APIs forwarding, routing to active cluster\",\n\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t\t\ttag.ActiveClusterName(currentActiveCluster),\n\t\t\t)\n\t\t\treturn currentActiveCluster, true\n\t\t}\n\t\tif policy.targetCluster == currentActiveCluster {\n\t\t\tpolicy.logger.Debug(\n\t\t\t\t\"All domain APIs forwarding, routing to active cluster\",\n\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t\t\ttag.ActiveClusterName(currentActiveCluster),\n\t\t\t)\n\t\t\treturn currentActiveCluster, true\n\t\t}\n\t\t// fallback to selected APIs if targetCluster is not empty and not the same as currentActiveCluster\n\t}\n\n\tif requestedConsistencyLevel == types.QueryConsistencyLevelStrong {\n\t\tpolicy.logger.Debug(\n\t\t\t\"Query requested strong consistency, routing to active cluster\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t\ttag.ActiveClusterName(currentActiveCluster),\n\t\t)\n\t\treturn currentActiveCluster, true\n\t}\n\n\t_, ok := policy.selectedAPIs[apiName]\n\tif !ok {\n\t\t// do not do dc redirection if API is not whitelisted\n\t\tpolicy.logger.Debug(\n\t\t\t\"API is not whitelisted, routing to current cluster\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.ClusterName(policy.currentClusterName),\n\t\t\ttag.OperationName(apiName),\n\t\t)\n\t\treturn policy.currentClusterName, false\n\t}\n\n\tpolicy.logger.Debug(\n\t\t\"API is whitelisted, routing to active cluster\",\n\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\ttag.ClusterName(policy.currentClusterName),\n\t\ttag.ActiveClusterName(currentActiveCluster),\n\t)\n\treturn currentActiveCluster, true\n}\n\nfunc (policy *selectedOrAllAPIsForwardingRedirectionPolicy) activeClusterForActiveActiveDomainRequest(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\trequestedActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n\tapiName string,\n) string {\n\tpolicy.logger.Debug(\"Determining active cluster for active-active domain request\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.Dynamic(\"execution\", workflowExecution), tag.OperationName(apiName))\n\tif apiName == \"SignalWithStartWorkflowExecution\" {\n\t\texistingActiveClusterSelectionPolicy, running, err := policy.activeClusterManager.GetActiveClusterSelectionPolicyForCurrentWorkflow(ctx, domainEntry.GetInfo().ID, workflowExecution.WorkflowID)\n\t\tif err != nil {\n\t\t\tpolicy.logger.Error(\"Failed to get active cluster selection policy for current workflow, using current cluster\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.OperationName(apiName), tag.Error(err))\n\t\t\treturn policy.currentClusterName\n\t\t}\n\t\t// if current workflow is still running, use the policy for the current workflow\n\t\tif running {\n\t\t\trequestedActiveClusterSelectionPolicy = existingActiveClusterSelectionPolicy\n\t\t}\n\t\treturn policy.activeClusterByClusterAttribute(ctx, domainEntry, requestedActiveClusterSelectionPolicy, apiName)\n\t} else if apiName == \"StartWorkflowExecution\" {\n\t\treturn policy.activeClusterByClusterAttribute(ctx, domainEntry, requestedActiveClusterSelectionPolicy, apiName)\n\t}\n\n\tif workflowExecution == nil || workflowExecution.WorkflowID == \"\" {\n\t\tpolicy.logger.Debug(\"Workflow execution is nil or workflow id is empty, using current cluster\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.OperationName(apiName))\n\t\treturn policy.currentClusterName\n\t}\n\n\tactiveClusterInfo, err := policy.activeClusterManager.GetActiveClusterInfoByWorkflow(ctx, domainEntry.GetInfo().ID, workflowExecution.WorkflowID, workflowExecution.RunID)\n\tif err != nil {\n\t\tpolicy.logger.Error(\"Failed to get active cluster of workflow, using current cluster\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.WorkflowID(workflowExecution.WorkflowID), tag.WorkflowRunID(workflowExecution.RunID), tag.OperationName(apiName), tag.Error(err))\n\t\treturn policy.currentClusterName\n\t}\n\n\tpolicy.logger.Debug(\"Get active cluster info by workflow for active-active domain request\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.WorkflowID(workflowExecution.WorkflowID), tag.WorkflowRunID(workflowExecution.RunID), tag.OperationName(apiName), tag.ActiveClusterName(activeClusterInfo.ActiveClusterName))\n\treturn activeClusterInfo.ActiveClusterName\n}\n\nfunc (policy *selectedOrAllAPIsForwardingRedirectionPolicy) activeClusterByClusterAttribute(ctx context.Context, domainEntry *cache.DomainCacheEntry, requestedActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy, apiName string) string {\n\tpolicy.logger.Debug(\"Active cluster selection policy by cluster attribute\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.OperationName(apiName), tag.Dynamic(\"policy\", requestedActiveClusterSelectionPolicy))\n\tactiveClusterInfo, err := policy.activeClusterManager.GetActiveClusterInfoByClusterAttribute(ctx, domainEntry.GetInfo().ID, requestedActiveClusterSelectionPolicy.GetClusterAttribute())\n\tif err != nil {\n\t\tpolicy.logger.Error(\"Failed to lookup active cluster by cluster attribute, using current cluster\", tag.WorkflowDomainName(domainEntry.GetInfo().Name), tag.OperationName(apiName), tag.Error(err))\n\t\treturn policy.currentClusterName\n\t}\n\treturn activeClusterInfo.ActiveClusterName\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/policy_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/frontend/wrappers/clusterredirection (interfaces: ClusterRedirectionPolicy)\n//\n// Generated by this command:\n//\n//\tmockgen -package clusterredirection -destination policy_mock.go -self_package github.com/uber/cadence/service/frontend/wrappers/clusterredirection github.com/uber/cadence/service/frontend/wrappers/clusterredirection ClusterRedirectionPolicy\n//\n\n// Package clusterredirection is a generated GoMock package.\npackage clusterredirection\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tcache \"github.com/uber/cadence/common/cache\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClusterRedirectionPolicy is a mock of ClusterRedirectionPolicy interface.\ntype MockClusterRedirectionPolicy struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClusterRedirectionPolicyMockRecorder\n\tisgomock struct{}\n}\n\n// MockClusterRedirectionPolicyMockRecorder is the mock recorder for MockClusterRedirectionPolicy.\ntype MockClusterRedirectionPolicyMockRecorder struct {\n\tmock *MockClusterRedirectionPolicy\n}\n\n// NewMockClusterRedirectionPolicy creates a new mock instance.\nfunc NewMockClusterRedirectionPolicy(ctrl *gomock.Controller) *MockClusterRedirectionPolicy {\n\tmock := &MockClusterRedirectionPolicy{ctrl: ctrl}\n\tmock.recorder = &MockClusterRedirectionPolicyMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClusterRedirectionPolicy) EXPECT() *MockClusterRedirectionPolicyMockRecorder {\n\treturn m.recorder\n}\n\n// Redirect mocks base method.\nfunc (m *MockClusterRedirectionPolicy) Redirect(ctx context.Context, domainEntry *cache.DomainCacheEntry, workflowExecution *types.WorkflowExecution, actClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy, apiName string, requestedConsistencyLevel types.QueryConsistencyLevel, call func(string) error) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Redirect\", ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, call)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Redirect indicates an expected call of Redirect.\nfunc (mr *MockClusterRedirectionPolicyMockRecorder) Redirect(ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, call any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Redirect\", reflect.TypeOf((*MockClusterRedirectionPolicy)(nil).Redirect), ctx, domainEntry, workflowExecution, actClSelPolicyForNewWF, apiName, requestedConsistencyLevel, call)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/policy_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage clusterredirection\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\tfrontendcfg \"github.com/uber/cadence/service/frontend/config\"\n)\n\ntype (\n\tnoopDCRedirectionPolicySuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcurrentClusterName string\n\t\tpolicy             *noopRedirectionPolicy\n\t}\n\n\tselectedAPIsForwardingRedirectionPolicySuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tactiveClusterManager *activecluster.MockManager\n\n\t\tdomainName             string\n\t\tdomainID               string\n\t\tcurrentClusterName     string\n\t\talternativeClusterName string\n\t\tmockConfig             *frontendcfg.Config\n\t\tmockMetricsClient      metrics.Client\n\n\t\tpolicy *selectedOrAllAPIsForwardingRedirectionPolicy\n\t}\n)\n\nfunc TestNoopDCRedirectionPolicySuite(t *testing.T) {\n\ts := new(noopDCRedirectionPolicySuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *noopDCRedirectionPolicySuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.currentClusterName = cluster.TestCurrentClusterName\n\ts.policy = newNoopRedirectionPolicy(s.currentClusterName)\n}\n\nfunc (s *noopDCRedirectionPolicySuite) TearDownTest() {\n\n}\n\nfunc (s *noopDCRedirectionPolicySuite) TestWithDomainRedirect() {\n\tdomainName := \"some random domain name\"\n\tdomainID := \"some random domain ID\"\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t&persistence.DomainConfig{},\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\tapiName := \"any random API name\"\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.Nil(err)\n\n\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.Nil(err)\n\n\ts.Equal(2, callCount)\n}\n\nfunc (s *noopDCRedirectionPolicySuite) TestWithDomainRedirectForAllowedAPIs() {\n\tdomainName := \"some random domain name\"\n\tdomainID := \"some random domain ID\"\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t&persistence.DomainConfig{},\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\t// Test all allowed APIs for deprecated domains\n\tallowedAPIs := []string{\n\t\t\"ListWorkflowExecutions\",\n\t\t\"CountWorkflowExecutions\",\n\t\t\"ScanWorkflowExecutions\",\n\t\t\"TerminateWorkflowExecution\",\n\t}\n\n\tfor _, apiName := range allowedAPIs {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\t// Verify that each API was tested for both domain ID and domain name redirects\n\ts.Equal(2*len(allowedAPIs), callCount)\n}\n\nfunc TestSelectedAPIsForwardingRedirectionPolicySuite(t *testing.T) {\n\ts := new(selectedAPIsForwardingRedirectionPolicySuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) SetupSuite() {\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TearDownSuite() {\n\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.domainName = \"some random domain name\"\n\ts.domainID = \"some random domain ID\"\n\ts.currentClusterName = cluster.TestCurrentClusterName\n\ts.alternativeClusterName = cluster.TestAlternativeClusterName\n\tlogger := testlogger.New(s.T())\n\n\ts.mockConfig = frontendcfg.NewConfig(dynamicconfig.NewCollection(\n\t\tdynamicconfig.NewNopClient(),\n\t\tlogger,\n\t),\n\t\t0,\n\t\tfalse,\n\t\t\"hostname\",\n\t\tlogger,\n\t)\n\n\ts.mockMetricsClient = metrics.NewNoopMetricsClient()\n\n\ts.activeClusterManager = activecluster.NewMockManager(s.controller)\n\n\ts.policy = newSelectedOrAllAPIsForwardingPolicy(\n\t\ts.currentClusterName,\n\t\ts.mockConfig,\n\t\tfalse,\n\t\tselectedAPIsForwardingRedirectionPolicyAPIAllowlist,\n\t\t\"\",\n\t\tlogger,\n\t\ts.activeClusterManager,\n\t\ts.mockMetricsClient,\n\t)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestWithDomainRedirect_LocalDomain() {\n\tdomainEntry := s.setupLocalDomain()\n\n\tapiName := \"any random API name\"\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.Nil(err)\n\n\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.Nil(err)\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\ts.Equal(2*(len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist)+1), callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestWithDomainRedirect_GlobalDomain_NoForwarding_DomainNotWhitelisted() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(false, true)\n\n\tdomainNotActiveErr := &types.DomainNotActiveError{\n\t\tCurrentCluster: s.currentClusterName,\n\t\tActiveCluster:  s.alternativeClusterName,\n\t}\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn domainNotActiveErr\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.NotNil(err)\n\t\ts.Equal(err.Error(), domainNotActiveErr.Error())\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.NotNil(err)\n\t\ts.Equal(err.Error(), domainNotActiveErr.Error())\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelStrong, callFn)\n\t\ts.NotNil(err)\n\t\ts.Equal(err.Error(), domainNotActiveErr.Error())\n\t}\n\n\ts.Equal(3*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestWithDomainRedirect_GlobalDomain_Forwarding_APINotWhitelisted() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, true)\n\n\tapiName := \"any random API name\"\n\tdomainNotActiveErr := &types.DomainNotActiveError{\n\t\tCurrentCluster: s.currentClusterName,\n\t\tActiveCluster:  s.alternativeClusterName,\n\t}\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn domainNotActiveErr\n\t}\n\n\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.NotNil(err)\n\ts.Equal(err.Error(), domainNotActiveErr.Error())\n\n\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\ts.NotNil(err)\n\ts.Equal(err.Error(), domainNotActiveErr.Error())\n\n\ts.Equal(2, callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_CurrentCluster() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, true)\n\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.currentClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_AlternativeCluster() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, false)\n\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.alternativeClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_AlternativeCluster_StrongConsistency() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, false)\n\n\tcallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tcallCount++\n\t\ts.Equal(s.alternativeClusterName, targetCluster)\n\t\treturn nil\n\t}\n\n\tapiName := \"any random API name\"\n\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelStrong, callFn)\n\ts.Nil(err)\n\n\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelStrong, callFn)\n\ts.Nil(err)\n\n\ts.Equal(2, callCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_CurrentClusterToAlternativeCluster() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, true)\n\n\tcurrentClustercallCount := 0\n\talternativeClustercallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tswitch targetCluster {\n\t\tcase s.currentClusterName:\n\t\t\tcurrentClustercallCount++\n\t\t\treturn &types.DomainNotActiveError{\n\t\t\t\tCurrentCluster: s.currentClusterName,\n\t\t\t\tActiveCluster:  s.alternativeClusterName,\n\t\t\t}\n\t\tcase s.alternativeClusterName:\n\t\t\talternativeClustercallCount++\n\t\t\treturn nil\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown cluster name %v\", targetCluster))\n\t\t}\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), currentClustercallCount)\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), alternativeClustercallCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_AlternativeClusterToCurrentCluster() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, false)\n\n\tcurrentClustercallCount := 0\n\talternativeClustercallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tswitch targetCluster {\n\t\tcase s.currentClusterName:\n\t\t\tcurrentClustercallCount++\n\t\t\treturn nil\n\t\tcase s.alternativeClusterName:\n\t\t\talternativeClustercallCount++\n\t\t\treturn &types.DomainNotActiveError{\n\t\t\t\tCurrentCluster: s.alternativeClusterName,\n\t\t\t\tActiveCluster:  s.currentClusterName,\n\t\t\t}\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown cluster name %v\", targetCluster))\n\t\t}\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Nil(err)\n\t}\n\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), currentClustercallCount)\n\ts.Equal(2*len(selectedAPIsForwardingRedirectionPolicyAPIAllowlist), alternativeClustercallCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_ToSameCluster_Skipped() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, false)\n\n\tcallFn := func(targetCluster string) error {\n\t\tswitch targetCluster {\n\t\tcase s.alternativeClusterName:\n\t\t\treturn &types.DomainNotActiveError{ // this shouldn't happen but if it does, we should skip the redirect\n\t\t\t\tCurrentCluster: s.alternativeClusterName,\n\t\t\t\tActiveCluster:  s.alternativeClusterName,\n\t\t\t}\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown cluster name %v\", targetCluster))\n\t\t}\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\tvar domainNotActiveErr *types.DomainNotActiveError\n\t\ts.ErrorAs(err, &domainNotActiveErr)\n\t\ts.Equal(s.alternativeClusterName, domainNotActiveErr.ActiveCluster)\n\t}\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_EmptyClusster_Skipped() {\n\tdomainEntry := s.setupGlobalDomainWithTwoReplicationCluster(true, false)\n\n\tcallFn := func(targetCluster string) error {\n\t\tswitch targetCluster {\n\t\tcase s.alternativeClusterName:\n\t\t\treturn &types.DomainNotActiveError{ // this shouldn't happen but if it does, we should skip the redirect\n\t\t\t\tCurrentCluster: s.alternativeClusterName,\n\t\t\t\tActiveCluster:  \"\",\n\t\t\t}\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown cluster name %v\", targetCluster))\n\t\t}\n\t}\n\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\tvar domainNotActiveErr *types.DomainNotActiveError\n\t\ts.ErrorAs(err, &domainNotActiveErr)\n\t\ts.Equal(\"\", domainNotActiveErr.ActiveCluster)\n\t}\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestGetTargetDataCenter_GlobalDomain_Forwarding_DeprecatedDomain() {\n\tdomainEntry := s.setupGlobalDeprecatedDomainWithTwoReplicationCluster(true, false)\n\n\tcurrentClustercallCount := 0\n\talternativeClustercallCount := 0\n\tcallFn := func(targetCluster string) error {\n\t\tswitch targetCluster {\n\t\tcase s.currentClusterName:\n\t\t\tcurrentClustercallCount++\n\t\t\treturn nil\n\t\tcase s.alternativeClusterName:\n\t\t\talternativeClustercallCount++\n\t\t\treturn &types.DomainNotActiveError{\n\t\t\t\tCurrentCluster: s.alternativeClusterName,\n\t\t\t\tActiveCluster:  s.currentClusterName,\n\t\t\t}\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown cluster name %v\", targetCluster))\n\t\t}\n\t}\n\n\t// Test non-allowed APIs\n\tfor apiName := range selectedAPIsForwardingRedirectionPolicyAPIAllowlist {\n\t\tif _, ok := allowedAPIsForDeprecatedDomains[apiName]; ok {\n\t\t\tcontinue // Skip allowed APIs\n\t\t}\n\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Error(err)\n\t\ts.Equal(\"domain is deprecated or deleted.\", err.Error())\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.Error(err)\n\t\ts.Equal(\"domain is deprecated or deleted.\", err.Error())\n\t}\n\ts.Equal(0, currentClustercallCount)\n\ts.Equal(0, alternativeClustercallCount)\n\n\t// Test allowed APIs\n\tfor apiName := range allowedAPIsForDeprecatedDomains {\n\t\terr := s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.NoError(err)\n\n\t\terr = s.policy.Redirect(context.Background(), domainEntry, nil, nil, apiName, types.QueryConsistencyLevelEventual, callFn)\n\t\ts.NoError(err)\n\t}\n\n\t// Verify that allowed APIs were called on the current cluster\n\ts.Equal(2*len(allowedAPIsForDeprecatedDomains), currentClustercallCount)\n\ts.Equal(2, alternativeClustercallCount)\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) setupLocalDomain() *cache.DomainCacheEntry {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.domainID, Name: s.domainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\treturn domainEntry\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) setupGlobalDomainWithTwoReplicationCluster(forwardingEnabled bool, isRecordActive bool) *cache.DomainCacheEntry {\n\tactiveCluster := s.alternativeClusterName\n\tif isRecordActive {\n\t\tactiveCluster = s.currentClusterName\n\t}\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.domainID, Name: s.domainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: activeCluster,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234, // not used\n\t)\n\n\ts.mockConfig.EnableDomainNotActiveAutoForwarding = dynamicproperties.GetBoolPropertyFnFilteredByDomain(forwardingEnabled)\n\treturn domainEntry\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) setupGlobalDeprecatedDomainWithTwoReplicationCluster(forwardingEnabled bool, isRecordActive bool) *cache.DomainCacheEntry {\n\tactiveCluster := s.alternativeClusterName\n\tif isRecordActive {\n\t\tactiveCluster = s.currentClusterName\n\t}\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.domainID, Name: s.domainName, Status: persistence.DomainStatusDeprecated},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: activeCluster,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234, // not used\n\t)\n\n\ts.mockConfig.EnableDomainNotActiveAutoForwarding = dynamicproperties.GetBoolPropertyFnFilteredByDomain(forwardingEnabled)\n\treturn domainEntry\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) setupActiveActiveDomainWithTwoReplicationCluster(forwardingEnabled bool) *cache.DomainCacheEntry {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.domainID, Name: s.domainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: s.currentClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t1234, // not used\n\t)\n\n\ts.mockConfig.EnableDomainNotActiveAutoForwarding = dynamicproperties.GetBoolPropertyFnFilteredByDomain(forwardingEnabled)\n\treturn domainEntry\n}\n\nfunc (s *selectedAPIsForwardingRedirectionPolicySuite) TestActiveClusterForActiveActiveDomainRequest() {\n\tdomainEntry := s.setupActiveActiveDomainWithTwoReplicationCluster(true)\n\n\tusEastStickyPlcy := &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-east\",\n\t\t},\n\t}\n\n\tusWestStickyPlcy := &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname                   string\n\t\tapiName                string\n\t\tdomainEntry            *cache.DomainCacheEntry\n\t\tworkflowExecution      *types.WorkflowExecution\n\t\tactClSelPolicyForNewWF *types.ActiveClusterSelectionPolicy\n\t\tmockFn                 func(activeClusterManager *activecluster.MockManager)\n\t\twant                   string\n\t}{\n\t\t{\n\t\t\tname:                   \"new workflow with policy\",\n\t\t\tapiName:                \"StartWorkflowExecution\",\n\t\t\tdomainEntry:            domainEntry,\n\t\t\tactClSelPolicyForNewWF: usWestStickyPlcy,\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), domainEntry.GetInfo().ID, usWestStickyPlcy.GetClusterAttribute()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: s.alternativeClusterName,\n\t\t},\n\t\t{\n\t\t\tname:                   \"new workflow with policy - lookup failed\",\n\t\t\tapiName:                \"StartWorkflowExecution\",\n\t\t\tdomainEntry:            domainEntry,\n\t\t\tactClSelPolicyForNewWF: usEastStickyPlcy,\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), domainEntry.GetInfo().ID, usEastStickyPlcy.GetClusterAttribute()).Return(nil, errors.New(\"lookup failed\"))\n\t\t\t},\n\t\t\twant: s.currentClusterName,\n\t\t},\n\t\t{\n\t\t\tname:        \"existing workflow - missing workflow execution\",\n\t\t\tapiName:     \"SignalWorkflowExecution\",\n\t\t\tdomainEntry: domainEntry,\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t},\n\t\t\twant: s.currentClusterName,\n\t\t},\n\t\t{\n\t\t\tname:        \"existing workflow - missing workflow id\",\n\t\t\tapiName:     \"SignalWorkflowExecution\",\n\t\t\tdomainEntry: domainEntry,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tRunID: \"run1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t},\n\t\t\twant: s.currentClusterName,\n\t\t},\n\t\t{\n\t\t\tname:        \"existing workflow - lookup failed\",\n\t\t\tapiName:     \"SignalWorkflowExecution\",\n\t\t\tdomainEntry: domainEntry,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf1\",\n\t\t\t\tRunID:      \"run1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainEntry.GetInfo().ID, \"wf1\", \"run1\").Return(nil, errors.New(\"lookup failed\"))\n\t\t\t},\n\t\t\twant: s.currentClusterName,\n\t\t},\n\t\t{\n\t\t\tname:                   \"SignalWithStartWorkflowExecution - workflow running, use current workflow policy\",\n\t\t\tapiName:                \"SignalWithStartWorkflowExecution\",\n\t\t\tdomainEntry:            domainEntry,\n\t\t\tactClSelPolicyForNewWF: usEastStickyPlcy, // This should be ignored when workflow is running\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\t// Returns the current workflow's policy and running=true\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterSelectionPolicyForCurrentWorkflow(gomock.Any(), domainEntry.GetInfo().ID, \"wf1\").Return(usWestStickyPlcy, true, nil)\n\t\t\t\t// Should use the west policy (from current workflow), not the east policy (from new workflow param)\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), domainEntry.GetInfo().ID, usWestStickyPlcy.GetClusterAttribute()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: s.alternativeClusterName,\n\t\t},\n\t\t{\n\t\t\tname:                   \"SignalWithStartWorkflowExecution - workflow not running, use new workflow policy\",\n\t\t\tapiName:                \"SignalWithStartWorkflowExecution\",\n\t\t\tdomainEntry:            domainEntry,\n\t\t\tactClSelPolicyForNewWF: usWestStickyPlcy,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\t// Returns policy but running=false\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterSelectionPolicyForCurrentWorkflow(gomock.Any(), domainEntry.GetInfo().ID, \"wf1\").Return(usEastStickyPlcy, false, nil)\n\t\t\t\t// Should use the west policy (from new workflow param), not the east policy (from current workflow)\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), domainEntry.GetInfo().ID, usWestStickyPlcy.GetClusterAttribute()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: s.alternativeClusterName,\n\t\t},\n\t\t{\n\t\t\tname:                   \"SignalWithStartWorkflowExecution - lookup failed, use current cluster\",\n\t\t\tapiName:                \"SignalWithStartWorkflowExecution\",\n\t\t\tdomainEntry:            domainEntry,\n\t\t\tactClSelPolicyForNewWF: usWestStickyPlcy,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\t// Lookup fails\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterSelectionPolicyForCurrentWorkflow(gomock.Any(), domainEntry.GetInfo().ID, \"wf1\").Return(nil, false, errors.New(\"lookup failed\"))\n\t\t\t},\n\t\t\twant: s.currentClusterName,\n\t\t},\n\t\t{\n\t\t\tname:        \"existing workflow - success\",\n\t\t\tapiName:     \"SignalWorkflowExecution\",\n\t\t\tdomainEntry: domainEntry,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf1\",\n\t\t\t\tRunID:      \"run1\",\n\t\t\t},\n\t\t\tmockFn: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainEntry.GetInfo().ID, \"wf1\", \"run1\").Return(&types.ActiveClusterInfo{\n\t\t\t\t\tActiveClusterName: s.alternativeClusterName,\n\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: s.alternativeClusterName,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\ts.Run(test.name, func() {\n\t\t\tactiveClusterManager := activecluster.NewMockManager(s.controller)\n\t\t\ttest.mockFn(activeClusterManager)\n\t\t\ts.policy.activeClusterManager = activeClusterManager\n\t\t\tapiName := test.apiName\n\t\t\tif apiName == \"\" {\n\t\t\t\tapiName = \"any random API name\"\n\t\t\t}\n\t\t\ts.Equal(test.want, s.policy.activeClusterForActiveActiveDomainRequest(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttest.domainEntry,\n\t\t\t\ttest.workflowExecution,\n\t\t\t\ttest.actClSelPolicyForNewWF,\n\t\t\t\tapiName,\n\t\t\t))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/utils.go",
    "content": "package clusterredirection\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc getRequestedConsistencyLevelFromContext(ctx context.Context) types.QueryConsistencyLevel {\n\tcall := yarpc.CallFromContext(ctx)\n\tif call == nil {\n\t\treturn types.QueryConsistencyLevelEventual\n\t}\n\tfeatureFlags := client.GetFeatureFlagsFromHeader(call)\n\tif featureFlags.AutoforwardingEnabled {\n\t\treturn types.QueryConsistencyLevelStrong\n\t}\n\treturn types.QueryConsistencyLevelEventual\n}\n"
  },
  {
    "path": "service/frontend/wrappers/clusterredirection/utils_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage clusterredirection\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc/yarpctest\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetRequestedConsistencyLevelFromContext(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tfeatureFlags apiv1.FeatureFlags\n\t\texpected     types.QueryConsistencyLevel\n\t}{\n\t\t{\n\t\t\tname:         \"empty feature flags\",\n\t\t\tfeatureFlags: apiv1.FeatureFlags{},\n\t\t\texpected:     types.QueryConsistencyLevelEventual,\n\t\t},\n\t\t{\n\t\t\tname:         \"auto forwarding disabled\",\n\t\t\tfeatureFlags: apiv1.FeatureFlags{AutoforwardingEnabled: false},\n\t\t\texpected:     types.QueryConsistencyLevelEventual,\n\t\t},\n\t\t{\n\t\t\tname:         \"autoforwarding enabled\",\n\t\t\tfeatureFlags: apiv1.FeatureFlags{AutoforwardingEnabled: true},\n\t\t\texpected:     types.QueryConsistencyLevelStrong,\n\t\t},\n\t\t{\n\t\t\tname: \"no autoforwarding field\",\n\t\t\tfeatureFlags: apiv1.FeatureFlags{\n\t\t\t\tWorkflowExecutionAlreadyCompletedErrorEnabled: true,\n\t\t\t},\n\t\t\texpected: types.QueryConsistencyLevelEventual,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctx := yarpctest.ContextWithCall(context.Background(), &yarpctest.Call{\n\t\t\t\tHeaders: map[string]string{common.ClientFeatureFlagsHeaderName: client.FeatureFlagsHeader(tt.featureFlags)},\n\t\t\t})\n\n\t\t\tresult := getRequestedConsistencyLevelFromContext(ctx)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/grpc/admin_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/frontend/admin\"\n)\n\ntype AdminHandler struct {\n\th admin.Handler\n}\n\nfunc NewAdminHandler(h admin.Handler) AdminHandler {\n\treturn AdminHandler{h}\n}\n\nfunc (g AdminHandler) AddSearchAttribute(ctx context.Context, request *adminv1.AddSearchAttributeRequest) (*adminv1.AddSearchAttributeResponse, error) {\n\terr := g.h.AddSearchAttribute(ctx, proto.ToAdminAddSearchAttributeRequest(request))\n\treturn &adminv1.AddSearchAttributeResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) CloseShard(ctx context.Context, request *adminv1.CloseShardRequest) (*adminv1.CloseShardResponse, error) {\n\terr := g.h.CloseShard(ctx, proto.ToAdminCloseShardRequest(request))\n\treturn &adminv1.CloseShardResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) CountDLQMessages(ctx context.Context, request *adminv1.CountDLQMessagesRequest) (*adminv1.CountDLQMessagesResponse, error) {\n\tresponse, err := g.h.CountDLQMessages(ctx, proto.ToAdminCountDLQMessagesRequest(request))\n\treturn proto.FromAdminCountDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DeleteWorkflow(ctx context.Context, request *adminv1.DeleteWorkflowRequest) (*adminv1.DeleteWorkflowResponse, error) {\n\tresponse, err := g.h.DeleteWorkflow(ctx, proto.ToAdminDeleteWorkflowRequest(request))\n\treturn proto.FromAdminDeleteWorkflowResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeCluster(ctx context.Context, request *adminv1.DescribeClusterRequest) (*adminv1.DescribeClusterResponse, error) {\n\tresponse, err := g.h.DescribeCluster(ctx)\n\treturn proto.FromAdminDescribeClusterResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeHistoryHost(ctx context.Context, request *adminv1.DescribeHistoryHostRequest) (*adminv1.DescribeHistoryHostResponse, error) {\n\tresponse, err := g.h.DescribeHistoryHost(ctx, proto.ToAdminDescribeHistoryHostRequest(request))\n\treturn proto.FromAdminDescribeHistoryHostResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeQueue(ctx context.Context, request *adminv1.DescribeQueueRequest) (*adminv1.DescribeQueueResponse, error) {\n\tresponse, err := g.h.DescribeQueue(ctx, proto.ToAdminDescribeQueueRequest(request))\n\treturn proto.FromAdminDescribeQueueResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeShardDistribution(ctx context.Context, request *adminv1.DescribeShardDistributionRequest) (*adminv1.DescribeShardDistributionResponse, error) {\n\tresponse, err := g.h.DescribeShardDistribution(ctx, proto.ToAdminDescribeShardDistributionRequest(request))\n\treturn proto.FromAdminDescribeShardDistributionResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeWorkflowExecution(ctx context.Context, request *adminv1.DescribeWorkflowExecutionRequest) (*adminv1.DescribeWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, proto.ToAdminDescribeWorkflowExecutionRequest(request))\n\treturn proto.FromAdminDescribeWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetCrossClusterTasks(ctx context.Context, request *adminv1.GetCrossClusterTasksRequest) (*adminv1.GetCrossClusterTasksResponse, error) {\n\tresponse, err := g.h.GetCrossClusterTasks(ctx, proto.ToAdminGetCrossClusterTasksRequest(request))\n\treturn proto.FromAdminGetCrossClusterTasksResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetDLQReplicationMessages(ctx context.Context, request *adminv1.GetDLQReplicationMessagesRequest) (*adminv1.GetDLQReplicationMessagesResponse, error) {\n\tresponse, err := g.h.GetDLQReplicationMessages(ctx, proto.ToAdminGetDLQReplicationMessagesRequest(request))\n\treturn proto.FromAdminGetDLQReplicationMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, request *adminv1.GetDomainAsyncWorkflowConfiguratonRequest) (*adminv1.GetDomainAsyncWorkflowConfiguratonResponse, error) {\n\tresponse, err := g.h.GetDomainAsyncWorkflowConfiguraton(ctx, proto.ToAdminGetDomainAsyncWorkflowConfiguratonRequest(request))\n\treturn proto.FromAdminGetDomainAsyncWorkflowConfiguratonResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainIsolationGroups(ctx context.Context, request *adminv1.GetDomainIsolationGroupsRequest) (*adminv1.GetDomainIsolationGroupsResponse, error) {\n\tresponse, err := g.h.GetDomainIsolationGroups(ctx, proto.ToAdminGetDomainIsolationGroupsRequest(request))\n\treturn proto.FromAdminGetDomainIsolationGroupsResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainReplicationMessages(ctx context.Context, request *adminv1.GetDomainReplicationMessagesRequest) (*adminv1.GetDomainReplicationMessagesResponse, error) {\n\tresponse, err := g.h.GetDomainReplicationMessages(ctx, proto.ToAdminGetDomainReplicationMessagesRequest(request))\n\treturn proto.FromAdminGetDomainReplicationMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetDynamicConfig(ctx context.Context, request *adminv1.GetDynamicConfigRequest) (*adminv1.GetDynamicConfigResponse, error) {\n\tresponse, err := g.h.GetDynamicConfig(ctx, proto.ToAdminGetDynamicConfigRequest(request))\n\treturn proto.FromAdminGetDynamicConfigResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetGlobalIsolationGroups(ctx context.Context, request *adminv1.GetGlobalIsolationGroupsRequest) (*adminv1.GetGlobalIsolationGroupsResponse, error) {\n\tresponse, err := g.h.GetGlobalIsolationGroups(ctx, proto.ToAdminGetGlobalIsolationGroupsRequest(request))\n\treturn proto.FromAdminGetGlobalIsolationGroupsResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetReplicationMessages(ctx context.Context, request *adminv1.GetReplicationMessagesRequest) (*adminv1.GetReplicationMessagesResponse, error) {\n\tresponse, err := g.h.GetReplicationMessages(ctx, proto.ToAdminGetReplicationMessagesRequest(request))\n\treturn proto.FromAdminGetReplicationMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) GetWorkflowExecutionRawHistoryV2(ctx context.Context, request *adminv1.GetWorkflowExecutionRawHistoryV2Request) (*adminv1.GetWorkflowExecutionRawHistoryV2Response, error) {\n\tresponse, err := g.h.GetWorkflowExecutionRawHistoryV2(ctx, proto.ToAdminGetWorkflowExecutionRawHistoryV2Request(request))\n\treturn proto.FromAdminGetWorkflowExecutionRawHistoryV2Response(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) ListDynamicConfig(ctx context.Context, request *adminv1.ListDynamicConfigRequest) (*adminv1.ListDynamicConfigResponse, error) {\n\tresponse, err := g.h.ListDynamicConfig(ctx, proto.ToAdminListDynamicConfigRequest(request))\n\treturn proto.FromAdminListDynamicConfigResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) MaintainCorruptWorkflow(ctx context.Context, request *adminv1.MaintainCorruptWorkflowRequest) (*adminv1.MaintainCorruptWorkflowResponse, error) {\n\tresponse, err := g.h.MaintainCorruptWorkflow(ctx, proto.ToAdminMaintainCorruptWorkflowRequest(request))\n\treturn proto.FromAdminMaintainCorruptWorkflowResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) MergeDLQMessages(ctx context.Context, request *adminv1.MergeDLQMessagesRequest) (*adminv1.MergeDLQMessagesResponse, error) {\n\tresponse, err := g.h.MergeDLQMessages(ctx, proto.ToAdminMergeDLQMessagesRequest(request))\n\treturn proto.FromAdminMergeDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) PurgeDLQMessages(ctx context.Context, request *adminv1.PurgeDLQMessagesRequest) (*adminv1.PurgeDLQMessagesResponse, error) {\n\terr := g.h.PurgeDLQMessages(ctx, proto.ToAdminPurgeDLQMessagesRequest(request))\n\treturn &adminv1.PurgeDLQMessagesResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) ReadDLQMessages(ctx context.Context, request *adminv1.ReadDLQMessagesRequest) (*adminv1.ReadDLQMessagesResponse, error) {\n\tresponse, err := g.h.ReadDLQMessages(ctx, proto.ToAdminReadDLQMessagesRequest(request))\n\treturn proto.FromAdminReadDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) ReapplyEvents(ctx context.Context, request *adminv1.ReapplyEventsRequest) (*adminv1.ReapplyEventsResponse, error) {\n\terr := g.h.ReapplyEvents(ctx, proto.ToAdminReapplyEventsRequest(request))\n\treturn &adminv1.ReapplyEventsResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) RefreshWorkflowTasks(ctx context.Context, request *adminv1.RefreshWorkflowTasksRequest) (*adminv1.RefreshWorkflowTasksResponse, error) {\n\terr := g.h.RefreshWorkflowTasks(ctx, proto.ToAdminRefreshWorkflowTasksRequest(request))\n\treturn &adminv1.RefreshWorkflowTasksResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) RemoveTask(ctx context.Context, request *adminv1.RemoveTaskRequest) (*adminv1.RemoveTaskResponse, error) {\n\terr := g.h.RemoveTask(ctx, proto.ToAdminRemoveTaskRequest(request))\n\treturn &adminv1.RemoveTaskResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) ResendReplicationTasks(ctx context.Context, request *adminv1.ResendReplicationTasksRequest) (*adminv1.ResendReplicationTasksResponse, error) {\n\terr := g.h.ResendReplicationTasks(ctx, proto.ToAdminResendReplicationTasksRequest(request))\n\treturn &adminv1.ResendReplicationTasksResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) ResetQueue(ctx context.Context, request *adminv1.ResetQueueRequest) (*adminv1.ResetQueueResponse, error) {\n\terr := g.h.ResetQueue(ctx, proto.ToAdminResetQueueRequest(request))\n\treturn &adminv1.ResetQueueResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) RespondCrossClusterTasksCompleted(ctx context.Context, request *adminv1.RespondCrossClusterTasksCompletedRequest) (*adminv1.RespondCrossClusterTasksCompletedResponse, error) {\n\tresponse, err := g.h.RespondCrossClusterTasksCompleted(ctx, proto.ToAdminRespondCrossClusterTasksCompletedRequest(request))\n\treturn proto.FromAdminRespondCrossClusterTasksCompletedResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) RestoreDynamicConfig(ctx context.Context, request *adminv1.RestoreDynamicConfigRequest) (*adminv1.RestoreDynamicConfigResponse, error) {\n\terr := g.h.RestoreDynamicConfig(ctx, proto.ToAdminRestoreDynamicConfigRequest(request))\n\treturn &adminv1.RestoreDynamicConfigResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, request *adminv1.UpdateDomainAsyncWorkflowConfiguratonRequest) (*adminv1.UpdateDomainAsyncWorkflowConfiguratonResponse, error) {\n\tresponse, err := g.h.UpdateDomainAsyncWorkflowConfiguraton(ctx, proto.ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(request))\n\treturn proto.FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDomainIsolationGroups(ctx context.Context, request *adminv1.UpdateDomainIsolationGroupsRequest) (*adminv1.UpdateDomainIsolationGroupsResponse, error) {\n\tresponse, err := g.h.UpdateDomainIsolationGroups(ctx, proto.ToAdminUpdateDomainIsolationGroupsRequest(request))\n\treturn proto.FromAdminUpdateDomainIsolationGroupsResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDynamicConfig(ctx context.Context, request *adminv1.UpdateDynamicConfigRequest) (*adminv1.UpdateDynamicConfigResponse, error) {\n\terr := g.h.UpdateDynamicConfig(ctx, proto.ToAdminUpdateDynamicConfigRequest(request))\n\treturn &adminv1.UpdateDynamicConfigResponse{}, proto.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateGlobalIsolationGroups(ctx context.Context, request *adminv1.UpdateGlobalIsolationGroupsRequest) (*adminv1.UpdateGlobalIsolationGroupsResponse, error) {\n\tresponse, err := g.h.UpdateGlobalIsolationGroups(ctx, proto.ToAdminUpdateGlobalIsolationGroupsRequest(request))\n\treturn proto.FromAdminUpdateGlobalIsolationGroupsResponse(response), proto.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateTaskListPartitionConfig(ctx context.Context, request *adminv1.UpdateTaskListPartitionConfigRequest) (*adminv1.UpdateTaskListPartitionConfigResponse, error) {\n\tresponse, err := g.h.UpdateTaskListPartitionConfig(ctx, proto.ToAdminUpdateTaskListPartitionConfigRequest(request))\n\treturn proto.FromAdminUpdateTaskListPartitionConfigResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/grpc/api_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n)\n\ntype APIHandler struct {\n\th api.Handler\n}\n\nfunc NewAPIHandler(h api.Handler) APIHandler {\n\treturn APIHandler{h}\n}\n\nfunc (g APIHandler) CountWorkflowExecutions(ctx context.Context, request *apiv1.CountWorkflowExecutionsRequest) (*apiv1.CountWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.CountWorkflowExecutions(ctx, proto.ToCountWorkflowExecutionsRequest(request))\n\treturn proto.FromCountWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) DeleteDomain(ctx context.Context, request *apiv1.DeleteDomainRequest) (*apiv1.DeleteDomainResponse, error) {\n\terr := g.h.DeleteDomain(ctx, proto.ToDeleteDomainRequest(request))\n\treturn &apiv1.DeleteDomainResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) DeprecateDomain(ctx context.Context, request *apiv1.DeprecateDomainRequest) (*apiv1.DeprecateDomainResponse, error) {\n\terr := g.h.DeprecateDomain(ctx, proto.ToDeprecateDomainRequest(request))\n\treturn &apiv1.DeprecateDomainResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) DescribeDomain(ctx context.Context, request *apiv1.DescribeDomainRequest) (*apiv1.DescribeDomainResponse, error) {\n\tresponse, err := g.h.DescribeDomain(ctx, proto.ToDescribeDomainRequest(request))\n\treturn proto.FromDescribeDomainResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) DescribeTaskList(ctx context.Context, request *apiv1.DescribeTaskListRequest) (*apiv1.DescribeTaskListResponse, error) {\n\tresponse, err := g.h.DescribeTaskList(ctx, proto.ToDescribeTaskListRequest(request))\n\treturn proto.FromDescribeTaskListResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) DescribeWorkflowExecution(ctx context.Context, request *apiv1.DescribeWorkflowExecutionRequest) (*apiv1.DescribeWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, proto.ToDescribeWorkflowExecutionRequest(request))\n\treturn proto.FromDescribeWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) DiagnoseWorkflowExecution(ctx context.Context, request *apiv1.DiagnoseWorkflowExecutionRequest) (*apiv1.DiagnoseWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.DiagnoseWorkflowExecution(ctx, proto.ToDiagnoseWorkflowExecutionRequest(request))\n\treturn proto.FromDiagnoseWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) FailoverDomain(ctx context.Context, request *apiv1.FailoverDomainRequest) (*apiv1.FailoverDomainResponse, error) {\n\tresponse, err := g.h.FailoverDomain(ctx, proto.ToFailoverDomainRequest(request))\n\treturn proto.FromFailoverDomainResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) GetClusterInfo(ctx context.Context, request *apiv1.GetClusterInfoRequest) (*apiv1.GetClusterInfoResponse, error) {\n\tresponse, err := g.h.GetClusterInfo(ctx)\n\treturn proto.FromGetClusterInfoResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) GetSearchAttributes(ctx context.Context, request *apiv1.GetSearchAttributesRequest) (*apiv1.GetSearchAttributesResponse, error) {\n\tresponse, err := g.h.GetSearchAttributes(ctx)\n\treturn proto.FromGetSearchAttributesResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) GetTaskListsByDomain(ctx context.Context, request *apiv1.GetTaskListsByDomainRequest) (*apiv1.GetTaskListsByDomainResponse, error) {\n\tresponse, err := g.h.GetTaskListsByDomain(ctx, proto.ToGetTaskListsByDomainRequest(request))\n\treturn proto.FromGetTaskListsByDomainResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) GetWorkflowExecutionHistory(ctx context.Context, request *apiv1.GetWorkflowExecutionHistoryRequest) (*apiv1.GetWorkflowExecutionHistoryResponse, error) {\n\tresponse, err := g.h.GetWorkflowExecutionHistory(ctx, proto.ToGetWorkflowExecutionHistoryRequest(request))\n\treturn proto.FromGetWorkflowExecutionHistoryResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListArchivedWorkflowExecutions(ctx context.Context, request *apiv1.ListArchivedWorkflowExecutionsRequest) (*apiv1.ListArchivedWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.ListArchivedWorkflowExecutions(ctx, proto.ToListArchivedWorkflowExecutionsRequest(request))\n\treturn proto.FromListArchivedWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListClosedWorkflowExecutions(ctx context.Context, request *apiv1.ListClosedWorkflowExecutionsRequest) (*apiv1.ListClosedWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.ListClosedWorkflowExecutions(ctx, proto.ToListClosedWorkflowExecutionsRequest(request))\n\treturn proto.FromListClosedWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListDomains(ctx context.Context, request *apiv1.ListDomainsRequest) (*apiv1.ListDomainsResponse, error) {\n\tresponse, err := g.h.ListDomains(ctx, proto.ToListDomainsRequest(request))\n\treturn proto.FromListDomainsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListFailoverHistory(ctx context.Context, request *apiv1.ListFailoverHistoryRequest) (*apiv1.ListFailoverHistoryResponse, error) {\n\tresponse, err := g.h.ListFailoverHistory(ctx, proto.ToListFailoverHistoryRequest(request))\n\treturn proto.FromListFailoverHistoryResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListOpenWorkflowExecutions(ctx context.Context, request *apiv1.ListOpenWorkflowExecutionsRequest) (*apiv1.ListOpenWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.ListOpenWorkflowExecutions(ctx, proto.ToListOpenWorkflowExecutionsRequest(request))\n\treturn proto.FromListOpenWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListTaskListPartitions(ctx context.Context, request *apiv1.ListTaskListPartitionsRequest) (*apiv1.ListTaskListPartitionsResponse, error) {\n\tresponse, err := g.h.ListTaskListPartitions(ctx, proto.ToListTaskListPartitionsRequest(request))\n\treturn proto.FromListTaskListPartitionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ListWorkflowExecutions(ctx context.Context, request *apiv1.ListWorkflowExecutionsRequest) (*apiv1.ListWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.ListWorkflowExecutions(ctx, proto.ToListWorkflowExecutionsRequest(request))\n\treturn proto.FromListWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) PollForActivityTask(ctx context.Context, request *apiv1.PollForActivityTaskRequest) (*apiv1.PollForActivityTaskResponse, error) {\n\tresponse, err := g.h.PollForActivityTask(ctx, proto.ToPollForActivityTaskRequest(request))\n\treturn proto.FromPollForActivityTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) PollForDecisionTask(ctx context.Context, request *apiv1.PollForDecisionTaskRequest) (*apiv1.PollForDecisionTaskResponse, error) {\n\tresponse, err := g.h.PollForDecisionTask(ctx, proto.ToPollForDecisionTaskRequest(request))\n\treturn proto.FromPollForDecisionTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) QueryWorkflow(ctx context.Context, request *apiv1.QueryWorkflowRequest) (*apiv1.QueryWorkflowResponse, error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, proto.ToQueryWorkflowRequest(request))\n\treturn proto.FromQueryWorkflowResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) RecordActivityTaskHeartbeat(ctx context.Context, request *apiv1.RecordActivityTaskHeartbeatRequest) (*apiv1.RecordActivityTaskHeartbeatResponse, error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeat(ctx, proto.ToRecordActivityTaskHeartbeatRequest(request))\n\treturn proto.FromRecordActivityTaskHeartbeatResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, request *apiv1.RecordActivityTaskHeartbeatByIDRequest) (*apiv1.RecordActivityTaskHeartbeatByIDResponse, error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeatByID(ctx, proto.ToRecordActivityTaskHeartbeatByIDRequest(request))\n\treturn proto.FromRecordActivityTaskHeartbeatByIDResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) RefreshWorkflowTasks(ctx context.Context, request *apiv1.RefreshWorkflowTasksRequest) (*apiv1.RefreshWorkflowTasksResponse, error) {\n\terr := g.h.RefreshWorkflowTasks(ctx, proto.ToRefreshWorkflowTasksRequest(request))\n\treturn &apiv1.RefreshWorkflowTasksResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RegisterDomain(ctx context.Context, request *apiv1.RegisterDomainRequest) (*apiv1.RegisterDomainResponse, error) {\n\terr := g.h.RegisterDomain(ctx, proto.ToRegisterDomainRequest(request))\n\treturn &apiv1.RegisterDomainResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RequestCancelWorkflowExecution(ctx context.Context, request *apiv1.RequestCancelWorkflowExecutionRequest) (*apiv1.RequestCancelWorkflowExecutionResponse, error) {\n\terr := g.h.RequestCancelWorkflowExecution(ctx, proto.ToRequestCancelWorkflowExecutionRequest(request))\n\treturn &apiv1.RequestCancelWorkflowExecutionResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) ResetStickyTaskList(ctx context.Context, request *apiv1.ResetStickyTaskListRequest) (*apiv1.ResetStickyTaskListResponse, error) {\n\tresponse, err := g.h.ResetStickyTaskList(ctx, proto.ToResetStickyTaskListRequest(request))\n\treturn proto.FromResetStickyTaskListResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ResetWorkflowExecution(ctx context.Context, request *apiv1.ResetWorkflowExecutionRequest) (*apiv1.ResetWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.ResetWorkflowExecution(ctx, proto.ToResetWorkflowExecutionRequest(request))\n\treturn proto.FromResetWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCanceled(ctx context.Context, request *apiv1.RespondActivityTaskCanceledRequest) (*apiv1.RespondActivityTaskCanceledResponse, error) {\n\terr := g.h.RespondActivityTaskCanceled(ctx, proto.ToRespondActivityTaskCanceledRequest(request))\n\treturn &apiv1.RespondActivityTaskCanceledResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCanceledByID(ctx context.Context, request *apiv1.RespondActivityTaskCanceledByIDRequest) (*apiv1.RespondActivityTaskCanceledByIDResponse, error) {\n\terr := g.h.RespondActivityTaskCanceledByID(ctx, proto.ToRespondActivityTaskCanceledByIDRequest(request))\n\treturn &apiv1.RespondActivityTaskCanceledByIDResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCompleted(ctx context.Context, request *apiv1.RespondActivityTaskCompletedRequest) (*apiv1.RespondActivityTaskCompletedResponse, error) {\n\terr := g.h.RespondActivityTaskCompleted(ctx, proto.ToRespondActivityTaskCompletedRequest(request))\n\treturn &apiv1.RespondActivityTaskCompletedResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCompletedByID(ctx context.Context, request *apiv1.RespondActivityTaskCompletedByIDRequest) (*apiv1.RespondActivityTaskCompletedByIDResponse, error) {\n\terr := g.h.RespondActivityTaskCompletedByID(ctx, proto.ToRespondActivityTaskCompletedByIDRequest(request))\n\treturn &apiv1.RespondActivityTaskCompletedByIDResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskFailed(ctx context.Context, request *apiv1.RespondActivityTaskFailedRequest) (*apiv1.RespondActivityTaskFailedResponse, error) {\n\terr := g.h.RespondActivityTaskFailed(ctx, proto.ToRespondActivityTaskFailedRequest(request))\n\treturn &apiv1.RespondActivityTaskFailedResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskFailedByID(ctx context.Context, request *apiv1.RespondActivityTaskFailedByIDRequest) (*apiv1.RespondActivityTaskFailedByIDResponse, error) {\n\terr := g.h.RespondActivityTaskFailedByID(ctx, proto.ToRespondActivityTaskFailedByIDRequest(request))\n\treturn &apiv1.RespondActivityTaskFailedByIDResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondDecisionTaskCompleted(ctx context.Context, request *apiv1.RespondDecisionTaskCompletedRequest) (*apiv1.RespondDecisionTaskCompletedResponse, error) {\n\tresponse, err := g.h.RespondDecisionTaskCompleted(ctx, proto.ToRespondDecisionTaskCompletedRequest(request))\n\treturn proto.FromRespondDecisionTaskCompletedResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondDecisionTaskFailed(ctx context.Context, request *apiv1.RespondDecisionTaskFailedRequest) (*apiv1.RespondDecisionTaskFailedResponse, error) {\n\terr := g.h.RespondDecisionTaskFailed(ctx, proto.ToRespondDecisionTaskFailedRequest(request))\n\treturn &apiv1.RespondDecisionTaskFailedResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RespondQueryTaskCompleted(ctx context.Context, request *apiv1.RespondQueryTaskCompletedRequest) (*apiv1.RespondQueryTaskCompletedResponse, error) {\n\terr := g.h.RespondQueryTaskCompleted(ctx, proto.ToRespondQueryTaskCompletedRequest(request))\n\treturn &apiv1.RespondQueryTaskCompletedResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) RestartWorkflowExecution(ctx context.Context, request *apiv1.RestartWorkflowExecutionRequest) (*apiv1.RestartWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.RestartWorkflowExecution(ctx, proto.ToRestartWorkflowExecutionRequest(request))\n\treturn proto.FromRestartWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) ScanWorkflowExecutions(ctx context.Context, request *apiv1.ScanWorkflowExecutionsRequest) (*apiv1.ScanWorkflowExecutionsResponse, error) {\n\tresponse, err := g.h.ScanWorkflowExecutions(ctx, proto.ToScanWorkflowExecutionsRequest(request))\n\treturn proto.FromScanWorkflowExecutionsResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) SignalWithStartWorkflowExecution(ctx context.Context, request *apiv1.SignalWithStartWorkflowExecutionRequest) (*apiv1.SignalWithStartWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecution(ctx, proto.ToSignalWithStartWorkflowExecutionRequest(request))\n\treturn proto.FromSignalWithStartWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, request *apiv1.SignalWithStartWorkflowExecutionAsyncRequest) (*apiv1.SignalWithStartWorkflowExecutionAsyncResponse, error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecutionAsync(ctx, proto.ToSignalWithStartWorkflowExecutionAsyncRequest(request))\n\treturn proto.FromSignalWithStartWorkflowExecutionAsyncResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) SignalWorkflowExecution(ctx context.Context, request *apiv1.SignalWorkflowExecutionRequest) (*apiv1.SignalWorkflowExecutionResponse, error) {\n\terr := g.h.SignalWorkflowExecution(ctx, proto.ToSignalWorkflowExecutionRequest(request))\n\treturn &apiv1.SignalWorkflowExecutionResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) StartWorkflowExecution(ctx context.Context, request *apiv1.StartWorkflowExecutionRequest) (*apiv1.StartWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.StartWorkflowExecution(ctx, proto.ToStartWorkflowExecutionRequest(request))\n\treturn proto.FromStartWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) StartWorkflowExecutionAsync(ctx context.Context, request *apiv1.StartWorkflowExecutionAsyncRequest) (*apiv1.StartWorkflowExecutionAsyncResponse, error) {\n\tresponse, err := g.h.StartWorkflowExecutionAsync(ctx, proto.ToStartWorkflowExecutionAsyncRequest(request))\n\treturn proto.FromStartWorkflowExecutionAsyncResponse(response), proto.FromError(err)\n}\n\nfunc (g APIHandler) TerminateWorkflowExecution(ctx context.Context, request *apiv1.TerminateWorkflowExecutionRequest) (*apiv1.TerminateWorkflowExecutionResponse, error) {\n\terr := g.h.TerminateWorkflowExecution(ctx, proto.ToTerminateWorkflowExecutionRequest(request))\n\treturn &apiv1.TerminateWorkflowExecutionResponse{}, proto.FromError(err)\n}\n\nfunc (g APIHandler) UpdateDomain(ctx context.Context, request *apiv1.UpdateDomainRequest) (*apiv1.UpdateDomainResponse, error) {\n\tresponse, err := g.h.UpdateDomain(ctx, proto.ToUpdateDomainRequest(request))\n\treturn proto.FromUpdateDomainResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/grpc/share.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage grpc\n\nimport (\n\t\"context\"\n\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g AdminHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(adminv1.BuildAdminAPIYARPCProcedures(g))\n}\n\nfunc (g APIHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(apiv1.BuildDomainAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildWorkflowAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildWorkerAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildVisibilityAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildMetaAPIYARPCProcedures(g))\n}\n\nfunc (g APIHandler) Health(ctx context.Context, request *apiv1.HealthRequest) (*apiv1.HealthResponse, error) {\n\tresponse, err := g.h.Health(ctx)\n\treturn proto.FromHealthResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/metered/api_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\n// apiHandler frontend handler wrapper for authentication and authorization\ntype apiHandler struct {\n\thandler         api.Handler\n\tlogger          log.Logger\n\tmetricsClient   metrics.Client\n\tdomainCache     cache.DomainCache\n\tcfg             *config.Config\n\ttokenSerializer common.TaskTokenSerializer\n}\n\n// NewAPIHandler creates frontend handler with metrics and logging\nfunc NewAPIHandler(handler api.Handler, logger log.Logger, metricsClient metrics.Client, domainCache cache.DomainCache, cfg *config.Config) api.Handler {\n\treturn &apiHandler{\n\t\thandler:         handler,\n\t\tlogger:          logger,\n\t\tmetricsClient:   metricsClient,\n\t\tdomainCache:     domainCache,\n\t\tcfg:             cfg,\n\t\ttokenSerializer: common.NewJSONTaskTokenSerializer(),\n\t}\n}\n\nfunc (h *apiHandler) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"CountWorkflowExecutions\")}\n\ttags = append(tags, toCountWorkflowExecutionsRequestTags(cp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendCountWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(cp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tcp2, err = h.handler.CountWorkflowExecutions(ctx, cp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn cp2, err\n}\nfunc (h *apiHandler) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DeleteDomain\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendDeleteDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.DeleteDomain(ctx, dp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DeprecateDomain\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendDeprecateDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.DeprecateDomain(ctx, dp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest) (dp2 *types.DescribeDomainResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DescribeDomain\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendDescribeDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tdp2, err = h.handler.DescribeDomain(ctx, dp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn dp2, err\n}\nfunc (h *apiHandler) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest) (dp2 *types.DescribeTaskListResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DescribeTaskList\")}\n\ttags = append(tags, toDescribeTaskListRequestTags(dp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendDescribeTaskListScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(dp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tdp2, err = h.handler.DescribeTaskList(ctx, dp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn dp2, err\n}\nfunc (h *apiHandler) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DescribeWorkflowExecution\")}\n\ttags = append(tags, toDescribeWorkflowExecutionRequestTags(dp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendDescribeWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(dp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tdp2, err = h.handler.DescribeWorkflowExecution(ctx, dp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn dp2, err\n}\nfunc (h *apiHandler) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"DiagnoseWorkflowExecution\")}\n\ttags = append(tags, toDiagnoseWorkflowExecutionRequestTags(dp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendDiagnoseWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(dp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tdp2, err = h.handler.DiagnoseWorkflowExecution(ctx, dp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn dp2, err\n}\nfunc (h *apiHandler) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest) (fp2 *types.FailoverDomainResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"FailoverDomain\")}\n\ttags = append(tags, toFailoverDomainRequestTags(fp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendFailoverDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(fp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tfp2, err = h.handler.FailoverDomain(ctx, fp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn fp2, err\n}\nfunc (h *apiHandler) GetClusterInfo(ctx context.Context) (cp1 *types.ClusterInfo, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"GetClusterInfo\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendGetClusterInfoScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tcp1, err = h.handler.GetClusterInfo(ctx)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn cp1, err\n}\nfunc (h *apiHandler) GetSearchAttributes(ctx context.Context) (gp1 *types.GetSearchAttributesResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"GetSearchAttributes\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendGetSearchAttributesScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tgp1, err = h.handler.GetSearchAttributes(ctx)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn gp1, err\n}\nfunc (h *apiHandler) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"GetTaskListsByDomain\")}\n\ttags = append(tags, toGetTaskListsByDomainRequestTags(gp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendGetTaskListsByDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(gp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tgp2, err = h.handler.GetTaskListsByDomain(ctx, gp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn gp2, err\n}\nfunc (h *apiHandler) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"GetWorkflowExecutionHistory\")}\n\ttags = append(tags, toGetWorkflowExecutionHistoryRequestTags(gp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendGetWorkflowExecutionHistoryScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(gp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tgp2, err = h.handler.GetWorkflowExecutionHistory(ctx, gp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn gp2, err\n}\nfunc (h *apiHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn h.handler.Health(ctx)\n}\nfunc (h *apiHandler) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListArchivedWorkflowExecutions\")}\n\ttags = append(tags, toListArchivedWorkflowExecutionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendListArchivedWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListArchivedWorkflowExecutions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListClosedWorkflowExecutions\")}\n\ttags = append(tags, toListClosedWorkflowExecutionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendListClosedWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListClosedWorkflowExecutions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest) (lp2 *types.ListDomainsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListDomains\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendListDomainsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListDomains(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListFailoverHistory\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendListFailoverHistoryScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListFailoverHistory(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListOpenWorkflowExecutions\")}\n\ttags = append(tags, toListOpenWorkflowExecutionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendListOpenWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListOpenWorkflowExecutions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListTaskListPartitions\")}\n\ttags = append(tags, toListTaskListPartitionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendListTaskListPartitionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListTaskListPartitions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ListWorkflowExecutions\")}\n\ttags = append(tags, toListWorkflowExecutionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendListWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ListWorkflowExecutions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"PollForActivityTask\")}\n\ttags = append(tags, toPollForActivityTaskRequestTags(pp1)...)\n\tscope := common.NewPerTaskListScope(pp1.Domain, pp1.TaskList.GetName(), pp1.TaskList.GetKind(), h.metricsClient, metrics.FrontendPollForActivityTaskScope).Tagged(metrics.GetContextTags(ctx)...)\n\tscope.IncCounter(metrics.CadenceRequestsPerTaskListWithoutRollup)\n\tsw := scope.StartTimer(metrics.CadenceLatencyPerTaskList)\n\tdefer sw.Stop()\n\tscopePerDomain := h.metricsClient.Scope(metrics.FrontendPollForActivityTaskScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(pp1.GetDomain()))...)\n\tscopePerDomain.IncCounter(metrics.CadenceRequests)\n\tswPerDomain := scopePerDomain.StartTimer(metrics.CadenceLatency)\n\tdefer swPerDomain.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tpp2, err = h.handler.PollForActivityTask(ctx, pp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn pp2, err\n}\nfunc (h *apiHandler) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"PollForDecisionTask\")}\n\ttags = append(tags, toPollForDecisionTaskRequestTags(pp1)...)\n\tscope := common.NewPerTaskListScope(pp1.Domain, pp1.TaskList.GetName(), pp1.TaskList.GetKind(), h.metricsClient, metrics.FrontendPollForDecisionTaskScope).Tagged(metrics.GetContextTags(ctx)...)\n\tscope.IncCounter(metrics.CadenceRequestsPerTaskListWithoutRollup)\n\tsw := scope.StartTimer(metrics.CadenceLatencyPerTaskList)\n\tdefer sw.Stop()\n\tscopePerDomain := h.metricsClient.Scope(metrics.FrontendPollForDecisionTaskScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(pp1.GetDomain()))...)\n\tscopePerDomain.IncCounter(metrics.CadenceRequests)\n\tswPerDomain := scopePerDomain.StartTimer(metrics.CadenceLatency)\n\tdefer swPerDomain.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tpp2, err = h.handler.PollForDecisionTask(ctx, pp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn pp2, err\n}\nfunc (h *apiHandler) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest) (qp2 *types.QueryWorkflowResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"QueryWorkflow\")}\n\ttags = append(tags, toQueryWorkflowRequestTags(qp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendQueryWorkflowScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(qp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tqp2, err = h.handler.QueryWorkflow(ctx, qp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn qp2, err\n}\nfunc (h *apiHandler) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RecordActivityTaskHeartbeat\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRecordActivityTaskHeartbeatScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.RecordActivityTaskHeartbeat(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RecordActivityTaskHeartbeatByID\")}\n\ttags = append(tags, toRecordActivityTaskHeartbeatByIDRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRecordActivityTaskHeartbeatByIDScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.RecordActivityTaskHeartbeatByID(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RefreshWorkflowTasks\")}\n\ttags = append(tags, toRefreshWorkflowTasksRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRefreshWorkflowTasksScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RefreshWorkflowTasks(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RegisterDomain\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendRegisterDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RegisterDomain(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RequestCancelWorkflowExecution\")}\n\ttags = append(tags, toRequestCancelWorkflowExecutionRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRequestCancelWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RequestCancelWorkflowExecution(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ResetStickyTaskList\")}\n\ttags = append(tags, toResetStickyTaskListRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendResetStickyTaskListScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.ResetStickyTaskList(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ResetWorkflowExecution\")}\n\ttags = append(tags, toResetWorkflowExecutionRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendResetWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.ResetWorkflowExecution(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskCanceled\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskCanceledScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskCanceled(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskCanceledByID\")}\n\ttags = append(tags, toRespondActivityTaskCanceledByIDRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskCanceledByIDScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskCanceledByID(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskCompleted\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskCompletedScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskCompleted(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskCompletedByID\")}\n\ttags = append(tags, toRespondActivityTaskCompletedByIDRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskCompletedByIDScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskCompletedByID(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskFailed\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskFailedScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskFailed(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondActivityTaskFailedByID\")}\n\ttags = append(tags, toRespondActivityTaskFailedByIDRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondActivityTaskFailedByIDScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondActivityTaskFailedByID(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondDecisionTaskCompleted\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondDecisionTaskCompletedScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.RespondDecisionTaskCompleted(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondDecisionTaskFailed\")}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondDecisionTaskFailedScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondDecisionTaskFailed(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RespondQueryTaskCompleted\")}\n\ttoken, err := h.tokenSerializer.DeserializeQueryTaskToken(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, tag.WorkflowDomainName(domainName))\n\tscope := h.metricsClient.Scope(metrics.FrontendRespondQueryTaskCompletedScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(domainName))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.RespondQueryTaskCompleted(ctx, rp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"RestartWorkflowExecution\")}\n\ttags = append(tags, toRestartWorkflowExecutionRequestTags(rp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendRestartWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(rp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\trp2, err = h.handler.RestartWorkflowExecution(ctx, rp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn rp2, err\n}\nfunc (h *apiHandler) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"ScanWorkflowExecutions\")}\n\ttags = append(tags, toScanWorkflowExecutionsRequestTags(lp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendScanWorkflowExecutionsScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(lp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tlp2, err = h.handler.ScanWorkflowExecutions(ctx, lp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn lp2, err\n}\nfunc (h *apiHandler) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"SignalWithStartWorkflowExecution\")}\n\ttags = append(tags, toSignalWithStartWorkflowExecutionRequestTags(sp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendSignalWithStartWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(sp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tsp2, err = h.handler.SignalWithStartWorkflowExecution(ctx, sp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn sp2, err\n}\nfunc (h *apiHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"SignalWithStartWorkflowExecutionAsync\")}\n\ttags = append(tags, toSignalWithStartWorkflowExecutionAsyncRequestTags(sp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendSignalWithStartWorkflowExecutionAsyncScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(sp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tsp2, err = h.handler.SignalWithStartWorkflowExecutionAsync(ctx, sp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn sp2, err\n}\nfunc (h *apiHandler) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\tctx = h.withSignalName(ctx, sp1.GetDomain(), sp1.GetSignalName())\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"SignalWorkflowExecution\")}\n\ttags = append(tags, toSignalWorkflowExecutionRequestTags(sp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendSignalWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(sp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.SignalWorkflowExecution(ctx, sp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"StartWorkflowExecution\")}\n\ttags = append(tags, toStartWorkflowExecutionRequestTags(sp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendStartWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(sp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tsp2, err = h.handler.StartWorkflowExecution(ctx, sp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn sp2, err\n}\nfunc (h *apiHandler) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"StartWorkflowExecutionAsync\")}\n\ttags = append(tags, toStartWorkflowExecutionAsyncRequestTags(sp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendStartWorkflowExecutionAsyncScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(sp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tsp2, err = h.handler.StartWorkflowExecutionAsync(ctx, sp1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn sp2, err\n}\nfunc (h *apiHandler) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"TerminateWorkflowExecution\")}\n\ttags = append(tags, toTerminateWorkflowExecutionRequestTags(tp1)...)\n\tscope := h.metricsClient.Scope(metrics.FrontendTerminateWorkflowExecutionScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainTag(tp1.GetDomain()))...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\terr = h.handler.TerminateWorkflowExecution(ctx, tp1)\n\tif err != nil {\n\t\treturn h.handleErr(err, scope, logger)\n\t}\n\treturn err\n}\nfunc (h *apiHandler) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest) (up2 *types.UpdateDomainResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\ttags := []tag.Tag{tag.WorkflowHandlerName(\"UpdateDomain\")}\n\tscope := h.metricsClient.Scope(metrics.FrontendUpdateDomainScope).Tagged(append(metrics.GetContextTags(ctx), metrics.DomainUnknownTag())...)\n\tscope.IncCounter(metrics.CadenceRequests)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tags...)\n\n\tup2, err = h.handler.UpdateDomain(ctx, up1)\n\tif err != nil {\n\t\treturn nil, h.handleErr(err, scope, logger)\n\t}\n\treturn up2, err\n}\n"
  },
  {
    "path": "service/frontend/wrappers/metered/metered.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\tcommonerrors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (h *apiHandler) handleErr(err error, scope metrics.Scope, logger log.Logger) error {\n\tlogger = logger.Helper()\n\n\t// attempt to extract hostname if possible\n\thostname, err := commonerrors.ExtractPeerHostname(err)\n\tif hostname != \"\" {\n\t\tlogger = logger.WithTags(tag.PeerHostname(hostname))\n\t}\n\n\tswitch err := err.(type) {\n\tcase *types.InternalServiceError:\n\t\tlogger.Error(\"Internal service error\", tag.Error(err))\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\treturn frontendInternalServiceError(\"cadence internal error, msg: %v\", err.Message)\n\tcase *types.BadRequestError:\n\t\tscope.IncCounter(metrics.CadenceErrBadRequestCounter)\n\t\treturn err\n\tcase *types.DomainNotActiveError:\n\t\tscope.IncCounter(metrics.CadenceErrBadRequestCounter)\n\t\treturn err\n\tcase *types.ServiceBusyError:\n\t\tscope.IncCounter(metrics.CadenceErrServiceBusyCounter)\n\t\treturn err\n\tcase *types.EntityNotExistsError:\n\t\tscope.IncCounter(metrics.CadenceErrEntityNotExistsCounter)\n\t\treturn err\n\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\tscope.IncCounter(metrics.CadenceErrWorkflowExecutionAlreadyCompletedCounter)\n\t\treturn err\n\tcase *types.WorkflowExecutionAlreadyStartedError:\n\t\tscope.IncCounter(metrics.CadenceErrExecutionAlreadyStartedCounter)\n\t\treturn err\n\tcase *types.DomainAlreadyExistsError:\n\t\tscope.IncCounter(metrics.CadenceErrDomainAlreadyExistsCounter)\n\t\treturn err\n\tcase *types.CancellationAlreadyRequestedError:\n\t\tscope.IncCounter(metrics.CadenceErrCancellationAlreadyRequestedCounter)\n\t\treturn err\n\tcase *types.QueryFailedError:\n\t\tscope.IncCounter(metrics.CadenceErrQueryFailedCounter)\n\t\treturn err\n\tcase *types.LimitExceededError:\n\t\tscope.IncCounter(metrics.CadenceErrLimitExceededCounter)\n\t\treturn err\n\tcase *types.ClientVersionNotSupportedError:\n\t\tscope.IncCounter(metrics.CadenceErrClientVersionNotSupportedCounter)\n\t\treturn err\n\tcase *yarpcerrors.Status:\n\t\tif err.Code() == yarpcerrors.CodeDeadlineExceeded {\n\t\t\tlogger.Error(\"Frontend request timedout\", tag.Error(err))\n\t\t\tscope.IncCounter(metrics.CadenceErrContextTimeoutCounter)\n\t\t\treturn err\n\t\t}\n\t\tif err.Code() == yarpcerrors.CodeCancelled {\n\t\t\tscope.IncCounter(metrics.CadenceErrGRPCConnectionClosingCounter)\n\t\t\tlogger.Warn(constants.GRPCConnectionClosingError, tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t}\n\tif errors.Is(err, context.DeadlineExceeded) {\n\t\tlogger.Error(\"Frontend request timedout\", tag.Error(err))\n\t\tscope.IncCounter(metrics.CadenceErrContextTimeoutCounter)\n\t\treturn err\n\t}\n\n\tlogger.Error(\"Uncategorized error\", tag.Error(err))\n\tscope.IncCounter(metrics.CadenceFailures)\n\treturn frontendInternalServiceError(\"cadence internal uncategorized error, msg: %v\", err.Error())\n}\n\nfunc (h *apiHandler) withSignalName(\n\tctx context.Context,\n\tdomainName string,\n\tsignalName string,\n) context.Context {\n\tif h.cfg.EmitSignalNameMetricsTag(domainName) {\n\t\treturn metrics.TagContext(ctx, metrics.SignalNameTag(signalName))\n\t}\n\treturn ctx\n}\n\nfunc frontendInternalServiceError(fmtStr string, args ...interface{}) error {\n\t// NOTE: For internal error, we can't return thrift error from cadence-frontend.\n\t// Because in uber internal metrics, thrift errors are counted as user errors.\n\treturn fmt.Errorf(fmtStr, args...)\n}\n\nfunc toCountWorkflowExecutionsRequestTags(req *types.CountWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toDescribeTaskListRequestTags(req *types.DescribeTaskListRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowTaskListName(req.GetTaskList().GetName()),\n\t\ttag.WorkflowTaskListType(int(req.GetTaskListType())),\n\t\ttag.WorkflowTaskListKind(int32(req.GetTaskList().GetKind())),\n\t}\n}\n\nfunc toDescribeWorkflowExecutionRequestTags(req *types.DescribeWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetExecution().GetRunID()),\n\t}\n}\n\nfunc toDiagnoseWorkflowExecutionRequestTags(req *types.DiagnoseWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toGetTaskListsByDomainRequestTags(req *types.GetTaskListsByDomainRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toGetWorkflowExecutionHistoryRequestTags(req *types.GetWorkflowExecutionHistoryRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toListArchivedWorkflowExecutionsRequestTags(req *types.ListArchivedWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toListClosedWorkflowExecutionsRequestTags(req *types.ListClosedWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toListOpenWorkflowExecutionsRequestTags(req *types.ListOpenWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toListTaskListPartitionsRequestTags(req *types.ListTaskListPartitionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowTaskListName(req.GetTaskList().GetName()),\n\t\ttag.WorkflowTaskListKind(int32(req.GetTaskList().GetKind())),\n\t}\n}\n\nfunc toListWorkflowExecutionsRequestTags(req *types.ListWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toPollForActivityTaskRequestTags(req *types.PollForActivityTaskRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowTaskListName(req.GetTaskList().GetName()),\n\t\ttag.WorkflowTaskListKind(int32(req.GetTaskList().GetKind())),\n\t}\n}\n\nfunc toPollForDecisionTaskRequestTags(req *types.PollForDecisionTaskRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowTaskListName(req.GetTaskList().GetName()),\n\t\ttag.WorkflowTaskListKind(int32(req.GetTaskList().GetKind())),\n\t}\n}\n\nfunc toQueryWorkflowRequestTags(req *types.QueryWorkflowRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetExecution().GetRunID()),\n\t}\n}\n\nfunc toRecordActivityTaskHeartbeatByIDRequestTags(req *types.RecordActivityTaskHeartbeatByIDRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetRunID()),\n\t}\n}\n\nfunc toRefreshWorkflowTasksRequestTags(req *types.RefreshWorkflowTasksRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetExecution().GetRunID()),\n\t}\n}\n\nfunc toRequestCancelWorkflowExecutionRequestTags(req *types.RequestCancelWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toResetStickyTaskListRequestTags(req *types.ResetStickyTaskListRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetExecution().GetRunID()),\n\t}\n}\n\nfunc toResetWorkflowExecutionRequestTags(req *types.ResetWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toRespondActivityTaskCanceledByIDRequestTags(req *types.RespondActivityTaskCanceledByIDRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetRunID()),\n\t}\n}\n\nfunc toRespondActivityTaskCompletedByIDRequestTags(req *types.RespondActivityTaskCompletedByIDRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetRunID()),\n\t}\n}\n\nfunc toRespondActivityTaskFailedByIDRequestTags(req *types.RespondActivityTaskFailedByIDRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetRunID()),\n\t}\n}\n\nfunc toRestartWorkflowExecutionRequestTags(req *types.RestartWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toSignalWithStartWorkflowExecutionRequestTags(req *types.SignalWithStartWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowType(req.WorkflowType.GetName()),\n\t\ttag.WorkflowSignalName(req.GetSignalName()),\n\t}\n}\n\nfunc toSignalWithStartWorkflowExecutionAsyncRequestTags(req *types.SignalWithStartWorkflowExecutionAsyncRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowType(req.WorkflowType.GetName()),\n\t\ttag.WorkflowSignalName(req.GetSignalName()),\n\t}\n}\n\nfunc toSignalWorkflowExecutionRequestTags(req *types.SignalWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t\ttag.WorkflowSignalName(req.GetSignalName()),\n\t}\n}\n\nfunc toStartWorkflowExecutionRequestTags(req *types.StartWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowType(req.WorkflowType.GetName()),\n\t\ttag.WorkflowCronSchedule(req.GetCronSchedule()),\n\t}\n}\n\nfunc toStartWorkflowExecutionAsyncRequestTags(req *types.StartWorkflowExecutionAsyncRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowID()),\n\t\ttag.WorkflowType(req.WorkflowType.GetName()),\n\t\ttag.WorkflowCronSchedule(req.GetCronSchedule()),\n\t}\n}\n\nfunc toTerminateWorkflowExecutionRequestTags(req *types.TerminateWorkflowExecutionRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t\ttag.WorkflowID(req.GetWorkflowExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(req.GetWorkflowExecution().GetRunID()),\n\t}\n}\n\nfunc toScanWorkflowExecutionsRequestTags(req *types.ListWorkflowExecutionsRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n\nfunc toFailoverDomainRequestTags(req *types.FailoverDomainRequest) []tag.Tag {\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainName(req.GetDomain()),\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/metered/metered_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\nfunc TestContextMetricsTags(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockHandler := api.NewMockHandler(ctrl)\n\tmockHandler.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{}, nil).Times(1)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\thandler := NewAPIHandler(mockHandler, testlogger.New(t), metricsClient, mockDomainCache, nil)\n\n\ttag := metrics.TransportTag(\"grpc\")\n\tctx := metrics.TagContext(context.Background(), tag)\n\thandler.CountWorkflowExecutions(ctx, nil)\n\n\tsnapshot := testScope.Snapshot()\n\tfor _, counter := range snapshot.Counters() {\n\t\tif counter.Name() == \"test.cadence_requests\" {\n\t\t\tassert.Equal(t, tag.Value(), counter.Tags()[tag.Key()])\n\t\t\treturn\n\t\t}\n\t}\n\tassert.Fail(t, \"counter not found\")\n}\n\nfunc TestSignalMetricHasSignalName(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockHandler := api.NewMockHandler(ctrl)\n\tmockHandler.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\thandler := NewAPIHandler(mockHandler, testlogger.New(t), metricsClient, mockDomainCache, &config.Config{EmitSignalNameMetricsTag: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)})\n\n\tsignalRequest := &types.SignalWorkflowExecutionRequest{\n\t\tSignalName: \"test_signal\",\n\t}\n\terr := handler.SignalWorkflowExecution(context.Background(), signalRequest)\n\tif err != nil {\n\t\treturn\n\t}\n\n\texpectedMetrics := make(map[string]bool)\n\texpectedMetrics[\"test.cadence_requests\"] = false\n\n\tsnapshot := testScope.Snapshot()\n\tfor _, counter := range snapshot.Counters() {\n\t\tif _, ok := expectedMetrics[counter.Name()]; ok {\n\t\t\texpectedMetrics[counter.Name()] = true\n\t\t}\n\t\tif val, ok := counter.Tags()[\"signalName\"]; ok {\n\t\t\tassert.Equal(t, val, \"test_signal\")\n\t\t} else {\n\t\t\tassert.Fail(t, \"Couldn't find signalName tag\")\n\t\t}\n\t}\n\tassert.True(t, expectedMetrics[\"test.cadence_requests\"])\n}\n\nfunc TestHandleErr_InternalServiceError(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\thandler := &apiHandler{}\n\n\terr := handler.handleErr(&types.InternalServiceError{Message: \"internal error\"}, metricsClient.Scope(0), logger)\n\n\tassert.NotNil(t, err)\n\tassert.Contains(t, err.Error(), \"cadence internal error\")\n}\n\nfunc TestHandleErr_ClientConnectionClosingError(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\thandler := &apiHandler{}\n\n\terr := handler.handleErr(errors.New(constants.GRPCConnectionClosingError), metricsClient.Scope(0), logger)\n\n\tassert.NotNil(t, err)\n\tassert.Contains(t, err.Error(), constants.GRPCConnectionClosingError)\n}\n\nfunc TestHandleErr_UncategorizedError(t *testing.T) {\n\tlogger := testlogger.New(t)\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\thandler := &apiHandler{}\n\n\terr := handler.handleErr(errors.New(\"unknown error\"), metricsClient.Scope(0), logger)\n\n\tassert.NotNil(t, err)\n\tassert.Contains(t, err.Error(), \"cadence internal uncategorized error\")\n}\n\nfunc TestToSignalWithStartWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"test-workflow-type\",\n\t\t},\n\t\tSignalName: \"test-signal\",\n\t}\n\n\ttags := toSignalWithStartWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowType(\"test-workflow-type\"),\n\t\ttag.WorkflowSignalName(\"test-signal\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToDescribeWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toDescribeWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToStartWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"test-workflow-type\",\n\t\t},\n\t\tCronSchedule: \"test-cron-schedule\",\n\t}\n\n\ttags := toStartWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowType(\"test-workflow-type\"),\n\t\ttag.WorkflowCronSchedule(\"test-cron-schedule\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToStartWorkflowExecutionAsyncRequestTags(t *testing.T) {\n\treq := &types.StartWorkflowExecutionAsyncRequest{\n\t\tStartWorkflowExecutionRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:     \"test-domain\",\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"test-workflow-type\",\n\t\t\t},\n\t\t\tCronSchedule: \"test-cron-schedule\",\n\t\t},\n\t}\n\n\ttags := toStartWorkflowExecutionAsyncRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowType(\"test-workflow-type\"),\n\t\ttag.WorkflowCronSchedule(\"test-cron-schedule\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToTerminateWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toTerminateWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToScanWorkflowExecutionsRequestTags(t *testing.T) {\n\treq := &types.ListWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toScanWorkflowExecutionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToListTaskListPartitionsRequestTags(t *testing.T) {\n\tkind := types.TaskListKindNormal\n\treq := &types.ListTaskListPartitionsRequest{\n\t\tDomain: \"test-domain\",\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"test-task-list\",\n\t\t\tKind: &kind,\n\t\t},\n\t}\n\n\ttags := toListTaskListPartitionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowTaskListName(\"test-task-list\"),\n\t\ttag.WorkflowTaskListKind(int32(types.TaskListKindNormal)),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToGetTaskListsByDomainRequestTags(t *testing.T) {\n\treq := &types.GetTaskListsByDomainRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toGetTaskListsByDomainRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToGetWorkflowExecutionHistoryRequestTags(t *testing.T) {\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toGetWorkflowExecutionHistoryRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToListArchivedWorkflowExecutionsRequestTags(t *testing.T) {\n\treq := &types.ListArchivedWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toListArchivedWorkflowExecutionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToListClosedWorkflowExecutionsRequestTags(t *testing.T) {\n\treq := &types.ListClosedWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toListClosedWorkflowExecutionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToListOpenWorkflowExecutionsRequestTags(t *testing.T) {\n\treq := &types.ListOpenWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toListOpenWorkflowExecutionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToRestartWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.RestartWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toRestartWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToRespondActivityTaskFailedByIDRequestTags(t *testing.T) {\n\treq := &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t}\n\n\ttags := toRespondActivityTaskFailedByIDRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToRespondActivityTaskCompletedByIDRequestTags(t *testing.T) {\n\treq := &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t}\n\n\ttags := toRespondActivityTaskCompletedByIDRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToRespondActivityTaskCanceledByIDRequestTags(t *testing.T) {\n\treq := &types.RespondActivityTaskCanceledByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t}\n\n\ttags := toRespondActivityTaskCanceledByIDRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToResetWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.ResetWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toResetWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToResetStickyTaskListRequestTags(t *testing.T) {\n\treq := &types.ResetStickyTaskListRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toResetStickyTaskListRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToRequestCancelWorkflowExecutionRequestTags(t *testing.T) {\n\treq := &types.RequestCancelWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toRequestCancelWorkflowExecutionRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToRefreshWorkflowTasksRequestTags(t *testing.T) {\n\treq := &types.RefreshWorkflowTasksRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toRefreshWorkflowTasksRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToRecordActivityTaskHeartbeatByIDRequestTags(t *testing.T) {\n\treq := &types.RecordActivityTaskHeartbeatByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t}\n\n\ttags := toRecordActivityTaskHeartbeatByIDRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToQueryWorkflowRequestTags(t *testing.T) {\n\treq := &types.QueryWorkflowRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}\n\n\ttags := toQueryWorkflowRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowRunID(\"test-run-id\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\nfunc TestToPollForDecisionTaskRequestTags(t *testing.T) {\n\tkind := types.TaskListKindNormal\n\treq := &types.PollForDecisionTaskRequest{\n\t\tDomain: \"test-domain\",\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"test-task-list\",\n\t\t\tKind: &kind,\n\t\t},\n\t}\n\n\ttags := toPollForDecisionTaskRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowTaskListName(\"test-task-list\"),\n\t\ttag.WorkflowTaskListKind(int32(types.TaskListKindNormal)),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToListWorkflowExecutionsRequestTags(t *testing.T) {\n\treq := &types.ListWorkflowExecutionsRequest{\n\t\tDomain: \"test-domain\",\n\t}\n\n\ttags := toListWorkflowExecutionsRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToPollForActivityTaskRequestTags(t *testing.T) {\n\tkind := types.TaskListKindNormal\n\treq := &types.PollForActivityTaskRequest{\n\t\tDomain: \"test-domain\",\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"test-task-list\",\n\t\t\tKind: &kind,\n\t\t},\n\t}\n\n\ttags := toPollForActivityTaskRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowTaskListName(\"test-task-list\"),\n\t\ttag.WorkflowTaskListKind(int32(types.TaskListKindNormal)),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToSignalWithStartWorkflowExecutionAsyncRequestTags(t *testing.T) {\n\treq := &types.SignalWithStartWorkflowExecutionAsyncRequest{\n\t\tSignalWithStartWorkflowExecutionRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     \"test-domain\",\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"test-workflow-type\",\n\t\t\t},\n\t\t\tSignalName: \"test-signal\",\n\t\t},\n\t}\n\n\ttags := toSignalWithStartWorkflowExecutionAsyncRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowID(\"test-workflow-id\"),\n\t\ttag.WorkflowType(\"test-workflow-type\"),\n\t\ttag.WorkflowSignalName(\"test-signal\"),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestToDescribeTaskListRequestTags(t *testing.T) {\n\tkind := types.TaskListKindNormal\n\ttaskListType := types.TaskListTypeDecision\n\treq := &types.DescribeTaskListRequest{\n\t\tDomain: \"test-domain\",\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"test-task-list\",\n\t\t\tKind: &kind,\n\t\t},\n\t\tTaskListType: &taskListType,\n\t}\n\n\ttags := toDescribeTaskListRequestTags(req)\n\n\texpectedTags := []tag.Tag{\n\t\ttag.WorkflowDomainName(\"test-domain\"),\n\t\ttag.WorkflowTaskListName(\"test-task-list\"),\n\t\ttag.WorkflowTaskListType(int(taskListType)),\n\t\ttag.WorkflowTaskListKind(int32(types.TaskListKindNormal)),\n\t}\n\n\tassert.ElementsMatch(t, expectedTags, tags)\n}\n\nfunc TestHandleErr(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\terr             error\n\t\texpectedErrType interface{}\n\t\texpectedErrMsg  string\n\t\texpectedCounter string\n\t}{\n\t\t{\n\t\t\tname:            \"BadRequestError\",\n\t\t\terr:             &types.BadRequestError{Message: \"bad request\"},\n\t\t\texpectedErrType: &types.BadRequestError{},\n\t\t\texpectedErrMsg:  \"bad request\",\n\t\t\texpectedCounter: \"test.cadence_err_bad_request_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"DomainNotActiveError\",\n\t\t\terr:             &types.DomainNotActiveError{Message: \"domain not active\"},\n\t\t\texpectedErrType: &types.DomainNotActiveError{},\n\t\t\texpectedErrMsg:  \"domain not active\",\n\t\t\texpectedCounter: \"test.cadence_err_bad_request_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"ServiceBusyError\",\n\t\t\terr:             &types.ServiceBusyError{Message: \"service busy\"},\n\t\t\texpectedErrType: &types.ServiceBusyError{},\n\t\t\texpectedErrMsg:  \"service busy\",\n\t\t\texpectedCounter: \"test.cadence_err_service_busy_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"EntityNotExistsError\",\n\t\t\terr:             &types.EntityNotExistsError{Message: \"entity not exists\"},\n\t\t\texpectedErrType: &types.EntityNotExistsError{},\n\t\t\texpectedErrMsg:  \"entity not exists\",\n\t\t\texpectedCounter: \"test.cadence_err_entity_not_exists_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"WorkflowExecutionAlreadyCompletedError\",\n\t\t\terr:             &types.WorkflowExecutionAlreadyCompletedError{Message: \"workflow execution already completed\"},\n\t\t\texpectedErrType: &types.WorkflowExecutionAlreadyCompletedError{},\n\t\t\texpectedErrMsg:  \"workflow execution already completed\",\n\t\t\texpectedCounter: \"test.cadence_err_workflow_execution_already_completed_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"WorkflowExecutionAlreadyStartedError\",\n\t\t\terr:             &types.WorkflowExecutionAlreadyStartedError{Message: \"workflow execution already started\"},\n\t\t\texpectedErrType: &types.WorkflowExecutionAlreadyStartedError{},\n\t\t\texpectedErrMsg:  \"workflow execution already started\",\n\t\t\texpectedCounter: \"test.cadence_err_execution_already_started_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"DomainAlreadyExistsError\",\n\t\t\terr:             &types.DomainAlreadyExistsError{Message: \"domain already exists\"},\n\t\t\texpectedErrType: &types.DomainAlreadyExistsError{},\n\t\t\texpectedErrMsg:  \"domain already exists\",\n\t\t\texpectedCounter: \"test.cadence_err_domain_already_exists_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"CancellationAlreadyRequestedError\",\n\t\t\terr:             &types.CancellationAlreadyRequestedError{Message: \"cancellation already requested\"},\n\t\t\texpectedErrType: &types.CancellationAlreadyRequestedError{},\n\t\t\texpectedErrMsg:  \"cancellation already requested\",\n\t\t\texpectedCounter: \"test.cadence_err_cancellation_already_requested_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"QueryFailedError\",\n\t\t\terr:             &types.QueryFailedError{Message: \"query failed\"},\n\t\t\texpectedErrType: &types.QueryFailedError{},\n\t\t\texpectedErrMsg:  \"query failed\",\n\t\t\texpectedCounter: \"test.cadence_err_query_failed_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"LimitExceededError\",\n\t\t\terr:             &types.LimitExceededError{Message: \"limit exceeded\"},\n\t\t\texpectedErrType: &types.LimitExceededError{},\n\t\t\texpectedErrMsg:  \"limit exceeded\",\n\t\t\texpectedCounter: \"test.cadence_err_limit_exceeded_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"ClientVersionNotSupportedError\",\n\t\t\terr:             &types.ClientVersionNotSupportedError{},\n\t\t\texpectedErrType: &types.ClientVersionNotSupportedError{},\n\t\t\texpectedErrMsg:  \"client version not supported\",\n\t\t\texpectedCounter: \"test.cadence_err_client_version_not_supported_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"YARPCDeadlineExceededError\",\n\t\t\terr:             yarpcerrors.Newf(yarpcerrors.CodeDeadlineExceeded, \"deadline exceeded\"),\n\t\t\texpectedErrType: &yarpcerrors.Status{},\n\t\t\texpectedErrMsg:  \"deadline exceeded\",\n\t\t\texpectedCounter: \"test.cadence_err_context_timeout_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"ContextDeadlineExceeded\",\n\t\t\terr:             context.DeadlineExceeded,\n\t\t\texpectedErrType: context.DeadlineExceeded,\n\t\t\texpectedErrMsg:  \"context deadline exceeded\",\n\t\t\texpectedCounter: \"test.cadence_err_context_timeout_counter+\",\n\t\t},\n\t\t{\n\t\t\tname:            \"UncategorizedError\",\n\t\t\terr:             errors.New(\"unknown error\"),\n\t\t\texpectedErrType: errors.New(\"unknown error\"),\n\t\t\texpectedErrMsg:  \"unknown error\",\n\t\t\texpectedCounter: \"test.cadence_failures+\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.Frontend, metrics.HistogramMigration{})\n\t\t\thandler := &apiHandler{}\n\n\t\t\terr := handler.handleErr(tt.err, metricsClient.Scope(0), logger)\n\n\t\t\tassert.NotNil(t, err)\n\t\t\tassert.IsType(t, tt.expectedErrType, err)\n\t\t\tassert.Contains(t, err.Error(), tt.expectedErrMsg)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/ratelimited/api_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n)\n\n// apiHandler implements api.Handler interface instrumented with rate limiter.\ntype apiHandler struct {\n\twrapped               api.Handler\n\ttokenSerializer       common.TaskTokenSerializer\n\tdomainCache           cache.DomainCache\n\tuserRateLimiter       quotas.Policy\n\tworkerRateLimiter     quotas.Policy\n\tvisibilityRateLimiter quotas.Policy\n\tasyncRateLimiter      quotas.Policy\n\tmaxWorkerPollDelay    dynamicproperties.DurationPropertyFnWithDomainFilter\n\tcallerBypass          quotas.CallerBypass\n}\n\n// NewAPIHandler creates a new instance of Handler with ratelimiter.\nfunc NewAPIHandler(\n\twrapped api.Handler,\n\tdomainCache cache.DomainCache,\n\tuserRateLimiter quotas.Policy,\n\tworkerRateLimiter quotas.Policy,\n\tvisibilityRateLimiter quotas.Policy,\n\tasyncRateLimiter quotas.Policy,\n\tmaxWorkerPollDelay dynamicproperties.DurationPropertyFnWithDomainFilter,\n\tcallerBypass quotas.CallerBypass,\n) api.Handler {\n\treturn &apiHandler{\n\t\twrapped:               wrapped,\n\t\ttokenSerializer:       common.NewJSONTaskTokenSerializer(),\n\t\tdomainCache:           domainCache,\n\t\tuserRateLimiter:       userRateLimiter,\n\t\tworkerRateLimiter:     workerRateLimiter,\n\t\tvisibilityRateLimiter: visibilityRateLimiter,\n\t\tasyncRateLimiter:      asyncRateLimiter,\n\t\tmaxWorkerPollDelay:    maxWorkerPollDelay,\n\t\tcallerBypass:          callerBypass,\n\t}\n}\n\nfunc (h *apiHandler) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\tif cp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif cp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, cp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.CountWorkflowExecutions(ctx, cp1)\n}\n\nfunc (h *apiHandler) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest) (err error) {\n\treturn h.wrapped.DeleteDomain(ctx, dp1)\n}\n\nfunc (h *apiHandler) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest) (err error) {\n\treturn h.wrapped.DeprecateDomain(ctx, dp1)\n}\n\nfunc (h *apiHandler) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest) (dp2 *types.DescribeDomainResponse, err error) {\n\treturn h.wrapped.DescribeDomain(ctx, dp1)\n}\n\nfunc (h *apiHandler) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest) (dp2 *types.DescribeTaskListResponse, err error) {\n\tif dp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif dp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, dp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.DescribeTaskList(ctx, dp1)\n}\n\nfunc (h *apiHandler) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\tif dp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif dp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, dp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.DescribeWorkflowExecution(ctx, dp1)\n}\n\nfunc (h *apiHandler) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\tif dp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif dp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, dp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.DiagnoseWorkflowExecution(ctx, dp1)\n}\n\nfunc (h *apiHandler) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest) (fp2 *types.FailoverDomainResponse, err error) {\n\tif fp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif fp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, fp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.FailoverDomain(ctx, fp1)\n}\n\nfunc (h *apiHandler) GetClusterInfo(ctx context.Context) (cp1 *types.ClusterInfo, err error) {\n\treturn h.wrapped.GetClusterInfo(ctx)\n}\n\nfunc (h *apiHandler) GetSearchAttributes(ctx context.Context) (gp1 *types.GetSearchAttributesResponse, err error) {\n\treturn h.wrapped.GetSearchAttributes(ctx)\n}\n\nfunc (h *apiHandler) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\tif gp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif gp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, gp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.GetTaskListsByDomain(ctx, gp1)\n}\n\nfunc (h *apiHandler) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\tif gp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif gp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, gp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.GetWorkflowExecutionHistory(ctx, gp1)\n}\n\nfunc (h *apiHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn h.wrapped.Health(ctx)\n}\n\nfunc (h *apiHandler) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ListArchivedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ListClosedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest) (lp2 *types.ListDomainsResponse, err error) {\n\treturn h.wrapped.ListDomains(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\treturn h.wrapped.ListFailoverHistory(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ListOpenWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ListTaskListPartitions(ctx, lp1)\n}\n\nfunc (h *apiHandler) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ListWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *apiHandler) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest) (pp2 *types.PollForActivityTaskResponse, err error) {\n\tif pp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif pp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeWorkerPoll, pp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.PollForActivityTask(ctx, pp1)\n}\n\nfunc (h *apiHandler) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\tif pp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif pp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeWorkerPoll, pp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.PollForDecisionTask(ctx, pp1)\n}\n\nfunc (h *apiHandler) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest) (qp2 *types.QueryWorkflowResponse, err error) {\n\tif qp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif qp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, qp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.QueryWorkflow(ctx, qp1)\n}\n\nfunc (h *apiHandler) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RecordActivityTaskHeartbeat(ctx, rp1)\n}\n\nfunc (h *apiHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, rp1.GetDomain())\n\treturn h.wrapped.RecordActivityTaskHeartbeatByID(ctx, rp1)\n}\n\nfunc (h *apiHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, rp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.RefreshWorkflowTasks(ctx, rp1)\n}\n\nfunc (h *apiHandler) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest) (err error) {\n\treturn h.wrapped.RegisterDomain(ctx, rp1)\n}\n\nfunc (h *apiHandler) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, rp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.RequestCancelWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *apiHandler) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, rp1.GetDomain())\n\treturn h.wrapped.ResetStickyTaskList(ctx, rp1)\n}\n\nfunc (h *apiHandler) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, rp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ResetWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondActivityTaskCanceled(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, rp1.GetDomain())\n\treturn h.wrapped.RespondActivityTaskCanceledByID(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondActivityTaskCompleted(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, rp1.GetDomain())\n\treturn h.wrapped.RespondActivityTaskCompletedByID(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondActivityTaskFailed(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, rp1.GetDomain())\n\treturn h.wrapped.RespondActivityTaskFailedByID(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondDecisionTaskCompleted(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.Deserialize(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondDecisionTaskFailed(ctx, rp1)\n}\n\nfunc (h *apiHandler) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest) (err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.TaskToken == nil {\n\t\terr = validate.ErrTaskTokenNotSet\n\t\treturn\n\t}\n\ttoken, err := h.tokenSerializer.DeserializeQueryTaskToken(rp1.TaskToken)\n\tif err != nil {\n\t\treturn\n\t}\n\tif token.DomainID == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tdomainName, err := h.domainCache.GetDomainName(token.DomainID)\n\tif err != nil {\n\t\treturn\n\t}\n\t// Count the request in the host RPS,\n\t// but we still accept it even if RPS is exceeded\n\th.allowDomain(ctx, ratelimitTypeWorker, domainName)\n\treturn h.wrapped.RespondQueryTaskCompleted(ctx, rp1)\n}\n\nfunc (h *apiHandler) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\tif rp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif rp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, rp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.RestartWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *apiHandler) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\tif lp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif lp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeVisibility, lp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.ScanWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *apiHandler) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tif sp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif sp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, sp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.SignalWithStartWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *apiHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tif sp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif sp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeAsync, sp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.SignalWithStartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (h *apiHandler) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest) (err error) {\n\tif sp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif sp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, sp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.SignalWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *apiHandler) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\tif sp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif sp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, sp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.StartWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *apiHandler) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\tif sp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif sp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeAsync, sp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.StartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (h *apiHandler) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest) (err error) {\n\tif tp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\tif tp1.GetDomain() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\tif limitErr := h.allowDomain(ctx, ratelimitTypeUser, tp1.GetDomain()); limitErr != nil {\n\t\terr = limitErr\n\t\treturn\n\t}\n\treturn h.wrapped.TerminateWorkflowExecution(ctx, tp1)\n}\n\nfunc (h *apiHandler) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest) (up2 *types.UpdateDomainResponse, err error) {\n\treturn h.wrapped.UpdateDomain(ctx, up1)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/ratelimited/ratelimit.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// ratelimitType differentiates between the three categories of ratelimiters\ntype ratelimitType int\n\nconst (\n\tratelimitTypeUser ratelimitType = iota + 1\n\tratelimitTypeWorker\n\tratelimitTypeWorkerPoll\n\tratelimitTypeVisibility\n\tratelimitTypeAsync\n)\n\nvar errRateLimited = types.ServiceBusyError{Message: \"Too many outstanding requests to the cadence service\"}\n\nfunc newErrRateLimited() error {\n\treturn &errRateLimited\n}\n\nfunc (h *apiHandler) allowDomain(ctx context.Context, requestType ratelimitType, domain string) error {\n\tvar policy quotas.Policy\n\tswitch requestType {\n\tcase ratelimitTypeUser:\n\t\tpolicy = h.userRateLimiter\n\tcase ratelimitTypeWorker, ratelimitTypeWorkerPoll:\n\t\tpolicy = h.workerRateLimiter\n\tcase ratelimitTypeVisibility:\n\t\tpolicy = h.visibilityRateLimiter\n\tcase ratelimitTypeAsync:\n\t\tpolicy = h.asyncRateLimiter\n\tdefault:\n\t\tpanic(\"coding error, unrecognized request ratelimit type value\")\n\t}\n\n\t// If it is a poll request and there is a maxWorkerPollDelay configured, use Wait()\n\t// to potentially wait for a token. Otherwise, use Allow() for an immediate check\n\tif requestType == ratelimitTypeWorkerPoll {\n\t\twaitTime := h.maxWorkerPollDelay(domain)\n\t\tif waitTime > 0 {\n\t\t\treturn h.waitForPolicy(ctx, waitTime, policy, domain)\n\t\t}\n\t}\n\tif !h.callerBypass.AllowPolicy(ctx, policy, quotas.Info{Domain: domain}) {\n\t\treturn newErrRateLimited()\n\t}\n\treturn nil\n}\n\nfunc (h *apiHandler) waitForPolicy(ctx context.Context, waitTime time.Duration, policy quotas.Policy, domain string) error {\n\twaitCtx, cancel := context.WithTimeout(ctx, waitTime)\n\tdefer cancel()\n\terr := policy.Wait(waitCtx, quotas.Info{Domain: domain})\n\tif err != nil {\n\t\tif ctx.Err() != nil {\n\t\t\treturn ctx.Err()\n\t\t}\n\t\twaitCtxErr := waitCtx.Err()\n\t\tswitch waitCtxErr {\n\t\tcase nil:\n\t\t\tif h.callerBypass.ShouldBypass(ctx) {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn newErrRateLimited()\n\t\tcase context.DeadlineExceeded:\n\t\t\t// Race condition: context deadline hit right around wait completion\n\t\t\tif !h.callerBypass.AllowPolicy(ctx, policy, quotas.Info{Domain: domain}) {\n\t\t\t\treturn newErrRateLimited()\n\t\t\t}\n\t\t\treturn nil\n\t\tdefault:\n\t\t\treturn waitCtxErr\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/frontend/wrappers/ratelimited/ratelimit_test.go",
    "content": "package ratelimited\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n)\n\nconst testDomain = \"test-domain\"\n\nfunc setupHandler(t *testing.T) *apiHandler {\n\treturn setupHandlerWithDC(t, nil)\n}\n\nfunc setupHandlerWithDC(t *testing.T, dc *dynamicconfig.Collection) *apiHandler {\n\tctrl := gomock.NewController(t)\n\tmockHandler := api.NewMockHandler(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\tvar callerBypass quotas.CallerBypass\n\tif dc != nil {\n\t\tcallerBypass = quotas.NewCallerBypass(dc.GetListProperty(dynamicproperties.RateLimiterBypassCallerTypes))\n\t} else {\n\t\tcallerBypass = quotas.NewCallerBypass(nil)\n\t}\n\n\treturn NewAPIHandler(\n\t\tmockHandler,\n\t\tmockDomainCache,\n\t\t&mockPolicy{}, // userRateLimiter\n\t\t&mockPolicy{}, // workerRateLimiter\n\t\t&mockPolicy{}, // visibilityRateLimiter\n\t\t&mockPolicy{}, // asyncRateLimiter\n\t\tfunc(domain string) time.Duration { return 0 }, // maxWorkerPollDelay\n\t\tcallerBypass,\n\t).(*apiHandler)\n}\n\nfunc TestAllowDomainRequestRouting(t *testing.T) {\n\tcases := []struct {\n\t\tname        string\n\t\trequestType ratelimitType\n\t\tsetupMock   func(*apiHandler)\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tname:        \"User request type uses user limiter with Allow\",\n\t\t\trequestType: ratelimitTypeUser,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.userRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"Worker request type uses worker limiter with Allow\",\n\t\t\trequestType: ratelimitTypeWorker,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.workerRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"Visibility request type uses visibility limiter with Allow\",\n\t\t\trequestType: ratelimitTypeVisibility,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.visibilityRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"WorkerPoll request with zero delay uses worker limiter with Allow\",\n\t\t\trequestType: ratelimitTypeWorkerPoll,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.maxWorkerPollDelay = func(domain string) time.Duration { return 0 }\n\t\t\t\th.workerRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"WorkerPoll request with delay uses worker limiter with Wait\",\n\t\t\trequestType: ratelimitTypeWorkerPoll,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.maxWorkerPollDelay = func(domain string) time.Duration { return 5 * time.Second }\n\t\t\t\th.workerRateLimiter.(*mockPolicy).On(\"Wait\", mock.Anything, quotas.Info{Domain: testDomain}).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"Allow blocking returns rate limit error\",\n\t\t\trequestType: ratelimitTypeUser,\n\t\t\tsetupMock: func(h *apiHandler) {\n\t\t\t\th.userRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(false).Once()\n\t\t\t},\n\t\t\texpectedErr: newErrRateLimited(),\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thandler := setupHandler(t)\n\t\t\ttc.setupMock(handler)\n\n\t\t\terr := handler.allowDomain(context.Background(), tc.requestType, testDomain)\n\n\t\t\tif tc.expectedErr != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.expectedErr != nil {\n\t\t\t\t\tassert.Equal(t, tc.expectedErr, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\thandler.userRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t\thandler.workerRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t\thandler.visibilityRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t})\n\t}\n}\n\nfunc TestHandlerRpcRouting(t *testing.T) {\n\tcases := []struct {\n\t\tname         string\n\t\toperation    func(*apiHandler) (interface{}, error)\n\t\tlimiterSetup func(*apiHandler)\n\t}{\n\t\t{\n\t\t\tname: \"PollForActivityTask uses worker limiter with Allow when no delay is configured\",\n\t\t\toperation: func(h *apiHandler) (interface{}, error) {\n\t\t\t\treturn h.PollForActivityTask(context.Background(), &types.PollForActivityTaskRequest{Domain: testDomain})\n\t\t\t},\n\t\t\tlimiterSetup: func(h *apiHandler) {\n\t\t\t\th.maxWorkerPollDelay = func(domain string) time.Duration { return 0 }\n\t\t\t\th.workerRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t\th.wrapped.(*api.MockHandler).EXPECT().PollForActivityTask(gomock.Any(), gomock.Any()).Return(&types.PollForActivityTaskResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PollForActivityTask uses worker limiter with Wait when delay is configured\",\n\t\t\toperation: func(h *apiHandler) (interface{}, error) {\n\t\t\t\treturn h.PollForActivityTask(context.Background(), &types.PollForActivityTaskRequest{Domain: testDomain})\n\t\t\t},\n\t\t\tlimiterSetup: func(h *apiHandler) {\n\t\t\t\th.maxWorkerPollDelay = func(domain string) time.Duration { return 1 * time.Millisecond }\n\t\t\t\th.workerRateLimiter.(*mockPolicy).On(\"Wait\", mock.Anything, quotas.Info{Domain: testDomain}).Return(nil).Once()\n\t\t\t\th.wrapped.(*api.MockHandler).EXPECT().PollForActivityTask(gomock.Any(), gomock.Any()).Return(&types.PollForActivityTaskResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CountWorkflowExecutions uses visibility limiter with Allow\",\n\t\t\toperation: func(h *apiHandler) (interface{}, error) {\n\t\t\t\treturn h.CountWorkflowExecutions(context.Background(), &types.CountWorkflowExecutionsRequest{Domain: testDomain})\n\t\t\t},\n\t\t\tlimiterSetup: func(h *apiHandler) {\n\t\t\t\th.visibilityRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t\th.wrapped.(*api.MockHandler).EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeTaskList uses user limiter with Allow\",\n\t\t\toperation: func(h *apiHandler) (interface{}, error) {\n\t\t\t\treturn h.DescribeTaskList(context.Background(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:   testDomain,\n\t\t\t\t\tTaskList: &types.TaskList{Name: \"test-tasklist\"},\n\t\t\t\t})\n\t\t\t},\n\t\t\tlimiterSetup: func(h *apiHandler) {\n\t\t\t\th.userRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t\th.wrapped.(*api.MockHandler).EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&types.DescribeTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thandler := setupHandler(t)\n\t\t\ttc.limiterSetup(handler)\n\n\t\t\tresp, err := tc.operation(handler)\n\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.NotNil(t, resp)\n\t\t\thandler.userRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t\thandler.workerRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t\thandler.visibilityRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t})\n\t}\n}\n\ntype mockPolicy struct {\n\tmock.Mock\n}\n\nfunc (m *mockPolicy) Allow(info quotas.Info) bool {\n\targs := m.Called(info)\n\treturn args.Bool(0)\n}\n\nfunc (m *mockPolicy) Wait(ctx context.Context, info quotas.Info) error {\n\targs := m.Called(ctx, info)\n\treturn args.Error(0)\n}\n\nfunc TestCallerTypeBypass(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tbypassCallerTypes []interface{}\n\t\tcallerType        types.CallerType\n\t\trateLimitBlocks   bool\n\t\texpectBypass      bool\n\t}{\n\t\t{\n\t\t\tname:              \"Bypass CLI caller when configured\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\trateLimitBlocks:   true,\n\t\t\texpectBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Bypass UI caller when configured\",\n\t\t\tbypassCallerTypes: []interface{}{\"ui\"},\n\t\t\tcallerType:        types.CallerTypeUI,\n\t\t\trateLimitBlocks:   true,\n\t\t\texpectBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Don't bypass when caller type not in list\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeUI,\n\t\t\trateLimitBlocks:   true,\n\t\t\texpectBypass:      false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Don't bypass with empty list\",\n\t\t\tbypassCallerTypes: []interface{}{},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\trateLimitBlocks:   true,\n\t\t\texpectBypass:      false,\n\t\t},\n\t\t{\n\t\t\tname:              \"Multiple caller types in bypass list\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\", \"ui\"},\n\t\t\tcallerType:        types.CallerTypeUI,\n\t\t\trateLimitBlocks:   true,\n\t\t\texpectBypass:      true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\tclient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, tt.bypassCallerTypes)\n\t\t\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\n\t\t\thandler := setupHandlerWithDC(t, dc)\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\n\t\t\tif tt.rateLimitBlocks {\n\t\t\t\thandler.userRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(false).Once()\n\t\t\t} else {\n\t\t\t\thandler.userRateLimiter.(*mockPolicy).On(\"Allow\", quotas.Info{Domain: testDomain}).Return(true).Once()\n\t\t\t}\n\n\t\t\terr := handler.allowDomain(ctx, ratelimitTypeUser, testDomain)\n\n\t\t\tif tt.expectBypass {\n\t\t\t\tassert.NoError(t, err, \"Expected bypass to allow request\")\n\t\t\t} else {\n\t\t\t\tif tt.rateLimitBlocks {\n\t\t\t\t\tassert.Error(t, err, \"Expected rate limit error\")\n\t\t\t\t\tassert.Equal(t, newErrRateLimited(), err)\n\t\t\t\t} else {\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\thandler.userRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t})\n\t}\n}\n\nfunc TestCallerTypeBypassWithWait(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tbypassCallerTypes []interface{}\n\t\tcallerType        types.CallerType\n\t\texpectBypass      bool\n\t}{\n\t\t{\n\t\t\tname:              \"Bypass in Wait nil error path when configured\",\n\t\t\tbypassCallerTypes: []interface{}{\"cli\"},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpectBypass:      true,\n\t\t},\n\t\t{\n\t\t\tname:              \"Don't bypass in Wait nil error path when not configured\",\n\t\t\tbypassCallerTypes: []interface{}{},\n\t\t\tcallerType:        types.CallerTypeCLI,\n\t\t\texpectBypass:      false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\tclient.UpdateValue(dynamicproperties.RateLimiterBypassCallerTypes, tt.bypassCallerTypes)\n\t\t\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\n\t\t\thandler := setupHandlerWithDC(t, dc)\n\t\t\thandler.maxWorkerPollDelay = func(domain string) time.Duration { return 1 * time.Millisecond }\n\t\t\tctx := types.ContextWithCallerInfo(context.Background(), types.NewCallerInfo(tt.callerType))\n\n\t\t\t// Wait returns an error, but waitCtx.Err() is nil (case nil path)\n\t\t\thandler.workerRateLimiter.(*mockPolicy).On(\"Wait\", mock.Anything, quotas.Info{Domain: testDomain}).Return(assert.AnError).Once()\n\n\t\t\terr := handler.allowDomain(ctx, ratelimitTypeWorkerPoll, testDomain)\n\n\t\t\tif tt.expectBypass {\n\t\t\t\tassert.NoError(t, err, \"Expected bypass to allow request\")\n\t\t\t} else {\n\t\t\t\tassert.Error(t, err, \"Expected rate limit error\")\n\t\t\t\tassert.Equal(t, newErrRateLimited(), err)\n\t\t\t}\n\n\t\t\thandler.workerRateLimiter.(*mockPolicy).AssertExpectations(t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/frontend/wrappers/thrift/admin_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/.gen/go/admin\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g AdminHandler) AddSearchAttribute(ctx context.Context, Request *admin.AddSearchAttributeRequest) (err error) {\n\terr = g.h.AddSearchAttribute(ctx, thrift.ToAdminAddSearchAttributeRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) CloseShard(ctx context.Context, Request *shared.CloseShardRequest) (err error) {\n\terr = g.h.CloseShard(ctx, thrift.ToAdminCloseShardRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DeleteWorkflow(ctx context.Context, Request *admin.AdminDeleteWorkflowRequest) (ap1 *admin.AdminDeleteWorkflowResponse, err error) {\n\tresponse, err := g.h.DeleteWorkflow(ctx, thrift.ToAdminDeleteWorkflowRequest(Request))\n\treturn thrift.FromAdminDeleteWorkflowResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeCluster(ctx context.Context) (dp1 *admin.DescribeClusterResponse, err error) {\n\tresponse, err := g.h.DescribeCluster(ctx)\n\treturn thrift.FromAdminDescribeClusterResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeHistoryHost(ctx context.Context, Request *shared.DescribeHistoryHostRequest) (dp1 *shared.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.h.DescribeHistoryHost(ctx, thrift.ToAdminDescribeHistoryHostRequest(Request))\n\treturn thrift.FromAdminDescribeHistoryHostResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeQueue(ctx context.Context, Request *shared.DescribeQueueRequest) (dp1 *shared.DescribeQueueResponse, err error) {\n\tresponse, err := g.h.DescribeQueue(ctx, thrift.ToAdminDescribeQueueRequest(Request))\n\treturn thrift.FromAdminDescribeQueueResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeShardDistribution(ctx context.Context, Request *shared.DescribeShardDistributionRequest) (dp1 *shared.DescribeShardDistributionResponse, err error) {\n\tresponse, err := g.h.DescribeShardDistribution(ctx, thrift.ToAdminDescribeShardDistributionRequest(Request))\n\treturn thrift.FromAdminDescribeShardDistributionResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) DescribeWorkflowExecution(ctx context.Context, Request *admin.DescribeWorkflowExecutionRequest) (dp1 *admin.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, thrift.ToAdminDescribeWorkflowExecutionRequest(Request))\n\treturn thrift.FromAdminDescribeWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetCrossClusterTasks(ctx context.Context, Request *shared.GetCrossClusterTasksRequest) (gp1 *shared.GetCrossClusterTasksResponse, err error) {\n\tresponse, err := g.h.GetCrossClusterTasks(ctx, thrift.ToAdminGetCrossClusterTasksRequest(Request))\n\treturn thrift.FromAdminGetCrossClusterTasksResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetDLQReplicationMessages(ctx context.Context, Request *replicator.GetDLQReplicationMessagesRequest) (gp1 *replicator.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.h.GetDLQReplicationMessages(ctx, thrift.ToAdminGetDLQReplicationMessagesRequest(Request))\n\treturn thrift.FromAdminGetDLQReplicationMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainAsyncWorkflowConfiguraton(ctx context.Context, Request *admin.GetDomainAsyncWorkflowConfiguratonRequest) (gp1 *admin.GetDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.h.GetDomainAsyncWorkflowConfiguraton(ctx, thrift.ToAdminGetDomainAsyncWorkflowConfiguratonRequest(Request))\n\treturn thrift.FromAdminGetDomainAsyncWorkflowConfiguratonResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainIsolationGroups(ctx context.Context, Request *admin.GetDomainIsolationGroupsRequest) (gp1 *admin.GetDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.h.GetDomainIsolationGroups(ctx, thrift.ToAdminGetDomainIsolationGroupsRequest(Request))\n\treturn thrift.FromAdminGetDomainIsolationGroupsResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetDomainReplicationMessages(ctx context.Context, Request *replicator.GetDomainReplicationMessagesRequest) (gp1 *replicator.GetDomainReplicationMessagesResponse, err error) {\n\tresponse, err := g.h.GetDomainReplicationMessages(ctx, thrift.ToAdminGetDomainReplicationMessagesRequest(Request))\n\treturn thrift.FromAdminGetDomainReplicationMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetDynamicConfig(ctx context.Context, Request *admin.GetDynamicConfigRequest) (gp1 *admin.GetDynamicConfigResponse, err error) {\n\tresponse, err := g.h.GetDynamicConfig(ctx, thrift.ToAdminGetDynamicConfigRequest(Request))\n\treturn thrift.FromAdminGetDynamicConfigResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetGlobalIsolationGroups(ctx context.Context, Request *admin.GetGlobalIsolationGroupsRequest) (gp1 *admin.GetGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.h.GetGlobalIsolationGroups(ctx, thrift.ToAdminGetGlobalIsolationGroupsRequest(Request))\n\treturn thrift.FromAdminGetGlobalIsolationGroupsResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetReplicationMessages(ctx context.Context, Request *replicator.GetReplicationMessagesRequest) (gp1 *replicator.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.h.GetReplicationMessages(ctx, thrift.ToAdminGetReplicationMessagesRequest(Request))\n\treturn thrift.FromAdminGetReplicationMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) GetWorkflowExecutionRawHistoryV2(ctx context.Context, GetRequest *admin.GetWorkflowExecutionRawHistoryV2Request) (gp1 *admin.GetWorkflowExecutionRawHistoryV2Response, err error) {\n\tresponse, err := g.h.GetWorkflowExecutionRawHistoryV2(ctx, thrift.ToAdminGetWorkflowExecutionRawHistoryV2Request(GetRequest))\n\treturn thrift.FromAdminGetWorkflowExecutionRawHistoryV2Response(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) ListDynamicConfig(ctx context.Context, Request *admin.ListDynamicConfigRequest) (lp1 *admin.ListDynamicConfigResponse, err error) {\n\tresponse, err := g.h.ListDynamicConfig(ctx, thrift.ToAdminListDynamicConfigRequest(Request))\n\treturn thrift.FromAdminListDynamicConfigResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) MaintainCorruptWorkflow(ctx context.Context, Request *admin.AdminMaintainWorkflowRequest) (ap1 *admin.AdminMaintainWorkflowResponse, err error) {\n\tresponse, err := g.h.MaintainCorruptWorkflow(ctx, thrift.ToAdminMaintainCorruptWorkflowRequest(Request))\n\treturn thrift.FromAdminMaintainCorruptWorkflowResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) MergeDLQMessages(ctx context.Context, Request *replicator.MergeDLQMessagesRequest) (mp1 *replicator.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.h.MergeDLQMessages(ctx, thrift.ToAdminMergeDLQMessagesRequest(Request))\n\treturn thrift.FromAdminMergeDLQMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) PurgeDLQMessages(ctx context.Context, Request *replicator.PurgeDLQMessagesRequest) (err error) {\n\terr = g.h.PurgeDLQMessages(ctx, thrift.ToAdminPurgeDLQMessagesRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) ReadDLQMessages(ctx context.Context, Request *replicator.ReadDLQMessagesRequest) (rp1 *replicator.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.h.ReadDLQMessages(ctx, thrift.ToAdminReadDLQMessagesRequest(Request))\n\treturn thrift.FromAdminReadDLQMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) ReapplyEvents(ctx context.Context, ReapplyEventsRequest *shared.ReapplyEventsRequest) (err error) {\n\terr = g.h.ReapplyEvents(ctx, thrift.ToAdminReapplyEventsRequest(ReapplyEventsRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) RefreshWorkflowTasks(ctx context.Context, Request *shared.RefreshWorkflowTasksRequest) (err error) {\n\terr = g.h.RefreshWorkflowTasks(ctx, thrift.ToAdminRefreshWorkflowTasksRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) RemoveTask(ctx context.Context, Request *shared.RemoveTaskRequest) (err error) {\n\terr = g.h.RemoveTask(ctx, thrift.ToAdminRemoveTaskRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) ResendReplicationTasks(ctx context.Context, Request *admin.ResendReplicationTasksRequest) (err error) {\n\terr = g.h.ResendReplicationTasks(ctx, thrift.ToAdminResendReplicationTasksRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) ResetQueue(ctx context.Context, Request *shared.ResetQueueRequest) (err error) {\n\terr = g.h.ResetQueue(ctx, thrift.ToAdminResetQueueRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) RespondCrossClusterTasksCompleted(ctx context.Context, Request *shared.RespondCrossClusterTasksCompletedRequest) (rp1 *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\tresponse, err := g.h.RespondCrossClusterTasksCompleted(ctx, thrift.ToAdminRespondCrossClusterTasksCompletedRequest(Request))\n\treturn thrift.FromAdminRespondCrossClusterTasksCompletedResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) RestoreDynamicConfig(ctx context.Context, Request *admin.RestoreDynamicConfigRequest) (err error) {\n\terr = g.h.RestoreDynamicConfig(ctx, thrift.ToAdminRestoreDynamicConfigRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDomainAsyncWorkflowConfiguraton(ctx context.Context, Request *admin.UpdateDomainAsyncWorkflowConfiguratonRequest) (up1 *admin.UpdateDomainAsyncWorkflowConfiguratonResponse, err error) {\n\tresponse, err := g.h.UpdateDomainAsyncWorkflowConfiguraton(ctx, thrift.ToAdminUpdateDomainAsyncWorkflowConfiguratonRequest(Request))\n\treturn thrift.FromAdminUpdateDomainAsyncWorkflowConfiguratonResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDomainIsolationGroups(ctx context.Context, Request *admin.UpdateDomainIsolationGroupsRequest) (up1 *admin.UpdateDomainIsolationGroupsResponse, err error) {\n\tresponse, err := g.h.UpdateDomainIsolationGroups(ctx, thrift.ToAdminUpdateDomainIsolationGroupsRequest(Request))\n\treturn thrift.FromAdminUpdateDomainIsolationGroupsResponse(response), thrift.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateDynamicConfig(ctx context.Context, Request *admin.UpdateDynamicConfigRequest) (err error) {\n\terr = g.h.UpdateDynamicConfig(ctx, thrift.ToAdminUpdateDynamicConfigRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g AdminHandler) UpdateGlobalIsolationGroups(ctx context.Context, Request *admin.UpdateGlobalIsolationGroupsRequest) (up1 *admin.UpdateGlobalIsolationGroupsResponse, err error) {\n\tresponse, err := g.h.UpdateGlobalIsolationGroups(ctx, thrift.ToAdminUpdateGlobalIsolationGroupsRequest(Request))\n\treturn thrift.FromAdminUpdateGlobalIsolationGroupsResponse(response), thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/thrift/admin_handler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/admin\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\tadminHandler \"github.com/uber/cadence/service/frontend/admin\"\n)\n\nfunc TestAdminThriftHandler(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\th := adminHandler.NewMockHandler(ctrl)\n\tth := NewAdminHandler(h)\n\tctx := context.Background()\n\tinternalErr := &types.InternalServiceError{Message: \"test\"}\n\texpectedErr := &shared.InternalServiceError{Message: \"test\"}\n\n\tt.Run(\"AddSearchAttribute\", func(t *testing.T) {\n\t\th.EXPECT().AddSearchAttribute(ctx, &types.AddSearchAttributeRequest{}).Return(internalErr).Times(1)\n\t\terr := th.AddSearchAttribute(ctx, &admin.AddSearchAttributeRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"CloseShard\", func(t *testing.T) {\n\t\th.EXPECT().CloseShard(ctx, &types.CloseShardRequest{}).Return(internalErr).Times(1)\n\t\terr := th.CloseShard(ctx, &shared.CloseShardRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeCluster\", func(t *testing.T) {\n\t\th.EXPECT().DescribeCluster(ctx).Return(&types.DescribeClusterResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeCluster(ctx)\n\t\tassert.Equal(t, admin.DescribeClusterResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeHistoryHost\", func(t *testing.T) {\n\t\th.EXPECT().DescribeHistoryHost(ctx, &types.DescribeHistoryHostRequest{}).Return(&types.DescribeHistoryHostResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeHistoryHost(ctx, &shared.DescribeHistoryHostRequest{})\n\t\tassert.Equal(t, shared.DescribeHistoryHostResponse{NumberOfShards: common.Int32Ptr(0), ShardControllerStatus: common.StringPtr(\"\"), Address: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeQueue\", func(t *testing.T) {\n\t\th.EXPECT().DescribeQueue(ctx, &types.DescribeQueueRequest{}).Return(&types.DescribeQueueResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeQueue(ctx, &shared.DescribeQueueRequest{})\n\t\tassert.Equal(t, shared.DescribeQueueResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().DescribeWorkflowExecution(ctx, &types.AdminDescribeWorkflowExecutionRequest{}).Return(nil, internalErr).Times(1)\n\t\tresp, err := th.DescribeWorkflowExecution(ctx, &admin.DescribeWorkflowExecutionRequest{})\n\t\tassert.Nil(t, resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetDLQReplicationMessages\", func(t *testing.T) {\n\t\th.EXPECT().GetDLQReplicationMessages(ctx, &types.GetDLQReplicationMessagesRequest{}).Return(&types.GetDLQReplicationMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetDLQReplicationMessages(ctx, &replicator.GetDLQReplicationMessagesRequest{})\n\t\tassert.Equal(t, replicator.GetDLQReplicationMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetDomainReplicationMessages\", func(t *testing.T) {\n\t\th.EXPECT().GetDomainReplicationMessages(ctx, &types.GetDomainReplicationMessagesRequest{}).Return(&types.GetDomainReplicationMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetDomainReplicationMessages(ctx, &replicator.GetDomainReplicationMessagesRequest{})\n\t\tassert.Equal(t, replicator.GetDomainReplicationMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetReplicationMessages\", func(t *testing.T) {\n\t\th.EXPECT().GetReplicationMessages(ctx, &types.GetReplicationMessagesRequest{}).Return(&types.GetReplicationMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetReplicationMessages(ctx, &replicator.GetReplicationMessagesRequest{})\n\t\tassert.Equal(t, replicator.GetReplicationMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetWorkflowExecutionRawHistoryV2\", func(t *testing.T) {\n\t\th.EXPECT().GetWorkflowExecutionRawHistoryV2(ctx, &types.GetWorkflowExecutionRawHistoryV2Request{}).Return(&types.GetWorkflowExecutionRawHistoryV2Response{}, internalErr).Times(1)\n\t\tresp, err := th.GetWorkflowExecutionRawHistoryV2(ctx, &admin.GetWorkflowExecutionRawHistoryV2Request{})\n\t\tassert.Equal(t, admin.GetWorkflowExecutionRawHistoryV2Response{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"MergeDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().MergeDLQMessages(ctx, &types.MergeDLQMessagesRequest{}).Return(&types.MergeDLQMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.MergeDLQMessages(ctx, &replicator.MergeDLQMessagesRequest{})\n\t\tassert.Equal(t, replicator.MergeDLQMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PurgeDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().PurgeDLQMessages(ctx, &types.PurgeDLQMessagesRequest{}).Return(internalErr).Times(1)\n\t\terr := th.PurgeDLQMessages(ctx, &replicator.PurgeDLQMessagesRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ReadDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().ReadDLQMessages(ctx, &types.ReadDLQMessagesRequest{}).Return(&types.ReadDLQMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ReadDLQMessages(ctx, &replicator.ReadDLQMessagesRequest{})\n\t\tassert.Equal(t, replicator.ReadDLQMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ReapplyEvents\", func(t *testing.T) {\n\t\th.EXPECT().ReapplyEvents(ctx, &types.ReapplyEventsRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ReapplyEvents(ctx, &shared.ReapplyEventsRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RefreshWorkflowTasks\", func(t *testing.T) {\n\t\th.EXPECT().RefreshWorkflowTasks(ctx, &types.RefreshWorkflowTasksRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RefreshWorkflowTasks(ctx, &shared.RefreshWorkflowTasksRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RemoveTask\", func(t *testing.T) {\n\t\th.EXPECT().RemoveTask(ctx, &types.RemoveTaskRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RemoveTask(ctx, &shared.RemoveTaskRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResendReplicationTasks\", func(t *testing.T) {\n\t\th.EXPECT().ResendReplicationTasks(ctx, &types.ResendReplicationTasksRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ResendReplicationTasks(ctx, &admin.ResendReplicationTasksRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetQueue\", func(t *testing.T) {\n\t\th.EXPECT().ResetQueue(ctx, &types.ResetQueueRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ResetQueue(ctx, &shared.ResetQueueRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetCrossClusterTasks\", func(t *testing.T) {\n\t\th.EXPECT().GetCrossClusterTasks(ctx, &types.GetCrossClusterTasksRequest{}).Return(&types.GetCrossClusterTasksResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetCrossClusterTasks(ctx, &shared.GetCrossClusterTasksRequest{})\n\t\tassert.Equal(t, shared.GetCrossClusterTasksResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n}\n"
  },
  {
    "path": "service/frontend/wrappers/thrift/api_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g APIHandler) CountWorkflowExecutions(ctx context.Context, CountRequest *shared.CountWorkflowExecutionsRequest) (cp1 *shared.CountWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.CountWorkflowExecutions(ctx, thrift.ToCountWorkflowExecutionsRequest(CountRequest))\n\treturn thrift.FromCountWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) DeleteDomain(ctx context.Context, DeleteRequest *shared.DeleteDomainRequest) (err error) {\n\terr = g.h.DeleteDomain(ctx, thrift.ToDeleteDomainRequest(DeleteRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) DeprecateDomain(ctx context.Context, DeprecateRequest *shared.DeprecateDomainRequest) (err error) {\n\terr = g.h.DeprecateDomain(ctx, thrift.ToDeprecateDomainRequest(DeprecateRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) DescribeDomain(ctx context.Context, DescribeRequest *shared.DescribeDomainRequest) (dp1 *shared.DescribeDomainResponse, err error) {\n\tresponse, err := g.h.DescribeDomain(ctx, thrift.ToDescribeDomainRequest(DescribeRequest))\n\treturn thrift.FromDescribeDomainResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) DescribeTaskList(ctx context.Context, Request *shared.DescribeTaskListRequest) (dp1 *shared.DescribeTaskListResponse, err error) {\n\tresponse, err := g.h.DescribeTaskList(ctx, thrift.ToDescribeTaskListRequest(Request))\n\treturn thrift.FromDescribeTaskListResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) DescribeWorkflowExecution(ctx context.Context, DescribeRequest *shared.DescribeWorkflowExecutionRequest) (dp1 *shared.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, thrift.ToDescribeWorkflowExecutionRequest(DescribeRequest))\n\treturn thrift.FromDescribeWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) DiagnoseWorkflowExecution(ctx context.Context, DiagnoseRequest *shared.DiagnoseWorkflowExecutionRequest) (dp1 *shared.DiagnoseWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.DiagnoseWorkflowExecution(ctx, thrift.ToDiagnoseWorkflowExecutionRequest(DiagnoseRequest))\n\treturn thrift.FromDiagnoseWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) FailoverDomain(ctx context.Context, FailoverRequest *shared.FailoverDomainRequest) (fp1 *shared.FailoverDomainResponse, err error) {\n\tresponse, err := g.h.FailoverDomain(ctx, thrift.ToFailoverDomainRequest(FailoverRequest))\n\treturn thrift.FromFailoverDomainResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) GetClusterInfo(ctx context.Context) (cp1 *shared.ClusterInfo, err error) {\n\tresponse, err := g.h.GetClusterInfo(ctx)\n\treturn thrift.FromGetClusterInfoResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) GetSearchAttributes(ctx context.Context) (gp1 *shared.GetSearchAttributesResponse, err error) {\n\tresponse, err := g.h.GetSearchAttributes(ctx)\n\treturn thrift.FromGetSearchAttributesResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) GetTaskListsByDomain(ctx context.Context, Request *shared.GetTaskListsByDomainRequest) (gp1 *shared.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.h.GetTaskListsByDomain(ctx, thrift.ToGetTaskListsByDomainRequest(Request))\n\treturn thrift.FromGetTaskListsByDomainResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) GetWorkflowExecutionHistory(ctx context.Context, GetRequest *shared.GetWorkflowExecutionHistoryRequest) (gp1 *shared.GetWorkflowExecutionHistoryResponse, err error) {\n\tresponse, err := g.h.GetWorkflowExecutionHistory(ctx, thrift.ToGetWorkflowExecutionHistoryRequest(GetRequest))\n\treturn thrift.FromGetWorkflowExecutionHistoryResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListArchivedWorkflowExecutions(ctx context.Context, ListRequest *shared.ListArchivedWorkflowExecutionsRequest) (lp1 *shared.ListArchivedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.ListArchivedWorkflowExecutions(ctx, thrift.ToListArchivedWorkflowExecutionsRequest(ListRequest))\n\treturn thrift.FromListArchivedWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListClosedWorkflowExecutions(ctx context.Context, ListRequest *shared.ListClosedWorkflowExecutionsRequest) (lp1 *shared.ListClosedWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.ListClosedWorkflowExecutions(ctx, thrift.ToListClosedWorkflowExecutionsRequest(ListRequest))\n\treturn thrift.FromListClosedWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListDomains(ctx context.Context, ListRequest *shared.ListDomainsRequest) (lp1 *shared.ListDomainsResponse, err error) {\n\tresponse, err := g.h.ListDomains(ctx, thrift.ToListDomainsRequest(ListRequest))\n\treturn thrift.FromListDomainsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListFailoverHistory(ctx context.Context, ListRequest *shared.ListFailoverHistoryRequest) (lp1 *shared.ListFailoverHistoryResponse, err error) {\n\tresponse, err := g.h.ListFailoverHistory(ctx, thrift.ToListFailoverHistoryRequest(ListRequest))\n\treturn thrift.FromListFailoverHistoryResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListOpenWorkflowExecutions(ctx context.Context, ListRequest *shared.ListOpenWorkflowExecutionsRequest) (lp1 *shared.ListOpenWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.ListOpenWorkflowExecutions(ctx, thrift.ToListOpenWorkflowExecutionsRequest(ListRequest))\n\treturn thrift.FromListOpenWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListTaskListPartitions(ctx context.Context, Request *shared.ListTaskListPartitionsRequest) (lp1 *shared.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.h.ListTaskListPartitions(ctx, thrift.ToListTaskListPartitionsRequest(Request))\n\treturn thrift.FromListTaskListPartitionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ListWorkflowExecutions(ctx context.Context, ListRequest *shared.ListWorkflowExecutionsRequest) (lp1 *shared.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.ListWorkflowExecutions(ctx, thrift.ToListWorkflowExecutionsRequest(ListRequest))\n\treturn thrift.FromListWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) PollForActivityTask(ctx context.Context, PollRequest *shared.PollForActivityTaskRequest) (pp1 *shared.PollForActivityTaskResponse, err error) {\n\tresponse, err := g.h.PollForActivityTask(ctx, thrift.ToPollForActivityTaskRequest(PollRequest))\n\treturn thrift.FromPollForActivityTaskResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) PollForDecisionTask(ctx context.Context, PollRequest *shared.PollForDecisionTaskRequest) (pp1 *shared.PollForDecisionTaskResponse, err error) {\n\tresponse, err := g.h.PollForDecisionTask(ctx, thrift.ToPollForDecisionTaskRequest(PollRequest))\n\treturn thrift.FromPollForDecisionTaskResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) QueryWorkflow(ctx context.Context, QueryRequest *shared.QueryWorkflowRequest) (qp1 *shared.QueryWorkflowResponse, err error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, thrift.ToQueryWorkflowRequest(QueryRequest))\n\treturn thrift.FromQueryWorkflowResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) RecordActivityTaskHeartbeat(ctx context.Context, HeartbeatRequest *shared.RecordActivityTaskHeartbeatRequest) (rp1 *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeat(ctx, thrift.ToRecordActivityTaskHeartbeatRequest(HeartbeatRequest))\n\treturn thrift.FromRecordActivityTaskHeartbeatResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, HeartbeatRequest *shared.RecordActivityTaskHeartbeatByIDRequest) (rp1 *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeatByID(ctx, thrift.ToRecordActivityTaskHeartbeatByIDRequest(HeartbeatRequest))\n\treturn thrift.FromRecordActivityTaskHeartbeatByIDResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) RefreshWorkflowTasks(ctx context.Context, Request *shared.RefreshWorkflowTasksRequest) (err error) {\n\terr = g.h.RefreshWorkflowTasks(ctx, thrift.ToRefreshWorkflowTasksRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RegisterDomain(ctx context.Context, RegisterRequest *shared.RegisterDomainRequest) (err error) {\n\terr = g.h.RegisterDomain(ctx, thrift.ToRegisterDomainRequest(RegisterRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RequestCancelWorkflowExecution(ctx context.Context, CancelRequest *shared.RequestCancelWorkflowExecutionRequest) (err error) {\n\terr = g.h.RequestCancelWorkflowExecution(ctx, thrift.ToRequestCancelWorkflowExecutionRequest(CancelRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) ResetStickyTaskList(ctx context.Context, ResetRequest *shared.ResetStickyTaskListRequest) (rp1 *shared.ResetStickyTaskListResponse, err error) {\n\tresponse, err := g.h.ResetStickyTaskList(ctx, thrift.ToResetStickyTaskListRequest(ResetRequest))\n\treturn thrift.FromResetStickyTaskListResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ResetWorkflowExecution(ctx context.Context, ResetRequest *shared.ResetWorkflowExecutionRequest) (rp1 *shared.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.ResetWorkflowExecution(ctx, thrift.ToResetWorkflowExecutionRequest(ResetRequest))\n\treturn thrift.FromResetWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCanceled(ctx context.Context, CanceledRequest *shared.RespondActivityTaskCanceledRequest) (err error) {\n\terr = g.h.RespondActivityTaskCanceled(ctx, thrift.ToRespondActivityTaskCanceledRequest(CanceledRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCanceledByID(ctx context.Context, CanceledRequest *shared.RespondActivityTaskCanceledByIDRequest) (err error) {\n\terr = g.h.RespondActivityTaskCanceledByID(ctx, thrift.ToRespondActivityTaskCanceledByIDRequest(CanceledRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCompleted(ctx context.Context, CompleteRequest *shared.RespondActivityTaskCompletedRequest) (err error) {\n\terr = g.h.RespondActivityTaskCompleted(ctx, thrift.ToRespondActivityTaskCompletedRequest(CompleteRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskCompletedByID(ctx context.Context, CompleteRequest *shared.RespondActivityTaskCompletedByIDRequest) (err error) {\n\terr = g.h.RespondActivityTaskCompletedByID(ctx, thrift.ToRespondActivityTaskCompletedByIDRequest(CompleteRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskFailed(ctx context.Context, FailRequest *shared.RespondActivityTaskFailedRequest) (err error) {\n\terr = g.h.RespondActivityTaskFailed(ctx, thrift.ToRespondActivityTaskFailedRequest(FailRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondActivityTaskFailedByID(ctx context.Context, FailRequest *shared.RespondActivityTaskFailedByIDRequest) (err error) {\n\terr = g.h.RespondActivityTaskFailedByID(ctx, thrift.ToRespondActivityTaskFailedByIDRequest(FailRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondDecisionTaskCompleted(ctx context.Context, CompleteRequest *shared.RespondDecisionTaskCompletedRequest) (rp1 *shared.RespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.h.RespondDecisionTaskCompleted(ctx, thrift.ToRespondDecisionTaskCompletedRequest(CompleteRequest))\n\treturn thrift.FromRespondDecisionTaskCompletedResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondDecisionTaskFailed(ctx context.Context, FailedRequest *shared.RespondDecisionTaskFailedRequest) (err error) {\n\terr = g.h.RespondDecisionTaskFailed(ctx, thrift.ToRespondDecisionTaskFailedRequest(FailedRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RespondQueryTaskCompleted(ctx context.Context, CompleteRequest *shared.RespondQueryTaskCompletedRequest) (err error) {\n\terr = g.h.RespondQueryTaskCompleted(ctx, thrift.ToRespondQueryTaskCompletedRequest(CompleteRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) RestartWorkflowExecution(ctx context.Context, RestartRequest *shared.RestartWorkflowExecutionRequest) (rp1 *shared.RestartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.RestartWorkflowExecution(ctx, thrift.ToRestartWorkflowExecutionRequest(RestartRequest))\n\treturn thrift.FromRestartWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) ScanWorkflowExecutions(ctx context.Context, ListRequest *shared.ListWorkflowExecutionsRequest) (lp1 *shared.ListWorkflowExecutionsResponse, err error) {\n\tresponse, err := g.h.ScanWorkflowExecutions(ctx, thrift.ToScanWorkflowExecutionsRequest(ListRequest))\n\treturn thrift.FromScanWorkflowExecutionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) SignalWithStartWorkflowExecution(ctx context.Context, SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionRequest) (sp1 *shared.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecution(ctx, thrift.ToSignalWithStartWorkflowExecutionRequest(SignalWithStartRequest))\n\treturn thrift.FromSignalWithStartWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, SignalWithStartRequest *shared.SignalWithStartWorkflowExecutionAsyncRequest) (sp1 *shared.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecutionAsync(ctx, thrift.ToSignalWithStartWorkflowExecutionAsyncRequest(SignalWithStartRequest))\n\treturn thrift.FromSignalWithStartWorkflowExecutionAsyncResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) SignalWorkflowExecution(ctx context.Context, SignalRequest *shared.SignalWorkflowExecutionRequest) (err error) {\n\terr = g.h.SignalWorkflowExecution(ctx, thrift.ToSignalWorkflowExecutionRequest(SignalRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) StartWorkflowExecution(ctx context.Context, StartRequest *shared.StartWorkflowExecutionRequest) (sp1 *shared.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.StartWorkflowExecution(ctx, thrift.ToStartWorkflowExecutionRequest(StartRequest))\n\treturn thrift.FromStartWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) StartWorkflowExecutionAsync(ctx context.Context, StartRequest *shared.StartWorkflowExecutionAsyncRequest) (sp1 *shared.StartWorkflowExecutionAsyncResponse, err error) {\n\tresponse, err := g.h.StartWorkflowExecutionAsync(ctx, thrift.ToStartWorkflowExecutionAsyncRequest(StartRequest))\n\treturn thrift.FromStartWorkflowExecutionAsyncResponse(response), thrift.FromError(err)\n}\n\nfunc (g APIHandler) TerminateWorkflowExecution(ctx context.Context, TerminateRequest *shared.TerminateWorkflowExecutionRequest) (err error) {\n\terr = g.h.TerminateWorkflowExecution(ctx, thrift.ToTerminateWorkflowExecutionRequest(TerminateRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g APIHandler) UpdateDomain(ctx context.Context, UpdateRequest *shared.UpdateDomainRequest) (up1 *shared.UpdateDomainResponse, err error) {\n\tresponse, err := g.h.UpdateDomain(ctx, thrift.ToUpdateDomainRequest(UpdateRequest))\n\treturn thrift.FromUpdateDomainResponse(response), thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/thrift/api_handler_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/health\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n)\n\nfunc TestThriftHandler(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\th := api.NewMockHandler(ctrl)\n\tth := NewAPIHandler(h)\n\tctx := context.Background()\n\tinternalErr := &types.InternalServiceError{Message: \"test\"}\n\texpectedErr := &shared.InternalServiceError{Message: \"test\"}\n\n\tt.Run(\"Health\", func(t *testing.T) {\n\t\th.EXPECT().Health(ctx).Return(&types.HealthStatus{}, internalErr).Times(1)\n\t\tresp, err := th.Health(ctx)\n\t\tassert.Equal(t, health.HealthStatus{Msg: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"CountWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().CountWorkflowExecutions(ctx, &types.CountWorkflowExecutionsRequest{}).Return(&types.CountWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.CountWorkflowExecutions(ctx, &shared.CountWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.CountWorkflowExecutionsResponse{Count: common.Int64Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DeprecateDomain\", func(t *testing.T) {\n\t\th.EXPECT().DeprecateDomain(ctx, &types.DeprecateDomainRequest{}).Return(internalErr).Times(1)\n\t\terr := th.DeprecateDomain(ctx, &shared.DeprecateDomainRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeDomain\", func(t *testing.T) {\n\t\th.EXPECT().DescribeDomain(ctx, &types.DescribeDomainRequest{}).Return(&types.DescribeDomainResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeDomain(ctx, &shared.DescribeDomainRequest{})\n\t\tassert.Equal(t, shared.DescribeDomainResponse{IsGlobalDomain: common.BoolPtr(false), FailoverVersion: common.Int64Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeTaskList\", func(t *testing.T) {\n\t\th.EXPECT().DescribeTaskList(ctx, &types.DescribeTaskListRequest{}).Return(&types.DescribeTaskListResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeTaskList(ctx, &shared.DescribeTaskListRequest{})\n\t\tassert.Equal(t, shared.DescribeTaskListResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{}).Return(&types.DescribeWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeWorkflowExecution(ctx, &shared.DescribeWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.DescribeWorkflowExecutionResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetClusterInfo\", func(t *testing.T) {\n\t\th.EXPECT().GetClusterInfo(ctx).Return(&types.ClusterInfo{}, internalErr).Times(1)\n\t\tresp, err := th.GetClusterInfo(ctx)\n\t\tassert.Equal(t, shared.ClusterInfo{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetSearchAttributes\", func(t *testing.T) {\n\t\th.EXPECT().GetSearchAttributes(ctx).Return(&types.GetSearchAttributesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetSearchAttributes(ctx)\n\t\tassert.Equal(t, shared.GetSearchAttributesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetWorkflowExecutionHistory\", func(t *testing.T) {\n\t\th.EXPECT().GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{}).Return(&types.GetWorkflowExecutionHistoryResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetWorkflowExecutionHistory(ctx, &shared.GetWorkflowExecutionHistoryRequest{})\n\t\tassert.Equal(t, shared.GetWorkflowExecutionHistoryResponse{Archived: common.BoolPtr(false)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListArchivedWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().ListArchivedWorkflowExecutions(ctx, &types.ListArchivedWorkflowExecutionsRequest{}).Return(&types.ListArchivedWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListArchivedWorkflowExecutions(ctx, &shared.ListArchivedWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.ListArchivedWorkflowExecutionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListClosedWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().ListClosedWorkflowExecutions(ctx, &types.ListClosedWorkflowExecutionsRequest{}).Return(&types.ListClosedWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListClosedWorkflowExecutions(ctx, &shared.ListClosedWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.ListClosedWorkflowExecutionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListDomains\", func(t *testing.T) {\n\t\th.EXPECT().ListDomains(ctx, &types.ListDomainsRequest{}).Return(&types.ListDomainsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListDomains(ctx, &shared.ListDomainsRequest{})\n\t\tassert.Equal(t, shared.ListDomainsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListOpenWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().ListOpenWorkflowExecutions(ctx, &types.ListOpenWorkflowExecutionsRequest{}).Return(&types.ListOpenWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListOpenWorkflowExecutions(ctx, &shared.ListOpenWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.ListOpenWorkflowExecutionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListTaskListPartitions\", func(t *testing.T) {\n\t\th.EXPECT().ListTaskListPartitions(ctx, &types.ListTaskListPartitionsRequest{}).Return(&types.ListTaskListPartitionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListTaskListPartitions(ctx, &shared.ListTaskListPartitionsRequest{})\n\t\tassert.Equal(t, shared.ListTaskListPartitionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().ListWorkflowExecutions(ctx, &types.ListWorkflowExecutionsRequest{}).Return(&types.ListWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListWorkflowExecutions(ctx, &shared.ListWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.ListWorkflowExecutionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PollForActivityTask\", func(t *testing.T) {\n\t\th.EXPECT().PollForActivityTask(ctx, &types.PollForActivityTaskRequest{}).Return(&types.PollForActivityTaskResponse{}, internalErr).Times(1)\n\t\tresp, err := th.PollForActivityTask(ctx, &shared.PollForActivityTaskRequest{})\n\t\tassert.Equal(t, shared.PollForActivityTaskResponse{WorkflowDomain: common.StringPtr(\"\"), ActivityId: common.StringPtr(\"\"), Attempt: common.Int32Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PollForDecisionTask\", func(t *testing.T) {\n\t\th.EXPECT().PollForDecisionTask(ctx, &types.PollForDecisionTaskRequest{}).Return(&types.PollForDecisionTaskResponse{}, internalErr).Times(1)\n\t\tresp, err := th.PollForDecisionTask(ctx, &shared.PollForDecisionTaskRequest{})\n\t\tassert.Equal(t, shared.PollForDecisionTaskResponse{StartedEventId: common.Int64Ptr(0), Attempt: common.Int64Ptr(0), BacklogCountHint: common.Int64Ptr(0), NextEventId: common.Int64Ptr(0), TotalHistoryBytes: common.Int64Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"QueryWorkflow\", func(t *testing.T) {\n\t\th.EXPECT().QueryWorkflow(ctx, &types.QueryWorkflowRequest{}).Return(&types.QueryWorkflowResponse{}, internalErr).Times(1)\n\t\tresp, err := th.QueryWorkflow(ctx, &shared.QueryWorkflowRequest{})\n\t\tassert.Equal(t, shared.QueryWorkflowResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordActivityTaskHeartbeat\", func(t *testing.T) {\n\t\th.EXPECT().RecordActivityTaskHeartbeat(ctx, &types.RecordActivityTaskHeartbeatRequest{}).Return(&types.RecordActivityTaskHeartbeatResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RecordActivityTaskHeartbeat(ctx, &shared.RecordActivityTaskHeartbeatRequest{})\n\t\tassert.Equal(t, shared.RecordActivityTaskHeartbeatResponse{CancelRequested: common.BoolPtr(false)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordActivityTaskHeartbeatByID\", func(t *testing.T) {\n\t\th.EXPECT().RecordActivityTaskHeartbeatByID(ctx, &types.RecordActivityTaskHeartbeatByIDRequest{}).Return(&types.RecordActivityTaskHeartbeatResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RecordActivityTaskHeartbeatByID(ctx, &shared.RecordActivityTaskHeartbeatByIDRequest{})\n\t\tassert.Equal(t, shared.RecordActivityTaskHeartbeatResponse{CancelRequested: common.BoolPtr(false)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RegisterDomain\", func(t *testing.T) {\n\t\th.EXPECT().RegisterDomain(ctx, &types.RegisterDomainRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RegisterDomain(ctx, &shared.RegisterDomainRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RequestCancelWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().RequestCancelWorkflowExecution(ctx, &types.RequestCancelWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RequestCancelWorkflowExecution(ctx, &shared.RequestCancelWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetStickyTaskList\", func(t *testing.T) {\n\t\th.EXPECT().ResetStickyTaskList(ctx, &types.ResetStickyTaskListRequest{}).Return(&types.ResetStickyTaskListResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ResetStickyTaskList(ctx, &shared.ResetStickyTaskListRequest{})\n\t\tassert.Equal(t, shared.ResetStickyTaskListResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{}).Return(&types.ResetWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ResetWorkflowExecution(ctx, &shared.ResetWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.ResetWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCanceled\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCanceled(ctx, &types.RespondActivityTaskCanceledRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCanceled(ctx, &shared.RespondActivityTaskCanceledRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCanceledByID\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCanceledByID(ctx, &types.RespondActivityTaskCanceledByIDRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCanceledByID(ctx, &shared.RespondActivityTaskCanceledByIDRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCompleted(ctx, &types.RespondActivityTaskCompletedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCompleted(ctx, &shared.RespondActivityTaskCompletedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCompletedByID\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCompletedByID(ctx, &types.RespondActivityTaskCompletedByIDRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCompletedByID(ctx, &shared.RespondActivityTaskCompletedByIDRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskFailed\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskFailed(ctx, &types.RespondActivityTaskFailedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskFailed(ctx, &shared.RespondActivityTaskFailedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskFailedByID\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskFailedByID(ctx, &types.RespondActivityTaskFailedByIDRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskFailedByID(ctx, &shared.RespondActivityTaskFailedByIDRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondDecisionTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondDecisionTaskCompleted(ctx, &types.RespondDecisionTaskCompletedRequest{}).Return(&types.RespondDecisionTaskCompletedResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RespondDecisionTaskCompleted(ctx, &shared.RespondDecisionTaskCompletedRequest{})\n\t\tassert.Equal(t, shared.RespondDecisionTaskCompletedResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondDecisionTaskFailed\", func(t *testing.T) {\n\t\th.EXPECT().RespondDecisionTaskFailed(ctx, &types.RespondDecisionTaskFailedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondDecisionTaskFailed(ctx, &shared.RespondDecisionTaskFailedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondQueryTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondQueryTaskCompleted(ctx, &types.RespondQueryTaskCompletedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondQueryTaskCompleted(ctx, &shared.RespondQueryTaskCompletedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ScanWorkflowExecutions\", func(t *testing.T) {\n\t\th.EXPECT().ScanWorkflowExecutions(ctx, &types.ListWorkflowExecutionsRequest{}).Return(&types.ListWorkflowExecutionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ScanWorkflowExecutions(ctx, &shared.ListWorkflowExecutionsRequest{})\n\t\tassert.Equal(t, shared.ListWorkflowExecutionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SignalWithStartWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().SignalWithStartWorkflowExecution(ctx, &types.SignalWithStartWorkflowExecutionRequest{}).Return(&types.StartWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.SignalWithStartWorkflowExecution(ctx, &shared.SignalWithStartWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.StartWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SignalWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.SignalWorkflowExecution(ctx, &shared.SignalWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"StartWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().StartWorkflowExecution(ctx, &types.StartWorkflowExecutionRequest{}).Return(&types.StartWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.StartWorkflowExecution(ctx, &shared.StartWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.StartWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"TerminateWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.TerminateWorkflowExecution(ctx, &shared.TerminateWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"UpdateDomain\", func(t *testing.T) {\n\t\th.EXPECT().UpdateDomain(ctx, &types.UpdateDomainRequest{}).Return(&types.UpdateDomainResponse{}, internalErr).Times(1)\n\t\tresp, err := th.UpdateDomain(ctx, &shared.UpdateDomainRequest{})\n\t\tassert.Equal(t, shared.UpdateDomainResponse{IsGlobalDomain: common.BoolPtr(false), FailoverVersion: common.Int64Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n}\n"
  },
  {
    "path": "service/frontend/wrappers/thrift/constructor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/admin/adminserviceserver\"\n\t\"github.com/uber/cadence/.gen/go/cadence/workflowserviceserver\"\n\t\"github.com/uber/cadence/.gen/go/health\"\n\t\"github.com/uber/cadence/.gen/go/health/metaserver\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/frontend/admin\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n)\n\ntype (\n\tAdminHandler struct {\n\t\th admin.Handler\n\t}\n\tAPIHandler struct {\n\t\th api.Handler\n\t}\n)\n\nfunc NewAdminHandler(h admin.Handler) AdminHandler {\n\treturn AdminHandler{h}\n}\n\nfunc NewAPIHandler(h api.Handler) APIHandler {\n\treturn APIHandler{h}\n}\n\nfunc (t AdminHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(adminserviceserver.New(t))\n}\n\nfunc (t APIHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(workflowserviceserver.New(t))\n\tdispatcher.Register(metaserver.New(t))\n}\n\nfunc (t APIHandler) Health(ctx context.Context) (*health.HealthStatus, error) {\n\tresponse, err := t.h.Health(ctx)\n\treturn thrift.FromHealthStatus(response), thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/frontend/wrappers/versioncheck/api_generated.go",
    "content": "package versioncheck\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/versioncheck.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/api\"\n\t\"github.com/uber/cadence/service/frontend/config\"\n)\n\ntype (\n\tversionCheckHandler struct {\n\t\tfrontendHandler api.Handler\n\t\tconfig          *config.Config\n\t\tversionChecker  client.VersionChecker\n\t}\n)\n\nfunc NewAPIHandler(\n\twfHandler api.Handler,\n\tconfig *config.Config,\n\tversionChecker client.VersionChecker,\n) api.Handler {\n\treturn &versionCheckHandler{\n\t\tfrontendHandler: wfHandler,\n\t\tconfig:          config,\n\t\tversionChecker:  versionChecker,\n\t}\n}\n\nfunc (h *versionCheckHandler) CountWorkflowExecutions(ctx context.Context, cp1 *types.CountWorkflowExecutionsRequest) (cp2 *types.CountWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.CountWorkflowExecutions(ctx, cp1)\n}\n\nfunc (h *versionCheckHandler) DeleteDomain(ctx context.Context, dp1 *types.DeleteDomainRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DeleteDomain(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) DeprecateDomain(ctx context.Context, dp1 *types.DeprecateDomainRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DeprecateDomain(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) DescribeDomain(ctx context.Context, dp1 *types.DescribeDomainRequest) (dp2 *types.DescribeDomainResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DescribeDomain(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) DescribeTaskList(ctx context.Context, dp1 *types.DescribeTaskListRequest) (dp2 *types.DescribeTaskListResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DescribeTaskList(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) DescribeWorkflowExecution(ctx context.Context, dp1 *types.DescribeWorkflowExecutionRequest) (dp2 *types.DescribeWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DescribeWorkflowExecution(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) DiagnoseWorkflowExecution(ctx context.Context, dp1 *types.DiagnoseWorkflowExecutionRequest) (dp2 *types.DiagnoseWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.DiagnoseWorkflowExecution(ctx, dp1)\n}\n\nfunc (h *versionCheckHandler) FailoverDomain(ctx context.Context, fp1 *types.FailoverDomainRequest) (fp2 *types.FailoverDomainResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.FailoverDomain(ctx, fp1)\n}\n\nfunc (h *versionCheckHandler) GetClusterInfo(ctx context.Context) (cp1 *types.ClusterInfo, err error) {\n\treturn h.frontendHandler.GetClusterInfo(ctx)\n}\n\nfunc (h *versionCheckHandler) GetSearchAttributes(ctx context.Context) (gp1 *types.GetSearchAttributesResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.GetSearchAttributes(ctx)\n}\n\nfunc (h *versionCheckHandler) GetTaskListsByDomain(ctx context.Context, gp1 *types.GetTaskListsByDomainRequest) (gp2 *types.GetTaskListsByDomainResponse, err error) {\n\treturn h.frontendHandler.GetTaskListsByDomain(ctx, gp1)\n}\n\nfunc (h *versionCheckHandler) GetWorkflowExecutionHistory(ctx context.Context, gp1 *types.GetWorkflowExecutionHistoryRequest) (gp2 *types.GetWorkflowExecutionHistoryResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.GetWorkflowExecutionHistory(ctx, gp1)\n}\n\nfunc (h *versionCheckHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn h.frontendHandler.Health(ctx)\n}\n\nfunc (h *versionCheckHandler) ListArchivedWorkflowExecutions(ctx context.Context, lp1 *types.ListArchivedWorkflowExecutionsRequest) (lp2 *types.ListArchivedWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListArchivedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListClosedWorkflowExecutions(ctx context.Context, lp1 *types.ListClosedWorkflowExecutionsRequest) (lp2 *types.ListClosedWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListClosedWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListDomains(ctx context.Context, lp1 *types.ListDomainsRequest) (lp2 *types.ListDomainsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListDomains(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListFailoverHistory(ctx context.Context, lp1 *types.ListFailoverHistoryRequest) (lp2 *types.ListFailoverHistoryResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListFailoverHistory(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListOpenWorkflowExecutions(ctx context.Context, lp1 *types.ListOpenWorkflowExecutionsRequest) (lp2 *types.ListOpenWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListOpenWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListTaskListPartitions(ctx context.Context, lp1 *types.ListTaskListPartitionsRequest) (lp2 *types.ListTaskListPartitionsResponse, err error) {\n\treturn h.frontendHandler.ListTaskListPartitions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) ListWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ListWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) PollForActivityTask(ctx context.Context, pp1 *types.PollForActivityTaskRequest) (pp2 *types.PollForActivityTaskResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.PollForActivityTask(ctx, pp1)\n}\n\nfunc (h *versionCheckHandler) PollForDecisionTask(ctx context.Context, pp1 *types.PollForDecisionTaskRequest) (pp2 *types.PollForDecisionTaskResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.PollForDecisionTask(ctx, pp1)\n}\n\nfunc (h *versionCheckHandler) QueryWorkflow(ctx context.Context, qp1 *types.QueryWorkflowRequest) (qp2 *types.QueryWorkflowResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.QueryWorkflow(ctx, qp1)\n}\n\nfunc (h *versionCheckHandler) RecordActivityTaskHeartbeat(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RecordActivityTaskHeartbeat(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RecordActivityTaskHeartbeatByID(ctx context.Context, rp1 *types.RecordActivityTaskHeartbeatByIDRequest) (rp2 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RecordActivityTaskHeartbeatByID(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RefreshWorkflowTasks(ctx context.Context, rp1 *types.RefreshWorkflowTasksRequest) (err error) {\n\treturn h.frontendHandler.RefreshWorkflowTasks(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RegisterDomain(ctx context.Context, rp1 *types.RegisterDomainRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RegisterDomain(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RequestCancelWorkflowExecution(ctx context.Context, rp1 *types.RequestCancelWorkflowExecutionRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RequestCancelWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) ResetStickyTaskList(ctx context.Context, rp1 *types.ResetStickyTaskListRequest) (rp2 *types.ResetStickyTaskListResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ResetStickyTaskList(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) ResetWorkflowExecution(ctx context.Context, rp1 *types.ResetWorkflowExecutionRequest) (rp2 *types.ResetWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ResetWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskCanceled(ctx context.Context, rp1 *types.RespondActivityTaskCanceledRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskCanceled(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskCanceledByID(ctx context.Context, rp1 *types.RespondActivityTaskCanceledByIDRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskCanceledByID(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskCompleted(ctx context.Context, rp1 *types.RespondActivityTaskCompletedRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskCompleted(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskCompletedByID(ctx context.Context, rp1 *types.RespondActivityTaskCompletedByIDRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskCompletedByID(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskFailed(ctx context.Context, rp1 *types.RespondActivityTaskFailedRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskFailed(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondActivityTaskFailedByID(ctx context.Context, rp1 *types.RespondActivityTaskFailedByIDRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondActivityTaskFailedByID(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondDecisionTaskCompleted(ctx context.Context, rp1 *types.RespondDecisionTaskCompletedRequest) (rp2 *types.RespondDecisionTaskCompletedResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondDecisionTaskCompleted(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondDecisionTaskFailed(ctx context.Context, rp1 *types.RespondDecisionTaskFailedRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondDecisionTaskFailed(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RespondQueryTaskCompleted(ctx context.Context, rp1 *types.RespondQueryTaskCompletedRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RespondQueryTaskCompleted(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) RestartWorkflowExecution(ctx context.Context, rp1 *types.RestartWorkflowExecutionRequest) (rp2 *types.RestartWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.RestartWorkflowExecution(ctx, rp1)\n}\n\nfunc (h *versionCheckHandler) ScanWorkflowExecutions(ctx context.Context, lp1 *types.ListWorkflowExecutionsRequest) (lp2 *types.ListWorkflowExecutionsResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.ScanWorkflowExecutions(ctx, lp1)\n}\n\nfunc (h *versionCheckHandler) SignalWithStartWorkflowExecution(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.SignalWithStartWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *versionCheckHandler) SignalWithStartWorkflowExecutionAsync(ctx context.Context, sp1 *types.SignalWithStartWorkflowExecutionAsyncRequest) (sp2 *types.SignalWithStartWorkflowExecutionAsyncResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.SignalWithStartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (h *versionCheckHandler) SignalWorkflowExecution(ctx context.Context, sp1 *types.SignalWorkflowExecutionRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.SignalWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *versionCheckHandler) StartWorkflowExecution(ctx context.Context, sp1 *types.StartWorkflowExecutionRequest) (sp2 *types.StartWorkflowExecutionResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.StartWorkflowExecution(ctx, sp1)\n}\n\nfunc (h *versionCheckHandler) StartWorkflowExecutionAsync(ctx context.Context, sp1 *types.StartWorkflowExecutionAsyncRequest) (sp2 *types.StartWorkflowExecutionAsyncResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.StartWorkflowExecutionAsync(ctx, sp1)\n}\n\nfunc (h *versionCheckHandler) TerminateWorkflowExecution(ctx context.Context, tp1 *types.TerminateWorkflowExecutionRequest) (err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.TerminateWorkflowExecution(ctx, tp1)\n}\n\nfunc (h *versionCheckHandler) UpdateDomain(ctx context.Context, up1 *types.UpdateDomainRequest) (up2 *types.UpdateDomainResponse, err error) {\n\terr = h.versionChecker.ClientSupported(ctx, h.config.EnableClientVersionCheck())\n\tif err != nil {\n\t\treturn\n\t}\n\treturn h.frontendHandler.UpdateDomain(ctx, up1)\n}\n"
  },
  {
    "path": "service/history/common/type.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage common\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/events\"\n)\n\ntype (\n\t// NotifyTaskInfo defines the info of task notification\n\tNotifyTaskInfo struct {\n\t\tExecutionInfo    *persistence.WorkflowExecutionInfo\n\t\tTasks            []persistence.Task\n\t\tVersionHistories *persistence.VersionHistories\n\t\tActivities       map[int64]*persistence.ActivityInfo\n\t\tHistory          events.PersistedBlobs\n\t\tPersistenceError bool\n\t}\n)\n"
  },
  {
    "path": "service/history/config/config.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\n// Config represents configuration for cadence-history service\ntype Config struct {\n\tNumberOfShards                   int\n\tIsAdvancedVisConfigExist         bool\n\tRPS                              dynamicproperties.IntPropertyFn\n\tMaxIDLengthWarnLimit             dynamicproperties.IntPropertyFn\n\tDomainNameMaxLength              dynamicproperties.IntPropertyFnWithDomainFilter\n\tIdentityMaxLength                dynamicproperties.IntPropertyFnWithDomainFilter\n\tWorkflowIDMaxLength              dynamicproperties.IntPropertyFnWithDomainFilter\n\tSignalNameMaxLength              dynamicproperties.IntPropertyFnWithDomainFilter\n\tWorkflowTypeMaxLength            dynamicproperties.IntPropertyFnWithDomainFilter\n\tRequestIDMaxLength               dynamicproperties.IntPropertyFnWithDomainFilter\n\tTaskListNameMaxLength            dynamicproperties.IntPropertyFnWithDomainFilter\n\tActivityIDMaxLength              dynamicproperties.IntPropertyFnWithDomainFilter\n\tActivityTypeMaxLength            dynamicproperties.IntPropertyFnWithDomainFilter\n\tMarkerNameMaxLength              dynamicproperties.IntPropertyFnWithDomainFilter\n\tTimerIDMaxLength                 dynamicproperties.IntPropertyFnWithDomainFilter\n\tPersistenceMaxQPS                dynamicproperties.IntPropertyFn\n\tPersistenceGlobalMaxQPS          dynamicproperties.IntPropertyFn\n\tEnableVisibilitySampling         dynamicproperties.BoolPropertyFn\n\tEnableReadFromClosedExecutionV2  dynamicproperties.BoolPropertyFn\n\tVisibilityOpenMaxQPS             dynamicproperties.IntPropertyFnWithDomainFilter\n\tVisibilityClosedMaxQPS           dynamicproperties.IntPropertyFnWithDomainFilter\n\tWriteVisibilityStoreName         dynamicproperties.StringPropertyFn\n\tEmitShardDiffLog                 dynamicproperties.BoolPropertyFn\n\tMaxAutoResetPoints               dynamicproperties.IntPropertyFnWithDomainFilter\n\tThrottledLogRPS                  dynamicproperties.IntPropertyFn\n\tEnableStickyQuery                dynamicproperties.BoolPropertyFnWithDomainFilter\n\tShutdownDrainDuration            dynamicproperties.DurationPropertyFn\n\tWorkflowDeletionJitterRange      dynamicproperties.IntPropertyFnWithDomainFilter\n\tDeleteHistoryEventContextTimeout dynamicproperties.IntPropertyFn\n\tMaxResponseSize                  int\n\n\t// HistoryCache settings\n\t// Change of these configs require shard restart\n\tHistoryCacheInitialSize              dynamicproperties.IntPropertyFn\n\tHistoryCacheMaxSize                  dynamicproperties.IntPropertyFn\n\tHistoryCacheTTL                      dynamicproperties.DurationPropertyFn\n\tEnableSizeBasedHistoryExecutionCache dynamicproperties.BoolPropertyFn\n\tExecutionCacheMaxByteSize            dynamicproperties.IntPropertyFn\n\n\t// EventsCache settings\n\t// Change of these configs require shard restart\n\tEventsCacheInitialCount          dynamicproperties.IntPropertyFn\n\tEventsCacheMaxCount              dynamicproperties.IntPropertyFn\n\tEventsCacheMaxSize               dynamicproperties.IntPropertyFn\n\tEventsCacheTTL                   dynamicproperties.DurationPropertyFn\n\tEventsCacheGlobalEnable          dynamicproperties.BoolPropertyFn\n\tEventsCacheGlobalInitialCount    dynamicproperties.IntPropertyFn\n\tEventsCacheGlobalMaxCount        dynamicproperties.IntPropertyFn\n\tEnableSizeBasedHistoryEventCache dynamicproperties.BoolPropertyFn\n\n\t// ShardController settings\n\tRangeSizeBits           uint\n\tAcquireShardInterval    dynamicproperties.DurationPropertyFn\n\tAcquireShardConcurrency dynamicproperties.IntPropertyFn\n\n\t// the artificial delay added to standby cluster's view of active cluster's time\n\tStandbyClusterDelay                  dynamicproperties.DurationPropertyFn\n\tStandbyTaskMissingEventsResendDelay  dynamicproperties.DurationPropertyFn\n\tStandbyTaskMissingEventsDiscardDelay dynamicproperties.DurationPropertyFn\n\n\t// Task process settings\n\tTaskProcessRPS                                    dynamicproperties.IntPropertyFnWithDomainFilter\n\tTaskSchedulerType                                 dynamicproperties.IntPropertyFn\n\tTaskSchedulerWorkerCount                          dynamicproperties.IntPropertyFn\n\tTaskSchedulerQueueSize                            dynamicproperties.IntPropertyFn\n\tTaskSchedulerDispatcherCount                      dynamicproperties.IntPropertyFn\n\tTaskSchedulerRoundRobinWeights                    dynamicproperties.MapPropertyFn\n\tTaskSchedulerDomainRoundRobinWeights              dynamicproperties.MapPropertyFnWithDomainFilter\n\tTaskSchedulerGlobalDomainRPS                      dynamicproperties.IntPropertyFnWithDomainFilter\n\tTaskSchedulerEnableRateLimiter                    dynamicproperties.BoolPropertyFn\n\tTaskSchedulerEnableRateLimiterShadowMode          dynamicproperties.BoolPropertyFnWithDomainFilter\n\tTaskCriticalRetryCount                            dynamicproperties.IntPropertyFn\n\tActiveTaskRedispatchInterval                      dynamicproperties.DurationPropertyFn\n\tStandbyTaskRedispatchInterval                     dynamicproperties.DurationPropertyFn\n\tStandbyTaskReReplicationContextTimeout            dynamicproperties.DurationPropertyFnWithDomainIDFilter\n\tEnableDropStuckTaskByDomainID                     dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tResurrectionCheckMinDelay                         dynamicproperties.DurationPropertyFnWithDomainFilter\n\tEnableHierarchicalWeightedRoundRobinTaskScheduler dynamicproperties.BoolPropertyFn\n\tEnableTaskListAwareTaskSchedulerByDomain          dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// History Queue (v2) settings\n\tEnableTimerQueueV2                         dynamicproperties.BoolPropertyFnWithShardIDFilter\n\tEnableTransferQueueV2                      dynamicproperties.BoolPropertyFnWithShardIDFilter\n\tQueueMaxPendingTaskCount                   dynamicproperties.IntPropertyFn\n\tEnableTimerQueueV2PendingTaskCountAlert    dynamicproperties.BoolPropertyFnWithShardIDFilter\n\tEnableTransferQueueV2PendingTaskCountAlert dynamicproperties.BoolPropertyFnWithShardIDFilter\n\tQueueCriticalPendingTaskCount              dynamicproperties.IntPropertyFn\n\tQueueMaxVirtualQueueCount                  dynamicproperties.IntPropertyFn\n\tVirtualSliceForceAppendInterval            dynamicproperties.DurationPropertyFn\n\n\t// QueueProcessor settings\n\tQueueProcessorEnableSplit                          dynamicproperties.BoolPropertyFn\n\tQueueProcessorSplitMaxLevel                        dynamicproperties.IntPropertyFn\n\tQueueProcessorEnableRandomSplitByDomainID          dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tQueueProcessorRandomSplitProbability               dynamicproperties.FloatPropertyFn\n\tQueueProcessorEnablePendingTaskSplitByDomainID     dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tQueueProcessorPendingTaskSplitThreshold            dynamicproperties.MapPropertyFn\n\tQueueProcessorEnableStuckTaskSplitByDomainID       dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tQueueProcessorStuckTaskSplitThreshold              dynamicproperties.MapPropertyFn\n\tQueueProcessorSplitLookAheadDurationByDomainID     dynamicproperties.DurationPropertyFnWithDomainIDFilter\n\tQueueProcessorPollBackoffInterval                  dynamicproperties.DurationPropertyFn\n\tQueueProcessorPollBackoffIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\tQueueProcessorEnablePersistQueueStates             dynamicproperties.BoolPropertyFn\n\tQueueProcessorEnableLoadQueueStates                dynamicproperties.BoolPropertyFn\n\tQueueProcessorEnableGracefulSyncShutdown           dynamicproperties.BoolPropertyFn\n\n\t// TimerQueueProcessor settings\n\tTimerTaskBatchSize                                dynamicproperties.IntPropertyFn\n\tTimerTaskDeleteBatchSize                          dynamicproperties.IntPropertyFn\n\tTimerProcessorGetFailureRetryCount                dynamicproperties.IntPropertyFn\n\tTimerProcessorCompleteTimerFailureRetryCount      dynamicproperties.IntPropertyFn\n\tTimerProcessorUpdateAckInterval                   dynamicproperties.DurationPropertyFn\n\tTimerProcessorUpdateAckIntervalJitterCoefficient  dynamicproperties.FloatPropertyFn\n\tTimerProcessorCompleteTimerInterval               dynamicproperties.DurationPropertyFn\n\tTimerProcessorFailoverMaxStartJitterInterval      dynamicproperties.DurationPropertyFn\n\tTimerProcessorFailoverMaxPollRPS                  dynamicproperties.IntPropertyFn\n\tTimerProcessorMaxPollRPS                          dynamicproperties.IntPropertyFn\n\tTimerProcessorMaxPollInterval                     dynamicproperties.DurationPropertyFn\n\tTimerProcessorMaxPollIntervalJitterCoefficient    dynamicproperties.FloatPropertyFn\n\tTimerProcessorSplitQueueInterval                  dynamicproperties.DurationPropertyFn\n\tTimerProcessorSplitQueueIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\tTimerProcessorMaxRedispatchQueueSize              dynamicproperties.IntPropertyFn\n\tTimerProcessorMaxTimeShift                        dynamicproperties.DurationPropertyFn\n\tTimerProcessorHistoryArchivalSizeLimit            dynamicproperties.IntPropertyFn\n\tTimerProcessorArchivalTimeLimit                   dynamicproperties.DurationPropertyFn\n\tDisableTimerFailoverQueue                         dynamicproperties.BoolPropertyFn\n\n\t// TransferQueueProcessor settings\n\tTransferTaskBatchSize                                dynamicproperties.IntPropertyFn\n\tTransferTaskDeleteBatchSize                          dynamicproperties.IntPropertyFn\n\tTransferProcessorCompleteTransferFailureRetryCount   dynamicproperties.IntPropertyFn\n\tTransferProcessorFailoverMaxStartJitterInterval      dynamicproperties.DurationPropertyFn\n\tTransferProcessorFailoverMaxPollRPS                  dynamicproperties.IntPropertyFn\n\tTransferProcessorMaxPollRPS                          dynamicproperties.IntPropertyFn\n\tTransferProcessorMaxPollInterval                     dynamicproperties.DurationPropertyFn\n\tTransferProcessorMaxPollIntervalJitterCoefficient    dynamicproperties.FloatPropertyFn\n\tTransferProcessorSplitQueueInterval                  dynamicproperties.DurationPropertyFn\n\tTransferProcessorSplitQueueIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\tTransferProcessorUpdateAckInterval                   dynamicproperties.DurationPropertyFn\n\tTransferProcessorUpdateAckIntervalJitterCoefficient  dynamicproperties.FloatPropertyFn\n\tTransferProcessorCompleteTransferInterval            dynamicproperties.DurationPropertyFn\n\tTransferProcessorMaxRedispatchQueueSize              dynamicproperties.IntPropertyFn\n\tTransferProcessorEnableValidator                     dynamicproperties.BoolPropertyFn\n\tTransferProcessorValidationInterval                  dynamicproperties.DurationPropertyFn\n\tTransferProcessorVisibilityArchivalTimeLimit         dynamicproperties.DurationPropertyFn\n\tDisableTransferFailoverQueue                         dynamicproperties.BoolPropertyFn\n\n\t// ReplicatorQueueProcessor settings\n\tReplicatorTaskDeleteBatchSize            dynamicproperties.IntPropertyFn\n\tReplicatorReadTaskMaxRetryCount          dynamicproperties.IntPropertyFn\n\tReplicatorProcessorFetchTasksBatchSize   dynamicproperties.IntPropertyFnWithShardIDFilter\n\tReplicatorProcessorMaxTaskBatchSize      dynamicproperties.IntPropertyFnWithShardIDFilter\n\tReplicatorProcessorMinTaskBatchSize      dynamicproperties.IntPropertyFnWithShardIDFilter\n\tReplicatorProcessorBatchSizeStepCount    dynamicproperties.IntPropertyFnWithShardIDFilter\n\tReplicatorUpperLatency                   dynamicproperties.DurationPropertyFn\n\tReplicatorCacheCapacity                  dynamicproperties.IntPropertyFn\n\tReplicatorCacheMaxSize                   dynamicproperties.IntPropertyFn\n\tReplicationBudgetManagerMaxSizeBytes     dynamicproperties.IntPropertyFn\n\tReplicationBudgetManagerMaxSizeCount     dynamicproperties.IntPropertyFn\n\tReplicationBudgetManagerSoftCapThreshold dynamicproperties.FloatPropertyFn\n\n\t// System Limits\n\tMaximumBufferedEventsBatch dynamicproperties.IntPropertyFn\n\tMaximumSignalsPerExecution dynamicproperties.IntPropertyFnWithDomainFilter\n\n\t// ShardUpdateMinInterval the minimal time interval which the shard info can be updated\n\tShardUpdateMinInterval dynamicproperties.DurationPropertyFn\n\t// ShardSyncMinInterval the minimal time interval which the shard info should be sync to remote\n\tShardSyncMinInterval            dynamicproperties.DurationPropertyFn\n\tShardSyncTimerJitterCoefficient dynamicproperties.FloatPropertyFn\n\n\t// Time to hold a poll request before returning an empty response\n\t// right now only used by GetMutableState\n\tLongPollExpirationInterval dynamicproperties.DurationPropertyFnWithDomainFilter\n\n\t// encoding the history events\n\tEventEncodingType dynamicproperties.StringPropertyFnWithDomainFilter\n\t// whether or not using ParentClosePolicy\n\tEnableParentClosePolicy dynamicproperties.BoolPropertyFnWithDomainFilter\n\t// whether or not enable system workers for processing parent close policy task\n\tEnableParentClosePolicyWorker dynamicproperties.BoolPropertyFn\n\t// parent close policy will be processed by sys workers(if enabled) if\n\t// the number of children greater than or equal to this threshold\n\tParentClosePolicyThreshold dynamicproperties.IntPropertyFnWithDomainFilter\n\t// the batch size of parent close policy processed by sys workers\n\tParentClosePolicyBatchSize dynamicproperties.IntPropertyFnWithDomainFilter\n\t// total number of parentClosePolicy system workflows\n\tNumParentClosePolicySystemWorkflows dynamicproperties.IntPropertyFn\n\n\t// Archival settings\n\tNumArchiveSystemWorkflows        dynamicproperties.IntPropertyFn\n\tArchiveRequestRPS                dynamicproperties.IntPropertyFn\n\tArchiveInlineHistoryRPS          dynamicproperties.IntPropertyFn\n\tArchiveInlineHistoryGlobalRPS    dynamicproperties.IntPropertyFn\n\tArchiveInlineVisibilityRPS       dynamicproperties.IntPropertyFn\n\tArchiveInlineVisibilityGlobalRPS dynamicproperties.IntPropertyFn\n\tAllowArchivingIncompleteHistory  dynamicproperties.BoolPropertyFn\n\n\t// Size limit related settings\n\tBlobSizeLimitError               dynamicproperties.IntPropertyFnWithDomainFilter\n\tBlobSizeLimitWarn                dynamicproperties.IntPropertyFnWithDomainFilter\n\tHistorySizeLimitError            dynamicproperties.IntPropertyFnWithDomainFilter\n\tHistorySizeLimitWarn             dynamicproperties.IntPropertyFnWithDomainFilter\n\tHistoryCountLimitError           dynamicproperties.IntPropertyFnWithDomainFilter\n\tHistoryCountLimitWarn            dynamicproperties.IntPropertyFnWithDomainFilter\n\tPendingActivitiesCountLimitError dynamicproperties.IntPropertyFn\n\tPendingActivitiesCountLimitWarn  dynamicproperties.IntPropertyFn\n\tPendingActivityValidationEnabled dynamicproperties.BoolPropertyFn\n\n\t// ValidSearchAttributes is legal indexed keys that can be used in list APIs\n\tEnableQueryAttributeValidation    dynamicproperties.BoolPropertyFn\n\tValidSearchAttributes             dynamicproperties.MapPropertyFn\n\tSearchAttributesNumberOfKeysLimit dynamicproperties.IntPropertyFnWithDomainFilter\n\tSearchAttributesSizeOfValueLimit  dynamicproperties.IntPropertyFnWithDomainFilter\n\tSearchAttributesTotalSizeLimit    dynamicproperties.IntPropertyFnWithDomainFilter\n\tSearchAttributesHiddenValueKeys   dynamicproperties.MapPropertyFn\n\n\t// Decision settings\n\t// StickyTTL is to expire a sticky tasklist if no update more than this duration\n\t// TODO https://github.com/uber/cadence/issues/2357\n\tStickyTTL dynamicproperties.DurationPropertyFnWithDomainFilter\n\t// DecisionHeartbeatTimeout is to timeout behavior of: RespondDecisionTaskComplete with ForceCreateNewDecisionTask == true without any decisions\n\t// So that decision will be scheduled to another worker(by clear stickyness)\n\tDecisionHeartbeatTimeout dynamicproperties.DurationPropertyFnWithDomainFilter\n\t// MaxDecisionStartToCloseSeconds is the StartToCloseSeconds for decision\n\tMaxDecisionStartToCloseSeconds           dynamicproperties.IntPropertyFnWithDomainFilter\n\tDecisionRetryCriticalAttempts            dynamicproperties.IntPropertyFn\n\tDecisionRetryMaxAttempts                 dynamicproperties.IntPropertyFnWithDomainFilter\n\tEnforceDecisionTaskAttempts              dynamicproperties.BoolPropertyFnWithDomainFilter\n\tNormalDecisionScheduleToStartMaxAttempts dynamicproperties.IntPropertyFnWithDomainFilter\n\tNormalDecisionScheduleToStartTimeout     dynamicproperties.DurationPropertyFnWithDomainFilter\n\n\t// The following is used by the new RPC replication stack\n\tReplicationTaskFetcherParallelism                    dynamicproperties.IntPropertyFn\n\tReplicationTaskFetcherAggregationInterval            dynamicproperties.DurationPropertyFn\n\tReplicationTaskFetcherTimerJitterCoefficient         dynamicproperties.FloatPropertyFn\n\tReplicationTaskFetcherErrorRetryWait                 dynamicproperties.DurationPropertyFn\n\tReplicationTaskFetcherServiceBusyWait                dynamicproperties.DurationPropertyFn\n\tReplicationTaskProcessorErrorRetryWait               dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorErrorRetryMaxAttempts        dynamicproperties.IntPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorErrorSecondRetryWait         dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorErrorSecondRetryMaxWait      dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorErrorSecondRetryExpiration   dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorNoTaskRetryWait              dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorCleanupInterval              dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorCleanupJitterCoefficient     dynamicproperties.FloatPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorStartWait                    dynamicproperties.DurationPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorStartWaitJitterCoefficient   dynamicproperties.FloatPropertyFnWithShardIDFilter\n\tReplicationTaskProcessorHostQPS                      dynamicproperties.FloatPropertyFn\n\tReplicationTaskProcessorShardQPS                     dynamicproperties.FloatPropertyFn\n\tReplicationTaskGenerationQPS                         dynamicproperties.FloatPropertyFn\n\tEnableReplicationTaskGeneration                      dynamicproperties.BoolPropertyFnWithDomainIDAndWorkflowIDFilter\n\tEnableRecordWorkflowExecutionUninitialized           dynamicproperties.BoolPropertyFnWithDomainFilter\n\tEnableCleanupOrphanedHistoryBranchOnWorkflowCreation dynamicproperties.BoolPropertyFnWithDomainFilter\n\tReplicationTaskProcessorLatencyLogThreshold          dynamicproperties.DurationPropertyFn\n\n\t// The following are used by the history workflowID cache\n\tWorkflowIDExternalRPS dynamicproperties.IntPropertyFnWithDomainFilter\n\tWorkflowIDInternalRPS dynamicproperties.IntPropertyFnWithDomainFilter\n\n\t// The following are used by consistent query\n\tEnableConsistentQuery         dynamicproperties.BoolPropertyFn\n\tEnableConsistentQueryByDomain dynamicproperties.BoolPropertyFnWithDomainFilter\n\tMaxBufferedQueryCount         dynamicproperties.IntPropertyFn\n\n\t// EnableContextHeaderInVisibility whether to enable indexing context header in visibility\n\tEnableContextHeaderInVisibility dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\tEnableCrossClusterOperationsForDomain dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// Data integrity check related config knobs\n\tMutableStateChecksumGenProbability    dynamicproperties.IntPropertyFnWithDomainFilter\n\tMutableStateChecksumVerifyProbability dynamicproperties.IntPropertyFnWithDomainFilter\n\tMutableStateChecksumInvalidateBefore  dynamicproperties.FloatPropertyFn\n\tEnableRetryForChecksumFailure         dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// History check for corruptions\n\tEnableHistoryCorruptionCheck dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// Failover marker heartbeat\n\tNotifyFailoverMarkerInterval               dynamicproperties.DurationPropertyFn\n\tNotifyFailoverMarkerTimerJitterCoefficient dynamicproperties.FloatPropertyFn\n\tEnableGracefulFailover                     dynamicproperties.BoolPropertyFn\n\n\t// Allows worker to dispatch activity tasks through local tunnel after decisions are made. This is an performance optimization to skip activity scheduling efforts.\n\tEnableActivityLocalDispatchByDomain dynamicproperties.BoolPropertyFnWithDomainFilter\n\t// Max # of activity tasks to dispatch to matching before creating transfer tasks. This is an performance optimization to skip activity scheduling efforts.\n\tMaxActivityCountDispatchByDomain dynamicproperties.IntPropertyFnWithDomainFilter\n\n\tActivityMaxScheduleToStartTimeoutForRetry dynamicproperties.DurationPropertyFnWithDomainFilter\n\n\t// Debugging configurations\n\tEnableDebugMode               bool // note that this value is initialized once on service start\n\tEnableTaskInfoLogByDomainID   dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tEnableTimerDebugLogByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\n\t// Hotshard stuff\n\tSampleLoggingRate                     dynamicproperties.IntPropertyFn\n\tEnableShardIDMetrics                  dynamicproperties.BoolPropertyFn\n\tLargeShardHistorySizeMetricThreshold  dynamicproperties.IntPropertyFn\n\tLargeShardHistoryEventMetricThreshold dynamicproperties.IntPropertyFn\n\tLargeShardHistoryBlobMetricThreshold  dynamicproperties.IntPropertyFn\n\n\tEnableStrongIdempotency            dynamicproperties.BoolPropertyFnWithDomainFilter\n\tEnableStrongIdempotencySanityCheck dynamicproperties.BoolPropertyFnWithDomainFilter\n\n\t// Global ratelimiter\n\tGlobalRatelimiterNewDataWeight  dynamicproperties.FloatPropertyFn\n\tGlobalRatelimiterUpdateInterval dynamicproperties.DurationPropertyFn\n\tGlobalRatelimiterDecayAfter     dynamicproperties.DurationPropertyFn\n\tGlobalRatelimiterGCAfter        dynamicproperties.DurationPropertyFn\n\n\t// HostName for machine running the service\n\tHostName string\n}\n\n// New returns new service config with default values\nfunc New(dc *dynamicconfig.Collection, numberOfShards int, maxMessageSize int, isAdvancedVisConfigExist bool, hostname string) *Config {\n\tcfg := &Config{\n\t\tNumberOfShards:                       numberOfShards,\n\t\tIsAdvancedVisConfigExist:             isAdvancedVisConfigExist,\n\t\tRPS:                                  dc.GetIntProperty(dynamicproperties.HistoryRPS),\n\t\tMaxIDLengthWarnLimit:                 dc.GetIntProperty(dynamicproperties.MaxIDLengthWarnLimit),\n\t\tDomainNameMaxLength:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.DomainNameMaxLength),\n\t\tIdentityMaxLength:                    dc.GetIntPropertyFilteredByDomain(dynamicproperties.IdentityMaxLength),\n\t\tWorkflowIDMaxLength:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowIDMaxLength),\n\t\tSignalNameMaxLength:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.SignalNameMaxLength),\n\t\tWorkflowTypeMaxLength:                dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowTypeMaxLength),\n\t\tRequestIDMaxLength:                   dc.GetIntPropertyFilteredByDomain(dynamicproperties.RequestIDMaxLength),\n\t\tTaskListNameMaxLength:                dc.GetIntPropertyFilteredByDomain(dynamicproperties.TaskListNameMaxLength),\n\t\tActivityIDMaxLength:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.ActivityIDMaxLength),\n\t\tActivityTypeMaxLength:                dc.GetIntPropertyFilteredByDomain(dynamicproperties.ActivityTypeMaxLength),\n\t\tMarkerNameMaxLength:                  dc.GetIntPropertyFilteredByDomain(dynamicproperties.MarkerNameMaxLength),\n\t\tTimerIDMaxLength:                     dc.GetIntPropertyFilteredByDomain(dynamicproperties.TimerIDMaxLength),\n\t\tPersistenceMaxQPS:                    dc.GetIntProperty(dynamicproperties.HistoryPersistenceMaxQPS),\n\t\tPersistenceGlobalMaxQPS:              dc.GetIntProperty(dynamicproperties.HistoryPersistenceGlobalMaxQPS),\n\t\tShutdownDrainDuration:                dc.GetDurationProperty(dynamicproperties.HistoryShutdownDrainDuration),\n\t\tEnableVisibilitySampling:             dc.GetBoolProperty(dynamicproperties.EnableVisibilitySampling),\n\t\tEnableReadFromClosedExecutionV2:      dc.GetBoolProperty(dynamicproperties.EnableReadFromClosedExecutionV2),\n\t\tVisibilityOpenMaxQPS:                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistoryVisibilityOpenMaxQPS),\n\t\tVisibilityClosedMaxQPS:               dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistoryVisibilityClosedMaxQPS),\n\t\tMaxAutoResetPoints:                   dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistoryMaxAutoResetPoints),\n\t\tMaxDecisionStartToCloseSeconds:       dc.GetIntPropertyFilteredByDomain(dynamicproperties.MaxDecisionStartToCloseSeconds),\n\t\tWriteVisibilityStoreName:             dc.GetStringProperty(dynamicproperties.WriteVisibilityStoreName),\n\t\tEmitShardDiffLog:                     dc.GetBoolProperty(dynamicproperties.EmitShardDiffLog),\n\t\tHistoryCacheInitialSize:              dc.GetIntProperty(dynamicproperties.HistoryCacheInitialSize),\n\t\tHistoryCacheMaxSize:                  dc.GetIntProperty(dynamicproperties.HistoryCacheMaxSize),\n\t\tExecutionCacheMaxByteSize:            dc.GetIntProperty(dynamicproperties.ExecutionCacheMaxByteSize),\n\t\tHistoryCacheTTL:                      dc.GetDurationProperty(dynamicproperties.HistoryCacheTTL),\n\t\tEnableSizeBasedHistoryExecutionCache: dc.GetBoolProperty(dynamicproperties.EnableSizeBasedHistoryExecutionCache),\n\t\tEventsCacheInitialCount:              dc.GetIntProperty(dynamicproperties.EventsCacheInitialCount),\n\t\tEventsCacheMaxCount:                  dc.GetIntProperty(dynamicproperties.EventsCacheMaxCount),\n\t\tEventsCacheMaxSize:                   dc.GetIntProperty(dynamicproperties.EventsCacheMaxSize),\n\t\tEventsCacheTTL:                       dc.GetDurationProperty(dynamicproperties.EventsCacheTTL),\n\t\tEventsCacheGlobalEnable:              dc.GetBoolProperty(dynamicproperties.EventsCacheGlobalEnable),\n\t\tEventsCacheGlobalInitialCount:        dc.GetIntProperty(dynamicproperties.EventsCacheGlobalInitialCount),\n\t\tEventsCacheGlobalMaxCount:            dc.GetIntProperty(dynamicproperties.EventsCacheGlobalMaxCount),\n\t\tEnableSizeBasedHistoryEventCache:     dc.GetBoolProperty(dynamicproperties.EnableSizeBasedHistoryEventCache),\n\t\tRangeSizeBits:                        20, // 20 bits for sequencer, 2^20 sequence number for any range\n\t\tAcquireShardInterval:                 dc.GetDurationProperty(dynamicproperties.AcquireShardInterval),\n\t\tAcquireShardConcurrency:              dc.GetIntProperty(dynamicproperties.AcquireShardConcurrency),\n\t\tStandbyClusterDelay:                  dc.GetDurationProperty(dynamicproperties.StandbyClusterDelay),\n\t\tStandbyTaskMissingEventsResendDelay:  dc.GetDurationProperty(dynamicproperties.StandbyTaskMissingEventsResendDelay),\n\t\tStandbyTaskMissingEventsDiscardDelay: dc.GetDurationProperty(dynamicproperties.StandbyTaskMissingEventsDiscardDelay),\n\t\tWorkflowDeletionJitterRange:          dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowDeletionJitterRange),\n\t\tDeleteHistoryEventContextTimeout:     dc.GetIntProperty(dynamicproperties.DeleteHistoryEventContextTimeout),\n\t\tMaxResponseSize:                      maxMessageSize,\n\n\t\tTaskProcessRPS:                                    dc.GetIntPropertyFilteredByDomain(dynamicproperties.TaskProcessRPS),\n\t\tTaskSchedulerType:                                 dc.GetIntProperty(dynamicproperties.TaskSchedulerType),\n\t\tTaskSchedulerWorkerCount:                          dc.GetIntProperty(dynamicproperties.TaskSchedulerWorkerCount),\n\t\tTaskSchedulerQueueSize:                            dc.GetIntProperty(dynamicproperties.TaskSchedulerQueueSize),\n\t\tTaskSchedulerDispatcherCount:                      dc.GetIntProperty(dynamicproperties.TaskSchedulerDispatcherCount),\n\t\tTaskSchedulerRoundRobinWeights:                    dc.GetMapProperty(dynamicproperties.TaskSchedulerRoundRobinWeights),\n\t\tTaskSchedulerDomainRoundRobinWeights:              dc.GetMapPropertyFilteredByDomain(dynamicproperties.TaskSchedulerDomainRoundRobinWeights),\n\t\tTaskSchedulerGlobalDomainRPS:                      dc.GetIntPropertyFilteredByDomain(dynamicproperties.TaskSchedulerGlobalDomainRPS),\n\t\tTaskSchedulerEnableRateLimiter:                    dc.GetBoolProperty(dynamicproperties.TaskSchedulerEnableRateLimiter),\n\t\tTaskSchedulerEnableRateLimiterShadowMode:          dc.GetBoolPropertyFilteredByDomain(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode),\n\t\tTaskCriticalRetryCount:                            dc.GetIntProperty(dynamicproperties.TaskCriticalRetryCount),\n\t\tActiveTaskRedispatchInterval:                      dc.GetDurationProperty(dynamicproperties.ActiveTaskRedispatchInterval),\n\t\tStandbyTaskRedispatchInterval:                     dc.GetDurationProperty(dynamicproperties.StandbyTaskRedispatchInterval),\n\t\tStandbyTaskReReplicationContextTimeout:            dc.GetDurationPropertyFilteredByDomainID(dynamicproperties.StandbyTaskReReplicationContextTimeout),\n\t\tEnableDropStuckTaskByDomainID:                     dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.EnableDropStuckTaskByDomainID),\n\t\tResurrectionCheckMinDelay:                         dc.GetDurationPropertyFilteredByDomain(dynamicproperties.ResurrectionCheckMinDelay),\n\t\tEnableHierarchicalWeightedRoundRobinTaskScheduler: dc.GetBoolProperty(dynamicproperties.EnableHierarchicalWeightedRoundRobinTaskScheduler),\n\t\tEnableTaskListAwareTaskSchedulerByDomain:          dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableTaskListAwareTaskSchedulerByDomain),\n\n\t\tEnableTimerQueueV2:                         dc.GetBoolPropertyFilteredByShardID(dynamicproperties.EnableTimerQueueV2),\n\t\tEnableTransferQueueV2:                      dc.GetBoolPropertyFilteredByShardID(dynamicproperties.EnableTransferQueueV2),\n\t\tQueueMaxPendingTaskCount:                   dc.GetIntProperty(dynamicproperties.QueueMaxPendingTaskCount),\n\t\tEnableTimerQueueV2PendingTaskCountAlert:    dc.GetBoolPropertyFilteredByShardID(dynamicproperties.EnableTimerQueueV2PendingTaskCountAlert),\n\t\tEnableTransferQueueV2PendingTaskCountAlert: dc.GetBoolPropertyFilteredByShardID(dynamicproperties.EnableTransferQueueV2PendingTaskCountAlert),\n\t\tQueueCriticalPendingTaskCount:              dc.GetIntProperty(dynamicproperties.QueueCriticalPendingTaskCount),\n\t\tQueueMaxVirtualQueueCount:                  dc.GetIntProperty(dynamicproperties.QueueMaxVirtualQueueCount),\n\t\tVirtualSliceForceAppendInterval:            dc.GetDurationProperty(dynamicproperties.VirtualSliceForceAppendInterval),\n\n\t\tQueueProcessorEnableSplit:                          dc.GetBoolProperty(dynamicproperties.QueueProcessorEnableSplit),\n\t\tQueueProcessorSplitMaxLevel:                        dc.GetIntProperty(dynamicproperties.QueueProcessorSplitMaxLevel),\n\t\tQueueProcessorEnableRandomSplitByDomainID:          dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.QueueProcessorEnableRandomSplitByDomainID),\n\t\tQueueProcessorRandomSplitProbability:               dc.GetFloat64Property(dynamicproperties.QueueProcessorRandomSplitProbability),\n\t\tQueueProcessorEnablePendingTaskSplitByDomainID:     dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.QueueProcessorEnablePendingTaskSplitByDomainID),\n\t\tQueueProcessorPendingTaskSplitThreshold:            dc.GetMapProperty(dynamicproperties.QueueProcessorPendingTaskSplitThreshold),\n\t\tQueueProcessorEnableStuckTaskSplitByDomainID:       dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.QueueProcessorEnableStuckTaskSplitByDomainID),\n\t\tQueueProcessorStuckTaskSplitThreshold:              dc.GetMapProperty(dynamicproperties.QueueProcessorStuckTaskSplitThreshold),\n\t\tQueueProcessorSplitLookAheadDurationByDomainID:     dc.GetDurationPropertyFilteredByDomainID(dynamicproperties.QueueProcessorSplitLookAheadDurationByDomainID),\n\t\tQueueProcessorPollBackoffInterval:                  dc.GetDurationProperty(dynamicproperties.QueueProcessorPollBackoffInterval),\n\t\tQueueProcessorPollBackoffIntervalJitterCoefficient: dc.GetFloat64Property(dynamicproperties.QueueProcessorPollBackoffIntervalJitterCoefficient),\n\t\tQueueProcessorEnablePersistQueueStates:             dc.GetBoolProperty(dynamicproperties.QueueProcessorEnablePersistQueueStates),\n\t\tQueueProcessorEnableLoadQueueStates:                dc.GetBoolProperty(dynamicproperties.QueueProcessorEnableLoadQueueStates),\n\t\tQueueProcessorEnableGracefulSyncShutdown:           dc.GetBoolProperty(dynamicproperties.QueueProcessorEnableGracefulSyncShutdown),\n\n\t\tTimerTaskBatchSize:                                   dc.GetIntProperty(dynamicproperties.TimerTaskBatchSize),\n\t\tTimerTaskDeleteBatchSize:                             dc.GetIntProperty(dynamicproperties.TimerTaskDeleteBatchSize),\n\t\tTimerProcessorGetFailureRetryCount:                   dc.GetIntProperty(dynamicproperties.TimerProcessorGetFailureRetryCount),\n\t\tTimerProcessorCompleteTimerFailureRetryCount:         dc.GetIntProperty(dynamicproperties.TimerProcessorCompleteTimerFailureRetryCount),\n\t\tTimerProcessorUpdateAckInterval:                      dc.GetDurationProperty(dynamicproperties.TimerProcessorUpdateAckInterval),\n\t\tTimerProcessorUpdateAckIntervalJitterCoefficient:     dc.GetFloat64Property(dynamicproperties.TimerProcessorUpdateAckIntervalJitterCoefficient),\n\t\tTimerProcessorCompleteTimerInterval:                  dc.GetDurationProperty(dynamicproperties.TimerProcessorCompleteTimerInterval),\n\t\tTimerProcessorFailoverMaxStartJitterInterval:         dc.GetDurationProperty(dynamicproperties.TimerProcessorFailoverMaxStartJitterInterval),\n\t\tTimerProcessorFailoverMaxPollRPS:                     dc.GetIntProperty(dynamicproperties.TimerProcessorFailoverMaxPollRPS),\n\t\tTimerProcessorMaxPollRPS:                             dc.GetIntProperty(dynamicproperties.TimerProcessorMaxPollRPS),\n\t\tTimerProcessorMaxPollInterval:                        dc.GetDurationProperty(dynamicproperties.TimerProcessorMaxPollInterval),\n\t\tTimerProcessorMaxPollIntervalJitterCoefficient:       dc.GetFloat64Property(dynamicproperties.TimerProcessorMaxPollIntervalJitterCoefficient),\n\t\tTimerProcessorSplitQueueInterval:                     dc.GetDurationProperty(dynamicproperties.TimerProcessorSplitQueueInterval),\n\t\tTimerProcessorSplitQueueIntervalJitterCoefficient:    dc.GetFloat64Property(dynamicproperties.TimerProcessorSplitQueueIntervalJitterCoefficient),\n\t\tTimerProcessorMaxRedispatchQueueSize:                 dc.GetIntProperty(dynamicproperties.TimerProcessorMaxRedispatchQueueSize),\n\t\tTimerProcessorMaxTimeShift:                           dc.GetDurationProperty(dynamicproperties.TimerProcessorMaxTimeShift),\n\t\tTimerProcessorHistoryArchivalSizeLimit:               dc.GetIntProperty(dynamicproperties.TimerProcessorHistoryArchivalSizeLimit),\n\t\tTimerProcessorArchivalTimeLimit:                      dc.GetDurationProperty(dynamicproperties.TimerProcessorArchivalTimeLimit),\n\t\tDisableTimerFailoverQueue:                            dc.GetBoolProperty(dynamicproperties.DisableTimerFailoverQueue),\n\t\tTransferTaskBatchSize:                                dc.GetIntProperty(dynamicproperties.TransferTaskBatchSize),\n\t\tTransferTaskDeleteBatchSize:                          dc.GetIntProperty(dynamicproperties.TransferTaskDeleteBatchSize),\n\t\tTransferProcessorFailoverMaxStartJitterInterval:      dc.GetDurationProperty(dynamicproperties.TransferProcessorFailoverMaxStartJitterInterval),\n\t\tTransferProcessorFailoverMaxPollRPS:                  dc.GetIntProperty(dynamicproperties.TransferProcessorFailoverMaxPollRPS),\n\t\tTransferProcessorMaxPollRPS:                          dc.GetIntProperty(dynamicproperties.TransferProcessorMaxPollRPS),\n\t\tTransferProcessorCompleteTransferFailureRetryCount:   dc.GetIntProperty(dynamicproperties.TransferProcessorCompleteTransferFailureRetryCount),\n\t\tTransferProcessorMaxPollInterval:                     dc.GetDurationProperty(dynamicproperties.TransferProcessorMaxPollInterval),\n\t\tTransferProcessorMaxPollIntervalJitterCoefficient:    dc.GetFloat64Property(dynamicproperties.TransferProcessorMaxPollIntervalJitterCoefficient),\n\t\tTransferProcessorSplitQueueInterval:                  dc.GetDurationProperty(dynamicproperties.TransferProcessorSplitQueueInterval),\n\t\tTransferProcessorSplitQueueIntervalJitterCoefficient: dc.GetFloat64Property(dynamicproperties.TransferProcessorSplitQueueIntervalJitterCoefficient),\n\t\tTransferProcessorUpdateAckInterval:                   dc.GetDurationProperty(dynamicproperties.TransferProcessorUpdateAckInterval),\n\t\tTransferProcessorUpdateAckIntervalJitterCoefficient:  dc.GetFloat64Property(dynamicproperties.TransferProcessorUpdateAckIntervalJitterCoefficient),\n\t\tTransferProcessorCompleteTransferInterval:            dc.GetDurationProperty(dynamicproperties.TransferProcessorCompleteTransferInterval),\n\t\tTransferProcessorMaxRedispatchQueueSize:              dc.GetIntProperty(dynamicproperties.TransferProcessorMaxRedispatchQueueSize),\n\t\tTransferProcessorEnableValidator:                     dc.GetBoolProperty(dynamicproperties.TransferProcessorEnableValidator),\n\t\tTransferProcessorValidationInterval:                  dc.GetDurationProperty(dynamicproperties.TransferProcessorValidationInterval),\n\t\tTransferProcessorVisibilityArchivalTimeLimit:         dc.GetDurationProperty(dynamicproperties.TransferProcessorVisibilityArchivalTimeLimit),\n\t\tDisableTransferFailoverQueue:                         dc.GetBoolProperty(dynamicproperties.DisableTransferFailoverQueue),\n\n\t\tReplicatorTaskDeleteBatchSize:            dc.GetIntProperty(dynamicproperties.ReplicatorTaskDeleteBatchSize),\n\t\tReplicatorReadTaskMaxRetryCount:          dc.GetIntProperty(dynamicproperties.ReplicatorReadTaskMaxRetryCount),\n\t\tReplicatorProcessorFetchTasksBatchSize:   dc.GetIntPropertyFilteredByShardID(dynamicproperties.ReplicatorTaskBatchSize),\n\t\tReplicatorProcessorMaxTaskBatchSize:      dc.GetIntPropertyFilteredByShardID(dynamicproperties.ReplicatorMaxTaskBatchSize),\n\t\tReplicatorProcessorMinTaskBatchSize:      dc.GetIntPropertyFilteredByShardID(dynamicproperties.ReplicatorMinTaskBatchSize),\n\t\tReplicatorProcessorBatchSizeStepCount:    dc.GetIntPropertyFilteredByShardID(dynamicproperties.ReplicatorTaskBatchStepCount),\n\t\tReplicatorUpperLatency:                   dc.GetDurationProperty(dynamicproperties.ReplicatorUpperLatency),\n\t\tReplicatorCacheCapacity:                  dc.GetIntProperty(dynamicproperties.ReplicatorCacheCapacity),\n\t\tReplicatorCacheMaxSize:                   dc.GetIntProperty(dynamicproperties.ReplicatorCacheMaxSize),\n\t\tReplicationBudgetManagerMaxSizeBytes:     dc.GetIntProperty(dynamicproperties.ReplicationBudgetManagerMaxSizeBytes),\n\t\tReplicationBudgetManagerMaxSizeCount:     dc.GetIntProperty(dynamicproperties.ReplicationBudgetManagerMaxSizeCount),\n\t\tReplicationBudgetManagerSoftCapThreshold: dc.GetFloat64Property(dynamicproperties.ReplicationBudgetManagerSoftCapThreshold),\n\n\t\tMaximumBufferedEventsBatch:      dc.GetIntProperty(dynamicproperties.MaximumBufferedEventsBatch),\n\t\tMaximumSignalsPerExecution:      dc.GetIntPropertyFilteredByDomain(dynamicproperties.MaximumSignalsPerExecution),\n\t\tShardUpdateMinInterval:          dc.GetDurationProperty(dynamicproperties.ShardUpdateMinInterval),\n\t\tShardSyncMinInterval:            dc.GetDurationProperty(dynamicproperties.ShardSyncMinInterval),\n\t\tShardSyncTimerJitterCoefficient: dc.GetFloat64Property(dynamicproperties.TransferProcessorMaxPollIntervalJitterCoefficient),\n\n\t\t// history client: client/history/client.go set the client timeout 30s\n\t\tLongPollExpirationInterval:          dc.GetDurationPropertyFilteredByDomain(dynamicproperties.HistoryLongPollExpirationInterval),\n\t\tEventEncodingType:                   dc.GetStringPropertyFilteredByDomain(dynamicproperties.DefaultEventEncoding),\n\t\tEnableParentClosePolicy:             dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableParentClosePolicy),\n\t\tNumParentClosePolicySystemWorkflows: dc.GetIntProperty(dynamicproperties.NumParentClosePolicySystemWorkflows),\n\t\tEnableParentClosePolicyWorker:       dc.GetBoolProperty(dynamicproperties.EnableParentClosePolicyWorker),\n\t\tParentClosePolicyThreshold:          dc.GetIntPropertyFilteredByDomain(dynamicproperties.ParentClosePolicyThreshold),\n\t\tParentClosePolicyBatchSize:          dc.GetIntPropertyFilteredByDomain(dynamicproperties.ParentClosePolicyBatchSize),\n\n\t\tNumArchiveSystemWorkflows:        dc.GetIntProperty(dynamicproperties.NumArchiveSystemWorkflows),\n\t\tArchiveRequestRPS:                dc.GetIntProperty(dynamicproperties.ArchiveRequestRPS),\n\t\tArchiveInlineHistoryRPS:          dc.GetIntProperty(dynamicproperties.ArchiveInlineHistoryRPS),\n\t\tArchiveInlineHistoryGlobalRPS:    dc.GetIntProperty(dynamicproperties.ArchiveInlineHistoryGlobalRPS),\n\t\tArchiveInlineVisibilityRPS:       dc.GetIntProperty(dynamicproperties.ArchiveInlineVisibilityRPS),\n\t\tArchiveInlineVisibilityGlobalRPS: dc.GetIntProperty(dynamicproperties.ArchiveInlineVisibilityGlobalRPS),\n\t\tAllowArchivingIncompleteHistory:  dc.GetBoolProperty(dynamicproperties.AllowArchivingIncompleteHistory),\n\n\t\tBlobSizeLimitError:               dc.GetIntPropertyFilteredByDomain(dynamicproperties.BlobSizeLimitError),\n\t\tBlobSizeLimitWarn:                dc.GetIntPropertyFilteredByDomain(dynamicproperties.BlobSizeLimitWarn),\n\t\tHistorySizeLimitError:            dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistorySizeLimitError),\n\t\tHistorySizeLimitWarn:             dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistorySizeLimitWarn),\n\t\tHistoryCountLimitError:           dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistoryCountLimitError),\n\t\tHistoryCountLimitWarn:            dc.GetIntPropertyFilteredByDomain(dynamicproperties.HistoryCountLimitWarn),\n\t\tPendingActivitiesCountLimitError: dc.GetIntProperty(dynamicproperties.PendingActivitiesCountLimitError),\n\t\tPendingActivitiesCountLimitWarn:  dc.GetIntProperty(dynamicproperties.PendingActivitiesCountLimitWarn),\n\t\tPendingActivityValidationEnabled: dc.GetBoolProperty(dynamicproperties.EnablePendingActivityValidation),\n\n\t\tThrottledLogRPS:   dc.GetIntProperty(dynamicproperties.HistoryThrottledLogRPS),\n\t\tEnableStickyQuery: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableStickyQuery),\n\n\t\tEnableQueryAttributeValidation:           dc.GetBoolProperty(dynamicproperties.EnableQueryAttributeValidation),\n\t\tValidSearchAttributes:                    dc.GetMapProperty(dynamicproperties.ValidSearchAttributes),\n\t\tSearchAttributesNumberOfKeysLimit:        dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesNumberOfKeysLimit),\n\t\tSearchAttributesSizeOfValueLimit:         dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesSizeOfValueLimit),\n\t\tSearchAttributesTotalSizeLimit:           dc.GetIntPropertyFilteredByDomain(dynamicproperties.SearchAttributesTotalSizeLimit),\n\t\tSearchAttributesHiddenValueKeys:          dc.GetMapProperty(dynamicproperties.SearchAttributesHiddenValueKeys),\n\t\tStickyTTL:                                dc.GetDurationPropertyFilteredByDomain(dynamicproperties.StickyTTL),\n\t\tDecisionHeartbeatTimeout:                 dc.GetDurationPropertyFilteredByDomain(dynamicproperties.DecisionHeartbeatTimeout),\n\t\tDecisionRetryCriticalAttempts:            dc.GetIntProperty(dynamicproperties.DecisionRetryCriticalAttempts),\n\t\tDecisionRetryMaxAttempts:                 dc.GetIntPropertyFilteredByDomain(dynamicproperties.DecisionRetryMaxAttempts),\n\t\tEnforceDecisionTaskAttempts:              dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnforceDecisionTaskAttempts),\n\t\tNormalDecisionScheduleToStartMaxAttempts: dc.GetIntPropertyFilteredByDomain(dynamicproperties.NormalDecisionScheduleToStartMaxAttempts),\n\t\tNormalDecisionScheduleToStartTimeout:     dc.GetDurationPropertyFilteredByDomain(dynamicproperties.NormalDecisionScheduleToStartTimeout),\n\n\t\tReplicationTaskFetcherParallelism:                    dc.GetIntProperty(dynamicproperties.ReplicationTaskFetcherParallelism),\n\t\tReplicationTaskFetcherAggregationInterval:            dc.GetDurationProperty(dynamicproperties.ReplicationTaskFetcherAggregationInterval),\n\t\tReplicationTaskFetcherTimerJitterCoefficient:         dc.GetFloat64Property(dynamicproperties.ReplicationTaskFetcherTimerJitterCoefficient),\n\t\tReplicationTaskFetcherErrorRetryWait:                 dc.GetDurationProperty(dynamicproperties.ReplicationTaskFetcherErrorRetryWait),\n\t\tReplicationTaskFetcherServiceBusyWait:                dc.GetDurationProperty(dynamicproperties.ReplicationTaskFetcherServiceBusyWait),\n\t\tReplicationTaskProcessorErrorRetryWait:               dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorErrorRetryWait),\n\t\tReplicationTaskProcessorErrorRetryMaxAttempts:        dc.GetIntPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorErrorRetryMaxAttempts),\n\t\tReplicationTaskProcessorErrorSecondRetryWait:         dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorErrorSecondRetryWait),\n\t\tReplicationTaskProcessorErrorSecondRetryMaxWait:      dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorErrorSecondRetryMaxWait),\n\t\tReplicationTaskProcessorErrorSecondRetryExpiration:   dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorErrorSecondRetryExpiration),\n\t\tReplicationTaskProcessorNoTaskRetryWait:              dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorNoTaskInitialWait),\n\t\tReplicationTaskProcessorCleanupInterval:              dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorCleanupInterval),\n\t\tReplicationTaskProcessorCleanupJitterCoefficient:     dc.GetFloat64PropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorCleanupJitterCoefficient),\n\t\tReplicationTaskProcessorStartWait:                    dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorStartWait),\n\t\tReplicationTaskProcessorStartWaitJitterCoefficient:   dc.GetFloat64PropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorStartWaitJitterCoefficient),\n\t\tReplicationTaskProcessorHostQPS:                      dc.GetFloat64Property(dynamicproperties.ReplicationTaskProcessorHostQPS),\n\t\tReplicationTaskProcessorShardQPS:                     dc.GetFloat64Property(dynamicproperties.ReplicationTaskProcessorShardQPS),\n\t\tReplicationTaskGenerationQPS:                         dc.GetFloat64Property(dynamicproperties.ReplicationTaskGenerationQPS),\n\t\tEnableReplicationTaskGeneration:                      dc.GetBoolPropertyFilteredByDomainIDAndWorkflowID(dynamicproperties.EnableReplicationTaskGeneration),\n\t\tEnableRecordWorkflowExecutionUninitialized:           dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableRecordWorkflowExecutionUninitialized),\n\t\tEnableCleanupOrphanedHistoryBranchOnWorkflowCreation: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableCleanupOrphanedHistoryBranchOnWorkflowCreation),\n\t\tReplicationTaskProcessorLatencyLogThreshold:          dc.GetDurationProperty(dynamicproperties.ReplicationTaskProcessorLatencyLogThreshold),\n\n\t\tWorkflowIDExternalRPS: dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowIDExternalRPS),\n\t\tWorkflowIDInternalRPS: dc.GetIntPropertyFilteredByDomain(dynamicproperties.WorkflowIDInternalRPS),\n\n\t\tEnableConsistentQuery:                 dc.GetBoolProperty(dynamicproperties.EnableConsistentQuery),\n\t\tEnableConsistentQueryByDomain:         dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableConsistentQueryByDomain),\n\t\tEnableContextHeaderInVisibility:       dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableContextHeaderInVisibility),\n\t\tEnableCrossClusterOperationsForDomain: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableCrossClusterOperationsForDomain),\n\t\tMaxBufferedQueryCount:                 dc.GetIntProperty(dynamicproperties.MaxBufferedQueryCount),\n\t\tMutableStateChecksumGenProbability:    dc.GetIntPropertyFilteredByDomain(dynamicproperties.MutableStateChecksumGenProbability),\n\t\tMutableStateChecksumVerifyProbability: dc.GetIntPropertyFilteredByDomain(dynamicproperties.MutableStateChecksumVerifyProbability),\n\t\tMutableStateChecksumInvalidateBefore:  dc.GetFloat64Property(dynamicproperties.MutableStateChecksumInvalidateBefore),\n\t\tEnableRetryForChecksumFailure:         dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableRetryForChecksumFailure),\n\n\t\tEnableHistoryCorruptionCheck: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableHistoryCorruptionCheck),\n\n\t\tNotifyFailoverMarkerInterval:               dc.GetDurationProperty(dynamicproperties.NotifyFailoverMarkerInterval),\n\t\tNotifyFailoverMarkerTimerJitterCoefficient: dc.GetFloat64Property(dynamicproperties.NotifyFailoverMarkerTimerJitterCoefficient),\n\t\tEnableGracefulFailover:                     dc.GetBoolProperty(dynamicproperties.EnableGracefulFailover),\n\n\t\tEnableActivityLocalDispatchByDomain: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableActivityLocalDispatchByDomain),\n\t\tMaxActivityCountDispatchByDomain:    dc.GetIntPropertyFilteredByDomain(dynamicproperties.MaxActivityCountDispatchByDomain),\n\n\t\tActivityMaxScheduleToStartTimeoutForRetry: dc.GetDurationPropertyFilteredByDomain(dynamicproperties.ActivityMaxScheduleToStartTimeoutForRetry),\n\n\t\tEnableDebugMode:               dc.GetBoolProperty(dynamicproperties.EnableDebugMode)(),\n\t\tEnableTaskInfoLogByDomainID:   dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.HistoryEnableTaskInfoLogByDomainID),\n\t\tEnableTimerDebugLogByDomainID: dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.EnableTimerDebugLogByDomainID),\n\n\t\tSampleLoggingRate:                     dc.GetIntProperty(dynamicproperties.SampleLoggingRate),\n\t\tEnableShardIDMetrics:                  dc.GetBoolProperty(dynamicproperties.EnableShardIDMetrics),\n\t\tLargeShardHistorySizeMetricThreshold:  dc.GetIntProperty(dynamicproperties.LargeShardHistorySizeMetricThreshold),\n\t\tLargeShardHistoryEventMetricThreshold: dc.GetIntProperty(dynamicproperties.LargeShardHistoryEventMetricThreshold),\n\t\tLargeShardHistoryBlobMetricThreshold:  dc.GetIntProperty(dynamicproperties.LargeShardHistoryBlobMetricThreshold),\n\n\t\tEnableStrongIdempotency:            dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableStrongIdempotency),\n\t\tEnableStrongIdempotencySanityCheck: dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableStrongIdempotencySanityCheck),\n\n\t\tGlobalRatelimiterNewDataWeight:  dc.GetFloat64Property(dynamicproperties.HistoryGlobalRatelimiterNewDataWeight),\n\t\tGlobalRatelimiterUpdateInterval: dc.GetDurationProperty(dynamicproperties.GlobalRatelimiterUpdateInterval),\n\t\tGlobalRatelimiterDecayAfter:     dc.GetDurationProperty(dynamicproperties.HistoryGlobalRatelimiterDecayAfter),\n\t\tGlobalRatelimiterGCAfter:        dc.GetDurationProperty(dynamicproperties.HistoryGlobalRatelimiterGCAfter),\n\n\t\tHostName: hostname,\n\t}\n\n\treturn cfg\n}\n\n// NewForTest create new history service config for test\nfunc NewForTest() *Config {\n\treturn NewForTestByShardNumber(1)\n}\n\n// NewForTestByShardNumber create new history service config for test\nfunc NewForTestByShardNumber(shardNumber int) *Config {\n\tpanicIfErr := func(err error) {\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\t}\n\tinMem := dynamicconfig.NewInMemoryClient()\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.HistoryLongPollExpirationInterval, 10*time.Second))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.EnableConsistentQueryByDomain, true))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ReplicationTaskProcessorHostQPS, float64(10000)))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ReplicationTaskProcessorCleanupInterval, 20*time.Millisecond))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ReplicatorTaskDeleteBatchSize, 50))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ShardSyncMinInterval, 20*time.Millisecond))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ReplicationTaskProcessorShardQPS, float64(10000)))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.ReplicationTaskProcessorStartWait, time.Nanosecond))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.EnableActivityLocalDispatchByDomain, true))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.MaxActivityCountDispatchByDomain, 0))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.EnableCrossClusterOperationsForDomain, true))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.NormalDecisionScheduleToStartMaxAttempts, 3))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.EnablePendingActivityValidation, true))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.QueueProcessorEnableGracefulSyncShutdown, true))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.QueueProcessorSplitMaxLevel, 2))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.QueueProcessorPendingTaskSplitThreshold, map[string]interface{}{\n\t\t\"0\": 1000,\n\t\t\"1\": 5000,\n\t}))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.QueueProcessorStuckTaskSplitThreshold, map[string]interface{}{\n\t\t\"0\": 10,\n\t\t\"1\": 50,\n\t}))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.QueueProcessorRandomSplitProbability, 0.5))\n\tpanicIfErr(inMem.UpdateValue(dynamicproperties.EnableStrongIdempotency, true))\n\n\tdc := dynamicconfig.NewCollection(inMem, log.NewNoop())\n\tconfig := New(dc, shardNumber, 1024*1024, false, \"\")\n\t// reduce the duration of long poll to increase test speed\n\tconfig.LongPollExpirationInterval = dc.GetDurationPropertyFilteredByDomain(dynamicproperties.HistoryLongPollExpirationInterval)\n\tconfig.EnableConsistentQueryByDomain = dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableConsistentQueryByDomain)\n\tconfig.ReplicationTaskProcessorHostQPS = dc.GetFloat64Property(dynamicproperties.ReplicationTaskProcessorHostQPS)\n\tconfig.ReplicationTaskProcessorCleanupInterval = dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorCleanupInterval)\n\tconfig.ReplicatorTaskDeleteBatchSize = dc.GetIntProperty(dynamicproperties.ReplicatorTaskDeleteBatchSize)\n\tconfig.ShardSyncMinInterval = dc.GetDurationProperty(dynamicproperties.ShardSyncMinInterval)\n\tconfig.ReplicationTaskProcessorShardQPS = dc.GetFloat64Property(dynamicproperties.ReplicationTaskProcessorShardQPS)\n\tconfig.ReplicationTaskProcessorStartWait = dc.GetDurationPropertyFilteredByShardID(dynamicproperties.ReplicationTaskProcessorStartWait)\n\tconfig.EnableActivityLocalDispatchByDomain = dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableActivityLocalDispatchByDomain)\n\tconfig.MaxActivityCountDispatchByDomain = dc.GetIntPropertyFilteredByDomain(dynamicproperties.MaxActivityCountDispatchByDomain)\n\tconfig.EnableCrossClusterOperationsForDomain = dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableCrossClusterOperationsForDomain)\n\tconfig.NormalDecisionScheduleToStartMaxAttempts = dc.GetIntPropertyFilteredByDomain(dynamicproperties.NormalDecisionScheduleToStartMaxAttempts)\n\tconfig.NormalDecisionScheduleToStartTimeout = dc.GetDurationPropertyFilteredByDomain(dynamicproperties.NormalDecisionScheduleToStartTimeout)\n\tconfig.PendingActivityValidationEnabled = dc.GetBoolProperty(dynamicproperties.EnablePendingActivityValidation)\n\tconfig.QueueProcessorEnableGracefulSyncShutdown = dc.GetBoolProperty(dynamicproperties.QueueProcessorEnableGracefulSyncShutdown)\n\tconfig.QueueProcessorSplitMaxLevel = dc.GetIntProperty(dynamicproperties.QueueProcessorSplitMaxLevel)\n\tconfig.QueueProcessorPendingTaskSplitThreshold = dc.GetMapProperty(dynamicproperties.QueueProcessorPendingTaskSplitThreshold)\n\tconfig.QueueProcessorStuckTaskSplitThreshold = dc.GetMapProperty(dynamicproperties.QueueProcessorStuckTaskSplitThreshold)\n\tconfig.QueueProcessorRandomSplitProbability = dc.GetFloat64Property(dynamicproperties.QueueProcessorRandomSplitProbability)\n\treturn config\n}\n\n// GetShardID return the corresponding shard ID for a given workflow ID\nfunc (config *Config) GetShardID(workflowID string) int {\n\treturn common.WorkflowIDToHistoryShard(workflowID, config.NumberOfShards)\n}\n"
  },
  {
    "path": "service/history/config/config_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype configTestCase struct {\n\tkey   dynamicproperties.Key\n\tvalue interface{}\n}\n\nfunc TestNewConfig(t *testing.T) {\n\thostname := \"hostname\"\n\tnumberOfShards := 8192\n\tmaxMessageSize := 1024\n\tisAdvancedVisConfigExist := true\n\tfields := map[string]configTestCase{\n\t\t\"NumberOfShards\":                                       {nil, numberOfShards},\n\t\t\"IsAdvancedVisConfigExist\":                             {nil, isAdvancedVisConfigExist},\n\t\t\"RPS\":                                                  {dynamicproperties.HistoryRPS, 1},\n\t\t\"MaxIDLengthWarnLimit\":                                 {dynamicproperties.MaxIDLengthWarnLimit, 2},\n\t\t\"DomainNameMaxLength\":                                  {dynamicproperties.DomainNameMaxLength, 3},\n\t\t\"IdentityMaxLength\":                                    {dynamicproperties.IdentityMaxLength, 4},\n\t\t\"WorkflowIDMaxLength\":                                  {dynamicproperties.WorkflowIDMaxLength, 5},\n\t\t\"SignalNameMaxLength\":                                  {dynamicproperties.SignalNameMaxLength, 6},\n\t\t\"WorkflowTypeMaxLength\":                                {dynamicproperties.WorkflowTypeMaxLength, 7},\n\t\t\"RequestIDMaxLength\":                                   {dynamicproperties.RequestIDMaxLength, 8},\n\t\t\"TaskListNameMaxLength\":                                {dynamicproperties.TaskListNameMaxLength, 9},\n\t\t\"ActivityIDMaxLength\":                                  {dynamicproperties.ActivityIDMaxLength, 10},\n\t\t\"ActivityTypeMaxLength\":                                {dynamicproperties.ActivityTypeMaxLength, 11},\n\t\t\"MarkerNameMaxLength\":                                  {dynamicproperties.MarkerNameMaxLength, 12},\n\t\t\"TimerIDMaxLength\":                                     {dynamicproperties.TimerIDMaxLength, 13},\n\t\t\"PersistenceMaxQPS\":                                    {dynamicproperties.HistoryPersistenceMaxQPS, 14},\n\t\t\"PersistenceGlobalMaxQPS\":                              {dynamicproperties.HistoryPersistenceGlobalMaxQPS, 15},\n\t\t\"EnableVisibilitySampling\":                             {dynamicproperties.EnableVisibilitySampling, true},\n\t\t\"EnableReadFromClosedExecutionV2\":                      {dynamicproperties.EnableReadFromClosedExecutionV2, true},\n\t\t\"VisibilityOpenMaxQPS\":                                 {dynamicproperties.HistoryVisibilityOpenMaxQPS, 16},\n\t\t\"VisibilityClosedMaxQPS\":                               {dynamicproperties.HistoryVisibilityClosedMaxQPS, 17},\n\t\t\"WriteVisibilityStoreName\":                             {dynamicproperties.WriteVisibilityStoreName, \"es\"},\n\t\t\"EmitShardDiffLog\":                                     {dynamicproperties.EmitShardDiffLog, true},\n\t\t\"MaxAutoResetPoints\":                                   {dynamicproperties.HistoryMaxAutoResetPoints, 18},\n\t\t\"ThrottledLogRPS\":                                      {dynamicproperties.HistoryThrottledLogRPS, 19},\n\t\t\"EnableStickyQuery\":                                    {dynamicproperties.EnableStickyQuery, true},\n\t\t\"ShutdownDrainDuration\":                                {dynamicproperties.HistoryShutdownDrainDuration, time.Second},\n\t\t\"WorkflowDeletionJitterRange\":                          {dynamicproperties.WorkflowDeletionJitterRange, 20},\n\t\t\"DeleteHistoryEventContextTimeout\":                     {dynamicproperties.DeleteHistoryEventContextTimeout, 21},\n\t\t\"MaxResponseSize\":                                      {nil, maxMessageSize},\n\t\t\"HistoryCacheInitialSize\":                              {dynamicproperties.HistoryCacheInitialSize, 22},\n\t\t\"HistoryCacheMaxSize\":                                  {dynamicproperties.HistoryCacheMaxSize, 23},\n\t\t\"HistoryCacheTTL\":                                      {dynamicproperties.HistoryCacheTTL, time.Second},\n\t\t\"EnableSizeBasedHistoryExecutionCache\":                 {dynamicproperties.EnableSizeBasedHistoryExecutionCache, true},\n\t\t\"EventsCacheInitialCount\":                              {dynamicproperties.EventsCacheInitialCount, 24},\n\t\t\"EventsCacheMaxCount\":                                  {dynamicproperties.EventsCacheMaxCount, 25},\n\t\t\"EventsCacheMaxSize\":                                   {dynamicproperties.EventsCacheMaxSize, 26},\n\t\t\"EventsCacheTTL\":                                       {dynamicproperties.EventsCacheTTL, time.Second},\n\t\t\"EventsCacheGlobalEnable\":                              {dynamicproperties.EventsCacheGlobalEnable, true},\n\t\t\"EventsCacheGlobalInitialCount\":                        {dynamicproperties.EventsCacheGlobalInitialCount, 27},\n\t\t\"EventsCacheGlobalMaxCount\":                            {dynamicproperties.EventsCacheGlobalMaxCount, 28},\n\t\t\"EnableSizeBasedHistoryEventCache\":                     {dynamicproperties.EnableSizeBasedHistoryEventCache, true},\n\t\t\"RangeSizeBits\":                                        {nil, uint(20)},\n\t\t\"AcquireShardInterval\":                                 {dynamicproperties.AcquireShardInterval, time.Second},\n\t\t\"AcquireShardConcurrency\":                              {dynamicproperties.AcquireShardConcurrency, 29},\n\t\t\"StandbyClusterDelay\":                                  {dynamicproperties.StandbyClusterDelay, time.Second},\n\t\t\"StandbyTaskMissingEventsResendDelay\":                  {dynamicproperties.StandbyTaskMissingEventsResendDelay, time.Second},\n\t\t\"StandbyTaskMissingEventsDiscardDelay\":                 {dynamicproperties.StandbyTaskMissingEventsDiscardDelay, time.Second},\n\t\t\"TaskProcessRPS\":                                       {dynamicproperties.TaskProcessRPS, 30},\n\t\t\"TaskSchedulerType\":                                    {dynamicproperties.TaskSchedulerType, 31},\n\t\t\"TaskSchedulerWorkerCount\":                             {dynamicproperties.TaskSchedulerWorkerCount, 32},\n\t\t\"TaskSchedulerQueueSize\":                               {dynamicproperties.TaskSchedulerQueueSize, 34},\n\t\t\"TaskSchedulerDispatcherCount\":                         {dynamicproperties.TaskSchedulerDispatcherCount, 35},\n\t\t\"TaskSchedulerRoundRobinWeights\":                       {dynamicproperties.TaskSchedulerRoundRobinWeights, map[string]interface{}{\"key\": 1}},\n\t\t\"TaskSchedulerDomainRoundRobinWeights\":                 {dynamicproperties.TaskSchedulerDomainRoundRobinWeights, map[string]interface{}{\"key\": 2}},\n\t\t\"TaskCriticalRetryCount\":                               {dynamicproperties.TaskCriticalRetryCount, 37},\n\t\t\"ActiveTaskRedispatchInterval\":                         {dynamicproperties.ActiveTaskRedispatchInterval, time.Second},\n\t\t\"StandbyTaskRedispatchInterval\":                        {dynamicproperties.StandbyTaskRedispatchInterval, time.Second},\n\t\t\"StandbyTaskReReplicationContextTimeout\":               {dynamicproperties.StandbyTaskReReplicationContextTimeout, time.Second},\n\t\t\"EnableDropStuckTaskByDomainID\":                        {dynamicproperties.EnableDropStuckTaskByDomainID, true},\n\t\t\"ResurrectionCheckMinDelay\":                            {dynamicproperties.ResurrectionCheckMinDelay, time.Second},\n\t\t\"QueueProcessorEnableSplit\":                            {dynamicproperties.QueueProcessorEnableSplit, true},\n\t\t\"QueueProcessorSplitMaxLevel\":                          {dynamicproperties.QueueProcessorSplitMaxLevel, 38},\n\t\t\"QueueProcessorEnableRandomSplitByDomainID\":            {dynamicproperties.QueueProcessorEnableRandomSplitByDomainID, true},\n\t\t\"QueueProcessorRandomSplitProbability\":                 {dynamicproperties.QueueProcessorRandomSplitProbability, 1.0},\n\t\t\"QueueProcessorEnablePendingTaskSplitByDomainID\":       {dynamicproperties.QueueProcessorEnablePendingTaskSplitByDomainID, true},\n\t\t\"QueueProcessorPendingTaskSplitThreshold\":              {dynamicproperties.QueueProcessorPendingTaskSplitThreshold, map[string]interface{}{\"a\": 100}},\n\t\t\"QueueProcessorEnableStuckTaskSplitByDomainID\":         {dynamicproperties.QueueProcessorEnableStuckTaskSplitByDomainID, true},\n\t\t\"QueueProcessorStuckTaskSplitThreshold\":                {dynamicproperties.QueueProcessorStuckTaskSplitThreshold, map[string]interface{}{\"b\": 1}},\n\t\t\"QueueProcessorSplitLookAheadDurationByDomainID\":       {dynamicproperties.QueueProcessorSplitLookAheadDurationByDomainID, time.Second},\n\t\t\"QueueProcessorPollBackoffInterval\":                    {dynamicproperties.QueueProcessorPollBackoffInterval, time.Second},\n\t\t\"QueueProcessorPollBackoffIntervalJitterCoefficient\":   {dynamicproperties.QueueProcessorPollBackoffIntervalJitterCoefficient, 1.0},\n\t\t\"QueueProcessorEnableLoadQueueStates\":                  {dynamicproperties.QueueProcessorEnableLoadQueueStates, true},\n\t\t\"QueueProcessorEnableGracefulSyncShutdown\":             {dynamicproperties.QueueProcessorEnableGracefulSyncShutdown, true},\n\t\t\"QueueProcessorEnablePersistQueueStates\":               {dynamicproperties.QueueProcessorEnablePersistQueueStates, true},\n\t\t\"TimerTaskBatchSize\":                                   {dynamicproperties.TimerTaskBatchSize, 39},\n\t\t\"TimerTaskDeleteBatchSize\":                             {dynamicproperties.TimerTaskDeleteBatchSize, 40},\n\t\t\"TimerProcessorGetFailureRetryCount\":                   {dynamicproperties.TimerProcessorGetFailureRetryCount, 41},\n\t\t\"TimerProcessorCompleteTimerFailureRetryCount\":         {dynamicproperties.TimerProcessorCompleteTimerFailureRetryCount, 42},\n\t\t\"TimerProcessorUpdateAckInterval\":                      {dynamicproperties.TimerProcessorUpdateAckInterval, time.Second},\n\t\t\"TimerProcessorUpdateAckIntervalJitterCoefficient\":     {dynamicproperties.TimerProcessorUpdateAckIntervalJitterCoefficient, 2.0},\n\t\t\"TimerProcessorCompleteTimerInterval\":                  {dynamicproperties.TimerProcessorCompleteTimerInterval, time.Second},\n\t\t\"TimerProcessorFailoverMaxStartJitterInterval\":         {dynamicproperties.TimerProcessorFailoverMaxStartJitterInterval, time.Second},\n\t\t\"TimerProcessorFailoverMaxPollRPS\":                     {dynamicproperties.TimerProcessorFailoverMaxPollRPS, 43},\n\t\t\"TimerProcessorMaxPollRPS\":                             {dynamicproperties.TimerProcessorMaxPollRPS, 44},\n\t\t\"TimerProcessorMaxPollInterval\":                        {dynamicproperties.TimerProcessorMaxPollInterval, time.Second},\n\t\t\"TimerProcessorMaxPollIntervalJitterCoefficient\":       {dynamicproperties.TimerProcessorMaxPollIntervalJitterCoefficient, 3.0},\n\t\t\"TimerProcessorSplitQueueInterval\":                     {dynamicproperties.TimerProcessorSplitQueueInterval, time.Second},\n\t\t\"TimerProcessorSplitQueueIntervalJitterCoefficient\":    {dynamicproperties.TimerProcessorSplitQueueIntervalJitterCoefficient, 4.0},\n\t\t\"TimerProcessorMaxRedispatchQueueSize\":                 {dynamicproperties.TimerProcessorMaxRedispatchQueueSize, 45},\n\t\t\"TimerProcessorMaxTimeShift\":                           {dynamicproperties.TimerProcessorMaxTimeShift, time.Second},\n\t\t\"TimerProcessorHistoryArchivalSizeLimit\":               {dynamicproperties.TimerProcessorHistoryArchivalSizeLimit, 46},\n\t\t\"TimerProcessorArchivalTimeLimit\":                      {dynamicproperties.TimerProcessorArchivalTimeLimit, time.Second},\n\t\t\"TransferTaskBatchSize\":                                {dynamicproperties.TransferTaskBatchSize, 47},\n\t\t\"TransferTaskDeleteBatchSize\":                          {dynamicproperties.TransferTaskDeleteBatchSize, 48},\n\t\t\"TransferProcessorCompleteTransferFailureRetryCount\":   {dynamicproperties.TransferProcessorCompleteTransferFailureRetryCount, 49},\n\t\t\"TransferProcessorFailoverMaxStartJitterInterval\":      {dynamicproperties.TransferProcessorFailoverMaxStartJitterInterval, time.Second},\n\t\t\"TransferProcessorFailoverMaxPollRPS\":                  {dynamicproperties.TransferProcessorFailoverMaxPollRPS, 50},\n\t\t\"TransferProcessorMaxPollRPS\":                          {dynamicproperties.TransferProcessorMaxPollRPS, 51},\n\t\t\"TransferProcessorMaxPollInterval\":                     {dynamicproperties.TransferProcessorMaxPollInterval, time.Second},\n\t\t\"TransferProcessorMaxPollIntervalJitterCoefficient\":    {dynamicproperties.TransferProcessorMaxPollIntervalJitterCoefficient, 8.0},\n\t\t\"TransferProcessorSplitQueueInterval\":                  {dynamicproperties.TransferProcessorSplitQueueInterval, time.Second},\n\t\t\"TransferProcessorSplitQueueIntervalJitterCoefficient\": {dynamicproperties.TransferProcessorSplitQueueIntervalJitterCoefficient, 6.0},\n\t\t\"TransferProcessorUpdateAckInterval\":                   {dynamicproperties.TransferProcessorUpdateAckInterval, time.Second},\n\t\t\"TransferProcessorUpdateAckIntervalJitterCoefficient\":  {dynamicproperties.TransferProcessorUpdateAckIntervalJitterCoefficient, 7.0},\n\t\t\"TransferProcessorCompleteTransferInterval\":            {dynamicproperties.TransferProcessorCompleteTransferInterval, time.Second},\n\t\t\"TransferProcessorMaxRedispatchQueueSize\":              {dynamicproperties.TransferProcessorMaxRedispatchQueueSize, 52},\n\t\t\"TransferProcessorEnableValidator\":                     {dynamicproperties.TransferProcessorEnableValidator, true},\n\t\t\"TransferProcessorValidationInterval\":                  {dynamicproperties.TransferProcessorValidationInterval, time.Second},\n\t\t\"TransferProcessorVisibilityArchivalTimeLimit\":         {dynamicproperties.TransferProcessorVisibilityArchivalTimeLimit, time.Second},\n\t\t\"ReplicatorTaskDeleteBatchSize\":                        {dynamicproperties.ReplicatorTaskDeleteBatchSize, 53},\n\t\t\"ReplicatorReadTaskMaxRetryCount\":                      {dynamicproperties.ReplicatorReadTaskMaxRetryCount, 54},\n\t\t\"ReplicatorProcessorFetchTasksBatchSize\":               {dynamicproperties.ReplicatorTaskBatchSize, 55},\n\t\t\"ReplicatorProcessorMinTaskBatchSize\":                  {dynamicproperties.ReplicatorMinTaskBatchSize, 1},\n\t\t\"ReplicatorProcessorMaxTaskBatchSize\":                  {dynamicproperties.ReplicatorMaxTaskBatchSize, 1000},\n\t\t\"ReplicatorProcessorBatchSizeStepCount\":                {dynamicproperties.ReplicatorTaskBatchStepCount, 10},\n\t\t\"ReplicatorUpperLatency\":                               {dynamicproperties.ReplicatorUpperLatency, time.Second},\n\t\t\"ReplicatorCacheCapacity\":                              {dynamicproperties.ReplicatorCacheCapacity, 56},\n\t\t\"ReplicatorCacheMaxSize\":                               {dynamicproperties.ReplicatorCacheMaxSize, 2000},\n\t\t\"ReplicationBudgetManagerMaxSizeBytes\":                 {dynamicproperties.ReplicationBudgetManagerMaxSizeBytes, 0},\n\t\t\"ReplicationBudgetManagerMaxSizeCount\":                 {dynamicproperties.ReplicationBudgetManagerMaxSizeCount, 0},\n\t\t\"ReplicationBudgetManagerSoftCapThreshold\":             {dynamicproperties.ReplicationBudgetManagerSoftCapThreshold, 1.0},\n\t\t\"MaximumBufferedEventsBatch\":                           {dynamicproperties.MaximumBufferedEventsBatch, 59},\n\t\t\"MaximumSignalsPerExecution\":                           {dynamicproperties.MaximumSignalsPerExecution, 60},\n\t\t\"ShardUpdateMinInterval\":                               {dynamicproperties.ShardUpdateMinInterval, time.Second},\n\t\t\"ShardSyncMinInterval\":                                 {dynamicproperties.ShardSyncMinInterval, time.Second},\n\t\t\"ShardSyncTimerJitterCoefficient\":                      {dynamicproperties.TransferProcessorMaxPollIntervalJitterCoefficient, 8.0},\n\t\t\"LongPollExpirationInterval\":                           {dynamicproperties.HistoryLongPollExpirationInterval, time.Second},\n\t\t\"EventEncodingType\":                                    {dynamicproperties.DefaultEventEncoding, \"eventEncodingType\"},\n\t\t\"EnableParentClosePolicy\":                              {dynamicproperties.EnableParentClosePolicy, true},\n\t\t\"EnableParentClosePolicyWorker\":                        {dynamicproperties.EnableParentClosePolicyWorker, true},\n\t\t\"ParentClosePolicyThreshold\":                           {dynamicproperties.ParentClosePolicyThreshold, 61},\n\t\t\"ParentClosePolicyBatchSize\":                           {dynamicproperties.ParentClosePolicyBatchSize, 62},\n\t\t\"NumParentClosePolicySystemWorkflows\":                  {dynamicproperties.NumParentClosePolicySystemWorkflows, 63},\n\t\t\"NumArchiveSystemWorkflows\":                            {dynamicproperties.NumArchiveSystemWorkflows, 64},\n\t\t\"ArchiveRequestRPS\":                                    {dynamicproperties.ArchiveRequestRPS, 65},\n\t\t\"ArchiveInlineHistoryRPS\":                              {dynamicproperties.ArchiveInlineHistoryRPS, 66},\n\t\t\"ArchiveInlineHistoryGlobalRPS\":                        {dynamicproperties.ArchiveInlineHistoryGlobalRPS, 67},\n\t\t\"ArchiveInlineVisibilityRPS\":                           {dynamicproperties.ArchiveInlineVisibilityRPS, 68},\n\t\t\"ArchiveInlineVisibilityGlobalRPS\":                     {dynamicproperties.ArchiveInlineVisibilityGlobalRPS, 69},\n\t\t\"AllowArchivingIncompleteHistory\":                      {dynamicproperties.AllowArchivingIncompleteHistory, true},\n\t\t\"BlobSizeLimitError\":                                   {dynamicproperties.BlobSizeLimitError, 70},\n\t\t\"BlobSizeLimitWarn\":                                    {dynamicproperties.BlobSizeLimitWarn, 71},\n\t\t\"HistorySizeLimitError\":                                {dynamicproperties.HistorySizeLimitError, 72},\n\t\t\"HistorySizeLimitWarn\":                                 {dynamicproperties.HistorySizeLimitWarn, 73},\n\t\t\"HistoryCountLimitError\":                               {dynamicproperties.HistoryCountLimitError, 74},\n\t\t\"HistoryCountLimitWarn\":                                {dynamicproperties.HistoryCountLimitWarn, 75},\n\t\t\"PendingActivitiesCountLimitError\":                     {dynamicproperties.PendingActivitiesCountLimitError, 76},\n\t\t\"PendingActivitiesCountLimitWarn\":                      {dynamicproperties.PendingActivitiesCountLimitWarn, 77},\n\t\t\"PendingActivityValidationEnabled\":                     {dynamicproperties.EnablePendingActivityValidation, true},\n\t\t\"EnableQueryAttributeValidation\":                       {dynamicproperties.EnableQueryAttributeValidation, true},\n\t\t\"ValidSearchAttributes\":                                {dynamicproperties.ValidSearchAttributes, map[string]interface{}{\"key\": 1}},\n\t\t\"SearchAttributesNumberOfKeysLimit\":                    {dynamicproperties.SearchAttributesNumberOfKeysLimit, 78},\n\t\t\"SearchAttributesSizeOfValueLimit\":                     {dynamicproperties.SearchAttributesSizeOfValueLimit, 79},\n\t\t\"SearchAttributesTotalSizeLimit\":                       {dynamicproperties.SearchAttributesTotalSizeLimit, 80},\n\t\t\"StickyTTL\":                                            {dynamicproperties.StickyTTL, time.Second},\n\t\t\"DecisionHeartbeatTimeout\":                             {dynamicproperties.DecisionHeartbeatTimeout, time.Second},\n\t\t\"MaxDecisionStartToCloseSeconds\":                       {dynamicproperties.MaxDecisionStartToCloseSeconds, 81},\n\t\t\"DecisionRetryCriticalAttempts\":                        {dynamicproperties.DecisionRetryCriticalAttempts, 82},\n\t\t\"DecisionRetryMaxAttempts\":                             {dynamicproperties.DecisionRetryMaxAttempts, 83},\n\t\t\"EnforceDecisionTaskAttempts\":                          {dynamicproperties.EnforceDecisionTaskAttempts, true},\n\t\t\"NormalDecisionScheduleToStartMaxAttempts\":             {dynamicproperties.NormalDecisionScheduleToStartMaxAttempts, 84},\n\t\t\"NormalDecisionScheduleToStartTimeout\":                 {dynamicproperties.NormalDecisionScheduleToStartTimeout, time.Second},\n\t\t\"ReplicationTaskFetcherParallelism\":                    {dynamicproperties.ReplicationTaskFetcherParallelism, 85},\n\t\t\"ReplicationTaskFetcherAggregationInterval\":            {dynamicproperties.ReplicationTaskFetcherAggregationInterval, time.Second},\n\t\t\"ReplicationTaskFetcherTimerJitterCoefficient\":         {dynamicproperties.ReplicationTaskFetcherTimerJitterCoefficient, 9.0},\n\t\t\"ReplicationTaskFetcherErrorRetryWait\":                 {dynamicproperties.ReplicationTaskFetcherErrorRetryWait, time.Second},\n\t\t\"ReplicationTaskFetcherServiceBusyWait\":                {dynamicproperties.ReplicationTaskFetcherServiceBusyWait, time.Second},\n\t\t\"ReplicationTaskProcessorErrorRetryWait\":               {dynamicproperties.ReplicationTaskProcessorErrorRetryWait, time.Second},\n\t\t\"ReplicationTaskProcessorErrorRetryMaxAttempts\":        {dynamicproperties.ReplicationTaskProcessorErrorRetryMaxAttempts, 86},\n\t\t\"ReplicationTaskProcessorErrorSecondRetryWait\":         {dynamicproperties.ReplicationTaskProcessorErrorSecondRetryWait, time.Second},\n\t\t\"ReplicationTaskProcessorErrorSecondRetryExpiration\":   {dynamicproperties.ReplicationTaskProcessorErrorSecondRetryExpiration, time.Second},\n\t\t\"ReplicationTaskProcessorErrorSecondRetryMaxWait\":      {dynamicproperties.ReplicationTaskProcessorErrorSecondRetryMaxWait, time.Second},\n\t\t\"ReplicationTaskProcessorNoTaskRetryWait\":              {dynamicproperties.ReplicationTaskProcessorNoTaskInitialWait, time.Second},\n\t\t\"ReplicationTaskProcessorCleanupInterval\":              {dynamicproperties.ReplicationTaskProcessorCleanupInterval, time.Second},\n\t\t\"ReplicationTaskProcessorCleanupJitterCoefficient\":     {dynamicproperties.ReplicationTaskProcessorCleanupJitterCoefficient, 10.0},\n\t\t\"ReplicationTaskProcessorStartWait\":                    {dynamicproperties.ReplicationTaskProcessorStartWait, time.Second},\n\t\t\"ReplicationTaskProcessorStartWaitJitterCoefficient\":   {dynamicproperties.ReplicationTaskProcessorStartWaitJitterCoefficient, 11.0},\n\t\t\"ReplicationTaskProcessorHostQPS\":                      {dynamicproperties.ReplicationTaskProcessorHostQPS, 12.0},\n\t\t\"ReplicationTaskProcessorShardQPS\":                     {dynamicproperties.ReplicationTaskProcessorShardQPS, 13.0},\n\t\t\"ReplicationTaskGenerationQPS\":                         {dynamicproperties.ReplicationTaskGenerationQPS, 14.0},\n\t\t\"EnableReplicationTaskGeneration\":                      {dynamicproperties.EnableReplicationTaskGeneration, true},\n\t\t\"EnableRecordWorkflowExecutionUninitialized\":           {dynamicproperties.EnableRecordWorkflowExecutionUninitialized, true},\n\t\t\"WorkflowIDExternalRPS\":                                {dynamicproperties.WorkflowIDExternalRPS, 87},\n\t\t\"WorkflowIDInternalRPS\":                                {dynamicproperties.WorkflowIDInternalRPS, 88},\n\t\t\"EnableConsistentQuery\":                                {dynamicproperties.EnableConsistentQuery, true},\n\t\t\"EnableConsistentQueryByDomain\":                        {dynamicproperties.EnableConsistentQueryByDomain, true},\n\t\t\"MaxBufferedQueryCount\":                                {dynamicproperties.MaxBufferedQueryCount, 89},\n\t\t\"EnableContextHeaderInVisibility\":                      {dynamicproperties.EnableContextHeaderInVisibility, true},\n\t\t\"EnableCrossClusterOperationsForDomain\":                {dynamicproperties.EnableCrossClusterOperationsForDomain, true},\n\t\t\"MutableStateChecksumGenProbability\":                   {dynamicproperties.MutableStateChecksumGenProbability, 90},\n\t\t\"MutableStateChecksumVerifyProbability\":                {dynamicproperties.MutableStateChecksumVerifyProbability, 91},\n\t\t\"MutableStateChecksumInvalidateBefore\":                 {dynamicproperties.MutableStateChecksumInvalidateBefore, 15.0},\n\t\t\"EnableRetryForChecksumFailure\":                        {dynamicproperties.EnableRetryForChecksumFailure, true},\n\t\t\"EnableHistoryCorruptionCheck\":                         {dynamicproperties.EnableHistoryCorruptionCheck, true},\n\t\t\"NotifyFailoverMarkerInterval\":                         {dynamicproperties.NotifyFailoverMarkerInterval, time.Second},\n\t\t\"NotifyFailoverMarkerTimerJitterCoefficient\":           {dynamicproperties.NotifyFailoverMarkerTimerJitterCoefficient, 16.0},\n\t\t\"EnableGracefulFailover\":                               {dynamicproperties.EnableGracefulFailover, true},\n\t\t\"EnableActivityLocalDispatchByDomain\":                  {dynamicproperties.EnableActivityLocalDispatchByDomain, true},\n\t\t\"MaxActivityCountDispatchByDomain\":                     {dynamicproperties.MaxActivityCountDispatchByDomain, 92},\n\t\t\"ActivityMaxScheduleToStartTimeoutForRetry\":            {dynamicproperties.ActivityMaxScheduleToStartTimeoutForRetry, time.Second},\n\t\t\"EnableDebugMode\":                                      {dynamicproperties.EnableDebugMode, true},\n\t\t\"EnableTaskInfoLogByDomainID\":                          {dynamicproperties.HistoryEnableTaskInfoLogByDomainID, true},\n\t\t\"EnableTimerDebugLogByDomainID\":                        {dynamicproperties.EnableTimerDebugLogByDomainID, true},\n\t\t\"SampleLoggingRate\":                                    {dynamicproperties.SampleLoggingRate, 93},\n\t\t\"EnableShardIDMetrics\":                                 {dynamicproperties.EnableShardIDMetrics, true},\n\t\t\"LargeShardHistorySizeMetricThreshold\":                 {dynamicproperties.LargeShardHistorySizeMetricThreshold, 94},\n\t\t\"LargeShardHistoryEventMetricThreshold\":                {dynamicproperties.LargeShardHistoryEventMetricThreshold, 95},\n\t\t\"LargeShardHistoryBlobMetricThreshold\":                 {dynamicproperties.LargeShardHistoryBlobMetricThreshold, 96},\n\t\t\"EnableStrongIdempotency\":                              {dynamicproperties.EnableStrongIdempotency, true},\n\t\t\"EnableStrongIdempotencySanityCheck\":                   {dynamicproperties.EnableStrongIdempotencySanityCheck, true},\n\t\t\"GlobalRatelimiterNewDataWeight\":                       {dynamicproperties.HistoryGlobalRatelimiterNewDataWeight, 17.0},\n\t\t\"GlobalRatelimiterUpdateInterval\":                      {dynamicproperties.GlobalRatelimiterUpdateInterval, time.Second},\n\t\t\"GlobalRatelimiterDecayAfter\":                          {dynamicproperties.HistoryGlobalRatelimiterDecayAfter, time.Second},\n\t\t\"GlobalRatelimiterGCAfter\":                             {dynamicproperties.HistoryGlobalRatelimiterGCAfter, time.Second},\n\t\t\"TaskSchedulerGlobalDomainRPS\":                         {dynamicproperties.TaskSchedulerGlobalDomainRPS, 97},\n\t\t\"TaskSchedulerEnableRateLimiterShadowMode\":             {dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false},\n\t\t\"TaskSchedulerEnableRateLimiter\":                       {dynamicproperties.TaskSchedulerEnableRateLimiter, true},\n\t\t\"HostName\":                                             {nil, hostname},\n\t\t\"SearchAttributesHiddenValueKeys\":                      {dynamicproperties.SearchAttributesHiddenValueKeys, map[string]interface{}{\"CustomStringField\": true}},\n\t\t\"ExecutionCacheMaxByteSize\":                            {dynamicproperties.ExecutionCacheMaxByteSize, 98},\n\t\t\"DisableTransferFailoverQueue\":                         {dynamicproperties.DisableTransferFailoverQueue, true},\n\t\t\"DisableTimerFailoverQueue\":                            {dynamicproperties.DisableTimerFailoverQueue, true},\n\t\t\"EnableTransferQueueV2\":                                {dynamicproperties.EnableTransferQueueV2, true},\n\t\t\"EnableTimerQueueV2\":                                   {dynamicproperties.EnableTimerQueueV2, true},\n\t\t\"QueueMaxPendingTaskCount\":                             {dynamicproperties.QueueMaxPendingTaskCount, 99},\n\t\t\"EnableTimerQueueV2PendingTaskCountAlert\":              {dynamicproperties.EnableTimerQueueV2PendingTaskCountAlert, true},\n\t\t\"EnableTransferQueueV2PendingTaskCountAlert\":           {dynamicproperties.EnableTransferQueueV2PendingTaskCountAlert, true},\n\t\t\"QueueCriticalPendingTaskCount\":                        {dynamicproperties.QueueCriticalPendingTaskCount, 100},\n\t\t\"QueueMaxVirtualQueueCount\":                            {dynamicproperties.QueueMaxVirtualQueueCount, 101},\n\t\t\"VirtualSliceForceAppendInterval\":                      {dynamicproperties.VirtualSliceForceAppendInterval, time.Second},\n\t\t\"ReplicationTaskProcessorLatencyLogThreshold\":          {dynamicproperties.ReplicationTaskProcessorLatencyLogThreshold, time.Duration(0)},\n\t\t\"EnableCleanupOrphanedHistoryBranchOnWorkflowCreation\": {dynamicproperties.EnableCleanupOrphanedHistoryBranchOnWorkflowCreation, true},\n\t\t\"EnableHierarchicalWeightedRoundRobinTaskScheduler\":    {dynamicproperties.EnableHierarchicalWeightedRoundRobinTaskScheduler, true},\n\t\t\"EnableTaskListAwareTaskSchedulerByDomain\":             {dynamicproperties.EnableTaskListAwareTaskSchedulerByDomain, true},\n\t}\n\tclient := dynamicconfig.NewInMemoryClient()\n\tfor fieldName, expected := range fields {\n\t\tif expected.key != nil {\n\t\t\terr := client.UpdateValue(expected.key, expected.value)\n\t\t\tif err != nil {\n\t\t\t\tt.Errorf(\"Failed to update config for %s: %s\", fieldName, err)\n\t\t\t}\n\t\t}\n\t}\n\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\n\tconfig := New(dc, numberOfShards, maxMessageSize, isAdvancedVisConfigExist, hostname)\n\n\tassertFieldsMatch(t, *config, fields)\n}\n\nfunc TestNewForTest(t *testing.T) {\n\tcfg := NewForTest()\n\tassert.NotNil(t, cfg)\n}\n\nfunc assertFieldsMatch(t *testing.T, config interface{}, fields map[string]configTestCase) {\n\tconfigType := reflect.ValueOf(config)\n\n\tfor i := 0; i < configType.NumField(); i++ {\n\t\tf := configType.Field(i)\n\t\tfieldName := configType.Type().Field(i).Name\n\n\t\tif expected, ok := fields[fieldName]; ok {\n\t\t\tactual := getValue(&f)\n\t\t\tif f.Kind() == reflect.Slice {\n\t\t\t\tassert.ElementsMatch(t, expected.value, actual, \"Incorrect value for field: %s\", fieldName)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, expected.value, actual, \"Incorrect value for field: %s\", fieldName)\n\t\t\t}\n\n\t\t} else {\n\t\t\tt.Errorf(\"Unknown property on Config: %s\", fieldName)\n\t\t}\n\t}\n}\n\nfunc getValue(f *reflect.Value) interface{} {\n\tswitch f.Kind() {\n\tcase reflect.Func:\n\t\tswitch fn := f.Interface().(type) {\n\t\tcase dynamicproperties.IntPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.IntPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.IntPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.BoolPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainIDFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.DurationPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.DurationPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.DurationPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.FloatPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.MapPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.StringPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.DurationPropertyFnWithDomainIDFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.IntPropertyFnWithShardIDFilter:\n\t\t\treturn fn(0)\n\t\tcase dynamicproperties.StringPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.DurationPropertyFnWithShardIDFilter:\n\t\t\treturn fn(0)\n\t\tcase dynamicproperties.FloatPropertyFnWithShardIDFilter:\n\t\t\treturn fn(0)\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainIDAndWorkflowIDFilter:\n\t\t\treturn fn(\"domain\", \"workflowID\")\n\t\tcase dynamicproperties.MapPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFnWithShardIDFilter:\n\t\t\treturn fn(0)\n\t\tcase func() []string:\n\t\t\treturn fn()\n\t\tdefault:\n\t\t\tpanic(\"Unable to handle type: \" + f.Type().Name())\n\t\t}\n\tdefault:\n\t\treturn f.Interface()\n\t}\n}\n\nfunc isolationGroupsHelper() []string {\n\treturn []string{\"zone-1\", \"zone-2\"}\n}\n"
  },
  {
    "path": "service/history/constants/constants.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage constants\n\nimport \"github.com/uber/cadence/common/types\"\n\nvar (\n\tErrDomainNotSet            = &types.BadRequestError{Message: \"Domain not set on request.\"}\n\tErrWorkflowExecutionNotSet = &types.BadRequestError{Message: \"WorkflowExecution not set on request.\"}\n\tErrTaskListNotSet          = &types.BadRequestError{Message: \"Tasklist not set.\"}\n\tErrRunIDNotValid           = &types.BadRequestError{Message: \"RunID is not valid UUID.\"}\n\tErrWorkflowIDNotSet        = &types.BadRequestError{Message: \"WorkflowId is not set on request.\"}\n\tErrSourceClusterNotSet     = &types.BadRequestError{Message: \"Source Cluster not set on request.\"}\n\tErrTimestampNotSet         = &types.BadRequestError{Message: \"Timestamp not set on request.\"}\n\tErrInvalidTaskType         = &types.BadRequestError{Message: \"Invalid task type\"}\n\tErrHistoryHostThrottle     = &types.ServiceBusyError{Message: \"History host rps exceeded\"}\n\tErrShuttingDown            = &types.InternalServiceError{Message: \"Shutting down\"}\n)\n"
  },
  {
    "path": "service/history/constants/test_constants.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage constants\n\nimport (\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// TestVersion is the workflow version for test\n\tTestVersion = cluster.TestCurrentClusterInitialFailoverVersion + (cluster.TestFailoverVersionIncrement * 5)\n\t// TestDomainID is the domainID for test\n\tTestDomainID = \"deadbeef-0123-4567-890a-bcdef0123456\"\n\t// TestActiveActiveDomainID is the active active domainID for test\n\tTestActiveActiveDomainID = \"965c78b7-1f6f-4122-9ba7-af6b7a55b27f\"\n\t// TestActiveActiveDomainName is the active active domain name for test\n\tTestActiveActiveDomainName = \"active-active-domain\"\n\t// TestDomainName is the domainName for test\n\tTestDomainName = \"some random domain name\"\n\t// TestRateLimitedDomainName is the domain name for testing task processing rate limits\n\tTestRateLimitedDomainName = \"rate-limited-domain-0123-4567-890a-bcdef0123456\"\n\t// TestRateLimitedDomainID is the domain ID for testing task processing rate limits\n\tTestRateLimitedDomainID = \"rate-limited-domain\"\n\t// TestUnknownDomainID is the domain ID for testing unknown domains\n\tTestUnknownDomainID = \"unknown-domain-id\"\n\t// TestWorkflowID is the workflowID for test\n\tTestWorkflowID = \"random-workflow-id\"\n\t// TestRunID is the workflow runID for test\n\tTestRunID = \"0d00698f-08e1-4d36-a3e2-3bf109f5d2d6\"\n\t// TestRequestID is the request ID for test\n\tTestRequestID = \"143b22cd-dfac-4d59-9398-893f89d89df6\"\n\t// CronSkip\n\tCronSkip = types.CronOverlapPolicySkipped\n\n\t// TestClusterMetadata is the cluster metadata for test\n\tTestClusterMetadata = cluster.GetTestClusterMetadata(true)\n\n\t// TestActiveClusterSelectionPolicy is the active cluster selection policy for test\n\tTestActiveClusterSelectionPolicy = types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-east\",\n\t\t},\n\t}\n\n\t// TestLocalDomainEntry is the local domain cache entry for test\n\tTestLocalDomainEntry = cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\t// TestGlobalDomainEntry is the global domain cache entry for test\n\tTestGlobalDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"test:///visibility/archival\",\n\t\t},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\tTestActiveActiveDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestActiveActiveDomainID, Name: TestActiveActiveDomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"test:///visibility/archival\",\n\t\t},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\t// TestRateLimitedDomainEntry is the global domain cache entry for test\n\tTestRateLimitedDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestRateLimitedDomainID, Name: TestRateLimitedDomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"test:///visibility/archival\",\n\t\t},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\t// TestGlobalParentDomainEntry is the global parent domain cache entry for test\n\tTestGlobalParentDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\t// TestGlobalTargetDomainEntry is the global target domain cache entry for test\n\tTestGlobalTargetDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\t// TestGlobalChildDomainEntry is the global child domain cache entry for test\n\tTestGlobalChildDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n\n\t// TestGlobalStandbyDomainEntry is the global standby domain cache entry for test\n\tTestGlobalStandbyDomainEntry = cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: TestDomainID, Name: TestDomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention:                1,\n\t\t\tVisibilityArchivalStatus: types.ArchivalStatusEnabled,\n\t\t\tVisibilityArchivalURI:    \"test:///visibility/archival\",\n\t\t},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tTestVersion,\n\t)\n)\n"
  },
  {
    "path": "service/history/decision/checker.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage decision\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/elasticsearch/validator\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\tattrValidator struct {\n\t\tconfig                    *config.Config\n\t\tdomainCache               cache.DomainCache\n\t\tmetricsClient             metrics.Client\n\t\tlogger                    log.Logger\n\t\tsearchAttributesValidator *validator.SearchAttributesValidator\n\t}\n\n\tworkflowSizeChecker struct {\n\t\tdomainName string\n\n\t\tblobSizeLimitWarn  int\n\t\tblobSizeLimitError int\n\n\t\thistorySizeLimitWarn  int\n\t\thistorySizeLimitError int\n\n\t\thistoryCountLimitWarn  int\n\t\thistoryCountLimitError int\n\n\t\tcompletedID    int64\n\t\tmutableState   execution.MutableState\n\t\texecutionStats *persistence.ExecutionStats\n\t\tmetricsScope   metrics.Scope\n\t\tlogger         log.Logger\n\t}\n)\n\nfunc newAttrValidator(\n\tdomainCache cache.DomainCache,\n\tmetricsClient metrics.Client,\n\tconfig *config.Config,\n\tlogger log.Logger,\n) *attrValidator {\n\treturn &attrValidator{\n\t\tconfig:        config,\n\t\tdomainCache:   domainCache,\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t\tsearchAttributesValidator: validator.NewSearchAttributesValidator(\n\t\t\tlogger,\n\t\t\tconfig.EnableQueryAttributeValidation,\n\t\t\tconfig.ValidSearchAttributes,\n\t\t\tconfig.SearchAttributesNumberOfKeysLimit,\n\t\t\tconfig.SearchAttributesSizeOfValueLimit,\n\t\t\tconfig.SearchAttributesTotalSizeLimit,\n\t\t),\n\t}\n}\n\nfunc newWorkflowSizeChecker(\n\tdomainName string,\n\tblobSizeLimitWarn int,\n\tblobSizeLimitError int,\n\thistorySizeLimitWarn int,\n\thistorySizeLimitError int,\n\thistoryCountLimitWarn int,\n\thistoryCountLimitError int,\n\tcompletedID int64,\n\tmutableState execution.MutableState,\n\texecutionStats *persistence.ExecutionStats,\n\tmetricsScope metrics.Scope,\n\tlogger log.Logger,\n) *workflowSizeChecker {\n\treturn &workflowSizeChecker{\n\t\tdomainName:             domainName,\n\t\tblobSizeLimitWarn:      blobSizeLimitWarn,\n\t\tblobSizeLimitError:     blobSizeLimitError,\n\t\thistorySizeLimitWarn:   historySizeLimitWarn,\n\t\thistorySizeLimitError:  historySizeLimitError,\n\t\thistoryCountLimitWarn:  historyCountLimitWarn,\n\t\thistoryCountLimitError: historyCountLimitError,\n\t\tcompletedID:            completedID,\n\t\tmutableState:           mutableState,\n\t\texecutionStats:         executionStats,\n\t\tmetricsScope:           metricsScope,\n\t\tlogger:                 logger,\n\t}\n}\n\nfunc (c *workflowSizeChecker) failWorkflowIfBlobSizeExceedsLimit(\n\tdecisionTypeTag metrics.Tag,\n\tblob []byte,\n\tmessage string,\n) (bool, error) {\n\n\texecutionInfo := c.mutableState.GetExecutionInfo()\n\terr := common.CheckEventBlobSizeLimit(\n\t\tlen(blob),\n\t\tc.blobSizeLimitWarn,\n\t\tc.blobSizeLimitError,\n\t\texecutionInfo.DomainID,\n\t\tc.domainName,\n\t\texecutionInfo.WorkflowID,\n\t\texecutionInfo.RunID,\n\t\tc.metricsScope.Tagged(decisionTypeTag),\n\t\tc.logger,\n\t\ttag.BlobSizeViolationOperation(decisionTypeTag.Value()),\n\t)\n\tif err == nil {\n\t\treturn false, nil\n\t}\n\n\tattributes := &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\tDetails: []byte(message),\n\t}\n\n\tif _, err := c.mutableState.AddFailWorkflowEvent(c.completedID, attributes); err != nil {\n\t\treturn false, err\n\t}\n\n\treturn true, nil\n}\n\nfunc (c *workflowSizeChecker) failWorkflowSizeExceedsLimit() (bool, error) {\n\thistoryCount := int(c.mutableState.GetNextEventID()) - 1\n\thistorySize := int(c.executionStats.HistorySize)\n\n\t// metricsScope already has domainName and operation: \"RespondDecisionTaskCompleted\"\n\tc.metricsScope.RecordTimer(metrics.HistorySize, time.Duration(historySize))\n\tc.metricsScope.RecordTimer(metrics.HistoryCount, time.Duration(historyCount))\n\n\tif historySize > c.historySizeLimitError || historyCount > c.historyCountLimitError {\n\t\texecutionInfo := c.mutableState.GetExecutionInfo()\n\t\tc.logger.Error(\"history size exceeds error limit.\",\n\t\t\ttag.WorkflowDomainName(c.domainName),\n\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\t\ttag.WorkflowHistorySize(historySize),\n\t\t\ttag.WorkflowEventCount(historyCount))\n\n\t\tattributes := &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\tReason:  common.StringPtr(common.FailureReasonSizeExceedsLimit),\n\t\t\tDetails: []byte(\"Workflow history size / count exceeds limit.\"),\n\t\t}\n\n\t\tif _, err := c.mutableState.AddFailWorkflowEvent(c.completedID, attributes); err != nil {\n\t\t\treturn false, err\n\t\t}\n\t\treturn true, nil\n\t}\n\n\tif historySize > c.historySizeLimitWarn || historyCount > c.historyCountLimitWarn {\n\t\texecutionInfo := c.mutableState.GetExecutionInfo()\n\t\tc.logger.Warn(\"history size exceeds warn limit.\",\n\t\t\ttag.WorkflowDomainName(c.domainName),\n\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\t\ttag.WorkflowHistorySize(historySize),\n\t\t\ttag.WorkflowEventCount(historyCount))\n\t\treturn false, nil\n\t}\n\n\treturn false, nil\n}\n\nfunc (v *attrValidator) validateActivityScheduleAttributes(\n\tdomainID string,\n\ttargetDomainID string,\n\tattributes *types.ScheduleActivityTaskDecisionAttributes,\n\texecutionInfo *persistence.WorkflowExecutionInfo,\n\tmetricsScope metrics.ScopeIdx,\n) error {\n\n\tif err := v.validateCrossDomainCall(\n\t\tdomainID,\n\t\ttargetDomainID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"ScheduleActivityTaskDecisionAttributes is not set on decision.\"}\n\t}\n\n\ttaskList, err := v.validatedTaskList(attributes.TaskList, &types.TaskList{Name: executionInfo.TaskList, Kind: executionInfo.TaskListKind.Ptr()}, metricsScope, attributes.GetDomain())\n\tif err != nil {\n\t\treturn err\n\t}\n\tattributes.TaskList = taskList\n\n\tif attributes.GetActivityID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"ActivityId is not set on decision.\"}\n\t}\n\n\tif attributes.ActivityType == nil || attributes.ActivityType.GetName() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"ActivityType is not set on decision.\"}\n\t}\n\n\tif err := common.ValidateRetryPolicy(attributes.RetryPolicy); err != nil {\n\t\treturn err\n\t}\n\n\tidLengthWarnLimit := v.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tattributes.GetActivityID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.ActivityIDMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrActivityIDExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeActivityID) {\n\t\treturn &types.BadRequestError{Message: \"ActivityID exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.GetActivityType().GetName(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.ActivityTypeMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrActivityTypeExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeActivityType) {\n\t\treturn &types.BadRequestError{Message: \"ActivityType exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.GetDomain(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.DomainNameMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn &types.BadRequestError{Message: \"Domain exceeds length limit.\"}\n\t}\n\n\t// Only attempt to deduce and fill in unspecified timeouts only when all timeouts are non-negative.\n\tif attributes.GetScheduleToCloseTimeoutSeconds() < 0 || attributes.GetScheduleToStartTimeoutSeconds() < 0 ||\n\t\tattributes.GetStartToCloseTimeoutSeconds() < 0 || attributes.GetHeartbeatTimeoutSeconds() < 0 {\n\t\treturn &types.BadRequestError{Message: \"A valid timeout may not be negative.\"}\n\t}\n\twfTimeout := executionInfo.WorkflowTimeout\n\n\t// ensure activity timeout never larger than workflow timeout\n\tif attributes.GetScheduleToCloseTimeoutSeconds() > wfTimeout {\n\t\tattributes.ScheduleToCloseTimeoutSeconds = common.Int32Ptr(wfTimeout)\n\t}\n\tif attributes.GetScheduleToStartTimeoutSeconds() > wfTimeout {\n\t\tattributes.ScheduleToStartTimeoutSeconds = common.Int32Ptr(wfTimeout)\n\t}\n\tif attributes.GetStartToCloseTimeoutSeconds() > wfTimeout {\n\t\tattributes.StartToCloseTimeoutSeconds = common.Int32Ptr(wfTimeout)\n\t}\n\tif attributes.GetHeartbeatTimeoutSeconds() > wfTimeout {\n\t\tattributes.HeartbeatTimeoutSeconds = common.Int32Ptr(wfTimeout)\n\t}\n\n\tvalidScheduleToClose := attributes.GetScheduleToCloseTimeoutSeconds() > 0\n\tvalidScheduleToStart := attributes.GetScheduleToStartTimeoutSeconds() > 0\n\tvalidStartToClose := attributes.GetStartToCloseTimeoutSeconds() > 0\n\n\tif validScheduleToClose {\n\t\tif !validScheduleToStart {\n\t\t\tattributes.ScheduleToStartTimeoutSeconds = common.Int32Ptr(attributes.GetScheduleToCloseTimeoutSeconds())\n\t\t}\n\t\tif !validStartToClose {\n\t\t\tattributes.StartToCloseTimeoutSeconds = common.Int32Ptr(attributes.GetScheduleToCloseTimeoutSeconds())\n\t\t}\n\t} else if validScheduleToStart && validStartToClose {\n\t\tattributes.ScheduleToCloseTimeoutSeconds = common.Int32Ptr(attributes.GetScheduleToStartTimeoutSeconds() + attributes.GetStartToCloseTimeoutSeconds())\n\t\tif attributes.GetScheduleToCloseTimeoutSeconds() > wfTimeout {\n\t\t\tattributes.ScheduleToCloseTimeoutSeconds = common.Int32Ptr(wfTimeout)\n\t\t}\n\t} else {\n\t\t// Deduction failed as there's not enough information to fill in missing timeouts.\n\t\treturn &types.BadRequestError{Message: \"A valid ScheduleToCloseTimeout is not set on decision.\"}\n\t}\n\n\t// ensure activity's SCHEDULE_TO_START and SCHEDULE_TO_CLOSE is as long as expiration on retry policy\n\t// if SCHEDULE_TO_START timeout is retryable\n\tp := attributes.RetryPolicy\n\tif p != nil {\n\t\tisScheduleToStartRetryable := true\n\t\tscheduleToStartErrorReason := execution.TimerTypeToReason(execution.TimerTypeScheduleToStart)\n\t\tfor _, reason := range p.GetNonRetriableErrorReasons() {\n\t\t\tif reason == scheduleToStartErrorReason {\n\t\t\t\tisScheduleToStartRetryable = false\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\texpiration := p.GetExpirationIntervalInSeconds()\n\t\tif expiration == 0 || expiration > wfTimeout {\n\t\t\texpiration = wfTimeout\n\t\t}\n\n\t\tif isScheduleToStartRetryable {\n\t\t\t// If schedule to start timeout is retryable, we don't need to fail the activity and schedule\n\t\t\t// it again on the same tasklist, as it's a no-op). Extending schedule to start timeout to achieve\n\t\t\t// the same thing.\n\t\t\t//\n\t\t\t// Theoretically, we can extend schedule to start to be as long as the expiration time,\n\t\t\t// but if user specifies a very long expiration time and activity task got lost after the activity is\n\t\t\t// scheduled, workflow will be stuck for a long time. So here, we cap the schedule to start timeout\n\t\t\t// to a maximum value, so that when the activity task got lost, timeout can happen sooner and schedule\n\t\t\t// the activity again.\n\n\t\t\tdomainName, _ := v.domainCache.GetDomainName(domainID) // if this call returns an error, we will just used the default value for max timeout\n\t\t\tmaximumScheduleToStartTimeoutForRetryInSeconds := int32(v.config.ActivityMaxScheduleToStartTimeoutForRetry(domainName).Seconds())\n\t\t\tscheduleToStartExpiration := min(expiration, maximumScheduleToStartTimeoutForRetryInSeconds)\n\t\t\tif attributes.GetScheduleToStartTimeoutSeconds() < scheduleToStartExpiration {\n\t\t\t\tattributes.ScheduleToStartTimeoutSeconds = common.Int32Ptr(scheduleToStartExpiration)\n\t\t\t}\n\n\t\t\t// TODO: uncomment the following code when the client side bug for calculating scheduleToClose deadline is fixed and\n\t\t\t// fully rolled out. Before that, we still need to extend scheduleToClose timeout to be as long as the expiration interval\n\t\t\t//\n\t\t\t// scheduleToCloseExpiration := common.MinInt32(expiration, scheduleToStartExpiration+attributes.GetStartToCloseTimeoutSeconds())\n\t\t\t// if attributes.GetScheduleToCloseTimeoutSeconds() < scheduleToCloseExpiration {\n\t\t\t// \tattributes.ScheduleToCloseTimeoutSeconds = common.Int32Ptr(scheduleToCloseExpiration)\n\t\t\t// }\n\t\t}\n\n\t\tif attributes.GetScheduleToCloseTimeoutSeconds() < expiration {\n\t\t\tattributes.ScheduleToCloseTimeoutSeconds = common.Int32Ptr(expiration)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateTimerScheduleAttributes(\n\tattributes *types.StartTimerDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"StartTimerDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.GetTimerID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"TimerId is not set on decision.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\tattributes.GetTimerID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.TimerIDMaxLength(domain),\n\t\tmetrics.CadenceErrTimerIDExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeTimerID) {\n\t\treturn &types.BadRequestError{Message: \"TimerId exceeds length limit.\"}\n\t}\n\tif attributes.GetStartToFireTimeoutSeconds() <= 0 {\n\t\treturn &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"Invalid StartToFireTimeoutSeconds: %v\", attributes.GetStartToFireTimeoutSeconds()),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateActivityCancelAttributes(\n\tattributes *types.RequestCancelActivityTaskDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"RequestCancelActivityTaskDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.GetActivityID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"ActivityId is not set on decision.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.GetActivityID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.ActivityIDMaxLength(domain),\n\t\tmetrics.CadenceErrActivityIDExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeActivityID) {\n\t\treturn &types.BadRequestError{Message: \"ActivityId exceeds length limit.\"}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateTimerCancelAttributes(\n\tattributes *types.CancelTimerDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"CancelTimerDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.GetTimerID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"TimerId is not set on decision.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\tattributes.GetTimerID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.TimerIDMaxLength(domain),\n\t\tmetrics.CadenceErrTimerIDExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeTimerID) {\n\t\treturn &types.BadRequestError{Message: \"TimerId exceeds length limit.\"}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateRecordMarkerAttributes(\n\tattributes *types.RecordMarkerDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"RecordMarkerDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.GetMarkerName() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"MarkerName is not set on decision.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\tattributes.GetMarkerName(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.MarkerNameMaxLength(domain),\n\t\tmetrics.CadenceErrMarkerNameExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeMarkerName) {\n\t\treturn &types.BadRequestError{Message: \"MarkerName exceeds length limit.\"}\n\t}\n\n\treturn nil\n}\n\nfunc (v *attrValidator) validateCompleteWorkflowExecutionAttributes(\n\tattributes *types.CompleteWorkflowExecutionDecisionAttributes,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"CompleteWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateFailWorkflowExecutionAttributes(\n\tattributes *types.FailWorkflowExecutionDecisionAttributes,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"FailWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.Reason == nil {\n\t\treturn &types.BadRequestError{Message: \"Reason is not set on decision.\"}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateCancelWorkflowExecutionAttributes(\n\tattributes *types.CancelWorkflowExecutionDecisionAttributes,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"CancelWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\treturn nil\n}\n\nfunc (v *attrValidator) validateCancelExternalWorkflowExecutionAttributes(\n\tdomainID string,\n\ttargetDomainID string,\n\tattributes *types.RequestCancelExternalWorkflowExecutionDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n) error {\n\n\tif err := v.validateCrossDomainCall(\n\t\tdomainID,\n\t\ttargetDomainID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"RequestCancelExternalWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.WorkflowID == \"\" {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId is not set on decision.\"}\n\t}\n\n\tidLengthWarnLimit := v.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tattributes.GetDomain(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.DomainNameMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn &types.BadRequestError{Message: \"Domain exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.GetWorkflowID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.WorkflowIDMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId exceeds length limit.\"}\n\t}\n\trunID := attributes.GetRunID()\n\tif runID != \"\" && uuid.Parse(runID) == nil {\n\t\treturn &types.BadRequestError{Message: \"Invalid RunId set on decision.\"}\n\t}\n\n\treturn nil\n}\n\nfunc (v *attrValidator) validateSignalExternalWorkflowExecutionAttributes(\n\tdomainID string,\n\ttargetDomainID string,\n\tattributes *types.SignalExternalWorkflowExecutionDecisionAttributes,\n\tmetricsScope metrics.ScopeIdx,\n) error {\n\n\tif err := v.validateCrossDomainCall(\n\t\tdomainID,\n\t\ttargetDomainID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"SignalExternalWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\tif attributes.Execution == nil {\n\t\treturn &types.BadRequestError{Message: \"Execution is nil on decision.\"}\n\t}\n\tif attributes.Execution.WorkflowID == \"\" {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId is not set on decision.\"}\n\t}\n\n\tidLengthWarnLimit := v.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tattributes.GetDomain(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.DomainNameMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn &types.BadRequestError{Message: \"Domain exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.Execution.GetWorkflowID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.WorkflowIDMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId exceeds length limit.\"}\n\t}\n\n\ttargetRunID := attributes.Execution.GetRunID()\n\tif targetRunID != \"\" && uuid.Parse(targetRunID) == nil {\n\t\treturn &types.BadRequestError{Message: \"Invalid RunId set on decision.\"}\n\t}\n\tif attributes.SignalName == \"\" {\n\t\treturn &types.BadRequestError{Message: \"SignalName is not set on decision.\"}\n\t}\n\n\treturn nil\n}\n\nfunc (v *attrValidator) validateUpsertWorkflowSearchAttributes(\n\tdomainName string,\n\tattributes *types.UpsertWorkflowSearchAttributesDecisionAttributes,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"UpsertWorkflowSearchAttributesDecisionAttributes is not set on decision.\"}\n\t}\n\n\tif attributes.SearchAttributes == nil {\n\t\treturn &types.BadRequestError{Message: \"SearchAttributes is not set on decision.\"}\n\t}\n\n\tif len(attributes.GetSearchAttributes().GetIndexedFields()) == 0 {\n\t\treturn &types.BadRequestError{Message: \"IndexedFields is empty on decision.\"}\n\t}\n\n\treturn v.searchAttributesValidator.ValidateSearchAttributes(attributes.GetSearchAttributes(), domainName)\n}\n\nfunc (v *attrValidator) validateContinueAsNewWorkflowExecutionAttributes(\n\tattributes *types.ContinueAsNewWorkflowExecutionDecisionAttributes,\n\texecutionInfo *persistence.WorkflowExecutionInfo,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) error {\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"ContinueAsNewWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\n\t// Inherit workflow type from previous execution if not provided on decision\n\tif attributes.WorkflowType == nil || attributes.WorkflowType.GetName() == \"\" {\n\t\tattributes.WorkflowType = &types.WorkflowType{Name: executionInfo.WorkflowTypeName}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.WorkflowType.GetName(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.WorkflowTypeMaxLength(domain),\n\t\tmetrics.CadenceErrWorkflowTypeExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeWorkflowType) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowType exceeds length limit.\"}\n\t}\n\n\t// Inherit Tasklist from previous execution if not provided on decision\n\ttaskList, err := v.validatedTaskList(attributes.TaskList, &types.TaskList{Name: executionInfo.TaskList, Kind: executionInfo.TaskListKind.Ptr()}, metricsScope, domain)\n\tif err != nil {\n\t\treturn err\n\t}\n\tattributes.TaskList = taskList\n\n\t// Inherit workflow timeout from previous execution if not provided on decision\n\tif attributes.GetExecutionStartToCloseTimeoutSeconds() <= 0 {\n\t\tattributes.ExecutionStartToCloseTimeoutSeconds = common.Int32Ptr(executionInfo.WorkflowTimeout)\n\t}\n\n\t// Inherit decision task timeout from previous execution if not provided on decision\n\tif attributes.GetTaskStartToCloseTimeoutSeconds() <= 0 {\n\t\tattributes.TaskStartToCloseTimeoutSeconds = common.Int32Ptr(executionInfo.DecisionStartToCloseTimeout)\n\t}\n\n\t// Check next run decision task delay\n\tif attributes.GetBackoffStartIntervalInSeconds() < 0 {\n\t\treturn &types.BadRequestError{Message: \"BackoffStartInterval is less than 0.\"}\n\t}\n\n\tdomainName, err := v.domainCache.GetDomainName(executionInfo.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn v.searchAttributesValidator.ValidateSearchAttributes(attributes.GetSearchAttributes(), domainName)\n}\n\nfunc (v *attrValidator) validateStartChildExecutionAttributes(\n\tdomainID string,\n\ttargetDomainID string,\n\tattributes *types.StartChildWorkflowExecutionDecisionAttributes,\n\tparentInfo *persistence.WorkflowExecutionInfo,\n\tmetricsScope metrics.ScopeIdx,\n) error {\n\n\tif err := v.validateCrossDomainCall(\n\t\tdomainID,\n\t\ttargetDomainID,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif attributes == nil {\n\t\treturn &types.BadRequestError{Message: \"StartChildWorkflowExecutionDecisionAttributes is not set on decision.\"}\n\t}\n\n\tif attributes.GetWorkflowID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Required field WorkflowID is not set on decision.\"}\n\t}\n\n\tif attributes.WorkflowType == nil || attributes.WorkflowType.GetName() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Required field WorkflowType is not set on decision.\"}\n\t}\n\n\tidLengthWarnLimit := v.config.MaxIDLengthWarnLimit()\n\tif !common.IsValidIDLength(\n\t\tattributes.GetDomain(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.DomainNameMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn &types.BadRequestError{Message: \"Domain exceeds length limit.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\tattributes.GetWorkflowID(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.WorkflowIDMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\tattributes.WorkflowType.GetName(),\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tidLengthWarnLimit,\n\t\tv.config.WorkflowTypeMaxLength(attributes.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowTypeExceededWarnLimit,\n\t\tattributes.GetDomain(),\n\t\tv.logger,\n\t\ttag.IDTypeWorkflowType) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowType exceeds length limit.\"}\n\t}\n\n\tif err := common.ValidateRetryPolicy(attributes.RetryPolicy); err != nil {\n\t\treturn err\n\t}\n\n\tif attributes.GetCronSchedule() != \"\" {\n\t\tif _, err := backoff.ValidateSchedule(attributes.GetCronSchedule()); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t// Inherit tasklist from parent workflow execution if not provided on decision\n\ttaskList, err := v.validatedTaskList(attributes.TaskList, &types.TaskList{Name: parentInfo.TaskList, Kind: parentInfo.TaskListKind.Ptr()}, metricsScope, attributes.GetDomain())\n\tif err != nil {\n\t\treturn err\n\t}\n\tattributes.TaskList = taskList\n\n\t// Inherit workflow timeout from parent workflow execution if not provided on decision\n\tif attributes.GetExecutionStartToCloseTimeoutSeconds() <= 0 {\n\t\tattributes.ExecutionStartToCloseTimeoutSeconds = common.Int32Ptr(parentInfo.WorkflowTimeout)\n\t}\n\n\t// Inherit decision task timeout from parent workflow execution if not provided on decision\n\tif attributes.GetTaskStartToCloseTimeoutSeconds() <= 0 {\n\t\tattributes.TaskStartToCloseTimeoutSeconds = common.Int32Ptr(parentInfo.DecisionStartToCloseTimeout)\n\t}\n\n\treturn nil\n}\n\nfunc (v *attrValidator) validatedTaskList(\n\ttaskList *types.TaskList,\n\tdefaultVal *types.TaskList,\n\tmetricsScope metrics.ScopeIdx,\n\tdomain string,\n) (*types.TaskList, error) {\n\n\tif taskList == nil {\n\t\ttaskList = &types.TaskList{}\n\t}\n\n\tif taskList.GetName() == \"\" {\n\t\tif defaultVal.GetName() == \"\" {\n\t\t\treturn taskList, &types.BadRequestError{Message: \"missing task list name\"}\n\t\t}\n\t\ttaskList.Name = defaultVal.GetName()\n\t\ttaskList.Kind = defaultVal.Kind\n\t\treturn taskList, nil\n\t}\n\tif taskList.GetKind() == types.TaskListKindNormal && defaultVal.GetKind() == types.TaskListKindEphemeral {\n\t\ttaskList.Kind = defaultVal.Kind\n\t}\n\n\tname := taskList.GetName()\n\tif !common.IsValidIDLength(\n\t\tname,\n\t\tv.metricsClient.Scope(metricsScope),\n\t\tv.config.MaxIDLengthWarnLimit(),\n\t\tv.config.TaskListNameMaxLength(domain),\n\t\tmetrics.CadenceErrTaskListNameExceededWarnLimit,\n\t\tdomain,\n\t\tv.logger,\n\t\ttag.IDTypeTaskListName) {\n\t\treturn taskList, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"task list name exceeds length limit of %v\", v.config.TaskListNameMaxLength(domain)),\n\t\t}\n\t}\n\n\tif strings.HasPrefix(name, constants.ReservedTaskListPrefix) {\n\t\treturn taskList, &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"task list name cannot start with reserved prefix %v\", constants.ReservedTaskListPrefix),\n\t\t}\n\t}\n\n\treturn taskList, nil\n}\n\nfunc (v *attrValidator) validateCrossDomainCall(\n\tsourceDomainID string,\n\ttargetDomainID string,\n) error {\n\n\t// same name, no check needed\n\tif sourceDomainID == targetDomainID {\n\t\treturn nil\n\t}\n\n\tsourceDomainEntry, err := v.domainCache.GetDomainByID(sourceDomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttargetDomainEntry, err := v.domainCache.GetDomainByID(targetDomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsourceClusters := sourceDomainEntry.GetReplicationConfig().Clusters\n\ttargetClusters := targetDomainEntry.GetReplicationConfig().Clusters\n\n\t// both \"local domain\"\n\t// here a domain is \"local domain\" when:\n\t// - IsGlobalDomain() returns false\n\t// - domainCluster contains only one cluster\n\t// case 1 can be actually be combined with this case\n\tif len(sourceClusters) == 1 && len(targetClusters) == 1 {\n\t\tif sourceClusters[0].ClusterName == targetClusters[0].ClusterName {\n\t\t\treturn nil\n\t\t}\n\t\treturn v.createCrossDomainCallError(sourceDomainEntry, targetDomainEntry)\n\t}\n\n\t// both global domain with > 1 replication cluster\n\t// when code reaches here, at least one domain has more than one cluster\n\tif len(sourceClusters) == len(targetClusters) &&\n\t\tv.config.EnableCrossClusterOperationsForDomain(sourceDomainEntry.GetInfo().Name) {\n\t\t// check if the source domain cluster matches those for the target domain\n\t\tfor _, sourceCluster := range sourceClusters {\n\t\t\tfound := false\n\t\t\tfor _, targetCluster := range targetClusters {\n\t\t\t\tif sourceCluster.ClusterName == targetCluster.ClusterName {\n\t\t\t\t\tfound = true\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tif !found {\n\t\t\t\treturn v.createCrossDomainCallError(sourceDomainEntry, targetDomainEntry)\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t}\n\n\treturn v.createCrossDomainCallError(sourceDomainEntry, targetDomainEntry)\n}\n\nfunc (v *attrValidator) createCrossDomainCallError(\n\tdomainEntry *cache.DomainCacheEntry,\n\ttargetDomainEntry *cache.DomainCacheEntry,\n) error {\n\treturn &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\"cannot make cross domain call between %v and %v\",\n\t\tdomainEntry.GetInfo().Name,\n\t\ttargetDomainEntry.GetInfo().Name,\n\t)}\n}\n"
  },
  {
    "path": "service/history/decision/checker_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage decision\n\nimport (\n\t\"sort\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\tattrValidatorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tmockDomainCache *cache.MockDomainCache\n\n\t\tvalidator *attrValidator\n\n\t\ttestDomainID       string\n\t\ttestTargetDomainID string\n\n\t\ttestActivityMaxScheduleToStartTimeoutForRetryInSeconds int32\n\t}\n)\n\nfunc TestAttrValidatorSuite(t *testing.T) {\n\ts := new(attrValidatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *attrValidatorSuite) SetupSuite() {\n\ts.testDomainID = \"test domain ID\"\n\ts.testTargetDomainID = \"test target domain ID\"\n\ts.testActivityMaxScheduleToStartTimeoutForRetryInSeconds = 1800\n}\n\nfunc (s *attrValidatorSuite) TearDownSuite() {\n}\n\nfunc (s *attrValidatorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\tconfig := &config.Config{\n\t\tMaxIDLengthWarnLimit:              dynamicproperties.GetIntPropertyFn(128),\n\t\tDomainNameMaxLength:               dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tIdentityMaxLength:                 dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tWorkflowIDMaxLength:               dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tSignalNameMaxLength:               dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tWorkflowTypeMaxLength:             dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tRequestIDMaxLength:                dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tTaskListNameMaxLength:             dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tActivityIDMaxLength:               dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tActivityTypeMaxLength:             dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tMarkerNameMaxLength:               dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tTimerIDMaxLength:                  dynamicproperties.GetIntPropertyFilteredByDomain(1000),\n\t\tValidSearchAttributes:             dynamicproperties.GetMapPropertyFn(definition.GetDefaultIndexedKeys()),\n\t\tEnableQueryAttributeValidation:    dynamicproperties.GetBoolPropertyFn(true),\n\t\tSearchAttributesNumberOfKeysLimit: dynamicproperties.GetIntPropertyFilteredByDomain(100),\n\t\tSearchAttributesSizeOfValueLimit:  dynamicproperties.GetIntPropertyFilteredByDomain(2 * 1024),\n\t\tSearchAttributesTotalSizeLimit:    dynamicproperties.GetIntPropertyFilteredByDomain(40 * 1024),\n\t\tActivityMaxScheduleToStartTimeoutForRetry: dynamicproperties.GetDurationPropertyFnFilteredByDomain(\n\t\t\ttime.Duration(s.testActivityMaxScheduleToStartTimeoutForRetryInSeconds) * time.Second,\n\t\t),\n\t\tEnableCrossClusterOperationsForDomain: dynamicproperties.GetBoolPropertyFnFilteredByDomain(false),\n\t}\n\ts.validator = newAttrValidator(\n\t\ts.mockDomainCache,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tconfig,\n\t\tlog.NewNoop(),\n\t)\n}\n\nfunc (s *attrValidatorSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *attrValidatorSuite) TestValidateSignalExternalWorkflowExecutionAttributes() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).AnyTimes()\n\n\tvar attributes *types.SignalExternalWorkflowExecutionDecisionAttributes\n\n\terr := s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.EqualError(err, \"SignalExternalWorkflowExecutionDecisionAttributes is not set on decision.\")\n\n\tattributes = &types.SignalExternalWorkflowExecutionDecisionAttributes{}\n\terr = s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.EqualError(err, \"Execution is nil on decision.\")\n\n\tattributes.Execution = &types.WorkflowExecution{}\n\tattributes.Execution.WorkflowID = \"workflow-id\"\n\terr = s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.EqualError(err, \"SignalName is not set on decision.\")\n\n\tattributes.Execution.RunID = \"run-id\"\n\terr = s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.EqualError(err, \"Invalid RunId set on decision.\")\n\tattributes.Execution.RunID = constants.TestRunID\n\n\tattributes.SignalName = \"my signal name\"\n\terr = s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.NoError(err)\n\n\tattributes.Input = []byte(\"test input\")\n\terr = s.validator.validateSignalExternalWorkflowExecutionAttributes(s.testDomainID, s.testTargetDomainID, attributes, metrics.HistoryRespondDecisionTaskCompletedScope)\n\ts.NoError(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateUpsertWorkflowSearchAttributes() {\n\tdomainName := \"testDomain\"\n\tvar attributes *types.UpsertWorkflowSearchAttributesDecisionAttributes\n\n\terr := s.validator.validateUpsertWorkflowSearchAttributes(domainName, attributes)\n\ts.EqualError(err, \"UpsertWorkflowSearchAttributesDecisionAttributes is not set on decision.\")\n\n\tattributes = &types.UpsertWorkflowSearchAttributesDecisionAttributes{}\n\terr = s.validator.validateUpsertWorkflowSearchAttributes(domainName, attributes)\n\ts.EqualError(err, \"SearchAttributes is not set on decision.\")\n\n\tattributes.SearchAttributes = &types.SearchAttributes{}\n\terr = s.validator.validateUpsertWorkflowSearchAttributes(domainName, attributes)\n\ts.EqualError(err, \"IndexedFields is empty on decision.\")\n\n\tattributes.SearchAttributes.IndexedFields = map[string][]byte{\"CustomKeywordField\": []byte(`\"bytes\"`)}\n\terr = s.validator.validateUpsertWorkflowSearchAttributes(domainName, attributes)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_LocalToLocal() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_LocalToEffectiveLocal_SameCluster() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_LocalToEffectiveLocal_DiffCluster() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestAlternativeClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_LocalToGlobal() {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_EffectiveLocalToLocal_SameCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_EffectiveLocalToLocal_DiffCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestAlternativeClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_EffectiveLocalToEffectiveLocal_SameCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}},\n\t\t},\n\t\t5678,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_EffectiveLocalToEffectiveLocal_DiffCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestAlternativeClusterName}},\n\t\t},\n\t\t5678,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_EffectiveLocalToGlobal() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t\t5678,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_GlobalToLocal() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_GlobalToEffectiveLocal() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t5678,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_GlobalToGlobal_SameDomain() {\n\ttargetDomainID := s.testDomainID\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, targetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_GlobalToGlobal_DiffDomain_SameCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(2)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(2)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n\n\ts.validator.config.EnableCrossClusterOperationsForDomain = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\terr = s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.Nil(err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateCrossDomainCall_GlobalToGlobal_DiffDomain_DiffCluster() {\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: \"cluster name for s.testDomainID\"},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\ttargetDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t{ClusterName: \"cluster name for s.testTargetDomainID\"},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateCrossDomainCall(s.testDomainID, s.testTargetDomainID)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *attrValidatorSuite) TestValidateTaskListName() {\n\ttaskList := func(name string) *types.TaskList {\n\t\tkind := types.TaskListKindNormal\n\t\treturn &types.TaskList{Name: name, Kind: &kind}\n\t}\n\tephemeralTasklist := func(name string) *types.TaskList {\n\t\tkind := types.TaskListKindEphemeral\n\t\treturn &types.TaskList{Name: name, Kind: &kind}\n\t}\n\n\ttestCases := []struct {\n\t\tdefaultVal  *types.TaskList\n\t\tinput       *types.TaskList\n\t\toutput      *types.TaskList\n\t\tisOutputErr bool\n\t}{\n\t\t{taskList(\"tl-1\"), nil, taskList(\"tl-1\"), false},\n\t\t{taskList(\"\"), taskList(\"tl-1\"), taskList(\"tl-1\"), false},\n\t\t{taskList(\"tl-1\"), taskList(\"tl-1\"), taskList(\"tl-1\"), false},\n\t\t{taskList(\"\"), taskList(\"/tl-1\"), taskList(\"/tl-1\"), false},\n\t\t{ephemeralTasklist(\"tl-1\"), nil, ephemeralTasklist(\"tl-1\"), false},\n\t\t{ephemeralTasklist(\"tl-1\"), taskList(\"tl-1\"), ephemeralTasklist(\"tl-1\"), false},\n\t\t{ephemeralTasklist(\"tl-1\"), taskList(\"tl-2\"), ephemeralTasklist(\"tl-2\"), false},\n\t\t{ephemeralTasklist(\"tl-1\"), taskList(\"\"), ephemeralTasklist(\"tl-1\"), false},\n\t\t{taskList(\"\"), taskList(\"/__cadence_sys\"), taskList(\"/__cadence_sys\"), false},\n\t\t{taskList(\"\"), nil, &types.TaskList{}, true},\n\t\t{taskList(\"\"), taskList(\"\"), taskList(\"\"), true},\n\t\t{taskList(\"\"), taskList(commonconstants.ReservedTaskListPrefix), taskList(commonconstants.ReservedTaskListPrefix), true},\n\t\t{taskList(\"tl-1\"), taskList(commonconstants.ReservedTaskListPrefix), taskList(commonconstants.ReservedTaskListPrefix), true},\n\t\t{taskList(\"\"), taskList(commonconstants.ReservedTaskListPrefix + \"tl-1\"), taskList(commonconstants.ReservedTaskListPrefix + \"tl-1\"), true},\n\t\t{taskList(\"tl-1\"), taskList(commonconstants.ReservedTaskListPrefix + \"tl-1\"), taskList(commonconstants.ReservedTaskListPrefix + \"tl-1\"), true},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tkey := tc.defaultVal.Name + \",\" + tc.defaultVal.Kind.String() + \"#\"\n\t\tif tc.input != nil {\n\t\t\tkey += tc.input.GetName()\n\t\t} else {\n\t\t\tkey += \"nil\"\n\t\t}\n\t\ts.Run(key, func() {\n\t\t\toutput, err := s.validator.validatedTaskList(tc.input, tc.defaultVal, metrics.HistoryRespondDecisionTaskCompletedScope, \"domain_name\")\n\t\t\tif tc.isOutputErr {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\ts.EqualValues(tc.output, output)\n\t\t})\n\t}\n}\n\nfunc (s *attrValidatorSuite) TestValidateActivityScheduleAttributes_NoRetryPolicy() {\n\twfTimeout := int32(5)\n\tattributes := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID: \"some random activityID\",\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: \"some random activity type\",\n\t\t},\n\t\tDomain: s.testDomainID,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"some random task list\",\n\t\t},\n\t\tInput:                         []byte{1, 2, 3},\n\t\tScheduleToCloseTimeoutSeconds: nil, // not set\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(3),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(3),  // ScheduleToStart + StartToClose > wfTimeout\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(10), // larger then wfTimeout\n\t}\n\n\texpectedAttributesAfterValidation := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    attributes.ActivityID,\n\t\tActivityType:                  attributes.ActivityType,\n\t\tDomain:                        attributes.Domain,\n\t\tTaskList:                      attributes.TaskList,\n\t\tInput:                         attributes.Input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(wfTimeout),\n\t\tScheduleToStartTimeoutSeconds: attributes.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    attributes.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(wfTimeout),\n\t}\n\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowTimeout: wfTimeout,\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateActivityScheduleAttributes(\n\t\ts.testDomainID,\n\t\ts.testTargetDomainID,\n\t\tattributes,\n\t\texecutionInfo,\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t)\n\ts.Nil(err)\n\ts.Equal(expectedAttributesAfterValidation, attributes)\n}\n\nfunc (s *attrValidatorSuite) TestValidateActivityScheduleAttributes_WithRetryPolicy_ScheduleToStartRetryable() {\n\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomainID).Return(\"some random domain name\", nil).Times(1)\n\n\twfTimeout := int32(3000)\n\tattributes := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID: \"some random activityID\",\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: \"some random activity type\",\n\t\t},\n\t\tDomain: s.testDomainID,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"some random task list\",\n\t\t},\n\t\tInput:                         []byte{1, 2, 3},\n\t\tScheduleToCloseTimeoutSeconds: nil, // not set\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(3),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(500), // extended ScheduleToStart + StartToClose > wfTimeout\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.1,\n\t\t\tExpirationIntervalInSeconds: s.testActivityMaxScheduleToStartTimeoutForRetryInSeconds + 1000, // larger than maximumScheduleToStartTimeoutForRetryInSeconds\n\t\t\tNonRetriableErrorReasons:    []string{\"non-retryable error\"},\n\t\t},\n\t}\n\n\texpectedAttributesAfterValidation := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    attributes.ActivityID,\n\t\tActivityType:                  attributes.ActivityType,\n\t\tDomain:                        attributes.Domain,\n\t\tTaskList:                      attributes.TaskList,\n\t\tInput:                         attributes.Input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(attributes.RetryPolicy.ExpirationIntervalInSeconds),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(s.testActivityMaxScheduleToStartTimeoutForRetryInSeconds),\n\t\tStartToCloseTimeoutSeconds:    attributes.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       attributes.HeartbeatTimeoutSeconds,\n\t\tRetryPolicy:                   attributes.RetryPolicy,\n\t}\n\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowTimeout: wfTimeout,\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateActivityScheduleAttributes(\n\t\ts.testDomainID,\n\t\ts.testTargetDomainID,\n\t\tattributes,\n\t\texecutionInfo,\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t)\n\ts.Nil(err)\n\ts.Equal(expectedAttributesAfterValidation, attributes)\n}\n\nfunc (s *attrValidatorSuite) TestValidateActivityScheduleAttributes_WithRetryPolicy_ScheduleToStartNonRetryable() {\n\twfTimeout := int32(1000)\n\tattributes := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID: \"some random activityID\",\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: \"some random activity type\",\n\t\t},\n\t\tDomain: s.testDomainID,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: \"some random task list\",\n\t\t},\n\t\tInput:                         []byte{1, 2, 3},\n\t\tScheduleToCloseTimeoutSeconds: nil, // not set\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(3),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(500), // extended ScheduleToStart + StartToClose > wfTimeout\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.1,\n\t\t\tExpirationIntervalInSeconds: s.testActivityMaxScheduleToStartTimeoutForRetryInSeconds + 1000, // larger than wfTimeout and maximumScheduleToStartTimeoutForRetryInSeconds\n\t\t\tNonRetriableErrorReasons:    []string{\"cadenceInternal:Timeout SCHEDULE_TO_START\"},\n\t\t},\n\t}\n\n\texpectedAttributesAfterValidation := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    attributes.ActivityID,\n\t\tActivityType:                  attributes.ActivityType,\n\t\tDomain:                        attributes.Domain,\n\t\tTaskList:                      attributes.TaskList,\n\t\tInput:                         attributes.Input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(wfTimeout),\n\t\tScheduleToStartTimeoutSeconds: attributes.ScheduleToStartTimeoutSeconds,\n\t\tStartToCloseTimeoutSeconds:    attributes.StartToCloseTimeoutSeconds,\n\t\tHeartbeatTimeoutSeconds:       attributes.HeartbeatTimeoutSeconds,\n\t\tRetryPolicy:                   attributes.RetryPolicy,\n\t}\n\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\ttargetDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: s.testTargetDomainID},\n\t\tnil,\n\t\tcluster.TestCurrentClusterName,\n\t)\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowTimeout: wfTimeout,\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testDomainID).Return(domainEntry, nil).Times(1)\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.testTargetDomainID).Return(targetDomainEntry, nil).Times(1)\n\n\terr := s.validator.validateActivityScheduleAttributes(\n\t\ts.testDomainID,\n\t\ts.testTargetDomainID,\n\t\tattributes,\n\t\texecutionInfo,\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t)\n\ts.Nil(err)\n\ts.Equal(expectedAttributesAfterValidation, attributes)\n}\n\nconst (\n\ttestDomainID   = \"test-domain-id\"\n\ttestDomainName = \"test-domain\"\n\ttestWorkflowID = \"test-workflow-id\"\n\ttestRunID      = \"test-run-id\"\n)\n\nfunc TestWorkflowSizeChecker_failWorkflowIfBlobSizeExceedsLimit(t *testing.T) {\n\tvar (\n\t\ttestDecisionTag = metrics.DecisionTypeTag(types.DecisionTypeCompleteWorkflowExecution.String())\n\t\ttestEventID     = int64(1)\n\t\ttestMessage     = \"test\"\n\t)\n\n\tfor name, tc := range map[string]struct {\n\t\tblobSizeLimitWarn    int\n\t\tblobSizeLimitError   int\n\t\tblob                 []byte\n\t\tassertLogsAndMetrics func(*testing.T, *observer.ObservedLogs, tally.TestScope)\n\t\texpectFail           bool\n\t}{\n\t\t\"no errors\": {\n\t\t\tblobSizeLimitWarn:  10,\n\t\t\tblobSizeLimitError: 20,\n\t\t\tblob:               []byte(\"test\"),\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tassert.Empty(t, logs.All())\n\t\t\t\t// ensure metrics with the size is emitted.\n\t\t\t\ttimerData := maps.Values(scope.Snapshot().Timers())\n\t\t\t\tassert.Len(t, timerData, 2)\n\t\t\t\tassert.Equal(t, \"test.event_blob_size\", timerData[0].Name())\n\t\t\t},\n\t\t},\n\t\t\"warn\": {\n\t\t\tblobSizeLimitWarn:  10,\n\t\t\tblobSizeLimitError: 20,\n\t\t\tblob:               []byte(\"should-warn\"),\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"Blob size close to the limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t},\n\t\t\"fail\": {\n\t\t\tblobSizeLimitWarn:  5,\n\t\t\tblobSizeLimitError: 10,\n\t\t\tblob:               []byte(\"should-fail\"),\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"Blob size exceeds limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t\texpectFail: true,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmutableState := execution.NewMockMutableState(ctrl)\n\t\t\tlogger, logs := testlogger.NewObserved(t)\n\t\t\tmetricsScope := tally.NewTestScope(\"test\", nil)\n\t\t\tchecker := &workflowSizeChecker{\n\t\t\t\tblobSizeLimitWarn:  tc.blobSizeLimitWarn,\n\t\t\t\tblobSizeLimitError: tc.blobSizeLimitError,\n\t\t\t\tcompletedID:        testEventID,\n\t\t\t\tmutableState:       mutableState,\n\t\t\t\tlogger:             logger,\n\t\t\t\tmetricsScope:       metrics.NewClient(metricsScope, metrics.History, metrics.HistogramMigration{}).Scope(metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DomainTag(testDomainName)),\n\t\t\t}\n\t\t\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:   testDomainID,\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testRunID,\n\t\t\t}).Times(1)\n\t\t\tif tc.expectFail {\n\t\t\t\tmutableState.EXPECT().AddFailWorkflowEvent(testEventID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(testMessage),\n\t\t\t\t}).Return(nil, nil).Times(1)\n\t\t\t}\n\t\t\tfailed, err := checker.failWorkflowIfBlobSizeExceedsLimit(testDecisionTag, tc.blob, testMessage)\n\t\t\trequire.NoError(t, err)\n\t\t\tif tc.assertLogsAndMetrics != nil {\n\t\t\t\ttc.assertLogsAndMetrics(t, logs, metricsScope)\n\t\t\t}\n\t\t\tassert.Equal(t, tc.expectFail, failed)\n\t\t})\n\t}\n\n}\n\nfunc TestWorkflowSizeChecker_failWorkflowSizeExceedsLimit(t *testing.T) {\n\tvar (\n\t\ttestEventID = int64(1)\n\t)\n\n\tfor name, tc := range map[string]struct {\n\t\thistoryCount           int\n\t\thistoryCountLimitWarn  int\n\t\thistoryCountLimitError int\n\n\t\thistorySize           int\n\t\thistorySizeLimitWarn  int\n\t\thistorySizeLimitError int\n\n\t\tnoExecutionCall bool\n\n\t\tassertLogsAndMetrics func(*testing.T, *observer.ObservedLogs, tally.TestScope)\n\t\texpectFail           bool\n\t}{\n\t\t\"no errors\": {\n\t\t\thistoryCount:           1,\n\t\t\thistoryCountLimitWarn:  10,\n\t\t\thistoryCountLimitError: 20,\n\t\t\thistorySize:            1,\n\t\t\thistorySizeLimitWarn:   10,\n\t\t\thistorySizeLimitError:  20,\n\t\t\tnoExecutionCall:        true,\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tassert.Empty(t, logs.All())\n\t\t\t\t// ensure metrics with the size is emitted.\n\t\t\t\ttimerData := maps.Values(scope.Snapshot().Timers())\n\t\t\t\tassert.Len(t, timerData, 4)\n\t\t\t\ttimerNames := make([]string, 0, 4)\n\t\t\t\tfor _, timer := range timerData {\n\t\t\t\t\ttimerNames = append(timerNames, timer.Name())\n\t\t\t\t}\n\t\t\t\tsort.Strings(timerNames)\n\n\t\t\t\t// timers are duplicated for specific domain and domain: all\n\t\t\t\tassert.Equal(t, []string{\"test.history_count\", \"test.history_count\", \"test.history_size\", \"test.history_size\"}, timerNames)\n\t\t\t},\n\t\t},\n\t\t\"count warn\": {\n\t\t\thistoryCount:           15,\n\t\t\thistoryCountLimitWarn:  10,\n\t\t\thistoryCountLimitError: 20,\n\n\t\t\thistorySize:           1,\n\t\t\thistorySizeLimitWarn:  10,\n\t\t\thistorySizeLimitError: 20,\n\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"history size exceeds warn limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t},\n\t\t\"count error\": {\n\t\t\thistoryCount:           25,\n\t\t\thistoryCountLimitWarn:  10,\n\t\t\thistoryCountLimitError: 20,\n\n\t\t\thistorySize:           1,\n\t\t\thistorySizeLimitWarn:  10,\n\t\t\thistorySizeLimitError: 20,\n\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"history size exceeds error limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t\texpectFail: true,\n\t\t},\n\t\t\"size warn\": {\n\t\t\thistoryCount:           1,\n\t\t\thistoryCountLimitWarn:  10,\n\t\t\thistoryCountLimitError: 20,\n\n\t\t\thistorySize:           15,\n\t\t\thistorySizeLimitWarn:  10,\n\t\t\thistorySizeLimitError: 20,\n\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"history size exceeds warn limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t},\n\t\t\"size error\": {\n\t\t\thistoryCount:           1,\n\t\t\thistoryCountLimitWarn:  10,\n\t\t\thistoryCountLimitError: 20,\n\n\t\t\thistorySize:           25,\n\t\t\thistorySizeLimitWarn:  10,\n\t\t\thistorySizeLimitError: 20,\n\n\t\t\tassertLogsAndMetrics: func(t *testing.T, logs *observer.ObservedLogs, scope tally.TestScope) {\n\t\t\t\tlogEntries := logs.All()\n\t\t\t\trequire.Len(t, logEntries, 1)\n\t\t\t\tassert.Equal(t, \"history size exceeds error limit.\", logEntries[0].Message)\n\t\t\t},\n\t\t\texpectFail: true,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmutableState := execution.NewMockMutableState(ctrl)\n\t\t\tlogger, logs := testlogger.NewObserved(t)\n\t\t\tmetricsScope := tally.NewTestScope(\"test\", nil)\n\n\t\t\tmutableState.EXPECT().GetNextEventID().Return(int64(tc.historyCount + 1)).Times(1)\n\t\t\tif !tc.noExecutionCall {\n\t\t\t\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t}).Times(1)\n\t\t\t}\n\t\t\tif tc.expectFail {\n\t\t\t\tmutableState.EXPECT().AddFailWorkflowEvent(testEventID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"Workflow history size / count exceeds limit.\"),\n\t\t\t\t}).Return(nil, nil).Times(1)\n\t\t\t}\n\n\t\t\tchecker := &workflowSizeChecker{\n\t\t\t\tcompletedID:            testEventID,\n\t\t\t\thistoryCountLimitWarn:  tc.historyCountLimitWarn,\n\t\t\t\thistoryCountLimitError: tc.historyCountLimitError,\n\t\t\t\thistorySizeLimitWarn:   tc.historySizeLimitWarn,\n\t\t\t\thistorySizeLimitError:  tc.historySizeLimitError,\n\t\t\t\tmutableState:           mutableState,\n\t\t\t\texecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\tHistorySize: int64(tc.historySize),\n\t\t\t\t},\n\t\t\t\tlogger:       logger,\n\t\t\t\tmetricsScope: metrics.NewClient(metricsScope, metrics.History, metrics.HistogramMigration{}).Scope(metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DomainTag(testDomainName)),\n\t\t\t}\n\t\t\tfailed, err := checker.failWorkflowSizeExceedsLimit()\n\t\t\trequire.NoError(t, err)\n\t\t\tif tc.assertLogsAndMetrics != nil {\n\t\t\t\ttc.assertLogsAndMetrics(t, logs, metricsScope)\n\t\t\t}\n\t\t\tassert.Equal(t, tc.expectFail, failed)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/decision/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage decision\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\ntype (\n\t// Handler contains decision business logic\n\tHandler interface {\n\t\tHandleDecisionTaskScheduled(context.Context, *types.ScheduleDecisionTaskRequest) error\n\t\tHandleDecisionTaskStarted(context.Context,\n\t\t\t*types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error)\n\t\tHandleDecisionTaskFailed(context.Context,\n\t\t\t*types.HistoryRespondDecisionTaskFailedRequest) error\n\t\tHandleDecisionTaskCompleted(context.Context,\n\t\t\t*types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error)\n\t\t// TODO also include the handle of decision timeout here\n\t}\n\n\thandlerImpl struct {\n\t\tconfig               *config.Config\n\t\tshard                shard.Context\n\t\ttimeSource           clock.TimeSource\n\t\tdomainCache          cache.DomainCache\n\t\texecutionCache       execution.Cache\n\t\ttokenSerializer      common.TaskTokenSerializer\n\t\tmetricsClient        metrics.Client\n\t\tlogger               log.Logger\n\t\tthrottledLogger      log.Logger\n\t\tattrValidator        *attrValidator\n\t\tversionChecker       client.VersionChecker\n\t\tactiveClusterManager activecluster.Manager\n\t}\n)\n\n// NewHandler creates a new Handler for handling decision business logic\nfunc NewHandler(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\ttokenSerializer common.TaskTokenSerializer,\n) Handler {\n\tconfig := shard.GetConfig()\n\tlogger := shard.GetLogger().WithTags(tag.ComponentDecisionHandler)\n\treturn &handlerImpl{\n\t\tconfig:               config,\n\t\tshard:                shard,\n\t\ttimeSource:           shard.GetTimeSource(),\n\t\tdomainCache:          shard.GetDomainCache(),\n\t\texecutionCache:       executionCache,\n\t\ttokenSerializer:      tokenSerializer,\n\t\tmetricsClient:        shard.GetMetricsClient(),\n\t\tlogger:               shard.GetLogger().WithTags(tag.ComponentDecisionHandler),\n\t\tactiveClusterManager: shard.GetActiveClusterManager(),\n\t\tthrottledLogger:      shard.GetThrottledLogger().WithTags(tag.ComponentDecisionHandler),\n\t\tattrValidator: newAttrValidator(\n\t\t\tshard.GetDomainCache(),\n\t\t\tshard.GetMetricsClient(),\n\t\t\tconfig,\n\t\t\tlogger,\n\t\t),\n\t\tversionChecker: client.NewVersionChecker(),\n\t}\n}\n\nfunc (handler *handlerImpl) HandleDecisionTaskScheduled(\n\tctx context.Context,\n\treq *types.ScheduleDecisionTaskRequest,\n) error {\n\n\tdomainEntry, err := handler.getActiveDomainByWorkflow(ctx, req.DomainUUID, req.WorkflowExecution.WorkflowID, req.WorkflowExecution.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: req.WorkflowExecution.WorkflowID,\n\t\tRunID:      req.WorkflowExecution.RunID,\n\t}\n\n\treturn workflow.UpdateWithActionFunc(\n\t\tctx,\n\t\thandler.logger,\n\t\thandler.executionCache,\n\t\tdomainID,\n\t\tworkflowExecution,\n\t\thandler.timeSource.Now(),\n\t\tfunc(context execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn nil, workflow.ErrNotExists\n\t\t\t}\n\n\t\t\tif mutableState.HasProcessedOrPendingDecision() {\n\t\t\t\treturn &workflow.UpdateAction{\n\t\t\t\t\tNoop: true,\n\t\t\t\t}, nil\n\t\t\t}\n\n\t\t\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tif err := mutableState.AddFirstDecisionTaskScheduled(\n\t\t\t\tstartEvent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\treturn &workflow.UpdateAction{}, nil\n\t\t},\n\t)\n}\n\nfunc (handler *handlerImpl) HandleDecisionTaskStarted(\n\tctx context.Context,\n\treq *types.RecordDecisionTaskStartedRequest,\n) (*types.RecordDecisionTaskStartedResponse, error) {\n\n\tdomainEntry, err := handler.getActiveDomainByWorkflow(ctx, req.DomainUUID, req.WorkflowExecution.WorkflowID, req.WorkflowExecution.RunID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: req.WorkflowExecution.WorkflowID,\n\t\tRunID:      req.WorkflowExecution.RunID,\n\t}\n\n\tscheduleID := req.GetScheduleID()\n\trequestID := req.GetRequestID()\n\n\tvar resp *types.RecordDecisionTaskStartedResponse\n\terr = workflow.UpdateWithActionFunc(\n\t\tctx,\n\t\thandler.logger,\n\t\thandler.executionCache,\n\t\tdomainID,\n\t\tworkflowExecution,\n\t\thandler.timeSource.Now(),\n\t\tfunc(context execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn nil, workflow.ErrNotExists\n\t\t\t}\n\n\t\t\tdecision, isRunning := mutableState.GetDecisionInfo(scheduleID)\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\thandler.metricsClient.IncCounter(metrics.HistoryRecordDecisionTaskStartedScope, metrics.StaleMutableStateCounter)\n\t\t\t\thandler.logger.Error(\"Encounter stale mutable state in RecordDecisionTaskStarted\",\n\t\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\t// Reload workflow execution history\n\t\t\t\t// ErrStaleState will trigger updateWorkflowExecutionWithAction function to reload the mutable state\n\t\t\t\treturn nil, workflow.ErrStaleState\n\t\t\t}\n\n\t\t\t// Check execution state to make sure task is in the list of outstanding tasks and it is not yet started.  If\n\t\t\t// task is not outstanding than it is most probably a duplicate and complete the task.\n\t\t\tif !isRunning {\n\t\t\t\t// Looks like DecisionTask already completed as a result of another call.\n\t\t\t\t// It is OK to drop the task at this point.\n\t\t\t\treturn nil, &types.EntityNotExistsError{Message: \"Decision task not found.\"}\n\t\t\t}\n\n\t\t\tupdateAction := &workflow.UpdateAction{}\n\n\t\t\tif decision.StartedID != constants.EmptyEventID {\n\t\t\t\t// If decision is started as part of the current request scope then return a positive response\n\t\t\t\tif decision.RequestID == requestID {\n\t\t\t\t\tresp, err = handler.createRecordDecisionTaskStartedResponse(domainID, mutableState, decision, req.PollRequest.GetIdentity())\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn nil, err\n\t\t\t\t\t}\n\t\t\t\t\tupdateAction.Noop = true\n\t\t\t\t\treturn updateAction, nil\n\t\t\t\t}\n\n\t\t\t\t// Looks like DecisionTask already started as a result of another call.\n\t\t\t\t// It is OK to drop the task at this point.\n\t\t\t\treturn nil, &types.EventAlreadyStartedError{Message: \"Decision task already started.\"}\n\t\t\t}\n\n\t\t\t_, decision, err = mutableState.AddDecisionTaskStartedEvent(scheduleID, requestID, req.PollRequest)\n\t\t\tif err != nil {\n\t\t\t\t// Unable to add DecisionTaskStarted event to history\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to add DecisionTaskStarted event to history.\"}\n\t\t\t}\n\n\t\t\tresp, err = handler.createRecordDecisionTaskStartedResponse(domainID, mutableState, decision, req.PollRequest.GetIdentity())\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treturn updateAction, nil\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (handler *handlerImpl) HandleDecisionTaskFailed(\n\tctx context.Context,\n\treq *types.HistoryRespondDecisionTaskFailedRequest,\n) (retError error) {\n\n\trequest := req.FailedRequest\n\ttoken, err := handler.tokenSerializer.Deserialize(request.TaskToken)\n\tif err != nil {\n\t\treturn workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := handler.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\treturn workflow.UpdateWithAction(ctx, handler.logger, handler.executionCache, domainID, workflowExecution, true, handler.timeSource.Now(),\n\t\tfunc(context execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\tscheduleID := token.ScheduleID\n\t\t\tdecision, isRunning := mutableState.GetDecisionInfo(scheduleID)\n\t\t\tif !isRunning || decision.Attempt != token.ScheduleAttempt || decision.StartedID == constants.EmptyEventID {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Decision task not found.\"}\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddDecisionTaskFailedEvent(decision.ScheduleID, decision.StartedID, request.GetCause(), request.Details,\n\t\t\t\trequest.GetIdentity(), \"\", request.GetBinaryChecksum(), \"\", \"\", 0, \"\")\n\t\t\treturn err\n\t\t})\n}\n\nfunc (handler *handlerImpl) HandleDecisionTaskCompleted(\n\tctx context.Context,\n\treq *types.HistoryRespondDecisionTaskCompletedRequest,\n) (resp *types.HistoryRespondDecisionTaskCompletedResponse, retError error) {\n\trequest := req.CompleteRequest\n\ttoken, err0 := handler.tokenSerializer.Deserialize(request.TaskToken)\n\tif err0 != nil {\n\t\treturn nil, workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := handler.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\tdomainName := domainEntry.GetInfo().Name\n\tlogger := handler.logger.WithTags(\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\ttag.WorkflowScheduleID(token.ScheduleID),\n\t)\n\tscope := handler.metricsClient.Scope(metrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DomainTag(domainName),\n\t\tmetrics.WorkflowTypeTag(token.WorkflowType))\n\n\tcall := yarpc.CallFromContext(ctx)\n\tclientLibVersion := call.Header(common.LibraryVersionHeaderName)\n\tclientFeatureVersion := call.Header(common.FeatureVersionHeaderName)\n\tclientImpl := call.Header(common.ClientImplHeaderName)\n\n\twfContext, release, err := handler.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, workflowExecution)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer func() { release(retError) }()\n\nUpdate_History_Loop:\n\tfor attempt := 0; attempt < workflow.ConditionalRetryCount; attempt++ {\n\t\tlogger.Debug(\"Update_History_Loop attempt\", tag.Attempt(int32(attempt)))\n\t\tmsBuilder, err := wfContext.LoadWorkflowExecution(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tif !msBuilder.IsWorkflowExecutionRunning() {\n\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t}\n\t\texecutionStats, err := wfContext.LoadExecutionStats(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\texecutionInfo := msBuilder.GetExecutionInfo()\n\t\tcurrentDecision, isRunning := msBuilder.GetDecisionInfo(token.ScheduleID)\n\n\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t// some extreme cassandra failure cases.\n\t\tif !isRunning && token.ScheduleID >= msBuilder.GetNextEventID() {\n\t\t\tscope.IncCounter(metrics.StaleMutableStateCounter)\n\t\t\tlogger.Error(\"Encounter stale mutable state in RespondDecisionTaskCompleted\", tag.WorkflowNextEventID(msBuilder.GetNextEventID()))\n\t\t\t// Reload workflow execution history\n\t\t\twfContext.Clear()\n\t\t\tcontinue Update_History_Loop\n\t\t}\n\n\t\tif !msBuilder.IsWorkflowExecutionRunning() || !isRunning || currentDecision.Attempt != token.ScheduleAttempt || currentDecision.StartedID == constants.EmptyEventID {\n\t\t\tlogger.Debugf(\"Decision task not found. IsWorkflowExecutionRunning: %v, isRunning: %v, currentDecision.Attempt: %v, token.ScheduleAttempt: %v, currentDecision.StartID: %v\",\n\t\t\t\tmsBuilder.IsWorkflowExecutionRunning(), isRunning, getDecisionInfoAttempt(currentDecision), token.ScheduleAttempt, getDecisionInfoStartedID(currentDecision))\n\t\t\treturn nil, &types.EntityNotExistsError{Message: \"Decision task not found.\"}\n\t\t}\n\n\t\tstartedID := currentDecision.StartedID\n\t\tmaxResetPoints := handler.config.MaxAutoResetPoints(domainEntry.GetInfo().Name)\n\t\tif msBuilder.GetExecutionInfo().AutoResetPoints != nil && maxResetPoints == len(msBuilder.GetExecutionInfo().AutoResetPoints.Points) {\n\t\t\tlogger.Debugf(\"Max reset points %d is exceeded\", maxResetPoints)\n\t\t\tscope.IncCounter(metrics.AutoResetPointsLimitExceededCounter)\n\t\t}\n\n\t\tdecisionHeartbeating := request.GetForceCreateNewDecisionTask() && len(request.Decisions) == 0\n\t\tvar decisionHeartbeatTimeout bool\n\t\tvar completedEvent *types.HistoryEvent\n\t\tif decisionHeartbeating {\n\t\t\ttimeout := handler.config.DecisionHeartbeatTimeout(domainName)\n\t\t\tif currentDecision.OriginalScheduledTimestamp > 0 && handler.timeSource.Now().After(time.Unix(0, currentDecision.OriginalScheduledTimestamp).Add(timeout)) {\n\t\t\t\tdecisionHeartbeatTimeout = true\n\t\t\t\tscope.IncCounter(metrics.DecisionHeartbeatTimeoutCounter)\n\t\t\t\tcompletedEvent, err = msBuilder.AddDecisionTaskTimedOutEvent(currentDecision.ScheduleID, currentDecision.StartedID)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Failed to add decision timeout event.\"}\n\t\t\t\t}\n\t\t\t\tmsBuilder.ClearStickyness()\n\t\t\t} else {\n\t\t\t\tlogger.Debug(\"Adding DecisionTaskCompletedEvent to mutable state for heartbeat\")\n\t\t\t\tcompletedEvent, err = msBuilder.AddDecisionTaskCompletedEvent(token.ScheduleID, startedID, request, maxResetPoints)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to add DecisionTaskCompleted event to history.\"}\n\t\t\t\t}\n\t\t\t}\n\t\t} else {\n\t\t\tcompletedEvent, err = msBuilder.AddDecisionTaskCompletedEvent(token.ScheduleID, startedID, request, maxResetPoints)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to add DecisionTaskCompleted event to history.\"}\n\t\t\t}\n\t\t}\n\n\t\tvar (\n\t\t\tfailDecision                bool\n\t\t\tfailCause                   types.DecisionTaskFailedCause\n\t\t\tfailMessage                 string\n\t\t\tactivityNotStartedCancelled bool\n\t\t\tcontinueAsNewBuilder        execution.MutableState\n\t\t\thasUnhandledEvents          bool\n\t\t\tdecisionResults             []*decisionResult\n\t\t)\n\t\thasUnhandledEvents = msBuilder.HasBufferedEvents()\n\n\t\tif request.StickyAttributes == nil || request.StickyAttributes.WorkerTaskList == nil {\n\t\t\tscope.IncCounter(metrics.CompleteDecisionWithStickyDisabledCounter)\n\t\t\texecutionInfo.StickyTaskList = \"\"\n\t\t\texecutionInfo.StickyScheduleToStartTimeout = 0\n\t\t} else {\n\t\t\tscope.IncCounter(metrics.CompleteDecisionWithStickyEnabledCounter)\n\t\t\texecutionInfo.StickyTaskList = request.StickyAttributes.WorkerTaskList.GetName()\n\t\t\texecutionInfo.StickyScheduleToStartTimeout = request.StickyAttributes.GetScheduleToStartTimeoutSeconds()\n\t\t}\n\t\texecutionInfo.ClientLibraryVersion = clientLibVersion\n\t\texecutionInfo.ClientFeatureVersion = clientFeatureVersion\n\t\texecutionInfo.ClientImpl = clientImpl\n\n\t\tbinChecksum := request.GetBinaryChecksum()\n\t\tif _, ok := domainEntry.GetConfig().BadBinaries.Binaries[binChecksum]; ok {\n\t\t\tfailDecision = true\n\t\t\tfailCause = types.DecisionTaskFailedCauseBadBinary\n\t\t\tfailMessage = fmt.Sprintf(\"binary %v is already marked as bad deployment\", binChecksum)\n\t\t} else {\n\t\t\tworkflowSizeChecker := newWorkflowSizeChecker(\n\t\t\t\tdomainName,\n\t\t\t\thandler.config.BlobSizeLimitWarn(domainName),\n\t\t\t\thandler.config.BlobSizeLimitError(domainName),\n\t\t\t\thandler.config.HistorySizeLimitWarn(domainName),\n\t\t\t\thandler.config.HistorySizeLimitError(domainName),\n\t\t\t\thandler.config.HistoryCountLimitWarn(domainName),\n\t\t\t\thandler.config.HistoryCountLimitError(domainName),\n\t\t\t\tcompletedEvent.ID,\n\t\t\t\tmsBuilder,\n\t\t\t\texecutionStats,\n\t\t\t\thandler.metricsClient.Scope(metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DomainTag(domainName)),\n\t\t\t\thandler.logger,\n\t\t\t)\n\n\t\t\tdecisionTaskHandler := newDecisionTaskHandler(\n\t\t\t\trequest.GetIdentity(),\n\t\t\t\tcompletedEvent.ID,\n\t\t\t\tdomainEntry,\n\t\t\t\tmsBuilder,\n\t\t\t\thandler.attrValidator,\n\t\t\t\tworkflowSizeChecker,\n\t\t\t\thandler.tokenSerializer,\n\t\t\t\thandler.logger,\n\t\t\t\thandler.domainCache,\n\t\t\t\thandler.metricsClient,\n\t\t\t\thandler.config,\n\t\t\t)\n\n\t\t\tif decisionResults, err = decisionTaskHandler.handleDecisions(\n\t\t\t\tctx,\n\t\t\t\trequest.ExecutionContext,\n\t\t\t\trequest.Decisions,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\t// set the vars used by following logic\n\t\t\t// further refactor should also clean up the vars used below\n\t\t\tfailDecision = decisionTaskHandler.failDecision\n\t\t\tif failDecision {\n\t\t\t\tfailCause = *decisionTaskHandler.failDecisionCause\n\t\t\t\tfailMessage = *decisionTaskHandler.failMessage\n\t\t\t}\n\n\t\t\t// failMessage is not used by decisionTaskHandler\n\t\t\tactivityNotStartedCancelled = decisionTaskHandler.activityNotStartedCancelled\n\t\t\t// continueAsNewTimerTasks is not used by decisionTaskHandler\n\t\t\tcontinueAsNewBuilder = decisionTaskHandler.continueAsNewBuilder\n\t\t\thasUnhandledEvents = decisionTaskHandler.hasUnhandledEventsBeforeDecisions\n\t\t}\n\n\t\tif failDecision {\n\t\t\tscope.IncCounter(metrics.FailedDecisionsCounter)\n\t\t\tlogger.Info(\"Failing the decision.\", tag.WorkflowDecisionFailCause(int64(failCause)))\n\t\t\tmsBuilder, err = handler.failDecisionHelper(\n\t\t\t\tctx, wfContext, token.ScheduleID, startedID, failCause, []byte(failMessage), request, domainEntry)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\thasUnhandledEvents = true\n\t\t\tcontinueAsNewBuilder = nil\n\t\t}\n\n\t\tcreateNewDecisionTask := msBuilder.IsWorkflowExecutionRunning() && (hasUnhandledEvents || request.GetForceCreateNewDecisionTask() || activityNotStartedCancelled)\n\t\tlogger.Debugf(\"createNewDecisionTask: %v, msBuilder.IsWorkflowExecutionRunning: %v, hasUnhandledEvents: %v, request.GetForceCreateNewDecisionTask: %v, activityNotStartedCancelled: %v\",\n\t\t\tcreateNewDecisionTask, msBuilder.IsWorkflowExecutionRunning(), hasUnhandledEvents, request.GetForceCreateNewDecisionTask(), activityNotStartedCancelled)\n\t\tvar newDecisionTaskScheduledID int64\n\t\tif createNewDecisionTask {\n\t\t\tvar newDecision *execution.DecisionInfo\n\t\t\tvar err error\n\t\t\tif decisionHeartbeating && !decisionHeartbeatTimeout {\n\t\t\t\tnewDecision, err = msBuilder.AddDecisionTaskScheduledEventAsHeartbeat(\n\t\t\t\t\trequest.GetReturnNewDecisionTask(),\n\t\t\t\t\tcurrentDecision.OriginalScheduledTimestamp,\n\t\t\t\t)\n\t\t\t} else {\n\t\t\t\tnewDecision, err = msBuilder.AddDecisionTaskScheduledEvent(\n\t\t\t\t\trequest.GetReturnNewDecisionTask(),\n\t\t\t\t)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Failed to add decision scheduled event.\"}\n\t\t\t}\n\n\t\t\tnewDecisionTaskScheduledID = newDecision.ScheduleID\n\t\t\t// skip transfer task for decision if request asking to return new decision task\n\t\t\tif request.GetReturnNewDecisionTask() {\n\t\t\t\tlogger.Debugf(\"Adding DecisionTaskStartedEvent to mutable state. new decision's ScheduleID: %d, TaskList: %s\", newDecisionTaskScheduledID, newDecision.TaskList)\n\t\t\t\t// start the new decision task if request asked to do so\n\t\t\t\t// TODO: replace the poll request\n\t\t\t\t_, _, err := msBuilder.AddDecisionTaskStartedEvent(newDecision.ScheduleID, \"request-from-RespondDecisionTaskCompleted\", &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{Name: newDecision.TaskList},\n\t\t\t\t\tIdentity: request.Identity,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\t// We apply the update to execution using optimistic concurrency.  If it fails due to a conflict then reload\n\t\t// the history and try the operation again.\n\t\tvar updateErr error\n\t\tif continueAsNewBuilder != nil {\n\t\t\tcontinueAsNewExecutionInfo := continueAsNewBuilder.GetExecutionInfo()\n\t\t\tlogger.Debugf(\"Updating execution with continue as new info. new wfid: %s, runid: %s\", continueAsNewExecutionInfo.WorkflowID, continueAsNewExecutionInfo.RunID)\n\t\t\tupdateErr = wfContext.UpdateWorkflowExecutionWithNewAsActive(\n\t\t\t\tctx,\n\t\t\t\thandler.shard.GetTimeSource().Now(),\n\t\t\t\texecution.NewContext(\n\t\t\t\t\tcontinueAsNewExecutionInfo.DomainID,\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: continueAsNewExecutionInfo.WorkflowID,\n\t\t\t\t\t\tRunID:      continueAsNewExecutionInfo.RunID,\n\t\t\t\t\t},\n\t\t\t\t\thandler.shard,\n\t\t\t\t\thandler.shard.GetExecutionManager(),\n\t\t\t\t\thandler.logger,\n\t\t\t\t),\n\t\t\t\tcontinueAsNewBuilder,\n\t\t\t)\n\t\t} else {\n\t\t\thandler.logger.Debug(\"HandleDecisionTaskCompleted calling UpdateWorkflowExecutionAsActive\", tag.WorkflowID(msBuilder.GetExecutionInfo().WorkflowID))\n\t\t\tupdateErr = wfContext.UpdateWorkflowExecutionAsActive(ctx, handler.shard.GetTimeSource().Now())\n\t\t}\n\n\t\tif updateErr != nil {\n\t\t\tif execution.IsConflictError(updateErr) {\n\t\t\t\tscope.IncCounter(metrics.ConcurrencyUpdateFailureCounter)\n\t\t\t\tlogger.Error(\"Encounter conflict error while updating workflow execution\",\n\t\t\t\t\ttag.WorkflowID(msBuilder.GetExecutionInfo().WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(msBuilder.GetExecutionInfo().RunID),\n\t\t\t\t\ttag.Error(updateErr),\n\t\t\t\t)\n\t\t\t\tcontinue Update_History_Loop\n\t\t\t}\n\n\t\t\t// if updateErr resulted in TransactionSizeLimitError then fail workflow\n\t\t\tswitch updateErr.(type) {\n\t\t\tcase *persistence.TransactionSizeLimitError:\n\t\t\t\t// must reload mutable state because the first call to updateWorkflowExecutionWithContext or continueAsNewWorkflowExecution\n\t\t\t\t// clears mutable state if error is returned\n\t\t\t\tmsBuilder, err = wfContext.LoadWorkflowExecution(ctx)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\n\t\t\t\teventBatchFirstEventID := msBuilder.GetNextEventID()\n\t\t\t\tif err := execution.TerminateWorkflow(\n\t\t\t\t\tmsBuilder,\n\t\t\t\t\teventBatchFirstEventID,\n\t\t\t\t\tcommon.FailureReasonTransactionSizeExceedsLimit,\n\t\t\t\t\t[]byte(updateErr.Error()),\n\t\t\t\t\texecution.IdentityHistoryService,\n\t\t\t\t); err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\n\t\t\t\thandler.logger.Debug(\"HandleDecisionTaskCompleted calling UpdateWorkflowExecutionAsActive\", tag.WorkflowID(msBuilder.GetExecutionInfo().WorkflowID))\n\t\t\t\tif err := wfContext.UpdateWorkflowExecutionAsActive(\n\t\t\t\t\tctx,\n\t\t\t\t\thandler.shard.GetTimeSource().Now(),\n\t\t\t\t); err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t}\n\n\t\t\treturn nil, updateErr\n\t\t}\n\n\t\thandler.handleBufferedQueries(\n\t\t\tmsBuilder,\n\t\t\tclientImpl,\n\t\t\tclientFeatureVersion,\n\t\t\treq.GetCompleteRequest().GetQueryResults(),\n\t\t\tcreateNewDecisionTask,\n\t\t\tdomainEntry,\n\t\t\tdecisionHeartbeating)\n\n\t\tif decisionHeartbeatTimeout {\n\t\t\t// at this point, update is successful, but we still return an error to client so that the worker will give up this workflow\n\t\t\treturn nil, &types.EntityNotExistsError{\n\t\t\t\tMessage: \"decision heartbeat timeout\",\n\t\t\t}\n\t\t}\n\n\t\tresp = &types.HistoryRespondDecisionTaskCompletedResponse{}\n\t\tif !msBuilder.IsWorkflowExecutionRunning() {\n\t\t\t// Workflow has been completed/terminated, so there is no need to dispatch more activity/decision tasks.\n\t\t\treturn resp, nil\n\t\t}\n\n\t\tactivitiesToDispatchLocally := make(map[string]*types.ActivityLocalDispatchInfo)\n\t\tfor _, dr := range decisionResults {\n\t\t\tif dr.activityDispatchInfo != nil {\n\t\t\t\tactivitiesToDispatchLocally[dr.activityDispatchInfo.ActivityID] = dr.activityDispatchInfo\n\t\t\t}\n\t\t}\n\t\tlogger.Debugf(\"%d activities will be dispatched locally on the client side\", len(activitiesToDispatchLocally))\n\t\tresp.ActivitiesToDispatchLocally = activitiesToDispatchLocally\n\n\t\tif request.GetReturnNewDecisionTask() && createNewDecisionTask {\n\t\t\tdecision, _ := msBuilder.GetDecisionInfo(newDecisionTaskScheduledID)\n\t\t\tresp.StartedResponse, err = handler.createRecordDecisionTaskStartedResponse(domainID, msBuilder, decision, request.GetIdentity())\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\t// sticky is always enabled when worker request for new decision task from RespondDecisionTaskCompleted\n\t\t\tresp.StartedResponse.StickyExecutionEnabled = true\n\t\t}\n\n\t\treturn resp, nil\n\t}\n\n\treturn nil, workflow.ErrMaxAttemptsExceeded\n}\n\nfunc (handler *handlerImpl) createRecordDecisionTaskStartedResponse(\n\tdomainID string,\n\tmsBuilder execution.MutableState,\n\tdecision *execution.DecisionInfo,\n\tidentity string,\n) (*types.RecordDecisionTaskStartedResponse, error) {\n\n\tresponse := &types.RecordDecisionTaskStartedResponse{}\n\tresponse.WorkflowType = msBuilder.GetWorkflowType()\n\texecutionInfo := msBuilder.GetExecutionInfo()\n\tif executionInfo.LastProcessedEvent != constants.EmptyEventID {\n\t\tresponse.PreviousStartedEventID = common.Int64Ptr(executionInfo.LastProcessedEvent)\n\t}\n\n\t// Starting decision could result in different scheduleID if decision was transient and new new events came in\n\t// before it was started.\n\tresponse.ScheduledEventID = decision.ScheduleID\n\tresponse.StartedEventID = decision.StartedID\n\t// if we call IsStickyTaskListEnabled then it's possible that the decision is a\n\t// sticky decision but due to TTL check, the field becomes false\n\t// NOTE: it's possible that StickyTaskList is empty is even if the decision is scheduled\n\t// on a sticky tasklist since stickiness can be cleared at anytime by the ResetStickyTaskList API.\n\t// When this field is false, we will send full workflow history to client\n\t// (see createPollForDecisionTaskResponse in workflowHandler.go)\n\t// This is actually desired since if that API is called, it basically means the client side\n\t// cache has been cleared for the workflow and full history is needed by the client. Even if\n\t// client side still has the cache, client library is still able to handle the situation.\n\tresponse.StickyExecutionEnabled = msBuilder.GetExecutionInfo().StickyTaskList != \"\"\n\tresponse.NextEventID = msBuilder.GetNextEventID()\n\tresponse.Attempt = decision.Attempt\n\tresponse.WorkflowExecutionTaskList = &types.TaskList{\n\t\tName: executionInfo.TaskList,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\tresponse.ScheduledTimestamp = common.Int64Ptr(decision.ScheduledTimestamp)\n\tresponse.StartedTimestamp = common.Int64Ptr(decision.StartedTimestamp)\n\n\tif decision.Attempt > 0 {\n\t\t// This decision is retried from mutable state\n\t\t// Also return schedule and started which are not written to history yet\n\t\tscheduledEvent, startedEvent := msBuilder.CreateTransientDecisionEvents(decision, identity)\n\t\tresponse.DecisionInfo = &types.TransientDecisionInfo{}\n\t\tresponse.DecisionInfo.ScheduledEvent = scheduledEvent\n\t\tresponse.DecisionInfo.StartedEvent = startedEvent\n\t}\n\tcurrentBranchToken, err := msBuilder.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresponse.BranchToken = currentBranchToken\n\n\tqr := msBuilder.GetQueryRegistry()\n\tbuffered := qr.GetBufferedIDs()\n\tqueries := make(map[string]*types.WorkflowQuery)\n\tfor _, id := range buffered {\n\t\tinput, err := qr.GetQueryInput(id)\n\t\tif err != nil {\n\t\t\tcontinue\n\t\t}\n\t\tqueries[id] = input\n\t}\n\tresponse.Queries = queries\n\tresponse.HistorySize = msBuilder.GetHistorySize()\n\treturn response, nil\n}\n\nfunc (handler *handlerImpl) handleBufferedQueries(\n\tmsBuilder execution.MutableState,\n\tclientImpl string,\n\tclientFeatureVersion string,\n\tqueryResults map[string]*types.WorkflowQueryResult,\n\tcreateNewDecisionTask bool,\n\tdomainEntry *cache.DomainCacheEntry,\n\tdecisionHeartbeating bool,\n) {\n\tqueryRegistry := msBuilder.GetQueryRegistry()\n\tif !queryRegistry.HasBufferedQuery() {\n\t\treturn\n\t}\n\n\tdomainID := domainEntry.GetInfo().ID\n\tdomain := domainEntry.GetInfo().Name\n\tworkflowID := msBuilder.GetExecutionInfo().WorkflowID\n\trunID := msBuilder.GetExecutionInfo().RunID\n\n\tscope := handler.metricsClient.Scope(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DomainTag(domainEntry.GetInfo().Name),\n\t\tmetrics.DecisionTypeTag(\"ConsistentQuery\"))\n\n\t// Consistent query requires both server and client worker support. If a consistent query was requested (meaning there are\n\t// buffered queries) but worker does not support consistent query then all buffered queries should be failed.\n\tversionErr := handler.versionChecker.SupportsConsistentQuery(clientImpl, clientFeatureVersion)\n\t// todo (David.Porter) remove the skip on version check for\n\t// clientImpl and clientFeatureVersion where they're nil\n\t// There's a bug, probably in matching somewhere which isn't\n\t// forwarding the client headers for version\n\t// info correctly making this call erroneously fail sometimes.\n\t// https://t3.uberinternal.com/browse/CDNC-8641\n\t// So defaulting just this flow to fail-open in the absence of headers.\n\tif versionErr != nil && clientImpl != \"\" && clientFeatureVersion != \"\" {\n\t\tscope.IncCounter(metrics.WorkerNotSupportsConsistentQueryCount)\n\t\tfailedTerminationState := &query.TerminationState{\n\t\t\tTerminationType: query.TerminationTypeFailed,\n\t\t\tFailure:         &types.BadRequestError{Message: versionErr.Error()},\n\t\t}\n\t\tbuffered := queryRegistry.GetBufferedIDs()\n\t\thandler.logger.Info(\n\t\t\t\"failing query because worker does not support consistent query\",\n\t\t\ttag.ClientImpl(clientImpl),\n\t\t\ttag.ClientFeatureVersion(clientFeatureVersion),\n\t\t\ttag.WorkflowDomainName(domain),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.Error(versionErr))\n\t\tfor _, id := range buffered {\n\t\t\tif err := queryRegistry.SetTerminationState(id, failedTerminationState); err != nil {\n\t\t\t\thandler.logger.Error(\n\t\t\t\t\t\"failed to set query termination state to failed\",\n\t\t\t\t\ttag.WorkflowDomainName(domain),\n\t\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\t\ttag.QueryID(id),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\t}\n\t\t}\n\t\treturn\n\t}\n\n\t// if its a heartbeat decision it means local activities may still be running on the worker\n\t// which were started by an external event which happened before the query\n\tif decisionHeartbeating {\n\t\treturn\n\t}\n\n\tsizeLimitError := handler.config.BlobSizeLimitError(domain)\n\tsizeLimitWarn := handler.config.BlobSizeLimitWarn(domain)\n\n\t// Complete or fail all queries we have results for\n\tfor id, result := range queryResults {\n\t\tif err := common.CheckEventBlobSizeLimit(\n\t\t\tlen(result.GetAnswer()),\n\t\t\tsizeLimitWarn,\n\t\t\tsizeLimitError,\n\t\t\tdomainID,\n\t\t\tdomain,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\tscope,\n\t\t\thandler.logger,\n\t\t\ttag.BlobSizeViolationOperation(\"ConsistentQuery\"),\n\t\t); err != nil {\n\t\t\thandler.logger.Info(\"failing query because query result size is too large\",\n\t\t\t\ttag.WorkflowDomainName(domain),\n\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\ttag.QueryID(id),\n\t\t\t\ttag.Error(err))\n\t\t\tfailedTerminationState := &query.TerminationState{\n\t\t\t\tTerminationType: query.TerminationTypeFailed,\n\t\t\t\tFailure:         err,\n\t\t\t}\n\t\t\tif err := queryRegistry.SetTerminationState(id, failedTerminationState); err != nil {\n\t\t\t\thandler.logger.Error(\n\t\t\t\t\t\"failed to set query termination state to failed\",\n\t\t\t\t\ttag.WorkflowDomainName(domain),\n\t\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\t\ttag.QueryID(id),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\t}\n\t\t} else {\n\t\t\tcompletedTerminationState := &query.TerminationState{\n\t\t\t\tTerminationType: query.TerminationTypeCompleted,\n\t\t\t\tQueryResult:     result,\n\t\t\t}\n\t\t\tif err := queryRegistry.SetTerminationState(id, completedTerminationState); err != nil {\n\t\t\t\thandler.logger.Error(\n\t\t\t\t\t\"failed to set query termination state to completed\",\n\t\t\t\t\ttag.WorkflowDomainName(domain),\n\t\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\t\ttag.QueryID(id),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\t}\n\t\t}\n\t}\n\n\t// If no decision task was created then it means no buffered events came in during this decision task's handling.\n\t// This means all unanswered buffered queries can be dispatched directly through matching at this point.\n\tif !createNewDecisionTask {\n\t\tbuffered := queryRegistry.GetBufferedIDs()\n\t\tfor _, id := range buffered {\n\t\t\tunblockTerminationState := &query.TerminationState{\n\t\t\t\tTerminationType: query.TerminationTypeUnblocked,\n\t\t\t}\n\t\t\tif err := queryRegistry.SetTerminationState(id, unblockTerminationState); err != nil {\n\t\t\t\thandler.logger.Error(\n\t\t\t\t\t\"failed to set query termination state to unblocked\",\n\t\t\t\t\ttag.WorkflowDomainName(domain),\n\t\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\t\ttag.QueryID(id),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (handler *handlerImpl) failDecisionHelper(\n\tctx context.Context,\n\twfContext execution.Context,\n\tscheduleID int64,\n\tstartedID int64,\n\tcause types.DecisionTaskFailedCause,\n\tdetails []byte,\n\trequest *types.RespondDecisionTaskCompletedRequest,\n\tdomainEntry *cache.DomainCacheEntry,\n) (execution.MutableState, error) {\n\n\t// Clear any updates we have accumulated so far\n\twfContext.Clear()\n\n\t// Reload workflow execution so we can apply the decision task failure event\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif _, err = mutableState.AddDecisionTaskFailedEvent(\n\t\tscheduleID, startedID, cause, details, request.GetIdentity(), \"\", request.GetBinaryChecksum(), \"\", \"\", 0, \"\",\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName := domainEntry.GetInfo().Name\n\tmaxAttempts := handler.config.DecisionRetryMaxAttempts(domainName)\n\tif maxAttempts > 0 && mutableState.GetExecutionInfo().DecisionAttempt > int64(maxAttempts) {\n\t\tmessage := fmt.Sprintf(\n\t\t\t\"Decision attempt exceeds limit. Last decision failure cause and details: %v - %v\",\n\t\t\tcause,\n\t\t\tdetails)\n\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\thandler.logger.Error(message,\n\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID))\n\t\thandler.metricsClient.IncCounter(metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DecisionRetriesExceededCounter)\n\n\t\tif _, err := mutableState.AddWorkflowExecutionTerminatedEvent(\n\t\t\tmutableState.GetNextEventID(),\n\t\t\tcommon.FailureReasonDecisionAttemptsExceedsLimit,\n\t\t\t[]byte(message),\n\t\t\texecution.IdentityHistoryService,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\t// Return new builder back to the caller for further updates\n\treturn mutableState, nil\n}\n\nfunc (handler *handlerImpl) getActiveDomainByWorkflow(ctx context.Context, domainID, workflowID, runID string) (*cache.DomainCacheEntry, error) {\n\tactiveClusterInfo, err := handler.activeClusterManager.GetActiveClusterInfoByWorkflow(ctx, domainID, workflowID, runID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomain, err := handler.domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif activeClusterInfo.ActiveClusterName == handler.shard.GetClusterMetadata().GetCurrentClusterName() {\n\t\treturn domain, nil\n\t}\n\treturn nil, domain.NewDomainNotActiveError(handler.shard.GetClusterMetadata().GetCurrentClusterName(), activeClusterInfo.ActiveClusterName)\n}\n\nfunc getDecisionInfoAttempt(di *execution.DecisionInfo) int64 {\n\tif di == nil {\n\t\treturn 0\n\t}\n\treturn di.Attempt\n}\n\nfunc getDecisionInfoStartedID(di *execution.DecisionInfo) int64 {\n\tif di == nil {\n\t\treturn 0\n\t}\n\treturn di.StartedID\n}\n"
  },
  {
    "path": "service/history/decision/handler_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage decision\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nconst (\n\ttestInvalidDomainUUID = \"some-invalid-UUID\"\n\ttestShardID           = 1234\n)\n\ntype (\n\tDecisionHandlerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tcontroller       *gomock.Controller\n\t\tmockMutableState *execution.MockMutableState\n\n\t\tdecisionHandler *handlerImpl\n\t\tqueryRegistry   query.Registry\n\t}\n)\n\nfunc TestDecisionHandlerSuite(t *testing.T) {\n\tsuite.Run(t, new(DecisionHandlerSuite))\n}\n\nfunc (s *DecisionHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\ts.decisionHandler = &handlerImpl{\n\t\tversionChecker: client.NewVersionChecker(),\n\t\tmetricsClient:  metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\tconfig:         config.NewForTest(),\n\t\tlogger:         testlogger.New(s.T()),\n\t\ttimeSource:     clock.NewRealTimeSource(),\n\t}\n\ts.queryRegistry = s.constructQueryRegistry(10)\n\ts.mockMutableState = execution.NewMockMutableState(s.controller)\n\tworkflowInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(workflowInfo).AnyTimes()\n}\n\nfunc (s *DecisionHandlerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *DecisionHandlerSuite) TestNewHandler() {\n\tshardContext := shard.NewMockContext(s.controller)\n\ttokenSerializer := common.NewMockTaskTokenSerializer(s.controller)\n\tshardContext.EXPECT().GetConfig().Times(1).Return(&config.Config{})\n\tshardContext.EXPECT().GetLogger().Times(2).Return(testlogger.New(s.T()))\n\tshardContext.EXPECT().GetTimeSource().Times(1)\n\tshardContext.EXPECT().GetDomainCache().Times(2)\n\tshardContext.EXPECT().GetMetricsClient().Times(2)\n\tshardContext.EXPECT().GetThrottledLogger().Times(1).Return(testlogger.New(s.T()))\n\tshardContext.EXPECT().GetActiveClusterManager().Times(1).Return(activecluster.NewMockManager(s.controller))\n\th := NewHandler(shardContext, execution.NewMockCache(s.controller), tokenSerializer)\n\ts.NotNil(h)\n\ts.Equal(\"handlerImpl\", reflect.ValueOf(h).Elem().Type().Name())\n}\n\nfunc TestHandleDecisionTaskScheduled(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tdomainID        string\n\t\tmutablestate    *persistence.WorkflowMutableState\n\t\tisfirstDecision bool\n\t\texpectCalls     func(ctrl *gomock.Controller, shardContext *shard.MockContext)\n\t\texpectErr       bool\n\t}{\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, shardContext *shard.MockContext) {\n\t\t\t\tshardContext.EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"completed workflow\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t// WorkflowStateCompleted = 2 from persistence WorkflowExecutionInfo.IsRunning()\n\t\t\t\t\tState: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, shardContext *shard.MockContext) {\n\t\t\t\tshardContext.EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"get start event failure\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t// execution has no event yet\n\t\t\t\t\tDecisionScheduleID: -23,\n\t\t\t\t\tLastProcessedEvent: -23,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, shardContext *shard.MockContext) {\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tshardContext.EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().\n\t\t\t\t\tGetEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tTimes(1).\n\t\t\t\t\tReturn(nil, &persistence.TimeoutError{Msg: \"failed to get start event: request timeout\"})\n\t\t\t\tshardContext.EXPECT().GetShardID().Return(testShardID).Times(1)\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"first decision task scheduled failure\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: -23,\n\t\t\t\t\tLastProcessedEvent: -23,\n\t\t\t\t},\n\t\t\t\tBufferedEvents: append([]*types.HistoryEvent{}, &types.HistoryEvent{}),\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, shardContext *shard.MockContext) {\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tshardContext.EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().\n\t\t\t\t\tGetEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tTimes(1).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, nil)\n\t\t\t\tshardContext.EXPECT().GetShardID().Return(testShardID).Times(1)\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(gomock.Any()).Times(1).Return([]int64{}, errors.New(\"some random error to avoid going too deep in call stack unrelated to this unit\"))\n\t\t\t},\n\t\t\texpectErr:       true,\n\t\t\tisfirstDecision: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"first decision task scheduled success\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: -23,\n\t\t\t\t\tLastProcessedEvent: -23,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, shardContext *shard.MockContext) {\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tshardContext.EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().\n\t\t\t\t\tGetEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tTimes(1).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, nil)\n\t\t\t\tshardContext.EXPECT().GetShardID().Return(testShardID).Times(1)\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(gomock.Any()).Times(1).Return([]int64{}, errors.New(\"some random error to avoid going too deep in call stack unrelated to this unit\"))\n\t\t\t},\n\t\t\texpectErr:       true,\n\t\t\tisfirstDecision: true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\trequest := &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: test.domainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t},\n\t\t\t\tIsFirstDecision: test.isfirstDecision,\n\t\t\t}\n\t\t\tdecisionHandler := &handlerImpl{\n\t\t\t\tconfig:               config.NewForTest(),\n\t\t\t\tshard:                shard.NewMockContext(ctrl),\n\t\t\t\ttimeSource:           clock.NewRealTimeSource(),\n\t\t\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t\tversionChecker:       client.NewVersionChecker(),\n\t\t\t\ttokenSerializer:      common.NewMockTaskTokenSerializer(ctrl),\n\t\t\t\tdomainCache:          cache.NewMockDomainCache(ctrl),\n\t\t\t\tactiveClusterManager: activecluster.NewMockManager(ctrl),\n\t\t\t}\n\t\t\texpectCommonCalls(decisionHandler, test.domainID)\n\t\t\texpectDefaultDomainCache(decisionHandler, test.domainID)\n\t\t\texpectGetWorkflowExecution(decisionHandler, test.domainID, test.mutablestate)\n\t\t\tif test.expectCalls != nil {\n\t\t\t\ttest.expectCalls(ctrl, decisionHandler.shard.(*shard.MockContext))\n\t\t\t}\n\t\t\tdecisionHandler.executionCache = execution.NewCache(decisionHandler.shard)\n\t\t\terr := decisionHandler.HandleDecisionTaskScheduled(context.Background(), request)\n\t\t\tassert.Equal(t, test.expectErr, err != nil)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionTaskFailed(t *testing.T) {\n\ttaskToken := []byte(\"test-token\")\n\ttests := []struct {\n\t\tname                        string\n\t\tdomainID                    string\n\t\tmutablestate                *persistence.WorkflowMutableState\n\t\texpectCalls                 func(ctrl *gomock.Controller, h *handlerImpl)\n\t\texpectErr                   bool\n\t\texpectNonDefaultDomainCache bool\n\t}{\n\t\t{\n\t\t\tname:     \"failure to deserialize token\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(taskToken).Return(nil, errors.New(\"unable to deserialize task token\"))\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t\texpectNonDefaultDomainCache: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\ttoken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\th.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(taskToken).Return(token, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(gomock.Any()).Return([]int64{0}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetShardID().Return(testShardID)\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEngine().Times(3).Return(engine)\n\t\t\t\tengine.EXPECT().NotifyNewHistoryEvent(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"completed workflow\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t// WorkflowStateCompleted = 2 from persistence WorkflowExecutionInfo.IsRunning()\n\t\t\t\t\tState: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\ttoken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\th.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(taskToken).Return(token, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"decision task not found\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: 0,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\ttoken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t}\n\t\t\t\th.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(taskToken).Return(token, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\trequest := &types.HistoryRespondDecisionTaskFailedRequest{\n\t\t\t\tDomainUUID: test.domainID,\n\t\t\t\tFailedRequest: &types.RespondDecisionTaskFailedRequest{\n\t\t\t\t\tTaskToken: taskToken,\n\t\t\t\t\tCause:     nil,\n\t\t\t\t\tDetails:   nil,\n\t\t\t\t},\n\t\t\t}\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tdecisionHandler := &handlerImpl{\n\t\t\t\tconfig:               config.NewForTest(),\n\t\t\t\tshard:                shardContext,\n\t\t\t\ttimeSource:           clock.NewRealTimeSource(),\n\t\t\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t\tversionChecker:       client.NewVersionChecker(),\n\t\t\t\ttokenSerializer:      common.NewMockTaskTokenSerializer(ctrl),\n\t\t\t\tdomainCache:          cache.NewMockDomainCache(ctrl),\n\t\t\t\tactiveClusterManager: activecluster.NewMockManager(ctrl),\n\t\t\t}\n\t\t\texpectCommonCalls(decisionHandler, test.domainID)\n\t\t\tif !test.expectNonDefaultDomainCache {\n\t\t\t\texpectDefaultDomainCache(decisionHandler, test.domainID)\n\t\t\t}\n\t\t\texpectGetWorkflowExecution(decisionHandler, test.domainID, test.mutablestate)\n\t\t\tdecisionHandler.executionCache = execution.NewCache(shardContext)\n\t\t\tif test.expectCalls != nil {\n\t\t\t\ttest.expectCalls(ctrl, decisionHandler)\n\t\t\t}\n\n\t\t\terr := decisionHandler.HandleDecisionTaskFailed(context.Background(), request)\n\t\t\tassert.Equal(t, test.expectErr, err != nil)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionTaskStarted(t *testing.T) {\n\ttestTaskListName := \"some-tasklist-name\"\n\ttestWorkflowTypeName := \"some-workflow-type-name\"\n\ttests := []struct {\n\t\tname               string\n\t\tdomainID           string\n\t\tmutablestate       *persistence.WorkflowMutableState\n\t\texpectCalls        func(ctrl *gomock.Controller, h *handlerImpl)\n\t\texpectErr          error\n\t\tassertResponseBody func(t *testing.T, response *types.RecordDecisionTaskStartedResponse)\n\t}{\n\t\t{\n\t\t\tname:     \"failure - decision task already started\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: &types.EventAlreadyStartedError{Message: \"Decision task already started.\"},\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"failure - workflow completed\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: workflow.ErrNotExists,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tState: 2, // 2 == WorkflowStateCompleted\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"failure - decision task already completed\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: &types.EntityNotExistsError{Message: \"Decision task not found.\"},\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: 1,\n\t\t\t\t\tNextEventID:        2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"failure - cached mutable state is stale\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\t// handler will attempt reloading mutable state at most 5 times\n\t\t\t\t// this test will fail all retries\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(5).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: workflow.ErrMaxAttemptsExceeded,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\texpectErr: nil,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tNextEventID:       3,\n\t\t\t\t\tDecisionRequestID: \"test-request-id\",\n\t\t\t\t\tDecisionAttempt:   1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertResponseBody: func(t *testing.T, resp *types.RecordDecisionTaskStartedResponse) {\n\t\t\t\t// expect test.mutablestate.ExecutionInfo.DecisionAttempt\n\t\t\t\tassert.Equal(t, int64(1), resp.DecisionInfo.ScheduledEvent.DecisionTaskScheduledEventAttributes.Attempt)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"success - decision startedID is empty\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texpectCalls: func(ctrl *gomock.Controller, h *handlerImpl) {\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(gomock.Any()).Times(1).Return([]int64{0}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().\n\t\t\t\t\tAppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID}).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetShardID().Return(testShardID)\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\th.shard.(*shard.MockContext).EXPECT().GetEngine().Times(3).Return(engine)\n\t\t\t\tengine.EXPECT().NotifyNewHistoryEvent(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\t\t\t},\n\t\t\texpectErr: nil,\n\t\t\tmutablestate: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionStartedID: -23,\n\t\t\t\t\tNextEventID:       2,\n\t\t\t\t\tWorkflowTypeName:  testWorkflowTypeName,\n\t\t\t\t\tTaskList:          testTaskListName,\n\t\t\t\t},\n\t\t\t},\n\t\t\tassertResponseBody: func(t *testing.T, resp *types.RecordDecisionTaskStartedResponse) {\n\t\t\t\tassert.Equal(t, testWorkflowTypeName, resp.WorkflowType.Name)\n\t\t\t\tassert.Equal(t, testTaskListName, resp.WorkflowExecutionTaskList.Name)\n\t\t\t\tassert.Equal(t, int64(0), resp.ScheduledEventID)\n\t\t\t\tassert.Equal(t, int64(3), resp.NextEventID)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\trequest := &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: test.domainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t},\n\t\t\t\tRequestID: \"test-request-id\",\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tDomain:   test.domainID,\n\t\t\t\t\tIdentity: \"test-identity\",\n\t\t\t\t},\n\t\t\t}\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tdecisionHandler := &handlerImpl{\n\t\t\t\tconfig:               config.NewForTest(),\n\t\t\t\tshard:                shardContext,\n\t\t\t\ttimeSource:           clock.NewRealTimeSource(),\n\t\t\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t\tversionChecker:       client.NewVersionChecker(),\n\t\t\t\tdomainCache:          cache.NewMockDomainCache(ctrl),\n\t\t\t\tactiveClusterManager: activecluster.NewMockManager(ctrl),\n\t\t\t}\n\t\t\texpectCommonCalls(decisionHandler, test.domainID)\n\t\t\texpectDefaultDomainCache(decisionHandler, test.domainID)\n\t\t\texpectGetWorkflowExecution(decisionHandler, test.domainID, test.mutablestate)\n\t\t\tdecisionHandler.executionCache = execution.NewCache(shardContext)\n\t\t\tif test.expectCalls != nil {\n\t\t\t\ttest.expectCalls(ctrl, decisionHandler)\n\t\t\t}\n\n\t\t\tresp, err := decisionHandler.HandleDecisionTaskStarted(context.Background(), request)\n\t\t\tassert.Equal(t, test.expectErr, err)\n\t\t\tif err == nil {\n\t\t\t\tassert.NotNil(t, resp)\n\t\t\t\tassert.Equal(t, test.mutablestate.ExecutionInfo.DecisionScheduleID, resp.ScheduledEventID)\n\t\t\t\tassert.Equal(t, test.mutablestate.ExecutionInfo.DecisionStartedID, resp.StartedEventID)\n\t\t\t\tassert.Equal(t, test.mutablestate.ExecutionInfo.NextEventID, resp.NextEventID)\n\t\t\t\tassert.Equal(t, test.mutablestate.ExecutionInfo.TaskList, resp.WorkflowExecutionTaskList.Name)\n\t\t\t\ttest.assertResponseBody(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionTaskCompleted(t *testing.T) {\n\tserializedTestToken := []byte(\"test-token\")\n\ttestTaskListName := \"some-tasklist-name\"\n\ttestWorkflowTypeName := \"some-workflow-type-name\"\n\ttests := []struct {\n\t\tname                        string\n\t\tdomainID                    string\n\t\texpectedErr                 error\n\t\texpectMockCalls             func(ctrl *gomock.Controller, decisionHandler *handlerImpl)\n\t\tassertResponseBody          func(t *testing.T, resp *types.HistoryRespondDecisionTaskCompletedResponse)\n\t\tmutableState                *persistence.WorkflowMutableState\n\t\trequest                     *types.HistoryRespondDecisionTaskCompletedRequest\n\t\texpectGetWorkflowExecution  bool\n\t\texpectNonDefaultDomainCache bool\n\t}{\n\t\t{\n\t\t\tname:        \"token deserialazation failure\",\n\t\t\tdomainID:    constants.TestDomainID,\n\t\t\texpectedErr: workflow.ErrDeserializingToken,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(nil, errors.New(\"unable to deserialize task token\"))\n\t\t\t},\n\t\t\texpectNonDefaultDomainCache: true,\n\t\t\texpectGetWorkflowExecution:  true,\n\t\t},\n\t\t{\n\t\t\tname:        \"get or create wf execution failure\",\n\t\t\tdomainID:    constants.TestDomainID,\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Can't load workflow execution.  WorkflowId not set.\"},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\ttaskToken := &common.TaskToken{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t\t// empty workflow ID to force decisionHandler.executionCache.GetOrCreateWorkflowExecution() failure\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(taskToken, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"success\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                nil,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 0,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Serialize(&common.TaskToken{\n\t\t\t\t\tDomainID:     constants.TestDomainID,\n\t\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t\t\t\tRunID:        constants.TestRunID,\n\t\t\t\t\tScheduleID:   1,\n\t\t\t\t\tActivityID:   \"some-activity-id\",\n\t\t\t\t\tActivityType: \"some-activity-name\",\n\t\t\t\t}).Return(serializedTestToken, nil)\n\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, int64(1), gomock.Any())\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(1).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(4).Return([]int64{0, 1, 2, 3}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(6).Return([]int64{0, 1, 2, 3, 4, 5}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, nil)\n\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEngine().Return(engine).Times(3)\n\t\t\t\tengine.EXPECT().NotifyNewHistoryEvent(events.NewNotification(constants.TestDomainID, &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\t0, 5, 0, 1, 0, nil))\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Times(1).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainID(constants.TestDomainName).Times(1).Return(constants.TestDomainID, nil)\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowTimeout: 600,\n\t\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\t\tPoints: func() []*types.ResetPointInfo {\n\t\t\t\t\t\t\tif historyMaxResetPoints, ok := dynamicproperties.IntKeys[dynamicproperties.HistoryMaxAutoResetPoints]; ok {\n\t\t\t\t\t\t\t\treturn make([]*types.ResetPointInfo, historyMaxResetPoints.DefaultValue)\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treturn []*types.ResetPointInfo{}\n\t\t\t\t\t\t}(),\n\t\t\t\t\t},\n\t\t\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\t\t\tTaskList:         testTaskListName,\n\t\t\t\t},\n\t\t\t\tChecksum:       checksum.Checksum{},\n\t\t\t\tBufferedEvents: append([]*types.HistoryEvent{}, &types.HistoryEvent{}),\n\t\t\t\tActivityInfos:  make(map[int64]*persistence.ActivityInfo),\n\t\t\t},\n\t\t\tassertResponseBody: func(t *testing.T, resp *types.HistoryRespondDecisionTaskCompletedResponse) {\n\t\t\t\tassert.True(t, resp.StartedResponse.StickyExecutionEnabled)\n\t\t\t\tassert.Equal(t, 1, len(resp.ActivitiesToDispatchLocally))\n\t\t\t\tassert.Equal(t, testWorkflowTypeName, resp.StartedResponse.WorkflowType.Name)\n\t\t\t\tassert.Equal(t, int64(0), resp.StartedResponse.Attempt)\n\t\t\t\tassert.Equal(t, testTaskListName, resp.StartedResponse.WorkflowExecutionTaskList.Name)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"decision task failure\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                &types.InternalServiceError{Message: \"add-decisiontask-failed-event operation failed\"},\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(2).Return(eventsCache)\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Times(1).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"workflow completed\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                workflow.ErrAlreadyCompleted,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tState: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"decision task not found\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                &types.EntityNotExistsError{Message: \"Decision task not found.\"},\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:        constants.TestDomainID,\n\t\t\t\t\tWorkflowID:      constants.TestWorkflowID,\n\t\t\t\t\tRunID:           constants.TestRunID,\n\t\t\t\t\tScheduleAttempt: 1,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"decision heartbeat time out\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                &types.EntityNotExistsError{Message: \"decision heartbeat timeout\"},\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken:                  serializedTestToken,\n\t\t\t\t\tDecisions:                  []*types.Decision{},\n\t\t\t\t\tReturnNewDecisionTask:      true,\n\t\t\t\t\tForceCreateNewDecisionTask: true,\n\t\t\t\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\t\t\t\tWorkerTaskList:                &types.TaskList{Name: testTaskListName},\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: func(i int32) *int32 { return &i }(10),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 0,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(eventsCache)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(1).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(1).Return([]int64{0}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, nil)\n\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEngine().Return(engine).Times(3)\n\t\t\t\tengine.EXPECT().NotifyNewHistoryEvent(events.NewNotification(constants.TestDomainID, &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\t0, 1, 0, 1, 0, nil))\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionOriginalScheduledTimestamp: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"update continueAsNew info failure - execution size limit exceeded\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                execution.ErrWorkflowFinished,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeContinueAsNewWorkflowExecution),\n\t\t\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: testWorkflowTypeName},\n\t\t\t\t\t\t\tTaskList:     &types.TaskList{Name: testTaskListName},\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t\tReturnNewDecisionTask: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 0,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(3).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().GetEvent(context.Background(), testShardID, constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, commonconstants.FirstEventID, commonconstants.FirstEventID, nil).Return(&types.HistoryEvent{}, nil)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, gomock.Any(), int64(1), gomock.Any()).Times(2)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(1).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(2).Times(1).Return([]int64{0, 1}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(nil, &persistence.TransactionSizeLimitError{Msg: fmt.Sprintf(\"transaction size exceeds limit\")})\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetExecutionManager().Times(1)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil)\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionOriginalScheduledTimestamp: 1,\n\t\t\t\t\tWorkflowTimeout:                    100,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"update continueAsNew info failure - conflict error\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                workflow.ErrAlreadyCompleted,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeCompleteWorkflowExecution),\n\t\t\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte{}},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(3).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().GetEvent(context.Background(), testShardID, constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, commonconstants.FirstEventID, commonconstants.FirstEventID, nil).\n\t\t\t\t\tReturn(&types.HistoryEvent{\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t}, nil).Times(3)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, gomock.Any(), int64(1), gomock.Any()).Times(2)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(3).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(2).Times(1).Return([]int64{0, 1}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(nil, execution.NewConflictError(new(testing.T), errors.New(\"some random conflict error\")))\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetExecutionManager().Times(1)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil)\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCronSchedule: \"0 1 * * 1\", // some random cron schedule\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"update continueAsNew info failure - load execution\",\n\t\t\tdomainID:    constants.TestDomainID,\n\t\t\texpectedErr: errors.New(\"some error occurred when loading workflow execution\"),\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeFailWorkflowExecution),\n\t\t\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tReason: func(reason string) *string { return &reason }(\"some reason to fail workflow execution\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(2).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().GetEvent(context.Background(), testShardID, constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, commonconstants.FirstEventID, commonconstants.FirstEventID, nil).\n\t\t\t\t\tReturn(&types.HistoryEvent{\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t}, nil).Times(3)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, gomock.Any(), int64(1), gomock.Any()).Times(2)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(3).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(2).Times(1).Return([]int64{0, 1}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(nil, &persistence.TransactionSizeLimitError{Msg: fmt.Sprintf(\"transaction size exceeds limit\")})\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetExecutionManager().Times(1)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil)\n\t\t\t\tfirstGetWfExecutionCall := decisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\t\tDomainID:     constants.TestDomainID,\n\t\t\t\t\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\t\t\t\t\tRunID:        constants.TestRunID,\n\t\t\t\t\t\t\t\tCronSchedule: \"0 1 * * 1\", // some random cron schedule\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t\t}, nil)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t\tlastGetWfExecutionCall := decisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error occurred when loading workflow execution\"))\n\t\t\t\tgomock.InOrder(firstGetWfExecutionCall, lastGetWfExecutionCall)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"update continueAsNew info failure - update execution\",\n\t\t\tdomainID:    constants.TestDomainID,\n\t\t\texpectedErr: errors.New(\"some error updating workflow execution\"),\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeFailWorkflowExecution),\n\t\t\t\t\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\t\tReason: func(reason string) *string { return &reason }(\"some reason to fail workflow execution\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(3).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().GetEvent(context.Background(), testShardID, constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, commonconstants.FirstEventID, commonconstants.FirstEventID, nil).\n\t\t\t\t\tReturn(&types.HistoryEvent{\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t}, nil).Times(3)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, gomock.Any(), int64(1), gomock.Any()).Times(3)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(3).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(2).Times(2).Return([]int64{0, 1}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(1).Times(1).Return([]int64{0}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(nil, &persistence.TransactionSizeLimitError{Msg: fmt.Sprintf(\"transaction size exceeds limit\")})\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetExecutionManager().Times(1)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx interface{}, request interface{}) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\t\t\treturn &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\t\t\tDomainID:     constants.TestDomainID,\n\t\t\t\t\t\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\t\t\t\t\t\tRunID:        constants.TestRunID,\n\t\t\t\t\t\t\t\t\tCronSchedule: \"0 1 * * 1\", // some random cron schedule\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(2)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error updating workflow execution\"))\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEngine().Return(engine).Times(2)\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                        \"bad binaries\",\n\t\t\tdomainID:                    constants.TestDomainID,\n\t\t\texpectedErr:                 errors.New(\"some error updating continue as new info\"),\n\t\t\texpectNonDefaultDomainCache: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName()}, nil)\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\teventsCache := events.NewMockCache(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(3).Return(eventsCache)\n\t\t\t\teventsCache.EXPECT().PutEvent(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID, int64(0), gomock.Any())\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(1).Return([]int64{0}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, gomock.Any()).Return(nil, errors.New(\"some error updating continue as new info\"))\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName},\n\t\t\t\t\t&persistence.DomainConfig{\n\t\t\t\t\t\tRetention:   1,\n\t\t\t\t\t\tBadBinaries: types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"test-binary-checksum\": {Reason: \"some reason\"}}},\n\t\t\t\t\t},\n\t\t\t\t\tcluster.TestCurrentClusterName)\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).AnyTimes().Return(domainEntry, nil)\n\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\tExecutionInfo:  &persistence.WorkflowExecutionInfo{DomainID: constants.TestDomainID, WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID, DecisionScheduleID: 1},\n\t\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\tExecutionInfo:  &persistence.WorkflowExecutionInfo{DomainID: constants.TestDomainID, WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\t\tExecutionInfo:  &persistence.WorkflowExecutionInfo{DomainID: constants.TestDomainID, WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID, DecisionAttempt: 2},\n\t\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t},\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken:                  serializedTestToken,\n\t\t\t\t\tBinaryChecksum:             \"test-binary-checksum\",\n\t\t\t\t\tForceCreateNewDecisionTask: true,\n\t\t\t\t\tReturnNewDecisionTask:      true,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"failure to load workflow execution - shard closed\",\n\t\t\tdomainID:    constants.TestDomainID,\n\t\t\texpectedErr: errors.New(\"some random error\"),\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Times(1).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                        \"failure to load workflow execution stats\",\n\t\t\tdomainID:                    constants.TestDomainID,\n\t\t\texpectedErr:                 errors.New(\"some random error\"),\n\t\t\texpectNonDefaultDomainCache: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdecisionHandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName()}, nil)\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).Times(2).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t\tdecisionHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).Times(1).Return(nil, errors.New(\"some random error\"))\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t\texpectGetWorkflowExecution(decisionHandler, constants.TestDomainID, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"task handler fail decisions\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                &types.InternalServiceError{Message: \"unable to change workflow state from 0 to 2, close status 3\"},\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 0,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeCancelWorkflowExecution),\n\t\t\t\t\t\tCancelWorkflowExecutionDecisionAttributes: &types.CancelWorkflowExecutionDecisionAttributes{Details: []byte{}},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"Update_History_Loop max attempt exceeded\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                workflow.ErrMaxAttemptsExceeded,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tScheduleID: 2,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(5).Return(events.NewMockCache(ctrl))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                       \"success with decision heartbeat\",\n\t\t\tdomainID:                   constants.TestDomainID,\n\t\t\texpectedErr:                nil,\n\t\t\texpectGetWorkflowExecution: true,\n\t\t\trequest: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken:                  serializedTestToken,\n\t\t\t\t\tDecisions:                  []*types.Decision{},\n\t\t\t\t\tReturnNewDecisionTask:      true,\n\t\t\t\t\tForceCreateNewDecisionTask: true,\n\t\t\t\t\tStickyAttributes: &types.StickyExecutionAttributes{\n\t\t\t\t\t\tWorkerTaskList: &types.TaskList{Name: testTaskListName},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectMockCalls: func(ctrl *gomock.Controller, decisionHandler *handlerImpl) {\n\t\t\t\tdeserializedTestToken := &common.TaskToken{\n\t\t\t\t\tDomainID:     constants.TestDomainID,\n\t\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\t\tRunID:        constants.TestRunID,\n\t\t\t\t\tWorkflowType: testWorkflowTypeName,\n\t\t\t\t}\n\t\t\t\tdecisionHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Deserialize(serializedTestToken).Return(deserializedTestToken, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEventsCache().Times(1).Return(events.NewMockCache(ctrl))\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetShardID().Times(1).Return(testShardID)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GenerateTaskIDs(3).Return([]int64{0, 1, 2}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), constants.TestDomainID, types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{}, nil)\n\t\t\t\tengine := engine.NewMockEngine(ctrl)\n\t\t\t\tdecisionHandler.shard.(*shard.MockContext).EXPECT().GetEngine().Return(engine).Times(3)\n\t\t\t\tengine.EXPECT().NotifyNewHistoryEvent(events.NewNotification(constants.TestDomainID, &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\t0, 3, 0, 1, 0, nil))\n\t\t\t\tengine.EXPECT().NotifyNewTransferTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewTimerTasks(gomock.Any())\n\t\t\t\tengine.EXPECT().NotifyNewReplicationTasks(gomock.Any())\n\t\t\t},\n\t\t\tassertResponseBody: func(t *testing.T, resp *types.HistoryRespondDecisionTaskCompletedResponse) {\n\t\t\t\tassert.True(t, resp.StartedResponse.StickyExecutionEnabled)\n\t\t\t\tassert.Equal(t, testWorkflowTypeName, resp.StartedResponse.WorkflowType.Name)\n\t\t\t\tassert.Equal(t, int64(0), resp.StartedResponse.Attempt)\n\t\t\t\tassert.Equal(t, testTaskListName, resp.StartedResponse.WorkflowExecutionTaskList.Name)\n\t\t\t},\n\t\t\tmutableState: &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowTypeName: testWorkflowTypeName,\n\t\t\t\t\tTaskList:         testTaskListName,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tshard := shard.NewMockContext(ctrl)\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\thandlerConfig := config.NewForTest()\n\t\t\thandlerConfig.MaxActivityCountDispatchByDomain = func(domain string) int { return 1 } // some value > 0\n\t\t\thandlerConfig.EnableActivityLocalDispatchByDomain = func(domain string) bool { return true }\n\t\t\thandlerConfig.DecisionRetryMaxAttempts = func(domain string) int { return 1 }\n\t\t\tdecisionHandler := &handlerImpl{\n\t\t\t\tconfig:               handlerConfig,\n\t\t\t\tshard:                shard,\n\t\t\t\ttimeSource:           clock.NewMockedTimeSource(),\n\t\t\t\tdomainCache:          domainCache,\n\t\t\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t\tversionChecker:       client.NewVersionChecker(),\n\t\t\t\ttokenSerializer:      common.NewMockTaskTokenSerializer(ctrl),\n\t\t\t\tattrValidator:        newAttrValidator(domainCache, metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), config.NewForTest(), testlogger.New(t)),\n\t\t\t\tactiveClusterManager: activecluster.NewMockManager(ctrl),\n\t\t\t}\n\t\t\texpectCommonCalls(decisionHandler, test.domainID)\n\t\t\tif !test.expectNonDefaultDomainCache {\n\t\t\t\texpectDefaultDomainCache(decisionHandler, test.domainID)\n\t\t\t}\n\t\t\tif test.expectGetWorkflowExecution {\n\t\t\t\texpectGetWorkflowExecution(decisionHandler, test.domainID, test.mutableState)\n\t\t\t}\n\t\t\tdecisionHandler.executionCache = execution.NewCache(shard)\n\n\t\t\trequest := &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: test.domainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: serializedTestToken,\n\t\t\t\t\tDecisions: []*types.Decision{{\n\t\t\t\t\t\tDecisionType: nil,\n\t\t\t\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\t\tActivityID:                    \"some-activity-id\",\n\t\t\t\t\t\t\tActivityType:                  &types.ActivityType{Name: \"some-activity-name\"},\n\t\t\t\t\t\t\tDomain:                        constants.TestDomainName,\n\t\t\t\t\t\t\tTaskList:                      &types.TaskList{Name: testTaskListName},\n\t\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: func(i int32) *int32 { return &i }(200),\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: func(i int32) *int32 { return &i }(100),\n\t\t\t\t\t\t\tStartToCloseTimeoutSeconds:    func(i int32) *int32 { return &i }(100),\n\t\t\t\t\t\t\tRequestLocalDispatch:          true,\n\t\t\t\t\t\t},\n\t\t\t\t\t}},\n\t\t\t\t\tReturnNewDecisionTask: true,\n\t\t\t\t},\n\t\t\t}\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(ctrl, decisionHandler)\n\t\t\t}\n\t\t\tif test.request != nil {\n\t\t\t\trequest = test.request\n\t\t\t}\n\t\t\tresp, err := decisionHandler.HandleDecisionTaskCompleted(context.Background(), request)\n\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t\tif err != nil {\n\t\t\t\tassert.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\ttest.assertResponseBody(t, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *DecisionHandlerSuite) TestCreateRecordDecisionTaskStartedResponse() {\n\ttests := []struct {\n\t\tname        string\n\t\texpectCalls func()\n\t\texpectedErr error\n\t\tindexes     []string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texpectCalls: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetWorkflowType().Return(&types.WorkflowType{})\n\t\t\t\ts.mockMutableState.EXPECT().GetNextEventID().Return(int64(1))\n\t\t\t\ts.mockMutableState.EXPECT().CreateTransientDecisionEvents(gomock.Any(), \"test-identity\").Return(&types.HistoryEvent{}, &types.HistoryEvent{})\n\t\t\t\ts.mockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte{}, nil)\n\t\t\t\tregistry := query.NewMockRegistry(s.controller)\n\t\t\t\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(registry)\n\t\t\t\tregistry.EXPECT().GetBufferedIDs().Return([]string{\"test-id\", \"test-id1\", \"test-id2\"})\n\t\t\t\tregistry.EXPECT().GetQueryInput(gomock.Any()).Return(&types.WorkflowQuery{}, nil).Times(2)\n\t\t\t\tregistry.EXPECT().GetQueryInput(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"query does not exist\"})\n\t\t\t\ts.mockMutableState.EXPECT().GetHistorySize()\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t\tindexes:     []string{\"test-id\", \"test-id1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"failure\",\n\t\t\texpectCalls: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetWorkflowType().Return(&types.WorkflowType{})\n\t\t\t\ts.mockMutableState.EXPECT().GetNextEventID().Return(int64(1))\n\t\t\t\ts.mockMutableState.EXPECT().CreateTransientDecisionEvents(gomock.Any(), \"test-identity\").Return(&types.HistoryEvent{}, &types.HistoryEvent{})\n\t\t\t\ts.mockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte{}, &types.BadRequestError{Message: fmt.Sprintf(\"getting branch index: %d, available branch count: %d\", 0, 0)})\n\t\t\t},\n\t\t\texpectedErr: &types.BadRequestError{Message: fmt.Sprintf(\"getting branch index: %d, available branch count: %d\", 0, 0)},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\ts.Run(test.name, func() {\n\t\t\tdecision := &execution.DecisionInfo{\n\t\t\t\tScheduleID: 1,\n\t\t\t\tStartedID:  2,\n\t\t\t\tRequestID:  constants.TestRequestID,\n\t\t\t\tTaskList:   \"test-tasklist\",\n\t\t\t\tAttempt:    1,\n\t\t\t}\n\t\t\ttest.expectCalls()\n\t\t\tresp, err := s.decisionHandler.createRecordDecisionTaskStartedResponse(constants.TestDomainID, s.mockMutableState, decision, \"test-identity\")\n\t\t\ts.Equal(test.expectedErr, err)\n\t\t\tif err != nil {\n\t\t\t\ts.Nil(resp)\n\t\t\t} else {\n\t\t\t\ts.Equal(&types.HistoryEvent{}, resp.DecisionInfo.ScheduledEvent)\n\t\t\t\ts.Equal(&types.HistoryEvent{}, resp.DecisionInfo.StartedEvent)\n\t\t\t\ts.Equal([]byte{}, resp.BranchToken)\n\t\t\t\tfor _, index := range test.indexes {\n\t\t\t\t\t_, ok := resp.Queries[index]\n\t\t\t\t\ts.True(ok)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_ClientNotSupports() {\n\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(s.queryRegistry)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, \"0.0.0\", nil, false, constants.TestGlobalDomainEntry, false)\n\ts.assertQueryCounts(s.queryRegistry, 0, 0, 0, 10)\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_HeartbeatDecision() {\n\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(s.queryRegistry)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n\tqueryResults := s.constructQueryResults(s.queryRegistry.GetBufferedIDs()[0:5], 10)\n\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, client.GoWorkerConsistentQueryVersion, queryResults, false, constants.TestGlobalDomainEntry, true)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_NewDecisionTask() {\n\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(s.queryRegistry)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n\tqueryResults := s.constructQueryResults(s.queryRegistry.GetBufferedIDs()[0:5], 10)\n\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, client.GoWorkerConsistentQueryVersion, queryResults, true, constants.TestGlobalDomainEntry, false)\n\ts.assertQueryCounts(s.queryRegistry, 5, 5, 0, 0)\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_NoNewDecisionTask() {\n\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(s.queryRegistry)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n\tqueryResults := s.constructQueryResults(s.queryRegistry.GetBufferedIDs()[0:5], 10)\n\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, client.GoWorkerConsistentQueryVersion, queryResults, false, constants.TestGlobalDomainEntry, false)\n\ts.assertQueryCounts(s.queryRegistry, 0, 5, 5, 0)\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_QueryTooLarge() {\n\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(s.queryRegistry)\n\ts.assertQueryCounts(s.queryRegistry, 10, 0, 0, 0)\n\tbufferedIDs := s.queryRegistry.GetBufferedIDs()\n\tqueryResults := s.constructQueryResults(bufferedIDs[0:5], 10)\n\tlargeQueryResults := s.constructQueryResults(bufferedIDs[5:10], 10*1024*1024)\n\tfor k, v := range largeQueryResults {\n\t\tqueryResults[k] = v\n\t}\n\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, client.GoWorkerConsistentQueryVersion, queryResults, false, constants.TestGlobalDomainEntry, false)\n\ts.assertQueryCounts(s.queryRegistry, 0, 5, 0, 5)\n}\n\nfunc (s *DecisionHandlerSuite) TestHandleBufferedQueries_QueryRegistryFailures() {\n\ttests := []struct {\n\t\tname                 string\n\t\texpectMockCalls      func()\n\t\tassertCalls          func(logs *observer.ObservedLogs)\n\t\tclientFeatureVersion string\n\t\tqueryResults         map[string]*types.WorkflowQueryResult\n\t}{\n\t\t{\n\t\t\tname: \"no buffered queries\",\n\t\t\texpectMockCalls: func() {\n\t\t\t\tqueryRegistry := query.NewMockRegistry(s.controller)\n\t\t\t\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(queryRegistry)\n\t\t\t\tqueryRegistry.EXPECT().HasBufferedQuery().Return(false)\n\t\t\t},\n\t\t\tassertCalls: func(logs *observer.ObservedLogs) {},\n\t\t},\n\t\t{\n\t\t\tname: \"set query termination state failed - client unsupported\",\n\t\t\texpectMockCalls: func() {\n\t\t\t\tqueryRegistry := query.NewMockRegistry(s.controller)\n\t\t\t\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(queryRegistry)\n\t\t\t\tqueryRegistry.EXPECT().HasBufferedQuery().Return(true)\n\t\t\t\tqueryRegistry.EXPECT().GetBufferedIDs().Return([]string{\"some-buffered-id\"})\n\t\t\t\tqueryRegistry.EXPECT().SetTerminationState(\"some-buffered-id\", gomock.Any()).Return(&types.InternalServiceError{Message: \"query does not exist\"})\n\t\t\t},\n\t\t\tassertCalls: func(logs *observer.ObservedLogs) {\n\t\t\t\ts.Equal(1, logs.FilterMessage(\"failed to set query termination state to failed\").Len())\n\t\t\t},\n\t\t\tclientFeatureVersion: \"0.0.0\",\n\t\t},\n\t\t{\n\t\t\tname: \"set query termination state failed - query too large\",\n\t\t\texpectMockCalls: func() {\n\t\t\t\tqueryRegistry := query.NewMockRegistry(s.controller)\n\t\t\t\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(queryRegistry)\n\t\t\t\tqueryRegistry.EXPECT().HasBufferedQuery().Return(true)\n\t\t\t\tqueryRegistry.EXPECT().GetBufferedIDs().Return([]string{\"some-id\"})\n\t\t\t\tqueryRegistry.EXPECT().SetTerminationState(\"some-id\", gomock.Any()).Return(&types.InternalServiceError{Message: \"query already in terminal state\"}).Times(2)\n\t\t\t},\n\t\t\tqueryResults:         s.constructQueryResults([]string{\"some-id\"}, 10*1024*1024),\n\t\t\tclientFeatureVersion: client.GoWorkerConsistentQueryVersion,\n\t\t\tassertCalls: func(logs *observer.ObservedLogs) {\n\t\t\t\ts.Equal(1, logs.FilterMessage(\"failed to set query termination state to failed\").Len())\n\t\t\t\ts.Equal(1, logs.FilterMessage(\"failed to set query termination state to unblocked\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"set query termination state unblocked\",\n\t\t\texpectMockCalls: func() {\n\t\t\t\tqueryRegistry := query.NewMockRegistry(s.controller)\n\t\t\t\ts.mockMutableState.EXPECT().GetQueryRegistry().Return(queryRegistry)\n\t\t\t\tqueryRegistry.EXPECT().HasBufferedQuery().Return(true)\n\t\t\t\tqueryRegistry.EXPECT().GetBufferedIDs().Return([]string{\"some-buffered-id\"})\n\t\t\t\tqueryRegistry.EXPECT().SetTerminationState(\"some-id\", gomock.Any()).Return(&types.InternalServiceError{Message: \"query does not exist\"})\n\t\t\t\tqueryRegistry.EXPECT().SetTerminationState(\"some-buffered-id\", gomock.Any()).Return(&types.InternalServiceError{Message: \"query already in terminal state\"})\n\t\t\t},\n\t\t\tclientFeatureVersion: client.GoWorkerConsistentQueryVersion,\n\t\t\tqueryResults:         map[string]*types.WorkflowQueryResult{\"some-id\": &types.WorkflowQueryResult{}},\n\t\t\tassertCalls: func(logs *observer.ObservedLogs) {\n\t\t\t\ts.Equal(1, logs.FilterMessage(\"failed to set query termination state to completed\").Len())\n\t\t\t\ts.Equal(1, logs.FilterMessage(\"failed to set query termination state to unblocked\").Len())\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\ts.Run(test.name, func() {\n\t\t\tcore, observedLogs := observer.New(zap.ErrorLevel)\n\t\t\tlogger := zap.New(core)\n\t\t\ts.decisionHandler.logger = log.NewLogger(logger, log.WithSampleFunc(func(int) bool { return true }))\n\n\t\t\ttest.expectMockCalls()\n\t\t\ts.decisionHandler.handleBufferedQueries(s.mockMutableState, client.GoSDK, test.clientFeatureVersion, test.queryResults, false, constants.TestGlobalDomainEntry, false)\n\t\t\ttest.assertCalls(observedLogs)\n\t\t})\n\t}\n}\n\nfunc (s *DecisionHandlerSuite) constructQueryResults(ids []string, resultSize int) map[string]*types.WorkflowQueryResult {\n\tresults := make(map[string]*types.WorkflowQueryResult)\n\tfor _, id := range ids {\n\t\tresults[id] = &types.WorkflowQueryResult{\n\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\tAnswer:     make([]byte, resultSize),\n\t\t}\n\t}\n\treturn results\n}\n\nfunc (s *DecisionHandlerSuite) constructQueryRegistry(numQueries int) query.Registry {\n\tqueryRegistry := query.NewRegistry()\n\tfor i := 0; i < numQueries; i++ {\n\t\tqueryRegistry.BufferQuery(&types.WorkflowQuery{})\n\t}\n\treturn queryRegistry\n}\n\nfunc (s *DecisionHandlerSuite) assertQueryCounts(queryRegistry query.Registry, buffered, completed, unblocked, failed int) {\n\ts.Len(queryRegistry.GetBufferedIDs(), buffered)\n\ts.Len(queryRegistry.GetCompletedIDs(), completed)\n\ts.Len(queryRegistry.GetUnblockedIDs(), unblocked)\n\ts.Len(queryRegistry.GetFailedIDs(), failed)\n}\n\nfunc expectCommonCalls(handler *handlerImpl, domainID string) {\n\thandler.shard.(*shard.MockContext).EXPECT().GetConfig().AnyTimes().Return(handler.config)\n\thandler.shard.(*shard.MockContext).EXPECT().GetLogger().AnyTimes().Return(handler.logger)\n\thandler.shard.(*shard.MockContext).EXPECT().GetTimeSource().AnyTimes().Return(handler.timeSource)\n\thandler.shard.(*shard.MockContext).EXPECT().GetDomainCache().AnyTimes().Return(handler.domainCache)\n\thandler.shard.(*shard.MockContext).EXPECT().GetClusterMetadata().AnyTimes().Return(constants.TestClusterMetadata)\n\thandler.shard.(*shard.MockContext).EXPECT().GetMetricsClient().AnyTimes().Return(handler.metricsClient)\n\thandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(domainID).AnyTimes().Return(constants.TestDomainName, nil)\n\thandler.shard.(*shard.MockContext).EXPECT().GetExecutionManager().Times(1)\n\thandler.shard.(*shard.MockContext).EXPECT().GetShardID().Return(testShardID).Times(1)\n\thandler.shard.(*shard.MockContext).EXPECT().GetActiveClusterManager().Return(handler.activeClusterManager).AnyTimes()\n}\n\nfunc expectGetWorkflowExecution(handler *handlerImpl, domainID string, state *persistence.WorkflowMutableState) {\n\tworkflowExecutionResponse := &persistence.GetWorkflowExecutionResponse{\n\t\tState:             state,\n\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t}\n\tif state == nil {\n\t\tworkflowExecutionResponse.State = &persistence.WorkflowMutableState{ExecutionInfo: &persistence.WorkflowExecutionInfo{}}\n\t}\n\tworkflowExecutionResponse.State.ExecutionStats = &persistence.ExecutionStats{}\n\tworkflowExecutionResponse.State.ExecutionInfo.DomainID = domainID\n\tworkflowExecutionResponse.State.ExecutionInfo.WorkflowID = constants.TestWorkflowID\n\tworkflowExecutionResponse.State.ExecutionInfo.RunID = constants.TestRunID\n\n\thandler.shard.(*shard.MockContext).EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tDomainName: constants.TestDomainName,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\tRunID:      constants.TestRunID,\n\t\t},\n\t}).AnyTimes().Return(workflowExecutionResponse, nil)\n\n\thandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n}\n\nfunc expectDefaultDomainCache(handler *handlerImpl, domainID string) {\n\thandler.activeClusterManager.(*activecluster.MockManager).EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName()}, nil)\n\thandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainByID(domainID).AnyTimes().Return(constants.TestLocalDomainEntry, nil)\n}\n"
  },
  {
    "path": "service/history/decision/task_handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage decision\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nconst (\n\tactivityCancellationMsgActivityIDUnknown  = \"ACTIVITY_ID_UNKNOWN\"\n\tactivityCancellationMsgActivityNotStarted = \"ACTIVITY_ID_NOT_STARTED\"\n)\n\ntype (\n\tattrValidationFn func() error\n\n\ttaskHandlerImpl struct {\n\t\tidentity                string\n\t\tdecisionTaskCompletedID int64\n\t\tdomainEntry             *cache.DomainCacheEntry\n\n\t\t// internal state\n\t\thasUnhandledEventsBeforeDecisions bool\n\t\tfailDecision                      bool\n\t\tfailDecisionCause                 *types.DecisionTaskFailedCause\n\t\tfailMessage                       *string\n\t\tactivityNotStartedCancelled       bool\n\t\tcontinueAsNewBuilder              execution.MutableState\n\t\tstopProcessing                    bool // should stop processing any more decisions\n\t\tmutableState                      execution.MutableState\n\n\t\t// validation\n\t\tattrValidator    *attrValidator\n\t\tsizeLimitChecker *workflowSizeChecker\n\n\t\ttokenSerializer common.TaskTokenSerializer\n\n\t\tlogger        log.Logger\n\t\tdomainCache   cache.DomainCache\n\t\tmetricsClient metrics.Client\n\t\tconfig        *config.Config\n\n\t\tactivityCountToDispatch int\n\t}\n\n\tdecisionResult struct {\n\t\tactivityDispatchInfo *types.ActivityLocalDispatchInfo\n\t}\n)\n\nfunc newDecisionTaskHandler(\n\tidentity string,\n\tdecisionTaskCompletedID int64,\n\tdomainEntry *cache.DomainCacheEntry,\n\tmutableState execution.MutableState,\n\tattrValidator *attrValidator,\n\tsizeLimitChecker *workflowSizeChecker,\n\ttokenSerializer common.TaskTokenSerializer,\n\tlogger log.Logger,\n\tdomainCache cache.DomainCache,\n\tmetricsClient metrics.Client,\n\tconfig *config.Config,\n) *taskHandlerImpl {\n\n\treturn &taskHandlerImpl{\n\t\tidentity:                identity,\n\t\tdecisionTaskCompletedID: decisionTaskCompletedID,\n\t\tdomainEntry:             domainEntry,\n\n\t\t// internal state\n\t\thasUnhandledEventsBeforeDecisions: mutableState.HasBufferedEvents(),\n\t\tfailDecision:                      false,\n\t\tfailDecisionCause:                 nil,\n\t\tfailMessage:                       nil,\n\t\tactivityNotStartedCancelled:       false,\n\t\tcontinueAsNewBuilder:              nil,\n\t\tstopProcessing:                    false,\n\t\tmutableState:                      mutableState,\n\n\t\t// validation\n\t\tattrValidator:    attrValidator,\n\t\tsizeLimitChecker: sizeLimitChecker,\n\n\t\ttokenSerializer: tokenSerializer,\n\n\t\tlogger:        logger,\n\t\tdomainCache:   domainCache,\n\t\tmetricsClient: metricsClient,\n\t\tconfig:        config,\n\n\t\tactivityCountToDispatch: config.MaxActivityCountDispatchByDomain(domainEntry.GetInfo().Name),\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecisions(\n\tctx context.Context,\n\texecutionContext []byte,\n\tdecisions []*types.Decision,\n) ([]*decisionResult, error) {\n\n\t// overall workflow size / count check\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowSizeExceedsLimit()\n\tif err != nil || failWorkflow {\n\t\treturn nil, err\n\t}\n\n\tvar results []*decisionResult\n\tfor _, decision := range decisions {\n\t\tresult, err := handler.handleDecisionWithResult(ctx, decision)\n\t\tif err != nil || handler.stopProcessing {\n\t\t\treturn nil, err\n\t\t} else if result != nil {\n\t\t\tresults = append(results, result)\n\t\t}\n\n\t}\n\thandler.mutableState.GetExecutionInfo().ExecutionContext = executionContext\n\treturn results, nil\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionWithResult(\n\tctx context.Context,\n\tdecision *types.Decision,\n) (*decisionResult, error) {\n\tswitch decision.GetDecisionType() {\n\tcase types.DecisionTypeScheduleActivityTask:\n\t\treturn handler.handleDecisionScheduleActivity(ctx, decision.ScheduleActivityTaskDecisionAttributes)\n\tdefault:\n\t\treturn nil, handler.handleDecision(ctx, decision)\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecision(\n\tctx context.Context,\n\tdecision *types.Decision,\n) error {\n\tswitch decision.GetDecisionType() {\n\n\tcase types.DecisionTypeCompleteWorkflowExecution:\n\t\treturn handler.handleDecisionCompleteWorkflow(ctx, decision.CompleteWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeFailWorkflowExecution:\n\t\treturn handler.handleDecisionFailWorkflow(ctx, decision.FailWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeCancelWorkflowExecution:\n\t\treturn handler.handleDecisionCancelWorkflow(ctx, decision.CancelWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeStartTimer:\n\t\treturn handler.handleDecisionStartTimer(ctx, decision.StartTimerDecisionAttributes)\n\n\tcase types.DecisionTypeRequestCancelActivityTask:\n\t\treturn handler.handleDecisionRequestCancelActivity(ctx, decision.RequestCancelActivityTaskDecisionAttributes)\n\n\tcase types.DecisionTypeCancelTimer:\n\t\treturn handler.handleDecisionCancelTimer(ctx, decision.CancelTimerDecisionAttributes)\n\n\tcase types.DecisionTypeRecordMarker:\n\t\treturn handler.handleDecisionRecordMarker(ctx, decision.RecordMarkerDecisionAttributes)\n\n\tcase types.DecisionTypeRequestCancelExternalWorkflowExecution:\n\t\treturn handler.handleDecisionRequestCancelExternalWorkflow(ctx, decision.RequestCancelExternalWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeSignalExternalWorkflowExecution:\n\t\treturn handler.handleDecisionSignalExternalWorkflow(ctx, decision.SignalExternalWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeContinueAsNewWorkflowExecution:\n\t\treturn handler.handleDecisionContinueAsNewWorkflow(ctx, decision.ContinueAsNewWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeStartChildWorkflowExecution:\n\t\treturn handler.handleDecisionStartChildWorkflow(ctx, decision.StartChildWorkflowExecutionDecisionAttributes)\n\n\tcase types.DecisionTypeUpsertWorkflowSearchAttributes:\n\t\treturn handler.handleDecisionUpsertWorkflowSearchAttributes(ctx, decision.UpsertWorkflowSearchAttributesDecisionAttributes)\n\n\tdefault:\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"Unknown decision type: %v\", decision.GetDecisionType())}\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionScheduleActivity(\n\tctx context.Context,\n\tattr *types.ScheduleActivityTaskDecisionAttributes,\n) (*decisionResult, error) {\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeScheduleActivityCounter,\n\t)\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\ttargetDomainID := domainID\n\tif attr.GetDomain() != \"\" {\n\t\ttargetDomainEntry, err := handler.domainCache.GetDomain(attr.GetDomain())\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Unable to schedule activity across domain %v.\", attr.GetDomain()),\n\t\t\t}\n\t\t}\n\t\ttargetDomainID = targetDomainEntry.GetInfo().ID\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateActivityScheduleAttributes(\n\t\t\t\tdomainID,\n\t\t\t\ttargetDomainID,\n\t\t\t\tattr,\n\t\t\t\texecutionInfo,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadScheduleActivityAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn nil, err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeScheduleActivityTask.String()),\n\t\tattr.Input,\n\t\t\"ScheduleActivityTaskDecisionAttributes.Input exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn nil, err\n\t}\n\n\tevent, ai, activityDispatchInfo, dispatched, started, err := handler.mutableState.AddActivityTaskScheduledEvent(\n\t\tctx, handler.decisionTaskCompletedID, attr, handler.activityCountToDispatch > 0)\n\tif dispatched {\n\t\thandler.activityCountToDispatch--\n\t}\n\tswitch err.(type) {\n\tcase nil:\n\t\tif activityDispatchInfo != nil || started {\n\t\t\tif _, err1 := handler.mutableState.AddActivityTaskStartedEvent(ai, event.ID, uuid.New(), handler.identity); err1 != nil {\n\t\t\t\treturn nil, err1\n\t\t\t}\n\t\t\tif started {\n\t\t\t\treturn nil, nil\n\t\t\t}\n\t\t\ttoken := &common.TaskToken{\n\t\t\t\tDomainID:        executionInfo.DomainID,\n\t\t\t\tWorkflowID:      executionInfo.WorkflowID,\n\t\t\t\tWorkflowType:    executionInfo.WorkflowTypeName,\n\t\t\t\tRunID:           executionInfo.RunID,\n\t\t\t\tScheduleID:      ai.ScheduleID,\n\t\t\t\tScheduleAttempt: 0,\n\t\t\t\tActivityID:      ai.ActivityID,\n\t\t\t\tActivityType:    attr.ActivityType.GetName(),\n\t\t\t}\n\t\t\tactivityDispatchInfo.TaskToken, err = handler.tokenSerializer.Serialize(token)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, workflow.ErrSerializingToken\n\t\t\t}\n\t\t\tactivityDispatchInfo.ScheduledTimestamp = common.Int64Ptr(ai.ScheduledTime.UnixNano())\n\t\t\tactivityDispatchInfo.ScheduledTimestampOfThisAttempt = common.Int64Ptr(ai.ScheduledTime.UnixNano())\n\t\t\tactivityDispatchInfo.StartedTimestamp = common.Int64Ptr(ai.StartedTime.UnixNano())\n\t\t\treturn &decisionResult{activityDispatchInfo: activityDispatchInfo}, nil\n\t\t}\n\t\treturn nil, nil\n\tcase *types.BadRequestError:\n\t\treturn nil, handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseScheduleActivityDuplicateID, \"\",\n\t\t)\n\tcase *types.InternalServiceError:\n\t\t// Check if this is ErrTooManyPendingActivities\n\t\tif err.Error() == execution.ErrTooManyPendingActivities.Error() {\n\t\t\treturn nil, handler.handleFailWorkflowError(common.FailureReasonPendingActivityExceedsLimit, err.Error())\n\t\t}\n\t\treturn nil, err\n\tdefault:\n\t\treturn nil, err\n\t}\n}\n\n// handleFailWorkflowError handles the certain types of error by failing the workflow\nfunc (handler *taskHandlerImpl) handleFailWorkflowError(failReason string, failDetails string) error {\n\t// Fail the workflow immediately instead of just returning the error\n\thandler.stopProcessing = true\n\n\tfailAttributes := &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  common.StringPtr(failReason),\n\t\tDetails: []byte(failDetails),\n\t}\n\n\tif _, failErr := handler.mutableState.AddFailWorkflowEvent(\n\t\thandler.decisionTaskCompletedID,\n\t\tfailAttributes,\n\t); failErr != nil {\n\t\treturn failErr\n\t}\n\n\treturn nil\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionRequestCancelActivity(\n\tctx context.Context,\n\tattr *types.RequestCancelActivityTaskDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeCancelActivityCounter,\n\t)\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateActivityCancelAttributes(\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t\thandler.domainEntry.GetInfo().Name)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelActivityAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tactivityID := attr.GetActivityID()\n\tactCancelReqEvent, ai, err := handler.mutableState.AddActivityTaskCancelRequestedEvent(\n\t\thandler.decisionTaskCompletedID,\n\t\tactivityID,\n\t\thandler.identity,\n\t)\n\tswitch err.(type) {\n\tcase nil:\n\t\tif ai.StartedID == constants.EmptyEventID {\n\t\t\t// We haven't started the activity yet, we can cancel the activity right away and\n\t\t\t// schedule a decision task to ensure the workflow makes progress.\n\t\t\t_, err = handler.mutableState.AddActivityTaskCanceledEvent(\n\t\t\t\tai.ScheduleID,\n\t\t\t\tai.StartedID,\n\t\t\t\tactCancelReqEvent.ID,\n\t\t\t\t[]byte(activityCancellationMsgActivityNotStarted),\n\t\t\t\thandler.identity,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\thandler.activityNotStartedCancelled = true\n\t\t}\n\t\treturn nil\n\tcase *types.BadRequestError:\n\t\t_, err = handler.mutableState.AddRequestCancelActivityTaskFailedEvent(\n\t\t\thandler.decisionTaskCompletedID,\n\t\t\tactivityID,\n\t\t\tactivityCancellationMsgActivityIDUnknown,\n\t\t)\n\t\treturn err\n\tdefault:\n\t\treturn err\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionStartTimer(\n\tctx context.Context,\n\tattr *types.StartTimerDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeStartTimerCounter,\n\t)\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateTimerScheduleAttributes(\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t\thandler.domainEntry.GetInfo().Name)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadStartTimerAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\t_, _, err := handler.mutableState.AddTimerStartedEvent(handler.decisionTaskCompletedID, attr)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn nil\n\tcase *types.BadRequestError:\n\t\treturn handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseStartTimerDuplicateID, \"\",\n\t\t)\n\tdefault:\n\t\treturn err\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionCompleteWorkflow(\n\tctx context.Context,\n\tattr *types.CompleteWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeCompleteWorkflowCounter,\n\t)\n\n\tif handler.hasUnhandledEventsBeforeDecisions {\n\t\treturn handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseUnhandledDecision,\n\t\t\t\"cannot complete workflow, new pending decisions were scheduled while this decision was processing\",\n\t\t)\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateCompleteWorkflowExecutionAttributes(attr)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeCompleteWorkflowExecution.String()),\n\t\tattr.Result,\n\t\t\"CompleteWorkflowExecutionDecisionAttributes.Result exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\t// If the decision has more than one completion event than just pick the first one\n\tif !handler.mutableState.IsWorkflowExecutionRunning() {\n\t\thandler.metricsClient.IncCounter(\n\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\tmetrics.MultipleCompletionDecisionsCounter,\n\t\t)\n\t\thandler.logger.Warn(\n\t\t\t\"Multiple completion decisions\",\n\t\t\ttag.WorkflowDecisionType(int64(types.DecisionTypeCompleteWorkflowExecution)),\n\t\t\ttag.ErrorTypeMultipleCompletionDecisions,\n\t\t)\n\t\treturn nil\n\t}\n\n\t// event ID is not relevant\n\tisCanceled, _ := handler.mutableState.IsCancelRequested()\n\n\t// check if this is a cron workflow\n\tcronBackoff, err := handler.mutableState.GetCronBackoffDuration(ctx)\n\tif err != nil {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\tif isCanceled || cronBackoff == backoff.NoBackoff {\n\t\t// canceled or not cron, so complete this workflow execution\n\t\tif _, err := handler.mutableState.AddCompletedWorkflowEvent(handler.decisionTaskCompletedID, attr); err != nil {\n\t\t\treturn &types.InternalServiceError{Message: \"Unable to add complete workflow event.\"}\n\t\t}\n\t\treturn nil\n\t}\n\n\t// this is a cron workflow\n\tstartEvent, err := handler.mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tstartAttributes := startEvent.WorkflowExecutionStartedEventAttributes\n\treturn handler.retryCronContinueAsNew(\n\t\tctx,\n\t\tstartAttributes,\n\t\tint32(cronBackoff.Seconds()),\n\t\ttypes.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t\tnil,\n\t\tnil,\n\t\tattr.Result,\n\t)\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionFailWorkflow(\n\tctx context.Context,\n\tattr *types.FailWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeFailWorkflowCounter,\n\t)\n\n\tif handler.hasUnhandledEventsBeforeDecisions {\n\t\treturn handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseUnhandledDecision,\n\t\t\t\"cannot complete workflow, new pending decisions were scheduled while this decision was processing\",\n\t\t)\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateFailWorkflowExecutionAttributes(attr)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeFailWorkflowExecution.String()),\n\t\tattr.Details,\n\t\t\"FailWorkflowExecutionDecisionAttributes.Details exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\t// If the decision has more than one completion event than just pick the first one\n\tif !handler.mutableState.IsWorkflowExecutionRunning() {\n\t\thandler.metricsClient.IncCounter(\n\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\tmetrics.MultipleCompletionDecisionsCounter,\n\t\t)\n\t\thandler.logger.Warn(\n\t\t\t\"Multiple completion decisions\",\n\t\t\ttag.WorkflowDecisionType(int64(types.DecisionTypeFailWorkflowExecution)),\n\t\t\ttag.ErrorTypeMultipleCompletionDecisions,\n\t\t)\n\t\treturn nil\n\t}\n\n\tif is, _ := handler.mutableState.IsCancelRequested(); is {\n\t\t// cancellation must be sticky, as it's telling things to stop.\n\t\t// this is particularly important for child workflows, as if they restart themselves after the parent\n\t\t// cancels its context, there is no way for the parent to cancel the new run.\n\t\tcancelAttrs := types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\t// TODO: serialize reason somehow, may deserve a new field / wrapped errors\n\t\t\tDetails: attr.Details,\n\t\t}\n\t\tif _, err := handler.mutableState.AddWorkflowExecutionCanceledEvent(handler.decisionTaskCompletedID, &cancelAttrs); err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\t// below will check whether to do continue as new based on backoff & backoff or cron\n\tbackoffInterval := handler.mutableState.GetRetryBackoffDuration(attr.GetReason())\n\tcontinueAsNewInitiator := types.ContinueAsNewInitiatorRetryPolicy\n\t// first check the backoff retry\n\tif backoffInterval == backoff.NoBackoff {\n\t\t// if no backoff retry, set the backoffInterval using cron schedule\n\t\tbackoffInterval, err = handler.mutableState.GetCronBackoffDuration(ctx)\n\t\tif err != nil {\n\t\t\thandler.stopProcessing = true\n\t\t\treturn err\n\t\t}\n\t\tcontinueAsNewInitiator = types.ContinueAsNewInitiatorCronSchedule\n\t}\n\t// second check the backoff / cron schedule\n\tif backoffInterval == backoff.NoBackoff {\n\t\t// no retry or cron\n\t\tif _, err := handler.mutableState.AddFailWorkflowEvent(handler.decisionTaskCompletedID, attr); err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\t// this is a cron / backoff workflow\n\tstartEvent, err := handler.mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tstartAttributes := startEvent.WorkflowExecutionStartedEventAttributes\n\treturn handler.retryCronContinueAsNew(\n\t\tctx,\n\t\tstartAttributes,\n\t\tint32(backoffInterval.Seconds()),\n\t\tcontinueAsNewInitiator.Ptr(),\n\t\tattr.Reason,\n\t\tattr.Details,\n\t\tstartAttributes.LastCompletionResult,\n\t)\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionCancelTimer(\n\tctx context.Context,\n\tattr *types.CancelTimerDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeCancelTimerCounter,\n\t)\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateTimerCancelAttributes(\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t\thandler.domainEntry.GetInfo().Name)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadCancelTimerAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\t_, err := handler.mutableState.AddTimerCanceledEvent(\n\t\thandler.decisionTaskCompletedID,\n\t\tattr,\n\t\thandler.identity)\n\tswitch err.(type) {\n\tcase nil:\n\t\t// timer deletion is a success, we may have deleted a fired timer in\n\t\t// which case we should reset hasBufferedEvents\n\t\t// TODO deletion of timer fired event refreshing hasUnhandledEventsBeforeDecisions\n\t\t//  is not entirely correct, since during these decisions processing, new event may appear\n\t\thandler.hasUnhandledEventsBeforeDecisions = handler.mutableState.HasBufferedEvents()\n\t\treturn nil\n\tcase *types.BadRequestError:\n\t\t_, err = handler.mutableState.AddCancelTimerFailedEvent(\n\t\t\thandler.decisionTaskCompletedID,\n\t\t\tattr,\n\t\t\thandler.identity,\n\t\t)\n\t\treturn err\n\tdefault:\n\t\treturn err\n\t}\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionCancelWorkflow(\n\tctx context.Context,\n\tattr *types.CancelWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(metrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeCancelWorkflowCounter)\n\n\tif handler.hasUnhandledEventsBeforeDecisions {\n\t\treturn handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseUnhandledDecision,\n\t\t\t\"cannot process cancellation, new pending decisions were scheduled while this decision was processing\",\n\t\t)\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateCancelWorkflowExecutionAttributes(attr)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\t// If the decision has more than one completion event than just pick the first one\n\tif !handler.mutableState.IsWorkflowExecutionRunning() {\n\t\thandler.metricsClient.IncCounter(\n\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\tmetrics.MultipleCompletionDecisionsCounter,\n\t\t)\n\t\thandler.logger.Warn(\n\t\t\t\"Multiple completion decisions\",\n\t\t\ttag.WorkflowDecisionType(int64(types.DecisionTypeCancelWorkflowExecution)),\n\t\t\ttag.ErrorTypeMultipleCompletionDecisions,\n\t\t)\n\t\treturn nil\n\t}\n\n\t_, err := handler.mutableState.AddWorkflowExecutionCanceledEvent(handler.decisionTaskCompletedID, attr)\n\treturn err\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionRequestCancelExternalWorkflow(\n\tctx context.Context,\n\tattr *types.RequestCancelExternalWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeCancelExternalWorkflowCounter,\n\t)\n\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\ttargetDomainID := domainID\n\tif attr.GetDomain() != \"\" {\n\t\ttargetDomainEntry, err := handler.domainCache.GetDomain(attr.GetDomain())\n\t\tif err != nil {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Unable to cancel workflow across domain: %v.\", attr.GetDomain()),\n\t\t\t}\n\t\t}\n\t\ttargetDomainID = targetDomainEntry.GetInfo().ID\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateCancelExternalWorkflowExecutionAttributes(\n\t\t\t\tdomainID,\n\t\t\t\ttargetDomainID,\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadRequestCancelExternalWorkflowExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tcancelRequestID := uuid.New()\n\t_, _, err := handler.mutableState.AddRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\thandler.decisionTaskCompletedID, cancelRequestID, attr,\n\t)\n\treturn err\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionRecordMarker(\n\tctx context.Context,\n\tattr *types.RecordMarkerDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeRecordMarkerCounter,\n\t)\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateRecordMarkerAttributes(\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t\thandler.domainEntry.GetInfo().Name)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadRecordMarkerAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeRecordMarker.String()),\n\t\tattr.Details,\n\t\t\"RecordMarkerDecisionAttributes.Details exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\t_, err = handler.mutableState.AddRecordMarkerEvent(handler.decisionTaskCompletedID, attr)\n\treturn err\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionContinueAsNewWorkflow(\n\tctx context.Context,\n\tattr *types.ContinueAsNewWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeContinueAsNewCounter,\n\t)\n\n\tif handler.hasUnhandledEventsBeforeDecisions {\n\t\treturn handler.handlerFailDecision(\n\t\t\ttypes.DecisionTaskFailedCauseUnhandledDecision,\n\t\t\t\"cannot complete workflow, new pending decisions were scheduled while this decision was processing\",\n\t\t)\n\t}\n\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateContinueAsNewWorkflowExecutionAttributes(\n\t\t\t\tattr,\n\t\t\t\texecutionInfo,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t\thandler.domainEntry.GetInfo().Name,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadContinueAsNewAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeContinueAsNewWorkflowExecution.String()),\n\t\tattr.Input,\n\t\t\"ContinueAsNewWorkflowExecutionDecisionAttributes. Input exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\t// If the decision has more than one completion event than just pick the first one\n\tif !handler.mutableState.IsWorkflowExecutionRunning() {\n\t\thandler.metricsClient.IncCounter(\n\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\tmetrics.MultipleCompletionDecisionsCounter,\n\t\t)\n\t\thandler.logger.Warn(\n\t\t\t\"Multiple completion decisions\",\n\t\t\ttag.WorkflowDecisionType(int64(types.DecisionTypeContinueAsNewWorkflowExecution)),\n\t\t\ttag.ErrorTypeMultipleCompletionDecisions,\n\t\t)\n\t\treturn nil\n\t}\n\n\tif is, _ := handler.mutableState.IsCancelRequested(); is {\n\t\t// cancellation must be sticky, as it's telling things to stop.\n\t\t// this is particularly important for child workflows, as if they restart themselves after the parent\n\t\t// cancels its context, there is no way for the parent to cancel the new run.\n\t\tcancelAttrs := types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\tDetails: nil, // TODO: serialize continue-as-new data somehow, may deserve a new field\n\t\t}\n\t\tif _, err := handler.mutableState.AddWorkflowExecutionCanceledEvent(handler.decisionTaskCompletedID, &cancelAttrs); err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\n\t// Extract parentDomainName so it can be passed down to next run of workflow execution\n\tvar parentDomainName string\n\tif handler.mutableState.HasParentExecution() {\n\t\tparentDomainID := executionInfo.ParentDomainID\n\t\tparentDomainName, err = handler.domainCache.GetDomainName(parentDomainID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t_, newStateBuilder, err := handler.mutableState.AddContinueAsNewEvent(\n\t\tctx,\n\t\thandler.decisionTaskCompletedID,\n\t\thandler.decisionTaskCompletedID,\n\t\tparentDomainName,\n\t\tattr,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\thandler.continueAsNewBuilder = newStateBuilder\n\treturn nil\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionStartChildWorkflow(\n\tctx context.Context,\n\tattr *types.StartChildWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeChildWorkflowCounter,\n\t)\n\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\ttargetDomainID := domainID\n\tif attr.GetDomain() != \"\" {\n\t\ttargetDomainEntry, err := handler.domainCache.GetDomain(attr.GetDomain())\n\t\tif err != nil {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Unable to schedule child execution across domain %v.\", attr.GetDomain()),\n\t\t\t}\n\t\t}\n\t\ttargetDomainID = targetDomainEntry.GetInfo().ID\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateStartChildExecutionAttributes(\n\t\t\t\tdomainID,\n\t\t\t\ttargetDomainID,\n\t\t\t\tattr,\n\t\t\t\texecutionInfo,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadStartChildExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeStartChildWorkflowExecution.String()),\n\t\tattr.Input,\n\t\t\"StartChildWorkflowExecutionDecisionAttributes.Input exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\tenabled := handler.config.EnableParentClosePolicy(handler.domainEntry.GetInfo().Name)\n\tif attr.ParentClosePolicy == nil {\n\t\t// for old clients, this field is empty. If they enable the feature, make default as terminate\n\t\tif enabled {\n\t\t\tattr.ParentClosePolicy = types.ParentClosePolicyTerminate.Ptr()\n\t\t} else {\n\t\t\tattr.ParentClosePolicy = types.ParentClosePolicyAbandon.Ptr()\n\t\t}\n\t} else {\n\t\t// for domains that haven't enabled the feature yet, need to use Abandon for backward-compatibility\n\t\tif !enabled {\n\t\t\tattr.ParentClosePolicy = types.ParentClosePolicyAbandon.Ptr()\n\t\t}\n\t}\n\n\trequestID := uuid.New()\n\t_, _, err = handler.mutableState.AddStartChildWorkflowExecutionInitiatedEvent(\n\t\thandler.decisionTaskCompletedID, requestID, attr,\n\t)\n\treturn err\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionSignalExternalWorkflow(\n\tctx context.Context,\n\tattr *types.SignalExternalWorkflowExecutionDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeSignalExternalWorkflowCounter,\n\t)\n\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\ttargetDomainID := domainID\n\tif attr.GetDomain() != \"\" {\n\t\ttargetDomainEntry, err := handler.domainCache.GetDomain(attr.GetDomain())\n\t\tif err != nil {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"Unable to signal workflow across domain: %v.\", attr.GetDomain()),\n\t\t\t}\n\t\t}\n\t\ttargetDomainID = targetDomainEntry.GetInfo().ID\n\t}\n\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateSignalExternalWorkflowExecutionAttributes(\n\t\t\t\tdomainID,\n\t\t\t\ttargetDomainID,\n\t\t\t\tattr,\n\t\t\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadSignalWorkflowExecutionAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeSignalExternalWorkflowExecution.String()),\n\t\tattr.Input,\n\t\t\"SignalExternalWorkflowExecutionDecisionAttributes.Input exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\tsignalRequestID := uuid.New() // for deduplicate\n\t_, _, err = handler.mutableState.AddSignalExternalWorkflowExecutionInitiatedEvent(\n\t\thandler.decisionTaskCompletedID, signalRequestID, attr,\n\t)\n\treturn err\n}\n\nfunc (handler *taskHandlerImpl) handleDecisionUpsertWorkflowSearchAttributes(\n\tctx context.Context,\n\tattr *types.UpsertWorkflowSearchAttributesDecisionAttributes,\n) error {\n\n\thandler.metricsClient.IncCounter(\n\t\tmetrics.HistoryRespondDecisionTaskCompletedScope,\n\t\tmetrics.DecisionTypeUpsertWorkflowSearchAttributesCounter,\n\t)\n\n\t// get domain name\n\texecutionInfo := handler.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\tdomainName, err := handler.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"Unable to get domain for domainID: %v.\", domainID),\n\t\t}\n\t}\n\n\t// valid search attributes for upsert\n\tif err := handler.validateDecisionAttr(\n\t\tfunc() error {\n\t\t\treturn handler.attrValidator.validateUpsertWorkflowSearchAttributes(\n\t\t\t\tdomainName,\n\t\t\t\tattr,\n\t\t\t)\n\t\t},\n\t\ttypes.DecisionTaskFailedCauseBadSearchAttributes,\n\t); err != nil || handler.stopProcessing {\n\t\treturn err\n\t}\n\n\t// blob size limit check\n\tfailWorkflow, err := handler.sizeLimitChecker.failWorkflowIfBlobSizeExceedsLimit(\n\t\tmetrics.DecisionTypeTag(types.DecisionTypeUpsertWorkflowSearchAttributes.String()),\n\t\tconvertSearchAttributesToByteArray(attr.GetSearchAttributes().GetIndexedFields()),\n\t\t\"UpsertWorkflowSearchAttributesDecisionAttributes exceeds size limit.\",\n\t)\n\tif err != nil || failWorkflow {\n\t\thandler.stopProcessing = true\n\t\treturn err\n\t}\n\n\t_, err = handler.mutableState.AddUpsertWorkflowSearchAttributesEvent(\n\t\thandler.decisionTaskCompletedID, attr,\n\t)\n\treturn err\n}\n\nfunc convertSearchAttributesToByteArray(fields map[string][]byte) []byte {\n\tresult := make([]byte, 0)\n\n\tfor k, v := range fields {\n\t\tresult = append(result, []byte(k)...)\n\t\tresult = append(result, v...)\n\t}\n\treturn result\n}\n\nfunc (handler *taskHandlerImpl) retryCronContinueAsNew(\n\tctx context.Context,\n\tattr *types.WorkflowExecutionStartedEventAttributes,\n\tbackoffInterval int32,\n\tcontinueAsNewIter *types.ContinueAsNewInitiator,\n\tfailureReason *string,\n\tfailureDetails []byte,\n\tlastCompletionResult []byte,\n) error {\n\tcontinueAsNewAttributes := &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        attr.WorkflowType,\n\t\tTaskList:                            attr.TaskList,\n\t\tRetryPolicy:                         attr.RetryPolicy,\n\t\tInput:                               attr.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: attr.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      attr.TaskStartToCloseTimeoutSeconds,\n\t\tCronSchedule:                        attr.CronSchedule,\n\t\tBackoffStartIntervalInSeconds:       common.Int32Ptr(backoffInterval),\n\t\tInitiator:                           continueAsNewIter,\n\t\tFailureReason:                       failureReason,\n\t\tFailureDetails:                      failureDetails,\n\t\tLastCompletionResult:                lastCompletionResult,\n\t\tHeader:                              attr.Header,\n\t\tMemo:                                attr.Memo,\n\t\tSearchAttributes:                    attr.SearchAttributes,\n\t\tJitterStartSeconds:                  attr.JitterStartSeconds,\n\t\tCronOverlapPolicy:                   attr.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        attr.ActiveClusterSelectionPolicy,\n\t}\n\n\t_, newStateBuilder, err := handler.mutableState.AddContinueAsNewEvent(\n\t\tctx,\n\t\thandler.decisionTaskCompletedID,\n\t\thandler.decisionTaskCompletedID,\n\t\tattr.GetParentWorkflowDomain(),\n\t\tcontinueAsNewAttributes,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\thandler.continueAsNewBuilder = newStateBuilder\n\treturn nil\n}\n\nfunc (handler *taskHandlerImpl) validateDecisionAttr(\n\tvalidationFn attrValidationFn,\n\tfailedCause types.DecisionTaskFailedCause,\n) error {\n\n\tif err := validationFn(); err != nil {\n\t\tif _, ok := err.(*types.BadRequestError); ok {\n\t\t\treturn handler.handlerFailDecision(failedCause, err.Error())\n\t\t}\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc (handler *taskHandlerImpl) handlerFailDecision(\n\tfailedCause types.DecisionTaskFailedCause,\n\tfailMessage string,\n) error {\n\thandler.failDecision = true\n\thandler.failDecisionCause = failedCause.Ptr()\n\thandler.failMessage = common.StringPtr(failMessage)\n\thandler.stopProcessing = true\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/decision/task_handler_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage decision\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nconst (\n\ttestTaskCompletedID = int64(123)\n)\n\nfunc TestHandleDecisionRequestCancelExternalWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl)\n\t\tattributes      *types.RequestCancelExternalWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, err error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tattributes: &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\tWorkflowID:        constants.TestWorkflowID,\n\t\t\t\tRunID:             constants.TestRunID,\n\t\t\t\tControl:           nil,\n\t\t\t\tChildWorkflowOnly: false,\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal service error\",\n\t\t\tattributes: &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\tWorkflowID:        constants.TestWorkflowID,\n\t\t\t\tRunID:             constants.TestRunID,\n\t\t\t\tControl:           nil,\n\t\t\t\tChildWorkflowOnly: false,\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(nil, errors.New(\"some error getting domain cache\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"Unable to cancel workflow across domain: some random domain name.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.NotNil(t, taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t\t// 9 is types.DecisionTaskFailedCause \"BAD_REQUEST_CANCEL_EXTERNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\t\t\t\t}(9), taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(1).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t})\n\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddRequestCancelExternalWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), test.attributes).AnyTimes()\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeRequestCancelExternalWorkflowExecution),\n\t\t\t\tRequestCancelExternalWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionRequestCancelActivity(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl)\n\t\tattributes      *types.RequestCancelActivityTaskDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.RequestCancelActivityTaskDecisionAttributes{ActivityID: testdata.ActivityID},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCancelRequestedEvent(\n\t\t\t\t\ttestTaskCompletedID,\n\t\t\t\t\ttestdata.ActivityID,\n\t\t\t\t\ttestdata.Identity,\n\t\t\t\t).Times(1).Return(&types.HistoryEvent{}, &persistence.ActivityInfo{StartedID: commonconstants.EmptyEventID}, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCanceledEvent(int64(0), int64(-23), int64(0), []byte(activityCancellationMsgActivityNotStarted), testdata.Identity).Return(nil, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"AddActivityTaskCanceledEvent failure\",\n\t\t\tattributes: &types.RequestCancelActivityTaskDecisionAttributes{ActivityID: testdata.ActivityID},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCancelRequestedEvent(\n\t\t\t\t\ttestTaskCompletedID,\n\t\t\t\t\ttestdata.ActivityID,\n\t\t\t\t\ttestdata.Identity,\n\t\t\t\t).Times(1).Return(&types.HistoryEvent{}, &persistence.ActivityInfo{StartedID: commonconstants.EmptyEventID}, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCanceledEvent(int64(0), int64(-23), int64(0), []byte(activityCancellationMsgActivityNotStarted), testdata.Identity).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some random error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"AddRequestCancelActivityTaskFailedEvent failure\",\n\t\t\tattributes: &types.RequestCancelActivityTaskDecisionAttributes{ActivityID: testdata.ActivityID},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCancelRequestedEvent(\n\t\t\t\t\ttestTaskCompletedID,\n\t\t\t\t\ttestdata.ActivityID,\n\t\t\t\t\ttestdata.Identity,\n\t\t\t\t).Times(1).Return(&types.HistoryEvent{}, &persistence.ActivityInfo{StartedID: commonconstants.EmptyEventID}, &types.BadRequestError{Message: \"some types.BadRequestError error\"})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddRequestCancelActivityTaskFailedEvent(testTaskCompletedID, testdata.ActivityID, activityCancellationMsgActivityIDUnknown).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some random error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"default switch case error\",\n\t\t\tattributes: &types.RequestCancelActivityTaskDecisionAttributes{ActivityID: testdata.ActivityID},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskCancelRequestedEvent(\n\t\t\t\t\ttestTaskCompletedID,\n\t\t\t\t\ttestdata.ActivityID,\n\t\t\t\t\ttestdata.Identity,\n\t\t\t\t).Times(1).Return(&types.HistoryEvent{}, &persistence.ActivityInfo{StartedID: commonconstants.EmptyEventID}, errors.New(\"some default error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some default error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.NotNil(t, taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t\t// 2 is types.DecisionTaskFailedCause \"BAD_REQUEST_CANCEL_ACTIVITY_ATTRIBUTES\"\n\t\t\t\t}(2), taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddRequestCancelExternalWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), test.attributes).AnyTimes()\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeRequestCancelActivityTask),\n\t\t\t\tRequestCancelActivityTaskDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionStartChildWorkflow(t *testing.T) {\n\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.StartChildWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname: \"success - ParentClosePolicy enabled\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t\tTaskList:     &types.TaskList{Name: testdata.TaskListName},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddStartChildWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), attr).Times(1)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.ParentClosePolicyTerminate.Ptr(), attr.ParentClosePolicy)\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - ParentClosePolicy disabled\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t\tTaskList:     &types.TaskList{Name: testdata.TaskListName},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.config.EnableParentClosePolicy = func(domain string) bool { return false }\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddStartChildWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), attr).Times(1)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.ParentClosePolicyAbandon.Ptr(), attr.ParentClosePolicy)\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - ParentClosePolicy non nil\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\tWorkflowID:        constants.TestWorkflowID,\n\t\t\t\tWorkflowType:      &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t\tTaskList:          &types.TaskList{Name: testdata.TaskListName},\n\t\t\t\tParentClosePolicy: new(types.ParentClosePolicy),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.config.EnableParentClosePolicy = func(domain string) bool { return false }\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddStartChildWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), attr).Times(1)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.ParentClosePolicyAbandon.Ptr(), attr.ParentClosePolicy)\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal service error\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t\tTaskList:     &types.TaskList{Name: testdata.TaskListName},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(nil, errors.New(\"some error getting domain cache\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"Unable to schedule child execution across domain some random domain name.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.Equal(t, \"missing task list name\", *taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t\t// 15 is types.DecisionTaskFailedCause \"BAD_START_CHILD_EXECUTION_ATTRIBUTES\"\n\t\t\t\t}(15), taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"size limit checker failure\",\n\t\t\tattributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\tWorkflowID:   constants.TestWorkflowID,\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\t\t\tTaskList:     &types.TaskList{Name: testdata.TaskListName},\n\t\t\t\tInput:        []byte(\"input\"),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 1\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 2\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"StartChildWorkflowExecutionDecisionAttributes.Input exceeds size limit.\"),\n\t\t\t\t}).Return(nil, errors.New(\"some error adding fail workflow event\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartChildWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error adding fail workflow event\"), err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().AnyTimes().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t})\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeStartChildWorkflowExecution),\n\t\t\t\tStartChildWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionCancelTimer(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes)\n\t\tattributes      *types.CancelTimerDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.CancelTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerCanceledEvent(testTaskCompletedID, attr, taskHandler.identity)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().HasBufferedEvents()\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"bad request error\",\n\t\t\tattributes: &types.CancelTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerCanceledEvent(testTaskCompletedID, attr, taskHandler.identity).Return(nil, &types.BadRequestError{\"some bad request error\"})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddCancelTimerFailedEvent(testTaskCompletedID, attr, taskHandler.identity)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"default error\",\n\t\t\tattributes: &types.CancelTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerCanceledEvent(testTaskCompletedID, attr, taskHandler.identity).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, errors.New(\"some random error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"attributes validation error\",\n\t\t\tattributes: &types.CancelTimerDecisionAttributes{},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelTimerDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.Equal(t, \"TimerId is not set on decision.\", *taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t}(4), taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType:                  common.Ptr(types.DecisionTypeCancelTimer),\n\t\t\t\tCancelTimerDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionSignalExternalWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.SignalExternalWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tattributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:     constants.TestDomainName,\n\t\t\t\tExecution:  &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\tSignalName: testdata.SignalName,\n\t\t\t\tInput:      []byte(\"some input data\"),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(2).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddSignalExternalWorkflowExecutionInitiatedEvent(testTaskCompletedID, gomock.Any(), attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal service error\",\n\t\t\tattributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:     constants.TestDomainName,\n\t\t\t\tExecution:  &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\tSignalName: testdata.SignalName,\n\t\t\t\tInput:      []byte(\"some input data\"),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(1).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(nil, errors.New(\"some error getting domain cache\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"Unable to signal workflow across domain: some random domain name.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tattributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:    constants.TestDomainName,\n\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\tInput:     []byte(\"some input data\"),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(1).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.Equal(t, \"SignalName is not set on decision.\", *taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t\t// 14 is types.DecisionTaskFailedCause \"BAD_SIGNAL_WORKFLOW_EXECUTION_ATTRIBUTES\"\n\t\t\t\t}(14), taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"size limit checker failure\",\n\t\t\tattributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\t\tDomain:     constants.TestDomainName,\n\t\t\t\tExecution:  &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\tSignalName: testdata.SignalName,\n\t\t\t\tInput:      []byte(\"some input data\"),\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 1\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 2\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(2).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"SignalExternalWorkflowExecutionDecisionAttributes.Input exceeds size limit.\"),\n\t\t\t\t}).Return(nil, errors.New(\"some error adding fail workflow event\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.SignalExternalWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error adding fail workflow event\"), err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeSignalExternalWorkflowExecution),\n\t\t\t\tSignalExternalWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionUpsertWorkflowSearchAttributes(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes)\n\t\tattributes      *types.UpsertWorkflowSearchAttributesDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tattributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{IndexedFields: map[string][]byte{\"some-key\": []byte(`\"some-value\"`)}},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(2).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddUpsertWorkflowSearchAttributesEvent(testTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, nil, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tattributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{IndexedFields: map[string][]byte{\"some-key\": []byte(\"some-value\")}},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(1).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.Equal(t, \"some-value is not a valid search attribute value for key some-key\", *taskHandler.failMessage)\n\t\t\t\tassert.Equal(t, func(i int32) *types.DecisionTaskFailedCause {\n\t\t\t\t\tcause := new(types.DecisionTaskFailedCause)\n\t\t\t\t\t*cause = types.DecisionTaskFailedCause(i)\n\t\t\t\t\treturn cause\n\t\t\t\t\t// 22 is types.DecisionTaskFailedCause \"BAD_SEARCH_ATTRIBUTES\"\n\t\t\t\t}(22), taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"size limit checker failure\",\n\t\t\tattributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{IndexedFields: map[string][]byte{\"some-key\": []byte(`\"some-value\"`)}},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 2\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 1\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(2).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"UpsertWorkflowSearchAttributesDecisionAttributes exceeds size limit.\"),\n\t\t\t\t}).Return(nil, errors.New(\"some random error adding workflow event\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes, err error) {\n\t\t\t\tassert.False(t, taskHandler.failDecision)\n\t\t\t\tassert.Empty(t, taskHandler.failMessage)\n\t\t\t\tassert.Nil(t, taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, errors.New(\"some random error adding workflow event\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"internal service error\",\n\t\t\tattributes: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{IndexedFields: map[string][]byte{\"some-key\": []byte(`\"some-value\"`)}},\n\t\t\t},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Times(1).Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(\"\", errors.New(\"some random error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.UpsertWorkflowSearchAttributesDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"Unable to get domain for domainID: deadbeef-0123-4567-890a-bcdef0123456.\"}, err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeUpsertWorkflowSearchAttributes),\n\t\t\t\tUpsertWorkflowSearchAttributesDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionStartTimer(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes)\n\t\tattributes      *types.StartTimerDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.StartTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes) {\n\t\t\t\tattr.StartToFireTimeoutSeconds = new(int64)\n\t\t\t\t*attr.StartToFireTimeoutSeconds = 60\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerStartedEvent(testTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"attributes validation failure\",\n\t\t\tattributes: &types.StartTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"bad request error\",\n\t\t\tattributes: &types.StartTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes) {\n\t\t\t\tattr.StartToFireTimeoutSeconds = new(int64)\n\t\t\t\t*attr.StartToFireTimeoutSeconds = 60\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerStartedEvent(testTaskCompletedID, attr).Return(nil, nil, &types.BadRequestError{Message: \"some types.BadRequestError error\"})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.True(t, taskHandler.failDecision)\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseStartTimerDuplicateID, *taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"default case error\",\n\t\t\tattributes: &types.StartTimerDecisionAttributes{TimerID: \"test-timer-id\"},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes) {\n\t\t\t\tattr.StartToFireTimeoutSeconds = new(int64)\n\t\t\t\t*attr.StartToFireTimeoutSeconds = 60\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddTimerStartedEvent(testTaskCompletedID, attr).Return(nil, nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.StartTimerDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, \"some random error\", err.Error())\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType:                 common.Ptr(types.DecisionTypeStartTimer),\n\t\t\t\tStartTimerDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionCompleteWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.CompleteWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"handler has unhandled events\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.hasUnhandledEventsBeforeDecisions = true\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseUnhandledDecision, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"cannot complete workflow, new pending decisions were scheduled while this decision was processing\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseBadCompleteWorkflowExecutionAttributes, *taskHandler.failDecisionCause)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"blob size limit check failure\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 5\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 3\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(taskHandler.sizeLimitChecker.completedID,\n\t\t\t\t\t&types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\t\tDetails: []byte(\"CompleteWorkflowExecutionDecisionAttributes.Result exceeds size limit.\"),\n\t\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"workflow not running\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(false)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"failure to get cron duration\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested()\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(time.Second, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"internal service error cancel requested\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(backoff.NoBackoff, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddCompletedWorkflowEvent(taskHandler.decisionTaskCompletedID, attr).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"Unable to add complete workflow event.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cancel requested - sucess\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(backoff.NoBackoff, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddCompletedWorkflowEvent(taskHandler.decisionTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"GetStartedEvent failure on cron workflow\",\n\t\t\tattributes: &types.CompleteWorkflowExecutionDecisionAttributes{Result: []byte(\"some-result\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested()\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background())\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetStartEvent(context.Background()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CompleteWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeCompleteWorkflowExecution),\n\t\t\t\tCompleteWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionFailWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.FailWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"handler has unhandled events\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.hasUnhandledEventsBeforeDecisions = true\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseUnhandledDecision, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"cannot complete workflow, new pending decisions were scheduled while this decision was processing\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseBadFailWorkflowExecutionAttributes, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"FailWorkflowExecutionDecisionAttributes is not set on decision.\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"blob size limit check failure\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 3\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 5\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(taskHandler.sizeLimitChecker.completedID,\n\t\t\t\t\t&types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\t\tDetails: []byte(\"FailWorkflowExecutionDecisionAttributes.Details exceeds size limit.\"),\n\t\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"workflow not running\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(false)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cancel requested - failure to add cancel event\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddWorkflowExecutionCanceledEvent(taskHandler.decisionTaskCompletedID,\n\t\t\t\t\t&types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tDetails: attr.Details,\n\t\t\t\t\t}).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cancel requested - success\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddWorkflowExecutionCanceledEvent(taskHandler.decisionTaskCompletedID,\n\t\t\t\t\t&types.CancelWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tDetails: attr.Details,\n\t\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"failure to get backoff duration on cron\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetRetryBackoffDuration(attr.GetReason()).Return(backoff.NoBackoff)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(time.Second, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"AddFailWorkflowEvent failure\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetRetryBackoffDuration(attr.GetReason()).Return(backoff.NoBackoff)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(backoff.NoBackoff, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(taskHandler.decisionTaskCompletedID, attr).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cron workflow - success\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetRetryBackoffDuration(attr.GetReason()).Return(backoff.NoBackoff)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetCronBackoffDuration(context.Background()).Return(backoff.NoBackoff, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(taskHandler.decisionTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"GetStartEvent failure\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetRetryBackoffDuration(attr.GetReason())\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetStartEvent(context.Background()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.FailWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes) {\n\t\t\t\tattr.Reason = new(string)\n\t\t\t\t*attr.Reason = \"some reason\"\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetRetryBackoffDuration(attr.GetReason())\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetStartEvent(context.Background()).Return(&types.HistoryEvent{WorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{}}, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddContinueAsNewEvent(context.Background(),\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any())\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.FailWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType:                            common.Ptr(types.DecisionTypeFailWorkflowExecution),\n\t\t\t\tFailWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionCancelWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.CancelWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"handler has unhandled events\",\n\t\t\tattributes: &types.CancelWorkflowExecutionDecisionAttributes{},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.hasUnhandledEventsBeforeDecisions = true\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseUnhandledDecision, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"cannot process cancellation, new pending decisions were scheduled while this decision was processing\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseBadCancelWorkflowExecutionAttributes, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"workflow not running\",\n\t\t\tattributes: &types.CancelWorkflowExecutionDecisionAttributes{Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.metricsClient = new(mocks.Client)\n\t\t\t\ttaskHandler.logger = log.NewMockLogger(gomock.NewController(t))\n\t\t\t\ttaskHandler.metricsClient.(*mocks.Client).On(\"IncCounter\", metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DecisionTypeCancelWorkflowCounter)\n\t\t\t\ttaskHandler.metricsClient.(*mocks.Client).On(\"IncCounter\", metrics.HistoryRespondDecisionTaskCompletedScope, metrics.MultipleCompletionDecisionsCounter)\n\t\t\t\ttaskHandler.logger.(*log.MockLogger).EXPECT().Warn(\"Multiple completion decisions\", []tag.Tag{tag.WorkflowDecisionType(int64(types.DecisionTypeCancelWorkflowExecution)), tag.ErrorTypeMultipleCompletionDecisions})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(false)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.CancelWorkflowExecutionDecisionAttributes{},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddWorkflowExecutionCanceledEvent(taskHandler.decisionTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.CancelWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeCancelWorkflowExecution),\n\t\t\t\tCancelWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionRecordMarker(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes)\n\t\tattributes      *types.RecordMarkerDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"blob size limit check failure\",\n\t\t\tattributes: &types.RecordMarkerDecisionAttributes{MarkerName: \"some-marker\", Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 5\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 3\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"RecordMarkerDecisionAttributes.Details exceeds size limit.\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseBadRecordMarkerAttributes, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"success\",\n\t\t\tattributes: &types.RecordMarkerDecisionAttributes{MarkerName: \"some-marker\", Details: []byte(\"some-details\")},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddRecordMarkerEvent(taskHandler.decisionTaskCompletedID, attr)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.RecordMarkerDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType:                   common.Ptr(types.DecisionTypeRecordMarker),\n\t\t\t\tRecordMarkerDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionScheduleActivity(t *testing.T) {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testdata.DomainID, Name: testdata.DomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention:   1,\n\t\t\tBadBinaries: types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{\"test-binary-checksum\": {Reason: \"some reason\"}}},\n\t\t},\n\t\tcluster.TestCurrentClusterName)\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:        testdata.DomainID,\n\t\tWorkflowID:      testdata.WorkflowID,\n\t\tWorkflowTimeout: 100,\n\t}\n\tvalidAttr := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tDomain:                        testdata.DomainName,\n\t\tTaskList:                      &types.TaskList{Name: testdata.TaskListName},\n\t\tActivityID:                    \"some-activity-id\",\n\t\tActivityType:                  &types.ActivityType{Name: testdata.ActivityTypeName},\n\t\tScheduleToCloseTimeoutSeconds: func(i int32) *int32 { return &i }(100),\n\t\tScheduleToStartTimeoutSeconds: func(i int32) *int32 { return &i }(20),\n\t\tStartToCloseTimeoutSeconds:    func(i int32) *int32 { return &i }(80),\n\t\tInput:                         []byte(\"some-input\"),\n\t}\n\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes)\n\t\tattributes      *types.ScheduleActivityTaskDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"internal service error\",\n\t\t\tattributes: &types.ScheduleActivityTaskDecisionAttributes{Domain: testdata.DomainName},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(nil, errors.New(\"some radom error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: fmt.Sprintf(\"Unable to schedule activity across domain %v.\", attr.GetDomain())}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"attributes validation failure\",\n\t\t\tattributes: &types.ScheduleActivityTaskDecisionAttributes{Domain: testdata.DomainName},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"blob size limit check failure\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 3\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 5\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID,\n\t\t\t\t\t&types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\t\tDetails: []byte(\"ScheduleActivityTaskDecisionAttributes.Input exceeds size limit.\"),\n\t\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"success - activity started\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, &persistence.ActivityInfo{}, &types.ActivityLocalDispatchInfo{}, true, true, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskStartedEvent(&persistence.ActivityInfo{}, int64(0), gomock.Any(), taskHandler.identity)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"token serialization failure\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, &persistence.ActivityInfo{}, &types.ActivityLocalDispatchInfo{}, true, false, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskStartedEvent(&persistence.ActivityInfo{}, int64(0), gomock.Any(), taskHandler.identity)\n\t\t\t\ttaskHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Serialize(&common.TaskToken{\n\t\t\t\t\tDomainID:     testdata.DomainID,\n\t\t\t\t\tWorkflowID:   testdata.WorkflowID,\n\t\t\t\t\tActivityType: testdata.ActivityTypeName,\n\t\t\t\t}).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Equal(t, workflow.ErrSerializingToken, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"success - decisionResult non nil\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, &persistence.ActivityInfo{}, &types.ActivityLocalDispatchInfo{}, true, false, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskStartedEvent(&persistence.ActivityInfo{}, int64(0), gomock.Any(), taskHandler.identity)\n\t\t\t\ttaskHandler.tokenSerializer.(*common.MockTaskTokenSerializer).EXPECT().Serialize(&common.TaskToken{\n\t\t\t\t\tDomainID:     testdata.DomainID,\n\t\t\t\t\tWorkflowID:   testdata.WorkflowID,\n\t\t\t\t\tActivityType: testdata.ActivityTypeName,\n\t\t\t\t}).Return([]byte(\"some-serialized-data\"), nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.NotNil(t, res)\n\t\t\t\tassert.Equal(t, []byte(\"some-serialized-data\"), res.activityDispatchInfo.TaskToken)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"failure to add activity task started\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, &persistence.ActivityInfo{}, &types.ActivityLocalDispatchInfo{}, true, true, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskStartedEvent(&persistence.ActivityInfo{}, int64(0), gomock.Any(), taskHandler.identity).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"activity scheduled - nil activityDispatchInfo\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(&types.HistoryEvent{}, &persistence.ActivityInfo{}, nil, true, false, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"bad request error\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(nil, nil, nil, false, false, &types.BadRequestError{\n\t\t\t\t\t\tMessage: \"some bad request error\",\n\t\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"default error case\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(context.Background(), taskHandler.decisionTaskCompletedID, attr, taskHandler.activityCountToDispatch > 0).\n\t\t\t\t\tReturn(nil, nil, nil, false, false, errors.New(\"some default error\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"some default error\", err.Error())\n\t\t\t\tassert.Nil(t, res)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType:                           common.Ptr(types.DecisionTypeScheduleActivityTask),\n\t\t\t\tScheduleActivityTaskDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\tassert.NotNil(t, err)\n\t\t\tassert.Equal(t, &types.BadRequestError{Message: fmt.Sprintf(\"Unknown decision type: %v\", decision.GetDecisionType())}, err)\n\n\t\t\tdecisionRes, err := taskHandler.handleDecisionWithResult(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, decisionRes, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionContinueAsNewWorkflow(t *testing.T) {\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:        testdata.DomainID,\n\t\tWorkflowID:      testdata.WorkflowID,\n\t\tWorkflowTimeout: 100,\n\t\tParentDomainID:  \"parent-domain-id\",\n\t}\n\tvalidAttr := &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        &types.WorkflowType{Name: testdata.WorkflowTypeName},\n\t\tTaskList:                            &types.TaskList{Name: testdata.TaskListName},\n\t\tInput:                               []byte(\"some-input\"),\n\t\tExecutionStartToCloseTimeoutSeconds: func(i int32) *int32 { return &i }(100),\n\t\tTaskStartToCloseTimeoutSeconds:      func(i int32) *int32 { return &i }(80),\n\t\tBackoffStartIntervalInSeconds:       new(int32),\n\t\tFailureReason:                       func(i string) *string { return &i }(\"some reason\"),\n\t\tSearchAttributes:                    &types.SearchAttributes{IndexedFields: map[string][]byte{\"some-key\": []byte(`\"some-value\"`)}},\n\t}\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes)\n\t\tattributes      *types.ContinueAsNewWorkflowExecutionDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"handler has unhandled events\",\n\t\t\tattributes: &types.ContinueAsNewWorkflowExecutionDecisionAttributes{},\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.hasUnhandledEventsBeforeDecisions = true\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseUnhandledDecision, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"cannot complete workflow, new pending decisions were scheduled while this decision was processing\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"attributes validation failure\",\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Equal(t, types.DecisionTaskFailedCauseBadContinueAsNewAttributes, *taskHandler.failDecisionCause)\n\t\t\t\tassert.Equal(t, \"ContinueAsNewWorkflowExecutionDecisionAttributes is not set on decision.\", *taskHandler.failMessage)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"blob size limit check failure\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitWarn = 3\n\t\t\t\ttaskHandler.sizeLimitChecker.blobSizeLimitError = 5\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(testTaskCompletedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tReason:  common.StringPtr(common.FailureReasonDecisionBlobSizeExceedsLimit),\n\t\t\t\t\tDetails: []byte(\"ContinueAsNewWorkflowExecutionDecisionAttributes. Input exceeds size limit.\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"workflow not running\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(false)\n\t\t\t\ttaskHandler.metricsClient = new(mocks.Client)\n\t\t\t\ttaskHandler.logger = log.NewMockLogger(gomock.NewController(t))\n\t\t\t\ttaskHandler.metricsClient.(*mocks.Client).On(\"IncCounter\", metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DecisionTypeContinueAsNewCounter)\n\t\t\t\ttaskHandler.metricsClient.(*mocks.Client).On(\"IncCounter\", metrics.HistoryRespondDecisionTaskCompletedScope, metrics.MultipleCompletionDecisionsCounter)\n\t\t\t\ttaskHandler.logger.(*log.MockLogger).EXPECT().Warn(\"Multiple completion decisions\", []tag.Tag{tag.WorkflowDecisionType(int64(types.DecisionTypeContinueAsNewWorkflowExecution)), tag.ErrorTypeMultipleCompletionDecisions})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cancel requested - failure to add cancel event\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddWorkflowExecutionCanceledEvent(taskHandler.decisionTaskCompletedID,\n\t\t\t\t\t&types.CancelWorkflowExecutionDecisionAttributes{}).Return(nil, errors.New(\"some error adding execution canceled event\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.ErrorContains(t, err, \"some error adding execution canceled event\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"cancel requested - success\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(true, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddWorkflowExecutionCanceledEvent(taskHandler.decisionTaskCompletedID,\n\t\t\t\t\t&types.CancelWorkflowExecutionDecisionAttributes{}).Return(&types.HistoryEvent{}, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"failure to extract parent domain name\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().HasParentExecution().Return(true)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(executionInfo.ParentDomainID).Return(\"\", errors.New(\"some error getting parent domain name\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.ErrorContains(t, err, \"some error getting parent domain name\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"failure to add continueAsNew event\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.attrValidator.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(testdata.DomainID).Return(testdata.DomainName, nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().IsCancelRequested().Return(false, \"\")\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().HasParentExecution().Return(true)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomainName(executionInfo.ParentDomainID).Return(\"parent-domain-name\", nil)\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddContinueAsNewEvent(context.Background(), taskHandler.decisionTaskCompletedID, taskHandler.decisionTaskCompletedID, \"parent-domain-name\", attr).\n\t\t\t\t\tReturn(nil, nil, errors.New(\"some error adding continueAsNew event\"))\n\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ContinueAsNewWorkflowExecutionDecisionAttributes, err error) {\n\t\t\t\tassert.ErrorContains(t, err, \"some error adding continueAsNew event\")\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\t\t\tdecision := &types.Decision{\n\t\t\t\tDecisionType: common.Ptr(types.DecisionTypeContinueAsNewWorkflowExecution),\n\t\t\t\tContinueAsNewWorkflowExecutionDecisionAttributes: test.attributes,\n\t\t\t}\n\t\t\terr := taskHandler.handleDecision(context.Background(), decision)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, err)\n\t\t})\n\t}\n}\n\nfunc TestValidateAttributes(t *testing.T) {\n\tt.Run(\"failure\", func(t *testing.T) {\n\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\tvalidationFn := func() error { return errors.New(\"some validation error\") }\n\t\tfailedCause := new(types.DecisionTaskFailedCause)\n\t\terr := taskHandler.validateDecisionAttr(validationFn, *failedCause)\n\t\tassert.ErrorContains(t, err, \"some validation error\")\n\t})\n}\n\nfunc TestHandleDecisions(t *testing.T) {\n\tt.Run(\"workflow size exceeds limit\", func(t *testing.T) {\n\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\ttaskHandler.sizeLimitChecker.historyCountLimitError = 10\n\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetNextEventID().Return(int64(12)) // nextEventID - 1 > historyCountLimit of 10\n\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(taskHandler.sizeLimitChecker.completedID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\tReason:  common.StringPtr(common.FailureReasonSizeExceedsLimit),\n\t\t\tDetails: []byte(\"Workflow history size / count exceeds limit.\"),\n\t\t}).Return(nil, errors.New(\"some error adding fail workflow event\"))\n\t\tres, err := taskHandler.handleDecisions(context.Background(), []byte{}, []*types.Decision{})\n\t\tassert.Nil(t, res)\n\t\tassert.ErrorContains(t, err, \"some error adding fail workflow event\")\n\t})\n}\n\nfunc newTaskHandlerForTest(t *testing.T) *taskHandlerImpl {\n\tctrl := gomock.NewController(t)\n\ttestLogger := testlogger.New(t)\n\ttestConfig := config.NewForTest()\n\ttestConfig.ValidSearchAttributes = func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\tvalidSearchAttr := make(map[string]interface{})\n\t\tvalidSearchAttr[\"some-key\"] = types.IndexedValueTypeString\n\t\treturn validSearchAttr\n\t}\n\tmockMutableState := execution.NewMockMutableState(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tworkflowSizeChecker := newWorkflowSizeChecker(\n\t\t\"testDomain\",\n\t\ttestConfig.BlobSizeLimitWarn(constants.TestDomainName),\n\t\ttestConfig.BlobSizeLimitError(constants.TestDomainName),\n\t\ttestConfig.HistorySizeLimitWarn(constants.TestDomainName),\n\t\ttestConfig.HistorySizeLimitError(constants.TestDomainName),\n\t\ttestConfig.HistoryCountLimitWarn(constants.TestDomainName),\n\t\ttestConfig.HistoryCountLimitError(constants.TestDomainName),\n\t\ttestTaskCompletedID,\n\t\tmockMutableState,\n\t\t&persistence.ExecutionStats{},\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}).Scope(metrics.HistoryRespondDecisionTaskCompletedScope, metrics.DomainTag(constants.TestDomainName)),\n\t\ttestLogger,\n\t)\n\tmockMutableState.EXPECT().HasBufferedEvents().Return(false)\n\ttaskHandler := newDecisionTaskHandler(\n\t\ttestdata.Identity,\n\t\ttestTaskCompletedID,\n\t\tconstants.TestLocalDomainEntry,\n\t\tmockMutableState,\n\t\tnewAttrValidator(mockDomainCache, metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), testConfig, testlogger.New(t)),\n\t\tworkflowSizeChecker,\n\t\tcommon.NewMockTaskTokenSerializer(ctrl),\n\t\ttestLogger,\n\t\tmockDomainCache,\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\ttestConfig,\n\t)\n\treturn taskHandler\n}\n\nfunc TestHandleFailWorkflowError(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl)\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, err error)\n\t}{\n\t\t{\n\t\t\tname: \"success - workflow fails due to too many pending activities\",\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&types.HistoryEvent{}, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failure - AddFailWorkflowEvent returns error\",\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl) {\n\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, errors.New(\"failed to add fail workflow event\"))\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, err error) {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, \"failed to add fail workflow event\", err.Error())\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler)\n\t\t\t}\n\n\t\t\terr := taskHandler.handleFailWorkflowError(common.FailureReasonPendingActivityExceedsLimit, execution.ErrTooManyPendingActivities.Error())\n\t\t\ttest.asserts(t, taskHandler, err)\n\t\t})\n\t}\n}\n\nfunc TestHandleDecisionScheduleActivityWithTooManyPendingActivities(t *testing.T) {\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testdata.DomainID, Name: testdata.DomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention: 1,\n\t\t},\n\t\tcluster.TestCurrentClusterName)\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:        testdata.DomainID,\n\t\tWorkflowID:      testdata.WorkflowID,\n\t\tWorkflowTimeout: 100,\n\t}\n\tvalidAttr := &types.ScheduleActivityTaskDecisionAttributes{\n\t\tDomain:                        testdata.DomainName,\n\t\tTaskList:                      &types.TaskList{Name: testdata.TaskListName},\n\t\tActivityID:                    \"some-activity-id\",\n\t\tActivityType:                  &types.ActivityType{Name: testdata.ActivityTypeName},\n\t\tScheduleToCloseTimeoutSeconds: func(i int32) *int32 { return &i }(100),\n\t\tScheduleToStartTimeoutSeconds: func(i int32) *int32 { return &i }(20),\n\t\tStartToCloseTimeoutSeconds:    func(i int32) *int32 { return &i }(80),\n\t\tInput:                         []byte(\"some-input\"),\n\t}\n\n\ttests := []struct {\n\t\tname            string\n\t\texpectMockCalls func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes)\n\t\tattributes      *types.ScheduleActivityTaskDecisionAttributes\n\t\tasserts         func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error)\n\t}{\n\t\t{\n\t\t\tname:       \"ErrTooManyPendingActivities - workflow fails\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\n\t\t\t\t// Mock AddActivityTaskScheduledEvent to return ErrTooManyPendingActivities\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tattr,\n\t\t\t\t\ttaskHandler.activityCountToDispatch > 0,\n\t\t\t\t).Return(nil, nil, nil, false, false, execution.ErrTooManyPendingActivities)\n\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddFailWorkflowEvent(\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&types.HistoryEvent{}, nil)\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.Nil(t, err)\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tassert.True(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"Other InternalServiceError - passes through\",\n\t\t\tattributes: validAttr,\n\t\t\texpectMockCalls: func(taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes) {\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().GetExecutionInfo().Return(executionInfo).Times(2)\n\t\t\t\ttaskHandler.domainCache.(*cache.MockDomainCache).EXPECT().GetDomain(attr.GetDomain()).Return(domainEntry, nil)\n\n\t\t\t\t// Mock AddActivityTaskScheduledEvent to return different InternalServiceError\n\t\t\t\ttaskHandler.mutableState.(*execution.MockMutableState).EXPECT().AddActivityTaskScheduledEvent(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\ttaskHandler.decisionTaskCompletedID,\n\t\t\t\t\tattr,\n\t\t\t\t\ttaskHandler.activityCountToDispatch > 0,\n\t\t\t\t).Return(nil, nil, nil, false, false, &types.InternalServiceError{Message: \"Some other internal service error\"})\n\t\t\t},\n\t\t\tasserts: func(t *testing.T, taskHandler *taskHandlerImpl, attr *types.ScheduleActivityTaskDecisionAttributes, res *decisionResult, err error) {\n\t\t\t\tassert.NotNil(t, err)\n\t\t\t\tassert.Equal(t, \"Some other internal service error\", err.Error())\n\t\t\t\tassert.Nil(t, res)\n\t\t\t\tassert.False(t, taskHandler.stopProcessing)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\ttaskHandler := newTaskHandlerForTest(t)\n\t\t\tif test.expectMockCalls != nil {\n\t\t\t\ttest.expectMockCalls(taskHandler, test.attributes)\n\t\t\t}\n\n\t\t\tres, err := taskHandler.handleDecisionScheduleActivity(context.Background(), test.attributes)\n\t\t\ttest.asserts(t, taskHandler, test.attributes, res, err)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/describe_mutable_state.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nfunc (e *historyEngineImpl) DescribeMutableState(\n\tctx context.Context,\n\trequest *types.DescribeMutableStateRequest,\n) (response *types.DescribeMutableStateResponse, retError error) {\n\n\tif err := common.ValidateDomainUUID(request.DomainUUID); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID := request.DomainUUID\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: request.Execution.WorkflowID,\n\t\tRunID:      request.Execution.RunID,\n\t}\n\n\tcacheCtx, dbCtx, release, cacheHit, err := e.executionCache.GetAndCreateWorkflowExecution(\n\t\tctx, domainID, execution,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer func() { release(retError) }()\n\n\tresponse = &types.DescribeMutableStateResponse{}\n\n\tif cacheHit {\n\t\tif msb := cacheCtx.GetWorkflowExecution(); msb != nil {\n\t\t\tresponse.MutableStateInCache, err = e.toMutableStateJSON(msb)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t}\n\n\tmsb, err := dbCtx.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresponse.MutableStateInDatabase, err = e.toMutableStateJSON(msb)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc (e *historyEngineImpl) toMutableStateJSON(msb execution.MutableState) (string, error) {\n\tms := msb.CopyToPersistence()\n\n\tjsonBytes, err := json.Marshal(ms)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn string(jsonBytes), nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/describe_mutable_state_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nfunc TestDescribeMutableState_Success(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tcachedExecutionInfo := getWorkflowMutableState()\n\tdbExecutionInfo := getWorkflowMutableState()\n\n\t// Make sure the db is different from the cache, so we can verify the correct value\n\t// goes to the correct field in the response\n\tdbExecutionInfo.ExecutionInfo.DomainID = constants.TestDomainID + \"other\"\n\n\t// Cache context mock\n\tmutableStateFromCacheMock := execution.NewMockMutableState(ctrl)\n\tmutableStateFromCacheMock.EXPECT().CopyToPersistence().Return(cachedExecutionInfo)\n\tcacheContextMock := execution.NewMockContext(ctrl)\n\tcacheContextMock.EXPECT().GetWorkflowExecution().Return(mutableStateFromCacheMock)\n\n\t// DB context mock\n\tmutableStateFromDBMock := execution.NewMockMutableState(ctrl)\n\tmutableStateFromDBMock.EXPECT().CopyToPersistence().Return(dbExecutionInfo)\n\tdbContextMock := execution.NewMockContext(ctrl)\n\tdbContextMock.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mutableStateFromDBMock, nil)\n\n\treleaseFunctionCalled := false\n\n\tcacheMock := execution.NewMockCache(ctrl)\n\tcacheMock.EXPECT().\n\t\tGetAndCreateWorkflowExecution(gomock.Any(), constants.TestDomainID, *getExpectedWFExecution()).\n\t\tReturn(cacheContextMock, dbContextMock, func(err error) {\n\t\t\treleaseFunctionCalled = true\n\t\t\tassert.NoError(t, err)\n\t\t}, true, nil)\n\n\tengine := &historyEngineImpl{\n\t\texecutionCache: cacheMock,\n\t}\n\n\tresp, err := engine.DescribeMutableState(context.TODO(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  getExpectedWFExecution(),\n\t})\n\tassert.NoError(t, err)\n\n\texpectedCachedExecutionInfo, err := json.Marshal(cachedExecutionInfo)\n\trequire.NoError(t, err)\n\texpectedDBExecutionInfo, err := json.Marshal(dbExecutionInfo)\n\trequire.NoError(t, err)\n\n\t// The response should contain the json representation of the execution info\n\tassert.Equal(t, string(expectedCachedExecutionInfo), resp.MutableStateInCache)\n\tassert.Equal(t, string(expectedDBExecutionInfo), resp.MutableStateInDatabase)\n\n\t// The release function should have been called\n\tassert.True(t, releaseFunctionCalled)\n}\n\nfunc TestDescribeMutableState_Error_UnknownDomain(t *testing.T) {\n\tengine := &historyEngineImpl{}\n\n\t_, err := engine.DescribeMutableState(context.TODO(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"This is not a uuid\",\n\t})\n\tassert.Error(t, err)\n\tassert.ErrorAs(t, err, new(*types.BadRequestError))\n\tassert.ErrorContains(t, err, \"Invalid domain UUID\")\n}\n\nfunc TestDescribeMutableState_Error_GetAndCreateError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tcacheMock := execution.NewMockCache(ctrl)\n\tcacheMock.EXPECT().\n\t\tGetAndCreateWorkflowExecution(gomock.Any(), constants.TestDomainID, *getExpectedWFExecution()).\n\t\tReturn(nil, nil, nil, true, assert.AnError)\n\n\tengine := &historyEngineImpl{\n\t\texecutionCache: cacheMock,\n\t}\n\n\t_, err := engine.DescribeMutableState(context.TODO(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  getExpectedWFExecution(),\n\t})\n\tassert.Error(t, err)\n\tassert.ErrorIs(t, err, assert.AnError)\n}\n\nfunc TestDescribeMutableState_Error_LoadFromDBError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// DB context mock returns error\n\tdbContextMock := execution.NewMockContext(ctrl)\n\tdbContextMock.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, assert.AnError)\n\n\treleaseFunctionCalled := false\n\tcacheMock := execution.NewMockCache(ctrl)\n\tcacheMock.EXPECT().\n\t\tGetAndCreateWorkflowExecution(gomock.Any(), constants.TestDomainID, *getExpectedWFExecution()).\n\t\tReturn(nil, dbContextMock, func(err error) {\n\t\t\treleaseFunctionCalled = true\n\t\t\tassert.ErrorIs(t, err, assert.AnError)\n\t\t}, false, nil)\n\n\tengine := &historyEngineImpl{\n\t\texecutionCache: cacheMock,\n\t}\n\n\t_, err := engine.DescribeMutableState(context.TODO(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  getExpectedWFExecution(),\n\t})\n\tassert.ErrorIs(t, err, assert.AnError)\n\n\t// release function should have been called\n\tassert.True(t, releaseFunctionCalled)\n}\n\n// The content of this struct is not important for the purpose of this test,\n// we just want to check that the method returns the json representation of the struct\nfunc getWorkflowMutableState() *persistence.WorkflowMutableState {\n\treturn &persistence.WorkflowMutableState{\n\t\tActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\tTimerInfos:          nil,\n\t\tChildExecutionInfos: nil,\n\t\tRequestCancelInfos:  nil,\n\t\tSignalInfos:         nil,\n\t\tSignalRequestedIDs:  nil,\n\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   constants.TestDomainID,\n\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\tRunID:      constants.TestRunID,\n\t\t},\n\t\tExecutionStats:   nil,\n\t\tBufferedEvents:   nil,\n\t\tVersionHistories: nil,\n\t\tReplicationState: nil,\n\t\tChecksum:         checksum.Checksum{},\n\t}\n}\n\nfunc getExpectedWFExecution() *types.WorkflowExecution {\n\treturn &types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/describe_queues.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/queue\"\n)\n\nfunc (e *historyEngineImpl) DescribeTransferQueue(\n\tctx context.Context,\n\tclusterName string,\n) (*types.DescribeQueueResponse, error) {\n\treturn e.describeQueue(ctx, persistence.HistoryTaskCategoryTransfer, clusterName)\n}\n\nfunc (e *historyEngineImpl) DescribeTimerQueue(\n\tctx context.Context,\n\tclusterName string,\n) (*types.DescribeQueueResponse, error) {\n\treturn e.describeQueue(ctx, persistence.HistoryTaskCategoryTimer, clusterName)\n}\n\nfunc (e *historyEngineImpl) describeQueue(\n\tctx context.Context,\n\tcategory persistence.HistoryTaskCategory,\n\tclusterName string,\n) (*types.DescribeQueueResponse, error) {\n\tqueueProcessor, ok := e.queueProcessors[category]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"queue processor not found for category %v\", category)\n\t}\n\tresp, err := queueProcessor.HandleAction(ctx, clusterName, queue.NewGetStateAction())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tserializedStates := make([]string, 0, len(resp.GetStateActionResult.States))\n\tfor _, state := range resp.GetStateActionResult.States {\n\t\tserializedStates = append(serializedStates, e.serializeQueueState(state))\n\t}\n\treturn &types.DescribeQueueResponse{\n\t\tProcessingQueueStates: serializedStates,\n\t}, nil\n}\n\nfunc (e *historyEngineImpl) serializeQueueState(\n\tstate queue.ProcessingQueueState,\n) string {\n\treturn fmt.Sprintf(\"%v\", state)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/describe_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\n// DescribeWorkflowExecution returns information about the specified workflow execution.\nfunc (e *historyEngineImpl) DescribeWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryDescribeWorkflowExecutionRequest,\n) (retResp *types.DescribeWorkflowExecutionResponse, retError error) {\n\tif err := validateDescribeWorkflowExecutionRequest(request); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID := request.DomainUUID\n\twfExecution := *request.Request.Execution\n\n\twfContext, release, err0 := e.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, wfExecution)\n\tif err0 != nil {\n\t\treturn nil, err0\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err1 := wfContext.LoadWorkflowExecution(ctx)\n\tif err1 != nil {\n\t\treturn nil, err1\n\t}\n\n\t// If history is corrupted, return an error to the end user\n\tif corrupted, err := e.checkForHistoryCorruptions(ctx, mutableState); err != nil {\n\t\treturn nil, err\n\t} else if corrupted {\n\t\treturn nil, &types.EntityNotExistsError{Message: \"Workflow execution corrupted.\"}\n\t}\n\n\tdomainCache := e.shard.GetDomainCache()\n\tresult, err := createDescribeWorkflowExecutionResponse(ctx, mutableState, domainCache)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn result, nil\n}\n\n// validateDescribeWorkflowExecutionRequest validates the input request\nfunc validateDescribeWorkflowExecutionRequest(request *types.HistoryDescribeWorkflowExecutionRequest) error {\n\treturn common.ValidateDomainUUID(request.DomainUUID)\n}\n\nfunc createDescribeWorkflowExecutionResponse(ctx context.Context, mutableState execution.MutableState, domainCache cache.DomainCache) (*types.DescribeWorkflowExecutionResponse, error) {\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionConfiguration, err := mapWorkflowExecutionConfiguration(executionInfo)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresult := &types.DescribeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: executionConfiguration,\n\t}\n\n\t// TODO: we need to consider adding execution time to mutable state\n\t// For now execution time will be calculated based on start time and cron schedule/retry policy\n\t// each time DescribeWorkflowExecution is called.\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar completionEvent *types.HistoryEvent\n\tif executionInfo.State == persistence.WorkflowStateCompleted {\n\t\tcompletionEvent, err = mutableState.GetCompletionEvent(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\thistoryLength := mutableState.GetNextEventID() - constants.FirstEventID\n\tworkflowExecutionInfo, err := mapWorkflowExecutionInfo(executionInfo, startEvent, domainCache, historyLength, completionEvent)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresult.WorkflowExecutionInfo = workflowExecutionInfo\n\n\tpendingActivityInfos := mutableState.GetPendingActivityInfos()\n\tfor _, ai := range pendingActivityInfos {\n\t\tscheduledEvent, err := mutableState.GetActivityScheduledEvent(ctx, ai.ScheduleID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tp := mapPendingActivityInfo(ai, scheduledEvent)\n\t\tresult.PendingActivities = append(result.PendingActivities, p)\n\t}\n\n\tchildExecutions := mutableState.GetPendingChildExecutionInfos()\n\tdomainEntry := mutableState.GetDomainEntry()\n\tfor _, childExecution := range childExecutions {\n\t\tpendingChild, err := mapPendingChildExecutionInfo(childExecution, domainEntry, domainCache)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult.PendingChildren = append(result.PendingChildren, pendingChild)\n\t}\n\n\tif di, ok := mutableState.GetPendingDecision(); ok {\n\t\tresult.PendingDecision = mapPendingDecisionInfo(di)\n\t}\n\n\treturn result, nil\n}\n\nfunc mapWorkflowExecutionConfiguration(executionInfo *persistence.WorkflowExecutionInfo) (*types.WorkflowExecutionConfiguration, error) {\n\treturn &types.WorkflowExecutionConfiguration{\n\t\tTaskList:                            mapDecisionInfoToTaskList(executionInfo),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(executionInfo.WorkflowTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(executionInfo.DecisionStartToCloseTimeout),\n\t}, nil\n}\n\nfunc mapWorkflowExecutionInfo(executionInfo *persistence.WorkflowExecutionInfo, startEvent *types.HistoryEvent, domainCache cache.DomainCache, historyLength int64, completionEvent *types.HistoryEvent) (*types.WorkflowExecutionInfo, error) {\n\tresult := &types.WorkflowExecutionInfo{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskList:                     mapDecisionInfoToTaskList(executionInfo),\n\t\tType:                         &types.WorkflowType{Name: executionInfo.WorkflowTypeName},\n\t\tStartTime:                    common.Int64Ptr(executionInfo.StartTimestamp.UnixNano()),\n\t\tHistoryLength:                historyLength,\n\t\tAutoResetPoints:              executionInfo.AutoResetPoints,\n\t\tMemo:                         &types.Memo{Fields: executionInfo.CopyMemo()},\n\t\tIsCron:                       len(executionInfo.CronSchedule) > 0,\n\t\tUpdateTime:                   common.Int64Ptr(executionInfo.LastUpdatedTimestamp.UnixNano()),\n\t\tSearchAttributes:             &types.SearchAttributes{IndexedFields: executionInfo.CopySearchAttributes()},\n\t\tPartitionConfig:              executionInfo.CopyPartitionConfig(),\n\t\tCronOverlapPolicy:            &executionInfo.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy: executionInfo.ActiveClusterSelectionPolicy,\n\t}\n\n\tbackoffDuration := time.Duration(startEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstDecisionTaskBackoffSeconds()) * time.Second\n\tresult.ExecutionTime = common.Int64Ptr(result.GetStartTime() + backoffDuration.Nanoseconds())\n\n\tif executionInfo.ParentRunID != \"\" {\n\t\tresult.ParentExecution = &types.WorkflowExecution{\n\t\t\tWorkflowID: executionInfo.ParentWorkflowID,\n\t\t\tRunID:      executionInfo.ParentRunID,\n\t\t}\n\t\tresult.ParentDomainID = common.StringPtr(executionInfo.ParentDomainID)\n\t\tresult.ParentInitiatedID = common.Int64Ptr(executionInfo.InitiatedID)\n\t\tparentDomain, err := domainCache.GetDomainName(executionInfo.ParentDomainID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult.ParentDomain = common.StringPtr(parentDomain)\n\t}\n\n\tif executionInfo.State == persistence.WorkflowStateCompleted {\n\t\tresult.CloseStatus = persistence.ToInternalWorkflowExecutionCloseStatus(executionInfo.CloseStatus)\n\t\tif completionEvent != nil {\n\t\t\tresult.CloseTime = common.Int64Ptr(completionEvent.GetTimestamp())\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\nfunc mapPendingActivityInfo(ai *persistence.ActivityInfo, activityScheduledEvent *types.HistoryEvent) *types.PendingActivityInfo {\n\tp := &types.PendingActivityInfo{\n\t\tActivityID: ai.ActivityID,\n\t\tScheduleID: ai.ScheduleID,\n\t}\n\n\tstate := types.PendingActivityStateScheduled\n\tif ai.CancelRequested {\n\t\tstate = types.PendingActivityStateCancelRequested\n\t} else if ai.StartedID != constants.EmptyEventID {\n\t\tstate = types.PendingActivityStateStarted\n\t}\n\tp.State = &state\n\n\tlastHeartbeatUnixNano := ai.LastHeartBeatUpdatedTime.UnixNano()\n\tif lastHeartbeatUnixNano > 0 {\n\t\tp.LastHeartbeatTimestamp = common.Int64Ptr(lastHeartbeatUnixNano)\n\t\tp.HeartbeatDetails = ai.Details\n\t}\n\n\tp.ActivityType = activityScheduledEvent.ActivityTaskScheduledEventAttributes.ActivityType\n\tif state == types.PendingActivityStateScheduled {\n\t\tp.ScheduledTimestamp = common.Int64Ptr(ai.ScheduledTime.UnixNano())\n\t} else {\n\t\tp.LastStartedTimestamp = common.Int64Ptr(ai.StartedTime.UnixNano())\n\t}\n\n\tif ai.HasRetryPolicy {\n\t\tp.Attempt = ai.Attempt\n\t\tif !ai.ExpirationTime.IsZero() {\n\t\t\tp.ExpirationTimestamp = common.Int64Ptr(ai.ExpirationTime.UnixNano())\n\t\t}\n\t\tif ai.MaximumAttempts != 0 {\n\t\t\tp.MaximumAttempts = ai.MaximumAttempts\n\t\t}\n\t\tif ai.LastFailureReason != \"\" {\n\t\t\tp.LastFailureReason = common.StringPtr(ai.LastFailureReason)\n\t\t\tp.LastFailureDetails = ai.LastFailureDetails\n\t\t}\n\t\tif ai.LastWorkerIdentity != \"\" {\n\t\t\tp.LastWorkerIdentity = ai.LastWorkerIdentity\n\t\t}\n\t\tif ai.StartedIdentity != \"\" {\n\t\t\tp.StartedWorkerIdentity = ai.StartedIdentity\n\t\t}\n\t}\n\n\treturn p\n}\n\nfunc mapPendingChildExecutionInfo(childExecution *persistence.ChildExecutionInfo, domainEntry *cache.DomainCacheEntry, domainCache cache.DomainCache) (*types.PendingChildExecutionInfo, error) {\n\tchildDomainName, err := execution.GetChildExecutionDomainName(\n\t\tchildExecution,\n\t\tdomainCache,\n\t\tdomainEntry,\n\t)\n\tif err != nil {\n\t\tif !common.IsEntityNotExistsError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\t// child domain already deleted, instead of failing the request,\n\t\t// return domainID instead since this field is only for information purpose\n\t\tchildDomainName = childExecution.DomainID\n\t}\n\treturn &types.PendingChildExecutionInfo{\n\t\tDomain:            childDomainName,\n\t\tWorkflowID:        childExecution.StartedWorkflowID,\n\t\tRunID:             childExecution.StartedRunID,\n\t\tWorkflowTypeName:  childExecution.WorkflowTypeName,\n\t\tInitiatedID:       childExecution.InitiatedID,\n\t\tParentClosePolicy: &childExecution.ParentClosePolicy,\n\t}, nil\n}\n\nfunc mapPendingDecisionInfo(di *execution.DecisionInfo) *types.PendingDecisionInfo {\n\tpendingDecision := &types.PendingDecisionInfo{\n\t\tState:                      types.PendingDecisionStateScheduled.Ptr(),\n\t\tScheduledTimestamp:         common.Int64Ptr(di.ScheduledTimestamp),\n\t\tAttempt:                    di.Attempt,\n\t\tOriginalScheduledTimestamp: common.Int64Ptr(di.OriginalScheduledTimestamp),\n\t\tScheduleID:                 di.ScheduleID,\n\t}\n\tif di.StartedID != constants.EmptyEventID {\n\t\tpendingDecision.State = types.PendingDecisionStateStarted.Ptr()\n\t\tpendingDecision.StartedTimestamp = common.Int64Ptr(di.StartedTimestamp)\n\t}\n\treturn pendingDecision\n}\n\nfunc mapDecisionInfoToTaskList(executionInfo *persistence.WorkflowExecutionInfo) *types.TaskList {\n\treturn &types.TaskList{\n\t\tName: executionInfo.TaskList,\n\t\tKind: executionInfo.TaskListKind.Ptr(),\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/describe_workflow_execution_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\thistoryConstants \"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestDescribeWorkflowExecution(t *testing.T) {\n\ttestDomainUUID := uuid.New()\n\ttestCases := []struct {\n\t\tname                  string\n\t\tsetupMocks            func(*execution.MockCache, *execution.MockContext, *execution.MockMutableState, *shard.MockContext, string)\n\t\tenableCorruptionCheck bool\n\t\texpectError           bool\n\t\terrorContains         string\n\t}{\n\t\t{\n\t\t\tname: \"Success - happy path\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\twfExecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, wfExecution).Return(mockContext, func(error) {}, nil)\n\t\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mockMutableState, nil)\n\n\t\t\t\t// Mock for checkForHistoryCorruptions (GetDomainEntry() call)\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: \"test-domain-name\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t\"test-cluster\",\n\t\t\t\t)\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\n\t\t\t\t// Mock for createDescribeWorkflowExecutionResponse\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(cache.NewMockDomainCache(gomock.NewController(t)))\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t})\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(&types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\t\t\t\tmockMutableState.EXPECT().GetPendingDecision().Return(nil, false)\n\t\t\t},\n\t\t\tenableCorruptionCheck: false,\n\t\t\texpectError:           false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - execution cache GetOrCreateWorkflowExecution fails\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\texecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, execution).Return(nil, nil, &types.InternalServiceError{Message: \"Cache error\"})\n\t\t\t},\n\t\t\tenableCorruptionCheck: false,\n\t\t\texpectError:           true,\n\t\t\terrorContains:         \"Cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - LoadWorkflowExecution fails\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\twfExecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, wfExecution).Return(mockContext, func(error) {}, nil)\n\t\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"Load error\"})\n\t\t\t},\n\t\t\tenableCorruptionCheck: false,\n\t\t\texpectError:           true,\n\t\t\terrorContains:         \"Load error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - createDescribeWorkflowExecutionResponse fails\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\twfExecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, wfExecution).Return(mockContext, func(error) {}, nil)\n\t\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mockMutableState, nil)\n\n\t\t\t\t// Mock for checkForHistoryCorruptions (GetDomainEntry() call)\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: \"test-domain-name\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t\"test-cluster\",\n\t\t\t\t)\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\n\t\t\t\t// Mock for createDescribeWorkflowExecutionResponse failure\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(cache.NewMockDomainCache(gomock.NewController(t)))\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"Start event error\"})\n\t\t\t},\n\t\t\tenableCorruptionCheck: false,\n\t\t\texpectError:           true,\n\t\t\terrorContains:         \"Start event error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - history corruption detected (missing start event)\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\twfExecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, wfExecution).Return(mockContext, func(error) {}, nil)\n\t\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mockMutableState, nil)\n\n\t\t\t\t// Mock for checkForHistoryCorruptions\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: \"test-domain-name\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t\"test-cluster\",\n\t\t\t\t)\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\n\t\t\t\t// This will trigger corruption detection\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(nil, execution.ErrMissingWorkflowStartEvent)\n\n\t\t\t\t// Mock GetExecutionInfo() call for corruption marking\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID:       \"test-wf\",\n\t\t\t\t\tRunID:            \"test-run\",\n\t\t\t\t\tWorkflowTypeName: \"test-type\",\n\t\t\t\t})\n\t\t\t},\n\t\t\tenableCorruptionCheck: true,\n\t\t\texpectError:           true,\n\t\t\terrorContains:         \"Workflow execution corrupted.\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - history corruption check fails with non-start-event error\",\n\t\t\tsetupMocks: func(mockExecutionCache *execution.MockCache, mockContext *execution.MockContext, mockMutableState *execution.MockMutableState, mockShard *shard.MockContext, domainUUID string) {\n\t\t\t\twfExecution := types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"}\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), domainUUID, wfExecution).Return(mockContext, func(error) {}, nil)\n\t\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mockMutableState, nil)\n\n\t\t\t\t// Mock for checkForHistoryCorruptions\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: \"test-domain-name\"},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t\"test-cluster\",\n\t\t\t\t)\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\n\t\t\t\t// This will trigger corruption check failure with different error\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"Database error\"})\n\n\t\t\t\t// Mock GetExecutionInfo() call for corruption marking\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID:       \"test-wf\",\n\t\t\t\t\tRunID:            \"test-run\",\n\t\t\t\t\tWorkflowTypeName: \"test-type\",\n\t\t\t\t})\n\t\t\t},\n\t\t\tenableCorruptionCheck: true,\n\t\t\texpectError:           true,\n\t\t\terrorContains:         \"Database error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockExecutionCache := execution.NewMockCache(ctrl)\n\t\t\tmockContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\n\t\t\ttc.setupMocks(mockExecutionCache, mockContext, mockMutableState, mockShard, testDomainUUID)\n\n\t\t\t// Set up config with explicit corruption check setting\n\t\t\tcfg := &config.Config{}\n\t\t\tcfg.EnableHistoryCorruptionCheck = func(domain string) bool {\n\t\t\t\treturn tc.enableCorruptionCheck\n\t\t\t}\n\n\t\t\tengine := &historyEngineImpl{\n\t\t\t\texecutionCache: mockExecutionCache,\n\t\t\t\tshard:          mockShard,\n\t\t\t\tconfig:         cfg,\n\t\t\t\tlogger:         testlogger.New(t),\n\t\t\t}\n\n\t\t\tresult, err := engine.DescribeWorkflowExecution(ctx.Background(), &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: testDomainUUID,\n\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain:    \"test-domain-name\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: \"test-wf\", RunID: \"test-run\"},\n\t\t\t\t},\n\t\t\t})\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.errorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tc.errorContains)\n\t\t\t\t}\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateDescribeWorkflowExecutionResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tsetupMocks    func(*execution.MockMutableState, *cache.MockDomainCache)\n\t\texpectError   bool\n\t\terrorContains string\n\t\tverifyResult  func(*testing.T, *types.DescribeWorkflowExecutionResponse)\n\t}{\n\t\t{\n\t\t\tname: \"Success - no pending activities, children, or decisions\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:                    \"test-domain-id\",\n\t\t\t\t\tWorkflowID:                  \"test-workflow-id\",\n\t\t\t\t\tRunID:                       \"test-run-id\",\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tWorkflowTypeName:            \"test-workflow-type\",\n\t\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(&cache.DomainCacheEntry{})\n\t\t\t\tmockMutableState.EXPECT().GetPendingDecision().Return(nil, false)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResult: func(t *testing.T, result *types.DescribeWorkflowExecutionResponse) {\n\t\t\t\tassert.NotNil(t, result.ExecutionConfiguration)\n\t\t\t\tassert.NotNil(t, result.WorkflowExecutionInfo)\n\t\t\t\tassert.Empty(t, result.PendingActivities)\n\t\t\t\tassert.Empty(t, result.PendingChildren)\n\t\t\t\tassert.Nil(t, result.PendingDecision)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - GetStartEvent fails\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"Failed to get start event\"})\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"Failed to get start event\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - GetCompletionEvent fails for completed workflow\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tState:                       persistence.WorkflowStateCompleted,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetCompletionEvent(gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"Failed to get completion event\"})\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"Failed to get completion event\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mapWorkflowExecutionInfo fails due to domain cache error\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tParentDomainID:              \"parent-domain-id\",\n\t\t\t\t\tParentWorkflowID:            \"parent-workflow-id\",\n\t\t\t\t\tParentRunID:                 \"parent-run-id\",\n\t\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"parent-domain-id\").Return(\"\", &types.InternalServiceError{Message: \"Domain cache error\"})\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"Domain cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - GetActivityScheduledEvent fails\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tactivityInfo := &persistence.ActivityInfo{\n\t\t\t\t\tScheduleID: 123,\n\t\t\t\t\tActivityID: \"test-activity\",\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{123: activityInfo})\n\t\t\t\tmockMutableState.EXPECT().GetActivityScheduledEvent(gomock.Any(), int64(123)).Return(nil, &types.InternalServiceError{Message: \"Failed to get activity scheduled event\"})\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"Failed to get activity scheduled event\",\n\t\t},\n\t\t{\n\t\t\tname: \"Error - mapPendingChildExecutionInfo fails\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tchildExecutionInfo := &persistence.ChildExecutionInfo{\n\t\t\t\t\tInitiatedID:       456,\n\t\t\t\t\tDomainID:          \"child-domain-id\",\n\t\t\t\t\tStartedWorkflowID: \"child-workflow-id\",\n\t\t\t\t\tStartedRunID:      \"child-run-id\",\n\t\t\t\t\tWorkflowTypeName:  \"child-workflow-type\",\n\t\t\t\t}\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{})\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{456: childExecutionInfo})\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"child-domain-id\").Return(\"\", &types.InternalServiceError{Message: \"Child domain cache error\"})\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\terrorContains: \"Child domain cache error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Success - with pending activities, children, and decision\",\n\t\t\tsetupMocks: func(mockMutableState *execution.MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t\t\tState:                       persistence.WorkflowStateRunning,\n\t\t\t\t}\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tactivityInfo := &persistence.ActivityInfo{\n\t\t\t\t\tScheduleID: 123,\n\t\t\t\t\tActivityID: \"test-activity\",\n\t\t\t\t}\n\t\t\t\tscheduledEvent := &types.HistoryEvent{\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityType: &types.ActivityType{Name: \"test-activity-type\"},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tchildExecutionInfo := &persistence.ChildExecutionInfo{\n\t\t\t\t\tInitiatedID:       456,\n\t\t\t\t\tDomainID:          \"child-domain-id\",\n\t\t\t\t\tStartedWorkflowID: \"child-workflow-id\",\n\t\t\t\t\tStartedRunID:      \"child-run-id\",\n\t\t\t\t\tWorkflowTypeName:  \"child-workflow-type\",\n\t\t\t\t}\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 789,\n\t\t\t\t\tStartedID:  constants.EmptyEventID,\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo)\n\t\t\t\tmockMutableState.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{123: activityInfo})\n\t\t\t\tmockMutableState.EXPECT().GetActivityScheduledEvent(gomock.Any(), int64(123)).Return(scheduledEvent, nil)\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{456: childExecutionInfo})\n\t\t\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"child-domain-id\").Return(\"child-domain-name\", nil)\n\t\t\t\tmockMutableState.EXPECT().GetPendingDecision().Return(decisionInfo, true)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tverifyResult: func(t *testing.T, result *types.DescribeWorkflowExecutionResponse) {\n\t\t\t\tassert.NotNil(t, result.ExecutionConfiguration)\n\t\t\t\tassert.NotNil(t, result.WorkflowExecutionInfo)\n\t\t\t\tassert.Len(t, result.PendingActivities, 1)\n\t\t\t\tassert.Len(t, result.PendingChildren, 1)\n\t\t\t\tassert.NotNil(t, result.PendingDecision)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\ttc.setupMocks(mockMutableState, mockDomainCache)\n\n\t\t\tresult, err := createDescribeWorkflowExecutionResponse(ctx.Background(), mockMutableState, mockDomainCache)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.errorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tc.errorContains)\n\t\t\t\t}\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t\tif tc.verifyResult != nil {\n\t\t\t\t\ttc.verifyResult(t, result)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMapWorkflowExecutionConfiguration(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\texpected      *types.WorkflowExecutionConfiguration\n\t}{\n\t\t{\n\t\t\tname: \"Success - all fields present\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tTaskList:                    \"test-task-list\",\n\t\t\t\tTaskListKind:                types.TaskListKindSticky,\n\t\t\t\tWorkflowTimeout:             1800,\n\t\t\t\tDecisionStartToCloseTimeout: 30,\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionConfiguration{\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1800),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(30),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"Success - zero values\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\texpected: &types.WorkflowExecutionConfiguration{\n\t\t\t\tTaskList:                            &types.TaskList{Name: \"\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(0),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(0),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := mapWorkflowExecutionConfiguration(tc.executionInfo)\n\t\t\tassert.NoError(t, err)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\n// TODO: More test cases with different states of the workflow\nfunc TestMapWorkflowExecutionInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\tstartEvent    *types.HistoryEvent\n\t\tsetupMock     func(*cache.MockDomainCache)\n\t\texpectError   bool\n\t\texpected      *types.WorkflowExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"Success - basic workflow with parent\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tWorkflowID:                   \"test-workflow-id\",\n\t\t\t\tRunID:                        \"test-run-id\",\n\t\t\t\tTaskList:                     \"test-task-list\",\n\t\t\t\tTaskListKind:                 types.TaskListKindNormal,\n\t\t\t\tWorkflowTypeName:             \"test-workflow-type\",\n\t\t\t\tStartTimestamp:               time.Unix(0, 1000000),\n\t\t\t\tLastUpdatedTimestamp:         time.Unix(0, 2000000),\n\t\t\t\tAutoResetPoints:              &types.ResetPoints{Points: []*types.ResetPointInfo{}},\n\t\t\t\tMemo:                         map[string][]byte{\"key\": []byte(\"value\")},\n\t\t\t\tSearchAttributes:             map[string][]byte{\"attr\": []byte(\"val\")},\n\t\t\t\tPartitionConfig:              map[string]string{\"partition\": \"config\"},\n\t\t\t\tCronOverlapPolicy:            historyConstants.CronSkip,\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{},\n\t\t\t\tParentWorkflowID:             \"parent-workflow-id\",\n\t\t\t\tParentRunID:                  \"parent-run-id\",\n\t\t\t\tParentDomainID:               \"parent-domain-id\",\n\t\t\t\tInitiatedID:                  123,\n\t\t\t\tState:                        persistence.WorkflowStateCompleted,\n\t\t\t\tCloseStatus:                  persistence.WorkflowCloseStatusCompleted,\n\t\t\t},\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(10),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"parent-domain-id\").Return(\"parent-domain-name\", nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t\t},\n\t\t\t\tType:                         &types.WorkflowType{Name: \"test-workflow-type\"},\n\t\t\t\tStartTime:                    common.Int64Ptr(1000000),\n\t\t\t\tHistoryLength:                10,\n\t\t\t\tAutoResetPoints:              &types.ResetPoints{Points: []*types.ResetPointInfo{}},\n\t\t\t\tMemo:                         &types.Memo{Fields: map[string][]byte{\"key\": []byte(\"value\")}},\n\t\t\t\tIsCron:                       false,\n\t\t\t\tUpdateTime:                   common.Int64Ptr(2000000),\n\t\t\t\tSearchAttributes:             &types.SearchAttributes{IndexedFields: map[string][]byte{\"attr\": []byte(\"val\")}},\n\t\t\t\tPartitionConfig:              map[string]string{\"partition\": \"config\"},\n\t\t\t\tCronOverlapPolicy:            &historyConstants.CronSkip,\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{},\n\t\t\t\tExecutionTime:                common.Int64Ptr(1000000 + (10 * time.Second).Nanoseconds()),\n\t\t\t\tParentExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"parent-workflow-id\",\n\t\t\t\t\tRunID:      \"parent-run-id\",\n\t\t\t\t},\n\t\t\t\tParentDomainID:    common.StringPtr(\"parent-domain-id\"),\n\t\t\t\tParentInitiatedID: common.Int64Ptr(123),\n\t\t\t\tParentDomain:      common.StringPtr(\"parent-domain-name\"),\n\t\t\t\tCloseStatus:       types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t\tCloseTime:         common.Int64Ptr(12345),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - basic workflow without parent\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tWorkflowID:           \"test-workflow-id\",\n\t\t\t\tRunID:                \"test-run-id\",\n\t\t\t\tTaskList:             \"test-task-list\",\n\t\t\t\tTaskListKind:         types.TaskListKindSticky,\n\t\t\t\tWorkflowTypeName:     \"test-workflow-type\",\n\t\t\t\tStartTimestamp:       time.Unix(0, 1000000),\n\t\t\t\tLastUpdatedTimestamp: time.Unix(0, 2000000),\n\t\t\t\tAutoResetPoints:      &types.ResetPoints{Points: []*types.ResetPointInfo{}},\n\t\t\t\tMemo:                 map[string][]byte{},\n\t\t\t\tSearchAttributes:     map[string][]byte{},\n\t\t\t\tPartitionConfig:      map[string]string{},\n\t\t\t\tCronSchedule:         \"0 0 * * *\",\n\t\t\t\tState:                persistence.WorkflowStateRunning,\n\t\t\t},\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(5),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock:   func(mockDomainCache *cache.MockDomainCache) {},\n\t\t\texpectError: false,\n\t\t\texpected: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t\tType:                         &types.WorkflowType{Name: \"test-workflow-type\"},\n\t\t\t\tStartTime:                    common.Int64Ptr(1000000),\n\t\t\t\tHistoryLength:                10,\n\t\t\t\tAutoResetPoints:              &types.ResetPoints{Points: []*types.ResetPointInfo{}},\n\t\t\t\tMemo:                         &types.Memo{Fields: map[string][]byte{}},\n\t\t\t\tIsCron:                       true,\n\t\t\t\tUpdateTime:                   common.Int64Ptr(2000000),\n\t\t\t\tSearchAttributes:             &types.SearchAttributes{IndexedFields: map[string][]byte{}},\n\t\t\t\tPartitionConfig:              map[string]string{},\n\t\t\t\tCronOverlapPolicy:            &historyConstants.CronSkip,\n\t\t\t\tActiveClusterSelectionPolicy: nil,\n\t\t\t\tExecutionTime:                common.Int64Ptr(1000000 + (5 * time.Second).Nanoseconds()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - parent domain name lookup fails\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tWorkflowID:                   \"test-workflow-id\",\n\t\t\t\tRunID:                        \"test-run-id\",\n\t\t\t\tTaskList:                     \"test-task-list\",\n\t\t\t\tTaskListKind:                 types.TaskListKindNormal,\n\t\t\t\tWorkflowTypeName:             \"test-workflow-type\",\n\t\t\t\tStartTimestamp:               time.Unix(0, 1000000),\n\t\t\t\tLastUpdatedTimestamp:         time.Unix(0, 2000000),\n\t\t\t\tAutoResetPoints:              &types.ResetPoints{Points: []*types.ResetPointInfo{}},\n\t\t\t\tMemo:                         map[string][]byte{\"key\": []byte(\"value\")},\n\t\t\t\tSearchAttributes:             map[string][]byte{\"attr\": []byte(\"val\")},\n\t\t\t\tPartitionConfig:              map[string]string{\"partition\": \"config\"},\n\t\t\t\tCronOverlapPolicy:            historyConstants.CronSkip,\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{},\n\t\t\t\tParentWorkflowID:             \"parent-workflow-id\",\n\t\t\t\tParentRunID:                  \"parent-run-id\",\n\t\t\t\tParentDomainID:               \"parent-domain-id\",\n\t\t\t\tInitiatedID:                  123,\n\t\t\t\tState:                        persistence.WorkflowStateRunning,\n\t\t\t},\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(10),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"parent-domain-id\").Return(\"\", &types.InternalServiceError{Message: \"Domain lookup failed\"})\n\t\t\t},\n\t\t\texpectError: true,\n\t\t\texpected:    nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\ttc.setupMock(mockDomainCache)\n\n\t\t\tresult, err := mapWorkflowExecutionInfo(tc.executionInfo, tc.startEvent, mockDomainCache, 10, &types.HistoryEvent{Timestamp: common.Int64Ptr(12345)})\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// TODO: More test cases for any other branches\nfunc TestMapPendingActivityInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                   string\n\t\tactivityInfo           *persistence.ActivityInfo\n\t\tactivityScheduledEvent *types.HistoryEvent\n\t\texpected               *types.PendingActivityInfo\n\t}{\n\t\t{\n\t\t\tname: \"Success - started activity with retry policy\",\n\t\t\tactivityInfo: &persistence.ActivityInfo{\n\t\t\t\tActivityID:               \"test-activity-id\",\n\t\t\t\tScheduleID:               123,\n\t\t\t\tStartedID:                124,\n\t\t\t\tCancelRequested:          false,\n\t\t\t\tStartedTime:              time.Unix(0, 2001),\n\t\t\t\tLastHeartBeatUpdatedTime: time.Unix(0, 2000),\n\t\t\t\tDetails:                  []byte(\"boom boom\"),\n\t\t\t\tHasRetryPolicy:           true,\n\t\t\t\tAttempt:                  2002,\n\t\t\t\tMaximumAttempts:          2003,\n\t\t\t\tExpirationTime:           time.Unix(0, 2004),\n\t\t\t\tLastFailureReason:        \"failure reason\",\n\t\t\t\tStartedIdentity:          \"StartedWorkerIdentity\",\n\t\t\t\tLastWorkerIdentity:       \"LastWorkerIdentity\",\n\t\t\t\tLastFailureDetails:       []byte(\"failure details\"),\n\t\t\t\tScheduledTime:            time.Unix(0, 1999),\n\t\t\t},\n\t\t\tactivityScheduledEvent: &types.HistoryEvent{\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"test-activity-type\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.PendingActivityInfo{\n\t\t\t\tActivityID:             \"test-activity-id\",\n\t\t\t\tScheduleID:             123,\n\t\t\t\tState:                  types.PendingActivityStateStarted.Ptr(),\n\t\t\t\tHeartbeatDetails:       []byte(\"boom boom\"),\n\t\t\t\tLastHeartbeatTimestamp: common.Int64Ptr(2000),\n\t\t\t\tLastStartedTimestamp:   common.Int64Ptr(2001),\n\t\t\t\tAttempt:                2002,\n\t\t\t\tMaximumAttempts:        2003,\n\t\t\t\tExpirationTimestamp:    common.Int64Ptr(2004),\n\t\t\t\tLastFailureReason:      common.StringPtr(\"failure reason\"),\n\t\t\t\tStartedWorkerIdentity:  \"StartedWorkerIdentity\",\n\t\t\t\tLastWorkerIdentity:     \"LastWorkerIdentity\",\n\t\t\t\tLastFailureDetails:     []byte(\"failure details\"),\n\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\tName: \"test-activity-type\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - scheduled activity without retry policy\",\n\t\t\tactivityInfo: &persistence.ActivityInfo{\n\t\t\t\tActivityID:      \"test-activity-id-2\",\n\t\t\t\tScheduleID:      125,\n\t\t\t\tStartedID:       constants.EmptyEventID,\n\t\t\t\tCancelRequested: false,\n\t\t\t\tScheduledTime:   time.Unix(0, 1999),\n\t\t\t\tHasRetryPolicy:  false,\n\t\t\t},\n\t\t\tactivityScheduledEvent: &types.HistoryEvent{\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"test-activity-type-2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.PendingActivityInfo{\n\t\t\t\tActivityID:         \"test-activity-id-2\",\n\t\t\t\tScheduleID:         125,\n\t\t\t\tState:              types.PendingActivityStateScheduled.Ptr(),\n\t\t\t\tScheduledTimestamp: common.Int64Ptr(1999),\n\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\tName: \"test-activity-type-2\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - cancel requested activity\",\n\t\t\tactivityInfo: &persistence.ActivityInfo{\n\t\t\t\tActivityID:      \"test-activity-id-3\",\n\t\t\t\tScheduleID:      126,\n\t\t\t\tStartedID:       127,\n\t\t\t\tCancelRequested: true,\n\t\t\t\tStartedTime:     time.Unix(0, 2001),\n\t\t\t\tHasRetryPolicy:  false,\n\t\t\t},\n\t\t\tactivityScheduledEvent: &types.HistoryEvent{\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"test-activity-type-3\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.PendingActivityInfo{\n\t\t\t\tActivityID:           \"test-activity-id-3\",\n\t\t\t\tScheduleID:           126,\n\t\t\t\tState:                types.PendingActivityStateCancelRequested.Ptr(),\n\t\t\t\tLastStartedTimestamp: common.Int64Ptr(2001),\n\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\tName: \"test-activity-type-3\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult := mapPendingActivityInfo(tc.activityInfo, tc.activityScheduledEvent)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestMapPendingChildExecutionInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tchildExecution *persistence.ChildExecutionInfo\n\t\tdomainEntry    *cache.DomainCacheEntry\n\t\tsetupMock      func(*cache.MockDomainCache)\n\t\texpectError    bool\n\t\texpected       *types.PendingChildExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"Success - child execution with DomainID\",\n\t\t\tchildExecution: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedID:       123,\n\t\t\t\tStartedWorkflowID: \"child-workflow-id\",\n\t\t\t\tStartedRunID:      \"child-run-id\",\n\t\t\t\tDomainID:          \"child-domain-id\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type\",\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t},\n\t\t\tdomainEntry: &cache.DomainCacheEntry{},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"child-domain-id\").Return(\"child-domain-name\", nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &types.PendingChildExecutionInfo{\n\t\t\t\tDomain:            \"child-domain-name\",\n\t\t\t\tWorkflowID:        \"child-workflow-id\",\n\t\t\t\tRunID:             \"child-run-id\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type\",\n\t\t\t\tInitiatedID:       123,\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - child execution with deprecated domain name\",\n\t\t\tchildExecution: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedID:          456,\n\t\t\t\tStartedWorkflowID:    \"child-workflow-id-2\",\n\t\t\t\tStartedRunID:         \"child-run-id-2\",\n\t\t\t\tDomainID:             \"\", // Empty DomainID\n\t\t\t\tDomainNameDEPRECATED: \"deprecated-domain-name\",\n\t\t\t\tWorkflowTypeName:     \"child-workflow-type-2\",\n\t\t\t\tParentClosePolicy:    types.ParentClosePolicyTerminate,\n\t\t\t},\n\t\t\tdomainEntry: &cache.DomainCacheEntry{},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// No mock expectations needed - uses deprecated domain name directly\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &types.PendingChildExecutionInfo{\n\t\t\t\tDomain:            \"deprecated-domain-name\",\n\t\t\t\tWorkflowID:        \"child-workflow-id-2\",\n\t\t\t\tRunID:             \"child-run-id-2\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type-2\",\n\t\t\t\tInitiatedID:       456,\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyTerminate.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - child execution using parent domain (fallback)\",\n\t\t\tchildExecution: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedID:          789,\n\t\t\t\tStartedWorkflowID:    \"child-workflow-id-3\",\n\t\t\t\tStartedRunID:         \"child-run-id-3\",\n\t\t\t\tDomainID:             \"\", // Empty DomainID\n\t\t\t\tDomainNameDEPRECATED: \"\", // Empty deprecated name\n\t\t\t\tWorkflowTypeName:     \"child-workflow-type-3\",\n\t\t\t\tParentClosePolicy:    types.ParentClosePolicyRequestCancel,\n\t\t\t},\n\t\t\tdomainEntry: cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{\n\t\t\t\t\tID:   \"parent-domain-id\",\n\t\t\t\t\tName: \"parent-domain-fallback\",\n\t\t\t\t},\n\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\"test-cluster\",\n\t\t\t),\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// No mock expectations needed - uses parent domain entry directly\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &types.PendingChildExecutionInfo{\n\t\t\t\tDomain:            \"parent-domain-fallback\",\n\t\t\t\tWorkflowID:        \"child-workflow-id-3\",\n\t\t\t\tRunID:             \"child-run-id-3\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type-3\",\n\t\t\t\tInitiatedID:       789,\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyRequestCancel.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - domain not exists error (uses domainID as fallback)\",\n\t\t\tchildExecution: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedID:       999,\n\t\t\t\tStartedWorkflowID: \"child-workflow-id-4\",\n\t\t\t\tStartedRunID:      \"child-run-id-4\",\n\t\t\t\tDomainID:          \"deleted-domain-id\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type-4\",\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t},\n\t\t\tdomainEntry: &cache.DomainCacheEntry{},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"deleted-domain-id\").Return(\"\", &types.EntityNotExistsError{Message: \"Domain not found\"})\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\texpected: &types.PendingChildExecutionInfo{\n\t\t\t\tDomain:            \"deleted-domain-id\", // Falls back to DomainID\n\t\t\t\tWorkflowID:        \"child-workflow-id-4\",\n\t\t\t\tRunID:             \"child-run-id-4\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type-4\",\n\t\t\t\tInitiatedID:       999,\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon.Ptr(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error - non-EntityNotExists error from domain cache\",\n\t\t\tchildExecution: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedID:       111,\n\t\t\t\tStartedWorkflowID: \"child-workflow-id-5\",\n\t\t\t\tStartedRunID:      \"child-run-id-5\",\n\t\t\t\tDomainID:          \"error-domain-id\",\n\t\t\t\tWorkflowTypeName:  \"child-workflow-type-5\",\n\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon,\n\t\t\t},\n\t\t\tdomainEntry: &cache.DomainCacheEntry{},\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"error-domain-id\").Return(\"\", &types.InternalServiceError{Message: \"Internal error\"})\n\t\t\t},\n\t\t\texpectError: true,\n\t\t\texpected:    nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\ttc.setupMock(mockDomainCache)\n\n\t\t\tresult, err := mapPendingChildExecutionInfo(tc.childExecution, tc.domainEntry, mockDomainCache)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMapPendingDecisionInfo(t *testing.T) {\n\ttestCases := []struct {\n\t\tname         string\n\t\tdecisionInfo *execution.DecisionInfo\n\t\texpected     *types.PendingDecisionInfo\n\t}{\n\t\t{\n\t\t\tname: \"Success - scheduled decision (not started)\",\n\t\t\tdecisionInfo: &execution.DecisionInfo{\n\t\t\t\tVersion:                    1,\n\t\t\t\tScheduleID:                 100,\n\t\t\t\tStartedID:                  constants.EmptyEventID, // Not started\n\t\t\t\tRequestID:                  \"request-id-1\",\n\t\t\t\tDecisionTimeout:            30,\n\t\t\t\tTaskList:                   \"decision-task-list\",\n\t\t\t\tAttempt:                    1,\n\t\t\t\tScheduledTimestamp:         1234567890,\n\t\t\t\tStartedTimestamp:           0, // Not started\n\t\t\t\tOriginalScheduledTimestamp: 1234567890,\n\t\t\t},\n\t\t\texpected: &types.PendingDecisionInfo{\n\t\t\t\tState:                      types.PendingDecisionStateScheduled.Ptr(),\n\t\t\t\tScheduledTimestamp:         common.Int64Ptr(1234567890),\n\t\t\t\tAttempt:                    1,\n\t\t\t\tOriginalScheduledTimestamp: common.Int64Ptr(1234567890),\n\t\t\t\tScheduleID:                 100,\n\t\t\t\tStartedTimestamp:           nil, // Should not be set for scheduled decision\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - started decision\",\n\t\t\tdecisionInfo: &execution.DecisionInfo{\n\t\t\t\tVersion:                    2,\n\t\t\t\tScheduleID:                 200,\n\t\t\t\tStartedID:                  201, // Started\n\t\t\t\tRequestID:                  \"request-id-2\",\n\t\t\t\tDecisionTimeout:            60,\n\t\t\t\tTaskList:                   \"decision-task-list-2\",\n\t\t\t\tAttempt:                    3,\n\t\t\t\tScheduledTimestamp:         2345678901,\n\t\t\t\tStartedTimestamp:           2345678950,\n\t\t\t\tOriginalScheduledTimestamp: 2345678900,\n\t\t\t},\n\t\t\texpected: &types.PendingDecisionInfo{\n\t\t\t\tState:                      types.PendingDecisionStateStarted.Ptr(),\n\t\t\t\tScheduledTimestamp:         common.Int64Ptr(2345678901),\n\t\t\t\tStartedTimestamp:           common.Int64Ptr(2345678950),\n\t\t\t\tAttempt:                    3,\n\t\t\t\tOriginalScheduledTimestamp: common.Int64Ptr(2345678900),\n\t\t\t\tScheduleID:                 200,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - decision with zero values\",\n\t\t\tdecisionInfo: &execution.DecisionInfo{\n\t\t\t\tVersion:                    0,\n\t\t\t\tScheduleID:                 0,\n\t\t\t\tStartedID:                  constants.EmptyEventID,\n\t\t\t\tRequestID:                  \"\",\n\t\t\t\tDecisionTimeout:            0,\n\t\t\t\tTaskList:                   \"\",\n\t\t\t\tAttempt:                    0,\n\t\t\t\tScheduledTimestamp:         0,\n\t\t\t\tStartedTimestamp:           0,\n\t\t\t\tOriginalScheduledTimestamp: 0,\n\t\t\t},\n\t\t\texpected: &types.PendingDecisionInfo{\n\t\t\t\tState:                      types.PendingDecisionStateScheduled.Ptr(),\n\t\t\t\tScheduledTimestamp:         common.Int64Ptr(0),\n\t\t\t\tAttempt:                    0,\n\t\t\t\tOriginalScheduledTimestamp: common.Int64Ptr(0),\n\t\t\t\tScheduleID:                 0,\n\t\t\t\tStartedTimestamp:           nil,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success - decision with high attempt count\",\n\t\t\tdecisionInfo: &execution.DecisionInfo{\n\t\t\t\tVersion:                    5,\n\t\t\t\tScheduleID:                 500,\n\t\t\t\tStartedID:                  501,\n\t\t\t\tRequestID:                  \"request-id-high-attempt\",\n\t\t\t\tDecisionTimeout:            90,\n\t\t\t\tTaskList:                   \"high-attempt-task-list\",\n\t\t\t\tAttempt:                    99,\n\t\t\t\tScheduledTimestamp:         3456789012,\n\t\t\t\tStartedTimestamp:           3456789100,\n\t\t\t\tOriginalScheduledTimestamp: 3456789000,\n\t\t\t},\n\t\t\texpected: &types.PendingDecisionInfo{\n\t\t\t\tState:                      types.PendingDecisionStateStarted.Ptr(),\n\t\t\t\tScheduledTimestamp:         common.Int64Ptr(3456789012),\n\t\t\t\tStartedTimestamp:           common.Int64Ptr(3456789100),\n\t\t\t\tAttempt:                    99,\n\t\t\t\tOriginalScheduledTimestamp: common.Int64Ptr(3456789000),\n\t\t\t\tScheduleID:                 500,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult := mapPendingDecisionInfo(tc.decisionInfo)\n\t\t\tassert.Equal(t, tc.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestValidateDescribeWorkflowExecutionRequest(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\trequest     *types.HistoryDescribeWorkflowExecutionRequest\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname: \"Success - valid UUID\",\n\t\t\trequest: &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: uuid.New(),\n\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - empty UUID\",\n\t\t\trequest: &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - invalid UUID format\",\n\t\t\trequest: &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"not-a-valid-uuid\",\n\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\terr := validateDescribeWorkflowExecutionRequest(tc.request)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/dlq_operations.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *historyEngineImpl) CountDLQMessages(ctx context.Context, forceFetch bool) (map[string]int64, error) {\n\treturn e.replicationDLQHandler.GetMessageCount(ctx, forceFetch)\n}\n\nfunc (e *historyEngineImpl) ReadDLQMessages(\n\tctx context.Context,\n\trequest *types.ReadDLQMessagesRequest,\n) (*types.ReadDLQMessagesResponse, error) {\n\n\ttasks, taskInfo, token, err := e.replicationDLQHandler.ReadMessages(\n\t\tctx,\n\t\trequest.GetSourceCluster(),\n\t\trequest.GetInclusiveEndMessageID(),\n\t\tint(request.GetMaximumPageSize()),\n\t\trequest.GetNextPageToken(),\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.ReadDLQMessagesResponse{\n\t\tType:                 request.GetType().Ptr(),\n\t\tReplicationTasks:     tasks,\n\t\tReplicationTasksInfo: taskInfo,\n\t\tNextPageToken:        token,\n\t}, nil\n}\n\nfunc (e *historyEngineImpl) PurgeDLQMessages(\n\tctx context.Context,\n\trequest *types.PurgeDLQMessagesRequest,\n) error {\n\n\treturn e.replicationDLQHandler.PurgeMessages(\n\t\tctx,\n\t\trequest.GetSourceCluster(),\n\t\trequest.GetInclusiveEndMessageID(),\n\t)\n}\n\nfunc (e *historyEngineImpl) MergeDLQMessages(\n\tctx context.Context,\n\trequest *types.MergeDLQMessagesRequest,\n) (*types.MergeDLQMessagesResponse, error) {\n\n\ttoken, err := e.replicationDLQHandler.MergeMessages(\n\t\tctx,\n\t\trequest.GetSourceCluster(),\n\t\trequest.GetInclusiveEndMessageID(),\n\t\tint(request.GetMaximumPageSize()),\n\t\trequest.GetNextPageToken(),\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.MergeDLQMessagesResponse{\n\t\tNextPageToken: token,\n\t}, nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/get_replication_messages.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *historyEngineImpl) GetReplicationMessages(\n\tctx context.Context,\n\tpollingCluster string,\n\tlastReadMessageID int64,\n) (*types.ReplicationMessages, error) {\n\n\tscope := metrics.HistoryGetReplicationMessagesScope\n\tsw := e.metricsClient.StartTimer(scope, metrics.GetReplicationMessagesForShardLatency)\n\tdefer sw.Stop()\n\n\treplicationMessages, err := e.replicationAckManager.GetTasks(\n\t\tctx,\n\t\tpollingCluster,\n\t\tlastReadMessageID,\n\t)\n\tif err != nil {\n\t\te.logger.Error(\"Failed to retrieve replication messages.\", tag.Error(err))\n\t\treturn nil, err\n\t}\n\n\t// Set cluster status for sync shard info\n\treplicationMessages.SyncShardStatus = &types.SyncShardStatus{\n\t\tTimestamp: common.Int64Ptr(e.timeSource.Now().UnixNano()),\n\t}\n\te.logger.Debug(\"Successfully fetched replication messages.\", tag.Counter(len(replicationMessages.ReplicationTasks)), tag.ClusterName(pollingCluster))\n\n\tif e.logger.DebugOn() {\n\t\tfor _, task := range replicationMessages.ReplicationTasks {\n\t\t\tdata, err := json.Marshal(task)\n\t\t\tif err != nil {\n\t\t\t\te.logger.Error(\"Failed to marshal replication task.\", tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\te.logger.Debugf(\"Replication task: %s\", string(data))\n\t\t}\n\t}\n\treturn replicationMessages, nil\n}\n\nfunc (e *historyEngineImpl) GetDLQReplicationMessages(\n\tctx context.Context,\n\ttaskInfos []*types.ReplicationTaskInfo,\n) ([]*types.ReplicationTask, error) {\n\n\tscope := metrics.HistoryGetDLQReplicationMessagesScope\n\tsw := e.metricsClient.StartTimer(scope, metrics.GetDLQReplicationMessagesLatency)\n\tdefer sw.Stop()\n\n\ttasks := make([]*types.ReplicationTask, 0, len(taskInfos))\n\tfor _, taskInfo := range taskInfos {\n\t\tt, err := convertToReplicationTask(taskInfo)\n\t\tif err != nil {\n\t\t\te.logger.Error(\"Failed to convert replication task.\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t\ttask, err := e.replicationHydrator.Hydrate(ctx, t)\n\t\tif err != nil {\n\t\t\te.logger.Error(\"Failed to fetch DLQ replication messages.\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t\tif task != nil {\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\t}\n\n\treturn tasks, nil\n}\n\nfunc convertToReplicationTask(taskInfo *types.ReplicationTaskInfo) (persistence.Task, error) {\n\tswitch taskInfo.TaskType {\n\tcase persistence.ReplicationTaskTypeHistory:\n\t\treturn &persistence.HistoryReplicationTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   taskInfo.DomainID,\n\t\t\t\tWorkflowID: taskInfo.WorkflowID,\n\t\t\t\tRunID:      taskInfo.RunID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID:  taskInfo.TaskID,\n\t\t\t\tVersion: taskInfo.Version,\n\t\t\t},\n\t\t\tFirstEventID: taskInfo.FirstEventID,\n\t\t\tNextEventID:  taskInfo.NextEventID,\n\t\t}, nil\n\tcase persistence.ReplicationTaskTypeSyncActivity:\n\t\treturn &persistence.SyncActivityTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   taskInfo.DomainID,\n\t\t\t\tWorkflowID: taskInfo.WorkflowID,\n\t\t\t\tRunID:      taskInfo.RunID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID:  taskInfo.TaskID,\n\t\t\t\tVersion: taskInfo.Version,\n\t\t\t},\n\t\t\tScheduledID: taskInfo.ScheduledID,\n\t\t}, nil\n\tcase persistence.ReplicationTaskTypeFailoverMarker:\n\t\treturn &persistence.FailoverMarkerTask{\n\t\t\tDomainID: taskInfo.DomainID,\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID:  taskInfo.TaskID,\n\t\t\t\tVersion: taskInfo.Version,\n\t\t\t},\n\t\t}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unsupported task type: %v\", taskInfo.TaskType)\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/history_engine.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/client/wrappers/retryable\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tcndc \"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/decision\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/failover\"\n\t\"github.com/uber/cadence/service/history/ndc\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/replication\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nconst (\n\tdefaultQueryFirstDecisionTaskWaitTime = time.Second\n\tqueryFirstDecisionTaskCheckInterval   = 200 * time.Millisecond\n\tcontextLockTimeout                    = 500 * time.Millisecond\n\tlongPollCompletionBuffer              = 50 * time.Millisecond\n\n\t// TerminateIfRunningReason reason for terminateIfRunning\n\tTerminateIfRunningReason = \"TerminateIfRunning Policy\"\n\t// TerminateIfRunningDetailsTemplate details template for terminateIfRunning\n\tTerminateIfRunningDetailsTemplate = \"New runID: %s\"\n)\n\nvar (\n\terrDomainDeprecated = &types.BadRequestError{Message: \"Domain is deprecated.\"}\n)\n\ntype historyEngineImpl struct {\n\tcurrentClusterName        string\n\tshard                     shard.Context\n\ttimeSource                clock.TimeSource\n\tdecisionHandler           decision.Handler\n\tclusterMetadata           cluster.Metadata\n\thistoryV2Mgr              persistence.HistoryManager\n\texecutionManager          persistence.ExecutionManager\n\tvisibilityMgr             persistence.VisibilityManager\n\tqueueProcessors           map[persistence.HistoryTaskCategory]queue.Processor\n\tnDCReplicator             ndc.HistoryReplicator\n\tnDCActivityReplicator     ndc.ActivityReplicator\n\thistoryEventNotifier      events.Notifier\n\ttokenSerializer           common.TaskTokenSerializer\n\texecutionCache            execution.Cache\n\tmetricsClient             metrics.Client\n\tlogger                    log.Logger\n\tthrottledLogger           log.Logger\n\tactiveClusterManager      activecluster.Manager\n\tconfig                    *config.Config\n\tarchivalClient            archiver.Client\n\tworkflowResetter          reset.WorkflowResetter\n\treplicationTaskProcessors []replication.TaskProcessor\n\treplicationAckManager     replication.TaskAckManager\n\treplicationTaskStore      *replication.TaskStore\n\treplicationHydrator       replication.TaskHydrator\n\treplicationMetricsEmitter *replication.MetricsEmitterImpl\n\teventsReapplier           ndc.EventsReapplier\n\tmatchingClient            matching.Client\n\trawMatchingClient         matching.Client\n\tclientChecker             client.VersionChecker\n\treplicationDLQHandler     replication.DLQHandler\n\tfailoverMarkerNotifier    failover.MarkerNotifier\n\n\tupdateWithActionFn func(\n\t\tcontext.Context,\n\t\tlog.Logger,\n\t\texecution.Cache,\n\t\tstring,\n\t\ttypes.WorkflowExecution,\n\t\tbool,\n\t\ttime.Time,\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error,\n\t) error\n}\n\nvar (\n\t// FailedWorkflowCloseState is a set of failed workflow close states, used for start workflow policy\n\t// for start workflow execution API\n\tFailedWorkflowCloseState = map[int]bool{\n\t\tpersistence.WorkflowCloseStatusFailed:     true,\n\t\tpersistence.WorkflowCloseStatusCanceled:   true,\n\t\tpersistence.WorkflowCloseStatusTerminated: true,\n\t\tpersistence.WorkflowCloseStatusTimedOut:   true,\n\t}\n)\n\n// NewEngineWithShardContext creates an instance of history engine\nfunc NewEngineWithShardContext(\n\tshard shard.Context,\n\tvisibilityMgr persistence.VisibilityManager,\n\tmatching matching.Client,\n\thistoryEventNotifier events.Notifier,\n\tconfig *config.Config,\n\treplicationTaskFetchers replication.TaskFetchers,\n\trawMatchingClient matching.Client,\n\tfailoverCoordinator failover.Coordinator,\n\tqueueFactories []queue.Factory,\n) engine.Engine {\n\tcurrentClusterName := shard.GetService().GetClusterMetadata().GetCurrentClusterName()\n\n\tlogger := shard.GetLogger()\n\texecutionManager := shard.GetExecutionManager()\n\thistoryV2Manager := shard.GetHistoryManager()\n\texecutionCache := execution.NewCache(shard)\n\tfailoverMarkerNotifier := failover.NewMarkerNotifier(shard, config, failoverCoordinator)\n\treplicationHydrator := replication.NewDeferredTaskHydrator(shard.GetShardID(), historyV2Manager, executionCache, shard.GetDomainCache())\n\treplicationTaskStore := replication.NewTaskStore(\n\t\tshard.GetConfig(),\n\t\tshard.GetClusterMetadata(),\n\t\tshard.GetDomainCache(),\n\t\tshard.GetMetricsClient(),\n\t\tshard.GetLogger(),\n\t\treplicationHydrator,\n\t\tshard.GetReplicationBudgetManager(),\n\t\tshard.GetShardID(),\n\t\tshard.GetTimeSource(),\n\t)\n\treplicationDynamicTaskBatchSizer := replication.NewDynamicTaskBatchSizer(shard.GetShardID(), logger, config, shard.GetMetricsClient())\n\treplicationReader := replication.NewTaskReader(shard.GetShardID(), executionManager)\n\n\thistoryEngImpl := &historyEngineImpl{\n\t\tcurrentClusterName:   currentClusterName,\n\t\tshard:                shard,\n\t\tclusterMetadata:      shard.GetClusterMetadata(),\n\t\ttimeSource:           shard.GetTimeSource(),\n\t\thistoryV2Mgr:         historyV2Manager,\n\t\texecutionManager:     executionManager,\n\t\tvisibilityMgr:        visibilityMgr,\n\t\ttokenSerializer:      common.NewJSONTaskTokenSerializer(),\n\t\texecutionCache:       executionCache,\n\t\tlogger:               logger.WithTags(tag.ComponentHistoryEngine),\n\t\tthrottledLogger:      shard.GetThrottledLogger().WithTags(tag.ComponentHistoryEngine),\n\t\tactiveClusterManager: shard.GetActiveClusterManager(),\n\t\tmetricsClient:        shard.GetMetricsClient(),\n\t\thistoryEventNotifier: historyEventNotifier,\n\t\tconfig:               config,\n\t\tworkflowResetter: reset.NewWorkflowResetter(\n\t\t\tshard,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t),\n\t\tmatchingClient:         matching,\n\t\trawMatchingClient:      rawMatchingClient,\n\t\tclientChecker:          client.NewVersionChecker(),\n\t\tfailoverMarkerNotifier: failoverMarkerNotifier,\n\t\treplicationHydrator:    replicationHydrator,\n\t\treplicationAckManager: replication.NewTaskAckManager(\n\t\t\tshard.GetShardID(),\n\t\t\tshard,\n\t\t\tshard.GetMetricsClient(),\n\t\t\tshard.GetLogger(),\n\t\t\treplicationReader,\n\t\t\treplicationTaskStore,\n\t\t\tshard.GetTimeSource(),\n\t\t\tconfig,\n\t\t\tproto.ReplicationMessagesSize,\n\t\t\treplicationDynamicTaskBatchSizer,\n\t\t),\n\t\treplicationTaskStore: replicationTaskStore,\n\t\treplicationMetricsEmitter: replication.NewMetricsEmitter(\n\t\t\tshard.GetShardID(), shard, replicationReader, shard.GetMetricsClient()),\n\t\tupdateWithActionFn: workflow.UpdateWithAction,\n\t\tqueueProcessors:    make(map[persistence.HistoryTaskCategory]queue.Processor),\n\t}\n\thistoryEngImpl.decisionHandler = decision.NewHandler(\n\t\tshard,\n\t\thistoryEngImpl.executionCache,\n\t\thistoryEngImpl.tokenSerializer,\n\t)\n\tpRetry := persistence.NewPersistenceRetryer(\n\t\tshard.GetExecutionManager(),\n\t\tshard.GetHistoryManager(),\n\t\tcommon.CreatePersistenceRetryPolicy(),\n\t)\n\topenExecutionCheck := invariant.NewConcreteExecutionExists(pRetry, shard.GetDomainCache())\n\n\tfor _, factory := range queueFactories {\n\t\thistoryEngImpl.queueProcessors[factory.Category()] = factory.CreateQueue(\n\t\t\tshard,\n\t\t\texecutionCache,\n\t\t\topenExecutionCheck,\n\t\t)\n\t}\n\n\thistoryEngImpl.eventsReapplier = ndc.NewEventsReapplier(shard.GetMetricsClient(), logger)\n\n\thistoryEngImpl.nDCReplicator = ndc.NewHistoryReplicator(\n\t\tshard,\n\t\texecutionCache,\n\t\thistoryEngImpl.eventsReapplier,\n\t\tlogger,\n\t)\n\thistoryEngImpl.nDCActivityReplicator = ndc.NewActivityReplicator(\n\t\tshard,\n\t\texecutionCache,\n\t\tlogger,\n\t)\n\n\tvar replicationTaskProcessors []replication.TaskProcessor\n\treplicationTaskExecutors := make(map[string]replication.TaskExecutor)\n\t// Intentionally use the raw client to create its own retry policy\n\thistoryRawClient := shard.GetService().GetClientBean().GetHistoryClient()\n\thistoryRetryableClient := retryable.NewHistoryClient(\n\t\thistoryRawClient,\n\t\tcommon.CreateReplicationServiceBusyRetryPolicy(),\n\t\tcommon.IsServiceBusyError,\n\t)\n\tresendFunc := func(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\treturn historyRetryableClient.ReplicateEventsV2(ctx, request)\n\t}\n\tfor _, replicationTaskFetcher := range replicationTaskFetchers.GetFetchers() {\n\t\tsourceCluster := replicationTaskFetcher.GetSourceCluster()\n\t\thistoryResender := cndc.NewHistoryResender(\n\t\t\tshard.GetDomainCache(),\n\t\t\tshard.GetService().GetClientBean(),\n\t\t\tresendFunc,\n\t\t\tnil,\n\t\t\topenExecutionCheck,\n\t\t\tshard.GetLogger().WithTags(tag.ComponentReplicatorQueue, tag.ActiveClusterName(sourceCluster)),\n\t\t)\n\t\treplicationTaskExecutor := replication.NewTaskExecutor(\n\t\t\tsourceCluster,\n\t\t\tshard,\n\t\t\tshard.GetDomainCache(),\n\t\t\thistoryResender,\n\t\t\thistoryEngImpl,\n\t\t\tshard.GetMetricsClient(),\n\t\t\tshard.GetLogger().WithTags(tag.ComponentReplicatorQueue, tag.ActiveClusterName(sourceCluster)),\n\t\t)\n\t\treplicationTaskExecutors[sourceCluster] = replicationTaskExecutor\n\n\t\treplicationTaskProcessor := replication.NewTaskProcessor(\n\t\t\tshard,\n\t\t\thistoryEngImpl,\n\t\t\tconfig,\n\t\t\tshard.GetMetricsClient(),\n\t\t\treplicationTaskFetcher,\n\t\t\treplicationTaskExecutor,\n\t\t\tshard.GetTimeSource(),\n\t\t)\n\t\treplicationTaskProcessors = append(replicationTaskProcessors, replicationTaskProcessor)\n\t}\n\thistoryEngImpl.replicationTaskProcessors = replicationTaskProcessors\n\treplicationMessageHandler := replication.NewDLQHandler(shard, replicationTaskExecutors)\n\thistoryEngImpl.replicationDLQHandler = replicationMessageHandler\n\n\tshard.SetEngine(historyEngImpl)\n\treturn historyEngImpl\n}\n\n// Start will spin up all the components needed to start serving this shard.\n// Make sure all the components are loaded lazily so start can return immediately.  This is important because\n// ShardController calls start sequentially for all the shards for a given host during startup.\nfunc (e *historyEngineImpl) Start() {\n\te.logger.Info(\"History engine state changed\", tag.LifeCycleStarting)\n\tdefer e.logger.Info(\"History engine state changed\", tag.LifeCycleStarted)\n\n\tfor _, processor := range e.queueProcessors {\n\t\tprocessor.Start()\n\t}\n\te.replicationDLQHandler.Start()\n\te.replicationMetricsEmitter.Start()\n\n\t// failover callback will try to create a failover queue processor to scan all inflight tasks\n\t// if domain needs to be failovered. However, in the multicursor queue logic, the scan range\n\t// can't be retrieved before the processor is started. If failover callback is registered\n\t// before queue processor is started, it may result in a deadline as to create the failover queue,\n\t// queue processor need to be started.\n\te.registerDomainFailoverCallback()\n\n\tfor _, replicationTaskProcessor := range e.replicationTaskProcessors {\n\t\treplicationTaskProcessor.Start()\n\t}\n\tif e.config.EnableGracefulFailover() {\n\t\te.failoverMarkerNotifier.Start()\n\t}\n\n}\n\n// Stop the service.\nfunc (e *historyEngineImpl) Stop() {\n\te.logger.Info(\"History engine state changed\", tag.LifeCycleStopping)\n\tdefer e.logger.Info(\"History engine state changed\", tag.LifeCycleStopped)\n\n\tfor _, processor := range e.queueProcessors {\n\t\tprocessor.Stop()\n\t}\n\te.replicationDLQHandler.Stop()\n\te.replicationMetricsEmitter.Stop()\n\n\tfor _, replicationTaskProcessor := range e.replicationTaskProcessors {\n\t\treplicationTaskProcessor.Stop()\n\t}\n\n\te.failoverMarkerNotifier.Stop()\n\n\t// unset the failover callback\n\te.shard.GetDomainCache().UnregisterDomainChangeCallback(createShardNameFromShardID(e.shard.GetShardID()))\n}\n\n// ScheduleDecisionTask schedules a decision if no outstanding decision found\nfunc (e *historyEngineImpl) ScheduleDecisionTask(ctx context.Context, req *types.ScheduleDecisionTaskRequest) error {\n\treturn e.decisionHandler.HandleDecisionTaskScheduled(ctx, req)\n}\n\nfunc (e *historyEngineImpl) ReplicateEventsV2(ctx context.Context, replicateRequest *types.ReplicateEventsV2Request) error {\n\treturn e.nDCReplicator.ApplyEvents(ctx, replicateRequest)\n}\n\nfunc (e *historyEngineImpl) SyncShardStatus(ctx context.Context, request *types.SyncShardStatusRequest) error {\n\n\tclusterName := request.GetSourceCluster()\n\tnow := time.Unix(0, request.GetTimestamp())\n\n\t// here there are 3 main things\n\t// 1. update the view of remote cluster's shard time\n\t// 2. notify the timer gate in the timer queue standby processor\n\t// 3. notify the transfer (essentially a no op, just put it here so it looks symmetric)\n\te.shard.SetCurrentTime(clusterName, now)\n\tfor _, processor := range e.queueProcessors {\n\t\tprocessor.NotifyNewTask(clusterName, &hcommon.NotifyTaskInfo{Tasks: []persistence.Task{}})\n\t}\n\treturn nil\n}\n\nfunc (e *historyEngineImpl) SyncActivity(ctx context.Context, request *types.SyncActivityRequest) (retError error) {\n\n\treturn e.nDCActivityReplicator.SyncActivity(ctx, request)\n}\n\nfunc (e *historyEngineImpl) newDomainNotActiveError(\n\tdomainEntry *cache.DomainCacheEntry,\n\tfailoverVersion int64,\n) error {\n\tactiveClusterName, err := e.shard.GetClusterMetadata().ClusterNameForFailoverVersion(failoverVersion)\n\tif err != nil {\n\t\tactiveClusterName = \"_unknown_\"\n\t}\n\n\treturn domainEntry.NewDomainNotActiveError(e.currentClusterName, activeClusterName)\n}\n\nfunc (e *historyEngineImpl) checkForHistoryCorruptions(ctx context.Context, mutableState execution.MutableState) (bool, error) {\n\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\tif !e.config.EnableHistoryCorruptionCheck(domainName) {\n\t\treturn false, nil\n\t}\n\n\t// Ensure that we can obtain start event. Failing to do so means corrupted history or resurrected mutable state record.\n\t_, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\tinfo := mutableState.GetExecutionInfo()\n\t\t// Mark workflow as corrupted. So that new one can be restarted.\n\t\tinfo.State = persistence.WorkflowStateCorrupted\n\n\t\te.logger.Error(\"history corruption check failed\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.WorkflowID(info.WorkflowID),\n\t\t\ttag.WorkflowRunID(info.RunID),\n\t\t\ttag.WorkflowType(info.WorkflowTypeName),\n\t\t\ttag.Error(err))\n\n\t\tif errors.Is(err, execution.ErrMissingWorkflowStartEvent) {\n\t\t\treturn true, nil\n\t\t}\n\t\treturn false, err\n\t}\n\n\treturn false, nil\n}\n\nfunc getScheduleID(activityID string, mutableState execution.MutableState) (int64, error) {\n\tif activityID == \"\" {\n\t\treturn 0, &types.BadRequestError{Message: \"Neither ActivityID nor ScheduleID is provided\"}\n\t}\n\tactivityInfo, ok := mutableState.GetActivityByActivityID(activityID)\n\tif !ok {\n\t\treturn 0, &types.BadRequestError{Message: \"Cannot locate Activity ScheduleID\"}\n\t}\n\treturn activityInfo.ScheduleID, nil\n}\n\nfunc (e *historyEngineImpl) getActiveDomainByWorkflow(ctx context.Context, domainID, workflowID, runID string) (*cache.DomainCacheEntry, error) {\n\tactiveClusterInfo, err := e.activeClusterManager.GetActiveClusterInfoByWorkflow(ctx, domainID, workflowID, runID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomain, err := e.shard.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif activeClusterInfo.ActiveClusterName == e.clusterMetadata.GetCurrentClusterName() {\n\t\treturn domain, nil\n\t}\n\treturn nil, domain.NewDomainNotActiveError(e.clusterMetadata.GetCurrentClusterName(), activeClusterInfo.ActiveClusterName)\n}\n\nfunc createShardNameFromShardID(shardID int) string {\n\treturn fmt.Sprintf(\"history-engine-%d\", shardID)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/history_engine2_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/decision\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\ntype (\n\tengine2Suite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockTxProcessor    *queue.MockProcessor\n\t\tmockTimerProcessor *queue.MockProcessor\n\t\tmockEventsCache    *events.MockCache\n\t\tmockDomainCache    *cache.MockDomainCache\n\n\t\thistoryEngine        *historyEngineImpl\n\t\tmockExecutionMgr     *mocks.ExecutionManager\n\t\tmockHistoryV2Mgr     *mocks.HistoryV2Manager\n\t\tmockShardManager     *mocks.ShardManager\n\t\tmockActiveClusterMgr *activecluster.MockManager\n\n\t\tconfig *config.Config\n\t\tlogger log.Logger\n\t}\n)\n\nfunc TestEngine2Suite(t *testing.T) {\n\ts := new(engine2Suite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *engine2Suite) SetupSuite() {\n\ts.config = config.NewForTest()\n}\n\nfunc (s *engine2Suite) TearDownSuite() {\n}\n\nfunc (s *engine2Suite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockTxProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockTimerProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockTxProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\ts.mockTimerProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&p.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ts.config,\n\t)\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\ts.mockShardManager = s.mockShard.Resource.ShardMgr\n\ts.mockEventsCache = s.mockShard.MockEventsCache\n\ts.mockActiveClusterMgr = s.mockShard.Resource.ActiveClusterMgr\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID}, &p.DomainConfig{}, \"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainID, nil).AnyTimes()\n\ts.mockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\n\texecutionCache := execution.NewCache(s.mockShard)\n\th := &historyEngineImpl{\n\t\tcurrentClusterName:   s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard:                s.mockShard,\n\t\tclusterMetadata:      s.mockShard.Resource.ClusterMetadata,\n\t\texecutionManager:     s.mockExecutionMgr,\n\t\thistoryV2Mgr:         s.mockHistoryV2Mgr,\n\t\texecutionCache:       executionCache,\n\t\tlogger:               s.logger,\n\t\tthrottledLogger:      s.logger,\n\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\ttokenSerializer:      common.NewJSONTaskTokenSerializer(),\n\t\tconfig:               s.config,\n\t\ttimeSource:           s.mockShard.GetTimeSource(),\n\t\thistoryEventNotifier: events.NewNotifier(clock.NewRealTimeSource(), metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), func(string) int { return 0 }),\n\t\tqueueProcessors: map[p.HistoryTaskCategory]queue.Processor{\n\t\t\tp.HistoryTaskCategoryTransfer: s.mockTxProcessor,\n\t\t\tp.HistoryTaskCategoryTimer:    s.mockTimerProcessor,\n\t\t},\n\t\tactiveClusterManager: s.mockActiveClusterMgr,\n\t}\n\ts.mockShard.SetEngine(h)\n\th.decisionHandler = decision.NewHandler(s.mockShard, h.executionCache, h.tokenSerializer)\n\n\ts.historyEngine = h\n}\n\nfunc (s *engine2Suite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedSuccessStickyExpired() {\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tstickyTl := \"stickyTaskList\"\n\tidentity := \"testIdentity\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\texecutionInfo := msBuilder.GetExecutionInfo()\n\texecutionInfo.StickyTaskList = stickyTl\n\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\trequest := types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &we,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: stickyTl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t}\n\n\texpectedResponse := types.RecordDecisionTaskStartedResponse{}\n\texpectedResponse.WorkflowType = msBuilder.GetWorkflowType()\n\texecutionInfo = msBuilder.GetExecutionInfo()\n\tif executionInfo.LastProcessedEvent != commonconstants.EmptyEventID {\n\t\texpectedResponse.PreviousStartedEventID = common.Int64Ptr(executionInfo.LastProcessedEvent)\n\t}\n\texpectedResponse.ScheduledEventID = di.ScheduleID\n\texpectedResponse.StartedEventID = di.ScheduleID + 1\n\texpectedResponse.StickyExecutionEnabled = false\n\texpectedResponse.NextEventID = msBuilder.GetNextEventID() + 1\n\texpectedResponse.Attempt = di.Attempt\n\texpectedResponse.WorkflowExecutionTaskList = &types.TaskList{\n\t\tName: executionInfo.TaskList,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\texpectedResponse.BranchToken, _ = msBuilder.GetCurrentBranchToken()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &request)\n\ts.Nil(err)\n\ts.NotNil(response)\n\texpectedResponse.StartedTimestamp = response.StartedTimestamp\n\texpectedResponse.ScheduledTimestamp = common.Int64Ptr(0)\n\tresponse.ScheduledTimestamp = common.Int64Ptr(0)\n\texpectedResponse.Queries = make(map[string]*types.WorkflowQuery)\n\ts.Equal(&expectedResponse, response)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedSuccessStickyEnabled() {\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tstickyTl := \"stickyTaskList\"\n\tidentity := \"testIdentity\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\texecutionInfo := msBuilder.GetExecutionInfo()\n\texecutionInfo.LastUpdatedTimestamp = time.Now()\n\texecutionInfo.StickyTaskList = stickyTl\n\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\trequest := types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &we,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: stickyTl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t}\n\n\texpectedResponse := types.RecordDecisionTaskStartedResponse{}\n\texpectedResponse.WorkflowType = msBuilder.GetWorkflowType()\n\texecutionInfo = msBuilder.GetExecutionInfo()\n\tif executionInfo.LastProcessedEvent != commonconstants.EmptyEventID {\n\t\texpectedResponse.PreviousStartedEventID = common.Int64Ptr(executionInfo.LastProcessedEvent)\n\t}\n\texpectedResponse.ScheduledEventID = di.ScheduleID\n\texpectedResponse.StartedEventID = di.ScheduleID + 1\n\texpectedResponse.StickyExecutionEnabled = true\n\texpectedResponse.NextEventID = msBuilder.GetNextEventID() + 1\n\texpectedResponse.Attempt = di.Attempt\n\texpectedResponse.WorkflowExecutionTaskList = &types.TaskList{\n\t\tName: executionInfo.TaskList,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\tcurrentBranchTokken, err := msBuilder.GetCurrentBranchToken()\n\ts.NoError(err)\n\texpectedResponse.BranchToken = currentBranchTokken\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &request)\n\ts.Nil(err)\n\ts.NotNil(response)\n\texpectedResponse.StartedTimestamp = response.StartedTimestamp\n\texpectedResponse.ScheduledTimestamp = common.Int64Ptr(0)\n\tresponse.ScheduledTimestamp = common.Int64Ptr(0)\n\texpectedResponse.Queries = make(map[string]*types.WorkflowQuery)\n\ts.Equal(&expectedResponse, response)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedIfNoExecution() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedIfGetExecutionFailed() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"FAILED\")).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedIfTaskAlreadyStarted() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.IsType(&types.EventAlreadyStartedError{}, err)\n\ts.logger.Error(\"RecordDecisionTaskStarted failed with\", tag.Error(err))\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedIfTaskAlreadyCompleted() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, int64(2), int64(3), nil, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n\ts.logger.Error(\"RecordDecisionTaskStarted failed with\", tag.Error(err))\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedConflictOnUpdate() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(3)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.ConditionFailedError{}).Once()\n\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &p.GetWorkflowExecutionResponse{State: ms2}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(\"wType\", response.WorkflowType.Name)\n\ts.True(response.PreviousStartedEventID == nil)\n\ts.Equal(int64(3), response.StartedEventID)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskRetrySameRequest() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\trequestID := \"testRecordDecisionTaskRetrySameRequestID\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(3)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.ConditionFailedError{}).Once()\n\n\tstartedEventID := test.AddDecisionTaskStartedEventWithRequestID(msBuilder, int64(2), requestID, tl, identity)\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &p.GetWorkflowExecutionResponse{State: ms2}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         requestID,\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(\"wType\", response.WorkflowType.Name)\n\ts.True(response.PreviousStartedEventID == nil)\n\ts.Equal(startedEventID.ID, response.StartedEventID)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskRetryDifferentRequest() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\trequestID := \"testRecordDecisionTaskRetrySameRequestID\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(3)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.ConditionFailedError{}).Once()\n\n\t// Add event.\n\ttest.AddDecisionTaskStartedEventWithRequestID(msBuilder, int64(2), \"some_other_req\", tl, identity)\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &p.GetWorkflowExecutionResponse{State: ms2}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         requestID,\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.IsType(&types.EventAlreadyStartedError{}, err)\n\ts.logger.Info(\"Failed with error\", tag.Error(err))\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskStartedMaxAttemptsExceeded() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(workflow.ConditionalRetryCount + 1)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tfor i := 0; i < workflow.ConditionalRetryCount; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t}\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Times(\n\t\tworkflow.ConditionalRetryCount)\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil,\n\t\t&p.ConditionFailedError{}).Times(workflow.ConditionalRetryCount)\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.NotNil(err)\n\ts.Nil(response)\n\ts.Equal(workflow.ErrMaxAttemptsExceeded, err)\n}\n\nfunc (s *engine2Suite) TestRecordDecisionTaskSuccess() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(3)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\t// load mutable state such that it already exists in memory when respond decision task is called\n\t// this enables us to set query registry on it\n\tctx, release, err := s.historyEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(constants.TestDomainID, workflowExecution)\n\ts.NoError(err)\n\tloadedMS, err := ctx.LoadWorkflowExecution(context.Background())\n\ts.NoError(err)\n\tqr := query.NewRegistry()\n\tid1, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tid2, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tid3, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tloadedMS.SetQueryRegistry(qr)\n\trelease(nil)\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(\"wType\", response.WorkflowType.Name)\n\ts.True(response.PreviousStartedEventID == nil)\n\ts.Equal(int64(3), response.StartedEventID)\n\texpectedQueryMap := map[string]*types.WorkflowQuery{\n\t\tid1: {},\n\t\tid2: {},\n\t\tid3: {},\n\t}\n\ts.Equal(expectedQueryMap, response.Queries)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedIfNoExecution() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\tresponse, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\tif err != nil {\n\t\ts.logger.Error(\"Unexpected Error\", tag.Error(err))\n\t}\n\ts.Nil(response)\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedSuccess() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, int64(2), int64(3), nil, identity)\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\ts.mockEventsCache.EXPECT().GetEvent(\n\t\tgomock.Any(), gomock.Any(), domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID(),\n\t\tdecisionCompletedEvent.ID, scheduledEvent.ID, gomock.Any(),\n\t).Return(scheduledEvent, nil)\n\tresponse, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(scheduledEvent, response.ScheduledEvent)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedResurrected() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID}\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\ttimeSource := clock.NewMockedTimeSource()\n\ts.historyEngine.timeSource = timeSource\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, int64(2), int64(3), nil, identity)\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, \"activity1_id\", \"activity_type1\", tl, []byte(\"input1\"), 100, 10, 1, 5)\n\n\t// Use mutable state snapshot before start/completion of the activity (to indicate resurrected state)\n\tmsSnapshot := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\n\tstartedEvent := test.AddActivityTaskStartedEvent(msBuilder, scheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, scheduledEvent.ID, startedEvent.ID, nil, identity)\n\n\t// Use full history after the activity start/completion\n\thistorySnapshot := msBuilder.GetHistoryBuilder().GetHistory()\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.GetWorkflowExecutionResponse{State: msSnapshot}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&p.ReadHistoryBranchResponse{HistoryEvents: historySnapshot.Events}, nil).Once()\n\n\t// Expect that mutable state will be updated to delete resurrected activity\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.UpdateWorkflowExecutionRequest) bool {\n\t\treturn len(request.UpdateWorkflowMutation.DeleteActivityInfos) == 1\n\t})).Return(&p.UpdateWorkflowExecutionResponse{}, nil).Once()\n\n\t// Ensure enough time passed\n\ttimeSource.Advance(time.Hour)\n\n\t_, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        scheduledEvent.ID,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest:       &types.PollForActivityTaskRequest{TaskList: &types.TaskList{Name: tl}, Identity: identity},\n\t})\n\n\ts.Equal(err, workflow.ErrActivityTaskNotFound)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedStaleState() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(workflow.ConditionalRetryCount + 1)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Times(workflow.ConditionalRetryCount)\n\n\tresponse, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.Error(err)\n\ts.Nil(response)\n\ts.Equal(workflow.ErrMaxAttemptsExceeded, err)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedActivityNotPending() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\n\tresponse, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        3,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\n\ts.Error(err)\n\ts.Nil(response)\n\ts.Equal(workflow.ErrActivityTaskNotFound, err)\n}\n\nfunc (s *engine2Suite) TestRecordActivityTaskStartedActivityAlreadyStarted() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(6)\n\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, true)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, int64(2), int64(3), nil, identity)\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\ts.mockEventsCache.EXPECT().GetEvent(\n\t\tgomock.Any(), gomock.Any(), domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID(),\n\t\tdecisionCompletedEvent.ID, scheduledEvent.ID, gomock.Any(),\n\t).Return(scheduledEvent, nil).Times(1)\n\n\t// start activity\n\tresponse, err := s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(scheduledEvent, response.ScheduledEvent)\n\n\t// another request made with the same scheduleID and same requestID\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\ts.mockEventsCache.EXPECT().GetEvent(\n\t\tgomock.Any(), gomock.Any(), domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID(),\n\t\tdecisionCompletedEvent.ID, scheduledEvent.ID, gomock.Any(),\n\t).Return(scheduledEvent, nil).Times(1)\n\n\tresponse, err = s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(response)\n\ts.Equal(scheduledEvent, response.ScheduledEvent)\n\n\t// another request made with the same scheduleID and different requestID\n\ts.mockEventsCache.EXPECT().GetEvent(\n\t\tgomock.Any(), gomock.Any(), domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID(),\n\t\tdecisionCompletedEvent.ID, scheduledEvent.ID, gomock.Any(),\n\t).Return(scheduledEvent, nil).Times(1)\n\n\tresponse, err = s.historyEngine.RecordActivityTaskStarted(context.Background(), &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tScheduleID:        5,\n\t\tTaskID:            100,\n\t\tRequestID:         \"otherReqId\",\n\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t})\n\ts.Error(err)\n\ts.Nil(response)\n\ts.Equal(&types.EventAlreadyStartedError{Message: \"Activity task already started.\"}, err)\n}\n\nfunc (s *engine2Suite) TestRequestCancelWorkflowExecutionSuccess() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\terr := s.historyEngine.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tIdentity: \"identity\",\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(domainID, workflowExecution)\n\ts.Equal(int64(4), executionBuilder.GetNextEventID())\n}\n\nfunc (s *engine2Suite) TestRequestCancelWorkflowExecutionDuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.DuplicateRequestError{RunID: \"test-run-id\"}).Once()\n\n\terr := s.historyEngine.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tIdentity: \"identity\",\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engine2Suite) TestRequestCancelWorkflowExecutionAlreadyCancelled_Success() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\tcancelRequestID := \"cancelrequestid\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tmsBuilder.GetExecutionInfo().State = p.WorkflowStateCompleted\n\tmsBuilder.GetExecutionInfo().CloseStatus = p.WorkflowCloseStatusCanceled\n\tmsBuilder.GetExecutionInfo().CancelRequested = true\n\tmsBuilder.GetExecutionInfo().CancelRequestID = cancelRequestID\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\n\terr := s.historyEngine.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tIdentity:  \"identity\",\n\t\t\tRequestID: cancelRequestID,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engine2Suite) TestRequestCancelWorkflowExecutionAlreadyCancelled_Fail() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\tcancelRequestID := \"cancelrequestid\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tmsBuilder.GetExecutionInfo().State = p.WorkflowStateCompleted\n\tmsBuilder.GetExecutionInfo().CancelRequested = true\n\tmsBuilder.GetExecutionInfo().CancelRequestID = cancelRequestID\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\n\terr := s.historyEngine.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tIdentity:  \"identity\",\n\t\t\tRequestID: cancelRequestID + \"xxx\",\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.WorkflowExecutionAlreadyCompletedError{}, err)\n}\n\nfunc (s *engine2Suite) TestRequestCancelWorkflowExecutionFail() {\n\tdomainID := constants.TestDomainID\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tidentity := \"testIdentity\"\n\ttl := \"testTaskList\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tmsBuilder := s.createExecutionStartedState(workflowExecution, tl, identity, false)\n\tmsBuilder.GetExecutionInfo().State = p.WorkflowStateCompleted\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &p.GetWorkflowExecutionResponse{State: ms1}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\n\terr := s.historyEngine.RequestCancelWorkflowExecution(context.Background(), &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tIdentity: \"identity\",\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.WorkflowExecutionAlreadyCompletedError{}, err)\n}\n\nfunc (s *engine2Suite) createExecutionStartedState(\n\twe types.WorkflowExecution,\n\ttl, identity string,\n\tstartDecision bool,\n) execution.MutableState {\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ts.logger,\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tif startDecision {\n\t\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\t}\n\t_ = msBuilder.SetHistoryTree(we.GetRunID())\n\n\treturn msBuilder\n}\n\n//nolint:unused\n//lint:ignore U1000 for printing within tests\nfunc (s *engine2Suite) printHistory(builder execution.MutableState) string {\n\treturn thrift.FromHistory(builder.GetHistoryBuilder().GetHistory()).String()\n}\n\nfunc (s *engine2Suite) TestRespondDecisionTaskCompletedRecordMarkerDecision() {\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\tmarkerDetails := []byte(\"marker details\")\n\tmarkerName := \"marker name\"\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(3)\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRecordMarker.Ptr(),\n\t\tRecordMarkerDecisionAttributes: &types.RecordMarkerDecisionAttributes{\n\t\t\tMarkerName: markerName,\n\t\t\tDetails:    markerDetails,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\t_, err := s.historyEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: domainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: nil,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(domainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(p.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_BrandNew() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\trequestID := uuid.New()\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\ts.Nil(err)\n\ts.NotNil(resp.RunID)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_BrandNew_DuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(nil, &p.DuplicateRequestError{RunID: \"test-run-id\"}).Once()\n\n\trequestID := uuid.New()\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\ts.NoError(err)\n\ts.Equal(\"test-run-id\", resp.RunID)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_BrandNew_DuplicateRequestError_TypeMismatch() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeSignal, RunID: \"test-run-id\"}).Once()\n\n\trequestID := uuid.New()\n\t_, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_StillRunning_Dedup() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\trequestID := \"requestID\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   requestID,\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateRunning,\n\t\tCloseStatus:      p.WorkflowCloseStatusNone,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}).Once()\n\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.Equal(runID, resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_StillRunning_NonDeDup() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   \"oldRequestID\",\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateRunning,\n\t\tCloseStatus:      p.WorkflowCloseStatusNone,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}).Once()\n\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           \"newRequestID\",\n\t\t},\n\t})\n\tif _, ok := err.(*types.WorkflowExecutionAlreadyStartedError); !ok {\n\t\ts.Fail(\"return err is not *types.WorkflowExecutionAlreadyStartedError\")\n\t}\n\ts.Nil(resp)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_NotRunning_PrevSuccess_DuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\n\t\t\"CreateWorkflowExecution\",\n\t\tmock.Anything,\n\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\treturn request.Mode == p.CreateWorkflowModeBrandNew &&\n\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t}),\n\t).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   \"oldRequestID\",\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateCompleted,\n\t\tCloseStatus:      p.WorkflowCloseStatusCompleted,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}).Once()\n\n\ts.mockExecutionMgr.On(\n\t\t\"CreateWorkflowExecution\",\n\t\tmock.Anything,\n\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\treturn request.Mode == p.CreateWorkflowModeWorkflowIDReuse &&\n\t\t\t\trequest.PreviousRunID == runID &&\n\t\t\t\trequest.PreviousLastWriteVersion == lastWriteVersion &&\n\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t}),\n\t).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeStart, RunID: \"test-run-id\"}).Once()\n\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           \"newRequestID\",\n\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\n\ts.Nil(err)\n\ts.Equal(\"test-run-id\", resp.RunID)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_NotRunning_PrevSuccess_DuplicateRequestError_TypeMismatch() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\n\t\t\"CreateWorkflowExecution\",\n\t\tmock.Anything,\n\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\treturn request.Mode == p.CreateWorkflowModeBrandNew &&\n\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t}),\n\t).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   \"oldRequestID\",\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateCompleted,\n\t\tCloseStatus:      p.WorkflowCloseStatusCompleted,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}).Once()\n\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionMgr.On(\n\t\t\"CreateWorkflowExecution\",\n\t\tmock.Anything,\n\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\treturn request.Mode == p.CreateWorkflowModeWorkflowIDReuse &&\n\t\t\t\trequest.PreviousRunID == runID &&\n\t\t\t\trequest.PreviousLastWriteVersion == lastWriteVersion &&\n\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t}),\n\t).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeSignal, RunID: \"test-run-id\"}).Once()\n\n\t_, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           \"newRequestID\",\n\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_NotRunning_PrevSuccess() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\trunID := \"runID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\n\toptions := []types.WorkflowIDReusePolicy{\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicateFailedOnly,\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicate,\n\t\ttypes.WorkflowIDReusePolicyRejectDuplicate,\n\t}\n\n\texpectedErrs := []bool{true, false, true}\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Times(3)\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Times(2)\n\ts.mockExecutionMgr.On(\n\t\t\"CreateWorkflowExecution\",\n\t\tmock.Anything,\n\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\treturn request.Mode == p.CreateWorkflowModeBrandNew &&\n\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t}),\n\t).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   \"oldRequestID\",\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateCompleted,\n\t\tCloseStatus:      p.WorkflowCloseStatusCompleted,\n\t\tLastWriteVersion: lastWriteVersion,\n\t}).Times(len(expectedErrs))\n\n\tfor index, option := range options {\n\t\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\t\tif !expectedErrs[index] {\n\t\t\ts.mockExecutionMgr.On(\n\t\t\t\t\"CreateWorkflowExecution\",\n\t\t\t\tmock.Anything,\n\t\t\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\t\t\treturn request.Mode == p.CreateWorkflowModeWorkflowIDReuse &&\n\t\t\t\t\t\trequest.PreviousRunID == runID &&\n\t\t\t\t\t\trequest.PreviousLastWriteVersion == lastWriteVersion &&\n\t\t\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t\t\t}),\n\t\t\t).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\t\t}\n\n\t\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\tDomain:                              domainID,\n\t\t\t\tWorkflowID:                          workflowID,\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\t\tIdentity:                            identity,\n\t\t\t\tRequestID:                           \"newRequestID\",\n\t\t\t\tWorkflowIDReusePolicy:               &option,\n\t\t\t},\n\t\t\tPartitionConfig: partitionConfig,\n\t\t})\n\n\t\tif expectedErrs[index] {\n\t\t\tif _, ok := err.(*types.WorkflowExecutionAlreadyStartedError); !ok {\n\t\t\t\ts.Fail(\"return err is not *types.WorkflowExecutionAlreadyStartedError\")\n\t\t\t}\n\t\t\ts.Nil(resp)\n\t\t} else {\n\t\t\ts.Nil(err)\n\t\t\ts.NotNil(resp)\n\t\t}\n\t}\n}\n\nfunc (s *engine2Suite) TestStartWorkflowExecution_NotRunning_PrevFail() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tlastWriteVersion := commonconstants.EmptyVersion\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\n\toptions := []types.WorkflowIDReusePolicy{\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicateFailedOnly,\n\t\ttypes.WorkflowIDReusePolicyAllowDuplicate,\n\t\ttypes.WorkflowIDReusePolicyRejectDuplicate,\n\t}\n\n\texpectedErrs := []bool{false, false, true}\n\n\tcloseStates := []int{\n\t\tp.WorkflowCloseStatusFailed,\n\t\tp.WorkflowCloseStatusCanceled,\n\t\tp.WorkflowCloseStatusTerminated,\n\t\tp.WorkflowCloseStatusTimedOut,\n\t}\n\trunIDs := []string{\"1\", \"2\", \"3\", \"4\"}\n\n\tfor i, closeState := range closeStates {\n\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Times(3)\n\t\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Times(1)\n\t\ts.mockExecutionMgr.On(\n\t\t\t\"CreateWorkflowExecution\",\n\t\t\tmock.Anything,\n\t\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\t\treturn request.Mode == p.CreateWorkflowModeBrandNew &&\n\t\t\t\t\t!request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() &&\n\t\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t\t}),\n\t\t).Return(nil, &p.WorkflowExecutionAlreadyStartedError{\n\t\t\tMsg:              \"random message\",\n\t\t\tStartRequestID:   \"oldRequestID\",\n\t\t\tRunID:            runIDs[i],\n\t\t\tState:            p.WorkflowStateCompleted,\n\t\t\tCloseStatus:      closeState,\n\t\t\tLastWriteVersion: lastWriteVersion,\n\t\t}).Times(len(expectedErrs))\n\n\t\tfor j, option := range options {\n\n\t\t\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil)\n\t\t\tif !expectedErrs[j] {\n\t\t\t\ts.mockExecutionMgr.On(\n\t\t\t\t\t\"CreateWorkflowExecution\",\n\t\t\t\t\tmock.Anything,\n\t\t\t\t\tmock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\t\t\t\t\treturn request.Mode == p.CreateWorkflowModeWorkflowIDReuse &&\n\t\t\t\t\t\t\trequest.PreviousRunID == runIDs[i] &&\n\t\t\t\t\t\t\trequest.PreviousLastWriteVersion == lastWriteVersion &&\n\t\t\t\t\t\t\treflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t\t\t\t\t}),\n\t\t\t\t).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\t\t\t}\n\n\t\t\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: domainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              domainID,\n\t\t\t\t\tWorkflowID:                          workflowID,\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\t\t\tIdentity:                            identity,\n\t\t\t\t\tRequestID:                           \"newRequestID\",\n\t\t\t\t\tWorkflowIDReusePolicy:               &option,\n\t\t\t\t},\n\t\t\t\tPartitionConfig: partitionConfig,\n\t\t\t})\n\n\t\t\tif expectedErrs[j] {\n\t\t\t\tif _, ok := err.(*types.WorkflowExecutionAlreadyStartedError); !ok {\n\t\t\t\t\ts.Fail(\"return err is not *types.WorkflowExecutionAlreadyStartedError\")\n\t\t\t\t}\n\t\t\t\ts.Nil(resp)\n\t\t\t} else {\n\t\t\t\ts.Nil(err)\n\t\t\t\ts.NotNil(resp)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_JustSignal() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tIdentity:   identity,\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.Equal(runID, resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_JustSignal_DuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tIdentity:   identity,\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeSignal, RunID: \"test-run-id\"}).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.Equal(\"test-run-id\", resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_JustSignal_DuplicateRequestError_TypeMismatch() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tIdentity:   identity,\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeStart, RunID: \"test-run-id\"}).Once()\n\n\t_, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_WorkflowNotExist() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tnotExistErr := &types.EntityNotExistsError{Message: \"Workflow not exist\"}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, notExistErr).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_WorkflowNotExist_DuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tnotExistErr := &types.EntityNotExistsError{Message: \"Workflow not exist\"}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, notExistErr).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(nil, &p.DuplicateRequestError{RunID: \"test-run-id\"}).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.Equal(\"test-run-id\", resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_WorkflowNotExist_DuplicateRequestError_TypeMismatch() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tnotExistErr := &types.EntityNotExistsError{Message: \"Workflow not exist\"}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, notExistErr).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(nil, &p.DuplicateRequestError{RequestType: p.WorkflowRequestTypeCancel, RunID: \"test-run-id\"}).Once()\n\n\t_, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Error(err)\n\ts.IsType(&p.DuplicateRequestError{}, err)\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_CreateTimeout() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t}\n\n\tnotExistErr := &types.EntityNotExistsError{Message: \"Workflow not exist\"}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, notExistErr).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.TimeoutError{}).Once()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.True(p.IsTimeoutError(err))\n\ts.NotNil(resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_WorkflowNotRunning() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\tpolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t\tWorkflowIDReusePolicy:               &policy,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = p.WorkflowStateCompleted\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n\ts.NotEqual(runID, resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_Start_DuplicateRequests() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := \"testRequestID\"\n\tpolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t\tWorkflowIDReusePolicy:               &policy,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = p.WorkflowStateCompleted\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\tworkflowAlreadyStartedErr := &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   requestID, // use same requestID\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateRunning,\n\t\tCloseStatus:      p.WorkflowCloseStatusNone,\n\t\tLastWriteVersion: commonconstants.EmptyVersion,\n\t}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, workflowAlreadyStartedErr).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n\ts.Equal(runID, resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_Start_DuplicateRequestError() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := \"testRequestID\"\n\tpolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t\tWorkflowIDReusePolicy:               &policy,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = p.WorkflowStateCompleted\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &p.DuplicateRequestError{RunID: \"test-run-id\"}).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.Equal(\"test-run-id\", resp.GetRunID())\n}\n\nfunc (s *engine2Suite) TestSignalWithStartWorkflowExecution_Start_WorkflowAlreadyStarted() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := \"testRequestID\"\n\tpolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t\tWorkflowIDReusePolicy:               &policy,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = p.WorkflowStateCompleted\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\tworkflowAlreadyStartedErr := &p.WorkflowExecutionAlreadyStartedError{\n\t\tMsg:              \"random message\",\n\t\tStartRequestID:   \"new request ID\",\n\t\tRunID:            runID,\n\t\tState:            p.WorkflowStateRunning,\n\t\tCloseStatus:      p.WorkflowCloseStatusNone,\n\t\tLastWriteVersion: commonconstants.EmptyVersion,\n\t}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, workflowAlreadyStartedErr).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(resp)\n\ts.NotNil(err)\n}\n\nfunc (s *engine2Suite) TestTerminateWorkflowExecution_Success() {\n\n}\n\nfunc (s *engine2Suite) TestNewChildContext() {\n\tctx := context.Background()\n\tchildCtx, childCancel := s.historyEngine.newChildContext(ctx)\n\tdefer childCancel()\n\t_, ok := childCtx.Deadline()\n\ts.True(ok)\n\n\tctx, cancel := context.WithTimeout(ctx, time.Hour)\n\tdefer cancel()\n\tchildCtx, childCancel = s.historyEngine.newChildContext(ctx)\n\tdefer childCancel()\n\tdeadline, ok := childCtx.Deadline()\n\ts.True(ok)\n\ts.True(time.Until(deadline) < 10*time.Minute)\n}\n\nfunc (s *engine2Suite) getBuilder(domainID string, we types.WorkflowExecution) execution.MutableState {\n\tcontext, release, err := s.historyEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\tif err != nil {\n\t\treturn nil\n\t}\n\tdefer release(nil)\n\n\treturn context.GetWorkflowExecution()\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/history_engine3_eventsv2_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/decision\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n)\n\ntype (\n\tengine3Suite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockTxProcessor    *queue.MockProcessor\n\t\tmockTimerProcessor *queue.MockProcessor\n\t\tmockEventsCache    *events.MockCache\n\t\tmockDomainCache    *cache.MockDomainCache\n\n\t\thistoryEngine    *historyEngineImpl\n\t\tmockExecutionMgr *mocks.ExecutionManager\n\t\tmockHistoryV2Mgr *mocks.HistoryV2Manager\n\n\t\tconfig *config.Config\n\t\tlogger log.Logger\n\t}\n)\n\nfunc TestEngine3Suite(t *testing.T) {\n\ts := new(engine3Suite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *engine3Suite) SetupSuite() {\n\ts.config = config.NewForTest()\n}\n\nfunc (s *engine3Suite) TearDownSuite() {\n}\n\nfunc (s *engine3Suite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockTxProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockTimerProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockTxProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\ts.mockTimerProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&p.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ts.config,\n\t)\n\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockEventsCache = s.mockShard.MockEventsCache\n\n\ts.mockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\n\th := &historyEngineImpl{\n\t\tcurrentClusterName:   s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard:                s.mockShard,\n\t\tclusterMetadata:      s.mockShard.Resource.ClusterMetadata,\n\t\texecutionManager:     s.mockExecutionMgr,\n\t\thistoryV2Mgr:         s.mockHistoryV2Mgr,\n\t\texecutionCache:       execution.NewCache(s.mockShard),\n\t\tlogger:               s.logger,\n\t\tthrottledLogger:      s.logger,\n\t\tmetricsClient:        metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\ttokenSerializer:      common.NewJSONTaskTokenSerializer(),\n\t\tconfig:               s.config,\n\t\ttimeSource:           s.mockShard.GetTimeSource(),\n\t\thistoryEventNotifier: events.NewNotifier(clock.NewRealTimeSource(), metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), func(string) int { return 0 }),\n\t\tqueueProcessors: map[p.HistoryTaskCategory]queue.Processor{\n\t\t\tp.HistoryTaskCategoryTransfer: s.mockTxProcessor,\n\t\t\tp.HistoryTaskCategoryTimer:    s.mockTimerProcessor,\n\t\t},\n\t\tactiveClusterManager: s.mockShard.Resource.ActiveClusterMgr,\n\t}\n\ts.mockShard.SetEngine(h)\n\th.decisionHandler = decision.NewHandler(s.mockShard, h.executionCache, h.tokenSerializer)\n\n\ts.historyEngine = h\n}\n\nfunc (s *engine3Suite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *engine3Suite) TestRecordDecisionTaskStartedSuccessStickyEnabled() {\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   testDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(2)\n\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tstickyTl := \"stickyTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\texecutionInfo := msBuilder.GetExecutionInfo()\n\texecutionInfo.LastUpdatedTimestamp = time.Now()\n\texecutionInfo.StickyTaskList = stickyTl\n\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\trequest := types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: &we,\n\t\tScheduleID:        2,\n\t\tTaskID:            100,\n\t\tRequestID:         \"reqId\",\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: stickyTl,\n\t\t\t},\n\t\t\tIdentity: identity,\n\t\t},\n\t}\n\n\texpectedResponse := types.RecordDecisionTaskStartedResponse{}\n\texpectedResponse.WorkflowType = msBuilder.GetWorkflowType()\n\texecutionInfo = msBuilder.GetExecutionInfo()\n\tif executionInfo.LastProcessedEvent != commonconstants.EmptyEventID {\n\t\texpectedResponse.PreviousStartedEventID = common.Int64Ptr(executionInfo.LastProcessedEvent)\n\t}\n\texpectedResponse.ScheduledEventID = di.ScheduleID\n\texpectedResponse.StartedEventID = di.ScheduleID + 1\n\texpectedResponse.StickyExecutionEnabled = true\n\texpectedResponse.NextEventID = msBuilder.GetNextEventID() + 1\n\texpectedResponse.Attempt = di.Attempt\n\texpectedResponse.WorkflowExecutionTaskList = &types.TaskList{\n\t\tName: executionInfo.TaskList,\n\t\tKind: types.TaskListKindNormal.Ptr(),\n\t}\n\texpectedResponse.BranchToken = msBuilder.GetExecutionInfo().BranchToken\n\n\tresponse, err := s.historyEngine.RecordDecisionTaskStarted(context.Background(), &request)\n\ts.Nil(err)\n\ts.NotNil(response)\n\texpectedResponse.StartedTimestamp = response.StartedTimestamp\n\texpectedResponse.ScheduledTimestamp = response.ScheduledTimestamp\n\texpectedResponse.Queries = make(map[string]*types.WorkflowQuery)\n\ts.Equal(&expectedResponse, response)\n}\n\nfunc (s *engine3Suite) TestStartWorkflowExecution_BrandNew() {\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig, partitionConfig)\n\t})).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\trequestID := uuid.New()\n\tresp, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t})\n\ts.Nil(err)\n\ts.NotNil(resp.RunID)\n}\n\nfunc (s *engine3Suite) TestStartWorkflowExecution_DeprecatedDomain() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"workflowID\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\trequestID := uuid.New()\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: domainID, Name: constants.TestDomainName, Status: p.DomainStatusDeprecated}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil)\n\n\t_, err := s.historyEngine.StartWorkflowExecution(context.Background(), &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t})\n\ts.IsType(&types.BadRequestError{}, err)\n}\nfunc (s *engine3Suite) TestSignalWithStartWorkflowExecution_JustSignal() {\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: testDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   testDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tIdentity:   identity,\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.historyEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &p.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &p.GetCurrentExecutionResponse{RunID: runID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &p.MutableStateUpdateSessionStats{},\n\t}, nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.Equal(runID, resp.GetRunID())\n}\n\nfunc (s *engine3Suite) TestSignalWithStartWorkflowExecution_WorkflowNotExist() {\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t\tPartitionConfig: partitionConfig,\n\t}\n\n\tnotExistErr := &types.EntityNotExistsError{Message: \"Workflow not exist\"}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, notExistErr).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&p.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(request *p.CreateWorkflowExecutionRequest) bool {\n\t\treturn !request.NewWorkflowSnapshot.ExecutionInfo.StartTimestamp.IsZero() && reflect.DeepEqual(partitionConfig, request.NewWorkflowSnapshot.ExecutionInfo.PartitionConfig)\n\t})).Return(&p.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\tresp, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.Nil(err)\n\ts.NotNil(resp.GetRunID())\n}\n\nfunc (s *engine3Suite) TestSignalWithStartWorkflowExecution_DeprecatedDomain() {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\tworkflowType := \"workflowType\"\n\ttaskList := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\trequestID := uuid.New()\n\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: domainID, Name: constants.TestDomainName, Status: p.DomainStatusDeprecated}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              domainID,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\t\tIdentity:                            identity,\n\t\t\tSignalName:                          signalName,\n\t\t\tInput:                               input,\n\t\t\tRequestID:                           requestID,\n\t\t},\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil)\n\n\t_, err := s.historyEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *engine3Suite) TestSignalWorkflowExecution_DeprecatedDomain() {\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t},\n\t}\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.historyEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&p.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName, Status: p.DomainStatusDeprecated}, &p.DomainConfig{Retention: 1}, \"\",\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil)\n\n\terr := s.historyEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/history_engine_start_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n)\n\nfunc TestHistoryEngineStartStop(t *testing.T) {\n\tt.Cleanup(func() { goleak.VerifyNone(t) })\n\n\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\n\teft.Engine.Start()\n\teft.Engine.Stop()\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/history_engine_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/api/encoding\"\n\t\"go.uber.org/yarpc/api/transport\"\n\n\thclient \"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcc \"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/decision\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/ndc\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\ntype (\n\tengineSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockShard            *shard.TestContext\n\t\tmockTxProcessor      *queue.MockProcessor\n\t\tmockTimerProcessor   *queue.MockProcessor\n\t\tmockDomainCache      *cache.MockDomainCache\n\t\tmockHistoryClient    *hclient.MockClient\n\t\tmockMatchingClient   *matching.MockClient\n\t\tmockEventsReapplier  *ndc.MockEventsReapplier\n\t\tmockWorkflowResetter *reset.MockWorkflowResetter\n\n\t\tmockHistoryEngine *historyEngineImpl\n\t\tmockExecutionMgr  *mocks.ExecutionManager\n\t\tmockHistoryV2Mgr  *mocks.HistoryV2Manager\n\t\tmockShardManager  *mocks.ShardManager\n\n\t\teventsCache events.Cache\n\t\tconfig      *config.Config\n\t}\n)\n\nfunc TestEngineSuite(t *testing.T) {\n\ts := new(engineSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *engineSuite) SetupSuite() {\n\ts.config = config.NewForTest()\n}\n\nfunc (s *engineSuite) TearDownSuite() {\n}\n\nfunc (s *engineSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockTxProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockTimerProcessor = queue.NewMockProcessor(s.controller)\n\ts.mockEventsReapplier = ndc.NewMockEventsReapplier(s.controller)\n\ts.mockWorkflowResetter = reset.NewMockWorkflowResetter(s.controller)\n\ts.mockTxProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\ts.mockTimerProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ts.config,\n\t)\n\ts.eventsCache = events.NewCache(\n\t\ts.mockShard.GetShardID(),\n\t\ts.mockShard.GetHistoryManager(),\n\t\ts.config,\n\t\ts.mockShard.GetLogger(),\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.mockDomainCache,\n\t)\n\ts.mockShard.SetEventsCache(s.eventsCache)\n\n\ts.mockMatchingClient = s.mockShard.Resource.MatchingClient\n\ts.mockHistoryClient = s.mockShard.Resource.HistoryClient\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\ts.mockShardManager = s.mockShard.Resource.ShardMgr\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestLocalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestActiveActiveDomainID).Return(constants.TestActiveActiveDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestActiveActiveDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\n\thistoryEventNotifier := events.NewNotifier(\n\t\tclock.NewRealTimeSource(),\n\t\ts.mockShard.Resource.MetricsClient,\n\t\tfunc(workflowID string) int {\n\t\t\treturn len(workflowID)\n\t\t},\n\t)\n\n\th := &historyEngineImpl{\n\t\tcurrentClusterName:   s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard:                s.mockShard,\n\t\ttimeSource:           s.mockShard.GetTimeSource(),\n\t\tclusterMetadata:      s.mockShard.Resource.ClusterMetadata,\n\t\texecutionManager:     s.mockExecutionMgr,\n\t\thistoryV2Mgr:         s.mockHistoryV2Mgr,\n\t\texecutionCache:       execution.NewCache(s.mockShard),\n\t\tlogger:               s.mockShard.GetLogger(),\n\t\tmetricsClient:        s.mockShard.GetMetricsClient(),\n\t\ttokenSerializer:      common.NewJSONTaskTokenSerializer(),\n\t\thistoryEventNotifier: historyEventNotifier,\n\t\tconfig:               config.NewForTest(),\n\t\tqueueProcessors: map[persistence.HistoryTaskCategory]queue.Processor{\n\t\t\tpersistence.HistoryTaskCategoryTransfer: s.mockTxProcessor,\n\t\t\tpersistence.HistoryTaskCategoryTimer:    s.mockTimerProcessor,\n\t\t},\n\t\tclientChecker:        cc.NewVersionChecker(),\n\t\teventsReapplier:      s.mockEventsReapplier,\n\t\tworkflowResetter:     s.mockWorkflowResetter,\n\t\tactiveClusterManager: s.mockShard.Resource.ActiveClusterMgr,\n\t}\n\ts.mockShard.SetEngine(h)\n\th.decisionHandler = decision.NewHandler(s.mockShard, h.executionCache, h.tokenSerializer)\n\n\th.historyEventNotifier.Start()\n\n\ts.mockHistoryEngine = h\n}\n\nfunc (s *engineSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n\ts.mockHistoryEngine.historyEventNotifier.Stop()\n}\n\nfunc (s *engineSuite) TestGetMutableStateSync() {\n\tctx := context.Background()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tmsBuilder.SetVersionHistories(&persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t// right now the next event ID is 4\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// test get the next event ID instantly\n\tresponse, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  &workflowExecution,\n\t})\n\ts.Nil(err)\n\ts.Equal(int64(4), response.GetNextEventID())\n}\n\nfunc (s *engineSuite) TestGetMutableState_IntestRunID() {\n\tctx := context.Background()\n\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      \"run-id-not-valid-uuid\",\n\t}\n\n\t_, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  &execution,\n\t})\n\ts.Equal(constants.ErrRunIDNotValid, err)\n}\n\nfunc (s *engineSuite) TestGetMutableState_EmptyRunID() {\n\tctx := context.Background()\n\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\t_, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  &execution,\n\t})\n\ts.Equal(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestGetMutableState_NoVersionHistories() {\n\tctx := context.Background()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\n\t// simulate having no version histories\n\ts.Require().NoError(msBuilder.SetVersionHistories(nil))\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// return immediately, since the expected next event ID appears\n\tmutableState, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tExecution:  &workflowExecution,\n\n\t\t// we request for the specific history version, but there are no history versions for the execution\n\t\tVersionHistoryItem: &types.VersionHistoryItem{\n\t\t\tEventID: 1,\n\t\t\tVersion: 1,\n\t\t},\n\t})\n\ts.ErrorContains(err, \"version histories do not exist\")\n\ts.Nil(mutableState)\n}\n\nfunc (s *engineSuite) TestGetMutableStateLongPoll() {\n\tctx := context.Background()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tmsBuilder.SetVersionHistories(&persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t// right now the next event ID is 4\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// test long poll on next event ID change\n\twaitGroup := &sync.WaitGroup{}\n\twaitGroup.Add(1)\n\tasycWorkflowUpdate := func(delay time.Duration) {\n\t\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t\tScheduleID: 2,\n\t\t})\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t\ttimer := time.NewTimer(delay)\n\n\t\t<-timer.C\n\t\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken: taskToken,\n\t\t\t\tIdentity:  identity,\n\t\t\t},\n\t\t})\n\t\ts.Nil(err)\n\t\twaitGroup.Done()\n\t\t// right now the next event ID is 5\n\t}\n\n\t// return immediately, since the expected next event ID appears\n\tresponse, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID:          constants.TestDomainID,\n\t\tExecution:           &workflowExecution,\n\t\tExpectedNextEventID: 3,\n\t\tVersionHistoryItem: &types.VersionHistoryItem{\n\t\t\tEventID: 1,\n\t\t\tVersion: 1,\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.Equal(int64(4), response.NextEventID)\n\n\t// long poll, new event happen before long poll timeout\n\tgo asycWorkflowUpdate(time.Second * 2)\n\tstart := time.Now()\n\tpollResponse, err := s.mockHistoryEngine.PollMutableState(ctx, &types.PollMutableStateRequest{\n\t\tDomainUUID:          constants.TestDomainID,\n\t\tExecution:           &workflowExecution,\n\t\tExpectedNextEventID: 4,\n\t\tVersionHistoryItem: &types.VersionHistoryItem{\n\t\t\tEventID: 1,\n\t\t\tVersion: 1,\n\t\t},\n\t})\n\ts.True(time.Now().After(start.Add(time.Second * 1)))\n\ts.Nil(err)\n\ts.Equal(int64(5), pollResponse.GetNextEventID())\n\twaitGroup.Wait()\n}\n\nfunc (s *engineSuite) TestGetMutableStateLongPoll_CurrentBranchChanged() {\n\tctx := context.Background()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tmsBuilder.SetVersionHistories(&persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t// right now the next event ID is 4\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// test long poll on next event ID change\n\tasyncBranchTokenUpdate := func(delay time.Duration) {\n\t\ttimer := time.NewTimer(delay)\n\t\t<-timer.C\n\t\tnewExecution := &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t}\n\t\ts.mockHistoryEngine.historyEventNotifier.NotifyNewHistoryEvent(events.NewNotification(\n\t\t\t\"constants.TestDomainID\",\n\t\t\tnewExecution,\n\t\t\tint64(1),\n\t\t\tint64(4),\n\t\t\tint64(1),\n\t\t\tpersistence.WorkflowStateCreated,\n\t\t\tpersistence.WorkflowCloseStatusNone,\n\t\t\tnil))\n\t}\n\n\t// return immediately, since the expected next event ID appears\n\tresponse0, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID:          constants.TestDomainID,\n\t\tExecution:           &workflowExecution,\n\t\tExpectedNextEventID: 3,\n\t})\n\ts.Nil(err)\n\ts.Equal(int64(4), response0.GetNextEventID())\n\n\t// long poll, new event happen before long poll timeout\n\tgo asyncBranchTokenUpdate(time.Second * 2)\n\tstart := time.Now()\n\tresponse1, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID:          constants.TestDomainID,\n\t\tExecution:           &workflowExecution,\n\t\tExpectedNextEventID: 10,\n\t})\n\ts.True(time.Now().After(start.Add(time.Second * 1)))\n\ts.Nil(err)\n\ts.Equal(response0.GetCurrentBranchToken(), response1.GetCurrentBranchToken())\n}\n\nfunc (s *engineSuite) TestGetMutableStateLongPollTimeout() {\n\tctx := context.Background()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-get-workflow-execution-event-id\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\n\tmsBuilder.SetVersionHistories(&persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t// right now the next event ID is 4\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// long poll, no event happen after long poll timeout\n\tresponse, err := s.mockHistoryEngine.GetMutableState(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID:          constants.TestDomainID,\n\t\tExecution:           &workflowExecution,\n\t\tExpectedNextEventID: 4,\n\t})\n\ts.Nil(err)\n\ts.Equal(int64(4), response.GetNextEventID())\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_RejectBasedOnNotEnabled() {\n\ts.mockHistoryEngine.config.EnableConsistentQueryByDomain = dynamicproperties.GetBoolPropertyFnFilteredByDomain(false)\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.Nil(resp)\n\ts.Equal(workflow.ErrConsistentQueryNotEnabled, err)\n\n\ts.mockHistoryEngine.config.EnableConsistentQueryByDomain = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\ts.mockHistoryEngine.config.EnableConsistentQuery = dynamicproperties.GetBoolPropertyFn(false)\n\tresp, err = s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.Nil(resp)\n\ts.Equal(workflow.ErrConsistentQueryNotEnabled, err)\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_RejectBasedOnCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_RejectBasedOnCompleted\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tevent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tdi.StartedID = event.ID\n\tevent = test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, di.StartedID, nil, \"some random identity\")\n\ttest.AddCompleteWorkflowEvent(msBuilder, event.ID, nil)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:            &workflowExecution,\n\t\t\tQuery:                &types.WorkflowQuery{},\n\t\t\tQueryRejectCondition: types.QueryRejectConditionNotOpen.Ptr(),\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.NoError(err)\n\ts.Nil(resp.GetResponse().QueryResult)\n\ts.NotNil(resp.GetResponse().QueryRejected)\n\ts.Equal(types.WorkflowExecutionCloseStatusCompleted.Ptr(), resp.GetResponse().GetQueryRejected().CloseStatus)\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_RejectBasedOnFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_RejectBasedOnFailed\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tevent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tdi.StartedID = event.ID\n\tevent = test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, di.StartedID, nil, \"some random identity\")\n\ttest.AddFailWorkflowEvent(msBuilder, event.ID, \"failure reason\", []byte{1, 2, 3})\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:            &workflowExecution,\n\t\t\tQuery:                &types.WorkflowQuery{},\n\t\t\tQueryRejectCondition: types.QueryRejectConditionNotOpen.Ptr(),\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.NoError(err)\n\ts.Nil(resp.GetResponse().QueryResult)\n\ts.NotNil(resp.GetResponse().QueryRejected)\n\ts.Equal(types.WorkflowExecutionCloseStatusFailed.Ptr(), resp.GetResponse().GetQueryRejected().CloseStatus)\n\n\trequest = &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:            &workflowExecution,\n\t\t\tQuery:                &types.WorkflowQuery{},\n\t\t\tQueryRejectCondition: types.QueryRejectConditionNotCompletedCleanly.Ptr(),\n\t\t},\n\t}\n\tresp, err = s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.NoError(err)\n\ts.Nil(resp.GetResponse().QueryResult)\n\ts.NotNil(resp.GetResponse().QueryRejected)\n\ts.Equal(types.WorkflowExecutionCloseStatusFailed.Ptr(), resp.GetResponse().GetQueryRejected().CloseStatus)\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_FirstDecisionNotCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_FirstDecisionNotCompleted\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution: &workflowExecution,\n\t\t\tQuery:     &types.WorkflowQuery{},\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.Equal(workflow.ErrQueryWorkflowBeforeFirstDecision, err)\n\ts.Nil(resp)\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_DirectlyThroughMatching() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_DirectlyThroughMatching\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\ts.mockMatchingClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(&types.MatchingQueryWorkflowResponse{QueryResult: []byte{1, 2, 3}}, nil)\n\ts.mockHistoryEngine.matchingClient = s.mockMatchingClient\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution: &workflowExecution,\n\t\t\tQuery:     &types.WorkflowQuery{},\n\t\t\t// since workflow is open this filter does not reject query\n\t\t\tQueryRejectCondition:  types.QueryRejectConditionNotOpen.Ptr(),\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelEventual.Ptr(),\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.NoError(err)\n\ts.NotNil(resp.GetResponse().QueryResult)\n\ts.Nil(resp.GetResponse().QueryRejected)\n\ts.Equal([]byte{1, 2, 3}, resp.GetResponse().GetQueryResult())\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_DecisionTaskDispatch_Timeout() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_DecisionTaskDispatch_Timeout\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution: &workflowExecution,\n\t\t\tQuery:     &types.WorkflowQuery{},\n\t\t\t// since workflow is open this filter does not reject query\n\t\t\tQueryRejectCondition:  types.QueryRejectConditionNotOpen.Ptr(),\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\n\twg := &sync.WaitGroup{}\n\twg.Add(1)\n\tgo func() {\n\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second*2)\n\t\tdefer cancel()\n\t\tresp, err := s.mockHistoryEngine.QueryWorkflow(ctx, request)\n\t\ts.Error(err)\n\t\ts.Nil(resp)\n\t\twg.Done()\n\t}()\n\n\t<-time.After(time.Second)\n\tbuilder := s.getBuilder(constants.TestDomainID, workflowExecution)\n\ts.NotNil(builder)\n\tqr := builder.GetQueryRegistry()\n\ts.True(qr.HasBufferedQuery())\n\ts.False(qr.HasCompletedQuery())\n\ts.False(qr.HasUnblockedQuery())\n\ts.False(qr.HasFailedQuery())\n\twg.Wait()\n\ts.False(qr.HasBufferedQuery())\n\ts.False(qr.HasCompletedQuery())\n\ts.False(qr.HasUnblockedQuery())\n\ts.False(qr.HasFailedQuery())\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_ConsistentQueryBufferFull() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_ConsistentQueryBufferFull\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\t// buffer query so that when types.QueryWorkflow is called buffer is already full\n\tctx, release, err := s.mockHistoryEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(constants.TestDomainID, workflowExecution)\n\ts.NoError(err)\n\tloadedMS, err := ctx.LoadWorkflowExecution(context.Background())\n\ts.NoError(err)\n\tqr := query.NewRegistry()\n\tqr.BufferQuery(&types.WorkflowQuery{})\n\tloadedMS.SetQueryRegistry(qr)\n\trelease(nil)\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:             &workflowExecution,\n\t\t\tQuery:                 &types.WorkflowQuery{},\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.Nil(resp)\n\ts.Equal(workflow.ErrConsistentQueryBufferExceeded, err)\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_DecisionTaskDispatch_Complete() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_DecisionTaskDispatch_Complete\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\twaitGroup := &sync.WaitGroup{}\n\twaitGroup.Add(1)\n\tasyncQueryUpdate := func(delay time.Duration, answer []byte) {\n\t\tdefer waitGroup.Done()\n\t\t<-time.After(delay)\n\t\tbuilder := s.getBuilder(constants.TestDomainID, workflowExecution)\n\t\ts.NotNil(builder)\n\t\tqr := builder.GetQueryRegistry()\n\t\tbuffered := qr.GetBufferedIDs()\n\t\tfor _, id := range buffered {\n\t\t\tresultType := types.QueryResultTypeAnswered\n\t\t\tcompletedTerminationState := &query.TerminationState{\n\t\t\t\tTerminationType: query.TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType: &resultType,\n\t\t\t\t\tAnswer:     answer,\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := qr.SetTerminationState(id, completedTerminationState)\n\t\t\ts.NoError(err)\n\t\t\tstate, err := qr.GetTerminationState(id)\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(query.TerminationTypeCompleted, state.TerminationType)\n\t\t}\n\t}\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:             &workflowExecution,\n\t\t\tQuery:                 &types.WorkflowQuery{},\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\tgo asyncQueryUpdate(time.Second*2, []byte{1, 2, 3})\n\tstart := time.Now()\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.True(time.Now().After(start.Add(time.Second)))\n\ts.NoError(err)\n\ts.Equal([]byte{1, 2, 3}, resp.GetResponse().GetQueryResult())\n\tbuilder := s.getBuilder(constants.TestDomainID, workflowExecution)\n\ts.NotNil(builder)\n\tqr := builder.GetQueryRegistry()\n\ts.False(qr.HasBufferedQuery())\n\ts.False(qr.HasCompletedQuery())\n\twaitGroup.Wait()\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_DecisionTaskDispatch_Complete_ActiveActive() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestActiveActiveDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestActiveActiveDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_DecisionTaskDispatch_Complete_ActiveActive\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestActiveActiveDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, &constants.TestActiveClusterSelectionPolicy)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\n\twaitGroup := &sync.WaitGroup{}\n\twaitGroup.Add(1)\n\tasyncQueryUpdate := func(delay time.Duration, answer []byte) {\n\t\tdefer waitGroup.Done()\n\t\t<-time.After(delay)\n\t\tbuilder := s.getBuilder(constants.TestActiveActiveDomainID, workflowExecution)\n\t\ts.NotNil(builder)\n\t\tqr := builder.GetQueryRegistry()\n\t\tbuffered := qr.GetBufferedIDs()\n\t\tfor _, id := range buffered {\n\t\t\tresultType := types.QueryResultTypeAnswered\n\t\t\tcompletedTerminationState := &query.TerminationState{\n\t\t\t\tTerminationType: query.TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType: &resultType,\n\t\t\t\t\tAnswer:     answer,\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := qr.SetTerminationState(id, completedTerminationState)\n\t\t\ts.NoError(err)\n\t\t\tstate, err := qr.GetTerminationState(id)\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(query.TerminationTypeCompleted, state.TerminationType)\n\t\t}\n\t}\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestActiveActiveDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:             &workflowExecution,\n\t\t\tQuery:                 &types.WorkflowQuery{},\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\tgo asyncQueryUpdate(time.Second*2, []byte{1, 2, 3})\n\tstart := time.Now()\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.True(time.Now().After(start.Add(time.Second)))\n\ts.NoError(err)\n\ts.Equal([]byte{1, 2, 3}, resp.GetResponse().GetQueryResult())\n\tbuilder := s.getBuilder(constants.TestActiveActiveDomainID, workflowExecution)\n\ts.NotNil(builder)\n\tqr := builder.GetQueryRegistry()\n\ts.False(qr.HasBufferedQuery())\n\ts.False(qr.HasCompletedQuery())\n\twaitGroup.Wait()\n}\n\nfunc (s *engineSuite) TestQueryWorkflow_DecisionTaskDispatch_Unblocked() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"TestQueryWorkflow_DecisionTaskDispatch_Unblocked\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\tdi = test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tasklist, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgweResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gweResponse, nil).Once()\n\ts.mockMatchingClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(&types.MatchingQueryWorkflowResponse{QueryResult: []byte{1, 2, 3}}, nil)\n\ts.mockHistoryEngine.matchingClient = s.mockMatchingClient\n\twaitGroup := &sync.WaitGroup{}\n\twaitGroup.Add(1)\n\tasyncQueryUpdate := func(delay time.Duration, answer []byte) {\n\t\tdefer waitGroup.Done()\n\t\t<-time.After(delay)\n\t\tbuilder := s.getBuilder(constants.TestDomainID, workflowExecution)\n\t\ts.NotNil(builder)\n\t\tqr := builder.GetQueryRegistry()\n\t\tbuffered := qr.GetBufferedIDs()\n\t\tfor _, id := range buffered {\n\t\t\ts.NoError(qr.SetTerminationState(id, &query.TerminationState{TerminationType: query.TerminationTypeUnblocked}))\n\t\t\tstate, err := qr.GetTerminationState(id)\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(query.TerminationTypeUnblocked, state.TerminationType)\n\t\t}\n\t}\n\n\trequest := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tExecution:             &workflowExecution,\n\t\t\tQuery:                 &types.WorkflowQuery{},\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\tgo asyncQueryUpdate(time.Second*2, []byte{1, 2, 3})\n\tstart := time.Now()\n\tresp, err := s.mockHistoryEngine.QueryWorkflow(context.Background(), request)\n\ts.True(time.Now().After(start.Add(time.Second)))\n\ts.NoError(err)\n\ts.Equal([]byte{1, 2, 3}, resp.GetResponse().GetQueryResult())\n\tbuilder := s.getBuilder(constants.TestDomainID, workflowExecution)\n\ts.NotNil(builder)\n\tqr := builder.GetQueryRegistry()\n\ts.False(qr.HasBufferedQuery())\n\ts.False(qr.HasCompletedQuery())\n\ts.False(qr.HasUnblockedQuery())\n\twaitGroup.Wait()\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedInvalidToken() {\n\n\tinvalidToken, _ := json.Marshal(\"bad token\")\n\tidentity := \"testIdentity\"\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        invalidToken,\n\t\t\tDecisions:        nil,\n\t\t\tExecutionContext: nil,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\n\ts.NotNil(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedIfNoExecution() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).Times(1)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedIfGetExecutionFailed() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"FAILED\")).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedUpdateExecutionFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, errors.New(\"FAILED\")).Once()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedIfTaskCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\ttest.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID, startedEvent.ID, nil, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedIfTaskNotStarted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedConflictOnUpdate() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tactivity1Result := []byte(\"activity1_result\")\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"activity_type2\"\n\tactivity2Input := []byte(\"input2\")\n\tactivity2Result := []byte(\"activity2_result\")\n\tactivity3ID := \"activity3\"\n\tactivity3Type := \"activity_type3\"\n\tactivity3Input := []byte(\"input3\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi1 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di1.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di1.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID,\n\t\tactivity1ID, activity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity2ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID,\n\t\tactivity2ID, activity2Type, tl, activity2Input, 100, 10, 1, 5)\n\tactivity1StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\tactivity2StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity2ScheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity1ScheduledEvent.ID,\n\t\tactivity1StartedEvent.ID, activity1Result, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent2 := test.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: di2.ScheduleID,\n\t})\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\tActivityID:                    activity3ID,\n\t\t\tActivityType:                  &types.ActivityType{Name: activity3Type},\n\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\tInput:                         activity3Input,\n\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity2ScheduledEvent.ID,\n\t\tactivity2StartedEvent.ID, activity2Result, identity)\n\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: ms2}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}},\n\t\t&persistence.ConditionFailedError{}).Once()\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\ts.Equal(int64(16), ms2.ExecutionInfo.NextEventID)\n\ts.Equal(decisionStartedEvent2.ID, ms2.ExecutionInfo.LastProcessedEvent)\n\ts.Equal(executionContext, ms2.ExecutionInfo.ExecutionContext)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\tactivity3Attributes := s.getActivityScheduledEvent(executionBuilder, 13).ActivityTaskScheduledEventAttributes\n\ts.Equal(activity3ID, activity3Attributes.ActivityID)\n\ts.Equal(activity3Type, activity3Attributes.ActivityType.Name)\n\ts.Equal(int64(12), activity3Attributes.DecisionTaskCompletedEventID)\n\ts.Equal(tl, activity3Attributes.TaskList.Name)\n\ts.Equal(activity3Input, activity3Attributes.Input)\n\ts.Equal(int32(100), *activity3Attributes.ScheduleToCloseTimeoutSeconds)\n\ts.Equal(int32(10), *activity3Attributes.ScheduleToStartTimeoutSeconds)\n\ts.Equal(int32(50), *activity3Attributes.StartToCloseTimeoutSeconds)\n\ts.Equal(int32(5), *activity3Attributes.HeartbeatTimeoutSeconds)\n\n\tdi, ok := executionBuilder.GetDecisionInfo(15)\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n}\n\nfunc (s *engineSuite) TestValidateSignalRequest() {\n\tworkflowType := \"testType\"\n\tinput := []byte(\"input\")\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tWorkflowID:                          \"ID\",\n\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\tTaskList:                            &types.TaskList{Name: \"taskptr\"},\n\t\tInput:                               input,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(10),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tIdentity:                            \"identity\",\n\t}\n\n\terr := s.mockHistoryEngine.validateStartWorkflowExecutionRequest(startRequest, 0)\n\ts.Error(err, \"startRequest doesn't have request id, it should error out\")\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedMaxAttemptsExceeded() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tinput := []byte(\"input\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\tActivityID:                    \"activity1\",\n\t\t\tActivityType:                  &types.ActivityType{Name: \"activity_type1\"},\n\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\tInput:                         input,\n\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t},\n\t}}\n\n\tfor i := 0; i < workflow.ConditionalRetryCount; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}},\n\t\t\t&persistence.ConditionFailedError{}).Once()\n\t}\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.Equal(workflow.ErrMaxAttemptsExceeded, err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedCompleteWorkflowFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tactivity1Result := []byte(\"activity1_result\")\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"activity_type2\"\n\tactivity2Input := []byte(\"input2\")\n\tactivity2Result := []byte(\"activity2_result\")\n\tworkflowResult := []byte(\"workflow result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 25, 200, identity, nil)\n\tdi1 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di1.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di1.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID,\n\t\tactivity1ID, activity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity2ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID,\n\t\tactivity2ID, activity2Type, tl, activity2Input, 100, 10, 1, 5)\n\tactivity1StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\tactivity2StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity2ScheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity1ScheduledEvent.ID,\n\t\tactivity1StartedEvent.ID, activity1Result, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity2ScheduledEvent.ID,\n\t\tactivity2StartedEvent.ID, activity2Result, identity)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: di2.ScheduleID,\n\t})\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\tResult: workflowResult,\n\t\t},\n\t}}\n\n\tfor i := 0; i < 2; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t}\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(15), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(decisionStartedEvent1.ID, executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Empty(executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi3, ok := executionBuilder.GetDecisionInfo(executionBuilder.GetExecutionInfo().NextEventID - 1)\n\ts.True(ok)\n\ts.Equal(executionBuilder.GetExecutionInfo().NextEventID-1, di3.ScheduleID)\n\ts.Equal(int64(0), di3.Attempt)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedFailWorkflowFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tactivity1Result := []byte(\"activity1_result\")\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"activity_type2\"\n\tactivity2Input := []byte(\"input2\")\n\tactivity2Result := []byte(\"activity2_result\")\n\treason := \"workflow fail reason\"\n\tdetails := []byte(\"workflow fail details\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 25, 200, identity, nil)\n\tdi1 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di1.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di1.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity1ID,\n\t\tactivity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity2ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity2ID,\n\t\tactivity2Type, tl, activity2Input, 100, 10, 1, 5)\n\tactivity1StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\tactivity2StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity2ScheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity1ScheduledEvent.ID,\n\t\tactivity1StartedEvent.ID, activity1Result, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity2ScheduledEvent.ID,\n\t\tactivity2StartedEvent.ID, activity2Result, identity)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: di2.ScheduleID,\n\t})\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\tReason:  &reason,\n\t\t\tDetails: details,\n\t\t},\n\t}}\n\n\tfor i := 0; i < 2; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t}\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(15), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(decisionStartedEvent1.ID, executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Empty(executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi3, ok := executionBuilder.GetDecisionInfo(executionBuilder.GetExecutionInfo().NextEventID - 1)\n\ts.True(ok)\n\ts.Equal(executionBuilder.GetExecutionInfo().NextEventID-1, di3.ScheduleID)\n\ts.Equal(int64(0), di3.Attempt)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedBadDecisionAttributes() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tactivity1Result := []byte(\"activity1_result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 25, 200, identity, nil)\n\tdi1 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di1.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di1.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity1ID,\n\t\tactivity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity1StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity1ScheduledEvent.ID,\n\t\tactivity1StartedEvent.ID, activity1Result, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: di2.ScheduleID,\n\t})\n\n\t// Decision with nil attributes\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t}}\n\n\tgwmsResponse1 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{\n\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil,\n\t).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\n// This test unit tests the activity schedule timeout validation logic of HistoryEngine's RespondDecisionTaskComplete function.\n// An scheduled activity decision has 3 timeouts: ScheduleToClose, ScheduleToStart and StartToClose.\n// This test verifies that when either ScheduleToClose or ScheduleToStart and StartToClose are specified,\n// HistoryEngine's validateActivityScheduleAttribute will deduce the missing timeout and fill it in\n// instead of returning a BadRequest error and only when all three are missing should a BadRequest be returned.\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedSingleActivityScheduledAttribute() {\n\tworkflowTimeout := int32(100)\n\ttestIterationVariables := []struct {\n\t\tscheduleToClose         *int32\n\t\tscheduleToStart         *int32\n\t\tstartToClose            *int32\n\t\theartbeat               *int32\n\t\texpectedScheduleToClose int32\n\t\texpectedScheduleToStart int32\n\t\texpectedStartToClose    int32\n\t\texpectDecisionFail      bool\n\t}{\n\t\t// No ScheduleToClose timeout, will use ScheduleToStart + StartToClose\n\t\t{nil, common.Int32Ptr(3), common.Int32Ptr(7), nil,\n\t\t\t3 + 7, 3, 7, false},\n\t\t// Has ScheduleToClose timeout but not ScheduleToStart or StartToClose,\n\t\t// will use ScheduleToClose for ScheduleToStart and StartToClose\n\t\t{common.Int32Ptr(7), nil, nil, nil,\n\t\t\t7, 7, 7, false},\n\t\t// No ScheduleToClose timeout, ScheduleToStart or StartToClose, expect error return\n\t\t{nil, nil, nil, nil,\n\t\t\t0, 0, 0, true},\n\t\t// Negative ScheduleToClose, expect error return\n\t\t{common.Int32Ptr(-1), nil, nil, nil,\n\t\t\t0, 0, 0, true},\n\t\t// Negative ScheduleToStart, expect error return\n\t\t{nil, common.Int32Ptr(-1), nil, nil,\n\t\t\t0, 0, 0, true},\n\t\t// Negative StartToClose, expect error return\n\t\t{nil, nil, common.Int32Ptr(-1), nil,\n\t\t\t0, 0, 0, true},\n\t\t// Negative HeartBeat, expect error return\n\t\t{nil, nil, nil, common.Int32Ptr(-1),\n\t\t\t0, 0, 0, true},\n\t\t// Use workflow timeout\n\t\t{common.Int32Ptr(workflowTimeout), nil, nil, nil,\n\t\t\tworkflowTimeout, workflowTimeout, workflowTimeout, false},\n\t\t// Timeout larger than workflow timeout\n\t\t{common.Int32Ptr(workflowTimeout + 1), nil, nil, nil,\n\t\t\tworkflowTimeout, workflowTimeout, workflowTimeout, false},\n\t\t{nil, common.Int32Ptr(workflowTimeout + 1), nil, nil,\n\t\t\t0, 0, 0, true},\n\t\t{nil, nil, common.Int32Ptr(workflowTimeout + 1), nil,\n\t\t\t0, 0, 0, true},\n\t\t{nil, nil, nil, common.Int32Ptr(workflowTimeout + 1),\n\t\t\t0, 0, 0, true},\n\t\t// No ScheduleToClose timeout, will use ScheduleToStart + StartToClose, but exceed limit\n\t\t{nil, common.Int32Ptr(workflowTimeout), common.Int32Ptr(10), nil,\n\t\t\tworkflowTimeout, workflowTimeout, 10, false},\n\t}\n\n\tfor _, iVar := range testIterationVariables {\n\t\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t\t}\n\t\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\t\twe := types.WorkflowExecution{\n\t\t\tWorkflowID: \"wId\",\n\t\t\tRunID:      constants.TestRunID,\n\t\t}\n\t\ttl := \"testTaskList\"\n\t\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\t\tWorkflowID: \"wId\",\n\t\t\tRunID:      we.GetRunID(),\n\t\t\tScheduleID: 2,\n\t\t})\n\t\tidentity := \"testIdentity\"\n\t\texecutionContext := []byte(\"context\")\n\t\tinput := []byte(\"input\")\n\n\t\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\t\ts.mockHistoryEngine.shard,\n\t\t\ttestlogger.New(s.Suite.T()),\n\t\t\twe.GetRunID(),\n\t\t\tconstants.TestLocalDomainEntry,\n\t\t)\n\t\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), workflowTimeout, 200, identity, nil)\n\t\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\t\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\t\tdecisions := []*types.Decision{{\n\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\tActivityID:                    \"activity1\",\n\t\t\t\tActivityType:                  &types.ActivityType{Name: \"activity_type1\"},\n\t\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\t\tInput:                         input,\n\t\t\t\tScheduleToCloseTimeoutSeconds: iVar.scheduleToClose,\n\t\t\t\tScheduleToStartTimeoutSeconds: iVar.scheduleToStart,\n\t\t\t\tStartToCloseTimeoutSeconds:    iVar.startToClose,\n\t\t\t\tHeartbeatTimeoutSeconds:       iVar.heartbeat,\n\t\t\t},\n\t\t}}\n\n\t\tgwmsResponse1 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\t\tif iVar.expectDecisionFail {\n\t\t\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\t\t}\n\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\tTaskToken:        taskToken,\n\t\t\t\tDecisions:        decisions,\n\t\t\t\tExecutionContext: executionContext,\n\t\t\t\tIdentity:         identity,\n\t\t\t},\n\t\t})\n\n\t\ts.Nil(err, s.printHistory(msBuilder))\n\t\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\t\tif !iVar.expectDecisionFail {\n\t\t\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\t\t\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\t\t\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\t\t\ts.False(executionBuilder.HasPendingDecision())\n\n\t\t\tactivity1Attributes := s.getActivityScheduledEvent(executionBuilder, int64(5)).ActivityTaskScheduledEventAttributes\n\t\t\ts.Equal(iVar.expectedScheduleToClose, activity1Attributes.GetScheduleToCloseTimeoutSeconds())\n\t\t\ts.Equal(iVar.expectedScheduleToStart, activity1Attributes.GetScheduleToStartTimeoutSeconds())\n\t\t\ts.Equal(iVar.expectedStartToClose, activity1Attributes.GetStartToCloseTimeoutSeconds())\n\t\t} else {\n\t\t\ts.Equal(int64(5), executionBuilder.GetExecutionInfo().NextEventID)\n\t\t\ts.Equal(commonconstants.EmptyEventID, executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\t\t\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\t\t\ts.True(executionBuilder.HasPendingDecision())\n\t\t}\n\t\ts.TearDownTest()\n\t\ts.SetupTest()\n\t}\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedBadBinary() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tdomainID := uuid.New()\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID, Name: constants.TestDomainName},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention: 2,\n\t\t\tBadBinaries: types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\t\"test-bad-binary\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tcluster.TestCurrentClusterName,\n\t)\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tdomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tvar decisions []*types.Decision\n\n\tgwmsResponse1 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: execution.CreatePersistenceMutableState(s.T(), msBuilder)}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(constants.TestDomainName, nil).AnyTimes()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: domainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t\tBinaryChecksum:   \"test-bad-binary\",\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(domainID, we)\n\ts.Equal(int64(5), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(commonconstants.EmptyEventID, executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Empty(executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedSingleActivityScheduledDecision() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tinput := []byte(\"input\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\tScheduleActivityTaskDecisionAttributes: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\tActivityID:                    \"activity1\",\n\t\t\tActivityType:                  &types.ActivityType{Name: \"activity_type1\"},\n\t\t\tTaskList:                      &types.TaskList{Name: tl},\n\t\t\tInput:                         input,\n\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(10),\n\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(5),\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\n\tactivity1Attributes := s.getActivityScheduledEvent(executionBuilder, int64(5)).ActivityTaskScheduledEventAttributes\n\ts.Equal(\"activity1\", activity1Attributes.ActivityID)\n\ts.Equal(\"activity_type1\", activity1Attributes.ActivityType.Name)\n\ts.Equal(int64(4), activity1Attributes.DecisionTaskCompletedEventID)\n\ts.Equal(tl, activity1Attributes.TaskList.Name)\n\ts.Equal(input, activity1Attributes.Input)\n\ts.Equal(int32(100), *activity1Attributes.ScheduleToCloseTimeoutSeconds)\n\ts.Equal(int32(10), *activity1Attributes.ScheduleToStartTimeoutSeconds)\n\ts.Equal(int32(50), *activity1Attributes.StartToCloseTimeoutSeconds)\n\ts.Equal(int32(5), *activity1Attributes.HeartbeatTimeoutSeconds)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompleted_DecisionHeartbeatTimeout() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tmsBuilder.GetExecutionInfo().DecisionOriginalScheduledTimestamp = time.Now().Add(-time.Hour).UnixNano()\n\n\tdecisions := []*types.Decision{}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tForceCreateNewDecisionTask: true,\n\t\t\tTaskToken:                  taskToken,\n\t\t\tDecisions:                  decisions,\n\t\t\tExecutionContext:           executionContext,\n\t\t\tIdentity:                   identity,\n\t\t},\n\t})\n\ts.Error(err, \"decision heartbeat timeout\")\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompleted_DecisionHeartbeatNotTimeout() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tmsBuilder.GetExecutionInfo().DecisionOriginalScheduledTimestamp = time.Now().Add(-time.Minute).UnixNano()\n\n\tdecisions := []*types.Decision{}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tForceCreateNewDecisionTask: true,\n\t\t\tTaskToken:                  taskToken,\n\t\t\tDecisions:                  decisions,\n\t\t\tExecutionContext:           executionContext,\n\t\t\tIdentity:                   identity,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompleted_DecisionHeartbeatNotTimeout_ZeroOrignalScheduledTime() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tmsBuilder.GetExecutionInfo().DecisionOriginalScheduledTimestamp = 0\n\n\tdecisions := []*types.Decision{}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tForceCreateNewDecisionTask: true,\n\t\t\tTaskToken:                  taskToken,\n\t\t\tDecisions:                  decisions,\n\t\t\tExecutionContext:           executionContext,\n\t\t\tIdentity:                   identity,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedCompleteWorkflowSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tworkflowResult := []byte(\"success\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\tResult: workflowResult,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateCompleted, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedFailWorkflowSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tdetails := []byte(\"fail workflow details\")\n\treason := \"fail workflow reason\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeFailWorkflowExecution.Ptr(),\n\t\tFailWorkflowExecutionDecisionAttributes: &types.FailWorkflowExecutionDecisionAttributes{\n\t\t\tReason:  &reason,\n\t\t\tDetails: details,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(persistence.WorkflowStateCompleted, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedSignalExternalWorkflowSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain: constants.TestDomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: we.WorkflowID,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte(\"test input\"),\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedStartChildWorkflowWithAbandonPolicy() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tabandon := types.ParentClosePolicyAbandon\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:     constants.TestDomainName,\n\t\t\tWorkflowID: \"child-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"child-workflow-type\",\n\t\t\t},\n\t\t\tParentClosePolicy: &abandon,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(int(1), len(executionBuilder.GetPendingChildExecutionInfos()))\n\tvar childID int64\n\tfor c := range executionBuilder.GetPendingChildExecutionInfos() {\n\t\tchildID = c\n\t\tbreak\n\t}\n\ts.Equal(\"child-workflow-id\", executionBuilder.GetPendingChildExecutionInfos()[childID].StartedWorkflowID)\n\ts.Equal(types.ParentClosePolicyAbandon, executionBuilder.GetPendingChildExecutionInfos()[childID].ParentClosePolicy)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedStartChildWorkflowWithTerminatePolicy() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tterminate := types.ParentClosePolicyTerminate\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeStartChildWorkflowExecution.Ptr(),\n\t\tStartChildWorkflowExecutionDecisionAttributes: &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:     constants.TestDomainName,\n\t\t\tWorkflowID: \"child-workflow-id\",\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"child-workflow-type\",\n\t\t\t},\n\t\t\tParentClosePolicy: &terminate,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(executionContext, executionBuilder.GetExecutionInfo().ExecutionContext)\n\ts.Equal(int(1), len(executionBuilder.GetPendingChildExecutionInfos()))\n\tvar childID int64\n\tfor c := range executionBuilder.GetPendingChildExecutionInfos() {\n\t\tchildID = c\n\t\tbreak\n\t}\n\ts.Equal(\"child-workflow-id\", executionBuilder.GetPendingChildExecutionInfos()[childID].StartedWorkflowID)\n\ts.Equal(types.ParentClosePolicyTerminate, executionBuilder.GetPendingChildExecutionInfos()[childID].ParentClosePolicy)\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedSignalExternalWorkflowFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      \"invalid run id\",\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.GetWorkflowID(),\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain: constants.TestDomainID,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: we.WorkflowID,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte(\"test input\"),\n\t\t},\n\t}}\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\n\ts.EqualError(err, \"RunID is not valid UUID.\")\n}\n\nfunc (s *engineSuite) TestRespondDecisionTaskCompletedSignalExternalWorkflowFailed_UnKnownDomain() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\texecutionContext := []byte(\"context\")\n\tforeignDomain := \"unknown domain\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeSignalExternalWorkflowExecution.Ptr(),\n\t\tSignalExternalWorkflowExecutionDecisionAttributes: &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain: foreignDomain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: we.WorkflowID,\n\t\t\t\tRunID:      we.RunID,\n\t\t\t},\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte(\"test input\"),\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockDomainCache.EXPECT().GetDomain(foreignDomain).Return(\n\t\tnil, errors.New(\"get foreign domain error\"),\n\t).Times(1)\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: executionContext,\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\n\ts.NotNil(err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedInvalidToken() {\n\n\tinvalidToken, _ := json.Marshal(\"bad token\")\n\tidentity := \"testIdentity\"\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: invalidToken,\n\t\t\tResult:    nil,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\n\ts.NotNil(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfNoExecution() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfNoRunID() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfGetExecutionFailed() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"FAILED\")).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfNoAIdProvided() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t})\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"Neither ActivityID nor ScheduleID is provided\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfNotFound() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: \"aid\",\n\t})\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Error(err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedUpdateExecutionFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, errors.New(\"FAILED\")).Once()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfTaskCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\tactivityStartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activityScheduledEvent.ID, activityStartedEvent.ID,\n\t\tactivityResult, identity)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedIfTaskNotStarted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 200, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedConflictOnUpdate() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tactivity1Result := []byte(\"activity1_result\")\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"activity_type2\"\n\tactivity2Input := []byte(\"input2\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity1ID,\n\t\tactivity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity2ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity2ID,\n\t\tactivity2Type, tl, activity2Input, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activity2ScheduledEvent.ID, identity)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &persistence.GetWorkflowExecutionResponse{State: ms1}\n\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: ms2}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, &persistence.ConditionFailedError{}).Once()\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activity1Result,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(10))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(10), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedMaxAttemptsExceeded() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tfor i := 0; i < workflow.ConditionalRetryCount; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, &persistence.ConditionFailedError{}).Once()\n\t}\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Equal(workflow.ErrMaxAttemptsExceeded, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(9), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(8))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(8), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCompletedByIdSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tactivityResult := []byte(\"activity result\")\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: activityID,\n\t})\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdecisionScheduledEvent := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, decisionScheduledEvent.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, decisionScheduledEvent.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: we.RunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCompleted(context.Background(), &types.HistoryRespondActivityTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tResult:    activityResult,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(9), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(8))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(8), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedInvalidToken() {\n\n\tinvalidToken, _ := json.Marshal(\"bad token\")\n\tidentity := \"testIdentity\"\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: invalidToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\n\ts.NotNil(err)\n\ts.IsType(&types.BadRequestError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedIfNoExecution() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil,\n\t\t&types.EntityNotExistsError{}).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedIfNoRunID() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil,\n\t\t&types.EntityNotExistsError{}).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedIfGetExecutionFailed() {\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil,\n\t\terrors.New(\"FAILED\")).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailededIfNoAIdProvided() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t})\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"Neither ActivityID nor ScheduleID is provided\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailededIfNotFound() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: \"aid\",\n\t})\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Error(err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedUpdateExecutionFailed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, errors.New(\"FAILED\")).Once()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"FAILED\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedIfTaskCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tfailReason := \"fail reason\"\n\tdetails := []byte(\"fail details\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\tactivityStartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\ttest.AddActivityTaskFailedEvent(msBuilder, activityScheduledEvent.ID, activityStartedEvent.ID,\n\t\tfailReason, details, identity)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tReason:    &failReason,\n\t\t\tDetails:   details,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedIfTaskNotStarted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedConflictOnUpdate() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"activity_type1\"\n\tactivity1Input := []byte(\"input1\")\n\tfailReason := \"fail reason\"\n\tdetails := []byte(\"fail details.\")\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"activity_type2\"\n\tactivity2Input := []byte(\"input2\")\n\tactivity2Result := []byte(\"activity2_result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 25, 25, identity, nil)\n\tdi1 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent1 := test.AddDecisionTaskStartedEvent(msBuilder, di1.ScheduleID, tl, identity)\n\tdecisionCompletedEvent1 := test.AddDecisionTaskCompletedEvent(msBuilder, di1.ScheduleID,\n\t\tdecisionStartedEvent1.ID, nil, identity)\n\tactivity1ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity1ID,\n\t\tactivity1Type, tl, activity1Input, 100, 10, 1, 5)\n\tactivity2ScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent1.ID, activity2ID,\n\t\tactivity2Type, tl, activity2Input, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activity1ScheduledEvent.ID, identity)\n\tactivity2StartedEvent := test.AddActivityTaskStartedEvent(msBuilder, activity2ScheduledEvent.ID, identity)\n\n\tms1 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse1 := &persistence.GetWorkflowExecutionResponse{State: ms1}\n\n\ttest.AddActivityTaskCompletedEvent(msBuilder, activity2ScheduledEvent.ID,\n\t\tactivity2StartedEvent.ID, activity2Result, identity)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms2 := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: ms2}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse1, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, &persistence.ConditionFailedError{}).Once()\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tReason:    &failReason,\n\t\t\tDetails:   details,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err, s.printHistory(msBuilder))\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(12), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(10))\n\ts.True(ok)\n\ts.Equal(int32(25), di.DecisionTimeout)\n\ts.Equal(int64(10), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedMaxAttemptsExceeded() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tfor i := 0; i < workflow.ConditionalRetryCount; i++ {\n\t\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, &persistence.ConditionFailedError{}).Once()\n\t}\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Equal(workflow.ErrMaxAttemptsExceeded, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tfailReason := \"failed\"\n\tfailDetails := []byte(\"fail details.\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tReason:    &failReason,\n\t\t\tDetails:   failDetails,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(9), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(8))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(8), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskFailedByIDSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tfailReason := \"failed\"\n\tfailDetails := []byte(\"fail details.\")\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: activityID,\n\t})\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdecisionScheduledEvent := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, decisionScheduledEvent.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, decisionScheduledEvent.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 5)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: we.RunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskFailed(context.Background(), &types.HistoryRespondActivityTaskFailedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tReason:    &failReason,\n\t\t\tDetails:   failDetails,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(9), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(8))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(8), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRecordActivityTaskHeartBeatSuccess_NoTimer() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 0)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\t// No HeartBeat timer running.\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tdetais := []byte(\"details\")\n\n\t_, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   detais,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestRecordActivityTaskHeartBeatSuccess_TimerRunning() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\t// HeartBeat timer running.\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tdetais := []byte(\"details\")\n\n\t_, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   detais,\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRecordActivityTaskHeartBeatByIDSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: activityID,\n\t})\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 0)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\n\t// No HeartBeat timer running.\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tdetais := []byte(\"details\")\n\n\t_, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   detais,\n\t\t},\n\t})\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceled_Scheduled() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceled_Started() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 5,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\t_, _, err := msBuilder.AddActivityTaskCancelRequestedEvent(decisionCompletedEvent.ID, activityID, identity)\n\ts.Nil(err)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr = s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(10), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(9))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(9), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceledByID_Started() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: activityID,\n\t})\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdecisionScheduledEvent := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, decisionScheduledEvent.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, decisionScheduledEvent.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\t_, _, err := msBuilder.AddActivityTaskCancelRequestedEvent(decisionCompletedEvent.ID, activityID, identity)\n\ts.Nil(err)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: we.RunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr = s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(10), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi, ok := executionBuilder.GetDecisionInfo(int64(9))\n\ts.True(ok)\n\ts.Equal(int32(100), di.DecisionTimeout)\n\ts.Equal(int64(9), di.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, di.StartedID)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceledIfNoRunID() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil)\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.NotNil(err)\n\ts.IsType(&types.EntityNotExistsError{}, err)\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceledIfNoAIdProvided() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t})\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.EqualError(err, \"Neither ActivityID nor ScheduleID is provided\")\n}\n\nfunc (s *engineSuite) TestRespondActivityTaskCanceledIfNotFound() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tScheduleID: commonconstants.EmptyEventID,\n\t\tActivityID: \"aid\",\n\t})\n\tidentity := \"testIdentity\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: taskToken,\n\t\t\tIdentity:  identity,\n\t\t},\n\t})\n\ts.Error(err)\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_NotScheduled() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_Scheduled() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 6,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(12), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi2, ok := executionBuilder.GetDecisionInfo(executionBuilder.GetExecutionInfo().NextEventID - 1)\n\ts.True(ok)\n\ts.Equal(executionBuilder.GetExecutionInfo().NextEventID-1, di2.ScheduleID)\n\ts.Equal(int64(0), di2.Attempt)\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_Started() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 7,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 0)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_Completed() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 6,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\tworkflowResult := []byte(\"workflow result\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 0)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tdecisions := []*types.Decision{\n\t\t{\n\t\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\t\tActivityID: activityID,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tDecisionType: types.DecisionTypeCompleteWorkflowExecution.Ptr(),\n\t\t\tCompleteWorkflowExecutionDecisionAttributes: &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\t\t\tResult: workflowResult,\n\t\t\t},\n\t\t},\n\t}\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateCompleted, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_NoHeartBeat() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 7,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 0)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\n\t// Try recording activity heartbeat\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tactivityTaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 5,\n\t})\n\n\thbResponse, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(hbResponse)\n\ts.True(hbResponse.CancelRequested)\n\n\t// Try cancelling the request.\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr = s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder = s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(13), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_Success() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 7,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\n\t// Try recording activity heartbeat\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tactivityTaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 5,\n\t})\n\n\thbResponse, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(hbResponse)\n\ts.True(hbResponse.CancelRequested)\n\n\t// Try cancelling the request.\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr = s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder = s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(13), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_SuccessWithQueries() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 7,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t// load mutable state such that it already exists in memory when respond decision task is called\n\t// this enables us to set query registry on it\n\tctx, release, err := s.mockHistoryEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(constants.TestDomainID, we)\n\ts.NoError(err)\n\tloadedMS, err := ctx.LoadWorkflowExecution(context.Background())\n\ts.NoError(err)\n\tqr := query.NewRegistry()\n\tid1, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tid2, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tid3, _ := qr.BufferQuery(&types.WorkflowQuery{})\n\tloadedMS.SetQueryRegistry(qr)\n\trelease(nil)\n\tresult1 := &types.WorkflowQueryResult{\n\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\tAnswer:     []byte{1, 2, 3},\n\t}\n\tresult2 := &types.WorkflowQueryResult{\n\t\tResultType:   types.QueryResultTypeFailed.Ptr(),\n\t\tErrorMessage: \"error reason\",\n\t}\n\tqueryResults := map[string]*types.WorkflowQueryResult{\n\t\tid1: result1,\n\t\tid2: result2,\n\t}\n\t_, err = s.mockHistoryEngine.RespondDecisionTaskCompleted(s.constructCallContext(cc.GoWorkerConsistentQueryVersion), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t\tQueryResults:     queryResults,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\ts.Len(qr.GetCompletedIDs(), 2)\n\tcompleted1, err := qr.GetTerminationState(id1)\n\ts.NoError(err)\n\ts.Equal(result1, completed1.QueryResult)\n\ts.Equal(query.TerminationTypeCompleted, completed1.TerminationType)\n\tcompleted2, err := qr.GetTerminationState(id2)\n\ts.NoError(err)\n\ts.Equal(result2, completed2.QueryResult)\n\ts.Equal(query.TerminationTypeCompleted, completed2.TerminationType)\n\ts.Len(qr.GetBufferedIDs(), 0)\n\ts.Len(qr.GetFailedIDs(), 0)\n\ts.Len(qr.GetUnblockedIDs(), 1)\n\tunblocked1, err := qr.GetTerminationState(id3)\n\ts.NoError(err)\n\ts.Nil(unblocked1.QueryResult)\n\ts.Equal(query.TerminationTypeUnblocked, unblocked1.TerminationType)\n\n\t// Try recording activity heartbeat\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\tactivityTaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      we.GetRunID(),\n\t\tScheduleID: 5,\n\t})\n\n\thbResponse, err := s.mockHistoryEngine.RecordActivityTaskHeartbeat(context.Background(), &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\ts.NotNil(hbResponse)\n\ts.True(hbResponse.CancelRequested)\n\n\t// Try cancelling the request.\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr = s.mockHistoryEngine.RespondActivityTaskCanceled(context.Background(), &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: activityTaskToken,\n\t\t\tIdentity:  identity,\n\t\t\tDetails:   []byte(\"details\"),\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder = s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(13), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestRequestCancel_RespondDecisionTaskCompleted_SuccessWithConsistentQueriesUnsupported() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 7,\n\t})\n\tidentity := \"testIdentity\"\n\tactivityID := \"activity1_id\"\n\tactivityType := \"activity_type1\"\n\tactivityInput := []byte(\"input1\")\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\tactivityScheduledEvent, _ := test.AddActivityTaskScheduledEvent(msBuilder, decisionCompletedEvent.ID, activityID,\n\t\tactivityType, tl, activityInput, 100, 10, 1, 1)\n\ttest.AddActivityTaskStartedEvent(msBuilder, activityScheduledEvent.ID, identity)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeRequestCancelActivityTask.Ptr(),\n\t\tRequestCancelActivityTaskDecisionAttributes: &types.RequestCancelActivityTaskDecisionAttributes{\n\t\t\tActivityID: activityID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t// load mutable state such that it already exists in memory when respond decision task is called\n\t// this enables us to set query registry on it\n\tctx, release, err := s.mockHistoryEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(constants.TestDomainID, we)\n\ts.NoError(err)\n\tloadedMS, err := ctx.LoadWorkflowExecution(context.Background())\n\ts.NoError(err)\n\tqr := query.NewRegistry()\n\tqr.BufferQuery(&types.WorkflowQuery{})\n\tqr.BufferQuery(&types.WorkflowQuery{})\n\tqr.BufferQuery(&types.WorkflowQuery{})\n\tloadedMS.SetQueryRegistry(qr)\n\trelease(nil)\n\t_, err = s.mockHistoryEngine.RespondDecisionTaskCompleted(s.constructCallContext(\"0.0.0\"), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(11), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(8), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\ts.Len(qr.GetBufferedIDs(), 0)\n\ts.Len(qr.GetCompletedIDs(), 0)\n\ts.Len(qr.GetUnblockedIDs(), 0)\n\ts.Len(qr.GetFailedIDs(), 3)\n}\n\nfunc (s *engineSuite) constructCallContext(featureVersion string) context.Context {\n\tctx := context.Background()\n\tctx, call := encoding.NewInboundCall(ctx)\n\terr := call.ReadFromRequest(&transport.Request{\n\t\tHeaders: transport.NewHeaders().With(common.ClientImplHeaderName, cc.GoSDK).With(common.FeatureVersionHeaderName, featureVersion),\n\t})\n\ts.NoError(err)\n\treturn ctx\n}\n\nfunc (s *engineSuite) TestStarTimer_DuplicateTimerID() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\ttimerID := \"t1\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeStartTimer.Ptr(),\n\t\tStartTimerDecisionAttributes: &types.StartTimerDecisionAttributes{\n\t\t\tTimerID:                   timerID,\n\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(1),\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\n\t// Try to add the same timer ID again.\n\tdi2 := test.AddDecisionTaskScheduledEvent(executionBuilder)\n\ttest.AddDecisionTaskStartedEvent(executionBuilder, di2.ScheduleID, tl, identity)\n\ttaskToken2, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: di2.ScheduleID,\n\t})\n\n\tms2 := execution.CreatePersistenceMutableState(s.T(), executionBuilder)\n\tgwmsResponse2 := &persistence.GetWorkflowExecutionResponse{State: ms2}\n\n\tdecisionFailedEvent := false\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse2, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Run(func(arguments mock.Arguments) {\n\t\treq := arguments.Get(1).(*persistence.AppendHistoryNodesRequest)\n\t\tdecTaskIndex := len(req.Events) - 1\n\t\tif decTaskIndex >= 0 && *req.Events[decTaskIndex].EventType == types.EventTypeDecisionTaskFailed {\n\t\t\tdecisionFailedEvent = true\n\t\t}\n\t}).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken2,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\ts.True(decisionFailedEvent)\n\texecutionBuilder = s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.True(executionBuilder.HasPendingDecision())\n\tdi3, ok := executionBuilder.GetDecisionInfo(executionBuilder.GetExecutionInfo().NextEventID)\n\ts.True(ok, \"DI.ScheduleID: %v, ScheduleID: %v, StartedID: %v\", di2.ScheduleID,\n\t\texecutionBuilder.GetExecutionInfo().DecisionScheduleID, executionBuilder.GetExecutionInfo().DecisionStartedID)\n\ts.Equal(executionBuilder.GetExecutionInfo().NextEventID, di3.ScheduleID)\n\ts.Equal(int64(1), di3.Attempt)\n}\n\nfunc (s *engineSuite) TestUserTimer_RespondDecisionTaskCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 6,\n\t})\n\tidentity := \"testIdentity\"\n\ttimerID := \"t1\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\t// Verify cancel timer with a start event.\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddTimerStartedEvent(msBuilder, decisionCompletedEvent.ID, timerID, 10)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCancelTimer.Ptr(),\n\t\tCancelTimerDecisionAttributes: &types.CancelTimerDecisionAttributes{\n\t\t\tTimerID: timerID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(10), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestCancelTimer_RespondDecisionTaskCompleted_NoStartTimer() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 2,\n\t})\n\tidentity := \"testIdentity\"\n\ttimerID := \"t1\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\t// Verify cancel timer with a start event.\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCancelTimer.Ptr(),\n\t\tCancelTimerDecisionAttributes: &types.CancelTimerDecisionAttributes{\n\t\t\tTimerID: timerID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err := s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(6), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(3), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n}\n\nfunc (s *engineSuite) TestCancelTimer_RespondDecisionTaskCompleted_TimerFired() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\ttaskToken, _ := json.Marshal(&common.TaskToken{\n\t\tWorkflowID: we.WorkflowID,\n\t\tRunID:      we.RunID,\n\t\tScheduleID: 6,\n\t})\n\tidentity := \"testIdentity\"\n\ttimerID := \"t1\"\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\t// Verify cancel timer with a start event.\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tl, []byte(\"input\"), 100, 100, identity, nil)\n\tdi := test.AddDecisionTaskScheduledEvent(msBuilder)\n\tdecisionStartedEvent := test.AddDecisionTaskStartedEvent(msBuilder, di.ScheduleID, tl, identity)\n\tdecisionCompletedEvent := test.AddDecisionTaskCompletedEvent(msBuilder, di.ScheduleID,\n\t\tdecisionStartedEvent.ID, nil, identity)\n\ttest.AddTimerStartedEvent(msBuilder, decisionCompletedEvent.ID, timerID, 10)\n\tdi2 := test.AddDecisionTaskScheduledEvent(msBuilder)\n\ttest.AddDecisionTaskStartedEvent(msBuilder, di2.ScheduleID, tl, identity)\n\ttest.AddTimerFiredEvent(msBuilder, timerID)\n\t_, _, err := msBuilder.CloseTransactionAsMutation(time.Now(), execution.TransactionPolicyActive)\n\ts.Nil(err)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\ts.True(len(gwmsResponse.State.BufferedEvents) > 0)\n\n\tdecisions := []*types.Decision{{\n\t\tDecisionType: types.DecisionTypeCancelTimer.Ptr(),\n\t\tCancelTimerDecisionAttributes: &types.CancelTimerDecisionAttributes{\n\t\t\tTimerID: timerID,\n\t\t},\n\t}}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(input *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\t// need to check whether the buffered events are cleared\n\t\ts.True(input.UpdateWorkflowMutation.ClearBufferedEvents)\n\t\treturn true\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.mockHistoryEngine.RespondDecisionTaskCompleted(context.Background(), &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken:        taskToken,\n\t\t\tDecisions:        decisions,\n\t\t\tExecutionContext: []byte(\"context\"),\n\t\t\tIdentity:         identity,\n\t\t},\n\t})\n\ts.Nil(err)\n\n\texecutionBuilder := s.getBuilder(constants.TestDomainID, we)\n\ts.Equal(int64(10), executionBuilder.GetExecutionInfo().NextEventID)\n\ts.Equal(int64(7), executionBuilder.GetExecutionInfo().LastProcessedEvent)\n\ts.Equal(persistence.WorkflowStateRunning, executionBuilder.GetExecutionInfo().State)\n\ts.False(executionBuilder.HasPendingDecision())\n\ts.False(executionBuilder.HasBufferedEvents())\n}\n\nfunc (s *engineSuite) TestSignalWorkflowExecution() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\ts.Nil(err)\n}\n\n// Test signal decision by adding request ID\nfunc (s *engineSuite) TestSignalWorkflowExecution_DuplicateRequest_WorkflowOpen() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name 2\"\n\tinput := []byte(\"test input 2\")\n\trequestID := uuid.New()\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t\tRequestID:         requestID,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t// assume duplicate request id\n\tms.SignalRequestedIDs = make(map[string]struct{})\n\tms.SignalRequestedIDs[requestID] = struct{}{}\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestSignalWorkflowExecution_DuplicateRequest_WorkflowCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name 2\"\n\tinput := []byte(\"test input 2\")\n\trequestID := uuid.New()\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t\tRequestID:         requestID,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\t// assume duplicate request id\n\tms.SignalRequestedIDs = make(map[string]struct{})\n\tms.SignalRequestedIDs[requestID] = struct{}{}\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tms.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestSignalWorkflowExecution_WorkflowCompleted() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := &types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t},\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, *we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\ts.EqualError(err, \"workflow execution already completed\")\n}\n\nfunc (s *engineSuite) TestSignalWorkflowExecution_DelayStart_NoDecisionScheduled() {\n\t// 1. Setup Cluster Info\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\t// 2. Setup Request Data\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t},\n\t}\n\n\t// 3. Build Mutable State - simulates DelayStart by NOT adding a decision task\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\t// Only add start event, no decision task scheduled — simulates DelayStart waiting\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tvar updateReq *persistence.UpdateWorkflowExecutionRequest\n\n\t// 4. Setup Mocks - capture the update request to verify DecisionScheduleID\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\tRun(func(args mock.Arguments) {\n\t\t\treq, _ := args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\tupdateReq = req\n\t\t}).\n\t\tReturn(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).\n\t\tOnce()\n\n\t// 5. Run the Signal Call\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\n\t// 6. Assertions\n\ts.Nil(err)\n\ts.NotNil(updateReq)\n\ts.NotNil(updateReq.UpdateWorkflowMutation)\n\ts.NotNil(updateReq.UpdateWorkflowMutation.ExecutionInfo)\n\n\t// Verify that NO decision was scheduled (ID is empty) because workflow hasn't processed first decision\n\ts.Equal(commonconstants.EmptyEventID, updateReq.UpdateWorkflowMutation.ExecutionInfo.DecisionScheduleID)\n}\n\nfunc (s *engineSuite) TestSignalWorkflowExecution_AfterFirstDecision_DecisionScheduled() {\n\t// Test that signals DO schedule a decision when workflow has already processed its first decision\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            constants.TestDomainID,\n\t\t\tWorkflowExecution: &we,\n\t\t\tIdentity:          identity,\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             input,\n\t\t},\n\t}\n\n\t// Build mutable state with a pending decision (proves HasProcessedOrPendingDecision returns true)\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\twe.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tvar updateReq *persistence.UpdateWorkflowExecutionRequest\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\tRun(func(args mock.Arguments) {\n\t\t\treq, _ := args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\tupdateReq = req\n\t\t}).\n\t\tReturn(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).\n\t\tOnce()\n\n\terr := s.mockHistoryEngine.SignalWorkflowExecution(context.Background(), signalRequest)\n\n\ts.Nil(err)\n\ts.NotNil(updateReq)\n\ts.NotNil(updateReq.UpdateWorkflowMutation)\n\ts.NotNil(updateReq.UpdateWorkflowMutation.ExecutionInfo)\n\n\t// Verify that a decision WAS scheduled because workflow has processed first decision\n\ts.NotEqual(commonconstants.EmptyEventID, updateReq.UpdateWorkflowMutation.ExecutionInfo.DecisionScheduleID)\n}\n\nfunc (s *engineSuite) TestSignalWithStartWorkflowExecution_DelayStart_NoDecisionScheduled() {\n\t// Test SignalWithStart on existing workflow waiting for DelayStart - should NOT schedule decision\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"wId\"\n\trunID := constants.TestRunID\n\tidentity := \"testIdentity\"\n\tsignalName := \"my signal name\"\n\tinput := []byte(\"test input\")\n\n\tsRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:     domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tIdentity:   identity,\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t},\n\t}\n\n\t// Build mutable state simulating DelayStart - no decision task\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\trunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\t// Add workflow started event but no decision task\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, we, \"wType\", \"testTaskList\", []byte(\"input\"), 100, 200, identity, nil)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: runID}\n\tvar updateReq *persistence.UpdateWorkflowExecutionRequest\n\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\tRun(func(args mock.Arguments) {\n\t\t\treq, _ := args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\tupdateReq = req\n\t\t}).\n\t\tReturn(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).\n\t\tOnce()\n\n\tresp, err := s.mockHistoryEngine.SignalWithStartWorkflowExecution(context.Background(), sRequest)\n\n\ts.Nil(err)\n\ts.Equal(runID, resp.GetRunID())\n\ts.NotNil(updateReq)\n\ts.NotNil(updateReq.UpdateWorkflowMutation)\n\ts.NotNil(updateReq.UpdateWorkflowMutation.ExecutionInfo)\n\n\t// Verify that NO decision was scheduled because workflow hasn't processed first decision\n\ts.Equal(commonconstants.EmptyEventID, updateReq.UpdateWorkflowMutation.ExecutionInfo.DecisionScheduleID)\n}\n\nfunc (s *engineSuite) TestRemoveSignalMutableState() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockHistoryEngine.clusterMetadata.GetCurrentClusterName(),\n\t\tFailoverVersion:   0,\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttasklist := \"testTaskList\"\n\tidentity := \"testIdentity\"\n\trequestID := uuid.New()\n\tremoveRequest := &types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID:        constants.TestDomainID,\n\t\tWorkflowExecution: &workflowExecution,\n\t\tRequestID:         requestID,\n\t}\n\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tconstants.TestRunID,\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\ttest.AddWorkflowExecutionStartedEvent(msBuilder, workflowExecution, \"wType\", tasklist, []byte(\"input\"), 100, 200, identity, nil)\n\ttest.AddDecisionTaskScheduledEvent(msBuilder)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.DomainID = constants.TestDomainID\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\terr := s.mockHistoryEngine.RemoveSignalMutableState(context.Background(), removeRequest)\n\ts.Nil(err)\n}\n\nfunc (s *engineSuite) TestReapplyEvents_ReturnSuccess() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-reapply\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\thistory := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        1,\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\tVersion:   1,\n\t\t},\n\t}\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockEventsReapplier.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\n\terr := s.mockHistoryEngine.ReapplyEvents(\n\t\tcontext.Background(),\n\t\tconstants.TestDomainID,\n\t\tworkflowExecution.WorkflowID,\n\t\tworkflowExecution.RunID,\n\t\thistory,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *engineSuite) TestReapplyEvents_IgnoreSameVersionEvents() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-reapply-same-version\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\thistory := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        1,\n\t\t\tEventType: types.EventTypeTimerStarted.Ptr(),\n\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t},\n\t}\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockEventsReapplier.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(0)\n\n\terr := s.mockHistoryEngine.ReapplyEvents(\n\t\tcontext.Background(),\n\t\tconstants.TestDomainID,\n\t\tworkflowExecution.WorkflowID,\n\t\tworkflowExecution.RunID,\n\t\thistory,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *engineSuite) TestReapplyEvents_ResetWorkflow() {\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestLocalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestLocalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"test-reapply-reset-workflow\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\thistory := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        1,\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\tVersion:   100,\n\t\t},\n\t}\n\tmsBuilder := execution.NewMutableStateBuilderWithEventV2(\n\t\ts.mockHistoryEngine.shard,\n\t\ttestlogger.New(s.Suite.T()),\n\t\tworkflowExecution.GetRunID(),\n\t\tconstants.TestLocalDomainEntry,\n\t)\n\tms := execution.CreatePersistenceMutableState(s.T(), msBuilder)\n\tms.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\tms.ExecutionInfo.LastProcessedEvent = 1\n\ttoken, err := msBuilder.GetCurrentBranchToken()\n\ts.NoError(err)\n\titem := persistence.NewVersionHistoryItem(1, 1)\n\tversionHistory := persistence.NewVersionHistory(token, []*persistence.VersionHistoryItem{item})\n\tms.VersionHistories = persistence.NewVersionHistories(versionHistory)\n\tgwmsResponse := &persistence.GetWorkflowExecutionResponse{State: ms}\n\tgceResponse := &persistence.GetCurrentExecutionResponse{RunID: constants.TestRunID}\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(gceResponse, nil).Once()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(gwmsResponse, nil).Once()\n\ts.mockEventsReapplier.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(0)\n\ts.mockWorkflowResetter.EXPECT().ResetWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\tgomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(),\n\t\tgomock.Any(),\n\t).Return(nil).Times(1)\n\terr = s.mockHistoryEngine.ReapplyEvents(\n\t\tcontext.Background(),\n\t\tconstants.TestDomainID,\n\t\tworkflowExecution.WorkflowID,\n\t\tworkflowExecution.RunID,\n\t\thistory,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *engineSuite) getBuilder(testDomainID string, we types.WorkflowExecution) execution.MutableState {\n\tcontext, release, err := s.mockHistoryEngine.executionCache.GetOrCreateWorkflowExecutionForBackground(testDomainID, we)\n\tif err != nil {\n\t\treturn nil\n\t}\n\tdefer release(nil)\n\n\treturn context.GetWorkflowExecution()\n}\n\nfunc (s *engineSuite) getActivityScheduledEvent(\n\tmsBuilder execution.MutableState,\n\tscheduleID int64,\n) *types.HistoryEvent {\n\tevent, _ := msBuilder.GetActivityScheduledEvent(context.Background(), scheduleID)\n\treturn event\n}\n\nfunc (s *engineSuite) printHistory(builder execution.MutableState) string {\n\treturn thrift.FromHistory(builder.GetHistoryBuilder().GetHistory()).String()\n}\n\nfunc TestRecordChildExecutionCompleted(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\trequest   *types.RecordChildExecutionCompletedRequest\n\t\tmockSetup func(*execution.MockMutableState)\n\t\twantErr   bool\n\t\tassertErr func(t *testing.T, err error)\n\t}{\n\t\t{\n\t\t\tname: \"workflow not running\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 1,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(false)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, workflow.ErrNotExists, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"stale mutable state - not containing initiated event\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 10,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\tms.EXPECT().GetChildExecutionInfo(int64(10)).Return(nil, false)\n\t\t\t\tms.EXPECT().GetNextEventID().Return(int64(10)).Times(2)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, workflow.ErrStaleState, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"pending child execution not found\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 10,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\tms.EXPECT().GetChildExecutionInfo(int64(10)).Return(nil, false)\n\t\t\t\tms.EXPECT().GetNextEventID().Return(int64(11)).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.EntityNotExistsError{Message: \"Pending child execution not found.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"stale mutable state - not containing started event\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 10,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\tms.EXPECT().GetChildExecutionInfo(int64(10)).Return(&persistence.ChildExecutionInfo{StartedID: commonconstants.EmptyEventID}, true)\n\t\t\t\tms.EXPECT().GetNextEventID().Return(int64(11)).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, workflow.ErrStaleState, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"pending child execution workflowID mismatch\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 10,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t\tStartedID: 11,\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\tms.EXPECT().GetChildExecutionInfo(int64(10)).Return(&persistence.ChildExecutionInfo{StartedID: 11, StartedWorkflowID: \"wid0\"}, true)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.EntityNotExistsError{Message: \"Pending child execution workflowID mismatch.\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - child workflow completed\",\n\t\t\trequest: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID:  \"58f7998e-9c00-4827-bbd3-6a891b3ca0ca\",\n\t\t\t\tInitiatedID: 10,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\tRunID:      \"4353ddce-ca34-4c78-9785-dc0b83af4bbc\",\n\t\t\t\t},\n\t\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wid1\",\n\t\t\t\t\tRunID:      \"312b2440-2859-4e50-a59f-d92a300a072d\",\n\t\t\t\t},\n\t\t\t\tCompletionEvent: &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\t\tResult: []byte(\"success\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tStartedID: 11,\n\t\t\t},\n\t\t\tmockSetup: func(ms *execution.MockMutableState) {\n\t\t\t\tms.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\t\t\t\tms.EXPECT().GetChildExecutionInfo(int64(10)).Return(&persistence.ChildExecutionInfo{StartedID: 11, StartedWorkflowID: \"wid1\"}, true)\n\t\t\t\tms.EXPECT().AddChildWorkflowExecutionCompletedEvent(int64(10), gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewTestContext(t, ctrl, &persistence.ShardInfo{}, config.NewForTest())\n\t\t\tmockDomainCache := mockShard.Resource.DomainCache\n\t\t\tmockActiveClusterManager := mockShard.Resource.ActiveClusterMgr\n\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{ActiveClusterName: mockShard.GetClusterMetadata().GetCurrentClusterName(), FailoverVersion: 0}, nil).Times(1)\n\t\t\ttestDomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: constants.TestDomainID}, &persistence.DomainConfig{}, \"\",\n\t\t\t)\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(testDomainEntry, nil).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainID, nil).AnyTimes()\n\t\t\tms := execution.NewMockMutableState(ctrl)\n\t\t\ttc.mockSetup(ms)\n\n\t\t\thistoryEngine := &historyEngineImpl{\n\t\t\t\tshard:           mockShard,\n\t\t\t\tclusterMetadata: mockShard.GetClusterMetadata(),\n\t\t\t\ttimeSource:      mockShard.GetTimeSource(),\n\t\t\t\tmetricsClient:   metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\t\t\tlogger:          mockShard.GetLogger(),\n\t\t\t\tupdateWithActionFn: func(_ context.Context, _ log.Logger, _ execution.Cache, _ string, _ types.WorkflowExecution, _ bool, _ time.Time, actionFn func(wfContext execution.Context, mutableState execution.MutableState) error) error {\n\t\t\t\t\treturn actionFn(nil, ms)\n\t\t\t\t},\n\t\t\t\tactiveClusterManager: mockActiveClusterManager,\n\t\t\t}\n\n\t\t\terr := historyEngine.RecordChildExecutionCompleted(context.Background(), tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\ttc.assertErr(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_createShardNameFromShardID(t *testing.T) {\n\tshardID := 1\n\tshardName := createShardNameFromShardID(shardID)\n\tassert.Equal(t, fmt.Sprintf(\"history-engine-%d\", shardID), shardName)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/notify_tasks.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/replication\"\n)\n\nfunc (e *historyEngineImpl) NotifyNewHistoryEvent(event *events.Notification) {\n\te.historyEventNotifier.NotifyNewHistoryEvent(event)\n}\n\nfunc (e *historyEngineImpl) NotifyNewTransferTasks(info *hcommon.NotifyTaskInfo) {\n\tif len(info.Tasks) == 0 {\n\t\treturn\n\t}\n\n\ttask := info.Tasks[0]\n\tclusterName, err := e.shard.GetClusterMetadata().ClusterNameForFailoverVersion(task.GetVersion())\n\tif err == nil {\n\t\ttransferProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTransfer]\n\t\tif !ok {\n\t\t\te.logger.Error(\"transfer processor not found\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\ttransferProcessor.NotifyNewTask(clusterName, info)\n\t}\n}\n\nfunc (e *historyEngineImpl) NotifyNewTimerTasks(info *hcommon.NotifyTaskInfo) {\n\tif len(info.Tasks) == 0 {\n\t\treturn\n\t}\n\n\ttask := info.Tasks[0]\n\tclusterName, err := e.shard.GetClusterMetadata().ClusterNameForFailoverVersion(task.GetVersion())\n\tif err == nil {\n\t\ttimerProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTimer]\n\t\tif !ok {\n\t\t\te.logger.Error(\"timer processor not found\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\t\ttimerProcessor.NotifyNewTask(clusterName, info)\n\t}\n}\n\nfunc (e *historyEngineImpl) NotifyNewReplicationTasks(info *hcommon.NotifyTaskInfo) {\n\tfor _, task := range info.Tasks {\n\t\thTask, err := hydrateReplicationTask(task, info.ExecutionInfo, info.VersionHistories, info.Activities, info.History)\n\t\tif err != nil {\n\t\t\te.logger.Error(\"failed to preemptively hydrate replication task\", tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\t\te.replicationTaskStore.Put(hTask)\n\t}\n}\n\nfunc hydrateReplicationTask(\n\ttask persistence.Task,\n\texec *persistence.WorkflowExecutionInfo,\n\tversionHistories *persistence.VersionHistories,\n\tactivities map[int64]*persistence.ActivityInfo,\n\thistory events.PersistedBlobs,\n) (*types.ReplicationTask, error) {\n\tinfo := persistence.ReplicationTaskInfo{\n\t\tDomainID:     exec.DomainID,\n\t\tWorkflowID:   exec.WorkflowID,\n\t\tRunID:        exec.RunID,\n\t\tTaskType:     task.GetTaskType(),\n\t\tCreationTime: task.GetVisibilityTimestamp().UnixNano(),\n\t\tTaskID:       task.GetTaskID(),\n\t\tVersion:      task.GetVersion(),\n\t}\n\n\tswitch t := task.(type) {\n\tcase *persistence.HistoryReplicationTask:\n\t\tinfo.BranchToken = t.BranchToken\n\t\tinfo.NewRunBranchToken = t.NewRunBranchToken\n\t\tinfo.FirstEventID = t.FirstEventID\n\t\tinfo.NextEventID = t.NextEventID\n\tcase *persistence.SyncActivityTask:\n\t\tinfo.ScheduledID = t.ScheduledID\n\tcase *persistence.FailoverMarkerTask:\n\t\t// No specific fields, but supported\n\tdefault:\n\t\treturn nil, errors.New(\"unknown replication task\")\n\t}\n\n\thydrator := replication.NewImmediateTaskHydrator(\n\t\texec.IsRunning(),\n\t\tversionHistories,\n\t\tactivities,\n\t\thistory.Find(info.BranchToken, info.FirstEventID),\n\t\thistory.Find(info.NewRunBranchToken, constants.FirstEventID),\n\t)\n\n\treturn hydrator.Hydrate(context.Background(), task)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/poll_mutable_state.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// GetMutableState retrieves the mutable state of the workflow execution\nfunc (e *historyEngineImpl) GetMutableState(ctx context.Context, request *types.GetMutableStateRequest) (*types.GetMutableStateResponse, error) {\n\treturn e.getMutableStateOrLongPoll(ctx, request)\n}\n\n// PollMutableState retrieves the mutable state of the workflow execution with long polling\nfunc (e *historyEngineImpl) PollMutableState(ctx context.Context, request *types.PollMutableStateRequest) (*types.PollMutableStateResponse, error) {\n\tresponse, err := e.getMutableStateOrLongPoll(ctx, &types.GetMutableStateRequest{\n\t\tDomainUUID:          request.DomainUUID,\n\t\tExecution:           request.Execution,\n\t\tExpectedNextEventID: request.ExpectedNextEventID,\n\t\tCurrentBranchToken:  request.CurrentBranchToken,\n\t\tVersionHistoryItem:  request.GetVersionHistoryItem(),\n\t})\n\n\tif err != nil {\n\t\treturn nil, e.updateEntityNotExistsErrorOnPassiveCluster(ctx, err, request.GetDomainUUID(), request.Execution)\n\t}\n\n\treturn &types.PollMutableStateResponse{\n\t\tExecution:                            response.Execution,\n\t\tWorkflowType:                         response.WorkflowType,\n\t\tNextEventID:                          response.NextEventID,\n\t\tPreviousStartedEventID:               response.PreviousStartedEventID,\n\t\tLastFirstEventID:                     response.LastFirstEventID,\n\t\tTaskList:                             response.TaskList,\n\t\tStickyTaskList:                       response.StickyTaskList,\n\t\tClientLibraryVersion:                 response.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 response.ClientFeatureVersion,\n\t\tClientImpl:                           response.ClientImpl,\n\t\tStickyTaskListScheduleToStartTimeout: response.StickyTaskListScheduleToStartTimeout,\n\t\tCurrentBranchToken:                   response.CurrentBranchToken,\n\t\tVersionHistories:                     response.VersionHistories,\n\t\tWorkflowState:                        response.WorkflowState,\n\t\tWorkflowCloseState:                   response.WorkflowCloseState,\n\t}, nil\n}\n\nfunc (e *historyEngineImpl) getMutableState(\n\tctx context.Context,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (retResp *types.GetMutableStateResponse, retError error) {\n\n\twfContext, release, retError := e.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, execution)\n\tif retError != nil {\n\t\treturn\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, retError := wfContext.LoadWorkflowExecution(ctx)\n\tif retError != nil {\n\t\treturn\n\t}\n\n\tcurrentBranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecution.RunID = wfContext.GetExecution().RunID\n\tworkflowState, workflowCloseState := mutableState.GetWorkflowStateCloseStatus()\n\tretResp = &types.GetMutableStateResponse{\n\t\tExecution:                            &execution,\n\t\tWorkflowType:                         &types.WorkflowType{Name: executionInfo.WorkflowTypeName},\n\t\tLastFirstEventID:                     mutableState.GetLastFirstEventID(),\n\t\tNextEventID:                          mutableState.GetNextEventID(),\n\t\tPreviousStartedEventID:               common.Int64Ptr(mutableState.GetPreviousStartedEventID()),\n\t\tTaskList:                             &types.TaskList{Name: executionInfo.TaskList},\n\t\tStickyTaskList:                       &types.TaskList{Name: executionInfo.StickyTaskList, Kind: types.TaskListKindSticky.Ptr()},\n\t\tClientLibraryVersion:                 executionInfo.ClientLibraryVersion,\n\t\tClientFeatureVersion:                 executionInfo.ClientFeatureVersion,\n\t\tClientImpl:                           executionInfo.ClientImpl,\n\t\tIsWorkflowRunning:                    mutableState.IsWorkflowExecutionRunning(),\n\t\tStickyTaskListScheduleToStartTimeout: common.Int32Ptr(executionInfo.StickyScheduleToStartTimeout),\n\t\tCurrentBranchToken:                   currentBranchToken,\n\t\tWorkflowState:                        common.Int32Ptr(int32(workflowState)),\n\t\tWorkflowCloseState:                   common.Int32Ptr(int32(workflowCloseState)),\n\t\tIsStickyTaskListEnabled:              mutableState.IsStickyTaskListEnabled(),\n\t\tHistorySize:                          mutableState.GetHistorySize(),\n\t}\n\tversionHistories := mutableState.GetVersionHistories()\n\tif versionHistories != nil {\n\t\tretResp.VersionHistories = versionHistories.ToInternalType()\n\t}\n\treturn\n}\n\nfunc (e *historyEngineImpl) updateEntityNotExistsErrorOnPassiveCluster(ctx context.Context, err error, domainID string, execution *types.WorkflowExecution) error {\n\tswitch err.(type) {\n\tcase *types.EntityNotExistsError:\n\t\tactiveClusterInfo, activeClusterErr := e.activeClusterManager.GetActiveClusterInfoByWorkflow(ctx, domainID, execution.GetWorkflowID(), execution.GetRunID())\n\t\tif activeClusterErr != nil {\n\t\t\treturn err // return original error if could not access active cluster manager\n\t\t}\n\t\tif activeClusterInfo.ActiveClusterName != e.clusterMetadata.GetCurrentClusterName() {\n\t\t\treturn &types.EntityNotExistsError{\n\t\t\t\tMessage:        \"Workflow execution not found in non-active cluster\",\n\t\t\t\tActiveCluster:  activeClusterInfo.ActiveClusterName,\n\t\t\t\tCurrentCluster: e.clusterMetadata.GetCurrentClusterName(),\n\t\t\t}\n\t\t}\n\t}\n\treturn err\n}\n\nfunc (e *historyEngineImpl) getMutableStateWithCurrentVersionHistory(\n\tctx context.Context,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (*types.GetMutableStateResponse, *persistence.VersionHistory, error) {\n\n\tmutableState, err := e.getMutableState(ctx, domainID, execution)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\t// must be corrupted workflow\n\tif mutableState.GetVersionHistories() == nil {\n\t\te.logger.Warn(\"version histories do not exist\")\n\t\treturn nil, nil, &types.InternalDataInconsistencyError{Message: \"version histories do not exist\"}\n\t}\n\n\tcurrentVersionHistory, err := persistence.NewVersionHistoriesFromInternalType(\n\t\tmutableState.GetVersionHistories(),\n\t).GetCurrentVersionHistory()\n\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"unable to get version history while looking up response for mutable state: %w\", err)\n\t}\n\n\treturn mutableState, currentVersionHistory, err\n}\n\nfunc (e *historyEngineImpl) getMutableStateOrLongPoll(\n\tctx context.Context,\n\trequest *types.GetMutableStateRequest,\n) (*types.GetMutableStateResponse, error) {\n\n\tif err := common.ValidateDomainUUID(request.DomainUUID); err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := request.DomainUUID\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: request.Execution.WorkflowID,\n\t\tRunID:      request.Execution.RunID,\n\t}\n\n\tmutableState, currentVersionHistory, err := e.getMutableStateWithCurrentVersionHistory(ctx, domainID, execution)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif request.VersionHistoryItem == nil {\n\t\tlastVersionHistoryItem, err := currentVersionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trequest.VersionHistoryItem = lastVersionHistoryItem.ToInternalType()\n\t}\n\t// Use the latest event id + event version as the branch identifier. This pair is unique across clusters.\n\t// We return the full version histories. Callers need to fetch the last version history item from current branch\n\t// and use the last version history item in following calls.\n\tif !currentVersionHistory.ContainsItem(persistence.NewVersionHistoryItemFromInternalType(request.VersionHistoryItem)) {\n\t\te.logger.Warn(\"current version history and requested one are different - found on first check\",\n\t\t\ttag.Dynamic(\"current-version-history\", currentVersionHistory),\n\t\t\ttag.Dynamic(\"requested-version-history\", request.VersionHistoryItem),\n\t\t)\n\t\treturn nil, &types.CurrentBranchChangedError{\n\t\t\tMessage:            \"current branch token and request branch token doesn't match\",\n\t\t\tCurrentBranchToken: mutableState.CurrentBranchToken}\n\t}\n\t// set the run id in case query the current running workflow\n\texecution.RunID = mutableState.Execution.RunID\n\n\t// expectedNextEventID is 0 when caller want to get the current next event ID without blocking\n\texpectedNextEventID := constants.FirstEventID\n\tif request.ExpectedNextEventID != 0 {\n\t\texpectedNextEventID = request.GetExpectedNextEventID()\n\t}\n\n\tif expectedNextEventID < mutableState.GetNextEventID() || !mutableState.GetIsWorkflowRunning() {\n\t\treturn mutableState, nil\n\t}\n\t// if caller decide to long poll on workflow execution\n\t// and the event ID we are looking for is smaller than current next event ID\n\treturn e.longPollForEventID(ctx, expectedNextEventID, domainID, execution, request.VersionHistoryItem)\n}\n\nfunc (e *historyEngineImpl) longPollForEventID(\n\tctx context.Context,\n\texpectedNextEventID int64,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\tversionHistoryItem *types.VersionHistoryItem,\n) (*types.GetMutableStateResponse, error) {\n\n\twfIdentifier := definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID())\n\tsubscriberID, channel, err := e.historyEventNotifier.WatchHistoryEvent(wfIdentifier)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer e.historyEventNotifier.UnwatchHistoryEvent(wfIdentifier, subscriberID) //nolint:errcheck\n\n\t// check again in case the next event ID is updated before we subscribed\n\tmutableState, currentVersionHistory, err := e.getMutableStateWithCurrentVersionHistory(ctx, domainID, execution)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif !currentVersionHistory.ContainsItem(persistence.NewVersionHistoryItemFromInternalType(versionHistoryItem)) {\n\t\te.logger.Warn(\"current version history and requested one are different - found on checking mutableState\",\n\t\t\ttag.Dynamic(\"current-version-history\", currentVersionHistory),\n\t\t\ttag.Dynamic(\"requested-version-history\", versionHistoryItem),\n\t\t)\n\t\treturn nil, &types.CurrentBranchChangedError{\n\t\t\tMessage:            \"current branch token and request branch token doesn't match\",\n\t\t\tCurrentBranchToken: mutableState.CurrentBranchToken,\n\t\t}\n\t}\n\n\tif expectedNextEventID < mutableState.GetNextEventID() || !mutableState.GetIsWorkflowRunning() {\n\t\treturn mutableState, nil\n\t}\n\n\tdomainName, err := e.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\texpirationInterval := e.shard.GetConfig().LongPollExpirationInterval(domainName)\n\tif deadline, ok := ctx.Deadline(); ok {\n\t\tremainingTime := deadline.Sub(e.shard.GetTimeSource().Now())\n\t\t// Here we return a safeguard error, to ensure that older clients are not stuck in long poll loop until context fully expires.\n\t\t// Otherwise, it results in multiple additional requests being made that returns empty responses.\n\t\t// Newer clients will not make request with too small timeout remaining.\n\t\tif remainingTime < longPollCompletionBuffer {\n\t\t\treturn nil, context.DeadlineExceeded\n\t\t}\n\t\t// longPollCompletionBuffer is here to leave some room to finish current request without its timeout.\n\t\texpirationInterval = min(\n\t\t\texpirationInterval,\n\t\t\tremainingTime-longPollCompletionBuffer,\n\t\t)\n\t}\n\tif expirationInterval <= 0 {\n\t\treturn mutableState, nil\n\t}\n\ttimer := time.NewTimer(expirationInterval)\n\tdefer timer.Stop()\n\tfor {\n\t\tselect {\n\t\tcase event := <-channel:\n\t\t\tmutableState.LastFirstEventID = event.LastFirstEventID\n\t\t\tmutableState.NextEventID = event.NextEventID\n\t\t\tmutableState.IsWorkflowRunning = event.WorkflowCloseState == persistence.WorkflowCloseStatusNone\n\t\t\tmutableState.PreviousStartedEventID = common.Int64Ptr(event.PreviousStartedEventID)\n\t\t\tmutableState.WorkflowState = common.Int32Ptr(int32(event.WorkflowState))\n\t\t\tmutableState.WorkflowCloseState = common.Int32Ptr(int32(event.WorkflowCloseState))\n\n\t\t\tcurrentVersionHistoryOnPoll, err := event.VersionHistories.GetCurrentVersionHistory()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"unexpected error getting current version history while polling for mutable state: %w\", err)\n\t\t\t}\n\t\t\tif !currentVersionHistoryOnPoll.ContainsItem(persistence.NewVersionHistoryItemFromInternalType(versionHistoryItem)) {\n\t\t\t\te.logger.Warn(\"current version history and requested one are different\",\n\t\t\t\t\ttag.Dynamic(\"current-version-history\", currentVersionHistoryOnPoll),\n\t\t\t\t\ttag.Dynamic(\"requested-version-history\", versionHistoryItem),\n\t\t\t\t)\n\t\t\t\treturn nil, &types.CurrentBranchChangedError{\n\t\t\t\t\tMessage:            \"current and requested version histories don't match - changed while polling\",\n\t\t\t\t\tCurrentBranchToken: mutableState.CurrentBranchToken,\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif expectedNextEventID < mutableState.GetNextEventID() || !mutableState.GetIsWorkflowRunning() {\n\t\t\t\treturn mutableState, nil\n\t\t\t}\n\t\tcase <-timer.C:\n\t\t\treturn mutableState, nil\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/query_workflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"strings\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc (e *historyEngineImpl) QueryWorkflow(\n\tctx context.Context,\n\trequest *types.HistoryQueryWorkflowRequest,\n) (retResp *types.HistoryQueryWorkflowResponse, retErr error) {\n\n\tscope := e.metricsClient.Scope(metrics.HistoryQueryWorkflowScope).Tagged(metrics.DomainTag(request.GetRequest().GetDomain()))\n\tshardMetricScope := e.metricsClient.Scope(metrics.HistoryQueryWorkflowScope, metrics.ShardIDTag(e.shard.GetShardID()))\n\n\tconsistentQueryEnabled := e.config.EnableConsistentQuery() && e.config.EnableConsistentQueryByDomain(request.GetRequest().GetDomain())\n\tif request.GetRequest().GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong {\n\t\tif !consistentQueryEnabled {\n\t\t\treturn nil, workflow.ErrConsistentQueryNotEnabled\n\t\t}\n\t\tshardMetricScope.IncCounter(metrics.ConsistentQueryPerShard)\n\t\te.logger.SampleInfo(\"History QueryWorkflow called with QueryConsistencyLevelStrong\", e.config.SampleLoggingRate(), tag.ShardID(e.shard.GetShardID()), tag.WorkflowID(request.GetRequest().Execution.WorkflowID), tag.WorkflowDomainName(request.GetRequest().Domain))\n\t}\n\n\texecution := *request.GetRequest().GetExecution()\n\n\tmutableStateResp, err := e.getMutableState(ctx, request.GetDomainUUID(), execution)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treq := request.GetRequest()\n\tif !mutableStateResp.GetIsWorkflowRunning() && req.QueryRejectCondition != nil {\n\t\tnotOpenReject := req.GetQueryRejectCondition() == types.QueryRejectConditionNotOpen\n\t\tcloseStatus := mutableStateResp.GetWorkflowCloseState()\n\t\tnotCompletedCleanlyReject := req.GetQueryRejectCondition() == types.QueryRejectConditionNotCompletedCleanly && closeStatus != persistence.WorkflowCloseStatusCompleted\n\t\tif notOpenReject || notCompletedCleanlyReject {\n\t\t\treturn &types.HistoryQueryWorkflowResponse{\n\t\t\t\tResponse: &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryRejected: &types.QueryRejected{\n\t\t\t\t\t\tCloseStatus: persistence.ToInternalWorkflowExecutionCloseStatus(int(closeStatus)),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\t}\n\n\t// query cannot be processed unless at least one decision task has finished\n\t// if first decision task has not finished wait for up to a second for it to complete\n\tqueryFirstDecisionTaskWaitTime := defaultQueryFirstDecisionTaskWaitTime\n\tctxDeadline, ok := ctx.Deadline()\n\tif ok {\n\t\tctxWaitTime := time.Until(ctxDeadline) - time.Second\n\t\tif ctxWaitTime > queryFirstDecisionTaskWaitTime {\n\t\t\tqueryFirstDecisionTaskWaitTime = ctxWaitTime\n\t\t}\n\t}\n\tdeadline := time.Now().Add(queryFirstDecisionTaskWaitTime)\n\tfor mutableStateResp.GetPreviousStartedEventID() <= 0 && time.Now().Before(deadline) {\n\t\ttime.Sleep(queryFirstDecisionTaskCheckInterval)\n\t\tmutableStateResp, err = e.getMutableState(ctx, request.GetDomainUUID(), execution)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tif mutableStateResp.GetPreviousStartedEventID() <= 0 {\n\t\tscope.IncCounter(metrics.QueryBeforeFirstDecisionCount)\n\t\treturn nil, workflow.ErrQueryWorkflowBeforeFirstDecision\n\t}\n\n\tde, err := e.shard.GetDomainCache().GetDomainByID(request.GetDomainUUID())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\twfContext, release, err := e.executionCache.GetOrCreateWorkflowExecution(ctx, request.GetDomainUUID(), execution)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer func() { release(retErr) }()\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// If history is corrupted, query will be rejected\n\tif corrupted, err := e.checkForHistoryCorruptions(ctx, mutableState); err != nil {\n\t\treturn nil, err\n\t} else if corrupted {\n\t\treturn nil, &types.EntityNotExistsError{Message: \"Workflow execution corrupted.\"}\n\t}\n\n\t// There are two ways in which queries get dispatched to decider. First, queries can be dispatched on decision tasks.\n\t// These decision tasks potentially contain new events and queries. The events are treated as coming before the query in time.\n\t// The second way in which queries are dispatched to decider is directly through matching; in this approach queries can be\n\t// dispatched to decider immediately even if there are outstanding events that came before the query. The following logic\n\t// is used to determine if a query can be safely dispatched directly through matching or if given the desired consistency\n\t// level must be dispatched on a decision task. There are four cases in which a query can be dispatched directly through\n\t// matching safely, without violating the desired consistency level:\n\t// 1. the domain is not active, in this case history is immutable so a query dispatched at any time is consistent\n\t// 2. the workflow is not running, whenever a workflow is not running dispatching query directly is consistent\n\t// 3. the client requested eventual consistency, in this case there are no consistency requirements so dispatching directly through matching is safe\n\t// 4. if there is no pending or started decision it means no events came before query arrived, so its safe to dispatch directly\n\tisActive := false\n\tclusterAttribute := mutableState.GetExecutionInfo().ActiveClusterSelectionPolicy.GetClusterAttribute()\n\tactiveClusterInfo, exists := de.GetActiveClusterInfoByClusterAttribute(clusterAttribute)\n\tif !exists {\n\t\t// TODO: this is only possible if the domain metadata doesn't have the cluster attribute, should we return an error instead?\n\t\te.logger.Warn(\"Failed to get active cluster info by cluster attribute, default to active\", tag.Dynamic(\"clusterAttribute\", clusterAttribute))\n\t\tisActive = true\n\t} else {\n\t\tisActive = activeClusterInfo.ActiveClusterName == e.clusterMetadata.GetCurrentClusterName()\n\t}\n\tsafeToDispatchDirectly := !isActive ||\n\t\t!mutableState.IsWorkflowExecutionRunning() ||\n\t\treq.GetQueryConsistencyLevel() == types.QueryConsistencyLevelEventual ||\n\t\t(!mutableState.HasPendingDecision() && !mutableState.HasInFlightDecision())\n\tif safeToDispatchDirectly {\n\t\trelease(nil)\n\t\tmsResp, err := e.getMutableState(ctx, request.GetDomainUUID(), execution)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treq.Execution.RunID = msResp.Execution.RunID\n\t\treturn e.queryDirectlyThroughMatching(ctx, msResp, request.GetDomainUUID(), req, scope, isActive)\n\t}\n\n\t// If we get here it means query could not be dispatched through matching directly, so it must block\n\t// until either an result has been obtained on a decision task response or until it is safe to dispatch directly through matching.\n\tsw := scope.StartTimer(metrics.DecisionTaskQueryLatency)\n\tdefer sw.Stop()\n\tqueryReg := mutableState.GetQueryRegistry()\n\tif len(queryReg.GetBufferedIDs()) >= e.config.MaxBufferedQueryCount() {\n\t\tscope.IncCounter(metrics.QueryBufferExceededCount)\n\t\treturn nil, workflow.ErrConsistentQueryBufferExceeded\n\t}\n\tqueryID, termCh := queryReg.BufferQuery(req.GetQuery())\n\tdefer queryReg.RemoveQuery(queryID)\n\trelease(nil)\n\tselect {\n\tcase <-termCh:\n\t\tstate, err := queryReg.GetTerminationState(queryID)\n\t\tif err != nil {\n\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\treturn nil, err\n\t\t}\n\t\tswitch state.TerminationType {\n\t\tcase query.TerminationTypeCompleted:\n\t\t\tresult := state.QueryResult\n\t\t\tswitch result.GetResultType() {\n\t\t\tcase types.QueryResultTypeAnswered:\n\t\t\t\treturn &types.HistoryQueryWorkflowResponse{\n\t\t\t\t\tResponse: &types.QueryWorkflowResponse{\n\t\t\t\t\t\tQueryResult: result.GetAnswer(),\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\tcase types.QueryResultTypeFailed:\n\t\t\t\treturn nil, &types.QueryFailedError{Message: result.GetErrorMessage()}\n\t\t\tdefault:\n\t\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\t\treturn nil, workflow.ErrQueryEnteredInvalidState\n\t\t\t}\n\t\tcase query.TerminationTypeUnblocked:\n\t\t\tmsResp, err := e.getMutableState(ctx, request.GetDomainUUID(), execution)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treq.Execution.RunID = msResp.Execution.RunID\n\t\t\treturn e.queryDirectlyThroughMatching(ctx, msResp, request.GetDomainUUID(), req, scope, isActive)\n\t\tcase query.TerminationTypeFailed:\n\t\t\treturn nil, state.Failure\n\t\tdefault:\n\t\t\tscope.IncCounter(metrics.QueryRegistryInvalidStateCount)\n\t\t\treturn nil, workflow.ErrQueryEnteredInvalidState\n\t\t}\n\tcase <-ctx.Done():\n\t\tscope.IncCounter(metrics.ConsistentQueryTimeoutCount)\n\t\treturn nil, ctx.Err()\n\t}\n}\n\nfunc (e *historyEngineImpl) queryDirectlyThroughMatching(\n\tctx context.Context,\n\tmsResp *types.GetMutableStateResponse,\n\tdomainID string,\n\tqueryRequest *types.QueryWorkflowRequest,\n\tscope metrics.Scope,\n\tisActive bool,\n) (*types.HistoryQueryWorkflowResponse, error) {\n\n\tsw := scope.StartTimer(metrics.DirectQueryDispatchLatency)\n\tdefer sw.Stop()\n\n\t// Sticky task list is not very useful in the standby cluster because the decider cache is\n\t// not updated by dispatching tasks to it (it is only updated in the case of query).\n\t// Additionally on the standby side we are not even able to clear sticky.\n\t// Stickiness might be outdated if the customer did a restart of their nodes causing a query\n\t// dispatched on the standby side on sticky to hang. We decided it made sense to simply not attempt\n\t// query on sticky task list at all on the passive side.\n\tsupportsStickyQuery := e.clientChecker.SupportsStickyQuery(msResp.GetClientImpl(), msResp.GetClientFeatureVersion()) == nil\n\tif msResp.GetIsStickyTaskListEnabled() &&\n\t\tlen(msResp.GetStickyTaskList().GetName()) != 0 &&\n\t\tsupportsStickyQuery &&\n\t\te.config.EnableStickyQuery(queryRequest.GetDomain()) &&\n\t\tisActive {\n\n\t\tstickyMatchingRequest := &types.MatchingQueryWorkflowRequest{\n\t\t\tDomainUUID:   domainID,\n\t\t\tQueryRequest: queryRequest,\n\t\t\tTaskList:     msResp.GetStickyTaskList(),\n\t\t}\n\n\t\t// using a clean new context in case customer provide a context which has\n\t\t// a really short deadline, causing we clear the stickiness\n\t\tstickyContext, cancel := context.WithTimeout(context.Background(), time.Duration(msResp.GetStickyTaskListScheduleToStartTimeout())*time.Second)\n\t\tstickyStopWatch := scope.StartTimer(metrics.DirectQueryDispatchStickyLatency)\n\t\tmatchingResp, err := e.rawMatchingClient.QueryWorkflow(stickyContext, stickyMatchingRequest)\n\t\tstickyStopWatch.Stop()\n\t\tcancel()\n\t\tif err == nil {\n\t\t\tscope.IncCounter(metrics.DirectQueryDispatchStickySuccessCount)\n\t\t\treturn &types.HistoryQueryWorkflowResponse{Response: &types.QueryWorkflowResponse{QueryResult: matchingResp.GetQueryResult(), QueryRejected: matchingResp.GetQueryRejected()}}, nil\n\t\t}\n\t\tswitch v := err.(type) {\n\t\tcase *types.StickyWorkerUnavailableError:\n\t\tcase *yarpcerrors.Status:\n\t\t\tif v.Code() != yarpcerrors.CodeDeadlineExceeded {\n\t\t\t\te.logger.Error(\"query directly though matching on sticky failed, will not attempt query on non-sticky\",\n\t\t\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\tdefault:\n\t\t\te.logger.Error(\"query directly though matching on sticky failed, will not attempt query on non-sticky\",\n\t\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\t\t\ttag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t\tif msResp.GetIsWorkflowRunning() {\n\t\t\te.logger.Info(\"query direct through matching failed on sticky, clearing sticky before attempting on non-sticky\",\n\t\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\t\t\ttag.Error(err))\n\t\t\tresetContext, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\t\t\tclearStickinessStopWatch := scope.StartTimer(metrics.DirectQueryDispatchClearStickinessLatency)\n\t\t\t_, err := e.ResetStickyTaskList(resetContext, &types.HistoryResetStickyTaskListRequest{\n\t\t\t\tDomainUUID: domainID,\n\t\t\t\tExecution:  queryRequest.GetExecution(),\n\t\t\t})\n\t\t\tclearStickinessStopWatch.Stop()\n\t\t\tcancel()\n\t\t\tif err != nil && err != workflow.ErrAlreadyCompleted && err != workflow.ErrNotExists {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tscope.IncCounter(metrics.DirectQueryDispatchClearStickinessSuccessCount)\n\t\t}\n\t}\n\n\tif err := common.IsValidContext(ctx); err != nil {\n\t\te.logger.Info(\"query context timed out before query on non-sticky task list could be attempted\",\n\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()))\n\t\tscope.IncCounter(metrics.DirectQueryDispatchTimeoutBeforeNonStickyCount)\n\t\treturn nil, err\n\t}\n\n\te.logger.Debug(\"query directly through matching on sticky timed out, attempting to query on non-sticky\",\n\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\ttag.WorkflowTaskListName(msResp.GetStickyTaskList().GetName()),\n\t\ttag.WorkflowNextEventID(msResp.GetNextEventID()))\n\n\tnonStickyMatchingRequest := &types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID:   domainID,\n\t\tQueryRequest: queryRequest,\n\t\tTaskList:     msResp.TaskList,\n\t}\n\n\tnonStickyStopWatch := scope.StartTimer(metrics.DirectQueryDispatchNonStickyLatency)\n\tmatchingResp, err := e.matchingClient.QueryWorkflow(ctx, nonStickyMatchingRequest)\n\tnonStickyStopWatch.Stop()\n\tif err != nil {\n\t\tif strings.Contains(err.Error(), \"unknown queryType\") {\n\t\t\te.logger.Info(\"user calls for unsupported query type\",\n\t\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\t\t\ttag.Error(err))\n\t\t} else {\n\t\t\te.logger.Error(\"query directly though matching on non-sticky failed\",\n\t\t\t\ttag.WorkflowDomainName(queryRequest.GetDomain()),\n\t\t\t\ttag.WorkflowID(queryRequest.Execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(queryRequest.Execution.GetRunID()),\n\t\t\t\ttag.WorkflowQueryType(queryRequest.Query.GetQueryType()),\n\t\t\t\ttag.Error(err))\n\t\t}\n\t\treturn nil, err\n\t}\n\tscope.IncCounter(metrics.DirectQueryDispatchNonStickySuccessCount)\n\treturn &types.HistoryQueryWorkflowResponse{Response: &types.QueryWorkflowResponse{QueryResult: matchingResp.GetQueryResult(), QueryRejected: matchingResp.GetQueryRejected()}}, err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reapply_events.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/ndc\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc (e *historyEngineImpl) ReapplyEvents(\n\tctx context.Context,\n\tdomainUUID string,\n\tworkflowID string,\n\trunID string,\n\treapplyEvents []*types.HistoryEvent,\n) error {\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, domainUUID, workflowID, runID)\n\tif err != nil {\n\t\tswitch {\n\t\t// TODO(active-active): should we remove this check for active-active domain?\n\t\t// It was introduced in https://github.com/cadence-workflow/cadence/pull/3502\n\t\tcase domainEntry != nil && domainEntry.IsDomainPendingActive():\n\t\t\treturn nil\n\t\tdefault:\n\t\t\treturn err\n\t\t}\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\t// remove run id from the execution so that reapply events to the current run\n\tcurrentExecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t}\n\n\treturn workflow.UpdateWithActionFunc(\n\t\tctx,\n\t\te.logger,\n\t\te.executionCache,\n\t\tdomainID,\n\t\tcurrentExecution,\n\t\te.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\t// Filter out reapply event from the same cluster\n\t\t\ttoReapplyEvents := make([]*types.HistoryEvent, 0, len(reapplyEvents))\n\t\t\tlastWriteVersion, err := mutableState.GetLastWriteVersion()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tfor _, event := range reapplyEvents {\n\t\t\t\tif event.Version == lastWriteVersion {\n\t\t\t\t\t// The reapply is from the same cluster. Ignoring.\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tdedupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\t\t\t\tif mutableState.IsResourceDuplicated(dedupResource) {\n\t\t\t\t\t// already apply the signal\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\ttoReapplyEvents = append(toReapplyEvents, event)\n\t\t\t}\n\t\t\tif len(toReapplyEvents) == 0 {\n\t\t\t\treturn &workflow.UpdateAction{\n\t\t\t\t\tNoop: true,\n\t\t\t\t}, nil\n\t\t\t}\n\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\t// need to reset target workflow (which is also the current workflow)\n\t\t\t\t// to accept events to be reapplied\n\t\t\t\tbaseRunID := mutableState.GetExecutionInfo().RunID\n\t\t\t\tresetRunID := uuid.New()\n\t\t\t\tbaseRebuildLastEventID := mutableState.GetPreviousStartedEventID()\n\n\t\t\t\t// TODO when https://github.com/uber/cadence/issues/2420 is finished, remove this block,\n\t\t\t\t//  since cannot reapply event to a finished workflow which had no decisions started\n\t\t\t\tif baseRebuildLastEventID == constants.EmptyEventID {\n\t\t\t\t\te.logger.Warn(\"cannot reapply event to a finished workflow\",\n\t\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\t\ttag.WorkflowID(currentExecution.GetWorkflowID()),\n\t\t\t\t\t)\n\t\t\t\t\te.metricsClient.IncCounter(metrics.HistoryReapplyEventsScope, metrics.EventReapplySkippedCount)\n\t\t\t\t\treturn &workflow.UpdateAction{Noop: true}, nil\n\t\t\t\t}\n\n\t\t\t\tbaseVersionHistories := mutableState.GetVersionHistories()\n\t\t\t\tif baseVersionHistories == nil {\n\t\t\t\t\treturn nil, execution.ErrMissingVersionHistories\n\t\t\t\t}\n\t\t\t\tbaseCurrentVersionHistory, err := baseVersionHistories.GetCurrentVersionHistory()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\tbaseRebuildLastEventVersion, err := baseCurrentVersionHistory.GetEventVersion(baseRebuildLastEventID)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\tbaseCurrentBranchToken := baseCurrentVersionHistory.GetBranchToken()\n\t\t\t\tbaseNextEventID := mutableState.GetNextEventID()\n\n\t\t\t\tif err = e.workflowResetter.ResetWorkflow(\n\t\t\t\t\tctx,\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\tbaseRunID,\n\t\t\t\t\tbaseCurrentBranchToken,\n\t\t\t\t\tbaseRebuildLastEventID,\n\t\t\t\t\tbaseRebuildLastEventVersion,\n\t\t\t\t\tbaseNextEventID,\n\t\t\t\t\tresetRunID,\n\t\t\t\t\tuuid.New(),\n\t\t\t\t\texecution.NewWorkflow(\n\t\t\t\t\t\tctx,\n\t\t\t\t\t\te.shard.GetClusterMetadata(),\n\t\t\t\t\t\twfContext,\n\t\t\t\t\t\tmutableState,\n\t\t\t\t\t\texecution.NoopReleaseFn,\n\t\t\t\t\t\te.logger,\n\t\t\t\t\t),\n\t\t\t\t\tndc.EventsReapplicationResetWorkflowReason,\n\t\t\t\t\ttoReapplyEvents,\n\t\t\t\t\tfalse,\n\t\t\t\t); err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\treturn &workflow.UpdateAction{\n\t\t\t\t\tNoop: true,\n\t\t\t\t}, nil\n\t\t\t}\n\n\t\t\tpostActions := &workflow.UpdateAction{\n\t\t\t\tCreateDecision: true,\n\t\t\t}\n\t\t\t// Do not create decision task when the workflow is cron and the cron has not been started yet\n\t\t\tif mutableState.GetExecutionInfo().CronSchedule != \"\" && !mutableState.HasProcessedOrPendingDecision() {\n\t\t\t\tpostActions.CreateDecision = false\n\t\t\t}\n\t\t\treappliedEvents, err := e.eventsReapplier.ReapplyEvents(\n\t\t\t\tctx,\n\t\t\t\tmutableState,\n\t\t\t\ttoReapplyEvents,\n\t\t\t\trunID,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\te.logger.Error(\"failed to re-apply stale events\", tag.Error(err))\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"unable to re-apply stale events\"}\n\t\t\t}\n\t\t\tif len(reappliedEvents) == 0 {\n\t\t\t\treturn &workflow.UpdateAction{\n\t\t\t\t\tNoop: true,\n\t\t\t\t}, nil\n\t\t\t}\n\t\t\treturn postActions, nil\n\t\t},\n\t)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/record_activity_task_started.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc (e *historyEngineImpl) RecordActivityTaskStarted(\n\tctx context.Context,\n\trequest *types.RecordActivityTaskStartedRequest,\n) (*types.RecordActivityTaskStartedResponse, error) {\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, request.DomainUUID, request.WorkflowExecution.WorkflowID, request.WorkflowExecution.RunID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainInfo := domainEntry.GetInfo()\n\n\tdomainID := domainInfo.ID\n\tdomainName := domainInfo.Name\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowExecution.WorkflowID,\n\t\tRunID:      request.WorkflowExecution.RunID,\n\t}\n\n\tvar resurrectError error\n\tresponse := &types.RecordActivityTaskStartedResponse{}\n\terr = workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, workflowExecution, false, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrNotExists\n\t\t\t}\n\n\t\t\tscheduleID := request.GetScheduleID()\n\t\t\trequestID := request.GetRequestID()\n\t\t\tai, isRunning := mutableState.GetActivityInfo(scheduleID)\n\n\t\t\t// RecordActivityTaskStarted is already past scheduleToClose timeout.\n\t\t\t// If at this point pending activity is still in mutable state it may be resurrected.\n\t\t\t// Otherwise it would be completed or timed out already.\n\t\t\tif isRunning && e.timeSource.Now().After(ai.ScheduledTime.Add(time.Duration(ai.ScheduleToCloseTimeout)*time.Second)) {\n\t\t\t\tresurrectedActivities, err := execution.GetResurrectedActivities(ctx, e.shard, mutableState)\n\t\t\t\tif err != nil {\n\t\t\t\t\te.logger.Error(\"Activity resurrection check failed\", tag.Error(err))\n\t\t\t\t\treturn err\n\t\t\t\t}\n\n\t\t\t\tif _, ok := resurrectedActivities[scheduleID]; ok {\n\t\t\t\t\t// found activity resurrection\n\t\t\t\t\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\t\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRecordActivityTaskStartedScope, metrics.ActivityResurrectionCounter)\n\t\t\t\t\te.logger.Error(\"Encounter resurrected activity, skip\",\n\t\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\t)\n\n\t\t\t\t\t// remove resurrected activity from mutable state\n\t\t\t\t\tif err := mutableState.DeleteActivity(scheduleID); err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\n\t\t\t\t\t// save resurrection error but return nil here, so that mutable state would get updated in DB\n\t\t\t\t\tresurrectError = workflow.ErrActivityTaskNotFound\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRecordActivityTaskStartedScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordActivityTaskStarted\",\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrStaleState\n\t\t\t}\n\n\t\t\t// Check execution state to make sure task is in the list of outstanding tasks and it is not yet started.  If\n\t\t\t// task is not outstanding than it is most probably a duplicate and complete the task.\n\t\t\tif !isRunning {\n\t\t\t\t// Looks like ActivityTask already completed as a result of another call.\n\t\t\t\t// It is OK to drop the task at this point.\n\t\t\t\te.logger.Debug(\"Potentially duplicate task.\", tag.TaskID(request.GetTaskID()), tag.WorkflowScheduleID(scheduleID), tag.TaskType(persistence.TransferTaskTypeActivityTask))\n\t\t\t\treturn workflow.ErrActivityTaskNotFound\n\t\t\t}\n\n\t\t\tscheduledEvent, err := mutableState.GetActivityScheduledEvent(ctx, scheduleID)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tresponse.ScheduledEvent = scheduledEvent\n\t\t\tresponse.ScheduledTimestampOfThisAttempt = common.Int64Ptr(ai.ScheduledTime.UnixNano())\n\n\t\t\tresponse.Attempt = int64(ai.Attempt)\n\t\t\tresponse.HeartbeatDetails = ai.Details\n\n\t\t\tresponse.WorkflowType = mutableState.GetWorkflowType()\n\t\t\tresponse.WorkflowDomain = domainName\n\n\t\t\tif ai.StartedID != constants.EmptyEventID {\n\t\t\t\t// If activity is started as part of the current request scope then return a positive response\n\t\t\t\tif ai.RequestID == requestID {\n\t\t\t\t\tresponse.StartedTimestamp = common.Int64Ptr(ai.StartedTime.UnixNano())\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\n\t\t\t\t// Looks like ActivityTask already started as a result of another call.\n\t\t\t\t// It is OK to drop the task at this point.\n\t\t\t\te.logger.Debug(\"Potentially duplicate task.\", tag.TaskID(request.GetTaskID()), tag.WorkflowScheduleID(scheduleID), tag.TaskType(persistence.TransferTaskTypeActivityTask))\n\t\t\t\treturn &types.EventAlreadyStartedError{Message: \"Activity task already started.\"}\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddActivityTaskStartedEvent(\n\t\t\t\tai, scheduleID, requestID, request.PollRequest.GetIdentity(),\n\t\t\t); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\t\tresponse.StartedTimestamp = common.Int64Ptr(ai.StartedTime.UnixNano())\n\n\t\t\treturn nil\n\t\t})\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif resurrectError != nil {\n\t\treturn nil, resurrectError\n\t}\n\n\treturn response, err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/record_child_execution_completed.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RecordChildExecutionCompleted records the completion of child execution into parent execution history\nfunc (e *historyEngineImpl) RecordChildExecutionCompleted(\n\tctx context.Context,\n\tcompletionRequest *types.RecordChildExecutionCompletedRequest,\n) error {\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, completionRequest.DomainUUID, completionRequest.WorkflowExecution.GetWorkflowID(), completionRequest.WorkflowExecution.GetRunID())\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: completionRequest.WorkflowExecution.GetWorkflowID(),\n\t\tRunID:      completionRequest.WorkflowExecution.GetRunID(),\n\t}\n\n\treturn e.updateWithActionFn(ctx, e.logger, e.executionCache, domainID, workflowExecution, true, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrNotExists\n\t\t\t}\n\n\t\t\tinitiatedID := completionRequest.InitiatedID\n\t\t\tstartedID := completionRequest.StartedID\n\t\t\tcompletedExecution := completionRequest.CompletedExecution\n\t\t\tcompletionEvent := completionRequest.CompletionEvent\n\n\t\t\t// Check mutable state to make sure child execution is in pending child executions\n\t\t\tci, isRunning := mutableState.GetChildExecutionInfo(initiatedID)\n\t\t\tif !isRunning {\n\t\t\t\tif initiatedID >= mutableState.GetNextEventID() {\n\t\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRecordChildExecutionCompletedScope, metrics.StaleMutableStateCounter)\n\t\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordChildExecutionCompleted\",\n\t\t\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\t\ttag.WorkflowInitiatedID(initiatedID),\n\t\t\t\t\t\ttag.WorkflowStartedID(startedID),\n\t\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t\t)\n\t\t\t\t\treturn workflow.ErrStaleState\n\t\t\t\t}\n\t\t\t\t// This can happen if the child execution is already completed, so it's deleted from mutable state\n\t\t\t\t// We must return a non-retryable error to avoid infinite retries\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Pending child execution not found.\"}\n\t\t\t}\n\t\t\tif ci.StartedID == constants.EmptyEventID {\n\t\t\t\t// This means that the child execution is initiated but not started in the view of the parent workflow,\n\t\t\t\t// but it receives a notification on child workflow completion.\n\t\t\t\t// This can happen if the parent workflow data is stale (due to shard movement), some bad guy is spoiling our API, or there is a bug.\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRecordChildExecutionCompletedScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordChildExecutionCompleted\",\n\t\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowInitiatedID(initiatedID),\n\t\t\t\t\ttag.WorkflowStartedID(startedID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrStaleState\n\t\t\t}\n\t\t\tif ci.StartedWorkflowID != completedExecution.GetWorkflowID() {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Pending child execution workflowID mismatch.\"}\n\t\t\t}\n\n\t\t\tswitch *completionEvent.EventType {\n\t\t\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\t\t\tattributes := completionEvent.WorkflowExecutionCompletedEventAttributes\n\t\t\t\t_, err = mutableState.AddChildWorkflowExecutionCompletedEvent(initiatedID, completedExecution, attributes)\n\t\t\tcase types.EventTypeWorkflowExecutionFailed:\n\t\t\t\tattributes := completionEvent.WorkflowExecutionFailedEventAttributes\n\t\t\t\t_, err = mutableState.AddChildWorkflowExecutionFailedEvent(initiatedID, completedExecution, attributes)\n\t\t\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\t\t\tattributes := completionEvent.WorkflowExecutionCanceledEventAttributes\n\t\t\t\t_, err = mutableState.AddChildWorkflowExecutionCanceledEvent(initiatedID, completedExecution, attributes)\n\t\t\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\t\t\tattributes := completionEvent.WorkflowExecutionTerminatedEventAttributes\n\t\t\t\t_, err = mutableState.AddChildWorkflowExecutionTerminatedEvent(initiatedID, completedExecution, attributes)\n\t\t\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\t\t\tattributes := completionEvent.WorkflowExecutionTimedOutEventAttributes\n\t\t\t\t_, err = mutableState.AddChildWorkflowExecutionTimedOutEvent(initiatedID, completedExecution, attributes)\n\t\t\t}\n\t\t\treturn err\n\t\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/record_decision_task_started.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// RecordDecisionTaskStarted starts a decision\nfunc (e *historyEngineImpl) RecordDecisionTaskStarted(ctx context.Context, request *types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error) {\n\treturn e.decisionHandler.HandleDecisionTaskStarted(ctx, request)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/refresh_workflow_tasks.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nfunc (e *historyEngineImpl) RefreshWorkflowTasks(\n\tctx context.Context,\n\tdomainUUID string,\n\tworkflowExecution types.WorkflowExecution,\n) (retError error) {\n\tdomainEntry, err := e.shard.GetDomainCache().GetDomainByID(domainUUID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\twfContext, release, err := e.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, workflowExecution)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tmutableStateTaskRefresher := execution.NewMutableStateTaskRefresher(\n\t\te.shard.GetConfig(),\n\t\te.shard.GetClusterMetadata(),\n\t\te.shard.GetDomainCache(),\n\t\te.shard.GetEventsCache(),\n\t\te.shard.GetShardID(),\n\t\te.logger,\n\t)\n\n\terr = mutableStateTaskRefresher.RefreshTasks(ctx, mutableState.GetExecutionInfo().StartTimestamp, mutableState)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\terr = wfContext.UpdateWorkflowExecutionTasks(ctx, e.shard.GetTimeSource().Now())\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/refresh_workflow_tasks_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n)\n\nfunc TestRefreshWorkflowTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\texecution         types.WorkflowExecution\n\t\tgetWFExecErr      error\n\t\treadHistBranchErr error\n\t\tupdateWFExecErr   error\n\t\twantErr           bool\n\t}{\n\t\t{\n\t\t\tname: \"runid is not uuid\",\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      \"not-a-uuid\",\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:         \"failed to get workflow execution\",\n\t\t\tgetWFExecErr: errors.New(\"some random error\"),\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname:              \"failed to get workflow start event\",\n\t\t\treadHistBranchErr: errors.New(\"some random error\"),\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to update workflow execution\",\n\t\t\t// returning TimeoutError because it doesn't get retried\n\t\t\tupdateWFExecErr: &persistence.TimeoutError{},\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\t// GetWorkflowExecution prep\n\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tExecution:  tc.execution,\n\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\tRangeID:    1,\n\t\t\t}\n\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t},\n\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t}\n\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\tReturn(getExecResp, tc.getWFExecErr).\n\t\t\t\tOnce()\n\n\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.\n\t\t\t\tEXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, tc.execution.WorkflowID, tc.execution.RunID).\n\t\t\t\tReturn(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).MaxTimes(1)\n\n\t\t\t// ReadHistoryBranch prep\n\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t// first event.\n\t\t\t\t\t{\n\t\t\t\t\t\tID:                                      1,\n\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\thistoryMgr.\n\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\tReturn(historyBranchResp, tc.readHistBranchErr).\n\t\t\t\tOnce()\n\n\t\t\t// UpdateWorkflowExecution prep\n\t\t\tvar gotUpdateExecReq *persistence.UpdateWorkflowExecutionRequest\n\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t}\n\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\tRun(func(args mock.Arguments) {\n\t\t\t\t\tvar ok bool\n\t\t\t\t\tgotUpdateExecReq, ok = args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\t\t\tif !ok {\n\t\t\t\t\t\tt.Fatalf(\"failed to cast input to *persistence.UpdateWorkflowExecutionRequest, type is %T\", args.Get(1))\n\t\t\t\t\t}\n\t\t\t\t}).\n\t\t\t\tReturn(updateExecResp, tc.updateWFExecErr).\n\t\t\t\tOnce()\n\n\t\t\t// UpdateShard prep. this is needed to update the shard's rangeID for failure cases.\n\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\tReturn(nil)\n\n\t\t\t// Call RefreshWorkflowTasks\n\t\t\terr := eft.Engine.RefreshWorkflowTasks(\n\t\t\t\tcontext.Background(),\n\t\t\t\tconstants.TestDomainID,\n\t\t\t\ttc.execution,\n\t\t\t)\n\n\t\t\t// Error validations\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"RefreshWorkflowTasks() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// UpdateWorkflowExecutionRequest validations\n\t\t\tif gotUpdateExecReq == nil {\n\t\t\t\tt.Fatal(\"UpdateWorkflowExecutionRequest is nil\")\n\t\t\t}\n\n\t\t\tif gotUpdateExecReq.RangeID != 1 {\n\t\t\t\tt.Errorf(\"got RangeID %v, want 1\", gotUpdateExecReq.RangeID)\n\t\t\t}\n\n\t\t\tif gotUpdateExecReq.DomainName != constants.TestDomainName {\n\t\t\t\tt.Errorf(\"got DomainName %v, want %v\", gotUpdateExecReq.DomainName, constants.TestDomainName)\n\t\t\t}\n\n\t\t\tif gotUpdateExecReq.Mode != persistence.UpdateWorkflowModeIgnoreCurrent {\n\t\t\t\tt.Errorf(\"got Mode %v, want %v\", gotUpdateExecReq.Mode, persistence.UpdateWorkflowModeIgnoreCurrent)\n\t\t\t}\n\n\t\t\tif len(gotUpdateExecReq.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]) != 2 {\n\t\t\t\tt.Errorf(\"got %v TimerTasks, want 2\", len(gotUpdateExecReq.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]))\n\t\t\t} else {\n\t\t\t\ttimer0, ok := gotUpdateExecReq.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer][0].(*persistence.WorkflowTimeoutTask)\n\t\t\t\tif !ok {\n\t\t\t\t\tt.Fatalf(\"failed to cast TimerTask[0] to *persistence.WorkflowTimeoutTask, type is %T\", timer0)\n\t\t\t\t}\n\n\t\t\t\ttimer1, ok := gotUpdateExecReq.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer][1].(*persistence.DecisionTimeoutTask)\n\t\t\t\tif !ok {\n\t\t\t\t\tt.Fatalf(\"failed to cast TimerTask[0] to *persistence.WorkflowTimeoutTask, type is %T\", timer1)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/register_domain_failover_callback.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n)\n\nfunc (e *historyEngineImpl) registerDomainFailoverCallback() {\n\n\t// NOTE: READ BEFORE MODIFICATION\n\t//\n\t// Tasks, e.g. transfer tasks and timer tasks, are created when holding the shard lock\n\t// meaning tasks -> release of shard lock\n\t//\n\t// Domain change notification follows the following steps, order matters\n\t// 1. lock all task processing.\n\t// 2. domain changes visible to everyone (Note: lock of task processing prevents task processing logic seeing the domain changes).\n\t// 3. failover min and max task levels are calculated, then update to shard.\n\t// 4. failover start & task processing unlock & shard domain version notification update. (order does not matter for this discussion)\n\t//\n\t// The above guarantees that task created during the failover will be processed.\n\t// If the task is created after domain change:\n\t// \t\tthen active processor will handle it. (simple case)\n\t// If the task is created before domain change:\n\t//\t\ttask -> release of shard lock\n\t//\t\tfailover min / max task levels calculated & updated to shard (using shard lock) -> failover start\n\t// above 2 guarantees that failover start is after persistence of the task.\n\n\tinitialNotificationVersion := e.shard.GetDomainNotificationVersion()\n\n\tcatchUpFn := func(domainCache cache.DomainCache, prepareCallback cache.PrepareCallbackFn, callback cache.CallbackFn) {\n\t\t// this section is trying to make the shard catch up with domain changes\n\t\tdomains := cache.DomainCacheEntries{}\n\t\tfor _, domain := range domainCache.GetAllDomain() {\n\t\t\tdomains = append(domains, domain)\n\t\t}\n\t\t// we must notify the change in an ordered fashion\n\t\t// since history shard has to update the shard info\n\t\t// with the domain change version.\n\t\tsort.Sort(domains)\n\n\t\tvar updatedEntries []*cache.DomainCacheEntry\n\t\tvar domainNames []string\n\t\tfor _, domain := range domains {\n\t\t\tif domain.GetNotificationVersion() >= initialNotificationVersion {\n\t\t\t\tupdatedEntries = append(updatedEntries, domain)\n\t\t\t\tdomainNames = append(domainNames, domain.GetInfo().Name)\n\t\t\t}\n\t\t}\n\t\tif len(updatedEntries) > 0 {\n\t\t\te.logger.Info(\"catchUpFn calling prepareCallback and callback\", tag.Dynamic(\"domain-names\", domainNames))\n\t\t\tprepareCallback()\n\t\t\tcallback(updatedEntries)\n\t\t}\n\t}\n\n\t// first set the failover callback\n\te.shard.GetDomainCache().RegisterDomainChangeCallback(\n\t\tcreateShardNameFromShardID(e.shard.GetShardID()),\n\t\tcatchUpFn,\n\t\te.lockTaskProcessingForDomainUpdate,\n\t\te.domainChangeCB,\n\t)\n}\n\nfunc (e *historyEngineImpl) domainChangeCB(nextDomains []*cache.DomainCacheEntry) {\n\tdefer func() {\n\t\te.unlockProcessingForDomainUpdate()\n\t}()\n\n\tif len(nextDomains) == 0 {\n\t\treturn\n\t}\n\n\tshardNotificationVersion := e.shard.GetDomainNotificationVersion()\n\tfailoverActivePassiveDomainIDs := map[string]struct{}{}\n\tfailoverActiveActiveDomainIDs := map[string]struct{}{}\n\n\tfor _, nextDomain := range nextDomains {\n\t\tdomainFailoverNotificationVersion := nextDomain.GetFailoverNotificationVersion()\n\t\tdomainActiveCluster := nextDomain.GetReplicationConfig().ActiveClusterName\n\n\t\tif !nextDomain.IsGlobalDomain() {\n\t\t\tcontinue\n\t\t}\n\n\t\tif !nextDomain.GetReplicationConfig().IsActiveActive() &&\n\t\t\tdomainFailoverNotificationVersion >= shardNotificationVersion &&\n\t\t\tdomainActiveCluster == e.currentClusterName {\n\t\t\tfailoverActivePassiveDomainIDs[nextDomain.GetInfo().ID] = struct{}{}\n\t\t}\n\t\tif nextDomain.GetReplicationConfig().IsActiveActive() &&\n\t\t\tdomainFailoverNotificationVersion >= shardNotificationVersion &&\n\t\t\tnextDomain.IsActiveIn(e.currentClusterName) {\n\t\t\tfailoverActiveActiveDomainIDs[nextDomain.GetInfo().ID] = struct{}{}\n\t\t}\n\t}\n\n\tif len(failoverActivePassiveDomainIDs) > 0 {\n\t\te.logger.Info(\"Domain Failover Start.\", tag.WorkflowDomainIDs(failoverActivePassiveDomainIDs))\n\n\t\t// Failover queues are not created for active-active domains. Will revisit after new queue framework implementation.\n\t\tfor _, processor := range e.queueProcessors {\n\t\t\tprocessor.FailoverDomain(failoverActivePassiveDomainIDs)\n\t\t}\n\t}\n\n\tif len(failoverActiveActiveDomainIDs) > 0 {\n\t\te.logger.Info(\"Active-Active Domain updated\", tag.WorkflowDomainIDs(failoverActiveActiveDomainIDs))\n\t}\n\n\t// Notify queues for any domain update. (active-passive and active-active)\n\tif len(failoverActivePassiveDomainIDs) > 0 || len(failoverActiveActiveDomainIDs) > 0 {\n\t\te.notifyQueues()\n\t}\n\n\tfailoverMarkerTasks := e.generateGracefulFailoverTasksForDomainUpdateCallback(shardNotificationVersion, nextDomains)\n\n\t// This is a debug metric\n\te.metricsClient.IncCounter(metrics.FailoverMarkerScope, metrics.HistoryFailoverCallbackCount)\n\tif len(failoverMarkerTasks) > 0 {\n\t\tif err := e.shard.ReplicateFailoverMarkers(\n\t\t\tcontext.Background(),\n\t\t\tfailoverMarkerTasks,\n\t\t); err != nil {\n\t\t\te.logger.Error(\"Failed to insert failover marker to replication queue.\", tag.Error(err))\n\t\t\te.metricsClient.IncCounter(metrics.FailoverMarkerScope, metrics.FailoverMarkerInsertFailure)\n\t\t\t// fail this failover callback and it retries on next domain cache refresh\n\t\t\treturn\n\t\t}\n\t}\n\n\t//nolint:errcheck\n\te.shard.UpdateDomainNotificationVersion(nextDomains[len(nextDomains)-1].GetNotificationVersion() + 1)\n}\n\nfunc (e *historyEngineImpl) notifyQueues() {\n\tnow := e.shard.GetTimeSource().Now()\n\t// the fake tasks will not be actually used, we just need to make sure\n\t// its length > 0 and has correct timestamp, to trigger a db scan\n\tfakeDecisionTask := []persistence.Task{&persistence.DecisionTask{}}\n\tfakeDecisionTimeoutTask := []persistence.Task{&persistence.DecisionTimeoutTask{TaskData: persistence.TaskData{VisibilityTimestamp: now}}}\n\ttransferProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTransfer]\n\tif !ok {\n\t\te.logger.Error(\"transfer processor not found\")\n\t\treturn\n\t}\n\ttransferProcessor.NotifyNewTask(e.currentClusterName, &hcommon.NotifyTaskInfo{Tasks: fakeDecisionTask})\n\ttimerProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTimer]\n\tif !ok {\n\t\te.logger.Error(\"timer processor not found\")\n\t\treturn\n\t}\n\ttimerProcessor.NotifyNewTask(e.currentClusterName, &hcommon.NotifyTaskInfo{Tasks: fakeDecisionTimeoutTask})\n}\n\nfunc (e *historyEngineImpl) generateGracefulFailoverTasksForDomainUpdateCallback(shardNotificationVersion int64, nextDomains []*cache.DomainCacheEntry) []*persistence.FailoverMarkerTask {\n\t// handle graceful failover on active to passive\n\t// make sure task processor failover the domain before inserting the failover marker\n\tfailoverMarkerTasks := []*persistence.FailoverMarkerTask{}\n\tfor _, nextDomain := range nextDomains {\n\t\tif nextDomain.GetReplicationConfig().IsActiveActive() {\n\t\t\t// Currently it's unclear whether graceful failover is working for active-passive domains. We don't use it in practice.\n\t\t\t// Don't try to make it work for active-active domains until we determine we need it.\n\t\t\t// We may potentially retire existing graceful failover implementation and provide \"sync replication\" instead.\n\t\t\tcontinue\n\t\t}\n\t\tdomainFailoverNotificationVersion := nextDomain.GetFailoverNotificationVersion()\n\t\tdomainActiveCluster := nextDomain.GetReplicationConfig().ActiveClusterName\n\t\tpreviousFailoverVersion := nextDomain.GetPreviousFailoverVersion()\n\t\tpreviousClusterName, err := e.clusterMetadata.ClusterNameForFailoverVersion(previousFailoverVersion)\n\t\tif err != nil && previousFailoverVersion != constants.InitialPreviousFailoverVersion {\n\t\t\te.logger.Error(\"Failed to handle graceful failover\", tag.WorkflowDomainID(nextDomain.GetInfo().ID), tag.Error(err))\n\t\t\tcontinue\n\t\t}\n\n\t\tif nextDomain.IsGlobalDomain() &&\n\t\t\tdomainFailoverNotificationVersion >= shardNotificationVersion &&\n\t\t\tdomainActiveCluster != e.currentClusterName &&\n\t\t\tpreviousFailoverVersion != constants.InitialPreviousFailoverVersion &&\n\t\t\tpreviousClusterName == e.currentClusterName {\n\t\t\t// the visibility timestamp will be set in shard context\n\t\t\tfailoverMarkerTasks = append(failoverMarkerTasks, &persistence.FailoverMarkerTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion: nextDomain.GetFailoverVersion(),\n\t\t\t\t},\n\t\t\t\tDomainID: nextDomain.GetInfo().ID,\n\t\t\t})\n\t\t\t// This is a debug metric\n\t\t\te.metricsClient.IncCounter(metrics.FailoverMarkerScope, metrics.FailoverMarkerCallbackCount)\n\t\t}\n\t}\n\treturn failoverMarkerTasks\n}\n\nfunc (e *historyEngineImpl) lockTaskProcessingForDomainUpdate() {\n\te.logger.Debug(\"Locking processing for domain update\")\n\tfor _, processor := range e.queueProcessors {\n\t\tprocessor.LockTaskProcessing()\n\t}\n}\n\nfunc (e *historyEngineImpl) unlockProcessingForDomainUpdate() {\n\te.logger.Debug(\"Unlocking processing for failover\")\n\tfor _, processor := range e.queueProcessors {\n\t\tprocessor.UnlockTaskProcessing()\n\t}\n}\n\nfunc (e *historyEngineImpl) failoverPredicate(shardNotificationVersion int64, nextDomain *cache.DomainCacheEntry, action func()) {\n\tdomainFailoverNotificationVersion := nextDomain.GetFailoverNotificationVersion()\n\tdomainActiveCluster := nextDomain.GetReplicationConfig().ActiveClusterName\n\n\tif nextDomain.IsGlobalDomain() &&\n\t\t!nextDomain.GetReplicationConfig().IsActiveActive() &&\n\t\tdomainFailoverNotificationVersion >= shardNotificationVersion &&\n\t\tdomainActiveCluster == e.currentClusterName {\n\t\taction()\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/register_domain_failover_callback_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestGenerateFailoverTasksForDomainCallback(t *testing.T) {\n\n\tdomainInfo := persistence.DomainInfo{\n\t\tID:         \"83b48dab-68cb-4f73-8752-c75d9271977f\",\n\t\tName:       \"samples-domain\",\n\t\tOwnerEmail: \"test4@test.com\",\n\t\tData:       map[string]string{},\n\t}\n\tdomainConfig := persistence.DomainConfig{\n\t\tRetention:             3,\n\t\tEmitMetric:            true,\n\t\tHistoryArchivalURI:    \"file:///tmp/cadence_archival/development\",\n\t\tVisibilityArchivalURI: \"file:///tmp/cadence_vis_archival/development\",\n\t\tBadBinaries:           types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:       types.IsolationGroupConfiguration{},\n\t}\n\treplicationConfig := persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"cluster0\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: \"cluster0\"},\n\t\t\t{ClusterName: \"cluster1\"},\n\t\t\t{ClusterName: \"cluster2\"},\n\t\t},\n\t}\n\n\tt1 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&replicationConfig,\n\t\t1,\n\t\tnil,\n\t\t2,\n\t\tconstants.InitialPreviousFailoverVersion,\n\t\t15,\n\t)\n\n\tt2 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&replicationConfig,\n\t\t10,\n\t\tnil,\n\t\t3,\n\t\tconstants.InitialPreviousFailoverVersion,\n\t\t15,\n\t)\n\n\tt3 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"cluster1\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: \"cluster0\"},\n\t\t\t\t{ClusterName: \"cluster1\"},\n\t\t\t\t{ClusterName: \"cluster2\"},\n\t\t\t},\n\t\t},\n\t\t10,\n\t\tcommon.Ptr(time.Now().Add(time.Second).UnixNano()),\n\t\t4,\n\t\t1,\n\t\t15,\n\t)\n\n\tnonFailoverDomainUpdateT1 := []*cache.DomainCacheEntry{\n\t\tt1,\n\t}\n\n\tfailoverUpdateT2 := []*cache.DomainCacheEntry{\n\t\tt2,\n\t}\n\n\tfailoverUpdateT3 := []*cache.DomainCacheEntry{\n\t\tt3,\n\t}\n\n\ttests := map[string]struct {\n\t\tdomainUpdates              []*cache.DomainCacheEntry\n\t\tcurrentNotificationVersion int64\n\t\texpectedRes                []*persistence.FailoverMarkerTask\n\t\texpectedErr                error\n\t}{\n\t\t\"non-failover domain update. This should generate no tasks as there's no failover taking place\": {\n\t\t\tdomainUpdates:              nonFailoverDomainUpdateT1,\n\t\t\tcurrentNotificationVersion: 1,\n\t\t\texpectedRes:                []*persistence.FailoverMarkerTask{},\n\t\t},\n\t\t\"non-graceful failover domain update. This should generate no tasks as there's no failover taking place\": {\n\t\t\tdomainUpdates:              failoverUpdateT2,\n\t\t\tcurrentNotificationVersion: 2,\n\t\t\texpectedRes:                []*persistence.FailoverMarkerTask{\n\t\t\t\t// no failoverMarker tasks are created for non-graceful failovers, they just occur immediately\n\t\t\t},\n\t\t},\n\t\t\"graceful failover domain update. This should generate failover marker tasks\": {\n\t\t\tdomainUpdates:              failoverUpdateT3,\n\t\t\tcurrentNotificationVersion: 3,\n\t\t\texpectedRes: []*persistence.FailoverMarkerTask{\n\t\t\t\t{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             10,\n\t\t\t\t\t\tTaskID:              0,\n\t\t\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t\t\t},\n\t\t\t\t\tDomainID: \"83b48dab-68cb-4f73-8752-c75d9271977f\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tcluster := cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 10,\n\t\t\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\t\"cluster0\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster1\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster2\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfunc(string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\the := historyEngineImpl{\n\t\t\t\tlogger:             log.NewNoop(),\n\t\t\t\tclusterMetadata:    cluster,\n\t\t\t\tcurrentClusterName: \"cluster0\",\n\t\t\t\tmetricsClient:      metrics.NewNoopMetricsClient(),\n\t\t\t}\n\t\t\tres := he.generateGracefulFailoverTasksForDomainUpdateCallback(td.currentNotificationVersion, td.domainUpdates)\n\t\t\tassert.Equal(t, td.expectedRes, res)\n\t\t})\n\t}\n}\n\nfunc TestDomainCallback(t *testing.T) {\n\n\t// This is a reference set of data from a series of steps, showing a data dump\n\t// from cluster0 of the domain failover values, counters and guards and how they work:\n\n\t// ---- Initial starting position -- active in cluster0\n\t//\n\t// Cluster0 has initialFailoverVersion of 1\n\t// Cluster1 has initialFailoverVersion of 0\n\t// Cluster2 has initialFailoverVersion of 2\n\t// failover increments are 10\n\n\t// ----------- t0 - initial starting position - the domain is initially active in cluster0\n\t// domains_partition             | 0\n\t// name                          | samples-domain\n\t// config                        | {retention: 3, emit_metric: True, archival_bucket: '', archival_status: 0, bad_binaries: 0x590d000a0b0c0000000000, bad_binaries_encoding: 'thriftrw', history_archival_status: 1, history_archival_uri: 'file:///tmp/cadence_archival/development', visibility_archival_status: 1, visibility_archival_uri: 'file:///tmp/cadence_vis_archival/development', isolation_groups: 0x5900, isolation_groups_encoding: 'thriftrw', async_workflow_config: 0x5902000a000b0014000000000b001e0000000000, async_workflow_config_encoding: 'thriftrw'}\n\t// config_version                | 0\n\t// domain                        | {id: ad1002f2-f87d-4128-afce-6d320c8871e7, name: 'samples-domain', status: 0, description: '', owner_email: '', data: null}\n\t// failover_end_time             | 0\n\t// failover_notification_version | 0\n\t// failover_version              | 1\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031237843277000\n\t// notification_version          | 1\n\t// previous_failover_version     | -1\n\t// replication_config            | {active_cluster_name: 'cluster0', clusters: [{cluster_name: 'cluster0'}, {cluster_name: 'cluster1'}, {cluster_name: 'cluster2'}]}\n\n\t// ------------ t1 domain update - no failover\n\t// domains_partition             | 0\n\t// name                          | samples-domain\n\t// config                        | {retention: 3, emit_metric: True, archival_bucket: '', archival_status: 0, bad_binaries: 0x590d000a0b0c0000000000, bad_binaries_encoding: 'thriftrw', history_archival_status: 1, history_archival_uri: 'file:///tmp/cadence_archival/development', visibility_archival_status: 1, visibility_archival_uri: 'file:///tmp/cadence_vis_archival/development', isolation_groups: 0x5900, isolation_groups_encoding: 'thriftrw', async_workflow_config: 0x5902000a000b0014000000000b001e0000000000, async_workflow_config_encoding: 'thriftrw'}\n\t// config_version                | 1\n\t// domain                        | {id: ad1002f2-f87d-4128-afce-6d320c8871e7, name: 'samples-domain', status: 0, description: '', owner_email: 'some-email@email.com', data: {}}\n\t// failover_end_time             | 0\n\t// failover_notification_version | 0\n\t// failover_version              | 1\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031406189428000\n\t// notification_version          | 2\n\t// previous_failover_version     | -1\n\t// replication_config            | {active_cluster_name: 'cluster0', clusters: [{cluster_name: 'cluster0'}, {cluster_name: 'cluster1'}, {cluster_name: 'cluster2'}]}\n\n\t// ------------- t2 Failover to cluster2 - non-graceful\n\t// domains_partition             | 0\n\t// name                          | samples-domain\n\t// config                        | {retention: 3, emit_metric: True, archival_bucket: '', archival_status: 0, bad_binaries: 0x590d000a0b0c0000000000, bad_binaries_encoding: 'thriftrw', history_archival_status: 1, history_archival_uri: 'file:///tmp/cadence_archival/development', visibility_archival_status: 1, visibility_archival_uri: 'file:///tmp/cadence_vis_archival/development', isolation_groups: 0x5900, isolation_groups_encoding: 'thriftrw', async_workflow_config: 0x5902000a000b0014000000000b001e0000000000, async_workflow_config_encoding: 'thriftrw'}\n\t// config_version                | 1\n\t// domain                        | {id: ad1002f2-f87d-4128-afce-6d320c8871e7, name: 'samples-domain', status: 0, description: '', owner_email: 'some-email@email.com', data: {'FailoverHistory': '[{\"eventTime\":\"2024-10-15T15:31:52.533009-07:00\",\"fromCluster\":\"cluster0\",\"toCluster\":\"cluster2\",\"failoverType\":\"Force\"}]'}}\n\t// failover_end_time             | 0\n\t// failover_notification_version | 3\n\t// failover_version              | 2\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031512533009000\n\t// notification_version          | 3\n\t// previous_failover_version     | -1\n\t// replication_config            | {active_cluster_name: 'cluster2', clusters: [{cluster_name: 'cluster0'}, {cluster_name: 'cluster1'}, {cluster_name: 'cluster2'}]}\n\n\t// ------------ t3 Failback to cluster0 - graceful\n\t// domains_partition             | 0\n\t// name                          | samples-domain\n\t// config                        | {retention: 3, emit_metric: True, archival_bucket: '', archival_status: 0, bad_binaries: 0x590d000a0b0c0000000000, bad_binaries_encoding: 'thriftrw', history_archival_status: 1, history_archival_uri: 'file:///tmp/cadence_archival/development', visibility_archival_status: 1, visibility_archival_uri: 'file:///tmp/cadence_vis_archival/development', isolation_groups: 0x5900, isolation_groups_encoding: 'thriftrw', async_workflow_config: 0x5902000a000b0014000000000b001e0000000000, async_workflow_config_encoding: 'thriftrw'}\n\t// config_version                | 1\n\t// domain                        | {id: ad1002f2-f87d-4128-afce-6d320c8871e7, name: 'samples-domain', status: 0, description: '', owner_email: 'some-email@email.com', data: {'FailoverHistory': '[{\"eventTime\":\"2024-10-15T15:33:39.83016-07:00\",\"fromCluster\":\"cluster2\",\"toCluster\":\"cluster0\",\"failoverType\":\"Grace\"},{\"eventTime\":\"2024-10-15T15:31:52.533009-07:00\",\"fromCluster\":\"cluster0\",\"toCluster\":\"cluster2\",\"failoverType\":\"Force\"}]'}}\n\t// failover_end_time             | 0\n\t// failover_notification_version | 4\n\t// failover_version              | 11\n\t// is_global_domain              | True\n\t// last_updated_time             | 0\n\t// notification_version          | 5\n\t// previous_failover_version     | 0\n\t// replication_config            | {active_cluster_name: 'cluster0', clusters: [{cluster_name: 'cluster0'}, {cluster_name: 'cluster1'}, {cluster_name: 'cluster2'}]}\n\n\t// ------------ t4 Failback to cluster2 - graceful\n\t// domains_partition             | 0\n\t// name                          | samples-domain\n\t// config                        | {retention: 3, emit_metric: True, archival_bucket: '', archival_status: 0, bad_binaries: 0x590d000a0b0c0000000000, bad_binaries_encoding: 'thriftrw', history_archival_status: 1, history_archival_uri: 'file:///tmp/cadence_archival/development', visibility_archival_status: 1, visibility_archival_uri: 'file:///tmp/cadence_vis_archival/development', isolation_groups: 0x5900, isolation_groups_encoding: 'thriftrw', async_workflow_config: 0x5902000a000b0014000000000b001e0000000000, async_workflow_config_encoding: 'thriftrw'}\n\t// config_version                | 1\n\t// domain                        | {id: ad1002f2-f87d-4128-afce-6d320c8871e7, name: 'samples-domain', status: 0, description: '', owner_email: 'some-email@email.com', data: {'FailoverHistory': '[{\"eventTime\":\"2024-10-15T15:47:56.967632-07:00\",\"fromCluster\":\"cluster0\",\"toCluster\":\"cluster2\",\"failoverType\":\"Grace\"}]'}}\n\t// failover_end_time             | 0\n\t// failover_notification_version | 5\n\t// failover_version              | 12\n\t// is_global_domain              | True\n\t// last_updated_time             | 0\n\t// notification_version          | 6\n\t// previous_failover_version     | 0\n\t// replication_config            | {active_cluster_name: 'cluster2', clusters: [{cluster_name: 'cluster0'}, {cluster_name: 'cluster1'}, {cluster_name: 'cluster2'}]}\n\tclusters := []*persistence.ClusterReplicationConfig{\n\t\t{ClusterName: \"cluster0\"},\n\t\t{ClusterName: \"cluster1\"},\n\t\t{ClusterName: \"cluster2\"},\n\t}\n\n\tdomainInfo := persistence.DomainInfo{\n\t\tID:         \"83b48dab-68cb-4f73-8752-c75d9271977f\",\n\t\tName:       \"samples-domain\",\n\t\tOwnerEmail: \"test4@test.com\",\n\t\tData:       map[string]string{},\n\t}\n\tdomainConfig := persistence.DomainConfig{\n\t\tRetention:             3,\n\t\tEmitMetric:            true,\n\t\tHistoryArchivalURI:    \"file:///tmp/cadence_archival/development\",\n\t\tVisibilityArchivalURI: \"file:///tmp/cadence_vis_archival/development\",\n\t\tBadBinaries:           types.BadBinaries{Binaries: map[string]*types.BadBinaryInfo{}},\n\t\tIsolationGroups:       types.IsolationGroupConfiguration{},\n\t}\n\treplicationConfig := persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"cluster0\",\n\t\tClusters:          clusters,\n\t}\n\n\t// failover_version              | 1\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031237843277000\n\t// notification_version          | 1\n\t// previous_failover_version     | -1\n\n\t// A domain update, no failover though\n\t// failover_end_time             | 0\n\t// failover_notification_version | 0\n\t// failover_version              | 1\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031406189428000\n\t// notification_version          | 2\n\t// previous_failover_version     | -1\n\tt1 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&replicationConfig,\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\tconstants.InitialPreviousFailoverVersion,\n\t\t2,\n\t)\n\n\t// failover to cluster2\n\t// failover_end_time             | 0\n\t// failover_notification_version | 3\n\t// failover_version              | 2\n\t// is_global_domain              | True\n\t// last_updated_time             | 1729031512533009000\n\t// notification_version          | 3\n\t// previous_failover_version     | -1\n\tt2 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"cluster2\",\n\t\t\tClusters:          clusters,\n\t\t},\n\t\t2,\n\t\tnil,\n\t\t3,\n\t\tconstants.InitialPreviousFailoverVersion,\n\t\t3,\n\t)\n\n\t// Graceful failover to cluster0 again (remember cluster0 is initialFailover version 1)\n\t// failover_end_time             | 0\n\t// failover_notification_version | 4\n\t// failover_version              | 11\n\t// is_global_domain              | True\n\t// last_updated_time             | 0\n\t// notification_version          | 5\n\t// previous_failover_version     | 0\n\tt3 := cache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t&domainConfig,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"cluster0\",\n\t\t\tClusters:          clusters,\n\t\t},\n\t\t11,\n\t\tcommon.Ptr(time.Now().Add(time.Second).UnixNano()),\n\t\t4,\n\t\t0,\n\t\t5, // this is 4 on all the other clusters, but on cluster0 its 5 because\n\t\t// graceful does two events it seems to increment it twice just locally\n\t)\n\n\tnonFailoverDomainUpdateT1 := []*cache.DomainCacheEntry{\n\t\tt1,\n\t}\n\n\tfailoverUpdateT2 := []*cache.DomainCacheEntry{\n\t\tt2,\n\t}\n\n\tfailoverUpdateT3 := []*cache.DomainCacheEntry{\n\t\tt3,\n\t}\n\n\tfailoverUpdateActiveActive := []*cache.DomainCacheEntry{\n\t\tcache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t\t&domainConfig,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tClusters: clusters,\n\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"region0\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t-1,\n\t\t\tnil,\n\t\t\t4,\n\t\t\t5,\n\t\t\t5,\n\t\t),\n\t}\n\n\tinvalidDomainUpdate := []*cache.DomainCacheEntry{\n\t\tcache.NewDomainCacheEntryForTest(&domainInfo,\n\t\t\t&domainConfig,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\tClusters:          clusters,\n\t\t\t},\n\t\t\t11,\n\t\t\tcommon.Ptr(time.Now().Add(time.Second).UnixNano()),\n\t\t\t4,\n\t\t\t5, // not a valid initial failover vrsion\n\t\t\t5,\n\t\t),\n\t}\n\n\ttimeSource := clock.NewMockedTimeSource()\n\n\ttests := map[string]struct {\n\t\tdomainUpdates []*cache.DomainCacheEntry\n\t\tasCluster     string\n\t\taffordances   func(\n\t\t\tshardCtx *shard.MockContext,\n\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\ttaskProcessor *task.MockProcessor,\n\t\t)\n\t\texpectedErr error\n\t}{\n\t\t\"t1 - non-failover domain update. - from the point of view of cluster 0\": {\n\t\t\tdomainUpdates: nonFailoverDomainUpdateT1,\n\t\t\tasCluster:     \"cluster0\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(3))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"t1 -> t2: non-graceful failover domain update. cluster0 POV\": {\n\t\t\tdomainUpdates: failoverUpdateT2,\n\t\t\tasCluster:     \"cluster0\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(4))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"t1 -> t2: non-graceful failover domain update. - cluster2 POV\": {\n\t\t\tdomainUpdates: failoverUpdateT2,\n\t\t\tasCluster:     \"cluster2\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().GetTimeSource().Return(timeSource)\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(4))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttxProcessor.EXPECT().FailoverDomain(gomock.Any())\n\t\t\t\ttxProcessor.EXPECT().NotifyNewTask(\"cluster2\", gomock.Any())\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().FailoverDomain(gomock.Any())\n\t\t\t\ttimerProcessor.EXPECT().NotifyNewTask(\"cluster2\", gomock.Any())\n\t\t\t},\n\t\t},\n\t\t\"t2 -> t3: graceful failover domain update. cluster0 POV\": {\n\t\t\tdomainUpdates: failoverUpdateT3,\n\t\t\tasCluster:     \"cluster0\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(6))\n\t\t\t\tshardCtx.EXPECT().GetTimeSource().Return(timeSource)\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttxProcessor.EXPECT().FailoverDomain(map[string]struct{}{\"83b48dab-68cb-4f73-8752-c75d9271977f\": struct{}{}})\n\t\t\t\ttxProcessor.EXPECT().NotifyNewTask(\"cluster0\", gomock.Any())\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().FailoverDomain(map[string]struct{}{\"83b48dab-68cb-4f73-8752-c75d9271977f\": struct{}{}})\n\t\t\t\ttimerProcessor.EXPECT().NotifyNewTask(\"cluster0\", gomock.Any())\n\t\t\t},\n\t\t},\n\t\t\"t2 -> t3: graceful failover domain update. - cluster2 POV\": {\n\t\t\tdomainUpdates: failoverUpdateT3,\n\t\t\tasCluster:     \"cluster2\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(6))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"graceful failover domain update. cluster1 POV\": {\n\t\t\tdomainUpdates: failoverUpdateT3,\n\t\t\tasCluster:     \"cluster1\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(6))\n\n\t\t\t\tshardCtx.EXPECT().ReplicateFailoverMarkers(gomock.Any(), []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion: 11,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"83b48dab-68cb-4f73-8752-c75d9271977f\",\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"graceful failover domain update. cluster1 POV - error on replication\": {\n\t\t\tdomainUpdates: failoverUpdateT3,\n\t\t\tasCluster:     \"cluster1\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\n\t\t\t\tshardCtx.EXPECT().ReplicateFailoverMarkers(gomock.Any(), []*persistence.FailoverMarkerTask{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion: 11,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tDomainID: \"83b48dab-68cb-4f73-8752-c75d9271977f\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(assert.AnError)\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"invalid failover version\": {\n\t\t\tdomainUpdates: invalidDomainUpdate,\n\t\t\tasCluster:     \"cluster1\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(6))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t},\n\t\t},\n\t\t\"active active domain failover update. cluster0 POV\": {\n\t\t\tdomainUpdates: failoverUpdateActiveActive,\n\t\t\tasCluster:     \"cluster0\",\n\t\t\taffordances: func(\n\t\t\t\tshardCtx *shard.MockContext,\n\t\t\t\ttxProcessor *queue.MockProcessor,\n\t\t\t\ttimerProcessor *queue.MockProcessor,\n\t\t\t\ttaskProcessor *task.MockProcessor) {\n\n\t\t\t\tshardCtx.EXPECT().GetDomainNotificationVersion().Return(int64(1))\n\t\t\t\tshardCtx.EXPECT().GetTimeSource().Return(timeSource)\n\t\t\t\tshardCtx.EXPECT().UpdateDomainNotificationVersion(int64(6))\n\n\t\t\t\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttxProcessor.EXPECT().NotifyNewTask(\"cluster0\", gomock.Any())\n\n\t\t\t\ttimerProcessor.EXPECT().UnlockTaskProcessing()\n\t\t\t\ttimerProcessor.EXPECT().NotifyNewTask(\"cluster0\", gomock.Any())\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tcluster := cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 10,\n\t\t\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\t\"cluster0\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster1\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster2\": config.ClusterInformation{\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfunc(string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\ttimeProcessor := queue.NewMockProcessor(ctrl)\n\t\t\tqueueTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\ttxProcessor := queue.NewMockProcessor(ctrl)\n\t\t\tshardCtx := shard.NewMockContext(ctrl)\n\n\t\t\ttd.affordances(shardCtx, txProcessor, timeProcessor, queueTaskProcessor)\n\n\t\t\the := historyEngineImpl{\n\t\t\t\tlogger:             log.NewNoop(),\n\t\t\t\tclusterMetadata:    cluster,\n\t\t\t\tcurrentClusterName: td.asCluster,\n\t\t\t\tmetricsClient:      metrics.NewNoopMetricsClient(),\n\t\t\t\tqueueProcessors: map[persistence.HistoryTaskCategory]queue.Processor{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: txProcessor,\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer:    timeProcessor,\n\t\t\t\t},\n\t\t\t\tshard: shardCtx,\n\t\t\t}\n\t\t\the.domainChangeCB(td.domainUpdates)\n\t\t})\n\t}\n}\n\nfunc TestDomainLocking(t *testing.T) {\n\n\tcluster := cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: 10,\n\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\"cluster0\": config.ClusterInformation{\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tfunc(string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n\n\tctrl := gomock.NewController(t)\n\ttimeProcessor := queue.NewMockProcessor(ctrl)\n\ttxProcessor := queue.NewMockProcessor(ctrl)\n\tshardCtx := shard.NewMockContext(ctrl)\n\n\ttimeProcessor.EXPECT().LockTaskProcessing()\n\ttimeProcessor.EXPECT().UnlockTaskProcessing()\n\n\ttxProcessor.EXPECT().LockTaskProcessing()\n\ttxProcessor.EXPECT().UnlockTaskProcessing()\n\n\the := historyEngineImpl{\n\t\tlogger:             log.NewNoop(),\n\t\tclusterMetadata:    cluster,\n\t\tcurrentClusterName: \"cluster0\",\n\t\tmetricsClient:      metrics.NewNoopMetricsClient(),\n\t\tqueueProcessors: map[persistence.HistoryTaskCategory]queue.Processor{\n\t\t\tpersistence.HistoryTaskCategoryTransfer: txProcessor,\n\t\t\tpersistence.HistoryTaskCategoryTimer:    timeProcessor,\n\t\t},\n\t\tshard: shardCtx,\n\t}\n\n\the.lockTaskProcessingForDomainUpdate()\n\the.unlockProcessingForDomainUpdate()\n}\n\nfunc TestHistoryEngine_registerDomainFailoverCallback_ClosureBehavior(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t// Define the initial shard notification version\n\tinitialShardVersion := int64(5)\n\tmockShard.EXPECT().GetDomainNotificationVersion().Return(initialShardVersion)\n\tmockShard.EXPECT().GetShardID().Return(456).Times(1)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\n\t// Capture the registered catchUpFn\n\tvar registeredCatchUpFn func(cache.DomainCache, cache.PrepareCallbackFn, cache.CallbackFn)\n\tmockDomainCache.EXPECT().RegisterDomainChangeCallback(\n\t\tcreateShardNameFromShardID(456), // id of the callback\n\t\tgomock.Any(),                    // catchUpFn\n\t\tgomock.Any(),                    // lockTaskProcessingForDomainUpdate\n\t\tgomock.Any(),                    // domainChangeCB\n\t).Do(func(_ string, catchUpFn, _, _ interface{}) {\n\t\tif fn, ok := catchUpFn.(cache.CatchUpFn); ok {\n\t\t\tregisteredCatchUpFn = fn\n\t\t} else {\n\t\t\tt.Fatalf(\"Failed to convert catchUpFn to cache.CatchUpFn: got type %T\", catchUpFn)\n\t\t}\n\t}).Times(1)\n\n\tcluster := cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: 10,\n\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\"cluster0\": config.ClusterInformation{\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t},\n\t\t\t\t\"cluster1\": config.ClusterInformation{\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\t},\n\t\t\t\t\"cluster2\": config.ClusterInformation{\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tfunc(string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t)\n\n\t// Create the history engine instance\n\tengine := &historyEngineImpl{\n\t\tshard:              mockShard,\n\t\tlogger:             log.NewNoop(),\n\t\tclusterMetadata:    cluster,    // Or a specific mock if needed\n\t\tcurrentClusterName: \"cluster0\", // Or a specific value\n\t\tmetricsClient:      metrics.NewNoopMetricsClient(),\n\t}\n\n\t// Call the method under test to register the callback\n\tengine.registerDomainFailoverCallback()\n\n\t// --- Test the behavior of the registered catchUpFn ---\n\n\tt.Run(\"catchUpFn - No updated domains\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\t\"uuid-domain1\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain1\", Name: \"domain1\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t1,\n\t\t\t),\n\t\t\t\"uuid-domain2\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain2\", Name: \"domain2\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t4,\n\t\t\t),\n\t\t})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func([]*cache.DomainCacheEntry) { callbackCalled = true }\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.False(t, prepareCalled, \"prepareCallback should not be called\")\n\t\t\tassert.False(t, callbackCalled, \"callback should not be called\")\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\t})\n\n\tt.Run(\"catchUpFn - Some updated domains\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\t\"uuid-domain1\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain1\", Name: \"domain1\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t7,\n\t\t\t),\n\t\t\t\"uuid-domain2\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain2\", Name: \"domain2\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t4,\n\t\t\t),\n\t\t})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tvar actualUpdatedEntries []*cache.DomainCacheEntry\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func(entries []*cache.DomainCacheEntry) {\n\t\t\tcallbackCalled = true\n\t\t\tactualUpdatedEntries = entries\n\t\t}\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.True(t, prepareCalled, \"prepareCallback should be called\")\n\t\t\tassert.True(t, callbackCalled, \"callback should be called\")\n\t\t\tassert.Len(t, actualUpdatedEntries, 1, \"should have one updated entry\")\n\t\t\tassert.Equal(t, int64(7), actualUpdatedEntries[0].GetNotificationVersion())\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\t})\n\n\tt.Run(\"catchUpFn - All domains should be updated\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\t\"uuid-domain1\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain1\", Name: \"domain1\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t7,\n\t\t\t),\n\t\t\t\"uuid-domain2\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain2\", Name: \"domain2\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t5,\n\t\t\t),\n\t\t})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tvar actualUpdatedEntries []*cache.DomainCacheEntry\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func(entries []*cache.DomainCacheEntry) {\n\t\t\tcallbackCalled = true\n\t\t\tactualUpdatedEntries = entries\n\t\t}\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.True(t, prepareCalled, \"prepareCallback should be called\")\n\t\t\tassert.True(t, callbackCalled, \"callback should be called\")\n\t\t\tassert.Len(t, actualUpdatedEntries, 2, \"should have two updated entries\")\n\t\t\tassert.Equal(t, int64(5), actualUpdatedEntries[0].GetNotificationVersion())\n\t\t\tassert.Equal(t, int64(7), actualUpdatedEntries[1].GetNotificationVersion())\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\t})\n\n\tt.Run(\"catchUpFn - Empty domain cache\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func([]*cache.DomainCacheEntry) { callbackCalled = true }\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.False(t, prepareCalled, \"prepareCallback should not be called\")\n\t\t\tassert.False(t, callbackCalled, \"callback should not be called\")\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/remove_signal_mutable_state.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RemoveSignalMutableState remove the signal request id in signal_requested for deduplicate\nfunc (e *historyEngineImpl) RemoveSignalMutableState(\n\tctx context.Context,\n\trequest *types.RemoveSignalMutableStateRequest,\n) error {\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, request.DomainUUID, request.WorkflowExecution.GetWorkflowID(), request.WorkflowExecution.GetRunID())\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowExecution.WorkflowID,\n\t\tRunID:      request.WorkflowExecution.RunID,\n\t}\n\n\treturn workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, workflowExecution, false, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrNotExists\n\t\t\t}\n\n\t\t\tmutableState.DeleteSignalRequested(request.GetRequestID())\n\n\t\t\treturn nil\n\t\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/request_cancel_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RequestCancelWorkflowExecution records request cancellation event for workflow execution\nfunc (e *historyEngineImpl) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\treq *types.HistoryRequestCancelWorkflowExecutionRequest,\n) error {\n\trequest := req.CancelRequest\n\tparentExecution := req.ExternalWorkflowExecution\n\tchildWorkflowOnly := req.GetChildWorkflowOnly()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowExecution.WorkflowID,\n\t}\n\t// If firstExecutionRunID is set on the request always try to cancel currently running execution\n\tif request.GetFirstExecutionRunID() == \"\" {\n\t\tworkflowExecution.RunID = request.WorkflowExecution.RunID\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, req.DomainUUID, workflowExecution.WorkflowID, workflowExecution.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\treturn workflow.UpdateCurrentWithActionFunc(ctx, e.logger, e.executionCache, e.executionManager, domainID, e.shard.GetDomainCache(), workflowExecution, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\tisCancelRequested, cancelRequestID := mutableState.IsCancelRequested()\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\t_, closeStatus := mutableState.GetWorkflowStateCloseStatus()\n\t\t\t\tif isCancelRequested && closeStatus == persistence.WorkflowCloseStatusCanceled {\n\t\t\t\t\tcancelRequest := req.CancelRequest\n\t\t\t\t\tif cancelRequest.RequestID != \"\" && cancelRequest.RequestID == cancelRequestID {\n\t\t\t\t\t\treturn &workflow.UpdateAction{Noop: true}, nil\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\t\tif request.GetFirstExecutionRunID() != \"\" {\n\t\t\t\tfirstRunID := executionInfo.FirstExecutionRunID\n\t\t\t\tif firstRunID == \"\" {\n\t\t\t\t\t// This is needed for backwards compatibility.  Workflow execution create with Cadence release v0.25.0 or earlier\n\t\t\t\t\t// does not have FirstExecutionRunID stored as part of mutable state.  If this is not set then load it from\n\t\t\t\t\t// workflow execution started event.\n\t\t\t\t\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn nil, err\n\t\t\t\t\t}\n\t\t\t\t\tfirstRunID = startEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstExecutionRunID()\n\t\t\t\t}\n\t\t\t\tif request.GetFirstExecutionRunID() != firstRunID {\n\t\t\t\t\treturn nil, &types.EntityNotExistsError{Message: \"Workflow execution not found\"}\n\t\t\t\t}\n\t\t\t}\n\t\t\tif childWorkflowOnly {\n\t\t\t\tparentWorkflowID := executionInfo.ParentWorkflowID\n\t\t\t\tparentRunID := executionInfo.ParentRunID\n\t\t\t\tif parentExecution.GetWorkflowID() != parentWorkflowID ||\n\t\t\t\t\tparentExecution.GetRunID() != parentRunID {\n\t\t\t\t\treturn nil, workflow.ErrParentMismatch\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif isCancelRequested {\n\t\t\t\tcancelRequest := req.CancelRequest\n\t\t\t\tif cancelRequest.RequestID != \"\" && cancelRequest.RequestID == cancelRequestID {\n\t\t\t\t\treturn workflow.UpdateWithNewDecision, nil\n\t\t\t\t}\n\t\t\t\t// if we consider workflow cancellation idempotent, then this error is redundant\n\t\t\t\t// this error maybe useful if this API is invoked by external, not decision from transfer queue\n\t\t\t\treturn nil, workflow.ErrCancellationAlreadyRequested\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddWorkflowExecutionCancelRequestedEvent(req.CancelRequest.Cause, req); err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to cancel workflow execution.\"}\n\t\t\t}\n\n\t\t\treturn workflow.UpdateWithNewDecision, nil\n\t\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reset_queues.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/queue\"\n)\n\nfunc (e *historyEngineImpl) ResetTransferQueue(\n\tctx context.Context,\n\tclusterName string,\n) error {\n\ttransferProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTransfer]\n\tif !ok {\n\t\treturn fmt.Errorf(\"transfer processor not found\")\n\t}\n\t_, err := transferProcessor.HandleAction(ctx, clusterName, queue.NewResetAction())\n\treturn err\n}\n\nfunc (e *historyEngineImpl) ResetTimerQueue(\n\tctx context.Context,\n\tclusterName string,\n) error {\n\ttimerProcessor, ok := e.queueProcessors[persistence.HistoryTaskCategoryTimer]\n\tif !ok {\n\t\treturn fmt.Errorf(\"timer processor not found\")\n\t}\n\t_, err := timerProcessor.HandleAction(ctx, clusterName, queue.NewResetAction())\n\treturn err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reset_sticky_tasklist.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// ResetStickyTaskList reset the volatile information in mutable state of a given types.\n// Volatile information are the information related to client, such as:\n// 1. StickyTaskList\n// 2. StickyScheduleToStartTimeout\n// 3. ClientLibraryVersion\n// 4. ClientFeatureVersion\n// 5. ClientImpl\nfunc (e *historyEngineImpl) ResetStickyTaskList(\n\tctx context.Context,\n\tresetRequest *types.HistoryResetStickyTaskListRequest,\n) (*types.HistoryResetStickyTaskListResponse, error) {\n\n\tif err := common.ValidateDomainUUID(resetRequest.DomainUUID); err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := resetRequest.DomainUUID\n\n\terr := workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, *resetRequest.Execution, false, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrAlreadyCompleted\n\t\t\t}\n\t\t\tmutableState.ClearStickyness()\n\t\t\treturn nil\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.HistoryResetStickyTaskListResponse{}, nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reset_sticky_tasklist_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc TestResetStickyTaskList(t *testing.T) {\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\tcases := []struct {\n\t\tname        string\n\t\trequest     *types.HistoryResetStickyTaskListRequest\n\t\tinit        func(engine *testdata.EngineForTest)\n\t\tassertions  func(engine *testdata.EngineForTest)\n\t\texpectedErr error\n\t}{\n\t\t{\n\t\t\tname: \"Invalid Domain\",\n\t\t\trequest: &types.HistoryResetStickyTaskListRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t\tExecution:  execution,\n\t\t\t},\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Missing domain UUID.\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Completed Workflow\",\n\t\t\trequest: &types.HistoryResetStickyTaskListRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tExecution:  execution,\n\t\t\t},\n\t\t\tinit: func(engine *testdata.EngineForTest) {\n\t\t\t\tengine.ShardCtx.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.GetWorkflowExecutionRequest) bool {\n\t\t\t\t\treturn req.Execution == *execution\n\t\t\t\t})).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\t\t\t\t\tRunID:      execution.RunID,\n\t\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\tChecksum:       checksum.Checksum{},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tengine.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, execution.WorkflowID, execution.RunID).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: workflow.ErrAlreadyCompleted,\n\t\t},\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\trequest: &types.HistoryResetStickyTaskListRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tExecution:  execution,\n\t\t\t},\n\t\t\tinit: func(engine *testdata.EngineForTest) {\n\t\t\t\tengine.ShardCtx.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.GetWorkflowExecutionRequest) bool {\n\t\t\t\t\treturn req.Execution == *execution\n\t\t\t\t})).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:       constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:     execution.WorkflowID,\n\t\t\t\t\t\t\tRunID:          execution.RunID,\n\t\t\t\t\t\t\tStickyTaskList: \"CLEAR ME PLEASE\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\tChecksum:       checksum.Checksum{},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tengine.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, execution.WorkflowID, execution.RunID).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).Times(1)\n\t\t\t\tengine.ShardCtx.Resource.ExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\t\t\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.WorkflowID == execution.WorkflowID &&\n\t\t\t\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.RunID == execution.RunID &&\n\t\t\t\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.StickyTaskList == \"\"\n\t\t\t\t})).Return(&persistence.UpdateWorkflowExecutionResponse{}, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, testCase := range cases {\n\t\tt.Run(testCase.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\n\t\t\tif testCase.init != nil {\n\t\t\t\ttestCase.init(eft)\n\t\t\t}\n\t\t\teft.Engine.Start()\n\t\t\tresult, err := eft.Engine.ResetStickyTaskList(ctx.Background(), testCase.request)\n\n\t\t\tif testCase.assertions != nil {\n\t\t\t\ttestCase.assertions(eft)\n\t\t\t}\n\t\t\teft.Engine.Stop()\n\n\t\t\tif testCase.expectedErr == nil {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, &types.HistoryResetStickyTaskListResponse{}, result)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, testCase.expectedErr, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reset_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nconst (\n\tDefaultPageSize = 100\n)\n\nfunc (e *historyEngineImpl) ResetWorkflowExecution(\n\tctx context.Context,\n\tresetRequest *types.HistoryResetWorkflowExecutionRequest,\n) (response *types.ResetWorkflowExecutionResponse, retError error) {\n\n\trequest := resetRequest.ResetRequest\n\tdomainID := resetRequest.GetDomainUUID()\n\tworkflowID := request.WorkflowExecution.GetWorkflowID()\n\tbaseRunID := request.WorkflowExecution.GetRunID()\n\n\tbaseContext, baseReleaseFn, err := e.executionCache.GetOrCreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      baseRunID,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer func() { baseReleaseFn(retError) }()\n\n\tbaseMutableState, err := baseContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif ok := baseMutableState.HasProcessedOrPendingDecision(); !ok {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"Cannot reset workflow without a decision task schedule.\",\n\t\t}\n\t}\n\tif request.GetDecisionFinishEventID() <= constants.FirstEventID ||\n\t\trequest.GetDecisionFinishEventID() > baseMutableState.GetNextEventID() {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"Decision finish ID must be > 1 && <= workflow next event ID.\",\n\t\t}\n\t}\n\tdomainName, err := e.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\t// also load the current run of the workflow, it can be different from the base runID\n\tresp, err := e.executionManager.GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: request.WorkflowExecution.GetWorkflowID(),\n\t\tDomainName: domainName,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcurrentRunID := resp.RunID\n\tvar currentContext execution.Context\n\tvar currentMutableState execution.MutableState\n\tvar currentReleaseFn execution.ReleaseFunc\n\tif currentRunID == baseRunID {\n\t\tcurrentContext = baseContext\n\t\tcurrentMutableState = baseMutableState\n\t} else {\n\t\tcurrentContext, currentReleaseFn, err = e.executionCache.GetOrCreateWorkflowExecution(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\ttypes.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      currentRunID,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdefer func() { currentReleaseFn(retError) }()\n\n\t\tcurrentMutableState, err = currentContext.LoadWorkflowExecution(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\t// dedup by requestID\n\tif currentMutableState.GetExecutionInfo().CreateRequestID == request.GetRequestID() {\n\t\te.logger.Info(\"Duplicated reset request\",\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(currentRunID),\n\t\t\ttag.WorkflowDomainID(domainID))\n\t\treturn &types.ResetWorkflowExecutionResponse{\n\t\t\tRunID: currentRunID,\n\t\t}, nil\n\t}\n\n\tresetRunID := uuid.New()\n\tbaseRebuildLastEventID := request.GetDecisionFinishEventID() - 1\n\tbaseVersionHistories := baseMutableState.GetVersionHistories()\n\tbaseCurrentBranchToken, err := baseMutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tbaseRebuildLastEventVersion := baseMutableState.GetCurrentVersion()\n\tbaseNextEventID := baseMutableState.GetNextEventID()\n\n\tif baseVersionHistories != nil {\n\t\tbaseCurrentVersionHistory, err := baseVersionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tbaseRebuildLastEventVersion, err = baseCurrentVersionHistory.GetEventVersion(baseRebuildLastEventID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tbaseCurrentBranchToken = baseCurrentVersionHistory.GetBranchToken()\n\t}\n\n\t// for reset workflow execution requests, the caller provides the decision finish event ID.\n\t// must validate the event ID to ensure it is a valid reset point\n\terr = e.validateResetPointForResetWorkflowExecutionRequest(ctx, request, baseCurrentBranchToken, domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := e.workflowResetter.ResetWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tbaseRunID,\n\t\tbaseCurrentBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tbaseNextEventID,\n\t\tresetRunID,\n\t\trequest.GetRequestID(),\n\t\texecution.NewWorkflow(\n\t\t\tctx,\n\t\t\te.shard.GetClusterMetadata(),\n\t\t\tcurrentContext,\n\t\t\tcurrentMutableState,\n\t\t\tcurrentReleaseFn,\n\t\t\te.logger,\n\t\t),\n\t\trequest.GetReason(),\n\t\tnil,\n\t\trequest.GetSkipSignalReapply(),\n\t); err != nil {\n\t\tif t, ok := persistence.AsDuplicateRequestError(err); ok {\n\t\t\tif t.RequestType == persistence.WorkflowRequestTypeReset {\n\t\t\t\treturn &types.ResetWorkflowExecutionResponse{\n\t\t\t\t\tRunID: t.RunID,\n\t\t\t\t}, nil\n\t\t\t}\n\t\t\te.logger.Error(\"A bug is detected for idempotency improvement\", tag.Dynamic(\"request-type\", t.RequestType))\n\t\t\treturn nil, t\n\t\t}\n\t\treturn nil, err\n\t}\n\treturn &types.ResetWorkflowExecutionResponse{\n\t\tRunID: resetRunID,\n\t}, nil\n}\n\nfunc (e *historyEngineImpl) validateResetPointForResetWorkflowExecutionRequest(\n\tctx context.Context,\n\trequest *types.ResetWorkflowExecutionRequest,\n\tbaseCurrentBranchToken []byte,\n\tdomainID string,\n) error {\n\titer := collection.NewPagingIterator(e.getPaginationFn(\n\t\tctx,\n\t\tconstants.FirstEventID,\n\t\trequest.GetDecisionFinishEventID()+1,\n\t\tbaseCurrentBranchToken,\n\t\tdomainID,\n\t))\n\n\tif !iter.HasNext() {\n\t\treturn fmt.Errorf(\"workflow has corrupted or missing history\")\n\t}\n\n\tvar events []*types.HistoryEvent\n\n\tfor iter.HasNext() {\n\t\tbatch, err := iter.Next()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tevents = batch.(*types.History).Events\n\n\t\t// get the batch of events that contains the reset event and exit the loop\n\t\tif events[len(events)-1].ID >= request.GetDecisionFinishEventID() {\n\t\t\tbreak\n\t\t}\n\t}\n\n\treturn checkResetEventType(events, request.GetDecisionFinishEventID())\n}\n\nfunc (e *historyEngineImpl) getPaginationFn(\n\tctx context.Context,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tbranchToken []byte,\n\tdomainID string,\n) collection.PaginationFn {\n\n\treturn func(paginationToken []byte) ([]interface{}, []byte, error) {\n\t\t_, historyBatches, token, _, err := persistenceutils.PaginateHistory(\n\t\t\tctx,\n\t\t\te.historyV2Mgr,\n\t\t\ttrue,\n\t\t\tbranchToken,\n\t\t\tfirstEventID,\n\t\t\tnextEventID,\n\t\t\tpaginationToken,\n\t\t\tDefaultPageSize,\n\t\t\tcommon.IntPtr(e.shard.GetShardID()),\n\t\t\tdomainID,\n\t\t\te.shard.GetDomainCache(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\n\t\tvar paginateItems []interface{}\n\t\tfor _, history := range historyBatches {\n\t\t\tpaginateItems = append(paginateItems, history)\n\t\t}\n\t\treturn paginateItems, token, nil\n\t}\n}\n\n// checkResetEventType checks the type of the reset event and only allows specific types to be resettable\nfunc checkResetEventType(events []*types.HistoryEvent, resetEventID int64) error {\n\tfor _, event := range events {\n\t\tif event.ID == resetEventID {\n\t\t\tswitch *event.EventType {\n\t\t\tcase types.EventTypeDecisionTaskStarted:\n\t\t\t\treturn nil\n\t\t\tcase types.EventTypeDecisionTaskTimedOut:\n\t\t\t\treturn nil\n\t\t\tcase types.EventTypeDecisionTaskFailed:\n\t\t\t\treturn nil\n\t\t\tcase types.EventTypeDecisionTaskCompleted:\n\t\t\t\treturn nil\n\t\t\tdefault:\n\t\t\t\treturn &types.BadRequestError{\n\t\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", event.EventType.String()),\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn fmt.Errorf(\"reset event ID %v not found in the history events\", resetEventID)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/reset_workflow_execution_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\tctx \"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n)\n\nvar (\n\ttestRequestID                = \"this is a test request\"\n\ttestRequestReason            = \"Test reason\"\n\ttestRequestSkipSignalReapply = true\n\tlatestRunID                  = constants.TestRunID\n\tlatestExecution              = &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: latestRunID}\n\tpreviousRunID                = \"bbbbbeef-0123-4567-890a-bcdef0123456\"\n\tpreviousExecution            = &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: previousRunID}\n\n\tversion         = int64(12)\n\tbranchToken     = []byte(\"other random branch token\")\n\tpartitionConfig = map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\tfirstEventID   = commonconstants.FirstEventID\n\tworkflowEvents = []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        1,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"some random workflow type\"},\n\t\t\t\tTaskList:                            &types.TaskList{Name: \"some random workflow type\"},\n\t\t\t\tInput:                               []byte(\"some random input\"),\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(123),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(233),\n\t\t\t\tIdentity:                            \"some random identity\",\n\t\t\t\tPartitionConfig:                     partitionConfig,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:                                   2,\n\t\t\tVersion:                              version,\n\t\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:        3,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 2,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        4,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 2,\n\t\t\t\tStartedEventID:   3,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        5,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\tActivityID: \"1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        6,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 5,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        7,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 5,\n\t\t\t\tStartedEventID:   6,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:                                   8,\n\t\t\tVersion:                              version,\n\t\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:        9,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\tScheduledEventID: 8,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        10,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\tTaskList: &types.TaskList{Name: \"some random workflow type\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        11,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 10,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        12,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\tScheduledEventID: 10,\n\t\t\t\tStartedEventID:   11,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:                                   13,\n\t\t\tVersion:                              version,\n\t\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:        14,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 13,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        15,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 13,\n\t\t\t\tStartedEventID:   14,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:                          16,\n\t\t\tVersion:                     version,\n\t\t\tEventType:                   types.EventTypeTimerStarted.Ptr(),\n\t\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:                        17,\n\t\t\tVersion:                   version,\n\t\t\tEventType:                 types.EventTypeTimerFired.Ptr(),\n\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:                                       18,\n\t\t\tVersion:                                  version,\n\t\t\tEventType:                                types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:                                   19,\n\t\t\tVersion:                              version,\n\t\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:        20,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 19,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        21,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeActivityTaskFailed.Ptr(),\n\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 19,\n\t\t\t\tStartedEventID:   20,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:                                   22,\n\t\t\tVersion:                              version,\n\t\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:        23,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 22,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        24,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 22,\n\t\t\t\tStartedEventID:   23,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        25,\n\t\t\tVersion:   version,\n\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{},\n\t\t},\n\t}\n\n\thistory = []testHistoryEvents{\n\t\t{\n\t\t\tEvents: workflowEvents[:2],\n\t\t\tSize:   1,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[2:3],\n\t\t\tSize:   2,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[3:6],\n\t\t\tSize:   3,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[6:8],\n\t\t\tSize:   4,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[8:10],\n\t\t\tSize:   5,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[10:11],\n\t\t\tSize:   6,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[11:13],\n\t\t\tSize:   7,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[13:14],\n\t\t\tSize:   8,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[14:16],\n\t\t\tSize:   9,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[16:19],\n\t\t\tSize:   10,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[19:22],\n\t\t\tSize:   11,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[22:23],\n\t\t\tSize:   12,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[23:24],\n\t\t\tSize:   13,\n\t\t},\n\t\t{\n\t\t\tEvents: workflowEvents[24:25],\n\t\t\tSize:   14,\n\t\t},\n\t}\n)\n\ntype (\n\tInitFn func(t *testing.T, engine *testdata.EngineForTest)\n\n\ttestHistoryEvents struct {\n\t\tEvents []*types.HistoryEvent\n\t\tSize   int\n\t}\n)\n\nfunc TestResetWorkflowExecution(t *testing.T) {\n\tcases := []struct {\n\t\tname        string\n\t\trequest     *types.HistoryResetWorkflowExecutionRequest\n\t\tinit        []InitFn\n\t\texpectedErr error\n\t\texpected    *types.ResetWorkflowExecutionResponse\n\t}{\n\t\t{\n\t\t\tname:    \"No processed or pending decision\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 100),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:           constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:         constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:              latestRunID,\n\t\t\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\t\t\tLastProcessedEvent: commonconstants.EmptyEventID,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t},\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Cannot reset workflow without a decision task schedule.\"},\n\t\t},\n\t\t{\n\t\t\tname:    \"Invalid DecisionFinishEventId\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 100),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t},\n\t\t\texpectedErr: &types.BadRequestError{Message: \"Decision finish ID must be > 1 && <= workflow next event ID.\"},\n\t\t},\n\t\t{\n\t\t\tname:    \"Duplicate Request\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:        constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:      constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:           latestRunID,\n\t\t\t\t\t\tNextEventID:     26,\n\t\t\t\t\t\tCreateRequestID: testRequestID,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t},\n\t\t\texpected: &types.ResetWorkflowExecutionResponse{\n\t\t\t\tRunID: latestRunID,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"Success\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\t\tgomock.Eq(int64(24)-1), // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(version),     // CurrentVersion\n\t\t\t\t\t\tgomock.Eq(int64(26)),   // NextEventID\n\t\t\t\t\t\tgomock.Any(),           // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\t// Can't assert on the result because the runID is random\n\t\t},\n\t\t{\n\t\t\tname: \"Success using version histories started in current cluster\",\n\t\t\t// This corresponds to VersionHistories.Histories.Items.EventID\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte(\"some other branch token\"),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\t\tVersion: 1000, // current cluster\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 23,\n\t\t\t\t\t\t\t\t\t\tVersion: 1001,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\tEventID: 24,\n\t\t\t\t\t\t\t\t\t\tVersion: 1002,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken), //VersionHistories.Histories.BranchToken\n\t\t\t\t\t\tgomock.Eq(int64(23)),   // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(int64(1001)), // VersionHistories.Histories.Items.Version\n\t\t\t\t\t\tgomock.Eq(int64(26)),   // NextEventID\n\t\t\t\t\t\tgomock.Any(),           // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\t// Can't assert on the result because the runID is random\n\t\t},\n\t\t{\n\t\t\tname:    \"Success using previous version\",\n\t\t\trequest: resetExecutionRequest(previousExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      latestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: 1,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithState(previousExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(previousExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\t\tgomock.Eq(int64(23)), // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(version),   // CurrentVersion\n\t\t\t\t\t\tgomock.Eq(int64(26)), // NextEventID\n\t\t\t\t\t\tgomock.Any(),         // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(nil).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\t// Can't assert on the result because the runID is random\n\t\t},\n\t\t{\n\t\t\tname:    \"Persistence Duplicate Request\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\t\tgomock.Eq(int64(23)), // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(version),   // CurrentVersion\n\t\t\t\t\t\tgomock.Eq(int64(26)), // NextEventID\n\t\t\t\t\t\tgomock.Any(),         // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(&persistence.DuplicateRequestError{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeReset,\n\t\t\t\t\t\tRunID:       \"errorID\",\n\t\t\t\t\t}).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.ResetWorkflowExecutionResponse{\n\t\t\t\tRunID: \"errorID\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"Persistence Duplicate Request Bug\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\t\tgomock.Eq(int64(23)), // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(version),   // CurrentVersion\n\t\t\t\t\t\tgomock.Eq(int64(26)), // NextEventID\n\t\t\t\t\t\tgomock.Any(),         // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(&persistence.DuplicateRequestError{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t\tRunID:       \"errorID\",\n\t\t\t\t\t}).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: &persistence.DuplicateRequestError{\n\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\tRunID:       \"errorID\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"Reset returns Err\",\n\t\t\trequest: resetExecutionRequest(latestExecution, 24),\n\t\t\tinit: []InitFn{\n\t\t\t\twithCurrentExecution(latestExecution),\n\t\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:       latestRunID,\n\t\t\t\t\t\tNextEventID: 26,\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t},\n\t\t\t\t\tReplicationState: &persistence.ReplicationState{\n\t\t\t\t\t\tCurrentVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 1},\n\t\t\t\t}),\n\t\t\t\twithActiveClusterInfo(constants.TestDomainID, latestExecution, &types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}),\n\t\t\t\twithHistoryPagination(branchToken, 24),\n\t\t\t\tfunc(t *testing.T, engine *testdata.EngineForTest) {\n\t\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\t\t\tengine.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\n\t\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\t\tgomock.Eq(int64(23)), // Request.DecisionFinishEventID - 1\n\t\t\t\t\t\tgomock.Eq(version),   // CurrentVersion\n\t\t\t\t\t\tgomock.Eq(int64(26)), // NextEventID\n\t\t\t\t\t\tgomock.Any(),         // random uuid\n\t\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t\t).Return(&types.BadRequestError{\n\t\t\t\t\t\tMessage: \"didn't work\",\n\t\t\t\t\t}).Times(1)\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: &types.BadRequestError{\n\t\t\t\tMessage: \"didn't work\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, testCase := range cases {\n\t\tt.Run(testCase.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\tfor _, setup := range testCase.init {\n\t\t\t\tsetup(t, eft)\n\t\t\t}\n\t\t\teft.Engine.Start()\n\t\t\tresult, err := eft.Engine.ResetWorkflowExecution(ctx.Background(), testCase.request)\n\t\t\teft.Engine.Stop()\n\n\t\t\tif testCase.expectedErr == nil {\n\t\t\t\tif assert.NotNil(t, result) {\n\t\t\t\t\tassert.NotEmpty(t, result.RunID)\n\t\t\t\t}\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, result)\n\t\t\t\tassert.Equal(t, testCase.expectedErr, err)\n\t\t\t}\n\n\t\t})\n\t}\n}\n\nfunc TestResetWorkflowExecution_ResetPointsValidation(t *testing.T) {\n\n\ttestCases := []struct {\n\t\tname         string\n\t\tresetEventID int64\n\t\tsetupMock    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64)\n\t\terr          error\n\t}{\n\t\t{\n\t\t\tname:         \"error - reset on decision task scheduled\",\n\t\t\tresetEventID: 10,\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[9].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on activity task scheduled\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 5,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[4].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on activity task started\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 6,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[5].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on activity task completed\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 7,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[6].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on timer started\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 16,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[15].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on timer fired\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 17,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[16].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on workflow execution signaled\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 18,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[17].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on activity task failed\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 21,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[20].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"error - reset on workflow execution completed\",\n\t\t\tsetupMock:    func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {},\n\t\t\tresetEventID: 25,\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: fmt.Sprintf(\"reset event must be of type DecisionTaskStarted, DecisionTaskTimedOut, DecisionTaskFailed, or DecisionTaskCompleted. Attempting to reset on event type: %v\", workflowEvents[24].EventType.String()),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - reset on decision task started\",\n\t\t\tsetupMock: func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {\n\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\tgomock.Eq(resetEventID-1), // Request.DecisionFinishEventID - 1\n\t\t\t\t\tgomock.Eq(version),        // CurrentVersion\n\t\t\t\t\tgomock.Eq(int64(26)),      // NextEventID\n\t\t\t\t\tgomock.Any(),              // random uuid\n\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tresetEventID: 23,\n\t\t\terr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - reset on decision task timed out\",\n\t\t\tsetupMock: func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {\n\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\tgomock.Eq(resetEventID-1), // Request.DecisionFinishEventID - 1\n\t\t\t\t\tgomock.Eq(version),        // CurrentVersion\n\t\t\t\t\tgomock.Eq(int64(26)),      // NextEventID\n\t\t\t\t\tgomock.Any(),              // random uuid\n\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tresetEventID: 9,\n\t\t\terr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - reset on decision task failed\",\n\t\t\tsetupMock: func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {\n\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\tgomock.Eq(resetEventID-1), // Request.DecisionFinishEventID - 1\n\t\t\t\t\tgomock.Eq(version),        // CurrentVersion\n\t\t\t\t\tgomock.Eq(int64(26)),      // NextEventID\n\t\t\t\t\tgomock.Any(),              // random uuid\n\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tresetEventID: 12,\n\t\t\terr:          nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - reset on decision task completed\",\n\t\t\tsetupMock: func(mockResetter *reset.MockWorkflowResetter, resetEventID int64) {\n\t\t\t\tmockResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(), // Context\n\t\t\t\t\tgomock.Eq(constants.TestDomainID),\n\t\t\t\t\tgomock.Eq(constants.TestWorkflowID),\n\t\t\t\t\tgomock.Eq(latestExecution.RunID),\n\t\t\t\t\tgomock.Eq(branchToken),\n\t\t\t\t\tgomock.Eq(resetEventID-1), // Request.DecisionFinishEventID - 1\n\t\t\t\t\tgomock.Eq(version),        // CurrentVersion\n\t\t\t\t\tgomock.Eq(int64(26)),      // NextEventID\n\t\t\t\t\tgomock.Any(),              // random uuid\n\t\t\t\t\tgomock.Eq(testRequestID),\n\t\t\t\t\t&workflowMatcher{latestExecution},\n\t\t\t\t\tgomock.Eq(testRequestReason),\n\t\t\t\t\tgomock.Nil(),\n\t\t\t\t\tgomock.Eq(testRequestSkipSignalReapply),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tresetEventID: 24,\n\t\t\terr:          nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tpageToken := []byte(\"some random pagination token\")\n\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\n\t\t\trequest := resetExecutionRequest(latestExecution, int(tc.resetEventID))\n\n\t\t\tmockResetter := reset.NewMockWorkflowResetter(ctrl)\n\t\t\teft.Engine.(*historyEngineImpl).workflowResetter = mockResetter\n\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, latestExecution.WorkflowID, latestExecution.RunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil)\n\n\t\t\ttc.setupMock(mockResetter, tc.resetEventID)\n\n\t\t\twithHistoryPagination(pageToken, tc.resetEventID)(t, eft)\n\n\t\t\twithState(latestExecution, &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:           constants.TestDomainID,\n\t\t\t\t\tWorkflowID:         constants.TestWorkflowID,\n\t\t\t\t\tRunID:              latestRunID,\n\t\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\t\tLastProcessedEvent: 25,\n\t\t\t\t\tNextEventID:        26,\n\t\t\t\t},\n\t\t\t\tExecutionStats: &persistence.ExecutionStats{HistorySize: 100},\n\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 25,\n\t\t\t\t\t\t\t\t\tVersion: version,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t})(t, eft)\n\n\t\t\twithCurrentExecution(latestExecution)(t, eft)\n\n\t\t\teft.Engine.Start()\n\t\t\tresult, err := eft.Engine.ResetWorkflowExecution(ctx.Background(), request)\n\t\t\teft.Engine.Stop()\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.EqualError(t, err, tc.err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc withHistoryPagination(pageToken []byte, resetEventID int64) InitFn {\n\treturn func(_ *testing.T, engine *testdata.EngineForTest) {\n\t\tcounter := int64(0)\n\t\thistorySize := 0\n\t\tfor index, historyBatch := range history {\n\t\t\ttoken := pageToken\n\n\t\t\tif index == 0 {\n\t\t\t\ttoken = nil\n\t\t\t}\n\n\t\t\tcounter += int64(len(historyBatch.Events))\n\n\t\t\tif counter >= resetEventID {\n\t\t\t\tpageToken = nil\n\t\t\t}\n\n\t\t\tengine.ShardCtx.GetHistoryManager().(*mocks.HistoryV2Manager).On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\t\t\tBranchToken:   branchToken,\n\t\t\t\tMinEventID:    firstEventID,\n\t\t\t\tMaxEventID:    resetEventID + 1, // Rebuild adds 1 to nextEventID\n\t\t\t\tPageSize:      DefaultPageSize,\n\t\t\t\tNextPageToken: token,\n\t\t\t\tShardID:       common.IntPtr(engine.ShardCtx.GetShardID()),\n\t\t\t\tDomainName:    constants.TestDomainName,\n\t\t\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\t\t\tHistory:       []*types.History{{Events: historyBatch.Events}},\n\t\t\t\tNextPageToken: pageToken,\n\t\t\t\tSize:          historyBatch.Size,\n\t\t\t}, nil).Once()\n\n\t\t\thistorySize += historyBatch.Size\n\n\t\t\tif counter >= resetEventID {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc withCurrentExecution(execution *types.WorkflowExecution) InitFn {\n\treturn func(_ *testing.T, engine *testdata.EngineForTest) {\n\t\tengine.ShardCtx.Resource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(&persistence.GetCurrentExecutionResponse{\n\t\t\tStartRequestID: \"CurrentExecutionStartRequestID\",\n\t\t\tRunID:          execution.RunID,\n\t\t\t// Other fields don't matter\n\t\t}, nil)\n\t}\n}\n\nfunc withState(execution *types.WorkflowExecution, state *persistence.WorkflowMutableState) InitFn {\n\treturn func(_ *testing.T, engine *testdata.EngineForTest) {\n\t\tengine.ShardCtx.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.GetWorkflowExecutionRequest) bool {\n\t\t\treturn req.Execution == *execution\n\t\t})).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\tState: state,\n\t\t}, nil)\n\t}\n}\n\nfunc withActiveClusterInfo(domainID string, execution *types.WorkflowExecution, activeClusterInfo *types.ActiveClusterInfo) InitFn {\n\treturn func(_ *testing.T, engine *testdata.EngineForTest) {\n\t\tengine.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, execution.WorkflowID, execution.RunID).Return(activeClusterInfo, nil)\n\t}\n}\n\nfunc resetExecutionRequest(execution *types.WorkflowExecution, decisionFinishEventID int) *types.HistoryResetWorkflowExecutionRequest {\n\treturn &types.HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID: constants.TestDomainID,\n\t\tResetRequest: &types.ResetWorkflowExecutionRequest{\n\t\t\tDomain:                constants.TestDomainName,\n\t\t\tWorkflowExecution:     execution,\n\t\t\tReason:                testRequestReason,\n\t\t\tDecisionFinishEventID: int64(decisionFinishEventID),\n\t\t\tRequestID:             testRequestID,\n\t\t\tSkipSignalReapply:     testRequestSkipSignalReapply,\n\t\t},\n\t}\n}\n\ntype workflowMatcher struct {\n\texecution *types.WorkflowExecution\n}\n\nfunc (m *workflowMatcher) Matches(obj interface{}) bool {\n\tif ex, ok := obj.(execution.Workflow); ok {\n\t\texecutionInfo := ex.GetMutableState().GetExecutionInfo()\n\t\treturn executionInfo.WorkflowID == m.execution.WorkflowID && executionInfo.RunID == m.execution.RunID\n\t}\n\treturn false\n}\n\nfunc (m *workflowMatcher) String() string {\n\treturn fmt.Sprintf(\"Workflow with WorkflowID %s and RunID %s\", m.execution.WorkflowID, m.execution.RunID)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_activity_task_canceled.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RespondActivityTaskCanceled completes an activity task failure.\nfunc (e *historyEngineImpl) RespondActivityTaskCanceled(\n\tctx context.Context,\n\treq *types.HistoryRespondActivityTaskCanceledRequest,\n) error {\n\n\trequest := req.CancelRequest\n\ttoken, err0 := e.tokenSerializer.Deserialize(request.TaskToken)\n\tif err0 != nil {\n\t\treturn workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\tdomainName := domainEntry.GetInfo().Name\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\tvar activityStartedTime time.Time\n\tvar taskList string\n\terr = workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, workflowExecution, true, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\tscheduleID := token.ScheduleID\n\t\t\tif scheduleID == constants.EmptyEventID { // client call CompleteActivityById, so get scheduleID by activityID\n\t\t\t\tscheduleID, err0 = getScheduleID(token.ActivityID, mutableState)\n\t\t\t\tif err0 != nil {\n\t\t\t\t\treturn err0\n\t\t\t\t}\n\t\t\t}\n\t\t\tai, isRunning := mutableState.GetActivityInfo(scheduleID)\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRespondActivityTaskCanceledScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordActivityTaskCanceled\",\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrStaleState\n\t\t\t}\n\n\t\t\tif !isRunning || ai.StartedID == constants.EmptyEventID ||\n\t\t\t\t(token.ScheduleID != constants.EmptyEventID && token.ScheduleAttempt != int64(ai.Attempt)) {\n\t\t\t\treturn workflow.ErrActivityTaskNotFound\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddActivityTaskCanceledEvent(\n\t\t\t\tscheduleID,\n\t\t\t\tai.StartedID,\n\t\t\t\tai.CancelRequestID,\n\t\t\t\trequest.Details,\n\t\t\t\trequest.Identity); err != nil {\n\t\t\t\t// Unable to add ActivityTaskCanceled event to history\n\t\t\t\treturn &types.InternalServiceError{Message: \"Unable to add ActivityTaskCanceled event to history.\"}\n\t\t\t}\n\n\t\t\tactivityStartedTime = ai.StartedTime\n\t\t\ttaskList = ai.TaskList\n\t\t\treturn nil\n\t\t})\n\tif err == nil && !activityStartedTime.IsZero() {\n\t\tscope := e.metricsClient.Scope(metrics.HistoryClientRespondActivityTaskCanceledScope).\n\t\t\tTagged(\n\t\t\t\tmetrics.DomainTag(domainName),\n\t\t\t\tmetrics.WorkflowTypeTag(token.WorkflowType),\n\t\t\t\tmetrics.ActivityTypeTag(token.ActivityType),\n\t\t\t\tmetrics.TaskListTag(taskList),\n\t\t\t)\n\t\tscope.RecordTimer(metrics.ActivityE2ELatency, time.Since(activityStartedTime))\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_activity_task_completed.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RespondActivityTaskCompleted completes an activity task.\nfunc (e *historyEngineImpl) RespondActivityTaskCompleted(\n\tctx context.Context,\n\treq *types.HistoryRespondActivityTaskCompletedRequest,\n) error {\n\trequest := req.CompleteRequest\n\ttoken, err0 := e.tokenSerializer.Deserialize(request.TaskToken)\n\tif err0 != nil {\n\t\treturn workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\tdomainName := domainEntry.GetInfo().Name\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\tvar activityStartedTime time.Time\n\tvar taskList string\n\terr = workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, workflowExecution, true, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\tscheduleID := token.ScheduleID\n\t\t\tif scheduleID == constants.EmptyEventID { // client call CompleteActivityById, so get scheduleID by activityID\n\t\t\t\tscheduleID, err0 = getScheduleID(token.ActivityID, mutableState)\n\t\t\t\tif err0 != nil {\n\t\t\t\t\treturn err0\n\t\t\t\t}\n\t\t\t}\n\t\t\tai, isRunning := mutableState.GetActivityInfo(scheduleID)\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRespondActivityTaskCompletedScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordActivityTaskCompleted\",\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrStaleState\n\t\t\t}\n\n\t\t\tif !isRunning || ai.StartedID == constants.EmptyEventID ||\n\t\t\t\t(token.ScheduleID != constants.EmptyEventID && token.ScheduleAttempt != int64(ai.Attempt)) {\n\t\t\t\te.logger.Warn(fmt.Sprintf(\n\t\t\t\t\t\"Encounter non existing activity in RecordActivityTaskCompleted: isRunning: %t, ai: %#v, token: %#v.\",\n\t\t\t\t\tisRunning, ai, token),\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrActivityTaskNotFound\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddActivityTaskCompletedEvent(scheduleID, ai.StartedID, request); err != nil {\n\t\t\t\t// Unable to add ActivityTaskCompleted event to history\n\t\t\t\treturn &types.InternalServiceError{Message: \"Unable to add ActivityTaskCompleted event to history.\"}\n\t\t\t}\n\t\t\tactivityStartedTime = ai.StartedTime\n\t\t\ttaskList = ai.TaskList\n\t\t\treturn nil\n\t\t})\n\tif err == nil && !activityStartedTime.IsZero() {\n\t\tscope := e.metricsClient.Scope(metrics.HistoryRespondActivityTaskCompletedScope).\n\t\t\tTagged(\n\t\t\t\tmetrics.DomainTag(domainName),\n\t\t\t\tmetrics.WorkflowTypeTag(token.WorkflowType),\n\t\t\t\tmetrics.ActivityTypeTag(token.ActivityType),\n\t\t\t\tmetrics.TaskListTag(taskList),\n\t\t\t)\n\t\tscope.RecordTimer(metrics.ActivityE2ELatency, time.Since(activityStartedTime))\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_activity_task_failed.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RespondActivityTaskFailed completes an activity task failure.\nfunc (e *historyEngineImpl) RespondActivityTaskFailed(\n\tctx context.Context,\n\treq *types.HistoryRespondActivityTaskFailedRequest,\n) error {\n\trequest := req.FailedRequest\n\ttoken, err0 := e.tokenSerializer.Deserialize(request.TaskToken)\n\tif err0 != nil {\n\t\treturn workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\tdomainName := domainEntry.GetInfo().Name\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\tvar activityStartedTime time.Time\n\tvar taskList string\n\terr = workflow.UpdateWithActionFunc(\n\t\tctx,\n\t\te.logger,\n\t\te.executionCache,\n\t\tdomainID,\n\t\tworkflowExecution,\n\t\te.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\tscheduleID := token.ScheduleID\n\t\t\tif scheduleID == constants.EmptyEventID { // client call CompleteActivityById, so get scheduleID by activityID\n\t\t\t\tscheduleID, err0 = getScheduleID(token.ActivityID, mutableState)\n\t\t\t\tif err0 != nil {\n\t\t\t\t\treturn nil, err0\n\t\t\t\t}\n\t\t\t}\n\t\t\tai, isRunning := mutableState.GetActivityInfo(scheduleID)\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRespondActivityTaskFailedScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordActivityTaskFailed\",\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn nil, workflow.ErrStaleState\n\t\t\t}\n\n\t\t\tif !isRunning || ai.StartedID == constants.EmptyEventID ||\n\t\t\t\t(token.ScheduleID != constants.EmptyEventID && token.ScheduleAttempt != int64(ai.Attempt)) {\n\t\t\t\te.logger.Warn(fmt.Sprintf(\n\t\t\t\t\t\"Encounter non existing activity in RecordActivityTaskFailed: isRunning: %t, ai: %#v, token: %#v.\",\n\t\t\t\t\tisRunning, ai, token),\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn nil, workflow.ErrActivityTaskNotFound\n\t\t\t}\n\n\t\t\tpostActions := &workflow.UpdateAction{}\n\t\t\tok, err := mutableState.RetryActivity(ai, req.FailedRequest.GetReason(), req.FailedRequest.GetDetails())\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tif !ok {\n\t\t\t\t// no more retry, and we want to record the failure event\n\t\t\t\tif _, err := mutableState.AddActivityTaskFailedEvent(scheduleID, ai.StartedID, request); err != nil {\n\t\t\t\t\t// Unable to add ActivityTaskFailed event to history\n\t\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to add ActivityTaskFailed event to history.\"}\n\t\t\t\t}\n\t\t\t\tpostActions.CreateDecision = true\n\t\t\t}\n\n\t\t\tactivityStartedTime = ai.StartedTime\n\t\t\ttaskList = ai.TaskList\n\t\t\treturn postActions, nil\n\t\t},\n\t)\n\tif err == nil && !activityStartedTime.IsZero() {\n\t\tscope := e.metricsClient.Scope(metrics.HistoryRespondActivityTaskFailedScope).\n\t\t\tTagged(\n\t\t\t\tmetrics.DomainTag(domainName),\n\t\t\t\tmetrics.WorkflowTypeTag(token.WorkflowType),\n\t\t\t\tmetrics.ActivityTypeTag(token.ActivityType),\n\t\t\t\tmetrics.TaskListTag(taskList),\n\t\t\t)\n\t\tscope.RecordTimer(metrics.ActivityE2ELatency, time.Since(activityStartedTime))\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_activity_task_heartbeat.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\n// RecordActivityTaskHeartbeat records an heartbeat for a task.\n// This method can be used for two purposes.\n// - For reporting liveness of the activity.\n// - For reporting progress of the activity, this can be done even if the liveness is not configured.\nfunc (e *historyEngineImpl) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\treq *types.HistoryRecordActivityTaskHeartbeatRequest,\n) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\trequest := req.HeartbeatRequest\n\ttoken, err0 := e.tokenSerializer.Deserialize(request.TaskToken)\n\tif err0 != nil {\n\t\treturn nil, workflow.ErrDeserializingToken\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, req.DomainUUID, token.WorkflowID, token.RunID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: token.WorkflowID,\n\t\tRunID:      token.RunID,\n\t}\n\n\tvar cancelRequested bool\n\terr = workflow.UpdateWithAction(ctx, e.logger, e.executionCache, domainID, workflowExecution, false, e.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\te.logger.Debug(\"Heartbeat failed\")\n\t\t\t\treturn workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\tscheduleID := token.ScheduleID\n\t\t\tif scheduleID == constants.EmptyEventID { // client call RecordActivityHeartbeatByID, so get scheduleID by activityID\n\t\t\t\tscheduleID, err0 = getScheduleID(token.ActivityID, mutableState)\n\t\t\t\tif err0 != nil {\n\t\t\t\t\treturn err0\n\t\t\t\t}\n\t\t\t}\n\t\t\tai, isRunning := mutableState.GetActivityInfo(scheduleID)\n\n\t\t\t// First check to see if cache needs to be refreshed as we could potentially have stale workflow execution in\n\t\t\t// some extreme cassandra failure cases.\n\t\t\tif !isRunning && scheduleID >= mutableState.GetNextEventID() {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryRecordActivityTaskHeartbeatScope, metrics.StaleMutableStateCounter)\n\t\t\t\te.logger.Error(\"Encounter stale mutable state in RecordActivityTaskHeartbeat\",\n\t\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\t\t\t\treturn workflow.ErrStaleState\n\t\t\t}\n\n\t\t\tif !isRunning || ai.StartedID == constants.EmptyEventID ||\n\t\t\t\t(token.ScheduleID != constants.EmptyEventID && token.ScheduleAttempt != int64(ai.Attempt)) {\n\t\t\t\te.logger.Warn(fmt.Sprintf(\n\t\t\t\t\t\"Encounter non existing activity in RecordActivityTaskHeartbeat: isRunning: %t, ai: %#v, token: %#v.\",\n\t\t\t\t\tisRunning, ai, token),\n\t\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t\t\t)\n\n\t\t\t\treturn workflow.ErrActivityTaskNotFound\n\t\t\t}\n\n\t\t\tcancelRequested = ai.CancelRequested\n\n\t\t\te.logger.Debug(fmt.Sprintf(\"Activity HeartBeat: scheduleEventID: %v, ActivityInfo: %+v, CancelRequested: %v\",\n\t\t\t\tscheduleID, ai, cancelRequested))\n\n\t\t\t// Save progress and last HB reported time.\n\t\t\tmutableState.UpdateActivityProgress(ai, request)\n\n\t\t\treturn nil\n\t\t})\n\n\tif err != nil {\n\t\treturn &types.RecordActivityTaskHeartbeatResponse{}, err\n\t}\n\n\treturn &types.RecordActivityTaskHeartbeatResponse{CancelRequested: cancelRequested}, nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_decision_task_completed.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// RespondDecisionTaskCompleted completes a decision task\nfunc (e *historyEngineImpl) RespondDecisionTaskCompleted(ctx context.Context, req *types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error) {\n\treturn e.decisionHandler.HandleDecisionTaskCompleted(ctx, req)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/respond_decision_task_failed.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// RespondDecisionTaskFailed fails a decision\nfunc (e *historyEngineImpl) RespondDecisionTaskFailed(ctx context.Context, req *types.HistoryRespondDecisionTaskFailedRequest) error {\n\treturn e.decisionHandler.HandleDecisionTaskFailed(ctx, req)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/signal_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc (e *historyEngineImpl) SignalWorkflowExecution(\n\tctx context.Context,\n\tsignalRequest *types.HistorySignalWorkflowExecutionRequest,\n) error {\n\trequest := signalRequest.SignalRequest\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowExecution.WorkflowID,\n\t\tRunID:      request.WorkflowExecution.RunID,\n\t}\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, signalRequest.DomainUUID, workflowExecution.WorkflowID, workflowExecution.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif domainEntry.GetInfo().Status != persistence.DomainStatusRegistered {\n\t\treturn errDomainDeprecated\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\tparentExecution := signalRequest.ExternalWorkflowExecution\n\tchildWorkflowOnly := signalRequest.GetChildWorkflowOnly()\n\n\treturn workflow.UpdateCurrentWithActionFunc(\n\t\tctx,\n\t\te.logger,\n\t\te.executionCache,\n\t\te.executionManager,\n\t\tdomainID,\n\t\te.shard.GetDomainCache(),\n\t\tworkflowExecution,\n\t\te.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\t// first deduplicate by request id for signal decision\n\t\t\t// this is done before workflow running check so that already completed error\n\t\t\t// won't be returned for duplicated signals even if the workflow is closed.\n\t\t\tif requestID := request.GetRequestID(); requestID != \"\" {\n\t\t\t\tif mutableState.IsSignalRequested(requestID) {\n\t\t\t\t\treturn &workflow.UpdateAction{\n\t\t\t\t\t\tNoop:           true,\n\t\t\t\t\t\tCreateDecision: false,\n\t\t\t\t\t}, nil\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\t// If history is corrupted, signal will be rejected\n\t\t\tif corrupted, err := e.checkForHistoryCorruptions(ctx, mutableState); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t} else if corrupted {\n\t\t\t\treturn nil, &types.EntityNotExistsError{Message: \"Workflow execution corrupted.\"}\n\t\t\t}\n\n\t\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\t\tcreateDecisionTask := true\n\t\t\tif !mutableState.HasProcessedOrPendingDecision() {\n\t\t\t\tcreateDecisionTask = false\n\t\t\t}\n\n\t\t\tmaxAllowedSignals := e.config.MaximumSignalsPerExecution(domainEntry.GetInfo().Name)\n\t\t\tif maxAllowedSignals > 0 && int(executionInfo.SignalCount) >= maxAllowedSignals {\n\t\t\t\te.logger.Info(\"Execution limit reached for maximum signals\", tag.WorkflowSignalCount(executionInfo.SignalCount),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\treturn nil, workflow.ErrSignalsLimitExceeded\n\t\t\t}\n\n\t\t\tif childWorkflowOnly {\n\t\t\t\tparentWorkflowID := executionInfo.ParentWorkflowID\n\t\t\t\tparentRunID := executionInfo.ParentRunID\n\t\t\t\tif parentExecution.GetWorkflowID() != parentWorkflowID ||\n\t\t\t\t\tparentExecution.GetRunID() != parentRunID {\n\t\t\t\t\treturn nil, workflow.ErrParentMismatch\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif requestID := request.GetRequestID(); requestID != \"\" {\n\t\t\t\tmutableState.AddSignalRequested(requestID)\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddWorkflowExecutionSignaled(\n\t\t\t\trequest.GetSignalName(),\n\t\t\t\trequest.GetInput(),\n\t\t\t\trequest.GetIdentity(),\n\t\t\t\trequest.GetRequestID(),\n\t\t\t); err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to signal workflow execution.\"}\n\t\t\t}\n\n\t\t\treturn &workflow.UpdateAction{\n\t\t\t\tNoop:           false,\n\t\t\t\tCreateDecision: createDecisionTask,\n\t\t\t}, nil\n\t\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/start_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nvar errClusterAttributeNotFound = &types.BadRequestError{Message: \"Cannot start workflow with a cluster attribute that is not found in the domain's metadata.\"}\n\n// for startWorkflowHelper be reused by signalWithStart\ntype signalWithStartArg struct {\n\tsignalWithStartRequest *types.HistorySignalWithStartWorkflowExecutionRequest\n\tprevMutableState       execution.MutableState\n}\n\n// StartWorkflowExecution starts a workflow execution\nfunc (e *historyEngineImpl) StartWorkflowExecution(\n\tctx context.Context,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n) (resp *types.StartWorkflowExecutionResponse, retError error) {\n\tdomainEntry, err := e.shard.GetDomainCache().GetDomainByID(startRequest.DomainUUID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, workflowExecution, historyBlob, err := e.startWorkflowHelper(\n\t\tctx,\n\t\tstartRequest,\n\t\tdomainEntry,\n\t\tmetrics.HistoryStartWorkflowExecutionScope,\n\t\tnil)\n\tif err != nil {\n\t\te.handleCreateWorkflowExecutionFailureCleanup(ctx,\n\t\t\tstartRequest,\n\t\t\tdomainEntry,\n\t\t\tworkflowExecution,\n\t\t\thistoryBlob,\n\t\t\tfalse,\n\t\t\terr,\n\t\t)\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc (e *historyEngineImpl) startWorkflowHelper(\n\tctx context.Context,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n\tdomainEntry *cache.DomainCacheEntry,\n\tmetricsScope metrics.ScopeIdx,\n\tsigWithStartArg *signalWithStartArg,\n) (_ *types.StartWorkflowExecutionResponse, _ *types.WorkflowExecution, _ *events.PersistedBlob, retError error) {\n\tif domainEntry.GetInfo().Status != persistence.DomainStatusRegistered {\n\t\treturn nil, nil, nil, errDomainDeprecated\n\t}\n\trequest := startRequest.StartRequest\n\n\terr := e.validateStartWorkflowExecutionRequest(request, metricsScope)\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\te.overrideTaskStartToCloseTimeoutSeconds(domainEntry, request, metricsScope)\n\n\tworkflowID := request.GetWorkflowID()\n\tdomainID := domainEntry.GetInfo().ID\n\tdomain := domainEntry.GetInfo().Name\n\n\t// grab the current context as a lock, nothing more\n\t// use a smaller context timeout to get the lock\n\tchildCtx, childCancel := e.newChildContext(ctx)\n\tdefer childCancel()\n\n\t_, currentRelease, err := e.executionCache.GetOrCreateCurrentWorkflowExecution(\n\t\tchildCtx,\n\t\tdomainID,\n\t\tworkflowID,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn nil, nil, nil, workflow.ErrConcurrentStartRequest\n\t\t}\n\t\treturn nil, nil, nil, err\n\t}\n\tdefer func() { currentRelease(retError) }()\n\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      uuid.New(),\n\t}\n\tcurMutableState, err := e.createMutableState(ctx, domainEntry, workflowExecution.GetRunID(), startRequest)\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\n\t// preprocess for signalWithStart\n\tvar prevMutableState execution.MutableState\n\tvar signalWithStartRequest *types.HistorySignalWithStartWorkflowExecutionRequest\n\tisSignalWithStart := sigWithStartArg != nil\n\tif isSignalWithStart {\n\t\tprevMutableState = sigWithStartArg.prevMutableState\n\t\tsignalWithStartRequest = sigWithStartArg.signalWithStartRequest\n\t}\n\tif prevMutableState != nil {\n\t\tprevLastWriteVersion, err := prevMutableState.GetLastWriteVersion()\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, err\n\t\t}\n\t\tif prevLastWriteVersion > curMutableState.GetCurrentVersion() {\n\t\t\tpolicy, err := e.shard.GetActiveClusterManager().GetActiveClusterSelectionPolicyForWorkflow(ctx, domainID, workflowID, prevMutableState.GetExecutionInfo().RunID)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, nil, nil, err\n\t\t\t}\n\t\t\tif policy.Equals(request.ActiveClusterSelectionPolicy) {\n\t\t\t\treturn nil, nil, nil, e.newDomainNotActiveError(\n\t\t\t\t\tdomainEntry,\n\t\t\t\t\tprevLastWriteVersion,\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\t\terr = e.applyWorkflowIDReusePolicyForSigWithStart(\n\t\t\tprevMutableState.GetExecutionInfo(),\n\t\t\t*workflowExecution,\n\t\t\trequest.GetWorkflowIDReusePolicy(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, err\n\t\t}\n\t} else if e.shard.GetConfig().EnableRecordWorkflowExecutionUninitialized(domainEntry.GetInfo().Name) && e.visibilityMgr != nil {\n\t\tuninitializedRequest := &persistence.RecordWorkflowExecutionUninitializedRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tDomain:     domain,\n\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t},\n\t\t\tWorkflowTypeName: request.WorkflowType.Name,\n\t\t\tUpdateTimestamp:  e.shard.GetTimeSource().Now().UnixNano(),\n\t\t\tShardID:          int64(e.shard.GetShardID()),\n\t\t}\n\n\t\tif err := e.visibilityMgr.RecordWorkflowExecutionUninitialized(ctx, uninitializedRequest); err != nil {\n\t\t\te.logger.Error(\"Failed to record uninitialized workflow execution\", tag.Error(err))\n\t\t}\n\t}\n\terr = e.addStartEventsAndTasks(\n\t\tcurMutableState,\n\t\t*workflowExecution,\n\t\tstartRequest,\n\t\tsignalWithStartRequest,\n\t)\n\tif err != nil {\n\t\tif e.shard.GetConfig().EnableRecordWorkflowExecutionUninitialized(domainEntry.GetInfo().Name) && e.visibilityMgr != nil {\n\t\t\t// delete the uninitialized workflow execution record since it failed to start the workflow\n\t\t\t// uninitialized record is used to find wfs that didn't make a progress or stuck during the start process\n\t\t\tif errVisibility := e.visibilityMgr.DeleteWorkflowExecution(ctx, &persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tDomain:     domain,\n\t\t\t\tRunID:      workflowExecution.RunID,\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t}); errVisibility != nil {\n\t\t\t\te.logger.Error(\"Failed to delete uninitialized workflow execution record\", tag.Error(errVisibility))\n\t\t\t}\n\t\t}\n\t\treturn nil, nil, nil, err\n\t}\n\twfContext := execution.NewContext(domainID, *workflowExecution, e.shard, e.executionManager, e.logger)\n\n\tnewWorkflow, newWorkflowEventsSeq, err := curMutableState.CloseTransactionAsSnapshot(\n\t\te.timeSource.Now(),\n\t\texecution.TransactionPolicyActive,\n\t)\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\tb, err := wfContext.PersistStartWorkflowBatchEvents(ctx, newWorkflowEventsSeq[0])\n\tif err != nil {\n\t\treturn nil, workflowExecution, nil, err\n\t}\n\thistoryBlob := &b\n\n\t// create as brand new\n\tcreateMode := persistence.CreateWorkflowModeBrandNew\n\tprevRunID := \"\"\n\tprevLastWriteVersion := int64(0)\n\t// overwrite in case of signalWithStart\n\tif prevMutableState != nil {\n\t\tcreateMode = persistence.CreateWorkflowModeWorkflowIDReuse\n\t\tinfo := prevMutableState.GetExecutionInfo()\n\t\t// For corrupted workflows use ContinueAsNew mode.\n\t\t// WorkflowIDReuse mode require workflows to be in completed state, which is not necessarily true for corrupted workflows.\n\t\tif info.State == persistence.WorkflowStateCorrupted {\n\t\t\tcreateMode = persistence.CreateWorkflowModeContinueAsNew\n\t\t}\n\t\tprevRunID = info.RunID\n\t\tprevLastWriteVersion, err = prevMutableState.GetLastWriteVersion()\n\t\tif err != nil {\n\t\t\treturn nil, workflowExecution, historyBlob, err\n\t\t}\n\t}\n\terr = wfContext.CreateWorkflowExecution(\n\t\tctx,\n\t\tnewWorkflow,\n\t\t*historyBlob,\n\t\tcreateMode,\n\t\tprevRunID,\n\t\tprevLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeNew,\n\t)\n\tif t, ok := persistence.AsDuplicateRequestError(err); ok {\n\t\tif t.RequestType == persistence.WorkflowRequestTypeStart || (isSignalWithStart && t.RequestType == persistence.WorkflowRequestTypeSignal) {\n\t\t\treturn &types.StartWorkflowExecutionResponse{\n\t\t\t\tRunID: t.RunID,\n\t\t\t}, workflowExecution, historyBlob, nil\n\t\t}\n\t\te.logger.Error(\"A bug is detected for idempotency improvement\", tag.Dynamic(\"request-type\", t.RequestType))\n\t\treturn nil, workflowExecution, historyBlob, t\n\t}\n\t// handle already started error\n\tif t, ok := err.(*persistence.WorkflowExecutionAlreadyStartedError); ok {\n\t\tif t.StartRequestID == request.GetRequestID() {\n\t\t\treturn &types.StartWorkflowExecutionResponse{\n\t\t\t\tRunID: t.RunID,\n\t\t\t}, workflowExecution, historyBlob, nil\n\t\t}\n\n\t\tif isSignalWithStart {\n\t\t\te.logger.Warn(\"signal-with-start might have left an orphaned history branch\",\n\t\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\t\ttag.Dynamic(\"debug-startRequest\", startRequest),\n\t\t\t\ttag.Dynamic(\"debug-historyBlob\", historyBlob),\n\t\t\t\ttag.Dynamic(\"debug-domainEntry\", domainEntry),\n\t\t\t\ttag.Dynamic(\"debug-workflowExecution\", workflowExecution),\n\t\t\t\ttag.Error(err))\n\t\t\treturn nil, workflowExecution, historyBlob, err\n\t\t}\n\n\t\tif curMutableState.GetCurrentVersion() < t.LastWriteVersion {\n\t\t\tpolicy, err := e.shard.GetActiveClusterManager().GetActiveClusterSelectionPolicyForWorkflow(ctx, domainID, workflowID, t.RunID)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, workflowExecution, historyBlob, err\n\t\t\t}\n\t\t\tif policy.Equals(request.ActiveClusterSelectionPolicy) {\n\t\t\t\treturn nil, workflowExecution, historyBlob, e.newDomainNotActiveError(\n\t\t\t\t\tdomainEntry,\n\t\t\t\t\tt.LastWriteVersion,\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\n\t\tprevRunID = t.RunID\n\t\tif shouldTerminateAndStart(startRequest, t.State) {\n\t\t\trunningWFCtx, err := workflow.LoadOnce(ctx, e.executionCache, domainID, workflowID, prevRunID)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, workflowExecution, historyBlob, err\n\t\t\t}\n\t\t\tdefer func() { runningWFCtx.GetReleaseFn()(retError) }()\n\n\t\t\tresp, err := e.terminateAndStartWorkflow(\n\t\t\t\tctx,\n\t\t\t\trunningWFCtx,\n\t\t\t\t*workflowExecution,\n\t\t\t\tdomainEntry,\n\t\t\t\tdomainID,\n\t\t\t\tstartRequest,\n\t\t\t\tnil,\n\t\t\t)\n\t\t\tswitch err.(type) {\n\t\t\t// By the time we try to terminate the workflow, it was already terminated\n\t\t\t// So continue as if we didn't need to terminate it in the first place\n\t\t\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\t\t\te.shard.GetLogger().Warn(\"Workflow completed while trying to terminate, will continue starting workflow\", tag.Error(err))\n\t\t\tdefault:\n\t\t\t\treturn resp, workflowExecution, historyBlob, err\n\t\t\t}\n\t\t}\n\t\tif err = e.applyWorkflowIDReusePolicyHelper(\n\t\t\tt.StartRequestID,\n\t\t\tprevRunID,\n\t\t\tt.State,\n\t\t\tt.CloseStatus,\n\t\t\t*workflowExecution,\n\t\t\tstartRequest.StartRequest.GetWorkflowIDReusePolicy(),\n\t\t); err != nil {\n\t\t\treturn nil, workflowExecution, historyBlob, err\n\t\t}\n\t\t// create as ID reuse\n\t\tcreateMode = persistence.CreateWorkflowModeWorkflowIDReuse\n\t\terr = wfContext.CreateWorkflowExecution(\n\t\t\tctx,\n\t\t\tnewWorkflow,\n\t\t\t*historyBlob,\n\t\t\tcreateMode,\n\t\t\tprevRunID,\n\t\t\tt.LastWriteVersion,\n\t\t\tpersistence.CreateWorkflowRequestModeNew,\n\t\t)\n\t\tif t, ok := persistence.AsDuplicateRequestError(err); ok {\n\t\t\tif t.RequestType == persistence.WorkflowRequestTypeStart || (isSignalWithStart && t.RequestType == persistence.WorkflowRequestTypeSignal) {\n\t\t\t\treturn &types.StartWorkflowExecutionResponse{\n\t\t\t\t\tRunID: t.RunID,\n\t\t\t\t}, workflowExecution, historyBlob, nil\n\t\t\t}\n\t\t\te.logger.Error(\"A bug is detected for idempotency improvement\", tag.Dynamic(\"request-type\", t.RequestType))\n\t\t\treturn nil, workflowExecution, historyBlob, t\n\t\t}\n\t}\n\tif err != nil {\n\t\treturn nil, workflowExecution, historyBlob, err\n\t}\n\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: workflowExecution.RunID,\n\t}, workflowExecution, historyBlob, nil\n}\n\nfunc (e *historyEngineImpl) handleCreateWorkflowExecutionFailureCleanup(\n\tctx context.Context,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowExecution *types.WorkflowExecution,\n\thistoryBlob *events.PersistedBlob,\n\tisSignalWithStart bool,\n\terr error,\n) {\n\n\tif workflowExecution == nil || historyBlob == nil {\n\t\t// expected behaviour, errors caused by validation will not have a workflow execution\n\t\treturn\n\t}\n\n\t// we cannot know if the workflow has succeeded, so the safest thing to do is\n\t// to do nothing. Deleting the history risks breaking a valid execution and prevrenting it from being restarted\n\tif persistence.IsTimeoutError(err) {\n\t\te.logger.Warn(\"timeout error detected when creating execution. If the excution was not created successfully this has probably left some orphaned history branch that needs cleaning up\",\n\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\ttag.Error(err))\n\t\te.metricsClient.IncCounter(metrics.HistoryEngineScope, metrics.WorkflowCreationFailedCleanupHaltedTimeoutCount)\n\t\treturn\n\t}\n\n\t// this is the set of errors for which we condider safe to history branch created\n\t// earlier when starting the workflow. It does not include any\n\t// errors where the state of the workflow is not known (like timeouts, DB internal errors)\n\tisKnownFailureRequiringCleanup := errors.As(err, new(*types.WorkflowExecutionAlreadyStartedError)) ||\n\t\terrors.As(err, new(*types.ShardOwnershipLostError)) ||\n\t\terrors.As(err, new(*persistence.CurrentWorkflowConditionFailedError)) ||\n\t\terrors.As(err, new(*persistence.ConditionFailedError)) ||\n\t\terrors.As(err, new(*persistence.ShardAlreadyExistError)) ||\n\t\terrors.As(err, new(*persistence.WorkflowExecutionAlreadyStartedError)) ||\n\t\terrors.As(err, new(*types.ShardOwnershipLostError)) ||\n\t\terrors.As(err, new(*types.WorkflowExecutionAlreadyStartedError)) ||\n\t\terrors.As(err, new(*types.WorkflowExecutionAlreadyCompletedError)) ||\n\t\terrors.As(err, new(*persistence.ShardOwnershipLostError)) ||\n\t\terrors.As(err, new(*persistence.DuplicateRequestError))\n\n\tif !isKnownFailureRequiringCleanup {\n\t\te.logger.Warn(\"unknown failure detected when creating execution. If the excution was not created successfully this has probably left some orphaned history branch that needs cleaning up\",\n\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\ttag.Error(err))\n\t\te.metricsClient.IncCounter(metrics.HistoryEngineScope, metrics.WorkflowCreationFailedCleanupUnknownCount)\n\t\treturn\n\t}\n\n\tif !e.shard.GetConfig().EnableCleanupOrphanedHistoryBranchOnWorkflowCreation(domainEntry.GetInfo().Name) {\n\t\te.logger.Warn(\"cleanup of orphaned history branch is disabled, but possible orphaned history branch was detected\",\n\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\ttag.Dynamic(\"debug-isSignalWithStart\", isSignalWithStart),\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn\n\t}\n\n\tif startRequest == nil || domainEntry == nil || workflowExecution.WorkflowID == \"\" || workflowExecution.RunID == \"\" {\n\t\te.logger.Error(\"some parameters are missing in handleCreateWorkflowExecutionFailureCleanup. This is a bug\",\n\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\ttag.Dynamic(\"debug-startRequest\", startRequest),\n\t\t\ttag.Dynamic(\"debug-historyBlob\", historyBlob),\n\t\t\ttag.Dynamic(\"debug-domainEntry\", domainEntry),\n\t\t\ttag.Dynamic(\"debug-workflowExecution\", workflowExecution),\n\t\t\ttag.Error(err))\n\t\treturn\n\t}\n\n\t// The key here is to delete the additational branch, since it has just been created earlier in this call\n\t// and is not used. However, we must be careful, there may be other existing branches that we must not touch\n\te.logger.Info(\"Deleting orphaned history branch during cleanup after identified failure during creation\",\n\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\ttag.Dynamic(\"debug-isSignalWithStart\", isSignalWithStart),\n\t\ttag.Error(err))\n\n\tcleanupErr := e.shard.GetHistoryManager().DeleteHistoryBranch(ctx, &persistence.DeleteHistoryBranchRequest{\n\t\tBranchToken: historyBlob.BranchToken,\n\t\tShardID:     common.IntPtr(e.shard.GetShardID()),\n\t\tDomainName:  domainEntry.GetInfo().Name,\n\t})\n\tif cleanupErr != nil {\n\t\te.logger.Warn(\"Failed to cleanup orphaned history branch\",\n\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\ttag.Error(cleanupErr))\n\t\te.metricsClient.IncCounter(metrics.HistoryEngineScope, metrics.WorkflowCreationFailedCleanupFailureCount)\n\t\treturn\n\t}\n\te.metricsClient.IncCounter(metrics.HistoryEngineScope, metrics.WorkflowCreationFailedCleanupSuccessCount)\n\tif e.shard.GetConfig().EnableRecordWorkflowExecutionUninitialized(domainEntry.GetInfo().Name) && e.visibilityMgr != nil {\n\t\t// delete the uninitialized workflow execution record since it failed to start the workflow\n\t\t// uninitialized record is used to find wfs that didn't make a progress or stuck during the start process\n\t\tif errVisibility := e.visibilityMgr.DeleteWorkflowExecution(ctx, &persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\t\tDomainID:   domainEntry.GetInfo().ID,\n\t\t\tDomain:     domainEntry.GetInfo().Name,\n\t\t\tRunID:      workflowExecution.RunID,\n\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t}); errVisibility != nil {\n\t\t\te.logger.Warn(\"Failed to delete uninitialized workflow execution record after failed CreateWorkflowExecution\",\n\t\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\t\ttag.Error(errVisibility))\n\t\t}\n\t}\n}\n\n// TODO(active-active): Review this method for active-active domains\nfunc (e *historyEngineImpl) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\tsignalWithStartRequest *types.HistorySignalWithStartWorkflowExecutionRequest,\n) (retResp *types.StartWorkflowExecutionResponse, retError error) {\n\tdomainEntry, err := e.shard.GetDomainCache().GetDomainByID(signalWithStartRequest.DomainUUID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif domainEntry.GetInfo().Status != persistence.DomainStatusRegistered {\n\t\treturn nil, errDomainDeprecated\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\tsRequest := signalWithStartRequest.SignalWithStartRequest\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: sRequest.WorkflowID,\n\t}\n\n\tvar prevMutableState execution.MutableState\n\tattempt := 0\n\n\twfContext, release, err0 := e.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, workflowExecution)\n\n\tif err0 == nil {\n\t\tdefer func() { release(retError) }()\n\tJust_Signal_Loop:\n\t\tfor ; attempt < workflow.ConditionalRetryCount; attempt++ {\n\t\t\t// workflow not exist, will create workflow then signal\n\t\t\tmutableState, err1 := wfContext.LoadWorkflowExecution(ctx)\n\t\t\tif err1 != nil {\n\t\t\t\tif _, ok := err1.(*types.EntityNotExistsError); ok {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\treturn nil, err1\n\t\t\t}\n\n\t\t\tif mutableState.IsSignalRequested(sRequest.GetRequestID()) {\n\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: wfContext.GetExecution().RunID}, nil\n\t\t\t}\n\n\t\t\t// workflow exist but not running, will restart workflow then signal\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\tprevMutableState = mutableState\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\t// workflow exists but history is corrupted, will restart workflow then signal\n\t\t\tif corrupted, err := e.checkForHistoryCorruptions(ctx, mutableState); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t} else if corrupted {\n\t\t\t\tprevMutableState = mutableState\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\t// workflow is running, if policy is TerminateIfRunning, terminate current run then signalWithStart\n\t\t\tif sRequest.GetWorkflowIDReusePolicy() == types.WorkflowIDReusePolicyTerminateIfRunning {\n\t\t\t\tworkflowExecution.RunID = uuid.New()\n\t\t\t\trunningWFCtx := workflow.NewContext(wfContext, release, mutableState)\n\t\t\t\tresp, errTerm := e.terminateAndStartWorkflow(\n\t\t\t\t\tctx,\n\t\t\t\t\trunningWFCtx,\n\t\t\t\t\tworkflowExecution,\n\t\t\t\t\tdomainEntry,\n\t\t\t\t\tdomainID,\n\t\t\t\t\tnil,\n\t\t\t\t\tsignalWithStartRequest,\n\t\t\t\t)\n\t\t\t\t// By the time we try to terminate the workflow, it was already terminated\n\t\t\t\t// So continue as if we didn't need to terminate it in the first place\n\t\t\t\tif _, ok := errTerm.(*types.WorkflowExecutionAlreadyCompletedError); !ok {\n\t\t\t\t\treturn resp, errTerm\n\t\t\t\t}\n\t\t\t}\n\n\t\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\t\tmaxAllowedSignals := e.config.MaximumSignalsPerExecution(domainEntry.GetInfo().Name)\n\t\t\tif maxAllowedSignals > 0 && int(executionInfo.SignalCount) >= maxAllowedSignals {\n\t\t\t\te.logger.Info(\"Execution limit reached for maximum signals\", tag.WorkflowSignalCount(executionInfo.SignalCount),\n\t\t\t\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\t\t\t\ttag.WorkflowRunID(workflowExecution.GetRunID()),\n\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\treturn nil, workflow.ErrSignalsLimitExceeded\n\t\t\t}\n\n\t\t\trequestID := sRequest.GetRequestID()\n\t\t\tif requestID != \"\" {\n\t\t\t\tmutableState.AddSignalRequested(requestID)\n\t\t\t}\n\n\t\t\tif _, err := mutableState.AddWorkflowExecutionSignaled(\n\t\t\t\tsRequest.GetSignalName(),\n\t\t\t\tsRequest.GetSignalInput(),\n\t\t\t\tsRequest.GetIdentity(),\n\t\t\t\tsRequest.GetRequestID(),\n\t\t\t); err != nil {\n\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Unable to signal workflow execution.\"}\n\t\t\t}\n\n\t\t\t// Create a transfer task to schedule a decision task\n\t\t\t// Do not schedule if the workflow hasn't processed its first decision yet\n\t\t\t// (e.g. waiting for DelayStart or Cron timer)\n\t\t\tif !mutableState.HasPendingDecision() && mutableState.HasProcessedOrPendingDecision() {\n\t\t\t\t_, err := mutableState.AddDecisionTaskScheduledEvent(false)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, &types.InternalServiceError{Message: \"Failed to add decision scheduled event.\"}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// We apply the update to execution using optimistic concurrency.  If it fails due to a conflict then reload\n\t\t\t// the history and try the operation again.\n\t\t\te.logger.Debugf(\"SignalWithStartWorkflowExecution calling UpdateWorkflowExecutionAsActive for wfID %s\",\n\t\t\t\tworkflowExecution.GetWorkflowID(),\n\t\t\t)\n\t\t\tif err := wfContext.UpdateWorkflowExecutionAsActive(ctx, e.shard.GetTimeSource().Now()); err != nil {\n\t\t\t\tif t, ok := persistence.AsDuplicateRequestError(err); ok {\n\t\t\t\t\tif t.RequestType == persistence.WorkflowRequestTypeSignal {\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: t.RunID}, nil\n\t\t\t\t\t}\n\t\t\t\t\te.logger.Error(\"A bug is detected for idempotency improvement\", tag.Dynamic(\"request-type\", t.RequestType))\n\t\t\t\t\treturn nil, t\n\t\t\t\t}\n\t\t\t\tif execution.IsConflictError(err) {\n\t\t\t\t\tcontinue Just_Signal_Loop\n\t\t\t\t}\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: wfContext.GetExecution().RunID}, nil\n\t\t} // end for Just_Signal_Loop\n\t\tif attempt == workflow.ConditionalRetryCount {\n\t\t\treturn nil, workflow.ErrMaxAttemptsExceeded\n\t\t}\n\t} else {\n\t\tif _, ok := err0.(*types.EntityNotExistsError); !ok {\n\t\t\treturn nil, err0\n\t\t}\n\t\t// workflow not exist, will create workflow then signal\n\t}\n\n\t// Start workflow and signal\n\tstartRequest, err := getStartRequest(domainID, sRequest, signalWithStartRequest.PartitionConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tsigWithStartArg := &signalWithStartArg{\n\t\tsignalWithStartRequest: signalWithStartRequest,\n\t\tprevMutableState:       prevMutableState,\n\t}\n\tresp, createdWFExecution, historyBlob, err := e.startWorkflowHelper(\n\t\tctx,\n\t\tstartRequest,\n\t\tdomainEntry,\n\t\tmetrics.HistorySignalWithStartWorkflowExecutionScope,\n\t\tsigWithStartArg,\n\t)\n\tif err != nil {\n\t\te.handleCreateWorkflowExecutionFailureCleanup(ctx,\n\t\t\tstartRequest,\n\t\t\tdomainEntry,\n\t\t\tcreatedWFExecution,\n\t\t\thistoryBlob,\n\t\t\ttrue,\n\t\t\terr,\n\t\t)\n\t\treturn nil, err\n\t}\n\treturn resp, nil\n}\n\nfunc getStartRequest(\n\tdomainID string,\n\trequest *types.SignalWithStartWorkflowExecutionRequest,\n\tpartitionConfig map[string]string,\n) (*types.HistoryStartWorkflowExecutionRequest, error) {\n\n\treq := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              request.Domain,\n\t\tWorkflowID:                          request.WorkflowID,\n\t\tWorkflowType:                        request.WorkflowType,\n\t\tTaskList:                            request.TaskList,\n\t\tInput:                               request.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: request.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      request.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            request.Identity,\n\t\tRequestID:                           request.RequestID,\n\t\tWorkflowIDReusePolicy:               request.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         request.RetryPolicy,\n\t\tCronSchedule:                        request.CronSchedule,\n\t\tCronOverlapPolicy:                   request.CronOverlapPolicy,\n\t\tMemo:                                request.Memo,\n\t\tSearchAttributes:                    request.SearchAttributes,\n\t\tHeader:                              request.Header,\n\t\tDelayStartSeconds:                   request.DelayStartSeconds,\n\t\tJitterStartSeconds:                  request.JitterStartSeconds,\n\t\tFirstRunAtTimeStamp:                 request.FirstRunAtTimestamp,\n\t\tActiveClusterSelectionPolicy:        request.ActiveClusterSelectionPolicy,\n\t}\n\n\treturn common.CreateHistoryStartWorkflowRequest(domainID, req, time.Now(), partitionConfig)\n}\n\nfunc shouldTerminateAndStart(startRequest *types.HistoryStartWorkflowExecutionRequest, state int) bool {\n\treturn startRequest.StartRequest.GetWorkflowIDReusePolicy() == types.WorkflowIDReusePolicyTerminateIfRunning &&\n\t\tpersistence.IsWorkflowRunning(state)\n}\n\nfunc (e *historyEngineImpl) validateStartWorkflowExecutionRequest(request *types.StartWorkflowExecutionRequest, metricsScope metrics.ScopeIdx) error {\n\tif len(request.GetRequestID()) == 0 {\n\t\treturn &types.BadRequestError{Message: \"Missing request ID.\"}\n\t}\n\tif request.ExecutionStartToCloseTimeoutSeconds == nil || request.GetExecutionStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn &types.BadRequestError{Message: \"Missing or invalid ExecutionStartToCloseTimeoutSeconds.\"}\n\t}\n\tif request.TaskStartToCloseTimeoutSeconds == nil || request.GetTaskStartToCloseTimeoutSeconds() <= 0 {\n\t\treturn &types.BadRequestError{Message: \"Missing or invalid TaskStartToCloseTimeoutSeconds.\"}\n\t}\n\tif request.TaskList == nil || request.TaskList.GetName() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Missing Tasklist.\"}\n\t}\n\tif request.WorkflowType == nil || request.WorkflowType.GetName() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Missing WorkflowType.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\trequest.GetDomain(),\n\t\te.metricsClient.Scope(metricsScope),\n\t\te.config.MaxIDLengthWarnLimit(),\n\t\te.config.DomainNameMaxLength(request.GetDomain()),\n\t\tmetrics.CadenceErrDomainNameExceededWarnLimit,\n\t\trequest.GetDomain(),\n\t\te.logger,\n\t\ttag.IDTypeDomainName) {\n\t\treturn &types.BadRequestError{Message: \"Domain exceeds length limit.\"}\n\t}\n\n\tif !common.IsValidIDLength(\n\t\trequest.GetWorkflowID(),\n\t\te.metricsClient.Scope(metricsScope),\n\t\te.config.MaxIDLengthWarnLimit(),\n\t\te.config.WorkflowIDMaxLength(request.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowIDExceededWarnLimit,\n\t\trequest.GetDomain(),\n\t\te.logger,\n\t\ttag.IDTypeWorkflowID) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowId exceeds length limit.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\trequest.TaskList.GetName(),\n\t\te.metricsClient.Scope(metricsScope),\n\t\te.config.MaxIDLengthWarnLimit(),\n\t\te.config.TaskListNameMaxLength(request.GetDomain()),\n\t\tmetrics.CadenceErrTaskListNameExceededWarnLimit,\n\t\trequest.GetDomain(),\n\t\te.logger,\n\t\ttag.IDTypeTaskListName) {\n\t\treturn &types.BadRequestError{Message: \"TaskList exceeds length limit.\"}\n\t}\n\tif !common.IsValidIDLength(\n\t\trequest.WorkflowType.GetName(),\n\t\te.metricsClient.Scope(metricsScope),\n\t\te.config.MaxIDLengthWarnLimit(),\n\t\te.config.WorkflowTypeMaxLength(request.GetDomain()),\n\t\tmetrics.CadenceErrWorkflowTypeExceededWarnLimit,\n\t\trequest.GetDomain(),\n\t\te.logger,\n\t\ttag.IDTypeWorkflowType) {\n\t\treturn &types.BadRequestError{Message: \"WorkflowType exceeds length limit.\"}\n\t}\n\n\treturn common.ValidateRetryPolicy(request.RetryPolicy)\n}\n\nfunc (e *historyEngineImpl) overrideTaskStartToCloseTimeoutSeconds(\n\tdomainEntry *cache.DomainCacheEntry,\n\trequest *types.StartWorkflowExecutionRequest,\n\tmetricsScope metrics.ScopeIdx,\n) {\n\tdomainName := domainEntry.GetInfo().Name\n\tmaxDecisionStartToCloseTimeoutSeconds := int32(e.config.MaxDecisionStartToCloseSeconds(domainName))\n\ttaskStartToCloseTimeoutSecs := request.GetTaskStartToCloseTimeoutSeconds()\n\ttaskStartToCloseTimeoutSecs = min(taskStartToCloseTimeoutSecs, maxDecisionStartToCloseTimeoutSeconds)\n\ttaskStartToCloseTimeoutSecs = min(taskStartToCloseTimeoutSecs, request.GetExecutionStartToCloseTimeoutSeconds())\n\n\tif taskStartToCloseTimeoutSecs != request.GetTaskStartToCloseTimeoutSeconds() {\n\t\trequest.TaskStartToCloseTimeoutSeconds = &taskStartToCloseTimeoutSecs\n\t\te.metricsClient.Scope(\n\t\t\tmetricsScope,\n\t\t\tmetrics.DomainTag(domainName),\n\t\t).IncCounter(metrics.DecisionStartToCloseTimeoutOverrideCount)\n\t}\n}\n\n// terminate running workflow then start a new run in one transaction\nfunc (e *historyEngineImpl) terminateAndStartWorkflow(\n\tctx context.Context,\n\trunningWFCtx workflow.Context,\n\tworkflowExecution types.WorkflowExecution,\n\tdomainEntry *cache.DomainCacheEntry,\n\tdomainID string,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n\tsignalWithStartRequest *types.HistorySignalWithStartWorkflowExecutionRequest,\n) (*types.StartWorkflowExecutionResponse, error) {\n\trunningMutableState := runningWFCtx.GetMutableState()\n\tvar err error\n\tif signalWithStartRequest != nil {\n\t\tstartRequest, err = getStartRequest(domainID, signalWithStartRequest.SignalWithStartRequest, signalWithStartRequest.PartitionConfig)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tactiveCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(runningMutableState.GetCurrentVersion())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif activeCluster != e.currentClusterName {\n\t\tif runningMutableState.GetExecutionInfo().ActiveClusterSelectionPolicy.Equals(startRequest.StartRequest.ActiveClusterSelectionPolicy) {\n\t\t\treturn nil, e.newDomainNotActiveError(domainEntry, runningMutableState.GetCurrentVersion())\n\t\t}\n\t\t// TODO(active-active): This is a short-term fix to handle this special case, because we don't have a way to terminate the existing workflow in a different cluster and start a new workflow in the current cluster\n\t\t// atomically in one transaction. We'll review this when we have time to implement a better solution.\n\t\treturn nil, &types.BadRequestError{Message: \"Cannot terminate the existing workflow and start a new workflow because it is active in a different cluster with a different active cluster selection policy.\"}\n\t}\nUpdateWorkflowLoop:\n\tfor attempt := 0; attempt < workflow.ConditionalRetryCount; attempt++ {\n\t\tif !runningMutableState.IsWorkflowExecutionRunning() {\n\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t}\n\n\t\tif err := execution.TerminateWorkflow(\n\t\t\trunningMutableState,\n\t\t\trunningMutableState.GetNextEventID(),\n\t\t\tTerminateIfRunningReason,\n\t\t\tgetTerminateIfRunningDetails(workflowExecution.GetRunID()),\n\t\t\texecution.IdentityHistoryService,\n\t\t); err != nil {\n\t\t\tif err == workflow.ErrStaleState {\n\t\t\t\t// Handler detected that cached workflow mutable could potentially be stale\n\t\t\t\t// Reload workflow execution history\n\t\t\t\trunningWFCtx.GetContext().Clear()\n\t\t\t\tif attempt != workflow.ConditionalRetryCount-1 {\n\t\t\t\t\t_, err = runningWFCtx.ReloadMutableState(ctx)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn nil, err\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tcontinue UpdateWorkflowLoop\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\n\t\t// new mutable state\n\t\tnewMutableState, err := e.createMutableState(ctx, domainEntry, workflowExecution.GetRunID(), startRequest)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\terr = e.addStartEventsAndTasks(\n\t\t\tnewMutableState,\n\t\t\tworkflowExecution,\n\t\t\tstartRequest,\n\t\t\tsignalWithStartRequest,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tupdateErr := runningWFCtx.GetContext().UpdateWorkflowExecutionWithNewAsActive(\n\t\t\tctx,\n\t\t\te.timeSource.Now(),\n\t\t\texecution.NewContext(\n\t\t\t\tdomainID,\n\t\t\t\tworkflowExecution,\n\t\t\t\te.shard,\n\t\t\t\te.shard.GetExecutionManager(),\n\t\t\t\te.logger,\n\t\t\t),\n\t\t\tnewMutableState,\n\t\t)\n\t\tif updateErr != nil {\n\t\t\tif execution.IsConflictError(updateErr) {\n\t\t\t\te.metricsClient.IncCounter(metrics.HistoryStartWorkflowExecutionScope, metrics.ConcurrencyUpdateFailureCounter)\n\t\t\t\tcontinue UpdateWorkflowLoop\n\t\t\t}\n\t\t\treturn nil, updateErr\n\t\t}\n\t\tbreak UpdateWorkflowLoop\n\t}\n\treturn &types.StartWorkflowExecutionResponse{\n\t\tRunID: workflowExecution.RunID,\n\t}, nil\n}\n\nfunc (e *historyEngineImpl) addStartEventsAndTasks(\n\tmutableState execution.MutableState,\n\tworkflowExecution types.WorkflowExecution,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n\tsignalWithStartRequest *types.HistorySignalWithStartWorkflowExecutionRequest,\n) error {\n\t// Add WF start event\n\tstartEvent, err := mutableState.AddWorkflowExecutionStartedEvent(\n\t\tworkflowExecution,\n\t\tstartRequest,\n\t)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"Failed to add workflow execution started event.\",\n\t\t}\n\t}\n\n\tif signalWithStartRequest != nil {\n\t\t// Add signal event\n\t\tsRequest := signalWithStartRequest.SignalWithStartRequest\n\t\tif sRequest.GetRequestID() != \"\" {\n\t\t\tmutableState.AddSignalRequested(sRequest.GetRequestID())\n\t\t}\n\t\t_, err := mutableState.AddWorkflowExecutionSignaled(\n\t\t\tsRequest.GetSignalName(),\n\t\t\tsRequest.GetSignalInput(),\n\t\t\tsRequest.GetIdentity(),\n\t\t\tsRequest.GetRequestID(),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn &types.InternalServiceError{Message: \"Failed to add workflow execution signaled event.\"}\n\t\t}\n\t}\n\n\t// Generate first decision task event if not child WF and no first decision task backoff\n\treturn e.generateFirstDecisionTask(\n\t\tmutableState,\n\t\tstartRequest.ParentExecutionInfo,\n\t\tstartEvent,\n\t)\n}\n\nfunc getTerminateIfRunningDetails(newRunID string) []byte {\n\treturn []byte(fmt.Sprintf(TerminateIfRunningDetailsTemplate, newRunID))\n}\n\nfunc (e *historyEngineImpl) applyWorkflowIDReusePolicyForSigWithStart(\n\tprevExecutionInfo *persistence.WorkflowExecutionInfo,\n\texecution types.WorkflowExecution,\n\twfIDReusePolicy types.WorkflowIDReusePolicy,\n) error {\n\n\tprevStartRequestID := prevExecutionInfo.CreateRequestID\n\tprevRunID := prevExecutionInfo.RunID\n\tprevState := prevExecutionInfo.State\n\tprevCloseState := prevExecutionInfo.CloseStatus\n\n\treturn e.applyWorkflowIDReusePolicyHelper(\n\t\tprevStartRequestID,\n\t\tprevRunID,\n\t\tprevState,\n\t\tprevCloseState,\n\t\texecution,\n\t\twfIDReusePolicy,\n\t)\n}\n\nfunc (e *historyEngineImpl) applyWorkflowIDReusePolicyHelper(\n\tprevStartRequestID,\n\tprevRunID string,\n\tprevState int,\n\tprevCloseState int,\n\texecution types.WorkflowExecution,\n\twfIDReusePolicy types.WorkflowIDReusePolicy,\n) error {\n\n\t// here we know some information about the prev workflow, i.e. either running right now\n\t// or has history check if the workflow is finished\n\tswitch prevState {\n\tcase persistence.WorkflowStateCreated,\n\t\tpersistence.WorkflowStateRunning:\n\t\tmsg := \"Workflow execution is already running. WorkflowId: %v, RunId: %v.\"\n\t\treturn getWorkflowAlreadyStartedError(msg, prevStartRequestID, execution.GetWorkflowID(), prevRunID)\n\tcase persistence.WorkflowStateCompleted:\n\t\t// previous workflow completed, proceed\n\tcase persistence.WorkflowStateCorrupted:\n\t\t// ignore workflow ID reuse policy for corrupted workflows, treat as they do not exist\n\t\treturn nil\n\tdefault:\n\t\t// persistence.WorkflowStateZombie or unknown type\n\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"Failed to process workflow, workflow has invalid state: %v.\", prevState)}\n\t}\n\n\tswitch wfIDReusePolicy {\n\tcase types.WorkflowIDReusePolicyAllowDuplicateFailedOnly:\n\t\tif _, ok := FailedWorkflowCloseState[prevCloseState]; !ok {\n\t\t\tmsg := \"Workflow execution already finished successfully. WorkflowId: %v, RunId: %v. Workflow ID reuse policy: allow duplicate workflow ID if last run failed.\"\n\t\t\treturn getWorkflowAlreadyStartedError(msg, prevStartRequestID, execution.GetWorkflowID(), prevRunID)\n\t\t}\n\tcase types.WorkflowIDReusePolicyAllowDuplicate,\n\t\ttypes.WorkflowIDReusePolicyTerminateIfRunning:\n\t\t// no check need here\n\tcase types.WorkflowIDReusePolicyRejectDuplicate:\n\t\tmsg := \"Workflow execution already finished. WorkflowId: %v, RunId: %v. Workflow ID reuse policy: reject duplicate workflow ID.\"\n\t\treturn getWorkflowAlreadyStartedError(msg, prevStartRequestID, execution.GetWorkflowID(), prevRunID)\n\tdefault:\n\t\treturn &types.InternalServiceError{Message: \"Failed to process start workflow reuse policy.\"}\n\t}\n\n\treturn nil\n}\n\nfunc getWorkflowAlreadyStartedError(errMsg string, createRequestID string, workflowID string, runID string) error {\n\treturn &types.WorkflowExecutionAlreadyStartedError{\n\t\tMessage:        fmt.Sprintf(errMsg, workflowID, runID),\n\t\tStartRequestID: createRequestID,\n\t\tRunID:          runID,\n\t}\n}\n\nfunc (e *historyEngineImpl) newChildContext(\n\tparentCtx context.Context,\n) (context.Context, context.CancelFunc) {\n\n\tctxTimeout := contextLockTimeout\n\tif deadline, ok := parentCtx.Deadline(); ok {\n\t\tnow := e.shard.GetTimeSource().Now()\n\t\tparentTimeout := deadline.Sub(now)\n\t\tif parentTimeout > 0 && parentTimeout < contextLockTimeout {\n\t\t\tctxTimeout = parentTimeout\n\t\t}\n\t}\n\treturn context.WithTimeout(context.Background(), ctxTimeout)\n}\n\nfunc (e *historyEngineImpl) createMutableState(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\trunID string,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n) (execution.MutableState, error) {\n\tactiveClusterInfo, err := e.shard.GetActiveClusterManager().GetActiveClusterInfoByClusterAttribute(ctx, domainEntry.GetInfo().ID, startRequest.StartRequest.ActiveClusterSelectionPolicy.GetClusterAttribute())\n\tif err != nil {\n\t\tvar errNotFound *activecluster.ClusterAttributeNotFoundError\n\t\tif !errors.As(err, &errNotFound) {\n\t\t\t// unexpected error\n\t\t\treturn nil, err\n\t\t}\n\t\te.logger.Warn(\"Failed to get active cluster info by cluster attribute, falling back to domain-level active cluster info\", tag.Error(err))\n\t\treturn nil, errClusterAttributeNotFound\n\t}\n\tif activeClusterInfo.ActiveClusterName != e.currentClusterName {\n\t\treturn nil, e.newDomainNotActiveError(domainEntry, activeClusterInfo.FailoverVersion)\n\t}\n\n\tnewMutableState := execution.NewMutableStateBuilderWithVersionHistories(\n\t\te.shard,\n\t\te.logger,\n\t\tdomainEntry,\n\t\tactiveClusterInfo.FailoverVersion,\n\t)\n\n\tif err := newMutableState.SetHistoryTree(runID); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn newMutableState, nil\n}\n\nfunc (e *historyEngineImpl) generateFirstDecisionTask(\n\tmutableState execution.MutableState,\n\tparentInfo *types.ParentExecutionInfo,\n\tstartEvent *types.HistoryEvent,\n) error {\n\n\tif parentInfo == nil {\n\t\t// DecisionTask is only created when it is not a Child Workflow and no backoff is needed\n\t\tif err := mutableState.AddFirstDecisionTaskScheduled(\n\t\t\tstartEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/start_workflow_execution_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/golang/mock/gomock\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n\t\"github.com/uber/cadence/service/history/events\"\n)\n\nfunc TestStartWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\trequest    *types.HistoryStartWorkflowExecutionRequest\n\t\tsetupMocks func(*testing.T, *testdata.EngineForTest)\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"start workflow execution success\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:       constants.TestDomainName,\n\t\t\t\t\tWorkflowID:   \"workflow-id\",\n\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"default-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tInput:                               []byte(\"workflow input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),   // 10 seconds\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id-for-start\",\n\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\tInitialIntervalInSeconds:    1,\n\t\t\t\t\t\tBackoffCoefficient:          2.0,\n\t\t\t\t\t\tMaximumIntervalInSeconds:    10,\n\t\t\t\t\t\tMaximumAttempts:             5,\n\t\t\t\t\t\tExpirationIntervalInSeconds: 3600, // 1 hour\n\t\t\t\t\t},\n\t\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\t\tFields: map[string][]byte{\n\t\t\t\t\t\t\t\"key1\": []byte(\"value1\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\t\"CustomKeywordField\": []byte(\"test\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.CreateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:                                      1,\n\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get workflow execution\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tInput:                               []byte(\"workflow input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),   // 10 seconds\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id-for-start\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"internal error\")).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"prev mutable state version conflict\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tInput:                               []byte(\"workflow input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"default-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tIdentity:              \"workflow-starter\",\n\t\t\t\t\tRequestID:             \"request-id-for-start\",\n\t\t\t\t\tWorkflowType:          &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tWorkflowIDReusePolicy: types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"version conflict\")).Once()\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"internal error\")).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"workflow ID reuse - terminate if running\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tInput:                               []byte(\"workflow input\"),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"default-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tIdentity:              \"workflow-starter\",\n\t\t\t\t\tRequestID:             \"request-id-for-start\",\n\t\t\t\t\tWorkflowType:          &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tWorkflowIDReusePolicy: types.WorkflowIDReusePolicyTerminateIfRunning.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\t// Simulate the termination and recreation process\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"TerminateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.CreateWorkflowExecutionResponse{}, nil).Once()\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"workflow ID reuse policy - reject duplicate\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"default-task-list\"},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),   // 10 seconds\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id-for-start\",\n\t\t\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyRejectDuplicate.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\tStartRequestID: \"existing-request-id\",\n\t\t\t\t\tRunID:          \"existing-run-id\",\n\t\t\t\t}).Once()\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t\thistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\ttc.setupMocks(t, eft)\n\n\t\t\t_, err := eft.Engine.StartWorkflowExecution(context.Background(), tc.request)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"%s: StartWorkflowExecution() error = %v, wantErr %v\", tc.name, err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestStartWorkflowExecution_OrphanedHistoryCleanup(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\trequest              *types.HistoryStartWorkflowExecutionRequest\n\t\tsetupMocks           func(*testing.T, *testdata.EngineForTest)\n\t\tenableCleanupFlag    bool\n\t\texpectHistoryCleanup bool\n\t\twantErr              bool\n\t}{\n\t\t{\n\t\t\tname: \"cleanup orphaned history on WorkflowExecutionAlreadyStartedError with flag enabled\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"default-task-list\"},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\t\t\t\tvar capturedBranchToken []byte\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tRun(func(args mock.Arguments) {\n\t\t\t\t\t\treq := args.Get(1).(*persistence.AppendHistoryNodesRequest)\n\t\t\t\t\t\tcapturedBranchToken = req.BranchToken\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil, &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\t\tStartRequestID: \"different-request-id\",\n\t\t\t\t\t\tRunID:          \"existing-run-id\",\n\t\t\t\t\t\tState:          persistence.WorkflowStateCompleted,\n\t\t\t\t\t}).Once()\n\n\t\t\t\thistoryV2Mgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.DeleteHistoryBranchRequest) bool {\n\t\t\t\t\treturn assert.Equal(t, capturedBranchToken, req.BranchToken) &&\n\t\t\t\t\t\tassert.Equal(t, constants.TestDomainName, req.DomainName)\n\t\t\t\t})).\n\t\t\t\t\tReturn(nil).Once()\n\t\t\t},\n\t\t\tenableCleanupFlag:    true,\n\t\t\texpectHistoryCleanup: true,\n\t\t\twantErr:              true,\n\t\t},\n\t\t{\n\t\t\tname: \"no cleanup when flag disabled on WorkflowExecutionAlreadyStartedError\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"default-task-list\"},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil, &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\t\t\tStartRequestID: \"different-request-id\",\n\t\t\t\t\t\tRunID:          \"existing-run-id\",\n\t\t\t\t\t\tState:          persistence.WorkflowStateCompleted,\n\t\t\t\t\t}).Once()\n\t\t\t},\n\t\t\tenableCleanupFlag:    false,\n\t\t\texpectHistoryCleanup: false,\n\t\t\twantErr:              true,\n\t\t},\n\t\t{\n\t\t\tname: \"no cleanup on DuplicateRequestError with WorkflowRequestTypeStart (returns success)\",\n\t\t\trequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"default-task-list\"},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tIdentity:                            \"workflow-starter\",\n\t\t\t\t\tRequestID:                           \"request-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{}).Once()\n\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil, &persistence.DuplicateRequestError{\n\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t\tRunID:       \"existing-run-id\",\n\t\t\t\t\t}).Once()\n\t\t\t\t// No DeleteHistoryBranch mock - cleanup doesn't happen for this case\n\t\t\t},\n\t\t\tenableCleanupFlag:    true,\n\t\t\texpectHistoryCleanup: false, // Cleanup doesn't happen because success is returned\n\t\t\twantErr:              false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\teft.ShardCtx.Resource.ShardMgr.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\t\t\teft.ShardCtx.GetConfig().EnableCleanupOrphanedHistoryBranchOnWorkflowCreation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(tc.enableCleanupFlag)\n\n\t\t\ttc.setupMocks(t, eft)\n\n\t\t\t_, err := eft.Engine.StartWorkflowExecution(context.Background(), tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\tif tc.expectHistoryCleanup {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.AssertCalled(t, \"DeleteHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.DeleteHistoryBranchRequest) bool {\n\t\t\t\t\treturn req.BranchToken != nil\n\t\t\t\t}))\n\t\t\t} else {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.AssertNotCalled(t, \"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\"))\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSignalWithStartWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tsetupMocks func(*testing.T, *testdata.EngineForTest)\n\t\trequest    *types.HistorySignalWithStartWorkflowExecutionRequest\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"signal and start workflow successfully\",\n\t\t\trequest: &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          \"workflow-id\",\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tSignalName:                          \"signal-name\",\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"default-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tRequestID:   \"request-id-for-start\",\n\t\t\t\t\tSignalInput: []byte(\"signal-input\"),\n\t\t\t\t\tIdentity:    \"tester\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\t// Mock GetCurrentExecution to simulate a non-existent current execution\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: \"workflow-id\",\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       \"\", // No current run ID indicates no current execution\n\t\t\t\t\tState:       persistence.WorkflowStateCompleted,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).Return(getCurrentExecResp, &types.EntityNotExistsError{}).Once()\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.CreateWorkflowExecutionResponse{}, nil)\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"terminate existing and start new workflow\",\n\t\t\trequest: &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                          constants.TestWorkflowID,\n\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\t\t\t\tSignalName:                          \"signal-name\",\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(3600), // 1 hour\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"default-task-list\",\n\t\t\t\t\t},\n\t\t\t\t\tRequestID:             \"request-id-for-start\",\n\t\t\t\t\tSignalInput:           []byte(\"signal-input\"),\n\t\t\t\t\tIdentity:              \"tester\",\n\t\t\t\t\tWorkflowIDReusePolicy: (*types.WorkflowIDReusePolicy)(common.Int32Ptr(3)),\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tdomainEntry := &cache.DomainCacheEntry{}\n\t\t\t\teft.ShardCtx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainEntry, nil).AnyTimes()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), constants.TestDomainID, nil).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\t\t\t\t// Simulate current workflow execution is running\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).Return(getCurrentExecResp, nil).Once()\n\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).\n\t\t\t\t\tOnce()\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t\tvar _ *persistence.UpdateWorkflowExecutionRequest\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tRun(func(args mock.Arguments) {\n\t\t\t\t\t\tvar ok bool\n\t\t\t\t\t\t_, ok = args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\t\t\t\tif !ok {\n\t\t\t\t\t\t\tt.Fatalf(\"failed to cast input to *persistence.UpdateWorkflowExecutionRequest, type is %T\", args.Get(1))\n\t\t\t\t\t\t}\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(updateExecResp, nil).\n\t\t\t\t\tOnce()\n\t\t\t\t// Expect termination of the current workflow\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"TerminateWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\t\t\t\t// Expect creation of a new workflow execution\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"CreateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.CreateWorkflowExecutionResponse{}, nil).Once()\n\n\t\t\t\t// Mocking additional interactions required by the workflow context and execution\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).Return(&persistence.AppendHistoryNodesResponse{}, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\ttc.setupMocks(t, eft)\n\n\t\t\tresponse, err := eft.Engine.SignalWithStartWorkflowExecution(context.Background(), tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, response)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateMutableState(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tdomainEntry    *cache.DomainCacheEntry\n\t\tmockFn         func(ac *activecluster.MockManager)\n\t\twantErr        bool\n\t\twantVersion    int64\n\t\twantErrMessage string\n\t}{\n\t\t{\n\t\t\tname: \"create mutable state successfully, failover version is looked up from active cluster manager\",\n\t\t\tdomainEntry: getDomainCacheEntry(\n\t\t\t\t0,\n\t\t\t\t&types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\tmockFn: func(ac *activecluster.MockManager) {\n\t\t\t\tac.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tFailoverVersion:   125,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantVersion: 125,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to create mutable state, current cluster is not the active cluster\",\n\t\t\tdomainEntry: getDomainCacheEntry(\n\t\t\t\t0,\n\t\t\t\t&types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\tmockFn: func(ac *activecluster.MockManager) {\n\t\t\t\tac.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t\tFailoverVersion:   125,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twantErr:        true,\n\t\t\twantErrMessage: \"is active in cluster(s): [cluster1 cluster2]\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to create mutable state. GetActiveClusterInfoByClusterAttribute failed\",\n\t\t\tdomainEntry: getDomainCacheEntry(\n\t\t\t\t0,\n\t\t\t\t&types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\tmockFn: func(ac *activecluster.MockManager) {\n\t\t\t\tac.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr:        true,\n\t\t\twantErrMessage: \"some error\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to create mutable state, cluster attribute not found\",\n\t\t\tdomainEntry: getDomainCacheEntry(\n\t\t\t\t0,\n\t\t\t\t&types.ActiveClusters{\n\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   0,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\"us-east\": {\n\t\t\t\t\t\t\t\t\tActiveClusterName: \"cluster2\",\n\t\t\t\t\t\t\t\t\tFailoverVersion:   2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}),\n\t\t\tmockFn: func(ac *activecluster.MockManager) {\n\t\t\t\tac.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &activecluster.ClusterAttributeNotFoundError{})\n\t\t\t},\n\t\t\twantErr:        true,\n\t\t\twantErrMessage: \"Cannot start workflow with a cluster attribute that is not found in the domain's metadata.\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\t\t\tengine := eft.Engine.(*historyEngineImpl)\n\n\t\t\tif tc.mockFn != nil {\n\t\t\t\ttc.mockFn(eft.ShardCtx.Resource.ActiveClusterMgr)\n\t\t\t}\n\n\t\t\tmutableState, err := engine.createMutableState(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.domainEntry,\n\t\t\t\t\"rid\",\n\t\t\t\t&types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t\t},\n\t\t\t)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.wantErrMessage)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, mutableState)\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tgotVer := mutableState.GetCurrentVersion()\n\t\t\tassert.Equal(t, tc.wantVersion, gotVer)\n\t\t})\n\t}\n}\n\nfunc TestHandleCreateWorkflowExecutionFailureCleanup(t *testing.T) {\n\t// Known error types that should trigger cleanup\n\tknownCleanupError := &persistence.WorkflowExecutionAlreadyStartedError{\n\t\tMsg: \"workflow already started\",\n\t}\n\n\ttests := []struct {\n\t\tname                          string\n\t\tenableCleanupFlag             bool\n\t\tenableUninitializedRecordFlag bool\n\t\terr                           error\n\t\tworkflowExecution             *types.WorkflowExecution\n\t\thistoryBlob                   []byte\n\t\tstartRequest                  *types.HistoryStartWorkflowExecutionRequest\n\t\tsetupMocks                    func(*testdata.EngineForTest)\n\t\texpectDeleteHistoryBranch     bool\n\t\texpectDeleteVisibility        bool\n\t}{\n\t\t{\n\t\t\tname:              \"workflowExecution is nil - returns early\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: nil,\n\t\t\thistoryBlob:       []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"historyBlob is nil - returns early\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: nil,\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"timeout error - returns early without cleanup\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               &persistence.TimeoutError{Msg: \"timeout\"},\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"unknown error - returns early without cleanup\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               errors.New(\"some unknown error\"),\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"cleanup disabled - returns early\",\n\t\t\tenableCleanupFlag: false,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"startRequest is nil - returns early with bug log\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob:               []byte(\"branch-token\"),\n\t\t\tstartRequest:              nil,\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"workflowID is empty - returns early with bug log\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"runID is empty - returns early with bug log\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks:                func(eft *testdata.EngineForTest) {},\n\t\t\texpectDeleteHistoryBranch: false,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"cleanup path with WorkflowExecutionAlreadyStartedError - deletes history branch successfully\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks: func(eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.DeleteHistoryBranchRequest) bool {\n\t\t\t\t\treturn string(req.BranchToken) == \"branch-token\" &&\n\t\t\t\t\t\treq.DomainName == constants.TestDomainName\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectDeleteHistoryBranch: true,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"cleanup path with DuplicateRequestError - deletes history branch successfully\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr: &persistence.DuplicateRequestError{\n\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\tRunID:       \"existing-run-id\",\n\t\t\t},\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks: func(eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\")).\n\t\t\t\t\tReturn(nil).Once()\n\t\t\t},\n\t\t\texpectDeleteHistoryBranch: true,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:              \"cleanup path - delete history branch fails gracefully\",\n\t\t\tenableCleanupFlag: true,\n\t\t\terr:               knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks: func(eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\")).\n\t\t\t\t\tReturn(errors.New(\"delete failed\")).Once()\n\t\t\t},\n\t\t\texpectDeleteHistoryBranch: true,\n\t\t\texpectDeleteVisibility:    false,\n\t\t},\n\t\t{\n\t\t\tname:                          \"cleanup path with visibility - deletes both history and visibility\",\n\t\t\tenableCleanupFlag:             true,\n\t\t\tenableUninitializedRecordFlag: true,\n\t\t\terr:                           knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks: func(eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\")).\n\t\t\t\t\tReturn(nil).Once()\n\t\t\t\teft.ShardCtx.Resource.VisibilityMgr.On(\"DeleteWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.VisibilityDeleteWorkflowExecutionRequest) bool {\n\t\t\t\t\treturn req.WorkflowID == \"wf-id\" &&\n\t\t\t\t\t\treq.RunID == \"run-id\" &&\n\t\t\t\t\t\treq.Domain == constants.TestDomainName\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectDeleteHistoryBranch: true,\n\t\t\texpectDeleteVisibility:    true,\n\t\t},\n\t\t{\n\t\t\tname:                          \"cleanup path with visibility - visibility delete fails gracefully\",\n\t\t\tenableCleanupFlag:             true,\n\t\t\tenableUninitializedRecordFlag: true,\n\t\t\terr:                           knownCleanupError,\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\tRunID:      \"run-id\",\n\t\t\t},\n\t\t\thistoryBlob: []byte(\"branch-token\"),\n\t\t\tstartRequest: &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{},\n\t\t\t},\n\t\t\tsetupMocks: func(eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.On(\"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\")).\n\t\t\t\t\tReturn(nil).Once()\n\t\t\t\teft.ShardCtx.Resource.VisibilityMgr.On(\"DeleteWorkflowExecution\", mock.Anything, mock.AnythingOfType(\"*persistence.VisibilityDeleteWorkflowExecutionRequest\")).\n\t\t\t\t\tReturn(errors.New(\"visibility delete failed\")).Once()\n\t\t\t},\n\t\t\texpectDeleteHistoryBranch: true,\n\t\t\texpectDeleteVisibility:    true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\t// Configure flags\n\t\t\teft.ShardCtx.GetConfig().EnableCleanupOrphanedHistoryBranchOnWorkflowCreation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(tc.enableCleanupFlag)\n\t\t\teft.ShardCtx.GetConfig().EnableRecordWorkflowExecutionUninitialized = dynamicproperties.GetBoolPropertyFnFilteredByDomain(tc.enableUninitializedRecordFlag)\n\n\t\t\t// Setup mocks\n\t\t\ttc.setupMocks(eft)\n\n\t\t\t// Get the engine implementation\n\t\t\tengine := eft.Engine.(*historyEngineImpl)\n\n\t\t\t// Create domain entry\n\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{\n\t\t\t\t\tID:   constants.TestDomainID,\n\t\t\t\t\tName: constants.TestDomainName,\n\t\t\t\t},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t1,\n\t\t\t\t1,\n\t\t\t\t1,\n\t\t\t)\n\n\t\t\t// Create history blob if provided\n\t\t\tvar historyBlob *events.PersistedBlob\n\t\t\tif tc.historyBlob != nil {\n\t\t\t\thistoryBlob = &events.PersistedBlob{\n\t\t\t\t\tBranchToken: tc.historyBlob,\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// Call the function\n\t\t\tengine.handleCreateWorkflowExecutionFailureCleanup(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttc.startRequest,\n\t\t\t\tdomainEntry,\n\t\t\t\ttc.workflowExecution,\n\t\t\t\thistoryBlob,\n\t\t\t\tfalse,\n\t\t\t\ttc.err,\n\t\t\t)\n\n\t\t\t// Assert mocks\n\t\t\tif tc.expectDeleteHistoryBranch {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.AssertCalled(t, \"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\"))\n\t\t\t} else {\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.AssertNotCalled(t, \"DeleteHistoryBranch\", mock.Anything, mock.AnythingOfType(\"*persistence.DeleteHistoryBranchRequest\"))\n\t\t\t}\n\n\t\t\tif tc.expectDeleteVisibility {\n\t\t\t\teft.ShardCtx.Resource.VisibilityMgr.AssertCalled(t, \"DeleteWorkflowExecution\", mock.Anything, mock.AnythingOfType(\"*persistence.VisibilityDeleteWorkflowExecutionRequest\"))\n\t\t\t} else if tc.enableUninitializedRecordFlag {\n\t\t\t\t// Only assert not called if the flag was enabled (otherwise it wouldn't be called anyway)\n\t\t\t\teft.ShardCtx.Resource.VisibilityMgr.AssertNotCalled(t, \"DeleteWorkflowExecution\", mock.Anything, mock.AnythingOfType(\"*persistence.VisibilityDeleteWorkflowExecutionRequest\"))\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getDomainCacheEntry(domainFailoverVersion int64, cfg *types.ActiveClusters) *cache.DomainCacheEntry {\n\t// only thing we care in domain cache entry is the active clusters config\n\treturn cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID: \"domain-id\",\n\t\t},\n\t\tnil,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters:    cfg,\n\t\t\tActiveClusterName: \"cluster0\",\n\t\t},\n\t\tdomainFailoverVersion,\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t1,\n\t)\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/terminate_workflow_execution.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage engineimpl\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/workflow\"\n)\n\nfunc (e *historyEngineImpl) TerminateWorkflowExecution(\n\tctx context.Context,\n\tterminateRequest *types.HistoryTerminateWorkflowExecutionRequest,\n) error {\n\trequest := terminateRequest.TerminateRequest\n\tparentExecution := terminateRequest.ExternalWorkflowExecution\n\tchildWorkflowOnly := terminateRequest.GetChildWorkflowOnly()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowExecution.WorkflowID,\n\t}\n\t// If firstExecutionRunID is set on the request always try to cancel currently running execution\n\tif request.GetFirstExecutionRunID() == \"\" {\n\t\tworkflowExecution.RunID = request.WorkflowExecution.RunID\n\t}\n\n\tdomainEntry, err := e.getActiveDomainByWorkflow(ctx, terminateRequest.DomainUUID, workflowExecution.WorkflowID, workflowExecution.RunID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomainID := domainEntry.GetInfo().ID\n\n\treturn workflow.UpdateCurrentWithActionFunc(\n\t\tctx,\n\t\te.logger,\n\t\te.executionCache,\n\t\te.executionManager,\n\t\tdomainID,\n\t\te.shard.GetDomainCache(),\n\t\tworkflowExecution,\n\t\te.timeSource.Now(),\n\t\tfunc(wfContext execution.Context, mutableState execution.MutableState) (*workflow.UpdateAction, error) {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn nil, workflow.ErrAlreadyCompleted\n\t\t\t}\n\n\t\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\t\tif request.GetFirstExecutionRunID() != \"\" {\n\t\t\t\tfirstRunID := executionInfo.FirstExecutionRunID\n\t\t\t\tif firstRunID == \"\" {\n\t\t\t\t\t// This is needed for backwards compatibility.  Workflow execution create with Cadence release v0.25.0 or earlier\n\t\t\t\t\t// does not have FirstExecutionRunID stored as part of mutable state.  If this is not set then load it from\n\t\t\t\t\t// workflow execution started event.\n\t\t\t\t\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn nil, err\n\t\t\t\t\t}\n\t\t\t\t\tfirstRunID = startEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstExecutionRunID()\n\t\t\t\t}\n\t\t\t\tif request.GetFirstExecutionRunID() != firstRunID {\n\t\t\t\t\treturn nil, &types.EntityNotExistsError{Message: \"Workflow execution not found\"}\n\t\t\t\t}\n\t\t\t}\n\t\t\tif childWorkflowOnly {\n\t\t\t\tparentWorkflowID := executionInfo.ParentWorkflowID\n\t\t\t\tparentRunID := executionInfo.ParentRunID\n\t\t\t\tif parentExecution.GetWorkflowID() != parentWorkflowID ||\n\t\t\t\t\tparentExecution.GetRunID() != parentRunID {\n\t\t\t\t\treturn nil, workflow.ErrParentMismatch\n\t\t\t\t}\n\t\t\t}\n\n\t\t\teventBatchFirstEventID := mutableState.GetNextEventID()\n\t\t\treturn workflow.UpdateWithoutDecision, execution.TerminateWorkflow(\n\t\t\t\tmutableState,\n\t\t\t\teventBatchFirstEventID,\n\t\t\t\trequest.GetReason(),\n\t\t\t\trequest.GetDetails(),\n\t\t\t\trequest.GetIdentity(),\n\t\t\t)\n\t\t})\n}\n"
  },
  {
    "path": "service/history/engine/engineimpl/terminate_workflow_execution_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\npackage engineimpl\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine/testdata\"\n)\n\nfunc TestTerminateWorkflowExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tterminationRequest types.HistoryTerminateWorkflowExecutionRequest\n\t\tsetupMocks         func(*testing.T, *testdata.EngineForTest)\n\t\twantErr            bool\n\t}{\n\t\t{\n\t\t\tname: \"domain is not active\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID:        constants.TestDomainID,\n\t\t\t\tChildWorkflowOnly: true,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:            \"Test termination\",\n\t\t\t\t\tIdentity:          \"testRunner\", // Specifically testing child workflow scenario\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"aaa\"}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"runid is not uuid\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: \"not-a-uuid\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, \"not-a-uuid\").Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil, errors.New(\"invalid UUID\")).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get workflow execution\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(nil, errors.New(\"some random error\")).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow parent mismatch\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID:        constants.TestDomainID,\n\t\t\t\tChildWorkflowOnly: true,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:            \"Test termination\",\n\t\t\t\t\tIdentity:          \"testRunner\", // Specifically testing child workflow scenario\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\t// Mock the retrieval of the workflow execution details\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:       constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:            constants.TestRunID,\n\t\t\t\t\t\t\tParentWorkflowID: \"other-parent-id\",\n\t\t\t\t\t\t\tParentRunID:      \"other-parent-runid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\t// Mock the retrieval of the workflow's history branch\n\t\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:                                      1,\n\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\t// Mock the update of the workflow execution\n\t\t\t\tvar _ *persistence.UpdateWorkflowExecutionRequest\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(updateExecResp, &types.EntityNotExistsError{Message: \"Workflow execution not found due to parent mismatch\"}).\n\t\t\t\t\tOnce()\n\n\t\t\t\t// Mock the update of the shard's range ID\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\n\t\t\t\t// Mock appending history nodes\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"valid first execution run ID\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:              constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution:   &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:              \"Test termination\",\n\t\t\t\t\tIdentity:            \"testRunner\",\n\t\t\t\t\tFirstExecutionRunID: \"\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:            constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:          constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:               constants.TestRunID,\n\t\t\t\t\t\t\tFirstExecutionRunID: \"\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.UpdateWorkflowExecutionResponse{}, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\t// Mock GetCurrentExecution call\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).\n\t\t\t\t\tReturn(getCurrentExecResp, nil).Once()\n\n\t\t\t\t// Mock the retrieval of the workflow's history branch\n\t\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\t\tFirstExecutionRunID: \"fetched-first-run-id\", // The ID to be fetched\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\teft.ShardCtx.Resource.HistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\t\t\t\t// Mock the update of the workflow execution\n\t\t\t\tvar _ *persistence.UpdateWorkflowExecutionRequest\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(updateExecResp, &types.EntityNotExistsError{Message: \"Workflow execution not found due to parent mismatch\"}).\n\t\t\t\t\tOnce()\n\n\t\t\t\t// Mock the update of the shard's range ID\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\n\t\t\t\t// Mock appending history nodes\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"successful termination of a running workflow\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain: constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tReason:   \"Test termination\",\n\t\t\t\t\tIdentity: \"testRunner\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\t// ReadHistoryBranch prep\n\t\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t// first event.\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID:                                      1,\n\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\t\t\t\tvar _ *persistence.UpdateWorkflowExecutionRequest\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tRun(func(args mock.Arguments) {\n\t\t\t\t\t\tvar ok bool\n\t\t\t\t\t\t_, ok = args.Get(1).(*persistence.UpdateWorkflowExecutionRequest)\n\t\t\t\t\t\tif !ok {\n\t\t\t\t\t\t\tt.Fatalf(\"failed to cast input to *persistence.UpdateWorkflowExecutionRequest, type is %T\", args.Get(1))\n\t\t\t\t\t\t}\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(updateExecResp, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\t// UpdateShard prep. this is needed to update the shard's rangeID for failure cases.\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once() // Adjust the return values based on your test case needs\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"first execution run ID matches\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:              constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution:   &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:              \"Test termination\",\n\t\t\t\t\tIdentity:            \"testRunner\",\n\t\t\t\t\tFirstExecutionRunID: \"matching-first-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:            constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:          constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:               constants.TestRunID,\n\t\t\t\t\t\t\tFirstExecutionRunID: \"matching-first-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, \"\").Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).\n\t\t\t\t\tReturn(getCurrentExecResp, nil).Once()\n\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{}, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(updateExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"first execution run ID does not match\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:              constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution:   &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:              \"Test termination\",\n\t\t\t\t\tIdentity:            \"testRunner\",\n\t\t\t\t\tFirstExecutionRunID: \"non-matching-first-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:            constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:          constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:               constants.TestRunID,\n\t\t\t\t\t\t\tFirstExecutionRunID: \"matching-first-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, \"\").Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).\n\t\t\t\t\tReturn(getCurrentExecResp, nil).Once()\n\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.ReadHistoryBranchResponse{}, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.UpdateWorkflowExecutionResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"load first execution run ID from start event\",\n\t\t\tterminationRequest: types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain:              constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution:   &types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tReason:              \"Test termination\",\n\t\t\t\t\tIdentity:            \"testRunner\",\n\t\t\t\t\tFirstExecutionRunID: \"fetched-first-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(t *testing.T, eft *testdata.EngineForTest) {\n\t\t\t\tgetExecReq := &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tExecution:  types.WorkflowExecution{WorkflowID: constants.TestWorkflowID, RunID: constants.TestRunID},\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t\tRangeID:    1,\n\t\t\t\t}\n\t\t\t\tgetExecResp := &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:            constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:          constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:               constants.TestRunID,\n\t\t\t\t\t\t\tFirstExecutionRunID: \"\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetWorkflowExecution\", mock.Anything, getExecReq).\n\t\t\t\t\tReturn(getExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, \"\").Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\t\t\t\teft.ShardCtx.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\t\t\t\tgetCurrentExecReq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tgetCurrentExecResp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"GetCurrentExecution\", mock.Anything, getCurrentExecReq).\n\t\t\t\t\tReturn(getCurrentExecResp, nil).Once()\n\n\t\t\t\thistoryBranchResp := &persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\t\t\tFirstExecutionRunID: \"fetched-first-run-id\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\thistoryMgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryMgr.\n\t\t\t\t\tOn(\"ReadHistoryBranch\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(historyBranchResp, nil).\n\t\t\t\t\tOnce()\n\n\t\t\t\tupdateExecResp := &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{},\n\t\t\t\t}\n\t\t\t\teft.ShardCtx.Resource.ExecutionMgr.\n\t\t\t\t\tOn(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(updateExecResp, nil).Once()\n\n\t\t\t\teft.ShardCtx.Resource.ShardMgr.\n\t\t\t\t\tOn(\"UpdateShard\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(nil)\n\n\t\t\t\thistoryV2Mgr := eft.ShardCtx.Resource.HistoryMgr\n\t\t\t\thistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.AnythingOfType(\"*persistence.AppendHistoryNodesRequest\")).\n\t\t\t\t\tReturn(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\teft := testdata.NewEngineForTest(t, NewEngineWithShardContext)\n\t\t\teft.Engine.Start()\n\t\t\tdefer eft.Engine.Stop()\n\n\t\t\ttc.setupMocks(t, eft)\n\n\t\t\terr := eft.Engine.TerminateWorkflowExecution(\n\t\t\t\tcontext.Background(),\n\t\t\t\t&tc.terminationRequest,\n\t\t\t)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"TerminateWorkflowExecution() error = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/engine/interface.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go\n\npackage engine\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/events\"\n)\n\ntype (\n\t// Engine represents an interface for managing workflow execution history.\n\tEngine interface {\n\t\tcommon.Daemon\n\n\t\tStartWorkflowExecution(ctx context.Context, request *types.HistoryStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\t\tGetMutableState(ctx context.Context, request *types.GetMutableStateRequest) (*types.GetMutableStateResponse, error)\n\t\tPollMutableState(ctx context.Context, request *types.PollMutableStateRequest) (*types.PollMutableStateResponse, error)\n\t\tDescribeMutableState(ctx context.Context, request *types.DescribeMutableStateRequest) (*types.DescribeMutableStateResponse, error)\n\t\tResetStickyTaskList(ctx context.Context, resetRequest *types.HistoryResetStickyTaskListRequest) (*types.HistoryResetStickyTaskListResponse, error)\n\t\tDescribeWorkflowExecution(ctx context.Context, request *types.HistoryDescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error)\n\t\tRecordDecisionTaskStarted(ctx context.Context, request *types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error)\n\t\tRecordActivityTaskStarted(ctx context.Context, request *types.RecordActivityTaskStartedRequest) (*types.RecordActivityTaskStartedResponse, error)\n\t\tRespondDecisionTaskCompleted(ctx context.Context, request *types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error)\n\t\tRespondDecisionTaskFailed(ctx context.Context, request *types.HistoryRespondDecisionTaskFailedRequest) error\n\t\tRespondActivityTaskCompleted(ctx context.Context, request *types.HistoryRespondActivityTaskCompletedRequest) error\n\t\tRespondActivityTaskFailed(ctx context.Context, request *types.HistoryRespondActivityTaskFailedRequest) error\n\t\tRespondActivityTaskCanceled(ctx context.Context, request *types.HistoryRespondActivityTaskCanceledRequest) error\n\t\tRecordActivityTaskHeartbeat(ctx context.Context, request *types.HistoryRecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error)\n\t\tRequestCancelWorkflowExecution(ctx context.Context, request *types.HistoryRequestCancelWorkflowExecutionRequest) error\n\t\tSignalWorkflowExecution(ctx context.Context, request *types.HistorySignalWorkflowExecutionRequest) error\n\t\tSignalWithStartWorkflowExecution(ctx context.Context, request *types.HistorySignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\t\tRemoveSignalMutableState(ctx context.Context, request *types.RemoveSignalMutableStateRequest) error\n\t\tTerminateWorkflowExecution(ctx context.Context, request *types.HistoryTerminateWorkflowExecutionRequest) error\n\t\tResetWorkflowExecution(ctx context.Context, request *types.HistoryResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error)\n\t\tScheduleDecisionTask(ctx context.Context, request *types.ScheduleDecisionTaskRequest) error\n\t\tRecordChildExecutionCompleted(ctx context.Context, request *types.RecordChildExecutionCompletedRequest) error\n\t\tReplicateEventsV2(ctx context.Context, request *types.ReplicateEventsV2Request) error\n\t\tSyncShardStatus(ctx context.Context, request *types.SyncShardStatusRequest) error\n\t\tSyncActivity(ctx context.Context, request *types.SyncActivityRequest) error\n\t\tGetReplicationMessages(ctx context.Context, pollingCluster string, lastReadMessageID int64) (*types.ReplicationMessages, error)\n\t\tGetDLQReplicationMessages(ctx context.Context, taskInfos []*types.ReplicationTaskInfo) ([]*types.ReplicationTask, error)\n\t\tQueryWorkflow(ctx context.Context, request *types.HistoryQueryWorkflowRequest) (*types.HistoryQueryWorkflowResponse, error)\n\t\tReapplyEvents(ctx context.Context, domainUUID string, workflowID string, runID string, events []*types.HistoryEvent) error\n\t\tCountDLQMessages(ctx context.Context, forceFetch bool) (map[string]int64, error)\n\t\tReadDLQMessages(ctx context.Context, messagesRequest *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error)\n\t\tPurgeDLQMessages(ctx context.Context, messagesRequest *types.PurgeDLQMessagesRequest) error\n\t\tMergeDLQMessages(ctx context.Context, messagesRequest *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error)\n\t\tRefreshWorkflowTasks(ctx context.Context, domainUUID string, execution types.WorkflowExecution) error\n\t\tResetTransferQueue(ctx context.Context, clusterName string) error\n\t\tResetTimerQueue(ctx context.Context, clusterName string) error\n\t\tDescribeTransferQueue(ctx context.Context, clusterName string) (*types.DescribeQueueResponse, error)\n\t\tDescribeTimerQueue(ctx context.Context, clusterName string) (*types.DescribeQueueResponse, error)\n\n\t\tNotifyNewHistoryEvent(event *events.Notification)\n\t\tNotifyNewTransferTasks(info *hcommon.NotifyTaskInfo)\n\t\tNotifyNewTimerTasks(info *hcommon.NotifyTaskInfo)\n\t\tNotifyNewReplicationTasks(info *hcommon.NotifyTaskInfo)\n\t}\n)\n"
  },
  {
    "path": "service/history/engine/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package engine -source interface.go -destination interface_mock.go\n//\n\n// Package engine is a generated GoMock package.\npackage engine\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n\tcommon \"github.com/uber/cadence/service/history/common\"\n\tevents \"github.com/uber/cadence/service/history/events\"\n)\n\n// MockEngine is a mock of Engine interface.\ntype MockEngine struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEngineMockRecorder\n\tisgomock struct{}\n}\n\n// MockEngineMockRecorder is the mock recorder for MockEngine.\ntype MockEngineMockRecorder struct {\n\tmock *MockEngine\n}\n\n// NewMockEngine creates a new mock instance.\nfunc NewMockEngine(ctrl *gomock.Controller) *MockEngine {\n\tmock := &MockEngine{ctrl: ctrl}\n\tmock.recorder = &MockEngineMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEngine) EXPECT() *MockEngineMockRecorder {\n\treturn m.recorder\n}\n\n// CountDLQMessages mocks base method.\nfunc (m *MockEngine) CountDLQMessages(ctx context.Context, forceFetch bool) (map[string]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountDLQMessages\", ctx, forceFetch)\n\tret0, _ := ret[0].(map[string]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountDLQMessages indicates an expected call of CountDLQMessages.\nfunc (mr *MockEngineMockRecorder) CountDLQMessages(ctx, forceFetch any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountDLQMessages\", reflect.TypeOf((*MockEngine)(nil).CountDLQMessages), ctx, forceFetch)\n}\n\n// DescribeMutableState mocks base method.\nfunc (m *MockEngine) DescribeMutableState(ctx context.Context, request *types.DescribeMutableStateRequest) (*types.DescribeMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeMutableState\", ctx, request)\n\tret0, _ := ret[0].(*types.DescribeMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeMutableState indicates an expected call of DescribeMutableState.\nfunc (mr *MockEngineMockRecorder) DescribeMutableState(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeMutableState\", reflect.TypeOf((*MockEngine)(nil).DescribeMutableState), ctx, request)\n}\n\n// DescribeTimerQueue mocks base method.\nfunc (m *MockEngine) DescribeTimerQueue(ctx context.Context, clusterName string) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTimerQueue\", ctx, clusterName)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTimerQueue indicates an expected call of DescribeTimerQueue.\nfunc (mr *MockEngineMockRecorder) DescribeTimerQueue(ctx, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTimerQueue\", reflect.TypeOf((*MockEngine)(nil).DescribeTimerQueue), ctx, clusterName)\n}\n\n// DescribeTransferQueue mocks base method.\nfunc (m *MockEngine) DescribeTransferQueue(ctx context.Context, clusterName string) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTransferQueue\", ctx, clusterName)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTransferQueue indicates an expected call of DescribeTransferQueue.\nfunc (mr *MockEngineMockRecorder) DescribeTransferQueue(ctx, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTransferQueue\", reflect.TypeOf((*MockEngine)(nil).DescribeTransferQueue), ctx, clusterName)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockEngine) DescribeWorkflowExecution(ctx context.Context, request *types.HistoryDescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*types.DescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) DescribeWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).DescribeWorkflowExecution), ctx, request)\n}\n\n// GetDLQReplicationMessages mocks base method.\nfunc (m *MockEngine) GetDLQReplicationMessages(ctx context.Context, taskInfos []*types.ReplicationTaskInfo) ([]*types.ReplicationTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", ctx, taskInfos)\n\tret0, _ := ret[0].([]*types.ReplicationTask)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQReplicationMessages indicates an expected call of GetDLQReplicationMessages.\nfunc (mr *MockEngineMockRecorder) GetDLQReplicationMessages(ctx, taskInfos any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQReplicationMessages\", reflect.TypeOf((*MockEngine)(nil).GetDLQReplicationMessages), ctx, taskInfos)\n}\n\n// GetMutableState mocks base method.\nfunc (m *MockEngine) GetMutableState(ctx context.Context, request *types.GetMutableStateRequest) (*types.GetMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMutableState\", ctx, request)\n\tret0, _ := ret[0].(*types.GetMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMutableState indicates an expected call of GetMutableState.\nfunc (mr *MockEngineMockRecorder) GetMutableState(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMutableState\", reflect.TypeOf((*MockEngine)(nil).GetMutableState), ctx, request)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockEngine) GetReplicationMessages(ctx context.Context, pollingCluster string, lastReadMessageID int64) (*types.ReplicationMessages, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", ctx, pollingCluster, lastReadMessageID)\n\tret0, _ := ret[0].(*types.ReplicationMessages)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockEngineMockRecorder) GetReplicationMessages(ctx, pollingCluster, lastReadMessageID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockEngine)(nil).GetReplicationMessages), ctx, pollingCluster, lastReadMessageID)\n}\n\n// MergeDLQMessages mocks base method.\nfunc (m *MockEngine) MergeDLQMessages(ctx context.Context, messagesRequest *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", ctx, messagesRequest)\n\tret0, _ := ret[0].(*types.MergeDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MergeDLQMessages indicates an expected call of MergeDLQMessages.\nfunc (mr *MockEngineMockRecorder) MergeDLQMessages(ctx, messagesRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeDLQMessages\", reflect.TypeOf((*MockEngine)(nil).MergeDLQMessages), ctx, messagesRequest)\n}\n\n// NotifyNewHistoryEvent mocks base method.\nfunc (m *MockEngine) NotifyNewHistoryEvent(event *events.Notification) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewHistoryEvent\", event)\n}\n\n// NotifyNewHistoryEvent indicates an expected call of NotifyNewHistoryEvent.\nfunc (mr *MockEngineMockRecorder) NotifyNewHistoryEvent(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewHistoryEvent\", reflect.TypeOf((*MockEngine)(nil).NotifyNewHistoryEvent), event)\n}\n\n// NotifyNewReplicationTasks mocks base method.\nfunc (m *MockEngine) NotifyNewReplicationTasks(info *common.NotifyTaskInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewReplicationTasks\", info)\n}\n\n// NotifyNewReplicationTasks indicates an expected call of NotifyNewReplicationTasks.\nfunc (mr *MockEngineMockRecorder) NotifyNewReplicationTasks(info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewReplicationTasks\", reflect.TypeOf((*MockEngine)(nil).NotifyNewReplicationTasks), info)\n}\n\n// NotifyNewTimerTasks mocks base method.\nfunc (m *MockEngine) NotifyNewTimerTasks(info *common.NotifyTaskInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewTimerTasks\", info)\n}\n\n// NotifyNewTimerTasks indicates an expected call of NotifyNewTimerTasks.\nfunc (mr *MockEngineMockRecorder) NotifyNewTimerTasks(info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewTimerTasks\", reflect.TypeOf((*MockEngine)(nil).NotifyNewTimerTasks), info)\n}\n\n// NotifyNewTransferTasks mocks base method.\nfunc (m *MockEngine) NotifyNewTransferTasks(info *common.NotifyTaskInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewTransferTasks\", info)\n}\n\n// NotifyNewTransferTasks indicates an expected call of NotifyNewTransferTasks.\nfunc (mr *MockEngineMockRecorder) NotifyNewTransferTasks(info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewTransferTasks\", reflect.TypeOf((*MockEngine)(nil).NotifyNewTransferTasks), info)\n}\n\n// PollMutableState mocks base method.\nfunc (m *MockEngine) PollMutableState(ctx context.Context, request *types.PollMutableStateRequest) (*types.PollMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollMutableState\", ctx, request)\n\tret0, _ := ret[0].(*types.PollMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollMutableState indicates an expected call of PollMutableState.\nfunc (mr *MockEngineMockRecorder) PollMutableState(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollMutableState\", reflect.TypeOf((*MockEngine)(nil).PollMutableState), ctx, request)\n}\n\n// PurgeDLQMessages mocks base method.\nfunc (m *MockEngine) PurgeDLQMessages(ctx context.Context, messagesRequest *types.PurgeDLQMessagesRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", ctx, messagesRequest)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PurgeDLQMessages indicates an expected call of PurgeDLQMessages.\nfunc (mr *MockEngineMockRecorder) PurgeDLQMessages(ctx, messagesRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PurgeDLQMessages\", reflect.TypeOf((*MockEngine)(nil).PurgeDLQMessages), ctx, messagesRequest)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockEngine) QueryWorkflow(ctx context.Context, request *types.HistoryQueryWorkflowRequest) (*types.HistoryQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", ctx, request)\n\tret0, _ := ret[0].(*types.HistoryQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockEngineMockRecorder) QueryWorkflow(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockEngine)(nil).QueryWorkflow), ctx, request)\n}\n\n// ReadDLQMessages mocks base method.\nfunc (m *MockEngine) ReadDLQMessages(ctx context.Context, messagesRequest *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", ctx, messagesRequest)\n\tret0, _ := ret[0].(*types.ReadDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadDLQMessages indicates an expected call of ReadDLQMessages.\nfunc (mr *MockEngineMockRecorder) ReadDLQMessages(ctx, messagesRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadDLQMessages\", reflect.TypeOf((*MockEngine)(nil).ReadDLQMessages), ctx, messagesRequest)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockEngine) ReapplyEvents(ctx context.Context, domainUUID, workflowID, runID string, events []*types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", ctx, domainUUID, workflowID, runID, events)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockEngineMockRecorder) ReapplyEvents(ctx, domainUUID, workflowID, runID, events any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockEngine)(nil).ReapplyEvents), ctx, domainUUID, workflowID, runID, events)\n}\n\n// RecordActivityTaskHeartbeat mocks base method.\nfunc (m *MockEngine) RecordActivityTaskHeartbeat(ctx context.Context, request *types.HistoryRecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", ctx, request)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeat indicates an expected call of RecordActivityTaskHeartbeat.\nfunc (mr *MockEngineMockRecorder) RecordActivityTaskHeartbeat(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeat\", reflect.TypeOf((*MockEngine)(nil).RecordActivityTaskHeartbeat), ctx, request)\n}\n\n// RecordActivityTaskStarted mocks base method.\nfunc (m *MockEngine) RecordActivityTaskStarted(ctx context.Context, request *types.RecordActivityTaskStartedRequest) (*types.RecordActivityTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskStarted\", ctx, request)\n\tret0, _ := ret[0].(*types.RecordActivityTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskStarted indicates an expected call of RecordActivityTaskStarted.\nfunc (mr *MockEngineMockRecorder) RecordActivityTaskStarted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskStarted\", reflect.TypeOf((*MockEngine)(nil).RecordActivityTaskStarted), ctx, request)\n}\n\n// RecordChildExecutionCompleted mocks base method.\nfunc (m *MockEngine) RecordChildExecutionCompleted(ctx context.Context, request *types.RecordChildExecutionCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordChildExecutionCompleted\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordChildExecutionCompleted indicates an expected call of RecordChildExecutionCompleted.\nfunc (mr *MockEngineMockRecorder) RecordChildExecutionCompleted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordChildExecutionCompleted\", reflect.TypeOf((*MockEngine)(nil).RecordChildExecutionCompleted), ctx, request)\n}\n\n// RecordDecisionTaskStarted mocks base method.\nfunc (m *MockEngine) RecordDecisionTaskStarted(ctx context.Context, request *types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordDecisionTaskStarted\", ctx, request)\n\tret0, _ := ret[0].(*types.RecordDecisionTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordDecisionTaskStarted indicates an expected call of RecordDecisionTaskStarted.\nfunc (mr *MockEngineMockRecorder) RecordDecisionTaskStarted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordDecisionTaskStarted\", reflect.TypeOf((*MockEngine)(nil).RecordDecisionTaskStarted), ctx, request)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockEngine) RefreshWorkflowTasks(ctx context.Context, domainUUID string, execution types.WorkflowExecution) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", ctx, domainUUID, execution)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockEngineMockRecorder) RefreshWorkflowTasks(ctx, domainUUID, execution any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockEngine)(nil).RefreshWorkflowTasks), ctx, domainUUID, execution)\n}\n\n// RemoveSignalMutableState mocks base method.\nfunc (m *MockEngine) RemoveSignalMutableState(ctx context.Context, request *types.RemoveSignalMutableStateRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RemoveSignalMutableState\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveSignalMutableState indicates an expected call of RemoveSignalMutableState.\nfunc (mr *MockEngineMockRecorder) RemoveSignalMutableState(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveSignalMutableState\", reflect.TypeOf((*MockEngine)(nil).RemoveSignalMutableState), ctx, request)\n}\n\n// ReplicateEventsV2 mocks base method.\nfunc (m *MockEngine) ReplicateEventsV2(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateEventsV2\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateEventsV2 indicates an expected call of ReplicateEventsV2.\nfunc (mr *MockEngineMockRecorder) ReplicateEventsV2(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateEventsV2\", reflect.TypeOf((*MockEngine)(nil).ReplicateEventsV2), ctx, request)\n}\n\n// RequestCancelWorkflowExecution mocks base method.\nfunc (m *MockEngine) RequestCancelWorkflowExecution(ctx context.Context, request *types.HistoryRequestCancelWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestCancelWorkflowExecution indicates an expected call of RequestCancelWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) RequestCancelWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).RequestCancelWorkflowExecution), ctx, request)\n}\n\n// ResetStickyTaskList mocks base method.\nfunc (m *MockEngine) ResetStickyTaskList(ctx context.Context, resetRequest *types.HistoryResetStickyTaskListRequest) (*types.HistoryResetStickyTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", ctx, resetRequest)\n\tret0, _ := ret[0].(*types.HistoryResetStickyTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetStickyTaskList indicates an expected call of ResetStickyTaskList.\nfunc (mr *MockEngineMockRecorder) ResetStickyTaskList(ctx, resetRequest any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetStickyTaskList\", reflect.TypeOf((*MockEngine)(nil).ResetStickyTaskList), ctx, resetRequest)\n}\n\n// ResetTimerQueue mocks base method.\nfunc (m *MockEngine) ResetTimerQueue(ctx context.Context, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetTimerQueue\", ctx, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetTimerQueue indicates an expected call of ResetTimerQueue.\nfunc (mr *MockEngineMockRecorder) ResetTimerQueue(ctx, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetTimerQueue\", reflect.TypeOf((*MockEngine)(nil).ResetTimerQueue), ctx, clusterName)\n}\n\n// ResetTransferQueue mocks base method.\nfunc (m *MockEngine) ResetTransferQueue(ctx context.Context, clusterName string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetTransferQueue\", ctx, clusterName)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetTransferQueue indicates an expected call of ResetTransferQueue.\nfunc (mr *MockEngineMockRecorder) ResetTransferQueue(ctx, clusterName any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetTransferQueue\", reflect.TypeOf((*MockEngine)(nil).ResetTransferQueue), ctx, clusterName)\n}\n\n// ResetWorkflowExecution mocks base method.\nfunc (m *MockEngine) ResetWorkflowExecution(ctx context.Context, request *types.HistoryResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*types.ResetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) ResetWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).ResetWorkflowExecution), ctx, request)\n}\n\n// RespondActivityTaskCanceled mocks base method.\nfunc (m *MockEngine) RespondActivityTaskCanceled(ctx context.Context, request *types.HistoryRespondActivityTaskCanceledRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceled indicates an expected call of RespondActivityTaskCanceled.\nfunc (mr *MockEngineMockRecorder) RespondActivityTaskCanceled(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceled\", reflect.TypeOf((*MockEngine)(nil).RespondActivityTaskCanceled), ctx, request)\n}\n\n// RespondActivityTaskCompleted mocks base method.\nfunc (m *MockEngine) RespondActivityTaskCompleted(ctx context.Context, request *types.HistoryRespondActivityTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompleted indicates an expected call of RespondActivityTaskCompleted.\nfunc (mr *MockEngineMockRecorder) RespondActivityTaskCompleted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompleted\", reflect.TypeOf((*MockEngine)(nil).RespondActivityTaskCompleted), ctx, request)\n}\n\n// RespondActivityTaskFailed mocks base method.\nfunc (m *MockEngine) RespondActivityTaskFailed(ctx context.Context, request *types.HistoryRespondActivityTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailed indicates an expected call of RespondActivityTaskFailed.\nfunc (mr *MockEngineMockRecorder) RespondActivityTaskFailed(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailed\", reflect.TypeOf((*MockEngine)(nil).RespondActivityTaskFailed), ctx, request)\n}\n\n// RespondDecisionTaskCompleted mocks base method.\nfunc (m *MockEngine) RespondDecisionTaskCompleted(ctx context.Context, request *types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", ctx, request)\n\tret0, _ := ret[0].(*types.HistoryRespondDecisionTaskCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondDecisionTaskCompleted indicates an expected call of RespondDecisionTaskCompleted.\nfunc (mr *MockEngineMockRecorder) RespondDecisionTaskCompleted(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskCompleted\", reflect.TypeOf((*MockEngine)(nil).RespondDecisionTaskCompleted), ctx, request)\n}\n\n// RespondDecisionTaskFailed mocks base method.\nfunc (m *MockEngine) RespondDecisionTaskFailed(ctx context.Context, request *types.HistoryRespondDecisionTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondDecisionTaskFailed indicates an expected call of RespondDecisionTaskFailed.\nfunc (mr *MockEngineMockRecorder) RespondDecisionTaskFailed(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskFailed\", reflect.TypeOf((*MockEngine)(nil).RespondDecisionTaskFailed), ctx, request)\n}\n\n// ScheduleDecisionTask mocks base method.\nfunc (m *MockEngine) ScheduleDecisionTask(ctx context.Context, request *types.ScheduleDecisionTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ScheduleDecisionTask\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ScheduleDecisionTask indicates an expected call of ScheduleDecisionTask.\nfunc (mr *MockEngineMockRecorder) ScheduleDecisionTask(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScheduleDecisionTask\", reflect.TypeOf((*MockEngine)(nil).ScheduleDecisionTask), ctx, request)\n}\n\n// SignalWithStartWorkflowExecution mocks base method.\nfunc (m *MockEngine) SignalWithStartWorkflowExecution(ctx context.Context, request *types.HistorySignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecution indicates an expected call of SignalWithStartWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) SignalWithStartWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).SignalWithStartWorkflowExecution), ctx, request)\n}\n\n// SignalWorkflowExecution mocks base method.\nfunc (m *MockEngine) SignalWorkflowExecution(ctx context.Context, request *types.HistorySignalWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SignalWorkflowExecution indicates an expected call of SignalWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) SignalWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).SignalWorkflowExecution), ctx, request)\n}\n\n// Start mocks base method.\nfunc (m *MockEngine) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockEngineMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockEngine)(nil).Start))\n}\n\n// StartWorkflowExecution mocks base method.\nfunc (m *MockEngine) StartWorkflowExecution(ctx context.Context, request *types.HistoryStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecution indicates an expected call of StartWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) StartWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).StartWorkflowExecution), ctx, request)\n}\n\n// Stop mocks base method.\nfunc (m *MockEngine) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockEngineMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockEngine)(nil).Stop))\n}\n\n// SyncActivity mocks base method.\nfunc (m *MockEngine) SyncActivity(ctx context.Context, request *types.SyncActivityRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SyncActivity\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncActivity indicates an expected call of SyncActivity.\nfunc (mr *MockEngineMockRecorder) SyncActivity(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncActivity\", reflect.TypeOf((*MockEngine)(nil).SyncActivity), ctx, request)\n}\n\n// SyncShardStatus mocks base method.\nfunc (m *MockEngine) SyncShardStatus(ctx context.Context, request *types.SyncShardStatusRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SyncShardStatus\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncShardStatus indicates an expected call of SyncShardStatus.\nfunc (mr *MockEngineMockRecorder) SyncShardStatus(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncShardStatus\", reflect.TypeOf((*MockEngine)(nil).SyncShardStatus), ctx, request)\n}\n\n// TerminateWorkflowExecution mocks base method.\nfunc (m *MockEngine) TerminateWorkflowExecution(ctx context.Context, request *types.HistoryTerminateWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TerminateWorkflowExecution indicates an expected call of TerminateWorkflowExecution.\nfunc (mr *MockEngineMockRecorder) TerminateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TerminateWorkflowExecution\", reflect.TypeOf((*MockEngine)(nil).TerminateWorkflowExecution), ctx, request)\n}\n"
  },
  {
    "path": "service/history/engine/testdata/engine_for_tests.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testdata\n\nimport (\n\t\"testing\"\n\n\t// client library cannot change from old gomock\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/failover\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/replication\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype EngineForTest struct {\n\tEngine engine.Engine\n\t// Add mocks or other fields here\n\tShardCtx *shard.TestContext\n}\n\n// NewEngineFn is defined as an alias for engineimpl.NewEngineWithShardContext to avoid circular dependency\ntype NewEngineFn func(\n\tshard shard.Context,\n\tvisibilityMgr persistence.VisibilityManager,\n\tmatching matching.Client,\n\thistoryEventNotifier events.Notifier,\n\tconfig *config.Config,\n\treplicationTaskFetchers replication.TaskFetchers,\n\trawMatchingClient matching.Client,\n\tfailoverCoordinator failover.Coordinator,\n\tqueueFactories []queue.Factory,\n) engine.Engine\n\nfunc NewEngineForTest(t *testing.T, newEngineFn NewEngineFn) *EngineForTest {\n\tt.Helper()\n\tcontroller := gomock.NewController(t)\n\thistoryCfg := config.NewForTest()\n\tshardCtx := shard.NewTestContext(\n\t\tt,\n\t\tcontroller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\thistoryCfg,\n\t)\n\n\tdomainCache := shardCtx.Resource.DomainCache\n\tdomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestLocalDomainEntry, nil).AnyTimes()\n\tdomainCache.EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\tdomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestLocalDomainEntry, nil).AnyTimes()\n\tdomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\tdomainCache.EXPECT().RegisterDomainChangeCallback(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\tdomainCache.EXPECT().UnregisterDomainChangeCallback(gomock.Any()).Times(1)\n\n\texecutionMgr := shardCtx.Resource.ExecutionMgr\n\t// RangeCompleteReplicationTask is called by taskProcessorImpl's background loop\n\texecutionMgr.\n\t\tOn(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, nil)\n\n\tmembershipResolver := shardCtx.Resource.MembershipResolver\n\tmembershipResolver.EXPECT().MemberCount(gomock.Any()).Return(1, nil).AnyTimes()\n\n\teventsCache := events.NewCache(\n\t\tshardCtx.GetShardID(),\n\t\tshardCtx.GetHistoryManager(),\n\t\thistoryCfg,\n\t\tshardCtx.GetLogger(),\n\t\tshardCtx.GetMetricsClient(),\n\t\tdomainCache,\n\t)\n\tshardCtx.SetEventsCache(eventsCache)\n\n\thistoryEventNotifier := events.NewNotifier(\n\t\tclock.NewRealTimeSource(),\n\t\tshardCtx.Resource.MetricsClient,\n\t\tfunc(workflowID string) int {\n\t\t\treturn len(workflowID)\n\t\t},\n\t)\n\n\treplicatonTaskFetchers := replication.NewMockTaskFetchers(controller)\n\treplicationTaskFetcher := replication.NewMockTaskFetcher(controller)\n\t// TODO: this should probably return another cluster name, not current\n\treplicationTaskFetcher.EXPECT().GetSourceCluster().Return(constants.TestClusterMetadata.GetCurrentClusterName()).AnyTimes()\n\treplicationTaskFetcher.EXPECT().GetRateLimiter().Return(quotas.NewDynamicRateLimiter(func() float64 { return 100 })).AnyTimes()\n\treplicationTaskFetcher.EXPECT().GetRequestChan(gomock.Any()).Return(nil).AnyTimes()\n\treplicatonTaskFetchers.EXPECT().GetFetchers().Return([]replication.TaskFetcher{replicationTaskFetcher}).AnyTimes()\n\n\tfailoverCoordinator := failover.NewMockCoordinator(controller)\n\n\ttimerQueueFactory := queue.NewMockFactory(controller)\n\ttimerQueueFactory.EXPECT().Category().Return(persistence.HistoryTaskCategoryTimer).AnyTimes()\n\ttimerQProcessor := queue.NewMockProcessor(controller)\n\ttimerQProcessor.EXPECT().Start().Return().Times(1)\n\ttimerQProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).Return().AnyTimes()\n\ttimerQProcessor.EXPECT().Stop().Return().Times(1)\n\ttimerQueueFactory.EXPECT().CreateQueue(gomock.Any(), gomock.Any(), gomock.Any()).Return(timerQProcessor).Times(1)\n\n\ttransferQueueFactory := queue.NewMockFactory(controller)\n\ttransferQueueFactory.EXPECT().Category().Return(persistence.HistoryTaskCategoryTransfer).AnyTimes()\n\ttransferQProcessor := queue.NewMockProcessor(controller)\n\ttransferQProcessor.EXPECT().Start().Return().Times(1)\n\ttransferQProcessor.EXPECT().NotifyNewTask(gomock.Any(), gomock.Any()).Return().AnyTimes()\n\ttransferQProcessor.EXPECT().Stop().Return().Times(1)\n\ttransferQueueFactory.EXPECT().CreateQueue(gomock.Any(), gomock.Any(), gomock.Any()).Return(transferQProcessor).Times(1)\n\n\tqueueFactories := []queue.Factory{timerQueueFactory, transferQueueFactory}\n\n\tengine := newEngineFn(\n\t\tshardCtx,\n\t\tshardCtx.Resource.VisibilityMgr,\n\t\tshardCtx.Resource.MatchingClient,\n\t\thistoryEventNotifier,\n\t\thistoryCfg,\n\t\treplicatonTaskFetchers,\n\t\tshardCtx.Resource.MatchingClient,\n\t\tfailoverCoordinator,\n\t\tqueueFactories,\n\t)\n\n\tshardCtx.SetEngine(engine)\n\n\thistoryEventNotifier.Start()\n\tt.Cleanup(historyEventNotifier.Stop)\n\n\treturn &EngineForTest{\n\t\tEngine:   engine,\n\t\tShardCtx: shardCtx,\n\t}\n}\n"
  },
  {
    "path": "service/history/events/blob.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2022 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage events\n\nimport (\n\t\"bytes\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// PersistedBlob is a wrapper on persistence.DataBlob with additional field indicating what was persisted.\n\t// Additional fields are used as an identification key among other blobs.\n\tPersistedBlob struct {\n\t\tpersistence.DataBlob\n\n\t\tBranchToken  []byte\n\t\tFirstEventID int64\n\t}\n\t// PersistedBlobs is a slice of PersistedBlob\n\tPersistedBlobs []PersistedBlob\n)\n\n// Find searches for persisted event blob. Returns nil when not found.\nfunc (blobs PersistedBlobs) Find(branchToken []byte, firstEventID int64) *persistence.DataBlob {\n\t// Linear search is ok here, as we will only have 1-2 persisted blobs per transaction\n\tfor _, blob := range blobs {\n\t\tif bytes.Equal(blob.BranchToken, branchToken) && blob.FirstEventID == firstEventID {\n\t\t\treturn &blob.DataBlob\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/events/blob_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2022 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage events\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestPersistedBlobs_Find(t *testing.T) {\n\tblob1 := persistence.DataBlob{Data: []byte{1, 2, 3}}\n\tblob2 := persistence.DataBlob{Data: []byte{4, 5, 6}}\n\tblob3 := persistence.DataBlob{Data: []byte{7, 8, 9}}\n\tbranchA := []byte{11, 11, 11}\n\tbranchB := []byte{22, 22, 22}\n\tpersistedBlobs := PersistedBlobs{\n\t\tPersistedBlob{BranchToken: branchA, FirstEventID: 100, DataBlob: blob1},\n\t\tPersistedBlob{BranchToken: branchA, FirstEventID: 105, DataBlob: blob2},\n\t\tPersistedBlob{BranchToken: branchB, FirstEventID: 100, DataBlob: blob3},\n\t}\n\tassert.Equal(t, blob1, *persistedBlobs.Find(branchA, 100))\n\tassert.Equal(t, blob2, *persistedBlobs.Find(branchA, 105))\n\tassert.Equal(t, blob3, *persistedBlobs.Find(branchB, 100))\n\tassert.Nil(t, persistedBlobs.Find(branchB, 105))\n\tassert.Nil(t, persistedBlobs.Find([]byte{99}, 100))\n}\n"
  },
  {
    "path": "service/history/events/cache.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination cache_mock.go\n\npackage events\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype (\n\t// Cache caches workflow history event\n\tCache interface {\n\t\tGetEvent(\n\t\t\tctx context.Context,\n\t\t\tshardID int,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\trunID string,\n\t\t\tfirstEventID int64,\n\t\t\teventID int64,\n\t\t\tbranchToken []byte,\n\t\t) (*types.HistoryEvent, error)\n\t\tPutEvent(\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\trunID string,\n\t\t\teventID int64,\n\t\t\tevent *types.HistoryEvent,\n\t\t)\n\t}\n\n\tcacheImpl struct {\n\t\tcache.Cache\n\t\tdomainCache    cache.DomainCache\n\t\thistoryManager persistence.HistoryManager\n\t\tdisabled       bool\n\t\tlogger         log.Logger\n\t\tmetricsClient  metrics.Client\n\t\tshardID        *int\n\t}\n\n\teventKey struct {\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\t\teventID    int64\n\t}\n)\n\nvar (\n\terrEventNotFoundInBatch = &types.EntityNotExistsError{Message: \"History event not found within expected batch\"}\n)\n\nvar _ Cache = (*cacheImpl)(nil)\n\n// NewGlobalCache creates a new global events cache\nfunc NewGlobalCache(\n\tinitialCount int,\n\tmaxCount int,\n\tttl time.Duration,\n\thistoryManager persistence.HistoryManager,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tenableSizeBasedCache dynamicproperties.BoolPropertyFn,\n\tmaxSize dynamicproperties.IntPropertyFn,\n\tdomainCache cache.DomainCache,\n) Cache {\n\treturn newCacheWithOption(\n\t\tnil,\n\t\tinitialCount,\n\t\tmaxCount,\n\t\tttl,\n\t\thistoryManager,\n\t\tfalse,\n\t\tlogger,\n\t\tmetricsClient,\n\t\tenableSizeBasedCache,\n\t\tmaxSize,\n\t\tdomainCache,\n\t)\n}\n\n// NewCache creates a new events cache\nfunc NewCache(\n\tshardID int,\n\thistoryManager persistence.HistoryManager,\n\tconfig *config.Config,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdomainCache cache.DomainCache,\n) Cache {\n\treturn newCacheWithOption(\n\t\t&shardID,\n\t\tconfig.EventsCacheInitialCount(),\n\t\tconfig.EventsCacheMaxCount(),\n\t\tconfig.EventsCacheTTL(),\n\t\thistoryManager,\n\t\tfalse,\n\t\tlogger,\n\t\tmetricsClient,\n\t\tconfig.EnableSizeBasedHistoryEventCache,\n\t\tconfig.EventsCacheMaxSize,\n\t\tdomainCache,\n\t)\n}\n\nfunc newCacheWithOption(\n\tshardID *int,\n\tinitialCount int,\n\tmaxCount int,\n\tttl time.Duration,\n\thistoryManager persistence.HistoryManager,\n\tdisabled bool,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tenableSizeBasedCache dynamicproperties.BoolPropertyFn,\n\tmaxSize dynamicproperties.IntPropertyFn,\n\tdomainCache cache.DomainCache,\n) *cacheImpl {\n\topts := &cache.Options{}\n\topts.InitialCapacity = initialCount\n\topts.TTL = ttl\n\topts.MaxCount = maxCount\n\topts.MetricsScope = metricsClient.Scope(metrics.EventsCacheGetEventScope)\n\tif shardID != nil {\n\t\topts.MetricsScope = opts.MetricsScope.Tagged(metrics.ShardIDTag(*shardID))\n\t}\n\n\topts.MaxSize = maxSize\n\topts.Logger = logger.WithTags(tag.ComponentEventsCache)\n\topts.IsSizeBased = enableSizeBasedCache\n\n\treturn &cacheImpl{\n\t\tCache:          cache.New(opts),\n\t\tdomainCache:    domainCache,\n\t\thistoryManager: historyManager,\n\t\tdisabled:       disabled,\n\t\tlogger:         logger.WithTags(tag.ComponentEventsCache),\n\t\tmetricsClient:  metricsClient,\n\t\tshardID:        shardID,\n\t}\n}\n\nfunc newEventKey(\n\tdomainID,\n\tworkflowID,\n\trunID string,\n\teventID int64,\n) eventKey {\n\treturn eventKey{\n\t\tdomainID:   domainID,\n\t\tworkflowID: workflowID,\n\t\trunID:      runID,\n\t\teventID:    eventID,\n\t}\n}\n\nfunc (e *cacheImpl) GetEvent(\n\tctx context.Context,\n\tshardID int,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tfirstEventID int64,\n\teventID int64,\n\tbranchToken []byte,\n) (*types.HistoryEvent, error) {\n\te.metricsClient.IncCounter(metrics.EventsCacheGetEventScope, metrics.CacheRequests)\n\tsw := e.metricsClient.StartTimer(metrics.EventsCacheGetEventScope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tkey := newEventKey(domainID, workflowID, runID, eventID)\n\t// Test hook for disabling cache\n\tif !e.disabled {\n\t\tevent, cacheHit := e.Cache.Get(key).(*types.HistoryEvent)\n\t\tif cacheHit {\n\t\t\treturn event, nil\n\t\t}\n\t}\n\n\te.metricsClient.IncCounter(metrics.EventsCacheGetEventScope, metrics.CacheMissCounter)\n\tevent, err := e.getHistoryEventFromStore(ctx, firstEventID, eventID, branchToken, shardID, domainID)\n\n\tif err != nil {\n\t\te.metricsClient.IncCounter(metrics.EventsCacheGetEventScope, metrics.CacheFailures)\n\t\te.logger.Error(\"EventsCache unable to retrieve event from store\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowEventID(eventID))\n\t\treturn nil, err\n\t}\n\n\te.Put(key, event)\n\treturn event, nil\n}\n\nfunc (e *cacheImpl) PutEvent(\n\tdomainID, workflowID,\n\trunID string,\n\teventID int64,\n\tevent *types.HistoryEvent,\n) {\n\te.metricsClient.IncCounter(metrics.EventsCachePutEventScope, metrics.CacheRequests)\n\tsw := e.metricsClient.StartTimer(metrics.EventsCachePutEventScope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tkey := newEventKey(domainID, workflowID, runID, eventID)\n\te.Put(key, event)\n}\n\nfunc (e *cacheImpl) getHistoryEventFromStore(\n\tctx context.Context,\n\tfirstEventID int64,\n\teventID int64,\n\tbranchToken []byte,\n\tshardID int,\n\tdomainID string,\n) (*types.HistoryEvent, error) {\n\te.metricsClient.IncCounter(metrics.EventsCacheGetFromStoreScope, metrics.CacheRequests)\n\tsw := e.metricsClient.StartTimer(metrics.EventsCacheGetFromStoreScope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tvar historyEvents []*types.HistoryEvent\n\tdomainName, err := e.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresponse, err := e.historyManager.ReadHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    eventID + 1,\n\t\tPageSize:      1,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(shardID),\n\t\tDomainName:    domainName,\n\t})\n\n\tif err != nil {\n\t\te.metricsClient.IncCounter(metrics.EventsCacheGetFromStoreScope, metrics.CacheFailures)\n\t\treturn nil, err\n\t}\n\n\thistoryEvents = response.HistoryEvents\n\n\t// find history event from batch and return back single event to caller\n\tfor _, e := range historyEvents {\n\t\tif e.ID == eventID {\n\t\t\treturn e, nil\n\t\t}\n\t}\n\n\treturn nil, errEventNotFoundInBatch\n}\n"
  },
  {
    "path": "service/history/events/cache_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: cache.go\n//\n// Generated by this command:\n//\n//\tmockgen -package events -source cache.go -destination cache_mock.go\n//\n\n// Package events is a generated GoMock package.\npackage events\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockCache is a mock of Cache interface.\ntype MockCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockCacheMockRecorder is the mock recorder for MockCache.\ntype MockCacheMockRecorder struct {\n\tmock *MockCache\n}\n\n// NewMockCache creates a new mock instance.\nfunc NewMockCache(ctrl *gomock.Controller) *MockCache {\n\tmock := &MockCache{ctrl: ctrl}\n\tmock.recorder = &MockCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCache) EXPECT() *MockCacheMockRecorder {\n\treturn m.recorder\n}\n\n// GetEvent mocks base method.\nfunc (m *MockCache) GetEvent(ctx context.Context, shardID int, domainID, workflowID, runID string, firstEventID, eventID int64, branchToken []byte) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEvent\", ctx, shardID, domainID, workflowID, runID, firstEventID, eventID, branchToken)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetEvent indicates an expected call of GetEvent.\nfunc (mr *MockCacheMockRecorder) GetEvent(ctx, shardID, domainID, workflowID, runID, firstEventID, eventID, branchToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEvent\", reflect.TypeOf((*MockCache)(nil).GetEvent), ctx, shardID, domainID, workflowID, runID, firstEventID, eventID, branchToken)\n}\n\n// PutEvent mocks base method.\nfunc (m *MockCache) PutEvent(domainID, workflowID, runID string, eventID int64, event *types.HistoryEvent) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"PutEvent\", domainID, workflowID, runID, eventID, event)\n}\n\n// PutEvent indicates an expected call of PutEvent.\nfunc (mr *MockCacheMockRecorder) PutEvent(domainID, workflowID, runID, eventID, event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PutEvent\", reflect.TypeOf((*MockCache)(nil).PutEvent), domainID, workflowID, runID, eventID, event)\n}\n"
  },
  {
    "path": "service/history/events/cache_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage events\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\teventsCacheSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tlogger             log.Logger\n\t\tmockHistoryManager *mocks.HistoryV2Manager\n\n\t\tcache       *cacheImpl\n\t\tctrl        *gomock.Controller\n\t\tdomainCache *cache.MockDomainCache\n\t}\n)\n\nfunc TestEventsCacheSuite(t *testing.T) {\n\ts := new(eventsCacheSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *eventsCacheSuite) SetupSuite() {\n\n}\n\nfunc (s *eventsCacheSuite) TearDownSuite() {\n\n}\n\nfunc (s *eventsCacheSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.logger = testlogger.New(s.Suite.T())\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\ts.mockHistoryManager = &mocks.HistoryV2Manager{}\n\ts.ctrl = gomock.NewController(s.T())\n\ts.domainCache = cache.NewMockDomainCache(s.ctrl)\n\ts.cache = s.newTestEventsCache()\n}\n\nfunc (s *eventsCacheSuite) TearDownTest() {\n\ts.mockHistoryManager.AssertExpectations(s.T())\n\ts.ctrl.Finish()\n}\n\nfunc (s *eventsCacheSuite) newTestEventsCache() *cacheImpl {\n\treturn newCacheWithOption(common.IntPtr(10), 16, 32, time.Minute, s.mockHistoryManager, false, s.logger,\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}), dynamicproperties.GetBoolPropertyFn(false), dynamicproperties.GetIntPropertyFn(1000), s.domainCache)\n}\n\nfunc (s *eventsCacheSuite) TestEventsCacheHitSuccess() {\n\tdomainID := \"events-cache-hit-success-domain\"\n\tworkflowID := \"events-cache-hit-success-workflow-id\"\n\trunID := \"events-cache-hit-success-run-id\"\n\teventID := int64(23)\n\tevent := &types.HistoryEvent{\n\t\tID:                                 eventID,\n\t\tEventType:                          types.EventTypeActivityTaskStarted.Ptr(),\n\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{},\n\t}\n\n\ts.cache.PutEvent(domainID, workflowID, runID, eventID, event)\n\tactualEvent, err := s.cache.GetEvent(context.Background(), 0, domainID, workflowID, runID, eventID, eventID, nil)\n\ts.Nil(err)\n\ts.Equal(event, actualEvent)\n}\n\nfunc (s *eventsCacheSuite) TestEventsCacheMissMultiEventsBatchV2Success() {\n\tdomainID := \"events-cache-miss-multi-events-batch-v2-success-domain\"\n\tworkflowID := \"events-cache-miss-multi-events-batch-v2-success-workflow-id\"\n\trunID := \"events-cache-miss-multi-events-batch-v2-success-run-id\"\n\tdomainName := \"events-cache-miss-multi-events-batch-v2-success-domainName\"\n\tevent1 := &types.HistoryEvent{\n\t\tID:                                   11,\n\t\tEventType:                            types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                   12,\n\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\tevent3 := &types.HistoryEvent{\n\t\tID:                                   13,\n\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\tevent4 := &types.HistoryEvent{\n\t\tID:                                   14,\n\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\tevent5 := &types.HistoryEvent{\n\t\tID:                                   15,\n\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\tevent6 := &types.HistoryEvent{\n\t\tID:                                   16,\n\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\n\tshardID := common.IntPtr(10)\n\ts.mockHistoryManager.On(\"ReadHistoryBranch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   []byte(\"store_token\"),\n\t\tMinEventID:    event1.ID,\n\t\tMaxEventID:    event6.ID + 1,\n\t\tPageSize:      1,\n\t\tNextPageToken: nil,\n\t\tShardID:       shardID,\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents:    []*types.HistoryEvent{event1, event2, event3, event4, event5, event6},\n\t\tNextPageToken:    nil,\n\t\tLastFirstEventID: event1.ID,\n\t}, nil)\n\n\ts.domainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.cache.PutEvent(domainID, workflowID, runID, event2.ID, event2)\n\tactualEvent, err := s.cache.GetEvent(context.Background(), *shardID, domainID, workflowID, runID, event1.ID, event6.ID, []byte(\"store_token\"))\n\ts.Nil(err)\n\ts.Equal(event6, actualEvent)\n}\n\nfunc (s *eventsCacheSuite) TestEventsCacheMissV2Failure() {\n\tdomainID := \"events-cache-miss-failure-domain\"\n\tworkflowID := \"events-cache-miss-failure-workflow-id\"\n\trunID := \"events-cache-miss-failure-run-id\"\n\tshardID := common.IntPtr(10)\n\tdomainName := \"events-cache-miss-failure-domainName\"\n\texpectedErr := errors.New(\"persistence call failed\")\n\ts.mockHistoryManager.On(\"ReadHistoryBranch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   []byte(\"store_token\"),\n\t\tMinEventID:    int64(11),\n\t\tMaxEventID:    int64(15),\n\t\tPageSize:      1,\n\t\tNextPageToken: nil,\n\t\tShardID:       shardID,\n\t\tDomainName:    domainName,\n\t}).Return(nil, expectedErr)\n\n\ts.domainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\tactualEvent, err := s.cache.GetEvent(context.Background(), *shardID, domainID, workflowID, runID, int64(11), int64(14), []byte(\"store_token\"))\n\ts.Nil(actualEvent)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *eventsCacheSuite) TestEventsCacheDisableSuccess() {\n\tdomainID := \"events-cache-disable-success-domain\"\n\tworkflowID := \"events-cache-disable-success-workflow-id\"\n\trunID := \"events-cache-disable-success-run-id\"\n\tshardID := common.IntPtr(10)\n\tdomainName := \"events-cache-disable-success-domainName\"\n\tevent1 := &types.HistoryEvent{\n\t\tID:                                 23,\n\t\tEventType:                          types.EventTypeActivityTaskStarted.Ptr(),\n\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                 32,\n\t\tEventType:                          types.EventTypeActivityTaskStarted.Ptr(),\n\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{},\n\t}\n\n\ts.mockHistoryManager.On(\"ReadHistoryBranch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   []byte(\"store_token\"),\n\t\tMinEventID:    event2.ID,\n\t\tMaxEventID:    event2.ID + 1,\n\t\tPageSize:      1,\n\t\tNextPageToken: nil,\n\t\tShardID:       shardID,\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents:    []*types.HistoryEvent{event2},\n\t\tNextPageToken:    nil,\n\t\tLastFirstEventID: event2.ID,\n\t}, nil)\n\n\ts.domainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.cache.PutEvent(domainID, workflowID, runID, event1.ID, event1)\n\ts.cache.PutEvent(domainID, workflowID, runID, event2.ID, event2)\n\ts.cache.disabled = true\n\tactualEvent, err := s.cache.GetEvent(context.Background(), *shardID, domainID, workflowID, runID, event2.ID, event2.ID, []byte(\"store_token\"))\n\ts.Nil(err)\n\ts.Equal(event2, actualEvent)\n}\n"
  },
  {
    "path": "service/history/events/notifier.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage events\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\teventsChanSize = 1000\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination notifier_mock.go -self_package github.com/uber/cadence/service/history/events\n\ntype (\n\t// Notifier is a pub-sub for sending and receiving notifications on new history events\n\tNotifier interface {\n\t\tcommon.Daemon\n\t\tNotifyNewHistoryEvent(event *Notification)\n\t\tWatchHistoryEvent(identifier definition.WorkflowIdentifier) (string, chan *Notification, error)\n\t\tUnwatchHistoryEvent(identifier definition.WorkflowIdentifier, subscriberID string) error\n\t}\n\n\t// Notification is the notification for new history events\n\tNotification struct {\n\t\tID                     definition.WorkflowIdentifier\n\t\tLastFirstEventID       int64\n\t\tNextEventID            int64\n\t\tPreviousStartedEventID int64\n\t\tTimestamp              time.Time\n\t\tWorkflowState          int\n\t\tWorkflowCloseState     int\n\t\tVersionHistories       *persistence.VersionHistories\n\t}\n\n\tnotifierImpl struct {\n\t\ttimeSource clock.TimeSource\n\t\tmetrics    metrics.Client\n\t\t// internal status indicator\n\t\tstatus int32\n\t\t// stop signal channel\n\t\tcloseChan chan bool\n\t\t// this channel will never close\n\t\teventsChan chan *Notification\n\t\t// function which calculate the shard ID from given workflow ID\n\t\tworkflowIDToShardID func(string) int\n\n\t\t// concurrent map with key workflowIdentifier, value map[string]chan *Notification.\n\t\t// the reason for the second map being non thread safe:\n\t\t// 1. expected number of subscriber per workflow is low, i.e. < 5\n\t\t// 2. update to this map is already guarded by GetAndDo API provided by ConcurrentTxMap\n\t\teventsPubsubs collection.ConcurrentTxMap\n\t}\n)\n\nvar _ Notifier = (*notifierImpl)(nil)\n\n// NewNotification creates a new history event notification\nfunc NewNotification(\n\tdomainID string,\n\tworkflowExecution *types.WorkflowExecution,\n\tlastFirstEventID int64,\n\tnextEventID int64,\n\tpreviousStartedEventID int64,\n\tworkflowState int,\n\tworkflowCloseState int,\n\tversionHistories *persistence.VersionHistories,\n) *Notification {\n\n\treturn &Notification{\n\t\tID: definition.NewWorkflowIdentifier(\n\t\t\tdomainID,\n\t\t\tworkflowExecution.GetWorkflowID(),\n\t\t\tworkflowExecution.GetRunID(),\n\t\t),\n\t\tLastFirstEventID:       lastFirstEventID,\n\t\tNextEventID:            nextEventID,\n\t\tPreviousStartedEventID: previousStartedEventID,\n\t\tWorkflowState:          workflowState,\n\t\tWorkflowCloseState:     workflowCloseState,\n\t\tVersionHistories:       versionHistories,\n\t}\n}\n\n// NewNotifier creates a new history event notifier\nfunc NewNotifier(\n\ttimeSource clock.TimeSource,\n\tmetrics metrics.Client,\n\tworkflowIDToShardID func(string) int,\n) Notifier {\n\n\thashFn := func(key interface{}) uint32 {\n\t\tnotification, ok := key.(Notification)\n\t\tif !ok {\n\t\t\treturn 0\n\t\t}\n\t\treturn uint32(workflowIDToShardID(notification.ID.WorkflowID))\n\t}\n\treturn &notifierImpl{\n\t\ttimeSource: timeSource,\n\t\tmetrics:    metrics,\n\t\tstatus:     common.DaemonStatusInitialized,\n\t\tcloseChan:  make(chan bool),\n\t\teventsChan: make(chan *Notification, eventsChanSize),\n\n\t\tworkflowIDToShardID: workflowIDToShardID,\n\n\t\teventsPubsubs: collection.NewShardedConcurrentTxMap(1024, hashFn),\n\t}\n}\n\nfunc (notifier *notifierImpl) WatchHistoryEvent(\n\tidentifier definition.WorkflowIdentifier) (string, chan *Notification, error) {\n\n\tchannel := make(chan *Notification, 1)\n\tsubscriberID := uuid.New()\n\tsubscribers := map[string]chan *Notification{\n\t\tsubscriberID: channel,\n\t}\n\n\t_, _, err := notifier.eventsPubsubs.PutOrDo(identifier, subscribers, func(key interface{}, value interface{}) error {\n\t\tsubscribers := value.(map[string]chan *Notification)\n\n\t\tif _, ok := subscribers[subscriberID]; ok {\n\t\t\t// UUID collision\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"Unable to watch on workflow execution.\",\n\t\t\t}\n\t\t}\n\t\tsubscribers[subscriberID] = channel\n\t\treturn nil\n\t})\n\n\tif err != nil {\n\t\treturn \"\", nil, err\n\t}\n\n\treturn subscriberID, channel, nil\n}\n\nfunc (notifier *notifierImpl) UnwatchHistoryEvent(\n\tidentifier definition.WorkflowIdentifier, subscriberID string) error {\n\n\tsuccess := true\n\tnotifier.eventsPubsubs.RemoveIf(identifier, func(key interface{}, value interface{}) bool {\n\t\tsubscribers := value.(map[string]chan *Notification)\n\n\t\tif _, ok := subscribers[subscriberID]; !ok {\n\t\t\t// cannot find the subscribe ID, which means there is a bug\n\t\t\tsuccess = false\n\t\t} else {\n\t\t\tdelete(subscribers, subscriberID)\n\t\t}\n\n\t\treturn len(subscribers) == 0\n\t})\n\n\tif !success {\n\t\t// cannot find the subscribe ID, which means there is a bug\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"Unable to unwatch on workflow execution.\",\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (notifier *notifierImpl) dispatchHistoryEventNotification(event *Notification) {\n\tidentifier := event.ID\n\n\ttimer := notifier.metrics.StartTimer(metrics.HistoryEventNotificationScope, metrics.HistoryEventNotificationFanoutLatency)\n\tdefer timer.Stop()\n\tnotifier.eventsPubsubs.GetAndDo(identifier, func(key interface{}, value interface{}) error { //nolint:errcheck\n\t\tsubscribers := value.(map[string]chan *Notification)\n\n\t\tfor _, channel := range subscribers {\n\t\t\tselect {\n\t\t\tcase channel <- event:\n\t\t\tdefault:\n\t\t\t\t// in case the channel is already filled with message\n\t\t\t\t// this should NOT happen, unless there is a bug or high load\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t})\n}\n\nfunc (notifier *notifierImpl) enqueueHistoryEventNotification(event *Notification) {\n\t// set the timestamp just before enqueuing the event\n\tevent.Timestamp = notifier.timeSource.Now()\n\tselect {\n\tcase notifier.eventsChan <- event:\n\tdefault:\n\t\t// in case the channel is already filled with message\n\t\t// this can be caused by high load\n\t\tnotifier.metrics.IncCounter(metrics.HistoryEventNotificationScope,\n\t\t\tmetrics.HistoryEventNotificationFailDeliveryCount)\n\t}\n}\n\nfunc (notifier *notifierImpl) dequeueHistoryEventNotifications() {\n\tfor {\n\t\t// send out metrics about the current number of messages in flight\n\t\tnotifier.metrics.UpdateGauge(metrics.HistoryEventNotificationScope,\n\t\t\tmetrics.HistoryEventNotificationInFlightMessageGauge, float64(len(notifier.eventsChan)))\n\t\tselect {\n\t\tcase event := <-notifier.eventsChan:\n\t\t\t// send out metrics about message processing delay\n\t\t\ttimeelapsed := time.Since(event.Timestamp)\n\t\t\tnotifier.metrics.RecordTimer(metrics.HistoryEventNotificationScope,\n\t\t\t\tmetrics.HistoryEventNotificationQueueingLatency, timeelapsed)\n\n\t\t\tnotifier.dispatchHistoryEventNotification(event)\n\t\tcase <-notifier.closeChan:\n\t\t\t// shutdown\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (notifier *notifierImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&notifier.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tgo notifier.dequeueHistoryEventNotifications()\n}\n\nfunc (notifier *notifierImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&notifier.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(notifier.closeChan)\n}\n\nfunc (notifier *notifierImpl) NotifyNewHistoryEvent(event *Notification) {\n\tnotifier.enqueueHistoryEventNotification(event)\n}\n"
  },
  {
    "path": "service/history/events/notifier_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: notifier.go\n//\n// Generated by this command:\n//\n//\tmockgen -package events -source notifier.go -destination notifier_mock.go -self_package github.com/uber/cadence/service/history/events\n//\n\n// Package events is a generated GoMock package.\npackage events\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tdefinition \"github.com/uber/cadence/common/definition\"\n)\n\n// MockNotifier is a mock of Notifier interface.\ntype MockNotifier struct {\n\tctrl     *gomock.Controller\n\trecorder *MockNotifierMockRecorder\n\tisgomock struct{}\n}\n\n// MockNotifierMockRecorder is the mock recorder for MockNotifier.\ntype MockNotifierMockRecorder struct {\n\tmock *MockNotifier\n}\n\n// NewMockNotifier creates a new mock instance.\nfunc NewMockNotifier(ctrl *gomock.Controller) *MockNotifier {\n\tmock := &MockNotifier{ctrl: ctrl}\n\tmock.recorder = &MockNotifierMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockNotifier) EXPECT() *MockNotifierMockRecorder {\n\treturn m.recorder\n}\n\n// NotifyNewHistoryEvent mocks base method.\nfunc (m *MockNotifier) NotifyNewHistoryEvent(event *Notification) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewHistoryEvent\", event)\n}\n\n// NotifyNewHistoryEvent indicates an expected call of NotifyNewHistoryEvent.\nfunc (mr *MockNotifierMockRecorder) NotifyNewHistoryEvent(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewHistoryEvent\", reflect.TypeOf((*MockNotifier)(nil).NotifyNewHistoryEvent), event)\n}\n\n// Start mocks base method.\nfunc (m *MockNotifier) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockNotifierMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockNotifier)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockNotifier) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockNotifierMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockNotifier)(nil).Stop))\n}\n\n// UnwatchHistoryEvent mocks base method.\nfunc (m *MockNotifier) UnwatchHistoryEvent(identifier definition.WorkflowIdentifier, subscriberID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UnwatchHistoryEvent\", identifier, subscriberID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UnwatchHistoryEvent indicates an expected call of UnwatchHistoryEvent.\nfunc (mr *MockNotifierMockRecorder) UnwatchHistoryEvent(identifier, subscriberID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UnwatchHistoryEvent\", reflect.TypeOf((*MockNotifier)(nil).UnwatchHistoryEvent), identifier, subscriberID)\n}\n\n// WatchHistoryEvent mocks base method.\nfunc (m *MockNotifier) WatchHistoryEvent(identifier definition.WorkflowIdentifier) (string, chan *Notification, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WatchHistoryEvent\", identifier)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(chan *Notification)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// WatchHistoryEvent indicates an expected call of WatchHistoryEvent.\nfunc (mr *MockNotifierMockRecorder) WatchHistoryEvent(identifier any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WatchHistoryEvent\", reflect.TypeOf((*MockNotifier)(nil).WatchHistoryEvent), identifier)\n}\n"
  },
  {
    "path": "service/history/events/notifier_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage events\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tnotifierSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\thistoryEventNotifier Notifier\n\t}\n)\n\nfunc TestHistoryEventNotifierSuite(t *testing.T) {\n\ts := new(notifierSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *notifierSuite) SetupSuite() {\n\n}\n\nfunc (s *notifierSuite) TearDownSuite() {\n\n}\n\nfunc (s *notifierSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.historyEventNotifier = NewNotifier(\n\t\tclock.NewRealTimeSource(),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\tfunc(workflowID string) int {\n\t\t\treturn len(workflowID)\n\t\t},\n\t)\n\ts.historyEventNotifier.Start()\n}\n\nfunc (s *notifierSuite) TearDownTest() {\n\ts.historyEventNotifier.Stop()\n}\n\nfunc (s *notifierSuite) TestSingleSubscriberWatchingEvents() {\n\tdomainID := \"domain ID\"\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"workflow ID\",\n\t\tRunID:      \"run ID\",\n\t}\n\tlastFirstEventID := int64(3)\n\tpreviousStartedEventID := int64(5)\n\tnextEventID := int64(18)\n\tworkflowState := persistence.WorkflowStateCreated\n\tworkflowCloseState := persistence.WorkflowCloseStatusNone\n\tversionHistory := persistence.VersionHistories{}\n\thistoryEvent := NewNotification(\n\t\tdomainID,\n\t\texecution,\n\t\tlastFirstEventID,\n\t\tnextEventID,\n\t\tpreviousStartedEventID,\n\t\tworkflowState,\n\t\tworkflowCloseState,\n\t\t&versionHistory,\n\t)\n\ttimerChan := time.NewTimer(time.Second * 2).C\n\n\tsubscriberID, channel, err := s.historyEventNotifier.WatchHistoryEvent(definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID()))\n\ts.Nil(err)\n\n\tgo func() {\n\t\t<-timerChan\n\t\ts.historyEventNotifier.NotifyNewHistoryEvent(historyEvent)\n\t}()\n\n\tmsg := <-channel\n\ts.Equal(historyEvent, msg)\n\n\terr = s.historyEventNotifier.UnwatchHistoryEvent(definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID()), subscriberID)\n\ts.Nil(err)\n}\n\nfunc (s *notifierSuite) TestMultipleSubscriberWatchingEvents() {\n\tdomainID := \"domain ID\"\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"workflow ID\",\n\t\tRunID:      \"run ID\",\n\t}\n\n\tlastFirstEventID := int64(3)\n\tpreviousStartedEventID := int64(5)\n\tnextEventID := int64(18)\n\tworkflowState := persistence.WorkflowStateCreated\n\tworkflowCloseState := persistence.WorkflowCloseStatusNone\n\tversionHistories := &persistence.VersionHistories{}\n\thistoryEvent := NewNotification(domainID, execution, lastFirstEventID, nextEventID, previousStartedEventID, workflowState, workflowCloseState, versionHistories)\n\ttimerChan := time.NewTimer(time.Second * 5).C\n\n\tsubscriberCount := 100\n\twaitGroup := sync.WaitGroup{}\n\twaitGroup.Add(subscriberCount)\n\n\twatchFunc := func() {\n\t\tsubscriberID, channel, err := s.historyEventNotifier.WatchHistoryEvent(definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID()))\n\t\ts.Nil(err)\n\n\t\ttimeourChan := time.NewTimer(time.Second * 10).C\n\n\t\tselect {\n\t\tcase msg := <-channel:\n\t\t\ts.Equal(historyEvent, msg)\n\t\tcase <-timeourChan:\n\t\t\ts.Fail(\"subscribe to new events timeout\")\n\t\t}\n\t\terr = s.historyEventNotifier.UnwatchHistoryEvent(definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID()), subscriberID)\n\t\ts.Nil(err)\n\t\twaitGroup.Done()\n\t}\n\n\tfor count := 0; count < subscriberCount; count++ {\n\t\tgo watchFunc()\n\t}\n\n\t<-timerChan\n\ts.historyEventNotifier.NotifyNewHistoryEvent(historyEvent)\n\twaitGroup.Wait()\n}\n"
  },
  {
    "path": "service/history/execution/cache.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination cache_mock.go\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n// Cache is a cache that holds workflow execution context\ntype Cache interface {\n\tcache.Cache\n\n\t// GetOrCreateCurrentWorkflowExecution gets or creates workflow execution context for the current run\n\tGetOrCreateCurrentWorkflowExecution(\n\t\tctx context.Context,\n\t\tdomainID string,\n\t\tworkflowID string,\n\t) (Context, ReleaseFunc, error)\n\n\t// GetAndCreateWorkflowExecution is for analyzing mutableState, it will try getting Context from cache\n\t// and also load from database\n\tGetAndCreateWorkflowExecution(\n\t\tctx context.Context,\n\t\tdomainID string,\n\t\texecution types.WorkflowExecution,\n\t) (Context, Context, ReleaseFunc, bool, error)\n\n\t// GetOrCreateWorkflowExecutionForBackground gets or creates workflow execution context with background context\n\tGetOrCreateWorkflowExecutionForBackground(\n\t\tdomainID string,\n\t\texecution types.WorkflowExecution,\n\t) (Context, ReleaseFunc, error)\n\n\t// GetOrCreateWorkflowExecutionWithTimeout gets or creates workflow execution context with timeout\n\tGetOrCreateWorkflowExecutionWithTimeout(\n\t\tdomainID string,\n\t\texecution types.WorkflowExecution,\n\t\ttimeout time.Duration,\n\t) (Context, ReleaseFunc, error)\n\n\t// GetOrCreateWorkflowExecution gets or creates workflow execution context\n\tGetOrCreateWorkflowExecution(\n\t\tctx context.Context,\n\t\tdomainID string,\n\t\texecution types.WorkflowExecution,\n\t) (Context, ReleaseFunc, error)\n}\n\ntype (\n\t// ReleaseFunc releases workflow execution context\n\tReleaseFunc func(err error)\n\n\t// Cache caches workflow execution context\n\tcacheImpl struct {\n\t\tcache.Cache\n\t\tshard            shard.Context\n\t\texecutionManager persistence.ExecutionManager\n\t\tdisabled         bool\n\t\tlogger           log.Logger\n\t\tmetricsClient    metrics.Client\n\t\tconfig           *config.Config\n\t}\n)\n\nvar (\n\t// NoopReleaseFn is an no-op implementation for the ReleaseFunc type\n\tNoopReleaseFn ReleaseFunc = func(err error) {}\n)\n\nconst (\n\tcacheNotReleased int32 = 0\n\tcacheReleased    int32 = 1\n)\n\n// NewCache creates a new workflow execution context cache\nfunc NewCache(shard shard.Context) Cache {\n\topts := &cache.Options{}\n\tconfig := shard.GetConfig()\n\topts.InitialCapacity = config.HistoryCacheInitialSize()\n\topts.TTL = config.HistoryCacheTTL()\n\topts.Pin = true\n\topts.MaxCount = config.HistoryCacheMaxSize()\n\topts.MetricsScope = shard.GetMetricsClient().Scope(metrics.HistoryExecutionCacheScope).Tagged(metrics.ShardIDTag(shard.GetShardID()))\n\topts.Logger = shard.GetLogger().WithTags(tag.ComponentHistoryCache)\n\topts.IsSizeBased = config.EnableSizeBasedHistoryExecutionCache\n\topts.MaxSize = config.ExecutionCacheMaxByteSize\n\n\treturn &cacheImpl{\n\t\tCache:            cache.New(opts),\n\t\tshard:            shard,\n\t\texecutionManager: shard.GetExecutionManager(),\n\t\tlogger:           shard.GetLogger().WithTags(tag.ComponentHistoryCache),\n\t\tmetricsClient:    shard.GetMetricsClient(),\n\t\tconfig:           config,\n\t}\n}\n\n// GetOrCreateCurrentWorkflowExecution gets or creates workflow execution context for the current run\nfunc (c *cacheImpl) GetOrCreateCurrentWorkflowExecution(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n) (Context, ReleaseFunc, error) {\n\n\tscope := metrics.HistoryCacheGetOrCreateCurrentScope\n\tc.metricsClient.IncCounter(scope, metrics.CacheRequests)\n\tsw := c.metricsClient.StartTimer(scope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\t// using empty run ID as current workflow run ID\n\trunID := \"\"\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\treturn c.getOrCreateWorkflowExecutionInternal(\n\t\tctx,\n\t\tdomainID,\n\t\texecution,\n\t\tscope,\n\t\ttrue,\n\t)\n}\n\n// GetAndCreateWorkflowExecution is for analyzing mutableState, it will try getting Context from cache\n// and also load from database\nfunc (c *cacheImpl) GetAndCreateWorkflowExecution(\n\tctx context.Context,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (Context, Context, ReleaseFunc, bool, error) {\n\n\tscope := metrics.HistoryCacheGetAndCreateScope\n\tc.metricsClient.IncCounter(scope, metrics.CacheRequests)\n\tsw := c.metricsClient.StartTimer(scope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tif err := c.validateWorkflowExecutionInfo(ctx, domainID, &execution); err != nil {\n\t\tc.metricsClient.IncCounter(scope, metrics.CacheFailures)\n\t\treturn nil, nil, nil, false, err\n\t}\n\n\tkey := definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID())\n\tcontextFromCache, cacheHit := c.Get(key).(Context)\n\t// TODO This will create a closure on every request.\n\t//  Consider revisiting this if it causes too much GC activity\n\treleaseFunc := NoopReleaseFn\n\t// If cache hit, we need to lock the cache to prevent race condition\n\tif cacheHit {\n\t\tif err := contextFromCache.Lock(ctx); err != nil {\n\t\t\t// ctx is done before lock can be acquired\n\t\t\tc.Release(key)\n\t\t\tc.metricsClient.IncCounter(metrics.HistoryCacheGetAndCreateScope, metrics.CacheFailures)\n\t\t\tc.metricsClient.IncCounter(metrics.HistoryCacheGetAndCreateScope, metrics.AcquireLockFailedCounter)\n\t\t\treturn nil, nil, nil, false, err\n\t\t}\n\t\treleaseFunc = c.makeReleaseFunc(key, contextFromCache, false)\n\t} else {\n\t\tc.metricsClient.IncCounter(metrics.HistoryCacheGetAndCreateScope, metrics.CacheMissCounter)\n\t}\n\n\t// Note, the one loaded from DB is not put into cache and don't affect any behavior\n\tcontextFromDB := NewContext(domainID, execution, c.shard, c.executionManager, c.logger)\n\treturn contextFromCache, contextFromDB, releaseFunc, cacheHit, nil\n}\n\n// GetOrCreateWorkflowExecutionForBackground gets or creates workflow execution context with background context\n// currently only used in tests\nfunc (c *cacheImpl) GetOrCreateWorkflowExecutionForBackground(\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (Context, ReleaseFunc, error) {\n\n\treturn c.GetOrCreateWorkflowExecution(context.Background(), domainID, execution)\n}\n\n// GetOrCreateWorkflowExecutionWithTimeout gets or creates workflow execution context with timeout\nfunc (c *cacheImpl) GetOrCreateWorkflowExecutionWithTimeout(\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\ttimeout time.Duration,\n) (Context, ReleaseFunc, error) {\n\n\tctx, cancel := context.WithTimeout(context.Background(), timeout)\n\tdefer cancel()\n\n\treturn c.GetOrCreateWorkflowExecution(ctx, domainID, execution)\n}\n\n// GetOrCreateWorkflowExecution gets or creates workflow execution context\nfunc (c *cacheImpl) GetOrCreateWorkflowExecution(\n\tctx context.Context,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (Context, ReleaseFunc, error) {\n\n\tscope := metrics.HistoryCacheGetOrCreateScope\n\tc.metricsClient.IncCounter(scope, metrics.CacheRequests)\n\tsw := c.metricsClient.StartTimer(scope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tif err := c.validateWorkflowExecutionInfo(ctx, domainID, &execution); err != nil {\n\t\tc.metricsClient.IncCounter(scope, metrics.CacheFailures)\n\t\treturn nil, nil, err\n\t}\n\n\treturn c.getOrCreateWorkflowExecutionInternal(\n\t\tctx,\n\t\tdomainID,\n\t\texecution,\n\t\tscope,\n\t\tfalse,\n\t)\n}\n\nfunc (c *cacheImpl) getOrCreateWorkflowExecutionInternal(\n\tctx context.Context,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\tscope metrics.ScopeIdx,\n\tforceClearContext bool,\n) (Context, ReleaseFunc, error) {\n\n\t// Test hook for disabling the cache\n\tif c.disabled {\n\t\treturn NewContext(domainID, execution, c.shard, c.executionManager, c.logger), NoopReleaseFn, nil\n\t}\n\n\tkey := definition.NewWorkflowIdentifier(domainID, execution.GetWorkflowID(), execution.GetRunID())\n\tworkflowCtx, cacheHit := c.Get(key).(Context)\n\tif !cacheHit {\n\t\tc.metricsClient.IncCounter(scope, metrics.CacheMissCounter)\n\t\t// Let's create the workflow execution workflowCtx\n\t\tworkflowCtx = NewContext(domainID, execution, c.shard, c.executionManager, c.logger)\n\t\telem, err := c.PutIfNotExist(key, workflowCtx)\n\t\tif err != nil {\n\t\t\tc.metricsClient.IncCounter(scope, metrics.CacheFailures)\n\t\t\treturn nil, nil, err\n\t\t}\n\t\tworkflowCtx = elem.(Context)\n\t}\n\n\t// TODO This will create a closure on every request.\n\t//  Consider revisiting this if it causes too much GC activity\n\treleaseFunc := c.makeReleaseFunc(key, workflowCtx, forceClearContext)\n\n\tif err := workflowCtx.Lock(ctx); err != nil {\n\t\t// ctx is done before lock can be acquired\n\t\tc.Release(key)\n\t\tc.metricsClient.IncCounter(scope, metrics.CacheFailures)\n\t\tc.metricsClient.IncCounter(scope, metrics.AcquireLockFailedCounter)\n\t\treturn nil, nil, err\n\t}\n\treturn workflowCtx, releaseFunc, nil\n}\n\nfunc (c *cacheImpl) validateWorkflowExecutionInfo(\n\tctx context.Context,\n\tdomainID string,\n\texecution *types.WorkflowExecution,\n) error {\n\n\tif execution.GetWorkflowID() == \"\" {\n\t\treturn &types.BadRequestError{Message: \"Can't load workflow execution.  WorkflowId not set.\"}\n\t}\n\tdomainName, err := c.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\t// RunID is not provided, lets try to retrieve the RunID for current active execution\n\tif execution.GetRunID() == \"\" {\n\t\tresponse, err := c.getCurrentExecutionWithRetry(ctx, &persistence.GetCurrentExecutionRequest{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: execution.GetWorkflowID(),\n\t\t\tDomainName: domainName,\n\t\t})\n\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\texecution.RunID = response.RunID\n\t} else if uuid.Parse(execution.GetRunID()) == nil { // immediately return if invalid runID\n\t\treturn &types.BadRequestError{Message: \"RunID is not valid UUID.\"}\n\t}\n\treturn nil\n}\n\nfunc (c *cacheImpl) makeReleaseFunc(\n\tkey definition.WorkflowIdentifier,\n\tcontext Context,\n\tforceClearContext bool,\n) func(error) {\n\n\tstatus := cacheNotReleased\n\treturn func(err error) {\n\t\tdefer func() {\n\t\t\tif atomic.CompareAndSwapInt32(&status, cacheNotReleased, cacheReleased) {\n\t\t\t\tif rec := recover(); rec != nil {\n\t\t\t\t\tcontext.Clear()\n\t\t\t\t\tcontext.Unlock()\n\t\t\t\t\tc.Release(key)\n\t\t\t\t\tpanic(rec)\n\t\t\t\t} else {\n\t\t\t\t\tif err != nil || forceClearContext {\n\t\t\t\t\t\t// TODO see issue #668, there are certain type or errors which can bypass the clear\n\t\t\t\t\t\tcontext.Clear()\n\t\t\t\t\t}\n\t\t\t\t\tcontext.Unlock()\n\t\t\t\t\tc.Release(key)\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n}\n\nfunc (c *cacheImpl) getCurrentExecutionWithRetry(\n\tctx context.Context,\n\trequest *persistence.GetCurrentExecutionRequest,\n) (*persistence.GetCurrentExecutionResponse, error) {\n\n\tc.metricsClient.IncCounter(metrics.HistoryCacheGetCurrentExecutionScope, metrics.CacheRequests)\n\tsw := c.metricsClient.StartTimer(metrics.HistoryCacheGetCurrentExecutionScope, metrics.CacheLatency)\n\tdefer sw.Stop()\n\n\tvar response *persistence.GetCurrentExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresponse, err = c.executionManager.GetCurrentExecution(ctx, request)\n\n\t\treturn err\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreatePersistenceRetryPolicy()),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\tif err != nil {\n\t\tc.metricsClient.IncCounter(metrics.HistoryCacheGetCurrentExecutionScope, metrics.CacheFailures)\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n"
  },
  {
    "path": "service/history/execution/cache_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: cache.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source cache.go -destination cache_mock.go\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tcache \"github.com/uber/cadence/common/cache\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockCache is a mock of Cache interface.\ntype MockCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockCacheMockRecorder is the mock recorder for MockCache.\ntype MockCacheMockRecorder struct {\n\tmock *MockCache\n}\n\n// NewMockCache creates a new mock instance.\nfunc NewMockCache(ctrl *gomock.Controller) *MockCache {\n\tmock := &MockCache{ctrl: ctrl}\n\tmock.recorder = &MockCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCache) EXPECT() *MockCacheMockRecorder {\n\treturn m.recorder\n}\n\n// Delete mocks base method.\nfunc (m *MockCache) Delete(key any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Delete\", key)\n}\n\n// Delete indicates an expected call of Delete.\nfunc (mr *MockCacheMockRecorder) Delete(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Delete\", reflect.TypeOf((*MockCache)(nil).Delete), key)\n}\n\n// Get mocks base method.\nfunc (m *MockCache) Get(key any) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Get\", key)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Get indicates an expected call of Get.\nfunc (mr *MockCacheMockRecorder) Get(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Get\", reflect.TypeOf((*MockCache)(nil).Get), key)\n}\n\n// GetAndCreateWorkflowExecution mocks base method.\nfunc (m *MockCache) GetAndCreateWorkflowExecution(ctx context.Context, domainID string, execution types.WorkflowExecution) (Context, Context, ReleaseFunc, bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAndCreateWorkflowExecution\", ctx, domainID, execution)\n\tret0, _ := ret[0].(Context)\n\tret1, _ := ret[1].(Context)\n\tret2, _ := ret[2].(ReleaseFunc)\n\tret3, _ := ret[3].(bool)\n\tret4, _ := ret[4].(error)\n\treturn ret0, ret1, ret2, ret3, ret4\n}\n\n// GetAndCreateWorkflowExecution indicates an expected call of GetAndCreateWorkflowExecution.\nfunc (mr *MockCacheMockRecorder) GetAndCreateWorkflowExecution(ctx, domainID, execution any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAndCreateWorkflowExecution\", reflect.TypeOf((*MockCache)(nil).GetAndCreateWorkflowExecution), ctx, domainID, execution)\n}\n\n// GetOrCreateCurrentWorkflowExecution mocks base method.\nfunc (m *MockCache) GetOrCreateCurrentWorkflowExecution(ctx context.Context, domainID, workflowID string) (Context, ReleaseFunc, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrCreateCurrentWorkflowExecution\", ctx, domainID, workflowID)\n\tret0, _ := ret[0].(Context)\n\tret1, _ := ret[1].(ReleaseFunc)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetOrCreateCurrentWorkflowExecution indicates an expected call of GetOrCreateCurrentWorkflowExecution.\nfunc (mr *MockCacheMockRecorder) GetOrCreateCurrentWorkflowExecution(ctx, domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrCreateCurrentWorkflowExecution\", reflect.TypeOf((*MockCache)(nil).GetOrCreateCurrentWorkflowExecution), ctx, domainID, workflowID)\n}\n\n// GetOrCreateWorkflowExecution mocks base method.\nfunc (m *MockCache) GetOrCreateWorkflowExecution(ctx context.Context, domainID string, execution types.WorkflowExecution) (Context, ReleaseFunc, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrCreateWorkflowExecution\", ctx, domainID, execution)\n\tret0, _ := ret[0].(Context)\n\tret1, _ := ret[1].(ReleaseFunc)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetOrCreateWorkflowExecution indicates an expected call of GetOrCreateWorkflowExecution.\nfunc (mr *MockCacheMockRecorder) GetOrCreateWorkflowExecution(ctx, domainID, execution any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrCreateWorkflowExecution\", reflect.TypeOf((*MockCache)(nil).GetOrCreateWorkflowExecution), ctx, domainID, execution)\n}\n\n// GetOrCreateWorkflowExecutionForBackground mocks base method.\nfunc (m *MockCache) GetOrCreateWorkflowExecutionForBackground(domainID string, execution types.WorkflowExecution) (Context, ReleaseFunc, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrCreateWorkflowExecutionForBackground\", domainID, execution)\n\tret0, _ := ret[0].(Context)\n\tret1, _ := ret[1].(ReleaseFunc)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetOrCreateWorkflowExecutionForBackground indicates an expected call of GetOrCreateWorkflowExecutionForBackground.\nfunc (mr *MockCacheMockRecorder) GetOrCreateWorkflowExecutionForBackground(domainID, execution any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrCreateWorkflowExecutionForBackground\", reflect.TypeOf((*MockCache)(nil).GetOrCreateWorkflowExecutionForBackground), domainID, execution)\n}\n\n// GetOrCreateWorkflowExecutionWithTimeout mocks base method.\nfunc (m *MockCache) GetOrCreateWorkflowExecutionWithTimeout(domainID string, execution types.WorkflowExecution, timeout time.Duration) (Context, ReleaseFunc, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrCreateWorkflowExecutionWithTimeout\", domainID, execution, timeout)\n\tret0, _ := ret[0].(Context)\n\tret1, _ := ret[1].(ReleaseFunc)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetOrCreateWorkflowExecutionWithTimeout indicates an expected call of GetOrCreateWorkflowExecutionWithTimeout.\nfunc (mr *MockCacheMockRecorder) GetOrCreateWorkflowExecutionWithTimeout(domainID, execution, timeout any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrCreateWorkflowExecutionWithTimeout\", reflect.TypeOf((*MockCache)(nil).GetOrCreateWorkflowExecutionWithTimeout), domainID, execution, timeout)\n}\n\n// Iterator mocks base method.\nfunc (m *MockCache) Iterator() cache.Iterator {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Iterator\")\n\tret0, _ := ret[0].(cache.Iterator)\n\treturn ret0\n}\n\n// Iterator indicates an expected call of Iterator.\nfunc (mr *MockCacheMockRecorder) Iterator() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Iterator\", reflect.TypeOf((*MockCache)(nil).Iterator))\n}\n\n// Put mocks base method.\nfunc (m *MockCache) Put(key, value any) any {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Put\", key, value)\n\tret0, _ := ret[0].(any)\n\treturn ret0\n}\n\n// Put indicates an expected call of Put.\nfunc (mr *MockCacheMockRecorder) Put(key, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Put\", reflect.TypeOf((*MockCache)(nil).Put), key, value)\n}\n\n// PutIfNotExist mocks base method.\nfunc (m *MockCache) PutIfNotExist(key, value any) (any, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PutIfNotExist\", key, value)\n\tret0, _ := ret[0].(any)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PutIfNotExist indicates an expected call of PutIfNotExist.\nfunc (mr *MockCacheMockRecorder) PutIfNotExist(key, value any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PutIfNotExist\", reflect.TypeOf((*MockCache)(nil).PutIfNotExist), key, value)\n}\n\n// Release mocks base method.\nfunc (m *MockCache) Release(key any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Release\", key)\n}\n\n// Release indicates an expected call of Release.\nfunc (mr *MockCacheMockRecorder) Release(key any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Release\", reflect.TypeOf((*MockCache)(nil).Release), key)\n}\n\n// Size mocks base method.\nfunc (m *MockCache) Size() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Size\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Size indicates an expected call of Size.\nfunc (mr *MockCacheMockRecorder) Size() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Size\", reflect.TypeOf((*MockCache)(nil).Size))\n}\n"
  },
  {
    "path": "service/history/execution/cache_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"reflect\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\thistoryCacheSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller *gomock.Controller\n\t\tmockShard  *shard.TestContext\n\n\t\tcache Cache\n\t}\n)\n\nfunc TestHistoryCacheSuite(t *testing.T) {\n\ts := new(historyCacheSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *historyCacheSuite) SetupSuite() {\n}\n\nfunc (s *historyCacheSuite) TearDownSuite() {\n}\n\nfunc (s *historyCacheSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n}\n\nfunc (s *historyCacheSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *historyCacheSuite) TestHistoryCacheBasic() {\n\ts.cache = NewCache(s.mockShard)\n\n\tdomainID := \"test_domain_id\"\n\tdomainName := \"test_domain_name\"\n\texecution1 := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\tmockMS1 := NewMockMutableState(s.controller)\n\tctx, release, err := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, execution1)\n\ts.Nil(err)\n\tctx.(*contextImpl).mutableState = mockMS1\n\trelease(nil)\n\tctx, release, err = s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, execution1)\n\ts.Nil(err)\n\ts.Equal(mockMS1, ctx.(*contextImpl).mutableState)\n\trelease(nil)\n\n\texecution2 := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tctx, release, err = s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, execution2)\n\ts.Nil(err)\n\ts.NotEqual(mockMS1, ctx.(*contextImpl).mutableState)\n\trelease(nil)\n}\n\nfunc (s *historyCacheSuite) TestHistoryCachePinning() {\n\ts.mockShard.GetConfig().HistoryCacheMaxSize = dynamicproperties.GetIntPropertyFn(1)\n\tdomainID := \"test_domain_id\"\n\ts.cache = NewCache(s.mockShard)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wf-cache-test-pinning\",\n\t\tRunID:      uuid.New(),\n\t}\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\tctx, release, err := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err)\n\n\twe2 := types.WorkflowExecution{\n\t\tWorkflowID: \"wf-cache-test-pinning\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\t// Cache is full because ctx is pinned, should get an error now\n\t_, _, err2 := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we2)\n\ts.NotNil(err2)\n\n\t// Now release the ctx, this should unpin it.\n\trelease(err2)\n\n\t_, release2, err3 := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we2)\n\ts.Nil(err3)\n\trelease2(err3)\n\n\t// Old ctx should be evicted.\n\tnewContext, release, err4 := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err4)\n\ts.False(ctx == newContext)\n\trelease(err4)\n}\n\nfunc (s *historyCacheSuite) TestHistoryCacheClear() {\n\ts.mockShard.GetConfig().HistoryCacheMaxSize = dynamicproperties.GetIntPropertyFn(20)\n\tdomainID := \"test_domain_id\"\n\ts.cache = NewCache(s.mockShard)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wf-cache-test-clear\",\n\t\tRunID:      uuid.New(),\n\t}\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\tctx, release, err := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err)\n\t// since we are just testing whether the release function will clear the cache\n\t// all we need is a fake msBuilder\n\tctx.(*contextImpl).mutableState = &mutableStateBuilder{}\n\trelease(nil)\n\n\t// since last time, the release function receive a nil error\n\t// the ms builder will not be cleared\n\tctx, release, err = s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err)\n\ts.NotNil(ctx.(*contextImpl).mutableState)\n\trelease(errors.New(\"some random error message\"))\n\n\t// since last time, the release function receive a non-nil error\n\t// the ms builder will be cleared\n\tctx, release, err = s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err)\n\ts.Nil(ctx.(*contextImpl).mutableState)\n\trelease(nil)\n}\n\nfunc (s *historyCacheSuite) TestHistoryCacheConcurrentAccess() {\n\ts.mockShard.GetConfig().HistoryCacheMaxSize = dynamicproperties.GetIntPropertyFn(20)\n\tdomainID := \"test_domain_id\"\n\ts.cache = NewCache(s.mockShard)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wf-cache-test-pinning\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tcoroutineCount := 50\n\twaitGroup := &sync.WaitGroup{}\n\tstopChan := make(chan struct{})\n\ttestFn := func() {\n\t\t<-stopChan\n\t\tctx, release, err := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\t\ts.Nil(err)\n\t\t// since each time the builder is reset to nil\n\t\ts.Nil(ctx.(*contextImpl).mutableState)\n\t\t// since we are just testing whether the release function will clear the cache\n\t\t// all we need is a fake msBuilder\n\t\tctx.(*contextImpl).mutableState = &mutableStateBuilder{}\n\t\trelease(errors.New(\"some random error message\"))\n\t\twaitGroup.Done()\n\t}\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\tfor i := 0; i < coroutineCount; i++ {\n\t\twaitGroup.Add(1)\n\t\tgo testFn()\n\t}\n\tclose(stopChan)\n\twaitGroup.Wait()\n\n\tctx, release, err := s.cache.GetOrCreateWorkflowExecutionForBackground(domainID, we)\n\ts.Nil(err)\n\t// since we are just testing whether the release function will clear the cache\n\t// all we need is a fake msBuilder\n\ts.Nil(ctx.(*contextImpl).mutableState)\n\trelease(nil)\n}\n\nfunc (s *historyCacheSuite) TestGetOrCreateCurrentWorkflowExecution() {\n\ttests := []struct {\n\t\tname          string\n\t\tmockSetup     func(c *cacheImpl, mockContext *MockContext, ctx context.Context)\n\t\tdisabled      bool\n\t\tcacheCtxNil   bool\n\t\tnoOpReleaseFn bool\n\t\terr           error\n\t}{\n\t\t{\n\t\t\tname:      \"success - cache enabled - cache miss\",\n\t\t\tmockSetup: func(_ *cacheImpl, _ *MockContext, _ context.Context) {},\n\t\t},\n\t\t{\n\t\t\tname:          \"success - cache disabled\",\n\t\t\tmockSetup:     func(_ *cacheImpl, _ *MockContext, _ context.Context) {},\n\t\t\tdisabled:      true,\n\t\t\tnoOpReleaseFn: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - cache enabled - cache hit\",\n\t\t\tmockSetup: func(c *cacheImpl, mockContext *MockContext, ctx context.Context) {\n\t\t\t\tmockContext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t\t\t\tkey := definition.NewWorkflowIdentifier(constants.TestDomainID, constants.TestWorkflowID, \"\")\n\t\t\t\t_, _ = c.Cache.PutIfNotExist(key, mockContext)\n\t\t\t\tmockContext.EXPECT().Lock(ctx).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error - cache enabled - cache hit error on lock\",\n\t\t\tmockSetup: func(cache *cacheImpl, mockContext *MockContext, ctx context.Context) {\n\t\t\t\tmockContext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t\t\t\tkey := definition.NewWorkflowIdentifier(constants.TestDomainID, constants.TestWorkflowID, \"\")\n\t\t\t\t_, _ = cache.Cache.PutIfNotExist(key, mockContext)\n\t\t\t\tmockContext.EXPECT().Lock(ctx).Return(errors.New(\"test-error\")).Times(1)\n\t\t\t},\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         errors.New(\"test-error\"),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\tc := &cacheImpl{\n\t\t\t\tCache: cache.New(&cache.Options{\n\t\t\t\t\tInitialCapacity: s.mockShard.GetConfig().HistoryCacheInitialSize(),\n\t\t\t\t\tTTL:             s.mockShard.GetConfig().HistoryCacheTTL(),\n\t\t\t\t\tPin:             true,\n\t\t\t\t\tMaxCount:        s.mockShard.GetConfig().HistoryCacheMaxSize(),\n\t\t\t\t\tLogger:          s.mockShard.GetLogger().WithTags(tag.ComponentHistoryCache),\n\t\t\t\t}),\n\t\t\t\tshard:            s.mockShard,\n\t\t\t\texecutionManager: s.mockShard.GetExecutionManager(),\n\t\t\t\tlogger:           s.mockShard.GetLogger().WithTags(tag.ComponentHistoryCache),\n\t\t\t\tmetricsClient:    s.mockShard.GetMetricsClient(),\n\t\t\t\tconfig:           s.mockShard.GetConfig(),\n\t\t\t}\n\t\t\tmockContext := NewMockContext(s.controller)\n\n\t\t\tc.disabled = tt.disabled\n\t\t\tctx := context.Background()\n\t\t\ttt.mockSetup(c, mockContext, ctx)\n\n\t\t\tcacheCtx, releaseFunc, err := c.GetOrCreateCurrentWorkflowExecution(ctx, constants.TestDomainID, constants.TestWorkflowID)\n\n\t\t\tif tt.cacheCtxNil {\n\t\t\t\ts.Nil(cacheCtx)\n\t\t\t} else {\n\t\t\t\ts.NotNil(cacheCtx)\n\t\t\t}\n\n\t\t\tif tt.noOpReleaseFn {\n\t\t\t\ts.True(reflect.ValueOf(NoopReleaseFn).Pointer() == reflect.ValueOf(releaseFunc).Pointer())\n\t\t\t}\n\n\t\t\tif tt.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tt.err, err)\n\t\t\t\ts.Nil(releaseFunc)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(releaseFunc)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *historyCacheSuite) TestGetOrCreateWorkflowExecution() {\n\ttests := []struct {\n\t\tname        string\n\t\tworkflowID  string\n\t\trunID       string\n\t\tcacheCtxNil bool\n\t\tmockSetup   func(mockShard *shard.TestContext)\n\t\terr         error\n\t}{\n\t\t{\n\t\t\tname:        \"error - empty workflow ID\",\n\t\t\tworkflowID:  \"\",\n\t\t\trunID:       constants.TestRunID,\n\t\t\tmockSetup:   func(_ *shard.TestContext) {},\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         &types.BadRequestError{Message: \"Can't load workflow execution.  WorkflowId not set.\"},\n\t\t},\n\t\t{\n\t\t\tname:       \"error - get domain cache error\",\n\t\t\tworkflowID: constants.TestWorkflowID,\n\t\t\trunID:      constants.TestRunID,\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(\"\", errors.New(\"test-error\")).Times(1)\n\t\t\t},\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         errors.New(\"test-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"error - not valid runID\",\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t},\n\t\t\tworkflowID:  constants.TestWorkflowID,\n\t\t\trunID:       \"invalid-run-id\",\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         &types.BadRequestError{Message: \"RunID is not valid UUID.\"},\n\t\t},\n\t\t{\n\t\t\tname: \"error - empty runID retry failed\",\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t\treq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tmockShard.GetExecutionManager().(*mocks.ExecutionManager).On(\"GetCurrentExecution\", mock.Anything, req).Return(nil, errors.New(\"test-error\")).Times(1)\n\t\t\t},\n\t\t\tworkflowID:  constants.TestWorkflowID,\n\t\t\trunID:       \"\",\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         errors.New(\"test-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"success - runID provided\",\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t},\n\t\t\tworkflowID: constants.TestWorkflowID,\n\t\t\trunID:      constants.TestRunID,\n\t\t\terr:        nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - runID not provided\",\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t\treq := &persistence.GetCurrentExecutionRequest{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainName: constants.TestDomainName,\n\t\t\t\t}\n\t\t\t\tresp := &persistence.GetCurrentExecutionResponse{\n\t\t\t\t\tRunID: constants.TestRunID,\n\t\t\t\t}\n\t\t\t\tmockShard.GetExecutionManager().(*mocks.ExecutionManager).On(\"GetCurrentExecution\", mock.Anything, req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\tworkflowID: constants.TestWorkflowID,\n\t\t\trunID:      \"\",\n\t\t\terr:        nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.cache = NewCache(s.mockShard)\n\n\t\t\tctx := context.Background()\n\t\t\ttt.mockSetup(s.mockShard)\n\n\t\t\tcacheCtx, releaseFunc, err := s.cache.GetOrCreateWorkflowExecution(ctx, constants.TestDomainID, types.WorkflowExecution{WorkflowID: tt.workflowID, RunID: tt.runID})\n\n\t\t\tif tt.cacheCtxNil {\n\t\t\t\ts.Nil(cacheCtx)\n\t\t\t} else {\n\t\t\t\ts.NotNil(cacheCtx)\n\t\t\t}\n\n\t\t\tif tt.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tt.err, err)\n\t\t\t\ts.Nil(releaseFunc)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(releaseFunc)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *historyCacheSuite) TestGetAndCreateWorkflowExecution() {\n\tctx := context.Background()\n\n\ttests := []struct {\n\t\tname        string\n\t\tdomainID    string\n\t\texecution   types.WorkflowExecution\n\t\tmockSetup   func(mockShard *shard.TestContext, mockContext *MockContext, c *cacheImpl)\n\t\tcacheHit    bool\n\t\tcacheCtxNil bool\n\t\terr         error\n\t}{\n\t\t{\n\t\t\tname:     \"error - could not validate workflow execution info\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.TestContext, _ *MockContext, _ *cacheImpl) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(\"\", errors.New(\"validate-error\")).Times(1)\n\t\t\t},\n\t\t\tcacheHit:    false,\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         errors.New(\"validate-error\"),\n\t\t},\n\t\t{\n\t\t\tname:     \"success - cache miss\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.TestContext, _ *MockContext, _ *cacheImpl) {\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t},\n\t\t\tcacheHit:    false,\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"error - cache hit error on lock\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.TestContext, mockContext *MockContext, c *cacheImpl) {\n\t\t\t\tmockContext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t\tkey := definition.NewWorkflowIdentifier(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\t\t\t\t_, _ = c.Cache.PutIfNotExist(key, mockContext)\n\t\t\t\tmockContext.EXPECT().Lock(ctx).Return(errors.New(\"lock-error\")).Times(1)\n\t\t\t},\n\t\t\tcacheHit:    false,\n\t\t\tcacheCtxNil: true,\n\t\t\terr:         errors.New(\"lock-error\"),\n\t\t},\n\t\t{\n\t\t\tname:     \"success - cache hit\",\n\t\t\tdomainID: constants.TestDomainID,\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.TestContext, mockContext *MockContext, c *cacheImpl) {\n\t\t\t\tmockContext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t\t\t\tmockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\t\t\t\tkey := definition.NewWorkflowIdentifier(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\t\t\t\t_, _ = c.Cache.PutIfNotExist(key, mockContext)\n\t\t\t\tmockContext.EXPECT().Lock(ctx).Return(nil).Times(1)\n\t\t\t},\n\t\t\tcacheHit: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.cache = NewCache(s.mockShard)\n\n\t\t\tmockContext := NewMockContext(s.controller)\n\n\t\t\ttt.mockSetup(s.mockShard, mockContext, s.cache.(*cacheImpl))\n\n\t\t\tcacheCtx, dbCtx, release, cacheHit, err := s.cache.GetAndCreateWorkflowExecution(ctx, tt.domainID, tt.execution)\n\n\t\t\ts.Equal(tt.cacheHit, cacheHit)\n\n\t\t\tif tt.cacheCtxNil {\n\t\t\t\ts.Nil(cacheCtx)\n\t\t\t} else {\n\t\t\t\ts.NotNil(cacheCtx)\n\t\t\t\ts.Equal(mockContext, cacheCtx)\n\t\t\t}\n\n\t\t\tif tt.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tt.err, err)\n\t\t\t\ts.Nil(dbCtx)\n\t\t\t\ts.Nil(release)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(dbCtx)\n\t\t\t\ts.NotNil(release)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *historyCacheSuite) TestGetOrCreateWorkflowExecutionWithTimeout() {\n\ts.cache = NewCache(s.mockShard)\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnewCtx := NewContext(constants.TestDomainID, workflowExecution, s.mockShard, s.mockShard.GetExecutionManager(), s.mockShard.GetLogger())\n\n\ts.mockShard.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).Times(1)\n\tkey := definition.NewWorkflowIdentifier(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\t_, _ = s.cache.(*cacheImpl).Cache.PutIfNotExist(key, newCtx)\n\n\t// getting the lock to guarantee that the context will time out\n\t_ = newCtx.Lock(context.Background())\n\tdefer newCtx.Unlock()\n\n\tcacheCtx, release, err := s.cache.GetOrCreateWorkflowExecutionWithTimeout(constants.TestDomainID, workflowExecution, 0)\n\n\ts.Nil(cacheCtx)\n\ts.Nil(release)\n\ts.Error(err)\n\ts.Equal(context.DeadlineExceeded, err)\n}\n"
  },
  {
    "path": "service/history/execution/checksum.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"fmt\"\n\t\"slices\"\n\n\tchecksumgen \"github.com/uber/cadence/.gen/go/checksum\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nconst (\n\tmutableStateChecksumPayloadV1 = 1\n)\n\nfunc generateMutableStateChecksum(ms MutableState) (checksum.Checksum, error) {\n\tpayload := newMutableStateChecksumPayload(ms)\n\tcsum, err := checksum.GenerateCRC32(payload, mutableStateChecksumPayloadV1)\n\tif err != nil {\n\t\treturn checksum.Checksum{}, err\n\t}\n\treturn csum, nil\n}\n\nfunc verifyMutableStateChecksum(\n\tms MutableState,\n\tcsum checksum.Checksum,\n) error {\n\tif csum.Version != mutableStateChecksumPayloadV1 {\n\t\treturn fmt.Errorf(\"invalid checksum payload version %v\", csum.Version)\n\t}\n\tpayload := newMutableStateChecksumPayload(ms)\n\treturn checksum.Verify(payload, csum)\n}\n\nfunc newMutableStateChecksumPayload(ms MutableState) *checksumgen.MutableStateChecksumPayload {\n\texecutionInfo := ms.GetExecutionInfo()\n\tpayload := &checksumgen.MutableStateChecksumPayload{\n\t\tCancelRequested:      common.BoolPtr(executionInfo.CancelRequested),\n\t\tState:                common.Int16Ptr(int16(executionInfo.State)),\n\t\tLastFirstEventID:     common.Int64Ptr(executionInfo.LastFirstEventID),\n\t\tNextEventID:          common.Int64Ptr(executionInfo.NextEventID),\n\t\tLastProcessedEventID: common.Int64Ptr(executionInfo.LastProcessedEvent),\n\t\tSignalCount:          common.Int64Ptr(int64(executionInfo.SignalCount)),\n\t\tDecisionAttempt:      common.Int32Ptr(int32(executionInfo.DecisionAttempt)),\n\t\tDecisionScheduledID:  common.Int64Ptr(executionInfo.DecisionScheduleID),\n\t\tDecisionStartedID:    common.Int64Ptr(executionInfo.DecisionStartedID),\n\t\tDecisionVersion:      common.Int64Ptr(executionInfo.DecisionVersion),\n\t\tStickyTaskListName:   common.StringPtr(executionInfo.StickyTaskList),\n\t}\n\n\tversionHistories := ms.GetVersionHistories()\n\tif versionHistories != nil {\n\t\tpayload.VersionHistories = thrift.FromVersionHistories(versionHistories.ToInternalType())\n\t}\n\n\t// for each of the pendingXXX ids below, sorting is needed to guarantee that\n\t// same serialized bytes can be generated during verification\n\tpendingTimerIDs := make([]int64, 0, len(ms.GetPendingTimerInfos()))\n\tfor _, ti := range ms.GetPendingTimerInfos() {\n\t\tpendingTimerIDs = append(pendingTimerIDs, ti.StartedID)\n\t}\n\tslices.Sort(pendingTimerIDs)\n\tpayload.PendingTimerStartedIDs = pendingTimerIDs\n\n\tpendingActivityIDs := make([]int64, 0, len(ms.GetPendingActivityInfos()))\n\tfor id := range ms.GetPendingActivityInfos() {\n\t\tpendingActivityIDs = append(pendingActivityIDs, id)\n\t}\n\tslices.Sort(pendingActivityIDs)\n\tpayload.PendingActivityScheduledIDs = pendingActivityIDs\n\n\tpendingChildIDs := make([]int64, 0, len(ms.GetPendingChildExecutionInfos()))\n\tfor id := range ms.GetPendingChildExecutionInfos() {\n\t\tpendingChildIDs = append(pendingChildIDs, id)\n\t}\n\tslices.Sort(pendingChildIDs)\n\tpayload.PendingChildInitiatedIDs = pendingChildIDs\n\n\tsignalIDs := make([]int64, 0, len(ms.GetPendingSignalExternalInfos()))\n\tfor id := range ms.GetPendingSignalExternalInfos() {\n\t\tsignalIDs = append(signalIDs, id)\n\t}\n\tslices.Sort(signalIDs)\n\tpayload.PendingSignalInitiatedIDs = signalIDs\n\n\trequestCancelIDs := make([]int64, 0, len(ms.GetPendingRequestCancelExternalInfos()))\n\tfor id := range ms.GetPendingRequestCancelExternalInfos() {\n\t\trequestCancelIDs = append(requestCancelIDs, id)\n\t}\n\tslices.Sort(requestCancelIDs)\n\tpayload.PendingReqCancelInitiatedIDs = requestCancelIDs\n\treturn payload\n}\n"
  },
  {
    "path": "service/history/execution/context.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination context_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/locks\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tdefaultRemoteCallTimeout = 30 * time.Second\n\tchecksumErrorRetryCount  = 3\n\tmaxLockDuration          = 1 * time.Second\n)\n\ntype conflictError struct {\n\tCause error\n}\n\nfunc (e *conflictError) Error() string {\n\treturn fmt.Sprintf(\"conditional update failed: %v\", e.Cause)\n}\n\nfunc (e *conflictError) Unwrap() error {\n\treturn e.Cause\n}\n\n// NewConflictError is only public because it used in workflow/util_test.go\n// TODO: refactor those tests\nfunc NewConflictError(_ *testing.T, cause error) error {\n\treturn &conflictError{cause}\n}\n\n// IsConflictError checks whether a conflict has occurred while updating a workflow execution\nfunc IsConflictError(err error) bool {\n\tvar e *conflictError\n\treturn errors.As(err, &e)\n}\n\ntype (\n\t// Context is the processing context for all operations on workflow execution\n\tContext interface {\n\t\tGetDomainName() string\n\t\tGetDomainID() string\n\t\tGetExecution() *types.WorkflowExecution\n\n\t\tGetWorkflowExecution() MutableState\n\t\tSetWorkflowExecution(mutableState MutableState)\n\t\tLoadWorkflowExecution(ctx context.Context) (MutableState, error)\n\t\tLoadWorkflowExecutionWithTaskVersion(ctx context.Context, incomingVersion int64) (MutableState, error)\n\t\tLoadExecutionStats(ctx context.Context) (*persistence.ExecutionStats, error)\n\t\tClear()\n\n\t\tLock(ctx context.Context) error\n\t\tUnlock()\n\n\t\tGetHistorySize() int64\n\t\tSetHistorySize(size int64)\n\n\t\tReapplyEvents(\n\t\t\teventBatches []*persistence.WorkflowEvents,\n\t\t) error\n\n\t\tPersistStartWorkflowBatchEvents(\n\t\t\tctx context.Context,\n\t\t\tworkflowEvents *persistence.WorkflowEvents,\n\t\t) (events.PersistedBlob, error)\n\t\tPersistNonStartWorkflowBatchEvents(\n\t\t\tctx context.Context,\n\t\t\tworkflowEvents *persistence.WorkflowEvents,\n\t\t) (events.PersistedBlob, error)\n\n\t\tCreateWorkflowExecution(\n\t\t\tctx context.Context,\n\t\t\tnewWorkflow *persistence.WorkflowSnapshot,\n\t\t\tpersistedHistory events.PersistedBlob,\n\t\t\tcreateMode persistence.CreateWorkflowMode,\n\t\t\tprevRunID string,\n\t\t\t// TODO(active-active): only passing prevLastWriteVersion might not be enough for active-active workflows, we may consider passing ActiveClusterSelectionPolicy as well\n\t\t\t// and include ActiveClusterSelectionPolicy in conditional update of current execution record\n\t\t\tprevLastWriteVersion int64,\n\t\t\tworkflowRequestMode persistence.CreateWorkflowRequestMode,\n\t\t) error\n\t\tConflictResolveWorkflowExecution(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tconflictResolveMode persistence.ConflictResolveWorkflowMode,\n\t\t\tresetMutableState MutableState,\n\t\t\tnewContext Context,\n\t\t\tnewMutableState MutableState,\n\t\t\tcurrentContext Context,\n\t\t\tcurrentMutableState MutableState,\n\t\t\tcurrentTransactionPolicy *TransactionPolicy,\n\t\t) error\n\t\tUpdateWorkflowExecutionAsActive(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t) error\n\t\tUpdateWorkflowExecutionWithNewAsActive(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tnewContext Context,\n\t\t\tnewMutableState MutableState,\n\t\t) error\n\t\tUpdateWorkflowExecutionAsPassive(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t) error\n\t\tUpdateWorkflowExecutionWithNewAsPassive(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tnewContext Context,\n\t\t\tnewMutableState MutableState,\n\t\t) error\n\t\tUpdateWorkflowExecutionWithNew(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tupdateMode persistence.UpdateWorkflowMode,\n\t\t\tnewContext Context,\n\t\t\tnewMutableState MutableState,\n\t\t\tcurrentWorkflowTransactionPolicy TransactionPolicy,\n\t\t\tnewWorkflowTransactionPolicy *TransactionPolicy,\n\t\t\tworkflowRequestMode persistence.CreateWorkflowRequestMode,\n\t\t) error\n\t\tUpdateWorkflowExecutionTasks(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t) error\n\n\t\tcache.Sizeable\n\t}\n)\n\ntype (\n\tcontextImpl struct {\n\t\tdomainID          string\n\t\tworkflowExecution types.WorkflowExecution\n\t\tshard             shard.Context\n\t\texecutionManager  persistence.ExecutionManager\n\t\tlogger            log.Logger\n\t\tmetricsClient     metrics.Client\n\n\t\tmutex           locks.Mutex\n\t\tlockTime        time.Time\n\t\tmaxLockDuration time.Duration\n\t\tmutableState    MutableState\n\t\tstats           *persistence.ExecutionStats\n\n\t\tappendHistoryNodesFn                  func(context.Context, string, types.WorkflowExecution, *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error)\n\t\tpersistStartWorkflowBatchEventsFn     func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tpersistNonStartWorkflowBatchEventsFn  func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tgetWorkflowExecutionFn                func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error)\n\t\tcreateWorkflowExecutionFn             func(context.Context, *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error)\n\t\tupdateWorkflowExecutionFn             func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)\n\t\tupdateWorkflowExecutionWithNewFn      func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\tnotifyTasksFromWorkflowSnapshotFn     func(*persistence.WorkflowSnapshot, events.PersistedBlobs, bool)\n\t\tnotifyTasksFromWorkflowMutationFn     func(*persistence.WorkflowMutation, events.PersistedBlobs, bool)\n\t\temitSessionUpdateStatsFn              func(string, *persistence.MutableStateUpdateSessionStats)\n\t\temitWorkflowHistoryStatsFn            func(string, int, int)\n\t\temitWorkflowCompletionStatsFn         func(string, string, string, string, string, *types.HistoryEvent)\n\t\tmergeContinueAsNewReplicationTasksFn  func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error\n\t\tupdateWorkflowExecutionEventReapplyFn func(persistence.UpdateWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error\n\t\tconflictResolveEventReapplyFn         func(persistence.ConflictResolveWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error\n\t\temitLargeWorkflowShardIDStatsFn       func(int64, int64, int64, int64)\n\t\temitWorkflowExecutionStatsFn          func(string, *persistence.MutableStateStats, int64)\n\t\tcreateMutableStateFn                  func(shard.Context, log.Logger, *cache.DomainCacheEntry) MutableState\n\t}\n)\n\nvar _ Context = (*contextImpl)(nil)\n\n// NewContext creates a new workflow execution context\nfunc NewContext(\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\tshard shard.Context,\n\texecutionManager persistence.ExecutionManager,\n\tlogger log.Logger,\n) Context {\n\tlogger = logger.WithTags(tag.WorkflowDomainID(domainID), tag.WorkflowID(execution.GetWorkflowID()), tag.WorkflowRunID(execution.GetRunID()))\n\tctx := &contextImpl{\n\t\tdomainID:          domainID,\n\t\tworkflowExecution: execution,\n\t\tshard:             shard,\n\t\texecutionManager:  executionManager,\n\t\tlogger:            logger,\n\t\tmetricsClient:     shard.GetMetricsClient(),\n\t\tmutex:             locks.NewMutex(),\n\t\tmaxLockDuration:   maxLockDuration,\n\t\tstats: &persistence.ExecutionStats{\n\t\t\tHistorySize: 0,\n\t\t},\n\n\t\tappendHistoryNodesFn: func(ctx context.Context, domainID string, workflowExecution types.WorkflowExecution, request *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\t\t\treturn appendHistoryV2EventsWithRetry(ctx, shard, common.CreatePersistenceRetryPolicy(), domainID, workflowExecution, request)\n\t\t},\n\t\tgetWorkflowExecutionFn: func(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\treturn getWorkflowExecutionWithRetry(ctx, shard, logger.Helper(), common.CreatePersistenceRetryPolicy(), request)\n\t\t},\n\t\tcreateWorkflowExecutionFn: func(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error) {\n\t\t\treturn createWorkflowExecutionWithRetry(ctx, shard, logger.Helper(), common.CreatePersistenceRetryPolicy(), request)\n\t\t},\n\t\tupdateWorkflowExecutionFn: func(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\t\t\treturn updateWorkflowExecutionWithRetry(ctx, shard, logger.Helper(), common.CreatePersistenceRetryPolicy(), request)\n\t\t},\n\t\tnotifyTasksFromWorkflowSnapshotFn: func(snapshot *persistence.WorkflowSnapshot, blobs events.PersistedBlobs, persistentError bool) {\n\t\t\tnotifyTasksFromWorkflowSnapshot(shard.GetEngine(), snapshot, blobs, persistentError)\n\t\t},\n\t\tnotifyTasksFromWorkflowMutationFn: func(snapshot *persistence.WorkflowMutation, blobs events.PersistedBlobs, persistentError bool) {\n\t\t\tnotifyTasksFromWorkflowMutation(shard.GetEngine(), snapshot, blobs, persistentError)\n\t\t},\n\t\temitSessionUpdateStatsFn: func(domainName string, stats *persistence.MutableStateUpdateSessionStats) {\n\t\t\temitSessionUpdateStats(shard.GetMetricsClient(), domainName, stats)\n\t\t},\n\t\temitWorkflowHistoryStatsFn: func(domainName string, historySize int, eventCount int) {\n\t\t\temitWorkflowHistoryStats(shard.GetMetricsClient(), domainName, historySize, eventCount)\n\t\t},\n\t\temitWorkflowCompletionStatsFn: func(domainName, workflowType, workflowID, runID, taskList string, event *types.HistoryEvent) {\n\t\t\temitWorkflowCompletionStats(shard.GetMetricsClient(), logger.Helper(), domainName, workflowType, workflowID, runID, taskList, event)\n\t\t},\n\t\temitWorkflowExecutionStatsFn: func(domainName string, stats *persistence.MutableStateStats, historySize int64) {\n\t\t\temitWorkflowExecutionStats(shard.GetMetricsClient(), domainName, stats, historySize)\n\t\t},\n\t\tmergeContinueAsNewReplicationTasksFn: func(updateMode persistence.UpdateWorkflowMode, mutation *persistence.WorkflowMutation, snapshot *persistence.WorkflowSnapshot) error {\n\t\t\treturn mergeContinueAsNewReplicationTasks(logger, updateMode, mutation, snapshot)\n\t\t},\n\t\tcreateMutableStateFn: NewMutableStateBuilder,\n\t}\n\tctx.persistStartWorkflowBatchEventsFn = ctx.PersistStartWorkflowBatchEvents\n\tctx.persistNonStartWorkflowBatchEventsFn = ctx.PersistNonStartWorkflowBatchEvents\n\tctx.updateWorkflowExecutionEventReapplyFn = ctx.updateWorkflowExecutionEventReapply\n\tctx.conflictResolveEventReapplyFn = ctx.conflictResolveEventReapply\n\tctx.emitLargeWorkflowShardIDStatsFn = ctx.emitLargeWorkflowShardIDStats\n\tctx.updateWorkflowExecutionWithNewFn = ctx.UpdateWorkflowExecutionWithNew\n\treturn ctx\n}\n\nfunc (c *contextImpl) Lock(ctx context.Context) error {\n\terr := c.mutex.Lock(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tc.lockTime = time.Now()\n\treturn nil\n}\n\nfunc (c *contextImpl) Unlock() {\n\tdefer c.mutex.Unlock()\n\n\tif c.lockTime.IsZero() { // skip logging if the lock is never acquired\n\t\treturn\n\t}\n\telapsed := time.Since(c.lockTime)\n\tc.metricsClient.RecordTimer(metrics.WorkflowContextScope, metrics.WorkflowContextLockLatency, elapsed)\n\tif elapsed > c.maxLockDuration {\n\t\tc.maxLockDuration = elapsed\n\t\tc.logger.Info(\"workflow context lock is released. this is logged only when it's longer than maxLockDuration\", tag.WorkflowContextLockLatency(elapsed))\n\t}\n}\n\nfunc (c *contextImpl) Clear() {\n\tc.metricsClient.IncCounter(metrics.WorkflowContextScope, metrics.WorkflowContextCleared)\n\tc.mutableState = nil\n\tc.stats = &persistence.ExecutionStats{\n\t\tHistorySize: 0,\n\t}\n}\n\nfunc (c *contextImpl) GetDomainID() string {\n\treturn c.domainID\n}\n\nfunc (c *contextImpl) GetExecution() *types.WorkflowExecution {\n\treturn &c.workflowExecution\n}\n\nfunc (c *contextImpl) GetDomainName() string {\n\tdomainName, err := c.shard.GetDomainCache().GetDomainName(c.domainID)\n\tif err != nil {\n\t\treturn \"\"\n\t}\n\treturn domainName\n}\n\nfunc (c *contextImpl) GetHistorySize() int64 {\n\treturn c.stats.HistorySize\n}\n\nfunc (c *contextImpl) SetHistorySize(size int64) {\n\tc.stats.HistorySize = size\n\tif c.mutableState != nil {\n\t\tc.mutableState.SetHistorySize(size)\n\t}\n}\n\nfunc (c *contextImpl) LoadExecutionStats(\n\tctx context.Context,\n) (*persistence.ExecutionStats, error) {\n\t_, err := c.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn c.stats, nil\n}\n\nfunc isChecksumError(err error) bool {\n\tif err == nil {\n\t\treturn false\n\t}\n\treturn strings.Contains(err.Error(), \"checksum mismatch error\")\n}\n\nfunc (c *contextImpl) LoadWorkflowExecutionWithTaskVersion(\n\tctx context.Context,\n\tincomingVersion int64,\n) (MutableState, error) {\n\tdomainEntry, err := c.shard.GetDomainCache().GetDomainByID(c.domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif c.mutableState == nil {\n\t\tvar response *persistence.GetWorkflowExecutionResponse\n\t\tfor i := 0; i < checksumErrorRetryCount; i++ {\n\t\t\tresponse, err = c.getWorkflowExecutionFn(ctx, &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tDomainID:   c.domainID,\n\t\t\t\tExecution:  c.workflowExecution,\n\t\t\t\tDomainName: domainEntry.GetInfo().Name,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tc.mutableState = c.createMutableStateFn(c.shard, c.logger, domainEntry)\n\t\t\terr = c.mutableState.Load(ctx, response.State)\n\t\t\tif err == nil {\n\t\t\t\tbreak\n\t\t\t} else if !isChecksumError(err) {\n\t\t\t\tc.logger.Error(\"failed to load mutable state\", tag.Error(err))\n\t\t\t\tbreak\n\t\t\t}\n\t\t\t// backoff before retry\n\t\t\tc.shard.GetTimeSource().Sleep(time.Millisecond * 100)\n\t\t}\n\t\tif isChecksumError(err) {\n\t\t\tc.metricsClient.IncCounter(metrics.WorkflowContextScope, metrics.StaleMutableStateCounter)\n\t\t\tc.logger.Error(\"encounter stale mutable state after retry\", tag.Error(err))\n\t\t}\n\n\t\tc.stats = response.State.ExecutionStats\n\n\t\t// finally emit execution and session stats\n\t\tc.emitWorkflowExecutionStatsFn(domainEntry.GetInfo().Name, response.MutableStateStats, c.stats.HistorySize)\n\t}\n\tflushBeforeReady, err := c.mutableState.StartTransaction(ctx, domainEntry, incomingVersion)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !flushBeforeReady {\n\t\treturn c.mutableState, nil\n\t}\n\tc.logger.Debug(\"LoadWorkflowExecutionWithTaskVersion calling UpdateWorkflowExecutionAsActive\", tag.WorkflowID(c.workflowExecution.GetWorkflowID()))\n\tif err = c.UpdateWorkflowExecutionAsActive(ctx, c.shard.GetTimeSource().Now()); err != nil {\n\t\treturn nil, err\n\t}\n\tflushBeforeReady, err = c.mutableState.StartTransaction(ctx, domainEntry, incomingVersion)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif flushBeforeReady {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"workflowExecutionContext counter flushBeforeReady status after loading mutable state from DB\",\n\t\t}\n\t}\n\treturn c.mutableState, nil\n}\n\n// GetWorkflowExecution should only be used in tests\nfunc (c *contextImpl) GetWorkflowExecution() MutableState {\n\treturn c.mutableState\n}\n\n// SetWorkflowExecution should only be used in tests\nfunc (c *contextImpl) SetWorkflowExecution(mutableState MutableState) {\n\tc.mutableState = mutableState\n}\n\nfunc (c *contextImpl) LoadWorkflowExecution(\n\tctx context.Context,\n) (MutableState, error) {\n\t// Use empty version to skip incoming task version validation\n\treturn c.LoadWorkflowExecutionWithTaskVersion(ctx, constants.EmptyVersion)\n}\n\nfunc (c *contextImpl) CreateWorkflowExecution(\n\tctx context.Context,\n\tnewWorkflow *persistence.WorkflowSnapshot,\n\tpersistedHistory events.PersistedBlob,\n\tcreateMode persistence.CreateWorkflowMode,\n\tprevRunID string,\n\tprevLastWriteVersion int64,\n\tworkflowRequestMode persistence.CreateWorkflowRequestMode,\n) (retError error) {\n\n\tdefer func() {\n\t\tif retError != nil {\n\t\t\tc.Clear()\n\t\t}\n\t}()\n\tdomain, errorDomainName := c.shard.GetDomainCache().GetDomainName(c.domainID)\n\tif errorDomainName != nil {\n\t\treturn errorDomainName\n\t}\n\terr := validateWorkflowRequestsAndMode(newWorkflow.WorkflowRequests, workflowRequestMode)\n\tif err != nil {\n\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\treturn err\n\t\t}\n\t\tc.logger.Error(\"workflow requests and mode validation error\", tag.Error(err))\n\t}\n\tcreateRequest := &persistence.CreateWorkflowExecutionRequest{\n\t\t// workflow create mode & prev run ID & version\n\t\tMode:                     createMode,\n\t\tPreviousRunID:            prevRunID,\n\t\tPreviousLastWriteVersion: prevLastWriteVersion,\n\t\tNewWorkflowSnapshot:      *newWorkflow,\n\t\tWorkflowRequestMode:      workflowRequestMode,\n\t\tDomainName:               domain,\n\t}\n\n\thistorySize := int64(len(persistedHistory.Data))\n\thistorySize += c.GetHistorySize()\n\tc.SetHistorySize(historySize)\n\tcreateRequest.NewWorkflowSnapshot.ExecutionStats = &persistence.ExecutionStats{\n\t\tHistorySize: historySize,\n\t}\n\n\tresp, err := c.createWorkflowExecutionFn(ctx, createRequest)\n\tif err != nil {\n\t\tif isOperationPossiblySuccessfulError(err) {\n\t\t\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, events.PersistedBlobs{persistedHistory}, true)\n\t\t}\n\t\treturn err\n\t}\n\n\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, events.PersistedBlobs{persistedHistory}, false)\n\n\t// finally emit session stats\n\tc.emitSessionUpdateStatsFn(domain, resp.MutableStateUpdateSessionStats)\n\n\treturn nil\n}\n\nfunc (c *contextImpl) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\tnow time.Time,\n\tconflictResolveMode persistence.ConflictResolveWorkflowMode,\n\tresetMutableState MutableState,\n\tnewContext Context,\n\tnewMutableState MutableState,\n\tcurrentContext Context,\n\tcurrentMutableState MutableState,\n\tcurrentTransactionPolicy *TransactionPolicy,\n) (retError error) {\n\tdefer func() {\n\t\tif retError != nil {\n\t\t\tc.Clear()\n\t\t}\n\t}()\n\n\tresetWorkflow, resetWorkflowEventsSeq, err := resetMutableState.CloseTransactionAsSnapshot(now, TransactionPolicyPassive)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, errorDomainName := c.shard.GetDomainCache().GetDomainName(c.domainID)\n\tif errorDomainName != nil {\n\t\treturn errorDomainName\n\t}\n\tvar persistedBlobs events.PersistedBlobs\n\tresetHistorySize := c.GetHistorySize()\n\tfor _, workflowEvents := range resetWorkflowEventsSeq {\n\t\tblob, err := c.persistNonStartWorkflowBatchEventsFn(ctx, workflowEvents)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tresetHistorySize += int64(len(blob.Data))\n\t\tpersistedBlobs = append(persistedBlobs, blob)\n\t}\n\tc.SetHistorySize(resetHistorySize)\n\tresetWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\tHistorySize: resetHistorySize,\n\t}\n\n\tvar newWorkflow *persistence.WorkflowSnapshot\n\tvar newWorkflowEventsSeq []*persistence.WorkflowEvents\n\tif newContext != nil && newMutableState != nil {\n\t\tdefer func() {\n\t\t\tif retError != nil {\n\t\t\t\tnewContext.Clear()\n\t\t\t}\n\t\t}()\n\t\tnewWorkflow, newWorkflowEventsSeq, err = newMutableState.CloseTransactionAsSnapshot(now, TransactionPolicyPassive)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif len(resetWorkflow.WorkflowRequests) != 0 && len(newWorkflow.WorkflowRequests) != 0 {\n\t\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\t\treturn &types.InternalServiceError{Message: \"workflow requests are only expected to be generated from either reset workflow or continue-as-new workflow for ConflictResolveWorkflowExecution\"}\n\t\t\t}\n\t\t\tc.logger.Error(\"workflow requests are only expected to be generated from either reset workflow or continue-as-new workflow for ConflictResolveWorkflowExecution\", tag.Number(int64(len(resetWorkflow.WorkflowRequests))), tag.NextNumber(int64(len(newWorkflow.WorkflowRequests))))\n\t\t}\n\t\tnewWorkflowSizeSize := newContext.GetHistorySize()\n\t\tstartEvents := newWorkflowEventsSeq[0]\n\t\tblob, err := c.persistStartWorkflowBatchEventsFn(ctx, startEvents)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tnewWorkflowSizeSize += int64(len(blob.Data))\n\t\tnewContext.SetHistorySize(newWorkflowSizeSize)\n\t\tnewWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\t\tHistorySize: newWorkflowSizeSize,\n\t\t}\n\t\tpersistedBlobs = append(persistedBlobs, blob)\n\t}\n\n\tvar currentWorkflow *persistence.WorkflowMutation\n\tvar currentWorkflowEventsSeq []*persistence.WorkflowEvents\n\tif currentContext != nil && currentMutableState != nil && currentTransactionPolicy != nil {\n\t\tdefer func() {\n\t\t\tif retError != nil {\n\t\t\t\tcurrentContext.Clear()\n\t\t\t}\n\t\t}()\n\t\tc.logger.Debug(\"ConflictResolveWorkflowExecution calling CloseTransactionAsMutation\", tag.WorkflowID(c.workflowExecution.GetWorkflowID()), tag.Dynamic(\"policy\", *currentTransactionPolicy))\n\t\tcurrentWorkflow, currentWorkflowEventsSeq, err = currentMutableState.CloseTransactionAsMutation(now, *currentTransactionPolicy)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif len(currentWorkflow.WorkflowRequests) != 0 {\n\t\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\t\treturn &types.InternalServiceError{Message: \"workflow requests are not expected from current workflow for ConflictResolveWorkflowExecution\"}\n\t\t\t}\n\t\t\tc.logger.Error(\"workflow requests are not expected from current workflow for ConflictResolveWorkflowExecution\", tag.Counter(len(currentWorkflow.WorkflowRequests)))\n\t\t}\n\t\tcurrentWorkflowSize := currentContext.GetHistorySize()\n\t\tfor _, workflowEvents := range currentWorkflowEventsSeq {\n\t\t\tblob, err := c.persistNonStartWorkflowBatchEventsFn(ctx, workflowEvents)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tcurrentWorkflowSize += int64(len(blob.Data))\n\t\t\tpersistedBlobs = append(persistedBlobs, blob)\n\t\t}\n\t\tcurrentContext.SetHistorySize(currentWorkflowSize)\n\t\tcurrentWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\t\tHistorySize: currentWorkflowSize,\n\t\t}\n\t}\n\n\tif err := c.conflictResolveEventReapplyFn(\n\t\tconflictResolveMode,\n\t\tresetWorkflowEventsSeq,\n\t\tnewWorkflowEventsSeq,\n\t\t// current workflow events will not participate in the events reapplication\n\t); err != nil {\n\t\treturn err\n\t}\n\tresp, err := c.shard.ConflictResolveWorkflowExecution(ctx, &persistence.ConflictResolveWorkflowExecutionRequest{\n\t\t// RangeID , this is set by shard context\n\t\tMode:                    conflictResolveMode,\n\t\tResetWorkflowSnapshot:   *resetWorkflow,\n\t\tNewWorkflowSnapshot:     newWorkflow,\n\t\tCurrentWorkflowMutation: currentWorkflow,\n\t\tWorkflowRequestMode:     persistence.CreateWorkflowRequestModeReplicated,\n\t\t// Encoding, this is set by shard context\n\t\tDomainName: domain,\n\t})\n\tif err != nil {\n\t\tif isOperationPossiblySuccessfulError(err) {\n\t\t\tc.notifyTasksFromWorkflowSnapshotFn(resetWorkflow, persistedBlobs, true)\n\t\t\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, persistedBlobs, true)\n\t\t\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, persistedBlobs, true)\n\t\t}\n\t\treturn err\n\t}\n\n\tworkflowState, workflowCloseState := resetMutableState.GetWorkflowStateCloseStatus()\n\t// Current branch changed and notify the watchers\n\tc.shard.GetEngine().NotifyNewHistoryEvent(events.NewNotification(\n\t\tc.domainID,\n\t\t&c.workflowExecution,\n\t\tresetMutableState.GetLastFirstEventID(),\n\t\tresetMutableState.GetNextEventID(),\n\t\tresetMutableState.GetPreviousStartedEventID(),\n\t\tworkflowState,\n\t\tworkflowCloseState,\n\t\tresetMutableState.GetVersionHistories().Duplicate(),\n\t))\n\n\tc.notifyTasksFromWorkflowSnapshotFn(resetWorkflow, persistedBlobs, false)\n\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, persistedBlobs, false)\n\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, persistedBlobs, false)\n\n\t// finally emit session stats\n\tc.emitWorkflowHistoryStatsFn(domain, int(c.stats.HistorySize), int(resetMutableState.GetNextEventID()-1))\n\tc.emitSessionUpdateStatsFn(domain, resp.MutableStateUpdateSessionStats)\n\t// emit workflow completion stats if any\n\tif resetWorkflow.ExecutionInfo.State == persistence.WorkflowStateCompleted {\n\t\tif event, err := resetMutableState.GetCompletionEvent(ctx); err == nil {\n\t\t\tworkflowType := resetWorkflow.ExecutionInfo.WorkflowTypeName\n\t\t\ttaskList := resetWorkflow.ExecutionInfo.TaskList\n\t\t\tc.emitWorkflowCompletionStatsFn(domain, workflowType, c.workflowExecution.GetWorkflowID(), c.workflowExecution.GetRunID(), taskList, event)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionAsActive(\n\tctx context.Context,\n\tnow time.Time,\n) error {\n\tc.logger.Debug(\"UpdateWorkflowExecutionAsActive calling UpdateWorkflowExecutionWithNew\",\n\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()),\n\t\ttag.Dynamic(\"current policy\", TransactionPolicyActive),\n\t\ttag.Dynamic(\"new policy\", nil),\n\t)\n\treturn c.updateWorkflowExecutionWithNewFn(ctx, now, persistence.UpdateWorkflowModeUpdateCurrent, nil, nil, TransactionPolicyActive, nil, persistence.CreateWorkflowRequestModeNew)\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionWithNewAsActive(\n\tctx context.Context,\n\tnow time.Time,\n\tnewContext Context,\n\tnewMutableState MutableState,\n) error {\n\tc.logger.Debug(\"UpdateWorkflowExecutionWithNewAsActive calling UpdateWorkflowExecutionWithNew\",\n\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()),\n\t\ttag.Dynamic(\"current policy\", TransactionPolicyActive),\n\t\ttag.Dynamic(\"new policy\", TransactionPolicyActive),\n\t)\n\treturn c.updateWorkflowExecutionWithNewFn(ctx, now, persistence.UpdateWorkflowModeUpdateCurrent, newContext, newMutableState, TransactionPolicyActive, TransactionPolicyActive.Ptr(), persistence.CreateWorkflowRequestModeNew)\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionAsPassive(\n\tctx context.Context,\n\tnow time.Time,\n) error {\n\tc.logger.Debug(\"UpdateWorkflowExecutionAsPassive calling UpdateWorkflowExecutionWithNew\",\n\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()),\n\t\ttag.Dynamic(\"current policy\", TransactionPolicyPassive),\n\t\ttag.Dynamic(\"new policy\", nil),\n\t)\n\treturn c.updateWorkflowExecutionWithNewFn(ctx, now, persistence.UpdateWorkflowModeUpdateCurrent, nil, nil, TransactionPolicyPassive, nil, persistence.CreateWorkflowRequestModeReplicated)\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionWithNewAsPassive(\n\tctx context.Context,\n\tnow time.Time,\n\tnewContext Context,\n\tnewMutableState MutableState,\n) error {\n\tc.logger.Debug(\"UpdateWorkflowExecutionWithNewAsPassive calling UpdateWorkflowExecutionWithNew\",\n\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()),\n\t\ttag.Dynamic(\"current policy\", TransactionPolicyPassive),\n\t\ttag.Dynamic(\"new policy\", TransactionPolicyPassive),\n\t)\n\treturn c.updateWorkflowExecutionWithNewFn(ctx, now, persistence.UpdateWorkflowModeUpdateCurrent, newContext, newMutableState, TransactionPolicyPassive, TransactionPolicyPassive.Ptr(), persistence.CreateWorkflowRequestModeReplicated)\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionTasks(\n\tctx context.Context,\n\tnow time.Time,\n) (retError error) {\n\tdefer func() {\n\t\tif retError != nil {\n\t\t\tc.Clear()\n\t\t}\n\t}()\n\n\tc.logger.Debug(\"UpdateWorkflowExecutionTask calling CloseTransactionAsMutation\", tag.WorkflowID(c.workflowExecution.GetWorkflowID()))\n\tcurrentWorkflow, currentWorkflowEventsSeq, err := c.mutableState.CloseTransactionAsMutation(now, TransactionPolicyPassive)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif len(currentWorkflowEventsSeq) != 0 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"UpdateWorkflowExecutionTask can only be used for persisting new workflow tasks, but found new history events\",\n\t\t}\n\t}\n\tdomainName, errorDomainName := c.shard.GetDomainCache().GetDomainName(c.domainID)\n\tif errorDomainName != nil {\n\t\treturn errorDomainName\n\t}\n\tif len(currentWorkflow.WorkflowRequests) != 0 {\n\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domainName) {\n\t\t\treturn &types.InternalServiceError{Message: \"UpdateWorkflowExecutionTask can only be used for persisting new workflow tasks, but found new workflow requests\"}\n\t\t}\n\t\tc.logger.Error(\"UpdateWorkflowExecutionTask can only be used for persisting new workflow tasks, but found new workflow requests\", tag.Counter(len(currentWorkflow.WorkflowRequests)))\n\t}\n\tcurrentWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\tHistorySize: c.GetHistorySize(),\n\t}\n\tresp, err := c.updateWorkflowExecutionFn(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\t// RangeID , this is set by shard context\n\t\tMode:                   persistence.UpdateWorkflowModeIgnoreCurrent,\n\t\tUpdateWorkflowMutation: *currentWorkflow,\n\t\t// Encoding, this is set by shard context\n\t\tDomainName: domainName,\n\t})\n\tif err != nil {\n\t\tif isOperationPossiblySuccessfulError(err) {\n\t\t\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, nil, true)\n\t\t}\n\t\treturn err\n\t}\n\t// notify current workflow tasks\n\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, nil, false)\n\tc.emitSessionUpdateStatsFn(domainName, resp.MutableStateUpdateSessionStats)\n\treturn nil\n}\n\nfunc (c *contextImpl) UpdateWorkflowExecutionWithNew(\n\tctx context.Context,\n\tnow time.Time,\n\tupdateMode persistence.UpdateWorkflowMode,\n\tnewContext Context,\n\tnewMutableState MutableState,\n\tcurrentWorkflowTransactionPolicy TransactionPolicy,\n\tnewWorkflowTransactionPolicy *TransactionPolicy,\n\tworkflowRequestMode persistence.CreateWorkflowRequestMode,\n) (retError error) {\n\tdefer func() {\n\t\tif retError != nil {\n\t\t\tc.Clear()\n\t\t}\n\t}()\n\n\tc.logger.Debug(\"UpdateWorkflowExecutionWithNew calling CloseTransactionAsMutation\", tag.WorkflowID(c.workflowExecution.GetWorkflowID()), tag.Dynamic(\"policy\", currentWorkflowTransactionPolicy))\n\n\tcurrentWorkflow, currentWorkflowEventsSeq, err := c.mutableState.CloseTransactionAsMutation(now, currentWorkflowTransactionPolicy)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, errorDomainName := c.shard.GetDomainCache().GetDomainName(c.domainID)\n\tif errorDomainName != nil {\n\t\treturn errorDomainName\n\t}\n\terr = validateWorkflowRequestsAndMode(currentWorkflow.WorkflowRequests, workflowRequestMode)\n\tif err != nil {\n\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\treturn err\n\t\t}\n\t\tc.logger.Error(\"workflow requests and mode validation error\", tag.Error(err))\n\t}\n\tvar persistedBlobs events.PersistedBlobs\n\tcurrentWorkflowSize := c.GetHistorySize()\n\toldWorkflowSize := currentWorkflowSize\n\tcurrentWorkflowHistoryCount := c.mutableState.GetNextEventID() - 1\n\toldWorkflowHistoryCount := currentWorkflowHistoryCount\n\tfor _, workflowEvents := range currentWorkflowEventsSeq {\n\t\tblob, err := c.persistNonStartWorkflowBatchEventsFn(ctx, workflowEvents)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tcurrentWorkflowHistoryCount += int64(len(workflowEvents.Events))\n\t\tcurrentWorkflowSize += int64(len(blob.Data))\n\t\tpersistedBlobs = append(persistedBlobs, blob)\n\t}\n\tc.SetHistorySize(currentWorkflowSize)\n\tcurrentWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\tHistorySize: currentWorkflowSize,\n\t}\n\n\tvar newWorkflow *persistence.WorkflowSnapshot\n\tvar newWorkflowEventsSeq []*persistence.WorkflowEvents\n\tif newContext != nil && newMutableState != nil && newWorkflowTransactionPolicy != nil {\n\t\tdefer func() {\n\t\t\tif retError != nil {\n\t\t\t\tnewContext.Clear()\n\t\t\t}\n\t\t}()\n\t\tnewWorkflow, newWorkflowEventsSeq, err = newMutableState.CloseTransactionAsSnapshot(\n\t\t\tnow,\n\t\t\t*newWorkflowTransactionPolicy,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif len(newWorkflow.WorkflowRequests) != 0 && len(currentWorkflow.WorkflowRequests) != 0 {\n\t\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\t\treturn &types.InternalServiceError{Message: \"workflow requests are only expected to be generated from one workflow for UpdateWorkflowExecution\"}\n\t\t\t}\n\t\t\tc.logger.Error(\"workflow requests are only expected to be generated from one workflow for UpdateWorkflowExecution\", tag.Number(int64(len(currentWorkflow.WorkflowRequests))), tag.NextNumber(int64(len(newWorkflow.WorkflowRequests))))\n\t\t}\n\n\t\terr := validateWorkflowRequestsAndMode(newWorkflow.WorkflowRequests, workflowRequestMode)\n\t\tif err != nil {\n\t\t\tif c.shard.GetConfig().EnableStrongIdempotencySanityCheck(domain) {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tc.logger.Error(\"workflow requests and mode validation error\", tag.Error(err))\n\t\t}\n\t\tnewWorkflowSizeSize := newContext.GetHistorySize()\n\t\tstartEvents := newWorkflowEventsSeq[0]\n\t\tfirstEventID := startEvents.Events[0].ID\n\t\tvar blob events.PersistedBlob\n\t\tif firstEventID == constants.FirstEventID {\n\t\t\tblob, err = c.persistStartWorkflowBatchEventsFn(ctx, startEvents)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\t// NOTE: This is the case for reset workflow, reset workflow already inserted a branch record\n\t\t\tblob, err = c.persistNonStartWorkflowBatchEventsFn(ctx, startEvents)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tpersistedBlobs = append(persistedBlobs, blob)\n\t\tnewWorkflowSizeSize += int64(len(blob.Data))\n\t\tnewContext.SetHistorySize(newWorkflowSizeSize)\n\t\tnewWorkflow.ExecutionStats = &persistence.ExecutionStats{\n\t\t\tHistorySize: newWorkflowSizeSize,\n\t\t}\n\t}\n\n\tif err := c.mergeContinueAsNewReplicationTasksFn(updateMode, currentWorkflow, newWorkflow); err != nil {\n\t\treturn err\n\t}\n\n\tif err := c.updateWorkflowExecutionEventReapplyFn(updateMode, currentWorkflowEventsSeq, newWorkflowEventsSeq); err != nil {\n\t\treturn err\n\t}\n\tresp, err := c.updateWorkflowExecutionFn(ctx, &persistence.UpdateWorkflowExecutionRequest{\n\t\t// RangeID , this is set by shard context\n\t\tMode:                   updateMode,\n\t\tUpdateWorkflowMutation: *currentWorkflow,\n\t\tNewWorkflowSnapshot:    newWorkflow,\n\t\tWorkflowRequestMode:    workflowRequestMode,\n\t\t// Encoding, this is set by shard context\n\t\tDomainName: domain,\n\t})\n\tif err != nil {\n\t\tif isOperationPossiblySuccessfulError(err) {\n\t\t\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, persistedBlobs, true)\n\t\t\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, persistedBlobs, true)\n\t\t}\n\t\treturn err\n\t}\n\n\t// for any change in the workflow, send a event\n\tworkflowState, workflowCloseState := c.mutableState.GetWorkflowStateCloseStatus()\n\tc.shard.GetEngine().NotifyNewHistoryEvent(events.NewNotification(\n\t\tc.domainID,\n\t\t&c.workflowExecution,\n\t\tc.mutableState.GetLastFirstEventID(),\n\t\tc.mutableState.GetNextEventID(),\n\t\tc.mutableState.GetPreviousStartedEventID(),\n\t\tworkflowState,\n\t\tworkflowCloseState,\n\t\tc.mutableState.GetVersionHistories().Duplicate(),\n\t))\n\n\t// notify current workflow tasks\n\tc.notifyTasksFromWorkflowMutationFn(currentWorkflow, persistedBlobs, false)\n\t// notify new workflow tasks\n\tc.notifyTasksFromWorkflowSnapshotFn(newWorkflow, persistedBlobs, false)\n\n\t// finally emit session stats\n\tc.emitWorkflowHistoryStatsFn(domain, int(c.stats.HistorySize), int(c.mutableState.GetNextEventID()-1))\n\tc.emitSessionUpdateStatsFn(domain, resp.MutableStateUpdateSessionStats)\n\tc.emitLargeWorkflowShardIDStatsFn(currentWorkflowSize-oldWorkflowSize, oldWorkflowHistoryCount, oldWorkflowSize, currentWorkflowHistoryCount)\n\t// emit workflow completion stats if any\n\tif currentWorkflow.ExecutionInfo.State == persistence.WorkflowStateCompleted {\n\t\tif event, err := c.mutableState.GetCompletionEvent(ctx); err == nil {\n\t\t\tworkflowType := currentWorkflow.ExecutionInfo.WorkflowTypeName\n\t\t\ttaskList := currentWorkflow.ExecutionInfo.TaskList\n\t\t\tc.emitWorkflowCompletionStatsFn(domain, workflowType, c.workflowExecution.GetWorkflowID(), c.workflowExecution.GetRunID(), taskList, event)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc notifyTasksFromWorkflowSnapshot(\n\tengine engine.Engine,\n\tworkflowSnapShot *persistence.WorkflowSnapshot,\n\thistory events.PersistedBlobs,\n\tpersistenceError bool,\n) {\n\tif workflowSnapShot == nil {\n\t\treturn\n\t}\n\n\tnotifyTasks(\n\t\tengine,\n\t\tworkflowSnapShot.ExecutionInfo,\n\t\tworkflowSnapShot.VersionHistories,\n\t\tworkflowSnapShot.ActivityInfos,\n\t\tworkflowSnapShot.TasksByCategory,\n\t\thistory,\n\t\tpersistenceError,\n\t)\n}\n\nfunc notifyTasksFromWorkflowMutation(\n\tengine engine.Engine,\n\tworkflowMutation *persistence.WorkflowMutation,\n\thistory events.PersistedBlobs,\n\tpersistenceError bool,\n) {\n\tif workflowMutation == nil {\n\t\treturn\n\t}\n\n\tnotifyTasks(\n\t\tengine,\n\t\tworkflowMutation.ExecutionInfo,\n\t\tworkflowMutation.VersionHistories,\n\t\tworkflowMutation.UpsertActivityInfos,\n\t\tworkflowMutation.TasksByCategory,\n\t\thistory,\n\t\tpersistenceError,\n\t)\n}\n\nfunc activityInfosToMap(ais []*persistence.ActivityInfo) map[int64]*persistence.ActivityInfo {\n\tm := make(map[int64]*persistence.ActivityInfo, len(ais))\n\tfor _, ai := range ais {\n\t\tm[ai.ScheduleID] = ai\n\t}\n\treturn m\n}\n\nfunc notifyTasks(\n\tengine engine.Engine,\n\texecutionInfo *persistence.WorkflowExecutionInfo,\n\tversionHistories *persistence.VersionHistories,\n\tactivities []*persistence.ActivityInfo,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]persistence.Task,\n\thistory events.PersistedBlobs,\n\tpersistenceError bool,\n) {\n\ttransferTaskInfo := &hcommon.NotifyTaskInfo{\n\t\tExecutionInfo:    executionInfo,\n\t\tTasks:            tasksByCategory[persistence.HistoryTaskCategoryTransfer],\n\t\tPersistenceError: persistenceError,\n\t}\n\ttimerTaskInfo := &hcommon.NotifyTaskInfo{\n\t\tExecutionInfo:    executionInfo,\n\t\tTasks:            tasksByCategory[persistence.HistoryTaskCategoryTimer],\n\t\tPersistenceError: persistenceError,\n\t}\n\treplicationTaskInfo := &hcommon.NotifyTaskInfo{\n\t\tExecutionInfo:    executionInfo,\n\t\tTasks:            tasksByCategory[persistence.HistoryTaskCategoryReplication],\n\t\tVersionHistories: versionHistories,\n\t\tActivities:       activityInfosToMap(activities),\n\t\tHistory:          history,\n\t\tPersistenceError: persistenceError,\n\t}\n\n\t// TODO: unify these methods\n\tengine.NotifyNewTransferTasks(transferTaskInfo)\n\tengine.NotifyNewTimerTasks(timerTaskInfo)\n\tengine.NotifyNewReplicationTasks(replicationTaskInfo)\n}\n\nfunc mergeContinueAsNewReplicationTasks(\n\tlogger log.Logger,\n\tupdateMode persistence.UpdateWorkflowMode,\n\tcurrentWorkflowMutation *persistence.WorkflowMutation,\n\tnewWorkflowSnapshot *persistence.WorkflowSnapshot,\n) error {\n\n\tif currentWorkflowMutation.ExecutionInfo.CloseStatus != persistence.WorkflowCloseStatusContinuedAsNew {\n\t\treturn nil\n\t} else if updateMode == persistence.UpdateWorkflowModeBypassCurrent && newWorkflowSnapshot == nil {\n\t\t// update current workflow as zombie & continue as new without new zombie workflow\n\t\t// this case can be valid if new workflow is already created by resend\n\t\treturn nil\n\t}\n\n\t// current workflow is doing continue as new\n\n\t// it is possible that continue as new is done as part of passive logic\n\tif len(currentWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryReplication]) == 0 {\n\t\tlogger.Debug(\"mergeContinueAsNewReplicationTasks: no replication task\",\n\t\t\ttag.WorkflowDomainID(currentWorkflowMutation.ExecutionInfo.DomainID),\n\t\t\ttag.WorkflowID(currentWorkflowMutation.ExecutionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(currentWorkflowMutation.ExecutionInfo.RunID),\n\t\t)\n\t\treturn nil\n\t}\n\n\tif newWorkflowSnapshot == nil || len(newWorkflowSnapshot.TasksByCategory[persistence.HistoryTaskCategoryReplication]) != 1 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"unable to find replication task from new workflow for continue as new replication\",\n\t\t}\n\t}\n\n\t// merge the new run first event batch replication task\n\t// to current event batch replication task\n\tnewRunTask := newWorkflowSnapshot.TasksByCategory[persistence.HistoryTaskCategoryReplication][0].(*persistence.HistoryReplicationTask)\n\tnewWorkflowSnapshot.TasksByCategory[persistence.HistoryTaskCategoryReplication] = nil\n\n\tnewRunBranchToken := newRunTask.BranchToken\n\ttaskUpdated := false\n\tfor _, replicationTask := range currentWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryReplication] {\n\t\tif task, ok := replicationTask.(*persistence.HistoryReplicationTask); ok {\n\t\t\ttaskUpdated = true\n\t\t\ttask.NewRunBranchToken = newRunBranchToken\n\t\t\tlogger.Debug(\"mergeContinueAsNewReplicationTasks: updated replication task\",\n\t\t\t\ttag.WorkflowDomainID(currentWorkflowMutation.ExecutionInfo.DomainID),\n\t\t\t\ttag.WorkflowID(currentWorkflowMutation.ExecutionInfo.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(currentWorkflowMutation.ExecutionInfo.RunID),\n\t\t\t\ttag.Dynamic(\"taskid\", task.TaskID),\n\t\t\t\ttag.Dynamic(\"version\", task.Version),\n\t\t\t\ttag.Dynamic(\"visibility_ts\", task.VisibilityTimestamp.Format(time.RFC3339)),\n\t\t\t)\n\t\t}\n\t}\n\tif !taskUpdated {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"unable to find replication task from current workflow for continue as new replication\",\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (c *contextImpl) PersistStartWorkflowBatchEvents(\n\tctx context.Context,\n\tworkflowEvents *persistence.WorkflowEvents,\n) (events.PersistedBlob, error) {\n\n\tif len(workflowEvents.Events) == 0 {\n\t\treturn events.PersistedBlob{}, &types.InternalServiceError{\n\t\t\tMessage: \"cannot persist first workflow events with empty events\",\n\t\t}\n\t}\n\n\tdomainID := workflowEvents.DomainID\n\tdomainName, err := c.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn events.PersistedBlob{}, err\n\t}\n\tworkflowID := workflowEvents.WorkflowID\n\trunID := workflowEvents.RunID\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowEvents.WorkflowID,\n\t\tRunID:      workflowEvents.RunID,\n\t}\n\n\tresp, err := c.appendHistoryNodesFn(\n\t\tctx,\n\t\tdomainID,\n\t\texecution,\n\t\t&persistence.AppendHistoryNodesRequest{\n\t\t\tIsNewBranch: true,\n\t\t\tInfo:        persistence.BuildHistoryGarbageCleanupInfo(domainID, workflowID, runID),\n\t\t\tBranchToken: workflowEvents.BranchToken,\n\t\t\tEvents:      workflowEvents.Events,\n\t\t\tDomainName:  domainName,\n\t\t\t// TransactionID is set by shard context\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn events.PersistedBlob{}, err\n\t}\n\treturn events.PersistedBlob{\n\t\tDataBlob:     resp.DataBlob,\n\t\tBranchToken:  workflowEvents.BranchToken,\n\t\tFirstEventID: workflowEvents.Events[0].ID,\n\t}, nil\n}\n\nfunc (c *contextImpl) PersistNonStartWorkflowBatchEvents(\n\tctx context.Context,\n\tworkflowEvents *persistence.WorkflowEvents,\n) (events.PersistedBlob, error) {\n\n\tif len(workflowEvents.Events) == 0 {\n\t\treturn events.PersistedBlob{}, nil // allow update workflow without events\n\t}\n\n\tdomainID := workflowEvents.DomainID\n\tdomainName, err := c.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn events.PersistedBlob{}, err\n\t}\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: workflowEvents.WorkflowID,\n\t\tRunID:      workflowEvents.RunID,\n\t}\n\n\tresp, err := c.appendHistoryNodesFn(\n\t\tctx,\n\t\tdomainID,\n\t\texecution,\n\t\t&persistence.AppendHistoryNodesRequest{\n\t\t\tIsNewBranch: false,\n\t\t\tBranchToken: workflowEvents.BranchToken,\n\t\t\tEvents:      workflowEvents.Events,\n\t\t\tDomainName:  domainName,\n\t\t\t// TransactionID is set by shard context\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn events.PersistedBlob{}, err\n\t}\n\treturn events.PersistedBlob{\n\t\tDataBlob:     resp.DataBlob,\n\t\tBranchToken:  workflowEvents.BranchToken,\n\t\tFirstEventID: workflowEvents.Events[0].ID,\n\t}, nil\n}\n\nfunc appendHistoryV2EventsWithRetry(\n\tctx context.Context,\n\tshardContext shard.Context,\n\tretryPolicy backoff.RetryPolicy,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\trequest *persistence.AppendHistoryNodesRequest,\n) (*persistence.AppendHistoryNodesResponse, error) {\n\n\tvar resp *persistence.AppendHistoryNodesResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = shardContext.AppendHistoryV2Events(ctx, request, domainID, execution)\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc createWorkflowExecutionWithRetry(\n\tctx context.Context,\n\tshardContext shard.Context,\n\tlogger log.Logger,\n\tretryPolicy backoff.RetryPolicy,\n\trequest *persistence.CreateWorkflowExecutionRequest,\n) (*persistence.CreateWorkflowExecutionResponse, error) {\n\tlogger = logger.Helper()\n\n\tvar resp *persistence.CreateWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = shardContext.CreateWorkflowExecution(ctx, request)\n\t\treturn err\n\t}\n\tisRetryable := func(err error) bool {\n\t\tif _, ok := err.(*persistence.TimeoutError); ok {\n\t\t\t// TODO: is timeout error retryable for create workflow?\n\t\t\t// if we treat it as retryable, user may receive workflowAlreadyRunning error\n\t\t\t// on the first start workflow execution request.\n\t\t\treturn false\n\t\t}\n\t\treturn persistence.IsTransientError(err)\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(isRetryable),\n\t)\n\n\terr := throttleRetry.Do(ctx, op)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn resp, nil\n\tcase *persistence.WorkflowExecutionAlreadyStartedError:\n\t\t// it is possible that workflow already exists and caller need to apply\n\t\t// workflow ID reuse policy\n\t\treturn nil, err\n\tdefault:\n\t\tlogger.Error(\n\t\t\t\"Persistent store operation failure\",\n\t\t\ttag.StoreOperationCreateWorkflowExecution,\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn nil, err\n\t}\n}\n\nfunc getWorkflowExecutionWithRetry(\n\tctx context.Context,\n\tshardContext shard.Context,\n\tlogger log.Logger,\n\tretryPolicy backoff.RetryPolicy,\n\trequest *persistence.GetWorkflowExecutionRequest,\n) (*persistence.GetWorkflowExecutionResponse, error) {\n\tlogger.Helper()\n\n\tvar resp *persistence.GetWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = shardContext.GetWorkflowExecution(ctx, request)\n\t\treturn err\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn resp, nil\n\tcase *types.EntityNotExistsError:\n\t\t// it is possible that workflow does not exists\n\t\treturn nil, err\n\tdefault:\n\t\t// If error is shard closed, only log error if shard has been closed for a while,\n\t\t// otherwise always log\n\t\tvar shardClosedError *shard.ErrShardClosed\n\t\tif !errors.As(err, &shardClosedError) || shardContext.GetTimeSource().Since(shardClosedError.ClosedAt) > shard.TimeBeforeShardClosedIsError {\n\t\t\tlogger.Error(\"Persistent fetch operation failure\", tag.StoreOperationGetWorkflowExecution, tag.Error(err))\n\t\t}\n\n\t\treturn nil, err\n\t}\n}\n\nfunc updateWorkflowExecutionWithRetry(\n\tctx context.Context,\n\tshardContext shard.Context,\n\tlogger log.Logger,\n\tretryPolicy backoff.RetryPolicy,\n\trequest *persistence.UpdateWorkflowExecutionRequest,\n) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\tlogger.Helper()\n\n\tvar resp *persistence.UpdateWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = shardContext.UpdateWorkflowExecution(ctx, request)\n\t\treturn err\n\t}\n\t// Preparation for the task Validation.\n\t// metricsClient := c.shard.GetMetricsClient()\n\t// domainCache := c.shard.GetDomainCache()\n\t// executionManager := c.shard.GetExecutionManager()\n\t// historymanager := c.shard.GetHistoryManager()\n\t// zapLogger, _ := zap.NewProduction()\n\t// checker, _ := taskvalidator.NewWfChecker(zapLogger, metricsClient, domainCache, executionManager, historymanager)\n\n\tisRetryable := func(err error) bool {\n\t\tif _, ok := err.(*persistence.TimeoutError); ok {\n\t\t\t// timeout error is not retryable for update workflow execution\n\t\t\treturn false\n\t\t}\n\t\treturn persistence.IsTransientError(err)\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(isRetryable),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn resp, nil\n\tcase *persistence.ConditionFailedError:\n\t\treturn nil, &conflictError{err}\n\tdefault:\n\t\tlogger.Error(\n\t\t\t\"Persistent store operation failure\",\n\t\t\ttag.StoreOperationUpdateWorkflowExecution,\n\t\t\ttag.Error(err),\n\t\t\ttag.Number(request.UpdateWorkflowMutation.Condition),\n\t\t)\n\t\t// TODO: Call the Task Validation here so that it happens whenever an error happen during Update.\n\t\t// err1 := checker.WorkflowCheckforValidation(\n\t\t//\tctx,\n\t\t//\tc.workflowExecution.GetWorkflowID(),\n\t\t//\tc.domainID,\n\t\t//\tc.GetDomainName(),\n\t\t//\tc.workflowExecution.GetRunID(),\n\t\t// )\n\t\t// if err1 != nil {\n\t\t//\treturn nil, err1\n\t\t// }\n\t\treturn nil, err\n\t}\n}\n\nfunc (c *contextImpl) updateWorkflowExecutionEventReapply(\n\tupdateMode persistence.UpdateWorkflowMode,\n\teventBatch1 []*persistence.WorkflowEvents,\n\teventBatch2 []*persistence.WorkflowEvents,\n) error {\n\n\tif updateMode != persistence.UpdateWorkflowModeBypassCurrent {\n\t\treturn nil\n\t}\n\n\tvar eventBatches []*persistence.WorkflowEvents\n\teventBatches = append(eventBatches, eventBatch1...)\n\teventBatches = append(eventBatches, eventBatch2...)\n\treturn c.ReapplyEvents(eventBatches)\n}\n\nfunc (c *contextImpl) conflictResolveEventReapply(\n\tconflictResolveMode persistence.ConflictResolveWorkflowMode,\n\teventBatch1 []*persistence.WorkflowEvents,\n\teventBatch2 []*persistence.WorkflowEvents,\n) error {\n\n\tif conflictResolveMode != persistence.ConflictResolveWorkflowModeBypassCurrent {\n\t\treturn nil\n\t}\n\n\tvar eventBatches []*persistence.WorkflowEvents\n\teventBatches = append(eventBatches, eventBatch1...)\n\teventBatches = append(eventBatches, eventBatch2...)\n\treturn c.ReapplyEvents(eventBatches)\n}\n\nfunc (c *contextImpl) ReapplyEvents(\n\teventBatches []*persistence.WorkflowEvents,\n) error {\n\n\t// NOTE: this function should only be used to workflow which is\n\t// not the caller, or otherwise deadlock will appear\n\n\tif len(eventBatches) == 0 {\n\t\treturn nil\n\t}\n\n\tdomainID := eventBatches[0].DomainID\n\tworkflowID := eventBatches[0].WorkflowID\n\trunID := eventBatches[0].RunID\n\tdomainCache := c.shard.GetDomainCache()\n\tdomainEntry, err := domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif domainEntry.IsDomainPendingActive() {\n\t\treturn nil\n\t}\n\tvar reapplyEvents []*types.HistoryEvent\n\tfor _, events := range eventBatches {\n\t\tif events.DomainID != domainID ||\n\t\t\tevents.WorkflowID != workflowID {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"workflowExecutionContext encounter mismatch domainID / workflowID in events reapplication.\",\n\t\t\t}\n\t\t}\n\n\t\tfor _, event := range events.Events {\n\t\t\tswitch event.GetEventType() {\n\t\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\t\treapplyEvents = append(reapplyEvents, event)\n\t\t\t}\n\t\t}\n\t}\n\tif len(reapplyEvents) == 0 {\n\t\treturn nil\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), defaultRemoteCallTimeout)\n\tdefer cancel()\n\n\tactiveClusterInfo, err := c.shard.GetActiveClusterManager().GetActiveClusterInfoByWorkflow(ctx, domainID, workflowID, runID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif activeClusterInfo.ActiveClusterName == c.shard.GetClusterMetadata().GetCurrentClusterName() {\n\t\treturn c.shard.GetEngine().ReapplyEvents(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\trunID,\n\t\t\treapplyEvents,\n\t\t)\n\t}\n\n\t// Reapply events only reapply to the current run.\n\t// The run id is only used for reapply event de-duplication\n\texecution := &types.WorkflowExecution{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\tclientBean := c.shard.GetService().GetClientBean()\n\tserializer := c.shard.GetService().GetPayloadSerializer()\n\t// The active cluster of the domain is the same as current cluster.\n\t// Use the history from the same cluster to reapply events\n\treapplyEventsDataBlob, err := serializer.SerializeBatchEvents(\n\t\treapplyEvents,\n\t\tconstants.EncodingTypeThriftRW,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\t// The active cluster of the domain is differ from the current cluster\n\t// Use frontend client to route this request to the active cluster\n\t// Reapplication only happens in active cluster\n\tsourceCluster, err := clientBean.GetRemoteAdminClient(activeClusterInfo.ActiveClusterName)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: err.Error(),\n\t\t}\n\t}\n\treturn sourceCluster.ReapplyEvents(\n\t\tctx,\n\t\t&types.ReapplyEventsRequest{\n\t\t\tDomainName:        domainEntry.GetInfo().Name,\n\t\t\tWorkflowExecution: execution,\n\t\t\tEvents:            reapplyEventsDataBlob.ToInternal(),\n\t\t},\n\t)\n}\n\nfunc (c *contextImpl) ByteSize() uint64 {\n\n\tvar size int\n\t// Estimate size of strings\n\tsize += len(c.domainID)\n\n\tsize += len(c.workflowExecution.GetWorkflowID()) + len(c.workflowExecution.GetRunID())\n\tsize += 3 * constants.StringSizeOverheadBytes\n\n\tsize += 3 * 8 // logger\n\tsize += 512   // MetricsClient estimation\n\tsize += 256   // ExecutionManager estimation\n\tsize += 8     // Mutex\n\tsize += 1024  // Mutable-state estimation\n\tsize += 8     // stats pointer\n\n\tsize += 18 * 8 // 18 function pointers with 8 bytes each\n\treturn uint64(size)\n}\n\nfunc isOperationPossiblySuccessfulError(err error) bool {\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn false\n\tcase *types.WorkflowExecutionAlreadyStartedError,\n\t\t*persistence.WorkflowExecutionAlreadyStartedError,\n\t\t*persistence.CurrentWorkflowConditionFailedError,\n\t\t*persistence.ConditionFailedError,\n\t\t*types.ServiceBusyError,\n\t\t*types.LimitExceededError,\n\t\t*persistence.ShardOwnershipLostError:\n\t\treturn false\n\tcase *persistence.TimeoutError:\n\t\treturn true\n\tdefault:\n\t\treturn !IsConflictError(err)\n\t}\n}\n\nfunc validateWorkflowRequestsAndMode(requests []*persistence.WorkflowRequest, mode persistence.CreateWorkflowRequestMode) error {\n\tif mode != persistence.CreateWorkflowRequestModeNew {\n\t\treturn nil\n\t}\n\tif len(requests) > 2 {\n\t\treturn &types.InternalServiceError{Message: \"too many workflow request entities generated from a single API request\"}\n\t} else if len(requests) == 2 {\n\t\t// SignalWithStartWorkflow API can generate 2 workflow requests\n\t\tif (requests[0].RequestType == persistence.WorkflowRequestTypeStart && requests[1].RequestType == persistence.WorkflowRequestTypeSignal) ||\n\t\t\t(requests[1].RequestType == persistence.WorkflowRequestTypeStart && requests[0].RequestType == persistence.WorkflowRequestTypeSignal) {\n\t\t\treturn nil\n\t\t}\n\t\treturn &types.InternalServiceError{Message: \"too many workflow request entities generated from a single API request\"}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/execution/context_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: context.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source context.go -destination context_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n\tevents \"github.com/uber/cadence/service/history/events\"\n)\n\n// MockContext is a mock of Context interface.\ntype MockContext struct {\n\tctrl     *gomock.Controller\n\trecorder *MockContextMockRecorder\n\tisgomock struct{}\n}\n\n// MockContextMockRecorder is the mock recorder for MockContext.\ntype MockContextMockRecorder struct {\n\tmock *MockContext\n}\n\n// NewMockContext creates a new mock instance.\nfunc NewMockContext(ctrl *gomock.Controller) *MockContext {\n\tmock := &MockContext{ctrl: ctrl}\n\tmock.recorder = &MockContextMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockContext) EXPECT() *MockContextMockRecorder {\n\treturn m.recorder\n}\n\n// ByteSize mocks base method.\nfunc (m *MockContext) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockContextMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockContext)(nil).ByteSize))\n}\n\n// Clear mocks base method.\nfunc (m *MockContext) Clear() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Clear\")\n}\n\n// Clear indicates an expected call of Clear.\nfunc (mr *MockContextMockRecorder) Clear() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Clear\", reflect.TypeOf((*MockContext)(nil).Clear))\n}\n\n// ConflictResolveWorkflowExecution mocks base method.\nfunc (m *MockContext) ConflictResolveWorkflowExecution(ctx context.Context, now time.Time, conflictResolveMode persistence.ConflictResolveWorkflowMode, resetMutableState MutableState, newContext Context, newMutableState MutableState, currentContext Context, currentMutableState MutableState, currentTransactionPolicy *TransactionPolicy) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ConflictResolveWorkflowExecution\", ctx, now, conflictResolveMode, resetMutableState, newContext, newMutableState, currentContext, currentMutableState, currentTransactionPolicy)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ConflictResolveWorkflowExecution indicates an expected call of ConflictResolveWorkflowExecution.\nfunc (mr *MockContextMockRecorder) ConflictResolveWorkflowExecution(ctx, now, conflictResolveMode, resetMutableState, newContext, newMutableState, currentContext, currentMutableState, currentTransactionPolicy any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ConflictResolveWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).ConflictResolveWorkflowExecution), ctx, now, conflictResolveMode, resetMutableState, newContext, newMutableState, currentContext, currentMutableState, currentTransactionPolicy)\n}\n\n// CreateWorkflowExecution mocks base method.\nfunc (m *MockContext) CreateWorkflowExecution(ctx context.Context, newWorkflow *persistence.WorkflowSnapshot, persistedHistory events.PersistedBlob, createMode persistence.CreateWorkflowMode, prevRunID string, prevLastWriteVersion int64, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateWorkflowExecution\", ctx, newWorkflow, persistedHistory, createMode, prevRunID, prevLastWriteVersion, workflowRequestMode)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateWorkflowExecution indicates an expected call of CreateWorkflowExecution.\nfunc (mr *MockContextMockRecorder) CreateWorkflowExecution(ctx, newWorkflow, persistedHistory, createMode, prevRunID, prevLastWriteVersion, workflowRequestMode any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).CreateWorkflowExecution), ctx, newWorkflow, persistedHistory, createMode, prevRunID, prevLastWriteVersion, workflowRequestMode)\n}\n\n// GetDomainID mocks base method.\nfunc (m *MockContext) GetDomainID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainID indicates an expected call of GetDomainID.\nfunc (mr *MockContextMockRecorder) GetDomainID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainID\", reflect.TypeOf((*MockContext)(nil).GetDomainID))\n}\n\n// GetDomainName mocks base method.\nfunc (m *MockContext) GetDomainName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainName indicates an expected call of GetDomainName.\nfunc (mr *MockContextMockRecorder) GetDomainName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainName\", reflect.TypeOf((*MockContext)(nil).GetDomainName))\n}\n\n// GetExecution mocks base method.\nfunc (m *MockContext) GetExecution() *types.WorkflowExecution {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecution\")\n\tret0, _ := ret[0].(*types.WorkflowExecution)\n\treturn ret0\n}\n\n// GetExecution indicates an expected call of GetExecution.\nfunc (mr *MockContextMockRecorder) GetExecution() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecution\", reflect.TypeOf((*MockContext)(nil).GetExecution))\n}\n\n// GetHistorySize mocks base method.\nfunc (m *MockContext) GetHistorySize() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistorySize\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetHistorySize indicates an expected call of GetHistorySize.\nfunc (mr *MockContextMockRecorder) GetHistorySize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistorySize\", reflect.TypeOf((*MockContext)(nil).GetHistorySize))\n}\n\n// GetWorkflowExecution mocks base method.\nfunc (m *MockContext) GetWorkflowExecution() MutableState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecution\")\n\tret0, _ := ret[0].(MutableState)\n\treturn ret0\n}\n\n// GetWorkflowExecution indicates an expected call of GetWorkflowExecution.\nfunc (mr *MockContextMockRecorder) GetWorkflowExecution() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).GetWorkflowExecution))\n}\n\n// LoadExecutionStats mocks base method.\nfunc (m *MockContext) LoadExecutionStats(ctx context.Context) (*persistence.ExecutionStats, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadExecutionStats\", ctx)\n\tret0, _ := ret[0].(*persistence.ExecutionStats)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LoadExecutionStats indicates an expected call of LoadExecutionStats.\nfunc (mr *MockContextMockRecorder) LoadExecutionStats(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadExecutionStats\", reflect.TypeOf((*MockContext)(nil).LoadExecutionStats), ctx)\n}\n\n// LoadWorkflowExecution mocks base method.\nfunc (m *MockContext) LoadWorkflowExecution(ctx context.Context) (MutableState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadWorkflowExecution\", ctx)\n\tret0, _ := ret[0].(MutableState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LoadWorkflowExecution indicates an expected call of LoadWorkflowExecution.\nfunc (mr *MockContextMockRecorder) LoadWorkflowExecution(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).LoadWorkflowExecution), ctx)\n}\n\n// LoadWorkflowExecutionWithTaskVersion mocks base method.\nfunc (m *MockContext) LoadWorkflowExecutionWithTaskVersion(ctx context.Context, incomingVersion int64) (MutableState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadWorkflowExecutionWithTaskVersion\", ctx, incomingVersion)\n\tret0, _ := ret[0].(MutableState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// LoadWorkflowExecutionWithTaskVersion indicates an expected call of LoadWorkflowExecutionWithTaskVersion.\nfunc (mr *MockContextMockRecorder) LoadWorkflowExecutionWithTaskVersion(ctx, incomingVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadWorkflowExecutionWithTaskVersion\", reflect.TypeOf((*MockContext)(nil).LoadWorkflowExecutionWithTaskVersion), ctx, incomingVersion)\n}\n\n// Lock mocks base method.\nfunc (m *MockContext) Lock(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Lock\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Lock indicates an expected call of Lock.\nfunc (mr *MockContextMockRecorder) Lock(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Lock\", reflect.TypeOf((*MockContext)(nil).Lock), ctx)\n}\n\n// PersistNonStartWorkflowBatchEvents mocks base method.\nfunc (m *MockContext) PersistNonStartWorkflowBatchEvents(ctx context.Context, workflowEvents *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PersistNonStartWorkflowBatchEvents\", ctx, workflowEvents)\n\tret0, _ := ret[0].(events.PersistedBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PersistNonStartWorkflowBatchEvents indicates an expected call of PersistNonStartWorkflowBatchEvents.\nfunc (mr *MockContextMockRecorder) PersistNonStartWorkflowBatchEvents(ctx, workflowEvents any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PersistNonStartWorkflowBatchEvents\", reflect.TypeOf((*MockContext)(nil).PersistNonStartWorkflowBatchEvents), ctx, workflowEvents)\n}\n\n// PersistStartWorkflowBatchEvents mocks base method.\nfunc (m *MockContext) PersistStartWorkflowBatchEvents(ctx context.Context, workflowEvents *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PersistStartWorkflowBatchEvents\", ctx, workflowEvents)\n\tret0, _ := ret[0].(events.PersistedBlob)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PersistStartWorkflowBatchEvents indicates an expected call of PersistStartWorkflowBatchEvents.\nfunc (mr *MockContextMockRecorder) PersistStartWorkflowBatchEvents(ctx, workflowEvents any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PersistStartWorkflowBatchEvents\", reflect.TypeOf((*MockContext)(nil).PersistStartWorkflowBatchEvents), ctx, workflowEvents)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockContext) ReapplyEvents(eventBatches []*persistence.WorkflowEvents) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", eventBatches)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockContextMockRecorder) ReapplyEvents(eventBatches any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockContext)(nil).ReapplyEvents), eventBatches)\n}\n\n// SetHistorySize mocks base method.\nfunc (m *MockContext) SetHistorySize(size int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetHistorySize\", size)\n}\n\n// SetHistorySize indicates an expected call of SetHistorySize.\nfunc (mr *MockContextMockRecorder) SetHistorySize(size any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetHistorySize\", reflect.TypeOf((*MockContext)(nil).SetHistorySize), size)\n}\n\n// SetWorkflowExecution mocks base method.\nfunc (m *MockContext) SetWorkflowExecution(mutableState MutableState) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetWorkflowExecution\", mutableState)\n}\n\n// SetWorkflowExecution indicates an expected call of SetWorkflowExecution.\nfunc (mr *MockContextMockRecorder) SetWorkflowExecution(mutableState any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).SetWorkflowExecution), mutableState)\n}\n\n// Unlock mocks base method.\nfunc (m *MockContext) Unlock() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Unlock\")\n}\n\n// Unlock indicates an expected call of Unlock.\nfunc (mr *MockContextMockRecorder) Unlock() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unlock\", reflect.TypeOf((*MockContext)(nil).Unlock))\n}\n\n// UpdateWorkflowExecutionAsActive mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionAsActive(ctx context.Context, now time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionAsActive\", ctx, now)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionAsActive indicates an expected call of UpdateWorkflowExecutionAsActive.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionAsActive(ctx, now any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionAsActive\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionAsActive), ctx, now)\n}\n\n// UpdateWorkflowExecutionAsPassive mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionAsPassive(ctx context.Context, now time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionAsPassive\", ctx, now)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionAsPassive indicates an expected call of UpdateWorkflowExecutionAsPassive.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionAsPassive(ctx, now any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionAsPassive\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionAsPassive), ctx, now)\n}\n\n// UpdateWorkflowExecutionTasks mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionTasks(ctx context.Context, now time.Time) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionTasks\", ctx, now)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionTasks indicates an expected call of UpdateWorkflowExecutionTasks.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionTasks(ctx, now any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionTasks\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionTasks), ctx, now)\n}\n\n// UpdateWorkflowExecutionWithNew mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionWithNew(ctx context.Context, now time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentWorkflowTransactionPolicy TransactionPolicy, newWorkflowTransactionPolicy *TransactionPolicy, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithNew\", ctx, now, updateMode, newContext, newMutableState, currentWorkflowTransactionPolicy, newWorkflowTransactionPolicy, workflowRequestMode)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithNew indicates an expected call of UpdateWorkflowExecutionWithNew.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionWithNew(ctx, now, updateMode, newContext, newMutableState, currentWorkflowTransactionPolicy, newWorkflowTransactionPolicy, workflowRequestMode any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithNew\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionWithNew), ctx, now, updateMode, newContext, newMutableState, currentWorkflowTransactionPolicy, newWorkflowTransactionPolicy, workflowRequestMode)\n}\n\n// UpdateWorkflowExecutionWithNewAsActive mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionWithNewAsActive(ctx context.Context, now time.Time, newContext Context, newMutableState MutableState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithNewAsActive\", ctx, now, newContext, newMutableState)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithNewAsActive indicates an expected call of UpdateWorkflowExecutionWithNewAsActive.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionWithNewAsActive(ctx, now, newContext, newMutableState any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithNewAsActive\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionWithNewAsActive), ctx, now, newContext, newMutableState)\n}\n\n// UpdateWorkflowExecutionWithNewAsPassive mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecutionWithNewAsPassive(ctx context.Context, now time.Time, newContext Context, newMutableState MutableState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecutionWithNewAsPassive\", ctx, now, newContext, newMutableState)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowExecutionWithNewAsPassive indicates an expected call of UpdateWorkflowExecutionWithNewAsPassive.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecutionWithNewAsPassive(ctx, now, newContext, newMutableState any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecutionWithNewAsPassive\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecutionWithNewAsPassive), ctx, now, newContext, newMutableState)\n}\n"
  },
  {
    "path": "service/history/execution/context_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestIsOperationPossiblySuccessfulError(t *testing.T) {\n\tassert.False(t, isOperationPossiblySuccessfulError(nil))\n\tassert.False(t, isOperationPossiblySuccessfulError(&types.WorkflowExecutionAlreadyStartedError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&persistence.WorkflowExecutionAlreadyStartedError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&persistence.CurrentWorkflowConditionFailedError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&persistence.ConditionFailedError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&types.ServiceBusyError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&types.LimitExceededError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(&persistence.ShardOwnershipLostError{}))\n\tassert.True(t, isOperationPossiblySuccessfulError(&persistence.TimeoutError{}))\n\tassert.False(t, isOperationPossiblySuccessfulError(NewConflictError(t, &persistence.ConditionFailedError{})))\n\tassert.True(t, isOperationPossiblySuccessfulError(context.DeadlineExceeded))\n}\n\nfunc TestMergeContinueAsNewReplicationTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                    string\n\t\tupdateMode              persistence.UpdateWorkflowMode\n\t\tcurrentWorkflowMutation *persistence.WorkflowMutation\n\t\tnewWorkflowSnapshot     *persistence.WorkflowSnapshot\n\t\twantErr                 bool\n\t\tassertErr               func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"current workflow does not continue as new\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"update workflow as zombie and continue as new without new zombie workflow\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateMode: persistence.UpdateWorkflowModeBypassCurrent,\n\t\t\twantErr:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"continue as new on the passive side\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateMode: persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\twantErr:    false,\n\t\t},\n\t\t{\n\t\t\tname: \"continue as new on the active side, but new workflow is not provided\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateMode: persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\twantErr:    true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.InternalServiceError{}, err)\n\t\t\t\tassert.Contains(t, err.Error(), \"unable to find replication task from new workflow for continue as new replication\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"continue as new on the active side, but new workflow has no replication task\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tnewWorkflowSnapshot: &persistence.WorkflowSnapshot{},\n\t\t\tupdateMode:          persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\twantErr:             true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.InternalServiceError{}, err)\n\t\t\t\tassert.Contains(t, err.Error(), \"unable to find replication task from new workflow for continue as new replication\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"continue as new on the active side, but current workflow has no history replication task\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.SyncActivityTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tnewWorkflowSnapshot: &persistence.WorkflowSnapshot{\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateMode: persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\twantErr:    true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.InternalServiceError{}, err)\n\t\t\t\tassert.Contains(t, err.Error(), \"unable to find replication task from current workflow for continue as new replication\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"continue as new on the active side\",\n\t\t\tcurrentWorkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusContinuedAsNew,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tnewWorkflowSnapshot: &persistence.WorkflowSnapshot{\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateMode: persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\twantErr:    false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlogger := log.NewNoop()\n\t\t\terr := mergeContinueAsNewReplicationTasks(logger, tc.updateMode, tc.currentWorkflowMutation, tc.newWorkflowSnapshot)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNotifyTasksFromWorkflowSnapshot(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\tworkflowSnapShot *persistence.WorkflowSnapshot\n\t\thistory          events.PersistedBlobs\n\t\tpersistenceError bool\n\t\tmockSetup        func(*engine.MockEngine)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflowSnapShot: &persistence.WorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityInfos: []*persistence.ActivityInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\tScheduleID: 11,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: {\n\t\t\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\t\t\tTaskList: \"test-tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer: {\n\t\t\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\t\t\tAttempt: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t\t\tNextEventID:  10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: events.PersistedBlobs{\n\t\t\t\tevents.PersistedBlob{},\n\t\t\t},\n\t\t\tpersistenceError: true,\n\t\t\tmockSetup: func(mockEngine *engine.MockEngine) {\n\t\t\t\tmockEngine.EXPECT().NotifyNewTransferTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\t\t\tTaskList: \"test-tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t\tmockEngine.EXPECT().NotifyNewTimerTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\t\t\tAttempt: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t\tmockEngine.EXPECT().NotifyNewReplicationTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t\t\tNextEventID:  10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tActivities: map[int64]*persistence.ActivityInfo{\n\t\t\t\t\t\t11: {\n\t\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\t\tScheduleID: 11,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tHistory: events.PersistedBlobs{\n\t\t\t\t\t\tevents.PersistedBlob{},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"nil snapshot\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockEngine := engine.NewMockEngine(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockEngine)\n\t\t\t}\n\t\t\tnotifyTasksFromWorkflowSnapshot(mockEngine, tc.workflowSnapShot, tc.history, tc.persistenceError)\n\t\t})\n\t}\n}\n\nfunc TestNotifyTasksFromWorkflowMutation(t *testing.T) {\n\ttestCases := []struct {\n\t\tname             string\n\t\tworkflowMutation *persistence.WorkflowMutation\n\t\thistory          events.PersistedBlobs\n\t\tpersistenceError bool\n\t\tmockSetup        func(*engine.MockEngine)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tworkflowMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tUpsertActivityInfos: []*persistence.ActivityInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\tScheduleID: 11,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: {\n\t\t\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\t\t\tTaskList: \"test-tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer: {\n\t\t\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\t\t\tAttempt: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: {\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t\t\tNextEventID:  10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: events.PersistedBlobs{\n\t\t\t\tevents.PersistedBlob{},\n\t\t\t},\n\t\t\tpersistenceError: true,\n\t\t\tmockSetup: func(mockEngine *engine.MockEngine) {\n\t\t\t\tmockEngine.EXPECT().NotifyNewTransferTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTask{\n\t\t\t\t\t\t\tTaskList: \"test-tl\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t\tmockEngine.EXPECT().NotifyNewTimerTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.ActivityTimeoutTask{\n\t\t\t\t\t\t\tAttempt: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t\tmockEngine.EXPECT().NotifyNewReplicationTasks(&hcommon.NotifyTaskInfo{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t\t\tNextEventID:  10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tVersionHistories: &persistence.VersionHistories{\n\t\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tActivities: map[int64]*persistence.ActivityInfo{\n\t\t\t\t\t\t11: {\n\t\t\t\t\t\t\tVersion:    1,\n\t\t\t\t\t\t\tScheduleID: 11,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tHistory: events.PersistedBlobs{\n\t\t\t\t\t\tevents.PersistedBlob{},\n\t\t\t\t\t},\n\t\t\t\t\tPersistenceError: true,\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"nil mutation\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockEngine := engine.NewMockEngine(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockEngine)\n\t\t\t}\n\t\t\tnotifyTasksFromWorkflowMutation(mockEngine, tc.workflowMutation, tc.history, tc.persistenceError)\n\t\t})\n\t}\n}\n\nfunc TestActivityInfosToMap(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\tactivities []*persistence.ActivityInfo\n\t\twant       map[int64]*persistence.ActivityInfo\n\t}{\n\t\t{\n\t\t\tname: \"non-empty\",\n\t\t\tactivities: []*persistence.ActivityInfo{\n\t\t\t\t{\n\t\t\t\t\tVersion:    1,\n\t\t\t\t\tScheduleID: 11,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tVersion:    2,\n\t\t\t\t\tScheduleID: 12,\n\t\t\t\t},\n\t\t\t},\n\t\t\twant: map[int64]*persistence.ActivityInfo{\n\t\t\t\t11: {\n\t\t\t\t\tVersion:    1,\n\t\t\t\t\tScheduleID: 11,\n\t\t\t\t},\n\t\t\t\t12: {\n\t\t\t\t\tVersion:    2,\n\t\t\t\t\tScheduleID: 12,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"empty slice\",\n\t\t\tactivities: []*persistence.ActivityInfo{},\n\t\t\twant:       map[int64]*persistence.ActivityInfo{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil slice\",\n\t\t\twant: map[int64]*persistence.ActivityInfo{},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tassert.Equal(t, tc.want, activityInfosToMap(tc.activities))\n\t}\n}\n\nfunc TestCreateWorkflowExecutionWithRetry(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\trequest   *persistence.CreateWorkflowExecutionRequest\n\t\tmockSetup func(*shard.MockContext)\n\t\twant      *persistence.CreateWorkflowExecutionResponse\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\trequest: &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().CreateWorkflowExecution(gomock.Any(), &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t}).Return(&persistence.CreateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.CreateWorkflowExecutionResponse{\n\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"workflow already started error\",\n\t\t\trequest: &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &persistence.WorkflowExecutionAlreadyStartedError{})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &persistence.WorkflowExecutionAlreadyStartedError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timeout error\",\n\t\t\trequest: &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &persistence.TimeoutError{})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &persistence.TimeoutError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"retry succeeds\",\n\t\t\trequest: &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{})\n\t\t\t\tmockShard.EXPECT().CreateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.CreateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.CreateWorkflowExecutionResponse{\n\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tpolicy := backoff.NewExponentialRetryPolicy(time.Millisecond)\n\t\t\tpolicy.SetMaximumAttempts(1)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard)\n\t\t\t}\n\t\t\tresp, err := createWorkflowExecutionWithRetry(context.Background(), mockShard, testlogger.New(t), policy, tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionWithRetry(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\trequest   *persistence.UpdateWorkflowExecutionRequest\n\t\tmockSetup func(*shard.MockContext)\n\t\twant      *persistence.UpdateWorkflowExecutionResponse\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\trequest: &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().UpdateWorkflowExecution(gomock.Any(), &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t}).Return(&persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"condition failed error\",\n\t\t\trequest: &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &persistence.ConditionFailedError{})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &conflictError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"timeout error\",\n\t\t\trequest: &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &persistence.TimeoutError{})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &persistence.TimeoutError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"retry succeeds\",\n\t\t\trequest: &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{})\n\t\t\t\tmockShard.EXPECT().UpdateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tpolicy := backoff.NewExponentialRetryPolicy(time.Millisecond)\n\t\t\tpolicy.SetMaximumAttempts(1)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard)\n\t\t\t}\n\t\t\tresp, err := updateWorkflowExecutionWithRetry(context.Background(), mockShard, testlogger.New(t), policy, tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAppendHistoryV2EventsWithRetry(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tdomainID  string\n\t\texecution types.WorkflowExecution\n\t\trequest   *persistence.AppendHistoryNodesRequest\n\t\tmockSetup func(*shard.MockContext)\n\t\twant      *persistence.AppendHistoryNodesResponse\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname:     \"Success case\",\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\trequest: &persistence.AppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().AppendHistoryV2Events(gomock.Any(), &persistence.AppendHistoryNodesRequest{\n\t\t\t\t\tIsNewBranch: true,\n\t\t\t\t}, \"test-domain-id\", types.WorkflowExecution{WorkflowID: \"test-workflow-id\", RunID: \"test-run-id\"}).Return(&persistence.AppendHistoryNodesResponse{\n\t\t\t\t\tDataBlob: persistence.DataBlob{},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.AppendHistoryNodesResponse{\n\t\t\t\tDataBlob: persistence.DataBlob{},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"retry success\",\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\trequest: &persistence.AppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{})\n\t\t\t\tmockShard.EXPECT().AppendHistoryV2Events(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&persistence.AppendHistoryNodesResponse{\n\t\t\t\t\tDataBlob: persistence.DataBlob{},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.AppendHistoryNodesResponse{\n\t\t\t\tDataBlob: persistence.DataBlob{},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"non retryable error\",\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\texecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\trequest: &persistence.AppendHistoryNodesRequest{\n\t\t\t\tIsNewBranch: true,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().AppendHistoryV2Events(gomock.Any(), &persistence.AppendHistoryNodesRequest{\n\t\t\t\t\tIsNewBranch: true,\n\t\t\t\t}, \"test-domain-id\", types.WorkflowExecution{WorkflowID: \"test-workflow-id\", RunID: \"test-run-id\"}).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tpolicy := backoff.NewExponentialRetryPolicy(time.Millisecond)\n\t\t\tpolicy.SetMaximumAttempts(1)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard)\n\t\t\t}\n\t\t\tresp, err := appendHistoryV2EventsWithRetry(context.Background(), mockShard, policy, tc.domainID, tc.execution, tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPersistStartWorkflowBatchEvents(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                     string\n\t\tworkflowEvents           *persistence.WorkflowEvents\n\t\tmockSetup                func(*shard.MockContext, *cache.MockDomainCache)\n\t\tmockAppendHistoryNodesFn func(context.Context, string, types.WorkflowExecution, *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error)\n\t\twantErr                  bool\n\t\twant                     events.PersistedBlob\n\t\tassertErr                func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:           \"empty events\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{},\n\t\t\twantErr:        true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &types.InternalServiceError{})\n\t\t\t\tassert.Contains(t, err.Error(), \"cannot persist first workflow events with empty events\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get domain name\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"\", errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, err, errors.New(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to append history nodes\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockAppendHistoryNodesFn: func(context.Context, string, types.WorkflowExecution, *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\t\t\t\treturn nil, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, err, errors.New(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockAppendHistoryNodesFn: func(ctx context.Context, domainID string, execution types.WorkflowExecution, req *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\t\t\t\tassert.Equal(t, &persistence.AppendHistoryNodesRequest{\n\t\t\t\t\tIsNewBranch: true,\n\t\t\t\t\tInfo:        \"::\",\n\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\t}, req)\n\t\t\t\treturn &persistence.AppendHistoryNodesResponse{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twant: events.PersistedBlob{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t},\n\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\tFirstEventID: 1,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tshard: mockShard,\n\t\t\t}\n\t\t\tif tc.mockAppendHistoryNodesFn != nil {\n\t\t\t\tctx.appendHistoryNodesFn = tc.mockAppendHistoryNodesFn\n\t\t\t}\n\t\t\tgot, err := ctx.PersistStartWorkflowBatchEvents(context.Background(), tc.workflowEvents)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPersistNonStartWorkflowBatchEvents(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                     string\n\t\tworkflowEvents           *persistence.WorkflowEvents\n\t\tmockSetup                func(*shard.MockContext, *cache.MockDomainCache)\n\t\tmockAppendHistoryNodesFn func(context.Context, string, types.WorkflowExecution, *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error)\n\t\twantErr                  bool\n\t\twant                     events.PersistedBlob\n\t\tassertErr                func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:           \"empty events\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{},\n\t\t\twantErr:        false,\n\t\t\twant:           events.PersistedBlob{},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get domain name\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"\", errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, err, errors.New(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to append history nodes\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockAppendHistoryNodesFn: func(context.Context, string, types.WorkflowExecution, *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\t\t\t\treturn nil, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, err, errors.New(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tworkflowEvents: &persistence.WorkflowEvents{\n\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockAppendHistoryNodesFn: func(ctx context.Context, domainID string, execution types.WorkflowExecution, req *persistence.AppendHistoryNodesRequest) (*persistence.AppendHistoryNodesResponse, error) {\n\t\t\t\tassert.Equal(t, &persistence.AppendHistoryNodesRequest{\n\t\t\t\t\tIsNewBranch: false,\n\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\t}, req)\n\t\t\t\treturn &persistence.AppendHistoryNodesResponse{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\twant: events.PersistedBlob{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t},\n\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\tFirstEventID: 1,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tshard: mockShard,\n\t\t\t}\n\t\t\tif tc.mockAppendHistoryNodesFn != nil {\n\t\t\t\tctx.appendHistoryNodesFn = tc.mockAppendHistoryNodesFn\n\t\t\t}\n\t\t\tgot, err := ctx.PersistNonStartWorkflowBatchEvents(context.Background(), tc.workflowEvents)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCreateWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                  string\n\t\tnewWorkflow                           *persistence.WorkflowSnapshot\n\t\thistory                               events.PersistedBlob\n\t\tcreateMode                            persistence.CreateWorkflowMode\n\t\tprevRunID                             string\n\t\tprevLastWriteVersion                  int64\n\t\tcreateWorkflowRequestMode             persistence.CreateWorkflowRequestMode\n\t\tmockCreateWorkflowExecutionFn         func(context.Context, *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error)\n\t\tmockNotifyTasksFromWorkflowSnapshotFn func(*persistence.WorkflowSnapshot, events.PersistedBlobs, bool)\n\t\tmockEmitSessionUpdateStatsFn          func(string, *persistence.MutableStateUpdateSessionStats)\n\t\twantErr                               bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to create workflow execution with possibly success error\",\n\t\t\tnewWorkflow: &persistence.WorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: events.PersistedBlob{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t},\n\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\tFirstEventID: 1,\n\t\t\t},\n\t\t\tcreateMode:                persistence.CreateWorkflowModeContinueAsNew,\n\t\t\tprevRunID:                 \"test-prev-run-id\",\n\t\t\tprevLastWriteVersion:      123,\n\t\t\tcreateWorkflowRequestMode: persistence.CreateWorkflowRequestModeReplicated,\n\t\t\tmockCreateWorkflowExecutionFn: func(context.Context, *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error) {\n\t\t\t\treturn nil, &types.InternalServiceError{}\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(_ *persistence.WorkflowSnapshot, _ events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to validate workflow requests\",\n\t\t\tnewWorkflow: &persistence.WorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{{}, {}},\n\t\t\t},\n\t\t\thistory: events.PersistedBlob{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t},\n\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\tFirstEventID: 1,\n\t\t\t},\n\t\t\tcreateMode:                persistence.CreateWorkflowModeContinueAsNew,\n\t\t\tprevRunID:                 \"test-prev-run-id\",\n\t\t\tprevLastWriteVersion:      123,\n\t\t\tcreateWorkflowRequestMode: persistence.CreateWorkflowRequestModeNew,\n\t\t\twantErr:                   true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tnewWorkflow: &persistence.WorkflowSnapshot{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: events.PersistedBlob{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t},\n\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\tFirstEventID: 1,\n\t\t\t},\n\t\t\tcreateMode:                persistence.CreateWorkflowModeContinueAsNew,\n\t\t\tprevRunID:                 \"test-prev-run-id\",\n\t\t\tprevLastWriteVersion:      123,\n\t\t\tcreateWorkflowRequestMode: persistence.CreateWorkflowRequestModeReplicated,\n\t\t\tmockCreateWorkflowExecutionFn: func(ctx context.Context, req *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error) {\n\t\t\t\tassert.Equal(t, &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\t\tMode:                     persistence.CreateWorkflowModeContinueAsNew,\n\t\t\t\t\tPreviousRunID:            \"test-prev-run-id\",\n\t\t\t\t\tPreviousLastWriteVersion: 123,\n\t\t\t\t\tNewWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 3,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tWorkflowRequestMode: persistence.CreateWorkflowRequestModeReplicated,\n\t\t\t\t\tDomainName:          \"test-domain\",\n\t\t\t\t}, req)\n\t\t\t\treturn &persistence.CreateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(newWorkflow *persistence.WorkflowSnapshot, history events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}, newWorkflow)\n\t\t\t\tassert.Equal(t, events.PersistedBlobs{\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte(\"123\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken:  []byte{1, 2, 3},\n\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t}, history)\n\t\t\t\tassert.Equal(t, false, persistenceError)\n\t\t\t},\n\t\t\tmockEmitSessionUpdateStatsFn: func(domainName string, stats *persistence.MutableStateUpdateSessionStats) {\n\t\t\t\tassert.Equal(t, \"test-domain\", domainName)\n\t\t\t\tassert.Equal(t, &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t}, stats)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\tEnableStrongIdempotencySanityCheck: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t}).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\tctx := &contextImpl{\n\t\t\t\tlogger:        testlogger.New(t),\n\t\t\t\tshard:         mockShard,\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t}\n\t\t\tif tc.mockCreateWorkflowExecutionFn != nil {\n\t\t\t\tctx.createWorkflowExecutionFn = tc.mockCreateWorkflowExecutionFn\n\t\t\t}\n\t\t\tif tc.mockNotifyTasksFromWorkflowSnapshotFn != nil {\n\t\t\t\tctx.notifyTasksFromWorkflowSnapshotFn = tc.mockNotifyTasksFromWorkflowSnapshotFn\n\t\t\t}\n\t\t\tif tc.mockEmitSessionUpdateStatsFn != nil {\n\t\t\t\tctx.emitSessionUpdateStatsFn = tc.mockEmitSessionUpdateStatsFn\n\t\t\t}\n\t\t\terr := ctx.CreateWorkflowExecution(context.Background(), tc.newWorkflow, tc.history, tc.createMode, tc.prevRunID, tc.prevLastWriteVersion, tc.createWorkflowRequestMode)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                  string\n\t\tmockSetup                             func(*shard.MockContext, *cache.MockDomainCache, *MockMutableState)\n\t\tmockUpdateWorkflowExecutionFn         func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)\n\t\tmockNotifyTasksFromWorkflowMutationFn func(*persistence.WorkflowMutation, events.PersistedBlobs, bool)\n\t\tmockEmitSessionUpdateStatsFn          func(string, *persistence.MutableStateUpdateSessionStats)\n\t\twantErr                               bool\n\t\tassertErr                             func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"CloseTransactionAsMutation failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"found unexpected new events\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{{}}, nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.InternalServiceError{}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"found unexpected workflow requests\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{{}},\n\t\t\t\t}, []*persistence.WorkflowEvents{}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tEnableStrongIdempotencySanityCheck: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t\t})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"UpdateWorkflowExecutionTask can only be used for persisting new workflow tasks, but found new workflow requests\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"\", errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"update workflow failed with possibly success error\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionFn: func(_ context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\t\t\t\treturn nil, &types.InternalServiceError{}\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(_ *persistence.WorkflowMutation, _ events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError, \"case: update workflow failed with possibly success error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.InternalServiceError{}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionFn: func(_ context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\t\t\t\tassert.Equal(t, &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t},\n\t\t\t\t\tMode:       persistence.UpdateWorkflowModeIgnoreCurrent,\n\t\t\t\t\tDomainName: \"test-domain\",\n\t\t\t\t}, request, \"case: success\")\n\t\t\t\treturn &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(mutation *persistence.WorkflowMutation, history events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t}, mutation, \"case: success\")\n\t\t\t\tassert.Nil(t, history, \"case: success\")\n\t\t\t\tassert.Equal(t, false, persistenceError, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitSessionUpdateStatsFn: func(domainName string, stats *persistence.MutableStateUpdateSessionStats) {\n\t\t\t\tassert.Equal(t, \"test-domain\", domainName, \"case: success\")\n\t\t\t\tassert.Equal(t, &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t}, stats, \"case: success\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache, mockMutableState)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tlogger:        testlogger.New(t),\n\t\t\t\tshard:         mockShard,\n\t\t\t\tmutableState:  mockMutableState,\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t}\n\t\t\tif tc.mockUpdateWorkflowExecutionFn != nil {\n\t\t\t\tctx.updateWorkflowExecutionFn = tc.mockUpdateWorkflowExecutionFn\n\t\t\t}\n\t\t\tif tc.mockNotifyTasksFromWorkflowMutationFn != nil {\n\t\t\t\tctx.notifyTasksFromWorkflowMutationFn = tc.mockNotifyTasksFromWorkflowMutationFn\n\t\t\t}\n\t\t\tif tc.mockEmitSessionUpdateStatsFn != nil {\n\t\t\t\tctx.emitSessionUpdateStatsFn = tc.mockEmitSessionUpdateStatsFn\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionTasks(context.Background(), time.Unix(0, 0))\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionWithNew(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                      string\n\t\tupdateMode                                persistence.UpdateWorkflowMode\n\t\tnewContext                                Context\n\t\tcurrentWorkflowTransactionPolicy          TransactionPolicy\n\t\tnewWorkflowTransactionPolicy              *TransactionPolicy\n\t\tworkflowRequestMode                       persistence.CreateWorkflowRequestMode\n\t\tmockSetup                                 func(*shard.MockContext, *cache.MockDomainCache, *MockMutableState, *MockMutableState, *engine.MockEngine)\n\t\tmockPersistNonStartWorkflowBatchEventsFn  func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tmockPersistStartWorkflowBatchEventsFn     func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tmockUpdateWorkflowExecutionFn             func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)\n\t\tmockNotifyTasksFromWorkflowMutationFn     func(*persistence.WorkflowMutation, events.PersistedBlobs, bool)\n\t\tmockNotifyTasksFromWorkflowSnapshotFn     func(*persistence.WorkflowSnapshot, events.PersistedBlobs, bool)\n\t\tmockEmitSessionUpdateStatsFn              func(string, *persistence.MutableStateUpdateSessionStats)\n\t\tmockEmitWorkflowHistoryStatsFn            func(string, int, int)\n\t\tmockEmitLargeWorkflowShardIDStatsFn       func(int64, int64, int64, int64)\n\t\tmockEmitWorkflowCompletionStatsFn         func(string, string, string, string, string, *types.HistoryEvent)\n\t\tmockMergeContinueAsNewReplicationTasksFn  func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error\n\t\tmockUpdateWorkflowExecutionEventReapplyFn func(persistence.UpdateWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error\n\t\twantErr                                   bool\n\t\tassertErr                                 func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:                             \"CloseTransactionAsMutation failed\",\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyPassive,\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                             \"PersistNonStartWorkflowBatchEvents failed\",\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyPassive,\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CloseTransactionAsSnapshot failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(gomock.Any())\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"both current workflow and new workflow generate workflow requests\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tEnableStrongIdempotencySanityCheck: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t\t})\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{{}},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(gomock.Any())\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{{}},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"workflow requests are only expected to be generated from one workflow for UpdateWorkflowExecution\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"mergeContinueAsNewReplicationTasks failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(gomock.Any())\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockMergeContinueAsNewReplicationTasksFn: func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"updateWorkflowExecutionEventReapply failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(gomock.Any())\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockMergeContinueAsNewReplicationTasksFn: func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionEventReapplyFn: func(persistence.UpdateWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"updateWorkflowExecution failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(gomock.Any())\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockMergeContinueAsNewReplicationTasksFn: func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionEventReapplyFn: func(persistence.UpdateWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionFn: func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\t\t\t\treturn nil, errors.New(\"some error\")\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(_ *persistence.WorkflowMutation, _ events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError, \"case: updateWorkflowExecution failed\")\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(_ *persistence.WorkflowSnapshot, _ events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError, \"case: updateWorkflowExecution failed\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tupdateMode:                       persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive,\n\t\t\tnewWorkflowTransactionPolicy:     TransactionPolicyActive.Ptr(),\n\t\t\tworkflowRequestMode:              persistence.CreateWorkflowRequestModeReplicated,\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), TransactionPolicyActive).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branchtoken\"),\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(11))\n\t\t\t\tmockMutableState.EXPECT().SetHistorySize(int64(5))\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), TransactionPolicyActive).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateCompleted, persistence.WorkflowCloseStatusCompleted)\n\t\t\t\tmockShard.EXPECT().GetEngine().Return(mockEngine)\n\t\t\t\tmockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any())\n\t\t\t\tmockMutableState.EXPECT().GetLastFirstEventID().Return(int64(1))\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10))\n\t\t\t\tmockMutableState.EXPECT().GetPreviousStartedEventID().Return(int64(12))\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(20))\n\t\t\t\tmockMutableState.EXPECT().GetCompletionEvent(gomock.Any()).Return(&types.HistoryEvent{\n\t\t\t\t\tID: 123,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowEvents{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t}, history, \"case: success\")\n\t\t\t\treturn events.PersistedBlob{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowEvents{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t}, history, \"case: success\")\n\t\t\t\treturn events.PersistedBlob{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte{4, 5},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockMergeContinueAsNewReplicationTasksFn: func(updateMode persistence.UpdateWorkflowMode, currentWorkflow *persistence.WorkflowMutation, newWorkflow *persistence.WorkflowSnapshot) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.Equal(t, &persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\tHistorySize: 5,\n\t\t\t\t\t},\n\t\t\t\t}, currentWorkflow, \"case: success\")\n\t\t\t\tassert.Equal(t, &persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\tHistorySize: 2,\n\t\t\t\t\t},\n\t\t\t\t}, newWorkflow, \"case: success\")\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionEventReapplyFn: func(updateMode persistence.UpdateWorkflowMode, currentEvents []*persistence.WorkflowEvents, newEvents []*persistence.WorkflowEvents) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.Equal(t, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, currentEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, newEvents, \"case: success\")\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionFn: func(_ context.Context, req *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\t\t\t\tassert.Equal(t, &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\t\tMode: persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 5,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tNewWorkflowSnapshot: &persistence.WorkflowSnapshot{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tWorkflowRequestMode: persistence.CreateWorkflowRequestModeReplicated,\n\t\t\t\t\tDomainName:          \"test-domain\",\n\t\t\t\t}, req, \"case: success\")\n\t\t\t\treturn &persistence.UpdateWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(currentWorkflow *persistence.WorkflowMutation, currentEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\tHistorySize: 5,\n\t\t\t\t\t},\n\t\t\t\t}, currentWorkflow, \"case: success\")\n\t\t\t\tassert.Equal(t, events.PersistedBlobs{\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, currentEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, false, persistenceError)\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(newWorkflow *persistence.WorkflowSnapshot, newEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\tHistorySize: 2,\n\t\t\t\t\t},\n\t\t\t\t}, newWorkflow, \"case: success\")\n\t\t\t\tassert.Equal(t, events.PersistedBlobs{\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, newEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, false, persistenceError, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitWorkflowHistoryStatsFn: func(domainName string, size int, count int) {\n\t\t\t\tassert.Equal(t, 5, size, \"case: success\")\n\t\t\t\tassert.Equal(t, 19, count, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitSessionUpdateStatsFn: func(domainName string, stats *persistence.MutableStateUpdateSessionStats) {\n\t\t\t\tassert.Equal(t, &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t}, stats, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitLargeWorkflowShardIDStatsFn: func(blobSize int64, oldHistoryCount int64, oldHistorySize int64, newHistoryCount int64) {\n\t\t\t\tassert.Equal(t, int64(5), blobSize, \"case: success\")\n\t\t\t\tassert.Equal(t, int64(10), oldHistoryCount, \"case: success\")\n\t\t\t\tassert.Equal(t, int64(0), oldHistorySize, \"case: success\")\n\t\t\t\tassert.Equal(t, int64(11), newHistoryCount, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitWorkflowCompletionStatsFn: func(domainName string, workflowType string, workflowID string, runID string, taskList string, lastEvent *types.HistoryEvent) {\n\t\t\t\tassert.Equal(t, &types.HistoryEvent{\n\t\t\t\t\tID: 123,\n\t\t\t\t}, lastEvent, \"case: success\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockNewMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockEngine := engine.NewMockEngine(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache, mockMutableState, mockNewMutableState, mockEngine)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tlogger:                                testlogger.New(t),\n\t\t\t\tshard:                                 mockShard,\n\t\t\t\tmutableState:                          mockMutableState,\n\t\t\t\tstats:                                 &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient:                         metrics.NewNoopMetricsClient(),\n\t\t\t\tpersistNonStartWorkflowBatchEventsFn:  tc.mockPersistNonStartWorkflowBatchEventsFn,\n\t\t\t\tpersistStartWorkflowBatchEventsFn:     tc.mockPersistStartWorkflowBatchEventsFn,\n\t\t\t\tupdateWorkflowExecutionFn:             tc.mockUpdateWorkflowExecutionFn,\n\t\t\t\tnotifyTasksFromWorkflowMutationFn:     tc.mockNotifyTasksFromWorkflowMutationFn,\n\t\t\t\tnotifyTasksFromWorkflowSnapshotFn:     tc.mockNotifyTasksFromWorkflowSnapshotFn,\n\t\t\t\temitSessionUpdateStatsFn:              tc.mockEmitSessionUpdateStatsFn,\n\t\t\t\temitWorkflowHistoryStatsFn:            tc.mockEmitWorkflowHistoryStatsFn,\n\t\t\t\tmergeContinueAsNewReplicationTasksFn:  tc.mockMergeContinueAsNewReplicationTasksFn,\n\t\t\t\tupdateWorkflowExecutionEventReapplyFn: tc.mockUpdateWorkflowExecutionEventReapplyFn,\n\t\t\t\temitLargeWorkflowShardIDStatsFn:       tc.mockEmitLargeWorkflowShardIDStatsFn,\n\t\t\t\temitWorkflowCompletionStatsFn:         tc.mockEmitWorkflowCompletionStatsFn,\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionWithNew(context.Background(), time.Unix(0, 0), tc.updateMode, tc.newContext, mockNewMutableState, tc.currentWorkflowTransactionPolicy, tc.newWorkflowTransactionPolicy, tc.workflowRequestMode)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConflictResolveWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                               string\n\t\tconflictResolveMode                                persistence.ConflictResolveWorkflowMode\n\t\tnewContext                                         Context\n\t\tcurrentContext                                     Context\n\t\tcurrentWorkflowTransactionPolicy                   *TransactionPolicy\n\t\tmockSetup                                          func(*shard.MockContext, *cache.MockDomainCache, *MockMutableState, *MockMutableState, *MockMutableState, *engine.MockEngine)\n\t\tmockPersistNonStartWorkflowBatchEventsFn           func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tmockPersistStartWorkflowBatchEventsFn              func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error)\n\t\tmockUpdateWorkflowExecutionFn                      func(context.Context, *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)\n\t\tmockNotifyTasksFromWorkflowMutationFn              func(*persistence.WorkflowMutation, events.PersistedBlobs, bool)\n\t\tmockNotifyTasksFromWorkflowSnapshotFn              func(*persistence.WorkflowSnapshot, events.PersistedBlobs, bool)\n\t\tmockEmitSessionUpdateStatsFn                       func(string, *persistence.MutableStateUpdateSessionStats)\n\t\tmockEmitWorkflowHistoryStatsFn                     func(string, int, int)\n\t\tmockEmitLargeWorkflowShardIDStatsFn                func(int64, int64, int64, int64)\n\t\tmockEmitWorkflowCompletionStatsFn                  func(string, string, string, string, string, *types.HistoryEvent)\n\t\tmockMergeContinueAsNewReplicationTasksFn           func(persistence.UpdateWorkflowMode, *persistence.WorkflowMutation, *persistence.WorkflowSnapshot) error\n\t\tmockConflictResolveWorkflowExecutionEventReapplyFn func(persistence.ConflictResolveWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error\n\t\twantErr                                            bool\n\t\tassertErr                                          func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"resetMutableState CloseTransactionAsSnapshot failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistNonStartWorkflowEvents failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"newMutableState CloseTransactionAsSnapshot failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"both reset workflow and new workflow generate workflow requests\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tEnableStrongIdempotencySanityCheck: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t\t})\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t\t\tRequestID:   \"test\",\n\t\t\t\t\t\t\tVersion:     1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t\t\tRequestID:   \"test\",\n\t\t\t\t\t\t\tVersion:     1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"workflow requests are only expected to be generated from either reset workflow or continue-as-new workflow for ConflictResolveWorkflowExecution\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistStartWorkflowEvents failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"currentMutableState CloseTransactionAsMutation failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(nil, nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"currentMutableState generates workflow requests\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tEnableStrongIdempotencySanityCheck: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t\t})\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tWorkflowRequests: []*persistence.WorkflowRequest{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\t\t\tRequestID:   \"test\",\n\t\t\t\t\t\t\tVersion:     1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tif history.BranchToken[0] == 1 {\n\t\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t\t}\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, &types.InternalServiceError{Message: \"workflow requests are not expected from current workflow for ConflictResolveWorkflowExecution\"}, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"currentMutableState persistNonStartWorkflowEvents failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tif history.BranchToken[0] == 1 {\n\t\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t\t}\n\t\t\t\treturn events.PersistedBlob{}, errors.New(\"some error\")\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"conflictResolveEventReapply failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockConflictResolveWorkflowExecutionEventReapplyFn: func(persistence.ConflictResolveWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ConflictResolveWorkflowExecution failed\",\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(context.Context, *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\treturn events.PersistedBlob{}, nil\n\t\t\t},\n\t\t\tmockConflictResolveWorkflowExecutionEventReapplyFn: func(persistence.ConflictResolveWorkflowMode, []*persistence.WorkflowEvents, []*persistence.WorkflowEvents) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(currentWorkflow *persistence.WorkflowMutation, currentEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError, \"case: ConflictResolveWorkflowExecution failed\")\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(newWorkflow *persistence.WorkflowSnapshot, newEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, true, persistenceError, \"case: ConflictResolveWorkflowExecution failed\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                \"ConflictResolveWorkflowExecution success\",\n\t\t\tconflictResolveMode: persistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\t\tnewContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentContext: &contextImpl{\n\t\t\t\tstats:         &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\t\t},\n\t\t\tcurrentWorkflowTransactionPolicy: TransactionPolicyActive.Ptr(),\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResetMutableState *MockMutableState, mockNewMutableState *MockMutableState, mockMutableState *MockMutableState, mockEngine *engine.MockEngine) {\n\t\t\t\tmockResetMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockNewMutableState.EXPECT().CloseTransactionAsSnapshot(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockMutableState.EXPECT().CloseTransactionAsMutation(gomock.Any(), gomock.Any()).Return(&persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id0\",\n\t\t\t\t\t},\n\t\t\t\t}, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\t\tmockShard.EXPECT().ConflictResolveWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.ConflictResolveWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockResetMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"123\"),\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 1,\n\t\t\t\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmockResetMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateCompleted, persistence.WorkflowCloseStatusCompleted)\n\t\t\t\tmockShard.EXPECT().GetEngine().Return(mockEngine)\n\t\t\t\tmockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any())\n\t\t\t\tmockResetMutableState.EXPECT().GetLastFirstEventID().Return(int64(123))\n\t\t\t\tmockResetMutableState.EXPECT().GetNextEventID().Return(int64(456))\n\t\t\t\tmockResetMutableState.EXPECT().GetPreviousStartedEventID().Return(int64(789))\n\t\t\t\tmockResetMutableState.EXPECT().GetNextEventID().Return(int64(1111))\n\t\t\t\tmockResetMutableState.EXPECT().GetCompletionEvent(gomock.Any()).Return(&types.HistoryEvent{\n\t\t\t\t\tID: 123,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tmockPersistNonStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tif history.BranchToken[0] == 1 {\n\t\t\t\t\tassert.Equal(t, &persistence.WorkflowEvents{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t}, history, \"case: success\")\n\t\t\t\t\treturn events.PersistedBlob{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil\n\t\t\t\t}\n\n\t\t\t\tassert.Equal(t, &persistence.WorkflowEvents{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tBranchToken: []byte{5, 6},\n\t\t\t\t}, history, \"case: success\")\n\t\t\t\treturn events.PersistedBlob{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte{1, 2},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockPersistStartWorkflowBatchEventsFn: func(_ context.Context, history *persistence.WorkflowEvents) (events.PersistedBlob, error) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowEvents{\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t}, history, \"case: success\")\n\t\t\t\treturn events.PersistedBlob{\n\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\tData: []byte{3, 2},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockConflictResolveWorkflowExecutionEventReapplyFn: func(mode persistence.ConflictResolveWorkflowMode, resetEvents []*persistence.WorkflowEvents, newEvents []*persistence.WorkflowEvents) error {\n\t\t\t\tassert.Equal(t, persistence.ConflictResolveWorkflowModeUpdateCurrent, mode, \"case: success\")\n\t\t\t\tassert.Equal(t, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t}, resetEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, []*persistence.WorkflowEvents{\n\t\t\t\t\t{\n\t\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tID: constants.FirstEventID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tBranchToken: []byte{4},\n\t\t\t\t\t},\n\t\t\t\t}, newEvents, \"case: success\")\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowMutationFn: func(currentWorkflow *persistence.WorkflowMutation, currentEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tassert.Equal(t, &persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id0\",\n\t\t\t\t\t},\n\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\tHistorySize: 2,\n\t\t\t\t\t},\n\t\t\t\t}, currentWorkflow, \"case: success\")\n\t\t\t\tassert.Equal(t, events.PersistedBlobs{\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{3, 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, currentEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, false, persistenceError, \"case: success\")\n\t\t\t},\n\t\t\tmockNotifyTasksFromWorkflowSnapshotFn: func(newWorkflow *persistence.WorkflowSnapshot, newEvents events.PersistedBlobs, persistenceError bool) {\n\t\t\t\tif newWorkflow.ExecutionInfo.RunID == \"test-run-id\" {\n\t\t\t\t\tassert.Equal(t, &persistence.WorkflowSnapshot{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t\tState:      persistence.WorkflowStateCompleted,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 5,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, newWorkflow, \"case: success\")\n\t\t\t\t} else {\n\t\t\t\t\tassert.Equal(t, &persistence.WorkflowSnapshot{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, newWorkflow, \"case: success\")\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, events.PersistedBlobs{\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2, 3, 4, 5},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{3, 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\t\t\tData: []byte{1, 2},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, newEvents, \"case: success\")\n\t\t\t\tassert.Equal(t, false, persistenceError, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitWorkflowHistoryStatsFn: func(domainName string, size int, count int) {\n\t\t\t\tassert.Equal(t, 5, size, \"case: success\")\n\t\t\t\tassert.Equal(t, 1110, count, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitSessionUpdateStatsFn: func(domainName string, stats *persistence.MutableStateUpdateSessionStats) {\n\t\t\t\tassert.Equal(t, &persistence.MutableStateUpdateSessionStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t}, stats, \"case: success\")\n\t\t\t},\n\t\t\tmockEmitWorkflowCompletionStatsFn: func(domainName string, workflowType string, workflowID string, runID string, taskList string, lastEvent *types.HistoryEvent) {\n\t\t\t\tassert.Equal(t, &types.HistoryEvent{\n\t\t\t\t\tID: 123,\n\t\t\t\t}, lastEvent, \"case: success\")\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockResetMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockNewMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockEngine := engine.NewMockEngine(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache, mockResetMutableState, mockNewMutableState, mockMutableState, mockEngine)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tlogger:                               testlogger.New(t),\n\t\t\t\tshard:                                mockShard,\n\t\t\t\tstats:                                &persistence.ExecutionStats{},\n\t\t\t\tmetricsClient:                        metrics.NewNoopMetricsClient(),\n\t\t\t\tpersistNonStartWorkflowBatchEventsFn: tc.mockPersistNonStartWorkflowBatchEventsFn,\n\t\t\t\tpersistStartWorkflowBatchEventsFn:    tc.mockPersistStartWorkflowBatchEventsFn,\n\t\t\t\tupdateWorkflowExecutionFn:            tc.mockUpdateWorkflowExecutionFn,\n\t\t\t\tnotifyTasksFromWorkflowMutationFn:    tc.mockNotifyTasksFromWorkflowMutationFn,\n\t\t\t\tnotifyTasksFromWorkflowSnapshotFn:    tc.mockNotifyTasksFromWorkflowSnapshotFn,\n\t\t\t\temitSessionUpdateStatsFn:             tc.mockEmitSessionUpdateStatsFn,\n\t\t\t\temitWorkflowHistoryStatsFn:           tc.mockEmitWorkflowHistoryStatsFn,\n\t\t\t\tmergeContinueAsNewReplicationTasksFn: tc.mockMergeContinueAsNewReplicationTasksFn,\n\t\t\t\tconflictResolveEventReapplyFn:        tc.mockConflictResolveWorkflowExecutionEventReapplyFn,\n\t\t\t\temitLargeWorkflowShardIDStatsFn:      tc.mockEmitLargeWorkflowShardIDStatsFn,\n\t\t\t\temitWorkflowCompletionStatsFn:        tc.mockEmitWorkflowCompletionStatsFn,\n\t\t\t}\n\t\t\terr := ctx.ConflictResolveWorkflowExecution(context.Background(), time.Unix(0, 0), tc.conflictResolveMode, mockResetMutableState, tc.newContext, mockNewMutableState, tc.currentContext, mockMutableState, tc.currentWorkflowTransactionPolicy)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReapplyEvents(t *testing.T) {\n\ttestCases := []struct {\n\t\tname         string\n\t\teventBatches []*persistence.WorkflowEvents\n\t\tmockSetup    func(*shard.MockContext, *cache.MockDomainCache, *resource.Test, *engine.MockEngine, *activecluster.MockManager)\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname:         \"empty input\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{},\n\t\t\twantErr:      false,\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, _ *resource.Test, _ *engine.MockEngine, _ *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"domain is pending active\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, _ *resource.Test, _ *engine.MockEngine, _ *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(cache.NewDomainCacheEntryForTest(nil, nil, true, nil, 0, common.Ptr(int64(1)), 0, 0, 0), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"domainID/workflowID mismatch\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, _ *resource.Test, _ *engine.MockEngine, _ *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(cache.NewDomainCacheEntryForTest(nil, nil, true, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"no signal events\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, _ *resource.Test, _ *engine.MockEngine, _ *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(cache.NewDomainCacheEntryForTest(nil, nil, true, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"lookup workflow error\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResource *resource.Test, mockEngine *engine.MockEngine, mockActiveClusterManager *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(\n\t\t\t\t\tcache.NewGlobalDomainCacheEntryForTest(nil, nil, &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, 0),\n\t\t\t\t\tnil)\n\t\t\t\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager)\n\t\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").Return(\n\t\t\t\t\tnil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - active-active apply to current cluster\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResource *resource.Test, mockEngine *engine.MockEngine, mockActiveClusterManager *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(\n\t\t\t\t\tcache.NewGlobalDomainCacheEntryForTest(nil, nil, &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, 0),\n\t\t\t\t\tnil)\n\t\t\t\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager)\n\t\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").Return(\n\t\t\t\t\t&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata)\n\t\t\t\tmockShard.EXPECT().GetEngine().Return(mockEngine)\n\t\t\t\tmockEngine.EXPECT().ReapplyEvents(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\", []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - active-active apply to remote cluster\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResource *resource.Test, mockEngine *engine.MockEngine, mockActiveClusterManager *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(\n\t\t\t\t\tcache.NewGlobalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"test-domain\"}, nil, &persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}, 0), nil)\n\t\t\t\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager)\n\t\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").Return(\n\t\t\t\t\t&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata)\n\t\t\t\tmockShard.EXPECT().GetService().Return(mockResource).Times(2)\n\t\t\t\tmockResource.RemoteAdminClient.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - apply to current cluster\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, _ *resource.Test, mockEngine *engine.MockEngine, mockActiveClusterManager *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(cache.NewGlobalDomainCacheEntryForTest(nil, nil, &persistence.DomainReplicationConfig{ActiveClusterName: cluster.TestCurrentClusterName}, 0), nil)\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata)\n\t\t\t\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager)\n\t\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").Return(\n\t\t\t\t\t&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().GetEngine().Return(mockEngine)\n\t\t\t\tmockEngine.EXPECT().ReapplyEvents(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\", []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success - apply to remote cluster\",\n\t\t\teventBatches: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockDomainCache *cache.MockDomainCache, mockResource *resource.Test, mockEngine *engine.MockEngine, mockActiveClusterManager *activecluster.MockManager) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"test-domain-id\").Return(cache.NewGlobalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"test-domain\"}, nil, &persistence.DomainReplicationConfig{ActiveClusterName: cluster.TestAlternativeClusterName}, 0), nil)\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata)\n\t\t\t\tmockShard.EXPECT().GetService().Return(mockResource).Times(2)\n\t\t\t\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager)\n\t\t\t\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"test-domain-id\", \"test-workflow-id\", \"test-run-id\").Return(\n\t\t\t\t\t&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\t\t\t}, nil)\n\t\t\t\tmockResource.RemoteAdminClient.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockEngine := engine.NewMockEngine(mockCtrl)\n\t\t\tmockActiveClusterManager := activecluster.NewMockManager(mockCtrl)\n\t\t\ttestResource := resource.NewTest(t, mockCtrl, metrics.Common)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockDomainCache, testResource, mockEngine, mockActiveClusterManager)\n\t\t\t}\n\t\t\tctx := &contextImpl{\n\t\t\t\tshard: mockShard,\n\t\t\t}\n\t\t\terr := ctx.ReapplyEvents(tc.eventBatches)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetWorkflowExecutionWithRetry(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\trequest   *persistence.GetWorkflowExecutionRequest\n\t\tmockSetup func(*shard.MockContext, *log.MockLogger, clock.MockedTimeSource)\n\t\twant      *persistence.GetWorkflowExecutionResponse\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), &persistence.GetWorkflowExecutionRequest{\n\t\t\t\t\tRangeID: 100,\n\t\t\t\t}).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tMutableStateStats: &persistence.MutableStateStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"entity not exists error\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, err, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"shard closed error recent\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &shard.ErrShardClosed{\n\t\t\t\t\tClosedAt: timeSource.Now().Add(-shard.TimeBeforeShardClosedIsError / 2),\n\t\t\t\t})\n\t\t\t\t// We do _not_ expect a log call\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.ErrorAs(t, err, new(*shard.ErrShardClosed))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"shard closed error\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\terr := &shard.ErrShardClosed{\n\t\t\t\t\tClosedAt: timeSource.Now().Add(-shard.TimeBeforeShardClosedIsError * 2),\n\t\t\t\t}\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, err)\n\t\t\t\texpectLog(mockLogger, err)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.ErrorAs(t, err, new(*shard.ErrShardClosed))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"non retryable error\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t\texpectLog(mockLogger, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, errors.New(\"some error\"), err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"retry succeeds\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tRangeID: 100,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockLogger *log.MockLogger, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{})\n\t\t\t\tmockShard.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tMutableStateStats: &persistence.MutableStateStats{\n\t\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\twant: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tMutableStateStats: &persistence.MutableStateStats{\n\t\t\t\t\tMutableStateSize: 123,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockLogger := log.NewMockLogger(gomock.NewController(t))\n\t\t\tmockLogger.EXPECT().Helper().Return(mockLogger)\n\t\t\ttimeSource := clock.NewMockedTimeSource()\n\t\t\tmockShard.EXPECT().GetTimeSource().Return(timeSource).AnyTimes()\n\t\t\tpolicy := backoff.NewExponentialRetryPolicy(time.Millisecond)\n\t\t\tpolicy.SetMaximumAttempts(1)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockLogger, timeSource)\n\t\t\t}\n\t\t\tresp, err := getWorkflowExecutionWithRetry(context.Background(), mockShard, mockLogger, policy, tc.request)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc expectLog(mockLogger *log.MockLogger, err error) *gomock.Call {\n\treturn mockLogger.EXPECT().Error(\n\t\t\"Persistent fetch operation failure\",\n\t\t[]tag.Tag{\n\t\t\ttag.StoreOperationGetWorkflowExecution,\n\t\t\ttag.Error(err),\n\t\t})\n}\n\nfunc TestLoadWorkflowExecutionWithTaskVersion(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                 string\n\t\tmockSetup                            func(*shard.MockContext, *MockMutableState, *cache.MockDomainCache)\n\t\tmockGetWorkflowExecutionFn           func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error)\n\t\tmockEmitWorkflowExecutionStatsFn     func(string, *persistence.MutableStateStats, int64)\n\t\tmockUpdateWorkflowExecutionWithNewFn func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\twantErr                              bool\n\t}{\n\t\t{\n\t\t\tname: \"domain cache failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"getWorkflowExecutionFn failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{\n\t\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\tnil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tmockGetWorkflowExecutionFn: func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\treturn nil, errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"StartTransaction failed\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{\n\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t}, nil, true, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t\tmockMutableState.EXPECT().Load(gomock.Any(), gomock.Any()).Return(errors.New(\"some error\"))\n\t\t\t\tmockMutableState.EXPECT().StartTransaction(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tmockGetWorkflowExecutionFn: func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\treturn &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 123,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockEmitWorkflowExecutionStatsFn: func(domainName string, stats *persistence.MutableStateStats, size int64) {\n\t\t\t\tassert.Equal(t, \"test-domain\", domainName)\n\t\t\t\tassert.Equal(t, int64(123), size)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"do not need to flush\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{\n\t\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0), nil)\n\t\t\t\tmockMutableState.EXPECT().Load(gomock.Any(), gomock.Any()).Return(errors.New(\"some error\"))\n\t\t\t\tmockMutableState.EXPECT().StartTransaction(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil)\n\t\t\t},\n\t\t\tmockGetWorkflowExecutionFn: func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\treturn &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 123,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockEmitWorkflowExecutionStatsFn: func(domainName string, stats *persistence.MutableStateStats, size int64) {\n\t\t\t\tassert.Equal(t, \"test-domain\", domainName)\n\t\t\t\tassert.Equal(t, int64(123), size)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"flush buffered events\",\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{\n\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t}, nil, true, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t\tmockMutableState.EXPECT().Load(gomock.Any(), gomock.Any()).Return(errors.New(\"some error\"))\n\t\t\t\tmockMutableState.EXPECT().StartTransaction(gomock.Any(), gomock.Any(), gomock.Any()).Return(true, nil)\n\t\t\t\tmockMutableState.EXPECT().StartTransaction(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil)\n\t\t\t\tmockShard.EXPECT().GetTimeSource().Return(clock.NewMockedTimeSource())\n\t\t\t},\n\t\t\tmockGetWorkflowExecutionFn: func(context.Context, *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\t\t\t\treturn &persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{\n\t\t\t\t\t\t\tHistorySize: 123,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t},\n\t\t\tmockEmitWorkflowExecutionStatsFn: func(domainName string, stats *persistence.MutableStateStats, size int64) {\n\t\t\t\tassert.Equal(t, \"test-domain\", domainName)\n\t\t\t\tassert.Equal(t, int64(123), size)\n\t\t\t},\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tif tc.mockSetup != nil {\n\t\t\t\ttc.mockSetup(mockShard, mockMutableState, mockDomainCache)\n\t\t\t}\n\n\t\t\tctx := &contextImpl{\n\t\t\t\tshard:  mockShard,\n\t\t\t\tlogger: testlogger.New(t),\n\t\t\t\tcreateMutableStateFn: func(shard.Context, log.Logger, *cache.DomainCacheEntry) MutableState {\n\t\t\t\t\treturn mockMutableState\n\t\t\t\t},\n\t\t\t\tgetWorkflowExecutionFn:           tc.mockGetWorkflowExecutionFn,\n\t\t\t\temitWorkflowExecutionStatsFn:     tc.mockEmitWorkflowExecutionStatsFn,\n\t\t\t\tupdateWorkflowExecutionWithNewFn: tc.mockUpdateWorkflowExecutionWithNewFn,\n\t\t\t}\n\t\t\tgot, err := ctx.LoadWorkflowExecutionWithTaskVersion(context.Background(), 123)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, mockMutableState, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionAsActive(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                 string\n\t\tmockUpdateWorkflowExecutionWithNewFn func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\twantErr                              bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.Nil(t, newContext)\n\t\t\t\tassert.Nil(t, newMutableState)\n\t\t\t\tassert.Equal(t, TransactionPolicyActive, currentPolicy)\n\t\t\t\tassert.Nil(t, newPolicy)\n\t\t\t\tassert.Equal(t, persistence.CreateWorkflowRequestModeNew, workflowRequestMode)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error case\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, _ persistence.CreateWorkflowRequestMode) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctx := &contextImpl{\n\t\t\t\tupdateWorkflowExecutionWithNewFn: tc.mockUpdateWorkflowExecutionWithNewFn,\n\t\t\t\tlogger:                           testlogger.New(t),\n\t\t\t\tworkflowExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionAsActive(context.Background(), time.Now())\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionWithNewAsActive(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                 string\n\t\tmockUpdateWorkflowExecutionWithNewFn func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\twantErr                              bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.NotNil(t, newContext)\n\t\t\t\tassert.NotNil(t, newMutableState)\n\t\t\t\tassert.Equal(t, TransactionPolicyActive, currentPolicy)\n\t\t\t\tassert.Equal(t, TransactionPolicyActive.Ptr(), newPolicy)\n\t\t\t\tassert.Equal(t, persistence.CreateWorkflowRequestModeNew, workflowRequestMode)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error case\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, _ persistence.CreateWorkflowRequestMode) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctx := &contextImpl{\n\t\t\t\tupdateWorkflowExecutionWithNewFn: tc.mockUpdateWorkflowExecutionWithNewFn,\n\t\t\t\tlogger:                           testlogger.New(t),\n\t\t\t\tworkflowExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionWithNewAsActive(context.Background(), time.Now(), &contextImpl{}, &mutableStateBuilder{})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionAsPassive(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                 string\n\t\tmockUpdateWorkflowExecutionWithNewFn func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\twantErr                              bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.Nil(t, newContext)\n\t\t\t\tassert.Nil(t, newMutableState)\n\t\t\t\tassert.Equal(t, TransactionPolicyPassive, currentPolicy)\n\t\t\t\tassert.Nil(t, newPolicy)\n\t\t\t\tassert.Equal(t, persistence.CreateWorkflowRequestModeReplicated, workflowRequestMode)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error case\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, _ persistence.CreateWorkflowRequestMode) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctx := &contextImpl{\n\t\t\t\tupdateWorkflowExecutionWithNewFn: tc.mockUpdateWorkflowExecutionWithNewFn,\n\t\t\t\tlogger:                           testlogger.New(t),\n\t\t\t\tworkflowExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionAsPassive(context.Background(), time.Now())\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateWorkflowExecutionWithNewAsPassive(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                 string\n\t\tmockUpdateWorkflowExecutionWithNewFn func(context.Context, time.Time, persistence.UpdateWorkflowMode, Context, MutableState, TransactionPolicy, *TransactionPolicy, persistence.CreateWorkflowRequestMode) error\n\t\twantErr                              bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, workflowRequestMode persistence.CreateWorkflowRequestMode) error {\n\t\t\t\tassert.Equal(t, persistence.UpdateWorkflowModeUpdateCurrent, updateMode)\n\t\t\t\tassert.NotNil(t, newContext)\n\t\t\t\tassert.NotNil(t, newMutableState)\n\t\t\t\tassert.Equal(t, TransactionPolicyPassive, currentPolicy)\n\t\t\t\tassert.Equal(t, TransactionPolicyPassive.Ptr(), newPolicy)\n\t\t\t\tassert.Equal(t, persistence.CreateWorkflowRequestModeReplicated, workflowRequestMode)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error case\",\n\t\t\tmockUpdateWorkflowExecutionWithNewFn: func(_ context.Context, _ time.Time, updateMode persistence.UpdateWorkflowMode, newContext Context, newMutableState MutableState, currentPolicy TransactionPolicy, newPolicy *TransactionPolicy, _ persistence.CreateWorkflowRequestMode) error {\n\t\t\t\treturn errors.New(\"some error\")\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctx := &contextImpl{\n\t\t\t\tupdateWorkflowExecutionWithNewFn: tc.mockUpdateWorkflowExecutionWithNewFn,\n\t\t\t\tlogger:                           testlogger.New(t),\n\t\t\t\tworkflowExecution: types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := ctx.UpdateWorkflowExecutionWithNewAsPassive(context.Background(), time.Now(), &contextImpl{}, &mutableStateBuilder{})\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateWorkflowRequestsAndMode(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\trequests []*persistence.WorkflowRequest\n\t\tmode     persistence.CreateWorkflowRequestMode\n\t\twantErr  bool\n\t}{\n\t\t{\n\t\t\tname:    \"Success - replicated mode\",\n\t\t\tmode:    persistence.CreateWorkflowRequestModeReplicated,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - new mode\",\n\t\t\trequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeCancel,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmode:    persistence.CreateWorkflowRequestModeNew,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - new mode, signal with start\",\n\t\t\trequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmode:    persistence.CreateWorkflowRequestModeNew,\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"too many requests\",\n\t\t\trequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmode:    persistence.CreateWorkflowRequestModeNew,\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"too many requests\",\n\t\t\trequests: []*persistence.WorkflowRequest{\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t\t\t\tRequestID:   \"abc\",\n\t\t\t\t\tVersion:     1,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmode:    persistence.CreateWorkflowRequestModeNew,\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\terr := validateWorkflowRequestsAndMode(tc.requests, tc.mode)\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/context_util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (c *contextImpl) emitLargeWorkflowShardIDStats(blobSize int64, oldHistoryCount int64, oldHistorySize int64, newHistoryCount int64) {\n\tif c.shard.GetConfig().EnableShardIDMetrics() {\n\t\tshardID := c.shard.GetShardID()\n\n\t\tblobSizeWarn := min(int64(c.shard.GetConfig().LargeShardHistoryBlobMetricThreshold()), int64(c.shard.GetConfig().BlobSizeLimitWarn(c.GetDomainName())))\n\t\t// check if blob size is larger than threshold in Dynamic config if so alert on it every time\n\t\tif blobSize > blobSizeWarn {\n\t\t\tc.logger.SampleInfo(\"Workflow writing a large blob\", c.shard.GetConfig().SampleLoggingRate(), tag.WorkflowDomainName(c.GetDomainName()),\n\t\t\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()), tag.ShardID(c.shard.GetShardID()), tag.WorkflowRunID(c.workflowExecution.GetRunID()))\n\t\t\tc.metricsClient.Scope(metrics.LargeExecutionBlobShardScope, metrics.ShardIDTag(shardID), metrics.DomainTag(c.GetDomainName())).IncCounter(metrics.LargeHistoryBlobCount)\n\t\t}\n\n\t\thistoryCountWarn := min(int64(c.shard.GetConfig().LargeShardHistoryEventMetricThreshold()), int64(c.shard.GetConfig().HistoryCountLimitWarn(c.GetDomainName())))\n\t\t// check if the new history count is greater than our threshold and only count/log it once when it passes it\n\t\t// this seems to double count and I can't figure out why but should be ok to get a rough idea and identify bad actors\n\t\tif oldHistoryCount < historyCountWarn && newHistoryCount >= historyCountWarn {\n\t\t\tc.logger.Warn(\"Workflow history event count is reaching dangerous levels\", tag.WorkflowDomainName(c.GetDomainName()),\n\t\t\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()), tag.ShardID(c.shard.GetShardID()), tag.WorkflowRunID(c.workflowExecution.GetRunID()))\n\t\t\tc.metricsClient.Scope(metrics.LargeExecutionCountShardScope, metrics.ShardIDTag(shardID), metrics.DomainTag(c.GetDomainName())).IncCounter(metrics.LargeHistoryEventCount)\n\t\t}\n\n\t\thistorySizeWarn := min(int64(c.shard.GetConfig().LargeShardHistorySizeMetricThreshold()), int64(c.shard.GetConfig().HistorySizeLimitWarn(c.GetDomainName())))\n\t\t// check if the new history size is greater than our threshold and only count/log it once when it passes it\n\t\tif oldHistorySize < historySizeWarn && c.stats.HistorySize >= historySizeWarn {\n\t\t\tc.logger.Warn(\"Workflow history event size is reaching dangerous levels\", tag.WorkflowDomainName(c.GetDomainName()),\n\t\t\t\ttag.WorkflowID(c.workflowExecution.GetWorkflowID()), tag.ShardID(c.shard.GetShardID()), tag.WorkflowRunID(c.workflowExecution.GetRunID()))\n\t\t\tc.metricsClient.Scope(metrics.LargeExecutionSizeShardScope, metrics.ShardIDTag(shardID), metrics.DomainTag(c.GetDomainName())).IncCounter(metrics.LargeHistorySizeCount)\n\t\t}\n\t}\n}\n\nfunc emitWorkflowHistoryStats(\n\tmetricsClient metrics.Client,\n\tdomainName string,\n\thistorySize int,\n\thistoryCount int,\n) {\n\n\tsizeScope := metricsClient.Scope(metrics.ExecutionSizeStatsScope, metrics.DomainTag(domainName))\n\tcountScope := metricsClient.Scope(metrics.ExecutionCountStatsScope, metrics.DomainTag(domainName))\n\n\tsizeScope.RecordTimer(metrics.HistorySize, time.Duration(historySize))\n\tcountScope.RecordTimer(metrics.HistoryCount, time.Duration(historyCount))\n}\n\nfunc emitWorkflowExecutionStats(\n\tmetricsClient metrics.Client,\n\tdomainName string,\n\tstats *persistence.MutableStateStats,\n\texecutionInfoHistorySize int64,\n) {\n\n\tif stats == nil {\n\t\treturn\n\t}\n\n\tsizeScope := metricsClient.Scope(metrics.ExecutionSizeStatsScope, metrics.DomainTag(domainName))\n\tcountScope := metricsClient.Scope(metrics.ExecutionCountStatsScope, metrics.DomainTag(domainName))\n\n\tsizeScope.RecordTimer(metrics.HistorySize, time.Duration(executionInfoHistorySize))\n\tsizeScope.RecordTimer(metrics.MutableStateSize, time.Duration(stats.MutableStateSize))\n\tsizeScope.RecordTimer(metrics.ExecutionInfoSize, time.Duration(stats.MutableStateSize))\n\tsizeScope.RecordTimer(metrics.ActivityInfoSize, time.Duration(stats.ActivityInfoSize))\n\tsizeScope.RecordTimer(metrics.TimerInfoSize, time.Duration(stats.TimerInfoSize))\n\tsizeScope.RecordTimer(metrics.ChildInfoSize, time.Duration(stats.ChildInfoSize))\n\tsizeScope.RecordTimer(metrics.SignalInfoSize, time.Duration(stats.SignalInfoSize))\n\tsizeScope.RecordTimer(metrics.BufferedEventsSize, time.Duration(stats.BufferedEventsSize))\n\n\tcountScope.RecordTimer(metrics.ActivityInfoCount, time.Duration(stats.ActivityInfoCount))\n\tcountScope.RecordTimer(metrics.TimerInfoCount, time.Duration(stats.TimerInfoCount))\n\tcountScope.RecordTimer(metrics.ChildInfoCount, time.Duration(stats.ChildInfoCount))\n\tcountScope.RecordTimer(metrics.SignalInfoCount, time.Duration(stats.SignalInfoCount))\n\tcountScope.RecordTimer(metrics.RequestCancelInfoCount, time.Duration(stats.RequestCancelInfoCount))\n\tcountScope.RecordTimer(metrics.BufferedEventsCount, time.Duration(stats.BufferedEventsCount))\n}\n\nfunc emitSessionUpdateStats(\n\tmetricsClient metrics.Client,\n\tdomainName string,\n\tstats *persistence.MutableStateUpdateSessionStats,\n) {\n\n\tif stats == nil {\n\t\treturn\n\t}\n\n\tsizeScope := metricsClient.Scope(metrics.SessionSizeStatsScope, metrics.DomainTag(domainName))\n\tcountScope := metricsClient.Scope(metrics.SessionCountStatsScope, metrics.DomainTag(domainName))\n\n\tsizeScope.RecordTimer(metrics.MutableStateSize, time.Duration(stats.MutableStateSize))\n\tsizeScope.RecordTimer(metrics.ExecutionInfoSize, time.Duration(stats.ExecutionInfoSize))\n\tsizeScope.RecordTimer(metrics.ActivityInfoSize, time.Duration(stats.ActivityInfoSize))\n\tsizeScope.RecordTimer(metrics.TimerInfoSize, time.Duration(stats.TimerInfoSize))\n\tsizeScope.RecordTimer(metrics.ChildInfoSize, time.Duration(stats.ChildInfoSize))\n\tsizeScope.RecordTimer(metrics.SignalInfoSize, time.Duration(stats.SignalInfoSize))\n\tsizeScope.RecordTimer(metrics.BufferedEventsSize, time.Duration(stats.BufferedEventsSize))\n\n\tcountScope.RecordTimer(metrics.ActivityInfoCount, time.Duration(stats.ActivityInfoCount))\n\tcountScope.RecordTimer(metrics.TimerInfoCount, time.Duration(stats.TimerInfoCount))\n\tcountScope.RecordTimer(metrics.ChildInfoCount, time.Duration(stats.ChildInfoCount))\n\tcountScope.RecordTimer(metrics.SignalInfoCount, time.Duration(stats.SignalInfoCount))\n\tcountScope.RecordTimer(metrics.RequestCancelInfoCount, time.Duration(stats.RequestCancelInfoCount))\n\tcountScope.RecordTimer(metrics.DeleteActivityInfoCount, time.Duration(stats.DeleteActivityInfoCount))\n\tcountScope.RecordTimer(metrics.DeleteTimerInfoCount, time.Duration(stats.DeleteTimerInfoCount))\n\tcountScope.RecordTimer(metrics.DeleteChildInfoCount, time.Duration(stats.DeleteChildInfoCount))\n\tcountScope.RecordTimer(metrics.DeleteSignalInfoCount, time.Duration(stats.DeleteSignalInfoCount))\n\tcountScope.RecordTimer(metrics.DeleteRequestCancelInfoCount, time.Duration(stats.DeleteRequestCancelInfoCount))\n\tcountScope.RecordTimer(metrics.TransferTasksCount, time.Duration(stats.TaskCountByCategory[persistence.HistoryTaskCategoryTransfer]))\n\tcountScope.RecordTimer(metrics.TimerTasksCount, time.Duration(stats.TaskCountByCategory[persistence.HistoryTaskCategoryTimer]))\n\tcountScope.RecordTimer(metrics.ReplicationTasksCount, time.Duration(stats.TaskCountByCategory[persistence.HistoryTaskCategoryReplication]))\n\tcountScope.IncCounter(metrics.UpdateWorkflowExecutionCount)\n}\n\nfunc emitWorkflowCompletionStats(\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tdomainName string,\n\tworkflowType string,\n\tworkflowID string,\n\trunID string,\n\ttaskList string,\n\tevent *types.HistoryEvent,\n) {\n\n\tif event.EventType == nil {\n\t\treturn\n\t}\n\n\tscope := metricsClient.Scope(\n\t\tmetrics.WorkflowCompletionStatsScope,\n\t\tmetrics.DomainTag(domainName),\n\t\tmetrics.WorkflowTypeTag(workflowType),\n\t\tmetrics.TaskListTag(taskList),\n\t)\n\n\tswitch *event.EventType {\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tscope.IncCounter(metrics.WorkflowSuccessCount)\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tscope.IncCounter(metrics.WorkflowCancelCount)\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tscope.IncCounter(metrics.WorkflowFailedCount)\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tscope.IncCounter(metrics.WorkflowTimeoutCount)\n\t\tlogger.Info(\"workflow execution timed out\",\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tscope.IncCounter(metrics.WorkflowTerminateCount)\n\t\tlogger.Info(\"workflow terminated\",\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tscope.IncCounter(metrics.WorkflowContinuedAsNew)\n\t\tlogger.Debug(\"workflow continued as new\",\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\tdefault:\n\t\tscope.IncCounter(metrics.WorkflowCompletedUnknownType)\n\t\tlogger.Warn(\"Workflow completed with an unknown event type\",\n\t\t\ttag.WorkflowEventType(event.EventType.String()),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/context_util_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\t_testDomain = \"testDomain\"\n)\n\nfunc createTestConfig() *config.Config {\n\treturn &config.Config{\n\t\tEnableShardIDMetrics:                  dynamicproperties.GetBoolPropertyFn(true),\n\t\tLargeShardHistoryBlobMetricThreshold:  dynamicproperties.GetIntPropertyFn(1024 * 1024 * 5),    // 5 MB\n\t\tBlobSizeLimitWarn:                     func(domainName string) int { return 1024 * 1024 * 5 }, // 5 MB\n\t\tLargeShardHistoryEventMetricThreshold: dynamicproperties.GetIntPropertyFn(1500),\n\t\tHistoryCountLimitWarn:                 func(domainName string) int { return 1500 },\n\t\tLargeShardHistorySizeMetricThreshold:  dynamicproperties.GetIntPropertyFn(1024 * 1024 * 2),    // 2 MB\n\t\tHistorySizeLimitWarn:                  func(domainName string) int { return 1024 * 1024 * 2 }, // 2 MB\n\t\tSampleLoggingRate:                     dynamicproperties.GetIntPropertyFn(100),\n\t}\n}\n\nfunc TestEmitLargeWorkflowShardIDStats(t *testing.T) {\n\ttests := []struct {\n\t\tname               string\n\t\tblobSize           int64\n\t\toldHistoryCount    int64\n\t\toldHistorySize     int64\n\t\tnewHistoryCount    int64\n\t\tshardConfig        *config.Config\n\t\tstats              *persistence.ExecutionStats\n\t\tloggerExpectations func(logger *log.MockLogger)\n\t\tassertMetrics      func(t *testing.T, snapshot tally.Snapshot)\n\t}{\n\t\t{\n\t\t\tname:            \"Blob size exceeds threshold\",\n\t\t\tblobSize:        1024 * 1024 * 10, // 10 MB\n\t\t\toldHistoryCount: 1000,\n\t\t\toldHistorySize:  1024 * 500, // 0.5 MB\n\t\t\tnewHistoryCount: 100,\n\t\t\tstats: &persistence.ExecutionStats{\n\t\t\t\tHistorySize: 0,\n\t\t\t},\n\t\t\tshardConfig: createTestConfig(),\n\t\t\tloggerExpectations: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().SampleInfo(\"Workflow writing a large blob\", 100, gomock.Any())\n\t\t\t},\n\t\t\tassertMetrics: func(t *testing.T, snapshot tally.Snapshot) {\n\t\t\t\tcountersSnapshot := snapshot.Counters()\n\n\t\t\t\trequire.Contains(t, countersSnapshot, \"test.large_history_blob_count+domain=testDomain,operation=LargeExecutionBlobShard,shard_id=1\")\n\n\t\t\t\tassert.Equal(t, int64(1), countersSnapshot[\"test.large_history_blob_count+domain=testDomain,operation=LargeExecutionBlobShard,shard_id=1\"].Value())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"History old size and old count already exceeds threshold\",\n\t\t\tblobSize:        1024 * 1024 * 10, // 10 MB\n\t\t\toldHistoryCount: 1500,\n\t\t\toldHistorySize:  1024 * 1024 * 2, // 0.5 MB\n\t\t\tnewHistoryCount: 2000,\n\t\t\tstats: &persistence.ExecutionStats{\n\t\t\t\tHistorySize: 1024 * 1024 * 3,\n\t\t\t},\n\t\t\tshardConfig: createTestConfig(),\n\t\t\tloggerExpectations: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().SampleInfo(\"Workflow writing a large blob\", 100, gomock.Any())\n\t\t\t},\n\t\t\tassertMetrics: func(t *testing.T, snapshot tally.Snapshot) {\n\t\t\t\tcountersSnapshot := snapshot.Counters()\n\n\t\t\t\trequire.NotContains(t, countersSnapshot, \"test.large_history_event_count+domain=testDomain,operation=LargeExecutionCountShard,shard_id=1\")\n\t\t\t\trequire.NotContains(t, countersSnapshot, \"test.large_history_size_count+domain=testDomain,operation=LargeExecutionSizeShard,shard_id=1\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"History old size and old count already exceeds threshold\",\n\t\t\tblobSize:        1024 * 1024 * 10, // 10 MB\n\t\t\toldHistoryCount: 1500,\n\t\t\toldHistorySize:  1024 * 1024 * 2, // 0.5 MB\n\t\t\tnewHistoryCount: 2000,\n\t\t\tstats: &persistence.ExecutionStats{\n\t\t\t\tHistorySize: 1024 * 1024 * 3,\n\t\t\t},\n\t\t\tshardConfig: createTestConfig(),\n\t\t\tloggerExpectations: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().SampleInfo(\"Workflow writing a large blob\", 100, gomock.Any())\n\t\t\t},\n\t\t\tassertMetrics: func(t *testing.T, snapshot tally.Snapshot) {\n\t\t\t\tcountersSnapshot := snapshot.Counters()\n\n\t\t\t\trequire.NotContains(t, countersSnapshot, \"test.large_history_event_count+domain=testDomain,operation=LargeExecutionCountShard,shard_id=1\")\n\t\t\t\trequire.NotContains(t, countersSnapshot, \"test.large_history_size_count+domain=testDomain,operation=LargeExecutionSizeShard,shard_id=1\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"History count and size within threshold\",\n\t\t\tblobSize:        1024 * 500, // 0.5 MB\n\t\t\toldHistoryCount: 500,\n\t\t\toldHistorySize:  1024 * 1024, // 1 MB\n\t\t\tnewHistoryCount: 800,\n\t\t\tstats:           &persistence.ExecutionStats{},\n\t\t\tshardConfig:     createTestConfig(),\n\t\t},\n\t\t{\n\t\t\tname:            \"Metrics disabled\",\n\t\t\tblobSize:        1024 * 1024 * 10, // 10 MB\n\t\t\toldHistoryCount: 1000,\n\t\t\toldHistorySize:  1024 * 1024 * 2, // 2 MB\n\t\t\tnewHistoryCount: 2000,\n\t\t\tshardConfig: &config.Config{\n\t\t\t\tEnableShardIDMetrics: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmetricScope := tally.NewTestScope(\"test\", make(map[string]string))\n\t\t\tmockMetricsClient := metrics.NewClient(metricScope, metrics.History, metrics.HistogramMigration{})\n\t\t\tmockLogger := log.NewMockLogger(gomock.NewController(t))\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tcontext := &contextImpl{\n\t\t\t\tshard:         mockShard,\n\t\t\t\tmetricsClient: mockMetricsClient,\n\t\t\t\tlogger:        mockLogger,\n\t\t\t\tstats:         tc.stats,\n\t\t\t}\n\t\t\tmockShard.EXPECT().GetShardID().Return(1).AnyTimes()\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(_testDomain, nil).AnyTimes()\n\t\t\tmockShard.EXPECT().GetConfig().Return(tc.shardConfig).AnyTimes()\n\n\t\t\tif tc.loggerExpectations != nil {\n\t\t\t\ttc.loggerExpectations(mockLogger)\n\t\t\t}\n\n\t\t\tcontext.emitLargeWorkflowShardIDStats(tc.blobSize, tc.oldHistoryCount, tc.oldHistorySize, tc.newHistoryCount)\n\n\t\t\tif tc.assertMetrics != nil {\n\t\t\t\ttc.assertMetrics(t, metricScope.Snapshot())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestEmitWorkflowHistoryStats(t *testing.T) {\n\n\tmockMetricsClient := new(mocks.Client)\n\tmockScope := new(mocks.Scope)\n\n\tmockMetricsClient.On(\"Scope\", mock.Anything, mock.Anything).Return(mockScope)\n\n\tmockScope.On(\"RecordTimer\", mock.Anything, mock.Anything).Return()\n\n\tdomainName := \"testDomain\"\n\thistorySize := 2048\n\thistoryCount := 150\n\n\temitWorkflowHistoryStats(mockMetricsClient, domainName, historySize, historyCount)\n\n\tmockMetricsClient.AssertExpectations(t)\n\tmockScope.AssertExpectations(t)\n}\n\nfunc TestEmitWorkflowExecutionStats(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tdomainName  string\n\t\tstats       *persistence.MutableStateStats\n\t\thistorySize int64\n\t\texpectCalls bool\n\t}{\n\t\t{\n\t\t\tname:       \"With valid stats\",\n\t\t\tdomainName: \"testDomain\",\n\t\t\tstats: &persistence.MutableStateStats{\n\t\t\t\tMutableStateSize: 1024,\n\t\t\t\tActivityInfoSize: 256,\n\t\t\t\tTimerInfoSize:    128,\n\t\t\t},\n\t\t\thistorySize: 2048,\n\t\t\texpectCalls: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"Nil stats\",\n\t\t\tdomainName:  \"testDomain\",\n\t\t\tstats:       nil,\n\t\t\thistorySize: 2048,\n\t\t\texpectCalls: false,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\n\t\t\tmockMetricsClient := new(mocks.Client)\n\t\t\tmockScope := new(mocks.Scope)\n\n\t\t\tif tc.expectCalls {\n\t\t\t\tmockMetricsClient.On(\"Scope\", metrics.ExecutionSizeStatsScope, mock.Anything).Return(mockScope)\n\t\t\t\tmockMetricsClient.On(\"Scope\", metrics.ExecutionCountStatsScope, mock.Anything).Return(mockScope)\n\t\t\t\tmockScope.On(\"RecordTimer\", mock.AnythingOfType(\"metrics.MetricIdx\"), mock.AnythingOfType(\"time.Duration\")).Return().Times(14)\n\t\t\t} else {\n\t\t\t\tmockScope.AssertNotCalled(t, \"RecordTimer\")\n\t\t\t}\n\n\t\t\temitWorkflowExecutionStats(mockMetricsClient, tc.domainName, tc.stats, tc.historySize)\n\n\t\t\tmockMetricsClient.AssertExpectations(t)\n\t\t\tmockScope.AssertExpectations(t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/history_builder.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// HistoryBuilder builds and stores workflow history events\n\tHistoryBuilder struct {\n\t\ttransientHistory []*types.HistoryEvent\n\t\thistory          []*types.HistoryEvent\n\t\tmsBuilder        MutableState\n\t}\n)\n\n// NewHistoryBuilder creates a new history builder\nfunc NewHistoryBuilder(msBuilder MutableState) *HistoryBuilder {\n\treturn &HistoryBuilder{\n\t\ttransientHistory: []*types.HistoryEvent{},\n\t\thistory:          []*types.HistoryEvent{},\n\t\tmsBuilder:        msBuilder,\n\t}\n}\n\n// NewHistoryBuilderFromEvents creates a new history builder based on the given workflow history events\nfunc NewHistoryBuilderFromEvents(history []*types.HistoryEvent, msBuilder MutableState) *HistoryBuilder {\n\treturn &HistoryBuilder{\n\t\thistory:   history,\n\t\tmsBuilder: msBuilder,\n\t}\n}\n\n// AddWorkflowExecutionStartedEvent adds WorkflowExecutionStarted event to history\n// originalRunID is the runID when the WorkflowExecutionStarted event is written\n// firstRunID is the very first runID along the chain of ContinueAsNew and Reset\nfunc (b *HistoryBuilder) AddWorkflowExecutionStartedEvent(\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n\tpreviousExecution *persistence.WorkflowExecutionInfo,\n\tfirstRunID string,\n\toriginalRunID string,\n\tfirstScheduledTime time.Time,\n) *types.HistoryEvent {\n\n\tvar prevRunID string\n\tvar resetPoints *types.ResetPoints\n\tif previousExecution != nil {\n\t\tprevRunID = previousExecution.RunID\n\t\tresetPoints = previousExecution.AutoResetPoints\n\t}\n\n\trequest := startRequest.StartRequest\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionStarted)\n\n\tvar scheduledTime *time.Time\n\tif request.CronSchedule != \"\" {\n\t\t// first scheduled time is only necessary for cron workflows.\n\t\tscheduledTime = &firstScheduledTime\n\t}\n\tattributes := &types.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        request.WorkflowType,\n\t\tTaskList:                            request.TaskList,\n\t\tHeader:                              request.Header,\n\t\tInput:                               request.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: request.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      request.TaskStartToCloseTimeoutSeconds,\n\t\tContinuedExecutionRunID:             prevRunID,\n\t\tPrevAutoResetPoints:                 resetPoints,\n\t\tIdentity:                            request.Identity,\n\t\tRetryPolicy:                         request.RetryPolicy,\n\t\tAttempt:                             startRequest.GetAttempt(),\n\t\tExpirationTimestamp:                 startRequest.ExpirationTimestamp,\n\t\tCronSchedule:                        request.CronSchedule,\n\t\tCronOverlapPolicy:                   request.CronOverlapPolicy,\n\t\tLastCompletionResult:                startRequest.LastCompletionResult,\n\t\tContinuedFailureReason:              startRequest.ContinuedFailureReason,\n\t\tContinuedFailureDetails:             startRequest.ContinuedFailureDetails,\n\t\tInitiator:                           startRequest.ContinueAsNewInitiator,\n\t\tFirstDecisionTaskBackoffSeconds:     startRequest.FirstDecisionTaskBackoffSeconds,\n\t\tFirstExecutionRunID:                 firstRunID,\n\t\tFirstScheduleTime:                   scheduledTime,\n\t\tOriginalExecutionRunID:              originalRunID,\n\t\tMemo:                                request.Memo,\n\t\tSearchAttributes:                    request.SearchAttributes,\n\t\tJitterStartSeconds:                  request.JitterStartSeconds,\n\t\tPartitionConfig:                     startRequest.PartitionConfig,\n\t\tRequestID:                           request.RequestID,\n\t\tActiveClusterSelectionPolicy:        request.ActiveClusterSelectionPolicy,\n\t}\n\tif parentInfo := startRequest.ParentExecutionInfo; parentInfo != nil {\n\t\tattributes.ParentWorkflowDomainID = &parentInfo.DomainUUID\n\t\tattributes.ParentWorkflowDomain = &parentInfo.Domain\n\t\tattributes.ParentWorkflowExecution = parentInfo.Execution\n\t\tattributes.ParentInitiatedEventID = &parentInfo.InitiatedID\n\t}\n\tevent.WorkflowExecutionStartedEventAttributes = attributes\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddDecisionTaskScheduledEvent adds DecisionTaskScheduled event to history\nfunc (b *HistoryBuilder) AddDecisionTaskScheduledEvent(taskList string,\n\tstartToCloseTimeoutSeconds int32, attempt int64) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeDecisionTaskScheduled)\n\tevent.DecisionTaskScheduledEventAttributes = getDecisionTaskScheduledEventAttributes(taskList, startToCloseTimeoutSeconds, attempt)\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTransientDecisionTaskScheduledEvent adds transient DecisionTaskScheduled event\nfunc (b *HistoryBuilder) AddTransientDecisionTaskScheduledEvent(taskList string,\n\tstartToCloseTimeoutSeconds int32, attempt int64, timestamp int64) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEventWithTimestamp(types.EventTypeDecisionTaskScheduled, timestamp)\n\tevent.DecisionTaskScheduledEventAttributes = getDecisionTaskScheduledEventAttributes(taskList, startToCloseTimeoutSeconds, attempt)\n\n\treturn b.addTransientEvent(event)\n}\n\n// AddDecisionTaskStartedEvent adds DecisionTaskStarted event to history\nfunc (b *HistoryBuilder) AddDecisionTaskStartedEvent(scheduleEventID int64, requestID string,\n\tidentity string) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeDecisionTaskStarted)\n\tevent.DecisionTaskStartedEventAttributes = getDecisionTaskStartedEventAttributes(scheduleEventID, requestID, identity)\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTransientDecisionTaskStartedEvent adds transient DecisionTaskStarted event\nfunc (b *HistoryBuilder) AddTransientDecisionTaskStartedEvent(scheduleEventID int64, requestID string,\n\tidentity string, timestamp int64) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEventWithTimestamp(types.EventTypeDecisionTaskStarted, timestamp)\n\tevent.DecisionTaskStartedEventAttributes = getDecisionTaskStartedEventAttributes(scheduleEventID, requestID, identity)\n\n\treturn b.addTransientEvent(event)\n}\n\n// AddDecisionTaskCompletedEvent adds DecisionTaskCompleted event to history\nfunc (b *HistoryBuilder) AddDecisionTaskCompletedEvent(scheduleEventID, StartedEventID int64,\n\trequest *types.RespondDecisionTaskCompletedRequest) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeDecisionTaskCompleted)\n\tevent.DecisionTaskCompletedEventAttributes = &types.DecisionTaskCompletedEventAttributes{\n\t\tExecutionContext: request.ExecutionContext,\n\t\tScheduledEventID: scheduleEventID,\n\t\tStartedEventID:   StartedEventID,\n\t\tIdentity:         request.Identity,\n\t\tBinaryChecksum:   request.BinaryChecksum,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddDecisionTaskTimedOutEvent adds DecisionTaskTimedOut event to history\nfunc (b *HistoryBuilder) AddDecisionTaskTimedOutEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\ttimeoutType types.TimeoutType,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\treason string,\n\tcause types.DecisionTaskTimedOutCause,\n\tresetRequestID string,\n) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeDecisionTaskTimedOut)\n\tevent.DecisionTaskTimedOutEventAttributes = &types.DecisionTaskTimedOutEventAttributes{\n\t\tScheduledEventID: scheduleEventID,\n\t\tStartedEventID:   startedEventID,\n\t\tTimeoutType:      timeoutType.Ptr(),\n\t\tBaseRunID:        baseRunID,\n\t\tNewRunID:         newRunID,\n\t\tForkEventVersion: forkEventVersion,\n\t\tReason:           reason,\n\t\tCause:            cause.Ptr(),\n\t\tRequestID:        resetRequestID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddDecisionTaskFailedEvent adds DecisionTaskFailed event to history\nfunc (b *HistoryBuilder) AddDecisionTaskFailedEvent(attr types.DecisionTaskFailedEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeDecisionTaskFailed)\n\tevent.DecisionTaskFailedEventAttributes = &attr\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskScheduledEvent adds ActivityTaskScheduled event to history\nfunc (b *HistoryBuilder) AddActivityTaskScheduledEvent(decisionCompletedEventID int64,\n\tattributes *types.ScheduleActivityTaskDecisionAttributes) *types.HistoryEvent {\n\n\tvar domain *string\n\tif attributes.Domain != \"\" {\n\t\t// for backward compatibility\n\t\t// old releases will encounter issues if Domain field is a pointer to an empty string.\n\t\tdomain = common.StringPtr(attributes.Domain)\n\t}\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskScheduled)\n\tevent.ActivityTaskScheduledEventAttributes = &types.ActivityTaskScheduledEventAttributes{\n\t\tActivityID:                    attributes.ActivityID,\n\t\tActivityType:                  attributes.ActivityType,\n\t\tTaskList:                      attributes.TaskList,\n\t\tHeader:                        attributes.Header,\n\t\tInput:                         attributes.Input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(common.Int32Default(attributes.ScheduleToCloseTimeoutSeconds)),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(common.Int32Default(attributes.ScheduleToStartTimeoutSeconds)),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(common.Int32Default(attributes.StartToCloseTimeoutSeconds)),\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(common.Int32Default(attributes.HeartbeatTimeoutSeconds)),\n\t\tDecisionTaskCompletedEventID:  decisionCompletedEventID,\n\t\tRetryPolicy:                   attributes.RetryPolicy,\n\t\tDomain:                        domain,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskStartedEvent adds ActivityTaskStarted event to history\nfunc (b *HistoryBuilder) AddActivityTaskStartedEvent(\n\tscheduleEventID int64,\n\tattempt int32,\n\trequestID string,\n\tidentity string,\n\tlastFailureReason string,\n\tlastFailureDetails []byte,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\tevent.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\tScheduledEventID:   scheduleEventID,\n\t\tAttempt:            attempt,\n\t\tIdentity:           identity,\n\t\tRequestID:          requestID,\n\t\tLastFailureReason:  common.StringPtr(lastFailureReason),\n\t\tLastFailureDetails: lastFailureDetails,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskCompletedEvent adds ActivityTaskCompleted event to history\nfunc (b *HistoryBuilder) AddActivityTaskCompletedEvent(scheduleEventID, startedEventID int64,\n\trequest *types.RespondActivityTaskCompletedRequest) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskCompleted)\n\tevent.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\tResult:           request.Result,\n\t\tScheduledEventID: scheduleEventID,\n\t\tStartedEventID:   startedEventID,\n\t\tIdentity:         request.Identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskFailedEvent adds ActivityTaskFailed event to history\nfunc (b *HistoryBuilder) AddActivityTaskFailedEvent(scheduleEventID, StartedEventID int64,\n\trequest *types.RespondActivityTaskFailedRequest) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskFailed)\n\tevent.ActivityTaskFailedEventAttributes = &types.ActivityTaskFailedEventAttributes{\n\t\tReason:           common.StringPtr(common.StringDefault(request.Reason)),\n\t\tDetails:          request.Details,\n\t\tScheduledEventID: scheduleEventID,\n\t\tStartedEventID:   StartedEventID,\n\t\tIdentity:         request.Identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskTimedOutEvent adds ActivityTaskTimedOut event to history\nfunc (b *HistoryBuilder) AddActivityTaskTimedOutEvent(\n\tscheduleEventID,\n\tstartedEventID int64,\n\ttimeoutType types.TimeoutType,\n\tlastHeartBeatDetails []byte,\n\tlastFailureReason string,\n\tlastFailureDetail []byte,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\tevent.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\tScheduledEventID:   scheduleEventID,\n\t\tStartedEventID:     startedEventID,\n\t\tTimeoutType:        &timeoutType,\n\t\tDetails:            lastHeartBeatDetails,\n\t\tLastFailureReason:  common.StringPtr(lastFailureReason),\n\t\tLastFailureDetails: lastFailureDetail,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddCompletedWorkflowEvent adds WorkflowExecutionCompleted event to history\nfunc (b *HistoryBuilder) AddCompletedWorkflowEvent(decisionCompletedEventID int64,\n\tattributes *types.CompleteWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionCompleted)\n\tevent.WorkflowExecutionCompletedEventAttributes = &types.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       attributes.Result,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddFailWorkflowEvent adds WorkflowExecutionFailed event to history\nfunc (b *HistoryBuilder) AddFailWorkflowEvent(decisionCompletedEventID int64,\n\tattributes *types.FailWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionFailed)\n\tevent.WorkflowExecutionFailedEventAttributes = &types.WorkflowExecutionFailedEventAttributes{\n\t\tReason:                       common.StringPtr(common.StringDefault(attributes.Reason)),\n\t\tDetails:                      attributes.Details,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTimeoutWorkflowEvent adds WorkflowExecutionTimedout event to history\nfunc (b *HistoryBuilder) AddTimeoutWorkflowEvent() *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionTimedOut)\n\tevent.WorkflowExecutionTimedOutEventAttributes = &types.WorkflowExecutionTimedOutEventAttributes{\n\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddWorkflowExecutionTerminatedEvent add WorkflowExecutionTerminated event to history\nfunc (b *HistoryBuilder) AddWorkflowExecutionTerminatedEvent(\n\treason string,\n\tdetails []byte,\n\tidentity string,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionTerminated)\n\tevent.WorkflowExecutionTerminatedEventAttributes = &types.WorkflowExecutionTerminatedEventAttributes{\n\t\tReason:   reason,\n\t\tDetails:  details,\n\t\tIdentity: identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddContinuedAsNewEvent adds WorkflowExecutionContinuedAsNew event to history\nfunc (b *HistoryBuilder) AddContinuedAsNewEvent(decisionCompletedEventID int64, newRunID string,\n\tattributes *types.ContinueAsNewWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\n\tinitiator := attributes.Initiator\n\tif initiator == nil {\n\t\tinitiator = types.ContinueAsNewInitiatorDecider.Ptr()\n\t}\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionContinuedAsNew)\n\tevent.WorkflowExecutionContinuedAsNewEventAttributes = &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\tNewExecutionRunID:                   newRunID,\n\t\tWorkflowType:                        attributes.WorkflowType,\n\t\tTaskList:                            attributes.TaskList,\n\t\tHeader:                              attributes.Header,\n\t\tInput:                               attributes.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: attributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      attributes.TaskStartToCloseTimeoutSeconds,\n\t\tDecisionTaskCompletedEventID:        decisionCompletedEventID,\n\t\tBackoffStartIntervalInSeconds:       common.Int32Ptr(attributes.GetBackoffStartIntervalInSeconds()),\n\t\tInitiator:                           initiator,\n\t\tFailureReason:                       attributes.FailureReason,\n\t\tFailureDetails:                      attributes.FailureDetails,\n\t\tLastCompletionResult:                attributes.LastCompletionResult,\n\t\tMemo:                                attributes.Memo,\n\t\tSearchAttributes:                    attributes.SearchAttributes,\n\t\tJitterStartSeconds:                  attributes.JitterStartSeconds,\n\t\tCronOverlapPolicy:                   attributes.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        attributes.ActiveClusterSelectionPolicy,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTimerStartedEvent adds TimerStart event to history\nfunc (b *HistoryBuilder) AddTimerStartedEvent(decisionCompletedEventID int64,\n\trequest *types.StartTimerDecisionAttributes) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeTimerStarted)\n\tevent.TimerStartedEventAttributes = &types.TimerStartedEventAttributes{\n\t\tTimerID:                      request.TimerID,\n\t\tStartToFireTimeoutSeconds:    request.StartToFireTimeoutSeconds,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTimerFiredEvent adds TimerFired event to history\nfunc (b *HistoryBuilder) AddTimerFiredEvent(\n\tStartedEventID int64,\n\tTimerID string,\n) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeTimerFired)\n\tevent.TimerFiredEventAttributes = &types.TimerFiredEventAttributes{\n\t\tTimerID:        TimerID,\n\t\tStartedEventID: StartedEventID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskCancelRequestedEvent add ActivityTaskCancelRequested event to history\nfunc (b *HistoryBuilder) AddActivityTaskCancelRequestedEvent(decisionCompletedEventID int64,\n\tactivityID string) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskCancelRequested)\n\tevent.ActivityTaskCancelRequestedEventAttributes = &types.ActivityTaskCancelRequestedEventAttributes{\n\t\tActivityID:                   activityID,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddRequestCancelActivityTaskFailedEvent add RequestCancelActivityTaskFailed event to history\nfunc (b *HistoryBuilder) AddRequestCancelActivityTaskFailedEvent(decisionCompletedEventID int64,\n\tactivityID string, cause string) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeRequestCancelActivityTaskFailed)\n\tevent.RequestCancelActivityTaskFailedEventAttributes = &types.RequestCancelActivityTaskFailedEventAttributes{\n\t\tActivityID:                   activityID,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t\tCause:                        cause,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddActivityTaskCanceledEvent adds ActivityTaskCanceled event to history\nfunc (b *HistoryBuilder) AddActivityTaskCanceledEvent(scheduleEventID, startedEventID int64,\n\tlatestCancelRequestedEventID int64, details []byte, identity string) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeActivityTaskCanceled)\n\tevent.ActivityTaskCanceledEventAttributes = &types.ActivityTaskCanceledEventAttributes{\n\t\tScheduledEventID:             scheduleEventID,\n\t\tStartedEventID:               startedEventID,\n\t\tLatestCancelRequestedEventID: latestCancelRequestedEventID,\n\t\tDetails:                      details,\n\t\tIdentity:                     identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddTimerCanceledEvent adds TimerCanceled event to history\nfunc (b *HistoryBuilder) AddTimerCanceledEvent(startedEventID int64,\n\tdecisionTaskCompletedEventID int64, timerID string, identity string) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeTimerCanceled)\n\tevent.TimerCanceledEventAttributes = &types.TimerCanceledEventAttributes{\n\t\tStartedEventID:               startedEventID,\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tTimerID:                      timerID,\n\t\tIdentity:                     identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddCancelTimerFailedEvent adds CancelTimerFailed event to history\nfunc (b *HistoryBuilder) AddCancelTimerFailedEvent(timerID string, decisionTaskCompletedEventID int64,\n\tcause string, identity string) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeCancelTimerFailed)\n\tevent.CancelTimerFailedEventAttributes = &types.CancelTimerFailedEventAttributes{\n\t\tTimerID:                      timerID,\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tCause:                        cause,\n\t\tIdentity:                     identity,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddWorkflowExecutionCancelRequestedEvent adds WorkflowExecutionCancelRequested event to history\nfunc (b *HistoryBuilder) AddWorkflowExecutionCancelRequestedEvent(\n\tcause string,\n\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionCancelRequested)\n\tevent.WorkflowExecutionCancelRequestedEventAttributes = &types.WorkflowExecutionCancelRequestedEventAttributes{\n\t\tCause:                     cause,\n\t\tIdentity:                  request.CancelRequest.Identity,\n\t\tExternalInitiatedEventID:  request.ExternalInitiatedEventID,\n\t\tExternalWorkflowExecution: request.ExternalWorkflowExecution,\n\t\tRequestID:                 request.CancelRequest.RequestID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddWorkflowExecutionCanceledEvent adds WorkflowExecutionCanceled event to history\nfunc (b *HistoryBuilder) AddWorkflowExecutionCanceledEvent(decisionTaskCompletedEventID int64,\n\tattributes *types.CancelWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionCanceled)\n\tevent.WorkflowExecutionCanceledEventAttributes = &types.WorkflowExecutionCanceledEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tDetails:                      attributes.Details,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddRequestCancelExternalWorkflowExecutionInitiatedEvent adds RequestCancelExternalWorkflowExecutionInitiated event to history\nfunc (b *HistoryBuilder) AddRequestCancelExternalWorkflowExecutionInitiatedEvent(decisionTaskCompletedEventID int64,\n\trequest *types.RequestCancelExternalWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeRequestCancelExternalWorkflowExecutionInitiated)\n\tevent.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes = &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tDomain:                       request.Domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: request.WorkflowID,\n\t\t\tRunID:      request.RunID,\n\t\t},\n\t\tControl:           request.Control,\n\t\tChildWorkflowOnly: request.ChildWorkflowOnly,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddRequestCancelExternalWorkflowExecutionFailedEvent adds RequestCancelExternalWorkflowExecutionFailed event to history\nfunc (b *HistoryBuilder) AddRequestCancelExternalWorkflowExecutionFailedEvent(decisionTaskCompletedEventID, initiatedEventID int64,\n\tdomain, workflowID, runID string, cause types.CancelExternalWorkflowExecutionFailedCause) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeRequestCancelExternalWorkflowExecutionFailed)\n\tevent.RequestCancelExternalWorkflowExecutionFailedEventAttributes = &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tInitiatedEventID:             initiatedEventID,\n\t\tDomain:                       domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tCause: cause.Ptr(),\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddExternalWorkflowExecutionCancelRequested adds ExternalWorkflowExecutionCancelRequested event to history\nfunc (b *HistoryBuilder) AddExternalWorkflowExecutionCancelRequested(initiatedEventID int64,\n\tdomain, workflowID, runID string) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeExternalWorkflowExecutionCancelRequested)\n\tevent.ExternalWorkflowExecutionCancelRequestedEventAttributes = &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{\n\t\tInitiatedEventID: initiatedEventID,\n\t\tDomain:           domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddSignalExternalWorkflowExecutionInitiatedEvent adds SignalExternalWorkflowExecutionInitiated event to history\nfunc (b *HistoryBuilder) AddSignalExternalWorkflowExecutionInitiatedEvent(decisionTaskCompletedEventID int64,\n\tattributes *types.SignalExternalWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeSignalExternalWorkflowExecutionInitiated)\n\tevent.SignalExternalWorkflowExecutionInitiatedEventAttributes = &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tDomain:                       attributes.Domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: attributes.Execution.WorkflowID,\n\t\t\tRunID:      attributes.Execution.RunID,\n\t\t},\n\t\tSignalName:        attributes.GetSignalName(),\n\t\tInput:             attributes.Input,\n\t\tControl:           attributes.Control,\n\t\tChildWorkflowOnly: attributes.ChildWorkflowOnly,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddUpsertWorkflowSearchAttributesEvent adds UpsertWorkflowSearchAttributes event to history\nfunc (b *HistoryBuilder) AddUpsertWorkflowSearchAttributesEvent(\n\tdecisionTaskCompletedEventID int64,\n\tattributes *types.UpsertWorkflowSearchAttributesDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeUpsertWorkflowSearchAttributes)\n\tevent.UpsertWorkflowSearchAttributesEventAttributes = &types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tSearchAttributes:             attributes.GetSearchAttributes(),\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddSignalExternalWorkflowExecutionFailedEvent adds SignalExternalWorkflowExecutionFailed event to history\nfunc (b *HistoryBuilder) AddSignalExternalWorkflowExecutionFailedEvent(decisionTaskCompletedEventID, initiatedEventID int64,\n\tdomain, workflowID, runID string, control []byte, cause types.SignalExternalWorkflowExecutionFailedCause) *types.HistoryEvent {\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeSignalExternalWorkflowExecutionFailed)\n\tevent.SignalExternalWorkflowExecutionFailedEventAttributes = &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\tDecisionTaskCompletedEventID: decisionTaskCompletedEventID,\n\t\tInitiatedEventID:             initiatedEventID,\n\t\tDomain:                       domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tCause:   cause.Ptr(),\n\t\tControl: control,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddExternalWorkflowExecutionSignaled adds ExternalWorkflowExecutionSignaled event to history\nfunc (b *HistoryBuilder) AddExternalWorkflowExecutionSignaled(initiatedEventID int64,\n\tdomain, workflowID, runID string, control []byte) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeExternalWorkflowExecutionSignaled)\n\tevent.ExternalWorkflowExecutionSignaledEventAttributes = &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\tInitiatedEventID: initiatedEventID,\n\t\tDomain:           domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tControl: control,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddMarkerRecordedEvent adds MarkerRecorded event to history\nfunc (b *HistoryBuilder) AddMarkerRecordedEvent(decisionCompletedEventID int64,\n\tattributes *types.RecordMarkerDecisionAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeMarkerRecorded)\n\tevent.MarkerRecordedEventAttributes = &types.MarkerRecordedEventAttributes{\n\t\tMarkerName:                   attributes.MarkerName,\n\t\tDetails:                      attributes.Details,\n\t\tDecisionTaskCompletedEventID: decisionCompletedEventID,\n\t\tHeader:                       attributes.Header,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddWorkflowExecutionSignaledEvent adds WorkflowExecutionSignaled event to history\nfunc (b *HistoryBuilder) AddWorkflowExecutionSignaledEvent(\n\tsignalName string,\n\tinput []byte,\n\tidentity string,\n\trequestID string,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeWorkflowExecutionSignaled)\n\tevent.WorkflowExecutionSignaledEventAttributes = &types.WorkflowExecutionSignaledEventAttributes{\n\t\tSignalName: signalName,\n\t\tInput:      input,\n\t\tIdentity:   identity,\n\t\tRequestID:  requestID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddStartChildWorkflowExecutionInitiatedEvent adds ChildWorkflowExecutionInitiated event to history\nfunc (b *HistoryBuilder) AddStartChildWorkflowExecutionInitiatedEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.StartChildWorkflowExecutionDecisionAttributes,\n\ttargetDomainName string,\n) *types.HistoryEvent {\n\n\tdomain := attributes.Domain\n\tif domain == \"\" {\n\t\tdomain = targetDomainName\n\t}\n\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeStartChildWorkflowExecutionInitiated)\n\tevent.StartChildWorkflowExecutionInitiatedEventAttributes = &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              domain,\n\t\tWorkflowID:                          attributes.WorkflowID,\n\t\tWorkflowType:                        attributes.WorkflowType,\n\t\tTaskList:                            attributes.TaskList,\n\t\tHeader:                              attributes.Header,\n\t\tInput:                               attributes.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: attributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      attributes.TaskStartToCloseTimeoutSeconds,\n\t\tControl:                             attributes.Control,\n\t\tDecisionTaskCompletedEventID:        decisionCompletedEventID,\n\t\tWorkflowIDReusePolicy:               attributes.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         attributes.RetryPolicy,\n\t\tCronSchedule:                        attributes.CronSchedule,\n\t\tMemo:                                attributes.Memo,\n\t\tSearchAttributes:                    attributes.SearchAttributes,\n\t\tParentClosePolicy:                   attributes.GetParentClosePolicy().Ptr(),\n\t\tCronOverlapPolicy:                   attributes.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        attributes.ActiveClusterSelectionPolicy,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionStartedEvent adds ChildWorkflowExecutionStarted event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionStartedEvent(\n\tdomain string,\n\texecution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType,\n\tinitiatedID int64,\n\theader *types.Header,\n) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionStarted)\n\tevent.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\tDomain:            domain,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tHeader:            header,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddStartChildWorkflowExecutionFailedEvent adds ChildWorkflowExecutionFailed event to history\nfunc (b *HistoryBuilder) AddStartChildWorkflowExecutionFailedEvent(initiatedID int64,\n\tcause types.ChildWorkflowExecutionFailedCause,\n\tinitiatedEventAttributes *types.StartChildWorkflowExecutionInitiatedEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeStartChildWorkflowExecutionFailed)\n\tevent.StartChildWorkflowExecutionFailedEventAttributes = &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:                       initiatedEventAttributes.Domain,\n\t\tWorkflowID:                   initiatedEventAttributes.WorkflowID,\n\t\tWorkflowType:                 initiatedEventAttributes.WorkflowType,\n\t\tInitiatedEventID:             initiatedID,\n\t\tDecisionTaskCompletedEventID: initiatedEventAttributes.DecisionTaskCompletedEventID,\n\t\tControl:                      initiatedEventAttributes.Control,\n\t\tCause:                        cause.Ptr(),\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionCompletedEvent adds ChildWorkflowExecutionCompleted event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionCompletedEvent(domain string, execution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType, initiatedID, startedID int64,\n\tcompletedAttributes *types.WorkflowExecutionCompletedEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\tevent.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\tDomain:            domain,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tStartedEventID:    startedID,\n\t\tResult:            completedAttributes.Result,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionFailedEvent adds ChildWorkflowExecutionFailed event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionFailedEvent(domain string, execution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType, initiatedID, startedID int64,\n\tfailedAttributes *types.WorkflowExecutionFailedEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionFailed)\n\tevent.ChildWorkflowExecutionFailedEventAttributes = &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\tDomain:            domain,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tStartedEventID:    startedID,\n\t\tReason:            common.StringPtr(common.StringDefault(failedAttributes.Reason)),\n\t\tDetails:           failedAttributes.Details,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionCanceledEvent adds ChildWorkflowExecutionCanceled event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionCanceledEvent(domain string, execution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType, initiatedID, startedID int64,\n\tcanceledAttributes *types.WorkflowExecutionCanceledEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCanceled)\n\tevent.ChildWorkflowExecutionCanceledEventAttributes = &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\tDomain:            domain,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tStartedEventID:    startedID,\n\t\tDetails:           canceledAttributes.Details,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionTerminatedEvent adds ChildWorkflowExecutionTerminated event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionTerminatedEvent(domain string, execution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType, initiatedID, startedID int64,\n\tterminatedAttributes *types.WorkflowExecutionTerminatedEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTerminated)\n\tevent.ChildWorkflowExecutionTerminatedEventAttributes = &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\tDomain:            domain,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tStartedEventID:    startedID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\n// AddChildWorkflowExecutionTimedOutEvent adds ChildWorkflowExecutionTimedOut event to history\nfunc (b *HistoryBuilder) AddChildWorkflowExecutionTimedOutEvent(domain string, execution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType, initiatedID, startedID int64,\n\ttimedOutAttributes *types.WorkflowExecutionTimedOutEventAttributes) *types.HistoryEvent {\n\tevent := b.msBuilder.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTimedOut)\n\tevent.ChildWorkflowExecutionTimedOutEventAttributes = &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\tDomain:            domain,\n\t\tTimeoutType:       timedOutAttributes.TimeoutType,\n\t\tWorkflowExecution: execution,\n\t\tWorkflowType:      workflowType,\n\t\tInitiatedEventID:  initiatedID,\n\t\tStartedEventID:    startedID,\n\t}\n\n\treturn b.addEventToHistory(event)\n}\n\nfunc (b *HistoryBuilder) addEventToHistory(event *types.HistoryEvent) *types.HistoryEvent {\n\tb.history = append(b.history, event)\n\treturn event\n}\n\nfunc (b *HistoryBuilder) addTransientEvent(event *types.HistoryEvent) *types.HistoryEvent {\n\tb.transientHistory = append(b.transientHistory, event)\n\treturn event\n}\n\nfunc newDecisionTaskScheduledEventWithInfo(eventID, timestamp int64, taskList string, startToCloseTimeoutSeconds int32, attempt int64) *types.HistoryEvent {\n\tevent := createNewHistoryEvent(eventID, types.EventTypeDecisionTaskScheduled, timestamp)\n\tevent.DecisionTaskScheduledEventAttributes = getDecisionTaskScheduledEventAttributes(taskList, startToCloseTimeoutSeconds, attempt)\n\treturn event\n}\n\nfunc newDecisionTaskStartedEventWithInfo(eventID, timestamp int64, scheduledEventID int64, requestID string, identity string) *types.HistoryEvent {\n\tevent := createNewHistoryEvent(eventID, types.EventTypeDecisionTaskStarted, timestamp)\n\tevent.DecisionTaskStartedEventAttributes = getDecisionTaskStartedEventAttributes(scheduledEventID, requestID, identity)\n\treturn event\n}\n\nfunc createNewHistoryEvent(eventID int64, eventType types.EventType, timestamp int64) *types.HistoryEvent {\n\treturn &types.HistoryEvent{\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(timestamp),\n\t\tEventType: eventType.Ptr(),\n\t}\n}\n\nfunc getDecisionTaskScheduledEventAttributes(taskList string, startToCloseTimeoutSeconds int32, attempt int64) *types.DecisionTaskScheduledEventAttributes {\n\treturn &types.DecisionTaskScheduledEventAttributes{\n\t\tTaskList: &types.TaskList{\n\t\t\tName: taskList,\n\t\t},\n\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(startToCloseTimeoutSeconds),\n\t\tAttempt:                    attempt,\n\t}\n}\n\nfunc getDecisionTaskStartedEventAttributes(scheduledEventID int64, requestID string, identity string) *types.DecisionTaskStartedEventAttributes {\n\treturn &types.DecisionTaskStartedEventAttributes{\n\t\tScheduledEventID: scheduledEventID,\n\t\tIdentity:         identity,\n\t\tRequestID:        requestID,\n\t}\n}\n\n// GetHistory gets workflow history stored inside history builder\nfunc (b *HistoryBuilder) GetHistory() *types.History {\n\thistory := types.History{Events: b.history}\n\treturn &history\n}\n"
  },
  {
    "path": "service/history/execution/history_builder_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\thistoryBuilderSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tmockShard       *shard.TestContext\n\t\tmockEventsCache *events.MockCache\n\t\tmockDomainCache *cache.MockDomainCache\n\n\t\tdomainID          string\n\t\tdomainName        string\n\t\tdomainEntry       *cache.DomainCacheEntry\n\t\ttargetDomainID    string\n\t\ttargetDomainName  string\n\t\ttargetDomainEntry *cache.DomainCacheEntry\n\t\tmsBuilder         MutableState\n\t\tbuilder           *HistoryBuilder\n\t\tlogger            log.Logger\n\t}\n)\n\nfunc TestHistoryBuilderSuite(t *testing.T) {\n\ts := new(historyBuilderSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *historyBuilderSuite) SetupTest() {\n\t// Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.logger = log.NewNoop()\n\n\ts.domainEntry = constants.TestLocalDomainEntry\n\ts.domainID = s.domainEntry.GetInfo().ID\n\ts.domainName = s.domainEntry.GetInfo().Name\n\n\ts.targetDomainEntry = constants.TestGlobalTargetDomainEntry\n\ts.targetDomainID = s.targetDomainEntry.GetInfo().ID\n\ts.targetDomainName = s.targetDomainEntry.GetInfo().Name\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockEventsCache = s.mockShard.MockEventsCache\n\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(s.domainID, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.domainID).Return(s.domainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(s.targetDomainName).Return(s.targetDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(s.targetDomainName).Return(s.targetDomainID, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.targetDomainID).Return(s.targetDomainEntry, nil).AnyTimes()\n\ts.mockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.msBuilder = NewMutableStateBuilder(s.mockShard, s.logger, s.domainEntry)\n\ts.builder = NewHistoryBuilder(s.msBuilder)\n}\n\nfunc (s *historyBuilderSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderDynamicSuccess() {\n\tid := \"dynamic-historybuilder-success-test-workflow-id\"\n\trid := \"dynamic-historybuilder-success-test-run-id\"\n\twt := \"dynamic-historybuilder-success-type\"\n\ttl := \"dynamic-historybuilder-success-tasklist\"\n\tidentity := \"dynamic-historybuilder-success-worker\"\n\tinput := []byte(\"dynamic-historybuilder-success-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tl, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning1)\n\ts.NotNil(di1)\n\tdecisionStartedID1 := di1.StartedID\n\ts.Equal(int64(3), decisionStartedID1)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionContext := []byte(\"dynamic-historybuilder-success-context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\t_, decisionRunning2 := s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivityTaskList := \"dynamic-historybuilder-success-activity-tasklist\"\n\tactivityTimeout := int32(60)\n\tqueueTimeout := int32(20)\n\thearbeatTimeout := int32(10)\n\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"dynamic-historybuilder-success-activity1-type\"\n\tactivity1Domain := \"\"\n\tactivity1Input := []byte(\"dynamic-historybuilder-success-activity1-input\")\n\tactivity1Result := []byte(\"dynamic-historybuilder-success-activity1-result\")\n\tactivity1ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity1ID, activity1Type,\n\t\tactivity1Domain, activityTaskList, activity1Input, activityTimeout, queueTimeout, hearbeatTimeout, nil, false)\n\ts.validateActivityTaskScheduledEvent(activity1ScheduledEvent, 5, 4, activity1ID, activity1Type,\n\t\tactivity1Domain, activityTaskList, activity1Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, false)\n\ts.Equal(int64(6), s.getNextEventID())\n\tai0, activity1Running0 := s.msBuilder.GetActivityInfo(5)\n\ts.True(activity1Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai0.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"dynamic-historybuilder-success-activity2-type\"\n\tactivity2Domain := \"\"\n\tactivity2Input := []byte(\"dynamic-historybuilder-success-activity2-input\")\n\tactivity2Reason := \"dynamic-historybuilder-success-activity2-failed\"\n\tactivity2Details := []byte(\"dynamic-historybuilder-success-activity2-callstack\")\n\tactivity2ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity2ID, activity2Type,\n\t\tactivity2Domain, activityTaskList, activity2Input, activityTimeout, queueTimeout, hearbeatTimeout, nil, false)\n\ts.validateActivityTaskScheduledEvent(activity2ScheduledEvent, 6, 4, activity2ID, activity2Type,\n\t\tactivity2Domain, activityTaskList, activity2Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, false)\n\ts.Equal(int64(7), s.getNextEventID())\n\tai2, activity2Running0 := s.msBuilder.GetActivityInfo(6)\n\ts.True(activity2Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai2.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity3ID := \"activity3\"\n\tactivity3Type := \"dynamic-historybuilder-success-activity3-type\"\n\tactivity3Domain := s.targetDomainName\n\tactivity3Input := []byte(\"dynamic-historybuilder-success-activity3-input\")\n\tactivity3RetryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tMaximumAttempts:             3,\n\t\tMaximumIntervalInSeconds:    1,\n\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\tBackoffCoefficient:          1,\n\t\tExpirationIntervalInSeconds: 100,\n\t}\n\tactivity3ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity3ID, activity3Type,\n\t\tactivity3Domain, activityTaskList, activity3Input, activityTimeout, queueTimeout, hearbeatTimeout, activity3RetryPolicy, false)\n\ts.validateActivityTaskScheduledEvent(activity3ScheduledEvent, 7, 4, activity3ID, activity3Type,\n\t\tactivity3Domain, activityTaskList, activity3Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, false)\n\ts.Equal(int64(8), s.getNextEventID())\n\tai2, activity3Running0 := s.msBuilder.GetActivityInfo(6)\n\ts.True(activity3Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai2.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity4ID := \"activity4\"\n\tactivity4Type := \"dynamic-historybuilder-success-activity4-type\"\n\tactivity4Domain := s.targetDomainName\n\tactivity4Input := []byte(\"dynamic-historybuilder-success-activity4-input\")\n\tactivity4Result := []byte(\"dynamic-historybuilder-success-activity4-result\")\n\tactivity4ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity4ID, activity4Type,\n\t\tactivity4Domain, activityTaskList, activity4Input, activityTimeout, queueTimeout, hearbeatTimeout, nil, true)\n\ts.validateActivityTaskScheduledEvent(activity4ScheduledEvent, 8, 4, activity4ID, activity4Type,\n\t\tactivity4Domain, activityTaskList, activity4Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, true)\n\ts.Equal(int64(9), s.getNextEventID())\n\tai4, activity4Running0 := s.msBuilder.GetActivityInfo(8)\n\ts.True(activity4Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai4.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity5ID := \"activity5\"\n\tactivity5Type := \"dynamic-historybuilder-success-activity5-type\"\n\tactivity5Domain := s.targetDomainName\n\tactivity5Input := []byte(\"dynamic-historybuilder-success-activity5-input\")\n\tactivity5RetryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    1,\n\t\tMaximumAttempts:             3,\n\t\tMaximumIntervalInSeconds:    1,\n\t\tNonRetriableErrorReasons:    []string{\"bad-bug\"},\n\t\tBackoffCoefficient:          1,\n\t\tExpirationIntervalInSeconds: 100,\n\t}\n\tactivity5ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity5ID, activity5Type,\n\t\tactivity5Domain, activityTaskList, activity5Input, activityTimeout, queueTimeout, hearbeatTimeout, activity5RetryPolicy, true)\n\ts.validateActivityTaskScheduledEvent(activity5ScheduledEvent, 9, 4, activity5ID, activity5Type,\n\t\tactivity5Domain, activityTaskList, activity5Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, true)\n\ts.Equal(int64(10), s.getNextEventID())\n\tai5, activity5Running0 := s.msBuilder.GetActivityInfo(9)\n\ts.True(activity5Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai5.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivityStartedEvent := s.addActivityTaskStartedEvent(5, activityTaskList, identity)\n\ts.validateActivityTaskStartedEvent(activityStartedEvent, commonconstants.BufferedEventID, 5, identity,\n\t\t0, \"\", nil)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskStartedEvent(activityStartedEvent, 10, 5, identity,\n\t\t0, \"\", nil)\n\ts.Equal(int64(11), s.getNextEventID())\n\tai3, activity1Running1 := s.msBuilder.GetActivityInfo(5)\n\ts.True(activity1Running1)\n\ts.Equal(int64(10), ai3.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivityCompletedEvent := s.addActivityTaskCompletedEvent(5, 10, activity1Result, identity)\n\ts.validateActivityTaskCompletedEvent(activityCompletedEvent, commonconstants.BufferedEventID, 5, 10, activity1Result,\n\t\tidentity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskCompletedEvent(activityCompletedEvent, 11, 5, 10, activity1Result,\n\t\tidentity)\n\ts.Equal(int64(12), s.getNextEventID())\n\t_, activity1Running2 := s.msBuilder.GetActivityInfo(5)\n\ts.False(activity1Running2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tdi2 := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di2, 12, tl, taskTimeout)\n\ts.Equal(int64(13), s.getNextEventID())\n\tdi3, decisionRunning3 := s.msBuilder.GetDecisionInfo(12)\n\ts.True(decisionRunning3)\n\ts.Equal(commonconstants.EmptyEventID, di3.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity2StartedEvent := s.addActivityTaskStartedEvent(6, activityTaskList, identity)\n\ts.validateActivityTaskStartedEvent(activity2StartedEvent, commonconstants.BufferedEventID, 6, identity,\n\t\t0, \"\", nil)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskStartedEvent(activity2StartedEvent, 13, 6, identity,\n\t\t0, \"\", nil)\n\ts.Equal(int64(14), s.getNextEventID())\n\tai4, activity2Running1 := s.msBuilder.GetActivityInfo(6)\n\ts.True(activity2Running1)\n\ts.Equal(int64(13), ai4.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity2FailedEvent := s.addActivityTaskFailedEvent(6, 13, activity2Reason, activity2Details,\n\t\tidentity)\n\ts.validateActivityTaskFailedEvent(activity2FailedEvent, commonconstants.BufferedEventID, 6, 13, activity2Reason,\n\t\tactivity2Details, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskFailedEvent(activity2FailedEvent, 14, 6, 13, activity2Reason,\n\t\tactivity2Details, identity)\n\ts.Equal(int64(15), s.getNextEventID())\n\t_, activity2Running3 := s.msBuilder.GetActivityInfo(6)\n\ts.False(activity2Running3)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity3StartedEvent := s.addActivityTaskStartedEvent(7, activityTaskList, identity)\n\ts.validateTransientActivityTaskStartedEvent(activity3StartedEvent, commonconstants.TransientEventID, 7, identity)\n\ts.Equal(int64(15), s.getNextEventID())\n\tai5, activity3Running1 := s.msBuilder.GetActivityInfo(7)\n\ts.True(activity3Running1)\n\ts.Equal(commonconstants.TransientEventID, ai5.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity3Reason := \"dynamic-historybuilder-success-activity3-failed\"\n\tactivity3Details := []byte(\"dynamic-historybuilder-success-activity3-callstack\")\n\ts.msBuilder.RetryActivity(ai5, activity3Reason, activity3Details)\n\tai6, activity3Running2 := s.msBuilder.GetActivityInfo(7)\n\ts.Equal(activity3Reason, ai6.LastFailureReason)\n\ts.Equal(activity3Details, ai6.LastFailureDetails)\n\ts.True(activity3Running2)\n\n\tactivity3StartedEvent2 := s.addActivityTaskStartedEvent(7, activityTaskList, identity)\n\ts.validateTransientActivityTaskStartedEvent(activity3StartedEvent2, commonconstants.TransientEventID, 7, identity)\n\ts.Equal(int64(15), s.getNextEventID())\n\tai7, activity3Running3 := s.msBuilder.GetActivityInfo(7)\n\ts.True(activity3Running3)\n\ts.Equal(commonconstants.TransientEventID, ai7.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity3Result := []byte(\"dynamic-historybuilder-success-activity1-result\")\n\tactivity3CompletedEvent := s.addActivityTaskCompletedEvent(7, commonconstants.TransientEventID, activity3Result, identity)\n\ts.validateActivityTaskCompletedEvent(activity3CompletedEvent, commonconstants.BufferedEventID, 7, commonconstants.TransientEventID,\n\t\tactivity3Result, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskCompletedEvent(activity3CompletedEvent, 16, 7, 15, activity3Result, identity)\n\ts.Equal(int64(17), s.getNextEventID())\n\tai8, activity3Running4 := s.msBuilder.GetActivityInfo(7)\n\ts.Nil(ai8)\n\ts.False(activity3Running4)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity4StartedEvent := s.addActivityTaskStartedEvent(8, activityTaskList, identity)\n\ts.validateActivityTaskStartedEvent(activity4StartedEvent, commonconstants.BufferedEventID, 8, identity,\n\t\t0, \"\", nil)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskStartedEvent(activity4StartedEvent, 17, 8, identity,\n\t\t0, \"\", nil)\n\ts.Equal(int64(18), s.getNextEventID())\n\tai4, activity4Running1 := s.msBuilder.GetActivityInfo(8)\n\ts.True(activity4Running1)\n\ts.Equal(int64(17), ai4.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity4CompletedEvent := s.addActivityTaskCompletedEvent(8, 17, activity4Result, identity)\n\ts.validateActivityTaskCompletedEvent(activity4CompletedEvent, commonconstants.BufferedEventID, 8, 17, activity4Result,\n\t\tidentity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskCompletedEvent(activity4CompletedEvent, 18, 8, 17, activity4Result,\n\t\tidentity)\n\ts.Equal(int64(19), s.getNextEventID())\n\t_, activity4Running2 := s.msBuilder.GetActivityInfo(8)\n\ts.False(activity4Running2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity5StartedEvent := s.addActivityTaskStartedEvent(9, activityTaskList, identity)\n\ts.validateTransientActivityTaskStartedEvent(activity5StartedEvent, commonconstants.TransientEventID, 9, identity)\n\ts.Equal(int64(19), s.getNextEventID())\n\tai5, activity5Running1 := s.msBuilder.GetActivityInfo(9)\n\ts.True(activity5Running1)\n\ts.Equal(commonconstants.TransientEventID, ai5.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity5Reason := \"dynamic-historybuilder-success-activity5-failed\"\n\tactivity5Details := []byte(\"dynamic-historybuilder-success-activity5-callstack\")\n\ts.msBuilder.RetryActivity(ai5, activity5Reason, activity5Details)\n\tai6, activity5Running2 := s.msBuilder.GetActivityInfo(9)\n\ts.Equal(activity5Reason, ai6.LastFailureReason)\n\ts.Equal(activity5Details, ai6.LastFailureDetails)\n\ts.True(activity5Running2)\n\n\tactivity5StartedEvent2 := s.addActivityTaskStartedEvent(9, activityTaskList, identity)\n\ts.validateTransientActivityTaskStartedEvent(activity5StartedEvent2, commonconstants.TransientEventID, 9, identity)\n\ts.Equal(int64(19), s.getNextEventID())\n\tai7, activity5Running3 := s.msBuilder.GetActivityInfo(9)\n\ts.True(activity5Running3)\n\ts.Equal(commonconstants.TransientEventID, ai7.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivity5Result := []byte(\"dynamic-historybuilder-success-activity5-result\")\n\tactivity5CompletedEvent := s.addActivityTaskCompletedEvent(9, commonconstants.TransientEventID, activity5Result, identity)\n\ts.validateActivityTaskCompletedEvent(activity5CompletedEvent, commonconstants.BufferedEventID, 9, commonconstants.TransientEventID,\n\t\tactivity5Result, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskCompletedEvent(activity5CompletedEvent, 20, 9, 19, activity5Result, identity)\n\ts.Equal(int64(21), s.getNextEventID())\n\tai8, activity5Running4 := s.msBuilder.GetActivityInfo(9)\n\ts.Nil(ai8)\n\ts.False(activity5Running4)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// Verify the last ActivityTaskStartedEvent which should show the error from the first attempt\n\thistoryEvents := s.msBuilder.GetHistoryBuilder().GetHistory().GetEvents()\n\ts.Len(historyEvents, 20)\n\ts.validateActivityTaskStartedEvent(historyEvents[14], 15, 7, identity, 1, activity3Reason, activity3Details)\n\n\tmarkerDetails := []byte(\"dynamic-historybuilder-success-marker-details\")\n\tmarkerHeaderField1 := []byte(\"dynamic-historybuilder-success-marker-header1\")\n\tmarkerHeaderField2 := []byte(\"dynamic-historybuilder-success-marker-header2\")\n\tmarkerHeader := map[string][]byte{\n\t\t\"name1\": markerHeaderField1,\n\t\t\"name2\": markerHeaderField2,\n\t}\n\tmarkerEvent := s.addMarkerRecordedEvent(4, \"testMarker\", markerDetails,\n\t\t&markerHeader)\n\ts.validateMarkerRecordedEvent(markerEvent, 21, 4, \"testMarker\", markerDetails, &markerHeader)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.Equal(int64(22), s.getNextEventID())\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderWorkflowStartFailures() {\n\tid := \"historybuilder-workflowstart-failures-test-workflow-id\"\n\trid := \"historybuilder-workflowstart-failures-test-run-id\"\n\twt := \"historybuilder-workflowstart-failures-type\"\n\ttl := \"historybuilder-workflowstart-failures-tasklist\"\n\tidentity := \"historybuilder-workflowstart-failures-worker\"\n\tinput := []byte(\"historybuilder-workflowstart-failures-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\t_, err := s.msBuilder.AddWorkflowExecutionStartedEvent(\n\t\twe,\n\t\t&types.HistoryStartWorkflowExecutionRequest{\n\t\t\tDomainUUID: s.domainID,\n\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\tWorkflowID:                          we.WorkflowID,\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: wt},\n\t\t\t\tTaskList:                            &types.TaskList{Name: tl},\n\t\t\t\tInput:                               input,\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(execTimeout),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(taskTimeout),\n\t\t\t\tIdentity:                            identity,\n\t\t\t},\n\t\t})\n\ts.NotNil(err)\n\n\ts.Equal(int64(3), s.getNextEventID(), s.printHistory())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning1)\n\ts.Equal(commonconstants.EmptyEventID, di1.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderDecisionScheduledFailures() {\n\tid := \"historybuilder-decisionscheduled-failures-test-workflow-id\"\n\trid := \"historybuilder-decisionscheduled-failures-test-run-id\"\n\twt := \"historybuilder-decisionscheduled-failures-type\"\n\ttl := \"historybuilder-decisionscheduled-failures-tasklist\"\n\tidentity := \"historybuilder-decisionscheduled-failures-worker\"\n\tinput := []byte(\"historybuilder-decisionscheduled-failures-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\t_, err := s.msBuilder.AddDecisionTaskScheduledEvent(false)\n\ts.NotNil(err)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning1)\n\ts.Equal(commonconstants.EmptyEventID, di1.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderDecisionScheduledTimedout() {\n\tid := \"historybuilder-decisionscheduled-failures-test-workflow-id\"\n\trid := \"historybuilder-decisionscheduled-failures-test-run-id\"\n\twt := \"historybuilder-decisionscheduled-failures-type\"\n\ttl := \"historybuilder-decisionscheduled-failures-tasklist\"\n\tidentity := \"historybuilder-decisionscheduled-failures-worker\"\n\tinput := []byte(\"historybuilder-decisionscheduled-failures-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionTimeoutEvent := s.addDecisionTaskTimedOutEvent(2)\n\ts.NotNil(decisionTimeoutEvent)\n\ts.validateDecisionTaskTimedoutEvent(decisionTimeoutEvent, 3, 2, 0, types.TimeoutTypeScheduleToStart, \"\", \"\", commonconstants.EmptyVersion, \"\", types.DecisionTaskTimedOutCauseTimeout)\n\ts.Equal(int64(4), s.getNextEventID())\n\n\tdi2 := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di2, 4, tl, taskTimeout)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(4)\n\ts.True(decisionRunning1)\n\ts.Equal(commonconstants.EmptyEventID, di1.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionTimeoutEvent = s.addDecisionTaskTimedOutEvent(4)\n\ts.Nil(decisionTimeoutEvent)\n\ts.Equal(int64(4), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderDecisionStartedTimedout() {\n\tid := \"historybuilder-decisionscheduled-failures-test-workflow-id\"\n\trid := \"historybuilder-decisionscheduled-failures-test-run-id\"\n\twt := \"historybuilder-decisionscheduled-failures-type\"\n\ttl := \"historybuilder-decisionscheduled-failures-tasklist\"\n\tidentity := \"historybuilder-decisionscheduled-failures-worker\"\n\tinput := []byte(\"historybuilder-decisionscheduled-failures-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tl, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionTimeoutEvent := s.addDecisionTaskStartedTimedOutEvent(2, 3)\n\ts.NotNil(decisionTimeoutEvent)\n\ts.validateDecisionTaskTimedoutEvent(decisionTimeoutEvent, 4, 2, 3, types.TimeoutTypeStartToClose, \"\", \"\", commonconstants.EmptyVersion, \"\", types.DecisionTaskTimedOutCauseTimeout)\n\ts.Equal(int64(5), s.getNextEventID())\n\n\tdi2 := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di2, 5, tl, taskTimeout)\n\ts.Equal(int64(5), s.getNextEventID())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(5)\n\ts.True(decisionRunning1)\n\ts.Equal(int64(1), di1.Attempt)\n\ts.Equal(commonconstants.EmptyEventID, di1.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent = s.addDecisionTaskStartedEvent(5, tl, identity)\n\ts.Nil(decisionStartedEvent)\n\ts.Equal(int64(5), s.getNextEventID())\n\n\tdecisionTimeoutEvent = s.addDecisionTaskStartedTimedOutEvent(5, 6)\n\ts.Equal(int64(5), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderDecisionStartedFailures() {\n\tid := \"historybuilder-decisionstarted-failures-test-workflow-id\"\n\trid := \"historybuilder-decisionstarted-failures-test-run-id\"\n\twt := \"historybuilder-decisionstarted-failures-type\"\n\ttl := \"historybuilder-decisionstarted-failures-tasklist\"\n\tidentity := \"historybuilder-decisionstarted-failures-worker\"\n\tinput := []byte(\"historybuilder-decisionstarted-failures-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\t_, _, err := s.msBuilder.AddDecisionTaskStartedEvent(2, uuid.New(), &types.PollForDecisionTaskRequest{\n\t\tTaskList: &types.TaskList{Name: tl},\n\t\tIdentity: identity,\n\t})\n\ts.NotNil(err)\n\ts.Equal(int64(2), s.getNextEventID())\n\t_, decisionRunning1 := s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning1)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\t_, _, err = s.msBuilder.AddDecisionTaskStartedEvent(100, uuid.New(), &types.PollForDecisionTaskRequest{\n\t\tTaskList: &types.TaskList{Name: tl},\n\t\tIdentity: identity,\n\t})\n\ts.NotNil(err)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi2, decisionRunning2 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning2)\n\ts.Equal(commonconstants.EmptyEventID, di2.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent2 := s.addDecisionTaskStartedEvent(2, tl, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent2, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdi3, decisionRunning3 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning3)\n\ts.Equal(int64(3), di3.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderFlushBufferedEvents() {\n\tid := \"flush-buffered-events-test-workflow-id\"\n\trid := \"flush-buffered-events-test-run-id\"\n\twt := \"flush-buffered-events-type\"\n\ttl := \"flush-buffered-events-tasklist\"\n\tidentity := \"flush-buffered-events-worker\"\n\tinput := []byte(\"flush-buffered-events-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1 execution started\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\t// 2 decision scheduled\n\tdi := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di, 2, tl, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdi0, decisionRunning0 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning0)\n\ts.Equal(commonconstants.EmptyEventID, di0.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\t// 3 decision started\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tl, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdi1, decisionRunning1 := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning1)\n\ts.NotNil(di1)\n\tdecisionStartedID1 := di1.StartedID\n\ts.Equal(int64(3), decisionStartedID1)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\t// 4 decision completed\n\tdecisionContext := []byte(\"flush-buffered-events-context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\t_, decisionRunning2 := s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tactivityTaskList := \"flush-buffered-events-activity-tasklist\"\n\tactivityTimeout := int32(60)\n\tqueueTimeout := int32(20)\n\thearbeatTimeout := int32(10)\n\n\t// 5 activity1 scheduled\n\tactivity1ID := \"activity1\"\n\tactivity1Type := \"flush-buffered-events-activity1-type\"\n\tactivity1Domain := \"\"\n\tactivity1Input := []byte(\"flush-buffered-events-activity1-input\")\n\tactivity1Result := []byte(\"flush-buffered-events-activity1-result\")\n\tactivity1ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity1ID, activity1Type,\n\t\tactivity1Domain, activityTaskList, activity1Input, activityTimeout, queueTimeout, hearbeatTimeout, nil, false)\n\ts.validateActivityTaskScheduledEvent(activity1ScheduledEvent, 5, 4, activity1ID, activity1Type,\n\t\tactivity1Domain, activityTaskList, activity1Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, false)\n\ts.Equal(int64(6), s.getNextEventID())\n\tai0, activity1Running0 := s.msBuilder.GetActivityInfo(5)\n\ts.True(activity1Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai0.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 6 activity 2 scheduled\n\tactivity2ID := \"activity2\"\n\tactivity2Type := \"flush-buffered-events-activity2-type\"\n\tactivity2Domain := s.targetDomainName\n\tactivity2Input := []byte(\"flush-buffered-events-activity2-input\")\n\tactivity2ScheduledEvent, _, activityDispatchInfo := s.addActivityTaskScheduledEvent(4, activity2ID, activity2Type,\n\t\tactivity2Domain, activityTaskList, activity2Input, activityTimeout, queueTimeout, hearbeatTimeout, nil, false)\n\ts.validateActivityTaskScheduledEvent(activity2ScheduledEvent, 6, 4, activity2ID, activity2Type,\n\t\tactivity2Domain, activityTaskList, activity2Input, activityTimeout, queueTimeout, hearbeatTimeout, activityDispatchInfo, false)\n\ts.Equal(int64(7), s.getNextEventID())\n\tai2, activity2Running0 := s.msBuilder.GetActivityInfo(6)\n\ts.True(activity2Running0)\n\ts.Equal(commonconstants.EmptyEventID, ai2.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 7 activity1 started\n\tactivityStartedEvent := s.addActivityTaskStartedEvent(5, activityTaskList, identity)\n\ts.validateActivityTaskStartedEvent(activityStartedEvent, commonconstants.BufferedEventID, 5, identity,\n\t\t0, \"\", nil)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskStartedEvent(activityStartedEvent, 7, 5, identity,\n\t\t0, \"\", nil)\n\ts.Equal(int64(8), s.getNextEventID())\n\tai3, activity1Running1 := s.msBuilder.GetActivityInfo(5)\n\ts.True(activity1Running1)\n\ts.Equal(int64(7), ai3.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 8 activity1 completed\n\tactivityCompletedEvent := s.addActivityTaskCompletedEvent(5, 7, activity1Result, identity)\n\ts.validateActivityTaskCompletedEvent(activityCompletedEvent, commonconstants.BufferedEventID, 5, 7, activity1Result, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateActivityTaskCompletedEvent(activityCompletedEvent, 8, 5, 7, activity1Result, identity)\n\ts.Equal(int64(9), s.getNextEventID())\n\t_, activity1Running2 := s.msBuilder.GetActivityInfo(5)\n\ts.False(activity1Running2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 9 decision2 scheduled\n\tdi2 := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(di2, 9, tl, taskTimeout)\n\ts.Equal(int64(10), s.getNextEventID())\n\tdi3, decisionRunning3 := s.msBuilder.GetDecisionInfo(9)\n\ts.True(decisionRunning3)\n\ts.Equal(commonconstants.EmptyEventID, di3.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 10 decision2 started\n\tdecision2StartedEvent := s.addDecisionTaskStartedEvent(9, tl, identity)\n\ts.validateDecisionTaskStartedEvent(decision2StartedEvent, 10, 9, identity)\n\ts.Equal(int64(11), s.getNextEventID())\n\tdi2, decision2Running := s.msBuilder.GetDecisionInfo(9)\n\ts.True(decision2Running)\n\ts.NotNil(di2)\n\tdecision2StartedID := di2.StartedID\n\ts.Equal(int64(10), decision2StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 11 (buffered) activity2 started\n\tactivity2StartedEvent := s.addActivityTaskStartedEvent(6, activityTaskList, identity)\n\ts.validateActivityTaskStartedEvent(activity2StartedEvent, commonconstants.BufferedEventID, 6, identity,\n\t\t0, \"\", nil)\n\ts.Equal(int64(11), s.getNextEventID())\n\tai4, activity2Running := s.msBuilder.GetActivityInfo(6)\n\ts.True(activity2Running)\n\ts.Equal(commonconstants.BufferedEventID, ai4.StartedID)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 12 (buffered) activity2 failed\n\tactivity2Reason := \"flush-buffered-events-activity2-failed\"\n\tactivity2Details := []byte(\"flush-buffered-events-activity2-callstack\")\n\tactivity2FailedEvent := s.addActivityTaskFailedEvent(6, commonconstants.BufferedEventID, activity2Reason, activity2Details, identity)\n\ts.validateActivityTaskFailedEvent(activity2FailedEvent, commonconstants.BufferedEventID, 6, commonconstants.BufferedEventID, activity2Reason,\n\t\tactivity2Details, identity)\n\ts.Equal(int64(11), s.getNextEventID())\n\t_, activity2Running2 := s.msBuilder.GetActivityInfo(6)\n\ts.False(activity2Running2)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\t// 13 (eventId will be 11) decision completed\n\tdecision2Context := []byte(\"flush-buffered-events-context\")\n\tdecision2CompletedEvent := s.addDecisionTaskCompletedEvent(9, 10, decision2Context, identity)\n\ts.validateDecisionTaskCompletedEvent(decision2CompletedEvent, 11, 9, 10, decision2Context, identity)\n\ts.Equal(int64(11), decision2CompletedEvent.ID)\n\ts.Equal(int64(12), s.getNextEventID())\n\t_, decision2Running2 := s.msBuilder.GetDecisionInfo(2)\n\ts.False(decision2Running2)\n\ts.Equal(int64(10), s.getPreviousDecisionStartedEventID())\n\n\t// flush buffered events. 12: Activity2Started, 13: Activity2Failed\n\ts.NoError(s.msBuilder.FlushBufferedEvents())\n\ts.Equal(int64(14), s.getNextEventID())\n\tactivity2StartedEvent2 := s.msBuilder.GetHistoryBuilder().history[11]\n\ts.Equal(int64(12), activity2StartedEvent2.ID)\n\ts.Equal(types.EventTypeActivityTaskStarted, activity2StartedEvent2.GetEventType())\n\n\tactivity2FailedEvent2 := s.msBuilder.GetHistoryBuilder().history[12]\n\ts.Equal(int64(13), activity2FailedEvent2.ID)\n\ts.Equal(types.EventTypeActivityTaskFailed, activity2FailedEvent2.GetEventType())\n\ts.Equal(int64(12), activity2FailedEvent2.ActivityTaskFailedEventAttributes.GetStartedEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderWorkflowCancellationRequested() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionContext := []byte(\"some random decision context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning)\n\ts.Nil(decisionInfo)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      \"some random target run ID\",\n\t}\n\tcancellationChildWorkflowOnly := true\n\tcancellationInitiatedEvent := s.addRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\t4, s.targetDomainName, targetExecution, cancellationChildWorkflowOnly,\n\t)\n\ts.validateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\tcancellationInitiatedEvent, 5, 4, s.targetDomainName, targetExecution, cancellationChildWorkflowOnly,\n\t)\n\ts.Equal(int64(6), s.getNextEventID())\n\n\tcancellationRequestedEvent := s.addExternalWorkflowExecutionCancelRequested(\n\t\t5, s.targetDomainName, targetExecution.GetWorkflowID(), targetExecution.GetRunID(),\n\t)\n\ts.validateExternalWorkflowExecutionCancelRequested(cancellationRequestedEvent, commonconstants.BufferedEventID, 5, s.targetDomainName, targetExecution)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateExternalWorkflowExecutionCancelRequested(cancellationRequestedEvent, 6, 5, s.targetDomainName, targetExecution)\n\ts.Equal(int64(7), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderWorkflowCancellationFailed() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionContext := []byte(\"some random decision context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning)\n\ts.Nil(decisionInfo)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      \"some random target run ID\",\n\t}\n\tcancellationChildWorkflowOnly := true\n\tcancellationFailedCause := types.CancelExternalWorkflowExecutionFailedCause(59)\n\tcancellationInitiatedEvent := s.addRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\t4, s.targetDomainName, targetExecution, cancellationChildWorkflowOnly,\n\t)\n\ts.validateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\tcancellationInitiatedEvent, 5, 4, s.targetDomainName, targetExecution, cancellationChildWorkflowOnly,\n\t)\n\ts.Equal(int64(6), s.getNextEventID())\n\n\tcancellationRequestedEvent := s.addRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\t4, 5, s.targetDomainName, targetExecution.GetWorkflowID(), targetExecution.GetRunID(), cancellationFailedCause,\n\t)\n\ts.validateRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\tcancellationRequestedEvent, commonconstants.BufferedEventID, 4, 5, s.targetDomainName, targetExecution, cancellationFailedCause,\n\t)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\tcancellationRequestedEvent, 6, 4, 5, s.targetDomainName, targetExecution, cancellationFailedCause,\n\t)\n\ts.Equal(int64(7), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilder_DecisionTaskResetTimedOut() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tbaseRunID := uuid.New()\n\tnewRunID := uuid.New()\n\tforkVersion := int64(10)\n\tresetRequestID := uuid.New()\n\tdecisionFailedEvent, err := s.msBuilder.AddDecisionTaskFailedEvent(2, 3, types.DecisionTaskFailedCauseResetWorkflow, nil, identity, \"\", \"\", baseRunID, newRunID, forkVersion, resetRequestID)\n\ts.NoError(err)\n\ts.NotNil(decisionFailedEvent.GetDecisionTaskFailedEventAttributes())\n\ts.Equal(baseRunID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetBaseRunID())\n\ts.Equal(newRunID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetNewRunID())\n\ts.Equal(forkVersion, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetForkEventVersion())\n\ts.Equal(resetRequestID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetRequestID())\n\ts.Equal(int64(5), s.getNextEventID())\n\n\tdecisionInfo = s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 5, tasklist, taskTimeout)\n\ts.Equal(int64(6), s.getNextEventID())\n\n\tdecisionStartedEvent = s.addDecisionTaskStartedEvent(5, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 6, 5, identity)\n\ts.Equal(int64(7), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(5)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(5), decisionInfo.ScheduleID)\n\ts.Equal(int64(6), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionFailedEvent, err = s.msBuilder.AddDecisionTaskFailedEvent(5, 6, types.DecisionTaskFailedCauseResetWorkflow, nil, identity, \"\", \"\", baseRunID, newRunID, forkVersion, resetRequestID)\n\ts.NoError(err)\n\ts.NotNil(decisionFailedEvent.GetDecisionTaskFailedEventAttributes())\n\ts.Equal(baseRunID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetBaseRunID())\n\ts.Equal(newRunID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetNewRunID())\n\ts.Equal(forkVersion, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetForkEventVersion())\n\ts.Equal(resetRequestID, decisionFailedEvent.GetDecisionTaskFailedEventAttributes().GetRequestID())\n\ts.Equal(int64(8), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilder_DecisionTaskResetFailed() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tbaseRunID := uuid.New()\n\tnewRunID := uuid.New()\n\tforkVersion := int64(10)\n\tresetRequestID := uuid.New()\n\tdecisionTimedOutEvent := s.addDecisionTaskResetTimedOutEvent(2, baseRunID, newRunID, forkVersion, resetRequestID)\n\ts.NotNil(decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes())\n\ts.Equal(int64(2), decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetScheduledEventID())\n\ts.Equal(baseRunID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetBaseRunID())\n\ts.Equal(newRunID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetNewRunID())\n\ts.Equal(forkVersion, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetForkEventVersion())\n\ts.Equal(resetRequestID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetRequestID())\n\ts.Equal(int64(5), s.getNextEventID())\n\n\tdecisionInfo = s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 5, tasklist, taskTimeout)\n\ts.Equal(int64(6), s.getNextEventID())\n\n\tdecisionTimedOutEvent = s.addDecisionTaskResetTimedOutEvent(5, baseRunID, newRunID, forkVersion, resetRequestID)\n\ts.NotNil(decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes())\n\ts.Equal(int64(5), decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetScheduledEventID())\n\ts.Equal(baseRunID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetBaseRunID())\n\ts.Equal(newRunID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetNewRunID())\n\ts.Equal(forkVersion, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetForkEventVersion())\n\ts.Equal(resetRequestID, decisionTimedOutEvent.GetDecisionTaskTimedOutEventAttributes().GetRequestID())\n\ts.Equal(int64(7), s.getNextEventID())\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderWorkflowExternalCancellationRequested() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionContext := []byte(\"some random decision context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning)\n\ts.Nil(decisionInfo)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tcause := \"cancel workflow\"\n\treq := &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tIdentity:  \"identity\",\n\t\t\tRequestID: \"b071cbe8-3a95-4223-a8ac-f308a42db383\",\n\t\t},\n\t}\n\tcancellationEvent := s.addWorkflowExecutionCancelRequestedEvent(cause, req)\n\ts.validateWorkflowExecutionCancelRequestedEvent(cancellationEvent, commonconstants.BufferedEventID, \"cancel workflow\", \"identity\", nil, types.WorkflowExecution{}, \"b071cbe8-3a95-4223-a8ac-f308a42db383\")\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateWorkflowExecutionCancelRequestedEvent(cancellationEvent, 5, \"cancel workflow\", \"identity\", nil, types.WorkflowExecution{}, \"b071cbe8-3a95-4223-a8ac-f308a42db383\")\n\ts.Equal(int64(6), s.getNextEventID())\n\t_, exists := s.msBuilder.(*mutableStateBuilder).workflowRequests[persistence.WorkflowRequest{\n\t\tRequestID:   \"b071cbe8-3a95-4223-a8ac-f308a42db383\",\n\t\tRequestType: persistence.WorkflowRequestTypeCancel,\n\t\tVersion:     s.msBuilder.GetCurrentVersion(),\n\t}]\n\ts.True(exists)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderWorkflowExternalSignaled() {\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tidentity := \"some random identity\"\n\tinput := []byte(\"some random workflow input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\tworkflowStartedEvent := s.addWorkflowExecutionStartedEvent(workflowExecution, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\ts.validateWorkflowExecutionStartedEvent(workflowStartedEvent, workflowType, tasklist, input, execTimeout, taskTimeout, identity, partitionConfig)\n\n\ts.Equal(int64(2), s.getNextEventID())\n\n\tdecisionInfo := s.addDecisionTaskScheduledEvent()\n\ts.validateDecisionTaskScheduledEvent(decisionInfo, 2, tasklist, taskTimeout)\n\ts.Equal(int64(3), s.getNextEventID())\n\tdecisionInfo, decisionRunning := s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionStartedEvent := s.addDecisionTaskStartedEvent(2, tasklist, identity)\n\ts.validateDecisionTaskStartedEvent(decisionStartedEvent, 3, 2, identity)\n\ts.Equal(int64(4), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.True(decisionRunning)\n\ts.NotNil(decisionInfo)\n\ts.Equal(int64(2), decisionInfo.ScheduleID)\n\ts.Equal(int64(3), decisionInfo.StartedID)\n\ts.Equal(commonconstants.EmptyEventID, s.getPreviousDecisionStartedEventID())\n\n\tdecisionContext := []byte(\"some random decision context\")\n\tdecisionCompletedEvent := s.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\ts.validateDecisionTaskCompletedEvent(decisionCompletedEvent, 4, 2, 3, decisionContext, identity)\n\ts.Equal(int64(5), s.getNextEventID())\n\tdecisionInfo, decisionRunning = s.msBuilder.GetDecisionInfo(2)\n\ts.False(decisionRunning)\n\ts.Nil(decisionInfo)\n\ts.Equal(int64(3), s.getPreviousDecisionStartedEventID())\n\n\tsignalName := \"test-signal\"\n\tsignalInput := []byte(\"input\")\n\tsignalIdentity := \"id\"\n\trequestID := \"3b8d0ec2-e1ff-4f61-915b-1ffca831361e\"\n\tsignalEvent := s.addWorkflowExecutionSignaledEvent(signalName, signalInput, signalIdentity, requestID)\n\ts.validateWorkflowExecutionSignaledEvent(signalEvent, commonconstants.BufferedEventID, \"test-signal\", []byte(\"input\"), \"id\", \"3b8d0ec2-e1ff-4f61-915b-1ffca831361e\")\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateWorkflowExecutionSignaledEvent(signalEvent, 5, \"test-signal\", []byte(\"input\"), \"id\", \"3b8d0ec2-e1ff-4f61-915b-1ffca831361e\")\n\ts.Equal(int64(6), s.getNextEventID())\n\t_, exists := s.msBuilder.(*mutableStateBuilder).workflowRequests[persistence.WorkflowRequest{\n\t\tRequestID:   \"3b8d0ec2-e1ff-4f61-915b-1ffca831361e\",\n\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\tVersion:     s.msBuilder.GetCurrentVersion(),\n\t}]\n\ts.True(exists)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderActivityTaskTimedOut() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\t// 2\n\ts.addDecisionTaskScheduledEvent()\n\t// 3\n\ts.addDecisionTaskStartedEvent(2, tl, identity)\n\tdecisionContext := []byte(\"historybuilder-context\")\n\t// 4\n\ts.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\tactivityID := \"activity1\"\n\tactivityType := \"activity1-type\"\n\tactivityDomain := \"\"\n\tactivityInput := []byte(\"activity1-input\")\n\tactivityTimeout := int32(60)\n\tqueueTimeout := int32(20)\n\thearbeatTimeout := int32(10)\n\t// 5\n\ts.addActivityTaskScheduledEvent(4, activityID, activityType, activityDomain, tl, activityInput, activityTimeout, queueTimeout, hearbeatTimeout, nil, false)\n\t// 6\n\ts.addActivityTaskStartedEvent(5, tl, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ttimeoutType := types.TimeoutTypeHeartbeat\n\tlastHeartbeat := []byte(\"last-heartbeat\")\n\t// 7\n\ttimedOut := s.addActivityTaskTimedOutEvent(5, 6, timeoutType, lastHeartbeat)\n\ts.validateActivityTaskTimedOutEvent(timedOut, commonconstants.BufferedEventID, 5, 6, timeoutType, lastHeartbeat, \"\", nil)\n\t// 8\n\ts.addDecisionTaskScheduledEvent()\n\t// 9\n\ts.addDecisionTaskStartedEvent(8, tl, identity)\n\t// 10\n\ts.addDecisionTaskCompletedEvent(8, 9, decisionContext, identity)\n\tfailReason := \"activity failed :(\"\n\tfailDetails := []byte(\"huge failure\")\n\t// 11\n\tfailed := s.addWorkflowExecutionFailed(10, &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  &failReason,\n\t\tDetails: failDetails,\n\t})\n\ts.validateWorkflowExecutionFailed(failed, 11, 10, failReason, failDetails)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderTimeoutWorkflow() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\t// 2\n\ttimedOut := s.addWorkflowExecutionTimedOut(1)\n\ts.validateWorkflowExecutionTimedOut(timedOut, 2)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderCompleteWorkflow() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\t// 2\n\ts.addDecisionTaskScheduledEvent()\n\t// 3\n\ts.addDecisionTaskStartedEvent(2, tl, identity)\n\tdecisionContext := []byte(\"historybuilder-context\")\n\t// 4\n\ts.addDecisionTaskCompletedEvent(2, 3, decisionContext, identity)\n\tdecisionResult := []byte(\"great success\")\n\t// 5\n\tevent := s.addWorkflowExecutionCompleted(4, &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: decisionResult,\n\t})\n\ts.validateWorkflowExecutionCompleted(event, 5, 4, decisionResult)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderTerminateWorkflow() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\t// 2\n\tterminated := s.addWorkflowExecutionTerminated(1, \"reason\", []byte(\"a really good reason\"), identity)\n\ts.validateWorkflowExecutionTerminated(terminated, 2, \"reason\", []byte(\"a really good reason\"), identity)\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderCancelWorkflow() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\treq := &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tIdentity:  \"identity\",\n\t\t\tRequestID: \"b071cbe8-3a95-4223-a8ac-f308a42db383\",\n\t\t},\n\t}\n\t// 2\n\ts.addWorkflowExecutionCancelRequestedEvent(\"because\", req)\n\t// 3\n\ts.addDecisionTaskScheduledEvent()\n\t// 4\n\ts.addDecisionTaskStartedEvent(3, tl, identity)\n\t// 5\n\ts.addDecisionTaskCompletedEvent(3, 4, []byte(\"context\"), identity)\n\t// 6\n\tevent := s.addWorkflowExecutionCanceled(5, &types.CancelWorkflowExecutionDecisionAttributes{\n\t\tDetails: []byte(\"details\"),\n\t})\n\ts.validateWorkflowExecutionCanceled(event, 6, 5, []byte(\"details\"))\n}\n\nfunc (s *historyBuilderSuite) TestHistoryBuilderTimers() {\n\tid := \"historybuilder-workflow-id\"\n\trid := \"historybuilder-test-run-id\"\n\twt := \"historybuilder-type\"\n\ttl := \"historybuilder-tasklist\"\n\tidentity := \"historybuilder-worker\"\n\tinput := []byte(\"historybuilder-input\")\n\texecTimeout := int32(60)\n\ttaskTimeout := int32(10)\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: id,\n\t\tRunID:      rid,\n\t}\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca1\",\n\t}\n\n\t// 1\n\ts.addWorkflowExecutionStartedEvent(we, wt, tl, input, execTimeout, taskTimeout, identity, partitionConfig)\n\t// 2\n\ts.addDecisionTaskScheduledEvent()\n\t// 3\n\ts.addDecisionTaskStartedEvent(2, tl, identity)\n\t// 4\n\ts.addDecisionTaskCompletedEvent(2, 3, []byte(\"context\"), identity)\n\t// 5\n\ttimer1Started := s.addTimerStartedEvent(4, &types.StartTimerDecisionAttributes{\n\t\tTimerID:                   \"timer1\",\n\t\tStartToFireTimeoutSeconds: common.Int64Ptr(int64(30)),\n\t})\n\t// 6\n\ttimer2Started := s.addTimerStartedEvent(4, &types.StartTimerDecisionAttributes{\n\t\tTimerID:                   \"timer2\",\n\t\tStartToFireTimeoutSeconds: common.Int64Ptr(int64(45)),\n\t})\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateTimerStartedEvent(timer1Started, 5, 4, 30, \"timer1\")\n\ts.validateTimerStartedEvent(timer2Started, 6, 4, 45, \"timer2\")\n\t// 7\n\ttimerFired := s.addTimerFiredEvent(\"timer1\")\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateTimerFiredEvent(timerFired, 7, 5, \"timer1\")\n\t// 8\n\ts.addDecisionTaskScheduledEvent()\n\t// 9\n\ts.addDecisionTaskStartedEvent(8, tl, identity)\n\t// 10\n\ts.addDecisionTaskCompletedEvent(8, 9, []byte(\"context\"), identity)\n\t// 11\n\t// Cancel timer2\n\tcancel := s.addTimerCanceledEvent(10, &types.CancelTimerDecisionAttributes{\n\t\tTimerID: \"timer2\",\n\t}, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\n\ts.validateTimerCanceledEvent(cancel, 11, 6, 10, \"timer2\", identity)\n\t// 12\n\t// Fail to cancel a timer that doesn't exist\n\tcancelFailed := s.addTimerCancelFailedEvent(10, &types.CancelTimerDecisionAttributes{\n\t\tTimerID: \"not-a-real-timer\",\n\t}, identity)\n\ts.Nil(s.msBuilder.FlushBufferedEvents())\n\ts.validateTimerCancelFailedEvent(cancelFailed, 12, 10, \"not-a-real-timer\", identity)\n}\n\nfunc (s *historyBuilderSuite) getNextEventID() int64 {\n\treturn s.msBuilder.GetExecutionInfo().NextEventID\n}\n\nfunc (s *historyBuilderSuite) getPreviousDecisionStartedEventID() int64 {\n\treturn s.msBuilder.GetExecutionInfo().LastProcessedEvent\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionStartedEvent(we types.WorkflowExecution, workflowType,\n\ttaskList string, input []byte, executionStartToCloseTimeout, taskStartToCloseTimeout int32,\n\tidentity string, partitionConfig map[string]string) *types.HistoryEvent {\n\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tWorkflowID:                          we.WorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\tInput:                               input,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(executionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(taskStartToCloseTimeout),\n\t\tIdentity:                            identity,\n\t}\n\n\tevent, err := s.msBuilder.AddWorkflowExecutionStartedEvent(\n\t\twe,\n\t\t&types.HistoryStartWorkflowExecutionRequest{\n\t\t\tDomainUUID:      s.domainID,\n\t\t\tStartRequest:    request,\n\t\t\tPartitionConfig: partitionConfig,\n\t\t},\n\t)\n\ts.Nil(err)\n\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionTimedOut(firstEvent int64) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddTimeoutWorkflowEvent(firstEvent)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionFailed(decisionCompletedEventID int64, attributes *types.FailWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddFailWorkflowEvent(decisionCompletedEventID, attributes)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionCompleted(decisionCompletedEventID int64, attributes *types.CompleteWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddCompletedWorkflowEvent(decisionCompletedEventID, attributes)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionTerminated(firstEventID int64, reason string, details []byte, identity string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddWorkflowExecutionTerminatedEvent(firstEventID, reason, details, identity)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionCanceled(decisionTaskCompletedEventID int64,\n\tattributes *types.CancelWorkflowExecutionDecisionAttributes) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddWorkflowExecutionCanceledEvent(decisionTaskCompletedEventID, attributes)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskScheduledEvent() *DecisionInfo {\n\tdi, err := s.msBuilder.AddDecisionTaskScheduledEvent(false)\n\ts.Nil(err)\n\treturn di\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskStartedEvent(\n\tscheduleID int64,\n\ttaskList string,\n\tidentity string,\n) *types.HistoryEvent {\n\n\tevent, _, err := s.msBuilder.AddDecisionTaskStartedEvent(scheduleID, uuid.New(), &types.PollForDecisionTaskRequest{\n\t\tTaskList: &types.TaskList{Name: taskList},\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskCompletedEvent(\n\tscheduleID int64,\n\tstartedID int64,\n\tcontext []byte,\n\tidentity string,\n) *types.HistoryEvent {\n\n\tevent, err := s.msBuilder.AddDecisionTaskCompletedEvent(scheduleID, startedID, &types.RespondDecisionTaskCompletedRequest{\n\t\tExecutionContext: context,\n\t\tIdentity:         identity,\n\t}, commonconstants.DefaultHistoryMaxAutoResetPoints)\n\ts.Nil(err)\n\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskTimedOutEvent(\n\tscheduleID int64,\n) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddDecisionTaskScheduleToStartTimeoutEvent(scheduleID)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskStartedTimedOutEvent(\n\tscheduleID int64,\n\tstartID int64,\n) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddDecisionTaskTimedOutEvent(scheduleID, startID)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addDecisionTaskResetTimedOutEvent(\n\tscheduleID int64,\n\tbaseRunID string,\n\tnewRunID string,\n\tnewRunVersion int64,\n\tresetRequestID string,\n) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddDecisionTaskResetTimeoutEvent(\n\t\tscheduleID,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tnewRunVersion,\n\t\t\"\",\n\t\tresetRequestID,\n\t)\n\ts.Nil(err)\n\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addTimerStartedEvent(decisionCompletedEventID int64, request *types.StartTimerDecisionAttributes) *types.HistoryEvent {\n\tevent, _, err := s.msBuilder.AddTimerStartedEvent(decisionCompletedEventID, request)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addTimerFiredEvent(timerID string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddTimerFiredEvent(timerID)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addTimerCanceledEvent(decisionCompletedEventID int64, attributes *types.CancelTimerDecisionAttributes, identity string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddTimerCanceledEvent(decisionCompletedEventID, attributes, identity)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addTimerCancelFailedEvent(decisionCompletedEventID int64, attributes *types.CancelTimerDecisionAttributes, identity string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddCancelTimerFailedEvent(decisionCompletedEventID, attributes, identity)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addActivityTaskScheduledEvent(\n\tdecisionCompletedID int64,\n\tactivityID, activityType, domain, taskList string,\n\tinput []byte,\n\ttimeout, queueTimeout, hearbeatTimeout int32,\n\tretryPolicy *types.RetryPolicy,\n\trequestLocalDispatch bool,\n) (*types.HistoryEvent, *persistence.ActivityInfo, *types.ActivityLocalDispatchInfo) {\n\tevent, ai, activityDispatchInfo, _, _, err := s.msBuilder.AddActivityTaskScheduledEvent(nil, decisionCompletedID,\n\t\t&types.ScheduleActivityTaskDecisionAttributes{\n\t\t\tActivityID:                    activityID,\n\t\t\tActivityType:                  &types.ActivityType{Name: activityType},\n\t\t\tDomain:                        domain,\n\t\t\tTaskList:                      &types.TaskList{Name: taskList},\n\t\t\tInput:                         input,\n\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(queueTimeout),\n\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(hearbeatTimeout),\n\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(1),\n\t\t\tRetryPolicy:                   retryPolicy,\n\t\t\tRequestLocalDispatch:          requestLocalDispatch,\n\t\t}, false,\n\t)\n\ts.Nil(err)\n\tif domain == \"\" {\n\t\ts.Equal(s.domainID, ai.DomainID)\n\t} else {\n\t\ts.Equal(s.targetDomainID, ai.DomainID)\n\t}\n\treturn event, ai, activityDispatchInfo\n}\n\nfunc (s *historyBuilderSuite) addActivityTaskStartedEvent(scheduleID int64, taskList,\n\tidentity string) *types.HistoryEvent {\n\tai, _ := s.msBuilder.GetActivityInfo(scheduleID)\n\tevent, err := s.msBuilder.AddActivityTaskStartedEvent(ai, scheduleID, uuid.New(), identity)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addActivityTaskCompletedEvent(scheduleID, startedID int64, result []byte,\n\tidentity string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddActivityTaskCompletedEvent(scheduleID, startedID, &types.RespondActivityTaskCompletedRequest{\n\t\tResult:   result,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addActivityTaskFailedEvent(scheduleID, startedID int64, reason string, details []byte,\n\tidentity string) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddActivityTaskFailedEvent(scheduleID, startedID, &types.RespondActivityTaskFailedRequest{\n\t\tReason:   common.StringPtr(reason),\n\t\tDetails:  details,\n\t\tIdentity: identity,\n\t})\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addActivityTaskTimedOutEvent(scheduleEventID,\n\tstartedEventID int64,\n\ttimeoutType types.TimeoutType,\n\tlastHeartBeatDetails []byte) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddActivityTaskTimedOutEvent(scheduleEventID, startedEventID, timeoutType, lastHeartBeatDetails)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addMarkerRecordedEvent(decisionCompletedEventID int64, markerName string, details []byte, header *map[string][]byte) *types.HistoryEvent {\n\tfields := make(map[string][]byte)\n\tif header != nil {\n\t\tfor name, value := range *header {\n\t\t\tfields[name] = value\n\t\t}\n\t}\n\tevent, err := s.msBuilder.AddRecordMarkerEvent(decisionCompletedEventID, &types.RecordMarkerDecisionAttributes{\n\t\tMarkerName: markerName,\n\t\tDetails:    details,\n\t\tHeader: &types.Header{\n\t\t\tFields: fields,\n\t\t},\n\t})\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\tdecisionCompletedEventID int64, targetDomain string, targetExecution types.WorkflowExecution,\n\tchildWorkflowOnly bool) *types.HistoryEvent {\n\tevent, _, err := s.msBuilder.AddRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\tdecisionCompletedEventID,\n\t\tuuid.New(),\n\t\t&types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:            targetDomain,\n\t\t\tWorkflowID:        targetExecution.WorkflowID,\n\t\t\tRunID:             targetExecution.RunID,\n\t\t\tChildWorkflowOnly: childWorkflowOnly,\n\t\t},\n\t)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addExternalWorkflowExecutionCancelRequested(\n\tinitiatedID int64, domain, workflowID, runID string) *types.HistoryEvent {\n\n\tevent, err := s.msBuilder.AddExternalWorkflowExecutionCancelRequested(\n\t\tinitiatedID, domain, workflowID, runID,\n\t)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addRequestCancelExternalWorkflowExecutionFailedEvent(\n\tdecisionTaskCompletedEventID, initiatedID int64,\n\tdomain, workflowID, runID string, cause types.CancelExternalWorkflowExecutionFailedCause) *types.HistoryEvent {\n\n\tevent, err := s.msBuilder.AddRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\tdecisionTaskCompletedEventID, initiatedID, domain, workflowID, runID, cause,\n\t)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionCancelRequestedEvent(\n\tcause string,\n\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddWorkflowExecutionCancelRequestedEvent(cause, request)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) addWorkflowExecutionSignaledEvent(\n\tsignalName string,\n\tinput []byte,\n\tidentity string,\n\trequestID string,\n) *types.HistoryEvent {\n\tevent, err := s.msBuilder.AddWorkflowExecutionSignaled(signalName, input, identity, requestID)\n\ts.Nil(err)\n\treturn event\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionStartedEvent(event *types.HistoryEvent, workflowType,\n\ttaskList string, input []byte, executionStartToCloseTimeout, taskStartToCloseTimeout int32, identity string, partitionConfig map[string]string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionStarted, *event.EventType)\n\ts.Equal(commonconstants.FirstEventID, event.ID)\n\tattributes := event.WorkflowExecutionStartedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(workflowType, attributes.WorkflowType.Name)\n\ts.Equal(taskList, attributes.TaskList.Name)\n\ts.Equal(input, attributes.Input)\n\ts.Equal(executionStartToCloseTimeout, *attributes.ExecutionStartToCloseTimeoutSeconds)\n\ts.Equal(taskStartToCloseTimeout, *attributes.TaskStartToCloseTimeoutSeconds)\n\ts.Equal(identity, attributes.Identity)\n\ts.Equal(partitionConfig, attributes.PartitionConfig)\n\tif attributes.CronSchedule == \"\" {\n\t\ts.Nil(attributes.FirstScheduleTime)\n\t} else {\n\t\ts.NotNil(attributes.FirstScheduleTime)\n\t}\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionFailed(event *types.HistoryEvent, eventID, decisionCompletedID int64, failReason string, failDetail []byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionFailed, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionFailedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionCompletedID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(failReason, *attributes.Reason)\n\ts.Equal(failDetail, attributes.Details)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionTimedOut(event *types.HistoryEvent, eventID int64) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionTimedOut, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionTimedOutEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(types.TimeoutTypeStartToClose, *attributes.TimeoutType)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionCompleted(event *types.HistoryEvent, eventID, decisionCompletedID int64, result []byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionCompleted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionCompletedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionCompletedID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(result, attributes.Result)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionTerminated(event *types.HistoryEvent, eventID int64, reason string, details []byte, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionTerminated, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionTerminatedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(reason, attributes.Reason)\n\ts.Equal(details, attributes.Details)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionCanceled(event *types.HistoryEvent, eventID, decisionTaskCompletedID int64, details []byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionCanceled, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionCanceledEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionTaskCompletedID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(details, attributes.Details)\n}\n\nfunc (s *historyBuilderSuite) validateDecisionTaskScheduledEvent(di *DecisionInfo, eventID int64,\n\ttaskList string, timeout int32) {\n\ts.NotNil(di)\n\ts.Equal(eventID, di.ScheduleID)\n\ts.Equal(taskList, di.TaskList)\n}\n\nfunc (s *historyBuilderSuite) validateDecisionTaskStartedEvent(event *types.HistoryEvent, eventID, scheduleID int64,\n\tidentity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeDecisionTaskStarted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.DecisionTaskStartedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateDecisionTaskCompletedEvent(event *types.HistoryEvent, eventID,\n\tscheduleID, startedID int64, context []byte, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeDecisionTaskCompleted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.DecisionTaskCompletedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(startedID, attributes.StartedEventID)\n\ts.Equal(context, attributes.ExecutionContext)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateDecisionTaskTimedoutEvent(\n\tevent *types.HistoryEvent,\n\teventID int64,\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\ttimeoutType types.TimeoutType,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\treason string,\n\tcause types.DecisionTaskTimedOutCause,\n) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeDecisionTaskTimedOut, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.DecisionTaskTimedOutEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleEventID, attributes.ScheduledEventID)\n\ts.Equal(startedEventID, attributes.StartedEventID)\n\ts.Equal(timeoutType, *attributes.TimeoutType)\n\ts.Equal(baseRunID, attributes.BaseRunID)\n\ts.Equal(newRunID, attributes.NewRunID)\n\ts.Equal(forkEventVersion, attributes.ForkEventVersion)\n\ts.Equal(reason, attributes.Reason)\n\ts.Equal(cause, *attributes.Cause)\n}\n\nfunc (s *historyBuilderSuite) validateActivityTaskScheduledEvent(\n\tevent *types.HistoryEvent,\n\teventID, decisionID int64,\n\tactivityID, activityType, domain, taskList string,\n\tinput []byte, timeout,\n\tqueueTimeout, hearbeatTimeout int32,\n\tactivityDispatchInfo *types.ActivityLocalDispatchInfo,\n\trequestLocalDispatch bool,\n) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeActivityTaskScheduled, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ActivityTaskScheduledEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(activityID, attributes.ActivityID)\n\ts.Equal(activityType, attributes.ActivityType.Name)\n\ts.Equal(taskList, attributes.TaskList.Name)\n\ts.Equal(input, attributes.Input)\n\ts.Equal(timeout, *attributes.ScheduleToCloseTimeoutSeconds)\n\ts.Equal(queueTimeout, *attributes.ScheduleToStartTimeoutSeconds)\n\ts.Equal(hearbeatTimeout, *attributes.HeartbeatTimeoutSeconds)\n\tif domain != \"\" {\n\t\ts.Equal(domain, *attributes.Domain)\n\t} else {\n\t\ts.Nil(attributes.Domain)\n\t}\n\tif requestLocalDispatch {\n\t\ts.NotNil(activityDispatchInfo)\n\t} else {\n\t\ts.Nil(activityDispatchInfo)\n\t}\n}\n\nfunc (s *historyBuilderSuite) validateActivityTaskStartedEvent(event *types.HistoryEvent, eventID, scheduleID int64,\n\tidentity string, attempt int64, lastFailureReason string, lastFailureDetails []byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeActivityTaskStarted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ActivityTaskStartedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(identity, attributes.Identity)\n\ts.Equal(lastFailureReason, *attributes.LastFailureReason)\n\ts.Equal(lastFailureDetails, attributes.LastFailureDetails)\n}\n\nfunc (s *historyBuilderSuite) validateTransientActivityTaskStartedEvent(event *types.HistoryEvent, eventID, scheduleID int64,\n\tidentity string) {\n\ts.Nil(event)\n\tai, ok := s.msBuilder.GetPendingActivityInfos()[scheduleID]\n\ts.True(ok)\n\ts.NotNil(ai)\n\ts.Equal(scheduleID, ai.ScheduleID)\n\ts.Equal(identity, ai.StartedIdentity)\n}\n\nfunc (s *historyBuilderSuite) validateActivityTaskCompletedEvent(event *types.HistoryEvent, eventID,\n\tscheduleID, startedID int64, result []byte, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeActivityTaskCompleted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ActivityTaskCompletedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(startedID, attributes.StartedEventID)\n\ts.Equal(result, attributes.Result)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateActivityTaskFailedEvent(event *types.HistoryEvent, eventID,\n\tscheduleID, startedID int64, reason string, details []byte, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeActivityTaskFailed, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ActivityTaskFailedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(startedID, attributes.StartedEventID)\n\ts.Equal(reason, *attributes.Reason)\n\ts.Equal(details, attributes.Details)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateActivityTaskTimedOutEvent(event *types.HistoryEvent, eventID, scheduleID, startedID int64, timeoutType types.TimeoutType,\n\tlastHeartBeatDetails []byte, lastFailureReason string, lastFailureDetails []byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeActivityTaskTimedOut, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ActivityTaskTimedOutEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(scheduleID, attributes.ScheduledEventID)\n\ts.Equal(startedID, attributes.StartedEventID)\n\ts.Equal(timeoutType, *attributes.TimeoutType)\n\ts.Equal(lastHeartBeatDetails, attributes.Details)\n\ts.Equal(lastFailureReason, *attributes.LastFailureReason)\n\ts.Equal(lastFailureDetails, attributes.LastFailureDetails)\n}\n\nfunc (s *historyBuilderSuite) validateMarkerRecordedEvent(\n\tevent *types.HistoryEvent, eventID, decisionTaskCompletedEventID int64,\n\tmarkerName string, details []byte, header *map[string][]byte) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeMarkerRecorded, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.MarkerRecordedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionTaskCompletedEventID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(markerName, attributes.GetMarkerName())\n\ts.Equal(details, attributes.Details)\n\tif header != nil {\n\t\tfor name, value := range attributes.Header.Fields {\n\t\t\ts.Equal((*header)[name], value)\n\t\t}\n\t} else {\n\t\ts.Nil(attributes.Header)\n\t}\n}\n\nfunc (s *historyBuilderSuite) validateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\tevent *types.HistoryEvent, eventID, decisionTaskCompletedEventID int64,\n\tdomain string, execution types.WorkflowExecution, childWorkflowOnly bool) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeRequestCancelExternalWorkflowExecutionInitiated, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionTaskCompletedEventID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(domain, attributes.GetDomain())\n\ts.Equal(execution.GetWorkflowID(), attributes.WorkflowExecution.GetWorkflowID())\n\ts.Equal(execution.GetRunID(), attributes.WorkflowExecution.GetRunID())\n\ts.Equal(childWorkflowOnly, attributes.ChildWorkflowOnly)\n}\n\nfunc (s *historyBuilderSuite) validateExternalWorkflowExecutionCancelRequested(\n\tevent *types.HistoryEvent, eventID, initiatedEventID int64,\n\tdomain string, execution types.WorkflowExecution) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeExternalWorkflowExecutionCancelRequested, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.ExternalWorkflowExecutionCancelRequestedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(initiatedEventID, attributes.GetInitiatedEventID())\n\ts.Equal(domain, attributes.GetDomain())\n\ts.Equal(execution.GetWorkflowID(), attributes.WorkflowExecution.GetWorkflowID())\n\ts.Equal(execution.GetRunID(), attributes.WorkflowExecution.GetRunID())\n}\n\nfunc (s *historyBuilderSuite) validateRequestCancelExternalWorkflowExecutionFailedEvent(\n\tevent *types.HistoryEvent, eventID, decisionTaskCompletedEventID, initiatedEventID int64,\n\tdomain string, execution types.WorkflowExecution, cause types.CancelExternalWorkflowExecutionFailedCause) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeRequestCancelExternalWorkflowExecutionFailed, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(decisionTaskCompletedEventID, attributes.GetDecisionTaskCompletedEventID())\n\ts.Equal(initiatedEventID, attributes.GetInitiatedEventID())\n\ts.Equal(domain, attributes.GetDomain())\n\ts.Equal(execution.GetWorkflowID(), attributes.WorkflowExecution.GetWorkflowID())\n\ts.Equal(execution.GetRunID(), attributes.WorkflowExecution.GetRunID())\n\ts.Equal(cause, *attributes.Cause)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionCancelRequestedEvent(\n\tevent *types.HistoryEvent,\n\teventID int64,\n\tcause string,\n\tidentity string,\n\texternalInitiatedEventID *int64,\n\texecution types.WorkflowExecution,\n\trequestID string,\n) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionCancelRequested, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionCancelRequestedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(cause, attributes.Cause)\n\ts.Equal(identity, attributes.Identity)\n\ts.Equal(externalInitiatedEventID, attributes.ExternalInitiatedEventID)\n\ts.Equal(execution.GetWorkflowID(), attributes.ExternalWorkflowExecution.GetWorkflowID())\n\ts.Equal(execution.GetRunID(), attributes.ExternalWorkflowExecution.GetRunID())\n\ts.Equal(requestID, attributes.RequestID)\n}\n\nfunc (s *historyBuilderSuite) validateWorkflowExecutionSignaledEvent(\n\tevent *types.HistoryEvent,\n\teventID int64,\n\tsignalName string,\n\tinput []byte,\n\tidentity string,\n\trequestID string,\n) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeWorkflowExecutionSignaled, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.WorkflowExecutionSignaledEventAttributes\n\ts.Equal(signalName, attributes.SignalName)\n\ts.Equal(input, attributes.Input)\n\ts.Equal(identity, attributes.Identity)\n\ts.Equal(requestID, attributes.RequestID)\n}\n\nfunc (s *historyBuilderSuite) validateTimerStartedEvent(event *types.HistoryEvent, eventID, decisionTaskCompleted, startToFireTimeoutSeconds int64, timerID string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeTimerStarted, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.TimerStartedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(timerID, attributes.TimerID)\n\ts.Equal(decisionTaskCompleted, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(startToFireTimeoutSeconds, *attributes.StartToFireTimeoutSeconds)\n}\n\nfunc (s *historyBuilderSuite) validateTimerFiredEvent(event *types.HistoryEvent, eventID, startedEventID int64, timerID string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeTimerFired, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.TimerFiredEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(timerID, attributes.TimerID)\n\ts.Equal(startedEventID, attributes.StartedEventID)\n}\n\nfunc (s *historyBuilderSuite) validateTimerCanceledEvent(event *types.HistoryEvent, eventID, startedEventID, decisionTaskCompletedID int64, timerID, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeTimerCanceled, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.TimerCanceledEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(timerID, attributes.TimerID)\n\ts.Equal(startedEventID, attributes.StartedEventID)\n\ts.Equal(decisionTaskCompletedID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(identity, attributes.Identity)\n}\n\nfunc (s *historyBuilderSuite) validateTimerCancelFailedEvent(event *types.HistoryEvent, eventID, decisionTaskCompletedID int64, timerID, identity string) {\n\ts.NotNil(event)\n\ts.Equal(types.EventTypeCancelTimerFailed, *event.EventType)\n\ts.Equal(eventID, event.ID)\n\tattributes := event.CancelTimerFailedEventAttributes\n\ts.NotNil(attributes)\n\ts.Equal(timerID, attributes.TimerID)\n\ts.Equal(decisionTaskCompletedID, attributes.DecisionTaskCompletedEventID)\n\ts.Equal(identity, attributes.Identity)\n\ts.Equal(timerCancellationMsgTimerIDUnknown, attributes.Cause)\n}\n\nfunc (s *historyBuilderSuite) printHistory() string {\n\treturn thrift.FromHistory(s.builder.GetHistory()).String()\n}\n"
  },
  {
    "path": "service/history/execution/integrity.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/collection\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n// GetResurrectedTimers returns a set of timers (timer IDs) that were resurrected.\n// Meaning timers that are still pending in mutable state, but were already completed based on event history.\nfunc GetResurrectedTimers(\n\tctx context.Context,\n\tshard shard.Context,\n\tmutableState MutableState,\n) (map[string]struct{}, error) {\n\t// 1. check if there is any pending timer\n\tpendingTimerInfos := mutableState.GetPendingTimerInfos()\n\tif len(pendingTimerInfos) == 0 {\n\t\treturn map[string]struct{}{}, nil\n\t}\n\n\t// 2. scan history from the beginning and see if any\n\t// TimerFiredEvent or TimerCancelledEvent matches pending timer\n\t// NOTE: We can't read from the middle of events branch, because\n\t// we don't know the last txn id of previous event from the middle.\n\t// Reading from the middle could get invalid nodes with invalid txn ids.\n\tresurrectedTimer := make(map[string]struct{})\n\tbranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := mutableState.GetExecutionInfo().DomainID\n\titer := collection.NewPagingIterator(getHistoryPaginationFn(\n\t\tctx,\n\t\tshard,\n\t\t1,\n\t\tmutableState.GetNextEventID(),\n\t\tbranchToken,\n\t\tdomainID,\n\t))\n\tfor iter.HasNext() {\n\t\titem, err := iter.Next()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tevent := item.(*types.HistoryEvent)\n\t\tvar timerID string\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeTimerFired:\n\t\t\ttimerID = event.TimerFiredEventAttributes.TimerID\n\t\tcase types.EventTypeTimerCanceled:\n\t\t\ttimerID = event.TimerCanceledEventAttributes.TimerID\n\t\t}\n\t\tif _, ok := pendingTimerInfos[timerID]; ok && timerID != \"\" {\n\t\t\tresurrectedTimer[timerID] = struct{}{}\n\t\t}\n\t}\n\treturn resurrectedTimer, nil\n}\n\n// GetResurrectedActivities returns a set of activities (schedule IDs) that were resurrected.\n// Meaning activities that are still pending in mutable state, but were already completed based on event history.\nfunc GetResurrectedActivities(\n\tctx context.Context,\n\tshard shard.Context,\n\tmutableState MutableState,\n) (map[int64]struct{}, error) {\n\t// 1. check if there is any pending activity\n\tpendingActivityInfos := mutableState.GetPendingActivityInfos()\n\tif len(pendingActivityInfos) == 0 {\n\t\treturn map[int64]struct{}{}, nil\n\t}\n\n\t// 2. scan history from the beginning and see if any\n\t// activity termination events matches pending activity\n\t// NOTE: We can't read from the middle of events branch, because\n\t// we don't know the last txn id of previous event from the middle.\n\t// Reading from the middle could get invalid nodes with invalid txn ids.\n\tresurrectedActivity := make(map[int64]struct{})\n\tbranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainID := mutableState.GetExecutionInfo().DomainID\n\titer := collection.NewPagingIterator(getHistoryPaginationFn(\n\t\tctx,\n\t\tshard,\n\t\t1,\n\t\tmutableState.GetNextEventID(),\n\t\tbranchToken,\n\t\tdomainID,\n\t))\n\tfor iter.HasNext() {\n\t\titem, err := iter.Next()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tevent := item.(*types.HistoryEvent)\n\t\tvar scheduledID int64\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeActivityTaskCompleted:\n\t\t\tscheduledID = event.ActivityTaskCompletedEventAttributes.ScheduledEventID\n\t\tcase types.EventTypeActivityTaskFailed:\n\t\t\tscheduledID = event.ActivityTaskFailedEventAttributes.ScheduledEventID\n\t\tcase types.EventTypeActivityTaskTimedOut:\n\t\t\tscheduledID = event.ActivityTaskTimedOutEventAttributes.ScheduledEventID\n\t\tcase types.EventTypeActivityTaskCanceled:\n\t\t\tscheduledID = event.ActivityTaskCanceledEventAttributes.ScheduledEventID\n\t\t}\n\t\tif _, ok := pendingActivityInfos[scheduledID]; ok && scheduledID != 0 {\n\t\t\tresurrectedActivity[scheduledID] = struct{}{}\n\t\t}\n\t}\n\treturn resurrectedActivity, nil\n}\n\nfunc getHistoryPaginationFn(\n\tctx context.Context,\n\tshard shard.Context,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tbranchToken []byte,\n\tdomainID string,\n) collection.PaginationFn {\n\tdomainCache := shard.GetDomainCache()\n\treturn func(token []byte) ([]interface{}, []byte, error) {\n\t\thistoryEvents, _, token, _, err := persistenceutils.PaginateHistory(\n\t\t\tctx,\n\t\t\tshard.GetHistoryManager(),\n\t\t\tfalse,\n\t\t\tbranchToken,\n\t\t\tfirstEventID,\n\t\t\tnextEventID,\n\t\t\ttoken,\n\t\t\tNDCDefaultPageSize,\n\t\t\tcommon.IntPtr(shard.GetShardID()),\n\t\t\tdomainID,\n\t\t\tdomainCache,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\n\t\tvar items []interface{}\n\t\tfor _, event := range historyEvents {\n\t\t\titems = append(items, event)\n\t\t}\n\t\treturn items, token, nil\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/integrity_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestGetResurrectedTimers(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\tsetup   func(mockShard *shard.MockContext, mockMutableState *MockMutableState, domainCache *cache.MockDomainCache, manager *persistence.MockHistoryManager)\n\t\twant    map[string]struct{}\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"No pending timers\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, domainCache *cache.MockDomainCache, manager *persistence.MockHistoryManager) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{}).Times(1)\n\t\t\t},\n\t\t\twant: map[string]struct{}{},\n\t\t},\n\t\t{\n\t\t\tname: \"Timers with no corresponding events\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, domainCache *cache.MockDomainCache, manager *persistence.MockHistoryManager) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\n\t\t\t\t\t\"timer1\": {\n\t\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\t\tExpiryTime: clock.NewRealTimeSource().Now().Add(10 * time.Minute),\n\t\t\t\t\t},\n\t\t\t\t}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte(\"branchToken\"), nil).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"testDomain\"}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10)).Times(1)\n\n\t\t\t\tmockShard.EXPECT().GetHistoryManager().Return(manager).Times(1)\n\t\t\t\tmanager.EXPECT().GetHistoryTree(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTreeResponse{\n\t\t\t\t\tBranches: []*workflow.HistoryBranch{\n\t\t\t\t\t\t{TreeID: common.StringPtr(\"treeID1\"), BranchID: common.StringPtr(\"branchID1\"), Ancestors: []*workflow.HistoryBranchRange{}},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\n\t\t\t\tmanager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tmockShard.EXPECT().GetShardID().Return(1).Times(1)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(domainCache).Times(1)\n\t\t\t\tdomainCache.EXPECT().GetDomainName(\"testDomain\").Return(\"Test Domain\", nil).Times(1)\n\t\t\t},\n\t\t\twant: map[string]struct{}{},\n\t\t},\n\t\t{\n\t\t\tname: \"Error on fetching branch token\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, domainCache *cache.MockDomainCache, manager *persistence.MockHistoryManager) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\"timer1\": {\n\t\t\t\t\tTimerID:    \"timer1\",\n\t\t\t\t\tExpiryTime: clock.NewRealTimeSource().Now().Add(10 * time.Minute),\n\t\t\t\t}}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"error fetching token\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Processing multiple events\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, domainCache *cache.MockDomainCache, manager *persistence.MockHistoryManager) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\n\t\t\t\t\t\"timer1\": {TimerID: \"timer1\", ExpiryTime: clock.NewRealTimeSource().Now().Add(10 * time.Minute)},\n\t\t\t\t\t\"timer2\": {TimerID: \"timer2\", ExpiryTime: clock.NewRealTimeSource().Now().Add(10 * time.Minute)},\n\t\t\t\t}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetCurrentBranchToken().Return(nil, nil).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"testDomain\"}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10)).Times(1)\n\t\t\t\tmockShard.EXPECT().GetShardID().Return(1).Times(1)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(domainCache).Times(1)\n\t\t\t\tdomainCache.EXPECT().GetDomainName(\"testDomain\").Return(\"testDomain\", nil).Times(1)\n\n\t\t\t\teventTypeTimerFired := types.EventTypeTimerFired\n\t\t\t\teventTypeTimerCanceled := types.EventTypeTimerCanceled\n\t\t\t\tevents := []*types.HistoryEvent{\n\t\t\t\t\t{EventType: &eventTypeTimerFired, TimerFiredEventAttributes: &types.TimerFiredEventAttributes{TimerID: \"timer1\"}},\n\t\t\t\t\t{EventType: &eventTypeTimerCanceled, TimerCanceledEventAttributes: &types.TimerCanceledEventAttributes{TimerID: \"timer2\"}},\n\t\t\t\t}\n\n\t\t\t\tmockShard.EXPECT().GetHistoryManager().Return(manager).Times(1)\n\t\t\t\tmanager.EXPECT().GetHistoryTree(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTreeResponse{\n\t\t\t\t\tBranches: []*workflow.HistoryBranch{\n\t\t\t\t\t\t{TreeID: common.StringPtr(\"treeID\"), BranchID: common.StringPtr(\"branchID\")},\n\t\t\t\t\t}}, nil).AnyTimes()\n\n\t\t\t\tmanager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: events,\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twant: map[string]struct{}{\"timer1\": {}, \"timer2\": {}},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(mockCtrl)\n\n\t\t\ttc.setup(mockShard, mockMutableState, mockDomainCache, mockHistoryManager)\n\t\t\tctx := context.Background()\n\t\t\tresurrectedTimers, err := GetResurrectedTimers(ctx, mockShard, mockMutableState)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"GetResurrectedTimers() should have returned an error\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"GetResurrectedTimers() should not have returned an error\")\n\t\t\t\tassert.Equal(t, tc.want, resurrectedTimers, \"Mismatch in expected and actual resurrected timers\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetResurrectedActivities(t *testing.T) {\n\ttests := []struct {\n\t\tname    string\n\t\tsetup   func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockHistoryManager *persistence.MockHistoryManager, mockDomainCache *cache.MockDomainCache)\n\t\twant    map[int64]struct{}\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tname: \"No pending activities\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockHistoryManager *persistence.MockHistoryManager, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{}).Times(1)\n\t\t\t},\n\t\t\twant: map[int64]struct{}{},\n\t\t},\n\t\t{\n\t\t\tname: \"With pending activities and matching events\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockHistoryManager *persistence.MockHistoryManager, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tpendingActivities := map[int64]*persistence.ActivityInfo{\n\t\t\t\t\t1: {ScheduleID: 1},\n\t\t\t\t\t2: {ScheduleID: 2},\n\t\t\t\t\t3: {ScheduleID: 3},\n\t\t\t\t\t4: {ScheduleID: 4},\n\t\t\t\t}\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(pendingActivities).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte(\"branchToken\"), nil).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"testDomain\"}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(int64(10)).Times(1)\n\n\t\t\t\ttaskCompleted := types.EventTypeActivityTaskCompleted\n\t\t\t\ttaskFailed := types.EventTypeActivityTaskFailed\n\t\t\t\ttaskTimedOut := types.EventTypeActivityTaskTimedOut\n\t\t\t\ttaskCanceled := types.EventTypeActivityTaskCanceled\n\n\t\t\t\tevents := []*types.HistoryEvent{\n\t\t\t\t\t{EventType: &taskCompleted, ActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{ScheduledEventID: 1}},\n\t\t\t\t\t{EventType: &taskFailed, ActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{ScheduledEventID: 2}},\n\t\t\t\t\t{EventType: &taskTimedOut, ActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{ScheduledEventID: 3}},\n\t\t\t\t\t{EventType: &taskCanceled, ActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{ScheduledEventID: 4}},\n\t\t\t\t}\n\n\t\t\t\tmockShard.EXPECT().GetShardID().Return(1).Times(1)\n\t\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"testDomain\", nil).Times(1)\n\n\t\t\t\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(1)\n\t\t\t\tmockHistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: events,\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t},\n\t\t\twant: map[int64]struct{}{1: {}, 2: {}, 3: {}, 4: {}},\n\t\t},\n\t\t{\n\t\t\tname: \"Error fetching branch token\",\n\t\t\tsetup: func(mockShard *shard.MockContext, mockMutableState *MockMutableState, mockHistoryManager *persistence.MockHistoryManager, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockMutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{1: {ScheduleID: 1}}).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"error fetching token\")).Times(1)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockShard := shard.NewMockContext(mockCtrl)\n\t\t\tmockMutableState := NewMockMutableState(mockCtrl)\n\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(mockCtrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\n\t\t\ttc.setup(mockShard, mockMutableState, mockHistoryManager, mockDomainCache)\n\n\t\t\tctx := context.Background()\n\t\t\tgot, err := GetResurrectedActivities(ctx, mockShard, mockMutableState)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err, \"GetResurrectedActivities() should have returned an error\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err, \"GetResurrectedActivities() should not have returned an error\")\n\t\t\t\tassert.Equal(t, tc.want, got, \"Mismatch in expected and actual resurrected activities\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mutable_state_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/query\"\n)\n\ntype (\n\t// DecisionInfo should be part of persistence layer\n\tDecisionInfo struct {\n\t\tVersion         int64\n\t\tScheduleID      int64\n\t\tStartedID       int64\n\t\tRequestID       string\n\t\tDecisionTimeout int32\n\t\tTaskList        string // This is only needed to communicate tasklist used after AddDecisionTaskScheduledEvent\n\t\tAttempt         int64\n\t\t// Scheduled and Started timestamps are useful for transient decision: when transient decision finally completes,\n\t\t// use these timestamp to create scheduled/started events.\n\t\t// Also used for recording latency metrics\n\t\tScheduledTimestamp int64\n\t\tStartedTimestamp   int64\n\t\t// OriginalScheduledTimestamp is to record the first scheduled decision during decision heartbeat.\n\t\t// Client may heartbeat decision by RespondDecisionTaskComplete with ForceCreateNewDecisionTask == true\n\t\t// In this case, OriginalScheduledTimestamp won't change. Then when current time - OriginalScheduledTimestamp exceeds\n\t\t// some threshold, server can interrupt the heartbeat by enforcing to timeout the decision.\n\t\tOriginalScheduledTimestamp int64\n\t}\n\n\t// MutableState contains the current workflow execution state\n\tMutableState interface {\n\t\tAddActivityTaskCancelRequestedEvent(int64, string, string) (*types.HistoryEvent, *persistence.ActivityInfo, error)\n\t\tAddActivityTaskCanceledEvent(int64, int64, int64, []uint8, string) (*types.HistoryEvent, error)\n\t\tAddActivityTaskCompletedEvent(int64, int64, *types.RespondActivityTaskCompletedRequest) (*types.HistoryEvent, error)\n\t\tAddActivityTaskFailedEvent(int64, int64, *types.RespondActivityTaskFailedRequest) (*types.HistoryEvent, error)\n\t\tAddActivityTaskScheduledEvent(context.Context, int64, *types.ScheduleActivityTaskDecisionAttributes, bool) (*types.HistoryEvent, *persistence.ActivityInfo, *types.ActivityLocalDispatchInfo, bool, bool, error)\n\t\tAddActivityTaskStartedEvent(*persistence.ActivityInfo, int64, string, string) (*types.HistoryEvent, error)\n\t\tAddActivityTaskTimedOutEvent(int64, int64, types.TimeoutType, []uint8) (*types.HistoryEvent, error)\n\t\tAddCancelTimerFailedEvent(int64, *types.CancelTimerDecisionAttributes, string) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionCanceledEvent(int64, *types.WorkflowExecution, *types.WorkflowExecutionCanceledEventAttributes) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionCompletedEvent(int64, *types.WorkflowExecution, *types.WorkflowExecutionCompletedEventAttributes) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionFailedEvent(int64, *types.WorkflowExecution, *types.WorkflowExecutionFailedEventAttributes) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionStartedEvent(string, *types.WorkflowExecution, *types.WorkflowType, int64, *types.Header) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionTerminatedEvent(int64, *types.WorkflowExecution, *types.WorkflowExecutionTerminatedEventAttributes) (*types.HistoryEvent, error)\n\t\tAddChildWorkflowExecutionTimedOutEvent(int64, *types.WorkflowExecution, *types.WorkflowExecutionTimedOutEventAttributes) (*types.HistoryEvent, error)\n\t\tAddCompletedWorkflowEvent(int64, *types.CompleteWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error)\n\t\tAddContinueAsNewEvent(context.Context, int64, int64, string, *types.ContinueAsNewWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, MutableState, error)\n\t\tAddDecisionTaskCompletedEvent(int64, int64, *types.RespondDecisionTaskCompletedRequest, int) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskFailedEvent(scheduleEventID int64, startedEventID int64, cause types.DecisionTaskFailedCause, details []byte, identity, reason, binChecksum, baseRunID, newRunID string, forkEventVersion int64, resetRequestID string) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskScheduleToStartTimeoutEvent(int64) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskResetTimeoutEvent(int64, string, string, int64, string, string) (*types.HistoryEvent, error)\n\t\tAddFirstDecisionTaskScheduled(*types.HistoryEvent) error\n\t\tAddDecisionTaskScheduledEvent(bypassTaskGeneration bool) (*DecisionInfo, error)\n\t\tAddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration bool, originalScheduledTimestamp int64) (*DecisionInfo, error)\n\t\tAddDecisionTaskStartedEvent(int64, string, *types.PollForDecisionTaskRequest) (*types.HistoryEvent, *DecisionInfo, error)\n\t\tAddDecisionTaskTimedOutEvent(int64, int64) (*types.HistoryEvent, error)\n\t\tAddExternalWorkflowExecutionCancelRequested(int64, string, string, string) (*types.HistoryEvent, error)\n\t\tAddExternalWorkflowExecutionSignaled(int64, string, string, string, []uint8) (*types.HistoryEvent, error)\n\t\tAddFailWorkflowEvent(int64, *types.FailWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error)\n\t\tAddRecordMarkerEvent(int64, *types.RecordMarkerDecisionAttributes) (*types.HistoryEvent, error)\n\t\tAddRequestCancelActivityTaskFailedEvent(int64, string, string) (*types.HistoryEvent, error)\n\t\tAddRequestCancelExternalWorkflowExecutionFailedEvent(int64, int64, string, string, string, types.CancelExternalWorkflowExecutionFailedCause) (*types.HistoryEvent, error)\n\t\tAddRequestCancelExternalWorkflowExecutionInitiatedEvent(int64, string, *types.RequestCancelExternalWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.RequestCancelInfo, error)\n\t\tAddSignalExternalWorkflowExecutionFailedEvent(int64, int64, string, string, string, []uint8, types.SignalExternalWorkflowExecutionFailedCause) (*types.HistoryEvent, error)\n\t\tAddSignalExternalWorkflowExecutionInitiatedEvent(int64, string, *types.SignalExternalWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.SignalInfo, error)\n\t\tAddSignalRequested(requestID string)\n\t\tAddStartChildWorkflowExecutionFailedEvent(int64, types.ChildWorkflowExecutionFailedCause, *types.StartChildWorkflowExecutionInitiatedEventAttributes) (*types.HistoryEvent, error)\n\t\tAddStartChildWorkflowExecutionInitiatedEvent(int64, string, *types.StartChildWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.ChildExecutionInfo, error)\n\t\tAddTimeoutWorkflowEvent(int64) (*types.HistoryEvent, error)\n\t\tAddTimerCanceledEvent(int64, *types.CancelTimerDecisionAttributes, string) (*types.HistoryEvent, error)\n\t\tAddTimerFiredEvent(string) (*types.HistoryEvent, error)\n\t\tAddTimerStartedEvent(int64, *types.StartTimerDecisionAttributes) (*types.HistoryEvent, *persistence.TimerInfo, error)\n\t\tAddUpsertWorkflowSearchAttributesEvent(int64, *types.UpsertWorkflowSearchAttributesDecisionAttributes) (*types.HistoryEvent, error)\n\t\tAddWorkflowExecutionCancelRequestedEvent(string, *types.HistoryRequestCancelWorkflowExecutionRequest) (*types.HistoryEvent, error)\n\t\tAddWorkflowExecutionCanceledEvent(int64, *types.CancelWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error)\n\t\tAddWorkflowExecutionSignaled(signalName string, input []byte, identity string, reqeustID string) (*types.HistoryEvent, error)\n\t\tAddWorkflowExecutionStartedEvent(types.WorkflowExecution, *types.HistoryStartWorkflowExecutionRequest) (*types.HistoryEvent, error)\n\t\tAddWorkflowExecutionTerminatedEvent(firstEventID int64, reason string, details []byte, identity string) (*types.HistoryEvent, error)\n\t\tClearStickyness()\n\t\tCheckResettable() error\n\t\tCopyToPersistence() *persistence.WorkflowMutableState\n\t\tRetryActivity(ai *persistence.ActivityInfo, failureReason string, failureDetails []byte) (bool, error)\n\t\tCreateNewHistoryEvent(eventType types.EventType) *types.HistoryEvent\n\t\tCreateNewHistoryEventWithTimestamp(eventType types.EventType, timestamp int64) *types.HistoryEvent\n\t\tCreateTransientDecisionEvents(di *DecisionInfo, identity string) (*types.HistoryEvent, *types.HistoryEvent)\n\t\tDeleteDecision()\n\t\tDeleteUserTimer(timerID string) error\n\t\tDeleteActivity(scheduleEventID int64) error\n\t\tDeleteSignalRequested(requestID string)\n\t\tFailDecision(bool)\n\t\tFlushBufferedEvents() error\n\t\tGetActivityByActivityID(string) (*persistence.ActivityInfo, bool)\n\t\tGetActivityInfo(int64) (*persistence.ActivityInfo, bool)\n\t\tGetActivityScheduledEvent(context.Context, int64) (*types.HistoryEvent, error)\n\t\tGetChildExecutionInfo(int64) (*persistence.ChildExecutionInfo, bool)\n\t\tGetChildExecutionInitiatedEvent(context.Context, int64) (*types.HistoryEvent, error)\n\t\tGetCompletionEvent(context.Context) (*types.HistoryEvent, error)\n\t\tGetDecisionInfo(int64) (*DecisionInfo, bool)\n\t\tGetDecisionScheduleToStartTimeout() time.Duration\n\t\tGetDomainEntry() *cache.DomainCacheEntry\n\t\tGetStartEvent(context.Context) (*types.HistoryEvent, error)\n\t\tGetCurrentBranchToken() ([]byte, error)\n\t\tGetVersionHistories() *persistence.VersionHistories\n\t\tGetCurrentVersion() int64\n\t\tGetExecutionInfo() *persistence.WorkflowExecutionInfo\n\t\tGetHistoryBuilder() *HistoryBuilder\n\t\tGetInFlightDecision() (*DecisionInfo, bool)\n\t\tGetPendingDecision() (*DecisionInfo, bool)\n\t\tGetLastFirstEventID() int64\n\t\tGetLastWriteVersion() (int64, error)\n\t\tGetNextEventID() int64\n\t\tGetPreviousStartedEventID() int64\n\t\tGetPendingActivityInfos() map[int64]*persistence.ActivityInfo\n\t\tGetPendingTimerInfos() map[string]*persistence.TimerInfo\n\t\tGetPendingChildExecutionInfos() map[int64]*persistence.ChildExecutionInfo\n\t\tGetPendingRequestCancelExternalInfos() map[int64]*persistence.RequestCancelInfo\n\t\tGetPendingSignalExternalInfos() map[int64]*persistence.SignalInfo\n\t\tGetRequestCancelInfo(int64) (*persistence.RequestCancelInfo, bool)\n\t\tGetRetryBackoffDuration(errReason string) time.Duration\n\t\tGetCronBackoffDuration(context.Context) (time.Duration, error)\n\t\tGetSignalInfo(int64) (*persistence.SignalInfo, bool)\n\t\tGetStartVersion() (int64, error)\n\t\tGetUserTimerInfoByEventID(int64) (*persistence.TimerInfo, bool)\n\t\tGetUserTimerInfo(string) (*persistence.TimerInfo, bool)\n\t\tGetWorkflowType() *types.WorkflowType\n\t\tGetWorkflowStateCloseStatus() (int, int)\n\t\tGetQueryRegistry() query.Registry\n\t\tSetQueryRegistry(query.Registry)\n\t\tHasBufferedEvents() bool\n\t\tHasInFlightDecision() bool\n\t\tHasParentExecution() bool\n\t\tHasPendingDecision() bool\n\t\tHasProcessedOrPendingDecision() bool\n\t\tIsCancelRequested() (bool, string)\n\t\tIsCurrentWorkflowGuaranteed() bool\n\t\tIsSignalRequested(requestID string) bool\n\t\tIsStickyTaskListEnabled() bool\n\t\tIsWorkflowExecutionRunning() bool\n\t\tIsWorkflowCompleted() bool\n\t\tIsResourceDuplicated(resourceDedupKey definition.DeduplicationID) bool\n\t\tUpdateDuplicatedResource(resourceDedupKey definition.DeduplicationID)\n\t\tLoad(context.Context, *persistence.WorkflowMutableState) error\n\t\tReplicateActivityInfo(*types.SyncActivityRequest, bool) error\n\t\tReplicateActivityTaskCancelRequestedEvent(*types.HistoryEvent) error\n\t\tReplicateActivityTaskCanceledEvent(*types.HistoryEvent) error\n\t\tReplicateActivityTaskCompletedEvent(*types.HistoryEvent) error\n\t\tReplicateActivityTaskFailedEvent(*types.HistoryEvent) error\n\t\tReplicateActivityTaskScheduledEvent(int64, *types.HistoryEvent, bool) (*persistence.ActivityInfo, error)\n\t\tReplicateActivityTaskStartedEvent(*types.HistoryEvent) error\n\t\tReplicateActivityTaskTimedOutEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionCanceledEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionCompletedEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionFailedEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionStartedEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionTerminatedEvent(*types.HistoryEvent) error\n\t\tReplicateChildWorkflowExecutionTimedOutEvent(*types.HistoryEvent) error\n\t\tReplicateDecisionTaskCompletedEvent(*types.HistoryEvent) error\n\t\tReplicateDecisionTaskFailedEvent(*types.HistoryEvent) error\n\t\tReplicateDecisionTaskScheduledEvent(int64, int64, string, int32, int64, int64, int64, bool) (*DecisionInfo, error)\n\t\tReplicateDecisionTaskStartedEvent(*DecisionInfo, int64, int64, int64, string, int64) (*DecisionInfo, error)\n\t\tReplicateDecisionTaskTimedOutEvent(*types.HistoryEvent) error\n\t\tReplicateExternalWorkflowExecutionCancelRequested(*types.HistoryEvent) error\n\t\tReplicateExternalWorkflowExecutionSignaled(*types.HistoryEvent) error\n\t\tReplicateRequestCancelExternalWorkflowExecutionFailedEvent(*types.HistoryEvent) error\n\t\tReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(int64, *types.HistoryEvent, string) (*persistence.RequestCancelInfo, error)\n\t\tReplicateSignalExternalWorkflowExecutionFailedEvent(*types.HistoryEvent) error\n\t\tReplicateSignalExternalWorkflowExecutionInitiatedEvent(int64, *types.HistoryEvent, string) (*persistence.SignalInfo, error)\n\t\tReplicateStartChildWorkflowExecutionFailedEvent(*types.HistoryEvent) error\n\t\tReplicateStartChildWorkflowExecutionInitiatedEvent(int64, *types.HistoryEvent, string) (*persistence.ChildExecutionInfo, error)\n\t\tReplicateTimerCanceledEvent(*types.HistoryEvent) error\n\t\tReplicateTimerFiredEvent(*types.HistoryEvent) error\n\t\tReplicateTimerStartedEvent(*types.HistoryEvent) (*persistence.TimerInfo, error)\n\t\tReplicateTransientDecisionTaskScheduled() error\n\t\tReplicateUpsertWorkflowSearchAttributesEvent(*types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionCancelRequestedEvent(*types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionCanceledEvent(int64, *types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionCompletedEvent(int64, *types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionContinuedAsNewEvent(int64, string, *types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionFailedEvent(int64, *types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionSignaled(*types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionStartedEvent(*string, types.WorkflowExecution, string, *types.HistoryEvent, bool) error\n\t\tReplicateWorkflowExecutionTerminatedEvent(int64, *types.HistoryEvent) error\n\t\tReplicateWorkflowExecutionTimedoutEvent(int64, *types.HistoryEvent) error\n\t\tSetCurrentBranchToken(branchToken []byte) error\n\t\tSetHistoryBuilder(hBuilder *HistoryBuilder)\n\t\tSetHistoryTree(treeID string) error\n\t\tSetVersionHistories(*persistence.VersionHistories) error\n\t\tUpdateActivity(*persistence.ActivityInfo) error\n\t\tUpdateActivityProgress(ai *persistence.ActivityInfo, request *types.RecordActivityTaskHeartbeatRequest)\n\t\tUpdateDecision(*DecisionInfo)\n\t\tUpdateUserTimer(*persistence.TimerInfo) error\n\t\tUpdateCurrentVersion(version int64, forceUpdate bool) error\n\t\tUpdateWorkflowStateCloseStatus(state int, closeStatus int) error\n\n\t\tAddTransferTasks(transferTasks ...persistence.Task)\n\t\tAddTimerTasks(timerTasks ...persistence.Task)\n\t\tGetTransferTasks() []persistence.Task\n\t\tGetTimerTasks() []persistence.Task\n\t\tDeleteTransferTasks()\n\t\tDeleteTimerTasks()\n\n\t\tSetUpdateCondition(int64)\n\t\tGetUpdateCondition() int64\n\n\t\tStartTransaction(ctx context.Context, entry *cache.DomainCacheEntry, incomingTaskVersion int64) (bool, error)\n\t\tCloseTransactionAsMutation(now time.Time, transactionPolicy TransactionPolicy) (*persistence.WorkflowMutation, []*persistence.WorkflowEvents, error)\n\t\tCloseTransactionAsSnapshot(now time.Time, transactionPolicy TransactionPolicy) (*persistence.WorkflowSnapshot, []*persistence.WorkflowEvents, error)\n\n\t\tGetHistorySize() int64\n\t\tSetHistorySize(size int64)\n\n\t\tcache.Sizeable\n\t}\n)\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"runtime/debug\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tmutableStateInvalidHistoryActionMsg         = \"invalid history builder state for action\"\n\tmutableStateInvalidHistoryActionMsgTemplate = mutableStateInvalidHistoryActionMsg + \": %v\"\n\n\ttimerCancellationMsgTimerIDUnknown = \"TIMER_ID_UNKNOWN\"\n)\n\nvar (\n\t// ErrWorkflowFinished indicates trying to mutate mutable state after workflow finished\n\tErrWorkflowFinished = &types.InternalServiceError{Message: \"invalid mutable state action: mutation after finish\"}\n\t// ErrMissingTimerInfo indicates missing timer info\n\tErrMissingTimerInfo = &types.InternalServiceError{Message: \"unable to get timer info\"}\n\t// ErrMissingActivityInfo indicates missing activity info\n\tErrMissingActivityInfo = &types.InternalServiceError{Message: \"unable to get activity info\"}\n\t// ErrMissingChildWorkflowInfo indicates missing child workflow info\n\tErrMissingChildWorkflowInfo = &types.InternalServiceError{Message: \"unable to get child workflow info\"}\n\t// ErrMissingWorkflowStartEvent indicates missing workflow start event\n\tErrMissingWorkflowStartEvent = &types.InternalServiceError{Message: \"unable to get workflow start event\"}\n\t// ErrMissingWorkflowCompletionEvent indicates missing workflow completion event\n\tErrMissingWorkflowCompletionEvent = &types.InternalServiceError{Message: \"unable to get workflow completion event\"}\n\t// ErrMissingActivityScheduledEvent indicates missing workflow activity scheduled event\n\tErrMissingActivityScheduledEvent = &types.InternalServiceError{Message: \"unable to get activity scheduled event\"}\n\t// ErrMissingChildWorkflowInitiatedEvent indicates missing child workflow initiated event\n\tErrMissingChildWorkflowInitiatedEvent = &types.InternalServiceError{Message: \"unable to get child workflow initiated event\"}\n\t// ErrEventsAfterWorkflowFinish is the error indicating server error trying to write events after workflow finish event\n\tErrEventsAfterWorkflowFinish = &types.InternalServiceError{Message: \"error validating last event being workflow finish event\"}\n\t// ErrMissingVersionHistories is the error indicating cadence failed to process 2dc workflow type.\n\tErrMissingVersionHistories = &types.BadRequestError{Message: \"versionHistories is empty, which is required for NDC feature. It's probably from deprecated 2dc workflows\"}\n\t// ErrTooManyPendingActivities is the error that currently there are too many pending activities in the workflow\n\tErrTooManyPendingActivities = &types.InternalServiceError{Message: \"Too many pending activities\"}\n)\n\ntype (\n\tmutableStateBuilder struct {\n\t\tpendingActivityInfoIDs     map[int64]*persistence.ActivityInfo // Schedule Event ID -> Activity Info.\n\t\tpendingActivityIDToEventID map[string]int64                    // Activity ID -> Schedule Event ID of the activity.\n\t\tupdateActivityInfos        map[int64]*persistence.ActivityInfo // Modified activities from last update.\n\t\tdeleteActivityInfos        map[int64]struct{}                  // Deleted activities from last update.\n\t\tsyncActivityTasks          map[int64]struct{}                  // Activity to be sync to remote\n\n\t\tpendingTimerInfoIDs     map[string]*persistence.TimerInfo // User Timer ID -> Timer Info.\n\t\tpendingTimerEventIDToID map[int64]string                  // User Timer Start Event ID -> User Timer ID.\n\t\tupdateTimerInfos        map[string]*persistence.TimerInfo // Modified timers from last update.\n\t\tdeleteTimerInfos        map[string]struct{}               // Deleted timers from last update.\n\n\t\tpendingChildExecutionInfoIDs map[int64]*persistence.ChildExecutionInfo // Initiated Event ID -> Child Execution Info\n\t\tupdateChildExecutionInfos    map[int64]*persistence.ChildExecutionInfo // Modified ChildExecution Infos since last update\n\t\tdeleteChildExecutionInfos    map[int64]struct{}                        // Deleted ChildExecution Infos since last update\n\n\t\tpendingRequestCancelInfoIDs map[int64]*persistence.RequestCancelInfo // Initiated Event ID -> RequestCancelInfo\n\t\tupdateRequestCancelInfos    map[int64]*persistence.RequestCancelInfo // Modified RequestCancel Infos since last update, for persistence update\n\t\tdeleteRequestCancelInfos    map[int64]struct{}                       // Deleted RequestCancel Infos since last update, for persistence update\n\n\t\tpendingSignalInfoIDs map[int64]*persistence.SignalInfo // Initiated Event ID -> SignalInfo\n\t\tupdateSignalInfos    map[int64]*persistence.SignalInfo // Modified SignalInfo since last update\n\t\tdeleteSignalInfos    map[int64]struct{}                // Deleted SignalInfos since last update\n\n\t\tpendingSignalRequestedIDs map[string]struct{} // Set of signaled requestIds\n\t\tupdateSignalRequestedIDs  map[string]struct{} // Set of signaled requestIds since last update\n\t\tdeleteSignalRequestedIDs  map[string]struct{} // Deleted signaled requestIds\n\n\t\tbufferedEvents       []*types.HistoryEvent // buffered history events that are already persisted\n\t\tupdateBufferedEvents []*types.HistoryEvent // buffered history events that needs to be persisted\n\t\tclearBufferedEvents  bool                  // delete buffered events from persistence\n\n\t\t// This section includes Workflow Execution Info parameters like StartTimestamp,\n\t\t// which are only visible after the workflow has begun.\n\t\t// However, there are other parameters such as LastEventTimestamp,\n\t\t// which are updated as the execution progresses through various cycles.\n\t\t// It's common to encounter null values for these timestamps initially,\n\t\t// as they are populated once the update cycles are initiated.\n\t\texecutionInfo    *persistence.WorkflowExecutionInfo // Workflow mutable state info.\n\t\tversionHistories *persistence.VersionHistories\n\t\t// TODO: remove this struct after all 2DC workflows complete\n\t\treplicationState *persistence.ReplicationState\n\t\thBuilder         *HistoryBuilder\n\n\t\t// in memory only attributes\n\t\t// indicate the current version\n\t\tcurrentVersion int64\n\t\t// indicates whether there are buffered events in persistence\n\t\thasBufferedEventsInDB bool\n\t\t// indicates the workflow state in DB, can be used to calculate\n\t\t// whether this workflow is pointed by current workflow record\n\t\tstateInDB int\n\t\t// indicates the next event ID in DB, for conditional update\n\t\tnextEventIDInDB int64\n\t\t// domain entry contains a snapshot of domain\n\t\t// NOTE: do not use the failover version inside, use currentVersion above\n\t\tdomainEntry *cache.DomainCacheEntry\n\t\t// record if a event has been applied to mutable state\n\t\t// TODO: persist this to db\n\t\tappliedEvents map[string]struct{}\n\n\t\tinsertTransferTasks    []persistence.Task\n\t\tinsertReplicationTasks []persistence.Task\n\t\tinsertTimerTasks       []persistence.Task\n\n\t\tworkflowRequests map[persistence.WorkflowRequest]struct{}\n\n\t\t// do not rely on this, this is only updated on\n\t\t// Load() and closeTransactionXXX methods. So when\n\t\t// a transaction is in progress, this value will be\n\t\t// wrong. This exist primarily for visibility via CLI\n\t\tchecksum checksum.Checksum\n\n\t\ttaskGenerator       MutableStateTaskGenerator\n\t\tdecisionTaskManager mutableStateDecisionTaskManager\n\t\tqueryRegistry       query.Registry\n\n\t\tshard                      shard.Context\n\t\tclusterMetadata            cluster.Metadata\n\t\teventsCache                events.Cache\n\t\tconfig                     *config.Config\n\t\ttimeSource                 clock.TimeSource\n\t\tlogger                     log.Logger\n\t\tmetricsClient              metrics.Client\n\t\tpendingActivityWarningSent bool\n\n\t\texecutionStats *persistence.ExecutionStats\n\t}\n)\n\nvar _ MutableState = (*mutableStateBuilder)(nil)\n\n// NewMutableStateBuilder creates a new workflow mutable state builder\nfunc NewMutableStateBuilder(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tdomainEntry *cache.DomainCacheEntry,\n) MutableState {\n\treturn newMutableStateBuilder(shard, logger, domainEntry, constants.EmptyVersion)\n}\n\nfunc newMutableStateBuilder(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tdomainEntry *cache.DomainCacheEntry,\n\tcurrentVersion int64,\n) *mutableStateBuilder {\n\ts := &mutableStateBuilder{\n\t\tupdateActivityInfos:        make(map[int64]*persistence.ActivityInfo),\n\t\tpendingActivityInfoIDs:     make(map[int64]*persistence.ActivityInfo),\n\t\tpendingActivityIDToEventID: make(map[string]int64),\n\t\tdeleteActivityInfos:        make(map[int64]struct{}),\n\t\tsyncActivityTasks:          make(map[int64]struct{}),\n\n\t\tpendingTimerInfoIDs:     make(map[string]*persistence.TimerInfo),\n\t\tpendingTimerEventIDToID: make(map[int64]string),\n\t\tupdateTimerInfos:        make(map[string]*persistence.TimerInfo),\n\t\tdeleteTimerInfos:        make(map[string]struct{}),\n\n\t\tupdateChildExecutionInfos:    make(map[int64]*persistence.ChildExecutionInfo),\n\t\tpendingChildExecutionInfoIDs: make(map[int64]*persistence.ChildExecutionInfo),\n\t\tdeleteChildExecutionInfos:    make(map[int64]struct{}),\n\n\t\tupdateRequestCancelInfos:    make(map[int64]*persistence.RequestCancelInfo),\n\t\tpendingRequestCancelInfoIDs: make(map[int64]*persistence.RequestCancelInfo),\n\t\tdeleteRequestCancelInfos:    make(map[int64]struct{}),\n\n\t\tupdateSignalInfos:    make(map[int64]*persistence.SignalInfo),\n\t\tpendingSignalInfoIDs: make(map[int64]*persistence.SignalInfo),\n\t\tdeleteSignalInfos:    make(map[int64]struct{}),\n\n\t\tupdateSignalRequestedIDs:  make(map[string]struct{}),\n\t\tpendingSignalRequestedIDs: make(map[string]struct{}),\n\t\tdeleteSignalRequestedIDs:  make(map[string]struct{}),\n\n\t\tcurrentVersion:        currentVersion,\n\t\thasBufferedEventsInDB: false,\n\t\tstateInDB:             persistence.WorkflowStateVoid,\n\t\tnextEventIDInDB:       0,\n\t\tdomainEntry:           domainEntry,\n\t\tappliedEvents:         make(map[string]struct{}),\n\n\t\tqueryRegistry: query.NewRegistry(),\n\n\t\tshard:           shard,\n\t\tclusterMetadata: shard.GetClusterMetadata(),\n\t\teventsCache:     shard.GetEventsCache(),\n\t\tconfig:          shard.GetConfig(),\n\t\ttimeSource:      shard.GetTimeSource(),\n\t\tlogger:          logger,\n\t\tmetricsClient:   shard.GetMetricsClient(),\n\n\t\tworkflowRequests: make(map[persistence.WorkflowRequest]struct{}),\n\t}\n\ts.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\tDecisionVersion:    constants.EmptyVersion,\n\t\tDecisionScheduleID: constants.EmptyEventID,\n\t\tDecisionStartedID:  constants.EmptyEventID,\n\t\tDecisionRequestID:  constants.EmptyUUID,\n\t\tDecisionTimeout:    0,\n\n\t\tNextEventID:        constants.FirstEventID,\n\t\tState:              persistence.WorkflowStateCreated,\n\t\tCloseStatus:        persistence.WorkflowCloseStatusNone,\n\t\tLastProcessedEvent: constants.EmptyEventID,\n\t\tCronOverlapPolicy:  types.CronOverlapPolicySkipped,\n\t}\n\ts.hBuilder = NewHistoryBuilder(s)\n\ts.taskGenerator = NewMutableStateTaskGenerator(shard.GetLogger(), shard.GetClusterMetadata(), shard.GetDomainCache(), s)\n\ts.decisionTaskManager = newMutableStateDecisionTaskManager(s)\n\ts.executionStats = &persistence.ExecutionStats{}\n\treturn s\n}\n\n// NewMutableStateBuilderWithVersionHistories creates mutable state builder with version history initialized\n// NOTE: currentVersion should be the failover version of the workflow, which is derived from domain metadata and\n// the active cluster selection policy of the workflow. For passive workflows, the currentVersion will be overridden by the event version,\n// so the input doesn't matter for them.\nfunc NewMutableStateBuilderWithVersionHistories(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tdomainEntry *cache.DomainCacheEntry,\n\tcurrentVersion int64,\n) MutableState {\n\n\ts := newMutableStateBuilder(shard, logger, domainEntry, currentVersion)\n\ts.versionHistories = persistence.NewVersionHistories(&persistence.VersionHistory{})\n\treturn s\n}\n\n// NewMutableStateBuilderWithEventV2 is used only in test\nfunc NewMutableStateBuilderWithEventV2(\n\tshard shard.Context,\n\tlogger log.Logger,\n\trunID string,\n\tdomainEntry *cache.DomainCacheEntry,\n) MutableState {\n\n\tmsBuilder := NewMutableStateBuilder(shard, logger, domainEntry)\n\t_ = msBuilder.SetHistoryTree(runID)\n\n\treturn msBuilder\n}\n\n// NewMutableStateBuilderWithVersionHistoriesWithEventV2 is used only in test\nfunc NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tversion int64,\n\trunID string,\n\tdomainEntry *cache.DomainCacheEntry,\n) MutableState {\n\n\tmsBuilder := NewMutableStateBuilderWithVersionHistories(shard, logger, domainEntry, domainEntry.GetFailoverVersion())\n\terr := msBuilder.UpdateCurrentVersion(version, false)\n\tif err != nil {\n\t\tlogger.Error(\"update current version error\", tag.Error(err))\n\t}\n\t_ = msBuilder.SetHistoryTree(runID)\n\n\treturn msBuilder\n}\n\n// Creates a shallow copy, not safe to use if the copied struct is mutated\nfunc (e *mutableStateBuilder) CopyToPersistence() *persistence.WorkflowMutableState {\n\tstate := &persistence.WorkflowMutableState{}\n\n\tstate.ActivityInfos = e.pendingActivityInfoIDs\n\tstate.TimerInfos = e.pendingTimerInfoIDs\n\tstate.ChildExecutionInfos = e.pendingChildExecutionInfoIDs\n\tstate.RequestCancelInfos = e.pendingRequestCancelInfoIDs\n\tstate.SignalInfos = e.pendingSignalInfoIDs\n\tstate.SignalRequestedIDs = e.pendingSignalRequestedIDs\n\tstate.ExecutionInfo = e.executionInfo\n\tstate.BufferedEvents = e.bufferedEvents\n\tstate.VersionHistories = e.versionHistories\n\tstate.Checksum = e.checksum\n\tstate.ReplicationState = e.replicationState\n\tstate.ExecutionStats = e.executionStats\n\n\treturn state\n}\n\nfunc (e *mutableStateBuilder) Load(\n\tctx context.Context,\n\tstate *persistence.WorkflowMutableState,\n) error {\n\n\te.pendingActivityInfoIDs = state.ActivityInfos\n\tfor _, activityInfo := range state.ActivityInfos {\n\t\te.pendingActivityIDToEventID[activityInfo.ActivityID] = activityInfo.ScheduleID\n\t}\n\te.pendingTimerInfoIDs = state.TimerInfos\n\tfor _, timerInfo := range state.TimerInfos {\n\t\te.pendingTimerEventIDToID[timerInfo.StartedID] = timerInfo.TimerID\n\t}\n\te.pendingChildExecutionInfoIDs = state.ChildExecutionInfos\n\te.pendingRequestCancelInfoIDs = state.RequestCancelInfos\n\te.pendingSignalInfoIDs = state.SignalInfos\n\te.pendingSignalRequestedIDs = state.SignalRequestedIDs\n\te.executionInfo = state.ExecutionInfo\n\te.bufferedEvents = e.reorderAndFilterDuplicateEvents(state.BufferedEvents, \"load\")\n\n\te.currentVersion = constants.EmptyVersion\n\te.hasBufferedEventsInDB = len(e.bufferedEvents) > 0\n\te.stateInDB = state.ExecutionInfo.State\n\te.nextEventIDInDB = state.ExecutionInfo.NextEventID\n\te.versionHistories = state.VersionHistories\n\t// TODO: remove this after all 2DC workflows complete\n\te.replicationState = state.ReplicationState\n\te.checksum = state.Checksum\n\te.executionStats = state.ExecutionStats\n\n\te.fillForBackwardsCompatibility()\n\n\tif len(state.Checksum.Value) > 0 {\n\t\tswitch {\n\t\tcase e.shouldInvalidateChecksum():\n\t\t\te.checksum = checksum.Checksum{}\n\t\t\te.metricsClient.IncCounter(metrics.WorkflowContextScope, metrics.MutableStateChecksumInvalidated)\n\t\tcase e.shouldVerifyChecksum():\n\t\t\tif err := verifyMutableStateChecksum(e, state.Checksum); err != nil {\n\t\t\t\t// we ignore checksum verification errors for now until this\n\t\t\t\t// feature is tested and/or we have mechanisms in place to deal\n\t\t\t\t// with these types of errors\n\t\t\t\te.metricsClient.IncCounter(metrics.WorkflowContextScope, metrics.MutableStateChecksumMismatch)\n\t\t\t\te.logError(\"mutable state checksum mismatch\",\n\t\t\t\t\ttag.WorkflowNextEventID(e.executionInfo.NextEventID),\n\t\t\t\t\ttag.WorkflowScheduleID(e.executionInfo.DecisionScheduleID),\n\t\t\t\t\ttag.WorkflowStartedID(e.executionInfo.DecisionStartedID),\n\t\t\t\t\ttag.Dynamic(\"timerIDs\", maps.Keys(e.pendingTimerInfoIDs)),\n\t\t\t\t\ttag.Dynamic(\"activityIDs\", maps.Keys(e.pendingActivityInfoIDs)),\n\t\t\t\t\ttag.Dynamic(\"childIDs\", maps.Keys(e.pendingChildExecutionInfoIDs)),\n\t\t\t\t\ttag.Dynamic(\"signalIDs\", maps.Keys(e.pendingSignalInfoIDs)),\n\t\t\t\t\ttag.Dynamic(\"cancelIDs\", maps.Keys(e.pendingRequestCancelInfoIDs)),\n\t\t\t\t\ttag.Error(err))\n\t\t\t\tif e.enableChecksumFailureRetry() {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) fillForBackwardsCompatibility() {\n\t// With https://github.com/uber/cadence/pull/4601 newly introduced DomainID may not be set for older workflows.\n\t// Here we will fill its value based on previously used domain name.\n\tfor _, info := range e.pendingChildExecutionInfoIDs {\n\t\tif info.DomainID == \"\" && info.DomainNameDEPRECATED != \"\" {\n\t\t\tdomainID, err := e.shard.GetDomainCache().GetDomainID(info.DomainNameDEPRECATED)\n\t\t\tif err != nil {\n\t\t\t\te.logError(\"failed to fill domainId for pending child executions\", tag.Error(err))\n\t\t\t}\n\n\t\t\tinfo.DomainID = domainID\n\t\t}\n\t}\n}\n\nfunc (e *mutableStateBuilder) GetCurrentBranchToken() ([]byte, error) {\n\tif e.versionHistories != nil {\n\t\tcurrentVersionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn currentVersionHistory.GetBranchToken(), nil\n\t}\n\treturn e.executionInfo.BranchToken, nil\n}\n\nfunc (e *mutableStateBuilder) GetVersionHistories() *persistence.VersionHistories {\n\treturn e.versionHistories\n}\n\n// set treeID/historyBranches\nfunc (e *mutableStateBuilder) SetHistoryTree(\n\ttreeID string,\n) error {\n\n\tinitialBranchToken, err := persistence.NewHistoryBranchToken(treeID)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn e.SetCurrentBranchToken(initialBranchToken)\n}\n\nfunc (e *mutableStateBuilder) SetCurrentBranchToken(\n\tbranchToken []byte,\n) error {\n\n\texeInfo := e.GetExecutionInfo()\n\tif e.versionHistories == nil {\n\t\texeInfo.BranchToken = branchToken\n\t\treturn nil\n\t}\n\n\tcurrentVersionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn currentVersionHistory.SetBranchToken(branchToken)\n}\n\nfunc (e *mutableStateBuilder) SetVersionHistories(\n\tversionHistories *persistence.VersionHistories,\n) error {\n\n\te.versionHistories = versionHistories\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) GetHistoryBuilder() *HistoryBuilder {\n\treturn e.hBuilder\n}\n\nfunc (e *mutableStateBuilder) SetHistoryBuilder(hBuilder *HistoryBuilder) {\n\te.hBuilder = hBuilder\n}\n\nfunc (e *mutableStateBuilder) GetExecutionInfo() *persistence.WorkflowExecutionInfo {\n\treturn e.executionInfo\n}\n\nfunc (e *mutableStateBuilder) FlushBufferedEvents() error {\n\t// put new events into 2 buckets:\n\t//  1) if the event was added while there was in-flight decision, then put it in buffered bucket\n\t//  2) otherwise, put it in committed bucket\n\tvar newBufferedEvents []*types.HistoryEvent\n\tvar newCommittedEvents []*types.HistoryEvent\n\tfor _, event := range e.hBuilder.history {\n\t\tif event.ID == constants.BufferedEventID {\n\t\t\tnewBufferedEvents = append(newBufferedEvents, event)\n\t\t} else {\n\t\t\tnewCommittedEvents = append(newCommittedEvents, event)\n\t\t}\n\t}\n\n\t// no decision in-flight, flush all buffered events to committed bucket\n\tif !e.HasInFlightDecision() {\n\t\tvar allBufferedEvents []*types.HistoryEvent\n\n\t\t// flush persisted buffered events\n\t\tif len(e.bufferedEvents) > 0 {\n\t\t\tallBufferedEvents = append(allBufferedEvents, e.bufferedEvents...)\n\t\t\te.bufferedEvents = nil\n\t\t}\n\t\tif e.hasBufferedEventsInDB {\n\t\t\te.clearBufferedEvents = true\n\t\t}\n\t\t// flush pending buffered events\n\t\tallBufferedEvents = append(allBufferedEvents, e.updateBufferedEvents...)\n\t\te.updateBufferedEvents = nil\n\n\t\t// Resolve issues with persistence duplicating or reordering buffered events\n\t\treorderedEvents := e.reorderAndFilterDuplicateEvents(allBufferedEvents, \"flush\")\n\n\t\t// Put back all the reordered buffer events at the end\n\t\tif len(reorderedEvents) > 0 {\n\t\t\tnewCommittedEvents = append(newCommittedEvents, reorderedEvents...)\n\t\t}\n\n\t\t// flush new buffered events\n\t\tnewCommittedEvents = append(newCommittedEvents, newBufferedEvents...)\n\t\tnewBufferedEvents = nil\n\t}\n\n\tnewCommittedEvents = e.trimEventsAfterWorkflowClose(newCommittedEvents)\n\te.hBuilder.history = newCommittedEvents\n\n\t// make sure all new committed events have correct EventID\n\te.assignEventIDToBufferedEvents()\n\tif err := e.assignTaskIDToEvents(); err != nil {\n\t\treturn err\n\t}\n\n\t// if decision is not closed yet, and there are new buffered events, then put those to the pending buffer\n\tif e.HasInFlightDecision() && len(newBufferedEvents) > 0 {\n\t\te.updateBufferedEvents = newBufferedEvents\n\t}\n\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) UpdateCurrentVersion(\n\tversion int64,\n\tforceUpdate bool,\n) error {\n\tbefore := e.currentVersion\n\tdefer func() {\n\t\te.logger.Debugf(\"UpdateCurrentVersion for domain %s, wfID %v, version before: %v, version after: %v, forceUpdate: %v\",\n\t\t\te.executionInfo.DomainID, e.executionInfo.WorkflowID, before, e.currentVersion, forceUpdate)\n\t}()\n\n\tif state, _ := e.GetWorkflowStateCloseStatus(); state == persistence.WorkflowStateCompleted {\n\t\t// always set current version to last write version when workflow is completed\n\t\tlastWriteVersion, err := e.GetLastWriteVersion()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\te.currentVersion = lastWriteVersion\n\t\treturn nil\n\t}\n\n\tif e.versionHistories != nil {\n\t\tversionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif !versionHistory.IsEmpty() {\n\t\t\t// this make sure current version >= last write version\n\t\t\tversionHistoryItem, err := versionHistory.GetLastItem()\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\te.currentVersion = versionHistoryItem.Version\n\t\t}\n\n\t\tif version > e.currentVersion || forceUpdate {\n\t\t\te.currentVersion = version\n\t\t}\n\n\t\treturn nil\n\t}\n\te.currentVersion = constants.EmptyVersion\n\treturn nil\n}\n\n// GetCurrentVersion indicates which cluster this workflow is considered active.\nfunc (e *mutableStateBuilder) GetCurrentVersion() int64 {\n\t// Legacy TODO: remove this after all 2DC workflows complete\n\tif e.replicationState != nil {\n\t\te.logger.Debugf(\"GetCurrentVersion replicationState.CurrentVersion=%v\", e.replicationState.CurrentVersion)\n\t\treturn e.replicationState.CurrentVersion\n\t}\n\n\tif e.versionHistories != nil {\n\t\te.logger.Debugf(\"GetCurrentVersion versionHistories.CurrentVersion=%v\", e.currentVersion)\n\t\treturn e.currentVersion\n\t}\n\n\te.logger.Debugf(\"GetCurrentVersion returning empty version=%v\", constants.EmptyVersion)\n\treturn constants.EmptyVersion\n}\n\n// TODO: Check all usages of this method and address active-active case if needed.\nfunc (e *mutableStateBuilder) GetStartVersion() (int64, error) {\n\n\tif e.versionHistories != nil {\n\t\tversionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn 0, err\n\t\t}\n\t\tfirstItem, err := versionHistory.GetFirstItem()\n\t\tif err != nil {\n\t\t\treturn 0, err\n\t\t}\n\t\treturn firstItem.Version, nil\n\t}\n\n\treturn constants.EmptyVersion, nil\n}\n\nfunc (e *mutableStateBuilder) GetLastWriteVersion() (int64, error) {\n\n\t// TODO: remove this after all 2DC workflows complete\n\tif e.replicationState != nil {\n\t\treturn e.replicationState.LastWriteVersion, nil\n\t}\n\n\tif e.versionHistories != nil {\n\t\tversionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn 0, err\n\t\t}\n\t\tlastItem, err := versionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn 0, err\n\t\t}\n\t\treturn lastItem.Version, nil\n\t}\n\n\treturn constants.EmptyVersion, nil\n}\n\nfunc (e *mutableStateBuilder) checkAndClearTimerFiredEvent(\n\ttimerID string,\n) *types.HistoryEvent {\n\n\tvar timerEvent *types.HistoryEvent\n\n\te.bufferedEvents, timerEvent = checkAndClearTimerFiredEvent(e.bufferedEvents, timerID)\n\tif timerEvent != nil {\n\t\treturn timerEvent\n\t}\n\te.updateBufferedEvents, timerEvent = checkAndClearTimerFiredEvent(e.updateBufferedEvents, timerID)\n\tif timerEvent != nil {\n\t\treturn timerEvent\n\t}\n\te.hBuilder.history, timerEvent = checkAndClearTimerFiredEvent(e.hBuilder.history, timerID)\n\treturn timerEvent\n}\n\nfunc (e *mutableStateBuilder) trimEventsAfterWorkflowClose(\n\tinput []*types.HistoryEvent,\n) []*types.HistoryEvent {\n\n\tif len(input) == 0 {\n\t\treturn input\n\t}\n\n\tnextIndex := 0\n\nloop:\n\tfor _, event := range input {\n\t\tnextIndex++\n\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionCompleted,\n\t\t\ttypes.EventTypeWorkflowExecutionFailed,\n\t\t\ttypes.EventTypeWorkflowExecutionTimedOut,\n\t\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew,\n\t\t\ttypes.EventTypeWorkflowExecutionCanceled:\n\n\t\t\tbreak loop\n\t\t}\n\t}\n\n\treturn input[0:nextIndex]\n}\n\nfunc (e *mutableStateBuilder) assignEventIDToBufferedEvents() {\n\tnewCommittedEvents := e.hBuilder.history\n\n\tscheduledIDToStartedID := make(map[int64]int64)\n\tfor _, event := range newCommittedEvents {\n\t\tif event.ID != constants.BufferedEventID {\n\t\t\tcontinue\n\t\t}\n\n\t\teventID := e.executionInfo.NextEventID\n\t\tevent.ID = eventID\n\t\te.executionInfo.IncreaseNextEventID()\n\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeActivityTaskStarted:\n\t\t\tattributes := event.ActivityTaskStartedEventAttributes\n\t\t\tscheduledID := attributes.GetScheduledEventID()\n\t\t\tscheduledIDToStartedID[scheduledID] = eventID\n\t\t\tif ai, ok := e.GetActivityInfo(scheduledID); ok {\n\t\t\t\tai.StartedID = eventID\n\t\t\t\te.updateActivityInfos[ai.ScheduleID] = ai\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\t\tattributes := event.ChildWorkflowExecutionStartedEventAttributes\n\t\t\tinitiatedID := attributes.GetInitiatedEventID()\n\t\t\tscheduledIDToStartedID[initiatedID] = eventID\n\t\t\tif ci, ok := e.GetChildExecutionInfo(initiatedID); ok {\n\t\t\t\tci.StartedID = eventID\n\t\t\t\te.updateChildExecutionInfos[ci.InitiatedID] = ci\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskCompleted:\n\t\t\tattributes := event.ActivityTaskCompletedEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetScheduledEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskFailed:\n\t\t\tattributes := event.ActivityTaskFailedEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetScheduledEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskTimedOut:\n\t\t\tattributes := event.ActivityTaskTimedOutEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetScheduledEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskCanceled:\n\t\t\tattributes := event.ActivityTaskCanceledEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetScheduledEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\t\tattributes := event.ChildWorkflowExecutionCompletedEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetInitiatedEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\t\tattributes := event.ChildWorkflowExecutionFailedEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetInitiatedEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\t\tattributes := event.ChildWorkflowExecutionTimedOutEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetInitiatedEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\t\tattributes := event.ChildWorkflowExecutionCanceledEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetInitiatedEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\t\t\tattributes := event.ChildWorkflowExecutionTerminatedEventAttributes\n\t\t\tif startedID, ok := scheduledIDToStartedID[attributes.GetInitiatedEventID()]; ok {\n\t\t\t\tattributes.StartedEventID = startedID\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (e *mutableStateBuilder) assignTaskIDToEvents() error {\n\n\t// assign task IDs to all history events\n\t// first transient events\n\tnumTaskIDs := len(e.hBuilder.transientHistory)\n\tif numTaskIDs > 0 {\n\t\ttaskIDs, err := e.shard.GenerateTaskIDs(numTaskIDs)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tfor index, event := range e.hBuilder.transientHistory {\n\t\t\tif event.TaskID == constants.EmptyEventTaskID {\n\t\t\t\ttaskID := taskIDs[index]\n\t\t\t\tevent.TaskID = taskID\n\t\t\t\te.executionInfo.LastEventTaskID = taskID\n\t\t\t}\n\t\t}\n\t}\n\n\t// then normal events\n\tnumTaskIDs = len(e.hBuilder.history)\n\tif numTaskIDs > 0 {\n\t\ttaskIDs, err := e.shard.GenerateTaskIDs(numTaskIDs)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tfor index, event := range e.hBuilder.history {\n\t\t\tif event.TaskID == constants.EmptyEventTaskID {\n\t\t\t\ttaskID := taskIDs[index]\n\t\t\t\tevent.TaskID = taskID\n\t\t\t\te.executionInfo.LastEventTaskID = taskID\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) IsCurrentWorkflowGuaranteed() bool {\n\t// stateInDB is used like a bloom filter:\n\t//\n\t// 1. stateInDB being created / running meaning that this workflow must be the current\n\t//  workflow (assuming there is no rebuild of mutable state).\n\t// 2. stateInDB being completed does not guarantee this workflow being the current workflow\n\t// 3. stateInDB being zombie guarantees this workflow not being the current workflow\n\t// 4. stateInDB cannot be void, void is only possible when mutable state is just initialized\n\n\tswitch e.stateInDB {\n\tcase persistence.WorkflowStateVoid:\n\t\treturn false\n\tcase persistence.WorkflowStateCreated:\n\t\treturn true\n\tcase persistence.WorkflowStateRunning:\n\t\treturn true\n\tcase persistence.WorkflowStateCompleted:\n\t\treturn false\n\tcase persistence.WorkflowStateZombie:\n\t\treturn false\n\tcase persistence.WorkflowStateCorrupted:\n\t\treturn false\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown workflow state: %v\", e.executionInfo.State))\n\t}\n}\n\nfunc (e *mutableStateBuilder) GetDomainEntry() *cache.DomainCacheEntry {\n\treturn e.domainEntry\n}\n\nfunc (e *mutableStateBuilder) IsStickyTaskListEnabled() bool {\n\tif e.executionInfo.StickyTaskList == \"\" {\n\t\treturn false\n\t}\n\tttl := e.config.StickyTTL(e.GetDomainEntry().GetInfo().Name)\n\treturn !e.timeSource.Now().After(e.executionInfo.LastUpdatedTimestamp.Add(ttl))\n}\n\nfunc (e *mutableStateBuilder) CreateNewHistoryEvent(\n\teventType types.EventType,\n) *types.HistoryEvent {\n\n\treturn e.CreateNewHistoryEventWithTimestamp(eventType, e.timeSource.Now().UnixNano())\n}\n\nfunc (e *mutableStateBuilder) CreateNewHistoryEventWithTimestamp(\n\teventType types.EventType,\n\ttimestamp int64,\n) *types.HistoryEvent {\n\teventID := e.executionInfo.NextEventID\n\tif e.shouldBufferEvent(eventType) {\n\t\teventID = constants.BufferedEventID\n\t} else {\n\t\t// only increase NextEventID if event is not buffered\n\t\te.executionInfo.IncreaseNextEventID()\n\t}\n\n\tts := common.Int64Ptr(timestamp)\n\thistoryEvent := &types.HistoryEvent{}\n\thistoryEvent.ID = eventID\n\thistoryEvent.Timestamp = ts\n\thistoryEvent.EventType = &eventType\n\thistoryEvent.Version = e.GetCurrentVersion()\n\thistoryEvent.TaskID = constants.EmptyEventTaskID\n\n\treturn historyEvent\n}\n\nfunc (e *mutableStateBuilder) shouldBufferEvent(\n\teventType types.EventType,\n) bool {\n\n\tswitch eventType {\n\tcase // do not buffer for workflow state change\n\t\ttypes.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeWorkflowExecutionCompleted,\n\t\ttypes.EventTypeWorkflowExecutionFailed,\n\t\ttypes.EventTypeWorkflowExecutionTimedOut,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew,\n\t\ttypes.EventTypeWorkflowExecutionCanceled:\n\t\treturn false\n\tcase // decision event should not be buffered\n\t\ttypes.EventTypeDecisionTaskScheduled,\n\t\ttypes.EventTypeDecisionTaskStarted,\n\t\ttypes.EventTypeDecisionTaskCompleted,\n\t\ttypes.EventTypeDecisionTaskFailed,\n\t\ttypes.EventTypeDecisionTaskTimedOut:\n\t\treturn false\n\tcase // events generated directly from decisions should not be buffered\n\t\t// workflow complete, failed, cancelled and continue-as-new events are duplication of above\n\t\t// just put is here for reference\n\t\t// types.EventTypeWorkflowExecutionCompleted,\n\t\t// types.EventTypeWorkflowExecutionFailed,\n\t\t// types.EventTypeWorkflowExecutionCanceled,\n\t\t// types.EventTypeWorkflowExecutionContinuedAsNew,\n\t\ttypes.EventTypeActivityTaskScheduled,\n\t\ttypes.EventTypeActivityTaskCancelRequested,\n\t\ttypes.EventTypeTimerStarted,\n\t\t// DecisionTypeCancelTimer is an exception. This decision will be mapped\n\t\t// to either types.EventTypeTimerCanceled, or types.EventTypeCancelTimerFailed.\n\t\t// So both should not be buffered. Ref: historyEngine, search for \"types.DecisionTypeCancelTimer\"\n\t\ttypes.EventTypeTimerCanceled,\n\t\ttypes.EventTypeCancelTimerFailed,\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionInitiated,\n\t\ttypes.EventTypeMarkerRecorded,\n\t\ttypes.EventTypeStartChildWorkflowExecutionInitiated,\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionInitiated,\n\t\ttypes.EventTypeUpsertWorkflowSearchAttributes:\n\t\t// do not buffer event if event is directly generated from a corresponding decision\n\n\t\t// sanity check there is no decision on the fly\n\t\tif e.HasInFlightDecision() {\n\t\t\tmsg := fmt.Sprintf(\"history mutable state is processing event: %v while there is decision pending. \"+\n\t\t\t\t\"domainID: %v, workflow ID: %v, run ID: %v.\", eventType, e.executionInfo.DomainID, e.executionInfo.WorkflowID, e.executionInfo.RunID)\n\t\t\tpanic(msg)\n\t\t}\n\t\treturn false\n\tdefault:\n\t\treturn true\n\t}\n}\n\nfunc (e *mutableStateBuilder) GetWorkflowType() *types.WorkflowType {\n\twType := &types.WorkflowType{}\n\twType.Name = e.executionInfo.WorkflowTypeName\n\n\treturn wType\n}\n\nfunc (e *mutableStateBuilder) GetQueryRegistry() query.Registry {\n\treturn e.queryRegistry\n}\n\nfunc (e *mutableStateBuilder) SetQueryRegistry(queryRegistry query.Registry) {\n\te.queryRegistry = queryRegistry\n}\n\nfunc (e *mutableStateBuilder) GetRetryBackoffDuration(\n\terrReason string,\n) time.Duration {\n\n\tinfo := e.executionInfo\n\tif !info.HasRetryPolicy {\n\t\treturn backoff.NoBackoff\n\t}\n\n\treturn getBackoffInterval(\n\t\te.timeSource.Now(),\n\t\tinfo.ExpirationTime,\n\t\tinfo.Attempt,\n\t\tinfo.MaximumAttempts,\n\t\tinfo.InitialInterval,\n\t\tinfo.MaximumInterval,\n\t\tinfo.BackoffCoefficient,\n\t\terrReason,\n\t\tinfo.NonRetriableErrors,\n\t)\n}\n\nfunc (e *mutableStateBuilder) GetCronBackoffDuration(\n\tctx context.Context,\n) (time.Duration, error) {\n\tinfo := e.executionInfo\n\tif len(info.CronSchedule) == 0 {\n\t\treturn backoff.NoBackoff, nil\n\t}\n\tsched, err := backoff.ValidateSchedule(info.CronSchedule)\n\tif err != nil {\n\t\treturn backoff.NoBackoff, err\n\t}\n\t// TODO: decide if we can add execution time in execution info.\n\texecutionTime := e.executionInfo.StartTimestamp\n\t// This only call when doing ContinueAsNew. At this point, the workflow should have a start event\n\tworkflowStartEvent, err := e.GetStartEvent(ctx)\n\tif err != nil {\n\t\te.logError(\"unable to find workflow start event\", tag.ErrorTypeInvalidHistoryAction)\n\t\treturn backoff.NoBackoff, err\n\t}\n\tfirstDecisionTaskBackoff :=\n\t\ttime.Duration(workflowStartEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstDecisionTaskBackoffSeconds()) * time.Second\n\texecutionTime = executionTime.Add(firstDecisionTaskBackoff)\n\tjitterStartSeconds := workflowStartEvent.GetWorkflowExecutionStartedEventAttributes().GetJitterStartSeconds()\n\n\treturn backoff.GetBackoffForNextSchedule(sched, executionTime, e.timeSource.Now(), jitterStartSeconds, info.CronOverlapPolicy)\n}\n\n// GetStartEvent retrieves the workflow start event from mutable state\nfunc (e *mutableStateBuilder) GetStartEvent(\n\tctx context.Context,\n) (*types.HistoryEvent, error) {\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tstartEvent, err := e.eventsCache.GetEvent(\n\t\tctx,\n\t\te.shard.GetShardID(),\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tconstants.FirstEventID,\n\t\tconstants.FirstEventID,\n\t\tcurrentBranchToken,\n\t)\n\tif err != nil {\n\t\t// do not return the original error\n\t\t// since original error can be of type entity not exists\n\t\t// which can cause task processing side to fail silently\n\t\t// However, if the error is a persistence transient error,\n\t\t// we return the original error, because we fail to get\n\t\t// the event because of failure from database\n\t\tif persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, ErrMissingWorkflowStartEvent\n\t}\n\treturn startEvent, nil\n}\n\n// DeletePendingRequestCancel deletes details about a RequestCancelInfo.\nfunc (e *mutableStateBuilder) DeletePendingRequestCancel(\n\tinitiatedEventID int64,\n) error {\n\n\tif _, ok := e.pendingRequestCancelInfoIDs[initiatedEventID]; ok {\n\t\tdelete(e.pendingRequestCancelInfoIDs, initiatedEventID)\n\t} else {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find request cancel external workflow event ID: %v in mutable state\", initiatedEventID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\t// log data inconsistency instead of returning an error\n\t\te.logDataInconsistency()\n\t}\n\n\tdelete(e.updateRequestCancelInfos, initiatedEventID)\n\te.deleteRequestCancelInfos[initiatedEventID] = struct{}{}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) writeEventToCache(\n\tevent *types.HistoryEvent,\n) {\n\n\t// For start event: store it within events cache so the recordWorkflowStarted transfer task doesn't need to\n\t// load it from database\n\t// For completion event: store it within events cache so we can communicate the result to parent execution\n\t// during the processing of DeleteTransferTask without loading this event from database\n\te.eventsCache.PutEvent(\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tevent.ID,\n\t\tevent,\n\t)\n}\n\nfunc (e *mutableStateBuilder) HasParentExecution() bool {\n\treturn e.executionInfo.ParentDomainID != \"\" && e.executionInfo.ParentWorkflowID != \"\"\n}\n\n// GetUserTimerInfoByEventID gives details about a user timer.\nfunc (e *mutableStateBuilder) GetUserTimerInfoByEventID(\n\tstartEventID int64,\n) (*persistence.TimerInfo, bool) {\n\n\ttimerID, ok := e.pendingTimerEventIDToID[startEventID]\n\tif !ok {\n\t\treturn nil, false\n\t}\n\treturn e.GetUserTimerInfo(timerID)\n}\n\nfunc (e *mutableStateBuilder) GetPendingRequestCancelExternalInfos() map[int64]*persistence.RequestCancelInfo {\n\treturn e.pendingRequestCancelInfoIDs\n}\n\nfunc (e *mutableStateBuilder) HasBufferedEvents() bool {\n\tif len(e.bufferedEvents) > 0 || len(e.updateBufferedEvents) > 0 {\n\t\treturn true\n\t}\n\n\tfor _, event := range e.hBuilder.history {\n\t\tif event.ID == constants.BufferedEventID {\n\t\t\treturn true\n\t\t}\n\t}\n\n\treturn false\n}\n\nfunc (e *mutableStateBuilder) ClearStickyness() {\n\te.executionInfo.StickyTaskList = \"\"\n\te.executionInfo.StickyScheduleToStartTimeout = 0\n\te.executionInfo.ClientLibraryVersion = \"\"\n\te.executionInfo.ClientFeatureVersion = \"\"\n\te.executionInfo.ClientImpl = \"\"\n}\n\n// GetLastFirstEventID returns last first event ID\n// first event ID is the ID of a batch of events in a single history events record\nfunc (e *mutableStateBuilder) GetLastFirstEventID() int64 {\n\treturn e.executionInfo.LastFirstEventID\n}\n\n// GetNextEventID returns next event ID\nfunc (e *mutableStateBuilder) GetNextEventID() int64 {\n\treturn e.executionInfo.NextEventID\n}\n\n// GetPreviousStartedEventID returns last started decision task event ID\nfunc (e *mutableStateBuilder) GetPreviousStartedEventID() int64 {\n\treturn e.executionInfo.LastProcessedEvent\n}\n\nfunc (e *mutableStateBuilder) IsWorkflowExecutionRunning() bool {\n\treturn e.executionInfo.IsRunning()\n}\n\nfunc (e *mutableStateBuilder) AddUpsertWorkflowSearchAttributesEvent(\n\tdecisionCompletedEventID int64,\n\trequest *types.UpsertWorkflowSearchAttributesDecisionAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionUpsertWorkflowSearchAttributes\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddUpsertWorkflowSearchAttributesEvent(decisionCompletedEventID, request)\n\tif err := e.ReplicateUpsertWorkflowSearchAttributesEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateUpsertWorkflowSearchAttributesEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tupsertSearchAttr := event.UpsertWorkflowSearchAttributesEventAttributes.GetSearchAttributes().GetIndexedFields()\n\tcurrentSearchAttr := e.GetExecutionInfo().SearchAttributes\n\n\te.executionInfo.SearchAttributes = mergeMapOfByteArray(currentSearchAttr, upsertSearchAttr)\n\n\treturn e.taskGenerator.GenerateWorkflowSearchAttrTasks()\n}\n\nfunc (e *mutableStateBuilder) AddRecordMarkerEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.RecordMarkerDecisionAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowRecordMarker\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn e.hBuilder.AddMarkerRecordedEvent(decisionCompletedEventID, attributes), nil\n}\n\nfunc (e *mutableStateBuilder) AddWorkflowExecutionTerminatedEvent(\n\tfirstEventID int64,\n\treason string,\n\tdetails []byte,\n\tidentity string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowTerminated\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionTerminatedEvent(reason, details, identity)\n\tif err := e.ReplicateWorkflowExecutionTerminatedEvent(firstEventID, event); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName := e.GetDomainEntry().GetInfo().Name\n\n\te.logger.Info(\n\t\t\"Workflow execution terminated.\",\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.WorkflowID(e.GetExecutionInfo().WorkflowID),\n\t\ttag.WorkflowRunID(e.GetExecutionInfo().RunID),\n\t\ttag.WorkflowTerminationReason(reason),\n\t)\n\n\tscopeWithDomainTag := e.metricsClient.Scope(metrics.HistoryTerminateWorkflowExecutionScope).\n\t\tTagged(metrics.DomainTag(domainName)).\n\t\tTagged(metrics.WorkflowTerminationReasonTag(reason))\n\tscopeWithDomainTag.IncCounter(metrics.WorkflowTerminateCounterPerDomain)\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionTerminatedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n) error {\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusTerminated,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(event)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(event, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n\nfunc (e *mutableStateBuilder) AddContinueAsNewEvent(\n\tctx context.Context,\n\tfirstEventID int64,\n\tdecisionCompletedEventID int64,\n\tparentDomainName string,\n\tattributes *types.ContinueAsNewWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, MutableState, error) {\n\n\topTag := tag.WorkflowActionWorkflowContinueAsNew\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tvar err error\n\tnewRunID := uuid.New()\n\tnewExecution := types.WorkflowExecution{\n\t\tWorkflowID: e.executionInfo.WorkflowID,\n\t\tRunID:      newRunID,\n\t}\n\n\t// Extract ParentExecutionInfo from current run so it can be passed down to the next\n\tvar parentInfo *types.ParentExecutionInfo\n\tif e.HasParentExecution() {\n\t\tparentInfo = &types.ParentExecutionInfo{\n\t\t\tDomainUUID: e.executionInfo.ParentDomainID,\n\t\t\tDomain:     parentDomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: e.executionInfo.ParentWorkflowID,\n\t\t\t\tRunID:      e.executionInfo.ParentRunID,\n\t\t\t},\n\t\t\tInitiatedID: e.executionInfo.InitiatedID,\n\t\t}\n\t}\n\n\tcontinueAsNewEvent := e.hBuilder.AddContinuedAsNewEvent(decisionCompletedEventID, newRunID, attributes)\n\tcurrentStartEvent, err := e.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tfirstRunID := e.executionInfo.FirstExecutionRunID\n\t// This is needed for backwards compatibility.  Workflow execution create with Cadence release v0.25.0 or earlier\n\t// does not have FirstExecutionRunID stored as part of mutable state.  If this is not set then load it from\n\t// workflow execution started event.\n\tif len(firstRunID) == 0 {\n\t\tfirstRunID = currentStartEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstExecutionRunID()\n\t}\n\tfirstScheduleTime := currentStartEvent.GetWorkflowExecutionStartedEventAttributes().GetFirstScheduledTime()\n\tdomainID := e.domainEntry.GetInfo().ID\n\tactiveClusterInfo, err := e.shard.GetActiveClusterManager().GetActiveClusterInfoByClusterAttribute(ctx, domainID, attributes.ActiveClusterSelectionPolicy.GetClusterAttribute())\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tif e.logger.DebugOn() {\n\t\te.logger.Debug(\"mutableStateBuilder.AddContinueAsNewEvent created newStateBuilder\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(e.executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(e.executionInfo.RunID),\n\t\t\ttag.WorkflowRunID(newRunID),\n\t\t\ttag.CurrentVersion(e.currentVersion),\n\t\t\ttag.Dynamic(\"activecluster-sel-policy\", attributes.ActiveClusterSelectionPolicy),\n\t\t\ttag.Dynamic(\"activecluster-info\", activeClusterInfo),\n\t\t)\n\t}\n\t// TODO: improve type casting\n\tnewStateBuilder := NewMutableStateBuilderWithVersionHistories(\n\t\te.shard,\n\t\te.logger,\n\t\te.domainEntry,\n\t\tactiveClusterInfo.FailoverVersion,\n\t).(*mutableStateBuilder)\n\n\tif _, err = newStateBuilder.addWorkflowExecutionStartedEventForContinueAsNew(\n\t\tparentInfo,\n\t\tnewExecution,\n\t\te,\n\t\tattributes,\n\t\tfirstRunID,\n\t\tfirstScheduleTime,\n\t); err != nil {\n\t\treturn nil, nil, &types.InternalServiceError{Message: \"Failed to add workflow execution started event.\"}\n\t}\n\n\tif err = e.ReplicateWorkflowExecutionContinuedAsNewEvent(\n\t\tfirstEventID,\n\t\tdomainID,\n\t\tcontinueAsNewEvent,\n\t); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\treturn continueAsNewEvent, newStateBuilder, nil\n}\n\nfunc rolloverAutoResetPointsWithExpiringTime(\n\tresetPoints *types.ResetPoints,\n\tprevRunID string,\n\tnowNano int64,\n\tdomainRetentionDays int32,\n) *types.ResetPoints {\n\n\tif resetPoints == nil || resetPoints.Points == nil {\n\t\treturn resetPoints\n\t}\n\tnewPoints := make([]*types.ResetPointInfo, 0, len(resetPoints.Points))\n\texpiringTimeNano := nowNano + int64(time.Duration(domainRetentionDays)*time.Hour*24)\n\tfor _, rp := range resetPoints.Points {\n\t\tif rp.GetRunID() == prevRunID {\n\t\t\trp.ExpiringTimeNano = common.Int64Ptr(expiringTimeNano)\n\t\t}\n\t\tnewPoints = append(newPoints, rp)\n\t}\n\treturn &types.ResetPoints{\n\t\tPoints: newPoints,\n\t}\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionContinuedAsNewEvent(\n\tfirstEventID int64,\n\tdomainID string,\n\tcontinueAsNewEvent *types.HistoryEvent,\n) error {\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusContinuedAsNew,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(continueAsNewEvent)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(continueAsNewEvent, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n\n// TODO mutable state should generate corresponding transfer / timer tasks according to\n//  updates accumulated, while currently all transfer / timer tasks are managed manually\n\n// TODO convert AddTransferTasks to prepareTransferTasks\nfunc (e *mutableStateBuilder) AddTransferTasks(\n\ttransferTasks ...persistence.Task,\n) {\n\n\te.insertTransferTasks = append(e.insertTransferTasks, transferTasks...)\n}\n\n// TODO convert AddTimerTasks to prepareTimerTasks\nfunc (e *mutableStateBuilder) AddTimerTasks(\n\ttimerTasks ...persistence.Task,\n) {\n\n\te.insertTimerTasks = append(e.insertTimerTasks, timerTasks...)\n}\n\nfunc (e *mutableStateBuilder) GetTransferTasks() []persistence.Task {\n\treturn e.insertTransferTasks\n}\n\nfunc (e *mutableStateBuilder) GetTimerTasks() []persistence.Task {\n\treturn e.insertTimerTasks\n}\n\nfunc (e *mutableStateBuilder) DeleteTransferTasks() {\n\te.insertTransferTasks = nil\n}\n\nfunc (e *mutableStateBuilder) DeleteTimerTasks() {\n\te.insertTimerTasks = nil\n}\n\nfunc (e *mutableStateBuilder) SetUpdateCondition(\n\tnextEventIDInDB int64,\n) {\n\n\te.nextEventIDInDB = nextEventIDInDB\n}\n\nfunc (e *mutableStateBuilder) GetUpdateCondition() int64 {\n\treturn e.nextEventIDInDB\n}\n\nfunc (e *mutableStateBuilder) GetWorkflowStateCloseStatus() (int, int) {\n\n\texecutionInfo := e.executionInfo\n\treturn executionInfo.State, executionInfo.CloseStatus\n}\n\nfunc (e *mutableStateBuilder) UpdateWorkflowStateCloseStatus(\n\tstate int,\n\tcloseStatus int,\n) error {\n\n\treturn e.executionInfo.UpdateWorkflowStateCloseStatus(state, closeStatus)\n}\n\nfunc (e *mutableStateBuilder) StartTransaction(\n\tctx context.Context,\n\tdomainEntry *cache.DomainCacheEntry,\n\tincomingTaskVersion int64,\n) (bool, error) {\n\tactiveClusterInfo, err := e.shard.GetActiveClusterManager().GetActiveClusterInfoByWorkflow(ctx, e.executionInfo.DomainID, e.executionInfo.WorkflowID, e.executionInfo.RunID)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tif e.logger.DebugOn() {\n\t\te.logger.Debugf(\"StartTransaction calling UpdateCurrentVersion for domain %s, wfID %v, incomingTaskVersion %v, version %v, stacktrace %v\",\n\t\t\tdomainEntry.GetInfo().Name, e.executionInfo.WorkflowID, incomingTaskVersion, activeClusterInfo.FailoverVersion, string(debug.Stack()))\n\t}\n\tif err := e.UpdateCurrentVersion(activeClusterInfo.FailoverVersion, false); err != nil {\n\t\treturn false, err\n\t}\n\n\treturn e.startTransactionHandleDecisionFailover(incomingTaskVersion)\n}\n\nfunc (e *mutableStateBuilder) CloseTransactionAsMutation(\n\tnow time.Time,\n\ttransactionPolicy TransactionPolicy,\n) (*persistence.WorkflowMutation, []*persistence.WorkflowEvents, error) {\n\n\tif err := e.prepareCloseTransaction(\n\t\ttransactionPolicy,\n\t); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tworkflowEventsSeq, err := e.prepareEventsAndReplicationTasks(transactionPolicy)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tif len(workflowEventsSeq) > 0 {\n\t\tlastEvents := workflowEventsSeq[len(workflowEventsSeq)-1].Events\n\t\tfirstEvent := lastEvents[0]\n\t\tlastEvent := lastEvents[len(lastEvents)-1]\n\t\te.updateWithLastFirstEvent(firstEvent)\n\t\tif err := e.updateWithLastWriteEvent(\n\t\t\tlastEvent,\n\t\t\ttransactionPolicy,\n\t\t); err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t}\n\n\t// update last update time\n\te.executionInfo.LastUpdatedTimestamp = now\n\n\t// we generate checksum here based on the assumption that the returned\n\t// snapshot object is considered immutable. As of this writing, the only\n\t// code that modifies the returned object lives inside workflowExecutionContext.resetWorkflowExecution\n\t// currently, the updates done inside workflowExecutionContext.resetWorkflowExecution doesn't\n\t// impact the checksum calculation\n\tchecksum := e.generateChecksum()\n\n\tworkflowMutation := &persistence.WorkflowMutation{\n\t\tExecutionInfo:    e.executionInfo,\n\t\tVersionHistories: e.versionHistories,\n\n\t\tUpsertActivityInfos:       maps.Values(e.updateActivityInfos),\n\t\tDeleteActivityInfos:       maps.Keys(e.deleteActivityInfos),\n\t\tUpsertTimerInfos:          maps.Values(e.updateTimerInfos),\n\t\tDeleteTimerInfos:          maps.Keys(e.deleteTimerInfos),\n\t\tUpsertChildExecutionInfos: maps.Values(e.updateChildExecutionInfos),\n\t\tDeleteChildExecutionInfos: maps.Keys(e.deleteChildExecutionInfos),\n\t\tUpsertRequestCancelInfos:  maps.Values(e.updateRequestCancelInfos),\n\t\tDeleteRequestCancelInfos:  maps.Keys(e.deleteRequestCancelInfos),\n\t\tUpsertSignalInfos:         maps.Values(e.updateSignalInfos),\n\t\tDeleteSignalInfos:         maps.Keys(e.deleteSignalInfos),\n\t\tUpsertSignalRequestedIDs:  maps.Keys(e.updateSignalRequestedIDs),\n\t\tDeleteSignalRequestedIDs:  maps.Keys(e.deleteSignalRequestedIDs),\n\t\tNewBufferedEvents:         e.updateBufferedEvents,\n\t\tClearBufferedEvents:       e.clearBufferedEvents,\n\n\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\tpersistence.HistoryTaskCategoryTransfer:    e.insertTransferTasks,\n\t\t\tpersistence.HistoryTaskCategoryReplication: e.insertReplicationTasks,\n\t\t\tpersistence.HistoryTaskCategoryTimer:       e.insertTimerTasks,\n\t\t},\n\n\t\tWorkflowRequests: convertWorkflowRequests(e.workflowRequests),\n\n\t\tCondition: e.nextEventIDInDB,\n\t\tChecksum:  checksum,\n\t}\n\n\te.checksum = checksum\n\tif err := e.cleanupTransaction(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn workflowMutation, workflowEventsSeq, nil\n}\n\nfunc (e *mutableStateBuilder) CloseTransactionAsSnapshot(\n\tnow time.Time,\n\ttransactionPolicy TransactionPolicy,\n) (*persistence.WorkflowSnapshot, []*persistence.WorkflowEvents, error) {\n\n\tif err := e.prepareCloseTransaction(\n\t\ttransactionPolicy,\n\t); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tworkflowEventsSeq, err := e.prepareEventsAndReplicationTasks(transactionPolicy)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tif len(workflowEventsSeq) > 1 {\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"cannot generate workflow snapshot with transient events\",\n\t\t}\n\t}\n\tif len(e.bufferedEvents) > 0 {\n\t\t// TODO do we need the functionality to generate snapshot with buffered events?\n\t\treturn nil, nil, &types.InternalServiceError{\n\t\t\tMessage: \"cannot generate workflow snapshot with buffered events\",\n\t\t}\n\t}\n\n\tif len(workflowEventsSeq) > 0 {\n\t\tlastEvents := workflowEventsSeq[len(workflowEventsSeq)-1].Events\n\t\tfirstEvent := lastEvents[0]\n\t\tlastEvent := lastEvents[len(lastEvents)-1]\n\t\te.updateWithLastFirstEvent(firstEvent)\n\t\tif err := e.updateWithLastWriteEvent(\n\t\t\tlastEvent,\n\t\t\ttransactionPolicy,\n\t\t); err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t}\n\n\t// update last update time\n\te.executionInfo.LastUpdatedTimestamp = now\n\n\t// we generate checksum here based on the assumption that the returned\n\t// snapshot object is considered immutable. As of this writing, the only\n\t// code that modifies the returned object lives inside workflowExecutionContext.resetWorkflowExecution\n\t// currently, the updates done inside workflowExecutionContext.resetWorkflowExecution doesn't\n\t// impact the checksum calculation\n\tchecksum := e.generateChecksum()\n\n\tworkflowSnapshot := &persistence.WorkflowSnapshot{\n\t\tExecutionInfo:    e.executionInfo,\n\t\tVersionHistories: e.versionHistories,\n\n\t\tActivityInfos:       maps.Values(e.pendingActivityInfoIDs),\n\t\tTimerInfos:          maps.Values(e.pendingTimerInfoIDs),\n\t\tChildExecutionInfos: maps.Values(e.pendingChildExecutionInfoIDs),\n\t\tRequestCancelInfos:  maps.Values(e.pendingRequestCancelInfoIDs),\n\t\tSignalInfos:         maps.Values(e.pendingSignalInfoIDs),\n\t\tSignalRequestedIDs:  maps.Keys(e.pendingSignalRequestedIDs),\n\n\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\tpersistence.HistoryTaskCategoryTransfer:    e.insertTransferTasks,\n\t\t\tpersistence.HistoryTaskCategoryReplication: e.insertReplicationTasks,\n\t\t\tpersistence.HistoryTaskCategoryTimer:       e.insertTimerTasks,\n\t\t},\n\n\t\tWorkflowRequests: convertWorkflowRequests(e.workflowRequests),\n\n\t\tCondition: e.nextEventIDInDB,\n\t\tChecksum:  checksum,\n\t}\n\n\te.checksum = checksum\n\tif err := e.cleanupTransaction(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn workflowSnapshot, workflowEventsSeq, nil\n}\n\nfunc (e *mutableStateBuilder) IsResourceDuplicated(\n\tresourceDedupKey definition.DeduplicationID,\n) bool {\n\tid := definition.GenerateDeduplicationKey(resourceDedupKey)\n\t_, duplicated := e.appliedEvents[id]\n\treturn duplicated\n}\n\nfunc (e *mutableStateBuilder) UpdateDuplicatedResource(\n\tresourceDedupKey definition.DeduplicationID,\n) {\n\tid := definition.GenerateDeduplicationKey(resourceDedupKey)\n\te.appliedEvents[id] = struct{}{}\n}\n\nfunc (e *mutableStateBuilder) GetHistorySize() int64 {\n\treturn e.executionStats.HistorySize\n}\n\nfunc (e *mutableStateBuilder) SetHistorySize(size int64) {\n\te.executionStats.HistorySize = size\n}\n\nfunc (e *mutableStateBuilder) ByteSize() uint64 {\n\t// TODO: To be implemented\n\treturn 0\n}\n\nfunc (e *mutableStateBuilder) prepareCloseTransaction(\n\ttransactionPolicy TransactionPolicy,\n) error {\n\n\tif err := e.closeTransactionWithPolicyCheck(\n\t\ttransactionPolicy,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := e.closeTransactionHandleBufferedEventsLimit(\n\t\ttransactionPolicy,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := e.closeTransactionHandleWorkflowReset(\n\t\ttransactionPolicy,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\t// flushing buffered events should happen at very last\n\tif transactionPolicy == TransactionPolicyActive {\n\t\tif err := e.FlushBufferedEvents(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t// NOTE: this function must be the last call\n\t//  since we only generate at most one activity & user timer,\n\t//  regardless of how many activity & user timer created\n\t//  so the calculation must be at the very end\n\treturn e.closeTransactionHandleActivityUserTimerTasks()\n}\n\nfunc (e *mutableStateBuilder) cleanupTransaction() error {\n\n\t// Clear all updates to prepare for the next session\n\te.hBuilder = NewHistoryBuilder(e)\n\n\te.updateActivityInfos = make(map[int64]*persistence.ActivityInfo)\n\te.deleteActivityInfos = make(map[int64]struct{})\n\te.syncActivityTasks = make(map[int64]struct{})\n\n\te.updateTimerInfos = make(map[string]*persistence.TimerInfo)\n\te.deleteTimerInfos = make(map[string]struct{})\n\n\te.updateChildExecutionInfos = make(map[int64]*persistence.ChildExecutionInfo)\n\te.deleteChildExecutionInfos = make(map[int64]struct{})\n\n\te.updateRequestCancelInfos = make(map[int64]*persistence.RequestCancelInfo)\n\te.deleteRequestCancelInfos = make(map[int64]struct{})\n\n\te.updateSignalInfos = make(map[int64]*persistence.SignalInfo)\n\te.deleteSignalInfos = make(map[int64]struct{})\n\n\te.updateSignalRequestedIDs = make(map[string]struct{})\n\te.deleteSignalRequestedIDs = make(map[string]struct{})\n\n\te.clearBufferedEvents = false\n\tif e.updateBufferedEvents != nil {\n\t\te.bufferedEvents = append(e.bufferedEvents, e.updateBufferedEvents...)\n\t\te.updateBufferedEvents = nil\n\t}\n\n\te.hasBufferedEventsInDB = len(e.bufferedEvents) > 0\n\te.stateInDB = e.executionInfo.State\n\te.nextEventIDInDB = e.GetNextEventID()\n\n\te.insertTransferTasks = nil\n\te.insertReplicationTasks = nil\n\te.insertTimerTasks = nil\n\n\te.workflowRequests = make(map[persistence.WorkflowRequest]struct{})\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) prepareEventsAndReplicationTasks(\n\ttransactionPolicy TransactionPolicy,\n) ([]*persistence.WorkflowEvents, error) {\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar workflowEventsSeq []*persistence.WorkflowEvents\n\tif len(e.hBuilder.transientHistory) != 0 {\n\t\tworkflowEventsSeq = append(workflowEventsSeq, &persistence.WorkflowEvents{\n\t\t\tDomainID:    e.executionInfo.DomainID,\n\t\t\tWorkflowID:  e.executionInfo.WorkflowID,\n\t\t\tRunID:       e.executionInfo.RunID,\n\t\t\tBranchToken: currentBranchToken,\n\t\t\tEvents:      e.hBuilder.transientHistory,\n\t\t})\n\t}\n\tif len(e.hBuilder.history) != 0 {\n\t\tworkflowEventsSeq = append(workflowEventsSeq, &persistence.WorkflowEvents{\n\t\t\tDomainID:    e.executionInfo.DomainID,\n\t\t\tWorkflowID:  e.executionInfo.WorkflowID,\n\t\t\tRunID:       e.executionInfo.RunID,\n\t\t\tBranchToken: currentBranchToken,\n\t\t\tEvents:      e.hBuilder.history,\n\t\t})\n\t}\n\n\tif err := e.validateNoEventsAfterWorkflowFinish(\n\t\ttransactionPolicy,\n\t\te.hBuilder.history,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tfor _, workflowEvents := range workflowEventsSeq {\n\t\treplicationTasks, err := e.eventsToReplicationTask(transactionPolicy, workflowEvents.Events)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\te.insertReplicationTasks = append(\n\t\t\te.insertReplicationTasks,\n\t\t\treplicationTasks...,\n\t\t)\n\t}\n\n\te.insertReplicationTasks = append(\n\t\te.insertReplicationTasks,\n\t\te.syncActivityToReplicationTask(transactionPolicy)...,\n\t)\n\n\tif transactionPolicy == TransactionPolicyPassive && len(e.insertReplicationTasks) > 0 {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"should not generate replication task when close transaction as passive\",\n\t\t}\n\t}\n\n\treturn workflowEventsSeq, nil\n}\n\nfunc (e *mutableStateBuilder) eventsToReplicationTask(\n\ttransactionPolicy TransactionPolicy,\n\tevents []*types.HistoryEvent,\n) ([]persistence.Task, error) {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\t!e.canReplicateEvents() ||\n\t\tlen(events) == 0 {\n\t\treturn emptyTasks, nil\n\t}\n\n\tfirstEvent := events[0]\n\tlastEvent := events[len(events)-1]\n\tversion := firstEvent.Version\n\n\t// Check all the events in the transaction belongs to the same cluster\n\tsourceCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(version)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tcurrentCluster := e.clusterMetadata.GetCurrentClusterName()\n\n\tif currentCluster != sourceCluster {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"mutableStateBuilder encounter contradicting version & transaction policy\",\n\t\t}\n\t}\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// the visibility timestamp will be set in shard context\n\treplicationTask := &persistence.HistoryReplicationTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   e.executionInfo.DomainID,\n\t\t\tWorkflowID: e.executionInfo.WorkflowID,\n\t\t\tRunID:      e.executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: firstEvent.Version,\n\t\t},\n\t\tFirstEventID:      firstEvent.ID,\n\t\tNextEventID:       lastEvent.ID + 1,\n\t\tBranchToken:       currentBranchToken,\n\t\tNewRunBranchToken: nil,\n\t}\n\n\te.logger.Debugf(\"eventsToReplicationTask returning replicationTask. Version: %v, FirstEventID: %v, NextEventID: %v, SourceCluster: %v, CurrentCluster: %v\",\n\t\treplicationTask.Version,\n\t\treplicationTask.FirstEventID,\n\t\treplicationTask.NextEventID,\n\t\tsourceCluster,\n\t\tcurrentCluster,\n\t)\n\n\treturn []persistence.Task{replicationTask}, nil\n}\n\nfunc (e *mutableStateBuilder) syncActivityToReplicationTask(\n\ttransactionPolicy TransactionPolicy,\n) []persistence.Task {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\t!e.canReplicateEvents() {\n\t\treturn emptyTasks\n\t}\n\n\treturn convertSyncActivityInfos(\n\t\te.executionInfo,\n\t\te.pendingActivityInfoIDs,\n\t\te.syncActivityTasks,\n\t)\n}\n\nfunc (e *mutableStateBuilder) updateWithLastWriteEvent(\n\tlastEvent *types.HistoryEvent,\n\ttransactionPolicy TransactionPolicy,\n) error {\n\n\tif transactionPolicy == TransactionPolicyPassive {\n\t\t// already handled in state builder\n\t\treturn nil\n\t}\n\n\te.GetExecutionInfo().LastEventTaskID = lastEvent.TaskID\n\n\tif e.versionHistories != nil {\n\t\tcurrentVersionHistory, err := e.versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := currentVersionHistory.AddOrUpdateItem(persistence.NewVersionHistoryItem(\n\t\t\tlastEvent.ID, lastEvent.Version,\n\t\t)); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) updateWithLastFirstEvent(\n\tlastFirstEvent *types.HistoryEvent,\n) {\n\te.GetExecutionInfo().SetLastFirstEventID(lastFirstEvent.ID)\n}\n\nfunc (e *mutableStateBuilder) canReplicateEvents() bool {\n\tif e.domainEntry.GetReplicationPolicy() == cache.ReplicationPolicyOneCluster {\n\t\treturn false\n\t}\n\t// ReplicationPolicyMultiCluster\n\tdomainID := e.domainEntry.GetInfo().ID\n\tworkflowID := e.GetExecutionInfo().WorkflowID\n\treturn e.shard.GetConfig().EnableReplicationTaskGeneration(domainID, workflowID)\n}\n\n// validateNoEventsAfterWorkflowFinish perform check on history event batch\n// NOTE: do not apply this check on every batch, since transient\n// decision && workflow finish will be broken (the first batch)\nfunc (e *mutableStateBuilder) validateNoEventsAfterWorkflowFinish(\n\ttransactionPolicy TransactionPolicy,\n\tevents []*types.HistoryEvent,\n) error {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\tlen(events) == 0 {\n\t\treturn nil\n\t}\n\n\t// only do check if workflow is finished\n\tif e.GetExecutionInfo().State != persistence.WorkflowStateCompleted {\n\t\treturn nil\n\t}\n\n\t// workflow close\n\t// this will perform check on the last event of last batch\n\t// NOTE: do not apply this check on every batch, since transient\n\t// decision && workflow finish will be broken (the first batch)\n\tlastEvent := events[len(events)-1]\n\tswitch lastEvent.GetEventType() {\n\tcase types.EventTypeWorkflowExecutionCompleted,\n\t\ttypes.EventTypeWorkflowExecutionFailed,\n\t\ttypes.EventTypeWorkflowExecutionTimedOut,\n\t\ttypes.EventTypeWorkflowExecutionTerminated,\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew,\n\t\ttypes.EventTypeWorkflowExecutionCanceled:\n\t\treturn nil\n\n\tdefault:\n\t\texecutionInfo := e.GetExecutionInfo()\n\t\te.logError(\n\t\t\t\"encounter case where events appears after workflow finish.\",\n\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\t)\n\t\treturn ErrEventsAfterWorkflowFinish\n\t}\n}\n\nfunc (e *mutableStateBuilder) startTransactionHandleDecisionFailover(\n\tincomingTaskVersion int64,\n) (bool, error) {\n\n\tif !e.IsWorkflowExecutionRunning() ||\n\t\t!e.canReplicateEvents() {\n\t\treturn false, nil\n\t}\n\n\t// NOTE:\n\t// the main idea here is to guarantee that once there is a decision task started\n\t// all events ending in the buffer should have the same version\n\n\t// Handling mutable state turn from standby to active, while having a decision on the fly\n\tdecision, ok := e.GetInFlightDecision()\n\tif !ok || decision.Version >= e.GetCurrentVersion() {\n\t\t// no pending decision, no buffered events\n\t\t// or decision has higher / equal version\n\t\treturn false, nil\n\t}\n\n\tlastWriteVersion, err := e.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tif lastWriteVersion != decision.Version {\n\t\treturn false, &types.InternalServiceError{Message: fmt.Sprintf(\n\t\t\t\"mutableStateBuilder encounter mismatch version, decision: %v, last write version %v\",\n\t\t\tdecision.Version,\n\t\t\tlastWriteVersion,\n\t\t)}\n\t}\n\n\tlastWriteSourceCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(lastWriteVersion)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tcurrentVersion := e.GetCurrentVersion()\n\tcurrentVersionCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(currentVersion)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tcurrentCluster := e.clusterMetadata.GetCurrentClusterName()\n\n\t// there are 5 cases for version changes (based on version from domain cache)\n\t// NOTE: domain cache version change may occur after seeing events with higher version\n\t//  meaning that the flush buffer logic in NDC branch manager should be kept.\n\t//\n\t// 1. active -> passive => fail decision & flush buffer using last write version\n\t// 2. active -> active => fail decision & flush buffer using last write version\n\t// 3. passive -> active => fail decision using current version, no buffered events\n\t// 4. passive -> passive => no buffered events, since always passive, nothing to be done\n\t// 5. special case: current cluster is passive. Due to some reason, the history generated by the current cluster\n\t// is missing and the missing history replicate back from remote cluster via resending approach => nothing to do\n\n\te.logger.Debugf(\"startTransactionHandleDecisionFailover incomingTaskVersion %v, lastWriteVersion %v, currentVersion %v, currentCluster %v, lastWriteSourceCluster %v, currentVersionCluster %v\",\n\t\tincomingTaskVersion,\n\t\tlastWriteVersion,\n\t\tcurrentVersion,\n\t\tcurrentCluster,\n\t\tlastWriteSourceCluster,\n\t\tcurrentVersionCluster,\n\t)\n\t// handle case 5\n\tincomingTaskSourceCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(incomingTaskVersion)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tif incomingTaskVersion != constants.EmptyVersion &&\n\t\tcurrentVersionCluster != currentCluster &&\n\t\tincomingTaskSourceCluster == currentCluster {\n\t\treturn false, nil\n\t}\n\n\t// handle case 4\n\tif lastWriteSourceCluster != currentCluster && currentVersionCluster != currentCluster {\n\t\t// do a sanity check on buffered events\n\t\tif e.HasBufferedEvents() {\n\t\t\treturn false, &types.InternalServiceError{\n\t\t\t\tMessage: \"mutableStateBuilder encounter previous passive workflow with buffered events\",\n\t\t\t}\n\t\t}\n\t\treturn false, nil\n\t}\n\n\t// handle case 1 & 2\n\tvar flushBufferVersion = lastWriteVersion\n\n\t// handle case 3\n\tif lastWriteSourceCluster != currentCluster && currentVersionCluster == currentCluster {\n\t\t// do a sanity check on buffered events\n\t\tif e.HasBufferedEvents() {\n\t\t\treturn false, &types.InternalServiceError{\n\t\t\t\tMessage: \"mutableStateBuilder encounter previous passive workflow with buffered events\",\n\t\t\t}\n\t\t}\n\t\tflushBufferVersion = currentVersion\n\t}\n\n\t// this workflow was previous active (whether it has buffered events or not),\n\t// the in flight decision must be failed to guarantee all events within same\n\t// event batch shard the same version\n\te.logger.Debugf(\"startTransactionHandleDecisionFailover calling UpdateCurrentVersion for domain %s, wfID %v, flushBufferVersion %v\",\n\t\te.executionInfo.DomainID, e.executionInfo.WorkflowID, flushBufferVersion)\n\tif err := e.UpdateCurrentVersion(flushBufferVersion, true); err != nil {\n\t\treturn false, err\n\t}\n\n\t// we have a decision with buffered events on the fly with a lower version, fail it\n\tif err := FailDecision(\n\t\te,\n\t\tdecision,\n\t\ttypes.DecisionTaskFailedCauseFailoverCloseDecision,\n\t); err != nil {\n\t\treturn false, err\n\t}\n\n\terr = ScheduleDecision(e)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\treturn true, nil\n}\n\nfunc (e *mutableStateBuilder) closeTransactionWithPolicyCheck(\n\ttransactionPolicy TransactionPolicy,\n) error {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\t!e.canReplicateEvents() {\n\t\treturn nil\n\t}\n\n\tactiveCluster, err := e.clusterMetadata.ClusterNameForFailoverVersion(e.GetCurrentVersion())\n\tif err != nil {\n\t\treturn err\n\t}\n\tcurrentCluster := e.clusterMetadata.GetCurrentClusterName()\n\n\tif activeCluster != currentCluster {\n\t\te.logger.Debugf(\"closeTransactionWithPolicyCheck activeCluster != currentCluster, activeCluster=%v, currentCluster=%v, e.GetCurrentVersion()=%v\", activeCluster, currentCluster, e.GetCurrentVersion())\n\t\treturn e.domainEntry.NewDomainNotActiveError(currentCluster, activeCluster)\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) closeTransactionHandleBufferedEventsLimit(\n\ttransactionPolicy TransactionPolicy,\n) error {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\t!e.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tif len(e.bufferedEvents) < e.config.MaximumBufferedEventsBatch() {\n\t\treturn nil\n\t}\n\n\t// Handling buffered events size issue\n\tif decision, ok := e.GetInFlightDecision(); ok {\n\t\t// we have a decision on the fly with a lower version, fail it\n\t\tif err := FailDecision(\n\t\t\te,\n\t\t\tdecision,\n\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\terr := ScheduleDecision(e)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) closeTransactionHandleWorkflowReset(\n\ttransactionPolicy TransactionPolicy,\n) error {\n\n\tif transactionPolicy == TransactionPolicyPassive ||\n\t\t!e.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\t// compare with bad client binary checksum and schedule a reset task\n\n\t// only schedule reset task if current doesn't have childWFs.\n\t// TODO: This will be removed once our reset allows childWFs\n\tif len(e.GetPendingChildExecutionInfos()) != 0 {\n\t\treturn nil\n\t}\n\n\texecutionInfo := e.GetExecutionInfo()\n\tdomainEntry, err := e.shard.GetDomainCache().GetDomainByID(executionInfo.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif _, pt := FindAutoResetPoint(\n\t\te.timeSource,\n\t\t&domainEntry.GetConfig().BadBinaries,\n\t\te.GetExecutionInfo().AutoResetPoints,\n\t); pt != nil {\n\t\tif err := e.taskGenerator.GenerateWorkflowResetTasks(); err != nil {\n\t\t\treturn err\n\t\t}\n\t\te.logInfo(\"Auto-Reset task is scheduled\",\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\t\ttag.WorkflowResetBaseRunID(pt.GetRunID()),\n\t\t\ttag.WorkflowEventID(pt.GetFirstDecisionCompletedID()),\n\t\t\ttag.WorkflowBinaryChecksum(pt.GetBinaryChecksum()),\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) closeTransactionHandleActivityUserTimerTasks() error {\n\tif !e.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tif err := e.taskGenerator.GenerateActivityTimerTasks(); err != nil {\n\t\treturn err\n\t}\n\n\treturn e.taskGenerator.GenerateUserTimerTasks()\n}\n\nfunc (e *mutableStateBuilder) checkMutability(\n\tactionTag tag.Tag,\n) error {\n\n\tif !e.IsWorkflowExecutionRunning() {\n\t\te.logWarn(\n\t\t\tmutableStateInvalidHistoryActionMsg,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowState(e.executionInfo.State),\n\t\t\tactionTag,\n\t\t)\n\t\treturn ErrWorkflowFinished\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) insertWorkflowRequest(request persistence.WorkflowRequest) {\n\tif e.domainEntry != nil && e.config.EnableStrongIdempotency(e.domainEntry.GetInfo().Name) && request.RequestID != \"\" {\n\t\tif _, ok := e.workflowRequests[request]; ok {\n\t\t\te.logWarn(\"error encountering duplicate request\", tag.WorkflowRequestID(request.RequestID))\n\t\t}\n\t\te.workflowRequests[request] = struct{}{}\n\t}\n}\n\nfunc (e *mutableStateBuilder) generateChecksum() checksum.Checksum {\n\tif !e.shouldGenerateChecksum() {\n\t\treturn checksum.Checksum{}\n\t}\n\tcsum, err := generateMutableStateChecksum(e)\n\tif err != nil {\n\t\te.logWarn(\"error generating mutableState checksum\", tag.Error(err))\n\t\treturn checksum.Checksum{}\n\t}\n\treturn csum\n}\n\nfunc (e *mutableStateBuilder) shouldGenerateChecksum() bool {\n\tif e.domainEntry == nil {\n\t\treturn false\n\t}\n\treturn rand.Intn(100) < e.config.MutableStateChecksumGenProbability(e.domainEntry.GetInfo().Name)\n}\n\nfunc (e *mutableStateBuilder) shouldVerifyChecksum() bool {\n\tif e.domainEntry == nil {\n\t\treturn false\n\t}\n\treturn rand.Intn(100) < e.config.MutableStateChecksumVerifyProbability(e.domainEntry.GetInfo().Name)\n}\n\nfunc (e *mutableStateBuilder) enableChecksumFailureRetry() bool {\n\tif e.domainEntry == nil {\n\t\treturn false\n\t}\n\treturn e.config.EnableRetryForChecksumFailure(e.domainEntry.GetInfo().Name)\n}\n\nfunc (e *mutableStateBuilder) shouldInvalidateChecksum() bool {\n\tinvalidateBeforeEpochSecs := int64(e.config.MutableStateChecksumInvalidateBefore())\n\tif invalidateBeforeEpochSecs > 0 {\n\t\tinvalidateBefore := time.Unix(invalidateBeforeEpochSecs, 0)\n\t\treturn e.executionInfo.LastUpdatedTimestamp.Before(invalidateBefore)\n\t}\n\treturn false\n}\n\nfunc (e *mutableStateBuilder) createInternalServerError(\n\tactionTag tag.Tag,\n) error {\n\n\treturn &types.InternalServiceError{Message: actionTag.Field().String + \" operation failed\"}\n}\n\nfunc (e *mutableStateBuilder) createCallerError(\n\tactionTag tag.Tag,\n) error {\n\n\treturn &types.BadRequestError{\n\t\tMessage: fmt.Sprintf(mutableStateInvalidHistoryActionMsgTemplate, actionTag.Field().String),\n\t}\n}\n\nfunc (e *mutableStateBuilder) unixNanoToTime(\n\ttimestampNanos int64,\n) time.Time {\n\n\treturn time.Unix(0, timestampNanos)\n}\n\nfunc (e *mutableStateBuilder) logInfo(msg string, tags ...tag.Tag) {\n\tif e == nil {\n\t\treturn\n\t}\n\tif e.executionInfo != nil {\n\t\ttags = append(tags, tag.WorkflowID(e.executionInfo.WorkflowID))\n\t\ttags = append(tags, tag.WorkflowRunID(e.executionInfo.RunID))\n\t\ttags = append(tags, tag.WorkflowDomainID(e.executionInfo.DomainID))\n\t}\n\te.logger.Info(msg, tags...)\n}\n\nfunc (e *mutableStateBuilder) logWarn(msg string, tags ...tag.Tag) {\n\tif e == nil {\n\t\treturn\n\t}\n\tif e.executionInfo != nil {\n\t\ttags = append(tags, tag.WorkflowID(e.executionInfo.WorkflowID))\n\t\ttags = append(tags, tag.WorkflowRunID(e.executionInfo.RunID))\n\t\ttags = append(tags, tag.WorkflowDomainID(e.executionInfo.DomainID))\n\t}\n\te.logger.Warn(msg, tags...)\n}\n\nfunc (e *mutableStateBuilder) logError(msg string, tags ...tag.Tag) {\n\tif e == nil {\n\t\treturn\n\t}\n\tif e.executionInfo != nil {\n\t\ttags = append(tags, tag.WorkflowID(e.executionInfo.WorkflowID))\n\t\ttags = append(tags, tag.WorkflowRunID(e.executionInfo.RunID))\n\t\ttags = append(tags, tag.WorkflowDomainID(e.executionInfo.DomainID))\n\t}\n\te.logger.Error(msg, tags...)\n}\n\nfunc (e *mutableStateBuilder) logDataInconsistency() {\n\tdomainID := e.executionInfo.DomainID\n\tworkflowID := e.executionInfo.WorkflowID\n\trunID := e.executionInfo.RunID\n\n\te.metricsClient.Scope(metrics.WorkflowContextScope).IncCounter(metrics.DataInconsistentCounter)\n\te.logger.Error(\"encounter mutable state data inconsistency\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(workflowID),\n\t\ttag.WorkflowRunID(runID),\n\t)\n}\nfunc (e *mutableStateBuilder) reorderAndFilterDuplicateEvents(events []*types.HistoryEvent, source string) []*types.HistoryEvent {\n\ttype eventUniquenessParams struct {\n\t\teventType        types.EventType\n\t\tscheduledEventID int64\n\t\tattempt          int32\n\t\tstartedEventID   int64\n\t}\n\n\tactivityTaskUniqueEvents := make(map[eventUniquenessParams]struct{})\n\n\tcheckEventUniqueness := func(event *types.HistoryEvent) bool {\n\t\tvar uniqueEventParams eventUniquenessParams\n\n\t\tvar scheduledEventID int64\n\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeActivityTaskStarted:\n\t\t\tscheduledEventID = event.ActivityTaskStartedEventAttributes.GetScheduledEventID()\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tattempt:          event.ActivityTaskStartedEventAttributes.Attempt,\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskCompleted:\n\t\t\tscheduledEventID = event.ActivityTaskCompletedEventAttributes.GetScheduledEventID()\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ActivityTaskCompletedEventAttributes.GetStartedEventID(),\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskFailed:\n\t\t\tscheduledEventID = event.ActivityTaskFailedEventAttributes.GetScheduledEventID()\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ActivityTaskFailedEventAttributes.GetStartedEventID(),\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskCanceled:\n\t\t\tscheduledEventID = event.ActivityTaskCanceledEventAttributes.GetScheduledEventID()\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ActivityTaskCanceledEventAttributes.StartedEventID,\n\t\t\t}\n\t\tcase types.EventTypeActivityTaskTimedOut:\n\t\t\tscheduledEventID = event.ActivityTaskTimedOutEventAttributes.GetScheduledEventID()\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ActivityTaskTimedOutEventAttributes.StartedEventID,\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\t\tscheduledEventID = event.ChildWorkflowExecutionStartedEventAttributes.InitiatedEventID\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\t\tscheduledEventID = event.ChildWorkflowExecutionCompletedEventAttributes.InitiatedEventID\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ChildWorkflowExecutionCompletedEventAttributes.StartedEventID,\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\t\tscheduledEventID = event.ChildWorkflowExecutionFailedEventAttributes.InitiatedEventID\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ChildWorkflowExecutionFailedEventAttributes.StartedEventID,\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\t\tscheduledEventID = event.ChildWorkflowExecutionTimedOutEventAttributes.InitiatedEventID\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ChildWorkflowExecutionTimedOutEventAttributes.StartedEventID,\n\t\t\t}\n\t\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\t\tscheduledEventID = event.ChildWorkflowExecutionCanceledEventAttributes.InitiatedEventID\n\t\t\tuniqueEventParams = eventUniquenessParams{\n\t\t\t\teventType:        event.GetEventType(),\n\t\t\t\tscheduledEventID: scheduledEventID,\n\t\t\t\tstartedEventID:   event.ChildWorkflowExecutionCanceledEventAttributes.StartedEventID,\n\t\t\t}\n\t\tdefault:\n\t\t\treturn true\n\t\t}\n\n\t\tif _, ok := activityTaskUniqueEvents[uniqueEventParams]; ok {\n\t\t\te.logger.Error(\"Duplicate event found\",\n\t\t\t\ttag.WorkflowDomainName(e.GetDomainEntry().GetInfo().Name),\n\t\t\t\ttag.WorkflowID(e.GetExecutionInfo().WorkflowID),\n\t\t\t\ttag.WorkflowRunID(e.GetExecutionInfo().RunID),\n\t\t\t\ttag.WorkflowScheduleID(scheduledEventID),\n\t\t\t\ttag.WorkflowEventType(event.GetEventType().String()),\n\t\t\t\ttag.Dynamic(\"duplication-source\", source),\n\t\t\t)\n\n\t\t\te.metricsClient.IncCounter(metrics.HistoryFlushBufferedEventsScope, metrics.DuplicateActivityTaskEventCounter)\n\t\t\treturn false\n\t\t}\n\t\tactivityTaskUniqueEvents[uniqueEventParams] = struct{}{}\n\t\treturn true\n\t}\n\n\tvar headEvents []*types.HistoryEvent\n\tvar tailEvents []*types.HistoryEvent\n\n\t// Sometimes we see buffered events are out of order when read back from database.  This is mostly not an issue\n\t// except in the Activity case where ActivityStarted and ActivityCompleted gets out of order.  The following code\n\t// is added to reorder buffered events to guarantee all activity completion events will always be processed at the end.\n\tfor _, event := range events {\n\t\t// We sometimes see duplicate events\n\t\tif unique := checkEventUniqueness(event); !unique {\n\t\t\tcontinue\n\t\t}\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeActivityTaskCompleted,\n\t\t\ttypes.EventTypeActivityTaskFailed,\n\t\t\ttypes.EventTypeActivityTaskCanceled,\n\t\t\ttypes.EventTypeActivityTaskTimedOut:\n\t\t\ttailEvents = append(tailEvents, event)\n\t\tcase types.EventTypeChildWorkflowExecutionCompleted,\n\t\t\ttypes.EventTypeChildWorkflowExecutionFailed,\n\t\t\ttypes.EventTypeChildWorkflowExecutionCanceled,\n\t\t\ttypes.EventTypeChildWorkflowExecutionTimedOut,\n\t\t\ttypes.EventTypeChildWorkflowExecutionTerminated:\n\t\t\ttailEvents = append(tailEvents, event)\n\t\tdefault:\n\t\t\theadEvents = append(headEvents, event)\n\t\t}\n\t}\n\treturn append(headEvents, tailEvents...)\n}\n\nfunc mergeMapOfByteArray(\n\tcurrent map[string][]byte,\n\tupsert map[string][]byte,\n) map[string][]byte {\n\n\tif current == nil {\n\t\tcurrent = make(map[string][]byte)\n\t}\n\tfor k, v := range upsert {\n\t\tcurrent[k] = v\n\t}\n\treturn current\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_add_continue_as_new_event_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/google/go-cmp/cmp/cmpopts\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n\tshardCtx \"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestAddContinueAsNewEvent(t *testing.T) {\n\tfirstEventID := int64(15)\n\tdecisionCompletedEventID := int64(15)\n\tdomainID := \"5391dbea-5b30-4323-82ca-e1c95339bb3e\"\n\tdomainFailoverVersion := int64(1)\n\tts0 := int64(123450)\n\tts1 := int64(123451)\n\tts2 := int64(123452)\n\tts3 := int64(123453)\n\tshardID := 123\n\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{},\n\t\tdomainFailoverVersion,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\tdomainEntryActiveActive := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t-1, // failover version is not used for active-active domain\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\t// the mutable state builder confusingly both returns a new builder with this fuction\n\t// as well as mutating its internal state, making it difficult to test repeatedly, since\n\t// the supplied inputs are muted per invocation. Wrapping them in a factor here to allow\n\t// for tests to be independent\n\tcreateStartingExecutionInfo := func() *persistence.WorkflowExecutionInfo {\n\t\treturn &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:                           \"5391dbea-5b30-4323-82ca-e1c95339bb3e\",\n\t\t\tWorkflowID:                         \"helloworld_b4db8bd0-74b7-4250-ade7-ac72a1efb171\",\n\t\t\tRunID:                              \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\tFirstExecutionRunID:                \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\tInitiatedID:                        -7,\n\t\t\tTaskList:                           \"helloWorldGroup\",\n\t\t\tWorkflowTypeName:                   \"helloWorldWorkflow\",\n\t\t\tWorkflowTimeout:                    60,\n\t\t\tDecisionStartToCloseTimeout:        60,\n\t\t\tState:                              1,\n\t\t\tLastFirstEventID:                   14,\n\t\t\tLastEventTaskID:                    15728673,\n\t\t\tNextEventID:                        16,\n\t\t\tLastProcessedEvent:                 14,\n\t\t\tStartTimestamp:                     time.Unix(0, ts0),\n\t\t\tLastUpdatedTimestamp:               time.Unix(0, ts2),\n\t\t\tCreateRequestID:                    \"b086d62c-dd2b-4bbc-9143-5940516acbfe\",\n\t\t\tDecisionVersion:                    -24,\n\t\t\tDecisionScheduleID:                 -23,\n\t\t\tDecisionStartedID:                  -23,\n\t\t\tDecisionRequestID:                  \"emptyUuid\",\n\t\t\tDecisionOriginalScheduledTimestamp: 1709872131542474000,\n\t\t\tStickyTaskList:                     \"david-porter-DVFG73D710:04be47fa-2381-469f-b2ea-1253271ad116\",\n\t\t\tStickyScheduleToStartTimeout:       5,\n\t\t\tClientLibraryVersion:               \"0.18.4\",\n\t\t\tClientFeatureVersion:               \"1.7.0\",\n\t\t\tClientImpl:                         \"uber-go\",\n\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\tPoints: []*types.ResetPointInfo{{\n\t\t\t\t\tBinaryChecksum:           \"6df03bf5110d681667852a8456519536\",\n\t\t\t\t\tRunID:                    \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\t\t\tFirstDecisionCompletedID: 4,\n\t\t\t\t\tCreatedTimeNano:          common.Ptr(int64(ts1)),\n\t\t\t\t\tResettable:               true,\n\t\t\t\t}},\n\t\t\t},\n\t\t\tSearchAttributes: map[string][]uint8{\"BinaryChecksums\": {91, 34, 54, 100, 102, 48, 51, 98, 102, 53, 49, 49, 48, 100, 54, 56, 49, 54, 54, 55, 56, 53, 50, 97, 56, 52, 53, 54, 53, 49, 57, 53, 51, 54, 34, 93}}}\n\n\t}\n\n\tcreateValidStartingHistory := func(version int64) []*types.HistoryEvent {\n\t\treturn []*types.HistoryEvent{{\n\t\t\tID:        15,\n\t\t\tTimestamp: common.Ptr(int64(1709872131580456000)),\n\t\t\tEventType: common.Ptr(types.EventTypeDecisionTaskCompleted),\n\t\t\tVersion:   version,\n\t\t\tTaskID:    -1234,\n\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tScheduledEventID: 13,\n\t\t\t\tStartedEventID:   14,\n\t\t\t\tIdentity:         \"27368@david-porter-DVFG73D710@helloWorldGroup@6027e9ee-048e-4f67-8d88-27883c496901\",\n\t\t\t\tBinaryChecksum:   \"6df03bf5110d681667852a8456519536\",\n\t\t\t},\n\t\t}}\n\t}\n\n\tcreateFetchedHistory := func(version int64) *types.HistoryEvent {\n\t\treturn &types.HistoryEvent{\n\t\t\tID:        1,\n\t\t\tTimestamp: common.Ptr(int64(1709938156435726000)),\n\t\t\tEventType: common.Ptr(types.EventTypeWorkflowExecutionStarted),\n\t\t\tVersion:   version,\n\t\t\tTaskID:    17826364,\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"helloWorldWorkflow\"},\n\t\t\t\tTaskList:                            &types.TaskList{Name: \"helloWorldGroup\"},\n\t\t\t\tInput:                               []uint8{110, 117, 108, 108, 10},\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Ptr(int32(60)),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Ptr(int32(60)),\n\t\t\t\tContinuedExecutionRunID:             \"96892ca6-975a-44b1-9726-cdb63acd8cda\",\n\t\t\t\tOriginalExecutionRunID:              \"befc5b41-fb06-4a99-bec2-91c3e98b17d7\",\n\t\t\t\tFirstExecutionRunID:                 \"bcdee7e4-cb21-4bbb-a8d1-43da79e3d252\",\n\t\t\t\tPrevAutoResetPoints: &types.ResetPoints{Points: []*types.ResetPointInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tBinaryChecksum:           \"6df03bf5110d681667852a8456519536\",\n\t\t\t\t\t\tRunID:                    \"bcdee7e4-cb21-4bbb-a8d1-43da79e3d252\",\n\t\t\t\t\t\tFirstDecisionCompletedID: 4,\n\t\t\t\t\t\tCreatedTimeNano:          common.Ptr(int64(1709938002170829000)),\n\t\t\t\t\t\tExpiringTimeNano:         common.Ptr(int64(1710197212347858000)),\n\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t},\n\t\t\t\t}},\n\t\t\t\tHeader: &types.Header{},\n\t\t\t},\n\t\t}\n\t}\n\n\texpectedEndingReturnExecutionStateFn := func(version int64) *persistence.WorkflowExecutionInfo {\n\t\treturn &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:                           \"5391dbea-5b30-4323-82ca-e1c95339bb3e\",\n\t\t\tWorkflowID:                         \"helloworld_b4db8bd0-74b7-4250-ade7-ac72a1efb171\",\n\t\t\tRunID:                              \"a run id\",\n\t\t\tFirstExecutionRunID:                \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\tInitiatedID:                        -23,\n\t\t\tTaskList:                           \"helloWorldGroup\",\n\t\t\tTaskListKind:                       types.TaskListKindNormal,\n\t\t\tWorkflowTypeName:                   \"helloWorldWorkflow\",\n\t\t\tWorkflowTimeout:                    60,\n\t\t\tDecisionStartToCloseTimeout:        60,\n\t\t\tState:                              1,\n\t\t\tLastFirstEventID:                   1,\n\t\t\tNextEventID:                        3,\n\t\t\tLastProcessedEvent:                 -23,\n\t\t\tStartTimestamp:                     time.Unix(0, ts3),\n\t\t\tCreateRequestID:                    \"4630bf04-5c64-41bf-92d9-576db2d535cb\",\n\t\t\tDecisionVersion:                    version,\n\t\t\tDecisionScheduleID:                 2,\n\t\t\tDecisionStartedID:                  -23,\n\t\t\tDecisionRequestID:                  \"emptyUuid\",\n\t\t\tDecisionTimeout:                    60,\n\t\t\tDecisionScheduledTimestamp:         ts3,\n\t\t\tDecisionOriginalScheduledTimestamp: ts3,\n\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\tPoints: []*types.ResetPointInfo{{\n\t\t\t\t\tBinaryChecksum:           \"6df03bf5110d681667852a8456519536\",\n\t\t\t\t\tRunID:                    \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\t\t\tFirstDecisionCompletedID: 4,\n\t\t\t\t\tCreatedTimeNano:          common.Ptr(int64(ts1)),\n\t\t\t\t\tExpiringTimeNano:         common.Ptr(int64(ts3)),\n\t\t\t\t\tResettable:               true,\n\t\t\t\t}},\n\t\t\t},\n\t\t}\n\t}\n\n\texpectedEndingReturnHistoryStateFn := func(version int64) []*types.HistoryEvent {\n\t\treturn []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tTimestamp: common.Ptr(int64(ts3)),\n\t\t\t\tEventType: common.Ptr(types.EventTypeWorkflowExecutionStarted),\n\t\t\t\tVersion:   version,\n\t\t\t\tTaskID:    -1234,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"helloWorldWorkflow\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"helloWorldGroup\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tInput:                               []uint8{110, 117, 108, 108, 10},\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Ptr(int32(60)),\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Ptr(int32(60)),\n\t\t\t\t\tContinuedExecutionRunID:             \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\t\t\tOriginalExecutionRunID:              \"a run id\",\n\t\t\t\t\tFirstExecutionRunID:                 \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\t\t\tPrevAutoResetPoints: &types.ResetPoints{Points: []*types.ResetPointInfo{{\n\t\t\t\t\t\tBinaryChecksum:           \"6df03bf5110d681667852a8456519536\",\n\t\t\t\t\t\tRunID:                    \"5adce5c5-b7b2-4418-9bf0-4207303f6343\",\n\t\t\t\t\t\tFirstDecisionCompletedID: 4,\n\t\t\t\t\t\tCreatedTimeNano:          common.Ptr(int64(ts1)),\n\t\t\t\t\t\tExpiringTimeNano:         common.Ptr(int64(ts3)),\n\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t}}},\n\t\t\t\t\tHeader: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tTimestamp: common.Ptr(int64(ts3)),\n\t\t\t\tEventType: common.Ptr(types.EventTypeDecisionTaskScheduled),\n\t\t\t\tVersion:   version,\n\t\t\t\tTaskID:    -1234,\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\tTaskList:                   &types.TaskList{Name: \"helloWorldGroup\"},\n\t\t\t\t\tStartToCloseTimeoutSeconds: common.Ptr(int32(60)),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t}\n\n\ttests := map[string]struct {\n\t\tdomainEntry   *cache.DomainCacheEntry\n\t\tstartingState *persistence.WorkflowExecutionInfo\n\t\t// history is a substruct of current state, but because they're both\n\t\t// pointing to each other, they're assembled at the test start\n\t\tstartingHistory []*types.HistoryEvent\n\n\t\t// expectations\n\t\thistoryManagerAffordance func(historyManager *persistence.MockHistoryManager)\n\t\ttaskgeneratorAffordance  func(taskGenerator *MockMutableStateTaskGenerator, msb *mutableStateBuilder)\n\t\tactClMgrAffordance       func(actClMgr *activecluster.MockManager)\n\t\texpectedReturnedState    *persistence.WorkflowExecutionInfo // this is returned\n\t\texpectedReturnedHistory  []*types.HistoryEvent\n\t\texpectedErr              error\n\t}{\n\t\t\"a continue-as-new event with no errors\": {\n\t\t\tdomainEntry:     domainEntry,\n\t\t\tstartingState:   createStartingExecutionInfo(),\n\t\t\tstartingHistory: createValidStartingHistory(domainFailoverVersion),\n\t\t\tactClMgrAffordance: func(actClMgr *activecluster.MockManager) {\n\t\t\t\tactClMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: domainFailoverVersion,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\t// when it goes to fetch the starting event\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\tcreateFetchedHistory(domainFailoverVersion),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\ttaskgeneratorAffordance: func(taskGenerator *MockMutableStateTaskGenerator, msb *mutableStateBuilder) {\n\t\t\t\ttaskGenerator.EXPECT().GenerateWorkflowCloseTasks(gomock.Any(), msb.config.WorkflowDeletionJitterRange(\"domain\"))\n\t\t\t},\n\t\t\texpectedReturnedState:   expectedEndingReturnExecutionStateFn(1),\n\t\t\texpectedReturnedHistory: expectedEndingReturnHistoryStateFn(1),\n\t\t},\n\t\t\"a continue-as-new event with no errors - active-active domain\": {\n\t\t\tdomainEntry:     domainEntryActiveActive,\n\t\t\tstartingState:   createStartingExecutionInfo(),\n\t\t\tstartingHistory: createValidStartingHistory(1),\n\t\t\tactClMgrAffordance: func(actClMgr *activecluster.MockManager) {\n\t\t\t\tactClMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: 2, // this version will be used by new mutable state builder for new tasks\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\tcreateFetchedHistory(2),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\ttaskgeneratorAffordance: func(taskGenerator *MockMutableStateTaskGenerator, msb *mutableStateBuilder) {\n\t\t\t\ttaskGenerator.EXPECT().GenerateWorkflowCloseTasks(gomock.Any(), msb.config.WorkflowDeletionJitterRange(\"domain\"))\n\t\t\t},\n\t\t\texpectedReturnedState:   expectedEndingReturnExecutionStateFn(2),\n\t\t\texpectedReturnedHistory: expectedEndingReturnHistoryStateFn(2),\n\t\t},\n\t\t\"a continue-as-new with failure to get the history event\": {\n\t\t\tdomainEntry:     domainEntry,\n\t\t\tstartingState:   createStartingExecutionInfo(),\n\t\t\tstartingHistory: createValidStartingHistory(domainFailoverVersion),\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"an error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"an error\"),\n\t\t},\n\t\t\"a continue-as-new with errors in replicating\": {\n\t\t\tdomainEntry:     domainEntry,\n\t\t\tstartingState:   createStartingExecutionInfo(),\n\t\t\tstartingHistory: createValidStartingHistory(domainFailoverVersion),\n\t\t\tactClMgrAffordance: func(actClMgr *activecluster.MockManager) {\n\t\t\t\tactClMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: domainFailoverVersion,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\tcreateFetchedHistory(domainFailoverVersion),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\ttaskgeneratorAffordance: func(taskGenerator *MockMutableStateTaskGenerator, msb *mutableStateBuilder) {\n\t\t\t\ttaskGenerator.EXPECT().GenerateWorkflowCloseTasks(gomock.Any(), msb.config.WorkflowDeletionJitterRange(\"domain\")).Return(errors.New(\"an error\"))\n\t\t\t},\n\t\t\texpectedErr: errors.New(\"an error\"),\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := log.NewNoop()\n\t\t\tmsb := &mutableStateBuilder{\n\t\t\t\tdomainEntry:    td.domainEntry,\n\t\t\t\texecutionInfo:  td.startingState,\n\t\t\t\tlogger:         logger,\n\t\t\t\tconfig:         config.NewForTest(),\n\t\t\t\tcurrentVersion: domainFailoverVersion,\n\t\t\t}\n\n\t\t\tactClMgr := activecluster.NewMockManager(ctrl)\n\t\t\tshardContext := shardCtx.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(logger).AnyTimes()\n\t\t\tshardContext.EXPECT().GetActiveClusterManager().Return(actClMgr).AnyTimes()\n\t\t\thistoryManager := persistence.NewMockHistoryManager(ctrl)\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\ttaskGenerator := NewMockMutableStateTaskGenerator(ctrl)\n\n\t\t\tmsb.timeSource = clock.NewMockedTimeSourceAt(time.Unix(0, ts3))\n\t\t\tmsb.eventsCache = events.NewCache(shardID,\n\t\t\t\thistoryManager,\n\t\t\t\tconfig.NewForTest(),\n\t\t\t\tlogger,\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tdomainCache)\n\t\t\tmsb.shard = shardContext\n\t\t\tmsb.executionInfo = td.startingState\n\t\t\tmsb.hBuilder = &HistoryBuilder{\n\t\t\t\thistory:   td.startingHistory,\n\t\t\t\tmsBuilder: msb,\n\t\t\t}\n\t\t\tmsb.taskGenerator = taskGenerator\n\n\t\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domain\", nil).AnyTimes()\n\n\t\t\tshardContext.EXPECT().GetShardID().Return(123).AnyTimes()\n\t\t\tshardContext.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).AnyTimes()\n\t\t\tshardContext.EXPECT().GetEventsCache().Return(msb.eventsCache).AnyTimes()\n\t\t\tshardContext.EXPECT().GetConfig().Return(msb.config).AnyTimes()\n\t\t\tshardContext.EXPECT().GetTimeSource().Return(msb.timeSource).AnyTimes()\n\t\t\tshardContext.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\t\t\tshardContext.EXPECT().GetDomainCache().Return(domainCache).AnyTimes()\n\n\t\t\tif td.historyManagerAffordance != nil {\n\t\t\t\ttd.historyManagerAffordance(historyManager)\n\t\t\t}\n\t\t\tif td.taskgeneratorAffordance != nil {\n\t\t\t\ttd.taskgeneratorAffordance(taskGenerator, msb)\n\t\t\t}\n\t\t\tif td.actClMgrAffordance != nil {\n\t\t\t\ttd.actClMgrAffordance(actClMgr)\n\t\t\t}\n\n\t\t\t_, returnedBuilder, err := msb.AddContinueAsNewEvent(context.Background(),\n\t\t\t\tfirstEventID,\n\t\t\t\tdecisionCompletedEventID,\n\t\t\t\t\"\",\n\t\t\t\t&types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\t\tName: \"helloWorldWorkflow\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"helloWorldGroup\",\n\t\t\t\t\t},\n\t\t\t\t\tInput: []uint8{110, 117, 108, 108, 10},\n\t\t\t\t})\n\n\t\t\tif td.expectedErr != nil {\n\t\t\t\tassert.ErrorAs(t, err, &td.expectedErr)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tresultExecutionInfo := returnedBuilder.GetExecutionInfo()\n\n\t\t\tassert.Empty(t, cmp.Diff(td.expectedReturnedState, resultExecutionInfo,\n\t\t\t\t// these are generated nondeterministically, with a plain guid generator\n\t\t\t\t// todo(david): make this mockable\n\t\t\t\tcmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"OriginalExecutionRunID\"),\n\t\t\t\tcmpopts.IgnoreFields(types.WorkflowExecution{}, \"RunID\"),\n\t\t\t\tcmpopts.IgnoreFields(persistence.WorkflowExecutionInfo{}, \"RunID\", \"CreateRequestID\"),\n\t\t\t))\n\n\t\t\tassert.Empty(t, cmp.Diff(td.expectedReturnedHistory, returnedBuilder.GetHistoryBuilder().history,\n\t\t\t\tcmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"OriginalExecutionRunID\"),\n\t\t\t\tcmpopts.IgnoreFields(types.WorkflowExecutionStartedEventAttributes{}, \"RequestID\")),\n\t\t\t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_activity.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// GetActivityInfo gives details about an activity that is currently in progress.\nfunc (e *mutableStateBuilder) GetActivityInfo(\n\tscheduleEventID int64,\n) (*persistence.ActivityInfo, bool) {\n\n\tai, ok := e.pendingActivityInfoIDs[scheduleEventID]\n\treturn ai, ok\n}\n\nfunc (e *mutableStateBuilder) GetPendingActivityInfos() map[int64]*persistence.ActivityInfo {\n\treturn e.pendingActivityInfoIDs\n}\n\n// GetActivityByActivityID gives details about an activity that is currently in progress.\nfunc (e *mutableStateBuilder) GetActivityByActivityID(\n\tactivityID string,\n) (*persistence.ActivityInfo, bool) {\n\n\teventID, ok := e.pendingActivityIDToEventID[activityID]\n\tif !ok {\n\t\treturn nil, false\n\t}\n\treturn e.GetActivityInfo(eventID)\n}\n\nfunc (e *mutableStateBuilder) UpdateActivityProgress(\n\tai *persistence.ActivityInfo,\n\trequest *types.RecordActivityTaskHeartbeatRequest,\n) {\n\tai.Version = e.GetCurrentVersion()\n\tai.Details = request.Details\n\tai.LastHeartBeatUpdatedTime = e.timeSource.Now()\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\te.syncActivityTasks[ai.ScheduleID] = struct{}{}\n}\n\n// ReplicateActivityInfo replicate the necessary activity information\nfunc (e *mutableStateBuilder) ReplicateActivityInfo(\n\trequest *types.SyncActivityRequest,\n\tresetActivityTimerTaskStatus bool,\n) error {\n\tai, ok := e.pendingActivityInfoIDs[request.GetScheduledID()]\n\tif !ok {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find activity event ID: %v in mutable state\", request.GetScheduledID()),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\treturn ErrMissingActivityInfo\n\t}\n\n\tai.Version = request.GetVersion()\n\tai.ScheduledTime = time.Unix(0, request.GetScheduledTime())\n\tai.StartedID = request.GetStartedID()\n\tai.LastHeartBeatUpdatedTime = time.Unix(0, request.GetLastHeartbeatTime())\n\tif ai.StartedID == constants.EmptyEventID {\n\t\tai.StartedTime = time.Time{}\n\t} else {\n\t\tai.StartedTime = time.Unix(0, request.GetStartedTime())\n\t}\n\tai.Details = request.GetDetails()\n\tai.Attempt = request.GetAttempt()\n\tai.LastFailureReason = request.GetLastFailureReason()\n\tai.LastWorkerIdentity = request.GetLastWorkerIdentity()\n\tai.LastFailureDetails = request.GetLastFailureDetails()\n\n\tif resetActivityTimerTaskStatus {\n\t\tai.TimerTaskStatus = TimerTaskStatusNone\n\t}\n\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\treturn nil\n}\n\n// UpdateActivity updates an activity\nfunc (e *mutableStateBuilder) UpdateActivity(\n\tai *persistence.ActivityInfo,\n) error {\n\n\tif _, ok := e.pendingActivityInfoIDs[ai.ScheduleID]; !ok {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find activity ID: %v in mutable state\", ai.ActivityID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\treturn ErrMissingActivityInfo\n\t}\n\n\te.pendingActivityInfoIDs[ai.ScheduleID] = ai\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\treturn nil\n}\n\n// DeleteActivity deletes details about an activity.\nfunc (e *mutableStateBuilder) DeleteActivity(\n\tscheduleEventID int64,\n) error {\n\n\tif activityInfo, ok := e.pendingActivityInfoIDs[scheduleEventID]; ok {\n\t\tdelete(e.pendingActivityInfoIDs, scheduleEventID)\n\n\t\tif _, ok = e.pendingActivityIDToEventID[activityInfo.ActivityID]; ok {\n\t\t\tdelete(e.pendingActivityIDToEventID, activityInfo.ActivityID)\n\t\t} else {\n\t\t\te.logError(\n\t\t\t\tfmt.Sprintf(\"unable to find activity ID: %v in mutable state\", activityInfo.ActivityID),\n\t\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t\t)\n\t\t\t// log data inconsistency instead of returning an error\n\t\t\te.logDataInconsistency()\n\t\t}\n\t} else {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find activity event id: %v in mutable state\", scheduleEventID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\t// log data inconsistency instead of returning an error\n\t\te.logDataInconsistency()\n\t}\n\n\tdelete(e.updateActivityInfos, scheduleEventID)\n\te.deleteActivityInfos[scheduleEventID] = struct{}{}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) GetActivityScheduledEvent(\n\tctx context.Context,\n\tscheduleEventID int64,\n) (*types.HistoryEvent, error) {\n\n\tai, ok := e.pendingActivityInfoIDs[scheduleEventID]\n\tif !ok {\n\t\treturn nil, ErrMissingActivityInfo\n\t}\n\n\t// Needed for backward compatibility reason\n\tif ai.ScheduledEvent != nil {\n\t\treturn ai.ScheduledEvent, nil\n\t}\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tscheduledEvent, err := e.eventsCache.GetEvent(\n\t\tctx,\n\t\te.shard.GetShardID(),\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tai.ScheduledEventBatchID,\n\t\tai.ScheduleID,\n\t\tcurrentBranchToken,\n\t)\n\tif err != nil {\n\t\t// do not return the original error\n\t\t// since original error can be of type entity not exists\n\t\t// which can cause task processing side to fail silently\n\t\t// However, if the error is a persistence transient error,\n\t\t// we return the original error, because we fail to get\n\t\t// the event because of failure from database\n\t\tif persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, ErrMissingActivityScheduledEvent\n\t}\n\treturn scheduledEvent, nil\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskScheduledEvent(\n\tctx context.Context,\n\tdecisionCompletedEventID int64,\n\tattributes *types.ScheduleActivityTaskDecisionAttributes,\n\tdispatch bool,\n) (*types.HistoryEvent, *persistence.ActivityInfo, *types.ActivityLocalDispatchInfo, bool, bool, error) {\n\n\topTag := tag.WorkflowActionActivityTaskScheduled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, nil, false, false, err\n\t}\n\n\t_, ok := e.GetActivityByActivityID(attributes.GetActivityID())\n\tif ok {\n\t\te.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction)\n\t\treturn nil, nil, nil, false, false, e.createCallerError(opTag)\n\t}\n\n\tpendingActivitiesCount := len(e.pendingActivityInfoIDs)\n\n\tif pendingActivitiesCount >= e.config.PendingActivitiesCountLimitError() {\n\t\te.logger.Error(\"Pending activity count exceeds error limit\",\n\t\t\ttag.WorkflowDomainName(e.GetDomainEntry().GetInfo().Name),\n\t\t\ttag.WorkflowID(e.executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(e.executionInfo.RunID),\n\t\t\ttag.Number(int64(pendingActivitiesCount)))\n\n\t\tif e.config.PendingActivityValidationEnabled() {\n\t\t\treturn nil, nil, nil, false, false, ErrTooManyPendingActivities\n\t\t}\n\t} else if pendingActivitiesCount >= e.config.PendingActivitiesCountLimitWarn() && !e.pendingActivityWarningSent {\n\t\te.logger.Warn(\"Pending activity count exceeds warn limit\",\n\t\t\ttag.WorkflowDomainName(e.GetDomainEntry().GetInfo().Name),\n\t\t\ttag.WorkflowID(e.executionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(e.executionInfo.RunID),\n\t\t\ttag.Number(int64(pendingActivitiesCount)))\n\n\t\te.pendingActivityWarningSent = true\n\t}\n\n\tevent := e.hBuilder.AddActivityTaskScheduledEvent(decisionCompletedEventID, attributes)\n\n\t// Write the event to cache only on active cluster for processing on activity started or retried\n\te.eventsCache.PutEvent(\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tevent.ID,\n\t\tevent,\n\t)\n\n\tai, err := e.ReplicateActivityTaskScheduledEvent(decisionCompletedEventID, event, true)\n\tif err != nil {\n\t\treturn nil, nil, nil, false, false, err\n\t}\n\tactivityStartedScope := e.metricsClient.Scope(metrics.HistoryRecordActivityTaskStartedScope)\n\tif e.config.EnableActivityLocalDispatchByDomain(e.domainEntry.GetInfo().Name) && attributes.RequestLocalDispatch {\n\t\tactivityStartedScope.IncCounter(metrics.CadenceRequests)\n\t\treturn event, ai, &types.ActivityLocalDispatchInfo{ActivityID: ai.ActivityID}, false, false, nil\n\t}\n\tstarted := false\n\tif dispatch {\n\t\tstarted = e.tryDispatchActivityTask(ctx, event, ai)\n\t}\n\tif started {\n\t\tactivityStartedScope.IncCounter(metrics.CadenceRequests)\n\t\treturn event, ai, nil, true, true, nil\n\t}\n\n\tif err := e.taskGenerator.GenerateActivityTransferTasks(event); err != nil {\n\t\treturn nil, nil, nil, dispatch, false, err\n\t}\n\n\treturn event, ai, nil, dispatch, false, err\n}\n\nfunc (e *mutableStateBuilder) tryDispatchActivityTask(\n\tctx context.Context,\n\tscheduledEvent *types.HistoryEvent,\n\tai *persistence.ActivityInfo,\n) bool {\n\ttaggedScope := e.metricsClient.Scope(metrics.HistoryScheduleDecisionTaskScope).Tagged(\n\t\tmetrics.DomainTag(e.domainEntry.GetInfo().Name),\n\t\tmetrics.WorkflowTypeTag(e.GetWorkflowType().Name),\n\t\tmetrics.TaskListTag(ai.TaskList))\n\ttaggedScope.IncCounter(metrics.DecisionTypeScheduleActivityDispatchCounter)\n\t_, err := e.shard.GetService().GetMatchingClient().AddActivityTask(ctx, &types.AddActivityTaskRequest{\n\t\tDomainUUID:       e.executionInfo.DomainID,\n\t\tSourceDomainUUID: e.domainEntry.GetInfo().ID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: e.executionInfo.WorkflowID,\n\t\t\tRunID:      e.executionInfo.RunID,\n\t\t},\n\t\tTaskList:                      &types.TaskList{Name: ai.TaskList},\n\t\tScheduleID:                    scheduledEvent.ID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(ai.ScheduleToStartTimeout),\n\t\tActivityTaskDispatchInfo: &types.ActivityTaskDispatchInfo{\n\t\t\tScheduledEvent:                  scheduledEvent,\n\t\t\tStartedTimestamp:                common.Int64Ptr(e.timeSource.Now().UnixNano()),\n\t\t\tWorkflowType:                    e.GetWorkflowType(),\n\t\t\tWorkflowDomain:                  e.GetDomainEntry().GetInfo().Name,\n\t\t\tScheduledTimestampOfThisAttempt: common.Int64Ptr(ai.ScheduledTime.UnixNano()),\n\t\t},\n\t\tPartitionConfig: e.executionInfo.PartitionConfig,\n\t})\n\tif err == nil {\n\t\ttaggedScope.IncCounter(metrics.DecisionTypeScheduleActivityDispatchSucceedCounter)\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskScheduledEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n\tskipTaskGeneration bool,\n) (*persistence.ActivityInfo, error) {\n\n\tattributes := event.ActivityTaskScheduledEventAttributes\n\ttargetDomainID := e.executionInfo.DomainID\n\tif attributes.GetDomain() != \"\" {\n\t\tvar err error\n\t\ttargetDomainID, err = e.shard.GetDomainCache().GetDomainID(attributes.GetDomain())\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tscheduleEventID := event.ID\n\tscheduleToCloseTimeout := attributes.GetScheduleToCloseTimeoutSeconds()\n\n\tai := &persistence.ActivityInfo{\n\t\tVersion:                  event.Version,\n\t\tScheduleID:               scheduleEventID,\n\t\tScheduledEventBatchID:    firstEventID,\n\t\tScheduledTime:            time.Unix(0, event.GetTimestamp()),\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               attributes.ActivityID,\n\t\tDomainID:                 targetDomainID,\n\t\tScheduleToStartTimeout:   attributes.GetScheduleToStartTimeoutSeconds(),\n\t\tScheduleToCloseTimeout:   scheduleToCloseTimeout,\n\t\tStartToCloseTimeout:      attributes.GetStartToCloseTimeoutSeconds(),\n\t\tHeartbeatTimeout:         attributes.GetHeartbeatTimeoutSeconds(),\n\t\tCancelRequested:          false,\n\t\tCancelRequestID:          constants.EmptyEventID,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tTaskList:                 attributes.TaskList.GetName(),\n\t\tTaskListKind:             attributes.TaskList.GetKind(),\n\t\tHasRetryPolicy:           attributes.RetryPolicy != nil,\n\t}\n\n\tif ai.HasRetryPolicy {\n\t\tai.InitialInterval = attributes.RetryPolicy.GetInitialIntervalInSeconds()\n\t\tai.BackoffCoefficient = attributes.RetryPolicy.GetBackoffCoefficient()\n\t\tai.MaximumInterval = attributes.RetryPolicy.GetMaximumIntervalInSeconds()\n\t\tai.MaximumAttempts = attributes.RetryPolicy.GetMaximumAttempts()\n\t\tai.NonRetriableErrors = attributes.RetryPolicy.NonRetriableErrorReasons\n\t\tif attributes.RetryPolicy.GetExpirationIntervalInSeconds() != 0 {\n\t\t\tai.ExpirationTime = ai.ScheduledTime.Add(time.Duration(attributes.RetryPolicy.GetExpirationIntervalInSeconds()) * time.Second)\n\t\t}\n\t}\n\n\te.pendingActivityInfoIDs[scheduleEventID] = ai\n\te.pendingActivityIDToEventID[ai.ActivityID] = scheduleEventID\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\n\tif !skipTaskGeneration {\n\t\treturn ai, e.taskGenerator.GenerateActivityTransferTasks(event)\n\t}\n\n\treturn ai, nil\n}\n\nfunc (e *mutableStateBuilder) addTransientActivityStartedEvent(\n\tscheduleEventID int64,\n) error {\n\n\tai, ok := e.GetActivityInfo(scheduleEventID)\n\tif !ok || ai.StartedID != constants.TransientEventID {\n\t\treturn nil\n\t}\n\n\t// activity task was started (as transient event), we need to add it now.\n\tevent := e.hBuilder.AddActivityTaskStartedEvent(scheduleEventID, ai.Attempt, ai.RequestID, ai.StartedIdentity,\n\t\tai.LastFailureReason, ai.LastFailureDetails)\n\tif !ai.StartedTime.IsZero() {\n\t\t// overwrite started event time to the one recorded in ActivityInfo\n\t\tevent.Timestamp = common.Int64Ptr(ai.StartedTime.UnixNano())\n\t}\n\treturn e.ReplicateActivityTaskStartedEvent(event)\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskStartedEvent(\n\tai *persistence.ActivityInfo,\n\tscheduleEventID int64,\n\trequestID string,\n\tidentity string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskStarted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif !ai.HasRetryPolicy {\n\t\tevent := e.hBuilder.AddActivityTaskStartedEvent(scheduleEventID, ai.Attempt, requestID, identity,\n\t\t\tai.LastFailureReason, ai.LastFailureDetails)\n\t\tif err := e.ReplicateActivityTaskStartedEvent(event); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn event, nil\n\t}\n\n\t// we might need to retry, so do not append started event just yet,\n\t// instead update mutable state and will record started event when activity task is closed\n\tai.Version = e.GetCurrentVersion()\n\tai.StartedID = constants.TransientEventID\n\tai.RequestID = requestID\n\tai.StartedTime = e.timeSource.Now()\n\tai.LastHeartBeatUpdatedTime = ai.StartedTime\n\tai.StartedIdentity = identity\n\tif err := e.UpdateActivity(ai); err != nil {\n\t\treturn nil, err\n\t}\n\te.syncActivityTasks[ai.ScheduleID] = struct{}{}\n\treturn nil, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskStartedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskStartedEventAttributes\n\tscheduleID := attributes.GetScheduledEventID()\n\tai, ok := e.GetActivityInfo(scheduleID)\n\tif !ok {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find activity event id: %v in mutable state\", scheduleID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\treturn ErrMissingActivityInfo\n\t}\n\n\tai.Version = event.Version\n\tai.StartedID = event.ID\n\tai.RequestID = attributes.GetRequestID()\n\tai.StartedTime = time.Unix(0, event.GetTimestamp())\n\tai.LastHeartBeatUpdatedTime = ai.StartedTime\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskCompletedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\trequest *types.RespondActivityTaskCompletedRequest,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskCompleted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif ai, ok := e.GetActivityInfo(scheduleEventID); !ok || ai.StartedID != startedEventID {\n\t\te.logger.Warn(\n\t\t\tmutableStateInvalidHistoryActionMsg,\n\t\t\topTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tif err := e.addTransientActivityStartedEvent(scheduleEventID); err != nil {\n\t\treturn nil, err\n\t}\n\tevent := e.hBuilder.AddActivityTaskCompletedEvent(scheduleEventID, startedEventID, request)\n\tif err := e.ReplicateActivityTaskCompletedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskCompletedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskCompletedEventAttributes\n\tscheduleID := attributes.GetScheduledEventID()\n\n\treturn e.DeleteActivity(scheduleID)\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskFailedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\trequest *types.RespondActivityTaskFailedRequest,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif ai, ok := e.GetActivityInfo(scheduleEventID); !ok || ai.StartedID != startedEventID {\n\t\te.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tif err := e.addTransientActivityStartedEvent(scheduleEventID); err != nil {\n\t\treturn nil, err\n\t}\n\tevent := e.hBuilder.AddActivityTaskFailedEvent(scheduleEventID, startedEventID, request)\n\tif err := e.ReplicateActivityTaskFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskFailedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskFailedEventAttributes\n\tscheduleID := attributes.GetScheduledEventID()\n\n\treturn e.DeleteActivity(scheduleID)\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskTimedOutEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\ttimeoutType types.TimeoutType,\n\tlastHeartBeatDetails []byte,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskTimedOut\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tai, ok := e.GetActivityInfo(scheduleEventID)\n\tif !ok || ai.StartedID != startedEventID || ((timeoutType == types.TimeoutTypeStartToClose ||\n\t\ttimeoutType == types.TimeoutTypeHeartbeat) && ai.StartedID == constants.EmptyEventID) {\n\t\te.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID),\n\t\t\ttag.WorkflowTimeoutType(int64(timeoutType)))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tif err := e.addTransientActivityStartedEvent(scheduleEventID); err != nil {\n\t\treturn nil, err\n\t}\n\tevent := e.hBuilder.AddActivityTaskTimedOutEvent(scheduleEventID, startedEventID, timeoutType, lastHeartBeatDetails, ai.LastFailureReason, ai.LastFailureDetails)\n\tif err := e.ReplicateActivityTaskTimedOutEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskTimedOutEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskTimedOutEventAttributes\n\tscheduleID := attributes.GetScheduledEventID()\n\n\treturn e.DeleteActivity(scheduleID)\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskCancelRequestedEvent(\n\tdecisionCompletedEventID int64,\n\tactivityID string,\n\tidentity string,\n) (*types.HistoryEvent, *persistence.ActivityInfo, error) {\n\n\topTag := tag.WorkflowActionActivityTaskCancelRequested\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\t// we need to add the cancel request event even if activity not in mutable state\n\t// if activity not in mutable state or already cancel requested,\n\t// we do not need to call the replication function\n\tactCancelReqEvent := e.hBuilder.AddActivityTaskCancelRequestedEvent(decisionCompletedEventID, activityID)\n\n\tai, ok := e.GetActivityByActivityID(activityID)\n\tif !ok || ai.CancelRequested {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowActivityID(activityID))\n\n\t\treturn nil, nil, e.createCallerError(opTag)\n\t}\n\n\tif err := e.ReplicateActivityTaskCancelRequestedEvent(actCancelReqEvent); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\treturn actCancelReqEvent, ai, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskCancelRequestedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskCancelRequestedEventAttributes\n\tactivityID := attributes.GetActivityID()\n\tai, ok := e.GetActivityByActivityID(activityID)\n\tif !ok {\n\t\t// On active side, if the ActivityTaskCancelRequested is invalid, it will created a RequestCancelActivityTaskFailed\n\t\t// Passive will rely on active side logic\n\t\treturn nil\n\t}\n\n\tai.Version = event.Version\n\n\t// - We have the activity dispatched to worker.\n\t// - The activity might not be heartbeating, but the activity can still call RecordActivityHeartBeat()\n\t//   to see cancellation while reporting progress of the activity.\n\tai.CancelRequested = true\n\n\tai.CancelRequestID = event.ID\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddRequestCancelActivityTaskFailedEvent(\n\tdecisionCompletedEventID int64,\n\tactivityID string,\n\tcause string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskCancelFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn e.hBuilder.AddRequestCancelActivityTaskFailedEvent(decisionCompletedEventID, activityID, cause), nil\n}\n\nfunc (e *mutableStateBuilder) AddActivityTaskCanceledEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\tlatestCancelRequestedEventID int64,\n\tdetails []byte,\n\tidentity string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionActivityTaskCanceled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tai, ok := e.GetActivityInfo(scheduleEventID)\n\tif !ok || ai.StartedID != startedEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\t// Verify cancel request as well.\n\tif !ai.CancelRequested {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowActivityID(ai.ActivityID),\n\t\t\ttag.WorkflowStartedID(ai.StartedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tif err := e.addTransientActivityStartedEvent(scheduleEventID); err != nil {\n\t\treturn nil, err\n\t}\n\tevent := e.hBuilder.AddActivityTaskCanceledEvent(scheduleEventID, startedEventID, latestCancelRequestedEventID,\n\t\tdetails, identity)\n\tif err := e.ReplicateActivityTaskCanceledEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateActivityTaskCanceledEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ActivityTaskCanceledEventAttributes\n\tscheduleID := attributes.GetScheduledEventID()\n\n\treturn e.DeleteActivity(scheduleID)\n}\n\nfunc (e *mutableStateBuilder) RetryActivity(\n\tai *persistence.ActivityInfo,\n\tfailureReason string,\n\tfailureDetails []byte,\n) (bool, error) {\n\n\topTag := tag.WorkflowActionActivityTaskRetry\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn false, err\n\t}\n\n\tif !ai.HasRetryPolicy || ai.CancelRequested {\n\t\treturn false, nil\n\t}\n\n\tnow := e.timeSource.Now()\n\n\tbackoffInterval := getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\tfailureReason,\n\t\tai.NonRetriableErrors,\n\t)\n\tif backoffInterval == backoff.NoBackoff {\n\t\treturn false, nil\n\t}\n\n\t// a retry is needed, update activity info for next retry\n\tai.Version = e.GetCurrentVersion()\n\tai.Attempt++\n\tai.ScheduledTime = now.Add(backoffInterval) // update to next schedule time\n\tai.StartedID = constants.EmptyEventID\n\tai.RequestID = \"\"\n\tai.StartedTime = time.Time{}\n\tai.TimerTaskStatus = TimerTaskStatusNone\n\tai.LastFailureReason = failureReason\n\tai.LastWorkerIdentity = ai.StartedIdentity\n\tai.LastFailureDetails = failureDetails\n\n\tif err := e.taskGenerator.GenerateActivityRetryTasks(\n\t\tai.ScheduleID,\n\t); err != nil {\n\t\treturn false, err\n\t}\n\n\te.updateActivityInfos[ai.ScheduleID] = ai\n\te.syncActivityTasks[ai.ScheduleID] = struct{}{}\n\treturn true, nil\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_activity_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nvar currentTime = time.Unix(0, 1)\n\nfunc testMutableStateBuilder(t *testing.T) *mutableStateBuilder {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewTestContext(\n\t\tt,\n\t\tctrl,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\tmockShard.Resource.TimeSource = clock.NewMockedTimeSourceAt(currentTime)\n\t// set the checksum probabilities to 100% for exercising during test\n\tmockShard.GetConfig().MutableStateChecksumGenProbability = func(domain string) int { return 100 }\n\tmockShard.GetConfig().MutableStateChecksumVerifyProbability = func(domain string) int { return 100 }\n\tmockShard.GetConfig().EnableRetryForChecksumFailure = func(domain string) bool { return true }\n\tlogger := log.NewNoop()\n\n\tmockShard.Resource.MatchingClient.EXPECT().AddActivityTask(gomock.Any(), gomock.Any()).Return(&types.AddActivityTaskResponse{}, nil).AnyTimes()\n\tmockShard.Resource.DomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\tmockShard.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(&cache.DomainCacheEntry{}, nil).AnyTimes()\n\treturn newMutableStateBuilder(mockShard, logger, constants.TestLocalDomainEntry, constants.TestLocalDomainEntry.GetFailoverVersion())\n}\n\nfunc Test__AddActivityTaskScheduledEvent(t *testing.T) {\n\tactivityType := &types.ActivityType{Name: \"activityType\"}\n\tretryPolicy := &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    11,\n\t\tBackoffCoefficient:          12,\n\t\tMaximumIntervalInSeconds:    13,\n\t\tMaximumAttempts:             14,\n\t\tNonRetriableErrorReasons:    []string{\"no retries please\"},\n\t\tExpirationIntervalInSeconds: 15,\n\t}\n\theader := &types.Header{Fields: map[string][]byte{\n\t\t\"key\": []byte(\"value\"),\n\t}}\n\tcases := []struct {\n\t\tname               string\n\t\tworkflowTaskList   *types.TaskList\n\t\tattr               *types.ScheduleActivityTaskDecisionAttributes\n\t\tdispatch           bool\n\t\texpectedAttributes *types.ActivityTaskScheduledEventAttributes\n\t\texpectedInfo       *persistence.ActivityInfo\n\t\texpectedDispatch   *types.ActivityLocalDispatchInfo\n\t\texpectedDispatched bool\n\t\texpectedStarted    bool\n\t\texpectedError      error\n\t}{\n\t\t{\n\t\t\tname:             \"success\",\n\t\t\tworkflowTaskList: &types.TaskList{Name: \"taskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\tattr: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\tActivityID:                    \"activityID\",\n\t\t\t\tActivityType:                  activityType,\n\t\t\t\tDomain:                        constants.TestDomainName,\n\t\t\t\tTaskList:                      &types.TaskList{Name: \"taskList\"},\n\t\t\t\tInput:                         []byte(\"input\"),\n\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(3),\n\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(4),\n\t\t\t\tRetryPolicy:                   retryPolicy,\n\t\t\t\tHeader:                        header,\n\t\t\t\tRequestLocalDispatch:          false,\n\t\t\t},\n\t\t\texpectedAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\tActivityID:                    \"activityID\",\n\t\t\t\tActivityType:                  activityType,\n\t\t\t\tDomain:                        &constants.TestDomainName,\n\t\t\t\tTaskList:                      &types.TaskList{Name: \"taskList\"},\n\t\t\t\tInput:                         []byte(\"input\"),\n\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(3),\n\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(4),\n\t\t\t\tDecisionTaskCompletedEventID:  0,\n\t\t\t\tRetryPolicy:                   retryPolicy,\n\t\t\t\tHeader:                        header,\n\t\t\t},\n\t\t\texpectedInfo: &persistence.ActivityInfo{\n\t\t\t\tVersion:                commonconstants.EmptyVersion,\n\t\t\t\tScheduleID:             1,\n\t\t\t\tScheduledEventBatchID:  0,\n\t\t\t\tScheduledTime:          currentTime,\n\t\t\t\tStartedID:              commonconstants.EmptyEventID,\n\t\t\t\tDomainID:               constants.TestDomainID,\n\t\t\t\tActivityID:             \"activityID\",\n\t\t\t\tScheduleToCloseTimeout: 1,\n\t\t\t\tScheduleToStartTimeout: 2,\n\t\t\t\tStartToCloseTimeout:    3,\n\t\t\t\tHeartbeatTimeout:       4,\n\t\t\t\tCancelRequestID:        commonconstants.EmptyEventID,\n\t\t\t\tTaskList:               \"taskList\",\n\t\t\t\tTaskListKind:           types.TaskListKindNormal,\n\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\tInitialInterval:        retryPolicy.InitialIntervalInSeconds,\n\t\t\t\tBackoffCoefficient:     retryPolicy.BackoffCoefficient,\n\t\t\t\tMaximumInterval:        retryPolicy.MaximumIntervalInSeconds,\n\t\t\t\tExpirationTime:         currentTime.Add(time.Duration(retryPolicy.ExpirationIntervalInSeconds) * time.Second),\n\t\t\t\tMaximumAttempts:        retryPolicy.MaximumAttempts,\n\t\t\t\tNonRetriableErrors:     retryPolicy.NonRetriableErrorReasons,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:             \"success - normal workflow with ephemeral activity\",\n\t\t\tworkflowTaskList: &types.TaskList{Name: \"taskList\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\tattr: &types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\tActivityID:                    \"activityID\",\n\t\t\t\tActivityType:                  activityType,\n\t\t\t\tDomain:                        constants.TestDomainName,\n\t\t\t\tTaskList:                      &types.TaskList{Name: \"taskList\", Kind: types.TaskListKindEphemeral.Ptr()},\n\t\t\t\tInput:                         []byte(\"input\"),\n\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(3),\n\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(4),\n\t\t\t\tRetryPolicy:                   retryPolicy,\n\t\t\t\tHeader:                        header,\n\t\t\t\tRequestLocalDispatch:          false,\n\t\t\t},\n\t\t\texpectedAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\tActivityID:                    \"activityID\",\n\t\t\t\tActivityType:                  activityType,\n\t\t\t\tDomain:                        &constants.TestDomainName,\n\t\t\t\tTaskList:                      &types.TaskList{Name: \"taskList\", Kind: types.TaskListKindEphemeral.Ptr()},\n\t\t\t\tInput:                         []byte(\"input\"),\n\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(3),\n\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(4),\n\t\t\t\tDecisionTaskCompletedEventID:  0,\n\t\t\t\tRetryPolicy:                   retryPolicy,\n\t\t\t\tHeader:                        header,\n\t\t\t},\n\t\t\texpectedInfo: &persistence.ActivityInfo{\n\t\t\t\tVersion:                commonconstants.EmptyVersion,\n\t\t\t\tScheduleID:             1,\n\t\t\t\tScheduledEventBatchID:  0,\n\t\t\t\tScheduledTime:          currentTime,\n\t\t\t\tStartedID:              commonconstants.EmptyEventID,\n\t\t\t\tDomainID:               constants.TestDomainID,\n\t\t\t\tActivityID:             \"activityID\",\n\t\t\t\tScheduleToCloseTimeout: 1,\n\t\t\t\tScheduleToStartTimeout: 2,\n\t\t\t\tStartToCloseTimeout:    3,\n\t\t\t\tHeartbeatTimeout:       4,\n\t\t\t\tCancelRequestID:        commonconstants.EmptyEventID,\n\t\t\t\tTaskList:               \"taskList\",\n\t\t\t\tTaskListKind:           types.TaskListKindEphemeral,\n\t\t\t\tHasRetryPolicy:         true,\n\t\t\t\tInitialInterval:        retryPolicy.InitialIntervalInSeconds,\n\t\t\t\tBackoffCoefficient:     retryPolicy.BackoffCoefficient,\n\t\t\t\tMaximumInterval:        retryPolicy.MaximumIntervalInSeconds,\n\t\t\t\tExpirationTime:         currentTime.Add(time.Duration(retryPolicy.ExpirationIntervalInSeconds) * time.Second),\n\t\t\t\tMaximumAttempts:        retryPolicy.MaximumAttempts,\n\t\t\t\tNonRetriableErrors:     retryPolicy.NonRetriableErrorReasons,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\n\t\t\tmb := testMutableStateBuilder(t)\n\t\t\tmockEventsCache := mb.shard.GetEventsCache().(*events.MockCache)\n\t\t\tmockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any())\n\t\t\tif tc.workflowTaskList != nil {\n\t\t\t\tmb.executionInfo.TaskList = tc.workflowTaskList.Name\n\t\t\t\tmb.executionInfo.TaskListKind = tc.workflowTaskList.GetKind()\n\t\t\t}\n\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t\t\tdefer cancel()\n\t\t\tevent, activityInfo, dispatchInfo, dispatched, started, err := mb.AddActivityTaskScheduledEvent(ctx, 0, tc.attr, tc.dispatch)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\tassert.ErrorIs(t, err, tc.expectedError)\n\t\t\t\tassert.Nil(t, event)\n\t\t\t\tassert.Nil(t, activityInfo)\n\t\t\t\tassert.Nil(t, dispatchInfo)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, types.EventTypeActivityTaskScheduled.Ptr(), event.EventType)\n\t\t\t\tassert.Equal(t, tc.expectedAttributes, event.ActivityTaskScheduledEventAttributes)\n\t\t\t\tassert.Equal(t, tc.expectedInfo, activityInfo)\n\t\t\t\tassert.Equal(t, tc.expectedDispatch, dispatchInfo)\n\t\t\t}\n\t\t\tassert.Equal(t, tc.expectedDispatched, dispatched)\n\t\t\tassert.Equal(t, tc.expectedStarted, started)\n\t\t})\n\t}\n}\n\nfunc Test__UpdateActivityProgress(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tai := &persistence.ActivityInfo{\n\t\tVersion:    1,\n\t\tScheduleID: 1,\n\t}\n\trequest := &types.RecordActivityTaskHeartbeatRequest{\n\t\tTaskToken: nil,\n\t\tDetails:   []byte{10, 0},\n\t\tIdentity:  \"\",\n\t}\n\tassert.Equal(t, int64(1), ai.Version)\n\tmb.UpdateActivityProgress(ai, request)\n\tassert.Equal(t, commonconstants.EmptyVersion, ai.Version)\n\tassert.Equal(t, request.Details, ai.Details)\n\tassert.Equal(t, ai, mb.updateActivityInfos[ai.ScheduleID])\n\tassert.NotNil(t, mb.syncActivityTasks[ai.ScheduleID])\n}\n\nfunc Test__ReplicateActivityInfo(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tnow := time.Now()\n\tnowUnix := now.UnixNano()\n\trequest := &types.SyncActivityRequest{\n\t\tScheduledID:       1,\n\t\tVersion:           1,\n\t\tScheduledTime:     &nowUnix,\n\t\tLastHeartbeatTime: &nowUnix,\n\t}\n\tai := &persistence.ActivityInfo{}\n\n\terr := mb.ReplicateActivityInfo(request, true)\n\tassert.Error(t, err)\n\tassert.Equal(t, ErrMissingActivityInfo, err)\n\n\tmb.pendingActivityInfoIDs[request.ScheduledID] = ai\n\terr = mb.ReplicateActivityInfo(request, true)\n\tassert.NoError(t, err)\n\tassert.Equal(t, int64(1), ai.Version)\n\tassert.Equal(t, now.UTC(), ai.ScheduledTime.UTC())\n\tassert.Equal(t, request.StartedID, ai.StartedID)\n\tassert.Equal(t, now.UTC(), ai.LastHeartBeatUpdatedTime.UTC())\n}\n\nfunc Test__UpdateActivity(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tai := &persistence.ActivityInfo{ScheduleID: 1}\n\tt.Run(\"error missing activity info\", func(t *testing.T) {\n\t\terr := mb.UpdateActivity(ai)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrMissingActivityInfo, err)\n\t})\n\tt.Run(\"update success\", func(t *testing.T) {\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\terr := mb.UpdateActivity(ai)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, ai, mb.updateActivityInfos[1])\n\t})\n}\n\nfunc Test__GetActivityScheduledEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tai := &persistence.ActivityInfo{\n\t\tScheduleID:     1,\n\t\tScheduledEvent: &types.HistoryEvent{},\n\t}\n\tt.Run(\"error missing activity info\", func(t *testing.T) {\n\t\t_, err := mb.GetActivityScheduledEvent(context.Background(), ai.ScheduleID)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrMissingActivityInfo, err)\n\t})\n\tt.Run(\"scheduled event from activity info\", func(t *testing.T) {\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tresult, err := mb.GetActivityScheduledEvent(context.Background(), ai.ScheduleID)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, ai.ScheduledEvent, result)\n\t})\n\tt.Run(\"scheduled event from events cache\", func(t *testing.T) {\n\t\tmockEventsCache := mb.shard.GetEventsCache().(*events.MockCache)\n\t\tmockEventsCache.EXPECT().GetEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.HistoryEvent{}, nil)\n\t\tmb.pendingActivityInfoIDs[1] = &persistence.ActivityInfo{\n\t\t\tScheduleID: 1,\n\t\t}\n\t\tresult, err := mb.GetActivityScheduledEvent(context.Background(), ai.ScheduleID)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, &types.HistoryEvent{}, result)\n\t})\n\tt.Run(\"error missing scheduled event\", func(t *testing.T) {\n\t\tmockEventsCache := mb.shard.GetEventsCache().(*events.MockCache)\n\t\tmockEventsCache.EXPECT().GetEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\tmb.pendingActivityInfoIDs[1] = &persistence.ActivityInfo{\n\t\t\tScheduleID: 1,\n\t\t}\n\t\t_, err := mb.GetActivityScheduledEvent(context.Background(), ai.ScheduleID)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrMissingActivityScheduledEvent, err)\n\t})\n}\n\nfunc Test__AddActivityTaskCompletedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\trequest := &types.RespondActivityTaskCompletedRequest{\n\t\tTaskToken: nil,\n\t\tResult:    nil,\n\t\tIdentity:  \"\",\n\t}\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddActivityTaskCompletedEvent(1, 1, request)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting activity info\", func(t *testing.T) {\n\t\t_, err := mb.AddActivityTaskCompletedEvent(1, 1, request)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-activitytask-completed-event operation failed\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tai := &persistence.ActivityInfo{\n\t\t\tScheduleID:     1,\n\t\t\tActivityID:     \"1\",\n\t\t\tScheduledEvent: &types.HistoryEvent{},\n\t\t\tStartedID:      1,\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\t\tmb.updateActivityInfos[1] = ai\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddActivityTaskCompletedEvent(1, 1, request)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.ActivityTaskCompletedEventAttributes.ScheduledEventID)\n\t})\n}\n\nfunc Test__tryDispatchActivityTask(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{}\n\tai := &persistence.ActivityInfo{}\n\tresult := mb.tryDispatchActivityTask(context.Background(), event, ai)\n\tassert.True(t, result)\n}\n\nfunc Test__ReplicateActivityTaskCanceledEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{\n\t\t\tScheduledEventID: 1,\n\t\t},\n\t}\n\tai := &persistence.ActivityInfo{\n\t\tActivityID: \"1\",\n\t}\n\tmb.pendingActivityInfoIDs[1] = ai\n\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\terr := mb.ReplicateActivityTaskCanceledEvent(event)\n\tassert.NoError(t, err)\n\t_, ok := mb.pendingActivityInfoIDs[1]\n\tassert.False(t, ok)\n\t_, ok = mb.pendingActivityIDToEventID[\"1\"]\n\tassert.False(t, ok)\n}\n\nfunc Test__ReplicateActivityTaskCancelRequestedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\tActivityTaskCancelRequestedEventAttributes: &types.ActivityTaskCancelRequestedEventAttributes{\n\t\t\tActivityID: \"1\",\n\t\t},\n\t}\n\tai := &persistence.ActivityInfo{\n\t\tActivityID: \"1\",\n\t\tScheduleID: 1,\n\t}\n\tmb.pendingActivityInfoIDs[1] = ai\n\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\terr := mb.ReplicateActivityTaskCancelRequestedEvent(event)\n\tassert.NoError(t, err)\n\tassert.Equal(t, ai, mb.updateActivityInfos[1])\n}\n\nfunc Test__ReplicateActivityTaskTimedOutEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\tScheduledEventID: 1,\n\t\t},\n\t}\n\tai := &persistence.ActivityInfo{\n\t\tActivityID: \"1\",\n\t}\n\tmb.pendingActivityInfoIDs[1] = ai\n\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\terr := mb.ReplicateActivityTaskTimedOutEvent(event)\n\tassert.NoError(t, err)\n\t_, ok := mb.pendingActivityInfoIDs[1]\n\tassert.False(t, ok)\n\t_, ok = mb.pendingActivityIDToEventID[\"1\"]\n\tassert.False(t, ok)\n}\n\nfunc Test__ReplicateActivityTaskFailedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeActivityTaskFailed.Ptr(),\n\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\tScheduledEventID: 1,\n\t\t},\n\t}\n\tai := &persistence.ActivityInfo{\n\t\tActivityID: \"1\",\n\t}\n\tmb.pendingActivityInfoIDs[1] = ai\n\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\terr := mb.ReplicateActivityTaskFailedEvent(event)\n\tassert.NoError(t, err)\n\t_, ok := mb.pendingActivityInfoIDs[1]\n\tassert.False(t, ok)\n\t_, ok = mb.pendingActivityIDToEventID[\"1\"]\n\tassert.False(t, ok)\n}\n\nfunc Test__ReplicateActivityTaskCompletedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\tScheduledEventID: 1,\n\t\t},\n\t}\n\tai := &persistence.ActivityInfo{\n\t\tActivityID: \"1\",\n\t}\n\tmb.pendingActivityInfoIDs[1] = ai\n\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\terr := mb.ReplicateActivityTaskCompletedEvent(event)\n\tassert.NoError(t, err)\n\t_, ok := mb.pendingActivityInfoIDs[1]\n\tassert.False(t, ok)\n\t_, ok = mb.pendingActivityIDToEventID[\"1\"]\n\tassert.False(t, ok)\n}\n\nfunc Test__AddActivityTaskCanceledEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddActivityTaskCanceledEvent(1, 1, 1, []byte{10}, \"test\")\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting activity info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, err := mb.AddActivityTaskCanceledEvent(1, 1, 1, []byte{10}, \"test\")\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-activitytask-canceled-event operation failed\", err.Error())\n\t})\n\tt.Run(\"error cancel not requested\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tai := &persistence.ActivityInfo{\n\t\t\tStartedID:       1,\n\t\t\tCancelRequested: false,\n\t\t\tStartedTime:     time.Now(),\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\t_, err := mb.AddActivityTaskCanceledEvent(1, 1, 1, []byte{10}, \"test\")\n\t\tassert.Error(t, err)\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tai := &persistence.ActivityInfo{\n\t\t\tStartedID:       1,\n\t\t\tCancelRequested: true,\n\t\t\tStartedTime:     time.Now(),\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tevent, err := mb.AddActivityTaskCanceledEvent(1, 1, 1, []byte{10}, \"test\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.ActivityTaskCanceledEventAttributes.ScheduledEventID)\n\t\tassert.Equal(t, \"test\", event.ActivityTaskCanceledEventAttributes.Identity)\n\t})\n}\n\nfunc Test__AddRequestCancelActivityTaskFailedEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddRequestCancelActivityTaskFailedEvent(1, \"1\", \"test\")\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddRequestCancelActivityTaskFailedEvent(1, \"1\", \"test\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.RequestCancelActivityTaskFailedEventAttributes.DecisionTaskCompletedEventID)\n\t\tassert.Equal(t, \"test\", event.RequestCancelActivityTaskFailedEventAttributes.Cause)\n\t})\n}\n\nfunc Test__AddActivityTaskCancelRequestedEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, _, err := mbCompleted.AddActivityTaskCancelRequestedEvent(1, \"1\", \"test\")\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting activity info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\t_, _, err := mb.AddActivityTaskCancelRequestedEvent(1, \"1\", \"test\")\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"invalid history builder state for action: add-activitytask-cancel-requested-event\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tai := &persistence.ActivityInfo{\n\t\t\tStartedID:   1,\n\t\t\tStartedTime: time.Now(),\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tmb.pendingActivityIDToEventID[\"1\"] = 1\n\t\tevent, ai, err := mb.AddActivityTaskCancelRequestedEvent(1, \"1\", \"test\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, \"1\", event.ActivityTaskCancelRequestedEventAttributes.ActivityID)\n\t})\n}\n\nfunc Test__AddActivityTaskTimedOutEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddActivityTaskTimedOutEvent(1, 1, types.TimeoutTypeHeartbeat, []byte{10})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting activity info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, err := mb.AddActivityTaskTimedOutEvent(1, 1, types.TimeoutTypeHeartbeat, []byte{10})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-activitytask-timed-event operation failed\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tai := &persistence.ActivityInfo{\n\t\t\tScheduleID:     1,\n\t\t\tActivityID:     \"1\",\n\t\t\tScheduledEvent: &types.HistoryEvent{},\n\t\t\tStartedID:      1,\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddActivityTaskTimedOutEvent(1, 1, types.TimeoutTypeHeartbeat, []byte{10})\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.ActivityTaskTimedOutEventAttributes.ScheduledEventID)\n\t})\n}\n\nfunc Test__AddActivityTaskFailedEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddActivityTaskFailedEvent(1, 1, &types.RespondActivityTaskFailedRequest{})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting activity info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, err := mb.AddActivityTaskFailedEvent(1, 1, &types.RespondActivityTaskFailedRequest{})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-activitytask-failed-event operation failed\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tai := &persistence.ActivityInfo{\n\t\t\tScheduleID:     1,\n\t\t\tActivityID:     \"1\",\n\t\t\tScheduledEvent: &types.HistoryEvent{},\n\t\t\tStartedID:      1,\n\t\t}\n\t\tmb.pendingActivityInfoIDs[1] = ai\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddActivityTaskFailedEvent(1, 1, &types.RespondActivityTaskFailedRequest{\n\t\t\tIdentity: \"test\",\n\t\t\tDetails:  make([]byte, 10),\n\t\t})\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.ActivityTaskFailedEventAttributes.ScheduledEventID)\n\n\t})\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_cancellation.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) IsCancelRequested() (bool, string) {\n\tif e.executionInfo.CancelRequested {\n\t\treturn e.executionInfo.CancelRequested, e.executionInfo.CancelRequestID\n\t}\n\n\treturn false, \"\"\n}\n\n// GetRequestCancelInfo gives details about a request cancellation that is currently in progress.\nfunc (e *mutableStateBuilder) GetRequestCancelInfo(\n\tinitiatedEventID int64,\n) (*persistence.RequestCancelInfo, bool) {\n\n\tri, ok := e.pendingRequestCancelInfoIDs[initiatedEventID]\n\treturn ri, ok\n}\n\nfunc (e *mutableStateBuilder) AddWorkflowExecutionCancelRequestedEvent(\n\tcause string,\n\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowCancelRequested\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif e.executionInfo.CancelRequested {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowState(e.executionInfo.State),\n\t\t\ttag.Bool(e.executionInfo.CancelRequested),\n\t\t\ttag.Key(e.executionInfo.CancelRequestID),\n\t\t)\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionCancelRequestedEvent(cause, request)\n\tif err := e.ReplicateWorkflowExecutionCancelRequestedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionCancelRequestedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\te.executionInfo.CancelRequested = true\n\trequestID := event.WorkflowExecutionCancelRequestedEventAttributes.RequestID\n\te.executionInfo.CancelRequestID = requestID\n\te.insertWorkflowRequest(persistence.WorkflowRequest{\n\t\tRequestID:   requestID,\n\t\tVersion:     event.Version,\n\t\tRequestType: persistence.WorkflowRequestTypeCancel,\n\t})\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddWorkflowExecutionCanceledEvent(\n\tdecisionTaskCompletedEventID int64,\n\tattributes *types.CancelWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowCanceled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionCanceledEvent(decisionTaskCompletedEventID, attributes)\n\tif err := e.ReplicateWorkflowExecutionCanceledEvent(decisionTaskCompletedEventID, event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionCanceledEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n) error {\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusCanceled,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(event)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(event, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n\nfunc (e *mutableStateBuilder) AddRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\tdecisionCompletedEventID int64,\n\tcancelRequestID string,\n\trequest *types.RequestCancelExternalWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, *persistence.RequestCancelInfo, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowCancelInitiated\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tevent := e.hBuilder.AddRequestCancelExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID, request)\n\trci, err := e.ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID, event, cancelRequestID)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn event, rci, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n\tcancelRequestID string,\n) (*persistence.RequestCancelInfo, error) {\n\n\t// TODO: Evaluate if we need cancelRequestID also part of history event\n\tinitiatedEventID := event.ID\n\trci := &persistence.RequestCancelInfo{\n\t\tVersion:               event.Version,\n\t\tInitiatedEventBatchID: firstEventID,\n\t\tInitiatedID:           initiatedEventID,\n\t\tCancelRequestID:       cancelRequestID,\n\t}\n\n\te.pendingRequestCancelInfoIDs[rci.InitiatedID] = rci\n\te.updateRequestCancelInfos[rci.InitiatedID] = rci\n\n\treturn rci, e.taskGenerator.GenerateRequestCancelExternalTasks(event)\n}\n\nfunc (e *mutableStateBuilder) AddExternalWorkflowExecutionCancelRequested(\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowCancelRequested\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\t_, ok := e.GetRequestCancelInfo(initiatedID)\n\tif !ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddExternalWorkflowExecutionCancelRequested(initiatedID, domain, workflowID, runID)\n\tif err := e.ReplicateExternalWorkflowExecutionCancelRequested(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateExternalWorkflowExecutionCancelRequested(\n\tevent *types.HistoryEvent,\n) error {\n\n\tinitiatedID := event.ExternalWorkflowExecutionCancelRequestedEventAttributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingRequestCancel(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddRequestCancelExternalWorkflowExecutionFailedEvent(\n\tdecisionTaskCompletedEventID int64,\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tcause types.CancelExternalWorkflowExecutionFailedCause,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowCancelFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\t_, ok := e.GetRequestCancelInfo(initiatedID)\n\tif !ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddRequestCancelExternalWorkflowExecutionFailedEvent(decisionTaskCompletedEventID, initiatedID,\n\t\tdomain, workflowID, runID, cause)\n\tif err := e.ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tinitiatedID := event.RequestCancelExternalWorkflowExecutionFailedEventAttributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingRequestCancel(initiatedID)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_child_workflow.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// GetChildExecutionInfo gives details about a child execution that is currently in progress.\nfunc (e *mutableStateBuilder) GetChildExecutionInfo(\n\tinitiatedEventID int64,\n) (*persistence.ChildExecutionInfo, bool) {\n\n\tci, ok := e.pendingChildExecutionInfoIDs[initiatedEventID]\n\treturn ci, ok\n}\n\n// GetChildExecutionInitiatedEvent reads out the ChildExecutionInitiatedEvent from mutable state for in-progress child\n// executions\nfunc (e *mutableStateBuilder) GetChildExecutionInitiatedEvent(\n\tctx context.Context,\n\tinitiatedEventID int64,\n) (*types.HistoryEvent, error) {\n\n\tci, ok := e.pendingChildExecutionInfoIDs[initiatedEventID]\n\tif !ok {\n\t\treturn nil, ErrMissingChildWorkflowInfo\n\t}\n\n\t// Needed for backward compatibility reason\n\tif ci.InitiatedEvent != nil {\n\t\treturn ci.InitiatedEvent, nil\n\t}\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tinitiatedEvent, err := e.eventsCache.GetEvent(\n\t\tctx,\n\t\te.shard.GetShardID(),\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tci.InitiatedEventBatchID,\n\t\tci.InitiatedID,\n\t\tcurrentBranchToken,\n\t)\n\tif err != nil {\n\t\t// do not return the original error\n\t\t// since original error can be of type entity not exists\n\t\t// which can cause task processing side to fail silently\n\t\t// However, if the error is a persistence transient error,\n\t\t// we return the original error, because we fail to get\n\t\t// the event because of failure from database\n\t\tif persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, ErrMissingChildWorkflowInitiatedEvent\n\t}\n\treturn initiatedEvent, nil\n}\n\nfunc (e *mutableStateBuilder) GetPendingChildExecutionInfos() map[int64]*persistence.ChildExecutionInfo {\n\treturn e.pendingChildExecutionInfoIDs\n}\n\n// DeletePendingChildExecution deletes details about a ChildExecutionInfo.\nfunc (e *mutableStateBuilder) DeletePendingChildExecution(\n\tinitiatedEventID int64,\n) error {\n\n\tif _, ok := e.pendingChildExecutionInfoIDs[initiatedEventID]; ok {\n\t\tdelete(e.pendingChildExecutionInfoIDs, initiatedEventID)\n\t} else {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find child workflow event ID: %v in mutable state\", initiatedEventID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\t// log data inconsistency instead of returning an error\n\t\te.logDataInconsistency()\n\t}\n\n\tdelete(e.updateChildExecutionInfos, initiatedEventID)\n\te.deleteChildExecutionInfos[initiatedEventID] = struct{}{}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddStartChildWorkflowExecutionInitiatedEvent(\n\tdecisionCompletedEventID int64,\n\tcreateRequestID string,\n\tattributes *types.StartChildWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, *persistence.ChildExecutionInfo, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowInitiated\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tevent := e.hBuilder.AddStartChildWorkflowExecutionInitiatedEvent(\n\t\tdecisionCompletedEventID,\n\t\tattributes,\n\t\te.GetDomainEntry().GetInfo().Name,\n\t)\n\t// Write the event to cache only on active cluster\n\te.eventsCache.PutEvent(e.executionInfo.DomainID, e.executionInfo.WorkflowID, e.executionInfo.RunID,\n\t\tevent.ID, event)\n\n\tci, err := e.ReplicateStartChildWorkflowExecutionInitiatedEvent(decisionCompletedEventID, event, createRequestID)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn event, ci, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateStartChildWorkflowExecutionInitiatedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n\tcreateRequestID string,\n) (*persistence.ChildExecutionInfo, error) {\n\n\tinitiatedEventID := event.ID\n\tattributes := event.StartChildWorkflowExecutionInitiatedEventAttributes\n\n\tdomainID := e.GetExecutionInfo().DomainID\n\tif domainName := attributes.GetDomain(); domainName != \"\" {\n\t\t// domainName may still be empty if two cadence clusters are running different versions\n\t\tvar err error\n\t\tdomainID, err = e.shard.GetDomainCache().GetDomainID(domainName)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tci := &persistence.ChildExecutionInfo{\n\t\tVersion:               event.Version,\n\t\tInitiatedID:           initiatedEventID,\n\t\tInitiatedEventBatchID: firstEventID,\n\t\tStartedID:             constants.EmptyEventID,\n\t\tStartedWorkflowID:     attributes.GetWorkflowID(),\n\t\tCreateRequestID:       createRequestID,\n\t\tDomainID:              domainID,\n\t\t// DomainName field is being deprecated\n\t\t// DomainName:            attributes.GetDomain(),\n\t\tWorkflowTypeName:  attributes.GetWorkflowType().GetName(),\n\t\tParentClosePolicy: attributes.GetParentClosePolicy(),\n\t}\n\n\te.pendingChildExecutionInfoIDs[ci.InitiatedID] = ci\n\te.updateChildExecutionInfos[ci.InitiatedID] = ci\n\n\treturn ci, e.taskGenerator.GenerateChildWorkflowTasks(event)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionStartedEvent(\n\tdomain string,\n\texecution *types.WorkflowExecution,\n\tworkflowType *types.WorkflowType,\n\tinitiatedID int64,\n\theader *types.Header,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowStarted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID != constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionStartedEvent(domain, execution, workflowType, initiatedID, header)\n\tif err := e.ReplicateChildWorkflowExecutionStartedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionStartedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionStartedEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok {\n\t\te.logError(\n\t\t\t\"Unable to find child workflow\",\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.WorkflowInitiatedID(initiatedID),\n\t\t)\n\t\treturn ErrMissingChildWorkflowInfo\n\t}\n\tci.StartedID = event.ID\n\tci.StartedRunID = attributes.GetWorkflowExecution().GetRunID()\n\te.updateChildExecutionInfos[ci.InitiatedID] = ci\n\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddStartChildWorkflowExecutionFailedEvent(\n\tinitiatedID int64,\n\tcause types.ChildWorkflowExecutionFailedCause,\n\tinitiatedEventAttributes *types.StartChildWorkflowExecutionInitiatedEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowInitiationFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID != constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddStartChildWorkflowExecutionFailedEvent(initiatedID, cause, initiatedEventAttributes)\n\tif err := e.ReplicateStartChildWorkflowExecutionFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateStartChildWorkflowExecutionFailedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.StartChildWorkflowExecutionFailedEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionCompletedEvent(\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionCompletedEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowCompleted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID == constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tchildDomainName, err := GetChildExecutionDomainName(ci, e.shard.GetDomainCache(), e.GetDomainEntry())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowType := &types.WorkflowType{\n\t\tName: ci.WorkflowTypeName,\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionCompletedEvent(\n\t\tchildDomainName,\n\t\tchildExecution,\n\t\tworkflowType,\n\t\tci.InitiatedID,\n\t\tci.StartedID,\n\t\tattributes,\n\t)\n\tif err := e.ReplicateChildWorkflowExecutionCompletedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionCompletedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionCompletedEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionFailedEvent(\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionFailedEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID == constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(!ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tdomainName, err := GetChildExecutionDomainName(ci, e.shard.GetDomainCache(), e.GetDomainEntry())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowType := &types.WorkflowType{\n\t\tName: ci.WorkflowTypeName,\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionFailedEvent(\n\t\tdomainName,\n\t\tchildExecution,\n\t\tworkflowType,\n\t\tci.InitiatedID,\n\t\tci.StartedID,\n\t\tattributes,\n\t)\n\tif err := e.ReplicateChildWorkflowExecutionFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionFailedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionFailedEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionCanceledEvent(\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionCanceledEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowCanceled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID == constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tdomainName, err := GetChildExecutionDomainName(ci, e.shard.GetDomainCache(), e.GetDomainEntry())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowType := &types.WorkflowType{\n\t\tName: ci.WorkflowTypeName,\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionCanceledEvent(\n\t\tdomainName,\n\t\tchildExecution,\n\t\tworkflowType,\n\t\tci.InitiatedID,\n\t\tci.StartedID,\n\t\tattributes,\n\t)\n\tif err := e.ReplicateChildWorkflowExecutionCanceledEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionCanceledEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionCanceledEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionTerminatedEvent(\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionTerminatedEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowTerminated\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID == constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tdomainName, err := GetChildExecutionDomainName(ci, e.shard.GetDomainCache(), e.GetDomainEntry())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowType := &types.WorkflowType{\n\t\tName: ci.WorkflowTypeName,\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionTerminatedEvent(\n\t\tdomainName,\n\t\tchildExecution,\n\t\tworkflowType,\n\t\tci.InitiatedID,\n\t\tci.StartedID,\n\t\tattributes,\n\t)\n\tif err := e.ReplicateChildWorkflowExecutionTerminatedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionTerminatedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionTerminatedEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddChildWorkflowExecutionTimedOutEvent(\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionTimedOutEventAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionChildWorkflowTimedOut\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tci, ok := e.GetChildExecutionInfo(initiatedID)\n\tif !ok || ci.StartedID == constants.EmptyEventID {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.Bool(ok),\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tdomainName, err := GetChildExecutionDomainName(ci, e.shard.GetDomainCache(), e.GetDomainEntry())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tworkflowType := &types.WorkflowType{\n\t\tName: ci.WorkflowTypeName,\n\t}\n\n\tevent := e.hBuilder.AddChildWorkflowExecutionTimedOutEvent(\n\t\tdomainName,\n\t\tchildExecution,\n\t\tworkflowType,\n\t\tci.InitiatedID,\n\t\tci.StartedID,\n\t\tattributes,\n\t)\n\tif err := e.ReplicateChildWorkflowExecutionTimedOutEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateChildWorkflowExecutionTimedOutEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.ChildWorkflowExecutionTimedOutEventAttributes\n\tinitiatedID := attributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingChildExecution(initiatedID)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_child_workflow_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nvar (\n\tcurrentBranchToken = []byte(\"branchToken\")\n\ttestShardID        = 42\n)\n\nfunc TestGetChildExecutionInfo(t *testing.T) {\n\tchildInfo := &persistence.ChildExecutionInfo{\n\t\tDomainID: constants.TestDomainID,\n\t}\n\tctx := createShardCtx(t)\n\tm := loadMutableState(t, ctx, &persistence.WorkflowMutableState{\n\t\tExecutionInfo: standardExecutionInfo(),\n\t\tChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t1: childInfo,\n\t\t}})\n\n\tresult, ok := m.GetChildExecutionInfo(1)\n\n\tassert.True(t, ok)\n\tassert.Equal(t, result, childInfo)\n}\n\nfunc TestGetChildExecutionInitiatedEvent(t *testing.T) {\n\tinitiatedEventID := int64(1)\n\tsampleEvent := &types.HistoryEvent{\n\t\tID: 2,\n\t}\n\tcases := []struct {\n\t\tname                 string\n\t\teventCacheAffordance func(e *events.MockCache)\n\t\tchildInfo            *persistence.ChildExecutionInfo\n\t\texpected             *types.HistoryEvent\n\t\texpectedErr          error\n\t}{\n\t\t{\n\t\t\tname:        \"Missing Child Workflow Info\",\n\t\t\texpectedErr: ErrMissingChildWorkflowInfo,\n\t\t},\n\t\t{\n\t\t\tname: \"Event stored on info\",\n\t\t\tchildInfo: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedEvent: sampleEvent,\n\t\t\t},\n\t\t\texpected: sampleEvent,\n\t\t},\n\t\t{\n\t\t\tname: \"Get from EventCache\",\n\t\t\tchildInfo: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedEventBatchID: 10,\n\t\t\t\tInitiatedID:           11,\n\t\t\t},\n\t\t\teventCacheAffordance: func(e *events.MockCache) {\n\t\t\t\te.EXPECT().GetEvent(\n\t\t\t\t\tgomock.Any(), // context\n\t\t\t\t\ttestShardID,\n\t\t\t\t\tconstants.TestDomainID,\n\t\t\t\t\tconstants.TestWorkflowID,\n\t\t\t\t\tconstants.TestRunID,\n\t\t\t\t\tint64(10), // InitiatedEventBatchID\n\t\t\t\t\tint64(11), // InitiatedID\n\t\t\t\t\tcurrentBranchToken,\n\t\t\t\t).Return(sampleEvent, nil)\n\t\t\t},\n\t\t\texpected: sampleEvent,\n\t\t},\n\t\t{\n\t\t\tname: \"Error from EventCache\",\n\t\t\tchildInfo: &persistence.ChildExecutionInfo{\n\t\t\t\tInitiatedEventBatchID: 10,\n\t\t\t\tInitiatedID:           11,\n\t\t\t},\n\t\t\teventCacheAffordance: func(e *events.MockCache) {\n\t\t\t\te.EXPECT().GetEvent(\n\t\t\t\t\tgomock.Any(), // context\n\t\t\t\t\ttestShardID,\n\t\t\t\t\tconstants.TestDomainID,\n\t\t\t\t\tconstants.TestWorkflowID,\n\t\t\t\t\tconstants.TestRunID,\n\t\t\t\t\tint64(10), // InitiatedEventBatchID\n\t\t\t\t\tint64(11), // InitiatedID\n\t\t\t\t\tcurrentBranchToken,\n\t\t\t\t).Return(nil, &types.AccessDeniedError{Message: \"Oh no!\"})\n\t\t\t},\n\t\t\texpectedErr: ErrMissingChildWorkflowInitiatedEvent,\n\t\t},\n\t}\n\n\tfor _, testCase := range cases {\n\t\tt.Run(testCase.name, func(t *testing.T) {\n\t\t\tctx := createShardCtx(t)\n\t\t\tif testCase.eventCacheAffordance != nil {\n\t\t\t\ttestCase.eventCacheAffordance(ctx.MockEventsCache)\n\t\t\t}\n\t\t\tchildExecutionInfo := map[int64]*persistence.ChildExecutionInfo{}\n\t\t\tif testCase.childInfo != nil {\n\t\t\t\tchildExecutionInfo[initiatedEventID] = testCase.childInfo\n\t\t\t}\n\t\t\tm := loadMutableState(t, ctx, &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo:       standardExecutionInfo(),\n\t\t\t\tChildExecutionInfos: childExecutionInfo,\n\t\t\t})\n\n\t\t\tresult, err := m.GetChildExecutionInitiatedEvent(context.Background(), initiatedEventID)\n\t\t\tassert.Equal(t, testCase.expected, result)\n\t\t\tassert.Equal(t, testCase.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestGetPendingChildExecutionInfos(t *testing.T) {\n\tchildExecutionInfos := map[int64]*persistence.ChildExecutionInfo{\n\t\t1: {\n\t\t\tDomainID: constants.TestDomainID,\n\t\t},\n\t}\n\tctx := createShardCtx(t)\n\n\tm := loadMutableState(t, ctx, &persistence.WorkflowMutableState{\n\t\tExecutionInfo:       standardExecutionInfo(),\n\t\tChildExecutionInfos: childExecutionInfos,\n\t})\n\n\tassert.Equal(t, childExecutionInfos, m.GetPendingChildExecutionInfos())\n}\n\nfunc TestDeletePendingChildExecution(t *testing.T) {\n\tcases := []struct {\n\t\tname      string\n\t\tchildInfo map[int64]*persistence.ChildExecutionInfo\n\t\ttoDelete  int64\n\t}{\n\t\t{\n\t\t\tname: \"DeletePresent\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t1: {\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttoDelete: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"DeleteNotFound\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t1: {\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttoDelete: 2,\n\t\t},\n\t}\n\tfor _, testCase := range cases {\n\t\tt.Run(testCase.name, func(t *testing.T) {\n\t\t\tctx := createShardCtx(t)\n\t\t\tm := loadMutableState(t, ctx, &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo:       standardExecutionInfo(),\n\t\t\t\tChildExecutionInfos: testCase.childInfo,\n\t\t\t})\n\n\t\t\terr := m.DeletePendingChildExecution(testCase.toDelete)\n\n\t\t\t// Even the failure case doesn't return an error\n\t\t\tassert.NoError(t, err)\n\t\t\t_, hasChild := m.GetPendingChildExecutionInfos()[testCase.toDelete]\n\t\t\t_, deletedChild := m.deleteChildExecutionInfos[testCase.toDelete]\n\n\t\t\tassert.Equal(t, false, hasChild)\n\t\t\tassert.Equal(t, true, deletedChild)\n\t\t})\n\t}\n}\n\nfunc TestAddChildEvents(t *testing.T) {\n\tparentWfID := \"parent\"\n\tparentRunID := \"1d00698f-08e1-4d36-a3e2-3bf109f5d2d6\"\n\tdecisionID := int64(0)\n\trequestID := \"request\"\n\tdecisionAttributes := &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:                              constants.TestDomainName,\n\t\tWorkflowID:                          constants.TestWorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: \"type\"},\n\t\tTaskList:                            &types.TaskList{},\n\t\tInput:                               []byte(\"input\"),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(2),\n\t\tParentClosePolicy:                   types.ParentClosePolicyAbandon.Ptr(),\n\t\tControl:                             []byte(\"control\"),\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\tRetryPolicy:                         nil,\n\t\tCronSchedule:                        \"\",\n\t\tHeader:                              &types.Header{Fields: map[string][]byte{\"headerKey\": []byte(\"headerValue\")}},\n\t\tMemo:                                &types.Memo{Fields: map[string][]byte{\"key\": []byte(\"value\")}},\n\t\tSearchAttributes:                    &types.SearchAttributes{IndexedFields: map[string][]byte{\"indexed\": []byte(\"field\")}},\n\t}\n\tinitiatedAttributes := &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\tDomain:                              decisionAttributes.Domain,\n\t\tWorkflowID:                          decisionAttributes.WorkflowID,\n\t\tWorkflowType:                        decisionAttributes.WorkflowType,\n\t\tTaskList:                            decisionAttributes.TaskList,\n\t\tInput:                               decisionAttributes.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: decisionAttributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      decisionAttributes.TaskStartToCloseTimeoutSeconds,\n\t\tParentClosePolicy:                   decisionAttributes.ParentClosePolicy,\n\t\tControl:                             decisionAttributes.Control,\n\t\tDecisionTaskCompletedEventID:        decisionID,\n\t\tWorkflowIDReusePolicy:               decisionAttributes.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         decisionAttributes.RetryPolicy,\n\t\tCronSchedule:                        decisionAttributes.CronSchedule,\n\t\tHeader:                              decisionAttributes.Header,\n\t\tMemo:                                decisionAttributes.Memo,\n\t\tSearchAttributes:                    decisionAttributes.SearchAttributes,\n\t}\n\tchildInitiatedID := int64(1)\n\tchildStartedID := int64(3)\n\tnextEventID := int64(2)\n\tcurrentTimestamp := int64(1_000_001)\n\tchildExecution := &types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\tchildType := &types.WorkflowType{Name: \"type\"}\n\n\tcases := []struct {\n\t\tname           string\n\t\tchildInfo      map[int64]*persistence.ChildExecutionInfo\n\t\twfState        int\n\t\tmockAllowances func(ctx *shard.TestContext)\n\t\teventFn        func(builder *mutableStateBuilder) (*types.HistoryEvent, error)\n\t\tassertFn       func(t *testing.T, builder *mutableStateBuilder)\n\t\texpectedEvent  *types.HistoryEvent\n\t\texpectedErr    error\n\t}{\n\t\t{\n\t\t\tname: \"StartChildWorkflowExecutionInitiated\",\n\t\t\tmockAllowances: func(ctx *shard.TestContext) {\n\t\t\t\tctx.MockEventsCache.EXPECT().PutEvent(constants.TestDomainID, parentWfID, parentRunID, nextEventID, gomock.Any())\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\tevent, _, e := builder.AddStartChildWorkflowExecutionInitiatedEvent(decisionID, requestID, decisionAttributes)\n\t\t\t\treturn event, e\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        nextEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeStartChildWorkflowExecutionInitiated.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: initiatedAttributes,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"StartChildWorkflowExecutionInitiated: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\tevent, _, e := builder.AddStartChildWorkflowExecutionInitiatedEvent(decisionID, requestID, decisionAttributes)\n\t\t\t\treturn event, e\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionStarted\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tStartedID: commonconstants.EmptyEventID,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionStartedEvent(constants.TestDomainID, &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}, &types.WorkflowType{Name: \"type\"}, childInitiatedID, nil)\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tDomain:            constants.TestDomainID,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tHeader:            nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionStarted: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionStartedEvent(constants.TestDomainID, &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t}, &types.WorkflowType{Name: \"type\"}, childInitiatedID, nil)\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionStarted: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionStartedEvent(constants.TestDomainID, childExecution, childType, childInitiatedID, nil)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowStarted.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"StartChildWorkflowExecutionFailed\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tStartedID: commonconstants.EmptyEventID,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddStartChildWorkflowExecutionFailedEvent(childInitiatedID, types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning, initiatedAttributes)\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeStartChildWorkflowExecutionFailed.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tStartChildWorkflowExecutionFailedEventAttributes: &types.StartChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tDomain:                       constants.TestDomainName,\n\t\t\t\t\tWorkflowID:                   constants.TestWorkflowID,\n\t\t\t\t\tWorkflowType:                 childType,\n\t\t\t\t\tCause:                        types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning.Ptr(),\n\t\t\t\t\tControl:                      []byte(\"control\"),\n\t\t\t\t\tInitiatedEventID:             childInitiatedID,\n\t\t\t\t\tDecisionTaskCompletedEventID: decisionID,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"StartChildWorkflowExecutionFailed: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddStartChildWorkflowExecutionFailedEvent(childInitiatedID, types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning, initiatedAttributes)\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"StartChildWorkflowExecutionFailed: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddStartChildWorkflowExecutionFailedEvent(childInitiatedID, types.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning, initiatedAttributes)\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowInitiationFailed.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionCompleted\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tInitiatedID:      childInitiatedID,\n\t\t\t\t\tStartedID:        childStartedID,\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowTypeName: childType.Name,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCompletedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult:                       []byte(\"result\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCompleted.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult:            []byte(\"result\"),\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tStartedEventID:    3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionCompleted: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCompletedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult:                       []byte(\"result\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname:      \"ChildWorkflowExecutionCompleted: missing child info\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCompletedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult:                       []byte(\"result\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowCompleted.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionFailed\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tInitiatedID:      childInitiatedID,\n\t\t\t\t\tStartedID:        childStartedID,\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowTypeName: childType.Name,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionFailedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tReason:                       common.StringPtr(\"failed\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionFailed.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tReason:            common.StringPtr(\"failed\"),\n\t\t\t\t\tDetails:           []byte(\"details\"),\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tStartedEventID:    childStartedID,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionFailed: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionFailedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tReason:                       common.StringPtr(\"failed\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionFailed: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionFailedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tReason:                       common.StringPtr(\"failed\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowFailed.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionCanceled\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tInitiatedID:      childInitiatedID,\n\t\t\t\t\tStartedID:        childStartedID,\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowTypeName: childType.Name,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCanceledEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCanceled.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails:           []byte(\"details\"),\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tStartedEventID:    childStartedID,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionCanceled: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCanceledEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionCanceled: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionCanceledEvent(childInitiatedID, childExecution, &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails:                      []byte(\"details\"),\n\t\t\t\t\tDecisionTaskCompletedEventID: 10, // Unused\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowCanceled.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTerminated\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tInitiatedID:      childInitiatedID,\n\t\t\t\t\tStartedID:        childStartedID,\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowTypeName: childType.Name,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\t// Attributes are entirely unused\n\t\t\t\treturn builder.AddChildWorkflowExecutionTerminatedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tReason:   \"reason\",\n\t\t\t\t\tDetails:  []byte(\"details\"),\n\t\t\t\t\tIdentity: \"identity\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTerminated.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionTerminatedEventAttributes: &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tStartedEventID:    childStartedID,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionTerminated: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\t// Attributes are entirely unused\n\t\t\t\treturn builder.AddChildWorkflowExecutionTerminatedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tReason:   \"reason\",\n\t\t\t\t\tDetails:  []byte(\"details\"),\n\t\t\t\t\tIdentity: \"identity\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTerminated: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\t// Attributes are entirely unused\n\t\t\t\treturn builder.AddChildWorkflowExecutionTerminatedEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tReason:   \"reason\",\n\t\t\t\t\tDetails:  []byte(\"details\"),\n\t\t\t\t\tIdentity: \"identity\",\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowTerminated.Field().String + \" operation failed\"},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTimedOut\",\n\t\t\tchildInfo: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\tchildInitiatedID: {\n\t\t\t\t\tInitiatedID:      childInitiatedID,\n\t\t\t\t\tStartedID:        childStartedID,\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowTypeName: childType.Name,\n\t\t\t\t},\n\t\t\t},\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionTimedOutEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\tTimestamp: &currentTimestamp,\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tDomain:            constants.TestDomainName,\n\t\t\t\t\tWorkflowExecution: childExecution,\n\t\t\t\t\tWorkflowType:      childType,\n\t\t\t\t\tInitiatedEventID:  childInitiatedID,\n\t\t\t\t\tStartedEventID:    childStartedID,\n\t\t\t\t\tTimeoutType:       types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"ChildWorkflowExecutionTimedOut: not mutable\",\n\t\t\twfState: persistence.WorkflowStateCompleted,\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionTimedOutEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: ErrWorkflowFinished,\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTimedOut: missing child info\",\n\t\t\teventFn: func(builder *mutableStateBuilder) (*types.HistoryEvent, error) {\n\t\t\t\treturn builder.AddChildWorkflowExecutionTimedOutEvent(childInitiatedID, childExecution, &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedErr: &types.InternalServiceError{Message: tag.WorkflowActionChildWorkflowTimedOut.Field().String + \" operation failed\"},\n\t\t},\n\t}\n\tfor _, testCase := range cases {\n\t\tt.Run(testCase.name, func(t *testing.T) {\n\t\t\tctx := createShardCtx(t)\n\t\t\tctx.Resource.TimeSource = clock.NewMockedTimeSourceAt(time.Unix(0, currentTimestamp))\n\t\t\tif testCase.mockAllowances != nil {\n\t\t\t\ttestCase.mockAllowances(ctx)\n\t\t\t}\n\n\t\t\tchildExecutionInfo := testCase.childInfo\n\t\t\tif childExecutionInfo == nil {\n\t\t\t\tchildExecutionInfo = map[int64]*persistence.ChildExecutionInfo{}\n\t\t\t}\n\n\t\t\tm := loadMutableState(t, ctx, &persistence.WorkflowMutableState{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowID:       parentWfID,\n\t\t\t\t\tRunID:            parentRunID,\n\t\t\t\t\tWorkflowTypeName: \"type\",\n\t\t\t\t\tBranchToken:      currentBranchToken,\n\t\t\t\t\tNextEventID:      nextEventID,\n\t\t\t\t\tState:            testCase.wfState,\n\t\t\t\t},\n\t\t\t\tChildExecutionInfos: childExecutionInfo,\n\t\t\t})\n\n\t\t\tevent, err := testCase.eventFn(m)\n\n\t\t\tassert.Equal(t, testCase.expectedEvent, event)\n\t\t\tassert.Equal(t, testCase.expectedErr, err)\n\t\t})\n\t}\n}\n\n// MutableState is all about mutating, avoid reusing objects\nfunc standardExecutionInfo() *persistence.WorkflowExecutionInfo {\n\treturn &persistence.WorkflowExecutionInfo{\n\t\tDomainID:         constants.TestDomainID,\n\t\tWorkflowID:       constants.TestWorkflowID,\n\t\tRunID:            constants.TestRunID,\n\t\tWorkflowTypeName: \"type\",\n\t\tBranchToken:      currentBranchToken,\n\t}\n}\n\nfunc createShardCtx(t *testing.T) *shard.TestContext {\n\treturn shard.NewTestContext(\n\t\tt,\n\t\tgomock.NewController(t),\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          testShardID,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n}\n\nfunc loadMutableState(t *testing.T, ctx *shard.TestContext, state *persistence.WorkflowMutableState) *mutableStateBuilder {\n\tdomain := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{},\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\tctx.Resource.DomainCache.EXPECT().GetDomainID(constants.TestDomainName).AnyTimes()\n\tctx.Resource.DomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(domain, nil).AnyTimes()\n\tctx.Resource.DomainCache.EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\tm := newMutableStateBuilder(ctx,\n\t\tlog.NewNoop(),\n\t\tdomain,\n\t\tdomain.GetFailoverVersion(),\n\t)\n\terr := m.Load(context.Background(), state)\n\tassert.NoError(t, err)\n\treturn m\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_completed.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) IsWorkflowCompleted() bool {\n\treturn e.executionInfo.State == persistence.WorkflowStateCompleted\n}\n\nfunc (e *mutableStateBuilder) AddCompletedWorkflowEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.CompleteWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowCompleted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddCompletedWorkflowEvent(decisionCompletedEventID, attributes)\n\tif err := e.ReplicateWorkflowExecutionCompletedEvent(decisionCompletedEventID, event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionCompletedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n) error {\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusCompleted,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(event)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(event, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n\nfunc (e *mutableStateBuilder) AddFailWorkflowEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.FailWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddFailWorkflowEvent(decisionCompletedEventID, attributes)\n\tif err := e.ReplicateWorkflowExecutionFailedEvent(decisionCompletedEventID, event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionFailedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n) error {\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusFailed,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(event)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(event, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n\n// GetCompletionEvent retrieves the workflow completion event from mutable state\nfunc (e *mutableStateBuilder) GetCompletionEvent(\n\tctx context.Context,\n) (*types.HistoryEvent, error) {\n\tif e.executionInfo.State != persistence.WorkflowStateCompleted {\n\t\treturn nil, ErrMissingWorkflowCompletionEvent\n\t}\n\n\t// Needed for backward compatibility reason\n\tif e.executionInfo.CompletionEvent != nil {\n\t\treturn e.executionInfo.CompletionEvent, nil\n\t}\n\n\t// Needed for backward compatibility reason\n\tif e.executionInfo.CompletionEventBatchID == constants.EmptyEventID {\n\t\treturn nil, ErrMissingWorkflowCompletionEvent\n\t}\n\n\tcurrentBranchToken, err := e.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Completion EventID is always one less than NextEventID after workflow is completed\n\tcompletionEventID := e.executionInfo.NextEventID - 1\n\tfirstEventID := e.executionInfo.CompletionEventBatchID\n\tcompletionEvent, err := e.eventsCache.GetEvent(\n\t\tctx,\n\t\te.shard.GetShardID(),\n\t\te.executionInfo.DomainID,\n\t\te.executionInfo.WorkflowID,\n\t\te.executionInfo.RunID,\n\t\tfirstEventID,\n\t\tcompletionEventID,\n\t\tcurrentBranchToken,\n\t)\n\tif err != nil {\n\t\t// do not return the original error\n\t\t// since original error can be of type entity not exists\n\t\t// which can cause task processing side to fail silently\n\t\t// However, if the error is a persistence transient error,\n\t\t// we return the original error, because we fail to get\n\t\t// the event because of failure from database\n\t\tif persistence.IsTransientError(err) {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn nil, ErrMissingWorkflowCompletionEvent\n\t}\n\n\treturn completionEvent, nil\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_decision.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"encoding/json\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// GetDecisionInfo returns details about the in-progress decision task\nfunc (e *mutableStateBuilder) GetDecisionInfo(\n\tscheduleEventID int64,\n) (*DecisionInfo, bool) {\n\treturn e.decisionTaskManager.GetDecisionInfo(scheduleEventID)\n}\n\n// UpdateDecision updates a decision task.\nfunc (e *mutableStateBuilder) UpdateDecision(\n\tdecision *DecisionInfo,\n) {\n\te.decisionTaskManager.UpdateDecision(decision)\n}\n\n// DeleteDecision deletes a decision task.\nfunc (e *mutableStateBuilder) DeleteDecision() {\n\te.decisionTaskManager.DeleteDecision()\n}\n\nfunc (e *mutableStateBuilder) FailDecision(\n\tincrementAttempt bool,\n) {\n\te.decisionTaskManager.FailDecision(incrementAttempt)\n}\n\nfunc (e *mutableStateBuilder) HasProcessedOrPendingDecision() bool {\n\treturn e.decisionTaskManager.HasProcessedOrPendingDecision()\n}\n\nfunc (e *mutableStateBuilder) HasPendingDecision() bool {\n\treturn e.decisionTaskManager.HasPendingDecision()\n}\n\nfunc (e *mutableStateBuilder) GetPendingDecision() (*DecisionInfo, bool) {\n\treturn e.decisionTaskManager.GetPendingDecision()\n}\n\nfunc (e *mutableStateBuilder) HasInFlightDecision() bool {\n\treturn e.decisionTaskManager.HasInFlightDecision()\n}\n\nfunc (e *mutableStateBuilder) GetInFlightDecision() (*DecisionInfo, bool) {\n\treturn e.decisionTaskManager.GetInFlightDecision()\n}\n\nfunc (e *mutableStateBuilder) GetDecisionScheduleToStartTimeout() time.Duration {\n\treturn e.decisionTaskManager.GetDecisionScheduleToStartTimeout()\n}\n\nfunc (e *mutableStateBuilder) AddFirstDecisionTaskScheduled(\n\tstartEvent *types.HistoryEvent,\n) error {\n\topTag := tag.WorkflowActionDecisionTaskScheduled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn err\n\t}\n\treturn e.decisionTaskManager.AddFirstDecisionTaskScheduled(startEvent)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskScheduledEvent(\n\tbypassTaskGeneration bool,\n) (*DecisionInfo, error) {\n\topTag := tag.WorkflowActionDecisionTaskScheduled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskScheduledEvent(bypassTaskGeneration)\n}\n\n// originalScheduledTimestamp is to record the first scheduled decision during decision heartbeat.\nfunc (e *mutableStateBuilder) AddDecisionTaskScheduledEventAsHeartbeat(\n\tbypassTaskGeneration bool,\n\toriginalScheduledTimestamp int64,\n) (*DecisionInfo, error) {\n\topTag := tag.WorkflowActionDecisionTaskScheduled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration, originalScheduledTimestamp)\n}\n\nfunc (e *mutableStateBuilder) ReplicateTransientDecisionTaskScheduled() error {\n\treturn e.decisionTaskManager.ReplicateTransientDecisionTaskScheduled()\n}\n\nfunc (e *mutableStateBuilder) ReplicateDecisionTaskScheduledEvent(\n\tversion int64,\n\tscheduleID int64,\n\ttaskList string,\n\tstartToCloseTimeoutSeconds int32,\n\tattempt int64,\n\tscheduleTimestamp int64,\n\toriginalScheduledTimestamp int64,\n\tbypassTaskGeneration bool,\n) (*DecisionInfo, error) {\n\treturn e.decisionTaskManager.ReplicateDecisionTaskScheduledEvent(version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, bypassTaskGeneration)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskStartedEvent(\n\tscheduleEventID int64,\n\trequestID string,\n\trequest *types.PollForDecisionTaskRequest,\n) (*types.HistoryEvent, *DecisionInfo, error) {\n\topTag := tag.WorkflowActionDecisionTaskStarted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskStartedEvent(scheduleEventID, requestID, request)\n}\n\nfunc (e *mutableStateBuilder) ReplicateDecisionTaskStartedEvent(\n\tdecision *DecisionInfo,\n\tversion int64,\n\tscheduleID int64,\n\tstartedID int64,\n\trequestID string,\n\ttimestamp int64,\n) (*DecisionInfo, error) {\n\n\treturn e.decisionTaskManager.ReplicateDecisionTaskStartedEvent(decision, version, scheduleID, startedID, requestID, timestamp)\n}\n\nfunc (e *mutableStateBuilder) CreateTransientDecisionEvents(\n\tdecision *DecisionInfo,\n\tidentity string,\n) (*types.HistoryEvent, *types.HistoryEvent) {\n\treturn e.decisionTaskManager.CreateTransientDecisionEvents(decision, identity)\n}\n\n// TODO: we will release the restriction when reset API allow those pending\nfunc (e *mutableStateBuilder) CheckResettable() error {\n\tif len(e.GetPendingChildExecutionInfos()) > 0 {\n\t\treturn &types.BadRequestError{\n\t\t\tMessage: \"it is not allowed resetting to a point that workflow has pending child types.\",\n\t\t}\n\t}\n\tif len(e.GetPendingRequestCancelExternalInfos()) > 0 {\n\t\treturn &types.BadRequestError{\n\t\t\tMessage: \"it is not allowed resetting to a point that workflow has pending request cancel.\",\n\t\t}\n\t}\n\tif len(e.GetPendingSignalExternalInfos()) > 0 {\n\t\treturn &types.BadRequestError{\n\t\t\tMessage: \"it is not allowed resetting to a point that workflow has pending signals to send.\",\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskCompletedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\trequest *types.RespondDecisionTaskCompletedRequest,\n\tmaxResetPoints int,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskCompleted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID, request, maxResetPoints)\n}\n\nfunc (e *mutableStateBuilder) ReplicateDecisionTaskCompletedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\treturn e.decisionTaskManager.ReplicateDecisionTaskCompletedEvent(event)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskTimedOutEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskTimedOutEvent(scheduleEventID, startedEventID)\n}\n\nfunc (e *mutableStateBuilder) ReplicateDecisionTaskTimedOutEvent(\n\tevent *types.HistoryEvent,\n) error {\n\treturn e.decisionTaskManager.ReplicateDecisionTaskTimedOutEvent(event)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskScheduleToStartTimeoutEvent(\n\tscheduleEventID int64,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskResetTimeoutEvent(\n\tscheduleEventID int64,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\treason string,\n\tresetRequestID string,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskResetTimeoutEvent(\n\t\tscheduleEventID,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tforkEventVersion,\n\t\treason,\n\t\tresetRequestID,\n\t)\n}\n\nfunc (e *mutableStateBuilder) AddDecisionTaskFailedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\tcause types.DecisionTaskFailedCause,\n\tdetails []byte,\n\tidentity string,\n\treason string,\n\tbinChecksum string,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\tresetRequestID string,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\treturn e.decisionTaskManager.AddDecisionTaskFailedEvent(\n\t\tscheduleEventID,\n\t\tstartedEventID,\n\t\tcause,\n\t\tdetails,\n\t\tidentity,\n\t\treason,\n\t\tbinChecksum,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tforkEventVersion,\n\t\tresetRequestID,\n\t)\n}\n\nfunc (e *mutableStateBuilder) ReplicateDecisionTaskFailedEvent(event *types.HistoryEvent) error {\n\treturn e.decisionTaskManager.ReplicateDecisionTaskFailedEvent(event)\n}\n\n// add BinaryCheckSum for the first decisionTaskCompletedID for auto-reset\nfunc (e *mutableStateBuilder) addBinaryCheckSumIfNotExists(\n\tevent *types.HistoryEvent,\n\tmaxResetPoints int,\n) error {\n\tbinChecksum := event.GetDecisionTaskCompletedEventAttributes().GetBinaryChecksum()\n\tif len(binChecksum) == 0 {\n\t\treturn nil\n\t}\n\texeInfo := e.executionInfo\n\tvar currResetPoints []*types.ResetPointInfo\n\tif exeInfo.AutoResetPoints != nil && exeInfo.AutoResetPoints.Points != nil {\n\t\tcurrResetPoints = e.executionInfo.AutoResetPoints.Points\n\t} else {\n\t\tcurrResetPoints = make([]*types.ResetPointInfo, 0, 1)\n\t}\n\n\t// List of all recent binary checksums associated with the types.\n\tvar recentBinaryChecksums []string\n\n\tfor _, rp := range currResetPoints {\n\t\trecentBinaryChecksums = append(recentBinaryChecksums, rp.GetBinaryChecksum())\n\t\tif rp.GetBinaryChecksum() == binChecksum {\n\t\t\t// this checksum already exists\n\t\t\treturn nil\n\t\t}\n\t}\n\n\trecentBinaryChecksums, currResetPoints = trimBinaryChecksums(recentBinaryChecksums, currResetPoints, maxResetPoints)\n\n\t// Adding current version of the binary checksum.\n\trecentBinaryChecksums = append(recentBinaryChecksums, binChecksum)\n\n\tresettable := true\n\terr := e.CheckResettable()\n\tif err != nil {\n\t\tresettable = false\n\t}\n\tinfo := &types.ResetPointInfo{\n\t\tBinaryChecksum:           binChecksum,\n\t\tRunID:                    exeInfo.RunID,\n\t\tFirstDecisionCompletedID: event.ID,\n\t\tCreatedTimeNano:          common.Int64Ptr(e.timeSource.Now().UnixNano()),\n\t\tResettable:               resettable,\n\t}\n\tcurrResetPoints = append(currResetPoints, info)\n\texeInfo.AutoResetPoints = &types.ResetPoints{\n\t\tPoints: currResetPoints,\n\t}\n\tbytes, err := json.Marshal(recentBinaryChecksums)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif exeInfo.SearchAttributes == nil {\n\t\texeInfo.SearchAttributes = make(map[string][]byte)\n\t}\n\texeInfo.SearchAttributes[definition.BinaryChecksums] = bytes\n\tif common.IsAdvancedVisibilityWritingEnabled(e.shard.GetConfig().WriteVisibilityStoreName(), e.shard.GetConfig().IsAdvancedVisConfigExist) {\n\t\treturn e.taskGenerator.GenerateWorkflowSearchAttrTasks()\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_decision_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc CreateDecisionInfo() *DecisionInfo {\n\treturn &DecisionInfo{\n\t\tVersion:                    123,\n\t\tScheduleID:                 123,\n\t\tStartedID:                  123,\n\t\tRequestID:                  \"123\",\n\t\tDecisionTimeout:            123,\n\t\tStartedTimestamp:           123,\n\t\tScheduledTimestamp:         123,\n\t\tOriginalScheduledTimestamp: 123,\n\t}\n}\n\nfunc TestGetDecisionInfoMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tscheduleEventID := int64(123)\n\trets := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().GetDecisionInfo(scheduleEventID).Return(rets, true).Times(1)\n\n\tdecisionInfo, ok := builder.GetDecisionInfo(scheduleEventID)\n\n\tassert.Equal(t, rets, decisionInfo)\n\tassert.True(t, ok)\n}\n\nfunc TestUpdateDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecision := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().UpdateDecision(decision).Times(1)\n\n\tbuilder.UpdateDecision(decision)\n}\n\nfunc TestDeleteDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().DeleteDecision().Times(1)\n\n\tbuilder.DeleteDecision()\n}\n\nfunc TestFailDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().FailDecision(true).Times(1)\n\n\tbuilder.FailDecision(true)\n}\n\nfunc TestHasProcessedOrPendingDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().HasProcessedOrPendingDecision().Return(true).Times(1)\n\n\thasProcessedOrPendingDecision := builder.HasProcessedOrPendingDecision()\n\n\tassert.True(t, hasProcessedOrPendingDecision)\n}\n\nfunc TestHasPendingDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().HasPendingDecision().Return(true).Times(1)\n\n\thasPendingDecision := builder.HasPendingDecision()\n\n\tassert.True(t, hasPendingDecision)\n}\n\nfunc TestGetPendingDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\trets := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().GetPendingDecision().Return(rets, true).Times(1)\n\n\tpendingDecision, ok := builder.GetPendingDecision()\n\n\tassert.Equal(t, rets, pendingDecision)\n\tassert.True(t, ok)\n}\n\nfunc TestHasInFlightDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().HasInFlightDecision().Return(true).Times(1)\n\n\thasInFlightDecision := builder.HasInFlightDecision()\n\n\tassert.True(t, hasInFlightDecision)\n}\n\nfunc TestGetInFlightDecisionMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\trets := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().GetInFlightDecision().Return(rets, true).Times(1)\n\n\tinFlightDecision, ok := builder.GetInFlightDecision()\n\n\tassert.Equal(t, rets, inFlightDecision)\n\tassert.True(t, ok)\n}\n\nfunc TestGetDecisionScheduleToStartTimeoutMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\trets := time.Duration(123)\n\n\tdecisionTaskManager.EXPECT().GetDecisionScheduleToStartTimeout().Return(rets).Times(1)\n\n\ttimeout := builder.GetDecisionScheduleToStartTimeout()\n\n\tassert.Equal(t, rets, timeout)\n}\n\nfunc TestAddFirstDecisionTaskScheduledMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tstartEvent := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddFirstDecisionTaskScheduled(startEvent).Return(nil).Times(1)\n\t\t\t}\n\n\t\t\terr := builder.AddFirstDecisionTaskScheduled(startEvent)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskScheduledEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\trets := CreateDecisionInfo()\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskScheduledEvent(true).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\tdecisionInfo, err := builder.AddDecisionTaskScheduledEvent(true)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, decisionInfo)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskScheduledEventAsHeartbeatMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\trets := CreateDecisionInfo()\n\t\t\toriginalScheduledTimestamp := int64(1)\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskScheduledEventAsHeartbeat(true, originalScheduledTimestamp).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\tdecisionInfo, err := builder.AddDecisionTaskScheduledEventAsHeartbeat(true, originalScheduledTimestamp)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, decisionInfo)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicateTransientDecisionTaskScheduledMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecisionTaskManager.EXPECT().ReplicateTransientDecisionTaskScheduled().Return(nil).Times(1)\n\n\terr := builder.ReplicateTransientDecisionTaskScheduled()\n\n\tassert.NoError(t, err)\n}\n\nfunc TestReplicateDecisionTaskScheduledEventMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tvar version, scheduleID, attempt, scheduleTimestamp, originalScheduledTimestamp int64 = 1, 2, 3, 4, 5\n\tstartToCloseTimeoutSeconds := int32(6)\n\ttaskList := \"taskList\"\n\trets := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().ReplicateDecisionTaskScheduledEvent(version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, true).Return(rets, nil).Times(1)\n\n\tdecisionInfo, err := builder.ReplicateDecisionTaskScheduledEvent(version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, true)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, rets, decisionInfo)\n}\n\nfunc TestAddDecisionTaskStartedEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\trequest       *types.PollForDecisionTaskRequest\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname:    \"success\",\n\t\t\trequest: &types.PollForDecisionTaskRequest{},\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"error due to state\",\n\t\t\trequest: &types.PollForDecisionTaskRequest{},\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tscheduleEventID := int64(123)\n\t\t\trequestID := \"requestID\"\n\t\t\trets0 := &types.HistoryEvent{}\n\t\t\trets1 := CreateDecisionInfo()\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskStartedEvent(scheduleEventID, requestID, tc.request).Return(rets0, rets1, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, decisionInfo, err := builder.AddDecisionTaskStartedEvent(scheduleEventID, requestID, tc.request)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets0, historyEvent)\n\t\t\t\tassert.Equal(t, rets1, decisionInfo)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicateDecisionTaskStartedEventMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tvar version, scheduleID, startedID, timestamp int64 = 1, 2, 3, 4\n\trequestID := \"requestID\"\n\trets := CreateDecisionInfo()\n\n\tdecisionTaskManager.EXPECT().ReplicateDecisionTaskStartedEvent(rets, version, scheduleID, startedID, requestID, timestamp).Return(rets, nil).Times(1)\n\n\tdecisionInfo, err := builder.ReplicateDecisionTaskStartedEvent(rets, version, scheduleID, startedID, requestID, timestamp)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, rets, decisionInfo)\n}\n\nfunc TestCreateTransientDecisionEventsMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tdecision := CreateDecisionInfo()\n\tidentity := \"identity\"\n\trets := &types.HistoryEvent{}\n\n\tdecisionTaskManager.EXPECT().CreateTransientDecisionEvents(decision, identity).Return(rets, rets).Times(1)\n\n\thistoryEvent0, historyEvent1 := builder.CreateTransientDecisionEvents(decision, identity)\n\n\tassert.Equal(t, rets, historyEvent0)\n\tassert.Equal(t, rets, historyEvent1)\n}\n\nfunc TestCheckResettableMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname                         string\n\t\tpendingChildExecutionInfoIDs map[int64]*persistence.ChildExecutionInfo\n\t\tpendingRequestCancelInfoIDs  map[int64]*persistence.RequestCancelInfo\n\t\tpendingSignalInfoIDs         map[int64]*persistence.SignalInfo\n\t\twantErr                      bool\n\t\terrMessage                   string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t},\n\t\t{\n\t\t\tname: \"error due to pending child execution\",\n\t\t\tpendingChildExecutionInfoIDs: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t1: {},\n\t\t\t},\n\t\t\twantErr:    true,\n\t\t\terrMessage: \"it is not allowed resetting to a point that workflow has pending child types.\",\n\t\t},\n\t\t{\n\t\t\tname: \"error due to pending request cancel external\",\n\t\t\tpendingRequestCancelInfoIDs: map[int64]*persistence.RequestCancelInfo{\n\t\t\t\t1: {},\n\t\t\t},\n\t\t\twantErr:    true,\n\t\t\terrMessage: \"it is not allowed resetting to a point that workflow has pending request cancel.\",\n\t\t},\n\t\t{\n\t\t\tname: \"error due to pending signal external\",\n\t\t\tpendingSignalInfoIDs: map[int64]*persistence.SignalInfo{\n\t\t\t\t1: {},\n\t\t\t},\n\t\t\twantErr:    true,\n\t\t\terrMessage: \"it is not allowed resetting to a point that workflow has pending signals to send.\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tpendingChildExecutionInfoIDs: tc.pendingChildExecutionInfoIDs,\n\t\t\t\tpendingRequestCancelInfoIDs:  tc.pendingRequestCancelInfoIDs,\n\t\t\t\tpendingSignalInfoIDs:         tc.pendingSignalInfoIDs,\n\t\t\t}\n\n\t\t\terr := builder.CheckResettable()\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.errMessage, err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n\n}\n\nfunc TestAddDecisionTaskCompletedEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tvar scheduleEventID, startedEventID int64 = 123, 234\n\t\t\trequest := &types.RespondDecisionTaskCompletedRequest{}\n\t\t\tmaxResetPoints := 1\n\t\t\trets := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID, request, maxResetPoints).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, err := builder.AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID, request, maxResetPoints)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, historyEvent)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicateDecisionTaskCompletedEventMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tevent := &types.HistoryEvent{}\n\n\tdecisionTaskManager.EXPECT().ReplicateDecisionTaskCompletedEvent(event).Return(nil).Times(1)\n\n\terr := builder.ReplicateDecisionTaskCompletedEvent(event)\n\n\tassert.NoError(t, err)\n}\n\nfunc TestAddDecisionTaskTimedOutEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tvar scheduleEventID, startedEventID int64 = 123, 234\n\t\t\trets := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskTimedOutEvent(scheduleEventID, startedEventID).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, err := builder.AddDecisionTaskTimedOutEvent(scheduleEventID, startedEventID)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, historyEvent)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicateDecisionTaskTimedOutEventMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tevent := &types.HistoryEvent{}\n\n\tdecisionTaskManager.EXPECT().ReplicateDecisionTaskTimedOutEvent(event).Return(nil).Times(1)\n\n\terr := builder.ReplicateDecisionTaskTimedOutEvent(event)\n\n\tassert.NoError(t, err)\n}\n\nfunc TestAddDecisionTaskScheduleToStartTimeoutEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tscheduleEventID := int64(123)\n\t\t\trets := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, err := builder.AddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, historyEvent)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskResetTimeoutEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tvar scheduleEventID, forkEventVersion int64 = 123, 1\n\t\t\tbaserRunID, newRunID, reason, resetRequestID := \"baseRunID\", \"newRunID\", \"reason\", \"resetRequestID\"\n\t\t\trets := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskResetTimeoutEvent(scheduleEventID, baserRunID, newRunID, forkEventVersion, reason, resetRequestID).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, err := builder.AddDecisionTaskResetTimeoutEvent(scheduleEventID, baserRunID, newRunID, forkEventVersion, reason, resetRequestID)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, historyEvent)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAddDecisionTaskFailedEventMutableStateBuilder(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\texecutionInfo *persistence.WorkflowExecutionInfo\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCreated,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error due to state\",\n\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionTaskManager,\n\t\t\t\texecutionInfo:       tc.executionInfo,\n\t\t\t\tlogger:              log.NewNoop(),\n\t\t\t}\n\n\t\t\tvar scheduleEventID, startedEventID, forkEventVersion int64 = 123, 234, 1\n\t\t\tcause := types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure\n\t\t\tdetails := []byte(\"details\")\n\t\t\tidentity, reason, binChecksum, baseRunID, newRunID, resetRequestID := \"identity\", \"reason\", \"checksum\", \"baseRunID\", \"newRunID\", \"resetRequestID\"\n\t\t\trets := &types.HistoryEvent{}\n\n\t\t\tif !tc.wantErr {\n\t\t\t\tdecisionTaskManager.EXPECT().AddDecisionTaskFailedEvent(scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID).Return(rets, nil).Times(1)\n\t\t\t}\n\n\t\t\thistoryEvent, err := builder.AddDecisionTaskFailedEvent(scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID)\n\n\t\t\tif tc.wantErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, rets, historyEvent)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestReplicateDecisionTaskFailedEventMutableStateBuilder(t *testing.T) {\n\tdecisionTaskManager := NewMockmutableStateDecisionTaskManager(gomock.NewController(t))\n\tbuilder := &mutableStateBuilder{\n\t\tdecisionTaskManager: decisionTaskManager,\n\t}\n\n\tevent := &types.HistoryEvent{}\n\n\tdecisionTaskManager.EXPECT().ReplicateDecisionTaskFailedEvent(event).Return(nil).Times(1)\n\n\terr := builder.ReplicateDecisionTaskFailedEvent(event)\n\n\tassert.NoError(t, err)\n}\n\nfunc TestAddBinaryCheckSumIfNotExistsMutableStateBuilder(t *testing.T) {\n\ttimeNow := time.Now()\n\trunID := \"runID\"\n\tcheckSum := \"checkSum\"\n\teventID := int64(1)\n\n\ttests := []struct {\n\t\tname                                 string\n\t\tdecisionTaskCompletedEventAttributes *types.DecisionTaskCompletedEventAttributes\n\t\tautoResetPoints                      *types.ResetPoints\n\t\tshardConfig                          *config.Config\n\t\tpendingChildExecutionInfoIDs         map[int64]*persistence.ChildExecutionInfo\n\t\twantWorkFlowExecutionInfo            *persistence.WorkflowExecutionInfo\n\t}{\n\t\t{\n\t\t\tname: \"binaryChecksum not added due to empty binChecksum\",\n\t\t\twantWorkFlowExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tRunID: runID,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"binaryChecksum not added due to existing checksum with AutoResetPoints and AutoResetPoints.Points in executionInfo\",\n\t\t\tdecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tBinaryChecksum: checkSum,\n\t\t\t},\n\t\t\tautoResetPoints: &types.ResetPoints{\n\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t{BinaryChecksum: checkSum},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantWorkFlowExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tRunID: runID,\n\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{BinaryChecksum: checkSum},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with existing distinct autoResetPoints\",\n\t\t\tdecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tBinaryChecksum: checkSum,\n\t\t\t},\n\t\t\tautoResetPoints: &types.ResetPoints{\n\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t{BinaryChecksum: \"anotherCheckSum\"},\n\t\t\t\t\t{BinaryChecksum: \"toBeTrimmed\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\tshardConfig: &config.Config{\n\t\t\t\tWriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(\"off\"),\n\t\t\t},\n\t\t\twantWorkFlowExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tRunID: runID,\n\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBinaryChecksum:           checkSum,\n\t\t\t\t\t\t\tRunID:                    runID,\n\t\t\t\t\t\t\tFirstDecisionCompletedID: eventID,\n\t\t\t\t\t\t\tCreatedTimeNano:          common.Int64Ptr(timeNow.UnixNano()),\n\t\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSearchAttributes: map[string][]byte{\n\t\t\t\t\tdefinition.BinaryChecksums: func() []byte {\n\t\t\t\t\t\tbytes, _ := json.Marshal([]string{checkSum})\n\t\t\t\t\t\treturn bytes\n\t\t\t\t\t}(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with AdvancedVisibilityWritingEnabled\",\n\t\t\tdecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tBinaryChecksum: checkSum,\n\t\t\t},\n\t\t\tshardConfig: &config.Config{\n\t\t\t\tWriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(\"es\"),\n\t\t\t\tIsAdvancedVisConfigExist: true,\n\t\t\t},\n\t\t\twantWorkFlowExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tRunID: runID,\n\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBinaryChecksum:           checkSum,\n\t\t\t\t\t\t\tRunID:                    runID,\n\t\t\t\t\t\t\tFirstDecisionCompletedID: eventID,\n\t\t\t\t\t\t\tCreatedTimeNano:          common.Int64Ptr(timeNow.UnixNano()),\n\t\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSearchAttributes: map[string][]byte{\n\t\t\t\t\tdefinition.BinaryChecksums: func() []byte {\n\t\t\t\t\t\tbytes, _ := json.Marshal([]string{checkSum})\n\t\t\t\t\t\treturn bytes\n\t\t\t\t\t}(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with CheckResettable error\",\n\t\t\tdecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\tBinaryChecksum: checkSum,\n\t\t\t},\n\t\t\tshardConfig: &config.Config{\n\t\t\t\tWriteVisibilityStoreName: dynamicproperties.GetStringPropertyFn(\"off\"),\n\t\t\t},\n\t\t\tpendingChildExecutionInfoIDs: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t1: {},\n\t\t\t},\n\t\t\twantWorkFlowExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tRunID: runID,\n\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBinaryChecksum:           checkSum,\n\t\t\t\t\t\t\tRunID:                    runID,\n\t\t\t\t\t\t\tFirstDecisionCompletedID: eventID,\n\t\t\t\t\t\t\tCreatedTimeNano:          common.Int64Ptr(timeNow.UnixNano()),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSearchAttributes: map[string][]byte{\n\t\t\t\t\tdefinition.BinaryChecksums: func() []byte {\n\t\t\t\t\t\tbytes, _ := json.Marshal([]string{checkSum})\n\t\t\t\t\t\treturn bytes\n\t\t\t\t\t}(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tbuilder := &mutableStateBuilder{\n\t\t\t\tlogger: log.NewNoop(),\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tAutoResetPoints: tc.autoResetPoints,\n\t\t\t\t\tRunID:           runID,\n\t\t\t\t},\n\t\t\t\ttimeSource:                   clock.NewMockedTimeSourceAt(timeNow),\n\t\t\t\tpendingChildExecutionInfoIDs: tc.pendingChildExecutionInfoIDs,\n\t\t\t\tshard:                        shard.NewMockContext(ctrl),\n\t\t\t\ttaskGenerator:                NewMockMutableStateTaskGenerator(ctrl),\n\t\t\t}\n\n\t\t\tevent := &types.HistoryEvent{\n\t\t\t\tID:                                   eventID,\n\t\t\t\tDecisionTaskCompletedEventAttributes: tc.decisionTaskCompletedEventAttributes,\n\t\t\t}\n\t\t\tmaxResetPoints := 1\n\n\t\t\tif tc.shardConfig != nil {\n\t\t\t\tbuilder.shard.(*shard.MockContext).EXPECT().GetConfig().Return(tc.shardConfig).Times(2)\n\t\t\t}\n\n\t\t\tif tc.shardConfig != nil && tc.shardConfig.IsAdvancedVisConfigExist {\n\t\t\t\tbuilder.taskGenerator.(*MockMutableStateTaskGenerator).EXPECT().GenerateWorkflowSearchAttrTasks().Return(nil).Times(1)\n\t\t\t}\n\n\t\t\terr := builder.addBinaryCheckSumIfNotExists(event, maxResetPoints)\n\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif diff := cmp.Diff(tc.wantWorkFlowExecutionInfo, builder.executionInfo); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_signal.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) IsSignalRequested(\n\trequestID string,\n) bool {\n\n\tif _, ok := e.pendingSignalRequestedIDs[requestID]; ok {\n\t\treturn true\n\t}\n\treturn false\n}\n\n// GetSignalInfo get details about a signal request that is currently in progress.\nfunc (e *mutableStateBuilder) GetSignalInfo(\n\tinitiatedEventID int64,\n) (*persistence.SignalInfo, bool) {\n\n\tri, ok := e.pendingSignalInfoIDs[initiatedEventID]\n\treturn ri, ok\n}\n\nfunc (e *mutableStateBuilder) GetPendingSignalExternalInfos() map[int64]*persistence.SignalInfo {\n\treturn e.pendingSignalInfoIDs\n\n}\n\nfunc (e *mutableStateBuilder) AddWorkflowExecutionSignaled(\n\tsignalName string,\n\tinput []byte,\n\tidentity string,\n\trequestID string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowSignaled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionSignaledEvent(signalName, input, identity, requestID)\n\tif err := e.ReplicateWorkflowExecutionSignaled(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionSignaled(\n\tevent *types.HistoryEvent,\n) error {\n\n\t// Increment signal count in mutable state for this workflow execution\n\te.executionInfo.SignalCount++\n\te.insertWorkflowRequest(persistence.WorkflowRequest{\n\t\tRequestID:   event.WorkflowExecutionSignaledEventAttributes.RequestID,\n\t\tVersion:     event.Version,\n\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t})\n\treturn nil\n}\n\nfunc (e *mutableStateBuilder) AddExternalWorkflowExecutionSignaled(\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tcontrol []byte,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowSignalRequested\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\t_, ok := e.GetSignalInfo(initiatedID)\n\tif !ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddExternalWorkflowExecutionSignaled(initiatedID, domain, workflowID, runID, control)\n\tif err := e.ReplicateExternalWorkflowExecutionSignaled(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateExternalWorkflowExecutionSignaled(\n\tevent *types.HistoryEvent,\n) error {\n\n\tinitiatedID := event.ExternalWorkflowExecutionSignaledEventAttributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingSignal(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddSignalExternalWorkflowExecutionFailedEvent(\n\tdecisionTaskCompletedEventID int64,\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tcontrol []byte,\n\tcause types.SignalExternalWorkflowExecutionFailedCause,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowSignalFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\t_, ok := e.GetSignalInfo(initiatedID)\n\tif !ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowInitiatedID(initiatedID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddSignalExternalWorkflowExecutionFailedEvent(decisionTaskCompletedEventID, initiatedID, domain,\n\t\tworkflowID, runID, control, cause)\n\tif err := e.ReplicateSignalExternalWorkflowExecutionFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateSignalExternalWorkflowExecutionFailedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tinitiatedID := event.SignalExternalWorkflowExecutionFailedEventAttributes.GetInitiatedEventID()\n\n\treturn e.DeletePendingSignal(initiatedID)\n}\n\nfunc (e *mutableStateBuilder) AddSignalRequested(\n\trequestID string,\n) {\n\n\tif e.pendingSignalRequestedIDs == nil {\n\t\te.pendingSignalRequestedIDs = make(map[string]struct{})\n\t}\n\tif e.updateSignalRequestedIDs == nil {\n\t\te.updateSignalRequestedIDs = make(map[string]struct{})\n\t}\n\te.pendingSignalRequestedIDs[requestID] = struct{}{} // add requestID to set\n\te.updateSignalRequestedIDs[requestID] = struct{}{}\n}\n\nfunc (e *mutableStateBuilder) DeleteSignalRequested(\n\trequestID string,\n) {\n\n\tdelete(e.pendingSignalRequestedIDs, requestID)\n\tdelete(e.updateSignalRequestedIDs, requestID)\n\te.deleteSignalRequestedIDs[requestID] = struct{}{}\n}\n\nfunc (e *mutableStateBuilder) AddSignalExternalWorkflowExecutionInitiatedEvent(\n\tdecisionCompletedEventID int64,\n\tsignalRequestID string,\n\trequest *types.SignalExternalWorkflowExecutionDecisionAttributes,\n) (*types.HistoryEvent, *persistence.SignalInfo, error) {\n\n\topTag := tag.WorkflowActionExternalWorkflowSignalInitiated\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tevent := e.hBuilder.AddSignalExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID, request)\n\tsi, err := e.ReplicateSignalExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID, event, signalRequestID)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn event, si, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateSignalExternalWorkflowExecutionInitiatedEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n\tsignalRequestID string,\n) (*persistence.SignalInfo, error) {\n\n\t// TODO: Consider also writing signalRequestID to history event\n\tinitiatedEventID := event.ID\n\tattributes := event.SignalExternalWorkflowExecutionInitiatedEventAttributes\n\tsi := &persistence.SignalInfo{\n\t\tVersion:               event.Version,\n\t\tInitiatedEventBatchID: firstEventID,\n\t\tInitiatedID:           initiatedEventID,\n\t\tSignalRequestID:       signalRequestID,\n\t\tSignalName:            attributes.GetSignalName(),\n\t\tInput:                 attributes.Input,\n\t\tControl:               attributes.Control,\n\t}\n\n\te.pendingSignalInfoIDs[si.InitiatedID] = si\n\te.updateSignalInfos[si.InitiatedID] = si\n\n\treturn si, e.taskGenerator.GenerateSignalExternalTasks(event)\n}\n\n// DeletePendingSignal deletes details about a SignalInfo\nfunc (e *mutableStateBuilder) DeletePendingSignal(\n\tinitiatedEventID int64,\n) error {\n\n\tif _, ok := e.pendingSignalInfoIDs[initiatedEventID]; ok {\n\t\tdelete(e.pendingSignalInfoIDs, initiatedEventID)\n\t} else {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find signal external workflow event ID: %v in mutable state\", initiatedEventID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\t// log data inconsistency instead of returning an error\n\t\te.logDataInconsistency()\n\t}\n\n\tdelete(e.updateSignalInfos, initiatedEventID)\n\te.deleteSignalInfos[initiatedEventID] = struct{}{}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_signal_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\nfunc Test__IsSignalRequested(t *testing.T) {\n\trequestID := \"101\"\n\tt.Run(\"signal not found\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tresult := mb.IsSignalRequested(requestID)\n\t\tassert.False(t, result)\n\t})\n\tt.Run(\"signal found\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.pendingSignalRequestedIDs[requestID] = struct{}{}\n\t\tresult := mb.IsSignalRequested(requestID)\n\t\tassert.True(t, result)\n\t})\n}\n\nfunc Test__GetSignalInfo(t *testing.T) {\n\tinitiatedEventID := int64(1)\n\tinfo := &persistence.SignalInfo{\n\t\tInitiatedID:     1,\n\t\tSignalRequestID: \"101\",\n\t}\n\tt.Run(\"signal not found\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, ok := mb.GetSignalInfo(initiatedEventID)\n\t\tassert.False(t, ok)\n\t})\n\tt.Run(\"signal found\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.pendingSignalInfoIDs[initiatedEventID] = info\n\t\tresult, ok := mb.GetSignalInfo(initiatedEventID)\n\t\tassert.True(t, ok)\n\t\tassert.Equal(t, info, result)\n\t})\n}\n\nfunc Test__ReplicateExternalWorkflowExecutionSignaled(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tExternalWorkflowExecutionSignaledEventAttributes: &types.ExternalWorkflowExecutionSignaledEventAttributes{\n\t\t\tInitiatedEventID: 1,\n\t\t},\n\t}\n\tinfo := &persistence.SignalInfo{\n\t\tInitiatedID:     1,\n\t\tSignalRequestID: \"101\",\n\t}\n\tmb.pendingSignalInfoIDs[int64(1)] = info\n\tmb.updateSignalInfos[int64(1)] = info\n\terr := mb.ReplicateExternalWorkflowExecutionSignaled(event)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, mb.deleteSignalInfos[int64(1)])\n\t_, ok := mb.pendingSignalInfoIDs[int64(1)]\n\tassert.False(t, ok)\n\t_, ok = mb.updateSignalInfos[int64(1)]\n\tassert.False(t, ok)\n}\n\nfunc Test__ReplicateSignalExternalWorkflowExecutionFailedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tevent := &types.HistoryEvent{\n\t\tSignalExternalWorkflowExecutionFailedEventAttributes: &types.SignalExternalWorkflowExecutionFailedEventAttributes{\n\t\t\tInitiatedEventID: 1,\n\t\t},\n\t}\n\tinfo := &persistence.SignalInfo{\n\t\tInitiatedID:     1,\n\t\tSignalRequestID: \"101\",\n\t}\n\tmb.pendingSignalInfoIDs[int64(1)] = info\n\tmb.updateSignalInfos[int64(1)] = info\n\terr := mb.ReplicateSignalExternalWorkflowExecutionFailedEvent(event)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, mb.deleteSignalInfos[int64(1)])\n\t_, ok := mb.pendingSignalInfoIDs[int64(1)]\n\tassert.False(t, ok)\n\t_, ok = mb.updateSignalInfos[int64(1)]\n\tassert.False(t, ok)\n}\n\nfunc Test__AddSignalRequested(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\trequestID := \"101\"\n\tmb.pendingSignalRequestedIDs = nil\n\tmb.updateSignalRequestedIDs = nil\n\tmb.AddSignalRequested(requestID)\n\tassert.NotNil(t, mb.pendingSignalRequestedIDs[requestID])\n\tassert.NotNil(t, mb.updateSignalRequestedIDs[requestID])\n}\n\nfunc Test__DeleteSignalRequested(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\trequestID := \"101\"\n\tmb.pendingSignalRequestedIDs[requestID] = struct{}{}\n\tmb.updateSignalRequestedIDs[requestID] = struct{}{}\n\tmb.DeleteSignalRequested(requestID)\n\tassert.NotNil(t, mb.deleteSignalRequestedIDs[requestID])\n}\n\nfunc Test__AddExternalWorkflowExecutionSignaled(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddExternalWorkflowExecutionSignaled(1, \"test-domain\", \"wid\", \"rid\", []byte{10})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting signal info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, err := mb.AddExternalWorkflowExecutionSignaled(1, \"test-domain\", \"wid\", \"rid\", []byte{10})\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-externalworkflow-signal-requested-event operation failed\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tsi := &persistence.SignalInfo{\n\t\t\tInitiatedID: 1,\n\t\t}\n\t\tmb.pendingSignalInfoIDs[1] = si\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddExternalWorkflowExecutionSignaled(1, \"test-domain\", \"wid\", \"rid\", []byte{10})\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.ExternalWorkflowExecutionSignaledEventAttributes.GetInitiatedEventID())\n\t})\n}\n\nfunc Test__AddSignalExternalWorkflowExecutionFailedEvent(t *testing.T) {\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, err := mbCompleted.AddSignalExternalWorkflowExecutionFailedEvent(1, 1, \"test-domain\", \"wid\", \"rid\", []byte{10}, types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"error getting signal info\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\t_, err := mb.AddSignalExternalWorkflowExecutionFailedEvent(1, 1, \"test-domain\", \"wid\", \"rid\", []byte{10}, types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, \"add-externalworkflow-signal-failed-event operation failed\", err.Error())\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tsi := &persistence.SignalInfo{\n\t\t\tInitiatedID: 1,\n\t\t}\n\t\tmb.pendingSignalInfoIDs[1] = si\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, err := mb.AddSignalExternalWorkflowExecutionFailedEvent(1, 1, \"test-domain\", \"wid\", \"rid\", []byte{10}, types.SignalExternalWorkflowExecutionFailedCauseWorkflowAlreadyCompleted)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int64(1), event.SignalExternalWorkflowExecutionFailedEventAttributes.GetInitiatedEventID())\n\t})\n}\n\nfunc Test__AddSignalExternalWorkflowExecutionInitiatedEvent(t *testing.T) {\n\trequest := &types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\tDomain: constants.TestDomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tSignalName: \"test-signal\",\n\t\tInput:      make([]byte, 0),\n\t}\n\tt.Run(\"error workflow finished\", func(t *testing.T) {\n\t\tmbCompleted := testMutableStateBuilder(t)\n\t\tmbCompleted.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t_, _, err := mbCompleted.AddSignalExternalWorkflowExecutionInitiatedEvent(1, \"101\", request)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrWorkflowFinished, err)\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb := testMutableStateBuilder(t)\n\t\tmb.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   constants.TestDomainID,\n\t\t\tWorkflowID: \"wid\",\n\t\t\tRunID:      \"rid\",\n\t\t}\n\t\tmb.hBuilder = NewHistoryBuilder(mb)\n\t\tevent, si, err := mb.AddSignalExternalWorkflowExecutionInitiatedEvent(1, \"101\", request)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, request.Execution, event.SignalExternalWorkflowExecutionInitiatedEventAttributes.GetWorkflowExecution())\n\t\tassert.Equal(t, \"101\", si.SignalRequestID)\n\t})\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_started.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) addWorkflowExecutionStartedEventForContinueAsNew(\n\tparentExecutionInfo *types.ParentExecutionInfo,\n\texecution types.WorkflowExecution,\n\tpreviousExecutionState MutableState,\n\tattributes *types.ContinueAsNewWorkflowExecutionDecisionAttributes,\n\tfirstRunID string,\n\tfirstScheduledTime time.Time,\n) (*types.HistoryEvent, error) {\n\n\tpreviousExecutionInfo := previousExecutionState.GetExecutionInfo()\n\ttl := &types.TaskList{\n\t\tName: previousExecutionInfo.TaskList,\n\t\tKind: previousExecutionInfo.TaskListKind.Ptr(),\n\t}\n\t// ContinueAsNew can change the name, not the kind\n\tif attributes.TaskList != nil {\n\t\ttl.Name = attributes.TaskList.Name\n\t}\n\n\tworkflowType := previousExecutionInfo.WorkflowTypeName\n\tif attributes.WorkflowType != nil {\n\t\tworkflowType = attributes.WorkflowType.GetName()\n\t}\n\twType := &types.WorkflowType{}\n\twType.Name = workflowType\n\n\tdecisionTimeout := previousExecutionInfo.DecisionStartToCloseTimeout\n\tif attributes.TaskStartToCloseTimeoutSeconds != nil {\n\t\tdecisionTimeout = attributes.GetTaskStartToCloseTimeoutSeconds()\n\t}\n\n\tcreateRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              e.domainEntry.GetInfo().Name,\n\t\tWorkflowID:                          execution.WorkflowID,\n\t\tTaskList:                            tl,\n\t\tWorkflowType:                        wType,\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(decisionTimeout),\n\t\tExecutionStartToCloseTimeoutSeconds: attributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tInput:                               attributes.Input,\n\t\tHeader:                              attributes.Header,\n\t\tRetryPolicy:                         attributes.RetryPolicy,\n\t\tCronSchedule:                        attributes.CronSchedule,\n\t\tMemo:                                attributes.Memo,\n\t\tSearchAttributes:                    attributes.SearchAttributes,\n\t\tJitterStartSeconds:                  attributes.JitterStartSeconds,\n\t\tCronOverlapPolicy:                   attributes.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        attributes.ActiveClusterSelectionPolicy,\n\t}\n\n\treq := &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID:                      e.domainEntry.GetInfo().ID,\n\t\tStartRequest:                    createRequest,\n\t\tParentExecutionInfo:             parentExecutionInfo,\n\t\tLastCompletionResult:            attributes.LastCompletionResult,\n\t\tContinuedFailureReason:          attributes.FailureReason,\n\t\tContinuedFailureDetails:         attributes.FailureDetails,\n\t\tContinueAsNewInitiator:          attributes.Initiator,\n\t\tFirstDecisionTaskBackoffSeconds: attributes.BackoffStartIntervalInSeconds,\n\t\tPartitionConfig:                 previousExecutionInfo.PartitionConfig,\n\t}\n\n\t// if ContinueAsNew as Cron or decider, recalculate the expiration timestamp and set attempts to 0\n\treq.Attempt = 0\n\tif attributes.RetryPolicy != nil && attributes.RetryPolicy.GetExpirationIntervalInSeconds() > 0 {\n\t\t// expirationTime calculates from first decision task schedule to the end of the workflow\n\t\texpirationInSeconds := attributes.RetryPolicy.GetExpirationIntervalInSeconds() + req.GetFirstDecisionTaskBackoffSeconds()\n\t\texpirationTime := e.timeSource.Now().Add(time.Second * time.Duration(expirationInSeconds))\n\t\treq.ExpirationTimestamp = common.Int64Ptr(expirationTime.UnixNano())\n\t}\n\t// if ContinueAsNew as retry use the same expiration timestamp and increment attempts from previous execution state\n\tif attributes.GetInitiator() == types.ContinueAsNewInitiatorRetryPolicy {\n\t\treq.Attempt = previousExecutionState.GetExecutionInfo().Attempt + 1\n\t\texpirationTime := previousExecutionState.GetExecutionInfo().ExpirationTime\n\t\tif !expirationTime.IsZero() {\n\t\t\treq.ExpirationTimestamp = common.Int64Ptr(expirationTime.UnixNano())\n\t\t}\n\t}\n\n\t// History event only has domainName so domainID has to be passed in explicitly to update the mutable state\n\tvar parentDomainID *string\n\tif parentExecutionInfo != nil {\n\t\tparentDomainID = &parentExecutionInfo.DomainUUID\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionStartedEvent(req, previousExecutionInfo, firstRunID, execution.GetRunID(), firstScheduledTime)\n\tif err := e.ReplicateWorkflowExecutionStartedEvent(\n\t\tparentDomainID,\n\t\texecution,\n\t\tcreateRequest.GetRequestID(),\n\t\tevent,\n\t\tfalse,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := e.SetHistoryTree(e.GetExecutionInfo().RunID); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := e.AddFirstDecisionTaskScheduled(\n\t\tevent,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) AddWorkflowExecutionStartedEvent(\n\texecution types.WorkflowExecution,\n\tstartRequest *types.HistoryStartWorkflowExecutionRequest,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowStarted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\trequest := startRequest.StartRequest\n\teventID := e.GetNextEventID()\n\tif eventID != constants.FirstEventID {\n\t\te.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(eventID),\n\t\t\ttag.ErrorTypeInvalidHistoryAction)\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddWorkflowExecutionStartedEvent(startRequest, nil, execution.GetRunID(), execution.GetRunID(),\n\t\ttime.Now())\n\n\tvar parentDomainID *string\n\tif startRequest.ParentExecutionInfo != nil {\n\t\tparentDomainID = &startRequest.ParentExecutionInfo.DomainUUID\n\t}\n\tif err := e.ReplicateWorkflowExecutionStartedEvent(\n\t\tparentDomainID,\n\t\texecution,\n\t\trequest.GetRequestID(),\n\t\tevent,\n\t\tfalse); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionStartedEvent(\n\tparentDomainID *string,\n\texecution types.WorkflowExecution,\n\trequestID string,\n\tstartEvent *types.HistoryEvent,\n\tgenerateDelayedDecisionTasks bool,\n) error {\n\n\tevent := startEvent.WorkflowExecutionStartedEventAttributes\n\tif event.GetRequestID() != \"\" {\n\t\t// prefer requestID from history event, ideally we should remove the requestID parameter\n\t\t// removing it may or may not be backward compatible, so keep it now\n\t\trequestID = event.GetRequestID()\n\t}\n\te.executionInfo.CreateRequestID = requestID\n\n\te.insertWorkflowRequest(persistence.WorkflowRequest{\n\t\tRequestID:   requestID,\n\t\tVersion:     startEvent.Version,\n\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t})\n\te.executionInfo.DomainID = e.domainEntry.GetInfo().ID\n\te.executionInfo.WorkflowID = execution.GetWorkflowID()\n\te.executionInfo.RunID = execution.GetRunID()\n\te.executionInfo.FirstExecutionRunID = event.GetFirstExecutionRunID()\n\tif e.executionInfo.FirstExecutionRunID == \"\" {\n\t\te.executionInfo.FirstExecutionRunID = execution.GetRunID()\n\t}\n\te.executionInfo.TaskList = event.TaskList.GetName()\n\te.executionInfo.TaskListKind = event.TaskList.GetKind()\n\te.executionInfo.WorkflowTypeName = event.WorkflowType.GetName()\n\te.executionInfo.WorkflowTimeout = event.GetExecutionStartToCloseTimeoutSeconds()\n\te.executionInfo.DecisionStartToCloseTimeout = event.GetTaskStartToCloseTimeoutSeconds()\n\te.executionInfo.StartTimestamp = e.unixNanoToTime(startEvent.GetTimestamp())\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCreated,\n\t\tpersistence.WorkflowCloseStatusNone,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.LastProcessedEvent = constants.EmptyEventID\n\te.executionInfo.LastFirstEventID = startEvent.ID\n\n\te.executionInfo.DecisionVersion = constants.EmptyVersion\n\te.executionInfo.DecisionScheduleID = constants.EmptyEventID\n\te.executionInfo.DecisionStartedID = constants.EmptyEventID\n\te.executionInfo.DecisionRequestID = constants.EmptyUUID\n\te.executionInfo.DecisionTimeout = 0\n\n\te.executionInfo.CronSchedule = event.GetCronSchedule()\n\tif event.CronOverlapPolicy != nil {\n\t\te.executionInfo.CronOverlapPolicy = *event.CronOverlapPolicy\n\t}\n\n\tif parentDomainID != nil {\n\t\te.executionInfo.ParentDomainID = *parentDomainID\n\t}\n\tif event.ParentWorkflowExecution != nil {\n\t\te.executionInfo.ParentWorkflowID = event.ParentWorkflowExecution.GetWorkflowID()\n\t\te.executionInfo.ParentRunID = event.ParentWorkflowExecution.GetRunID()\n\t}\n\tif event.ParentInitiatedEventID != nil {\n\t\te.executionInfo.InitiatedID = event.GetParentInitiatedEventID()\n\t} else {\n\t\te.executionInfo.InitiatedID = constants.EmptyEventID\n\t}\n\n\tif event.CronOverlapPolicy != nil {\n\t\te.executionInfo.CronOverlapPolicy = *event.CronOverlapPolicy\n\t}\n\n\te.executionInfo.Attempt = event.GetAttempt()\n\tif event.GetExpirationTimestamp() != 0 {\n\t\te.executionInfo.ExpirationTime = time.Unix(0, event.GetExpirationTimestamp())\n\t}\n\tif event.RetryPolicy != nil {\n\t\te.executionInfo.HasRetryPolicy = true\n\t\te.executionInfo.BackoffCoefficient = event.RetryPolicy.GetBackoffCoefficient()\n\t\te.executionInfo.ExpirationSeconds = event.RetryPolicy.GetExpirationIntervalInSeconds()\n\t\te.executionInfo.InitialInterval = event.RetryPolicy.GetInitialIntervalInSeconds()\n\t\te.executionInfo.MaximumAttempts = event.RetryPolicy.GetMaximumAttempts()\n\t\te.executionInfo.MaximumInterval = event.RetryPolicy.GetMaximumIntervalInSeconds()\n\t\te.executionInfo.NonRetriableErrors = event.RetryPolicy.NonRetriableErrorReasons\n\t}\n\n\te.executionInfo.AutoResetPoints = rolloverAutoResetPointsWithExpiringTime(\n\t\tevent.GetPrevAutoResetPoints(),\n\t\tevent.GetContinuedExecutionRunID(),\n\t\tstartEvent.GetTimestamp(),\n\t\te.domainEntry.GetRetentionDays(e.executionInfo.WorkflowID),\n\t)\n\n\tif event.Memo != nil {\n\t\te.executionInfo.Memo = event.Memo.GetFields()\n\t}\n\tif event.SearchAttributes != nil {\n\t\te.executionInfo.SearchAttributes = event.SearchAttributes.GetIndexedFields()\n\t}\n\te.executionInfo.PartitionConfig = event.PartitionConfig\n\te.executionInfo.ActiveClusterSelectionPolicy = event.ActiveClusterSelectionPolicy\n\n\te.writeEventToCache(startEvent)\n\n\tif err := e.taskGenerator.GenerateWorkflowStartTasks(e.unixNanoToTime(startEvent.GetTimestamp()), startEvent); err != nil {\n\t\treturn err\n\t}\n\tif err := e.taskGenerator.GenerateRecordWorkflowStartedTasks(startEvent); err != nil {\n\t\treturn err\n\t}\n\tif generateDelayedDecisionTasks && event.GetFirstDecisionTaskBackoffSeconds() > 0 {\n\t\tif err := e.taskGenerator.GenerateDelayedDecisionTasks(startEvent); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_timedout.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) AddTimeoutWorkflowEvent(\n\tfirstEventID int64,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionWorkflowTimeout\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tevent := e.hBuilder.AddTimeoutWorkflowEvent()\n\tif err := e.ReplicateWorkflowExecutionTimedoutEvent(firstEventID, event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateWorkflowExecutionTimedoutEvent(\n\tfirstEventID int64,\n\tevent *types.HistoryEvent,\n) error {\n\n\tif err := e.UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateCompleted,\n\t\tpersistence.WorkflowCloseStatusTimedOut,\n\t); err != nil {\n\t\treturn err\n\t}\n\te.executionInfo.CompletionEventBatchID = firstEventID // Used when completion event needs to be loaded from database\n\te.ClearStickyness()\n\te.writeEventToCache(event)\n\n\treturn e.taskGenerator.GenerateWorkflowCloseTasks(event, e.config.WorkflowDeletionJitterRange(e.domainEntry.GetInfo().Name))\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_timer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (e *mutableStateBuilder) GetPendingTimerInfos() map[string]*persistence.TimerInfo {\n\treturn e.pendingTimerInfoIDs\n}\n\nfunc (e *mutableStateBuilder) AddTimerStartedEvent(\n\tdecisionCompletedEventID int64,\n\trequest *types.StartTimerDecisionAttributes,\n) (*types.HistoryEvent, *persistence.TimerInfo, error) {\n\n\topTag := tag.WorkflowActionTimerStarted\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\ttimerID := request.GetTimerID()\n\t_, ok := e.GetUserTimerInfo(timerID)\n\tif ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowTimerID(timerID))\n\t\treturn nil, nil, e.createCallerError(opTag)\n\t}\n\n\tevent := e.hBuilder.AddTimerStartedEvent(decisionCompletedEventID, request)\n\tti, err := e.ReplicateTimerStartedEvent(event)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\treturn event, ti, err\n}\n\nfunc (e *mutableStateBuilder) ReplicateTimerStartedEvent(\n\tevent *types.HistoryEvent,\n) (*persistence.TimerInfo, error) {\n\n\tattributes := event.TimerStartedEventAttributes\n\ttimerID := attributes.GetTimerID()\n\n\tstartToFireTimeout := attributes.GetStartToFireTimeoutSeconds()\n\tfireTimeout := time.Duration(startToFireTimeout) * time.Second\n\t// TODO: Time skew need to be taken in to account.\n\texpiryTime := time.Unix(0, event.GetTimestamp()).Add(fireTimeout) // should use the event time, not now\n\tti := &persistence.TimerInfo{\n\t\tVersion:    event.Version,\n\t\tTimerID:    timerID,\n\t\tExpiryTime: expiryTime,\n\t\tStartedID:  event.ID,\n\t\tTaskStatus: TimerTaskStatusNone,\n\t}\n\n\te.pendingTimerInfoIDs[ti.TimerID] = ti\n\te.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\te.updateTimerInfos[ti.TimerID] = ti\n\n\treturn ti, nil\n}\n\nfunc (e *mutableStateBuilder) AddTimerFiredEvent(\n\ttimerID string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionTimerFired\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\ttimerInfo, ok := e.GetUserTimerInfo(timerID)\n\tif !ok {\n\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowTimerID(timerID))\n\t\treturn nil, e.createInternalServerError(opTag)\n\t}\n\n\t// Timer is running.\n\tevent := e.hBuilder.AddTimerFiredEvent(timerInfo.StartedID, timerID)\n\tif err := e.ReplicateTimerFiredEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateTimerFiredEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.TimerFiredEventAttributes\n\ttimerID := attributes.GetTimerID()\n\n\treturn e.DeleteUserTimer(timerID)\n}\n\nfunc (e *mutableStateBuilder) AddTimerCanceledEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.CancelTimerDecisionAttributes,\n\tidentity string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionTimerCanceled\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar timerStartedID int64\n\ttimerID := attributes.GetTimerID()\n\tti, ok := e.GetUserTimerInfo(timerID)\n\tif !ok {\n\t\t// if timer is not running then check if it has fired in the mutable state.\n\t\t// If so clear the timer from the mutable state. We need to check both the\n\t\t// bufferedEvents and the history builder\n\t\ttimerFiredEvent := e.checkAndClearTimerFiredEvent(timerID)\n\t\tif timerFiredEvent == nil {\n\t\t\te.logWarn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\t\ttag.WorkflowEventID(e.GetNextEventID()),\n\t\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\t\ttag.WorkflowTimerID(timerID))\n\t\t\treturn nil, e.createCallerError(opTag)\n\t\t}\n\t\ttimerStartedID = timerFiredEvent.TimerFiredEventAttributes.GetStartedEventID()\n\t} else {\n\t\ttimerStartedID = ti.StartedID\n\t}\n\n\t// Timer is running.\n\tevent := e.hBuilder.AddTimerCanceledEvent(timerStartedID, decisionCompletedEventID, timerID, identity)\n\tif ok {\n\t\tif err := e.ReplicateTimerCanceledEvent(event); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn event, nil\n}\n\nfunc (e *mutableStateBuilder) ReplicateTimerCanceledEvent(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattributes := event.TimerCanceledEventAttributes\n\ttimerID := attributes.GetTimerID()\n\n\treturn e.DeleteUserTimer(timerID)\n}\n\nfunc (e *mutableStateBuilder) AddCancelTimerFailedEvent(\n\tdecisionCompletedEventID int64,\n\tattributes *types.CancelTimerDecisionAttributes,\n\tidentity string,\n) (*types.HistoryEvent, error) {\n\n\topTag := tag.WorkflowActionTimerCancelFailed\n\tif err := e.checkMutability(opTag); err != nil {\n\t\treturn nil, err\n\t}\n\n\t// No Operation: We couldn't cancel it probably TIMER_ID_UNKNOWN\n\ttimerID := attributes.GetTimerID()\n\treturn e.hBuilder.AddCancelTimerFailedEvent(timerID, decisionCompletedEventID,\n\t\ttimerCancellationMsgTimerIDUnknown, identity), nil\n}\n\n// GetUserTimerInfo gives details about a user timer.\nfunc (e *mutableStateBuilder) GetUserTimerInfo(\n\ttimerID string,\n) (*persistence.TimerInfo, bool) {\n\n\ttimerInfo, ok := e.pendingTimerInfoIDs[timerID]\n\treturn timerInfo, ok\n}\n\n// UpdateUserTimer updates the user timer in progress.\nfunc (e *mutableStateBuilder) UpdateUserTimer(\n\tti *persistence.TimerInfo,\n) error {\n\n\ttimerID, ok := e.pendingTimerEventIDToID[ti.StartedID]\n\tif !ok {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find timer event ID: %v in mutable state\", ti.StartedID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\treturn ErrMissingTimerInfo\n\t}\n\n\tif _, ok := e.pendingTimerInfoIDs[timerID]; !ok {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find timer ID: %v in mutable state\", timerID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\treturn ErrMissingTimerInfo\n\t}\n\n\te.pendingTimerInfoIDs[ti.TimerID] = ti\n\te.updateTimerInfos[ti.TimerID] = ti\n\treturn nil\n}\n\n// DeleteUserTimer deletes an user timer.\nfunc (e *mutableStateBuilder) DeleteUserTimer(\n\ttimerID string,\n) error {\n\n\tif timerInfo, ok := e.pendingTimerInfoIDs[timerID]; ok {\n\t\tdelete(e.pendingTimerInfoIDs, timerID)\n\n\t\tif _, ok = e.pendingTimerEventIDToID[timerInfo.StartedID]; ok {\n\t\t\tdelete(e.pendingTimerEventIDToID, timerInfo.StartedID)\n\t\t} else {\n\t\t\te.logError(\n\t\t\t\tfmt.Sprintf(\"unable to find timer event ID: %v in mutable state\", timerID),\n\t\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t\t)\n\t\t\t// log data inconsistency instead of returning an error\n\t\t\te.logDataInconsistency()\n\t\t}\n\t} else {\n\t\te.logError(\n\t\t\tfmt.Sprintf(\"unable to find timer ID: %v in mutable state\", timerID),\n\t\t\ttag.ErrorTypeInvalidMutableStateAction,\n\t\t)\n\t\t// log data inconsistency instead of returning an error\n\t\te.logDataInconsistency()\n\t}\n\n\tdelete(e.updateTimerInfos, timerID)\n\te.deleteTimerInfos[timerID] = struct{}{}\n\treturn nil\n}\n\nfunc checkAndClearTimerFiredEvent(\n\tevents []*types.HistoryEvent,\n\ttimerID string,\n) ([]*types.HistoryEvent, *types.HistoryEvent) {\n\t// go over all history events. if we find a timer fired event for the given\n\t// timerID, clear it\n\ttimerFiredIdx := -1\n\tfor idx, event := range events {\n\t\tif event.GetEventType() == types.EventTypeTimerFired &&\n\t\t\tevent.GetTimerFiredEventAttributes().GetTimerID() == timerID {\n\t\t\ttimerFiredIdx = idx\n\t\t\tbreak\n\t\t}\n\t}\n\tif timerFiredIdx == -1 {\n\t\treturn events, nil\n\t}\n\n\ttimerEvent := events[timerFiredIdx]\n\treturn append(events[:timerFiredIdx], events[timerFiredIdx+1:]...), timerEvent\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_methods_timer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc Test__checkAndClearTimerFiredEvent(t *testing.T) {\n\tt.Run(\"no timer fired event to clear\", func(t *testing.T) {\n\t\tevents := []*types.HistoryEvent{{\n\t\t\tID:        1,\n\t\t\tTimestamp: nil,\n\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t}}\n\t\tremainingEvents, timerEvent := checkAndClearTimerFiredEvent(events, \"1\")\n\t\tassert.Nil(t, timerEvent)\n\t\tassert.Equal(t, len(events), len(remainingEvents))\n\t})\n\tt.Run(\"timer fired event cleared\", func(t *testing.T) {\n\t\ttimerEvent := &types.HistoryEvent{\n\t\t\tID:        2,\n\t\t\tTimestamp: nil,\n\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\tTimerID:        \"101\",\n\t\t\t\tStartedEventID: 0,\n\t\t\t},\n\t\t}\n\t\tevents := []*types.HistoryEvent{{\n\t\t\tID:        1,\n\t\t\tTimestamp: nil,\n\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t},\n\t\t\ttimerEvent,\n\t\t}\n\t\tremainingEvents, clearedEvent := checkAndClearTimerFiredEvent(events, timerEvent.TimerFiredEventAttributes.TimerID)\n\t\tassert.NotNil(t, timerEvent)\n\t\tassert.Equal(t, timerEvent, clearedEvent)\n\t\tassert.Equal(t, len(events)-1, len(remainingEvents))\n\t})\n}\n\nfunc Test__DeleteUserTimer(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tti := &persistence.TimerInfo{\n\t\tTimerID:   \"101\",\n\t\tStartedID: 1,\n\t}\n\tmb.pendingTimerInfoIDs[ti.TimerID] = ti\n\tmb.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\terr := mb.DeleteUserTimer(ti.TimerID)\n\tassert.NoError(t, err)\n}\n\nfunc Test__UpdateUserTimer(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tti := &persistence.TimerInfo{\n\t\tTimerID:   \"101\",\n\t\tStartedID: 1,\n\t}\n\tt.Run(\"missing timer info\", func(t *testing.T) {\n\t\terr := mb.UpdateUserTimer(ti)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrMissingTimerInfo, err)\n\n\t\tmb.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\t\terr = mb.UpdateUserTimer(ti)\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, ErrMissingTimerInfo, err)\n\t})\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmb.pendingTimerInfoIDs[ti.TimerID] = ti\n\t\tmb.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\t\terr := mb.UpdateUserTimer(ti)\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc Test__GetUserTimerInfo(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tti := &persistence.TimerInfo{\n\t\tTimerID:   \"101\",\n\t\tStartedID: 1,\n\t}\n\tmb.pendingTimerInfoIDs[ti.TimerID] = ti\n\tinfo, ok := mb.GetUserTimerInfo(ti.TimerID)\n\tassert.Equal(t, ti, info)\n\tassert.True(t, ok)\n}\n\nfunc Test__ReplicateTimerCanceledEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\ttimerEvent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tTimestamp: nil,\n\t\tEventType: types.EventTypeTimerCanceled.Ptr(),\n\t\tTimerCanceledEventAttributes: &types.TimerCanceledEventAttributes{\n\t\t\tTimerID:        \"101\",\n\t\t\tStartedEventID: 0,\n\t\t},\n\t}\n\tti := &persistence.TimerInfo{\n\t\tTimerID:   \"101\",\n\t\tStartedID: 1,\n\t}\n\tmb.pendingTimerInfoIDs[ti.TimerID] = ti\n\tmb.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\terr := mb.ReplicateTimerCanceledEvent(timerEvent)\n\tassert.NoError(t, err)\n}\n\nfunc Test__ReplicateTimerFiredEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\ttimerEvent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tTimestamp: nil,\n\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\tTimerID:        \"101\",\n\t\t\tStartedEventID: 0,\n\t\t},\n\t}\n\tti := &persistence.TimerInfo{\n\t\tTimerID:   \"101\",\n\t\tStartedID: 1,\n\t}\n\tmb.pendingTimerInfoIDs[ti.TimerID] = ti\n\tmb.pendingTimerEventIDToID[ti.StartedID] = ti.TimerID\n\terr := mb.ReplicateTimerFiredEvent(timerEvent)\n\tassert.NoError(t, err)\n}\n\nfunc Test__ReplicateTimerStartedEvent(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tstartToFireTimeoutSeconds := int64(5)\n\tnow := time.Now()\n\tnowUnix := now.UnixNano()\n\ttimerEvent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tVersion:   0,\n\t\tTimestamp: &nowUnix,\n\t\tEventType: types.EventTypeTimerStarted.Ptr(),\n\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{\n\t\t\tTimerID:                   \"101\",\n\t\t\tStartToFireTimeoutSeconds: &startToFireTimeoutSeconds,\n\t\t},\n\t}\n\texpectedTimerInfo := &persistence.TimerInfo{\n\t\tVersion:    0,\n\t\tTimerID:    \"101\",\n\t\tStartedID:  1,\n\t\tExpiryTime: now.Add(time.Second * time.Duration(int64(5))),\n\t}\n\n\tti, err := mb.ReplicateTimerStartedEvent(timerEvent)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expectedTimerInfo.ExpiryTime.UTC(), ti.ExpiryTime.UTC())\n\tassert.Equal(t, expectedTimerInfo.TimerID, ti.TimerID)\n}\n\nfunc Test__GetPendingTimerInfos(t *testing.T) {\n\tmb := testMutableStateBuilder(t)\n\tpendingTimerInfo := map[string]*persistence.TimerInfo{\n\t\t\"101\": {\n\t\t\tVersion:   0,\n\t\t\tTimerID:   \"101\",\n\t\t\tStartedID: 1,\n\t\t},\n\t}\n\tmb.pendingTimerInfoIDs = pendingTimerInfo\n\tresult := mb.GetPendingTimerInfos()\n\tassert.Equal(t, pendingTimerInfo, result)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_builder_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/checksum\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/testing/testdatagen\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/query\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\tshardCtx \"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tmutableStateSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tmockShard       *shard.TestContext\n\t\tmockEventsCache *events.MockCache\n\n\t\tmsBuilder *mutableStateBuilder\n\t\tlogger    log.Logger\n\t\ttestScope tally.TestScope\n\t}\n)\n\nfunc TestMutableStateSuite(t *testing.T) {\n\ts := new(mutableStateSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *mutableStateSuite) SetupSuite() {\n\n}\n\nfunc (s *mutableStateSuite) TearDownSuite() {\n\n}\n\nfunc (s *mutableStateSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\t// set the checksum probabilities to 100% for exercising during test\n\ts.mockShard.GetConfig().MutableStateChecksumGenProbability = func(domain string) int { return 100 }\n\ts.mockShard.GetConfig().MutableStateChecksumVerifyProbability = func(domain string) int { return 100 }\n\ts.mockShard.GetConfig().EnableRetryForChecksumFailure = func(domain string) bool { return true }\n\n\ts.mockEventsCache = s.mockShard.GetEventsCache().(*events.MockCache)\n\n\ts.testScope = s.mockShard.Resource.MetricsScope\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\n\ts.msBuilder = newMutableStateBuilder(s.mockShard, s.logger, constants.TestLocalDomainEntry, constants.TestLocalDomainEntry.GetFailoverVersion())\n}\n\nfunc (s *mutableStateSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *mutableStateSuite) TestErrorReturnedWhenSchedulingTooManyPendingActivities() {\n\tfor i := 0; i < s.msBuilder.config.PendingActivitiesCountLimitError(); i++ {\n\t\ts.msBuilder.pendingActivityInfoIDs[int64(i)] = &persistence.ActivityInfo{}\n\t}\n\n\t_, _, _, _, _, err := s.msBuilder.AddActivityTaskScheduledEvent(nil, 1, &types.ScheduleActivityTaskDecisionAttributes{}, false)\n\tassert.Equal(s.T(), \"Too many pending activities\", err.Error())\n}\n\nfunc (s *mutableStateSuite) TestTransientDecisionCompletionFirstBatchReplicated_ReplicateDecisionCompleted() {\n\tversion := int64(12)\n\trunID := uuid.New()\n\ts.msBuilder = NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\tversion,\n\t\trunID,\n\t\tconstants.TestGlobalDomainEntry,\n\t).(*mutableStateBuilder)\n\n\tnewDecisionScheduleEvent, newDecisionStartedEvent := s.prepareTransientDecisionCompletionFirstBatchReplicated(version, runID)\n\n\tnewDecisionCompletedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        newDecisionStartedEvent.ID + 1,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\tScheduledEventID: newDecisionScheduleEvent.ID,\n\t\t\tStartedEventID:   newDecisionStartedEvent.ID,\n\t\t\tIdentity:         \"some random identity\",\n\t\t},\n\t}\n\terr := s.msBuilder.ReplicateDecisionTaskCompletedEvent(newDecisionCompletedEvent)\n\ts.NoError(err)\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().transientHistory))\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().history))\n}\n\nfunc (s *mutableStateSuite) TestTransientDecisionCompletionFirstBatchReplicated_FailoverDecisionTimeout() {\n\tversion := int64(12)\n\trunID := uuid.New()\n\ts.msBuilder = NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\tversion,\n\t\trunID,\n\t\tconstants.TestGlobalDomainEntry,\n\t).(*mutableStateBuilder)\n\n\tnewDecisionScheduleEvent, newDecisionStartedEvent := s.prepareTransientDecisionCompletionFirstBatchReplicated(version, runID)\n\n\ts.NotNil(s.msBuilder.AddDecisionTaskTimedOutEvent(newDecisionScheduleEvent.ID, newDecisionStartedEvent.ID))\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().transientHistory))\n\ts.Equal(1, len(s.msBuilder.GetHistoryBuilder().history))\n}\n\nfunc (s *mutableStateSuite) TestTransientDecisionCompletionFirstBatchReplicated_FailoverDecisionFailed() {\n\tversion := int64(12)\n\trunID := uuid.New()\n\ts.msBuilder = NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\tversion,\n\t\trunID,\n\t\tconstants.TestGlobalDomainEntry,\n\t).(*mutableStateBuilder)\n\n\tnewDecisionScheduleEvent, newDecisionStartedEvent := s.prepareTransientDecisionCompletionFirstBatchReplicated(version, runID)\n\n\ts.NotNil(s.msBuilder.AddDecisionTaskFailedEvent(\n\t\tnewDecisionScheduleEvent.ID,\n\t\tnewDecisionStartedEvent.ID,\n\t\ttypes.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure,\n\t\t[]byte(\"some random decision failure details\"),\n\t\t\"some random decision failure identity\",\n\t\t\"\", \"\", \"\", \"\", 0, \"\",\n\t))\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().transientHistory))\n\ts.Equal(1, len(s.msBuilder.GetHistoryBuilder().history))\n}\n\nfunc (s *mutableStateSuite) TestShouldBufferEvent() {\n\t// workflow status events will be assign event ID immediately\n\tworkflowEvents := map[types.EventType]bool{\n\t\ttypes.EventTypeWorkflowExecutionStarted:        true,\n\t\ttypes.EventTypeWorkflowExecutionCompleted:      true,\n\t\ttypes.EventTypeWorkflowExecutionFailed:         true,\n\t\ttypes.EventTypeWorkflowExecutionTimedOut:       true,\n\t\ttypes.EventTypeWorkflowExecutionTerminated:     true,\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew: true,\n\t\ttypes.EventTypeWorkflowExecutionCanceled:       true,\n\t}\n\n\t// decision events will be assign event ID immediately\n\tdecisionTaskEvents := map[types.EventType]bool{\n\t\ttypes.EventTypeDecisionTaskScheduled: true,\n\t\ttypes.EventTypeDecisionTaskStarted:   true,\n\t\ttypes.EventTypeDecisionTaskCompleted: true,\n\t\ttypes.EventTypeDecisionTaskFailed:    true,\n\t\ttypes.EventTypeDecisionTaskTimedOut:  true,\n\t}\n\n\t// events corresponding to decisions from client will be assign event ID immediately\n\tdecisionEvents := map[types.EventType]bool{\n\t\ttypes.EventTypeWorkflowExecutionCompleted:                      true,\n\t\ttypes.EventTypeWorkflowExecutionFailed:                         true,\n\t\ttypes.EventTypeWorkflowExecutionCanceled:                       true,\n\t\ttypes.EventTypeWorkflowExecutionContinuedAsNew:                 true,\n\t\ttypes.EventTypeActivityTaskScheduled:                           true,\n\t\ttypes.EventTypeActivityTaskCancelRequested:                     true,\n\t\ttypes.EventTypeTimerStarted:                                    true,\n\t\ttypes.EventTypeTimerCanceled:                                   true,\n\t\ttypes.EventTypeCancelTimerFailed:                               true,\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionInitiated: true,\n\t\ttypes.EventTypeMarkerRecorded:                                  true,\n\t\ttypes.EventTypeStartChildWorkflowExecutionInitiated:            true,\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionInitiated:        true,\n\t\ttypes.EventTypeUpsertWorkflowSearchAttributes:                  true,\n\t}\n\n\t// other events will not be assign event ID immediately\n\totherEvents := map[types.EventType]bool{}\nOtherEventsLoop:\n\tfor _, eventType := range types.EventTypeValues() {\n\t\tif _, ok := workflowEvents[eventType]; ok {\n\t\t\tcontinue OtherEventsLoop\n\t\t}\n\t\tif _, ok := decisionTaskEvents[eventType]; ok {\n\t\t\tcontinue OtherEventsLoop\n\t\t}\n\t\tif _, ok := decisionEvents[eventType]; ok {\n\t\t\tcontinue OtherEventsLoop\n\t\t}\n\t\totherEvents[eventType] = true\n\t}\n\n\t// test workflowEvents, decisionTaskEvents, decisionEvents will return true\n\tfor eventType := range workflowEvents {\n\t\ts.False(s.msBuilder.shouldBufferEvent(eventType))\n\t}\n\tfor eventType := range decisionTaskEvents {\n\t\ts.False(s.msBuilder.shouldBufferEvent(eventType))\n\t}\n\tfor eventType := range decisionEvents {\n\t\ts.False(s.msBuilder.shouldBufferEvent(eventType))\n\t}\n\t// other events will return false\n\tfor eventType := range otherEvents {\n\t\ts.True(s.msBuilder.shouldBufferEvent(eventType))\n\t}\n\n\t// +1 is because DecisionTypeCancelTimer will be mapped\n\t// to either types.EventTypeTimerCanceled, or types.EventTypeCancelTimerFailed.\n\ts.Equal(len(types.DecisionTypeValues())+1, len(decisionEvents),\n\t\t\"This assertaion will be broken a new decision is added and no corresponding logic added to shouldBufferEvent()\")\n}\n\nfunc (s *mutableStateSuite) TestReorderEvents() {\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tactivityID := \"activity_id\"\n\tactivityResult := []byte(\"activity_result\")\n\n\tinfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  we.GetWorkflowID(),\n\t\tRunID:                       we.GetRunID(),\n\t\tTaskList:                    tl,\n\t\tWorkflowTypeName:            \"wType\",\n\t\tWorkflowTimeout:             200,\n\t\tDecisionStartToCloseTimeout: 100,\n\t\tState:                       persistence.WorkflowStateRunning,\n\t\tCloseStatus:                 persistence.WorkflowCloseStatusNone,\n\t\tNextEventID:                 int64(8),\n\t\tLastProcessedEvent:          int64(3),\n\t\tLastUpdatedTimestamp:        time.Now(),\n\t\tDecisionVersion:             commonconstants.EmptyVersion,\n\t\tDecisionScheduleID:          commonconstants.EmptyEventID,\n\t\tDecisionStartedID:           commonconstants.EmptyEventID,\n\t\tDecisionTimeout:             100,\n\t}\n\n\tactivityInfos := map[int64]*persistence.ActivityInfo{\n\t\t5: {\n\t\t\tVersion:                int64(1),\n\t\t\tScheduleID:             int64(5),\n\t\t\tScheduledTime:          time.Now(),\n\t\t\tStartedID:              commonconstants.EmptyEventID,\n\t\t\tStartedTime:            time.Now(),\n\t\t\tActivityID:             activityID,\n\t\t\tScheduleToStartTimeout: 100,\n\t\t\tScheduleToCloseTimeout: 200,\n\t\t\tStartToCloseTimeout:    300,\n\t\t\tHeartbeatTimeout:       50,\n\t\t},\n\t}\n\n\tbufferedEvents := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\t\tVersion:   1,\n\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\tResult:           []byte(activityResult),\n\t\t\t\tScheduledEventID: 5,\n\t\t\t\tStartedEventID:   commonconstants.BufferedEventID,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\tVersion:   1,\n\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\tScheduledEventID: 5,\n\t\t\t},\n\t\t},\n\t}\n\n\tdbState := &persistence.WorkflowMutableState{\n\t\tExecutionInfo:  info,\n\t\tActivityInfos:  activityInfos,\n\t\tBufferedEvents: bufferedEvents,\n\t}\n\n\terr := s.msBuilder.Load(context.Background(), dbState)\n\ts.Nil(err)\n\n\ts.Equal(types.EventTypeActivityTaskStarted, s.msBuilder.bufferedEvents[0].GetEventType())\n\ts.Equal(int64(-123), s.msBuilder.bufferedEvents[0].ID)\n\ts.Equal(int64(5), s.msBuilder.bufferedEvents[0].ActivityTaskStartedEventAttributes.GetScheduledEventID())\n\ts.Equal(types.EventTypeActivityTaskCompleted, s.msBuilder.bufferedEvents[1].GetEventType())\n\ts.Equal(int64(-123), s.msBuilder.bufferedEvents[1].ID)\n\ts.Equal(int64(-123), s.msBuilder.bufferedEvents[1].ActivityTaskCompletedEventAttributes.GetStartedEventID())\n\ts.Equal(int64(5), s.msBuilder.bufferedEvents[1].ActivityTaskCompletedEventAttributes.GetScheduledEventID())\n\n\terr = s.msBuilder.FlushBufferedEvents()\n\ts.Nil(err)\n\n\ts.Equal(types.EventTypeActivityTaskStarted, s.msBuilder.hBuilder.history[0].GetEventType())\n\ts.Equal(int64(8), s.msBuilder.hBuilder.history[0].ID)\n\ts.Equal(int64(5), s.msBuilder.hBuilder.history[0].ActivityTaskStartedEventAttributes.GetScheduledEventID())\n\ts.Equal(types.EventTypeActivityTaskCompleted, s.msBuilder.hBuilder.history[1].GetEventType())\n\ts.Equal(int64(9), s.msBuilder.hBuilder.history[1].ID)\n\ts.Equal(int64(8), s.msBuilder.hBuilder.history[1].ActivityTaskCompletedEventAttributes.GetStartedEventID())\n\ts.Equal(int64(5), s.msBuilder.hBuilder.history[1].ActivityTaskCompletedEventAttributes.GetScheduledEventID())\n}\n\nfunc (s *mutableStateSuite) TestChecksum() {\n\ttestCases := []struct {\n\t\tname                 string\n\t\tenableBufferedEvents bool\n\t\tcloseTxFunc          func(ms *mutableStateBuilder) (checksum.Checksum, error)\n\t}{\n\t\t{\n\t\t\tname: \"closeTransactionAsSnapshot\",\n\t\t\tcloseTxFunc: func(ms *mutableStateBuilder) (checksum.Checksum, error) {\n\t\t\t\tsnapshot, _, err := ms.CloseTransactionAsSnapshot(time.Now(), TransactionPolicyPassive)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn checksum.Checksum{}, err\n\t\t\t\t}\n\t\t\t\treturn snapshot.Checksum, err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                 \"closeTransactionAsMutation\",\n\t\t\tenableBufferedEvents: true,\n\t\t\tcloseTxFunc: func(ms *mutableStateBuilder) (checksum.Checksum, error) {\n\t\t\t\tmutation, _, err := ms.CloseTransactionAsMutation(time.Now(), TransactionPolicyPassive)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn checksum.Checksum{}, err\n\t\t\t\t}\n\t\t\t\treturn mutation.Checksum, err\n\t\t\t},\n\t\t},\n\t}\n\n\tloadErrorsFunc := func() int64 {\n\t\tcounter := s.testScope.Snapshot().Counters()[\"test.mutable_state_checksum_mismatch+operation=WorkflowContext\"]\n\t\tif counter != nil {\n\t\t\treturn counter.Value()\n\t\t}\n\t\treturn 0\n\t}\n\n\tvar loadErrors int64\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\tdbState := buildWorkflowMutableState()\n\t\t\tif !tc.enableBufferedEvents {\n\t\t\t\tdbState.BufferedEvents = nil\n\t\t\t}\n\n\t\t\t// create mutable state and verify checksum is generated on close\n\t\t\tloadErrors = loadErrorsFunc()\n\t\t\ts.msBuilder.Load(context.Background(), dbState)\n\t\t\ts.Equal(loadErrors, loadErrorsFunc()) // no errors expected\n\t\t\ts.EqualValues(dbState.Checksum, s.msBuilder.checksum)\n\t\t\ts.msBuilder.domainEntry = s.newDomainCacheEntry()\n\t\t\tcsum, err := tc.closeTxFunc(s.msBuilder)\n\t\t\ts.Nil(err)\n\t\t\ts.NotNil(csum.Value)\n\t\t\ts.Equal(checksum.FlavorIEEECRC32OverThriftBinary, csum.Flavor)\n\t\t\ts.Equal(mutableStateChecksumPayloadV1, csum.Version)\n\t\t\ts.EqualValues(csum, s.msBuilder.checksum)\n\n\t\t\t// verify checksum is verified on Load\n\t\t\tdbState.Checksum = csum\n\t\t\terr = s.msBuilder.Load(context.Background(), dbState)\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(loadErrors, loadErrorsFunc())\n\n\t\t\t// generate checksum again and verify its the same\n\t\t\tcsum, err = tc.closeTxFunc(s.msBuilder)\n\t\t\ts.Nil(err)\n\t\t\ts.NotNil(csum.Value)\n\t\t\ts.Equal(dbState.Checksum.Value, csum.Value)\n\n\t\t\t// modify checksum and verify Load fails\n\t\t\tdbState.Checksum.Value[0]++\n\t\t\terr = s.msBuilder.Load(context.Background(), dbState)\n\t\t\ts.Error(err)\n\t\t\ts.Equal(loadErrors+1, loadErrorsFunc())\n\t\t\ts.EqualValues(dbState.Checksum, s.msBuilder.checksum)\n\n\t\t\t// test checksum is invalidated\n\t\t\tloadErrors = loadErrorsFunc()\n\t\t\ts.mockShard.GetConfig().MutableStateChecksumInvalidateBefore = func(...dynamicproperties.FilterOption) float64 {\n\t\t\t\treturn float64((s.msBuilder.executionInfo.LastUpdatedTimestamp.UnixNano() / int64(time.Second)) + 1)\n\t\t\t}\n\t\t\terr = s.msBuilder.Load(context.Background(), dbState)\n\t\t\ts.NoError(err)\n\t\t\ts.Equal(loadErrors, loadErrorsFunc())\n\t\t\ts.EqualValues(checksum.Checksum{}, s.msBuilder.checksum)\n\n\t\t\t// revert the config value for the next test case\n\t\t\ts.mockShard.GetConfig().MutableStateChecksumInvalidateBefore = func(...dynamicproperties.FilterOption) float64 {\n\t\t\t\treturn float64(0)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateSuite) TestChecksumProbabilities() {\n\tfor _, prob := range []int{0, 100} {\n\t\ts.mockShard.GetConfig().MutableStateChecksumGenProbability = func(domain string) int { return prob }\n\t\ts.mockShard.GetConfig().MutableStateChecksumVerifyProbability = func(domain string) int { return prob }\n\t\tfor i := 0; i < 100; i++ {\n\t\t\tshouldGenerate := s.msBuilder.shouldGenerateChecksum()\n\t\t\tshouldVerify := s.msBuilder.shouldVerifyChecksum()\n\t\t\ts.Equal(prob == 100, shouldGenerate)\n\t\t\ts.Equal(prob == 100, shouldVerify)\n\t\t}\n\t}\n}\n\nfunc (s *mutableStateSuite) TestChecksumShouldInvalidate() {\n\ts.mockShard.GetConfig().MutableStateChecksumInvalidateBefore = func(...dynamicproperties.FilterOption) float64 { return 0 }\n\ts.False(s.msBuilder.shouldInvalidateChecksum())\n\ts.msBuilder.executionInfo.LastUpdatedTimestamp = time.Now()\n\ts.mockShard.GetConfig().MutableStateChecksumInvalidateBefore = func(...dynamicproperties.FilterOption) float64 {\n\t\treturn float64((s.msBuilder.executionInfo.LastUpdatedTimestamp.UnixNano() / int64(time.Second)) + 1)\n\t}\n\ts.True(s.msBuilder.shouldInvalidateChecksum())\n\ts.mockShard.GetConfig().MutableStateChecksumInvalidateBefore = func(...dynamicproperties.FilterOption) float64 {\n\t\treturn float64((s.msBuilder.executionInfo.LastUpdatedTimestamp.UnixNano() / int64(time.Second)) - 1)\n\t}\n\ts.False(s.msBuilder.shouldInvalidateChecksum())\n}\n\nfunc (s *mutableStateSuite) TestTrimEvents() {\n\tvar input []*types.HistoryEvent\n\toutput := s.msBuilder.trimEventsAfterWorkflowClose(input)\n\ts.Equal(input, output)\n\n\tinput = []*types.HistoryEvent{}\n\toutput = s.msBuilder.trimEventsAfterWorkflowClose(input)\n\ts.Equal(input, output)\n\n\tinput = []*types.HistoryEvent{\n\t\t{\n\t\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t},\n\t\t{\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t},\n\t}\n\toutput = s.msBuilder.trimEventsAfterWorkflowClose(input)\n\ts.Equal(input, output)\n\n\tinput = []*types.HistoryEvent{\n\t\t{\n\t\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t},\n\t\t{\n\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t},\n\t}\n\toutput = s.msBuilder.trimEventsAfterWorkflowClose(input)\n\ts.Equal(input, output)\n\n\tinput = []*types.HistoryEvent{\n\t\t{\n\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t},\n\t\t{\n\t\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t},\n\t}\n\toutput = s.msBuilder.trimEventsAfterWorkflowClose(input)\n\ts.Equal([]*types.HistoryEvent{\n\t\t{\n\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t},\n\t}, output)\n}\n\nfunc (s *mutableStateSuite) TestMergeMapOfByteArray() {\n\tvar currentMap map[string][]byte\n\tvar newMap map[string][]byte\n\tresultMap := mergeMapOfByteArray(currentMap, newMap)\n\ts.Equal(make(map[string][]byte), resultMap)\n\n\tnewMap = map[string][]byte{\"key\": []byte(\"val\")}\n\tresultMap = mergeMapOfByteArray(currentMap, newMap)\n\ts.Equal(newMap, resultMap)\n\n\tcurrentMap = map[string][]byte{\"number\": []byte(\"1\")}\n\tresultMap = mergeMapOfByteArray(currentMap, newMap)\n\ts.Equal(2, len(resultMap))\n}\n\nfunc (s *mutableStateSuite) TestEventReapplied() {\n\trunID := uuid.New()\n\teventID := int64(1)\n\tversion := int64(2)\n\tdedupResource := definition.NewEventReappliedID(runID, eventID, version)\n\tisReapplied := s.msBuilder.IsResourceDuplicated(dedupResource)\n\ts.False(isReapplied)\n\ts.msBuilder.UpdateDuplicatedResource(dedupResource)\n\tisReapplied = s.msBuilder.IsResourceDuplicated(dedupResource)\n\ts.True(isReapplied)\n}\n\nfunc (s *mutableStateSuite) TestTransientDecisionTaskSchedule_CurrentVersionChanged() {\n\tversion := int64(2000)\n\trunID := uuid.New()\n\ts.msBuilder = NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\tversion,\n\t\trunID,\n\t\tconstants.TestGlobalDomainEntry,\n\t).(*mutableStateBuilder)\n\tdecisionScheduleEvent, decisionStartedEvent := s.prepareTransientDecisionCompletionFirstBatchReplicated(version, runID)\n\tdecisionFailedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        3,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\tScheduledEventID: decisionScheduleEvent.ID,\n\t\t\tStartedEventID:   decisionStartedEvent.ID,\n\t\t},\n\t}\n\terr := s.msBuilder.ReplicateDecisionTaskFailedEvent(decisionFailedEvent)\n\ts.NoError(err)\n\n\terr = s.msBuilder.UpdateCurrentVersion(version+1, true)\n\ts.NoError(err)\n\tversionHistories := s.msBuilder.GetVersionHistories()\n\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\ts.NoError(err)\n\tversionHistory.AddOrUpdateItem(&persistence.VersionHistoryItem{\n\t\tEventID: 3,\n\t\tVersion: version,\n\t})\n\n\tnow := time.Now()\n\tdi, err := s.msBuilder.AddDecisionTaskScheduledEventAsHeartbeat(true, now.UnixNano())\n\ts.NoError(err)\n\ts.NotNil(di)\n\n\ts.Equal(int64(0), s.msBuilder.GetExecutionInfo().DecisionAttempt)\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().transientHistory))\n\ts.Equal(1, len(s.msBuilder.GetHistoryBuilder().history))\n}\n\nfunc (s *mutableStateSuite) TestTransientDecisionTaskStart_CurrentVersionChanged() {\n\tversion := int64(2000)\n\trunID := uuid.New()\n\ts.msBuilder = NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\tversion,\n\t\trunID,\n\t\tconstants.TestGlobalDomainEntry,\n\t).(*mutableStateBuilder)\n\tdecisionScheduleEvent, decisionStartedEvent := s.prepareTransientDecisionCompletionFirstBatchReplicated(version, runID)\n\tdecisionFailedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        3,\n\t\tTimestamp: common.Int64Ptr(time.Now().UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\tScheduledEventID: decisionScheduleEvent.ID,\n\t\t\tStartedEventID:   decisionStartedEvent.ID,\n\t\t},\n\t}\n\terr := s.msBuilder.ReplicateDecisionTaskFailedEvent(decisionFailedEvent)\n\ts.NoError(err)\n\n\tdecisionScheduleID := int64(4)\n\tnow := time.Now()\n\ttasklist := \"some random tasklist\"\n\tdecisionTimeoutSecond := int32(11)\n\tdecisionAttempt := int64(2)\n\tnewDecisionScheduleEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        decisionScheduleID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tAttempt:                    decisionAttempt,\n\t\t},\n\t}\n\tdi, err := s.msBuilder.ReplicateDecisionTaskScheduledEvent(\n\t\tnewDecisionScheduleEvent.Version,\n\t\tnewDecisionScheduleEvent.ID,\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.TaskList.GetName(),\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetStartToCloseTimeoutSeconds(),\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetAttempt(),\n\t\t0,\n\t\t0,\n\t\tfalse,\n\t)\n\ts.NoError(err)\n\ts.NotNil(di)\n\n\terr = s.msBuilder.UpdateCurrentVersion(version+1, true)\n\ts.NoError(err)\n\tversionHistories := s.msBuilder.GetVersionHistories()\n\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\ts.NoError(err)\n\tversionHistory.AddOrUpdateItem(&persistence.VersionHistoryItem{\n\t\tEventID: 3,\n\t\tVersion: version,\n\t})\n\n\t_, _, err = s.msBuilder.AddDecisionTaskStartedEvent(\n\t\tdecisionScheduleID,\n\t\tuuid.New(),\n\t\t&types.PollForDecisionTaskRequest{\n\t\t\tIdentity: IdentityHistoryService,\n\t\t},\n\t)\n\ts.NoError(err)\n\n\ts.Equal(0, len(s.msBuilder.GetHistoryBuilder().transientHistory))\n\ts.Equal(2, len(s.msBuilder.GetHistoryBuilder().history))\n}\n\nfunc (s *mutableStateSuite) prepareTransientDecisionCompletionFirstBatchReplicated(version int64, runID string) (*types.HistoryEvent, *types.HistoryEvent) {\n\tdomainID := constants.TestDomainID\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      runID,\n\t}\n\n\tnow := time.Now()\n\tworkflowType := \"some random workflow type\"\n\ttasklist := \"some random tasklist\"\n\tworkflowTimeoutSecond := int32(222)\n\tdecisionTimeoutSecond := int32(11)\n\tdecisionAttempt := int64(0)\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"dca\",\n\t}\n\n\teventID := int64(1)\n\tworkflowStartEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\tInput:                               nil,\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tPartitionConfig:                     partitionConfig,\n\t\t},\n\t}\n\teventID++\n\n\tdecisionScheduleEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tAttempt:                    decisionAttempt,\n\t\t},\n\t}\n\teventID++\n\n\tdecisionStartedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\tScheduledEventID: decisionScheduleEvent.ID,\n\t\t\tRequestID:        uuid.New(),\n\t\t},\n\t}\n\teventID++\n\n\tdecisionFailedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\tScheduledEventID: decisionScheduleEvent.ID,\n\t\t\tStartedEventID:   decisionStartedEvent.ID,\n\t\t},\n\t}\n\teventID++\n\n\ts.mockEventsCache.EXPECT().PutEvent(\n\t\tdomainID, execution.GetWorkflowID(), execution.GetRunID(),\n\t\tworkflowStartEvent.ID, workflowStartEvent,\n\t).Times(1)\n\terr := s.msBuilder.ReplicateWorkflowExecutionStartedEvent(\n\t\tnil,\n\t\texecution,\n\t\tuuid.New(),\n\t\tworkflowStartEvent,\n\t\tfalse,\n\t)\n\ts.Nil(err)\n\n\t// setup transient decision\n\tdi, err := s.msBuilder.ReplicateDecisionTaskScheduledEvent(\n\t\tdecisionScheduleEvent.Version,\n\t\tdecisionScheduleEvent.ID,\n\t\tdecisionScheduleEvent.DecisionTaskScheduledEventAttributes.TaskList.GetName(),\n\t\tdecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetStartToCloseTimeoutSeconds(),\n\t\tdecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetAttempt(),\n\t\t0,\n\t\t0,\n\t\tfalse,\n\t)\n\ts.Nil(err)\n\ts.NotNil(di)\n\n\tdi, err = s.msBuilder.ReplicateDecisionTaskStartedEvent(nil,\n\t\tdecisionStartedEvent.Version,\n\t\tdecisionScheduleEvent.ID,\n\t\tdecisionStartedEvent.ID,\n\t\tdecisionStartedEvent.DecisionTaskStartedEventAttributes.GetRequestID(),\n\t\tdecisionStartedEvent.GetTimestamp(),\n\t)\n\ts.Nil(err)\n\ts.NotNil(di)\n\n\terr = s.msBuilder.ReplicateDecisionTaskFailedEvent(decisionFailedEvent)\n\ts.Nil(err)\n\n\tdecisionAttempt = int64(123)\n\tnewDecisionScheduleEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tAttempt:                    decisionAttempt,\n\t\t},\n\t}\n\teventID++\n\n\tnewDecisionStartedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        eventID,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\tScheduledEventID: decisionScheduleEvent.ID,\n\t\t\tRequestID:        uuid.New(),\n\t\t},\n\t}\n\teventID++ //nolint:ineffassign\n\n\tdi, err = s.msBuilder.ReplicateDecisionTaskScheduledEvent(\n\t\tnewDecisionScheduleEvent.Version,\n\t\tnewDecisionScheduleEvent.ID,\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.TaskList.GetName(),\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetStartToCloseTimeoutSeconds(),\n\t\tnewDecisionScheduleEvent.DecisionTaskScheduledEventAttributes.GetAttempt(),\n\t\t0,\n\t\t0,\n\t\tfalse,\n\t)\n\ts.Nil(err)\n\ts.NotNil(di)\n\n\tdi, err = s.msBuilder.ReplicateDecisionTaskStartedEvent(nil,\n\t\tnewDecisionStartedEvent.Version,\n\t\tnewDecisionScheduleEvent.ID,\n\t\tnewDecisionStartedEvent.ID,\n\t\tnewDecisionStartedEvent.DecisionTaskStartedEventAttributes.GetRequestID(),\n\t\tnewDecisionStartedEvent.GetTimestamp(),\n\t)\n\ts.Nil(err)\n\ts.NotNil(di)\n\n\treturn newDecisionScheduleEvent, newDecisionStartedEvent\n}\n\nfunc (s *mutableStateSuite) TestLoad_BackwardsCompatibility() {\n\tmutableState := buildWorkflowMutableState()\n\n\ts.msBuilder.Load(context.Background(), mutableState)\n\n\ts.Equal(constants.TestDomainID, s.msBuilder.pendingChildExecutionInfoIDs[81].DomainID)\n}\n\nfunc (s *mutableStateSuite) TestUpdateCurrentVersion_WorkflowOpen() {\n\tmutableState := buildWorkflowMutableState()\n\n\ts.msBuilder.Load(context.Background(), mutableState)\n\ts.Equal(commonconstants.EmptyVersion, s.msBuilder.GetCurrentVersion())\n\n\tversion := int64(2000)\n\ts.msBuilder.UpdateCurrentVersion(version, false)\n\ts.Equal(version, s.msBuilder.GetCurrentVersion())\n}\n\nfunc (s *mutableStateSuite) TestUpdateCurrentVersion_WorkflowClosed() {\n\tmutableState := buildWorkflowMutableState()\n\tmutableState.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\tmutableState.ExecutionInfo.CloseStatus = persistence.WorkflowCloseStatusCompleted\n\n\ts.msBuilder.Load(context.Background(), mutableState)\n\ts.Equal(commonconstants.EmptyVersion, s.msBuilder.GetCurrentVersion())\n\n\tversionHistory, err := mutableState.VersionHistories.GetCurrentVersionHistory()\n\ts.NoError(err)\n\tlastItem, err := versionHistory.GetLastItem()\n\ts.NoError(err)\n\tlastWriteVersion := lastItem.Version\n\n\tversion := int64(2000)\n\ts.msBuilder.UpdateCurrentVersion(version, false)\n\ts.Equal(lastWriteVersion, s.msBuilder.GetCurrentVersion())\n}\n\nfunc (s *mutableStateSuite) newDomainCacheEntry() *cache.DomainCacheEntry {\n\treturn cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: \"mutableStateTest\"},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{},\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n}\n\nfunc buildWorkflowMutableState() *persistence.WorkflowMutableState {\n\tdomainID := constants.TestDomainID\n\twe := types.WorkflowExecution{\n\t\tWorkflowID: \"wId\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttl := \"testTaskList\"\n\tfailoverVersion := int64(300)\n\tpartitionConfig := map[string]string{\n\t\t\"zone\": \"phx\",\n\t}\n\n\tinfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:                    domainID,\n\t\tWorkflowID:                  we.GetWorkflowID(),\n\t\tRunID:                       we.GetRunID(),\n\t\tTaskList:                    tl,\n\t\tWorkflowTypeName:            \"wType\",\n\t\tWorkflowTimeout:             200,\n\t\tDecisionStartToCloseTimeout: 100,\n\t\tState:                       persistence.WorkflowStateRunning,\n\t\tCloseStatus:                 persistence.WorkflowCloseStatusNone,\n\t\tNextEventID:                 int64(101),\n\t\tLastProcessedEvent:          int64(99),\n\t\tLastUpdatedTimestamp:        time.Now(),\n\t\tDecisionVersion:             failoverVersion,\n\t\tDecisionScheduleID:          commonconstants.EmptyEventID,\n\t\tDecisionStartedID:           commonconstants.EmptyEventID,\n\t\tDecisionTimeout:             100,\n\t\tPartitionConfig:             partitionConfig,\n\t}\n\n\tactivityInfos := map[int64]*persistence.ActivityInfo{\n\t\t5: {\n\t\t\tVersion:                failoverVersion,\n\t\t\tScheduleID:             int64(5),\n\t\t\tScheduledTime:          time.Now(),\n\t\t\tStartedID:              commonconstants.EmptyEventID,\n\t\t\tStartedTime:            time.Now(),\n\t\t\tActivityID:             \"activityID_5\",\n\t\t\tScheduleToStartTimeout: 100,\n\t\t\tScheduleToCloseTimeout: 200,\n\t\t\tStartToCloseTimeout:    300,\n\t\t\tHeartbeatTimeout:       50,\n\t\t},\n\t}\n\n\ttimerInfos := map[string]*persistence.TimerInfo{\n\t\t\"25\": {\n\t\t\tVersion:    failoverVersion,\n\t\t\tTimerID:    \"25\",\n\t\t\tStartedID:  85,\n\t\t\tExpiryTime: time.Now().Add(time.Hour),\n\t\t},\n\t}\n\n\tchildInfos := map[int64]*persistence.ChildExecutionInfo{\n\t\t80: {\n\t\t\tVersion:               failoverVersion,\n\t\t\tInitiatedID:           80,\n\t\t\tInitiatedEventBatchID: 20,\n\t\t\tInitiatedEvent:        &types.HistoryEvent{},\n\t\t\tStartedID:             commonconstants.EmptyEventID,\n\t\t\tCreateRequestID:       uuid.New(),\n\t\t\tDomainID:              constants.TestDomainID,\n\t\t\tWorkflowTypeName:      \"code.uber.internal/test/foobar\",\n\t\t},\n\t\t81: {\n\t\t\tVersion:               failoverVersion,\n\t\t\tInitiatedID:           80,\n\t\t\tInitiatedEventBatchID: 20,\n\t\t\tInitiatedEvent:        &types.HistoryEvent{},\n\t\t\tStartedID:             commonconstants.EmptyEventID,\n\t\t\tCreateRequestID:       uuid.New(),\n\t\t\tDomainNameDEPRECATED:  constants.TestDomainName,\n\t\t\tWorkflowTypeName:      \"code.uber.internal/test/foobar\",\n\t\t},\n\t}\n\n\tsignalInfos := map[int64]*persistence.SignalInfo{\n\t\t75: {\n\t\t\tVersion:               failoverVersion,\n\t\t\tInitiatedID:           75,\n\t\t\tInitiatedEventBatchID: 17,\n\t\t\tSignalRequestID:       uuid.New(),\n\t\t\tSignalName:            \"test-signal-75\",\n\t\t\tInput:                 []byte(\"signal-input-75\"),\n\t\t},\n\t}\n\n\tsignalRequestIDs := map[string]struct{}{\n\t\tuuid.New(): {},\n\t}\n\n\tbufferedEvents := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\tVersion:   failoverVersion,\n\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\t\tSignalName: \"test-signal-buffered\",\n\t\t\t\tInput:      []byte(\"test-signal-buffered-input\"),\n\t\t\t},\n\t\t},\n\t}\n\n\tversionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"token#1\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{EventID: 1, Version: 300},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\treturn &persistence.WorkflowMutableState{\n\t\tExecutionInfo:       info,\n\t\tActivityInfos:       activityInfos,\n\t\tTimerInfos:          timerInfos,\n\t\tChildExecutionInfos: childInfos,\n\t\tSignalInfos:         signalInfos,\n\t\tSignalRequestedIDs:  signalRequestIDs,\n\t\tBufferedEvents:      bufferedEvents,\n\t\tVersionHistories:    versionHistories,\n\t}\n}\n\nfunc TestNewMutableStateBuilderWithEventV2(t *testing.T) {\n\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt,\n\t\tctrl,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\tdomainCache := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: \"mutableStateTest\"},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{},\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\tNewMutableStateBuilderWithEventV2(mockShard, log.NewNoop(), \"A82146B5-7A5C-4660-9195-E80E5161EC56\", domainCache)\n}\n\nvar (\n\tdomainID = \"A6338800-D143-4FEF-8A49-9BBB31386C5F\"\n\twfID     = \"879A361B-B435-491D-8A3B-ACF3BAD30F4B\"\n\trunID    = \"81DFCB6B-ACD4-46D1-89C2-804388203880\"\n\tts1      = int64(1234)\n\tshardID  = 123\n)\n\n// Guiding real data example: ie:\n// `select execution from executions where run_id = <run-id> ALLOW FILTERING;`\n//\n// executions.execution {\n// domainID: \"A6338800-D143-4FEF-8A49-9BBB31386C5F\",\n// wfID: \"879A361B-B435-491D-8A3B-ACF3BAD30F4B\",\n// runID: \"81DFCB6B-ACD4-46D1-89C2-804388203880\",\n// initiated_id: -7,\n// completion_event: null,\n// state: 2,\n// close_status: 1,\n// next_event_id: 12,\n// last_processed_event: 9,\n// decision_schedule_id: -23,\n// decision_started_id: -23,\n// last_first_event_id: 10,\n// decision_version: -24,\n// completion_event_batch_id: 10,\n// last_event_task_id: 4194328,\n// }\nvar exampleMutableStateForClosedWF = &mutableStateBuilder{\n\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\tWorkflowID:             wfID,\n\t\tDomainID:               domainID,\n\t\tRunID:                  runID,\n\t\tNextEventID:            12,\n\t\tState:                  persistence.WorkflowStateCompleted,\n\t\tCompletionEventBatchID: 10,\n\t\tBranchToken:            []byte(\"branch-token\"),\n\t},\n}\n\nvar exampleCompletionEvent = &types.HistoryEvent{\n\tID:        11,\n\tTaskID:    4194328,\n\tVersion:   1,\n\tTimestamp: &ts1,\n\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{\n\t\tResult:                       []byte(\"some random workflow completion result\"),\n\t\tDecisionTaskCompletedEventID: 10,\n\t},\n}\n\nvar exampleStartEvent = &types.HistoryEvent{\n\tID:        1,\n\tTaskID:    4194328,\n\tVersion:   1,\n\tTimestamp: &ts1,\n\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\tWorkflowType:                        &types.WorkflowType{Name: \"workflow-type\"},\n\t\tTaskList:                            &types.TaskList{Name: \"tasklist\"},\n\t\tInput:                               []byte(\"some random workflow input\"),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\tOriginalExecutionRunID:              runID,\n\t\tIdentity:                            \"123@some-hostname@@uuid\",\n\t},\n}\n\nfunc TestGetCompletionEvent(t *testing.T) {\n\ttests := map[string]struct {\n\t\tcurrentState *mutableStateBuilder\n\n\t\thistoryManagerAffordance func(historyManager *persistence.MockHistoryManager)\n\n\t\texpectedResult *types.HistoryEvent\n\t\texpectedErr    error\n\t}{\n\t\t\"Getting a completed event from a normal, completed workflow - taken from a real example\": {\n\t\t\tcurrentState: exampleMutableStateForClosedWF,\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(),\n\t\t\t\t\t&persistence.ReadHistoryBranchRequest{\n\t\t\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\t\t\tMinEventID:    10,\n\t\t\t\t\t\tMaxEventID:    12, // nextEventID +1\n\t\t\t\t\t\tPageSize:      1,\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t\tShardID:       common.IntPtr(shardID),\n\t\t\t\t\t\tDomainName:    \"domain\",\n\t\t\t\t\t}).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\texampleCompletionEvent,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t},\n\n\t\t\texpectedResult: exampleCompletionEvent,\n\t\t},\n\t\t\"An unexpected error while fetchhing history, such as not found err\": {\n\t\t\tcurrentState: &mutableStateBuilder{\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID:             wfID,\n\t\t\t\t\tDomainID:               domainID,\n\t\t\t\t\tRunID:                  runID,\n\t\t\t\t\tNextEventID:            12,\n\t\t\t\t\tState:                  persistence.WorkflowStateCompleted,\n\t\t\t\t\tCompletionEventBatchID: 10,\n\t\t\t\t\tBranchToken:            []byte(\"branch-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"a transient random error\"))\n\t\t\t},\n\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    &types.InternalServiceError{Message: \"unable to get workflow completion event\"},\n\t\t},\n\t\t\"A 'transient' internal service error, this should be returned to the caller\": {\n\t\t\tcurrentState: &mutableStateBuilder{\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID:             wfID,\n\t\t\t\t\tDomainID:               domainID,\n\t\t\t\t\tRunID:                  runID,\n\t\t\t\t\tNextEventID:            12,\n\t\t\t\t\tState:                  persistence.WorkflowStateCompleted,\n\t\t\t\t\tCompletionEventBatchID: 10,\n\t\t\t\t\tBranchToken:            []byte(\"branch-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.InternalServiceError{Message: \"an err\"})\n\t\t\t},\n\n\t\t\texpectedResult: nil,\n\t\t\texpectedErr:    &types.InternalServiceError{Message: \"an err\"},\n\t\t},\n\t\t\"initial validation: An invalid starting mutable state should return an error\": {\n\t\t\tcurrentState: &mutableStateBuilder{\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {},\n\t\t\texpectedResult:           nil,\n\t\t\texpectedErr:              &types.InternalServiceError{Message: \"unable to get workflow completion event\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shardCtx.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetShardID().Return(123).AnyTimes() // this isn't called on a few of the validation failures\n\t\t\thistoryManager := persistence.NewMockHistoryManager(ctrl)\n\t\t\ttd.historyManagerAffordance(historyManager)\n\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domain\", nil).AnyTimes() // this isn't called on validation\n\n\t\t\ttd.currentState.eventsCache = events.NewCache(shardID, historyManager, config.NewForTest(), log.NewNoop(), metrics.NewNoopMetricsClient(), domainCache)\n\t\t\ttd.currentState.shard = shardContext\n\n\t\t\tres, err := td.currentState.GetCompletionEvent(context.Background())\n\n\t\t\tassert.Equal(t, td.expectedResult, res)\n\t\t\tif td.expectedErr != nil {\n\t\t\t\tassert.ErrorAs(t, td.expectedErr, &err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetStartEvent(t *testing.T) {\n\ttests := map[string]struct {\n\t\tcurrentState *mutableStateBuilder\n\n\t\thistoryManagerAffordance func(historyManager *persistence.MockHistoryManager)\n\n\t\texpectedResult *types.HistoryEvent\n\t\texpectedErr    error\n\t}{\n\t\t\"Getting a start event from a normal, completed workflow - taken from a real example\": {\n\t\t\tcurrentState: exampleMutableStateForClosedWF,\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(),\n\t\t\t\t\t&persistence.ReadHistoryBranchRequest{\n\t\t\t\t\t\tBranchToken:   []byte(\"branch-token\"),\n\t\t\t\t\t\tMinEventID:    1,\n\t\t\t\t\t\tMaxEventID:    2,\n\t\t\t\t\t\tPageSize:      1,\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t\tShardID:       common.IntPtr(shardID),\n\t\t\t\t\t\tDomainName:    \"domain\",\n\t\t\t\t\t}).Return(&persistence.ReadHistoryBranchResponse{\n\t\t\t\t\tHistoryEvents: []*types.HistoryEvent{\n\t\t\t\t\t\texampleStartEvent,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t},\n\n\t\t\texpectedResult: exampleStartEvent,\n\t\t},\n\t\t\"Getting a start event but hitting an error when reaching into history\": {\n\t\t\tcurrentState: exampleMutableStateForClosedWF,\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"an error\"))\n\t\t\t},\n\t\t\texpectedErr: types.InternalServiceError{Message: \"unable to get workflow start event\"},\n\t\t},\n\t\t\"Getting a start event but hitting a 'transient' error when reaching into history. This should be passed back up the call stack\": {\n\t\t\tcurrentState: exampleMutableStateForClosedWF,\n\t\t\thistoryManagerAffordance: func(historyManager *persistence.MockHistoryManager) {\n\t\t\t\thistoryManager.EXPECT().ReadHistoryBranch(gomock.Any(), gomock.Any()).Return(nil, &types.InternalServiceError{Message: \"an error\"})\n\t\t\t},\n\t\t\texpectedErr: types.InternalServiceError{Message: \"an error\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shardCtx.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetShardID().Return(123).AnyTimes() // this isn't called on a few of the validation failures\n\t\t\thistoryManager := persistence.NewMockHistoryManager(ctrl)\n\t\t\ttd.historyManagerAffordance(historyManager)\n\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domain\", nil).AnyTimes() // this isn't called on validation\n\n\t\t\ttd.currentState.eventsCache = events.NewCache(shardID, historyManager, config.NewForTest(), log.NewNoop(), metrics.NewNoopMetricsClient(), domainCache)\n\t\t\ttd.currentState.shard = shardContext\n\n\t\t\tres, err := td.currentState.GetStartEvent(context.Background())\n\n\t\t\tassert.Equal(t, td.expectedResult, res)\n\t\t\tif td.expectedErr != nil {\n\t\t\t\tassert.ErrorAs(t, err, &td.expectedErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLoggingNilAndInvalidHandling(t *testing.T) {\n\tgen := testdatagen.New(t)\n\n\texecutionInfo := persistence.WorkflowExecutionInfo{}\n\n\tgen.Fuzz(&executionInfo)\n\tmsb := mutableStateBuilder{\n\t\texecutionInfo: &executionInfo,\n\t\tlogger:        log.NewNoop(),\n\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t}\n\n\tmsbInvalid := mutableStateBuilder{logger: log.NewNoop()}\n\n\tassert.NotPanics(t, func() {\n\t\tmsbInvalid.logWarn(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsbInvalid.logError(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsbInvalid.logInfo(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsb.logWarn(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsb.logError(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsb.logInfo(\"test\", tag.WorkflowDomainID(\"test\"))\n\t\tmsb.logDataInconsistency()\n\t})\n}\n\nfunc TestAssignEventIDToBufferedEvents(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tstartingEventID                   int64\n\t\tpendingActivityInfo               map[int64]*persistence.ActivityInfo\n\t\tpendingChildExecutionInfoIDs      map[int64]*persistence.ChildExecutionInfo\n\t\tstartingHistoryEntries            []*types.HistoryEvent\n\t\texpectedUpdateActivityInfos       map[int64]*persistence.ActivityInfo\n\t\texpectedEndingHistoryEntries      []*types.HistoryEvent\n\t\texpectedNextEventID               int64\n\t\texpectedUpdateChildExecutionInfos map[int64]*persistence.ChildExecutionInfo\n\t}{\n\t\t\"Timer Fired - this should increment the nextevent ID counter\": {\n\t\t\tstartingEventID: 12,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID:        \"1\",\n\t\t\t\t\t\tStartedEventID: 11,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tID:        12,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID:        \"1\",\n\t\t\t\t\t\tStartedEventID: 11,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               13,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity completed and started - this should update any buffered activities\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tID:        4,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\t\tID:        5,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityID: \"0\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 5,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tID:        4,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\t\tID:        5,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityID: \"0\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 5,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               7,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity task started and a pending activity is updated - this should be put to the updatedActivityInfos map with all the other counters incremented\": {\n\t\t\tstartingEventID: 6,\n\t\t\tpendingActivityInfo: map[int64]*persistence.ActivityInfo{\n\t\t\t\t5: {\n\t\t\t\t\tScheduleID: 5,\n\t\t\t\t\tStartedID:  6,\n\t\t\t\t},\n\t\t\t},\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 5,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 5,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID: 7,\n\t\t\texpectedUpdateActivityInfos: map[int64]*persistence.ActivityInfo{\n\t\t\t\t5: {\n\t\t\t\t\tScheduleID: 5,\n\t\t\t\t\tStartedID:  6,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity task started and then completed\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   4567,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   6,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity task started and then Cancelled\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\t\tStartedEventID:   6,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity task started and then failed\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskFailed.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskFailed.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   6,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Activity task started and then timed out\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\t\tStartedEventID:   6,\n\t\t\t\t\t\tScheduledEventID: 3456,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Child workflow scheduled and then completed\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCompleted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCompleted.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Child workflow scheduled and then Cancelled - where there is a pending execution that requires an update\": {\n\t\t\tstartingEventID: 6,\n\t\t\tpendingChildExecutionInfoIDs: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t123: {\n\t\t\t\t\tInitiatedID: 321,\n\t\t\t\t},\n\t\t\t},\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCanceled.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\t\tStartedEventID:   321,\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCanceled.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\t\tStartedEventID:   6,\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:         8,\n\t\t\texpectedUpdateActivityInfos: map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t321: {\n\t\t\t\t\tInitiatedID: 321,\n\t\t\t\t\tStartedID:   6,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"Child workflow scheduled and then Failed\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionFailed.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionFailed.Ptr().Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Child workflow scheduled and then Timed out\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t\t\"Child workflow scheduled and then Terminated\": {\n\t\t\tstartingEventID: 6,\n\t\t\tstartingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTerminated.Ptr(),\n\t\t\t\t\tID:        commonconstants.BufferedEventID,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionTerminatedEventAttributes: &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedEndingHistoryEntries: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tID:        6,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 123,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTerminated.Ptr(),\n\t\t\t\t\tID:        7,\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t\tChildWorkflowExecutionTerminatedEventAttributes: &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\t\tStartedEventID:   123,\n\t\t\t\t\t\tInitiatedEventID: 2345,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedNextEventID:               8,\n\t\t\texpectedUpdateActivityInfos:       map[int64]*persistence.ActivityInfo{},\n\t\t\texpectedUpdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmsb := &mutableStateBuilder{\n\t\t\t\tpendingChildExecutionInfoIDs: td.pendingChildExecutionInfoIDs,\n\t\t\t\tpendingActivityInfoIDs:       td.pendingActivityInfo,\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tNextEventID: td.startingEventID,\n\t\t\t\t},\n\t\t\t\thBuilder: &HistoryBuilder{\n\t\t\t\t\thistory: td.startingHistoryEntries,\n\t\t\t\t},\n\t\t\t\tupdateActivityInfos:       make(map[int64]*persistence.ActivityInfo),\n\t\t\t\tupdateChildExecutionInfos: make(map[int64]*persistence.ChildExecutionInfo),\n\t\t\t}\n\n\t\t\tmsb.assignEventIDToBufferedEvents()\n\n\t\t\tassert.Equal(t, td.expectedEndingHistoryEntries, msb.hBuilder.history)\n\t\t\tassert.Equal(t, td.expectedNextEventID, msb.executionInfo.NextEventID)\n\t\t\tassert.Equal(t, td.expectedUpdateActivityInfos, msb.updateActivityInfos)\n\t\t\tassert.Equal(t, td.expectedUpdateChildExecutionInfos, msb.updateChildExecutionInfos)\n\t\t})\n\t}\n}\n\n// This is only for passing the coverage\nfunc TestLog(t *testing.T) {\n\tvar e *mutableStateBuilder\n\tassert.NotPanics(t, func() { e.logInfo(\"a\") })\n\tassert.NotPanics(t, func() { e.logWarn(\"a\") })\n\tassert.NotPanics(t, func() { e.logError(\"a\") })\n}\n\nfunc TestMutableStateBuilder_CopyToPersistence_roundtrip(t *testing.T) {\n\n\tfor i := 0; i <= 100; i++ {\n\t\tctrl := gomock.NewController(t)\n\n\t\tseed := int64(rand.Int())\n\t\tfuzzer := testdatagen.NewWithNilChance(t, seed, 0)\n\n\t\texecution := &persistence.WorkflowMutableState{}\n\t\tfuzzer.Fuzz(&execution)\n\n\t\t// checksum is a calculated value, zero it out because\n\t\t// it'll be overwridden during the constructor setup\n\t\texecution.Checksum = checksum.Checksum{}\n\n\t\tshardContext := shard.NewMockContext(ctrl)\n\t\tmockCache := events.NewMockCache(ctrl)\n\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\tmockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"some-domain-id\", nil).AnyTimes()\n\n\t\tshardContext.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).Times(2)\n\t\tshardContext.EXPECT().GetEventsCache().Return(mockCache)\n\t\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\t\tNumberOfShards:                        2,\n\t\t\tIsAdvancedVisConfigExist:              false,\n\t\t\tMaxResponseSize:                       0,\n\t\t\tMutableStateChecksumInvalidateBefore:  dynamicproperties.GetFloatPropertyFn(10),\n\t\t\tMutableStateChecksumVerifyProbability: dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\t\tHostName:                              \"test-host\",\n\t\t}).Times(1)\n\t\tshardContext.EXPECT().GetTimeSource().Return(clock.NewMockedTimeSource())\n\t\tshardContext.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient())\n\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\n\t\tactiveClusterManager := activecluster.NewMockManager(ctrl)\n\t\tshardContext.EXPECT().GetActiveClusterManager().Return(activeClusterManager).AnyTimes()\n\n\t\tmsb := newMutableStateBuilder(shardContext, log.NewNoop(), constants.TestGlobalDomainEntry, constants.TestGlobalDomainEntry.GetFailoverVersion())\n\n\t\tmsb.Load(context.Background(), execution)\n\n\t\tout := msb.CopyToPersistence()\n\n\t\tassert.Equal(t, execution.ActivityInfos, out.ActivityInfos, \"activityinfos mismatch\")\n\t\tassert.Equal(t, execution.TimerInfos, out.TimerInfos, \"timerinfos mismatch\")\n\t\tassert.Equal(t, execution.ChildExecutionInfos, out.ChildExecutionInfos, \"child executino info mismatches\")\n\t\tassert.Equal(t, execution.RequestCancelInfos, out.RequestCancelInfos, \"request cancellantion info mismatches\")\n\t\tassert.Equal(t, execution.SignalInfos, out.SignalInfos, \"signal info mismatches\")\n\t\tassert.Equal(t, execution.SignalRequestedIDs, out.SignalRequestedIDs, \"signal request ids mismaches\")\n\t\tassert.Equal(t, execution.ExecutionInfo, out.ExecutionInfo, \"execution info mismatches\")\n\t\tassert.Equal(t, execution.BufferedEvents, out.BufferedEvents, \"buffered events mismatch\")\n\t\tassert.Equal(t, execution.VersionHistories, out.VersionHistories, \"version histories\")\n\t\tassert.Equal(t, execution.Checksum, out.Checksum, \"checksum mismatch\")\n\t\tassert.Equal(t, execution.ReplicationState, out.ReplicationState, \"replication state mismatch\")\n\t\tassert.Equal(t, execution.ExecutionStats, out.ExecutionStats, \"execution stats mismatch\")\n\n\t\tassert.Equal(t, execution, out)\n\n\t}\n}\n\nfunc TestMutableStateBuilder_closeTransactionHandleWorkflowReset(t *testing.T) {\n\tt1 := time.Unix(123, 0)\n\tnow := time.Unix(500, 0)\n\n\tbadBinaryID := \"bad-binary-id\"\n\n\tmockDomainEntryWithBadBinary := cache.NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"domain\"}, &persistence.DomainConfig{\n\t\tBadBinaries: types.BadBinaries{\n\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\tbadBinaryID: &types.BadBinaryInfo{\n\t\t\t\t\tReason:          \"some-reason\",\n\t\t\t\t\tOperator:        \"\",\n\t\t\t\t\tCreatedTimeNano: common.Ptr(t1.UnixNano()),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, \"cluster0\")\n\n\tmockDomainEntryWithoutBadBinary := cache.NewLocalDomainCacheEntryForTest(nil, &persistence.DomainConfig{\n\t\tBadBinaries: types.BadBinaries{\n\t\t\tBinaries: map[string]*types.BadBinaryInfo{},\n\t\t},\n\t}, \"cluster0\")\n\n\ttests := map[string]struct {\n\t\tpolicyIn                         TransactionPolicy\n\t\tshardContextExpectations         func(mockCache *events.MockCache, shard *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache)\n\t\tmutableStateBuilderStartingState func(m *mutableStateBuilder)\n\n\t\texpectedEndState func(t *testing.T, m *mutableStateBuilder)\n\t\texpectedErr      error\n\t}{\n\t\t\"a workflow with reset point which is running - the expectation is that this should be able to successfully find the domain to reset and add a transfer task\": {\n\n\t\t\tpolicyIn: TransactionPolicyActive,\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\t// the workflow's running\n\t\t\t\tm.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tDomainID:    \"some-domain-id\",\n\t\t\t\t\tWorkflowID:  \"wf-id\",\n\t\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBinaryChecksum:           badBinaryID,\n\t\t\t\t\t\t\t\tRunID:                    \"\",\n\t\t\t\t\t\t\t\tFirstDecisionCompletedID: 0,\n\t\t\t\t\t\t\t\tCreatedTimeNano:          common.Ptr(t1.UnixNano()),\n\t\t\t\t\t\t\t\tExpiringTimeNano:         nil,\n\t\t\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(mockDomainEntryWithBadBinary, nil)\n\t\t\t},\n\t\t\texpectedEndState: func(t *testing.T, m *mutableStateBuilder) {\n\t\t\t\tassert.Equal(t, []persistence.Task{\n\t\t\t\t\t&persistence.ResetWorkflowTask{\n\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\tDomainID:   \"some-domain-id\",\n\t\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\tVersion: commonconstants.EmptyVersion,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, m.insertTransferTasks)\n\t\t\t},\n\t\t},\n\t\t\"a workflow with reset point which is running for a domain without a bad binary - the expectation is this will not add any transfer tasks\": {\n\n\t\t\tpolicyIn: TransactionPolicyActive,\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\t// the workflow's running\n\t\t\t\tm.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tDomainID:    \"some-domain-id\",\n\t\t\t\t\tWorkflowID:  \"wf-id\",\n\t\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBinaryChecksum:           badBinaryID,\n\t\t\t\t\t\t\t\tRunID:                    \"\",\n\t\t\t\t\t\t\t\tFirstDecisionCompletedID: 0,\n\t\t\t\t\t\t\t\tCreatedTimeNano:          common.Ptr(t1.UnixNano()),\n\t\t\t\t\t\t\t\tExpiringTimeNano:         nil,\n\t\t\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(mockDomainEntryWithoutBadBinary, nil)\n\t\t\t},\n\t\t\texpectedEndState: func(t *testing.T, m *mutableStateBuilder) {\n\t\t\t\tassert.Equal(t, []persistence.Task(nil), m.insertTransferTasks)\n\t\t\t},\n\t\t},\n\t\t\"a workflow withithout auto-reset point which is running for a domain\": {\n\n\t\t\tpolicyIn: TransactionPolicyActive,\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\t// the workflow's running\n\t\t\t\tm.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tDomainID:    \"some-domain-id\",\n\t\t\t\t\tWorkflowID:  \"wf-id\",\n\t\t\t\t}\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(mockDomainEntryWithoutBadBinary, nil)\n\t\t\t},\n\t\t\texpectedEndState: func(t *testing.T, m *mutableStateBuilder) {\n\t\t\t\tassert.Equal(t, []persistence.Task(nil), m.insertTransferTasks)\n\t\t\t},\n\t\t},\n\t\t\"a workflow with reset point which is running but which has child workflows\": {\n\t\t\tpolicyIn: TransactionPolicyActive,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t},\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\t// the workflow's running\n\t\t\t\tm.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\n\t\t\t\t// there's some child workflow that's due to be updated\n\t\t\t\tm.pendingChildExecutionInfoIDs = map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t\t1: &persistence.ChildExecutionInfo{},\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedEndState: func(t *testing.T, m *mutableStateBuilder) {\n\t\t\t\tassert.Equal(t, []persistence.Task(nil), m.insertTransferTasks)\n\t\t\t},\n\t\t},\n\t\t\"Transaction policy passive - no expected resets\": {\n\t\t\tpolicyIn: TransactionPolicyPassive,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t},\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\t// the workflow's running\n\t\t\t\tm.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tCloseStatus: persistence.WorkflowCloseStatusNone,\n\t\t\t\t}\n\n\t\t\t\t// there's some child workflow that's due to be updated\n\t\t\t\tm.pendingChildExecutionInfoIDs = map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t\t1: &persistence.ChildExecutionInfo{},\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedEndState: func(t *testing.T, m *mutableStateBuilder) {\n\t\t\t\tassert.Equal(t, []persistence.Task(nil), m.insertTransferTasks)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\ttd.shardContextExpectations(mockCache, shardContext, mockDomainCache)\n\n\t\t\tnowClock := clock.NewMockedTimeSourceAt(now)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\ttd.mutableStateBuilderStartingState(msb)\n\n\t\t\tmsb.timeSource = nowClock\n\t\t\terr := msb.closeTransactionHandleWorkflowReset(td.policyIn)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t\ttd.expectedEndState(t, msb)\n\t\t})\n\t}\n}\n\nfunc TestMutableStateBuilder_GetVersionHistoriesStart(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmutableStateBuilderStartingState func(m *mutableStateBuilder)\n\t\texpectedVersion                  int64\n\t\texpectedErr                      error\n\t}{\n\t\t\"A mutable state with version history\": {\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\tm.versionHistories = &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 100,\n\t\t\t\t\t\t\t\t\tVersion: 23,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 401,\n\t\t\t\t\t\t\t\t\tVersion: 424,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 200,\n\t\t\t\t\t\t\t\t\tVersion: 123,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 201,\n\t\t\t\t\t\t\t\t\tVersion: 124,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedVersion: 23,\n\t\t},\n\t\t\"invalid / partial version history \": {\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\tm.versionHistories = &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token2\"),\n\t\t\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedErr:     &types.BadRequestError{Message: \"version history is empty.\"},\n\t\t\texpectedVersion: 0,\n\t\t},\n\t\t\"invalid / partial version history - branch not available\": {\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t\tm.versionHistories = &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 10,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedErr:     &types.BadRequestError{Message: \"getting branch index: 10, available branch count: 2\"},\n\t\t\texpectedVersion: 0,\n\t\t},\n\t\t\"nil version history\": {\n\t\t\tmutableStateBuilderStartingState: func(m *mutableStateBuilder) {\n\t\t\t},\n\t\t\texpectedVersion: commonconstants.EmptyVersion,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\ttd.mutableStateBuilderStartingState(msb)\n\n\t\t\tres, err := msb.GetStartVersion()\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t\tassert.Equal(t, td.expectedVersion, res)\n\t\t})\n\t}\n}\n\nfunc TestIsCurrentWorkflowGuaranteed(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tstateInDB      int\n\t\texpectedResult bool\n\t}{\n\t\t{\n\t\t\tname:           \"Workflow is created\",\n\t\t\tstateInDB:      persistence.WorkflowStateCreated,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Workflow is running\",\n\t\t\tstateInDB:      persistence.WorkflowStateRunning,\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Workflow is completed\",\n\t\t\tstateInDB:      persistence.WorkflowStateCompleted,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"Workflow is zombie\",\n\t\t\tstateInDB:      persistence.WorkflowStateZombie,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"Workflow is void\",\n\t\t\tstateInDB:      persistence.WorkflowStateVoid,\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"Workflow is corrupted\",\n\t\t\tstateInDB:      persistence.WorkflowStateCorrupted,\n\t\t\texpectedResult: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmsb := mutableStateBuilder{}\n\t\t\tmsb.stateInDB = tt.stateInDB\n\t\t\tresult := msb.IsCurrentWorkflowGuaranteed()\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n\n\tassert.Panics(t, func() {\n\t\tmsb := mutableStateBuilder{}\n\t\tmsb.stateInDB = 123\n\t\tmsb.IsCurrentWorkflowGuaranteed()\n\t})\n}\n\n// this is a pretty poor test, the actual logic is better tested in the\n// unit tests for getBackoffInterval()\nfunc TestGetRetryBackoffDuration(t *testing.T) {\n\n\ttests := []struct {\n\t\tname            string\n\t\tretryPolicy     *persistence.WorkflowExecutionInfo\n\t\terrorReason     string\n\t\texpectedBackoff time.Duration\n\t}{\n\t\t{\n\t\t\tname: \"NoRetryPolicy\",\n\t\t\tretryPolicy: &persistence.WorkflowExecutionInfo{\n\t\t\t\tHasRetryPolicy: false,\n\t\t\t},\n\t\t\terrorReason:     \"some error reason\",\n\t\t\texpectedBackoff: backoff.NoBackoff,\n\t\t},\n\t\t{\n\t\t\tname: \"WithRetryPolicy\",\n\t\t\tretryPolicy: &persistence.WorkflowExecutionInfo{\n\t\t\t\tHasRetryPolicy:     true,\n\t\t\t\tExpirationTime:     time.Now().Add(time.Hour),\n\t\t\t\tAttempt:            1,\n\t\t\t\tMaximumAttempts:    5,\n\t\t\t\tBackoffCoefficient: 2.0,\n\t\t\t\tInitialInterval:    12,\n\t\t\t\tNonRetriableErrors: []string{\"non-retriable-error\"},\n\t\t\t},\n\t\t\terrorReason:     \"some error reason\",\n\t\t\texpectedBackoff: 24 * time.Second,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\n\t\t\tt1 := time.Unix(1730247795, 0)\n\n\t\t\tmsb := mutableStateBuilder{}\n\n\t\t\tmsb.executionInfo = tt.retryPolicy\n\t\t\tmsb.timeSource = clock.NewMockedTimeSourceAt(t1)\n\n\t\t\tduration := msb.GetRetryBackoffDuration(tt.errorReason)\n\t\t\tassert.Equal(t, tt.expectedBackoff, duration)\n\t\t})\n\t}\n}\n\nfunc TestGetCronRetryBackoffDuration(t *testing.T) {\n\n\tt1 := time.Unix(1730247795, 0)\n\n\tsampleVersionHistory := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 100,\n\t\t\t\t\t\tVersion: 23,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 401,\n\t\t\t\t\t\tVersion: 424,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tBranchToken: []byte(\"branch-token1\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 200,\n\t\t\t\t\t\tVersion: 123,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 201,\n\t\t\t\t\t\tVersion: 124,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tstartEvent := &types.HistoryEvent{\n\t\tID:                                      1,\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t}\n\n\ttests := map[string]struct {\n\t\tstartingExecutionInfo    *persistence.WorkflowExecutionInfo\n\t\texpectedErr              bool\n\t\tshardContextExpectations func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache)\n\t\texpectedBackoff          time.Duration\n\t}{\n\t\t\"with simple, valid cron schedule\": {\n\t\t\tstartingExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:       \"domain-id\",\n\t\t\t\tCronSchedule:   \"* * * * *\",\n\t\t\t\tRunID:          \"run-id\",\n\t\t\t\tWorkflowID:     \"wid\",\n\t\t\t\tStartTimestamp: t1,\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetShardID().Return(12)\n\t\t\t\tmockCache.EXPECT().GetEvent(gomock.Any(), 12, \"domain-id\", \"wid\", \"run-id\", int64(1), int64(1), []byte(\"branch-token1\")).Return(startEvent, nil)\n\t\t\t},\n\t\t\texpectedBackoff: 45 * time.Second,\n\t\t},\n\t\t\"with no cron schedule\": {\n\t\t\tstartingExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:       \"domain-id\",\n\t\t\t\tRunID:          \"run-id\",\n\t\t\t\tWorkflowID:     \"wid\",\n\t\t\t\tStartTimestamp: t1,\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t},\n\t\t\texpectedBackoff: backoff.NoBackoff,\n\t\t},\n\t\t\"with invalid start event\": {\n\t\t\tstartingExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:       \"domain-id\",\n\t\t\t\tRunID:          \"run-id\",\n\t\t\t\tWorkflowID:     \"wid\",\n\t\t\t\tCronSchedule:   \"* * * * *\",\n\t\t\t\tStartTimestamp: t1,\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetShardID().Return(12)\n\t\t\t\tmockCache.EXPECT().GetEvent(gomock.Any(), 12, \"domain-id\", \"wid\", \"run-id\", int64(1), int64(1), []byte(\"branch-token1\")).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedBackoff: backoff.NoBackoff,\n\t\t\texpectedErr:     true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\ttd.shardContextExpectations(mockCache, shardContext, mockDomainCache)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\tmsb.executionInfo = td.startingExecutionInfo\n\t\t\tmsb.versionHistories = sampleVersionHistory\n\t\t\tmsb.timeSource = clock.NewMockedTimeSourceAt(t1)\n\n\t\t\tduration, err := msb.GetCronBackoffDuration(context.Background())\n\t\t\tassert.Equal(t, td.expectedBackoff, duration)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestStartTransactionHandleFailover(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tincomingTaskVersion       int64\n\t\tcurrentVersion            int64\n\t\tdecisionManagerAffordance func(m *MockmutableStateDecisionTaskManager)\n\t\texpectFlushBeforeReady    bool\n\t\texpectedErr               bool\n\t}{\n\t\t\"Failing over from cluster2 to cluster1 - passive -> passive: There's an inflight decision, but it's from an earlier version\": {\n\t\t\tincomingTaskVersion: 10,\n\t\t\tcurrentVersion:      2,\n\t\t\tdecisionManagerAffordance: func(m *MockmutableStateDecisionTaskManager) {\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(&DecisionInfo{\n\t\t\t\t\tVersion: 2,\n\t\t\t\t}, true)\n\t\t\t},\n\t\t\texpectFlushBeforeReady: false,\n\t\t},\n\t\t// todo: David.porter - look a bit more into why this could occur and write a better description\n\t\t// about what the intent is, because this is a unit test without a clear intent or outcome.\n\t\t// At the time of writing this test I believe this is a migration case, but I'm not 100% sure and\n\t\t// need to do some runtime debugging.\n\t\t\"empty version\": {\n\t\t\tincomingTaskVersion: commonconstants.EmptyVersion,\n\t\t\tcurrentVersion:      2,\n\t\t\tdecisionManagerAffordance: func(m *MockmutableStateDecisionTaskManager) {\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(&DecisionInfo{\n\t\t\t\t\tVersion: 2,\n\t\t\t\t}, true)\n\t\t\t},\n\t\t\texpectFlushBeforeReady: false,\n\t\t},\n\t\t\"active -> passive - when there's an inflight decision from an earlier version\": {\n\t\t\tincomingTaskVersion: 12,\n\t\t\tcurrentVersion:      11,\n\t\t\tdecisionManagerAffordance: func(m *MockmutableStateDecisionTaskManager) {\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(&DecisionInfo{\n\t\t\t\t\tVersion:    2,\n\t\t\t\t\tStartedID:  123,\n\t\t\t\t\tScheduleID: 124,\n\t\t\t\t\tRequestID:  \"requestID\",\n\t\t\t\t}, true)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(int64(124), int64(123), types.DecisionTaskFailedCauseFailoverCloseDecision, gomock.Any(), \"history-service\", gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any())\n\n\t\t\t\tm.EXPECT().HasInFlightDecision().Return(true)\n\t\t\t\tm.EXPECT().HasInFlightDecision().Return(true)\n\t\t\t\tm.EXPECT().HasPendingDecision().Return(true)\n\t\t\t},\n\t\t\texpectFlushBeforeReady: true,\n\t\t},\n\t\t\"There's a decision for for the same level as the failover version\": {\n\t\t\tincomingTaskVersion: 10,\n\t\t\tcurrentVersion:      10,\n\t\t\tdecisionManagerAffordance: func(m *MockmutableStateDecisionTaskManager) {\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(&DecisionInfo{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}, true)\n\t\t\t},\n\t\t\texpectFlushBeforeReady: false,\n\t\t\texpectedErr:            true,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tclusterMetadata := cluster.NewMetadata(\n\t\t\t\tcommonConfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 10,\n\t\t\t\t\tPrimaryClusterName:       \"cluster0\",\n\t\t\t\t\tCurrentClusterName:       \"cluster0\",\n\t\t\t\t\tClusterGroup: map[string]commonConfig.ClusterInformation{\n\t\t\t\t\t\t\"cluster0\": {\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster1\": {\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 0,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"cluster2\": {\n\t\t\t\t\t\t\tEnabled:                true,\n\t\t\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tfunc(string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\tshardContext := shardCtx.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\n\t\t\tactiveClusterManager := activecluster.NewMockManager(ctrl)\n\t\t\tshardContext.EXPECT().GetActiveClusterManager().Return(activeClusterManager).AnyTimes()\n\n\t\t\tdecisionManager := NewMockmutableStateDecisionTaskManager(ctrl)\n\t\t\ttd.decisionManagerAffordance(decisionManager)\n\n\t\t\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\tNumberOfShards:                        3,\n\t\t\t\tIsAdvancedVisConfigExist:              false,\n\t\t\t\tMaxResponseSize:                       0,\n\t\t\t\tMutableStateChecksumInvalidateBefore:  dynamicproperties.GetFloatPropertyFn(10),\n\t\t\t\tMutableStateChecksumVerifyProbability: dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\t\t\tEnableReplicationTaskGeneration:       func(_ string, _ string) bool { return true },\n\t\t\t\tHostName:                              \"test-host\",\n\t\t\t}).Times(1)\n\n\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{\n\t\t\t\tID:   \"domain-id\",\n\t\t\t\tName: \"domain\",\n\t\t\t},\n\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t{ClusterName: \"cluster0\"},\n\t\t\t\t\t\t{ClusterName: \"cluster1\"},\n\t\t\t\t\t\t{ClusterName: \"cluster2\"},\n\t\t\t\t\t},\n\t\t\t\t}, 0, nil, 0, 0, 0)\n\n\t\t\tmsb := mutableStateBuilder{\n\t\t\t\tdecisionTaskManager: decisionManager,\n\t\t\t\tshard:               shardContext,\n\t\t\t\tdomainEntry:         domainEntry,\n\t\t\t\tclusterMetadata:     clusterMetadata,\n\t\t\t\tcurrentVersion:      td.currentVersion,\n\t\t\t\tversionHistories: &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tBranchToken: []byte(\"token\"),\n\t\t\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 3,\n\t\t\t\t\t\t\t\t\tVersion: 10,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:               \"domainID\",\n\t\t\t\t\tWorkflowID:             \"workflowID\",\n\t\t\t\t\tRunID:                  \"some-example-run\",\n\t\t\t\t\tFirstExecutionRunID:    \"\",\n\t\t\t\t\tParentDomainID:         \"\",\n\t\t\t\t\tParentWorkflowID:       \"\",\n\t\t\t\t\tParentRunID:            \"\",\n\t\t\t\t\tInitiatedID:            0,\n\t\t\t\t\tCompletionEventBatchID: 0,\n\t\t\t\t\tCompletionEvent:        nil,\n\t\t\t\t\tState:                  0,\n\t\t\t\t\tCloseStatus:            0,\n\t\t\t\t\tLastFirstEventID:       0,\n\t\t\t\t\tLastEventTaskID:        0,\n\t\t\t\t\tNextEventID:            0,\n\t\t\t\t\tLastProcessedEvent:     0,\n\t\t\t\t},\n\t\t\t\tlogger: testlogger.New(t),\n\t\t\t}\n\n\t\t\tmsb.hBuilder = NewHistoryBuilder(&msb)\n\n\t\t\tflushBeforeReady, err := msb.startTransactionHandleDecisionFailover(td.incomingTaskVersion)\n\t\t\tif td.expectedErr {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, td.expectFlushBeforeReady, flushBeforeReady)\n\t\t})\n\t}\n}\n\nfunc TestSimpleGetters(t *testing.T) {\n\n\tmsb := createMSB(t)\n\tassert.Equal(t, msb.versionHistories, msb.GetVersionHistories())\n\n\tbranchToken, err := msb.GetCurrentBranchToken()\n\tassert.Equal(t, msb.versionHistories.Histories[0].BranchToken, branchToken)\n\tassert.NoError(t, err)\n\tassert.Equal(t, msb.currentVersion, msb.GetCurrentVersion())\n\tassert.Equal(t, msb.domainEntry, msb.GetDomainEntry())\n\tassert.Equal(t, msb.executionInfo, msb.GetExecutionInfo())\n\tassert.Equal(t, msb.hBuilder, msb.GetHistoryBuilder())\n\tassert.Equal(t, msb.executionStats.HistorySize, msb.GetHistorySize())\n\tassert.Equal(t, msb.executionInfo.LastFirstEventID, msb.GetLastFirstEventID())\n\tlastWriteVersion, err := msb.GetLastWriteVersion()\n\n\titem, err := msb.versionHistories.Histories[0].GetLastItem()\n\tassert.NoError(t, err)\n\tassert.Equal(t, item.Version, lastWriteVersion)\n\n\tassert.Equal(t, msb.executionInfo.NextEventID, msb.GetNextEventID())\n\tassert.Equal(t, msb.pendingRequestCancelInfoIDs, msb.GetPendingRequestCancelExternalInfos())\n\tassert.Equal(t, msb.executionInfo.LastProcessedEvent, msb.GetPreviousStartedEventID())\n\tassert.Equal(t, msb.queryRegistry, msb.GetQueryRegistry())\n\n\tstartVersion, err := msb.GetStartVersion()\n\tassert.NoError(t, err)\n\tassert.Equal(t, msb.versionHistories.Histories[0].Items[0].Version, startVersion)\n\tassert.Equal(t, msb.insertTimerTasks, msb.GetTimerTasks())\n\tassert.Equal(t, msb.insertTransferTasks, msb.GetTransferTasks())\n\tassert.Equal(t, msb.nextEventIDInDB, msb.GetUpdateCondition())\n\tassert.Equal(t, msb.versionHistories, msb.GetVersionHistories())\n\n\tstate, closeStatus := msb.GetWorkflowStateCloseStatus()\n\tassert.Equal(t, msb.executionInfo.CloseStatus, closeStatus)\n\tassert.Equal(t, msb.executionInfo.State, state)\n\tassert.Equal(t, &types.WorkflowType{Name: msb.executionInfo.WorkflowTypeName}, msb.GetWorkflowType())\n\n\tpendingActivityInfo, activityInfoIsPresent := msb.GetActivityInfo(1232)\n\tassert.Equal(t, msb.pendingActivityInfoIDs[1232], pendingActivityInfo)\n\tassert.True(t, activityInfoIsPresent)\n\n\tassert.Equal(t, msb.pendingActivityInfoIDs, msb.GetPendingActivityInfos())\n\n\tpendingRequestCancelledInfo, ok := msb.GetRequestCancelInfo(13)\n\tassert.Equal(t, msb.pendingRequestCancelInfoIDs[13], pendingRequestCancelledInfo)\n\tassert.True(t, ok)\n\n\tpendingChildExecutions, ok := msb.GetChildExecutionInfo(1)\n\tassert.Equal(t, msb.pendingChildExecutionInfoIDs[1], pendingChildExecutions)\n\tassert.True(t, ok)\n\n}\n\nfunc TestMutableState_IsCurrentWorkflowGuaranteed(t *testing.T) {\n\ttests := map[string]struct {\n\t\tstate    int\n\t\texpected bool\n\t}{\n\t\t\"created\": {\n\t\t\tstate:    persistence.WorkflowStateCreated,\n\t\t\texpected: true,\n\t\t},\n\t\t\"running\": {\n\t\t\tstate:    persistence.WorkflowStateCreated,\n\t\t\texpected: true,\n\t\t},\n\t\t\"completed\": {\n\t\t\tstate:    persistence.WorkflowStateCompleted,\n\t\t\texpected: false,\n\t\t},\n\t\t\"void\": {\n\t\t\tstate:    persistence.WorkflowStateVoid,\n\t\t\texpected: false,\n\t\t},\n\t\t\"zombie state\": {\n\t\t\tstate:    persistence.WorkflowStateZombie,\n\t\t\texpected: false,\n\t\t},\n\t\t\"corrupted state\": {\n\t\t\tstate:    persistence.WorkflowStateCorrupted,\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmsb := mutableStateBuilder{\n\t\t\t\tstateInDB: td.state,\n\t\t\t\tlogger:    testlogger.New(t),\n\t\t\t}\n\t\t\tassert.Equal(t, td.expected, msb.IsCurrentWorkflowGuaranteed())\n\t\t})\n\t}\n}\n\nfunc createMSB(t *testing.T) mutableStateBuilder {\n\n\tsampleDomain := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"domain-id\", Name: \"domain\"}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\n\treturn mutableStateBuilder{\n\t\tpendingActivityInfoIDs: map[int64]*persistence.ActivityInfo{\n\t\t\t1232: &persistence.ActivityInfo{ActivityID: \"activityID\"},\n\t\t},\n\t\tpendingActivityIDToEventID: map[string]int64{\n\t\t\t\"activityID\": 6,\n\t\t},\n\t\tupdateActivityInfos: map[int64]*persistence.ActivityInfo{\n\t\t\t7: &persistence.ActivityInfo{DomainID: \"domainID\"},\n\t\t},\n\t\tdeleteActivityInfos: map[int64]struct{}{\n\t\t\t8: struct{}{},\n\t\t},\n\t\tsyncActivityTasks: map[int64]struct{}{},\n\t\tpendingTimerInfoIDs: map[string]*persistence.TimerInfo{\n\t\t\t\"testdata-pendingTimerInfoIDs\": &persistence.TimerInfo{\n\t\t\t\tVersion: 1,\n\t\t\t\tTimerID: \"1232\",\n\t\t\t},\n\t\t},\n\t\tpendingTimerEventIDToID: map[int64]string{},\n\t\tupdateTimerInfos: map[string]*persistence.TimerInfo{\n\t\t\t\"testdata-updatedtimerinfos\": &persistence.TimerInfo{\n\t\t\t\tVersion: 1,\n\t\t\t\tTimerID: \"1232\",\n\t\t\t},\n\t\t},\n\t\tdeleteTimerInfos: map[string]struct{}{},\n\t\tpendingChildExecutionInfoIDs: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t1: &persistence.ChildExecutionInfo{\n\t\t\t\tWorkflowTypeName: \"sample-workflow\",\n\t\t\t},\n\t\t},\n\t\tupdateChildExecutionInfos: map[int64]*persistence.ChildExecutionInfo{\n\t\t\t8: &persistence.ChildExecutionInfo{DomainID: \"updateChildInfosDomainID\"},\n\t\t},\n\t\tdeleteChildExecutionInfos: map[int64]struct{}{\n\t\t\t12: struct{}{},\n\t\t},\n\t\tpendingRequestCancelInfoIDs: map[int64]*persistence.RequestCancelInfo{\n\t\t\t13: &persistence.RequestCancelInfo{InitiatedID: 16},\n\t\t},\n\t\tupdateRequestCancelInfos: map[int64]*persistence.RequestCancelInfo{},\n\t\tdeleteRequestCancelInfos: map[int64]struct{}{\n\t\t\t15: struct{}{},\n\t\t},\n\t\tpendingSignalInfoIDs:      map[int64]*persistence.SignalInfo{},\n\t\tupdateSignalInfos:         map[int64]*persistence.SignalInfo{},\n\t\tdeleteSignalInfos:         map[int64]struct{}{},\n\t\tpendingSignalRequestedIDs: map[string]struct{}{},\n\t\tupdateSignalRequestedIDs:  map[string]struct{}{},\n\t\tdeleteSignalRequestedIDs:  map[string]struct{}{},\n\t\tbufferedEvents:            []*types.HistoryEvent{},\n\t\tupdateBufferedEvents:      []*types.HistoryEvent{},\n\t\tclearBufferedEvents:       false,\n\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:                           \"d9cbf563-3056-4387-b2ac-5fddd868fe4d\",\n\t\t\tWorkflowID:                         \"53fc235c-093e-4b15-9d9d-045e61354b91\",\n\t\t\tRunID:                              \"a2901718-ac12-443e-873d-b100f45d55d8\",\n\t\t\tFirstExecutionRunID:                \"a2901718-ac12-443e-873d-b100f45d55d8\",\n\t\t\tInitiatedID:                        -7,\n\t\t\tTaskList:                           \"tl\",\n\t\t\tWorkflowTypeName:                   \"test\",\n\t\t\tWorkflowTimeout:                    600000000,\n\t\t\tDecisionStartToCloseTimeout:        10,\n\t\t\tState:                              1,\n\t\t\tLastFirstEventID:                   1,\n\t\t\tLastEventTaskID:                    2097153,\n\t\t\tNextEventID:                        3,\n\t\t\tLastProcessedEvent:                 -23,\n\t\t\tStartTimestamp:                     time.Date(2024, 10, 21, 20, 58, 1, 275000000, time.UTC),\n\t\t\tLastUpdatedTimestamp:               time.Date(2024, 10, 21, 20, 58, 1, 275000000, time.UTC),\n\t\t\tCreateRequestID:                    \"f33ee669-9ff6-4221-a2b0-feb2959667b8\",\n\t\t\tDecisionVersion:                    1,\n\t\t\tDecisionScheduleID:                 2,\n\t\t\tDecisionStartedID:                  -23,\n\t\t\tDecisionRequestID:                  \"emptyUuid\",\n\t\t\tDecisionTimeout:                    10,\n\t\t\tDecisionScheduledTimestamp:         1729544281275414000,\n\t\t\tDecisionOriginalScheduledTimestamp: 1729544281275414000,\n\t\t\tAutoResetPoints:                    &types.ResetPoints{},\n\t\t},\n\t\tversionHistories: &persistence.VersionHistories{\n\t\t\tHistories: []*persistence.VersionHistory{\n\t\t\t\t{\n\t\t\t\t\tBranchToken: []byte(\"a branch token\"),\n\t\t\t\t\tItems: []*persistence.VersionHistoryItem{{\n\t\t\t\t\t\tEventID: 2,\n\t\t\t\t\t\tVersion: 1,\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tcurrentVersion:        int64(-24),\n\t\thasBufferedEventsInDB: false,\n\t\tstateInDB:             int(1),\n\t\tnextEventIDInDB:       int64(3),\n\t\tdomainEntry:           sampleDomain,\n\t\tappliedEvents:         map[string]struct{}{},\n\t\tinsertTransferTasks: []persistence.Task{\n\t\t\t&persistence.DecisionTask{\n\t\t\t\tTargetDomainID: \"decsion task\",\n\t\t\t},\n\t\t},\n\t\tinsertReplicationTasks: []persistence.Task{},\n\t\tinsertTimerTasks: []persistence.Task{\n\t\t\t&persistence.ActivityRetryTimerTask{\n\t\t\t\tTaskData: persistence.TaskData{},\n\t\t\t\tEventID:  123,\n\t\t\t\tAttempt:  4,\n\t\t\t},\n\t\t},\n\t\tworkflowRequests: map[persistence.WorkflowRequest]struct{}{},\n\t\tchecksum:         checksum.Checksum{},\n\t\texecutionStats:   &persistence.ExecutionStats{HistorySize: 403},\n\t\tqueryRegistry:    query.NewRegistry(),\n\t\tlogger:           testlogger.New(t),\n\t}\n}\n\nfunc TestMutableStateBuilder_GetTransferTasks(t *testing.T) {\n\tmsb := &mutableStateBuilder{\n\t\tinsertTransferTasks: []persistence.Task{\n\t\t\t&persistence.ActivityTask{},\n\t\t\t&persistence.DecisionTask{},\n\t\t},\n\t}\n\ttasks := msb.GetTransferTasks()\n\tassert.Equal(t, 2, len(tasks))\n\tassert.IsType(t, &persistence.ActivityTask{}, tasks[0])\n\tassert.IsType(t, &persistence.DecisionTask{}, tasks[1])\n}\n\nfunc TestMutableStateBuilder_GetTimerTasks(t *testing.T) {\n\tmsb := &mutableStateBuilder{\n\t\tinsertTimerTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{},\n\t\t},\n\t}\n\ttasks := msb.GetTimerTasks()\n\tassert.Equal(t, 1, len(tasks))\n\tassert.IsType(t, &persistence.UserTimerTask{}, tasks[0])\n}\n\nfunc TestMutableStateBuilder_DeleteTransferTasks(t *testing.T) {\n\tmsb := &mutableStateBuilder{\n\t\tinsertTransferTasks: []persistence.Task{\n\t\t\t&persistence.ActivityTask{},\n\t\t},\n\t}\n\tmsb.DeleteTransferTasks()\n\tassert.Nil(t, msb.insertTransferTasks)\n}\n\nfunc TestMutableStateBuilder_DeleteTimerTasks(t *testing.T) {\n\tmsb := &mutableStateBuilder{\n\t\tinsertTimerTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{},\n\t\t},\n\t}\n\tmsb.DeleteTimerTasks()\n\tassert.Nil(t, msb.insertTimerTasks)\n}\n\nfunc TestMutableStateBuilder_SetUpdateCondition(t *testing.T) {\n\tmsb := &mutableStateBuilder{}\n\tmsb.SetUpdateCondition(123)\n\tassert.Equal(t, int64(123), msb.nextEventIDInDB)\n}\n\nfunc TestMutableStateBuilder_GetUpdateCondition(t *testing.T) {\n\tmsb := &mutableStateBuilder{\n\t\tnextEventIDInDB: 123,\n\t}\n\tassert.Equal(t, int64(123), msb.GetUpdateCondition())\n}\n\nfunc TestCheckAndClearTimerFiredEvent(t *testing.T) {\n\ttests := []struct {\n\t\tname                         string\n\t\ttimerID                      string\n\t\tbufferedEvents               []*types.HistoryEvent\n\t\tupdateBufferedEvents         []*types.HistoryEvent\n\t\thistory                      []*types.HistoryEvent\n\t\texpectedTimerEvent           *types.HistoryEvent\n\t\texpectedBufferedEvents       []*types.HistoryEvent\n\t\texpectedUpdateBufferedEvents []*types.HistoryEvent\n\t\texpectedHistory              []*types.HistoryEvent\n\t}{\n\t\t{\n\t\t\tname:    \"TimerFiredEventInBufferedEvents\",\n\t\t\ttimerID: \"timer1\",\n\t\t\tbufferedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tupdateBufferedEvents: []*types.HistoryEvent{},\n\t\t\thistory:              []*types.HistoryEvent{},\n\t\t\texpectedTimerEvent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedBufferedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedUpdateBufferedEvents: []*types.HistoryEvent{},\n\t\t\texpectedHistory:              []*types.HistoryEvent{},\n\t\t},\n\t\t{\n\t\t\tname:           \"TimerFiredEventInUpdateBufferedEvents\",\n\t\t\ttimerID:        \"timer2\",\n\t\t\tbufferedEvents: []*types.HistoryEvent{},\n\t\t\tupdateBufferedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer3\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: []*types.HistoryEvent{},\n\t\t\texpectedTimerEvent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\tTimerID: \"timer2\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedBufferedEvents: []*types.HistoryEvent{},\n\t\t\texpectedUpdateBufferedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer3\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedHistory: []*types.HistoryEvent{},\n\t\t},\n\t\t{\n\t\t\tname:                 \"TimerFiredEventInHistory\",\n\t\t\ttimerID:              \"timer3\",\n\t\t\tbufferedEvents:       []*types.HistoryEvent{},\n\t\t\tupdateBufferedEvents: []*types.HistoryEvent{},\n\t\t\thistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer3\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedTimerEvent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\tTimerID: \"timer3\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedBufferedEvents:       []*types.HistoryEvent{},\n\t\t\texpectedUpdateBufferedEvents: []*types.HistoryEvent{},\n\t\t\texpectedHistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeTimerFired.Ptr(),\n\t\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{\n\t\t\t\t\t\tTimerID: \"timer1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                         \"NoTimerFiredEvent\",\n\t\t\ttimerID:                      \"timer4\",\n\t\t\tbufferedEvents:               []*types.HistoryEvent{},\n\t\t\tupdateBufferedEvents:         []*types.HistoryEvent{},\n\t\t\thistory:                      []*types.HistoryEvent{},\n\t\t\texpectedTimerEvent:           nil,\n\t\t\texpectedBufferedEvents:       []*types.HistoryEvent{},\n\t\t\texpectedUpdateBufferedEvents: []*types.HistoryEvent{},\n\t\t\texpectedHistory:              []*types.HistoryEvent{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmsb := &mutableStateBuilder{\n\t\t\t\tbufferedEvents:       tt.bufferedEvents,\n\t\t\t\tupdateBufferedEvents: tt.updateBufferedEvents,\n\t\t\t\thBuilder:             &HistoryBuilder{history: tt.history},\n\t\t\t}\n\n\t\t\ttimerEvent := msb.checkAndClearTimerFiredEvent(tt.timerID)\n\n\t\t\tassert.Equal(t, tt.expectedTimerEvent, timerEvent)\n\t\t\tassert.Equal(t, tt.expectedBufferedEvents, msb.bufferedEvents)\n\t\t\tassert.Equal(t, tt.expectedUpdateBufferedEvents, msb.updateBufferedEvents)\n\t\t\tassert.Equal(t, tt.expectedHistory, msb.hBuilder.history)\n\t\t})\n\t}\n}\n\nfunc TestAssignTaskIDToTransientHistoryEvents(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\ttransientHistory         []*types.HistoryEvent\n\t\ttaskID                   int64\n\t\tshardContextExpectations func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache)\n\t\texpectedEvents           []*types.HistoryEvent\n\t\texpectedErr              error\n\t}{\n\t\t\"AssignTaskIDToSingleEvent - transient\": {\n\t\t\ttransientHistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 123,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(1).Return([]int64{123}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    123,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"AssignTaskIDToMultipleEvents - transient\": {\n\t\t\ttransientHistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 456,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(2).Return([]int64{123, 124}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    123,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t\tTaskID:    124,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"NoEvents - transient events\": {\n\t\t\ttransientHistory: []*types.HistoryEvent{},\n\t\t\ttaskID:           789,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{},\n\t\t},\n\t\t\"error returned\": {\n\t\t\ttransientHistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 456,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(1).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: assert.AnError,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\ttd.shardContextExpectations(mockCache, shardContext, mockDomainCache)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\tmsb.hBuilder.transientHistory = td.transientHistory\n\n\t\t\terr := msb.assignTaskIDToEvents()\n\n\t\t\tassert.Equal(t, td.expectedEvents, msb.hBuilder.transientHistory)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestAssignTaskIDToHistoryEvents(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\thistory                  []*types.HistoryEvent\n\t\ttaskID                   int64\n\t\tshardContextExpectations func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache)\n\t\texpectedEvents           []*types.HistoryEvent\n\t\texpectedErr              error\n\t}{\n\t\t\"AssignTaskIDToSingleEvent\": {\n\t\t\thistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 123,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(1).Return([]int64{123}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    123,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"AssignTaskIDToMultipleEvents\": {\n\t\t\thistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 456,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(2).Return([]int64{123, 124}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    123,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t\tTaskID:    124,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"NoEvents - transient events\": {\n\t\t\thistory: []*types.HistoryEvent{},\n\t\t\ttaskID:  789,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{},\n\t\t},\n\t\t\"error returned\": {\n\t\t\thistory: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\ttaskID: 456,\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(1).Return(nil, assert.AnError).Times(1)\n\t\t\t},\n\t\t\texpectedEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: assert.AnError,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\ttd.shardContextExpectations(mockCache, shardContext, mockDomainCache)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\tmsb.hBuilder.history = td.history\n\n\t\t\terr := msb.assignTaskIDToEvents()\n\n\t\t\tassert.Equal(t, td.expectedEvents, msb.hBuilder.history)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestAddUpsertWorkflowSearchAttributesEvent(t *testing.T) {\n\n\tnow := time.Unix(1730353941, 0)\n\n\ttests := map[string]struct {\n\t\tdecisionCompletedEventID int64\n\t\trequest                  *types.UpsertWorkflowSearchAttributesDecisionAttributes\n\t\tmutableStateBuilderSetup func(m *mutableStateBuilder)\n\t\texpectedEvent            *types.HistoryEvent\n\t\texpectedErr              error\n\t}{\n\t\t\"successful upsert\": {\n\t\t\tdecisionCompletedEventID: 123,\n\t\t\trequest: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\"CustomKeywordField\": []byte(\"keyword\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmutableStateBuilderSetup: func(m *mutableStateBuilder) {\n\t\t\t},\n\t\t\texpectedEvent: &types.HistoryEvent{\n\t\t\t\tID:        1,\n\t\t\t\tEventType: types.EventTypeUpsertWorkflowSearchAttributes.Ptr(),\n\t\t\t\tUpsertWorkflowSearchAttributesEventAttributes: &types.UpsertWorkflowSearchAttributesEventAttributes{\n\t\t\t\t\tDecisionTaskCompletedEventID: 123,\n\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\t\"CustomKeywordField\": []byte(\"keyword\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTaskID:    commonconstants.EmptyEventTaskID,\n\t\t\t\tVersion:   commonconstants.EmptyVersion,\n\t\t\t\tTimestamp: common.Ptr(now.UnixNano()),\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t\"mutability check fails\": {\n\t\t\tdecisionCompletedEventID: 123,\n\t\t\trequest: &types.UpsertWorkflowSearchAttributesDecisionAttributes{\n\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\"CustomKeywordField\": []byte(\"keyword\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmutableStateBuilderSetup: func(m *mutableStateBuilder) {\n\t\t\t\tm.executionInfo.State = persistence.WorkflowStateCompleted\n\t\t\t},\n\t\t\texpectedEvent: nil,\n\t\t\texpectedErr:   &types.InternalServiceError{Message: \"invalid mutable state action: mutation after finish\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tnowClock := clock.NewMockedTimeSourceAt(now)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\tmsb.hBuilder = &HistoryBuilder{\n\t\t\t\thistory:   []*types.HistoryEvent{},\n\t\t\t\tmsBuilder: msb,\n\t\t\t}\n\n\t\t\ttd.mutableStateBuilderSetup(msb)\n\n\t\t\tmsb.timeSource = nowClock\n\n\t\t\tevent, err := msb.AddUpsertWorkflowSearchAttributesEvent(td.decisionCompletedEventID, td.request)\n\n\t\t\tassert.Equal(t, td.expectedEvent, event)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestCloseTransactionAsMutation(t *testing.T) {\n\n\tnow := time.Unix(500, 0)\n\n\tmockDomain := cache.NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{Name: \"domain\"}, &persistence.DomainConfig{\n\t\tBadBinaries: types.BadBinaries{},\n\t}, \"cluster0\")\n\n\ttests := map[string]struct {\n\t\tmutableStateSetup        func(ms *mutableStateBuilder)\n\t\tshardContextExpectations func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache)\n\t\ttransactionPolicy        TransactionPolicy\n\t\texpectedMutation         *persistence.WorkflowMutation\n\t\texpectedEvent            []*persistence.WorkflowEvents\n\t\texpectedErr              error\n\t}{\n\t\t\"no buffered events\": {\n\t\t\tmutableStateSetup: func(ms *mutableStateBuilder) {\n\t\t\t\tms.executionInfo.DomainID = \"some-domain-id\"\n\t\t\t\tms.executionInfo.NextEventID = 10\n\t\t\t\tms.executionInfo.LastProcessedEvent = 5\n\t\t\t\tms.executionInfo.State = persistence.WorkflowStateRunning\n\t\t\t\tms.executionInfo.CloseStatus = persistence.WorkflowCloseStatusNone\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tNumberOfShards:                        2,\n\t\t\t\t\tIsAdvancedVisConfigExist:              false,\n\t\t\t\t\tMaxResponseSize:                       0,\n\t\t\t\t\tMutableStateChecksumInvalidateBefore:  dynamicproperties.GetFloatPropertyFn(10),\n\t\t\t\t\tMutableStateChecksumVerifyProbability: dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\t\t\t\tHostName:                              \"test-host\",\n\t\t\t\t\tEnableReplicationTaskGeneration:       func(string, string) bool { return true },\n\t\t\t\t\tMaximumBufferedEventsBatch:            func(...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t}).Times(2)\n\n\t\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(mockDomain, nil)\n\n\t\t\t},\n\t\t\texpectedMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:             \"some-domain-id\",\n\t\t\t\t\tNextEventID:          10,\n\t\t\t\t\tLastProcessedEvent:   5,\n\t\t\t\t\tState:                persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus:          persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tLastUpdatedTimestamp: now,\n\t\t\t\t\tDecisionVersion:      commonconstants.EmptyVersion,\n\t\t\t\t\tDecisionScheduleID:   commonconstants.EmptyEventID,\n\t\t\t\t\tDecisionRequestID:    commonconstants.EmptyUUID,\n\t\t\t\t\tDecisionStartedID:    commonconstants.EmptyEventID,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer:    nil,\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer:       nil,\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: nil,\n\t\t\t\t},\n\t\t\t\tUpsertActivityInfos:       []*persistence.ActivityInfo{},\n\t\t\t\tDeleteActivityInfos:       []int64{},\n\t\t\t\tUpsertTimerInfos:          []*persistence.TimerInfo{},\n\t\t\t\tDeleteTimerInfos:          []string{},\n\t\t\t\tUpsertChildExecutionInfos: []*persistence.ChildExecutionInfo{},\n\t\t\t\tUpsertRequestCancelInfos:  []*persistence.RequestCancelInfo{},\n\t\t\t\tDeleteRequestCancelInfos:  []int64{},\n\t\t\t\tUpsertSignalInfos:         []*persistence.SignalInfo{},\n\t\t\t\tDeleteSignalInfos:         []int64{},\n\t\t\t\tUpsertSignalRequestedIDs:  []string{},\n\t\t\t\tDeleteSignalRequestedIDs:  []string{},\n\t\t\t\tDeleteChildExecutionInfos: []int64{},\n\t\t\t\tWorkflowRequests:          []*persistence.WorkflowRequest{},\n\t\t\t\tCondition:                 0,\n\t\t\t},\n\t\t\texpectedEvent: nil,\n\t\t\texpectedErr:   nil,\n\t\t},\n\t\t\"with buffered events\": {\n\t\t\tmutableStateSetup: func(ms *mutableStateBuilder) {\n\t\t\t\tms.executionInfo.DomainID = \"some-domain-id\"\n\t\t\t\tms.executionInfo.NextEventID = 10\n\t\t\t\tms.executionInfo.LastProcessedEvent = 5\n\t\t\t\tms.executionInfo.State = persistence.WorkflowStateRunning\n\t\t\t\tms.executionInfo.CloseStatus = persistence.WorkflowCloseStatusNone\n\t\t\t\tms.bufferedEvents = []*types.HistoryEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:        1,\n\t\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tshardContextExpectations: func(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tNumberOfShards:                        2,\n\t\t\t\t\tIsAdvancedVisConfigExist:              false,\n\t\t\t\t\tMaxResponseSize:                       0,\n\t\t\t\t\tMutableStateChecksumInvalidateBefore:  dynamicproperties.GetFloatPropertyFn(10),\n\t\t\t\t\tMutableStateChecksumVerifyProbability: dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\t\t\t\tHostName:                              \"test-host\",\n\t\t\t\t\tEnableReplicationTaskGeneration:       func(string, string) bool { return true },\n\t\t\t\t\tMaximumBufferedEventsBatch:            func(...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t}).Times(3)\n\n\t\t\t\tshardContext.EXPECT().GenerateTaskIDs(1).Return([]int64{123}, nil).Times(1)\n\t\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(mockDomain, nil)\n\n\t\t\t},\n\t\t\texpectedMutation: &persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:             \"some-domain-id\",\n\t\t\t\t\tNextEventID:          10,\n\t\t\t\t\tLastProcessedEvent:   5,\n\t\t\t\t\tState:                persistence.WorkflowStateRunning,\n\t\t\t\t\tCloseStatus:          persistence.WorkflowCloseStatusNone,\n\t\t\t\t\tLastUpdatedTimestamp: now,\n\t\t\t\t\tDecisionVersion:      commonconstants.EmptyVersion,\n\t\t\t\t\tDecisionScheduleID:   commonconstants.EmptyEventID,\n\t\t\t\t\tDecisionRequestID:    commonconstants.EmptyUUID,\n\t\t\t\t\tDecisionStartedID:    commonconstants.EmptyEventID,\n\t\t\t\t\tLastFirstEventID:     1,\n\t\t\t\t},\n\t\t\t\tTasksByCategory: map[persistence.HistoryTaskCategory][]persistence.Task{\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer: nil,\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer:    nil,\n\t\t\t\t\tpersistence.HistoryTaskCategoryReplication: []persistence.Task{\n\t\t\t\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\t\tDomainID:   \"some-domain-id\",\n\t\t\t\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFirstEventID: 1,\n\t\t\t\t\t\t\tNextEventID:  2,\n\t\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\t\tVersion:             0,\n\t\t\t\t\t\t\t\tTaskID:              0,\n\t\t\t\t\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tUpsertActivityInfos:       []*persistence.ActivityInfo{},\n\t\t\t\tDeleteActivityInfos:       []int64{},\n\t\t\t\tUpsertTimerInfos:          []*persistence.TimerInfo{},\n\t\t\t\tDeleteTimerInfos:          []string{},\n\t\t\t\tUpsertChildExecutionInfos: []*persistence.ChildExecutionInfo{},\n\t\t\t\tUpsertRequestCancelInfos:  []*persistence.RequestCancelInfo{},\n\t\t\t\tDeleteRequestCancelInfos:  []int64{},\n\t\t\t\tUpsertSignalInfos:         []*persistence.SignalInfo{},\n\t\t\t\tDeleteSignalInfos:         []int64{},\n\t\t\t\tUpsertSignalRequestedIDs:  []string{},\n\t\t\t\tDeleteSignalRequestedIDs:  []string{},\n\t\t\t\tDeleteChildExecutionInfos: []int64{},\n\t\t\t\tWorkflowRequests:          []*persistence.WorkflowRequest{},\n\t\t\t\tCondition:                 0,\n\t\t\t},\n\t\t\texpectedEvent: []*persistence.WorkflowEvents{\n\t\t\t\t{\n\t\t\t\t\tDomainID: \"some-domain-id\",\n\t\t\t\t\tEvents: []*types.HistoryEvent{{\n\t\t\t\t\t\tID: 1, EventType: types.EventTypeWorkflowExecutionStarted.Ptr()},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\n\t\t\tactiveClusterManager := activecluster.NewMockManager(ctrl)\n\t\t\tshardContext.EXPECT().GetActiveClusterManager().Return(activeClusterManager).AnyTimes()\n\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tms := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\t\t\ttd.mutableStateSetup(ms)\n\t\t\ttd.shardContextExpectations(mockCache, shardContext, mockDomainCache)\n\n\t\t\tmutation, workflowEvents, err := ms.CloseTransactionAsMutation(now, td.transactionPolicy)\n\t\t\tassert.Equal(t, td.expectedMutation, mutation)\n\t\t\tassert.Equal(t, td.expectedEvent, workflowEvents)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc Test__reorderAndFilterDuplicateEvents(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tbuildEvents func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent)\n\t\tassertions  func(*testing.T, *observer.ObservedLogs)\n\t}{\n\t\t{\n\t\t\tname: \"no duplicates\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent1.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCompleted)\n\t\t\t\tevent2.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1, event2}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 0, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activity started event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent1.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent2.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activty completed event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCompleted)\n\t\t\t\tevent1.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCompleted)\n\t\t\t\tevent2.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activity canceled event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCanceled)\n\t\t\t\tevent1.ActivityTaskCanceledEventAttributes = &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCanceled)\n\t\t\t\tevent2.ActivityTaskCanceledEventAttributes = &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activity failed event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskFailed)\n\t\t\t\tevent1.ActivityTaskFailedEventAttributes = &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskFailed)\n\t\t\t\tevent2.ActivityTaskFailedEventAttributes = &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"activity timed out event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent1.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent2.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow started event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionStarted)\n\t\t\t\tevent1.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionStarted)\n\t\t\t\tevent2.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow completed event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\t\t\t\tevent1.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\t\t\t\tevent2.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow failed event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionFailed)\n\t\t\t\tevent1.ChildWorkflowExecutionFailedEventAttributes = &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionFailed)\n\t\t\t\tevent2.ChildWorkflowExecutionFailedEventAttributes = &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow timed out event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTimedOut)\n\t\t\t\tevent1.ChildWorkflowExecutionTimedOutEventAttributes = &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTimedOut)\n\t\t\t\tevent2.ChildWorkflowExecutionTimedOutEventAttributes = &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"child workflow canceled event duplicated\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCanceled)\n\t\t\t\tevent1.ChildWorkflowExecutionCanceledEventAttributes = &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCanceled)\n\t\t\t\tevent2.ChildWorkflowExecutionCanceledEventAttributes = &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event1, event2}, []*types.HistoryEvent{event1}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, logs *observer.ObservedLogs) {\n\t\t\t\tassert.Equal(t, 1, logs.FilterMessage(\"Duplicate event found\").Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"reorder events\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent1.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent2.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event2, event1}, []*types.HistoryEvent{event1, event2}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"reorder all event types\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent1.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCompleted)\n\t\t\t\tevent2.ActivityTaskCompletedEventAttributes = &types.ActivityTaskCompletedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent3 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskFailed)\n\t\t\t\tevent3.ActivityTaskFailedEventAttributes = &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent4 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskCanceled)\n\t\t\t\tevent4.ActivityTaskCanceledEventAttributes = &types.ActivityTaskCanceledEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent5 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent5.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent6 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\t\t\t\tevent6.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tStartedEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent7 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionFailed)\n\t\t\t\tevent7.ChildWorkflowExecutionFailedEventAttributes = &types.ChildWorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tStartedEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent8 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCanceled)\n\t\t\t\tevent8.ChildWorkflowExecutionCanceledEventAttributes = &types.ChildWorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tStartedEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent9 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTimedOut)\n\t\t\t\tevent9.ChildWorkflowExecutionTimedOutEventAttributes = &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tStartedEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent10 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionTerminated)\n\t\t\t\tevent10.ChildWorkflowExecutionTerminatedEventAttributes = &types.ChildWorkflowExecutionTerminatedEventAttributes{\n\t\t\t\t\tStartedEventID: 1,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event2, event3, event4, event5, event6, event7, event8, event9, event10, event1}, []*types.HistoryEvent{event1, event2, event3, event4, event5, event6, event7, event8, event9, event10}\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"reorder and remove duplicate events\",\n\t\t\tbuildEvents: func(msb *mutableStateBuilder) ([]*types.HistoryEvent, []*types.HistoryEvent) {\n\t\t\t\tevent1 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskStarted)\n\t\t\t\tevent1.ActivityTaskStartedEventAttributes = &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\tAttempt:          1,\n\t\t\t\t}\n\t\t\t\tevent2 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent2.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent3 := msb.CreateNewHistoryEvent(types.EventTypeActivityTaskTimedOut)\n\t\t\t\tevent3.ActivityTaskTimedOutEventAttributes = &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t}\n\t\t\t\tevent4 := msb.CreateNewHistoryEvent(types.EventTypeStartChildWorkflowExecutionInitiated)\n\t\t\t\tevent4.StartChildWorkflowExecutionInitiatedEventAttributes = &types.StartChildWorkflowExecutionInitiatedEventAttributes{}\n\t\t\t\tevent5 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionStarted)\n\t\t\t\tevent5.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 4,\n\t\t\t\t}\n\t\t\t\tevent6 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionStarted)\n\t\t\t\tevent6.ChildWorkflowExecutionStartedEventAttributes = &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 4,\n\t\t\t\t}\n\t\t\t\tevent7 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\t\t\t\tevent7.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 4,\n\t\t\t\t\tStartedEventID:   5,\n\t\t\t\t}\n\t\t\t\tevent8 := msb.CreateNewHistoryEvent(types.EventTypeChildWorkflowExecutionCompleted)\n\t\t\t\tevent8.ChildWorkflowExecutionCompletedEventAttributes = &types.ChildWorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tInitiatedEventID: 4,\n\t\t\t\t\tStartedEventID:   5,\n\t\t\t\t}\n\n\t\t\t\treturn []*types.HistoryEvent{event2, event1, event3, event4, event5, event6, event7, event8}, []*types.HistoryEvent{event1, event4, event5, event2, event7}\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tcore, observedLogs := observer.New(zap.WarnLevel)\n\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, nil)\n\n\t\t\tmsb.logger = log.NewLogger(zap.New(core))\n\n\t\t\tmsb.executionInfo.DomainID = \"some-domain-id\"\n\t\t\tmsb.executionInfo.WorkflowID = \"some-workflow-id\"\n\t\t\tmsb.executionInfo.RunID = \"some-run-id\"\n\n\t\t\tallEvents, nonDuplicate := tc.buildEvents(msb)\n\n\t\t\tresult := msb.reorderAndFilterDuplicateEvents(allEvents, \"test\")\n\n\t\t\tassert.Equal(t, nonDuplicate, result)\n\t\t\tif tc.assertions != nil {\n\t\t\t\ttc.assertions(t, observedLogs)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc createMSBWithMocks(mockCache *events.MockCache, shardContext *shardCtx.MockContext, mockDomainCache *cache.MockDomainCache, domainEntry *cache.DomainCacheEntry) *mutableStateBuilder {\n\t// the MSB constructor calls a bunch of endpoints on the mocks, so\n\t// put them in here as a set of fixed expectations so the actual mocking\n\t// code can just make expectations on the calls on the returned MSB object\n\t// and not get cluttered with constructor calls\n\tshardContext.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).Times(2)\n\tshardContext.EXPECT().GetEventsCache().Return(mockCache)\n\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                        2,\n\t\tIsAdvancedVisConfigExist:              false,\n\t\tMaxResponseSize:                       0,\n\t\tMutableStateChecksumInvalidateBefore:  dynamicproperties.GetFloatPropertyFn(10),\n\t\tMutableStateChecksumVerifyProbability: dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\tMutableStateChecksumGenProbability:    dynamicproperties.GetIntPropertyFilteredByDomain(0.0),\n\t\tHostName:                              \"test-host\",\n\t\tEnableReplicationTaskGeneration:       func(string, string) bool { return true },\n\t\tMaximumBufferedEventsBatch:            func(...dynamicproperties.FilterOption) int { return 100 },\n\t}).Times(1)\n\tshardContext.EXPECT().GetTimeSource().Return(clock.NewMockedTimeSource())\n\tshardContext.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient())\n\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).Times(1)\n\n\tif domainEntry == nil {\n\t\tdomainEntry = constants.TestGlobalDomainEntry\n\t}\n\n\tmsb := newMutableStateBuilder(shardContext, log.NewNoop(), domainEntry, domainEntry.GetFailoverVersion())\n\treturn msb\n}\n\nfunc TestLoad_ActiveActive(t *testing.T) {\n\tdomainID := \"test-domain-id\"\n\tworkflowID := \"test-workflow-id\"\n\trunID := \"test-run-id\"\n\n\t// Create domain entries for different test scenarios\n\tactiveActiveDomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   domainID,\n\t\t\tName: \"test-domain\",\n\t\t},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"cityID\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"seattle\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"sydney\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"regionID\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\tnonActiveActiveDomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   domainID,\n\t\t\tName: \"test-domain\",\n\t\t},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"cluster0\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: \"cluster0\"},\n\t\t\t},\n\t\t},\n\t\t1,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\t// Create base mutable state for testing\n\tbaseMutableState := buildWorkflowMutableState()\n\tbaseMutableState.ExecutionInfo = &persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tState:      persistence.WorkflowStateRunning,\n\t}\n\n\t// Create base mutable state for testing (without version histories)\n\tbaseMutableStateNoVersionHistory := buildWorkflowMutableState()\n\tbaseMutableStateNoVersionHistory.ExecutionInfo = &persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tState:      persistence.WorkflowStateRunning,\n\t}\n\tbaseMutableStateNoVersionHistory.VersionHistories = nil\n\n\ttests := map[string]struct {\n\t\tdomainEntry                    *cache.DomainCacheEntry\n\t\tmutableState                   *persistence.WorkflowMutableState\n\t\tactiveClusterManagerAffordance func(activeClusterManager *activecluster.MockManager)\n\t\texpectedCurrentVersion         int64\n\t\texpectedErr                    error\n\t}{\n\t\t\"Non-active-active domain\": {\n\t\t\tdomainEntry:  nonActiveActiveDomainEntry,\n\t\t\tmutableState: baseMutableState,\n\t\t\tactiveClusterManagerAffordance: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t},\n\t\t\texpectedCurrentVersion: commonconstants.EmptyVersion,\n\t\t\texpectedErr:            nil,\n\t\t},\n\t\t\"Active-active domain with nil version history - should return EmptyVersion\": {\n\t\t\tdomainEntry:  activeActiveDomainEntry,\n\t\t\tmutableState: baseMutableStateNoVersionHistory,\n\t\t\tactiveClusterManagerAffordance: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t},\n\t\t\texpectedCurrentVersion: commonconstants.EmptyVersion, // GetCurrentVersion returns EmptyVersion when versionHistories is nil\n\t\t\texpectedErr:            nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Setup mocks\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tactiveClusterManager := activecluster.NewMockManager(ctrl)\n\n\t\t\t// Set up shard context expectations\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tshardContext.EXPECT().GetActiveClusterManager().Return(activeClusterManager).AnyTimes()\n\n\t\t\tmsb := createMSBWithMocks(mockCache, shardContext, mockDomainCache, td.domainEntry)\n\n\t\t\t// modify expectation from .Times(1) to .AnyTimes() for the load function\n\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\n\t\t\t// Set up domain cache expectations\n\t\t\tmockDomainCache.EXPECT().GetDomainID(gomock.Any()).Return(\"some-domain-id\", nil).AnyTimes()\n\n\t\t\t// Set up active cluster manager expectations based on test case\n\t\t\ttd.activeClusterManagerAffordance(activeClusterManager)\n\n\t\t\t// Execute Load function\n\t\t\terr := msb.Load(context.Background(), td.mutableState)\n\n\t\t\t// Verify results\n\t\t\tif td.expectedErr != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Equal(t, td.expectedErr.Error(), err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, td.expectedCurrentVersion, msb.GetCurrentVersion())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestStartTransaction(t *testing.T) {\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"test-workflow\"\n\trunID := constants.TestRunID\n\n\t// Create domain entries for different test scenarios\n\tregularDomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   domainID,\n\t\t\tName: \"test-domain\",\n\t\t},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: \"cluster0\",\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: \"cluster0\"},\n\t\t\t},\n\t\t},\n\t\t123, // failover version\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\tactiveActiveDomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   domainID,\n\t\t\tName: \"test-aa-domain\",\n\t\t},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"region0\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster1\",\n\t\t\t\t\t\t\t\tFailoverVersion:   200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t456, // failover version\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\ttestCases := map[string]struct {\n\t\tdomainEntry               *cache.DomainCacheEntry\n\t\tsetupActiveClusterManager func(activeClusterManager *activecluster.MockManager)\n\t\tsetupMutableStateBuilder  func(msBuilder *mutableStateBuilder)\n\t\tincomingTaskVersion       int64\n\t\texpectedFlushDecision     bool\n\t\texpectedVersion           int64\n\t\texpectedError             bool\n\t\texpectedErrorContains     string\n\t}{\n\t\t\"it should successfully update the failover version\": {\n\t\t\tdomainEntry: regularDomainEntry,\n\t\t\tsetupActiveClusterManager: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: int64(123),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tsetupMutableStateBuilder: func(msBuilder *mutableStateBuilder) {},\n\t\t\tincomingTaskVersion:      int64(100),\n\t\t\texpectedFlushDecision:    false,\n\t\t\texpectedVersion:          123,\n\t\t\texpectedError:            false,\n\t\t},\n\t\t\"when the domain is active-active domain it should update the failover version\": {\n\t\t\tdomainEntry: activeActiveDomainEntry,\n\t\t\tsetupActiveClusterManager: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: int64(999),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tsetupMutableStateBuilder: func(msBuilder *mutableStateBuilder) {},\n\t\t\tincomingTaskVersion:      int64(200),\n\t\t\texpectedFlushDecision:    false,\n\t\t\texpectedVersion:          999,\n\t\t\texpectedError:            false,\n\t\t},\n\t\t\"when the domain is active-active and workflow lookup fails it should return an error\": {\n\t\t\tdomainEntry: activeActiveDomainEntry,\n\t\t\tsetupActiveClusterManager: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t).Return(nil, errors.New(\"cluster lookup failed\")).Times(1)\n\t\t\t},\n\t\t\tsetupMutableStateBuilder: func(msBuilder *mutableStateBuilder) {},\n\t\t\tincomingTaskVersion:      int64(300),\n\t\t\texpectedFlushDecision:    false,\n\t\t\texpectedVersion:          0, // version won't be set due to error\n\t\t\texpectedError:            true,\n\t\t\texpectedErrorContains:    \"cluster lookup failed\",\n\t\t},\n\t\t\"when unable to update current version it should return an error\": {\n\t\t\tdomainEntry: regularDomainEntry,\n\t\t\tsetupActiveClusterManager: func(activeClusterManager *activecluster.MockManager) {\n\t\t\t\tactiveClusterManager.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t).Return(&types.ActiveClusterInfo{\n\t\t\t\t\tFailoverVersion: int64(123),\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\tsetupMutableStateBuilder: func(msBuilder *mutableStateBuilder) {\n\t\t\t\t// Create empty version histories to trigger GetCurrentVersionHistory error\n\t\t\t\tmsBuilder.versionHistories = &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories:                  []*persistence.VersionHistory{},\n\t\t\t\t}\n\t\t\t},\n\t\t\tincomingTaskVersion:   int64(400),\n\t\t\texpectedFlushDecision: false,\n\t\t\texpectedVersion:       0, // version won't be updated due to error\n\t\t\texpectedError:         true,\n\t\t\texpectedErrorContains: \"getting branch index\",\n\t\t},\n\t}\n\n\tfor name, tc := range testCases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Setup mocks\n\t\t\tshardContext := shard.NewMockContext(ctrl)\n\t\t\tmockCache := events.NewMockCache(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tactiveClusterManager := activecluster.NewMockManager(ctrl)\n\n\t\t\t// Set up shard context expectations\n\t\t\tshardContext.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\t\t\tshardContext.EXPECT().GetActiveClusterManager().Return(activeClusterManager).AnyTimes()\n\n\t\t\t// Create mutable state builder with mocks\n\t\t\tmsBuilder := createMSBWithMocks(mockCache, shardContext, mockDomainCache, tc.domainEntry)\n\n\t\t\t// Override shard context expectations for multiple calls\n\t\t\tshardContext.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tshardContext.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\tHostName:                        \"test-host\",\n\t\t\t\tEnableReplicationTaskGeneration: func(string, string) bool { return true },\n\t\t\t\tMaximumBufferedEventsBatch:      func(...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t}).AnyTimes()\n\n\t\t\t// Set up basic execution info\n\t\t\tmsBuilder.executionInfo = &persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:    domainID,\n\t\t\t\tWorkflowID:  workflowID,\n\t\t\t\tRunID:       runID,\n\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\tNextEventID: int64(5),\n\t\t\t}\n\n\t\t\t// Set up version histories for UdpateCurrentVersion\n\t\t\tmsBuilder.versionHistories = persistence.NewVersionHistories(&persistence.VersionHistory{\n\t\t\t\tBranchToken: []byte(\"test-branch-token\"),\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{EventID: 1, Version: 1},\n\t\t\t\t},\n\t\t\t})\n\n\t\t\t// Apply test-specific setup\n\t\t\ttc.setupActiveClusterManager(activeClusterManager)\n\t\t\ttc.setupMutableStateBuilder(msBuilder)\n\n\t\t\t// Execute the function under test\n\t\t\tflushDecision, err := msBuilder.StartTransaction(context.Background(), tc.domainEntry, tc.incomingTaskVersion)\n\n\t\t\t// Verify results\n\t\t\tif tc.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif tc.expectedErrorContains != \"\" {\n\t\t\t\t\tassert.Contains(t, err.Error(), tc.expectedErrorContains)\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tc.expectedFlushDecision, flushDecision)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedFlushDecision, flushDecision)\n\n\t\t\t\t// Verify domain entry was updated\n\t\t\t\tassert.Equal(t, tc.domainEntry, msBuilder.domainEntry)\n\n\t\t\t\t// Verify current version was set to expected version\n\t\t\t\tassert.Equal(t, tc.expectedVersion, msBuilder.GetCurrentVersion())\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_decision_task_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mutable_state_decision_task_manager_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"fmt\"\n\t\"runtime/debug\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tmutableStateDecisionTaskManager interface {\n\t\tReplicateDecisionTaskScheduledEvent(\n\t\t\tversion int64,\n\t\t\tscheduleID int64,\n\t\t\ttaskList string,\n\t\t\tstartToCloseTimeoutSeconds int32,\n\t\t\tattempt int64,\n\t\t\tscheduleTimestamp int64,\n\t\t\toriginalScheduledTimestamp int64,\n\t\t\tbypassTaskGeneration bool,\n\t\t) (*DecisionInfo, error)\n\t\tReplicateTransientDecisionTaskScheduled() error\n\t\tReplicateDecisionTaskStartedEvent(\n\t\t\tdecision *DecisionInfo,\n\t\t\tversion int64,\n\t\t\tscheduleID int64,\n\t\t\tstartedID int64,\n\t\t\trequestID string,\n\t\t\ttimestamp int64,\n\t\t) (*DecisionInfo, error)\n\t\tReplicateDecisionTaskCompletedEvent(event *types.HistoryEvent) error\n\t\tReplicateDecisionTaskFailedEvent(*types.HistoryEvent) error\n\t\tReplicateDecisionTaskTimedOutEvent(*types.HistoryEvent) error\n\n\t\tAddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID int64) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskScheduledEventAsHeartbeat(\n\t\t\tbypassTaskGeneration bool,\n\t\t\toriginalScheduledTimestamp int64,\n\t\t) (*DecisionInfo, error)\n\t\tAddDecisionTaskScheduledEvent(bypassTaskGeneration bool) (*DecisionInfo, error)\n\t\tAddFirstDecisionTaskScheduled(startEvent *types.HistoryEvent) error\n\t\tAddDecisionTaskStartedEvent(\n\t\t\tscheduleEventID int64,\n\t\t\trequestID string,\n\t\t\trequest *types.PollForDecisionTaskRequest,\n\t\t) (*types.HistoryEvent, *DecisionInfo, error)\n\t\tAddDecisionTaskCompletedEvent(\n\t\t\tscheduleEventID int64,\n\t\t\tstartedEventID int64,\n\t\t\trequest *types.RespondDecisionTaskCompletedRequest,\n\t\t\tmaxResetPoints int,\n\t\t) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskFailedEvent(\n\t\t\tscheduleEventID int64,\n\t\t\tstartedEventID int64,\n\t\t\tcause types.DecisionTaskFailedCause,\n\t\t\tdetails []byte,\n\t\t\tidentity string,\n\t\t\treason string,\n\t\t\tbinChecksum string,\n\t\t\tbaseRunID string,\n\t\t\tnewRunID string,\n\t\t\tforkEventVersion int64,\n\t\t\tresetRequestID string,\n\t\t) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskTimedOutEvent(scheduleEventID int64, startedEventID int64) (*types.HistoryEvent, error)\n\t\tAddDecisionTaskResetTimeoutEvent(\n\t\t\tscheduleEventID int64,\n\t\t\tbaseRunID string,\n\t\t\tnewRunID string,\n\t\t\tforkEventVersion int64,\n\t\t\treason string,\n\t\t\tresetRequestID string,\n\t\t) (*types.HistoryEvent, error)\n\n\t\tFailDecision(incrementAttempt bool)\n\t\tDeleteDecision()\n\t\tUpdateDecision(decision *DecisionInfo)\n\n\t\tHasPendingDecision() bool\n\t\tGetPendingDecision() (*DecisionInfo, bool)\n\t\tHasInFlightDecision() bool\n\t\tGetInFlightDecision() (*DecisionInfo, bool)\n\t\tHasProcessedOrPendingDecision() bool\n\t\tGetDecisionInfo(scheduleEventID int64) (*DecisionInfo, bool)\n\t\tGetDecisionScheduleToStartTimeout() time.Duration\n\n\t\tCreateTransientDecisionEvents(decision *DecisionInfo, identity string) (*types.HistoryEvent, *types.HistoryEvent)\n\t}\n\n\tmutableStateDecisionTaskManagerImpl struct {\n\t\tmsb *mutableStateBuilder\n\t}\n)\n\nfunc newMutableStateDecisionTaskManager(msb *mutableStateBuilder) mutableStateDecisionTaskManager {\n\treturn &mutableStateDecisionTaskManagerImpl{\n\t\tmsb: msb,\n\t}\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateDecisionTaskScheduledEvent(\n\tversion int64,\n\tscheduleID int64,\n\ttaskList string,\n\tstartToCloseTimeoutSeconds int32,\n\tattempt int64,\n\tscheduleTimestamp int64,\n\toriginalScheduledTimestamp int64,\n\tbypassTaskGeneration bool,\n) (*DecisionInfo, error) {\n\n\t// set workflow state to running, since decision is scheduled\n\t// NOTE: for zombie workflow, should not change the state\n\tstate, _ := m.msb.GetWorkflowStateCloseStatus()\n\tif state != persistence.WorkflowStateZombie {\n\t\tif err := m.msb.UpdateWorkflowStateCloseStatus(\n\t\t\tpersistence.WorkflowStateRunning,\n\t\t\tpersistence.WorkflowCloseStatusNone,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tdecision := &DecisionInfo{\n\t\tVersion:                    version,\n\t\tScheduleID:                 scheduleID,\n\t\tStartedID:                  constants.EmptyEventID,\n\t\tRequestID:                  constants.EmptyUUID,\n\t\tDecisionTimeout:            startToCloseTimeoutSeconds,\n\t\tTaskList:                   taskList,\n\t\tAttempt:                    attempt,\n\t\tScheduledTimestamp:         scheduleTimestamp,\n\t\tStartedTimestamp:           0,\n\t\tOriginalScheduledTimestamp: originalScheduledTimestamp,\n\t}\n\n\tm.UpdateDecision(decision)\n\n\tif !bypassTaskGeneration {\n\t\tif err := m.msb.taskGenerator.GenerateDecisionScheduleTasks(decision.ScheduleID); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn decision, nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateTransientDecisionTaskScheduled() error {\n\tif m.HasPendingDecision() || m.msb.GetExecutionInfo().DecisionAttempt == 0 {\n\t\treturn nil\n\t}\n\n\t// the schedule ID for this decision is guaranteed to be wrong\n\t// since the next event ID is assigned at the very end of when\n\t// all events are applied for replication.\n\t// this is OK\n\t// 1. if a failover happen just after this transient decision,\n\t// AddDecisionTaskStartedEvent will handle the correction of schedule ID\n\t// and set the attempt to 0\n\t// 2. if no failover happen during the life time of this transient decision\n\t// then ReplicateDecisionTaskScheduledEvent will overwrite everything\n\t// including the decision schedule ID\n\tdecision := &DecisionInfo{\n\t\tVersion:            m.msb.GetCurrentVersion(),\n\t\tScheduleID:         m.msb.GetNextEventID(),\n\t\tStartedID:          constants.EmptyEventID,\n\t\tRequestID:          constants.EmptyUUID,\n\t\tDecisionTimeout:    m.msb.GetExecutionInfo().DecisionStartToCloseTimeout,\n\t\tTaskList:           m.msb.GetExecutionInfo().TaskList,\n\t\tAttempt:            m.msb.GetExecutionInfo().DecisionAttempt,\n\t\tScheduledTimestamp: m.msb.timeSource.Now().UnixNano(),\n\t\tStartedTimestamp:   0,\n\t}\n\n\tm.UpdateDecision(decision)\n\n\treturn m.msb.taskGenerator.GenerateDecisionScheduleTasks(decision.ScheduleID)\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateDecisionTaskStartedEvent(\n\tdecision *DecisionInfo,\n\tversion int64,\n\tscheduleID int64,\n\tstartedID int64,\n\trequestID string,\n\ttimestamp int64,\n) (*DecisionInfo, error) {\n\t// Replicator calls it with a nil decision info, and it is safe to always lookup the decision in this case as it\n\t// does not have to deal with transient decision case.\n\tvar ok bool\n\tif decision == nil {\n\t\tdecision, ok = m.GetDecisionInfo(scheduleID)\n\t\tif !ok {\n\t\t\treturn nil, errors.NewInternalFailureError(fmt.Sprintf(\"unable to find decision: %v\", scheduleID))\n\t\t}\n\t\t// setting decision attempt to 0 for decision task replication\n\t\t// this mainly handles transient decision completion\n\t\t// for transient decision, active side will write 2 batch in a \"transaction\"\n\t\t// 1. decision task scheduled & decision task started\n\t\t// 2. decision task completed & other events\n\t\t// since we need to treat each individual event batch as one transaction\n\t\t// certain \"magic\" needs to be done, i.e. setting attempt to 0 so\n\t\t// if first batch is replicated, but not the second one, decision can be correctly timed out\n\t\tdecision.Attempt = 0\n\t}\n\n\t// Update mutable decision state\n\tdecision = &DecisionInfo{\n\t\tVersion:                    version,\n\t\tScheduleID:                 scheduleID,\n\t\tStartedID:                  startedID,\n\t\tRequestID:                  requestID,\n\t\tDecisionTimeout:            decision.DecisionTimeout,\n\t\tAttempt:                    decision.Attempt,\n\t\tStartedTimestamp:           timestamp,\n\t\tScheduledTimestamp:         decision.ScheduledTimestamp,\n\t\tTaskList:                   decision.TaskList,\n\t\tOriginalScheduledTimestamp: decision.OriginalScheduledTimestamp,\n\t}\n\n\tm.UpdateDecision(decision)\n\treturn decision, m.msb.taskGenerator.GenerateDecisionStartTasks(scheduleID)\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateDecisionTaskCompletedEvent(\n\tevent *types.HistoryEvent,\n) error {\n\tm.beforeAddDecisionTaskCompletedEvent()\n\tmaxResetPoints := constants.DefaultHistoryMaxAutoResetPoints // use default when it is not set in the config\n\tif m.msb.GetDomainEntry() != nil && m.msb.GetDomainEntry().GetInfo() != nil && m.msb.config != nil {\n\t\tdomainName := m.msb.GetDomainEntry().GetInfo().Name\n\t\tmaxResetPoints = m.msb.config.MaxAutoResetPoints(domainName)\n\t}\n\treturn m.afterAddDecisionTaskCompletedEvent(event, maxResetPoints)\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateDecisionTaskFailedEvent(event *types.HistoryEvent) error {\n\tif event != nil && event.DecisionTaskFailedEventAttributes.GetCause() == types.DecisionTaskFailedCauseResetWorkflow {\n\t\tm.msb.insertWorkflowRequest(persistence.WorkflowRequest{\n\t\t\tRequestID:   event.DecisionTaskFailedEventAttributes.RequestID,\n\t\t\tVersion:     event.Version,\n\t\t\tRequestType: persistence.WorkflowRequestTypeReset,\n\t\t})\n\t}\n\tm.FailDecision(true)\n\treturn nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) ReplicateDecisionTaskTimedOutEvent(\n\tevent *types.HistoryEvent,\n) error {\n\ttimeoutType := event.DecisionTaskTimedOutEventAttributes.GetTimeoutType()\n\tincrementAttempt := true\n\t// Do not increment decision attempt in the case of sticky scheduleToStart timeout to\n\t// prevent creating next decision as transient\n\t// Note: this is just best effort, stickiness can be cleared before the timer fires,\n\t// and we can't tell is the decision that is having scheduleToStart timeout is sticky\n\t// or not.\n\tif timeoutType == types.TimeoutTypeScheduleToStart &&\n\t\tm.msb.executionInfo.StickyTaskList != \"\" {\n\t\tincrementAttempt = false\n\t}\n\tif event.DecisionTaskTimedOutEventAttributes.GetCause() == types.DecisionTaskTimedOutCauseReset {\n\t\tm.msb.insertWorkflowRequest(persistence.WorkflowRequest{\n\t\t\tRequestID:   event.DecisionTaskTimedOutEventAttributes.GetRequestID(),\n\t\t\tVersion:     event.Version,\n\t\t\tRequestType: persistence.WorkflowRequestTypeReset,\n\t\t})\n\t}\n\tm.FailDecision(incrementAttempt)\n\treturn nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskScheduleToStartTimeoutEvent(\n\tscheduleEventID int64,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tif m.msb.executionInfo.DecisionScheduleID != scheduleEventID || m.msb.executionInfo.DecisionStartedID > 0 {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t)\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tvar event *types.HistoryEvent\n\t// stickyness will be cleared in ReplicateDecisionTaskTimedOutEvent\n\t// Avoid creating new history events when decisions are continuously timing out\n\tif m.msb.executionInfo.DecisionAttempt == 0 {\n\t\tevent = m.msb.hBuilder.AddDecisionTaskTimedOutEvent(\n\t\t\tscheduleEventID,\n\t\t\t0,\n\t\t\ttypes.TimeoutTypeScheduleToStart,\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\tconstants.EmptyVersion,\n\t\t\t\"\",\n\t\t\ttypes.DecisionTaskTimedOutCauseTimeout,\n\t\t\t\"\",\n\t\t)\n\t\tif err := m.ReplicateDecisionTaskTimedOutEvent(event); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\tif err := m.ReplicateDecisionTaskTimedOutEvent(&types.HistoryEvent{\n\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\tTimeoutType: types.TimeoutTypeScheduleToStart.Ptr(),\n\t\t\t},\n\t\t}); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn event, nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskResetTimeoutEvent(\n\tscheduleEventID int64,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\treason string,\n\tresetRequestID string,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tif m.msb.executionInfo.DecisionScheduleID != scheduleEventID {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t)\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tevent := m.msb.hBuilder.AddDecisionTaskTimedOutEvent(\n\t\tscheduleEventID,\n\t\t0,\n\t\ttypes.TimeoutTypeScheduleToStart,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tforkEventVersion,\n\t\treason,\n\t\ttypes.DecisionTaskTimedOutCauseReset,\n\t\tresetRequestID,\n\t)\n\n\tif err := m.ReplicateDecisionTaskTimedOutEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\t// always clear decision attempt for reset\n\tm.msb.executionInfo.DecisionAttempt = 0\n\treturn event, nil\n}\n\n// originalScheduledTimestamp is to record the first scheduled decision during decision heartbeat.\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskScheduledEventAsHeartbeat(\n\tbypassTaskGeneration bool,\n\toriginalScheduledTimestamp int64,\n) (*DecisionInfo, error) {\n\topTag := tag.WorkflowActionDecisionTaskScheduled\n\tif m.HasPendingDecision() {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(m.msb.executionInfo.DecisionScheduleID))\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\t// Tasklist and decision timeout should already be set from workflow execution started event\n\ttaskList := m.msb.executionInfo.TaskList\n\tif m.msb.IsStickyTaskListEnabled() {\n\t\ttaskList = m.msb.executionInfo.StickyTaskList\n\t} else {\n\t\t// It can be because stickyness has expired due to StickyTTL config\n\t\t// In that case we need to clear stickyness so that the LastUpdateTimestamp is not corrupted.\n\t\t// In other cases, clearing stickyness shouldn't hurt anything.\n\t\t// TODO: https://github.com/uber/cadence/issues/2357:\n\t\t//  if we can use a new field(LastDecisionUpdateTimestamp), then we could get rid of it.\n\t\tm.msb.ClearStickyness()\n\t}\n\tstartToCloseTimeoutSeconds := m.msb.executionInfo.DecisionStartToCloseTimeout\n\n\t// Flush any buffered events before creating the decision, otherwise it will result in invalid IDs for transient\n\t// decision and will cause in timeout processing to not work for transient decisions\n\tif m.msb.HasBufferedEvents() {\n\t\t// if creating a decision and in the mean time events are flushed from buffered events\n\t\t// than this decision cannot be a transient decision\n\t\tm.msb.executionInfo.DecisionAttempt = 0\n\t\tif err := m.msb.FlushBufferedEvents(); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tvar newDecisionEvent *types.HistoryEvent\n\tscheduleID := m.msb.GetNextEventID() // we will generate the schedule event later for repeatedly failing decisions\n\t// Avoid creating new history events when decisions are continuously failing\n\tscheduleTime := m.msb.timeSource.Now().UnixNano()\n\tuseNonTransientDecision := m.shouldUpdateLastWriteVersion()\n\n\tif m.msb.executionInfo.DecisionAttempt == 0 || useNonTransientDecision {\n\t\tnewDecisionEvent = m.msb.hBuilder.AddDecisionTaskScheduledEvent(\n\t\t\ttaskList,\n\t\t\tstartToCloseTimeoutSeconds,\n\t\t\tm.msb.executionInfo.DecisionAttempt)\n\t\tscheduleID = newDecisionEvent.ID\n\t\tscheduleTime = newDecisionEvent.GetTimestamp()\n\t\tm.msb.executionInfo.DecisionAttempt = 0\n\t}\n\n\treturn m.ReplicateDecisionTaskScheduledEvent(\n\t\tm.msb.GetCurrentVersion(),\n\t\tscheduleID,\n\t\ttaskList,\n\t\tstartToCloseTimeoutSeconds,\n\t\tm.msb.executionInfo.DecisionAttempt,\n\t\tscheduleTime,\n\t\toriginalScheduledTimestamp,\n\t\tbypassTaskGeneration,\n\t)\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskScheduledEvent(\n\tbypassTaskGeneration bool,\n) (*DecisionInfo, error) {\n\treturn m.AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration, m.msb.timeSource.Now().UnixNano())\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddFirstDecisionTaskScheduled(\n\tstartEvent *types.HistoryEvent,\n) error {\n\t// handle first decision case, i.e. possible delayed decision\n\t//\n\t// below handles the following cases:\n\t// 1. if not continue as new & if workflow has no parent\n\t//   -> schedule decision & schedule delayed decision\n\t// 2. if not continue as new & if workflow has parent\n\t//   -> this function should not be called during workflow start, but should be called as\n\t//      part of schedule decision in 2 phase commit\n\t//\n\t// if continue as new\n\t//  1. whether has parent workflow or not\n\t//   -> schedule decision & schedule delayed decision\n\t//\n\tstartAttr := startEvent.WorkflowExecutionStartedEventAttributes\n\tdecisionBackoffDuration := time.Duration(startAttr.GetFirstDecisionTaskBackoffSeconds()) * time.Second\n\n\tvar err error\n\tif decisionBackoffDuration != 0 {\n\t\tif err = m.msb.taskGenerator.GenerateDelayedDecisionTasks(\n\t\t\tstartEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\tif _, err = m.AddDecisionTaskScheduledEvent(\n\t\t\tfalse,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskStartedEvent(\n\tscheduleEventID int64,\n\trequestID string,\n\trequest *types.PollForDecisionTaskRequest,\n) (*types.HistoryEvent, *DecisionInfo, error) {\n\topTag := tag.WorkflowActionDecisionTaskStarted\n\tdecision, ok := m.GetDecisionInfo(scheduleEventID)\n\tif !ok || decision.StartedID != constants.EmptyEventID {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID))\n\t\treturn nil, nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tvar event *types.HistoryEvent\n\tscheduleID := decision.ScheduleID\n\tstartedID := scheduleID + 1\n\ttasklist := request.TaskList.GetName()\n\tstartTime := m.msb.timeSource.Now().UnixNano()\n\tuseNonTransientDecision := m.shouldUpdateLastWriteVersion()\n\n\t// First check to see if new events came since transient decision was scheduled\n\tif decision.Attempt > 0 && (decision.ScheduleID != m.msb.GetNextEventID() || useNonTransientDecision) {\n\t\t// Also create a new DecisionTaskScheduledEvent since new events came in when it was scheduled\n\t\tscheduleEvent := m.msb.hBuilder.AddDecisionTaskScheduledEvent(tasklist, decision.DecisionTimeout, 0)\n\t\tscheduleID = scheduleEvent.ID\n\t\tdecision.Attempt = 0\n\t}\n\n\t// Avoid creating new history events when decisions are continuously failing\n\tif decision.Attempt == 0 {\n\t\t// Now create DecisionTaskStartedEvent\n\t\tevent = m.msb.hBuilder.AddDecisionTaskStartedEvent(scheduleID, requestID, request.GetIdentity())\n\t\tstartedID = event.ID\n\t\tstartTime = event.GetTimestamp()\n\t}\n\n\tdecision, err := m.ReplicateDecisionTaskStartedEvent(decision, m.msb.GetCurrentVersion(), scheduleID, startedID, requestID, startTime)\n\treturn event, decision, err\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskCompletedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\trequest *types.RespondDecisionTaskCompletedRequest,\n\tmaxResetPoints int,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskCompleted\n\tdecision, ok := m.GetDecisionInfo(scheduleEventID)\n\tif !ok || decision.StartedID != startedEventID {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID))\n\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tm.beforeAddDecisionTaskCompletedEvent()\n\tif decision.Attempt > 0 {\n\t\t// Create corresponding DecisionTaskSchedule and DecisionTaskStarted events for decisions we have been retrying\n\t\tscheduledEvent := m.msb.hBuilder.AddTransientDecisionTaskScheduledEvent(m.msb.executionInfo.TaskList, decision.DecisionTimeout,\n\t\t\tdecision.Attempt, decision.ScheduledTimestamp)\n\t\tstartedEvent := m.msb.hBuilder.AddTransientDecisionTaskStartedEvent(scheduledEvent.ID, decision.RequestID,\n\t\t\trequest.GetIdentity(), decision.StartedTimestamp)\n\t\tstartedEventID = startedEvent.ID\n\t}\n\t// Now write the completed event\n\tevent := m.msb.hBuilder.AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID, request)\n\n\terr := m.afterAddDecisionTaskCompletedEvent(event, maxResetPoints)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskFailedEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n\tcause types.DecisionTaskFailedCause,\n\tdetails []byte,\n\tidentity string,\n\treason string,\n\tbinChecksum string,\n\tbaseRunID string,\n\tnewRunID string,\n\tforkEventVersion int64,\n\tresetRequestID string,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskFailed\n\tattr := types.DecisionTaskFailedEventAttributes{\n\t\tScheduledEventID: scheduleEventID,\n\t\tStartedEventID:   startedEventID,\n\t\tCause:            cause.Ptr(),\n\t\tDetails:          details,\n\t\tIdentity:         identity,\n\t\tReason:           common.StringPtr(reason),\n\t\tBinaryChecksum:   binChecksum,\n\t\tBaseRunID:        baseRunID,\n\t\tNewRunID:         newRunID,\n\t\tForkEventVersion: forkEventVersion,\n\t\tRequestID:        resetRequestID,\n\t}\n\n\tdt, ok := m.GetDecisionInfo(scheduleEventID)\n\tif !ok || dt.StartedID != startedEventID {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID))\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tvar event *types.HistoryEvent\n\t// Only emit DecisionTaskFailedEvent for the very first time\n\tif dt.Attempt == 0 {\n\t\tevent = m.msb.hBuilder.AddDecisionTaskFailedEvent(attr)\n\t}\n\n\tif err := m.ReplicateDecisionTaskFailedEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\n\t// always clear decision attempt for reset\n\tif cause == types.DecisionTaskFailedCauseResetWorkflow ||\n\t\tcause == types.DecisionTaskFailedCauseFailoverCloseDecision {\n\t\tm.msb.executionInfo.DecisionAttempt = 0\n\t}\n\treturn event, nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) AddDecisionTaskTimedOutEvent(\n\tscheduleEventID int64,\n\tstartedEventID int64,\n) (*types.HistoryEvent, error) {\n\topTag := tag.WorkflowActionDecisionTaskTimedOut\n\tdt, ok := m.GetDecisionInfo(scheduleEventID)\n\tif !ok || dt.StartedID != startedEventID {\n\t\tm.msb.logger.Warn(mutableStateInvalidHistoryActionMsg, opTag,\n\t\t\ttag.WorkflowEventID(m.msb.GetNextEventID()),\n\t\t\ttag.ErrorTypeInvalidHistoryAction,\n\t\t\ttag.WorkflowScheduleID(scheduleEventID),\n\t\t\ttag.WorkflowStartedID(startedEventID))\n\t\treturn nil, m.msb.createInternalServerError(opTag)\n\t}\n\n\tevent := &types.HistoryEvent{\n\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t},\n\t}\n\t// Avoid creating new history events when decisions are continuously timing out\n\tif dt.Attempt == 0 {\n\t\tevent = m.msb.hBuilder.AddDecisionTaskTimedOutEvent(\n\t\t\tscheduleEventID,\n\t\t\tstartedEventID,\n\t\t\ttypes.TimeoutTypeStartToClose,\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\tconstants.EmptyVersion,\n\t\t\t\"\",\n\t\t\ttypes.DecisionTaskTimedOutCauseTimeout,\n\t\t\t\"\",\n\t\t)\n\t}\n\n\tif err := m.ReplicateDecisionTaskTimedOutEvent(event); err != nil {\n\t\treturn nil, err\n\t}\n\treturn event, nil\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) FailDecision(\n\tincrementAttempt bool,\n) {\n\t// Clear stickiness whenever decision fails\n\tm.msb.ClearStickyness()\n\n\tfailDecisionInfo := &DecisionInfo{\n\t\tVersion:                    constants.EmptyVersion,\n\t\tScheduleID:                 constants.EmptyEventID,\n\t\tStartedID:                  constants.EmptyEventID,\n\t\tRequestID:                  constants.EmptyUUID,\n\t\tDecisionTimeout:            0,\n\t\tStartedTimestamp:           0,\n\t\tTaskList:                   \"\",\n\t\tOriginalScheduledTimestamp: 0,\n\t}\n\n\tif incrementAttempt {\n\t\tfailDecisionInfo.Attempt = m.msb.executionInfo.DecisionAttempt + 1\n\t\tfailDecisionInfo.ScheduledTimestamp = m.msb.timeSource.Now().UnixNano()\n\n\t\tif failDecisionInfo.Attempt >= int64(m.msb.shard.GetConfig().DecisionRetryCriticalAttempts()) {\n\t\t\tdomainName := m.msb.GetDomainEntry().GetInfo().Name\n\t\t\tdomainTag := metrics.DomainTag(domainName)\n\t\t\tm.msb.metricsClient.Scope(metrics.WorkflowContextScope, domainTag).RecordTimer(metrics.DecisionAttemptTimer, time.Duration(failDecisionInfo.Attempt))\n\t\t\tm.msb.logger.Warn(\"Critical error processing decision task, retrying.\",\n\t\t\t\ttag.WorkflowDomainName(m.msb.GetDomainEntry().GetInfo().Name),\n\t\t\t\ttag.WorkflowID(m.msb.GetExecutionInfo().WorkflowID),\n\t\t\t\ttag.WorkflowRunID(m.msb.GetExecutionInfo().RunID),\n\t\t\t)\n\t\t}\n\t}\n\tm.UpdateDecision(failDecisionInfo)\n}\n\n// DeleteDecision deletes a decision task.\nfunc (m *mutableStateDecisionTaskManagerImpl) DeleteDecision() {\n\tresetDecisionInfo := &DecisionInfo{\n\t\tVersion:            constants.EmptyVersion,\n\t\tScheduleID:         constants.EmptyEventID,\n\t\tStartedID:          constants.EmptyEventID,\n\t\tRequestID:          constants.EmptyUUID,\n\t\tDecisionTimeout:    0,\n\t\tAttempt:            0,\n\t\tStartedTimestamp:   0,\n\t\tScheduledTimestamp: 0,\n\t\tTaskList:           \"\",\n\t\t// Keep the last original scheduled timestamp, so that AddDecisionAsHeartbeat can continue with it.\n\t\tOriginalScheduledTimestamp: m.getDecisionInfo().OriginalScheduledTimestamp,\n\t}\n\tm.UpdateDecision(resetDecisionInfo)\n}\n\n// UpdateDecision updates a decision task.\nfunc (m *mutableStateDecisionTaskManagerImpl) UpdateDecision(\n\tdecision *DecisionInfo,\n) {\n\tm.msb.executionInfo.DecisionVersion = decision.Version\n\tm.msb.executionInfo.DecisionScheduleID = decision.ScheduleID\n\tm.msb.executionInfo.DecisionStartedID = decision.StartedID\n\tm.msb.executionInfo.DecisionRequestID = decision.RequestID\n\tm.msb.executionInfo.DecisionTimeout = decision.DecisionTimeout\n\tm.msb.executionInfo.DecisionAttempt = decision.Attempt\n\tm.msb.executionInfo.DecisionStartedTimestamp = decision.StartedTimestamp\n\tm.msb.executionInfo.DecisionScheduledTimestamp = decision.ScheduledTimestamp\n\tm.msb.executionInfo.DecisionOriginalScheduledTimestamp = decision.OriginalScheduledTimestamp\n\n\t// NOTE: do not update tasklist in execution info\n\n\tif m.msb.logger.DebugOn() {\n\t\tm.msb.logger.Debugf(\n\t\t\t\"Decision Updated: {Schedule: %v, Started: %v, ID: %v, Timeout: %v, Attempt: %v, Timestamp: %v}, Stacktrace: %v\",\n\t\t\tdecision.ScheduleID,\n\t\t\tdecision.StartedID,\n\t\t\tdecision.RequestID,\n\t\t\tdecision.DecisionTimeout,\n\t\t\tdecision.Attempt,\n\t\t\tdecision.StartedTimestamp,\n\t\t\tdebug.Stack(),\n\t\t)\n\t}\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) HasPendingDecision() bool {\n\treturn m.msb.executionInfo.DecisionScheduleID != constants.EmptyEventID\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) GetPendingDecision() (*DecisionInfo, bool) {\n\tif m.msb.executionInfo.DecisionScheduleID == constants.EmptyEventID {\n\t\treturn nil, false\n\t}\n\n\tdecision := m.getDecisionInfo()\n\treturn decision, true\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) HasInFlightDecision() bool {\n\treturn m.msb.executionInfo.DecisionStartedID > 0\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) GetInFlightDecision() (*DecisionInfo, bool) {\n\tif m.msb.executionInfo.DecisionScheduleID == constants.EmptyEventID ||\n\t\tm.msb.executionInfo.DecisionStartedID == constants.EmptyEventID {\n\t\treturn nil, false\n\t}\n\n\tdecision := m.getDecisionInfo()\n\treturn decision, true\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) HasProcessedOrPendingDecision() bool {\n\treturn m.HasPendingDecision() || m.msb.GetPreviousStartedEventID() != constants.EmptyEventID\n}\n\n// GetDecisionInfo returns details about the in-progress decision task\nfunc (m *mutableStateDecisionTaskManagerImpl) GetDecisionInfo(\n\tscheduleEventID int64,\n) (*DecisionInfo, bool) {\n\tdecision := m.getDecisionInfo()\n\tif scheduleEventID == decision.ScheduleID {\n\t\treturn decision, true\n\t}\n\treturn nil, false\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) GetDecisionScheduleToStartTimeout() time.Duration {\n\t// we should not call IsStickyTaskListEnabled which may be false\n\t// if sticky TTL has expired\n\t// NOTE: this function is called in the same mutable state transaction as the one generating the decision task\n\t// we stickiness won't be cleared between creating the decision and getting the timeout\n\tif m.msb.executionInfo.StickyTaskList != \"\" {\n\t\treturn time.Duration(\n\t\t\tm.msb.executionInfo.StickyScheduleToStartTimeout,\n\t\t) * time.Second\n\t}\n\n\tdomainName := m.msb.GetDomainEntry().GetInfo().Name\n\tif m.msb.executionInfo.DecisionAttempt <\n\t\tint64(m.msb.config.NormalDecisionScheduleToStartMaxAttempts(domainName)) {\n\t\treturn m.msb.config.NormalDecisionScheduleToStartTimeout(domainName)\n\t}\n\treturn 0\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) CreateTransientDecisionEvents(\n\tdecision *DecisionInfo,\n\tidentity string,\n) (*types.HistoryEvent, *types.HistoryEvent) {\n\ttasklist := m.msb.executionInfo.TaskList\n\tscheduledEvent := newDecisionTaskScheduledEventWithInfo(\n\t\tdecision.ScheduleID,\n\t\tdecision.ScheduledTimestamp,\n\t\ttasklist,\n\t\tdecision.DecisionTimeout,\n\t\tdecision.Attempt,\n\t)\n\n\tstartedEvent := newDecisionTaskStartedEventWithInfo(\n\t\tdecision.StartedID,\n\t\tdecision.StartedTimestamp,\n\t\tdecision.ScheduleID,\n\t\tdecision.RequestID,\n\t\tidentity,\n\t)\n\n\treturn scheduledEvent, startedEvent\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) getDecisionInfo() *DecisionInfo {\n\ttaskList := m.msb.executionInfo.TaskList\n\tif m.msb.executionInfo.StickyTaskList != \"\" {\n\t\ttaskList = m.msb.executionInfo.StickyTaskList\n\t}\n\treturn &DecisionInfo{\n\t\tVersion:                    m.msb.executionInfo.DecisionVersion,\n\t\tScheduleID:                 m.msb.executionInfo.DecisionScheduleID,\n\t\tStartedID:                  m.msb.executionInfo.DecisionStartedID,\n\t\tRequestID:                  m.msb.executionInfo.DecisionRequestID,\n\t\tDecisionTimeout:            m.msb.executionInfo.DecisionTimeout,\n\t\tAttempt:                    m.msb.executionInfo.DecisionAttempt,\n\t\tStartedTimestamp:           m.msb.executionInfo.DecisionStartedTimestamp,\n\t\tScheduledTimestamp:         m.msb.executionInfo.DecisionScheduledTimestamp,\n\t\tTaskList:                   taskList,\n\t\tOriginalScheduledTimestamp: m.msb.executionInfo.DecisionOriginalScheduledTimestamp,\n\t}\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) beforeAddDecisionTaskCompletedEvent() {\n\t// Make sure to delete decision before adding events.  Otherwise they are buffered rather than getting appended\n\tm.DeleteDecision()\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) afterAddDecisionTaskCompletedEvent(\n\tevent *types.HistoryEvent,\n\tmaxResetPoints int,\n) error {\n\tm.msb.executionInfo.LastProcessedEvent = event.GetDecisionTaskCompletedEventAttributes().GetStartedEventID()\n\treturn m.msb.addBinaryCheckSumIfNotExists(event, maxResetPoints)\n}\n\nfunc (m *mutableStateDecisionTaskManagerImpl) shouldUpdateLastWriteVersion() bool {\n\n\tcurrentVersion := m.msb.GetCurrentVersion()\n\tlastWriteVersion, err := m.msb.GetLastWriteVersion()\n\tif err != nil {\n\t\t// The error is version history has no item. This is expected for the first batch of a workflow.\n\t\treturn false\n\t}\n\treturn currentVersion != lastWriteVersion\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_decision_task_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: mutable_state_decision_task_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source mutable_state_decision_task_manager.go -destination mutable_state_decision_task_manager_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockmutableStateDecisionTaskManager is a mock of mutableStateDecisionTaskManager interface.\ntype MockmutableStateDecisionTaskManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockmutableStateDecisionTaskManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockmutableStateDecisionTaskManagerMockRecorder is the mock recorder for MockmutableStateDecisionTaskManager.\ntype MockmutableStateDecisionTaskManagerMockRecorder struct {\n\tmock *MockmutableStateDecisionTaskManager\n}\n\n// NewMockmutableStateDecisionTaskManager creates a new mock instance.\nfunc NewMockmutableStateDecisionTaskManager(ctrl *gomock.Controller) *MockmutableStateDecisionTaskManager {\n\tmock := &MockmutableStateDecisionTaskManager{ctrl: ctrl}\n\tmock.recorder = &MockmutableStateDecisionTaskManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockmutableStateDecisionTaskManager) EXPECT() *MockmutableStateDecisionTaskManagerMockRecorder {\n\treturn m.recorder\n}\n\n// AddDecisionTaskCompletedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID int64, request *types.RespondDecisionTaskCompletedRequest, maxResetPoints int) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskCompletedEvent\", scheduleEventID, startedEventID, request, maxResetPoints)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskCompletedEvent indicates an expected call of AddDecisionTaskCompletedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskCompletedEvent(scheduleEventID, startedEventID, request, maxResetPoints any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskCompletedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskCompletedEvent), scheduleEventID, startedEventID, request, maxResetPoints)\n}\n\n// AddDecisionTaskFailedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskFailedEvent(scheduleEventID, startedEventID int64, cause types.DecisionTaskFailedCause, details []byte, identity, reason, binChecksum, baseRunID, newRunID string, forkEventVersion int64, resetRequestID string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskFailedEvent\", scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskFailedEvent indicates an expected call of AddDecisionTaskFailedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskFailedEvent(scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskFailedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskFailedEvent), scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID)\n}\n\n// AddDecisionTaskResetTimeoutEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskResetTimeoutEvent(scheduleEventID int64, baseRunID, newRunID string, forkEventVersion int64, reason, resetRequestID string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskResetTimeoutEvent\", scheduleEventID, baseRunID, newRunID, forkEventVersion, reason, resetRequestID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskResetTimeoutEvent indicates an expected call of AddDecisionTaskResetTimeoutEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskResetTimeoutEvent(scheduleEventID, baseRunID, newRunID, forkEventVersion, reason, resetRequestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskResetTimeoutEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskResetTimeoutEvent), scheduleEventID, baseRunID, newRunID, forkEventVersion, reason, resetRequestID)\n}\n\n// AddDecisionTaskScheduleToStartTimeoutEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduleToStartTimeoutEvent\", scheduleEventID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduleToStartTimeoutEvent indicates an expected call of AddDecisionTaskScheduleToStartTimeoutEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskScheduleToStartTimeoutEvent(scheduleEventID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduleToStartTimeoutEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskScheduleToStartTimeoutEvent), scheduleEventID)\n}\n\n// AddDecisionTaskScheduledEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskScheduledEvent(bypassTaskGeneration bool) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduledEvent\", bypassTaskGeneration)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduledEvent indicates an expected call of AddDecisionTaskScheduledEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskScheduledEvent(bypassTaskGeneration any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduledEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskScheduledEvent), bypassTaskGeneration)\n}\n\n// AddDecisionTaskScheduledEventAsHeartbeat mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration bool, originalScheduledTimestamp int64) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduledEventAsHeartbeat\", bypassTaskGeneration, originalScheduledTimestamp)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduledEventAsHeartbeat indicates an expected call of AddDecisionTaskScheduledEventAsHeartbeat.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration, originalScheduledTimestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduledEventAsHeartbeat\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskScheduledEventAsHeartbeat), bypassTaskGeneration, originalScheduledTimestamp)\n}\n\n// AddDecisionTaskStartedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskStartedEvent(scheduleEventID int64, requestID string, request *types.PollForDecisionTaskRequest) (*types.HistoryEvent, *DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskStartedEvent\", scheduleEventID, requestID, request)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*DecisionInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddDecisionTaskStartedEvent indicates an expected call of AddDecisionTaskStartedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskStartedEvent(scheduleEventID, requestID, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskStartedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskStartedEvent), scheduleEventID, requestID, request)\n}\n\n// AddDecisionTaskTimedOutEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddDecisionTaskTimedOutEvent(scheduleEventID, startedEventID int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskTimedOutEvent\", scheduleEventID, startedEventID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskTimedOutEvent indicates an expected call of AddDecisionTaskTimedOutEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddDecisionTaskTimedOutEvent(scheduleEventID, startedEventID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskTimedOutEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddDecisionTaskTimedOutEvent), scheduleEventID, startedEventID)\n}\n\n// AddFirstDecisionTaskScheduled mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) AddFirstDecisionTaskScheduled(startEvent *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddFirstDecisionTaskScheduled\", startEvent)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AddFirstDecisionTaskScheduled indicates an expected call of AddFirstDecisionTaskScheduled.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) AddFirstDecisionTaskScheduled(startEvent any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddFirstDecisionTaskScheduled\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).AddFirstDecisionTaskScheduled), startEvent)\n}\n\n// CreateTransientDecisionEvents mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) CreateTransientDecisionEvents(decision *DecisionInfo, identity string) (*types.HistoryEvent, *types.HistoryEvent) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTransientDecisionEvents\", decision, identity)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*types.HistoryEvent)\n\treturn ret0, ret1\n}\n\n// CreateTransientDecisionEvents indicates an expected call of CreateTransientDecisionEvents.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) CreateTransientDecisionEvents(decision, identity any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTransientDecisionEvents\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).CreateTransientDecisionEvents), decision, identity)\n}\n\n// DeleteDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) DeleteDecision() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DeleteDecision\")\n}\n\n// DeleteDecision indicates an expected call of DeleteDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) DeleteDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).DeleteDecision))\n}\n\n// FailDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) FailDecision(incrementAttempt bool) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"FailDecision\", incrementAttempt)\n}\n\n// FailDecision indicates an expected call of FailDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) FailDecision(incrementAttempt any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).FailDecision), incrementAttempt)\n}\n\n// GetDecisionInfo mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) GetDecisionInfo(scheduleEventID int64) (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDecisionInfo\", scheduleEventID)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetDecisionInfo indicates an expected call of GetDecisionInfo.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) GetDecisionInfo(scheduleEventID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDecisionInfo\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).GetDecisionInfo), scheduleEventID)\n}\n\n// GetDecisionScheduleToStartTimeout mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) GetDecisionScheduleToStartTimeout() time.Duration {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDecisionScheduleToStartTimeout\")\n\tret0, _ := ret[0].(time.Duration)\n\treturn ret0\n}\n\n// GetDecisionScheduleToStartTimeout indicates an expected call of GetDecisionScheduleToStartTimeout.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) GetDecisionScheduleToStartTimeout() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDecisionScheduleToStartTimeout\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).GetDecisionScheduleToStartTimeout))\n}\n\n// GetInFlightDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) GetInFlightDecision() (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetInFlightDecision\")\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetInFlightDecision indicates an expected call of GetInFlightDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) GetInFlightDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetInFlightDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).GetInFlightDecision))\n}\n\n// GetPendingDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) GetPendingDecision() (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingDecision\")\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetPendingDecision indicates an expected call of GetPendingDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) GetPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).GetPendingDecision))\n}\n\n// HasInFlightDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) HasInFlightDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasInFlightDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasInFlightDecision indicates an expected call of HasInFlightDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) HasInFlightDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasInFlightDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).HasInFlightDecision))\n}\n\n// HasPendingDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) HasPendingDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasPendingDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasPendingDecision indicates an expected call of HasPendingDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) HasPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasPendingDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).HasPendingDecision))\n}\n\n// HasProcessedOrPendingDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) HasProcessedOrPendingDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasProcessedOrPendingDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasProcessedOrPendingDecision indicates an expected call of HasProcessedOrPendingDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) HasProcessedOrPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasProcessedOrPendingDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).HasProcessedOrPendingDecision))\n}\n\n// ReplicateDecisionTaskCompletedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateDecisionTaskCompletedEvent(event *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskCompletedEvent\", event)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskCompletedEvent indicates an expected call of ReplicateDecisionTaskCompletedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateDecisionTaskCompletedEvent(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskCompletedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateDecisionTaskCompletedEvent), event)\n}\n\n// ReplicateDecisionTaskFailedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateDecisionTaskFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskFailedEvent indicates an expected call of ReplicateDecisionTaskFailedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateDecisionTaskFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskFailedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateDecisionTaskFailedEvent), arg0)\n}\n\n// ReplicateDecisionTaskScheduledEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateDecisionTaskScheduledEvent(version, scheduleID int64, taskList string, startToCloseTimeoutSeconds int32, attempt, scheduleTimestamp, originalScheduledTimestamp int64, bypassTaskGeneration bool) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskScheduledEvent\", version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, bypassTaskGeneration)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateDecisionTaskScheduledEvent indicates an expected call of ReplicateDecisionTaskScheduledEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateDecisionTaskScheduledEvent(version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, bypassTaskGeneration any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskScheduledEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateDecisionTaskScheduledEvent), version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, bypassTaskGeneration)\n}\n\n// ReplicateDecisionTaskStartedEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateDecisionTaskStartedEvent(decision *DecisionInfo, version, scheduleID, startedID int64, requestID string, timestamp int64) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskStartedEvent\", decision, version, scheduleID, startedID, requestID, timestamp)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateDecisionTaskStartedEvent indicates an expected call of ReplicateDecisionTaskStartedEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateDecisionTaskStartedEvent(decision, version, scheduleID, startedID, requestID, timestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskStartedEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateDecisionTaskStartedEvent), decision, version, scheduleID, startedID, requestID, timestamp)\n}\n\n// ReplicateDecisionTaskTimedOutEvent mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateDecisionTaskTimedOutEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskTimedOutEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskTimedOutEvent indicates an expected call of ReplicateDecisionTaskTimedOutEvent.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateDecisionTaskTimedOutEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskTimedOutEvent\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateDecisionTaskTimedOutEvent), arg0)\n}\n\n// ReplicateTransientDecisionTaskScheduled mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) ReplicateTransientDecisionTaskScheduled() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateTransientDecisionTaskScheduled\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateTransientDecisionTaskScheduled indicates an expected call of ReplicateTransientDecisionTaskScheduled.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) ReplicateTransientDecisionTaskScheduled() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateTransientDecisionTaskScheduled\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).ReplicateTransientDecisionTaskScheduled))\n}\n\n// UpdateDecision mocks base method.\nfunc (m *MockmutableStateDecisionTaskManager) UpdateDecision(decision *DecisionInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateDecision\", decision)\n}\n\n// UpdateDecision indicates an expected call of UpdateDecision.\nfunc (mr *MockmutableStateDecisionTaskManagerMockRecorder) UpdateDecision(decision any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDecision\", reflect.TypeOf((*MockmutableStateDecisionTaskManager)(nil).UpdateDecision), decision)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_decision_task_manager_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestReplicateDecisionTaskCompletedEvent(t *testing.T) {\n\tmockShard := shard.NewTestContext(\n\t\tt,\n\t\tgomock.NewController(t),\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\tmockShard.GetConfig().MutableStateChecksumGenProbability = func(domain string) int { return 100 }\n\tmockShard.GetConfig().MutableStateChecksumVerifyProbability = func(domain string) int { return 100 }\n\tlogger := mockShard.GetLogger()\n\tmockShard.Resource.DomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\n\tm := &mutableStateDecisionTaskManagerImpl{\n\t\tmsb: newMutableStateBuilder(mockShard, logger, constants.TestLocalDomainEntry, constants.TestLocalDomainEntry.GetFailoverVersion()),\n\t}\n\teventType := types.EventTypeActivityTaskCompleted\n\te := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: &eventType,\n\t}\n\terr := m.ReplicateDecisionTaskCompletedEvent(e)\n\tassert.NoError(t, err)\n\n\t// test when domainEntry is missed\n\tm.msb.domainEntry = nil\n\terr = m.ReplicateDecisionTaskCompletedEvent(e)\n\tassert.NoError(t, err)\n\n\t// test when config is nil\n\tm.msb = newMutableStateBuilder(mockShard, logger, constants.TestLocalDomainEntry, constants.TestLocalDomainEntry.GetFailoverVersion())\n\tm.msb.config = nil\n\terr = m.ReplicateDecisionTaskCompletedEvent(e)\n\tassert.NoError(t, err)\n}\n\nfunc TestReplicateDecisionTaskScheduledEvent(t *testing.T) {\n\tversion := int64(123)\n\tscheduleID := int64(1)\n\ttaskList := \"task-list\"\n\tstartToCloseTimeoutSeconds := int32(100)\n\tattempt := int64(1)\n\tscheduleTimestamp := int64(1)\n\toriginalScheduledTimestamp := int64(0)\n\tbypassTaskGeneration := false\n\ttests := []struct {\n\t\tname         string\n\t\tassertions   func(t *testing.T, info *DecisionInfo, err error, logs *observer.ObservedLogs)\n\t\texpectations func(mgr *mutableStateDecisionTaskManagerImpl)\n\t\tnewMsb       func(t *testing.T) *mutableStateBuilder\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tnewMsb: func(t *testing.T) *mutableStateBuilder {\n\t\t\t\treturn &mutableStateBuilder{\n\t\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: 0, // persistence.WorkflowStateCreated\n\t\t\t\t\t},\n\t\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectations: func(mgr *mutableStateDecisionTaskManagerImpl) {\n\t\t\t\tmgr.msb.taskGenerator.(*MockMutableStateTaskGenerator).EXPECT().GenerateDecisionScheduleTasks(scheduleID)\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, info *DecisionInfo, err error, observedLogs *observer.ObservedLogs) {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, version, info.Version)\n\t\t\t\tassert.Equal(t, scheduleID, info.ScheduleID)\n\t\t\t\tassert.Equal(t, taskList, info.TaskList)\n\t\t\t\tassert.Equal(t, attempt, info.Attempt)\n\t\t\t\tassert.Equal(t, scheduleTimestamp, info.ScheduledTimestamp)\n\t\t\t\tassert.Equal(t, originalScheduledTimestamp, info.OriginalScheduledTimestamp)\n\t\t\t\tassert.Equal(t, 1, observedLogs.FilterMessageSnippet(fmt.Sprintf(\n\t\t\t\t\t\"Decision Updated: {Schedule: %v, Started: %v, ID: %v, Timeout: %v, Attempt: %v, Timestamp: %v}\",\n\t\t\t\t\tscheduleID,\n\t\t\t\t\tcommonconstants.EmptyEventID,\n\t\t\t\t\tcommonconstants.EmptyUUID,\n\t\t\t\t\tstartToCloseTimeoutSeconds,\n\t\t\t\t\tattempt,\n\t\t\t\t\t0,\n\t\t\t\t)).Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowStateCloseStatus failure\",\n\t\t\tnewMsb: func(t *testing.T) *mutableStateBuilder {\n\t\t\t\treturn &mutableStateBuilder{\n\t\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: 2, // persistence.WorkflowStateCompleted\n\t\t\t\t\t},\n\t\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\t}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, info *DecisionInfo, err error, observedLogs *observer.ObservedLogs) {\n\t\t\t\tassert.ErrorContains(t, err, \"unable to change workflow state from 2 to 1, close status 0\")\n\t\t\t\tassert.Nil(t, info)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GenerateDecisionScheduleTasks failure\",\n\t\t\tnewMsb: func(t *testing.T) *mutableStateBuilder {\n\t\t\t\treturn &mutableStateBuilder{\n\t\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tState: 0, // persistence.WorkflowStateCreated\n\t\t\t\t\t},\n\t\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectations: func(mgr *mutableStateDecisionTaskManagerImpl) {\n\t\t\t\tmgr.msb.taskGenerator.(*MockMutableStateTaskGenerator).EXPECT().GenerateDecisionScheduleTasks(scheduleID).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, info *DecisionInfo, err error, observedLogs *observer.ObservedLogs) {\n\t\t\t\tassert.ErrorContains(t, err, \"some error\")\n\t\t\t\tassert.Nil(t, info)\n\t\t\t\tassert.Equal(t, 1, observedLogs.FilterMessageSnippet(fmt.Sprintf(\n\t\t\t\t\t\"Decision Updated: {Schedule: %v, Started: %v, ID: %v, Timeout: %v, Attempt: %v, Timestamp: %v}\",\n\t\t\t\t\tscheduleID,\n\t\t\t\t\tcommonconstants.EmptyEventID,\n\t\t\t\t\tcommonconstants.EmptyUUID,\n\t\t\t\t\tstartToCloseTimeoutSeconds,\n\t\t\t\t\tattempt,\n\t\t\t\t\t0,\n\t\t\t\t)).Len())\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tm := &mutableStateDecisionTaskManagerImpl{msb: test.newMsb(t)}\n\t\t\tcore, observedLogs := observer.New(zap.DebugLevel)\n\t\t\tm.msb.logger = log.NewLogger(zap.New(core))\n\t\t\tif test.expectations != nil {\n\t\t\t\ttest.expectations(m)\n\t\t\t}\n\t\t\tinfo, err := m.ReplicateDecisionTaskScheduledEvent(version, scheduleID, taskList, startToCloseTimeoutSeconds, attempt, scheduleTimestamp, originalScheduledTimestamp, bypassTaskGeneration)\n\t\t\ttest.assertions(t, info, err, observedLogs)\n\t\t})\n\t}\n}\n\nfunc TestReplicateTransientDecisionTaskScheduled(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\texpectations func(mgr *mutableStateDecisionTaskManagerImpl)\n\t\tnewMsb       func(t *testing.T) *mutableStateBuilder\n\t\tassertions   func(t *testing.T, err error, logs *observer.ObservedLogs)\n\t}{\n\t\t{\n\t\t\tname: \"success - decision updated\",\n\t\t\tnewMsb: func(t *testing.T) *mutableStateBuilder {\n\t\t\t\treturn &mutableStateBuilder{\n\t\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\t\t\tDecisionAttempt:    1,\n\t\t\t\t\t},\n\t\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\t\ttimeSource:    clock.NewMockedTimeSource(),\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectations: func(mgr *mutableStateDecisionTaskManagerImpl) {\n\t\t\t\tmgr.msb.taskGenerator.(*MockMutableStateTaskGenerator).EXPECT().GenerateDecisionScheduleTasks(int64(0))\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, err error, observedLogs *observer.ObservedLogs) {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, 1, observedLogs.FilterMessageSnippet(fmt.Sprintf(\n\t\t\t\t\t\"Decision Updated: {Schedule: %v, Started: %v, ID: %v, Timeout: %v, Attempt: %v, Timestamp: %v}\",\n\t\t\t\t\t0, commonconstants.EmptyEventID, commonconstants.EmptyUUID, 0, 1, 0)).Len())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - decision need no update\",\n\t\t\tnewMsb: func(t *testing.T) *mutableStateBuilder {\n\t\t\t\treturn &mutableStateBuilder{\n\t\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDecisionScheduleID: 0, // pending decisions\n\t\t\t\t\t\tDecisionAttempt:    0,\n\t\t\t\t\t},\n\t\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\t\ttimeSource:    clock.NewMockedTimeSource(),\n\t\t\t\t}\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, err error, observedLogs *observer.ObservedLogs) {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tm := &mutableStateDecisionTaskManagerImpl{msb: test.newMsb(t)}\n\t\t\tcore, observedLogs := observer.New(zap.DebugLevel)\n\t\t\tm.msb.logger = log.NewLogger(zap.New(core))\n\t\t\tif test.expectations != nil {\n\t\t\t\ttest.expectations(m)\n\t\t\t}\n\t\t\terr := m.ReplicateTransientDecisionTaskScheduled()\n\t\t\ttest.assertions(t, err, observedLogs)\n\t\t})\n\t}\n}\n\nfunc TestCreateTransientDecisionEvents(t *testing.T) {\n\tm := &mutableStateDecisionTaskManagerImpl{\n\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{TaskList: \"some-task-list\"}},\n\t}\n\tdecision := &DecisionInfo{\n\t\tScheduleID:                 0,\n\t\tStartedID:                  1,\n\t\tRequestID:                  \"some-requestID\",\n\t\tDecisionTimeout:            100,\n\t\tAttempt:                    1,\n\t\tScheduledTimestamp:         10,\n\t\tStartedTimestamp:           10,\n\t\tOriginalScheduledTimestamp: 10,\n\t}\n\tscheduledEvent, startedEvent := m.CreateTransientDecisionEvents(decision, \"identity\")\n\trequire.NotNil(t, scheduledEvent)\n\tassert.Equal(t, decision.ScheduleID, scheduledEvent.ID)\n\tassert.Equal(t, &decision.ScheduledTimestamp, scheduledEvent.Timestamp)\n\tassert.Equal(t, m.msb.executionInfo.TaskList, scheduledEvent.DecisionTaskScheduledEventAttributes.TaskList.Name)\n\tassert.Equal(t, &decision.DecisionTimeout, scheduledEvent.DecisionTaskScheduledEventAttributes.StartToCloseTimeoutSeconds)\n\tassert.Equal(t, decision.Attempt, scheduledEvent.DecisionTaskScheduledEventAttributes.Attempt)\n\n\trequire.NotNil(t, startedEvent)\n\tassert.Equal(t, decision.StartedID, startedEvent.ID)\n\tassert.Equal(t, &decision.StartedTimestamp, startedEvent.Timestamp)\n\tassert.Equal(t, decision.ScheduleID, startedEvent.DecisionTaskStartedEventAttributes.ScheduledEventID)\n\tassert.Equal(t, decision.RequestID, startedEvent.DecisionTaskStartedEventAttributes.RequestID)\n\tassert.Equal(t, \"identity\", startedEvent.DecisionTaskStartedEventAttributes.Identity)\n}\n\nfunc TestGetDecisionScheduleToStartTimeout(t *testing.T) {\n\tt.Run(\"sticky taskList\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tStickyTaskList:               \"some-sticky-task-list\",\n\t\t\t\tStickyScheduleToStartTimeout: 100,\n\t\t\t}},\n\t\t}\n\t\tduration := m.GetDecisionScheduleToStartTimeout()\n\t\tassert.Equal(t, time.Duration(m.msb.executionInfo.StickyScheduleToStartTimeout)*time.Second, duration)\n\t})\n}\n\nfunc TestHasProcessedOrPendingDecision(t *testing.T) {\n\tt.Run(\"true\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDecisionScheduleID: 0, // has pending decisions\n\t\t\t}},\n\t\t}\n\t\tok := m.HasProcessedOrPendingDecision()\n\t\tassert.True(t, ok)\n\t})\n\n\tt.Run(\"false\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t// has no pending decisions\n\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\tLastProcessedEvent: commonconstants.EmptyEventID,\n\t\t\t}},\n\t\t}\n\t\tok := m.HasProcessedOrPendingDecision()\n\t\tassert.False(t, ok)\n\t})\n}\n\nfunc TestGetInFlightDecision(t *testing.T) {\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDecisionScheduleID: 0,\n\t\t\t\tDecisionStartedID:  1,\n\t\t\t\tStickyTaskList:     \"some-sticky-task-list\",\n\t\t\t}},\n\t\t}\n\t\tdecision, ok := m.GetInFlightDecision()\n\t\trequire.NotNil(t, decision)\n\t\tassert.True(t, ok)\n\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionVersion, decision.Version)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionScheduleID, decision.ScheduleID)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionStartedID, decision.StartedID)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionRequestID, decision.RequestID)\n\t\tassert.Equal(t, int64(m.msb.executionInfo.Attempt), decision.Attempt)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionStartedTimestamp, decision.StartedTimestamp)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionScheduledTimestamp, decision.ScheduledTimestamp)\n\t\tassert.Equal(t, m.msb.executionInfo.StickyTaskList, decision.TaskList)\n\t})\n\n\tt.Run(\"failure\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\tDecisionStartedID:  commonconstants.EmptyEventID,\n\t\t\t}},\n\t\t}\n\t\tdecision, value := m.GetInFlightDecision()\n\t\trequire.Nil(t, decision)\n\t\tassert.False(t, value)\n\t})\n}\n\nfunc TestGetPendingDecision(t *testing.T) {\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDecisionScheduleID: 0,\n\t\t\t\tDecisionStartedID:  1,\n\t\t\t\tStickyTaskList:     \"some-sticky-task-list\",\n\t\t\t}},\n\t\t}\n\t\tdecision, ok := m.GetPendingDecision()\n\t\trequire.NotNil(t, decision)\n\t\tassert.True(t, ok)\n\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionVersion, decision.Version)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionScheduleID, decision.ScheduleID)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionStartedID, decision.StartedID)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionRequestID, decision.RequestID)\n\t\tassert.Equal(t, int64(m.msb.executionInfo.Attempt), decision.Attempt)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionStartedTimestamp, decision.StartedTimestamp)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionScheduledTimestamp, decision.ScheduledTimestamp)\n\t\tassert.Equal(t, m.msb.executionInfo.StickyTaskList, decision.TaskList)\n\t})\n\n\tt.Run(\"failure\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{executionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t}},\n\t\t}\n\t\tdecision, value := m.GetPendingDecision()\n\t\trequire.Nil(t, decision)\n\t\tassert.False(t, value)\n\t})\n}\n\nfunc TestReplicateDecisionTaskStartedEvent(t *testing.T) {\n\tvar version int64 = 123\n\tvar scheduleID int64 = 1\n\tvar startedID int64 = 2\n\trequestID := \"some-request-id\"\n\tvar timeStamp int64 = 1\n\tvar originalTimeStamp int64 = 1\n\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tcore, observedLogs := observer.New(zap.DebugLevel)\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID:                 scheduleID,\n\t\t\t\t\tDecisionVersion:                    version,\n\t\t\t\t\tDecisionStartedID:                  startedID,\n\t\t\t\t\tDecisionRequestID:                  requestID,\n\t\t\t\t\tDecisionStartedTimestamp:           timeStamp,\n\t\t\t\t\tDecisionOriginalScheduledTimestamp: originalTimeStamp,\n\t\t\t\t\tTaskList:                           \"some-taskList\",\n\t\t\t\t},\n\t\t\t\ttaskGenerator: NewMockMutableStateTaskGenerator(gomock.NewController(t)),\n\t\t\t\ttimeSource:    clock.NewMockedTimeSource(),\n\t\t\t\tlogger:        log.NewLogger(zap.New(core)),\n\t\t\t},\n\t\t}\n\t\tvar decision *DecisionInfo\n\t\tm.msb.taskGenerator.(*MockMutableStateTaskGenerator).EXPECT().GenerateDecisionStartTasks(scheduleID)\n\t\tresult, err := m.ReplicateDecisionTaskStartedEvent(decision, version, scheduleID, startedID, requestID, timeStamp)\n\t\trequire.NoError(t, err)\n\t\trequire.NotNil(t, result)\n\t\tassert.Equal(t, 1, observedLogs.FilterMessageSnippet(fmt.Sprintf(\n\t\t\t\"Decision Updated: {Schedule: %v, Started: %v, ID: %v, Timeout: %v, Attempt: %v, Timestamp: %v}\",\n\t\t\tscheduleID, startedID, requestID, 0, m.msb.executionInfo.Attempt, timeStamp)).Len())\n\t\tassert.Equal(t, version, result.Version)\n\t\tassert.Equal(t, scheduleID, result.ScheduleID)\n\t\tassert.Equal(t, startedID, result.StartedID)\n\t\tassert.Equal(t, int64(m.msb.executionInfo.Attempt), result.Attempt)\n\t\tassert.Equal(t, requestID, result.RequestID)\n\t\tassert.Equal(t, m.msb.executionInfo.TaskList, result.TaskList)\n\t\tassert.Equal(t, timeStamp, result.StartedTimestamp)\n\t\tassert.Equal(t, m.msb.executionInfo.DecisionOriginalScheduledTimestamp, result.OriginalScheduledTimestamp)\n\t})\n\n\tt.Run(\"failure\", func(t *testing.T) {\n\t\tm := &mutableStateDecisionTaskManagerImpl{\n\t\t\tmsb: &mutableStateBuilder{\n\t\t\t\texecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDecisionScheduleID: commonconstants.EmptyEventID,\n\t\t\t\t\tDecisionAttempt:    1,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tvar decision *DecisionInfo\n\t\tresult, err := m.ReplicateDecisionTaskStartedEvent(decision, version, scheduleID, startedID, requestID, timeStamp)\n\t\tassert.ErrorContains(t, err, fmt.Sprintf(\"unable to find decision: %v\", scheduleID))\n\t\trequire.Nil(t, result)\n\t})\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: mutable_state.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source mutable_state.go -destination mutable_state_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tcache \"github.com/uber/cadence/common/cache\"\n\tdefinition \"github.com/uber/cadence/common/definition\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n\tquery \"github.com/uber/cadence/service/history/query\"\n)\n\n// MockMutableState is a mock of MutableState interface.\ntype MockMutableState struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMutableStateMockRecorder\n\tisgomock struct{}\n}\n\n// MockMutableStateMockRecorder is the mock recorder for MockMutableState.\ntype MockMutableStateMockRecorder struct {\n\tmock *MockMutableState\n}\n\n// NewMockMutableState creates a new mock instance.\nfunc NewMockMutableState(ctrl *gomock.Controller) *MockMutableState {\n\tmock := &MockMutableState{ctrl: ctrl}\n\tmock.recorder = &MockMutableStateMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMutableState) EXPECT() *MockMutableStateMockRecorder {\n\treturn m.recorder\n}\n\n// AddActivityTaskCancelRequestedEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskCancelRequestedEvent(arg0 int64, arg1, arg2 string) (*types.HistoryEvent, *persistence.ActivityInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskCancelRequestedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.ActivityInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddActivityTaskCancelRequestedEvent indicates an expected call of AddActivityTaskCancelRequestedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskCancelRequestedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskCancelRequestedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskCancelRequestedEvent), arg0, arg1, arg2)\n}\n\n// AddActivityTaskCanceledEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskCanceledEvent(arg0, arg1, arg2 int64, arg3 []uint8, arg4 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskCanceledEvent\", arg0, arg1, arg2, arg3, arg4)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTaskCanceledEvent indicates an expected call of AddActivityTaskCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskCanceledEvent(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskCanceledEvent), arg0, arg1, arg2, arg3, arg4)\n}\n\n// AddActivityTaskCompletedEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskCompletedEvent(arg0, arg1 int64, arg2 *types.RespondActivityTaskCompletedRequest) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskCompletedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTaskCompletedEvent indicates an expected call of AddActivityTaskCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskCompletedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskCompletedEvent), arg0, arg1, arg2)\n}\n\n// AddActivityTaskFailedEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskFailedEvent(arg0, arg1 int64, arg2 *types.RespondActivityTaskFailedRequest) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskFailedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTaskFailedEvent indicates an expected call of AddActivityTaskFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskFailedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskFailedEvent), arg0, arg1, arg2)\n}\n\n// AddActivityTaskScheduledEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskScheduledEvent(arg0 context.Context, arg1 int64, arg2 *types.ScheduleActivityTaskDecisionAttributes, arg3 bool) (*types.HistoryEvent, *persistence.ActivityInfo, *types.ActivityLocalDispatchInfo, bool, bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskScheduledEvent\", arg0, arg1, arg2, arg3)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.ActivityInfo)\n\tret2, _ := ret[2].(*types.ActivityLocalDispatchInfo)\n\tret3, _ := ret[3].(bool)\n\tret4, _ := ret[4].(bool)\n\tret5, _ := ret[5].(error)\n\treturn ret0, ret1, ret2, ret3, ret4, ret5\n}\n\n// AddActivityTaskScheduledEvent indicates an expected call of AddActivityTaskScheduledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskScheduledEvent(arg0, arg1, arg2, arg3 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskScheduledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskScheduledEvent), arg0, arg1, arg2, arg3)\n}\n\n// AddActivityTaskStartedEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskStartedEvent(arg0 *persistence.ActivityInfo, arg1 int64, arg2, arg3 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskStartedEvent\", arg0, arg1, arg2, arg3)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTaskStartedEvent indicates an expected call of AddActivityTaskStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskStartedEvent(arg0, arg1, arg2, arg3 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskStartedEvent), arg0, arg1, arg2, arg3)\n}\n\n// AddActivityTaskTimedOutEvent mocks base method.\nfunc (m *MockMutableState) AddActivityTaskTimedOutEvent(arg0, arg1 int64, arg2 types.TimeoutType, arg3 []uint8) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTaskTimedOutEvent\", arg0, arg1, arg2, arg3)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTaskTimedOutEvent indicates an expected call of AddActivityTaskTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) AddActivityTaskTimedOutEvent(arg0, arg1, arg2, arg3 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTaskTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).AddActivityTaskTimedOutEvent), arg0, arg1, arg2, arg3)\n}\n\n// AddCancelTimerFailedEvent mocks base method.\nfunc (m *MockMutableState) AddCancelTimerFailedEvent(arg0 int64, arg1 *types.CancelTimerDecisionAttributes, arg2 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddCancelTimerFailedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddCancelTimerFailedEvent indicates an expected call of AddCancelTimerFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddCancelTimerFailedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddCancelTimerFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddCancelTimerFailedEvent), arg0, arg1, arg2)\n}\n\n// AddChildWorkflowExecutionCanceledEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionCanceledEvent(arg0 int64, arg1 *types.WorkflowExecution, arg2 *types.WorkflowExecutionCanceledEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionCanceledEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionCanceledEvent indicates an expected call of AddChildWorkflowExecutionCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionCanceledEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionCanceledEvent), arg0, arg1, arg2)\n}\n\n// AddChildWorkflowExecutionCompletedEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionCompletedEvent(arg0 int64, arg1 *types.WorkflowExecution, arg2 *types.WorkflowExecutionCompletedEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionCompletedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionCompletedEvent indicates an expected call of AddChildWorkflowExecutionCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionCompletedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionCompletedEvent), arg0, arg1, arg2)\n}\n\n// AddChildWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionFailedEvent(arg0 int64, arg1 *types.WorkflowExecution, arg2 *types.WorkflowExecutionFailedEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionFailedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionFailedEvent indicates an expected call of AddChildWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionFailedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionFailedEvent), arg0, arg1, arg2)\n}\n\n// AddChildWorkflowExecutionStartedEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionStartedEvent(arg0 string, arg1 *types.WorkflowExecution, arg2 *types.WorkflowType, arg3 int64, arg4 *types.Header) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionStartedEvent\", arg0, arg1, arg2, arg3, arg4)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionStartedEvent indicates an expected call of AddChildWorkflowExecutionStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionStartedEvent(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionStartedEvent), arg0, arg1, arg2, arg3, arg4)\n}\n\n// AddChildWorkflowExecutionTerminatedEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionTerminatedEvent(arg0 int64, arg1 *types.WorkflowExecution, arg2 *types.WorkflowExecutionTerminatedEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionTerminatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionTerminatedEvent indicates an expected call of AddChildWorkflowExecutionTerminatedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionTerminatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionTerminatedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionTerminatedEvent), arg0, arg1, arg2)\n}\n\n// AddChildWorkflowExecutionTimedOutEvent mocks base method.\nfunc (m *MockMutableState) AddChildWorkflowExecutionTimedOutEvent(arg0 int64, arg1 *types.WorkflowExecution, arg2 *types.WorkflowExecutionTimedOutEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddChildWorkflowExecutionTimedOutEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddChildWorkflowExecutionTimedOutEvent indicates an expected call of AddChildWorkflowExecutionTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) AddChildWorkflowExecutionTimedOutEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddChildWorkflowExecutionTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).AddChildWorkflowExecutionTimedOutEvent), arg0, arg1, arg2)\n}\n\n// AddCompletedWorkflowEvent mocks base method.\nfunc (m *MockMutableState) AddCompletedWorkflowEvent(arg0 int64, arg1 *types.CompleteWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddCompletedWorkflowEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddCompletedWorkflowEvent indicates an expected call of AddCompletedWorkflowEvent.\nfunc (mr *MockMutableStateMockRecorder) AddCompletedWorkflowEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddCompletedWorkflowEvent\", reflect.TypeOf((*MockMutableState)(nil).AddCompletedWorkflowEvent), arg0, arg1)\n}\n\n// AddContinueAsNewEvent mocks base method.\nfunc (m *MockMutableState) AddContinueAsNewEvent(arg0 context.Context, arg1, arg2 int64, arg3 string, arg4 *types.ContinueAsNewWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, MutableState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddContinueAsNewEvent\", arg0, arg1, arg2, arg3, arg4)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(MutableState)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddContinueAsNewEvent indicates an expected call of AddContinueAsNewEvent.\nfunc (mr *MockMutableStateMockRecorder) AddContinueAsNewEvent(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddContinueAsNewEvent\", reflect.TypeOf((*MockMutableState)(nil).AddContinueAsNewEvent), arg0, arg1, arg2, arg3, arg4)\n}\n\n// AddDecisionTaskCompletedEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskCompletedEvent(arg0, arg1 int64, arg2 *types.RespondDecisionTaskCompletedRequest, arg3 int) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskCompletedEvent\", arg0, arg1, arg2, arg3)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskCompletedEvent indicates an expected call of AddDecisionTaskCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskCompletedEvent(arg0, arg1, arg2, arg3 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskCompletedEvent), arg0, arg1, arg2, arg3)\n}\n\n// AddDecisionTaskFailedEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskFailedEvent(scheduleEventID, startedEventID int64, cause types.DecisionTaskFailedCause, details []byte, identity, reason, binChecksum, baseRunID, newRunID string, forkEventVersion int64, resetRequestID string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskFailedEvent\", scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskFailedEvent indicates an expected call of AddDecisionTaskFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskFailedEvent(scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskFailedEvent), scheduleEventID, startedEventID, cause, details, identity, reason, binChecksum, baseRunID, newRunID, forkEventVersion, resetRequestID)\n}\n\n// AddDecisionTaskResetTimeoutEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskResetTimeoutEvent(arg0 int64, arg1, arg2 string, arg3 int64, arg4, arg5 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskResetTimeoutEvent\", arg0, arg1, arg2, arg3, arg4, arg5)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskResetTimeoutEvent indicates an expected call of AddDecisionTaskResetTimeoutEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskResetTimeoutEvent(arg0, arg1, arg2, arg3, arg4, arg5 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskResetTimeoutEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskResetTimeoutEvent), arg0, arg1, arg2, arg3, arg4, arg5)\n}\n\n// AddDecisionTaskScheduleToStartTimeoutEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskScheduleToStartTimeoutEvent(arg0 int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduleToStartTimeoutEvent\", arg0)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduleToStartTimeoutEvent indicates an expected call of AddDecisionTaskScheduleToStartTimeoutEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskScheduleToStartTimeoutEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduleToStartTimeoutEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskScheduleToStartTimeoutEvent), arg0)\n}\n\n// AddDecisionTaskScheduledEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskScheduledEvent(bypassTaskGeneration bool) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduledEvent\", bypassTaskGeneration)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduledEvent indicates an expected call of AddDecisionTaskScheduledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskScheduledEvent(bypassTaskGeneration any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskScheduledEvent), bypassTaskGeneration)\n}\n\n// AddDecisionTaskScheduledEventAsHeartbeat mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration bool, originalScheduledTimestamp int64) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskScheduledEventAsHeartbeat\", bypassTaskGeneration, originalScheduledTimestamp)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskScheduledEventAsHeartbeat indicates an expected call of AddDecisionTaskScheduledEventAsHeartbeat.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskScheduledEventAsHeartbeat(bypassTaskGeneration, originalScheduledTimestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskScheduledEventAsHeartbeat\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskScheduledEventAsHeartbeat), bypassTaskGeneration, originalScheduledTimestamp)\n}\n\n// AddDecisionTaskStartedEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskStartedEvent(arg0 int64, arg1 string, arg2 *types.PollForDecisionTaskRequest) (*types.HistoryEvent, *DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskStartedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*DecisionInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddDecisionTaskStartedEvent indicates an expected call of AddDecisionTaskStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskStartedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskStartedEvent), arg0, arg1, arg2)\n}\n\n// AddDecisionTaskTimedOutEvent mocks base method.\nfunc (m *MockMutableState) AddDecisionTaskTimedOutEvent(arg0, arg1 int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTaskTimedOutEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTaskTimedOutEvent indicates an expected call of AddDecisionTaskTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) AddDecisionTaskTimedOutEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTaskTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).AddDecisionTaskTimedOutEvent), arg0, arg1)\n}\n\n// AddExternalWorkflowExecutionCancelRequested mocks base method.\nfunc (m *MockMutableState) AddExternalWorkflowExecutionCancelRequested(arg0 int64, arg1, arg2, arg3 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddExternalWorkflowExecutionCancelRequested\", arg0, arg1, arg2, arg3)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddExternalWorkflowExecutionCancelRequested indicates an expected call of AddExternalWorkflowExecutionCancelRequested.\nfunc (mr *MockMutableStateMockRecorder) AddExternalWorkflowExecutionCancelRequested(arg0, arg1, arg2, arg3 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddExternalWorkflowExecutionCancelRequested\", reflect.TypeOf((*MockMutableState)(nil).AddExternalWorkflowExecutionCancelRequested), arg0, arg1, arg2, arg3)\n}\n\n// AddExternalWorkflowExecutionSignaled mocks base method.\nfunc (m *MockMutableState) AddExternalWorkflowExecutionSignaled(arg0 int64, arg1, arg2, arg3 string, arg4 []uint8) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddExternalWorkflowExecutionSignaled\", arg0, arg1, arg2, arg3, arg4)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddExternalWorkflowExecutionSignaled indicates an expected call of AddExternalWorkflowExecutionSignaled.\nfunc (mr *MockMutableStateMockRecorder) AddExternalWorkflowExecutionSignaled(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddExternalWorkflowExecutionSignaled\", reflect.TypeOf((*MockMutableState)(nil).AddExternalWorkflowExecutionSignaled), arg0, arg1, arg2, arg3, arg4)\n}\n\n// AddFailWorkflowEvent mocks base method.\nfunc (m *MockMutableState) AddFailWorkflowEvent(arg0 int64, arg1 *types.FailWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddFailWorkflowEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddFailWorkflowEvent indicates an expected call of AddFailWorkflowEvent.\nfunc (mr *MockMutableStateMockRecorder) AddFailWorkflowEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddFailWorkflowEvent\", reflect.TypeOf((*MockMutableState)(nil).AddFailWorkflowEvent), arg0, arg1)\n}\n\n// AddFirstDecisionTaskScheduled mocks base method.\nfunc (m *MockMutableState) AddFirstDecisionTaskScheduled(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddFirstDecisionTaskScheduled\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AddFirstDecisionTaskScheduled indicates an expected call of AddFirstDecisionTaskScheduled.\nfunc (mr *MockMutableStateMockRecorder) AddFirstDecisionTaskScheduled(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddFirstDecisionTaskScheduled\", reflect.TypeOf((*MockMutableState)(nil).AddFirstDecisionTaskScheduled), arg0)\n}\n\n// AddRecordMarkerEvent mocks base method.\nfunc (m *MockMutableState) AddRecordMarkerEvent(arg0 int64, arg1 *types.RecordMarkerDecisionAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddRecordMarkerEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddRecordMarkerEvent indicates an expected call of AddRecordMarkerEvent.\nfunc (mr *MockMutableStateMockRecorder) AddRecordMarkerEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddRecordMarkerEvent\", reflect.TypeOf((*MockMutableState)(nil).AddRecordMarkerEvent), arg0, arg1)\n}\n\n// AddRequestCancelActivityTaskFailedEvent mocks base method.\nfunc (m *MockMutableState) AddRequestCancelActivityTaskFailedEvent(arg0 int64, arg1, arg2 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddRequestCancelActivityTaskFailedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddRequestCancelActivityTaskFailedEvent indicates an expected call of AddRequestCancelActivityTaskFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddRequestCancelActivityTaskFailedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddRequestCancelActivityTaskFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddRequestCancelActivityTaskFailedEvent), arg0, arg1, arg2)\n}\n\n// AddRequestCancelExternalWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) AddRequestCancelExternalWorkflowExecutionFailedEvent(arg0, arg1 int64, arg2, arg3, arg4 string, arg5 types.CancelExternalWorkflowExecutionFailedCause) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddRequestCancelExternalWorkflowExecutionFailedEvent\", arg0, arg1, arg2, arg3, arg4, arg5)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddRequestCancelExternalWorkflowExecutionFailedEvent indicates an expected call of AddRequestCancelExternalWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddRequestCancelExternalWorkflowExecutionFailedEvent(arg0, arg1, arg2, arg3, arg4, arg5 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddRequestCancelExternalWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddRequestCancelExternalWorkflowExecutionFailedEvent), arg0, arg1, arg2, arg3, arg4, arg5)\n}\n\n// AddRequestCancelExternalWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) AddRequestCancelExternalWorkflowExecutionInitiatedEvent(arg0 int64, arg1 string, arg2 *types.RequestCancelExternalWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.RequestCancelInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddRequestCancelExternalWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.RequestCancelInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddRequestCancelExternalWorkflowExecutionInitiatedEvent indicates an expected call of AddRequestCancelExternalWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddRequestCancelExternalWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddRequestCancelExternalWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddRequestCancelExternalWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// AddSignalExternalWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) AddSignalExternalWorkflowExecutionFailedEvent(arg0, arg1 int64, arg2, arg3, arg4 string, arg5 []uint8, arg6 types.SignalExternalWorkflowExecutionFailedCause) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddSignalExternalWorkflowExecutionFailedEvent\", arg0, arg1, arg2, arg3, arg4, arg5, arg6)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddSignalExternalWorkflowExecutionFailedEvent indicates an expected call of AddSignalExternalWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddSignalExternalWorkflowExecutionFailedEvent(arg0, arg1, arg2, arg3, arg4, arg5, arg6 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddSignalExternalWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddSignalExternalWorkflowExecutionFailedEvent), arg0, arg1, arg2, arg3, arg4, arg5, arg6)\n}\n\n// AddSignalExternalWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) AddSignalExternalWorkflowExecutionInitiatedEvent(arg0 int64, arg1 string, arg2 *types.SignalExternalWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.SignalInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddSignalExternalWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.SignalInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddSignalExternalWorkflowExecutionInitiatedEvent indicates an expected call of AddSignalExternalWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddSignalExternalWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddSignalExternalWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddSignalExternalWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// AddSignalRequested mocks base method.\nfunc (m *MockMutableState) AddSignalRequested(requestID string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddSignalRequested\", requestID)\n}\n\n// AddSignalRequested indicates an expected call of AddSignalRequested.\nfunc (mr *MockMutableStateMockRecorder) AddSignalRequested(requestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddSignalRequested\", reflect.TypeOf((*MockMutableState)(nil).AddSignalRequested), requestID)\n}\n\n// AddStartChildWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) AddStartChildWorkflowExecutionFailedEvent(arg0 int64, arg1 types.ChildWorkflowExecutionFailedCause, arg2 *types.StartChildWorkflowExecutionInitiatedEventAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddStartChildWorkflowExecutionFailedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddStartChildWorkflowExecutionFailedEvent indicates an expected call of AddStartChildWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddStartChildWorkflowExecutionFailedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddStartChildWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddStartChildWorkflowExecutionFailedEvent), arg0, arg1, arg2)\n}\n\n// AddStartChildWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) AddStartChildWorkflowExecutionInitiatedEvent(arg0 int64, arg1 string, arg2 *types.StartChildWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, *persistence.ChildExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddStartChildWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.ChildExecutionInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddStartChildWorkflowExecutionInitiatedEvent indicates an expected call of AddStartChildWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddStartChildWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddStartChildWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddStartChildWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// AddTimeoutWorkflowEvent mocks base method.\nfunc (m *MockMutableState) AddTimeoutWorkflowEvent(arg0 int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddTimeoutWorkflowEvent\", arg0)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddTimeoutWorkflowEvent indicates an expected call of AddTimeoutWorkflowEvent.\nfunc (mr *MockMutableStateMockRecorder) AddTimeoutWorkflowEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTimeoutWorkflowEvent\", reflect.TypeOf((*MockMutableState)(nil).AddTimeoutWorkflowEvent), arg0)\n}\n\n// AddTimerCanceledEvent mocks base method.\nfunc (m *MockMutableState) AddTimerCanceledEvent(arg0 int64, arg1 *types.CancelTimerDecisionAttributes, arg2 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddTimerCanceledEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddTimerCanceledEvent indicates an expected call of AddTimerCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddTimerCanceledEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTimerCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddTimerCanceledEvent), arg0, arg1, arg2)\n}\n\n// AddTimerFiredEvent mocks base method.\nfunc (m *MockMutableState) AddTimerFiredEvent(arg0 string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddTimerFiredEvent\", arg0)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddTimerFiredEvent indicates an expected call of AddTimerFiredEvent.\nfunc (mr *MockMutableStateMockRecorder) AddTimerFiredEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTimerFiredEvent\", reflect.TypeOf((*MockMutableState)(nil).AddTimerFiredEvent), arg0)\n}\n\n// AddTimerStartedEvent mocks base method.\nfunc (m *MockMutableState) AddTimerStartedEvent(arg0 int64, arg1 *types.StartTimerDecisionAttributes) (*types.HistoryEvent, *persistence.TimerInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddTimerStartedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*persistence.TimerInfo)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// AddTimerStartedEvent indicates an expected call of AddTimerStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddTimerStartedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTimerStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddTimerStartedEvent), arg0, arg1)\n}\n\n// AddTimerTasks mocks base method.\nfunc (m *MockMutableState) AddTimerTasks(timerTasks ...persistence.Task) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range timerTasks {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"AddTimerTasks\", varargs...)\n}\n\n// AddTimerTasks indicates an expected call of AddTimerTasks.\nfunc (mr *MockMutableStateMockRecorder) AddTimerTasks(timerTasks ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTimerTasks\", reflect.TypeOf((*MockMutableState)(nil).AddTimerTasks), timerTasks...)\n}\n\n// AddTransferTasks mocks base method.\nfunc (m *MockMutableState) AddTransferTasks(transferTasks ...persistence.Task) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range transferTasks {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"AddTransferTasks\", varargs...)\n}\n\n// AddTransferTasks indicates an expected call of AddTransferTasks.\nfunc (mr *MockMutableStateMockRecorder) AddTransferTasks(transferTasks ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTransferTasks\", reflect.TypeOf((*MockMutableState)(nil).AddTransferTasks), transferTasks...)\n}\n\n// AddUpsertWorkflowSearchAttributesEvent mocks base method.\nfunc (m *MockMutableState) AddUpsertWorkflowSearchAttributesEvent(arg0 int64, arg1 *types.UpsertWorkflowSearchAttributesDecisionAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddUpsertWorkflowSearchAttributesEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddUpsertWorkflowSearchAttributesEvent indicates an expected call of AddUpsertWorkflowSearchAttributesEvent.\nfunc (mr *MockMutableStateMockRecorder) AddUpsertWorkflowSearchAttributesEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddUpsertWorkflowSearchAttributesEvent\", reflect.TypeOf((*MockMutableState)(nil).AddUpsertWorkflowSearchAttributesEvent), arg0, arg1)\n}\n\n// AddWorkflowExecutionCancelRequestedEvent mocks base method.\nfunc (m *MockMutableState) AddWorkflowExecutionCancelRequestedEvent(arg0 string, arg1 *types.HistoryRequestCancelWorkflowExecutionRequest) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddWorkflowExecutionCancelRequestedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddWorkflowExecutionCancelRequestedEvent indicates an expected call of AddWorkflowExecutionCancelRequestedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddWorkflowExecutionCancelRequestedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddWorkflowExecutionCancelRequestedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddWorkflowExecutionCancelRequestedEvent), arg0, arg1)\n}\n\n// AddWorkflowExecutionCanceledEvent mocks base method.\nfunc (m *MockMutableState) AddWorkflowExecutionCanceledEvent(arg0 int64, arg1 *types.CancelWorkflowExecutionDecisionAttributes) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddWorkflowExecutionCanceledEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddWorkflowExecutionCanceledEvent indicates an expected call of AddWorkflowExecutionCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) AddWorkflowExecutionCanceledEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddWorkflowExecutionCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).AddWorkflowExecutionCanceledEvent), arg0, arg1)\n}\n\n// AddWorkflowExecutionSignaled mocks base method.\nfunc (m *MockMutableState) AddWorkflowExecutionSignaled(signalName string, input []byte, identity, reqeustID string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddWorkflowExecutionSignaled\", signalName, input, identity, reqeustID)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddWorkflowExecutionSignaled indicates an expected call of AddWorkflowExecutionSignaled.\nfunc (mr *MockMutableStateMockRecorder) AddWorkflowExecutionSignaled(signalName, input, identity, reqeustID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddWorkflowExecutionSignaled\", reflect.TypeOf((*MockMutableState)(nil).AddWorkflowExecutionSignaled), signalName, input, identity, reqeustID)\n}\n\n// AddWorkflowExecutionStartedEvent mocks base method.\nfunc (m *MockMutableState) AddWorkflowExecutionStartedEvent(arg0 types.WorkflowExecution, arg1 *types.HistoryStartWorkflowExecutionRequest) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddWorkflowExecutionStartedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddWorkflowExecutionStartedEvent indicates an expected call of AddWorkflowExecutionStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddWorkflowExecutionStartedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddWorkflowExecutionStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddWorkflowExecutionStartedEvent), arg0, arg1)\n}\n\n// AddWorkflowExecutionTerminatedEvent mocks base method.\nfunc (m *MockMutableState) AddWorkflowExecutionTerminatedEvent(firstEventID int64, reason string, details []byte, identity string) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddWorkflowExecutionTerminatedEvent\", firstEventID, reason, details, identity)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddWorkflowExecutionTerminatedEvent indicates an expected call of AddWorkflowExecutionTerminatedEvent.\nfunc (mr *MockMutableStateMockRecorder) AddWorkflowExecutionTerminatedEvent(firstEventID, reason, details, identity any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddWorkflowExecutionTerminatedEvent\", reflect.TypeOf((*MockMutableState)(nil).AddWorkflowExecutionTerminatedEvent), firstEventID, reason, details, identity)\n}\n\n// ByteSize mocks base method.\nfunc (m *MockMutableState) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockMutableStateMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockMutableState)(nil).ByteSize))\n}\n\n// CheckResettable mocks base method.\nfunc (m *MockMutableState) CheckResettable() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CheckResettable\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CheckResettable indicates an expected call of CheckResettable.\nfunc (mr *MockMutableStateMockRecorder) CheckResettable() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CheckResettable\", reflect.TypeOf((*MockMutableState)(nil).CheckResettable))\n}\n\n// ClearStickyness mocks base method.\nfunc (m *MockMutableState) ClearStickyness() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ClearStickyness\")\n}\n\n// ClearStickyness indicates an expected call of ClearStickyness.\nfunc (mr *MockMutableStateMockRecorder) ClearStickyness() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ClearStickyness\", reflect.TypeOf((*MockMutableState)(nil).ClearStickyness))\n}\n\n// CloseTransactionAsMutation mocks base method.\nfunc (m *MockMutableState) CloseTransactionAsMutation(now time.Time, transactionPolicy TransactionPolicy) (*persistence.WorkflowMutation, []*persistence.WorkflowEvents, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CloseTransactionAsMutation\", now, transactionPolicy)\n\tret0, _ := ret[0].(*persistence.WorkflowMutation)\n\tret1, _ := ret[1].([]*persistence.WorkflowEvents)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// CloseTransactionAsMutation indicates an expected call of CloseTransactionAsMutation.\nfunc (mr *MockMutableStateMockRecorder) CloseTransactionAsMutation(now, transactionPolicy any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseTransactionAsMutation\", reflect.TypeOf((*MockMutableState)(nil).CloseTransactionAsMutation), now, transactionPolicy)\n}\n\n// CloseTransactionAsSnapshot mocks base method.\nfunc (m *MockMutableState) CloseTransactionAsSnapshot(now time.Time, transactionPolicy TransactionPolicy) (*persistence.WorkflowSnapshot, []*persistence.WorkflowEvents, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CloseTransactionAsSnapshot\", now, transactionPolicy)\n\tret0, _ := ret[0].(*persistence.WorkflowSnapshot)\n\tret1, _ := ret[1].([]*persistence.WorkflowEvents)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// CloseTransactionAsSnapshot indicates an expected call of CloseTransactionAsSnapshot.\nfunc (mr *MockMutableStateMockRecorder) CloseTransactionAsSnapshot(now, transactionPolicy any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseTransactionAsSnapshot\", reflect.TypeOf((*MockMutableState)(nil).CloseTransactionAsSnapshot), now, transactionPolicy)\n}\n\n// CopyToPersistence mocks base method.\nfunc (m *MockMutableState) CopyToPersistence() *persistence.WorkflowMutableState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CopyToPersistence\")\n\tret0, _ := ret[0].(*persistence.WorkflowMutableState)\n\treturn ret0\n}\n\n// CopyToPersistence indicates an expected call of CopyToPersistence.\nfunc (mr *MockMutableStateMockRecorder) CopyToPersistence() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CopyToPersistence\", reflect.TypeOf((*MockMutableState)(nil).CopyToPersistence))\n}\n\n// CreateNewHistoryEvent mocks base method.\nfunc (m *MockMutableState) CreateNewHistoryEvent(eventType types.EventType) *types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateNewHistoryEvent\", eventType)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\treturn ret0\n}\n\n// CreateNewHistoryEvent indicates an expected call of CreateNewHistoryEvent.\nfunc (mr *MockMutableStateMockRecorder) CreateNewHistoryEvent(eventType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateNewHistoryEvent\", reflect.TypeOf((*MockMutableState)(nil).CreateNewHistoryEvent), eventType)\n}\n\n// CreateNewHistoryEventWithTimestamp mocks base method.\nfunc (m *MockMutableState) CreateNewHistoryEventWithTimestamp(eventType types.EventType, timestamp int64) *types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateNewHistoryEventWithTimestamp\", eventType, timestamp)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\treturn ret0\n}\n\n// CreateNewHistoryEventWithTimestamp indicates an expected call of CreateNewHistoryEventWithTimestamp.\nfunc (mr *MockMutableStateMockRecorder) CreateNewHistoryEventWithTimestamp(eventType, timestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateNewHistoryEventWithTimestamp\", reflect.TypeOf((*MockMutableState)(nil).CreateNewHistoryEventWithTimestamp), eventType, timestamp)\n}\n\n// CreateTransientDecisionEvents mocks base method.\nfunc (m *MockMutableState) CreateTransientDecisionEvents(di *DecisionInfo, identity string) (*types.HistoryEvent, *types.HistoryEvent) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateTransientDecisionEvents\", di, identity)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(*types.HistoryEvent)\n\treturn ret0, ret1\n}\n\n// CreateTransientDecisionEvents indicates an expected call of CreateTransientDecisionEvents.\nfunc (mr *MockMutableStateMockRecorder) CreateTransientDecisionEvents(di, identity any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateTransientDecisionEvents\", reflect.TypeOf((*MockMutableState)(nil).CreateTransientDecisionEvents), di, identity)\n}\n\n// DeleteActivity mocks base method.\nfunc (m *MockMutableState) DeleteActivity(scheduleEventID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteActivity\", scheduleEventID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteActivity indicates an expected call of DeleteActivity.\nfunc (mr *MockMutableStateMockRecorder) DeleteActivity(scheduleEventID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteActivity\", reflect.TypeOf((*MockMutableState)(nil).DeleteActivity), scheduleEventID)\n}\n\n// DeleteDecision mocks base method.\nfunc (m *MockMutableState) DeleteDecision() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DeleteDecision\")\n}\n\n// DeleteDecision indicates an expected call of DeleteDecision.\nfunc (mr *MockMutableStateMockRecorder) DeleteDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteDecision\", reflect.TypeOf((*MockMutableState)(nil).DeleteDecision))\n}\n\n// DeleteSignalRequested mocks base method.\nfunc (m *MockMutableState) DeleteSignalRequested(requestID string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DeleteSignalRequested\", requestID)\n}\n\n// DeleteSignalRequested indicates an expected call of DeleteSignalRequested.\nfunc (mr *MockMutableStateMockRecorder) DeleteSignalRequested(requestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteSignalRequested\", reflect.TypeOf((*MockMutableState)(nil).DeleteSignalRequested), requestID)\n}\n\n// DeleteTimerTasks mocks base method.\nfunc (m *MockMutableState) DeleteTimerTasks() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DeleteTimerTasks\")\n}\n\n// DeleteTimerTasks indicates an expected call of DeleteTimerTasks.\nfunc (mr *MockMutableStateMockRecorder) DeleteTimerTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTimerTasks\", reflect.TypeOf((*MockMutableState)(nil).DeleteTimerTasks))\n}\n\n// DeleteTransferTasks mocks base method.\nfunc (m *MockMutableState) DeleteTransferTasks() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DeleteTransferTasks\")\n}\n\n// DeleteTransferTasks indicates an expected call of DeleteTransferTasks.\nfunc (mr *MockMutableStateMockRecorder) DeleteTransferTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteTransferTasks\", reflect.TypeOf((*MockMutableState)(nil).DeleteTransferTasks))\n}\n\n// DeleteUserTimer mocks base method.\nfunc (m *MockMutableState) DeleteUserTimer(timerID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteUserTimer\", timerID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteUserTimer indicates an expected call of DeleteUserTimer.\nfunc (mr *MockMutableStateMockRecorder) DeleteUserTimer(timerID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteUserTimer\", reflect.TypeOf((*MockMutableState)(nil).DeleteUserTimer), timerID)\n}\n\n// FailDecision mocks base method.\nfunc (m *MockMutableState) FailDecision(arg0 bool) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"FailDecision\", arg0)\n}\n\n// FailDecision indicates an expected call of FailDecision.\nfunc (mr *MockMutableStateMockRecorder) FailDecision(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailDecision\", reflect.TypeOf((*MockMutableState)(nil).FailDecision), arg0)\n}\n\n// FlushBufferedEvents mocks base method.\nfunc (m *MockMutableState) FlushBufferedEvents() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FlushBufferedEvents\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// FlushBufferedEvents indicates an expected call of FlushBufferedEvents.\nfunc (mr *MockMutableStateMockRecorder) FlushBufferedEvents() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FlushBufferedEvents\", reflect.TypeOf((*MockMutableState)(nil).FlushBufferedEvents))\n}\n\n// GetActivityByActivityID mocks base method.\nfunc (m *MockMutableState) GetActivityByActivityID(arg0 string) (*persistence.ActivityInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActivityByActivityID\", arg0)\n\tret0, _ := ret[0].(*persistence.ActivityInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetActivityByActivityID indicates an expected call of GetActivityByActivityID.\nfunc (mr *MockMutableStateMockRecorder) GetActivityByActivityID(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActivityByActivityID\", reflect.TypeOf((*MockMutableState)(nil).GetActivityByActivityID), arg0)\n}\n\n// GetActivityInfo mocks base method.\nfunc (m *MockMutableState) GetActivityInfo(arg0 int64) (*persistence.ActivityInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActivityInfo\", arg0)\n\tret0, _ := ret[0].(*persistence.ActivityInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetActivityInfo indicates an expected call of GetActivityInfo.\nfunc (mr *MockMutableStateMockRecorder) GetActivityInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActivityInfo\", reflect.TypeOf((*MockMutableState)(nil).GetActivityInfo), arg0)\n}\n\n// GetActivityScheduledEvent mocks base method.\nfunc (m *MockMutableState) GetActivityScheduledEvent(arg0 context.Context, arg1 int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActivityScheduledEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetActivityScheduledEvent indicates an expected call of GetActivityScheduledEvent.\nfunc (mr *MockMutableStateMockRecorder) GetActivityScheduledEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActivityScheduledEvent\", reflect.TypeOf((*MockMutableState)(nil).GetActivityScheduledEvent), arg0, arg1)\n}\n\n// GetChildExecutionInfo mocks base method.\nfunc (m *MockMutableState) GetChildExecutionInfo(arg0 int64) (*persistence.ChildExecutionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetChildExecutionInfo\", arg0)\n\tret0, _ := ret[0].(*persistence.ChildExecutionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetChildExecutionInfo indicates an expected call of GetChildExecutionInfo.\nfunc (mr *MockMutableStateMockRecorder) GetChildExecutionInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetChildExecutionInfo\", reflect.TypeOf((*MockMutableState)(nil).GetChildExecutionInfo), arg0)\n}\n\n// GetChildExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) GetChildExecutionInitiatedEvent(arg0 context.Context, arg1 int64) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetChildExecutionInitiatedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetChildExecutionInitiatedEvent indicates an expected call of GetChildExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) GetChildExecutionInitiatedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetChildExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).GetChildExecutionInitiatedEvent), arg0, arg1)\n}\n\n// GetCompletionEvent mocks base method.\nfunc (m *MockMutableState) GetCompletionEvent(arg0 context.Context) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCompletionEvent\", arg0)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCompletionEvent indicates an expected call of GetCompletionEvent.\nfunc (mr *MockMutableStateMockRecorder) GetCompletionEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCompletionEvent\", reflect.TypeOf((*MockMutableState)(nil).GetCompletionEvent), arg0)\n}\n\n// GetCronBackoffDuration mocks base method.\nfunc (m *MockMutableState) GetCronBackoffDuration(arg0 context.Context) (time.Duration, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCronBackoffDuration\", arg0)\n\tret0, _ := ret[0].(time.Duration)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCronBackoffDuration indicates an expected call of GetCronBackoffDuration.\nfunc (mr *MockMutableStateMockRecorder) GetCronBackoffDuration(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCronBackoffDuration\", reflect.TypeOf((*MockMutableState)(nil).GetCronBackoffDuration), arg0)\n}\n\n// GetCurrentBranchToken mocks base method.\nfunc (m *MockMutableState) GetCurrentBranchToken() ([]byte, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentBranchToken\")\n\tret0, _ := ret[0].([]byte)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCurrentBranchToken indicates an expected call of GetCurrentBranchToken.\nfunc (mr *MockMutableStateMockRecorder) GetCurrentBranchToken() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentBranchToken\", reflect.TypeOf((*MockMutableState)(nil).GetCurrentBranchToken))\n}\n\n// GetCurrentVersion mocks base method.\nfunc (m *MockMutableState) GetCurrentVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetCurrentVersion indicates an expected call of GetCurrentVersion.\nfunc (mr *MockMutableStateMockRecorder) GetCurrentVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentVersion\", reflect.TypeOf((*MockMutableState)(nil).GetCurrentVersion))\n}\n\n// GetDecisionInfo mocks base method.\nfunc (m *MockMutableState) GetDecisionInfo(arg0 int64) (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDecisionInfo\", arg0)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetDecisionInfo indicates an expected call of GetDecisionInfo.\nfunc (mr *MockMutableStateMockRecorder) GetDecisionInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDecisionInfo\", reflect.TypeOf((*MockMutableState)(nil).GetDecisionInfo), arg0)\n}\n\n// GetDecisionScheduleToStartTimeout mocks base method.\nfunc (m *MockMutableState) GetDecisionScheduleToStartTimeout() time.Duration {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDecisionScheduleToStartTimeout\")\n\tret0, _ := ret[0].(time.Duration)\n\treturn ret0\n}\n\n// GetDecisionScheduleToStartTimeout indicates an expected call of GetDecisionScheduleToStartTimeout.\nfunc (mr *MockMutableStateMockRecorder) GetDecisionScheduleToStartTimeout() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDecisionScheduleToStartTimeout\", reflect.TypeOf((*MockMutableState)(nil).GetDecisionScheduleToStartTimeout))\n}\n\n// GetDomainEntry mocks base method.\nfunc (m *MockMutableState) GetDomainEntry() *cache.DomainCacheEntry {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainEntry\")\n\tret0, _ := ret[0].(*cache.DomainCacheEntry)\n\treturn ret0\n}\n\n// GetDomainEntry indicates an expected call of GetDomainEntry.\nfunc (mr *MockMutableStateMockRecorder) GetDomainEntry() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainEntry\", reflect.TypeOf((*MockMutableState)(nil).GetDomainEntry))\n}\n\n// GetExecutionInfo mocks base method.\nfunc (m *MockMutableState) GetExecutionInfo() *persistence.WorkflowExecutionInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionInfo\")\n\tret0, _ := ret[0].(*persistence.WorkflowExecutionInfo)\n\treturn ret0\n}\n\n// GetExecutionInfo indicates an expected call of GetExecutionInfo.\nfunc (mr *MockMutableStateMockRecorder) GetExecutionInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionInfo\", reflect.TypeOf((*MockMutableState)(nil).GetExecutionInfo))\n}\n\n// GetHistoryBuilder mocks base method.\nfunc (m *MockMutableState) GetHistoryBuilder() *HistoryBuilder {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryBuilder\")\n\tret0, _ := ret[0].(*HistoryBuilder)\n\treturn ret0\n}\n\n// GetHistoryBuilder indicates an expected call of GetHistoryBuilder.\nfunc (mr *MockMutableStateMockRecorder) GetHistoryBuilder() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryBuilder\", reflect.TypeOf((*MockMutableState)(nil).GetHistoryBuilder))\n}\n\n// GetHistorySize mocks base method.\nfunc (m *MockMutableState) GetHistorySize() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistorySize\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetHistorySize indicates an expected call of GetHistorySize.\nfunc (mr *MockMutableStateMockRecorder) GetHistorySize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistorySize\", reflect.TypeOf((*MockMutableState)(nil).GetHistorySize))\n}\n\n// GetInFlightDecision mocks base method.\nfunc (m *MockMutableState) GetInFlightDecision() (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetInFlightDecision\")\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetInFlightDecision indicates an expected call of GetInFlightDecision.\nfunc (mr *MockMutableStateMockRecorder) GetInFlightDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetInFlightDecision\", reflect.TypeOf((*MockMutableState)(nil).GetInFlightDecision))\n}\n\n// GetLastFirstEventID mocks base method.\nfunc (m *MockMutableState) GetLastFirstEventID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastFirstEventID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetLastFirstEventID indicates an expected call of GetLastFirstEventID.\nfunc (mr *MockMutableStateMockRecorder) GetLastFirstEventID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastFirstEventID\", reflect.TypeOf((*MockMutableState)(nil).GetLastFirstEventID))\n}\n\n// GetLastWriteVersion mocks base method.\nfunc (m *MockMutableState) GetLastWriteVersion() (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastWriteVersion\")\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetLastWriteVersion indicates an expected call of GetLastWriteVersion.\nfunc (mr *MockMutableStateMockRecorder) GetLastWriteVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastWriteVersion\", reflect.TypeOf((*MockMutableState)(nil).GetLastWriteVersion))\n}\n\n// GetNextEventID mocks base method.\nfunc (m *MockMutableState) GetNextEventID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetNextEventID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetNextEventID indicates an expected call of GetNextEventID.\nfunc (mr *MockMutableStateMockRecorder) GetNextEventID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetNextEventID\", reflect.TypeOf((*MockMutableState)(nil).GetNextEventID))\n}\n\n// GetPendingActivityInfos mocks base method.\nfunc (m *MockMutableState) GetPendingActivityInfos() map[int64]*persistence.ActivityInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingActivityInfos\")\n\tret0, _ := ret[0].(map[int64]*persistence.ActivityInfo)\n\treturn ret0\n}\n\n// GetPendingActivityInfos indicates an expected call of GetPendingActivityInfos.\nfunc (mr *MockMutableStateMockRecorder) GetPendingActivityInfos() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingActivityInfos\", reflect.TypeOf((*MockMutableState)(nil).GetPendingActivityInfos))\n}\n\n// GetPendingChildExecutionInfos mocks base method.\nfunc (m *MockMutableState) GetPendingChildExecutionInfos() map[int64]*persistence.ChildExecutionInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingChildExecutionInfos\")\n\tret0, _ := ret[0].(map[int64]*persistence.ChildExecutionInfo)\n\treturn ret0\n}\n\n// GetPendingChildExecutionInfos indicates an expected call of GetPendingChildExecutionInfos.\nfunc (mr *MockMutableStateMockRecorder) GetPendingChildExecutionInfos() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingChildExecutionInfos\", reflect.TypeOf((*MockMutableState)(nil).GetPendingChildExecutionInfos))\n}\n\n// GetPendingDecision mocks base method.\nfunc (m *MockMutableState) GetPendingDecision() (*DecisionInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingDecision\")\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetPendingDecision indicates an expected call of GetPendingDecision.\nfunc (mr *MockMutableStateMockRecorder) GetPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingDecision\", reflect.TypeOf((*MockMutableState)(nil).GetPendingDecision))\n}\n\n// GetPendingRequestCancelExternalInfos mocks base method.\nfunc (m *MockMutableState) GetPendingRequestCancelExternalInfos() map[int64]*persistence.RequestCancelInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingRequestCancelExternalInfos\")\n\tret0, _ := ret[0].(map[int64]*persistence.RequestCancelInfo)\n\treturn ret0\n}\n\n// GetPendingRequestCancelExternalInfos indicates an expected call of GetPendingRequestCancelExternalInfos.\nfunc (mr *MockMutableStateMockRecorder) GetPendingRequestCancelExternalInfos() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingRequestCancelExternalInfos\", reflect.TypeOf((*MockMutableState)(nil).GetPendingRequestCancelExternalInfos))\n}\n\n// GetPendingSignalExternalInfos mocks base method.\nfunc (m *MockMutableState) GetPendingSignalExternalInfos() map[int64]*persistence.SignalInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingSignalExternalInfos\")\n\tret0, _ := ret[0].(map[int64]*persistence.SignalInfo)\n\treturn ret0\n}\n\n// GetPendingSignalExternalInfos indicates an expected call of GetPendingSignalExternalInfos.\nfunc (mr *MockMutableStateMockRecorder) GetPendingSignalExternalInfos() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingSignalExternalInfos\", reflect.TypeOf((*MockMutableState)(nil).GetPendingSignalExternalInfos))\n}\n\n// GetPendingTimerInfos mocks base method.\nfunc (m *MockMutableState) GetPendingTimerInfos() map[string]*persistence.TimerInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingTimerInfos\")\n\tret0, _ := ret[0].(map[string]*persistence.TimerInfo)\n\treturn ret0\n}\n\n// GetPendingTimerInfos indicates an expected call of GetPendingTimerInfos.\nfunc (mr *MockMutableStateMockRecorder) GetPendingTimerInfos() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingTimerInfos\", reflect.TypeOf((*MockMutableState)(nil).GetPendingTimerInfos))\n}\n\n// GetPreviousStartedEventID mocks base method.\nfunc (m *MockMutableState) GetPreviousStartedEventID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPreviousStartedEventID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetPreviousStartedEventID indicates an expected call of GetPreviousStartedEventID.\nfunc (mr *MockMutableStateMockRecorder) GetPreviousStartedEventID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPreviousStartedEventID\", reflect.TypeOf((*MockMutableState)(nil).GetPreviousStartedEventID))\n}\n\n// GetQueryRegistry mocks base method.\nfunc (m *MockMutableState) GetQueryRegistry() query.Registry {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueryRegistry\")\n\tret0, _ := ret[0].(query.Registry)\n\treturn ret0\n}\n\n// GetQueryRegistry indicates an expected call of GetQueryRegistry.\nfunc (mr *MockMutableStateMockRecorder) GetQueryRegistry() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueryRegistry\", reflect.TypeOf((*MockMutableState)(nil).GetQueryRegistry))\n}\n\n// GetRequestCancelInfo mocks base method.\nfunc (m *MockMutableState) GetRequestCancelInfo(arg0 int64) (*persistence.RequestCancelInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRequestCancelInfo\", arg0)\n\tret0, _ := ret[0].(*persistence.RequestCancelInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetRequestCancelInfo indicates an expected call of GetRequestCancelInfo.\nfunc (mr *MockMutableStateMockRecorder) GetRequestCancelInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRequestCancelInfo\", reflect.TypeOf((*MockMutableState)(nil).GetRequestCancelInfo), arg0)\n}\n\n// GetRetryBackoffDuration mocks base method.\nfunc (m *MockMutableState) GetRetryBackoffDuration(errReason string) time.Duration {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRetryBackoffDuration\", errReason)\n\tret0, _ := ret[0].(time.Duration)\n\treturn ret0\n}\n\n// GetRetryBackoffDuration indicates an expected call of GetRetryBackoffDuration.\nfunc (mr *MockMutableStateMockRecorder) GetRetryBackoffDuration(errReason any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRetryBackoffDuration\", reflect.TypeOf((*MockMutableState)(nil).GetRetryBackoffDuration), errReason)\n}\n\n// GetSignalInfo mocks base method.\nfunc (m *MockMutableState) GetSignalInfo(arg0 int64) (*persistence.SignalInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSignalInfo\", arg0)\n\tret0, _ := ret[0].(*persistence.SignalInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetSignalInfo indicates an expected call of GetSignalInfo.\nfunc (mr *MockMutableStateMockRecorder) GetSignalInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSignalInfo\", reflect.TypeOf((*MockMutableState)(nil).GetSignalInfo), arg0)\n}\n\n// GetStartEvent mocks base method.\nfunc (m *MockMutableState) GetStartEvent(arg0 context.Context) (*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStartEvent\", arg0)\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStartEvent indicates an expected call of GetStartEvent.\nfunc (mr *MockMutableStateMockRecorder) GetStartEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStartEvent\", reflect.TypeOf((*MockMutableState)(nil).GetStartEvent), arg0)\n}\n\n// GetStartVersion mocks base method.\nfunc (m *MockMutableState) GetStartVersion() (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetStartVersion\")\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetStartVersion indicates an expected call of GetStartVersion.\nfunc (mr *MockMutableStateMockRecorder) GetStartVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetStartVersion\", reflect.TypeOf((*MockMutableState)(nil).GetStartVersion))\n}\n\n// GetTimerTasks mocks base method.\nfunc (m *MockMutableState) GetTimerTasks() []persistence.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTimerTasks\")\n\tret0, _ := ret[0].([]persistence.Task)\n\treturn ret0\n}\n\n// GetTimerTasks indicates an expected call of GetTimerTasks.\nfunc (mr *MockMutableStateMockRecorder) GetTimerTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTimerTasks\", reflect.TypeOf((*MockMutableState)(nil).GetTimerTasks))\n}\n\n// GetTransferTasks mocks base method.\nfunc (m *MockMutableState) GetTransferTasks() []persistence.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTransferTasks\")\n\tret0, _ := ret[0].([]persistence.Task)\n\treturn ret0\n}\n\n// GetTransferTasks indicates an expected call of GetTransferTasks.\nfunc (mr *MockMutableStateMockRecorder) GetTransferTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTransferTasks\", reflect.TypeOf((*MockMutableState)(nil).GetTransferTasks))\n}\n\n// GetUpdateCondition mocks base method.\nfunc (m *MockMutableState) GetUpdateCondition() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetUpdateCondition\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetUpdateCondition indicates an expected call of GetUpdateCondition.\nfunc (mr *MockMutableStateMockRecorder) GetUpdateCondition() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetUpdateCondition\", reflect.TypeOf((*MockMutableState)(nil).GetUpdateCondition))\n}\n\n// GetUserTimerInfo mocks base method.\nfunc (m *MockMutableState) GetUserTimerInfo(arg0 string) (*persistence.TimerInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetUserTimerInfo\", arg0)\n\tret0, _ := ret[0].(*persistence.TimerInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetUserTimerInfo indicates an expected call of GetUserTimerInfo.\nfunc (mr *MockMutableStateMockRecorder) GetUserTimerInfo(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetUserTimerInfo\", reflect.TypeOf((*MockMutableState)(nil).GetUserTimerInfo), arg0)\n}\n\n// GetUserTimerInfoByEventID mocks base method.\nfunc (m *MockMutableState) GetUserTimerInfoByEventID(arg0 int64) (*persistence.TimerInfo, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetUserTimerInfoByEventID\", arg0)\n\tret0, _ := ret[0].(*persistence.TimerInfo)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetUserTimerInfoByEventID indicates an expected call of GetUserTimerInfoByEventID.\nfunc (mr *MockMutableStateMockRecorder) GetUserTimerInfoByEventID(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetUserTimerInfoByEventID\", reflect.TypeOf((*MockMutableState)(nil).GetUserTimerInfoByEventID), arg0)\n}\n\n// GetVersionHistories mocks base method.\nfunc (m *MockMutableState) GetVersionHistories() *persistence.VersionHistories {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVersionHistories\")\n\tret0, _ := ret[0].(*persistence.VersionHistories)\n\treturn ret0\n}\n\n// GetVersionHistories indicates an expected call of GetVersionHistories.\nfunc (mr *MockMutableStateMockRecorder) GetVersionHistories() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVersionHistories\", reflect.TypeOf((*MockMutableState)(nil).GetVersionHistories))\n}\n\n// GetWorkflowStateCloseStatus mocks base method.\nfunc (m *MockMutableState) GetWorkflowStateCloseStatus() (int, int) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowStateCloseStatus\")\n\tret0, _ := ret[0].(int)\n\tret1, _ := ret[1].(int)\n\treturn ret0, ret1\n}\n\n// GetWorkflowStateCloseStatus indicates an expected call of GetWorkflowStateCloseStatus.\nfunc (mr *MockMutableStateMockRecorder) GetWorkflowStateCloseStatus() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowStateCloseStatus\", reflect.TypeOf((*MockMutableState)(nil).GetWorkflowStateCloseStatus))\n}\n\n// GetWorkflowType mocks base method.\nfunc (m *MockMutableState) GetWorkflowType() *types.WorkflowType {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowType\")\n\tret0, _ := ret[0].(*types.WorkflowType)\n\treturn ret0\n}\n\n// GetWorkflowType indicates an expected call of GetWorkflowType.\nfunc (mr *MockMutableStateMockRecorder) GetWorkflowType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowType\", reflect.TypeOf((*MockMutableState)(nil).GetWorkflowType))\n}\n\n// HasBufferedEvents mocks base method.\nfunc (m *MockMutableState) HasBufferedEvents() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasBufferedEvents\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasBufferedEvents indicates an expected call of HasBufferedEvents.\nfunc (mr *MockMutableStateMockRecorder) HasBufferedEvents() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasBufferedEvents\", reflect.TypeOf((*MockMutableState)(nil).HasBufferedEvents))\n}\n\n// HasInFlightDecision mocks base method.\nfunc (m *MockMutableState) HasInFlightDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasInFlightDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasInFlightDecision indicates an expected call of HasInFlightDecision.\nfunc (mr *MockMutableStateMockRecorder) HasInFlightDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasInFlightDecision\", reflect.TypeOf((*MockMutableState)(nil).HasInFlightDecision))\n}\n\n// HasParentExecution mocks base method.\nfunc (m *MockMutableState) HasParentExecution() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasParentExecution\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasParentExecution indicates an expected call of HasParentExecution.\nfunc (mr *MockMutableStateMockRecorder) HasParentExecution() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasParentExecution\", reflect.TypeOf((*MockMutableState)(nil).HasParentExecution))\n}\n\n// HasPendingDecision mocks base method.\nfunc (m *MockMutableState) HasPendingDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasPendingDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasPendingDecision indicates an expected call of HasPendingDecision.\nfunc (mr *MockMutableStateMockRecorder) HasPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasPendingDecision\", reflect.TypeOf((*MockMutableState)(nil).HasPendingDecision))\n}\n\n// HasProcessedOrPendingDecision mocks base method.\nfunc (m *MockMutableState) HasProcessedOrPendingDecision() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasProcessedOrPendingDecision\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasProcessedOrPendingDecision indicates an expected call of HasProcessedOrPendingDecision.\nfunc (mr *MockMutableStateMockRecorder) HasProcessedOrPendingDecision() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasProcessedOrPendingDecision\", reflect.TypeOf((*MockMutableState)(nil).HasProcessedOrPendingDecision))\n}\n\n// IsCancelRequested mocks base method.\nfunc (m *MockMutableState) IsCancelRequested() (bool, string) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsCancelRequested\")\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(string)\n\treturn ret0, ret1\n}\n\n// IsCancelRequested indicates an expected call of IsCancelRequested.\nfunc (mr *MockMutableStateMockRecorder) IsCancelRequested() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsCancelRequested\", reflect.TypeOf((*MockMutableState)(nil).IsCancelRequested))\n}\n\n// IsCurrentWorkflowGuaranteed mocks base method.\nfunc (m *MockMutableState) IsCurrentWorkflowGuaranteed() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsCurrentWorkflowGuaranteed\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsCurrentWorkflowGuaranteed indicates an expected call of IsCurrentWorkflowGuaranteed.\nfunc (mr *MockMutableStateMockRecorder) IsCurrentWorkflowGuaranteed() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsCurrentWorkflowGuaranteed\", reflect.TypeOf((*MockMutableState)(nil).IsCurrentWorkflowGuaranteed))\n}\n\n// IsResourceDuplicated mocks base method.\nfunc (m *MockMutableState) IsResourceDuplicated(resourceDedupKey definition.DeduplicationID) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsResourceDuplicated\", resourceDedupKey)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsResourceDuplicated indicates an expected call of IsResourceDuplicated.\nfunc (mr *MockMutableStateMockRecorder) IsResourceDuplicated(resourceDedupKey any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsResourceDuplicated\", reflect.TypeOf((*MockMutableState)(nil).IsResourceDuplicated), resourceDedupKey)\n}\n\n// IsSignalRequested mocks base method.\nfunc (m *MockMutableState) IsSignalRequested(requestID string) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsSignalRequested\", requestID)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsSignalRequested indicates an expected call of IsSignalRequested.\nfunc (mr *MockMutableStateMockRecorder) IsSignalRequested(requestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsSignalRequested\", reflect.TypeOf((*MockMutableState)(nil).IsSignalRequested), requestID)\n}\n\n// IsStickyTaskListEnabled mocks base method.\nfunc (m *MockMutableState) IsStickyTaskListEnabled() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsStickyTaskListEnabled\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsStickyTaskListEnabled indicates an expected call of IsStickyTaskListEnabled.\nfunc (mr *MockMutableStateMockRecorder) IsStickyTaskListEnabled() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsStickyTaskListEnabled\", reflect.TypeOf((*MockMutableState)(nil).IsStickyTaskListEnabled))\n}\n\n// IsWorkflowCompleted mocks base method.\nfunc (m *MockMutableState) IsWorkflowCompleted() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowCompleted\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsWorkflowCompleted indicates an expected call of IsWorkflowCompleted.\nfunc (mr *MockMutableStateMockRecorder) IsWorkflowCompleted() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowCompleted\", reflect.TypeOf((*MockMutableState)(nil).IsWorkflowCompleted))\n}\n\n// IsWorkflowExecutionRunning mocks base method.\nfunc (m *MockMutableState) IsWorkflowExecutionRunning() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsWorkflowExecutionRunning\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsWorkflowExecutionRunning indicates an expected call of IsWorkflowExecutionRunning.\nfunc (mr *MockMutableStateMockRecorder) IsWorkflowExecutionRunning() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsWorkflowExecutionRunning\", reflect.TypeOf((*MockMutableState)(nil).IsWorkflowExecutionRunning))\n}\n\n// Load mocks base method.\nfunc (m *MockMutableState) Load(arg0 context.Context, arg1 *persistence.WorkflowMutableState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Load\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Load indicates an expected call of Load.\nfunc (mr *MockMutableStateMockRecorder) Load(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Load\", reflect.TypeOf((*MockMutableState)(nil).Load), arg0, arg1)\n}\n\n// ReplicateActivityInfo mocks base method.\nfunc (m *MockMutableState) ReplicateActivityInfo(arg0 *types.SyncActivityRequest, arg1 bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityInfo\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityInfo indicates an expected call of ReplicateActivityInfo.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityInfo(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityInfo\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityInfo), arg0, arg1)\n}\n\n// ReplicateActivityTaskCancelRequestedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskCancelRequestedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskCancelRequestedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskCancelRequestedEvent indicates an expected call of ReplicateActivityTaskCancelRequestedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskCancelRequestedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskCancelRequestedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskCancelRequestedEvent), arg0)\n}\n\n// ReplicateActivityTaskCanceledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskCanceledEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskCanceledEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskCanceledEvent indicates an expected call of ReplicateActivityTaskCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskCanceledEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskCanceledEvent), arg0)\n}\n\n// ReplicateActivityTaskCompletedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskCompletedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskCompletedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskCompletedEvent indicates an expected call of ReplicateActivityTaskCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskCompletedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskCompletedEvent), arg0)\n}\n\n// ReplicateActivityTaskFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskFailedEvent indicates an expected call of ReplicateActivityTaskFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskFailedEvent), arg0)\n}\n\n// ReplicateActivityTaskScheduledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskScheduledEvent(arg0 int64, arg1 *types.HistoryEvent, arg2 bool) (*persistence.ActivityInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskScheduledEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*persistence.ActivityInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateActivityTaskScheduledEvent indicates an expected call of ReplicateActivityTaskScheduledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskScheduledEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskScheduledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskScheduledEvent), arg0, arg1, arg2)\n}\n\n// ReplicateActivityTaskStartedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskStartedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskStartedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskStartedEvent indicates an expected call of ReplicateActivityTaskStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskStartedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskStartedEvent), arg0)\n}\n\n// ReplicateActivityTaskTimedOutEvent mocks base method.\nfunc (m *MockMutableState) ReplicateActivityTaskTimedOutEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateActivityTaskTimedOutEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateActivityTaskTimedOutEvent indicates an expected call of ReplicateActivityTaskTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateActivityTaskTimedOutEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateActivityTaskTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateActivityTaskTimedOutEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionCanceledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionCanceledEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionCanceledEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionCanceledEvent indicates an expected call of ReplicateChildWorkflowExecutionCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionCanceledEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionCanceledEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionCompletedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionCompletedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionCompletedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionCompletedEvent indicates an expected call of ReplicateChildWorkflowExecutionCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionCompletedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionCompletedEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionFailedEvent indicates an expected call of ReplicateChildWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionFailedEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionStartedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionStartedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionStartedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionStartedEvent indicates an expected call of ReplicateChildWorkflowExecutionStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionStartedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionStartedEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionTerminatedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionTerminatedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionTerminatedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionTerminatedEvent indicates an expected call of ReplicateChildWorkflowExecutionTerminatedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionTerminatedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionTerminatedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionTerminatedEvent), arg0)\n}\n\n// ReplicateChildWorkflowExecutionTimedOutEvent mocks base method.\nfunc (m *MockMutableState) ReplicateChildWorkflowExecutionTimedOutEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateChildWorkflowExecutionTimedOutEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateChildWorkflowExecutionTimedOutEvent indicates an expected call of ReplicateChildWorkflowExecutionTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateChildWorkflowExecutionTimedOutEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateChildWorkflowExecutionTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateChildWorkflowExecutionTimedOutEvent), arg0)\n}\n\n// ReplicateDecisionTaskCompletedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateDecisionTaskCompletedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskCompletedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskCompletedEvent indicates an expected call of ReplicateDecisionTaskCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateDecisionTaskCompletedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateDecisionTaskCompletedEvent), arg0)\n}\n\n// ReplicateDecisionTaskFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateDecisionTaskFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskFailedEvent indicates an expected call of ReplicateDecisionTaskFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateDecisionTaskFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateDecisionTaskFailedEvent), arg0)\n}\n\n// ReplicateDecisionTaskScheduledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateDecisionTaskScheduledEvent(arg0, arg1 int64, arg2 string, arg3 int32, arg4, arg5, arg6 int64, arg7 bool) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskScheduledEvent\", arg0, arg1, arg2, arg3, arg4, arg5, arg6, arg7)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateDecisionTaskScheduledEvent indicates an expected call of ReplicateDecisionTaskScheduledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateDecisionTaskScheduledEvent(arg0, arg1, arg2, arg3, arg4, arg5, arg6, arg7 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskScheduledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateDecisionTaskScheduledEvent), arg0, arg1, arg2, arg3, arg4, arg5, arg6, arg7)\n}\n\n// ReplicateDecisionTaskStartedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateDecisionTaskStartedEvent(arg0 *DecisionInfo, arg1, arg2, arg3 int64, arg4 string, arg5 int64) (*DecisionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskStartedEvent\", arg0, arg1, arg2, arg3, arg4, arg5)\n\tret0, _ := ret[0].(*DecisionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateDecisionTaskStartedEvent indicates an expected call of ReplicateDecisionTaskStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateDecisionTaskStartedEvent(arg0, arg1, arg2, arg3, arg4, arg5 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateDecisionTaskStartedEvent), arg0, arg1, arg2, arg3, arg4, arg5)\n}\n\n// ReplicateDecisionTaskTimedOutEvent mocks base method.\nfunc (m *MockMutableState) ReplicateDecisionTaskTimedOutEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateDecisionTaskTimedOutEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateDecisionTaskTimedOutEvent indicates an expected call of ReplicateDecisionTaskTimedOutEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateDecisionTaskTimedOutEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateDecisionTaskTimedOutEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateDecisionTaskTimedOutEvent), arg0)\n}\n\n// ReplicateExternalWorkflowExecutionCancelRequested mocks base method.\nfunc (m *MockMutableState) ReplicateExternalWorkflowExecutionCancelRequested(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateExternalWorkflowExecutionCancelRequested\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateExternalWorkflowExecutionCancelRequested indicates an expected call of ReplicateExternalWorkflowExecutionCancelRequested.\nfunc (mr *MockMutableStateMockRecorder) ReplicateExternalWorkflowExecutionCancelRequested(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateExternalWorkflowExecutionCancelRequested\", reflect.TypeOf((*MockMutableState)(nil).ReplicateExternalWorkflowExecutionCancelRequested), arg0)\n}\n\n// ReplicateExternalWorkflowExecutionSignaled mocks base method.\nfunc (m *MockMutableState) ReplicateExternalWorkflowExecutionSignaled(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateExternalWorkflowExecutionSignaled\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateExternalWorkflowExecutionSignaled indicates an expected call of ReplicateExternalWorkflowExecutionSignaled.\nfunc (mr *MockMutableStateMockRecorder) ReplicateExternalWorkflowExecutionSignaled(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateExternalWorkflowExecutionSignaled\", reflect.TypeOf((*MockMutableState)(nil).ReplicateExternalWorkflowExecutionSignaled), arg0)\n}\n\n// ReplicateRequestCancelExternalWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateRequestCancelExternalWorkflowExecutionFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateRequestCancelExternalWorkflowExecutionFailedEvent indicates an expected call of ReplicateRequestCancelExternalWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateRequestCancelExternalWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateRequestCancelExternalWorkflowExecutionFailedEvent), arg0)\n}\n\n// ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(arg0 int64, arg1 *types.HistoryEvent, arg2 string) (*persistence.RequestCancelInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*persistence.RequestCancelInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent indicates an expected call of ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// ReplicateSignalExternalWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateSignalExternalWorkflowExecutionFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateSignalExternalWorkflowExecutionFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateSignalExternalWorkflowExecutionFailedEvent indicates an expected call of ReplicateSignalExternalWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateSignalExternalWorkflowExecutionFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateSignalExternalWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateSignalExternalWorkflowExecutionFailedEvent), arg0)\n}\n\n// ReplicateSignalExternalWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateSignalExternalWorkflowExecutionInitiatedEvent(arg0 int64, arg1 *types.HistoryEvent, arg2 string) (*persistence.SignalInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateSignalExternalWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*persistence.SignalInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateSignalExternalWorkflowExecutionInitiatedEvent indicates an expected call of ReplicateSignalExternalWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateSignalExternalWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateSignalExternalWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateSignalExternalWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// ReplicateStartChildWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateStartChildWorkflowExecutionFailedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateStartChildWorkflowExecutionFailedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateStartChildWorkflowExecutionFailedEvent indicates an expected call of ReplicateStartChildWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateStartChildWorkflowExecutionFailedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateStartChildWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateStartChildWorkflowExecutionFailedEvent), arg0)\n}\n\n// ReplicateStartChildWorkflowExecutionInitiatedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateStartChildWorkflowExecutionInitiatedEvent(arg0 int64, arg1 *types.HistoryEvent, arg2 string) (*persistence.ChildExecutionInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateStartChildWorkflowExecutionInitiatedEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(*persistence.ChildExecutionInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateStartChildWorkflowExecutionInitiatedEvent indicates an expected call of ReplicateStartChildWorkflowExecutionInitiatedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateStartChildWorkflowExecutionInitiatedEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateStartChildWorkflowExecutionInitiatedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateStartChildWorkflowExecutionInitiatedEvent), arg0, arg1, arg2)\n}\n\n// ReplicateTimerCanceledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateTimerCanceledEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateTimerCanceledEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateTimerCanceledEvent indicates an expected call of ReplicateTimerCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateTimerCanceledEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateTimerCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateTimerCanceledEvent), arg0)\n}\n\n// ReplicateTimerFiredEvent mocks base method.\nfunc (m *MockMutableState) ReplicateTimerFiredEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateTimerFiredEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateTimerFiredEvent indicates an expected call of ReplicateTimerFiredEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateTimerFiredEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateTimerFiredEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateTimerFiredEvent), arg0)\n}\n\n// ReplicateTimerStartedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateTimerStartedEvent(arg0 *types.HistoryEvent) (*persistence.TimerInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateTimerStartedEvent\", arg0)\n\tret0, _ := ret[0].(*persistence.TimerInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReplicateTimerStartedEvent indicates an expected call of ReplicateTimerStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateTimerStartedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateTimerStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateTimerStartedEvent), arg0)\n}\n\n// ReplicateTransientDecisionTaskScheduled mocks base method.\nfunc (m *MockMutableState) ReplicateTransientDecisionTaskScheduled() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateTransientDecisionTaskScheduled\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateTransientDecisionTaskScheduled indicates an expected call of ReplicateTransientDecisionTaskScheduled.\nfunc (mr *MockMutableStateMockRecorder) ReplicateTransientDecisionTaskScheduled() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateTransientDecisionTaskScheduled\", reflect.TypeOf((*MockMutableState)(nil).ReplicateTransientDecisionTaskScheduled))\n}\n\n// ReplicateUpsertWorkflowSearchAttributesEvent mocks base method.\nfunc (m *MockMutableState) ReplicateUpsertWorkflowSearchAttributesEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateUpsertWorkflowSearchAttributesEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateUpsertWorkflowSearchAttributesEvent indicates an expected call of ReplicateUpsertWorkflowSearchAttributesEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateUpsertWorkflowSearchAttributesEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateUpsertWorkflowSearchAttributesEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateUpsertWorkflowSearchAttributesEvent), arg0)\n}\n\n// ReplicateWorkflowExecutionCancelRequestedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionCancelRequestedEvent(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionCancelRequestedEvent\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionCancelRequestedEvent indicates an expected call of ReplicateWorkflowExecutionCancelRequestedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionCancelRequestedEvent(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionCancelRequestedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionCancelRequestedEvent), arg0)\n}\n\n// ReplicateWorkflowExecutionCanceledEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionCanceledEvent(arg0 int64, arg1 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionCanceledEvent\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionCanceledEvent indicates an expected call of ReplicateWorkflowExecutionCanceledEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionCanceledEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionCanceledEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionCanceledEvent), arg0, arg1)\n}\n\n// ReplicateWorkflowExecutionCompletedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionCompletedEvent(arg0 int64, arg1 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionCompletedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionCompletedEvent indicates an expected call of ReplicateWorkflowExecutionCompletedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionCompletedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionCompletedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionCompletedEvent), arg0, arg1)\n}\n\n// ReplicateWorkflowExecutionContinuedAsNewEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionContinuedAsNewEvent(arg0 int64, arg1 string, arg2 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionContinuedAsNewEvent\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionContinuedAsNewEvent indicates an expected call of ReplicateWorkflowExecutionContinuedAsNewEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionContinuedAsNewEvent(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionContinuedAsNewEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionContinuedAsNewEvent), arg0, arg1, arg2)\n}\n\n// ReplicateWorkflowExecutionFailedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionFailedEvent(arg0 int64, arg1 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionFailedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionFailedEvent indicates an expected call of ReplicateWorkflowExecutionFailedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionFailedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionFailedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionFailedEvent), arg0, arg1)\n}\n\n// ReplicateWorkflowExecutionSignaled mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionSignaled(arg0 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionSignaled\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionSignaled indicates an expected call of ReplicateWorkflowExecutionSignaled.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionSignaled(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionSignaled\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionSignaled), arg0)\n}\n\n// ReplicateWorkflowExecutionStartedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionStartedEvent(arg0 *string, arg1 types.WorkflowExecution, arg2 string, arg3 *types.HistoryEvent, arg4 bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionStartedEvent\", arg0, arg1, arg2, arg3, arg4)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionStartedEvent indicates an expected call of ReplicateWorkflowExecutionStartedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionStartedEvent(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionStartedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionStartedEvent), arg0, arg1, arg2, arg3, arg4)\n}\n\n// ReplicateWorkflowExecutionTerminatedEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionTerminatedEvent(arg0 int64, arg1 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionTerminatedEvent\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionTerminatedEvent indicates an expected call of ReplicateWorkflowExecutionTerminatedEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionTerminatedEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionTerminatedEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionTerminatedEvent), arg0, arg1)\n}\n\n// ReplicateWorkflowExecutionTimedoutEvent mocks base method.\nfunc (m *MockMutableState) ReplicateWorkflowExecutionTimedoutEvent(arg0 int64, arg1 *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateWorkflowExecutionTimedoutEvent\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateWorkflowExecutionTimedoutEvent indicates an expected call of ReplicateWorkflowExecutionTimedoutEvent.\nfunc (mr *MockMutableStateMockRecorder) ReplicateWorkflowExecutionTimedoutEvent(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateWorkflowExecutionTimedoutEvent\", reflect.TypeOf((*MockMutableState)(nil).ReplicateWorkflowExecutionTimedoutEvent), arg0, arg1)\n}\n\n// RetryActivity mocks base method.\nfunc (m *MockMutableState) RetryActivity(ai *persistence.ActivityInfo, failureReason string, failureDetails []byte) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RetryActivity\", ai, failureReason, failureDetails)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RetryActivity indicates an expected call of RetryActivity.\nfunc (mr *MockMutableStateMockRecorder) RetryActivity(ai, failureReason, failureDetails any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RetryActivity\", reflect.TypeOf((*MockMutableState)(nil).RetryActivity), ai, failureReason, failureDetails)\n}\n\n// SetCurrentBranchToken mocks base method.\nfunc (m *MockMutableState) SetCurrentBranchToken(branchToken []byte) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SetCurrentBranchToken\", branchToken)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SetCurrentBranchToken indicates an expected call of SetCurrentBranchToken.\nfunc (mr *MockMutableStateMockRecorder) SetCurrentBranchToken(branchToken any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetCurrentBranchToken\", reflect.TypeOf((*MockMutableState)(nil).SetCurrentBranchToken), branchToken)\n}\n\n// SetHistoryBuilder mocks base method.\nfunc (m *MockMutableState) SetHistoryBuilder(hBuilder *HistoryBuilder) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetHistoryBuilder\", hBuilder)\n}\n\n// SetHistoryBuilder indicates an expected call of SetHistoryBuilder.\nfunc (mr *MockMutableStateMockRecorder) SetHistoryBuilder(hBuilder any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetHistoryBuilder\", reflect.TypeOf((*MockMutableState)(nil).SetHistoryBuilder), hBuilder)\n}\n\n// SetHistorySize mocks base method.\nfunc (m *MockMutableState) SetHistorySize(size int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetHistorySize\", size)\n}\n\n// SetHistorySize indicates an expected call of SetHistorySize.\nfunc (mr *MockMutableStateMockRecorder) SetHistorySize(size any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetHistorySize\", reflect.TypeOf((*MockMutableState)(nil).SetHistorySize), size)\n}\n\n// SetHistoryTree mocks base method.\nfunc (m *MockMutableState) SetHistoryTree(treeID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SetHistoryTree\", treeID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SetHistoryTree indicates an expected call of SetHistoryTree.\nfunc (mr *MockMutableStateMockRecorder) SetHistoryTree(treeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetHistoryTree\", reflect.TypeOf((*MockMutableState)(nil).SetHistoryTree), treeID)\n}\n\n// SetQueryRegistry mocks base method.\nfunc (m *MockMutableState) SetQueryRegistry(arg0 query.Registry) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetQueryRegistry\", arg0)\n}\n\n// SetQueryRegistry indicates an expected call of SetQueryRegistry.\nfunc (mr *MockMutableStateMockRecorder) SetQueryRegistry(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetQueryRegistry\", reflect.TypeOf((*MockMutableState)(nil).SetQueryRegistry), arg0)\n}\n\n// SetUpdateCondition mocks base method.\nfunc (m *MockMutableState) SetUpdateCondition(arg0 int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetUpdateCondition\", arg0)\n}\n\n// SetUpdateCondition indicates an expected call of SetUpdateCondition.\nfunc (mr *MockMutableStateMockRecorder) SetUpdateCondition(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetUpdateCondition\", reflect.TypeOf((*MockMutableState)(nil).SetUpdateCondition), arg0)\n}\n\n// SetVersionHistories mocks base method.\nfunc (m *MockMutableState) SetVersionHistories(arg0 *persistence.VersionHistories) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SetVersionHistories\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SetVersionHistories indicates an expected call of SetVersionHistories.\nfunc (mr *MockMutableStateMockRecorder) SetVersionHistories(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVersionHistories\", reflect.TypeOf((*MockMutableState)(nil).SetVersionHistories), arg0)\n}\n\n// StartTransaction mocks base method.\nfunc (m *MockMutableState) StartTransaction(ctx context.Context, entry *cache.DomainCacheEntry, incomingTaskVersion int64) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"StartTransaction\", ctx, entry, incomingTaskVersion)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartTransaction indicates an expected call of StartTransaction.\nfunc (mr *MockMutableStateMockRecorder) StartTransaction(ctx, entry, incomingTaskVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartTransaction\", reflect.TypeOf((*MockMutableState)(nil).StartTransaction), ctx, entry, incomingTaskVersion)\n}\n\n// UpdateActivity mocks base method.\nfunc (m *MockMutableState) UpdateActivity(arg0 *persistence.ActivityInfo) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateActivity\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateActivity indicates an expected call of UpdateActivity.\nfunc (mr *MockMutableStateMockRecorder) UpdateActivity(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateActivity\", reflect.TypeOf((*MockMutableState)(nil).UpdateActivity), arg0)\n}\n\n// UpdateActivityProgress mocks base method.\nfunc (m *MockMutableState) UpdateActivityProgress(ai *persistence.ActivityInfo, request *types.RecordActivityTaskHeartbeatRequest) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateActivityProgress\", ai, request)\n}\n\n// UpdateActivityProgress indicates an expected call of UpdateActivityProgress.\nfunc (mr *MockMutableStateMockRecorder) UpdateActivityProgress(ai, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateActivityProgress\", reflect.TypeOf((*MockMutableState)(nil).UpdateActivityProgress), ai, request)\n}\n\n// UpdateCurrentVersion mocks base method.\nfunc (m *MockMutableState) UpdateCurrentVersion(version int64, forceUpdate bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateCurrentVersion\", version, forceUpdate)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateCurrentVersion indicates an expected call of UpdateCurrentVersion.\nfunc (mr *MockMutableStateMockRecorder) UpdateCurrentVersion(version, forceUpdate any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateCurrentVersion\", reflect.TypeOf((*MockMutableState)(nil).UpdateCurrentVersion), version, forceUpdate)\n}\n\n// UpdateDecision mocks base method.\nfunc (m *MockMutableState) UpdateDecision(arg0 *DecisionInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateDecision\", arg0)\n}\n\n// UpdateDecision indicates an expected call of UpdateDecision.\nfunc (mr *MockMutableStateMockRecorder) UpdateDecision(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDecision\", reflect.TypeOf((*MockMutableState)(nil).UpdateDecision), arg0)\n}\n\n// UpdateDuplicatedResource mocks base method.\nfunc (m *MockMutableState) UpdateDuplicatedResource(resourceDedupKey definition.DeduplicationID) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UpdateDuplicatedResource\", resourceDedupKey)\n}\n\n// UpdateDuplicatedResource indicates an expected call of UpdateDuplicatedResource.\nfunc (mr *MockMutableStateMockRecorder) UpdateDuplicatedResource(resourceDedupKey any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDuplicatedResource\", reflect.TypeOf((*MockMutableState)(nil).UpdateDuplicatedResource), resourceDedupKey)\n}\n\n// UpdateUserTimer mocks base method.\nfunc (m *MockMutableState) UpdateUserTimer(arg0 *persistence.TimerInfo) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateUserTimer\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateUserTimer indicates an expected call of UpdateUserTimer.\nfunc (mr *MockMutableStateMockRecorder) UpdateUserTimer(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateUserTimer\", reflect.TypeOf((*MockMutableState)(nil).UpdateUserTimer), arg0)\n}\n\n// UpdateWorkflowStateCloseStatus mocks base method.\nfunc (m *MockMutableState) UpdateWorkflowStateCloseStatus(state, closeStatus int) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowStateCloseStatus\", state, closeStatus)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateWorkflowStateCloseStatus indicates an expected call of UpdateWorkflowStateCloseStatus.\nfunc (mr *MockMutableStateMockRecorder) UpdateWorkflowStateCloseStatus(state, closeStatus any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowStateCloseStatus\", reflect.TypeOf((*MockMutableState)(nil).UpdateWorkflowStateCloseStatus), state, closeStatus)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_generator.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mutable_state_task_generator_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// MutableStateTaskGenerator generates workflow transfer and timer tasks\n\tMutableStateTaskGenerator interface {\n\t\t// for workflow reset startTime should be the reset time instead of\n\t\t// the startEvent time, so that workflow timeout timestamp can be\n\t\t// re-calculated.\n\t\tGenerateWorkflowStartTasks(\n\t\t\tstartTime time.Time,\n\t\t\tstartEvent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateWorkflowCloseTasks(\n\t\t\tcloseEvent *types.HistoryEvent,\n\t\t\tworkflowDeletionTaskJitterRange int,\n\t\t) error\n\t\tGenerateRecordWorkflowStartedTasks(\n\t\t\tstartEvent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateDelayedDecisionTasks(\n\t\t\tstartEvent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateDecisionScheduleTasks(\n\t\t\tdecisionScheduleID int64,\n\t\t) error\n\t\tGenerateDecisionStartTasks(\n\t\t\tdecisionScheduleID int64,\n\t\t) error\n\t\tGenerateActivityTransferTasks(\n\t\t\tevent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateActivityRetryTasks(\n\t\t\tactivityScheduleID int64,\n\t\t) error\n\t\tGenerateChildWorkflowTasks(\n\t\t\tevent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateRequestCancelExternalTasks(\n\t\t\tevent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateSignalExternalTasks(\n\t\t\tevent *types.HistoryEvent,\n\t\t) error\n\t\tGenerateWorkflowSearchAttrTasks() error\n\t\tGenerateWorkflowResetTasks() error\n\t\t// these 2 APIs should only be called when mutable state transaction is being closed\n\t\tGenerateActivityTimerTasks() error\n\t\tGenerateUserTimerTasks() error\n\t}\n\n\tmutableStateTaskGeneratorImpl struct {\n\t\tlogger          log.Logger\n\t\tclusterMetadata cluster.Metadata\n\t\tdomainCache     cache.DomainCache\n\n\t\tmutableState MutableState\n\t}\n)\n\nconst (\n\tdefaultWorkflowRetentionInDays      int32 = 1\n\tdefaultInitIntervalForDecisionRetry       = 1 * time.Minute\n\tdefaultMaxIntervalForDecisionRetry        = 5 * time.Minute\n\tdefaultJitterCoefficient                  = 0.2\n)\n\nvar _ MutableStateTaskGenerator = (*mutableStateTaskGeneratorImpl)(nil)\n\n// NewMutableStateTaskGenerator creates a new task generator for mutable state\nfunc NewMutableStateTaskGenerator(\n\tlogger log.Logger,\n\tclusterMetadata cluster.Metadata,\n\tdomainCache cache.DomainCache,\n\tmutableState MutableState,\n) MutableStateTaskGenerator {\n\n\treturn &mutableStateTaskGeneratorImpl{\n\t\tlogger:          logger,\n\t\tclusterMetadata: clusterMetadata,\n\t\tdomainCache:     domainCache,\n\t\tmutableState:    mutableState,\n\t}\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateWorkflowStartTasks(\n\tstartTime time.Time,\n\tstartEvent *types.HistoryEvent,\n) error {\n\tattr := startEvent.WorkflowExecutionStartedEventAttributes\n\tfirstDecisionDelayDuration := time.Duration(attr.GetFirstDecisionTaskBackoffSeconds()) * time.Second\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tstartVersion := startEvent.Version\n\n\tworkflowTimeoutDuration := time.Duration(executionInfo.WorkflowTimeout) * time.Second\n\tworkflowTimeoutTimestamp := startTime.Add(workflowTimeoutDuration + firstDecisionDelayDuration)\n\t// ensure that the first attempt does not time out early based on retry policy timeout\n\tif attr.Attempt > 0 && !executionInfo.ExpirationTime.IsZero() && workflowTimeoutTimestamp.After(executionInfo.ExpirationTime) {\n\t\tworkflowTimeoutTimestamp = executionInfo.ExpirationTime\n\t}\n\tr.mutableState.AddTimerTasks(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: workflowTimeoutTimestamp,\n\t\t\tVersion:             startVersion,\n\t\t},\n\t\tTaskList: executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateWorkflowCloseTasks(\n\tcloseEvent *types.HistoryEvent,\n\tworkflowDeletionTaskJitterRange int,\n) error {\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\ttaskList := executionInfo.TaskList\n\tr.mutableState.AddTransferTasks(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: closeEvent.Version,\n\t\t},\n\t\tTaskList: taskList,\n\t})\n\n\tretentionInDays := defaultWorkflowRetentionInDays\n\tdomainEntry, err := r.domainCache.GetDomainByID(executionInfo.DomainID)\n\tswitch err.(type) {\n\tcase nil:\n\t\tretentionInDays = domainEntry.GetRetentionDays(executionInfo.WorkflowID)\n\tcase *types.EntityNotExistsError:\n\t\t// domain is not accessible, use default value above\n\tdefault:\n\t\treturn err\n\t}\n\n\tcloseTimestamp := time.Unix(0, closeEvent.GetTimestamp())\n\tretentionDuration := (time.Duration(retentionInDays) * time.Hour * 24)\n\tif workflowDeletionTaskJitterRange > 1 {\n\t\tretentionDuration += time.Duration(rand.Intn(workflowDeletionTaskJitterRange*60)) * time.Second\n\t}\n\n\tr.logger.Debug(\"GenerateWorkflowCloseTasks\",\n\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\ttag.WorkflowRunID(executionInfo.RunID),\n\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\ttag.Timestamp(closeTimestamp),\n\t)\n\tr.mutableState.AddTimerTasks(&persistence.DeleteHistoryEventTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: closeTimestamp.Add(retentionDuration),\n\t\t\tVersion:             closeEvent.Version,\n\t\t},\n\t\tTaskList: taskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateDelayedDecisionTasks(\n\tstartEvent *types.HistoryEvent,\n) error {\n\n\tstartVersion := startEvent.Version\n\tstartTimestamp := time.Unix(0, startEvent.GetTimestamp())\n\tstartAttr := startEvent.WorkflowExecutionStartedEventAttributes\n\tdecisionBackoffDuration := time.Duration(startAttr.GetFirstDecisionTaskBackoffSeconds()) * time.Second\n\texecutionTimestamp := startTimestamp.Add(decisionBackoffDuration)\n\n\t// noParentWorkflow case\n\tfirstDecisionDelayType := persistence.WorkflowBackoffTimeoutTypeCron\n\t// continue as new case\n\tif startAttr.Initiator != nil {\n\t\tswitch startAttr.GetInitiator() {\n\t\tcase types.ContinueAsNewInitiatorRetryPolicy:\n\t\t\tfirstDecisionDelayType = persistence.WorkflowBackoffTimeoutTypeRetry\n\t\tcase types.ContinueAsNewInitiatorCronSchedule:\n\t\t\tfirstDecisionDelayType = persistence.WorkflowBackoffTimeoutTypeCron\n\t\tcase types.ContinueAsNewInitiatorDecider:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"encounter continue as new iterator & first decision delay not 0\",\n\t\t\t}\n\t\tdefault:\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"unknown iterator retry policy: %v\", startAttr.GetInitiator()),\n\t\t\t}\n\t\t}\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTimerTasks(&persistence.WorkflowBackoffTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: executionTimestamp,\n\t\t\tVersion:             startVersion,\n\t\t},\n\t\tTimeoutType: firstDecisionDelayType,\n\t\tTaskList:    executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateRecordWorkflowStartedTasks(\n\tstartEvent *types.HistoryEvent,\n) error {\n\n\tstartVersion := startEvent.Version\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTransferTasks(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: startVersion,\n\t\t},\n\t\tTaskList: executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateDecisionScheduleTasks(\n\tdecisionScheduleID int64,\n) error {\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tdecision, ok := r.mutableState.GetDecisionInfo(\n\t\tdecisionScheduleID,\n\t)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending decision: %v\", decisionScheduleID),\n\t\t}\n\t}\n\n\toriginalTaskList := executionInfo.TaskList\n\tr.mutableState.AddTransferTasks(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: decision.Version,\n\t\t},\n\t\tTargetDomainID:       executionInfo.DomainID,\n\t\tTaskList:             decision.TaskList,\n\t\tScheduleID:           decision.ScheduleID,\n\t\tOriginalTaskList:     originalTaskList,\n\t\tOriginalTaskListKind: executionInfo.TaskListKind,\n\t})\n\n\tif scheduleToStartTimeout := r.mutableState.GetDecisionScheduleToStartTimeout(); scheduleToStartTimeout != 0 {\n\t\tscheduledTime := time.Unix(0, decision.ScheduledTimestamp)\n\t\tr.mutableState.AddTimerTasks(&persistence.DecisionTimeoutTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\t\tRunID:      executionInfo.RunID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t// TaskID is set by shard\n\t\t\t\tVisibilityTimestamp: scheduledTime.Add(scheduleToStartTimeout),\n\t\t\t\tVersion:             decision.Version,\n\t\t\t},\n\t\t\tTimeoutType:     int(TimerTypeScheduleToStart),\n\t\t\tEventID:         decision.ScheduleID,\n\t\t\tScheduleAttempt: decision.Attempt,\n\t\t\tTaskList:        originalTaskList,\n\t\t})\n\t}\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateDecisionStartTasks(\n\tdecisionScheduleID int64,\n) error {\n\n\tdecision, ok := r.mutableState.GetDecisionInfo(\n\t\tdecisionScheduleID,\n\t)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending decision: %v\", decisionScheduleID),\n\t\t}\n\t}\n\n\tstartedTime := time.Unix(0, decision.StartedTimestamp)\n\tstartToCloseTimeout := time.Duration(\n\t\tdecision.DecisionTimeout,\n\t) * time.Second\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\t// schedule timer exponentially if decision keeps failing\n\tif decision.Attempt > 1 {\n\t\tdefaultStartToCloseTimeout := executionInfo.DecisionStartToCloseTimeout\n\t\tstartToCloseTimeout = getNextDecisionTimeout(decision.Attempt, time.Duration(defaultStartToCloseTimeout)*time.Second)\n\t\tdecision.DecisionTimeout = int32(startToCloseTimeout.Seconds()) // override decision timeout\n\t\tr.mutableState.UpdateDecision(decision)\n\t}\n\n\tr.mutableState.AddTimerTasks(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: startedTime.Add(startToCloseTimeout),\n\t\t\tVersion:             decision.Version,\n\t\t},\n\t\tTimeoutType:     int(TimerTypeStartToClose),\n\t\tEventID:         decision.ScheduleID,\n\t\tScheduleAttempt: decision.Attempt,\n\t\tTaskList:        executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateActivityTransferTasks(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattr := event.ActivityTaskScheduledEventAttributes\n\tactivityScheduleID := event.ID\n\n\tactivityInfo, ok := r.mutableState.GetActivityInfo(activityScheduleID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending activity: %v\", activityScheduleID),\n\t\t}\n\t}\n\n\tvar targetDomainID string\n\tvar err error\n\tif activityInfo.DomainID != \"\" {\n\t\ttargetDomainID = activityInfo.DomainID\n\t} else {\n\t\t// TODO remove this block after Mar, 1th, 2020\n\t\t//  previously, DomainID in activity info is not used, so need to get\n\t\t//  schedule event from DB checking whether activity to be scheduled\n\t\t//  belongs to this domain\n\t\ttargetDomainID, err = r.getTargetDomainID(attr.GetDomain())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTransferTasks(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: activityInfo.Version,\n\t\t},\n\t\tTargetDomainID: targetDomainID,\n\t\tTaskList:       activityInfo.TaskList,\n\t\tScheduleID:     activityInfo.ScheduleID,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateActivityRetryTasks(\n\tactivityScheduleID int64,\n) error {\n\n\tai, ok := r.mutableState.GetActivityInfo(activityScheduleID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending activity: %v\", activityScheduleID),\n\t\t}\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTimerTasks(&persistence.ActivityRetryTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVersion:             ai.Version,\n\t\t\tVisibilityTimestamp: ai.ScheduledTime,\n\t\t},\n\t\tEventID:  ai.ScheduleID,\n\t\tAttempt:  int64(ai.Attempt),\n\t\tTaskList: ai.TaskList,\n\t})\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateChildWorkflowTasks(\n\tevent *types.HistoryEvent,\n) error {\n\n\tchildWorkflowScheduleID := event.ID\n\n\tchildWorkflowInfo, ok := r.mutableState.GetChildExecutionInfo(childWorkflowScheduleID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending child workflow: %v\", childWorkflowScheduleID),\n\t\t}\n\t}\n\n\tmsbDomainID := r.mutableState.GetDomainEntry().GetInfo().ID\n\n\ttargetDomainID := childWorkflowInfo.DomainID\n\tif childWorkflowInfo.DomainID == \"\" {\n\t\ttargetDomainID = msbDomainID\n\t}\n\n\terr := r.validateChildWorkflowParameters(msbDomainID, targetDomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tstartChildExecutionTask := &persistence.StartChildExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: childWorkflowInfo.Version,\n\t\t},\n\t\tTargetDomainID:   targetDomainID,\n\t\tTargetWorkflowID: childWorkflowInfo.StartedWorkflowID,\n\t\tInitiatedID:      childWorkflowInfo.InitiatedID,\n\t\tTaskList:         executionInfo.TaskList,\n\t}\n\n\tr.mutableState.AddTransferTasks(startChildExecutionTask)\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateRequestCancelExternalTasks(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattr := event.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\tscheduleID := event.ID\n\tversion := event.Version\n\ttargetDomainName := attr.GetDomain()\n\ttargetWorkflowID := attr.GetWorkflowExecution().GetWorkflowID()\n\ttargetRunID := attr.GetWorkflowExecution().GetRunID()\n\ttargetChildOnly := attr.GetChildWorkflowOnly()\n\n\t_, ok := r.mutableState.GetRequestCancelInfo(scheduleID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending request cancel external workflow: %v\", scheduleID),\n\t\t}\n\t}\n\n\ttargetDomainID, err := r.getTargetDomainID(targetDomainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tcancelExecutionTask := &persistence.CancelExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: version,\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: targetChildOnly,\n\t\tInitiatedID:             scheduleID,\n\t\tTaskList:                executionInfo.TaskList,\n\t}\n\n\tr.mutableState.AddTransferTasks(cancelExecutionTask)\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateSignalExternalTasks(\n\tevent *types.HistoryEvent,\n) error {\n\n\tattr := event.SignalExternalWorkflowExecutionInitiatedEventAttributes\n\tscheduleID := event.ID\n\tversion := event.Version\n\ttargetDomainName := attr.GetDomain()\n\ttargetWorkflowID := attr.GetWorkflowExecution().GetWorkflowID()\n\ttargetRunID := attr.GetWorkflowExecution().GetRunID()\n\ttargetChildOnly := attr.GetChildWorkflowOnly()\n\n\t_, ok := r.mutableState.GetSignalInfo(scheduleID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending signal external workflow: %v\", scheduleID),\n\t\t}\n\t}\n\n\ttargetDomainID, err := r.getTargetDomainID(targetDomainName)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tsignalExecutionTask := &persistence.SignalExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: version,\n\t\t},\n\t\tTargetDomainID:          targetDomainID,\n\t\tTargetWorkflowID:        targetWorkflowID,\n\t\tTargetRunID:             targetRunID,\n\t\tTargetChildWorkflowOnly: targetChildOnly,\n\t\tInitiatedID:             scheduleID,\n\t\tTaskList:                executionInfo.TaskList,\n\t}\n\n\tr.mutableState.AddTransferTasks(signalExecutionTask)\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateWorkflowSearchAttrTasks() error {\n\n\tcurrentVersion := r.mutableState.GetCurrentVersion()\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTransferTasks(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: currentVersion, // task processing does not check this version\n\t\t},\n\t\tTaskList: executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateWorkflowResetTasks() error {\n\n\tcurrentVersion := r.mutableState.GetCurrentVersion()\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tr.mutableState.AddTransferTasks(&persistence.ResetWorkflowTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID and VisibilityTimestamp are set by shard context\n\t\t\tVersion: currentVersion,\n\t\t},\n\t\tTaskList: executionInfo.TaskList,\n\t})\n\n\treturn nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateActivityTimerTasks() error {\n\n\t_, err := NewTimerSequence(r.mutableState).CreateNextActivityTimer()\n\treturn err\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) GenerateUserTimerTasks() error {\n\n\t_, err := NewTimerSequence(r.mutableState).CreateNextUserTimer()\n\treturn err\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) getTargetDomainID(\n\ttargetDomainName string,\n) (string, error) {\n\tif targetDomainName != \"\" {\n\t\treturn r.domainCache.GetDomainID(targetDomainName)\n\t}\n\n\treturn r.mutableState.GetExecutionInfo().DomainID, nil\n}\n\nfunc (r *mutableStateTaskGeneratorImpl) validateChildWorkflowParameters(msbDomainID string, targetDomainID string) error {\n\t// standard case\n\tif msbDomainID == targetDomainID {\n\t\treturn nil\n\t}\n\n\tthisDomain := r.mutableState.GetDomainEntry()\n\ttargetDomain, err := r.domainCache.GetDomainByID(targetDomainID)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"cannot get target domain for child workflow: %w\", err)\n\t}\n\n\t// Generally, cross-domain calls are not allowed for launching child workflows in global domains due to\n\t// the fact that we have removed cross-cluster calls (due to their overhead and limited use).\n\t//\n\t// There is a limited exception for local domains, which do not suffer from this problem can be\n\t// handled as an exception where the transfer task may be picked up by another domain in-cluster\n\t// without risk of the child workflow may end up in a different cluster.\n\tif thisDomain.IsGlobalDomain() || targetDomain.IsGlobalDomain() {\n\t\treturn &types.BadRequestError{\n\t\t\tMessage: fmt.Sprintf(\"The child workflow is \"+\n\t\t\t\t\"trying to use domain %s but it's running in domain %s. \"+\n\t\t\t\t\"Cross-cluster and cross domain child workflows are not supported for global domains\",\n\t\t\t\ttargetDomainID, msbDomainID),\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc getNextDecisionTimeout(attempt int64, defaultStartToCloseTimeout time.Duration) time.Duration {\n\tif attempt <= 1 {\n\t\treturn defaultStartToCloseTimeout\n\t}\n\n\tnextInterval := float64(defaultInitIntervalForDecisionRetry) * math.Pow(2, float64(attempt-2))\n\tnextInterval = math.Min(nextInterval, float64(defaultMaxIntervalForDecisionRetry))\n\tjitterPortion := int(defaultJitterCoefficient * nextInterval)\n\tif jitterPortion < 1 {\n\t\tjitterPortion = 1\n\t}\n\tnextInterval = nextInterval*(1-defaultJitterCoefficient) + float64(rand.Intn(jitterPortion))\n\treturn time.Duration(nextInterval)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_generator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: mutable_state_task_generator.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source mutable_state_task_generator.go -destination mutable_state_task_generator_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockMutableStateTaskGenerator is a mock of MutableStateTaskGenerator interface.\ntype MockMutableStateTaskGenerator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMutableStateTaskGeneratorMockRecorder\n\tisgomock struct{}\n}\n\n// MockMutableStateTaskGeneratorMockRecorder is the mock recorder for MockMutableStateTaskGenerator.\ntype MockMutableStateTaskGeneratorMockRecorder struct {\n\tmock *MockMutableStateTaskGenerator\n}\n\n// NewMockMutableStateTaskGenerator creates a new mock instance.\nfunc NewMockMutableStateTaskGenerator(ctrl *gomock.Controller) *MockMutableStateTaskGenerator {\n\tmock := &MockMutableStateTaskGenerator{ctrl: ctrl}\n\tmock.recorder = &MockMutableStateTaskGeneratorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMutableStateTaskGenerator) EXPECT() *MockMutableStateTaskGeneratorMockRecorder {\n\treturn m.recorder\n}\n\n// GenerateActivityRetryTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateActivityRetryTasks(activityScheduleID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateActivityRetryTasks\", activityScheduleID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateActivityRetryTasks indicates an expected call of GenerateActivityRetryTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateActivityRetryTasks(activityScheduleID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateActivityRetryTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateActivityRetryTasks), activityScheduleID)\n}\n\n// GenerateActivityTimerTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateActivityTimerTasks() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateActivityTimerTasks\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateActivityTimerTasks indicates an expected call of GenerateActivityTimerTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateActivityTimerTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateActivityTimerTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateActivityTimerTasks))\n}\n\n// GenerateActivityTransferTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateActivityTransferTasks(event *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateActivityTransferTasks\", event)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateActivityTransferTasks indicates an expected call of GenerateActivityTransferTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateActivityTransferTasks(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateActivityTransferTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateActivityTransferTasks), event)\n}\n\n// GenerateChildWorkflowTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateChildWorkflowTasks(event *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateChildWorkflowTasks\", event)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateChildWorkflowTasks indicates an expected call of GenerateChildWorkflowTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateChildWorkflowTasks(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateChildWorkflowTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateChildWorkflowTasks), event)\n}\n\n// GenerateDecisionScheduleTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateDecisionScheduleTasks(decisionScheduleID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateDecisionScheduleTasks\", decisionScheduleID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateDecisionScheduleTasks indicates an expected call of GenerateDecisionScheduleTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateDecisionScheduleTasks(decisionScheduleID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateDecisionScheduleTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateDecisionScheduleTasks), decisionScheduleID)\n}\n\n// GenerateDecisionStartTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateDecisionStartTasks(decisionScheduleID int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateDecisionStartTasks\", decisionScheduleID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateDecisionStartTasks indicates an expected call of GenerateDecisionStartTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateDecisionStartTasks(decisionScheduleID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateDecisionStartTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateDecisionStartTasks), decisionScheduleID)\n}\n\n// GenerateDelayedDecisionTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateDelayedDecisionTasks(startEvent *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateDelayedDecisionTasks\", startEvent)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateDelayedDecisionTasks indicates an expected call of GenerateDelayedDecisionTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateDelayedDecisionTasks(startEvent any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateDelayedDecisionTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateDelayedDecisionTasks), startEvent)\n}\n\n// GenerateRecordWorkflowStartedTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateRecordWorkflowStartedTasks(startEvent *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateRecordWorkflowStartedTasks\", startEvent)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateRecordWorkflowStartedTasks indicates an expected call of GenerateRecordWorkflowStartedTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateRecordWorkflowStartedTasks(startEvent any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateRecordWorkflowStartedTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateRecordWorkflowStartedTasks), startEvent)\n}\n\n// GenerateRequestCancelExternalTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateRequestCancelExternalTasks(event *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateRequestCancelExternalTasks\", event)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateRequestCancelExternalTasks indicates an expected call of GenerateRequestCancelExternalTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateRequestCancelExternalTasks(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateRequestCancelExternalTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateRequestCancelExternalTasks), event)\n}\n\n// GenerateSignalExternalTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateSignalExternalTasks(event *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateSignalExternalTasks\", event)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateSignalExternalTasks indicates an expected call of GenerateSignalExternalTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateSignalExternalTasks(event any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateSignalExternalTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateSignalExternalTasks), event)\n}\n\n// GenerateUserTimerTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateUserTimerTasks() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateUserTimerTasks\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateUserTimerTasks indicates an expected call of GenerateUserTimerTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateUserTimerTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateUserTimerTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateUserTimerTasks))\n}\n\n// GenerateWorkflowCloseTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateWorkflowCloseTasks(closeEvent *types.HistoryEvent, workflowDeletionTaskJitterRange int) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateWorkflowCloseTasks\", closeEvent, workflowDeletionTaskJitterRange)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateWorkflowCloseTasks indicates an expected call of GenerateWorkflowCloseTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateWorkflowCloseTasks(closeEvent, workflowDeletionTaskJitterRange any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateWorkflowCloseTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateWorkflowCloseTasks), closeEvent, workflowDeletionTaskJitterRange)\n}\n\n// GenerateWorkflowResetTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateWorkflowResetTasks() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateWorkflowResetTasks\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateWorkflowResetTasks indicates an expected call of GenerateWorkflowResetTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateWorkflowResetTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateWorkflowResetTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateWorkflowResetTasks))\n}\n\n// GenerateWorkflowSearchAttrTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateWorkflowSearchAttrTasks() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateWorkflowSearchAttrTasks\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateWorkflowSearchAttrTasks indicates an expected call of GenerateWorkflowSearchAttrTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateWorkflowSearchAttrTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateWorkflowSearchAttrTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateWorkflowSearchAttrTasks))\n}\n\n// GenerateWorkflowStartTasks mocks base method.\nfunc (m *MockMutableStateTaskGenerator) GenerateWorkflowStartTasks(startTime time.Time, startEvent *types.HistoryEvent) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateWorkflowStartTasks\", startTime, startEvent)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// GenerateWorkflowStartTasks indicates an expected call of GenerateWorkflowStartTasks.\nfunc (mr *MockMutableStateTaskGeneratorMockRecorder) GenerateWorkflowStartTasks(startTime, startEvent any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateWorkflowStartTasks\", reflect.TypeOf((*MockMutableStateTaskGenerator)(nil).GenerateWorkflowStartTasks), startTime, startEvent)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_generator_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\ntype (\n\tmutableStateTaskGeneratorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockDomainCache  *cache.MockDomainCache\n\t\tmockMutableState *MockMutableState\n\n\t\ttaskGenerator *mutableStateTaskGeneratorImpl\n\t}\n)\n\nfunc TestMutableStateTaskGeneratorSuite(t *testing.T) {\n\ts := new(mutableStateTaskGeneratorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockMutableState = NewMockMutableState(s.controller)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\n\ts.taskGenerator = NewMutableStateTaskGenerator(\n\t\tlog.NewNoop(),\n\t\tconstants.TestClusterMetadata,\n\t\ts.mockDomainCache,\n\t\ts.mockMutableState,\n\t).(*mutableStateTaskGeneratorImpl)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowCloseTasks_JitteredDeletion() {\n\tnow := time.Now()\n\tversion := int64(123)\n\tcloseEvent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tVersion:   version,\n\t}\n\tdomainEntry, err := s.mockDomainCache.GetDomainByID(constants.TestDomainID)\n\ts.NoError(err)\n\tretention := time.Duration(domainEntry.GetRetentionDays(constants.TestWorkflowID)) * time.Hour * 24\n\ttestCases := GenerateWorkflowCloseTasksTestCases(retention, closeEvent, now)\n\n\tfor _, tc := range testCases {\n\t\t// create new mockMutableState so can we can setup separete mock for each test case\n\t\tmockMutableState := NewMockMutableState(s.controller)\n\t\ttaskGenerator := NewMutableStateTaskGenerator(\n\t\t\tlog.NewNoop(),\n\t\t\tconstants.TestClusterMetadata,\n\t\t\ts.mockDomainCache,\n\t\t\tmockMutableState,\n\t\t)\n\n\t\tvar transferTasks []persistence.Task\n\t\tvar timerTasks []persistence.Task\n\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry).AnyTimes()\n\t\tmockMutableState.EXPECT().AddTransferTasks(gomock.Any()).Do(func(tasks ...persistence.Task) {\n\t\t\ttransferTasks = tasks\n\t\t}).MaxTimes(1)\n\t\tmockMutableState.EXPECT().AddTimerTasks(gomock.Any()).Do(func(tasks ...persistence.Task) {\n\t\t\ttimerTasks = tasks\n\t\t}).MaxTimes(1)\n\n\t\ttc.setupFn(mockMutableState)\n\t\terr := taskGenerator.GenerateWorkflowCloseTasks(closeEvent, 60)\n\t\ts.NoError(err)\n\n\t\tactualGeneratedTasks := transferTasks\n\t\tfor _, task := range actualGeneratedTasks {\n\t\t\t// force set visibility timestamp since that field is not assigned\n\t\t\t// for transfer and cross cluster in GenerateWorkflowCloseTasks\n\t\t\t// it will be set by shard context\n\t\t\t// set it to now so that we can easily test if other fields are equal\n\t\t\ttask.SetVisibilityTimestamp(now)\n\t\t}\n\t\tfor _, task := range timerTasks {\n\t\t\t// force set timer tasks because with jittering the timertask visibility time stamp\n\t\t\t// is not consistent for each run.\n\t\t\t// as long as code doesn't break during generation we should be ok\n\t\t\ttask.SetVisibilityTimestamp(time.Unix(0, closeEvent.GetTimestamp()).Add(retention))\n\t\t}\n\t\tactualGeneratedTasks = append(actualGeneratedTasks, timerTasks...)\n\t\ts.Equal(tc.generatedTasks, actualGeneratedTasks)\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowCloseTasks() {\n\tnow := time.Now()\n\tversion := int64(123)\n\tcloseEvent := &types.HistoryEvent{\n\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tVersion:   version,\n\t}\n\tdomainEntry, err := s.mockDomainCache.GetDomainByID(constants.TestDomainID)\n\ts.NoError(err)\n\tretention := time.Duration(domainEntry.GetRetentionDays(constants.TestWorkflowID)) * time.Hour * 24\n\ttestCases := GenerateWorkflowCloseTasksTestCases(retention, closeEvent, now)\n\n\tfor _, tc := range testCases {\n\t\t// create new mockMutableState so can we can setup separete mock for each test case\n\t\tmockMutableState := NewMockMutableState(s.controller)\n\t\ttaskGenerator := NewMutableStateTaskGenerator(\n\t\t\tlog.NewNoop(),\n\t\t\tconstants.TestClusterMetadata,\n\t\t\ts.mockDomainCache,\n\t\t\tmockMutableState,\n\t\t)\n\n\t\tvar transferTasks []persistence.Task\n\t\tvar timerTasks []persistence.Task\n\t\tmockMutableState.EXPECT().GetDomainEntry().Return(domainEntry).AnyTimes()\n\t\tmockMutableState.EXPECT().AddTransferTasks(gomock.Any()).Do(func(tasks ...persistence.Task) {\n\t\t\ttransferTasks = tasks\n\t\t}).MaxTimes(1)\n\t\tmockMutableState.EXPECT().AddTimerTasks(gomock.Any()).Do(func(tasks ...persistence.Task) {\n\t\t\ttimerTasks = tasks\n\t\t}).MaxTimes(1)\n\n\t\ttc.setupFn(mockMutableState)\n\t\terr := taskGenerator.GenerateWorkflowCloseTasks(closeEvent, 1)\n\t\ts.NoError(err)\n\n\t\tactualGeneratedTasks := transferTasks\n\t\tfor _, task := range actualGeneratedTasks {\n\t\t\t// force set visibility timestamp since that field is not assigned\n\t\t\t// for transfer and cross cluster in GenerateWorkflowCloseTasks\n\t\t\t// it will be set by shard context\n\t\t\t// set it to now so that we can easily test if other fields are equal\n\t\t\ttask.SetVisibilityTimestamp(now)\n\t\t}\n\t\tactualGeneratedTasks = append(actualGeneratedTasks, timerTasks...)\n\t\ts.Equal(tc.generatedTasks, actualGeneratedTasks)\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowCloseTasks_NotActive() {\n\tcloseEvent := &types.HistoryEvent{\n\t\tVersion:   constants.TestVersion,\n\t\tTimestamp: common.Ptr(time.Unix(1719224698, 0).UnixNano()),\n\t}\n\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"some-domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"rid\",\n\t\tTaskList:   \"task-list\",\n\t}).Times(1)\n\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{},\n\t\t&persistence.DomainConfig{\n\t\t\tRetention: defaultWorkflowRetentionInDays,\n\t\t},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1,\n\t)\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(\"some-domain-id\").Return(domainEntry, nil).Times(1)\n\n\tvar transferTasks []persistence.Task\n\ttransferTasks = append(transferTasks, &persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"some-domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: constants.TestVersion,\n\t\t},\n\t\tTaskList: \"task-list\",\n\t})\n\n\ts.mockMutableState.EXPECT().AddTransferTasks(transferTasks).Times(1)\n\n\texpectedDeletionTS := time.Unix(0, closeEvent.GetTimestamp()).\n\t\tAdd(time.Duration(defaultWorkflowRetentionInDays) * time.Hour * 24)\n\n\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.DeleteHistoryEventTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"some-domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: expectedDeletionTS,\n\t\t\tVersion:             closeEvent.Version,\n\t\t},\n\t\tTaskList: \"task-list\",\n\t})\n\n\terr := s.taskGenerator.GenerateWorkflowCloseTasks(closeEvent, 1)\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateFromTransferTask() {\n\tnow := time.Now()\n\ttestCases := []struct {\n\t\ttransferTask *persistence.TransferTaskInfo\n\t\texpectError  bool\n\t}{\n\t\t{\n\t\t\ttransferTask: &persistence.TransferTaskInfo{\n\t\t\t\tTaskType: persistence.TransferTaskTypeActivityTask,\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\ttransferTask: &persistence.TransferTaskInfo{\n\t\t\t\tTaskType:                persistence.TransferTaskTypeCancelExecution,\n\t\t\t\tTargetDomainID:          constants.TestDomainID,\n\t\t\t\tTargetWorkflowID:        constants.TestWorkflowID,\n\t\t\t\tTargetRunID:             constants.TestRunID,\n\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\tScheduleID:              int64(123),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\ttransferTask: &persistence.TransferTaskInfo{\n\t\t\t\tTaskType:                persistence.TransferTaskTypeSignalExecution,\n\t\t\t\tTargetDomainID:          constants.TestDomainID,\n\t\t\t\tTargetWorkflowID:        constants.TestWorkflowID,\n\t\t\t\tTargetRunID:             constants.TestRunID,\n\t\t\t\tTargetChildWorkflowOnly: false,\n\t\t\t\tScheduleID:              int64(123),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\ttransferTask: &persistence.TransferTaskInfo{\n\t\t\t\tTaskType:         persistence.TransferTaskTypeStartChildExecution,\n\t\t\t\tTargetDomainID:   constants.TestDomainID,\n\t\t\t\tTargetWorkflowID: constants.TestWorkflowID,\n\t\t\t\tScheduleID:       int64(123),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tif !tc.expectError {\n\t\t\ttc.transferTask.Version = int64(101)\n\t\t\ttc.transferTask.VisibilityTimestamp = now\n\t\t}\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGetNextDecisionTimeout() {\n\tdefaultStartToCloseTimeout := 10 * time.Second\n\texpectedResult := []time.Duration{\n\t\tdefaultStartToCloseTimeout,\n\t\tdefaultStartToCloseTimeout,\n\t\tdefaultInitIntervalForDecisionRetry,\n\t\tdefaultInitIntervalForDecisionRetry * 2,\n\t\tdefaultInitIntervalForDecisionRetry * 4,\n\t\tdefaultMaxIntervalForDecisionRetry,\n\t\tdefaultMaxIntervalForDecisionRetry,\n\t\tdefaultMaxIntervalForDecisionRetry,\n\t}\n\tfor i := 0; i < len(expectedResult); i++ {\n\t\tnext := getNextDecisionTimeout(int64(i), defaultStartToCloseTimeout)\n\t\texpected := expectedResult[i]\n\t\tmin, max := getNextBackoffRange(expected)\n\t\ts.True(next >= min, \"NextBackoff too low: actual: %v, expected: %v\", next, expected)\n\t\ts.True(next <= max, \"NextBackoff too high: actual: %v, expected: %v\", next, expected)\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowStartTasks() {\n\tstartTime := time.Now()\n\texpirationTime := startTime.Add(5 * time.Second)\n\n\ttestCases := []struct {\n\t\tname                string\n\t\tstartEvent          *types.HistoryEvent\n\t\tworkflowTimeout     int32\n\t\tvisibilityTimestamp time.Time\n\t}{\n\t\t{\n\t\t\tname: \"Success case - first attempt\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{Attempt: 0},\n\t\t\t\tVersion:                                 constants.TestVersion,\n\t\t\t},\n\t\t\tworkflowTimeout:     10,\n\t\t\tvisibilityTimestamp: startTime.Add(time.Duration(10) * time.Second),\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - second attempt and workflowTimeoutTimestamp before expirationTime\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{Attempt: 1},\n\t\t\t\tVersion:                                 constants.TestVersion,\n\t\t\t},\n\t\t\tworkflowTimeout:     1,\n\t\t\tvisibilityTimestamp: startTime.Add(time.Duration(1) * time.Second),\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - second attempt and workflowTimeoutTimestamp after expirationTime\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{Attempt: 1},\n\t\t\t\tVersion:                                 constants.TestVersion,\n\t\t\t},\n\t\t\tworkflowTimeout:     6,\n\t\t\tvisibilityTimestamp: expirationTime,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{WorkflowTimeout: tc.workflowTimeout, ExpirationTime: expirationTime, TaskList: \"task-list\"}).Times(1)\n\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.WorkflowTimeoutTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: tc.visibilityTimestamp,\n\t\t\t\t\tVersion:             tc.startEvent.Version,\n\t\t\t\t},\n\t\t\t\tTaskList: \"task-list\",\n\t\t\t}).Times(1)\n\n\t\t\terr := s.taskGenerator.GenerateWorkflowStartTasks(startTime, tc.startEvent)\n\n\t\t\ts.NoError(err)\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateDelayedDecisionTasks() {\n\ttimestamp := common.Int64Ptr(time.Now().UnixNano())\n\tfirstDecisionTaskBackoffSeconds := common.Int32Ptr(1)\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tstartEvent *types.HistoryEvent\n\t\tsetupMock  func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - nil initiator\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiator:                       nil,\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: firstDecisionTaskBackoffSeconds,\n\t\t\t\t},\n\t\t\t\tTimestamp: timestamp,\n\t\t\t},\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.WorkflowBackoffTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, *timestamp).Add(time.Duration(*firstDecisionTaskBackoffSeconds) * time.Second),\n\t\t\t\t\t\tVersion:             constants.TestVersion,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType: persistence.WorkflowBackoffTimeoutTypeCron,\n\t\t\t\t\tTaskList:    \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - retry policy initiator\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiator:                       types.ContinueAsNewInitiatorRetryPolicy.Ptr(),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: firstDecisionTaskBackoffSeconds,\n\t\t\t\t},\n\t\t\t\tTimestamp: timestamp,\n\t\t\t},\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.WorkflowBackoffTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, *timestamp).Add(time.Duration(*firstDecisionTaskBackoffSeconds) * time.Second),\n\t\t\t\t\t\tVersion:             constants.TestVersion,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType: persistence.WorkflowBackoffTimeoutTypeRetry,\n\t\t\t\t\tTaskList:    \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - cron initiator\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiator:                       types.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: firstDecisionTaskBackoffSeconds,\n\t\t\t\t},\n\t\t\t\tTimestamp: timestamp,\n\t\t\t},\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.WorkflowBackoffTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, *timestamp).Add(time.Duration(*firstDecisionTaskBackoffSeconds) * time.Second),\n\t\t\t\t\t\tVersion:             constants.TestVersion,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType: persistence.WorkflowBackoffTimeoutTypeCron,\n\t\t\t\t\tTaskList:    \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - decider initiator\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiator:                       types.ContinueAsNewInitiatorDecider.Ptr(),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: firstDecisionTaskBackoffSeconds,\n\t\t\t\t},\n\t\t\t\tTimestamp: timestamp,\n\t\t\t},\n\t\t\tsetupMock: func() {},\n\t\t\terr:       &types.InternalServiceError{Message: \"encounter continue as new iterator & first decision delay not 0\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - unknown initiator\",\n\t\t\tstartEvent: &types.HistoryEvent{\n\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tInitiator:                       types.ContinueAsNewInitiator(3).Ptr(),\n\t\t\t\t\tFirstDecisionTaskBackoffSeconds: firstDecisionTaskBackoffSeconds,\n\t\t\t\t},\n\t\t\t\tTimestamp: timestamp,\n\t\t\t},\n\t\t\tsetupMock: func() {},\n\t\t\terr:       &types.InternalServiceError{Message: fmt.Sprintf(\"unknown iterator retry policy: %v\", types.ContinueAsNewInitiator(3).Ptr())},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateDelayedDecisionTasks(tc.startEvent)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateRecordWorkflowStartedTasks() {\n\tstartEvent := &types.HistoryEvent{\n\t\tVersion: constants.TestVersion,\n\t}\n\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"rid\",\n\t\tTaskList:   \"task-list\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"rid\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: startEvent.Version,\n\t\t},\n\t\tTaskList: \"task-list\",\n\t}).Times(1)\n\n\terr := s.taskGenerator.GenerateRecordWorkflowStartedTasks(startEvent)\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateDecisionScheduleTasks() {\n\tdecisionScheduleID := int64(123)\n\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:     constants.TestDomainID,\n\t\tWorkflowID:   constants.TestWorkflowID,\n\t\tRunID:        constants.TestRunID,\n\t\tTaskList:     \"task-list\",\n\t\tTaskListKind: types.TaskListKindEphemeral,\n\t}\n\n\tdecision := &DecisionInfo{\n\t\tTaskList:           \"taskList\",\n\t\tScheduleID:         123,\n\t\tVersion:            constants.TestVersion,\n\t\tScheduledTimestamp: time.Now().UnixNano(),\n\t\tAttempt:            1,\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - scheduleToStartTimeout 0\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(decision, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: decision.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:       executionInfo.DomainID,\n\t\t\t\t\tTaskList:             decision.TaskList,\n\t\t\t\t\tScheduleID:           decision.ScheduleID,\n\t\t\t\t\tOriginalTaskList:     \"task-list\",\n\t\t\t\t\tOriginalTaskListKind: types.TaskListKindEphemeral,\n\t\t\t\t}).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionScheduleToStartTimeout().Return(time.Duration(0)).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - scheduleToStartTimeout not 0\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(decision, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: decision.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:       executionInfo.DomainID,\n\t\t\t\t\tTaskList:             decision.TaskList,\n\t\t\t\t\tScheduleID:           decision.ScheduleID,\n\t\t\t\t\tOriginalTaskList:     \"task-list\",\n\t\t\t\t\tOriginalTaskListKind: types.TaskListKindEphemeral,\n\t\t\t\t}).Times(1)\n\t\t\t\tscheduleToStartTimeout := time.Duration(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionScheduleToStartTimeout().Return(scheduleToStartTimeout).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.DecisionTimeoutTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, decision.ScheduledTimestamp).Add(scheduleToStartTimeout),\n\t\t\t\t\t\tVersion:             decision.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType:     int(TimerTypeScheduleToStart),\n\t\t\t\t\tEventID:         decision.ScheduleID,\n\t\t\t\t\tScheduleAttempt: decision.Attempt,\n\t\t\t\t\tTaskList:        \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetDecisionInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending decision: %v\", decisionScheduleID),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).Times(1)\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateDecisionScheduleTasks(decisionScheduleID)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateDecisionStartTasks() {\n\tseed := int64(1)\n\trand.Seed(seed)\n\tdecisionScheduleID := int64(123)\n\tgetDecision := func() *DecisionInfo {\n\t\treturn &DecisionInfo{\n\t\t\tVersion:          constants.TestVersion,\n\t\t\tScheduleID:       123,\n\t\t\tStartedTimestamp: time.Now().UnixNano(),\n\t\t}\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - attempt greater than 1\",\n\t\t\tsetupMock: func() {\n\t\t\t\tdecision := getDecision()\n\t\t\t\tdecision.Attempt = 2\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(decision, true).Times(1)\n\t\t\t\tdefaultStartToCloseTimeout := int32(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:                    \"domain-id\",\n\t\t\t\t\tWorkflowID:                  \"wf-id\",\n\t\t\t\t\tRunID:                       \"rid\",\n\t\t\t\t\tDecisionStartToCloseTimeout: defaultStartToCloseTimeout,\n\t\t\t\t\tTaskList:                    \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t\tstartToCloseTimeout := getNextDecisionTimeout(decision.Attempt, time.Duration(defaultStartToCloseTimeout)*time.Second)\n\t\t\t\tdecision.DecisionTimeout = int32(startToCloseTimeout.Seconds())\n\t\t\t\ts.mockMutableState.EXPECT().UpdateDecision(decision).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.DecisionTimeoutTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, decision.StartedTimestamp).Add(startToCloseTimeout),\n\t\t\t\t\t\tVersion:             decision.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType:     int(TimerTypeStartToClose),\n\t\t\t\t\tEventID:         decision.ScheduleID,\n\t\t\t\t\tScheduleAttempt: decision.Attempt,\n\t\t\t\t\tTaskList:        \"task-list\",\n\t\t\t\t})\n\t\t\t\trand.Seed(seed)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - attempt less or equal to 1\",\n\t\t\tsetupMock: func() {\n\t\t\t\tdecision := getDecision()\n\t\t\t\tdecision.DecisionTimeout = 1\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(decision, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.DecisionTimeoutTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, decision.StartedTimestamp).Add(time.Duration(decision.DecisionTimeout) * time.Second),\n\t\t\t\t\t\tVersion:             decision.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTimeoutType:     int(TimerTypeStartToClose),\n\t\t\t\t\tEventID:         decision.ScheduleID,\n\t\t\t\t\tScheduleAttempt: decision.Attempt,\n\t\t\t\t\tTaskList:        \"task-list\",\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetDecisionInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetDecisionInfo(decisionScheduleID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending decision: %v\", decisionScheduleID),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateDecisionStartTasks(decisionScheduleID)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateActivityTransferTasks() {\n\tdomain := constants.TestDomainName\n\tevent := &types.HistoryEvent{\n\t\tID: 123,\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\tDomain: &domain,\n\t\t},\n\t}\n\n\tgetActivityInfo := func() *persistence.ActivityInfo {\n\t\treturn &persistence.ActivityInfo{\n\t\t\tVersion:    constants.TestVersion,\n\t\t\tTaskList:   \"taskList\",\n\t\t\tScheduleID: 456,\n\t\t}\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - DomainID is not empty\",\n\t\t\tsetupMock: func() {\n\t\t\t\tactivityInfo := getActivityInfo()\n\t\t\t\tactivityInfo.DomainID = constants.TestDomainID\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(event.ID).Return(activityInfo, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.ActivityTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: activityInfo.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: activityInfo.DomainID,\n\t\t\t\t\tTaskList:       activityInfo.TaskList,\n\t\t\t\t\tScheduleID:     activityInfo.ScheduleID,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - DomainID is empty\",\n\t\t\tsetupMock: func() {\n\t\t\t\tactivityInfo := getActivityInfo()\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(event.ID).Return(activityInfo, true).Times(1)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(event.ActivityTaskScheduledEventAttributes.GetDomain()).Return(event.ActivityTaskScheduledEventAttributes.GetDomain(), nil).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.ActivityTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: activityInfo.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID: event.ActivityTaskScheduledEventAttributes.GetDomain(),\n\t\t\t\t\tTaskList:       activityInfo.TaskList,\n\t\t\t\t\tScheduleID:     activityInfo.ScheduleID,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetActivityInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(event.ID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending activity: %v\", event.ID),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - getTargetDomainID error\",\n\t\t\tsetupMock: func() {\n\t\t\t\tactivityInfo := getActivityInfo()\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(event.ID).Return(activityInfo, true).Times(1)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(event.ActivityTaskScheduledEventAttributes.GetDomain()).\n\t\t\t\t\tReturn(\"\", errors.New(\"get-target-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"get-target-domain-id-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateActivityTransferTasks(event)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateActivityRetryTasks() {\n\tactivityScheduleID := int64(123)\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMock: func() {\n\t\t\t\tai := &persistence.ActivityInfo{\n\t\t\t\t\tVersion:       constants.TestVersion,\n\t\t\t\t\tScheduledTime: time.Now(),\n\t\t\t\t\tScheduleID:    activityScheduleID,\n\t\t\t\t\tAttempt:       1,\n\t\t\t\t\tTaskList:      \"task-list2\",\n\t\t\t\t}\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(activityScheduleID).Return(ai, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.ActivityRetryTimerTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion:             ai.Version,\n\t\t\t\t\t\tVisibilityTimestamp: ai.ScheduledTime,\n\t\t\t\t\t},\n\t\t\t\t\tEventID:  ai.ScheduleID,\n\t\t\t\t\tAttempt:  int64(ai.Attempt),\n\t\t\t\t\tTaskList: \"task-list2\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetActivityInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetActivityInfo(activityScheduleID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending activity: %v\", activityScheduleID),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateActivityRetryTasks(activityScheduleID)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateChildWorkflowTasks() {\n\teventID := int64(123)\n\n\tgetChildWorkflowInfo := func() *persistence.ChildExecutionInfo {\n\t\treturn &persistence.ChildExecutionInfo{\n\t\t\tVersion:           constants.TestVersion,\n\t\t\tInitiatedID:       123,\n\t\t\tStartedWorkflowID: constants.TestWorkflowID,\n\t\t}\n\t}\n\n\tparentDomain := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{\n\t\t\tID:   constants.TestDomainID,\n\t\t\tName: constants.TestDomainName,\n\t\t},\n\t\t&persistence.DomainConfig{},\n\t\tnil,\n\t\t0,\n\t)\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMock  func()\n\t\tdomainName string\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - targetDomainID is not empty and isCrossClusterTask is false\",\n\t\t\tsetupMock: func() {\n\t\t\t\tchildWorkflowInfo := getChildWorkflowInfo()\n\t\t\t\tchildWorkflowInfo.DomainID = constants.TestDomainID\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetDomainEntry().Return(parentDomain).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetChildExecutionInfo(eventID).Return(childWorkflowInfo, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.StartChildExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: childWorkflowInfo.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:   childWorkflowInfo.DomainID,\n\t\t\t\t\tTargetWorkflowID: childWorkflowInfo.StartedWorkflowID,\n\t\t\t\t\tInitiatedID:      childWorkflowInfo.InitiatedID,\n\t\t\t\t\tTaskList:         \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"Success case - targetDomainID is empty - the expectations is that this should default to the existing parent workflow's domain\",\n\t\t\tdomainName: \"\",\n\t\t\tsetupMock: func() {\n\t\t\t\tchildWorkflowInfo := getChildWorkflowInfo()\n\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetDomainEntry().Return(parentDomain).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetChildExecutionInfo(eventID).Return(childWorkflowInfo, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.StartChildExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: childWorkflowInfo.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:   constants.TestDomainID,\n\t\t\t\t\tTargetWorkflowID: childWorkflowInfo.StartedWorkflowID,\n\t\t\t\t\tInitiatedID:      childWorkflowInfo.InitiatedID,\n\t\t\t\t\tTaskList:         \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"targetDomainID different to child's - this is invalid and should be an error\",\n\t\t\tdomainName: \"child-domain-B\",\n\t\t\tsetupMock: func() {\n\n\t\t\t\tchildWorkflowInfo := &persistence.ChildExecutionInfo{\n\t\t\t\t\tVersion:           constants.TestVersion,\n\t\t\t\t\tInitiatedID:       123,\n\t\t\t\t\tStartedWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tDomainID:          \"child-domain-B\",\n\t\t\t\t}\n\n\t\t\t\tcacheEntry := cache.NewGlobalDomainCacheEntryForTest(nil, nil, nil, 0)\n\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cacheEntry, nil).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetChildExecutionInfo(eventID).Return(childWorkflowInfo, true).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().GetDomainEntry().Return(parentDomain).Times(2)\n\t\t\t},\n\t\t\terr: &types.BadRequestError{\n\t\t\t\tMessage: \"The child workflow is trying to use domain child-domain-B but it's running in domain deadbeef-0123-4567-890a-bcdef0123456. Cross-cluster and cross domain child workflows are not supported for global domains\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetChildExecutionInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetChildExecutionInfo(eventID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending child workflow: %v\", eventID),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\tevent := &types.HistoryEvent{\n\t\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\t\tDomain: tc.domainName,\n\t\t\t\t},\n\t\t\t\tID: eventID,\n\t\t\t}\n\n\t\t\terr := s.taskGenerator.GenerateChildWorkflowTasks(event)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateRequestCancelExternalTasks() {\n\tevent := &types.HistoryEvent{\n\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain: constants.TestDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t},\n\t\tID:      123,\n\t\tVersion: constants.TestVersion,\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Error case - GetRequestCancelInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetRequestCancelInfo(event.ID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending request cancel external workflow: %v\", event.ID),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - getTargetDomainID error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetRequestCancelInfo(event.ID).Return(&persistence.RequestCancelInfo{}, true).Times(1)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(event.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes.GetDomain()).\n\t\t\t\t\tReturn(\"\", errors.New(\"get-target-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"get-target-domain-id-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateRequestCancelExternalTasks(event)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateSignalExternalTasks() {\n\tevent := &types.HistoryEvent{\n\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes: &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain: constants.TestDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t},\n\t\tID:      123,\n\t\tVersion: constants.TestVersion,\n\t}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func()\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMock: func() {\n\t\t\t\ttargetDomainID := \"target-domain-id\"\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t})\n\t\t\t\ts.mockMutableState.EXPECT().GetSignalInfo(event.ID).Return(nil, true).Times(1)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(event.SignalExternalWorkflowExecutionInitiatedEventAttributes.GetDomain()).\n\t\t\t\t\tReturn(targetDomainID, nil).Times(1)\n\t\t\t\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.SignalExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\t\tRunID:      \"rid\",\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVersion: event.Version,\n\t\t\t\t\t},\n\t\t\t\t\tTargetDomainID:   targetDomainID,\n\t\t\t\t\tTargetWorkflowID: event.SignalExternalWorkflowExecutionInitiatedEventAttributes.WorkflowExecution.GetWorkflowID(),\n\t\t\t\t\tTargetRunID:      event.SignalExternalWorkflowExecutionInitiatedEventAttributes.WorkflowExecution.GetRunID(),\n\t\t\t\t\tInitiatedID:      event.ID,\n\t\t\t\t\tTaskList:         \"task-list\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetSignalInfo error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetSignalInfo(event.ID).Return(nil, false).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{\n\t\t\t\tMessage: fmt.Sprintf(\"it could be a bug, cannot get pending signal external workflow: %v\", event.ID),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - getTargetDomainID error\",\n\t\t\tsetupMock: func() {\n\t\t\t\ts.mockMutableState.EXPECT().GetSignalInfo(event.ID).Return(&persistence.SignalInfo{}, true).Times(1)\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(event.SignalExternalWorkflowExecutionInitiatedEventAttributes.GetDomain()).\n\t\t\t\t\tReturn(\"\", errors.New(\"get-target-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"get-target-domain-id-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock()\n\n\t\t\terr := s.taskGenerator.GenerateSignalExternalTasks(event)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowSearchAttrTasks() {\n\tversion := int64(123)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"run-id\",\n\t\tTaskList:   \"task-list\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(version).Times(1)\n\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"run-id\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: version,\n\t\t},\n\t\tTaskList: \"task-list\",\n\t}).Times(1)\n\n\terr := s.taskGenerator.GenerateWorkflowSearchAttrTasks()\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateWorkflowResetTasks() {\n\tversion := int64(123)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"run-id\",\n\t\tTaskList:   \"task-list\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(version).Times(1)\n\ts.mockMutableState.EXPECT().AddTransferTasks(&persistence.ResetWorkflowTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"run-id\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: version,\n\t\t},\n\t\tTaskList: \"task-list\",\n\t}).Times(1)\n\n\terr := s.taskGenerator.GenerateWorkflowResetTasks()\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateActivityTimerTasks() {\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(nil).Times(1)\n\n\terr := s.taskGenerator.GenerateActivityTimerTasks()\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGenerateUserTimerTasks() {\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(nil).Times(1)\n\n\terr := s.taskGenerator.GenerateUserTimerTasks()\n\n\ts.NoError(err)\n}\n\nfunc (s *mutableStateTaskGeneratorSuite) TestGetTargetDomainID() {\n\ttestCases := []struct {\n\t\tname             string\n\t\tsetupMock        func(targetDomainName string, domainID string)\n\t\ttargetDomainName string\n\t\tdomainID         string\n\t\terr              error\n\t}{\n\t\t{\n\t\t\tname: \"Success case - empty targetDomainName\",\n\t\t\tsetupMock: func(targetDomainName string, domainID string) {\n\t\t\t\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: domainID}).Times(1)\n\t\t\t},\n\t\t\tdomainID:         constants.TestDomainID,\n\t\t\ttargetDomainName: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"Success case - targetDomainName is not empty\",\n\t\t\tsetupMock: func(targetDomainName string, domainID string) {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(targetDomainName).Return(domainID, nil).Times(1)\n\t\t\t},\n\t\t\tdomainID:         constants.TestDomainID,\n\t\t\ttargetDomainName: constants.TestDomainName,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetDomainID error\",\n\t\t\tsetupMock: func(targetDomainName string, domainID string) {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainID(targetDomainName).Return(\"\", errors.New(\"get-domain-id-error\")).Times(1)\n\t\t\t},\n\t\t\ttargetDomainName: constants.TestDomainName,\n\t\t\terr:              errors.New(\"get-domain-id-error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMock(tc.targetDomainName, tc.domainID)\n\n\t\t\ttargetDomainID, err := s.taskGenerator.getTargetDomainID(tc.targetDomainName)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(tc.domainID, targetDomainID)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc getNextBackoffRange(duration time.Duration) (time.Duration, time.Duration) {\n\trangeMin := time.Duration((1 - defaultJitterCoefficient) * float64(duration))\n\treturn rangeMin, duration\n}\n\nfunc TestValidationOfChildWorkflowParameters(t *testing.T) {\n\n\tg1 := cache.NewGlobalDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"g1\", Name: \"g1\"}, nil, nil, 0)\n\tg2 := cache.NewGlobalDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"g2\", Name: \"g2\"}, nil, nil, 0)\n\tl1 := cache.NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"l1\", Name: \"l1\"}, nil, \"\")\n\tl2 := cache.NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"l2\", Name: \"l2\"}, nil, \"\")\n\n\ttests := map[string]struct {\n\t\tthisDomain    *cache.DomainCacheEntry\n\t\tchildWorkflow *cache.DomainCacheEntry\n\t\tsetupCache    func(mockCache *cache.MockDomainCache)\n\t\texpectedError error\n\t}{\n\t\t\"Normal case: a child workflow running from the same domain as the parent shouldn't see any errors\": {\n\t\t\tthisDomain:    g1,\n\t\t\tchildWorkflow: g1,\n\t\t\tsetupCache:    func(mockCache *cache.MockDomainCache) {},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"local domains, cross-domain call\": {\n\t\t\tthisDomain:    l1,\n\t\t\tchildWorkflow: l2,\n\t\t\tsetupCache: func(cache *cache.MockDomainCache) {\n\t\t\t\tcache.EXPECT().GetDomainByID(l2.GetInfo().ID).Return(l2, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Global domains cross domain call. This is not permitted\": {\n\t\t\tthisDomain:    g1,\n\t\t\tchildWorkflow: g2,\n\t\t\tsetupCache: func(cache *cache.MockDomainCache) {\n\t\t\t\tcache.EXPECT().GetDomainByID(g2.GetInfo().ID).Return(g2, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.BadRequestError{Message: \"The child workflow is trying to use domain g2 but it's running in domain g1. Cross-cluster and cross domain child workflows are not supported for global domains\"},\n\t\t},\n\t\t\"Global domains cross domain call 2. This is not permitted\": {\n\t\t\tthisDomain:    l1,\n\t\t\tchildWorkflow: g2,\n\t\t\tsetupCache: func(cache *cache.MockDomainCache) {\n\t\t\t\tcache.EXPECT().GetDomainByID(g2.GetInfo().ID).Return(g2, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.BadRequestError{Message: \"The child workflow is trying to use domain g2 but it's running in domain l1. Cross-cluster and cross domain child workflows are not supported for global domains\"},\n\t\t},\n\t\t\"Global domains cross domain call 3. This is not permitted\": {\n\t\t\tthisDomain:    g1,\n\t\t\tchildWorkflow: l2,\n\t\t\tsetupCache: func(cache *cache.MockDomainCache) {\n\t\t\t\tcache.EXPECT().GetDomainByID(l2.GetInfo().ID).Return(g2, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: &types.BadRequestError{Message: \"The child workflow is trying to use domain l2 but it's running in domain g1. Cross-cluster and cross domain child workflows are not supported for global domains\"},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\n\t\t\tmsb := mutableStateBuilder{\n\t\t\t\tdomainEntry: td.thisDomain,\n\t\t\t}\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\ttd.setupCache(domainCache)\n\n\t\t\tmstb := mutableStateTaskGeneratorImpl{\n\t\t\t\tmutableState: &msb,\n\t\t\t\tdomainCache:  domainCache,\n\t\t\t}\n\n\t\t\terr := mstb.validateChildWorkflowParameters(td.thisDomain.GetInfo().ID, td.childWorkflow.GetInfo().ID)\n\t\t\tassert.Equal(t, td.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc GenerateWorkflowCloseTasksTestCases(retention time.Duration, closeEvent *types.HistoryEvent, now time.Time) []struct {\n\tsetupFn        func(mockMutableState *MockMutableState)\n\tgeneratedTasks []persistence.Task\n} {\n\tversion := int64(123)\n\treturn []struct {\n\t\tsetupFn        func(mockMutableState *MockMutableState)\n\t\tgeneratedTasks []persistence.Task\n\t}{\n\t\t{\n\t\t\t// no parent, no children\n\t\t\tsetupFn: func(mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().HasParentExecution().Return(false).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tgeneratedTasks: []persistence.Task{\n\t\t\t\t&persistence.CloseExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t\t&persistence.DeleteHistoryEventTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, closeEvent.GetTimestamp()).Add(retention),\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// parent and children all active in current cluster\n\t\t\tsetupFn: func(mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:         constants.TestDomainID,\n\t\t\t\t\tWorkflowID:       constants.TestWorkflowID,\n\t\t\t\t\tRunID:            constants.TestRunID,\n\t\t\t\t\tParentDomainID:   constants.TestDomainID,\n\t\t\t\t\tParentWorkflowID: \"parent workflowID\",\n\t\t\t\t\tParentRunID:      \"parent runID\",\n\t\t\t\t\tInitiatedID:      101,\n\t\t\t\t\tCloseStatus:      persistence.WorkflowCloseStatusCompleted,\n\t\t\t\t\tTaskList:         \"task-list\",\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().HasParentExecution().Return(true).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{\n\t\t\t\t\t102: {DomainID: constants.TestDomainID, ParentClosePolicy: types.ParentClosePolicyTerminate},\n\t\t\t\t\t103: {DomainID: constants.TestDomainID, ParentClosePolicy: types.ParentClosePolicyAbandon},\n\t\t\t\t}).AnyTimes()\n\t\t\t},\n\t\t\tgeneratedTasks: []persistence.Task{\n\t\t\t\t&persistence.CloseExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t\t&persistence.DeleteHistoryEventTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, closeEvent.GetTimestamp()).Add(retention),\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// no parent, no children\n\t\t\tsetupFn: func(mockMutableState *MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\tTaskList:   \"task-list\",\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().HasParentExecution().Return(false).AnyTimes()\n\t\t\t\tmockMutableState.EXPECT().GetPendingChildExecutionInfos().Return(nil).AnyTimes()\n\t\t\t},\n\t\t\tgeneratedTasks: []persistence.Task{\n\t\t\t\t&persistence.CloseExecutionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: now,\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t\t&persistence.DeleteHistoryEventTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tVisibilityTimestamp: time.Unix(0, closeEvent.GetTimestamp()).Add(retention),\n\t\t\t\t\t\tVersion:             version,\n\t\t\t\t\t},\n\t\t\t\t\tTaskList: \"task-list\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_refresher.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination mutable_state_task_refresher_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n)\n\nvar emptyTasks = []persistence.Task{}\n\ntype (\n\t// MutableStateTaskRefresher refreshes workflow transfer and timer tasks\n\tMutableStateTaskRefresher interface {\n\t\tRefreshTasks(ctx context.Context, startTime time.Time, mutableState MutableState) error\n\t}\n\n\tmutableStateTaskRefresherImpl struct {\n\t\tconfig          *config.Config\n\t\tclusterMetadata cluster.Metadata\n\t\tdomainCache     cache.DomainCache\n\t\teventsCache     events.Cache\n\t\tshardID         int\n\t\tlogger          log.Logger\n\n\t\tnewMutableStateTaskGeneratorFn                 func(log.Logger, cluster.Metadata, cache.DomainCache, MutableState) MutableStateTaskGenerator\n\t\trefreshTasksForWorkflowStartFn                 func(context.Context, time.Time, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForWorkflowCloseFn                 func(context.Context, MutableState, MutableStateTaskGenerator, int) error\n\t\trefreshTasksForRecordWorkflowStartedFn         func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForDecisionFn                      func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForActivityFn                      func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache, func(MutableState) TimerSequence) error\n\t\trefreshTasksForTimerFn                         func(context.Context, MutableState, MutableStateTaskGenerator, func(MutableState) TimerSequence) error\n\t\trefreshTasksForChildWorkflowFn                 func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForRequestCancelExternalWorkflowFn func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForSignalExternalWorkflowFn        func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForWorkflowSearchAttrFn            func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t}\n)\n\n// NewMutableStateTaskRefresher creates a new task refresher for mutable state\nfunc NewMutableStateTaskRefresher(\n\tconfig *config.Config,\n\tclusterMetadata cluster.Metadata,\n\tdomainCache cache.DomainCache,\n\teventsCache events.Cache,\n\tshardID int,\n\tlogger log.Logger,\n) MutableStateTaskRefresher {\n\treturn &mutableStateTaskRefresherImpl{\n\t\tconfig:          config,\n\t\tclusterMetadata: clusterMetadata,\n\t\tdomainCache:     domainCache,\n\t\teventsCache:     eventsCache,\n\t\tshardID:         shardID,\n\t\tlogger:          logger,\n\n\t\tnewMutableStateTaskGeneratorFn:                 NewMutableStateTaskGenerator,\n\t\trefreshTasksForWorkflowStartFn:                 refreshTasksForWorkflowStart,\n\t\trefreshTasksForWorkflowCloseFn:                 refreshTasksForWorkflowClose,\n\t\trefreshTasksForRecordWorkflowStartedFn:         refreshTasksForRecordWorkflowStarted,\n\t\trefreshTasksForDecisionFn:                      refreshTasksForDecision,\n\t\trefreshTasksForActivityFn:                      refreshTasksForActivity,\n\t\trefreshTasksForTimerFn:                         refreshTasksForTimer,\n\t\trefreshTasksForChildWorkflowFn:                 refreshTasksForChildWorkflow,\n\t\trefreshTasksForRequestCancelExternalWorkflowFn: refreshTasksForRequestCancelExternalWorkflow,\n\t\trefreshTasksForSignalExternalWorkflowFn:        refreshTasksForSignalExternalWorkflow,\n\t\trefreshTasksForWorkflowSearchAttrFn:            refreshTasksForWorkflowSearchAttr,\n\t}\n}\n\nfunc (r *mutableStateTaskRefresherImpl) RefreshTasks(\n\tctx context.Context,\n\tstartTime time.Time,\n\tmutableState MutableState,\n) error {\n\ttaskGenerator := r.newMutableStateTaskGeneratorFn(\n\t\tr.logger,\n\t\tr.clusterMetadata,\n\t\tr.domainCache,\n\t\tmutableState,\n\t)\n\n\tif err := r.refreshTasksForWorkflowStartFn(\n\t\tctx,\n\t\tstartTime,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForWorkflowCloseFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tr.config.WorkflowDeletionJitterRange(mutableState.GetDomainEntry().GetInfo().Name),\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForRecordWorkflowStartedFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForDecisionFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForActivityFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tr.shardID,\n\t\tr.eventsCache,\n\t\tNewTimerSequence,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForTimerFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tNewTimerSequence,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForChildWorkflowFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tr.shardID,\n\t\tr.eventsCache,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForRequestCancelExternalWorkflowFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tr.shardID,\n\t\tr.eventsCache,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif err := r.refreshTasksForSignalExternalWorkflowFn(\n\t\tctx,\n\t\tmutableState,\n\t\ttaskGenerator,\n\t\tr.shardID,\n\t\tr.eventsCache,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tif common.IsAdvancedVisibilityWritingEnabled(r.config.WriteVisibilityStoreName(), r.config.IsAdvancedVisConfigExist) {\n\t\tif err := r.refreshTasksForWorkflowSearchAttrFn(\n\t\t\tctx,\n\t\t\tmutableState,\n\t\t\ttaskGenerator,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc refreshTasksForWorkflowStart(\n\tctx context.Context,\n\tstartTime time.Time,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n) error {\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := taskGenerator.GenerateWorkflowStartTasks(\n\t\tstartTime,\n\t\tstartEvent,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tstartAttr := startEvent.WorkflowExecutionStartedEventAttributes\n\tif !mutableState.HasProcessedOrPendingDecision() && startAttr.GetFirstDecisionTaskBackoffSeconds() > 0 {\n\t\tif err := taskGenerator.GenerateDelayedDecisionTasks(\n\t\t\tstartEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc refreshTasksForWorkflowClose(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tworkflowDeletionTaskJitterRange int,\n) error {\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tif executionInfo.CloseStatus != persistence.WorkflowCloseStatusNone {\n\t\tcloseEvent, err := mutableState.GetCompletionEvent(ctx)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn taskGenerator.GenerateWorkflowCloseTasks(\n\t\t\tcloseEvent,\n\t\t\tworkflowDeletionTaskJitterRange,\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc refreshTasksForRecordWorkflowStarted(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n) error {\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tif executionInfo.CloseStatus == persistence.WorkflowCloseStatusNone {\n\t\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn taskGenerator.GenerateRecordWorkflowStartedTasks(\n\t\t\tstartEvent,\n\t\t)\n\t}\n\treturn nil\n}\n\nfunc refreshTasksForDecision(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n) error {\n\tif !mutableState.HasPendingDecision() {\n\t\t// no decision task at all\n\t\treturn nil\n\t}\n\n\tdecision, ok := mutableState.GetPendingDecision()\n\tif !ok {\n\t\treturn &types.InternalServiceError{Message: \"it could be a bug, cannot get pending decision\"}\n\t}\n\n\t// decision already started\n\tif decision.StartedID != constants.EmptyEventID {\n\t\treturn taskGenerator.GenerateDecisionStartTasks(\n\t\t\tdecision.ScheduleID,\n\t\t)\n\t}\n\n\t// decision only scheduled\n\treturn taskGenerator.GenerateDecisionScheduleTasks(\n\t\tdecision.ScheduleID,\n\t)\n}\n\nfunc refreshTasksForActivity(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tshardID int,\n\teventsCache events.Cache,\n\tnewTimerSequenceFn func(MutableState) TimerSequence,\n) error {\n\tcurrentBranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn err\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tpendingActivityInfos := mutableState.GetPendingActivityInfos()\n\tfor _, activityInfo := range pendingActivityInfos {\n\t\t// clear all activity timer task mask for later activity timer task re-generation\n\t\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\t\t// need to update activity timer task mask for which task is generated\n\t\tif err := mutableState.UpdateActivity(\n\t\t\tactivityInfo,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif activityInfo.StartedID != constants.EmptyEventID {\n\t\t\tcontinue\n\t\t}\n\t\tscheduleEvent, err := eventsCache.GetEvent(\n\t\t\tctx,\n\t\t\tshardID,\n\t\t\texecutionInfo.DomainID,\n\t\t\texecutionInfo.WorkflowID,\n\t\t\texecutionInfo.RunID,\n\t\t\tactivityInfo.ScheduledEventBatchID,\n\t\t\tactivityInfo.ScheduleID,\n\t\t\tcurrentBranchToken,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := taskGenerator.GenerateActivityTransferTasks(\n\t\t\tscheduleEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif _, err := newTimerSequenceFn(\n\t\tmutableState,\n\t).CreateNextActivityTimer(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc refreshTasksForTimer(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tnewTimerSequenceFn func(MutableState) TimerSequence,\n) error {\n\tpendingTimerInfos := mutableState.GetPendingTimerInfos()\n\n\tfor _, timerInfo := range pendingTimerInfos {\n\t\t// clear all timer task mask for later timer task re-generation\n\t\ttimerInfo.TaskStatus = TimerTaskStatusNone\n\n\t\t// need to update user timer task mask for which task is generated\n\t\tif err := mutableState.UpdateUserTimer(\n\t\t\ttimerInfo,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif _, err := newTimerSequenceFn(mutableState).CreateNextUserTimer(); err != nil {\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc refreshTasksForChildWorkflow(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tshardID int,\n\teventsCache events.Cache,\n) error {\n\tcurrentBranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn err\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tpendingChildWorkflowInfos := mutableState.GetPendingChildExecutionInfos()\n\tfor _, childWorkflowInfo := range pendingChildWorkflowInfos {\n\t\tif childWorkflowInfo.StartedID != constants.EmptyEventID {\n\t\t\tcontinue\n\t\t}\n\t\tscheduleEvent, err := eventsCache.GetEvent(\n\t\t\tctx,\n\t\t\tshardID,\n\t\t\texecutionInfo.DomainID,\n\t\t\texecutionInfo.WorkflowID,\n\t\t\texecutionInfo.RunID,\n\t\t\tchildWorkflowInfo.InitiatedEventBatchID,\n\t\t\tchildWorkflowInfo.InitiatedID,\n\t\t\tcurrentBranchToken,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := taskGenerator.GenerateChildWorkflowTasks(\n\t\t\tscheduleEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc refreshTasksForRequestCancelExternalWorkflow(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tshardID int,\n\teventsCache events.Cache,\n) error {\n\tcurrentBranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn err\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tpendingRequestCancelInfos := mutableState.GetPendingRequestCancelExternalInfos()\n\tfor _, requestCancelInfo := range pendingRequestCancelInfos {\n\t\tinitiateEvent, err := eventsCache.GetEvent(\n\t\t\tctx,\n\t\t\tshardID,\n\t\t\texecutionInfo.DomainID,\n\t\t\texecutionInfo.WorkflowID,\n\t\t\texecutionInfo.RunID,\n\t\t\trequestCancelInfo.InitiatedEventBatchID,\n\t\t\trequestCancelInfo.InitiatedID,\n\t\t\tcurrentBranchToken,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := taskGenerator.GenerateRequestCancelExternalTasks(\n\t\t\tinitiateEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc refreshTasksForSignalExternalWorkflow(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n\tshardID int,\n\teventsCache events.Cache,\n) error {\n\tcurrentBranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn err\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tpendingSignalInfos := mutableState.GetPendingSignalExternalInfos()\n\tfor _, signalInfo := range pendingSignalInfos {\n\t\tinitiateEvent, err := eventsCache.GetEvent(\n\t\t\tctx,\n\t\t\tshardID,\n\t\t\texecutionInfo.DomainID,\n\t\t\texecutionInfo.WorkflowID,\n\t\t\texecutionInfo.RunID,\n\t\t\tsignalInfo.InitiatedEventBatchID,\n\t\t\tsignalInfo.InitiatedID,\n\t\t\tcurrentBranchToken,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif err := taskGenerator.GenerateSignalExternalTasks(\n\t\t\tinitiateEvent,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc refreshTasksForWorkflowSearchAttr(\n\tctx context.Context,\n\tmutableState MutableState,\n\ttaskGenerator MutableStateTaskGenerator,\n) error {\n\treturn taskGenerator.GenerateWorkflowSearchAttrTasks()\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_refresher_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: mutable_state_task_refresher.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source mutable_state_task_refresher.go -destination mutable_state_task_refresher_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockMutableStateTaskRefresher is a mock of MutableStateTaskRefresher interface.\ntype MockMutableStateTaskRefresher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMutableStateTaskRefresherMockRecorder\n\tisgomock struct{}\n}\n\n// MockMutableStateTaskRefresherMockRecorder is the mock recorder for MockMutableStateTaskRefresher.\ntype MockMutableStateTaskRefresherMockRecorder struct {\n\tmock *MockMutableStateTaskRefresher\n}\n\n// NewMockMutableStateTaskRefresher creates a new mock instance.\nfunc NewMockMutableStateTaskRefresher(ctrl *gomock.Controller) *MockMutableStateTaskRefresher {\n\tmock := &MockMutableStateTaskRefresher{ctrl: ctrl}\n\tmock.recorder = &MockMutableStateTaskRefresherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMutableStateTaskRefresher) EXPECT() *MockMutableStateTaskRefresherMockRecorder {\n\treturn m.recorder\n}\n\n// RefreshTasks mocks base method.\nfunc (m *MockMutableStateTaskRefresher) RefreshTasks(ctx context.Context, startTime time.Time, mutableState MutableState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshTasks\", ctx, startTime, mutableState)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshTasks indicates an expected call of RefreshTasks.\nfunc (mr *MockMutableStateTaskRefresherMockRecorder) RefreshTasks(ctx, startTime, mutableState any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshTasks\", reflect.TypeOf((*MockMutableStateTaskRefresher)(nil).RefreshTasks), ctx, startTime, mutableState)\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_task_refresher_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n)\n\nfunc TestRefreshTasksForWorkflowStart(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get start event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate start tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateWorkflowStartTasks(gomock.Any(), &types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate delayed decision tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Ptr[int32](10),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmtg.EXPECT().GenerateWorkflowStartTasks(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tms.EXPECT().HasProcessedOrPendingDecision().Return(false)\n\t\t\t\tmtg.EXPECT().GenerateDelayedDecisionTasks(startEvent).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tstartEvent := &types.HistoryEvent{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tFirstDecisionTaskBackoffSeconds: common.Ptr[int32](10),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(startEvent, nil)\n\t\t\t\tmtg.EXPECT().GenerateWorkflowStartTasks(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tms.EXPECT().HasProcessedOrPendingDecision().Return(false)\n\t\t\t\tmtg.EXPECT().GenerateDelayedDecisionTasks(startEvent).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\ttc.mockSetup(ms, mtg)\n\t\t\terr := refreshTasksForWorkflowStart(context.Background(), time.Now(), ms, mtg)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForWorkflowStart err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForWorkflowClose(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get completion event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusCompleted})\n\t\t\t\tms.EXPECT().GetCompletionEvent(gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate close tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusCompleted})\n\t\t\t\tms.EXPECT().GetCompletionEvent(gomock.Any()).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateWorkflowCloseTasks(&types.HistoryEvent{ID: 1}, gomock.Any()).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - open workflow\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusNone})\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\ttc.mockSetup(ms, mtg)\n\t\t\terr := refreshTasksForWorkflowClose(context.Background(), ms, mtg, 100)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForWorkflowClose err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForRecordWorkflowStarted(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get start event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusNone})\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate record started tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetStartEvent(gomock.Any()).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusNone})\n\t\t\t\tmtg.EXPECT().GenerateRecordWorkflowStartedTasks(&types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - closed workflow\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{CloseStatus: persistence.WorkflowCloseStatusCompleted})\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\ttc.mockSetup(ms, mtg)\n\t\t\terr := refreshTasksForRecordWorkflowStarted(context.Background(), ms, mtg)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForRecordWorkflowStarted err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForDecision(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"success - no pending decision\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().HasPendingDecision().Return(false)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"bug - cannot get pending decision\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().HasPendingDecision().Return(true)\n\t\t\t\tms.EXPECT().GetPendingDecision().Return(nil, false)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - generate decision started task\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().HasPendingDecision().Return(true)\n\t\t\t\tms.EXPECT().GetPendingDecision().Return(&DecisionInfo{ScheduleID: 2, StartedID: 3}, true)\n\t\t\t\tmtg.EXPECT().GenerateDecisionStartTasks(int64(2)).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"success - generate decision started task\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tms.EXPECT().HasPendingDecision().Return(true)\n\t\t\t\tms.EXPECT().GetPendingDecision().Return(&DecisionInfo{ScheduleID: 2, StartedID: constants.EmptyEventID}, true)\n\t\t\t\tmtg.EXPECT().GenerateDecisionScheduleTasks(int64(2)).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\ttc.mockSetup(ms, mtg)\n\t\t\terr := refreshTasksForDecision(context.Background(), ms, mtg)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForDecision err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForActivity(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator, *events.MockCache, *MockTimerSequence)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get current branch token\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to update activity\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{})\n\t\t\t\tms.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{1: {Version: 1, TimerTaskStatus: TimerTaskStatusCreated}})\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 1, TimerTaskStatus: TimerTaskStatusNone}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{1: {Version: 1, TimerTaskStatus: TimerTaskStatusCreated, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}})\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 1, TimerTaskStatus: TimerTaskStatusNone, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}).Return(nil)\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(11), int64(12), []byte(\"token\")).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate activity tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{1: {Version: 1, TimerTaskStatus: TimerTaskStatusCreated, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}})\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 1, TimerTaskStatus: TimerTaskStatusNone, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}).Return(nil)\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(11), int64(12), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateActivityTransferTasks(&types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to create activity timer\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{1: {Version: 1, TimerTaskStatus: TimerTaskStatusCreated, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}})\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 1, TimerTaskStatus: TimerTaskStatusNone, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}).Return(nil)\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(11), int64(12), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateActivityTransferTasks(&types.HistoryEvent{ID: 1}).Return(nil)\n\t\t\t\tmt.EXPECT().CreateNextActivityTimer().Return(false, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{10: {Version: 0, StartedID: 10}, 1: {Version: 1, TimerTaskStatus: TimerTaskStatusCreated, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}})\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 0, StartedID: 10, TimerTaskStatus: TimerTaskStatusNone}).Return(nil)\n\t\t\t\tms.EXPECT().UpdateActivity(&persistence.ActivityInfo{Version: 1, TimerTaskStatus: TimerTaskStatusNone, ScheduledEventBatchID: 11, ScheduleID: 12, StartedID: constants.EmptyEventID}).Return(nil)\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(11), int64(12), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateActivityTransferTasks(&types.HistoryEvent{ID: 1}).Return(nil)\n\t\t\t\tmt.EXPECT().CreateNextActivityTimer().Return(true, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tmc := events.NewMockCache(ctrl)\n\t\t\tmt := NewMockTimerSequence(ctrl)\n\t\t\ttc.mockSetup(ms, mtg, mc, mt)\n\t\t\terr := refreshTasksForActivity(context.Background(), ms, mtg, 1, mc, func(MutableState) TimerSequence { return mt })\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForActivity err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForTimer(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator, *MockTimerSequence)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to update user timer\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\"0\": &persistence.TimerInfo{}})\n\t\t\t\tms.EXPECT().UpdateUserTimer(gomock.Any()).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to create user timer\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\"0\": &persistence.TimerInfo{}})\n\t\t\t\tms.EXPECT().UpdateUserTimer(gomock.Any()).Return(nil)\n\t\t\t\tmt.EXPECT().CreateNextUserTimer().Return(false, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mt *MockTimerSequence) {\n\t\t\t\tms.EXPECT().GetPendingTimerInfos().Return(map[string]*persistence.TimerInfo{\"0\": &persistence.TimerInfo{}})\n\t\t\t\tms.EXPECT().UpdateUserTimer(gomock.Any()).Return(nil)\n\t\t\t\tmt.EXPECT().CreateNextUserTimer().Return(false, nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tmt := NewMockTimerSequence(ctrl)\n\t\t\ttc.mockSetup(ms, mtg, mt)\n\t\t\terr := refreshTasksForTimer(context.Background(), ms, mtg, func(MutableState) TimerSequence { return mt })\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForTimer err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForChildWorkflow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator, *events.MockCache)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get current branch token\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, StartedID: constants.EmptyEventID, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate child workflow tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, StartedID: constants.EmptyEventID, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateChildWorkflowTasks(&types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingChildExecutionInfos().Return(map[int64]*persistence.ChildExecutionInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, StartedID: constants.EmptyEventID, Version: 1}, 11: {StartedID: 12}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateChildWorkflowTasks(&types.HistoryEvent{ID: 1}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tmc := events.NewMockCache(ctrl)\n\t\t\ttc.mockSetup(ms, mtg, mc)\n\t\t\terr := refreshTasksForChildWorkflow(context.Background(), ms, mtg, 1, mc)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForChildWorkflow err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForRequestCancelExternalWorkflow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator, *events.MockCache)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get current branch token\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingRequestCancelExternalInfos().Return(map[int64]*persistence.RequestCancelInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate child workflow tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingRequestCancelExternalInfos().Return(map[int64]*persistence.RequestCancelInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateRequestCancelExternalTasks(&types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingRequestCancelExternalInfos().Return(map[int64]*persistence.RequestCancelInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateRequestCancelExternalTasks(&types.HistoryEvent{ID: 1}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tmc := events.NewMockCache(ctrl)\n\t\t\ttc.mockSetup(ms, mtg, mc)\n\t\t\terr := refreshTasksForRequestCancelExternalWorkflow(context.Background(), ms, mtg, 1, mc)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForRequestCancelExternalWorkflow err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForSignalExternalWorkflow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableState, *MockMutableStateTaskGenerator, *events.MockCache)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to get current branch token\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get event\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingSignalExternalInfos().Return(map[int64]*persistence.SignalInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to generate child workflow tasks\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingSignalExternalInfos().Return(map[int64]*persistence.SignalInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateSignalExternalTasks(&types.HistoryEvent{ID: 1}).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(ms *MockMutableState, mtg *MockMutableStateTaskGenerator, mc *events.MockCache) {\n\t\t\t\tms.EXPECT().GetCurrentBranchToken().Return([]byte(\"token\"), nil)\n\t\t\t\tms.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{DomainID: \"domain-id\", WorkflowID: \"wf-id\", RunID: \"run-id\"})\n\t\t\t\tms.EXPECT().GetPendingSignalExternalInfos().Return(map[int64]*persistence.SignalInfo{1: {InitiatedEventBatchID: 1, InitiatedID: 2, Version: 1}})\n\t\t\t\tmc.EXPECT().GetEvent(gomock.Any(), gomock.Any(), \"domain-id\", \"wf-id\", \"run-id\", int64(1), int64(2), []byte(\"token\")).Return(&types.HistoryEvent{ID: 1}, nil)\n\t\t\t\tmtg.EXPECT().GenerateSignalExternalTasks(&types.HistoryEvent{ID: 1}).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tmc := events.NewMockCache(ctrl)\n\t\t\ttc.mockSetup(ms, mtg, mc)\n\t\t\terr := refreshTasksForSignalExternalWorkflow(context.Background(), ms, mtg, 1, mc)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForSignalExternalWorkflow err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasksForWorkflowSearchAttr(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*MockMutableStateTaskGenerator)\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"failed to generate workflow search attribute tasks\",\n\t\t\tmockSetup: func(mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tmtg.EXPECT().GenerateWorkflowSearchAttrTasks().Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(mtg *MockMutableStateTaskGenerator) {\n\t\t\t\tmtg.EXPECT().GenerateWorkflowSearchAttrTasks().Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\ttc.mockSetup(mtg)\n\t\t\terr := refreshTasksForWorkflowSearchAttr(context.Background(), nil, mtg)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"refreshTasksForWorkflowSearchAttr err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                                           string\n\t\trefreshTasksForWorkflowStartFn                 func(context.Context, time.Time, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForWorkflowCloseFn                 func(context.Context, MutableState, MutableStateTaskGenerator, int) error\n\t\trefreshTasksForRecordWorkflowStartedFn         func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForDecisionFn                      func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t\trefreshTasksForActivityFn                      func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache, func(MutableState) TimerSequence) error\n\t\trefreshTasksForTimerFn                         func(context.Context, MutableState, MutableStateTaskGenerator, func(MutableState) TimerSequence) error\n\t\trefreshTasksForChildWorkflowFn                 func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForRequestCancelExternalWorkflowFn func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForSignalExternalWorkflowFn        func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error\n\t\trefreshTasksForWorkflowSearchAttrFn            func(context.Context, MutableState, MutableStateTaskGenerator) error\n\t\twantErr                                        bool\n\t}{\n\t\t{\n\t\t\tname:                                   \"success\",\n\t\t\trefreshTasksForWorkflowStartFn:         func(context.Context, time.Time, MutableState, MutableStateTaskGenerator) error { return nil },\n\t\t\trefreshTasksForWorkflowCloseFn:         func(context.Context, MutableState, MutableStateTaskGenerator, int) error { return nil },\n\t\t\trefreshTasksForRecordWorkflowStartedFn: func(context.Context, MutableState, MutableStateTaskGenerator) error { return nil },\n\t\t\trefreshTasksForDecisionFn:              func(context.Context, MutableState, MutableStateTaskGenerator) error { return nil },\n\t\t\trefreshTasksForActivityFn: func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache, func(MutableState) TimerSequence) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\trefreshTasksForTimerFn: func(context.Context, MutableState, MutableStateTaskGenerator, func(MutableState) TimerSequence) error {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\trefreshTasksForChildWorkflowFn:                 func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error { return nil },\n\t\t\trefreshTasksForRequestCancelExternalWorkflowFn: func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error { return nil },\n\t\t\trefreshTasksForSignalExternalWorkflowFn:        func(context.Context, MutableState, MutableStateTaskGenerator, int, events.Cache) error { return nil },\n\t\t\trefreshTasksForWorkflowSearchAttrFn:            func(context.Context, MutableState, MutableStateTaskGenerator) error { return nil },\n\t\t\twantErr:                                        false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tms := NewMockMutableState(ctrl)\n\t\t\tmtg := NewMockMutableStateTaskGenerator(ctrl)\n\t\t\tms.EXPECT().GetDomainEntry().Return(cache.NewLocalDomainCacheEntryForTest(&persistence.DomainInfo{ID: \"domain-id\"}, nil, \"test\")).AnyTimes()\n\t\t\trefresher := &mutableStateTaskRefresherImpl{\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tWriteVisibilityStoreName:    dynamicproperties.GetStringPropertyFn(constants.VisibilityModeES),\n\t\t\t\t\tWorkflowDeletionJitterRange: dynamicproperties.GetIntPropertyFilteredByDomain(1),\n\t\t\t\t\tIsAdvancedVisConfigExist:    true,\n\t\t\t\t},\n\t\t\t\tnewMutableStateTaskGeneratorFn: func(log.Logger, cluster.Metadata, cache.DomainCache, MutableState) MutableStateTaskGenerator {\n\t\t\t\t\treturn mtg\n\t\t\t\t},\n\t\t\t\trefreshTasksForWorkflowStartFn:                 tc.refreshTasksForWorkflowStartFn,\n\t\t\t\trefreshTasksForWorkflowCloseFn:                 tc.refreshTasksForWorkflowCloseFn,\n\t\t\t\trefreshTasksForRecordWorkflowStartedFn:         tc.refreshTasksForRecordWorkflowStartedFn,\n\t\t\t\trefreshTasksForDecisionFn:                      tc.refreshTasksForDecisionFn,\n\t\t\t\trefreshTasksForActivityFn:                      tc.refreshTasksForActivityFn,\n\t\t\t\trefreshTasksForTimerFn:                         tc.refreshTasksForTimerFn,\n\t\t\t\trefreshTasksForChildWorkflowFn:                 tc.refreshTasksForChildWorkflowFn,\n\t\t\t\trefreshTasksForRequestCancelExternalWorkflowFn: tc.refreshTasksForRequestCancelExternalWorkflowFn,\n\t\t\t\trefreshTasksForSignalExternalWorkflowFn:        tc.refreshTasksForSignalExternalWorkflowFn,\n\t\t\t\trefreshTasksForWorkflowSearchAttrFn:            tc.refreshTasksForWorkflowSearchAttrFn,\n\t\t\t}\n\t\t\terr := refresher.RefreshTasks(context.Background(), time.Now(), ms)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"RefreshTasks err = %v, wantErr %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_util.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// TransactionPolicy is the policy used for updating workflow execution\n\tTransactionPolicy int\n)\n\nconst (\n\t// TransactionPolicyActive updates workflow execution as active\n\tTransactionPolicyActive TransactionPolicy = 0\n\t// TransactionPolicyPassive updates workflow execution as passive\n\tTransactionPolicyPassive TransactionPolicy = 1\n)\n\n// Ptr returns a pointer to the current transaction policy\nfunc (policy TransactionPolicy) Ptr() *TransactionPolicy {\n\treturn &policy\n}\n\nfunc convertSyncActivityInfos(\n\texecutionInfo *persistence.WorkflowExecutionInfo,\n\tactivityInfos map[int64]*persistence.ActivityInfo,\n\tinputs map[int64]struct{},\n) []persistence.Task {\n\toutputs := make([]persistence.Task, 0, len(inputs))\n\tfor item := range inputs {\n\t\tactivityInfo, ok := activityInfos[item]\n\t\tif ok {\n\t\t\t// the visibility timestamp will be set in shard context\n\t\t\toutputs = append(outputs, &persistence.SyncActivityTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\t\t\tRunID:      executionInfo.RunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion: activityInfo.Version,\n\t\t\t\t},\n\t\t\t\tScheduledID: activityInfo.ScheduleID,\n\t\t\t})\n\t\t}\n\t}\n\treturn outputs\n}\n\nfunc convertWorkflowRequests(inputs map[persistence.WorkflowRequest]struct{}) []*persistence.WorkflowRequest {\n\toutputs := make([]*persistence.WorkflowRequest, 0, len(inputs))\n\tfor key := range inputs {\n\t\tkey := key // TODO: remove this trick once we upgrade go to 1.22\n\t\toutputs = append(outputs, &key)\n\t}\n\treturn outputs\n}\n\n// FailDecision fails the current decision task\nfunc FailDecision(\n\tmutableState MutableState,\n\tdecision *DecisionInfo,\n\tdecisionFailureCause types.DecisionTaskFailedCause,\n) error {\n\n\tif _, err := mutableState.AddDecisionTaskFailedEvent(\n\t\tdecision.ScheduleID,\n\t\tdecision.StartedID,\n\t\tdecisionFailureCause,\n\t\tnil,\n\t\tIdentityHistoryService,\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\t0,\n\t\t\"\",\n\t); err != nil {\n\t\treturn err\n\t}\n\n\treturn mutableState.FlushBufferedEvents()\n}\n\n// ScheduleDecision schedules a new decision task\nfunc ScheduleDecision(\n\tmutableState MutableState,\n) error {\n\n\tif mutableState.HasPendingDecision() {\n\t\treturn nil\n\t}\n\n\t_, err := mutableState.AddDecisionTaskScheduledEvent(false)\n\tif err != nil {\n\t\treturn &types.InternalServiceError{Message: \"Failed to add decision scheduled event.\"}\n\t}\n\treturn nil\n}\n\n// FindAutoResetPoint returns the auto reset point\nfunc FindAutoResetPoint(\n\ttimeSource clock.TimeSource,\n\tbadBinaries *types.BadBinaries,\n\tautoResetPoints *types.ResetPoints,\n) (string, *types.ResetPointInfo) {\n\tif badBinaries == nil || badBinaries.Binaries == nil || autoResetPoints == nil || autoResetPoints.Points == nil {\n\t\treturn \"\", nil\n\t}\n\tnowNano := timeSource.Now().UnixNano()\n\tfor _, p := range autoResetPoints.Points {\n\t\tbin, ok := badBinaries.Binaries[p.GetBinaryChecksum()]\n\t\tif ok && p.GetResettable() {\n\t\t\tif p.GetExpiringTimeNano() > 0 && nowNano > p.GetExpiringTimeNano() {\n\t\t\t\t// reset point has expired and we may already deleted the history\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treturn bin.GetReason(), p\n\t\t}\n\t}\n\treturn \"\", nil\n}\n\n// GetChildExecutionDomainName gets domain name for the child workflow\n// NOTE: DomainName in ChildExecutionInfo is being deprecated, and\n// we should always use DomainID field instead.\n// this function exists for backward compatibility reason\nfunc GetChildExecutionDomainName(\n\tchildInfo *persistence.ChildExecutionInfo,\n\tdomainCache cache.DomainCache,\n\tparentDomainEntry *cache.DomainCacheEntry,\n) (string, error) {\n\tif childInfo.DomainID != \"\" {\n\t\treturn domainCache.GetDomainName(childInfo.DomainID)\n\t}\n\n\tif childInfo.DomainNameDEPRECATED != \"\" {\n\t\treturn childInfo.DomainNameDEPRECATED, nil\n\t}\n\n\treturn parentDomainEntry.GetInfo().Name, nil\n}\n\n// GetChildExecutionDomainID gets domainID for the child workflow\n// NOTE: DomainName in ChildExecutionInfo is being deprecated, and\n// we should always use DomainID field instead.\n// this function exists for backward compatibility reason\nfunc GetChildExecutionDomainID(\n\tchildInfo *persistence.ChildExecutionInfo,\n\tdomainCache cache.DomainCache,\n\tparentDomainEntry *cache.DomainCacheEntry,\n) (string, error) {\n\tif childInfo.DomainID != \"\" {\n\t\treturn childInfo.DomainID, nil\n\t}\n\n\tif childInfo.DomainNameDEPRECATED != \"\" {\n\t\treturn domainCache.GetDomainID(childInfo.DomainNameDEPRECATED)\n\t}\n\n\treturn parentDomainEntry.GetInfo().ID, nil\n}\n\nfunc trimBinaryChecksums(recentBinaryChecksums []string, currResetPoints []*types.ResetPointInfo, maxResetPoints int) ([]string, []*types.ResetPointInfo) {\n\tnumResetPoints := len(currResetPoints)\n\tif numResetPoints >= maxResetPoints {\n\t\t// If exceeding the max limit, do rotation by taking the oldest ones out.\n\t\t// startIndex plus one here because it needs to make space for the new binary checksum for the current run\n\t\tstartIndex := numResetPoints - maxResetPoints + 1\n\t\tcurrResetPoints = currResetPoints[startIndex:]\n\t\trecentBinaryChecksums = recentBinaryChecksums[startIndex:]\n\t}\n\n\treturn recentBinaryChecksums, currResetPoints\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_util_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/testing/testdatagen/idlfuzzedtestdata\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestCopyActivityInfo(t *testing.T) {\n\tt.Run(\"test CopyActivityInfo Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.ActivityInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopyActivityInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestCopyWorkflowExecutionInfo(t *testing.T) {\n\tt.Run(\"test ExecutionInfo Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.WorkflowExecutionInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopyWorkflowExecutionInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestCopyTimerInfoMapping(t *testing.T) {\n\tt.Run(\"test Timer info Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.TimerInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopyTimerInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestChildWorkflowMapping(t *testing.T) {\n\tt.Run(\"test child workflwo info Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.ChildExecutionInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopyChildInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestCopySignalInfo(t *testing.T) {\n\tt.Run(\"test signal info Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.SignalInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopySignalInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestCopyCancellationInfo(t *testing.T) {\n\tt.Run(\"test signal info Mapping\", func(t *testing.T) {\n\t\tf := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\n\t\td1 := persistence.RequestCancelInfo{}\n\t\tf.Fuzz(&d1)\n\t\td2 := CopyCancellationInfo(t, &d1)\n\n\t\tassert.Equal(t, &d1, d2)\n\t})\n}\n\nfunc TestFindAutoResetPoint(t *testing.T) {\n\ttimeSource := clock.NewRealTimeSource()\n\n\t// case 1: nil\n\t_, pt := FindAutoResetPoint(timeSource, nil, nil)\n\tassert.Nil(t, pt)\n\n\t// case 2: empty\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{}, &types.ResetPoints{})\n\tassert.Nil(t, pt)\n\n\tpt0 := &types.ResetPointInfo{\n\t\tBinaryChecksum: \"abc\",\n\t\tResettable:     true,\n\t}\n\tpt1 := &types.ResetPointInfo{\n\t\tBinaryChecksum: \"def\",\n\t\tResettable:     true,\n\t}\n\tpt3 := &types.ResetPointInfo{\n\t\tBinaryChecksum: \"ghi\",\n\t\tResettable:     false,\n\t}\n\n\texpiredNowNano := time.Now().UnixNano() - int64(time.Hour)\n\tnotExpiredNowNano := time.Now().UnixNano() + int64(time.Hour)\n\tpt4 := &types.ResetPointInfo{\n\t\tBinaryChecksum:   \"expired\",\n\t\tResettable:       true,\n\t\tExpiringTimeNano: common.Int64Ptr(expiredNowNano),\n\t}\n\n\tpt5 := &types.ResetPointInfo{\n\t\tBinaryChecksum:   \"notExpired\",\n\t\tResettable:       true,\n\t\tExpiringTimeNano: common.Int64Ptr(notExpiredNowNano),\n\t}\n\n\t// case 3: two intersection\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"abc\": {},\n\t\t\t\"def\": {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt0, pt1, pt3,\n\t\t},\n\t})\n\tassert.Equal(t, pt, pt0)\n\n\t// case 4: one intersection\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"none\":    {},\n\t\t\t\"def\":     {},\n\t\t\t\"expired\": {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt4, pt0, pt1, pt3,\n\t\t},\n\t})\n\tassert.Equal(t, pt, pt1)\n\n\t// case 4: no intersection\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"none1\": {},\n\t\t\t\"none2\": {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt0, pt1, pt3,\n\t\t},\n\t})\n\tassert.Nil(t, pt)\n\n\t// case 5: not resettable\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"none1\": {},\n\t\t\t\"ghi\":   {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt0, pt1, pt3,\n\t\t},\n\t})\n\tassert.Nil(t, pt)\n\n\t// case 6: one intersection of expired\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"none\":    {},\n\t\t\t\"expired\": {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt0, pt1, pt3, pt4, pt5,\n\t\t},\n\t})\n\tassert.Nil(t, pt)\n\n\t// case 7: one intersection of not expired\n\t_, pt = FindAutoResetPoint(timeSource, &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"none\":       {},\n\t\t\t\"notExpired\": {},\n\t\t},\n\t}, &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\tpt0, pt1, pt3, pt4, pt5,\n\t\t},\n\t})\n\tassert.Equal(t, pt, pt5)\n}\n\nfunc TestTrimBinaryChecksums(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tmaxResetPoints int\n\t\texpected       []string\n\t}{\n\t\t{\n\t\t\tname:           \"not reach limit\",\n\t\t\tmaxResetPoints: 6,\n\t\t\texpected:       []string{\"checksum1\", \"checksum2\", \"checksum3\", \"checksum4\", \"checksum5\"},\n\t\t},\n\t\t{\n\t\t\tname:           \"reach at limit\",\n\t\t\tmaxResetPoints: 5,\n\t\t\texpected:       []string{\"checksum2\", \"checksum3\", \"checksum4\", \"checksum5\"},\n\t\t},\n\t\t{\n\t\t\tname:           \"exceeds limit\",\n\t\t\tmaxResetPoints: 2,\n\t\t\texpected:       []string{\"checksum5\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcurrResetPoints := []*types.ResetPointInfo{\n\t\t\t\t{\n\t\t\t\t\tBinaryChecksum: \"checksum1\",\n\t\t\t\t\tRunID:          \"run1\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBinaryChecksum: \"checksum2\",\n\t\t\t\t\tRunID:          \"run2\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBinaryChecksum: \"checksum3\",\n\t\t\t\t\tRunID:          \"run3\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBinaryChecksum: \"checksum4\",\n\t\t\t\t\tRunID:          \"run4\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tBinaryChecksum: \"checksum5\",\n\t\t\t\t\tRunID:          \"run5\",\n\t\t\t\t},\n\t\t\t}\n\t\t\tvar recentBinaryChecksums []string\n\t\t\tfor _, rp := range currResetPoints {\n\t\t\t\trecentBinaryChecksums = append(recentBinaryChecksums, rp.GetBinaryChecksum())\n\t\t\t}\n\t\t\trecentBinaryChecksums, currResetPoints = trimBinaryChecksums(recentBinaryChecksums, currResetPoints, tc.maxResetPoints)\n\t\t\tassert.Equal(t, tc.expected, recentBinaryChecksums)\n\t\t\tassert.Equal(t, len(tc.expected), len(currResetPoints))\n\t\t})\n\t}\n\n\t// test empty case\n\tcurrResetPoints := make([]*types.ResetPointInfo, 0, 1)\n\tvar recentBinaryChecksums []string\n\tfor _, rp := range currResetPoints {\n\t\trecentBinaryChecksums = append(recentBinaryChecksums, rp.GetBinaryChecksum())\n\t}\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tt.Errorf(\"The function panicked: %v\", r)\n\t\t}\n\t}()\n\ttrimedBinaryChecksums, trimedResetPoints := trimBinaryChecksums(recentBinaryChecksums, currResetPoints, 2)\n\tassert.Equal(t, recentBinaryChecksums, trimedBinaryChecksums)\n\tassert.Equal(t, currResetPoints, trimedResetPoints)\n}\n\nfunc TestConvertWorkflowRequests(t *testing.T) {\n\tinputs := map[persistence.WorkflowRequest]struct{}{}\n\tinputs[persistence.WorkflowRequest{RequestID: \"aaa\", Version: 1, RequestType: persistence.WorkflowRequestTypeStart}] = struct{}{}\n\tinputs[persistence.WorkflowRequest{RequestID: \"aaa\", Version: 1, RequestType: persistence.WorkflowRequestTypeSignal}] = struct{}{}\n\n\texpectedOutputs := []*persistence.WorkflowRequest{\n\t\t{\n\t\t\tRequestID:   \"aaa\",\n\t\t\tVersion:     1,\n\t\t\tRequestType: persistence.WorkflowRequestTypeStart,\n\t\t},\n\t\t{\n\t\t\tRequestID:   \"aaa\",\n\t\t\tVersion:     1,\n\t\t\tRequestType: persistence.WorkflowRequestTypeSignal,\n\t\t},\n\t}\n\n\tassert.ElementsMatch(t, expectedOutputs, convertWorkflowRequests(inputs))\n}\n\nfunc TestCreatePersistenceMutableState(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockShardContext := shard.NewMockContext(ctrl)\n\tlogger := testlogger.New(t)\n\tmockEventsCache := events.NewMockCache(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShardContext.EXPECT().GetClusterMetadata().Return(constants.TestClusterMetadata).Times(2)\n\tmockShardContext.EXPECT().GetEventsCache().Return(mockEventsCache)\n\tmockShardContext.EXPECT().GetConfig().Return(config.NewForTest())\n\tmockShardContext.EXPECT().GetTimeSource().Return(clock.NewMockedTimeSource())\n\tmockShardContext.EXPECT().GetMetricsClient().Return(metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}))\n\tmockShardContext.EXPECT().GetDomainCache().Return(mockDomainCache)\n\tmockShardContext.EXPECT().GetLogger().Return(logger).AnyTimes()\n\n\tbuilder := newMutableStateBuilder(mockShardContext, logger, constants.TestLocalDomainEntry, constants.TestLocalDomainEntry.GetFailoverVersion())\n\tbuilder.pendingActivityInfoIDs[0] = &persistence.ActivityInfo{}\n\tbuilder.pendingTimerInfoIDs[\"some-key\"] = &persistence.TimerInfo{}\n\tbuilder.pendingSignalInfoIDs[0] = &persistence.SignalInfo{}\n\tbuilder.pendingChildExecutionInfoIDs[0] = &persistence.ChildExecutionInfo{}\n\tbuilder.bufferedEvents = []*types.HistoryEvent{{}}\n\tbuilder.updateBufferedEvents = []*types.HistoryEvent{{}}\n\tbuilder.versionHistories = &persistence.VersionHistories{CurrentVersionHistoryIndex: 0, Histories: []*persistence.VersionHistory{}}\n\tbuilder.pendingRequestCancelInfoIDs[0] = &persistence.RequestCancelInfo{}\n\tbuilder.decisionTaskManager.(*mutableStateDecisionTaskManagerImpl).HasInFlightDecision()\n\tbuilder.executionInfo.DecisionStartedID = 1\n\n\tmutableState := CreatePersistenceMutableState(t, builder)\n\tassert.NotNil(t, mutableState)\n\tassert.Equal(t, builder.executionInfo, mutableState.ExecutionInfo)\n\tassert.Equal(t, builder.pendingActivityInfoIDs, mutableState.ActivityInfos)\n\tassert.Equal(t, builder.pendingSignalInfoIDs, mutableState.SignalInfos)\n\tassert.Equal(t, builder.pendingTimerInfoIDs, mutableState.TimerInfos)\n\tassert.Equal(t, builder.pendingRequestCancelInfoIDs, mutableState.RequestCancelInfos)\n\tassert.Equal(t, len(builder.bufferedEvents)+len(builder.updateBufferedEvents), len(mutableState.BufferedEvents))\n}\n\nfunc TestGetChildExecutionDomainName(t *testing.T) {\n\tt.Run(\"nonempty domain ID\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainID: testdata.DomainID}\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\texpected := testdata.DomainName\n\t\tmockDomainCache.EXPECT().GetDomainName(childInfo.DomainID).Return(expected, nil)\n\t\tname, err := GetChildExecutionDomainName(childInfo, mockDomainCache, constants.TestLocalDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, expected, name)\n\t})\n\n\tt.Run(\"nonempty domain name\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainNameDEPRECATED: testdata.DomainName}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tname, err := GetChildExecutionDomainName(childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, testdata.DomainName, name)\n\t})\n\n\tt.Run(\"empty childInfo\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tname, err := GetChildExecutionDomainName(childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, parentDomainEntry.GetInfo().Name, name)\n\t})\n}\n\nfunc TestGetChildExecutionDomainID(t *testing.T) {\n\tt.Run(\"nonempty domain ID\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainID: testdata.DomainID}\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tname, err := GetChildExecutionDomainID(childInfo, mockDomainCache, constants.TestLocalDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, testdata.DomainID, name)\n\t})\n\n\tt.Run(\"nonempty domain name\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainNameDEPRECATED: testdata.DomainName}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tmockDomainCache.EXPECT().GetDomainID(testdata.DomainName).Return(testdata.DomainID, nil)\n\t\tname, err := GetChildExecutionDomainID(childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, testdata.DomainID, name)\n\t})\n\n\tt.Run(\"empty childInfo\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tname, err := GetChildExecutionDomainID(childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, parentDomainEntry.GetInfo().ID, name)\n\t})\n}\n\nfunc TestGetChildExecutionDomainEntry(t *testing.T) {\n\tt.Run(\"nonempty domain ID\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainID: testdata.DomainID}\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\texpected := &cache.DomainCacheEntry{}\n\t\tmockDomainCache.EXPECT().GetDomainByID(childInfo.DomainID).Return(expected, nil)\n\t\tentry, err := GetChildExecutionDomainEntry(t, childInfo, mockDomainCache, constants.TestLocalDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, expected, entry)\n\t})\n\n\tt.Run(\"nonempty domain name\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{DomainNameDEPRECATED: testdata.DomainName}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\texpected := &cache.DomainCacheEntry{}\n\t\tmockDomainCache.EXPECT().GetDomain(childInfo.DomainNameDEPRECATED).Return(expected, nil)\n\t\tentry, err := GetChildExecutionDomainEntry(t, childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, expected, entry)\n\t})\n\n\tt.Run(\"empty childInfo\", func(t *testing.T) {\n\t\tchildInfo := &persistence.ChildExecutionInfo{}\n\t\tparentDomainEntry := constants.TestLocalDomainEntry\n\t\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\t\tentry, err := GetChildExecutionDomainEntry(t, childInfo, mockDomainCache, parentDomainEntry)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, parentDomainEntry, entry)\n\t})\n}\n\nfunc TestConvert(t *testing.T) {\n\tt.Run(\"convertSyncActivityInfos\", func(t *testing.T) {\n\t\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   \"some-domain-id\",\n\t\t\tWorkflowID: \"some-workflow-id\",\n\t\t\tRunID:      \"some-run-id\",\n\t\t}\n\t\tactivityInfos := map[int64]*persistence.ActivityInfo{1: {Version: 1, ScheduleID: 1}}\n\t\tinputs := map[int64]struct{}{1: {}}\n\t\toutputs := convertSyncActivityInfos(executionInfo, activityInfos, inputs)\n\t\tassert.NotNil(t, outputs)\n\t\tassert.Equal(t, 1, len(outputs))\n\t\tassert.Equal(t, int64(1), outputs[0].(*persistence.SyncActivityTask).ScheduledID)\n\t\tassert.Equal(t, int64(1), outputs[0].GetVersion())\n\t\tassert.Equal(t, executionInfo.DomainID, outputs[0].(*persistence.SyncActivityTask).DomainID)\n\t\tassert.Equal(t, executionInfo.WorkflowID, outputs[0].(*persistence.SyncActivityTask).WorkflowID)\n\t\tassert.Equal(t, executionInfo.RunID, outputs[0].(*persistence.SyncActivityTask).RunID)\n\t})\n}\n\nfunc TestScheduleDecision(t *testing.T) {\n\tt.Run(\"mutable state has pending decision\", func(t *testing.T) {\n\t\tmockMutableState := NewMockMutableState(gomock.NewController(t))\n\t\tmockMutableState.EXPECT().HasPendingDecision().Return(true)\n\t\terr := ScheduleDecision(mockMutableState)\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"internal service error\", func(t *testing.T) {\n\t\tmockMutableState := NewMockMutableState(gomock.NewController(t))\n\t\tmockMutableState.EXPECT().HasPendingDecision().Return(false)\n\t\tmockMutableState.EXPECT().AddDecisionTaskScheduledEvent(false).Return(nil, errors.New(\"some error\"))\n\t\terr := ScheduleDecision(mockMutableState)\n\t\tassert.NotNil(t, err)\n\t\tassert.Equal(t, \"Failed to add decision scheduled event.\", err.Error())\n\t})\n\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmockMutableState := NewMockMutableState(gomock.NewController(t))\n\t\tmockMutableState.EXPECT().HasPendingDecision().Return(false)\n\t\tmockMutableState.EXPECT().AddDecisionTaskScheduledEvent(false).Return(nil, nil)\n\t\terr := ScheduleDecision(mockMutableState)\n\t\trequire.NoError(t, err)\n\t})\n}\n\nfunc TestFailDecision(t *testing.T) {\n\tt.Run(\"success\", func(t *testing.T) {\n\t\tmockMutableState := NewMockMutableState(gomock.NewController(t))\n\t\tdecision := &DecisionInfo{}\n\t\tfailureCause := new(types.DecisionTaskFailedCause)\n\t\tmockMutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\tdecision.ScheduleID,\n\t\t\tdecision.StartedID,\n\t\t\t*failureCause,\n\t\t\tnil,\n\t\t\tIdentityHistoryService,\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\tint64(0),\n\t\t\t\"\",\n\t\t).Return(nil, nil)\n\t\tmockMutableState.EXPECT().FlushBufferedEvents() // only on success\n\t\terr := FailDecision(mockMutableState, decision, *failureCause)\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"failure\", func(t *testing.T) {\n\t\tmockMutableState := NewMockMutableState(gomock.NewController(t))\n\t\tdecision := &DecisionInfo{}\n\t\tfailureCause := new(types.DecisionTaskFailedCause)\n\t\tmockMutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\tdecision.ScheduleID,\n\t\t\tdecision.StartedID,\n\t\t\t*failureCause,\n\t\t\tnil,\n\t\t\tIdentityHistoryService,\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\t\"\",\n\t\t\tint64(0),\n\t\t\t\"\",\n\t\t).Return(nil, errors.New(\"some error adding failed event\"))\n\t\terr := FailDecision(mockMutableState, decision, *failureCause)\n\t\tassert.NotNil(t, err)\n\t\tassert.Equal(t, \"some error adding failed event\", err.Error())\n\t})\n}\n"
  },
  {
    "path": "service/history/execution/mutable_state_util_test_helpers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage execution\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"golang.org/x/exp/slices\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// CreatePersistenceMutableState creates a persistence mutable state based on the its in-memory version\nfunc CreatePersistenceMutableState(t *testing.T, ms MutableState) *persistence.WorkflowMutableState {\n\tbuilder := ms.(*mutableStateBuilder)\n\tbuilder.FlushBufferedEvents() //nolint:errcheck\n\tinfo := CopyWorkflowExecutionInfo(t, builder.GetExecutionInfo())\n\tstats := &persistence.ExecutionStats{}\n\tactivityInfos := make(map[int64]*persistence.ActivityInfo)\n\tfor id, info := range builder.GetPendingActivityInfos() {\n\t\tactivityInfos[id] = CopyActivityInfo(t, info)\n\t}\n\ttimerInfos := make(map[string]*persistence.TimerInfo)\n\tfor id, info := range builder.GetPendingTimerInfos() {\n\t\ttimerInfos[id] = CopyTimerInfo(t, info)\n\t}\n\tcancellationInfos := make(map[int64]*persistence.RequestCancelInfo)\n\tfor id, info := range builder.GetPendingRequestCancelExternalInfos() {\n\t\tcancellationInfos[id] = CopyCancellationInfo(t, info)\n\t}\n\tsignalInfos := make(map[int64]*persistence.SignalInfo)\n\tfor id, info := range builder.GetPendingSignalExternalInfos() {\n\t\tsignalInfos[id] = CopySignalInfo(t, info)\n\t}\n\tchildInfos := make(map[int64]*persistence.ChildExecutionInfo)\n\tfor id, info := range builder.GetPendingChildExecutionInfos() {\n\t\tchildInfos[id] = CopyChildInfo(t, info)\n\t}\n\n\tbuilder.FlushBufferedEvents() //nolint:errcheck\n\tvar bufferedEvents []*types.HistoryEvent\n\tif len(builder.bufferedEvents) > 0 {\n\t\tbufferedEvents = append(bufferedEvents, builder.bufferedEvents...)\n\t}\n\tif len(builder.updateBufferedEvents) > 0 {\n\t\tbufferedEvents = append(bufferedEvents, builder.updateBufferedEvents...)\n\t}\n\n\tvar versionHistories *persistence.VersionHistories\n\tif ms.GetVersionHistories() != nil {\n\t\tversionHistories = ms.GetVersionHistories().Duplicate()\n\t}\n\treturn &persistence.WorkflowMutableState{\n\t\tExecutionInfo:       info,\n\t\tExecutionStats:      stats,\n\t\tActivityInfos:       activityInfos,\n\t\tTimerInfos:          timerInfos,\n\t\tBufferedEvents:      bufferedEvents,\n\t\tSignalInfos:         signalInfos,\n\t\tRequestCancelInfos:  cancellationInfos,\n\t\tChildExecutionInfos: childInfos,\n\t\tVersionHistories:    versionHistories,\n\t}\n}\n\n// CopyWorkflowExecutionInfo copies WorkflowExecutionInfo\nfunc CopyWorkflowExecutionInfo(t *testing.T, sourceInfo *persistence.WorkflowExecutionInfo) *persistence.WorkflowExecutionInfo {\n\treturn &persistence.WorkflowExecutionInfo{\n\t\tDomainID:                           sourceInfo.DomainID,\n\t\tWorkflowID:                         sourceInfo.WorkflowID,\n\t\tRunID:                              sourceInfo.RunID,\n\t\tFirstExecutionRunID:                sourceInfo.FirstExecutionRunID,\n\t\tParentDomainID:                     sourceInfo.ParentDomainID,\n\t\tParentWorkflowID:                   sourceInfo.ParentWorkflowID,\n\t\tParentRunID:                        sourceInfo.ParentRunID,\n\t\tIsCron:                             sourceInfo.IsCron,\n\t\tInitiatedID:                        sourceInfo.InitiatedID,\n\t\tCompletionEventBatchID:             sourceInfo.CompletionEventBatchID,\n\t\tCompletionEvent:                    sourceInfo.CompletionEvent,\n\t\tTaskList:                           sourceInfo.TaskList,\n\t\tTaskListKind:                       sourceInfo.TaskListKind,\n\t\tStickyTaskList:                     sourceInfo.StickyTaskList,\n\t\tStickyScheduleToStartTimeout:       sourceInfo.StickyScheduleToStartTimeout,\n\t\tWorkflowTypeName:                   sourceInfo.WorkflowTypeName,\n\t\tWorkflowTimeout:                    sourceInfo.WorkflowTimeout,\n\t\tDecisionStartToCloseTimeout:        sourceInfo.DecisionStartToCloseTimeout,\n\t\tExecutionContext:                   sourceInfo.ExecutionContext,\n\t\tState:                              sourceInfo.State,\n\t\tCloseStatus:                        sourceInfo.CloseStatus,\n\t\tLastFirstEventID:                   sourceInfo.LastFirstEventID,\n\t\tLastEventTaskID:                    sourceInfo.LastEventTaskID,\n\t\tNextEventID:                        sourceInfo.NextEventID,\n\t\tLastProcessedEvent:                 sourceInfo.LastProcessedEvent,\n\t\tStartTimestamp:                     sourceInfo.StartTimestamp,\n\t\tLastUpdatedTimestamp:               sourceInfo.LastUpdatedTimestamp,\n\t\tCreateRequestID:                    sourceInfo.CreateRequestID,\n\t\tSignalCount:                        sourceInfo.SignalCount,\n\t\tDecisionVersion:                    sourceInfo.DecisionVersion,\n\t\tDecisionScheduleID:                 sourceInfo.DecisionScheduleID,\n\t\tDecisionStartedID:                  sourceInfo.DecisionStartedID,\n\t\tDecisionRequestID:                  sourceInfo.DecisionRequestID,\n\t\tDecisionTimeout:                    sourceInfo.DecisionTimeout,\n\t\tDecisionAttempt:                    sourceInfo.DecisionAttempt,\n\t\tDecisionScheduledTimestamp:         sourceInfo.DecisionScheduledTimestamp,\n\t\tDecisionStartedTimestamp:           sourceInfo.DecisionStartedTimestamp,\n\t\tDecisionOriginalScheduledTimestamp: sourceInfo.DecisionOriginalScheduledTimestamp,\n\t\tCancelRequested:                    sourceInfo.CancelRequested,\n\t\tCancelRequestID:                    sourceInfo.CancelRequestID,\n\t\tCronSchedule:                       sourceInfo.CronSchedule,\n\t\tClientLibraryVersion:               sourceInfo.ClientLibraryVersion,\n\t\tClientFeatureVersion:               sourceInfo.ClientFeatureVersion,\n\t\tClientImpl:                         sourceInfo.ClientImpl,\n\t\tAutoResetPoints:                    sourceInfo.AutoResetPoints,\n\t\tMemo:                               sourceInfo.Memo,\n\t\tSearchAttributes:                   sourceInfo.SearchAttributes,\n\t\tPartitionConfig:                    sourceInfo.PartitionConfig,\n\t\tExecutionStatus:                    sourceInfo.ExecutionStatus,\n\t\tScheduledExecutionTimestamp:        sourceInfo.ScheduledExecutionTimestamp,\n\t\tAttempt:                            sourceInfo.Attempt,\n\t\tHasRetryPolicy:                     sourceInfo.HasRetryPolicy,\n\t\tInitialInterval:                    sourceInfo.InitialInterval,\n\t\tBackoffCoefficient:                 sourceInfo.BackoffCoefficient,\n\t\tMaximumInterval:                    sourceInfo.MaximumInterval,\n\t\tExpirationTime:                     sourceInfo.ExpirationTime,\n\t\tMaximumAttempts:                    sourceInfo.MaximumAttempts,\n\t\tNonRetriableErrors:                 sourceInfo.NonRetriableErrors,\n\t\tBranchToken:                        sourceInfo.BranchToken,\n\t\tExpirationSeconds:                  sourceInfo.ExpirationSeconds,\n\t\tCronOverlapPolicy:                  sourceInfo.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:       sourceInfo.ActiveClusterSelectionPolicy,\n\t}\n}\n\n// CopyActivityInfo copies ActivityInfo\nfunc CopyActivityInfo(t *testing.T, sourceInfo *persistence.ActivityInfo) *persistence.ActivityInfo {\n\tdetails := slices.Clone(sourceInfo.Details)\n\n\treturn &persistence.ActivityInfo{\n\t\tVersion:                  sourceInfo.Version,\n\t\tScheduleID:               sourceInfo.ScheduleID,\n\t\tScheduledEventBatchID:    sourceInfo.ScheduledEventBatchID,\n\t\tScheduledEvent:           deepCopyHistoryEvent(t, sourceInfo.ScheduledEvent),\n\t\tStartedID:                sourceInfo.StartedID,\n\t\tStartedEvent:             deepCopyHistoryEvent(t, sourceInfo.StartedEvent),\n\t\tActivityID:               sourceInfo.ActivityID,\n\t\tRequestID:                sourceInfo.RequestID,\n\t\tDetails:                  details,\n\t\tScheduledTime:            sourceInfo.ScheduledTime,\n\t\tStartedTime:              sourceInfo.StartedTime,\n\t\tScheduleToStartTimeout:   sourceInfo.ScheduleToStartTimeout,\n\t\tScheduleToCloseTimeout:   sourceInfo.ScheduleToCloseTimeout,\n\t\tStartToCloseTimeout:      sourceInfo.StartToCloseTimeout,\n\t\tHeartbeatTimeout:         sourceInfo.HeartbeatTimeout,\n\t\tLastHeartBeatUpdatedTime: sourceInfo.LastHeartBeatUpdatedTime,\n\t\tCancelRequested:          sourceInfo.CancelRequested,\n\t\tCancelRequestID:          sourceInfo.CancelRequestID,\n\t\tTimerTaskStatus:          sourceInfo.TimerTaskStatus,\n\t\tAttempt:                  sourceInfo.Attempt,\n\t\tDomainID:                 sourceInfo.DomainID,\n\t\tStartedIdentity:          sourceInfo.StartedIdentity,\n\t\tTaskList:                 sourceInfo.TaskList,\n\t\tTaskListKind:             sourceInfo.TaskListKind,\n\t\tHasRetryPolicy:           sourceInfo.HasRetryPolicy,\n\t\tInitialInterval:          sourceInfo.InitialInterval,\n\t\tBackoffCoefficient:       sourceInfo.BackoffCoefficient,\n\t\tMaximumInterval:          sourceInfo.MaximumInterval,\n\t\tExpirationTime:           sourceInfo.ExpirationTime,\n\t\tMaximumAttempts:          sourceInfo.MaximumAttempts,\n\t\tNonRetriableErrors:       sourceInfo.NonRetriableErrors,\n\t\tLastFailureReason:        sourceInfo.LastFailureReason,\n\t\tLastWorkerIdentity:       sourceInfo.LastWorkerIdentity,\n\t\tLastFailureDetails:       sourceInfo.LastFailureDetails,\n\t\t// Not written to database - This is used only for deduping heartbeat timer creation\n\t\tLastHeartbeatTimeoutVisibilityInSeconds: sourceInfo.LastHeartbeatTimeoutVisibilityInSeconds,\n\t}\n}\n\n// CopyTimerInfo copies TimerInfo\nfunc CopyTimerInfo(t *testing.T, sourceInfo *persistence.TimerInfo) *persistence.TimerInfo {\n\treturn &persistence.TimerInfo{\n\t\tVersion:    sourceInfo.Version,\n\t\tTimerID:    sourceInfo.TimerID,\n\t\tStartedID:  sourceInfo.StartedID,\n\t\tExpiryTime: sourceInfo.ExpiryTime,\n\t\tTaskStatus: sourceInfo.TaskStatus,\n\t}\n}\n\n// CopyCancellationInfo copies RequestCancelInfo\nfunc CopyCancellationInfo(t *testing.T, sourceInfo *persistence.RequestCancelInfo) *persistence.RequestCancelInfo {\n\treturn &persistence.RequestCancelInfo{\n\t\tVersion:               sourceInfo.Version,\n\t\tInitiatedID:           sourceInfo.InitiatedID,\n\t\tInitiatedEventBatchID: sourceInfo.InitiatedEventBatchID,\n\t\tCancelRequestID:       sourceInfo.CancelRequestID,\n\t}\n}\n\n// CopySignalInfo copies SignalInfo\nfunc CopySignalInfo(t *testing.T, sourceInfo *persistence.SignalInfo) *persistence.SignalInfo {\n\treturn &persistence.SignalInfo{\n\t\tVersion:               sourceInfo.Version,\n\t\tInitiatedEventBatchID: sourceInfo.InitiatedEventBatchID,\n\t\tInitiatedID:           sourceInfo.InitiatedID,\n\t\tSignalRequestID:       sourceInfo.SignalRequestID,\n\t\tSignalName:            sourceInfo.SignalName,\n\t\tInput:                 slices.Clone(sourceInfo.Input),\n\t\tControl:               slices.Clone(sourceInfo.Control),\n\t}\n}\n\n// CopyChildInfo copies ChildExecutionInfo\nfunc CopyChildInfo(t *testing.T, sourceInfo *persistence.ChildExecutionInfo) *persistence.ChildExecutionInfo {\n\treturn &persistence.ChildExecutionInfo{\n\t\tVersion:               sourceInfo.Version,\n\t\tInitiatedID:           sourceInfo.InitiatedID,\n\t\tInitiatedEventBatchID: sourceInfo.InitiatedEventBatchID,\n\t\tStartedID:             sourceInfo.StartedID,\n\t\tStartedWorkflowID:     sourceInfo.StartedWorkflowID,\n\t\tStartedRunID:          sourceInfo.StartedRunID,\n\t\tCreateRequestID:       sourceInfo.CreateRequestID,\n\t\tDomainID:              sourceInfo.DomainID,\n\t\tDomainNameDEPRECATED:  sourceInfo.DomainNameDEPRECATED,\n\t\tWorkflowTypeName:      sourceInfo.WorkflowTypeName,\n\t\tParentClosePolicy:     sourceInfo.ParentClosePolicy,\n\t\tInitiatedEvent:        deepCopyHistoryEvent(t, sourceInfo.InitiatedEvent),\n\t\tStartedEvent:          deepCopyHistoryEvent(t, sourceInfo.StartedEvent),\n\t}\n}\n\nfunc deepCopyHistoryEvent(t *testing.T, e *types.HistoryEvent) *types.HistoryEvent {\n\tif e == nil {\n\t\treturn nil\n\t}\n\tbytes, err := json.Marshal(e)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tvar copy types.HistoryEvent\n\terr = json.Unmarshal(bytes, &copy)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn &copy\n}\n\n// GetChildExecutionDomainEntry get domain entry for the child workflow\n// NOTE: DomainName in ChildExecutionInfo is being deprecated, and\n// we should always use DomainID field instead.\n// this function exists for backward compatibility reason\nfunc GetChildExecutionDomainEntry(\n\tt *testing.T,\n\tchildInfo *persistence.ChildExecutionInfo,\n\tdomainCache cache.DomainCache,\n\tparentDomainEntry *cache.DomainCacheEntry,\n) (*cache.DomainCacheEntry, error) {\n\tif childInfo.DomainID != \"\" {\n\t\treturn domainCache.GetDomainByID(childInfo.DomainID)\n\t}\n\n\tif childInfo.DomainNameDEPRECATED != \"\" {\n\t\treturn domainCache.GetDomain(childInfo.DomainNameDEPRECATED)\n\t}\n\n\treturn parentDomainEntry, nil\n}\n"
  },
  {
    "path": "service/history/execution/retry.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"math\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n)\n\nfunc getBackoffInterval(\n\tnow time.Time,\n\texpirationTime time.Time,\n\tcurrAttempt int32,\n\tmaxAttempts int32,\n\tinitInterval int32,\n\tmaxInterval int32,\n\tbackoffCoefficient float64,\n\tfailureReason string,\n\tnonRetriableErrors []string,\n) time.Duration {\n\n\tif maxAttempts == 0 && expirationTime.IsZero() {\n\t\treturn backoff.NoBackoff\n\t}\n\n\tif maxAttempts > 0 && currAttempt >= maxAttempts-1 {\n\t\t// currAttempt starts from 0.\n\t\t// MaximumAttempts is the total attempts, including initial (non-retry) attempt.\n\t\treturn backoff.NoBackoff\n\t}\n\n\tnextInterval := int64(float64(initInterval) * math.Pow(backoffCoefficient, float64(currAttempt)))\n\tif nextInterval <= 0 {\n\t\t// math.Pow() could overflow\n\t\tif maxInterval > 0 {\n\t\t\tnextInterval = int64(maxInterval)\n\t\t} else {\n\t\t\treturn backoff.NoBackoff\n\t\t}\n\t}\n\n\tif maxInterval > 0 && nextInterval > int64(maxInterval) {\n\t\t// cap next interval to MaxInterval\n\t\tnextInterval = int64(maxInterval)\n\t}\n\n\tbackoffInterval := time.Duration(nextInterval) * time.Second\n\tnextScheduleTime := now.Add(backoffInterval)\n\tif !expirationTime.IsZero() && nextScheduleTime.After(expirationTime) {\n\t\treturn backoff.NoBackoff\n\t}\n\n\t// make sure we don't retry size exceeded error reasons. Note that FailureReasonFailureDetailsExceedsLimit is retryable.\n\tif failureReason == common.FailureReasonCancelDetailsExceedsLimit ||\n\t\tfailureReason == common.FailureReasonCompleteResultExceedsLimit ||\n\t\tfailureReason == common.FailureReasonHeartbeatExceedsLimit ||\n\t\tfailureReason == common.FailureReasonDecisionBlobSizeExceedsLimit {\n\t\treturn backoff.NoBackoff\n\t}\n\n\t// check if error is non-retriable\n\tfor _, er := range nonRetriableErrors {\n\t\tif er == failureReason {\n\t\t\treturn backoff.NoBackoff\n\t\t}\n\t}\n\n\treturn backoffInterval\n}\n"
  },
  {
    "path": "service/history/execution/retry_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc Test_NextRetry2(t *testing.T) {\n\t// a := assert.New(t)\n\tnow, _ := time.Parse(time.RFC3339, \"2018-04-13T16:08:08+00:00\")\n\treason := \"good-reason\"\n\tidentity := \"some-worker-identity\"\n\n\t// no retry without retry policy\n\tai := &persistence.ActivityInfo{\n\t\tScheduleToStartTimeout: int32((30 * time.Minute).Seconds()),\n\t\tScheduleToCloseTimeout: int32((30 * time.Minute).Seconds()),\n\t\tStartToCloseTimeout:    int32((30 * time.Minute).Seconds()),\n\t\tHasRetryPolicy:         true,\n\t\tNonRetriableErrors:     []string{\"bad-reason\", \"ugly-reason\"},\n\t\tStartedIdentity:        identity,\n\t\tMaximumAttempts:        0,\n\t\tInitialInterval:        60,\n\t\tBackoffCoefficient:     1,\n\t\tMaximumInterval:        6000,\n\t\tExpirationTime:         now.Add(86400 * time.Second),\n\t\tAttempt:                5,\n\t}\n\n\tdur := getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t)\n\n\tt.Logf(\"dur: %v\", dur)\n}\n\nfunc Test_NextRetry(t *testing.T) {\n\ta := assert.New(t)\n\tnow, _ := time.Parse(time.RFC3339, \"2018-04-13T16:08:08+00:00\")\n\treason := \"good-reason\"\n\tidentity := \"some-worker-identity\"\n\n\t// no retry without retry policy\n\tai := &persistence.ActivityInfo{\n\t\tScheduleToStartTimeout: 5,\n\t\tScheduleToCloseTimeout: 30,\n\t\tStartToCloseTimeout:    25,\n\t\tHasRetryPolicy:         false,\n\t\tNonRetriableErrors:     []string{\"bad-reason\", \"ugly-reason\"},\n\t\tStartedIdentity:        identity,\n\t}\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tclock.NewRealTimeSource().Now(),\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// no retry if cancel requested\n\tai.HasRetryPolicy = true\n\tai.CancelRequested = true\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tclock.NewRealTimeSource().Now(),\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// no retry if both MaximumAttempts and ExpirationTime are not set\n\tai.CancelRequested = false\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tclock.NewRealTimeSource().Now(),\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// no retry if MaximumAttempts is 1 (for initial attempt)\n\tai.InitialInterval = 1\n\tai.MaximumAttempts = 1\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tclock.NewRealTimeSource().Now(),\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// backoff retry, intervals: 1s, 2s, 4s, 8s.\n\tai.MaximumAttempts = 5\n\tai.BackoffCoefficient = 2\n\ta.Equal(time.Second, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\ta.Equal(time.Second*2, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\ta.Equal(time.Second*4, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\t// test non-retriable error\n\treason = \"bad-reason\"\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\treason = \"good-reason\"\n\n\ta.Equal(time.Second*8, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\t// no retry as max attempt reached\n\ta.Equal(ai.MaximumAttempts-1, ai.Attempt)\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// increase max attempts, with max interval cap at 10s\n\tai.MaximumAttempts = 6\n\tai.MaximumInterval = 10\n\ta.Equal(time.Second*10, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\t// no retry because expiration time before next interval\n\tai.MaximumAttempts = 8\n\tai.ExpirationTime = now.Add(time.Second * 5)\n\ta.Equal(backoff.NoBackoff, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\n\t// extend expiration, next interval should be 10s\n\tai.ExpirationTime = now.Add(time.Minute)\n\ta.Equal(time.Second*10, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n\n\t// with big max retry, math.Pow() could overflow, verify that it uses the MaxInterval\n\tai.Attempt = 64\n\tai.MaximumAttempts = 100\n\ta.Equal(time.Second*10, getBackoffInterval(\n\t\tnow,\n\t\tai.ExpirationTime,\n\t\tai.Attempt,\n\t\tai.MaximumAttempts,\n\t\tai.InitialInterval,\n\t\tai.MaximumInterval,\n\t\tai.BackoffCoefficient,\n\t\treason,\n\t\tai.NonRetriableErrors,\n\t))\n\tai.Attempt++\n}\n"
  },
  {
    "path": "service/history/execution/state_builder.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination state_builder_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\t// StateBuilder is the mutable state builder\n\tStateBuilder interface {\n\t\tApplyEvents(\n\t\t\tdomainID string,\n\t\t\trequestID string,\n\t\t\tworkflowExecution types.WorkflowExecution,\n\t\t\thistory []*types.HistoryEvent,\n\t\t\tnewRunHistory []*types.HistoryEvent,\n\t\t) (MutableState, error)\n\n\t\tGetMutableState() MutableState\n\t}\n\n\tstateBuilderImpl struct {\n\t\tshard           shard.Context\n\t\tclusterMetadata cluster.Metadata\n\t\tdomainCache     cache.DomainCache\n\t\tlogger          log.Logger\n\n\t\tmutableState MutableState\n\t}\n)\n\nconst (\n\terrMessageHistorySizeZero = \"encounter history size being zero\"\n)\n\nvar _ StateBuilder = (*stateBuilderImpl)(nil)\n\n// NewStateBuilder creates a state builder\nfunc NewStateBuilder(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tmutableState MutableState,\n) StateBuilder {\n\n\treturn &stateBuilderImpl{\n\t\tshard:           shard,\n\t\tclusterMetadata: shard.GetService().GetClusterMetadata(),\n\t\tdomainCache:     shard.GetDomainCache(),\n\t\tlogger:          logger,\n\t\tmutableState:    mutableState,\n\t}\n}\n\nfunc (b *stateBuilderImpl) ApplyEvents(\n\tdomainID string,\n\trequestID string,\n\tworkflowExecution types.WorkflowExecution,\n\thistory []*types.HistoryEvent,\n\tnewRunHistory []*types.HistoryEvent,\n) (MutableState, error) {\n\n\tif len(history) == 0 {\n\t\treturn nil, errors.NewInternalFailureError(errMessageHistorySizeZero)\n\t}\n\tfirstEvent := history[0]\n\tlastEvent := history[len(history)-1]\n\tvar newRunMutableStateBuilder MutableState\n\n\tb.logger.Debugf(\"stateBuilderImpl Applying events for domain %s, wfID %v, first event [id:%v, version:%v], last event [id:%v, version:%v]\",\n\t\tdomainID, workflowExecution.WorkflowID, firstEvent.ID, firstEvent.Version, lastEvent.ID, lastEvent.Version)\n\n\t// need to clear the stickiness since workflow turned to passive\n\tb.mutableState.ClearStickyness()\n\n\thistoryLength := len(history)\n\tfor i, event := range history {\n\t\tb.logger.Debugf(\"stateBuilderImpl Applying event %v of %v. Calling UpdateCurrentVersion for domain %s, wfID %v, event [id:%v, version:%v]\",\n\t\t\ti+1, historyLength, domainID, workflowExecution.WorkflowID, event.ID, event.Version)\n\n\t\t// NOTE: stateBuilder is also being used in the active side\n\t\tif err := b.mutableState.UpdateCurrentVersion(event.Version, true); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tversionHistories := b.mutableState.GetVersionHistories()\n\t\tif versionHistories == nil {\n\t\t\treturn nil, ErrMissingVersionHistories\n\t\t}\n\t\tversionHistory, err := versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tif err := versionHistory.AddOrUpdateItem(persistence.NewVersionHistoryItem(\n\t\t\tevent.ID,\n\t\t\tevent.Version,\n\t\t)); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tb.mutableState.GetExecutionInfo().LastEventTaskID = event.TaskID\n\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionStarted:\n\t\t\tattributes := event.WorkflowExecutionStartedEventAttributes\n\t\t\tvar parentDomainID *string\n\t\t\t// If ParentWorkflowDomainID is present use it, otherwise fallback to ParentWorkflowDomain\n\t\t\t// as ParentWorkflowDomainID will not be present on older histories.\n\t\t\tif attributes.ParentWorkflowDomainID != nil {\n\t\t\t\tparentDomainID = attributes.ParentWorkflowDomainID\n\t\t\t} else if attributes.GetParentWorkflowDomain() != \"\" {\n\t\t\t\tparentDomainEntry, err := b.domainCache.GetDomain(\n\t\t\t\t\tattributes.GetParentWorkflowDomain(),\n\t\t\t\t)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\tparentDomainID = &parentDomainEntry.GetInfo().ID\n\t\t\t}\n\n\t\t\tb.logger.Debug(\"stateBuilderImpl calling ReplicateWorkflowExecutionStartedEvent\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\t\ttag.Dynamic(\"attributes.activecluster-sel-policy-nil\", attributes.ActiveClusterSelectionPolicy == nil),\n\t\t\t)\n\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionStartedEvent(\n\t\t\t\tparentDomainID,\n\t\t\t\tworkflowExecution,\n\t\t\t\trequestID,\n\t\t\t\tevent,\n\t\t\t\ttrue,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tif err := b.mutableState.SetHistoryTree(\n\t\t\t\tworkflowExecution.GetRunID(),\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeDecisionTaskScheduled:\n\t\t\tattributes := event.DecisionTaskScheduledEventAttributes\n\t\t\t// use event.GetTimestamp() as DecisionOriginalScheduledTimestamp, because the heartbeat is not happening here.\n\t\t\t_, err := b.mutableState.ReplicateDecisionTaskScheduledEvent(\n\t\t\t\tevent.Version,\n\t\t\t\tevent.ID,\n\t\t\t\tattributes.TaskList.GetName(),\n\t\t\t\tattributes.GetStartToCloseTimeoutSeconds(),\n\t\t\t\tattributes.GetAttempt(),\n\t\t\t\tevent.GetTimestamp(),\n\t\t\t\tevent.GetTimestamp(),\n\t\t\t\tfalse,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeDecisionTaskStarted:\n\t\t\tattributes := event.DecisionTaskStartedEventAttributes\n\t\t\t_, err := b.mutableState.ReplicateDecisionTaskStartedEvent(\n\t\t\t\tnil,\n\t\t\t\tevent.Version,\n\t\t\t\tattributes.GetScheduledEventID(),\n\t\t\t\tevent.ID,\n\t\t\t\tattributes.GetRequestID(),\n\t\t\t\tevent.GetTimestamp(),\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeDecisionTaskCompleted:\n\t\t\tif err := b.mutableState.ReplicateDecisionTaskCompletedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeDecisionTaskTimedOut:\n\t\t\tif err := b.mutableState.ReplicateDecisionTaskTimedOutEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\t// this is for transient decision\n\t\t\terr := b.mutableState.ReplicateTransientDecisionTaskScheduled()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeDecisionTaskFailed:\n\t\t\tif err := b.mutableState.ReplicateDecisionTaskFailedEvent(event); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\t// this is for transient decision\n\t\t\terr := b.mutableState.ReplicateTransientDecisionTaskScheduled()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskScheduled:\n\t\t\tif _, err := b.mutableState.ReplicateActivityTaskScheduledEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t\tfalse,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskStarted:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskStartedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskCompleted:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskCompletedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskFailed:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskFailedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskTimedOut:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskTimedOutEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskCancelRequested:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskCancelRequestedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeActivityTaskCanceled:\n\t\t\tif err := b.mutableState.ReplicateActivityTaskCanceledEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeRequestCancelActivityTaskFailed:\n\t\t\t// No mutable state action is needed\n\n\t\tcase types.EventTypeTimerStarted:\n\t\t\tif _, err := b.mutableState.ReplicateTimerStartedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeTimerFired:\n\t\t\tif err := b.mutableState.ReplicateTimerFiredEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeTimerCanceled:\n\t\t\tif err := b.mutableState.ReplicateTimerCanceledEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeCancelTimerFailed:\n\t\t\t// no mutable state action is needed\n\n\t\tcase types.EventTypeStartChildWorkflowExecutionInitiated:\n\t\t\tif _, err := b.mutableState.ReplicateStartChildWorkflowExecutionInitiatedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t\t// create a new request ID which is used by transfer queue processor\n\t\t\t\t// if domain is failed over at this point\n\t\t\t\tuuid.New(),\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeStartChildWorkflowExecutionFailed:\n\t\t\tif err := b.mutableState.ReplicateStartChildWorkflowExecutionFailedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionStartedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionCompletedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionFailedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionCanceledEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionTimedOutEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\t\t\tif err := b.mutableState.ReplicateChildWorkflowExecutionTerminatedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\t\tif _, err := b.mutableState.ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t\t// create a new request ID which is used by transfer queue processor\n\t\t\t\t// if domain is failed over at this point\n\t\t\t\tuuid.New(),\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\t\tif err := b.mutableState.ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeExternalWorkflowExecutionCancelRequested:\n\t\t\tif err := b.mutableState.ReplicateExternalWorkflowExecutionCancelRequested(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\t\t// Create a new request ID which is used by transfer queue processor if domain is failed over at this point\n\t\t\tsignalRequestID := uuid.New()\n\t\t\tif _, err := b.mutableState.ReplicateSignalExternalWorkflowExecutionInitiatedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t\tsignalRequestID,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\t\tif err := b.mutableState.ReplicateSignalExternalWorkflowExecutionFailedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeExternalWorkflowExecutionSignaled:\n\t\t\tif err := b.mutableState.ReplicateExternalWorkflowExecutionSignaled(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeMarkerRecorded:\n\t\t\t// No mutable state action is needed\n\n\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionSignaled(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionCancelRequested:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionCancelRequestedEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeUpsertWorkflowSearchAttributes:\n\t\t\tif err := b.mutableState.ReplicateUpsertWorkflowSearchAttributesEvent(\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionCompletedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionFailed:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionFailedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionTimedoutEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionCanceledEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\t\tif err := b.mutableState.ReplicateWorkflowExecutionTerminatedEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tevent,\n\t\t\t); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\n\t\t\t// The length of newRunHistory can be zero in resend case\n\t\t\tif len(newRunHistory) != 0 {\n\t\t\t\tdomainEntry := b.mutableState.GetDomainEntry()\n\t\t\t\tnewRunMutableStateBuilder = NewMutableStateBuilderWithVersionHistories(\n\t\t\t\t\tb.shard,\n\t\t\t\t\tb.logger,\n\t\t\t\t\tdomainEntry,\n\t\t\t\t\tconstants.EmptyVersion,\n\t\t\t\t)\n\t\t\t\tnewRunStateBuilder := NewStateBuilder(b.shard, b.logger, newRunMutableStateBuilder)\n\t\t\t\tnewRunID := event.WorkflowExecutionContinuedAsNewEventAttributes.GetNewExecutionRunID()\n\t\t\t\tnewExecution := types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: workflowExecution.WorkflowID,\n\t\t\t\t\tRunID:      newRunID,\n\t\t\t\t}\n\t\t\t\t_, err := newRunStateBuilder.ApplyEvents(\n\t\t\t\t\tdomainID,\n\t\t\t\t\tuuid.New(),\n\t\t\t\t\tnewExecution,\n\t\t\t\t\tnewRunHistory,\n\t\t\t\t\tnil,\n\t\t\t\t)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tb.logger.Debug(\"stateBuilderImpl calling ReplicateWorkflowExecutionContinuedAsNewEvent\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(workflowExecution.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(workflowExecution.RunID),\n\t\t\t\ttag.Dynamic(\"newRunHistorySize\", len(newRunHistory)),\n\t\t\t\ttag.Dynamic(\"activecluster-sel-policy-nil\", event.WorkflowExecutionContinuedAsNewEventAttributes.ActiveClusterSelectionPolicy == nil),\n\t\t\t)\n\n\t\t\terr := b.mutableState.ReplicateWorkflowExecutionContinuedAsNewEvent(\n\t\t\t\tfirstEvent.ID,\n\t\t\t\tdomainID,\n\t\t\t\tevent,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\tdefault:\n\t\t\treturn nil, &types.BadRequestError{Message: \"Unknown event type\"}\n\t\t}\n\n\t\tb.logger.Debugf(\"Applied event %v of %v for domain %s, wfID %v, event [id:%v, version:%v]\",\n\t\t\ti+1, historyLength, domainID, workflowExecution.WorkflowID, event.ID, event.Version)\n\t}\n\n\tb.mutableState.GetExecutionInfo().SetLastFirstEventID(firstEvent.ID)\n\tb.mutableState.GetExecutionInfo().SetNextEventID(lastEvent.ID + 1)\n\n\tb.mutableState.SetHistoryBuilder(NewHistoryBuilderFromEvents(history, b.mutableState))\n\n\treturn newRunMutableStateBuilder, nil\n}\n\nfunc (b *stateBuilderImpl) GetMutableState() MutableState {\n\n\treturn b.mutableState\n}\n\nfunc (b *stateBuilderImpl) unixNanoToTime(\n\tunixNano int64,\n) time.Time {\n\n\treturn time.Unix(0, unixNano)\n}\n"
  },
  {
    "path": "service/history/execution/state_builder_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: state_builder.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source state_builder.go -destination state_builder_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockStateBuilder is a mock of StateBuilder interface.\ntype MockStateBuilder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockStateBuilderMockRecorder\n\tisgomock struct{}\n}\n\n// MockStateBuilderMockRecorder is the mock recorder for MockStateBuilder.\ntype MockStateBuilderMockRecorder struct {\n\tmock *MockStateBuilder\n}\n\n// NewMockStateBuilder creates a new mock instance.\nfunc NewMockStateBuilder(ctrl *gomock.Controller) *MockStateBuilder {\n\tmock := &MockStateBuilder{ctrl: ctrl}\n\tmock.recorder = &MockStateBuilderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockStateBuilder) EXPECT() *MockStateBuilderMockRecorder {\n\treturn m.recorder\n}\n\n// ApplyEvents mocks base method.\nfunc (m *MockStateBuilder) ApplyEvents(domainID, requestID string, workflowExecution types.WorkflowExecution, history, newRunHistory []*types.HistoryEvent) (MutableState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ApplyEvents\", domainID, requestID, workflowExecution, history, newRunHistory)\n\tret0, _ := ret[0].(MutableState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ApplyEvents indicates an expected call of ApplyEvents.\nfunc (mr *MockStateBuilderMockRecorder) ApplyEvents(domainID, requestID, workflowExecution, history, newRunHistory any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ApplyEvents\", reflect.TypeOf((*MockStateBuilder)(nil).ApplyEvents), domainID, requestID, workflowExecution, history, newRunHistory)\n}\n\n// GetMutableState mocks base method.\nfunc (m *MockStateBuilder) GetMutableState() MutableState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMutableState\")\n\tret0, _ := ret[0].(MutableState)\n\treturn ret0\n}\n\n// GetMutableState indicates an expected call of GetMutableState.\nfunc (mr *MockStateBuilderMockRecorder) GetMutableState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMutableState\", reflect.TypeOf((*MockStateBuilder)(nil).GetMutableState))\n}\n"
  },
  {
    "path": "service/history/execution/state_builder_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tstateBuilderSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockShard        *shard.TestContext\n\t\tmockEventsCache  *events.MockCache\n\t\tmockDomainCache  *cache.MockDomainCache\n\t\tmockMutableState *MockMutableState\n\t\tlogger           log.Logger\n\n\t\tsourceCluster string\n\t\tstateBuilder  *stateBuilderImpl\n\t}\n)\n\nfunc TestStateBuilderSuite(t *testing.T) {\n\ts := new(stateBuilderSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *stateBuilderSuite) SetupSuite() {\n\n}\n\nfunc (s *stateBuilderSuite) TearDownSuite() {\n\n}\n\nfunc (s *stateBuilderSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockMutableState = NewMockMutableState(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockEventsCache = s.mockShard.MockEventsCache\n\ts.mockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(persistence.NewVersionHistories(&persistence.VersionHistory{})).AnyTimes()\n\n\ts.stateBuilder = NewStateBuilder(\n\t\ts.mockShard,\n\t\ts.logger,\n\t\ts.mockMutableState,\n\t).(*stateBuilderImpl)\n\ts.sourceCluster = \"some random source cluster\"\n}\n\nfunc (s *stateBuilderSuite) TearDownTest() {\n\ts.stateBuilder = nil\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *stateBuilderSuite) mockUpdateVersion(events ...*types.HistoryEvent) {\n\tfor _, event := range events {\n\t\ts.mockMutableState.EXPECT().UpdateCurrentVersion(event.Version, true).Times(1)\n\t}\n\ts.mockMutableState.EXPECT().SetHistoryBuilder(NewHistoryBuilderFromEvents(events, s.mockMutableState)).Times(1)\n}\n\nfunc (s *stateBuilderSuite) toHistory(events ...*types.HistoryEvent) []*types.HistoryEvent {\n\treturn events\n}\n\n// workflow operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionStarted_NoCronSchedule() {\n\tcronSchedule := \"\"\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowTimeout: 100,\n\t\tCronSchedule:    cronSchedule,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionStarted\n\tstartWorkflowAttribute := &types.WorkflowExecutionStartedEventAttributes{\n\t\tParentWorkflowDomain: common.StringPtr(constants.TestDomainName),\n\t}\n\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                 version,\n\t\tID:                                      1,\n\t\tTimestamp:                               common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                               &evenType,\n\t\tWorkflowExecutionStartedEventAttributes: startWorkflowAttribute,\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestGlobalParentDomainEntry, nil).Times(1)\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionStartedEvent(&constants.TestDomainID, workflowExecution, requestID, event, true).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\ts.mockMutableState.EXPECT().SetHistoryTree(constants.TestRunID).Return(nil).Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionStarted_WithCronSchedule() {\n\tcronSchedule := \"* * * * *\"\n\tparsedSchedule, err := backoff.ValidateSchedule(cronSchedule)\n\trequire.NoError(s.T(), err)\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tWorkflowTimeout: 100,\n\t\tCronSchedule:    cronSchedule,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionStarted\n\tnext, err := backoff.GetBackoffForNextSchedule(parsedSchedule, now, now, 0, types.CronOverlapPolicySkipped)\n\trequire.NoError(s.T(), err)\n\tstartWorkflowAttribute := &types.WorkflowExecutionStartedEventAttributes{\n\t\tParentWorkflowDomainID: common.StringPtr(constants.TestDomainID),\n\t\tParentWorkflowDomain:   common.StringPtr(constants.TestDomainName),\n\t\tInitiator:              types.ContinueAsNewInitiatorCronSchedule.Ptr(),\n\t\tFirstDecisionTaskBackoffSeconds: common.Int32Ptr(\n\t\t\tint32(next.Seconds()),\n\t\t),\n\t}\n\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                 version,\n\t\tID:                                      1,\n\t\tTimestamp:                               common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                               &evenType,\n\t\tWorkflowExecutionStartedEventAttributes: startWorkflowAttribute,\n\t}\n\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestGlobalParentDomainEntry, nil).Times(0)\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionStartedEvent(&constants.TestDomainID, workflowExecution, requestID, event, true).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\ts.mockMutableState.EXPECT().SetHistoryTree(constants.TestRunID).Return(nil).Times(1)\n\n\t_, err = s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionTimedOut() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionTimedOut\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                  version,\n\t\tID:                                       130,\n\t\tTimestamp:                                common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                                &evenType,\n\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionTimedoutEvent(event.ID, event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionTerminated() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionTerminated\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tWorkflowExecutionTerminatedEventAttributes: &types.WorkflowExecutionTerminatedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionTerminatedEvent(event.ID, event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                version,\n\t\tID:                                     130,\n\t\tTimestamp:                              common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                              &evenType,\n\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionFailedEvent(event.ID, event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionCompleted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionCompleted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionCompletedEvent(event.ID, event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionCanceled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionCanceled\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                  version,\n\t\tID:                                       130,\n\t\tTimestamp:                                common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                                &evenType,\n\t\tWorkflowExecutionCanceledEventAttributes: &types.WorkflowExecutionCanceledEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionCanceledEvent(event.ID, event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionContinuedAsNew() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\tparentWorkflowID := \"some random parent workflow ID\"\n\tparentRunID := uuid.New()\n\tparentInitiatedEventID := int64(144)\n\n\tnow := time.Now()\n\ttasklist := \"some random tasklist\"\n\tworkflowType := \"some random workflow type\"\n\tworkflowTimeoutSecond := int32(110)\n\tdecisionTimeoutSecond := int32(11)\n\tnewRunID := uuid.New()\n\n\tcontinueAsNewEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tNewExecutionRunID: newRunID,\n\t\t},\n\t}\n\n\tnewRunStartedEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        1,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tParentWorkflowDomainID: common.StringPtr(constants.TestDomainID),\n\t\t\tParentWorkflowDomain:   common.StringPtr(constants.TestDomainName),\n\t\t\tParentWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: parentWorkflowID,\n\t\t\t\tRunID:      parentRunID,\n\t\t\t},\n\t\t\tParentInitiatedEventID:              common.Int64Ptr(parentInitiatedEventID),\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t},\n\t}\n\n\tnewRunSignalEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        2,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tSignalName: \"some random signal name\",\n\t\t\tInput:      []byte(\"some random signal input\"),\n\t\t\tIdentity:   \"some random identity\",\n\t\t},\n\t}\n\n\tnewRunDecisionAttempt := int64(123)\n\tnewRunDecisionEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        3,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(decisionTimeoutSecond),\n\t\t\tAttempt:                    newRunDecisionAttempt,\n\t\t},\n\t}\n\tnewRunEvents := []*types.HistoryEvent{\n\t\tnewRunStartedEvent, newRunSignalEvent, newRunDecisionEvent,\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionContinuedAsNewEvent(\n\t\tcontinueAsNewEvent.ID,\n\t\tconstants.TestDomainID,\n\t\tcontinueAsNewEvent,\n\t).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockUpdateVersion(continueAsNewEvent)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t// new workflow domain\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestGlobalParentDomainEntry, nil).AnyTimes()\n\n\tnewRunStateBuilder, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(continueAsNewEvent), newRunEvents)\n\ts.Nil(err)\n\ts.NotNil(newRunStateBuilder)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionContinuedAsNew_EmptyNewRunHistory() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tnewRunID := uuid.New()\n\n\tcontinueAsNewEvent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tNewExecutionRunID: newRunID,\n\t\t},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionContinuedAsNewEvent(\n\t\tcontinueAsNewEvent.ID,\n\t\tconstants.TestDomainID,\n\t\tcontinueAsNewEvent,\n\t).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).AnyTimes()\n\ts.mockUpdateVersion(continueAsNewEvent)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t// new workflow domain\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestGlobalParentDomainEntry, nil).AnyTimes()\n\tnewRunStateBuilder, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(continueAsNewEvent), nil)\n\ts.Nil(err)\n\ts.Nil(newRunStateBuilder)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionSignaled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionSignaled\n\tevent := &types.HistoryEvent{\n\t\tVersion:                                  version,\n\t\tID:                                       130,\n\t\tTimestamp:                                common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                                &evenType,\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{},\n\t}\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionSignaled(event).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeWorkflowExecutionCancelRequested() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\tnow := time.Now()\n\tevenType := types.EventTypeWorkflowExecutionCancelRequested\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tWorkflowExecutionCancelRequestedEventAttributes: &types.WorkflowExecutionCancelRequestedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateWorkflowExecutionCancelRequestedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeUpsertWorkflowSearchAttributes() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeUpsertWorkflowSearchAttributes\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tUpsertWorkflowSearchAttributesEventAttributes: &types.UpsertWorkflowSearchAttributesEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateUpsertWorkflowSearchAttributesEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeMarkerRecorded() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeMarkerRecorded\n\tevent := &types.HistoryEvent{\n\t\tVersion:                       version,\n\t\tID:                            130,\n\t\tTimestamp:                     common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                     &evenType,\n\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{},\n\t}\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// decision operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeDecisionTaskScheduled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\ttasklist := \"some random tasklist\"\n\ttimeoutSecond := int32(11)\n\tevenType := types.EventTypeDecisionTaskScheduled\n\tdecisionAttempt := int64(111)\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\tTaskList:                   &types.TaskList{Name: tasklist},\n\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(timeoutSecond),\n\t\t\tAttempt:                    decisionAttempt,\n\t\t},\n\t}\n\tdi := &DecisionInfo{\n\t\tVersion:         event.Version,\n\t\tScheduleID:      event.ID,\n\t\tStartedID:       commonconstants.EmptyEventID,\n\t\tRequestID:       commonconstants.EmptyUUID,\n\t\tDecisionTimeout: timeoutSecond,\n\t\tTaskList:        tasklist,\n\t\tAttempt:         decisionAttempt,\n\t}\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tTaskList: tasklist,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateDecisionTaskScheduledEvent(\n\t\tevent.Version, event.ID, tasklist, timeoutSecond, decisionAttempt, event.GetTimestamp(), event.GetTimestamp(), false,\n\t).Return(di, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeDecisionTaskStarted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\ttasklist := \"some random tasklist\"\n\ttimeoutSecond := int32(11)\n\tscheduleID := int64(111)\n\tdecisionRequestID := uuid.New()\n\tevenType := types.EventTypeDecisionTaskStarted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{\n\t\t\tScheduledEventID: scheduleID,\n\t\t\tRequestID:        decisionRequestID,\n\t\t},\n\t}\n\tdi := &DecisionInfo{\n\t\tVersion:         event.Version,\n\t\tScheduleID:      scheduleID,\n\t\tStartedID:       event.ID,\n\t\tRequestID:       decisionRequestID,\n\t\tDecisionTimeout: timeoutSecond,\n\t\tTaskList:        tasklist,\n\t\tAttempt:         0,\n\t}\n\ts.mockMutableState.EXPECT().ReplicateDecisionTaskStartedEvent(\n\t\t(*DecisionInfo)(nil), event.Version, scheduleID, event.ID, decisionRequestID, event.GetTimestamp(),\n\t).Return(di, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeDecisionTaskTimedOut() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tscheduleID := int64(12)\n\tstartedID := int64(28)\n\tevenType := types.EventTypeDecisionTaskTimedOut\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\tScheduledEventID: scheduleID,\n\t\t\tStartedEventID:   startedID,\n\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateDecisionTaskTimedOutEvent(event).Return(nil).Times(1)\n\ttasklist := \"some random tasklist\"\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tTaskList: tasklist,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateTransientDecisionTaskScheduled().Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeDecisionTaskFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tscheduleID := int64(12)\n\tstartedID := int64(28)\n\tevenType := types.EventTypeDecisionTaskFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\tScheduledEventID: scheduleID,\n\t\t\tStartedEventID:   startedID,\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateDecisionTaskFailedEvent(event).Return(nil).Times(1)\n\ttasklist := \"some random tasklist\"\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tTaskList: tasklist,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateTransientDecisionTaskScheduled().Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeDecisionTaskCompleted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tscheduleID := int64(12)\n\tstartedID := int64(28)\n\tevenType := types.EventTypeDecisionTaskCompleted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\tScheduledEventID: scheduleID,\n\t\t\tStartedEventID:   startedID,\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateDecisionTaskCompletedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// user timer operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeTimerStarted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\ttimerID := \"timer ID\"\n\ttimeoutSecond := int64(10)\n\tevenType := types.EventTypeTimerStarted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{\n\t\t\tTimerID:                   timerID,\n\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(timeoutSecond),\n\t\t},\n\t}\n\tti := &persistence.TimerInfo{\n\t\tVersion:    event.Version,\n\t\tTimerID:    timerID,\n\t\tExpiryTime: time.Unix(0, event.GetTimestamp()).Add(time.Duration(timeoutSecond) * time.Second),\n\t\tStartedID:  event.ID,\n\t\tTaskStatus: TimerTaskStatusNone,\n\t}\n\ts.mockMutableState.EXPECT().ReplicateTimerStartedEvent(event).Return(ti, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeTimerFired() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeTimerFired\n\tevent := &types.HistoryEvent{\n\t\tVersion:                   version,\n\t\tID:                        130,\n\t\tTimestamp:                 common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                 &evenType,\n\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateTimerFiredEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeCancelTimerFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeCancelTimerFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:                          version,\n\t\tID:                               130,\n\t\tTimestamp:                        common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                        &evenType,\n\t\tCancelTimerFailedEventAttributes: &types.CancelTimerFailedEventAttributes{},\n\t}\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeTimerCanceled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\n\tevenType := types.EventTypeTimerCanceled\n\tevent := &types.HistoryEvent{\n\t\tVersion:                      version,\n\t\tID:                           130,\n\t\tTimestamp:                    common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                    &evenType,\n\t\tTimerCanceledEventAttributes: &types.TimerCanceledEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateTimerCanceledEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// activity operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskScheduled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tactivityID := \"activity ID\"\n\ttasklist := \"some random tasklist\"\n\ttimeoutSecond := int32(10)\n\tevenType := types.EventTypeActivityTaskScheduled\n\tevent := &types.HistoryEvent{\n\t\tVersion:                              version,\n\t\tID:                                   130,\n\t\tTimestamp:                            common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                            &evenType,\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\n\tai := &persistence.ActivityInfo{\n\t\tVersion:                  event.Version,\n\t\tScheduleID:               event.ID,\n\t\tScheduledEventBatchID:    event.ID,\n\t\tScheduledEvent:           event,\n\t\tScheduledTime:            time.Unix(0, event.GetTimestamp()),\n\t\tStartedID:                commonconstants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               activityID,\n\t\tScheduleToStartTimeout:   timeoutSecond,\n\t\tScheduleToCloseTimeout:   timeoutSecond,\n\t\tStartToCloseTimeout:      timeoutSecond,\n\t\tHeartbeatTimeout:         timeoutSecond,\n\t\tCancelRequested:          false,\n\t\tCancelRequestID:          commonconstants.EmptyEventID,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tTaskList:                 tasklist,\n\t}\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tTaskList: tasklist,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskScheduledEvent(event.ID, event, false).Return(ai, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskStarted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\ttasklist := \"some random tasklist\"\n\tevenType := types.EventTypeActivityTaskScheduled\n\tscheduledEvent := &types.HistoryEvent{\n\t\tVersion:                              version,\n\t\tID:                                   130,\n\t\tTimestamp:                            common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                            &evenType,\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\n\tevenType = types.EventTypeActivityTaskStarted\n\tstartedEvent := &types.HistoryEvent{\n\t\tVersion:                            version,\n\t\tID:                                 scheduledEvent.ID + 1,\n\t\tTimestamp:                          common.Int64Ptr(scheduledEvent.GetTimestamp() + 1000),\n\t\tEventType:                          &evenType,\n\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{},\n\t}\n\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tTaskList: tasklist,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskStartedEvent(startedEvent).Return(nil).Times(1)\n\ts.mockUpdateVersion(startedEvent)\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(startedEvent), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskTimedOut() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeActivityTaskTimedOut\n\tevent := &types.HistoryEvent{\n\t\tVersion:                             version,\n\t\tID:                                  130,\n\t\tTimestamp:                           common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                           &evenType,\n\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskTimedOutEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t//\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeActivityTaskFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:                           version,\n\t\tID:                                130,\n\t\tTimestamp:                         common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                         &evenType,\n\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskFailedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskCompleted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeActivityTaskCompleted\n\tevent := &types.HistoryEvent{\n\t\tVersion:                              version,\n\t\tID:                                   130,\n\t\tTimestamp:                            common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                            &evenType,\n\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskCompletedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskCancelRequested() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeActivityTaskCancelRequested\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tActivityTaskCancelRequestedEventAttributes: &types.ActivityTaskCancelRequestedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskCancelRequestedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeRequestCancelActivityTaskFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeRequestCancelActivityTaskFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tRequestCancelActivityTaskFailedEventAttributes: &types.RequestCancelActivityTaskFailedEventAttributes{},\n\t}\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeActivityTaskCanceled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeActivityTaskCanceled\n\tevent := &types.HistoryEvent{\n\t\tVersion:                             version,\n\t\tID:                                  130,\n\t\tTimestamp:                           common.Int64Ptr(now.UnixNano()),\n\t\tEventType:                           &evenType,\n\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{},\n\t}\n\n\ts.mockMutableState.EXPECT().ReplicateActivityTaskCanceledEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\t// assertion on timer generated is in `mockUpdateVersion` function, since activity / user timer\n\t// need to be refreshed each time\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// child workflow operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeStartChildWorkflowExecutionInitiated() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttargetWorkflowID := \"some random target workflow ID\"\n\n\tnow := time.Now()\n\tcreateRequestID := uuid.New()\n\tevenType := types.EventTypeStartChildWorkflowExecutionInitiated\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tStartChildWorkflowExecutionInitiatedEventAttributes: &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain:     constants.TestDomainName,\n\t\t\tWorkflowID: targetWorkflowID,\n\t\t},\n\t}\n\n\tci := &persistence.ChildExecutionInfo{\n\t\tVersion:               event.Version,\n\t\tInitiatedID:           event.ID,\n\t\tInitiatedEventBatchID: event.ID,\n\t\tStartedID:             commonconstants.EmptyEventID,\n\t\tCreateRequestID:       createRequestID,\n\t\tDomainID:              constants.TestDomainID,\n\t\tDomainNameDEPRECATED:  constants.TestDomainName,\n\t}\n\n\t// the create request ID is generated inside, cannot assert equal\n\ts.mockMutableState.EXPECT().ReplicateStartChildWorkflowExecutionInitiatedEvent(\n\t\tevent.ID, event, gomock.Any(),\n\t).Return(ci, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeStartChildWorkflowExecutionFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeStartChildWorkflowExecutionFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tStartChildWorkflowExecutionFailedEventAttributes: &types.StartChildWorkflowExecutionFailedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateStartChildWorkflowExecutionFailedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionStarted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionStarted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionStartedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionTimedOut() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionTimedOut\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionTimedOutEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionTerminated() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionTerminated\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionTerminatedEventAttributes: &types.ChildWorkflowExecutionTerminatedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionTerminatedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionFailedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionCompleted() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionCompleted\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionCompletedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// cancel external workflow operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeRequestCancelExternalWorkflowExecutionInitiated() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\ttargetWorkflowID := \"some random target workflow ID\"\n\ttargetRunID := uuid.New()\n\tchildWorkflowOnly := true\n\n\tnow := time.Now()\n\tcancellationRequestID := uuid.New()\n\tcontrol := []byte(\"some random control\")\n\tevenType := types.EventTypeRequestCancelExternalWorkflowExecutionInitiated\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain: constants.TestDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: targetWorkflowID,\n\t\t\t\tRunID:      targetRunID,\n\t\t\t},\n\t\t\tChildWorkflowOnly: childWorkflowOnly,\n\t\t\tControl:           control,\n\t\t},\n\t}\n\trci := &persistence.RequestCancelInfo{\n\t\tVersion:         event.Version,\n\t\tInitiatedID:     event.ID,\n\t\tCancelRequestID: cancellationRequestID,\n\t}\n\n\t// the cancellation request ID is generated inside, cannot assert equal\n\ts.mockMutableState.EXPECT().ReplicateRequestCancelExternalWorkflowExecutionInitiatedEvent(\n\t\tevent.ID, event, gomock.Any(),\n\t).Return(rci, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeRequestCancelExternalWorkflowExecutionFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeRequestCancelExternalWorkflowExecutionFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes: &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateRequestCancelExternalWorkflowExecutionFailedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeExternalWorkflowExecutionCancelRequested() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeExternalWorkflowExecutionCancelRequested\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tExternalWorkflowExecutionCancelRequestedEventAttributes: &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateExternalWorkflowExecutionCancelRequested(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeChildWorkflowExecutionCanceled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeChildWorkflowExecutionCanceled\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateChildWorkflowExecutionCanceledEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\n// signal external workflow operations\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeSignalExternalWorkflowExecutionInitiated() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\ttargetWorkflowID := \"some random target workflow ID\"\n\ttargetRunID := uuid.New()\n\tchildWorkflowOnly := true\n\n\tnow := time.Now()\n\tsignalRequestID := uuid.New()\n\tsignalName := \"some random signal name\"\n\tsignalInput := []byte(\"some random signal input\")\n\tcontrol := []byte(\"some random control\")\n\tevenType := types.EventTypeSignalExternalWorkflowExecutionInitiated\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes: &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{\n\t\t\tDomain: constants.TestDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: targetWorkflowID,\n\t\t\t\tRunID:      targetRunID,\n\t\t\t},\n\t\t\tSignalName:        signalName,\n\t\t\tInput:             signalInput,\n\t\t\tChildWorkflowOnly: childWorkflowOnly,\n\t\t},\n\t}\n\tsi := &persistence.SignalInfo{\n\t\tVersion:         event.Version,\n\t\tInitiatedID:     event.ID,\n\t\tSignalRequestID: signalRequestID,\n\t\tSignalName:      signalName,\n\t\tInput:           signalInput,\n\t\tControl:         control,\n\t}\n\n\t// the cancellation request ID is generated inside, cannot assert equal\n\ts.mockMutableState.EXPECT().ReplicateSignalExternalWorkflowExecutionInitiatedEvent(\n\t\tevent.ID, event, gomock.Any(),\n\t).Return(si, nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeSignalExternalWorkflowExecutionFailed() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeSignalExternalWorkflowExecutionFailed\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tSignalExternalWorkflowExecutionFailedEventAttributes: &types.SignalExternalWorkflowExecutionFailedEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateSignalExternalWorkflowExecutionFailedEvent(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEvents_EventTypeExternalWorkflowExecutionSignaled() {\n\tversion := int64(1)\n\trequestID := uuid.New()\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      constants.TestRunID,\n\t}\n\n\tnow := time.Now()\n\tevenType := types.EventTypeExternalWorkflowExecutionSignaled\n\tevent := &types.HistoryEvent{\n\t\tVersion:   version,\n\t\tID:        130,\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t\tEventType: &evenType,\n\t\tExternalWorkflowExecutionSignaledEventAttributes: &types.ExternalWorkflowExecutionSignaledEventAttributes{},\n\t}\n\ts.mockMutableState.EXPECT().ReplicateExternalWorkflowExecutionSignaled(event).Return(nil).Times(1)\n\ts.mockUpdateVersion(event)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().ClearStickyness().Times(1)\n\n\t_, err := s.stateBuilder.ApplyEvents(constants.TestDomainID, requestID, workflowExecution, s.toHistory(event), nil)\n\ts.Nil(err)\n}\n\nfunc (s *stateBuilderSuite) TestApplyEventsNewEventsNotHandled() {\n\teventTypes := types.EventTypeValues()\n\ts.Equal(42, len(eventTypes), \"If you see this error, you are adding new event type. \"+\n\t\t\"Before updating the number to make this test pass, please make sure you update stateBuilderImpl.ApplyEvents method \"+\n\t\t\"to handle the new decision type. Otherwise cross dc will not work on the new event.\")\n}\n"
  },
  {
    "path": "service/history/execution/state_rebuilder.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination state_rebuilder_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\t// NDCDefaultPageSize is the default pagination size for ndc\n\tNDCDefaultPageSize = 100\n)\n\ntype (\n\t// StateRebuilder is a mutable state builder to ndc state rebuild\n\tStateRebuilder interface {\n\t\tRebuild(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tbaseWorkflowIdentifier definition.WorkflowIdentifier,\n\t\t\tbaseBranchToken []byte,\n\t\t\tbaseLastEventID int64,\n\t\t\tbaseLastEventVersion int64,\n\t\t\ttargetWorkflowIdentifier definition.WorkflowIdentifier,\n\t\t\ttargetBranchFn func() ([]byte, error),\n\t\t\trequestID string,\n\t\t) (MutableState, int64, error)\n\t}\n\n\tstateRebuilderImpl struct {\n\t\tshard              shard.Context\n\t\tdomainCache        cache.DomainCache\n\t\tclusterMetadata    cluster.Metadata\n\t\thistoryV2Mgr       persistence.HistoryManager\n\t\ttaskRefresher      MutableStateTaskRefresher\n\t\trebuiltHistorySize int64\n\t\tlogger             log.Logger\n\t}\n)\n\nvar _ StateRebuilder = (*stateRebuilderImpl)(nil)\n\n// NewStateRebuilder creates a state rebuilder\nfunc NewStateRebuilder(\n\tshard shard.Context,\n\tlogger log.Logger,\n) StateRebuilder {\n\n\treturn &stateRebuilderImpl{\n\t\tshard:           shard,\n\t\tdomainCache:     shard.GetDomainCache(),\n\t\tclusterMetadata: shard.GetService().GetClusterMetadata(),\n\t\thistoryV2Mgr:    shard.GetHistoryManager(),\n\t\ttaskRefresher: NewMutableStateTaskRefresher(\n\t\t\tshard.GetConfig(),\n\t\t\tshard.GetClusterMetadata(),\n\t\t\tshard.GetDomainCache(),\n\t\t\tshard.GetEventsCache(),\n\t\t\tshard.GetShardID(),\n\t\t\tlogger,\n\t\t),\n\t\trebuiltHistorySize: 0,\n\t\tlogger:             logger,\n\t}\n}\n\nfunc (r *stateRebuilderImpl) Rebuild(\n\tctx context.Context,\n\tnow time.Time,\n\tbaseWorkflowIdentifier definition.WorkflowIdentifier,\n\tbaseBranchToken []byte,\n\tbaseLastEventID int64,\n\tbaseLastEventVersion int64,\n\ttargetWorkflowIdentifier definition.WorkflowIdentifier,\n\ttargetBranchFn func() ([]byte, error),\n\trequestID string,\n) (MutableState, int64, error) {\n\n\titer := collection.NewPagingIterator(r.getPaginationFn(\n\t\tctx,\n\t\tconstants.FirstEventID,\n\t\tbaseLastEventID+1,\n\t\tbaseBranchToken,\n\t\ttargetWorkflowIdentifier.DomainID,\n\t))\n\n\tdomainEntry, err := r.domainCache.GetDomainByID(targetWorkflowIdentifier.DomainID)\n\tif err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\t// Corrupt data handling\n\tif !iter.HasNext() {\n\t\treturn nil, 0, fmt.Errorf(\"Attempting to build history state but the iterator has found no history\")\n\t}\n\t// need to specially handling the first batch, to initialize mutable state & state builder\n\tbatch, err := iter.Next()\n\tif err != nil {\n\t\treturn nil, 0, err\n\t}\n\tfirstEventBatch := batch.(*types.History).Events\n\trebuiltMutableState, stateBuilder := r.initializeBuilders(\n\t\tdomainEntry,\n\t)\n\tif err := r.applyEvents(targetWorkflowIdentifier, stateBuilder, firstEventBatch, requestID); err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\tfor iter.HasNext() {\n\t\tbatch, err := iter.Next()\n\t\tif err != nil {\n\t\t\treturn nil, 0, err\n\t\t}\n\t\tevents := batch.(*types.History).Events\n\t\tif err := r.applyEvents(targetWorkflowIdentifier, stateBuilder, events, requestID); err != nil {\n\t\t\treturn nil, 0, err\n\t\t}\n\t}\n\n\trebuildVersionHistories := rebuiltMutableState.GetVersionHistories()\n\tif rebuildVersionHistories != nil {\n\t\tcurrentVersionHistory, err := rebuildVersionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, 0, err\n\t\t}\n\t\tlastItem, err := currentVersionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn nil, 0, err\n\t\t}\n\t\tif !lastItem.Equals(persistence.NewVersionHistoryItem(\n\t\t\tbaseLastEventID,\n\t\t\tbaseLastEventVersion,\n\t\t)) {\n\t\t\treturn nil, 0, &types.BadRequestError{Message: fmt.Sprintf(\n\t\t\t\t\"nDCStateRebuilder unable to rebuild mutable state to event ID: %v, version: %v, \"+\n\t\t\t\t\t\"baseLastEventID + baseLastEventVersion is not the same as the last event of the last \"+\n\t\t\t\t\t\"batch, event ID: %v, version :%v ,typically because of attemptting to rebuild to a middle of a batch\",\n\t\t\t\tbaseLastEventID,\n\t\t\t\tbaseLastEventVersion,\n\t\t\t\tlastItem.EventID,\n\t\t\t\tlastItem.Version,\n\t\t\t)}\n\t\t}\n\t}\n\n\ttargetBranchToken, err := targetBranchFn()\n\tif err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\tif err := rebuiltMutableState.SetCurrentBranchToken(targetBranchToken); err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\t// close rebuilt mutable state transaction clearing all generated tasks, workflow requests, etc.\n\t_, _, err = rebuiltMutableState.CloseTransactionAsSnapshot(now, TransactionPolicyPassive)\n\tif err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\t// refresh tasks to be generated\n\tif err := r.taskRefresher.RefreshTasks(ctx, now, rebuiltMutableState); err != nil {\n\t\treturn nil, 0, err\n\t}\n\n\t// mutable state rebuild should use the same time stamp\n\trebuiltMutableState.GetExecutionInfo().StartTimestamp = now\n\treturn rebuiltMutableState, r.rebuiltHistorySize, nil\n}\n\nfunc (r *stateRebuilderImpl) initializeBuilders(\n\tdomainEntry *cache.DomainCacheEntry,\n) (MutableState, StateBuilder) {\n\tresetMutableStateBuilder := NewMutableStateBuilderWithVersionHistories(\n\t\tr.shard,\n\t\tr.logger,\n\t\tdomainEntry,\n\t\tconstants.EmptyVersion,\n\t)\n\tstateBuilder := NewStateBuilder(\n\t\tr.shard,\n\t\tr.logger,\n\t\tresetMutableStateBuilder,\n\t)\n\treturn resetMutableStateBuilder, stateBuilder\n}\n\nfunc (r *stateRebuilderImpl) applyEvents(\n\tworkflowIdentifier definition.WorkflowIdentifier,\n\tstateBuilder StateBuilder,\n\tevents []*types.HistoryEvent,\n\trequestID string,\n) error {\n\n\t_, err := stateBuilder.ApplyEvents(\n\t\tworkflowIdentifier.DomainID,\n\t\trequestID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: workflowIdentifier.WorkflowID,\n\t\t\tRunID:      workflowIdentifier.RunID,\n\t\t},\n\t\tevents,\n\t\tnil, // no new run history when rebuilding mutable state\n\t)\n\tif err != nil {\n\t\tr.logger.Error(\"nDCStateRebuilder unable to rebuild mutable state.\", tag.Error(err))\n\t}\n\n\treturn err\n}\n\nfunc (r *stateRebuilderImpl) getPaginationFn(\n\tctx context.Context,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tbranchToken []byte,\n\tdomainID string,\n) collection.PaginationFn {\n\n\treturn func(paginationToken []byte) ([]interface{}, []byte, error) {\n\n\t\t_, historyBatches, token, size, err := persistenceutils.PaginateHistory(\n\t\t\tctx,\n\t\t\tr.historyV2Mgr,\n\t\t\ttrue,\n\t\t\tbranchToken,\n\t\t\tfirstEventID,\n\t\t\tnextEventID,\n\t\t\tpaginationToken,\n\t\t\tNDCDefaultPageSize,\n\t\t\tcommon.IntPtr(r.shard.GetShardID()),\n\t\t\tdomainID,\n\t\t\tr.domainCache,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\tr.rebuiltHistorySize += int64(size)\n\n\t\tvar paginateItems []interface{}\n\t\tfor _, history := range historyBatches {\n\t\t\tpaginateItems = append(paginateItems, history)\n\t\t}\n\t\treturn paginateItems, token, nil\n\t}\n}\n"
  },
  {
    "path": "service/history/execution/state_rebuilder_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: state_rebuilder.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source state_rebuilder.go -destination state_rebuilder_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tdefinition \"github.com/uber/cadence/common/definition\"\n)\n\n// MockStateRebuilder is a mock of StateRebuilder interface.\ntype MockStateRebuilder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockStateRebuilderMockRecorder\n\tisgomock struct{}\n}\n\n// MockStateRebuilderMockRecorder is the mock recorder for MockStateRebuilder.\ntype MockStateRebuilderMockRecorder struct {\n\tmock *MockStateRebuilder\n}\n\n// NewMockStateRebuilder creates a new mock instance.\nfunc NewMockStateRebuilder(ctrl *gomock.Controller) *MockStateRebuilder {\n\tmock := &MockStateRebuilder{ctrl: ctrl}\n\tmock.recorder = &MockStateRebuilderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockStateRebuilder) EXPECT() *MockStateRebuilderMockRecorder {\n\treturn m.recorder\n}\n\n// Rebuild mocks base method.\nfunc (m *MockStateRebuilder) Rebuild(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowIdentifier, baseBranchToken []byte, baseLastEventID, baseLastEventVersion int64, targetWorkflowIdentifier definition.WorkflowIdentifier, targetBranchFn func() ([]byte, error), requestID string) (MutableState, int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Rebuild\", ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchFn, requestID)\n\tret0, _ := ret[0].(MutableState)\n\tret1, _ := ret[1].(int64)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// Rebuild indicates an expected call of Rebuild.\nfunc (mr *MockStateRebuilderMockRecorder) Rebuild(ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchFn, requestID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Rebuild\", reflect.TypeOf((*MockStateRebuilder)(nil).Rebuild), ctx, now, baseWorkflowIdentifier, baseBranchToken, baseLastEventID, baseLastEventVersion, targetWorkflowIdentifier, targetBranchFn, requestID)\n}\n"
  },
  {
    "path": "service/history/execution/state_rebuilder_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/collection\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tstateRebuilderSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller               *gomock.Controller\n\t\tmockShard                *shard.TestContext\n\t\tmockEventsCache          *events.MockCache\n\t\tmockTaskRefresher        *MockMutableStateTaskRefresher\n\t\tmockDomainCache          *cache.MockDomainCache\n\t\tmockActiveClusterManager *activecluster.MockManager\n\t\tmockHistoryV2Mgr         *mocks.HistoryV2Manager\n\t\tlogger                   log.Logger\n\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\n\t\tnDCStateRebuilder *stateRebuilderImpl\n\t}\n)\n\nfunc TestStateRebuilderSuite(t *testing.T) {\n\ts := new(stateRebuilderSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *stateRebuilderSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockTaskRefresher = NewMockMutableStateTaskRefresher(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockEventsCache = s.mockShard.MockEventsCache\n\ts.mockEventsCache.EXPECT().PutEvent(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.workflowID = \"some random workflow ID\"\n\ts.runID = uuid.New()\n\ts.nDCStateRebuilder = NewStateRebuilder(\n\t\ts.mockShard,\n\t\ts.logger,\n\t).(*stateRebuilderImpl)\n\ts.nDCStateRebuilder.taskRefresher = s.mockTaskRefresher\n}\n\nfunc (s *stateRebuilderSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *stateRebuilderSuite) TestInitializeBuilders() {\n\tmutableState, stateBuilder := s.nDCStateRebuilder.initializeBuilders(constants.TestGlobalDomainEntry)\n\ts.NotNil(mutableState)\n\ts.NotNil(stateBuilder)\n\ts.NotNil(mutableState.GetVersionHistories())\n}\n\nfunc (s *stateRebuilderSuite) TestApplyEvents() {\n\n\trequestID := uuid.New()\n\tevents := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:                                      1,\n\t\t\tEventType:                               types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tID:                                       2,\n\t\t\tEventType:                                types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{},\n\t\t},\n\t}\n\n\tworkflowIdentifier := definition.NewWorkflowIdentifier(s.domainID, s.workflowID, s.runID)\n\n\tmockStateBuilder := NewMockStateBuilder(s.controller)\n\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\ts.domainID,\n\t\trequestID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: s.workflowID,\n\t\t\tRunID:      s.runID,\n\t\t},\n\t\tevents,\n\t\t[]*types.HistoryEvent(nil),\n\t).Return(nil, nil).Times(1)\n\n\terr := s.nDCStateRebuilder.applyEvents(workflowIdentifier, mockStateBuilder, events, requestID)\n\ts.NoError(err)\n}\n\nfunc (s *stateRebuilderSuite) TestPagination() {\n\tfirstEventID := commonconstants.FirstEventID\n\tnextEventID := int64(101)\n\tbranchToken := []byte(\"some random branch token\")\n\tdomainName := \"some random domain name\"\n\n\tevent1 := &types.HistoryEvent{\n\t\tID:                                      1,\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                   2,\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tevent3 := &types.HistoryEvent{\n\t\tID:                                 3,\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t}\n\tevent4 := &types.HistoryEvent{\n\t\tID:                                   4,\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tevent5 := &types.HistoryEvent{\n\t\tID:                                   5,\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\thistory1 := []*types.History{{Events: []*types.HistoryEvent{event1, event2, event3}}}\n\thistory2 := []*types.History{{Events: []*types.HistoryEvent{event4, event5}}}\n\thistory := append(history1, history2...)\n\tpageToken := []byte(\"some random token\")\n\ts.mockDomainCache.EXPECT().GetDomainName(s.domainID).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history1,\n\t\tNextPageToken: pageToken,\n\t\tSize:          12345,\n\t}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      NDCDefaultPageSize,\n\t\tNextPageToken: pageToken,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history2,\n\t\tNextPageToken: nil,\n\t\tSize:          67890,\n\t}, nil).Once()\n\n\tpaginationFn := s.nDCStateRebuilder.getPaginationFn(context.Background(), firstEventID, nextEventID, branchToken, s.domainID)\n\titer := collection.NewPagingIterator(paginationFn)\n\n\tresult := []*types.History{}\n\tfor iter.HasNext() {\n\t\titem, err := iter.Next()\n\t\ts.NoError(err)\n\t\tresult = append(result, item.(*types.History))\n\t}\n\n\ts.Equal(history, result)\n}\n\nfunc (s *stateRebuilderSuite) TestRebuild() {\n\trequestID := uuid.New()\n\tversion := int64(12)\n\tlastEventID := int64(2)\n\tbranchToken := []byte(\"other random branch token\")\n\ttargetBranchToken := []byte(\"some other random branch token\")\n\tnow := time.Now()\n\tpartitionConfig := map[string]string{\n\t\t\"userid\": uuid.New(),\n\t}\n\n\ttargetDomainID := uuid.New()\n\ttargetDomainName := \"other random domain name\"\n\ttargetWorkflowID := \"other random workflow ID\"\n\ttargetRunID := uuid.New()\n\n\tfirstEventID := commonconstants.FirstEventID\n\tnextEventID := lastEventID + 1\n\tevents1 := []*types.HistoryEvent{{\n\t\tID:        1,\n\t\tVersion:   version,\n\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"some random workflow type\"},\n\t\t\tTaskList:                            &types.TaskList{Name: \"some random workflow type\"},\n\t\t\tInput:                               []byte(\"some random input\"),\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(123),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(233),\n\t\t\tIdentity:                            \"some random identity\",\n\t\t\tPartitionConfig:                     partitionConfig,\n\t\t},\n\t}}\n\tevents2 := []*types.HistoryEvent{{\n\t\tID:        2,\n\t\tVersion:   version,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tSignalName: \"some random signal name\",\n\t\t\tInput:      []byte(\"some random signal input\"),\n\t\t\tIdentity:   \"some random identity\",\n\t\t},\n\t}}\n\thistory1 := []*types.History{{Events: events1}}\n\thistory2 := []*types.History{{Events: events2}}\n\tpageToken := []byte(\"some random pagination token\")\n\n\thistorySize1 := 12345\n\thistorySize2 := 67890\n\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(targetDomainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    targetDomainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history1,\n\t\tNextPageToken: pageToken,\n\t\tSize:          historySize1,\n\t}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      NDCDefaultPageSize,\n\t\tNextPageToken: pageToken,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    targetDomainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history2,\n\t\tNextPageToken: nil,\n\t\tSize:          historySize2,\n\t}, nil).Once()\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(targetDomainID).Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: targetDomainID, Name: targetDomainName},\n\t\t&persistence.DomainConfig{},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t), nil).AnyTimes()\n\ts.mockTaskRefresher.EXPECT().RefreshTasks(gomock.Any(), now, gomock.Any()).Return(nil).Times(1)\n\n\ttargetBranchTokenFn := func() ([]byte, error) {\n\t\treturn targetBranchToken, nil\n\t}\n\n\trebuildMutableState, rebuiltHistorySize, err := s.nDCStateRebuilder.Rebuild(\n\t\tcontext.Background(),\n\t\tnow,\n\t\tdefinition.NewWorkflowIdentifier(s.domainID, s.workflowID, s.runID),\n\t\tbranchToken,\n\t\tlastEventID,\n\t\tversion,\n\t\tdefinition.NewWorkflowIdentifier(targetDomainID, targetWorkflowID, targetRunID),\n\t\ttargetBranchTokenFn,\n\t\trequestID,\n\t)\n\ts.NoError(err)\n\ts.NotNil(rebuildMutableState)\n\trebuildExecutionInfo := rebuildMutableState.GetExecutionInfo()\n\ts.Equal(targetDomainID, rebuildExecutionInfo.DomainID)\n\ts.Equal(targetWorkflowID, rebuildExecutionInfo.WorkflowID)\n\ts.Equal(targetRunID, rebuildExecutionInfo.RunID)\n\ts.Equal(partitionConfig, rebuildExecutionInfo.PartitionConfig)\n\ts.Equal(int64(historySize1+historySize2), rebuiltHistorySize)\n\ts.Equal(persistence.NewVersionHistories(\n\t\tpersistence.NewVersionHistory(\n\t\t\ttargetBranchToken,\n\t\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID, version)},\n\t\t),\n\t), rebuildMutableState.GetVersionHistories())\n\ts.Equal(rebuildMutableState.GetExecutionInfo().StartTimestamp, now)\n}\n\nfunc (s *stateRebuilderSuite) TestInvalidStateHandling() {\n\trequestID := uuid.New()\n\tversion := int64(12)\n\tlastEventID := int64(2)\n\tbranchToken := []byte(\"other random branch token\")\n\ttargetBranchToken := []byte(\"some other random branch token\")\n\tnow := time.Now()\n\n\ttargetDomainID := uuid.New()\n\ttargetDomainName := \"other random domain name\"\n\ttargetWorkflowID := \"other random workflow ID\"\n\ttargetRunID := uuid.New()\n\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(targetDomainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(targetDomainID).Return(cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: targetDomainID, Name: targetDomainName},\n\t\t&persistence.DomainConfig{},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t), nil).AnyTimes()\n\n\ttargetBranchTokenFn := func() ([]byte, error) {\n\t\treturn targetBranchToken, nil\n\t}\n\n\ts.nDCStateRebuilder.Rebuild(\n\t\tcontext.Background(),\n\t\tnow,\n\t\tdefinition.NewWorkflowIdentifier(s.domainID, s.workflowID, s.runID),\n\t\tbranchToken,\n\t\tlastEventID,\n\t\tversion,\n\t\tdefinition.NewWorkflowIdentifier(targetDomainID, targetWorkflowID, targetRunID),\n\t\ttargetBranchTokenFn,\n\t\trequestID,\n\t)\n}\n"
  },
  {
    "path": "service/history/execution/timer_sequence.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination timer_sequence_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// TimerType indicates timer type\n\tTimerType int32\n)\n\nconst (\n\t// TimerTypeStartToClose is the timer type for activity startToClose timer\n\tTimerTypeStartToClose = TimerType(types.TimeoutTypeStartToClose)\n\t// TimerTypeScheduleToStart is the timer type for activity scheduleToStart timer\n\tTimerTypeScheduleToStart = TimerType(types.TimeoutTypeScheduleToStart)\n\t// TimerTypeScheduleToClose is the timer type for activity scheduleToClose timer\n\tTimerTypeScheduleToClose = TimerType(types.TimeoutTypeScheduleToClose)\n\t// TimerTypeHeartbeat is the timer type for activity heartbeat timer\n\tTimerTypeHeartbeat = TimerType(types.TimeoutTypeHeartbeat)\n)\n\nconst (\n\t// TimerTaskStatusNone indicates activity / user timer task has not been created\n\tTimerTaskStatusNone = iota\n\t// TimerTaskStatusCreated indicates user timer task has been created\n\tTimerTaskStatusCreated\n)\n\nconst (\n\t// TimerTaskStatusCreatedStartToClose indicates activity startToClose timer has been created\n\tTimerTaskStatusCreatedStartToClose = 1 << iota\n\t// TimerTaskStatusCreatedScheduleToStart indicates activity scheduleToStart timer has been created\n\tTimerTaskStatusCreatedScheduleToStart\n\t// TimerTaskStatusCreatedScheduleToClose indicates activity scheduleToClose timer has been created\n\tTimerTaskStatusCreatedScheduleToClose\n\t// TimerTaskStatusCreatedHeartbeat indicates activity heartbeat timer has been created\n\tTimerTaskStatusCreatedHeartbeat\n)\n\ntype (\n\t// TimerSequenceID describes user / activity timer and defines an order among timers\n\tTimerSequenceID struct {\n\t\tEventID      int64\n\t\tTimestamp    time.Time\n\t\tTimerType    TimerType\n\t\tTimerCreated bool\n\t\tAttempt      int32\n\t}\n\n\t// TimerSequenceIDs is a list of TimerSequenceID\n\tTimerSequenceIDs []TimerSequenceID\n\n\t// TimerSequence manages user / activity timer\n\tTimerSequence interface {\n\t\tIsExpired(referenceTime time.Time, TimerSequenceID TimerSequenceID) (time.Duration, bool)\n\n\t\tCreateNextUserTimer() (bool, error)\n\t\tCreateNextActivityTimer() (bool, error)\n\n\t\tLoadAndSortUserTimers() []TimerSequenceID\n\t\tLoadAndSortActivityTimers() []TimerSequenceID\n\t}\n\n\ttimerSequenceImpl struct {\n\t\tmutableState MutableState\n\t}\n)\n\nvar _ TimerSequence = (*timerSequenceImpl)(nil)\n\n// NewTimerSequence creates a new timer sequence\nfunc NewTimerSequence(\n\tmutableState MutableState,\n) TimerSequence {\n\treturn &timerSequenceImpl{\n\t\tmutableState: mutableState,\n\t}\n}\n\nfunc (t *timerSequenceImpl) IsExpired(\n\treferenceTime time.Time,\n\tTimerSequenceID TimerSequenceID,\n) (time.Duration, bool) {\n\n\t// Cassandra timestamp resolution is in millisecond\n\t// here we do the check in terms of second resolution.\n\n\ttimerFireTimeInSecond := TimerSequenceID.Timestamp.Unix()\n\treferenceTimeInSecond := referenceTime.Unix()\n\n\tif timerFireTimeInSecond <= referenceTimeInSecond {\n\t\treturn time.Duration(referenceTimeInSecond-timerFireTimeInSecond) * time.Second, true\n\t}\n\n\treturn 0, false\n}\n\nfunc (t *timerSequenceImpl) CreateNextUserTimer() (bool, error) {\n\n\tsequenceIDs := t.LoadAndSortUserTimers()\n\tif len(sequenceIDs) == 0 {\n\t\treturn false, nil\n\t}\n\n\tfirstTimerTask := sequenceIDs[0]\n\n\t// timer has already been created\n\tif firstTimerTask.TimerCreated {\n\t\treturn false, nil\n\t}\n\n\ttimerInfo, ok := t.mutableState.GetUserTimerInfoByEventID(firstTimerTask.EventID)\n\tif !ok {\n\t\treturn false, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unable to load activity info %v\", firstTimerTask.EventID),\n\t\t}\n\t}\n\t// mark timer task mask as indication that timer task is generated\n\t// here TaskID is misleading attr, should be called timer created flag or something\n\ttimerInfo.TaskStatus = TimerTaskStatusCreated\n\tif err := t.mutableState.UpdateUserTimer(timerInfo); err != nil {\n\t\treturn false, err\n\t}\n\texecutionInfo := t.mutableState.GetExecutionInfo()\n\tt.mutableState.AddTimerTasks(&persistence.UserTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: firstTimerTask.Timestamp,\n\t\t\tVersion:             t.mutableState.GetCurrentVersion(),\n\t\t},\n\t\tEventID:  firstTimerTask.EventID,\n\t\tTaskList: executionInfo.TaskList,\n\t})\n\treturn true, nil\n}\n\nfunc (t *timerSequenceImpl) CreateNextActivityTimer() (bool, error) {\n\n\tsequenceIDs := t.LoadAndSortActivityTimers()\n\tif len(sequenceIDs) == 0 {\n\t\treturn false, nil\n\t}\n\n\tfirstTimerTask := sequenceIDs[0]\n\n\t// timer has already been created\n\tif firstTimerTask.TimerCreated {\n\t\treturn false, nil\n\t}\n\n\tactivityInfo, ok := t.mutableState.GetActivityInfo(firstTimerTask.EventID)\n\tif !ok {\n\t\treturn false, &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unable to load activity info %v\", firstTimerTask.EventID),\n\t\t}\n\t}\n\t// mark timer task mask as indication that timer task is generated\n\tactivityInfo.TimerTaskStatus |= TimerTypeToTimerMask(firstTimerTask.TimerType)\n\tif firstTimerTask.TimerType == TimerTypeHeartbeat {\n\t\tactivityInfo.LastHeartbeatTimeoutVisibilityInSeconds = firstTimerTask.Timestamp.Unix()\n\t}\n\tif err := t.mutableState.UpdateActivity(activityInfo); err != nil {\n\t\treturn false, err\n\t}\n\texecutionInfo := t.mutableState.GetExecutionInfo()\n\tt.mutableState.AddTimerTasks(&persistence.ActivityTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\tRunID:      executionInfo.RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: firstTimerTask.Timestamp,\n\t\t\tVersion:             t.mutableState.GetCurrentVersion(),\n\t\t},\n\t\tTimeoutType: int(firstTimerTask.TimerType),\n\t\tEventID:     firstTimerTask.EventID,\n\t\tAttempt:     int64(firstTimerTask.Attempt),\n\t\tTaskList:    activityInfo.TaskList,\n\t})\n\treturn true, nil\n}\n\nfunc (t *timerSequenceImpl) LoadAndSortUserTimers() []TimerSequenceID {\n\n\tpendingTimers := t.mutableState.GetPendingTimerInfos()\n\ttimers := make(TimerSequenceIDs, 0, len(pendingTimers))\n\n\tfor _, timerInfo := range pendingTimers {\n\n\t\tif sequenceID := t.getUserTimerTimeout(\n\t\t\ttimerInfo,\n\t\t); sequenceID != nil {\n\t\t\ttimers = append(timers, *sequenceID)\n\t\t}\n\t}\n\n\tsort.Sort(timers)\n\treturn timers\n}\n\nfunc (t *timerSequenceImpl) LoadAndSortActivityTimers() []TimerSequenceID {\n\t// there can be 4 timer per activity\n\t// see TimerType\n\tpendingActivities := t.mutableState.GetPendingActivityInfos()\n\tactivityTimers := make(TimerSequenceIDs, 0, len(pendingActivities)*4)\n\n\tfor _, activityInfo := range pendingActivities {\n\n\t\tif sequenceID := t.getActivityScheduleToCloseTimeout(\n\t\t\tactivityInfo,\n\t\t); sequenceID != nil {\n\t\t\tactivityTimers = append(activityTimers, *sequenceID)\n\t\t}\n\n\t\tif sequenceID := t.getActivityScheduleToStartTimeout(\n\t\t\tactivityInfo,\n\t\t); sequenceID != nil {\n\t\t\tactivityTimers = append(activityTimers, *sequenceID)\n\t\t}\n\n\t\tif sequenceID := t.getActivityStartToCloseTimeout(\n\t\t\tactivityInfo,\n\t\t); sequenceID != nil {\n\t\t\tactivityTimers = append(activityTimers, *sequenceID)\n\t\t}\n\n\t\tif sequenceID := t.getActivityHeartbeatTimeout(\n\t\t\tactivityInfo,\n\t\t); sequenceID != nil {\n\t\t\tactivityTimers = append(activityTimers, *sequenceID)\n\t\t}\n\t}\n\n\tsort.Sort(activityTimers)\n\treturn activityTimers\n}\n\nfunc (t *timerSequenceImpl) getUserTimerTimeout(\n\ttimerInfo *persistence.TimerInfo,\n) *TimerSequenceID {\n\n\treturn &TimerSequenceID{\n\t\tEventID:      timerInfo.StartedID,\n\t\tTimestamp:    timerInfo.ExpiryTime,\n\t\tTimerType:    TimerTypeStartToClose,\n\t\tTimerCreated: timerInfo.TaskStatus == TimerTaskStatusCreated,\n\t\tAttempt:      0,\n\t}\n}\n\nfunc (t *timerSequenceImpl) getActivityScheduleToStartTimeout(\n\tactivityInfo *persistence.ActivityInfo,\n) *TimerSequenceID {\n\n\t// activity is not scheduled yet, probably due to retry & backoff\n\tif activityInfo.ScheduleID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\t// activity is already started\n\tif activityInfo.StartedID != constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\tstartTimeout := activityInfo.ScheduledTime.Add(\n\t\ttime.Duration(activityInfo.ScheduleToStartTimeout) * time.Second,\n\t)\n\n\treturn &TimerSequenceID{\n\t\tEventID:      activityInfo.ScheduleID,\n\t\tTimestamp:    startTimeout,\n\t\tTimerType:    TimerTypeScheduleToStart,\n\t\tTimerCreated: (activityInfo.TimerTaskStatus & TimerTaskStatusCreatedScheduleToStart) > 0,\n\t\tAttempt:      activityInfo.Attempt,\n\t}\n}\n\nfunc (t *timerSequenceImpl) getActivityScheduleToCloseTimeout(\n\tactivityInfo *persistence.ActivityInfo,\n) *TimerSequenceID {\n\n\t// activity is not scheduled yet, probably due to retry & backoff\n\tif activityInfo.ScheduleID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\tcloseTimeout := activityInfo.ScheduledTime.Add(\n\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t)\n\n\treturn &TimerSequenceID{\n\t\tEventID:      activityInfo.ScheduleID,\n\t\tTimestamp:    closeTimeout,\n\t\tTimerType:    TimerTypeScheduleToClose,\n\t\tTimerCreated: (activityInfo.TimerTaskStatus & TimerTaskStatusCreatedScheduleToClose) > 0,\n\t\tAttempt:      activityInfo.Attempt,\n\t}\n}\n\nfunc (t *timerSequenceImpl) getActivityStartToCloseTimeout(\n\tactivityInfo *persistence.ActivityInfo,\n) *TimerSequenceID {\n\n\t// activity is not scheduled yet, probably due to retry & backoff\n\tif activityInfo.ScheduleID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\t// activity is not started yet\n\tif activityInfo.StartedID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\tcloseTimeout := activityInfo.StartedTime.Add(\n\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t)\n\n\treturn &TimerSequenceID{\n\t\tEventID:      activityInfo.ScheduleID,\n\t\tTimestamp:    closeTimeout,\n\t\tTimerType:    TimerTypeStartToClose,\n\t\tTimerCreated: (activityInfo.TimerTaskStatus & TimerTaskStatusCreatedStartToClose) > 0,\n\t\tAttempt:      activityInfo.Attempt,\n\t}\n}\n\nfunc (t *timerSequenceImpl) getActivityHeartbeatTimeout(\n\tactivityInfo *persistence.ActivityInfo,\n) *TimerSequenceID {\n\n\t// activity is not scheduled yet, probably due to retry & backoff\n\tif activityInfo.ScheduleID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\t// activity is not started yet\n\tif activityInfo.StartedID == constants.EmptyEventID {\n\t\treturn nil\n\t}\n\n\t// not heartbeat timeout configured\n\tif activityInfo.HeartbeatTimeout <= 0 {\n\t\treturn nil\n\t}\n\n\t// use the latest time as last heartbeat time\n\tlastHeartbeat := activityInfo.StartedTime\n\tif activityInfo.LastHeartBeatUpdatedTime.After(lastHeartbeat) {\n\t\tlastHeartbeat = activityInfo.LastHeartBeatUpdatedTime\n\t}\n\n\theartbeatTimeout := lastHeartbeat.Add(\n\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t)\n\n\treturn &TimerSequenceID{\n\t\tEventID:      activityInfo.ScheduleID,\n\t\tTimestamp:    heartbeatTimeout,\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: (activityInfo.TimerTaskStatus & TimerTaskStatusCreatedHeartbeat) > 0,\n\t\tAttempt:      activityInfo.Attempt,\n\t}\n}\n\n// TimerTypeToTimerMask converts TimerType into the TimerTaskStatus flag\nfunc TimerTypeToTimerMask(\n\tTimerType TimerType,\n) int32 {\n\n\tswitch TimerType {\n\tcase TimerTypeStartToClose:\n\t\treturn TimerTaskStatusCreatedStartToClose\n\tcase TimerTypeScheduleToStart:\n\t\treturn TimerTaskStatusCreatedScheduleToStart\n\tcase TimerTypeScheduleToClose:\n\t\treturn TimerTaskStatusCreatedScheduleToClose\n\tcase TimerTypeHeartbeat:\n\t\treturn TimerTaskStatusCreatedHeartbeat\n\tdefault:\n\t\tpanic(\"invalid timeout type\")\n\t}\n}\n\n// TimerTypeToInternal converts TimeType to its internal representation\nfunc TimerTypeToInternal(\n\tTimerType TimerType,\n) types.TimeoutType {\n\n\tswitch TimerType {\n\tcase TimerTypeStartToClose:\n\t\treturn types.TimeoutTypeStartToClose\n\tcase TimerTypeScheduleToStart:\n\t\treturn types.TimeoutTypeScheduleToStart\n\tcase TimerTypeScheduleToClose:\n\t\treturn types.TimeoutTypeScheduleToClose\n\tcase TimerTypeHeartbeat:\n\t\treturn types.TimeoutTypeHeartbeat\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"invalid timer type: %v\", TimerType))\n\t}\n}\n\n// TimerTypeFromInternal gets TimerType from internal type\nfunc TimerTypeFromInternal(\n\tTimerType types.TimeoutType,\n) TimerType {\n\n\tswitch TimerType {\n\tcase types.TimeoutTypeStartToClose:\n\t\treturn TimerTypeStartToClose\n\tcase types.TimeoutTypeScheduleToStart:\n\t\treturn TimerTypeScheduleToStart\n\tcase types.TimeoutTypeScheduleToClose:\n\t\treturn TimerTypeScheduleToClose\n\tcase types.TimeoutTypeHeartbeat:\n\t\treturn TimerTypeHeartbeat\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"invalid timeout type: %v\", TimerType))\n\t}\n}\n\n// TimerTypeToReason creates timeout reason based on the TimeType\nfunc TimerTypeToReason(\n\ttimerType TimerType,\n) string {\n\treturn fmt.Sprintf(\"cadenceInternal:Timeout %v\", TimerTypeToInternal(timerType))\n}\n\n// Len implements sort.Interface\nfunc (s TimerSequenceIDs) Len() int {\n\treturn len(s)\n}\n\n// Swap implements sort.Interface.\nfunc (s TimerSequenceIDs) Swap(\n\tthis int,\n\tthat int,\n) {\n\ts[this], s[that] = s[that], s[this]\n}\n\n// Less implements sort.Interface\nfunc (s TimerSequenceIDs) Less(\n\tthis int,\n\tthat int,\n) bool {\n\n\tthisSequenceID := s[this]\n\tthatSequenceID := s[that]\n\n\t// order: timeout time, event ID, timeout type\n\n\tif thisSequenceID.Timestamp.Before(thatSequenceID.Timestamp) {\n\t\treturn true\n\t} else if thisSequenceID.Timestamp.After(thatSequenceID.Timestamp) {\n\t\treturn false\n\t}\n\n\t// timeout time are the same\n\tif thisSequenceID.EventID < thatSequenceID.EventID {\n\t\treturn true\n\t} else if thisSequenceID.EventID > thatSequenceID.EventID {\n\t\treturn false\n\t}\n\n\t// timeout time & event ID are the same\n\tif thisSequenceID.TimerType < thatSequenceID.TimerType {\n\t\treturn true\n\t} else if thisSequenceID.TimerType > thatSequenceID.TimerType {\n\t\treturn false\n\t}\n\n\t// thisSequenceID && thatSequenceID are the same\n\treturn true\n}\n"
  },
  {
    "path": "service/history/execution/timer_sequence_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: timer_sequence.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source timer_sequence.go -destination timer_sequence_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockTimerSequence is a mock of TimerSequence interface.\ntype MockTimerSequence struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTimerSequenceMockRecorder\n\tisgomock struct{}\n}\n\n// MockTimerSequenceMockRecorder is the mock recorder for MockTimerSequence.\ntype MockTimerSequenceMockRecorder struct {\n\tmock *MockTimerSequence\n}\n\n// NewMockTimerSequence creates a new mock instance.\nfunc NewMockTimerSequence(ctrl *gomock.Controller) *MockTimerSequence {\n\tmock := &MockTimerSequence{ctrl: ctrl}\n\tmock.recorder = &MockTimerSequenceMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTimerSequence) EXPECT() *MockTimerSequenceMockRecorder {\n\treturn m.recorder\n}\n\n// CreateNextActivityTimer mocks base method.\nfunc (m *MockTimerSequence) CreateNextActivityTimer() (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateNextActivityTimer\")\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateNextActivityTimer indicates an expected call of CreateNextActivityTimer.\nfunc (mr *MockTimerSequenceMockRecorder) CreateNextActivityTimer() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateNextActivityTimer\", reflect.TypeOf((*MockTimerSequence)(nil).CreateNextActivityTimer))\n}\n\n// CreateNextUserTimer mocks base method.\nfunc (m *MockTimerSequence) CreateNextUserTimer() (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateNextUserTimer\")\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateNextUserTimer indicates an expected call of CreateNextUserTimer.\nfunc (mr *MockTimerSequenceMockRecorder) CreateNextUserTimer() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateNextUserTimer\", reflect.TypeOf((*MockTimerSequence)(nil).CreateNextUserTimer))\n}\n\n// IsExpired mocks base method.\nfunc (m *MockTimerSequence) IsExpired(referenceTime time.Time, TimerSequenceID TimerSequenceID) (time.Duration, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsExpired\", referenceTime, TimerSequenceID)\n\tret0, _ := ret[0].(time.Duration)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// IsExpired indicates an expected call of IsExpired.\nfunc (mr *MockTimerSequenceMockRecorder) IsExpired(referenceTime, TimerSequenceID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsExpired\", reflect.TypeOf((*MockTimerSequence)(nil).IsExpired), referenceTime, TimerSequenceID)\n}\n\n// LoadAndSortActivityTimers mocks base method.\nfunc (m *MockTimerSequence) LoadAndSortActivityTimers() []TimerSequenceID {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadAndSortActivityTimers\")\n\tret0, _ := ret[0].([]TimerSequenceID)\n\treturn ret0\n}\n\n// LoadAndSortActivityTimers indicates an expected call of LoadAndSortActivityTimers.\nfunc (mr *MockTimerSequenceMockRecorder) LoadAndSortActivityTimers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadAndSortActivityTimers\", reflect.TypeOf((*MockTimerSequence)(nil).LoadAndSortActivityTimers))\n}\n\n// LoadAndSortUserTimers mocks base method.\nfunc (m *MockTimerSequence) LoadAndSortUserTimers() []TimerSequenceID {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadAndSortUserTimers\")\n\tret0, _ := ret[0].([]TimerSequenceID)\n\treturn ret0\n}\n\n// LoadAndSortUserTimers indicates an expected call of LoadAndSortUserTimers.\nfunc (mr *MockTimerSequenceMockRecorder) LoadAndSortUserTimers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadAndSortUserTimers\", reflect.TypeOf((*MockTimerSequence)(nil).LoadAndSortUserTimers))\n}\n"
  },
  {
    "path": "service/history/execution/timer_sequence_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\ttimerSequenceSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockMutableState *MockMutableState\n\n\t\ttimerSequence *timerSequenceImpl\n\t}\n)\n\nfunc TestTimerSequenceSuite(t *testing.T) {\n\ts := new(timerSequenceSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerSequenceSuite) SetupSuite() {\n\n}\n\nfunc (s *timerSequenceSuite) TearDownSuite() {\n\n}\n\nfunc (s *timerSequenceSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockMutableState = NewMockMutableState(s.controller)\n\n\ts.timerSequence = NewTimerSequence(s.mockMutableState).(*timerSequenceImpl)\n}\n\nfunc (s *timerSequenceSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *timerSequenceSuite) TestCreateNextUserTimer_AlreadyCreated() {\n\tnow := time.Now()\n\ttimerInfo := &persistence.TimerInfo{\n\t\tVersion:    123,\n\t\tTimerID:    \"some random timer ID\",\n\t\tStartedID:  456,\n\t\tExpiryTime: now.Add(100 * time.Second),\n\t\tTaskStatus: TimerTaskStatusCreated,\n\t}\n\ttimerInfos := map[string]*persistence.TimerInfo{timerInfo.TimerID: timerInfo}\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(timerInfos).Times(1)\n\n\tmodified, err := s.timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.False(modified)\n}\n\nfunc (s *timerSequenceSuite) TestCreateNextUserTimer_NotCreated() {\n\tnow := time.Now()\n\tcurrentVersion := int64(999)\n\ttimerInfo := &persistence.TimerInfo{\n\t\tVersion:    123,\n\t\tTimerID:    \"some random timer ID\",\n\t\tStartedID:  456,\n\t\tExpiryTime: now.Add(100 * time.Second),\n\t\tTaskStatus: TimerTaskStatusNone,\n\t}\n\ttimerInfos := map[string]*persistence.TimerInfo{timerInfo.TimerID: timerInfo}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"run-id\",\n\t\tTaskList:   \"task-list\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(timerInfos).Times(1)\n\ts.mockMutableState.EXPECT().GetUserTimerInfoByEventID(timerInfo.StartedID).Return(timerInfo, true).Times(1)\n\n\tvar timerInfoUpdated = *timerInfo // make a copy\n\ttimerInfoUpdated.TaskStatus = TimerTaskStatusCreated\n\ts.mockMutableState.EXPECT().UpdateUserTimer(&timerInfoUpdated).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(currentVersion).Times(1)\n\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.UserTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"run-id\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: timerInfo.ExpiryTime,\n\t\t\tVersion:             currentVersion,\n\t\t},\n\t\tEventID:  timerInfo.StartedID,\n\t\tTaskList: \"task-list\",\n\t}).Times(1)\n\n\tmodified, err := s.timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n}\n\nfunc (s *timerSequenceSuite) TestCreateNextActivityTimer_AlreadyCreated() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedScheduleToStart,\n\t\tAttempt:                  12,\n\t\tTaskList:                 \"task-list\",\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tmodified, err := s.timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.False(modified)\n}\n\nfunc (s *timerSequenceSuite) TestCreateNextActivityTimer_NotCreated() {\n\tnow := time.Now()\n\tcurrentVersion := int64(999)\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t\tTaskList:                 \"task-list\",\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"run-id\",\n\t\tTaskList:   \"task-list2\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\ts.mockMutableState.EXPECT().GetActivityInfo(activityInfo.ScheduleID).Return(activityInfo, true).Times(1)\n\n\tvar activityInfoUpdated = *activityInfo // make a copy\n\tactivityInfoUpdated.TimerTaskStatus = TimerTaskStatusCreatedScheduleToStart\n\ts.mockMutableState.EXPECT().UpdateActivity(&activityInfoUpdated).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(currentVersion).Times(1)\n\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.ActivityTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"run-id\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToStartTimeout) * time.Second,\n\t\t\t),\n\t\t\tVersion: currentVersion,\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeScheduleToStart),\n\t\tEventID:     activityInfo.ScheduleID,\n\t\tAttempt:     int64(activityInfo.Attempt),\n\t\tTaskList:    \"task-list\",\n\t}).Times(1)\n\n\tmodified, err := s.timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n}\n\nfunc (s *timerSequenceSuite) TestCreateNextActivityTimer_HeartbeatTimer() {\n\tnow := time.Now()\n\tcurrentVersion := int64(999)\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t\tTaskList:                 \"task-list\",\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   \"domain-id\",\n\t\tWorkflowID: \"wf-id\",\n\t\tRunID:      \"run-id\",\n\t\tTaskList:   \"task-list2\",\n\t}).Times(1)\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\ts.mockMutableState.EXPECT().GetActivityInfo(activityInfo.ScheduleID).Return(activityInfo, true).Times(1)\n\n\ttaskVisibilityTimestamp := activityInfo.StartedTime.Add(\n\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t)\n\n\tvar activityInfoUpdated = *activityInfo // make a copy\n\tactivityInfoUpdated.TimerTaskStatus = TimerTaskStatusCreatedHeartbeat\n\tactivityInfoUpdated.LastHeartbeatTimeoutVisibilityInSeconds = taskVisibilityTimestamp.Unix()\n\ts.mockMutableState.EXPECT().UpdateActivity(&activityInfoUpdated).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(currentVersion).Times(1)\n\ts.mockMutableState.EXPECT().AddTimerTasks(&persistence.ActivityTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   \"domain-id\",\n\t\t\tWorkflowID: \"wf-id\",\n\t\t\tRunID:      \"run-id\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\t// TaskID is set by shard\n\t\t\tVisibilityTimestamp: taskVisibilityTimestamp,\n\t\t\tVersion:             currentVersion,\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeHeartbeat),\n\t\tEventID:     activityInfo.ScheduleID,\n\t\tAttempt:     int64(activityInfo.Attempt),\n\t\tTaskList:    \"task-list\",\n\t}).Times(1)\n\n\tmodified, err := s.timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortUserTimers_None() {\n\ttimerInfos := map[string]*persistence.TimerInfo{}\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(timerInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortUserTimers()\n\ts.Empty(TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortUserTimers_One() {\n\tnow := time.Now()\n\ttimerInfo := &persistence.TimerInfo{\n\t\tVersion:    123,\n\t\tTimerID:    \"some random timer ID\",\n\t\tStartedID:  456,\n\t\tExpiryTime: now.Add(100 * time.Second),\n\t\tTaskStatus: TimerTaskStatusCreated,\n\t}\n\ttimerInfos := map[string]*persistence.TimerInfo{timerInfo.TimerID: timerInfo}\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(timerInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortUserTimers()\n\ts.Equal([]TimerSequenceID{{\n\t\tEventID:      timerInfo.StartedID,\n\t\tTimestamp:    timerInfo.ExpiryTime,\n\t\tTimerType:    TimerTypeStartToClose,\n\t\tTimerCreated: true,\n\t\tAttempt:      0,\n\t}}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortUserTimers_Multiple() {\n\tnow := time.Now()\n\ttimerInfo1 := &persistence.TimerInfo{\n\t\tVersion:    123,\n\t\tTimerID:    \"some random timer ID\",\n\t\tStartedID:  456,\n\t\tExpiryTime: now.Add(100 * time.Second),\n\t\tTaskStatus: TimerTaskStatusCreated,\n\t}\n\ttimerInfo2 := &persistence.TimerInfo{\n\t\tVersion:    1234,\n\t\tTimerID:    \"other random timer ID\",\n\t\tStartedID:  4567,\n\t\tExpiryTime: now.Add(200 * time.Second),\n\t\tTaskStatus: TimerTaskStatusNone,\n\t}\n\ttimerInfos := map[string]*persistence.TimerInfo{\n\t\ttimerInfo1.TimerID: timerInfo1,\n\t\ttimerInfo2.TimerID: timerInfo2,\n\t}\n\ts.mockMutableState.EXPECT().GetPendingTimerInfos().Return(timerInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortUserTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID:      timerInfo1.StartedID,\n\t\t\tTimestamp:    timerInfo1.ExpiryTime,\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      0,\n\t\t},\n\t\t{\n\t\t\tEventID:      timerInfo2.StartedID,\n\t\t\tTimestamp:    timerInfo2.ExpiryTime,\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: false,\n\t\t\tAttempt:      0,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_None() {\n\tactivityInfos := map[int64]*persistence.ActivityInfo{}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Empty(TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_NotScheduled() {\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               constants.EmptyEventID,\n\t\tScheduledTime:            time.Time{},\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Empty(TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_Scheduled_NotStarted() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedScheduleToStart,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToStartTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToStart,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_Scheduled_Started_WithHeartbeatTimeout() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedStartToClose | TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeHeartbeat,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_Scheduled_Started_WithoutHeartbeatTimeout() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedStartToClose,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_Scheduled_Started_Heartbeated_WithHeartbeatTimeout() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedStartToClose | TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.LastHeartBeatUpdatedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeHeartbeat,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_One_Scheduled_Started_Heartbeated_WithoutHeartbeatTimeout() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose | TimerTaskStatusCreatedStartToClose,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo.ScheduleID,\n\t\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: true,\n\t\t\tAttempt:      activityInfo.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestLoadAndSortActivityTimers_Multiple() {\n\tnow := time.Now()\n\tactivityInfo1 := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\tactivityInfo2 := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               2345,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"other random activity ID\",\n\t\tScheduleToStartTimeout:   11,\n\t\tScheduleToCloseTimeout:   1001,\n\t\tStartToCloseTimeout:      101,\n\t\tHeartbeatTimeout:         6,\n\t\tLastHeartBeatUpdatedTime: now.Add(800 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  21,\n\t}\n\tactivityInfos := map[int64]*persistence.ActivityInfo{\n\t\tactivityInfo1.ScheduleID: activityInfo1,\n\t\tactivityInfo2.ScheduleID: activityInfo2,\n\t}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).Times(1)\n\n\tTimerSequenceIDs := s.timerSequence.LoadAndSortActivityTimers()\n\ts.Equal([]TimerSequenceID{\n\t\t{\n\t\t\tEventID: activityInfo2.ScheduleID,\n\t\t\tTimestamp: activityInfo2.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo2.ScheduleToStartTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToStart,\n\t\t\tTimerCreated: false,\n\t\t\tAttempt:      activityInfo2.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo1.ScheduleID,\n\t\t\tTimestamp: activityInfo1.StartedTime.Add(\n\t\t\t\ttime.Duration(activityInfo1.StartToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeStartToClose,\n\t\t\tTimerCreated: false,\n\t\t\tAttempt:      activityInfo1.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo1.ScheduleID,\n\t\t\tTimestamp: activityInfo1.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo1.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: false,\n\t\t\tAttempt:      activityInfo1.Attempt,\n\t\t},\n\t\t{\n\t\t\tEventID: activityInfo2.ScheduleID,\n\t\t\tTimestamp: activityInfo2.ScheduledTime.Add(\n\t\t\t\ttime.Duration(activityInfo2.ScheduleToCloseTimeout) * time.Second,\n\t\t\t),\n\t\t\tTimerType:    TimerTypeScheduleToClose,\n\t\t\tTimerCreated: false,\n\t\t\tAttempt:      activityInfo2.Attempt,\n\t\t},\n\t}, TimerSequenceIDs)\n}\n\nfunc (s *timerSequenceSuite) TestGetUserTimerTimeout() {\n\tnow := time.Now()\n\ttimerInfo := &persistence.TimerInfo{\n\t\tVersion:    123,\n\t\tTimerID:    \"some random timer ID\",\n\t\tStartedID:  456,\n\t\tExpiryTime: now.Add(100 * time.Second),\n\t\tTaskStatus: TimerTaskStatusCreated,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID:      timerInfo.StartedID,\n\t\tTimestamp:    timerInfo.ExpiryTime,\n\t\tTimerType:    TimerTypeStartToClose,\n\t\tTimerCreated: true,\n\t\tAttempt:      0,\n\t}\n\n\ttimerSequence := s.timerSequence.getUserTimerTimeout(timerInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\ttimerInfo.TaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getUserTimerTimeout(timerInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityScheduleToStartTimeout_NotScheduled() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               constants.EmptyEventID,\n\t\tScheduledTime:            time.Time{},\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityScheduleToStartTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityScheduleToStartTimeout_Scheduled_NotStarted() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToStart,\n\t\tAttempt:                  12,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID: activityInfo.ScheduleID,\n\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\ttime.Duration(activityInfo.ScheduleToStartTimeout) * time.Second,\n\t\t),\n\t\tTimerType:    TimerTypeScheduleToStart,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityScheduleToStartTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getActivityScheduleToStartTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityScheduleToStartTimeout_Scheduled_Started() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Second),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToStart,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityScheduleToStartTimeout(activityInfo)\n\ts.Empty(timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\ttimerSequence = s.timerSequence.getActivityScheduleToStartTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityScheduleToCloseTimeout_NotScheduled() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               constants.EmptyEventID,\n\t\tScheduledTime:            time.Time{},\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityScheduleToCloseTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityScheduleToCloseTimeout_Scheduled() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedScheduleToClose,\n\t\tAttempt:                  12,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID: activityInfo.ScheduleID,\n\t\tTimestamp: activityInfo.ScheduledTime.Add(\n\t\t\ttime.Duration(activityInfo.ScheduleToCloseTimeout) * time.Second,\n\t\t),\n\t\tTimerType:    TimerTypeScheduleToClose,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityScheduleToCloseTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getActivityScheduleToCloseTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityStartToCloseTimeout_NotStarted() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityStartToCloseTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityStartToCloseTimeout_Started() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedStartToClose,\n\t\tAttempt:                  12,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID: activityInfo.ScheduleID,\n\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\ttime.Duration(activityInfo.StartToCloseTimeout) * time.Second,\n\t\t),\n\t\tTimerType:    TimerTypeStartToClose,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityStartToCloseTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getActivityStartToCloseTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithHeartbeat_NotStarted() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithHeartbeat_Started_NoHeartbeat() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID: activityInfo.ScheduleID,\n\t\tTimestamp: activityInfo.StartedTime.Add(\n\t\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t\t),\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithHeartbeat_Started_Heartbeated() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         1,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\n\texpectedTimerSequence := &TimerSequenceID{\n\t\tEventID: activityInfo.ScheduleID,\n\t\tTimestamp: activityInfo.LastHeartBeatUpdatedTime.Add(\n\t\t\ttime.Duration(activityInfo.HeartbeatTimeout) * time.Second,\n\t\t),\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\texpectedTimerSequence.TimerCreated = false\n\ttimerSequence = s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Equal(expectedTimerSequence, timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithoutHeartbeat_NotStarted() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                constants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusNone,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithoutHeartbeat_Started_NoHeartbeat() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\ttimerSequence = s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestGetActivityHeartbeatTimeout_WithoutHeartbeat_Started_Heartbeated() {\n\tnow := time.Now()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:                  123,\n\t\tScheduleID:               234,\n\t\tScheduledTime:            now,\n\t\tStartedID:                345,\n\t\tStartedTime:              now.Add(200 * time.Millisecond),\n\t\tActivityID:               \"some random activity ID\",\n\t\tScheduleToStartTimeout:   10,\n\t\tScheduleToCloseTimeout:   1000,\n\t\tStartToCloseTimeout:      100,\n\t\tHeartbeatTimeout:         0,\n\t\tLastHeartBeatUpdatedTime: now.Add(400 * time.Millisecond),\n\t\tTimerTaskStatus:          TimerTaskStatusCreatedHeartbeat,\n\t\tAttempt:                  12,\n\t}\n\n\ttimerSequence := s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n\n\tactivityInfo.TimerTaskStatus = TimerTaskStatusNone\n\ttimerSequence = s.timerSequence.getActivityHeartbeatTimeout(activityInfo)\n\ts.Empty(timerSequence)\n}\n\nfunc (s *timerSequenceSuite) TestConversion() {\n\ts.Equal(types.TimeoutTypeStartToClose, TimerTypeToInternal(TimerTypeStartToClose))\n\ts.Equal(types.TimeoutTypeScheduleToStart, TimerTypeToInternal(TimerTypeScheduleToStart))\n\ts.Equal(types.TimeoutTypeScheduleToClose, TimerTypeToInternal(TimerTypeScheduleToClose))\n\ts.Equal(types.TimeoutTypeHeartbeat, TimerTypeToInternal(TimerTypeHeartbeat))\n\n\ts.Equal(TimerTypeFromInternal(types.TimeoutTypeStartToClose), TimerTypeStartToClose)\n\ts.Equal(TimerTypeFromInternal(types.TimeoutTypeScheduleToStart), TimerTypeScheduleToStart)\n\ts.Equal(TimerTypeFromInternal(types.TimeoutTypeScheduleToClose), TimerTypeScheduleToClose)\n\ts.Equal(TimerTypeFromInternal(types.TimeoutTypeHeartbeat), TimerTypeHeartbeat)\n\n\ts.Equal(int32(TimerTaskStatusCreatedStartToClose), TimerTypeToTimerMask(TimerTypeStartToClose))\n\ts.Equal(int32(TimerTaskStatusCreatedScheduleToStart), TimerTypeToTimerMask(TimerTypeScheduleToStart))\n\ts.Equal(int32(TimerTaskStatusCreatedScheduleToClose), TimerTypeToTimerMask(TimerTypeScheduleToClose))\n\ts.Equal(int32(TimerTaskStatusCreatedHeartbeat), TimerTypeToTimerMask(TimerTypeHeartbeat))\n\n\ts.Equal(TimerTaskStatusNone, 0)\n\ts.Equal(TimerTaskStatusCreated, 1)\n\ts.Equal(TimerTaskStatusCreatedStartToClose, 1)\n\ts.Equal(TimerTaskStatusCreatedScheduleToStart, 2)\n\ts.Equal(TimerTaskStatusCreatedScheduleToClose, 4)\n\ts.Equal(TimerTaskStatusCreatedHeartbeat, 8)\n}\n\nfunc (s *timerSequenceSuite) TestLess_CompareTime() {\n\tnow := time.Now()\n\ttimerSequenceID1 := TimerSequenceID{\n\t\tEventID:      123,\n\t\tTimestamp:    now,\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequenceID2 := TimerSequenceID{\n\t\tEventID:      123,\n\t\tTimestamp:    now.Add(time.Second),\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\tTimerSequenceIDs := TimerSequenceIDs([]TimerSequenceID{timerSequenceID1, timerSequenceID2})\n\ts.True(TimerSequenceIDs.Less(0, 1))\n\ts.False(TimerSequenceIDs.Less(1, 0))\n}\n\nfunc (s *timerSequenceSuite) TestLess_CompareEventID() {\n\tnow := time.Now()\n\ttimerSequenceID1 := TimerSequenceID{\n\t\tEventID:      122,\n\t\tTimestamp:    now,\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequenceID2 := TimerSequenceID{\n\t\tEventID:      123,\n\t\tTimestamp:    now,\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\tTimerSequenceIDs := TimerSequenceIDs([]TimerSequenceID{timerSequenceID1, timerSequenceID2})\n\ts.True(TimerSequenceIDs.Less(0, 1))\n\ts.False(TimerSequenceIDs.Less(1, 0))\n}\n\nfunc (s *timerSequenceSuite) TestLess_CompareType() {\n\tnow := time.Now()\n\ttimerSequenceID1 := TimerSequenceID{\n\t\tEventID:      123,\n\t\tTimestamp:    now,\n\t\tTimerType:    TimerTypeScheduleToClose,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\ttimerSequenceID2 := TimerSequenceID{\n\t\tEventID:      123,\n\t\tTimestamp:    now,\n\t\tTimerType:    TimerTypeHeartbeat,\n\t\tTimerCreated: true,\n\t\tAttempt:      12,\n\t}\n\n\tTimerSequenceIDs := TimerSequenceIDs([]TimerSequenceID{timerSequenceID1, timerSequenceID2})\n\ts.True(TimerSequenceIDs.Less(0, 1))\n\ts.False(TimerSequenceIDs.Less(1, 0))\n}\n"
  },
  {
    "path": "service/history/execution/workflow.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination workflow_mock.go -self_package github.com/uber/cadence/service/history/execution\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// IdentityHistoryService is the service role identity\n\tIdentityHistoryService = \"history-service\"\n\t// WorkflowTerminationIdentity is the component which decides to terminate the workflow\n\tWorkflowTerminationIdentity = \"worker-service\"\n\t// WorkflowTerminationReason is the reason for terminating workflow due to version conflict\n\tWorkflowTerminationReason = \"Terminate Workflow Due To Version Conflict.\"\n)\n\ntype (\n\t// Workflow is the interface for NDC workflow\n\tWorkflow interface {\n\t\tGetContext() Context\n\t\tGetMutableState() MutableState\n\t\tGetReleaseFn() ReleaseFunc\n\t\tGetVectorClock() (WorkflowVectorClock, error)\n\t\tHappensAfter(that Workflow) (bool, error)\n\t\tRevive() error\n\t\tSuppressBy(incomingWorkflow Workflow) (TransactionPolicy, error)\n\t\tFlushBufferedEvents() error\n\t}\n\n\tworkflowImpl struct {\n\t\tlogger          log.Logger\n\t\tclusterMetadata cluster.Metadata\n\n\t\tctx          context.Context\n\t\tcontext      Context\n\t\tmutableState MutableState\n\t\treleaseFn    ReleaseFunc\n\t}\n\n\tWorkflowVectorClock struct {\n\t\tActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy\n\t\tLastWriteVersion             int64\n\t\tLastEventTaskID              int64\n\t\tStartTimestamp               time.Time\n\t\tRunID                        string\n\t}\n)\n\n// NewWorkflow creates a new NDC workflow\nfunc NewWorkflow(\n\tctx context.Context,\n\tclusterMetadata cluster.Metadata,\n\tcontext Context,\n\tmutableState MutableState,\n\treleaseFn ReleaseFunc,\n\tlogger log.Logger,\n) Workflow {\n\n\treturn &workflowImpl{\n\t\tctx:             ctx,\n\t\tclusterMetadata: clusterMetadata,\n\t\tlogger:          logger,\n\t\tcontext:         context,\n\t\tmutableState:    mutableState,\n\t\treleaseFn:       releaseFn,\n\t}\n}\n\nfunc (r *workflowImpl) GetContext() Context {\n\treturn r.context\n}\n\nfunc (r *workflowImpl) GetMutableState() MutableState {\n\treturn r.mutableState\n}\n\nfunc (r *workflowImpl) GetReleaseFn() ReleaseFunc {\n\treturn r.releaseFn\n}\n\nfunc (r *workflowImpl) GetVectorClock() (WorkflowVectorClock, error) {\n\n\tlastWriteVersion, err := r.mutableState.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn WorkflowVectorClock{}, err\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\treturn WorkflowVectorClock{\n\t\tActiveClusterSelectionPolicy: executionInfo.ActiveClusterSelectionPolicy,\n\t\tLastWriteVersion:             lastWriteVersion,\n\t\tLastEventTaskID:              executionInfo.LastEventTaskID,\n\t\tStartTimestamp:               executionInfo.StartTimestamp,\n\t\tRunID:                        executionInfo.RunID,\n\t}, nil\n}\n\nfunc (r *workflowImpl) HappensAfter(\n\tthat Workflow,\n) (bool, error) {\n\n\tthisVectorClock, err := r.GetVectorClock()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tthatVectorClock, err := that.GetVectorClock()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\treturn workflowHappensAfter(\n\t\tthisVectorClock,\n\t\tthatVectorClock,\n\t), nil\n}\n\nfunc (r *workflowImpl) Revive() error {\n\n\tstate, _ := r.mutableState.GetWorkflowStateCloseStatus()\n\tif state != persistence.WorkflowStateZombie {\n\t\treturn nil\n\t} else if state == persistence.WorkflowStateCompleted {\n\t\t// workflow already finished\n\t\treturn nil\n\t}\n\n\t// workflow is in zombie state, need to set the state correctly accordingly\n\tstate = persistence.WorkflowStateCreated\n\tif r.mutableState.HasProcessedOrPendingDecision() {\n\t\tstate = persistence.WorkflowStateRunning\n\t}\n\treturn r.mutableState.UpdateWorkflowStateCloseStatus(\n\t\tstate,\n\t\tpersistence.WorkflowCloseStatusNone,\n\t)\n}\n\nfunc (r *workflowImpl) SuppressBy(\n\tincomingWorkflow Workflow,\n) (TransactionPolicy, error) {\n\n\t// NOTE: READ BEFORE MODIFICATION\n\t//\n\t// if the workflow to be suppressed has last write version being local active\n\t//  then use active logic to terminate this workflow\n\t// if the workflow to be suppressed has last write version being remote active\n\t//  then turn this workflow into a zombie\n\n\tcurrentVectorClock, err := r.GetVectorClock()\n\tif err != nil {\n\t\treturn TransactionPolicyActive, err\n\t}\n\tincomingVectorClock, err := incomingWorkflow.GetVectorClock()\n\tif err != nil {\n\t\treturn TransactionPolicyActive, err\n\t}\n\n\tif workflowHappensAfter(\n\t\tcurrentVectorClock,\n\t\tincomingVectorClock,\n\t) {\n\t\treturn TransactionPolicyActive, &types.InternalServiceError{\n\t\t\tMessage: \"nDCWorkflow cannot suppress workflow by older workflow\",\n\t\t}\n\t}\n\n\t// if workflow is in zombie or finished state, keep as is\n\tif !r.mutableState.IsWorkflowExecutionRunning() {\n\t\treturn TransactionPolicyPassive, nil\n\t}\n\n\tlastWriteCluster, err := r.clusterMetadata.ClusterNameForFailoverVersion(currentVectorClock.LastWriteVersion)\n\tif err != nil {\n\t\treturn TransactionPolicyActive, err\n\t}\n\tcurrentCluster := r.clusterMetadata.GetCurrentClusterName()\n\n\tif currentCluster == lastWriteCluster {\n\t\treturn TransactionPolicyActive, r.terminateWorkflow(currentVectorClock.LastWriteVersion, incomingVectorClock.LastWriteVersion, WorkflowTerminationReason)\n\t}\n\treturn TransactionPolicyPassive, r.zombiefyWorkflow()\n}\n\nfunc (r *workflowImpl) FlushBufferedEvents() error {\n\n\tif !r.mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tif !r.mutableState.HasBufferedEvents() {\n\t\treturn nil\n\t}\n\n\tcurrentVectorClock, err := r.GetVectorClock()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tlastWriteCluster, err := r.clusterMetadata.ClusterNameForFailoverVersion(currentVectorClock.LastWriteVersion)\n\tif err != nil {\n\t\t// TODO: add a test for this\n\t\treturn err\n\t}\n\tcurrentCluster := r.clusterMetadata.GetCurrentClusterName()\n\n\tif lastWriteCluster != currentCluster {\n\t\t// TODO: add a test for this\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"nDCWorkflow encounter workflow with buffered events but last write not from current cluster\",\n\t\t}\n\t}\n\n\treturn r.failDecision(currentVectorClock.LastWriteVersion, true)\n}\n\nfunc (r *workflowImpl) failDecision(\n\tlastWriteVersion int64,\n\tscheduleNewDecision bool,\n) error {\n\n\t// do not persist the change right now, NDC requires transaction\n\tr.logger.Debugf(\"failDecision calling UpdateCurrentVersion for domain %s, wfID %v, lastWriteVersion %v\",\n\t\tr.mutableState.GetExecutionInfo().DomainID, r.mutableState.GetExecutionInfo().WorkflowID, lastWriteVersion)\n\tif err := r.mutableState.UpdateCurrentVersion(lastWriteVersion, true); err != nil {\n\t\treturn err\n\t}\n\n\tdecision, ok := r.mutableState.GetInFlightDecision()\n\tif !ok {\n\t\treturn nil\n\t}\n\n\tif err := FailDecision(r.mutableState, decision, types.DecisionTaskFailedCauseFailoverCloseDecision); err != nil {\n\t\treturn err\n\t}\n\tif scheduleNewDecision {\n\t\treturn ScheduleDecision(r.mutableState)\n\t}\n\treturn nil\n}\n\nfunc (r *workflowImpl) terminateWorkflow(\n\tlastWriteVersion int64,\n\tincomingLastWriteVersion int64,\n\tterminationReason string,\n) error {\n\n\teventBatchFirstEventID := r.GetMutableState().GetNextEventID()\n\tif err := r.failDecision(lastWriteVersion, false); err != nil {\n\t\treturn err\n\t}\n\n\t// do not persist the change right now, NDC requires transaction\n\tr.logger.Debugf(\"terminateWorkflow calling UpdateCurrentVersion for domain %s, wfID %v, lastWriteVersion %v\",\n\t\tr.mutableState.GetExecutionInfo().DomainID, r.mutableState.GetExecutionInfo().WorkflowID, lastWriteVersion)\n\tif err := r.mutableState.UpdateCurrentVersion(lastWriteVersion, true); err != nil {\n\t\treturn err\n\t}\n\n\t_, err := r.mutableState.AddWorkflowExecutionTerminatedEvent(\n\t\teventBatchFirstEventID,\n\t\tterminationReason,\n\t\t[]byte(fmt.Sprintf(\"terminated by version: %v\", incomingLastWriteVersion)),\n\t\tWorkflowTerminationIdentity,\n\t)\n\n\treturn err\n}\n\nfunc (r *workflowImpl) zombiefyWorkflow() error {\n\n\treturn r.mutableState.GetExecutionInfo().UpdateWorkflowStateCloseStatus(\n\t\tpersistence.WorkflowStateZombie,\n\t\tpersistence.WorkflowCloseStatusNone,\n\t)\n}\n\n// Conflict resolution for workflow replication requires determining which workflow event is \"newer.\"\n// This decision must be made deterministically across all clusters without coordination,\n// ensuring that each cluster independently resolves conflicts to the same final state.\n//\n// Active-Passive Domains:\n//   - These domains do not use active cluster selection policies.\n//   - The event with the larger failover version is considered newer, since the failover version is\n//     a monotonically increasing logical clock for the domain.\n//   - After a failover, any event with a higher failover version wins conflict resolution.\n//   - Conflicts between events with the same failover version should not occur, as they originate\n//     from the same active cluster. In case of a tie, the replication task event ID is used as a tiebreaker.\n//\n// Active-Active Domains (Same Selection Policy):\n//   - Treated the same as active-passive domains.\n//   - The event generated after failover (larger failover version) wins.\n//\n// Active-Active Domains (Different Selection Policies):\n//   - These represent workflows started concurrently in different active clusters.\n//   - The event with the larger start timestamp is considered newer.\n//   - Clock skew between clusters is expected to be small, making this a reasonable rule.\n//   - In case of a tie on start time, the RunID is used as a final tiebreaker.\nfunc workflowHappensAfter(\n\tthisVectorClock WorkflowVectorClock,\n\tthatVectorClock WorkflowVectorClock,\n) bool {\n\tif !thisVectorClock.ActiveClusterSelectionPolicy.Equals(thatVectorClock.ActiveClusterSelectionPolicy) {\n\t\tif !thisVectorClock.StartTimestamp.Equal(thatVectorClock.StartTimestamp) {\n\t\t\treturn thatVectorClock.StartTimestamp.Before(thisVectorClock.StartTimestamp)\n\t\t}\n\t\treturn thisVectorClock.RunID > thatVectorClock.RunID\n\t}\n\n\tif thisVectorClock.LastWriteVersion != thatVectorClock.LastWriteVersion {\n\t\treturn thisVectorClock.LastWriteVersion > thatVectorClock.LastWriteVersion\n\t}\n\n\t// thisVectorClock.LastWriteVersion == thatVectorClock.LastWriteVersion\n\treturn thisVectorClock.LastEventTaskID > thatVectorClock.LastEventTaskID\n}\n"
  },
  {
    "path": "service/history/execution/workflow_execution_util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport \"github.com/uber/cadence/common/types\"\n\n// TerminateWorkflow is a helper function to terminate workflow\nfunc TerminateWorkflow(\n\tmutableState MutableState,\n\teventBatchFirstEventID int64,\n\tterminateReason string,\n\tterminateDetails []byte,\n\tterminateIdentity string,\n) error {\n\n\tif decision, ok := mutableState.GetInFlightDecision(); ok {\n\t\tif err := FailDecision(\n\t\t\tmutableState,\n\t\t\tdecision,\n\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t_, err := mutableState.AddWorkflowExecutionTerminatedEvent(\n\t\teventBatchFirstEventID,\n\t\tterminateReason,\n\t\tterminateDetails,\n\t\tterminateIdentity,\n\t)\n\treturn err\n}\n"
  },
  {
    "path": "service/history/execution/workflow_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: workflow.go\n//\n// Generated by this command:\n//\n//\tmockgen -package execution -source workflow.go -destination workflow_mock.go -self_package github.com/uber/cadence/service/history/execution\n//\n\n// Package execution is a generated GoMock package.\npackage execution\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockWorkflow is a mock of Workflow interface.\ntype MockWorkflow struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWorkflowMockRecorder\n\tisgomock struct{}\n}\n\n// MockWorkflowMockRecorder is the mock recorder for MockWorkflow.\ntype MockWorkflowMockRecorder struct {\n\tmock *MockWorkflow\n}\n\n// NewMockWorkflow creates a new mock instance.\nfunc NewMockWorkflow(ctrl *gomock.Controller) *MockWorkflow {\n\tmock := &MockWorkflow{ctrl: ctrl}\n\tmock.recorder = &MockWorkflowMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWorkflow) EXPECT() *MockWorkflowMockRecorder {\n\treturn m.recorder\n}\n\n// FlushBufferedEvents mocks base method.\nfunc (m *MockWorkflow) FlushBufferedEvents() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"FlushBufferedEvents\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// FlushBufferedEvents indicates an expected call of FlushBufferedEvents.\nfunc (mr *MockWorkflowMockRecorder) FlushBufferedEvents() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FlushBufferedEvents\", reflect.TypeOf((*MockWorkflow)(nil).FlushBufferedEvents))\n}\n\n// GetContext mocks base method.\nfunc (m *MockWorkflow) GetContext() Context {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetContext\")\n\tret0, _ := ret[0].(Context)\n\treturn ret0\n}\n\n// GetContext indicates an expected call of GetContext.\nfunc (mr *MockWorkflowMockRecorder) GetContext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetContext\", reflect.TypeOf((*MockWorkflow)(nil).GetContext))\n}\n\n// GetMutableState mocks base method.\nfunc (m *MockWorkflow) GetMutableState() MutableState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMutableState\")\n\tret0, _ := ret[0].(MutableState)\n\treturn ret0\n}\n\n// GetMutableState indicates an expected call of GetMutableState.\nfunc (mr *MockWorkflowMockRecorder) GetMutableState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMutableState\", reflect.TypeOf((*MockWorkflow)(nil).GetMutableState))\n}\n\n// GetReleaseFn mocks base method.\nfunc (m *MockWorkflow) GetReleaseFn() ReleaseFunc {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReleaseFn\")\n\tret0, _ := ret[0].(ReleaseFunc)\n\treturn ret0\n}\n\n// GetReleaseFn indicates an expected call of GetReleaseFn.\nfunc (mr *MockWorkflowMockRecorder) GetReleaseFn() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReleaseFn\", reflect.TypeOf((*MockWorkflow)(nil).GetReleaseFn))\n}\n\n// GetVectorClock mocks base method.\nfunc (m *MockWorkflow) GetVectorClock() (WorkflowVectorClock, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVectorClock\")\n\tret0, _ := ret[0].(WorkflowVectorClock)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetVectorClock indicates an expected call of GetVectorClock.\nfunc (mr *MockWorkflowMockRecorder) GetVectorClock() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVectorClock\", reflect.TypeOf((*MockWorkflow)(nil).GetVectorClock))\n}\n\n// HappensAfter mocks base method.\nfunc (m *MockWorkflow) HappensAfter(that Workflow) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HappensAfter\", that)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// HappensAfter indicates an expected call of HappensAfter.\nfunc (mr *MockWorkflowMockRecorder) HappensAfter(that any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HappensAfter\", reflect.TypeOf((*MockWorkflow)(nil).HappensAfter), that)\n}\n\n// Revive mocks base method.\nfunc (m *MockWorkflow) Revive() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Revive\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Revive indicates an expected call of Revive.\nfunc (mr *MockWorkflowMockRecorder) Revive() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Revive\", reflect.TypeOf((*MockWorkflow)(nil).Revive))\n}\n\n// SuppressBy mocks base method.\nfunc (m *MockWorkflow) SuppressBy(incomingWorkflow Workflow) (TransactionPolicy, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SuppressBy\", incomingWorkflow)\n\tret0, _ := ret[0].(TransactionPolicy)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SuppressBy indicates an expected call of SuppressBy.\nfunc (mr *MockWorkflowMockRecorder) SuppressBy(incomingWorkflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SuppressBy\", reflect.TypeOf((*MockWorkflow)(nil).SuppressBy), incomingWorkflow)\n}\n"
  },
  {
    "path": "service/history/execution/workflow_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage execution\n\nimport (\n\t\"context\"\n\t\"reflect\"\n\t\"runtime\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tworkflowSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockContext      *MockContext\n\t\tmockMutableState *MockMutableState\n\n\t\tdomainID   string\n\t\tdomainName string\n\t\tworkflowID string\n\t\trunID      string\n\t}\n)\n\nfunc TestWorkflowSuite(t *testing.T) {\n\ts := new(workflowSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *workflowSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockContext = NewMockContext(s.controller)\n\ts.mockMutableState = NewMockMutableState(s.controller)\n\ts.domainID = uuid.New()\n\ts.domainName = \"domain-name\"\n\ts.workflowID = \"some random workflow ID\"\n\ts.runID = uuid.New()\n}\n\nfunc (s *workflowSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *workflowSuite) TestGetMethods() {\n\tlastEventTaskID := int64(144)\n\tlastEventVersion := int64(12)\n\tstartTimestamp := time.Now()\n\tactiveClusterSelectionPolicy := &types.ActiveClusterSelectionPolicy{\n\t\tActiveClusterSelectionStrategy: types.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\tStickyRegion:                   \"region-1\",\n\t}\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastEventVersion, nil).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:                     s.domainID,\n\t\tWorkflowID:                   s.workflowID,\n\t\tRunID:                        s.runID,\n\t\tLastEventTaskID:              lastEventTaskID,\n\t\tStartTimestamp:               startTimestamp,\n\t\tActiveClusterSelectionPolicy: activeClusterSelectionPolicy,\n\t}).AnyTimes()\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\n\ts.Equal(s.mockContext, nDCWorkflow.GetContext())\n\ts.Equal(s.mockMutableState, nDCWorkflow.GetMutableState())\n\t// NOTE golang does not seem to let people compare functions, easily\n\t//  link: https://github.com/stretchr/testify/issues/182\n\t// this is a hack to compare 2 functions, being the same\n\texpectedReleaseFn := runtime.FuncForPC(reflect.ValueOf(NoopReleaseFn).Pointer()).Name()\n\tactualReleaseFn := runtime.FuncForPC(reflect.ValueOf(nDCWorkflow.GetReleaseFn()).Pointer()).Name()\n\ts.Equal(expectedReleaseFn, actualReleaseFn)\n\tvectorClock, err := nDCWorkflow.GetVectorClock()\n\ts.NoError(err)\n\texpectedVectorClock := WorkflowVectorClock{\n\t\tActiveClusterSelectionPolicy: activeClusterSelectionPolicy,\n\t\tLastWriteVersion:             lastEventVersion,\n\t\tLastEventTaskID:              lastEventTaskID,\n\t\tStartTimestamp:               startTimestamp,\n\t\tRunID:                        s.runID,\n\t}\n\ts.Equal(expectedVectorClock, vectorClock)\n}\n\nfunc (s *workflowSuite) TestSuppressWorkflowBy_Error() {\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\n\tincomingMockContext := NewMockContext(s.controller)\n\tincomingMockMutableState := NewMockMutableState(s.controller)\n\tincomingNDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\tincomingMockContext,\n\t\tincomingMockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\n\t// cannot suppress by older workflow\n\tlastEventTaskID := int64(144)\n\tlastEventVersion := int64(12)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastEventVersion, nil).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           s.runID,\n\t\tLastEventTaskID: lastEventTaskID,\n\t}).AnyTimes()\n\n\tincomingRunID := uuid.New()\n\tincomingLastEventTaskID := int64(144)\n\tincomingLastEventVersion := lastEventVersion - 1\n\tincomingMockMutableState.EXPECT().GetLastWriteVersion().Return(incomingLastEventVersion, nil).AnyTimes()\n\tincomingMockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           incomingRunID,\n\t\tLastEventTaskID: incomingLastEventTaskID,\n\t}).AnyTimes()\n\n\t_, err := nDCWorkflow.SuppressBy(incomingNDCWorkflow)\n\ts.Error(err)\n}\n\nfunc (s *workflowSuite) TestSuppressWorkflowBy_Terminate() {\n\tlastEventID := int64(2)\n\tlastEventTaskID := int64(144)\n\tlastEventVersion := cluster.TestCurrentClusterInitialFailoverVersion\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(lastEventID + 1).AnyTimes()\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastEventVersion, nil).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           s.runID,\n\t\tLastEventTaskID: lastEventTaskID,\n\t}).AnyTimes()\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\n\tincomingRunID := uuid.New()\n\tincomingLastEventTaskID := int64(144)\n\tincomingLastEventVersion := lastEventVersion + 1\n\tincomingMockContext := NewMockContext(s.controller)\n\tincomingMockMutableState := NewMockMutableState(s.controller)\n\tincomingNDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\tincomingMockContext,\n\t\tincomingMockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\tincomingMockMutableState.EXPECT().GetLastWriteVersion().Return(incomingLastEventVersion, nil).AnyTimes()\n\tincomingMockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           incomingRunID,\n\t\tLastEventTaskID: incomingLastEventTaskID,\n\t}).AnyTimes()\n\n\ts.mockMutableState.EXPECT().UpdateCurrentVersion(lastEventVersion, true).Return(nil).AnyTimes()\n\tinFlightDecision := &DecisionInfo{\n\t\tVersion:    1234,\n\t\tScheduleID: 5678,\n\t\tStartedID:  9012,\n\t}\n\ts.mockMutableState.EXPECT().GetInFlightDecision().Return(inFlightDecision, true).Times(1)\n\ts.mockMutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\tinFlightDecision.ScheduleID,\n\t\tinFlightDecision.StartedID,\n\t\ttypes.DecisionTaskFailedCauseFailoverCloseDecision,\n\t\t[]byte(nil),\n\t\tIdentityHistoryService,\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\tint64(0),\n\t\t\"\",\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\ts.mockMutableState.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\n\ts.mockMutableState.EXPECT().AddWorkflowExecutionTerminatedEvent(\n\t\tlastEventID+1, WorkflowTerminationReason, gomock.Any(), WorkflowTerminationIdentity,\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\n\t// if workflow is in zombie or finished state, keep as is\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).Times(1)\n\tpolicy, err := nDCWorkflow.SuppressBy(incomingNDCWorkflow)\n\ts.NoError(err)\n\ts.Equal(TransactionPolicyPassive, policy)\n\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).Times(1)\n\tpolicy, err = nDCWorkflow.SuppressBy(incomingNDCWorkflow)\n\ts.NoError(err)\n\ts.Equal(TransactionPolicyActive, policy)\n}\n\nfunc (s *workflowSuite) TestSuppressWorkflowBy_Zombiefy() {\n\tlastEventTaskID := int64(144)\n\tlastEventVersion := cluster.TestAlternativeClusterInitialFailoverVersion\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastEventVersion, nil).AnyTimes()\n\texecutionInfo := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           s.runID,\n\t\tLastEventTaskID: lastEventTaskID,\n\t\tState:           persistence.WorkflowStateRunning,\n\t\tCloseStatus:     persistence.WorkflowCloseStatusNone,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(executionInfo).AnyTimes()\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\n\tincomingRunID := uuid.New()\n\tincomingLastEventTaskID := int64(144)\n\tincomingLastEventVersion := lastEventVersion + 1\n\tincomingMockContext := NewMockContext(s.controller)\n\tincomingMockMutableState := NewMockMutableState(s.controller)\n\tincomingNDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\tincomingMockContext,\n\t\tincomingMockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\tincomingMockMutableState.EXPECT().GetLastWriteVersion().Return(incomingLastEventVersion, nil).AnyTimes()\n\tincomingMockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           incomingRunID,\n\t\tLastEventTaskID: incomingLastEventTaskID,\n\t}).AnyTimes()\n\n\t// if workflow is in zombie or finished state, keep as is\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).Times(1)\n\tpolicy, err := nDCWorkflow.SuppressBy(incomingNDCWorkflow)\n\ts.NoError(err)\n\ts.Equal(TransactionPolicyPassive, policy)\n\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).Times(1)\n\tpolicy, err = nDCWorkflow.SuppressBy(incomingNDCWorkflow)\n\ts.NoError(err)\n\ts.Equal(TransactionPolicyPassive, policy)\n\ts.Equal(persistence.WorkflowStateZombie, executionInfo.State)\n\ts.Equal(persistence.WorkflowCloseStatusNone, executionInfo.CloseStatus)\n}\n\nfunc (s *workflowSuite) TestRevive_Zombie_Error() {\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateZombie, persistence.WorkflowCloseStatusNone).Times(1)\n\ts.mockMutableState.EXPECT().HasProcessedOrPendingDecision().Return(true).Times(1)\n\ts.mockMutableState.EXPECT().UpdateWorkflowStateCloseStatus(persistence.WorkflowStateRunning, persistence.WorkflowCloseStatusNone).Return(&types.InternalServiceError{Message: \"error\"}).Times(1)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.Revive()\n\ts.Error(err)\n}\n\nfunc (s *workflowSuite) TestRevive_Zombie_Success() {\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateZombie, persistence.WorkflowCloseStatusNone).Times(1)\n\ts.mockMutableState.EXPECT().HasProcessedOrPendingDecision().Return(true).Times(1)\n\ts.mockMutableState.EXPECT().UpdateWorkflowStateCloseStatus(persistence.WorkflowStateRunning, persistence.WorkflowCloseStatusNone).Return(nil).Times(1)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.Revive()\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestRevive_NonZombie_Success() {\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateCompleted, persistence.WorkflowCloseStatusNone).Times(1)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.Revive()\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestFlushBufferedEvents_Success() {\n\tlastWriteVersion := cluster.TestCurrentClusterInitialFailoverVersion\n\tlastEventTaskID := int64(144)\n\tdecision := &DecisionInfo{\n\t\tScheduleID: 1,\n\t\tStartedID:  2,\n\t}\n\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(true)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastWriteVersion, nil)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{LastEventTaskID: lastEventTaskID}).AnyTimes()\n\ts.mockMutableState.EXPECT().UpdateCurrentVersion(lastWriteVersion, true).Return(nil)\n\ts.mockMutableState.EXPECT().GetInFlightDecision().Return(decision, true)\n\ts.mockMutableState.EXPECT().AddDecisionTaskFailedEvent(decision.ScheduleID, decision.StartedID, types.DecisionTaskFailedCauseFailoverCloseDecision, nil, IdentityHistoryService, \"\", \"\", \"\", \"\", int64(0), \"\").Return(&types.HistoryEvent{}, nil)\n\ts.mockMutableState.EXPECT().FlushBufferedEvents().Return(nil)\n\ts.mockMutableState.EXPECT().HasPendingDecision().Return(false)\n\ts.mockMutableState.EXPECT().AddDecisionTaskScheduledEvent(false).Return(&DecisionInfo{}, nil)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.FlushBufferedEvents()\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestFlushBufferedEvents_NoBuffer_Success() {\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(false)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.FlushBufferedEvents()\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestFlushBufferedEvents_NoDecision_Success() {\n\tlastWriteVersion := cluster.TestCurrentClusterInitialFailoverVersion\n\tlastEventTaskID := int64(144)\n\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(true)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastWriteVersion, nil)\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(\n\t\t&persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:        s.domainID,\n\t\t\tWorkflowID:      s.workflowID,\n\t\t\tRunID:           s.runID,\n\t\t\tLastEventTaskID: lastEventTaskID,\n\t\t},\n\t).AnyTimes()\n\ts.mockMutableState.EXPECT().UpdateCurrentVersion(lastWriteVersion, true).Return(nil)\n\ts.mockMutableState.EXPECT().GetInFlightDecision().Return(nil, false)\n\n\tnDCWorkflow := NewWorkflow(\n\t\tcontext.Background(),\n\t\tcluster.TestActiveClusterMetadata,\n\t\ts.mockContext,\n\t\ts.mockMutableState,\n\t\tNoopReleaseFn,\n\t\ttestlogger.New(s.T()),\n\t)\n\terr := nDCWorkflow.FlushBufferedEvents()\n\ts.NoError(err)\n}\n\nfunc TestWorkflowHappensAfter(t *testing.T) {\n\tbaseTime := time.Date(2023, 1, 1, 0, 0, 0, 0, time.FixedZone(\"UTC-8\", -8*60*60))\n\tutcBaseTime := baseTime.UTC()\n\tlaterTime := baseTime.Add(time.Hour)\n\n\t// Helper function to create ActiveClusterSelectionPolicy\n\tcreatePolicy := func(scope, name string) *types.ActiveClusterSelectionPolicy {\n\t\treturn &types.ActiveClusterSelectionPolicy{\n\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\tScope: scope,\n\t\t\t\tName:  name,\n\t\t\t},\n\t\t}\n\t}\n\n\t// Helper function to create WorkflowVectorClock\n\tcreateVectorClock := func(policy *types.ActiveClusterSelectionPolicy, lastWriteVersion, lastEventTaskID int64, startTime time.Time, runID string) WorkflowVectorClock {\n\t\treturn WorkflowVectorClock{\n\t\t\tActiveClusterSelectionPolicy: policy,\n\t\t\tLastWriteVersion:             lastWriteVersion,\n\t\t\tLastEventTaskID:              lastEventTaskID,\n\t\t\tStartTimestamp:               startTime,\n\t\t\tRunID:                        runID,\n\t\t}\n\t}\n\n\tpolicy1 := createPolicy(\"region\", \"region-1\")\n\tpolicy2 := createPolicy(\"region\", \"region-2\")\n\n\ttests := []struct {\n\t\tname            string\n\t\tthisVectorClock WorkflowVectorClock\n\t\tthatVectorClock WorkflowVectorClock\n\t\texpected        bool\n\t}{\n\t\t{\n\t\t\tname:            \"Different policies - this happens after based on start timestamp\",\n\t\t\tthisVectorClock: createVectorClock(policy1, 100, 10, laterTime, \"run-1\"),\n\t\t\tthatVectorClock: createVectorClock(policy2, 200, 20, baseTime, \"run-2\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"Different policies - same start timestamp - this happens after based on RunID\",\n\t\t\tthisVectorClock: createVectorClock(policy1, 100, 10, baseTime, \"run-z\"),\n\t\t\tthatVectorClock: createVectorClock(policy2, 200, 20, baseTime, \"run-a\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"Different policies - same start timestamp in different time zone - this happens after based on RunID\",\n\t\t\tthisVectorClock: createVectorClock(policy1, 100, 10, baseTime, \"run-z\"),\n\t\t\tthatVectorClock: createVectorClock(policy2, 200, 20, utcBaseTime, \"run-a\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"Same policies - this happens after based on LastWriteVersion\",\n\t\t\tthisVectorClock: createVectorClock(policy1, 200, 10, baseTime, \"run-1\"),\n\t\t\tthatVectorClock: createVectorClock(policy1, 100, 20, baseTime, \"run-2\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"Same policies and LastWriteVersion - this happens after based on LastEventTaskID\",\n\t\t\tthisVectorClock: createVectorClock(policy1, 100, 20, baseTime, \"run-1\"),\n\t\t\tthatVectorClock: createVectorClock(policy1, 100, 10, baseTime, \"run-2\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"Nil policies - this happens after based on LastWriteVersion\",\n\t\t\tthisVectorClock: createVectorClock(nil, 200, 10, laterTime, \"run-1\"),\n\t\t\tthatVectorClock: createVectorClock(nil, 100, 20, baseTime, \"run-2\"),\n\t\t\texpected:        true,\n\t\t},\n\t\t{\n\t\t\tname:            \"One nil policy, one non-nil - this happens after based on start timestamp\",\n\t\t\tthisVectorClock: createVectorClock(nil, 100, 10, laterTime, \"run-1\"),\n\t\t\tthatVectorClock: createVectorClock(policy1, 200, 20, baseTime, \"run-2\"),\n\t\t\texpected:        true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := workflowHappensAfter(tt.thisVectorClock, tt.thatVectorClock)\n\t\t\tassert.Equal(t, tt.expected, result, \"workflowHappensAfter result mismatch for test case: %s\", tt.name)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/failover/coordinator.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination coordinator_mock.go -self_package github.com/uber/cadence/service/history/failover\n\npackage failover\n\nimport (\n\tctx \"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\nconst (\n\tnotificationChanBufferSize       = 1000\n\treceiveChanBufferSize            = 1000\n\tcleanupMarkerInterval            = 30 * time.Minute\n\tinvalidMarkerDuration            = 1 * time.Hour\n\tupdateDomainRetryInitialInterval = 50 * time.Millisecond\n\tupdateDomainRetryCoefficient     = 2.0\n\tupdateDomainMaxRetry             = 2\n)\n\nvar (\n\terrRecordNotFound = &types.EntityNotExistsError{Message: \"Graceful failover record not found in shard coordinator\"}\n)\n\ntype (\n\t// Coordinator manages the failover markers on sending and receiving\n\tCoordinator interface {\n\t\tcommon.Daemon\n\n\t\tNotifyFailoverMarkers(shardID int32, markers []*types.FailoverMarkerAttributes)\n\t\tReceiveFailoverMarkers(shardIDs []int32, marker *types.FailoverMarkerAttributes)\n\t\tGetFailoverInfo(domainID string) (*types.GetFailoverInfoResponse, error)\n\t}\n\n\tcoordinatorImpl struct {\n\t\tstatus           int32\n\t\tnotificationChan chan *notificationRequest\n\t\treceiveChan      chan *receiveRequest\n\t\tshutdownChan     chan struct{}\n\t\tretryPolicy      backoff.RetryPolicy\n\n\t\trecorderLock sync.Mutex\n\t\trecorder     map[string]*failoverRecord\n\n\t\tdomainManager persistence.DomainManager\n\t\thistoryClient history.Client\n\t\tconfig        *config.Config\n\t\ttimeSource    clock.TimeSource\n\t\tdomainCache   cache.DomainCache\n\t\tscope         metrics.Scope\n\t\tlogger        log.Logger\n\t}\n\n\tnotificationRequest struct {\n\t\tshardID int32\n\t\tmarkers []*types.FailoverMarkerAttributes\n\t}\n\n\treceiveRequest struct {\n\t\tshardIDs []int32\n\t\tmarker   *types.FailoverMarkerAttributes\n\t}\n\n\tfailoverRecord struct {\n\t\tfailoverVersion int64\n\t\tshards          map[int32]struct{}\n\t\tlastUpdatedTime time.Time\n\t}\n)\n\n// NewCoordinator initialize a failover coordinator\nfunc NewCoordinator(\n\tdomainManager persistence.DomainManager,\n\thistoryClient history.Client,\n\ttimeSource clock.TimeSource,\n\tdomainCache cache.DomainCache,\n\tconfig *config.Config,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) Coordinator {\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(updateDomainRetryInitialInterval)\n\tretryPolicy.SetBackoffCoefficient(updateDomainRetryCoefficient)\n\tretryPolicy.SetMaximumAttempts(updateDomainMaxRetry)\n\n\treturn &coordinatorImpl{\n\t\tstatus:           common.DaemonStatusInitialized,\n\t\trecorder:         make(map[string]*failoverRecord),\n\t\tnotificationChan: make(chan *notificationRequest, notificationChanBufferSize),\n\t\treceiveChan:      make(chan *receiveRequest, receiveChanBufferSize),\n\t\tshutdownChan:     make(chan struct{}),\n\t\tretryPolicy:      retryPolicy,\n\t\tdomainManager:    domainManager,\n\t\thistoryClient:    historyClient,\n\t\ttimeSource:       timeSource,\n\t\tdomainCache:      domainCache,\n\t\tconfig:           config,\n\t\tscope:            metricsClient.Scope(metrics.FailoverMarkerScope),\n\t\tlogger:           logger.WithTags(tag.ComponentFailoverCoordinator),\n\t}\n}\n\nfunc (c *coordinatorImpl) Start() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&c.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\tgo c.receiveFailoverMarkersLoop()\n\tgo c.notifyFailoverMarkerLoop()\n\n\tc.logger.Info(\"Coordinator state changed\", tag.LifeCycleStarted)\n}\n\nfunc (c *coordinatorImpl) Stop() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&c.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\tclose(c.shutdownChan)\n\tc.logger.Info(\"Coordinator state changed\", tag.LifeCycleStopped)\n}\n\nfunc (c *coordinatorImpl) NotifyFailoverMarkers(\n\tshardID int32,\n\tmarkers []*types.FailoverMarkerAttributes,\n) {\n\n\tc.notificationChan <- &notificationRequest{\n\t\tshardID: shardID,\n\t\tmarkers: markers,\n\t}\n}\n\nfunc (c *coordinatorImpl) ReceiveFailoverMarkers(\n\tshardIDs []int32,\n\tmarker *types.FailoverMarkerAttributes,\n) {\n\n\tc.receiveChan <- &receiveRequest{\n\t\tshardIDs: shardIDs,\n\t\tmarker:   marker,\n\t}\n}\n\nfunc (c *coordinatorImpl) GetFailoverInfo(\n\tdomainID string,\n) (*types.GetFailoverInfoResponse, error) {\n\tc.recorderLock.Lock()\n\tdefer c.recorderLock.Unlock()\n\n\trecord, ok := c.recorder[domainID]\n\tif !ok {\n\t\treturn nil, errRecordNotFound\n\t}\n\n\tvar pendingShards []int32\n\tfor i := 0; i < c.config.NumberOfShards; i++ {\n\t\tif _, ok := record.shards[int32(i)]; !ok {\n\t\t\tpendingShards = append(pendingShards, int32(i))\n\t\t}\n\t}\n\treturn &types.GetFailoverInfoResponse{\n\t\tCompletedShardCount: int32(len(record.shards)),\n\t\tPendingShards:       pendingShards,\n\t}, nil\n}\n\nfunc (c *coordinatorImpl) receiveFailoverMarkersLoop() {\n\n\tticker := time.NewTicker(cleanupMarkerInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-c.shutdownChan:\n\t\t\treturn\n\t\tcase <-ticker.C:\n\t\t\tc.cleanupInvalidMarkers()\n\t\tcase request := <-c.receiveChan:\n\t\t\tc.handleFailoverMarkers(request)\n\t\t}\n\t}\n}\n\nfunc (c *coordinatorImpl) notifyFailoverMarkerLoop() {\n\n\ttimer := time.NewTimer(backoff.JitDuration(\n\t\tc.config.NotifyFailoverMarkerInterval(),\n\t\tc.config.NotifyFailoverMarkerTimerJitterCoefficient(),\n\t))\n\tdefer timer.Stop()\n\trequestByMarker := make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\n\tfor {\n\t\tselect {\n\t\tcase <-c.shutdownChan:\n\t\t\treturn\n\t\tcase notificationReq := <-c.notificationChan:\n\t\t\t// if there is a shard movement happen, it is fine to have duplicate shard ID in the request\n\t\t\t// The receiver side will de-dup the shard IDs. See: handleFailoverMarkers\n\t\t\taggregateNotificationRequests(notificationReq, requestByMarker)\n\t\tcase <-timer.C:\n\t\t\tif err := c.notifyRemoteCoordinator(requestByMarker); err == nil {\n\t\t\t\trequestByMarker = make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\t\t\t}\n\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\tc.config.NotifyFailoverMarkerInterval(),\n\t\t\t\tc.config.NotifyFailoverMarkerTimerJitterCoefficient(),\n\t\t\t))\n\t\t}\n\t}\n}\n\nfunc (c *coordinatorImpl) handleFailoverMarkers(\n\trequest *receiveRequest,\n) {\n\n\tc.recorderLock.Lock()\n\tdefer c.recorderLock.Unlock()\n\n\tmarker := request.marker\n\tdomainID := marker.GetDomainID()\n\tif record, ok := c.recorder[domainID]; ok {\n\t\t// if the local failover version is smaller than the new received marker,\n\t\t// it means there is another failover happened and the local one should be invalid.\n\t\tif record.failoverVersion < marker.GetFailoverVersion() {\n\t\t\tdelete(c.recorder, domainID)\n\t\t}\n\n\t\t// if the local failover version is larger than the new received marker,\n\t\t// ignore the incoming marker\n\t\tif record.failoverVersion > marker.GetFailoverVersion() {\n\t\t\treturn\n\t\t}\n\t}\n\n\tif _, ok := c.recorder[domainID]; !ok {\n\t\t// initialize the failover record\n\t\tc.recorder[marker.GetDomainID()] = &failoverRecord{\n\t\t\tfailoverVersion: marker.GetFailoverVersion(),\n\t\t\tshards:          make(map[int32]struct{}),\n\t\t}\n\t}\n\n\trecord := c.recorder[domainID]\n\trecord.lastUpdatedTime = c.timeSource.Now()\n\tfor _, shardID := range request.shardIDs {\n\t\trecord.shards[shardID] = struct{}{}\n\t}\n\n\tdomainName, err := c.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\tc.logger.Error(\"Coordinator failed to get domain after receiving all failover markers\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.Error(err),\n\t\t)\n\n\t\tc.scope.Tagged(metrics.DomainTag(domainName)).IncCounter(metrics.CadenceFailures)\n\t\treturn\n\t}\n\n\tif len(record.shards) == c.config.NumberOfShards {\n\t\tif err := domain.CleanPendingActiveState(\n\t\t\tc.domainManager,\n\t\t\tdomainID,\n\t\t\trecord.failoverVersion,\n\t\t\tc.retryPolicy,\n\t\t); err != nil {\n\t\t\tc.logger.Error(\"Coordinator failed to update domain after receiving all failover markers\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tc.scope.IncCounter(metrics.CadenceFailures)\n\t\t\treturn\n\t\t}\n\t\tdelete(c.recorder, domainID)\n\t\tnow := c.timeSource.Now()\n\t\tc.scope.Tagged(\n\t\t\tmetrics.DomainTag(domainName),\n\t\t).RecordTimer(\n\t\t\tmetrics.GracefulFailoverLatency,\n\t\t\tnow.Sub(time.Unix(0, marker.GetCreationTime())),\n\t\t)\n\t\tc.logger.Info(\"Updated domain from pending-active to active\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.FailoverVersion(marker.FailoverVersion),\n\t\t)\n\t} else {\n\t\tc.scope.Tagged(\n\t\t\tmetrics.DomainTag(domainName),\n\t\t).RecordTimer(\n\t\t\tmetrics.FailoverMarkerCount,\n\t\t\ttime.Duration(len(record.shards)),\n\t\t)\n\t}\n}\n\nfunc (c *coordinatorImpl) cleanupInvalidMarkers() {\n\tc.recorderLock.Lock()\n\tdefer c.recorderLock.Unlock()\n\n\tfor domainID, record := range c.recorder {\n\t\tif c.timeSource.Now().Sub(record.lastUpdatedTime) > invalidMarkerDuration {\n\t\t\tdelete(c.recorder, domainID)\n\t\t}\n\t}\n}\n\nfunc (c *coordinatorImpl) notifyRemoteCoordinator(\n\trequestByMarker map[types.FailoverMarkerAttributes]*receiveRequest,\n) error {\n\n\tif len(requestByMarker) > 0 {\n\t\tvar tokens []*types.FailoverMarkerToken\n\t\tfor _, request := range requestByMarker {\n\t\t\ttokens = append(tokens, &types.FailoverMarkerToken{\n\t\t\t\tShardIDs:       request.shardIDs,\n\t\t\t\tFailoverMarker: request.marker,\n\t\t\t})\n\t\t}\n\n\t\terr := c.historyClient.NotifyFailoverMarkers(\n\t\t\tctx.Background(),\n\t\t\t&types.NotifyFailoverMarkersRequest{\n\t\t\t\tFailoverMarkerTokens: tokens,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\tc.scope.IncCounter(metrics.FailoverMarkerNotificationFailure)\n\t\t\tc.logger.Error(\"Failed to notify failover markers\", tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc aggregateNotificationRequests(\n\trequest *notificationRequest,\n\trequestByMarker map[types.FailoverMarkerAttributes]*receiveRequest,\n) {\n\n\tfor _, marker := range request.markers {\n\t\tmarkerMask := types.FailoverMarkerAttributes{\n\t\t\tDomainID:        marker.DomainID,\n\t\t\tFailoverVersion: marker.FailoverVersion,\n\t\t}\n\t\tif _, ok := requestByMarker[markerMask]; !ok {\n\t\t\trequestByMarker[markerMask] = &receiveRequest{\n\t\t\t\tshardIDs: []int32{},\n\t\t\t\tmarker:   marker,\n\t\t\t}\n\t\t}\n\t\treq := requestByMarker[markerMask]\n\t\treq.shardIDs = append(req.shardIDs, request.shardID)\n\t}\n}\n"
  },
  {
    "path": "service/history/failover/coordinator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: coordinator.go\n//\n// Generated by this command:\n//\n//\tmockgen -package failover -source coordinator.go -destination coordinator_mock.go -self_package github.com/uber/cadence/service/history/failover\n//\n\n// Package failover is a generated GoMock package.\npackage failover\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockCoordinator is a mock of Coordinator interface.\ntype MockCoordinator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCoordinatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockCoordinatorMockRecorder is the mock recorder for MockCoordinator.\ntype MockCoordinatorMockRecorder struct {\n\tmock *MockCoordinator\n}\n\n// NewMockCoordinator creates a new mock instance.\nfunc NewMockCoordinator(ctrl *gomock.Controller) *MockCoordinator {\n\tmock := &MockCoordinator{ctrl: ctrl}\n\tmock.recorder = &MockCoordinatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCoordinator) EXPECT() *MockCoordinatorMockRecorder {\n\treturn m.recorder\n}\n\n// GetFailoverInfo mocks base method.\nfunc (m *MockCoordinator) GetFailoverInfo(domainID string) (*types.GetFailoverInfoResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFailoverInfo\", domainID)\n\tret0, _ := ret[0].(*types.GetFailoverInfoResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetFailoverInfo indicates an expected call of GetFailoverInfo.\nfunc (mr *MockCoordinatorMockRecorder) GetFailoverInfo(domainID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFailoverInfo\", reflect.TypeOf((*MockCoordinator)(nil).GetFailoverInfo), domainID)\n}\n\n// NotifyFailoverMarkers mocks base method.\nfunc (m *MockCoordinator) NotifyFailoverMarkers(shardID int32, markers []*types.FailoverMarkerAttributes) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyFailoverMarkers\", shardID, markers)\n}\n\n// NotifyFailoverMarkers indicates an expected call of NotifyFailoverMarkers.\nfunc (mr *MockCoordinatorMockRecorder) NotifyFailoverMarkers(shardID, markers any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyFailoverMarkers\", reflect.TypeOf((*MockCoordinator)(nil).NotifyFailoverMarkers), shardID, markers)\n}\n\n// ReceiveFailoverMarkers mocks base method.\nfunc (m *MockCoordinator) ReceiveFailoverMarkers(shardIDs []int32, marker *types.FailoverMarkerAttributes) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ReceiveFailoverMarkers\", shardIDs, marker)\n}\n\n// ReceiveFailoverMarkers indicates an expected call of ReceiveFailoverMarkers.\nfunc (mr *MockCoordinatorMockRecorder) ReceiveFailoverMarkers(shardIDs, marker any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReceiveFailoverMarkers\", reflect.TypeOf((*MockCoordinator)(nil).ReceiveFailoverMarkers), shardIDs, marker)\n}\n\n// Start mocks base method.\nfunc (m *MockCoordinator) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockCoordinatorMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockCoordinator)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockCoordinator) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockCoordinatorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockCoordinator)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/failover/coordinator_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage failover\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype (\n\tcoordinatorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller          *gomock.Controller\n\t\tmockResource        *resource.Test\n\t\tmockMetadataManager *mmocks.MetadataManager\n\t\thistoryClient       *history.MockClient\n\t\tconfig              *config.Config\n\t\tcoordinator         *coordinatorImpl\n\t}\n)\n\nfunc TestCoordinatorSuite(t *testing.T) {\n\ts := new(coordinatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *coordinatorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.History)\n\ts.mockMetadataManager = s.mockResource.MetadataMgr\n\ts.historyClient = s.mockResource.HistoryClient\n\ts.config = config.NewForTest()\n\ts.config.NumberOfShards = 2\n\ts.config.NotifyFailoverMarkerInterval = dynamicproperties.GetDurationPropertyFn(10 * time.Millisecond)\n\ts.config.NotifyFailoverMarkerTimerJitterCoefficient = dynamicproperties.GetFloatPropertyFn(0.01)\n\ts.mockResource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test\", nil).AnyTimes()\n\n\ts.coordinator = NewCoordinator(\n\t\ts.mockMetadataManager,\n\t\ts.historyClient,\n\t\ts.mockResource.GetTimeSource(),\n\t\ts.mockResource.GetDomainCache(),\n\t\ts.config,\n\t\ts.mockResource.GetMetricsClient(),\n\t\ts.mockResource.GetLogger(),\n\t).(*coordinatorImpl)\n}\n\nfunc (s *coordinatorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n\ts.coordinator.Stop()\n\ts.mockMetadataManager.AssertExpectations(s.T())\n}\n\nfunc (s *coordinatorSuite) TestNotifyFailoverMarkers() {\n\tdoneCh := make(chan struct{})\n\tattributes := &types.FailoverMarkerAttributes{\n\t\tDomainID:        uuid.New(),\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\ts.historyClient.EXPECT().NotifyFailoverMarkers(\n\t\tcontext.Background(), &types.NotifyFailoverMarkersRequest{\n\t\t\tFailoverMarkerTokens: []*types.FailoverMarkerToken{\n\t\t\t\t{\n\t\t\t\t\tShardIDs:       []int32{1, 2},\n\t\t\t\t\tFailoverMarker: attributes,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t).DoAndReturn(func(ctx context.Context, request *types.NotifyFailoverMarkersRequest, opts ...yarpc.CallOption) error {\n\t\tclose(doneCh)\n\t\treturn nil\n\t}).Times(1)\n\n\ts.coordinator.NotifyFailoverMarkers(\n\t\t1,\n\t\t[]*types.FailoverMarkerAttributes{attributes},\n\t)\n\ts.coordinator.NotifyFailoverMarkers(\n\t\t2,\n\t\t[]*types.FailoverMarkerAttributes{attributes},\n\t)\n\ts.coordinator.Start()\n\t<-doneCh\n}\n\nfunc (s *coordinatorSuite) TestNotifyRemoteCoordinator_Empty() {\n\trequestByMarker := make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\ts.historyClient.EXPECT().NotifyFailoverMarkers(context.Background(), gomock.Any()).Times(0)\n\terr := s.coordinator.notifyRemoteCoordinator(requestByMarker)\n\ts.NoError(err)\n}\n\nfunc (s *coordinatorSuite) TestNotifyRemoteCoordinator() {\n\n\trequestByMarker := make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\tattributes := types.FailoverMarkerAttributes{\n\t\tDomainID:        uuid.New(),\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequestByMarker[attributes] = &receiveRequest{\n\t\tshardIDs: []int32{1, 2, 3},\n\t\tmarker:   &attributes,\n\t}\n\n\ts.historyClient.EXPECT().NotifyFailoverMarkers(\n\t\tcontext.Background(), &types.NotifyFailoverMarkersRequest{\n\t\t\tFailoverMarkerTokens: []*types.FailoverMarkerToken{\n\t\t\t\t{\n\t\t\t\t\tShardIDs:       []int32{1, 2, 3},\n\t\t\t\t\tFailoverMarker: &attributes,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t).Return(nil).Times(1)\n\terr := s.coordinator.notifyRemoteCoordinator(requestByMarker)\n\ts.NoError(err)\n\ts.Equal(1, len(requestByMarker))\n}\n\nfunc (s *coordinatorSuite) TestNotifyRemoteCoordinator_Error() {\n\n\trequestByMarker := make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\tattributes := types.FailoverMarkerAttributes{\n\t\tDomainID:        uuid.New(),\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequestByMarker[attributes] = &receiveRequest{\n\t\tshardIDs: []int32{1, 2, 3},\n\t\tmarker:   &attributes,\n\t}\n\n\ts.historyClient.EXPECT().NotifyFailoverMarkers(\n\t\tcontext.Background(), &types.NotifyFailoverMarkersRequest{\n\t\t\tFailoverMarkerTokens: []*types.FailoverMarkerToken{\n\t\t\t\t{\n\t\t\t\t\tShardIDs:       []int32{1, 2, 3},\n\t\t\t\t\tFailoverMarker: &attributes,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t).Return(fmt.Errorf(\"test\")).Times(1)\n\terr := s.coordinator.notifyRemoteCoordinator(requestByMarker)\n\ts.Error(err)\n}\n\nfunc (s *coordinatorSuite) TestAggregateNotificationRequests() {\n\trequestByMarker := make(map[types.FailoverMarkerAttributes]*receiveRequest)\n\tattributes1 := types.FailoverMarkerAttributes{\n\t\tDomainID:        uuid.New(),\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\tattributes2 := types.FailoverMarkerAttributes{\n\t\tDomainID:        uuid.New(),\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(2),\n\t}\n\trequest1 := &notificationRequest{\n\t\tshardID: 1,\n\t\tmarkers: []*types.FailoverMarkerAttributes{&attributes1},\n\t}\n\taggregateNotificationRequests(request1, requestByMarker)\n\trequest2 := &notificationRequest{\n\t\tshardID: 2,\n\t\tmarkers: []*types.FailoverMarkerAttributes{&attributes1},\n\t}\n\taggregateNotificationRequests(request2, requestByMarker)\n\trequest3 := &notificationRequest{\n\t\tshardID: 3,\n\t\tmarkers: []*types.FailoverMarkerAttributes{&attributes1, &attributes2},\n\t}\n\taggregateNotificationRequests(request3, requestByMarker)\n\tattributes1.CreationTime = nil\n\tattributes2.CreationTime = nil\n\ts.Equal([]int32{1, 2, 3}, requestByMarker[attributes1].shardIDs)\n\ts.Equal([]int32{3}, requestByMarker[attributes2].shardIDs)\n}\n\nfunc (s *coordinatorSuite) TestHandleFailoverMarkers_DeleteExpiredFailoverMarker() {\n\tdomainID := uuid.New()\n\tattributes1 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\tattributes2 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequest1 := &receiveRequest{\n\t\tshardIDs: []int32{1},\n\t\tmarker:   attributes1,\n\t}\n\trequest2 := &receiveRequest{\n\t\tshardIDs: []int32{2},\n\t\tmarker:   attributes2,\n\t}\n\n\ts.coordinator.handleFailoverMarkers(request1)\n\ts.coordinator.handleFailoverMarkers(request2)\n\ts.Equal(1, len(s.coordinator.recorder))\n}\n\nfunc (s *coordinatorSuite) TestHandleFailoverMarkers_IgnoreExpiredFailoverMarker() {\n\tdomainID := uuid.New()\n\tattributes1 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\tattributes2 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequest1 := &receiveRequest{\n\t\tshardIDs: []int32{1},\n\t\tmarker:   attributes1,\n\t}\n\trequest2 := &receiveRequest{\n\t\tshardIDs: []int32{2},\n\t\tmarker:   attributes2,\n\t}\n\n\ts.coordinator.handleFailoverMarkers(request2)\n\ts.coordinator.handleFailoverMarkers(request1)\n\ts.Equal(1, len(s.coordinator.recorder))\n}\n\nfunc (s *coordinatorSuite) TestHandleFailoverMarkers_CleanPendingActiveState_Success() {\n\tdomainID := uuid.New()\n\tattributes1 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\tattributes2 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequest1 := &receiveRequest{\n\t\tshardIDs: []int32{1},\n\t\tmarker:   attributes1,\n\t}\n\trequest2 := &receiveRequest{\n\t\tshardIDs: []int32{2},\n\t\tmarker:   attributes2,\n\t}\n\tinfo := &persistence.DomainInfo{\n\t\tID:          domainID,\n\t\tName:        uuid.New(),\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"some random description\",\n\t\tOwnerEmail:  \"some random email\",\n\t\tData:        nil,\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention:  1,\n\t\tEmitMetric: true,\n\t}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\t\"active\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockMetadataManager.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: 1,\n\t}, nil)\n\ts.mockMetadataManager.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainID,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             common.Int64Ptr(1),\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\ts.mockMetadataManager.On(\"UpdateDomain\", mock.Anything, &persistence.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             nil,\n\t\tNotificationVersion:         1,\n\t}).Return(nil).Times(1)\n\n\ts.coordinator.handleFailoverMarkers(request1)\n\ts.coordinator.handleFailoverMarkers(request2)\n\ts.Equal(0, len(s.coordinator.recorder))\n}\n\nfunc (s *coordinatorSuite) TestHandleFailoverMarkers_CleanPendingActiveState_Error() {\n\tdomainID := uuid.New()\n\tattributes1 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\tattributes2 := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequest1 := &receiveRequest{\n\t\tshardIDs: []int32{1},\n\t\tmarker:   attributes1,\n\t}\n\trequest2 := &receiveRequest{\n\t\tshardIDs: []int32{2},\n\t\tmarker:   attributes2,\n\t}\n\tinfo := &persistence.DomainInfo{\n\t\tID:          domainID,\n\t\tName:        uuid.New(),\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"some random description\",\n\t\tOwnerEmail:  \"some random email\",\n\t\tData:        nil,\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention:  1,\n\t\tEmitMetric: true,\n\t}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\t\"active\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.mockMetadataManager.On(\"GetMetadata\", mock.Anything).Return(&persistence.GetMetadataResponse{\n\t\tNotificationVersion: 1,\n\t}, nil)\n\ts.mockMetadataManager.On(\"GetDomain\", mock.Anything, &persistence.GetDomainRequest{\n\t\tID: domainID,\n\t}).Return(&persistence.GetDomainResponse{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tIsGlobalDomain:              true,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             common.Int64Ptr(1),\n\t\tNotificationVersion:         1,\n\t}, nil).Times(1)\n\ts.mockMetadataManager.On(\"UpdateDomain\", mock.Anything, &persistence.UpdateDomainRequest{\n\t\tInfo:                        info,\n\t\tConfig:                      domainConfig,\n\t\tReplicationConfig:           replicationConfig,\n\t\tConfigVersion:               1,\n\t\tFailoverVersion:             2,\n\t\tFailoverNotificationVersion: 2,\n\t\tFailoverEndTime:             nil,\n\t\tNotificationVersion:         1,\n\t}).Return(fmt.Errorf(\"test error\")).Times(3)\n\n\ts.coordinator.handleFailoverMarkers(request1)\n\ts.coordinator.handleFailoverMarkers(request2)\n\ts.Equal(1, len(s.coordinator.recorder))\n}\n\nfunc (s *coordinatorSuite) TestGetFailoverInfo_Success() {\n\tdomainID := uuid.New()\n\n\t// Add failover marker\n\tattributes := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 2,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\trequest := &receiveRequest{\n\t\tshardIDs: []int32{1},\n\t\tmarker:   attributes,\n\t}\n\ts.coordinator.handleFailoverMarkers(request)\n\n\tresp, err := s.coordinator.GetFailoverInfo(domainID)\n\ts.NoError(err)\n\ts.Equal(int32(1), resp.GetCompletedShardCount())\n\ts.Contains(resp.GetPendingShards(), int32(0))\n}\n\nfunc (s *coordinatorSuite) TestGetFailoverInfo_DomainIDNotFound_Error() {\n\tdomainID := uuid.New()\n\tresp, err := s.coordinator.GetFailoverInfo(domainID)\n\ts.Nil(resp)\n\ts.Error(err)\n}\n"
  },
  {
    "path": "service/history/failover/marker_notifier.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination marker_notifier_mock.go -self_package github.com/uber/cadence/service/history/failover\n\npackage failover\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\t// MarkerNotifier notifies failover markers to the remote failover coordinator\n\tMarkerNotifier interface {\n\t\tcommon.Daemon\n\t}\n\n\tmarkerNotifierImpl struct {\n\t\tstatus              int32\n\t\tshutdownCh          chan struct{}\n\t\tshard               shard.Context\n\t\tconfig              *config.Config\n\t\tfailoverCoordinator Coordinator\n\t\tlogger              log.Logger\n\t\tmetrics             metrics.Client\n\t}\n)\n\n// NewMarkerNotifier creates a new instance of failover marker notifier\nfunc NewMarkerNotifier(\n\tshard shard.Context,\n\tconfig *config.Config,\n\tfailoverCoordinator Coordinator,\n) MarkerNotifier {\n\n\treturn &markerNotifierImpl{\n\t\tstatus:              common.DaemonStatusInitialized,\n\t\tshutdownCh:          make(chan struct{}, 1),\n\t\tshard:               shard,\n\t\tconfig:              config,\n\t\tfailoverCoordinator: failoverCoordinator,\n\t\tlogger:              shard.GetLogger().WithTags(tag.ComponentFailoverMarkerNotifier),\n\t\tmetrics:             shard.GetMetricsClient(),\n\t}\n}\n\nfunc (m *markerNotifierImpl) Start() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&m.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\tgo m.notifyPendingFailoverMarker()\n\tm.logger.Info(\"Marker notifier state changed\", tag.LifeCycleStarted)\n}\n\nfunc (m *markerNotifierImpl) Stop() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&m.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\tclose(m.shutdownCh)\n\tm.logger.Info(\"Marker notifier state changed\", tag.LifeCycleStopped)\n}\n\nfunc (m *markerNotifierImpl) notifyPendingFailoverMarker() {\n\n\tticker := time.NewTicker(m.config.NotifyFailoverMarkerInterval())\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-m.shutdownCh:\n\t\t\treturn\n\t\tcase <-ticker.C:\n\t\t\tmarkers, err := m.shard.ValidateAndUpdateFailoverMarkers()\n\t\t\tif err != nil {\n\t\t\t\tm.metrics.IncCounter(metrics.FailoverMarkerScope, metrics.FailoverMarkerUpdateShardFailure)\n\t\t\t\tm.logger.Error(\"Failed to update pending failover markers in shard info.\", tag.Error(err))\n\t\t\t}\n\n\t\t\tif len(markers) > 0 {\n\t\t\t\tm.failoverCoordinator.NotifyFailoverMarkers(int32(m.shard.GetShardID()), markers)\n\t\t\t}\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/failover/marker_notifier_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: marker_notifier.go\n//\n// Generated by this command:\n//\n//\tmockgen -package failover -source marker_notifier.go -destination marker_notifier_mock.go -self_package github.com/uber/cadence/service/history/failover\n//\n\n// Package failover is a generated GoMock package.\npackage failover\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockMarkerNotifier is a mock of MarkerNotifier interface.\ntype MockMarkerNotifier struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMarkerNotifierMockRecorder\n\tisgomock struct{}\n}\n\n// MockMarkerNotifierMockRecorder is the mock recorder for MockMarkerNotifier.\ntype MockMarkerNotifierMockRecorder struct {\n\tmock *MockMarkerNotifier\n}\n\n// NewMockMarkerNotifier creates a new mock instance.\nfunc NewMockMarkerNotifier(ctrl *gomock.Controller) *MockMarkerNotifier {\n\tmock := &MockMarkerNotifier{ctrl: ctrl}\n\tmock.recorder = &MockMarkerNotifierMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMarkerNotifier) EXPECT() *MockMarkerNotifierMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockMarkerNotifier) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockMarkerNotifierMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockMarkerNotifier)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockMarkerNotifier) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockMarkerNotifierMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockMarkerNotifier)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/failover/marker_notifier_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage failover\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tmarkerNotifierSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tcoordinator     *MockCoordinator\n\t\tmockShard       *shard.TestContext\n\t\tmockDomainCache *cache.MockDomainCache\n\t\tclusterMetadata cluster.Metadata\n\t\tmarkerNotifier  *markerNotifierImpl\n\t}\n)\n\nfunc TestMarkerNotifierSuite(t *testing.T) {\n\ts := new(markerNotifierSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *markerNotifierSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\n\tconfig := config.NewForTest()\n\tconfig.NotifyFailoverMarkerInterval = dynamicproperties.GetDurationPropertyFn(time.Millisecond)\n\ts.coordinator = NewMockCoordinator(s.controller)\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig,\n\t)\n\ts.clusterMetadata = s.mockShard.Resource.ClusterMetadata\n\tmockShardManager := s.mockShard.Resource.ShardMgr\n\tmockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\n\ts.markerNotifier = NewMarkerNotifier(\n\t\ts.mockShard,\n\t\tconfig,\n\t\ts.coordinator,\n\t).(*markerNotifierImpl)\n}\n\nfunc (s *markerNotifierSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.markerNotifier.Stop()\n}\n\nfunc (s *markerNotifierSuite) TestNotifyPendingFailoverMarker_Shutdown() {\n\tclose(s.markerNotifier.shutdownCh)\n\ts.coordinator.EXPECT().NotifyFailoverMarkers(gomock.Any(), gomock.Any()).Times(0)\n\ts.markerNotifier.notifyPendingFailoverMarker()\n}\n\nfunc (s *markerNotifierSuite) TestNotifyPendingFailoverMarker() {\n\tdomainID := uuid.New()\n\tinfo := &persistence.DomainInfo{\n\t\tID:          domainID,\n\t\tName:        domainID,\n\t\tStatus:      persistence.DomainStatusRegistered,\n\t\tDescription: \"some random description\",\n\t\tOwnerEmail:  \"some random email\",\n\t\tData:        nil,\n\t}\n\tdomainConfig := &persistence.DomainConfig{\n\t\tRetention:  1,\n\t\tEmitMetric: true,\n\t}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: s.clusterMetadata.GetCurrentClusterName(),\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{\n\t\t\t\tClusterName: s.clusterMetadata.GetCurrentClusterName(),\n\t\t\t},\n\t\t},\n\t}\n\tendTime := common.Int64Ptr(time.Now().UnixNano())\n\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\tinfo,\n\t\tdomainConfig,\n\t\ttrue,\n\t\treplicationConfig,\n\t\t1,\n\t\tendTime,\n\t\t0, 0, 0,\n\t)\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil).AnyTimes()\n\ttask := &types.FailoverMarkerAttributes{\n\t\tDomainID:        domainID,\n\t\tFailoverVersion: 1,\n\t\tCreationTime:    common.Int64Ptr(1),\n\t}\n\ttasks := []*types.FailoverMarkerAttributes{task}\n\trespCh := make(chan error, 1)\n\terr := s.mockShard.AddingPendingFailoverMarker(task)\n\ts.NoError(err)\n\n\tcount := 0\n\ts.coordinator.EXPECT().NotifyFailoverMarkers(\n\t\tint32(s.mockShard.GetShardID()),\n\t\ttasks,\n\t).Do(\n\t\tfunc(\n\t\t\tshardID int32,\n\t\t\tmarkers []*types.FailoverMarkerAttributes,\n\t\t) {\n\t\t\tif count == 0 {\n\t\t\t\tcount++\n\t\t\t\trespCh <- nil\n\t\t\t}\n\t\t\tif count == 1 {\n\t\t\t\tclose(s.markerNotifier.shutdownCh)\n\t\t\t}\n\t\t},\n\t)\n\n\ts.markerNotifier.notifyPendingFailoverMarker()\n}\n"
  },
  {
    "path": "service/history/handler/handler.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"slices\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\t\"golang.org/x/exp/maps\"\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/engine/engineimpl\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/failover\"\n\t\"github.com/uber/cadence/service/history/lookup\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/queuev2\"\n\t\"github.com/uber/cadence/service/history/replication\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n)\n\nconst (\n\tshardOwnershipTransferDelay = 5 * time.Second\n)\n\ntype (\n\t// handlerImpl is an implementation for history service independent of wire protocol\n\thandlerImpl struct {\n\t\tresource.Resource\n\n\t\tshuttingDown             int32\n\t\tcontroller               shard.Controller\n\t\ttokenSerializer          common.TaskTokenSerializer\n\t\tstartWG                  sync.WaitGroup\n\t\tconfig                   *config.Config\n\t\thistoryEventNotifier     events.Notifier\n\t\trateLimiter              quotas.Limiter\n\t\treplicationTaskFetchers  replication.TaskFetchers\n\t\tqueueTaskProcessor       task.Processor\n\t\tfailoverCoordinator      failover.Coordinator\n\t\tworkflowIDCache          workflowcache.WFCache\n\t\tratelimitAggregator      algorithm.RequestWeighted\n\t\tqueueFactories           []queue.Factory\n\t\treplicationBudgetManager cache.Manager\n\t}\n)\n\nvar _ Handler = (*handlerImpl)(nil)\nvar _ shard.EngineFactory = (*handlerImpl)(nil)\n\n// NewHandler creates a thrift handler for the history service\nfunc NewHandler(\n\tresource resource.Resource,\n\tconfig *config.Config,\n\twfCache workflowcache.WFCache,\n) Handler {\n\thandler := &handlerImpl{\n\t\tResource:            resource,\n\t\tconfig:              config,\n\t\ttokenSerializer:     common.NewJSONTaskTokenSerializer(),\n\t\trateLimiter:         quotas.NewDynamicRateLimiter(config.RPS.AsFloat64()),\n\t\tworkflowIDCache:     wfCache,\n\t\tratelimitAggregator: resource.GetRatelimiterAlgorithm(),\n\t}\n\n\t// prevent us from trying to serve requests before shard controller is started and ready\n\thandler.startWG.Add(1)\n\treturn handler\n}\n\n// Start starts the handler\nfunc (h *handlerImpl) Start() {\n\tvar err error\n\th.replicationTaskFetchers, err = replication.NewTaskFetchers(\n\t\th.GetLogger(),\n\t\th.config,\n\t\th.GetClusterMetadata(),\n\t\th.GetClientBean(),\n\t\th.GetMetricsClient(),\n\t)\n\tif err != nil {\n\t\th.GetLogger().Fatal(\"Creating replication task fetchers failed\", tag.Error(err))\n\t}\n\n\th.replicationTaskFetchers.Start()\n\n\ttaskPriorityAssigner := task.NewPriorityAssigner(\n\t\th.GetClusterMetadata().GetCurrentClusterName(),\n\t\th.GetDomainCache(),\n\t\th.GetActiveClusterManager(),\n\t\th.GetLogger(),\n\t\th.GetMetricsClient(),\n\t\th.config,\n\t)\n\n\th.replicationBudgetManager = cache.NewBudgetManager(\n\t\t\"replication-budget-manager\",\n\t\th.config.ReplicationBudgetManagerMaxSizeBytes,\n\t\th.config.ReplicationBudgetManagerMaxSizeCount,\n\t\tcache.AdmissionOptimistic,\n\t\t0,\n\t\th.GetMetricsClient().Scope(metrics.ReplicatorCacheManagerScope, metrics.HostTag(h.config.HostName)),\n\t\th.GetLogger(),\n\t\th.config.ReplicationBudgetManagerSoftCapThreshold,\n\t)\n\n\th.controller = shard.NewShardController(\n\t\th.Resource,\n\t\th,\n\t\th.config,\n\t\th.replicationBudgetManager,\n\t)\n\n\tvar taskProcessor task.Processor\n\ttaskProcessor, err = task.NewProcessor(\n\t\ttaskPriorityAssigner,\n\t\th.config,\n\t\th.GetLogger(),\n\t\th.GetMetricsClient(),\n\t\th.GetTimeSource(),\n\t\th.GetDomainCache(),\n\t)\n\tif err != nil {\n\t\th.GetLogger().Fatal(\"Creating priority task processor failed\", tag.Error(err))\n\t}\n\ttaskRateLimiter := task.NewRateLimiter(\n\t\th.GetLogger(),\n\t\th.GetMetricsClient(),\n\t\th.GetDomainCache(),\n\t\th.config,\n\t\th.controller,\n\t)\n\th.queueTaskProcessor = task.NewRateLimitedProcessor(taskProcessor, taskRateLimiter)\n\th.queueTaskProcessor.Start()\n\n\th.queueFactories = []queue.Factory{\n\t\tqueuev2.NewTransferQueueFactory(\n\t\t\th.queueTaskProcessor,\n\t\t\th.GetArchiverClient(),\n\t\t\th.workflowIDCache,\n\t\t),\n\t\tqueuev2.NewTimerQueueFactory(\n\t\t\th.queueTaskProcessor,\n\t\t\th.GetArchiverClient(),\n\t\t),\n\t}\n\n\th.historyEventNotifier = events.NewNotifier(h.GetTimeSource(), h.GetMetricsClient(), h.config.GetShardID)\n\t// events notifier must starts before controller\n\th.historyEventNotifier.Start()\n\n\th.failoverCoordinator = failover.NewCoordinator(\n\t\th.GetDomainManager(),\n\t\th.GetHistoryClient(),\n\t\th.GetTimeSource(),\n\t\th.GetDomainCache(),\n\t\th.config,\n\t\th.GetMetricsClient(),\n\t\th.GetLogger(),\n\t)\n\tif h.config.EnableGracefulFailover() {\n\t\th.failoverCoordinator.Start()\n\t}\n\n\th.controller.Start()\n\n\th.startWG.Done()\n}\n\n// Stop stops the handler\nfunc (h *handlerImpl) Stop() {\n\th.prepareToShutDown()\n\tif h.replicationBudgetManager != nil {\n\t\th.replicationBudgetManager.Stop()\n\t}\n\th.replicationTaskFetchers.Stop()\n\th.controller.Stop()\n\th.queueTaskProcessor.Stop()\n\th.historyEventNotifier.Stop()\n\th.failoverCoordinator.Stop()\n}\n\n// PrepareToStop starts graceful traffic drain in preparation for shutdown\nfunc (h *handlerImpl) PrepareToStop(remainingTime time.Duration) time.Duration {\n\th.GetLogger().Info(\"ShutdownHandler: Initiating shardController shutdown\")\n\th.controller.PrepareToStop()\n\th.GetLogger().Info(\"ShutdownHandler: Waiting for traffic to drain\")\n\tremainingTime = common.SleepWithMinDuration(shardOwnershipTransferDelay, remainingTime)\n\th.GetLogger().Info(\"ShutdownHandler: No longer taking rpc requests\")\n\th.prepareToShutDown()\n\treturn remainingTime\n}\n\nfunc (h *handlerImpl) prepareToShutDown() {\n\tatomic.StoreInt32(&h.shuttingDown, 1)\n}\n\nfunc (h *handlerImpl) isShuttingDown() bool {\n\treturn atomic.LoadInt32(&h.shuttingDown) != 0\n}\n\n// CreateEngine is implementation for HistoryEngineFactory used for creating the engine instance for shard\nfunc (h *handlerImpl) CreateEngine(\n\tshardContext shard.Context,\n) engine.Engine {\n\treturn engineimpl.NewEngineWithShardContext(\n\t\tshardContext,\n\t\th.GetVisibilityManager(),\n\t\th.GetMatchingClient(),\n\t\th.historyEventNotifier,\n\t\th.config,\n\t\th.replicationTaskFetchers,\n\t\th.GetMatchingRawClient(),\n\t\th.failoverCoordinator,\n\t\th.queueFactories,\n\t)\n}\n\n// Health is for health check\nfunc (h *handlerImpl) Health(ctx context.Context) (*types.HealthStatus, error) {\n\th.startWG.Wait()\n\th.GetLogger().Debug(\"History health check endpoint reached.\")\n\ths := &types.HealthStatus{Ok: true, Msg: \"OK\"}\n\treturn hs, nil\n}\n\n// RecordActivityTaskHeartbeat - Record Activity Task Heart beat.\nfunc (h *handlerImpl) RecordActivityTaskHeartbeat(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRecordActivityTaskHeartbeatRequest,\n) (resp *types.RecordActivityTaskHeartbeatResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRecordActivityTaskHeartbeatScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\theartbeatRequest := wrappedRequest.HeartbeatRequest\n\ttoken, err0 := h.tokenSerializer.Deserialize(heartbeatRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn nil, h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn nil, h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, \"\")\n\t}\n\n\tresponse, err2 := engine.RecordActivityTaskHeartbeat(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, \"\")\n\t}\n\n\treturn response, nil\n}\n\n// RecordActivityTaskStarted - Record Activity Task started.\nfunc (h *handlerImpl) RecordActivityTaskStarted(\n\tctx context.Context,\n\trecordRequest *types.RecordActivityTaskStartedRequest,\n) (resp *types.RecordActivityTaskStartedResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRecordActivityTaskStartedScope)\n\tdefer sw.Stop()\n\n\tdomainID := recordRequest.GetDomainUUID()\n\tworkflowExecution := recordRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\n\th.emitInfoOrDebugLog(\n\t\tdomainID,\n\t\t\"RecordActivityTaskStarted\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(recordRequest.WorkflowExecution.RunID),\n\t\ttag.WorkflowScheduleID(recordRequest.GetScheduleID()),\n\t)\n\n\tif recordRequest.GetDomainUUID() == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, workflowID, \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, workflowID, \"\")\n\t}\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, \"\")\n\t}\n\n\tresponse, err2 := engine.RecordActivityTaskStarted(ctx, recordRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, \"\")\n\t}\n\n\treturn response, nil\n}\n\n// RecordDecisionTaskStarted - Record Decision Task started.\nfunc (h *handlerImpl) RecordDecisionTaskStarted(\n\tctx context.Context,\n\trecordRequest *types.RecordDecisionTaskStartedRequest,\n) (resp *types.RecordDecisionTaskStartedResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRecordDecisionTaskStartedScope)\n\tdefer sw.Stop()\n\n\tdomainID := recordRequest.GetDomainUUID()\n\tworkflowExecution := recordRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\n\th.emitInfoOrDebugLog(\n\t\tdomainID,\n\t\t\"RecordDecisionTaskStarted\",\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(workflowExecution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(recordRequest.WorkflowExecution.RunID),\n\t\ttag.WorkflowScheduleID(recordRequest.GetScheduleID()),\n\t)\n\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, workflowID, runID)\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, workflowID, runID)\n\t}\n\n\tif recordRequest.PollRequest == nil || recordRequest.PollRequest.TaskList.GetName() == \"\" {\n\t\treturn nil, h.error(constants.ErrTaskListNotSet, scope, domainID, workflowID, runID)\n\t}\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\th.GetLogger().Error(\"RecordDecisionTaskStarted failed.\",\n\t\t\ttag.Error(err1),\n\t\t\ttag.WorkflowID(recordRequest.WorkflowExecution.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainName(domainID),\n\t\t\ttag.WorkflowRunID(recordRequest.WorkflowExecution.GetRunID()),\n\t\t\ttag.WorkflowScheduleID(recordRequest.GetScheduleID()),\n\t\t)\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresponse, err2 := engine.RecordDecisionTaskStarted(ctx, recordRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn response, nil\n}\n\n// RespondActivityTaskCompleted - records completion of an activity task\nfunc (h *handlerImpl) RespondActivityTaskCompleted(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRespondActivityTaskCompletedRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRespondActivityTaskCompletedScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tcompleteRequest := wrappedRequest.CompleteRequest\n\ttoken, err0 := h.tokenSerializer.Deserialize(completeRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\trunID := token.RunID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RespondActivityTaskCompleted(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// RespondActivityTaskFailed - records failure of an activity task\nfunc (h *handlerImpl) RespondActivityTaskFailed(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRespondActivityTaskFailedRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRespondActivityTaskFailedScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tfailRequest := wrappedRequest.FailedRequest\n\ttoken, err0 := h.tokenSerializer.Deserialize(failRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\trunID := token.RunID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RespondActivityTaskFailed(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// RespondActivityTaskCanceled - records failure of an activity task\nfunc (h *handlerImpl) RespondActivityTaskCanceled(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRespondActivityTaskCanceledRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRespondActivityTaskCanceledScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tcancelRequest := wrappedRequest.CancelRequest\n\ttoken, err0 := h.tokenSerializer.Deserialize(cancelRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\trunID := token.RunID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RespondActivityTaskCanceled(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// RespondDecisionTaskCompleted - records completion of a decision task\nfunc (h *handlerImpl) RespondDecisionTaskCompleted(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRespondDecisionTaskCompletedRequest,\n) (resp *types.HistoryRespondDecisionTaskCompletedResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRespondDecisionTaskCompletedScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tcompleteRequest := wrappedRequest.CompleteRequest\n\tif len(completeRequest.Decisions) == 0 {\n\t\tscope.IncCounter(metrics.EmptyCompletionDecisionsCounter)\n\t}\n\ttoken, err0 := h.tokenSerializer.Deserialize(completeRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn nil, h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\th.GetLogger().Debug(fmt.Sprintf(\"RespondDecisionTaskCompleted. DomainID: %v, WorkflowID: %v, RunID: %v, ScheduleID: %v\",\n\t\ttoken.DomainID,\n\t\ttoken.WorkflowID,\n\t\ttoken.RunID,\n\t\ttoken.ScheduleID))\n\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn nil, h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\trunID := token.RunID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresponse, err2 := engine.RespondDecisionTaskCompleted(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn response, nil\n}\n\n// RespondDecisionTaskFailed - failed response to decision task\nfunc (h *handlerImpl) RespondDecisionTaskFailed(\n\tctx context.Context,\n\twrappedRequest *types.HistoryRespondDecisionTaskFailedRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRespondDecisionTaskFailedScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tfailedRequest := wrappedRequest.FailedRequest\n\ttoken, err0 := h.tokenSerializer.Deserialize(failedRequest.TaskToken)\n\tif err0 != nil {\n\t\terr0 = &types.BadRequestError{Message: fmt.Sprintf(\"Error deserializing task token. Error: %v\", err0)}\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\n\th.GetLogger().Debug(fmt.Sprintf(\"RespondDecisionTaskFailed. DomainID: %v, WorkflowID: %v, RunID: %v, ScheduleID: %v\",\n\t\ttoken.DomainID,\n\t\ttoken.WorkflowID,\n\t\ttoken.RunID,\n\t\ttoken.ScheduleID))\n\n\tif failedRequest != nil && failedRequest.GetCause() == types.DecisionTaskFailedCauseUnhandledDecision {\n\t\th.GetLogger().Info(\"Non-Deterministic Error\", tag.WorkflowDomainID(token.DomainID), tag.WorkflowID(token.WorkflowID), tag.WorkflowRunID(token.RunID))\n\t\tdomainName, err := h.GetDomainCache().GetDomainName(token.DomainID)\n\t\tvar domainTag metrics.Tag\n\n\t\tif err == nil {\n\t\t\tdomainTag = metrics.DomainTag(domainName)\n\t\t} else {\n\t\t\tdomainTag = metrics.DomainUnknownTag()\n\t\t}\n\n\t\tscope.Tagged(domainTag).IncCounter(metrics.CadenceErrNonDeterministicCounter)\n\t}\n\terr0 = validateTaskToken(token)\n\tif err0 != nil {\n\t\treturn h.error(err0, scope, domainID, \"\", \"\")\n\t}\n\tworkflowID := token.WorkflowID\n\trunID := token.RunID\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RespondDecisionTaskFailed(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// StartWorkflowExecution - creates a new workflow execution\nfunc (h *handlerImpl) StartWorkflowExecution(\n\tctx context.Context,\n\twrappedRequest *types.HistoryStartWorkflowExecutionRequest,\n) (resp *types.StartWorkflowExecutionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryStartWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tstartRequest := wrappedRequest.StartRequest\n\tworkflowID := startRequest.GetWorkflowID()\n\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, \"\")\n\t}\n\n\tresponse, err2 := engine.StartWorkflowExecution(ctx, wrappedRequest)\n\trunID := response.GetRunID()\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn response, nil\n}\n\n// DescribeHistoryHost returns information about the internal states of a history host\nfunc (h *handlerImpl) DescribeHistoryHost(\n\tctx context.Context,\n\trequest *types.DescribeHistoryHostRequest,\n) (resp *types.DescribeHistoryHostResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tnumOfItemsInCacheByID, numOfItemsInCacheByName := h.GetDomainCache().GetCacheSize()\n\tstatus := \"\"\n\tswitch h.controller.Status() {\n\tcase common.DaemonStatusInitialized:\n\t\tstatus = \"initialized\"\n\tcase common.DaemonStatusStarted:\n\t\tstatus = \"started\"\n\tcase common.DaemonStatusStopped:\n\t\tstatus = \"stopped\"\n\t}\n\n\tresp = &types.DescribeHistoryHostResponse{\n\t\tNumberOfShards: int32(h.controller.NumShards()),\n\t\tShardIDs:       h.controller.ShardIDs(),\n\t\tDomainCache: &types.DomainCacheInfo{\n\t\t\tNumOfItemsInCacheByID:   numOfItemsInCacheByID,\n\t\t\tNumOfItemsInCacheByName: numOfItemsInCacheByName,\n\t\t},\n\t\tShardControllerStatus: status,\n\t\tAddress:               h.GetHostInfo().GetAddress(),\n\t}\n\treturn resp, nil\n}\n\n// RemoveTask returns information about the internal states of a history host\nfunc (h *handlerImpl) RemoveTask(\n\tctx context.Context,\n\trequest *types.RemoveTaskRequest,\n) (retError error) {\n\texecutionMgr, err := h.GetExecutionManager(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tswitch taskType := commonconstants.TaskType(request.GetType()); taskType {\n\tcase commonconstants.TaskTypeTransfer:\n\t\treturn executionMgr.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\tTaskKey:      persistence.NewImmediateTaskKey(request.GetTaskID()),\n\t\t})\n\tcase commonconstants.TaskTypeTimer:\n\t\treturn executionMgr.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(0, request.GetVisibilityTimestamp()), request.GetTaskID()),\n\t\t})\n\tcase commonconstants.TaskTypeReplication:\n\t\treturn executionMgr.CompleteHistoryTask(ctx, &persistence.CompleteHistoryTaskRequest{\n\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\tTaskKey:      persistence.NewImmediateTaskKey(request.GetTaskID()),\n\t\t})\n\tdefault:\n\t\treturn constants.ErrInvalidTaskType\n\t}\n}\n\n// CloseShard closes a shard hosted by this instance\nfunc (h *handlerImpl) CloseShard(\n\tctx context.Context,\n\trequest *types.CloseShardRequest,\n) (retError error) {\n\th.controller.RemoveEngineForShard(int(request.GetShardID()))\n\treturn nil\n}\n\n// ResetQueue resets processing queue states\nfunc (h *handlerImpl) ResetQueue(\n\tctx context.Context,\n\trequest *types.ResetQueueRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryResetQueueScope)\n\tdefer sw.Stop()\n\n\tengine, err := h.controller.GetEngineForShard(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\tswitch taskType := commonconstants.TaskType(request.GetType()); taskType {\n\tcase commonconstants.TaskTypeTransfer:\n\t\terr = engine.ResetTransferQueue(ctx, request.GetClusterName())\n\tcase commonconstants.TaskTypeTimer:\n\t\terr = engine.ResetTimerQueue(ctx, request.GetClusterName())\n\tdefault:\n\t\terr = constants.ErrInvalidTaskType\n\t}\n\n\tif err != nil {\n\t\treturn h.error(err, scope, \"\", \"\", \"\")\n\t}\n\treturn nil\n}\n\n// DescribeQueue describes processing queue states\nfunc (h *handlerImpl) DescribeQueue(\n\tctx context.Context,\n\trequest *types.DescribeQueueRequest,\n) (resp *types.DescribeQueueResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryDescribeQueueScope)\n\tdefer sw.Stop()\n\n\tengine, err := h.controller.GetEngineForShard(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\tswitch taskType := commonconstants.TaskType(request.GetType()); taskType {\n\tcase commonconstants.TaskTypeTransfer:\n\t\tresp, err = engine.DescribeTransferQueue(ctx, request.GetClusterName())\n\tcase commonconstants.TaskTypeTimer:\n\t\tresp, err = engine.DescribeTimerQueue(ctx, request.GetClusterName())\n\tdefault:\n\t\terr = constants.ErrInvalidTaskType\n\t}\n\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, \"\", \"\", \"\")\n\t}\n\treturn resp, nil\n}\n\n// DescribeMutableState - returns the internal analysis of workflow execution state\nfunc (h *handlerImpl) DescribeMutableState(\n\tctx context.Context,\n\trequest *types.DescribeMutableStateRequest,\n) (resp *types.DescribeMutableStateResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryDescribeMutabelStateScope)\n\tdefer sw.Stop()\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := request.Execution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.DescribeMutableState(ctx, request)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\treturn resp, nil\n}\n\n// GetMutableState - returns the id of the next event in the execution's history\nfunc (h *handlerImpl) GetMutableState(\n\tctx context.Context,\n\tgetRequest *types.GetMutableStateRequest,\n) (resp *types.GetMutableStateResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryGetMutableStateScope)\n\tdefer sw.Stop()\n\n\tdomainID := getRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := getRequest.Execution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetWorkflowID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.GetMutableState(ctx, getRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\treturn resp, nil\n}\n\n// PollMutableState - returns the id of the next event in the execution's history\nfunc (h *handlerImpl) PollMutableState(\n\tctx context.Context,\n\tgetRequest *types.PollMutableStateRequest,\n) (resp *types.PollMutableStateResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryPollMutableStateScope)\n\tdefer sw.Stop()\n\n\tdomainID := getRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := getRequest.Execution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.PollMutableState(ctx, getRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\treturn resp, nil\n}\n\n// DescribeWorkflowExecution returns information about the specified workflow execution.\nfunc (h *handlerImpl) DescribeWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryDescribeWorkflowExecutionRequest,\n) (resp *types.DescribeWorkflowExecutionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryDescribeWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := request.Request.Execution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.DescribeWorkflowExecution(ctx, request)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\treturn resp, nil\n}\n\n// RequestCancelWorkflowExecution - requests cancellation of a workflow\nfunc (h *handlerImpl) RequestCancelWorkflowExecution(\n\tctx context.Context,\n\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRequestCancelWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" || request.CancelRequest.GetDomain() == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tcancelRequest := request.CancelRequest\n\th.GetLogger().Debug(fmt.Sprintf(\"RequestCancelWorkflowExecution. DomainID: %v/%v, WorkflowID: %v, RunID: %v.\",\n\t\tcancelRequest.GetDomain(),\n\t\trequest.GetDomainUUID(),\n\t\tcancelRequest.WorkflowExecution.GetWorkflowID(),\n\t\tcancelRequest.WorkflowExecution.GetRunID()))\n\n\tworkflowID := cancelRequest.WorkflowExecution.GetWorkflowID()\n\trunID := cancelRequest.WorkflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RequestCancelWorkflowExecution(ctx, request)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// SignalWorkflowExecution is used to send a signal event to running workflow execution.  This results in\n// WorkflowExecutionSignaled event recorded in the history and a decision task being created for the execution.\nfunc (h *handlerImpl) SignalWorkflowExecution(\n\tctx context.Context,\n\twrappedRequest *types.HistorySignalWorkflowExecutionRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistorySignalWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := wrappedRequest.SignalRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.SignalWorkflowExecution(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// SignalWithStartWorkflowExecution is used to ensure sending a signal event to a workflow execution.\n// If workflow is running, this results in WorkflowExecutionSignaled event recorded in the history\n// and a decision task being created for the execution.\n// If workflow is not running or not found, this results in WorkflowExecutionStarted and WorkflowExecutionSignaled\n// event recorded in history, and a decision task being created for the execution\nfunc (h *handlerImpl) SignalWithStartWorkflowExecution(\n\tctx context.Context,\n\twrappedRequest *types.HistorySignalWithStartWorkflowExecutionRequest,\n) (resp *types.StartWorkflowExecutionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistorySignalWithStartWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tsignalWithStartRequest := wrappedRequest.SignalWithStartRequest\n\tworkflowID := signalWithStartRequest.GetWorkflowID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, \"\")\n\t}\n\n\tresp, err2 := engine.SignalWithStartWorkflowExecution(ctx, wrappedRequest)\n\tif err2 == nil {\n\t\treturn resp, nil\n\t}\n\t// Two simultaneous SignalWithStart requests might try to start a workflow at the same time.\n\t// This can result in one of the requests failing with one of two possible errors:\n\t//    1) If it is a brand new WF ID, one of the requests can fail with WorkflowExecutionAlreadyStartedError\n\t//       (createMode is persistence.CreateWorkflowModeBrandNew)\n\t//    2) If it an already existing WF ID, one of the requests can fail with a CurrentWorkflowConditionFailedError\n\t//       (createMode is persisetence.CreateWorkflowModeWorkflowIDReuse)\n\t// If either error occurs, just go ahead and retry. It should succeed on the subsequent attempt.\n\tvar e1 *persistence.WorkflowExecutionAlreadyStartedError\n\tvar e2 *persistence.CurrentWorkflowConditionFailedError\n\tif !errors.As(err2, &e1) && !errors.As(err2, &e2) {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, resp.GetRunID())\n\t}\n\n\tresp, err2 = engine.SignalWithStartWorkflowExecution(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, resp.GetRunID())\n\t}\n\treturn resp, nil\n}\n\n// RemoveSignalMutableState is used to remove a signal request ID that was previously recorded.  This is currently\n// used to clean execution info when signal decision finished.\nfunc (h *handlerImpl) RemoveSignalMutableState(\n\tctx context.Context,\n\twrappedRequest *types.RemoveSignalMutableStateRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRemoveSignalMutableStateScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := wrappedRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RemoveSignalMutableState(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// TerminateWorkflowExecution terminates an existing workflow execution by recording WorkflowExecutionTerminated event\n// in the history and immediately terminating the execution instance.\nfunc (h *handlerImpl) TerminateWorkflowExecution(\n\tctx context.Context,\n\twrappedRequest *types.HistoryTerminateWorkflowExecutionRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryTerminateWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := wrappedRequest.TerminateRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.TerminateWorkflowExecution(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// ResetWorkflowExecution reset an existing workflow execution\n// in the history and immediately terminating the execution instance.\nfunc (h *handlerImpl) ResetWorkflowExecution(\n\tctx context.Context,\n\twrappedRequest *types.HistoryResetWorkflowExecutionRequest,\n) (resp *types.ResetWorkflowExecutionResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryResetWorkflowExecutionScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tdomainID := wrappedRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := wrappedRequest.ResetRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.ResetWorkflowExecution(ctx, wrappedRequest)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn resp, nil\n}\n\n// QueryWorkflow queries a types.\nfunc (h *handlerImpl) QueryWorkflow(\n\tctx context.Context,\n\trequest *types.HistoryQueryWorkflowRequest,\n) (resp *types.HistoryQueryWorkflowResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryQueryWorkflowScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowID := request.GetRequest().GetExecution().GetWorkflowID()\n\trunID := request.GetRequest().GetExecution().GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn nil, h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err2 := engine.QueryWorkflow(ctx, request)\n\tif err2 != nil {\n\t\treturn nil, h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn resp, nil\n}\n\n// ScheduleDecisionTask is used for creating a decision task for already started workflow execution.  This is mainly\n// used by transfer queue processor during the processing of StartChildWorkflowExecution task, where it first starts\n// child execution without creating the decision task and then calls this API after updating the mutable state of\n// parent execution.\nfunc (h *handlerImpl) ScheduleDecisionTask(\n\tctx context.Context,\n\trequest *types.ScheduleDecisionTaskRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryScheduleDecisionTaskScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tif request.WorkflowExecution == nil {\n\t\treturn h.error(constants.ErrWorkflowExecutionNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := request.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.ScheduleDecisionTask(ctx, request)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// RecordChildExecutionCompleted is used for reporting the completion of child workflow execution to parent.\n// This is mainly called by transfer queue processor during the processing of DeleteExecution task.\nfunc (h *handlerImpl) RecordChildExecutionCompleted(\n\tctx context.Context,\n\trequest *types.RecordChildExecutionCompletedRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRecordChildExecutionCompletedScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tif request.WorkflowExecution == nil {\n\t\treturn h.error(constants.ErrWorkflowExecutionNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := request.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.RecordChildExecutionCompleted(ctx, request)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// ResetStickyTaskList reset the volatile information in mutable state of a given types.\n// Volatile information are the information related to client, such as:\n// 1. StickyTaskList\n// 2. StickyScheduleToStartTimeout\n// 3. ClientLibraryVersion\n// 4. ClientFeatureVersion\n// 5. ClientImpl\nfunc (h *handlerImpl) ResetStickyTaskList(\n\tctx context.Context,\n\tresetRequest *types.HistoryResetStickyTaskListRequest,\n) (resp *types.HistoryResetStickyTaskListResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryResetStickyTaskListScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tdomainID := resetRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn nil, h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn nil, h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowID := resetRequest.Execution.GetWorkflowID()\n\trunID := resetRequest.Execution.GetRunID()\n\tengine, err := h.controller.GetEngine(workflowID)\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\tresp, err = engine.ResetStickyTaskList(ctx, resetRequest)\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\treturn resp, nil\n}\n\n// ReplicateEventsV2 is called by processor to replicate history events for passive domains\nfunc (h *handlerImpl) ReplicateEventsV2(\n\tctx context.Context,\n\treplicateRequest *types.ReplicateEventsV2Request,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryReplicateEventsV2Scope)\n\tdefer sw.Stop()\n\n\tdomainID := replicateRequest.GetDomainUUID()\n\tif domainID == \"\" {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowExecution := replicateRequest.WorkflowExecution\n\tworkflowID := workflowExecution.GetWorkflowID()\n\trunID := workflowExecution.GetRunID()\n\tengine, err1 := h.controller.GetEngine(workflowID)\n\tif err1 != nil {\n\t\treturn h.error(err1, scope, domainID, workflowID, runID)\n\t}\n\n\terr2 := engine.ReplicateEventsV2(ctx, replicateRequest)\n\tif err2 != nil {\n\t\treturn h.error(err2, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// SyncShardStatus is called by processor to sync history shard information from another cluster\nfunc (h *handlerImpl) SyncShardStatus(\n\tctx context.Context,\n\tsyncShardStatusRequest *types.SyncShardStatusRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistorySyncShardStatusScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, \"\", \"\", \"\")\n\t}\n\n\tif syncShardStatusRequest.SourceCluster == \"\" {\n\t\treturn h.error(constants.ErrSourceClusterNotSet, scope, \"\", \"\", \"\")\n\t}\n\n\tif syncShardStatusRequest.Timestamp == nil {\n\t\treturn h.error(constants.ErrTimestampNotSet, scope, \"\", \"\", \"\")\n\t}\n\n\t// shard ID is already provided in the request\n\tengine, err := h.controller.GetEngineForShard(int(syncShardStatusRequest.GetShardID()))\n\tif err != nil {\n\t\treturn h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\terr = engine.SyncShardStatus(ctx, syncShardStatusRequest)\n\tif err != nil {\n\t\treturn h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\treturn nil\n}\n\n// SyncActivity is called by processor to sync activity\nfunc (h *handlerImpl) SyncActivity(\n\tctx context.Context,\n\tsyncActivityRequest *types.SyncActivityRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistorySyncActivityScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := syncActivityRequest.GetDomainID()\n\tif syncActivityRequest.DomainID == \"\" || uuid.Parse(syncActivityRequest.GetDomainID()) == nil {\n\t\treturn h.error(constants.ErrDomainNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif ok := h.rateLimiter.Allow(); !ok {\n\t\treturn h.error(constants.ErrHistoryHostThrottle, scope, domainID, \"\", \"\")\n\t}\n\n\tif syncActivityRequest.WorkflowID == \"\" {\n\t\treturn h.error(constants.ErrWorkflowIDNotSet, scope, domainID, \"\", \"\")\n\t}\n\n\tif syncActivityRequest.RunID == \"\" || uuid.Parse(syncActivityRequest.GetRunID()) == nil {\n\t\treturn h.error(constants.ErrRunIDNotValid, scope, domainID, \"\", \"\")\n\t}\n\n\tworkflowID := syncActivityRequest.GetWorkflowID()\n\trunID := syncActivityRequest.GetRunID()\n\tengine, err := h.controller.GetEngine(workflowID)\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\terr = engine.SyncActivity(ctx, syncActivityRequest)\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// GetReplicationMessages is called by remote peers to get replicated messages for cross DC replication\nfunc (h *handlerImpl) GetReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetReplicationMessagesRequest,\n) (resp *types.GetReplicationMessagesResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\th.GetLogger().Debug(\"Received GetReplicationMessages call.\")\n\n\tmetricsScope, sw := h.startRequestProfile(ctx, metrics.HistoryGetReplicationMessagesScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tmsgs := h.getReplicationShardMessages(ctx, request)\n\tresponse := h.buildGetReplicationMessagesResponse(metricsScope, msgs)\n\n\th.GetLogger().Debug(\"GetReplicationMessages succeeded.\")\n\treturn response, nil\n}\n\n// getReplicationShardMessages gets replication messages from all the shards of the request\n// it queries the replication tasks from each shard in parallel\nfunc (h *handlerImpl) getReplicationShardMessages(\n\tctx context.Context,\n\trequest *types.GetReplicationMessagesRequest,\n) []replicationShardMessages {\n\tvar (\n\t\twg      sync.WaitGroup\n\t\tmx      sync.Mutex\n\t\tresults = make([]replicationShardMessages, 0, len(request.Tokens))\n\t)\n\n\twg.Add(len(request.Tokens))\n\tfor _, token := range request.Tokens {\n\t\tgo func(token *types.ReplicationToken) {\n\t\t\tdefer wg.Done()\n\n\t\t\tengine, err := h.controller.GetEngineForShard(int(token.GetShardID()))\n\t\t\tif err != nil {\n\t\t\t\th.GetLogger().Warn(\"History engine not found for shard\", tag.Error(err))\n\t\t\t\treturn\n\t\t\t}\n\t\t\tmsgs, err := engine.GetReplicationMessages(\n\t\t\t\tctx,\n\t\t\t\trequest.GetClusterName(),\n\t\t\t\ttoken.GetLastRetrievedMessageID(),\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\th.GetLogger().Warn(\"Failed to get replication tasks for shard\", tag.Error(err))\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tmx.Lock()\n\t\t\tdefer mx.Unlock()\n\n\t\t\tresults = append(results, replicationShardMessages{\n\t\t\t\tReplicationMessages:  msgs,\n\t\t\t\tshardID:              token.GetShardID(),\n\t\t\t\tsize:                 proto.FromReplicationMessages(msgs).Size(),\n\t\t\t\tearliestCreationTime: msgs.GetEarliestCreationTime(),\n\t\t\t})\n\t\t}(token)\n\t}\n\n\twg.Wait()\n\treturn results\n}\n\n// buildGetReplicationMessagesResponse builds a new GetReplicationMessagesResponse from shard results\n// The response can be partial if the total size of the response exceeds the max size.\n// In this case, responses with oldest replication tasks will be returned\nfunc (h *handlerImpl) buildGetReplicationMessagesResponse(metricsScope metrics.Scope, msgs []replicationShardMessages) *types.GetReplicationMessagesResponse {\n\t// Shards with large messages can cause the response to exceed the max size.\n\t// In this case, we need to skip some shard messages to make sure the result response size is within the limit.\n\t// To prevent a replication lag in the future, we should return the messages with the oldest replication task.\n\t// So we sort the shard messages by the earliest creation time of the replication task.\n\t// If the earliest creation time is the same, we compare the size of the message.\n\t// This will sure that shards with the oldest replication tasks will be processed first.\n\tsortReplicationShardMessages(msgs)\n\n\tvar (\n\t\tresponseSize    = 0\n\t\tmaxResponseSize = h.config.MaxResponseSize\n\t\tmessagesByShard = make(map[int32]*types.ReplicationMessages, len(msgs))\n\t)\n\n\tfor _, m := range msgs {\n\t\tif (responseSize + m.size) >= maxResponseSize {\n\t\t\tmetricsScope.Tagged(metrics.ShardIDTag(int(m.shardID))).IncCounter(metrics.ReplicationMessageTooLargePerShard)\n\n\t\t\t// Log shards that did not fit for debugging purposes\n\t\t\th.GetLogger().Warn(\"Replication messages did not fit in the response (history host)\",\n\t\t\t\ttag.ShardID(int(m.shardID)),\n\t\t\t\ttag.ResponseSize(m.size),\n\t\t\t\ttag.ResponseTotalSize(responseSize),\n\t\t\t\ttag.ResponseMaxSize(maxResponseSize),\n\t\t\t)\n\n\t\t\tcontinue\n\t\t}\n\t\tresponseSize += m.size\n\t\tmessagesByShard[m.shardID] = m.ReplicationMessages\n\t}\n\treturn &types.GetReplicationMessagesResponse{MessagesByShard: messagesByShard}\n}\n\n// replicationShardMessages wraps types.ReplicationMessages\n// and contains some metadata of the ReplicationMessages\ntype replicationShardMessages struct {\n\t*types.ReplicationMessages\n\t// shardID of the ReplicationMessages\n\tshardID int32\n\t// size of proto payload of ReplicationMessages\n\tsize int\n\t// earliestCreationTime of ReplicationMessages\n\tearliestCreationTime *int64\n}\n\n// sortReplicationShardMessages sorts the peer responses by the earliest creation time of the replication tasks\nfunc sortReplicationShardMessages(msgs []replicationShardMessages) {\n\tslices.SortStableFunc(msgs, cmpReplicationShardMessages)\n}\n\n// cmpReplicationShardMessages compares\n// two replicationShardMessages objects by earliest creation time\n// it can be used as a comparison func for slices.SortStableFunc\n// if a's or b's earliestCreationTime is nil, slices.SortStableFunc will put them to the end of a slice\n// otherwise it will compare the earliestCreationTime of the replication tasks\n// if earliestCreationTime is equal, it will compare the size of the response\nfunc cmpReplicationShardMessages(a, b replicationShardMessages) int {\n\t// a > b\n\tif a.earliestCreationTime == nil {\n\t\treturn 1\n\t}\n\t// a < b\n\tif b.earliestCreationTime == nil {\n\t\treturn -1\n\t}\n\n\t// if both are not nil, compare the creation time\n\tif *a.earliestCreationTime < *b.earliestCreationTime {\n\t\treturn -1\n\t}\n\n\tif *a.earliestCreationTime > *b.earliestCreationTime {\n\t\treturn 1\n\t}\n\n\t// if both equal, compare the size\n\tif a.size < b.size {\n\t\treturn -1\n\t}\n\n\tif a.size > b.size {\n\t\treturn 1\n\t}\n\n\treturn 0\n}\n\n// GetDLQReplicationMessages is called by remote peers to get replicated messages for DLQ merging\nfunc (h *handlerImpl) GetDLQReplicationMessages(\n\tctx context.Context,\n\trequest *types.GetDLQReplicationMessagesRequest,\n) (resp *types.GetDLQReplicationMessagesResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\t_, sw := h.startRequestProfile(ctx, metrics.HistoryGetDLQReplicationMessagesScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\ttaskInfoPerExecution := map[definition.WorkflowIdentifier][]*types.ReplicationTaskInfo{}\n\t// do batch based on workflow ID and run ID\n\tfor _, taskInfo := range request.GetTaskInfos() {\n\t\tidentity := definition.NewWorkflowIdentifier(\n\t\t\ttaskInfo.GetDomainID(),\n\t\t\ttaskInfo.GetWorkflowID(),\n\t\t\ttaskInfo.GetRunID(),\n\t\t)\n\t\tif _, ok := taskInfoPerExecution[identity]; !ok {\n\t\t\ttaskInfoPerExecution[identity] = []*types.ReplicationTaskInfo{}\n\t\t}\n\t\ttaskInfoPerExecution[identity] = append(taskInfoPerExecution[identity], taskInfo)\n\t}\n\n\tvar wg sync.WaitGroup\n\twg.Add(len(taskInfoPerExecution))\n\ttasksChan := make(chan *types.ReplicationTask, len(request.GetTaskInfos()))\n\thandleTaskInfoPerExecution := func(taskInfos []*types.ReplicationTaskInfo) {\n\t\tdefer wg.Done()\n\t\tif len(taskInfos) == 0 {\n\t\t\treturn\n\t\t}\n\n\t\tengine, err := h.controller.GetEngine(\n\t\t\ttaskInfos[0].GetWorkflowID(),\n\t\t)\n\t\tif err != nil {\n\t\t\th.GetLogger().Warn(\"History engine not found for workflow ID.\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\n\t\ttasks, err := engine.GetDLQReplicationMessages(\n\t\t\tctx,\n\t\t\ttaskInfos,\n\t\t)\n\t\tif err != nil {\n\t\t\th.GetLogger().Error(\"Failed to get dlq replication tasks.\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\n\t\tfor _, task := range tasks {\n\t\t\ttasksChan <- task\n\t\t}\n\t}\n\n\tfor _, replicationTaskInfos := range taskInfoPerExecution {\n\t\tgo handleTaskInfoPerExecution(replicationTaskInfos)\n\t}\n\twg.Wait()\n\tclose(tasksChan)\n\n\treplicationTasks := make([]*types.ReplicationTask, 0, len(tasksChan))\n\tfor task := range tasksChan {\n\t\treplicationTasks = append(replicationTasks, task)\n\t}\n\treturn &types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: replicationTasks,\n\t}, nil\n}\n\n// ReapplyEvents applies stale events to the current workflow and the current run\nfunc (h *handlerImpl) ReapplyEvents(\n\tctx context.Context,\n\trequest *types.HistoryReapplyEventsRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryReapplyEventsScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\tworkflowID := request.GetRequest().GetWorkflowExecution().GetWorkflowID()\n\trunID := request.GetRequest().GetWorkflowExecution().GetRunID()\n\tengine, err := h.controller.GetEngine(workflowID)\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\t// deserialize history event object\n\thistoryEvents, err := h.GetPayloadSerializer().DeserializeBatchEvents(&persistence.DataBlob{\n\t\tEncoding: commonconstants.EncodingTypeThriftRW,\n\t\tData:     request.GetRequest().GetEvents().GetData(),\n\t})\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\texecution := request.GetRequest().GetWorkflowExecution()\n\tif err := engine.ReapplyEvents(\n\t\tctx,\n\t\trequest.GetDomainUUID(),\n\t\texecution.GetWorkflowID(),\n\t\texecution.GetRunID(),\n\t\thistoryEvents,\n\t); err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\treturn nil\n}\n\nfunc (h *handlerImpl) CountDLQMessages(\n\tctx context.Context,\n\trequest *types.CountDLQMessagesRequest,\n) (resp *types.HistoryCountDLQMessagesResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryCountDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tg := &errgroup.Group{}\n\tvar mu sync.Mutex\n\tentries := map[types.HistoryDLQCountKey]int64{}\n\tfor _, shardID := range h.controller.ShardIDs() {\n\t\tshardID := shardID\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &e) }()\n\n\t\t\tengine, err := h.controller.GetEngineForShard(int(shardID))\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"dlq count for shard %d: %w\", shardID, err)\n\t\t\t}\n\n\t\t\tcounts, err := engine.CountDLQMessages(ctx, request.ForceFetch)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"dlq count for shard %d: %w\", shardID, err)\n\t\t\t}\n\n\t\t\tmu.Lock()\n\t\t\tdefer mu.Unlock()\n\t\t\tfor sourceCluster, count := range counts {\n\t\t\t\tkey := types.HistoryDLQCountKey{ShardID: shardID, SourceCluster: sourceCluster}\n\t\t\t\tentries[key] = count\n\t\t\t}\n\t\t\treturn nil\n\t\t})\n\t}\n\n\terr := g.Wait()\n\treturn &types.HistoryCountDLQMessagesResponse{Entries: entries}, h.error(err, scope, \"\", \"\", \"\")\n}\n\n// ReadDLQMessages reads replication DLQ messages\nfunc (h *handlerImpl) ReadDLQMessages(\n\tctx context.Context,\n\trequest *types.ReadDLQMessagesRequest,\n) (resp *types.ReadDLQMessagesResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryReadDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tengine, err := h.controller.GetEngineForShard(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\treturn engine.ReadDLQMessages(ctx, request)\n}\n\n// PurgeDLQMessages deletes replication DLQ messages\nfunc (h *handlerImpl) PurgeDLQMessages(\n\tctx context.Context,\n\trequest *types.PurgeDLQMessagesRequest,\n) (retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryPurgeDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tengine, err := h.controller.GetEngineForShard(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\treturn engine.PurgeDLQMessages(ctx, request)\n}\n\n// MergeDLQMessages reads and applies replication DLQ messages\nfunc (h *handlerImpl) MergeDLQMessages(\n\tctx context.Context,\n\trequest *types.MergeDLQMessagesRequest,\n) (resp *types.MergeDLQMessagesResponse, retError error) {\n\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryMergeDLQMessagesScope)\n\tdefer sw.Stop()\n\n\tengine, err := h.controller.GetEngineForShard(int(request.GetShardID()))\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, \"\", \"\", \"\")\n\t}\n\n\treturn engine.MergeDLQMessages(ctx, request)\n}\n\n// RefreshWorkflowTasks refreshes all the tasks of a workflow\nfunc (h *handlerImpl) RefreshWorkflowTasks(\n\tctx context.Context,\n\trequest *types.HistoryRefreshWorkflowTasksRequest) (retError error) {\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRefreshWorkflowTasksScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn constants.ErrShuttingDown\n\t}\n\n\tdomainID := request.DomainUIID\n\texecution := request.GetRequest().GetExecution()\n\tworkflowID := execution.GetWorkflowID()\n\trunID := execution.GetWorkflowID()\n\tengine, err := h.controller.GetEngine(workflowID)\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\terr = engine.RefreshWorkflowTasks(\n\t\tctx,\n\t\tdomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\tRunID:      execution.RunID,\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn h.error(err, scope, domainID, workflowID, runID)\n\t}\n\n\treturn nil\n}\n\n// NotifyFailoverMarkers sends the failover markers to failover coordinator.\n// The coordinator decides when the failover finishes based on received failover marker.\nfunc (h *handlerImpl) NotifyFailoverMarkers(\n\tctx context.Context,\n\trequest *types.NotifyFailoverMarkersRequest,\n) (retError error) {\n\n\t_, sw := h.startRequestProfile(ctx, metrics.HistoryNotifyFailoverMarkersScope)\n\tdefer sw.Stop()\n\n\tfor _, token := range request.GetFailoverMarkerTokens() {\n\t\tmarker := token.GetFailoverMarker()\n\t\th.GetLogger().Debug(\"Handling failover maker\", tag.WorkflowDomainID(marker.GetDomainID()))\n\t\th.failoverCoordinator.ReceiveFailoverMarkers(token.GetShardIDs(), token.GetFailoverMarker())\n\t}\n\treturn nil\n}\n\nfunc (h *handlerImpl) GetCrossClusterTasks(\n\tctx context.Context,\n\trequest *types.GetCrossClusterTasksRequest,\n) (resp *types.GetCrossClusterTasksResponse, retError error) {\n\treturn nil, types.BadRequestError{Message: \"The cross-cluster feature has been deprecated.\"}\n}\n\nfunc (h *handlerImpl) RespondCrossClusterTasksCompleted(\n\tctx context.Context,\n\trequest *types.RespondCrossClusterTasksCompletedRequest,\n) (resp *types.RespondCrossClusterTasksCompletedResponse, retError error) {\n\treturn nil, types.BadRequestError{Message: \"The cross-cluster feature has been deprecated\"}\n}\n\nfunc (h *handlerImpl) GetFailoverInfo(\n\tctx context.Context,\n\trequest *types.GetFailoverInfoRequest,\n) (resp *types.GetFailoverInfoResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryGetFailoverInfoScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\tresp, err := h.failoverCoordinator.GetFailoverInfo(request.GetDomainID())\n\tif err != nil {\n\t\treturn nil, h.error(err, scope, request.GetDomainID(), \"\", \"\")\n\t}\n\treturn resp, nil\n}\n\nfunc (h *handlerImpl) RatelimitUpdate(\n\tctx context.Context,\n\trequest *types.RatelimitUpdateRequest,\n) (_ *types.RatelimitUpdateResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.GetLogger(), &retError) }()\n\th.startWG.Wait()\n\n\tscope, sw := h.startRequestProfile(ctx, metrics.HistoryRatelimitUpdateScope)\n\tdefer sw.Stop()\n\n\tif h.isShuttingDown() {\n\t\treturn nil, constants.ErrShuttingDown\n\t}\n\n\t// for now there is just one ratelimit-rpc type and one algorithm that makes use of it.\n\t// unpack the arg and pass it to the aggregator.\n\t// in the future, this should select the algorithm by the Any.ValueType, via a registry of some kind.\n\targ, err := rpc.AnyToAggregatorUpdate(request.Any)\n\tif err != nil {\n\t\treturn nil, h.error(fmt.Errorf(\"failed to map data to args: %w\", err), scope, \"\", \"\", \"\")\n\t}\n\terr = h.ratelimitAggregator.Update(arg)\n\tif err != nil {\n\t\treturn nil, h.error(fmt.Errorf(\"failed to update ratelimits: %w\", err), scope, \"\", \"\", \"\")\n\t}\n\n\t// collect the response data and pack it into an Any for the response.\n\t// like unpacking, this will eventually be handled by the registry above.\n\t//\n\t// \"_\" is ignoring \"used RPS\" data here.  it is likely useful for being friendlier\n\t// to brief, bursty-but-within-limits load, but that has not yet been built.\n\tweights, err := h.ratelimitAggregator.HostUsage(arg.ID, maps.Keys(arg.Load))\n\tif err != nil {\n\t\treturn nil, h.error(fmt.Errorf(\"failed to retrieve updated weights: %w\", err), scope, \"\", \"\", \"\")\n\t}\n\tresAny, err := rpc.AggregatorWeightsToAny(weights)\n\tif err != nil {\n\t\treturn nil, h.error(fmt.Errorf(\"failed to Any-package response: %w\", err), scope, \"\", \"\", \"\")\n\t}\n\n\treturn &types.RatelimitUpdateResponse{\n\t\tAny: resAny,\n\t}, nil\n}\n\n// convertError is a helper method to convert ShardOwnershipLostError from persistence layer returned by various\n// HistoryEngine API calls to ShardOwnershipLost error return by HistoryService for client to be redirected to the\n// correct shard.\nfunc (h *handlerImpl) convertError(err error) error {\n\tswitch err := err.(type) {\n\tcase *persistence.ShardOwnershipLostError:\n\t\tinfo, err2 := lookup.HistoryServerByShardID(h.GetMembershipResolver(), err.ShardID)\n\t\tif err2 != nil {\n\t\t\treturn shard.CreateShardOwnershipLostError(h.GetHostInfo(), membership.HostInfo{})\n\t\t}\n\n\t\treturn shard.CreateShardOwnershipLostError(h.GetHostInfo(), info)\n\tcase *persistence.WorkflowExecutionAlreadyStartedError:\n\t\treturn &types.InternalServiceError{Message: err.Msg}\n\tcase *persistence.CurrentWorkflowConditionFailedError:\n\t\treturn &types.InternalServiceError{Message: err.Msg}\n\tcase *persistence.TimeoutError:\n\t\treturn &types.InternalServiceError{Message: err.Msg}\n\tcase *persistence.TransactionSizeLimitError:\n\t\treturn &types.BadRequestError{Message: err.Msg}\n\t}\n\n\treturn err\n}\n\nfunc (h *handlerImpl) updateErrorMetric(\n\tscope metrics.Scope,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\terr error,\n) {\n\tlogger := h.GetLogger().Helper()\n\n\tvar yarpcE *yarpcerrors.Status\n\n\tvar shardOwnershipLostError *types.ShardOwnershipLostError\n\tvar eventAlreadyStartedError *types.EventAlreadyStartedError\n\tvar badRequestError *types.BadRequestError\n\tvar domainNotActiveError *types.DomainNotActiveError\n\tvar workflowExecutionAlreadyStartedError *types.WorkflowExecutionAlreadyStartedError\n\tvar entityNotExistsError *types.EntityNotExistsError\n\tvar workflowExecutionAlreadyCompletedError *types.WorkflowExecutionAlreadyCompletedError\n\tvar cancellationAlreadyRequestedError *types.CancellationAlreadyRequestedError\n\tvar limitExceededError *types.LimitExceededError\n\tvar retryTaskV2Error *types.RetryTaskV2Error\n\tvar serviceBusyError *types.ServiceBusyError\n\tvar internalServiceError *types.InternalServiceError\n\tvar queryFailedError *types.QueryFailedError\n\n\tif errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) {\n\t\tscope.IncCounter(metrics.CadenceErrContextTimeoutCounter)\n\t\treturn\n\t}\n\n\tif errors.As(err, &shardOwnershipLostError) {\n\t\tscope.IncCounter(metrics.CadenceErrShardOwnershipLostCounter)\n\n\t} else if errors.As(err, &eventAlreadyStartedError) {\n\t\tscope.IncCounter(metrics.CadenceErrEventAlreadyStartedCounter)\n\n\t} else if errors.As(err, &badRequestError) {\n\t\tscope.IncCounter(metrics.CadenceErrBadRequestCounter)\n\n\t} else if errors.As(err, &domainNotActiveError) {\n\t\tscope.IncCounter(metrics.CadenceErrDomainNotActiveCounter)\n\n\t} else if errors.As(err, &workflowExecutionAlreadyStartedError) {\n\t\tscope.IncCounter(metrics.CadenceErrExecutionAlreadyStartedCounter)\n\n\t} else if errors.As(err, &entityNotExistsError) {\n\t\tscope.IncCounter(metrics.CadenceErrEntityNotExistsCounter)\n\n\t} else if errors.As(err, &workflowExecutionAlreadyCompletedError) {\n\t\tscope.IncCounter(metrics.CadenceErrWorkflowExecutionAlreadyCompletedCounter)\n\n\t} else if errors.As(err, &cancellationAlreadyRequestedError) {\n\t\tscope.IncCounter(metrics.CadenceErrCancellationAlreadyRequestedCounter)\n\n\t} else if errors.As(err, &limitExceededError) {\n\t\tscope.IncCounter(metrics.CadenceErrLimitExceededCounter)\n\n\t} else if errors.As(err, &retryTaskV2Error) {\n\t\tscope.IncCounter(metrics.CadenceErrRetryTaskCounter)\n\n\t} else if errors.As(err, &serviceBusyError) {\n\t\tscope.IncCounter(metrics.CadenceErrServiceBusyCounter)\n\n\t} else if errors.As(err, &queryFailedError) {\n\t\tscope.IncCounter(metrics.CadenceErrQueryFailedCounter)\n\n\t} else if errors.As(err, &yarpcE) {\n\t\tif yarpcE.Code() == yarpcerrors.CodeDeadlineExceeded {\n\t\t\tscope.IncCounter(metrics.CadenceErrContextTimeoutCounter)\n\t\t}\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\n\t} else if errors.As(err, &internalServiceError) {\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\tlogger.Error(\"Internal service error\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainID(domainID))\n\n\t} else {\n\t\t// Default / unknown error fallback\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\tlogger.Error(\"Uncategorized error\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.WorkflowRunID(runID),\n\t\t\ttag.WorkflowDomainID(domainID))\n\t}\n}\n\nfunc (h *handlerImpl) error(\n\terr error,\n\tscope metrics.Scope,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) error {\n\terr = h.convertError(err)\n\n\th.updateErrorMetric(scope, domainID, workflowID, runID, err)\n\treturn err\n}\n\nfunc (h *handlerImpl) emitInfoOrDebugLog(\n\tdomainID string,\n\tmsg string,\n\ttags ...tag.Tag,\n) {\n\tif h.config.EnableDebugMode && h.config.EnableTaskInfoLogByDomainID(domainID) {\n\t\th.GetLogger().Info(msg, tags...)\n\t} else {\n\t\th.GetLogger().Debug(msg, tags...)\n\t}\n}\n\nfunc (h *handlerImpl) startRequestProfile(ctx context.Context, scope metrics.ScopeIdx) (metrics.Scope, metrics.Stopwatch) {\n\tmetricsScope := h.GetMetricsClient().Scope(scope, metrics.GetContextTags(ctx)...)\n\tmetricsScope.IncCounter(metrics.CadenceRequests)\n\tsw := metricsScope.StartTimer(metrics.CadenceLatency)\n\treturn metricsScope, sw\n}\n\nfunc validateTaskToken(token *common.TaskToken) error {\n\tif token.WorkflowID == \"\" {\n\t\treturn constants.ErrWorkflowIDNotSet\n\t}\n\tif token.RunID != \"\" && uuid.Parse(token.RunID) == nil {\n\t\treturn constants.ErrRunIDNotValid\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/history/handler/handler_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/global/rpc\"\n\t\"github.com/uber/cadence/common/quotas/global/shared\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/failover\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n)\n\nconst (\n\ttestWorkflowID    = \"test-workflow-id\"\n\ttestWorkflowRunID = \"test-workflow-run-id\"\n\ttestDomainID      = \"BF80C53A-ED56-4DD9-84EB-BE9AD4E45867\"\n\ttestValidUUID     = \"FCD00931-EBD4-4028-B67E-4DE624641255\"\n)\n\ntype (\n\thandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller               *gomock.Controller\n\t\tmockResource             *resource.Test\n\t\tmockShardController      *shard.MockController\n\t\tmockEngine               *engine.MockEngine\n\t\tmockWFCache              *workflowcache.MockWFCache\n\t\tmockTokenSerializer      *common.MockTaskTokenSerializer\n\t\tmockHistoryEventNotifier *events.MockNotifier\n\t\tmockRatelimiter          *quotas.MockLimiter\n\t\tmockFailoverCoordinator  *failover.MockCoordinator\n\n\t\thandler *handlerImpl\n\t}\n)\n\nfunc TestHandlerSuite(t *testing.T) {\n\ts := new(handlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *handlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.History)\n\ts.mockResource.Logger = testlogger.New(s.Suite.T())\n\ts.mockShardController = shard.NewMockController(s.controller)\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.mockWFCache = workflowcache.NewMockWFCache(s.controller)\n\ts.mockFailoverCoordinator = failover.NewMockCoordinator(s.controller)\n\ts.handler = NewHandler(s.mockResource, config.NewForTest(), s.mockWFCache).(*handlerImpl)\n\ts.handler.controller = s.mockShardController\n\ts.mockTokenSerializer = common.NewMockTaskTokenSerializer(s.controller)\n\ts.mockRatelimiter = quotas.NewMockLimiter(s.controller)\n\ts.handler.rateLimiter = s.mockRatelimiter\n\ts.handler.tokenSerializer = s.mockTokenSerializer\n\ts.handler.startWG.Done()\n}\n\nfunc (s *handlerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *handlerSuite) TestHealth() {\n\ths, err := s.handler.Health(context.Background())\n\ts.NoError(err)\n\ts.Equal(&types.HealthStatus{Ok: true, Msg: \"OK\"}, hs)\n}\n\nfunc (s *handlerSuite) TestRecordActivityTaskHeartbeat() {\n\ttestInput := map[string]struct {\n\t\tcaseName      string\n\t\tinput         *types.HistoryRecordActivityTaskHeartbeatRequest\n\t\texpected      *types.RecordActivityTaskHeartbeatResponse\n\t\texpectedError bool\n\t}{\n\t\t\"valid input\": {\n\t\t\tcaseName: \"valid input\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      &types.RecordActivityTaskHeartbeatResponse{CancelRequested: false},\n\t\t\texpectedError: false,\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tcaseName: \"empty domainID\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tcaseName: \"ratelimit exceeded\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tcaseName: \"token deserialization error\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tcaseName: \"invalid task token\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tcaseName: \"get engine error\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error\": {\n\t\t\tcaseName: \"engine error\",\n\t\t\tinput: &types.HistoryRecordActivityTaskHeartbeatRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tHeartbeatRequest: &types.RecordActivityTaskHeartbeatRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tswitch input.caseName {\n\t\t\tcase \"valid input\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), input.input).Return(input.expected, nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"empty domainID\":\n\t\t\tcase \"ratelimit exceeded\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\tcase \"token deserialization error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\tcase \"invalid task token\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\tcase \"get engine error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\tcase \"engine error\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordActivityTaskHeartbeat(gomock.Any(), input.input).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t}\n\t\t\tresponse, err := s.handler.RecordActivityTaskHeartbeat(context.Background(), input.input)\n\t\t\ts.Equal(input.expected, response)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRecordActivityTaskStarted() {\n\ttestInput := map[string]struct {\n\t\tcaseName      string\n\t\tinput         *types.RecordActivityTaskStartedRequest\n\t\texpected      *types.RecordActivityTaskStartedResponse\n\t\texpectedError bool\n\t}{\n\t\t\"valid input\": {\n\t\t\tcaseName: \"valid input\",\n\t\t\tinput: &types.RecordActivityTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      &types.RecordActivityTaskStartedResponse{Attempt: 1},\n\t\t\texpectedError: false,\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tcaseName: \"empty domainID\",\n\t\t\tinput: &types.RecordActivityTaskStartedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tcaseName: \"ratelimit exceeded\",\n\t\t\tinput: &types.RecordActivityTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tcaseName: \"get engine error\",\n\t\t\tinput: &types.RecordActivityTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error\": {\n\t\t\tcaseName: \"engine error\",\n\t\t\tinput: &types.RecordActivityTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tswitch input.caseName {\n\t\t\tcase \"valid input\":\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordActivityTaskStarted(gomock.Any(), input.input).Return(input.expected, nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"empty domainID\":\n\t\t\tcase \"ratelimit exceeded\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\tcase \"get engine error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\tcase \"engine error\":\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordActivityTaskStarted(gomock.Any(), input.input).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t}\n\t\t\tresponse, err := s.handler.RecordActivityTaskStarted(context.Background(), input.input)\n\t\t\ts.Equal(input.expected, response)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRecordDecisionTaskStarted() {\n\ttestInput := map[string]struct {\n\t\tinput         *types.RecordDecisionTaskStartedRequest\n\t\texpected      *types.RecordDecisionTaskStartedResponse\n\t\texpectedError bool\n\t}{\n\t\t\"valid input\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: &types.RecordDecisionTaskStartedResponse{\n\t\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t},\n\t\t\t\tAttempt: 1,\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error with ShardOwnershipLost\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: \"test-task-list\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"empty poll request\": {\n\t\t\tinput: &types.RecordDecisionTaskStartedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected:      nil,\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tswitch name {\n\t\t\tcase \"valid input\":\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordDecisionTaskStarted(gomock.Any(), input.input).Return(input.expected, nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"empty domainID\":\n\t\t\tcase \"ratelimit exceeded\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\tcase \"get engine error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\tcase \"engine error\":\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordDecisionTaskStarted(gomock.Any(), input.input).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"engine error with ShardOwnershipLost\":\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordDecisionTaskStarted(gomock.Any(), input.input).Return(nil, &persistence.ShardOwnershipLostError{ShardID: 123}).Times(1)\n\t\t\t\ts.mockResource.MembershipResolver.EXPECT().Lookup(service.History, string(rune(123)))\n\t\t\tcase \"empty poll request\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t}\n\n\t\t\tresponse, err := s.handler.RecordDecisionTaskStarted(context.Background(), input.input)\n\t\t\ts.Equal(input.expected, response)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondActivityTaskCompleted() {\n\ttestInput := map[string]struct {\n\t\tcaseName      string\n\t\tinput         *types.HistoryRespondActivityTaskCompletedRequest\n\t\texpectedError bool\n\t}{\n\t\t\"valid input\": {\n\t\t\tcaseName: \"valid input\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tcaseName: \"empty domainID\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tcaseName: \"ratelimit exceeded\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tcaseName: \"token deserialization error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tcaseName: \"invalid task token\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tcaseName: \"get engine error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error\": {\n\t\t\tcaseName: \"engine error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondActivityTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tResult:    []byte(\"result\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tswitch input.caseName {\n\t\t\tcase \"valid input\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskCompleted(gomock.Any(), input.input).Return(nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"empty domainID\":\n\t\t\tcase \"ratelimit exceeded\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\tcase \"token deserialization error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\tcase \"invalid task token\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\tcase \"get engine error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\tcase \"engine error\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskCompleted(gomock.Any(), input.input).Return(errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t}\n\t\t\terr := s.handler.RespondActivityTaskCompleted(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondActivityTaskFailed() {\n\ttestInput := map[string]struct {\n\t\tcaseName      string\n\t\tinput         *types.HistoryRespondActivityTaskFailedRequest\n\t\texpectedError bool\n\t}{\n\t\t\"valid input\": {\n\t\t\tcaseName: \"valid input\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tcaseName: \"empty domainID\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tcaseName: \"ratelimit exceeded\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tcaseName: \"token deserialization error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tcaseName: \"invalid task token\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tcaseName: \"get engine error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t\t\"engine error\": {\n\t\t\tcaseName: \"engine error\",\n\t\t\tinput: &types.HistoryRespondActivityTaskFailedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tFailedRequest: &types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tswitch input.caseName {\n\t\t\tcase \"valid input\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskFailed(gomock.Any(), input.input).Return(nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\tcase \"empty domainID\":\n\t\t\tcase \"ratelimit exceeded\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\tcase \"token deserialization error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\tcase \"invalid task token\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\tcase \"get engine error\":\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\tcase \"engine error\":\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskFailed(gomock.Any(), input.input).Return(errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t}\n\t\t\terr := s.handler.RespondActivityTaskFailed(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondActivityTaskCanceled() {\n\tvalidInput := &types.HistoryRespondActivityTaskCanceledRequest{\n\t\tDomainUUID: testDomainID,\n\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\tDetails:   []byte(\"Details\"),\n\t\t\tIdentity:  \"identity\",\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryRespondActivityTaskCanceledRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskCanceled(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput: &types.HistoryRespondActivityTaskCanceledRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCancelRequest: &types.RespondActivityTaskCanceledRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDetails:   []byte(\"Details\"),\n\t\t\t\t\tIdentity:  \"identity\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondActivityTaskCanceled(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RespondActivityTaskCanceled(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondDecisionTaskCompleted() {\n\tvalidReq := &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\tDomainUUID: testDomainID,\n\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\tDecisions: []*types.Decision{\n\t\t\t\t{\n\t\t\t\t\tDecisionType: types.DecisionTypeScheduleActivityTask.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tExecutionContext: nil,\n\t\t\tIdentity:         \"identity\",\n\t\t},\n\t}\n\tvalidResp := &types.HistoryRespondDecisionTaskCompletedResponse{\n\t\tStartedResponse: &types.RecordDecisionTaskStartedResponse{\n\t\t\tWorkflowType: &types.WorkflowType{},\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryRespondDecisionTaskCompletedRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"valid input\": {\n\t\t\tinput:         validReq,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), validReq).Return(validResp, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validReq,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tinput: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDecisions: []*types.Decision{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tinput: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDecisions: []*types.Decision{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput: &types.HistoryRespondDecisionTaskCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tCompleteRequest: &types.RespondDecisionTaskCompletedRequest{\n\t\t\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\t\t\tDecisions: []*types.Decision{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validReq,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskCompleted(gomock.Any(), validReq).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.RespondDecisionTaskCompleted(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondDecisionTaskFailed() {\n\tvalidInput := &types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID: testDomainID,\n\t\tFailedRequest: &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\tCause:     types.DecisionTaskFailedCauseBadBinary.Ptr(),\n\t\t\tDetails:   []byte(\"Details\"),\n\t\t\tIdentity:  \"identity\",\n\t\t},\n\t}\n\tspecialInput := &types.HistoryRespondDecisionTaskFailedRequest{\n\t\tDomainUUID: testDomainID,\n\t\tFailedRequest: &types.RespondDecisionTaskFailedRequest{\n\t\t\tTaskToken: []byte(\"task-token\"),\n\t\t\tCause:     types.DecisionTaskFailedCauseUnhandledDecision.Ptr(),\n\t\t\tDetails:   []byte(\"Details\"),\n\t\t\tIdentity:  \"identity\",\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryRespondDecisionTaskFailedRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskFailed(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryRespondDecisionTaskFailedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"token deserialization error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(nil, errors.New(\"some random error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"invalid task token\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: \"\",\n\t\t\t\t\tRunID:      \"\",\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskFailed(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"special domain\": {\n\t\t\tinput:         specialInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"name\", nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskFailed(gomock.Any(), specialInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"special domain2\": {\n\t\t\tinput:         specialInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockTokenSerializer.EXPECT().Deserialize(gomock.Any()).Return(&common.TaskToken{\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"\", errors.New(\"error\")).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RespondDecisionTaskFailed(gomock.Any(), specialInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RespondDecisionTaskFailed(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestDescribeHistoryHost() {\n\trequest := &types.DescribeHistoryHostRequest{\n\t\tHostAddress: common.StringPtr(\"test\"),\n\t}\n\n\tmockStatus := map[string]int32{\n\t\t\"initialized\": 0,\n\t\t\"started\":     1,\n\t\t\"stopped\":     2,\n\t}\n\n\tfor status, value := range mockStatus {\n\t\ts.mockResource.DomainCache.EXPECT().GetCacheSize().Return(int64(2), int64(3)).Times(1)\n\t\ts.mockShardController.EXPECT().Status().Return(value).Times(1)\n\t\ts.mockShardController.EXPECT().NumShards().Return(1)\n\t\ts.mockShardController.EXPECT().ShardIDs().Return([]int32{0})\n\t\tresp, err := s.handler.DescribeHistoryHost(context.Background(), request)\n\t\ts.NoError(err)\n\t\ts.Equal(resp.DomainCache, &types.DomainCacheInfo{\n\t\t\tNumOfItemsInCacheByID:   2,\n\t\t\tNumOfItemsInCacheByName: 3,\n\t\t})\n\t\ts.Equal(resp.ShardControllerStatus, status)\n\t}\n}\n\nfunc (s *handlerSuite) TestRemoveTask() {\n\tnow := time.Now()\n\ttestInput := map[string]struct {\n\t\trequest       *types.RemoveTaskRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"transfer task\": {\n\t\t\trequest: &types.RemoveTaskRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeTransfer)),\n\t\t\t\tTaskID:  int64(1),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockResource.ExecutionMgr.On(\"CompleteHistoryTask\", mock.Anything, &persistence.CompleteHistoryTaskRequest{\n\t\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(1),\n\t\t\t\t}).Return(nil).Once()\n\t\t\t},\n\t\t},\n\t\t\"timer task\": {\n\t\t\trequest: &types.RemoveTaskRequest{\n\t\t\t\tShardID:             0,\n\t\t\t\tType:                common.Int32Ptr(int32(commonconstants.TaskTypeTimer)),\n\t\t\t\tTaskID:              int64(1),\n\t\t\t\tVisibilityTimestamp: common.Int64Ptr(int64(now.UnixNano())),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockResource.ExecutionMgr.On(\"CompleteHistoryTask\", mock.Anything, &persistence.CompleteHistoryTaskRequest{\n\t\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\tTaskKey:      persistence.NewHistoryTaskKey(time.Unix(0, int64(now.UnixNano())), 1),\n\t\t\t\t}).Return(nil).Once()\n\t\t\t},\n\t\t},\n\t\t\"replication task\": {\n\t\t\trequest: &types.RemoveTaskRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeReplication)),\n\t\t\t\tTaskID:  int64(1),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockResource.ExecutionMgr.On(\"CompleteHistoryTask\", mock.Anything, &persistence.CompleteHistoryTaskRequest{\n\t\t\t\t\tTaskCategory: persistence.HistoryTaskCategoryReplication,\n\t\t\t\t\tTaskKey:      persistence.NewImmediateTaskKey(1),\n\t\t\t\t}).Return(nil).Once()\n\t\t\t},\n\t\t},\n\t\t\"invalid\": {\n\t\t\trequest: &types.RemoveTaskRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(100)),\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RemoveTask(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestCloseShard() {\n\trequest := &types.CloseShardRequest{\n\t\tShardID: 0,\n\t}\n\n\ts.mockShardController.EXPECT().RemoveEngineForShard(0).Return().Times(1)\n\terr := s.handler.CloseShard(context.Background(), request)\n\ts.NoError(err)\n}\n\nfunc (s *handlerSuite) TestResetQueue() {\n\ttestInput := map[string]struct {\n\t\trequest       *types.ResetQueueRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"getEngine error\": {\n\t\t\trequest: &types.ResetQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"transfer task\": {\n\t\t\trequest: &types.ResetQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeTransfer)),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetTransferQueue(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"timer task\": {\n\t\t\trequest: &types.ResetQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeTimer)),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetTimerQueue(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"invalid task\": {\n\t\t\trequest: &types.ResetQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(100)),\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.ResetQueue(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestDescribeQueue() {\n\ttestInput := map[string]struct {\n\t\trequest       *types.DescribeQueueRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"getEngine error\": {\n\t\t\trequest: &types.DescribeQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"transfer task\": {\n\t\t\trequest: &types.DescribeQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeTransfer)),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeTransferQueue(gomock.Any(), gomock.Any()).Return(&types.DescribeQueueResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"timer task\": {\n\t\t\trequest: &types.DescribeQueueRequest{\n\t\t\t\tShardID: 0,\n\t\t\t\tType:    common.Int32Ptr(int32(commonconstants.TaskTypeTimer)),\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeTimerQueue(gomock.Any(), gomock.Any()).Return(&types.DescribeQueueResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"invalid task\": {\n\t\t\trequest: &types.DescribeQueueRequest{\n\t\t\t\tType: common.Int32Ptr(int32(100)),\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(0).Return(s.mockEngine, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.DescribeQueue(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestDescribeMutableState() {\n\tvalidInput := &types.DescribeMutableStateRequest{\n\t\tDomainUUID: testDomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\trequest       *types.DescribeMutableStateRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"empty domainID\": {\n\t\t\trequest: &types.DescribeMutableStateRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getMutableState error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeMutableState(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeMutableState(gomock.Any(), validInput).Return(&types.DescribeMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.DescribeMutableState(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestGetMutableState() {\n\tvalidInput := &types.GetMutableStateRequest{\n\t\tDomainUUID: testDomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\trequest       *types.GetMutableStateRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"empty domainID\": {\n\t\t\trequest: &types.GetMutableStateRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getMutableState error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetMutableState(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetMutableState(gomock.Any(), validInput).Return(&types.GetMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.GetMutableState(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestPollMutableState() {\n\tvalidInput := &types.PollMutableStateRequest{\n\t\tDomainUUID: testDomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\trequest       *types.PollMutableStateRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"empty domainID\": {\n\t\t\trequest: &types.PollMutableStateRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getMutableState error\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollMutableState(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\trequest:       validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollMutableState(gomock.Any(), validInput).Return(&types.PollMutableStateResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.PollMutableState(context.Background(), input.request)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestDescribeWorkflowExecution() {\n\tvalidInput := &types.HistoryDescribeWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryDescribeWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeWorkflowExecution(gomock.Any(), validInput).Return(&types.DescribeWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeWorkflowExecution(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.DescribeWorkflowExecution(context.Background(), input.input)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRequestCancelWorkflowExecution() {\n\tvalidInput := &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tDomain: \"domain\",\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryRequestCancelWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RequestCancelWorkflowExecution(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestSignalWorkflowExecution() {\n\tvalidInput := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain: \"domain\",\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistorySignalWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWorkflowExecution(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistorySignalWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWorkflowExecution(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.SignalWorkflowExecution(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestSignalWithStartWorkflowExecution() {\n\tvalidInput := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistorySignalWithStartWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"special engine error and retry failure\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(nil, &persistence.WorkflowExecutionAlreadyStartedError{}).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"special engine error and retry success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(nil, &persistence.CurrentWorkflowConditionFailedError{}).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), validInput).Return(&types.StartWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.SignalWithStartWorkflowExecution(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRemoveSignalMutableState() {\n\tvalidInput := &types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID: testDomainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.RemoveSignalMutableStateRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RemoveSignalMutableState(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.RemoveSignalMutableStateRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RemoveSignalMutableState(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RemoveSignalMutableState(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestTerminateWorkflowExecution() {\n\tvalidInput := &types.HistoryTerminateWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\tDomain: \"domain\",\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryTerminateWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().TerminateWorkflowExecution(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().TerminateWorkflowExecution(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.TerminateWorkflowExecution(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestResetWorkflowExecution() {\n\tvalidInput := &types.HistoryResetWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tResetRequest: &types.ResetWorkflowExecutionRequest{\n\t\t\tDomain: \"domain\",\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t\tReason:                \"test\",\n\t\t\tDecisionFinishEventID: 1,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryResetWorkflowExecutionRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"valid input\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetWorkflowExecution(gomock.Any(), validInput).Return(&types.ResetWorkflowExecutionResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryResetWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetWorkflowExecution(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.ResetWorkflowExecution(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestQueryWorkflow() {\n\tvalidInput := &types.HistoryQueryWorkflowRequest{\n\t\tDomainUUID: testDomainID,\n\t\tRequest: &types.QueryWorkflowRequest{\n\t\t\tDomain: \"domain\",\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryQueryWorkflowRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryQueryWorkflowRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"queryWorkflow error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().QueryWorkflow(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().QueryWorkflow(gomock.Any(), validInput).Return(&types.HistoryQueryWorkflowResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.QueryWorkflow(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestScheduleDecisionTask() {\n\tvalidInput := &types.ScheduleDecisionTaskRequest{\n\t\tDomainUUID: testDomainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.ScheduleDecisionTaskRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"scheduleDecisionTask error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ScheduleDecisionTask(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ScheduleDecisionTask(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty execution\": {\n\t\t\tinput: &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.ScheduleDecisionTask(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRecordChildExecutionCompleted() {\n\tvalidInput := &types.RecordChildExecutionCompletedRequest{\n\t\tDomainUUID:  testDomainID,\n\t\tInitiatedID: 1,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.RecordChildExecutionCompletedRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"recordChildExecutionCompleted error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordChildExecutionCompleted(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RecordChildExecutionCompleted(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty execution\": {\n\t\t\tinput: &types.RecordChildExecutionCompletedRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RecordChildExecutionCompleted(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestResetStickyTaskList() {\n\tvalidInput := &types.HistoryResetStickyTaskListRequest{\n\t\tDomainUUID: testDomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryResetStickyTaskListRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.HistoryResetStickyTaskListRequest{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"resetStickyTaskList error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetStickyTaskList(gomock.Any(), validInput).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ResetStickyTaskList(gomock.Any(), validInput).Return(&types.HistoryResetStickyTaskListResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.ResetStickyTaskList(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestReplicateEventsV2() {\n\tvalidInput := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: testDomainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testValidUUID,\n\t\t},\n\t\tVersionHistoryItems: []*types.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: 1,\n\t\t\t\tVersion: 1,\n\t\t\t},\n\t\t},\n\t\tEvents: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\tData:         []byte{1, 2, 3},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.ReplicateEventsV2Request\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"getEngine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"replicateEventsV2 error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.ReplicateEventsV2(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestSyncShardStatus() {\n\tvalidInput := &types.SyncShardStatusRequest{\n\t\tSourceCluster: \"test\",\n\t\tShardID:       1,\n\t\tTimestamp:     common.Int64Ptr(time.Now().UnixNano()),\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.SyncShardStatusRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"get shard engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.ShardID)).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"syncShardStatus error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SyncShardStatus(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SyncShardStatus(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty sourceCluster\": {\n\t\t\tinput: &types.SyncShardStatusRequest{\n\t\t\t\tSourceCluster: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"missing timestamp\": {\n\t\t\tinput: &types.SyncShardStatusRequest{\n\t\t\t\tSourceCluster: \"test\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.SyncShardStatus(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestSyncActivity() {\n\tvalidInput := &types.SyncActivityRequest{\n\t\tDomainID:    testDomainID,\n\t\tWorkflowID:  testWorkflowID,\n\t\tRunID:       testValidUUID,\n\t\tVersion:     1,\n\t\tScheduledID: 1,\n\t\tDetails:     []byte{1, 2, 3},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.SyncActivityRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"ratelimit exceeded\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty domainID\": {\n\t\t\tinput: &types.SyncActivityRequest{\n\t\t\t\tDomainID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn:        func() {},\n\t\t},\n\t\t\"empty workflowID\": {\n\t\t\tinput: &types.SyncActivityRequest{\n\t\t\t\tDomainID:   testDomainID,\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty runID\": {\n\t\t\tinput: &types.SyncActivityRequest{\n\t\t\t\tDomainID:   testDomainID,\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      \"\",\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"syncActivity error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), validInput).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), validInput).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.SyncActivity(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestGetReplicationMessages() {\n\tvalidInput := &types.GetReplicationMessagesRequest{\n\t\tClusterName: \"test\",\n\t\tTokens: []*types.ReplicationToken{\n\t\t\t{\n\t\t\t\tShardID:                1,\n\t\t\t\tLastRetrievedMessageID: 1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tShardID:                2,\n\t\t\t\tLastRetrievedMessageID: 2,\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput            *types.GetReplicationMessagesRequest\n\t\tmockFn           func()\n\t\texpectedError    bool\n\t\texpectedResponse *types.GetReplicationMessagesResponse\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t\texpectedResponse: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[0].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[0].LastRetrievedMessageID).Return(&types.ReplicationMessages{}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[1].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[1].LastRetrievedMessageID).Return(&types.ReplicationMessages{}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{\n\t\t\t\t\t1: {},\n\t\t\t\t\t2: {},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine and cannot get task\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[0].ShardID)).Return(nil, errors.New(\"errors\")).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[1].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[1].LastRetrievedMessageID).Return(nil, errors.New(\"errors\")).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{},\n\t\t\t},\n\t\t},\n\t\t\"maxSize exceeds\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.config.MaxResponseSize = 0\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[0].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[0].LastRetrievedMessageID).Return(&types.ReplicationMessages{\n\t\t\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType: types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType: types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[1].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[1].LastRetrievedMessageID).Return(&types.ReplicationMessages{\n\t\t\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType: types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType: types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{},\n\t\t\t},\n\t\t},\n\t\t\"only first shard in response\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\tfirstShardMessages := &types.ReplicationMessages{\n\t\t\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(1000),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(1000),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tsecondShardMessages := &types.ReplicationMessages{\n\t\t\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(100),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(100),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\t// we want to allow only the second shard messages to be returned\n\t\t\t\ts.handler.config.MaxResponseSize = proto.FromReplicationMessages(secondShardMessages).Size() + 1\n\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[0].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[0].LastRetrievedMessageID).Return(firstShardMessages, nil).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(int(validInput.Tokens[1].ShardID)).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetReplicationMessages(gomock.Any(), validInput.ClusterName, validInput.Tokens[1].LastRetrievedMessageID).Return(secondShardMessages, nil).Times(1)\n\t\t\t},\n\t\t\texpectedResponse: &types.GetReplicationMessagesResponse{\n\t\t\t\tMessagesByShard: map[int32]*types.ReplicationMessages{\n\t\t\t\t\t// second shard is older than first shard\n\t\t\t\t\t// so it should only return the second shard messages\n\t\t\t\t\t// because the first shard messages will exceed the max response size\n\t\t\t\t\t2: {\n\t\t\t\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(100),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\t\t\t\t\t\t\tCreationTime: common.Int64Ptr(100),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.GetReplicationMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.Equal(input.expectedResponse, resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\tgoleak.VerifyNone(s.T())\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestGetDLQReplicationMessages() {\n\tvalidInput := &types.GetDLQReplicationMessagesRequest{\n\t\tTaskInfos: []*types.ReplicationTaskInfo{\n\t\t\t{\n\t\t\t\tDomainID:   testDomainID,\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\tmockResp := make([]*types.ReplicationTask, 0, 10)\n\tmockResp = append(mockResp, &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistory.Ptr(),\n\t})\n\n\tmockEmptyResp := make([]*types.ReplicationTask, 0)\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.GetDLQReplicationMessagesRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any()).Return(mockResp, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(gomock.Any()).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get task\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"empty task response\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetDLQReplicationMessages(gomock.Any(), gomock.Any()).Return(mockEmptyResp, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.GetDLQReplicationMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t\tgoleak.VerifyNone(s.T())\n\t\t})\n\n\t}\n}\n\nfunc (s *handlerSuite) TestReapplyEvents() {\n\tvalidInput := &types.HistoryReapplyEventsRequest{\n\t\tDomainUUID: testDomainID,\n\t\tRequest: &types.ReapplyEventsRequest{\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t\tEvents: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\tData:         []byte{},\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryReapplyEventsRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get serialized\": {\n\t\t\tinput: &types.HistoryReapplyEventsRequest{\n\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\tRequest: &types.ReapplyEventsRequest{\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t\tRunID:      testValidUUID,\n\t\t\t\t\t},\n\t\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\tData:         []byte{1, 2, 3, 4},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"reapplyEvents error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReapplyEvents(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.ReapplyEvents(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestCountDLQMessages() {\n\tvalidInput := &types.CountDLQMessagesRequest{\n\t\tForceFetch: true,\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.CountDLQMessagesRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().ShardIDs().Return([]int32{0}).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(gomock.Any()).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"countDLQMessages error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().ShardIDs().Return([]int32{0}).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any()).Return(map[string]int64{}, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().ShardIDs().Return([]int32{0}).Times(1)\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(gomock.Any()).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().CountDLQMessages(gomock.Any(), gomock.Any()).Return(map[string]int64{\n\t\t\t\t\t\"test\":  1,\n\t\t\t\t\t\"test2\": 2,\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\t_, err := s.handler.CountDLQMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestReadDLQMessages() {\n\tvalidInput := &types.ReadDLQMessagesRequest{\n\t\tShardID: 1,\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.ReadDLQMessagesRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"get shard engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"readDLQMessages error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReadDLQMessages(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\tresp := &types.ReadDLQMessagesResponse{}\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ReadDLQMessages(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\tresp, err := s.handler.ReadDLQMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Nil(resp)\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NotNil(resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestPurgeDLQMessages() {\n\tvalidInput := &types.PurgeDLQMessagesRequest{\n\t\tShardID: 1,\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.PurgeDLQMessagesRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"get shard engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"purgeDLQMessages error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PurgeDLQMessages(gomock.Any(), gomock.Any()).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PurgeDLQMessages(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.PurgeDLQMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestMergeDLQMessages() {\n\tvalidInput := &types.MergeDLQMessagesRequest{\n\t\tShardID: 1,\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.MergeDLQMessagesRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"get shard engine error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"mergeDLQMessages error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().MergeDLQMessages(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngineForShard(1).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().MergeDLQMessages(gomock.Any(), gomock.Any()).Return(&types.MergeDLQMessagesResponse{}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\t_, err := s.handler.MergeDLQMessages(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRefreshWorkflowTasks() {\n\tvalidInput := &types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: testDomainID,\n\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t},\n\t}\n\n\ttestInput := map[string]struct {\n\t\tinput         *types.HistoryRefreshWorkflowTasksRequest\n\t\texpectedError bool\n\t\tmockFn        func()\n\t}{\n\t\t\"shutting down\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.handler.shuttingDown = int32(1)\n\t\t\t},\n\t\t},\n\t\t\"cannot get engine\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(nil, errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"refreshWorkflowTasks error\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: true,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any(), gomock.Any()).Return(errors.New(\"error\")).Times(1)\n\t\t\t},\n\t\t},\n\t\t\"success\": {\n\t\t\tinput:         validInput,\n\t\t\texpectedError: false,\n\t\t\tmockFn: func() {\n\t\t\t\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\tinput.mockFn()\n\t\t\terr := s.handler.RefreshWorkflowTasks(context.Background(), input.input)\n\t\t\ts.handler.shuttingDown = int32(0)\n\t\t\tif input.expectedError {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestStartWorkflowExecution() {\n\n\trequest := &types.HistoryStartWorkflowExecutionRequest{\n\t\tDomainUUID: testDomainID,\n\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\tWorkflowID: testWorkflowID,\n\t\t},\n\t}\n\n\texpectedResponse := &types.StartWorkflowExecutionResponse{\n\t\tRunID: testWorkflowRunID,\n\t}\n\n\ts.mockShardController.EXPECT().GetEngine(testWorkflowID).Return(s.mockEngine, nil).Times(1)\n\ts.mockRatelimiter.EXPECT().Allow().Return(true).Times(1)\n\ts.mockEngine.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(expectedResponse, nil).Times(1)\n\n\tresponse, err := s.handler.StartWorkflowExecution(context.Background(), request)\n\ts.Equal(expectedResponse, response)\n\ts.Nil(err)\n}\n\nfunc (s *handlerSuite) TestEmitInfoOrDebugLog() {\n\t// test emitInfoOrDebugLog\n\ts.mockResource.Logger = testlogger.New(s.Suite.T())\n\ts.handler.emitInfoOrDebugLog(\"domain1\", \"test log\")\n}\n\nfunc (s *handlerSuite) TestValidateTaskToken() {\n\ttestInput := map[string]struct {\n\t\ttaskToken     *common.TaskToken\n\t\texpectedError error\n\t}{\n\t\t\"valid task token\": {\n\t\t\ttaskToken: &common.TaskToken{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testValidUUID,\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"empty workflow id\": {\n\t\t\ttaskToken: &common.TaskToken{\n\t\t\t\tWorkflowID: \"\",\n\t\t\t},\n\t\t\texpectedError: constants.ErrWorkflowIDNotSet,\n\t\t},\n\t\t\"invalid run id\": {\n\t\t\ttaskToken: &common.TaskToken{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      \"invalid\",\n\t\t\t},\n\t\t\texpectedError: constants.ErrRunIDNotValid,\n\t\t},\n\t}\n\n\tfor name, input := range testInput {\n\t\ts.Run(name, func() {\n\t\t\terr := validateTaskToken(input.taskToken)\n\t\t\ts.Equal(input.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestCorrectUseOfErrorHandling() {\n\n\ttests := map[string]struct {\n\t\tinput       error\n\t\texpectation func(scope *mocks.Scope)\n\t}{\n\t\t\"A deadline exceeded error\": {\n\t\t\tinput: context.DeadlineExceeded,\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrContextTimeoutCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"A cancelled error\": {\n\t\t\tinput: context.Canceled,\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrContextTimeoutCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"A shard ownership lost error\": {\n\t\t\tinput: &types.ShardOwnershipLostError{\n\t\t\t\tMessage: \"something is lost\",\n\t\t\t\tOwner:   \"owner\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrShardOwnershipLostCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"a workflow is already started\": {\n\t\t\tinput: &types.EventAlreadyStartedError{\n\t\t\t\tMessage: \"workflow already running\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrEventAlreadyStartedCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"a bad request\": {\n\t\t\tinput: &types.BadRequestError{\n\t\t\t\tMessage: \"bad request\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrBadRequestCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"domain is not active\": {\n\t\t\tinput: &types.DomainNotActiveError{\n\t\t\t\tMessage: \"domain not active\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrDomainNotActiveCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"workflow is already started err\": {\n\t\t\tinput: &types.WorkflowExecutionAlreadyStartedError{\n\t\t\t\tMessage: \"bad already started\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrExecutionAlreadyStartedCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"does not exist\": {\n\t\t\tinput: &types.EntityNotExistsError{\n\t\t\t\tMessage: \"the workflow doesn't exist\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrEntityNotExistsCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"already completed\": {\n\t\t\tinput: &types.WorkflowExecutionAlreadyCompletedError{\n\t\t\t\tMessage: \"the workflow is done\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrWorkflowExecutionAlreadyCompletedCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"Cancellation already requested\": {\n\t\t\tinput: &types.CancellationAlreadyRequestedError{\n\t\t\t\tMessage: \"the workflow is cancelled already\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrCancellationAlreadyRequestedCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"rate-limits\": {\n\t\t\tinput: &types.LimitExceededError{\n\t\t\t\tMessage: \"limits exceeded\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrLimitExceededCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"retry tasks\": {\n\t\t\tinput: &types.RetryTaskV2Error{\n\t\t\t\tMessage: \"limits exceeded\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrRetryTaskCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"service busy error\": {\n\t\t\tinput: &types.ServiceBusyError{\n\t\t\t\tMessage: \"limits exceeded - service is busy\",\n\t\t\t},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrServiceBusyCounter).Once()\n\t\t\t},\n\t\t},\n\t\t\"deadline exceeded\": {\n\t\t\tinput: yarpcerrors.DeadlineExceededErrorf(\"some deadline exceeded err\"),\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceErrContextTimeoutCounter).Once()\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceFailures).Once()\n\t\t\t},\n\t\t},\n\t\t\"internal error\": {\n\t\t\tinput: types.InternalServiceError{Message: \"internal error\"},\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceFailures).Once()\n\t\t\t},\n\t\t},\n\t\t\"uncategorized error\": {\n\t\t\tinput: errors.New(\"some random error\"),\n\t\t\texpectation: func(scope *mocks.Scope) {\n\t\t\t\tscope.Mock.On(\"IncCounter\", metrics.CadenceFailures).Once()\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\ts.Run(name, func() {\n\t\t\tscope := mocks.Scope{}\n\t\t\ttd.expectation(&scope)\n\t\t\th := handlerImpl{\n\t\t\t\tResource: resource.NewTest(s.T(), gomock.NewController(s.T()), 0),\n\t\t\t}\n\t\t\th.error(td.input, &scope, \"some-domain\", \"some-wf\", \"some-run\")\n\t\t\t// we're doing the args assertion in the On, so using mock.Anything to avoid having to duplicate this\n\t\t\t// a wrong metric being emitted will fail the mock.On() expectation. This will catch missing calls\n\t\t\tscope.Mock.AssertCalled(s.T(), \"IncCounter\", mock.Anything)\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestConvertError() {\n\ttestCases := []struct {\n\t\tname     string\n\t\tinput    error\n\t\texpected error\n\t}{\n\t\t{\n\t\t\tname: \"workflow already started error\",\n\t\t\tinput: &persistence.WorkflowExecutionAlreadyStartedError{\n\t\t\t\tMsg: \"workflow already started\",\n\t\t\t},\n\t\t\texpected: &types.InternalServiceError{\n\t\t\t\tMessage: \"workflow already started\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"current workflow condition failed error\",\n\t\t\tinput: &persistence.CurrentWorkflowConditionFailedError{\n\t\t\t\tMsg: \"current workflow condition failed\",\n\t\t\t},\n\t\t\texpected: &types.InternalServiceError{\n\t\t\t\tMessage: \"current workflow condition failed\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistence timeout error\",\n\t\t\tinput: &persistence.TimeoutError{\n\t\t\t\tMsg: \"persistence timeout\",\n\t\t\t},\n\t\t\texpected: &types.InternalServiceError{\n\t\t\t\tMessage: \"persistence timeout\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"transaction size limit error\",\n\t\t\tinput: &persistence.TransactionSizeLimitError{\n\t\t\t\tMsg: \"transaction size limit\",\n\t\t\t},\n\t\t\texpected: &types.BadRequestError{\n\t\t\t\tMessage: \"transaction size limit\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"shard ownership lost error\",\n\t\t\tinput: &persistence.ShardOwnershipLostError{\n\t\t\t\tShardID: 1,\n\t\t\t},\n\t\t\texpected: &types.ShardOwnershipLostError{\n\t\t\t\tOwner:   \"127.0.0.1:1234\",\n\t\t\t\tMessage: \"Shard is not owned by host: test_host\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.mockResource.MembershipResolver.EXPECT().Lookup(gomock.Any(), gomock.Any()).Return(membership.NewHostInfo(\"127.0.0.1:1234\"), nil).AnyTimes()\n\t\terr := s.handler.convertError(tc.input)\n\t\ts.Equal(tc.expected, err, tc.name)\n\t}\n}\n\nfunc TestRatelimitUpdate(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tres := resource.NewMockResource(ctrl)\n\tres.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tres.EXPECT().GetLogger().Return(testlogger.New(t)).AnyTimes()\n\tupdate, err := rpc.TestUpdateToAny(t, \"testhost\", time.Second, map[shared.GlobalKey]rpc.Calls{\n\t\t\"test:domain-user-limit\": {\n\t\t\tAllowed:  10,\n\t\t\tRejected: 5,\n\t\t},\n\t})\n\trequire.NoError(t, err)\n\talg, err := algorithm.New(\n\t\tmetrics.NewNoopMetricsClient(),\n\t\ttestlogger.New(t),\n\t\talgorithm.Config{\n\t\t\tNewDataWeight:  func(opts ...dynamicproperties.FilterOption) float64 { return 0.5 },\n\t\t\tUpdateInterval: func(opts ...dynamicproperties.FilterOption) time.Duration { return 3 * time.Second },\n\t\t\tDecayAfter:     func(opts ...dynamicproperties.FilterOption) time.Duration { return 6 * time.Second },\n\t\t\tGcAfter:        func(opts ...dynamicproperties.FilterOption) time.Duration { return time.Minute },\n\t\t},\n\t)\n\trequire.NoError(t, err)\n\th := &handlerImpl{\n\t\tResource:            res,\n\t\tratelimitAggregator: alg,\n\t}\n\n\tresp, err := h.RatelimitUpdate(context.Background(), &types.RatelimitUpdateRequest{\n\t\tAny: update,\n\t})\n\trequire.NoError(t, err)\n\tw, err := rpc.TestAnyToWeights(t, resp.Any)\n\trequire.NoError(t, err)\n\tassert.Equalf(t,\n\t\tmap[shared.GlobalKey]rpc.UpdateEntry{\n\t\t\t\"test:domain-user-limit\": {\n\t\t\t\tWeight: 1,\n\t\t\t\t// re 10 vs 15: used RPS only tracks accepted.\n\t\t\t\t//\n\t\t\t\t// this way we don't consider any incorrectly-rejected requests\n\t\t\t\t// when calculating our new limits.\n\t\t\t\tUsedRPS: 10,\n\t\t\t},\n\t\t},\n\t\tw,\n\t\t\"unexpected weights returned from aggregator or serialization.  if values differ in a reasonable way, possibly aggregator behavior changed?\",\n\t)\n}\n\nfunc Test_cmpReplicationShardMessages(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\ta, b replicationShardMessages\n\t\twant int\n\t}{\n\t\t\"a time is nil, b is empty\": {\n\t\t\ta: replicationShardMessages{earliestCreationTime: nil}, want: 1,\n\t\t},\n\t\t\"a time is not nil, b is empty\": {\n\t\t\ta: replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)}, want: -1,\n\t\t},\n\t\t\"a time is not nil, b time is nil\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: nil},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a time less b time\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a time greater b time\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\twant: 1,\n\t\t},\n\t\t\"a size less b size\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10), size: 10},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10), size: 20},\n\t\t\twant: -1,\n\t\t},\n\t\t\"a size greater b size\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10), size: 20},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10), size: 10},\n\t\t\twant: 1,\n\t\t},\n\t\t\"a equal b\": {\n\t\t\ta:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\tb:    replicationShardMessages{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\twant: 0,\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, c.want, cmpReplicationShardMessages(c.a, c.b))\n\t\t})\n\t}\n}\n\nfunc Test_sortReplicationShardMessages(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tmsgs []replicationShardMessages\n\t\twant []replicationShardMessages\n\t}{\n\t\t\"empty\": {},\n\t\t\"multiple nil, non nil earliestCreationTime\": {\n\t\t\tmsgs: []replicationShardMessages{\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\t},\n\t\t\twant: []replicationShardMessages{\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(10)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t},\n\t\t},\n\t\t\"multiple nil, non nil same earliestCreationTime, different size\": {\n\t\t\tmsgs: []replicationShardMessages{\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 50},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 30},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t},\n\t\t\twant: []replicationShardMessages{\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(20)},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 30},\n\t\t\t\t{earliestCreationTime: common.Int64Ptr(100), size: 50},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t\t{earliestCreationTime: nil},\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tsortReplicationShardMessages(c.msgs)\n\t\t\tassert.Equal(t, c.want, c.msgs)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/handler/interface.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -package handler github.com/uber/cadence/service/history/handler Handler\n//go:generate gowrap gen -g -p . -i Handler -t ../../templates/grpc.tmpl -o ../wrappers/grpc/grpc_handler_generated.go -v handler=GRPC -v package=historyv1 -v path=github.com/uber/cadence/.gen/proto/history/v1 -v prefix=History\n//go:generate gowrap gen -g -p ../../../.gen/go/history/historyserviceserver -i Interface -t ../../templates/thrift.tmpl -o ../wrappers/thrift/thrift_handler_generated.go -v handler=Thrift -v prefix=History\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// Handler interface for history service\ntype Handler interface {\n\t// Do not use embeded methods, otherwise, we got the following error from gowrap\n\t// and we only get this error from history/interface.go, not sure why\n\t// failed to parse interface declaration: Daemon: target declaration not found\n\t// service/history/interface.go:22: running \"gowrap\": exit status 1\n\t// common.Daemon\n\tStart()\n\tStop()\n\n\tPrepareToStop(time.Duration) time.Duration\n\tHealth(context.Context) (*types.HealthStatus, error)\n\tCloseShard(context.Context, *types.CloseShardRequest) error\n\tDescribeHistoryHost(context.Context, *types.DescribeHistoryHostRequest) (*types.DescribeHistoryHostResponse, error)\n\tDescribeMutableState(context.Context, *types.DescribeMutableStateRequest) (*types.DescribeMutableStateResponse, error)\n\tDescribeQueue(context.Context, *types.DescribeQueueRequest) (*types.DescribeQueueResponse, error)\n\tDescribeWorkflowExecution(context.Context, *types.HistoryDescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error)\n\tGetCrossClusterTasks(context.Context, *types.GetCrossClusterTasksRequest) (*types.GetCrossClusterTasksResponse, error)\n\tCountDLQMessages(context.Context, *types.CountDLQMessagesRequest) (*types.HistoryCountDLQMessagesResponse, error)\n\tGetDLQReplicationMessages(context.Context, *types.GetDLQReplicationMessagesRequest) (*types.GetDLQReplicationMessagesResponse, error)\n\tGetMutableState(context.Context, *types.GetMutableStateRequest) (*types.GetMutableStateResponse, error)\n\tGetReplicationMessages(context.Context, *types.GetReplicationMessagesRequest) (*types.GetReplicationMessagesResponse, error)\n\tMergeDLQMessages(context.Context, *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error)\n\tNotifyFailoverMarkers(context.Context, *types.NotifyFailoverMarkersRequest) error\n\tPollMutableState(context.Context, *types.PollMutableStateRequest) (*types.PollMutableStateResponse, error)\n\tPurgeDLQMessages(context.Context, *types.PurgeDLQMessagesRequest) error\n\tQueryWorkflow(context.Context, *types.HistoryQueryWorkflowRequest) (*types.HistoryQueryWorkflowResponse, error)\n\tReadDLQMessages(context.Context, *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error)\n\tReapplyEvents(context.Context, *types.HistoryReapplyEventsRequest) error\n\tRecordActivityTaskHeartbeat(context.Context, *types.HistoryRecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error)\n\tRecordActivityTaskStarted(context.Context, *types.RecordActivityTaskStartedRequest) (*types.RecordActivityTaskStartedResponse, error)\n\tRecordChildExecutionCompleted(context.Context, *types.RecordChildExecutionCompletedRequest) error\n\tRecordDecisionTaskStarted(context.Context, *types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error)\n\tRefreshWorkflowTasks(context.Context, *types.HistoryRefreshWorkflowTasksRequest) error\n\tRemoveSignalMutableState(context.Context, *types.RemoveSignalMutableStateRequest) error\n\tRemoveTask(context.Context, *types.RemoveTaskRequest) error\n\tReplicateEventsV2(context.Context, *types.ReplicateEventsV2Request) error\n\tRequestCancelWorkflowExecution(context.Context, *types.HistoryRequestCancelWorkflowExecutionRequest) error\n\tResetQueue(context.Context, *types.ResetQueueRequest) error\n\tResetStickyTaskList(context.Context, *types.HistoryResetStickyTaskListRequest) (*types.HistoryResetStickyTaskListResponse, error)\n\tResetWorkflowExecution(context.Context, *types.HistoryResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error)\n\tRespondActivityTaskCanceled(context.Context, *types.HistoryRespondActivityTaskCanceledRequest) error\n\tRespondActivityTaskCompleted(context.Context, *types.HistoryRespondActivityTaskCompletedRequest) error\n\tRespondActivityTaskFailed(context.Context, *types.HistoryRespondActivityTaskFailedRequest) error\n\tRespondCrossClusterTasksCompleted(context.Context, *types.RespondCrossClusterTasksCompletedRequest) (*types.RespondCrossClusterTasksCompletedResponse, error)\n\tRespondDecisionTaskCompleted(context.Context, *types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error)\n\tRespondDecisionTaskFailed(context.Context, *types.HistoryRespondDecisionTaskFailedRequest) error\n\tScheduleDecisionTask(context.Context, *types.ScheduleDecisionTaskRequest) error\n\tSignalWithStartWorkflowExecution(context.Context, *types.HistorySignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\tSignalWorkflowExecution(context.Context, *types.HistorySignalWorkflowExecutionRequest) error\n\tStartWorkflowExecution(context.Context, *types.HistoryStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error)\n\tSyncActivity(context.Context, *types.SyncActivityRequest) error\n\tSyncShardStatus(context.Context, *types.SyncShardStatusRequest) error\n\tTerminateWorkflowExecution(context.Context, *types.HistoryTerminateWorkflowExecutionRequest) error\n\tGetFailoverInfo(context.Context, *types.GetFailoverInfoRequest) (*types.GetFailoverInfoResponse, error)\n\tRatelimitUpdate(context.Context, *types.RatelimitUpdateRequest) (*types.RatelimitUpdateResponse, error)\n}\n"
  },
  {
    "path": "service/history/handler/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package handler -source interface.go -destination interface_mock.go -package handler github.com/uber/cadence/service/history/handler Handler\n//\n\n// Package handler is a generated GoMock package.\npackage handler\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// CloseShard mocks base method.\nfunc (m *MockHandler) CloseShard(arg0 context.Context, arg1 *types.CloseShardRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CloseShard\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CloseShard indicates an expected call of CloseShard.\nfunc (mr *MockHandlerMockRecorder) CloseShard(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CloseShard\", reflect.TypeOf((*MockHandler)(nil).CloseShard), arg0, arg1)\n}\n\n// CountDLQMessages mocks base method.\nfunc (m *MockHandler) CountDLQMessages(arg0 context.Context, arg1 *types.CountDLQMessagesRequest) (*types.HistoryCountDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CountDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryCountDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CountDLQMessages indicates an expected call of CountDLQMessages.\nfunc (mr *MockHandlerMockRecorder) CountDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CountDLQMessages\", reflect.TypeOf((*MockHandler)(nil).CountDLQMessages), arg0, arg1)\n}\n\n// DescribeHistoryHost mocks base method.\nfunc (m *MockHandler) DescribeHistoryHost(arg0 context.Context, arg1 *types.DescribeHistoryHostRequest) (*types.DescribeHistoryHostResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeHistoryHost\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeHistoryHostResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeHistoryHost indicates an expected call of DescribeHistoryHost.\nfunc (mr *MockHandlerMockRecorder) DescribeHistoryHost(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeHistoryHost\", reflect.TypeOf((*MockHandler)(nil).DescribeHistoryHost), arg0, arg1)\n}\n\n// DescribeMutableState mocks base method.\nfunc (m *MockHandler) DescribeMutableState(arg0 context.Context, arg1 *types.DescribeMutableStateRequest) (*types.DescribeMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeMutableState\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeMutableState indicates an expected call of DescribeMutableState.\nfunc (mr *MockHandlerMockRecorder) DescribeMutableState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeMutableState\", reflect.TypeOf((*MockHandler)(nil).DescribeMutableState), arg0, arg1)\n}\n\n// DescribeQueue mocks base method.\nfunc (m *MockHandler) DescribeQueue(arg0 context.Context, arg1 *types.DescribeQueueRequest) (*types.DescribeQueueResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeQueue\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeQueueResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeQueue indicates an expected call of DescribeQueue.\nfunc (mr *MockHandlerMockRecorder) DescribeQueue(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeQueue\", reflect.TypeOf((*MockHandler)(nil).DescribeQueue), arg0, arg1)\n}\n\n// DescribeWorkflowExecution mocks base method.\nfunc (m *MockHandler) DescribeWorkflowExecution(arg0 context.Context, arg1 *types.HistoryDescribeWorkflowExecutionRequest) (*types.DescribeWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) DescribeWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).DescribeWorkflowExecution), arg0, arg1)\n}\n\n// GetCrossClusterTasks mocks base method.\nfunc (m *MockHandler) GetCrossClusterTasks(arg0 context.Context, arg1 *types.GetCrossClusterTasksRequest) (*types.GetCrossClusterTasksResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCrossClusterTasks\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetCrossClusterTasksResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCrossClusterTasks indicates an expected call of GetCrossClusterTasks.\nfunc (mr *MockHandlerMockRecorder) GetCrossClusterTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCrossClusterTasks\", reflect.TypeOf((*MockHandler)(nil).GetCrossClusterTasks), arg0, arg1)\n}\n\n// GetDLQReplicationMessages mocks base method.\nfunc (m *MockHandler) GetDLQReplicationMessages(arg0 context.Context, arg1 *types.GetDLQReplicationMessagesRequest) (*types.GetDLQReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDLQReplicationMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetDLQReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetDLQReplicationMessages indicates an expected call of GetDLQReplicationMessages.\nfunc (mr *MockHandlerMockRecorder) GetDLQReplicationMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDLQReplicationMessages\", reflect.TypeOf((*MockHandler)(nil).GetDLQReplicationMessages), arg0, arg1)\n}\n\n// GetFailoverInfo mocks base method.\nfunc (m *MockHandler) GetFailoverInfo(arg0 context.Context, arg1 *types.GetFailoverInfoRequest) (*types.GetFailoverInfoResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFailoverInfo\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetFailoverInfoResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetFailoverInfo indicates an expected call of GetFailoverInfo.\nfunc (mr *MockHandlerMockRecorder) GetFailoverInfo(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFailoverInfo\", reflect.TypeOf((*MockHandler)(nil).GetFailoverInfo), arg0, arg1)\n}\n\n// GetMutableState mocks base method.\nfunc (m *MockHandler) GetMutableState(arg0 context.Context, arg1 *types.GetMutableStateRequest) (*types.GetMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMutableState\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetMutableState indicates an expected call of GetMutableState.\nfunc (mr *MockHandlerMockRecorder) GetMutableState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMutableState\", reflect.TypeOf((*MockHandler)(nil).GetMutableState), arg0, arg1)\n}\n\n// GetReplicationMessages mocks base method.\nfunc (m *MockHandler) GetReplicationMessages(arg0 context.Context, arg1 *types.GetReplicationMessagesRequest) (*types.GetReplicationMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetReplicationMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetReplicationMessages indicates an expected call of GetReplicationMessages.\nfunc (mr *MockHandlerMockRecorder) GetReplicationMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationMessages\", reflect.TypeOf((*MockHandler)(nil).GetReplicationMessages), arg0, arg1)\n}\n\n// Health mocks base method.\nfunc (m *MockHandler) Health(arg0 context.Context) (*types.HealthStatus, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Health\", arg0)\n\tret0, _ := ret[0].(*types.HealthStatus)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Health indicates an expected call of Health.\nfunc (mr *MockHandlerMockRecorder) Health(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Health\", reflect.TypeOf((*MockHandler)(nil).Health), arg0)\n}\n\n// MergeDLQMessages mocks base method.\nfunc (m *MockHandler) MergeDLQMessages(arg0 context.Context, arg1 *types.MergeDLQMessagesRequest) (*types.MergeDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MergeDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MergeDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MergeDLQMessages indicates an expected call of MergeDLQMessages.\nfunc (mr *MockHandlerMockRecorder) MergeDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeDLQMessages\", reflect.TypeOf((*MockHandler)(nil).MergeDLQMessages), arg0, arg1)\n}\n\n// NotifyFailoverMarkers mocks base method.\nfunc (m *MockHandler) NotifyFailoverMarkers(arg0 context.Context, arg1 *types.NotifyFailoverMarkersRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NotifyFailoverMarkers\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// NotifyFailoverMarkers indicates an expected call of NotifyFailoverMarkers.\nfunc (mr *MockHandlerMockRecorder) NotifyFailoverMarkers(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyFailoverMarkers\", reflect.TypeOf((*MockHandler)(nil).NotifyFailoverMarkers), arg0, arg1)\n}\n\n// PollMutableState mocks base method.\nfunc (m *MockHandler) PollMutableState(arg0 context.Context, arg1 *types.PollMutableStateRequest) (*types.PollMutableStateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollMutableState\", arg0, arg1)\n\tret0, _ := ret[0].(*types.PollMutableStateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollMutableState indicates an expected call of PollMutableState.\nfunc (mr *MockHandlerMockRecorder) PollMutableState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollMutableState\", reflect.TypeOf((*MockHandler)(nil).PollMutableState), arg0, arg1)\n}\n\n// PrepareToStop mocks base method.\nfunc (m *MockHandler) PrepareToStop(arg0 time.Duration) time.Duration {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PrepareToStop\", arg0)\n\tret0, _ := ret[0].(time.Duration)\n\treturn ret0\n}\n\n// PrepareToStop indicates an expected call of PrepareToStop.\nfunc (mr *MockHandlerMockRecorder) PrepareToStop(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PrepareToStop\", reflect.TypeOf((*MockHandler)(nil).PrepareToStop), arg0)\n}\n\n// PurgeDLQMessages mocks base method.\nfunc (m *MockHandler) PurgeDLQMessages(arg0 context.Context, arg1 *types.PurgeDLQMessagesRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PurgeDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// PurgeDLQMessages indicates an expected call of PurgeDLQMessages.\nfunc (mr *MockHandlerMockRecorder) PurgeDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PurgeDLQMessages\", reflect.TypeOf((*MockHandler)(nil).PurgeDLQMessages), arg0, arg1)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockHandler) QueryWorkflow(arg0 context.Context, arg1 *types.HistoryQueryWorkflowRequest) (*types.HistoryQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockHandlerMockRecorder) QueryWorkflow(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockHandler)(nil).QueryWorkflow), arg0, arg1)\n}\n\n// RatelimitUpdate mocks base method.\nfunc (m *MockHandler) RatelimitUpdate(arg0 context.Context, arg1 *types.RatelimitUpdateRequest) (*types.RatelimitUpdateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RatelimitUpdate\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RatelimitUpdateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RatelimitUpdate indicates an expected call of RatelimitUpdate.\nfunc (mr *MockHandlerMockRecorder) RatelimitUpdate(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RatelimitUpdate\", reflect.TypeOf((*MockHandler)(nil).RatelimitUpdate), arg0, arg1)\n}\n\n// ReadDLQMessages mocks base method.\nfunc (m *MockHandler) ReadDLQMessages(arg0 context.Context, arg1 *types.ReadDLQMessagesRequest) (*types.ReadDLQMessagesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadDLQMessages\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ReadDLQMessagesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadDLQMessages indicates an expected call of ReadDLQMessages.\nfunc (mr *MockHandlerMockRecorder) ReadDLQMessages(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadDLQMessages\", reflect.TypeOf((*MockHandler)(nil).ReadDLQMessages), arg0, arg1)\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockHandler) ReapplyEvents(arg0 context.Context, arg1 *types.HistoryReapplyEventsRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockHandlerMockRecorder) ReapplyEvents(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockHandler)(nil).ReapplyEvents), arg0, arg1)\n}\n\n// RecordActivityTaskHeartbeat mocks base method.\nfunc (m *MockHandler) RecordActivityTaskHeartbeat(arg0 context.Context, arg1 *types.HistoryRecordActivityTaskHeartbeatRequest) (*types.RecordActivityTaskHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskHeartbeat\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RecordActivityTaskHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskHeartbeat indicates an expected call of RecordActivityTaskHeartbeat.\nfunc (mr *MockHandlerMockRecorder) RecordActivityTaskHeartbeat(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskHeartbeat\", reflect.TypeOf((*MockHandler)(nil).RecordActivityTaskHeartbeat), arg0, arg1)\n}\n\n// RecordActivityTaskStarted mocks base method.\nfunc (m *MockHandler) RecordActivityTaskStarted(arg0 context.Context, arg1 *types.RecordActivityTaskStartedRequest) (*types.RecordActivityTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordActivityTaskStarted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RecordActivityTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordActivityTaskStarted indicates an expected call of RecordActivityTaskStarted.\nfunc (mr *MockHandlerMockRecorder) RecordActivityTaskStarted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordActivityTaskStarted\", reflect.TypeOf((*MockHandler)(nil).RecordActivityTaskStarted), arg0, arg1)\n}\n\n// RecordChildExecutionCompleted mocks base method.\nfunc (m *MockHandler) RecordChildExecutionCompleted(arg0 context.Context, arg1 *types.RecordChildExecutionCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordChildExecutionCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordChildExecutionCompleted indicates an expected call of RecordChildExecutionCompleted.\nfunc (mr *MockHandlerMockRecorder) RecordChildExecutionCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordChildExecutionCompleted\", reflect.TypeOf((*MockHandler)(nil).RecordChildExecutionCompleted), arg0, arg1)\n}\n\n// RecordDecisionTaskStarted mocks base method.\nfunc (m *MockHandler) RecordDecisionTaskStarted(arg0 context.Context, arg1 *types.RecordDecisionTaskStartedRequest) (*types.RecordDecisionTaskStartedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordDecisionTaskStarted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RecordDecisionTaskStartedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RecordDecisionTaskStarted indicates an expected call of RecordDecisionTaskStarted.\nfunc (mr *MockHandlerMockRecorder) RecordDecisionTaskStarted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordDecisionTaskStarted\", reflect.TypeOf((*MockHandler)(nil).RecordDecisionTaskStarted), arg0, arg1)\n}\n\n// RefreshWorkflowTasks mocks base method.\nfunc (m *MockHandler) RefreshWorkflowTasks(arg0 context.Context, arg1 *types.HistoryRefreshWorkflowTasksRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshWorkflowTasks\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshWorkflowTasks indicates an expected call of RefreshWorkflowTasks.\nfunc (mr *MockHandlerMockRecorder) RefreshWorkflowTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshWorkflowTasks\", reflect.TypeOf((*MockHandler)(nil).RefreshWorkflowTasks), arg0, arg1)\n}\n\n// RemoveSignalMutableState mocks base method.\nfunc (m *MockHandler) RemoveSignalMutableState(arg0 context.Context, arg1 *types.RemoveSignalMutableStateRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RemoveSignalMutableState\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveSignalMutableState indicates an expected call of RemoveSignalMutableState.\nfunc (mr *MockHandlerMockRecorder) RemoveSignalMutableState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveSignalMutableState\", reflect.TypeOf((*MockHandler)(nil).RemoveSignalMutableState), arg0, arg1)\n}\n\n// RemoveTask mocks base method.\nfunc (m *MockHandler) RemoveTask(arg0 context.Context, arg1 *types.RemoveTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RemoveTask\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveTask indicates an expected call of RemoveTask.\nfunc (mr *MockHandlerMockRecorder) RemoveTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveTask\", reflect.TypeOf((*MockHandler)(nil).RemoveTask), arg0, arg1)\n}\n\n// ReplicateEventsV2 mocks base method.\nfunc (m *MockHandler) ReplicateEventsV2(arg0 context.Context, arg1 *types.ReplicateEventsV2Request) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateEventsV2\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateEventsV2 indicates an expected call of ReplicateEventsV2.\nfunc (mr *MockHandlerMockRecorder) ReplicateEventsV2(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateEventsV2\", reflect.TypeOf((*MockHandler)(nil).ReplicateEventsV2), arg0, arg1)\n}\n\n// RequestCancelWorkflowExecution mocks base method.\nfunc (m *MockHandler) RequestCancelWorkflowExecution(arg0 context.Context, arg1 *types.HistoryRequestCancelWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestCancelWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestCancelWorkflowExecution indicates an expected call of RequestCancelWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) RequestCancelWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestCancelWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).RequestCancelWorkflowExecution), arg0, arg1)\n}\n\n// ResetQueue mocks base method.\nfunc (m *MockHandler) ResetQueue(arg0 context.Context, arg1 *types.ResetQueueRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetQueue\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetQueue indicates an expected call of ResetQueue.\nfunc (mr *MockHandlerMockRecorder) ResetQueue(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetQueue\", reflect.TypeOf((*MockHandler)(nil).ResetQueue), arg0, arg1)\n}\n\n// ResetStickyTaskList mocks base method.\nfunc (m *MockHandler) ResetStickyTaskList(arg0 context.Context, arg1 *types.HistoryResetStickyTaskListRequest) (*types.HistoryResetStickyTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetStickyTaskList\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryResetStickyTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetStickyTaskList indicates an expected call of ResetStickyTaskList.\nfunc (mr *MockHandlerMockRecorder) ResetStickyTaskList(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetStickyTaskList\", reflect.TypeOf((*MockHandler)(nil).ResetStickyTaskList), arg0, arg1)\n}\n\n// ResetWorkflowExecution mocks base method.\nfunc (m *MockHandler) ResetWorkflowExecution(arg0 context.Context, arg1 *types.HistoryResetWorkflowExecutionRequest) (*types.ResetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ResetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) ResetWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).ResetWorkflowExecution), arg0, arg1)\n}\n\n// RespondActivityTaskCanceled mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCanceled(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskCanceledRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCanceled\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCanceled indicates an expected call of RespondActivityTaskCanceled.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCanceled(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCanceled\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCanceled), arg0, arg1)\n}\n\n// RespondActivityTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondActivityTaskCompleted(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskCompleted indicates an expected call of RespondActivityTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskCompleted), arg0, arg1)\n}\n\n// RespondActivityTaskFailed mocks base method.\nfunc (m *MockHandler) RespondActivityTaskFailed(arg0 context.Context, arg1 *types.HistoryRespondActivityTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondActivityTaskFailed\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondActivityTaskFailed indicates an expected call of RespondActivityTaskFailed.\nfunc (mr *MockHandlerMockRecorder) RespondActivityTaskFailed(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondActivityTaskFailed\", reflect.TypeOf((*MockHandler)(nil).RespondActivityTaskFailed), arg0, arg1)\n}\n\n// RespondCrossClusterTasksCompleted mocks base method.\nfunc (m *MockHandler) RespondCrossClusterTasksCompleted(arg0 context.Context, arg1 *types.RespondCrossClusterTasksCompletedRequest) (*types.RespondCrossClusterTasksCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondCrossClusterTasksCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.RespondCrossClusterTasksCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondCrossClusterTasksCompleted indicates an expected call of RespondCrossClusterTasksCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondCrossClusterTasksCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondCrossClusterTasksCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondCrossClusterTasksCompleted), arg0, arg1)\n}\n\n// RespondDecisionTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondDecisionTaskCompleted(arg0 context.Context, arg1 *types.HistoryRespondDecisionTaskCompletedRequest) (*types.HistoryRespondDecisionTaskCompletedResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(*types.HistoryRespondDecisionTaskCompletedResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RespondDecisionTaskCompleted indicates an expected call of RespondDecisionTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondDecisionTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondDecisionTaskCompleted), arg0, arg1)\n}\n\n// RespondDecisionTaskFailed mocks base method.\nfunc (m *MockHandler) RespondDecisionTaskFailed(arg0 context.Context, arg1 *types.HistoryRespondDecisionTaskFailedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondDecisionTaskFailed\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondDecisionTaskFailed indicates an expected call of RespondDecisionTaskFailed.\nfunc (mr *MockHandlerMockRecorder) RespondDecisionTaskFailed(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondDecisionTaskFailed\", reflect.TypeOf((*MockHandler)(nil).RespondDecisionTaskFailed), arg0, arg1)\n}\n\n// ScheduleDecisionTask mocks base method.\nfunc (m *MockHandler) ScheduleDecisionTask(arg0 context.Context, arg1 *types.ScheduleDecisionTaskRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ScheduleDecisionTask\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ScheduleDecisionTask indicates an expected call of ScheduleDecisionTask.\nfunc (mr *MockHandlerMockRecorder) ScheduleDecisionTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ScheduleDecisionTask\", reflect.TypeOf((*MockHandler)(nil).ScheduleDecisionTask), arg0, arg1)\n}\n\n// SignalWithStartWorkflowExecution mocks base method.\nfunc (m *MockHandler) SignalWithStartWorkflowExecution(arg0 context.Context, arg1 *types.HistorySignalWithStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWithStartWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SignalWithStartWorkflowExecution indicates an expected call of SignalWithStartWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) SignalWithStartWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWithStartWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).SignalWithStartWorkflowExecution), arg0, arg1)\n}\n\n// SignalWorkflowExecution mocks base method.\nfunc (m *MockHandler) SignalWorkflowExecution(arg0 context.Context, arg1 *types.HistorySignalWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SignalWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SignalWorkflowExecution indicates an expected call of SignalWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) SignalWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SignalWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).SignalWorkflowExecution), arg0, arg1)\n}\n\n// Start mocks base method.\nfunc (m *MockHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockHandler)(nil).Start))\n}\n\n// StartWorkflowExecution mocks base method.\nfunc (m *MockHandler) StartWorkflowExecution(arg0 context.Context, arg1 *types.HistoryStartWorkflowExecutionRequest) (*types.StartWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"StartWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(*types.StartWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// StartWorkflowExecution indicates an expected call of StartWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) StartWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"StartWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).StartWorkflowExecution), arg0, arg1)\n}\n\n// Stop mocks base method.\nfunc (m *MockHandler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockHandlerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockHandler)(nil).Stop))\n}\n\n// SyncActivity mocks base method.\nfunc (m *MockHandler) SyncActivity(arg0 context.Context, arg1 *types.SyncActivityRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SyncActivity\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncActivity indicates an expected call of SyncActivity.\nfunc (mr *MockHandlerMockRecorder) SyncActivity(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncActivity\", reflect.TypeOf((*MockHandler)(nil).SyncActivity), arg0, arg1)\n}\n\n// SyncShardStatus mocks base method.\nfunc (m *MockHandler) SyncShardStatus(arg0 context.Context, arg1 *types.SyncShardStatusRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SyncShardStatus\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncShardStatus indicates an expected call of SyncShardStatus.\nfunc (mr *MockHandlerMockRecorder) SyncShardStatus(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncShardStatus\", reflect.TypeOf((*MockHandler)(nil).SyncShardStatus), arg0, arg1)\n}\n\n// TerminateWorkflowExecution mocks base method.\nfunc (m *MockHandler) TerminateWorkflowExecution(arg0 context.Context, arg1 *types.HistoryTerminateWorkflowExecutionRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TerminateWorkflowExecution\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// TerminateWorkflowExecution indicates an expected call of TerminateWorkflowExecution.\nfunc (mr *MockHandlerMockRecorder) TerminateWorkflowExecution(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TerminateWorkflowExecution\", reflect.TypeOf((*MockHandler)(nil).TerminateWorkflowExecution), arg0, arg1)\n}\n"
  },
  {
    "path": "service/history/lookup/lookup.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lookup\n\nimport (\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\n// HistoryServerByShardID calls resolver.Lookup with key based on provided shardID\nfunc HistoryServerByShardID(resolver membership.Resolver, shardID int) (membership.HostInfo, error) {\n\treturn resolver.Lookup(service.History, string(rune(shardID)))\n}\n"
  },
  {
    "path": "service/history/lookup/lookup_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage lookup\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestHistoryServerByShardID_Succeeds(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\n\tmockResolver.EXPECT().Lookup(service.History, string(rune(65))).\n\t\tReturn(membership.NewHostInfo(\"127.0.0.1:1234\"), nil)\n\n\thost, err := HistoryServerByShardID(mockResolver, 65)\n\trequire.NoError(t, err)\n\tassert.Equal(t, \"127.0.0.1:1234\", host.GetAddress())\n}\n\nfunc TestHistoryServerByShardID_PreservesError(t *testing.T) {\n\tlookupError := errors.New(\"lookup failed\")\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\n\tmockResolver.EXPECT().Lookup(service.History, gomock.Any()).\n\t\tReturn(membership.HostInfo{}, lookupError)\n\n\thost, err := HistoryServerByShardID(mockResolver, 65)\n\tassert.Equal(t, lookupError, err, \"error should not be modified\")\n\tassert.Empty(t, host)\n}\n"
  },
  {
    "path": "service/history/ndc/activity_replicator.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination activity_replicator_mock.go\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tresendMissingEventMessage  = \"Resend missed sync activity events\"\n\tresendHigherVersionMessage = \"Resend sync activity events due to a higher version received\"\n\terrRetrySyncActivityMsg    = \"retry on applying sync activity\"\n)\n\ntype (\n\t// ActivityReplicator handles sync activity process\n\tActivityReplicator interface {\n\t\tSyncActivity(\n\t\t\tctx ctx.Context,\n\t\t\trequest *types.SyncActivityRequest,\n\t\t) error\n\t}\n\n\tactivityReplicatorImpl struct {\n\t\texecutionCache  execution.Cache\n\t\tclusterMetadata cluster.Metadata\n\t\tlogger          log.Logger\n\t}\n)\n\nvar _ ActivityReplicator = (*activityReplicatorImpl)(nil)\n\n// NewActivityReplicator creates activity replicator\nfunc NewActivityReplicator(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\tlogger log.Logger,\n) ActivityReplicator {\n\n\treturn &activityReplicatorImpl{\n\t\texecutionCache:  executionCache,\n\t\tclusterMetadata: shard.GetService().GetClusterMetadata(),\n\t\tlogger:          logger.WithTags(tag.ComponentHistoryReplicator),\n\t}\n}\n\nfunc (r *activityReplicatorImpl) SyncActivity(\n\tctx ctx.Context,\n\trequest *types.SyncActivityRequest,\n) (retError error) {\n\n\t// sync activity info will only be sent from active side, when\n\t// 1. activity has retry policy and activity got started\n\t// 2. activity heart beat\n\t// no sync activity task will be sent when active side fail / timeout activity,\n\t// since standby side does not have activity retry timer\n\tdomainID := request.GetDomainID()\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: request.WorkflowID,\n\t\tRunID:      request.RunID,\n\t}\n\n\tcontext, release, err := r.executionCache.GetOrCreateWorkflowExecution(ctx, domainID, workflowExecution)\n\tif err != nil {\n\t\t// for get workflow execution context, with valid run id\n\t\t// err will not be of type EntityNotExistsError\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := context.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\n\t\t// this can happen if the workflow start event and this sync activity task are out of order\n\t\t// or the target workflow is long gone\n\t\t// the safe solution to this is to throw away the sync activity task\n\t\t// or otherwise, worker attempt will exceeds limit and put this message to DLQ\n\t\treturn nil\n\t}\n\n\tversion := request.GetVersion()\n\tscheduleID := request.GetScheduledID()\n\tshouldApply, err := r.shouldApplySyncActivity(\n\t\tdomainID,\n\t\tworkflowExecution.GetWorkflowID(),\n\t\tworkflowExecution.GetRunID(),\n\t\tscheduleID,\n\t\tversion,\n\t\tmutableState,\n\t\trequest.GetVersionHistory(),\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !shouldApply {\n\t\treturn nil\n\t}\n\n\tai, ok := mutableState.GetActivityInfo(scheduleID)\n\tif !ok {\n\t\t// this should not retry, can be caused by out of order delivery\n\t\t// since the activity is already finished\n\t\treturn nil\n\t}\n\n\tif ai.Version > request.GetVersion() {\n\t\t// this should not retry, can be caused by failover or reset\n\t\treturn nil\n\t}\n\n\tif ai.Version == request.GetVersion() {\n\t\tif ai.Attempt > request.GetAttempt() {\n\t\t\t// this should not retry, can be caused by failover or reset\n\t\t\treturn nil\n\t\t}\n\t\tif ai.Attempt == request.GetAttempt() {\n\t\t\tlastHeartbeatTime := time.Unix(0, request.GetLastHeartbeatTime())\n\t\t\tif ai.LastHeartBeatUpdatedTime.After(lastHeartbeatTime) {\n\t\t\t\t// this should not retry, can be caused by out of order delivery\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\t// version equal & attempt equal & last heartbeat after existing heartbeat\n\t\t\t// should update activity\n\t\t}\n\t\t// version equal & attempt larger then existing, should update activity\n\t}\n\t// version larger then existing, should update activity\n\n\t// calculate whether to reset the activity timer task status bits\n\t// reset timer task status bits if\n\t// 1. same source cluster & attempt changes\n\t// 2. different source cluster\n\tresetActivityTimerTaskStatus := false\n\tif !r.clusterMetadata.IsVersionFromSameCluster(request.GetVersion(), ai.Version) {\n\t\tresetActivityTimerTaskStatus = true\n\t} else if ai.Attempt < request.GetAttempt() {\n\t\tresetActivityTimerTaskStatus = true\n\t}\n\terr = mutableState.ReplicateActivityInfo(request, resetActivityTimerTaskStatus)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// see whether we need to refresh the activity timer\n\teventTime := request.GetScheduledTime()\n\tif eventTime < request.GetStartedTime() {\n\t\teventTime = request.GetStartedTime()\n\t}\n\tif eventTime < request.GetLastHeartbeatTime() {\n\t\teventTime = request.GetLastHeartbeatTime()\n\t}\n\n\t// passive logic need to explicitly call create timer\n\tnow := time.Unix(0, eventTime)\n\tif _, err := execution.NewTimerSequence(\n\t\tmutableState,\n\t).CreateNextActivityTimer(); err != nil {\n\t\treturn err\n\t}\n\n\tupdateMode := persistence.UpdateWorkflowModeUpdateCurrent\n\tif state, _ := mutableState.GetWorkflowStateCloseStatus(); state == persistence.WorkflowStateZombie {\n\t\tupdateMode = persistence.UpdateWorkflowModeBypassCurrent\n\t}\n\n\tr.logger.Debugf(\"SyncActivity calling UpdateWorkflowExecutionWithNew for wfID %s, updateMode %v, current policy %v, new policy %v\",\n\t\tworkflowExecution.GetWorkflowID(),\n\t\tupdateMode,\n\t\texecution.TransactionPolicyPassive,\n\t\tnil,\n\t)\n\treturn context.UpdateWorkflowExecutionWithNew(\n\t\tctx,\n\t\tnow,\n\t\tupdateMode,\n\t\tnil, // no new workflow\n\t\tnil, // no new workflow\n\t\texecution.TransactionPolicyPassive,\n\t\tnil,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n}\n\nfunc (r *activityReplicatorImpl) shouldApplySyncActivity(\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tscheduleID int64,\n\tactivityVersion int64,\n\tmutableState execution.MutableState,\n\tincomingRawVersionHistory *types.VersionHistory,\n) (bool, error) {\n\n\tif mutableState.GetVersionHistories() != nil {\n\t\tif state, _ := mutableState.GetWorkflowStateCloseStatus(); state == persistence.WorkflowStateCompleted {\n\t\t\treturn false, nil\n\t\t}\n\n\t\tcurrentVersionHistory, err := mutableState.GetVersionHistories().GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn false, err\n\t\t}\n\n\t\tlastLocalItem, err := currentVersionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn false, err\n\t\t}\n\n\t\tincomingVersionHistory := persistence.NewVersionHistoryFromInternalType(incomingRawVersionHistory)\n\t\tlastIncomingItem, err := incomingVersionHistory.GetLastItem()\n\t\tif err != nil {\n\t\t\treturn false, err\n\t\t}\n\n\t\tlcaItem, err := currentVersionHistory.FindLCAItem(incomingVersionHistory)\n\t\tif err != nil {\n\t\t\treturn false, err\n\t\t}\n\n\t\t// case 1: local version history is superset of incoming version history\n\t\t// or incoming version history is superset of local version history\n\t\t// resend the missing event if local version history doesn't have the schedule event\n\n\t\t// case 2: local version history and incoming version history diverged\n\t\t// case 2-1: local version history has the higher version and discard the incoming event\n\t\t// case 2-2: incoming version history has the higher version and resend the missing incoming events\n\t\tif currentVersionHistory.IsLCAAppendable(lcaItem) || incomingVersionHistory.IsLCAAppendable(lcaItem) {\n\t\t\t// case 1\n\t\t\tif scheduleID > lcaItem.EventID {\n\t\t\t\treturn false, newNDCRetryTaskErrorWithHint(\n\t\t\t\t\tresendMissingEventMessage,\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t\tcommon.Int64Ptr(lcaItem.EventID),\n\t\t\t\t\tcommon.Int64Ptr(lcaItem.Version),\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t)\n\t\t\t}\n\t\t} else {\n\t\t\t// case 2\n\t\t\tif lastIncomingItem.Version < lastLocalItem.Version {\n\t\t\t\t// case 2-1\n\t\t\t\treturn false, nil\n\t\t\t} else if lastIncomingItem.Version > lastLocalItem.Version {\n\t\t\t\t// case 2-2\n\t\t\t\treturn false, newNDCRetryTaskErrorWithHint(\n\t\t\t\t\tresendHigherVersionMessage,\n\t\t\t\t\tdomainID,\n\t\t\t\t\tworkflowID,\n\t\t\t\t\trunID,\n\t\t\t\t\tcommon.Int64Ptr(lcaItem.EventID),\n\t\t\t\t\tcommon.Int64Ptr(lcaItem.Version),\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\t} else {\n\t\treturn false, &types.InternalServiceError{Message: \"The workflow version histories is corrupted.\"}\n\t}\n\n\treturn true, nil\n}\n"
  },
  {
    "path": "service/history/ndc/activity_replicator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: activity_replicator.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source activity_replicator.go -destination activity_replicator_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockActivityReplicator is a mock of ActivityReplicator interface.\ntype MockActivityReplicator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockActivityReplicatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockActivityReplicatorMockRecorder is the mock recorder for MockActivityReplicator.\ntype MockActivityReplicatorMockRecorder struct {\n\tmock *MockActivityReplicator\n}\n\n// NewMockActivityReplicator creates a new mock instance.\nfunc NewMockActivityReplicator(ctrl *gomock.Controller) *MockActivityReplicator {\n\tmock := &MockActivityReplicator{ctrl: ctrl}\n\tmock.recorder = &MockActivityReplicatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockActivityReplicator) EXPECT() *MockActivityReplicatorMockRecorder {\n\treturn m.recorder\n}\n\n// SyncActivity mocks base method.\nfunc (m *MockActivityReplicator) SyncActivity(ctx context.Context, request *types.SyncActivityRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SyncActivity\", ctx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SyncActivity indicates an expected call of SyncActivity.\nfunc (mr *MockActivityReplicatorMockRecorder) SyncActivity(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SyncActivity\", reflect.TypeOf((*MockActivityReplicator)(nil).SyncActivity), ctx, request)\n}\n"
  },
  {
    "path": "service/history/ndc/activity_replicator_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tactivityReplicatorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockShard        *shard.TestContext\n\t\tmockEngine       *engine.MockEngine\n\t\tmockDomainCache  *cache.MockDomainCache\n\t\tmockMutableState *execution.MockMutableState\n\n\t\tmockExecutionMgr *mocks.ExecutionManager\n\n\t\tlogger         log.Logger\n\t\texecutionCache execution.Cache\n\n\t\tactivityReplicator ActivityReplicator\n\t}\n)\n\nfunc TestActivityReplicatorSuite(t *testing.T) {\n\ts := new(activityReplicatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *activityReplicatorSuite) SetupSuite() {\n\n}\n\nfunc (s *activityReplicatorSuite) TearDownSuite() {\n\n}\n\nfunc (s *activityReplicatorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockMutableState = execution.NewMockMutableState(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.executionCache = execution.NewCache(s.mockShard)\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.mockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTransferTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTimerTasks(gomock.Any()).AnyTimes()\n\ts.mockShard.SetEngine(s.mockEngine)\n\n\ts.activityReplicator = NewActivityReplicator(\n\t\ts.mockShard,\n\t\ts.executionCache,\n\t\ts.logger,\n\t)\n}\n\nfunc (s *activityReplicatorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_WorkflowNotFound() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: domainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tDomainName: domainName,\n\t\tRangeID:    1,\n\t}).Return(nil, &types.EntityNotExistsError{})\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr := s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_WorkflowClosed() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tversionHistories := &persistence.VersionHistories{}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateCompleted, persistence.WorkflowCloseStatusCompleted)\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_IncomingScheduleIDLarger_IncomingVersionSmaller() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(100)\n\tlastWriteVersion := version + 100\n\tnextEventID := scheduleID - 10\n\tversionHistoryItem0 := persistence.NewVersionHistoryItem(1, 1)\n\tversionHistoryItem1 := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem0,\n\t\tversionHistoryItem1,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\n\tversionHistoryItem2 := persistence.NewVersionHistoryItem(scheduleID+1, version-1)\n\tversionHistory2 := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem0,\n\t\tversionHistoryItem2,\n\t})\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: versionHistory2.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_IncomingScheduleIDLarger_IncomingVersionLarger() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(100)\n\tlastWriteVersion := version - 100\n\tnextEventID := scheduleID - 10\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:    domainID,\n\t\tWorkflowID:  workflowID,\n\t\tRunID:       runID,\n\t\tVersion:     version,\n\t\tScheduledID: scheduleID,\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastWriteVersion, nil).AnyTimes()\n\tvar versionHistories *persistence.VersionHistories\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\t_ = s.activityReplicator.SyncActivity(ctx.Background(), request)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_VersionHistories_IncomingVersionSmaller_DiscardTask() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(99)\n\n\tlastWriteVersion := version - 100\n\tincomingVersionHistory := persistence.VersionHistory{\n\t\tBranchToken: []byte{},\n\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: scheduleID - 1,\n\t\t\t\tVersion: version - 1,\n\t\t\t},\n\t\t\t{\n\t\t\t\tEventID: scheduleID,\n\t\t\t\tVersion: version,\n\t\t\t},\n\t\t},\n\t}\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: incomingVersionHistory.ToInternalType(),\n\t}\n\tlocalVersionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{},\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID - 1,\n\t\t\t\t\t\tVersion: version - 1,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID + 1,\n\t\t\t\t\t\tVersion: version + 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(localVersionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateRunning, persistence.WorkflowCloseStatusNone)\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_DifferentVersionHistories_IncomingVersionLarger_ReturnRetryError() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(100)\n\tlastWriteVersion := version - 100\n\n\tincomingVersionHistory := persistence.VersionHistory{\n\t\tBranchToken: []byte{},\n\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: 50,\n\t\t\t\tVersion: 2,\n\t\t\t},\n\t\t\t{\n\t\t\t\tEventID: scheduleID,\n\t\t\t\tVersion: version,\n\t\t\t},\n\t\t},\n\t}\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: incomingVersionHistory.ToInternalType(),\n\t}\n\tlocalVersionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{},\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: 100,\n\t\t\t\t\t\tVersion: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(localVersionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateRunning, persistence.WorkflowCloseStatusNone)\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Equal(newNDCRetryTaskErrorWithHint(\n\t\tresendHigherVersionMessage,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tcommon.Int64Ptr(50),\n\t\tcommon.Int64Ptr(2),\n\t\tnil,\n\t\tnil,\n\t),\n\t\terr,\n\t)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_VersionHistories_IncomingScheduleIDLarger_ReturnRetryError() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(99)\n\tversion := int64(100)\n\n\tlastWriteVersion := version - 100\n\tincomingVersionHistory := persistence.VersionHistory{\n\t\tBranchToken: []byte{},\n\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: 50,\n\t\t\t\tVersion: 2,\n\t\t\t},\n\t\t\t{\n\t\t\t\tEventID: scheduleID,\n\t\t\t\tVersion: version,\n\t\t\t},\n\t\t\t{\n\t\t\t\tEventID: scheduleID + 100,\n\t\t\t\tVersion: version + 100,\n\t\t\t},\n\t\t},\n\t}\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: incomingVersionHistory.ToInternalType(),\n\t}\n\tlocalVersionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{},\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID - 10,\n\t\t\t\t\t\tVersion: version,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(localVersionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(persistence.WorkflowStateRunning, persistence.WorkflowCloseStatusNone)\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Equal(newNDCRetryTaskErrorWithHint(\n\t\tresendMissingEventMessage,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tcommon.Int64Ptr(scheduleID-10),\n\t\tcommon.Int64Ptr(version),\n\t\tnil,\n\t\tnil,\n\t),\n\t\terr,\n\t)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_VersionHistories_SameScheduleID() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(99)\n\tversion := int64(100)\n\n\tlastWriteVersion := version - 100\n\tincomingVersionHistory := persistence.VersionHistory{\n\t\tBranchToken: []byte{},\n\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: 50,\n\t\t\t\tVersion: 2,\n\t\t\t},\n\t\t\t{\n\t\t\t\tEventID: scheduleID,\n\t\t\t\tVersion: version,\n\t\t\t},\n\t\t},\n\t}\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: incomingVersionHistory.ToInternalType(),\n\t}\n\tlocalVersionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{},\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID,\n\t\t\t\t\t\tVersion: version,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(localVersionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(nil, false).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().\n\t\tReturn(persistence.WorkflowStateCreated, persistence.WorkflowCloseStatusNone).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_VersionHistories_LocalVersionHistoryWin() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(99)\n\tversion := int64(100)\n\n\tlastWriteVersion := version - 100\n\tincomingVersionHistory := persistence.VersionHistory{\n\t\tBranchToken: []byte{},\n\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t{\n\t\t\t\tEventID: scheduleID,\n\t\t\t\tVersion: version,\n\t\t\t},\n\t\t},\n\t}\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: incomingVersionHistory.ToInternalType(),\n\t}\n\tlocalVersionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: []byte{},\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID,\n\t\t\t\t\t\tVersion: version,\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tEventID: scheduleID + 1,\n\t\t\t\t\t\tVersion: version + 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(localVersionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(nil, false).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().\n\t\tReturn(persistence.WorkflowStateCreated, persistence.WorkflowCloseStatusNone).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityCompleted() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(100)\n\tlastWriteVersion := version\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(nil, false).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning_LocalActivityVersionLarger() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tscheduleID := int64(144)\n\tversion := int64(100)\n\tlastWriteVersion := version + 10\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().AnyTimes()\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:       domainID,\n\t\tWorkflowID:     workflowID,\n\t\tRunID:          runID,\n\t\tVersion:        version,\n\t\tScheduledID:    scheduleID,\n\t\tVersionHistory: versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tlastWriteVersion,\n\t\t), nil,\n\t).AnyTimes()\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(&persistence.ActivityInfo{\n\t\tVersion: lastWriteVersion - 1,\n\t}, true).AnyTimes()\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Nil(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning_Update_SameVersionSameAttempt() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\tscheduleID := int64(144)\n\tscheduledTime := time.Now()\n\tstartedID := scheduleID + 1\n\tstartedTime := scheduledTime.Add(time.Minute)\n\theartBeatUpdatedTime := startedTime.Add(time.Minute)\n\tattempt := int32(0)\n\tdetails := []byte(\"some random activity heartbeat progress\")\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().Times(1)\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tVersion:           version,\n\t\tScheduledID:       scheduleID,\n\t\tScheduledTime:     common.Int64Ptr(scheduledTime.UnixNano()),\n\t\tStartedID:         startedID,\n\t\tStartedTime:       common.Int64Ptr(startedTime.UnixNano()),\n\t\tAttempt:           attempt,\n\t\tLastHeartbeatTime: common.Int64Ptr(heartBeatUpdatedTime.UnixNano()),\n\t\tDetails:           details,\n\t\tVersionHistory:    versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:    version,\n\t\tScheduleID: scheduleID,\n\t\tAttempt:    attempt,\n\t}\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(activityInfo, true).AnyTimes()\n\n\texpectedErr := errors.New(\"this is error is used to by pass lots of mocking\")\n\ts.mockMutableState.EXPECT().ReplicateActivityInfo(request, false).Return(expectedErr).Times(1)\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning_Update_SameVersionLargerAttempt() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\tscheduleID := int64(144)\n\tscheduledTime := time.Now()\n\tstartedID := scheduleID + 1\n\tstartedTime := scheduledTime.Add(time.Minute)\n\theartBeatUpdatedTime := startedTime.Add(time.Minute)\n\tattempt := int32(100)\n\tdetails := []byte(\"some random activity heartbeat progress\")\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().Times(1)\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tVersion:           version,\n\t\tScheduledID:       scheduleID,\n\t\tScheduledTime:     common.Int64Ptr(scheduledTime.UnixNano()),\n\t\tStartedID:         startedID,\n\t\tStartedTime:       common.Int64Ptr(startedTime.UnixNano()),\n\t\tAttempt:           attempt,\n\t\tLastHeartbeatTime: common.Int64Ptr(heartBeatUpdatedTime.UnixNano()),\n\t\tDetails:           details,\n\t\tVersionHistory:    versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:    version,\n\t\tScheduleID: scheduleID,\n\t\tAttempt:    attempt - 1,\n\t}\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(activityInfo, true).AnyTimes()\n\n\texpectedErr := errors.New(\"this is error is used to by pass lots of mocking\")\n\ts.mockMutableState.EXPECT().ReplicateActivityInfo(request, true).Return(expectedErr).Times(1)\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning_Update_LargerVersion() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\tscheduleID := int64(144)\n\tscheduledTime := time.Now()\n\tstartedID := scheduleID + 1\n\tstartedTime := scheduledTime.Add(time.Minute)\n\theartBeatUpdatedTime := startedTime.Add(time.Minute)\n\tattempt := int32(100)\n\tdetails := []byte(\"some random activity heartbeat progress\")\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().Clear().Times(1)\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tVersion:           version,\n\t\tScheduledID:       scheduleID,\n\t\tScheduledTime:     common.Int64Ptr(scheduledTime.UnixNano()),\n\t\tStartedID:         startedID,\n\t\tStartedTime:       common.Int64Ptr(startedTime.UnixNano()),\n\t\tAttempt:           attempt,\n\t\tLastHeartbeatTime: common.Int64Ptr(heartBeatUpdatedTime.UnixNano()),\n\t\tDetails:           details,\n\t\tVersionHistory:    versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:    version - 1,\n\t\tScheduleID: scheduleID,\n\t\tAttempt:    attempt + 1,\n\t}\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(activityInfo, true).AnyTimes()\n\n\texpectedErr := errors.New(\"this is error is used to by pass lots of mocking\")\n\ts.mockMutableState.EXPECT().ReplicateActivityInfo(request, true).Return(expectedErr).Times(1)\n\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\tscheduleID := int64(144)\n\tscheduledTime := time.Now()\n\tstartedID := scheduleID + 1\n\tstartedTime := scheduledTime.Add(time.Minute)\n\theartBeatUpdatedTime := startedTime.Add(time.Minute)\n\tattempt := int32(100)\n\tdetails := []byte(\"some random activity heartbeat progress\")\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tVersion:           version,\n\t\tScheduledID:       scheduleID,\n\t\tScheduledTime:     common.Int64Ptr(scheduledTime.UnixNano()),\n\t\tStartedID:         startedID,\n\t\tStartedTime:       common.Int64Ptr(startedTime.UnixNano()),\n\t\tAttempt:           attempt,\n\t\tLastHeartbeatTime: common.Int64Ptr(heartBeatUpdatedTime.UnixNano()),\n\t\tDetails:           details,\n\t\tVersionHistory:    versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(1, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:    version - 1,\n\t\tScheduleID: scheduleID,\n\t\tAttempt:    attempt + 1,\n\t}\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(activityInfo, true).AnyTimes()\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).AnyTimes()\n\n\ts.mockMutableState.EXPECT().ReplicateActivityInfo(request, true).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().UpdateActivity(activityInfo).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(int64(1)).Times(1)\n\ts.mockMutableState.EXPECT().AddTimerTasks(gomock.Any()).Times(1)\n\tnow := time.Unix(0, request.GetLastHeartbeatTime())\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeUpdateCurrent,\n\t\tnil,\n\t\tnil,\n\t\texecution.TransactionPolicyPassive,\n\t\tnil,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.NoError(err)\n}\n\nfunc (s *activityReplicatorSuite) TestSyncActivity_ActivityRunning_ZombieWorkflow() {\n\tdomainName := \"some random domain name\"\n\tdomainID := constants.TestDomainID\n\tworkflowID := \"some random workflow ID\"\n\trunID := uuid.New()\n\tversion := int64(100)\n\tscheduleID := int64(144)\n\tscheduledTime := time.Now()\n\tstartedID := scheduleID + 1\n\tstartedTime := scheduledTime.Add(time.Minute)\n\theartBeatUpdatedTime := startedTime.Add(time.Minute)\n\tattempt := int32(100)\n\tdetails := []byte(\"some random activity heartbeat progress\")\n\tnextEventID := scheduleID + 10\n\tversionHistoryItem := persistence.NewVersionHistoryItem(scheduleID, version)\n\tversionHistory := persistence.NewVersionHistory([]byte{}, []*persistence.VersionHistoryItem{\n\t\tversionHistoryItem,\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tkey := definition.NewWorkflowIdentifier(domainID, workflowID, runID)\n\tcontext := execution.NewMockContext(s.controller)\n\tcontext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(s.mockMutableState, nil).Times(1)\n\tcontext.EXPECT().Lock(gomock.Any()).Return(nil)\n\tcontext.EXPECT().Unlock().Times(1)\n\tcontext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\t_, err := s.executionCache.PutIfNotExist(key, context)\n\ts.NoError(err)\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tVersion:           version,\n\t\tScheduledID:       scheduleID,\n\t\tScheduledTime:     common.Int64Ptr(scheduledTime.UnixNano()),\n\t\tStartedID:         startedID,\n\t\tStartedTime:       common.Int64Ptr(startedTime.UnixNano()),\n\t\tAttempt:           attempt,\n\t\tLastHeartbeatTime: common.Int64Ptr(heartBeatUpdatedTime.UnixNano()),\n\t\tDetails:           details,\n\t\tVersionHistory:    versionHistory.ToInternalType(),\n\t}\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetWorkflowStateCloseStatus().Return(3, 0).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(domainID).Return(\n\t\tcache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{ID: domainID, Name: domainName},\n\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t\t},\n\t\t\t},\n\t\t\tversion,\n\t\t), nil,\n\t).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t})\n\tactivityInfo := &persistence.ActivityInfo{\n\t\tVersion:    version - 1,\n\t\tScheduleID: scheduleID,\n\t\tAttempt:    attempt + 1,\n\t}\n\ts.mockMutableState.EXPECT().GetActivityInfo(scheduleID).Return(activityInfo, true).AnyTimes()\n\tactivityInfos := map[int64]*persistence.ActivityInfo{activityInfo.ScheduleID: activityInfo}\n\ts.mockMutableState.EXPECT().GetPendingActivityInfos().Return(activityInfos).AnyTimes()\n\n\ts.mockMutableState.EXPECT().ReplicateActivityInfo(request, true).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().UpdateActivity(activityInfo).Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().GetCurrentVersion().Return(int64(1)).Times(1)\n\ts.mockMutableState.EXPECT().AddTimerTasks(gomock.Any()).Times(1)\n\tnow := time.Unix(0, request.GetLastHeartbeatTime())\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeBypassCurrent,\n\t\tnil,\n\t\tnil,\n\t\texecution.TransactionPolicyPassive,\n\t\tnil,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr = s.activityReplicator.SyncActivity(ctx.Background(), request)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "service/history/ndc/branch_manager.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination branch_manager_mock.go\n\npackage ndc\n\nimport (\n\t\"context\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\toutOfOrderDeliveryMessage = \"Resend events due to out of order delivery\"\n)\n\ntype (\n\tbranchManager interface {\n\t\tprepareVersionHistory(\n\t\t\tctx context.Context,\n\t\t\tincomingVersionHistory *persistence.VersionHistory,\n\t\t\tincomingFirstEventID int64,\n\t\t\tincomingFirstEventVersion int64,\n\t\t) (bool, int, error)\n\t}\n\n\tbranchManagerImpl struct {\n\t\tshard           shard.Context\n\t\tclusterMetadata cluster.Metadata\n\t\thistoryV2Mgr    persistence.HistoryManager\n\n\t\tcontext      execution.Context\n\t\tmutableState execution.MutableState\n\t\tlogger       log.Logger\n\t}\n)\n\nvar _ branchManager = (*branchManagerImpl)(nil)\n\nfunc newBranchManager(\n\tshard shard.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tlogger log.Logger,\n) branchManager {\n\n\treturn &branchManagerImpl{\n\t\tshard:           shard,\n\t\tclusterMetadata: shard.GetService().GetClusterMetadata(),\n\t\thistoryV2Mgr:    shard.GetHistoryManager(),\n\n\t\tcontext:      context,\n\t\tmutableState: mutableState,\n\t\tlogger:       logger,\n\t}\n}\n\nfunc (r *branchManagerImpl) prepareVersionHistory(\n\tctx context.Context,\n\tincomingVersionHistory *persistence.VersionHistory,\n\tincomingFirstEventID int64,\n\tincomingFirstEventVersion int64,\n) (bool, int, error) {\n\n\tversionHistoryIndex, lcaVersionHistoryItem, err := r.flushBufferedEvents(ctx, incomingVersionHistory)\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\tlocalVersionHistories := r.mutableState.GetVersionHistories()\n\tif localVersionHistories == nil {\n\t\treturn false, 0, execution.ErrMissingVersionHistories\n\t}\n\tversionHistory, err := localVersionHistories.GetVersionHistory(versionHistoryIndex)\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\t// if can directly append to a branch\n\tif versionHistory.IsLCAAppendable(lcaVersionHistoryItem) {\n\t\tdoContinue, err := r.verifyEventsOrder(\n\t\t\tctx,\n\t\t\tversionHistory,\n\t\t\tincomingFirstEventID,\n\t\t\tincomingFirstEventVersion,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn false, 0, err\n\t\t}\n\t\treturn doContinue, versionHistoryIndex, nil\n\t}\n\n\tnewVersionHistory, err := versionHistory.DuplicateUntilLCAItem(lcaVersionHistoryItem)\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\t// if cannot directly append to the new branch to be created\n\tdoContinue, err := r.verifyEventsOrder(\n\t\tctx,\n\t\tnewVersionHistory,\n\t\tincomingFirstEventID,\n\t\tincomingFirstEventVersion,\n\t)\n\tif err != nil || !doContinue {\n\t\treturn false, 0, err\n\t}\n\n\tnewVersionHistoryIndex, err := r.createNewBranch(\n\t\tctx,\n\t\tversionHistory.GetBranchToken(),\n\t\tlcaVersionHistoryItem.EventID,\n\t\tnewVersionHistory,\n\t)\n\tif err != nil {\n\t\treturn false, 0, err\n\t}\n\n\treturn true, newVersionHistoryIndex, nil\n}\n\nfunc (r *branchManagerImpl) flushBufferedEvents(\n\tctx context.Context,\n\tincomingVersionHistory *persistence.VersionHistory,\n) (int, *persistence.VersionHistoryItem, error) {\n\n\tlocalVersionHistories := r.mutableState.GetVersionHistories()\n\tif localVersionHistories == nil {\n\t\treturn 0, nil, execution.ErrMissingVersionHistories\n\t}\n\tversionHistoryIndex, lcaVersionHistoryItem, err := localVersionHistories.FindLCAVersionHistoryIndexAndItem(\n\t\tincomingVersionHistory,\n\t)\n\tif err != nil {\n\t\treturn 0, nil, err\n\t}\n\n\t// check whether there are buffered events, if so, flush it\n\t// NOTE: buffered events does not show in version history or next event id\n\tif !r.mutableState.HasBufferedEvents() {\n\t\treturn versionHistoryIndex, lcaVersionHistoryItem, nil\n\t}\n\n\ttargetWorkflow := execution.NewWorkflow(\n\t\tctx,\n\t\tr.clusterMetadata,\n\t\tr.context,\n\t\tr.mutableState,\n\t\texecution.NoopReleaseFn,\n\t\tr.logger,\n\t)\n\tif err := targetWorkflow.FlushBufferedEvents(); err != nil {\n\t\treturn 0, nil, err\n\t}\n\t// the workflow must be updated as active, to send out replication tasks\n\tr.logger.Debug(\"flushBufferedEvents calling UpdateWorkflowExecutionAsActive\", tag.WorkflowID(r.mutableState.GetExecutionInfo().WorkflowID))\n\tif err := targetWorkflow.GetContext().UpdateWorkflowExecutionAsActive(\n\t\tctx,\n\t\tr.shard.GetTimeSource().Now(),\n\t); err != nil {\n\t\treturn 0, nil, err\n\t}\n\n\tr.context = targetWorkflow.GetContext()\n\tr.mutableState = targetWorkflow.GetMutableState()\n\n\tlocalVersionHistories = r.mutableState.GetVersionHistories()\n\treturn localVersionHistories.FindLCAVersionHistoryIndexAndItem(incomingVersionHistory)\n}\n\nfunc (r *branchManagerImpl) verifyEventsOrder(\n\tctx context.Context,\n\tlocalVersionHistory *persistence.VersionHistory,\n\tincomingFirstEventID int64,\n\tincomingFirstEventVersion int64,\n) (bool, error) {\n\n\tlastVersionHistoryItem, err := localVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tnextEventID := lastVersionHistoryItem.EventID + 1\n\n\tif incomingFirstEventID < nextEventID {\n\t\t// duplicate replication task\n\t\treturn false, nil\n\t}\n\tif incomingFirstEventID > nextEventID {\n\t\texecutionInfo := r.mutableState.GetExecutionInfo()\n\t\treturn false, newNDCRetryTaskErrorWithHint(\n\t\t\toutOfOrderDeliveryMessage,\n\t\t\texecutionInfo.DomainID,\n\t\t\texecutionInfo.WorkflowID,\n\t\t\texecutionInfo.RunID,\n\t\t\tcommon.Int64Ptr(lastVersionHistoryItem.EventID),\n\t\t\tcommon.Int64Ptr(lastVersionHistoryItem.Version),\n\t\t\tcommon.Int64Ptr(incomingFirstEventID),\n\t\t\tcommon.Int64Ptr(incomingFirstEventVersion))\n\t}\n\t// task.getFirstEvent().GetEventId() == nextEventID\n\treturn true, nil\n}\n\nfunc (r *branchManagerImpl) createNewBranch(\n\tctx context.Context,\n\tbaseBranchToken []byte,\n\tbaseBranchLastEventID int64,\n\tnewVersionHistory *persistence.VersionHistory,\n) (newVersionHistoryIndex int, retError error) {\n\n\tshardID := r.shard.GetShardID()\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\tworkflowID := executionInfo.WorkflowID\n\tdomainName, err := r.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\tresp, err := r.historyV2Mgr.ForkHistoryBranch(ctx, &persistence.ForkHistoryBranchRequest{\n\t\tForkBranchToken: baseBranchToken,\n\t\tForkNodeID:      baseBranchLastEventID + 1,\n\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(domainID, workflowID, uuid.New()),\n\t\tShardID:         common.IntPtr(shardID),\n\t\tDomainName:      domainName,\n\t})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\n\tif err := newVersionHistory.SetBranchToken(resp.NewBranchToken); err != nil {\n\t\treturn 0, err\n\t}\n\tversionHistory := r.mutableState.GetVersionHistories()\n\tif versionHistory == nil {\n\t\treturn 0, execution.ErrMissingVersionHistories\n\t}\n\tbranchChanged, newIndex, err := versionHistory.AddVersionHistory(\n\t\tnewVersionHistory,\n\t)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tif branchChanged {\n\t\treturn 0, &types.BadRequestError{\n\t\t\tMessage: \"nDCBranchMgr encounter branch change during conflict resolution\",\n\t\t}\n\t}\n\n\treturn newIndex, nil\n}\n"
  },
  {
    "path": "service/history/ndc/branch_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: branch_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source branch_manager.go -destination branch_manager_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockbranchManager is a mock of branchManager interface.\ntype MockbranchManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockbranchManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockbranchManagerMockRecorder is the mock recorder for MockbranchManager.\ntype MockbranchManagerMockRecorder struct {\n\tmock *MockbranchManager\n}\n\n// NewMockbranchManager creates a new mock instance.\nfunc NewMockbranchManager(ctrl *gomock.Controller) *MockbranchManager {\n\tmock := &MockbranchManager{ctrl: ctrl}\n\tmock.recorder = &MockbranchManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockbranchManager) EXPECT() *MockbranchManagerMockRecorder {\n\treturn m.recorder\n}\n\n// prepareVersionHistory mocks base method.\nfunc (m *MockbranchManager) prepareVersionHistory(ctx context.Context, incomingVersionHistory *persistence.VersionHistory, incomingFirstEventID, incomingFirstEventVersion int64) (bool, int, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"prepareVersionHistory\", ctx, incomingVersionHistory, incomingFirstEventID, incomingFirstEventVersion)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(int)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// prepareVersionHistory indicates an expected call of prepareVersionHistory.\nfunc (mr *MockbranchManagerMockRecorder) prepareVersionHistory(ctx, incomingVersionHistory, incomingFirstEventID, incomingFirstEventVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"prepareVersionHistory\", reflect.TypeOf((*MockbranchManager)(nil).prepareVersionHistory), ctx, incomingVersionHistory, incomingFirstEventID, incomingFirstEventVersion)\n}\n"
  },
  {
    "path": "service/history/ndc/branch_manager_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tbranchManagerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockShard        *shard.TestContext\n\t\tmockContext      *execution.MockContext\n\t\tmockMutableState *execution.MockMutableState\n\n\t\tmockHistoryV2Manager *mocks.HistoryV2Manager\n\n\t\tlogger log.Logger\n\n\t\tbranchIndex int\n\t\tdomainID    string\n\t\tworkflowID  string\n\t\trunID       string\n\n\t\tbranchManager *branchManagerImpl\n\t}\n)\n\nfunc TestBranchManagerSuite(t *testing.T) {\n\ts := new(branchManagerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *branchManagerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockContext = execution.NewMockContext(s.controller)\n\ts.mockMutableState = execution.NewMockMutableState(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockHistoryV2Manager = s.mockShard.Resource.HistoryMgr\n\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.domainID = uuid.New()\n\ts.workflowID = \"some random workflow ID\"\n\ts.runID = uuid.New()\n\ts.branchIndex = 0\n\ts.branchManager = newBranchManager(\n\t\ts.mockShard, s.mockContext, s.mockMutableState, s.logger,\n\t).(*branchManagerImpl)\n}\n\nfunc (s *branchManagerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *branchManagerSuite) TestCreateNewBranch() {\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\tbaseBranchLCAEventVersion := int64(200)\n\tbaseBranchLCAEventID := int64(1394)\n\tbaseBranchLastEventVersion := int64(400)\n\tbaseBranchLastEventID := int64(2333)\n\tdomainName := \"test-domain-name\"\n\tversionHistory := persistence.NewVersionHistory(baseBranchToken, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID, baseBranchLCAEventVersion),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLastEventID, baseBranchLastEventVersion),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tnewBranchToken := []byte(\"some random new branch token\")\n\tnewVersionHistory, err := versionHistory.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID, baseBranchLCAEventVersion),\n\t)\n\ts.NoError(err)\n\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Manager.On(\"ForkHistoryBranch\", mock.Anything, mock.MatchedBy(func(input *persistence.ForkHistoryBranchRequest) bool {\n\t\tinput.Info = \"\"\n\t\ts.Equal(&persistence.ForkHistoryBranchRequest{\n\t\t\tForkBranchToken: baseBranchToken,\n\t\t\tForkNodeID:      baseBranchLCAEventID + 1,\n\t\t\tInfo:            \"\",\n\t\t\tShardID:         common.IntPtr(s.mockShard.GetShardID()),\n\t\t\tDomainName:      domainName,\n\t\t}, input)\n\t\treturn true\n\t})).Return(&persistence.ForkHistoryBranchResponse{\n\t\tNewBranchToken: newBranchToken,\n\t}, nil).Once()\n\n\tnewIndex, err := s.branchManager.createNewBranch(ctx.Background(), baseBranchToken, baseBranchLCAEventID, newVersionHistory)\n\ts.Nil(err)\n\ts.Equal(1, newIndex)\n\n\tcompareVersionHistory, err := versionHistory.DuplicateUntilLCAItem(\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID, baseBranchLCAEventVersion),\n\t)\n\ts.NoError(err)\n\ts.NoError(compareVersionHistory.SetBranchToken(newBranchToken))\n\tnewVersionHistory, err = versionHistories.GetVersionHistory(newIndex)\n\ts.NoError(err)\n\ts.True(compareVersionHistory.Equals(newVersionHistory))\n}\n\nfunc (s *branchManagerSuite) TestFlushBufferedEvents() {\n\n\tlastWriteVersion := int64(300)\n\tversionHistory := persistence.NewVersionHistory([]byte(\"some random base branch token\"), []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(100, 200),\n\t\tpersistence.NewVersionHistoryItem(150, 300),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tincomingVersionHistory := versionHistory.Duplicate()\n\terr := incomingVersionHistory.AddOrUpdateItem(\n\t\tpersistence.NewVersionHistoryItem(200, 300),\n\t)\n\ts.NoError(err)\n\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(lastWriteVersion, nil).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\ts.mockMutableState.EXPECT().UpdateCurrentVersion(lastWriteVersion, true).Return(nil).Times(1)\n\tdecisionInfo := &execution.DecisionInfo{\n\t\tScheduleID: 1234,\n\t\tStartedID:  2345,\n\t}\n\ts.mockMutableState.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t// GetExecutionInfo's return value is not used by this test\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\ts.mockMutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\tdecisionInfo.ScheduleID,\n\t\tdecisionInfo.StartedID,\n\t\ttypes.DecisionTaskFailedCauseFailoverCloseDecision,\n\t\t[]byte(nil),\n\t\texecution.IdentityHistoryService,\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\tint64(0),\n\t\t\"\",\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\ts.mockMutableState.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\ts.mockMutableState.EXPECT().HasPendingDecision().Return(false).Times(1)\n\ts.mockMutableState.EXPECT().AddDecisionTaskScheduledEvent(false).Return(&execution.DecisionInfo{}, nil).Times(1)\n\n\ts.mockContext.EXPECT().UpdateWorkflowExecutionAsActive(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\n\tctx := ctx.Background()\n\n\t_, _, err = s.branchManager.flushBufferedEvents(ctx, incomingVersionHistory)\n\ts.NoError(err)\n}\n\nfunc (s *branchManagerSuite) TestPrepareVersionHistory_BranchAppendable_NoMissingEventInBetween() {\n\n\tversionHistory := persistence.NewVersionHistory([]byte(\"some random base branch token\"), []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(100, 200),\n\t\tpersistence.NewVersionHistoryItem(150, 300),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tincomingVersionHistory := versionHistory.Duplicate()\n\terr := incomingVersionHistory.AddOrUpdateItem(\n\t\tpersistence.NewVersionHistoryItem(200, 300),\n\t)\n\ts.NoError(err)\n\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(false).AnyTimes()\n\n\tdoContinue, index, err := s.branchManager.prepareVersionHistory(\n\t\tctx.Background(),\n\t\tincomingVersionHistory,\n\t\t150+1,\n\t\t300)\n\ts.NoError(err)\n\ts.True(doContinue)\n\ts.Equal(0, index)\n}\n\nfunc (s *branchManagerSuite) TestPrepareVersionHistory_BranchAppendable_MissingEventInBetween() {\n\n\tversionHistory := persistence.NewVersionHistory([]byte(\"some random base branch token\"), []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(100, 200),\n\t\tpersistence.NewVersionHistoryItem(150, 300),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tincomingVersionHistory := versionHistory.Duplicate()\n\tincomingFirstEventVersionHistoryItem := persistence.NewVersionHistoryItem(200, 300)\n\terr := incomingVersionHistory.AddOrUpdateItem(\n\t\tincomingFirstEventVersionHistoryItem,\n\t)\n\ts.NoError(err)\n\n\texecution := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(false).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(execution).AnyTimes()\n\n\t_, _, err = s.branchManager.prepareVersionHistory(\n\t\tctx.Background(),\n\t\tincomingVersionHistory,\n\t\t150+2,\n\t\t300)\n\ts.IsType(&types.RetryTaskV2Error{}, err)\n}\n\nfunc (s *branchManagerSuite) TestPrepareVersionHistory_BranchNotAppendable_NoMissingEventInBetween() {\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\tbaseBranchLCAEventID := int64(85)\n\tbaseBranchLCAEventVersion := int64(200)\n\tdomainName := \"test-domainName\"\n\n\tversionHistory := persistence.NewVersionHistory(baseBranchToken, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID+10, baseBranchLCAEventVersion),\n\t\tpersistence.NewVersionHistoryItem(150, 300),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tincomingVersionHistory := persistence.NewVersionHistory(nil, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID, baseBranchLCAEventVersion),\n\t\tpersistence.NewVersionHistoryItem(200, 400),\n\t})\n\n\tnewBranchToken := []byte(\"some random new branch token\")\n\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(false).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Manager.On(\"ForkHistoryBranch\", mock.Anything, mock.MatchedBy(func(input *persistence.ForkHistoryBranchRequest) bool {\n\t\tinput.Info = \"\"\n\t\ts.Equal(&persistence.ForkHistoryBranchRequest{\n\t\t\tForkBranchToken: baseBranchToken,\n\t\t\tForkNodeID:      baseBranchLCAEventID + 1,\n\t\t\tInfo:            \"\",\n\t\t\tShardID:         common.IntPtr(s.mockShard.GetShardID()),\n\t\t\tDomainName:      domainName,\n\t\t}, input)\n\t\treturn true\n\t})).Return(&persistence.ForkHistoryBranchResponse{\n\t\tNewBranchToken: newBranchToken,\n\t}, nil).Once()\n\n\tdoContinue, index, err := s.branchManager.prepareVersionHistory(\n\t\tctx.Background(),\n\t\tincomingVersionHistory,\n\t\tbaseBranchLCAEventID+1,\n\t\tbaseBranchLCAEventVersion,\n\t)\n\ts.NoError(err)\n\ts.True(doContinue)\n\ts.Equal(1, index)\n}\n\nfunc (s *branchManagerSuite) TestPrepareVersionHistory_BranchNotAppendable_MissingEventInBetween() {\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\tbaseBranchLCAEventID := int64(85)\n\tbaseBranchLCAEventVersion := int64(200)\n\tbaseBranchLastEventID := int64(150)\n\tbaseBranchLastEventVersion := int64(300)\n\n\tversionHistory := persistence.NewVersionHistory(baseBranchToken, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID+10, baseBranchLCAEventVersion),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLastEventID, baseBranchLastEventVersion),\n\t})\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tincomingVersionHistory := persistence.NewVersionHistory(nil, []*persistence.VersionHistoryItem{\n\t\tpersistence.NewVersionHistoryItem(10, 0),\n\t\tpersistence.NewVersionHistoryItem(50, 100),\n\t\tpersistence.NewVersionHistoryItem(baseBranchLCAEventID, baseBranchLCAEventVersion),\n\t\tpersistence.NewVersionHistoryItem(200, 400),\n\t})\n\n\texecution := &persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().HasBufferedEvents().Return(false).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(execution).AnyTimes()\n\n\t_, _, err := s.branchManager.prepareVersionHistory(\n\t\tctx.Background(),\n\t\tincomingVersionHistory,\n\t\tbaseBranchLCAEventID+2,\n\t\tbaseBranchLCAEventVersion,\n\t)\n\ts.IsType(&types.RetryTaskV2Error{}, err)\n}\n"
  },
  {
    "path": "service/history/ndc/conflict_resolver.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination conflict_resolver_mock.go\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tconflictResolver interface {\n\t\tprepareMutableState(\n\t\t\tctx ctx.Context,\n\t\t\tbranchIndex int,\n\t\t\tincomingVersion int64,\n\t\t) (execution.MutableState, bool, error)\n\t}\n\n\tconflictResolverImpl struct {\n\t\tshard          shard.Context\n\t\tstateRebuilder execution.StateRebuilder\n\n\t\tcontext      execution.Context\n\t\tmutableState execution.MutableState\n\t\tlogger       log.Logger\n\t}\n)\n\nvar _ conflictResolver = (*conflictResolverImpl)(nil)\n\nfunc newConflictResolver(\n\tshard shard.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tlogger log.Logger,\n) conflictResolver {\n\n\treturn &conflictResolverImpl{\n\t\tshard:          shard,\n\t\tstateRebuilder: execution.NewStateRebuilder(shard, logger),\n\n\t\tcontext:      context,\n\t\tmutableState: mutableState,\n\t\tlogger:       logger,\n\t}\n}\n\nfunc (r *conflictResolverImpl) prepareMutableState(\n\tctx ctx.Context,\n\tbranchIndex int,\n\tincomingVersion int64,\n) (execution.MutableState, bool, error) {\n\n\tversionHistories := r.mutableState.GetVersionHistories()\n\tif versionHistories == nil {\n\t\treturn nil, false, execution.ErrMissingVersionHistories\n\t}\n\tcurrentVersionHistoryIndex := versionHistories.GetCurrentVersionHistoryIndex()\n\n\t// replication task to be applied to current branch\n\tif branchIndex == currentVersionHistoryIndex {\n\t\treturn r.mutableState, false, nil\n\t}\n\n\tcurrentVersionHistory, err := versionHistories.GetVersionHistory(currentVersionHistoryIndex)\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\tcurrentLastItem, err := currentVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\n\t// mutable state does not need rebuild\n\tif incomingVersion < currentLastItem.Version {\n\t\treturn r.mutableState, false, nil\n\t}\n\n\tif incomingVersion == currentLastItem.Version {\n\t\treturn nil, false, &types.BadRequestError{\n\t\t\tMessage: \"nDCConflictResolver encounter replication task version == current branch last write version\",\n\t\t}\n\t}\n\n\t// task.getVersion() > currentLastItem\n\t// incoming replication task, after application, will become the current branch\n\t// (because higher version wins), we need to rebuild the mutable state for that\n\trebuiltMutableState, err := r.rebuild(ctx, branchIndex, uuid.New())\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\treturn rebuiltMutableState, true, nil\n}\n\nfunc (r *conflictResolverImpl) rebuild(\n\tctx ctx.Context,\n\tbranchIndex int,\n\trequestID string,\n) (execution.MutableState, error) {\n\n\tversionHistories := r.mutableState.GetVersionHistories()\n\tif versionHistories == nil {\n\t\treturn nil, execution.ErrMissingVersionHistories\n\t}\n\treplayVersionHistory, err := versionHistories.GetVersionHistory(branchIndex)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastItem, err := replayVersionHistory.GetLastItem()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\texecutionInfo := r.mutableState.GetExecutionInfo()\n\tworkflowIdentifier := definition.NewWorkflowIdentifier(\n\t\texecutionInfo.DomainID,\n\t\texecutionInfo.WorkflowID,\n\t\texecutionInfo.RunID,\n\t)\n\n\trebuildMutableState, rebuiltHistorySize, err := r.stateRebuilder.Rebuild(\n\t\tctx,\n\t\texecutionInfo.StartTimestamp,\n\t\tworkflowIdentifier,\n\t\treplayVersionHistory.GetBranchToken(),\n\t\tlastItem.EventID,\n\t\tlastItem.Version,\n\t\tworkflowIdentifier,\n\t\tfunc() ([]byte, error) { return replayVersionHistory.GetBranchToken(), nil },\n\t\trequestID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// after rebuilt verification\n\trebuildVersionHistories := rebuildMutableState.GetVersionHistories()\n\tif rebuildVersionHistories == nil {\n\t\treturn nil, execution.ErrMissingVersionHistories\n\t}\n\trebuildVersionHistory, err := rebuildVersionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif !rebuildVersionHistory.Equals(replayVersionHistory) {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"nDCConflictResolver encounter mismatch version history after rebuild\",\n\t\t}\n\t}\n\n\t// set the current branch index to target branch index\n\t// set the version history back\n\t//\n\t// caller can use the IsRebuilt function in VersionHistories\n\t// telling whether mutable state is rebuilt, before apply new history events\n\tif err := versionHistories.SetCurrentVersionHistoryIndex(branchIndex); err != nil {\n\t\treturn nil, err\n\t}\n\tif err = rebuildMutableState.SetVersionHistories(versionHistories); err != nil {\n\t\treturn nil, err\n\t}\n\t// set the update condition from original mutable state\n\trebuildMutableState.SetUpdateCondition(r.mutableState.GetUpdateCondition())\n\n\tr.context.Clear()\n\tr.context.SetHistorySize(rebuiltHistorySize)\n\treturn rebuildMutableState, nil\n}\n"
  },
  {
    "path": "service/history/ndc/conflict_resolver_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: conflict_resolver.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source conflict_resolver.go -destination conflict_resolver_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MockconflictResolver is a mock of conflictResolver interface.\ntype MockconflictResolver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockconflictResolverMockRecorder\n\tisgomock struct{}\n}\n\n// MockconflictResolverMockRecorder is the mock recorder for MockconflictResolver.\ntype MockconflictResolverMockRecorder struct {\n\tmock *MockconflictResolver\n}\n\n// NewMockconflictResolver creates a new mock instance.\nfunc NewMockconflictResolver(ctrl *gomock.Controller) *MockconflictResolver {\n\tmock := &MockconflictResolver{ctrl: ctrl}\n\tmock.recorder = &MockconflictResolverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockconflictResolver) EXPECT() *MockconflictResolverMockRecorder {\n\treturn m.recorder\n}\n\n// prepareMutableState mocks base method.\nfunc (m *MockconflictResolver) prepareMutableState(ctx context.Context, branchIndex int, incomingVersion int64) (execution.MutableState, bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"prepareMutableState\", ctx, branchIndex, incomingVersion)\n\tret0, _ := ret[0].(execution.MutableState)\n\tret1, _ := ret[1].(bool)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// prepareMutableState indicates an expected call of prepareMutableState.\nfunc (mr *MockconflictResolverMockRecorder) prepareMutableState(ctx, branchIndex, incomingVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"prepareMutableState\", reflect.TypeOf((*MockconflictResolver)(nil).prepareMutableState), ctx, branchIndex, incomingVersion)\n}\n"
  },
  {
    "path": "service/history/ndc/conflict_resolver_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/net/context\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tconflictResolverSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockShard        *shard.TestContext\n\t\tmockContext      *execution.MockContext\n\t\tmockMutableState *execution.MockMutableState\n\t\tmockStateBuilder *execution.MockStateRebuilder\n\n\t\tlogger log.Logger\n\n\t\tdomainID   string\n\t\tdomainName string\n\t\tworkflowID string\n\t\trunID      string\n\n\t\tnDCConflictResolver *conflictResolverImpl\n\t}\n)\n\nfunc TestConflictResolverSuite(t *testing.T) {\n\ts := new(conflictResolverSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *conflictResolverSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockContext = execution.NewMockContext(s.controller)\n\ts.mockMutableState = execution.NewMockMutableState(s.controller)\n\ts.mockStateBuilder = execution.NewMockStateRebuilder(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.domainID = uuid.New()\n\ts.domainName = \"some random domain name\"\n\ts.workflowID = \"some random workflow ID\"\n\ts.runID = uuid.New()\n\n\ts.nDCConflictResolver = newConflictResolver(\n\t\ts.mockShard, s.mockContext, s.mockMutableState, s.logger,\n\t).(*conflictResolverImpl)\n\ts.nDCConflictResolver.stateRebuilder = s.mockStateBuilder\n}\n\nfunc (s *conflictResolverSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *conflictResolverSuite) TestRebuild() {\n\tctx := ctx.Background()\n\tupdateCondition := int64(59)\n\trequestID := uuid.New()\n\tversion := int64(12)\n\thistorySize := int64(12345)\n\n\tbranchToken0 := []byte(\"some random branch token\")\n\tlastEventID0 := int64(5)\n\tversionHistory0 := persistence.NewVersionHistory(\n\t\tbranchToken0,\n\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID0, version)},\n\t)\n\tbranchToken1 := []byte(\"other random branch token\")\n\tlastEventID1 := int64(2)\n\tversionHistory1 := persistence.NewVersionHistory(\n\t\tbranchToken1,\n\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID1, version)},\n\t)\n\tversionHistories := persistence.NewVersionHistories(versionHistory0)\n\t_, _, err := versionHistories.AddVersionHistory(versionHistory1)\n\ts.NoError(err)\n\n\ts.mockMutableState.EXPECT().GetUpdateCondition().Return(updateCondition).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}).AnyTimes()\n\n\tworkflowIdentifier := definition.NewWorkflowIdentifier(\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.runID,\n\t)\n\tmockRebuildMutableState := execution.NewMockMutableState(s.controller)\n\tmockRebuildMutableState.EXPECT().GetVersionHistories().Return(\n\t\tpersistence.NewVersionHistories(\n\t\t\tpersistence.NewVersionHistory(\n\t\t\t\tbranchToken1,\n\t\t\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID1, version)},\n\t\t\t),\n\t\t),\n\t).Times(1)\n\tmockRebuildMutableState.EXPECT().SetVersionHistories(versionHistories).Return(nil).Times(1)\n\tmockRebuildMutableState.EXPECT().SetUpdateCondition(updateCondition).Times(1)\n\n\ts.mockStateBuilder.EXPECT().Rebuild(\n\t\tctx,\n\t\tgomock.Any(),\n\t\tworkflowIdentifier,\n\t\tbranchToken1,\n\t\tlastEventID1,\n\t\tversion,\n\t\tworkflowIdentifier,\n\t\tgomock.Any(),\n\t\trequestID,\n\t).DoAndReturn(func(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowIdentifier, baseBranchToken []byte, baseRebuildLastEventID int64, baseRebuildLastEventVersion int64, targetWorkflowIdentifier definition.WorkflowIdentifier, targetBranchFn func() ([]byte, error), requestID string) (execution.MutableState, int64, error) {\n\t\ttargetBranchToken, err := targetBranchFn()\n\t\ts.NoError(err)\n\t\ts.Equal(branchToken1, targetBranchToken)\n\n\t\treturn mockRebuildMutableState, historySize, nil\n\t}).Times(1)\n\n\ts.mockContext.EXPECT().Clear().Times(1)\n\ts.mockContext.EXPECT().SetHistorySize(historySize).Times(1)\n\trebuiltMutableState, err := s.nDCConflictResolver.rebuild(ctx, 1, requestID)\n\ts.NoError(err)\n\ts.NotNil(rebuiltMutableState)\n\ts.Equal(1, versionHistories.GetCurrentVersionHistoryIndex())\n}\n\nfunc (s *conflictResolverSuite) TestPrepareMutableState_NoRebuild() {\n\tbranchToken := []byte(\"some random branch token\")\n\tlastEventID := int64(2)\n\tversion := int64(12)\n\tversionHistoryItem := persistence.NewVersionHistoryItem(lastEventID, version)\n\tversionHistory := persistence.NewVersionHistory(\n\t\tbranchToken,\n\t\t[]*persistence.VersionHistoryItem{versionHistoryItem},\n\t)\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\n\trebuiltMutableState, isRebuilt, err := s.nDCConflictResolver.prepareMutableState(ctx.Background(), 0, version)\n\ts.NoError(err)\n\ts.False(isRebuilt)\n\ts.Equal(s.mockMutableState, rebuiltMutableState)\n}\n\nfunc (s *conflictResolverSuite) TestPrepareMutableState_Rebuild() {\n\tctx := ctx.Background()\n\tupdateCondition := int64(59)\n\tversion := int64(12)\n\tincomingVersion := version + 1\n\thistorySize := int64(12345)\n\n\t// current branch\n\tbranchToken0 := []byte(\"some random branch token\")\n\tlastEventID0 := int64(2)\n\n\tversionHistoryItem0 := persistence.NewVersionHistoryItem(lastEventID0, version)\n\tversionHistory0 := persistence.NewVersionHistory(\n\t\tbranchToken0,\n\t\t[]*persistence.VersionHistoryItem{versionHistoryItem0},\n\t)\n\n\t// stale branch, used for rebuild\n\tbranchToken1 := []byte(\"other random branch token\")\n\tlastEventID1 := lastEventID0 - 1\n\tversionHistoryItem1 := persistence.NewVersionHistoryItem(lastEventID1, version)\n\tversionHistory1 := persistence.NewVersionHistory(\n\t\tbranchToken1,\n\t\t[]*persistence.VersionHistoryItem{versionHistoryItem1},\n\t)\n\n\tversionHistories := persistence.NewVersionHistories(versionHistory0)\n\t_, _, err := versionHistories.AddVersionHistory(versionHistory1)\n\ts.Nil(err)\n\n\ts.mockMutableState.EXPECT().GetUpdateCondition().Return(updateCondition).AnyTimes()\n\ts.mockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   s.domainID,\n\t\tWorkflowID: s.workflowID,\n\t\tRunID:      s.runID,\n\t}).AnyTimes()\n\n\tworkflowIdentifier := definition.NewWorkflowIdentifier(\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.runID,\n\t)\n\tmockRebuildMutableState := execution.NewMockMutableState(s.controller)\n\tmockRebuildMutableState.EXPECT().GetVersionHistories().Return(\n\t\tpersistence.NewVersionHistories(\n\t\t\tpersistence.NewVersionHistory(\n\t\t\t\tbranchToken1,\n\t\t\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID1, version)},\n\t\t\t),\n\t\t),\n\t).Times(1)\n\tmockRebuildMutableState.EXPECT().SetVersionHistories(versionHistories).Return(nil).Times(1)\n\tmockRebuildMutableState.EXPECT().SetUpdateCondition(updateCondition).Times(1)\n\n\ts.mockStateBuilder.EXPECT().Rebuild(\n\t\tctx,\n\t\tgomock.Any(),\n\t\tworkflowIdentifier,\n\t\tbranchToken1,\n\t\tlastEventID1,\n\t\tversion,\n\t\tworkflowIdentifier,\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t).DoAndReturn(func(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowIdentifier, baseBranchToken []byte, baseRebuildLastEventID int64, baseRebuildLastEventVersion int64, targetWorkflowIdentifier definition.WorkflowIdentifier, targetBranchFn func() ([]byte, error), requestID string) (execution.MutableState, int64, error) {\n\t\ttargetBranchToken, err := targetBranchFn()\n\t\ts.NoError(err)\n\t\ts.Equal(branchToken1, targetBranchToken)\n\n\t\treturn mockRebuildMutableState, historySize, nil\n\t}).Times(1)\n\n\ts.mockContext.EXPECT().Clear().Times(1)\n\ts.mockContext.EXPECT().SetHistorySize(int64(historySize)).Times(1)\n\trebuiltMutableState, isRebuilt, err := s.nDCConflictResolver.prepareMutableState(ctx, 1, incomingVersion)\n\ts.NoError(err)\n\ts.NotNil(rebuiltMutableState)\n\ts.True(isRebuilt)\n}\n"
  },
  {
    "path": "service/history/ndc/events_reapplier.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination events_reapplier_mock.go\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\t// EventsReapplier handles event re-application\n\tEventsReapplier interface {\n\t\tReapplyEvents(\n\t\t\tctx ctx.Context,\n\t\t\tmsBuilder execution.MutableState,\n\t\t\thistoryEvents []*types.HistoryEvent,\n\t\t\trunID string,\n\t\t) ([]*types.HistoryEvent, error)\n\t}\n\n\teventsReapplierImpl struct {\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t}\n)\n\nvar _ EventsReapplier = (*eventsReapplierImpl)(nil)\n\n// NewEventsReapplier creates events reapplier\nfunc NewEventsReapplier(\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) EventsReapplier {\n\n\treturn &eventsReapplierImpl{\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t}\n}\n\nfunc (r *eventsReapplierImpl) ReapplyEvents(\n\tctx ctx.Context,\n\tmsBuilder execution.MutableState,\n\thistoryEvents []*types.HistoryEvent,\n\trunID string,\n) ([]*types.HistoryEvent, error) {\n\n\tvar reappliedEvents []*types.HistoryEvent\n\tfor _, event := range historyEvents {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\tdedupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\t\t\tif msBuilder.IsResourceDuplicated(dedupResource) {\n\t\t\t\t// skip already applied event\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treappliedEvents = append(reappliedEvents, event)\n\t\t}\n\t}\n\n\tif len(reappliedEvents) == 0 {\n\t\treturn nil, nil\n\t}\n\n\t// sanity check workflow still running\n\tif !msBuilder.IsWorkflowExecutionRunning() {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"unable to reapply events to closed workflow.\",\n\t\t}\n\t}\n\n\tfor _, event := range reappliedEvents {\n\t\tsignal := event.GetWorkflowExecutionSignaledEventAttributes()\n\t\tif _, err := msBuilder.AddWorkflowExecutionSignaled(\n\t\t\tsignal.GetSignalName(),\n\t\t\tsignal.GetInput(),\n\t\t\tsignal.GetIdentity(),\n\t\t\t\"\", // Do not set requestID for requests reapplied, because they have already been applied previously\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdeDupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\t\tmsBuilder.UpdateDuplicatedResource(deDupResource)\n\t}\n\treturn reappliedEvents, nil\n}\n"
  },
  {
    "path": "service/history/ndc/events_reapplier_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: events_reapplier.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source events_reapplier.go -destination events_reapplier_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MockEventsReapplier is a mock of EventsReapplier interface.\ntype MockEventsReapplier struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEventsReapplierMockRecorder\n\tisgomock struct{}\n}\n\n// MockEventsReapplierMockRecorder is the mock recorder for MockEventsReapplier.\ntype MockEventsReapplierMockRecorder struct {\n\tmock *MockEventsReapplier\n}\n\n// NewMockEventsReapplier creates a new mock instance.\nfunc NewMockEventsReapplier(ctrl *gomock.Controller) *MockEventsReapplier {\n\tmock := &MockEventsReapplier{ctrl: ctrl}\n\tmock.recorder = &MockEventsReapplierMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEventsReapplier) EXPECT() *MockEventsReapplierMockRecorder {\n\treturn m.recorder\n}\n\n// ReapplyEvents mocks base method.\nfunc (m *MockEventsReapplier) ReapplyEvents(ctx context.Context, msBuilder execution.MutableState, historyEvents []*types.HistoryEvent, runID string) ([]*types.HistoryEvent, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReapplyEvents\", ctx, msBuilder, historyEvents, runID)\n\tret0, _ := ret[0].([]*types.HistoryEvent)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReapplyEvents indicates an expected call of ReapplyEvents.\nfunc (mr *MockEventsReapplierMockRecorder) ReapplyEvents(ctx, msBuilder, historyEvents, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReapplyEvents\", reflect.TypeOf((*MockEventsReapplier)(nil).ReapplyEvents), ctx, msBuilder, historyEvents, runID)\n}\n"
  },
  {
    "path": "service/history/ndc/events_reapplier_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\teventReapplicationSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller *gomock.Controller\n\n\t\treapplication EventsReapplier\n\t}\n)\n\nfunc TestEventReapplicationSuite(t *testing.T) {\n\ts := new(eventReapplicationSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *eventReapplicationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\tlogger := testlogger.New(s.Suite.T())\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.reapplication = NewEventsReapplier(\n\t\tmetricsClient,\n\t\tlogger,\n\t)\n}\n\nfunc (s *eventReapplicationSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *eventReapplicationSuite) TestReapplyEvents_AppliedEvent() {\n\trunID := uuid.New()\n\tworkflowExecution := &persistence.WorkflowExecutionInfo{\n\t\tDomainID: uuid.New(),\n\t}\n\tevent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tIdentity:   \"test\",\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte{},\n\t\t\tRequestID:  \"b90794a5-e9f4-4f41-9ebf-38aeedefd4ef\",\n\t\t},\n\t}\n\tattr := event.WorkflowExecutionSignaledEventAttributes\n\n\tmsBuilderCurrent := execution.NewMockMutableState(s.controller)\n\tmsBuilderCurrent.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\tmsBuilderCurrent.EXPECT().GetLastWriteVersion().Return(int64(1), nil).AnyTimes()\n\tmsBuilderCurrent.EXPECT().GetExecutionInfo().Return(workflowExecution).AnyTimes()\n\tmsBuilderCurrent.EXPECT().AddWorkflowExecutionSignaled(\n\t\tattr.GetSignalName(),\n\t\tattr.GetInput(),\n\t\tattr.GetIdentity(),\n\t\t\"\",\n\t).Return(event, nil).Times(1)\n\tdedupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\tmsBuilderCurrent.EXPECT().IsResourceDuplicated(dedupResource).Return(false).Times(1)\n\tmsBuilderCurrent.EXPECT().UpdateDuplicatedResource(dedupResource).Times(1)\n\tevents := []*types.HistoryEvent{\n\t\t{EventType: types.EventTypeWorkflowExecutionStarted.Ptr()},\n\t\tevent,\n\t}\n\tappliedEvent, err := s.reapplication.ReapplyEvents(context.Background(), msBuilderCurrent, events, runID)\n\ts.NoError(err)\n\ts.Equal(1, len(appliedEvent))\n}\n\nfunc (s *eventReapplicationSuite) TestReapplyEvents_Noop() {\n\trunID := uuid.New()\n\tevent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tIdentity:   \"test\",\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte{},\n\t\t},\n\t}\n\n\tmsBuilderCurrent := execution.NewMockMutableState(s.controller)\n\tdedupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\tmsBuilderCurrent.EXPECT().IsResourceDuplicated(dedupResource).Return(true).Times(1)\n\tevents := []*types.HistoryEvent{\n\t\t{EventType: types.EventTypeWorkflowExecutionStarted.Ptr()},\n\t\tevent,\n\t}\n\tappliedEvent, err := s.reapplication.ReapplyEvents(context.Background(), msBuilderCurrent, events, runID)\n\ts.NoError(err)\n\ts.Equal(0, len(appliedEvent))\n}\n\nfunc (s *eventReapplicationSuite) TestReapplyEvents_PartialAppliedEvent() {\n\trunID := uuid.New()\n\tworkflowExecution := &persistence.WorkflowExecutionInfo{\n\t\tDomainID: uuid.New(),\n\t}\n\tevent1 := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tIdentity:   \"test\",\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte{},\n\t\t\tRequestID:  \"3eb0594e-82dd-4335-8284-855c99d61c74\",\n\t\t},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:        2,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tIdentity:   \"test\",\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte{},\n\t\t\tRequestID:  \"2d2cae90-1ae4-4bcb-99a8-30b1cce64e3e\",\n\t\t},\n\t}\n\tattr1 := event1.WorkflowExecutionSignaledEventAttributes\n\n\tmsBuilderCurrent := execution.NewMockMutableState(s.controller)\n\tmsBuilderCurrent.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\tmsBuilderCurrent.EXPECT().GetLastWriteVersion().Return(int64(1), nil).AnyTimes()\n\tmsBuilderCurrent.EXPECT().GetExecutionInfo().Return(workflowExecution).AnyTimes()\n\tmsBuilderCurrent.EXPECT().AddWorkflowExecutionSignaled(\n\t\tattr1.GetSignalName(),\n\t\tattr1.GetInput(),\n\t\tattr1.GetIdentity(),\n\t\t\"\",\n\t).Return(event1, nil).Times(1)\n\tdedupResource1 := definition.NewEventReappliedID(runID, event1.ID, event1.Version)\n\tmsBuilderCurrent.EXPECT().IsResourceDuplicated(dedupResource1).Return(false).Times(1)\n\tdedupResource2 := definition.NewEventReappliedID(runID, event2.ID, event2.Version)\n\tmsBuilderCurrent.EXPECT().IsResourceDuplicated(dedupResource2).Return(true).Times(1)\n\tmsBuilderCurrent.EXPECT().UpdateDuplicatedResource(dedupResource1).Times(1)\n\tevents := []*types.HistoryEvent{\n\t\t{EventType: types.EventTypeWorkflowExecutionStarted.Ptr()},\n\t\tevent1,\n\t\tevent2,\n\t}\n\tappliedEvent, err := s.reapplication.ReapplyEvents(context.Background(), msBuilderCurrent, events, runID)\n\ts.NoError(err)\n\ts.Equal(1, len(appliedEvent))\n}\n\nfunc (s *eventReapplicationSuite) TestReapplyEvents_Error() {\n\trunID := uuid.New()\n\tworkflowExecution := &persistence.WorkflowExecutionInfo{\n\t\tDomainID: uuid.New(),\n\t}\n\tevent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tIdentity:   \"test\",\n\t\t\tSignalName: \"signal\",\n\t\t\tInput:      []byte{},\n\t\t\tRequestID:  \"1ece6551-27ac-4a9f-a086-5a780dea10f7\",\n\t\t},\n\t}\n\tattr := event.WorkflowExecutionSignaledEventAttributes\n\n\tmsBuilderCurrent := execution.NewMockMutableState(s.controller)\n\tmsBuilderCurrent.EXPECT().IsWorkflowExecutionRunning().Return(true)\n\tmsBuilderCurrent.EXPECT().GetLastWriteVersion().Return(int64(1), nil).AnyTimes()\n\tmsBuilderCurrent.EXPECT().GetExecutionInfo().Return(workflowExecution).AnyTimes()\n\tmsBuilderCurrent.EXPECT().AddWorkflowExecutionSignaled(\n\t\tattr.GetSignalName(),\n\t\tattr.GetInput(),\n\t\tattr.GetIdentity(),\n\t\t\"\",\n\t).Return(nil, fmt.Errorf(\"test\")).Times(1)\n\tdedupResource := definition.NewEventReappliedID(runID, event.ID, event.Version)\n\tmsBuilderCurrent.EXPECT().IsResourceDuplicated(dedupResource).Return(false).Times(1)\n\tevents := []*types.HistoryEvent{\n\t\t{EventType: types.EventTypeWorkflowExecutionStarted.Ptr()},\n\t\tevent,\n\t}\n\tappliedEvent, err := s.reapplication.ReapplyEvents(context.Background(), msBuilderCurrent, events, runID)\n\ts.Error(err)\n\ts.Equal(0, len(appliedEvent))\n}\n"
  },
  {
    "path": "service/history/ndc/existing_workflow_transaction_manager.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination existing_workflow_transaction_manager_mock.go\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\ttransactionManagerForExistingWorkflow interface {\n\t\tdispatchForExistingWorkflow(\n\t\t\tctx ctx.Context,\n\t\t\tnow time.Time,\n\t\t\tisWorkflowRebuilt bool,\n\t\t\ttargetWorkflow execution.Workflow,\n\t\t\tnewWorkflow execution.Workflow,\n\t\t) error\n\t}\n\n\ttransactionManagerForExistingWorkflowImpl struct {\n\t\ttransactionManager transactionManager\n\t\tlogger             log.Logger\n\t}\n)\n\nvar _ transactionManagerForExistingWorkflow = (*transactionManagerForExistingWorkflowImpl)(nil)\n\nfunc newTransactionManagerForExistingWorkflow(\n\ttransactionManager transactionManager,\n\tlogger log.Logger,\n) transactionManagerForExistingWorkflow {\n\n\treturn &transactionManagerForExistingWorkflowImpl{\n\t\ttransactionManager: transactionManager,\n\t\tlogger:             logger,\n\t}\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) dispatchForExistingWorkflow(\n\tctx ctx.Context,\n\tnow time.Time,\n\tisWorkflowRebuilt bool,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\t// NOTE: this function does NOT mutate current workflow, target workflow or new workflow,\n\t//  workflow mutation is done in methods within executeTransaction function\n\n\t// this is a performance optimization so most update does not need to\n\t// check whether target workflow is current workflow by calling DB API\n\tif !isWorkflowRebuilt && targetWorkflow.GetMutableState().IsCurrentWorkflowGuaranteed() {\n\t\t// NOTE: if target workflow is rebuilt, then IsCurrentWorkflowGuaranteed is not trustworthy\n\n\t\t// update to current record, since target workflow is pointed by current record\n\t\treturn r.dispatchWorkflowUpdateAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tisWorkflowRebuilt,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\t}\n\n\ttargetExecutionInfo := targetWorkflow.GetMutableState().GetExecutionInfo()\n\tdomainID := targetExecutionInfo.DomainID\n\tworkflowID := targetExecutionInfo.WorkflowID\n\ttargetRunID := targetExecutionInfo.RunID\n\n\t// the target workflow is rebuilt\n\t// we need to check the current workflow execution\n\tcurrentRunID, err := r.transactionManager.getCurrentWorkflowRunID(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif currentRunID == \"\" {\n\t\t// this means a bug in our code or DB is inconsistent...\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"nDCTransactionManager: unable to locate current workflow during update\",\n\t\t}\n\t}\n\n\tif currentRunID == targetRunID {\n\t\tif !isWorkflowRebuilt {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"nDCTransactionManager: encounter workflow not rebuilt & current workflow not guaranteed\",\n\t\t\t}\n\t\t}\n\n\t\t// update to current record, since target workflow is pointed by current record\n\t\treturn r.dispatchWorkflowUpdateAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tisWorkflowRebuilt,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\t}\n\n\t// there exists a current workflow, need additional check\n\tcurrentWorkflow, err := r.transactionManager.loadNDCWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tcurrentRunID,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttargetWorkflowIsNewer, err := targetWorkflow.HappensAfter(currentWorkflow)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !targetWorkflowIsNewer {\n\t\t// target workflow is older than current workflow, need to suppress the target workflow\n\t\treturn r.dispatchWorkflowUpdateAsZombie(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tisWorkflowRebuilt,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\t}\n\n\t// isWorkflowRebuilt is irrelevant here, because the DB API to be used\n\t// will set target workflow using snapshot\n\treturn r.executeTransaction(\n\t\tctx,\n\t\tnow,\n\t\ttransactionPolicySuppressCurrentAndUpdateAsCurrent,\n\t\tcurrentWorkflow,\n\t\ttargetWorkflow,\n\t\tnewWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) dispatchWorkflowUpdateAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\tisWorkflowRebuilt bool,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\tif !isWorkflowRebuilt {\n\t\treturn r.executeTransaction(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttransactionPolicyUpdateAsCurrent,\n\t\t\tnil,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\t}\n\n\treturn r.executeTransaction(\n\t\tctx,\n\t\tnow,\n\t\ttransactionPolicyConflictResolveAsCurrent,\n\t\tnil,\n\t\ttargetWorkflow,\n\t\tnewWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) dispatchWorkflowUpdateAsZombie(\n\tctx ctx.Context,\n\tnow time.Time,\n\tisWorkflowRebuilt bool,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\tif !isWorkflowRebuilt {\n\t\treturn r.executeTransaction(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttransactionPolicyUpdateAsZombie,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\t}\n\n\treturn r.executeTransaction(\n\t\tctx,\n\t\tnow,\n\t\ttransactionPolicyConflictResolveAsZombie,\n\t\tcurrentWorkflow,\n\t\ttargetWorkflow,\n\t\tnewWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) updateAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\tif newWorkflow == nil {\n\t\treturn targetWorkflow.GetContext().UpdateWorkflowExecutionAsPassive(ctx, now)\n\t}\n\n\treturn targetWorkflow.GetContext().UpdateWorkflowExecutionWithNewAsPassive(\n\t\tctx,\n\t\tnow,\n\t\tnewWorkflow.GetContext(),\n\t\tnewWorkflow.GetMutableState(),\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) updateAsZombie(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\ttargetPolicy, err := targetWorkflow.SuppressBy(\n\t\tcurrentWorkflow,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif targetPolicy != execution.TransactionPolicyPassive {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"nDCTransactionManagerForExistingWorkflow updateAsZombie encounter target workflow policy not being passive\",\n\t\t}\n\t}\n\n\tvar newContext execution.Context\n\tvar newMutableState execution.MutableState\n\tvar newTransactionPolicy *execution.TransactionPolicy\n\tif newWorkflow != nil {\n\t\tnewWorkflowPolicy, err := newWorkflow.SuppressBy(\n\t\t\tcurrentWorkflow,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif newWorkflowPolicy != execution.TransactionPolicyPassive {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"nDCTransactionManagerForExistingWorkflow updateAsZombie encounter new workflow policy not being passive\",\n\t\t\t}\n\t\t}\n\n\t\t// sanity check if new workflow is already created\n\t\t// since workflow resend can have already created the new workflow\n\t\tnewExecutionInfo := newWorkflow.GetMutableState().GetExecutionInfo()\n\t\tnewWorkflowExists, err := r.transactionManager.checkWorkflowExists(\n\t\t\tctx,\n\t\t\tnewExecutionInfo.DomainID,\n\t\t\tnewExecutionInfo.WorkflowID,\n\t\t\tnewExecutionInfo.RunID,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif newWorkflowExists {\n\t\t\t// new workflow already exists, do not create again\n\t\t\tnewContext = nil\n\t\t\tnewMutableState = nil\n\t\t\tnewTransactionPolicy = nil\n\t\t} else {\n\t\t\t// new workflow does not exists, continue\n\t\t\tnewContext = newWorkflow.GetContext()\n\t\t\tnewMutableState = newWorkflow.GetMutableState()\n\t\t\tnewTransactionPolicy = execution.TransactionPolicyPassive.Ptr()\n\t\t}\n\t}\n\n\t// release lock on current workflow, since current cluster maybe the active cluster\n\t//  and events maybe reapplied to current workflow\n\tcurrentWorkflow.GetReleaseFn()(nil)\n\tcurrentWorkflow = nil\n\n\tr.logger.Debugf(\"updateAsZombie calling UpdateWorkflowExecutionWithNew for wfID %s, current policy %v, new policy %v\",\n\t\ttargetWorkflow.GetMutableState().GetExecutionInfo().WorkflowID,\n\t\texecution.TransactionPolicyPassive,\n\t\tnewTransactionPolicy,\n\t)\n\treturn targetWorkflow.GetContext().UpdateWorkflowExecutionWithNew(\n\t\tctx,\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeBypassCurrent,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\texecution.TransactionPolicyPassive,\n\t\tnewTransactionPolicy,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) suppressCurrentAndUpdateAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\tvar err error\n\n\tcurrentWorkflowPolicy := execution.TransactionPolicyPassive\n\tif currentWorkflow.GetMutableState().IsWorkflowExecutionRunning() {\n\t\tcurrentWorkflowPolicy, err = currentWorkflow.SuppressBy(\n\t\t\ttargetWorkflow,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tif err := targetWorkflow.Revive(); err != nil {\n\t\treturn err\n\t}\n\n\tvar newContext execution.Context\n\tvar newMutableState execution.MutableState\n\tif newWorkflow != nil {\n\t\tnewContext = newWorkflow.GetContext()\n\t\tnewMutableState = newWorkflow.GetMutableState()\n\t\tif err := newWorkflow.Revive(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn targetWorkflow.GetContext().ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetWorkflow.GetMutableState(),\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tcurrentWorkflow.GetContext(),\n\t\tcurrentWorkflow.GetMutableState(),\n\t\tcurrentWorkflowPolicy.Ptr(),\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) conflictResolveAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\tvar newContext execution.Context\n\tvar newMutableState execution.MutableState\n\tif newWorkflow != nil {\n\t\tnewContext = newWorkflow.GetContext()\n\t\tnewMutableState = newWorkflow.GetMutableState()\n\t}\n\n\treturn targetWorkflow.GetContext().ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetWorkflow.GetMutableState(),\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) conflictResolveAsZombie(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\ttargetWorkflowPolicy, err := targetWorkflow.SuppressBy(\n\t\tcurrentWorkflow,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif targetWorkflowPolicy != execution.TransactionPolicyPassive {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"nDCTransactionManagerForExistingWorkflow conflictResolveAsZombie encounter target workflow policy not being passive\",\n\t\t}\n\t}\n\n\tvar newContext execution.Context\n\tvar newMutableState execution.MutableState\n\tif newWorkflow != nil {\n\t\tnewWorkflowPolicy, err := newWorkflow.SuppressBy(\n\t\t\tcurrentWorkflow,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif newWorkflowPolicy != execution.TransactionPolicyPassive {\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"nDCTransactionManagerForExistingWorkflow conflictResolveAsZombie encounter new workflow policy not being passive\",\n\t\t\t}\n\t\t}\n\n\t\t// sanity check if new workflow is already created\n\t\t// since workflow resend can have already created the new workflow\n\t\tnewExecutionInfo := newWorkflow.GetMutableState().GetExecutionInfo()\n\t\tnewWorkflowExists, err := r.transactionManager.checkWorkflowExists(\n\t\t\tctx,\n\t\t\tnewExecutionInfo.DomainID,\n\t\t\tnewExecutionInfo.WorkflowID,\n\t\t\tnewExecutionInfo.RunID,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif newWorkflowExists {\n\t\t\t// new workflow already exists, do not create again\n\t\t\tnewContext = nil\n\t\t\tnewMutableState = nil\n\t\t} else {\n\t\t\t// new workflow does not exists, continue\n\t\t\tnewContext = newWorkflow.GetContext()\n\t\t\tnewMutableState = newWorkflow.GetMutableState()\n\t\t}\n\t}\n\n\t// release lock on current workflow, since current cluster maybe the active cluster\n\t//  and events maybe reapplied to current workflow\n\tcurrentWorkflow.GetReleaseFn()(nil)\n\tcurrentWorkflow = nil\n\n\treturn targetWorkflow.GetContext().ConflictResolveWorkflowExecution(\n\t\tctx,\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\ttargetWorkflow.GetMutableState(),\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) executeTransaction(\n\tctx ctx.Context,\n\tnow time.Time,\n\ttransactionPolicy transactionPolicy,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) (retError error) {\n\n\tdefer func() {\n\t\tif rec := recover(); rec != nil {\n\t\t\tr.cleanupTransaction(currentWorkflow, targetWorkflow, newWorkflow, errPanic)\n\t\t\tpanic(rec)\n\t\t} else {\n\t\t\tr.cleanupTransaction(currentWorkflow, targetWorkflow, newWorkflow, retError)\n\t\t}\n\t}()\n\n\tswitch transactionPolicy {\n\tcase transactionPolicyUpdateAsCurrent:\n\t\treturn r.updateAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\n\tcase transactionPolicyUpdateAsZombie:\n\t\treturn r.updateAsZombie(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\n\tcase transactionPolicySuppressCurrentAndUpdateAsCurrent:\n\t\treturn r.suppressCurrentAndUpdateAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\n\tcase transactionPolicyConflictResolveAsCurrent:\n\t\treturn r.conflictResolveAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\n\tcase transactionPolicyConflictResolveAsZombie:\n\t\treturn r.conflictResolveAsZombie(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t\tnewWorkflow,\n\t\t)\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"nDCTransactionManager: encounter unknown transaction type: %v\", transactionPolicy),\n\t\t}\n\t}\n}\n\nfunc (r *transactionManagerForExistingWorkflowImpl) cleanupTransaction(\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n\terr error,\n) {\n\n\tif currentWorkflow != nil {\n\t\tcurrentWorkflow.GetReleaseFn()(err)\n\t}\n\tif targetWorkflow != nil {\n\t\ttargetWorkflow.GetReleaseFn()(err)\n\t}\n\tif newWorkflow != nil {\n\t\tnewWorkflow.GetReleaseFn()(err)\n\t}\n}\n"
  },
  {
    "path": "service/history/ndc/existing_workflow_transaction_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: existing_workflow_transaction_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source existing_workflow_transaction_manager.go -destination existing_workflow_transaction_manager_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MocktransactionManagerForExistingWorkflow is a mock of transactionManagerForExistingWorkflow interface.\ntype MocktransactionManagerForExistingWorkflow struct {\n\tctrl     *gomock.Controller\n\trecorder *MocktransactionManagerForExistingWorkflowMockRecorder\n\tisgomock struct{}\n}\n\n// MocktransactionManagerForExistingWorkflowMockRecorder is the mock recorder for MocktransactionManagerForExistingWorkflow.\ntype MocktransactionManagerForExistingWorkflowMockRecorder struct {\n\tmock *MocktransactionManagerForExistingWorkflow\n}\n\n// NewMocktransactionManagerForExistingWorkflow creates a new mock instance.\nfunc NewMocktransactionManagerForExistingWorkflow(ctrl *gomock.Controller) *MocktransactionManagerForExistingWorkflow {\n\tmock := &MocktransactionManagerForExistingWorkflow{ctrl: ctrl}\n\tmock.recorder = &MocktransactionManagerForExistingWorkflowMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MocktransactionManagerForExistingWorkflow) EXPECT() *MocktransactionManagerForExistingWorkflowMockRecorder {\n\treturn m.recorder\n}\n\n// dispatchForExistingWorkflow mocks base method.\nfunc (m *MocktransactionManagerForExistingWorkflow) dispatchForExistingWorkflow(ctx context.Context, now time.Time, isWorkflowRebuilt bool, targetWorkflow, newWorkflow execution.Workflow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"dispatchForExistingWorkflow\", ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// dispatchForExistingWorkflow indicates an expected call of dispatchForExistingWorkflow.\nfunc (mr *MocktransactionManagerForExistingWorkflowMockRecorder) dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"dispatchForExistingWorkflow\", reflect.TypeOf((*MocktransactionManagerForExistingWorkflow)(nil).dispatchForExistingWorkflow), ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n}\n"
  },
  {
    "path": "service/history/ndc/existing_workflow_transaction_manager_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\ttransactionManagerForExistingWorkflowSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockTransactionMgr *MocktransactionManager\n\n\t\tupdateMgr *transactionManagerForExistingWorkflowImpl\n\t}\n)\n\nfunc TestTransactionMgrForExistingWorkflowSuite(t *testing.T) {\n\ts := new(transactionManagerForExistingWorkflowSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockTransactionMgr = NewMocktransactionManager(s.controller)\n\n\ts.updateMgr = newTransactionManagerForExistingWorkflow(\n\t\ts.mockTransactionMgr,\n\t\ttestlogger.New(s.T()),\n\t).(*transactionManagerForExistingWorkflowImpl)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowGuaranteed() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tisWorkflowRebuilt := false\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(true).AnyTimes()\n\n\ttargetContext.EXPECT().UpdateWorkflowExecutionWithNewAsPassive(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tnewContext,\n\t\tnewMutableState,\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowNotGuaranteed_IsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\n\tisWorkflowRebuilt := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(targetRunID, nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.Error(err)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowNotGuaranteed_NotCurrent_CurrentRunning_UpdateAsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := false\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\tnewWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(true, nil)\n\tcurrentWorkflowPolicy := execution.TransactionPolicyPassive\n\tcurrentMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\tcurrentWorkflow.EXPECT().SuppressBy(targetWorkflow).Return(currentWorkflowPolicy, nil).Times(1)\n\ttargetWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetMutableState,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tcurrentContext,\n\t\tcurrentMutableState,\n\t\tcurrentWorkflowPolicy.Ptr(),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowNotGuaranteed_NotCurrent_CurrentComplete_UpdateAsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := false\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\tnewWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(true, nil)\n\tcurrentWorkflowPolicy := execution.TransactionPolicyPassive\n\tcurrentMutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tcurrentWorkflow.EXPECT().SuppressBy(targetWorkflow).Return(currentWorkflowPolicy, nil).Times(0)\n\ttargetWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetMutableState,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tcurrentContext,\n\t\tcurrentMutableState,\n\t\tcurrentWorkflowPolicy.Ptr(),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowNotGuaranteed_NotCurrent_UpdateAsZombie_NewRunDoesNotExists() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tnewRunID := \"some random new run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := false\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\tnewMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      newRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().checkWorkflowExists(ctx, domainID, workflowID, newRunID).Return(false, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\tnewWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeBypassCurrent,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\texecution.TransactionPolicyPassive,\n\t\texecution.TransactionPolicyPassive.Ptr(),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_NoRebuild_CurrentWorkflowNotGuaranteed_NotCurrent_UpdateAsZombie_NewRunDoesExists() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tnewRunID := \"some random new run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := false\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\tnewMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      newRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().checkWorkflowExists(ctx, domainID, workflowID, newRunID).Return(true, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\tnewWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeBypassCurrent,\n\t\t(execution.Context)(nil),\n\t\t(execution.MutableState)(nil),\n\t\texecution.TransactionPolicyPassive,\n\t\t(*execution.TransactionPolicy)(nil),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_Rebuild_IsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\n\tisWorkflowRebuilt := true\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(targetRunID, nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetMutableState,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\t(execution.Context)(nil),\n\t\t(execution.MutableState)(nil),\n\t\t(*execution.TransactionPolicy)(nil),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_Rebuild_NotCurrent_UpdateAsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := true\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\tnewWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(true, nil)\n\tcurrentWorkflowPolicy := execution.TransactionPolicyActive\n\tcurrentMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\tcurrentWorkflow.EXPECT().SuppressBy(targetWorkflow).Return(currentWorkflowPolicy, nil).Times(1)\n\ttargetWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeUpdateCurrent,\n\t\ttargetMutableState,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\tcurrentContext,\n\t\tcurrentMutableState,\n\t\tcurrentWorkflowPolicy.Ptr(),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_Rebuild_NotCurrent_UpdateAsZombie_NewRunDoesNotExists() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tnewRunID := \"some random new run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := true\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\tnewMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      newRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().checkWorkflowExists(ctx, domainID, workflowID, newRunID).Return(false, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\tnewWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\ttargetMutableState,\n\t\tnewContext,\n\t\tnewMutableState,\n\t\t(execution.Context)(nil),\n\t\t(execution.MutableState)(nil),\n\t\t(*execution.TransactionPolicy)(nil),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForExistingWorkflowSuite) TestDispatchForExistingWorkflow_Rebuild_NotCurrent_UpdateAsZombie_NewRunDoesExists() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tnewRunID := \"some random new run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\tisWorkflowRebuilt := true\n\n\ttargetReleaseCalled := false\n\tnewReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewContext := execution.NewMockContext(s.controller)\n\tnewMutableState := execution.NewMockMutableState(s.controller)\n\tvar newReleaseFn execution.ReleaseFunc = func(error) { newReleaseCalled = true }\n\tnewWorkflow.EXPECT().GetContext().Return(newContext).AnyTimes()\n\tnewWorkflow.EXPECT().GetMutableState().Return(newMutableState).AnyTimes()\n\tnewWorkflow.EXPECT().GetReleaseFn().Return(newReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\tnewMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      newRunID,\n\t}).AnyTimes()\n\ts.mockTransactionMgr.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\ts.mockTransactionMgr.EXPECT().checkWorkflowExists(ctx, domainID, workflowID, newRunID).Return(true, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\tnewWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().ConflictResolveWorkflowExecution(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.ConflictResolveWorkflowModeBypassCurrent,\n\t\ttargetMutableState,\n\t\t(execution.Context)(nil),\n\t\t(execution.MutableState)(nil),\n\t\t(execution.Context)(nil),\n\t\t(execution.MutableState)(nil),\n\t\t(*execution.TransactionPolicy)(nil),\n\t).Return(nil).Times(1)\n\n\terr := s.updateMgr.dispatchForExistingWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(newReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n"
  },
  {
    "path": "service/history/ndc/history_replicator.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tmutableStateMissingMessage = \"Resend events due to missing mutable state\"\n)\n\ntype (\n\t// HistoryReplicator handles history replication task\n\tHistoryReplicator interface {\n\t\tApplyEvents(\n\t\t\tctx ctx.Context,\n\t\t\trequest *types.ReplicateEventsV2Request,\n\t\t) error\n\t}\n\n\thistoryReplicatorImpl struct {\n\t\tshard              shard.Context\n\t\tclusterMetadata    cluster.Metadata\n\t\thistoryV2Manager   persistence.HistoryManager\n\t\thistorySerializer  persistence.PayloadSerializer\n\t\tmetricsClient      metrics.Client\n\t\tdomainCache        cache.DomainCache\n\t\texecutionCache     execution.Cache\n\t\teventsReapplier    EventsReapplier\n\t\ttransactionManager transactionManager\n\t\tlogger             log.Logger\n\n\t\tnewBranchManagerFn    newBranchManagerFn\n\t\tnewConflictResolverFn newConflictResolverFn\n\t\tnewWorkflowResetterFn newWorkflowResetterFn\n\t\tnewStateBuilderFn     newStateBuilderFn\n\t\tnewMutableStateFn     newMutableStateFn\n\n\t\t// refactored functions for a better testability\n\t\tnewReplicationTaskFn                                         newReplicationTaskFn\n\t\tapplyStartEventsFn                                           applyStartEventsFn\n\t\tapplyNonStartEventsPrepareBranchFn                           applyNonStartEventsPrepareBranchFn\n\t\tapplyNonStartEventsPrepareMutableStateFn                     applyNonStartEventsPrepareMutableStateFn\n\t\tapplyNonStartEventsToCurrentBranchFn                         applyNonStartEventsToCurrentBranchFn\n\t\tapplyNonStartEventsToNoneCurrentBranchFn                     applyNonStartEventsToNoneCurrentBranchFn\n\t\tapplyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn    applyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn\n\t\tapplyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn\n\t\tapplyNonStartEventsMissingMutableStateFn                     applyNonStartEventsMissingMutableStateFn\n\t\tapplyNonStartEventsResetWorkflowFn                           applyNonStartEventsResetWorkflowFn\n\t}\n\n\tnewStateBuilderFn func(\n\t\tmutableState execution.MutableState,\n\t\tlogger log.Logger) execution.StateBuilder\n\n\tnewMutableStateFn func(\n\t\tdomainEntry *cache.DomainCacheEntry,\n\t\tlogger log.Logger,\n\t) execution.MutableState\n\n\tnewBranchManagerFn func(\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tlogger log.Logger,\n\t) branchManager\n\n\tnewConflictResolverFn func(\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tlogger log.Logger,\n\t) conflictResolver\n\n\tnewWorkflowResetterFn func(\n\t\tdomainID string,\n\t\tworkflowID string,\n\t\tbaseRunID string,\n\t\tnewContext execution.Context,\n\t\tnewRunID string,\n\t\tlogger log.Logger,\n\t) WorkflowResetter\n\n\tnewReplicationTaskFn func(\n\t\tclusterMetadata cluster.Metadata,\n\t\thistorySerializer persistence.PayloadSerializer,\n\t\ttaskStartTime time.Time,\n\t\tlogger log.Logger,\n\t\trequest *types.ReplicateEventsV2Request,\n\t) (replicationTask, error)\n\n\tapplyStartEventsFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\treleaseFn execution.ReleaseFunc,\n\t\ttask replicationTask,\n\t\tdomainCache cache.DomainCache,\n\t\tnewMutableState newMutableStateFn,\n\t\tnewStateBuilder newStateBuilderFn,\n\t\ttransactionManager transactionManager,\n\t\tlogger log.Logger,\n\t\tshard shard.Context,\n\t\tclusterMetadata cluster.Metadata,\n\t) (retError error)\n\n\tapplyNonStartEventsPrepareBranchFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\ttask replicationTask,\n\t\tnewBranchManager newBranchManagerFn,\n\t) (bool, int, error)\n\n\tapplyNonStartEventsPrepareMutableStateFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tbranchIndex int,\n\t\ttask replicationTask,\n\t\tnewConflictResolver newConflictResolverFn,\n\t) (execution.MutableState, bool, error)\n\n\tapplyNonStartEventsToCurrentBranchFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tisRebuilt bool,\n\t\treleaseFn execution.ReleaseFunc,\n\t\ttask replicationTask,\n\t\tnewStateBuilder newStateBuilderFn,\n\t\tclusterMetadata cluster.Metadata,\n\t\tshard shard.Context,\n\t\tlogger log.Logger,\n\t\ttransactionManager transactionManager,\n\t) error\n\n\tapplyNonStartEventsToNoneCurrentBranchFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tbranchIndex int,\n\t\treleaseFn execution.ReleaseFunc,\n\t\ttask replicationTask,\n\t\tr *historyReplicatorImpl,\n\t\tlogger log.Logger,\n\t) error\n\n\tapplyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\treleaseFn execution.ReleaseFunc,\n\t\ttask replicationTask,\n\t\tr *historyReplicatorImpl,\n\t) error\n\n\tapplyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\tbranchIndex int,\n\t\treleaseFn execution.ReleaseFunc,\n\t\ttask replicationTask,\n\t\ttransactionManager transactionManager,\n\t\tclusterMetadata cluster.Metadata,\n\t\tshard shard.Context,\n\t\tlogger log.Logger,\n\t) error\n\n\tapplyNonStartEventsMissingMutableStateFn func(\n\t\tctx ctx.Context,\n\t\tnewContext execution.Context,\n\t\ttask replicationTask,\n\t\tnewWorkflowResetter newWorkflowResetterFn,\n\t) (execution.MutableState, error)\n\n\tapplyNonStartEventsResetWorkflowFn func(\n\t\tctx ctx.Context,\n\t\tcontext execution.Context,\n\t\tmutableState execution.MutableState,\n\t\ttask replicationTask,\n\t\tnewStateBuilder newStateBuilderFn,\n\t\ttransactionManager transactionManager,\n\t\tclusterMetadata cluster.Metadata,\n\t\tlogger log.Logger,\n\t\tshard shard.Context,\n\t) error\n)\n\nvar _ HistoryReplicator = (*historyReplicatorImpl)(nil)\n\nvar errPanic = errors.NewInternalFailureError(\"encounter panic\")\n\n// NewHistoryReplicator creates history replicator\nfunc NewHistoryReplicator(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\teventsReapplier EventsReapplier,\n\tlogger log.Logger,\n) HistoryReplicator {\n\n\ttransactionManager := newTransactionManager(shard, executionCache, eventsReapplier, logger)\n\treplicator := &historyReplicatorImpl{\n\t\tshard:              shard,\n\t\tclusterMetadata:    shard.GetService().GetClusterMetadata(),\n\t\thistoryV2Manager:   shard.GetHistoryManager(),\n\t\thistorySerializer:  persistence.NewPayloadSerializer(),\n\t\tmetricsClient:      shard.GetMetricsClient(),\n\t\tdomainCache:        shard.GetDomainCache(),\n\t\texecutionCache:     executionCache,\n\t\ttransactionManager: transactionManager,\n\t\teventsReapplier:    eventsReapplier,\n\t\tlogger:             logger.WithTags(tag.ComponentHistoryReplicator),\n\n\t\tnewBranchManagerFn: func(\n\t\t\tcontext execution.Context,\n\t\t\tmutableState execution.MutableState,\n\t\t\tlogger log.Logger,\n\t\t) branchManager {\n\t\t\treturn newBranchManager(shard, context, mutableState, logger)\n\t\t},\n\t\tnewConflictResolverFn: func(\n\t\t\tcontext execution.Context,\n\t\t\tmutableState execution.MutableState,\n\t\t\tlogger log.Logger,\n\t\t) conflictResolver {\n\t\t\treturn newConflictResolver(shard, context, mutableState, logger)\n\t\t},\n\t\tnewWorkflowResetterFn: func(\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\tbaseRunID string,\n\t\t\tnewContext execution.Context,\n\t\t\tnewRunID string,\n\t\t\tlogger log.Logger,\n\t\t) WorkflowResetter {\n\t\t\treturn NewWorkflowResetter(shard, transactionManager, domainID, workflowID, baseRunID, newContext, newRunID, logger)\n\t\t},\n\t\tnewStateBuilderFn: func(\n\t\t\tstate execution.MutableState,\n\t\t\tlogger log.Logger,\n\t\t) execution.StateBuilder {\n\n\t\t\treturn execution.NewStateBuilder(\n\t\t\t\tshard,\n\t\t\t\tlogger,\n\t\t\t\tstate,\n\t\t\t)\n\t\t},\n\t\tnewMutableStateFn: func(\n\t\t\tdomainEntry *cache.DomainCacheEntry,\n\t\t\tlogger log.Logger,\n\t\t) execution.MutableState {\n\t\t\treturn execution.NewMutableStateBuilderWithVersionHistories(\n\t\t\t\tshard,\n\t\t\t\tlogger,\n\t\t\t\tdomainEntry,\n\t\t\t\tconstants.EmptyVersion,\n\t\t\t)\n\t\t},\n\t\tnewReplicationTaskFn:                                         newReplicationTask,\n\t\tapplyStartEventsFn:                                           applyStartEvents,\n\t\tapplyNonStartEventsPrepareBranchFn:                           applyNonStartEventsPrepareBranch,\n\t\tapplyNonStartEventsPrepareMutableStateFn:                     applyNonStartEventsPrepareMutableState,\n\t\tapplyNonStartEventsToCurrentBranchFn:                         applyNonStartEventsToCurrentBranch,\n\t\tapplyNonStartEventsToNoneCurrentBranchFn:                     applyNonStartEventsToNoneCurrentBranch,\n\t\tapplyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn:    applyNonStartEventsToNoneCurrentBranchWithContinueAsNew,\n\t\tapplyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn: applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew,\n\t\tapplyNonStartEventsMissingMutableStateFn:                     applyNonStartEventsMissingMutableState,\n\t\tapplyNonStartEventsResetWorkflowFn:                           applyNonStartEventsResetWorkflow,\n\t}\n\n\treturn replicator\n}\n\nfunc (r *historyReplicatorImpl) ApplyEvents(\n\tctx ctx.Context,\n\trequest *types.ReplicateEventsV2Request,\n) (retError error) {\n\n\tstartTime := time.Now()\n\ttask, err := r.newReplicationTaskFn(\n\t\tr.clusterMetadata,\n\t\tr.historySerializer,\n\t\tstartTime,\n\t\tr.logger,\n\t\trequest,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn r.applyEvents(ctx, task)\n}\n\nfunc (r *historyReplicatorImpl) applyEvents(\n\tctx ctx.Context,\n\ttask replicationTask,\n) (retError error) {\n\n\tcontext, releaseFn, err := r.executionCache.GetOrCreateWorkflowExecution(\n\t\tctx,\n\t\ttask.getDomainID(),\n\t\t*task.getExecution(),\n\t)\n\tif err != nil {\n\t\t// for get workflow execution context, with valid run id\n\t\t// err will not be of type EntityNotExistsError\n\t\treturn err\n\t}\n\tdefer func() {\n\t\tif rec := recover(); rec != nil {\n\t\t\treleaseFn(errPanic)\n\t\t\tpanic(rec)\n\t\t} else {\n\t\t\treleaseFn(retError)\n\t\t}\n\t}()\n\n\tswitch task.getFirstEvent().GetEventType() {\n\tcase types.EventTypeWorkflowExecutionStarted:\n\t\treturn r.applyStartEventsFn(ctx, context, releaseFn, task, r.domainCache,\n\t\t\tr.newMutableStateFn, r.newStateBuilderFn, r.transactionManager, r.logger, r.shard, r.clusterMetadata)\n\n\tdefault:\n\t\t// apply events, other than simple start workflow execution\n\t\t// the continue as new + start workflow execution combination will also be processed here\n\t\tmutableState, err := context.LoadWorkflowExecutionWithTaskVersion(ctx, task.getVersion())\n\t\tswitch err.(type) {\n\t\tcase nil:\n\t\t\t// Sanity check to make only 3DC mutable state here\n\t\t\tif mutableState.GetVersionHistories() == nil {\n\t\t\t\treturn execution.ErrMissingVersionHistories\n\t\t\t}\n\n\t\t\tdoContinue, branchIndex, err := r.applyNonStartEventsPrepareBranchFn(ctx, context, mutableState, task, r.newBranchManagerFn)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t} else if !doContinue {\n\t\t\t\tr.metricsClient.IncCounter(metrics.ReplicateHistoryEventsScope, metrics.DuplicateReplicationEventsCounter)\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tmutableState, isRebuilt, err := r.applyNonStartEventsPrepareMutableStateFn(ctx, context, mutableState, branchIndex, task, r.newConflictResolverFn)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\t\tif mutableState.GetVersionHistories().GetCurrentVersionHistoryIndex() == branchIndex {\n\t\t\t\treturn r.applyNonStartEventsToCurrentBranchFn(ctx, context, mutableState, isRebuilt, releaseFn, task,\n\t\t\t\t\tr.newStateBuilderFn, r.clusterMetadata, r.shard, r.logger, r.transactionManager)\n\t\t\t}\n\t\t\t// passed in r because there's a recursive call within applyNonStartEventsToNoneCurrentBranchWithContinueAsNew\n\t\t\treturn r.applyNonStartEventsToNoneCurrentBranchFn(ctx, context, mutableState, branchIndex, releaseFn, task, r, r.logger)\n\n\t\tcase *types.EntityNotExistsError:\n\t\t\t// mutable state not created, check if is workflow reset\n\t\t\tmutableState, err := r.applyNonStartEventsMissingMutableStateFn(ctx, context, task, r.newWorkflowResetterFn)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\n\t\t\treturn r.applyNonStartEventsResetWorkflowFn(ctx, context, mutableState, task,\n\t\t\t\tr.newStateBuilderFn, r.transactionManager, r.clusterMetadata, r.logger, r.shard)\n\n\t\tdefault:\n\t\t\t// unable to get mutable state, return err so we can retry the task later\n\t\t\treturn err\n\t\t}\n\t}\n}\n\nfunc applyStartEvents(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\treleaseFn execution.ReleaseFunc,\n\ttask replicationTask,\n\tdomainCache cache.DomainCache,\n\tnewMutableState newMutableStateFn,\n\tnewStateBuilder newStateBuilderFn,\n\ttransactionManager transactionManager,\n\tlogger log.Logger,\n\tshard shard.Context,\n\tclusterMetadata cluster.Metadata,\n) (retError error) {\n\n\tdomainEntry, err := domainCache.GetDomainByID(task.getDomainID())\n\tif err != nil {\n\t\treturn err\n\t}\n\trequestID := uuid.New() // requestID used for start workflow execution request.  This is not on the history event.\n\t// since it's replicated from the other cluster, we don't care the active cluster policy, because the failover version will be updated after ApplyEvents\n\tmutableState := newMutableState(domainEntry, task.getLogger())\n\tstateBuilder := newStateBuilder(mutableState, task.getLogger())\n\n\t// use state builder for workflow mutable state mutation\n\t_, err = stateBuilder.ApplyEvents(\n\t\ttask.getDomainID(),\n\t\trequestID,\n\t\t*task.getExecution(),\n\t\ttask.getEvents(),\n\t\ttask.getNewEvents(),\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to apply events when applyStartEvents\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\n\terr = transactionManager.createWorkflow(\n\t\tctx,\n\t\ttask.getEventTime(),\n\t\texecution.NewWorkflow(\n\t\t\tctx,\n\t\t\tclusterMetadata,\n\t\t\tcontext,\n\t\t\tmutableState,\n\t\t\treleaseFn,\n\t\t\tlogger,\n\t\t),\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to create workflow when applyStartEvents\",\n\t\t\ttag.Error(err),\n\t\t)\n\t} else {\n\t\tnotify(task.getSourceCluster(), task.getEventTime(), logger, shard, clusterMetadata)\n\t}\n\treturn err\n}\n\nfunc applyNonStartEventsPrepareBranch(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\ttask replicationTask,\n\tnewBranchManager newBranchManagerFn,\n) (bool, int, error) {\n\n\tincomingVersionHistory := task.getVersionHistory()\n\tbranchManager := newBranchManager(context, mutableState, task.getLogger())\n\tdoContinue, versionHistoryIndex, err := branchManager.prepareVersionHistory(\n\t\tctx,\n\t\tincomingVersionHistory,\n\t\ttask.getFirstEvent().ID,\n\t\ttask.getFirstEvent().Version,\n\t)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn doContinue, versionHistoryIndex, nil\n\tcase *types.RetryTaskV2Error:\n\t\t// replication message can arrive out of order\n\t\t// do not log\n\t\treturn false, 0, err\n\tdefault:\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to prepare version history when applyNonStartEventsPrepareBranch\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn false, 0, err\n\t}\n}\n\nfunc applyNonStartEventsPrepareMutableState(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tbranchIndex int,\n\ttask replicationTask,\n\tnewConflictResolver newConflictResolverFn,\n) (execution.MutableState, bool, error) {\n\n\tincomingVersion := task.getVersion()\n\tconflictResolver := newConflictResolver(context, mutableState, task.getLogger())\n\tmutableState, isRebuilt, err := conflictResolver.prepareMutableState(\n\t\tctx,\n\t\tbranchIndex,\n\t\tincomingVersion,\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to prepare mutable state when applyNonStartEventsPrepareMutableState\",\n\t\t\ttag.Error(err),\n\t\t)\n\t}\n\treturn mutableState, isRebuilt, err\n}\n\nfunc applyNonStartEventsToCurrentBranch(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tisRebuilt bool,\n\treleaseFn execution.ReleaseFunc,\n\ttask replicationTask,\n\tnewStateBuilder newStateBuilderFn,\n\tclusterMetadata cluster.Metadata,\n\tshard shard.Context,\n\tlogger log.Logger,\n\ttransactionManager transactionManager,\n) error {\n\n\trequestID := uuid.New() // requestID used for start workflow execution request.  This is not on the history event.\n\tstateBuilder := newStateBuilder(mutableState, task.getLogger())\n\tnewMutableState, err := stateBuilder.ApplyEvents(\n\t\ttask.getDomainID(),\n\t\trequestID,\n\t\t*task.getExecution(),\n\t\ttask.getEvents(),\n\t\ttask.getNewEvents(),\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to apply events when applyNonStartEventsToCurrentBranch\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\n\ttargetWorkflow := execution.NewWorkflow(\n\t\tctx,\n\t\tclusterMetadata,\n\t\tcontext,\n\t\tmutableState,\n\t\treleaseFn,\n\t\tlogger,\n\t)\n\n\tvar newWorkflow execution.Workflow\n\tif newMutableState != nil {\n\t\tnewExecutionInfo := newMutableState.GetExecutionInfo()\n\t\tnewContext := execution.NewContext(\n\t\t\tnewExecutionInfo.DomainID,\n\t\t\ttypes.WorkflowExecution{\n\t\t\t\tWorkflowID: newExecutionInfo.WorkflowID,\n\t\t\t\tRunID:      newExecutionInfo.RunID,\n\t\t\t},\n\t\t\tshard,\n\t\t\tshard.GetExecutionManager(),\n\t\t\tlogger,\n\t\t)\n\n\t\tnewWorkflow = execution.NewWorkflow(\n\t\t\tctx,\n\t\t\tclusterMetadata,\n\t\t\tnewContext,\n\t\t\tnewMutableState,\n\t\t\texecution.NoopReleaseFn,\n\t\t\tlogger,\n\t\t)\n\t}\n\n\terr = transactionManager.updateWorkflow(\n\t\tctx,\n\t\ttask.getEventTime(),\n\t\tisRebuilt,\n\t\ttargetWorkflow,\n\t\tnewWorkflow,\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to update workflow when applyNonStartEventsToCurrentBranch\",\n\t\t\ttag.Error(err),\n\t\t)\n\t} else {\n\t\tnotify(task.getSourceCluster(), task.getEventTime(), logger, shard, clusterMetadata)\n\t}\n\treturn err\n}\n\nfunc applyNonStartEventsToNoneCurrentBranch(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tbranchIndex int,\n\treleaseFn execution.ReleaseFunc,\n\ttask replicationTask,\n\tr *historyReplicatorImpl,\n\tlogger log.Logger,\n) error {\n\n\tif len(task.getNewEvents()) != 0 {\n\t\treturn r.applyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn(\n\t\t\tctx,\n\t\t\tcontext,\n\t\t\treleaseFn,\n\t\t\ttask,\n\t\t\tr,\n\t\t)\n\t}\n\n\treturn r.applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn(\n\t\tctx,\n\t\tcontext,\n\t\tmutableState,\n\t\tbranchIndex,\n\t\treleaseFn,\n\t\ttask,\n\t\tr.transactionManager,\n\t\tr.clusterMetadata,\n\t\tr.shard,\n\t\tlogger,\n\t)\n}\n\nfunc applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\tbranchIndex int,\n\treleaseFn execution.ReleaseFunc,\n\ttask replicationTask,\n\ttransactionManager transactionManager,\n\tclusterMetadata cluster.Metadata,\n\tshard shard.Context,\n\tlogger log.Logger,\n) error {\n\n\tversionHistoryItem := persistence.NewVersionHistoryItem(\n\t\ttask.getLastEvent().ID,\n\t\ttask.getLastEvent().Version,\n\t)\n\tversionHistories := mutableState.GetVersionHistories()\n\tif versionHistories == nil {\n\t\treturn execution.ErrMissingVersionHistories\n\t}\n\tversionHistory, err := versionHistories.GetVersionHistory(branchIndex)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif err = versionHistory.AddOrUpdateItem(versionHistoryItem); err != nil {\n\t\treturn err\n\t}\n\n\terr = transactionManager.backfillWorkflow(\n\t\tctx,\n\t\ttask.getEventTime(),\n\t\texecution.NewWorkflow(\n\t\t\tctx,\n\t\t\tclusterMetadata,\n\t\t\tcontext,\n\t\t\tmutableState,\n\t\t\treleaseFn,\n\t\t\tlogger,\n\t\t),\n\t\t&persistence.WorkflowEvents{\n\t\t\tDomainID:    task.getDomainID(),\n\t\t\tWorkflowID:  task.getExecution().GetWorkflowID(),\n\t\t\tRunID:       task.getExecution().GetRunID(),\n\t\t\tBranchToken: versionHistory.GetBranchToken(),\n\t\t\tEvents:      task.getEvents(),\n\t\t},\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to backfill workflow when applyNonStartEventsToNoneCurrentBranch\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc applyNonStartEventsToNoneCurrentBranchWithContinueAsNew(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\treleaseFn execution.ReleaseFunc,\n\ttask replicationTask,\n\tr *historyReplicatorImpl,\n) error {\n\n\t// workflow backfill to non current branch with continue as new\n\t// first, release target workflow lock & create the new workflow as zombie\n\t// NOTE: need to release target workflow due to target workflow\n\t//  can potentially be the current workflow causing deadlock\n\n\t// 1. clear all in memory changes & release target workflow lock\n\t// 2. apply new workflow first\n\t// 3. apply target workflow\n\n\t// step 1\n\tcontext.Clear()\n\treleaseFn(nil)\n\n\t// step 2\n\tstartTime := time.Now()\n\ttask, newTask, err := task.splitTask(startTime)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif err := r.applyEvents(ctx, newTask); err != nil {\n\t\tnewTask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to create new workflow when applyNonStartEventsToNoneCurrentBranchWithContinueAsNew\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\n\t// step 3\n\tif err := r.applyEvents(ctx, task); err != nil {\n\t\tnewTask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to create target workflow when applyNonStartEventsToNoneCurrentBranchWithContinueAsNew\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc applyNonStartEventsMissingMutableState(\n\tctx ctx.Context,\n\tnewContext execution.Context,\n\ttask replicationTask,\n\tnewWorkflowResetter newWorkflowResetterFn,\n) (execution.MutableState, error) {\n\n\t// for non reset workflow execution replication task, just do re-replication\n\tif !task.isWorkflowReset() {\n\t\tfirstEvent := task.getFirstEvent()\n\t\treturn nil, newNDCRetryTaskErrorWithHint(\n\t\t\tmutableStateMissingMessage,\n\t\t\ttask.getDomainID(),\n\t\t\ttask.getWorkflowID(),\n\t\t\ttask.getRunID(),\n\t\t\tnil,\n\t\t\tnil,\n\t\t\tcommon.Int64Ptr(firstEvent.ID),\n\t\t\tcommon.Int64Ptr(firstEvent.Version),\n\t\t)\n\t}\n\n\tdecisionTaskEvent := task.getFirstEvent()\n\tbaseEventID := decisionTaskEvent.ID - 1\n\tbaseRunID, newRunID, baseEventVersion, _ := task.getWorkflowResetMetadata()\n\n\tworkflowResetter := newWorkflowResetter(\n\t\ttask.getDomainID(),\n\t\ttask.getWorkflowID(),\n\t\tbaseRunID,\n\t\tnewContext,\n\t\tnewRunID,\n\t\ttask.getLogger(),\n\t)\n\n\tresetMutableState, err := workflowResetter.ResetWorkflow(\n\t\tctx,\n\t\ttask.getEventTime(),\n\t\tbaseEventID,\n\t\tbaseEventVersion,\n\t\ttask.getFirstEvent().ID,\n\t\ttask.getVersion(),\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to reset workflow when applyNonStartEventsMissingMutableState\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn nil, err\n\t}\n\treturn resetMutableState, nil\n}\n\nfunc applyNonStartEventsResetWorkflow(\n\tctx ctx.Context,\n\tcontext execution.Context,\n\tmutableState execution.MutableState,\n\ttask replicationTask,\n\tnewStateBuilder newStateBuilderFn,\n\ttransactionManager transactionManager,\n\tclusterMetadata cluster.Metadata,\n\tlogger log.Logger,\n\tshard shard.Context,\n) error {\n\n\trequestID := uuid.New() // requestID used for start workflow execution request.  This is not on the history event.\n\tstateBuilder := newStateBuilder(mutableState, task.getLogger())\n\t_, err := stateBuilder.ApplyEvents(\n\t\ttask.getDomainID(),\n\t\trequestID,\n\t\t*task.getExecution(),\n\t\ttask.getEvents(),\n\t\ttask.getNewEvents(),\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to apply events when applyNonStartEventsResetWorkflow\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn err\n\t}\n\n\ttargetWorkflow := execution.NewWorkflow(\n\t\tctx,\n\t\tclusterMetadata,\n\t\tcontext,\n\t\tmutableState,\n\t\texecution.NoopReleaseFn,\n\t\tlogger,\n\t)\n\n\terr = transactionManager.createWorkflow(\n\t\tctx,\n\t\ttask.getEventTime(),\n\t\ttargetWorkflow,\n\t)\n\tif err != nil {\n\t\ttask.getLogger().Error(\n\t\t\t\"nDCHistoryReplicator unable to create workflow when applyNonStartEventsResetWorkflow\",\n\t\t\ttag.Error(err),\n\t\t)\n\t} else {\n\t\tnotify(task.getSourceCluster(), task.getEventTime(), logger, shard, clusterMetadata)\n\t}\n\treturn err\n}\n\nfunc notify(\n\tclusterName string,\n\tnow time.Time,\n\tlogger log.Logger,\n\tshard shard.Context,\n\tclusterMetadata cluster.Metadata,\n) {\n\tif clusterName == clusterMetadata.GetCurrentClusterName() {\n\t\t// this is a valid use case for testing, but not for production\n\t\tlogger.Warn(\"nDCHistoryReplicator applying events generated by current cluster\")\n\t\treturn\n\t}\n\tnow = now.Add(-shard.GetConfig().StandbyClusterDelay())\n\tshard.SetCurrentTime(clusterName, now)\n\tlogger.Debugf(\"History replicator setting current time to: %v for clusterName %v\", now, clusterName)\n}\n\nfunc newNDCRetryTaskErrorWithHint(\n\tmessage string,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tstartEventID *int64,\n\tstartEventVersion *int64,\n\tendEventID *int64,\n\tendEventVersion *int64,\n) error {\n\n\treturn &types.RetryTaskV2Error{\n\t\tMessage:           message,\n\t\tDomainID:          domainID,\n\t\tWorkflowID:        workflowID,\n\t\tRunID:             runID,\n\t\tStartEventID:      startEventID,\n\t\tStartEventVersion: startEventVersion,\n\t\tEndEventID:        endEventID,\n\t\tEndEventVersion:   endEventVersion,\n\t}\n}\n"
  },
  {
    "path": "service/history/ndc/history_replicator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nvar (\n\ttestShardID = 1234\n)\n\nfunc createTestHistoryReplicator(t *testing.T, domainID string) historyReplicatorImpl {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(1)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).AnyTimes()\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tclusterMetadata := cluster.Metadata{}\n\tmockShard.EXPECT().GetClusterMetadata().Return(clusterMetadata).Times(2)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(3)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(2)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(2)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(clusterMetadata).Times(1)\n\n\treplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\treplicatorImpl := replicator.(*historyReplicatorImpl)\n\treturn *replicatorImpl\n}\n\nfunc TestNewHistoryReplicator(t *testing.T) {\n\tassert.NotNil(t, createTestHistoryReplicator(t, uuid.New()))\n}\n\nfunc TestNewHistoryReplicator_newBranchManager(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(1)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).Times(1)\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(4)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(3)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(2)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\n\ttestReplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\ttestReplicatorImpl := testReplicator.(*historyReplicatorImpl)\n\n\t// test newBranchManagerFn function in history replicator\n\tmockExecutionContext := execution.NewMockContext(ctrl)\n\tmockExecutionMutableState := execution.NewMockMutableState(ctrl)\n\tassert.NotNil(t, testReplicatorImpl.newBranchManagerFn(mockExecutionContext, mockExecutionMutableState, log.NewNoop()))\n}\n\nfunc TestNewHistoryReplicator_newConflictResolver(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(2)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).Times(1)\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(3)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(4)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(3)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(4)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\n\ttestReplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\ttestReplicatorImpl := testReplicator.(*historyReplicatorImpl)\n\n\t// test newConflictResolverFn function in history replicator\n\tmockEventsCache := events.NewMockCache(ctrl)\n\tmockShard.EXPECT().GetEventsCache().Return(mockEventsCache).Times(1)\n\n\tmockExecutionContext := execution.NewMockContext(ctrl)\n\tmockExecutionMutableState := execution.NewMockMutableState(ctrl)\n\tassert.NotNil(t, testReplicatorImpl.newConflictResolverFn(mockExecutionContext, mockExecutionMutableState, log.NewNoop()))\n}\n\nfunc TestNewHistoryReplicator_newWorkflowResetter(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(2)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).Times(1)\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(3)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(5)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(3)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(4)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\n\ttestReplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\ttestReplicatorImpl := testReplicator.(*historyReplicatorImpl)\n\n\t// test newWorkflowResetterFn function in history replicator\n\tmockEventsCache := events.NewMockCache(ctrl)\n\tmockShard.EXPECT().GetEventsCache().Return(mockEventsCache).Times(1)\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\tmockExecutionContext := execution.NewMockContext(ctrl)\n\tassert.NotNil(t, testReplicatorImpl.newWorkflowResetterFn(\n\t\t\"test-domain-id\",\n\t\t\"test-workflow-id\",\n\t\t\"test-base-run-id\",\n\t\tmockExecutionContext,\n\t\t\"test-run-id\",\n\t\tlog.NewNoop(),\n\t))\n}\n\nfunc TestNewHistoryReplicator_newStateBuilder(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(1)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).Times(1)\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(3)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(3)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(3)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(2)\n\n\ttestReplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\ttestReplicatorImpl := testReplicator.(*historyReplicatorImpl)\n\n\t// test newStateBuilderFn function in history replicator\n\tmockExecutionMutableState := execution.NewMockMutableState(ctrl)\n\tassert.NotNil(t, testReplicatorImpl.newStateBuilderFn(mockExecutionMutableState, log.NewNoop()))\n}\n\nfunc TestNewHistoryReplicator_newMutableState(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\tNumberOfShards:                       0,\n\t\tIsAdvancedVisConfigExist:             false,\n\t\tMaxResponseSize:                      0,\n\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\tHostName:                             \"test-host\",\n\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t}).Times(2)\n\n\t// before going into NewHistoryReplicator\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).Times(1)\n\tmockShard.EXPECT().GetLogger().Return(log.NewNoop()).AnyTimes()\n\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).AnyTimes()\n\tmockShard.EXPECT().GetShardID().Return(testShardID).AnyTimes()\n\n\ttestExecutionCache := execution.NewCache(mockShard)\n\tmockEventsReapplier := NewMockEventsReapplier(ctrl)\n\n\t// going into NewHistoryReplicator -> newTransactionManager()\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(4)\n\tmockActiveClusterManager := activecluster.NewMockManager(ctrl)\n\tmockShard.EXPECT().GetActiveClusterManager().Return(mockActiveClusterManager).Times(2)\n\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\tmockShard.EXPECT().GetHistoryManager().Return(mockHistoryManager).Times(3)\n\n\tmockHistoryResource := resource.NewMockResource(ctrl)\n\tmockShard.EXPECT().GetService().Return(mockHistoryResource).Times(2)\n\tmockPayloadSerializer := persistence.NewMockPayloadSerializer(ctrl)\n\tmockHistoryResource.EXPECT().GetPayloadSerializer().Return(mockPayloadSerializer).Times(1)\n\n\t// going into NewHistoryReplicator -> newTransactionManager -> reset.NewWorkflowResetter\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).Times(3)\n\n\t// going back to NewHistoryReplicator\n\tmockHistoryResource.EXPECT().GetClusterMetadata().Return(cluster.Metadata{}).Times(1)\n\n\ttestReplicator := NewHistoryReplicator(mockShard, testExecutionCache, mockEventsReapplier, log.NewNoop())\n\ttestReplicatorImpl := testReplicator.(*historyReplicatorImpl)\n\n\t// test newMutableStateFn function in history replicator\n\tdeadline := int64(0)\n\tmockShard.EXPECT().GetTimeSource().Return(clock.NewMockedTimeSource()).Times(1)\n\tmockEventsCache := events.NewMockCache(ctrl)\n\tmockShard.EXPECT().GetEventsCache().Return(mockEventsCache).Times(1)\n\tmockDomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{Name: \"test-domain\"},\n\t\tnil,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"test-active-cluster\"},\n\t\t0,\n\t\t&deadline,\n\t\t0, 0, 0,\n\t)\n\tassert.NotNil(t, testReplicatorImpl.newMutableStateFn(mockDomainCacheEntry, log.NewNoop()))\n}\n\nfunc TestApplyEvents(t *testing.T) {\n\treplicator := createTestHistoryReplicator(t, uuid.New())\n\treplicator.newReplicationTaskFn = func(\n\t\tclusterMetadata cluster.Metadata,\n\t\thistorySerializer persistence.PayloadSerializer,\n\t\ttaskStartTime time.Time,\n\t\tlogger log.Logger,\n\t\trequest *types.ReplicateEventsV2Request,\n\t) (replicationTask, error) {\n\t\treturn nil, nil\n\t}\n\n\t// Intentionally panic result. Will test applyEvents function seperately\n\tassert.Panics(t, func() { replicator.ApplyEvents(nil, nil) })\n}\n\nfunc Test_applyEvents_EventTypeWorkflowExecutionStarted(t *testing.T) {\n\tworkflowExecutionStartedType := types.EventTypeWorkflowExecutionStarted\n\n\ttests := map[string]struct {\n\t\tmockExecutionCacheAffordance  func(mockExecutionCache *execution.MockCache)\n\t\tmockReplicationTaskAffordance func(mockReplicationTask *MockreplicationTask)\n\t\texpectError                   error\n\t}{\n\t\t\"Case1: success case\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionStartedType,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: error case\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, func(err error) {}, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\t// mock objects\n\t\t\treplicator := createTestHistoryReplicator(t, uuid.New())\n\t\t\tmockReplicationTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionCache := execution.NewMockCache(ctrl)\n\t\t\treplicator.executionCache = mockExecutionCache\n\t\t\t// mock functions\n\t\t\ttest.mockReplicationTaskAffordance(mockReplicationTask)\n\t\t\ttest.mockExecutionCacheAffordance(mockExecutionCache)\n\n\t\t\t// parameter functions affordance\n\t\t\treplicator.applyStartEventsFn = func(\n\t\t\t\tctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\ttask replicationTask,\n\t\t\t\tdomainCache cache.DomainCache,\n\t\t\t\tnewMutableState newMutableStateFn,\n\t\t\t\tnewStateBuilder newStateBuilderFn,\n\t\t\t\ttransactionManager transactionManager,\n\t\t\t\tlogger log.Logger,\n\t\t\t\tshard shard.Context,\n\t\t\t\tclusterMetadata cluster.Metadata,\n\t\t\t) error {\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tassert.Equal(t, replicator.applyEvents(ctx.Background(), mockReplicationTask), test.expectError)\n\t\t})\n\t}\n}\n\nfunc Test_applyEvents_defaultCase_noErrorBranch(t *testing.T) {\n\tworkflowExecutionType := types.EventTypeWorkflowExecutionCompleted\n\n\ttests := map[string]struct {\n\t\tmockExecutionCacheAffordance           func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext)\n\t\tmockReplicationTaskAffordance          func(mockReplicationTask *MockreplicationTask)\n\t\tmockExecutionContextAffordance         func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState)\n\t\tmockMutableStateAffordance             func(mockExecutionMutableState *execution.MockMutableState)\n\t\tmockApplyNonStartEventsPrepareBranchFn func(ctx ctx.Context,\n\t\t\tcontext execution.Context,\n\t\t\tmutableState execution.MutableState,\n\t\t\ttask replicationTask,\n\t\t\tnewBranchManager newBranchManagerFn,\n\t\t) (bool, int, error)\n\t\tapplyNonStartEventsPrepareMutableStateFnAffordance func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\tcontext execution.Context,\n\t\t\tmutableState execution.MutableState,\n\t\t\tbranchIndex int,\n\t\t\ttask replicationTask,\n\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t) (execution.MutableState, bool, error)\n\t\texpectError error\n\t}{\n\t\t\"Case1: case nil with GetVersionHistories is nil\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn false, 0, nil\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\treturn nil, false, fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: execution.ErrMissingVersionHistories,\n\t\t},\n\t\t\"Case2: case nil with applyNonStartEventsPrepareBranchFn error\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories:                  nil,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn false, 0, fmt.Errorf(\"test error\")\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\treturn nil, false, fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case3: case nil with applyNonStartEventsPrepareBranchFn no error and doContinue is false\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories:                  nil,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn false, 0, nil\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\treturn nil, false, fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case4: case nil with applyNonStartEventsPrepareMutableStateFn error\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories:                  nil,\n\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn true, 5, nil\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\tassert.Equal(t, branchIndex, 5)\n\t\t\t\t\treturn nil, false, fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case5: case nil with CurrentVersionHistoryIndex() == branchIndex\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories:                  nil,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn true, 1, nil\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\tassert.Equal(t, branchIndex, 1)\n\t\t\t\t\treturn mockExecutionMutableState, false, nil\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case6: case nil with CurrentVersionHistoryIndex() != branchIndex\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionMutableState.EXPECT().GetVersionHistories().Return(&persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 1,\n\t\t\t\t\tHistories:                  nil,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsPrepareBranchFn: func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewBranchManager newBranchManagerFn,\n\t\t\t) (bool, int, error) {\n\t\t\t\treturn true, 2, nil\n\t\t\t},\n\t\t\tapplyNonStartEventsPrepareMutableStateFnAffordance: func(mockExecutionMutableState *execution.MockMutableState) func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\tfn := func(ctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tnewConflictResolver newConflictResolverFn,\n\t\t\t\t) (execution.MutableState, bool, error) {\n\t\t\t\t\tassert.Equal(t, branchIndex, 2)\n\t\t\t\t\treturn mockExecutionMutableState, false, nil\n\t\t\t\t}\n\t\t\t\treturn fn\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\t// mock objects\n\t\t\treplicator := createTestHistoryReplicator(t, uuid.New())\n\t\t\tmockReplicationTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionCache := execution.NewMockCache(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockExecutionMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockMetricsClient := metrics.NewNoopMetricsClient()\n\t\t\treplicator.executionCache = mockExecutionCache\n\t\t\treplicator.metricsClient = mockMetricsClient\n\n\t\t\t// mock functions\n\t\t\ttest.mockReplicationTaskAffordance(mockReplicationTask)\n\t\t\ttest.mockExecutionCacheAffordance(mockExecutionCache, mockExecutionContext)\n\t\t\ttest.mockExecutionContextAffordance(mockExecutionContext, mockExecutionMutableState)\n\t\t\ttest.mockMutableStateAffordance(mockExecutionMutableState)\n\n\t\t\t// parameter functions affordance\n\t\t\treplicator.applyNonStartEventsPrepareBranchFn = test.mockApplyNonStartEventsPrepareBranchFn\n\t\t\treplicator.applyNonStartEventsPrepareMutableStateFn = test.applyNonStartEventsPrepareMutableStateFnAffordance(mockExecutionMutableState)\n\t\t\treplicator.applyNonStartEventsToCurrentBranchFn = func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tisRebuilt bool,\n\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewStateBuilder newStateBuilderFn,\n\t\t\t\tclusterMetadata cluster.Metadata,\n\t\t\t\tshard shard.Context,\n\t\t\t\tlogger log.Logger,\n\t\t\t\ttransactionManager transactionManager,\n\t\t\t) error {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treplicator.applyNonStartEventsToNoneCurrentBranchFn = func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\tbranchIndex int,\n\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\ttask replicationTask,\n\t\t\t\tr *historyReplicatorImpl,\n\t\t\t\tlogger log.Logger,\n\t\t\t) error {\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tassert.Equal(t, replicator.applyEvents(ctx.Background(), mockReplicationTask), test.expectError)\n\t\t})\n\t}\n}\n\nfunc Test_applyEvents_defaultCase_errorAndDefault(t *testing.T) {\n\tworkflowExecutionType := types.EventTypeWorkflowExecutionCompleted\n\n\ttests := map[string]struct {\n\t\tmockExecutionCacheAffordance                 func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext)\n\t\tmockReplicationTaskAffordance                func(mockReplicationTask *MockreplicationTask)\n\t\tmockExecutionContextAffordance               func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState)\n\t\tmockMutableStateAffordance                   func(mockExecutionMutableState *execution.MockMutableState)\n\t\tmockApplyNonStartEventsMissingMutableStateFn func(ctx ctx.Context,\n\t\t\tnewContext execution.Context,\n\t\t\ttask replicationTask,\n\t\t\tnewWorkflowResetter newWorkflowResetterFn,\n\t\t) (execution.MutableState, error)\n\t\texpectError error\n\t}{\n\t\t\"Case1-1: case EntityNotExistsError + applyNonStartEventsMissingMutableStateFn error\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, &types.EntityNotExistsError{}).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\treturn\n\t\t\t},\n\t\t\tmockApplyNonStartEventsMissingMutableStateFn: func(ctx ctx.Context,\n\t\t\t\tnewContext execution.Context,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewWorkflowResetter newWorkflowResetterFn,\n\t\t\t) (execution.MutableState, error) {\n\t\t\t\treturn nil, fmt.Errorf(\"test error\")\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case1-2: case EntityNotExistsError\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, &types.EntityNotExistsError{}).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\treturn\n\t\t\t},\n\t\t\tmockApplyNonStartEventsMissingMutableStateFn: func(ctx ctx.Context,\n\t\t\t\tnewContext execution.Context,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewWorkflowResetter newWorkflowResetterFn,\n\t\t\t) (execution.MutableState, error) {\n\t\t\t\treturn nil, nil\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case1-3: case other errors\": {\n\t\t\tmockExecutionCacheAffordance: func(mockExecutionCache *execution.MockCache, mockExecutionContext *execution.MockContext) {\n\t\t\t\tmockExecutionCache.EXPECT().GetOrCreateWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionContext, func(err error) {}, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tEventType: &workflowExecutionType,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockExecutionContextAffordance: func(mockExecutionContext *execution.MockContext, mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\tmockExecutionContext.EXPECT().LoadWorkflowExecutionWithTaskVersion(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecutionMutableState, fmt.Errorf(\"test-error\")).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockExecutionMutableState *execution.MockMutableState) {\n\t\t\t\treturn\n\t\t\t},\n\t\t\tmockApplyNonStartEventsMissingMutableStateFn: func(ctx ctx.Context,\n\t\t\t\tnewContext execution.Context,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewWorkflowResetter newWorkflowResetterFn,\n\t\t\t) (execution.MutableState, error) {\n\t\t\t\treturn nil, nil\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test-error\"),\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\t// mock objects\n\t\t\treplicator := createTestHistoryReplicator(t, uuid.New())\n\t\t\tmockReplicationTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionCache := execution.NewMockCache(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockExecutionMutableState := execution.NewMockMutableState(ctrl)\n\t\t\treplicator.executionCache = mockExecutionCache\n\n\t\t\t// mock functions\n\t\t\ttest.mockReplicationTaskAffordance(mockReplicationTask)\n\t\t\ttest.mockExecutionCacheAffordance(mockExecutionCache, mockExecutionContext)\n\t\t\ttest.mockExecutionContextAffordance(mockExecutionContext, mockExecutionMutableState)\n\t\t\ttest.mockMutableStateAffordance(mockExecutionMutableState)\n\n\t\t\t// parameter functions affordance\n\t\t\treplicator.applyNonStartEventsMissingMutableStateFn = test.mockApplyNonStartEventsMissingMutableStateFn\n\t\t\treplicator.applyNonStartEventsResetWorkflowFn = func(ctx ctx.Context,\n\t\t\t\tcontext execution.Context,\n\t\t\t\tmutableState execution.MutableState,\n\t\t\t\ttask replicationTask,\n\t\t\t\tnewStateBuilder newStateBuilderFn,\n\t\t\t\ttransactionManager transactionManager,\n\t\t\t\tclusterMetadata cluster.Metadata,\n\t\t\t\tlogger log.Logger,\n\t\t\t\tshard shard.Context,\n\t\t\t) error {\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\tassert.Equal(t, replicator.applyEvents(ctx.Background(), mockReplicationTask), test.expectError)\n\t\t})\n\t}\n}\n\nfunc Test_applyStartEvents(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockDomainCacheAffordance        func(mockDomainCache *cache.MockDomainCache)\n\t\tmockStateBuilderAffordance       func(mockStateBuilder *execution.MockStateBuilder)\n\t\tmockReplicationTaskAffordance    func(mockReplicationTask *MockreplicationTask)\n\t\tmockTransactionManagerAffordance func(mockTransactionManager *MocktransactionManager)\n\t\tmockShardContextAffordance       func(mockShardContext *shard.MockContext)\n\t\texpectError                      error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockDomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).\n\t\t\t\t\tReturn(cache.NewGlobalDomainCacheEntryForTest(nil, nil, nil, 1), nil).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(2)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getEventTime().Return(time.Now()).Times(2)\n\t\t\t\tmockReplicationTask.EXPECT().getSourceCluster().Return(\"test-source-cluster\").Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().createWorkflow(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockShardContextAffordance: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tNumberOfShards:                       0,\n\t\t\t\t\tIsAdvancedVisConfigExist:             false,\n\t\t\t\t\tMaxResponseSize:                      0,\n\t\t\t\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tHostName:                             \"test-host\",\n\t\t\t\t\tStandbyClusterDelay:                  dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}).Times(1)\n\t\t\t\tmockShard.EXPECT().SetCurrentTime(gomock.Any(), gomock.Any()).Times(1)\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: error when GetDomainByID fails\": {\n\t\t\tmockDomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\tmockShardContextAffordance:       func(mockShard *shard.MockContext) {},\n\t\t\texpectError:                      fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case3: error during createWorkflow\": {\n\t\t\tmockDomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).\n\t\t\t\t\tReturn(cache.NewGlobalDomainCacheEntryForTest(nil, nil, nil, 1), nil).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getLogger().Return(log.NewNoop()).Times(3)\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(2)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().createWorkflow(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockShardContextAffordance: func(mockShard *shard.MockContext) {},\n\t\t\texpectError:                fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case4: error when calling stateBuilder.ApplyEvents\": {\n\t\t\tmockDomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).\n\t\t\t\t\tReturn(cache.NewGlobalDomainCacheEntryForTest(nil, nil, nil, 1), nil).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockReplicationTaskAffordance: func(mockReplicationTask *MockreplicationTask) {\n\t\t\t\tmockReplicationTask.EXPECT().getLogger().Return(log.NewNoop()).Times(3)\n\t\t\t\tmockReplicationTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(2)\n\t\t\t\tmockReplicationTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockReplicationTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\tmockShardContextAffordance:       func(mockShard *shard.MockContext) {},\n\t\t\texpectError:                      fmt.Errorf(\"test error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockStateBuilder := execution.NewMockStateBuilder(ctrl)\n\t\t\tmockReplicationTask := NewMockreplicationTask(ctrl)\n\t\t\tmockTransactionManager := NewMocktransactionManager(ctrl)\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tlogger := log.NewNoop()\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockDomainCacheAffordance(mockDomainCache)\n\t\t\ttest.mockStateBuilderAffordance(mockStateBuilder)\n\t\t\ttest.mockReplicationTaskAffordance(mockReplicationTask)\n\t\t\ttest.mockTransactionManagerAffordance(mockTransactionManager)\n\t\t\ttest.mockShardContextAffordance(mockShard)\n\n\t\t\t// Mock functions\n\t\t\tmockNewMutableStateFn := func(\n\t\t\t\tdomainEntry *cache.DomainCacheEntry,\n\t\t\t\tlogger log.Logger,\n\t\t\t) execution.MutableState {\n\t\t\t\treturn mockMutableState\n\t\t\t}\n\n\t\t\tmockNewStateBuilderFn := func(\n\t\t\t\tstate execution.MutableState,\n\t\t\t\tlogger log.Logger,\n\t\t\t) execution.StateBuilder {\n\t\t\t\treturn mockStateBuilder\n\t\t\t}\n\n\t\t\t// Call the function under test\n\t\t\terr := applyStartEvents(ctx.Background(), mockExecutionContext, func(err error) {}, mockReplicationTask,\n\t\t\t\tmockDomainCache, mockNewMutableStateFn, mockNewStateBuilderFn, mockTransactionManager, logger, mockShard, cluster.Metadata{})\n\n\t\t\t// Assertions\n\t\t\t// can't change it to ErrorIs since ErrorIs need error chains\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsPrepareBranch(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance          func(mockTask *MockreplicationTask)\n\t\tmockBranchManagerAffordance func(mockBranchManager *MockbranchManager)\n\t\tmockNewBranchManagerFn      func(context execution.Context, mutableState execution.MutableState, logger log.Logger) branchManager\n\t\texpectDoContinue            bool\n\t\texpectVersionHistoryIndex   int\n\t\texpectError                 error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getVersionHistory().Return(&persistence.VersionHistory{}).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockBranchManagerAffordance: func(mockBranchManager *MockbranchManager) {\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tint64(1),\n\t\t\t\t\tint64(1),\n\t\t\t\t).Return(true, 5, nil).Times(1)\n\t\t\t},\n\t\t\tmockNewBranchManagerFn: func(context execution.Context, mutableState execution.MutableState, logger log.Logger) branchManager {\n\t\t\t\tmockBranchManager := NewMockbranchManager(gomock.NewController(t))\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(true, 5, nil).Times(1)\n\t\t\t\treturn mockBranchManager\n\t\t\t},\n\t\t\texpectDoContinue:          true,\n\t\t\texpectVersionHistoryIndex: 5,\n\t\t\texpectError:               nil,\n\t\t},\n\t\t\"Case2: RetryTaskV2Error case\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getVersionHistory().Return(&persistence.VersionHistory{}).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockBranchManagerAffordance: func(mockBranchManager *MockbranchManager) {\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tint64(1),\n\t\t\t\t\tint64(1),\n\t\t\t\t).Return(false, 0, &types.RetryTaskV2Error{}).Times(1)\n\t\t\t},\n\t\t\tmockNewBranchManagerFn: func(context execution.Context, mutableState execution.MutableState, logger log.Logger) branchManager {\n\t\t\t\tmockBranchManager := NewMockbranchManager(gomock.NewController(t))\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(false, 0, &types.RetryTaskV2Error{}).Times(1)\n\t\t\t\treturn mockBranchManager\n\t\t\t},\n\t\t\texpectDoContinue:          false,\n\t\t\texpectVersionHistoryIndex: 0,\n\t\t\texpectError:               &types.RetryTaskV2Error{},\n\t\t},\n\t\t\"Case3: unknown error case\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getVersionHistory().Return(&persistence.VersionHistory{}).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockBranchManagerAffordance: func(mockBranchManager *MockbranchManager) {\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tint64(1),\n\t\t\t\t\tint64(1),\n\t\t\t\t).Return(false, 0, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockNewBranchManagerFn: func(context execution.Context, mutableState execution.MutableState, logger log.Logger) branchManager {\n\t\t\t\tmockBranchManager := NewMockbranchManager(gomock.NewController(t))\n\t\t\t\tmockBranchManager.EXPECT().prepareVersionHistory(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(false, 0, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t\treturn mockBranchManager\n\t\t\t},\n\t\t\texpectDoContinue:          false,\n\t\t\texpectVersionHistoryIndex: 0,\n\t\t\texpectError:               fmt.Errorf(\"test error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\tmockBranchManager := NewMockbranchManager(ctrl)\n\t\t\ttest.mockBranchManagerAffordance(mockBranchManager)\n\n\t\t\t// Mock newBranchManagerFn\n\t\t\tnewBranchManagerFn := func(context execution.Context, mutableState execution.MutableState, logger log.Logger) branchManager {\n\t\t\t\treturn mockBranchManager\n\t\t\t}\n\n\t\t\t// Call the function under test\n\t\t\tdoContinue, versionHistoryIndex, err := applyNonStartEventsPrepareBranch(ctx.Background(), mockExecutionContext, mockMutableState, mockTask, newBranchManagerFn)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectDoContinue, doContinue)\n\t\t\tassert.Equal(t, test.expectVersionHistoryIndex, versionHistoryIndex)\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsPrepareMutableState(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance             func(mockTask *MockreplicationTask)\n\t\tmockConflictResolverAffordance func(mockConflictResolver *MockconflictResolver)\n\t\tmockNewConflictResolverFn      func(context execution.Context, mutableState execution.MutableState, logger log.Logger) conflictResolver\n\t\texpectMutableState             execution.MutableState\n\t\texpectIsRebuilt                bool\n\t\texpectError                    error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t},\n\t\t\tmockConflictResolverAffordance: func(mockConflictResolver *MockconflictResolver) {\n\t\t\t\tmockMutableState := execution.NewMockMutableState(gomock.NewController(t)) // Define mockMutableState\n\t\t\t\tmockConflictResolver.EXPECT().prepareMutableState(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t5,        // branchIndex\n\t\t\t\t\tint64(1), // incomingVersion\n\t\t\t\t).Return(mockMutableState, true, nil).Times(1)\n\t\t\t},\n\t\t\tmockNewConflictResolverFn: func(context execution.Context, mutableState execution.MutableState, logger log.Logger) conflictResolver {\n\t\t\t\tmockConflictResolver := NewMockconflictResolver(gomock.NewController(t))\n\t\t\t\treturn mockConflictResolver\n\t\t\t},\n\t\t\texpectMutableState: execution.NewMockMutableState(gomock.NewController(t)), // Use same mock for expectations\n\t\t\texpectIsRebuilt:    true,\n\t\t\texpectError:        nil,\n\t\t},\n\t\t\"Case2: error during prepareMutableState\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2) // Error logging case\n\t\t\t},\n\t\t\tmockConflictResolverAffordance: func(mockConflictResolver *MockconflictResolver) {\n\t\t\t\tmockConflictResolver.EXPECT().prepareMutableState(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t5,        // branchIndex\n\t\t\t\t\tint64(1), // incomingVersion\n\t\t\t\t).Return(nil, false, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockNewConflictResolverFn: func(context execution.Context, mutableState execution.MutableState, logger log.Logger) conflictResolver {\n\t\t\t\tmockConflictResolver := NewMockconflictResolver(gomock.NewController(t))\n\t\t\t\treturn mockConflictResolver\n\t\t\t},\n\t\t\texpectMutableState: nil,\n\t\t\texpectIsRebuilt:    false,\n\t\t\texpectError:        fmt.Errorf(\"test error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl) // Define mockMutableState\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\tmockConflictResolver := NewMockconflictResolver(ctrl)\n\t\t\ttest.mockConflictResolverAffordance(mockConflictResolver)\n\n\t\t\t// Mock newConflictResolverFn\n\t\t\tnewConflictResolverFn := func(context execution.Context, mutableState execution.MutableState, logger log.Logger) conflictResolver {\n\t\t\t\treturn mockConflictResolver\n\t\t\t}\n\n\t\t\t// Call the function under test\n\t\t\tmutableState, isRebuilt, err := applyNonStartEventsPrepareMutableState(ctx.Background(), mockExecutionContext, mockMutableState, 5, mockTask, newConflictResolverFn)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectMutableState, mutableState)\n\t\t\tassert.Equal(t, test.expectIsRebuilt, isRebuilt)\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsToCurrentBranch(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance               func(mockTask *MockreplicationTask)\n\t\tmockStateBuilderAffordance       func(mockStateBuilder *execution.MockStateBuilder, mockMutableState *execution.MockMutableState)\n\t\tmockTransactionManagerAffordance func(mockTransactionManager *MocktransactionManager)\n\t\tmockShardAffordance              func(mockShard *shard.MockContext)\n\t\tmockNewStateBuilderFn            func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder\n\t\texpectError                      error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(2)\n\t\t\t\tmockTask.EXPECT().getSourceCluster().Return(\"test-source-cluster\").Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(mockMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().updateWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttrue,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockShardAffordance: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(nil).Times(1)\n\t\t\t\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).Times(1)\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tNumberOfShards:                       0,\n\t\t\t\t\tIsAdvancedVisConfigExist:             false,\n\t\t\t\t\tMaxResponseSize:                      0,\n\t\t\t\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tHostName:                             \"test-host\",\n\t\t\t\t\tStandbyClusterDelay:                  dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}).Times(1)\n\t\t\t\tmockShard.EXPECT().SetCurrentTime(gomock.Any(), gomock.Any()).Times(1)\n\t\t\t},\n\t\t\tmockNewStateBuilderFn: func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder {\n\t\t\t\tmockStateBuilder := execution.NewMockStateBuilder(gomock.NewController(t))\n\t\t\t\treturn mockStateBuilder\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: error during ApplyEvents\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(nil, fmt.Errorf(\"test error\")).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\tmockShardAffordance:              func(mockShard *shard.MockContext) {},\n\t\t\tmockNewStateBuilderFn: func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder {\n\t\t\t\tmockStateBuilder := execution.NewMockStateBuilder(gomock.NewController(t))\n\t\t\t\treturn mockStateBuilder\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case3: error during updateWorkflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\tDomainID:   \"test-domain-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(mockMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().updateWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttrue,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(fmt.Errorf(\"test update error\")).Times(1)\n\t\t\t},\n\t\t\tmockShardAffordance: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(nil).Times(1)\n\t\t\t\tmockShard.EXPECT().GetMetricsClient().Return(metrics.NewNoopMetricsClient()).Times(1)\n\t\t\t},\n\t\t\tmockNewStateBuilderFn: func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder {\n\t\t\t\tmockStateBuilder := execution.NewMockStateBuilder(gomock.NewController(t))\n\t\t\t\treturn mockStateBuilder\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test update error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockTransactionManager := NewMocktransactionManager(ctrl)\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tlogger := log.NewNoop()\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\tmockStateBuilder := execution.NewMockStateBuilder(ctrl)\n\t\t\ttest.mockStateBuilderAffordance(mockStateBuilder, mockMutableState)\n\t\t\ttest.mockTransactionManagerAffordance(mockTransactionManager)\n\t\t\ttest.mockShardAffordance(mockShard)\n\n\t\t\t// Mock newStateBuilderFn\n\t\t\tnewStateBuilderFn := func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder {\n\t\t\t\treturn mockStateBuilder\n\t\t\t}\n\n\t\t\t// Call the function under test\n\t\t\terr := applyNonStartEventsToCurrentBranch(ctx.Background(), mockExecutionContext, mockMutableState, true, func(error) {}, mockTask, newStateBuilderFn, cluster.Metadata{}, mockShard, logger, mockTransactionManager)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsToNoneCurrentBranch(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance                                    func(mockTask *MockreplicationTask)\n\t\tmockApplyNonStartEventsWithContinueAsNewAffordance    func(replicator *historyReplicatorImpl)\n\t\tmockApplyNonStartEventsWithoutContinueAsNewAffordance func(replicator *historyReplicatorImpl)\n\t\texpectError                                           error\n\t}{\n\t\t\"Case1: with NewEvents, should call applyNonStartEventsToNoneCurrentBranchWithContinueAsNew\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return([]*types.HistoryEvent{{}}).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {\n\t\t\t\treplicator.applyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn = func(\n\t\t\t\t\tctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tr *historyReplicatorImpl,\n\t\t\t\t) error {\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithoutContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: without NewEvents, should call applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {},\n\t\t\tmockApplyNonStartEventsWithoutContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {\n\t\t\t\treplicator.applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn = func(\n\t\t\t\t\tctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\ttransactionManager transactionManager,\n\t\t\t\t\tclusterMetadata cluster.Metadata,\n\t\t\t\t\tshard shard.Context,\n\t\t\t\t\tlogger log.Logger,\n\t\t\t\t) error {\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case3: error case for applyNonStartEventsToNoneCurrentBranchWithContinueAsNew\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return([]*types.HistoryEvent{{}}).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {\n\t\t\t\treplicator.applyNonStartEventsToNoneCurrentBranchWithContinueAsNewFn = func(\n\t\t\t\t\tctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\tr *historyReplicatorImpl,\n\t\t\t\t) error {\n\t\t\t\t\treturn fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithoutContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t\t\"Case4: error case for applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockApplyNonStartEventsWithContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {},\n\t\t\tmockApplyNonStartEventsWithoutContinueAsNewAffordance: func(replicator *historyReplicatorImpl) {\n\t\t\t\treplicator.applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNewFn = func(\n\t\t\t\t\tctx ctx.Context,\n\t\t\t\t\tcontext execution.Context,\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tbranchIndex int,\n\t\t\t\t\treleaseFn execution.ReleaseFunc,\n\t\t\t\t\ttask replicationTask,\n\t\t\t\t\ttransactionManager transactionManager,\n\t\t\t\t\tclusterMetadata cluster.Metadata,\n\t\t\t\t\tshard shard.Context,\n\t\t\t\t\tlogger log.Logger,\n\t\t\t\t) error {\n\t\t\t\t\treturn fmt.Errorf(\"test error\")\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"test error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockReleaseFn := func(error) {}\n\n\t\t\t// Create the replicator using createTestHistoryReplicator\n\t\t\treplicator := createTestHistoryReplicator(t, uuid.New())\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\ttest.mockApplyNonStartEventsWithContinueAsNewAffordance(&replicator)\n\t\t\ttest.mockApplyNonStartEventsWithoutContinueAsNewAffordance(&replicator)\n\n\t\t\t// Call the function under test\n\t\t\terr := applyNonStartEventsToNoneCurrentBranch(ctx.Background(), mockExecutionContext, mockMutableState, 1, mockReleaseFn, mockTask, &replicator, replicator.logger)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance               func(mockTask *MockreplicationTask)\n\t\tmockMutableStateAffordance       func(mockMutableState *execution.MockMutableState) *persistence.VersionHistories\n\t\tmockTransactionManagerAffordance func(mockTransactionManager *MocktransactionManager)\n\t\texpectError                      error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLastEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockMutableState *execution.MockMutableState) *persistence.VersionHistories {\n\t\t\t\t// Create a VersionHistory and VersionHistories structure\n\t\t\t\tversionHistory := &persistence.VersionHistory{\n\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories:                  []*persistence.VersionHistory{versionHistory},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).Times(1)\n\t\t\t\treturn versionHistories\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().backfillWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.WorkflowEvents{\n\t\t\t\t\t\tDomainID:    \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID:  \"test-workflow-id\",\n\t\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\t\tEvents:      nil,\n\t\t\t\t\t},\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: missing version histories\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLastEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockMutableState *execution.MockMutableState) *persistence.VersionHistories {\n\t\t\t\tmockMutableState.EXPECT().GetVersionHistories().Return(nil).Times(1)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\texpectError:                      execution.ErrMissingVersionHistories,\n\t\t},\n\t\t\"Case3: error when adding or updating version history item\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLastEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockMutableState *execution.MockMutableState) *persistence.VersionHistories {\n\t\t\t\t// Create a VersionHistory and VersionHistories structure with a higher event ID and version\n\t\t\t\tversionHistory := &persistence.VersionHistory{\n\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tEventID: 2, // Higher event ID\n\t\t\t\t\t\t\tVersion: 2, // Higher version\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories:                  []*persistence.VersionHistory{versionHistory},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).Times(1)\n\t\t\t\treturn versionHistories\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\texpectError:                      &types.BadRequestError{Message: \"cannot update version history with a lower version 1. Last version: 2\"},\n\t\t},\n\t\t\"Case4: error during backfill workflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLastEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      1,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t},\n\t\t\tmockMutableStateAffordance: func(mockMutableState *execution.MockMutableState) *persistence.VersionHistories {\n\t\t\t\tversionHistory := &persistence.VersionHistory{\n\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\tItems:       []*persistence.VersionHistoryItem{},\n\t\t\t\t}\n\t\t\t\tversionHistories := &persistence.VersionHistories{\n\t\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\t\tHistories:                  []*persistence.VersionHistory{versionHistory},\n\t\t\t\t}\n\n\t\t\t\tmockMutableState.EXPECT().GetVersionHistories().Return(versionHistories).Times(1)\n\t\t\t\treturn versionHistories\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().backfillWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.WorkflowEvents{\n\t\t\t\t\t\tDomainID:    \"test-domain-id\",\n\t\t\t\t\t\tWorkflowID:  \"test-workflow-id\",\n\t\t\t\t\t\tRunID:       \"test-run-id\",\n\t\t\t\t\t\tBranchToken: []byte(\"branch-token\"),\n\t\t\t\t\t\tEvents:      nil,\n\t\t\t\t\t},\n\t\t\t\t).Return(fmt.Errorf(\"backfill error\")).Times(1)\n\t\t\t},\n\t\t\texpectError: fmt.Errorf(\"backfill error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockTransactionManager := NewMocktransactionManager(ctrl)\n\t\t\tmockReleaseFn := func(error) {}\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockMutableStateAffordance(mockMutableState)\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\ttest.mockTransactionManagerAffordance(mockTransactionManager)\n\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t// Call the function under test\n\t\t\terr := applyNonStartEventsToNoneCurrentBranchWithoutContinueAsNew(\n\t\t\t\tctx.Background(),\n\t\t\t\tmockExecutionContext,\n\t\t\t\tmockMutableState,\n\t\t\t\t0, // Ensure branchIndex is valid\n\t\t\t\tmockReleaseFn,\n\t\t\t\tmockTask,\n\t\t\t\tmockTransactionManager,\n\t\t\t\tcluster.Metadata{},\n\t\t\t\tmockShard,\n\t\t\t\ttestlogger.New(t),\n\t\t\t)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsMissingMutableState(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance             func(mockTask *MockreplicationTask)\n\t\tmockWorkflowResetterAffordance func(mockWorkflowResetter *MockWorkflowResetter)\n\t\tmockNewWorkflowResetterFn      func(mockTask *MockreplicationTask, mockWorkflowResetter *MockWorkflowResetter) newWorkflowResetterFn\n\t\texpectMutableState             execution.MutableState\n\t\tvalidateError                  func(t *testing.T, err error)\n\t}{\n\t\t\"Case1: non-reset workflow, should return retry task error\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().isWorkflowReset().Return(false).Times(1)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      10,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getWorkflowID().Return(\"test-workflow-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getRunID().Return(\"test-run-id\").Times(1)\n\t\t\t},\n\t\t\tmockWorkflowResetterAffordance: func(mockWorkflowResetter *MockWorkflowResetter) {},\n\t\t\tmockNewWorkflowResetterFn: func(mockTask *MockreplicationTask, mockWorkflowResetter *MockWorkflowResetter) newWorkflowResetterFn {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\texpectMutableState: nil,\n\t\t\tvalidateError: func(t *testing.T, err error) {\n\t\t\t\tassert.IsType(t, &types.RetryTaskV2Error{}, err)\n\t\t\t\tretryError := err.(*types.RetryTaskV2Error)\n\t\t\t\tassert.Equal(t, \"Resend events due to missing mutable state\", retryError.Message)\n\t\t\t\tassert.Equal(t, \"test-domain-id\", retryError.DomainID)\n\t\t\t\tassert.Equal(t, \"test-workflow-id\", retryError.WorkflowID)\n\t\t\t\tassert.Equal(t, \"test-run-id\", retryError.RunID)\n\t\t\t},\n\t\t},\n\t\t\"Case2: reset workflow, should return reset mutable state\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().isWorkflowReset().Return(true).Times(1)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      10,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getWorkflowResetMetadata().Return(\"base-run-id\", \"new-run-id\", int64(1), false).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getWorkflowID().Return(\"test-workflow-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t\tmockTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockWorkflowResetterAffordance: func(mockWorkflowResetter *MockWorkflowResetter) {\n\t\t\t\tmockMutableState := execution.NewMockMutableState(gomock.NewController(t)) // Ensure consistent MutableState\n\t\t\t\tmockWorkflowResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tint64(9),\n\t\t\t\t\tint64(1),\n\t\t\t\t\tint64(10),\n\t\t\t\t\tint64(1),\n\t\t\t\t).Return(mockMutableState, nil).Times(1)\n\t\t\t},\n\t\t\tmockNewWorkflowResetterFn: func(mockTask *MockreplicationTask, mockWorkflowResetter *MockWorkflowResetter) newWorkflowResetterFn {\n\t\t\t\t// Return the already created mockWorkflowResetter\n\t\t\t\treturn func(domainID, workflowID, baseRunID string, newContext execution.Context, newRunID string, logger log.Logger) WorkflowResetter {\n\t\t\t\t\treturn mockWorkflowResetter\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectMutableState: execution.NewMockMutableState(gomock.NewController(t)), // Match returned value\n\t\t\tvalidateError:      func(t *testing.T, err error) { assert.NoError(t, err) },\n\t\t},\n\t\t\"Case3: error during workflow reset\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().isWorkflowReset().Return(true).Times(1)\n\t\t\t\tmockTask.EXPECT().getFirstEvent().Return(&types.HistoryEvent{\n\t\t\t\t\tID:      10,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t}).Times(2)\n\t\t\t\tmockTask.EXPECT().getWorkflowResetMetadata().Return(\"base-run-id\", \"new-run-id\", int64(1), false).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getWorkflowID().Return(\"test-workflow-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t\tmockTask.EXPECT().getVersion().Return(int64(1)).Times(1)\n\t\t\t},\n\t\t\tmockWorkflowResetterAffordance: func(mockWorkflowResetter *MockWorkflowResetter) {\n\t\t\t\tmockWorkflowResetter.EXPECT().ResetWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tint64(9),\n\t\t\t\t\tint64(1),\n\t\t\t\t\tint64(10),\n\t\t\t\t\tint64(1),\n\t\t\t\t).Return(nil, fmt.Errorf(\"reset error\")).Times(1)\n\t\t\t},\n\t\t\tmockNewWorkflowResetterFn: func(mockTask *MockreplicationTask, mockWorkflowResetter *MockWorkflowResetter) newWorkflowResetterFn {\n\t\t\t\t// Return the already created mockWorkflowResetter\n\t\t\t\treturn func(domainID, workflowID, baseRunID string, newContext execution.Context, newRunID string, logger log.Logger) WorkflowResetter {\n\t\t\t\t\treturn mockWorkflowResetter\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectMutableState: nil,\n\t\t\tvalidateError:      func(t *testing.T, err error) { assert.EqualError(t, err, \"reset error\") },\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockWorkflowResetter := NewMockWorkflowResetter(ctrl)\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\ttest.mockWorkflowResetterAffordance(mockWorkflowResetter)\n\t\t\tnewWorkflowResetterFn := test.mockNewWorkflowResetterFn(mockTask, mockWorkflowResetter)\n\n\t\t\t// Call the function under test\n\t\t\tmutableState, err := applyNonStartEventsMissingMutableState(\n\t\t\t\tctx.Background(),\n\t\t\t\tmockExecutionContext,\n\t\t\t\tmockTask,\n\t\t\t\tnewWorkflowResetterFn,\n\t\t\t)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectMutableState, mutableState)\n\t\t\ttest.validateError(t, err)\n\t\t})\n\t}\n}\n\nfunc Test_applyNonStartEventsResetWorkflow(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance               func(mockTask *MockreplicationTask)\n\t\tmockStateBuilderAffordance       func(mockStateBuilder *execution.MockStateBuilder)\n\t\tmockTransactionManagerAffordance func(mockTransactionManager *MocktransactionManager)\n\t\tmockShardContextAffordance       func(mockShard *shard.MockContext)\n\t\texpectError                      error\n\t}{\n\t\t\"Case1: success case with no errors\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(1)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(2)\n\t\t\t\tmockTask.EXPECT().getSourceCluster().Return(\"test-source-cluster\").Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().createWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockShardContextAffordance: func(mockShard *shard.MockContext) {\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tNumberOfShards:                       0,\n\t\t\t\t\tIsAdvancedVisConfigExist:             false,\n\t\t\t\t\tMaxResponseSize:                      0,\n\t\t\t\t\tHistoryCacheInitialSize:              dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheMaxSize:                  dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tHistoryCacheTTL:                      dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tHostName:                             \"test-host\",\n\t\t\t\t\tStandbyClusterDelay:                  dynamicproperties.GetDurationPropertyFn(10),\n\t\t\t\t\tEnableSizeBasedHistoryExecutionCache: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t}).Times(1)\n\t\t\t\tmockShard.EXPECT().SetCurrentTime(gomock.Any(), gomock.Any()).Times(1)\n\t\t\t},\n\t\t\texpectError: nil,\n\t\t},\n\t\t\"Case2: error during ApplyEvents\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(nil, fmt.Errorf(\"applyEvents error\")).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {},\n\t\t\tmockShardContextAffordance:       func(mockShard *shard.MockContext) {},\n\t\t\texpectError:                      fmt.Errorf(\"applyEvents error\"),\n\t\t},\n\t\t\"Case3: error during createWorkflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *MockreplicationTask) {\n\t\t\t\tmockTask.EXPECT().getLogger().Return(log.NewNoop()).Times(2)\n\t\t\t\tmockTask.EXPECT().getDomainID().Return(\"test-domain-id\").Times(1)\n\t\t\t\tmockTask.EXPECT().getExecution().Return(&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}).Times(1)\n\t\t\t\tmockTask.EXPECT().getEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getNewEvents().Return(nil).Times(1)\n\t\t\t\tmockTask.EXPECT().getEventTime().Return(time.Now()).Times(1)\n\t\t\t},\n\t\t\tmockStateBuilderAffordance: func(mockStateBuilder *execution.MockStateBuilder) {\n\t\t\t\tmockStateBuilder.EXPECT().ApplyEvents(\n\t\t\t\t\t\"test-domain-id\",\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\tmockTransactionManagerAffordance: func(mockTransactionManager *MocktransactionManager) {\n\t\t\t\tmockTransactionManager.EXPECT().createWorkflow(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(fmt.Errorf(\"createWorkflow error\")).Times(1)\n\t\t\t},\n\t\t\tmockShardContextAffordance: func(mockShard *shard.MockContext) {},\n\t\t\texpectError:                fmt.Errorf(\"createWorkflow error\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Mock objects\n\t\t\tmockTask := NewMockreplicationTask(ctrl)\n\t\t\tmockExecutionContext := execution.NewMockContext(ctrl)\n\t\t\tmockMutableState := execution.NewMockMutableState(ctrl)\n\t\t\tmockStateBuilder := execution.NewMockStateBuilder(ctrl)\n\t\t\tmockTransactionManager := NewMocktransactionManager(ctrl)\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tlogger := log.NewNoop()\n\n\t\t\t// Mock affordances\n\t\t\ttest.mockTaskAffordance(mockTask)\n\t\t\ttest.mockStateBuilderAffordance(mockStateBuilder)\n\t\t\ttest.mockTransactionManagerAffordance(mockTransactionManager)\n\t\t\ttest.mockShardContextAffordance(mockShard)\n\n\t\t\t// Mock functions\n\t\t\tmockNewStateBuilderFn := func(mutableState execution.MutableState, logger log.Logger) execution.StateBuilder {\n\t\t\t\treturn mockStateBuilder\n\t\t\t}\n\n\t\t\t// Call the function under test\n\t\t\terr := applyNonStartEventsResetWorkflow(\n\t\t\t\tctx.Background(),\n\t\t\t\tmockExecutionContext,\n\t\t\t\tmockMutableState,\n\t\t\t\tmockTask,\n\t\t\t\tmockNewStateBuilderFn,\n\t\t\t\tmockTransactionManager,\n\t\t\t\tcluster.Metadata{},\n\t\t\t\tlogger,\n\t\t\t\tmockShard,\n\t\t\t)\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectError, err)\n\t\t})\n\t}\n}\n\nfunc Test_notify(t *testing.T) {\n\ttests := map[string]struct {\n\t\tclusterName          string\n\t\tnow                  time.Time\n\t\tcurrentClusterName   string\n\t\tprimaryClusterName   string\n\t\texpectSetCurrentTime bool\n\t}{\n\t\t\"Case1: event from current cluster, should log a warning\": {\n\t\t\tclusterName:          \"current-cluster\",\n\t\t\tnow:                  time.Now(),\n\t\t\tcurrentClusterName:   \"current-cluster\",\n\t\t\tprimaryClusterName:   \"primary-cluster\",\n\t\t\texpectSetCurrentTime: false,\n\t\t},\n\t\t\"Case2: event from different cluster, should update shard time\": {\n\t\t\tclusterName:          \"other-cluster\",\n\t\t\tnow:                  time.Now(),\n\t\t\tcurrentClusterName:   \"current-cluster\",\n\t\t\tprimaryClusterName:   \"primary-cluster\",\n\t\t\texpectSetCurrentTime: true,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\t// Create ClusterInformation instances\n\t\t\tclusterGroup := map[string]commonConfig.ClusterInformation{\n\t\t\t\t\"current-cluster\": {\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 1,\n\t\t\t\t\tRPCName:                \"current-cluster-rpc\",\n\t\t\t\t\tRPCAddress:             \"127.0.0.1:8080\",\n\t\t\t\t\tRPCTransport:           \"tchannel\",\n\t\t\t\t},\n\t\t\t\t\"other-cluster\": {\n\t\t\t\t\tEnabled:                true,\n\t\t\t\t\tInitialFailoverVersion: 2,\n\t\t\t\t\tRPCName:                \"other-cluster-rpc\",\n\t\t\t\t\tRPCAddress:             \"127.0.0.1:8081\",\n\t\t\t\t\tRPCTransport:           \"grpc\",\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Create Metadata instance\n\t\t\tclusterMetadata := cluster.NewMetadata(\n\t\t\t\tcommonConfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 1,\n\t\t\t\t\tPrimaryClusterName:       test.primaryClusterName,\n\t\t\t\t\tCurrentClusterName:       test.currentClusterName,\n\t\t\t\t\tClusterGroup:             clusterGroup,\n\t\t\t\t},\n\t\t\t\tdynamicproperties.GetBoolPropertyFnFilteredByDomain(false),\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)\n\n\t\t\t// Mock Shard Context\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tif test.expectSetCurrentTime {\n\t\t\t\tmockShard.EXPECT().GetConfig().Return(&config.Config{\n\t\t\t\t\tStandbyClusterDelay: dynamicproperties.GetDurationPropertyFn(5 * time.Minute),\n\t\t\t\t}).Times(1)\n\t\t\t\tmockShard.EXPECT().SetCurrentTime(test.clusterName, gomock.Any()).Times(1)\n\t\t\t}\n\n\t\t\t// Use Noop logger\n\t\t\tlogger := log.NewNoop()\n\n\t\t\t// Call the function under test\n\t\t\tnotify(\n\t\t\t\ttest.clusterName,\n\t\t\t\ttest.now,\n\t\t\t\tlogger,\n\t\t\t\tmockShard,\n\t\t\t\tclusterMetadata,\n\t\t\t)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/ndc/new_workflow_transaction_mamanger_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: new_workflow_transaction_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source new_workflow_transaction_manager.go -destination new_workflow_transaction_mamanger_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MocktransactionManagerForNewWorkflow is a mock of transactionManagerForNewWorkflow interface.\ntype MocktransactionManagerForNewWorkflow struct {\n\tctrl     *gomock.Controller\n\trecorder *MocktransactionManagerForNewWorkflowMockRecorder\n\tisgomock struct{}\n}\n\n// MocktransactionManagerForNewWorkflowMockRecorder is the mock recorder for MocktransactionManagerForNewWorkflow.\ntype MocktransactionManagerForNewWorkflowMockRecorder struct {\n\tmock *MocktransactionManagerForNewWorkflow\n}\n\n// NewMocktransactionManagerForNewWorkflow creates a new mock instance.\nfunc NewMocktransactionManagerForNewWorkflow(ctrl *gomock.Controller) *MocktransactionManagerForNewWorkflow {\n\tmock := &MocktransactionManagerForNewWorkflow{ctrl: ctrl}\n\tmock.recorder = &MocktransactionManagerForNewWorkflowMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MocktransactionManagerForNewWorkflow) EXPECT() *MocktransactionManagerForNewWorkflowMockRecorder {\n\treturn m.recorder\n}\n\n// dispatchForNewWorkflow mocks base method.\nfunc (m *MocktransactionManagerForNewWorkflow) dispatchForNewWorkflow(ctx context.Context, now time.Time, targetWorkflow execution.Workflow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"dispatchForNewWorkflow\", ctx, now, targetWorkflow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// dispatchForNewWorkflow indicates an expected call of dispatchForNewWorkflow.\nfunc (mr *MocktransactionManagerForNewWorkflowMockRecorder) dispatchForNewWorkflow(ctx, now, targetWorkflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"dispatchForNewWorkflow\", reflect.TypeOf((*MocktransactionManagerForNewWorkflow)(nil).dispatchForNewWorkflow), ctx, now, targetWorkflow)\n}\n"
  },
  {
    "path": "service/history/ndc/new_workflow_transaction_manager.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination new_workflow_transaction_mamanger_mock.go\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\ttransactionManagerForNewWorkflow interface {\n\t\tdispatchForNewWorkflow(\n\t\t\tctx ctx.Context,\n\t\t\tnow time.Time,\n\t\t\ttargetWorkflow execution.Workflow,\n\t\t) error\n\t}\n\n\ttransactionManagerForNewWorkflowImpl struct {\n\t\ttransactionManager transactionManager\n\t\tlogger             log.Logger\n\t}\n)\n\nvar _ transactionManagerForNewWorkflow = (*transactionManagerForNewWorkflowImpl)(nil)\n\nfunc newTransactionManagerForNewWorkflow(\n\ttransactionManager transactionManager,\n\tlogger log.Logger,\n) transactionManagerForNewWorkflow {\n\n\treturn &transactionManagerForNewWorkflowImpl{\n\t\ttransactionManager: transactionManager,\n\t\tlogger:             logger,\n\t}\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) dispatchForNewWorkflow(\n\tctx ctx.Context,\n\tnow time.Time,\n\ttargetWorkflow execution.Workflow,\n) error {\n\t// NOTE: this function does NOT mutate current workflow or target workflow,\n\t//  workflow mutation is done in methods within executeTransaction function\n\n\ttargetExecutionInfo := targetWorkflow.GetMutableState().GetExecutionInfo()\n\tdomainID := targetExecutionInfo.DomainID\n\tworkflowID := targetExecutionInfo.WorkflowID\n\ttargetRunID := targetExecutionInfo.RunID\n\n\t// we need to check the current workflow execution\n\tcurrentRunID, err := r.transactionManager.getCurrentWorkflowRunID(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t)\n\tif err != nil || currentRunID == targetRunID {\n\t\t// error out or workflow already created\n\t\treturn err\n\t}\n\n\tif currentRunID == \"\" {\n\t\t// current record does not exists\n\t\treturn r.executeTransaction(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttransactionPolicyCreateAsCurrent,\n\t\t\tnil,\n\t\t\ttargetWorkflow,\n\t\t)\n\t}\n\n\t// there exists a current workflow, need additional check\n\tcurrentWorkflow, err := r.transactionManager.loadNDCWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tcurrentRunID,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttargetWorkflowIsNewer, err := targetWorkflow.HappensAfter(currentWorkflow)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif !targetWorkflowIsNewer {\n\t\t// target workflow is older than current workflow, need to suppress the target workflow\n\t\treturn r.executeTransaction(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttransactionPolicyCreateAsZombie,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t)\n\t}\n\n\t// target workflow is newer than current workflow\n\tif !currentWorkflow.GetMutableState().IsWorkflowExecutionRunning() {\n\t\t// current workflow is completed\n\t\t// proceed to create workflow\n\t\treturn r.executeTransaction(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\ttransactionPolicyCreateAsCurrent,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t)\n\t}\n\n\t// current workflow is still running, need to suppress the current workflow\n\treturn r.executeTransaction(\n\t\tctx,\n\t\tnow,\n\t\ttransactionPolicySuppressCurrentAndCreateAsCurrent,\n\t\tcurrentWorkflow,\n\t\ttargetWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) createAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n) error {\n\n\ttargetWorkflowSnapshot, targetWorkflowEventsSeq, err := targetWorkflow.GetMutableState().CloseTransactionAsSnapshot(\n\t\tnow,\n\t\texecution.TransactionPolicyPassive,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar targetWorkflowHistoryBlob events.PersistedBlob\n\tif len(targetWorkflowEventsSeq[0].Events) > 0 {\n\t\tif targetWorkflowEventsSeq[0].Events[0].GetEventType() == types.EventTypeWorkflowExecutionStarted {\n\t\t\ttargetWorkflowHistoryBlob, err = targetWorkflow.GetContext().PersistStartWorkflowBatchEvents(\n\t\t\t\tctx,\n\t\t\t\ttargetWorkflowEventsSeq[0],\n\t\t\t)\n\t\t} else { // reset workflows fall into else branch\n\t\t\ttargetWorkflowHistoryBlob, err = targetWorkflow.GetContext().PersistNonStartWorkflowBatchEvents(\n\t\t\t\tctx,\n\t\t\t\ttargetWorkflowEventsSeq[0],\n\t\t\t)\n\t\t}\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// target workflow to be created as current\n\tif currentWorkflow != nil {\n\t\t// current workflow exists, need to do compare and swap\n\t\tcreateMode := persistence.CreateWorkflowModeWorkflowIDReuse\n\t\tprevRunID := currentWorkflow.GetMutableState().GetExecutionInfo().RunID\n\t\tprevVectorClock, err := currentWorkflow.GetVectorClock()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn targetWorkflow.GetContext().CreateWorkflowExecution(\n\t\t\tctx,\n\t\t\ttargetWorkflowSnapshot,\n\t\t\ttargetWorkflowHistoryBlob,\n\t\t\tcreateMode,\n\t\t\tprevRunID,\n\t\t\tprevVectorClock.LastWriteVersion,\n\t\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t\t)\n\t}\n\n\t// current workflow does not exists, create as brand new\n\tcreateMode := persistence.CreateWorkflowModeBrandNew\n\tprevRunID := \"\"\n\tprevLastWriteVersion := int64(0)\n\treturn targetWorkflow.GetContext().CreateWorkflowExecution(\n\t\tctx,\n\t\ttargetWorkflowSnapshot,\n\t\ttargetWorkflowHistoryBlob,\n\t\tcreateMode,\n\t\tprevRunID,\n\t\tprevLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) createAsZombie(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n) error {\n\n\ttargetWorkflowPolicy, err := targetWorkflow.SuppressBy(\n\t\tcurrentWorkflow,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif targetWorkflowPolicy != execution.TransactionPolicyPassive {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"nDCTransactionManagerForNewWorkflow createAsZombie encounter target workflow policy not being passive\",\n\t\t}\n\t}\n\n\ttargetWorkflowSnapshot, targetWorkflowEventsSeq, err := targetWorkflow.GetMutableState().CloseTransactionAsSnapshot(\n\t\tnow,\n\t\ttargetWorkflowPolicy,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tvar targetWorkflowHistoryBlob events.PersistedBlob\n\tif len(targetWorkflowEventsSeq[0].Events) > 0 {\n\t\tif targetWorkflowEventsSeq[0].Events[0].GetEventType() == types.EventTypeWorkflowExecutionStarted {\n\t\t\ttargetWorkflowHistoryBlob, err = targetWorkflow.GetContext().PersistStartWorkflowBatchEvents(\n\t\t\t\tctx,\n\t\t\t\ttargetWorkflowEventsSeq[0],\n\t\t\t)\n\t\t} else { // reset workflows fall into else branch\n\t\t\ttargetWorkflowHistoryBlob, err = targetWorkflow.GetContext().PersistNonStartWorkflowBatchEvents(\n\t\t\t\tctx,\n\t\t\t\ttargetWorkflowEventsSeq[0],\n\t\t\t)\n\t\t}\n\t}\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// release lock on current workflow, since current cluster maybe the active cluster\n\t// and events maybe reapplied to current workflow\n\t// TODO: add functional test for this case.\n\tcurrentWorkflow.GetReleaseFn()(nil)\n\tcurrentWorkflow = nil\n\n\tif err := targetWorkflow.GetContext().ReapplyEvents(\n\t\ttargetWorkflowEventsSeq,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tcreateMode := persistence.CreateWorkflowModeZombie\n\tprevRunID := \"\"\n\tprevLastWriteVersion := int64(0)\n\terr = targetWorkflow.GetContext().CreateWorkflowExecution(\n\t\tctx,\n\t\ttargetWorkflowSnapshot,\n\t\ttargetWorkflowHistoryBlob,\n\t\tcreateMode,\n\t\tprevRunID,\n\t\tprevLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn nil\n\tcase *persistence.WorkflowExecutionAlreadyStartedError:\n\t\t// workflow already created\n\t\treturn nil\n\tdefault:\n\t\treturn err\n\t}\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) suppressCurrentAndCreateAsCurrent(\n\tctx ctx.Context,\n\tnow time.Time,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n) error {\n\n\tcurrentWorkflowPolicy, err := currentWorkflow.SuppressBy(\n\t\ttargetWorkflow,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif err := targetWorkflow.Revive(); err != nil {\n\t\treturn err\n\t}\n\n\tr.logger.Debugf(\"suppressCurrentAndCreateAsCurrent calling UpdateWorkflowExecutionWithNew for wfID %s, current policy %v, new policy %v\",\n\t\tcurrentWorkflow.GetMutableState().GetExecutionInfo().WorkflowID,\n\t\tcurrentWorkflowPolicy,\n\t\texecution.TransactionPolicyPassive,\n\t)\n\treturn currentWorkflow.GetContext().UpdateWorkflowExecutionWithNew(\n\t\tctx,\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeUpdateCurrent,\n\t\ttargetWorkflow.GetContext(),\n\t\ttargetWorkflow.GetMutableState(),\n\t\tcurrentWorkflowPolicy,\n\t\texecution.TransactionPolicyPassive.Ptr(),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) executeTransaction(\n\tctx ctx.Context,\n\tnow time.Time,\n\ttransactionPolicy transactionPolicy,\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n) (retError error) {\n\n\tdefer func() {\n\t\tif rec := recover(); rec != nil {\n\t\t\tr.cleanupTransaction(currentWorkflow, targetWorkflow, errPanic)\n\t\t\tpanic(rec)\n\t\t} else {\n\t\t\tr.cleanupTransaction(currentWorkflow, targetWorkflow, retError)\n\t\t}\n\t}()\n\n\tswitch transactionPolicy {\n\tcase transactionPolicyCreateAsCurrent:\n\t\treturn r.createAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t)\n\n\tcase transactionPolicyCreateAsZombie:\n\t\treturn r.createAsZombie(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t)\n\n\tcase transactionPolicySuppressCurrentAndCreateAsCurrent:\n\t\treturn r.suppressCurrentAndCreateAsCurrent(\n\t\t\tctx,\n\t\t\tnow,\n\t\t\tcurrentWorkflow,\n\t\t\ttargetWorkflow,\n\t\t)\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"nDCTransactionManager: encounter unknown transaction type: %v\", transactionPolicy),\n\t\t}\n\t}\n}\n\nfunc (r *transactionManagerForNewWorkflowImpl) cleanupTransaction(\n\tcurrentWorkflow execution.Workflow,\n\ttargetWorkflow execution.Workflow,\n\terr error,\n) {\n\n\tif currentWorkflow != nil {\n\t\tcurrentWorkflow.GetReleaseFn()(err)\n\t}\n\tif targetWorkflow != nil {\n\t\ttargetWorkflow.GetReleaseFn()(err)\n\t}\n}\n"
  },
  {
    "path": "service/history/ndc/new_workflow_transaction_manager_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\ttransactionManagerForNewWorkflowSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller             *gomock.Controller\n\t\tmockTransactionManager *MocktransactionManager\n\n\t\tcreateManager *transactionManagerForNewWorkflowImpl\n\t}\n)\n\nfunc TestTransactionManagerForNewWorkflowSuite(t *testing.T) {\n\ts := new(transactionManagerForNewWorkflowSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockTransactionManager = NewMocktransactionManager(s.controller)\n\n\ts.createManager = newTransactionManagerForNewWorkflow(\n\t\ts.mockTransactionManager,\n\t\ttestlogger.New(s.T()),\n\t).(*transactionManagerForNewWorkflowImpl)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_Dup() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(runID, nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, workflow)\n\ts.NoError(err)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_BrandNew() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\tworkflowSnapshot := &persistence.WorkflowSnapshot{}\n\tworkflowStartedType := types.EventTypeWorkflowExecutionStarted\n\tworkflowEventsSeq := []*persistence.WorkflowEvents{\n\t\t&persistence.WorkflowEvents{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: &workflowStartedType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tworkflowHistory := events.PersistedBlob{DataBlob: persistence.DataBlob{Data: make([]byte, 12345)}}\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\tmutableState.EXPECT().CloseTransactionAsSnapshot(now, execution.TransactionPolicyPassive).Return(\n\t\tworkflowSnapshot, workflowEventsSeq, nil,\n\t).Times(1)\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(\n\t\tctx, domainID, workflowID,\n\t).Return(\"\", nil).Times(1)\n\n\tcontext.EXPECT().PersistStartWorkflowBatchEvents(\n\t\tgomock.Any(),\n\t\tworkflowEventsSeq[0],\n\t).Return(workflowHistory, nil).Times(1)\n\tcontext.EXPECT().CreateWorkflowExecution(\n\t\tgomock.Any(),\n\t\tworkflowSnapshot,\n\t\tworkflowHistory,\n\t\tpersistence.CreateWorkflowModeBrandNew,\n\t\t\"\",\n\t\tint64(0),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, workflow)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_CreateAsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\tcurrentLastWriteVersion := int64(4321)\n\n\ttargetReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetWorkflowSnapshot := &persistence.WorkflowSnapshot{}\n\tworkflowNonStartedType := types.EventTypeDecisionTaskScheduled // non workflow started event\n\ttargetWorkflowEventsSeq := []*persistence.WorkflowEvents{\n\t\t&persistence.WorkflowEvents{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: &workflowNonStartedType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ttargetWorkflowHistory := events.PersistedBlob{DataBlob: persistence.DataBlob{Data: make([]byte, 12345)}}\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ttargetMutableState.EXPECT().CloseTransactionAsSnapshot(now, execution.TransactionPolicyPassive).Return(\n\t\ttargetWorkflowSnapshot, targetWorkflowEventsSeq, nil,\n\t).Times(1)\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionManager.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(true, nil)\n\tcurrentMutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tcurrentMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      currentRunID,\n\t}).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetVectorClock().Return(execution.WorkflowVectorClock{\n\t\tLastWriteVersion: currentLastWriteVersion,\n\t}, nil)\n\n\ttargetContext.EXPECT().PersistNonStartWorkflowBatchEvents(\n\t\tgomock.Any(),\n\t\ttargetWorkflowEventsSeq[0],\n\t).Return(targetWorkflowHistory, nil).Times(1)\n\ttargetContext.EXPECT().CreateWorkflowExecution(\n\t\tgomock.Any(),\n\t\ttargetWorkflowSnapshot,\n\t\ttargetWorkflowHistory,\n\t\tpersistence.CreateWorkflowModeWorkflowIDReuse,\n\t\tcurrentRunID,\n\t\tcurrentLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, targetWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_CreateAsZombie() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\ttargetReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetWorkflowSnapshot := &persistence.WorkflowSnapshot{\n\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t},\n\t}\n\tworkflowStartedType := types.EventTypeWorkflowExecutionStarted\n\ttargetWorkflowEventsSeq := []*persistence.WorkflowEvents{\n\t\t&persistence.WorkflowEvents{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: &workflowStartedType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ttargetWorkflowHistory := events.PersistedBlob{DataBlob: persistence.DataBlob{Data: make([]byte, 12345)}}\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ttargetMutableState.EXPECT().CloseTransactionAsSnapshot(now, execution.TransactionPolicyPassive).Return(\n\t\ttargetWorkflowSnapshot, targetWorkflowEventsSeq, nil,\n\t).Times(1)\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionManager.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().PersistStartWorkflowBatchEvents(\n\t\tgomock.Any(),\n\t\ttargetWorkflowEventsSeq[0],\n\t).Return(targetWorkflowHistory, nil).Times(1)\n\ttargetContext.EXPECT().CreateWorkflowExecution(\n\t\tgomock.Any(),\n\t\ttargetWorkflowSnapshot,\n\t\ttargetWorkflowHistory,\n\t\tpersistence.CreateWorkflowModeZombie,\n\t\t\"\",\n\t\tint64(0),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\ttargetContext.EXPECT().ReapplyEvents(targetWorkflowEventsSeq).Return(nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, targetWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_CreateAsZombie_Dedup() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\ttargetReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetWorkflowSnapshot := &persistence.WorkflowSnapshot{\n\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t},\n\t}\n\tworkflowNonStartedType := types.EventTypeDecisionTaskScheduled // non workflow started event\n\ttargetWorkflowEventsSeq := []*persistence.WorkflowEvents{\n\t\t&persistence.WorkflowEvents{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: &workflowNonStartedType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ttargetWorkflowHistory := events.PersistedBlob{DataBlob: persistence.DataBlob{Data: make([]byte, 12345)}}\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\ttargetMutableState.EXPECT().CloseTransactionAsSnapshot(now, execution.TransactionPolicyPassive).Return(\n\t\ttargetWorkflowSnapshot, targetWorkflowEventsSeq, nil,\n\t).Times(1)\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionManager.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(false, nil)\n\ttargetWorkflow.EXPECT().SuppressBy(currentWorkflow).Return(execution.TransactionPolicyPassive, nil).Times(1)\n\n\ttargetContext.EXPECT().PersistNonStartWorkflowBatchEvents(\n\t\tgomock.Any(),\n\t\ttargetWorkflowEventsSeq[0],\n\t).Return(targetWorkflowHistory, nil).Times(1)\n\ttargetContext.EXPECT().CreateWorkflowExecution(\n\t\tgomock.Any(),\n\t\ttargetWorkflowSnapshot,\n\t\ttargetWorkflowHistory,\n\t\tpersistence.CreateWorkflowModeZombie,\n\t\t\"\",\n\t\tint64(0),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(&persistence.WorkflowExecutionAlreadyStartedError{}).Times(1)\n\ttargetContext.EXPECT().ReapplyEvents(targetWorkflowEventsSeq).Return(nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, targetWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n\nfunc (s *transactionManagerForNewWorkflowSuite) TestDispatchForNewWorkflow_SuppressCurrentAndCreateAsCurrent() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\ttargetRunID := \"some random run ID\"\n\tcurrentRunID := \"other random runID\"\n\n\ttargetReleaseCalled := false\n\tcurrentReleaseCalled := false\n\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\ttargetContext := execution.NewMockContext(s.controller)\n\ttargetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { targetReleaseCalled = true }\n\ttargetWorkflow.EXPECT().GetContext().Return(targetContext).AnyTimes()\n\ttargetWorkflow.EXPECT().GetMutableState().Return(targetMutableState).AnyTimes()\n\ttargetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tcurrentMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t}).AnyTimes()\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\ttargetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      targetRunID,\n\t}).AnyTimes()\n\n\ts.mockTransactionManager.EXPECT().getCurrentWorkflowRunID(ctx, domainID, workflowID).Return(currentRunID, nil).Times(1)\n\ts.mockTransactionManager.EXPECT().loadNDCWorkflow(ctx, domainID, workflowID, currentRunID).Return(currentWorkflow, nil).Times(1)\n\n\ttargetWorkflow.EXPECT().HappensAfter(currentWorkflow).Return(true, nil)\n\tcurrentMutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\tcurrentWorkflowPolicy := execution.TransactionPolicyActive\n\tcurrentWorkflow.EXPECT().SuppressBy(targetWorkflow).Return(currentWorkflowPolicy, nil).Times(1)\n\ttargetWorkflow.EXPECT().Revive().Return(nil).Times(1)\n\n\tcurrentContext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(),\n\t\tnow,\n\t\tpersistence.UpdateWorkflowModeUpdateCurrent,\n\t\ttargetContext,\n\t\ttargetMutableState,\n\t\tcurrentWorkflowPolicy,\n\t\texecution.TransactionPolicyPassive.Ptr(),\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.createManager.dispatchForNewWorkflow(ctx, now, targetWorkflow)\n\ts.NoError(err)\n\ts.True(targetReleaseCalled)\n\ts.True(currentReleaseCalled)\n}\n"
  },
  {
    "path": "service/history/ndc/replication_task.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination replication_task_mock.go -self_package github.com/uber/cadence/service/history/ndc\n\npackage ndc\n\nimport (\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\treplicationTask interface {\n\t\tgetDomainID() string\n\t\tgetExecution() *types.WorkflowExecution\n\t\tgetWorkflowID() string\n\t\tgetRunID() string\n\t\tgetEventTime() time.Time\n\t\tgetFirstEvent() *types.HistoryEvent\n\t\tgetLastEvent() *types.HistoryEvent\n\t\tgetVersion() int64\n\t\tgetSourceCluster() string\n\t\tgetEvents() []*types.HistoryEvent\n\t\tgetNewEvents() []*types.HistoryEvent\n\t\tgetLogger() log.Logger\n\t\tgetVersionHistory() *persistence.VersionHistory\n\t\tisWorkflowReset() bool\n\t\tgetWorkflowResetMetadata() (string, string, int64, bool)\n\n\t\tsplitTask(taskStartTime time.Time) (replicationTask, replicationTask, error)\n\t}\n\n\treplicationTaskImpl struct {\n\t\tsourceCluster  string\n\t\tdomainID       string\n\t\texecution      *types.WorkflowExecution\n\t\tversion        int64\n\t\tfirstEvent     *types.HistoryEvent\n\t\tlastEvent      *types.HistoryEvent\n\t\teventTime      time.Time\n\t\tevents         []*types.HistoryEvent\n\t\tnewEvents      []*types.HistoryEvent\n\t\tversionHistory *persistence.VersionHistory\n\n\t\tstartTime time.Time\n\t\tlogger    log.Logger\n\t}\n)\n\nvar (\n\t// ErrInvalidDomainID is returned if domain ID is invalid\n\tErrInvalidDomainID = &types.BadRequestError{Message: \"invalid domain ID\"}\n\t// ErrInvalidExecution is returned if execution is invalid\n\tErrInvalidExecution = &types.BadRequestError{Message: \"invalid execution\"}\n\t// ErrInvalidRunID is returned if run ID is invalid\n\tErrInvalidRunID = &types.BadRequestError{Message: \"invalid run ID\"}\n\t// ErrEventIDMismatch is returned if event ID mis-matched\n\tErrEventIDMismatch = &types.BadRequestError{Message: \"event ID mismatch\"}\n\t// ErrEventVersionMismatch is returned if event version mis-matched\n\tErrEventVersionMismatch = &types.BadRequestError{Message: \"event version mismatch\"}\n\t// ErrNoNewRunHistory is returned if there is no new run history\n\tErrNoNewRunHistory = &types.BadRequestError{Message: \"no new run history events\"}\n\t// ErrLastEventIsNotContinueAsNew is returned if the last event is not continue as new\n\tErrLastEventIsNotContinueAsNew = &types.BadRequestError{Message: \"last event is not continue as new\"}\n\t// ErrEmptyHistoryRawEventBatch indicate that one single batch of history raw events is of size 0\n\tErrEmptyHistoryRawEventBatch = &types.BadRequestError{Message: \"encounter empty history batch\"}\n)\n\nfunc newReplicationTask(\n\tclusterMetadata cluster.Metadata,\n\thistorySerializer persistence.PayloadSerializer,\n\ttaskStartTime time.Time,\n\tlogger log.Logger,\n\trequest *types.ReplicateEventsV2Request,\n) (replicationTask, error) {\n\n\tevents, newEvents, err := validateReplicateEventsRequest(\n\t\thistorySerializer,\n\t\trequest,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainID := request.GetDomainUUID()\n\texecution := request.WorkflowExecution\n\tversionHistory := &types.VersionHistory{\n\t\tBranchToken: nil,\n\t\tItems:       request.VersionHistoryItems,\n\t}\n\n\tfirstEvent := events[0]\n\tlastEvent := events[len(events)-1]\n\tversion := firstEvent.Version\n\n\tsourceCluster, err := clusterMetadata.ClusterNameForFailoverVersion(version)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\teventTime := int64(0)\n\tfor _, event := range events {\n\t\tif event.GetTimestamp() > eventTime {\n\t\t\teventTime = event.GetTimestamp()\n\t\t}\n\t}\n\tfor _, event := range newEvents {\n\t\tif event.GetTimestamp() > eventTime {\n\t\t\teventTime = event.GetTimestamp()\n\t\t}\n\t}\n\n\tlogger = logger.WithTags(\n\t\ttag.WorkflowID(execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(execution.GetRunID()),\n\t\ttag.SourceCluster(sourceCluster),\n\t\ttag.IncomingVersion(version),\n\t\ttag.WorkflowFirstEventID(firstEvent.ID),\n\t\ttag.WorkflowNextEventID(lastEvent.ID+1),\n\t)\n\n\treturn &replicationTaskImpl{\n\t\tsourceCluster:  sourceCluster,\n\t\tdomainID:       domainID,\n\t\texecution:      execution,\n\t\tversion:        version,\n\t\tfirstEvent:     firstEvent,\n\t\tlastEvent:      lastEvent,\n\t\teventTime:      time.Unix(0, eventTime),\n\t\tevents:         events,\n\t\tnewEvents:      newEvents,\n\t\tversionHistory: persistence.NewVersionHistoryFromInternalType(versionHistory),\n\n\t\tstartTime: taskStartTime,\n\t\tlogger:    logger,\n\t}, nil\n}\n\nfunc (t *replicationTaskImpl) getDomainID() string {\n\treturn t.domainID\n}\n\nfunc (t *replicationTaskImpl) getExecution() *types.WorkflowExecution {\n\treturn t.execution\n}\n\nfunc (t *replicationTaskImpl) getWorkflowID() string {\n\treturn t.execution.GetWorkflowID()\n}\n\nfunc (t *replicationTaskImpl) getRunID() string {\n\treturn t.execution.GetRunID()\n}\n\nfunc (t *replicationTaskImpl) getEventTime() time.Time {\n\treturn t.eventTime\n}\n\nfunc (t *replicationTaskImpl) getFirstEvent() *types.HistoryEvent {\n\treturn t.firstEvent\n}\n\nfunc (t *replicationTaskImpl) getLastEvent() *types.HistoryEvent {\n\treturn t.lastEvent\n}\n\nfunc (t *replicationTaskImpl) getVersion() int64 {\n\treturn t.version\n}\n\nfunc (t *replicationTaskImpl) getSourceCluster() string {\n\treturn t.sourceCluster\n}\n\nfunc (t *replicationTaskImpl) getEvents() []*types.HistoryEvent {\n\treturn t.events\n}\n\nfunc (t *replicationTaskImpl) getNewEvents() []*types.HistoryEvent {\n\treturn t.newEvents\n}\n\nfunc (t *replicationTaskImpl) getLogger() log.Logger {\n\treturn t.logger\n}\n\nfunc (t *replicationTaskImpl) getVersionHistory() *persistence.VersionHistory {\n\treturn t.versionHistory\n}\n\nfunc (t *replicationTaskImpl) isWorkflowReset() bool {\n\n\tbaseRunID, newRunID, baseEventVersion, isReset := t.getWorkflowResetMetadata()\n\treturn len(baseRunID) > 0 && baseEventVersion != constants.EmptyVersion && len(newRunID) > 0 && isReset\n}\n\nfunc (t *replicationTaskImpl) getWorkflowResetMetadata() (string, string, int64, bool) {\n\n\tvar baseRunID string\n\tvar newRunID string\n\tvar baseEventVersion = constants.EmptyVersion\n\tvar isReset bool\n\tswitch t.getFirstEvent().GetEventType() {\n\tcase types.EventTypeDecisionTaskFailed:\n\t\tdecisionTaskFailedEvent := t.getFirstEvent()\n\t\tattr := decisionTaskFailedEvent.GetDecisionTaskFailedEventAttributes()\n\t\tbaseRunID = attr.GetBaseRunID()\n\t\tbaseEventVersion = attr.GetForkEventVersion()\n\t\tnewRunID = attr.GetNewRunID()\n\t\tif attr.GetCause() == types.DecisionTaskFailedCauseResetWorkflow {\n\t\t\tisReset = true\n\t\t}\n\n\tcase types.EventTypeDecisionTaskTimedOut:\n\t\tdecisionTaskTimedOutEvent := t.getFirstEvent()\n\t\tattr := decisionTaskTimedOutEvent.GetDecisionTaskTimedOutEventAttributes()\n\n\t\tbaseRunID = attr.GetBaseRunID()\n\t\tbaseEventVersion = attr.GetForkEventVersion()\n\t\tnewRunID = attr.GetNewRunID()\n\t\tif attr.GetCause() == types.DecisionTaskTimedOutCauseReset {\n\t\t\tisReset = true\n\t\t}\n\t}\n\treturn baseRunID, newRunID, baseEventVersion, isReset\n}\n\nfunc (t *replicationTaskImpl) splitTask(\n\ttaskStartTime time.Time,\n) (replicationTask, replicationTask, error) {\n\n\tif len(t.newEvents) == 0 {\n\t\treturn nil, nil, ErrNoNewRunHistory\n\t}\n\tnewHistoryEvents := t.newEvents\n\n\tif t.getLastEvent().GetEventType() != types.EventTypeWorkflowExecutionContinuedAsNew ||\n\t\tt.getLastEvent().WorkflowExecutionContinuedAsNewEventAttributes == nil {\n\t\treturn nil, nil, ErrLastEventIsNotContinueAsNew\n\t}\n\tnewRunID := t.getLastEvent().WorkflowExecutionContinuedAsNewEventAttributes.GetNewExecutionRunID()\n\n\tnewFirstEvent := newHistoryEvents[0]\n\tnewLastEvent := newHistoryEvents[len(newHistoryEvents)-1]\n\n\tnewEventTime := int64(0)\n\tfor _, event := range newHistoryEvents {\n\t\tif event.GetTimestamp() > newEventTime {\n\t\t\tnewEventTime = event.GetTimestamp()\n\t\t}\n\t}\n\n\tnewVersionHistory := persistence.NewVersionHistoryFromInternalType(&types.VersionHistory{\n\t\tBranchToken: nil,\n\t\tItems: []*types.VersionHistoryItem{{\n\t\t\tEventID: newLastEvent.ID,\n\t\t\tVersion: newLastEvent.Version,\n\t\t}},\n\t})\n\n\tlogger := t.logger.WithTags(\n\t\ttag.WorkflowID(t.getExecution().GetWorkflowID()),\n\t\ttag.WorkflowRunID(newRunID),\n\t\ttag.SourceCluster(t.sourceCluster),\n\t\ttag.IncomingVersion(t.version),\n\t\ttag.WorkflowFirstEventID(newFirstEvent.ID),\n\t\ttag.WorkflowNextEventID(newLastEvent.ID+1),\n\t)\n\n\tnewRunTask := &replicationTaskImpl{\n\t\tsourceCluster: t.sourceCluster,\n\t\tdomainID:      t.domainID,\n\t\texecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: t.execution.WorkflowID,\n\t\t\tRunID:      newRunID,\n\t\t},\n\t\tversion:        t.version,\n\t\tfirstEvent:     newFirstEvent,\n\t\tlastEvent:      newLastEvent,\n\t\teventTime:      time.Unix(0, newEventTime),\n\t\tevents:         newHistoryEvents,\n\t\tnewEvents:      []*types.HistoryEvent{},\n\t\tversionHistory: newVersionHistory,\n\n\t\tstartTime: taskStartTime,\n\t\tlogger:    logger,\n\t}\n\tt.newEvents = nil\n\n\treturn t, newRunTask, nil\n}\n\nfunc validateReplicateEventsRequest(\n\thistorySerializer persistence.PayloadSerializer,\n\trequest *types.ReplicateEventsV2Request,\n) ([]*types.HistoryEvent, []*types.HistoryEvent, error) {\n\n\t// TODO add validation on version history\n\n\tif valid := validateUUID(request.GetDomainUUID()); !valid {\n\t\treturn nil, nil, ErrInvalidDomainID\n\t}\n\tif request.WorkflowExecution == nil {\n\t\treturn nil, nil, ErrInvalidExecution\n\t}\n\tif valid := validateUUID(request.WorkflowExecution.GetRunID()); !valid {\n\t\treturn nil, nil, ErrInvalidRunID\n\t}\n\n\tevents, err := deserializeBlob(historySerializer, request.Events)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tif len(events) == 0 {\n\t\treturn nil, nil, ErrEmptyHistoryRawEventBatch\n\t}\n\n\tversion, err := validateEvents(events)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tif request.NewRunEvents == nil {\n\t\treturn events, nil, nil\n\t}\n\n\tnewRunEvents, err := deserializeBlob(historySerializer, request.NewRunEvents)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tnewRunVersion, err := validateEvents(newRunEvents)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tif version != newRunVersion {\n\t\treturn nil, nil, ErrEventVersionMismatch\n\t}\n\treturn events, newRunEvents, nil\n}\n\nfunc validateUUID(input string) bool {\n\treturn uuid.Parse(input) != nil\n}\n\nfunc validateEvents(events []*types.HistoryEvent) (int64, error) {\n\n\tfirstEvent := events[0]\n\tfirstEventID := firstEvent.ID\n\tversion := firstEvent.Version\n\n\tfor index, event := range events {\n\t\tif event.ID != firstEventID+int64(index) {\n\t\t\treturn 0, ErrEventIDMismatch\n\t\t}\n\t\tif event.Version != version {\n\t\t\treturn 0, ErrEventVersionMismatch\n\t\t}\n\t}\n\treturn version, nil\n}\n\nfunc deserializeBlob(\n\thistorySerializer persistence.PayloadSerializer,\n\tblob *types.DataBlob,\n) ([]*types.HistoryEvent, error) {\n\n\tif blob == nil {\n\t\treturn nil, nil\n\t}\n\n\tinternalEvents, err := historySerializer.DeserializeBatchEvents(&persistence.DataBlob{\n\t\tEncoding: constants.EncodingTypeThriftRW,\n\t\tData:     blob.Data,\n\t})\n\treturn internalEvents, err\n}\n"
  },
  {
    "path": "service/history/ndc/replication_task_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: replication_task.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source replication_task.go -destination replication_task_mock.go -self_package github.com/uber/cadence/service/history/ndc\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tlog \"github.com/uber/cadence/common/log\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockreplicationTask is a mock of replicationTask interface.\ntype MockreplicationTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockreplicationTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockreplicationTaskMockRecorder is the mock recorder for MockreplicationTask.\ntype MockreplicationTaskMockRecorder struct {\n\tmock *MockreplicationTask\n}\n\n// NewMockreplicationTask creates a new mock instance.\nfunc NewMockreplicationTask(ctrl *gomock.Controller) *MockreplicationTask {\n\tmock := &MockreplicationTask{ctrl: ctrl}\n\tmock.recorder = &MockreplicationTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockreplicationTask) EXPECT() *MockreplicationTaskMockRecorder {\n\treturn m.recorder\n}\n\n// getDomainID mocks base method.\nfunc (m *MockreplicationTask) getDomainID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getDomainID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// getDomainID indicates an expected call of getDomainID.\nfunc (mr *MockreplicationTaskMockRecorder) getDomainID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getDomainID\", reflect.TypeOf((*MockreplicationTask)(nil).getDomainID))\n}\n\n// getEventTime mocks base method.\nfunc (m *MockreplicationTask) getEventTime() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getEventTime\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// getEventTime indicates an expected call of getEventTime.\nfunc (mr *MockreplicationTaskMockRecorder) getEventTime() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getEventTime\", reflect.TypeOf((*MockreplicationTask)(nil).getEventTime))\n}\n\n// getEvents mocks base method.\nfunc (m *MockreplicationTask) getEvents() []*types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getEvents\")\n\tret0, _ := ret[0].([]*types.HistoryEvent)\n\treturn ret0\n}\n\n// getEvents indicates an expected call of getEvents.\nfunc (mr *MockreplicationTaskMockRecorder) getEvents() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getEvents\", reflect.TypeOf((*MockreplicationTask)(nil).getEvents))\n}\n\n// getExecution mocks base method.\nfunc (m *MockreplicationTask) getExecution() *types.WorkflowExecution {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getExecution\")\n\tret0, _ := ret[0].(*types.WorkflowExecution)\n\treturn ret0\n}\n\n// getExecution indicates an expected call of getExecution.\nfunc (mr *MockreplicationTaskMockRecorder) getExecution() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getExecution\", reflect.TypeOf((*MockreplicationTask)(nil).getExecution))\n}\n\n// getFirstEvent mocks base method.\nfunc (m *MockreplicationTask) getFirstEvent() *types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getFirstEvent\")\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\treturn ret0\n}\n\n// getFirstEvent indicates an expected call of getFirstEvent.\nfunc (mr *MockreplicationTaskMockRecorder) getFirstEvent() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getFirstEvent\", reflect.TypeOf((*MockreplicationTask)(nil).getFirstEvent))\n}\n\n// getLastEvent mocks base method.\nfunc (m *MockreplicationTask) getLastEvent() *types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getLastEvent\")\n\tret0, _ := ret[0].(*types.HistoryEvent)\n\treturn ret0\n}\n\n// getLastEvent indicates an expected call of getLastEvent.\nfunc (mr *MockreplicationTaskMockRecorder) getLastEvent() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getLastEvent\", reflect.TypeOf((*MockreplicationTask)(nil).getLastEvent))\n}\n\n// getLogger mocks base method.\nfunc (m *MockreplicationTask) getLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// getLogger indicates an expected call of getLogger.\nfunc (mr *MockreplicationTaskMockRecorder) getLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getLogger\", reflect.TypeOf((*MockreplicationTask)(nil).getLogger))\n}\n\n// getNewEvents mocks base method.\nfunc (m *MockreplicationTask) getNewEvents() []*types.HistoryEvent {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getNewEvents\")\n\tret0, _ := ret[0].([]*types.HistoryEvent)\n\treturn ret0\n}\n\n// getNewEvents indicates an expected call of getNewEvents.\nfunc (mr *MockreplicationTaskMockRecorder) getNewEvents() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getNewEvents\", reflect.TypeOf((*MockreplicationTask)(nil).getNewEvents))\n}\n\n// getRunID mocks base method.\nfunc (m *MockreplicationTask) getRunID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getRunID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// getRunID indicates an expected call of getRunID.\nfunc (mr *MockreplicationTaskMockRecorder) getRunID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getRunID\", reflect.TypeOf((*MockreplicationTask)(nil).getRunID))\n}\n\n// getSourceCluster mocks base method.\nfunc (m *MockreplicationTask) getSourceCluster() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getSourceCluster\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// getSourceCluster indicates an expected call of getSourceCluster.\nfunc (mr *MockreplicationTaskMockRecorder) getSourceCluster() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getSourceCluster\", reflect.TypeOf((*MockreplicationTask)(nil).getSourceCluster))\n}\n\n// getVersion mocks base method.\nfunc (m *MockreplicationTask) getVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// getVersion indicates an expected call of getVersion.\nfunc (mr *MockreplicationTaskMockRecorder) getVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getVersion\", reflect.TypeOf((*MockreplicationTask)(nil).getVersion))\n}\n\n// getVersionHistory mocks base method.\nfunc (m *MockreplicationTask) getVersionHistory() *persistence.VersionHistory {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getVersionHistory\")\n\tret0, _ := ret[0].(*persistence.VersionHistory)\n\treturn ret0\n}\n\n// getVersionHistory indicates an expected call of getVersionHistory.\nfunc (mr *MockreplicationTaskMockRecorder) getVersionHistory() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getVersionHistory\", reflect.TypeOf((*MockreplicationTask)(nil).getVersionHistory))\n}\n\n// getWorkflowID mocks base method.\nfunc (m *MockreplicationTask) getWorkflowID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getWorkflowID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// getWorkflowID indicates an expected call of getWorkflowID.\nfunc (mr *MockreplicationTaskMockRecorder) getWorkflowID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getWorkflowID\", reflect.TypeOf((*MockreplicationTask)(nil).getWorkflowID))\n}\n\n// getWorkflowResetMetadata mocks base method.\nfunc (m *MockreplicationTask) getWorkflowResetMetadata() (string, string, int64, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getWorkflowResetMetadata\")\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(string)\n\tret2, _ := ret[2].(int64)\n\tret3, _ := ret[3].(bool)\n\treturn ret0, ret1, ret2, ret3\n}\n\n// getWorkflowResetMetadata indicates an expected call of getWorkflowResetMetadata.\nfunc (mr *MockreplicationTaskMockRecorder) getWorkflowResetMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getWorkflowResetMetadata\", reflect.TypeOf((*MockreplicationTask)(nil).getWorkflowResetMetadata))\n}\n\n// isWorkflowReset mocks base method.\nfunc (m *MockreplicationTask) isWorkflowReset() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"isWorkflowReset\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// isWorkflowReset indicates an expected call of isWorkflowReset.\nfunc (mr *MockreplicationTaskMockRecorder) isWorkflowReset() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"isWorkflowReset\", reflect.TypeOf((*MockreplicationTask)(nil).isWorkflowReset))\n}\n\n// splitTask mocks base method.\nfunc (m *MockreplicationTask) splitTask(taskStartTime time.Time) (replicationTask, replicationTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"splitTask\", taskStartTime)\n\tret0, _ := ret[0].(replicationTask)\n\tret1, _ := ret[1].(replicationTask)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// splitTask indicates an expected call of splitTask.\nfunc (mr *MockreplicationTaskMockRecorder) splitTask(taskStartTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"splitTask\", reflect.TypeOf((*MockreplicationTask)(nil).splitTask), taskStartTime)\n}\n"
  },
  {
    "path": "service/history/ndc/replication_task_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ndc\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestReplicationTaskResetEvent(t *testing.T) {\n\ttaskStartTime := time.Now()\n\tdomainID := uuid.New()\n\tworkflowID := uuid.New()\n\trunID := uuid.New()\n\tlogger := testlogger.New(t)\n\tclusterMetadata := cluster.GetTestClusterMetadata(true)\n\n\thistorySerializer := persistence.NewPayloadSerializer()\n\tversionHistoryItems := []*types.VersionHistoryItem{}\n\tversionHistoryItems = append(versionHistoryItems, persistence.NewVersionHistoryItem(3, 0).ToInternalType())\n\n\teventType := types.EventTypeDecisionTaskFailed\n\tresetCause := types.DecisionTaskFailedCauseResetWorkflow\n\tevent := &types.HistoryEvent{\n\t\tID:        3,\n\t\tEventType: &eventType,\n\t\tVersion:   0,\n\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\tBaseRunID:        uuid.New(),\n\t\t\tForkEventVersion: 0,\n\t\t\tNewRunID:         runID,\n\t\t\tCause:            &resetCause,\n\t\t},\n\t}\n\tevents := []*types.HistoryEvent{}\n\tevents = append(events, event)\n\teventsBlob, err := historySerializer.SerializeBatchEvents(events, constants.EncodingTypeThriftRW)\n\trequire.NoError(t, err)\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tVersionHistoryItems: versionHistoryItems,\n\t\tEvents:              eventsBlob.ToInternal(),\n\t\tNewRunEvents:        nil,\n\t}\n\n\ttask, err := newReplicationTask(clusterMetadata, historySerializer, taskStartTime, logger, request)\n\trequire.NoError(t, err)\n\tassert.True(t, task.isWorkflowReset())\n}\n\nfunc TestNewReplicationTask(t *testing.T) {\n\tencodingType1 := types.EncodingTypeJSON\n\tencodingType2 := types.EncodingTypeThriftRW\n\n\thistoryEvent1 := &types.HistoryEvent{\n\t\tVersion: 0,\n\t}\n\thistoryEvent2 := &types.HistoryEvent{\n\t\tVersion: 1,\n\t}\n\n\thistoryEvents := []*types.HistoryEvent{historyEvent1, historyEvent2}\n\tserializer := persistence.NewPayloadSerializer()\n\tserializedEvents, err := serializer.SerializeBatchEvents(historyEvents, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\n\thistoryEvent3 := &types.HistoryEvent{\n\t\tID:      0,\n\t\tVersion: 2,\n\t}\n\thistoryEvent4 := &types.HistoryEvent{\n\t\tID:      1,\n\t\tVersion: 3,\n\t}\n\n\thistoryEvents2 := []*types.HistoryEvent{historyEvent3}\n\tserializedEvents2, err := serializer.SerializeBatchEvents(historyEvents2, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\n\thistoryEvents3 := []*types.HistoryEvent{historyEvent3, historyEvent4}\n\tserializedEvents3, err := serializer.SerializeBatchEvents(historyEvents3, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\n\thistoryEvents4 := []*types.HistoryEvent{historyEvent4}\n\tserializedEvents4, err := serializer.SerializeBatchEvents(historyEvents4, constants.EncodingTypeThriftRW)\n\tassert.NoError(t, err)\n\n\ttests := map[string]struct {\n\t\trequest          *types.ReplicateEventsV2Request\n\t\teventsAffordance func() []*types.HistoryEvent\n\t\texpectedErrorMsg string\n\t}{\n\t\t\"Case1: empty case\": {\n\t\t\trequest:          &types.ReplicateEventsV2Request{},\n\t\t\texpectedErrorMsg: \"invalid domain ID\",\n\t\t},\n\t\t\"Case2: nil case\": {\n\t\t\trequest:          nil,\n\t\t\texpectedErrorMsg: \"invalid domain ID\",\n\t\t},\n\t\t\"Case3-1: fail case with invalid domain id\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents:              nil,\n\t\t\t\tNewRunEvents:        nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"invalid domain ID\",\n\t\t},\n\t\t\"Case3-2: fail case in validateReplicateEventsRequest with invalid run id\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents:              nil,\n\t\t\t\tNewRunEvents:        nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"invalid run ID\",\n\t\t},\n\t\t\"Case3-3: fail case in validateReplicateEventsRequest with invalid workflow execution\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID:          \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution:   nil,\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents:              nil,\n\t\t\t\tNewRunEvents:        nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"invalid execution\",\n\t\t},\n\t\t\"Case3-4: fail case in validateReplicateEventsRequest with event is empty\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"12345678-1234-5678-9012-123456789012\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents:              nil,\n\t\t\t\tNewRunEvents:        nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"encounter empty history batch\",\n\t\t},\n\t\t\"Case3-5: fail case in validateReplicateEventsRequest with DeserializeBatchEvents error\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"12345678-1234-5678-9012-123456789012\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType1,\n\t\t\t\t\tData:         []byte(\"test-data\"),\n\t\t\t\t},\n\t\t\t\tNewRunEvents: nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"cadence deserialization error: DeserializeBatchEvents encoding: \\\"thriftrw\\\", error: Invalid binary encoding version.\",\n\t\t},\n\t\t\"Case3-6: fail case in validateReplicateEventsRequest with event ID mismatch\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"12345678-1234-5678-9012-123456789012\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType2,\n\t\t\t\t\tData:         serializedEvents.Data,\n\t\t\t\t},\n\t\t\t\tNewRunEvents: nil,\n\t\t\t},\n\t\t\texpectedErrorMsg: \"event ID mismatch\",\n\t\t},\n\t\t\"Case3-7: fail case in validateReplicateEventsRequest with event version mismatch\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"12345678-1234-5678-9012-123456789012\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType2,\n\t\t\t\t\tData:         serializedEvents2.Data,\n\t\t\t\t},\n\t\t\t\tNewRunEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType2,\n\t\t\t\t\tData:         serializedEvents3.Data,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErrorMsg: \"event version mismatch\",\n\t\t},\n\t\t\"Case3-8: fail case in validateReplicateEventsRequest with ErrEventVersionMismatch\": {\n\t\t\trequest: &types.ReplicateEventsV2Request{\n\t\t\t\tDomainUUID: \"12345678-1234-5678-9012-123456789011\",\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"12345678-1234-5678-9012-123456789012\",\n\t\t\t\t},\n\t\t\t\tVersionHistoryItems: nil,\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType2,\n\t\t\t\t\tData:         serializedEvents2.Data,\n\t\t\t\t},\n\t\t\t\tNewRunEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: &encodingType2,\n\t\t\t\t\tData:         serializedEvents4.Data,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedErrorMsg: \"event version mismatch\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tdomainID := \"\"\n\t\t\tif test.request != nil {\n\t\t\t\tdomainID = test.request.DomainUUID\n\t\t\t}\n\t\t\treplicator := createTestHistoryReplicator(t, domainID)\n\t\t\t_, err := newReplicationTask(\n\t\t\t\treplicator.clusterMetadata,\n\t\t\t\treplicator.historySerializer,\n\t\t\t\ttime.Now(),\n\t\t\t\treplicator.logger,\n\t\t\t\ttest.request,\n\t\t\t)\n\t\t\tassert.Equal(t, test.expectedErrorMsg, err.Error())\n\t\t})\n\t}\n}\n\nfunc Test_getWorkflowResetMetadata(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance     func(mockTask *replicationTaskImpl)\n\t\texpectBaseRunID        string\n\t\texpectNewRunID         string\n\t\texpectBaseEventVersion int64\n\t\texpectIsReset          bool\n\t}{\n\t\t\"Case1: DecisionTaskFailed with reset workflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\tdecisionTaskFailedEvent := &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\t\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\t\t\tBaseRunID:        \"base-run-id\",\n\t\t\t\t\t\tForkEventVersion: 1,\n\t\t\t\t\t\tNewRunID:         \"new-run-id\",\n\t\t\t\t\t\tCause:            types.DecisionTaskFailedCauseResetWorkflow.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockTask.firstEvent = decisionTaskFailedEvent\n\t\t\t},\n\t\t\texpectBaseRunID:        \"base-run-id\",\n\t\t\texpectNewRunID:         \"new-run-id\",\n\t\t\texpectBaseEventVersion: 1,\n\t\t\texpectIsReset:          true,\n\t\t},\n\t\t\"Case2: DecisionTaskTimedOut with reset workflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\tdecisionTaskTimedOutEvent := &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\t\tBaseRunID:        \"base-run-id-timedout\",\n\t\t\t\t\t\tForkEventVersion: 2,\n\t\t\t\t\t\tNewRunID:         \"new-run-id-timedout\",\n\t\t\t\t\t\tCause:            types.DecisionTaskTimedOutCauseReset.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockTask.firstEvent = decisionTaskTimedOutEvent\n\t\t\t},\n\t\t\texpectBaseRunID:        \"base-run-id-timedout\",\n\t\t\texpectNewRunID:         \"new-run-id-timedout\",\n\t\t\texpectBaseEventVersion: 2,\n\t\t\texpectIsReset:          true,\n\t\t},\n\t\t\"Case3: DecisionTaskFailed without reset workflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\tdecisionTaskFailedEvent := &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskFailed.Ptr(),\n\t\t\t\t\tDecisionTaskFailedEventAttributes: &types.DecisionTaskFailedEventAttributes{\n\t\t\t\t\t\tBaseRunID:        \"base-run-id\",\n\t\t\t\t\t\tForkEventVersion: 1,\n\t\t\t\t\t\tNewRunID:         \"new-run-id\",\n\t\t\t\t\t\tCause:            types.DecisionTaskFailedCause.Ptr(0),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockTask.firstEvent = decisionTaskFailedEvent\n\t\t\t},\n\t\t\texpectBaseRunID:        \"base-run-id\",\n\t\t\texpectNewRunID:         \"new-run-id\",\n\t\t\texpectBaseEventVersion: 1,\n\t\t\texpectIsReset:          false,\n\t\t},\n\t\t\"Case4: DecisionTaskTimedOut without reset workflow\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\tdecisionTaskTimedOutEvent := &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\t\tBaseRunID:        \"base-run-id-timedout\",\n\t\t\t\t\t\tForkEventVersion: 2,\n\t\t\t\t\t\tNewRunID:         \"new-run-id-timedout\",\n\t\t\t\t\t\tCause:            types.DecisionTaskTimedOutCause.Ptr(0),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockTask.firstEvent = decisionTaskTimedOutEvent\n\t\t\t},\n\t\t\texpectBaseRunID:        \"base-run-id-timedout\",\n\t\t\texpectNewRunID:         \"new-run-id-timedout\",\n\t\t\texpectBaseEventVersion: 2,\n\t\t\texpectIsReset:          false,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Create a mock task\n\t\t\tmockTask := &replicationTaskImpl{}\n\n\t\t\t// Apply test case mock behavior\n\t\t\ttest.mockTaskAffordance(mockTask)\n\n\t\t\t// Call the method under test\n\t\t\tbaseRunID, newRunID, baseEventVersion, isReset := mockTask.getWorkflowResetMetadata()\n\n\t\t\t// Assertions\n\t\t\tassert.Equal(t, test.expectBaseRunID, baseRunID)\n\t\t\tassert.Equal(t, test.expectNewRunID, newRunID)\n\t\t\tassert.Equal(t, test.expectBaseEventVersion, baseEventVersion)\n\t\t\tassert.Equal(t, test.expectIsReset, isReset)\n\t\t})\n\t}\n}\n\nfunc Test_splitTask(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockTaskAffordance func(mockTask *replicationTaskImpl)\n\t\ttaskStartTime      time.Time\n\t\texpectedNewRunTask bool\n\t\texpectedError      error\n\t}{\n\t\t\"Case1: success case with valid newEvents and continuedAsNew event\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\ttimestamp1 := time.Now().UnixNano()\n\t\t\t\ttimestamp2 := time.Now().Add(1 * time.Second).UnixNano()\n\n\t\t\t\tmockTask.newEvents = []*types.HistoryEvent{\n\t\t\t\t\t{ID: 1, Version: 1, Timestamp: &timestamp1},\n\t\t\t\t\t{ID: 2, Version: 1, Timestamp: &timestamp2},\n\t\t\t\t}\n\t\t\t\tmockTask.firstEvent = &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\t\t\t\tNewExecutionRunID: \"new-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockTask.lastEvent = &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\t\t\t\tID:        2,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\t\t\t\tNewExecutionRunID: \"new-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\t// Initialize execution to avoid nil pointer error\n\t\t\t\tmockTask.execution = &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}\n\t\t\t\t// Assign a logger\n\t\t\t\tmockTask.logger = log.NewNoop()\n\t\t\t},\n\t\t\ttaskStartTime:      time.Now(),\n\t\t\texpectedNewRunTask: true,\n\t\t\texpectedError:      nil,\n\t\t},\n\t\t\"Case2: error when no newEvents\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\tmockTask.newEvents = nil\n\t\t\t\t// Initialize execution to avoid nil pointer error\n\t\t\t\tmockTask.execution = &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}\n\t\t\t\tmockTask.logger = log.NewNoop() // Assign a logger\n\t\t\t},\n\t\t\ttaskStartTime:      time.Now(),\n\t\t\texpectedNewRunTask: false,\n\t\t\texpectedError:      ErrNoNewRunHistory,\n\t\t},\n\t\t\"Case3: error when lastEvent is not continuedAsNew\": {\n\t\t\tmockTaskAffordance: func(mockTask *replicationTaskImpl) {\n\t\t\t\ttimestamp1 := time.Now().UnixNano()\n\n\t\t\t\tmockTask.newEvents = []*types.HistoryEvent{\n\t\t\t\t\t{ID: 1, Version: 1, Timestamp: &timestamp1},\n\t\t\t\t}\n\t\t\t\tmockTask.lastEvent = &types.HistoryEvent{\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(), // Not continuedAsNew\n\t\t\t\t\tID:        1,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t}\n\t\t\t\t// Initialize execution to avoid nil pointer error\n\t\t\t\tmockTask.execution = &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t}\n\t\t\t\tmockTask.logger = log.NewNoop() // Assign a logger\n\t\t\t},\n\t\t\ttaskStartTime:      time.Now(),\n\t\t\texpectedNewRunTask: false,\n\t\t\texpectedError:      ErrLastEventIsNotContinueAsNew,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Create a mock task\n\t\t\tmockTask := &replicationTaskImpl{}\n\n\t\t\t// Apply the mock behavior based on the test case\n\t\t\ttest.mockTaskAffordance(mockTask)\n\n\t\t\t// Call the method under test\n\t\t\t_, newRunTask, err := mockTask.splitTask(test.taskStartTime)\n\n\t\t\t// Assertions\n\t\t\tif test.expectedError != nil {\n\t\t\t\tassert.Equal(t, test.expectedError, err)\n\t\t\t\tassert.Nil(t, newRunTask)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, newRunTask)\n\t\t\t\tassert.Equal(t, \"new-run-id\", newRunTask.getRunID())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_replicationTaskImpl_Getters(t *testing.T) {\n\t// Setup some common values for testing\n\tworkflowExecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t}\n\thistoryEvents := []*types.HistoryEvent{\n\t\t{ID: 1, Version: 1},\n\t\t{ID: 2, Version: 1},\n\t}\n\tnewHistoryEvents := []*types.HistoryEvent{\n\t\t{ID: 3, Version: 1},\n\t\t{ID: 4, Version: 1},\n\t}\n\tlogger := log.NewNoop()\n\tversionHistory := persistence.NewVersionHistory(nil, []*persistence.VersionHistoryItem{\n\t\t{EventID: 1, Version: 1},\n\t})\n\n\ttask := &replicationTaskImpl{\n\t\tdomainID:       \"test-domain-id\",\n\t\texecution:      workflowExecution,\n\t\tversion:        123,\n\t\tsourceCluster:  \"test-cluster\",\n\t\teventTime:      time.Now(),\n\t\tevents:         historyEvents,\n\t\tnewEvents:      newHistoryEvents,\n\t\tlogger:         logger,\n\t\tversionHistory: versionHistory,\n\t}\n\n\ttests := map[string]struct {\n\t\ttestFunc       func() interface{}\n\t\texpectedResult interface{}\n\t}{\n\t\t\"getDomainID\": {\n\t\t\ttestFunc:       func() interface{} { return task.getDomainID() },\n\t\t\texpectedResult: \"test-domain-id\",\n\t\t},\n\t\t\"getWorkflowID\": {\n\t\t\ttestFunc:       func() interface{} { return task.getWorkflowID() },\n\t\t\texpectedResult: \"test-workflow-id\",\n\t\t},\n\t\t\"getRunID\": {\n\t\t\ttestFunc:       func() interface{} { return task.getRunID() },\n\t\t\texpectedResult: \"test-run-id\",\n\t\t},\n\t\t\"getEventTime\": {\n\t\t\ttestFunc:       func() interface{} { return task.getEventTime() },\n\t\t\texpectedResult: task.eventTime,\n\t\t},\n\t\t\"getVersion\": {\n\t\t\ttestFunc:       func() interface{} { return task.getVersion() },\n\t\t\texpectedResult: int64(123),\n\t\t},\n\t\t\"getSourceCluster\": {\n\t\t\ttestFunc:       func() interface{} { return task.getSourceCluster() },\n\t\t\texpectedResult: \"test-cluster\",\n\t\t},\n\t\t\"getEvents\": {\n\t\t\ttestFunc:       func() interface{} { return task.getEvents() },\n\t\t\texpectedResult: historyEvents,\n\t\t},\n\t\t\"getNewEvents\": {\n\t\t\ttestFunc:       func() interface{} { return task.getNewEvents() },\n\t\t\texpectedResult: newHistoryEvents,\n\t\t},\n\t\t\"getLogger\": {\n\t\t\ttestFunc:       func() interface{} { return task.getLogger() },\n\t\t\texpectedResult: logger,\n\t\t},\n\t\t\"getVersionHistory\": {\n\t\t\ttestFunc:       func() interface{} { return task.getVersionHistory() },\n\t\t\texpectedResult: versionHistory,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, test.expectedResult, test.testFunc())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/ndc/transaction_manager.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination transaction_manager_mock.go\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n// NOTE: terminology\n//\n// 1. currentWorkflow means current running / closed workflow\n//  pointed by the current record in DB\n//\n// 2. targetWorkflow means the workflow to be replicated\n//  pointed by the replication task\n//\n// 3. newWorkflow means the workflow to be replicated, as part of continue as new\n//  pointed by the replication task\n//\n// 4. if target workflow and current workflow are the same\n//  then target workflow is set, current workflow is nil\n//\n// 5. suppress a workflow means turn a workflow into a zombie\n//  or terminate a workflow\n\n// Cases to be handled by this file:\n//\n// create path (there will be only current branch)\n// 1. create as current\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyCreateAsCurrent\n// 2. create as zombie\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyCreateAsZombie\n//\n// create path (there will be only current branch) + suppress current\n// 1. create as current & suppress current\t\t\t\t\t\t\t\t\t\t-> transactionPolicySuppressCurrentAndCreateAsCurrent\n//\n// update to current branch path\n// 1. update as current\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyUpdateAsCurrent\n// 2. update as current & new created as current\t\t\t\t\t\t\t\t-> transactionPolicyUpdateAsCurrent\n// 3. update as zombie\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyUpdateAsZombie\n// 4. update as zombie & new created as zombie\t\t\t\t\t\t\t\t\t-> transactionPolicyUpdateAsZombie\n//\n// backfill to non current branch path\n// 1. backfill as is\n// 2. backfill as is & new created as zombie\n//\n// conflict resolve path\n// 1. conflict resolve as current\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyConflictResolveAsCurrent\n// 2. conflict resolve as zombie\t\t\t\t\t\t\t\t\t\t\t\t-> transactionPolicyConflictResolveAsZombie\n//\n// conflict resolve path + suppress current\n// 1. update from zombie to current & suppress current\t\t\t\t\t\t\t-> transactionPolicySuppressCurrentAndUpdateAsCurrent\n// 2. update from zombie to current & new created as current & suppress current\t-> transactionPolicySuppressCurrentAndUpdateAsCurrent\n\ntype transactionPolicy int\n\nconst (\n\ttransactionPolicyCreateAsCurrent transactionPolicy = iota\n\ttransactionPolicyCreateAsZombie\n\ttransactionPolicySuppressCurrentAndCreateAsCurrent\n\n\ttransactionPolicyUpdateAsCurrent\n\ttransactionPolicyUpdateAsZombie\n\n\ttransactionPolicyConflictResolveAsCurrent\n\ttransactionPolicyConflictResolveAsZombie\n\n\ttransactionPolicySuppressCurrentAndUpdateAsCurrent\n\n\t// EventsReapplicationResetWorkflowReason is the reason for reset workflow during reapplication\n\tEventsReapplicationResetWorkflowReason = \"events-reapplication\"\n)\n\ntype (\n\ttransactionManager interface {\n\t\tcreateWorkflow(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\ttargetWorkflow execution.Workflow,\n\t\t) error\n\t\tupdateWorkflow(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tisWorkflowRebuilt bool,\n\t\t\ttargetWorkflow execution.Workflow,\n\t\t\tnewWorkflow execution.Workflow,\n\t\t) error\n\t\tbackfillWorkflow(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\ttargetWorkflow execution.Workflow,\n\t\t\ttargetWorkflowEvents *persistence.WorkflowEvents,\n\t\t) error\n\n\t\tcheckWorkflowExists(\n\t\t\tctx context.Context,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\trunID string,\n\t\t) (bool, error)\n\t\tgetCurrentWorkflowRunID(\n\t\t\tctx context.Context,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t) (string, error)\n\t\tloadNDCWorkflow(\n\t\t\tctx context.Context,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\trunID string,\n\t\t) (execution.Workflow, error)\n\t}\n\n\ttransactionManagerImpl struct {\n\t\tshard                shard.Context\n\t\texecutionCache       execution.Cache\n\t\tclusterMetadata      cluster.Metadata\n\t\tactiveClusterManager activecluster.Manager\n\t\thistoryV2Manager     persistence.HistoryManager\n\t\tserializer           persistence.PayloadSerializer\n\t\tmetricsClient        metrics.Client\n\t\tworkflowResetter     reset.WorkflowResetter\n\t\teventsReapplier      EventsReapplier\n\t\tlogger               log.Logger\n\n\t\tcreateManager transactionManagerForNewWorkflow\n\t\tupdateManager transactionManagerForExistingWorkflow\n\t}\n)\n\nvar _ transactionManager = (*transactionManagerImpl)(nil)\n\nfunc newTransactionManager(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\teventsReapplier EventsReapplier,\n\tlogger log.Logger,\n) *transactionManagerImpl {\n\n\ttransactionManager := &transactionManagerImpl{\n\t\tshard:                shard,\n\t\texecutionCache:       executionCache,\n\t\tclusterMetadata:      shard.GetClusterMetadata(),\n\t\tactiveClusterManager: shard.GetActiveClusterManager(),\n\t\thistoryV2Manager:     shard.GetHistoryManager(),\n\t\tserializer:           shard.GetService().GetPayloadSerializer(),\n\t\tmetricsClient:        shard.GetMetricsClient(),\n\t\tworkflowResetter: reset.NewWorkflowResetter(\n\t\t\tshard,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t),\n\t\teventsReapplier: eventsReapplier,\n\t\tlogger:          logger.WithTags(tag.ComponentHistoryReplicator),\n\n\t\tcreateManager: nil,\n\t\tupdateManager: nil,\n\t}\n\ttransactionManager.createManager = newTransactionManagerForNewWorkflow(transactionManager, logger)\n\ttransactionManager.updateManager = newTransactionManagerForExistingWorkflow(transactionManager, logger)\n\treturn transactionManager\n}\n\nfunc (r *transactionManagerImpl) createWorkflow(\n\tctx context.Context,\n\tnow time.Time,\n\ttargetWorkflow execution.Workflow,\n) error {\n\n\treturn r.createManager.dispatchForNewWorkflow(\n\t\tctx,\n\t\tnow,\n\t\ttargetWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerImpl) updateWorkflow(\n\tctx context.Context,\n\tnow time.Time,\n\tisWorkflowRebuilt bool,\n\ttargetWorkflow execution.Workflow,\n\tnewWorkflow execution.Workflow,\n) error {\n\n\treturn r.updateManager.dispatchForExistingWorkflow(\n\t\tctx,\n\t\tnow,\n\t\tisWorkflowRebuilt,\n\t\ttargetWorkflow,\n\t\tnewWorkflow,\n\t)\n}\n\nfunc (r *transactionManagerImpl) backfillWorkflow(\n\tctx context.Context,\n\tnow time.Time,\n\ttargetWorkflow execution.Workflow,\n\ttargetWorkflowEvents *persistence.WorkflowEvents,\n) (retError error) {\n\n\tdefer func() {\n\t\tif rec := recover(); rec != nil {\n\t\t\ttargetWorkflow.GetReleaseFn()(errPanic)\n\t\t\tpanic(rec)\n\t\t} else {\n\t\t\ttargetWorkflow.GetReleaseFn()(retError)\n\t\t}\n\t}()\n\n\tif _, err := targetWorkflow.GetContext().PersistNonStartWorkflowBatchEvents(\n\t\tctx,\n\t\ttargetWorkflowEvents,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tupdateMode, transactionPolicy, err := r.backfillWorkflowEventsReapply(\n\t\tctx,\n\t\ttargetWorkflow,\n\t\ttargetWorkflowEvents,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tr.logger.Debugf(\"backfillWorkflowEventsReapply calling UpdateWorkflowExecutionWithNew for wfID %s, updateMode %v, current policy %v, new policy %v\",\n\t\ttargetWorkflow.GetMutableState().GetExecutionInfo().WorkflowID,\n\t\tupdateMode,\n\t\ttransactionPolicy,\n\t\tnil,\n\t)\n\n\treturn targetWorkflow.GetContext().UpdateWorkflowExecutionWithNew(\n\t\tctx,\n\t\tnow,\n\t\tupdateMode,\n\t\tnil,\n\t\tnil,\n\t\ttransactionPolicy,\n\t\tnil,\n\t\tpersistence.CreateWorkflowRequestModeReplicated,\n\t)\n}\n\nfunc (r *transactionManagerImpl) backfillWorkflowEventsReapply(\n\tctx context.Context,\n\ttargetWorkflow execution.Workflow,\n\ttargetWorkflowEvents *persistence.WorkflowEvents,\n) (persistence.UpdateWorkflowMode, execution.TransactionPolicy, error) {\n\n\tisCurrentWorkflow, err := r.isWorkflowCurrent(ctx, targetWorkflow)\n\tif err != nil {\n\t\treturn 0, execution.TransactionPolicyActive, err\n\t}\n\tisWorkflowRunning := targetWorkflow.GetMutableState().IsWorkflowExecutionRunning()\n\tactiveClusterSelectionPolicy := targetWorkflow.GetMutableState().GetExecutionInfo().ActiveClusterSelectionPolicy\n\tactiveClusterInfo, err := r.activeClusterManager.GetActiveClusterInfoByClusterAttribute(ctx, targetWorkflow.GetMutableState().GetExecutionInfo().DomainID, activeClusterSelectionPolicy.GetClusterAttribute())\n\tif err != nil {\n\t\treturn 0, execution.TransactionPolicyActive, err\n\t}\n\ttargetWorkflowActiveCluster := activeClusterInfo.ActiveClusterName\n\tcurrentCluster := r.clusterMetadata.GetCurrentClusterName()\n\tisActiveCluster := targetWorkflowActiveCluster == currentCluster\n\n\t// workflow events reapplication\n\t// we need to handle 3 cases\n\t// 1. target workflow is self & self being current & active\n\t//  a. workflow still running -> just reapply\n\t//  b. workflow closed -> reset current workflow & reapply\n\t// 2. anything not case 1 -> find the current & active workflow to reapply\n\n\t// case 1\n\tif isCurrentWorkflow && isActiveCluster {\n\t\t// case 1.a\n\t\tif isWorkflowRunning {\n\t\t\tif _, err := r.eventsReapplier.ReapplyEvents(\n\t\t\t\tctx,\n\t\t\t\ttargetWorkflow.GetMutableState(),\n\t\t\t\ttargetWorkflowEvents.Events,\n\t\t\t\ttargetWorkflow.GetMutableState().GetExecutionInfo().RunID,\n\t\t\t); err != nil {\n\t\t\t\treturn 0, execution.TransactionPolicyActive, err\n\t\t\t}\n\t\t\treturn persistence.UpdateWorkflowModeUpdateCurrent, execution.TransactionPolicyActive, nil\n\t\t}\n\n\t\t// case 1.b\n\t\t// need to reset target workflow (which is also the current workflow)\n\t\t// to accept events to be reapplied\n\t\tbaseMutableState := targetWorkflow.GetMutableState()\n\t\tdomainID := baseMutableState.GetExecutionInfo().DomainID\n\t\tworkflowID := baseMutableState.GetExecutionInfo().WorkflowID\n\t\tbaseRunID := baseMutableState.GetExecutionInfo().RunID\n\t\tresetRunID := uuid.New()\n\t\tbaseRebuildLastEventID := baseMutableState.GetPreviousStartedEventID()\n\n\t\t// TODO when https://github.com/uber/cadence/issues/2420 is finished, remove this block,\n\t\t//  since cannot reapply event to a finished workflow which had no decisions started\n\t\tif baseRebuildLastEventID == constants.EmptyEventID {\n\t\t\tr.logger.Warn(\"cannot reapply event to a finished workflow\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t)\n\t\t\tr.metricsClient.IncCounter(metrics.HistoryReapplyEventsScope, metrics.EventReapplySkippedCount)\n\t\t\treturn persistence.UpdateWorkflowModeBypassCurrent, execution.TransactionPolicyPassive, nil\n\t\t}\n\n\t\tbaseVersionHistories := baseMutableState.GetVersionHistories()\n\t\tif baseVersionHistories == nil {\n\t\t\treturn 0, execution.TransactionPolicyActive, execution.ErrMissingVersionHistories\n\t\t}\n\t\tbaseCurrentVersionHistory, err := baseVersionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn 0, execution.TransactionPolicyActive, err\n\t\t}\n\t\tbaseRebuildLastEventVersion, err := baseCurrentVersionHistory.GetEventVersion(baseRebuildLastEventID)\n\t\tif err != nil {\n\t\t\treturn 0, execution.TransactionPolicyActive, err\n\t\t}\n\t\tbaseCurrentBranchToken := baseCurrentVersionHistory.GetBranchToken()\n\t\tbaseNextEventID := baseMutableState.GetNextEventID()\n\n\t\tif err = r.workflowResetter.ResetWorkflow(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tbaseRunID,\n\t\t\tbaseCurrentBranchToken,\n\t\t\tbaseRebuildLastEventID,\n\t\t\tbaseRebuildLastEventVersion,\n\t\t\tbaseNextEventID,\n\t\t\tresetRunID,\n\t\t\tuuid.New(),\n\t\t\ttargetWorkflow,\n\t\t\tEventsReapplicationResetWorkflowReason,\n\t\t\ttargetWorkflowEvents.Events,\n\t\t\tfalse,\n\t\t); err != nil {\n\t\t\treturn 0, execution.TransactionPolicyActive, err\n\t\t}\n\t\t// after the reset of target workflow (current workflow) with additional events to be reapplied\n\t\t// target workflow is no longer the current workflow\n\t\treturn persistence.UpdateWorkflowModeBypassCurrent, execution.TransactionPolicyPassive, nil\n\t}\n\n\t// case 2\n\t//  find the current & active workflow to reapply\n\tif err := targetWorkflow.GetContext().ReapplyEvents(\n\t\t[]*persistence.WorkflowEvents{targetWorkflowEvents},\n\t); err != nil {\n\t\treturn 0, execution.TransactionPolicyActive, err\n\t}\n\n\tif isCurrentWorkflow {\n\t\treturn persistence.UpdateWorkflowModeUpdateCurrent, execution.TransactionPolicyPassive, nil\n\t}\n\treturn persistence.UpdateWorkflowModeBypassCurrent, execution.TransactionPolicyPassive, nil\n}\n\nfunc (r *transactionManagerImpl) checkWorkflowExists(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) (bool, error) {\n\tdomainName, errorDomainName := r.shard.GetDomainCache().GetDomainName(domainID)\n\tif errorDomainName != nil {\n\t\treturn false, errorDomainName\n\t}\n\t_, err := r.shard.GetWorkflowExecution(\n\t\tctx,\n\t\t&persistence.GetWorkflowExecutionRequest{\n\t\t\tDomainID: domainID,\n\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tDomainName: domainName,\n\t\t},\n\t)\n\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn true, nil\n\tcase *types.EntityNotExistsError:\n\t\treturn false, nil\n\tdefault:\n\t\treturn false, err\n\t}\n}\n\nfunc (r *transactionManagerImpl) getCurrentWorkflowRunID(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n) (string, error) {\n\n\tdomainName, errorDomainName := r.shard.GetDomainCache().GetDomainName(domainID)\n\tif errorDomainName != nil {\n\t\treturn \"\", errorDomainName\n\t}\n\tresp, err := r.shard.GetExecutionManager().GetCurrentExecution(\n\t\tctx,\n\t\t&persistence.GetCurrentExecutionRequest{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tDomainName: domainName,\n\t\t},\n\t)\n\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn resp.RunID, nil\n\tcase *types.EntityNotExistsError:\n\t\treturn \"\", nil\n\tdefault:\n\t\treturn \"\", err\n\t}\n}\n\nfunc (r *transactionManagerImpl) loadNDCWorkflow(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) (execution.Workflow, error) {\n\n\t// we need to check the current workflow execution\n\tcontext, release, err := r.executionCache.GetOrCreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmsBuilder, err := context.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\t// no matter what error happen, we need to retry\n\t\trelease(err)\n\t\treturn nil, err\n\t}\n\treturn execution.NewWorkflow(ctx, r.clusterMetadata, context, msBuilder, release, r.logger), nil\n}\n\nfunc (r *transactionManagerImpl) isWorkflowCurrent(\n\tctx context.Context,\n\ttargetWorkflow execution.Workflow,\n) (bool, error) {\n\n\t// since we are not rebuilding the mutable state (when doing backfill) then we\n\t// can trust the result from IsCurrentWorkflowGuaranteed\n\tif targetWorkflow.GetMutableState().IsCurrentWorkflowGuaranteed() {\n\t\treturn true, nil\n\t}\n\n\t// target workflow is not guaranteed to be current workflow, do additional check\n\texecutionInfo := targetWorkflow.GetMutableState().GetExecutionInfo()\n\tdomainID := executionInfo.DomainID\n\tworkflowID := executionInfo.WorkflowID\n\trunID := executionInfo.RunID\n\n\tcurrentRunID, err := r.getCurrentWorkflowRunID(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\treturn currentRunID == runID, nil\n}\n"
  },
  {
    "path": "service/history/ndc/transaction_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: transaction_manager.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source transaction_manager.go -destination transaction_manager_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MocktransactionManager is a mock of transactionManager interface.\ntype MocktransactionManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MocktransactionManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MocktransactionManagerMockRecorder is the mock recorder for MocktransactionManager.\ntype MocktransactionManagerMockRecorder struct {\n\tmock *MocktransactionManager\n}\n\n// NewMocktransactionManager creates a new mock instance.\nfunc NewMocktransactionManager(ctrl *gomock.Controller) *MocktransactionManager {\n\tmock := &MocktransactionManager{ctrl: ctrl}\n\tmock.recorder = &MocktransactionManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MocktransactionManager) EXPECT() *MocktransactionManagerMockRecorder {\n\treturn m.recorder\n}\n\n// backfillWorkflow mocks base method.\nfunc (m *MocktransactionManager) backfillWorkflow(ctx context.Context, now time.Time, targetWorkflow execution.Workflow, targetWorkflowEvents *persistence.WorkflowEvents) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"backfillWorkflow\", ctx, now, targetWorkflow, targetWorkflowEvents)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// backfillWorkflow indicates an expected call of backfillWorkflow.\nfunc (mr *MocktransactionManagerMockRecorder) backfillWorkflow(ctx, now, targetWorkflow, targetWorkflowEvents any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"backfillWorkflow\", reflect.TypeOf((*MocktransactionManager)(nil).backfillWorkflow), ctx, now, targetWorkflow, targetWorkflowEvents)\n}\n\n// checkWorkflowExists mocks base method.\nfunc (m *MocktransactionManager) checkWorkflowExists(ctx context.Context, domainID, workflowID, runID string) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"checkWorkflowExists\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// checkWorkflowExists indicates an expected call of checkWorkflowExists.\nfunc (mr *MocktransactionManagerMockRecorder) checkWorkflowExists(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"checkWorkflowExists\", reflect.TypeOf((*MocktransactionManager)(nil).checkWorkflowExists), ctx, domainID, workflowID, runID)\n}\n\n// createWorkflow mocks base method.\nfunc (m *MocktransactionManager) createWorkflow(ctx context.Context, now time.Time, targetWorkflow execution.Workflow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"createWorkflow\", ctx, now, targetWorkflow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// createWorkflow indicates an expected call of createWorkflow.\nfunc (mr *MocktransactionManagerMockRecorder) createWorkflow(ctx, now, targetWorkflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"createWorkflow\", reflect.TypeOf((*MocktransactionManager)(nil).createWorkflow), ctx, now, targetWorkflow)\n}\n\n// getCurrentWorkflowRunID mocks base method.\nfunc (m *MocktransactionManager) getCurrentWorkflowRunID(ctx context.Context, domainID, workflowID string) (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"getCurrentWorkflowRunID\", ctx, domainID, workflowID)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// getCurrentWorkflowRunID indicates an expected call of getCurrentWorkflowRunID.\nfunc (mr *MocktransactionManagerMockRecorder) getCurrentWorkflowRunID(ctx, domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"getCurrentWorkflowRunID\", reflect.TypeOf((*MocktransactionManager)(nil).getCurrentWorkflowRunID), ctx, domainID, workflowID)\n}\n\n// loadNDCWorkflow mocks base method.\nfunc (m *MocktransactionManager) loadNDCWorkflow(ctx context.Context, domainID, workflowID, runID string) (execution.Workflow, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"loadNDCWorkflow\", ctx, domainID, workflowID, runID)\n\tret0, _ := ret[0].(execution.Workflow)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// loadNDCWorkflow indicates an expected call of loadNDCWorkflow.\nfunc (mr *MocktransactionManagerMockRecorder) loadNDCWorkflow(ctx, domainID, workflowID, runID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"loadNDCWorkflow\", reflect.TypeOf((*MocktransactionManager)(nil).loadNDCWorkflow), ctx, domainID, workflowID, runID)\n}\n\n// updateWorkflow mocks base method.\nfunc (m *MocktransactionManager) updateWorkflow(ctx context.Context, now time.Time, isWorkflowRebuilt bool, targetWorkflow, newWorkflow execution.Workflow) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"updateWorkflow\", ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// updateWorkflow indicates an expected call of updateWorkflow.\nfunc (mr *MocktransactionManagerMockRecorder) updateWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"updateWorkflow\", reflect.TypeOf((*MocktransactionManager)(nil).updateWorkflow), ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n}\n"
  },
  {
    "path": "service/history/ndc/transaction_manager_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\ttransactionManagerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockShard            *shard.TestContext\n\t\tmockCreateManager    *MocktransactionManagerForNewWorkflow\n\t\tmockUpdateManager    *MocktransactionManagerForExistingWorkflow\n\t\tmockEventsReapplier  *MockEventsReapplier\n\t\tmockWorkflowResetter *reset.MockWorkflowResetter\n\n\t\tmockExecutionManager *mocks.ExecutionManager\n\n\t\tlogger      log.Logger\n\t\tdomainEntry *cache.DomainCacheEntry\n\n\t\ttransactionManager *transactionManagerImpl\n\t}\n)\n\nfunc TestTransactionManagerSuite(t *testing.T) {\n\ts := new(transactionManagerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transactionManagerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockCreateManager = NewMocktransactionManagerForNewWorkflow(s.controller)\n\ts.mockUpdateManager = NewMocktransactionManagerForExistingWorkflow(s.controller)\n\ts.mockEventsReapplier = NewMockEventsReapplier(s.controller)\n\ts.mockWorkflowResetter = reset.NewMockWorkflowResetter(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockExecutionManager = s.mockShard.Resource.ExecutionMgr\n\n\ts.logger = s.mockShard.GetLogger()\n\ts.domainEntry = constants.TestGlobalDomainEntry\n\n\t// Setup mock for GetActiveClusterInfoByClusterAttribute to avoid errors\n\tmockActiveClusterManager := s.mockShard.Resource.ActiveClusterMgr\n\tmockActiveClusterManager.EXPECT().GetActiveClusterInfoByClusterAttribute(\n\t\tgomock.Any(), gomock.Any(), gomock.Any(),\n\t).Return(&types.ActiveClusterInfo{\n\t\tActiveClusterName: s.mockShard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tFailoverVersion:   int64(0),\n\t}, nil).AnyTimes()\n\n\ts.transactionManager = newTransactionManager(s.mockShard, execution.NewCache(s.mockShard), s.mockEventsReapplier, s.logger)\n\ts.transactionManager.createManager = s.mockCreateManager\n\ts.transactionManager.updateManager = s.mockUpdateManager\n\ts.transactionManager.workflowResetter = s.mockWorkflowResetter\n}\n\nfunc (s *transactionManagerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *transactionManagerSuite) TestCreateWorkflow() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\n\ts.mockCreateManager.EXPECT().dispatchForNewWorkflow(\n\t\tctx, now, targetWorkflow,\n\t).Return(nil).Times(1)\n\n\terr := s.transactionManager.createWorkflow(ctx, now, targetWorkflow)\n\ts.NoError(err)\n}\n\nfunc (s *transactionManagerSuite) TestUpdateWorkflow() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\tisWorkflowRebuilt := true\n\ttargetWorkflow := execution.NewMockWorkflow(s.controller)\n\tnewWorkflow := execution.NewMockWorkflow(s.controller)\n\n\ts.mockUpdateManager.EXPECT().dispatchForExistingWorkflow(\n\t\tctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow,\n\t).Return(nil).Times(1)\n\n\terr := s.transactionManager.updateWorkflow(ctx, now, isWorkflowRebuilt, targetWorkflow, newWorkflow)\n\ts.NoError(err)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_CurrentWorkflow_Active_Open() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\treleaseCalled := false\n\trunID := uuid.New()\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{\n\t\tEvents: []*types.HistoryEvent{{ID: 1}},\n\t}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\ts.mockEventsReapplier.EXPECT().ReapplyEvents(ctx, mutableState, workflowEvents.Events, runID).Return(workflowEvents.Events, nil).Times(1)\n\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(true).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{RunID: runID}).AnyTimes()\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeUpdateCurrent, nil, nil, execution.TransactionPolicyActive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_CurrentWorkflow_Active_Closed() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tdomainName := \"some random domainName\"\n\tlastDecisionTaskStartedEventID := int64(9999)\n\tnextEventID := lastDecisionTaskStartedEventID * 2\n\tlastDecisionTaskStartedVersion := s.domainEntry.GetFailoverVersion()\n\tversionHistory := persistence.NewVersionHistory([]byte(\"branch token\"), []*persistence.VersionHistoryItem{\n\t\t{EventID: lastDecisionTaskStartedEventID, Version: lastDecisionTaskStartedVersion},\n\t})\n\thistories := persistence.NewVersionHistories(versionHistory)\n\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\tmutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\tmutableState.EXPECT().GetPreviousStartedEventID().Return(lastDecisionTaskStartedEventID).Times(1)\n\tmutableState.EXPECT().GetVersionHistories().Return(histories).Times(1)\n\n\ts.mockWorkflowResetter.EXPECT().ResetWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\trunID,\n\t\tversionHistory.GetBranchToken(),\n\t\tlastDecisionTaskStartedEventID,\n\t\tlastDecisionTaskStartedVersion,\n\t\tnextEventID,\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t\tworkflow,\n\t\tEventsReapplicationResetWorkflowReason,\n\t\tworkflowEvents.Events,\n\t\tfalse,\n\t).Return(nil).Times(1)\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(&persistence.GetCurrentExecutionResponse{RunID: runID}, nil).Once()\n\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeBypassCurrent, nil, nil, execution.TransactionPolicyPassive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_CurrentWorkflow_Passive_Open() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{\n\t\tEvents: []*types.HistoryEvent{{ID: 1}},\n\t}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\ts.transactionManager.clusterMetadata = cluster.TestPassiveClusterMetadata\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(true).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(true).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\tcontext.EXPECT().ReapplyEvents([]*persistence.WorkflowEvents{workflowEvents}).Times(1)\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeUpdateCurrent, nil, nil, execution.TransactionPolicyPassive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_CurrentWorkflow_Passive_Closed() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tdomainName := \"some random domainName\"\n\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\ts.transactionManager.clusterMetadata = cluster.TestPassiveClusterMetadata\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(&persistence.GetCurrentExecutionResponse{RunID: runID}, nil).Once()\n\tcontext.EXPECT().ReapplyEvents([]*persistence.WorkflowEvents{workflowEvents}).Times(1)\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeUpdateCurrent, nil, nil, execution.TransactionPolicyPassive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_NotCurrentWorkflow_Active() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tcurrentRunID := \"other random run ID\"\n\tdomainName := \"some random domainName\"\n\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{\n\t\tEvents: []*types.HistoryEvent{{\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t}},\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(&persistence.GetCurrentExecutionResponse{RunID: currentRunID}, nil).Once()\n\tcontext.EXPECT().ReapplyEvents([]*persistence.WorkflowEvents{workflowEvents}).Times(1)\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeBypassCurrent, nil, nil, execution.TransactionPolicyPassive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestBackfillWorkflow_NotCurrentWorkflow_Passive() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tcurrentRunID := \"other random run ID\"\n\tdomainName := \"some random domainName\"\n\treleaseCalled := false\n\n\tworkflow := execution.NewMockWorkflow(s.controller)\n\tcontext := execution.NewMockContext(s.controller)\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tvar releaseFn execution.ReleaseFunc = func(error) { releaseCalled = true }\n\n\tworkflowEvents := &persistence.WorkflowEvents{\n\t\tEvents: []*types.HistoryEvent{{\n\t\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t}},\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t}\n\n\tworkflow.EXPECT().GetContext().Return(context).AnyTimes()\n\tworkflow.EXPECT().GetMutableState().Return(mutableState).AnyTimes()\n\tworkflow.EXPECT().GetReleaseFn().Return(releaseFn).AnyTimes()\n\n\tmutableState.EXPECT().IsCurrentWorkflowGuaranteed().Return(false).AnyTimes()\n\tmutableState.EXPECT().IsWorkflowExecutionRunning().Return(false).AnyTimes()\n\tmutableState.EXPECT().GetDomainEntry().Return(s.domainEntry).AnyTimes()\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(&persistence.GetCurrentExecutionResponse{RunID: currentRunID}, nil).Once()\n\tcontext.EXPECT().ReapplyEvents([]*persistence.WorkflowEvents{workflowEvents}).Times(1)\n\tcontext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), workflowEvents).Return(events.PersistedBlob{}, nil).Times(1)\n\tcontext.EXPECT().UpdateWorkflowExecutionWithNew(\n\t\tgomock.Any(), now, persistence.UpdateWorkflowModeBypassCurrent, nil, nil, execution.TransactionPolicyPassive, (*execution.TransactionPolicy)(nil), persistence.CreateWorkflowRequestModeReplicated,\n\t).Return(nil).Times(1)\n\terr := s.transactionManager.backfillWorkflow(ctx, now, workflow, workflowEvents)\n\ts.NoError(err)\n\ts.True(releaseCalled)\n}\n\nfunc (s *transactionManagerSuite) TestCheckWorkflowExists_DoesNotExists() {\n\tctx := ctx.Background()\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tdomainName := \"some random domainName\"\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil)\n\ts.mockExecutionManager.On(\"GetWorkflowExecution\", mock.Anything, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: domainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tDomainName: domainName,\n\t\tRangeID:    1,\n\t}).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\texists, err := s.transactionManager.checkWorkflowExists(ctx, domainID, workflowID, runID)\n\ts.NoError(err)\n\ts.False(exists)\n}\n\nfunc (s *transactionManagerSuite) TestCheckWorkflowExists_DoesExists() {\n\tctx := ctx.Background()\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tdomainName := \"some random Domain Name\"\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetWorkflowExecution\", mock.Anything, &persistence.GetWorkflowExecutionRequest{\n\t\tDomainID: domainID,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tDomainName: domainName,\n\t\tRangeID:    1,\n\t}).Return(&persistence.GetWorkflowExecutionResponse{}, nil).Once()\n\n\texists, err := s.transactionManager.checkWorkflowExists(ctx, domainID, workflowID, runID)\n\ts.NoError(err)\n\ts.True(exists)\n}\n\nfunc (s *transactionManagerSuite) TestGetWorkflowCurrentRunID_Missing() {\n\tctx := ctx.Background()\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\tdomainName := \"some random domainName\"\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\tcurrentRunID, err := s.transactionManager.getCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(\"\", currentRunID)\n}\n\nfunc (s *transactionManagerSuite) TestGetWorkflowCurrentRunID_Exists() {\n\tctx := ctx.Background()\n\tdomainID := \"some random domain ID\"\n\tworkflowID := \"some random workflow ID\"\n\trunID := \"some random run ID\"\n\tdomainName := \"some random domainName\"\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\ts.mockExecutionManager.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tDomainName: domainName,\n\t}).Return(&persistence.GetCurrentExecutionResponse{RunID: runID}, nil).Once()\n\n\tcurrentRunID, err := s.transactionManager.getCurrentWorkflowRunID(ctx, domainID, workflowID)\n\ts.NoError(err)\n\ts.Equal(runID, currentRunID)\n}\n"
  },
  {
    "path": "service/history/ndc/workflow_resetter.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination workflow_resetter_mock.go\n\npackage ndc\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tresendOnResetWorkflowMessage = \"Resend events due to reset workflow\"\n)\n\ntype (\n\t// WorkflowResetter handles workflow reset for NDC\n\tWorkflowResetter interface {\n\t\tResetWorkflow(\n\t\t\tctx context.Context,\n\t\t\tnow time.Time,\n\t\t\tbaseLastEventID int64,\n\t\t\tbaseLastEventVersion int64,\n\t\t\tincomingFirstEventID int64,\n\t\t\tincomingFirstEventVersion int64,\n\t\t) (execution.MutableState, error)\n\t}\n\n\tworkflowResetterImpl struct {\n\t\tshard              shard.Context\n\t\ttransactionManager transactionManager\n\t\thistoryV2Manager   persistence.HistoryManager\n\t\tstateRebuilder     execution.StateRebuilder\n\n\t\tdomainID   string\n\t\tworkflowID string\n\t\tbaseRunID  string\n\t\tnewContext execution.Context\n\t\tnewRunID   string\n\n\t\tlogger log.Logger\n\t}\n)\n\nvar _ WorkflowResetter = (*workflowResetterImpl)(nil)\n\n// NewWorkflowResetter creates workflow resetter\nfunc NewWorkflowResetter(\n\tshard shard.Context,\n\ttransactionManager transactionManager,\n\tdomainID string,\n\tworkflowID string,\n\tbaseRunID string,\n\tnewContext execution.Context,\n\tnewRunID string,\n\tlogger log.Logger,\n) WorkflowResetter {\n\n\treturn &workflowResetterImpl{\n\t\tshard:              shard,\n\t\ttransactionManager: transactionManager,\n\t\thistoryV2Manager:   shard.GetHistoryManager(),\n\t\tstateRebuilder:     execution.NewStateRebuilder(shard, logger),\n\n\t\tdomainID:   domainID,\n\t\tworkflowID: workflowID,\n\t\tbaseRunID:  baseRunID,\n\t\tnewContext: newContext,\n\t\tnewRunID:   newRunID,\n\t\tlogger:     logger,\n\t}\n}\n\nfunc (r *workflowResetterImpl) ResetWorkflow(\n\tctx context.Context,\n\tnow time.Time,\n\tbaseLastEventID int64,\n\tbaseLastEventVersion int64,\n\tincomingFirstEventID int64,\n\tincomingFirstEventVersion int64,\n) (execution.MutableState, error) {\n\n\tbaseBranchToken, err := r.getBaseBranchToken(\n\t\tctx,\n\t\tbaseLastEventID,\n\t\tbaseLastEventVersion,\n\t\tincomingFirstEventID,\n\t\tincomingFirstEventVersion,\n\t)\n\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresetBranchTokenFn := func() ([]byte, error) {\n\t\tresetBranchToken, err := r.getResetBranchToken(ctx, baseBranchToken, baseLastEventID)\n\n\t\treturn resetBranchToken, err\n\t}\n\n\trequestID := uuid.New()\n\trebuildMutableState, rebuiltHistorySize, err := r.stateRebuilder.Rebuild(\n\t\tctx,\n\t\tnow,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\tr.domainID,\n\t\t\tr.workflowID,\n\t\t\tr.baseRunID,\n\t\t),\n\t\tbaseBranchToken,\n\t\tbaseLastEventID,\n\t\tbaseLastEventVersion,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\tr.domainID,\n\t\t\tr.workflowID,\n\t\t\tr.newRunID,\n\t\t),\n\t\tresetBranchTokenFn,\n\t\trequestID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tr.newContext.Clear()\n\tr.newContext.SetHistorySize(rebuiltHistorySize)\n\treturn rebuildMutableState, nil\n}\n\nfunc (r *workflowResetterImpl) getBaseBranchToken(\n\tctx context.Context,\n\tbaseLastEventID int64,\n\tbaseLastEventVersion int64,\n\tincomingFirstEventID int64,\n\tincomingFirstEventVersion int64,\n) (baseBranchToken []byte, retError error) {\n\n\tbaseWorkflow, err := r.transactionManager.loadNDCWorkflow(\n\t\tctx,\n\t\tr.domainID,\n\t\tr.workflowID,\n\t\tr.baseRunID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer func() {\n\t\tbaseWorkflow.GetReleaseFn()(retError)\n\t}()\n\n\tmutableState := baseWorkflow.GetMutableState()\n\tbranchToken, err := mutableState.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tbaseVersionHistories := mutableState.GetVersionHistories()\n\tif baseVersionHistories != nil {\n\n\t\t_, baseVersionHistory, err := baseVersionHistories.FindFirstVersionHistoryByItem(\n\t\t\tpersistence.NewVersionHistoryItem(baseLastEventID, baseLastEventVersion),\n\t\t)\n\t\tif err != nil {\n\t\t\t// the base event and incoming event are from different branch\n\t\t\t// only re-replicate the gap on the incoming branch\n\t\t\t// the base branch event will eventually arrived\n\t\t\treturn nil, newNDCRetryTaskErrorWithHint(\n\t\t\t\tresendOnResetWorkflowMessage,\n\t\t\t\tr.domainID,\n\t\t\t\tr.workflowID,\n\t\t\t\tr.newRunID,\n\t\t\t\tnil,\n\t\t\t\tnil,\n\t\t\t\tcommon.Int64Ptr(incomingFirstEventID),\n\t\t\t\tcommon.Int64Ptr(incomingFirstEventVersion),\n\t\t\t)\n\t\t}\n\n\t\tbranchToken = baseVersionHistory.GetBranchToken()\n\t}\n\treturn branchToken, nil\n}\n\nfunc (r *workflowResetterImpl) getResetBranchToken(\n\tctx context.Context,\n\tbaseBranchToken []byte,\n\tbaseLastEventID int64,\n) ([]byte, error) {\n\n\t// fork a new history branch\n\tshardID := r.shard.GetShardID()\n\tdomainName, err := r.shard.GetDomainCache().GetDomainName(r.domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := r.historyV2Manager.ForkHistoryBranch(\n\t\tctx,\n\t\t&persistence.ForkHistoryBranchRequest{\n\t\t\tForkBranchToken: baseBranchToken,\n\t\t\tForkNodeID:      baseLastEventID + 1,\n\t\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(r.domainID, r.workflowID, r.newRunID),\n\t\t\tShardID:         common.IntPtr(shardID),\n\t\t\tDomainName:      domainName,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp.NewBranchToken, nil\n}\n"
  },
  {
    "path": "service/history/ndc/workflow_resetter_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: workflow_resetter.go\n//\n// Generated by this command:\n//\n//\tmockgen -package ndc -source workflow_resetter.go -destination workflow_resetter_mock.go\n//\n\n// Package ndc is a generated GoMock package.\npackage ndc\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MockWorkflowResetter is a mock of WorkflowResetter interface.\ntype MockWorkflowResetter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWorkflowResetterMockRecorder\n\tisgomock struct{}\n}\n\n// MockWorkflowResetterMockRecorder is the mock recorder for MockWorkflowResetter.\ntype MockWorkflowResetterMockRecorder struct {\n\tmock *MockWorkflowResetter\n}\n\n// NewMockWorkflowResetter creates a new mock instance.\nfunc NewMockWorkflowResetter(ctrl *gomock.Controller) *MockWorkflowResetter {\n\tmock := &MockWorkflowResetter{ctrl: ctrl}\n\tmock.recorder = &MockWorkflowResetterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWorkflowResetter) EXPECT() *MockWorkflowResetterMockRecorder {\n\treturn m.recorder\n}\n\n// ResetWorkflow mocks base method.\nfunc (m *MockWorkflowResetter) ResetWorkflow(ctx context.Context, now time.Time, baseLastEventID, baseLastEventVersion, incomingFirstEventID, incomingFirstEventVersion int64) (execution.MutableState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetWorkflow\", ctx, now, baseLastEventID, baseLastEventVersion, incomingFirstEventID, incomingFirstEventVersion)\n\tret0, _ := ret[0].(execution.MutableState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ResetWorkflow indicates an expected call of ResetWorkflow.\nfunc (mr *MockWorkflowResetterMockRecorder) ResetWorkflow(ctx, now, baseLastEventID, baseLastEventVersion, incomingFirstEventID, incomingFirstEventVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflow\", reflect.TypeOf((*MockWorkflowResetter)(nil).ResetWorkflow), ctx, now, baseLastEventID, baseLastEventVersion, incomingFirstEventID, incomingFirstEventVersion)\n}\n"
  },
  {
    "path": "service/history/ndc/workflow_resetter_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage ndc\n\nimport (\n\tctx \"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/net/context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tworkflowResetterSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller              *gomock.Controller\n\t\tmockShard               *shard.TestContext\n\t\tmockBaseMutableState    *execution.MockMutableState\n\t\tmockRebuiltMutableState *execution.MockMutableState\n\t\tmockTransactionMgr      *MocktransactionManager\n\t\tmockStateRebuilder      *execution.MockStateRebuilder\n\n\t\tlogger           log.Logger\n\t\tmockHistoryV2Mgr *mocks.HistoryV2Manager\n\n\t\tdomainID   string\n\t\tdomainName string\n\t\tworkflowID string\n\t\tbaseRunID  string\n\t\tnewContext execution.Context\n\t\tnewRunID   string\n\n\t\tworkflowResetter *workflowResetterImpl\n\t}\n)\n\nfunc TestWorkflowResetterSuite(t *testing.T) {\n\ts := new(workflowResetterSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *workflowResetterSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockBaseMutableState = execution.NewMockMutableState(s.controller)\n\ts.mockRebuiltMutableState = execution.NewMockMutableState(s.controller)\n\ts.mockTransactionMgr = NewMocktransactionManager(s.controller)\n\ts.mockStateRebuilder = execution.NewMockStateRebuilder(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\n\ts.logger = s.mockShard.GetLogger()\n\n\ts.domainID = uuid.New()\n\ts.domainName = \"some random domain name\"\n\ts.workflowID = \"some random workflow ID\"\n\ts.baseRunID = uuid.New()\n\ts.newContext = execution.NewContext(\n\t\ts.domainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: s.workflowID,\n\t\t\tRunID:      s.newRunID,\n\t\t},\n\t\ts.mockShard,\n\t\tnil,\n\t\ts.logger,\n\t)\n\ts.newRunID = uuid.New()\n\n\ts.workflowResetter = NewWorkflowResetter(\n\t\ts.mockShard, s.mockTransactionMgr, s.domainID, s.workflowID, s.baseRunID, s.newContext, s.newRunID, s.logger,\n\t).(*workflowResetterImpl)\n\ts.workflowResetter.stateRebuilder = s.mockStateRebuilder\n}\n\nfunc (s *workflowResetterSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *workflowResetterSuite) TestResetWorkflow_NoError() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tbranchToken := []byte(\"some random branch token\")\n\tlastEventID := int64(500)\n\tversion := int64(123)\n\tversionHistory := persistence.NewVersionHistory(\n\t\tbranchToken,\n\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID, version)},\n\t)\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\n\tbaseEventID := lastEventID - 100\n\tbaseVersion := version\n\tincomingFirstEventID := baseEventID + 12\n\tincomingVersion := baseVersion + 3\n\n\trebuiltHistorySize := int64(9999)\n\tresetBranchToken := []byte(\"other random branch token\")\n\tdomainName := \"test-domainName\"\n\n\ts.mockBaseMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockBaseMutableState.EXPECT().GetCurrentBranchToken().Return(branchToken, nil).AnyTimes()\n\n\tmockBaseWorkflowReleaseFnCalled := false\n\tmockBaseWorkflowReleaseFn := func(err error) {\n\t\tmockBaseWorkflowReleaseFnCalled = true\n\t}\n\tmockBaseWorkflow := execution.NewMockWorkflow(s.controller)\n\tmockBaseWorkflow.EXPECT().GetMutableState().Return(s.mockBaseMutableState).AnyTimes()\n\tmockBaseWorkflow.EXPECT().GetReleaseFn().Return(mockBaseWorkflowReleaseFn).Times(1)\n\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(\n\t\tctx,\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.baseRunID,\n\t).Return(mockBaseWorkflow, nil).Times(1)\n\n\ts.mockStateRebuilder.EXPECT().Rebuild(\n\t\tctx,\n\t\tnow,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\ts.domainID,\n\t\t\ts.workflowID,\n\t\t\ts.baseRunID,\n\t\t),\n\t\tbranchToken,\n\t\tbaseEventID,\n\t\tbaseVersion,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\ts.domainID,\n\t\t\ts.workflowID,\n\t\t\ts.newRunID,\n\t\t),\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t).DoAndReturn(func(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowIdentifier, baseBranchToken []byte, baseRebuildLastEventID int64, baseRebuildLastEventVersion int64, targetWorkflowIdentifier definition.WorkflowIdentifier, targetBranchFn func() ([]byte, error), requestID string) (execution.MutableState, int64, error) {\n\t\ttargetBranchToken, err := targetBranchFn()\n\t\ts.NoError(err)\n\t\ts.Equal(resetBranchToken, targetBranchToken)\n\n\t\treturn s.mockRebuiltMutableState, rebuiltHistorySize, nil\n\t}).Times(1)\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ForkHistoryBranch\", mock.Anything, &persistence.ForkHistoryBranchRequest{\n\t\tForkBranchToken: branchToken,\n\t\tForkNodeID:      baseEventID + 1,\n\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(s.domainID, s.workflowID, s.newRunID),\n\t\tShardID:         common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:      domainName,\n\t}).Return(&persistence.ForkHistoryBranchResponse{NewBranchToken: resetBranchToken}, nil).Times(1)\n\n\trebuiltMutableState, err := s.workflowResetter.ResetWorkflow(\n\t\tctx,\n\t\tnow,\n\t\tbaseEventID,\n\t\tbaseVersion,\n\t\tincomingFirstEventID,\n\t\tincomingVersion,\n\t)\n\ts.NoError(err)\n\ts.Equal(s.mockRebuiltMutableState, rebuiltMutableState)\n\ts.Equal(s.newContext.GetHistorySize(), rebuiltHistorySize)\n\ts.True(mockBaseWorkflowReleaseFnCalled)\n}\n\nfunc (s *workflowResetterSuite) TestResetWorkflow_Error() {\n\tctx := ctx.Background()\n\tnow := time.Now()\n\n\tbranchToken := []byte(\"some random branch token\")\n\tlastEventID := int64(500)\n\tversion := int64(123)\n\tversionHistory := persistence.NewVersionHistory(\n\t\tbranchToken,\n\t\t[]*persistence.VersionHistoryItem{persistence.NewVersionHistoryItem(lastEventID, version)},\n\t)\n\tversionHistories := persistence.NewVersionHistories(versionHistory)\n\tbaseEventID := lastEventID + 100\n\tbaseVersion := version\n\tincomingFirstEventID := baseEventID + 12\n\tincomingFirstEventVersion := baseVersion + 3\n\n\ts.mockBaseMutableState.EXPECT().GetVersionHistories().Return(versionHistories).AnyTimes()\n\ts.mockBaseMutableState.EXPECT().GetCurrentBranchToken().Return(branchToken, nil).AnyTimes()\n\n\tmockBaseWorkflowReleaseFn := func(err error) {\n\t}\n\tmockBaseWorkflow := execution.NewMockWorkflow(s.controller)\n\tmockBaseWorkflow.EXPECT().GetMutableState().Return(s.mockBaseMutableState).AnyTimes()\n\tmockBaseWorkflow.EXPECT().GetReleaseFn().Return(mockBaseWorkflowReleaseFn).Times(1)\n\n\ts.mockTransactionMgr.EXPECT().loadNDCWorkflow(\n\t\tctx,\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.baseRunID,\n\t).Return(mockBaseWorkflow, nil).Times(1)\n\n\trebuiltMutableState, err := s.workflowResetter.ResetWorkflow(\n\t\tctx,\n\t\tnow,\n\t\tbaseEventID,\n\t\tbaseVersion,\n\t\tincomingFirstEventID,\n\t\tincomingFirstEventVersion,\n\t)\n\ts.Error(err)\n\ts.IsType(&types.RetryTaskV2Error{}, err)\n\ts.Nil(rebuiltMutableState)\n\n\tretryErr, isRetryError := err.(*types.RetryTaskV2Error)\n\ts.True(isRetryError)\n\texpectedErr := &types.RetryTaskV2Error{\n\t\tMessage:         resendOnResetWorkflowMessage,\n\t\tDomainID:        s.domainID,\n\t\tWorkflowID:      s.workflowID,\n\t\tRunID:           s.newRunID,\n\t\tEndEventID:      common.Int64Ptr(incomingFirstEventID),\n\t\tEndEventVersion: common.Int64Ptr(incomingFirstEventVersion),\n\t}\n\ts.Equal(retryErr, expectedErr)\n}\n"
  },
  {
    "path": "service/history/query/query.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"sync/atomic\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// TerminationTypeCompleted means a query reaches its termination state because it has been completed\n\tTerminationTypeCompleted TerminationType = iota\n\t// TerminationTypeUnblocked means a query reaches its termination state because it has been unblocked\n\tTerminationTypeUnblocked\n\t// TerminationTypeFailed means a query reaches its termination state because it has failed\n\tTerminationTypeFailed\n)\n\nvar (\n\terrTerminationStateInvalid = &types.InternalServiceError{Message: \"query termination state invalid\"}\n\terrAlreadyInTerminalState  = &types.InternalServiceError{Message: \"query already in terminal state\"}\n\terrQueryNotInTerminalState = &types.InternalServiceError{Message: \"query not in terminal state\"}\n)\n\ntype (\n\t// TerminationType is the type of a query's termination state\n\tTerminationType int\n\n\t// TerminationState describes a query's termination state\n\tTerminationState struct {\n\t\tTerminationType TerminationType\n\t\tQueryResult     *types.WorkflowQueryResult\n\t\tFailure         error\n\t}\n\n\tquery interface {\n\t\tgetQueryID() string\n\t\tgetQueryTermCh() <-chan struct{}\n\t\tgetQueryInput() *types.WorkflowQuery\n\t\tgetTerminationState() (*TerminationState, error)\n\t\tsetTerminationState(*TerminationState) error\n\t}\n\n\tqueryImpl struct {\n\t\tid         string\n\t\tqueryInput *types.WorkflowQuery\n\t\ttermCh     chan struct{}\n\n\t\tterminationState atomic.Value\n\t}\n)\n\nfunc newQuery(queryInput *types.WorkflowQuery) query {\n\treturn &queryImpl{\n\t\tid:         uuid.New(),\n\t\tqueryInput: queryInput,\n\t\ttermCh:     make(chan struct{}),\n\t}\n}\n\nfunc (q *queryImpl) getQueryID() string {\n\treturn q.id\n}\n\nfunc (q *queryImpl) getQueryTermCh() <-chan struct{} {\n\treturn q.termCh\n}\n\nfunc (q *queryImpl) getQueryInput() *types.WorkflowQuery {\n\treturn q.queryInput\n}\n\nfunc (q *queryImpl) getTerminationState() (*TerminationState, error) {\n\tts := q.terminationState.Load()\n\tif ts == nil {\n\t\treturn nil, errQueryNotInTerminalState\n\t}\n\treturn ts.(*TerminationState), nil\n}\n\nfunc (q *queryImpl) setTerminationState(terminationState *TerminationState) error {\n\tif err := q.validateTerminationState(terminationState); err != nil {\n\t\treturn err\n\t}\n\tcurrTerminationState, _ := q.getTerminationState()\n\tif currTerminationState != nil {\n\t\treturn errAlreadyInTerminalState\n\t}\n\tq.terminationState.Store(terminationState)\n\tclose(q.termCh)\n\treturn nil\n}\n\nfunc (q *queryImpl) validateTerminationState(\n\tterminationState *TerminationState,\n) error {\n\tif terminationState == nil {\n\t\treturn errTerminationStateInvalid\n\t}\n\tswitch terminationState.TerminationType {\n\tcase TerminationTypeCompleted:\n\t\tif terminationState.QueryResult == nil || terminationState.Failure != nil {\n\t\t\treturn errTerminationStateInvalid\n\t\t}\n\t\tqueryResult := terminationState.QueryResult\n\t\tvalidAnswered := queryResult.GetResultType() == types.QueryResultTypeAnswered &&\n\t\t\tqueryResult.Answer != nil &&\n\t\t\tqueryResult.ErrorMessage == \"\"\n\t\tvalidFailed := queryResult.GetResultType() == types.QueryResultTypeFailed &&\n\t\t\tqueryResult.Answer == nil &&\n\t\t\tqueryResult.ErrorMessage != \"\"\n\t\tif !validAnswered && !validFailed {\n\t\t\treturn errTerminationStateInvalid\n\t\t}\n\t\treturn nil\n\tcase TerminationTypeUnblocked:\n\t\tif terminationState.QueryResult != nil || terminationState.Failure != nil {\n\t\t\treturn errTerminationStateInvalid\n\t\t}\n\t\treturn nil\n\tcase TerminationTypeFailed:\n\t\tif terminationState.QueryResult != nil || terminationState.Failure == nil {\n\t\t\treturn errTerminationStateInvalid\n\t\t}\n\t\treturn nil\n\tdefault:\n\t\treturn errTerminationStateInvalid\n\t}\n}\n"
  },
  {
    "path": "service/history/query/query_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype QuerySuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestQuerySuite(t *testing.T) {\n\tsuite.Run(t, new(QuerySuite))\n}\n\nfunc (s *QuerySuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *QuerySuite) TestValidateTerminationState() {\n\ttestCases := []struct {\n\t\tts        *TerminationState\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tts:        nil,\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult:     &types.WorkflowQueryResult{},\n\t\t\t\tFailure:         errors.New(\"err\"),\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType:   types.QueryResultTypeAnswered.Ptr(),\n\t\t\t\t\tAnswer:       []byte{1, 2, 3},\n\t\t\t\t\tErrorMessage: \"err\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType: types.QueryResultTypeFailed.Ptr(),\n\t\t\t\t\tAnswer:     []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType:   types.QueryResultTypeFailed.Ptr(),\n\t\t\t\t\tErrorMessage: \"err\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\t\t\tAnswer:     []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeUnblocked,\n\t\t\t\tQueryResult:     &types.WorkflowQueryResult{},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeUnblocked,\n\t\t\t\tFailure:         errors.New(\"err\"),\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeUnblocked,\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeFailed,\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeFailed,\n\t\t\t\tQueryResult:     &types.WorkflowQueryResult{},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tts: &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeFailed,\n\t\t\t\tFailure:         errors.New(\"err\"),\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tqueryImpl := &queryImpl{}\n\tfor _, tc := range testCases {\n\t\tif tc.expectErr {\n\t\t\ts.Error(queryImpl.validateTerminationState(tc.ts))\n\t\t} else {\n\t\t\ts.NoError(queryImpl.validateTerminationState(tc.ts))\n\t\t}\n\t}\n}\n\nfunc (s *QuerySuite) TestTerminationState_Failed() {\n\tfailedTerminationState := &TerminationState{\n\t\tTerminationType: TerminationTypeFailed,\n\t\tFailure:         errors.New(\"err\"),\n\t}\n\ts.testSetTerminationState(failedTerminationState)\n}\n\nfunc (s *QuerySuite) TestTerminationState_Completed() {\n\tansweredTerminationState := &TerminationState{\n\t\tTerminationType: TerminationTypeCompleted,\n\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\tAnswer:     []byte{1, 2, 3},\n\t\t},\n\t}\n\ts.testSetTerminationState(answeredTerminationState)\n}\n\nfunc (s *QuerySuite) TestTerminationState_Unblocked() {\n\tunblockedTerminationState := &TerminationState{\n\t\tTerminationType: TerminationTypeUnblocked,\n\t}\n\ts.testSetTerminationState(unblockedTerminationState)\n}\n\nfunc (s *QuerySuite) testSetTerminationState(terminationState *TerminationState) {\n\tquery := newQuery(nil)\n\tts, err := query.getTerminationState()\n\ts.Equal(errQueryNotInTerminalState, err)\n\ts.Nil(ts)\n\ts.False(closed(query.getQueryTermCh()))\n\ts.Equal(errTerminationStateInvalid, query.setTerminationState(nil))\n\ts.NoError(query.setTerminationState(terminationState))\n\ts.True(closed(query.getQueryTermCh()))\n\tactualTerminationState, err := query.getTerminationState()\n\ts.NoError(err)\n\ts.assertTerminationStateEqual(terminationState, actualTerminationState)\n}\n\nfunc (s *QuerySuite) assertTerminationStateEqual(expected *TerminationState, actual *TerminationState) {\n\ts.Equal(expected.TerminationType, actual.TerminationType)\n\tif expected.Failure != nil {\n\t\ts.Equal(expected.Failure.Error(), actual.Failure.Error())\n\t}\n\tif expected.QueryResult != nil {\n\t\ts.Equal(expected.QueryResult, actual.QueryResult)\n\t}\n}\n\nfunc closed(ch <-chan struct{}) bool {\n\tselect {\n\tcase <-ch:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "service/history/query/registry.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination registry_mock.go -self_package github.com/uber/cadence/service/history/query\n\npackage query\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\terrQueryNotExists = &types.InternalServiceError{Message: \"query does not exist\"}\n)\n\ntype (\n\t// Registry manages all the queries for a workflow\n\tRegistry interface {\n\t\tHasBufferedQuery() bool\n\t\tGetBufferedIDs() []string\n\t\tHasCompletedQuery() bool\n\t\tGetCompletedIDs() []string\n\t\tHasUnblockedQuery() bool\n\t\tGetUnblockedIDs() []string\n\t\tHasFailedQuery() bool\n\t\tGetFailedIDs() []string\n\n\t\tGetQueryTermCh(string) (<-chan struct{}, error)\n\t\tGetQueryInput(string) (*types.WorkflowQuery, error)\n\t\tGetTerminationState(string) (*TerminationState, error)\n\n\t\tBufferQuery(queryInput *types.WorkflowQuery) (string, <-chan struct{})\n\t\tSetTerminationState(string, *TerminationState) error\n\t\tRemoveQuery(id string)\n\t}\n\n\tregistryImpl struct {\n\t\tsync.RWMutex\n\n\t\tbuffered  map[string]query\n\t\tcompleted map[string]query\n\t\tunblocked map[string]query\n\t\tfailed    map[string]query\n\t}\n)\n\n// NewRegistry creates a new query registry\nfunc NewRegistry() Registry {\n\treturn &registryImpl{\n\t\tbuffered:  make(map[string]query),\n\t\tcompleted: make(map[string]query),\n\t\tunblocked: make(map[string]query),\n\t\tfailed:    make(map[string]query),\n\t}\n}\n\nfunc (r *registryImpl) HasBufferedQuery() bool {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn len(r.buffered) > 0\n}\n\nfunc (r *registryImpl) GetBufferedIDs() []string {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn r.getIDs(r.buffered)\n}\n\nfunc (r *registryImpl) HasCompletedQuery() bool {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn len(r.completed) > 0\n}\n\nfunc (r *registryImpl) GetCompletedIDs() []string {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn r.getIDs(r.completed)\n}\n\nfunc (r *registryImpl) HasUnblockedQuery() bool {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn len(r.unblocked) > 0\n}\n\nfunc (r *registryImpl) GetUnblockedIDs() []string {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn r.getIDs(r.unblocked)\n}\n\nfunc (r *registryImpl) HasFailedQuery() bool {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn len(r.failed) > 0\n}\n\nfunc (r *registryImpl) GetFailedIDs() []string {\n\tr.RLock()\n\tdefer r.RUnlock()\n\treturn r.getIDs(r.failed)\n}\n\nfunc (r *registryImpl) GetQueryTermCh(id string) (<-chan struct{}, error) {\n\tr.RLock()\n\tdefer r.RUnlock()\n\tq, err := r.getQueryNoLock(id)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn q.getQueryTermCh(), nil\n}\n\nfunc (r *registryImpl) GetQueryInput(id string) (*types.WorkflowQuery, error) {\n\tr.RLock()\n\tdefer r.RUnlock()\n\tq, err := r.getQueryNoLock(id)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn q.getQueryInput(), nil\n}\n\nfunc (r *registryImpl) GetTerminationState(id string) (*TerminationState, error) {\n\tr.RLock()\n\tdefer r.RUnlock()\n\tq, err := r.getQueryNoLock(id)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn q.getTerminationState()\n}\n\nfunc (r *registryImpl) BufferQuery(queryInput *types.WorkflowQuery) (string, <-chan struct{}) {\n\tr.Lock()\n\tdefer r.Unlock()\n\tq := newQuery(queryInput)\n\tid := q.getQueryID()\n\tr.buffered[id] = q\n\treturn id, q.getQueryTermCh()\n}\n\nfunc (r *registryImpl) SetTerminationState(id string, TerminationState *TerminationState) error {\n\tr.Lock()\n\tdefer r.Unlock()\n\tq, ok := r.buffered[id]\n\tif !ok {\n\t\treturn errQueryNotExists\n\t}\n\tif err := q.setTerminationState(TerminationState); err != nil {\n\t\treturn err\n\t}\n\tdelete(r.buffered, id)\n\tswitch TerminationState.TerminationType {\n\tcase TerminationTypeCompleted:\n\t\tr.completed[id] = q\n\tcase TerminationTypeUnblocked:\n\t\tr.unblocked[id] = q\n\tcase TerminationTypeFailed:\n\t\tr.failed[id] = q\n\t}\n\treturn nil\n}\n\nfunc (r *registryImpl) RemoveQuery(id string) {\n\tr.Lock()\n\tdefer r.Unlock()\n\tdelete(r.buffered, id)\n\tdelete(r.completed, id)\n\tdelete(r.unblocked, id)\n\tdelete(r.failed, id)\n}\n\nfunc (r *registryImpl) getQueryNoLock(id string) (query, error) {\n\tif q, ok := r.buffered[id]; ok {\n\t\treturn q, nil\n\t}\n\tif q, ok := r.completed[id]; ok {\n\t\treturn q, nil\n\t}\n\tif q, ok := r.unblocked[id]; ok {\n\t\treturn q, nil\n\t}\n\tif q, ok := r.failed[id]; ok {\n\t\treturn q, nil\n\t}\n\treturn nil, errQueryNotExists\n}\n\nfunc (r *registryImpl) getIDs(m map[string]query) []string {\n\tresult := make([]string, len(m))\n\tindex := 0\n\tfor id := range m {\n\t\tresult[index] = id\n\t\tindex++\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "service/history/query/registry_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: registry.go\n//\n// Generated by this command:\n//\n//\tmockgen -package query -source registry.go -destination registry_mock.go -self_package github.com/uber/cadence/service/history/query\n//\n\n// Package query is a generated GoMock package.\npackage query\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockRegistry is a mock of Registry interface.\ntype MockRegistry struct {\n\tctrl     *gomock.Controller\n\trecorder *MockRegistryMockRecorder\n\tisgomock struct{}\n}\n\n// MockRegistryMockRecorder is the mock recorder for MockRegistry.\ntype MockRegistryMockRecorder struct {\n\tmock *MockRegistry\n}\n\n// NewMockRegistry creates a new mock instance.\nfunc NewMockRegistry(ctrl *gomock.Controller) *MockRegistry {\n\tmock := &MockRegistry{ctrl: ctrl}\n\tmock.recorder = &MockRegistryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRegistry) EXPECT() *MockRegistryMockRecorder {\n\treturn m.recorder\n}\n\n// BufferQuery mocks base method.\nfunc (m *MockRegistry) BufferQuery(queryInput *types.WorkflowQuery) (string, <-chan struct{}) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"BufferQuery\", queryInput)\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(<-chan struct{})\n\treturn ret0, ret1\n}\n\n// BufferQuery indicates an expected call of BufferQuery.\nfunc (mr *MockRegistryMockRecorder) BufferQuery(queryInput any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"BufferQuery\", reflect.TypeOf((*MockRegistry)(nil).BufferQuery), queryInput)\n}\n\n// GetBufferedIDs mocks base method.\nfunc (m *MockRegistry) GetBufferedIDs() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBufferedIDs\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetBufferedIDs indicates an expected call of GetBufferedIDs.\nfunc (mr *MockRegistryMockRecorder) GetBufferedIDs() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBufferedIDs\", reflect.TypeOf((*MockRegistry)(nil).GetBufferedIDs))\n}\n\n// GetCompletedIDs mocks base method.\nfunc (m *MockRegistry) GetCompletedIDs() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCompletedIDs\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetCompletedIDs indicates an expected call of GetCompletedIDs.\nfunc (mr *MockRegistryMockRecorder) GetCompletedIDs() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCompletedIDs\", reflect.TypeOf((*MockRegistry)(nil).GetCompletedIDs))\n}\n\n// GetFailedIDs mocks base method.\nfunc (m *MockRegistry) GetFailedIDs() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFailedIDs\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetFailedIDs indicates an expected call of GetFailedIDs.\nfunc (mr *MockRegistryMockRecorder) GetFailedIDs() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFailedIDs\", reflect.TypeOf((*MockRegistry)(nil).GetFailedIDs))\n}\n\n// GetQueryInput mocks base method.\nfunc (m *MockRegistry) GetQueryInput(arg0 string) (*types.WorkflowQuery, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueryInput\", arg0)\n\tret0, _ := ret[0].(*types.WorkflowQuery)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueryInput indicates an expected call of GetQueryInput.\nfunc (mr *MockRegistryMockRecorder) GetQueryInput(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueryInput\", reflect.TypeOf((*MockRegistry)(nil).GetQueryInput), arg0)\n}\n\n// GetQueryTermCh mocks base method.\nfunc (m *MockRegistry) GetQueryTermCh(arg0 string) (<-chan struct{}, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueryTermCh\", arg0)\n\tret0, _ := ret[0].(<-chan struct{})\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueryTermCh indicates an expected call of GetQueryTermCh.\nfunc (mr *MockRegistryMockRecorder) GetQueryTermCh(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueryTermCh\", reflect.TypeOf((*MockRegistry)(nil).GetQueryTermCh), arg0)\n}\n\n// GetTerminationState mocks base method.\nfunc (m *MockRegistry) GetTerminationState(arg0 string) (*TerminationState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTerminationState\", arg0)\n\tret0, _ := ret[0].(*TerminationState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTerminationState indicates an expected call of GetTerminationState.\nfunc (mr *MockRegistryMockRecorder) GetTerminationState(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTerminationState\", reflect.TypeOf((*MockRegistry)(nil).GetTerminationState), arg0)\n}\n\n// GetUnblockedIDs mocks base method.\nfunc (m *MockRegistry) GetUnblockedIDs() []string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetUnblockedIDs\")\n\tret0, _ := ret[0].([]string)\n\treturn ret0\n}\n\n// GetUnblockedIDs indicates an expected call of GetUnblockedIDs.\nfunc (mr *MockRegistryMockRecorder) GetUnblockedIDs() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetUnblockedIDs\", reflect.TypeOf((*MockRegistry)(nil).GetUnblockedIDs))\n}\n\n// HasBufferedQuery mocks base method.\nfunc (m *MockRegistry) HasBufferedQuery() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasBufferedQuery\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasBufferedQuery indicates an expected call of HasBufferedQuery.\nfunc (mr *MockRegistryMockRecorder) HasBufferedQuery() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasBufferedQuery\", reflect.TypeOf((*MockRegistry)(nil).HasBufferedQuery))\n}\n\n// HasCompletedQuery mocks base method.\nfunc (m *MockRegistry) HasCompletedQuery() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasCompletedQuery\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasCompletedQuery indicates an expected call of HasCompletedQuery.\nfunc (mr *MockRegistryMockRecorder) HasCompletedQuery() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasCompletedQuery\", reflect.TypeOf((*MockRegistry)(nil).HasCompletedQuery))\n}\n\n// HasFailedQuery mocks base method.\nfunc (m *MockRegistry) HasFailedQuery() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasFailedQuery\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasFailedQuery indicates an expected call of HasFailedQuery.\nfunc (mr *MockRegistryMockRecorder) HasFailedQuery() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasFailedQuery\", reflect.TypeOf((*MockRegistry)(nil).HasFailedQuery))\n}\n\n// HasUnblockedQuery mocks base method.\nfunc (m *MockRegistry) HasUnblockedQuery() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasUnblockedQuery\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasUnblockedQuery indicates an expected call of HasUnblockedQuery.\nfunc (mr *MockRegistryMockRecorder) HasUnblockedQuery() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasUnblockedQuery\", reflect.TypeOf((*MockRegistry)(nil).HasUnblockedQuery))\n}\n\n// RemoveQuery mocks base method.\nfunc (m *MockRegistry) RemoveQuery(id string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RemoveQuery\", id)\n}\n\n// RemoveQuery indicates an expected call of RemoveQuery.\nfunc (mr *MockRegistryMockRecorder) RemoveQuery(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveQuery\", reflect.TypeOf((*MockRegistry)(nil).RemoveQuery), id)\n}\n\n// SetTerminationState mocks base method.\nfunc (m *MockRegistry) SetTerminationState(arg0 string, arg1 *TerminationState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SetTerminationState\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SetTerminationState indicates an expected call of SetTerminationState.\nfunc (mr *MockRegistryMockRecorder) SetTerminationState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetTerminationState\", reflect.TypeOf((*MockRegistry)(nil).SetTerminationState), arg0, arg1)\n}\n"
  },
  {
    "path": "service/history/query/registry_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage query\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype QueryRegistrySuite struct {\n\tsuite.Suite\n\t*require.Assertions\n}\n\nfunc TestQueryRegistrySuite(t *testing.T) {\n\tsuite.Run(t, new(QueryRegistrySuite))\n}\n\nfunc (s *QueryRegistrySuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *QueryRegistrySuite) TestQueryRegistry() {\n\tqr := NewRegistry()\n\tids := make([]string, 100)\n\ttermChans := make([]<-chan struct{}, 100)\n\tfor i := 0; i < 100; i++ {\n\t\tids[i], termChans[i] = qr.BufferQuery(&types.WorkflowQuery{})\n\t}\n\ts.assertBufferedState(qr, ids...)\n\ts.assertHasQueries(qr, true, false, false, false)\n\ts.assertQuerySizes(qr, 100, 0, 0, 0)\n\ts.assertChanState(false, termChans...)\n\n\tfor i := 0; i < 25; i++ {\n\t\terr := qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\tQueryResult: &types.WorkflowQueryResult{\n\t\t\t\tResultType: types.QueryResultTypeAnswered.Ptr(),\n\t\t\t\tAnswer:     []byte{1, 2, 3},\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\t}\n\ts.assertCompletedState(qr, ids[0:25]...)\n\ts.assertBufferedState(qr, ids[25:]...)\n\ts.assertHasQueries(qr, true, true, false, false)\n\ts.assertQuerySizes(qr, 75, 25, 0, 0)\n\ts.assertChanState(true, termChans[0:25]...)\n\ts.assertChanState(false, termChans[25:]...)\n\n\tfor i := 25; i < 50; i++ {\n\t\terr := qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\tTerminationType: TerminationTypeUnblocked,\n\t\t})\n\t\ts.NoError(err)\n\t}\n\ts.assertCompletedState(qr, ids[0:25]...)\n\ts.assertUnblockedState(qr, ids[25:50]...)\n\ts.assertBufferedState(qr, ids[50:]...)\n\ts.assertHasQueries(qr, true, true, true, false)\n\ts.assertQuerySizes(qr, 50, 25, 25, 0)\n\ts.assertChanState(true, termChans[0:50]...)\n\ts.assertChanState(false, termChans[50:]...)\n\n\tfor i := 50; i < 75; i++ {\n\t\terr := qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\tTerminationType: TerminationTypeFailed,\n\t\t\tFailure:         errors.New(\"err\"),\n\t\t})\n\t\ts.NoError(err)\n\t}\n\ts.assertCompletedState(qr, ids[0:25]...)\n\ts.assertUnblockedState(qr, ids[25:50]...)\n\ts.assertFailedState(qr, ids[50:75]...)\n\ts.assertBufferedState(qr, ids[75:]...)\n\ts.assertHasQueries(qr, true, true, true, true)\n\ts.assertQuerySizes(qr, 25, 25, 25, 25)\n\ts.assertChanState(true, termChans[0:75]...)\n\ts.assertChanState(false, termChans[75:]...)\n\n\tfor i := 0; i < 75; i++ {\n\t\tswitch i % 3 {\n\t\tcase 0:\n\t\t\ts.Equal(errQueryNotExists, qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeCompleted,\n\t\t\t\tQueryResult:     &types.WorkflowQueryResult{},\n\t\t\t}))\n\t\tcase 1:\n\t\t\ts.Equal(errQueryNotExists, qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeUnblocked,\n\t\t\t}))\n\t\tcase 2:\n\t\t\ts.Equal(errQueryNotExists, qr.SetTerminationState(ids[i], &TerminationState{\n\t\t\t\tTerminationType: TerminationTypeFailed,\n\t\t\t\tFailure:         errors.New(\"err\"),\n\t\t\t}))\n\t\t}\n\t}\n\ts.assertCompletedState(qr, ids[0:25]...)\n\ts.assertUnblockedState(qr, ids[25:50]...)\n\ts.assertFailedState(qr, ids[50:75]...)\n\ts.assertBufferedState(qr, ids[75:]...)\n\ts.assertHasQueries(qr, true, true, true, true)\n\ts.assertQuerySizes(qr, 25, 25, 25, 25)\n\ts.assertChanState(true, termChans[0:75]...)\n\ts.assertChanState(false, termChans[75:]...)\n\n\tfor i := 0; i < 25; i++ {\n\t\tqr.RemoveQuery(ids[i])\n\t\ts.assertHasQueries(qr, true, i < 24, true, true)\n\t\ts.assertQuerySizes(qr, 25, 25-i-1, 25, 25)\n\t}\n\tfor i := 25; i < 50; i++ {\n\t\tqr.RemoveQuery(ids[i])\n\t\ts.assertHasQueries(qr, true, false, i < 49, true)\n\t\ts.assertQuerySizes(qr, 25, 0, 50-i-1, 25)\n\t}\n\tfor i := 50; i < 75; i++ {\n\t\tqr.RemoveQuery(ids[i])\n\t\ts.assertHasQueries(qr, true, false, false, i < 74)\n\t\ts.assertQuerySizes(qr, 25, 0, 0, 75-i-1)\n\t}\n\tfor i := 75; i < 100; i++ {\n\t\tqr.RemoveQuery(ids[i])\n\t\ts.assertHasQueries(qr, i < 99, false, false, false)\n\t\ts.assertQuerySizes(qr, 100-i-1, 0, 0, 0)\n\t}\n\ts.assertChanState(true, termChans[0:75]...)\n\ts.assertChanState(false, termChans[75:]...)\n}\n\nfunc (s *QueryRegistrySuite) assertBufferedState(qr Registry, ids ...string) {\n\tfor _, id := range ids {\n\t\ttermCh, err := qr.GetQueryTermCh(id)\n\t\ts.NoError(err)\n\t\ts.False(closed(termCh))\n\t\tinput, err := qr.GetQueryInput(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(input)\n\t\ttermState, err := qr.GetTerminationState(id)\n\t\ts.Equal(errQueryNotInTerminalState, err)\n\t\ts.Nil(termState)\n\t}\n}\n\nfunc (s *QueryRegistrySuite) assertCompletedState(qr Registry, ids ...string) {\n\tfor _, id := range ids {\n\t\ttermCh, err := qr.GetQueryTermCh(id)\n\t\ts.NoError(err)\n\t\ts.True(closed(termCh))\n\t\tinput, err := qr.GetQueryInput(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(input)\n\t\ttermState, err := qr.GetTerminationState(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(termState)\n\t\ts.Equal(TerminationTypeCompleted, termState.TerminationType)\n\t\ts.NotNil(termState.QueryResult)\n\t\ts.Nil(termState.Failure)\n\t}\n}\n\nfunc (s *QueryRegistrySuite) assertUnblockedState(qr Registry, ids ...string) {\n\tfor _, id := range ids {\n\t\ttermCh, err := qr.GetQueryTermCh(id)\n\t\ts.NoError(err)\n\t\ts.True(closed(termCh))\n\t\tinput, err := qr.GetQueryInput(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(input)\n\t\ttermState, err := qr.GetTerminationState(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(termState)\n\t\ts.Equal(TerminationTypeUnblocked, termState.TerminationType)\n\t\ts.Nil(termState.QueryResult)\n\t\ts.Nil(termState.Failure)\n\t}\n}\n\nfunc (s *QueryRegistrySuite) assertFailedState(qr Registry, ids ...string) {\n\tfor _, id := range ids {\n\t\ttermCh, err := qr.GetQueryTermCh(id)\n\t\ts.NoError(err)\n\t\ts.True(closed(termCh))\n\t\tinput, err := qr.GetQueryInput(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(input)\n\t\ttermState, err := qr.GetTerminationState(id)\n\t\ts.NoError(err)\n\t\ts.NotNil(termState)\n\t\ts.Equal(TerminationTypeFailed, termState.TerminationType)\n\t\ts.Nil(termState.QueryResult)\n\t\ts.NotNil(termState.Failure)\n\t}\n}\n\nfunc (s *QueryRegistrySuite) assertHasQueries(qr Registry, buffered, completed, unblocked, failed bool) {\n\ts.Equal(buffered, qr.HasBufferedQuery())\n\ts.Equal(completed, qr.HasCompletedQuery())\n\ts.Equal(unblocked, qr.HasUnblockedQuery())\n\ts.Equal(failed, qr.HasFailedQuery())\n}\n\nfunc (s *QueryRegistrySuite) assertQuerySizes(qr Registry, buffered, completed, unblocked, failed int) {\n\ts.Len(qr.GetBufferedIDs(), buffered)\n\ts.Len(qr.GetCompletedIDs(), completed)\n\ts.Len(qr.GetUnblockedIDs(), unblocked)\n\ts.Len(qr.GetFailedIDs(), failed)\n}\n\nfunc (s *QueryRegistrySuite) assertChanState(expectedClosed bool, chans ...<-chan struct{}) {\n\tfor _, ch := range chans {\n\t\ts.Equal(expectedClosed, closed(ch))\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/action.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport \"github.com/uber/cadence/common/types\"\n\ntype (\n\t// ActionType specifies the type of the Action\n\tActionType int\n\n\t// Action specifies the Action should be performed\n\tAction struct {\n\t\tActionType               ActionType\n\t\tResetActionAttributes    *ResetActionAttributes\n\t\tGetStateActionAttributes *GetStateActionAttributes\n\t\tGetTasksAttributes       *GetTasksAttributes\n\t\tUpdateTaskAttributes     *UpdateTasksAttributes\n\t\t// add attributes for other action types here\n\t}\n\n\t// ActionResult is the result for performing an Action\n\tActionResult struct {\n\t\tActionType           ActionType\n\t\tResetActionResult    *ResetActionResult\n\t\tGetStateActionResult *GetStateActionResult\n\t\tGetTasksResult       *GetTasksResult\n\t\tUpdateTaskResult     *UpdateTasksResult\n\t}\n\n\t// ResetActionAttributes contains the parameter for performing Reset Action\n\tResetActionAttributes struct{}\n\t// ResetActionResult is the result for performing Reset Action\n\tResetActionResult struct{}\n\n\t// GetStateActionAttributes contains the parameter for performing GetState Action\n\tGetStateActionAttributes struct{}\n\t// GetStateActionResult is the result for performing GetState Action\n\tGetStateActionResult struct {\n\t\tStates []ProcessingQueueState\n\t}\n\n\t// GetTasksAttributes contains the parameter to get tasks\n\tGetTasksAttributes struct{}\n\t// GetTasksResult is the result for performing GetTasks Action\n\tGetTasksResult struct {\n\t\tTaskRequests []*types.CrossClusterTaskRequest\n\t}\n\t// UpdateTasksAttributes contains the parameter to update task\n\tUpdateTasksAttributes struct {\n\t\tTaskResponses []*types.CrossClusterTaskResponse\n\t}\n\t// UpdateTasksResult is the result for performing UpdateTask Action\n\tUpdateTasksResult struct {\n\t}\n)\n\nconst (\n\t// ActionTypeReset is the ActionType for reseting processing queue states\n\tActionTypeReset ActionType = iota + 1\n\t// ActionTypeGetState is the ActionType for reading processing queue states\n\tActionTypeGetState\n\t// ActionTypeGetTasks is the ActionType for get cross cluster tasks\n\tActionTypeGetTasks\n\t// ActionTypeUpdateTask is the ActionType to update outstanding task\n\tActionTypeUpdateTask\n\t// add more ActionType here\n)\n\n// NewResetAction creates a new action for reseting processing queue states\nfunc NewResetAction() *Action {\n\treturn &Action{\n\t\tActionType:            ActionTypeReset,\n\t\tResetActionAttributes: &ResetActionAttributes{},\n\t}\n}\n\n// NewGetStateAction reads all processing queue states in the processor\nfunc NewGetStateAction() *Action {\n\treturn &Action{\n\t\tActionType:               ActionTypeGetState,\n\t\tGetStateActionAttributes: &GetStateActionAttributes{},\n\t}\n}\n\n// NewGetTasksAction creates a queue action for fetching cross cluster tasks\nfunc NewGetTasksAction() *Action {\n\treturn &Action{\n\t\tActionType:         ActionTypeGetTasks,\n\t\tGetTasksAttributes: &GetTasksAttributes{},\n\t}\n}\n\n// NewUpdateTasksAction creates a queue action for responding cross cluster task\n// processing results\nfunc NewUpdateTasksAction(\n\ttaskResponses []*types.CrossClusterTaskResponse,\n) *Action {\n\treturn &Action{\n\t\tActionType: ActionTypeUpdateTask,\n\t\tUpdateTaskAttributes: &UpdateTasksAttributes{\n\t\t\tTaskResponses: taskResponses,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/constants.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport \"time\"\n\nconst (\n\tdefaultProcessingQueueLevel = 0\n\n\t// gracefulShutdownTimeout is the the hardcoded timeout for queue components to wrap up shutting down.\n\t// This is not ideal because we should have a top level deadline for shutdown propagating down to all components.\n\tgracefulShutdownTimeout = time.Minute\n)\n"
  },
  {
    "path": "service/history/queue/domain_filter.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\n// NewDomainFilter creates a new domain filter\nfunc NewDomainFilter(\n\tdomainIDs map[string]struct{},\n\treverseMatch bool,\n) DomainFilter {\n\tif domainIDs == nil {\n\t\tdomainIDs = make(map[string]struct{})\n\t}\n\n\treturn DomainFilter{\n\t\tDomainIDs:    domainIDs,\n\t\tReverseMatch: reverseMatch,\n\t}\n}\n\n// Filter returns true if domainID is in the domainID set specified by the filter\nfunc (f DomainFilter) Filter(domainID string) bool {\n\t_, ok := f.DomainIDs[domainID]\n\tif f.ReverseMatch {\n\t\tok = !ok\n\t}\n\treturn ok\n}\n\n// Include adds more domainIDs to the domainID set specified by the filter\nfunc (f DomainFilter) Include(domainIDs map[string]struct{}) DomainFilter {\n\tfilter := f.copy()\n\tfor domainID := range domainIDs {\n\t\tif !filter.ReverseMatch {\n\t\t\tfilter.DomainIDs[domainID] = struct{}{}\n\t\t} else {\n\t\t\tdelete(filter.DomainIDs, domainID)\n\t\t}\n\t}\n\treturn filter\n}\n\n// Exclude removes domainIDs from the domainID set specified by the filter\nfunc (f DomainFilter) Exclude(domainIDs map[string]struct{}) DomainFilter {\n\tfilter := f.copy()\n\tfor domainID := range domainIDs {\n\t\tif !filter.ReverseMatch {\n\t\t\tdelete(filter.DomainIDs, domainID)\n\t\t} else {\n\t\t\tfilter.DomainIDs[domainID] = struct{}{}\n\t\t}\n\t}\n\treturn filter\n}\n\n// Merge merges the domainID sets specified by two domain filters\nfunc (f DomainFilter) Merge(f2 DomainFilter) DomainFilter {\n\t// case 1: ReverseMatch field is false for both filters\n\tif !f.ReverseMatch && !f2.ReverseMatch {\n\t\t// union the domainIDs field\n\t\tfilter := f.copy()\n\t\tfor domainID := range f2.DomainIDs {\n\t\t\tfilter.DomainIDs[domainID] = struct{}{}\n\t\t}\n\t\treturn filter\n\t}\n\n\t// for the following three cases, ReverseMatch field is always true\n\n\t// case 2: ReverseMatch field is true for both filters\n\tif f.ReverseMatch && f2.ReverseMatch {\n\t\t// intersect the domainIDs field\n\t\tfilter := DomainFilter{\n\t\t\tDomainIDs:    make(map[string]struct{}),\n\t\t\tReverseMatch: true,\n\t\t}\n\t\tfor domainID := range f.DomainIDs {\n\t\t\tif _, ok := f2.DomainIDs[domainID]; ok {\n\t\t\t\tfilter.DomainIDs[domainID] = struct{}{}\n\t\t\t}\n\t\t}\n\t\treturn filter\n\t}\n\n\t// case 3, 4: one of the filters has ReverseMatch equals true\n\tvar filter DomainFilter\n\tvar includeDomainIDs map[string]struct{}\n\tif f.ReverseMatch {\n\t\tfilter = f.copy()\n\t\tincludeDomainIDs = f2.DomainIDs\n\t} else {\n\t\tfilter = f2.copy()\n\t\tincludeDomainIDs = f.DomainIDs\n\t}\n\n\tfor domainID := range includeDomainIDs {\n\t\tdelete(filter.DomainIDs, domainID)\n\t}\n\n\treturn filter\n}\n\nfunc (f DomainFilter) copy() DomainFilter {\n\tdomainIDs := make(map[string]struct{})\n\tfor domainID := range f.DomainIDs {\n\t\tdomainIDs[domainID] = struct{}{}\n\t}\n\treturn NewDomainFilter(domainIDs, f.ReverseMatch)\n}\n\nfunc covertToDomainIDSet(\n\tdomainIDs []string,\n) map[string]struct{} {\n\tdomainIDsMap := make(map[string]struct{})\n\tfor _, domainID := range domainIDs {\n\t\tdomainIDsMap[domainID] = struct{}{}\n\t}\n\treturn domainIDsMap\n}\n"
  },
  {
    "path": "service/history/queue/domain_filter_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tdomainFilterSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t}\n)\n\nfunc TestDomainFilterSuite(t *testing.T) {\n\ts := new(domainFilterSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainFilterSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *domainFilterSuite) TestDomainFilter_Filter() {\n\ttestCases := []struct {\n\t\tdomainIDs      map[string]struct{}\n\t\treverseMatch   bool\n\t\ttestDomainIDs  []string\n\t\texpectedResult []bool\n\t}{\n\t\t{\n\t\t\tdomainIDs:      nil,\n\t\t\treverseMatch:   false,\n\t\t\ttestDomainIDs:  []string{\"some random domain\"},\n\t\t\texpectedResult: []bool{false},\n\t\t},\n\t\t{\n\t\t\tdomainIDs:      covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:   false,\n\t\t\ttestDomainIDs:  []string{\"testDomain1\", \"some random domain\"},\n\t\t\texpectedResult: []bool{true, false},\n\t\t},\n\t\t{\n\t\t\tdomainIDs:      covertToDomainIDSet([]string{}),\n\t\t\treverseMatch:   true,\n\t\t\ttestDomainIDs:  []string{\"any domainID\"},\n\t\t\texpectedResult: []bool{true},\n\t\t},\n\t\t{\n\t\t\tdomainIDs:      covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:   true,\n\t\t\ttestDomainIDs:  []string{\"testDomain1\", \"some random domain\"},\n\t\t\texpectedResult: []bool{false, true},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tfilter := NewDomainFilter(tc.domainIDs, tc.reverseMatch)\n\t\tfor i, testDomain := range tc.testDomainIDs {\n\t\t\tresult := filter.Filter(testDomain)\n\t\t\ts.Equal(tc.expectedResult[i], result)\n\t\t}\n\t}\n}\n\nfunc (s *domainFilterSuite) TestDomainFilter_Include() {\n\ttestCases := []struct {\n\t\tdomainIDs         map[string]struct{}\n\t\treverseMatch      bool\n\t\tnewDomainIDs      map[string]struct{}\n\t\texpectedDomainIDs map[string]struct{}\n\t}{\n\t\t{\n\t\t\tdomainIDs:         covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:      false,\n\t\t\tnewDomainIDs:      covertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\texpectedDomainIDs: covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\", \"testDomain3\"}),\n\t\t},\n\t\t{\n\t\t\tdomainIDs:         covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:      true,\n\t\t\tnewDomainIDs:      covertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\texpectedDomainIDs: covertToDomainIDSet([]string{\"testDomain1\"}),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tbaseFilter := NewDomainFilter(tc.domainIDs, tc.reverseMatch)\n\t\tnewFilter := baseFilter.Include(tc.newDomainIDs)\n\n\t\t// check if the base filter got modified\n\t\ts.Equal(tc.domainIDs, baseFilter.DomainIDs)\n\t\ts.Equal(tc.reverseMatch, baseFilter.ReverseMatch)\n\n\t\ts.Equal(tc.expectedDomainIDs, newFilter.DomainIDs)\n\t\ts.Equal(tc.reverseMatch, newFilter.ReverseMatch)\n\t}\n}\n\nfunc (s *domainFilterSuite) TestDomainFilter_Exclude() {\n\ttestCases := []struct {\n\t\tdomainIDs         map[string]struct{}\n\t\treverseMatch      bool\n\t\tnewDomainIDs      map[string]struct{}\n\t\texpectedDomainIDs map[string]struct{}\n\t}{\n\t\t{\n\t\t\tdomainIDs:         covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:      false,\n\t\t\tnewDomainIDs:      covertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\texpectedDomainIDs: covertToDomainIDSet([]string{\"testDomain1\"}),\n\t\t},\n\t\t{\n\t\t\tdomainIDs:         covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\treverseMatch:      true,\n\t\t\tnewDomainIDs:      covertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\texpectedDomainIDs: covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\", \"testDomain3\"}),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tbaseFilter := NewDomainFilter(tc.domainIDs, tc.reverseMatch)\n\t\tnewFilter := baseFilter.Exclude(tc.newDomainIDs)\n\n\t\t// check if the base filter got modified\n\t\ts.Equal(tc.domainIDs, baseFilter.DomainIDs)\n\t\ts.Equal(tc.reverseMatch, baseFilter.ReverseMatch)\n\n\t\ts.Equal(tc.expectedDomainIDs, newFilter.DomainIDs)\n\t\ts.Equal(tc.reverseMatch, newFilter.ReverseMatch)\n\t}\n}\n\nfunc (s *domainFilterSuite) TestDomainFilter_Merge() {\n\ttestCases := []struct {\n\t\tdomainIDs            []map[string]struct{}\n\t\treverseMatch         []bool\n\t\texpectedDomainIDs    map[string]struct{}\n\t\texpectedReverseMatch bool\n\t}{\n\t\t{\n\t\t\tdomainIDs: []map[string]struct{}{\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\t},\n\t\t\treverseMatch:         []bool{false, false},\n\t\t\texpectedDomainIDs:    covertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\", \"testDomain3\"}),\n\t\t\texpectedReverseMatch: false,\n\t\t},\n\t\t{\n\t\t\tdomainIDs: []map[string]struct{}{\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\t},\n\t\t\treverseMatch:         []bool{true, true},\n\t\t\texpectedDomainIDs:    covertToDomainIDSet([]string{\"testDomain2\"}),\n\t\t\texpectedReverseMatch: true,\n\t\t},\n\t\t{\n\t\t\tdomainIDs: []map[string]struct{}{\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\t},\n\t\t\treverseMatch:         []bool{true, false},\n\t\t\texpectedDomainIDs:    covertToDomainIDSet([]string{\"testDomain1\"}),\n\t\t\texpectedReverseMatch: true,\n\t\t},\n\t\t{\n\t\t\tdomainIDs: []map[string]struct{}{\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain1\", \"testDomain2\"}),\n\t\t\t\tcovertToDomainIDSet([]string{\"testDomain2\", \"testDomain3\"}),\n\t\t\t},\n\t\t\treverseMatch:         []bool{false, true},\n\t\t\texpectedDomainIDs:    covertToDomainIDSet([]string{\"testDomain3\"}),\n\t\t\texpectedReverseMatch: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tvar filters []DomainFilter\n\t\tfor i, domainIDs := range tc.domainIDs {\n\t\t\tfilters = append(filters, NewDomainFilter(domainIDs, tc.reverseMatch[i]))\n\t\t}\n\n\t\ts.NotEmpty(filters)\n\t\tmergedFilter := filters[0]\n\t\tfor _, f := range filters[1:] {\n\t\t\tmergedFilter = mergedFilter.Merge(f)\n\t\t}\n\n\t\ts.Equal(tc.expectedDomainIDs, mergedFilter.DomainIDs)\n\t\ts.Equal(tc.expectedReverseMatch, mergedFilter.ReverseMatch)\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/factory.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination factory_mock.go -self_package github.com/uber/cadence/service/history/queue\n\ntype (\n\tFactory interface {\n\t\tCategory() persistence.HistoryTaskCategory\n\t\tCreateQueue(shard.Context, execution.Cache, invariant.Invariant) Processor\n\t}\n\n\ttransferQueueFactory struct {\n\t\ttaskProcessor  task.Processor\n\t\tarchivalClient archiver.Client\n\t\twfIDCache      workflowcache.WFCache\n\t}\n\n\ttimerQueueFactory struct {\n\t\ttaskProcessor  task.Processor\n\t\tarchivalClient archiver.Client\n\t}\n)\n\nfunc NewTransferQueueFactory(\n\ttaskProcessor task.Processor,\n\tarchivalClient archiver.Client,\n\twfIDCache workflowcache.WFCache,\n) Factory {\n\treturn &transferQueueFactory{\n\t\ttaskProcessor:  taskProcessor,\n\t\tarchivalClient: archivalClient,\n\t\twfIDCache:      wfIDCache,\n\t}\n}\n\nfunc (f *transferQueueFactory) Category() persistence.HistoryTaskCategory {\n\treturn persistence.HistoryTaskCategoryTransfer\n}\n\nfunc (f *transferQueueFactory) CreateQueue(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) Processor {\n\tworkflowResetter := reset.NewWorkflowResetter(shard, executionCache, shard.GetLogger())\n\treturn NewTransferQueueProcessor(\n\t\tshard,\n\t\tf.taskProcessor,\n\t\texecutionCache,\n\t\tworkflowResetter,\n\t\tf.archivalClient,\n\t\topenExecutionCheck,\n\t\tf.wfIDCache,\n\t)\n}\n\nfunc (f *timerQueueFactory) Category() persistence.HistoryTaskCategory {\n\treturn persistence.HistoryTaskCategoryTimer\n}\n\nfunc NewTimerQueueFactory(\n\ttaskProcessor task.Processor,\n\tarchivalClient archiver.Client,\n) Factory {\n\treturn &timerQueueFactory{\n\t\ttaskProcessor:  taskProcessor,\n\t\tarchivalClient: archivalClient,\n\t}\n}\n\nfunc (f *timerQueueFactory) CreateQueue(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) Processor {\n\treturn NewTimerQueueProcessor(\n\t\tshard,\n\t\tf.taskProcessor,\n\t\texecutionCache,\n\t\tf.archivalClient,\n\t\topenExecutionCheck,\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/factory_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: factory.go\n//\n// Generated by this command:\n//\n//\tmockgen -package queue -source factory.go -destination factory_mock.go -self_package github.com/uber/cadence/service/history/queue\n//\n\n// Package queue is a generated GoMock package.\npackage queue\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\tinvariant \"github.com/uber/cadence/common/reconciliation/invariant\"\n\texecution \"github.com/uber/cadence/service/history/execution\"\n\tshard \"github.com/uber/cadence/service/history/shard\"\n)\n\n// MockFactory is a mock of Factory interface.\ntype MockFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockFactoryMockRecorder is the mock recorder for MockFactory.\ntype MockFactoryMockRecorder struct {\n\tmock *MockFactory\n}\n\n// NewMockFactory creates a new mock instance.\nfunc NewMockFactory(ctrl *gomock.Controller) *MockFactory {\n\tmock := &MockFactory{ctrl: ctrl}\n\tmock.recorder = &MockFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFactory) EXPECT() *MockFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// Category mocks base method.\nfunc (m *MockFactory) Category() persistence.HistoryTaskCategory {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Category\")\n\tret0, _ := ret[0].(persistence.HistoryTaskCategory)\n\treturn ret0\n}\n\n// Category indicates an expected call of Category.\nfunc (mr *MockFactoryMockRecorder) Category() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Category\", reflect.TypeOf((*MockFactory)(nil).Category))\n}\n\n// CreateQueue mocks base method.\nfunc (m *MockFactory) CreateQueue(arg0 shard.Context, arg1 execution.Cache, arg2 invariant.Invariant) Processor {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateQueue\", arg0, arg1, arg2)\n\tret0, _ := ret[0].(Processor)\n\treturn ret0\n}\n\n// CreateQueue indicates an expected call of CreateQueue.\nfunc (mr *MockFactoryMockRecorder) CreateQueue(arg0, arg1, arg2 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateQueue\", reflect.TypeOf((*MockFactory)(nil).CreateQueue), arg0, arg1, arg2)\n}\n"
  },
  {
    "path": "service/history/queue/factory_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nfunc TestTransferQueueFactory(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\tdefer mockShard.Finish(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockArchiver := archiver.NewMockClient(ctrl)\n\tmockInvariant := invariant.NewMockInvariant(ctrl)\n\tmockWorkflowCache := workflowcache.NewMockWFCache(ctrl)\n\n\tf := NewTransferQueueFactory(mockProcessor, mockArchiver, mockWorkflowCache)\n\n\tprocessor := f.CreateQueue(mockShard, execution.NewCache(mockShard), mockInvariant)\n\n\tif processor == nil {\n\t\tt.Error(\"NewTransferQueueProcessor returned nil\")\n\t}\n}\n\nfunc TestTimerQueueFactory(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\tdefer mockShard.Finish(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockArchiver := archiver.NewMockClient(ctrl)\n\tmockInvariant := invariant.NewMockInvariant(ctrl)\n\n\tf := NewTimerQueueFactory(mockProcessor, mockArchiver)\n\tprocessor := f.CreateQueue(mockShard, execution.NewCache(mockShard), mockInvariant)\n\n\tif processor == nil {\n\t\tt.Error(\"NewTimerQueueProcessor returned nil\")\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/interface.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/service/history/queue\n\npackage queue\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\t// DomainFilter filters domain\n\tDomainFilter struct {\n\t\tDomainIDs map[string]struct{}\n\t\t// by default, a DomainFilter matches domains listed in the Domains field\n\t\t// if reverseMatch is true then the DomainFilter matches domains that are\n\t\t// not in the Domains field.\n\t\tReverseMatch bool\n\t}\n\n\t// ProcessingQueueState indicates the scope of a task processing queue and its current progress\n\tProcessingQueueState interface {\n\t\tLevel() int\n\t\tAckLevel() task.Key\n\t\tReadLevel() task.Key\n\t\tMaxLevel() task.Key\n\t\tDomainFilter() DomainFilter\n\t}\n\n\t// ProcessingQueue is responsible for keeping track of the state of tasks\n\t// within the scope defined by its state; it can also be split into multiple\n\t// ProcessingQueues with non-overlapping scope or be merged with another\n\t// ProcessingQueue\n\tProcessingQueue interface {\n\t\tState() ProcessingQueueState\n\t\tSplit(ProcessingQueueSplitPolicy) []ProcessingQueue\n\t\tMerge(ProcessingQueue) []ProcessingQueue\n\t\tAddTasks(map[task.Key]task.Task, task.Key)\n\t\tGetTask(task.Key) (task.Task, error)\n\t\tGetTasks() []task.Task\n\t\tUpdateAckLevel() (task.Key, int) // return new ack level and number of pending tasks\n\t\t// TODO: add Offload() method\n\t}\n\n\t// ProcessingQueueSplitPolicy determines if one ProcessingQueue should be split\n\t// into multiple ProcessingQueues\n\tProcessingQueueSplitPolicy interface {\n\t\tEvaluate(ProcessingQueue) []ProcessingQueueState\n\t}\n\n\t// ProcessingQueueCollection manages a list of non-overlapping ProcessingQueues\n\t// and keep track of the current active ProcessingQueue\n\tProcessingQueueCollection interface {\n\t\tLevel() int\n\t\tQueues() []ProcessingQueue\n\t\tActiveQueue() ProcessingQueue\n\t\tAddTasks(map[task.Key]task.Task, task.Key)\n\t\tGetTask(task.Key) (task.Task, error)\n\t\tGetTasks() []task.Task\n\t\tUpdateAckLevels() (task.Key, int) // return min of all new ack levels and number of total pending tasks\n\t\tSplit(ProcessingQueueSplitPolicy) []ProcessingQueue\n\t\tMerge([]ProcessingQueue)\n\t\t// TODO: add Offload() method\n\t}\n\n\t// Processor is the interface for task queue processor\n\tProcessor interface {\n\t\tcommon.Daemon\n\t\tFailoverDomain(domainIDs map[string]struct{})\n\t\tNotifyNewTask(clusterName string, info *hcommon.NotifyTaskInfo)\n\t\tHandleAction(ctx context.Context, clusterName string, action *Action) (*ActionResult, error)\n\t\tLockTaskProcessing()\n\t\tUnlockTaskProcessing()\n\t}\n)\n"
  },
  {
    "path": "service/history/queue/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package queue -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/service/history/queue\n//\n\n// Package queue is a generated GoMock package.\npackage queue\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tcommon \"github.com/uber/cadence/service/history/common\"\n\ttask \"github.com/uber/cadence/service/history/task\"\n)\n\n// MockProcessingQueueState is a mock of ProcessingQueueState interface.\ntype MockProcessingQueueState struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessingQueueStateMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessingQueueStateMockRecorder is the mock recorder for MockProcessingQueueState.\ntype MockProcessingQueueStateMockRecorder struct {\n\tmock *MockProcessingQueueState\n}\n\n// NewMockProcessingQueueState creates a new mock instance.\nfunc NewMockProcessingQueueState(ctrl *gomock.Controller) *MockProcessingQueueState {\n\tmock := &MockProcessingQueueState{ctrl: ctrl}\n\tmock.recorder = &MockProcessingQueueStateMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessingQueueState) EXPECT() *MockProcessingQueueStateMockRecorder {\n\treturn m.recorder\n}\n\n// AckLevel mocks base method.\nfunc (m *MockProcessingQueueState) AckLevel() task.Key {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AckLevel\")\n\tret0, _ := ret[0].(task.Key)\n\treturn ret0\n}\n\n// AckLevel indicates an expected call of AckLevel.\nfunc (mr *MockProcessingQueueStateMockRecorder) AckLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AckLevel\", reflect.TypeOf((*MockProcessingQueueState)(nil).AckLevel))\n}\n\n// DomainFilter mocks base method.\nfunc (m *MockProcessingQueueState) DomainFilter() DomainFilter {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DomainFilter\")\n\tret0, _ := ret[0].(DomainFilter)\n\treturn ret0\n}\n\n// DomainFilter indicates an expected call of DomainFilter.\nfunc (mr *MockProcessingQueueStateMockRecorder) DomainFilter() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DomainFilter\", reflect.TypeOf((*MockProcessingQueueState)(nil).DomainFilter))\n}\n\n// Level mocks base method.\nfunc (m *MockProcessingQueueState) Level() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Level\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Level indicates an expected call of Level.\nfunc (mr *MockProcessingQueueStateMockRecorder) Level() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Level\", reflect.TypeOf((*MockProcessingQueueState)(nil).Level))\n}\n\n// MaxLevel mocks base method.\nfunc (m *MockProcessingQueueState) MaxLevel() task.Key {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MaxLevel\")\n\tret0, _ := ret[0].(task.Key)\n\treturn ret0\n}\n\n// MaxLevel indicates an expected call of MaxLevel.\nfunc (mr *MockProcessingQueueStateMockRecorder) MaxLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MaxLevel\", reflect.TypeOf((*MockProcessingQueueState)(nil).MaxLevel))\n}\n\n// ReadLevel mocks base method.\nfunc (m *MockProcessingQueueState) ReadLevel() task.Key {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadLevel\")\n\tret0, _ := ret[0].(task.Key)\n\treturn ret0\n}\n\n// ReadLevel indicates an expected call of ReadLevel.\nfunc (mr *MockProcessingQueueStateMockRecorder) ReadLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadLevel\", reflect.TypeOf((*MockProcessingQueueState)(nil).ReadLevel))\n}\n\n// MockProcessingQueue is a mock of ProcessingQueue interface.\ntype MockProcessingQueue struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessingQueueMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessingQueueMockRecorder is the mock recorder for MockProcessingQueue.\ntype MockProcessingQueueMockRecorder struct {\n\tmock *MockProcessingQueue\n}\n\n// NewMockProcessingQueue creates a new mock instance.\nfunc NewMockProcessingQueue(ctrl *gomock.Controller) *MockProcessingQueue {\n\tmock := &MockProcessingQueue{ctrl: ctrl}\n\tmock.recorder = &MockProcessingQueueMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessingQueue) EXPECT() *MockProcessingQueueMockRecorder {\n\treturn m.recorder\n}\n\n// AddTasks mocks base method.\nfunc (m *MockProcessingQueue) AddTasks(arg0 map[task.Key]task.Task, arg1 task.Key) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddTasks\", arg0, arg1)\n}\n\n// AddTasks indicates an expected call of AddTasks.\nfunc (mr *MockProcessingQueueMockRecorder) AddTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTasks\", reflect.TypeOf((*MockProcessingQueue)(nil).AddTasks), arg0, arg1)\n}\n\n// GetTask mocks base method.\nfunc (m *MockProcessingQueue) GetTask(arg0 task.Key) (task.Task, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTask\", arg0)\n\tret0, _ := ret[0].(task.Task)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTask indicates an expected call of GetTask.\nfunc (mr *MockProcessingQueueMockRecorder) GetTask(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTask\", reflect.TypeOf((*MockProcessingQueue)(nil).GetTask), arg0)\n}\n\n// GetTasks mocks base method.\nfunc (m *MockProcessingQueue) GetTasks() []task.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\")\n\tret0, _ := ret[0].([]task.Task)\n\treturn ret0\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockProcessingQueueMockRecorder) GetTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockProcessingQueue)(nil).GetTasks))\n}\n\n// Merge mocks base method.\nfunc (m *MockProcessingQueue) Merge(arg0 ProcessingQueue) []ProcessingQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Merge\", arg0)\n\tret0, _ := ret[0].([]ProcessingQueue)\n\treturn ret0\n}\n\n// Merge indicates an expected call of Merge.\nfunc (mr *MockProcessingQueueMockRecorder) Merge(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Merge\", reflect.TypeOf((*MockProcessingQueue)(nil).Merge), arg0)\n}\n\n// Split mocks base method.\nfunc (m *MockProcessingQueue) Split(arg0 ProcessingQueueSplitPolicy) []ProcessingQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Split\", arg0)\n\tret0, _ := ret[0].([]ProcessingQueue)\n\treturn ret0\n}\n\n// Split indicates an expected call of Split.\nfunc (mr *MockProcessingQueueMockRecorder) Split(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Split\", reflect.TypeOf((*MockProcessingQueue)(nil).Split), arg0)\n}\n\n// State mocks base method.\nfunc (m *MockProcessingQueue) State() ProcessingQueueState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"State\")\n\tret0, _ := ret[0].(ProcessingQueueState)\n\treturn ret0\n}\n\n// State indicates an expected call of State.\nfunc (mr *MockProcessingQueueMockRecorder) State() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"State\", reflect.TypeOf((*MockProcessingQueue)(nil).State))\n}\n\n// UpdateAckLevel mocks base method.\nfunc (m *MockProcessingQueue) UpdateAckLevel() (task.Key, int) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevel\")\n\tret0, _ := ret[0].(task.Key)\n\tret1, _ := ret[1].(int)\n\treturn ret0, ret1\n}\n\n// UpdateAckLevel indicates an expected call of UpdateAckLevel.\nfunc (mr *MockProcessingQueueMockRecorder) UpdateAckLevel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevel\", reflect.TypeOf((*MockProcessingQueue)(nil).UpdateAckLevel))\n}\n\n// MockProcessingQueueSplitPolicy is a mock of ProcessingQueueSplitPolicy interface.\ntype MockProcessingQueueSplitPolicy struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessingQueueSplitPolicyMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessingQueueSplitPolicyMockRecorder is the mock recorder for MockProcessingQueueSplitPolicy.\ntype MockProcessingQueueSplitPolicyMockRecorder struct {\n\tmock *MockProcessingQueueSplitPolicy\n}\n\n// NewMockProcessingQueueSplitPolicy creates a new mock instance.\nfunc NewMockProcessingQueueSplitPolicy(ctrl *gomock.Controller) *MockProcessingQueueSplitPolicy {\n\tmock := &MockProcessingQueueSplitPolicy{ctrl: ctrl}\n\tmock.recorder = &MockProcessingQueueSplitPolicyMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessingQueueSplitPolicy) EXPECT() *MockProcessingQueueSplitPolicyMockRecorder {\n\treturn m.recorder\n}\n\n// Evaluate mocks base method.\nfunc (m *MockProcessingQueueSplitPolicy) Evaluate(arg0 ProcessingQueue) []ProcessingQueueState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Evaluate\", arg0)\n\tret0, _ := ret[0].([]ProcessingQueueState)\n\treturn ret0\n}\n\n// Evaluate indicates an expected call of Evaluate.\nfunc (mr *MockProcessingQueueSplitPolicyMockRecorder) Evaluate(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Evaluate\", reflect.TypeOf((*MockProcessingQueueSplitPolicy)(nil).Evaluate), arg0)\n}\n\n// MockProcessingQueueCollection is a mock of ProcessingQueueCollection interface.\ntype MockProcessingQueueCollection struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessingQueueCollectionMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessingQueueCollectionMockRecorder is the mock recorder for MockProcessingQueueCollection.\ntype MockProcessingQueueCollectionMockRecorder struct {\n\tmock *MockProcessingQueueCollection\n}\n\n// NewMockProcessingQueueCollection creates a new mock instance.\nfunc NewMockProcessingQueueCollection(ctrl *gomock.Controller) *MockProcessingQueueCollection {\n\tmock := &MockProcessingQueueCollection{ctrl: ctrl}\n\tmock.recorder = &MockProcessingQueueCollectionMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessingQueueCollection) EXPECT() *MockProcessingQueueCollectionMockRecorder {\n\treturn m.recorder\n}\n\n// ActiveQueue mocks base method.\nfunc (m *MockProcessingQueueCollection) ActiveQueue() ProcessingQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ActiveQueue\")\n\tret0, _ := ret[0].(ProcessingQueue)\n\treturn ret0\n}\n\n// ActiveQueue indicates an expected call of ActiveQueue.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) ActiveQueue() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ActiveQueue\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).ActiveQueue))\n}\n\n// AddTasks mocks base method.\nfunc (m *MockProcessingQueueCollection) AddTasks(arg0 map[task.Key]task.Task, arg1 task.Key) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddTasks\", arg0, arg1)\n}\n\n// AddTasks indicates an expected call of AddTasks.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) AddTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTasks\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).AddTasks), arg0, arg1)\n}\n\n// GetTask mocks base method.\nfunc (m *MockProcessingQueueCollection) GetTask(arg0 task.Key) (task.Task, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTask\", arg0)\n\tret0, _ := ret[0].(task.Task)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTask indicates an expected call of GetTask.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) GetTask(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTask\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).GetTask), arg0)\n}\n\n// GetTasks mocks base method.\nfunc (m *MockProcessingQueueCollection) GetTasks() []task.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\")\n\tret0, _ := ret[0].([]task.Task)\n\treturn ret0\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) GetTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).GetTasks))\n}\n\n// Level mocks base method.\nfunc (m *MockProcessingQueueCollection) Level() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Level\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Level indicates an expected call of Level.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) Level() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Level\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Level))\n}\n\n// Merge mocks base method.\nfunc (m *MockProcessingQueueCollection) Merge(arg0 []ProcessingQueue) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Merge\", arg0)\n}\n\n// Merge indicates an expected call of Merge.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) Merge(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Merge\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Merge), arg0)\n}\n\n// Queues mocks base method.\nfunc (m *MockProcessingQueueCollection) Queues() []ProcessingQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Queues\")\n\tret0, _ := ret[0].([]ProcessingQueue)\n\treturn ret0\n}\n\n// Queues indicates an expected call of Queues.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) Queues() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Queues\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Queues))\n}\n\n// Split mocks base method.\nfunc (m *MockProcessingQueueCollection) Split(arg0 ProcessingQueueSplitPolicy) []ProcessingQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Split\", arg0)\n\tret0, _ := ret[0].([]ProcessingQueue)\n\treturn ret0\n}\n\n// Split indicates an expected call of Split.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) Split(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Split\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).Split), arg0)\n}\n\n// UpdateAckLevels mocks base method.\nfunc (m *MockProcessingQueueCollection) UpdateAckLevels() (task.Key, int) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAckLevels\")\n\tret0, _ := ret[0].(task.Key)\n\tret1, _ := ret[1].(int)\n\treturn ret0, ret1\n}\n\n// UpdateAckLevels indicates an expected call of UpdateAckLevels.\nfunc (mr *MockProcessingQueueCollectionMockRecorder) UpdateAckLevels() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAckLevels\", reflect.TypeOf((*MockProcessingQueueCollection)(nil).UpdateAckLevels))\n}\n\n// MockProcessor is a mock of Processor interface.\ntype MockProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessorMockRecorder is the mock recorder for MockProcessor.\ntype MockProcessorMockRecorder struct {\n\tmock *MockProcessor\n}\n\n// NewMockProcessor creates a new mock instance.\nfunc NewMockProcessor(ctrl *gomock.Controller) *MockProcessor {\n\tmock := &MockProcessor{ctrl: ctrl}\n\tmock.recorder = &MockProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessor) EXPECT() *MockProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// FailoverDomain mocks base method.\nfunc (m *MockProcessor) FailoverDomain(domainIDs map[string]struct{}) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"FailoverDomain\", domainIDs)\n}\n\n// FailoverDomain indicates an expected call of FailoverDomain.\nfunc (mr *MockProcessorMockRecorder) FailoverDomain(domainIDs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"FailoverDomain\", reflect.TypeOf((*MockProcessor)(nil).FailoverDomain), domainIDs)\n}\n\n// HandleAction mocks base method.\nfunc (m *MockProcessor) HandleAction(ctx context.Context, clusterName string, action *Action) (*ActionResult, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleAction\", ctx, clusterName, action)\n\tret0, _ := ret[0].(*ActionResult)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// HandleAction indicates an expected call of HandleAction.\nfunc (mr *MockProcessorMockRecorder) HandleAction(ctx, clusterName, action any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleAction\", reflect.TypeOf((*MockProcessor)(nil).HandleAction), ctx, clusterName, action)\n}\n\n// LockTaskProcessing mocks base method.\nfunc (m *MockProcessor) LockTaskProcessing() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"LockTaskProcessing\")\n}\n\n// LockTaskProcessing indicates an expected call of LockTaskProcessing.\nfunc (mr *MockProcessorMockRecorder) LockTaskProcessing() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LockTaskProcessing\", reflect.TypeOf((*MockProcessor)(nil).LockTaskProcessing))\n}\n\n// NotifyNewTask mocks base method.\nfunc (m *MockProcessor) NotifyNewTask(clusterName string, info *common.NotifyTaskInfo) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"NotifyNewTask\", clusterName, info)\n}\n\n// NotifyNewTask indicates an expected call of NotifyNewTask.\nfunc (mr *MockProcessorMockRecorder) NotifyNewTask(clusterName, info any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NotifyNewTask\", reflect.TypeOf((*MockProcessor)(nil).NotifyNewTask), clusterName, info)\n}\n\n// Start mocks base method.\nfunc (m *MockProcessor) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockProcessorMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockProcessor)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockProcessor)(nil).Stop))\n}\n\n// UnlockTaskProcessing mocks base method.\nfunc (m *MockProcessor) UnlockTaskProcessing() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"UnlockTaskProcessing\")\n}\n\n// UnlockTaskProcessing indicates an expected call of UnlockTaskProcessing.\nfunc (mr *MockProcessorMockRecorder) UnlockTaskProcessing() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UnlockTaskProcessing\", reflect.TypeOf((*MockProcessor)(nil).UnlockTaskProcessing))\n}\n"
  },
  {
    "path": "service/history/queue/main_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"os\"\n\t\"testing\"\n\n\t\"go.uber.org/goleak\"\n)\n\nfunc TestMain(m *testing.M) {\n\tdefer goleak.VerifyTestMain(m)\n\n\tos.Exit(m.Run())\n}\n"
  },
  {
    "path": "service/history/queue/processing_queue.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tt \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nvar (\n\terrTaskNotFound = errors.New(\"task not found\")\n)\n\ntype (\n\tprocessingQueueImpl struct {\n\t\tstate            *processingQueueStateImpl\n\t\toutstandingTasks map[task.Key]task.Task\n\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client // TODO: emit metrics\n\t}\n)\n\n// NewProcessingQueue creates a new processing queue based on its state\nfunc NewProcessingQueue(\n\tstate ProcessingQueueState,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) ProcessingQueue {\n\treturn newProcessingQueue(\n\t\tstate,\n\t\tnil,\n\t\tlogger,\n\t\tmetricsClient,\n\t)\n}\n\nfunc newProcessingQueue(\n\tstate ProcessingQueueState,\n\toutstandingTasks map[task.Key]task.Task,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) *processingQueueImpl {\n\tif outstandingTasks == nil {\n\t\toutstandingTasks = make(map[task.Key]task.Task)\n\t}\n\n\tqueue := &processingQueueImpl{\n\t\toutstandingTasks: outstandingTasks,\n\t\tlogger:           logger,\n\t\tmetricsClient:    metricsClient,\n\t}\n\n\t// convert state to *processingQueueStateImpl type so that\n\t// queue implementation can change the state value\n\tif stateImpl, ok := state.(*processingQueueStateImpl); ok {\n\t\tqueue.state = stateImpl\n\t} else {\n\t\tqueue.state = copyQueueState(state)\n\t}\n\n\tif queue.state.readLevel.Less(queue.state.ackLevel) {\n\t\tlogger.Fatal(\"ack level larger than readlevel when creating processing queue\", tag.Error(\n\t\t\tfmt.Errorf(\"ack level: %v, read level: %v\", queue.state.ackLevel, queue.state.readLevel),\n\t\t))\n\t}\n\n\treturn queue\n}\n\nfunc (q *processingQueueImpl) State() ProcessingQueueState {\n\treturn q.state\n}\n\nfunc (q *processingQueueImpl) Split(policy ProcessingQueueSplitPolicy) []ProcessingQueue {\n\tnewQueueStates := policy.Evaluate(q)\n\tif len(newQueueStates) == 0 {\n\t\t// no need to split, return self\n\t\treturn []ProcessingQueue{q}\n\t}\n\n\treturn splitProcessingQueue([]*processingQueueImpl{q}, newQueueStates, q.logger, q.metricsClient)\n}\n\nfunc (q *processingQueueImpl) Merge(queue ProcessingQueue) []ProcessingQueue {\n\tq1, q2 := q, queue.(*processingQueueImpl)\n\n\tif q1.State().Level() != q2.State().Level() {\n\t\terrMsg := \"Processing queue encountered a queue from different level during merge\"\n\t\tq.logger.Error(errMsg, tag.Error(\n\t\t\tfmt.Errorf(\"current queue level: %v, incoming queue level: %v\", q1.state.level, q2.state.level),\n\t\t))\n\t\tpanic(errMsg)\n\t}\n\n\tif !q1.state.ackLevel.Less(q2.state.maxLevel) ||\n\t\t!q2.state.ackLevel.Less(q1.state.maxLevel) {\n\t\t// one queue's ackLevel is larger or equal than the other one's maxLevel\n\t\t// this means there's no overlap between two queues\n\t\treturn []ProcessingQueue{q1, q2}\n\t}\n\n\t// generate new queue states for merged queues\n\tnewQueueStates := []ProcessingQueueState{}\n\tif !taskKeyEquals(q1.state.ackLevel, q2.state.ackLevel) {\n\t\tif q2.state.ackLevel.Less(q1.state.ackLevel) {\n\t\t\tq1, q2 = q2, q1\n\t\t}\n\n\t\tnewQueueStates = append(newQueueStates, newProcessingQueueState(\n\t\t\tq1.state.level,\n\t\t\tq1.state.ackLevel,\n\t\t\tminTaskKey(q1.state.readLevel, q2.state.ackLevel),\n\t\t\tq2.state.ackLevel,\n\t\t\tq1.state.domainFilter.copy(),\n\t\t))\n\t}\n\n\tif !taskKeyEquals(q1.state.maxLevel, q2.state.maxLevel) {\n\t\tif q1.state.maxLevel.Less(q2.state.maxLevel) {\n\t\t\tq1, q2 = q2, q1\n\t\t}\n\n\t\tnewQueueStates = append(newQueueStates, newProcessingQueueState(\n\t\t\tq1.state.level,\n\t\t\tq2.state.maxLevel,\n\t\t\tmaxTaskKey(q1.state.readLevel, q2.state.maxLevel),\n\t\t\tq1.state.maxLevel,\n\t\t\tq1.state.domainFilter.copy(),\n\t\t))\n\t}\n\n\toverlappingQueueAckLevel := maxTaskKey(q1.state.ackLevel, q2.state.ackLevel)\n\tnewQueueStates = append(newQueueStates, newProcessingQueueState(\n\t\tq1.state.level,\n\t\toverlappingQueueAckLevel,\n\t\tmaxTaskKey(minTaskKey(q1.state.readLevel, q2.state.readLevel), overlappingQueueAckLevel),\n\t\tminTaskKey(q1.state.maxLevel, q2.state.maxLevel),\n\t\tq1.state.domainFilter.Merge(q2.state.domainFilter),\n\t))\n\n\tfor _, state := range newQueueStates {\n\t\tif state.ReadLevel().Less(state.AckLevel()) || state.MaxLevel().Less(state.ReadLevel()) {\n\t\t\tq.logger.Fatal(\"invalid processing queue merge result\", tag.Error(\n\t\t\t\tfmt.Errorf(\"q1: %v, q2: %v, merge result: %v\", q1.state, q2.state, newQueueStates),\n\t\t\t))\n\t\t}\n\t}\n\n\treturn splitProcessingQueue([]*processingQueueImpl{q1, q2}, newQueueStates, q.logger, q.metricsClient)\n}\n\nfunc (q *processingQueueImpl) AddTasks(tasks map[task.Key]task.Task, newReadLevel task.Key) {\n\tif newReadLevel.Less(q.state.readLevel) {\n\t\tq.logger.Fatal(\"processing queue read level moved backward\", tag.Error(\n\t\t\tfmt.Errorf(\"current read level: %v, new read level: %v\", q.state.readLevel, newReadLevel),\n\t\t))\n\t}\n\n\tfor key, task := range tasks {\n\t\tif _, loaded := q.outstandingTasks[key]; loaded {\n\t\t\t// TODO: this means the task has been submitted before, we should mark the task state accordingly and\n\t\t\t// do not submit this task again in transfer/timer queue processor base\n\t\t\tq.logger.Debug(fmt.Sprintf(\"Skipping task: %+v. DomainID: %v, WorkflowID: %v, RunID: %v, Type: %v\",\n\t\t\t\tkey, task.GetDomainID(), task.GetWorkflowID(), task.GetRunID(), task.GetTaskType()))\n\t\t\tcontinue\n\t\t}\n\n\t\tif !taskBelongsToProcessQueue(q.state, key, task) {\n\t\t\terrMsg := \"Processing queue encountered a task doesn't belong to its scope\"\n\t\t\tq.logger.Error(errMsg, tag.Error(\n\t\t\t\tfmt.Errorf(\"processing queue state: %+v, key: %+v, task: %+v\", q.state, key, task),\n\t\t\t))\n\t\t\tcontinue\n\t\t}\n\n\t\tq.outstandingTasks[key] = task\n\t}\n\n\tq.state.readLevel = newReadLevel\n}\n\nfunc (q *processingQueueImpl) GetTask(key task.Key) (task.Task, error) {\n\tif task, ok := q.outstandingTasks[key]; ok {\n\t\treturn task, nil\n\t}\n\treturn nil, errTaskNotFound\n}\n\nfunc (q *processingQueueImpl) GetTasks() []task.Task {\n\tvar outstandingTask []task.Task\n\tfor _, task := range q.outstandingTasks {\n\t\toutstandingTask = append(outstandingTask, task)\n\t}\n\treturn outstandingTask\n}\n\nfunc (q *processingQueueImpl) UpdateAckLevel() (task.Key, int) {\n\tkeys := make([]task.Key, 0, len(q.outstandingTasks))\n\tfor key := range q.outstandingTasks {\n\t\tkeys = append(keys, key)\n\t}\n\n\tsort.Slice(keys, func(i, j int) bool {\n\t\treturn keys[i].Less(keys[j])\n\t})\n\n\tvar idx int\n\tvar key task.Key\n\tfor idx, key = range keys {\n\t\tif q.state.readLevel.Less(key) {\n\t\t\t// this can happen as during merge read level can move backward.\n\t\t\t// besides that, for timer task key, readLevel is expected to be less than task key\n\t\t\t// as the taskID for read level is always 0. This means we can potentially buffer\n\t\t\t// more timer tasks in memory. If this becomes a problem, we can change this logic.\n\t\t\tbreak\n\t\t}\n\n\t\tif q.outstandingTasks[key].State() != t.TaskStateAcked {\n\t\t\tbreak\n\t\t}\n\n\t\tq.state.ackLevel = key\n\t\tdelete(q.outstandingTasks, key)\n\t}\n\n\t// The following loop attempts to delete tasks beyond ack level but has already been acked.\n\t// To ensure ack level can be advanced as quick as possible, for a series of acked tasks,\n\t// we still keep the last one in memory so that when previous pending tasks are acked, ack level\n\t// can be advanced without wait for other tasks.\n\t// Also only delete tasks less than read level as the sequence beyond read level may change.\n\t//\n\t// As an example, say currently we have 9 tasks: 1 2 3 4 5 6 7 9 10. Ack level is 0, read level is 7,\n\t// task 1 2 6 10 is pending and the rest have been acked.\n\t// We can delete task 3 4 but not 5 as otherwise even if task 1 and 2 were acked later, ack level would\n\t// at most be 2 until task 6 is acked, while ideally it should be 5.\n\t// We also can't delete task 7, because it's possible that task 8 will be loaded later. If task 7 got deleted,\n\t// ack level can only be advanced to 6 instead of 7.\n\tfor idx < len(keys)-1 && keys[idx].Less(q.state.readLevel) {\n\t\tif q.outstandingTasks[keys[idx]].State() == t.TaskStateAcked && q.outstandingTasks[keys[idx+1]].State() == t.TaskStateAcked {\n\t\t\tdelete(q.outstandingTasks, keys[idx])\n\t\t}\n\n\t\tidx++\n\t}\n\n\tif len(q.outstandingTasks) == 0 {\n\t\tq.state.ackLevel = q.state.readLevel\n\t}\n\n\tif timerKey, ok := q.state.ackLevel.(timerTaskKey); ok {\n\t\tq.state.ackLevel = newTimerTaskKey(timerKey.visibilityTimestamp, 0)\n\t}\n\n\tif q.state.readLevel.Less(q.state.ackLevel) {\n\t\tq.logger.Fatal(\"ack level moved beyond read level\", tag.Error(\n\t\t\tfmt.Errorf(\"processing queue state: %v\", q.state),\n\t\t))\n\t}\n\n\treturn q.state.ackLevel, len(q.outstandingTasks)\n}\n\nfunc splitProcessingQueue(\n\tqueues []*processingQueueImpl,\n\tnewQueueStates []ProcessingQueueState,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) []ProcessingQueue {\n\tnewQueueTasks := make([]map[task.Key]task.Task, 0, len(newQueueStates))\n\tfor i := 0; i != len(newQueueStates); i++ {\n\t\tnewQueueTasks = append(newQueueTasks, make(map[task.Key]task.Task))\n\t}\n\n\tfor _, queue := range queues {\n\tSplitTaskLoop:\n\t\tfor key, task := range queue.outstandingTasks {\n\t\t\tfor i, state := range newQueueStates {\n\t\t\t\tif taskBelongsToProcessQueue(state, key, task) {\n\t\t\t\t\tnewQueueTasks[i][key] = task\n\t\t\t\t\tcontinue SplitTaskLoop\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// if code reaches there it means the task doesn't belongs to any new queue.\n\t\t\t// there's must be a bug in the code for generating the newQueueStates\n\t\t\t// log error, skip the split and return current queues as result\n\t\t\tcurrentQueues := make([]ProcessingQueue, 0, len(newQueueStates))\n\t\t\tcurrentQueueStates := make([]ProcessingQueueState, 0, len(newQueueStates))\n\t\t\tfor _, q := range queues {\n\t\t\t\tcurrentQueues = append(currentQueues, q)\n\t\t\t\tcurrentQueueStates = append(currentQueueStates, queue.State())\n\t\t\t}\n\t\t\tlogger.Error(\"Processing queue encountered an error during split or merge.\", tag.Error(\n\t\t\t\tfmt.Errorf(\"current queue state: %+v, new queue state: %+v\", currentQueueStates, newQueueStates),\n\t\t\t))\n\t\t\treturn currentQueues\n\t\t}\n\t}\n\n\tnewQueues := make([]ProcessingQueue, 0, len(newQueueStates))\n\tfor i, state := range newQueueStates {\n\t\tqueue := newProcessingQueue(\n\t\t\tstate,\n\t\t\tnewQueueTasks[i],\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t)\n\t\tnewQueues = append(newQueues, queue)\n\t}\n\n\treturn newQueues\n}\n\nfunc taskBelongsToProcessQueue(state ProcessingQueueState, key task.Key, task task.Task) bool {\n\treturn state.DomainFilter().Filter(task.GetDomainID()) &&\n\t\tstate.AckLevel().Less(key) &&\n\t\t!state.MaxLevel().Less(key)\n}\n\nfunc taskKeyEquals(key1 task.Key, key2 task.Key) bool {\n\treturn !key1.Less(key2) && !key2.Less(key1)\n}\n\nfunc minTaskKey(key1 task.Key, key2 task.Key) task.Key {\n\tif key1.Less(key2) {\n\t\treturn key1\n\t}\n\treturn key2\n}\n\nfunc maxTaskKey(key1 task.Key, key2 task.Key) task.Key {\n\tif key1.Less(key2) {\n\t\treturn key2\n\t}\n\treturn key1\n}\n"
  },
  {
    "path": "service/history/queue/processing_queue_collection.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype processingQueueCollection struct {\n\tlevel       int\n\tqueues      []ProcessingQueue\n\tactiveQueue ProcessingQueue\n}\n\n// NewProcessingQueueCollection creates a new collection for non-overlapping queues\nfunc NewProcessingQueueCollection(level int, queues []ProcessingQueue) ProcessingQueueCollection {\n\tsortProcessingQueue(queues)\n\tqueueCollection := &processingQueueCollection{\n\t\tlevel:  level,\n\t\tqueues: queues,\n\t}\n\tqueueCollection.resetActiveQueue()\n\treturn queueCollection\n}\n\nfunc newProcessingQueueCollections(\n\tprocessingQueueStates []ProcessingQueueState,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) []ProcessingQueueCollection {\n\tprocessingQueuesMap := make(map[int][]ProcessingQueue) // level -> state\n\tfor _, queueState := range processingQueueStates {\n\t\tprocessingQueuesMap[queueState.Level()] = append(processingQueuesMap[queueState.Level()], NewProcessingQueue(\n\t\t\tqueueState,\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t))\n\t}\n\tprocessingQueueCollections := make([]ProcessingQueueCollection, 0, len(processingQueuesMap))\n\tfor level, queues := range processingQueuesMap {\n\t\tprocessingQueueCollections = append(processingQueueCollections, NewProcessingQueueCollection(\n\t\t\tlevel,\n\t\t\tqueues,\n\t\t))\n\t}\n\n\treturn processingQueueCollections\n}\n\nfunc (c *processingQueueCollection) Level() int {\n\treturn c.level\n}\n\nfunc (c *processingQueueCollection) Queues() []ProcessingQueue {\n\treturn c.queues\n}\n\nfunc (c *processingQueueCollection) ActiveQueue() ProcessingQueue {\n\treturn c.activeQueue\n}\n\nfunc (c *processingQueueCollection) AddTasks(tasks map[task.Key]task.Task, newReadLevel task.Key) {\n\tactiveQueue := c.ActiveQueue()\n\tactiveQueue.AddTasks(tasks, newReadLevel)\n\n\tif taskKeyEquals(activeQueue.State().ReadLevel(), activeQueue.State().MaxLevel()) {\n\t\tc.resetActiveQueue()\n\t}\n}\n\nfunc (c *processingQueueCollection) GetTask(key task.Key) (task.Task, error) {\n\tfor _, queue := range c.Queues() {\n\t\tif task, err := queue.GetTask(key); err == nil {\n\t\t\treturn task, nil\n\t\t}\n\t}\n\treturn nil, errTaskNotFound\n}\n\nfunc (c *processingQueueCollection) GetTasks() []task.Task {\n\tvar outstandingTask []task.Task\n\tfor _, queue := range c.Queues() {\n\t\toutstandingTask = append(outstandingTask, queue.GetTasks()...)\n\t}\n\treturn outstandingTask\n}\n\nfunc (c *processingQueueCollection) UpdateAckLevels() (task.Key, int) {\n\tif len(c.queues) == 0 {\n\t\treturn nil, 0\n\t}\n\n\tremainingQueues := make([]ProcessingQueue, 0, len(c.queues))\n\ttotalPendingTasks := 0\n\tvar minAckLevel task.Key\n\n\tfor _, queue := range c.queues {\n\t\tackLevel, numPendingTasks := queue.UpdateAckLevel()\n\t\tif taskKeyEquals(ackLevel, queue.State().MaxLevel()) {\n\t\t\tcontinue\n\t\t}\n\n\t\tremainingQueues = append(remainingQueues, queue)\n\t\ttotalPendingTasks += numPendingTasks\n\t\tif minAckLevel == nil {\n\t\t\tminAckLevel = ackLevel\n\t\t} else {\n\t\t\tminAckLevel = minTaskKey(minAckLevel, ackLevel)\n\t\t}\n\t}\n\n\tc.queues = remainingQueues\n\treturn minAckLevel, totalPendingTasks\n}\n\nfunc (c *processingQueueCollection) Split(policy ProcessingQueueSplitPolicy) []ProcessingQueue {\n\tif len(c.queues) == 0 {\n\t\treturn nil\n\t}\n\n\tnewQueues := make([]ProcessingQueue, 0, len(c.queues))\n\tnextLevelQueues := []ProcessingQueue{}\n\n\tfor _, queue := range c.queues {\n\t\tsplitQueues := queue.Split(policy)\n\t\tsortProcessingQueue(splitQueues)\n\t\tfor _, splitQueue := range splitQueues {\n\t\t\tif splitQueue.State().Level() != c.level {\n\t\t\t\tnextLevelQueues = append(nextLevelQueues, splitQueue)\n\t\t\t} else {\n\t\t\t\tnewQueues = append(newQueues, splitQueue)\n\t\t\t}\n\t\t}\n\t}\n\n\tc.queues = newQueues\n\n\tc.resetActiveQueue()\n\n\treturn nextLevelQueues\n}\n\nfunc (c *processingQueueCollection) Merge(incomingQueues []ProcessingQueue) {\n\tsortProcessingQueue(incomingQueues)\n\n\tnewQueues := make([]ProcessingQueue, 0, len(c.queues)+len(incomingQueues))\n\n\tcurrentQueueIdx := 0\n\tincomingQueueIdx := 0\n\tfor incomingQueueIdx < len(incomingQueues) && currentQueueIdx < len(c.queues) {\n\t\tmergedQueues := c.queues[currentQueueIdx].Merge(incomingQueues[incomingQueueIdx])\n\t\tsortProcessingQueue(mergedQueues)\n\t\tnewQueues = append(newQueues, mergedQueues[:len(mergedQueues)-1]...)\n\n\t\tlastMergedQueue := mergedQueues[len(mergedQueues)-1]\n\t\toverlapWithCurrentQueue := currentQueueIdx+1 != len(c.queues) &&\n\t\t\tc.queues[currentQueueIdx+1].State().AckLevel().Less(lastMergedQueue.State().MaxLevel())\n\t\toverlapWithIncomingQueue := incomingQueueIdx+1 != len(incomingQueues) &&\n\t\t\tincomingQueues[incomingQueueIdx+1].State().AckLevel().Less(lastMergedQueue.State().MaxLevel())\n\n\t\tif !overlapWithCurrentQueue && !overlapWithIncomingQueue {\n\t\t\tnewQueues = append(newQueues, lastMergedQueue)\n\t\t\tincomingQueueIdx++\n\t\t\tcurrentQueueIdx++\n\t\t} else if overlapWithCurrentQueue {\n\t\t\tincomingQueues[incomingQueueIdx] = lastMergedQueue\n\t\t\tcurrentQueueIdx++\n\t\t} else {\n\t\t\tc.queues[currentQueueIdx] = lastMergedQueue\n\t\t\tincomingQueueIdx++\n\t\t}\n\t}\n\n\tif incomingQueueIdx < len(incomingQueues) {\n\t\tnewQueues = append(newQueues, incomingQueues[incomingQueueIdx:]...)\n\t}\n\n\tif currentQueueIdx < len(c.queues) {\n\t\tnewQueues = append(newQueues, c.queues[currentQueueIdx:]...)\n\t}\n\n\tc.queues = newQueues\n\n\t// make sure the result is ordered and disjoint\n\tfor idx := 0; idx < len(c.queues)-1; idx++ {\n\t\tif c.queues[idx+1].State().AckLevel().Less(c.queues[idx].State().MaxLevel()) {\n\t\t\terrMsg := \"\"\n\t\t\tfor _, q := range c.queues {\n\t\t\t\terrMsg += fmt.Sprintf(\"%v \", q)\n\t\t\t}\n\t\t\tpanic(\"invalid processing queue merge result: \" + errMsg)\n\t\t}\n\t}\n\n\tc.resetActiveQueue()\n}\n\nfunc (c *processingQueueCollection) resetActiveQueue() {\n\tfor _, queue := range c.queues {\n\t\tif !taskKeyEquals(queue.State().ReadLevel(), queue.State().MaxLevel()) {\n\t\t\tc.activeQueue = queue\n\t\t\treturn\n\t\t}\n\t}\n\tc.activeQueue = nil\n}\n\nfunc sortProcessingQueue(queues []ProcessingQueue) {\n\tsort.Slice(queues, func(i, j int) bool {\n\t\tif queues[i].State().Level() == queues[j].State().Level() {\n\t\t\treturn queues[i].State().AckLevel().Less(queues[j].State().AckLevel())\n\t\t}\n\t\treturn queues[i].State().Level() < queues[j].State().Level()\n\t})\n}\n"
  },
  {
    "path": "service/history/queue/processing_queue_collection_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tprocessingQueueCollectionSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller *gomock.Controller\n\n\t\tlevel int\n\t}\n)\n\nfunc TestProcessingQueueCollectionSuite(t *testing.T) {\n\ts := new(processingQueueCollectionSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *processingQueueCollectionSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.level = 0\n}\n\nfunc (s *processingQueueCollectionSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *processingQueueCollectionSuite) TestNewCollection_EmptyQueues() {\n\tqueueCollection := NewProcessingQueueCollection(s.level, nil).(*processingQueueCollection)\n\n\ts.Nil(queueCollection.ActiveQueue())\n}\n\nfunc (s *processingQueueCollectionSuite) TestNewCollection_OutOfOrderQueues() {\n\ttotalQueues := 4\n\tmockQueues := []ProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tmockQueues[0].(*MockProcessingQueue).EXPECT().State().Return(newProcessingQueueState(\n\t\ts.level,\n\t\ttestKey{ID: 20},\n\t\ttestKey{ID: 25},\n\t\ttestKey{ID: 30},\n\t\tDomainFilter{},\n\t)).AnyTimes()\n\tmockQueues[1].(*MockProcessingQueue).EXPECT().State().Return(newProcessingQueueState(\n\t\ts.level,\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 10},\n\t\ttestKey{ID: 10},\n\t\tDomainFilter{},\n\t)).AnyTimes()\n\tmockQueues[2].(*MockProcessingQueue).EXPECT().State().Return(newProcessingQueueState(\n\t\ts.level,\n\t\ttestKey{ID: 30},\n\t\ttestKey{ID: 30},\n\t\ttestKey{ID: 40},\n\t\tDomainFilter{},\n\t)).AnyTimes()\n\tmockQueues[3].(*MockProcessingQueue).EXPECT().State().Return(newProcessingQueueState(\n\t\ts.level,\n\t\ttestKey{ID: 10},\n\t\ttestKey{ID: 20},\n\t\ttestKey{ID: 20},\n\t\tDomainFilter{},\n\t)).AnyTimes()\n\texpectedActiveQueue := mockQueues[0]\n\n\tqueueCollection := NewProcessingQueueCollection(s.level, mockQueues).(*processingQueueCollection)\n\n\ts.Equal(expectedActiveQueue.State(), queueCollection.ActiveQueue().State())\n\ts.True(s.isQueuesSorted(queueCollection.queues))\n}\n\nfunc (s *processingQueueCollectionSuite) TestAddTasks_ReadNotFinished() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\tnewReadLevel := testKey{ID: 9}\n\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tmockQueues[currentActiveIdx].EXPECT().AddTasks(gomock.Any(), newReadLevel).Times(1)\n\tmockQueues[currentActiveIdx].EXPECT().State().Return(newProcessingQueueState(\n\t\ts.level,\n\t\ttestKey{ID: 3},\n\t\tnewReadLevel,\n\t\ttestKey{ID: 10},\n\t\tDomainFilter{},\n\t)).AnyTimes()\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tqueueCollection.AddTasks(map[task.Key]task.Task{}, newReadLevel)\n\ts.Equal(mockQueues[currentActiveIdx].State(), queueCollection.ActiveQueue().State())\n}\n\nfunc (s *processingQueueCollectionSuite) TestAddTask_ReadFinished() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\tnewReadLevel := testKey{ID: 10}\n\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tmockQueues[currentActiveIdx].EXPECT().AddTasks(gomock.Any(), newReadLevel).Times(1)\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues[i].EXPECT().State().Return(newProcessingQueueState(\n\t\t\ts.level,\n\t\t\ttestKey{ID: 3},\n\t\t\tnewReadLevel,\n\t\t\ttestKey{ID: 10},\n\t\t\tDomainFilter{},\n\t\t)).AnyTimes()\n\t}\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tqueueCollection.AddTasks(map[task.Key]task.Task{}, newReadLevel)\n\ts.Nil(queueCollection.ActiveQueue())\n}\n\nfunc (s *processingQueueCollectionSuite) TestGetTask_Success() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\n\tmockTask := task.NewMockTask(s.controller)\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tmockQueues[0].EXPECT().GetTask(gomock.Any()).Return(mockTask, nil).Times(1)\n\tfor i := 1; i != totalQueues; i++ {\n\t\tmockQueues[i].EXPECT().GetTask(gomock.Any()).Return(nil, errTaskNotFound).AnyTimes()\n\t}\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tnewTask, err := queueCollection.GetTask(testKey{ID: 1})\n\ts.NoError(err)\n\ts.Equal(mockTask, newTask)\n}\n\nfunc (s *processingQueueCollectionSuite) TestGetTask_NoTaskFound_Fail() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues[i].EXPECT().GetTask(gomock.Any()).Return(nil, errTaskNotFound).AnyTimes()\n\t}\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tnewTask, err := queueCollection.GetTask(testKey{ID: 1})\n\ts.Error(err)\n\ts.Nil(newTask)\n}\n\nfunc (s *processingQueueCollectionSuite) TestGetTasks_Success() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\n\tmockQueues := []*MockProcessingQueue{}\n\tmockTasks := []task.Task{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t\tmockTasks = append(mockTasks, task.NewMockTask(s.controller))\n\t}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues[i].EXPECT().GetTasks().Return([]task.Task{mockTasks[i]}).Times(1)\n\t}\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tnewTasks := queueCollection.GetTasks()\n\ts.Equal(mockTasks, newTasks)\n}\n\nfunc (s *processingQueueCollectionSuite) TestGetTasks_EmptyTask() {\n\ttotalQueues := 4\n\tcurrentActiveIdx := 1\n\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues[i].EXPECT().GetTasks().Return([]task.Task{}).Times(1)\n\t}\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tnewTasks := queueCollection.GetTasks()\n\ts.Equal(0, len(newTasks))\n}\n\nfunc (s *processingQueueCollectionSuite) TestUpdateAckLevels() {\n\ttotalQueues := 5\n\tcurrentActiveIdx := 1\n\tmockQueues := []*MockProcessingQueue{}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tmockQueues = append(mockQueues, NewMockProcessingQueue(s.controller))\n\t}\n\n\tfinishedQueueIdx := map[int]struct{}{0: {}, 2: {}, 3: {}}\n\tfor i := 0; i != totalQueues; i++ {\n\t\tif _, ok := finishedQueueIdx[i]; ok {\n\t\t\tmockQueues[i].EXPECT().UpdateAckLevel().Return(testKey{ID: i}, 0).Times(1)\n\t\t\tmockQueues[i].EXPECT().State().Return(newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: i},\n\t\t\t\ttestKey{ID: i},\n\t\t\t\ttestKey{ID: i},\n\t\t\t\tDomainFilter{},\n\t\t\t)).AnyTimes()\n\t\t} else {\n\t\t\tmockQueues[i].EXPECT().UpdateAckLevel().Return(testKey{ID: i - i}, 1).Times(1)\n\t\t\tmockQueues[i].EXPECT().State().Return(newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: i - 1},\n\t\t\t\ttestKey{ID: i},\n\t\t\t\ttestKey{ID: i},\n\t\t\t\tDomainFilter{},\n\t\t\t)).AnyTimes()\n\t\t}\n\t}\n\texpectedActiveQueue := mockQueues[1]\n\n\tqueueCollection := s.newTestProcessingQueueCollection(s.level, mockQueues)\n\tqueueCollection.activeQueue = mockQueues[currentActiveIdx]\n\n\tackLevel, totalPendingTasks := queueCollection.UpdateAckLevels()\n\ts.Equal(testKey{ID: 0}, ackLevel)\n\ts.Equal(totalQueues-len(finishedQueueIdx), totalPendingTasks)\n\ts.Len(queueCollection.queues, totalQueues-len(finishedQueueIdx))\n\ts.Equal(expectedActiveQueue.State(), queueCollection.ActiveQueue().State())\n}\n\nfunc (s *processingQueueCollectionSuite) TestSplit() {\n\ttestCases := []struct {\n\t\tcurrentQueueStates []ProcessingQueueState\n\t\tsplitResults       [][]ProcessingQueueState\n\n\t\texpectedActiveQueueState     ProcessingQueueState\n\t\texpectedNewQueueStates       []ProcessingQueueState\n\t\texpectedNextLevelQueueStates []ProcessingQueueState\n\t}{\n\t\t{}, // empty queue collection\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tsplitResults: [][]ProcessingQueueState{\n\t\t\t\t{\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level,\n\t\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t\t),\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level+1,\n\t\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedNextLevelQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level+1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tsplitResults: [][]ProcessingQueueState{\n\t\t\t\t{\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level+1,\n\t\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t\t),\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level,\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t\t\t),\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level,\n\t\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t\ts.level,\n\t\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\ttestKey{ID: 15},\n\t\t\t\ttestKey{ID: 20},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}, \"domain3\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedNextLevelQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level+1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tmockQueues := []ProcessingQueue{}\n\t\tfor idx, queueState := range tc.currentQueueStates {\n\t\t\tmockQueue := NewMockProcessingQueue(s.controller)\n\t\t\tmockQueue.EXPECT().State().Return(queueState).AnyTimes()\n\n\t\t\tsplitMockQueues := []ProcessingQueue{}\n\t\t\tfor _, splitQueueState := range tc.splitResults[idx] {\n\t\t\t\tsplitMockQueue := NewMockProcessingQueue(s.controller)\n\t\t\t\tsplitMockQueue.EXPECT().State().Return(splitQueueState).AnyTimes()\n\t\t\t\tsplitMockQueues = append(splitMockQueues, splitMockQueue)\n\t\t\t}\n\t\t\tmockQueue.EXPECT().Split(gomock.Any()).Return(splitMockQueues)\n\n\t\t\tmockQueues = append(mockQueues, mockQueue)\n\t\t}\n\n\t\tqueueCollection := NewProcessingQueueCollection(s.level, mockQueues).(*processingQueueCollection)\n\t\tnextLevelQueues := queueCollection.Split(NewMockProcessingQueueSplitPolicy(s.controller))\n\n\t\tif tc.expectedActiveQueueState != nil {\n\t\t\ts.Equal(tc.expectedActiveQueueState, queueCollection.ActiveQueue().State())\n\t\t}\n\t\tfor idx, expectedState := range tc.expectedNewQueueStates {\n\t\t\ts.Equal(expectedState, queueCollection.queues[idx].State())\n\t\t}\n\t\tfor idx, expectedState := range tc.expectedNextLevelQueueStates {\n\t\t\ts.Equal(expectedState, nextLevelQueues[idx].State())\n\t\t}\n\t}\n}\n\nfunc (s *processingQueueCollectionSuite) TestMerge() {\n\ttestCases := []struct {\n\t\tcurrentQueueStates  []ProcessingQueueState\n\t\tincomingQueueStates []ProcessingQueueState\n\n\t\texpectedActiveQueueState ProcessingQueueState\n\t\texpectedNewQueueStates   []ProcessingQueueState\n\t}{\n\t\t{}, // empty queue collection\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tincomingQueueStates:      nil,\n\t\t\texpectedActiveQueueState: nil,\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 25},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tincomingQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 20},\n\t\t\t\ttestKey{ID: 25},\n\t\t\t\ttestKey{ID: 30},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 25},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tincomingQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\ttestKey{ID: 20},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\ttestKey{ID: 75},\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tincomingQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 8},\n\t\t\t\t\ttestKey{ID: 35},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 80},\n\t\t\t\t\ttestKey{ID: 90},\n\t\t\t\t\ttestKey{ID: 100},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 8},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 8},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 8},\n\t\t\t\t\ttestKey{ID: 8},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\ttestKey{ID: 35},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\ttestKey{ID: 75},\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 80},\n\t\t\t\t\ttestKey{ID: 90},\n\t\t\t\t\ttestKey{ID: 100},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\ttestKey{ID: 40},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\ttestKey{ID: 65},\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\tincomingQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 18},\n\t\t\t\t\ttestKey{ID: 18},\n\t\t\t\t\ttestKey{ID: 100},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedActiveQueueState: newProcessingQueueState(\n\t\t\t\ts.level,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t),\n\t\t\texpectedNewQueueStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 18},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 18},\n\t\t\t\t\ttestKey{ID: 18},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain1\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\ttestKey{ID: 30},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\ttestKey{ID: 60},\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain2\": {}, \"domain3\": {}}},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\ts.level,\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\ttestKey{ID: 70},\n\t\t\t\t\ttestKey{ID: 100},\n\t\t\t\t\tDomainFilter{DomainIDs: map[string]struct{}{\"domain3\": {}}},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tqueues := []ProcessingQueue{}\n\t\tfor _, queueState := range tc.currentQueueStates {\n\t\t\tqueue := NewProcessingQueue(queueState, nil, nil)\n\t\t\tqueues = append(queues, queue)\n\t\t}\n\n\t\tincomingQueues := []ProcessingQueue{}\n\t\tfor _, queueState := range tc.incomingQueueStates {\n\t\t\tincomingQueue := NewProcessingQueue(queueState, nil, nil)\n\t\t\tincomingQueues = append(incomingQueues, incomingQueue)\n\t\t}\n\n\t\tqueueCollection := NewProcessingQueueCollection(s.level, queues).(*processingQueueCollection)\n\t\tqueueCollection.Merge(incomingQueues)\n\n\t\tif tc.expectedActiveQueueState != nil {\n\t\t\ts.Equal(tc.expectedActiveQueueState, queueCollection.ActiveQueue().State())\n\t\t}\n\t\tfor idx, expectedState := range tc.expectedNewQueueStates {\n\t\t\ts.Equal(expectedState, queueCollection.queues[idx].State())\n\t\t}\n\t}\n}\n\nfunc (s *processingQueueCollectionSuite) isQueuesSorted(\n\tqueues []ProcessingQueue,\n) bool {\n\tif len(queues) <= 1 {\n\t\treturn true\n\t}\n\n\tfor i := 0; i != len(queues)-1; i++ {\n\t\tif !queues[i].State().AckLevel().Less(queues[i+1].State().AckLevel()) ||\n\t\t\tqueues[i+1].State().AckLevel().Less(queues[i].State().MaxLevel()) {\n\t\t\treturn false\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc (s *processingQueueCollectionSuite) newTestProcessingQueueCollection(\n\tlevel int,\n\tmockQueues []*MockProcessingQueue,\n) *processingQueueCollection {\n\tqueues := make([]ProcessingQueue, 0, len(mockQueues))\n\tfor _, mockQueue := range mockQueues {\n\t\tqueues = append(queues, mockQueue)\n\t}\n\n\treturn &processingQueueCollection{\n\t\tlevel:  level,\n\t\tqueues: queues,\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/processing_queue_state.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype processingQueueStateImpl struct {\n\tlevel        int\n\tackLevel     task.Key\n\treadLevel    task.Key\n\tmaxLevel     task.Key\n\tdomainFilter DomainFilter\n}\n\n// NewProcessingQueueState creates a new state instance for processing queue\n// readLevel will be set to the same value as ackLevel\nfunc NewProcessingQueueState(\n\tlevel int,\n\tackLevel task.Key,\n\tmaxLevel task.Key,\n\tdomainFilter DomainFilter,\n) ProcessingQueueState {\n\treturn newProcessingQueueState(\n\t\tlevel,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tdomainFilter,\n\t)\n}\n\nfunc newProcessingQueueState(\n\tlevel int,\n\tackLevel task.Key,\n\treadLevel task.Key,\n\tmaxLevel task.Key,\n\tdomainFilter DomainFilter,\n) *processingQueueStateImpl {\n\treturn &processingQueueStateImpl{\n\t\tlevel:        level,\n\t\tackLevel:     ackLevel,\n\t\treadLevel:    readLevel,\n\t\tmaxLevel:     maxLevel,\n\t\tdomainFilter: domainFilter,\n\t}\n}\n\nfunc (s *processingQueueStateImpl) Level() int {\n\treturn s.level\n}\n\nfunc (s *processingQueueStateImpl) MaxLevel() task.Key {\n\treturn s.maxLevel\n}\n\nfunc (s *processingQueueStateImpl) AckLevel() task.Key {\n\treturn s.ackLevel\n}\n\nfunc (s *processingQueueStateImpl) ReadLevel() task.Key {\n\treturn s.readLevel\n}\n\nfunc (s *processingQueueStateImpl) DomainFilter() DomainFilter {\n\treturn s.domainFilter\n}\n\nfunc (s *processingQueueStateImpl) String() string {\n\treturn fmt.Sprintf(\"&{level: %+v, ackLevel: %+v, readLevel: %+v, maxLevel: %+v, domainFilter: %+v}\",\n\t\ts.level, s.ackLevel, s.readLevel, s.maxLevel, s.domainFilter,\n\t)\n}\n\nfunc copyQueueState(state ProcessingQueueState) *processingQueueStateImpl {\n\treturn newProcessingQueueState(\n\t\tstate.Level(),\n\t\tstate.AckLevel(),\n\t\tstate.ReadLevel(),\n\t\tstate.MaxLevel(),\n\t\tstate.DomainFilter(),\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/processing_queue_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"sort\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tt \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tprocessingQueueSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller *gomock.Controller\n\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t}\n\n\ttestKey struct {\n\t\tID int\n\t}\n)\n\nfunc TestProcessingQueueSuite(t *testing.T) {\n\ts := new(processingQueueSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *processingQueueSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n}\n\nfunc (s *processingQueueSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *processingQueueSuite) TestAddTasks() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 9},\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor _, key := range taskKeys {\n\t\tmockTask := task.NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetDomainID().Return(\"some random domainID\").AnyTimes()\n\t\tmockTask.EXPECT().GetWorkflowID().Return(\"some random workflowID\").AnyTimes()\n\t\tmockTask.EXPECT().GetRunID().Return(\"some random runID\").AnyTimes()\n\t\tmockTask.EXPECT().GetTaskType().Return(0).AnyTimes()\n\t\ttasks[key] = mockTask\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\tmake(map[task.Key]task.Task),\n\t)\n\n\tnewReadLevel := testKey{ID: 10}\n\tqueue.AddTasks(tasks, newReadLevel)\n\ts.Len(queue.outstandingTasks, len(taskKeys))\n\ts.Equal(newReadLevel, queue.state.readLevel)\n\n\t// add the same set of tasks again, should have no effect\n\tqueue.AddTasks(tasks, newReadLevel)\n\ts.Len(queue.outstandingTasks, len(taskKeys))\n\ts.Equal(newReadLevel, queue.state.readLevel)\n}\n\nfunc (s *processingQueueSuite) TestGetTask_TaskNotFound_Error() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 9},\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor _, key := range taskKeys {\n\t\tmockTask := task.NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetDomainID().Return(\"some random domainID\").AnyTimes()\n\t\tmockTask.EXPECT().GetWorkflowID().Return(\"some random workflowID\").AnyTimes()\n\t\tmockTask.EXPECT().GetRunID().Return(\"some random runID\").AnyTimes()\n\t\tmockTask.EXPECT().GetTaskType().Return(0).AnyTimes()\n\t\ttasks[key] = mockTask\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\tmake(map[task.Key]task.Task),\n\t)\n\n\tnewReadLevel := testKey{ID: 10}\n\tqueue.AddTasks(tasks, newReadLevel)\n\ts.Len(queue.outstandingTasks, len(taskKeys))\n\n\ttask, err := queue.GetTask(testKey{ID: 100})\n\ts.Nil(task)\n\ts.Error(err)\n}\n\nfunc (s *processingQueueSuite) TestGetTask_Success() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 9},\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor _, key := range taskKeys {\n\t\tmockTask := task.NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetDomainID().Return(\"some random domainID\").AnyTimes()\n\t\tmockTask.EXPECT().GetWorkflowID().Return(\"some random workflowID\").AnyTimes()\n\t\tmockTask.EXPECT().GetRunID().Return(\"some random runID\").AnyTimes()\n\t\tmockTask.EXPECT().GetTaskType().Return(0).AnyTimes()\n\t\ttasks[key] = mockTask\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\tmake(map[task.Key]task.Task),\n\t)\n\n\tnewReadLevel := testKey{ID: 10}\n\tqueue.AddTasks(tasks, newReadLevel)\n\ts.Len(queue.outstandingTasks, len(taskKeys))\n\n\ttask, err := queue.GetTask(testKey{ID: 3})\n\ts.NoError(err)\n\ts.Equal(tasks[testKey{ID: 3}], task)\n}\n\nfunc (s *processingQueueSuite) TestGetTasks_Success() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 9},\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor _, key := range taskKeys {\n\t\tmockTask := task.NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetTaskID().Return(int64(key.(testKey).ID)).AnyTimes()\n\t\tmockTask.EXPECT().GetDomainID().Return(\"some random domainID\").AnyTimes()\n\t\tmockTask.EXPECT().GetWorkflowID().Return(\"some random workflowID\").AnyTimes()\n\t\tmockTask.EXPECT().GetRunID().Return(\"some random runID\").AnyTimes()\n\t\tmockTask.EXPECT().GetTaskType().Return(0).AnyTimes()\n\t\ttasks[key] = mockTask\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\tmake(map[task.Key]task.Task),\n\t)\n\n\tnewReadLevel := testKey{ID: 10}\n\tqueue.AddTasks(tasks, newReadLevel)\n\ts.Len(queue.outstandingTasks, len(taskKeys))\n\n\toutstandingTasks := queue.GetTasks()\n\tfor _, outstandingTask := range outstandingTasks {\n\t\ttask := tasks[testKey{ID: int(outstandingTask.GetTaskID())}]\n\t\ts.Equal(outstandingTask, task)\n\t}\n}\n\nfunc (s *processingQueueSuite) TestGetTasks_EmptyResult() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\tackLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\tmake(map[task.Key]task.Task),\n\t)\n\n\toutstandingTasks := queue.GetTasks()\n\ts.Equal(0, len(outstandingTasks))\n}\n\nfunc (s *processingQueueSuite) TestUpdateAckLevel_WithPendingTasks() {\n\tackLevel := testKey{ID: 1}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 8},\n\t\ttestKey{ID: 10},\n\t}\n\ttaskStates := []t.State{\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor i, key := range taskKeys {\n\t\ttask := task.NewMockTask(s.controller)\n\t\ttask.EXPECT().State().Return(taskStates[i]).AnyTimes()\n\t\ttasks[key] = task\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\ttaskKeys[len(taskKeys)-1],\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\ttasks,\n\t)\n\n\tnewAckLevel, pendingTasks := queue.UpdateAckLevel()\n\ts.Equal(testKey{ID: 3}, newAckLevel)\n\ts.Equal(testKey{ID: 3}, queue.state.ackLevel)\n\ts.Equal(3, pendingTasks)\n}\n\nfunc (s *processingQueueSuite) TestUpdateAckLevel_NoPendingTasks() {\n\tackLevel := testKey{ID: 1}\n\treadLevel := testKey{ID: 9}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 8},\n\t}\n\ttaskStates := []t.State{\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor i, key := range taskKeys {\n\t\ttask := task.NewMockTask(s.controller)\n\t\ttask.EXPECT().State().Return(taskStates[i]).AnyTimes()\n\t\ttasks[key] = task\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\treadLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\ttasks,\n\t)\n\n\tnewAckLevel, pendingTasks := queue.UpdateAckLevel()\n\ts.Equal(readLevel, newAckLevel)\n\ts.Equal(readLevel, queue.state.ackLevel)\n\ts.Equal(0, pendingTasks)\n}\n\nfunc (s *processingQueueSuite) TestUpdateAckLevel_TaskKeyLargerThanReadLevel() {\n\tackLevel := testKey{ID: 1}\n\treadLevel := testKey{ID: 5}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 8},\n\t\ttestKey{ID: 9},\n\t}\n\ttaskStates := []t.State{\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor i, key := range taskKeys {\n\t\ttask := task.NewMockTask(s.controller)\n\t\ttask.EXPECT().State().Return(taskStates[i]).AnyTimes()\n\t\ttasks[key] = task\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\treadLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\ttasks,\n\t)\n\n\tnewAckLevel, pendingTasks := queue.UpdateAckLevel()\n\ts.Equal(readLevel, newAckLevel)\n\ts.Equal(readLevel, queue.state.ackLevel)\n\ts.Equal(2, pendingTasks)\n}\n\nfunc (s *processingQueueSuite) TestUpdateAckLevel_DeleteTaskBeyondAckLevel() {\n\tackLevel := testKey{ID: 0}\n\treadLevel := testKey{ID: 7}\n\tmaxLevel := testKey{ID: 10}\n\n\ttaskKeys := []task.Key{\n\t\ttestKey{ID: 1},\n\t\ttestKey{ID: 2},\n\t\ttestKey{ID: 3},\n\t\ttestKey{ID: 4},\n\t\ttestKey{ID: 5},\n\t\ttestKey{ID: 6},\n\t\ttestKey{ID: 7},\n\t\ttestKey{ID: 9},\n\t\ttestKey{ID: 10},\n\t}\n\ttaskStates := []t.State{\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStateAcked,\n\t\tt.TaskStatePending,\n\t}\n\ttasks := make(map[task.Key]task.Task)\n\tfor i, key := range taskKeys {\n\t\ttask := task.NewMockTask(s.controller)\n\t\ttask.EXPECT().State().Return(taskStates[i]).AnyTimes()\n\t\ttasks[key] = task\n\t}\n\n\tqueue := s.newTestProcessingQueue(\n\t\t0,\n\t\tackLevel,\n\t\treadLevel,\n\t\tmaxLevel,\n\t\tNewDomainFilter(nil, true),\n\t\ttasks,\n\t)\n\n\tnewAckLevel, pendingTasks := queue.UpdateAckLevel()\n\ts.Equal(testKey{ID: 1}, newAckLevel)\n\ts.Equal(testKey{ID: 1}, queue.state.ackLevel)\n\ts.Equal(6, pendingTasks) // 2 5 6 7 9 10\n}\n\nfunc (s *processingQueueSuite) TestSplit() {\n\ttestCases := []struct {\n\t\tqueue             *processingQueueImpl\n\t\tpolicyResult      []ProcessingQueueState\n\t\texpectedNewQueues []*processingQueueImpl\n\t}{\n\t\t{\n\t\t\t// test 1: no split needed\n\t\t\tqueue: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 3},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 2}, testKey{ID: 3}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tpolicyResult: nil,\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 3},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 2}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 2: split two domains to another level, doesn't change range\n\t\t\tqueue: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 2}, testKey{ID: 3}, testKey{ID: 5}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain2\", \"testDomain3\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tpolicyResult: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 2}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 5}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 3: split into multiple new levels, while keeping the existing range\n\t\t\tqueue: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    make(map[string]struct{}),\n\t\t\t\t\tReverseMatch: true,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 2}, testKey{ID: 3}, testKey{ID: 5}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain2\", \"testDomain3\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tpolicyResult: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t2,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 2}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t2,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 5}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 4: change the queue range\n\t\t\tqueue: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 7},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    make(map[string]struct{}),\n\t\t\t\t\tReverseMatch: true,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 2}, testKey{ID: 3}, testKey{ID: 5}, testKey{ID: 6}, testKey{ID: 7}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain2\", \"testDomain3\", \"testDomain1\", \"testDomain3\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tpolicyResult: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 7},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    make(map[string]struct{}),\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 2}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 7},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    make(map[string]struct{}),\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 6}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 7}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 5}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tmockPolicy := NewMockProcessingQueueSplitPolicy(s.controller)\n\t\tmockPolicy.EXPECT().Evaluate(ProcessingQueue(tc.queue)).Return(tc.policyResult).Times(1)\n\n\t\tnewQueues := tc.queue.Split(mockPolicy)\n\n\t\ts.assertQueuesEqual(tc.expectedNewQueues, newQueues)\n\t}\n}\n\nfunc (s *processingQueueSuite) TestMerge() {\n\ttestCases := []struct {\n\t\tqueue1            *processingQueueImpl\n\t\tqueue2            *processingQueueImpl\n\t\texpectedNewQueues []*processingQueueImpl\n\t}{\n\t\t{\n\t\t\t// test 1: no overlap in range\n\t\t\tqueue1: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 1},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}},\n\t\t\t\t\t[]string{\"testDomain1\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tqueue2: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\ttestKey{ID: 50},\n\t\t\t\ttestKey{ID: 100},\n\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 50}},\n\t\t\t\t\t[]string{\"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 1},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 50},\n\t\t\t\t\ttestKey{ID: 100},\n\t\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 50}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 2: same ack level\n\t\t\tqueue1: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 7},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 4}, testKey{ID: 7}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain1\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tqueue2: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 3},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 2}, testKey{ID: 3}},\n\t\t\t\t\t[]string{\"testDomain2\", \"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 3},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 2}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 4}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 7},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 7}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 3: same max level\n\t\t\tqueue1: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 7},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 4}, testKey{ID: 7}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain1\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tqueue2: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 9},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 6}, testKey{ID: 8}, testKey{ID: 9}},\n\t\t\t\t\t[]string{\"testDomain2\", \"testDomain2\", \"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 4}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 7},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 6}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 7}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 8}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 9}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 4: one queue contain another\n\t\t\tqueue1: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 7},\n\t\t\t\ttestKey{ID: 20},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 4}, testKey{ID: 7}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\", \"testDomain1\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tqueue2: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 9},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 6}, testKey{ID: 8}, testKey{ID: 9}},\n\t\t\t\t\t[]string{\"testDomain2\", \"testDomain2\", \"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 4}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 7},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 6}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 7}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 8}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 9}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t// test 5: general case\n\t\t\tqueue1: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 3},\n\t\t\t\ttestKey{ID: 15},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 1}, testKey{ID: 3}},\n\t\t\t\t\t[]string{\"testDomain1\", \"testDomain1\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\tqueue2: s.newTestProcessingQueue(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 17},\n\t\t\t\ttestKey{ID: 20},\n\t\t\t\tDomainFilter{\n\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\tReverseMatch: false,\n\t\t\t\t},\n\t\t\t\ts.newMockTasksForDomain(\n\t\t\t\t\t[]task.Key{testKey{ID: 6}, testKey{ID: 8}, testKey{ID: 9}, testKey{ID: 17}},\n\t\t\t\t\t[]string{\"testDomain2\", \"testDomain2\", \"testDomain2\", \"testDomain2\"},\n\t\t\t\t),\n\t\t\t),\n\t\t\texpectedNewQueues: []*processingQueueImpl{\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 3},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 1}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 3}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 6}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 8}: task.NewMockTask(s.controller),\n\t\t\t\t\t\ttestKey{ID: 9}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t\ts.newTestProcessingQueue(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 15},\n\t\t\t\t\ttestKey{ID: 17},\n\t\t\t\t\ttestKey{ID: 20},\n\t\t\t\t\tDomainFilter{\n\t\t\t\t\t\tDomainIDs:    map[string]struct{}{\"testDomain2\": {}},\n\t\t\t\t\t\tReverseMatch: false,\n\t\t\t\t\t},\n\t\t\t\t\tmap[task.Key]task.Task{\n\t\t\t\t\t\ttestKey{ID: 17}: task.NewMockTask(s.controller),\n\t\t\t\t\t},\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tqueue1 := s.copyProcessingQueue(tc.queue1)\n\t\tqueue2 := s.copyProcessingQueue(tc.queue2)\n\t\ts.assertQueuesEqual(tc.expectedNewQueues, queue1.Merge(queue2))\n\n\t\tqueue1 = s.copyProcessingQueue(tc.queue1)\n\t\tqueue2 = s.copyProcessingQueue(tc.queue2)\n\t\ts.assertQueuesEqual(tc.expectedNewQueues, queue2.Merge(queue1))\n\t}\n}\n\nfunc (s *processingQueueSuite) assertQueuesEqual(\n\texpectedQueues []*processingQueueImpl,\n\tactual []ProcessingQueue,\n) {\n\ts.Equal(len(expectedQueues), len(actual))\n\n\tactualQueues := make([]*processingQueueImpl, 0, len(actual))\n\tfor _, queue := range actual {\n\t\tactualQueues = append(actualQueues, queue.(*processingQueueImpl))\n\t}\n\n\tcompFn := func(q1, q2 *processingQueueImpl) bool {\n\t\tif taskKeyEquals(q1.state.ackLevel, q2.state.ackLevel) {\n\t\t\treturn q1.state.level < q2.state.level\n\t\t}\n\n\t\treturn q1.state.ackLevel.Less(q2.state.ackLevel)\n\t}\n\n\tsort.Slice(expectedQueues, func(i, j int) bool {\n\t\treturn compFn(expectedQueues[i], expectedQueues[j])\n\t})\n\tsort.Slice(actualQueues, func(i, j int) bool {\n\t\treturn compFn(actualQueues[i], actualQueues[j])\n\t})\n\n\tfor i := 0; i != len(expectedQueues); i++ {\n\t\ts.assertQueueEqual(expectedQueues[i], actualQueues[i])\n\t}\n}\n\nfunc (s *processingQueueSuite) assertQueueEqual(\n\texpected *processingQueueImpl,\n\tactual *processingQueueImpl,\n) {\n\ts.Equal(expected.state, actual.state)\n\ts.Equal(len(expected.outstandingTasks), len(actual.outstandingTasks))\n\texpectedKeys := make([]task.Key, 0, len(expected.outstandingTasks))\n\tfor key := range expected.outstandingTasks {\n\t\texpectedKeys = append(expectedKeys, key)\n\t}\n\tactualKeys := make([]task.Key, 0, len(actual.outstandingTasks))\n\tfor key := range actual.outstandingTasks {\n\t\tactualKeys = append(actualKeys, key)\n\t}\n\tsort.Slice(expectedKeys, func(i, j int) bool {\n\t\treturn expectedKeys[i].Less(expectedKeys[j])\n\t})\n\tsort.Slice(actualKeys, func(i, j int) bool {\n\t\treturn actualKeys[i].Less(actualKeys[j])\n\t})\n\tfor i := 0; i != len(expectedKeys); i++ {\n\t\ts.True(taskKeyEquals(expectedKeys[i], actualKeys[i]))\n\t}\n}\n\nfunc (s *processingQueueSuite) copyProcessingQueue(\n\tqueue *processingQueueImpl,\n) *processingQueueImpl {\n\ttasks := make(map[task.Key]task.Task)\n\tfor key, task := range queue.outstandingTasks {\n\t\ttasks[key] = task\n\t}\n\n\treturn s.newTestProcessingQueue(\n\t\tqueue.state.level,\n\t\tqueue.state.ackLevel,\n\t\tqueue.state.readLevel,\n\t\tqueue.state.maxLevel,\n\t\tqueue.state.domainFilter.copy(),\n\t\ttasks,\n\t)\n}\n\nfunc (s *processingQueueSuite) newTestProcessingQueue(\n\tlevel int,\n\tackLevel task.Key,\n\treadLevel task.Key,\n\tmaxLevel task.Key,\n\tdomainFilter DomainFilter,\n\toutstandingTasks map[task.Key]task.Task,\n) *processingQueueImpl {\n\treturn newProcessingQueue(\n\t\t&processingQueueStateImpl{\n\t\t\tlevel:        level,\n\t\t\tackLevel:     ackLevel,\n\t\t\treadLevel:    readLevel,\n\t\t\tmaxLevel:     maxLevel,\n\t\t\tdomainFilter: domainFilter,\n\t\t},\n\t\toutstandingTasks,\n\t\ts.logger,\n\t\ts.metricsClient,\n\t)\n}\n\nfunc (s *processingQueueSuite) newMockTasksForDomain(\n\tkeys []task.Key,\n\tdomainID []string,\n) map[task.Key]task.Task {\n\ttasks := make(map[task.Key]task.Task)\n\ts.Equal(len(keys), len(domainID))\n\n\tfor i := 0; i != len(keys); i++ {\n\t\tmockTask := task.NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetDomainID().Return(domainID[i]).AnyTimes()\n\t\ttasks[keys[i]] = mockTask\n\t}\n\n\treturn tasks\n}\n\nfunc (k testKey) Less(key task.Key) bool {\n\treturn k.ID < key.(testKey).ID\n}\n"
  },
  {
    "path": "service/history/queue/processor_base.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\twarnPendingTasks = 2000\n)\n\ntype (\n\tupdateMaxReadLevelFn          func() task.Key\n\tupdateClusterAckLevelFn       func(task.Key) error // TODO: deprecate this in favor of updateProcessingQueueStatesFn\n\tupdateProcessingQueueStatesFn func([]ProcessingQueueState) error\n\tqueueShutdownFn               func() error\n\n\tactionNotification struct {\n\t\tctx                  context.Context\n\t\taction               *Action\n\t\tresultNotificationCh chan actionResultNotification\n\t}\n\n\tactionResultNotification struct {\n\t\tresult *ActionResult\n\t\terr    error\n\t}\n\n\tprocessorBase struct {\n\t\tshard         shard.Context\n\t\ttaskProcessor task.Processor\n\t\tredispatcher  task.Redispatcher\n\n\t\toptions                     *queueProcessorOptions\n\t\tupdateMaxReadLevel          updateMaxReadLevelFn\n\t\tupdateClusterAckLevel       updateClusterAckLevelFn\n\t\tupdateProcessingQueueStates updateProcessingQueueStatesFn\n\t\tqueueShutdown               queueShutdownFn\n\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tmetricsScope  metrics.Scope\n\n\t\trateLimiter quotas.Limiter\n\n\t\tstatus         int32\n\t\tshutdownWG     sync.WaitGroup\n\t\tshutdownCh     chan struct{}\n\t\tactionNotifyCh chan actionNotification\n\n\t\tprocessingQueueCollections []ProcessingQueueCollection\n\t}\n)\n\nfunc newProcessorBase(\n\tshard shard.Context,\n\tprocessingQueueStates []ProcessingQueueState,\n\ttaskProcessor task.Processor,\n\toptions *queueProcessorOptions,\n\tupdateMaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\tqueueShutdown queueShutdownFn,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) *processorBase {\n\tmetricsScope := metricsClient.Scope(options.MetricScope).Tagged(metrics.ShardIDTag(shard.GetShardID()))\n\treturn &processorBase{\n\t\tshard:         shard,\n\t\ttaskProcessor: taskProcessor,\n\t\tredispatcher: task.NewRedispatcher(\n\t\t\ttaskProcessor,\n\t\t\tshard.GetTimeSource(),\n\t\t\t&task.RedispatcherOptions{\n\t\t\t\tTaskRedispatchInterval: options.RedispatchInterval,\n\t\t\t},\n\t\t\tlogger,\n\t\t\tmetricsScope,\n\t\t),\n\t\toptions:                     options,\n\t\tupdateMaxReadLevel:          updateMaxReadLevel,\n\t\tupdateClusterAckLevel:       updateClusterAckLevel,\n\t\tupdateProcessingQueueStates: updateProcessingQueueStates,\n\t\tqueueShutdown:               queueShutdown,\n\t\tlogger:                      logger,\n\t\tmetricsClient:               metricsClient,\n\t\tmetricsScope:                metricsScope,\n\t\trateLimiter:                 quotas.NewDynamicRateLimiter(options.MaxPollRPS.AsFloat64()),\n\t\tstatus:                      common.DaemonStatusInitialized,\n\t\tshutdownCh:                  make(chan struct{}),\n\t\tactionNotifyCh:              make(chan actionNotification),\n\t\tprocessingQueueCollections:  newProcessingQueueCollections(processingQueueStates, logger, metricsClient),\n\t}\n}\n\nfunc (p *processorBase) updateAckLevel() (bool, task.Key, error) {\n\tp.metricsScope.IncCounter(metrics.AckLevelUpdateCounter)\n\tvar minAckLevel task.Key\n\ttotalPengingTasks := 0\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tackLevel, numPendingTasks := queueCollection.UpdateAckLevels()\n\t\tif ackLevel == nil {\n\t\t\t// ack level may be nil if the queueCollection doesn't contain any processing queue\n\t\t\t// after updating ack levels\n\t\t\tcontinue\n\t\t}\n\n\t\ttotalPengingTasks += numPendingTasks\n\t\tif minAckLevel == nil {\n\t\t\tminAckLevel = ackLevel\n\t\t} else {\n\t\t\tminAckLevel = minTaskKey(minAckLevel, ackLevel)\n\t\t}\n\t}\n\n\tif minAckLevel == nil {\n\t\t// note that only failover processor will meet this condition\n\t\terr := p.queueShutdown()\n\t\tif err != nil {\n\t\t\tp.logger.Error(\"Error shutdown queue\", tag.Error(err))\n\t\t\t// return error so that shutdown callback can be retried\n\t\t\treturn false, nil, err\n\t\t}\n\t\treturn true, nil, nil\n\t}\n\n\tif totalPengingTasks > warnPendingTasks {\n\t\tp.logger.Warn(\"Too many pending tasks.\", tag.Number(int64(totalPengingTasks)))\n\t}\n\t// TODO: consider move pendingTasksTime metrics from shardInfoScope to queue processor scope\n\tp.metricsClient.RecordTimer(metrics.ShardInfoScope, getPendingTasksMetricIdx(p.options.MetricScope), time.Duration(totalPengingTasks))\n\n\tif p.options.EnablePersistQueueStates() && p.updateProcessingQueueStates != nil {\n\t\tstates := p.getProcessingQueueStates().GetStateActionResult.States\n\t\tif err := p.updateProcessingQueueStates(states); err != nil {\n\t\t\tp.logger.Error(\"Error persisting processing queue states\", tag.Error(err), tag.OperationFailed)\n\t\t\tp.metricsScope.IncCounter(metrics.AckLevelUpdateFailedCounter)\n\t\t\treturn false, minAckLevel, err\n\t\t}\n\t} else {\n\t\tif err := p.updateClusterAckLevel(minAckLevel); err != nil {\n\t\t\tp.logger.Error(\"Error updating ack level for shard\", tag.Error(err), tag.OperationFailed)\n\t\t\tp.metricsScope.IncCounter(metrics.AckLevelUpdateFailedCounter)\n\t\t\treturn false, minAckLevel, err\n\t\t}\n\t}\n\n\treturn false, minAckLevel, nil\n}\n\nfunc (p *processorBase) initializeSplitPolicy(lookAheadFunc lookAheadFunc) ProcessingQueueSplitPolicy {\n\tif !p.options.EnableSplit() {\n\t\treturn nil\n\t}\n\n\t// note the order of policies matters, check the comment for aggregated split policy\n\tvar policies []ProcessingQueueSplitPolicy\n\tmaxNewQueueLevel := p.options.SplitMaxLevel()\n\n\tpendingTaskThresholds, err := dynamicproperties.ConvertDynamicConfigMapPropertyToIntMap(p.options.PendingTaskSplitThreshold())\n\tif err != nil {\n\t\tp.logger.Error(\"Failed to convert pending task threshold\", tag.Error(err))\n\t} else {\n\t\tpolicies = append(policies, NewPendingTaskSplitPolicy(\n\t\t\tpendingTaskThresholds,\n\t\t\tp.options.EnablePendingTaskSplitByDomainID,\n\t\t\tlookAheadFunc,\n\t\t\tmaxNewQueueLevel,\n\t\t\tp.logger,\n\t\t\tp.metricsScope,\n\t\t))\n\t}\n\n\ttaskAttemptThresholds, err := dynamicproperties.ConvertDynamicConfigMapPropertyToIntMap(p.options.StuckTaskSplitThreshold())\n\tif err != nil {\n\t\tp.logger.Error(\"Failed to convert stuck task threshold\", tag.Error(err))\n\t} else {\n\t\tpolicies = append(policies, NewStuckTaskSplitPolicy(\n\t\t\ttaskAttemptThresholds,\n\t\t\tp.options.EnableStuckTaskSplitByDomainID,\n\t\t\tmaxNewQueueLevel,\n\t\t\tp.logger,\n\t\t\tp.metricsScope,\n\t\t))\n\t}\n\n\trandomSplitProbability := p.options.RandomSplitProbability()\n\tif randomSplitProbability != float64(0) {\n\t\tpolicies = append(policies, NewRandomSplitPolicy(\n\t\t\trandomSplitProbability,\n\t\t\tp.options.EnableRandomSplitByDomainID,\n\t\t\tmaxNewQueueLevel,\n\t\t\tlookAheadFunc,\n\t\t\tp.logger,\n\t\t\tp.metricsScope,\n\t\t))\n\t}\n\n\tif len(policies) == 0 {\n\t\treturn nil\n\t}\n\n\treturn NewAggregatedSplitPolicy(policies...)\n}\n\nfunc (p *processorBase) splitProcessingQueueCollection(splitPolicy ProcessingQueueSplitPolicy, upsertPollTimeFn func(int, time.Time)) {\n\tdefer p.emitProcessingQueueMetrics()\n\n\tif splitPolicy == nil {\n\t\treturn\n\t}\n\n\tnewQueuesMap := make(map[int][][]ProcessingQueue)\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tcurrentNewQueuesMap := make(map[int][]ProcessingQueue)\n\t\tnewQueues := queueCollection.Split(splitPolicy)\n\t\tfor _, newQueue := range newQueues {\n\t\t\tnewQueueLevel := newQueue.State().Level()\n\t\t\tcurrentNewQueuesMap[newQueueLevel] = append(currentNewQueuesMap[newQueueLevel], newQueue)\n\t\t}\n\n\t\tfor newQueueLevel, queues := range currentNewQueuesMap {\n\t\t\tnewQueuesMap[newQueueLevel] = append(newQueuesMap[newQueueLevel], queues)\n\t\t}\n\t}\n\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tif queuesList, ok := newQueuesMap[queueCollection.Level()]; ok {\n\t\t\tfor _, queues := range queuesList {\n\t\t\t\tqueueCollection.Merge(queues)\n\t\t\t}\n\t\t}\n\t\tdelete(newQueuesMap, queueCollection.Level())\n\t}\n\n\tfor level, newQueuesList := range newQueuesMap {\n\t\tnewQueueCollection := NewProcessingQueueCollection(\n\t\t\tlevel,\n\t\t\t[]ProcessingQueue{},\n\t\t)\n\t\tfor _, newQueues := range newQueuesList {\n\t\t\tnewQueueCollection.Merge(newQueues)\n\t\t}\n\t\tp.processingQueueCollections = append(p.processingQueueCollections, newQueueCollection)\n\t\tdelete(newQueuesMap, level)\n\t}\n\n\t// there can be new queue collections created or new queues added to an existing collection\n\tfor _, queueCollections := range p.processingQueueCollections {\n\t\tupsertPollTimeFn(queueCollections.Level(), time.Time{})\n\t}\n}\n\nfunc (p *processorBase) emitProcessingQueueMetrics() {\n\tnumProcessingQueues := 0\n\tmaxProcessingQueueLevel := 0\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tsize := len(queueCollection.Queues())\n\t\tnumProcessingQueues += size\n\t\tif size != 0 && queueCollection.Level() > maxProcessingQueueLevel {\n\t\t\tmaxProcessingQueueLevel = queueCollection.Level()\n\t\t}\n\t}\n\tp.metricsScope.RecordTimer(metrics.ProcessingQueueNumTimer, time.Duration(numProcessingQueues))\n\tp.metricsScope.RecordTimer(metrics.ProcessingQueueMaxLevelTimer, time.Duration(maxProcessingQueueLevel))\n}\n\nfunc (p *processorBase) addAction(ctx context.Context, action *Action) (chan actionResultNotification, bool) {\n\tresultNotificationCh := make(chan actionResultNotification, 1)\n\tif ctx == nil {\n\t\tctx = context.Background()\n\t}\n\tselect {\n\tcase p.actionNotifyCh <- actionNotification{\n\t\tctx:                  ctx,\n\t\taction:               action,\n\t\tresultNotificationCh: resultNotificationCh,\n\t}:\n\t\treturn resultNotificationCh, true\n\tcase <-p.shutdownCh:\n\t\tclose(resultNotificationCh)\n\t\treturn nil, false\n\tcase <-ctx.Done():\n\t\tclose(resultNotificationCh)\n\t\treturn nil, false\n\t}\n}\n\nfunc (p *processorBase) handleActionNotification(notification actionNotification, postActionFn func()) {\n\tvar result *ActionResult\n\tvar err error\n\tswitch notification.action.ActionType {\n\tcase ActionTypeReset:\n\t\tresult, err = p.resetProcessingQueueStates()\n\tcase ActionTypeGetState:\n\t\tresult = p.getProcessingQueueStates()\n\tdefault:\n\t\terr = fmt.Errorf(\"unknown queue action type: %v\", notification.action.ActionType)\n\t}\n\n\tnotification.resultNotificationCh <- actionResultNotification{\n\t\tresult: result,\n\t\terr:    err,\n\t}\n\n\tclose(notification.resultNotificationCh)\n\n\tif err == nil {\n\t\t// only run post action when the action complete successfully\n\t\tpostActionFn()\n\t}\n}\n\nfunc (p *processorBase) resetProcessingQueueStates() (*ActionResult, error) {\n\tvar minAckLevel task.Key\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tackLevel, _ := queueCollection.UpdateAckLevels()\n\t\tif ackLevel == nil {\n\t\t\t// ack level may be nil if the queueCollection doesn't contain any processing queue\n\t\t\t// after updating ack levels\n\t\t\tcontinue\n\t\t}\n\n\t\tif minAckLevel == nil {\n\t\t\tminAckLevel = ackLevel\n\t\t} else {\n\t\t\tminAckLevel = minTaskKey(minAckLevel, ackLevel)\n\t\t}\n\t}\n\n\tif minAckLevel == nil {\n\t\t// reset queue can't be invoked for failover queue, so if this happens, there's must be a\n\t\t// bug in the queue split implementation\n\t\tp.logger.Fatal(\"unable to find minAckLevel during reset\", tag.Value(p.processingQueueCollections))\n\t}\n\n\tvar maxReadLevel task.Key\n\tswitch p.options.MetricScope {\n\tcase metrics.TransferActiveQueueProcessorScope, metrics.TransferStandbyQueueProcessorScope:\n\t\tmaxReadLevel = maximumTransferTaskKey\n\tcase metrics.TimerActiveQueueProcessorScope, metrics.TimerStandbyQueueProcessorScope:\n\t\tmaxReadLevel = maximumTimerTaskKey\n\t}\n\n\tp.processingQueueCollections = newProcessingQueueCollections(\n\t\t[]ProcessingQueueState{\n\t\t\tNewProcessingQueueState(\n\t\t\t\tdefaultProcessingQueueLevel,\n\t\t\t\tminAckLevel,\n\t\t\t\tmaxReadLevel,\n\t\t\t\tNewDomainFilter(nil, true),\n\t\t\t),\n\t\t},\n\t\tp.logger,\n\t\tp.metricsClient,\n\t)\n\n\treturn &ActionResult{\n\t\tActionType:        ActionTypeReset,\n\t\tResetActionResult: &ResetActionResult{},\n\t}, nil\n}\n\nfunc (p *processorBase) getProcessingQueueStates() *ActionResult {\n\tvar queueStates []ProcessingQueueState\n\tfor _, queueCollection := range p.processingQueueCollections {\n\t\tfor _, queue := range queueCollection.Queues() {\n\t\t\tqueueStates = append(queueStates, copyQueueState(queue.State()))\n\t\t}\n\t}\n\n\treturn &ActionResult{\n\t\tActionType: ActionTypeGetState,\n\t\tGetStateActionResult: &GetStateActionResult{\n\t\t\tStates: queueStates,\n\t\t},\n\t}\n}\n\nfunc (p *processorBase) submitTask(task task.Task) (bool, error) {\n\tsubmitted, err := p.taskProcessor.TrySubmit(task)\n\tif err != nil {\n\t\tselect {\n\t\tcase <-p.shutdownCh:\n\t\t\t// if error is due to shard shutdown\n\t\t\treturn false, err\n\t\tdefault:\n\t\t\t// otherwise it might be error from domain cache etc, add\n\t\t\t// the task to redispatch queue so that it can be retried\n\t\t\tp.logger.Error(\"Failed to submit task\", tag.Error(err))\n\t\t}\n\t}\n\tif err != nil || !submitted {\n\t\tp.redispatcher.AddTask(task)\n\t\treturn false, nil\n\t}\n\n\treturn true, nil\n}\n\nfunc getPendingTasksMetricIdx(scopeIdx metrics.ScopeIdx) metrics.MetricIdx {\n\tswitch scopeIdx {\n\tcase metrics.TimerActiveQueueProcessorScope:\n\t\treturn metrics.ShardInfoTimerActivePendingTasksTimer\n\tcase metrics.TimerStandbyQueueProcessorScope:\n\t\treturn metrics.ShardInfoTimerStandbyPendingTasksTimer\n\tcase metrics.TransferActiveQueueProcessorScope:\n\t\treturn metrics.ShardInfoTransferActivePendingTasksTimer\n\tcase metrics.TransferStandbyQueueProcessorScope:\n\t\treturn metrics.ShardInfoTransferStandbyPendingTasksTimer\n\tcase metrics.CrossClusterQueueProcessorScope:\n\t\treturn metrics.ShardInfoCrossClusterPendingTasksTimer\n\tcase metrics.ReplicatorQueueProcessorScope:\n\t\treturn metrics.ShardInfoReplicationPendingTasksTimer\n\tdefault:\n\t\tpanic(\"unknown queue processor metric scope\")\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/processor_base_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"sort\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tprocessorBaseSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller        *gomock.Controller\n\t\tmockShard         *shard.TestContext\n\t\tmockTaskProcessor *task.MockProcessor\n\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tmetricsScope  metrics.Scope\n\t}\n)\n\nfunc TestProcessorBaseSuite(t *testing.T) {\n\ts := new(processorBaseSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *processorBaseSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockTaskProcessor = task.NewMockProcessor(s.controller)\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.metricsScope = s.metricsClient.Scope(metrics.TransferQueueProcessorScope)\n}\n\nfunc (s *processorBaseSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *processorBaseSuite) TestSplitQueue() {\n\tmockQueueSplitPolicy := NewMockProcessingQueueSplitPolicy(s.controller)\n\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(0),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, true),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTransferTaskKey(0),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{}, true),\n\t\t),\n\t}\n\tmockQueueSplitPolicy.EXPECT().Evaluate(NewProcessingQueue(processingQueueStates[0], s.logger, s.metricsClient)).Return(nil).Times(1)\n\tmockQueueSplitPolicy.EXPECT().Evaluate(NewProcessingQueue(processingQueueStates[1], s.logger, s.metricsClient)).Return([]ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTransferTaskKey(0),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}).Times(1)\n\tmockQueueSplitPolicy.EXPECT().Evaluate(NewProcessingQueue(processingQueueStates[2], s.logger, s.metricsClient)).Return([]ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}, \"testDomain3\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain2\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain3\": {}}, false),\n\t\t),\n\t}).Times(1)\n\n\tprocessorBase := s.newTestProcessorBase(\n\t\tprocessingQueueStates,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tnextPollTime := make(map[int]time.Time)\n\tprocessorBase.splitProcessingQueueCollection(\n\t\tmockQueueSplitPolicy,\n\t\tfunc(level int, pollTime time.Time) {\n\t\t\tnextPollTime[level] = pollTime\n\t\t},\n\t)\n\n\tprocessingQueueCollections := processorBase.processingQueueCollections\n\tsort.Slice(processingQueueCollections, func(i, j int) bool {\n\t\treturn processingQueueCollections[i].Level() < processingQueueCollections[j].Level()\n\t})\n\ts.Len(processingQueueCollections, 3)\n\ts.Len(processingQueueCollections[0].Queues(), 2)\n\ts.Len(processingQueueCollections[1].Queues(), 1)\n\ts.Len(processingQueueCollections[2].Queues(), 2)\n\tfor idx := 1; idx != len(processingQueueCollections)-1; idx++ {\n\t\ts.Less(\n\t\t\tprocessingQueueCollections[idx-1].Level(),\n\t\t\tprocessingQueueCollections[idx].Level(),\n\t\t)\n\t}\n\ts.Len(nextPollTime, 3)\n\tfor _, nextPollTime := range nextPollTime {\n\t\ts.Zero(nextPollTime)\n\t}\n}\n\nfunc (s *processorBaseSuite) TestUpdateAckLevel_Transfer_ProcessedFinished() {\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}}, true),\n\t\t),\n\t}\n\tqueueShutdown := false\n\tqueueShutdownFn := func() error {\n\t\tqueueShutdown = true\n\t\treturn nil\n\t}\n\n\tprocessorBase := s.newTestProcessorBase(\n\t\tprocessingQueueStates,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tqueueShutdownFn,\n\t)\n\n\tprocessFinished, ackLevel, err := processorBase.updateAckLevel()\n\ts.NoError(err)\n\ts.Nil(ackLevel)\n\ts.True(processFinished)\n\ts.True(queueShutdown)\n}\n\nfunc (s *processorBaseSuite) TestUpdateAckLevel_Tranfer_ProcessNotFinished() {\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTransferTaskKey(5),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTransferTaskKey(2),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}}, true),\n\t\t),\n\t}\n\tupdateAckLevel := int64(0)\n\tupdateTransferAckLevelFn := func(ackLevel task.Key) error {\n\t\tupdateAckLevel = ackLevel.(transferTaskKey).taskID\n\t\treturn nil\n\t}\n\n\tprocessorBase := s.newTestProcessorBase(\n\t\tprocessingQueueStates,\n\t\tnil,\n\t\tupdateTransferAckLevelFn,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tprocessFinished, ackLevel, err := processorBase.updateAckLevel()\n\ts.NoError(err)\n\ts.False(processFinished)\n\ts.Equal(int64(2), updateAckLevel)\n\ts.Equal(int64(2), ackLevel.(transferTaskKey).taskID)\n}\n\nfunc (s *processorBaseSuite) TestUpdateAckLevel_Timer_UpdateAckLevel() {\n\tnow := time.Now()\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTimerTaskKey(now.Add(-5*time.Second), 0),\n\t\t\tnewTimerTaskKey(now, 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTimerTaskKey(now.Add(-3*time.Second), 0),\n\t\t\tnewTimerTaskKey(now.Add(5*time.Second), 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTimerTaskKey(now.Add(-1*time.Second), 0),\n\t\t\tnewTimerTaskKey(now.Add(100*time.Second), 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}}, true),\n\t\t),\n\t}\n\tupdateAckLevel := time.Time{}\n\tupdateTransferAckLevelFn := func(ackLevel task.Key) error {\n\t\tupdateAckLevel = ackLevel.(timerTaskKey).visibilityTimestamp\n\t\treturn nil\n\t}\n\n\ttimerQueueProcessBase := s.newTestProcessorBase(processingQueueStates, nil, updateTransferAckLevelFn, nil, nil)\n\ttimerQueueProcessBase.options.EnablePersistQueueStates = dynamicproperties.GetBoolPropertyFn(true)\n\tprocessFinished, ackLevel, err := timerQueueProcessBase.updateAckLevel()\n\ts.NoError(err)\n\ts.False(processFinished)\n\ts.Equal(now.Add(-5*time.Second), updateAckLevel)\n\ts.Equal(now.Add(-5*time.Second), ackLevel.(timerTaskKey).visibilityTimestamp)\n}\n\nfunc (s *processorBaseSuite) TestUpdateAckLevel_Timer_UpdateQueueStates() {\n\tnow := time.Now()\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t2,\n\t\t\tnewTimerTaskKey(now.Add(-5*time.Second), 0),\n\t\t\tnewTimerTaskKey(now, 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTimerTaskKey(now.Add(-3*time.Second), 0),\n\t\t\tnewTimerTaskKey(now.Add(5*time.Second), 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTimerTaskKey(now.Add(-1*time.Second), 0),\n\t\t\tnewTimerTaskKey(now.Add(100*time.Second), 0),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}}, true),\n\t\t),\n\t}\n\n\tvar pState []*types.ProcessingQueueState\n\tupdateProcessingQueueStates := func(states []ProcessingQueueState) error {\n\t\tpState = convertToPersistenceTimerProcessingQueueStates(states)\n\t\treturn nil\n\t}\n\n\ttimerQueueProcessBase := s.newTestProcessorBase(processingQueueStates, nil, nil, updateProcessingQueueStates, nil)\n\ttimerQueueProcessBase.options.EnablePersistQueueStates = dynamicproperties.GetBoolPropertyFn(true)\n\tprocessFinished, ackLevel, err := timerQueueProcessBase.updateAckLevel()\n\ts.NoError(err)\n\ts.False(processFinished)\n\ts.Equal(len(processingQueueStates), len(pState))\n\ts.Equal(now.Add(-5*time.Second), ackLevel.(timerTaskKey).visibilityTimestamp)\n}\n\nfunc (s *processorBaseSuite) TestInitializeSplitPolicy_Disabled() {\n\tprocessorBase := s.newTestProcessorBase(\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tsplitPolicy := processorBase.initializeSplitPolicy(nil)\n\ts.Nil(splitPolicy, \"got non-nil split policy, want nil because it's disabled\")\n}\n\nfunc (s *processorBaseSuite) TestInitializeSplitPolicy_Enabled() {\n\tprocessorBase := s.newTestProcessorBase(nil, nil, nil, nil, nil)\n\n\tprocessorBase.options.EnableSplit = dynamicproperties.GetBoolPropertyFn(true)\n\n\tsplitPolicy := processorBase.initializeSplitPolicy(nil)\n\ts.NotNil(splitPolicy, \"got nil split policy, want non-nil\")\n\taggPolicy, ok := splitPolicy.(*aggregatedSplitPolicy)\n\ts.True(ok, \"got %T, want *aggregatedSplitPolicy\", splitPolicy)\n\ts.Equal(3, len(aggPolicy.policies), \"got %v policies, want 3: pending task policy, stuck task policy and random split policy\", len(aggPolicy.policies))\n}\n\nfunc (s *processorBaseSuite) TestResetProcessingQueueStates() {\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(0),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, true),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t1,\n\t\t\tnewTransferTaskKey(0),\n\t\t\tnewTransferTaskKey(100),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\t0,\n\t\t\tnewTransferTaskKey(100),\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tNewDomainFilter(map[string]struct{}{}, true),\n\t\t),\n\t}\n\tprocessorBase := s.newTestProcessorBase(processingQueueStates, nil, nil, nil, nil)\n\n\tres, err := processorBase.resetProcessingQueueStates()\n\ts.NoError(err, \"no error expected\")\n\ts.Equal(ActionTypeReset, res.ActionType, \"got action type %v, want %v\", res.ActionType, ActionTypeReset)\n}\n\nfunc (s *processorBaseSuite) newTestProcessorBase(\n\tprocessingQueueStates []ProcessingQueueState,\n\tupdateMaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\tqueueShutdown queueShutdownFn,\n) *processorBase {\n\treturn newProcessorBase(\n\t\ts.mockShard,\n\t\tprocessingQueueStates,\n\t\ts.mockTaskProcessor,\n\t\tnewTransferQueueProcessorOptions(s.mockShard.GetConfig(), true, false),\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\ts.logger,\n\t\ts.metricsClient,\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/processor_options.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype queueProcessorOptions struct {\n\tBatchSize                            dynamicproperties.IntPropertyFn\n\tDeleteBatchSize                      dynamicproperties.IntPropertyFn\n\tMaxPollRPS                           dynamicproperties.IntPropertyFn\n\tMaxPollInterval                      dynamicproperties.DurationPropertyFn\n\tMaxPollIntervalJitterCoefficient     dynamicproperties.FloatPropertyFn\n\tUpdateAckInterval                    dynamicproperties.DurationPropertyFn\n\tUpdateAckIntervalJitterCoefficient   dynamicproperties.FloatPropertyFn\n\tRedispatchInterval                   dynamicproperties.DurationPropertyFn\n\tMaxRedispatchQueueSize               dynamicproperties.IntPropertyFn\n\tMaxStartJitterInterval               dynamicproperties.DurationPropertyFn\n\tSplitQueueInterval                   dynamicproperties.DurationPropertyFn\n\tSplitQueueIntervalJitterCoefficient  dynamicproperties.FloatPropertyFn\n\tEnableSplit                          dynamicproperties.BoolPropertyFn\n\tSplitMaxLevel                        dynamicproperties.IntPropertyFn\n\tEnableRandomSplitByDomainID          dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tRandomSplitProbability               dynamicproperties.FloatPropertyFn\n\tEnablePendingTaskSplitByDomainID     dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tPendingTaskSplitThreshold            dynamicproperties.MapPropertyFn\n\tEnableStuckTaskSplitByDomainID       dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\tStuckTaskSplitThreshold              dynamicproperties.MapPropertyFn\n\tSplitLookAheadDurationByDomainID     dynamicproperties.DurationPropertyFnWithDomainIDFilter\n\tPollBackoffInterval                  dynamicproperties.DurationPropertyFn\n\tPollBackoffIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\tEnablePersistQueueStates             dynamicproperties.BoolPropertyFn\n\tEnableLoadQueueStates                dynamicproperties.BoolPropertyFn\n\tEnableGracefulSyncShutdown           dynamicproperties.BoolPropertyFn\n\tEnableValidator                      dynamicproperties.BoolPropertyFn\n\tValidationInterval                   dynamicproperties.DurationPropertyFn\n\t// MaxPendingTaskSize is used in cross cluster queue to limit the pending task count\n\tMaxPendingTaskSize dynamicproperties.IntPropertyFn\n\tMetricScope        metrics.ScopeIdx\n}\n"
  },
  {
    "path": "service/history/queue/queue_processor_util.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc convertToPersistenceTransferProcessingQueueStates(states []ProcessingQueueState) []*types.ProcessingQueueState {\n\tpStates := make([]*types.ProcessingQueueState, 0, len(states))\n\tfor _, state := range states {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        common.Int32Ptr(int32(state.Level())),\n\t\t\tAckLevel:     common.Int64Ptr(state.AckLevel().(transferTaskKey).taskID),\n\t\t\tMaxLevel:     common.Int64Ptr(state.MaxLevel().(transferTaskKey).taskID),\n\t\t\tDomainFilter: convertToPersistenceDomainFilter(state.DomainFilter()),\n\t\t})\n\t}\n\n\treturn pStates\n}\n\nfunc convertFromPersistenceTransferProcessingQueueStates(pStates []*types.ProcessingQueueState) []ProcessingQueueState {\n\tstates := make([]ProcessingQueueState, 0, len(pStates))\n\tfor _, pState := range pStates {\n\t\tstates = append(states, NewProcessingQueueState(\n\t\t\tint(pState.GetLevel()),\n\t\t\tnewTransferTaskKey(pState.GetAckLevel()),\n\t\t\tnewTransferTaskKey(pState.GetMaxLevel()),\n\t\t\tconvertFromPersistenceDomainFilter(pState.DomainFilter),\n\t\t))\n\t}\n\n\treturn states\n}\n\nfunc convertToPersistenceTimerProcessingQueueStates(states []ProcessingQueueState) []*types.ProcessingQueueState {\n\tpStates := make([]*types.ProcessingQueueState, 0, len(states))\n\tfor _, state := range states {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        common.Int32Ptr(int32(state.Level())),\n\t\t\tAckLevel:     common.Int64Ptr(state.AckLevel().(timerTaskKey).visibilityTimestamp.UnixNano()),\n\t\t\tMaxLevel:     common.Int64Ptr(state.MaxLevel().(timerTaskKey).visibilityTimestamp.UnixNano()),\n\t\t\tDomainFilter: convertToPersistenceDomainFilter(state.DomainFilter()),\n\t\t})\n\t}\n\n\treturn pStates\n}\n\nfunc convertFromPersistenceTimerProcessingQueueStates(pStates []*types.ProcessingQueueState) []ProcessingQueueState {\n\tstates := make([]ProcessingQueueState, 0, len(pStates))\n\tfor _, pState := range pStates {\n\t\tstates = append(states, NewProcessingQueueState(\n\t\t\tint(pState.GetLevel()),\n\t\t\tnewTimerTaskKey(time.Unix(0, pState.GetAckLevel()), 0),\n\t\t\tnewTimerTaskKey(time.Unix(0, pState.GetMaxLevel()), 0),\n\t\t\tconvertFromPersistenceDomainFilter(pState.DomainFilter),\n\t\t))\n\t}\n\n\treturn states\n}\n\nfunc convertToPersistenceDomainFilter(domainFilter DomainFilter) *types.DomainFilter {\n\tdomainIDs := make([]string, 0, len(domainFilter.DomainIDs))\n\tfor domainID := range domainFilter.DomainIDs {\n\t\tdomainIDs = append(domainIDs, domainID)\n\t}\n\n\treturn &types.DomainFilter{\n\t\tDomainIDs:    domainIDs,\n\t\tReverseMatch: domainFilter.ReverseMatch,\n\t}\n}\n\nfunc convertFromPersistenceDomainFilter(domainFilter *types.DomainFilter) DomainFilter {\n\tdomainIDs := make(map[string]struct{})\n\tfor _, domainID := range domainFilter.DomainIDs {\n\t\tdomainIDs[domainID] = struct{}{}\n\t}\n\n\treturn NewDomainFilter(domainIDs, domainFilter.GetReverseMatch())\n}\n\nfunc validateProcessingQueueStates(pStates []*types.ProcessingQueueState, ackLevel interface{}) bool {\n\tif len(pStates) == 0 {\n\t\treturn false\n\t}\n\n\tminAckLevel := pStates[0].GetAckLevel()\n\tfor _, pState := range pStates {\n\t\tminAckLevel = min(minAckLevel, pState.GetAckLevel())\n\t}\n\n\tswitch ackLevel := ackLevel.(type) {\n\tcase int64:\n\t\treturn minAckLevel == ackLevel\n\tcase time.Time:\n\t\treturn minAckLevel == ackLevel.UnixNano()\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/queue_processor_util_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tqueueProcessorUtilSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t}\n)\n\nfunc TestQueueProcessorUtilSuite(t *testing.T) {\n\ts := new(queueProcessorUtilSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *queueProcessorUtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *queueProcessorUtilSuite) TestConvertToPersistenceTransferProcessingQueueStates() {\n\tlevels := []int{0, 1}\n\tackLevels := []int64{123, 456}\n\tmaxLevels := []int64{456, 789}\n\tdomainFilters := []DomainFilter{\n\t\tNewDomainFilter(nil, true),\n\t\tNewDomainFilter(map[string]struct{}{\"domain 1\": {}, \"domain 2\": {}}, false),\n\t}\n\tstates := []ProcessingQueueState{}\n\tfor i := 0; i != len(levels); i++ {\n\t\tstates = append(states, NewProcessingQueueState(\n\t\t\tlevels[i],\n\t\t\tnewTransferTaskKey(ackLevels[i]),\n\t\t\tnewTransferTaskKey(maxLevels[i]),\n\t\t\tdomainFilters[i],\n\t\t))\n\t}\n\n\tpStates := convertToPersistenceTransferProcessingQueueStates(states)\n\ts.Equal(len(states), len(pStates))\n\tfor i := 0; i != len(states); i++ {\n\t\ts.assertProcessingQueueStateEqual(states[i], pStates[i])\n\t}\n}\n\nfunc (s *queueProcessorUtilSuite) TestConvertFromPersistenceTransferProcessingQueueStates() {\n\tlevels := []int32{0, 1}\n\tackLevels := []int64{123, 456}\n\tmaxLevels := []int64{456, 789}\n\tdomainFilters := []*types.DomainFilter{\n\t\t{\n\t\t\tDomainIDs:    nil,\n\t\t\tReverseMatch: true,\n\t\t},\n\t\t{\n\t\t\tDomainIDs:    []string{\"domain 1\", \"domain 2\"},\n\t\t\tReverseMatch: false,\n\t\t},\n\t}\n\tpStates := []*types.ProcessingQueueState{}\n\tfor i := 0; i != len(levels); i++ {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        common.Int32Ptr(levels[i]),\n\t\t\tAckLevel:     common.Int64Ptr(ackLevels[i]),\n\t\t\tMaxLevel:     common.Int64Ptr(maxLevels[i]),\n\t\t\tDomainFilter: domainFilters[i],\n\t\t})\n\t}\n\n\tstates := convertFromPersistenceTransferProcessingQueueStates(pStates)\n\ts.Equal(len(pStates), len(states))\n\tfor i := 0; i != len(pStates); i++ {\n\t\ts.assertProcessingQueueStateEqual(states[i], pStates[i])\n\t}\n}\n\nfunc (s *queueProcessorUtilSuite) TestConvertToPersistenceTimerProcessingQueueStates() {\n\tlevels := []int{0, 1}\n\tackLevels := []time.Time{time.Now(), time.Now().Add(-time.Minute)}\n\tmaxLevels := []time.Time{time.Now().Add(time.Hour), time.Now().Add(time.Nanosecond)}\n\tdomainFilters := []DomainFilter{\n\t\tNewDomainFilter(nil, true),\n\t\tNewDomainFilter(map[string]struct{}{\"domain 1\": {}, \"domain 2\": {}}, false),\n\t}\n\tstates := []ProcessingQueueState{}\n\tfor i := 0; i != len(levels); i++ {\n\t\tstates = append(states, NewProcessingQueueState(\n\t\t\tlevels[i],\n\t\t\tnewTimerTaskKey(ackLevels[i], 0),\n\t\t\tnewTimerTaskKey(maxLevels[i], 0),\n\t\t\tdomainFilters[i],\n\t\t))\n\t}\n\n\tpStates := convertToPersistenceTimerProcessingQueueStates(states)\n\ts.Equal(len(states), len(pStates))\n\tfor i := 0; i != len(states); i++ {\n\t\ts.assertProcessingQueueStateEqual(states[i], pStates[i])\n\t}\n}\n\nfunc (s *queueProcessorUtilSuite) TestConvertFromPersistenceTimerProcessingQueueStates() {\n\tlevels := []int32{0, 1}\n\tackLevels := []time.Time{time.Now(), time.Now().Add(-time.Minute)}\n\tmaxLevels := []time.Time{time.Now().Add(time.Hour), time.Now().Add(time.Nanosecond)}\n\tdomainFilters := []*types.DomainFilter{\n\t\t{\n\t\t\tDomainIDs:    nil,\n\t\t\tReverseMatch: true,\n\t\t},\n\t\t{\n\t\t\tDomainIDs:    []string{\"domain 1\", \"domain 2\"},\n\t\t\tReverseMatch: false,\n\t\t},\n\t}\n\tpStates := []*types.ProcessingQueueState{}\n\tfor i := 0; i != len(levels); i++ {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        common.Int32Ptr(levels[i]),\n\t\t\tAckLevel:     common.Int64Ptr(ackLevels[i].UnixNano()),\n\t\t\tMaxLevel:     common.Int64Ptr(maxLevels[i].UnixNano()),\n\t\t\tDomainFilter: domainFilters[i],\n\t\t})\n\t}\n\n\tstates := convertFromPersistenceTimerProcessingQueueStates(pStates)\n\ts.Equal(len(pStates), len(states))\n\tfor i := 0; i != len(pStates); i++ {\n\t\ts.assertProcessingQueueStateEqual(states[i], pStates[i])\n\t}\n}\n\nfunc (s *queueProcessorUtilSuite) assertProcessingQueueStateEqual(\n\tstate ProcessingQueueState,\n\tpState *types.ProcessingQueueState,\n) {\n\tvar ackLevel, maxLevel int64\n\tswitch taskKey := state.AckLevel().(type) {\n\tcase transferTaskKey:\n\t\tackLevel = taskKey.taskID\n\tcase timerTaskKey:\n\t\tackLevel = taskKey.visibilityTimestamp.UnixNano()\n\t\ts.Zero(taskKey.taskID)\n\t}\n\tswitch taskKey := state.MaxLevel().(type) {\n\tcase transferTaskKey:\n\t\tmaxLevel = taskKey.taskID\n\tcase timerTaskKey:\n\t\tmaxLevel = taskKey.visibilityTimestamp.UnixNano()\n\t\ts.Zero(taskKey.taskID)\n\t}\n\ts.Equal(state.Level(), int(pState.GetLevel()))\n\ts.Equal(ackLevel, pState.GetAckLevel())\n\ts.Equal(maxLevel, pState.GetMaxLevel())\n\ts.assertDomainFilterEqual(state.DomainFilter(), pState.GetDomainFilter())\n}\n\nfunc (s *queueProcessorUtilSuite) TestValidateProcessingQueueStates_Fail() {\n\tackLevels := []int64{123, 456}\n\tpStates := []*types.ProcessingQueueState{}\n\tfor i := 0; i != len(ackLevels); i++ {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        nil,\n\t\t\tAckLevel:     common.Int64Ptr(ackLevels[i]),\n\t\t\tMaxLevel:     nil,\n\t\t\tDomainFilter: nil,\n\t\t})\n\t}\n\n\tisValid := validateProcessingQueueStates(pStates, 789)\n\ts.False(isValid)\n}\n\nfunc (s *queueProcessorUtilSuite) TestValidateProcessingQueueStates_Success() {\n\tackLevels := []time.Time{time.Now(), time.Now().Add(-time.Minute)}\n\tpStates := []*types.ProcessingQueueState{}\n\tfor i := 0; i != len(ackLevels); i++ {\n\t\tpStates = append(pStates, &types.ProcessingQueueState{\n\t\t\tLevel:        nil,\n\t\t\tAckLevel:     common.Int64Ptr(ackLevels[i].UnixNano()),\n\t\t\tMaxLevel:     nil,\n\t\t\tDomainFilter: nil,\n\t\t})\n\t}\n\n\tisValid := validateProcessingQueueStates(pStates, ackLevels[1])\n\ts.True(isValid)\n}\n\nfunc (s *queueProcessorUtilSuite) assertDomainFilterEqual(\n\tdomainFilter DomainFilter,\n\tpDomainFilter *types.DomainFilter,\n) {\n\ts.Equal(domainFilter.ReverseMatch, pDomainFilter.GetReverseMatch())\n\ts.Equal(len(domainFilter.DomainIDs), len(pDomainFilter.GetDomainIDs()))\n\tfor _, domainID := range pDomainFilter.GetDomainIDs() {\n\t\t_, ok := domainFilter.DomainIDs[domainID]\n\t\ts.True(ok)\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/split_policy.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"fmt\"\n\t\"math/rand\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tt \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\tpolicyTypePendingTask int = iota + 1\n\tpolicyTypeStuckTask\n\tpolicyTypeSelectedDomain\n\tpolicyTypeRandom\n)\n\ntype (\n\tlookAheadFunc func(task.Key, string) task.Key\n\n\tpendingTaskSplitPolicy struct {\n\t\tpendingTaskThreshold map[int]int // queue level -> threshold\n\t\tenabledByDomainID    dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\t\tmaxNewQueueLevel     int\n\t\tlookAheadFunc        lookAheadFunc\n\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t}\n\n\tstuckTaskSplitPolicy struct {\n\t\tattemptThreshold  map[int]int // queue level -> threshold\n\t\tenabledByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\t\tmaxNewQueueLevel  int\n\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t}\n\n\tselectedDomainSplitPolicy struct {\n\t\tdomainIDs     map[string]struct{}\n\t\tnewQueueLevel int\n\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t}\n\n\trandomSplitPolicy struct {\n\t\tsplitProbability  float64\n\t\tenabledByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\t\tmaxNewQueueLevel  int\n\t\tlookAheadFunc     lookAheadFunc\n\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t}\n\n\taggregatedSplitPolicy struct {\n\t\tpolicies []ProcessingQueueSplitPolicy\n\t}\n)\n\n// NewPendingTaskSplitPolicy creates a new processing queue split policy\n// based on the number of pending tasks\nfunc NewPendingTaskSplitPolicy(\n\tpendingTaskThreshold map[int]int,\n\tenabledByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter,\n\tlookAheadFunc lookAheadFunc,\n\tmaxNewQueueLevel int,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) ProcessingQueueSplitPolicy {\n\treturn &pendingTaskSplitPolicy{\n\t\tpendingTaskThreshold: pendingTaskThreshold,\n\t\tenabledByDomainID:    enabledByDomainID,\n\t\tlookAheadFunc:        lookAheadFunc,\n\t\tmaxNewQueueLevel:     maxNewQueueLevel,\n\t\tlogger:               logger,\n\t\tmetricsScope:         metricsScope,\n\t}\n}\n\n// NewStuckTaskSplitPolicy creates a new processing queue split policy\n// based on the number of task attempts tasks\nfunc NewStuckTaskSplitPolicy(\n\tattemptThreshold map[int]int,\n\tenabledByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter,\n\tmaxNewQueueLevel int,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) ProcessingQueueSplitPolicy {\n\treturn &stuckTaskSplitPolicy{\n\t\tattemptThreshold:  attemptThreshold,\n\t\tenabledByDomainID: enabledByDomainID,\n\t\tmaxNewQueueLevel:  maxNewQueueLevel,\n\t\tlogger:            logger,\n\t\tmetricsScope:      metricsScope,\n\t}\n}\n\n// NewSelectedDomainSplitPolicy creates a new processing queue split policy\n// that splits out specific domainIDs\nfunc NewSelectedDomainSplitPolicy(\n\tdomainIDs map[string]struct{},\n\tnewQueueLevel int,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) ProcessingQueueSplitPolicy {\n\treturn &selectedDomainSplitPolicy{\n\t\tdomainIDs:     domainIDs,\n\t\tnewQueueLevel: newQueueLevel,\n\t\tlogger:        logger,\n\t\tmetricsScope:  metricsScope,\n\t}\n}\n\n// NewRandomSplitPolicy creates a split policy that will randomly split one\n// or more domains into a new processing queue\nfunc NewRandomSplitPolicy(\n\tsplitProbability float64,\n\tenabledByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter,\n\tmaxNewQueueLevel int,\n\tlookAheadFunc lookAheadFunc,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) ProcessingQueueSplitPolicy {\n\treturn &randomSplitPolicy{\n\t\tsplitProbability:  splitProbability,\n\t\tenabledByDomainID: enabledByDomainID,\n\t\tmaxNewQueueLevel:  maxNewQueueLevel,\n\t\tlookAheadFunc:     lookAheadFunc,\n\t\tlogger:            logger,\n\t\tmetricsScope:      metricsScope,\n\t}\n}\n\n// NewAggregatedSplitPolicy creates a new processing queue split policy\n// that which combines other policies. Policies are evaluated in the order\n// they passed in, and if one policy returns an non-empty result, that result\n// will be returned as is and policies after that one will not be evaluated\nfunc NewAggregatedSplitPolicy(policies ...ProcessingQueueSplitPolicy) ProcessingQueueSplitPolicy {\n\treturn &aggregatedSplitPolicy{\n\t\tpolicies: policies,\n\t}\n}\n\nfunc (p *pendingTaskSplitPolicy) Evaluate(queue ProcessingQueue) []ProcessingQueueState {\n\tqueueImpl := queue.(*processingQueueImpl)\n\n\tif queueImpl.state.level == p.maxNewQueueLevel {\n\t\t// already reaches max level, skip splitting\n\t\treturn nil\n\t}\n\n\tthreshold, ok := p.pendingTaskThreshold[queueImpl.state.level]\n\tif !ok {\n\t\t// no threshold specified for the level, skip splitting\n\t\treturn nil\n\t}\n\n\tpendingTasksPerDomain := make(map[string]int) // domainID -> # of pending tasks\n\tfor _, task := range queueImpl.outstandingTasks {\n\t\tif task.State() != t.TaskStateAcked {\n\t\t\tpendingTasksPerDomain[task.GetDomainID()]++\n\t\t}\n\t}\n\n\tdomainToSplit := make(map[string]struct{})\n\tfor domainID, pendingTasks := range pendingTasksPerDomain {\n\t\tif pendingTasks > threshold && p.enabledByDomainID(domainID) {\n\t\t\tdomainToSplit[domainID] = struct{}{}\n\t\t}\n\t}\n\n\tif len(domainToSplit) == 0 {\n\t\treturn nil\n\t}\n\n\tnewQueueLevel := queueImpl.state.level + 1 // split stuck tasks to current level + 1\n\tp.logger.Info(\"Split processing queue\",\n\t\ttag.QueueLevel(newQueueLevel),\n\t\ttag.PreviousQueueLevel(queueImpl.state.level),\n\t\ttag.WorkflowDomainIDs(domainToSplit),\n\t\ttag.QueueSplitPolicyType(policyTypePendingTask),\n\t)\n\tp.metricsScope.IncCounter(metrics.ProcessingQueuePendingTaskSplitCounter)\n\n\treturn splitQueueHelper(\n\t\tqueueImpl,\n\t\tdomainToSplit,\n\t\tnewQueueLevel,\n\t\tp.lookAheadFunc,\n\t)\n}\n\nfunc (p *stuckTaskSplitPolicy) Evaluate(queue ProcessingQueue) []ProcessingQueueState {\n\tqueueImpl := queue.(*processingQueueImpl)\n\n\tif queueImpl.state.level == p.maxNewQueueLevel {\n\t\t// already reaches max level, skip splitting\n\t\treturn nil\n\t}\n\n\tthreshold, ok := p.attemptThreshold[queueImpl.state.level]\n\tif !ok {\n\t\t// no threshold specified for the level, skip splitting\n\t\treturn nil\n\t}\n\n\tdomainToSplit := make(map[string]struct{})\n\tfor _, task := range queueImpl.outstandingTasks {\n\t\tdomainID := task.GetDomainID()\n\t\tattempt := task.GetAttempt()\n\t\tif attempt > threshold && p.enabledByDomainID(domainID) {\n\t\t\tdomainToSplit[domainID] = struct{}{}\n\t\t}\n\t}\n\n\tif len(domainToSplit) == 0 {\n\t\treturn nil\n\t}\n\n\tnewQueueLevel := queueImpl.state.level + 1 // split stuck tasks to current level + 1\n\tp.logger.Info(\"Split processing queue\",\n\t\ttag.QueueLevel(newQueueLevel),\n\t\ttag.PreviousQueueLevel(queueImpl.state.level),\n\t\ttag.WorkflowDomainIDs(domainToSplit),\n\t\ttag.QueueSplitPolicyType(policyTypeStuckTask),\n\t)\n\tp.metricsScope.IncCounter(metrics.ProcessingQueueStuckTaskSplitCounter)\n\n\treturn splitQueueHelper(\n\t\tqueueImpl,\n\t\tdomainToSplit,\n\t\tnewQueueLevel,\n\t\tnil, // no need to look ahead\n\t)\n}\n\nfunc (p *selectedDomainSplitPolicy) Evaluate(queue ProcessingQueue) []ProcessingQueueState {\n\tdomainBelongsToQueue := false\n\tcurrentQueueState := queue.State()\n\tcurrentDomainFilter := currentQueueState.DomainFilter()\n\tfor domainID := range p.domainIDs {\n\t\tif currentDomainFilter.Filter(domainID) {\n\t\t\tdomainBelongsToQueue = true\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif !domainBelongsToQueue {\n\t\t// no split needed\n\t\treturn nil\n\t}\n\n\tp.logger.Info(\"Split processing queue\",\n\t\ttag.QueueLevel(currentQueueState.Level()),\n\t\ttag.PreviousQueueLevel(p.newQueueLevel),\n\t\ttag.WorkflowDomainIDs(p.domainIDs),\n\t\ttag.QueueSplitPolicyType(policyTypeSelectedDomain),\n\t)\n\tp.metricsScope.IncCounter(metrics.ProcessingQueueSelectedDomainSplitCounter)\n\n\treturn []ProcessingQueueState{\n\t\tnewProcessingQueueState(\n\t\t\tcurrentQueueState.Level(),\n\t\t\tcurrentQueueState.AckLevel(),\n\t\t\tcurrentQueueState.ReadLevel(),\n\t\t\tcurrentQueueState.MaxLevel(),\n\t\t\tcurrentDomainFilter.Exclude(p.domainIDs),\n\t\t),\n\t\tnewProcessingQueueState(\n\t\t\tp.newQueueLevel,\n\t\t\tcurrentQueueState.AckLevel(),\n\t\t\tcurrentQueueState.ReadLevel(),\n\t\t\tcurrentQueueState.MaxLevel(),\n\t\t\t// make a copy here so that it won't be accidentally modified when p.domainID is changed\n\t\t\tNewDomainFilter(p.domainIDs, false).copy(),\n\t\t),\n\t}\n}\n\nfunc (p *randomSplitPolicy) Evaluate(queue ProcessingQueue) []ProcessingQueueState {\n\tqueueImpl := queue.(*processingQueueImpl)\n\n\tif queueImpl.state.level == p.maxNewQueueLevel {\n\t\t// already reaches max level, skip splitting\n\t\treturn nil\n\t}\n\n\tdomainIDs := make(map[string]struct{})\n\tfor _, task := range queueImpl.outstandingTasks {\n\t\tdomainIDs[task.GetDomainID()] = struct{}{}\n\t}\n\n\tif len(domainIDs) == 0 {\n\t\treturn nil\n\t}\n\n\tdomainToSplit := make(map[string]struct{})\n\tfor domainID := range domainIDs {\n\t\tif !p.enabledByDomainID(domainID) {\n\t\t\tcontinue\n\t\t}\n\t\tif !shouldSplit(p.splitProbability) {\n\t\t\tcontinue\n\t\t}\n\n\t\tdomainToSplit[domainID] = struct{}{}\n\t}\n\n\tif len(domainToSplit) == 0 {\n\t\treturn nil\n\t}\n\n\tnewQueueLevel := queueImpl.state.level + 1 // split stuck tasks to current level + 1\n\tp.logger.Info(\"Split processing queue\",\n\t\ttag.QueueLevel(newQueueLevel),\n\t\ttag.PreviousQueueLevel(queueImpl.state.level),\n\t\ttag.WorkflowDomainIDs(domainToSplit),\n\t\ttag.QueueSplitPolicyType(policyTypeRandom),\n\t)\n\tp.metricsScope.IncCounter(metrics.ProcessingQueueRandomSplitCounter)\n\n\treturn splitQueueHelper(\n\t\tqueueImpl,\n\t\tdomainToSplit,\n\t\tnewQueueLevel,\n\t\tp.lookAheadFunc,\n\t)\n}\n\nfunc (p *aggregatedSplitPolicy) Evaluate(queue ProcessingQueue) []ProcessingQueueState {\n\tfor _, policy := range p.policies {\n\t\tnewStates := policy.Evaluate(queue)\n\t\tif len(newStates) != 0 {\n\t\t\treturn newStates\n\t\t}\n\t}\n\treturn nil\n}\n\n// splitQueueHelper assumes domainToSplit is not empty\nfunc splitQueueHelper(\n\tqueueImpl *processingQueueImpl,\n\tdomainToSplit map[string]struct{},\n\tnewQueueLevel int,\n\tlookAheadFunc lookAheadFunc,\n) []ProcessingQueueState {\n\tnewMaxLevel := queueImpl.state.readLevel\n\tif lookAheadFunc != nil {\n\t\tfor domainID := range domainToSplit {\n\t\t\tnewMaxLevel = maxTaskKey(newMaxLevel, lookAheadFunc(queueImpl.state.readLevel, domainID))\n\t\t}\n\t}\n\tif queueImpl.state.maxLevel.Less(newMaxLevel) {\n\t\tnewMaxLevel = queueImpl.state.maxLevel\n\t}\n\n\tnewQueueStates := []ProcessingQueueState{\n\t\tnewProcessingQueueState(\n\t\t\tnewQueueLevel,\n\t\t\tqueueImpl.state.ackLevel,\n\t\t\tqueueImpl.state.readLevel,\n\t\t\tnewMaxLevel,\n\t\t\tNewDomainFilter(domainToSplit, false),\n\t\t),\n\t}\n\n\texcludedDomainFilter := queueImpl.state.domainFilter.Exclude(domainToSplit)\n\tif excludedDomainFilter.ReverseMatch || len(excludedDomainFilter.DomainIDs) != 0 {\n\t\t// this means the new domain filter still matches at least one domain\n\t\tnewQueueStates = append(newQueueStates, newProcessingQueueState(\n\t\t\tqueueImpl.state.level,\n\t\t\tqueueImpl.state.ackLevel,\n\t\t\tqueueImpl.state.readLevel,\n\t\t\tnewMaxLevel,\n\t\t\texcludedDomainFilter,\n\t\t))\n\t}\n\n\tif !taskKeyEquals(newMaxLevel, queueImpl.state.maxLevel) {\n\t\tnewQueueStates = append(newQueueStates, newProcessingQueueState(\n\t\t\tqueueImpl.state.level,\n\t\t\tnewMaxLevel,\n\t\t\tnewMaxLevel,\n\t\t\tqueueImpl.state.maxLevel,\n\t\t\tqueueImpl.state.domainFilter.copy(),\n\t\t))\n\t}\n\n\tfor _, state := range newQueueStates {\n\t\tif state.ReadLevel().Less(state.AckLevel()) || state.MaxLevel().Less(state.ReadLevel()) {\n\t\t\tpanic(fmt.Sprintf(\"invalid processing queue split result: %v, state before split: %v, newMaxLevel: %v\", state, queueImpl.state, newMaxLevel))\n\t\t}\n\t}\n\n\treturn newQueueStates\n}\n\nfunc shouldSplit(probability float64) bool {\n\tif probability <= 0 {\n\t\treturn false\n\t}\n\tif probability >= 1.0 {\n\t\treturn true\n\t}\n\treturn rand.Intn(int(1.0/probability)) == 0\n}\n"
  },
  {
    "path": "service/history/queue/split_policy_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"sort\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tt \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tsplitPolicySuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller *gomock.Controller\n\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\t}\n)\n\nfunc TestSplitPolicySuite(t *testing.T) {\n\ts := new(splitPolicySuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *splitPolicySuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.metricsScope = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}).Scope(metrics.TimerQueueProcessorScope)\n}\n\nfunc (s *splitPolicySuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *splitPolicySuite) TestPendingTaskSplitPolicy() {\n\tmaxNewQueueLevel := 3\n\tpendingTaskThreshold := map[int]int{\n\t\t0: 10,\n\t\t1: 100,\n\t\t2: 1000,\n\t\t3: 10000,\n\t}\n\tlookAheadTasks := 5\n\tlookAheadFunc := func(key task.Key, _ string) task.Key {\n\t\tcurrentID := key.(testKey).ID\n\t\treturn testKey{ID: currentID + lookAheadTasks}\n\t}\n\tpendingTaskSplitPolicy := NewPendingTaskSplitPolicy(\n\t\tpendingTaskThreshold,\n\t\tdynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tlookAheadFunc,\n\t\tmaxNewQueueLevel,\n\t\ts.logger,\n\t\ts.metricsScope,\n\t)\n\n\ttestCases := []struct {\n\t\tcurrentState             ProcessingQueueState\n\t\tnumPendingTasksPerDomain map[string]int // domainID -> number of pending tasks\n\t\texpectedNewStates        []ProcessingQueueState\n\t}{\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t101, // a level which has no threshold specified\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 1000},\n\t\t\t\ttestKey{ID: 10000},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 1000,\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t3, // maxNewQueueLevel\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 100000},\n\t\t\t\ttestKey{ID: 1000000},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain2\": 100000,\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t1,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 198},\n\t\t\t\ttestKey{ID: 200},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 99,\n\t\t\t\t\"testDomain2\": 99,\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t2,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 2002},\n\t\t\t\ttestKey{ID: 2002 + lookAheadTasks + 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 1001,\n\t\t\t\t\"testDomain2\": 1001,\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t3,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 2002},\n\t\t\t\t\ttestKey{ID: 2002 + lookAheadTasks},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t2,\n\t\t\t\t\ttestKey{ID: 2002 + lookAheadTasks},\n\t\t\t\t\ttestKey{ID: 2002 + lookAheadTasks},\n\t\t\t\t\ttestKey{ID: 2002 + lookAheadTasks + 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t2,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 1001},\n\t\t\t\ttestKey{ID: 1001 + lookAheadTasks},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 1001,\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t3,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 1001},\n\t\t\t\t\ttestKey{ID: 1001 + lookAheadTasks},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 109},\n\t\t\t\ttestKey{ID: 109 + lookAheadTasks + 100},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\ttrue,\n\t\t\t\t),\n\t\t\t),\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain2\": 9,\n\t\t\t\t\"testDomain3\": 100,\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 109},\n\t\t\t\t\ttestKey{ID: 109 + lookAheadTasks},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 109 + lookAheadTasks},\n\t\t\t\t\ttestKey{ID: 109 + lookAheadTasks},\n\t\t\t\t\ttestKey{ID: 109 + lookAheadTasks + 100},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 109},\n\t\t\t\t\ttestKey{ID: 109 + lookAheadTasks},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain3\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\toutstandingTasks := make(map[task.Key]task.Task)\n\t\tfor domainID, numPendingTasks := range tc.numPendingTasksPerDomain {\n\t\t\tfor i := 0; i != numPendingTasks; i++ {\n\t\t\t\tmockTask := task.NewMockTask(s.controller)\n\t\t\t\tmockTask.EXPECT().GetDomainID().Return(domainID).MaxTimes(1)\n\t\t\t\tmockTask.EXPECT().State().Return(t.TaskStatePending).MaxTimes(1)\n\t\t\t\toutstandingTasks[task.NewMockKey(s.controller)] = mockTask\n\t\t\t}\n\t\t}\n\n\t\tqueue := newProcessingQueue(\n\t\t\ttc.currentState,\n\t\t\toutstandingTasks,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\n\t\ts.assertQueueStatesEqual(tc.expectedNewStates, pendingTaskSplitPolicy.Evaluate(queue))\n\t}\n}\n\nfunc (s *splitPolicySuite) TestStuckTaskSplitPolicy() {\n\tmaxNewQueueLevel := 3\n\tattemptThreshold := map[int]int{\n\t\t0: 10,\n\t\t1: 100,\n\t\t2: 1000,\n\t\t3: 10000,\n\t}\n\n\tstuckTaskSplitPolicy := NewStuckTaskSplitPolicy(\n\t\tattemptThreshold,\n\t\tdynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tmaxNewQueueLevel,\n\t\ts.logger,\n\t\ts.metricsScope,\n\t)\n\n\ttestCases := []struct {\n\t\tcurrentState        ProcessingQueueState\n\t\tpendingTaskAttempts map[string][]int // domainID -> list of task attempts\n\t\texpectedNewStates   []ProcessingQueueState\n\t}{\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t101, // a level which has no threshold specified\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 3},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain1\": {1, 1000, 10000},\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t3, // maxNewQueueLevel\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 1},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain2\": {100000},\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t1,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 4},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain1\": {0, 99},\n\t\t\t\t\"testDomain2\": {0, 99},\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t2,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 4},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain1\": {0, 99, 1001},\n\t\t\t\t\"testDomain2\": {1001},\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t3,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 4},\n\t\t\t\t\ttestKey{ID: 4},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t2,\n\t\t\t\t\ttestKey{ID: 4},\n\t\t\t\t\ttestKey{ID: 4},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t2,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain1\": {0, 1, 99, 1001, 0},\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t3,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\ttrue,\n\t\t\t\t),\n\t\t\t),\n\t\t\tpendingTaskAttempts: map[string][]int{\n\t\t\t\t\"testDomain2\": {0, 1, 9},\n\t\t\t\t\"testDomain3\": {1, 100},\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain3\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\toutstandingTasks := make(map[task.Key]task.Task)\n\t\tfor domainID, taskAttempts := range tc.pendingTaskAttempts {\n\t\t\tfor _, attempt := range taskAttempts {\n\t\t\t\tmockTask := task.NewMockTask(s.controller)\n\t\t\t\tmockTask.EXPECT().GetDomainID().Return(domainID).MaxTimes(1)\n\t\t\t\tmockTask.EXPECT().GetAttempt().Return(attempt).MaxTimes(1)\n\t\t\t\toutstandingTasks[task.NewMockKey(s.controller)] = mockTask\n\t\t\t}\n\t\t}\n\n\t\tqueue := newProcessingQueue(\n\t\t\ttc.currentState,\n\t\t\toutstandingTasks,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\n\t\ts.assertQueueStatesEqual(tc.expectedNewStates, stuckTaskSplitPolicy.Evaluate(queue))\n\t}\n}\n\nfunc (s *splitPolicySuite) TestSelectedDomainSplitPolicy() {\n\tnewQueueLevel := 123\n\n\ttestCases := []struct {\n\t\tcurrentState      ProcessingQueueState\n\t\tdomainToSplit     map[string]struct{}\n\t\texpectedNewStates []ProcessingQueueState\n\t}{\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tdomainToSplit:     map[string]struct{}{\"testDomain3\": {}, \"testDomain4\": {}},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\ttrue,\n\t\t\t\t),\n\t\t\t),\n\t\t\tdomainToSplit: map[string]struct{}{\"testDomain3\": {}},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t0,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}, \"testDomain3\": {}},\n\t\t\t\t\t\ttrue,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\tnewQueueLevel,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain3\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tqueue := NewProcessingQueue(tc.currentState, nil, nil)\n\t\tsplitPolicy := NewSelectedDomainSplitPolicy(tc.domainToSplit, newQueueLevel, s.logger, s.metricsScope)\n\n\t\ts.assertQueueStatesEqual(tc.expectedNewStates, splitPolicy.Evaluate(queue))\n\t}\n}\n\nfunc (s *splitPolicySuite) TestRandomSplitPolicy() {\n\tmaxNewQueueLevel := 3\n\tlookAheadFunc := func(key task.Key, _ string) task.Key {\n\t\tcurrentID := key.(testKey).ID\n\t\treturn testKey{ID: currentID + 10}\n\t}\n\n\ttestCases := []struct {\n\t\tcurrentState             ProcessingQueueState\n\t\tsplitProbability         float64\n\t\tnumPendingTasksPerDomain map[string]int // domainID -> number of pending tasks\n\t\texpectedNewStates        []ProcessingQueueState\n\t}{\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tsplitProbability:         0,\n\t\t\tnumPendingTasksPerDomain: nil,\n\t\t\texpectedNewStates:        nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t3,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}, \"testDomain2\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tsplitProbability: 1,\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 2,\n\t\t\t\t\"testDomain2\": 3,\n\t\t\t},\n\t\t\texpectedNewStates: nil,\n\t\t},\n\t\t{\n\t\t\tcurrentState: newProcessingQueueState(\n\t\t\t\t0,\n\t\t\t\ttestKey{ID: 0},\n\t\t\t\ttestKey{ID: 5},\n\t\t\t\ttestKey{ID: 10},\n\t\t\t\tNewDomainFilter(\n\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\tfalse,\n\t\t\t\t),\n\t\t\t),\n\t\t\tsplitProbability: 1,\n\t\t\tnumPendingTasksPerDomain: map[string]int{\n\t\t\t\t\"testDomain1\": 5,\n\t\t\t},\n\t\t\texpectedNewStates: []ProcessingQueueState{\n\t\t\t\tnewProcessingQueueState(\n\t\t\t\t\t1,\n\t\t\t\t\ttestKey{ID: 0},\n\t\t\t\t\ttestKey{ID: 5},\n\t\t\t\t\ttestKey{ID: 10},\n\t\t\t\t\tNewDomainFilter(\n\t\t\t\t\t\tmap[string]struct{}{\"testDomain1\": {}},\n\t\t\t\t\t\tfalse,\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\toutstandingTasks := make(map[task.Key]task.Task)\n\t\tfor domainID, numPendingTasks := range tc.numPendingTasksPerDomain {\n\t\t\tfor i := 0; i != numPendingTasks; i++ {\n\t\t\t\tmockTask := task.NewMockTask(s.controller)\n\t\t\t\tmockTask.EXPECT().GetDomainID().Return(domainID).MaxTimes(1)\n\t\t\t\toutstandingTasks[task.NewMockKey(s.controller)] = mockTask\n\t\t\t}\n\t\t}\n\n\t\tqueue := newProcessingQueue(\n\t\t\ttc.currentState,\n\t\t\toutstandingTasks,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\t\tsplitPolicy := NewRandomSplitPolicy(\n\t\t\ttc.splitProbability,\n\t\t\tdynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\tmaxNewQueueLevel,\n\t\t\tlookAheadFunc,\n\t\t\ts.logger,\n\t\t\ts.metricsScope,\n\t\t)\n\n\t\ts.assertQueueStatesEqual(tc.expectedNewStates, splitPolicy.Evaluate(queue))\n\t}\n}\n\nfunc (s *splitPolicySuite) TestAggregatedSplitPolicy() {\n\texpectedNewStates := []ProcessingQueueState{\n\t\tNewMockProcessingQueueState(s.controller),\n\t\tNewMockProcessingQueueState(s.controller),\n\t\tNewMockProcessingQueueState(s.controller),\n\t}\n\n\tmockProcessingQueue := NewMockProcessingQueue(s.controller)\n\n\ttotalPolicyNum := 4\n\tpolicyEvaluationResults := [][]ProcessingQueueState{\n\t\tnil,\n\t\t{},\n\t\texpectedNewStates,\n\t}\n\n\tmockSplitPolicies := []ProcessingQueueSplitPolicy{}\n\tfor i := 0; i != totalPolicyNum; i++ {\n\t\tmockSplitPolicy := NewMockProcessingQueueSplitPolicy(s.controller)\n\t\tif i < len(policyEvaluationResults) {\n\t\t\tmockSplitPolicy.EXPECT().Evaluate(mockProcessingQueue).Return(policyEvaluationResults[i]).Times(1)\n\t\t}\n\t\tmockSplitPolicies = append(mockSplitPolicies, mockSplitPolicy)\n\t}\n\n\taggregatedSplitPolicy := NewAggregatedSplitPolicy(mockSplitPolicies...)\n\ts.Equal(expectedNewStates, aggregatedSplitPolicy.Evaluate(mockProcessingQueue))\n}\n\nfunc (s *splitPolicySuite) assertQueueStatesEqual(\n\texpected []ProcessingQueueState,\n\tactual []ProcessingQueueState,\n) {\n\ts.Equal(len(expected), len(actual))\n\n\tif len(expected) == 0 {\n\t\treturn\n\t}\n\n\tcompFn := func(s1, s2 ProcessingQueueState) bool {\n\t\tif taskKeyEquals(s1.AckLevel(), s2.AckLevel()) {\n\t\t\treturn s1.Level() < s2.Level()\n\t\t}\n\n\t\treturn s1.AckLevel().Less(s2.AckLevel())\n\t}\n\n\tsort.Slice(expected, func(i, j int) bool {\n\t\treturn compFn(expected[i], expected[j])\n\t})\n\tsort.Slice(actual, func(i, j int) bool {\n\t\treturn compFn(actual[i], actual[j])\n\t})\n\n\tfor i := 0; i != len(expected); i++ {\n\t\ts.Equal(expected[i], actual[i])\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/task_allocator.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"runtime/debug\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\thtask \"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\t// TaskAllocator verifies if a task should be processed or not\n\tTaskAllocator interface {\n\t\tVerifyActiveTask(domainID, wfID, rID string, task interface{}) (bool, error)\n\t\tVerifyFailoverActiveTask(targetDomainIDs map[string]struct{}, domainID, wfID, rID string, task interface{}) (bool, error)\n\t\tVerifyStandbyTask(standbyCluster string, domainID, wfID, rID string, task interface{}) (bool, error)\n\t\tLock()\n\t\tUnlock()\n\t}\n\n\ttaskAllocatorImpl struct {\n\t\tcurrentClusterName string\n\t\tshard              shard.Context\n\t\tdomainCache        cache.DomainCache\n\t\tactiveClusterMgr   activecluster.Manager\n\t\tlogger             log.Logger\n\n\t\tlocker sync.RWMutex\n\t}\n)\n\n// NewTaskAllocator create a new task allocator\nfunc NewTaskAllocator(shard shard.Context) TaskAllocator {\n\treturn &taskAllocatorImpl{\n\t\tcurrentClusterName: shard.GetService().GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard:              shard,\n\t\tdomainCache:        shard.GetDomainCache(),\n\t\tactiveClusterMgr:   shard.GetActiveClusterManager(),\n\t\tlogger:             shard.GetLogger(),\n\t}\n}\n\n// VerifyActiveTask, will return true if task activeness check is successful\nfunc (t *taskAllocatorImpl) VerifyActiveTask(domainID, wfID, rID string, task interface{}) (bool, error) {\n\treturn t.verifyTaskActiveness(t.currentClusterName, domainID, wfID, rID, task, false)\n}\n\n// VerifyFailoverActiveTask, will return true if task activeness check is successful\nfunc (t *taskAllocatorImpl) VerifyFailoverActiveTask(targetDomainIDs map[string]struct{}, domainID, wfID, rID string, task interface{}) (bool, error) {\n\t_, ok := targetDomainIDs[domainID]\n\tif !ok {\n\t\treturn false, nil\n\t}\n\n\treturn t.verifyTaskActiveness(\"\", domainID, wfID, rID, task, true)\n}\n\n// VerifyStandbyTask, will return true if task standbyness check is successful\nfunc (t *taskAllocatorImpl) VerifyStandbyTask(standbyCluster string, domainID, wfID, rID string, task interface{}) (bool, error) {\n\treturn t.verifyTaskActiveness(standbyCluster, domainID, wfID, rID, task, true)\n}\n\n// verifyTaskActiveness verifies if a task should be processed or not based on domain's state\n//   - If failed to fetch the domain, it returns (false, err) indicating the task should be retried\n//   - If domain is not found, it returns (false, nil) indicating the task should be skipped\n//   - If domain is local, return (!skipLocalDomain, nil) indicating the task should be skipped if skipLocalDomain is true\n//   - If domain is pending active, it returns (false, ErrTaskPendingActive) indicating the task should be retried\n//   - If domain is active in the given cluster, it returns (true, nil) indicating the task should be processed\n//     Special case: if it's a failover queue (cluster == \"\"), it returns (true, nil) indicating the task should be processed in any cluster\nfunc (t *taskAllocatorImpl) verifyTaskActiveness(cluster string, domainID, wfID, rID string, task interface{}, skipLocalDomain bool) (b bool, e error) {\n\tif t.logger.DebugOn() {\n\t\tdefer func() {\n\t\t\ttaskString := \"nil\"\n\t\t\tif task != nil {\n\t\t\t\tdata, err := json.Marshal(task)\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.logger.Error(\"Failed to marshal task.\", tag.Error(err))\n\t\t\t\t\ttaskString = \"nil\"\n\t\t\t\t} else {\n\t\t\t\t\ttaskString = string(data)\n\t\t\t\t}\n\t\t\t}\n\t\t\tt.logger.Debugf(\"verifyTaskActiveness returning (%v, %v) for cluster %s, domainID %s, wfID %s, rID %s, task %s, stacktrace %s\",\n\t\t\t\tb,\n\t\t\t\te,\n\t\t\t\tcluster,\n\t\t\t\tdomainID,\n\t\t\t\twfID,\n\t\t\t\trID,\n\t\t\t\ttaskString,\n\t\t\t\tstring(debug.Stack()),\n\t\t\t)\n\t\t}()\n\t}\n\tt.locker.RLock()\n\tdefer t.locker.RUnlock()\n\n\tdomainEntry, err := t.domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\t// it is possible that the domain is deleted\n\t\t// we should treat that domain as not active\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\tt.logger.Warn(\"Failed to get domain from cache\", tag.WorkflowDomainID(domainID), tag.Error(err))\n\t\t\treturn false, err\n\t\t}\n\t\tt.logger.Warn(\"Cannot find domain, default to not process task.\", tag.WorkflowDomainID(domainID), tag.Value(task))\n\t\treturn false, nil\n\t}\n\n\t// handle local domain\n\tif !domainEntry.IsGlobalDomain() {\n\t\t// only active in domain's cluster but should be skipped if skipLocalDomain is true\n\t\treturn !skipLocalDomain && t.currentClusterName == cluster, nil\n\t}\n\n\t// return error for pending active domain so the task can be retried\n\tif err := t.checkDomainPendingActive(\n\t\tdomainEntry,\n\t\tdomainID,\n\t\ttask,\n\t); err != nil {\n\t\treturn false, err\n\t}\n\n\tif cluster == \"\" { // failover queue task. Revisit this logic. It's copied from previous implementation\n\t\treturn true, nil\n\t}\n\n\t// handle active-active domain\n\tif domainEntry.GetReplicationConfig().IsActiveActive() {\n\t\tresp, err := t.activeClusterMgr.GetActiveClusterInfoByWorkflow(context.Background(), domainID, wfID, rID)\n\t\tif err != nil {\n\t\t\tt.logger.Warn(\"Failed to lookup active cluster\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(wfID),\n\t\t\t\ttag.WorkflowRunID(rID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\treturn false, err\n\t\t}\n\t\tif resp.ActiveClusterName != cluster {\n\t\t\tt.logger.Debug(\"Skip task because workflow is not active on the given cluster\",\n\t\t\t\ttag.WorkflowID(wfID),\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.ClusterName(cluster),\n\t\t\t)\n\t\t\treturn false, nil\n\t\t}\n\n\t\tt.logger.Debug(\"Active cluster for given task\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.ClusterName(resp.ActiveClusterName),\n\t\t)\n\t\treturn true, nil\n\t}\n\n\t// handle active-passive domain\n\tif domainEntry.GetReplicationConfig().ActiveClusterName != cluster {\n\t\tt.logger.Debug(\"Domain is not active in the given cluster, skip task.\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.WorkflowID(wfID),\n\t\t\ttag.WorkflowRunID(rID),\n\t\t\ttag.ClusterName(cluster),\n\t\t)\n\t\treturn false, nil\n\t}\n\n\treturn true, nil\n}\n\nfunc (t *taskAllocatorImpl) checkDomainPendingActive(domainEntry *cache.DomainCacheEntry, taskDomainID string, task interface{}) error {\n\tif domainEntry.IsGlobalDomain() && domainEntry.GetFailoverEndTime() != nil {\n\t\t// the domain is pending active, pause on processing this task\n\t\tt.logger.Debug(\"Domain is not in pending active, skip task.\", tag.WorkflowDomainID(taskDomainID), tag.Value(task))\n\t\treturn htask.ErrTaskPendingActive\n\t}\n\treturn nil\n}\n\n// Lock block all task allocation\nfunc (t *taskAllocatorImpl) Lock() {\n\tt.locker.Lock()\n}\n\n// Unlock resume the task allocator\nfunc (t *taskAllocatorImpl) Unlock() {\n\tt.locker.Unlock()\n}\n\n// isDomainNotRegistered checks either if domain does not exist or is in deprecated or deleted status\nfunc isDomainNotRegistered(shard shard.Context, domainID string) (bool, error) {\n\tdomainEntry, err := shard.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\t// error in finding a domain\n\t\treturn false, err\n\t}\n\tinfo := domainEntry.GetInfo()\n\tif info == nil {\n\t\treturn false, errors.New(\"domain info is nil in cache\")\n\t}\n\treturn info.Status == persistence.DomainStatusDeprecated || info.Status == persistence.DomainStatusDeleted, nil\n}\n"
  },
  {
    "path": "service/history/queue/task_allocator_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\thtask \"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestVerifyActiveTask(t *testing.T) {\n\tdomainID := \"testDomainID\"\n\ttask := \"testTask\"\n\ttests := []struct {\n\t\tname                string\n\t\tsetupMocks          func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager)\n\t\texpectedResult      bool\n\t\texpectedErrorString string\n\t}{\n\t\t{\n\t\t\tname: \"Failed to get domain from cache, non-EntityNotExistsError\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedResult:      false,\n\t\t\texpectedErrorString: \"some error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Domain not found, EntityNotExistsError\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global and not active in current cluster\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: domainID + \"name\"},\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"otherCluster\",\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global and pending active in current cluster\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tendtime := int64(1)\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: domainID + \"name\"},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"currentCluster\",\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t\t&endtime, 1, 1, 1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult:      false,\n\t\t\texpectedErrorString: \"the domain is pending-active\",\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global and active in current cluster\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Name: domainID + \"name\"},\n\t\t\t\t\tnil,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"currentCluster\",\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, task is not active in current cluster so should be skipped\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: \"another-cluster\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, task is active in current cluster so should be processed\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: \"currentCluster\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, activeness lookup returns error\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedResult:      false,\n\t\t\texpectedErrorString: \"some error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mocks\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\t// Setup mock shard to return mock domain cache and logger\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tmockShard.EXPECT().GetService().Return(nil).AnyTimes() // Adjust based on your implementation\n\n\t\t\tactiveClusterMgr := activecluster.NewMockManager(ctrl)\n\n\t\t\t// Create the task allocator\n\t\t\tallocator := &taskAllocatorImpl{\n\t\t\t\tcurrentClusterName: \"currentCluster\",\n\t\t\t\tshard:              mockShard,\n\t\t\t\tdomainCache:        mockDomainCache,\n\t\t\t\tlogger:             log.NewNoop(),\n\t\t\t\tactiveClusterMgr:   activeClusterMgr,\n\t\t\t}\n\n\t\t\ttt.setupMocks(mockDomainCache, activeClusterMgr)\n\t\t\tresult, err := allocator.VerifyActiveTask(domainID, \"wfid\", \"rid\", task)\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t\tif tt.expectedErrorString != \"\" {\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedErrorString)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVerifyFailoverActiveTask(t *testing.T) {\n\tdomainID := \"testDomainID\"\n\ttask := \"testTask\"\n\n\ttests := []struct {\n\t\tname                string\n\t\ttargetDomainIDs     map[string]struct{}\n\t\tsetupMocks          func(mockDomainCache *cache.MockDomainCache)\n\t\texpectedResult      bool\n\t\texpectedError       error\n\t\texpectedErrorString string\n\t}{\n\t\t{\n\t\t\tname: \"Domain not in targetDomainIDs\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\t\"someOtherDomainID\": {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// No mocks needed since the domain is not in targetDomainIDs\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain in targetDomainIDs, GetDomainByID returns non-EntityNotExistsError\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\tdomainID: {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedResult:      false,\n\t\t\texpectedError:       errors.New(\"some error\"),\n\t\t\texpectedErrorString: \"some error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Domain in targetDomainIDs, GetDomainByID returns EntityNotExistsError\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\tdomainID: {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain in targetDomainIDs, domain is pending active\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\tdomainID: {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// Set up a domainEntry that is global and has a non-nil FailoverEndTime\n\t\t\t\tendtime := int64(1)\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue, // IsGlobalDomain\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t1,\n\t\t\t\t\t&endtime, // FailoverEndTime is non-nil\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  htask.ErrTaskPendingActive,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain in targetDomainIDs, checkDomainPendingActive returns nil\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\tdomainID: {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// Set up a domainEntry that is global and has a nil FailoverEndTime\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue, // IsGlobalDomain\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t1,\n\t\t\t\t\tnil, // FailoverEndTime is nil\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is local, should be skipped\",\n\t\t\ttargetDomainIDs: map[string]struct{}{\n\t\t\t\tdomainID: {},\n\t\t\t},\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\t// Set up a domainEntry that is global and has a nil FailoverEndTime\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t\tfalse, // IsGlobalDomain\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mocks\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\t// Setup mock shard to return mock domain cache and logger\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tmockShard.EXPECT().GetService().Return(nil).AnyTimes() // Adjust based on your implementation\n\n\t\t\t// Create the task allocator\n\t\t\tallocator := &taskAllocatorImpl{\n\t\t\t\tcurrentClusterName: \"currentCluster\",\n\t\t\t\tshard:              mockShard,\n\t\t\t\tdomainCache:        mockDomainCache,\n\t\t\t\tlogger:             log.NewNoop(),\n\t\t\t}\n\n\t\t\ttt.setupMocks(mockDomainCache)\n\t\t\tresult, err := allocator.VerifyFailoverActiveTask(tt.targetDomainIDs, domainID, \"wfid\", \"rid\", task)\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t\t} else if tt.expectedErrorString != \"\" {\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedErrorString)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVerifyStandbyTask(t *testing.T) {\n\tdomainID := \"testDomainID\"\n\ttask := \"testTask\"\n\ttests := []struct {\n\t\tname                string\n\t\tsetupMocks          func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager)\n\t\tstandbyCluster      string\n\t\texpectedResult      bool\n\t\texpectedError       error\n\t\texpectedErrorString string\n\t}{\n\t\t{\n\t\t\tname: \"GetDomainByID returns non-EntityNotExistsError\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tstandbyCluster:      \"standbyCluster\",\n\t\t\texpectedResult:      false,\n\t\t\texpectedErrorString: \"some error\",\n\t\t},\n\t\t{\n\t\t\tname: \"GetDomainByID returns EntityNotExistsError\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is not global\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t\"\",\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global but not standby (active cluster name does not match standbyCluster)\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"anotherCluster\",\n\t\t\t\t\t},\n\t\t\t\t\t0,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global and standby, but checkDomainPendingActive returns error\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tendTime := time.Now().Add(time.Hour).UnixNano()\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(nil, nil, true, &persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: \"currentCluster\",\n\t\t\t\t}, 1, &endTime, 1, 1, 1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tstandbyCluster:      \"currentCluster\",\n\t\t\texpectedResult:      false,\n\t\t\texpectedErrorString: htask.ErrTaskPendingActive.Error(),\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is global and standby, checkDomainPendingActive returns nil\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: \"standbyCluster\",\n\t\t\t\t\t},\n\t\t\t\t\t0, // FailoverEndTime is zero\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: true,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is local, should be skipped\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\t// Set up a domainEntry that is global and has a nil FailoverEndTime\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\tnil,\n\t\t\t\t\tnil,\n\t\t\t\t\tfalse, // IsGlobalDomain\n\t\t\t\t\t&persistence.DomainReplicationConfig{},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t\t1,\n\t\t\t\t)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, task is not active in standby cluster so should be skipped\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: \"another-cluster\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, task is active in standby cluster so should be processed\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: \"standbyCluster\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: true,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Domain is active-active mode, activeness lookup returns error\",\n\t\t\tsetupMocks: func(mockDomainCache *cache.MockDomainCache, activeClusterMgr *activecluster.MockManager) {\n\t\t\t\tdomainEntry := activeActiveDomainEntry()\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(domainEntry, nil)\n\t\t\t\tactiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\tstandbyCluster: \"standbyCluster\",\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  errors.New(\"some error\"),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mocks\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\t// Setup mock shard to return mock domain cache and logger\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tmockShard.EXPECT().GetService().Return(nil).AnyTimes() // Adjust based on your implementation\n\n\t\t\tactiveClusterMgr := activecluster.NewMockManager(ctrl)\n\n\t\t\t// Create the task allocator\n\t\t\tallocator := &taskAllocatorImpl{\n\t\t\t\tcurrentClusterName: \"currentCluster\",\n\t\t\t\tshard:              mockShard,\n\t\t\t\tdomainCache:        mockDomainCache,\n\t\t\t\tlogger:             log.NewNoop(),\n\t\t\t\tactiveClusterMgr:   activeClusterMgr,\n\t\t\t}\n\n\t\t\ttt.setupMocks(mockDomainCache, activeClusterMgr)\n\t\t\tresult, err := allocator.VerifyStandbyTask(tt.standbyCluster, domainID, \"wfid\", \"rid\", task)\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t\t} else if tt.expectedErrorString != \"\" {\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedErrorString)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsDomainNotRegistered(t *testing.T) {\n\ttests := []struct {\n\t\tname                string\n\t\tdomainID            string\n\t\tmockFn              func(mockDomainCache *cache.MockDomainCache)\n\t\texpectedErrorString string\n\t}{\n\t\t{\n\t\t\tname: \"domainID return error\",\n\t\t\tmockFn: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"\").Return(nil, fmt.Errorf(\"testError\"))\n\t\t\t},\n\t\t\tdomainID:            \"\",\n\t\t\texpectedErrorString: \"testError\",\n\t\t},\n\t\t{\n\t\t\tname:     \"cannot get info\",\n\t\t\tdomainID: \"testDomainID\",\n\t\t\tmockFn: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(nil, nil, false, nil, 0, nil, 0, 0, 0)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"testDomainID\").Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedErrorString: \"domain info is nil in cache\",\n\t\t},\n\t\t{\n\t\t\tname:     \"domain is deprecated\",\n\t\t\tdomainID: \"testDomainID\",\n\t\t\tmockFn: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tdomainEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Status: persistence.DomainStatusDeprecated}, nil, false, nil, 0, nil, 0, 0, 0)\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(\"testDomainID\").Return(domainEntry, nil)\n\t\t\t},\n\t\t\texpectedErrorString: \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mocks\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\t// Setup mock shard to return mock domain cache and logger\n\t\t\tmockShard.EXPECT().GetDomainCache().Return(mockDomainCache).AnyTimes()\n\t\t\tmockShard.EXPECT().GetService().Return(nil).AnyTimes() // Adjust based on your implementation\n\n\t\t\ttt.mockFn(mockDomainCache)\n\t\t\tres, err := isDomainNotRegistered(mockShard, tt.domainID)\n\t\t\tif tt.expectedErrorString != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedErrorString)\n\t\t\t\tassert.False(t, res)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.True(t, res)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestLockUnlock(t *testing.T) {\n\t// basic validation to ensure lock/unlock doesn't panic and get blocked\n\tallocator := &taskAllocatorImpl{}\n\tallocator.Lock()\n\tdefer allocator.Unlock()\n}\n\nfunc activeActiveDomainEntry() *cache.DomainCacheEntry {\n\t// The specific versions are not important, as long as it's active-active.\n\t// Having a DomainReplicationConfig with non-zero length ActiveClusters is enough to make it active-active.\n\treturn cache.NewDomainCacheEntryForTest(\n\t\tnil,\n\t\tnil,\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\t// Populate ActiveClusters to make it active-active\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t1,\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t1,\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_active_processor.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc newTimerQueueActiveProcessor(\n\tclusterName string,\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n) *timerQueueProcessorBase {\n\tconfig := shard.GetConfig()\n\toptions := newTimerQueueProcessorOptions(config, true, false)\n\n\tlogger = logger.WithTags(tag.ClusterName(clusterName))\n\n\ttaskFilter := func(timer persistence.Task) (bool, error) {\n\t\tif timer.GetTaskCategory() != persistence.HistoryTaskCategoryTimer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shard, timer.GetDomainID()); notRegistered && err == nil {\n\t\t\t// Allow deletion tasks for deprecated domains\n\t\t\tif timer.GetTaskType() == persistence.TaskTypeDeleteHistoryEvent {\n\t\t\t\treturn true, nil\n\t\t\t}\n\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in active timer queue.\", tag.WorkflowDomainID(timer.GetDomainID()), tag.Value(timer))\n\t\t\treturn false, nil\n\t\t}\n\n\t\treturn taskAllocator.VerifyActiveTask(timer.GetDomainID(), timer.GetWorkflowID(), timer.GetRunID(), timer)\n\t}\n\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTimerTaskKey(shard.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, clusterName).GetScheduledTime(), 0)\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\treturn shard.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, clusterName, persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0))\n\t}\n\n\tupdateProcessingQueueStates := func(states []ProcessingQueueState) error {\n\t\tpStates := convertToPersistenceTimerProcessingQueueStates(states)\n\t\treturn shard.UpdateTimerProcessingQueueStates(clusterName, pStates)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn nil\n\t}\n\n\treturn newTimerQueueProcessorBase(\n\t\tclusterName,\n\t\tshard,\n\t\tloadTimerProcessingQueueStates(clusterName, shard, options, logger),\n\t\ttaskProcessor,\n\t\tclock.NewTimerGate(shard.GetTimeSource()),\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_failover_processor.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc newTimerQueueFailoverProcessor(\n\tstandbyClusterName string,\n\tshardContext shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tminLevel, maxLevel time.Time,\n\tdomainIDs map[string]struct{},\n) (updateClusterAckLevelFn, *timerQueueProcessorBase) {\n\tconfig := shardContext.GetConfig()\n\toptions := newTimerQueueProcessorOptions(config, true, true)\n\n\tcurrentClusterName := shardContext.GetService().GetClusterMetadata().GetCurrentClusterName()\n\tfailoverStartTime := shardContext.GetTimeSource().Now()\n\tfailoverUUID := uuid.New()\n\tlogger = logger.WithTags(\n\t\ttag.ClusterName(currentClusterName),\n\t\ttag.WorkflowDomainIDs(domainIDs),\n\t\ttag.FailoverMsg(\"from: \"+standbyClusterName),\n\t)\n\n\ttaskFilter := func(timer persistence.Task) (bool, error) {\n\t\tif timer.GetTaskCategory() != persistence.HistoryTaskCategoryTimer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shardContext, timer.GetDomainID()); notRegistered && err == nil {\n\t\t\t// Allow deletion tasks for deprecated domains\n\t\t\tif timer.GetTaskType() == persistence.TaskTypeDeleteHistoryEvent {\n\t\t\t\treturn true, nil\n\t\t\t}\n\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in failover timer queue.\", tag.WorkflowDomainID(timer.GetDomainID()), tag.Value(timer))\n\t\t\treturn false, nil\n\t\t}\n\t\treturn taskAllocator.VerifyFailoverActiveTask(domainIDs, timer.GetDomainID(), timer.GetWorkflowID(), timer.GetRunID(), timer)\n\t}\n\n\tmaxReadLevelTaskKey := newTimerTaskKey(maxLevel, 0)\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn maxReadLevelTaskKey // this is a const\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\treturn shardContext.UpdateFailoverLevel(\n\t\t\tpersistence.HistoryTaskCategoryTimer,\n\t\t\tfailoverUUID,\n\t\t\tpersistence.FailoverLevel{\n\t\t\t\tStartTime:    failoverStartTime,\n\t\t\t\tMinLevel:     persistence.NewHistoryTaskKey(minLevel, 0),\n\t\t\t\tCurrentLevel: persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\t\t\tMaxLevel:     persistence.NewHistoryTaskKey(maxLevel, 0),\n\t\t\t\tDomainIDs:    domainIDs,\n\t\t\t},\n\t\t)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn shardContext.DeleteFailoverLevel(\n\t\t\tpersistence.HistoryTaskCategoryTimer,\n\t\t\tfailoverUUID,\n\t\t)\n\t}\n\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tdefaultProcessingQueueLevel,\n\t\t\tnewTimerTaskKey(minLevel, 0),\n\t\t\tmaxReadLevelTaskKey,\n\t\t\tNewDomainFilter(domainIDs, false),\n\t\t),\n\t}\n\n\treturn updateClusterAckLevel, newTimerQueueProcessorBase(\n\t\tcurrentClusterName, // should use current cluster's time when doing domain failover\n\t\tshardContext,\n\t\tprocessingQueueStates,\n\t\ttaskProcessor,\n\t\tclock.NewTimerGate(shardContext.GetTimeSource()),\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tnil,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshardContext.GetMetricsClient(),\n\t)\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_processor.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype timerQueueProcessor struct {\n\tshard         shard.Context\n\thistoryEngine engine.Engine\n\ttaskProcessor task.Processor\n\n\tconfig             *config.Config\n\tcurrentClusterName string\n\n\tmetricsClient metrics.Client\n\tlogger        log.Logger\n\n\tstatus       int32\n\tshutdownChan chan struct{}\n\tshutdownWG   sync.WaitGroup\n\n\tackLevel                time.Time\n\ttaskAllocator           TaskAllocator\n\tactiveTaskExecutor      task.Executor\n\tactiveQueueProcessor    *timerQueueProcessorBase\n\tstandbyQueueProcessors  map[string]*timerQueueProcessorBase\n\tstandbyTaskExecutors    []task.Executor\n\tstandbyQueueTimerGates  map[string]clock.EventTimerGate\n\tfailoverQueueProcessors []*timerQueueProcessorBase\n}\n\n// NewTimerQueueProcessor creates a new timer QueueProcessor\nfunc NewTimerQueueProcessor(\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\texecutionCache execution.Cache,\n\tarchivalClient archiver.Client,\n\texecutionCheck invariant.Invariant,\n) Processor {\n\tlogger := shard.GetLogger().WithTags(tag.ComponentTimerQueue)\n\tcurrentClusterName := shard.GetClusterMetadata().GetCurrentClusterName()\n\tconfig := shard.GetConfig()\n\ttaskAllocator := NewTaskAllocator(shard)\n\n\tactiveLogger := logger.WithTags(tag.QueueTypeActive)\n\n\tactiveTaskExecutor := task.NewTimerActiveTaskExecutor(\n\t\tshard,\n\t\tarchivalClient,\n\t\texecutionCache,\n\t\tactiveLogger,\n\t\tshard.GetMetricsClient(),\n\t\tconfig,\n\t)\n\n\tactiveQueueProcessor := newTimerQueueActiveProcessor(\n\t\tcurrentClusterName,\n\t\tshard,\n\t\ttaskProcessor,\n\t\ttaskAllocator,\n\t\tactiveTaskExecutor,\n\t\tactiveLogger,\n\t)\n\n\tstandbyTaskExecutors := make([]task.Executor, 0, len(shard.GetClusterMetadata().GetRemoteClusterInfo()))\n\tstandbyQueueProcessors := make(map[string]*timerQueueProcessorBase)\n\tstandbyQueueTimerGates := make(map[string]clock.EventTimerGate)\n\tfor clusterName := range shard.GetClusterMetadata().GetRemoteClusterInfo() {\n\t\thistoryResender := ndc.NewHistoryResender(\n\t\t\tshard.GetDomainCache(),\n\t\t\tshard.GetService().GetClientBean(),\n\t\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\t\treturn shard.GetEngine().ReplicateEventsV2(ctx, request)\n\t\t\t},\n\t\t\tconfig.StandbyTaskReReplicationContextTimeout,\n\t\t\texecutionCheck,\n\t\t\tshard.GetLogger(),\n\t\t)\n\t\tstandByLogger := logger.WithTags(tag.QueueTypeStandby, tag.ActiveClusterName(clusterName))\n\t\tstandbyTaskExecutor := task.NewTimerStandbyTaskExecutor(\n\t\t\tshard,\n\t\t\tarchivalClient,\n\t\t\texecutionCache,\n\t\t\thistoryResender,\n\t\t\tstandByLogger,\n\t\t\tshard.GetMetricsClient(),\n\t\t\tclusterName,\n\t\t\tconfig,\n\t\t)\n\t\tstandbyTaskExecutors = append(standbyTaskExecutors, standbyTaskExecutor)\n\t\tstandbyQueueProcessors[clusterName], standbyQueueTimerGates[clusterName] = newTimerQueueStandbyProcessor(\n\t\t\tclusterName,\n\t\t\tshard,\n\t\t\ttaskProcessor,\n\t\t\ttaskAllocator,\n\t\t\tstandbyTaskExecutor,\n\t\t\tstandByLogger,\n\t\t)\n\t}\n\n\treturn &timerQueueProcessor{\n\t\tshard:         shard,\n\t\ttaskProcessor: taskProcessor,\n\n\t\tconfig:             config,\n\t\tcurrentClusterName: currentClusterName,\n\n\t\tmetricsClient: shard.GetMetricsClient(),\n\t\tlogger:        logger,\n\n\t\tstatus:       common.DaemonStatusInitialized,\n\t\tshutdownChan: make(chan struct{}),\n\n\t\tackLevel:               shard.GetQueueAckLevel(persistence.HistoryTaskCategoryTimer).GetScheduledTime(),\n\t\ttaskAllocator:          taskAllocator,\n\t\tactiveTaskExecutor:     activeTaskExecutor,\n\t\tactiveQueueProcessor:   activeQueueProcessor,\n\t\tstandbyQueueProcessors: standbyQueueProcessors,\n\t\tstandbyQueueTimerGates: standbyQueueTimerGates,\n\t\tstandbyTaskExecutors:   standbyTaskExecutors,\n\t}\n}\n\nfunc (t *timerQueueProcessor) Start() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Starting timer queue processor\")\n\tdefer t.logger.Info(\"Timer queue processor started\")\n\n\tt.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tt.shutdownWG.Add(1)\n\tgo t.completeTimerLoop()\n}\n\nfunc (t *timerQueueProcessor) Stop() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tif !t.shard.GetConfig().QueueProcessorEnableGracefulSyncShutdown() {\n\t\tt.logger.Info(\"Stopping timer queue processor non-gracefully\")\n\t\tdefer t.logger.Info(\"Timer queue processor stopped non-gracefully\")\n\t\tt.activeQueueProcessor.Stop()\n\t\t// stop active executor after queue processor\n\t\tt.activeTaskExecutor.Stop()\n\t\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\t// stop standby executors after queue processors\n\t\tfor _, standbyTaskExecutor := range t.standbyTaskExecutors {\n\t\t\tstandbyTaskExecutor.Stop()\n\t\t}\n\n\t\tclose(t.shutdownChan)\n\t\tif !common.AwaitWaitGroup(&t.shutdownWG, time.Minute) {\n\t\t\tt.logger.Warn(\"timerQueueProcessor timed out on shut down\", tag.LifeCycleStopTimedout)\n\t\t}\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Stopping timer queue processor gracefully\")\n\tdefer t.logger.Info(\"Timer queue processor stopped gracefully\")\n\n\t// close the shutdown channel first so processor pumps drains tasks\n\t// and then stop the processors\n\tclose(t.shutdownChan)\n\tif !common.AwaitWaitGroup(&t.shutdownWG, gracefulShutdownTimeout) {\n\t\tt.logger.Warn(\"transferQueueProcessor timed out on shut down\", tag.LifeCycleStopTimedout)\n\t}\n\tt.activeQueueProcessor.Stop()\n\n\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Stop()\n\t}\n\n\t// stop standby executors after queue processors\n\tfor _, standbyTaskExecutor := range t.standbyTaskExecutors {\n\t\tstandbyTaskExecutor.Stop()\n\t}\n\n\tif len(t.failoverQueueProcessors) > 0 {\n\t\tt.logger.Info(\"Shutting down failover timer queues\", tag.Counter(len(t.failoverQueueProcessors)))\n\t\tfor _, failoverQueueProcessor := range t.failoverQueueProcessors {\n\t\t\tfailoverQueueProcessor.Stop()\n\t\t}\n\t}\n\n\tt.activeTaskExecutor.Stop()\n}\n\nfunc (t *timerQueueProcessor) NotifyNewTask(clusterName string, info *hcommon.NotifyTaskInfo) {\n\tif clusterName == t.currentClusterName {\n\t\tt.activeQueueProcessor.notifyNewTimers(info.Tasks)\n\t\treturn\n\t}\n\n\tstandbyQueueProcessor, ok := t.standbyQueueProcessors[clusterName]\n\tif !ok {\n\t\tpanic(fmt.Sprintf(\"Cannot find standby timer processor for %s.\", clusterName))\n\t}\n\n\tstandbyQueueTimerGate, ok := t.standbyQueueTimerGates[clusterName]\n\tif !ok {\n\t\tpanic(fmt.Sprintf(\"Cannot find standby timer gate for %s.\", clusterName))\n\t}\n\n\tcurTime := t.shard.GetCurrentTime(clusterName)\n\tstandbyQueueTimerGate.SetCurrentTime(curTime)\n\tt.logger.Debug(\"Current time for standby queue timergate is updated\", tag.ClusterName(clusterName), tag.Timestamp(curTime))\n\tstandbyQueueProcessor.notifyNewTimers(info.Tasks)\n}\n\nfunc (t *timerQueueProcessor) FailoverDomain(domainIDs map[string]struct{}) {\n\tif t.shard.GetConfig().DisableTimerFailoverQueue() {\n\t\treturn\n\t}\n\n\t// Failover queue is used to scan all inflight tasks, if queue processor is not\n\t// started, there's no inflight task and we don't need to create a failover processor.\n\t// Also the HandleAction will be blocked if queue processor processing loop is not running.\n\tif atomic.LoadInt32(&t.status) != common.DaemonStatusStarted {\n\t\treturn\n\t}\n\n\tminLevel := t.shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, t.currentClusterName).GetScheduledTime()\n\tstandbyClusterName := t.currentClusterName\n\tfor clusterName := range t.shard.GetClusterMetadata().GetEnabledClusterInfo() {\n\t\tackLevel := t.shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, clusterName).GetScheduledTime()\n\t\tif ackLevel.Before(minLevel) {\n\t\t\tminLevel = ackLevel\n\t\t\tstandbyClusterName = clusterName\n\t\t}\n\t}\n\n\tif standbyClusterName != t.currentClusterName {\n\t\tt.logger.Debugf(\"Timer queue failover will use minLevel: %v from standbyClusterName: %s\", minLevel, standbyClusterName)\n\t} else {\n\t\tt.logger.Debugf(\"Timer queue failover will use minLevel: %v from current cluster: %s\", minLevel, t.currentClusterName)\n\t}\n\n\tmaxReadLevel := time.Time{}\n\tactionResult, err := t.HandleAction(context.Background(), t.currentClusterName, NewGetStateAction())\n\tif err != nil {\n\t\tt.logger.Error(\"Timer failover failed while getting queue states\", tag.WorkflowDomainIDs(domainIDs), tag.Error(err))\n\t\tif err == errProcessorShutdown {\n\t\t\t// processor/shard already shutdown, we don't need to create failover queue processor\n\t\t\treturn\n\t\t}\n\t\t// other errors should never be returned for GetStateAction\n\t\tpanic(fmt.Sprintf(\"unknown error for GetStateAction: %v\", err))\n\t}\n\n\tvar maxReadLevelQueueLevel int\n\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\tqueueReadLevel := queueState.ReadLevel().(timerTaskKey).visibilityTimestamp\n\t\tif maxReadLevel.Before(queueReadLevel) {\n\t\t\tmaxReadLevel = queueReadLevel\n\t\t\tmaxReadLevelQueueLevel = queueState.Level()\n\t\t}\n\t}\n\n\tif !maxReadLevel.IsZero() {\n\t\tt.logger.Debugf(\"Timer queue failover will use maxReadLevel: %v from queue at level: %v\", maxReadLevel, maxReadLevelQueueLevel)\n\t}\n\n\t// TODO: Below Add call has no effect, understand the underlying intent and fix it.\n\tmaxReadLevel.Add(1 * time.Millisecond)\n\n\tt.logger.Info(\"Timer Failover Triggered\",\n\t\ttag.WorkflowDomainIDs(domainIDs),\n\t\ttag.MinLevel(minLevel.UnixNano()),\n\t\ttag.MaxLevel(maxReadLevel.UnixNano()),\n\t)\n\n\tupdateClusterAckLevelFn, failoverQueueProcessor := newTimerQueueFailoverProcessor(\n\t\tstandbyClusterName,\n\t\tt.shard,\n\t\tt.taskProcessor,\n\t\tt.taskAllocator,\n\t\tt.activeTaskExecutor,\n\t\tt.logger,\n\t\tminLevel,\n\t\tmaxReadLevel,\n\t\tdomainIDs,\n\t)\n\n\t// NOTE: READ REF BEFORE MODIFICATION\n\t// ref: historyEngine.go registerDomainFailoverCallback function\n\terr = updateClusterAckLevelFn(newTimerTaskKey(minLevel, 0))\n\tif err != nil {\n\t\tt.logger.Error(\"Error update shard ack level\", tag.Error(err))\n\t}\n\n\t// Failover queue processors are started on the fly when domains are failed over.\n\t// Failover queue processors will be stopped when the timer queue instance is stopped (due to restart or shard movement).\n\t// This means the failover queue processor might not finish its job.\n\t// There is no mechanism to re-start ongoing failover queue processors in the new shard owner.\n\tt.failoverQueueProcessors = append(t.failoverQueueProcessors, failoverQueueProcessor)\n\tfailoverQueueProcessor.Start()\n}\n\nfunc (t *timerQueueProcessor) HandleAction(ctx context.Context, clusterName string, action *Action) (*ActionResult, error) {\n\tvar resultNotificationCh chan actionResultNotification\n\tvar added bool\n\tif clusterName == t.currentClusterName {\n\t\tresultNotificationCh, added = t.activeQueueProcessor.addAction(ctx, action)\n\t} else {\n\t\tfound := false\n\t\tfor standbyClusterName, standbyProcessor := range t.standbyQueueProcessors {\n\t\t\tif clusterName == standbyClusterName {\n\t\t\t\tresultNotificationCh, added = standbyProcessor.addAction(ctx, action)\n\t\t\t\tfound = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tif !found {\n\t\t\treturn nil, fmt.Errorf(\"unknown cluster name: %v\", clusterName)\n\t\t}\n\t}\n\n\tif !added {\n\t\tif ctxErr := ctx.Err(); ctxErr != nil {\n\t\t\treturn nil, ctxErr\n\t\t}\n\t\treturn nil, errProcessorShutdown\n\t}\n\n\tselect {\n\tcase resultNotification := <-resultNotificationCh:\n\t\treturn resultNotification.result, resultNotification.err\n\tcase <-t.shutdownChan:\n\t\treturn nil, errProcessorShutdown\n\tcase <-ctx.Done():\n\t\treturn nil, ctx.Err()\n\t}\n}\n\nfunc (t *timerQueueProcessor) LockTaskProcessing() {\n\tt.logger.Debug(\"Timer queue processor locking task processing\")\n\tt.taskAllocator.Lock()\n}\n\nfunc (t *timerQueueProcessor) UnlockTaskProcessing() {\n\tt.logger.Debug(\"Timer queue processor unlocking task processing\")\n\tt.taskAllocator.Unlock()\n}\n\nfunc (t *timerQueueProcessor) drain() {\n\tif !t.shard.GetConfig().QueueProcessorEnableGracefulSyncShutdown() {\n\t\tif err := t.completeTimer(context.Background()); err != nil {\n\t\t\tt.logger.Error(\"Failed to complete timer task during drain\", tag.Error(err))\n\t\t}\n\t\treturn\n\t}\n\n\t// when graceful shutdown is enabled for queue processor, use a context with timeout\n\tctx, cancel := context.WithTimeout(context.Background(), gracefulShutdownTimeout)\n\tdefer cancel()\n\tif err := t.completeTimer(ctx); err != nil {\n\t\tt.logger.Error(\"Failed to complete timer task during drain\", tag.Error(err))\n\t}\n}\n\nfunc (t *timerQueueProcessor) completeTimerLoop() {\n\tdefer t.shutdownWG.Done()\n\n\tt.logger.Info(\"Timer queue processor completeTimerLoop\")\n\tdefer t.logger.Info(\"Timer queue processor completeTimerLoop completed\")\n\n\tcompleteTimer := time.NewTimer(t.config.TimerProcessorCompleteTimerInterval())\n\tdefer completeTimer.Stop()\n\n\t// Create a retryTimer once, and reset it as needed\n\tretryTimer := time.NewTimer(0)\n\tdefer retryTimer.Stop()\n\t// Stop it immediately because we don't want it to fire initially\n\tif !retryTimer.Stop() {\n\t\t<-retryTimer.C\n\t}\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownChan:\n\t\t\tt.drain()\n\t\t\treturn\n\t\tcase <-completeTimer.C:\n\t\t\tfor attempt := 0; attempt < t.config.TimerProcessorCompleteTimerFailureRetryCount(); attempt++ {\n\t\t\t\terr := t.completeTimer(context.Background())\n\t\t\t\tif err == nil {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\n\t\t\t\tt.logger.Error(\"Failed to complete timer task\", tag.Error(err), tag.Attempt(int32(attempt)))\n\t\t\t\tvar errShardClosed *shard.ErrShardClosed\n\t\t\t\tif errors.As(err, &errShardClosed) {\n\t\t\t\t\tif !t.shard.GetConfig().QueueProcessorEnableGracefulSyncShutdown() {\n\t\t\t\t\t\tgo t.Stop()\n\t\t\t\t\t\treturn\n\t\t\t\t\t}\n\n\t\t\t\t\tt.Stop()\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\t// Reset the retryTimer for the delay between attempts\n\t\t\t\t// TODO: the first retry has 0 backoff, revisit it to see if it's expected\n\t\t\t\tretryDuration := time.Duration(attempt*100) * time.Millisecond\n\t\t\t\tretryTimer.Reset(retryDuration)\n\t\t\t\tselect {\n\t\t\t\tcase <-t.shutdownChan:\n\t\t\t\t\tt.drain()\n\t\t\t\t\treturn\n\t\t\t\tcase <-retryTimer.C:\n\t\t\t\t\t// do nothing. retry loop will continue\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tcompleteTimer.Reset(t.config.TimerProcessorCompleteTimerInterval())\n\t\t}\n\t}\n}\n\nfunc (t *timerQueueProcessor) completeTimer(ctx context.Context) error {\n\tnewAckLevel := maximumTimerTaskKey\n\tactionResult, err := t.HandleAction(ctx, t.currentClusterName, NewGetStateAction())\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\tnewAckLevel = minTaskKey(newAckLevel, queueState.AckLevel())\n\t}\n\n\tfor standbyClusterName := range t.standbyQueueProcessors {\n\t\tactionResult, err := t.HandleAction(ctx, standbyClusterName, NewGetStateAction())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\t\tnewAckLevel = minTaskKey(newAckLevel, queueState.AckLevel())\n\t\t}\n\t}\n\n\tfor _, failoverInfo := range t.shard.GetAllFailoverLevels(persistence.HistoryTaskCategoryTimer) {\n\t\tfailoverLevel := newTimerTaskKey(failoverInfo.MinLevel.GetScheduledTime(), 0)\n\t\tnewAckLevel = minTaskKey(newAckLevel, failoverLevel)\n\t}\n\n\tif newAckLevel == maximumTimerTaskKey {\n\t\tpanic(\"Unable to get timer queue processor ack level\")\n\t}\n\n\tnewAckLevelTimestamp := newAckLevel.(timerTaskKey).visibilityTimestamp\n\tif !t.ackLevel.Before(newAckLevelTimestamp) {\n\t\tt.logger.Debugf(\"Skipping timer task completion because new ack level %v is not before ack level %v\", newAckLevelTimestamp, t.ackLevel)\n\t\treturn nil\n\t}\n\n\tt.logger.Debugf(\"Start completing timer task from: %v, to %v\", t.ackLevel, newAckLevelTimestamp)\n\tt.metricsClient.Scope(metrics.TimerQueueProcessorScope).\n\t\tTagged(metrics.ShardIDTag(t.shard.GetShardID())).\n\t\tIncCounter(metrics.TaskBatchCompleteCounter)\n\n\ttotalDeleted := 0\n\tfor {\n\t\tpageSize := t.config.TimerTaskDeleteBatchSize()\n\t\tresp, err := t.shard.GetExecutionManager().RangeCompleteHistoryTask(ctx, &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(t.ackLevel, 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(newAckLevelTimestamp, 0),\n\t\t\tPageSize:            pageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\ttotalDeleted += resp.TasksCompleted\n\t\tt.logger.Debug(\"Timer task batch deletion\", tag.Dynamic(\"page-size\", pageSize), tag.Dynamic(\"total-deleted\", totalDeleted))\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, pageSize) {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tt.ackLevel = newAckLevelTimestamp\n\n\treturn t.shard.UpdateQueueAckLevel(persistence.HistoryTaskCategoryTimer, persistence.NewHistoryTaskKey(t.ackLevel, 0))\n}\n\nfunc loadTimerProcessingQueueStates(\n\tclusterName string,\n\tshard shard.Context,\n\toptions *queueProcessorOptions,\n\tlogger log.Logger,\n) []ProcessingQueueState {\n\tackLevel := shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, clusterName).GetScheduledTime()\n\tif options.EnableLoadQueueStates() {\n\t\tpStates := shard.GetTimerProcessingQueueStates(clusterName)\n\t\tif validateProcessingQueueStates(pStates, ackLevel) {\n\t\t\treturn convertFromPersistenceTimerProcessingQueueStates(pStates)\n\t\t}\n\n\t\tlogger.Error(\"Incompatible processing queue states and ackLevel\",\n\t\t\ttag.Value(pStates),\n\t\t\ttag.ShardTimerAcks(ackLevel),\n\t\t)\n\t}\n\n\treturn []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tdefaultProcessingQueueLevel,\n\t\t\tnewTimerTaskKey(ackLevel, 0),\n\t\t\tmaximumTimerTaskKey,\n\t\t\tNewDomainFilter(nil, true),\n\t\t),\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_processor_base.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nvar (\n\tmaximumTimerTaskKey = newTimerTaskKey(\n\t\ttime.Unix(0, math.MaxInt64),\n\t\t0,\n\t)\n\ttimerTaskOperationRetryPolicy = common.CreatePersistenceRetryPolicy()\n)\n\ntype (\n\ttimerTaskKey struct {\n\t\tvisibilityTimestamp time.Time\n\t\ttaskID              int64\n\t}\n\n\ttimeTaskReadProgress struct {\n\t\tcurrentQueue  ProcessingQueue\n\t\treadLevel     task.Key\n\t\tmaxReadLevel  task.Key\n\t\tnextPageToken []byte\n\t}\n\n\ttimerQueueProcessorBase struct {\n\t\t*processorBase\n\n\t\ttaskInitializer task.Initializer\n\t\tclusterName     string\n\t\tpollTimeLock    sync.Mutex\n\t\tbackoffTimer    map[int]*time.Timer\n\t\tnextPollTime    map[int]time.Time\n\t\ttimerGate       clock.TimerGate\n\n\t\t// timer notification\n\t\tnewTimerCh  chan struct{}\n\t\tnewTimeLock sync.Mutex\n\t\tnewTime     time.Time\n\n\t\tprocessingQueueReadProgress map[int]timeTaskReadProgress\n\n\t\tupdateAckLevelFn                 func() (bool, task.Key, error)\n\t\tsplitProcessingQueueCollectionFn func(splitPolicy ProcessingQueueSplitPolicy, upsertPollTimeFn func(int, time.Time))\n\t}\n\n\tfilteredTimerTasksResponse struct {\n\t\ttimerTasks         []persistence.Task\n\t\tlookAheadTimestamp time.Time\n\t\tnextPageToken      []byte\n\t}\n)\n\nfunc newTimerQueueProcessorBase(\n\tclusterName string,\n\tshard shard.Context,\n\tprocessingQueueStates []ProcessingQueueState,\n\ttaskProcessor task.Processor,\n\ttimerGate clock.TimerGate,\n\toptions *queueProcessorOptions,\n\tupdateMaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\tqueueShutdown queueShutdownFn,\n\ttaskFilter task.Filter,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) *timerQueueProcessorBase {\n\tprocessorBase := newProcessorBase(\n\t\tshard,\n\t\tprocessingQueueStates,\n\t\ttaskProcessor,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\tlogger.WithTags(tag.ComponentTimerQueue),\n\t\tmetricsClient,\n\t)\n\n\tqueueType := task.QueueTypeActiveTimer\n\tif options.MetricScope == metrics.TimerStandbyQueueProcessorScope {\n\t\tqueueType = task.QueueTypeStandbyTimer\n\t}\n\n\tt := &timerQueueProcessorBase{\n\t\tprocessorBase: processorBase,\n\t\ttaskInitializer: func(taskInfo persistence.Task) task.Task {\n\t\t\treturn task.NewHistoryTask(\n\t\t\t\tshard,\n\t\t\t\ttaskInfo,\n\t\t\t\tqueueType,\n\t\t\t\ttask.InitializeLoggerForTask(shard.GetShardID(), taskInfo, logger),\n\t\t\t\ttaskFilter,\n\t\t\t\ttaskExecutor,\n\t\t\t\ttaskProcessor,\n\t\t\t\tprocessorBase.redispatcher,\n\t\t\t\tshard.GetConfig().TaskCriticalRetryCount,\n\t\t\t)\n\t\t},\n\t\tclusterName:                 clusterName,\n\t\tbackoffTimer:                make(map[int]*time.Timer),\n\t\tnextPollTime:                make(map[int]time.Time),\n\t\ttimerGate:                   timerGate,\n\t\tnewTimerCh:                  make(chan struct{}, 1),\n\t\tprocessingQueueReadProgress: make(map[int]timeTaskReadProgress),\n\t}\n\n\tt.updateAckLevelFn = t.updateAckLevel\n\tt.splitProcessingQueueCollectionFn = t.splitProcessingQueueCollection\n\treturn t\n}\n\nfunc (t *timerQueueProcessorBase) Start() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Timer queue processor state changed\", tag.LifeCycleStarting)\n\tdefer t.logger.Info(\"Timer queue processor state changed\", tag.LifeCycleStarted)\n\n\tt.redispatcher.Start()\n\n\tnewPollTime := time.Time{}\n\tif startJitter := t.options.MaxStartJitterInterval(); startJitter > 0 {\n\t\tnow := t.shard.GetTimeSource().Now()\n\t\tnewPollTime = now.Add(time.Duration(rand.Int63n(int64(startJitter))))\n\t}\n\tfor _, queueCollections := range t.processingQueueCollections {\n\t\tt.upsertPollTime(queueCollections.Level(), newPollTime)\n\t}\n\n\tt.shutdownWG.Add(1)\n\tgo t.processorPump()\n}\n\n// Edge Case: Stop doesn't stop TimerGate if timerQueueProcessorBase is only initiliazed without starting\n// As a result, TimerGate needs to be stopped separately\n// One way to fix this is to make sure TimerGate doesn't start daemon loop on initilization and requires explicit Start\nfunc (t *timerQueueProcessorBase) Stop() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Timer queue processor state changed\", tag.LifeCycleStopping)\n\n\tt.timerGate.Stop()\n\tclose(t.shutdownCh)\n\tt.pollTimeLock.Lock()\n\tfor _, timer := range t.backoffTimer {\n\t\ttimer.Stop()\n\t}\n\tt.pollTimeLock.Unlock()\n\n\tif success := common.AwaitWaitGroup(&t.shutdownWG, gracefulShutdownTimeout); !success {\n\t\tt.logger.Warn(\"timerQueueProcessorBase timed out on shut down\", tag.LifeCycleStopTimedout)\n\t}\n\n\tt.redispatcher.Stop()\n\tt.logger.Info(\"Timer queue processor state changed\", tag.LifeCycleStopped)\n}\n\nfunc (t *timerQueueProcessorBase) processorPump() {\n\tdefer t.shutdownWG.Done()\n\tupdateAckTimer := time.NewTimer(backoff.JitDuration(t.options.UpdateAckInterval(), t.options.UpdateAckIntervalJitterCoefficient()))\n\tdefer updateAckTimer.Stop()\n\tsplitQueueTimer := time.NewTimer(backoff.JitDuration(t.options.SplitQueueInterval(), t.options.SplitQueueIntervalJitterCoefficient()))\n\tdefer splitQueueTimer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownCh:\n\t\t\treturn\n\t\tcase <-t.timerGate.Chan():\n\t\t\tt.updateTimerGates()\n\t\tcase <-updateAckTimer.C:\n\t\t\tif stopPump := t.handleAckLevelUpdate(updateAckTimer); stopPump {\n\t\t\t\tif !t.options.EnableGracefulSyncShutdown() {\n\t\t\t\t\tgo t.Stop()\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\tt.Stop()\n\t\t\t\treturn\n\t\t\t}\n\t\tcase <-t.newTimerCh:\n\t\t\tt.handleNewTimer()\n\t\tcase <-splitQueueTimer.C:\n\t\t\tt.splitQueue(splitQueueTimer)\n\t\tcase notification := <-t.actionNotifyCh:\n\t\t\tt.handleActionNotification(notification)\n\t\t}\n\t}\n}\n\nfunc (t *timerQueueProcessorBase) processQueueCollections(levels map[int]struct{}) {\n\tfor _, queueCollection := range t.processingQueueCollections {\n\t\tlevel := queueCollection.Level()\n\t\tif _, ok := levels[level]; !ok {\n\t\t\tcontinue\n\t\t}\n\n\t\tactiveQueue := queueCollection.ActiveQueue()\n\t\tif activeQueue == nil {\n\t\t\t// process for this queue collection has finished\n\t\t\t// it's possible that new queue will be added to this collection later though,\n\t\t\t// pollTime will be updated after split/merge\n\t\t\tt.logger.Debug(\"Active queue is nil for timer queue at this level\", tag.QueueLevel(level))\n\t\t\tcontinue\n\t\t}\n\n\t\tt.upsertPollTime(level, t.shard.GetCurrentTime(t.clusterName).Add(backoff.JitDuration(\n\t\t\tt.options.MaxPollInterval(),\n\t\t\tt.options.MaxPollIntervalJitterCoefficient(),\n\t\t)))\n\n\t\tvar nextPageToken []byte\n\t\treadLevel := activeQueue.State().ReadLevel()\n\t\tmaxReadLevel := minTaskKey(activeQueue.State().MaxLevel(), t.updateMaxReadLevel())\n\t\tdomainFilter := activeQueue.State().DomainFilter()\n\n\t\tif progress, ok := t.processingQueueReadProgress[level]; ok {\n\t\t\tif progress.currentQueue == activeQueue {\n\t\t\t\treadLevel = progress.readLevel\n\t\t\t\tmaxReadLevel = progress.maxReadLevel\n\t\t\t\tnextPageToken = progress.nextPageToken\n\t\t\t}\n\t\t\tdelete(t.processingQueueReadProgress, level)\n\t\t}\n\n\t\tif !readLevel.Less(maxReadLevel) {\n\t\t\t// notify timer gate about the min time\n\t\t\tt.upsertPollTime(level, readLevel.(timerTaskKey).visibilityTimestamp)\n\t\t\tt.logger.Debug(\"Skipping processing timer queue at this level because readLevel >= maxReadLevel\", tag.QueueLevel(level))\n\t\t\tcontinue\n\t\t}\n\n\t\tctx, cancel := context.WithTimeout(context.Background(), loadQueueTaskThrottleRetryDelay)\n\t\tif err := t.rateLimiter.Wait(ctx); err != nil {\n\t\t\tcancel()\n\t\t\tif level == defaultProcessingQueueLevel {\n\t\t\t\tt.upsertPollTime(level, time.Time{})\n\t\t\t} else {\n\t\t\t\tt.setupBackoffTimer(level)\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\t\tcancel()\n\n\t\tresp, err := t.readAndFilterTasks(readLevel, maxReadLevel, nextPageToken)\n\t\tif err != nil {\n\t\t\tt.logger.Error(\"Processor unable to retrieve tasks\", tag.Error(err))\n\t\t\tt.upsertPollTime(level, time.Time{}) // re-enqueue the event\n\t\t\tcontinue\n\t\t}\n\n\t\ttasks := make(map[task.Key]task.Task)\n\t\ttaskChFull := false\n\t\tsubmittedCount := 0\n\t\tnow := t.shard.GetTimeSource().Now()\n\t\tfor _, taskInfo := range resp.timerTasks {\n\t\t\tif !domainFilter.Filter(taskInfo.GetDomainID()) {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tif persistence.IsTaskCorrupted(taskInfo) {\n\t\t\t\tt.logger.Error(\"Processing queue encountered a corrupted task\", tag.Dynamic(\"task\", taskInfo))\n\t\t\t\tt.metricsScope.IncCounter(metrics.CorruptedHistoryTaskCounter)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\ttask := t.taskInitializer(taskInfo)\n\t\t\ttasks[newTimerTaskKey(taskInfo.GetVisibilityTimestamp(), taskInfo.GetTaskID())] = task\n\t\t\t// shard level metrics for the duration between a task being written to a queue and being fetched from it\n\t\t\tt.metricsScope.RecordHistogramDuration(metrics.TaskEnqueueToFetchLatency, now.Sub(taskInfo.GetVisibilityTimestamp()))\n\t\t\tsubmitted, err := t.submitTask(task)\n\t\t\tif err != nil {\n\t\t\t\t// only err here is due to the fact that processor has been shutdown\n\t\t\t\t// return instead of continue\n\t\t\t\treturn\n\t\t\t}\n\t\t\ttaskChFull = taskChFull || !submitted\n\t\t\tif submitted {\n\t\t\t\tsubmittedCount++\n\t\t\t}\n\t\t}\n\t\tt.logger.Debugf(\"Submitted %d timer tasks successfully out of %d tasks\", submittedCount, len(resp.timerTasks))\n\n\t\tvar newReadLevel task.Key\n\t\tif len(resp.nextPageToken) == 0 {\n\t\t\tnewReadLevel = maxReadLevel\n\t\t\tif !resp.lookAheadTimestamp.IsZero() {\n\t\t\t\t// lookAheadTask may exist only when nextPageToken is empty\n\t\t\t\t// notice that lookAheadTask.VisibilityTimestamp may be larger than shard max read level,\n\t\t\t\t// which means new tasks can be generated before that timestamp. This issue is solved by\n\t\t\t\t// upsertPollTime whenever there are new tasks\n\t\t\t\tlookAheadTimestamp := resp.lookAheadTimestamp\n\t\t\t\tt.upsertPollTime(level, lookAheadTimestamp)\n\t\t\t\tnewReadLevel = minTaskKey(newReadLevel, newTimerTaskKey(lookAheadTimestamp, 0))\n\t\t\t\tt.logger.Debugf(\"nextPageToken is empty for timer queue at level %d so setting newReadLevel to max(lookAheadTask.timestamp: %v, maxReadLevel: %v)\", level, lookAheadTimestamp, maxReadLevel)\n\t\t\t} else {\n\t\t\t\t// else we have no idea when the next poll should happen\n\t\t\t\t// rely on notifyNewTask to trigger the next poll even for non-default queue.\n\t\t\t\t// another option for non-default queue is that we can setup a backoff timer to check back later\n\t\t\t\tt.logger.Debugf(\"nextPageToken is empty for timer queue at level %d and there' no lookAheadTask. setting readLevel to maxReadLevel: %v\", level, maxReadLevel)\n\t\t\t}\n\t\t} else {\n\t\t\t// more tasks should be loaded for this processing queue\n\t\t\t// record the current progress and update the poll time\n\t\t\tif level == defaultProcessingQueueLevel || !taskChFull {\n\t\t\t\tt.logger.Debugf(\"upserting poll time for timer queue at level %d because nextPageToken is not empty and !taskChFull\", level)\n\t\t\t\tt.upsertPollTime(level, time.Time{})\n\t\t\t} else {\n\t\t\t\tt.logger.Debugf(\"setting up backoff timer for timer queue at level %d because nextPageToken is not empty and taskChFull\", level)\n\t\t\t\tt.setupBackoffTimer(level)\n\t\t\t}\n\t\t\tt.processingQueueReadProgress[level] = timeTaskReadProgress{\n\t\t\t\tcurrentQueue:  activeQueue,\n\t\t\t\treadLevel:     readLevel,\n\t\t\t\tmaxReadLevel:  maxReadLevel,\n\t\t\t\tnextPageToken: resp.nextPageToken,\n\t\t\t}\n\t\t\tif len(resp.timerTasks) > 0 {\n\t\t\t\tnewReadLevel = newTimerTaskKey(resp.timerTasks[len(resp.timerTasks)-1].GetVisibilityTimestamp(), 0)\n\t\t\t}\n\t\t}\n\t\tqueueCollection.AddTasks(tasks, newReadLevel)\n\t}\n}\n\n// splitQueue splits the processing queue collection based on some policy\n// and resets the timer with jitter for next run\nfunc (t *timerQueueProcessorBase) splitQueue(splitQueueTimer *time.Timer) {\n\tsplitPolicy := t.initializeSplitPolicy(\n\t\tfunc(key task.Key, domainID string) task.Key {\n\t\t\treturn newTimerTaskKey(\n\t\t\t\tkey.(timerTaskKey).visibilityTimestamp.Add(\n\t\t\t\t\tt.options.SplitLookAheadDurationByDomainID(domainID),\n\t\t\t\t),\n\t\t\t\t0,\n\t\t\t)\n\t\t},\n\t)\n\n\tt.splitProcessingQueueCollectionFn(splitPolicy, t.upsertPollTime)\n\n\tsplitQueueTimer.Reset(backoff.JitDuration(\n\t\tt.options.SplitQueueInterval(),\n\t\tt.options.SplitQueueIntervalJitterCoefficient(),\n\t))\n}\n\n// handleAckLevelUpdate updates ack level and resets timer with jitter.\n// returns true if processing should be terminated\nfunc (t *timerQueueProcessorBase) handleAckLevelUpdate(updateAckTimer *time.Timer) bool {\n\tprocessFinished, _, err := t.updateAckLevelFn()\n\tvar errShardClosed *shard.ErrShardClosed\n\tif errors.As(err, &errShardClosed) || (err == nil && processFinished) {\n\t\treturn true\n\t}\n\tupdateAckTimer.Reset(backoff.JitDuration(\n\t\tt.options.UpdateAckInterval(),\n\t\tt.options.UpdateAckIntervalJitterCoefficient(),\n\t))\n\treturn false\n}\n\nfunc (t *timerQueueProcessorBase) updateTimerGates() {\n\tmaxRedispatchQueueSize := t.options.MaxRedispatchQueueSize()\n\tif t.redispatcher.Size() > maxRedispatchQueueSize {\n\t\tt.redispatcher.Redispatch(maxRedispatchQueueSize)\n\t\tif t.redispatcher.Size() > maxRedispatchQueueSize {\n\t\t\t// if redispatcher still has a large number of tasks\n\t\t\t// this only happens when system is under very high load\n\t\t\t// we should backoff here instead of keeping submitting tasks to task processor\n\t\t\t// don't call t.timerGate.Update(time.Now() + loadQueueTaskThrottleRetryDelay) as the time in\n\t\t\t// standby timer processor is not real time and is managed separately\n\t\t\ttime.Sleep(backoff.JitDuration(\n\t\t\t\tt.options.PollBackoffInterval(),\n\t\t\t\tt.options.PollBackoffIntervalJitterCoefficient(),\n\t\t\t))\n\t\t}\n\t\tt.timerGate.Update(time.Time{})\n\t\treturn\n\t}\n\n\tt.pollTimeLock.Lock()\n\tlevels := make(map[int]struct{})\n\tnow := t.shard.GetCurrentTime(t.clusterName)\n\tfor level, pollTime := range t.nextPollTime {\n\t\tif !now.Before(pollTime) {\n\t\t\tlevels[level] = struct{}{}\n\t\t\tdelete(t.nextPollTime, level)\n\t\t} else {\n\t\t\tt.timerGate.Update(pollTime)\n\t\t}\n\t}\n\tt.pollTimeLock.Unlock()\n\n\tt.processQueueCollections(levels)\n}\n\nfunc (t *timerQueueProcessorBase) handleNewTimer() {\n\tt.newTimeLock.Lock()\n\tnewTime := t.newTime\n\tt.newTime = time.Time{}\n\tt.newTimeLock.Unlock()\n\n\t// New Timer has arrived.\n\tt.metricsScope.IncCounter(metrics.NewTimerNotifyCounter)\n\t// notify all queue collections as they are waiting for the notification when there's\n\t// no more task to process. For non-default queue, we choose to do periodic polling\n\t// in the future, then we don't need to notify them.\n\tfor _, queueCollection := range t.processingQueueCollections {\n\t\tt.upsertPollTime(queueCollection.Level(), newTime)\n\t}\n}\n\nfunc (t *timerQueueProcessorBase) handleActionNotification(notification actionNotification) {\n\tt.processorBase.handleActionNotification(notification, func() {\n\t\tswitch notification.action.ActionType {\n\t\tcase ActionTypeReset:\n\t\t\tt.upsertPollTime(defaultProcessingQueueLevel, time.Time{})\n\t\t}\n\t})\n}\n\nfunc (t *timerQueueProcessorBase) readAndFilterTasks(readLevel, maxReadLevel task.Key, nextPageToken []byte) (*filteredTimerTasksResponse, error) {\n\tresp, err := t.getTimerTasks(readLevel, maxReadLevel, nextPageToken, t.options.BatchSize())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar lookAheadTimestamp time.Time\n\tfilteredTasks := []persistence.Task{}\n\tfor _, timerTask := range resp.Tasks {\n\t\tif !t.isProcessNow(timerTask.GetVisibilityTimestamp()) {\n\t\t\t// found the first task that is not ready to be processed yet.\n\t\t\t// reset NextPageToken so we can load more tasks starting from this lookAheadTask next time.\n\t\t\tlookAheadTimestamp = timerTask.GetVisibilityTimestamp()\n\t\t\tresp.NextPageToken = nil\n\t\t\tbreak\n\t\t}\n\t\tfilteredTasks = append(filteredTasks, timerTask)\n\t}\n\n\tif len(resp.NextPageToken) == 0 && lookAheadTimestamp.IsZero() {\n\t\t// only look ahead within the processing queue boundary\n\t\tlookAheadTask, err := t.readLookAheadTask(maxReadLevel, maximumTimerTaskKey)\n\t\tif err != nil {\n\t\t\t// we don't know if look ahead task exists or not, but we know if it exists,\n\t\t\t// it's visibility timestamp is larger than or equal to maxReadLevel.\n\t\t\t// so, create a fake look ahead task so another load can be triggered at that time.\n\t\t\tlookAheadTimestamp = maxReadLevel.(timerTaskKey).visibilityTimestamp\n\t\t} else if lookAheadTask != nil {\n\t\t\tlookAheadTimestamp = lookAheadTask.GetVisibilityTimestamp()\n\t\t}\n\t}\n\n\tt.logger.Debugf(\"readAndFilterTasks returning %d tasks and lookAheadTimestamp: %#v for readLevel: %#v, maxReadLevel: %#v\", len(filteredTasks), lookAheadTimestamp, readLevel, maxReadLevel)\n\treturn &filteredTimerTasksResponse{\n\t\ttimerTasks:         filteredTasks,\n\t\tlookAheadTimestamp: lookAheadTimestamp,\n\t\tnextPageToken:      resp.NextPageToken,\n\t}, nil\n}\n\nfunc (t *timerQueueProcessorBase) readLookAheadTask(lookAheadStartLevel task.Key, lookAheadMaxLevel task.Key) (persistence.Task, error) {\n\tresp, err := t.getTimerTasks(lookAheadStartLevel, lookAheadMaxLevel, nil, 1)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif len(resp.Tasks) == 1 {\n\t\treturn resp.Tasks[0], nil\n\t}\n\treturn nil, nil\n}\n\nfunc (t *timerQueueProcessorBase) getTimerTasks(readLevel, maxReadLevel task.Key, nextPageToken []byte, batchSize int) (*persistence.GetHistoryTasksResponse, error) {\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            batchSize,\n\t\tNextPageToken:       nextPageToken,\n\t}\n\n\tvar response *persistence.GetHistoryTasksResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresponse, err = t.shard.GetExecutionManager().GetHistoryTasks(ctx, request)\n\t\treturn err\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(timerTaskOperationRetryPolicy),\n\t\tbackoff.WithRetryableError(func(err error) bool { return true }),\n\t)\n\terr := throttleRetry.Do(context.Background(), op)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response, nil\n}\n\nfunc (t *timerQueueProcessorBase) isProcessNow(expiryTime time.Time) bool {\n\tif expiryTime.IsZero() {\n\t\t// return true, but somewhere probably have bug creating empty timerTask.\n\t\tt.logger.Warn(\"Timer task has timestamp zero\")\n\t}\n\treturn !t.shard.GetCurrentTime(t.clusterName).Before(expiryTime)\n}\n\nfunc (t *timerQueueProcessorBase) notifyNewTimers(timerTasks []persistence.Task) {\n\tif len(timerTasks) == 0 {\n\t\treturn\n\t}\n\n\tisActive := t.options.MetricScope == metrics.TimerActiveQueueProcessorScope\n\tminNewTime := timerTasks[0].GetVisibilityTimestamp()\n\tshardIDTag := metrics.ShardIDTag(t.shard.GetShardID())\n\tfor _, timerTask := range timerTasks {\n\t\tts := timerTask.GetVisibilityTimestamp()\n\t\tif ts.Before(minNewTime) {\n\t\t\tminNewTime = ts\n\t\t}\n\n\t\ttaskScopeIdx := task.GetTimerTaskMetricScope(\n\t\t\ttimerTask.GetTaskType(),\n\t\t\tisActive,\n\t\t)\n\t\tt.metricsClient.Scope(taskScopeIdx).Tagged(shardIDTag).IncCounter(metrics.NewTimerNotifyCounter)\n\t}\n\n\tt.notifyNewTimer(minNewTime)\n}\n\nfunc (t *timerQueueProcessorBase) notifyNewTimer(newTime time.Time) {\n\tt.newTimeLock.Lock()\n\tdefer t.newTimeLock.Unlock()\n\n\tif t.newTime.IsZero() || newTime.Before(t.newTime) {\n\t\tt.logger.Debugf(\"Updating newTime from %v to %v\", t.newTime, newTime)\n\t\tt.newTime = newTime\n\t\tselect {\n\t\tcase t.newTimerCh <- struct{}{}:\n\t\t\t// Notified about new time.\n\t\tdefault:\n\t\t\t// Channel \"full\" -> drop and move on, this will happen only if service is in high load.\n\t\t}\n\t}\n}\n\nfunc (t *timerQueueProcessorBase) upsertPollTime(level int, newPollTime time.Time) {\n\tt.pollTimeLock.Lock()\n\tdefer t.pollTimeLock.Unlock()\n\n\tif _, ok := t.backoffTimer[level]; ok {\n\t\t// honor existing backoff timer\n\t\tt.logger.Debugf(\"Skipping upsertPollTime for timer queue at level %d because there's a backoff timer\", level)\n\t\treturn\n\t}\n\n\tcurrentPollTime, ok := t.nextPollTime[level]\n\tif !ok || newPollTime.Before(currentPollTime) {\n\t\tt.logger.Debugf(\"Updating poll timer for timer queue at level %d. CurrentPollTime: %v, newPollTime: %v\", level, currentPollTime, newPollTime)\n\t\tt.nextPollTime[level] = newPollTime\n\t\tt.timerGate.Update(newPollTime)\n\t\treturn\n\t}\n\n\tt.logger.Debugf(\"Skipping upsertPollTime for level %d because currentPollTime %v is before newPollTime %v\", level, currentPollTime, newPollTime)\n}\n\n// setupBackoffTimer will trigger a poll for the specified processing queue collection\n// after a certain period of (real) time. This means for standby timer, even if the cluster time\n// has not been updated, the poll will still be triggered when the timer fired. Use this function\n// for delaying the load for processing queue. If a poll should be triggered immediately\n// use upsertPollTime.\nfunc (t *timerQueueProcessorBase) setupBackoffTimer(level int) {\n\tt.pollTimeLock.Lock()\n\tdefer t.pollTimeLock.Unlock()\n\n\tif _, ok := t.backoffTimer[level]; ok {\n\t\t// honor existing backoff timer\n\t\treturn\n\t}\n\n\tt.metricsScope.IncCounter(metrics.ProcessingQueueThrottledCounter)\n\tt.logger.Info(\"Throttled processing queue\", tag.QueueLevel(level))\n\tbackoffDuration := backoff.JitDuration(\n\t\tt.options.PollBackoffInterval(),\n\t\tt.options.PollBackoffIntervalJitterCoefficient(),\n\t)\n\tt.backoffTimer[level] = time.AfterFunc(backoffDuration, func() {\n\t\tselect {\n\t\tcase <-t.shutdownCh:\n\t\t\treturn\n\t\tdefault:\n\t\t}\n\n\t\tt.pollTimeLock.Lock()\n\t\tdefer t.pollTimeLock.Unlock()\n\n\t\tt.nextPollTime[level] = time.Time{}\n\t\tt.timerGate.Update(time.Time{})\n\t\tdelete(t.backoffTimer, level)\n\t})\n}\n\nfunc newTimerTaskKey(visibilityTimestamp time.Time, taskID int64) task.Key {\n\treturn timerTaskKey{\n\t\tvisibilityTimestamp: visibilityTimestamp,\n\t\ttaskID:              taskID,\n\t}\n}\n\nfunc (k timerTaskKey) Less(\n\tkey task.Key,\n) bool {\n\ttimerKey := key.(timerTaskKey)\n\tif k.visibilityTimestamp.Equal(timerKey.visibilityTimestamp) {\n\t\treturn k.taskID < timerKey.taskID\n\t}\n\treturn k.visibilityTimestamp.Before(timerKey.visibilityTimestamp)\n}\n\nfunc (k timerTaskKey) String() string {\n\treturn fmt.Sprintf(\"{visibilityTimestamp: %v, taskID: %v}\", k.visibilityTimestamp, k.taskID)\n}\n\nfunc newTimerQueueProcessorOptions(\n\tconfig *config.Config,\n\tisActive bool,\n\tisFailover bool,\n) *queueProcessorOptions {\n\toptions := &queueProcessorOptions{\n\t\tBatchSize:                            config.TimerTaskBatchSize,\n\t\tDeleteBatchSize:                      config.TimerTaskDeleteBatchSize,\n\t\tMaxPollRPS:                           config.TimerProcessorMaxPollRPS,\n\t\tMaxPollInterval:                      config.TimerProcessorMaxPollInterval,\n\t\tMaxPollIntervalJitterCoefficient:     config.TimerProcessorMaxPollIntervalJitterCoefficient,\n\t\tUpdateAckInterval:                    config.TimerProcessorUpdateAckInterval,\n\t\tUpdateAckIntervalJitterCoefficient:   config.TimerProcessorUpdateAckIntervalJitterCoefficient,\n\t\tMaxRedispatchQueueSize:               config.TimerProcessorMaxRedispatchQueueSize,\n\t\tSplitQueueInterval:                   config.TimerProcessorSplitQueueInterval,\n\t\tSplitQueueIntervalJitterCoefficient:  config.TimerProcessorSplitQueueIntervalJitterCoefficient,\n\t\tPollBackoffInterval:                  config.QueueProcessorPollBackoffInterval,\n\t\tPollBackoffIntervalJitterCoefficient: config.QueueProcessorPollBackoffIntervalJitterCoefficient,\n\t\tEnableGracefulSyncShutdown:           config.QueueProcessorEnableGracefulSyncShutdown,\n\t}\n\n\tif isFailover {\n\t\t// disable queue split for failover processor\n\t\toptions.EnableSplit = dynamicproperties.GetBoolPropertyFn(false)\n\n\t\t// disable persist and load processing queue states for failover processor as it will never be split\n\t\toptions.EnablePersistQueueStates = dynamicproperties.GetBoolPropertyFn(false)\n\t\toptions.EnableLoadQueueStates = dynamicproperties.GetBoolPropertyFn(false)\n\n\t\toptions.MaxStartJitterInterval = config.TimerProcessorFailoverMaxStartJitterInterval\n\t} else {\n\t\toptions.EnableSplit = config.QueueProcessorEnableSplit\n\t\toptions.SplitMaxLevel = config.QueueProcessorSplitMaxLevel\n\t\toptions.EnableRandomSplitByDomainID = config.QueueProcessorEnableRandomSplitByDomainID\n\t\toptions.RandomSplitProbability = config.QueueProcessorRandomSplitProbability\n\t\toptions.EnablePendingTaskSplitByDomainID = config.QueueProcessorEnablePendingTaskSplitByDomainID\n\t\toptions.PendingTaskSplitThreshold = config.QueueProcessorPendingTaskSplitThreshold\n\t\toptions.EnableStuckTaskSplitByDomainID = config.QueueProcessorEnableStuckTaskSplitByDomainID\n\t\toptions.StuckTaskSplitThreshold = config.QueueProcessorStuckTaskSplitThreshold\n\t\toptions.SplitLookAheadDurationByDomainID = config.QueueProcessorSplitLookAheadDurationByDomainID\n\n\t\toptions.EnablePersistQueueStates = config.QueueProcessorEnablePersistQueueStates\n\t\toptions.EnableLoadQueueStates = config.QueueProcessorEnableLoadQueueStates\n\n\t\toptions.MaxStartJitterInterval = dynamicproperties.GetDurationPropertyFn(0)\n\t}\n\n\tif isActive {\n\t\toptions.MetricScope = metrics.TimerActiveQueueProcessorScope\n\t\toptions.RedispatchInterval = config.ActiveTaskRedispatchInterval\n\t} else {\n\t\toptions.MetricScope = metrics.TimerStandbyQueueProcessorScope\n\t\toptions.RedispatchInterval = config.StandbyTaskRedispatchInterval\n\t}\n\n\treturn options\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_processor_base_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\ttimerQueueProcessorBaseSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockShard            *shard.TestContext\n\t\tmockTaskProcessor    *task.MockProcessor\n\t\tmockQueueSplitPolicy *MockProcessingQueueSplitPolicy\n\n\t\tclusterName   string\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tmetricsScope  metrics.Scope\n\t}\n)\n\nfunc TestTimerQueueProcessorBaseSuite(t *testing.T) {\n\ts := new(timerQueueProcessorBaseSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockQueueSplitPolicy = NewMockProcessingQueueSplitPolicy(s.controller)\n\ts.mockTaskProcessor = task.NewMockProcessor(s.controller)\n\n\ts.clusterName = cluster.TestCurrentClusterName\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.metricsScope = s.metricsClient.Scope(metrics.TimerQueueProcessorScope)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n\tdefer goleak.VerifyNone(s.T())\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestIsProcessNow() {\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\ts.True(timerQueueProcessBase.isProcessNow(time.Time{}))\n\n\tnow := s.mockShard.GetCurrentTime(s.clusterName)\n\ts.True(timerQueueProcessBase.isProcessNow(now))\n\n\ttimeBefore := now.Add(-10 * time.Second)\n\ts.True(timerQueueProcessBase.isProcessNow(timeBefore))\n\n\ttimeAfter := now.Add(10 * time.Second)\n\ts.False(timerQueueProcessBase.isProcessNow(timeAfter))\n\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestGetTimerTasks_More() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(10*time.Second), 0)\n\tbatchSize := 10\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            batchSize,\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"some random output next page token\"),\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.getTimerTasks(readLevel, maxReadLevel, request.NextPageToken, batchSize)\n\ts.Nil(err)\n\ts.Equal(response.Tasks, got.Tasks)\n\ts.Equal(response.NextPageToken, got.NextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestGetTimerTasks_NoMore() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(10*time.Second), 0)\n\tbatchSize := 10\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            batchSize,\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.getTimerTasks(readLevel, maxReadLevel, request.NextPageToken, batchSize)\n\ts.Nil(err)\n\ts.Equal(response.Tasks, got.Tasks)\n\ts.Empty(got.NextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadLookAheadTask() {\n\tshardMaxReadLevel := s.mockShard.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, s.clusterName).GetScheduledTime()\n\treadLevel := newTimerTaskKey(shardMaxReadLevel, 0)\n\tmaxReadLevel := newTimerTaskKey(shardMaxReadLevel.Add(10*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            1,\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: shardMaxReadLevel,\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(79),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"some random next page token\"),\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tlookAheadTask, err := timerQueueProcessBase.readLookAheadTask(readLevel, maxReadLevel)\n\ts.Nil(err)\n\ts.Equal(response.Tasks[0], lookAheadTask)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadAndFilterTasks_NoLookAhead_NoNextPage() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(1*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tlookAheadRequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maximumTimerTaskKey.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            1,\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, lookAheadRequest).Return(&persistence.GetHistoryTasksResponse{}, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.readAndFilterTasks(readLevel, maxReadLevel, request.NextPageToken)\n\ts.Nil(err)\n\ts.Equal(response.Tasks, got.timerTasks)\n\ts.Zero(got.lookAheadTimestamp)\n\ts.Nil(got.nextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadAndFilterTasks_NoLookAhead_HasNextPage() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(1*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"some random next page token\"),\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.readAndFilterTasks(readLevel, maxReadLevel, request.NextPageToken)\n\ts.Nil(err)\n\ts.Equal(response.Tasks, got.timerTasks)\n\ts.Zero(got.lookAheadTimestamp)\n\ts.Equal(response.NextPageToken, got.nextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadAndFilterTasks_HasLookAhead_NoNextPage() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(1*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(500 * time.Millisecond),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.readAndFilterTasks(readLevel, maxReadLevel, request.NextPageToken)\n\ts.Nil(err)\n\ts.Equal([]persistence.Task{response.Tasks[0]}, got.timerTasks)\n\ts.Equal(response.Tasks[1].GetVisibilityTimestamp(), got.lookAheadTimestamp)\n\ts.Nil(got.nextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadAndFilterTasks_HasLookAhead_HasNextPage() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(1*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(500 * time.Millisecond),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"some random next page token\"),\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.readAndFilterTasks(readLevel, maxReadLevel, request.NextPageToken)\n\ts.Nil(err)\n\ts.Equal([]persistence.Task{response.Tasks[0]}, got.timerTasks)\n\ts.Equal(response.Tasks[1].GetVisibilityTimestamp(), got.lookAheadTimestamp)\n\ts.Equal(response.NextPageToken, got.nextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestReadAndFilterTasks_LookAheadFailed_NoNextPage() {\n\treadLevel := newTimerTaskKey(time.Now().Add(-10*time.Second), 0)\n\tmaxReadLevel := newTimerTaskKey(time.Now().Add(1*time.Second), 0)\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(readLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       []byte(\"some random input next page token\"),\n\t}\n\n\tlookAheadRequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(maxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maximumTimerTaskKey.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            1,\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-5 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Now().Add(-500 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, lookAheadRequest).Return(nil, errors.New(\"some random error\")).Times(0)\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\tgot, err := timerQueueProcessBase.readAndFilterTasks(readLevel, maxReadLevel, request.NextPageToken)\n\ts.Nil(err)\n\ts.Equal(response.Tasks, got.timerTasks)\n\ts.Equal(maxReadLevel.(timerTaskKey).visibilityTimestamp, got.lookAheadTimestamp)\n\ts.Nil(got.nextPageToken)\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestNotifyNewTimes() {\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(nil, nil, nil, nil, nil)\n\tdefer done()\n\n\t// assert the initial state\n\ts.True(timerQueueProcessBase.newTime.IsZero())\n\tselect {\n\tcase <-timerQueueProcessBase.newTimerCh:\n\tdefault:\n\t}\n\n\tnow := time.Now()\n\ttimerQueueProcessBase.notifyNewTimers([]persistence.Task{\n\t\t&persistence.UserTimerTask{\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVisibilityTimestamp: now.Add(5 * time.Second),\n\t\t\t\tTaskID:              int64(59),\n\t\t\t},\n\t\t\tEventID: int64(28),\n\t\t},\n\t\t&persistence.UserTimerTask{\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVisibilityTimestamp: now.Add(1 * time.Second),\n\t\t\t\tTaskID:              int64(59),\n\t\t\t},\n\t\t\tEventID: int64(28),\n\t\t},\n\t})\n\tselect {\n\tcase <-timerQueueProcessBase.newTimerCh:\n\t\ts.Equal(now.Add(1*time.Second), timerQueueProcessBase.newTime)\n\tdefault:\n\t\ts.Fail(\"should notify new timer\")\n\t}\n\n\ttimerQueueProcessBase.notifyNewTimers([]persistence.Task{\n\t\t&persistence.UserTimerTask{\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVisibilityTimestamp: now.Add(10 * time.Second),\n\t\t\t\tTaskID:              int64(59),\n\t\t\t},\n\t\t\tEventID: int64(28),\n\t\t},\n\t})\n\tselect {\n\tcase <-timerQueueProcessBase.newTimerCh:\n\t\ts.Fail(\"should not notify new timer\")\n\tdefault:\n\t\ts.Equal(now.Add(1*time.Second), timerQueueProcessBase.newTime)\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestProcessQueueCollections_SkipRead() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tshardMaxReadLevel := newTimerTaskKey(now, 0)\n\tackLevel := newTimerTaskKey(now.Add(50*time.Millisecond), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn shardMaxReadLevel\n\t}\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tdefer done()\n\ttimerQueueProcessBase.processQueueCollections(map[int]struct{}{queueLevel: {}})\n\n\ts.Len(timerQueueProcessBase.processingQueueCollections, 1)\n\ts.Len(timerQueueProcessBase.processingQueueCollections[0].Queues(), 1)\n\tactiveQueue := timerQueueProcessBase.processingQueueCollections[0].ActiveQueue()\n\ts.NotNil(activeQueue)\n\ts.Equal(ackLevel, activeQueue.State().AckLevel())\n\ts.Equal(ackLevel, activeQueue.State().ReadLevel())\n\ts.Equal(maxLevel, activeQueue.State().MaxLevel())\n\n\ts.Empty(timerQueueProcessBase.processingQueueReadProgress)\n\n\ts.Empty(timerQueueProcessBase.backoffTimer)\n\ttime.Sleep(100 * time.Millisecond)\n\ts.True(timerQueueProcessBase.nextPollTime[queueLevel].Before(s.mockShard.GetTimeSource().Now()))\n\tselect {\n\tcase <-timerQueueProcessBase.timerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer gate should fire\")\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestProcessBatch_HasNextPage() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tackLevel := newTimerTaskKey(now.Add(-5*time.Second), 0)\n\tshardMaxReadLevel := newTimerTaskKey(now.Add(1*time.Second), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"excludedDomain\": {}}, true),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn shardMaxReadLevel\n\t}\n\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(shardMaxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: now.Add(-3 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"excludedDomain\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: now.Add(-2 * time.Second),\n\t\t\t\t\tTaskID:              int64(60),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"some random next page token\"),\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tdefer done()\n\ttimerQueueProcessBase.processQueueCollections(map[int]struct{}{queueLevel: {}})\n\n\ts.Len(timerQueueProcessBase.processingQueueCollections, 1)\n\ts.Len(timerQueueProcessBase.processingQueueCollections[0].Queues(), 1)\n\tactiveQueue := timerQueueProcessBase.processingQueueCollections[0].ActiveQueue()\n\ts.NotNil(activeQueue)\n\ts.Equal(ackLevel, activeQueue.State().AckLevel())\n\ts.Equal(newTimerTaskKey(response.Tasks[1].GetVisibilityTimestamp(), 0), activeQueue.State().ReadLevel())\n\ts.Equal(maxLevel, activeQueue.State().MaxLevel())\n\ts.Len(activeQueue.(*processingQueueImpl).outstandingTasks, 1)\n\n\ts.Len(timerQueueProcessBase.processingQueueReadProgress, 1)\n\ts.Equal(timeTaskReadProgress{\n\t\tcurrentQueue:  activeQueue,\n\t\treadLevel:     ackLevel,\n\t\tmaxReadLevel:  shardMaxReadLevel,\n\t\tnextPageToken: response.NextPageToken,\n\t}, timerQueueProcessBase.processingQueueReadProgress[0])\n\n\ts.True(timerQueueProcessBase.nextPollTime[queueLevel].IsZero())\n\ts.Empty(timerQueueProcessBase.backoffTimer)\n\ttime.Sleep(100 * time.Millisecond)\n\tselect {\n\tcase <-timerQueueProcessBase.timerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer gate should fire\")\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestProcessBatch_NoNextPage_HasLookAhead() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tackLevel := newTimerTaskKey(now.Add(-5*time.Second), 0)\n\tshardMaxReadLevel := newTimerTaskKey(now.Add(1*time.Second), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"excludedDomain\": {}}, true),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn shardMaxReadLevel\n\t}\n\n\trequestNextPageToken := []byte(\"some random input next page token\")\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(shardMaxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       requestNextPageToken,\n\t}\n\n\tlookAheadTaskTimestamp := now.Add(50 * time.Millisecond)\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: now.Add(-3 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"excludedDomain\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: lookAheadTaskTimestamp,\n\t\t\t\t\tTaskID:              int64(60),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tdefer done()\n\ttimerQueueProcessBase.processingQueueReadProgress[0] = timeTaskReadProgress{\n\t\tcurrentQueue:  timerQueueProcessBase.processingQueueCollections[0].ActiveQueue(),\n\t\treadLevel:     ackLevel,\n\t\tmaxReadLevel:  shardMaxReadLevel,\n\t\tnextPageToken: requestNextPageToken,\n\t}\n\ttimerQueueProcessBase.processQueueCollections(map[int]struct{}{queueLevel: {}})\n\n\ts.Len(timerQueueProcessBase.processingQueueCollections, 1)\n\ts.Len(timerQueueProcessBase.processingQueueCollections[0].Queues(), 1)\n\tactiveQueue := timerQueueProcessBase.processingQueueCollections[0].ActiveQueue()\n\ts.NotNil(activeQueue)\n\ts.Equal(ackLevel, activeQueue.State().AckLevel())\n\ts.Equal(newTimerTaskKey(lookAheadTaskTimestamp, 0), activeQueue.State().ReadLevel())\n\ts.Equal(maxLevel, activeQueue.State().MaxLevel())\n\ts.Len(activeQueue.(*processingQueueImpl).outstandingTasks, 1)\n\n\ts.Empty(timerQueueProcessBase.processingQueueReadProgress)\n\n\ts.Empty(timerQueueProcessBase.backoffTimer)\n\ts.Equal(lookAheadTaskTimestamp, timerQueueProcessBase.nextPollTime[queueLevel])\n\ttime.Sleep(100 * time.Millisecond)\n\tselect {\n\tcase <-timerQueueProcessBase.timerGate.Chan():\n\tdefault:\n\t\ts.Fail(\"timer gate should fire\")\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestProcessBatch_NoNextPage_NoLookAhead() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tackLevel := newTimerTaskKey(now.Add(-5*time.Second), 0)\n\tshardMaxReadLevel := newTimerTaskKey(now.Add(1*time.Second), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"excludedDomain\": {}}, true),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn shardMaxReadLevel\n\t}\n\n\trequestNextPageToken := []byte(\"some random input next page token\")\n\trequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(shardMaxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            s.mockShard.GetConfig().TimerTaskBatchSize(),\n\t\tNextPageToken:       requestNextPageToken,\n\t}\n\n\tlookAheadRequest := &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(shardMaxReadLevel.(timerTaskKey).visibilityTimestamp, 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(maximumTimerTaskKey.(timerTaskKey).visibilityTimestamp, 0),\n\t\tPageSize:            1,\n\t\tNextPageToken:       nil,\n\t}\n\n\tresponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"some random domain ID\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: now.Add(-3 * time.Second),\n\t\t\t\t\tTaskID:              int64(59),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t\t&persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"excludedDomain\",\n\t\t\t\t\tWorkflowID: \"some random workflow ID\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: now.Add(-2 * time.Second),\n\t\t\t\t\tTaskID:              int64(60),\n\t\t\t\t\tVersion:             int64(1),\n\t\t\t\t},\n\t\t\t\tEventID: int64(28),\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n\n\tmockExecutionMgr := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, request).Return(response, nil).Once()\n\tmockExecutionMgr.On(\"GetHistoryTasks\", mock.Anything, lookAheadRequest).Return(&persistence.GetHistoryTasksResponse{}, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\ttimerQueueProcessBase, done := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tdefer done()\n\ttimerQueueProcessBase.processingQueueReadProgress[0] = timeTaskReadProgress{\n\t\tcurrentQueue:  timerQueueProcessBase.processingQueueCollections[0].ActiveQueue(),\n\t\treadLevel:     ackLevel,\n\t\tmaxReadLevel:  shardMaxReadLevel,\n\t\tnextPageToken: requestNextPageToken,\n\t}\n\ttimerQueueProcessBase.processQueueCollections(map[int]struct{}{queueLevel: {}})\n\n\ts.Len(timerQueueProcessBase.processingQueueCollections, 1)\n\ts.Len(timerQueueProcessBase.processingQueueCollections[0].Queues(), 1)\n\tactiveQueue := timerQueueProcessBase.processingQueueCollections[0].ActiveQueue()\n\ts.NotNil(activeQueue)\n\ts.Equal(ackLevel, activeQueue.State().AckLevel())\n\ts.Equal(shardMaxReadLevel, activeQueue.State().ReadLevel())\n\ts.Equal(maxLevel, activeQueue.State().MaxLevel())\n\ts.Len(activeQueue.(*processingQueueImpl).outstandingTasks, 1)\n\n\ts.Empty(timerQueueProcessBase.processingQueueReadProgress)\n\n\t_, ok := timerQueueProcessBase.nextPollTime[queueLevel]\n\ts.True(ok) // this is the poll time for max poll interval\n\ttime.Sleep(100 * time.Millisecond)\n\tselect {\n\tcase <-timerQueueProcessBase.timerGate.Chan():\n\t\ts.Fail(\"timer gate should not fire\")\n\tdefault:\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestTimerProcessorPump_HandleAckLevelUpdate() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tackLevel := newTimerTaskKey(now.Add(50*time.Millisecond), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTimerTaskKey(now, 0)\n\t}\n\n\ttimerQueueProcessBase, _ := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\ttimerQueueProcessBase.options.UpdateAckInterval = dynamicproperties.GetDurationPropertyFn(1 * time.Millisecond)\n\tupdatedCh := make(chan struct{}, 1)\n\ttimerQueueProcessBase.updateAckLevelFn = func() (bool, task.Key, error) {\n\t\tupdatedCh <- struct{}{}\n\t\treturn false, nil, nil\n\t}\n\ttimerQueueProcessBase.Start()\n\tdefer timerQueueProcessBase.Stop()\n\n\tselect {\n\tcase <-updatedCh:\n\t\treturn\n\tcase <-time.After(100 * time.Millisecond):\n\t\ts.Fail(\"Ack level update not called\")\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) TestTimerProcessorPump_SplitQueue() {\n\tnow := time.Now()\n\tqueueLevel := 0\n\tackLevel := newTimerTaskKey(now.Add(50*time.Millisecond), 0)\n\tmaxLevel := newTimerTaskKey(now.Add(10*time.Second), 0)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTimerTaskKey(now, 0)\n\t}\n\n\ttimerQueueProcessBase, _ := s.newTestTimerQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\ttimerQueueProcessBase.options.SplitQueueInterval = dynamicproperties.GetDurationPropertyFn(1 * time.Millisecond)\n\tsplittedCh := make(chan struct{}, 1)\n\ttimerQueueProcessBase.splitProcessingQueueCollectionFn = func(splitPolicy ProcessingQueueSplitPolicy, upsertPollTimeFn func(int, time.Time)) {\n\t\tsplittedCh <- struct{}{}\n\t}\n\ttimerQueueProcessBase.Start()\n\tdefer timerQueueProcessBase.Stop()\n\n\tselect {\n\tcase <-splittedCh:\n\t\treturn\n\tcase <-time.After(100 * time.Millisecond):\n\t\ts.Fail(\"splitProcessingQueueCollectionFn not called\")\n\t}\n}\n\nfunc (s *timerQueueProcessorBaseSuite) newTestTimerQueueProcessorBase(\n\tprocessingQueueStates []ProcessingQueueState,\n\tupdateMaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\tqueueShutdown queueShutdownFn,\n) (*timerQueueProcessorBase, func()) {\n\ttimerGate := clock.NewTimerGate(s.mockShard.GetTimeSource())\n\n\treturn newTimerQueueProcessorBase(\n\t\t\ts.clusterName,\n\t\t\ts.mockShard,\n\t\t\tprocessingQueueStates,\n\t\t\ts.mockTaskProcessor,\n\t\t\ttimerGate,\n\t\t\tnewTimerQueueProcessorOptions(s.mockShard.GetConfig(), true, false),\n\t\t\tupdateMaxReadLevel,\n\t\t\tupdateClusterAckLevel,\n\t\t\tupdateProcessingQueueStates,\n\t\t\tqueueShutdown,\n\t\t\tnil,\n\t\t\tnil,\n\t\t\ts.logger,\n\t\t\ts.metricsClient,\n\t\t), func() {\n\t\t\ttimerGate.Stop()\n\t\t}\n}\n"
  },
  {
    "path": "service/history/queue/timer_queue_standby_processor.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc newTimerQueueStandbyProcessor(\n\tclusterName string,\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n) (*timerQueueProcessorBase, clock.EventTimerGate) {\n\tconfig := shard.GetConfig()\n\toptions := newTimerQueueProcessorOptions(config, false, false)\n\n\tlogger = logger.WithTags(tag.ClusterName(clusterName))\n\n\ttaskFilter := func(timer persistence.Task) (bool, error) {\n\t\tif timer.GetTaskCategory() != persistence.HistoryTaskCategoryTimer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shard, timer.GetDomainID()); notRegistered && err == nil {\n\t\t\t// Allow deletion tasks for deprecated domains\n\t\t\tif timer.GetTaskType() == persistence.TaskTypeDeleteHistoryEvent {\n\t\t\t\treturn true, nil\n\t\t\t}\n\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in standby timer queue.\", tag.WorkflowDomainID(timer.GetDomainID()), tag.Value(timer))\n\t\t\treturn false, nil\n\t\t}\n\t\tif timer.GetTaskType() == persistence.TaskTypeWorkflowTimeout ||\n\t\t\ttimer.GetTaskType() == persistence.TaskTypeDeleteHistoryEvent {\n\t\t\tdomainEntry, err := shard.GetDomainCache().GetDomainByID(timer.GetDomainID())\n\t\t\tif err == nil {\n\t\t\t\tif domainEntry.HasReplicationCluster(clusterName) {\n\t\t\t\t\t// guarantee the processing of workflow execution history deletion\n\t\t\t\t\treturn true, nil\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\t\t\t// retry the task if failed to find the domain\n\t\t\t\t\tlogger.Warn(\"Cannot find domain\", tag.WorkflowDomainID(timer.GetDomainID()))\n\t\t\t\t\treturn false, err\n\t\t\t\t}\n\t\t\t\tlogger.Warn(\"Cannot find domain, default to not process task.\", tag.WorkflowDomainID(timer.GetDomainID()), tag.Value(timer))\n\t\t\t\treturn false, nil\n\t\t\t}\n\t\t}\n\t\treturn taskAllocator.VerifyStandbyTask(clusterName, timer.GetDomainID(), timer.GetWorkflowID(), timer.GetRunID(), timer)\n\t}\n\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTimerTaskKey(shard.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, clusterName).GetScheduledTime(), 0)\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\treturn shard.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, clusterName, persistence.NewHistoryTaskKey(ackLevel.(timerTaskKey).visibilityTimestamp, 0))\n\t}\n\n\tupdateProcessingQueueStates := func(states []ProcessingQueueState) error {\n\t\tpStates := convertToPersistenceTimerProcessingQueueStates(states)\n\t\treturn shard.UpdateTimerProcessingQueueStates(clusterName, pStates)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn nil\n\t}\n\n\tremoteTimerGate := clock.NewEventTimerGate(shard.GetCurrentTime(clusterName))\n\n\treturn newTimerQueueProcessorBase(\n\t\tclusterName,\n\t\tshard,\n\t\tloadTimerProcessingQueueStates(clusterName, shard, options, logger),\n\t\ttaskProcessor,\n\t\tremoteTimerGate,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t), remoteTimerGate\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_processor.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nvar (\n\terrUnexpectedQueueTask = errors.New(\"unexpected queue task\")\n\terrProcessorShutdown   = errors.New(\"queue processor has been shutdown\")\n\n\tmaximumTransferTaskKey = newTransferTaskKey(math.MaxInt64)\n)\n\ntype transferQueueProcessor struct {\n\tshard         shard.Context\n\thistoryEngine engine.Engine\n\ttaskProcessor task.Processor\n\n\tconfig             *config.Config\n\tcurrentClusterName string\n\n\tmetricsClient metrics.Client\n\tlogger        log.Logger\n\n\tstatus       int32\n\tshutdownChan chan struct{}\n\tshutdownWG   sync.WaitGroup\n\n\tackLevel                int64\n\ttaskAllocator           TaskAllocator\n\tactiveTaskExecutor      task.Executor\n\tactiveQueueProcessor    *transferQueueProcessorBase\n\tstandbyQueueProcessors  map[string]*transferQueueProcessorBase\n\tfailoverQueueProcessors []*transferQueueProcessorBase\n}\n\n// NewTransferQueueProcessor creates a new transfer QueueProcessor\nfunc NewTransferQueueProcessor(\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\texecutionCache execution.Cache,\n\tworkflowResetter reset.WorkflowResetter,\n\tarchivalClient archiver.Client,\n\texecutionCheck invariant.Invariant,\n\twfIDCache workflowcache.WFCache,\n) Processor {\n\tlogger := shard.GetLogger().WithTags(tag.ComponentTransferQueue)\n\tcurrentClusterName := shard.GetClusterMetadata().GetCurrentClusterName()\n\tconfig := shard.GetConfig()\n\ttaskAllocator := NewTaskAllocator(shard)\n\n\tactiveLogger := logger.WithTags(tag.QueueTypeActive)\n\n\tactiveTaskExecutor := task.NewTransferActiveTaskExecutor(\n\t\tshard,\n\t\tarchivalClient,\n\t\texecutionCache,\n\t\tworkflowResetter,\n\t\tactiveLogger,\n\t\tconfig,\n\t\twfIDCache,\n\t)\n\n\tactiveQueueProcessor := newTransferQueueActiveProcessor(\n\t\tshard,\n\t\ttaskProcessor,\n\t\ttaskAllocator,\n\t\tactiveTaskExecutor,\n\t\tactiveLogger,\n\t)\n\n\tstandbyQueueProcessors := make(map[string]*transferQueueProcessorBase)\n\tfor clusterName := range shard.GetClusterMetadata().GetRemoteClusterInfo() {\n\t\thistoryResender := ndc.NewHistoryResender(\n\t\t\tshard.GetDomainCache(),\n\t\t\tshard.GetService().GetClientBean(),\n\t\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\t\treturn shard.GetEngine().ReplicateEventsV2(ctx, request)\n\t\t\t},\n\t\t\tconfig.StandbyTaskReReplicationContextTimeout,\n\t\t\texecutionCheck,\n\t\t\tshard.GetLogger(),\n\t\t)\n\t\tstandByLogger := logger.WithTags(tag.QueueTypeStandby, tag.ActiveClusterName(clusterName))\n\t\tstandbyTaskExecutor := task.NewTransferStandbyTaskExecutor(\n\t\t\tshard,\n\t\t\tarchivalClient,\n\t\t\texecutionCache,\n\t\t\thistoryResender,\n\t\t\tstandByLogger,\n\t\t\tclusterName,\n\t\t\tconfig,\n\t\t)\n\t\tstandbyQueueProcessors[clusterName] = newTransferQueueStandbyProcessor(\n\t\t\tclusterName,\n\t\t\tshard,\n\t\t\ttaskProcessor,\n\t\t\ttaskAllocator,\n\t\t\tstandbyTaskExecutor,\n\t\t\tstandByLogger,\n\t\t)\n\t}\n\n\treturn &transferQueueProcessor{\n\t\tshard:                  shard,\n\t\ttaskProcessor:          taskProcessor,\n\t\tconfig:                 config,\n\t\tcurrentClusterName:     currentClusterName,\n\t\tmetricsClient:          shard.GetMetricsClient(),\n\t\tlogger:                 logger,\n\t\tstatus:                 common.DaemonStatusInitialized,\n\t\tshutdownChan:           make(chan struct{}),\n\t\tackLevel:               shard.GetQueueAckLevel(persistence.HistoryTaskCategoryTransfer).GetTaskID(),\n\t\ttaskAllocator:          taskAllocator,\n\t\tactiveTaskExecutor:     activeTaskExecutor,\n\t\tactiveQueueProcessor:   activeQueueProcessor,\n\t\tstandbyQueueProcessors: standbyQueueProcessors,\n\t}\n}\n\nfunc (t *transferQueueProcessor) Start() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Starting transfer queue processor\")\n\tdefer t.logger.Info(\"Transfer queue processor started\")\n\n\tt.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tt.shutdownWG.Add(1)\n\tgo t.completeTransferLoop()\n}\n\nfunc (t *transferQueueProcessor) Stop() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tif !t.shard.GetConfig().QueueProcessorEnableGracefulSyncShutdown() {\n\t\tt.logger.Info(\"Stopping transfer queue processor non-gracefully\")\n\t\tdefer t.logger.Info(\"Transfer queue processor stopped non-gracefully\")\n\t\tt.activeQueueProcessor.Stop()\n\t\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\tclose(t.shutdownChan)\n\t\tif !common.AwaitWaitGroup(&t.shutdownWG, time.Minute) {\n\t\t\tt.logger.Warn(\"transferQueueProcessor timed out on shut down\", tag.LifeCycleStopTimedout)\n\t\t}\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Stopping transfer queue processor gracefully\")\n\tdefer t.logger.Info(\"Transfer queue processor stopped gracefully\")\n\n\t// close the shutdown channel so processor pump goroutine drains tasks and then stop the processors\n\tclose(t.shutdownChan)\n\tif !common.AwaitWaitGroup(&t.shutdownWG, gracefulShutdownTimeout) {\n\t\tt.logger.Warn(\"transferQueueProcessor timed out on shut down\", tag.LifeCycleStopTimedout)\n\t}\n\tt.activeQueueProcessor.Stop()\n\tfor _, standbyQueueProcessor := range t.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Stop()\n\t}\n\n\tif len(t.failoverQueueProcessors) > 0 {\n\t\tt.logger.Info(\"Shutting down failover transfer queues\", tag.Counter(len(t.failoverQueueProcessors)))\n\t\tfor _, failoverQueueProcessor := range t.failoverQueueProcessors {\n\t\t\tfailoverQueueProcessor.Stop()\n\t\t}\n\t}\n}\n\nfunc (t *transferQueueProcessor) NotifyNewTask(clusterName string, info *hcommon.NotifyTaskInfo) {\n\tif len(info.Tasks) == 0 {\n\t\treturn\n\t}\n\n\tif clusterName == t.currentClusterName {\n\t\tt.activeQueueProcessor.notifyNewTask(info)\n\t\treturn\n\t}\n\n\tstandbyQueueProcessor, ok := t.standbyQueueProcessors[clusterName]\n\tif !ok {\n\t\tpanic(fmt.Sprintf(\"Cannot find transfer processor for %s.\", clusterName))\n\t}\n\tstandbyQueueProcessor.notifyNewTask(info)\n}\n\nfunc (t *transferQueueProcessor) FailoverDomain(domainIDs map[string]struct{}) {\n\tif t.shard.GetConfig().DisableTransferFailoverQueue() {\n\t\treturn\n\t}\n\n\t// Failover queue is used to scan all inflight tasks, if queue processor is not\n\t// started, there's no inflight task and we don't need to create a failover processor.\n\t// Also the HandleAction will be blocked if queue processor processing loop is not running.\n\tif atomic.LoadInt32(&t.status) != common.DaemonStatusStarted {\n\t\treturn\n\t}\n\n\tminLevel := t.shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, t.currentClusterName).GetTaskID()\n\tstandbyClusterName := t.currentClusterName\n\tfor clusterName := range t.shard.GetClusterMetadata().GetEnabledClusterInfo() {\n\t\tackLevel := t.shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, clusterName).GetTaskID()\n\t\tif ackLevel < minLevel {\n\t\t\tminLevel = ackLevel\n\t\t\tstandbyClusterName = clusterName\n\t\t}\n\t}\n\n\tmaxReadLevel := int64(0)\n\tactionResult, err := t.HandleAction(context.Background(), t.currentClusterName, NewGetStateAction())\n\tif err != nil {\n\t\tt.logger.Error(\"Transfer Failover Failed\", tag.WorkflowDomainIDs(domainIDs), tag.Error(err))\n\t\tif err == errProcessorShutdown {\n\t\t\t// processor/shard already shutdown, we don't need to create failover queue processor\n\t\t\treturn\n\t\t}\n\t\t// other errors should never be returned for GetStateAction\n\t\tpanic(fmt.Sprintf(\"unknown error for GetStateAction: %v\", err))\n\t}\n\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\tqueueReadLevel := queueState.ReadLevel().(transferTaskKey).taskID\n\t\tif maxReadLevel < queueReadLevel {\n\t\t\tmaxReadLevel = queueReadLevel\n\t\t}\n\t}\n\t// maxReadLevel is exclusive, so add 1\n\tmaxReadLevel++\n\n\tt.logger.Info(\"Transfer Failover Triggered\",\n\t\ttag.WorkflowDomainIDs(domainIDs),\n\t\ttag.MinLevel(minLevel),\n\t\ttag.MaxLevel(maxReadLevel))\n\n\tupdateShardAckLevel, failoverQueueProcessor := newTransferQueueFailoverProcessor(\n\t\tt.shard,\n\t\tt.taskProcessor,\n\t\tt.taskAllocator,\n\t\tt.activeTaskExecutor,\n\t\tt.logger,\n\t\tminLevel,\n\t\tmaxReadLevel,\n\t\tdomainIDs,\n\t\tstandbyClusterName,\n\t)\n\n\t// NOTE: READ REF BEFORE MODIFICATION\n\t// ref: historyEngine.go registerDomainFailoverCallback function\n\terr = updateShardAckLevel(newTransferTaskKey(minLevel))\n\tif err != nil {\n\t\tt.logger.Error(\"Error update shard ack level\", tag.Error(err))\n\t}\n\n\t// Failover queue processors are started on the fly when domains are failed over.\n\t// Failover queue processors will be stopped when the transfer queue instance is stopped (due to restart or shard movement).\n\t// This means the failover queue processor might not finish its job.\n\t// There is no mechanism to re-start ongoing failover queue processors in the new shard owner.\n\tt.failoverQueueProcessors = append(t.failoverQueueProcessors, failoverQueueProcessor)\n\tfailoverQueueProcessor.Start()\n}\n\nfunc (t *transferQueueProcessor) HandleAction(\n\tctx context.Context,\n\tclusterName string,\n\taction *Action,\n) (*ActionResult, error) {\n\tvar resultNotificationCh chan actionResultNotification\n\tvar added bool\n\tif clusterName == t.currentClusterName {\n\t\tresultNotificationCh, added = t.activeQueueProcessor.addAction(ctx, action)\n\t} else {\n\t\tfound := false\n\t\tfor standbyClusterName, standbyProcessor := range t.standbyQueueProcessors {\n\t\t\tif clusterName == standbyClusterName {\n\t\t\t\tresultNotificationCh, added = standbyProcessor.addAction(ctx, action)\n\t\t\t\tfound = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tif !found {\n\t\t\treturn nil, fmt.Errorf(\"unknown cluster name: %v\", clusterName)\n\t\t}\n\t}\n\n\tif !added {\n\t\tif ctxErr := ctx.Err(); ctxErr != nil {\n\t\t\treturn nil, ctxErr\n\t\t}\n\t\treturn nil, errProcessorShutdown\n\t}\n\n\tselect {\n\tcase resultNotification := <-resultNotificationCh:\n\t\treturn resultNotification.result, resultNotification.err\n\tcase <-t.shutdownChan:\n\t\treturn nil, errProcessorShutdown\n\tcase <-ctx.Done():\n\t\treturn nil, ctx.Err()\n\t}\n}\n\nfunc (t *transferQueueProcessor) LockTaskProcessing() {\n\tt.logger.Debug(\"Transfer queue processor locking task processing\")\n\tt.taskAllocator.Lock()\n}\n\nfunc (t *transferQueueProcessor) UnlockTaskProcessing() {\n\tt.logger.Debug(\"Transfer queue processor unlocking task processing\")\n\tt.taskAllocator.Unlock()\n}\n\nfunc (t *transferQueueProcessor) drain() {\n\t// before shutdown, make sure the ack level is up to date\n\tif err := t.completeTransfer(); err != nil {\n\t\tt.logger.Error(\"Failed to complete transfer task during shutdown\", tag.Error(err))\n\t}\n}\n\nfunc (t *transferQueueProcessor) completeTransferLoop() {\n\tdefer t.shutdownWG.Done()\n\n\tt.logger.Info(\"Transfer queue processor completeTransferLoop\")\n\tdefer t.logger.Info(\"Transfer queue processor completeTransferLoop completed\")\n\n\tcompleteTimer := time.NewTimer(t.config.TransferProcessorCompleteTransferInterval())\n\tdefer completeTimer.Stop()\n\n\t// Create a retryTimer once, and reset it as needed\n\tretryTimer := time.NewTimer(0)\n\tdefer retryTimer.Stop()\n\t// Stop it immediately because we don't want it to fire initially\n\tif !retryTimer.Stop() {\n\t\t<-retryTimer.C\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownChan:\n\t\t\tt.drain()\n\t\t\treturn\n\t\tcase <-completeTimer.C:\n\t\t\tfor attempt := 0; attempt < t.config.TransferProcessorCompleteTransferFailureRetryCount(); attempt++ {\n\t\t\t\terr := t.completeTransfer()\n\t\t\t\tif err == nil {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\n\t\t\t\tt.logger.Error(\"Failed to complete transfer task\", tag.Error(err), tag.Attempt(int32(attempt)))\n\t\t\t\tvar errShardClosed *shard.ErrShardClosed\n\t\t\t\tif errors.As(err, &errShardClosed) {\n\t\t\t\t\t// shard closed, trigger shutdown and bail out\n\t\t\t\t\tif !t.shard.GetConfig().QueueProcessorEnableGracefulSyncShutdown() {\n\t\t\t\t\t\tgo t.Stop()\n\t\t\t\t\t\treturn\n\t\t\t\t\t}\n\n\t\t\t\t\tt.Stop()\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\t// Reset the retryTimer for the delay between attempts\n\t\t\t\t// TODO: the first retry has 0 backoff, revisit it to see if it's expected\n\t\t\t\tretryDuration := time.Duration(attempt*100) * time.Millisecond\n\t\t\t\tretryTimer.Reset(retryDuration)\n\t\t\t\tselect {\n\t\t\t\tcase <-t.shutdownChan:\n\t\t\t\t\tt.drain()\n\t\t\t\t\treturn\n\t\t\t\tcase <-retryTimer.C:\n\t\t\t\t\t// do nothing. retry loop will continue\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tcompleteTimer.Reset(t.config.TransferProcessorCompleteTransferInterval())\n\t\t}\n\t}\n}\n\nfunc (t *transferQueueProcessor) completeTransfer() error {\n\tnewAckLevel := maximumTransferTaskKey\n\tactionResult, err := t.HandleAction(context.Background(), t.currentClusterName, NewGetStateAction())\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\tnewAckLevel = minTaskKey(newAckLevel, queueState.AckLevel())\n\t}\n\n\tfor standbyClusterName := range t.standbyQueueProcessors {\n\t\tactionResult, err := t.HandleAction(context.Background(), standbyClusterName, NewGetStateAction())\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfor _, queueState := range actionResult.GetStateActionResult.States {\n\t\t\tnewAckLevel = minTaskKey(newAckLevel, queueState.AckLevel())\n\t\t}\n\t}\n\n\tfor _, failoverInfo := range t.shard.GetAllFailoverLevels(persistence.HistoryTaskCategoryTransfer) {\n\t\tfailoverLevel := newTransferTaskKey(failoverInfo.MinLevel.GetTaskID())\n\t\tif newAckLevel == nil {\n\t\t\tnewAckLevel = failoverLevel\n\t\t} else {\n\t\t\tnewAckLevel = minTaskKey(newAckLevel, failoverLevel)\n\t\t}\n\t}\n\n\tif newAckLevel == nil {\n\t\tpanic(\"Unable to get transfer queue processor ack level\")\n\t}\n\n\tnewAckLevelTaskID := newAckLevel.(transferTaskKey).taskID\n\tt.logger.Debugf(\"Start completing transfer task from: %v, to %v.\", t.ackLevel, newAckLevelTaskID)\n\tif t.ackLevel >= newAckLevelTaskID {\n\t\treturn nil\n\t}\n\n\tt.metricsClient.Scope(metrics.TransferQueueProcessorScope).\n\t\tTagged(metrics.ShardIDTag(t.shard.GetShardID())).\n\t\tIncCounter(metrics.TaskBatchCompleteCounter)\n\n\tfor {\n\t\tpageSize := t.config.TransferTaskDeleteBatchSize()\n\t\t// we're switching from exclusive begin/end to inclusive min/exclusive max,\n\t\t// so we need to adjust the taskID, for example, if the original range is (1, 10],\n\t\t// the new range should be [2, 11), so we add 1 to both the min and max taskID\n\t\tresp, err := t.shard.GetExecutionManager().RangeCompleteHistoryTask(context.Background(), &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(t.ackLevel + 1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(newAckLevelTaskID + 1),\n\t\t\tPageSize:            pageSize,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, pageSize) {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tt.ackLevel = newAckLevelTaskID\n\n\treturn t.shard.UpdateQueueAckLevel(persistence.HistoryTaskCategoryTransfer, persistence.NewImmediateTaskKey(newAckLevelTaskID))\n}\n\nfunc newTransferQueueActiveProcessor(\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n) *transferQueueProcessorBase {\n\tconfig := shard.GetConfig()\n\toptions := newTransferQueueProcessorOptions(config, true, false)\n\n\tcurrentClusterName := shard.GetClusterMetadata().GetCurrentClusterName()\n\tlogger = logger.WithTags(tag.ClusterName(currentClusterName))\n\n\ttaskFilter := func(task persistence.Task) (bool, error) {\n\t\tif task.GetTaskCategory() != persistence.HistoryTaskCategoryTransfer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shard, task.GetDomainID()); notRegistered && err == nil {\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in active transfer queue.\", tag.WorkflowDomainID(task.GetDomainID()), tag.Value(task))\n\t\t\treturn false, nil\n\t\t}\n\t\treturn taskAllocator.VerifyActiveTask(task.GetDomainID(), task.GetWorkflowID(), task.GetRunID(), task)\n\t}\n\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(shard.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTransfer, currentClusterName).GetTaskID())\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\ttaskID := ackLevel.(transferTaskKey).taskID\n\t\treturn shard.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, currentClusterName, persistence.NewImmediateTaskKey(taskID))\n\t}\n\n\tupdateProcessingQueueStates := func(states []ProcessingQueueState) error {\n\t\tpStates := convertToPersistenceTransferProcessingQueueStates(states)\n\t\treturn shard.UpdateTransferProcessingQueueStates(currentClusterName, pStates)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn nil\n\t}\n\n\treturn newTransferQueueProcessorBase(\n\t\tshard,\n\t\tloadTransferProcessingQueueStates(currentClusterName, shard, options, logger),\n\t\ttaskProcessor,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t)\n}\n\nfunc newTransferQueueStandbyProcessor(\n\tclusterName string,\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n) *transferQueueProcessorBase {\n\tconfig := shard.GetConfig()\n\toptions := newTransferQueueProcessorOptions(config, false, false)\n\n\tlogger = logger.WithTags(tag.ClusterName(clusterName))\n\n\ttaskFilter := func(task persistence.Task) (bool, error) {\n\t\tif task.GetTaskCategory() != persistence.HistoryTaskCategoryTransfer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shard, task.GetDomainID()); notRegistered && err == nil {\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in standby transfer queue.\", tag.WorkflowDomainID(task.GetDomainID()), tag.Value(task))\n\t\t\treturn false, nil\n\t\t}\n\t\tif task.GetTaskType() == persistence.TransferTaskTypeCloseExecution ||\n\t\t\ttask.GetTaskType() == persistence.TransferTaskTypeRecordWorkflowClosed {\n\t\t\tdomainEntry, err := shard.GetDomainCache().GetDomainByID(task.GetDomainID())\n\t\t\tif err == nil {\n\t\t\t\tif domainEntry.HasReplicationCluster(clusterName) {\n\t\t\t\t\t// guarantee the processing of workflow execution close\n\t\t\t\t\treturn true, nil\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\t\t\t// retry the task if failed to find the domain\n\t\t\t\t\tlogger.Warn(\"Cannot find domain\", tag.WorkflowDomainID(task.GetDomainID()))\n\t\t\t\t\treturn false, err\n\t\t\t\t}\n\t\t\t\tlogger.Warn(\"Cannot find domain, default to not process task.\", tag.WorkflowDomainID(task.GetDomainID()), tag.Value(task))\n\t\t\t\treturn false, nil\n\t\t\t}\n\t\t}\n\t\treturn taskAllocator.VerifyStandbyTask(clusterName, task.GetDomainID(), task.GetWorkflowID(), task.GetRunID(), task)\n\t}\n\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(shard.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTransfer, clusterName).GetTaskID())\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\ttaskID := ackLevel.(transferTaskKey).taskID\n\t\treturn shard.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, clusterName, persistence.NewImmediateTaskKey(taskID))\n\t}\n\n\tupdateProcessingQueueStates := func(states []ProcessingQueueState) error {\n\t\tpStates := convertToPersistenceTransferProcessingQueueStates(states)\n\t\treturn shard.UpdateTransferProcessingQueueStates(clusterName, pStates)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn nil\n\t}\n\n\treturn newTransferQueueProcessorBase(\n\t\tshard,\n\t\tloadTransferProcessingQueueStates(clusterName, shard, options, logger),\n\t\ttaskProcessor,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t)\n}\n\nfunc newTransferQueueFailoverProcessor(\n\tshardContext shard.Context,\n\ttaskProcessor task.Processor,\n\ttaskAllocator TaskAllocator,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tminLevel, maxLevel int64,\n\tdomainIDs map[string]struct{},\n\tstandbyClusterName string,\n) (updateClusterAckLevelFn, *transferQueueProcessorBase) {\n\tconfig := shardContext.GetConfig()\n\toptions := newTransferQueueProcessorOptions(config, true, true)\n\n\tcurrentClusterName := shardContext.GetService().GetClusterMetadata().GetCurrentClusterName()\n\tfailoverUUID := uuid.New()\n\tlogger = logger.WithTags(\n\t\ttag.ClusterName(currentClusterName),\n\t\ttag.WorkflowDomainIDs(domainIDs),\n\t\ttag.FailoverMsg(\"from: \"+standbyClusterName),\n\t)\n\n\ttaskFilter := func(task persistence.Task) (bool, error) {\n\t\tif task.GetTaskCategory() != persistence.HistoryTaskCategoryTransfer {\n\t\t\treturn false, errUnexpectedQueueTask\n\t\t}\n\t\tif notRegistered, err := isDomainNotRegistered(shardContext, task.GetDomainID()); notRegistered && err == nil {\n\t\t\tlogger.Info(\"Domain is not in registered status, skip task in failover transfer queue.\", tag.WorkflowDomainID(task.GetDomainID()), tag.Value(task))\n\t\t\treturn false, nil\n\t\t}\n\t\treturn taskAllocator.VerifyFailoverActiveTask(domainIDs, task.GetDomainID(), task.GetWorkflowID(), task.GetRunID(), task)\n\t}\n\n\tmaxReadLevelTaskKey := newTransferTaskKey(maxLevel)\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn maxReadLevelTaskKey // this is a const\n\t}\n\n\tupdateClusterAckLevel := func(ackLevel task.Key) error {\n\t\ttaskID := ackLevel.(transferTaskKey).taskID\n\t\treturn shardContext.UpdateFailoverLevel(\n\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\tfailoverUUID,\n\t\t\tpersistence.FailoverLevel{\n\t\t\t\tStartTime:    shardContext.GetTimeSource().Now(),\n\t\t\t\tMinLevel:     persistence.NewImmediateTaskKey(minLevel),\n\t\t\t\tCurrentLevel: persistence.NewImmediateTaskKey(taskID),\n\t\t\t\tMaxLevel:     persistence.NewImmediateTaskKey(maxLevel),\n\t\t\t\tDomainIDs:    domainIDs,\n\t\t\t},\n\t\t)\n\t}\n\n\tqueueShutdown := func() error {\n\t\treturn shardContext.DeleteFailoverLevel(\n\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\tfailoverUUID,\n\t\t)\n\t}\n\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tdefaultProcessingQueueLevel,\n\t\t\tnewTransferTaskKey(minLevel),\n\t\t\tmaxReadLevelTaskKey,\n\t\t\tNewDomainFilter(domainIDs, false),\n\t\t),\n\t}\n\n\treturn updateClusterAckLevel, newTransferQueueProcessorBase(\n\t\tshardContext,\n\t\tprocessingQueueStates,\n\t\ttaskProcessor,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tnil,\n\t\tqueueShutdown,\n\t\ttaskFilter,\n\t\ttaskExecutor,\n\t\tlogger,\n\t\tshardContext.GetMetricsClient(),\n\t)\n}\n\nfunc loadTransferProcessingQueueStates(\n\tclusterName string,\n\tshard shard.Context,\n\toptions *queueProcessorOptions,\n\tlogger log.Logger,\n) []ProcessingQueueState {\n\tackLevel := shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, clusterName).GetTaskID()\n\tif options.EnableLoadQueueStates() {\n\t\tpStates := shard.GetTransferProcessingQueueStates(clusterName)\n\t\tif validateProcessingQueueStates(pStates, ackLevel) {\n\t\t\treturn convertFromPersistenceTransferProcessingQueueStates(pStates)\n\t\t}\n\n\t\tlogger.Error(\"Incompatible processing queue states and ackLevel\",\n\t\t\ttag.Value(pStates),\n\t\t\ttag.ShardTransferAcks(ackLevel),\n\t\t)\n\t}\n\n\t// LoadQueueStates is disabled or sanity check failed\n\t// fallback to use ackLevel\n\treturn []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tdefaultProcessingQueueLevel,\n\t\t\tnewTransferTaskKey(ackLevel),\n\t\t\tmaximumTransferTaskKey,\n\t\t\tNewDomainFilter(nil, true),\n\t\t),\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_processor_base.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\tnumTasksEstimationDecay = 0.6\n)\n\nvar (\n\tloadQueueTaskThrottleRetryDelay = 5 * time.Second\n\tpersistenceOperationRetryPolicy = common.CreatePersistenceRetryPolicy()\n)\n\ntype (\n\ttransferTaskKey struct {\n\t\ttaskID int64\n\t}\n\n\ttransferQueueProcessorBase struct {\n\t\t*processorBase\n\n\t\ttaskInitializer task.Initializer\n\n\t\tnotifyCh  chan struct{}\n\t\tprocessCh chan struct{}\n\n\t\t// for managing if a processing queue collection should be processed\n\t\tstartJitterTimer *time.Timer\n\t\tprocessingLock   sync.Mutex\n\t\tbackoffTimer     map[int]*time.Timer\n\t\tshouldProcess    map[int]bool\n\n\t\t// for estimating the look ahead taskID during split\n\t\tlastSplitTime           time.Time\n\t\tlastMaxReadLevel        int64\n\t\testimatedTasksPerMinute int64\n\n\t\t// for validating if the queue failed to load any tasks\n\t\tvalidator *transferQueueValidator\n\n\t\tprocessQueueCollectionsFn func()\n\t\tupdateAckLevelFn          func() (bool, task.Key, error)\n\t}\n)\n\nfunc newTransferQueueProcessorBase(\n\tshard shard.Context,\n\tprocessingQueueStates []ProcessingQueueState,\n\ttaskProcessor task.Processor,\n\toptions *queueProcessorOptions,\n\tupdateMaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\tqueueShutdown queueShutdownFn,\n\ttaskFilter task.Filter,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) *transferQueueProcessorBase {\n\tprocessorBase := newProcessorBase(\n\t\tshard,\n\t\tprocessingQueueStates,\n\t\ttaskProcessor,\n\t\toptions,\n\t\tupdateMaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\tqueueShutdown,\n\t\tlogger.WithTags(tag.ComponentTransferQueue),\n\t\tmetricsClient,\n\t)\n\n\tqueueType := task.QueueTypeActiveTransfer\n\tif options.MetricScope == metrics.TransferStandbyQueueProcessorScope {\n\t\tqueueType = task.QueueTypeStandbyTransfer\n\t}\n\n\ttransferQueueProcessorBase := &transferQueueProcessorBase{\n\t\tprocessorBase: processorBase,\n\t\ttaskInitializer: func(taskInfo persistence.Task) task.Task {\n\t\t\treturn task.NewHistoryTask(\n\t\t\t\tshard,\n\t\t\t\ttaskInfo,\n\t\t\t\tqueueType,\n\t\t\t\ttask.InitializeLoggerForTask(shard.GetShardID(), taskInfo, logger),\n\t\t\t\ttaskFilter,\n\t\t\t\ttaskExecutor,\n\t\t\t\ttaskProcessor,\n\t\t\t\tprocessorBase.redispatcher,\n\t\t\t\tshard.GetConfig().TaskCriticalRetryCount,\n\t\t\t)\n\t\t},\n\t\tnotifyCh:         make(chan struct{}, 1),\n\t\tprocessCh:        make(chan struct{}, 1),\n\t\tbackoffTimer:     make(map[int]*time.Timer),\n\t\tshouldProcess:    make(map[int]bool),\n\t\tlastSplitTime:    time.Time{},\n\t\tlastMaxReadLevel: 0,\n\t}\n\n\ttransferQueueProcessorBase.processQueueCollectionsFn = transferQueueProcessorBase.processQueueCollections\n\ttransferQueueProcessorBase.updateAckLevelFn = transferQueueProcessorBase.updateAckLevel\n\n\tif shard.GetConfig().EnableDebugMode && options.EnableValidator() {\n\t\ttransferQueueProcessorBase.validator = newTransferQueueValidator(\n\t\t\ttransferQueueProcessorBase,\n\t\t\toptions.ValidationInterval,\n\t\t\tlogger,\n\t\t\tprocessorBase.metricsScope,\n\t\t)\n\t}\n\n\treturn transferQueueProcessorBase\n}\n\nfunc (t *transferQueueProcessorBase) Start() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Transfer queue processor state changed\", tag.LifeCycleStarting)\n\tdefer t.logger.Info(\"Transfer queue processor state changed\", tag.LifeCycleStarted)\n\n\tt.redispatcher.Start()\n\n\t// trigger an initial (maybe delayed) load of tasks\n\tif startJitter := t.options.MaxStartJitterInterval(); startJitter > 0 {\n\t\tt.startJitterTimer = time.AfterFunc(\n\t\t\ttime.Duration(rand.Int63n(int64(startJitter))),\n\t\t\tfunc() {\n\t\t\t\tt.notifyAllQueueCollections()\n\t\t\t},\n\t\t)\n\t} else {\n\t\tt.notifyAllQueueCollections()\n\t}\n\n\tt.shutdownWG.Add(1)\n\tgo t.processorPump()\n}\n\nfunc (t *transferQueueProcessorBase) Stop() {\n\tif !atomic.CompareAndSwapInt32(&t.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tt.logger.Info(\"Transfer queue processor state changed\", tag.LifeCycleStopping)\n\n\tclose(t.shutdownCh)\n\tif t.startJitterTimer != nil {\n\t\tt.startJitterTimer.Stop()\n\t}\n\tt.processingLock.Lock()\n\tfor _, timer := range t.backoffTimer {\n\t\ttimer.Stop()\n\t}\n\tfor level := range t.shouldProcess {\n\t\tt.shouldProcess[level] = false\n\t}\n\tt.processingLock.Unlock()\n\n\tif success := common.AwaitWaitGroup(&t.shutdownWG, gracefulShutdownTimeout); !success {\n\t\tt.logger.Warn(\"transferQueueProcessorBase timed out on shut down\", tag.LifeCycleStopTimedout)\n\t}\n\n\tt.redispatcher.Stop()\n\tt.logger.Info(\"Transfer queue processor state changed\", tag.LifeCycleStopped)\n}\n\nfunc (t *transferQueueProcessorBase) notifyNewTask(info *hcommon.NotifyTaskInfo) {\n\tselect {\n\tcase t.notifyCh <- struct{}{}:\n\tdefault:\n\t}\n\n\tif info.ExecutionInfo != nil && t.validator != nil {\n\t\t// executionInfo will be nil when notifyNewTask is called to trigger a scan, for example during domain failover or sync shard.\n\t\tt.validator.addTasks(info)\n\t}\n}\n\nfunc (t *transferQueueProcessorBase) readyForProcess(level int) {\n\tt.processingLock.Lock()\n\tdefer t.processingLock.Unlock()\n\n\tif _, ok := t.backoffTimer[level]; ok {\n\t\t// current level is being throttled\n\t\treturn\n\t}\n\n\tt.shouldProcess[level] = true\n\n\t// trigger the actual processing\n\tselect {\n\tcase t.processCh <- struct{}{}:\n\tdefault:\n\t}\n}\n\nfunc (t *transferQueueProcessorBase) setupBackoffTimer(level int) {\n\tt.processingLock.Lock()\n\tdefer t.processingLock.Unlock()\n\n\tif _, ok := t.backoffTimer[level]; ok {\n\t\t// there's an existing backoff timer, no-op\n\t\t// this case should not happen\n\t\treturn\n\t}\n\n\tt.metricsScope.IncCounter(metrics.ProcessingQueueThrottledCounter)\n\tt.logger.Info(\"Throttled processing queue\", tag.QueueLevel(level))\n\n\tbackoffDuration := backoff.JitDuration(\n\t\tt.options.PollBackoffInterval(),\n\t\tt.options.PollBackoffIntervalJitterCoefficient(),\n\t)\n\tt.backoffTimer[level] = time.AfterFunc(backoffDuration, func() {\n\t\tselect {\n\t\tcase <-t.shutdownCh:\n\t\t\treturn\n\t\tdefault:\n\t\t}\n\n\t\tt.processingLock.Lock()\n\t\tdefer t.processingLock.Unlock()\n\n\t\tt.shouldProcess[level] = true\n\t\tdelete(t.backoffTimer, level)\n\n\t\t// trigger the actual processing\n\t\tselect {\n\t\tcase t.processCh <- struct{}{}:\n\t\tdefault:\n\t\t}\n\t})\n}\n\nfunc (t *transferQueueProcessorBase) processorPump() {\n\tdefer t.shutdownWG.Done()\n\n\tupdateAckTimer := time.NewTimer(backoff.JitDuration(\n\t\tt.options.UpdateAckInterval(),\n\t\tt.options.UpdateAckIntervalJitterCoefficient(),\n\t))\n\tdefer updateAckTimer.Stop()\n\n\tsplitQueueTimer := time.NewTimer(backoff.JitDuration(\n\t\tt.options.SplitQueueInterval(),\n\t\tt.options.SplitQueueIntervalJitterCoefficient(),\n\t))\n\tdefer splitQueueTimer.Stop()\n\n\tmaxPollTimer := time.NewTimer(backoff.JitDuration(\n\t\tt.options.MaxPollInterval(),\n\t\tt.options.MaxPollIntervalJitterCoefficient(),\n\t))\n\tdefer maxPollTimer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-t.shutdownCh:\n\t\t\treturn\n\t\tcase <-t.notifyCh:\n\t\t\t// notify all queue collections as they are waiting for the notification when there's\n\t\t\t// no more task to process. For non-default queue, if we choose to do periodic polling\n\t\t\t// in the future, then we don't need to notify them.\n\t\t\tt.notifyAllQueueCollections()\n\t\tcase <-maxPollTimer.C:\n\t\t\tt.notifyAllQueueCollections()\n\t\t\tmaxPollTimer.Reset(backoff.JitDuration(\n\t\t\t\tt.options.MaxPollInterval(),\n\t\t\t\tt.options.MaxPollIntervalJitterCoefficient(),\n\t\t\t))\n\t\tcase <-t.processCh:\n\t\t\tmaxRedispatchQueueSize := t.options.MaxRedispatchQueueSize()\n\t\t\tif redispathSize := t.redispatcher.Size(); redispathSize > maxRedispatchQueueSize {\n\t\t\t\tt.logger.Debugf(\"Transfer queue has too many pending tasks in re-dispatch queue: %v > maxRedispatchQueueSize: %v, block loading tasks from persistence\", redispathSize, maxRedispatchQueueSize)\n\t\t\t\tt.redispatcher.Redispatch(maxRedispatchQueueSize)\n\t\t\t\tif redispathSize := t.redispatcher.Size(); redispathSize > maxRedispatchQueueSize {\n\t\t\t\t\t// if redispatcher still has a large number of tasks\n\t\t\t\t\t// this only happens when system is under very high load\n\t\t\t\t\t// we should backoff here instead of keeping submitting tasks to task processor\n\t\t\t\t\tt.logger.Debugf(\"Transfer queue still has too many pending tasks in re-dispatch queue: %v > maxRedispatchQueueSize: %v, backing off for %v\", redispathSize, maxRedispatchQueueSize, t.options.PollBackoffInterval())\n\t\t\t\t\ttime.Sleep(backoff.JitDuration(\n\t\t\t\t\t\tt.options.PollBackoffInterval(),\n\t\t\t\t\t\tt.options.PollBackoffIntervalJitterCoefficient(),\n\t\t\t\t\t))\n\t\t\t\t}\n\t\t\t\t// re-enqueue the event to see if we need keep re-dispatching or load new tasks from persistence\n\t\t\t\tselect {\n\t\t\t\tcase t.processCh <- struct{}{}:\n\t\t\t\tdefault:\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tt.processQueueCollectionsFn()\n\t\t\t}\n\t\tcase <-updateAckTimer.C:\n\t\t\tprocessFinished, _, err := t.updateAckLevelFn()\n\t\t\tvar errShardClosed *shard.ErrShardClosed\n\t\t\tif errors.As(err, &errShardClosed) || (err == nil && processFinished) {\n\t\t\t\tif !t.options.EnableGracefulSyncShutdown() {\n\t\t\t\t\tgo t.Stop()\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\tt.Stop()\n\t\t\t\treturn\n\t\t\t}\n\t\t\tupdateAckTimer.Reset(backoff.JitDuration(\n\t\t\t\tt.options.UpdateAckInterval(),\n\t\t\t\tt.options.UpdateAckIntervalJitterCoefficient(),\n\t\t\t))\n\t\tcase <-splitQueueTimer.C:\n\t\t\tt.splitQueue()\n\t\t\tsplitQueueTimer.Reset(backoff.JitDuration(\n\t\t\t\tt.options.SplitQueueInterval(),\n\t\t\t\tt.options.SplitQueueIntervalJitterCoefficient(),\n\t\t\t))\n\t\tcase notification := <-t.actionNotifyCh:\n\t\t\tt.handleActionNotification(notification)\n\t\t}\n\t}\n}\n\nfunc (t *transferQueueProcessorBase) notifyAllQueueCollections() {\n\tfor _, queueCollection := range t.processingQueueCollections {\n\t\tt.readyForProcess(queueCollection.Level())\n\t}\n}\n\nfunc (t *transferQueueProcessorBase) processQueueCollections() {\n\tfor _, queueCollection := range t.processingQueueCollections {\n\t\tlevel := queueCollection.Level()\n\t\tt.processingLock.Lock()\n\t\tif shouldProcess, ok := t.shouldProcess[level]; !ok || !shouldProcess {\n\t\t\tt.processingLock.Unlock()\n\t\t\tcontinue\n\t\t}\n\t\tt.shouldProcess[level] = false\n\t\tt.processingLock.Unlock()\n\n\t\tactiveQueue := queueCollection.ActiveQueue()\n\t\tif activeQueue == nil {\n\t\t\t// process for this queue collection has finished\n\t\t\t// it's possible that new queue will be added to this collection later though,\n\t\t\t// pollTime will be updated after split/merge\n\t\t\tcontinue\n\t\t}\n\n\t\treadLevel := activeQueue.State().ReadLevel()\n\t\tmaxReadLevel := minTaskKey(activeQueue.State().MaxLevel(), t.updateMaxReadLevel())\n\t\tdomainFilter := activeQueue.State().DomainFilter()\n\n\t\tif !readLevel.Less(maxReadLevel) {\n\t\t\t// no task need to be processed for now, wait for new task notification\n\t\t\t// note that if taskID for new task is still less than readLevel, the notification\n\t\t\t// will just be a no-op and there's no DB requests.\n\t\t\tcontinue\n\t\t}\n\n\t\tctx, cancel := context.WithTimeout(context.Background(), loadQueueTaskThrottleRetryDelay)\n\t\tif err := t.rateLimiter.Wait(ctx); err != nil {\n\t\t\tcancel()\n\t\t\tif level != defaultProcessingQueueLevel {\n\t\t\t\tt.setupBackoffTimer(level)\n\t\t\t} else {\n\t\t\t\tt.readyForProcess(level)\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\t\tcancel()\n\n\t\ttransferTaskInfos, more, err := t.readTasks(readLevel, maxReadLevel)\n\t\tif err != nil {\n\t\t\tt.logger.Error(\"Processor unable to retrieve tasks\", tag.Error(err))\n\t\t\tt.readyForProcess(level) // re-enqueue the event\n\t\t\tcontinue\n\t\t}\n\t\tt.logger.Debug(\"load transfer tasks from database\",\n\t\t\ttag.ReadLevel(readLevel.(transferTaskKey).taskID),\n\t\t\ttag.MaxLevel(maxReadLevel.(transferTaskKey).taskID),\n\t\t\ttag.Counter(len(transferTaskInfos)))\n\n\t\ttasks := make(map[task.Key]task.Task)\n\t\ttaskChFull := false\n\t\tnow := t.shard.GetTimeSource().Now()\n\t\tfor _, taskInfo := range transferTaskInfos {\n\t\t\tif !domainFilter.Filter(taskInfo.GetDomainID()) {\n\t\t\t\tt.logger.Debug(\"transfer task filtered\", tag.TaskID(taskInfo.GetTaskID()))\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tif persistence.IsTaskCorrupted(taskInfo) {\n\t\t\t\tt.logger.Error(\"Processing queue encountered a corrupted task\", tag.Dynamic(\"task\", taskInfo))\n\t\t\t\tt.metricsScope.IncCounter(metrics.CorruptedHistoryTaskCounter)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\ttask := t.taskInitializer(taskInfo)\n\t\t\ttasks[newTransferTaskKey(taskInfo.GetTaskID())] = task\n\t\t\tt.metricsScope.RecordHistogramDuration(metrics.TaskEnqueueToFetchLatency, now.Sub(taskInfo.GetVisibilityTimestamp()))\n\t\t\tsubmitted, err := t.submitTask(task)\n\t\t\tif err != nil {\n\t\t\t\t// only err here is due to the fact that processor has been shutdown\n\t\t\t\t// return instead of continue\n\t\t\t\treturn\n\t\t\t}\n\t\t\ttaskChFull = taskChFull || !submitted\n\t\t}\n\n\t\tvar newReadLevel task.Key\n\t\tif !more {\n\t\t\tnewReadLevel = maxReadLevel\n\t\t} else {\n\t\t\tnewReadLevel = newTransferTaskKey(transferTaskInfos[len(transferTaskInfos)-1].GetTaskID())\n\t\t}\n\t\tqueueCollection.AddTasks(tasks, newReadLevel)\n\t\tif t.validator != nil {\n\t\t\tt.logger.Debug(\"ack transfer tasks\",\n\t\t\t\ttag.ReadLevel(readLevel.(transferTaskKey).taskID),\n\t\t\t\ttag.MaxLevel(newReadLevel.(transferTaskKey).taskID),\n\t\t\t\ttag.Counter(len(tasks)))\n\t\t\tt.validator.ackTasks(level, readLevel, newReadLevel, tasks)\n\t\t}\n\n\t\tnewActiveQueue := queueCollection.ActiveQueue()\n\t\tif more || (newActiveQueue != nil && newActiveQueue != activeQueue) {\n\t\t\t// more tasks for the current active queue or the active queue has changed\n\t\t\tif level != defaultProcessingQueueLevel && taskChFull {\n\t\t\t\tt.setupBackoffTimer(level)\n\t\t\t} else {\n\t\t\t\tt.readyForProcess(level)\n\t\t\t}\n\t\t}\n\n\t\t// else it means we don't have tasks to process for now\n\t\t// wait for new task notification\n\t\t// another option for non-default queue is that we can setup a backoff timer to check back later\n\t}\n}\n\nfunc (t *transferQueueProcessorBase) splitQueue() {\n\tcurrentTime := t.shard.GetTimeSource().Now()\n\tcurrentMaxReadLevel := t.updateMaxReadLevel().(transferTaskKey).taskID\n\tdefer func() {\n\t\tt.lastSplitTime = currentTime\n\t\tt.lastMaxReadLevel = currentMaxReadLevel\n\t}()\n\n\tif currentMaxReadLevel-t.lastMaxReadLevel < 2<<(t.shard.GetConfig().RangeSizeBits-1) {\n\t\t// only update the estimation when rangeID is not renewed\n\t\t// note the threshold here is only an estimation. If the read level increased too much\n\t\t// we will drop that data point.\n\t\tnumTasksPerMinute := (currentMaxReadLevel - t.lastMaxReadLevel) / int64(currentTime.Sub(t.lastSplitTime).Seconds()) * int64(time.Minute.Seconds())\n\n\t\tif t.estimatedTasksPerMinute == 0 {\n\t\t\t// set the initial value for the estimation\n\t\t\tt.estimatedTasksPerMinute = numTasksPerMinute\n\t\t} else {\n\t\t\tt.estimatedTasksPerMinute = int64(numTasksEstimationDecay*float64(t.estimatedTasksPerMinute) + (1-numTasksEstimationDecay)*float64(numTasksPerMinute))\n\t\t}\n\t}\n\n\tif t.lastSplitTime.IsZero() || t.estimatedTasksPerMinute == 0 {\n\t\t// skip the split as we can't estimate the look ahead taskID\n\t\treturn\n\t}\n\n\tsplitPolicy := t.initializeSplitPolicy(\n\t\tfunc(key task.Key, domainID string) task.Key {\n\t\t\ttotalLookAhead := t.estimatedTasksPerMinute * int64(t.options.SplitLookAheadDurationByDomainID(domainID).Minutes())\n\t\t\t// ensure the above calculation doesn't overflow and cap the maximun look ahead interval\n\t\t\ttotalLookAhead = max(min(totalLookAhead, 2<<t.shard.GetConfig().RangeSizeBits), 0)\n\t\t\treturn newTransferTaskKey(key.(transferTaskKey).taskID + totalLookAhead)\n\t\t},\n\t)\n\n\tt.splitProcessingQueueCollection(splitPolicy, func(level int, _ time.Time) {\n\t\tt.readyForProcess(level)\n\t})\n}\n\nfunc (t *transferQueueProcessorBase) handleActionNotification(notification actionNotification) {\n\tt.processorBase.handleActionNotification(notification, func() {\n\t\tswitch notification.action.ActionType {\n\t\tcase ActionTypeReset:\n\t\t\tt.readyForProcess(defaultProcessingQueueLevel)\n\t\t}\n\t})\n}\n\nfunc (t *transferQueueProcessorBase) readTasks(\n\treadLevel task.Key,\n\tmaxReadLevel task.Key,\n) ([]persistence.Task, bool, error) {\n\n\tvar response *persistence.GetHistoryTasksResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresponse, err = t.shard.GetExecutionManager().GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(readLevel.(transferTaskKey).taskID + 1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxReadLevel.(transferTaskKey).taskID + 1),\n\t\t\tPageSize:            t.options.BatchSize(),\n\t\t})\n\t\treturn err\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(persistenceOperationRetryPolicy),\n\t\tbackoff.WithRetryableError(persistence.IsBackgroundTransientError),\n\t)\n\terr := throttleRetry.Do(context.Background(), op)\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\n\treturn response.Tasks, len(response.NextPageToken) != 0, nil\n}\n\nfunc newTransferTaskKey(taskID int64) task.Key {\n\treturn transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n}\n\nfunc (k transferTaskKey) Less(\n\tkey task.Key,\n) bool {\n\treturn k.taskID < key.(transferTaskKey).taskID\n}\n\nfunc newTransferQueueProcessorOptions(\n\tconfig *config.Config,\n\tisActive bool,\n\tisFailover bool,\n) *queueProcessorOptions {\n\toptions := &queueProcessorOptions{\n\t\tBatchSize:                            config.TransferTaskBatchSize,\n\t\tDeleteBatchSize:                      config.TransferTaskDeleteBatchSize,\n\t\tMaxPollRPS:                           config.TransferProcessorMaxPollRPS,\n\t\tMaxPollInterval:                      config.TransferProcessorMaxPollInterval,\n\t\tMaxPollIntervalJitterCoefficient:     config.TransferProcessorMaxPollIntervalJitterCoefficient,\n\t\tUpdateAckInterval:                    config.TransferProcessorUpdateAckInterval,\n\t\tUpdateAckIntervalJitterCoefficient:   config.TransferProcessorUpdateAckIntervalJitterCoefficient,\n\t\tMaxRedispatchQueueSize:               config.TransferProcessorMaxRedispatchQueueSize,\n\t\tSplitQueueInterval:                   config.TransferProcessorSplitQueueInterval,\n\t\tSplitQueueIntervalJitterCoefficient:  config.TransferProcessorSplitQueueIntervalJitterCoefficient,\n\t\tPollBackoffInterval:                  config.QueueProcessorPollBackoffInterval,\n\t\tPollBackoffIntervalJitterCoefficient: config.QueueProcessorPollBackoffIntervalJitterCoefficient,\n\t\tEnableValidator:                      config.TransferProcessorEnableValidator,\n\t\tValidationInterval:                   config.TransferProcessorValidationInterval,\n\t\tEnableGracefulSyncShutdown:           config.QueueProcessorEnableGracefulSyncShutdown,\n\t}\n\n\tif isFailover {\n\t\t// disable queue split for failover processor\n\t\toptions.EnableSplit = dynamicproperties.GetBoolPropertyFn(false)\n\n\t\t// disable persist and load processing queue states for failover processor as it will never be split\n\t\toptions.EnablePersistQueueStates = dynamicproperties.GetBoolPropertyFn(false)\n\t\toptions.EnableLoadQueueStates = dynamicproperties.GetBoolPropertyFn(false)\n\n\t\toptions.MaxStartJitterInterval = config.TransferProcessorFailoverMaxStartJitterInterval\n\t} else {\n\t\toptions.EnableSplit = config.QueueProcessorEnableSplit\n\t\toptions.SplitMaxLevel = config.QueueProcessorSplitMaxLevel\n\t\toptions.EnableRandomSplitByDomainID = config.QueueProcessorEnableRandomSplitByDomainID\n\t\toptions.RandomSplitProbability = config.QueueProcessorRandomSplitProbability\n\t\toptions.EnablePendingTaskSplitByDomainID = config.QueueProcessorEnablePendingTaskSplitByDomainID\n\t\toptions.PendingTaskSplitThreshold = config.QueueProcessorPendingTaskSplitThreshold\n\t\toptions.EnableStuckTaskSplitByDomainID = config.QueueProcessorEnableStuckTaskSplitByDomainID\n\t\toptions.StuckTaskSplitThreshold = config.QueueProcessorStuckTaskSplitThreshold\n\t\toptions.SplitLookAheadDurationByDomainID = config.QueueProcessorSplitLookAheadDurationByDomainID\n\n\t\toptions.EnablePersistQueueStates = config.QueueProcessorEnablePersistQueueStates\n\t\toptions.EnableLoadQueueStates = config.QueueProcessorEnableLoadQueueStates\n\n\t\toptions.MaxStartJitterInterval = dynamicproperties.GetDurationPropertyFn(0)\n\t}\n\n\tif isActive {\n\t\toptions.MetricScope = metrics.TransferActiveQueueProcessorScope\n\t\toptions.RedispatchInterval = config.ActiveTaskRedispatchInterval\n\t} else {\n\t\toptions.MetricScope = metrics.TransferStandbyQueueProcessorScope\n\t\toptions.RedispatchInterval = config.StandbyTaskRedispatchInterval\n\t}\n\n\treturn options\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_processor_base_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\ttransferQueueProcessorBaseSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockShard            *shard.TestContext\n\t\tmockTaskProcessor    *task.MockProcessor\n\t\tmockQueueSplitPolicy *MockProcessingQueueSplitPolicy\n\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tmetricsScope  metrics.Scope\n\t}\n)\n\nfunc TestTransferQueueProcessorBaseSuite(t *testing.T) {\n\ts := new(transferQueueProcessorBaseSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transferQueueProcessorBaseSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockQueueSplitPolicy = NewMockProcessingQueueSplitPolicy(s.controller)\n\ts.mockTaskProcessor = task.NewMockProcessor(s.controller)\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.metricsScope = s.metricsClient.Scope(metrics.TransferQueueProcessorScope)\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestProcessQueueCollections_NoNextPage_WithNextQueue() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tnewTransferTaskKey(10000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\ttaskInfos := []persistence.Task{\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 1,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain2\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 10,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 100,\n\t\t\t},\n\t\t},\n\t}\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(ackLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\tTasks:         taskInfos,\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tprocessingQueueStates,\n\t\tupdateMaxReadLevel,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tprocessorBase.processQueueCollections()\n\n\tqueueCollection := processorBase.processingQueueCollections[0]\n\ts.NotNil(queueCollection.ActiveQueue())\n\ts.True(taskKeyEquals(maxLevel, queueCollection.Queues()[0].State().ReadLevel()))\n\n\ts.True(processorBase.shouldProcess[queueLevel])\n\tselect {\n\tcase <-processorBase.processCh:\n\tdefault:\n\t\ts.Fail(\"processCh should be unblocked\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestProcessQueueCollections_NoNextPage_NoNextQueue() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tshardMaxLevel := newTransferTaskKey(500)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn shardMaxLevel\n\t}\n\ttaskInfos := []persistence.Task{\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 1,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain2\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 10,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 100,\n\t\t\t},\n\t\t},\n\t}\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(ackLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(shardMaxLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\tTasks:         taskInfos,\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tprocessingQueueStates,\n\t\tupdateMaxReadLevel,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tprocessorBase.processQueueCollections()\n\n\tqueueCollection := processorBase.processingQueueCollections[0]\n\ts.NotNil(queueCollection.ActiveQueue())\n\ts.True(taskKeyEquals(shardMaxLevel, queueCollection.Queues()[0].State().ReadLevel()))\n\n\tshouldProcess, ok := processorBase.shouldProcess[queueLevel]\n\tif ok {\n\t\ts.False(shouldProcess)\n\t}\n\tselect {\n\tcase <-processorBase.processCh:\n\t\ts.Fail(\"processCh should be blocked\")\n\tdefault:\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestProcessQueueCollections_WithNextPage_NotThrottled() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\ttaskInfos := []persistence.Task{\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 1,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain2\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 10,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain1\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 100,\n\t\t\t},\n\t\t},\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: \"testDomain2\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 500,\n\t\t\t},\n\t\t},\n\t}\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(ackLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\tTasks:         taskInfos,\n\t\tNextPageToken: []byte{1, 2, 3},\n\t}, nil).Once()\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).AnyTimes()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tprocessingQueueStates,\n\t\tupdateMaxReadLevel,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\tprocessorBase.processQueueCollections()\n\n\tqueueCollection := processorBase.processingQueueCollections[0]\n\ts.NotNil(queueCollection.ActiveQueue())\n\ts.True(taskKeyEquals(newTransferTaskKey(500), queueCollection.Queues()[0].State().ReadLevel()))\n\n\ts.True(processorBase.shouldProcess[queueLevel])\n\tselect {\n\tcase <-processorBase.processCh:\n\tdefault:\n\t\ts.Fail(\"processCh should be unblocked\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestProcessQueueCollections_WithNextPage_Throttled() {\n\tqueueLevel := defaultProcessingQueueLevel + 1\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\ttaskInfos := []persistence.Task{\n\t\t&persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   \"testDomain1\",\n\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\tRunID:      \"testRunID\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tTaskID: 500,\n\t\t\t},\n\t\t},\n\t}\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(ackLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\tTasks:         taskInfos,\n\t\tNextPageToken: []byte{1, 2, 3},\n\t}, nil).Once()\n\n\t// return false to indicate that taskCh is full\n\t// so the queue should backoff\n\ts.mockTaskProcessor.EXPECT().TrySubmit(gomock.Any()).Return(false, nil).AnyTimes()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tprocessingQueueStates,\n\t\tupdateMaxReadLevel,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\tprocessorBase.options.PollBackoffInterval = dynamicproperties.GetDurationPropertyFn(time.Millisecond * 100)\n\n\tprocessorBase.processQueueCollections()\n\n\tqueueCollection := processorBase.processingQueueCollections[0]\n\ts.NotNil(queueCollection.ActiveQueue())\n\ts.True(taskKeyEquals(newTransferTaskKey(500), queueCollection.Queues()[0].State().ReadLevel()))\n\n\tprocessorBase.processingLock.Lock()\n\ts.False(processorBase.shouldProcess[queueLevel])\n\t_, ok := processorBase.backoffTimer[queueLevel]\n\ts.True(ok)\n\tprocessorBase.processingLock.Unlock()\n\tselect {\n\tcase <-processorBase.processCh:\n\t\ts.Fail(\"processCh should be blocked before the backoff timer fires\")\n\tdefault:\n\t}\n\n\ttime.Sleep(time.Millisecond * 300)\n\tprocessorBase.processingLock.Lock()\n\ts.True(processorBase.shouldProcess[queueLevel])\n\t_, ok = processorBase.backoffTimer[queueLevel]\n\ts.False(ok)\n\tprocessorBase.processingLock.Unlock()\n\tselect {\n\tcase <-processorBase.processCh:\n\tdefault:\n\t\ts.Fail(\"processCh should be unblocked after the backoff timer fires\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestReadTasks_NoNextPage() {\n\treadLevel := newTransferTaskKey(3)\n\tmaxReadLevel := newTransferTaskKey(100)\n\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tgetTransferTaskResponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks:         []persistence.Task{&persistence.DecisionTask{}, &persistence.DecisionTask{}, &persistence.DecisionTask{}},\n\t\tNextPageToken: nil,\n\t}\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(readLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxReadLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(getTransferTaskResponse, nil).Once()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\ttasks, more, err := processorBase.readTasks(readLevel, maxReadLevel)\n\ts.NoError(err)\n\ts.Len(tasks, len(getTransferTaskResponse.Tasks))\n\ts.False(more)\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestReadTasks_WithNextPage() {\n\treadLevel := newTransferTaskKey(3)\n\tmaxReadLevel := newTransferTaskKey(10)\n\n\tmockExecutionManager := s.mockShard.Resource.ExecutionMgr\n\tgetTransferTaskResponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks:         []persistence.Task{&persistence.DecisionTask{}, &persistence.DecisionTask{}, &persistence.DecisionTask{}},\n\t\tNextPageToken: []byte{1, 2, 3},\n\t}\n\tmockExecutionManager.On(\"GetHistoryTasks\", mock.Anything, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(readLevel.(transferTaskKey).taskID + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxReadLevel.(transferTaskKey).taskID + 1),\n\t\tPageSize:            s.mockShard.GetConfig().TransferTaskBatchSize(),\n\t}).Return(getTransferTaskResponse, nil).Once()\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t\tnil,\n\t)\n\n\ttasks, more, err := processorBase.readTasks(readLevel, maxReadLevel)\n\ts.NoError(err)\n\ts.Len(tasks, len(getTransferTaskResponse.Tasks))\n\ts.True(more)\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestTransferProcessorPump_ProcessChHasItem() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tnewTransferTaskKey(10000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tprocessedCh := make(chan struct{}, 1)\n\tprocessorBase.processQueueCollectionsFn = func() {\n\t\tprocessedCh <- struct{}{}\n\t}\n\n\tprocessorBase.Start()\n\tdefer processorBase.Stop()\n\n\tprocessorBase.processCh <- struct{}{}\n\n\tselect {\n\tcase <-processedCh:\n\t\treturn\n\tcase <-time.After(100 * time.Millisecond):\n\t\ts.Fail(\"processQueueCollectionsFn not called\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestTransferProcessorPump_NotifyChHasItem() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tnewTransferTaskKey(10000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tprocessedCh := make(chan struct{}, 1)\n\tprocessorBase.processQueueCollectionsFn = func() {\n\t\tprocessedCh <- struct{}{}\n\t}\n\n\tprocessorBase.Start()\n\tdefer processorBase.Stop()\n\n\tprocessorBase.notifyCh <- struct{}{}\n\n\tselect {\n\tcase <-processedCh:\n\t\treturn\n\tcase <-time.After(100 * time.Millisecond):\n\t\ts.Fail(\"processQueueCollectionsFn not called\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) TestTransferProcessorPump_UpdateAckLevel() {\n\tqueueLevel := 0\n\tackLevel := newTransferTaskKey(0)\n\tmaxLevel := newTransferTaskKey(1000)\n\tprocessingQueueStates := []ProcessingQueueState{\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tackLevel,\n\t\t\tmaxLevel,\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t\tNewProcessingQueueState(\n\t\t\tqueueLevel,\n\t\t\tnewTransferTaskKey(1000),\n\t\t\tnewTransferTaskKey(10000),\n\t\t\tNewDomainFilter(map[string]struct{}{\"testDomain1\": {}}, false),\n\t\t),\n\t}\n\tupdateMaxReadLevel := func() task.Key {\n\t\treturn newTransferTaskKey(10000)\n\t}\n\n\tprocessorBase := s.newTestTransferQueueProcessorBase(processingQueueStates, updateMaxReadLevel, nil, nil, nil)\n\tprocessorBase.options.UpdateAckInterval = dynamicproperties.GetDurationPropertyFn(1 * time.Millisecond)\n\tupdatedCh := make(chan struct{}, 1)\n\tprocessorBase.processQueueCollectionsFn = func() {}\n\tprocessorBase.updateAckLevelFn = func() (bool, task.Key, error) {\n\t\tupdatedCh <- struct{}{}\n\t\treturn false, nil, nil\n\t}\n\n\tprocessorBase.Start()\n\tdefer processorBase.Stop()\n\n\tselect {\n\tcase <-updatedCh:\n\t\treturn\n\tcase <-time.After(100 * time.Millisecond):\n\t\ts.Fail(\"updateAckLevelFn not called\")\n\t}\n}\n\nfunc (s *transferQueueProcessorBaseSuite) newTestTransferQueueProcessorBase(\n\tprocessingQueueStates []ProcessingQueueState,\n\tmaxReadLevel updateMaxReadLevelFn,\n\tupdateClusterAckLevel updateClusterAckLevelFn,\n\tupdateProcessingQueueStates updateProcessingQueueStatesFn,\n\ttransferQueueShutdown queueShutdownFn,\n) *transferQueueProcessorBase {\n\tprocessorBase := newTransferQueueProcessorBase(\n\t\ts.mockShard,\n\t\tprocessingQueueStates,\n\t\ts.mockTaskProcessor,\n\t\tnewTransferQueueProcessorOptions(s.mockShard.GetConfig(), true, false),\n\t\tmaxReadLevel,\n\t\tupdateClusterAckLevel,\n\t\tupdateProcessingQueueStates,\n\t\ttransferQueueShutdown,\n\t\tnil,\n\t\tnil,\n\t\ts.logger,\n\t\ts.metricsClient,\n\t)\n\tfor _, queueCollections := range processorBase.processingQueueCollections {\n\t\tprocessorBase.shouldProcess[queueCollections.Level()] = true\n\t}\n\treturn processorBase\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_processor_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nfunc setupTransferQueueProcessor(t *testing.T, cfg *config.Config) (*gomock.Controller, *transferQueueProcessor) {\n\tctrl := gomock.NewController(t)\n\n\tif cfg == nil {\n\t\tcfg = config.NewForTest()\n\t}\n\n\tmockShard := shard.NewTestContext(\n\t\tt,\n\t\tctrl,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tcfg,\n\t)\n\n\treturn ctrl, NewTransferQueueProcessor(\n\t\tmockShard,\n\t\ttask.NewMockProcessor(ctrl),\n\t\texecution.NewCache(mockShard),\n\t\treset.NewMockWorkflowResetter(ctrl),\n\t\tarchiver.NewMockClient(ctrl),\n\t\tinvariant.NewMockInvariant(ctrl),\n\t\tworkflowcache.NewMockWFCache(ctrl),\n\t).(*transferQueueProcessor)\n}\n\nfunc TestTransferQueueProcessor_StartStop(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tassert.Equal(t, common.DaemonStatusInitialized, processor.status)\n\n\tprocessor.Start()\n\tassert.Equal(t, common.DaemonStatusStarted, processor.status)\n\n\t// noop start\n\tprocessor.Start()\n\n\tprocessor.Stop()\n\tassert.Equal(t, common.DaemonStatusStopped, processor.status)\n\n\t// noop stop\n\tprocessor.Stop()\n}\n\nfunc TestTransferQueueProcessor_StartNotGracefulStop(t *testing.T) {\n\tcfg := config.NewForTest()\n\tcfg.QueueProcessorEnableGracefulSyncShutdown = dynamicproperties.GetBoolPropertyFn(false)\n\n\tctrl, processor := setupTransferQueueProcessor(t, cfg)\n\tdefer ctrl.Finish()\n\n\tassert.Equal(t, common.DaemonStatusInitialized, processor.status)\n\tprocessor.Start()\n\n\tassert.Equal(t, common.DaemonStatusStarted, processor.status)\n\n\tprocessor.Stop()\n\tassert.Equal(t, common.DaemonStatusStopped, processor.status)\n}\n\nfunc TestTransferQueueProcessor_NotifyNewTask(t *testing.T) {\n\ttests := map[string]struct {\n\t\ttasks             []persistence.Task\n\t\tclusterName       string\n\t\tcheckNotification func(processor *transferQueueProcessor)\n\t\tshouldPanic       bool\n\t}{\n\t\t\"no task\": {\n\t\t\ttasks:             []persistence.Task{},\n\t\t\tclusterName:       constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\tcheckNotification: func(processor *transferQueueProcessor) {},\n\t\t},\n\t\t\"notify active queue processor\": {\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{},\n\t\t\t},\n\t\t\tclusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\tcheckNotification: func(processor *transferQueueProcessor) {\n\t\t\t\t<-processor.activeQueueProcessor.notifyCh\n\t\t\t},\n\t\t},\n\t\t\"notify standby queue processor\": {\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{},\n\t\t\t},\n\t\t\tclusterName: \"standby\",\n\t\t\tcheckNotification: func(processor *transferQueueProcessor) {\n\t\t\t\t<-processor.standbyQueueProcessors[\"standby\"].notifyCh\n\t\t\t},\n\t\t},\n\t\t\"panic on unknown cluster\": {\n\t\t\ttasks: []persistence.Task{\n\t\t\t\t&persistence.ActivityTask{},\n\t\t\t},\n\t\t\tclusterName:       \"unknown\",\n\t\t\tcheckNotification: func(processor *transferQueueProcessor) {},\n\t\t\tshouldPanic:       true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tinfo := &hcommon.NotifyTaskInfo{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t},\n\t\t\t\tTasks: tc.tasks,\n\t\t\t}\n\n\t\t\tif tc.shouldPanic {\n\t\t\t\tassert.Panics(t, func() {\n\t\t\t\t\tprocessor.NotifyNewTask(tc.clusterName, info)\n\t\t\t\t})\n\t\t\t} else {\n\t\t\t\tprocessor.NotifyNewTask(tc.clusterName, info)\n\t\t\t\ttc.checkNotification(processor)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTransferQueueProcessor_FailoverDomain(t *testing.T) {\n\ttests := map[string]struct {\n\t\tdomainIDs        map[string]struct{}\n\t\tsetupMocks       func(mockShard *shard.TestContext)\n\t\tprocessorStarted bool\n\t}{\n\t\t\"processor not started\": {\n\t\t\tdomainIDs:        map[string]struct{}{},\n\t\t\tsetupMocks:       func(mockShard *shard.TestContext) {},\n\t\t\tprocessorStarted: false,\n\t\t},\n\t\t\"processor started\": {\n\t\t\tdomainIDs: map[string]struct{}{\"domainID\": {}},\n\t\t\tsetupMocks: func(mockShard *shard.TestContext) {\n\t\t\t\tresponse := &persistence.GetHistoryTasksResponse{\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.DecisionTask{\n\t\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockShard.GetExecutionManager().(*mocks.ExecutionManager).On(\"GetHistoryTasks\", mock.Anything, mock.Anything).Return(response, nil).Once()\n\t\t\t},\n\t\t\tprocessorStarted: true,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tif tc.processorStarted {\n\t\t\t\tdefer processor.Stop()\n\t\t\t\tprocessor.Start()\n\t\t\t}\n\n\t\t\ttc.setupMocks(processor.shard.(*shard.TestContext))\n\n\t\t\tprocessor.FailoverDomain(tc.domainIDs)\n\n\t\t\tprocessor.ackLevel = 10\n\n\t\t\tif tc.processorStarted {\n\t\t\t\tassert.Equal(t, 1, len(processor.failoverQueueProcessors))\n\t\t\t\tassert.Equal(t, common.DaemonStatusStarted, processor.failoverQueueProcessors[0].status)\n\t\t\t}\n\n\t\t\tif tc.processorStarted {\n\t\t\t\tprocessor.drain()\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTransferQueueProcessor_HandleAction(t *testing.T) {\n\ttests := map[string]struct {\n\t\tclusterName string\n\t\terr         error\n\t}{\n\t\t\"active cluster\": {\n\t\t\tclusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t},\n\t\t\"standby cluster\": {\n\t\t\tclusterName: \"standby\",\n\t\t},\n\t\t\"unknown cluster\": {\n\t\t\tclusterName: \"unknown\",\n\t\t\terr:         errors.New(\"unknown cluster name: unknown\"),\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tdefer processor.Stop()\n\t\t\tprocessor.Start()\n\n\t\t\tctx := context.Background()\n\n\t\t\taction := &Action{\n\t\t\t\tActionType:               ActionTypeGetState,\n\t\t\t\tGetStateActionAttributes: &GetStateActionAttributes{},\n\t\t\t}\n\n\t\t\tactionResult, err := processor.HandleAction(ctx, tc.clusterName, action)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Nil(t, actionResult)\n\t\t\t\tassert.ErrorContains(t, err, tc.err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.NotNil(t, actionResult)\n\t\t\t\tassert.Equal(t, action.ActionType, actionResult.ActionType)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTransferQueueProcessor_LockTaskProcessing(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tlocked := make(chan struct{}, 1)\n\n\tprocessor.LockTaskProcessing()\n\n\tgo func() {\n\t\tdefer processor.taskAllocator.Unlock()\n\t\tprocessor.taskAllocator.Lock()\n\t\tlocked <- struct{}{}\n\t}()\n\n\tselect {\n\tcase <-locked:\n\t\tassert.Fail(t, \"Expected mutex to be locked, but it was unlocked\")\n\tcase <-time.After(50 * time.Millisecond):\n\t\tprocessor.UnlockTaskProcessing()\n\t\tassert.True(t, true, \"Mutex is locked as expected\")\n\t}\n}\n\nfunc TestTransferQueueProcessor_completeTransfer(t *testing.T) {\n\ttests := map[string]struct {\n\t\tackLevel  int64\n\t\tmockSetup func(*shard.TestContext)\n\t\terr       error\n\t}{\n\t\t\"noop - ackLevel >= newAckLevelTaskID\": {\n\t\t\tackLevel:  10,\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {},\n\t\t\terr:       nil,\n\t\t},\n\t\t\"error - ackLevel < newAckLevelTaskID - RangeCompleteTransferTask error\": {\n\t\t\tackLevel: 1,\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, assert.AnError).Once()\n\t\t\t},\n\t\t\terr: assert.AnError,\n\t\t},\n\t\t\"success - ackLevel < newAckLevelTaskID\": {\n\t\t\tackLevel: 1,\n\t\t\tmockSetup: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\t\t\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, nil).Once()\n\t\t\t\tmockShard.GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tt := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tprocessor.ackLevel = tt.ackLevel\n\n\t\t\ttt.mockSetup(processor.shard.(*shard.TestContext))\n\n\t\t\tdefer processor.Stop()\n\t\t\tprocessor.Start()\n\n\t\t\terr := processor.completeTransfer()\n\n\t\t\tif tt.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorIs(t, err, tt.err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTransferQueueProcessor_completeTransferLoop(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tprocessor.config.TransferProcessorCompleteTransferInterval = dynamicproperties.GetDurationPropertyFn(10 * time.Millisecond)\n\n\tprocessor.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tprocessor.shard.(*shard.TestContext).Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, nil)\n\n\tprocessor.shard.(*shard.TestContext).GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\n\tprocessor.shutdownWG.Add(1)\n\n\tgo func() {\n\t\ttime.Sleep(200 * time.Millisecond)\n\t\tprocessor.activeQueueProcessor.Stop()\n\t\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\tclose(processor.shutdownChan)\n\t\tcommon.AwaitWaitGroup(&processor.shutdownWG, time.Minute)\n\t}()\n\n\tprocessor.completeTransferLoop()\n}\n\nfunc TestTransferQueueProcessor_completeTransferLoop_ErrShardClosed(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tprocessor.config.TransferProcessorCompleteTransferInterval = dynamicproperties.GetDurationPropertyFn(30 * time.Millisecond)\n\n\tprocessor.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tprocessor.shard.(*shard.TestContext).Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, &shard.ErrShardClosed{}).Once()\n\n\tprocessor.shutdownWG.Add(1)\n\n\tgo func() {\n\t\ttime.Sleep(50 * time.Millisecond)\n\t\tprocessor.activeQueueProcessor.Stop()\n\t\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\tclose(processor.shutdownChan)\n\t\tcommon.AwaitWaitGroup(&processor.shutdownWG, time.Minute)\n\t}()\n\n\tprocessor.completeTransferLoop()\n}\n\nfunc TestTransferQueueProcessor_completeTransferLoop_ErrShardClosedNotGraceful(t *testing.T) {\n\tcfg := config.NewForTest()\n\tcfg.QueueProcessorEnableGracefulSyncShutdown = dynamicproperties.GetBoolPropertyFn(false)\n\n\tctrl, processor := setupTransferQueueProcessor(t, cfg)\n\tdefer ctrl.Finish()\n\n\tprocessor.config.TransferProcessorCompleteTransferInterval = dynamicproperties.GetDurationPropertyFn(30 * time.Millisecond)\n\n\tprocessor.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tprocessor.shard.(*shard.TestContext).Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, &shard.ErrShardClosed{}).Once()\n\n\tprocessor.shutdownWG.Add(1)\n\n\tgo func() {\n\t\ttime.Sleep(50 * time.Millisecond)\n\t\tprocessor.activeQueueProcessor.Stop()\n\t\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\tclose(processor.shutdownChan)\n\t\tcommon.AwaitWaitGroup(&processor.shutdownWG, time.Minute)\n\t}()\n\n\tprocessor.completeTransferLoop()\n}\n\nfunc TestTransferQueueProcessor_completeTransferLoop_OtherError(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tprocessor.config.TransferProcessorCompleteTransferInterval = dynamicproperties.GetDurationPropertyFn(30 * time.Millisecond)\n\n\tprocessor.activeQueueProcessor.Start()\n\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\tstandbyQueueProcessor.Start()\n\t}\n\n\tprocessor.shard.(*shard.TestContext).Resource.ExecutionMgr.On(\"RangeCompleteHistoryTask\", mock.Anything, mock.Anything).\n\t\tReturn(&persistence.RangeCompleteHistoryTaskResponse{}, assert.AnError)\n\n\tprocessor.shutdownWG.Add(1)\n\n\tgo func() {\n\t\ttime.Sleep(50 * time.Millisecond)\n\t\tprocessor.activeQueueProcessor.Stop()\n\t\tfor _, standbyQueueProcessor := range processor.standbyQueueProcessors {\n\t\t\tstandbyQueueProcessor.Stop()\n\t\t}\n\n\t\tclose(processor.shutdownChan)\n\t\tcommon.AwaitWaitGroup(&processor.shutdownWG, time.Minute)\n\t}()\n\n\tprocessor.completeTransferLoop()\n}\n\nfunc Test_transferQueueActiveProcessor_taskFilter(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockSetup func(*shard.TestContext)\n\t\ttask      persistence.Task\n\t\terr       error\n\t}{\n\t\t\"noop - domain not registered\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{Status: persistence.DomainStatusDeprecated}, nil, true, nil, 1, nil, 0, 0, 0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(1)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t\"taskFilter success\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).\n\t\t\t\t\tReturn(constants.TestDomainName, nil).Times(1)\n\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(2)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\t// Error to Execute only. Since taskImpl is not exported, and the filter is a private field, had to use the Execute method to execute the filter function.\n\t\t\t// The filter returned no error\n\t\t\terr: &types.BadRequestError{Message: \"Can't load workflow execution.  WorkflowId not set.\"},\n\t\t},\n\t}\n\n\tfor name, tt := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\ttt.mockSetup(processor.shard.(*shard.TestContext))\n\n\t\t\terr := processor.activeQueueProcessor.taskInitializer(tt.task).Execute()\n\n\t\t\tif tt.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorContains(t, err, tt.err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_transferQueueActiveProcessor_updateClusterAckLevel(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\ttaskID := int64(11)\n\n\tkey := transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n\n\tprocessor.shard.(*shard.TestContext).GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := processor.activeQueueProcessor.processorBase.updateClusterAckLevel(key)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, taskID, processor.shard.(*shard.TestContext).ShardInfo().ClusterTransferAckLevel[constants.TestClusterMetadata.GetCurrentClusterName()])\n}\n\nfunc Test_transferQueueActiveProcessor_updateProcessingQueueStates(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\ttaskID := int64(11)\n\n\tkey := transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n\n\tstate := NewProcessingQueueState(12, key, key, DomainFilter{})\n\n\tstates := []ProcessingQueueState{state}\n\n\tprocessor.shard.(*shard.TestContext).GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := processor.activeQueueProcessor.processorBase.updateProcessingQueueStates(states)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, taskID, processor.shard.(*shard.TestContext).ShardInfo().ClusterTransferAckLevel[constants.TestClusterMetadata.GetCurrentClusterName()])\n\tassert.Equal(t, 1, len(processor.shard.(*shard.TestContext).ShardInfo().TransferProcessingQueueStates.StatesByCluster[constants.TestClusterMetadata.GetCurrentClusterName()]))\n\tassert.Equal(t, int32(state.Level()), *processor.shard.(*shard.TestContext).ShardInfo().TransferProcessingQueueStates.StatesByCluster[constants.TestClusterMetadata.GetCurrentClusterName()][0].Level)\n}\n\nfunc Test_transferQueueActiveProcessor_queueShutdown(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\terr := processor.activeQueueProcessor.queueShutdown()\n\n\tassert.NoError(t, err)\n}\n\nfunc Test_transferQueueStandbyProcessor_taskFilter(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockSetup func(*shard.TestContext)\n\t\ttask      persistence.Task\n\t\terr       error\n\t}{\n\t\t\"noop - domain not registered\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{Status: persistence.DomainStatusDeprecated}, nil, true, nil, 1, nil, 0, 0, 0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(1)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t\"no error - TransferTaskTypeCloseExecution or TransferTaskTypeRecordWorkflowClosed\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainName(constants.TestDomainID).\n\t\t\t\t\tReturn(constants.TestDomainName, nil).Times(1)\n\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t\t\t{ClusterName: constants.TestClusterMetadata.GetCurrentClusterName()},\n\t\t\t\t\t\t\t{ClusterName: \"standby\"},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(2)\n\t\t\t},\n\t\t\ttask: &persistence.CloseExecutionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\t// Error to Execute only. Since taskImpl is not exported, and the filter is a private field, had to use the Execute method to execute the filter function.\n\t\t\t// The filter returned no error\n\t\t\terr: &types.BadRequestError{Message: \"Can't load workflow execution.  WorkflowId not set.\"},\n\t\t},\n\t\t\"error - TransferTaskTypeCloseExecution or TransferTaskTypeRecordWorkflowClosed - cannot find domain - retry\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(nil, assert.AnError).Times(2)\n\t\t\t},\n\t\t\ttask: &persistence.CloseExecutionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: assert.AnError,\n\t\t},\n\t\t\"noop - TransferTaskTypeCloseExecution or TransferTaskTypeRecordWorkflowClosed - EntityNotExistsError\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(nil, &types.EntityNotExistsError{Message: \"domain doesn't exist\"}).Times(2)\n\t\t\t},\n\t\t\ttask: &persistence.CloseExecutionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t\"taskFilter success\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(2)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\n\tfor name, tt := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\ttt.mockSetup(processor.shard.(*shard.TestContext))\n\n\t\t\terr := processor.standbyQueueProcessors[\"standby\"].taskInitializer(tt.task).Execute()\n\n\t\t\tif tt.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorContains(t, err, tt.err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_transferQueueStandbyProcessor_updateClusterAckLevel(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\ttaskID := int64(11)\n\n\tkey := transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n\n\tprocessor.shard.(*shard.TestContext).GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := processor.standbyQueueProcessors[\"standby\"].processorBase.updateClusterAckLevel(key)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, taskID, processor.shard.(*shard.TestContext).ShardInfo().ClusterTransferAckLevel[\"standby\"])\n}\n\nfunc Test_transferQueueStandbyProcessor_updateProcessingQueueStates(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\ttaskID := int64(11)\n\n\tkey := transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n\n\tstate := NewProcessingQueueState(12, key, key, DomainFilter{})\n\n\tstates := []ProcessingQueueState{state}\n\n\tprocessor.shard.(*shard.TestContext).GetShardManager().(*mocks.ShardManager).On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil).Once()\n\n\terr := processor.standbyQueueProcessors[\"standby\"].processorBase.updateProcessingQueueStates(states)\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, taskID, processor.shard.(*shard.TestContext).ShardInfo().ClusterTransferAckLevel[\"standby\"])\n\tassert.Equal(t, 1, len(processor.shard.(*shard.TestContext).ShardInfo().TransferProcessingQueueStates.StatesByCluster[\"standby\"]))\n\tassert.Equal(t, int32(state.Level()), *processor.shard.(*shard.TestContext).ShardInfo().TransferProcessingQueueStates.StatesByCluster[\"standby\"][0].Level)\n}\n\nfunc Test_transferQueueStandbyProcessor_queueShutdown(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\terr := processor.standbyQueueProcessors[\"standby\"].queueShutdown()\n\n\tassert.NoError(t, err)\n}\n\nfunc Test_transferQueueFailoverProcessor_taskFilter(t *testing.T) {\n\ttests := map[string]struct {\n\t\tmockSetup func(*shard.TestContext)\n\t\ttask      persistence.Task\n\t\terr       error\n\t}{\n\t\t\"noop - domain not registered\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{Status: persistence.DomainStatusDeprecated}, nil, true, nil, 1, nil, 0, 0, 0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(1)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t\"taskFilter success\": {\n\t\t\tmockSetup: func(testContext *shard.TestContext) {\n\t\t\t\tcacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{Status: persistence.DomainStatusRegistered},\n\t\t\t\t\tnil,\n\t\t\t\t\ttrue,\n\t\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\t\tActiveClusterName: constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\t\t\t},\n\t\t\t\t\t1,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0)\n\n\t\t\t\ttestContext.GetDomainCache().(*cache.MockDomainCache).EXPECT().GetDomainByID(constants.TestDomainID).\n\t\t\t\t\tReturn(cacheEntry, nil).Times(1)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: constants.TestDomainID,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\n\tfor name, tt := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\ttt.mockSetup(processor.shard.(*shard.TestContext))\n\n\t\t\tdomainIDs := map[string]struct{}{\"standby\": {}}\n\n\t\t\t_, failoverQueueProcessor := newTransferQueueFailoverProcessor(\n\t\t\t\tprocessor.shard,\n\t\t\t\tprocessor.taskProcessor,\n\t\t\t\tprocessor.taskAllocator,\n\t\t\t\tprocessor.activeTaskExecutor,\n\t\t\t\tprocessor.logger,\n\t\t\t\t0,\n\t\t\t\t10,\n\t\t\t\tdomainIDs,\n\t\t\t\tconstants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\t)\n\n\t\t\terr := failoverQueueProcessor.taskInitializer(tt.task).Execute()\n\n\t\t\tif tt.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorContains(t, err, tt.err.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_transferQueueFailoverProcessor_updateClusterAckLevel(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\ttaskID := int64(11)\n\n\tkey := transferTaskKey{\n\t\ttaskID: taskID,\n\t}\n\n\tdomainIDs := map[string]struct{}{\"standby\": {}}\n\n\tupdateClusterAckLevel, _ := newTransferQueueFailoverProcessor(\n\t\tprocessor.shard,\n\t\tprocessor.taskProcessor,\n\t\tprocessor.taskAllocator,\n\t\tprocessor.activeTaskExecutor,\n\t\tprocessor.logger,\n\t\t0,\n\t\t10,\n\t\tdomainIDs,\n\t\tconstants.TestClusterMetadata.GetCurrentClusterName(),\n\t)\n\n\terr := updateClusterAckLevel(key)\n\n\tassert.NoError(t, err)\n}\n\nfunc Test_transferQueueFailoverProcessor_queueShutdown(t *testing.T) {\n\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\tdefer ctrl.Finish()\n\n\tdomainIDs := map[string]struct{}{\"standby\": {}}\n\n\t_, failoverQueueProcessor := newTransferQueueFailoverProcessor(\n\t\tprocessor.shard,\n\t\tprocessor.taskProcessor,\n\t\tprocessor.taskAllocator,\n\t\tprocessor.activeTaskExecutor,\n\t\tprocessor.logger,\n\t\t0,\n\t\t10,\n\t\tdomainIDs,\n\t\tconstants.TestClusterMetadata.GetCurrentClusterName(),\n\t)\n\n\terr := failoverQueueProcessor.queueShutdown()\n\n\tassert.NoError(t, err)\n}\n\nfunc Test_loadTransferProcessingQueueStates(t *testing.T) {\n\ttests := map[string]struct {\n\t\tenableLoadQueueStates bool\n\t\tclusterName           string\n\t\ttaskID                func(testContext *shard.TestContext) int64\n\t}{\n\t\t\"load queue states true\": {\n\t\t\tenableLoadQueueStates: true,\n\t\t\tclusterName:           constants.TestClusterMetadata.GetCurrentClusterName(),\n\t\t\ttaskID: func(testContext *shard.TestContext) int64 {\n\t\t\t\treturn testContext.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, constants.TestClusterMetadata.GetCurrentClusterName()).GetTaskID()\n\t\t\t},\n\t\t},\n\t\t\"load queue states false\": {\n\t\t\tenableLoadQueueStates: false,\n\t\t\tclusterName:           \"standby\",\n\t\t\ttaskID: func(testContext *shard.TestContext) int64 {\n\t\t\t\treturn testContext.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, \"standby\").GetTaskID()\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tt := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl, processor := setupTransferQueueProcessor(t, nil)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\topts := &queueProcessorOptions{\n\t\t\t\tEnableLoadQueueStates: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn tt.enableLoadQueueStates\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tpqs := loadTransferProcessingQueueStates(tt.clusterName, processor.shard.(*shard.TestContext), opts, processor.shard.(*shard.TestContext).GetLogger())\n\n\t\t\tassert.NotNil(t, pqs)\n\t\t\tassert.Equal(t, 1, len(pqs))\n\t\t\tassert.Equal(t, tt.taskID(processor.shard.(*shard.TestContext)), pqs[0].AckLevel().(transferTaskKey).taskID)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_validator.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\tdefaultMaxPendingTasksSize = 5000\n)\n\ntype (\n\tpendingTaskInfo struct {\n\t\texecutionInfo          *persistence.WorkflowExecutionInfo\n\t\ttask                   persistence.Task\n\t\tpotentialFalsePositive bool\n\t}\n\n\ttransferQueueValidator struct {\n\t\tsync.Mutex\n\n\t\tprocessor    *transferQueueProcessorBase\n\t\ttimeSource   clock.TimeSource\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\n\t\tpendingTaskInfos   map[int64]pendingTaskInfo\n\t\tmaxReadLevels      map[int]task.Key\n\t\tminReadTaskID      int64\n\t\tlastValidateTime   time.Time\n\t\tvalidationInterval dynamicproperties.DurationPropertyFn\n\t}\n)\n\nfunc newTransferQueueValidator(\n\tprocessor *transferQueueProcessorBase,\n\tvalidationInterval dynamicproperties.DurationPropertyFn,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) *transferQueueValidator {\n\ttimeSource := processor.shard.GetTimeSource()\n\treturn &transferQueueValidator{\n\t\tprocessor:          processor,\n\t\ttimeSource:         timeSource,\n\t\tlogger:             logger,\n\t\tmetricsScope:       metricsScope,\n\t\tpendingTaskInfos:   make(map[int64]pendingTaskInfo),\n\t\tmaxReadLevels:      make(map[int]task.Key),\n\t\tminReadTaskID:      0,\n\t\tlastValidateTime:   timeSource.Now(),\n\t\tvalidationInterval: validationInterval,\n\t}\n}\n\nfunc (v *transferQueueValidator) addTasks(info *hcommon.NotifyTaskInfo) {\n\tv.Lock()\n\tdefer v.Unlock()\n\n\tnumTaskToAdd := len(info.Tasks)\n\tif numTaskToAdd+len(v.pendingTaskInfos) > defaultMaxPendingTasksSize {\n\t\tnumTaskToAdd = defaultMaxPendingTasksSize - len(v.pendingTaskInfos)\n\t\tvar taskDump strings.Builder\n\t\tdroppedTasks := info.Tasks[numTaskToAdd:]\n\t\tfor _, task := range droppedTasks {\n\t\t\ttaskDump.WriteString(fmt.Sprintf(\"%+v\\n\", task))\n\t\t}\n\t\tv.logger.Warn(\n\t\t\t\"Too many pending transfer tasks, dropping new tasks\",\n\t\t\ttag.WorkflowDomainID(info.ExecutionInfo.DomainID),\n\t\t\ttag.WorkflowID(info.ExecutionInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(info.ExecutionInfo.RunID),\n\t\t\ttag.Key(\"dropped-transfer-tasks\"),\n\t\t\ttag.Value(taskDump.String()),\n\t\t)\n\t\tv.metricsScope.AddCounter(metrics.QueueValidatorDropTaskCounter, int64(len(droppedTasks)))\n\t}\n\n\tfor _, task := range info.Tasks[:numTaskToAdd] {\n\t\t// It is possible that a task is acked before it is added to the validator\n\t\t// In that case, the lost task could be a potential false positive case\n\t\tpotentialFalsePositive := info.PersistenceError || task.GetTaskID() <= v.minReadTaskID\n\t\tv.pendingTaskInfos[task.GetTaskID()] = pendingTaskInfo{\n\t\t\texecutionInfo:          info.ExecutionInfo,\n\t\t\ttask:                   task,\n\t\t\tpotentialFalsePositive: potentialFalsePositive,\n\t\t}\n\t}\n}\n\nfunc (v *transferQueueValidator) ackTasks(\n\tqueueLevel int,\n\treadLevel task.Key,\n\tmaxReadLevel task.Key,\n\tloadedTasks map[task.Key]task.Task,\n) {\n\tv.Lock()\n\tdefer v.Unlock()\n\n\tfor _, task := range loadedTasks {\n\t\t// note that loadedTasks will contain tasks not in pendingTaskInfos\n\t\t// either due to the retries when updating mutable state or the fact that we\n\t\t// have two processors for the same queue in DB\n\t\tdelete(v.pendingTaskInfos, task.GetTaskID())\n\t}\n\n\tif queueLevel == defaultProcessingQueueLevel {\n\t\tif expectedReadLevel, ok := v.maxReadLevels[queueLevel]; ok && expectedReadLevel.Less(readLevel) {\n\t\t\t// TODO: implement an event logger for queue processor and dump all events when this validation fails.\n\t\t\tv.logger.Error(\"Transfer queue processor load request is not continuous\")\n\t\t\tv.metricsScope.IncCounter(metrics.QueueValidatorInvalidLoadCounter)\n\t\t}\n\t}\n\tv.maxReadLevels[queueLevel] = maxReadLevel\n\n\tif v.timeSource.Now().After(v.lastValidateTime.Add(v.validationInterval())) {\n\t\tv.validatePendingTasks()\n\t\tv.lastValidateTime = v.timeSource.Now()\n\t}\n}\n\nfunc (v *transferQueueValidator) validatePendingTasks() {\n\tv.metricsScope.IncCounter(metrics.QueueValidatorValidationCounter)\n\n\t// first find the minimal read level across all processing queue levels\n\tminReadLevel := maximumTransferTaskKey\n\tfor _, queueCollection := range v.processor.processingQueueCollections {\n\t\tif activeQueue := queueCollection.ActiveQueue(); activeQueue != nil {\n\t\t\tminReadLevel = minTaskKey(minReadLevel, activeQueue.State().ReadLevel())\n\t\t}\n\t}\n\n\t// all pending tasks with taskID <= minReadLevel will never be loaded,\n\t// log those tasks, emit metrics, and delete them from pending tasks\n\t//\n\t// NOTE: this may contain false positives as when the persistence operation for\n\t// updating workflow execution times out, task notification will still be sent,\n\t// but those tasks may not be persisted.\n\t//\n\t// As a result, when lost task metric is emitted, first check if there's corresponding\n\t// persistence operation errors.\n\tminReadTaskID := minReadLevel.(transferTaskKey).taskID\n\tfor taskID, taskInfo := range v.pendingTaskInfos {\n\t\tif taskID <= minReadTaskID {\n\t\t\tv.logger.Error(\"Failed to load transfer task\",\n\t\t\t\ttag.TaskID(taskID),\n\t\t\t\ttag.TaskVisibilityTimestamp(taskInfo.task.GetVisibilityTimestamp().UnixNano()),\n\t\t\t\ttag.FailoverVersion(taskInfo.task.GetVersion()),\n\t\t\t\ttag.TaskType(taskInfo.task.GetTaskType()),\n\t\t\t\ttag.WorkflowDomainID(taskInfo.executionInfo.DomainID),\n\t\t\t\ttag.WorkflowID(taskInfo.executionInfo.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(taskInfo.executionInfo.RunID),\n\t\t\t\ttag.Bool(taskInfo.potentialFalsePositive),\n\t\t\t)\n\t\t\tv.metricsScope.IncCounter(metrics.QueueValidatorLostTaskCounter)\n\t\t\tdelete(v.pendingTaskInfos, taskID)\n\t\t}\n\t}\n\tv.minReadTaskID = minReadTaskID\n}\n"
  },
  {
    "path": "service/history/queue/transfer_queue_validator_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queue\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\ttransferQueueValidatorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tmockShard       *shard.TestContext\n\t\tmockLogger      *log.MockLogger\n\t\tmockMetricScope *mocks.Scope\n\n\t\tprocessor *transferQueueProcessorBase\n\t\tvalidator *transferQueueValidator\n\t}\n)\n\nconst (\n\ttestValidationInterval = 100 * time.Millisecond\n)\n\nfunc TestTransferQueueValidatorSuite(t *testing.T) {\n\ts := new(transferQueueValidatorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transferQueueValidatorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockLogger = log.NewMockLogger(gomock.NewController(s.T()))\n\ts.mockMetricScope = &mocks.Scope{}\n\n\ts.processor = &transferQueueProcessorBase{\n\t\tprocessorBase: &processorBase{\n\t\t\tshard: s.mockShard,\n\t\t\tprocessingQueueCollections: newProcessingQueueCollections(\n\t\t\t\t[]ProcessingQueueState{\n\t\t\t\t\tNewProcessingQueueState(\n\t\t\t\t\t\tdefaultProcessingQueueLevel,\n\t\t\t\t\t\tnewTransferTaskKey(0),\n\t\t\t\t\t\tmaximumTransferTaskKey,\n\t\t\t\t\t\tNewDomainFilter(map[string]struct{}{}, true),\n\t\t\t\t\t),\n\t\t\t\t},\n\t\t\t\tnil,\n\t\t\t\tnil,\n\t\t\t),\n\t\t},\n\t}\n\ts.validator = newTransferQueueValidator(\n\t\ts.processor,\n\t\tdynamicproperties.GetDurationPropertyFn(testValidationInterval),\n\t\ts.mockLogger,\n\t\ts.mockMetricScope,\n\t)\n}\n\nfunc (s *transferQueueValidatorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockMetricScope.AssertExpectations(s.T())\n}\n\nfunc (s *transferQueueValidatorSuite) TestAddTasks_NoTaskDropped() {\n\texecutionInfo := &persistence.WorkflowExecutionInfo{}\n\ttasks := []persistence.Task{\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 0}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 1}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 2}},\n\t}\n\texpectedPendingTasksLen := len(tasks)\n\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: tasks, PersistenceError: false})\n\ts.Len(s.validator.pendingTaskInfos, expectedPendingTasksLen)\n\n\ttasks = []persistence.Task{\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 4}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 5}},\n\t}\n\texpectedPendingTasksLen += len(tasks)\n\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: tasks, PersistenceError: false})\n\ts.Len(s.validator.pendingTaskInfos, expectedPendingTasksLen)\n}\n\nfunc (s *transferQueueValidatorSuite) TestAddTasks_TaskDropped() {\n\texecutionInfo := &persistence.WorkflowExecutionInfo{}\n\ttasks := []persistence.Task{\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 0}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 1}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 2}},\n\t}\n\texpectedPendingTasksLen := len(tasks)\n\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: tasks, PersistenceError: false})\n\ts.Len(s.validator.pendingTaskInfos, expectedPendingTasksLen)\n\n\ttasks = []persistence.Task{}\n\tfor i := 0; i != defaultMaxPendingTasksSize; i++ {\n\t\ttasks = append(tasks, &persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: int64(i + expectedPendingTasksLen)}})\n\t}\n\n\tnumDroppedTasks := expectedPendingTasksLen + len(tasks) - defaultMaxPendingTasksSize\n\ts.mockLogger.EXPECT().Warn(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\ts.mockMetricScope.On(\"AddCounter\", metrics.QueueValidatorDropTaskCounter, int64(numDroppedTasks)).Times(1)\n\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: tasks, PersistenceError: false})\n\ts.Len(s.validator.pendingTaskInfos, defaultMaxPendingTasksSize)\n}\n\nfunc (s *transferQueueValidatorSuite) TestAckTasks_NoTaskLost() {\n\texecutionInfo := &persistence.WorkflowExecutionInfo{}\n\tpendingTasks := []persistence.Task{\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 0}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 1}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 2}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 100}},\n\t}\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: pendingTasks, PersistenceError: false})\n\n\tloadedTasks := make(map[task.Key]task.Task, len(pendingTasks))\n\tfor _, pendingTask := range pendingTasks[:len(pendingTasks)-1] {\n\t\tloadedTasks[newTransferTaskKey(pendingTask.GetTaskID())] = task.NewHistoryTask(\n\t\t\ts.mockShard,\n\t\t\t&persistence.DecisionTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID: pendingTask.GetTaskID(),\n\t\t\t\t},\n\t\t\t},\n\t\t\ttask.QueueTypeActiveTransfer,\n\t\t\tnil, nil, nil, nil, nil, nil,\n\t\t)\n\t}\n\n\ttime.Sleep(testValidationInterval)\n\ts.mockMetricScope.On(\"IncCounter\", metrics.QueueValidatorValidationCounter).Times(1)\n\n\treadLevel := newTransferTaskKey(0)\n\tmaxReadLevel := newTransferTaskKey(10)\n\ts.processor.processingQueueCollections[0].ActiveQueue().State().(*processingQueueStateImpl).readLevel = maxReadLevel\n\ts.validator.ackTasks(defaultProcessingQueueLevel, readLevel, maxReadLevel, loadedTasks)\n\ts.Len(s.validator.pendingTaskInfos, 1)\n}\n\nfunc (s *transferQueueValidatorSuite) TestAckTasks_TaskLost() {\n\texecutionInfo := &persistence.WorkflowExecutionInfo{}\n\tpendingTasks := []persistence.Task{\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 0}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 1}},\n\t\t&persistence.DecisionTask{TaskData: persistence.TaskData{TaskID: 2}},\n\t}\n\ts.validator.addTasks(&hcommon.NotifyTaskInfo{ExecutionInfo: executionInfo, Tasks: pendingTasks, PersistenceError: false})\n\n\tloadedTasks := make(map[task.Key]task.Task, len(pendingTasks))\n\tfor _, pendingTask := range pendingTasks[1:] {\n\t\tloadedTasks[newTransferTaskKey(pendingTask.GetTaskID())] = task.NewHistoryTask(\n\t\t\ts.mockShard,\n\t\t\t&persistence.DecisionTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID: pendingTask.GetTaskID(),\n\t\t\t\t},\n\t\t\t},\n\t\t\ttask.QueueTypeActiveTransfer,\n\t\t\tnil, nil, nil, nil, nil, nil,\n\t\t)\n\t}\n\n\ttime.Sleep(testValidationInterval)\n\ts.mockLogger.EXPECT().Error(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\ts.mockMetricScope.On(\"IncCounter\", metrics.QueueValidatorValidationCounter).Times(1)\n\ts.mockMetricScope.On(\"IncCounter\", metrics.QueueValidatorLostTaskCounter).Times(1)\n\n\treadLevel := newTransferTaskKey(0)\n\tmaxReadLevel := newTransferTaskKey(10)\n\ts.processor.processingQueueCollections[0].ActiveQueue().State().(*processingQueueStateImpl).readLevel = maxReadLevel\n\ts.validator.ackTasks(defaultProcessingQueueLevel, readLevel, maxReadLevel, loadedTasks)\n\ts.Empty(s.validator.pendingTaskInfos)\n}\n\nfunc (s *transferQueueValidatorSuite) TestAckTasks_LostRequestNotContinuous() {\n\treadLevel := newTransferTaskKey(0)\n\tmaxReadLevel := newTransferTaskKey(10)\n\ts.validator.ackTasks(defaultProcessingQueueLevel, readLevel, maxReadLevel, nil)\n\n\treadLevel = newTransferTaskKey(10)\n\tmaxReadLevel = newTransferTaskKey(15)\n\ts.validator.ackTasks(defaultProcessingQueueLevel, readLevel, maxReadLevel, nil)\n\n\ts.mockLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\ts.mockMetricScope.On(\"IncCounter\", metrics.QueueValidatorInvalidLoadCounter).Times(1)\n\n\treadLevel = newTransferTaskKey(16)\n\tmaxReadLevel = newTransferTaskKey(25)\n\ts.validator.ackTasks(defaultProcessingQueueLevel, readLevel, maxReadLevel, nil)\n}\n"
  },
  {
    "path": "service/history/queuev2/alert.go",
    "content": "package queuev2\n\ntype (\n\t// Alert is created by a Monitor when some statistics of the Queue is abnormal\n\tAlert struct {\n\t\tAlertType                            AlertType\n\t\tAlertAttributesQueuePendingTaskCount *AlertAttributesQueuePendingTaskCount\n\t}\n\n\tAlertType int\n\n\tAlertAttributesQueuePendingTaskCount struct {\n\t\tCurrentPendingTaskCount  int\n\t\tCriticalPendingTaskCount int\n\t}\n)\n\nconst (\n\tAlertTypeUnspecified AlertType = iota\n\tAlertTypeQueuePendingTaskCount\n)\n"
  },
  {
    "path": "service/history/queuev2/convert.go",
    "content": "package queuev2\n\nimport (\n\t\"fmt\"\n\t\"slices\"\n\t\"time\"\n\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc FromPersistenceQueueState(state *types.QueueState) *QueueState {\n\tvirtualQueueStates := make(map[int64][]VirtualSliceState)\n\tfor k, v := range state.VirtualQueueStates {\n\t\tvirtualQueueStates[k] = FromPersistenceVirtualQueueState(v)\n\t}\n\treturn &QueueState{\n\t\tVirtualQueueStates:    virtualQueueStates,\n\t\tExclusiveMaxReadLevel: FromPersistenceTaskKey(state.ExclusiveMaxReadLevel),\n\t}\n}\n\nfunc ToPersistenceQueueState(state *QueueState) *types.QueueState {\n\tvirtualQueueStates := make(map[int64]*types.VirtualQueueState)\n\tfor k, v := range state.VirtualQueueStates {\n\t\tvirtualQueueStates[k] = ToPersistenceVirtualQueueState(v)\n\t}\n\treturn &types.QueueState{\n\t\tVirtualQueueStates:    virtualQueueStates,\n\t\tExclusiveMaxReadLevel: ToPersistenceTaskKey(state.ExclusiveMaxReadLevel),\n\t}\n}\n\nfunc FromPersistenceVirtualQueueState(state *types.VirtualQueueState) []VirtualSliceState {\n\tstates := make([]VirtualSliceState, 0, len(state.VirtualSliceStates))\n\tfor _, v := range state.VirtualSliceStates {\n\t\tstates = append(states, FromPersistenceVirtualSliceState(v))\n\t}\n\treturn states\n}\n\nfunc ToPersistenceVirtualQueueState(state []VirtualSliceState) *types.VirtualQueueState {\n\tstates := make([]*types.VirtualSliceState, 0, len(state))\n\tfor _, v := range state {\n\t\tstates = append(states, ToPersistenceVirtualSliceState(v))\n\t}\n\treturn &types.VirtualQueueState{\n\t\tVirtualSliceStates: states,\n\t}\n}\n\nfunc FromPersistenceVirtualSliceState(state *types.VirtualSliceState) VirtualSliceState {\n\treturn VirtualSliceState{\n\t\tRange:     FromPersistenceTaskRange(state.TaskRange),\n\t\tPredicate: FromPersistencePredicate(state.Predicate),\n\t}\n}\n\nfunc ToPersistenceVirtualSliceState(state VirtualSliceState) *types.VirtualSliceState {\n\treturn &types.VirtualSliceState{\n\t\tTaskRange: ToPersistenceTaskRange(state.Range),\n\t\tPredicate: ToPersistencePredicate(state.Predicate),\n\t}\n}\n\nfunc FromPersistenceTaskRange(state *types.TaskRange) Range {\n\treturn Range{\n\t\tInclusiveMinTaskKey: FromPersistenceTaskKey(state.InclusiveMin),\n\t\tExclusiveMaxTaskKey: FromPersistenceTaskKey(state.ExclusiveMax),\n\t}\n}\n\nfunc ToPersistenceTaskRange(r Range) *types.TaskRange {\n\treturn &types.TaskRange{\n\t\tInclusiveMin: ToPersistenceTaskKey(r.InclusiveMinTaskKey),\n\t\tExclusiveMax: ToPersistenceTaskKey(r.ExclusiveMaxTaskKey),\n\t}\n}\n\nfunc FromPersistenceTaskKey(key *types.TaskKey) persistence.HistoryTaskKey {\n\treturn persistence.NewHistoryTaskKey(time.Unix(0, key.ScheduledTimeNano).UTC(), key.TaskID)\n}\n\nfunc ToPersistenceTaskKey(key persistence.HistoryTaskKey) *types.TaskKey {\n\treturn &types.TaskKey{\n\t\tTaskID:            key.GetTaskID(),\n\t\tScheduledTimeNano: key.GetScheduledTime().UnixNano(),\n\t}\n}\n\nfunc FromPersistencePredicate(predicate *types.Predicate) Predicate {\n\tif predicate == nil {\n\t\treturn NewUniversalPredicate()\n\t}\n\tswitch predicate.PredicateType {\n\tcase types.PredicateTypeUniversal:\n\t\treturn NewUniversalPredicate()\n\tcase types.PredicateTypeEmpty:\n\t\treturn NewEmptyPredicate()\n\tcase types.PredicateTypeDomainID:\n\t\treturn NewDomainIDPredicate(predicate.GetDomainIDPredicateAttributes().DomainIDs, predicate.GetDomainIDPredicateAttributes().GetIsExclusive())\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown predicate type: %v\", predicate.PredicateType))\n\t}\n}\n\nfunc ToPersistencePredicate(predicate Predicate) *types.Predicate {\n\tswitch p := predicate.(type) {\n\tcase *universalPredicate:\n\t\treturn &types.Predicate{PredicateType: types.PredicateTypeUniversal, UniversalPredicateAttributes: &types.UniversalPredicateAttributes{}}\n\tcase *emptyPredicate:\n\t\treturn &types.Predicate{PredicateType: types.PredicateTypeEmpty, EmptyPredicateAttributes: &types.EmptyPredicateAttributes{}}\n\tcase *domainIDPredicate:\n\t\tdomainIDs := maps.Keys(p.domainIDs)\n\t\tslices.Sort(domainIDs)\n\t\treturn &types.Predicate{PredicateType: types.PredicateTypeDomainID, DomainIDPredicateAttributes: &types.DomainIDPredicateAttributes{DomainIDs: domainIDs, IsExclusive: &p.isExclusive}}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p))\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/convert_test.go",
    "content": "package queuev2\n\nimport (\n\t\"math/rand\"\n\t\"sort\"\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestConvertTaskKey(t *testing.T) {\n\tf := fuzz.New()\n\tfor i := 0; i < 1000; i++ {\n\t\tvar key types.TaskKey\n\t\tf.Fuzz(&key)\n\t\tpersistenceKey := FromPersistenceTaskKey(&key)\n\t\tconvertedKey := ToPersistenceTaskKey(persistenceKey)\n\t\tassert.Equal(t, key, *convertedKey)\n\t}\n}\n\nfunc TestConvertTaskRange(t *testing.T) {\n\tf := fuzz.New().NilChance(0)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar r types.TaskRange\n\t\tf.Fuzz(&r)\n\t\tpersistenceRange := FromPersistenceTaskRange(&r)\n\t\tconvertedRange := ToPersistenceTaskRange(persistenceRange)\n\t\tassert.Equal(t, r, *convertedRange)\n\t}\n}\n\nfunc predicateFuzzGenerator(t *types.Predicate, c fuzz.Continue) {\n\tswitch c.Intn(int(types.NumPredicateTypes)) {\n\tcase 0:\n\t\tt.PredicateType = types.PredicateTypeUniversal\n\t\tc.Fuzz(&t.UniversalPredicateAttributes)\n\tcase 1:\n\t\tt.PredicateType = types.PredicateTypeEmpty\n\t\tc.Fuzz(&t.EmptyPredicateAttributes)\n\tcase 2:\n\t\tt.PredicateType = types.PredicateTypeDomainID\n\t\tc.Fuzz(&t.DomainIDPredicateAttributes)\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc domainIDPredicateAttributesFuzzGenerator(t *types.DomainIDPredicateAttributes, c fuzz.Continue) {\n\tconst maxCount = 10 // adjust as needed\n\tcount := rand.Intn(maxCount) + 1\n\n\tseen := make(map[string]struct{})\n\tt.DomainIDs = make([]string, 0, count)\n\n\tfor len(t.DomainIDs) < count {\n\t\tvar s string\n\t\tc.Fuzz(&s)\n\t\tif s != \"\" && len(s) >= 3 {\n\t\t\tif _, exists := seen[s]; !exists {\n\t\t\t\tseen[s] = struct{}{}\n\t\t\t\tt.DomainIDs = append(t.DomainIDs, s)\n\t\t\t}\n\t\t}\n\t}\n\n\tsort.Strings(t.DomainIDs)\n\n\tvar b bool\n\tc.Fuzz(&b)\n\tt.IsExclusive = &b\n}\n\nfunc TestConvertVirtualSliceState(t *testing.T) {\n\tf := fuzz.New().NilChance(0).Funcs(predicateFuzzGenerator, domainIDPredicateAttributesFuzzGenerator)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar s types.VirtualSliceState\n\t\tf.Fuzz(&s)\n\t\tpersistenceState := FromPersistenceVirtualSliceState(&s)\n\t\tconvertedState := ToPersistenceVirtualSliceState(persistenceState)\n\t\tassert.Equal(t, s, *convertedState)\n\t}\n}\n\nfunc TestConvertVirtualQueueState(t *testing.T) {\n\tf := fuzz.New().NilChance(0).Funcs(predicateFuzzGenerator, domainIDPredicateAttributesFuzzGenerator)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar s types.VirtualQueueState\n\t\tf.Fuzz(&s)\n\t\tpersistenceState := FromPersistenceVirtualQueueState(&s)\n\t\tconvertedState := ToPersistenceVirtualQueueState(persistenceState)\n\t\tassert.Equal(t, s, *convertedState)\n\t}\n}\n\nfunc TestConvertQueueState(t *testing.T) {\n\tf := fuzz.New().NilChance(0).Funcs(predicateFuzzGenerator, domainIDPredicateAttributesFuzzGenerator)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar s types.QueueState\n\t\tf.Fuzz(&s)\n\t\tpersistenceState := FromPersistenceQueueState(&s)\n\t\tconvertedState := ToPersistenceQueueState(persistenceState)\n\t\tassert.Equal(t, s, *convertedState)\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/queue\"\n)\n\ntype (\n\tQueue interface {\n\t\tcommon.Daemon\n\t\tCategory() persistence.HistoryTaskCategory\n\t\tNotifyNewTask(string, *hcommon.NotifyTaskInfo)\n\n\t\tFailoverDomain(map[string]struct{})\n\t\tHandleAction(context.Context, string, *queue.Action) (*queue.ActionResult, error)\n\t\tLockTaskProcessing()\n\t\tUnlockTaskProcessing()\n\t}\n)\n"
  },
  {
    "path": "service/history/queuev2/mitigator.go",
    "content": "//go:generate mockgen -package $GOPACKAGE -destination mitigator_mock.go github.com/uber/cadence/service/history/queuev2 Mitigator\npackage queuev2\n\nimport (\n\t\"maps\"\n\t\"slices\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\ttargetLoadFactor           = 0.8\n\tclearSliceThrottleDuration = 10 * time.Second\n)\n\ntype (\n\tMitigator interface {\n\t\tMitigate(Alert)\n\t}\n\n\tMitigatorOptions struct {\n\t\tMaxVirtualQueueCount dynamicproperties.IntPropertyFn\n\t}\n\n\tmitigatorImpl struct {\n\t\tvirtualQueueManager VirtualQueueManager\n\t\tmonitor             Monitor\n\t\tlogger              log.Logger\n\t\tmetricsScope        metrics.Scope\n\t\toptions             *MitigatorOptions\n\n\t\thandlers map[AlertType]func(Alert)\n\t}\n\n\tpendingTaskStats struct {\n\t\ttotalPendingTaskCount             int\n\t\tpendingTaskCountPerDomain         map[string]int\n\t\tpendingTaskCountPerDomainPerSlice map[VirtualSlice]map[string]int\n\t\tslicesPerDomain                   map[string][]VirtualSlice\n\t}\n)\n\nfunc NewMitigator(\n\tvirtualQueueManager VirtualQueueManager,\n\tmonitor Monitor,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n\toptions *MitigatorOptions,\n) Mitigator {\n\tm := &mitigatorImpl{\n\t\tvirtualQueueManager: virtualQueueManager,\n\t\tmonitor:             monitor,\n\t\tlogger:              logger,\n\t\tmetricsScope:        metricsScope,\n\t\toptions:             options,\n\t}\n\tm.handlers = map[AlertType]func(Alert){\n\t\tAlertTypeQueuePendingTaskCount: m.handleQueuePendingTaskCount,\n\t}\n\treturn m\n}\n\nfunc (m *mitigatorImpl) Mitigate(alert Alert) {\n\thandler, ok := m.handlers[alert.AlertType]\n\tif !ok {\n\t\tm.logger.Error(\"unknown queue alert type\", tag.AlertType(int(alert.AlertType)))\n\t\treturn\n\t}\n\thandler(alert)\n\n\tm.monitor.ResolveAlert(alert.AlertType)\n\tm.logger.Info(\"mitigated queue alert\", tag.AlertType(int(alert.AlertType)))\n}\n\nfunc (m *mitigatorImpl) handleQueuePendingTaskCount(alert Alert) {\n\t// First, try cleaning up tasks that has already been acknowledged to see if we can reduce the pending task count\n\tvirtualQueues := m.virtualQueueManager.VirtualQueues()\n\tfor _, virtualQueue := range virtualQueues {\n\t\tvirtualQueue.UpdateAndGetState()\n\t}\n\tif m.monitor.GetTotalPendingTaskCount() <= alert.AlertAttributesQueuePendingTaskCount.CriticalPendingTaskCount {\n\t\tm.logger.Debug(\"mitigating queue alert, skip mitigation because the alert is no longer valid\")\n\t\treturn\n\t}\n\t// Second, getting the stats of pending tasks. We need:\n\tstats := m.collectPendingTaskStats()\n\n\t// Third, find virtual slices to split given the target pending task count and the stats of pending tasks\n\ttargetPendingTaskCount := int(float64(alert.AlertAttributesQueuePendingTaskCount.CriticalPendingTaskCount) * targetLoadFactor)\n\tif m.logger.DebugOn() {\n\t\tsliceStatesPerDomain := make(map[string][]*types.VirtualSliceState)\n\t\tfor domain, slices := range stats.slicesPerDomain {\n\t\t\tfor _, s := range slices {\n\t\t\t\tsliceStatesPerDomain[domain] = append(sliceStatesPerDomain[domain], ToPersistenceVirtualSliceState(s.GetState()))\n\t\t\t}\n\t\t}\n\t\tfor s, domainStats := range stats.pendingTaskCountPerDomainPerSlice {\n\t\t\tm.logger.Debug(\"mitigating queue alert, get task stats per slice\", tag.Dynamic(\"slice\", ToPersistenceVirtualSliceState(s.GetState())), tag.Dynamic(\"domain-stats\", domainStats))\n\t\t}\n\t\tm.logger.Debug(\"mitigating queue alert, get task stats\",\n\t\t\ttag.AlertType(int(alert.AlertType)),\n\t\t\ttag.Dynamic(\"pending-task-count-per-domain\", stats.pendingTaskCountPerDomain),\n\t\t\ttag.Dynamic(\"slices-per-domain\", sliceStatesPerDomain),\n\t\t\ttag.Dynamic(\"pending-task-count\", stats.totalPendingTaskCount),\n\t\t\ttag.Dynamic(\"target-task-count\", targetPendingTaskCount),\n\t\t)\n\t}\n\tdomainsToClearPerSlice := m.findDomainsToClear(stats, targetPendingTaskCount)\n\tif m.logger.DebugOn() {\n\t\tfor s, domains := range domainsToClearPerSlice {\n\t\t\tm.logger.Debug(\"mitigating queue alert, get domains to clear\", tag.Dynamic(\"slice\", ToPersistenceVirtualSliceState(s.GetState())), tag.WorkflowDomainIDs(domains))\n\t\t}\n\t}\n\n\t// Finally, split and clear the slices\n\tm.processQueueSplitsAndClear(virtualQueues, domainsToClearPerSlice)\n\tif m.logger.DebugOn() {\n\t\tvirtualQueues := m.virtualQueueManager.VirtualQueues()\n\t\tstate := make(map[int64]*types.VirtualQueueState)\n\t\tfor queueID, vq := range virtualQueues {\n\t\t\tstate[queueID] = ToPersistenceVirtualQueueState(vq.GetState())\n\t\t}\n\t\tm.logger.Debug(\"mitigating queue alert, get queue state after mitigation\", tag.Dynamic(\"queue-state\", state))\n\t}\n}\n\n// The stats of pending tasks are used to calculate the domains to clear. We need:\n// 1. The total number of pending tasks per domain\n// 2. The number of pending tasks per domain per slice\n// 3. The slices that contains the tasks for each domain\nfunc (m *mitigatorImpl) collectPendingTaskStats() pendingTaskStats {\n\tstats := pendingTaskStats{\n\t\tpendingTaskCountPerDomain:         make(map[string]int),\n\t\tpendingTaskCountPerDomainPerSlice: make(map[VirtualSlice]map[string]int),\n\t\tslicesPerDomain:                   make(map[string][]VirtualSlice),\n\t}\n\n\tfor _, virtualQueue := range m.virtualQueueManager.VirtualQueues() {\n\t\tvirtualQueue.IterateSlices(func(slice VirtualSlice) {\n\t\t\tperDomain := slice.PendingTaskStats().PendingTaskCountPerDomain\n\t\t\tstats.pendingTaskCountPerDomainPerSlice[slice] = perDomain\n\t\t\tfor domain, count := range perDomain {\n\t\t\t\tstats.totalPendingTaskCount += count\n\t\t\t\tstats.pendingTaskCountPerDomain[domain] += count\n\t\t\t\tstats.slicesPerDomain[domain] = append(stats.slicesPerDomain[domain], slice)\n\t\t\t}\n\t\t})\n\t}\n\n\tfor _, slicesList := range stats.slicesPerDomain {\n\t\tslices.SortFunc(slicesList, func(a, b VirtualSlice) int {\n\t\t\treturn b.GetState().Range.InclusiveMinTaskKey.Compare(a.GetState().Range.InclusiveMinTaskKey)\n\t\t})\n\t}\n\treturn stats\n}\n\nfunc (m *mitigatorImpl) findDomainsToClear(stats pendingTaskStats, targetCount int) map[VirtualSlice][]string {\n\tdomainsToClear := make(map[VirtualSlice][]string)\n\n\tpq := collection.NewPriorityQueue(\n\t\tfunc(a, b string) bool {\n\t\t\treturn stats.pendingTaskCountPerDomain[a] > stats.pendingTaskCountPerDomain[b]\n\t\t},\n\t\tslices.Collect(maps.Keys(stats.pendingTaskCountPerDomain))...,\n\t)\n\n\tfor stats.totalPendingTaskCount > targetCount && !pq.IsEmpty() {\n\t\tdomain, err := pq.Remove()\n\t\tif err != nil {\n\t\t\t// this should never happen because we check the priority queue is not empty before calling Remove\n\t\t\t// but just want to be future proof\n\t\t\tm.logger.Error(\"failed to remove domain from priority queue with unexpected error\", tag.Error(err))\n\t\t\tpanic(err)\n\t\t}\n\t\tif len(stats.slicesPerDomain[domain]) == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tslice := stats.slicesPerDomain[domain][0]\n\t\tstats.slicesPerDomain[domain] = stats.slicesPerDomain[domain][1:]\n\n\t\ttaskCount := stats.pendingTaskCountPerDomainPerSlice[slice][domain]\n\t\tstats.totalPendingTaskCount -= taskCount\n\t\tstats.pendingTaskCountPerDomain[domain] -= taskCount\n\t\tif stats.pendingTaskCountPerDomain[domain] > 0 {\n\t\t\tpq.Add(domain)\n\t\t}\n\t\tdomainsToClear[slice] = append(domainsToClear[slice], domain)\n\t}\n\treturn domainsToClear\n}\n\nfunc (m *mitigatorImpl) processQueueSplitsAndClear(virtualQueues map[int64]VirtualQueue, domainsToClear map[VirtualSlice][]string) {\n\tmaxQueueID := m.options.MaxVirtualQueueCount() - 1\n\tfor queueID, vq := range virtualQueues {\n\t\tif queueID >= int64(maxQueueID) {\n\t\t\t// Clear slices in the last queue\n\t\t\tcleared := false\n\t\t\tvq.ClearSlices(func(slice VirtualSlice) bool {\n\t\t\t\t_, ok := domainsToClear[slice]\n\t\t\t\tcleared = cleared || ok\n\t\t\t\treturn ok\n\t\t\t})\n\t\t\tif cleared {\n\t\t\t\tvq.Pause(clearSliceThrottleDuration)\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\n\t\tvar slicesToMove []VirtualSlice\n\t\tvq.SplitSlices(func(slice VirtualSlice) ([]VirtualSlice, bool) {\n\t\t\tdomains := domainsToClear[slice]\n\t\t\tif len(domains) == 0 {\n\t\t\t\treturn nil, false\n\t\t\t}\n\t\t\tpredicate := NewDomainIDPredicate(domains, false)\n\t\t\tsplitSlice, remainingSlice, ok := slice.TrySplitByPredicate(predicate)\n\t\t\tif !ok {\n\t\t\t\tslice.Clear()\n\t\t\t\tslicesToMove = append(slicesToMove, slice)\n\t\t\t\treturn nil, true\n\t\t\t}\n\t\t\tsplitSlice.Clear()\n\t\t\tslicesToMove = append(slicesToMove, splitSlice)\n\t\t\treturn []VirtualSlice{remainingSlice}, true\n\t\t})\n\n\t\tif len(slicesToMove) > 0 {\n\t\t\tnextQueue := m.virtualQueueManager.GetOrCreateVirtualQueue(queueID + 1)\n\t\t\tnextQueue.Pause(clearSliceThrottleDuration)\n\t\t\tnextQueue.MergeSlices(slicesToMove...)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/mitigator_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: Mitigator)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination mitigator_mock.go github.com/uber/cadence/service/history/queuev2 Mitigator\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockMitigator is a mock of Mitigator interface.\ntype MockMitigator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMitigatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockMitigatorMockRecorder is the mock recorder for MockMitigator.\ntype MockMitigatorMockRecorder struct {\n\tmock *MockMitigator\n}\n\n// NewMockMitigator creates a new mock instance.\nfunc NewMockMitigator(ctrl *gomock.Controller) *MockMitigator {\n\tmock := &MockMitigator{ctrl: ctrl}\n\tmock.recorder = &MockMitigatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMitigator) EXPECT() *MockMitigatorMockRecorder {\n\treturn m.recorder\n}\n\n// Mitigate mocks base method.\nfunc (m *MockMitigator) Mitigate(arg0 Alert) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Mitigate\", arg0)\n}\n\n// Mitigate indicates an expected call of Mitigate.\nfunc (mr *MockMitigatorMockRecorder) Mitigate(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Mitigate\", reflect.TypeOf((*MockMitigator)(nil).Mitigate), arg0)\n}\n"
  },
  {
    "path": "service/history/queuev2/mitigator_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestNewMitigator(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\tlogger := testlogger.New(t)\n\tmetricsScope := metrics.NoopScope\n\toptions := &MitigatorOptions{\n\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(10),\n\t}\n\n\tmitigator := NewMitigator(\n\t\tmockVirtualQueueManager,\n\t\tmockMonitor,\n\t\tlogger,\n\t\tmetricsScope,\n\t\toptions,\n\t)\n\n\trequire.NotNil(t, mitigator)\n\n\t// Verify internal structure\n\timpl, ok := mitigator.(*mitigatorImpl)\n\trequire.True(t, ok)\n\tassert.Equal(t, mockMonitor, impl.monitor)\n\tassert.Equal(t, logger, impl.logger)\n\tassert.Equal(t, metricsScope, impl.metricsScope)\n\tassert.Equal(t, options, impl.options)\n\n\t// Verify handlers are properly initialized\n\tassert.NotNil(t, impl.handlers)\n\tassert.Len(t, impl.handlers, 1)\n\t_, exists := impl.handlers[AlertTypeQueuePendingTaskCount]\n\tassert.True(t, exists)\n}\n\nfunc TestMitigator_Mitigate_KnownAlertType(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\tlogger := testlogger.New(t)\n\tmetricsScope := metrics.NoopScope\n\toptions := &MitigatorOptions{\n\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(10),\n\t}\n\n\tmitigator := NewMitigator(\n\t\tmockVirtualQueueManager,\n\t\tmockMonitor,\n\t\tlogger,\n\t\tmetricsScope,\n\t\toptions,\n\t)\n\timpl, ok := mitigator.(*mitigatorImpl)\n\trequire.True(t, ok)\n\thandlerCalled := false\n\timpl.handlers[AlertTypeQueuePendingTaskCount] = func(alert Alert) {\n\t\thandlerCalled = true\n\t}\n\n\talert := Alert{\n\t\tAlertType: AlertTypeQueuePendingTaskCount,\n\t\tAlertAttributesQueuePendingTaskCount: &AlertAttributesQueuePendingTaskCount{\n\t\t\tCurrentPendingTaskCount:  150,\n\t\t\tCriticalPendingTaskCount: 100,\n\t\t},\n\t}\n\n\t// Expect ResolveAlert to be called on the monitor\n\tmockMonitor.EXPECT().ResolveAlert(AlertTypeQueuePendingTaskCount).Times(1)\n\n\tmitigator.Mitigate(alert)\n\tassert.True(t, handlerCalled)\n}\n\nfunc TestMitigator_Mitigate_UnknownAlertType(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\tlogger := testlogger.New(t)\n\tmetricsScope := metrics.NoopScope\n\toptions := &MitigatorOptions{\n\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(10),\n\t}\n\n\tmitigator := NewMitigator(\n\t\tmockVirtualQueueManager,\n\t\tmockMonitor,\n\t\tlogger,\n\t\tmetricsScope,\n\t\toptions,\n\t)\n\n\t// Create an alert with an unknown/unhandled alert type\n\tunknownAlertType := AlertType(999)\n\talert := Alert{\n\t\tAlertType: unknownAlertType,\n\t}\n\n\tmitigator.Mitigate(alert)\n}\n\nfunc TestMitigator_collectPendingTaskStats(t *testing.T) {\n\ttests := []struct {\n\t\tname                              string\n\t\tsetupMocks                        func(*gomock.Controller) (*MockVirtualQueueManager, map[string][]VirtualSlice, map[VirtualSlice]map[string]int)\n\t\texpectedTotalPendingTaskCount     int\n\t\texpectedPendingTaskCountPerDomain map[string]int\n\t\texpectedSlicesPerDomainLength     map[string]int\n\t\tvalidateResults                   func(*testing.T, pendingTaskStats, map[string][]VirtualSlice, map[VirtualSlice]map[string]int)\n\t}{\n\t\t{\n\t\t\tname: \"empty virtual queues\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*MockVirtualQueueManager, map[string][]VirtualSlice, map[VirtualSlice]map[string]int) {\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockVirtualQueueManager.EXPECT().VirtualQueues().Return(map[int64]VirtualQueue{}).Times(1)\n\t\t\t\treturn mockVirtualQueueManager, map[string][]VirtualSlice{}, map[VirtualSlice]map[string]int{}\n\t\t\t},\n\t\t\texpectedTotalPendingTaskCount:     0,\n\t\t\texpectedPendingTaskCountPerDomain: map[string]int{},\n\t\t\texpectedSlicesPerDomainLength:     map[string]int{},\n\t\t\tvalidateResults: func(t *testing.T, stats pendingTaskStats, expectedSlicesPerDomain map[string][]VirtualSlice, expectedPendingTaskCountPerDomainPerSlice map[VirtualSlice]map[string]int) {\n\t\t\t\tassert.Empty(t, stats.pendingTaskCountPerDomain)\n\t\t\t\tassert.Empty(t, stats.pendingTaskCountPerDomainPerSlice)\n\t\t\t\tassert.Empty(t, stats.slicesPerDomain)\n\t\t\t\tassert.Empty(t, stats.pendingTaskCountPerDomainPerSlice)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"single queue single slice\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*MockVirtualQueueManager, map[string][]VirtualSlice, map[VirtualSlice]map[string]int) {\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockVirtualQueue := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockVirtualSlice := NewMockVirtualSlice(ctrl)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{0: mockVirtualQueue}\n\t\t\t\tpendingTaskStats := PendingTaskStats{\n\t\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\t\"domain1\": 10,\n\t\t\t\t\t\t\"domain2\": 5,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().VirtualQueues().Return(virtualQueues).Times(1)\n\t\t\t\tmockVirtualQueue.EXPECT().IterateSlices(gomock.Any()).DoAndReturn(func(f func(VirtualSlice)) {\n\t\t\t\t\tf(mockVirtualSlice)\n\t\t\t\t}).Times(1)\n\t\t\t\tmockVirtualSlice.EXPECT().PendingTaskStats().Return(pendingTaskStats).Times(1)\n\t\t\t\tmockVirtualSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\texpectedSlicesPerDomain := map[string][]VirtualSlice{\n\t\t\t\t\t\"domain1\": {mockVirtualSlice},\n\t\t\t\t\t\"domain2\": {mockVirtualSlice},\n\t\t\t\t}\n\n\t\t\t\texpectedPendingTaskCountPerDomainPerSlice := map[VirtualSlice]map[string]int{\n\t\t\t\t\tmockVirtualSlice: {\n\t\t\t\t\t\t\"domain1\": 10,\n\t\t\t\t\t\t\"domain2\": 5,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\treturn mockVirtualQueueManager, expectedSlicesPerDomain, expectedPendingTaskCountPerDomainPerSlice\n\t\t\t},\n\t\t\texpectedTotalPendingTaskCount: 15,\n\t\t\texpectedPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\"domain1\": 10,\n\t\t\t\t\"domain2\": 5,\n\t\t\t},\n\t\t\texpectedSlicesPerDomainLength: map[string]int{\n\t\t\t\t\"domain1\": 1,\n\t\t\t\t\"domain2\": 1,\n\t\t\t},\n\t\t\tvalidateResults: func(t *testing.T, stats pendingTaskStats, expectedSlicesPerDomain map[string][]VirtualSlice, expectedPendingTaskCountPerDomainPerSlice map[VirtualSlice]map[string]int) {\n\t\t\t\tassert.Equal(t, expectedSlicesPerDomain, stats.slicesPerDomain)\n\t\t\t\tassert.Equal(t, expectedPendingTaskCountPerDomainPerSlice, stats.pendingTaskCountPerDomainPerSlice)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple queues multiple slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*MockVirtualQueueManager, map[string][]VirtualSlice, map[VirtualSlice]map[string]int) {\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockVirtualQueue1 := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockVirtualQueue2 := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVirtualSlice3 := NewMockVirtualSlice(ctrl)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t0: mockVirtualQueue1,\n\t\t\t\t\t1: mockVirtualQueue2,\n\t\t\t\t}\n\n\t\t\t\tpendingTaskStats1 := PendingTaskStats{\n\t\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\t\"domain1\": 10,\n\t\t\t\t\t\t\"domain2\": 5,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tpendingTaskStats2 := PendingTaskStats{\n\t\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\t\"domain1\": 8,\n\t\t\t\t\t\t\"domain3\": 3,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tpendingTaskStats3 := PendingTaskStats{\n\t\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\t\"domain2\": 7,\n\t\t\t\t\t\t\"domain3\": 4,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().VirtualQueues().Return(virtualQueues).Times(1)\n\t\t\t\tmockVirtualQueue1.EXPECT().IterateSlices(gomock.Any()).DoAndReturn(func(f func(VirtualSlice)) {\n\t\t\t\t\tf(mockVirtualSlice1)\n\t\t\t\t\tf(mockVirtualSlice2)\n\t\t\t\t}).Times(1)\n\t\t\t\tmockVirtualQueue2.EXPECT().IterateSlices(gomock.Any()).DoAndReturn(func(f func(VirtualSlice)) {\n\t\t\t\t\tf(mockVirtualSlice3)\n\t\t\t\t}).Times(1)\n\n\t\t\t\tmockVirtualSlice1.EXPECT().PendingTaskStats().Return(pendingTaskStats1).Times(1)\n\t\t\t\tmockVirtualSlice2.EXPECT().PendingTaskStats().Return(pendingTaskStats2).Times(1)\n\t\t\t\tmockVirtualSlice3.EXPECT().PendingTaskStats().Return(pendingTaskStats3).Times(1)\n\n\t\t\t\tmockVirtualSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmockVirtualSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmockVirtualSlice3.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(400),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\t// Expected slices sorted by InclusiveMinTaskKey in descending order\n\t\t\t\t// slice1=0, slice2=200, slice3=100 -> sorted: slice2(200), slice3(100), slice1(0)\n\t\t\t\texpectedSlicesPerDomain := map[string][]VirtualSlice{\n\t\t\t\t\t\"domain1\": {mockVirtualSlice2, mockVirtualSlice1}, // slice2[200] > slice1[0]\n\t\t\t\t\t\"domain2\": {mockVirtualSlice3, mockVirtualSlice1}, // slice3[100] > slice1[0]\n\t\t\t\t\t\"domain3\": {mockVirtualSlice2, mockVirtualSlice3}, // slice2[200] > slice3[100]\n\t\t\t\t}\n\n\t\t\t\texpectedPendingTaskCountPerDomainPerSlice := map[VirtualSlice]map[string]int{\n\t\t\t\t\tmockVirtualSlice1: {\n\t\t\t\t\t\t\"domain1\": 10,\n\t\t\t\t\t\t\"domain2\": 5,\n\t\t\t\t\t},\n\t\t\t\t\tmockVirtualSlice2: {\n\t\t\t\t\t\t\"domain1\": 8,\n\t\t\t\t\t\t\"domain3\": 3,\n\t\t\t\t\t},\n\t\t\t\t\tmockVirtualSlice3: {\n\t\t\t\t\t\t\"domain2\": 7,\n\t\t\t\t\t\t\"domain3\": 4,\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\treturn mockVirtualQueueManager, expectedSlicesPerDomain, expectedPendingTaskCountPerDomainPerSlice\n\t\t\t},\n\t\t\texpectedTotalPendingTaskCount: 37, // 10+5 + 8+3 + 7+4 = 37\n\t\t\texpectedPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\"domain1\": 18, // 10+8\n\t\t\t\t\"domain2\": 12, // 5+7\n\t\t\t\t\"domain3\": 7,  // 3+4\n\t\t\t},\n\t\t\texpectedSlicesPerDomainLength: map[string]int{\n\t\t\t\t\"domain1\": 2, // slice1, slice2\n\t\t\t\t\"domain2\": 2, // slice1, slice3\n\t\t\t\t\"domain3\": 2, // slice2, slice3\n\t\t\t},\n\t\t\tvalidateResults: func(t *testing.T, stats pendingTaskStats, expectedSlicesPerDomain map[string][]VirtualSlice, expectedPendingTaskCountPerDomainPerSlice map[VirtualSlice]map[string]int) {\n\t\t\t\t// Verify the exact slice order\n\t\t\t\tassert.Equal(t, expectedSlicesPerDomain, stats.slicesPerDomain)\n\t\t\t\t// Verify the pendingTaskCountPerDomainPerSlice\n\t\t\t\tassert.Equal(t, expectedPendingTaskCountPerDomainPerSlice, stats.pendingTaskCountPerDomainPerSlice)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*MockVirtualQueueManager, map[string][]VirtualSlice, map[VirtualSlice]map[string]int) {\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockVirtualQueue := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockVirtualSlice := NewMockVirtualSlice(ctrl)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{0: mockVirtualQueue}\n\t\t\t\temptyPendingTaskStats := PendingTaskStats{\n\t\t\t\t\tPendingTaskCountPerDomain: map[string]int{},\n\t\t\t\t}\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().VirtualQueues().Return(virtualQueues).Times(1)\n\t\t\t\tmockVirtualQueue.EXPECT().IterateSlices(gomock.Any()).DoAndReturn(func(f func(VirtualSlice)) {\n\t\t\t\t\tf(mockVirtualSlice)\n\t\t\t\t}).Times(1)\n\t\t\t\tmockVirtualSlice.EXPECT().PendingTaskStats().Return(emptyPendingTaskStats).Times(1)\n\t\t\t\tmockVirtualSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\texpectedSlicesPerDomain := map[string][]VirtualSlice{}\n\t\t\t\texpectedPendingTaskCountPerDomainPerSlice := map[VirtualSlice]map[string]int{\n\t\t\t\t\tmockVirtualSlice: {},\n\t\t\t\t}\n\n\t\t\t\treturn mockVirtualQueueManager, expectedSlicesPerDomain, expectedPendingTaskCountPerDomainPerSlice\n\t\t\t},\n\t\t\texpectedTotalPendingTaskCount:     0,\n\t\t\texpectedPendingTaskCountPerDomain: map[string]int{},\n\t\t\texpectedSlicesPerDomainLength:     map[string]int{},\n\t\t\tvalidateResults: func(t *testing.T, stats pendingTaskStats, expectedSlicesPerDomain map[string][]VirtualSlice, expectedPendingTaskCountPerDomainPerSlice map[VirtualSlice]map[string]int) {\n\t\t\t\tassert.Empty(t, stats.slicesPerDomain)\n\t\t\t\tassert.Equal(t, expectedPendingTaskCountPerDomainPerSlice, stats.pendingTaskCountPerDomainPerSlice)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tmetricsScope := metrics.NoopScope\n\t\t\toptions := &MitigatorOptions{\n\t\t\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(10),\n\t\t\t}\n\n\t\t\tmockVirtualQueueManager, expectedSlicesPerDomain, expectedPendingTaskCountPerDomainPerSlice := tt.setupMocks(ctrl)\n\n\t\t\tmitigator := NewMitigator(\n\t\t\t\tmockVirtualQueueManager,\n\t\t\t\tmockMonitor,\n\t\t\t\tlogger,\n\t\t\t\tmetricsScope,\n\t\t\t\toptions,\n\t\t\t)\n\n\t\t\timpl, ok := mitigator.(*mitigatorImpl)\n\t\t\trequire.True(t, ok)\n\t\t\timpl.virtualQueueManager = mockVirtualQueueManager\n\n\t\t\tstats := impl.collectPendingTaskStats()\n\n\t\t\t// Verify basic aggregated data\n\t\t\tassert.Equal(t, tt.expectedTotalPendingTaskCount, stats.totalPendingTaskCount)\n\t\t\tassert.Equal(t, tt.expectedPendingTaskCountPerDomain, stats.pendingTaskCountPerDomain)\n\n\t\t\t// Verify slices per domain lengths\n\t\t\tfor domain, expectedLength := range tt.expectedSlicesPerDomainLength {\n\t\t\t\tassert.Len(t, stats.slicesPerDomain[domain], expectedLength, \"domain: %s\", domain)\n\t\t\t}\n\n\t\t\t// Run custom validation if provided\n\t\t\tif tt.validateResults != nil {\n\t\t\t\ttt.validateResults(t, stats, expectedSlicesPerDomain, expectedPendingTaskCountPerDomainPerSlice)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMitigator_findDomainsToClear(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tsetupStats  func(*gomock.Controller) (pendingTaskStats, map[VirtualSlice][]string)\n\t\ttargetCount int\n\t}{\n\t\t{\n\t\t\tname: \"target count zero - clear everything\",\n\t\t\tsetupStats: func(ctrl *gomock.Controller) (pendingTaskStats, map[VirtualSlice][]string) {\n\t\t\t\tmockSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSlice3 := NewMockVirtualSlice(ctrl)\n\t\t\t\tstats := pendingTaskStats{\n\t\t\t\t\ttotalPendingTaskCount: 142,\n\t\t\t\t\tpendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\t\"domain1\": 35, // higher priority\n\t\t\t\t\t\t\"domain2\": 45,\n\t\t\t\t\t\t\"domain3\": 62,\n\t\t\t\t\t},\n\t\t\t\t\tpendingTaskCountPerDomainPerSlice: map[VirtualSlice]map[string]int{\n\t\t\t\t\t\tmockSlice1: {\"domain1\": 21, \"domain2\": 34, \"domain3\": 55},\n\t\t\t\t\t\tmockSlice2: {\"domain1\": 13, \"domain2\": 8, \"domain3\": 5},\n\t\t\t\t\t\tmockSlice3: {\"domain1\": 1, \"domain2\": 3, \"domain3\": 2},\n\t\t\t\t\t},\n\t\t\t\t\tslicesPerDomain: map[string][]VirtualSlice{\n\t\t\t\t\t\t\"domain1\": {mockSlice1, mockSlice2, mockSlice3},\n\t\t\t\t\t\t\"domain2\": {mockSlice2, mockSlice3, mockSlice1},\n\t\t\t\t\t\t\"domain3\": {mockSlice3, mockSlice1, mockSlice2},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\texpectedResult := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice1: {\"domain3\", \"domain1\", \"domain2\"},\n\t\t\t\t\tmockSlice2: {\"domain2\", \"domain1\", \"domain3\"},\n\t\t\t\t\tmockSlice3: {\"domain3\", \"domain2\", \"domain1\"},\n\t\t\t\t}\n\t\t\t\treturn stats, expectedResult\n\t\t\t},\n\t\t\ttargetCount: 0,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tmetricsScope := metrics.NoopScope\n\t\t\toptions := &MitigatorOptions{\n\t\t\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(10),\n\t\t\t}\n\n\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\tmitigator := NewMitigator(\n\t\t\t\tmockVirtualQueueManager,\n\t\t\t\tmockMonitor,\n\t\t\t\tlogger,\n\t\t\t\tmetricsScope,\n\t\t\t\toptions,\n\t\t\t)\n\n\t\t\timpl, ok := mitigator.(*mitigatorImpl)\n\t\t\trequire.True(t, ok)\n\n\t\t\tstats, expectedResult := tt.setupStats(ctrl)\n\t\t\tresult := impl.findDomainsToClear(stats, tt.targetCount)\n\n\t\t\t// Verify the actual result content\n\t\t\tassert.Equal(t, expectedResult, result, \"Result should contain the expected slice-to-domains mapping\")\n\t\t})\n\t}\n}\n\nfunc TestMitigator_processQueueSplitsAndClear(t *testing.T) {\n\ttests := []struct {\n\t\tname       string\n\t\tsetupMocks func(*gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int)\n\t}{\n\t\t{\n\t\t\tname: \"no domains to clear - no operations\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\tmockVQ := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// SplitSlices is still called, but the split function should return false for all slices\n\t\t\t\tmockVQ.EXPECT().SplitSlices(gomock.Any()).Do(func(splitFunc func(VirtualSlice) ([]VirtualSlice, bool)) {\n\t\t\t\t\t// Since domainsToClear is empty, any slice should return split=false\n\t\t\t\t\tmockSlice := NewMockVirtualSlice(nil)\n\t\t\t\t\tremaining, split := splitFunc(mockSlice)\n\t\t\t\t\trequire.False(t, split)\n\t\t\t\t\trequire.Nil(t, remaining)\n\t\t\t\t}).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t0: mockVQ,\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{}\n\t\t\t\tmaxQueueCount := 3\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"clear slices in last queue - no splitting\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\tmockSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Should call ClearSlices with a predicate function\n\t\t\t\tmockVQ.EXPECT().ClearSlices(gomock.Any()).Do(func(predicate func(VirtualSlice) bool) {\n\t\t\t\t\t// Verify the predicate returns true for slices in domainsToClear\n\t\t\t\t\tclearedCount := 0\n\t\t\t\t\tfor _, slice := range []VirtualSlice{mockSlice1, mockSlice2} {\n\t\t\t\t\t\tif predicate(slice) {\n\t\t\t\t\t\t\tclearedCount++\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\t// Since both slices are in domainsToClear, both should be cleared\n\t\t\t\t\trequire.Equal(t, 1, clearedCount)\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// Should pause the queue after clearing\n\t\t\t\tmockVQ.EXPECT().Pause(clearSliceThrottleDuration).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t2: mockVQ, // queueID >= maxQueueCount-1 (2)\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice1: {\"domain1\"},\n\t\t\t\t}\n\t\t\t\tmaxQueueCount := 3\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"split and move slices - successful split\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\tmockSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSplitSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockRemainingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Set up split behavior\n\t\t\t\tmockSlice.EXPECT().TrySplitByPredicate(gomock.Any()).Return(mockSplitSlice, mockRemainingSlice, true).AnyTimes()\n\t\t\t\tmockSplitSlice.EXPECT().Clear().AnyTimes()\n\n\t\t\t\t// Should call SplitSlices with a function\n\t\t\t\tmockVQ.EXPECT().SplitSlices(gomock.Any()).Do(func(splitFunc func(VirtualSlice) ([]VirtualSlice, bool)) {\n\t\t\t\t\t// Call the split function with our mock slice\n\t\t\t\t\tremaining, split := splitFunc(mockSlice)\n\t\t\t\t\t// Should return split=true and remaining slice\n\t\t\t\t\trequire.True(t, split)\n\t\t\t\t\trequire.Len(t, remaining, 1)\n\t\t\t\t}).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t0: mockVQ, // queueID < maxQueueCount-1\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice: {\"domain1\", \"domain2\"},\n\t\t\t\t}\n\t\t\t\tmaxQueueCount := 3\n\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"clear slice when split fails\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\tmockSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Set up split behavior to fail\n\t\t\t\tmockSlice.EXPECT().TrySplitByPredicate(gomock.Any()).Return(nil, nil, false).AnyTimes()\n\t\t\t\tmockSlice.EXPECT().Clear().AnyTimes()\n\n\t\t\t\tmockVQ.EXPECT().SplitSlices(gomock.Any()).Do(func(splitFunc func(VirtualSlice) ([]VirtualSlice, bool)) {\n\t\t\t\t\tremaining, split := splitFunc(mockSlice)\n\t\t\t\t\trequire.True(t, split)\n\t\t\t\t\trequire.Nil(t, remaining)\n\t\t\t\t}).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t1: mockVQ,\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice: {\"domain1\"},\n\t\t\t\t}\n\t\t\t\tmaxQueueCount := 4\n\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple queues with mixed operations\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\t// Queue 0: split and move\n\t\t\t\tmockSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSplitSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockRemainingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ0 := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Queue 1: split and move\n\t\t\t\tmockSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockSplitSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockRemainingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ1 := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Queue 2: clear (last queue)\n\t\t\t\tmockSlice3 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ2 := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Set up split behaviors\n\t\t\t\tmockSlice1.EXPECT().TrySplitByPredicate(gomock.Any()).Return(mockSplitSlice1, mockRemainingSlice1, true).AnyTimes()\n\t\t\t\tmockSplitSlice1.EXPECT().Clear().AnyTimes()\n\n\t\t\t\tmockSlice2.EXPECT().TrySplitByPredicate(gomock.Any()).Return(mockSplitSlice2, mockRemainingSlice2, true).AnyTimes()\n\t\t\t\tmockSplitSlice2.EXPECT().Clear().AnyTimes()\n\n\t\t\t\t// Queue 0: should split\n\t\t\t\tmockVQ0.EXPECT().SplitSlices(gomock.Any()).Times(1)\n\n\t\t\t\t// Queue 1: should split\n\t\t\t\tmockVQ1.EXPECT().SplitSlices(gomock.Any()).Times(1)\n\n\t\t\t\t// Queue 2: should clear (last queue)\n\t\t\t\tvar cleared bool\n\t\t\t\tmockVQ2.EXPECT().ClearSlices(gomock.Any()).Do(func(predicate func(VirtualSlice) bool) {\n\t\t\t\t\t// Simulate that mockSlice3 exists in the queue and is checked\n\t\t\t\t\tif predicate(mockSlice3) {\n\t\t\t\t\t\tcleared = true\n\t\t\t\t\t}\n\t\t\t\t}).Times(1)\n\n\t\t\t\t// Pause should only be called if cleared is true\n\t\t\t\tmockVQ2.EXPECT().Pause(clearSliceThrottleDuration).Do(func(duration interface{}) {\n\t\t\t\t\trequire.True(t, cleared, \"Pause should only be called if slices were cleared\")\n\t\t\t\t}).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t0: mockVQ0,\n\t\t\t\t\t1: mockVQ1,\n\t\t\t\t\t2: mockVQ2, // Last queue (>= maxQueueCount-1)\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice1: {\"domain1\"},\n\t\t\t\t\tmockSlice2: {\"domain2\"},\n\t\t\t\t\tmockSlice3: {\"domain3\"},\n\t\t\t\t}\n\t\t\t\tmaxQueueCount := 3\n\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty domains to clear for some slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (map[int64]VirtualQueue, map[VirtualSlice][]string, int) {\n\t\t\t\tmockSlice1 := NewMockVirtualSlice(ctrl) // Has domains to clear\n\t\t\t\tmockSplitSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockRemainingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmockVQ := NewMockVirtualQueue(ctrl)\n\n\t\t\t\t// Set up split behavior for slice that has domains to clear\n\t\t\t\tmockSlice1.EXPECT().TrySplitByPredicate(gomock.Any()).Return(mockSplitSlice1, mockRemainingSlice1, true).AnyTimes()\n\t\t\t\tmockSplitSlice1.EXPECT().Clear().AnyTimes()\n\n\t\t\t\tmockVQ.EXPECT().SplitSlices(gomock.Any()).Do(func(splitFunc func(VirtualSlice) ([]VirtualSlice, bool)) {\n\t\t\t\t\t// Test with slice that has domains to clear\n\t\t\t\t\tremaining, split := splitFunc(mockSlice1)\n\t\t\t\t\trequire.True(t, split)\n\t\t\t\t\trequire.Len(t, remaining, 1)\n\n\t\t\t\t\t// Test with slice that has no domains to clear\n\t\t\t\t\tmockSlice2 := NewMockVirtualSlice(nil)\n\t\t\t\t\tremaining, split = splitFunc(mockSlice2)\n\t\t\t\t\trequire.False(t, split)\n\t\t\t\t\trequire.Nil(t, remaining)\n\t\t\t\t}).Times(1)\n\n\t\t\t\tvirtualQueues := map[int64]VirtualQueue{\n\t\t\t\t\t0: mockVQ,\n\t\t\t\t}\n\t\t\t\tdomainsToClear := map[VirtualSlice][]string{\n\t\t\t\t\tmockSlice1: {\"domain1\"},\n\t\t\t\t\t// mockSlice2 intentionally not in domainsToClear\n\t\t\t\t}\n\t\t\t\tmaxQueueCount := 3\n\n\t\t\t\treturn virtualQueues, domainsToClear, maxQueueCount\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\tlogger := testlogger.New(t)\n\t\t\tmetricsScope := metrics.NoopScope\n\t\t\toptions := &MitigatorOptions{\n\t\t\t\tMaxVirtualQueueCount: dynamicproperties.GetIntPropertyFn(3),\n\t\t\t}\n\n\t\t\tvirtualQueues, domainsToClear, maxQueueCount := tt.setupMocks(ctrl)\n\t\t\toptions.MaxVirtualQueueCount = dynamicproperties.GetIntPropertyFn(maxQueueCount)\n\n\t\t\t// Create a mock VirtualQueueManager that returns our test queues\n\t\t\tmockVQManager := NewMockVirtualQueueManager(ctrl)\n\n\t\t\t// Set up expectations for GetOrCreateVirtualQueue calls if needed\n\t\t\tfor queueID := range virtualQueues {\n\t\t\t\tif queueID < int64(maxQueueCount-1) {\n\t\t\t\t\t// Expect calls to create next queues for moving slices\n\t\t\t\t\tnextQueueID := queueID + 1\n\t\t\t\t\tif _, exists := virtualQueues[nextQueueID]; !exists {\n\t\t\t\t\t\tmockNextVQ := NewMockVirtualQueue(ctrl)\n\t\t\t\t\t\tmockNextVQ.EXPECT().Pause(clearSliceThrottleDuration).AnyTimes()\n\t\t\t\t\t\tmockNextVQ.EXPECT().MergeSlices(gomock.Any()).AnyTimes()\n\t\t\t\t\t\tmockVQManager.EXPECT().GetOrCreateVirtualQueue(nextQueueID).Return(mockNextVQ).AnyTimes()\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tmitigator := &mitigatorImpl{\n\t\t\t\tvirtualQueueManager: mockVQManager,\n\t\t\t\tmonitor:             mockMonitor,\n\t\t\t\tlogger:              logger,\n\t\t\t\tmetricsScope:        metricsScope,\n\t\t\t\toptions:             options,\n\t\t\t}\n\n\t\t\t// Execute the method\n\t\t\tmitigator.processQueueSplitsAndClear(virtualQueues, domainsToClear)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/monitor.go",
    "content": "//go:generate mockgen -package $GOPACKAGE -destination monitor_mock.go github.com/uber/cadence/service/history/queuev2 Monitor\npackage queuev2\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tMonitor interface {\n\t\tSubscribe(chan<- *Alert)\n\t\tUnsubscribe()\n\t\tGetTotalPendingTaskCount() int\n\t\tGetSlicePendingTaskCount(VirtualSlice) int\n\t\tSetSlicePendingTaskCount(VirtualSlice, int)\n\t\tRemoveSlice(VirtualSlice)\n\t\tResolveAlert(AlertType)\n\t}\n\n\tMonitorOptions struct {\n\t\tEnablePendingTaskCountAlert func() bool\n\t\tCriticalPendingTaskCount    dynamicproperties.IntPropertyFn\n\t}\n\n\tmonitorImpl struct {\n\t\tsync.Mutex\n\n\t\tcategory persistence.HistoryTaskCategory\n\t\toptions  *MonitorOptions\n\n\t\tsubscriber            chan<- *Alert\n\t\tpendingAlerts         map[AlertType]struct{}\n\t\ttotalPendingTaskCount int\n\t\tslicePendingTaskCount map[VirtualSlice]int\n\t}\n)\n\nfunc NewMonitor(category persistence.HistoryTaskCategory, options *MonitorOptions) Monitor {\n\treturn &monitorImpl{\n\t\tcategory: category,\n\t\toptions:  options,\n\n\t\tpendingAlerts:         make(map[AlertType]struct{}),\n\t\ttotalPendingTaskCount: 0,\n\t\tslicePendingTaskCount: make(map[VirtualSlice]int),\n\t}\n}\n\nfunc (m *monitorImpl) Subscribe(subscriber chan<- *Alert) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tm.subscriber = subscriber\n}\n\nfunc (m *monitorImpl) Unsubscribe() {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tm.subscriber = nil\n}\nfunc (m *monitorImpl) GetTotalPendingTaskCount() int {\n\tm.Lock()\n\tdefer m.Unlock()\n\treturn m.totalPendingTaskCount\n}\n\nfunc (m *monitorImpl) GetSlicePendingTaskCount(slice VirtualSlice) int {\n\tm.Lock()\n\tdefer m.Unlock()\n\treturn m.slicePendingTaskCount[slice]\n}\n\nfunc (m *monitorImpl) SetSlicePendingTaskCount(slice VirtualSlice, count int) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tcurrentSliceCount := m.slicePendingTaskCount[slice]\n\tm.totalPendingTaskCount += count - currentSliceCount\n\tm.slicePendingTaskCount[slice] = count\n\n\tcriticalPendingTaskCount := m.options.CriticalPendingTaskCount()\n\tif m.options.EnablePendingTaskCountAlert() && criticalPendingTaskCount > 0 && m.totalPendingTaskCount > criticalPendingTaskCount {\n\t\tm.sendAlertLocked(&Alert{\n\t\t\tAlertType: AlertTypeQueuePendingTaskCount,\n\t\t\tAlertAttributesQueuePendingTaskCount: &AlertAttributesQueuePendingTaskCount{\n\t\t\t\tCurrentPendingTaskCount:  m.totalPendingTaskCount,\n\t\t\t\tCriticalPendingTaskCount: criticalPendingTaskCount,\n\t\t\t},\n\t\t})\n\t}\n}\n\nfunc (m *monitorImpl) RemoveSlice(slice VirtualSlice) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tif currentSliceCount, ok := m.slicePendingTaskCount[slice]; ok {\n\t\tm.totalPendingTaskCount -= currentSliceCount\n\t\tdelete(m.slicePendingTaskCount, slice)\n\t}\n}\n\nfunc (m *monitorImpl) ResolveAlert(alertType AlertType) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tdelete(m.pendingAlerts, alertType)\n}\n\nfunc (m *monitorImpl) sendAlertLocked(alert *Alert) {\n\t// deduplicate alerts\n\tif _, ok := m.pendingAlerts[alert.AlertType]; ok {\n\t\treturn\n\t}\n\n\tselect {\n\tcase m.subscriber <- alert:\n\t\tm.pendingAlerts[alert.AlertType] = struct{}{}\n\tdefault:\n\t\t// do not block if subscriber is not ready\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/monitor_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: Monitor)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination monitor_mock.go github.com/uber/cadence/service/history/queuev2 Monitor\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockMonitor is a mock of Monitor interface.\ntype MockMonitor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockMonitorMockRecorder\n\tisgomock struct{}\n}\n\n// MockMonitorMockRecorder is the mock recorder for MockMonitor.\ntype MockMonitorMockRecorder struct {\n\tmock *MockMonitor\n}\n\n// NewMockMonitor creates a new mock instance.\nfunc NewMockMonitor(ctrl *gomock.Controller) *MockMonitor {\n\tmock := &MockMonitor{ctrl: ctrl}\n\tmock.recorder = &MockMonitorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockMonitor) EXPECT() *MockMonitorMockRecorder {\n\treturn m.recorder\n}\n\n// GetSlicePendingTaskCount mocks base method.\nfunc (m *MockMonitor) GetSlicePendingTaskCount(arg0 VirtualSlice) int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSlicePendingTaskCount\", arg0)\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetSlicePendingTaskCount indicates an expected call of GetSlicePendingTaskCount.\nfunc (mr *MockMonitorMockRecorder) GetSlicePendingTaskCount(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSlicePendingTaskCount\", reflect.TypeOf((*MockMonitor)(nil).GetSlicePendingTaskCount), arg0)\n}\n\n// GetTotalPendingTaskCount mocks base method.\nfunc (m *MockMonitor) GetTotalPendingTaskCount() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTotalPendingTaskCount\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetTotalPendingTaskCount indicates an expected call of GetTotalPendingTaskCount.\nfunc (mr *MockMonitorMockRecorder) GetTotalPendingTaskCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTotalPendingTaskCount\", reflect.TypeOf((*MockMonitor)(nil).GetTotalPendingTaskCount))\n}\n\n// RemoveSlice mocks base method.\nfunc (m *MockMonitor) RemoveSlice(arg0 VirtualSlice) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RemoveSlice\", arg0)\n}\n\n// RemoveSlice indicates an expected call of RemoveSlice.\nfunc (mr *MockMonitorMockRecorder) RemoveSlice(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveSlice\", reflect.TypeOf((*MockMonitor)(nil).RemoveSlice), arg0)\n}\n\n// ResolveAlert mocks base method.\nfunc (m *MockMonitor) ResolveAlert(arg0 AlertType) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ResolveAlert\", arg0)\n}\n\n// ResolveAlert indicates an expected call of ResolveAlert.\nfunc (mr *MockMonitorMockRecorder) ResolveAlert(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResolveAlert\", reflect.TypeOf((*MockMonitor)(nil).ResolveAlert), arg0)\n}\n\n// SetSlicePendingTaskCount mocks base method.\nfunc (m *MockMonitor) SetSlicePendingTaskCount(arg0 VirtualSlice, arg1 int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetSlicePendingTaskCount\", arg0, arg1)\n}\n\n// SetSlicePendingTaskCount indicates an expected call of SetSlicePendingTaskCount.\nfunc (mr *MockMonitorMockRecorder) SetSlicePendingTaskCount(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetSlicePendingTaskCount\", reflect.TypeOf((*MockMonitor)(nil).SetSlicePendingTaskCount), arg0, arg1)\n}\n\n// Subscribe mocks base method.\nfunc (m *MockMonitor) Subscribe(arg0 chan<- *Alert) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Subscribe\", arg0)\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockMonitorMockRecorder) Subscribe(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockMonitor)(nil).Subscribe), arg0)\n}\n\n// Unsubscribe mocks base method.\nfunc (m *MockMonitor) Unsubscribe() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Unsubscribe\")\n}\n\n// Unsubscribe indicates an expected call of Unsubscribe.\nfunc (mr *MockMonitorMockRecorder) Unsubscribe() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unsubscribe\", reflect.TypeOf((*MockMonitor)(nil).Unsubscribe))\n}\n"
  },
  {
    "path": "service/history/queuev2/monitor_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestMonitorPendingTaskCount(t *testing.T) {\n\tmonitor := NewMonitor(persistence.HistoryTaskCategoryTimer, &MonitorOptions{\n\t\tCriticalPendingTaskCount:    dynamicproperties.GetIntPropertyFn(100),\n\t\tEnablePendingTaskCountAlert: func() bool { return true },\n\t})\n\n\tassert.Equal(t, 0, monitor.GetTotalPendingTaskCount())\n\n\tslice1 := &virtualSliceImpl{}\n\tslice2 := &virtualSliceImpl{}\n\tslice3 := &virtualSliceImpl{}\n\n\t// set pending task count for slices\n\tmonitor.SetSlicePendingTaskCount(slice1, 10)\n\tassert.Equal(t, 10, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 10, monitor.GetSlicePendingTaskCount(slice1))\n\n\tmonitor.SetSlicePendingTaskCount(slice2, 20)\n\tassert.Equal(t, 30, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 20, monitor.GetSlicePendingTaskCount(slice2))\n\n\tmonitor.SetSlicePendingTaskCount(slice3, 30)\n\tassert.Equal(t, 60, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 30, monitor.GetSlicePendingTaskCount(slice3))\n\n\t// update pending task count for slices\n\tmonitor.SetSlicePendingTaskCount(slice1, 15)\n\tassert.Equal(t, 65, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 15, monitor.GetSlicePendingTaskCount(slice1))\n\n\tmonitor.SetSlicePendingTaskCount(slice2, 21)\n\tassert.Equal(t, 66, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 21, monitor.GetSlicePendingTaskCount(slice2))\n\n\tmonitor.RemoveSlice(slice1)\n\tassert.Equal(t, 51, monitor.GetTotalPendingTaskCount())\n\n\tmonitor.RemoveSlice(slice2)\n\tassert.Equal(t, 30, monitor.GetTotalPendingTaskCount())\n\n\tmonitor.RemoveSlice(slice3)\n\tassert.Equal(t, 0, monitor.GetTotalPendingTaskCount())\n\n\talertCh := make(chan *Alert, alertChSize)\n\tmonitor.Subscribe(alertCh)\n\n\tmonitor.SetSlicePendingTaskCount(slice1, 101)\n\tassert.Equal(t, 101, monitor.GetTotalPendingTaskCount())\n\tassert.Equal(t, 101, monitor.GetSlicePendingTaskCount(slice1))\n\n\talert := <-alertCh\n\tassert.Equal(t, AlertTypeQueuePendingTaskCount, alert.AlertType)\n\tassert.Equal(t, 101, alert.AlertAttributesQueuePendingTaskCount.CurrentPendingTaskCount)\n\tassert.Equal(t, 100, alert.AlertAttributesQueuePendingTaskCount.CriticalPendingTaskCount)\n\n\t_, ok := monitor.(*monitorImpl).pendingAlerts[AlertTypeQueuePendingTaskCount]\n\tassert.True(t, ok)\n}\n\nfunc TestMonitorSubscribeAndUnsubscribe(t *testing.T) {\n\tmonitor := NewMonitor(persistence.HistoryTaskCategoryTimer, &MonitorOptions{})\n\n\talertCh := make(chan *Alert, alertChSize)\n\tmonitor.Subscribe(alertCh)\n\tmonitor.(*monitorImpl).subscriber <- &Alert{AlertType: AlertTypeQueuePendingTaskCount}\n\talert := <-alertCh\n\tassert.Equal(t, AlertTypeQueuePendingTaskCount, alert.AlertType)\n\n\tmonitor.Unsubscribe()\n\tassert.Nil(t, monitor.(*monitorImpl).subscriber)\n}\n\nfunc TestMonitorResolveAlert(t *testing.T) {\n\tmonitor := NewMonitor(persistence.HistoryTaskCategoryTimer, &MonitorOptions{})\n\n\tmonitor.(*monitorImpl).pendingAlerts[AlertTypeQueuePendingTaskCount] = struct{}{}\n\tassert.Equal(t, 1, len(monitor.(*monitorImpl).pendingAlerts))\n\n\tmonitor.ResolveAlert(AlertTypeQueuePendingTaskCount)\n\tassert.Equal(t, 0, len(monitor.(*monitorImpl).pendingAlerts))\n}\n"
  },
  {
    "path": "service/history/queuev2/pause_controller.go",
    "content": "//go:generate mockgen -package $GOPACKAGE -destination pause_controller_mock.go github.com/uber/cadence/service/history/queuev2 PauseController\npackage queuev2\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\t// PauseController is a controller that allows to pause and resume a background job.\n\t// For example, if you have a background job like this:\n\t// func run(ctx context.Context, wg *sync.WaitGroup, ...) {\n\t//   defer wg.Done()\n\t//   for {\n\t//     select {\n\t//     case <-ctx.Done():\n\t//       return\n\t//     case <-notifyCh:\n\t//       doSomething(...)\n\t//     }\n\t//   }\n\t// }\n\t// you can integrate the pause controller into the run function like this:\n\t// func run(ctx context.Context, wg *sync.WaitGroup, pauseController PauseController, ...) {\n\t//   defer wg.Done()\n\t//   pauseController.Subscribe(\"run\", notifyCh)\n\t//   for {\n\t//     select {\n\t//     case <-ctx.Done():\n\t//       return\n\t//     case <-notifyCh:\n\t//       doSomething(pauseController, ...)\n\t//     }\n\t//   }\n\t// }\n\t//\n\t// func doSomething(pauseController PauseController, ...) {\n\t//   if someCondition {\n\t//     pauseController.Pause(10 * time.Second)\n\t//   }\n\t//   if pauseController.IsPaused() {\n\t//     return\n\t//   }\n\t//   // do the actual work\n\t// }\n\tPauseController interface {\n\t\t// Stop the pause controller but don't send notification to the subscribers. no-op if it's not paused.\n\t\tStop()\n\t\t// Pause the job for the given duration. Zero and negative durations are ignored.\n\t\t// If it's already paused, the pause duration can only be updated to a longer duration.\n\t\tPause(time.Duration)\n\t\t// Resume the job immediately. If it's not paused, this is a no-op.\n\t\tResume()\n\t\t// Check if the job is paused.\n\t\tIsPaused() bool\n\t\t// Subscribe to the pause controller.\n\t\tSubscribe(string, chan<- struct{})\n\t\t// Unsubscribe from the pause controller.\n\t\tUnsubscribe(string)\n\t}\n\n\tpauseControllerImpl struct {\n\t\tsync.Mutex\n\t\tsubscribers map[string]chan<- struct{}\n\t\ttimeSource  clock.TimeSource\n\t\tpauseUntil  time.Time\n\t\ttimer       clock.Timer\n\t}\n)\n\nfunc NewPauseController(timeSource clock.TimeSource) PauseController {\n\treturn &pauseControllerImpl{\n\t\ttimeSource:  timeSource,\n\t\tsubscribers: make(map[string]chan<- struct{}),\n\t}\n}\n\nfunc (p *pauseControllerImpl) IsPaused() bool {\n\tp.Lock()\n\tdefer p.Unlock()\n\treturn p.timer != nil\n}\n\nfunc (p *pauseControllerImpl) Subscribe(id string, ch chan<- struct{}) {\n\tp.Lock()\n\tdefer p.Unlock()\n\tp.subscribers[id] = ch\n}\n\nfunc (p *pauseControllerImpl) Unsubscribe(id string) {\n\tp.Lock()\n\tdefer p.Unlock()\n\tdelete(p.subscribers, id)\n}\n\nfunc (p *pauseControllerImpl) Stop() {\n\tp.Lock()\n\tdefer p.Unlock()\n\tp.stopTimerLocked()\n\tp.timer = nil\n\tp.pauseUntil = time.Time{}\n}\n\nfunc (p *pauseControllerImpl) Pause(duration time.Duration) {\n\tif duration <= 0 {\n\t\treturn\n\t}\n\n\tp.Lock()\n\tdefer p.Unlock()\n\n\tnewPauseUntil := p.timeSource.Now().Add(duration)\n\tif newPauseUntil.Before(p.pauseUntil) {\n\t\treturn\n\t}\n\n\tp.stopTimerLocked()\n\tp.timer = p.timeSource.AfterFunc(duration, func() {\n\t\tp.Lock()\n\t\tdefer p.Unlock()\n\t\tp.timer = nil\n\t\tp.pauseUntil = time.Time{}\n\t\tp.notifySubscribers()\n\t})\n\tp.pauseUntil = newPauseUntil\n}\n\nfunc (p *pauseControllerImpl) Resume() {\n\tp.Lock()\n\tdefer p.Unlock()\n\n\tif p.timer == nil {\n\t\treturn\n\t}\n\tp.stopTimerLocked()\n\tp.timer = nil\n\tp.pauseUntil = time.Time{}\n\tp.notifySubscribers()\n}\n\nfunc (p *pauseControllerImpl) stopTimerLocked() {\n\tif p.timer != nil {\n\t\tp.timer.Stop()\n\t}\n}\n\nfunc (p *pauseControllerImpl) notifySubscribers() {\n\tfor _, ch := range p.subscribers {\n\t\tselect {\n\t\tcase ch <- struct{}{}:\n\t\tdefault:\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/pause_controller_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: PauseController)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination pause_controller_mock.go github.com/uber/cadence/service/history/queuev2 PauseController\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPauseController is a mock of PauseController interface.\ntype MockPauseController struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPauseControllerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPauseControllerMockRecorder is the mock recorder for MockPauseController.\ntype MockPauseControllerMockRecorder struct {\n\tmock *MockPauseController\n}\n\n// NewMockPauseController creates a new mock instance.\nfunc NewMockPauseController(ctrl *gomock.Controller) *MockPauseController {\n\tmock := &MockPauseController{ctrl: ctrl}\n\tmock.recorder = &MockPauseControllerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPauseController) EXPECT() *MockPauseControllerMockRecorder {\n\treturn m.recorder\n}\n\n// IsPaused mocks base method.\nfunc (m *MockPauseController) IsPaused() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsPaused\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsPaused indicates an expected call of IsPaused.\nfunc (mr *MockPauseControllerMockRecorder) IsPaused() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsPaused\", reflect.TypeOf((*MockPauseController)(nil).IsPaused))\n}\n\n// Pause mocks base method.\nfunc (m *MockPauseController) Pause(arg0 time.Duration) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Pause\", arg0)\n}\n\n// Pause indicates an expected call of Pause.\nfunc (mr *MockPauseControllerMockRecorder) Pause(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Pause\", reflect.TypeOf((*MockPauseController)(nil).Pause), arg0)\n}\n\n// Resume mocks base method.\nfunc (m *MockPauseController) Resume() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Resume\")\n}\n\n// Resume indicates an expected call of Resume.\nfunc (mr *MockPauseControllerMockRecorder) Resume() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Resume\", reflect.TypeOf((*MockPauseController)(nil).Resume))\n}\n\n// Stop mocks base method.\nfunc (m *MockPauseController) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockPauseControllerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockPauseController)(nil).Stop))\n}\n\n// Subscribe mocks base method.\nfunc (m *MockPauseController) Subscribe(arg0 string, arg1 chan<- struct{}) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Subscribe\", arg0, arg1)\n}\n\n// Subscribe indicates an expected call of Subscribe.\nfunc (mr *MockPauseControllerMockRecorder) Subscribe(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Subscribe\", reflect.TypeOf((*MockPauseController)(nil).Subscribe), arg0, arg1)\n}\n\n// Unsubscribe mocks base method.\nfunc (m *MockPauseController) Unsubscribe(arg0 string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Unsubscribe\", arg0)\n}\n\n// Unsubscribe indicates an expected call of Unsubscribe.\nfunc (mr *MockPauseControllerMockRecorder) Unsubscribe(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unsubscribe\", reflect.TypeOf((*MockPauseController)(nil).Unsubscribe), arg0)\n}\n"
  },
  {
    "path": "service/history/queuev2/pause_controller_test.go",
    "content": "package queuev2\n\nimport (\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nfunc TestPauseController_Basic(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\tch := make(chan struct{}, 100)\n\tcontroller.Subscribe(\"test\", ch)\n\t// Initially not paused\n\tassert.False(t, controller.IsPaused())\n\n\t// pause for 100ms\n\tcontroller.Pause(100 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\n\t// advance 50ms, should still be paused\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(50 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\n\t// pause for 200ms and this should reset the pause duration\n\tcontroller.Pause(200 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\n\t// Advance time to trigger timer\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(200 * time.Millisecond)\n\t// wait for the timer to expire\n\t<-ch\n\tassert.False(t, controller.IsPaused())\n\n\t// Resume when not paused should be no-op\n\tcontroller.Resume()\n\tassert.False(t, controller.IsPaused())\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"resume when not paused should be no-op\")\n\tdefault:\n\t\t// Expected\n\t}\n\n\t// Pause then resume\n\tcontroller.Pause(100 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\n\tcontroller.Resume()\n\tassert.False(t, controller.IsPaused())\n\tselect {\n\tcase <-ch:\n\tdefault:\n\t\tt.Fatal(\"channel should receive notification after resume\")\n\t}\n\n\t// Resume again should be no-op\n\tcontroller.Resume()\n\tassert.False(t, controller.IsPaused())\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"resume when not paused should be no-op\")\n\tdefault:\n\t\t// Expected\n\t}\n\n\t// Stop when not paused\n\tcontroller.Stop()\n\tassert.False(t, controller.IsPaused())\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"channel should not receive notification after stop\")\n\tdefault:\n\t\t// Expected\n\t}\n\n\t// Pause then stop\n\tcontroller.Pause(100 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\tcontroller.Stop()\n\tassert.False(t, controller.IsPaused())\n\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"channel should not receive notification after stop\")\n\tdefault:\n\t\t// Expected\n\t}\n\n\tcontroller.Pause(100 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\n\t// Pause duration is less than the previous pause duration, should not reset the pause duration\n\tcontroller.Pause(10 * time.Millisecond)\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(20 * time.Millisecond)\n\tassert.True(t, controller.IsPaused())\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"channel should not receive notification\")\n\tdefault:\n\t\t// Expected\n\t}\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(100 * time.Millisecond)\n\t// wait for the timer to expire\n\t<-ch\n\tassert.False(t, controller.IsPaused())\n}\n\nfunc TestPauseController_Subscribe_Unsubscribe(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\t// Subscribe a channel\n\tch1 := make(chan struct{}, 1)\n\tcontroller.Subscribe(\"sub1\", ch1)\n\n\t// Subscribe another channel\n\tch2 := make(chan struct{}, 1)\n\tcontroller.Subscribe(\"sub2\", ch2)\n\n\t// Unsubscribe one\n\tcontroller.Unsubscribe(\"sub1\")\n\n\t// Pause and resume to trigger notifications\n\tcontroller.Pause(100 * time.Millisecond)\n\tcontroller.Resume()\n\n\t// Only ch2 should receive notification\n\tselect {\n\tcase <-ch1:\n\t\tt.Fatal(\"ch1 should not receive notification after unsubscribe\")\n\tdefault:\n\t\t// Expected\n\t}\n\n\tselect {\n\tcase <-ch2:\n\t\t// Expected\n\tdefault:\n\t\tt.Fatal(\"ch2 should receive notification\")\n\t}\n}\n\nfunc TestPauseController_NotifySubscribers(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\t// Subscribe multiple channels\n\tch1 := make(chan struct{}, 1)\n\tch2 := make(chan struct{}, 1)\n\tch3 := make(chan struct{}, 1)\n\n\tcontroller.Subscribe(\"sub1\", ch1)\n\tcontroller.Subscribe(\"sub2\", ch2)\n\tcontroller.Subscribe(\"sub3\", ch3)\n\n\t// Pause and resume to trigger notifications\n\tcontroller.Pause(100 * time.Millisecond)\n\tcontroller.Resume()\n\n\t// All channels should receive notifications\n\tselect {\n\tcase <-ch1:\n\t\t// Expected\n\tdefault:\n\t\tt.Fatal(\"ch1 should receive notification\")\n\t}\n\n\tselect {\n\tcase <-ch2:\n\t\t// Expected\n\tdefault:\n\t\tt.Fatal(\"ch2 should receive notification\")\n\t}\n\n\tselect {\n\tcase <-ch3:\n\t\t// Expected\n\tdefault:\n\t\tt.Fatal(\"ch3 should receive notification\")\n\t}\n}\n\nfunc TestPauseController_SubscriberChannelBlocking(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\t// Subscribe a channel with no buffer (will block)\n\tch := make(chan struct{})\n\tcontroller.Subscribe(\"sub1\", ch)\n\n\t// Pause and resume to trigger notification\n\tcontroller.Pause(100 * time.Millisecond)\n\tcontroller.Resume()\n\n\t// The notification should be sent non-blocking\n\t// If the channel is full or blocked, it should not cause deadlock\n\t// This test verifies that the select with default case works correctly\n\tassert.False(t, controller.IsPaused())\n}\n\nfunc TestPauseController_ZeroDurationPause(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\t// Subscribe a channel\n\tch := make(chan struct{}, 1)\n\tcontroller.Subscribe(\"sub1\", ch)\n\n\t// Pause with zero duration\n\tcontroller.Pause(0)\n\tassert.False(t, controller.IsPaused())\n\n\t// Channel should not receive notification\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"channel should not receive notification for zero duration pause\")\n\tdefault:\n\t\t// Expected\n\t}\n}\n\nfunc TestPauseController_NegativeDurationPause(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\t// Subscribe a channel\n\tch := make(chan struct{}, 1)\n\tcontroller.Subscribe(\"sub1\", ch)\n\n\t// Pause with negative duration\n\tcontroller.Pause(-100 * time.Millisecond)\n\tassert.False(t, controller.IsPaused())\n\n\t// Channel should not receive notification\n\tselect {\n\tcase <-ch:\n\t\tt.Fatal(\"channel should not receive notification for negative duration pause\")\n\tdefault:\n\t\t// Expected\n\t}\n}\n\nfunc TestPauseController_ConcurrentAccess(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\tcontroller := NewPauseController(timeSource)\n\n\tvar wg sync.WaitGroup\n\tnumGoroutines := 100\n\n\t// Test concurrent pause/resume operations\n\tfor i := 0; i < numGoroutines; i++ {\n\t\twg.Add(1)\n\t\tgo func(id int) {\n\t\t\tdefer wg.Done()\n\t\t\tch := make(chan struct{}, 1)\n\t\t\tcontroller.Subscribe(\"sub\"+string(rune(id)), ch)\n\t\t\tcontroller.Pause(time.Duration(id+1) * time.Millisecond)\n\t\t\tcontroller.Resume()\n\t\t\tcontroller.IsPaused()\n\t\t\tcontroller.Unsubscribe(\"sub\" + string(rune(id)))\n\t\t}(i)\n\t}\n\n\twg.Wait()\n}\n"
  },
  {
    "path": "service/history/queuev2/pending_task_tracker.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination pending_task_tracker_mock.go github.com/uber/cadence/service/history/queuev2 PendingTaskTracker\npackage queuev2\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n\tctask \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\t// PendingTaskTracker tracks the pending tasks in a virtual slice.\n\tPendingTaskTracker interface {\n\t\t// AddTask adds a task to the pending task tracker.\n\t\tAddTask(task.Task)\n\t\t// PruneAckedTasks prunes the acked tasks from the pending task tracker.\n\t\tPruneAckedTasks() int\n\t\t// GetMinimumTaskKey returns the minimum task key in the pending task tracker, if there are no pending tasks, it returns MaximumHistoryTaskKey.\n\t\tGetMinimumTaskKey() (persistence.HistoryTaskKey, bool)\n\t\t// GetTasks returns all the tasks in the pending task tracker, the result should be read-only.\n\t\tGetTasks() map[persistence.HistoryTaskKey]task.Task\n\t\t// GetPendingTaskCount returns the number of pending tasks in the pending task tracker.\n\t\tGetPendingTaskCount() int\n\t\t// GetPerDomainPendingTaskCount returns the number of pending tasks per domain.\n\t\tGetPerDomainPendingTaskCount() map[string]int\n\t\t// Clear clears the pending task tracker.\n\t\tClear()\n\t}\n\n\tpendingTaskTrackerImpl struct {\n\t\ttaskMap            map[persistence.HistoryTaskKey]task.Task\n\t\ttaskCountPerDomain map[string]int // domainID -> task count\n\t\tminTaskKey         persistence.HistoryTaskKey\n\t}\n)\n\nfunc NewPendingTaskTracker() PendingTaskTracker {\n\treturn &pendingTaskTrackerImpl{\n\t\ttaskMap:            make(map[persistence.HistoryTaskKey]task.Task),\n\t\ttaskCountPerDomain: make(map[string]int),\n\t\tminTaskKey:         persistence.MaximumHistoryTaskKey,\n\t}\n}\n\nfunc (t *pendingTaskTrackerImpl) AddTask(task task.Task) {\n\tif len(t.taskMap) == 0 {\n\t\tt.minTaskKey = task.GetTaskKey()\n\t} else if t.minTaskKey.Compare(task.GetTaskKey()) > 0 {\n\t\tt.minTaskKey = task.GetTaskKey()\n\t}\n\n\tt.taskMap[task.GetTaskKey()] = task\n\tt.taskCountPerDomain[task.GetDomainID()]++\n}\n\nfunc (t *pendingTaskTrackerImpl) GetMinimumTaskKey() (persistence.HistoryTaskKey, bool) {\n\tif len(t.taskMap) == 0 {\n\t\treturn persistence.MaximumHistoryTaskKey, false\n\t}\n\treturn t.minTaskKey, true\n}\n\nfunc (t *pendingTaskTrackerImpl) GetTasks() map[persistence.HistoryTaskKey]task.Task {\n\treturn t.taskMap\n}\n\nfunc (t *pendingTaskTrackerImpl) PruneAckedTasks() int {\n\tprunedCount := 0\n\tminTaskKey := persistence.MaximumHistoryTaskKey\n\tfor key, task := range t.taskMap {\n\t\tif task.State() == ctask.TaskStateAcked {\n\t\t\tdelete(t.taskMap, key)\n\t\t\tt.taskCountPerDomain[task.GetDomainID()]--\n\t\t\tprunedCount++\n\t\t\tcontinue\n\t\t}\n\n\t\tif key.Compare(minTaskKey) < 0 {\n\t\t\tminTaskKey = key\n\t\t}\n\t}\n\tt.minTaskKey = minTaskKey\n\treturn prunedCount\n}\n\nfunc (t *pendingTaskTrackerImpl) GetPendingTaskCount() int {\n\treturn len(t.taskMap)\n}\n\nfunc (t *pendingTaskTrackerImpl) GetPerDomainPendingTaskCount() map[string]int {\n\treturn t.taskCountPerDomain\n}\n\nfunc (t *pendingTaskTrackerImpl) Clear() {\n\tfor _, task := range t.taskMap {\n\t\ttask.Cancel()\n\t}\n\tt.taskMap = make(map[persistence.HistoryTaskKey]task.Task)\n\tt.taskCountPerDomain = make(map[string]int)\n\tt.minTaskKey = persistence.MaximumHistoryTaskKey\n}\n"
  },
  {
    "path": "service/history/queuev2/pending_task_tracker_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: PendingTaskTracker)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination pending_task_tracker_mock.go github.com/uber/cadence/service/history/queuev2 PendingTaskTracker\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttask \"github.com/uber/cadence/service/history/task\"\n)\n\n// MockPendingTaskTracker is a mock of PendingTaskTracker interface.\ntype MockPendingTaskTracker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPendingTaskTrackerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPendingTaskTrackerMockRecorder is the mock recorder for MockPendingTaskTracker.\ntype MockPendingTaskTrackerMockRecorder struct {\n\tmock *MockPendingTaskTracker\n}\n\n// NewMockPendingTaskTracker creates a new mock instance.\nfunc NewMockPendingTaskTracker(ctrl *gomock.Controller) *MockPendingTaskTracker {\n\tmock := &MockPendingTaskTracker{ctrl: ctrl}\n\tmock.recorder = &MockPendingTaskTrackerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPendingTaskTracker) EXPECT() *MockPendingTaskTrackerMockRecorder {\n\treturn m.recorder\n}\n\n// AddTask mocks base method.\nfunc (m *MockPendingTaskTracker) AddTask(arg0 task.Task) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddTask\", arg0)\n}\n\n// AddTask indicates an expected call of AddTask.\nfunc (mr *MockPendingTaskTrackerMockRecorder) AddTask(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTask\", reflect.TypeOf((*MockPendingTaskTracker)(nil).AddTask), arg0)\n}\n\n// Clear mocks base method.\nfunc (m *MockPendingTaskTracker) Clear() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Clear\")\n}\n\n// Clear indicates an expected call of Clear.\nfunc (mr *MockPendingTaskTrackerMockRecorder) Clear() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Clear\", reflect.TypeOf((*MockPendingTaskTracker)(nil).Clear))\n}\n\n// GetMinimumTaskKey mocks base method.\nfunc (m *MockPendingTaskTracker) GetMinimumTaskKey() (persistence.HistoryTaskKey, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMinimumTaskKey\")\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// GetMinimumTaskKey indicates an expected call of GetMinimumTaskKey.\nfunc (mr *MockPendingTaskTrackerMockRecorder) GetMinimumTaskKey() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMinimumTaskKey\", reflect.TypeOf((*MockPendingTaskTracker)(nil).GetMinimumTaskKey))\n}\n\n// GetPendingTaskCount mocks base method.\nfunc (m *MockPendingTaskTracker) GetPendingTaskCount() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingTaskCount\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetPendingTaskCount indicates an expected call of GetPendingTaskCount.\nfunc (mr *MockPendingTaskTrackerMockRecorder) GetPendingTaskCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingTaskCount\", reflect.TypeOf((*MockPendingTaskTracker)(nil).GetPendingTaskCount))\n}\n\n// GetPerDomainPendingTaskCount mocks base method.\nfunc (m *MockPendingTaskTracker) GetPerDomainPendingTaskCount() map[string]int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPerDomainPendingTaskCount\")\n\tret0, _ := ret[0].(map[string]int)\n\treturn ret0\n}\n\n// GetPerDomainPendingTaskCount indicates an expected call of GetPerDomainPendingTaskCount.\nfunc (mr *MockPendingTaskTrackerMockRecorder) GetPerDomainPendingTaskCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPerDomainPendingTaskCount\", reflect.TypeOf((*MockPendingTaskTracker)(nil).GetPerDomainPendingTaskCount))\n}\n\n// GetTasks mocks base method.\nfunc (m *MockPendingTaskTracker) GetTasks() map[persistence.HistoryTaskKey]task.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\")\n\tret0, _ := ret[0].(map[persistence.HistoryTaskKey]task.Task)\n\treturn ret0\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockPendingTaskTrackerMockRecorder) GetTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockPendingTaskTracker)(nil).GetTasks))\n}\n\n// PruneAckedTasks mocks base method.\nfunc (m *MockPendingTaskTracker) PruneAckedTasks() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PruneAckedTasks\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// PruneAckedTasks indicates an expected call of PruneAckedTasks.\nfunc (mr *MockPendingTaskTrackerMockRecorder) PruneAckedTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PruneAckedTasks\", reflect.TypeOf((*MockPendingTaskTracker)(nil).PruneAckedTasks))\n}\n"
  },
  {
    "path": "service/history/queuev2/pending_task_tracker_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\tctask \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestPendingTaskTracker(t *testing.T) {\n\ttestTime := time.Unix(0, 0)\n\ttests := []struct {\n\t\tname             string\n\t\tsetupTasks       func(ctrl *gomock.Controller) []*task.MockTask\n\t\tpruneAcked       bool\n\t\tpruneAckedCount  int\n\t\tclear            bool\n\t\twantMinKey       persistence.HistoryTaskKey\n\t\twantHasMinKey    bool\n\t\twantTaskCount    int\n\t\twantDomainCounts map[string]int\n\t}{\n\t\t{\n\t\t\tname:             \"empty tracker\",\n\t\t\tsetupTasks:       func(ctrl *gomock.Controller) []*task.MockTask { return []*task.MockTask{} },\n\t\t\twantMinKey:       persistence.MaximumHistoryTaskKey,\n\t\t\twantHasMinKey:    false,\n\t\t\twantTaskCount:    0,\n\t\t\twantDomainCounts: map[string]int{},\n\t\t},\n\t\t{\n\t\t\tname: \"single task\",\n\t\t\tsetupTasks: func(ctrl *gomock.Controller) []*task.MockTask {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\tmockTask.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 1)).AnyTimes()\n\t\t\t\tmockTask.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).AnyTimes()\n\t\t\t\treturn []*task.MockTask{mockTask}\n\t\t\t},\n\t\t\twantMinKey:       persistence.NewHistoryTaskKey(testTime, 1),\n\t\t\twantHasMinKey:    true,\n\t\t\twantTaskCount:    1,\n\t\t\twantDomainCounts: map[string]int{\"domain1\": 1},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple tasks\",\n\t\t\tsetupTasks: func(ctrl *gomock.Controller) []*task.MockTask {\n\t\t\t\ttask1 := task.NewMockTask(ctrl)\n\t\t\t\ttask1.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 3)).AnyTimes()\n\t\t\t\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask1.EXPECT().State().Return(ctask.TaskStatePending).AnyTimes()\n\n\t\t\t\ttask2 := task.NewMockTask(ctrl)\n\t\t\t\ttask2.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 1)).AnyTimes()\n\t\t\t\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\t\ttask2.EXPECT().State().Return(ctask.TaskStatePending).AnyTimes()\n\n\t\t\t\ttask3 := task.NewMockTask(ctrl)\n\t\t\t\ttask3.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 2)).AnyTimes()\n\t\t\t\ttask3.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask3.EXPECT().State().Return(ctask.TaskStatePending).AnyTimes()\n\n\t\t\t\treturn []*task.MockTask{task1, task2, task3}\n\t\t\t},\n\t\t\twantMinKey:       persistence.NewHistoryTaskKey(testTime, 1),\n\t\t\twantHasMinKey:    true,\n\t\t\twantTaskCount:    3,\n\t\t\twantDomainCounts: map[string]int{\"domain1\": 2, \"domain2\": 1},\n\t\t},\n\t\t{\n\t\t\tname: \"prune acked tasks\",\n\t\t\tsetupTasks: func(ctrl *gomock.Controller) []*task.MockTask {\n\t\t\t\ttask1 := task.NewMockTask(ctrl)\n\t\t\t\ttask1.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 1)).AnyTimes()\n\t\t\t\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask1.EXPECT().State().Return(ctask.TaskStateAcked).AnyTimes()\n\n\t\t\t\ttask2 := task.NewMockTask(ctrl)\n\t\t\t\ttask2.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 2)).AnyTimes()\n\t\t\t\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\t\ttask2.EXPECT().State().Return(ctask.TaskStatePending).AnyTimes()\n\n\t\t\t\ttask3 := task.NewMockTask(ctrl)\n\t\t\t\ttask3.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 3)).AnyTimes()\n\t\t\t\ttask3.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask3.EXPECT().State().Return(ctask.TaskStateAcked).AnyTimes()\n\n\t\t\t\treturn []*task.MockTask{task1, task2, task3}\n\t\t\t},\n\t\t\tpruneAcked:       true,\n\t\t\tpruneAckedCount:  2,\n\t\t\twantMinKey:       persistence.NewHistoryTaskKey(testTime, 2),\n\t\t\twantHasMinKey:    true,\n\t\t\twantTaskCount:    1,\n\t\t\twantDomainCounts: map[string]int{\"domain1\": 0, \"domain2\": 1},\n\t\t},\n\t\t{\n\t\t\tname: \"all tasks acked\",\n\t\t\tsetupTasks: func(ctrl *gomock.Controller) []*task.MockTask {\n\t\t\t\ttask1 := task.NewMockTask(ctrl)\n\t\t\t\ttask1.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 1)).AnyTimes()\n\t\t\t\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask1.EXPECT().State().Return(ctask.TaskStateAcked).AnyTimes()\n\n\t\t\t\ttask2 := task.NewMockTask(ctrl)\n\t\t\t\ttask2.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 2)).AnyTimes()\n\t\t\t\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\t\ttask2.EXPECT().State().Return(ctask.TaskStateAcked).AnyTimes()\n\n\t\t\t\treturn []*task.MockTask{task1, task2}\n\t\t\t},\n\t\t\tpruneAcked:       true,\n\t\t\tpruneAckedCount:  2,\n\t\t\twantMinKey:       persistence.MaximumHistoryTaskKey,\n\t\t\twantHasMinKey:    false,\n\t\t\twantTaskCount:    0,\n\t\t\twantDomainCounts: map[string]int{\"domain1\": 0, \"domain2\": 0},\n\t\t},\n\t\t{\n\t\t\tname: \"clear all tasks\",\n\t\t\tsetupTasks: func(ctrl *gomock.Controller) []*task.MockTask {\n\t\t\t\ttask1 := task.NewMockTask(ctrl)\n\t\t\t\ttask1.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 1)).AnyTimes()\n\t\t\t\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask1.EXPECT().Cancel().AnyTimes()\n\t\t\t\ttask1.EXPECT().State().Return(ctask.TaskStateCanceled).Times(1)\n\n\t\t\t\ttask2 := task.NewMockTask(ctrl)\n\t\t\t\ttask2.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 2)).AnyTimes()\n\t\t\t\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\t\ttask2.EXPECT().Cancel().AnyTimes()\n\t\t\t\ttask2.EXPECT().State().Return(ctask.TaskStateCanceled).Times(1)\n\n\t\t\t\ttask3 := task.NewMockTask(ctrl)\n\t\t\t\ttask3.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(testTime, 3)).AnyTimes()\n\t\t\t\ttask3.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\t\ttask3.EXPECT().Cancel().AnyTimes()\n\t\t\t\ttask3.EXPECT().State().Return(ctask.TaskStateCanceled).Times(1)\n\n\t\t\t\treturn []*task.MockTask{task1, task2, task3}\n\t\t\t},\n\t\t\tclear:            true,\n\t\t\twantMinKey:       persistence.MaximumHistoryTaskKey,\n\t\t\twantHasMinKey:    false,\n\t\t\twantTaskCount:    0,\n\t\t\twantDomainCounts: map[string]int{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttracker := NewPendingTaskTracker()\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tinputTasks := tt.setupTasks(ctrl)\n\t\t\t// Setup tasks\n\t\t\tfor _, task := range inputTasks {\n\t\t\t\ttracker.AddTask(task)\n\t\t\t}\n\n\t\t\t// Prune acked tasks if needed\n\t\t\tif tt.pruneAcked {\n\t\t\t\tprunedCount := tracker.PruneAckedTasks()\n\t\t\t\tassert.Equal(t, tt.pruneAckedCount, prunedCount)\n\t\t\t}\n\n\t\t\t// Clear all tasks if needed\n\t\t\tif tt.clear {\n\t\t\t\ttracker.Clear()\n\t\t\t}\n\n\t\t\t// Test GetMinimumTaskKey\n\t\t\tgotMinKey, gotHasMinKey := tracker.GetMinimumTaskKey()\n\t\t\tassert.Equal(t, tt.wantMinKey, gotMinKey)\n\t\t\tassert.Equal(t, tt.wantHasMinKey, gotHasMinKey)\n\n\t\t\t// Test GetTasks\n\t\t\ttasks := tracker.GetTasks()\n\t\t\tassert.Equal(t, tt.wantTaskCount, tracker.GetPendingTaskCount())\n\n\t\t\t// Test GetPerDomainPendingTaskCount\n\t\t\tdomainCounts := tracker.GetPerDomainPendingTaskCount()\n\t\t\tassert.Equal(t, tt.wantDomainCounts, domainCounts, \"Per-domain task counts should match expected values\")\n\n\t\t\t// Verify all tasks are in the map\n\t\t\tfor _, task := range inputTasks {\n\t\t\t\tif tt.clear {\n\t\t\t\t\t// After clear, no tasks should be in the map\n\t\t\t\t\t_, exists := tasks[task.GetTaskKey()]\n\t\t\t\t\tassert.False(t, exists, \"After clear, no task should be in the map\")\n\t\t\t\t\tassert.Equal(t, ctask.TaskStateCanceled, task.State())\n\t\t\t\t} else if tt.pruneAcked && task.State() == ctask.TaskStateAcked {\n\t\t\t\t\t_, exists := tasks[task.GetTaskKey()]\n\t\t\t\t\tassert.False(t, exists, \"Acked task should not be in the map\")\n\t\t\t\t} else {\n\t\t\t\t\t_, exists := tasks[task.GetTaskKey()]\n\t\t\t\t\tassert.True(t, exists, \"Task should be in the map\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/predicate.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination predicate_mock.go github.com/uber/cadence/service/history/queuev2 Predicate\n\npackage queuev2\n\nimport (\n\t\"maps\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// Predicate defines a predicate that can be used to filter tasks\n\tPredicate interface {\n\t\t// IsEmpty returns true if no task satisfies the predicate\n\t\tIsEmpty() bool\n\t\t// Check returns true if the task satisfies the predicate\n\t\tCheck(task persistence.Task) bool\n\t\t// Equals returns true if the predicate is the same as the other predicate\n\t\tEquals(other Predicate) bool\n\t}\n\n\tdomainIDPredicate struct {\n\t\tdomainIDs   map[string]struct{}\n\t\tisExclusive bool\n\t}\n\n\tuniversalPredicate struct{}\n\n\temptyPredicate struct{}\n)\n\nfunc NewUniversalPredicate() Predicate {\n\treturn &universalPredicate{}\n}\n\nfunc (p *universalPredicate) IsEmpty() bool {\n\treturn false\n}\n\nfunc (p *universalPredicate) Check(task persistence.Task) bool {\n\treturn true\n}\n\nfunc (p *universalPredicate) Equals(other Predicate) bool {\n\t_, ok := other.(*universalPredicate)\n\treturn ok\n}\n\nfunc NewEmptyPredicate() Predicate {\n\treturn &emptyPredicate{}\n}\n\nfunc (p *emptyPredicate) IsEmpty() bool {\n\treturn true\n}\n\nfunc (p *emptyPredicate) Check(task persistence.Task) bool {\n\treturn false\n}\n\nfunc (p *emptyPredicate) Equals(other Predicate) bool {\n\t_, ok := other.(*emptyPredicate)\n\treturn ok\n}\n\nfunc NewDomainIDPredicate(domainIDs []string, isExclusive bool) Predicate {\n\tdomainIDSet := make(map[string]struct{})\n\tfor _, domainID := range domainIDs {\n\t\tdomainIDSet[domainID] = struct{}{}\n\t}\n\treturn &domainIDPredicate{\n\t\tdomainIDs:   domainIDSet,\n\t\tisExclusive: isExclusive,\n\t}\n}\n\nfunc (p *domainIDPredicate) IsEmpty() bool {\n\treturn len(p.domainIDs) == 0 && !p.isExclusive\n}\n\nfunc (p *domainIDPredicate) Check(task persistence.Task) bool {\n\tif _, ok := p.domainIDs[task.GetDomainID()]; ok {\n\t\treturn !p.isExclusive\n\t}\n\treturn p.isExclusive\n}\n\nfunc (p *domainIDPredicate) Equals(other Predicate) bool {\n\to, ok := other.(*domainIDPredicate)\n\tif !ok {\n\t\treturn false\n\t}\n\treturn p.isExclusive == o.isExclusive && maps.Equal(p.domainIDs, o.domainIDs)\n}\n"
  },
  {
    "path": "service/history/queuev2/predicate_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: Predicate)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination predicate_mock.go github.com/uber/cadence/service/history/queuev2 Predicate\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockPredicate is a mock of Predicate interface.\ntype MockPredicate struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPredicateMockRecorder\n\tisgomock struct{}\n}\n\n// MockPredicateMockRecorder is the mock recorder for MockPredicate.\ntype MockPredicateMockRecorder struct {\n\tmock *MockPredicate\n}\n\n// NewMockPredicate creates a new mock instance.\nfunc NewMockPredicate(ctrl *gomock.Controller) *MockPredicate {\n\tmock := &MockPredicate{ctrl: ctrl}\n\tmock.recorder = &MockPredicateMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPredicate) EXPECT() *MockPredicateMockRecorder {\n\treturn m.recorder\n}\n\n// Check mocks base method.\nfunc (m *MockPredicate) Check(task persistence.Task) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Check\", task)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Check indicates an expected call of Check.\nfunc (mr *MockPredicateMockRecorder) Check(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Check\", reflect.TypeOf((*MockPredicate)(nil).Check), task)\n}\n\n// Equals mocks base method.\nfunc (m *MockPredicate) Equals(other Predicate) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Equals\", other)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Equals indicates an expected call of Equals.\nfunc (mr *MockPredicateMockRecorder) Equals(other any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Equals\", reflect.TypeOf((*MockPredicate)(nil).Equals), other)\n}\n\n// IsEmpty mocks base method.\nfunc (m *MockPredicate) IsEmpty() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsEmpty\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsEmpty indicates an expected call of IsEmpty.\nfunc (mr *MockPredicateMockRecorder) IsEmpty() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsEmpty\", reflect.TypeOf((*MockPredicate)(nil).IsEmpty))\n}\n"
  },
  {
    "path": "service/history/queuev2/predicate_operation.go",
    "content": "package queuev2\n\nimport \"fmt\"\n\nfunc Not(predicate Predicate) Predicate {\n\tswitch p := predicate.(type) {\n\tcase *universalPredicate:\n\t\treturn &emptyPredicate{}\n\tcase *emptyPredicate:\n\t\treturn &universalPredicate{}\n\tcase *domainIDPredicate:\n\t\tisExclusive := !p.isExclusive\n\t\tif isExclusive && len(p.domainIDs) == 0 {\n\t\t\treturn &universalPredicate{}\n\t\t}\n\t\treturn &domainIDPredicate{\n\t\t\tdomainIDs:   p.domainIDs,\n\t\t\tisExclusive: isExclusive,\n\t\t}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p))\n\t}\n}\n\nfunc And(p1, p2 Predicate) Predicate {\n\tswitch p1 := p1.(type) {\n\tcase *universalPredicate:\n\t\treturn p2\n\tcase *emptyPredicate:\n\t\treturn p1\n\tcase *domainIDPredicate:\n\t\tswitch p2 := p2.(type) {\n\t\tcase *universalPredicate:\n\t\t\treturn p1\n\t\tcase *emptyPredicate:\n\t\t\treturn p2\n\t\tcase *domainIDPredicate:\n\t\t\tif p1.isExclusive {\n\t\t\t\tif p2.isExclusive {\n\t\t\t\t\tdomainIDs := unionStringSet(p1.domainIDs, p2.domainIDs)\n\t\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\t\treturn &emptyPredicate{}\n\t\t\t\t\t}\n\t\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\t\tisExclusive: true,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tdomainIDs := map[string]struct{}{}\n\t\t\t\tfor domainID := range p2.domainIDs {\n\t\t\t\t\tif _, ok := p1.domainIDs[domainID]; !ok {\n\t\t\t\t\t\tdomainIDs[domainID] = struct{}{}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\treturn &emptyPredicate{}\n\t\t\t\t}\n\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\tisExclusive: false,\n\t\t\t\t}\n\t\t\t}\n\t\t\tif p2.isExclusive {\n\t\t\t\tdomainIDs := map[string]struct{}{}\n\t\t\t\tfor domainID := range p1.domainIDs {\n\t\t\t\t\tif _, ok := p2.domainIDs[domainID]; !ok {\n\t\t\t\t\t\tdomainIDs[domainID] = struct{}{}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\treturn &emptyPredicate{}\n\t\t\t\t}\n\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\tisExclusive: false,\n\t\t\t\t}\n\t\t\t}\n\t\t\tdomainIDs := intersectStringSet(p1.domainIDs, p2.domainIDs)\n\t\t\tif len(domainIDs) == 0 {\n\t\t\t\treturn &emptyPredicate{}\n\t\t\t}\n\t\t\treturn &domainIDPredicate{\n\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\tisExclusive: false,\n\t\t\t}\n\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p2))\n\t\t}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p1))\n\t}\n}\n\nfunc Or(p1, p2 Predicate) Predicate {\n\tswitch p1 := p1.(type) {\n\tcase *universalPredicate:\n\t\treturn p1\n\tcase *emptyPredicate:\n\t\treturn p2\n\tcase *domainIDPredicate:\n\t\tswitch p2 := p2.(type) {\n\t\tcase *universalPredicate:\n\t\t\treturn p2\n\t\tcase *emptyPredicate:\n\t\t\treturn p1\n\t\tcase *domainIDPredicate:\n\t\t\tif p1.isExclusive {\n\t\t\t\tif p2.isExclusive {\n\t\t\t\t\tdomainIDs := intersectStringSet(p1.domainIDs, p2.domainIDs)\n\t\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\t\treturn &universalPredicate{}\n\t\t\t\t\t}\n\t\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\t\tisExclusive: true,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tdomainIDs := diffStringSet(p1.domainIDs, p2.domainIDs)\n\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\treturn &universalPredicate{}\n\t\t\t\t}\n\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\tisExclusive: true,\n\t\t\t\t}\n\t\t\t}\n\t\t\tif p2.isExclusive {\n\t\t\t\tdomainIDs := diffStringSet(p2.domainIDs, p1.domainIDs)\n\t\t\t\tif len(domainIDs) == 0 {\n\t\t\t\t\treturn &universalPredicate{}\n\t\t\t\t}\n\t\t\t\treturn &domainIDPredicate{\n\t\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\t\tisExclusive: true,\n\t\t\t\t}\n\t\t\t}\n\t\t\tdomainIDs := unionStringSet(p1.domainIDs, p2.domainIDs)\n\t\t\tif len(domainIDs) == 0 {\n\t\t\t\treturn &emptyPredicate{}\n\t\t\t}\n\t\t\treturn &domainIDPredicate{\n\t\t\t\tdomainIDs:   domainIDs,\n\t\t\t\tisExclusive: false,\n\t\t\t}\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p2))\n\t\t}\n\tdefault:\n\t\tpanic(fmt.Sprintf(\"unknown predicate type: %T\", p1))\n\t}\n}\n\nfunc unionStringSet(set1, set2 map[string]struct{}) map[string]struct{} {\n\tresult := make(map[string]struct{})\n\tfor domainID := range set1 {\n\t\tresult[domainID] = struct{}{}\n\t}\n\tfor domainID := range set2 {\n\t\tresult[domainID] = struct{}{}\n\t}\n\treturn result\n}\n\nfunc intersectStringSet(set1, set2 map[string]struct{}) map[string]struct{} {\n\tresult := make(map[string]struct{})\n\tfor domainID := range set1 {\n\t\tif _, ok := set2[domainID]; ok {\n\t\t\tresult[domainID] = struct{}{}\n\t\t}\n\t}\n\treturn result\n}\n\nfunc diffStringSet(set1, set2 map[string]struct{}) map[string]struct{} {\n\tresult := make(map[string]struct{})\n\tfor domainID := range set1 {\n\t\tif _, ok := set2[domainID]; !ok {\n\t\t\tresult[domainID] = struct{}{}\n\t\t}\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "service/history/queuev2/predicate_operation_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\n\tfuzz \"github.com/google/gofuzz\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestNot(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tinput    Predicate\n\t\texpected Predicate\n\t}{\n\t\t{\n\t\t\tname:     \"universalPredicate returns emptyPredicate\",\n\t\t\tinput:    NewUniversalPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"emptyPredicate returns universalPredicate\",\n\t\t\tinput:    NewEmptyPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with isExclusive=true returns isExclusive=false\",\n\t\t\tinput:    NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with isExclusive=false returns isExclusive=true\",\n\t\t\tinput:    NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with empty domains and isExclusive=true\",\n\t\t\tinput:    NewDomainIDPredicate([]string{}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with empty domains and isExclusive=false\",\n\t\t\tinput:    NewDomainIDPredicate([]string{}, false),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with single domain\",\n\t\t\tinput:    NewDomainIDPredicate([]string{\"single-domain\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"single-domain\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"domainIDPredicate with multiple domains\",\n\t\t\tinput:    NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, true),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := Not(tt.input)\n\t\t\tassert.NotNil(t, result)\n\n\t\t\t// Use the Equals method to compare predicates\n\t\t\tassert.True(t, tt.expected.Equals(result),\n\t\t\t\t\"expected %+v, got %+v\", tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestAnd(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tp1       Predicate\n\t\tp2       Predicate\n\t\texpected Predicate\n\t}{\n\t\t// universalPredicate cases\n\t\t{\n\t\t\tname:     \"universalPredicate AND universalPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"universalPredicate AND emptyPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"universalPredicate AND domainIDPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\n\t\t// emptyPredicate cases\n\t\t{\n\t\t\tname:     \"emptyPredicate AND universalPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"emptyPredicate AND emptyPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"emptyPredicate AND domainIDPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\n\t\t// domainIDPredicate AND universalPredicate\n\t\t{\n\t\t\tname:     \"domainIDPredicate AND universalPredicate\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\n\t\t// domainIDPredicate AND emptyPredicate\n\t\t{\n\t\t\tname:     \"domainIDPredicate AND emptyPredicate\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\n\t\t// domainIDPredicate AND domainIDPredicate cases\n\t\t{\n\t\t\tname:     \"exclusive AND exclusive - union domains\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive AND exclusive - overlapping domains\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive AND inclusive - p2 domains not in p1\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive AND inclusive - all p2 domains in p1\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive AND exclusive - p1 domains not in p2\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive AND exclusive - all p1 domains in p2\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive AND inclusive - intersection\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive AND inclusive - no intersection\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive AND inclusive - same domains\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\n\t\t// Edge cases with empty domain lists\n\t\t{\n\t\t\tname:     \"empty exclusive AND non-empty exclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty inclusive AND non-empty inclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty exclusive AND non-empty inclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"non-empty inclusive AND empty exclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := And(tt.p1, tt.p2)\n\t\t\tassert.NotNil(t, result)\n\n\t\t\t// Use the Equals method to compare predicates\n\t\t\tassert.True(t, tt.expected.Equals(result),\n\t\t\t\t\"expected %+v, got %+v\", tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc predicateOperationFuzzGenerator(p *Predicate, c fuzz.Continue) {\n\tswitch c.Intn(3) {\n\tcase 0:\n\t\t*p = NewUniversalPredicate()\n\tcase 1:\n\t\t*p = NewEmptyPredicate()\n\tcase 2:\n\t\tvar domainIDPredicate domainIDPredicate\n\t\tc.Fuzz(&domainIDPredicate)\n\t\t*p = &domainIDPredicate\n\tdefault:\n\t\tpanic(\"invalid predicate type\")\n\t}\n}\n\nfunc TestAnd_Commutativity(t *testing.T) {\n\tf := fuzz.New().Funcs(predicateOperationFuzzGenerator)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar p1, p2 Predicate\n\t\tf.Fuzz(&p1)\n\t\tf.Fuzz(&p2)\n\n\t\tresult1 := And(p1, p2)\n\t\tresult2 := And(p2, p1)\n\n\t\tassert.True(t, result1.Equals(result2),\n\t\t\t\"And should be commutative: And(p1, p2) should equal And(p2, p1)\")\n\t}\n}\n\nfunc TestAnd_LogicalCorrectness(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Test data with different domain IDs\n\ttestTasks := []*persistence.MockTask{\n\t\t// Task with domain1\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain2\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain3\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain3\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain4\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain4\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t}\n\n\t// Test cases with different predicate combinations\n\ttestCases := []struct {\n\t\tname string\n\t\tp1   Predicate\n\t\tp2   Predicate\n\t}{\n\t\t// Universal and Empty predicates\n\t\t{\n\t\t\tname: \"universal AND universal\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"universal AND empty\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty AND universal\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty AND empty\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\n\t\t// Universal with domain predicates\n\t\t{\n\t\t\tname: \"universal AND domain inclusive\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"universal AND domain exclusive\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"domain inclusive AND universal\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, false),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"domain exclusive AND universal\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, true),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\n\t\t// Empty with domain predicates\n\t\t{\n\t\t\tname: \"empty AND domain inclusive\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"empty AND domain exclusive\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"domain inclusive AND empty\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, false),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"domain exclusive AND empty\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\n\t\t// Domain predicate combinations\n\t\t{\n\t\t\tname: \"inclusive AND inclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"inclusive AND inclusive - disjoint\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive AND exclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive AND exclusive - disjoint\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"inclusive AND exclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive AND inclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t},\n\n\t\t// Edge cases with empty domain lists\n\t\t{\n\t\t\tname: \"empty inclusive AND non-empty inclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"empty exclusive AND non-empty exclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"empty inclusive AND empty exclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{}, true),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tandPredicate := And(tc.p1, tc.p2)\n\n\t\t\t// Test the logical property for each task\n\t\t\tfor i, task := range testTasks {\n\t\t\t\tp1Result := tc.p1.Check(task)\n\t\t\t\tp2Result := tc.p2.Check(task)\n\t\t\t\texpectedResult := p1Result && p2Result\n\t\t\t\tactualResult := andPredicate.Check(task)\n\n\t\t\t\tassert.Equal(t, expectedResult, actualResult,\n\t\t\t\t\t\"For task %d (domain=%s): And(p1, p2).Check(task) should equal p1.Check(task) && p2.Check(task). \"+\n\t\t\t\t\t\t\"p1.Check(task)=%t, p2.Check(task)=%t, expected=%t, actual=%t\",\n\t\t\t\t\ti, task.GetDomainID(), p1Result, p2Result, expectedResult, actualResult)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAnd_LogicalCorrectness_FuzzTesting(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Create a variety of test tasks with different domain IDs\n\tdomains := []string{\"domain1\", \"domain2\", \"domain3\", \"domain4\", \"domain5\", \"\"}\n\ttestTasks := make([]*persistence.MockTask, len(domains))\n\tfor i, domain := range domains {\n\t\ttask := persistence.NewMockTask(ctrl)\n\t\ttask.EXPECT().GetDomainID().Return(domain).AnyTimes()\n\t\ttestTasks[i] = task\n\t}\n\n\tf := fuzz.New().Funcs(predicateOperationFuzzGenerator)\n\tfor i := 0; i < 500; i++ {\n\t\tvar p1, p2 Predicate\n\t\tf.Fuzz(&p1)\n\t\tf.Fuzz(&p2)\n\n\t\tandPredicate := And(p1, p2)\n\n\t\t// Test the logical property for each task\n\t\tfor _, task := range testTasks {\n\t\t\tp1Result := p1.Check(task)\n\t\t\tp2Result := p2.Check(task)\n\t\t\texpectedResult := p1Result && p2Result\n\t\t\tactualResult := andPredicate.Check(task)\n\n\t\t\tassert.Equal(t, expectedResult, actualResult,\n\t\t\t\t\"Fuzz test iteration %d: And(p1, p2).Check(task) should equal p1.Check(task) && p2.Check(task). \"+\n\t\t\t\t\t\"Task domain: %s, p1.Check(task)=%t, p2.Check(task)=%t, expected=%t, actual=%t, \"+\n\t\t\t\t\t\"p1=%+v, p2=%+v\",\n\t\t\t\ti, task.GetDomainID(), p1Result, p2Result, expectedResult, actualResult, p1, p2)\n\t\t}\n\t}\n}\n\nfunc TestOr(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tp1       Predicate\n\t\tp2       Predicate\n\t\texpected Predicate\n\t}{\n\t\t// universalPredicate cases\n\t\t{\n\t\t\tname:     \"universalPredicate OR universalPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"universalPredicate OR emptyPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"universalPredicate OR domainIDPredicate\",\n\t\t\tp1:       NewUniversalPredicate(),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\n\t\t// emptyPredicate cases\n\t\t{\n\t\t\tname:     \"emptyPredicate OR universalPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"emptyPredicate OR emptyPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"emptyPredicate OR domainIDPredicate\",\n\t\t\tp1:       NewEmptyPredicate(),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\n\t\t// domainIDPredicate OR universalPredicate\n\t\t{\n\t\t\tname:     \"domainIDPredicate OR universalPredicate\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewUniversalPredicate(),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\n\t\t// domainIDPredicate OR emptyPredicate\n\t\t{\n\t\t\tname:     \"domainIDPredicate OR emptyPredicate\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewEmptyPredicate(),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\n\t\t// domainIDPredicate OR domainIDPredicate cases\n\t\t{\n\t\t\tname:     \"exclusive OR exclusive - intersection exists\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive OR exclusive - no intersection\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, true),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive OR inclusive - p1 excludes some of p2\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive OR inclusive - p1 excludes all of p2\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"exclusive OR inclusive - p1 excludes none of p2\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive OR exclusive - p2 excludes some of p1\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain4\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive OR exclusive - p2 excludes all of p1\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive OR exclusive - p2 excludes none of p1\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive OR inclusive - union\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"inclusive OR inclusive - same domains\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\n\t\t// Edge cases with empty domain lists\n\t\t{\n\t\t\tname:     \"empty exclusive OR non-empty exclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty inclusive OR non-empty inclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\texpected: NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty exclusive OR non-empty inclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"non-empty inclusive OR empty exclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{}, true),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty exclusive OR empty exclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:       NewDomainIDPredicate([]string{}, true),\n\t\t\texpected: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname:     \"empty inclusive OR empty inclusive\",\n\t\t\tp1:       NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:       NewDomainIDPredicate([]string{}, false),\n\t\t\texpected: NewEmptyPredicate(),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := Or(tt.p1, tt.p2)\n\t\t\tassert.NotNil(t, result)\n\n\t\t\t// Use the Equals method to compare predicates\n\t\t\tassert.True(t, tt.expected.Equals(result),\n\t\t\t\t\"expected %+v, got %+v\", tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestOr_Commutativity(t *testing.T) {\n\tf := fuzz.New().Funcs(predicateOperationFuzzGenerator)\n\tfor i := 0; i < 1000; i++ {\n\t\tvar p1, p2 Predicate\n\t\tf.Fuzz(&p1)\n\t\tf.Fuzz(&p2)\n\n\t\tresult1 := Or(p1, p2)\n\t\tresult2 := Or(p2, p1)\n\n\t\tassert.True(t, result1.Equals(result2),\n\t\t\t\"Or should be commutative: Or(p1, p2) should equal Or(p2, p1)\")\n\t}\n}\n\nfunc TestOr_LogicalCorrectness(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Test data with different domain IDs\n\ttestTasks := []*persistence.MockTask{\n\t\t// Task with domain1\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain2\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain3\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain3\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t\t// Task with domain4\n\t\tfunc() *persistence.MockTask {\n\t\t\ttask := persistence.NewMockTask(ctrl)\n\t\t\ttask.EXPECT().GetDomainID().Return(\"domain4\").AnyTimes()\n\t\t\treturn task\n\t\t}(),\n\t}\n\n\t// Test cases with different predicate combinations\n\ttestCases := []struct {\n\t\tname string\n\t\tp1   Predicate\n\t\tp2   Predicate\n\t}{\n\t\t// Universal and Empty predicates\n\t\t{\n\t\t\tname: \"universal OR universal\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"universal OR empty\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty OR universal\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"empty OR empty\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\n\t\t// Universal with domain predicates\n\t\t{\n\t\t\tname: \"universal OR domain inclusive\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"universal OR domain exclusive\",\n\t\t\tp1:   NewUniversalPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"domain inclusive OR universal\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, false),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"domain exclusive OR universal\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, true),\n\t\t\tp2:   NewUniversalPredicate(),\n\t\t},\n\n\t\t// Empty with domain predicates\n\t\t{\n\t\t\tname: \"empty OR domain inclusive\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"empty OR domain exclusive\",\n\t\t\tp1:   NewEmptyPredicate(),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"domain inclusive OR empty\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, false),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\t\t{\n\t\t\tname: \"domain exclusive OR empty\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t\tp2:   NewEmptyPredicate(),\n\t\t},\n\n\t\t// Domain predicate combinations\n\t\t{\n\t\t\tname: \"inclusive OR inclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"inclusive OR inclusive - disjoint\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive OR exclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive OR exclusive - disjoint\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain3\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"inclusive OR exclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain4\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"exclusive OR inclusive - overlapping\",\n\t\t\tp1:   NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t},\n\n\t\t// Edge cases with empty domain lists\n\t\t{\n\t\t\tname: \"empty inclusive OR non-empty inclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t},\n\t\t{\n\t\t\tname: \"empty exclusive OR non-empty exclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, true),\n\t\t\tp2:   NewDomainIDPredicate([]string{\"domain1\"}, true),\n\t\t},\n\t\t{\n\t\t\tname: \"empty inclusive OR empty exclusive\",\n\t\t\tp1:   NewDomainIDPredicate([]string{}, false),\n\t\t\tp2:   NewDomainIDPredicate([]string{}, true),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\torPredicate := Or(tc.p1, tc.p2)\n\n\t\t\t// Test the logical property for each task\n\t\t\tfor i, task := range testTasks {\n\t\t\t\tp1Result := tc.p1.Check(task)\n\t\t\t\tp2Result := tc.p2.Check(task)\n\t\t\t\texpectedResult := p1Result || p2Result\n\t\t\t\tactualResult := orPredicate.Check(task)\n\n\t\t\t\tassert.Equal(t, expectedResult, actualResult,\n\t\t\t\t\t\"For task %d (domain=%s): Or(p1, p2).Check(task) should equal p1.Check(task) || p2.Check(task). \"+\n\t\t\t\t\t\t\"p1.Check(task)=%t, p2.Check(task)=%t, expected=%t, actual=%t\",\n\t\t\t\t\ti, task.GetDomainID(), p1Result, p2Result, expectedResult, actualResult)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestOr_LogicalCorrectness_FuzzTesting(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Create a variety of test tasks with different domain IDs\n\tdomains := []string{\"domain1\", \"domain2\", \"domain3\", \"domain4\", \"domain5\", \"\"}\n\ttestTasks := make([]*persistence.MockTask, len(domains))\n\tfor i, domain := range domains {\n\t\ttask := persistence.NewMockTask(ctrl)\n\t\ttask.EXPECT().GetDomainID().Return(domain).AnyTimes()\n\t\ttestTasks[i] = task\n\t}\n\n\tf := fuzz.New().Funcs(predicateOperationFuzzGenerator)\n\tfor i := 0; i < 500; i++ {\n\t\tvar p1, p2 Predicate\n\t\tf.Fuzz(&p1)\n\t\tf.Fuzz(&p2)\n\n\t\torPredicate := Or(p1, p2)\n\n\t\t// Test the logical property for each task\n\t\tfor _, task := range testTasks {\n\t\t\tp1Result := p1.Check(task)\n\t\t\tp2Result := p2.Check(task)\n\t\t\texpectedResult := p1Result || p2Result\n\t\t\tactualResult := orPredicate.Check(task)\n\n\t\t\tassert.Equal(t, expectedResult, actualResult,\n\t\t\t\t\"Fuzz test iteration %d: Or(p1, p2).Check(task) should equal p1.Check(task) || p2.Check(task). \"+\n\t\t\t\t\t\"Task domain: %s, p1.Check(task)=%t, p2.Check(task)=%t, expected=%t, actual=%t, \"+\n\t\t\t\t\t\"p1=%+v, p2=%+v\",\n\t\t\t\ti, task.GetDomainID(), p1Result, p2Result, expectedResult, actualResult, p1, p2)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/predicate_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestNewUniversalPredicate(t *testing.T) {\n\tpredicate := NewUniversalPredicate()\n\tassert.NotNil(t, predicate)\n\tassert.IsType(t, &universalPredicate{}, predicate)\n}\n\nfunc TestUniversalPredicate_IsEmpty(t *testing.T) {\n\tpredicate := NewUniversalPredicate()\n\tassert.False(t, predicate.IsEmpty())\n}\n\nfunc TestUniversalPredicate_Check(t *testing.T) {\n\tpredicate := NewUniversalPredicate()\n\n\t// Test with nil task\n\tassert.True(t, predicate.Check(nil))\n\n\t// Test with mock task\n\tctrl := gomock.NewController(t)\n\n\ttask := persistence.NewMockTask(ctrl)\n\ttask.EXPECT().GetDomainID().Return(\"test-domain\").AnyTimes()\n\tassert.True(t, predicate.Check(task))\n\n\t// Test with different domain ID\n\ttask2 := persistence.NewMockTask(ctrl)\n\ttask2.EXPECT().GetDomainID().Return(\"different-domain\").AnyTimes()\n\tassert.True(t, predicate.Check(task2))\n}\n\nfunc TestUniversalPredicate_Equals(t *testing.T) {\n\tpredicate1 := NewUniversalPredicate()\n\tpredicate2 := NewUniversalPredicate()\n\temptyPredicate := NewEmptyPredicate()\n\tdomainPredicate := NewDomainIDPredicate([]string{\"test\"}, false)\n\n\t// Same type should be equal\n\tassert.True(t, predicate1.Equals(predicate2))\n\tassert.True(t, predicate2.Equals(predicate1))\n\n\t// Different types should not be equal\n\tassert.False(t, predicate1.Equals(emptyPredicate))\n\tassert.False(t, predicate1.Equals(domainPredicate))\n\tassert.False(t, predicate1.Equals(nil))\n}\n\nfunc TestNewEmptyPredicate(t *testing.T) {\n\tpredicate := NewEmptyPredicate()\n\tassert.NotNil(t, predicate)\n\tassert.IsType(t, &emptyPredicate{}, predicate)\n}\n\nfunc TestEmptyPredicate_IsEmpty(t *testing.T) {\n\tpredicate := NewEmptyPredicate()\n\tassert.True(t, predicate.IsEmpty())\n}\n\nfunc TestEmptyPredicate_Check(t *testing.T) {\n\tpredicate := NewEmptyPredicate()\n\n\t// Test with nil task\n\tassert.False(t, predicate.Check(nil))\n\n\t// Test with mock task\n\tctrl := gomock.NewController(t)\n\n\ttask := persistence.NewMockTask(ctrl)\n\ttask.EXPECT().GetDomainID().Return(\"test-domain\").AnyTimes()\n\tassert.False(t, predicate.Check(task))\n\n\t// Test with different domain ID\n\ttask2 := persistence.NewMockTask(ctrl)\n\ttask2.EXPECT().GetDomainID().Return(\"different-domain\").AnyTimes()\n\tassert.False(t, predicate.Check(task2))\n}\n\nfunc TestEmptyPredicate_Equals(t *testing.T) {\n\tpredicate1 := NewEmptyPredicate()\n\tpredicate2 := NewEmptyPredicate()\n\tuniversalPredicate := NewUniversalPredicate()\n\tdomainPredicate := NewDomainIDPredicate([]string{\"test\"}, false)\n\n\t// Same type should be equal\n\tassert.True(t, predicate1.Equals(predicate2))\n\tassert.True(t, predicate2.Equals(predicate1))\n\n\t// Different types should not be equal\n\tassert.False(t, predicate1.Equals(universalPredicate))\n\tassert.False(t, predicate1.Equals(domainPredicate))\n\tassert.False(t, predicate1.Equals(nil))\n}\n\nfunc TestNewDomainIDPredicate(t *testing.T) {\n\t// Test with empty domain IDs\n\tpredicate := NewDomainIDPredicate([]string{}, false)\n\tassert.NotNil(t, predicate)\n\tassert.IsType(t, &domainIDPredicate{}, predicate)\n\n\t// Test with single domain ID\n\tpredicate = NewDomainIDPredicate([]string{\"domain1\"}, false)\n\tassert.NotNil(t, predicate)\n\n\t// Test with multiple domain IDs\n\tpredicate = NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\"}, true)\n\tassert.NotNil(t, predicate)\n\n\t// Test with duplicate domain IDs (should be deduplicated)\n\tpredicate = NewDomainIDPredicate([]string{\"domain1\", \"domain1\", \"domain2\"}, false)\n\tassert.NotNil(t, predicate)\n}\n\nfunc TestDomainIDPredicate_IsEmpty(t *testing.T) {\n\t// Empty domain IDs with inclusive mode\n\tpredicate := NewDomainIDPredicate([]string{}, false)\n\tassert.True(t, predicate.IsEmpty())\n\n\t// Empty domain IDs with exclusive mode\n\tpredicate = NewDomainIDPredicate([]string{}, true)\n\tassert.False(t, predicate.IsEmpty())\n\n\t// Non-empty domain IDs with inclusive mode\n\tpredicate = NewDomainIDPredicate([]string{\"domain1\"}, false)\n\tassert.False(t, predicate.IsEmpty())\n\n\t// Non-empty domain IDs with exclusive mode\n\tpredicate = NewDomainIDPredicate([]string{\"domain1\"}, true)\n\tassert.False(t, predicate.IsEmpty())\n}\n\nfunc TestDomainIDPredicate_Check_Inclusive(t *testing.T) {\n\t// Test inclusive mode (isExclusive = false)\n\tpredicate := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false)\n\n\tctrl := gomock.NewController(t)\n\n\t// Task with domain in the list should pass\n\ttask1 := persistence.NewMockTask(ctrl)\n\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\tassert.True(t, predicate.Check(task1))\n\n\ttask2 := persistence.NewMockTask(ctrl)\n\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\tassert.True(t, predicate.Check(task2))\n\n\t// Task with domain not in the list should fail\n\ttask3 := persistence.NewMockTask(ctrl)\n\ttask3.EXPECT().GetDomainID().Return(\"domain3\").AnyTimes()\n\tassert.False(t, predicate.Check(task3))\n\n\t// Test with empty domain ID\n\ttask4 := persistence.NewMockTask(ctrl)\n\ttask4.EXPECT().GetDomainID().Return(\"\").AnyTimes()\n\tassert.False(t, predicate.Check(task4))\n}\n\nfunc TestDomainIDPredicate_Check_Exclusive(t *testing.T) {\n\t// Test exclusive mode (isExclusive = true)\n\tpredicate := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true)\n\n\tctrl := gomock.NewController(t)\n\n\t// Task with domain in the list should fail\n\ttask1 := persistence.NewMockTask(ctrl)\n\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\tassert.False(t, predicate.Check(task1))\n\n\ttask2 := persistence.NewMockTask(ctrl)\n\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\tassert.False(t, predicate.Check(task2))\n\n\t// Task with domain not in the list should pass\n\ttask3 := persistence.NewMockTask(ctrl)\n\ttask3.EXPECT().GetDomainID().Return(\"domain3\").AnyTimes()\n\tassert.True(t, predicate.Check(task3))\n\n\t// Test with empty domain ID\n\ttask4 := persistence.NewMockTask(ctrl)\n\ttask4.EXPECT().GetDomainID().Return(\"\").AnyTimes()\n\tassert.True(t, predicate.Check(task4))\n}\n\nfunc TestDomainIDPredicate_Check_EmptyList(t *testing.T) {\n\t// Test with empty domain list and inclusive mode\n\tpredicate := NewDomainIDPredicate([]string{}, false)\n\n\tctrl := gomock.NewController(t)\n\n\ttask := persistence.NewMockTask(ctrl)\n\ttask.EXPECT().GetDomainID().Return(\"any-domain\").AnyTimes()\n\tassert.False(t, predicate.Check(task)) // No domains in list, so inclusive returns false\n\n\t// Test with empty domain list and exclusive mode\n\tpredicate = NewDomainIDPredicate([]string{}, true)\n\tassert.True(t, predicate.Check(task)) // No domains in list, so exclusive returns true\n}\n\nfunc TestDomainIDPredicate_Equals(t *testing.T) {\n\t// Test equal predicates\n\tpredicate1 := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false)\n\tpredicate2 := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false)\n\tassert.True(t, predicate1.Equals(predicate2))\n\n\t// Test equal predicates with different order\n\tpredicate3 := NewDomainIDPredicate([]string{\"domain2\", \"domain1\"}, false)\n\tassert.True(t, predicate1.Equals(predicate3))\n\n\t// Test different domain sets\n\tpredicate4 := NewDomainIDPredicate([]string{\"domain1\", \"domain3\"}, false)\n\tassert.False(t, predicate1.Equals(predicate4))\n\n\t// Test different domain count\n\tpredicate5 := NewDomainIDPredicate([]string{\"domain1\"}, false)\n\tassert.False(t, predicate1.Equals(predicate5))\n\n\t// Test different exclusion mode\n\tpredicate6 := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, true)\n\tassert.False(t, predicate1.Equals(predicate6))\n\n\t// Test against different predicate types\n\tuniversalPredicate := NewUniversalPredicate()\n\temptyPredicate := NewEmptyPredicate()\n\tassert.False(t, predicate1.Equals(universalPredicate))\n\tassert.False(t, predicate1.Equals(emptyPredicate))\n\tassert.False(t, predicate1.Equals(nil))\n}\n\nfunc TestDomainIDPredicate_Equals_EmptyLists(t *testing.T) {\n\t// Test with both empty lists\n\tpredicate1 := NewDomainIDPredicate([]string{}, false)\n\tpredicate2 := NewDomainIDPredicate([]string{}, false)\n\tassert.True(t, predicate1.Equals(predicate2))\n\n\t// Test with one empty, one non-empty\n\tpredicate3 := NewDomainIDPredicate([]string{\"domain1\"}, false)\n\tassert.False(t, predicate1.Equals(predicate3))\n\n\t// Test with both empty but different exclusion mode\n\tpredicate4 := NewDomainIDPredicate([]string{}, true)\n\tassert.False(t, predicate1.Equals(predicate4))\n}\n\nfunc TestDomainIDPredicate_DuplicateDomains(t *testing.T) {\n\t// Test that duplicate domains are handled correctly\n\tpredicate := NewDomainIDPredicate([]string{\"domain1\", \"domain1\", \"domain2\", \"domain2\"}, false)\n\n\tctrl := gomock.NewController(t)\n\n\ttask1 := persistence.NewMockTask(ctrl)\n\ttask1.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\ttask2 := persistence.NewMockTask(ctrl)\n\ttask2.EXPECT().GetDomainID().Return(\"domain2\").AnyTimes()\n\ttask3 := persistence.NewMockTask(ctrl)\n\ttask3.EXPECT().GetDomainID().Return(\"domain3\").AnyTimes()\n\n\tassert.True(t, predicate.Check(task1))\n\tassert.True(t, predicate.Check(task2))\n\tassert.False(t, predicate.Check(task3))\n\n\t// Test equals with duplicate domains\n\tpredicate2 := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false)\n\tassert.True(t, predicate.Equals(predicate2))\n}\n\nfunc TestPredicateIntegration(t *testing.T) {\n\t// Test integration between different predicate types\n\tuniversal := NewUniversalPredicate()\n\tempty := NewEmptyPredicate()\n\tdomainInclusive := NewDomainIDPredicate([]string{\"domain1\"}, false)\n\tdomainExclusive := NewDomainIDPredicate([]string{\"domain1\"}, true)\n\n\tctrl := gomock.NewController(t)\n\n\ttask := persistence.NewMockTask(ctrl)\n\ttask.EXPECT().GetDomainID().Return(\"domain1\").AnyTimes()\n\n\t// Universal should always pass\n\tassert.True(t, universal.Check(task))\n\tassert.False(t, universal.IsEmpty())\n\n\t// Empty should always fail\n\tassert.False(t, empty.Check(task))\n\tassert.True(t, empty.IsEmpty())\n\n\t// Domain inclusive should pass for matching domain\n\tassert.True(t, domainInclusive.Check(task))\n\tassert.False(t, domainInclusive.IsEmpty())\n\n\t// Domain exclusive should fail for matching domain\n\tassert.False(t, domainExclusive.Check(task))\n\tassert.False(t, domainExclusive.IsEmpty())\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_base.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\talertChSize = 10\n\t// Non-default readers will use critical pending task count * this coefficient\n\t// as its max pending task count so that their loading will never trigger pending\n\t// task alert & action\n\tnonRootQueueMaxPendingTaskCoefficient = 0.8\n)\n\ntype (\n\tOptions struct {\n\t\tPageSize                             dynamicproperties.IntPropertyFn\n\t\tDeleteBatchSize                      dynamicproperties.IntPropertyFn\n\t\tMaxPollRPS                           dynamicproperties.IntPropertyFn\n\t\tMaxPollInterval                      dynamicproperties.DurationPropertyFn\n\t\tMaxPollIntervalJitterCoefficient     dynamicproperties.FloatPropertyFn\n\t\tUpdateAckInterval                    dynamicproperties.DurationPropertyFn\n\t\tUpdateAckIntervalJitterCoefficient   dynamicproperties.FloatPropertyFn\n\t\tRedispatchInterval                   dynamicproperties.DurationPropertyFn\n\t\tMaxPendingTasksCount                 dynamicproperties.IntPropertyFn\n\t\tPollBackoffInterval                  dynamicproperties.DurationPropertyFn\n\t\tPollBackoffIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\t\tVirtualSliceForceAppendInterval      dynamicproperties.DurationPropertyFn\n\t\t// monitor & mitigator options\n\t\tCriticalPendingTaskCount    dynamicproperties.IntPropertyFn\n\t\tEnablePendingTaskCountAlert func() bool\n\t\tMaxVirtualQueueCount        dynamicproperties.IntPropertyFn\n\n\t\tEnableValidator        dynamicproperties.BoolPropertyFn\n\t\tValidationInterval     dynamicproperties.DurationPropertyFn\n\t\tMaxStartJitterInterval dynamicproperties.DurationPropertyFn\n\t}\n\n\tqueueBase struct {\n\t\tshard           shard.Context\n\t\ttaskProcessor   task.Processor\n\t\tlogger          log.Logger\n\t\tmetricsClient   metrics.Client\n\t\tmetricsScope    metrics.Scope\n\t\tcategory        persistence.HistoryTaskCategory\n\t\toptions         *Options\n\t\ttimeSource      clock.TimeSource\n\t\ttaskInitializer task.Initializer\n\n\t\trescheduler           task.Rescheduler\n\t\tqueueReader           QueueReader\n\t\tmonitor               Monitor\n\t\tmitigator             Mitigator\n\t\tupdateQueueStateTimer clock.Timer\n\t\tvirtualQueueManager   VirtualQueueManager\n\t\texclusiveAckLevel     persistence.HistoryTaskKey\n\t\talertCh               chan *Alert\n\t\tnewVirtualSliceState  VirtualSliceState\n\n\t\tupdateQueueStateFn func(ctx context.Context)\n\t}\n)\n\nfunc newQueueBase(\n\tshard shard.Context,\n\ttaskProcessor task.Processor,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tmetricsScope metrics.Scope,\n\tcategory persistence.HistoryTaskCategory,\n\ttaskExecutor task.Executor,\n\toptions *Options,\n) *queueBase {\n\ttimeSource := shard.GetTimeSource()\n\tpersistenceQueueState, err := shard.GetQueueState(category)\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to get queue state, probably task category is not supported\", tag.Error(err), tag.Dynamic(\"category\", category))\n\t}\n\tlogger.Info(\"loading queue state\", tag.Dynamic(\"queue-state\", persistenceQueueState))\n\tqueueState := FromPersistenceQueueState(persistenceQueueState)\n\texclusiveAckLevel, _ := getExclusiveAckLevelAndMaxQueueIDFromQueueState(queueState)\n\n\trescheduler := task.NewRescheduler(\n\t\ttaskProcessor,\n\t\ttimeSource,\n\t\tlogger,\n\t\tmetricsScope,\n\t)\n\tvar queueType task.QueueType\n\tif category == persistence.HistoryTaskCategoryTransfer {\n\t\tqueueType = task.QueueTypeTransfer\n\t} else if category == persistence.HistoryTaskCategoryTimer {\n\t\tqueueType = task.QueueTypeTimer\n\t}\n\ttaskInitializer := func(t persistence.Task) task.Task {\n\t\treturn task.NewHistoryTaskV2(\n\t\t\tshard,\n\t\t\tt,\n\t\t\tqueueType,\n\t\t\ttask.InitializeLoggerForTask(shard.GetShardID(), t, logger),\n\t\t\ttaskExecutor,\n\t\t\ttaskProcessor,\n\t\t\trescheduler,\n\t\t\tshard.GetConfig().TaskCriticalRetryCount,\n\t\t)\n\t}\n\tqueueReader := NewQueueReader(\n\t\tshard,\n\t\tcategory,\n\t)\n\tmonitor := NewMonitor(\n\t\tcategory,\n\t\t&MonitorOptions{\n\t\t\tCriticalPendingTaskCount:    options.CriticalPendingTaskCount,\n\t\t\tEnablePendingTaskCountAlert: options.EnablePendingTaskCountAlert,\n\t\t},\n\t)\n\tvirtualQueueManager := NewVirtualQueueManager(\n\t\ttaskProcessor,\n\t\trescheduler,\n\t\ttaskInitializer,\n\t\tqueueReader,\n\t\tlogger,\n\t\tmetricsScope,\n\t\ttimeSource,\n\t\tquotas.NewDynamicRateLimiter(options.MaxPollRPS.AsFloat64()),\n\t\tmonitor,\n\t\t&VirtualQueueManagerOptions{\n\t\t\tRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\tPageSize:                             options.PageSize,\n\t\t\t\tMaxPendingTasksCount:                 options.MaxPendingTasksCount,\n\t\t\t\tPollBackoffInterval:                  options.PollBackoffInterval,\n\t\t\t\tPollBackoffIntervalJitterCoefficient: options.PollBackoffIntervalJitterCoefficient,\n\t\t\t},\n\t\t\tNonRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\tPageSize: options.PageSize,\n\t\t\t\t// non-root queues should not trigger task unloading\n\t\t\t\t// otherwise those virtual queues will keep loading, hit pending task count limit, unload, throttle, load, etc...\n\t\t\t\t// use a limit lower than the critical pending task count instead\n\t\t\t\tMaxPendingTasksCount: func(opts ...dynamicproperties.FilterOption) int {\n\t\t\t\t\treturn int(float64(options.CriticalPendingTaskCount(opts...)) * nonRootQueueMaxPendingTaskCoefficient)\n\t\t\t\t},\n\t\t\t\tPollBackoffInterval:                  options.PollBackoffInterval,\n\t\t\t\tPollBackoffIntervalJitterCoefficient: options.PollBackoffIntervalJitterCoefficient,\n\t\t\t},\n\t\t\tVirtualSliceForceAppendInterval: options.VirtualSliceForceAppendInterval,\n\t\t},\n\t\tqueueState.VirtualQueueStates,\n\t)\n\tmitigator := NewMitigator(\n\t\tvirtualQueueManager,\n\t\tmonitor,\n\t\tlogger,\n\t\tmetricsScope,\n\t\t&MitigatorOptions{\n\t\t\tMaxVirtualQueueCount: options.MaxVirtualQueueCount,\n\t\t},\n\t)\n\tq := &queueBase{\n\t\tshard:               shard,\n\t\ttaskProcessor:       taskProcessor,\n\t\tlogger:              logger,\n\t\tmetricsClient:       metricsClient,\n\t\tmetricsScope:        metricsScope,\n\t\tcategory:            category,\n\t\toptions:             options,\n\t\ttimeSource:          timeSource,\n\t\ttaskInitializer:     taskInitializer,\n\t\trescheduler:         rescheduler,\n\t\tqueueReader:         queueReader,\n\t\tmonitor:             monitor,\n\t\tmitigator:           mitigator,\n\t\texclusiveAckLevel:   exclusiveAckLevel,\n\t\tvirtualQueueManager: virtualQueueManager,\n\t\talertCh:             make(chan *Alert, alertChSize),\n\t\tnewVirtualSliceState: VirtualSliceState{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: queueState.ExclusiveMaxReadLevel,\n\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tq.updateQueueStateFn = q.updateQueueState\n\treturn q\n}\n\nfunc (q *queueBase) Start() {\n\tq.rescheduler.Start()\n\tq.virtualQueueManager.Start()\n\n\tq.updateQueueStateTimer = q.timeSource.NewTimer(backoff.JitDuration(\n\t\tq.options.UpdateAckInterval(),\n\t\tq.options.UpdateAckIntervalJitterCoefficient(),\n\t))\n\n\tq.monitor.Subscribe(q.alertCh)\n}\n\nfunc (q *queueBase) Stop() {\n\tq.monitor.Unsubscribe()\n\tq.updateQueueStateTimer.Stop()\n\tq.virtualQueueManager.Stop()\n\tq.rescheduler.Stop()\n}\n\nfunc (q *queueBase) Category() persistence.HistoryTaskCategory {\n\treturn q.category\n}\n\nfunc (q *queueBase) FailoverDomain(domainIDs map[string]struct{}) {\n\tq.rescheduler.RescheduleDomains(domainIDs)\n}\n\nfunc (q *queueBase) HandleAction(ctx context.Context, clusterName string, action *queue.Action) (*queue.ActionResult, error) {\n\treturn nil, nil\n}\n\nfunc (q *queueBase) LockTaskProcessing() {}\n\nfunc (q *queueBase) UnlockTaskProcessing() {}\n\nfunc (q *queueBase) processNewTasks() bool {\n\tnewExclusiveMaxTaskKey := q.shard.UpdateIfNeededAndGetQueueMaxReadLevel(q.category, q.shard.GetClusterMetadata().GetCurrentClusterName())\n\tif q.category.Type() == persistence.HistoryTaskCategoryTypeImmediate {\n\t\tnewExclusiveMaxTaskKey = persistence.NewImmediateTaskKey(newExclusiveMaxTaskKey.GetTaskID() + 1)\n\t}\n\n\tnewVirtualSliceState, remainingVirtualSliceState, ok := q.newVirtualSliceState.TrySplitByTaskKey(newExclusiveMaxTaskKey)\n\tif !ok {\n\t\treturn false\n\t}\n\tq.newVirtualSliceState = remainingVirtualSliceState\n\n\tnewVirtualSlice := NewVirtualSlice(newVirtualSliceState, q.taskInitializer, q.queueReader, NewPendingTaskTracker(), q.logger)\n\n\tq.logger.Debug(\"processing new tasks\", tag.Dynamic(\"inclusiveMinTaskKey\", newVirtualSliceState.Range.InclusiveMinTaskKey), tag.Dynamic(\"exclusiveMaxTaskKey\", newVirtualSliceState.Range.ExclusiveMaxTaskKey))\n\tq.virtualQueueManager.AddNewVirtualSliceToRootQueue(newVirtualSlice)\n\treturn true\n}\n\nfunc (q *queueBase) updateQueueState(ctx context.Context) {\n\tq.metricsScope.IncCounter(metrics.AckLevelUpdateCounter)\n\tqueueState := &QueueState{\n\t\tVirtualQueueStates:    q.virtualQueueManager.UpdateAndGetState(),\n\t\tExclusiveMaxReadLevel: q.newVirtualSliceState.Range.InclusiveMinTaskKey,\n\t}\n\tnewExclusiveAckLevel, maxQueueID := getExclusiveAckLevelAndMaxQueueIDFromQueueState(queueState)\n\tq.metricsScope.UpdateGauge(metrics.VirtualQueueCountGauge, float64(maxQueueID+1))\n\n\t// for backward compatibility, we record the timer metrics in shard info scope\n\tpendingTaskCount := q.monitor.GetTotalPendingTaskCount()\n\tif q.category == persistence.HistoryTaskCategoryTransfer {\n\t\tq.metricsClient.RecordTimer(metrics.ShardInfoScope, metrics.ShardInfoTransferActivePendingTasksTimer, time.Duration(pendingTaskCount))\n\t} else if q.category == persistence.HistoryTaskCategoryTimer {\n\t\tq.metricsClient.RecordTimer(metrics.ShardInfoScope, metrics.ShardInfoTimerActivePendingTasksTimer, time.Duration(pendingTaskCount))\n\t}\n\n\t// we emit the metrics in the queue scope and experiment with gauge metrics\n\t// TODO: review the metrics and remove this comment or change the metric from gauge to histogram\n\tq.metricsScope.UpdateGauge(metrics.PendingTaskGauge, float64(pendingTaskCount))\n\n\tq.logger.Debug(\"complete history tasks\", tag.Dynamic(\"oldExclusiveAckLevel\", q.exclusiveAckLevel), tag.Dynamic(\"newExclusiveAckLevel\", newExclusiveAckLevel))\n\tif newExclusiveAckLevel.Compare(q.exclusiveAckLevel) > 0 {\n\t\tq.metricsScope.IncCounter(metrics.TaskBatchCompleteCounter)\n\t\tinclusiveMinTaskKey := q.exclusiveAckLevel\n\t\texclusiveMaxTaskKey := newExclusiveAckLevel\n\t\tif q.category.Type() == persistence.HistoryTaskCategoryTypeScheduled {\n\t\t\tinclusiveMinTaskKey = persistence.NewHistoryTaskKey(inclusiveMinTaskKey.GetScheduledTime(), 0)\n\t\t\texclusiveMaxTaskKey = persistence.NewHistoryTaskKey(exclusiveMaxTaskKey.GetScheduledTime(), 0)\n\t\t}\n\t\tfor {\n\t\t\tpageSize := q.options.DeleteBatchSize()\n\t\t\tresp, err := q.shard.GetExecutionManager().RangeCompleteHistoryTask(ctx, &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        q.category,\n\t\t\t\tInclusiveMinTaskKey: inclusiveMinTaskKey,\n\t\t\t\tExclusiveMaxTaskKey: exclusiveMaxTaskKey,\n\t\t\t\tPageSize:            pageSize,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tq.logger.Error(\"Failed to range complete history tasks\", tag.Error(err))\n\t\t\t\tq.updateQueueStateTimer.Reset(backoff.JitDuration(\n\t\t\t\t\tq.options.UpdateAckInterval(),\n\t\t\t\t\tq.options.UpdateAckIntervalJitterCoefficient(),\n\t\t\t\t))\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, pageSize) {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tq.exclusiveAckLevel = newExclusiveAckLevel\n\t}\n\n\t// even though the ack level is not updated, we still need to update the queue state\n\tpersistenceQueueState := ToPersistenceQueueState(queueState)\n\tq.logger.Debug(\"store queue state\", tag.Dynamic(\"queue-state\", persistenceQueueState), tag.PendingTaskCount(pendingTaskCount))\n\terr := q.shard.UpdateQueueState(q.category, persistenceQueueState)\n\tif err != nil {\n\t\tq.logger.Error(\"Failed to update queue state\", tag.Error(err))\n\t\tq.metricsScope.IncCounter(metrics.AckLevelUpdateFailedCounter)\n\t}\n\n\tq.updateQueueStateTimer.Reset(backoff.JitDuration(\n\t\tq.options.UpdateAckInterval(),\n\t\tq.options.UpdateAckIntervalJitterCoefficient(),\n\t))\n}\n\nfunc (q *queueBase) handleAlert(ctx context.Context, alert *Alert) {\n\tif alert == nil {\n\t\treturn\n\t}\n\n\tq.mitigator.Mitigate(*alert)\n\tq.updateQueueStateFn(ctx)\n}\n\nfunc getExclusiveAckLevelAndMaxQueueIDFromQueueState(state *QueueState) (persistence.HistoryTaskKey, int64) {\n\tmaxQueueID := int64(0)\n\tnewExclusiveAckLevel := state.ExclusiveMaxReadLevel\n\tfor queueID, virtualQueueState := range state.VirtualQueueStates {\n\t\tif len(virtualQueueState) != 0 {\n\t\t\tnewExclusiveAckLevel = persistence.MinHistoryTaskKey(newExclusiveAckLevel, virtualQueueState[0].Range.InclusiveMinTaskKey)\n\t\t}\n\t\tmaxQueueID = max(maxQueueID, queueID)\n\t}\n\treturn newExclusiveAckLevel, maxQueueID\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_base_test.go",
    "content": "package queuev2\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestQueueBase_ProcessNewTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname                         string\n\t\tcategory                     persistence.HistoryTaskCategory\n\t\tinitialVirtualSlice          VirtualSliceState\n\t\texpectedVirtualSlices        []VirtualSlice\n\t\texpectedNewVirtualSliceState VirtualSliceState\n\t\texpectError                  bool\n\t\tsetupMocks                   func(*gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager)\n\t}{\n\t\t{\n\t\t\tname:     \"Successfully process new tasks for immediate category\",\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialVirtualSlice: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedNewVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(201),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\t\t\t\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\t\t\t\tmockShard.EXPECT().UpdateIfNeededAndGetQueueMaxReadLevel(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tcluster.TestCurrentClusterName,\n\t\t\t\t).Return(persistence.NewImmediateTaskKey(200))\n\t\t\t\tmockVirtualQueueManager.EXPECT().AddNewVirtualSliceToRootQueue(gomock.Any()).DoAndReturn(func(s VirtualSlice) {\n\t\t\t\t\tassert.Equal(t, s.GetState().Range.InclusiveMinTaskKey, persistence.NewImmediateTaskKey(100))\n\t\t\t\t\tassert.Equal(t, s.GetState().Range.ExclusiveMaxTaskKey, persistence.NewImmediateTaskKey(201))\n\t\t\t\t})\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"Successfully process new tasks for scheduled category\",\n\t\t\tcategory: persistence.HistoryTaskCategoryTimer,\n\t\t\tinitialVirtualSlice: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(100, 0), 100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedNewVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(201, 0), 201),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\t\t\t\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\t\t\t\tmockShard.EXPECT().UpdateIfNeededAndGetQueueMaxReadLevel(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTimer,\n\t\t\t\t\tcluster.TestCurrentClusterName,\n\t\t\t\t).Return(persistence.NewHistoryTaskKey(time.Unix(201, 0), 201))\n\t\t\t\tmockVirtualQueueManager.EXPECT().AddNewVirtualSliceToRootQueue(gomock.Any()).DoAndReturn(func(s VirtualSlice) {\n\t\t\t\t\tassert.Equal(t, s.GetState().Range.InclusiveMinTaskKey, persistence.NewHistoryTaskKey(time.Unix(100, 0), 100))\n\t\t\t\t\tassert.Equal(t, s.GetState().Range.ExclusiveMaxTaskKey, persistence.NewHistoryTaskKey(time.Unix(201, 0), 201))\n\t\t\t\t})\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"No new tasks to process\",\n\t\t\tcategory: persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialVirtualSlice: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedNewVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\n\t\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\t\t\t\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\t\t\t\tmockShard.EXPECT().UpdateIfNeededAndGetQueueMaxReadLevel(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tcluster.TestCurrentClusterName,\n\t\t\t\t).Return(persistence.NewImmediateTaskKey(99))\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Setup\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager := tt.setupMocks(ctrl)\n\n\t\t\tqueueBase := &queueBase{\n\t\t\t\tshard:                mockShard,\n\t\t\t\ttaskProcessor:        mockTaskProcessor,\n\t\t\t\tmetricsClient:        metrics.NoopClient,\n\t\t\t\tmetricsScope:         metrics.NoopScope,\n\t\t\t\tlogger:               testlogger.New(t),\n\t\t\t\tcategory:             tt.category,\n\t\t\t\ttimeSource:           mockTimeSource,\n\t\t\t\tvirtualQueueManager:  mockVirtualQueueManager,\n\t\t\t\tnewVirtualSliceState: tt.initialVirtualSlice,\n\t\t\t}\n\n\t\t\t// Execute\n\t\t\tqueueBase.processNewTasks()\n\n\t\t\t// Verify\n\t\t\tif !tt.expectError {\n\t\t\t\tassert.Equal(t, queueBase.newVirtualSliceState, tt.expectedNewVirtualSliceState)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestQueueBase_UpdateQueueState(t *testing.T) {\n\ttests := []struct {\n\t\tname                      string\n\t\tcategory                  persistence.HistoryTaskCategory\n\t\tinitialExclusiveAckLevel  persistence.HistoryTaskKey\n\t\tinitialVirtualSliceState  VirtualSliceState\n\t\texpectedExclusiveAckLevel persistence.HistoryTaskKey\n\t\texpectError               bool\n\t\tsetupMocks                func(*gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager, *MockMonitor, *MockMitigator)\n\t}{\n\t\t{\n\t\t\tname:                     \"Successfully update queue state with new ack level\",\n\t\t\tcategory:                 persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialExclusiveAckLevel: persistence.NewImmediateTaskKey(100),\n\t\t\tinitialVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1000),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedExclusiveAckLevel: persistence.NewImmediateTaskKey(200),\n\t\t\texpectError:               false,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager, *MockMonitor, *MockMitigator) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\t\tmockMitigator := NewMockMitigator(ctrl)\n\n\t\t\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(100).Times(1)\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).AnyTimes()\n\t\t\t\tmockExecutionManager.EXPECT().RangeCompleteHistoryTask(gomock.Any(), &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\tPageSize:            100,\n\t\t\t\t}).Return(&persistence.RangeCompleteHistoryTaskResponse{\n\t\t\t\t\tTasksCompleted: 10,\n\t\t\t\t}, nil)\n\t\t\t\tmockShard.EXPECT().UpdateQueueState(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil)\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().UpdateAndGetState().Return(map[int64][]VirtualSliceState{\n\t\t\t\t\t1: {\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager, mockMonitor, mockMitigator\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                     \"Failed to range complete history tasks\",\n\t\t\tcategory:                 persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialExclusiveAckLevel: persistence.NewImmediateTaskKey(100),\n\t\t\tinitialVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1000),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedExclusiveAckLevel: persistence.NewImmediateTaskKey(100),\n\t\t\texpectError:               true,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager, *MockMonitor, *MockMitigator) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\t\tmockMitigator := NewMockMitigator(ctrl)\n\n\t\t\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(100).Times(1)\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).AnyTimes()\n\t\t\t\tmockExecutionManager.EXPECT().RangeCompleteHistoryTask(gomock.Any(), &persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\tPageSize:            100,\n\t\t\t\t}).Return(nil, assert.AnError)\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().UpdateAndGetState().Return(map[int64][]VirtualSliceState{\n\t\t\t\t\t1: {\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager, mockMonitor, mockMitigator\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                     \"Successfully update queue state with no new ack level\",\n\t\t\tcategory:                 persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialExclusiveAckLevel: persistence.NewImmediateTaskKey(100),\n\t\t\tinitialVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1000),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedExclusiveAckLevel: persistence.NewImmediateTaskKey(100),\n\t\t\texpectError:               false,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager, *MockMonitor, *MockMitigator) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\t\tmockMitigator := NewMockMitigator(ctrl)\n\n\t\t\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(100).Times(1)\n\t\t\t\tmockShard.EXPECT().UpdateQueueState(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(nil)\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().UpdateAndGetState().Return(map[int64][]VirtualSliceState{\n\t\t\t\t\t1: {\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager, mockMonitor, mockMitigator\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                     \"Failed to update queue state\",\n\t\t\tcategory:                 persistence.HistoryTaskCategoryTransfer,\n\t\t\tinitialExclusiveAckLevel: persistence.NewImmediateTaskKey(300),\n\t\t\tinitialVirtualSliceState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1000),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedExclusiveAckLevel: persistence.NewImmediateTaskKey(200),\n\t\t\texpectError:               true,\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (*shard.MockContext, *task.MockProcessor, clock.TimeSource, *MockVirtualQueueManager, *MockMonitor, *MockMitigator) {\n\t\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\t\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\t\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\t\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\t\t\t\tmockMonitor := NewMockMonitor(ctrl)\n\t\t\t\tmockMitigator := NewMockMitigator(ctrl)\n\n\t\t\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(100).Times(1)\n\t\t\t\tmockShard.EXPECT().UpdateQueueState(\n\t\t\t\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(assert.AnError)\n\n\t\t\t\tmockVirtualQueueManager.EXPECT().UpdateAndGetState().Return(map[int64][]VirtualSliceState{\n\t\t\t\t\t1: {\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(200),\n\t\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\treturn mockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager, mockMonitor, mockMitigator\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Setup\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockShard, mockTaskProcessor, mockTimeSource, mockVirtualQueueManager, mockMonitor, mockMitigator := tt.setupMocks(ctrl)\n\n\t\t\tqueueBase := &queueBase{\n\t\t\t\tshard:                 mockShard,\n\t\t\t\ttaskProcessor:         mockTaskProcessor,\n\t\t\t\tmetricsClient:         metrics.NoopClient,\n\t\t\t\tmetricsScope:          metrics.NoopScope,\n\t\t\t\tlogger:                testlogger.New(t),\n\t\t\t\tcategory:              tt.category,\n\t\t\t\ttimeSource:            mockTimeSource,\n\t\t\t\tmonitor:               mockMonitor,\n\t\t\t\tmitigator:             mockMitigator,\n\t\t\t\tvirtualQueueManager:   mockVirtualQueueManager,\n\t\t\t\texclusiveAckLevel:     tt.initialExclusiveAckLevel,\n\t\t\t\tnewVirtualSliceState:  tt.initialVirtualSliceState,\n\t\t\t\tupdateQueueStateTimer: mockTimeSource.NewTimer(time.Second * 10),\n\t\t\t\toptions: &Options{\n\t\t\t\t\tDeleteBatchSize:                    dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\tUpdateAckInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\t\tUpdateAckIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.1),\n\t\t\t\t},\n\t\t\t}\n\n\t\t\t// Execute\n\t\t\tqueueBase.updateQueueState(context.Background())\n\n\t\t\t// Verify\n\t\t\tif !tt.expectError {\n\t\t\t\tassert.Equal(t, tt.expectedExclusiveAckLevel, queueBase.exclusiveAckLevel)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestQueueBase_HandleAlert(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockVirtualQueueManager := NewMockVirtualQueueManager(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\tmockMitigator := NewMockMitigator(ctrl)\n\tmockMitigator.EXPECT().Mitigate(Alert{AlertType: AlertTypeQueuePendingTaskCount})\n\n\tupdateQueueStateCalled := false\n\tqueueBase := &queueBase{\n\t\tshard:               mockShard,\n\t\ttaskProcessor:       mockTaskProcessor,\n\t\tmetricsClient:       metrics.NoopClient,\n\t\tmetricsScope:        metrics.NoopScope,\n\t\tlogger:              testlogger.New(t),\n\t\tcategory:            persistence.HistoryTaskCategoryTransfer,\n\t\ttimeSource:          mockTimeSource,\n\t\tmonitor:             mockMonitor,\n\t\tmitigator:           mockMitigator,\n\t\tvirtualQueueManager: mockVirtualQueueManager,\n\t\texclusiveAckLevel:   persistence.NewImmediateTaskKey(100),\n\t\tnewVirtualSliceState: VirtualSliceState{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1000),\n\t\t\t\tExclusiveMaxTaskKey: persistence.MaximumHistoryTaskKey,\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t\tupdateQueueStateTimer: mockTimeSource.NewTimer(time.Second * 10),\n\t\toptions: &Options{\n\t\t\tDeleteBatchSize:                    dynamicproperties.GetIntPropertyFn(100),\n\t\t\tUpdateAckInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tUpdateAckIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.1),\n\t\t},\n\t\tupdateQueueStateFn: func(ctx context.Context) {\n\t\t\tupdateQueueStateCalled = true\n\t\t},\n\t}\n\n\tqueueBase.handleAlert(context.Background(), &Alert{AlertType: AlertTypeQueuePendingTaskCount})\n\n\tassert.True(t, updateQueueStateCalled)\n}\n\nfunc TestNewQueueBase(t *testing.T) {\n\tqueueState := &types.QueueState{\n\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\tTaskID: 400,\n\t\t},\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\trootQueueID: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 100,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 200,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 300,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t1: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 300,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 400,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\tmockShard.EXPECT().GetQueueState(persistence.HistoryTaskCategoryTransfer).Return(queueState, nil)\n\n\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource)\n\n\tqueueBase := newQueueBase(\n\t\tmockShard,\n\t\tmockTaskProcessor,\n\t\ttestlogger.New(t),\n\t\tmetrics.NoopClient,\n\t\tmetrics.NoopScope,\n\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\tnil,\n\t\t&Options{\n\t\t\tDeleteBatchSize:    dynamicproperties.GetIntPropertyFn(100),\n\t\t\tRedispatchInterval: dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tMaxPollRPS:         dynamicproperties.GetIntPropertyFn(100),\n\t\t},\n\t)\n\n\tassert.Equal(t, persistence.NewImmediateTaskKey(400), queueBase.newVirtualSliceState.Range.InclusiveMinTaskKey)\n\tvirtualQueues := queueBase.virtualQueueManager.VirtualQueues()\n\tstates := make(map[int64][]VirtualSliceState)\n\tfor queueID, virtualQueue := range virtualQueues {\n\t\tstates[queueID] = virtualQueue.GetState()\n\t}\n\tassert.Equal(t, map[int64][]VirtualSliceState{\n\t\trootQueueID: {\n\t\t\t{\n\t\t\t\tRange:     Range{InclusiveMinTaskKey: persistence.NewImmediateTaskKey(100), ExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(200)},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tRange:     Range{InclusiveMinTaskKey: persistence.NewImmediateTaskKey(200), ExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(300)},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t\t1: {\n\t\t\t{\n\t\t\t\tRange:     Range{InclusiveMinTaskKey: persistence.NewImmediateTaskKey(300), ExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(400)},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t}, states)\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_immediate.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\timmediateQueue struct {\n\t\tbase     *queueBase\n\t\tnotifyCh chan struct{}\n\n\t\tstatus       int32\n\t\tshutdownWG   sync.WaitGroup\n\t\tctx          context.Context\n\t\tcancel       func()\n\t\tpollTimer    clock.Timer\n\t\tlastPollTime time.Time\n\t}\n)\n\nfunc NewImmediateQueue(\n\tshard shard.Context,\n\tcategory persistence.HistoryTaskCategory,\n\ttaskProcessor task.Processor,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tmetricsScope metrics.Scope,\n\toptions *Options,\n) Queue {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &immediateQueue{\n\t\tbase: newQueueBase(\n\t\t\tshard,\n\t\t\ttaskProcessor,\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\tmetricsScope,\n\t\t\tcategory,\n\t\t\ttaskExecutor,\n\t\t\toptions,\n\t\t),\n\t\tnotifyCh: make(chan struct{}, 1),\n\t\tctx:      ctx,\n\t\tcancel:   cancel,\n\t}\n}\n\nfunc (q *immediateQueue) Start() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tq.base.logger.Info(\"History queue state changed\", tag.LifeCycleStarting)\n\tdefer q.base.logger.Info(\"History queue state changed\", tag.LifeCycleStarted)\n\n\tq.base.Start()\n\n\tq.shutdownWG.Add(1)\n\tgo q.processEventLoop()\n\n\tq.notify()\n}\n\nfunc (q *immediateQueue) Stop() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tq.base.logger.Info(\"History queue state changed\", tag.LifeCycleStopping)\n\tdefer q.base.logger.Info(\"History queue state changed\", tag.LifeCycleStopped)\n\n\tq.cancel()\n\tq.shutdownWG.Wait()\n\n\tq.base.Stop()\n}\n\nfunc (q *immediateQueue) Category() persistence.HistoryTaskCategory {\n\treturn q.base.Category()\n}\n\nfunc (q *immediateQueue) FailoverDomain(domainIDs map[string]struct{}) {\n\tq.base.FailoverDomain(domainIDs)\n}\n\nfunc (q *immediateQueue) HandleAction(ctx context.Context, clusterName string, action *queue.Action) (*queue.ActionResult, error) {\n\treturn q.base.HandleAction(ctx, clusterName, action)\n}\n\nfunc (q *immediateQueue) UnlockTaskProcessing() {\n\tq.base.UnlockTaskProcessing()\n}\n\nfunc (q *immediateQueue) LockTaskProcessing() {\n\tq.base.LockTaskProcessing()\n}\n\nfunc (q *immediateQueue) NotifyNewTask(clusterName string, info *hcommon.NotifyTaskInfo) {\n\tnumTasks := len(info.Tasks)\n\tif numTasks == 0 {\n\t\treturn\n\t}\n\n\tq.notify()\n\tq.base.metricsScope.AddCounter(metrics.NewHistoryTaskCounter, int64(numTasks))\n}\n\nfunc (q *immediateQueue) notify() {\n\tselect {\n\tcase q.notifyCh <- struct{}{}:\n\tdefault:\n\t}\n}\n\nfunc (q *immediateQueue) processPollTimer() {\n\t// NOTE: ideally when new tasks are written to the queue, a notification should be sent to the notifyCh,\n\t// thus this periodic poll is not needed, but we keep it for now to provide a fallback mechanism in case\n\t// there is a bug in the notification mechanism\n\tif q.lastPollTime.Add(q.base.options.PollBackoffInterval()).Before(q.base.timeSource.Now()) {\n\t\tnewTasks := q.base.processNewTasks()\n\t\tif newTasks {\n\t\t\t// TODO: consider changing it to warn level\n\t\t\tq.base.logger.Info(\"processing new tasks because poll timer fired\")\n\t\t}\n\t\tq.lastPollTime = q.base.timeSource.Now()\n\t}\n\n\tq.pollTimer.Reset(backoff.JitDuration(\n\t\tq.base.options.MaxPollInterval(),\n\t\tq.base.options.MaxPollIntervalJitterCoefficient(),\n\t))\n}\n\nfunc (q *immediateQueue) processEventLoop() {\n\tdefer q.shutdownWG.Done()\n\n\tq.pollTimer = q.base.timeSource.NewTimer(backoff.JitDuration(\n\t\tq.base.options.MaxPollInterval(),\n\t\tq.base.options.MaxPollIntervalJitterCoefficient(),\n\t))\n\tdefer q.pollTimer.Stop()\n\tfor {\n\t\tselect {\n\t\tcase <-q.notifyCh:\n\t\t\tq.base.processNewTasks()\n\t\t\tq.lastPollTime = q.base.timeSource.Now()\n\t\tcase <-q.pollTimer.Chan():\n\t\t\tq.processPollTimer()\n\t\tcase <-q.base.updateQueueStateTimer.Chan():\n\t\t\tq.base.updateQueueState(q.ctx)\n\t\tcase alert := <-q.base.alertCh:\n\t\t\tq.base.handleAlert(q.ctx, alert)\n\t\tcase <-q.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_immediate_test.go",
    "content": "package queuev2\n\nimport (\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestImmediateQueue_LifeCycle(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\tmockTaskExecutor := task.NewMockExecutor(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsClient := metrics.NoopClient\n\tmockMetricsScope := metrics.NoopScope\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\n\t// Setup mock expectations\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\tmockShard.EXPECT().GetQueueState(persistence.HistoryTaskCategoryTransfer).Return(&types.QueueState{\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\trootQueueID: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\tTaskID: 10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\tTaskID: 10,\n\t\t},\n\t}, nil).AnyTimes()\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).AnyTimes()\n\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, nil).AnyTimes()\n\tmockExecutionManager.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(&persistence.RangeCompleteHistoryTaskResponse{}, nil).AnyTimes()\n\tmockShard.EXPECT().UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTransfer, cluster.TestCurrentClusterName).Return(persistence.NewImmediateTaskKey(10)).AnyTimes()\n\tmockShard.EXPECT().UpdateQueueState(persistence.HistoryTaskCategoryTransfer, gomock.Any()).Return(nil).AnyTimes()\n\n\toptions := &Options{\n\t\tDeleteBatchSize:                      dynamicproperties.GetIntPropertyFn(100),\n\t\tRedispatchInterval:                   dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tPageSize:                             dynamicproperties.GetIntPropertyFn(100),\n\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\tMaxPollInterval:                      dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tMaxPollIntervalJitterCoefficient:     dynamicproperties.GetFloatPropertyFn(0.1),\n\t\tUpdateAckInterval:                    dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tUpdateAckIntervalJitterCoefficient:   dynamicproperties.GetFloatPropertyFn(0.1),\n\t\tMaxPollRPS:                           dynamicproperties.GetIntPropertyFn(100),\n\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\tCriticalPendingTaskCount:             dynamicproperties.GetIntPropertyFn(90),\n\t\tVirtualSliceForceAppendInterval:      dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tEnablePendingTaskCountAlert:          func() bool { return true },\n\t\tMaxVirtualQueueCount:                 dynamicproperties.GetIntPropertyFn(2),\n\t}\n\n\tqueue := NewImmediateQueue(\n\t\tmockShard,\n\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\tmockTaskProcessor,\n\t\tmockTaskExecutor,\n\t\tmockLogger,\n\t\tmockMetricsClient,\n\t\tmockMetricsScope,\n\t\toptions,\n\t).(*immediateQueue)\n\n\t// Test Start\n\tqueue.Start()\n\tassert.Equal(t, common.DaemonStatusStarted, atomic.LoadInt32(&queue.status))\n\n\t// Test NotifyNewTask\n\tqueue.NotifyNewTask(\"test-cluster\", &hcommon.NotifyTaskInfo{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.DecisionTask{},\n\t\t},\n\t})\n\n\t// Advance time to trigger poll\n\tmockTimeSource.Advance(options.MaxPollInterval() * 2)\n\n\t// Advance time to trigger update ack\n\tmockTimeSource.Advance(options.UpdateAckInterval() * 2)\n\n\t// Test Stop\n\tqueue.Stop()\n\tassert.Equal(t, common.DaemonStatusStopped, atomic.LoadInt32(&queue.status))\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_reader.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination queue_reader_mock.go github.com/uber/cadence/service/history/queuev2 QueueReader\npackage queuev2\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tQueueReader interface {\n\t\tGetTask(context.Context, *GetTaskRequest) (*GetTaskResponse, error)\n\t}\n\n\tGetTaskRequest struct {\n\t\tProgress  *GetTaskProgress\n\t\tPredicate Predicate\n\t\tPageSize  int\n\t}\n\n\t// GetTaskProgress contains the range of the slice to read, the next page token, and the next task key\n\tGetTaskProgress struct {\n\t\tRange\n\t\tNextPageToken []byte\n\t\tNextTaskKey   persistence.HistoryTaskKey\n\t}\n\n\tGetTaskResponse struct {\n\t\tTasks    []persistence.Task\n\t\tProgress *GetTaskProgress\n\t}\n\n\tsimpleQueueReader struct {\n\t\tshard    shard.Context\n\t\tcategory persistence.HistoryTaskCategory\n\t}\n)\n\nfunc NewQueueReader(\n\tshard shard.Context,\n\tcategory persistence.HistoryTaskCategory,\n) QueueReader {\n\treturn &simpleQueueReader{\n\t\tshard:    shard,\n\t\tcategory: category,\n\t}\n}\n\nfunc (r *simpleQueueReader) GetTask(ctx context.Context, req *GetTaskRequest) (*GetTaskResponse, error) {\n\tresp, err := r.shard.GetExecutionManager().GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        r.category,\n\t\tInclusiveMinTaskKey: req.Progress.InclusiveMinTaskKey,\n\t\tExclusiveMaxTaskKey: req.Progress.ExclusiveMaxTaskKey,\n\t\tPageSize:            req.PageSize,\n\t\tNextPageToken:       req.Progress.NextPageToken,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnextTaskKey := req.Progress.ExclusiveMaxTaskKey\n\ttasks := make([]persistence.Task, 0, len(resp.Tasks))\n\tfor _, task := range resp.Tasks {\n\t\t// filter out tasks that don't match the predicate\n\t\tif req.Predicate.Check(task) {\n\t\t\ttasks = append(tasks, task)\n\t\t}\n\t}\n\t// If there are more tasks to read, set the next task key to the next task key of the last task\n\tif len(resp.NextPageToken) != 0 && len(resp.Tasks) > 0 {\n\t\tnextTaskKey = resp.Tasks[len(resp.Tasks)-1].GetTaskKey().Next()\n\t}\n\n\treturn &GetTaskResponse{\n\t\tTasks: tasks,\n\t\tProgress: &GetTaskProgress{\n\t\t\tRange:         req.Progress.Range,\n\t\t\tNextPageToken: resp.NextPageToken,\n\t\t\tNextTaskKey:   nextTaskKey,\n\t\t},\n\t}, nil\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_reader_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: QueueReader)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination queue_reader_mock.go github.com/uber/cadence/service/history/queuev2 QueueReader\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockQueueReader is a mock of QueueReader interface.\ntype MockQueueReader struct {\n\tctrl     *gomock.Controller\n\trecorder *MockQueueReaderMockRecorder\n\tisgomock struct{}\n}\n\n// MockQueueReaderMockRecorder is the mock recorder for MockQueueReader.\ntype MockQueueReaderMockRecorder struct {\n\tmock *MockQueueReader\n}\n\n// NewMockQueueReader creates a new mock instance.\nfunc NewMockQueueReader(ctrl *gomock.Controller) *MockQueueReader {\n\tmock := &MockQueueReader{ctrl: ctrl}\n\tmock.recorder = &MockQueueReaderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockQueueReader) EXPECT() *MockQueueReaderMockRecorder {\n\treturn m.recorder\n}\n\n// GetTask mocks base method.\nfunc (m *MockQueueReader) GetTask(arg0 context.Context, arg1 *GetTaskRequest) (*GetTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTask\", arg0, arg1)\n\tret0, _ := ret[0].(*GetTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTask indicates an expected call of GetTask.\nfunc (mr *MockQueueReaderMockRecorder) GetTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTask\", reflect.TypeOf((*MockQueueReader)(nil).GetTask), arg0, arg1)\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_reader_test.go",
    "content": "package queuev2\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestQueueReader_GetTask(t *testing.T) {\n\thistoryTasks := []persistence.Task{\n\t\t&persistence.DecisionTimeoutTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\tRunID:      \"test-run\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVersion:             1,\n\t\t\t\tTaskID:              1,\n\t\t\t\tVisibilityTimestamp: time.Now().UTC(),\n\t\t\t},\n\t\t},\n\t\t&persistence.ActivityTimeoutTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\tWorkflowID: \"test-workflow\",\n\t\t\t\tRunID:      \"test-run\",\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVersion:             1,\n\t\t\t\tTaskID:              2,\n\t\t\t\tVisibilityTimestamp: time.Now().UTC(),\n\t\t\t},\n\t\t},\n\t}\n\ttests := []struct {\n\t\tname            string\n\t\trequest         *GetTaskRequest\n\t\tmockSetup       func(*shard.MockContext, *persistence.MockExecutionManager, *MockPredicate)\n\t\texpectedTasks   []persistence.Task\n\t\texpectedToken   []byte\n\t\texpectedNextKey persistence.HistoryTaskKey\n\t\texpectedError   error\n\t}{\n\t\t{\n\t\t\tname: \"successful task retrieval\",\n\t\t\trequest: &GetTaskRequest{\n\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t},\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockExec *persistence.MockExecutionManager, mockPredicate *MockPredicate) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExec)\n\t\t\t\tmockExec.EXPECT().GetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\tPageSize:            10,\n\t\t\t\t\tNextPageToken:       nil,\n\t\t\t\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\tTasks:         historyTasks,\n\t\t\t\t\tNextPageToken: []byte(\"next-page\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockPredicate.EXPECT().Check(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\texpectedTasks:   historyTasks,\n\t\t\texpectedToken:   []byte(\"next-page\"),\n\t\t\texpectedNextKey: historyTasks[1].GetTaskKey().Next(),\n\t\t},\n\t\t{\n\t\t\tname: \"successful task retrieval with empty next page token\",\n\t\t\trequest: &GetTaskRequest{\n\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t},\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockExec *persistence.MockExecutionManager, mockPredicate *MockPredicate) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExec)\n\t\t\t\tmockExec.EXPECT().GetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\tPageSize:            10,\n\t\t\t\t\tNextPageToken:       nil,\n\t\t\t\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\tTasks:         historyTasks,\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}, nil)\n\t\t\t\tmockPredicate.EXPECT().Check(gomock.Any()).Return(true).AnyTimes()\n\t\t\t},\n\t\t\texpectedTasks:   historyTasks,\n\t\t\texpectedToken:   nil,\n\t\t\texpectedNextKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t},\n\t\t{\n\t\t\tname: \"task filtering with predicate\",\n\t\t\trequest: &GetTaskRequest{\n\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t},\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockExec *persistence.MockExecutionManager, mockPredicate *MockPredicate) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExec)\n\t\t\t\tmockExec.EXPECT().GetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\tPageSize:            10,\n\t\t\t\t\tNextPageToken:       nil,\n\t\t\t\t}).Return(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\tTasks:         historyTasks,\n\t\t\t\t\tNextPageToken: []byte(\"next-page\"),\n\t\t\t\t}, nil)\n\t\t\t\tmockPredicate.EXPECT().Check(historyTasks[0]).Return(true)\n\t\t\t\tmockPredicate.EXPECT().Check(historyTasks[1]).Return(false)\n\t\t\t},\n\t\t\texpectedTasks:   []persistence.Task{historyTasks[0]},\n\t\t\texpectedToken:   []byte(\"next-page\"),\n\t\t\texpectedNextKey: historyTasks[1].GetTaskKey().Next(),\n\t\t},\n\t\t{\n\t\t\tname: \"error from execution manager\",\n\t\t\trequest: &GetTaskRequest{\n\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t},\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockExec *persistence.MockExecutionManager, mockPredicate *MockPredicate) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExec)\n\t\t\t\tmockExec.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedError: assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname: \"empty task list\",\n\t\t\trequest: &GetTaskRequest{\n\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\t\t\t},\n\t\t\t\tPageSize: 10,\n\t\t\t},\n\t\t\tmockSetup: func(mockShard *shard.MockContext, mockExec *persistence.MockExecutionManager, mockPredicate *MockPredicate) {\n\t\t\t\tmockShard.EXPECT().GetExecutionManager().Return(mockExec)\n\t\t\t\tmockExec.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\tTasks:         []persistence.Task{},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedTasks:   []persistence.Task{},\n\t\t\texpectedToken:   nil,\n\t\t\texpectedNextKey: persistence.NewHistoryTaskKey(time.Unix(0, math.MaxInt64), 10),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tmockShard := shard.NewMockContext(controller)\n\t\t\tmockExec := persistence.NewMockExecutionManager(controller)\n\t\t\tmockPredicate := NewMockPredicate(controller)\n\t\t\ttt.mockSetup(mockShard, mockExec, mockPredicate)\n\n\t\t\treader := NewQueueReader(mockShard, persistence.HistoryTaskCategoryTimer)\n\t\t\ttt.request.Predicate = mockPredicate\n\t\t\tresp, err := reader.GetTask(context.Background(), tt.request)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tt.expectedTasks, resp.Tasks)\n\t\t\tassert.Equal(t, tt.expectedToken, resp.Progress.NextPageToken)\n\t\t\tassert.Equal(t, tt.expectedNextKey, resp.Progress.NextTaskKey)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_scheduled.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tscheduledQueue struct {\n\t\tbase *queueBase\n\n\t\ttimerGate    clock.TimerGate\n\t\tnewTimerCh   chan struct{}\n\t\tnextTimeLock sync.RWMutex\n\t\tnextTime     time.Time\n\n\t\tstatus     int32\n\t\tshutdownWG sync.WaitGroup\n\t\tctx        context.Context\n\t\tcancel     func()\n\t}\n)\n\nfunc NewScheduledQueue(\n\tshard shard.Context,\n\tcategory persistence.HistoryTaskCategory,\n\ttaskProcessor task.Processor,\n\ttaskExecutor task.Executor,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tmetricsScope metrics.Scope,\n\toptions *Options,\n) Queue {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &scheduledQueue{\n\t\tbase: newQueueBase(\n\t\t\tshard,\n\t\t\ttaskProcessor,\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\tmetricsScope,\n\t\t\tcategory,\n\t\t\ttaskExecutor,\n\t\t\toptions,\n\t\t),\n\t\ttimerGate:  clock.NewTimerGate(shard.GetTimeSource()),\n\t\tnewTimerCh: make(chan struct{}, 1),\n\t\tctx:        ctx,\n\t\tcancel:     cancel,\n\t\tstatus:     common.DaemonStatusInitialized,\n\t}\n}\n\nfunc (q *scheduledQueue) Start() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tq.base.logger.Info(\"History queue state changed\", tag.LifeCycleStarting)\n\tdefer q.base.logger.Info(\"History queue state changed\", tag.LifeCycleStarted)\n\n\tq.base.Start()\n\n\tq.shutdownWG.Add(1)\n\tgo q.processEventLoop()\n\n\tq.notify(time.Time{})\n}\n\nfunc (q *scheduledQueue) Stop() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tq.base.logger.Info(\"History queue state changed\", tag.LifeCycleStopping)\n\tdefer q.base.logger.Info(\"History queue state changed\", tag.LifeCycleStopped)\n\n\tq.cancel()\n\tq.timerGate.Stop()\n\tq.shutdownWG.Wait()\n\tq.base.Stop()\n}\n\nfunc (q *scheduledQueue) Category() persistence.HistoryTaskCategory {\n\treturn q.base.Category()\n}\n\nfunc (q *scheduledQueue) FailoverDomain(domainIDs map[string]struct{}) {\n\tq.base.FailoverDomain(domainIDs)\n}\n\nfunc (q *scheduledQueue) HandleAction(ctx context.Context, clusterName string, action *queue.Action) (*queue.ActionResult, error) {\n\treturn q.base.HandleAction(ctx, clusterName, action)\n}\n\nfunc (q *scheduledQueue) LockTaskProcessing() {\n\tq.base.LockTaskProcessing()\n}\n\nfunc (q *scheduledQueue) UnlockTaskProcessing() {\n\tq.base.UnlockTaskProcessing()\n}\n\nfunc (q *scheduledQueue) NotifyNewTask(clusterName string, info *hcommon.NotifyTaskInfo) {\n\tnumTasks := len(info.Tasks)\n\tif numTasks == 0 {\n\t\treturn\n\t}\n\n\tnextTime := info.Tasks[0].GetVisibilityTimestamp()\n\tfor i := 1; i < numTasks; i++ {\n\t\tts := info.Tasks[i].GetVisibilityTimestamp()\n\t\tif ts.Before(nextTime) {\n\t\t\tnextTime = ts\n\t\t}\n\t}\n\n\tq.notify(nextTime)\n\tq.base.metricsScope.AddCounter(metrics.NewHistoryTaskCounter, int64(numTasks))\n}\n\nfunc (q *scheduledQueue) notify(t time.Time) {\n\tq.nextTimeLock.Lock()\n\tdefer q.nextTimeLock.Unlock()\n\n\tif q.nextTime.IsZero() || t.Before(q.nextTime) {\n\t\tq.nextTime = t\n\t\tselect {\n\t\tcase q.newTimerCh <- struct{}{}:\n\t\tdefault:\n\t\t}\n\t}\n}\n\nfunc (q *scheduledQueue) processNextTime() {\n\tq.nextTimeLock.Lock()\n\tnextTime := q.nextTime\n\tq.nextTime = time.Time{}\n\tq.nextTimeLock.Unlock()\n\n\tq.timerGate.Update(nextTime)\n}\n\nfunc (q *scheduledQueue) processEventLoop() {\n\tdefer q.shutdownWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-q.newTimerCh:\n\t\t\tq.processNextTime()\n\t\tcase <-q.timerGate.Chan():\n\t\t\tq.base.processNewTasks()\n\t\t\tq.lookAheadTask()\n\t\tcase <-q.base.updateQueueStateTimer.Chan():\n\t\t\tq.base.updateQueueState(q.ctx)\n\t\tcase alert := <-q.base.alertCh:\n\t\t\tq.base.handleAlert(q.ctx, alert)\n\t\tcase <-q.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (q *scheduledQueue) lookAheadTask() {\n\tlookAheadMinTime := q.base.newVirtualSliceState.Range.InclusiveMinTaskKey.GetScheduledTime()\n\tlookAheadMaxTime := lookAheadMinTime.Add(backoff.JitDuration(\n\t\tq.base.options.MaxPollInterval(),\n\t\tq.base.options.MaxPollIntervalJitterCoefficient(),\n\t))\n\n\tresp, err := q.base.queueReader.GetTask(q.ctx, &GetTaskRequest{\n\t\tProgress: &GetTaskProgress{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(lookAheadMinTime, 0),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(lookAheadMaxTime, 0),\n\t\t\t},\n\t\t\tNextTaskKey: persistence.NewHistoryTaskKey(lookAheadMaxTime, 0),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t\tPageSize:  1,\n\t})\n\tif err != nil {\n\t\tq.timerGate.Update(lookAheadMinTime)\n\t\tq.base.logger.Error(\"Failed to look ahead task\", tag.Error(err))\n\t\treturn\n\t}\n\n\tif len(resp.Tasks) == 0 {\n\t\tq.timerGate.Update(lookAheadMaxTime)\n\t\treturn\n\t}\n\n\tq.timerGate.Update(resp.Tasks[0].GetVisibilityTimestamp())\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_scheduled_test.go",
    "content": "package queuev2\n\nimport (\n\t\"errors\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\thcommon \"github.com/uber/cadence/service/history/common\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestScheduledQueue_LifeCycle(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\n\tmockShard := shard.NewMockContext(ctrl)\n\tmockTaskProcessor := task.NewMockProcessor(ctrl)\n\tmockTaskExecutor := task.NewMockExecutor(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsClient := metrics.NoopClient\n\tmockMetricsScope := metrics.NoopScope\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\n\t// Setup mock expectations\n\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\tmockShard.EXPECT().GetQueueState(persistence.HistoryTaskCategoryTimer).Return(&types.QueueState{\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\trootQueueID: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{\n\t\t\t\t\t\t\t\tScheduledTimeNano: mockTimeSource.Now().Add(-1 * time.Hour).UnixNano(),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{\n\t\t\t\t\t\t\t\tScheduledTimeNano: mockTimeSource.Now().UnixNano(),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tExclusiveMaxReadLevel: &types.TaskKey{\n\t\t\tScheduledTimeNano: mockTimeSource.Now().UnixNano(),\n\t\t},\n\t}, nil).AnyTimes()\n\tmockShard.EXPECT().GetExecutionManager().Return(mockExecutionManager).AnyTimes()\n\tmockExecutionManager.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Return(&persistence.GetHistoryTasksResponse{}, nil).AnyTimes()\n\tmockExecutionManager.EXPECT().RangeCompleteHistoryTask(gomock.Any(), gomock.Any()).Return(&persistence.RangeCompleteHistoryTaskResponse{}, nil).AnyTimes()\n\tmockShard.EXPECT().UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, cluster.TestCurrentClusterName).Return(persistence.NewHistoryTaskKey(time.Now(), 10)).AnyTimes()\n\tmockShard.EXPECT().UpdateQueueState(persistence.HistoryTaskCategoryTimer, gomock.Any()).Return(nil).AnyTimes()\n\n\toptions := &Options{\n\t\tDeleteBatchSize:                      dynamicproperties.GetIntPropertyFn(100),\n\t\tRedispatchInterval:                   dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tPageSize:                             dynamicproperties.GetIntPropertyFn(100),\n\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tMaxPollInterval:                      dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tMaxPollIntervalJitterCoefficient:     dynamicproperties.GetFloatPropertyFn(0.1),\n\t\tUpdateAckInterval:                    dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tUpdateAckIntervalJitterCoefficient:   dynamicproperties.GetFloatPropertyFn(0.1),\n\t\tMaxPollRPS:                           dynamicproperties.GetIntPropertyFn(100),\n\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\tVirtualSliceForceAppendInterval:      dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\tCriticalPendingTaskCount:             dynamicproperties.GetIntPropertyFn(90),\n\t\tEnablePendingTaskCountAlert:          func() bool { return true },\n\t\tMaxVirtualQueueCount:                 dynamicproperties.GetIntPropertyFn(2),\n\t}\n\n\tqueue := NewScheduledQueue(\n\t\tmockShard,\n\t\tpersistence.HistoryTaskCategoryTimer,\n\t\tmockTaskProcessor,\n\t\tmockTaskExecutor,\n\t\tmockLogger,\n\t\tmockMetricsClient,\n\t\tmockMetricsScope,\n\t\toptions,\n\t).(*scheduledQueue)\n\n\t// Test Start\n\tqueue.Start()\n\tassert.Equal(t, common.DaemonStatusStarted, atomic.LoadInt32(&queue.status))\n\n\tqueue.NotifyNewTask(\"test-cluster\", &hcommon.NotifyTaskInfo{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.DecisionTimeoutTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVisibilityTimestamp: time.Time{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t})\n\n\t// Advance time to trigger poll\n\tmockTimeSource.Advance(options.MaxPollInterval() * 2)\n\n\t// Advance time to trigger update ack\n\tmockTimeSource.Advance(options.UpdateAckInterval() * 2)\n\n\t// Test Stop\n\tqueue.Stop()\n\tassert.Equal(t, common.DaemonStatusStopped, atomic.LoadInt32(&queue.status))\n}\n\nfunc TestScheduledQueue_LookAheadTask(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(*gomock.Controller, *MockQueueReader, clock.TimeSource, *clock.MockTimerGate)\n\t\texpectedUpdate time.Time\n\t}{\n\t\t{\n\t\t\tname: \"successful look ahead with future task\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, mockReader *MockQueueReader, mockTimeSource clock.TimeSource, mockTimerGate *clock.MockTimerGate) {\n\t\t\t\tnow := mockTimeSource.Now()\n\t\t\t\tfutureTime := now.Add(time.Hour)\n\t\t\t\tmockReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(now, 0),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t&persistence.DecisionTimeoutTask{\n\t\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\t\tVisibilityTimestamp: futureTime,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockTimerGate.EXPECT().Update(futureTime)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"no tasks found\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, mockReader *MockQueueReader, mockTimeSource clock.TimeSource, mockTimerGate *clock.MockTimerGate) {\n\t\t\t\tnow := mockTimeSource.Now()\n\t\t\t\tmockReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(now, 0),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{},\n\t\t\t\t}, nil)\n\t\t\t\tmockTimerGate.EXPECT().Update(now.Add(time.Second * 10))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"error during look ahead\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller, mockReader *MockQueueReader, mockTimeSource clock.TimeSource, mockTimerGate *clock.MockTimerGate) {\n\t\t\t\tnow := mockTimeSource.Now()\n\t\t\t\tmockReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(now, 0),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Second*10), 0),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(nil, errors.New(\"test error\"))\n\t\t\t\tmockTimerGate.EXPECT().Update(now)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockShard := shard.NewMockContext(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockMetricsScope := metrics.NoopScope\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockTimerGate := clock.NewMockTimerGate(ctrl)\n\n\t\t\t// Setup mock expectations\n\t\t\tmockShard.EXPECT().GetTimeSource().Return(mockTimeSource).AnyTimes()\n\t\t\tmockShard.EXPECT().GetClusterMetadata().Return(cluster.TestActiveClusterMetadata).AnyTimes()\n\n\t\t\toptions := &Options{\n\t\t\t\tDeleteBatchSize:                    dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tRedispatchInterval:                 dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\tPageSize:                           dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tPollBackoffInterval:                dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\tMaxPollInterval:                    dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\tMaxPollIntervalJitterCoefficient:   dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t\t\tUpdateAckInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\tUpdateAckIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.1),\n\t\t\t}\n\n\t\t\tnow := mockTimeSource.Now()\n\t\t\t// Create scheduled queue directly\n\t\t\tqueue := &scheduledQueue{\n\t\t\t\tbase: &queueBase{\n\t\t\t\t\tlogger:       mockLogger,\n\t\t\t\t\tmetricsScope: mockMetricsScope,\n\t\t\t\t\tcategory:     persistence.HistoryTaskCategoryTimer,\n\t\t\t\t\toptions:      options,\n\t\t\t\t\tqueueReader:  mockQueueReader,\n\t\t\t\t\tnewVirtualSliceState: VirtualSliceState{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(now, 0),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(now.Add(time.Hour), 0),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\ttimerGate:  mockTimerGate,\n\t\t\t\tnewTimerCh: make(chan struct{}, 1),\n\t\t\t}\n\n\t\t\t// Setup test-specific mocks and get expected update time\n\t\t\ttt.setupMocks(ctrl, mockQueueReader, mockTimeSource, mockTimerGate)\n\n\t\t\t// Execute lookAheadTask\n\t\t\tqueue.lookAheadTask()\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_state.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport \"github.com/uber/cadence/common/persistence\"\n\ntype QueueState struct {\n\tVirtualQueueStates    map[int64][]VirtualSliceState\n\tExclusiveMaxReadLevel persistence.HistoryTaskKey\n}\n\ntype VirtualSliceState struct {\n\tRange     Range\n\tPredicate Predicate\n}\n\nfunc (s *VirtualSliceState) IsEmpty() bool {\n\treturn s.Range.IsEmpty() || s.Predicate.IsEmpty()\n}\n\nfunc (s *VirtualSliceState) Contains(task persistence.Task) bool {\n\treturn s.Range.Contains(task.GetTaskKey()) && s.Predicate.Check(task)\n}\n\nfunc (s *VirtualSliceState) TrySplitByTaskKey(taskKey persistence.HistoryTaskKey) (VirtualSliceState, VirtualSliceState, bool) {\n\tif !s.Range.CanSplitByTaskKey(taskKey) {\n\t\treturn VirtualSliceState{}, VirtualSliceState{}, false\n\t}\n\n\treturn VirtualSliceState{\n\t\t\tRange:     Range{InclusiveMinTaskKey: s.Range.InclusiveMinTaskKey, ExclusiveMaxTaskKey: taskKey},\n\t\t\tPredicate: s.Predicate,\n\t\t}, VirtualSliceState{\n\t\t\tRange:     Range{InclusiveMinTaskKey: taskKey, ExclusiveMaxTaskKey: s.Range.ExclusiveMaxTaskKey},\n\t\t\tPredicate: s.Predicate,\n\t\t}, true\n}\n\nfunc (s *VirtualSliceState) TrySplitByPredicate(predicate Predicate) (VirtualSliceState, VirtualSliceState, bool) {\n\tif predicate.Equals(&universalPredicate{}) || predicate.Equals(&emptyPredicate{}) || predicate.Equals(s.Predicate) {\n\t\treturn VirtualSliceState{}, VirtualSliceState{}, false\n\t}\n\treturn VirtualSliceState{\n\t\t\tRange:     s.Range,\n\t\t\tPredicate: And(s.Predicate, predicate),\n\t\t}, VirtualSliceState{\n\t\t\tRange:     s.Range,\n\t\t\tPredicate: And(s.Predicate, Not(predicate)),\n\t\t}, true\n}\n"
  },
  {
    "path": "service/history/queuev2/queue_state_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestQueueState_IsEmpty(t *testing.T) {\n\tstate := &VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t}\n\tassert.True(t, state.IsEmpty())\n\n\tctrl := gomock.NewController(t)\n\tmockPredicate := NewMockPredicate(ctrl)\n\tmockPredicate.EXPECT().IsEmpty().Return(true)\n\n\tstate = &VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t\t},\n\t\tPredicate: mockPredicate,\n\t}\n\tassert.True(t, state.IsEmpty())\n}\n\nfunc TestVirtualSliceState_Contains(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockPredicate := NewMockPredicate(ctrl)\n\tmockPredicate.EXPECT().Check(gomock.Any()).Return(true).AnyTimes()\n\n\tstate := &VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t\t},\n\t\tPredicate: mockPredicate,\n\t}\n\tassert.True(t, state.Contains(&persistence.DecisionTimeoutTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              1,\n\t\t\tVisibilityTimestamp: time.Unix(0, 1),\n\t\t},\n\t}))\n\tassert.False(t, state.Contains(&persistence.DecisionTimeoutTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              100,\n\t\t\tVisibilityTimestamp: time.Unix(0, 10),\n\t\t},\n\t}))\n\tassert.False(t, state.Contains(&persistence.DecisionTimeoutTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              101,\n\t\t\tVisibilityTimestamp: time.Unix(0, 10),\n\t\t},\n\t}))\n\tassert.False(t, state.Contains(&persistence.DecisionTimeoutTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              101,\n\t\t\tVisibilityTimestamp: time.Unix(0, 0),\n\t\t},\n\t}))\n}\n\nfunc TestVirtualSliceState_TrySplitByTaskKey(t *testing.T) {\n\tstate := &VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t}\n\n\tsplit1, split2, ok := state.TrySplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 5), 50))\n\tassert.True(t, ok)\n\tassert.Equal(t, Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 5), 50),\n\t}, split1.Range)\n\tassert.Equal(t, Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 5), 50),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}, split2.Range)\n\n\t_, _, ok = state.TrySplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 10), 100))\n\tassert.False(t, ok)\n\n\t_, _, ok = state.TrySplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 1), 0))\n\tassert.False(t, ok)\n\n\t_, _, ok = state.TrySplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 11), 100))\n\tassert.False(t, ok)\n\n\t_, _, ok = state.TrySplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 0), 101))\n\tassert.False(t, ok)\n}\n\nfunc TestVirtualSliceState_TrySplitByPredicate(t *testing.T) {\n\tbaseRange := Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}\n\tbasePredicate := NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false)\n\n\ttests := []struct {\n\t\tname           string\n\t\tstate          VirtualSliceState\n\t\tsplitPredicate Predicate\n\t\texpectedSplit  bool\n\t\texpectedFirst  VirtualSliceState\n\t\texpectedSecond VirtualSliceState\n\t}{\n\t\t{\n\t\t\tname: \"universal predicate should not split\",\n\t\t\tstate: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: basePredicate,\n\t\t\t},\n\t\t\tsplitPredicate: NewUniversalPredicate(),\n\t\t\texpectedSplit:  false,\n\t\t\texpectedFirst:  VirtualSliceState{},\n\t\t\texpectedSecond: VirtualSliceState{},\n\t\t},\n\t\t{\n\t\t\tname: \"empty predicate should not split\",\n\t\t\tstate: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: basePredicate,\n\t\t\t},\n\t\t\tsplitPredicate: NewEmptyPredicate(),\n\t\t\texpectedSplit:  false,\n\t\t\texpectedFirst:  VirtualSliceState{},\n\t\t\texpectedSecond: VirtualSliceState{},\n\t\t},\n\t\t{\n\t\t\tname: \"identical predicate should not split\",\n\t\t\tstate: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: basePredicate,\n\t\t\t},\n\t\t\tsplitPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpectedSplit:  false,\n\t\t\texpectedFirst:  VirtualSliceState{},\n\t\t\texpectedSecond: VirtualSliceState{},\n\t\t},\n\t\t{\n\t\t\tname: \"different predicate should split successfully\",\n\t\t\tstate: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: basePredicate,\n\t\t\t},\n\t\t\tsplitPredicate: NewDomainIDPredicate([]string{\"domain3\"}, false),\n\t\t\texpectedSplit:  true,\n\t\t\texpectedFirst: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: And(basePredicate, NewDomainIDPredicate([]string{\"domain3\"}, false)),\n\t\t\t},\n\t\t\texpectedSecond: VirtualSliceState{\n\t\t\t\tRange:     baseRange,\n\t\t\t\tPredicate: And(basePredicate, Not(NewDomainIDPredicate([]string{\"domain3\"}, false))),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tfirst, second, split := tt.state.TrySplitByPredicate(tt.splitPredicate)\n\n\t\t\tassert.Equal(t, tt.expectedSplit, split)\n\n\t\t\tif tt.expectedSplit {\n\t\t\t\tassert.Equal(t, tt.expectedFirst.Range, first.Range)\n\t\t\t\tassert.Equal(t, tt.expectedSecond.Range, second.Range)\n\t\t\t\t// For predicates, we check if they produce the same results rather than exact equality\n\t\t\t\t// since the And and Not operations create new predicate instances\n\t\t\t\tassert.True(t, tt.expectedFirst.Predicate.Equals(first.Predicate))\n\t\t\t\tassert.True(t, tt.expectedSecond.Predicate.Equals(second.Predicate))\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.expectedFirst, first)\n\t\t\t\tassert.Equal(t, tt.expectedSecond, second)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/range.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype Range struct {\n\tInclusiveMinTaskKey persistence.HistoryTaskKey\n\tExclusiveMaxTaskKey persistence.HistoryTaskKey\n}\n\nfunc (r *Range) IsEmpty() bool {\n\treturn r.InclusiveMinTaskKey.Compare(r.ExclusiveMaxTaskKey) >= 0\n}\n\nfunc (r *Range) Contains(taskKey persistence.HistoryTaskKey) bool {\n\treturn taskKey.Compare(r.InclusiveMinTaskKey) >= 0 && taskKey.Compare(r.ExclusiveMaxTaskKey) < 0\n}\n\nfunc (r *Range) ContainsRange(other Range) bool {\n\treturn r.InclusiveMinTaskKey.Compare(other.InclusiveMinTaskKey) <= 0 && r.ExclusiveMaxTaskKey.Compare(other.ExclusiveMaxTaskKey) >= 0\n}\n\nfunc (r *Range) CanMerge(other Range) bool {\n\treturn r.InclusiveMinTaskKey.Compare(other.ExclusiveMaxTaskKey) <= 0 && r.ExclusiveMaxTaskKey.Compare(other.InclusiveMinTaskKey) >= 0\n}\n\nfunc (r *Range) CanSplitByTaskKey(taskKey persistence.HistoryTaskKey) bool {\n\treturn taskKey.Compare(r.InclusiveMinTaskKey) > 0 && taskKey.Compare(r.ExclusiveMaxTaskKey) < 0\n}\n"
  },
  {
    "path": "service/history/queuev2/range_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestRange_IsEmpty(t *testing.T) {\n\tr := Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(2),\n\t}\n\tassert.False(t, r.IsEmpty())\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(1),\n\t}\n\tassert.True(t, r.IsEmpty())\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t}\n\tassert.True(t, r.IsEmpty())\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(1, 0), 1),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(2, 0), 1),\n\t}\n\tassert.False(t, r.IsEmpty())\n\n}\n\nfunc TestRange_Contains(t *testing.T) {\n\tr := Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(2),\n\t}\n\tassert.True(t, r.Contains(persistence.NewImmediateTaskKey(1)))\n\tassert.False(t, r.Contains(persistence.NewImmediateTaskKey(2)))\n\tassert.False(t, r.Contains(persistence.NewImmediateTaskKey(0)))\n\tassert.False(t, r.Contains(persistence.NewImmediateTaskKey(3)))\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 100), 0),\n\t}\n\tassert.False(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 0), 0)))\n\tassert.True(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 1), 0)))\n\tassert.True(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 1), 1000000)))\n\tassert.True(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 99), 1000000)))\n\tassert.False(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 100), 0)))\n\tassert.False(t, r.Contains(persistence.NewHistoryTaskKey(time.Unix(0, 101), 0)))\n}\n\nfunc TestRange_ContainsRange(t *testing.T) {\n\tr := Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t}\n\tassert.True(t, r.ContainsRange(r))\n\tassert.True(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t}))\n\tassert.True(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t}))\n\tassert.False(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t}))\n\tassert.False(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(1),\n\t}))\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}\n\tassert.True(t, r.ContainsRange(r))\n\tassert.True(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 5), 0),\n\t}))\n\tassert.True(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 5), 101),\n\t}))\n\tassert.True(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 5), 1110),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}))\n\tassert.False(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 15), 100),\n\t}))\n\tassert.False(t, r.ContainsRange(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 100),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 101),\n\t}))\n}\n\nfunc TestRange_CanMerge(t *testing.T) {\n\tr := Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t}\n\tassert.True(t, r.CanMerge(r))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(12),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(10),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(1),\n\t}))\n\tassert.False(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(0),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(0),\n\t}))\n\tassert.False(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t}))\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}\n\tassert.True(t, r.CanMerge(r))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 1),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 99),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 100), 100),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 100),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t}))\n\tassert.True(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 100), 100),\n\t}))\n\tassert.False(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 101),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 100), 100),\n\t}))\n\tassert.False(t, r.CanMerge(Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 0), 1),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), -1),\n\t}))\n}\n\nfunc TestRange_CanSplitByTaskKey(t *testing.T) {\n\tr := Range{\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t}\n\tassert.True(t, r.CanSplitByTaskKey(persistence.NewImmediateTaskKey(5)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewImmediateTaskKey(1)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewImmediateTaskKey(10)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewImmediateTaskKey(0)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewImmediateTaskKey(11)))\n\n\tr = Range{\n\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 1), 0),\n\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(time.Unix(0, 10), 100),\n\t}\n\tassert.True(t, r.CanSplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 5), 100)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 1), 0)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 10), 100)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 0), 0)))\n\tassert.False(t, r.CanSplitByTaskKey(persistence.NewHistoryTaskKey(time.Unix(0, 10), 101)))\n}\n"
  },
  {
    "path": "service/history/queuev2/timer_queue_factory.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\ttimerQueueFactory struct {\n\t\ttaskProcessor  task.Processor\n\t\tarchivalClient archiver.Client\n\t}\n)\n\nfunc NewTimerQueueFactory(\n\ttaskProcessor task.Processor,\n\tarchivalClient archiver.Client,\n) queue.Factory {\n\treturn &timerQueueFactory{\n\t\ttaskProcessor:  taskProcessor,\n\t\tarchivalClient: archivalClient,\n\t}\n}\n\nfunc (f *timerQueueFactory) Category() persistence.HistoryTaskCategory {\n\treturn persistence.HistoryTaskCategoryTimer\n}\n\nfunc (f *timerQueueFactory) isQueueV2Enabled(shard shard.Context) bool {\n\treturn shard.GetConfig().EnableTimerQueueV2(shard.GetShardID())\n}\n\nfunc (f *timerQueueFactory) CreateQueue(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\tif f.isQueueV2Enabled(shard) {\n\t\treturn f.createQueuev2(shard, executionCache, openExecutionCheck)\n\t}\n\treturn f.createQueuev1(shard, executionCache, openExecutionCheck)\n}\n\nfunc (f *timerQueueFactory) createQueuev1(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\treturn queue.NewTimerQueueProcessor(\n\t\tshard,\n\t\tf.taskProcessor,\n\t\texecutionCache,\n\t\tf.archivalClient,\n\t\topenExecutionCheck,\n\t)\n}\n\nfunc (f *timerQueueFactory) createQueuev2(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\tlogger := shard.GetLogger().WithTags(tag.ComponentTimerQueueV2)\n\tactiveTaskExecutor := task.NewTimerActiveTaskExecutor(\n\t\tshard,\n\t\tf.archivalClient,\n\t\texecutionCache,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t\tshard.GetConfig(),\n\t)\n\thistoryResender := ndc.NewHistoryResender(\n\t\tshard.GetDomainCache(),\n\t\tshard.GetService().GetClientBean(),\n\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\treturn shard.GetEngine().ReplicateEventsV2(ctx, request)\n\t\t},\n\t\tshard.GetConfig().StandbyTaskReReplicationContextTimeout,\n\t\topenExecutionCheck,\n\t\tlogger,\n\t)\n\tstandbyTaskExecutor := task.NewTimerStandbyTaskExecutor(\n\t\tshard,\n\t\tf.archivalClient,\n\t\texecutionCache,\n\t\thistoryResender,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t\tshard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard.GetConfig(),\n\t)\n\texecutorWrapper := task.NewExecutorWrapper(\n\t\tshard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard.GetActiveClusterManager(),\n\t\tactiveTaskExecutor,\n\t\tstandbyTaskExecutor,\n\t\tlogger,\n\t)\n\tconfig := shard.GetConfig()\n\treturn NewScheduledQueue(\n\t\tshard,\n\t\tpersistence.HistoryTaskCategoryTimer,\n\t\tf.taskProcessor,\n\t\texecutorWrapper,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t\tshard.GetMetricsClient().Scope(metrics.TimerQueueProcessorV2Scope).Tagged(metrics.ShardIDTag(shard.GetShardID())),\n\t\t&Options{\n\t\t\tPageSize:                             config.TimerTaskBatchSize,\n\t\t\tDeleteBatchSize:                      config.TimerTaskDeleteBatchSize,\n\t\t\tMaxPollRPS:                           config.TimerProcessorMaxPollRPS,\n\t\t\tMaxPollInterval:                      config.TimerProcessorMaxPollInterval,\n\t\t\tMaxPollIntervalJitterCoefficient:     config.TimerProcessorMaxPollIntervalJitterCoefficient,\n\t\t\tUpdateAckInterval:                    config.TimerProcessorUpdateAckInterval,\n\t\t\tUpdateAckIntervalJitterCoefficient:   config.TimerProcessorUpdateAckIntervalJitterCoefficient,\n\t\t\tMaxPendingTasksCount:                 config.QueueMaxPendingTaskCount,\n\t\t\tPollBackoffInterval:                  config.QueueProcessorPollBackoffInterval,\n\t\t\tPollBackoffIntervalJitterCoefficient: config.QueueProcessorPollBackoffIntervalJitterCoefficient,\n\t\t\tVirtualSliceForceAppendInterval:      config.VirtualSliceForceAppendInterval,\n\t\t\tMaxStartJitterInterval:               dynamicproperties.GetDurationPropertyFn(0),\n\t\t\tRedispatchInterval:                   config.ActiveTaskRedispatchInterval,\n\t\t\tCriticalPendingTaskCount:             config.QueueCriticalPendingTaskCount,\n\t\t\tEnablePendingTaskCountAlert:          func() bool { return config.EnableTimerQueueV2PendingTaskCountAlert(shard.GetShardID()) },\n\t\t\tMaxVirtualQueueCount:                 config.QueueMaxVirtualQueueCount,\n\t\t},\n\t)\n}\n"
  },
  {
    "path": "service/history/queuev2/timer_queue_factory_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nfunc TestTimerQueueFactory_CreateQueuev2(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\n\t// Create the factory\n\tfactory := &timerQueueFactory{\n\t\ttaskProcessor:  task.NewMockProcessor(ctrl),\n\t\tarchivalClient: archiver.NewMockClient(ctrl),\n\t}\n\n\t// Test the createQueuev2 method\n\tprocessor := factory.createQueuev2(mockShard, execution.NewMockCache(ctrl), invariant.NewMockInvariant(ctrl))\n\n\t// Verify the result\n\tassert.NotNil(t, processor)\n\tassert.Implements(t, (*queue.Processor)(nil), processor)\n}\n\nfunc TestTimerQueueFactory_Category(t *testing.T) {\n\tfactory := &timerQueueFactory{}\n\n\tcategory := factory.Category()\n\n\tassert.Equal(t, persistence.HistoryTaskCategoryTimer, category)\n}\n\nfunc TestTimerQueueFactory_IsQueueV2Enabled(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\n\tfactory := &timerQueueFactory{}\n\n\t// Test the isQueueV2Enabled method\n\t// by default, queue v2 is disabled\n\tenabled := factory.isQueueV2Enabled(mockShard)\n\tassert.False(t, enabled)\n}\n"
  },
  {
    "path": "service/history/queuev2/transfer_queue_factory.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage queuev2\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\ttransferQueueFactory struct {\n\t\ttaskProcessor  task.Processor\n\t\tarchivalClient archiver.Client\n\t\twfIDCache      workflowcache.WFCache\n\t}\n)\n\nfunc NewTransferQueueFactory(\n\ttaskProcessor task.Processor,\n\tarchivalClient archiver.Client,\n\twfIDCache workflowcache.WFCache,\n) queue.Factory {\n\treturn &transferQueueFactory{taskProcessor, archivalClient, wfIDCache}\n}\n\nfunc (f *transferQueueFactory) Category() persistence.HistoryTaskCategory {\n\treturn persistence.HistoryTaskCategoryTransfer\n}\n\nfunc (f *transferQueueFactory) isQueueV2Enabled(shard shard.Context) bool {\n\treturn shard.GetConfig().EnableTransferQueueV2(shard.GetShardID())\n}\n\nfunc (f *transferQueueFactory) CreateQueue(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\tif f.isQueueV2Enabled(shard) {\n\t\treturn f.createQueuev2(shard, executionCache, openExecutionCheck)\n\t}\n\treturn f.createQueuev1(shard, executionCache, openExecutionCheck)\n}\n\nfunc (f *transferQueueFactory) createQueuev1(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\tworkflowResetter := reset.NewWorkflowResetter(shard, executionCache, shard.GetLogger())\n\treturn queue.NewTransferQueueProcessor(\n\t\tshard,\n\t\tf.taskProcessor,\n\t\texecutionCache,\n\t\tworkflowResetter,\n\t\tf.archivalClient,\n\t\topenExecutionCheck,\n\t\tf.wfIDCache,\n\t)\n}\n\nfunc (f *transferQueueFactory) createQueuev2(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\topenExecutionCheck invariant.Invariant,\n) queue.Processor {\n\tlogger := shard.GetLogger().WithTags(tag.ComponentTransferQueueV2)\n\tworkflowResetter := reset.NewWorkflowResetter(shard, executionCache, logger)\n\tactiveTaskExecutor := task.NewTransferActiveTaskExecutor(\n\t\tshard,\n\t\tf.archivalClient,\n\t\texecutionCache,\n\t\tworkflowResetter,\n\t\tlogger,\n\t\tshard.GetConfig(),\n\t\tf.wfIDCache,\n\t)\n\n\thistoryResender := ndc.NewHistoryResender(\n\t\tshard.GetDomainCache(),\n\t\tshard.GetService().GetClientBean(),\n\t\tfunc(ctx context.Context, request *types.ReplicateEventsV2Request) error {\n\t\t\treturn shard.GetEngine().ReplicateEventsV2(ctx, request)\n\t\t},\n\t\tshard.GetConfig().StandbyTaskReReplicationContextTimeout,\n\t\topenExecutionCheck,\n\t\tlogger,\n\t)\n\tstandbyTaskExecutor := task.NewTransferStandbyTaskExecutor(\n\t\tshard,\n\t\tf.archivalClient,\n\t\texecutionCache,\n\t\thistoryResender,\n\t\tlogger,\n\t\tshard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard.GetConfig(),\n\t)\n\n\texecutorWrapper := task.NewExecutorWrapper(\n\t\tshard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tshard.GetActiveClusterManager(),\n\t\tactiveTaskExecutor,\n\t\tstandbyTaskExecutor,\n\t\tlogger,\n\t)\n\tconfig := shard.GetConfig()\n\treturn NewImmediateQueue(\n\t\tshard,\n\t\tpersistence.HistoryTaskCategoryTransfer,\n\t\tf.taskProcessor,\n\t\texecutorWrapper,\n\t\tlogger,\n\t\tshard.GetMetricsClient(),\n\t\tshard.GetMetricsClient().Scope(metrics.TransferQueueProcessorV2Scope).Tagged(metrics.ShardIDTag(shard.GetShardID())),\n\t\t&Options{\n\t\t\tPageSize:                             config.TransferTaskBatchSize,\n\t\t\tDeleteBatchSize:                      config.TransferTaskDeleteBatchSize,\n\t\t\tMaxPollRPS:                           config.TransferProcessorMaxPollRPS,\n\t\t\tMaxPollInterval:                      config.TransferProcessorMaxPollInterval,\n\t\t\tMaxPollIntervalJitterCoefficient:     config.TransferProcessorMaxPollIntervalJitterCoefficient,\n\t\t\tUpdateAckInterval:                    config.TransferProcessorUpdateAckInterval,\n\t\t\tUpdateAckIntervalJitterCoefficient:   config.TransferProcessorUpdateAckIntervalJitterCoefficient,\n\t\t\tMaxPendingTasksCount:                 config.QueueMaxPendingTaskCount,\n\t\t\tPollBackoffInterval:                  config.QueueProcessorPollBackoffInterval,\n\t\t\tPollBackoffIntervalJitterCoefficient: config.QueueProcessorPollBackoffIntervalJitterCoefficient,\n\t\t\tVirtualSliceForceAppendInterval:      config.VirtualSliceForceAppendInterval,\n\t\t\tEnableValidator:                      config.TransferProcessorEnableValidator,\n\t\t\tValidationInterval:                   config.TransferProcessorValidationInterval,\n\t\t\tMaxStartJitterInterval:               dynamicproperties.GetDurationPropertyFn(0),\n\t\t\tRedispatchInterval:                   config.ActiveTaskRedispatchInterval,\n\t\t\tCriticalPendingTaskCount:             config.QueueCriticalPendingTaskCount,\n\t\t\tEnablePendingTaskCountAlert:          func() bool { return config.EnableTransferQueueV2PendingTaskCountAlert(shard.GetShardID()) },\n\t\t\tMaxVirtualQueueCount:                 config.QueueMaxVirtualQueueCount,\n\t\t},\n\t)\n}\n"
  },
  {
    "path": "service/history/queuev2/transfer_queue_factory_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/queue\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/task\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nfunc TestTransferQueueFactory_CreateQueuev2(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\n\t// Create the factory\n\tfactory := &transferQueueFactory{\n\t\ttaskProcessor:  task.NewMockProcessor(ctrl),\n\t\tarchivalClient: archiver.NewMockClient(ctrl),\n\t}\n\n\t// Test the createQueuev2 method\n\tprocessor := factory.createQueuev2(mockShard, execution.NewMockCache(ctrl), invariant.NewMockInvariant(ctrl))\n\n\t// Verify the result\n\tassert.NotNil(t, processor)\n\tassert.Implements(t, (*queue.Processor)(nil), processor)\n}\n\nfunc TestTransferQueueFactory_CreateQueuev1(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\n\t// Create the factory\n\tfactory := &transferQueueFactory{\n\t\ttaskProcessor:  task.NewMockProcessor(ctrl),\n\t\tarchivalClient: archiver.NewMockClient(ctrl),\n\t}\n\n\t// Test the createQueuev1 method\n\tprocessor := factory.createQueuev1(mockShard, execution.NewMockCache(ctrl), invariant.NewMockInvariant(ctrl))\n\n\t// Verify the result\n\tassert.NotNil(t, processor)\n\tassert.Implements(t, (*queue.Processor)(nil), processor)\n}\n\nfunc TestTransferQueueFactory_Category(t *testing.T) {\n\tfactory := &transferQueueFactory{}\n\n\tcategory := factory.Category()\n\n\tassert.Equal(t, persistence.HistoryTaskCategoryTransfer, category)\n}\n\nfunc TestTransferQueueFactory_IsQueueV2Enabled(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockShard := shard.NewTestContext(\n\t\tt, ctrl, &persistence.ShardInfo{\n\t\t\tShardID:          10,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest())\n\n\tfactory := &transferQueueFactory{}\n\n\t// Test the isQueueV2Enabled method\n\t// by default, queue v2 is disabled\n\tenabled := factory.isQueueV2Enabled(mockShard)\n\tassert.False(t, enabled)\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination virtual_queue_mock.go github.com/uber/cadence/service/history/queuev2 VirtualQueue\npackage queuev2\n\nimport (\n\t\"container/list\"\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nvar (\n\ttaskSchedulerThrottleBackoffInterval = time.Second * 5\n\ttaskReaderErrorBackoffInterval       = time.Second\n)\n\ntype (\n\tVirtualQueue interface {\n\t\tcommon.Daemon\n\t\t// GetState return the current state of the virtual queue\n\t\tGetState() []VirtualSliceState\n\t\t// UpdateAndGetState update the state of the virtual queue and return the current state\n\t\tUpdateAndGetState() []VirtualSliceState\n\t\t// MergeSlices merge the incoming slices into the virtual queue, this is used when we want to merge slices to a non-root virtual queue\n\t\tMergeSlices(...VirtualSlice)\n\t\t// MergeWithLastSlice merge the incoming slice with the last slice in the virtual queue, this is used when we want to add a new slice to the root virtual queue to avoid nullify the effect of AppendSlices\n\t\tMergeWithLastSlice(VirtualSlice)\n\t\t// AppendSlices append the incoming slices to the virtual queue, this is used when we want to add a new slice to the root virtual queue to prevent infinite growth of the virtual slice\n\t\tAppendSlices(...VirtualSlice)\n\t\t// IterateSlices iterate over the slices in the virtual queue\n\t\tIterateSlices(func(VirtualSlice))\n\t\t// ClearSlices calls the Clear method of the slices that satisfy the predicate function\n\t\tClearSlices(func(VirtualSlice) bool)\n\t\t// SplitSlices applies the split function to the slices in the virtual queue and return the remaining slices that should be kept in the virtual queue and whether the split is applied\n\t\tSplitSlices(func(VirtualSlice) (remaining []VirtualSlice, split bool))\n\t\t// Pause pauses the virtual queue for a while\n\t\tPause(time.Duration)\n\t}\n\n\tVirtualQueueOptions struct {\n\t\tPageSize                             dynamicproperties.IntPropertyFn\n\t\tMaxPendingTasksCount                 dynamicproperties.IntPropertyFn\n\t\tPollBackoffInterval                  dynamicproperties.DurationPropertyFn\n\t\tPollBackoffIntervalJitterCoefficient dynamicproperties.FloatPropertyFn\n\t}\n\n\tvirtualQueueImpl struct {\n\t\tqueueOptions        *VirtualQueueOptions\n\t\tprocessor           task.Processor\n\t\trescheduler         task.Rescheduler\n\t\tlogger              log.Logger\n\t\tmetricsScope        metrics.Scope\n\t\ttimeSource          clock.TimeSource\n\t\ttaskLoadRateLimiter quotas.Limiter\n\t\tmonitor             Monitor\n\n\t\tsync.RWMutex\n\t\tstatus          int32\n\t\twg              sync.WaitGroup\n\t\tctx             context.Context\n\t\tcancel          func()\n\t\tnotifyCh        chan struct{}\n\t\tpauseController PauseController\n\t\tvirtualSlices   *list.List\n\t\tsliceToRead     *list.Element\n\t}\n)\n\nfunc NewVirtualQueue(\n\tprocessor task.Processor,\n\trescheduler task.Rescheduler,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n\ttimeSource clock.TimeSource,\n\ttaskLoadRateLimiter quotas.Limiter,\n\tmonitor Monitor,\n\tvirtualSlices []VirtualSlice,\n\tqueueOptions *VirtualQueueOptions,\n) VirtualQueue {\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tsliceList := list.New()\n\tfor _, slice := range virtualSlices {\n\t\tsliceList.PushBack(slice)\n\t}\n\n\treturn &virtualQueueImpl{\n\t\tqueueOptions:        queueOptions,\n\t\tprocessor:           processor,\n\t\trescheduler:         rescheduler,\n\t\tlogger:              logger,\n\t\tmetricsScope:        metricsScope,\n\t\ttimeSource:          timeSource,\n\t\ttaskLoadRateLimiter: taskLoadRateLimiter,\n\t\tmonitor:             monitor,\n\n\t\tstatus:          common.DaemonStatusInitialized,\n\t\tctx:             ctx,\n\t\tcancel:          cancel,\n\t\tnotifyCh:        make(chan struct{}, 1),\n\t\tpauseController: NewPauseController(timeSource),\n\t\tvirtualSlices:   sliceList,\n\t\tsliceToRead:     sliceList.Front(),\n\t}\n}\n\nfunc (q *virtualQueueImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tq.pauseController.Subscribe(\"virtual-queue\", q.notifyCh)\n\tq.wg.Add(1)\n\tgo q.run()\n\n\tq.notify()\n\n\tq.logger.Info(\"Virtual queue state changed\", tag.LifeCycleStarted)\n}\n\nfunc (q *virtualQueueImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&q.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tq.pauseController.Unsubscribe(\"virtual-queue\")\n\tq.pauseController.Stop()\n\n\tq.cancel()\n\tq.wg.Wait()\n\n\tq.RLock()\n\tdefer q.RUnlock()\n\tfor e := q.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\tslice := e.Value.(VirtualSlice)\n\t\tslice.Clear()\n\t}\n\n\tq.logger.Info(\"Virtual queue state changed\", tag.LifeCycleStopped)\n}\n\nfunc (q *virtualQueueImpl) GetState() []VirtualSliceState {\n\tq.RLock()\n\tdefer q.RUnlock()\n\n\tstates := make([]VirtualSliceState, 0, q.virtualSlices.Len())\n\tfor e := q.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\tstates = append(states, e.Value.(VirtualSlice).GetState())\n\t}\n\treturn states\n}\n\nfunc (q *virtualQueueImpl) UpdateAndGetState() []VirtualSliceState {\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tstates := make([]VirtualSliceState, 0, q.virtualSlices.Len())\n\tvar next *list.Element\n\tfor e := q.virtualSlices.Front(); e != nil; e = next {\n\t\tnext = e.Next()\n\t\tslice := e.Value.(VirtualSlice)\n\t\tstate := slice.UpdateAndGetState()\n\t\tif slice.IsEmpty() {\n\t\t\tq.virtualSlices.Remove(e)\n\t\t\tq.monitor.RemoveSlice(slice)\n\t\t} else {\n\t\t\tstates = append(states, state)\n\t\t\tq.monitor.SetSlicePendingTaskCount(slice, slice.GetPendingTaskCount())\n\t\t}\n\t}\n\treturn states\n}\n\nfunc (q *virtualQueueImpl) MergeSlices(incomingSlices ...VirtualSlice) {\n\tif len(incomingSlices) == 0 {\n\t\treturn\n\t}\n\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tmergedSlices := list.New()\n\n\tcurrentSliceElement := q.virtualSlices.Front()\n\tincomingSliceIdx := 0\n\n\tfor currentSliceElement != nil && incomingSliceIdx < len(incomingSlices) {\n\t\tcurrentSlice := currentSliceElement.Value.(VirtualSlice)\n\t\tincomingSlice := incomingSlices[incomingSliceIdx]\n\n\t\tif currentSlice.GetState().Range.InclusiveMinTaskKey.Compare(incomingSlice.GetState().Range.InclusiveMinTaskKey) < 0 {\n\t\t\tq.appendOrMergeSlice(mergedSlices, currentSlice)\n\t\t\tcurrentSliceElement = currentSliceElement.Next()\n\t\t} else {\n\t\t\tq.appendOrMergeSlice(mergedSlices, incomingSlice)\n\t\t\tincomingSliceIdx++\n\t\t}\n\t}\n\tfor ; currentSliceElement != nil; currentSliceElement = currentSliceElement.Next() {\n\t\tq.appendOrMergeSlice(mergedSlices, currentSliceElement.Value.(VirtualSlice))\n\t}\n\tfor _, slice := range incomingSlices[incomingSliceIdx:] {\n\t\tq.appendOrMergeSlice(mergedSlices, slice)\n\t}\n\n\tq.virtualSlices.Init()\n\tq.virtualSlices = mergedSlices\n\tq.resetNextReadSliceLocked()\n}\n\nfunc (q *virtualQueueImpl) MergeWithLastSlice(incomingSlice VirtualSlice) {\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tq.appendOrMergeSlice(q.virtualSlices, incomingSlice)\n\tq.resetNextReadSliceLocked()\n}\n\nfunc (q *virtualQueueImpl) AppendSlices(incomingSlices ...VirtualSlice) {\n\tif len(incomingSlices) == 0 {\n\t\treturn\n\t}\n\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tfor _, slice := range incomingSlices {\n\t\tq.virtualSlices.PushBack(slice)\n\t}\n\n\tq.resetNextReadSliceLocked()\n}\n\nfunc (q *virtualQueueImpl) IterateSlices(f func(VirtualSlice)) {\n\tq.RLock()\n\tdefer q.RUnlock()\n\n\tfor e := q.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\tf(e.Value.(VirtualSlice))\n\t}\n}\n\nfunc (q *virtualQueueImpl) ClearSlices(f func(VirtualSlice) bool) {\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tfor e := q.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\tslice := e.Value.(VirtualSlice)\n\t\tif f(slice) {\n\t\t\tslice.Clear()\n\t\t\tq.monitor.SetSlicePendingTaskCount(slice, slice.GetPendingTaskCount())\n\t\t}\n\t}\n\n\tq.resetNextReadSliceLocked()\n}\n\nfunc (q *virtualQueueImpl) SplitSlices(f func(VirtualSlice) (remaining []VirtualSlice, split bool)) {\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tremainingSlices := list.New()\n\tfor e := q.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\tslice := e.Value.(VirtualSlice)\n\t\tremaining, split := f(slice)\n\t\tif !split {\n\t\t\tremainingSlices.PushBack(slice)\n\t\t\tcontinue\n\t\t}\n\n\t\tq.monitor.RemoveSlice(slice)\n\n\t\tfor _, remainingSlice := range remaining {\n\t\t\tremainingSlices.PushBack(remainingSlice)\n\t\t\tq.monitor.SetSlicePendingTaskCount(remainingSlice, remainingSlice.GetPendingTaskCount())\n\t\t}\n\t}\n\n\tq.virtualSlices.Init()\n\tq.virtualSlices = remainingSlices\n\tq.resetNextReadSliceLocked()\n}\n\nfunc (q *virtualQueueImpl) Pause(duration time.Duration) {\n\tq.pauseController.Pause(duration)\n}\n\nfunc (q *virtualQueueImpl) notify() {\n\tselect {\n\tcase q.notifyCh <- struct{}{}:\n\tdefault:\n\t}\n}\n\nfunc (q *virtualQueueImpl) run() {\n\tdefer q.wg.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-q.ctx.Done():\n\t\t\treturn\n\t\tcase <-q.notifyCh:\n\t\t\tq.loadAndSubmitTasks()\n\t\t}\n\t}\n}\n\nfunc (q *virtualQueueImpl) loadAndSubmitTasks() {\n\tif err := q.taskLoadRateLimiter.Wait(q.ctx); err != nil {\n\t\tif q.ctx.Err() != nil {\n\t\t\treturn\n\t\t}\n\t\t// this should never happen, but we log it for debugging purposes\n\t\tq.logger.Error(\"Virtual queue failed to wait for rate limiter\", tag.Error(err))\n\t}\n\n\tq.Lock()\n\tdefer q.Unlock()\n\n\tif q.sliceToRead == nil {\n\t\treturn\n\t}\n\n\tpendingTaskCount := q.monitor.GetTotalPendingTaskCount()\n\tmaxTaskCount := q.queueOptions.MaxPendingTasksCount()\n\t// TODO: review the metrics and remove this comment or change the metric from gauge to histogram\n\tq.metricsScope.UpdateGauge(metrics.PendingTaskGauge, float64(pendingTaskCount))\n\tif pendingTaskCount >= maxTaskCount {\n\t\tq.logger.Warn(\"Too many pending tasks, pause loading tasks for a while\", tag.PendingTaskCount(pendingTaskCount), tag.MaxTaskCount(maxTaskCount))\n\t\tq.pauseController.Pause(q.queueOptions.PollBackoffInterval())\n\t}\n\n\tif q.pauseController.IsPaused() {\n\t\t// emit a metric indicating that the virtual queue is paused\n\t\tq.metricsScope.UpdateGauge(metrics.VirtualQueuePausedGauge, 1.0)\n\t\tq.logger.Debug(\"virtual queue is paused\", tag.PendingTaskCount(pendingTaskCount), tag.MaxTaskCount(maxTaskCount))\n\t\treturn\n\t}\n\n\t// emit a metric indicating that the virtual queue is alive\n\tq.metricsScope.UpdateGauge(metrics.VirtualQueueRunningGauge, 1.0)\n\tsliceToRead := q.sliceToRead.Value.(VirtualSlice)\n\n\t// This logic is to avoid the loop of loading tasks from max virtual queue -> pending task count exceeds critical task count -> unload tasks from max virtual queue\n\t// for non-root virtual queue, we know that maxTaskCount < criticalTaskCount\n\tremainingSize := maxTaskCount - pendingTaskCount\n\tif remainingSize <= 0 {\n\t\tremainingSize = 1\n\t\tq.logger.Error(\"unexpected error, virtual queue is not paused when pending task count exceeds max task count limit\", tag.PendingTaskCount(pendingTaskCount), tag.MaxTaskCount(maxTaskCount))\n\t}\n\tpageSize := min(q.queueOptions.PageSize(), remainingSize)\n\tq.logger.Debug(\"getting tasks from virtual queue\", tag.PendingTaskCount(pendingTaskCount), tag.MaxTaskCount(maxTaskCount), tag.Counter(pageSize))\n\ttasks, err := sliceToRead.GetTasks(q.ctx, pageSize)\n\tif err != nil {\n\t\tq.logger.Error(\"Virtual queue failed to get tasks\", tag.Error(err))\n\t\tq.pauseController.Pause(taskReaderErrorBackoffInterval)\n\t\treturn\n\t}\n\tq.logger.Debug(\"got tasks from virtual queue\", tag.Counter(len(tasks)))\n\n\tq.monitor.SetSlicePendingTaskCount(sliceToRead, sliceToRead.GetPendingTaskCount())\n\n\tnow := q.timeSource.Now()\n\tfor _, task := range tasks {\n\t\tif persistence.IsTaskCorrupted(task) {\n\t\t\tq.logger.Error(\"Virtual queue encountered a corrupted task\", tag.Dynamic(\"task\", task))\n\t\t\tq.metricsScope.IncCounter(metrics.CorruptedHistoryTaskCounter)\n\t\t\ttask.Ack()\n\t\t\tcontinue\n\t\t}\n\n\t\tscheduledTime := task.GetTaskKey().GetScheduledTime()\n\t\t// if the scheduled time is in the future, we need to reschedule the task\n\t\tif now.Before(scheduledTime) {\n\t\t\tq.rescheduler.RescheduleTask(task, scheduledTime)\n\t\t\tcontinue\n\t\t}\n\t\t// shard level metrics for the duration between a task being written to a queue and being fetched from it\n\t\tq.metricsScope.RecordHistogramDuration(metrics.TaskEnqueueToFetchLatency, now.Sub(task.GetVisibilityTimestamp()))\n\t\ttask.SetInitialSubmitTime(now)\n\t\tsubmitted, err := q.processor.TrySubmit(task)\n\t\tif err != nil {\n\t\t\tselect {\n\t\t\tcase <-q.ctx.Done():\n\t\t\t\treturn\n\t\t\tdefault:\n\t\t\t\tq.logger.Error(\"Virtual queue failed to submit task\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t\tif !submitted {\n\t\t\tq.metricsScope.IncCounter(metrics.ProcessingQueueThrottledCounter)\n\t\t\tq.rescheduler.RescheduleTask(task, q.timeSource.Now().Add(taskSchedulerThrottleBackoffInterval))\n\t\t}\n\t}\n\n\tif sliceToRead.HasMoreTasks() {\n\t\tq.notify()\n\t\treturn\n\t}\n\n\tq.sliceToRead = q.sliceToRead.Next()\n\tif q.sliceToRead != nil {\n\t\tq.notify()\n\t}\n}\n\nfunc (q *virtualQueueImpl) resetNextReadSliceLocked() {\n\tq.sliceToRead = nil\n\tfor element := q.virtualSlices.Front(); element != nil; element = element.Next() {\n\t\tif element.Value.(VirtualSlice).HasMoreTasks() {\n\t\t\tq.sliceToRead = element\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif q.sliceToRead != nil {\n\t\tq.notify()\n\t}\n}\n\nfunc (q *virtualQueueImpl) appendOrMergeSlice(slices *list.List, incomingSlice VirtualSlice) {\n\tif slices.Len() == 0 {\n\t\tslices.PushBack(incomingSlice)\n\t\tq.monitor.SetSlicePendingTaskCount(incomingSlice, incomingSlice.GetPendingTaskCount())\n\t\treturn\n\t}\n\n\tlastElement := slices.Back()\n\tlastSlice := lastElement.Value.(VirtualSlice)\n\tmergedSlices, merged := lastSlice.TryMergeWithVirtualSlice(incomingSlice)\n\tif !merged {\n\t\tslices.PushBack(incomingSlice)\n\t\tq.monitor.SetSlicePendingTaskCount(incomingSlice, incomingSlice.GetPendingTaskCount())\n\t\treturn\n\t}\n\n\tslices.Remove(lastElement)\n\tq.monitor.RemoveSlice(lastSlice)\n\tq.monitor.RemoveSlice(incomingSlice) // incomingSlice may already be tracked by the monitor, so we need to remove it if it's tracked\n\tfor _, mergedSlice := range mergedSlices {\n\t\tslices.PushBack(mergedSlice)\n\t\tq.monitor.SetSlicePendingTaskCount(mergedSlice, mergedSlice.GetPendingTaskCount())\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue_manager.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -destination virtual_queue_manager_mock.go -package queuev2 github.com/uber/cadence/service/history/queuev2 VirtualQueueManager\npackage queuev2\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nconst (\n\trootQueueID = 0\n)\n\ntype (\n\tVirtualQueueManagerOptions struct {\n\t\tRootQueueOptions                *VirtualQueueOptions\n\t\tNonRootQueueOptions             *VirtualQueueOptions\n\t\tVirtualSliceForceAppendInterval dynamicproperties.DurationPropertyFn\n\t}\n\tVirtualQueueManager interface {\n\t\tcommon.Daemon\n\t\tVirtualQueues() map[int64]VirtualQueue\n\t\tGetOrCreateVirtualQueue(int64) VirtualQueue\n\t\tUpdateAndGetState() map[int64][]VirtualSliceState\n\t\t// Add a new virtual slice to the root queue. This is used when new tasks are generated and max read level is updated.\n\t\t// By default, all new tasks belong to the root queue, so we need to add a new virtual slice to the root queue.\n\t\tAddNewVirtualSliceToRootQueue(VirtualSlice)\n\t}\n\n\tvirtualQueueManagerImpl struct {\n\t\tprocessor           task.Processor\n\t\ttaskInitializer     task.Initializer\n\t\trescheduler         task.Rescheduler\n\t\tqueueReader         QueueReader\n\t\tlogger              log.Logger\n\t\tmetricsScope        metrics.Scope\n\t\ttimeSource          clock.TimeSource\n\t\ttaskLoadRateLimiter quotas.Limiter\n\t\tmonitor             Monitor\n\t\tqueueManagerOptions *VirtualQueueManagerOptions\n\n\t\tsync.RWMutex\n\t\tstatus               int32\n\t\tvirtualQueues        map[int64]VirtualQueue\n\t\tcreateVirtualQueueFn func(int64, ...VirtualSlice) VirtualQueue\n\n\t\tnextForceNewSliceTime time.Time\n\t}\n)\n\nfunc NewVirtualQueueManager(\n\tprocessor task.Processor,\n\trescheduler task.Rescheduler,\n\ttaskInitializer task.Initializer,\n\tqueueReader QueueReader,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n\ttimeSource clock.TimeSource,\n\ttaskLoadRateLimiter quotas.Limiter,\n\tmonitor Monitor,\n\tqueueManagerOptions *VirtualQueueManagerOptions,\n\tvirtualQueueStates map[int64][]VirtualSliceState,\n) VirtualQueueManager {\n\tvirtualQueues := make(map[int64]VirtualQueue)\n\tfor queueID, states := range virtualQueueStates {\n\t\tvirtualSlices := make([]VirtualSlice, len(states))\n\t\tfor i, state := range states {\n\t\t\tvirtualSlices[i] = NewVirtualSlice(state, taskInitializer, queueReader, NewPendingTaskTracker(), logger)\n\t\t}\n\t\tvar opts *VirtualQueueOptions\n\t\tif queueID == rootQueueID {\n\t\t\topts = queueManagerOptions.RootQueueOptions\n\t\t} else {\n\t\t\topts = queueManagerOptions.NonRootQueueOptions\n\t\t}\n\t\tvirtualQueues[queueID] = NewVirtualQueue(processor, rescheduler, logger.WithTags(tag.VirtualQueueID(queueID)), metricsScope, timeSource, taskLoadRateLimiter, monitor, virtualSlices, opts)\n\t}\n\treturn &virtualQueueManagerImpl{\n\t\tprocessor:           processor,\n\t\ttaskInitializer:     taskInitializer,\n\t\tqueueReader:         queueReader,\n\t\trescheduler:         rescheduler,\n\t\tlogger:              logger,\n\t\tmetricsScope:        metricsScope,\n\t\ttimeSource:          timeSource,\n\t\ttaskLoadRateLimiter: taskLoadRateLimiter,\n\t\tmonitor:             monitor,\n\t\tqueueManagerOptions: queueManagerOptions,\n\t\tstatus:              common.DaemonStatusInitialized,\n\t\tvirtualQueues:       virtualQueues,\n\t\tcreateVirtualQueueFn: func(queueID int64, s ...VirtualSlice) VirtualQueue {\n\t\t\tvar opts *VirtualQueueOptions\n\t\t\tif queueID == rootQueueID {\n\t\t\t\topts = queueManagerOptions.RootQueueOptions\n\t\t\t} else {\n\t\t\t\topts = queueManagerOptions.NonRootQueueOptions\n\t\t\t}\n\t\t\treturn NewVirtualQueue(processor, rescheduler, logger.WithTags(tag.VirtualQueueID(queueID)), metricsScope, timeSource, taskLoadRateLimiter, monitor, s, opts)\n\t\t},\n\t}\n}\n\nfunc (m *virtualQueueManagerImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&m.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\tfor _, vq := range m.virtualQueues {\n\t\tvq.Start()\n\t}\n}\n\nfunc (m *virtualQueueManagerImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&m.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\tfor _, vq := range m.virtualQueues {\n\t\tvq.Stop()\n\t}\n}\n\nfunc (m *virtualQueueManagerImpl) VirtualQueues() map[int64]VirtualQueue {\n\tm.RLock()\n\tdefer m.RUnlock()\n\treturn m.virtualQueues\n}\n\nfunc (m *virtualQueueManagerImpl) GetOrCreateVirtualQueue(queueID int64) VirtualQueue {\n\tm.RLock()\n\tif vq, ok := m.virtualQueues[queueID]; ok {\n\t\tm.RUnlock()\n\t\treturn vq\n\t}\n\tm.RUnlock()\n\n\tm.Lock()\n\tdefer m.Unlock()\n\tif vq, ok := m.virtualQueues[queueID]; ok {\n\t\treturn vq\n\t}\n\tm.virtualQueues[queueID] = m.createVirtualQueueFn(queueID)\n\tm.virtualQueues[queueID].Start()\n\treturn m.virtualQueues[queueID]\n}\n\nfunc (m *virtualQueueManagerImpl) UpdateAndGetState() map[int64][]VirtualSliceState {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tvirtualQueueStates := make(map[int64][]VirtualSliceState)\n\tfor key, vq := range m.virtualQueues {\n\t\tstate := vq.UpdateAndGetState()\n\t\tif len(state) > 0 {\n\t\t\tvirtualQueueStates[key] = state\n\t\t} else if key != rootQueueID {\n\t\t\tvq.Stop()\n\t\t\tdelete(m.virtualQueues, key)\n\t\t}\n\t}\n\treturn virtualQueueStates\n}\n\nfunc (m *virtualQueueManagerImpl) AddNewVirtualSliceToRootQueue(s VirtualSlice) {\n\tm.RLock()\n\tif vq, ok := m.virtualQueues[rootQueueID]; ok {\n\t\tm.RUnlock()\n\t\tm.appendOrMergeSlice(vq, s)\n\t\treturn\n\t}\n\tm.RUnlock()\n\n\tm.Lock()\n\tdefer m.Unlock()\n\tif vq, ok := m.virtualQueues[rootQueueID]; ok {\n\t\tm.appendOrMergeSlice(vq, s)\n\t\treturn\n\t}\n\n\tm.virtualQueues[rootQueueID] = m.createVirtualQueueFn(rootQueueID, s)\n\tm.virtualQueues[rootQueueID].Start()\n}\n\nfunc (m *virtualQueueManagerImpl) appendOrMergeSlice(vq VirtualQueue, s VirtualSlice) {\n\tnow := m.timeSource.Now()\n\tnewVirtualSliceState := s.GetState()\n\t// TODO: we should set a limit on the number of virtual slices to prevent the size of queue state from being too large to be stored in database\n\tif now.After(m.nextForceNewSliceTime) {\n\t\tm.logger.Debug(\"append new slice to virtual queue\", tag.Dynamic(\"currentTime\", now), tag.Dynamic(\"nextForceNewSliceTime\", m.nextForceNewSliceTime), tag.Dynamic(\"inclusiveMinTaskKey\", newVirtualSliceState.Range.InclusiveMinTaskKey), tag.Dynamic(\"exclusiveMaxTaskKey\", newVirtualSliceState.Range.ExclusiveMaxTaskKey))\n\t\tvq.AppendSlices(s)\n\t\tm.nextForceNewSliceTime = now.Add(m.queueManagerOptions.VirtualSliceForceAppendInterval())\n\t\treturn\n\t}\n\tm.logger.Debug(\"merge slice to virtual queue\", tag.Dynamic(\"currentTime\", now), tag.Dynamic(\"nextForceNewSliceTime\", m.nextForceNewSliceTime), tag.Dynamic(\"inclusiveMinTaskKey\", newVirtualSliceState.Range.InclusiveMinTaskKey), tag.Dynamic(\"exclusiveMaxTaskKey\", newVirtualSliceState.Range.ExclusiveMaxTaskKey))\n\tvq.MergeWithLastSlice(s)\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue_manager_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: VirtualQueueManager)\n//\n// Generated by this command:\n//\n//\tmockgen -destination virtual_queue_manager_mock.go -package queuev2 github.com/uber/cadence/service/history/queuev2 VirtualQueueManager\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockVirtualQueueManager is a mock of VirtualQueueManager interface.\ntype MockVirtualQueueManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVirtualQueueManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockVirtualQueueManagerMockRecorder is the mock recorder for MockVirtualQueueManager.\ntype MockVirtualQueueManagerMockRecorder struct {\n\tmock *MockVirtualQueueManager\n}\n\n// NewMockVirtualQueueManager creates a new mock instance.\nfunc NewMockVirtualQueueManager(ctrl *gomock.Controller) *MockVirtualQueueManager {\n\tmock := &MockVirtualQueueManager{ctrl: ctrl}\n\tmock.recorder = &MockVirtualQueueManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVirtualQueueManager) EXPECT() *MockVirtualQueueManagerMockRecorder {\n\treturn m.recorder\n}\n\n// AddNewVirtualSliceToRootQueue mocks base method.\nfunc (m *MockVirtualQueueManager) AddNewVirtualSliceToRootQueue(arg0 VirtualSlice) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddNewVirtualSliceToRootQueue\", arg0)\n}\n\n// AddNewVirtualSliceToRootQueue indicates an expected call of AddNewVirtualSliceToRootQueue.\nfunc (mr *MockVirtualQueueManagerMockRecorder) AddNewVirtualSliceToRootQueue(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddNewVirtualSliceToRootQueue\", reflect.TypeOf((*MockVirtualQueueManager)(nil).AddNewVirtualSliceToRootQueue), arg0)\n}\n\n// GetOrCreateVirtualQueue mocks base method.\nfunc (m *MockVirtualQueueManager) GetOrCreateVirtualQueue(arg0 int64) VirtualQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOrCreateVirtualQueue\", arg0)\n\tret0, _ := ret[0].(VirtualQueue)\n\treturn ret0\n}\n\n// GetOrCreateVirtualQueue indicates an expected call of GetOrCreateVirtualQueue.\nfunc (mr *MockVirtualQueueManagerMockRecorder) GetOrCreateVirtualQueue(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOrCreateVirtualQueue\", reflect.TypeOf((*MockVirtualQueueManager)(nil).GetOrCreateVirtualQueue), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockVirtualQueueManager) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockVirtualQueueManagerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockVirtualQueueManager)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockVirtualQueueManager) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockVirtualQueueManagerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockVirtualQueueManager)(nil).Stop))\n}\n\n// UpdateAndGetState mocks base method.\nfunc (m *MockVirtualQueueManager) UpdateAndGetState() map[int64][]VirtualSliceState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAndGetState\")\n\tret0, _ := ret[0].(map[int64][]VirtualSliceState)\n\treturn ret0\n}\n\n// UpdateAndGetState indicates an expected call of UpdateAndGetState.\nfunc (mr *MockVirtualQueueManagerMockRecorder) UpdateAndGetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAndGetState\", reflect.TypeOf((*MockVirtualQueueManager)(nil).UpdateAndGetState))\n}\n\n// VirtualQueues mocks base method.\nfunc (m *MockVirtualQueueManager) VirtualQueues() map[int64]VirtualQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"VirtualQueues\")\n\tret0, _ := ret[0].(map[int64]VirtualQueue)\n\treturn ret0\n}\n\n// VirtualQueues indicates an expected call of VirtualQueues.\nfunc (mr *MockVirtualQueueManagerMockRecorder) VirtualQueues() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"VirtualQueues\", reflect.TypeOf((*MockVirtualQueueManager)(nil).VirtualQueues))\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue_manager_test.go",
    "content": "package queuev2\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestVirtualQueueManager_VirtualQueues(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tinitialStates   map[int64][]VirtualSliceState\n\t\texpectedStates  map[int64][]VirtualSliceState\n\t\tsetupMockQueues func(map[int64]*MockVirtualQueue)\n\t}{\n\t\t{\n\t\t\tname:           \"empty virtual queues\",\n\t\t\tinitialStates:  map[int64][]VirtualSliceState{},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\t// No mocks to set up for empty case\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"single queue with single slice\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().GetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple queues with multiple slices\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().GetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmocks[2].EXPECT().GetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"queue with empty state\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().GetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmocks[2].EXPECT().GetState().Return([]VirtualSliceState{})\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mock dependencies\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockTaskInitializer := func(t persistence.Task) task.Task {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\tmockTask.EXPECT().GetTaskID().Return(t.GetTaskID())\n\t\t\t\treturn mockTask\n\t\t\t}\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockLogger := log.NewNoop()\n\t\t\tmockMetricsScope := metrics.NoopScope\n\n\t\t\t// Create virtual queues map with mocks\n\t\t\tvirtualQueues := make(map[int64]VirtualQueue)\n\t\t\tmockQueues := make(map[int64]*MockVirtualQueue)\n\t\t\tfor queueID := range tt.initialStates {\n\t\t\t\tmockQueue := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockQueues[queueID] = mockQueue\n\t\t\t\tvirtualQueues[queueID] = mockQueue\n\t\t\t}\n\n\t\t\t// Set up mock expectations\n\t\t\ttt.setupMockQueues(mockQueues)\n\n\t\t\t// Create manager instance\n\t\t\tmanager := &virtualQueueManagerImpl{\n\t\t\t\tprocessor:       mockProcessor,\n\t\t\t\ttaskInitializer: mockTaskInitializer,\n\t\t\t\trescheduler:     mockRescheduler,\n\t\t\t\tqueueReader:     mockQueueReader,\n\t\t\t\tlogger:          mockLogger,\n\t\t\t\tmetricsScope:    mockMetricsScope,\n\t\t\t\tqueueManagerOptions: &VirtualQueueManagerOptions{\n\t\t\t\t\tRootQueueOptions: &VirtualQueueOptions{},\n\t\t\t\t\tNonRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\t\t\tPageSize: dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tstatus:        common.DaemonStatusInitialized,\n\t\t\t\tvirtualQueues: virtualQueues,\n\t\t\t}\n\n\t\t\t// Execute test\n\t\t\tvqs := manager.VirtualQueues()\n\t\t\tstates := make(map[int64][]VirtualSliceState)\n\t\t\tfor queueID, virtualQueue := range vqs {\n\t\t\t\tstates[queueID] = virtualQueue.GetState()\n\t\t\t}\n\n\t\t\t// Verify results\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\t\t})\n\t}\n}\n\nfunc TestVirtualQueueManager_UpdateAndGetState(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tinitialStates   map[int64][]VirtualSliceState\n\t\texpectedStates  map[int64][]VirtualSliceState\n\t\tsetupMockQueues func(map[int64]*MockVirtualQueue)\n\t}{\n\t\t{\n\t\t\tname:           \"empty virtual queues\",\n\t\t\tinitialStates:  map[int64][]VirtualSliceState{},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\t// No mocks to set up for empty case\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"single queue with single slice\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple queues with multiple slices\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmocks[2].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"queue with empty state gets removed\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tmocks[2].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{})\n\t\t\t\tmocks[2].EXPECT().Stop()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"all queues empty get removed\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t1: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t2: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[1].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{})\n\t\t\t\tmocks[1].EXPECT().Stop()\n\t\t\t\tmocks[2].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{})\n\t\t\t\tmocks[2].EXPECT().Stop()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"empty root queue is not removed\",\n\t\t\tinitialStates: map[int64][]VirtualSliceState{\n\t\t\t\t0: {\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedStates: map[int64][]VirtualSliceState{},\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue) {\n\t\t\t\tmocks[0].EXPECT().UpdateAndGetState().Return([]VirtualSliceState{})\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mock dependencies\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockTaskInitializer := func(t persistence.Task) task.Task {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\tmockTask.EXPECT().GetTaskID().Return(t.GetTaskID())\n\t\t\t\treturn mockTask\n\t\t\t}\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockLogger := log.NewNoop()\n\t\t\tmockMetricsScope := metrics.NoopScope\n\n\t\t\t// Create virtual queues map with mocks\n\t\t\tvirtualQueues := make(map[int64]VirtualQueue)\n\t\t\tmockQueues := make(map[int64]*MockVirtualQueue)\n\t\t\tfor queueID := range tt.initialStates {\n\t\t\t\tmockQueue := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockQueues[queueID] = mockQueue\n\t\t\t\tvirtualQueues[queueID] = mockQueue\n\t\t\t}\n\n\t\t\t// Set up mock expectations\n\t\t\ttt.setupMockQueues(mockQueues)\n\n\t\t\t// Create manager instance\n\t\t\tmanager := &virtualQueueManagerImpl{\n\t\t\t\tprocessor:       mockProcessor,\n\t\t\t\ttaskInitializer: mockTaskInitializer,\n\t\t\t\trescheduler:     mockRescheduler,\n\t\t\t\tqueueReader:     mockQueueReader,\n\t\t\t\tlogger:          mockLogger,\n\t\t\t\tmetricsScope:    mockMetricsScope,\n\t\t\t\tqueueManagerOptions: &VirtualQueueManagerOptions{\n\t\t\t\t\tRootQueueOptions: &VirtualQueueOptions{},\n\t\t\t\t\tNonRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\t\t\tMaxPendingTasksCount: dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tstatus:        common.DaemonStatusInitialized,\n\t\t\t\tvirtualQueues: virtualQueues,\n\t\t\t}\n\n\t\t\t// Execute test\n\t\t\tstates := manager.UpdateAndGetState()\n\n\t\t\t// Verify results\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\t\t})\n\t}\n}\n\nfunc TestVirtualQueueManager_AddNewVirtualSlice(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\tinitialQueues   map[int64]VirtualQueue\n\t\tnewSlice        VirtualSlice\n\t\tsetupMockQueues func(map[int64]*MockVirtualQueue, *MockVirtualSlice)\n\t\tverifyQueues    func(*testing.T, map[int64]VirtualQueue)\n\t\tappendSlice     bool\n\t}{\n\t\t{\n\t\t\tname: \"merge slice to existing root queue\",\n\t\t\tinitialQueues: map[int64]VirtualQueue{\n\t\t\t\trootQueueID: nil, // Will be replaced with mock\n\t\t\t},\n\t\t\tnewSlice: nil, // Will be replaced with mock\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue, slice *MockVirtualSlice) {\n\t\t\t\tmocks[rootQueueID].EXPECT().MergeWithLastSlice(slice)\n\t\t\t\tslice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tverifyQueues: func(t *testing.T, queues map[int64]VirtualQueue) {\n\t\t\t\tassert.Contains(t, queues, int64(rootQueueID))\n\t\t\t\tassert.Len(t, queues, 1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"append slice to existing root queue\",\n\t\t\tinitialQueues: map[int64]VirtualQueue{\n\t\t\t\trootQueueID: nil, // Will be replaced with mock\n\t\t\t},\n\t\t\tnewSlice: nil, // Will be replaced with mock\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue, slice *MockVirtualSlice) {\n\t\t\t\tmocks[rootQueueID].EXPECT().AppendSlices(slice)\n\t\t\t\tslice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\tverifyQueues: func(t *testing.T, queues map[int64]VirtualQueue) {\n\t\t\t\tassert.Contains(t, queues, int64(rootQueueID))\n\t\t\t\tassert.Len(t, queues, 1)\n\t\t\t},\n\t\t\tappendSlice: true,\n\t\t},\n\t\t{\n\t\t\tname:          \"create new root queue when none exists\",\n\t\t\tinitialQueues: map[int64]VirtualQueue{},\n\t\t\tnewSlice:      nil, // Will be replaced with mock\n\t\t\tsetupMockQueues: func(mocks map[int64]*MockVirtualQueue, slice *MockVirtualSlice) {\n\t\t\t\t// No expectations needed as we're creating a new queue\n\t\t\t},\n\t\t\tverifyQueues: func(t *testing.T, queues map[int64]VirtualQueue) {\n\t\t\t\tassert.Contains(t, queues, int64(rootQueueID))\n\t\t\t\tassert.Len(t, queues, 1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Create mock dependencies\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockTaskInitializer := func(t persistence.Task) task.Task {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\tmockTask.EXPECT().GetTaskID().Return(t.GetTaskID())\n\t\t\t\treturn mockTask\n\t\t\t}\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockLogger := log.NewNoop()\n\t\t\tmockMetricsScope := metrics.NoopScope\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\n\t\t\t// Create mock slice\n\t\t\tmockSlice := NewMockVirtualSlice(ctrl)\n\n\t\t\t// Create virtual queues map with mocks\n\t\t\tvirtualQueues := make(map[int64]VirtualQueue)\n\t\t\tmockQueues := make(map[int64]*MockVirtualQueue)\n\t\t\tfor queueID := range tt.initialQueues {\n\t\t\t\tmockQueue := NewMockVirtualQueue(ctrl)\n\t\t\t\tmockQueues[queueID] = mockQueue\n\t\t\t\tvirtualQueues[queueID] = mockQueue\n\t\t\t}\n\n\t\t\t// Set up mock expectations\n\t\t\ttt.setupMockQueues(mockQueues, mockSlice)\n\n\t\t\tforceNewSliceDuration := time.Minute\n\t\t\t// Create manager instance\n\t\t\tmanager := &virtualQueueManagerImpl{\n\t\t\t\tprocessor:       mockProcessor,\n\t\t\t\ttaskInitializer: mockTaskInitializer,\n\t\t\t\trescheduler:     mockRescheduler,\n\t\t\t\tqueueReader:     mockQueueReader,\n\t\t\t\tlogger:          mockLogger,\n\t\t\t\tmetricsScope:    mockMetricsScope,\n\t\t\t\ttimeSource:      mockTimeSource,\n\t\t\t\tqueueManagerOptions: &VirtualQueueManagerOptions{\n\t\t\t\t\tRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\t\t\tPageSize: dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\t},\n\t\t\t\t\tNonRootQueueOptions: &VirtualQueueOptions{\n\t\t\t\t\t\tPageSize: dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\t},\n\t\t\t\t\tVirtualSliceForceAppendInterval: dynamicproperties.GetDurationPropertyFn(forceNewSliceDuration),\n\t\t\t\t},\n\t\t\t\tstatus:        common.DaemonStatusInitialized,\n\t\t\t\tvirtualQueues: virtualQueues,\n\t\t\t\tcreateVirtualQueueFn: func(queueID int64, s ...VirtualSlice) VirtualQueue {\n\t\t\t\t\tvq := NewMockVirtualQueue(ctrl)\n\t\t\t\t\tvq.EXPECT().Start()\n\t\t\t\t\treturn vq\n\t\t\t\t},\n\t\t\t\tnextForceNewSliceTime: mockTimeSource.Now(),\n\t\t\t}\n\n\t\t\t// Execute test\n\t\t\tif tt.appendSlice {\n\t\t\t\tmockTimeSource.Advance(forceNewSliceDuration)\n\t\t\t}\n\t\t\tmanager.AddNewVirtualSliceToRootQueue(mockSlice)\n\n\t\t\t// Verify results\n\t\t\ttt.verifyQueues(t, manager.virtualQueues)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: VirtualQueue)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination virtual_queue_mock.go github.com/uber/cadence/service/history/queuev2 VirtualQueue\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockVirtualQueue is a mock of VirtualQueue interface.\ntype MockVirtualQueue struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVirtualQueueMockRecorder\n\tisgomock struct{}\n}\n\n// MockVirtualQueueMockRecorder is the mock recorder for MockVirtualQueue.\ntype MockVirtualQueueMockRecorder struct {\n\tmock *MockVirtualQueue\n}\n\n// NewMockVirtualQueue creates a new mock instance.\nfunc NewMockVirtualQueue(ctrl *gomock.Controller) *MockVirtualQueue {\n\tmock := &MockVirtualQueue{ctrl: ctrl}\n\tmock.recorder = &MockVirtualQueueMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVirtualQueue) EXPECT() *MockVirtualQueueMockRecorder {\n\treturn m.recorder\n}\n\n// AppendSlices mocks base method.\nfunc (m *MockVirtualQueue) AppendSlices(arg0 ...VirtualSlice) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"AppendSlices\", varargs...)\n}\n\n// AppendSlices indicates an expected call of AppendSlices.\nfunc (mr *MockVirtualQueueMockRecorder) AppendSlices(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AppendSlices\", reflect.TypeOf((*MockVirtualQueue)(nil).AppendSlices), arg0...)\n}\n\n// ClearSlices mocks base method.\nfunc (m *MockVirtualQueue) ClearSlices(arg0 func(VirtualSlice) bool) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"ClearSlices\", arg0)\n}\n\n// ClearSlices indicates an expected call of ClearSlices.\nfunc (mr *MockVirtualQueueMockRecorder) ClearSlices(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ClearSlices\", reflect.TypeOf((*MockVirtualQueue)(nil).ClearSlices), arg0)\n}\n\n// GetState mocks base method.\nfunc (m *MockVirtualQueue) GetState() []VirtualSliceState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetState\")\n\tret0, _ := ret[0].([]VirtualSliceState)\n\treturn ret0\n}\n\n// GetState indicates an expected call of GetState.\nfunc (mr *MockVirtualQueueMockRecorder) GetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetState\", reflect.TypeOf((*MockVirtualQueue)(nil).GetState))\n}\n\n// IterateSlices mocks base method.\nfunc (m *MockVirtualQueue) IterateSlices(arg0 func(VirtualSlice)) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"IterateSlices\", arg0)\n}\n\n// IterateSlices indicates an expected call of IterateSlices.\nfunc (mr *MockVirtualQueueMockRecorder) IterateSlices(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IterateSlices\", reflect.TypeOf((*MockVirtualQueue)(nil).IterateSlices), arg0)\n}\n\n// MergeSlices mocks base method.\nfunc (m *MockVirtualQueue) MergeSlices(arg0 ...VirtualSlice) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{}\n\tfor _, a := range arg0 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tm.ctrl.Call(m, \"MergeSlices\", varargs...)\n}\n\n// MergeSlices indicates an expected call of MergeSlices.\nfunc (mr *MockVirtualQueueMockRecorder) MergeSlices(arg0 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeSlices\", reflect.TypeOf((*MockVirtualQueue)(nil).MergeSlices), arg0...)\n}\n\n// MergeWithLastSlice mocks base method.\nfunc (m *MockVirtualQueue) MergeWithLastSlice(arg0 VirtualSlice) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"MergeWithLastSlice\", arg0)\n}\n\n// MergeWithLastSlice indicates an expected call of MergeWithLastSlice.\nfunc (mr *MockVirtualQueueMockRecorder) MergeWithLastSlice(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MergeWithLastSlice\", reflect.TypeOf((*MockVirtualQueue)(nil).MergeWithLastSlice), arg0)\n}\n\n// Pause mocks base method.\nfunc (m *MockVirtualQueue) Pause(arg0 time.Duration) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Pause\", arg0)\n}\n\n// Pause indicates an expected call of Pause.\nfunc (mr *MockVirtualQueueMockRecorder) Pause(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Pause\", reflect.TypeOf((*MockVirtualQueue)(nil).Pause), arg0)\n}\n\n// SplitSlices mocks base method.\nfunc (m *MockVirtualQueue) SplitSlices(arg0 func(VirtualSlice) ([]VirtualSlice, bool)) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SplitSlices\", arg0)\n}\n\n// SplitSlices indicates an expected call of SplitSlices.\nfunc (mr *MockVirtualQueueMockRecorder) SplitSlices(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SplitSlices\", reflect.TypeOf((*MockVirtualQueue)(nil).SplitSlices), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockVirtualQueue) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockVirtualQueueMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockVirtualQueue)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockVirtualQueue) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockVirtualQueueMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockVirtualQueue)(nil).Stop))\n}\n\n// UpdateAndGetState mocks base method.\nfunc (m *MockVirtualQueue) UpdateAndGetState() []VirtualSliceState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAndGetState\")\n\tret0, _ := ret[0].([]VirtualSliceState)\n\treturn ret0\n}\n\n// UpdateAndGetState indicates an expected call of UpdateAndGetState.\nfunc (mr *MockVirtualQueueMockRecorder) UpdateAndGetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAndGetState\", reflect.TypeOf((*MockVirtualQueue)(nil).UpdateAndGetState))\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_queue_test.go",
    "content": "package queuev2\n\nimport (\n\t\"container/list\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestVirtualQueueImpl_GetState(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\n\tmockVirtualSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t)\n\n\tstates := queue.GetState()\n\texpectedStates := []VirtualSliceState{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tassert.Equal(t, expectedStates, states)\n}\n\nfunc TestVirtualQueueImpl_UpdateAndGetState(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\n\tmockVirtualSlice1.EXPECT().UpdateAndGetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice1.EXPECT().GetPendingTaskCount().Return(1)\n\tmockVirtualSlice1.EXPECT().IsEmpty().Return(false)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice1, 1)\n\n\tmockVirtualSlice2.EXPECT().UpdateAndGetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice2.EXPECT().IsEmpty().Return(true)\n\tmockMonitor.EXPECT().RemoveSlice(mockVirtualSlice2)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t)\n\n\tstates := queue.UpdateAndGetState()\n\texpectedStates := []VirtualSliceState{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tassert.Equal(t, expectedStates, states)\n}\n\nfunc TestVirtualQueue_MergeSlices(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor)\n\t\texpectedStates []VirtualSliceState\n\t}{\n\t\t{\n\t\t\tname: \"Merge non-overlapping slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\texistingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().HasMoreTasks().Return(true)\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{}, false)\n\t\t\t\texistingSlice.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(existingSlice, 1)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().TryMergeWithVirtualSlice(existingSlice2).Return([]VirtualSlice{}, false)\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(2)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 2)\n\n\t\t\t\texistingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice2.EXPECT().GetPendingTaskCount().Return(3)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(existingSlice2, 3)\n\n\t\t\t\treturn []VirtualSlice{existingSlice, existingSlice2}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Merge overlapping slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\texistingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(gomock.Any()).Return([]VirtualSlice{mergedSlice}, true)\n\t\t\t\texistingSlice.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(existingSlice, 1)\n\t\t\t\tmergedSlice.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice, 1)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice.EXPECT().TryMergeWithVirtualSlice(existingSlice2).Return([]VirtualSlice{mergedSlice2}, true)\n\t\t\t\tmergedSlice2.EXPECT().GetPendingTaskCount().Return(2)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice2, 2)\n\n\t\t\t\texistingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice2.EXPECT().HasMoreTasks().Return(true).AnyTimes()\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice2)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(mergedSlice)\n\n\t\t\t\treturn []VirtualSlice{existingSlice, existingSlice2}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Merge empty queue with new slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 1)\n\t\t\t\tincomingSlice.EXPECT().HasMoreTasks().Return(true).AnyTimes()\n\n\t\t\t\treturn []VirtualSlice{}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockMetricsScope := metrics.NoopScope\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\t\t\texistingSlices, incomingSlices, monitor := tt.setupMocks(ctrl)\n\n\t\t\tqueue := NewVirtualQueue(\n\t\t\t\tmockProcessor,\n\t\t\t\tmockRescheduler,\n\t\t\t\tmockLogger,\n\t\t\t\tmockMetricsScope,\n\t\t\t\tmockTimeSource,\n\t\t\t\tmockRateLimiter,\n\t\t\t\tmonitor,\n\t\t\t\texistingSlices,\n\t\t\t\t&VirtualQueueOptions{\n\t\t\t\t\tPageSize:                             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t\t\t},\n\t\t\t)\n\n\t\t\tqueue.MergeSlices(incomingSlices...)\n\t\t\tstates := queue.GetState()\n\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\t\t})\n\t}\n}\n\nfunc TestAppendOrMergeSlice(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(ctrl *gomock.Controller) (VirtualSlice, VirtualSlice, Monitor)\n\t\texpectedStates []VirtualSliceState\n\t}{\n\t\t{\n\t\t\tname: \"Append when no merge possible\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{}, false)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(2)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 2)\n\n\t\t\t\treturn existingSlice, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Merge when slices overlap\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{mergedSlice}, true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmergedSlice.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice, 1)\n\n\t\t\t\treturn existingSlice, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Append to empty list\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(10)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 10)\n\n\t\t\t\treturn nil, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with multiple resulting slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) (VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{mergedSlice1, mergedSlice2}, true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmergedSlice1.EXPECT().GetPendingTaskCount().Return(1)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice1, 1)\n\t\t\t\tmergedSlice2.EXPECT().GetPendingTaskCount().Return(2)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice2, 2)\n\n\t\t\t\treturn existingSlice, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\texistingSlice, incomingSlice, monitor := tt.setupMocks(ctrl)\n\t\t\tslices := list.New()\n\n\t\t\tif existingSlice != nil {\n\t\t\t\tslices.PushBack(existingSlice)\n\t\t\t}\n\n\t\t\tvirtualQueue := &virtualQueueImpl{\n\t\t\t\tvirtualSlices: slices,\n\t\t\t\tmonitor:       monitor,\n\t\t\t}\n\n\t\t\tvirtualQueue.appendOrMergeSlice(slices, incomingSlice)\n\n\t\t\t// Convert list to slice of states for comparison\n\t\t\tvar states []VirtualSliceState\n\t\t\tfor e := slices.Front(); e != nil; e = e.Next() {\n\t\t\t\tstates = append(states, e.Value.(VirtualSlice).GetState())\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\t\t})\n\t}\n}\n\nfunc TestVirtualQueue_LoadAndSubmitTasks(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\tmockRateLimiter.EXPECT().Wait(gomock.Any()).Return(nil).AnyTimes()\n\tmockMonitor := NewMockMonitor(ctrl)\n\tmockPauseController := NewMockPauseController(ctrl)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\n\tmockTask1 := task.NewMockTask(ctrl)\n\tmockTask1.EXPECT().GetDomainID().Return(\"some random domainID\")\n\tmockTask1.EXPECT().GetWorkflowID().Return(\"some random workflowID\")\n\tmockTask1.EXPECT().GetRunID().Return(\"some random runID\")\n\tmockTask1.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(mockTimeSource.Now().Add(time.Second*-1), 1))\n\tmockTask1.EXPECT().GetVisibilityTimestamp().Return(mockTimeSource.Now().Add(time.Second * -1))\n\tmockTask1.EXPECT().SetInitialSubmitTime(gomock.Any()).Times(1)\n\tmockTask2 := task.NewMockTask(ctrl)\n\tmockTask2.EXPECT().GetDomainID().Return(\"some random domainID\")\n\tmockTask2.EXPECT().GetWorkflowID().Return(\"some random workflowID\")\n\tmockTask2.EXPECT().GetRunID().Return(\"some random runID\")\n\tmockTask2.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(mockTimeSource.Now().Add(time.Second*1), 2))\n\tmockTask3 := task.NewMockTask(ctrl)\n\tmockTask3.EXPECT().GetDomainID().Return(\"some random domainID\")\n\tmockTask3.EXPECT().GetWorkflowID().Return(\"some random workflowID\")\n\tmockTask3.EXPECT().GetRunID().Return(\"some random runID\")\n\tmockTask3.EXPECT().GetTaskKey().Return(persistence.NewHistoryTaskKey(mockTimeSource.Now().Add(time.Second*-1), 1))\n\tmockTask3.EXPECT().GetVisibilityTimestamp().Return(mockTimeSource.Now().Add(time.Second * -1))\n\tmockTask3.EXPECT().SetInitialSubmitTime(gomock.Any()).Times(1)\n\n\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(0)\n\tmockPauseController.EXPECT().IsPaused().Return(false)\n\tmockVirtualSlice1.EXPECT().GetTasks(gomock.Any(), 10).Return([]task.Task{mockTask1, mockTask2}, nil)\n\tmockVirtualSlice1.EXPECT().GetPendingTaskCount().Return(2)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice1, 2)\n\tmockVirtualSlice1.EXPECT().HasMoreTasks().Return(false)\n\n\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(0)\n\tmockPauseController.EXPECT().IsPaused().Return(false)\n\tmockVirtualSlice2.EXPECT().GetTasks(gomock.Any(), 10).Return([]task.Task{mockTask3}, nil)\n\tmockVirtualSlice2.EXPECT().HasMoreTasks().Return(false)\n\tmockVirtualSlice2.EXPECT().GetPendingTaskCount().Return(1)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice2, 1)\n\tmockProcessor.EXPECT().TrySubmit(mockTask3).Return(false, nil)\n\n\tmockProcessor.EXPECT().TrySubmit(mockTask1).Return(true, nil)\n\tmockRescheduler.EXPECT().RescheduleTask(mockTask2, mockTimeSource.Now().Add(time.Second*1))\n\tmockRescheduler.EXPECT().RescheduleTask(mockTask3, mockTimeSource.Now().Add(taskSchedulerThrottleBackoffInterval))\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t).(*virtualQueueImpl)\n\n\tqueue.pauseController = mockPauseController\n\n\tqueue.loadAndSubmitTasks()\n\n\tqueue.loadAndSubmitTasks()\n\n\tassert.Nil(t, queue.sliceToRead)\n}\n\nfunc TestVirtualQueue_LifeCycle(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\tmockRateLimiter.EXPECT().Wait(gomock.Any()).Return(nil).AnyTimes()\n\tmockMonitor := NewMockMonitor(ctrl)\n\tmockPauseController := NewMockPauseController(ctrl)\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t}\n\n\tmockVirtualSlice1.EXPECT().GetTasks(gomock.Any(), 10).Return([]task.Task{}, nil).MaxTimes(1)\n\tmockVirtualSlice1.EXPECT().HasMoreTasks().Return(false).MaxTimes(1)\n\tmockVirtualSlice1.EXPECT().GetPendingTaskCount().Return(0).MaxTimes(1)\n\tmockVirtualSlice1.EXPECT().Clear().Times(1)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice1, 0).MaxTimes(1)\n\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(0).MaxTimes(1)\n\n\tmockPauseController.EXPECT().Subscribe(gomock.Any(), gomock.Any()).Times(1)\n\tmockPauseController.EXPECT().IsPaused().Return(false).AnyTimes()\n\tmockPauseController.EXPECT().Unsubscribe(gomock.Any()).Times(1)\n\tmockPauseController.EXPECT().Stop().Times(1)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t).(*virtualQueueImpl)\n\n\tqueue.pauseController = mockPauseController\n\n\tqueue.Start()\n\tqueue.Stop()\n}\n\nfunc TestVirtualQueue_LifeCycle_Pause(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\tmockRateLimiter.EXPECT().Wait(gomock.Any()).Return(nil).AnyTimes()\n\tmockMonitor := NewMockMonitor(ctrl)\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t}\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t).(*virtualQueueImpl)\n\n\tgomock.InOrder(\n\t\t// first time we call loadAndSubmitTasks, we should pause, so set the total pending task count to be larger than MaxPendingTasksCount\n\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(101).Times(1),\n\t\t// then we should resume from pause and load the tasks\n\t\t// to simplify the test, we just assume that there is no more tasks to load\n\t\tmockMonitor.EXPECT().GetTotalPendingTaskCount().Return(0).MaxTimes(1),\n\t\tmockVirtualSlice1.EXPECT().GetTasks(gomock.Any(), 10).Return([]task.Task{}, nil).MaxTimes(1),\n\t\tmockVirtualSlice1.EXPECT().GetPendingTaskCount().Return(0).MaxTimes(1),\n\t\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice1, 0).MaxTimes(1),\n\t\tmockVirtualSlice1.EXPECT().HasMoreTasks().Return(false).MaxTimes(1),\n\t\tmockVirtualSlice1.EXPECT().Clear().Times(1),\n\t)\n\n\tqueue.Start()\n\n\t// wait for the pause controller to resume\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(time.Second * 10)\n\n\tqueue.Stop()\n}\n\nfunc TestVirtualQueue_IterateSlices(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\n\tmockVirtualSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\tmockMonitor := NewMockMonitor(ctrl)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t)\n\n\tstates := []VirtualSliceState{}\n\tqueue.IterateSlices(func(slice VirtualSlice) {\n\t\tstates = append(states, slice.GetState())\n\t})\n\n\texpectedStates := []VirtualSliceState{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tassert.Equal(t, expectedStates, states)\n}\n\nfunc TestVirtualQueue_ClearSlices(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\tmockMonitor := NewMockMonitor(ctrl)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\n\tmockVirtualSlice1.EXPECT().Clear().Times(1)\n\tmockVirtualSlice1.EXPECT().GetPendingTaskCount().Return(0).Times(1)\n\tmockVirtualSlice1.EXPECT().HasMoreTasks().Return(false).Times(1)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice1, 0).Times(1)\n\tmockVirtualSlice2.EXPECT().Clear().Times(1)\n\tmockVirtualSlice2.EXPECT().GetPendingTaskCount().Return(0).Times(1)\n\tmockVirtualSlice2.EXPECT().HasMoreTasks().Return(true).Times(1)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice2, 0).Times(1)\n\tmockVirtualSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t)\n\n\tqueue.ClearSlices(func(slice VirtualSlice) bool {\n\t\treturn true\n\t})\n\n\tstates := queue.GetState()\n\texpectedStates := []VirtualSliceState{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tassert.Equal(t, expectedStates, states)\n\tassert.Equal(t, mockVirtualSlice2, queue.(*virtualQueueImpl).sliceToRead.Value.(VirtualSlice))\n}\n\nfunc TestVirtualQueue_SplitSlices(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockProcessor := task.NewMockProcessor(ctrl)\n\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\tmockLogger := testlogger.New(t)\n\tmockMetricsScope := metrics.NoopScope\n\tmockPageSize := dynamicproperties.GetIntPropertyFn(10)\n\n\tmockVirtualSlice1 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice2 := NewMockVirtualSlice(ctrl)\n\tmockVirtualSlice3 := NewMockVirtualSlice(ctrl)\n\n\tmockVirtualSlices := []VirtualSlice{\n\t\tmockVirtualSlice1,\n\t\tmockVirtualSlice2,\n\t}\n\tmockMonitor := NewMockMonitor(ctrl)\n\n\tmockVirtualSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice1.EXPECT().HasMoreTasks().Return(true).Times(1)\n\tmockMonitor.EXPECT().RemoveSlice(mockVirtualSlice2).Times(1)\n\tmockVirtualSlice3.EXPECT().GetState().Return(VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t},\n\t\tPredicate: NewUniversalPredicate(),\n\t})\n\tmockVirtualSlice3.EXPECT().GetPendingTaskCount().Return(0).Times(1)\n\tmockMonitor.EXPECT().SetSlicePendingTaskCount(mockVirtualSlice3, 0).Times(1)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\tqueue := NewVirtualQueue(\n\t\tmockProcessor,\n\t\tmockRescheduler,\n\t\tmockLogger,\n\t\tmockMetricsScope,\n\t\tmockTimeSource,\n\t\tmockRateLimiter,\n\t\tmockMonitor,\n\t\tmockVirtualSlices,\n\t\t&VirtualQueueOptions{\n\t\t\tPageSize:                             mockPageSize,\n\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t},\n\t)\n\n\tqueue.SplitSlices(func(slice VirtualSlice) (remaining []VirtualSlice, split bool) {\n\t\tif slice == mockVirtualSlice1 {\n\t\t\treturn nil, false\n\t\t} else if slice == mockVirtualSlice2 {\n\t\t\treturn []VirtualSlice{mockVirtualSlice3}, true\n\t\t}\n\t\treturn nil, false\n\t})\n\n\tstates := queue.GetState()\n\texpectedStates := []VirtualSliceState{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t},\n\t\t\tPredicate: NewUniversalPredicate(),\n\t\t},\n\t}\n\tassert.Equal(t, expectedStates, states)\n\tassert.Equal(t, mockVirtualSlice1, queue.(*virtualQueueImpl).sliceToRead.Value.(VirtualSlice))\n}\n\nfunc TestVirtualQueue_AppendSlices(t *testing.T) {\n\ttests := []struct {\n\t\tname                   string\n\t\tsetupMocks             func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor)\n\t\texpectedStates         []VirtualSliceState\n\t\texpectedSliceToReadIdx *int // nil if sliceToRead should be nil, otherwise index of expected slice\n\t}{\n\t\t{\n\t\t\tname: \"Append empty slice list\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, []VirtualSlice{}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0),\n\t\t},\n\t\t{\n\t\t\tname: \"Append single slice to empty queue\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\tincomingSlice.EXPECT().HasMoreTasks().Return(true).Times(1)\n\n\t\t\t\treturn []VirtualSlice{}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0),\n\t\t},\n\t\t{\n\t\t\tname: \"Append multiple slices to empty queue\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\tincomingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\tincomingSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\tincomingSlice1.EXPECT().HasMoreTasks().Return(true).Times(1)\n\n\t\t\t\tincomingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\n\t\t\t\treturn []VirtualSlice{}, []VirtualSlice{incomingSlice1, incomingSlice2}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0),\n\t\t},\n\t\t{\n\t\t\tname: \"Append slices to queue with existing slices\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\texistingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\texistingSlice1.EXPECT().HasMoreTasks().Return(false).Times(1)\n\n\t\t\t\texistingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\texistingSlice2.EXPECT().HasMoreTasks().Return(true).Times(1)\n\n\t\t\t\tincomingSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(21),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\n\t\t\t\tincomingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(31),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(40),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\n\t\t\t\treturn []VirtualSlice{existingSlice1, existingSlice2}, []VirtualSlice{incomingSlice1, incomingSlice2}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(21),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(31),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(40),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(1),\n\t\t},\n\t\t{\n\t\t\tname: \"Append slices when no existing slice has more tasks\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\texistingSlice.EXPECT().HasMoreTasks().Return(false).Times(1)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\tincomingSlice.EXPECT().HasMoreTasks().Return(true).Times(1)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(1),\n\t\t},\n\t\t{\n\t\t\tname: \"Append slices when all slices have no more tasks\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, []VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\texistingSlice.EXPECT().HasMoreTasks().Return(false).Times(1)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).Times(1)\n\t\t\t\tincomingSlice.EXPECT().HasMoreTasks().Return(false).Times(1)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, []VirtualSlice{incomingSlice}, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(11),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockMetricsScope := metrics.NoopScope\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\t\t\texistingSlices, incomingSlices, monitor := tt.setupMocks(ctrl)\n\n\t\t\tqueue := NewVirtualQueue(\n\t\t\t\tmockProcessor,\n\t\t\t\tmockRescheduler,\n\t\t\t\tmockLogger,\n\t\t\t\tmockMetricsScope,\n\t\t\t\tmockTimeSource,\n\t\t\t\tmockRateLimiter,\n\t\t\t\tmonitor,\n\t\t\t\texistingSlices,\n\t\t\t\t&VirtualQueueOptions{\n\t\t\t\t\tPageSize:                             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t\t\t},\n\t\t\t)\n\n\t\t\tqueue.AppendSlices(incomingSlices...)\n\t\t\tstates := queue.GetState()\n\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\n\t\t\t// Check sliceToRead\n\t\t\tqueueImpl := queue.(*virtualQueueImpl)\n\t\t\tif tt.expectedSliceToReadIdx == nil {\n\t\t\t\tassert.Nil(t, queueImpl.sliceToRead, \"sliceToRead should be nil\")\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, queueImpl.sliceToRead, \"sliceToRead should not be nil\")\n\t\t\t\tif queueImpl.sliceToRead != nil {\n\t\t\t\t\texpectedSlice := append(existingSlices, incomingSlices...)[*tt.expectedSliceToReadIdx]\n\t\t\t\t\tassert.Equal(t, expectedSlice, queueImpl.sliceToRead.Value.(VirtualSlice))\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestVirtualQueue_MergeWithLastSlice(t *testing.T) {\n\ttests := []struct {\n\t\tname                   string\n\t\tsetupMocks             func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor)\n\t\texpectedStates         []VirtualSliceState\n\t\texpectedSliceToReadIdx *int // nil if sliceToRead should be nil, otherwise index of expected slice\n\t}{\n\t\t{\n\t\t\tname: \"Merge with empty queue\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(5)\n\t\t\t\tincomingSlice.EXPECT().HasMoreTasks().Return(true)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 5)\n\n\t\t\t\treturn []VirtualSlice{}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0),\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with last slice - no merge possible\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\texistingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice1.EXPECT().HasMoreTasks().Return(false)\n\n\t\t\t\texistingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice2.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{}, false)\n\t\t\t\texistingSlice2.EXPECT().HasMoreTasks().Return(true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(3)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 3)\n\n\t\t\t\treturn []VirtualSlice{existingSlice1, existingSlice2}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(1), // existingSlice1 has more tasks\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with last slice - successful merge\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\texistingSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice1.EXPECT().HasMoreTasks().Return(true)\n\n\t\t\t\texistingSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice2.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{mergedSlice}, true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice.EXPECT().GetPendingTaskCount().Return(7)\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice2)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice, 7)\n\n\t\t\t\treturn []VirtualSlice{existingSlice1, existingSlice2}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0), // existingSlice1 has more tasks\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with last slice - multiple merged slices result\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice1 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice2 := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{mergedSlice1, mergedSlice2}, true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice1.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice1.EXPECT().GetPendingTaskCount().Return(4)\n\t\t\t\tmergedSlice1.EXPECT().HasMoreTasks().Return(true)\n\n\t\t\t\tmergedSlice2.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice2.EXPECT().GetPendingTaskCount().Return(6)\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice1, 4)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice2, 6)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0), // mergedSlice1 has more tasks\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with single existing slice - no merge possible\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{}, false)\n\t\t\t\texistingSlice.EXPECT().HasMoreTasks().Return(true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tincomingSlice.EXPECT().GetPendingTaskCount().Return(2)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(incomingSlice, 2)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(15),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0), // existingSlice has more tasks\n\t\t},\n\t\t{\n\t\t\tname: \"Merge with single existing slice - successful merge\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) ([]VirtualSlice, VirtualSlice, Monitor) {\n\t\t\t\texistingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tincomingSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmergedSlice := NewMockVirtualSlice(ctrl)\n\t\t\t\tmonitor := NewMockMonitor(ctrl)\n\n\t\t\t\texistingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\texistingSlice.EXPECT().TryMergeWithVirtualSlice(incomingSlice).Return([]VirtualSlice{mergedSlice}, true)\n\n\t\t\t\tincomingSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\n\t\t\t\tmergedSlice.EXPECT().GetState().Return(VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t}).AnyTimes()\n\t\t\t\tmergedSlice.EXPECT().GetPendingTaskCount().Return(5)\n\t\t\t\tmergedSlice.EXPECT().HasMoreTasks().Return(true)\n\n\t\t\t\tmonitor.EXPECT().RemoveSlice(existingSlice)\n\t\t\t\tmonitor.EXPECT().RemoveSlice(incomingSlice)\n\t\t\t\tmonitor.EXPECT().SetSlicePendingTaskCount(mergedSlice, 5)\n\n\t\t\t\treturn []VirtualSlice{existingSlice}, incomingSlice, monitor\n\t\t\t},\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedSliceToReadIdx: common.Ptr(0), // mergedSlice has more tasks\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockProcessor := task.NewMockProcessor(ctrl)\n\t\t\tmockRescheduler := task.NewMockRescheduler(ctrl)\n\t\t\tmockLogger := testlogger.New(t)\n\t\t\tmockMetricsScope := metrics.NoopScope\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\tmockRateLimiter := quotas.NewMockLimiter(ctrl)\n\n\t\t\texistingSlices, incomingSlice, monitor := tt.setupMocks(ctrl)\n\n\t\t\tqueue := NewVirtualQueue(\n\t\t\t\tmockProcessor,\n\t\t\t\tmockRescheduler,\n\t\t\t\tmockLogger,\n\t\t\t\tmockMetricsScope,\n\t\t\t\tmockTimeSource,\n\t\t\t\tmockRateLimiter,\n\t\t\t\tmonitor,\n\t\t\t\texistingSlices,\n\t\t\t\t&VirtualQueueOptions{\n\t\t\t\t\tPageSize:                             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\t\tMaxPendingTasksCount:                 dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\t\tPollBackoffInterval:                  dynamicproperties.GetDurationPropertyFn(time.Second * 10),\n\t\t\t\t\tPollBackoffIntervalJitterCoefficient: dynamicproperties.GetFloatPropertyFn(0.0),\n\t\t\t\t},\n\t\t\t)\n\n\t\t\tqueue.MergeWithLastSlice(incomingSlice)\n\t\t\tstates := queue.GetState()\n\n\t\t\tassert.Equal(t, tt.expectedStates, states)\n\n\t\t\t// Check sliceToRead\n\t\t\tqueueImpl := queue.(*virtualQueueImpl)\n\t\t\tif tt.expectedSliceToReadIdx == nil {\n\t\t\t\tassert.Nil(t, queueImpl.sliceToRead, \"sliceToRead should be nil\")\n\t\t\t} else {\n\t\t\t\tassert.NotNil(t, queueImpl.sliceToRead, \"sliceToRead should not be nil\")\n\t\t\t\tif queueImpl.sliceToRead != nil {\n\t\t\t\t\t// For MergeWithLastSlice, we need to build the final slice list differently\n\t\t\t\t\t// since the incoming slice is merged/appended to the existing slices\n\t\t\t\t\tvar finalSlices []VirtualSlice\n\t\t\t\t\tfor _, state := range states {\n\t\t\t\t\t\t// Find the slice that matches this state\n\t\t\t\t\t\tfor e := queueImpl.virtualSlices.Front(); e != nil; e = e.Next() {\n\t\t\t\t\t\t\tslice := e.Value.(VirtualSlice)\n\t\t\t\t\t\t\tif slice.GetState() == state {\n\t\t\t\t\t\t\t\tfinalSlices = append(finalSlices, slice)\n\t\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\texpectedSlice := finalSlices[*tt.expectedSliceToReadIdx]\n\t\t\t\t\tassert.Equal(t, expectedSlice, queueImpl.sliceToRead.Value.(VirtualSlice))\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_slice.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination virtual_slice_mock.go github.com/uber/cadence/service/history/queuev2 VirtualSlice\npackage queuev2\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\ntype (\n\tVirtualSlice interface {\n\t\tGetState() VirtualSliceState\n\t\tIsEmpty() bool\n\t\tGetTasks(context.Context, int) ([]task.Task, error)\n\t\tHasMoreTasks() bool\n\t\tUpdateAndGetState() VirtualSliceState\n\t\tGetPendingTaskCount() int\n\t\tClear()\n\t\tPendingTaskStats() PendingTaskStats\n\n\t\tTrySplitByTaskKey(persistence.HistoryTaskKey) (VirtualSlice, VirtualSlice, bool)\n\t\tTrySplitByPredicate(Predicate) (VirtualSlice, VirtualSlice, bool)\n\t\tTryMergeWithVirtualSlice(VirtualSlice) ([]VirtualSlice, bool)\n\t}\n\n\tPendingTaskStats struct {\n\t\tPendingTaskCountPerDomain map[string]int\n\t}\n\n\tvirtualSliceImpl struct {\n\t\tstate              VirtualSliceState\n\t\ttaskInitializer    task.Initializer\n\t\tqueueReader        QueueReader\n\t\tpendingTaskTracker PendingTaskTracker\n\t\tlogger             log.Logger\n\n\t\t// progress tracks the read progress of the slice, sorted by the inclusive min task key of the range, ranges are not overlapping\n\t\t// For a virtual slice, the progress is a task key pointing to the next task to read and the next page token\n\t\t// In most cases, there is only one GetTaskProgress item in the progress slice.\n\t\t// However, when 2 virtual slices are merged, the progress slice may contain 2 items, because even though the range of the slices have overlap,\n\t\t// their GetTaskProgress cannot be merged, because the next page tokens are coupled with the range of the original slices.\n\t\tprogress []*GetTaskProgress\n\t}\n)\n\nfunc NewVirtualSlice(\n\tstate VirtualSliceState,\n\ttaskInitializer task.Initializer,\n\tqueueReader QueueReader,\n\tpendingTaskTracker PendingTaskTracker,\n\tlogger log.Logger,\n) VirtualSlice {\n\treturn &virtualSliceImpl{\n\t\tstate:              state,\n\t\ttaskInitializer:    taskInitializer,\n\t\tqueueReader:        queueReader,\n\t\tpendingTaskTracker: pendingTaskTracker,\n\t\tlogger:             logger,\n\t\tprogress: []*GetTaskProgress{\n\t\t\t{\n\t\t\t\tRange:         state.Range,\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   state.Range.InclusiveMinTaskKey,\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc (s *virtualSliceImpl) GetState() VirtualSliceState {\n\treturn s.state\n}\n\nfunc (s *virtualSliceImpl) GetPendingTaskCount() int {\n\treturn s.pendingTaskTracker.GetPendingTaskCount()\n}\n\nfunc (s *virtualSliceImpl) IsEmpty() bool {\n\treturn s.state.IsEmpty() && !s.HasMoreTasks()\n}\n\nfunc (s *virtualSliceImpl) Clear() {\n\ts.UpdateAndGetState()\n\ts.pendingTaskTracker.Clear()\n\ts.progress = []*GetTaskProgress{\n\t\t{\n\t\t\tRange:         s.state.Range,\n\t\t\tNextPageToken: nil,\n\t\t\tNextTaskKey:   s.state.Range.InclusiveMinTaskKey,\n\t\t},\n\t}\n}\n\nfunc (s *virtualSliceImpl) GetTasks(ctx context.Context, pageSize int) ([]task.Task, error) {\n\tif len(s.progress) == 0 {\n\t\treturn nil, nil\n\t}\n\n\ttasks := make([]task.Task, 0, pageSize)\n\tfor len(tasks) < pageSize && len(s.progress) > 0 {\n\t\tresp, err := s.queueReader.GetTask(ctx, &GetTaskRequest{\n\t\t\tProgress:  s.progress[0],\n\t\t\tPredicate: s.state.Predicate,\n\t\t\tPageSize:  pageSize - len(tasks),\n\t\t})\n\t\tif err != nil {\n\t\t\t// NOTE: we must return the tasks here to let them either be submitted to scheduler or rescheduler\n\t\t\t// because they are already added to pending task tracker. Otherwise, they will become zombie tasks,\n\t\t\t// and won't be processed until shard restart.\n\t\t\t// The number of tasks returned here doesn't need to be the same as the page size even if there is still more tasks to read.\n\t\t\t// HasMoreTasks() method will still return true in this case.\n\t\t\tif len(tasks) > 0 {\n\t\t\t\treturn tasks, nil\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\n\t\tfor _, t := range resp.Tasks {\n\t\t\ttask := s.taskInitializer(t)\n\t\t\ttasks = append(tasks, task)\n\t\t\ts.pendingTaskTracker.AddTask(task)\n\t\t}\n\n\t\t// The persistence layer may return non-empty next page token even if there are no more tasks to read\n\t\t// We compare the next task key with the exclusive max task key to determine if there are more tasks to read instead\n\t\tif resp.Progress.NextTaskKey.Compare(s.progress[0].ExclusiveMaxTaskKey) < 0 {\n\t\t\ts.progress[0] = resp.Progress\n\t\t} else {\n\t\t\ts.progress = s.progress[1:]\n\t\t}\n\t}\n\n\treturn tasks, nil\n}\n\nfunc (s *virtualSliceImpl) HasMoreTasks() bool {\n\treturn len(s.progress) > 0\n}\n\nfunc (s *virtualSliceImpl) PendingTaskStats() PendingTaskStats {\n\treturn PendingTaskStats{\n\t\tPendingTaskCountPerDomain: s.pendingTaskTracker.GetPerDomainPendingTaskCount(),\n\t}\n}\n\nfunc (s *virtualSliceImpl) UpdateAndGetState() VirtualSliceState {\n\tprunedCount := s.pendingTaskTracker.PruneAckedTasks()\n\tnextTaskKey := s.state.Range.ExclusiveMaxTaskKey\n\tif len(s.progress) > 0 {\n\t\tnextTaskKey = s.progress[0].NextTaskKey\n\t}\n\ts.logger.Debug(\"pruned acked tasks\", tag.Counter(prunedCount), tag.Dynamic(\"inclusiveMinTaskKey\", s.state.Range.InclusiveMinTaskKey), tag.Dynamic(\"exclusiveMaxTaskKey\", s.state.Range.ExclusiveMaxTaskKey), tag.Dynamic(\"nextTaskKey\", nextTaskKey))\n\tminPendingTaskKey, ok := s.pendingTaskTracker.GetMinimumTaskKey()\n\tif !ok {\n\t\tif len(s.progress) > 0 { // no pending tasks, and there are more tasks to read\n\t\t\ts.state.Range.InclusiveMinTaskKey = s.progress[0].NextTaskKey\n\t\t} else { // no pending tasks, and no more tasks to read\n\t\t\ts.state.Range.InclusiveMinTaskKey = s.state.Range.ExclusiveMaxTaskKey\n\t\t}\n\t} else {\n\t\tif len(s.progress) > 0 { // there are pending tasks, and there are more tasks to read\n\t\t\ts.state.Range.InclusiveMinTaskKey = persistence.MinHistoryTaskKey(minPendingTaskKey, s.progress[0].NextTaskKey)\n\t\t} else { // there are pending tasks, and no more tasks to read\n\t\t\ts.state.Range.InclusiveMinTaskKey = minPendingTaskKey\n\t\t}\n\t}\n\treturn s.state\n}\n\nfunc (s *virtualSliceImpl) TrySplitByTaskKey(taskKey persistence.HistoryTaskKey) (VirtualSlice, VirtualSlice, bool) {\n\tleftState, rightState, ok := s.state.TrySplitByTaskKey(taskKey)\n\tif !ok {\n\t\treturn nil, nil, false\n\t}\n\n\tleftTracker := NewPendingTaskTracker()\n\trightTracker := NewPendingTaskTracker()\n\n\ttaskMap := s.pendingTaskTracker.GetTasks()\n\tfor taskKey, task := range taskMap {\n\t\tif leftState.Range.Contains(taskKey) {\n\t\t\tleftTracker.AddTask(task)\n\t\t} else {\n\t\t\trightTracker.AddTask(task)\n\t\t}\n\t}\n\n\tleftProgress := []*GetTaskProgress{}\n\trightProgress := []*GetTaskProgress{}\n\n\tfor _, progress := range s.progress {\n\t\tif leftState.Range.ContainsRange(progress.Range) {\n\t\t\tleftProgress = append(leftProgress, progress)\n\t\t\tcontinue\n\t\t}\n\t\tif rightState.Range.ContainsRange(progress.Range) {\n\t\t\trightProgress = append(rightProgress, progress)\n\t\t\tcontinue\n\t\t}\n\n\t\tif leftState.Range.Contains(progress.NextTaskKey) {\n\t\t\tleftProgress = append(leftProgress, &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: progress.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: leftState.Range.ExclusiveMaxTaskKey,\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   progress.NextTaskKey,\n\t\t\t})\n\t\t\trightProgress = append(rightProgress, &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: rightState.Range.InclusiveMinTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: progress.Range.ExclusiveMaxTaskKey,\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   rightState.Range.InclusiveMinTaskKey,\n\t\t\t})\n\t\t} else {\n\t\t\trightProgress = append(rightProgress, &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: progress.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: progress.Range.ExclusiveMaxTaskKey,\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   progress.NextTaskKey,\n\t\t\t})\n\t\t}\n\t}\n\n\tleftSlice := &virtualSliceImpl{\n\t\tstate:              leftState,\n\t\ttaskInitializer:    s.taskInitializer,\n\t\tqueueReader:        s.queueReader,\n\t\tpendingTaskTracker: leftTracker,\n\t\tprogress:           leftProgress,\n\t\tlogger:             s.logger,\n\t}\n\n\trightSlice := &virtualSliceImpl{\n\t\tstate:              rightState,\n\t\ttaskInitializer:    s.taskInitializer,\n\t\tqueueReader:        s.queueReader,\n\t\tpendingTaskTracker: rightTracker,\n\t\tprogress:           rightProgress,\n\t\tlogger:             s.logger,\n\t}\n\n\treturn leftSlice, rightSlice, true\n}\n\nfunc (s *virtualSliceImpl) TrySplitByPredicate(predicate Predicate) (VirtualSlice, VirtualSlice, bool) {\n\tpassState, failState, ok := s.state.TrySplitByPredicate(predicate)\n\tif !ok {\n\t\treturn nil, nil, false\n\t}\n\tpassTracker := NewPendingTaskTracker()\n\tfailTracker := NewPendingTaskTracker()\n\n\ttaskMap := s.pendingTaskTracker.GetTasks()\n\tfor _, task := range taskMap {\n\t\tif passState.Predicate.Check(task) {\n\t\t\tpassTracker.AddTask(task)\n\t\t} else {\n\t\t\tfailTracker.AddTask(task)\n\t\t}\n\t}\n\n\tpassProgress := make([]*GetTaskProgress, len(s.progress))\n\tfailProgress := make([]*GetTaskProgress, len(s.progress))\n\tcopy(passProgress, s.progress)\n\tcopy(failProgress, s.progress)\n\tpassSlice := &virtualSliceImpl{\n\t\tstate:              passState,\n\t\ttaskInitializer:    s.taskInitializer,\n\t\tqueueReader:        s.queueReader,\n\t\tpendingTaskTracker: passTracker,\n\t\tprogress:           passProgress,\n\t\tlogger:             s.logger,\n\t}\n\tfailSlice := &virtualSliceImpl{\n\t\tstate:              failState,\n\t\ttaskInitializer:    s.taskInitializer,\n\t\tqueueReader:        s.queueReader,\n\t\tpendingTaskTracker: failTracker,\n\t\tprogress:           failProgress,\n\t\tlogger:             s.logger,\n\t}\n\treturn passSlice, failSlice, true\n}\n\nfunc (s *virtualSliceImpl) TryMergeWithVirtualSlice(other VirtualSlice) ([]VirtualSlice, bool) {\n\totherImpl, ok := other.(*virtualSliceImpl)\n\tif !ok {\n\t\treturn nil, false\n\t}\n\n\tif s == other || !s.state.Range.CanMerge(other.GetState().Range) {\n\t\treturn nil, false\n\t}\n\n\tif s.state.Predicate.Equals(other.GetState().Predicate) {\n\t\treturn []VirtualSlice{mergeVirtualSlicesByRange(s, otherImpl)}, true\n\t}\n\treturn mergeVirtualSlicesWithDifferentPredicate(s, otherImpl)\n}\n\nfunc mergeVirtualSlicesByRange(left, right *virtualSliceImpl) VirtualSlice {\n\tif left.state.Range.InclusiveMinTaskKey.Compare(right.state.Range.InclusiveMinTaskKey) > 0 {\n\t\tleft, right = right, left\n\t}\n\tmergedState := VirtualSliceState{\n\t\tRange: Range{\n\t\t\tInclusiveMinTaskKey: left.state.Range.InclusiveMinTaskKey,\n\t\t\tExclusiveMaxTaskKey: persistence.MaxHistoryTaskKey(left.state.Range.ExclusiveMaxTaskKey, right.state.Range.ExclusiveMaxTaskKey),\n\t\t},\n\t\tPredicate: left.state.Predicate, // left and right have the same predicate\n\t}\n\tpendingTaskTracker := left.pendingTaskTracker\n\ttaskMap := right.pendingTaskTracker.GetTasks()\n\tfor _, task := range taskMap {\n\t\tpendingTaskTracker.AddTask(task)\n\t}\n\tmergedProgress := mergeGetTaskProgressWithSamePredicate(left.progress, right.progress)\n\n\treturn &virtualSliceImpl{\n\t\tstate:              mergedState,\n\t\ttaskInitializer:    left.taskInitializer,\n\t\tqueueReader:        left.queueReader,\n\t\tpendingTaskTracker: pendingTaskTracker,\n\t\tprogress:           mergedProgress,\n\t\tlogger:             left.logger,\n\t}\n}\n\nfunc mergeGetTaskProgressWithSamePredicate(left, right []*GetTaskProgress) []*GetTaskProgress {\n\tmergedProgress := []*GetTaskProgress{}\n\tleftIndex := 0\n\trightIndex := 0\n\tfor leftIndex < len(left) && rightIndex < len(right) {\n\t\tif left[leftIndex].Range.InclusiveMinTaskKey.Compare(right[rightIndex].Range.InclusiveMinTaskKey) <= 0 {\n\t\t\tmergedProgress = appendOrMergeProgressWithSamePredicate(mergedProgress, left[leftIndex])\n\t\t\tleftIndex++\n\t\t} else {\n\t\t\tmergedProgress = appendOrMergeProgressWithSamePredicate(mergedProgress, right[rightIndex])\n\t\t\trightIndex++\n\t\t}\n\t}\n\tfor leftIndex < len(left) {\n\t\tmergedProgress = appendOrMergeProgressWithSamePredicate(mergedProgress, left[leftIndex])\n\t\tleftIndex++\n\t}\n\tfor rightIndex < len(right) {\n\t\tmergedProgress = appendOrMergeProgressWithSamePredicate(mergedProgress, right[rightIndex])\n\t\trightIndex++\n\t}\n\treturn mergedProgress\n}\n\nfunc appendOrMergeProgressWithSamePredicate(mergedProgress []*GetTaskProgress, progress *GetTaskProgress) []*GetTaskProgress {\n\tif len(mergedProgress) == 0 {\n\t\treturn append(mergedProgress, progress)\n\t}\n\n\tlastProgress := mergedProgress[len(mergedProgress)-1]\n\tmergedProgress = mergedProgress[:len(mergedProgress)-1] // remove the last progress\n\treturn append(mergedProgress, mergeProgressWithSamePredicate(lastProgress, progress)...)\n}\n\n// mergeProgress merges two progress with the same predicate\n// Assuming the inclusive key, next key, max key of the 2 progress are [a, b, c] and [x, y, z] where a <= b <= c and x <= y <= z,\n// also assuming that a <= x, otherwise we can swap the left and right progress\n// There are 10 different cases regarding the order of a, b, c, x, y, z, and here are the cases and merged results:\n// [a, b, c, x, y, z] -> [b, b, c] and [y, y, z]\n// [a, b, x, c, y, z] -> [b, b, x] and [y, y, z]\n// [a, b, x, y, c, z] -> [b, b, x] and [y, y, z]\n// [a, b, x, y, z, c] -> [b, b, x] and [y, y, c]\n// [a, x, b, c, y ,z] -> [y, y, z]\n// [a, x, b, y, c, z] -> [y, y, z]\n// [a, x, b, y, z, c] -> [y, y, c]\n// [a, x, y, b, c, z] -> [b, b, z]\n// [a, x, y, b, z, c] -> [b, b, c]\n// [a, x, y, z, b, c] -> [b, b, c]\n// we only need to consider the range that hasn't been read yet, the merged result can be represented as\n// [b, b, min(c, x)], [max(b, y), max(b, y), max(c, z)], and if b >= x, [b, b, min(c, x)] will be an empty progress so it's omitted\nfunc mergeProgressWithSamePredicate(left, right *GetTaskProgress) []*GetTaskProgress {\n\tif left.Range.InclusiveMinTaskKey.Compare(right.Range.InclusiveMinTaskKey) > 0 {\n\t\tleft, right = right, left\n\t}\n\tif left.NextTaskKey.Compare(right.InclusiveMinTaskKey) < 0 {\n\t\treturn []*GetTaskProgress{\n\t\t\t{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: left.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MinHistoryTaskKey(left.ExclusiveMaxTaskKey, right.InclusiveMinTaskKey),\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   left.NextTaskKey,\n\t\t\t},\n\t\t\t{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: right.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.MaxHistoryTaskKey(left.ExclusiveMaxTaskKey, right.ExclusiveMaxTaskKey),\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   persistence.MaxHistoryTaskKey(left.NextTaskKey, right.NextTaskKey),\n\t\t\t},\n\t\t}\n\t}\n\treturn []*GetTaskProgress{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: persistence.MaxHistoryTaskKey(left.NextTaskKey, right.NextTaskKey),\n\t\t\t\tExclusiveMaxTaskKey: persistence.MaxHistoryTaskKey(left.ExclusiveMaxTaskKey, right.ExclusiveMaxTaskKey),\n\t\t\t},\n\t\t\tNextPageToken: nil,\n\t\t\tNextTaskKey:   persistence.MaxHistoryTaskKey(left.NextTaskKey, right.NextTaskKey),\n\t\t},\n\t}\n}\n\nfunc mergeGetTaskProgressWithDifferentPredicate(left, right []*GetTaskProgress) []*GetTaskProgress {\n\tmergedProgress := []*GetTaskProgress{}\n\tleftIndex := 0\n\trightIndex := 0\n\tfor leftIndex < len(left) && rightIndex < len(right) {\n\t\tif left[leftIndex].NextTaskKey.Compare(right[rightIndex].NextTaskKey) <= 0 {\n\t\t\tmergedProgress = appendOrMergeProgressWithDifferentPredicate(mergedProgress, left[leftIndex])\n\t\t\tleftIndex++\n\t\t} else {\n\t\t\tmergedProgress = appendOrMergeProgressWithDifferentPredicate(mergedProgress, right[rightIndex])\n\t\t\trightIndex++\n\t\t}\n\t}\n\tfor leftIndex < len(left) {\n\t\tmergedProgress = appendOrMergeProgressWithDifferentPredicate(mergedProgress, left[leftIndex])\n\t\tleftIndex++\n\t}\n\tfor rightIndex < len(right) {\n\t\tmergedProgress = appendOrMergeProgressWithDifferentPredicate(mergedProgress, right[rightIndex])\n\t\trightIndex++\n\t}\n\treturn mergedProgress\n}\n\nfunc appendOrMergeProgressWithDifferentPredicate(mergedProgress []*GetTaskProgress, progress *GetTaskProgress) []*GetTaskProgress {\n\tif len(mergedProgress) == 0 {\n\t\treturn append(mergedProgress, progress)\n\t}\n\n\tlastProgress := mergedProgress[len(mergedProgress)-1]\n\tmergedProgress = mergedProgress[:len(mergedProgress)-1] // remove the last progress\n\treturn append(mergedProgress, mergeProgressWithDifferentPredicate(lastProgress, progress)...)\n}\n\n// mergeProgress merges two progress with different predicates\n// Assuming the inclusive key, next key, max key of the 2 progress are [a, b, c] and [x, y, z] where a <= b <= c and x <= y <= z,\n// also assuming that b <= y, otherwise we can swap the left and right progress\n// There are 7 different cases regarding the order of a, b, c, x, y, z, and here are the cases and merged results:\n// [a, b, c, x, y, z] -> [b, b, c] and [y, y, z], (c < y)\n// [a, b, x, c, y, z] -> [b, b, c] and [y, y, z], (c < y)\n// [a, b, x, y, c, z] -> [b, b, z]\n// [a, b, x, y, z, c] -> [b, b, c]\n// [a, x, b, c, y ,z] -> [b, b, c] and [y, y, z], (c < y)\n// [a, x, b, y, c, z] -> [b, b, z]\n// [a, x, b, y, z, c] -> [b, b, c]\n// the idea is to get the union of [next task key, exclusive max task key) of the 2 progress,\n// and then set inclusive min task key to next task key, next page token to nil\nfunc mergeProgressWithDifferentPredicate(left, right *GetTaskProgress) []*GetTaskProgress {\n\tif left.NextTaskKey.Compare(right.NextTaskKey) > 0 {\n\t\tleft, right = right, left\n\t}\n\n\tif left.Range.ExclusiveMaxTaskKey.Compare(right.NextTaskKey) < 0 {\n\t\treturn []*GetTaskProgress{\n\t\t\t{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: left.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: left.ExclusiveMaxTaskKey,\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   left.NextTaskKey,\n\t\t\t},\n\t\t\t{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: right.NextTaskKey,\n\t\t\t\t\tExclusiveMaxTaskKey: right.ExclusiveMaxTaskKey,\n\t\t\t\t},\n\t\t\t\tNextPageToken: nil,\n\t\t\t\tNextTaskKey:   right.NextTaskKey,\n\t\t\t},\n\t\t}\n\t}\n\treturn []*GetTaskProgress{\n\t\t{\n\t\t\tRange: Range{\n\t\t\t\tInclusiveMinTaskKey: left.NextTaskKey,\n\t\t\t\tExclusiveMaxTaskKey: persistence.MaxHistoryTaskKey(left.ExclusiveMaxTaskKey, right.ExclusiveMaxTaskKey),\n\t\t\t},\n\t\t\tNextPageToken: nil,\n\t\t\tNextTaskKey:   left.NextTaskKey,\n\t\t},\n\t}\n}\n\nfunc mergeVirtualSlicesByPredicate(this, that *virtualSliceImpl) VirtualSlice {\n\tmergedState := VirtualSliceState{\n\t\tRange:     this.state.Range, // this and that have the same range\n\t\tPredicate: Or(this.state.Predicate, that.state.Predicate),\n\t}\n\tpendingTaskTracker := this.pendingTaskTracker\n\ttaskMap := that.pendingTaskTracker.GetTasks()\n\tfor _, task := range taskMap {\n\t\tpendingTaskTracker.AddTask(task)\n\t}\n\tmergedProgress := mergeGetTaskProgressWithDifferentPredicate(this.progress, that.progress)\n\n\treturn &virtualSliceImpl{\n\t\tstate:              mergedState,\n\t\ttaskInitializer:    this.taskInitializer,\n\t\tqueueReader:        this.queueReader,\n\t\tpendingTaskTracker: pendingTaskTracker,\n\t\tprogress:           mergedProgress,\n\t\tlogger:             this.logger,\n\t}\n}\n\n// merge slices with different predicates, the general idea is to find the overlap range of the 2 slices,\n// combine the predicates of the overlap range and leave the rest as is\nfunc mergeVirtualSlicesWithDifferentPredicate(this, that *virtualSliceImpl) ([]VirtualSlice, bool) {\n\tcompareInclusiveMin := this.state.Range.InclusiveMinTaskKey.Compare(that.state.Range.InclusiveMinTaskKey)\n\tif compareInclusiveMin > 0 {\n\t\tthis, that = that, this\n\t} else if compareInclusiveMin == 0 {\n\t\tif this.state.Range.ExclusiveMaxTaskKey.Compare(that.state.Range.ExclusiveMaxTaskKey) > 0 {\n\t\t\tthis, that = that, this\n\t\t}\n\t}\n\t// Use a, b to to represent the inclusive min task key and exclusive max task key of `this`\n\t// Use x, y to to represent the inclusive min task key and exclusive max task key of `that`\n\t// At this point, we know that a <= x, (in actual world, we also know that x <= b because we know that the 2 slices can be merged, but it doesn't affect the logic), so there are 5 cases to consider:\n\t// 1. {a, b, x, y} (x >= b, a < x) -> don't merge\n\t// 2. {a, x, b, y} (a < x < b < y) -> [a, x) and [x, b) and [b, y)\n\t// 3. {a, x, b, y} (a < x < b == y) -> [a, x) and [x, b)\n\t// 4. {a, x, y, b} (a < x < y < b) -> [a, x) and [x, y) and [y, b)\n\t// 5. {x, a, b, y} (x == a < b < y) -> [x, b) and [b, y)\n\t// 6. {x, a, y, b} (x == a < y == b) -> [x, b)\n\tif compareInclusiveMin == 0 {\n\t\tthatLeft, thatRight, ok := that.TrySplitByTaskKey(this.state.Range.ExclusiveMaxTaskKey)\n\t\tif !ok {\n\t\t\t// Case 6\n\t\t\treturn []VirtualSlice{mergeVirtualSlicesByPredicate(this, that)}, true\n\t\t}\n\t\t// Case 5\n\t\treturn []VirtualSlice{mergeVirtualSlicesByPredicate(this, thatLeft.(*virtualSliceImpl)), thatRight}, true\n\t}\n\tthisLeft, thisRight, ok := this.TrySplitByTaskKey(that.state.Range.InclusiveMinTaskKey)\n\tif !ok {\n\t\t// Case 1\n\t\treturn nil, false\n\t}\n\tmergedVirtualSlices := make([]VirtualSlice, 0, 3)\n\tmergedVirtualSlices = append(mergedVirtualSlices, thisLeft)\n\tthatLeft, thatRight, ok := that.TrySplitByTaskKey(this.state.Range.ExclusiveMaxTaskKey)\n\tif !ok {\n\t\tthisRightLeft, thisRightRight, ok := thisRight.TrySplitByTaskKey(that.state.Range.ExclusiveMaxTaskKey)\n\t\tif !ok {\n\t\t\t// Case 3: combine predicates of thisRight and that\n\t\t\tmergedVirtualSlices = append(mergedVirtualSlices, mergeVirtualSlicesByPredicate(thisRight.(*virtualSliceImpl), that))\n\t\t\treturn mergedVirtualSlices, true\n\t\t}\n\t\t// Case 4: combine predicates of thisRightLeft and that\n\t\tmergedVirtualSlices = append(mergedVirtualSlices, mergeVirtualSlicesByPredicate(thisRightLeft.(*virtualSliceImpl), that))\n\t\tmergedVirtualSlices = append(mergedVirtualSlices, thisRightRight)\n\t\treturn mergedVirtualSlices, true\n\t}\n\t// Case 2: combine predicates of thisRight and thatLeft\n\tmergedVirtualSlices = append(mergedVirtualSlices, mergeVirtualSlicesByPredicate(thisRight.(*virtualSliceImpl), thatLeft.(*virtualSliceImpl)))\n\tmergedVirtualSlices = append(mergedVirtualSlices, thatRight)\n\treturn mergedVirtualSlices, true\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_slice_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/queuev2 (interfaces: VirtualSlice)\n//\n// Generated by this command:\n//\n//\tmockgen -package queuev2 -destination virtual_slice_mock.go github.com/uber/cadence/service/history/queuev2 VirtualSlice\n//\n\n// Package queuev2 is a generated GoMock package.\npackage queuev2\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttask \"github.com/uber/cadence/service/history/task\"\n)\n\n// MockVirtualSlice is a mock of VirtualSlice interface.\ntype MockVirtualSlice struct {\n\tctrl     *gomock.Controller\n\trecorder *MockVirtualSliceMockRecorder\n\tisgomock struct{}\n}\n\n// MockVirtualSliceMockRecorder is the mock recorder for MockVirtualSlice.\ntype MockVirtualSliceMockRecorder struct {\n\tmock *MockVirtualSlice\n}\n\n// NewMockVirtualSlice creates a new mock instance.\nfunc NewMockVirtualSlice(ctrl *gomock.Controller) *MockVirtualSlice {\n\tmock := &MockVirtualSlice{ctrl: ctrl}\n\tmock.recorder = &MockVirtualSliceMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockVirtualSlice) EXPECT() *MockVirtualSliceMockRecorder {\n\treturn m.recorder\n}\n\n// Clear mocks base method.\nfunc (m *MockVirtualSlice) Clear() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Clear\")\n}\n\n// Clear indicates an expected call of Clear.\nfunc (mr *MockVirtualSliceMockRecorder) Clear() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Clear\", reflect.TypeOf((*MockVirtualSlice)(nil).Clear))\n}\n\n// GetPendingTaskCount mocks base method.\nfunc (m *MockVirtualSlice) GetPendingTaskCount() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPendingTaskCount\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetPendingTaskCount indicates an expected call of GetPendingTaskCount.\nfunc (mr *MockVirtualSliceMockRecorder) GetPendingTaskCount() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPendingTaskCount\", reflect.TypeOf((*MockVirtualSlice)(nil).GetPendingTaskCount))\n}\n\n// GetState mocks base method.\nfunc (m *MockVirtualSlice) GetState() VirtualSliceState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetState\")\n\tret0, _ := ret[0].(VirtualSliceState)\n\treturn ret0\n}\n\n// GetState indicates an expected call of GetState.\nfunc (mr *MockVirtualSliceMockRecorder) GetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetState\", reflect.TypeOf((*MockVirtualSlice)(nil).GetState))\n}\n\n// GetTasks mocks base method.\nfunc (m *MockVirtualSlice) GetTasks(arg0 context.Context, arg1 int) ([]task.Task, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTasks\", arg0, arg1)\n\tret0, _ := ret[0].([]task.Task)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTasks indicates an expected call of GetTasks.\nfunc (mr *MockVirtualSliceMockRecorder) GetTasks(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTasks\", reflect.TypeOf((*MockVirtualSlice)(nil).GetTasks), arg0, arg1)\n}\n\n// HasMoreTasks mocks base method.\nfunc (m *MockVirtualSlice) HasMoreTasks() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasMoreTasks\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasMoreTasks indicates an expected call of HasMoreTasks.\nfunc (mr *MockVirtualSliceMockRecorder) HasMoreTasks() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasMoreTasks\", reflect.TypeOf((*MockVirtualSlice)(nil).HasMoreTasks))\n}\n\n// IsEmpty mocks base method.\nfunc (m *MockVirtualSlice) IsEmpty() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsEmpty\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsEmpty indicates an expected call of IsEmpty.\nfunc (mr *MockVirtualSliceMockRecorder) IsEmpty() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsEmpty\", reflect.TypeOf((*MockVirtualSlice)(nil).IsEmpty))\n}\n\n// PendingTaskStats mocks base method.\nfunc (m *MockVirtualSlice) PendingTaskStats() PendingTaskStats {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PendingTaskStats\")\n\tret0, _ := ret[0].(PendingTaskStats)\n\treturn ret0\n}\n\n// PendingTaskStats indicates an expected call of PendingTaskStats.\nfunc (mr *MockVirtualSliceMockRecorder) PendingTaskStats() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PendingTaskStats\", reflect.TypeOf((*MockVirtualSlice)(nil).PendingTaskStats))\n}\n\n// TryMergeWithVirtualSlice mocks base method.\nfunc (m *MockVirtualSlice) TryMergeWithVirtualSlice(arg0 VirtualSlice) ([]VirtualSlice, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TryMergeWithVirtualSlice\", arg0)\n\tret0, _ := ret[0].([]VirtualSlice)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// TryMergeWithVirtualSlice indicates an expected call of TryMergeWithVirtualSlice.\nfunc (mr *MockVirtualSliceMockRecorder) TryMergeWithVirtualSlice(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TryMergeWithVirtualSlice\", reflect.TypeOf((*MockVirtualSlice)(nil).TryMergeWithVirtualSlice), arg0)\n}\n\n// TrySplitByPredicate mocks base method.\nfunc (m *MockVirtualSlice) TrySplitByPredicate(arg0 Predicate) (VirtualSlice, VirtualSlice, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TrySplitByPredicate\", arg0)\n\tret0, _ := ret[0].(VirtualSlice)\n\tret1, _ := ret[1].(VirtualSlice)\n\tret2, _ := ret[2].(bool)\n\treturn ret0, ret1, ret2\n}\n\n// TrySplitByPredicate indicates an expected call of TrySplitByPredicate.\nfunc (mr *MockVirtualSliceMockRecorder) TrySplitByPredicate(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TrySplitByPredicate\", reflect.TypeOf((*MockVirtualSlice)(nil).TrySplitByPredicate), arg0)\n}\n\n// TrySplitByTaskKey mocks base method.\nfunc (m *MockVirtualSlice) TrySplitByTaskKey(arg0 persistence.HistoryTaskKey) (VirtualSlice, VirtualSlice, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TrySplitByTaskKey\", arg0)\n\tret0, _ := ret[0].(VirtualSlice)\n\tret1, _ := ret[1].(VirtualSlice)\n\tret2, _ := ret[2].(bool)\n\treturn ret0, ret1, ret2\n}\n\n// TrySplitByTaskKey indicates an expected call of TrySplitByTaskKey.\nfunc (mr *MockVirtualSliceMockRecorder) TrySplitByTaskKey(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TrySplitByTaskKey\", reflect.TypeOf((*MockVirtualSlice)(nil).TrySplitByTaskKey), arg0)\n}\n\n// UpdateAndGetState mocks base method.\nfunc (m *MockVirtualSlice) UpdateAndGetState() VirtualSliceState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateAndGetState\")\n\tret0, _ := ret[0].(VirtualSliceState)\n\treturn ret0\n}\n\n// UpdateAndGetState indicates an expected call of UpdateAndGetState.\nfunc (mr *MockVirtualSliceMockRecorder) UpdateAndGetState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateAndGetState\", reflect.TypeOf((*MockVirtualSlice)(nil).UpdateAndGetState))\n}\n"
  },
  {
    "path": "service/history/queuev2/virtual_slice_test.go",
    "content": "package queuev2\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\tgomock \"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/history/task\"\n)\n\nfunc TestMergeProgressWithSamePredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tleft     *GetTaskProgress\n\t\tright    *GetTaskProgress\n\t\texpected []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname: \"Case 1: [a,b,c,x,y,z] - Non-overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 2: [a,b,x,c,y,z] - Partially overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 3: [a,b,x,y,c,z] - Overlapping ranges with interleaved keys\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 4: [a,b,x,y,z,c] - Overlapping ranges with right extending beyond\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 5: [a,x,b,c,y,z] - Left range contains right range\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 6: [a,x,b,y,c,z] - Overlapping ranges with interleaved keys\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 7: [a,x,b,y,z,c] - Overlapping ranges with left extending beyond\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 8: [a,x,y,b,c,z] - Right range contains left range\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 9: [a,x,y,b,z,c] - Overlapping ranges with right extending beyond\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 10: [a,x,y,z,b,c] - Right range completely contains left range\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := mergeProgressWithSamePredicate(tt.left, tt.right)\n\t\t\tassert.Equal(t, tt.expected, result)\n\n\t\t\tresult = mergeProgressWithSamePredicate(tt.right, tt.left)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestAppendOrMergeProgressWithSamePredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmergedProgress []*GetTaskProgress\n\t\tprogress       *GetTaskProgress\n\t\texpected       []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname:           \"Empty slice - should append\",\n\t\t\tmergedProgress: []*GetTaskProgress{},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Non-overlapping ranges - should append\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Overlapping ranges - should merge\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Contained ranges - should merge\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple existing progress - should merge with last\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t},\n\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := appendOrMergeProgressWithSamePredicate(tt.mergedProgress, tt.progress)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestMergeGetTaskProgress(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tleft     []*GetTaskProgress\n\t\tright    []*GetTaskProgress\n\t\texpected []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname:     \"Empty slices\",\n\t\t\tleft:     []*GetTaskProgress{},\n\t\t\tright:    []*GetTaskProgress{},\n\t\t\texpected: []*GetTaskProgress{},\n\t\t},\n\t\t{\n\t\t\tname: \"Left empty, right non-empty\",\n\t\t\tleft: []*GetTaskProgress{},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Left non-empty, right empty\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Non-overlapping ranges\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Overlapping ranges\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple progress items with overlaps\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Contained ranges\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := mergeGetTaskProgressWithSamePredicate(tt.left, tt.right)\n\t\t\tassert.Equal(t, tt.expected, result)\n\n\t\t\tresult = mergeGetTaskProgressWithSamePredicate(tt.right, tt.left)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestMergeVirtualSlicesByRange(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tleft     *virtualSliceImpl\n\t\tright    *virtualSliceImpl\n\t\texpected VirtualSliceState\n\t}{\n\t\t{\n\t\t\tname: \"Non-overlapping ranges\",\n\t\t\tleft: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Overlapping ranges\",\n\t\t\tleft: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Contained ranges\",\n\t\t\tleft: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Identical ranges\",\n\t\t\tleft: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Adjacent ranges\",\n\t\t\tleft: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPendingTaskTracker0 := NewMockPendingTaskTracker(ctrl)\n\t\t\tmockPendingTaskTracker1 := NewMockPendingTaskTracker(ctrl)\n\t\t\tmockPendingTaskTracker1.EXPECT().GetTasks().Return(map[persistence.HistoryTaskKey]task.Task{\n\t\t\t\tpersistence.NewImmediateTaskKey(1): task.NewMockTask(ctrl),\n\t\t\t\tpersistence.NewImmediateTaskKey(2): task.NewMockTask(ctrl),\n\t\t\t})\n\t\t\tmockPendingTaskTracker0.EXPECT().AddTask(gomock.Any()).Times(2)\n\t\t\ttt.left.pendingTaskTracker = mockPendingTaskTracker0\n\t\t\ttt.right.pendingTaskTracker = mockPendingTaskTracker1\n\t\t\tresult := mergeVirtualSlicesByRange(tt.left, tt.right)\n\t\t\tassert.Equal(t, tt.expected, result.GetState())\n\t\t})\n\t}\n}\n\nfunc TestTrySplitByTaskKey(t *testing.T) {\n\ttests := []struct {\n\t\tname                  string\n\t\tslice                 *virtualSliceImpl\n\t\tsplitKey              persistence.HistoryTaskKey\n\t\texpectedLeft          VirtualSliceState\n\t\texpectedRight         VirtualSliceState\n\t\texpectedOk            bool\n\t\texpectedLeftProgress  []*GetTaskProgress\n\t\texpectedRightProgress []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname: \"Split at middle of range\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitKey: persistence.NewImmediateTaskKey(3),\n\t\t\texpectedLeft: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedRight: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedLeftProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRightProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Split at middle of range with multiple progress items\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(40),\n\t\t\t\t\t\tNextPageToken: []byte{4, 5, 6},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(70),\n\t\t\t\t\t\tNextPageToken: []byte{7, 8, 9},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitKey: persistence.NewImmediateTaskKey(50),\n\t\t\texpectedLeft: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedRight: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedLeftProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(40),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(40),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRightProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(70),\n\t\t\t\t\tNextPageToken: []byte{7, 8, 9},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Split at middle of range with multiple progress items - 2\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(55),\n\t\t\t\t\t\tNextPageToken: []byte{4, 5, 6},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(70),\n\t\t\t\t\t\tNextPageToken: []byte{7, 8, 9},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitKey: persistence.NewImmediateTaskKey(50),\n\t\t\texpectedLeft: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedRight: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(50),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedLeftProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRightProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(55),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(55),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(70),\n\t\t\t\t\tNextPageToken: []byte{7, 8, 9},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Split key outside range - should fail\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitKey:   persistence.NewImmediateTaskKey(6),\n\t\t\texpectedOk: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\n\t\t\ttt.slice.taskInitializer = func(t persistence.Task) task.Task {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\treturn mockTask\n\t\t\t}\n\t\t\ttt.slice.queueReader = mockQueueReader\n\t\t\ttt.slice.pendingTaskTracker = NewPendingTaskTracker()\n\n\t\t\tleft, right, ok := tt.slice.TrySplitByTaskKey(tt.splitKey)\n\t\t\tassert.Equal(t, tt.expectedOk, ok)\n\n\t\t\tif ok {\n\t\t\t\tassert.Equal(t, tt.expectedLeft, left.GetState())\n\t\t\t\tassert.Equal(t, tt.expectedRight, right.GetState())\n\n\t\t\t\t// Verify progress\n\t\t\t\tleftImpl, ok := left.(*virtualSliceImpl)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Equal(t, tt.expectedLeftProgress, leftImpl.progress)\n\n\t\t\t\trightImpl, ok := right.(*virtualSliceImpl)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Equal(t, tt.expectedRightProgress, rightImpl.progress)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, left)\n\t\t\t\tassert.Nil(t, right)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateAndGetState(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tslice         *virtualSliceImpl\n\t\texpectedState VirtualSliceState\n\t\tsetupMock     func(*MockPendingTaskTracker)\n\t}{\n\t\t{\n\t\t\tname: \"No pending tasks, no more tasks to read\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\tsetupMock: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().PruneAckedTasks()\n\t\t\t\tmock.EXPECT().GetMinimumTaskKey().Return(persistence.MaximumHistoryTaskKey, false)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"No pending tasks, has more tasks to read\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\tsetupMock: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().PruneAckedTasks()\n\t\t\t\tmock.EXPECT().GetMinimumTaskKey().Return(persistence.MaximumHistoryTaskKey, false)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Has pending tasks, no more tasks to read\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\tsetupMock: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().PruneAckedTasks()\n\t\t\t\tmock.EXPECT().GetMinimumTaskKey().Return(persistence.NewImmediateTaskKey(2), true)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Has pending tasks, has more tasks to read\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedState: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t},\n\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t},\n\t\t\tsetupMock: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().PruneAckedTasks()\n\t\t\t\tmock.EXPECT().GetMinimumTaskKey().Return(persistence.NewImmediateTaskKey(5), true)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockPendingTaskTracker := NewMockPendingTaskTracker(ctrl)\n\n\t\t\ttt.slice.queueReader = mockQueueReader\n\t\t\ttt.slice.pendingTaskTracker = mockPendingTaskTracker\n\t\t\ttt.slice.logger = testlogger.New(t)\n\n\t\t\t// Setup mock expectations using the setupMock function\n\t\t\ttt.setupMock(mockPendingTaskTracker)\n\n\t\t\tresult := tt.slice.UpdateAndGetState()\n\t\t\tassert.Equal(t, tt.expectedState, result)\n\t\t})\n\t}\n}\n\nfunc TestGetTasks(t *testing.T) {\n\thistoryTasks := []persistence.Task{\n\t\t&persistence.DecisionTask{},\n\t\t&persistence.ActivityTask{},\n\t\t&persistence.ActivityTask{},\n\t}\n\ttests := []struct {\n\t\tname               string\n\t\tslice              *virtualSliceImpl\n\t\tpageSize           int\n\t\tsetupMock          func(*MockQueueReader, *MockPendingTaskTracker)\n\t\texpectedTasksCount int\n\t\texpectedError      error\n\t\texpectedProgress   []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname: \"Empty progress - should return empty tasks\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{},\n\t\t\t},\n\t\t\tpageSize: 10,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\t// No expectations needed as progress is empty\n\t\t\t},\n\t\t\texpectedTasksCount: 0,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress:   []*GetTaskProgress{},\n\t\t},\n\t\t{\n\t\t\tname: \"Single page of tasks\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 2,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  2,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[0], historyTasks[1]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockPendingTaskTracker.EXPECT().AddTask(gomock.Any()).Times(2)\n\t\t\t},\n\t\t\texpectedTasksCount: 2,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple pages of tasks\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 3,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\t// First page\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  3,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[0], historyTasks[1]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\t// Second page\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[2]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockPendingTaskTracker.EXPECT().AddTask(gomock.Any()).Times(3)\n\t\t\t},\n\t\t\texpectedTasksCount: 3,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(6),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error from queue reader\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 2,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedTasksCount: 0,\n\t\t\texpectedError:      assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname: \"Single page of tasks with error from queue reader\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 2,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  2,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[0]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockPendingTaskTracker.EXPECT().AddTask(gomock.Any()).Times(1)\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\texpectedTasksCount: 1,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple pages of tasks with error from queue reader\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 3,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\t// First page\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  3,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[0], historyTasks[1]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\t// Second page\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  1,\n\t\t\t\t}).Return(nil, assert.AnError)\n\n\t\t\t\tmockPendingTaskTracker.EXPECT().AddTask(gomock.Any()).Times(2)\n\t\t\t},\n\t\t\texpectedTasksCount: 2,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Single page of tasks with next task key equal to exclusive max task key but non-empty next page token\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tpageSize: 2,\n\t\t\tsetupMock: func(mockQueueReader *MockQueueReader, mockPendingTaskTracker *MockPendingTaskTracker) {\n\t\t\t\tmockQueueReader.EXPECT().GetTask(gomock.Any(), &GetTaskRequest{\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t\tPageSize:  2,\n\t\t\t\t}).Return(&GetTaskResponse{\n\t\t\t\t\tTasks: []persistence.Task{historyTasks[0]},\n\t\t\t\t\tProgress: &GetTaskProgress{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\tmockPendingTaskTracker.EXPECT().AddTask(gomock.Any()).Times(1)\n\t\t\t},\n\t\t\texpectedTasksCount: 1,\n\t\t\texpectedError:      nil,\n\t\t\texpectedProgress:   []*GetTaskProgress{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\t\t\tmockPendingTaskTracker := NewMockPendingTaskTracker(ctrl)\n\n\t\t\ttt.slice.queueReader = mockQueueReader\n\t\t\ttt.slice.pendingTaskTracker = mockPendingTaskTracker\n\t\t\ttt.slice.taskInitializer = func(t persistence.Task) task.Task {\n\t\t\t\treturn task.NewMockTask(ctrl)\n\t\t\t}\n\n\t\t\t// Setup mock expectations\n\t\t\ttt.setupMock(mockQueueReader, mockPendingTaskTracker)\n\n\t\t\ttasks, err := tt.slice.GetTasks(context.Background(), tt.pageSize)\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedError, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Len(t, tasks, tt.expectedTasksCount)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMergeProgressWithDifferentPredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tleft     *GetTaskProgress\n\t\tright    *GetTaskProgress\n\t\texpected []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname: \"Case 1: Non-overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 1,2: Adjacent ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 2: Overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 3: Adjacent ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: nil,\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 3: Overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(4),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 4: Completely overlapping ranges - left contains right\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 3,4: Equal exclusive max task keys\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(4),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 5: Overlapping ranges\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 6: Overlapping ranges with same next task key\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 7: Completely overlapping ranges - left contains right\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Identical ranges and next task keys\",\n\t\t\tleft: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t},\n\t\t\tright: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// make sure that these test inputs are valid\n\t\t\tif tt.left.NextTaskKey.Compare(tt.left.Range.InclusiveMinTaskKey) == 0 {\n\t\t\t\trequire.Nil(t, tt.left.NextPageToken)\n\t\t\t} else {\n\t\t\t\trequire.NotNil(t, tt.left.NextPageToken)\n\t\t\t}\n\t\t\tif tt.right.NextTaskKey.Compare(tt.right.Range.InclusiveMinTaskKey) == 0 {\n\t\t\t\trequire.Nil(t, tt.right.NextPageToken)\n\t\t\t} else {\n\t\t\t\trequire.NotNil(t, tt.right.NextPageToken)\n\t\t\t}\n\n\t\t\t// test the merge function\n\t\t\tresult := mergeProgressWithDifferentPredicate(tt.left, tt.right)\n\t\t\tassert.Equal(t, tt.expected, result)\n\n\t\t\t// Test commutativity - the function should handle argument order internally\n\t\t\tresult2 := mergeProgressWithDifferentPredicate(tt.right, tt.left)\n\t\t\tassert.Equal(t, tt.expected, result2)\n\t\t})\n\t}\n}\n\nfunc TestAppendOrMergeProgressWithDifferentPredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmergedProgress []*GetTaskProgress\n\t\tprogress       *GetTaskProgress\n\t\texpected       []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname:           \"Empty slice - should append\",\n\t\t\tmergedProgress: []*GetTaskProgress{},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Non-overlapping ranges - should append both with reset tokens\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Adjacent ranges - should merge into single range\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: nil,\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Overlapping ranges - should merge with earliest next task key\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Completely contained ranges - should merge to outer range\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Same next task key - different ranges\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple existing progress - should merge with last only\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(6),\n\t\t\t\tNextPageToken: []byte(\"token3\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple existing progress - last range non-overlapping\",\n\t\t\tmergedProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tprogress: &GetTaskProgress{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t},\n\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(9),\n\t\t\t\tNextPageToken: []byte(\"token3\"),\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(9),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(9),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := appendOrMergeProgressWithDifferentPredicate(tt.mergedProgress, tt.progress)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestMergeGetTaskProgressWithDifferentPredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tleft     []*GetTaskProgress\n\t\tright    []*GetTaskProgress\n\t\texpected []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname:     \"Empty slices\",\n\t\t\tleft:     []*GetTaskProgress{},\n\t\t\tright:    []*GetTaskProgress{},\n\t\t\texpected: []*GetTaskProgress{},\n\t\t},\n\t\t{\n\t\t\tname: \"Left empty, right non-empty\",\n\t\t\tleft: []*GetTaskProgress{},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Left non-empty, right empty\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Non-overlapping ranges - ordered by NextTaskKey\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Adjacent ranges by NextTaskKey\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(4), // comes after left's range\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple progress items - complex merge by NextTaskKey\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(7),\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tNextPageToken: []byte(\"token3\"),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(9),\n\t\t\t\t\tNextPageToken: []byte(\"token4\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t// Merged based on NextTaskKey order: 2, 4, 7, 9\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(2),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(4),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(7),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(7),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(9),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(9),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Overlapping ranges with interleaved NextTaskKeys\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(5),\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Same NextTaskKey - should merge ranges\",\n\t\t\tleft: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: []byte(\"token1\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tright: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3), // Same NextTaskKey\n\t\t\t\t\tNextPageToken: []byte(\"token2\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(3),\n\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := mergeGetTaskProgressWithDifferentPredicate(tt.left, tt.right)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestVirtualSliceImpl_TrySplitByPredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname                  string\n\t\tslice                 *virtualSliceImpl\n\t\tsplitPredicate        Predicate\n\t\texpectedLeft          VirtualSliceState\n\t\texpectedRight         VirtualSliceState\n\t\texpectedOk            bool\n\t\texpectedLeftProgress  []*GetTaskProgress\n\t\texpectedRightProgress []*GetTaskProgress\n\t}{\n\t\t{\n\t\t\tname: \"Universal predicate should not split\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitPredicate: NewUniversalPredicate(),\n\t\t\texpectedOk:     false,\n\t\t},\n\t\t{\n\t\t\tname: \"Empty predicate should not split\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitPredicate: NewEmptyPredicate(),\n\t\t\texpectedOk:     false,\n\t\t},\n\t\t{\n\t\t\tname: \"Identical predicate should not split\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\texpectedOk:     false,\n\t\t},\n\t\t{\n\t\t\tname: \"Different predicate should split successfully\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitPredicate: NewDomainIDPredicate([]string{\"domain3\"}, false),\n\t\t\texpectedOk:     true,\n\t\t\texpectedLeft: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: And(NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false), NewDomainIDPredicate([]string{\"domain3\"}, false)),\n\t\t\t},\n\t\t\texpectedRight: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t},\n\t\t\t\tPredicate: And(NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false), Not(NewDomainIDPredicate([]string{\"domain3\"}, false))),\n\t\t\t},\n\t\t\texpectedLeftProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRightProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Split with multiple progress items\",\n\t\t\tslice: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(40),\n\t\t\t\t\t\tNextPageToken: []byte{4, 5, 6},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsplitPredicate: NewDomainIDPredicate([]string{\"domain1\"}, false),\n\t\t\texpectedOk:     true,\n\t\t\texpectedLeft: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t},\n\t\t\t\tPredicate: And(NewUniversalPredicate(), NewDomainIDPredicate([]string{\"domain1\"}, false)),\n\t\t\t},\n\t\t\texpectedRight: VirtualSliceState{\n\t\t\t\tRange: Range{\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(100),\n\t\t\t\t},\n\t\t\t\tPredicate: And(NewUniversalPredicate(), Not(NewDomainIDPredicate([]string{\"domain1\"}, false))),\n\t\t\t},\n\t\t\texpectedLeftProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(40),\n\t\t\t\t\tNextPageToken: []byte{4, 5, 6},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedRightProgress: []*GetTaskProgress{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(20),\n\t\t\t\t\tNextPageToken: []byte{1, 2, 3},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(30),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(60),\n\t\t\t\t\t},\n\t\t\t\t\tNextTaskKey:   persistence.NewImmediateTaskKey(40),\n\t\t\t\t\tNextPageToken: []byte{4, 5, 6},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockQueueReader := NewMockQueueReader(ctrl)\n\n\t\t\ttt.slice.taskInitializer = func(t persistence.Task) task.Task {\n\t\t\t\tmockTask := task.NewMockTask(ctrl)\n\t\t\t\treturn mockTask\n\t\t\t}\n\t\t\ttt.slice.queueReader = mockQueueReader\n\t\t\ttt.slice.pendingTaskTracker = NewPendingTaskTracker()\n\n\t\t\tleft, right, ok := tt.slice.TrySplitByPredicate(tt.splitPredicate)\n\t\t\tassert.Equal(t, tt.expectedOk, ok)\n\n\t\t\tif ok {\n\t\t\t\tassert.Equal(t, tt.expectedLeft, left.GetState())\n\t\t\t\tassert.Equal(t, tt.expectedRight, right.GetState())\n\n\t\t\t\t// Verify progress\n\t\t\t\tleftImpl, ok := left.(*virtualSliceImpl)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Equal(t, tt.expectedLeftProgress, leftImpl.progress)\n\n\t\t\t\trightImpl, ok := right.(*virtualSliceImpl)\n\t\t\t\tassert.True(t, ok)\n\t\t\t\tassert.Equal(t, tt.expectedRightProgress, rightImpl.progress)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, left)\n\t\t\t\tassert.Nil(t, right)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestPendingTaskStats(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tmockSetup     func(*MockPendingTaskTracker)\n\t\texpectedStats PendingTaskStats\n\t}{\n\t\t{\n\t\t\tname: \"Empty pending task tracker - should return empty stats\",\n\t\t\tmockSetup: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().GetPerDomainPendingTaskCount().Return(map[string]int{})\n\t\t\t},\n\t\t\texpectedStats: PendingTaskStats{\n\t\t\t\tPendingTaskCountPerDomain: map[string]int{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Single domain with tasks - should return correct stats\",\n\t\t\tmockSetup: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().GetPerDomainPendingTaskCount().Return(map[string]int{\n\t\t\t\t\t\"domain1\": 5,\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedStats: PendingTaskStats{\n\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\"domain1\": 5,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Multiple domains with tasks - should return correct stats\",\n\t\t\tmockSetup: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().GetPerDomainPendingTaskCount().Return(map[string]int{\n\t\t\t\t\t\"domain1\": 3,\n\t\t\t\t\t\"domain2\": 7,\n\t\t\t\t\t\"domain3\": 2,\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedStats: PendingTaskStats{\n\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\"domain1\": 3,\n\t\t\t\t\t\"domain2\": 7,\n\t\t\t\t\t\"domain3\": 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Domain with zero tasks - should include zero counts\",\n\t\t\tmockSetup: func(mock *MockPendingTaskTracker) {\n\t\t\t\tmock.EXPECT().GetPerDomainPendingTaskCount().Return(map[string]int{\n\t\t\t\t\t\"domain1\": 5,\n\t\t\t\t\t\"domain2\": 0,\n\t\t\t\t\t\"domain3\": 3,\n\t\t\t\t})\n\t\t\t},\n\t\t\texpectedStats: PendingTaskStats{\n\t\t\t\tPendingTaskCountPerDomain: map[string]int{\n\t\t\t\t\t\"domain1\": 5,\n\t\t\t\t\t\"domain2\": 0,\n\t\t\t\t\t\"domain3\": 3,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockPendingTaskTracker := NewMockPendingTaskTracker(ctrl)\n\n\t\t\t// Setup the virtual slice with mock dependencies\n\t\t\tslice := &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewUniversalPredicate(),\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: mockPendingTaskTracker,\n\t\t\t}\n\n\t\t\t// Setup mock expectations\n\t\t\ttt.mockSetup(mockPendingTaskTracker)\n\n\t\t\t// Call the method under test\n\t\t\tresult := slice.PendingTaskStats()\n\n\t\t\t// Verify the result\n\t\t\tassert.Equal(t, tt.expectedStats, result)\n\t\t})\n\t}\n}\n\nfunc TestMergeVirtualSlicesWithDifferentPredicate(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tthis           *virtualSliceImpl\n\t\tthat           *virtualSliceImpl\n\t\texpectedStates []VirtualSliceState\n\t\texpectedOk     bool\n\t}{\n\t\t{\n\t\t\tname: \"Case 1: Non-overlapping ranges\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(4),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Case 2: Overlapping ranges\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(8),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 3: Completely overlapping ranges - left contains right and has the same exclusive max task key\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(5),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 4: Completely overlapping ranges - left contains right and has different exclusive max task key\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(3),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 5: Completely overlapping ranges - left contains right and has different exclusive max task key\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(6),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Case 6: Same ranges\",\n\t\t\tthis: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\tthat: &virtualSliceImpl{\n\t\t\t\tstate: VirtualSliceState{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t\tprogress: []*GetTaskProgress{\n\t\t\t\t\t{\n\t\t\t\t\t\tRange: Range{\n\t\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextTaskKey: persistence.NewImmediateTaskKey(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tpendingTaskTracker: NewPendingTaskTracker(),\n\t\t\t},\n\t\t\texpectedOk: true,\n\t\t\texpectedStates: []VirtualSliceState{\n\t\t\t\t{\n\t\t\t\t\tRange: Range{\n\t\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(1),\n\t\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(10),\n\t\t\t\t\t},\n\t\t\t\t\tPredicate: NewDomainIDPredicate([]string{\"domain1\", \"domain2\", \"domain3\", \"domain4\"}, false),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmergedSlices, ok := mergeVirtualSlicesWithDifferentPredicate(tt.this, tt.that)\n\t\t\tassert.Equal(t, tt.expectedOk, ok)\n\t\t\tif ok {\n\t\t\t\tactualStates := make([]VirtualSliceState, len(mergedSlices))\n\t\t\t\tfor i, slice := range mergedSlices {\n\t\t\t\t\tactualStates[i] = slice.GetState()\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tt.expectedStates, actualStates)\n\t\t\t}\n\t\t\t// Test commutativity - the function should handle argument order internally\n\n\t\t\tmergedSlices, ok = mergeVirtualSlicesWithDifferentPredicate(tt.that, tt.this)\n\t\t\tassert.Equal(t, tt.expectedOk, ok)\n\t\t\tif ok {\n\t\t\t\tactualStates := make([]VirtualSliceState, len(mergedSlices))\n\t\t\t\tfor i, slice := range mergedSlices {\n\t\t\t\t\tactualStates[i] = slice.GetState()\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tt.expectedStates, actualStates)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/replication/dlq_handler.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tdefaultBeginningMessageID = -1\n)\n\nvar (\n\terrInvalidCluster = &types.BadRequestError{Message: \"Invalid target cluster name.\"}\n)\n\ntype (\n\t// DLQHandler is the interface handles replication DLQ messages\n\tDLQHandler interface {\n\t\tcommon.Daemon\n\n\t\tGetMessageCount(\n\t\t\tctx context.Context,\n\t\t\tforceFetch bool,\n\t\t) (map[string]int64, error)\n\t\tReadMessages(\n\t\t\tctx context.Context,\n\t\t\tsourceCluster string,\n\t\t\tlastMessageID int64,\n\t\t\tpageSize int,\n\t\t\tpageToken []byte,\n\t\t) ([]*types.ReplicationTask, []*types.ReplicationTaskInfo, []byte, error)\n\t\tPurgeMessages(\n\t\t\tctx context.Context,\n\t\t\tsourceCluster string,\n\t\t\tlastMessageID int64,\n\t\t) error\n\t\tMergeMessages(\n\t\t\tctx context.Context,\n\t\t\tsourceCluster string,\n\t\t\tlastMessageID int64,\n\t\t\tpageSize int,\n\t\t\tpageToken []byte,\n\t\t) ([]byte, error)\n\t}\n\n\tdlqHandlerImpl struct {\n\t\ttaskExecutors map[string]TaskExecutor\n\t\tshard         shard.Context\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tdone          chan struct{}\n\t\tstatus        int32\n\t\ttimeSource    clock.TimeSource\n\n\t\tmu           sync.Mutex\n\t\tlatestCounts map[string]int64\n\t}\n)\n\nvar _ DLQHandler = (*dlqHandlerImpl)(nil)\n\n// NewDLQHandler initialize the replication message DLQ handler\nfunc NewDLQHandler(\n\tshard shard.Context,\n\ttaskExecutors map[string]TaskExecutor,\n) DLQHandler {\n\n\tif taskExecutors == nil {\n\t\tpanic(\"Failed to initialize replication DLQ handler due to nil task executors\")\n\t}\n\n\treturn &dlqHandlerImpl{\n\t\tshard:         shard,\n\t\ttaskExecutors: taskExecutors,\n\t\tlogger:        shard.GetLogger(),\n\t\tmetricsClient: shard.GetMetricsClient(),\n\t\tdone:          make(chan struct{}),\n\t\ttimeSource:    clock.NewRealTimeSource(),\n\t}\n}\n\n// Start starts the DLQ handler\nfunc (r *dlqHandlerImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tgo r.emitDLQSizeMetricsLoop()\n\tr.logger.Info(\"DLQ handler started.\")\n}\n\n// Stop stops the DLQ handler\nfunc (r *dlqHandlerImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tr.logger.Debug(\"DLQ handler shutting down.\")\n\tclose(r.done)\n}\n\nfunc (r *dlqHandlerImpl) GetMessageCount(ctx context.Context, forceFetch bool) (map[string]int64, error) {\n\tif forceFetch || r.latestCounts == nil {\n\t\tif err := r.fetchAndEmitMessageCount(ctx); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn r.latestCounts, nil\n}\n\nfunc (r *dlqHandlerImpl) fetchAndEmitMessageCount(ctx context.Context) error {\n\tshardID := strconv.Itoa(r.shard.GetShardID())\n\tresult := map[string]int64{}\n\tfor sourceCluster := range r.taskExecutors {\n\t\trequest := persistence.GetReplicationDLQSizeRequest{SourceClusterName: sourceCluster}\n\t\tresponse, err := r.shard.GetExecutionManager().GetReplicationDLQSize(ctx, &request)\n\t\tif err != nil {\n\t\t\tr.logger.Error(\"failed to get replication DLQ size\", tag.Error(err))\n\t\t\tr.metricsClient.Scope(metrics.ReplicationDLQStatsScope).IncCounter(metrics.ReplicationDLQProbeFailed)\n\t\t\treturn err\n\t\t}\n\t\tr.metricsClient.Scope(\n\t\t\tmetrics.ReplicationDLQStatsScope,\n\t\t\tmetrics.SourceClusterTag(sourceCluster),\n\t\t\tmetrics.InstanceTag(shardID),\n\t\t).UpdateGauge(metrics.ReplicationDLQSize, float64(response.Size))\n\n\t\tif response.Size > 0 {\n\t\t\tresult[sourceCluster] = response.Size\n\t\t}\n\t}\n\n\tr.mu.Lock()\n\tr.latestCounts = result\n\tr.mu.Unlock()\n\n\treturn nil\n}\n\nfunc (r *dlqHandlerImpl) emitDLQSizeMetricsLoop() {\n\tgetInterval := func() time.Duration {\n\t\treturn backoff.JitDuration(\n\t\t\tdlqMetricsEmitTimerInterval,\n\t\t\tdlqMetricsEmitTimerCoefficient,\n\t\t)\n\t}\n\n\ttimer := r.timeSource.NewTimer(getInterval())\n\tdefer timer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-timer.Chan():\n\t\t\tr.fetchAndEmitMessageCount(context.Background())\n\t\t\ttimer.Reset(getInterval())\n\t\tcase <-r.done:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (r *dlqHandlerImpl) ReadMessages(\n\tctx context.Context,\n\tsourceCluster string,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*types.ReplicationTask, []*types.ReplicationTaskInfo, []byte, error) {\n\n\treturn r.readMessagesWithAckLevel(\n\t\tctx,\n\t\tsourceCluster,\n\t\tlastMessageID,\n\t\tpageSize,\n\t\tpageToken,\n\t)\n}\n\nfunc (r *dlqHandlerImpl) readMessagesWithAckLevel(\n\tctx context.Context,\n\tsourceCluster string,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]*types.ReplicationTask, []*types.ReplicationTaskInfo, []byte, error) {\n\n\tresp, err := r.shard.GetExecutionManager().GetReplicationTasksFromDLQ(\n\t\tctx,\n\t\t&persistence.GetReplicationTasksFromDLQRequest{\n\t\t\tSourceClusterName: sourceCluster,\n\t\t\tReadLevel:         defaultBeginningMessageID + 1,\n\t\t\tMaxReadLevel:      lastMessageID + 1,\n\t\t\tBatchSize:         pageSize,\n\t\t\tNextPageToken:     pageToken,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\n\tremoteAdminClient, err := r.shard.GetService().GetClientBean().GetRemoteAdminClient(sourceCluster)\n\tif err != nil {\n\t\treturn nil, nil, nil, err\n\t}\n\n\ttaskInfo := make([]*types.ReplicationTaskInfo, 0, len(resp.Tasks))\n\tfor _, task := range resp.Tasks {\n\t\tti, err := task.ToInternalReplicationTaskInfo()\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, err\n\t\t}\n\t\ttaskInfo = append(taskInfo, ti)\n\t}\n\tresponse := &types.GetDLQReplicationMessagesResponse{}\n\tif len(taskInfo) > 0 {\n\t\tresponse, err = remoteAdminClient.GetDLQReplicationMessages(\n\t\t\tctx,\n\t\t\t&types.GetDLQReplicationMessagesRequest{\n\t\t\t\tTaskInfos: taskInfo,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, nil, err\n\t\t}\n\t}\n\n\treturn response.ReplicationTasks, taskInfo, resp.NextPageToken, nil\n}\n\nfunc (r *dlqHandlerImpl) PurgeMessages(\n\tctx context.Context,\n\tsourceCluster string,\n\tlastMessageID int64,\n) error {\n\n\t_, err := r.shard.GetExecutionManager().RangeDeleteReplicationTaskFromDLQ(\n\t\tctx,\n\t\t&persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\tSourceClusterName:    sourceCluster,\n\t\t\tInclusiveBeginTaskID: defaultBeginningMessageID + 1,\n\t\t\tExclusiveEndTaskID:   lastMessageID + 1,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (r *dlqHandlerImpl) MergeMessages(\n\tctx context.Context,\n\tsourceCluster string,\n\tlastMessageID int64,\n\tpageSize int,\n\tpageToken []byte,\n) ([]byte, error) {\n\n\tif _, ok := r.taskExecutors[sourceCluster]; !ok {\n\t\treturn nil, errInvalidCluster\n\t}\n\n\ttasks, rawTasks, token, err := r.readMessagesWithAckLevel(\n\t\tctx,\n\t\tsourceCluster,\n\t\tlastMessageID,\n\t\tpageSize,\n\t\tpageToken,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treplicationTasks := map[int64]*types.ReplicationTask{}\n\tfor _, task := range tasks {\n\t\treplicationTasks[task.SourceTaskID] = task\n\t}\n\n\tlastMessageID = defaultBeginningMessageID\n\tfor _, raw := range rawTasks {\n\t\tif task, ok := replicationTasks[raw.TaskID]; ok {\n\t\t\tif _, err := r.taskExecutors[sourceCluster].execute(task, true); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\n\t\t// If hydrated replication task does not exist in remote cluster - continue merging\n\t\t// Record lastMessageID with raw task id, so that they can be purged after.\n\t\tif lastMessageID < raw.TaskID {\n\t\t\tlastMessageID = raw.TaskID\n\t\t}\n\t}\n\n\t_, err = r.shard.GetExecutionManager().RangeDeleteReplicationTaskFromDLQ(\n\t\tctx,\n\t\t&persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\tSourceClusterName:    sourceCluster,\n\t\t\tInclusiveBeginTaskID: defaultBeginningMessageID + 1,\n\t\t\tExclusiveEndTaskID:   lastMessageID + 1,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn token, nil\n}\n"
  },
  {
    "path": "service/history/replication/dlq_handler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tdlqHandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t\tcontroller *gomock.Controller\n\n\t\tmockShard        *shard.TestContext\n\t\tconfig           *config.Config\n\t\tmockClientBean   *client.MockBean\n\t\tadminClient      *admin.MockClient\n\t\texecutionManager *mocks.ExecutionManager\n\t\tshardManager     *mocks.ShardManager\n\t\ttaskExecutor     *fakeTaskExecutor\n\t\ttaskExecutors    map[string]TaskExecutor\n\t\tsourceCluster    string\n\n\t\tmessageHandler *dlqHandlerImpl\n\t}\n)\n\nfunc TestDLQMessageHandlerSuite(t *testing.T) {\n\ts := new(dlqHandlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *dlqHandlerSuite) SetupSuite() {\n\n}\n\nfunc (s *dlqHandlerSuite) TearDownSuite() {\n\n}\n\nfunc (s *dlqHandlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.config = config.NewForTest()\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:                0,\n\t\t\tRangeID:                1,\n\t\t\tReplicationDLQAckLevel: map[string]int64{\"test\": -1},\n\t\t},\n\t\ts.config,\n\t)\n\n\ts.mockClientBean = s.mockShard.Resource.ClientBean\n\ts.adminClient = s.mockShard.Resource.RemoteAdminClient\n\ts.executionManager = s.mockShard.Resource.ExecutionMgr\n\ts.shardManager = s.mockShard.Resource.ShardMgr\n\n\ts.taskExecutors = make(map[string]TaskExecutor)\n\ts.taskExecutor = &fakeTaskExecutor{}\n\ts.sourceCluster = \"test\"\n\ts.taskExecutors[s.sourceCluster] = s.taskExecutor\n\n\ts.messageHandler = NewDLQHandler(\n\t\ts.mockShard,\n\t\ts.taskExecutors,\n\t).(*dlqHandlerImpl)\n}\n\nfunc (s *dlqHandlerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *dlqHandlerSuite) TestNewDLQHandler_panic() {\n\ts.Panics(func() { NewDLQHandler(s.mockShard, nil) }, \"Failed to initialize replication DLQ handler due to nil task executors\")\n}\n\nfunc (s *dlqHandlerSuite) TestStartStop() {\n\ttests := []struct {\n\t\tname   string\n\t\tstatus int32\n\t}{\n\t\t{\n\t\t\tname:   \"started\",\n\t\t\tstatus: common.DaemonStatusInitialized,\n\t\t},\n\t\t{\n\t\t\tname:   \"not started\",\n\t\t\tstatus: common.DaemonStatusStopped,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\tdefer goleak.VerifyNone(t)\n\n\t\t\ts.messageHandler.status = tc.status\n\n\t\t\ts.messageHandler.Start()\n\n\t\t\ts.messageHandler.Stop()\n\t\t})\n\t}\n}\n\nfunc (s *dlqHandlerSuite) TestGetMessageCount() {\n\tsize := int64(1)\n\ttests := []struct {\n\t\tname         string\n\t\tlatestCounts map[string]int64\n\t\tforceFetch   bool\n\t\terr          error\n\t}{\n\t\t{\n\t\t\tname:         \"success\",\n\t\t\tlatestCounts: map[string]int64{s.sourceCluster: size},\n\t\t},\n\t\t{\n\t\t\tname:       \"success with fetchAndEmitMessageCount call\",\n\t\t\tforceFetch: true,\n\t\t},\n\t\t{\n\t\t\tname:       \"error\",\n\t\t\tforceFetch: true,\n\t\t\terr:        errors.New(\"fetchAndEmitMessageCount error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ts.messageHandler.latestCounts = tc.latestCounts\n\n\t\t\tif tc.forceFetch || tc.latestCounts == nil {\n\t\t\t\ts.executionManager.On(\"GetReplicationDLQSize\", mock.Anything, mock.Anything).Return(&persistence.GetReplicationDLQSizeResponse{Size: size}, tc.err).Times(1)\n\t\t\t}\n\n\t\t\tcounts, err := s.messageHandler.GetMessageCount(context.Background(), tc.forceFetch)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else if tc.latestCounts != nil {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(size, counts[s.sourceCluster])\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *dlqHandlerSuite) TestFetchAndEmitMessageCount() {\n\ttests := []struct {\n\t\tname string\n\t\terr  error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\terr:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\terr:  errors.New(\"error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\tsize := int64(3)\n\t\t\trets := &persistence.GetReplicationDLQSizeResponse{Size: size}\n\t\t\ts.messageHandler.latestCounts = make(map[string]int64)\n\n\t\t\ts.executionManager.On(\"GetReplicationDLQSize\", context.Background(), mock.Anything).Return(rets, tc.err).Times(1)\n\n\t\t\terr := s.messageHandler.fetchAndEmitMessageCount(context.Background())\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(len(s.messageHandler.latestCounts), len(s.taskExecutors))\n\t\t\t\ts.Equal(size, s.messageHandler.latestCounts[s.sourceCluster])\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *dlqHandlerSuite) TestEmitDLQSizeMetricsLoop_FetchesAndEmitsMetricsPeriodically() {\n\tdefer goleak.VerifyNone(s.T())\n\n\temissionNumber := 2\n\n\ts.messageHandler.status = common.DaemonStatusStarted\n\ts.executionManager.On(\"GetReplicationDLQSize\", mock.Anything, mock.Anything).Return(&persistence.GetReplicationDLQSizeResponse{Size: 1}, nil).Times(emissionNumber)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\ts.messageHandler.timeSource = mockTimeSource\n\n\tgo s.messageHandler.emitDLQSizeMetricsLoop()\n\n\tfor i := 0; i < emissionNumber; i++ {\n\t\tmockTimeSource.BlockUntil(1)\n\n\t\t// Advance time to trigger the next emission\n\t\tmockTimeSource.Advance(dlqMetricsEmitTimerInterval + time.Duration(int64(float64(dlqMetricsEmitTimerInterval)*(1+dlqMetricsEmitTimerCoefficient))))\n\t}\n\n\ts.messageHandler.Stop()\n\n\ts.Equal(common.DaemonStatusStopped, s.messageHandler.status)\n}\n\nfunc (s *dlqHandlerSuite) TestReadMessages_OK() {\n\tctx := context.Background()\n\tlastMessageID := int64(1)\n\tpageSize := 1\n\tvar pageToken []byte\n\n\tresp := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   uuid.New(),\n\t\t\t\t\tWorkflowID: uuid.New(),\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         0,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}).Return(resp, nil).Times(1)\n\n\ts.mockClientBean.EXPECT().GetRemoteAdminClient(s.sourceCluster).Return(s.adminClient, nil).AnyTimes()\n\ts.adminClient.EXPECT().\n\t\tGetDLQReplicationMessages(ctx, gomock.Any()).\n\t\tReturn(&types.GetDLQReplicationMessagesResponse{}, nil)\n\ttasks, info, token, err := s.messageHandler.ReadMessages(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\ts.NoError(err)\n\ts.Nil(token)\n\ts.Equal(resp.Tasks[0].GetDomainID(), info[0].GetDomainID())\n\ts.Equal(resp.Tasks[0].GetWorkflowID(), info[0].GetWorkflowID())\n\ts.Equal(resp.Tasks[0].GetRunID(), info[0].GetRunID())\n\ts.Nil(tasks)\n}\n\nfunc (s *dlqHandlerSuite) TestReadMessagesWithAckLevel_OK() {\n\treplicationTasksResponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   \"domainID\",\n\t\t\t\t\tWorkflowID: \"workflowID\",\n\t\t\t\t\tRunID:      \"runID\",\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID:  123,\n\t\t\t\t\tVersion: 1,\n\t\t\t\t},\n\t\t\t\tFirstEventID: 1,\n\t\t\t\tNextEventID:  2,\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"token\"),\n\t}\n\n\tDLQReplicationMessagesResponse := &types.GetDLQReplicationMessagesResponse{\n\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t{\n\t\t\t\tSourceTaskID: 123,\n\t\t\t},\n\t\t},\n\t}\n\n\tctx := context.Background()\n\tlastMessageID := int64(123)\n\tpageSize := 12\n\tpageToken := []byte(\"token\")\n\n\treq := &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         defaultBeginningMessageID + 1,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}\n\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", ctx, req).Return(replicationTasksResponse, nil).Times(1)\n\n\ts.adminClient.EXPECT().\n\t\tGetDLQReplicationMessages(ctx, gomock.Any()).\n\t\tReturn(DLQReplicationMessagesResponse, nil).Times(1)\n\n\treplicationTasks, taskInfo, nextPageToken, err := s.messageHandler.readMessagesWithAckLevel(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\n\ts.NoError(err)\n\ts.Equal(replicationTasks, DLQReplicationMessagesResponse.ReplicationTasks)\n\ts.Len(taskInfo, len(replicationTasksResponse.Tasks))\n\t// testing content of taskInfo because it's assembled in the method using tasks from replicationTasksFromDLQ\n\tfor i, task := range taskInfo {\n\t\tt, err := replicationTasksResponse.Tasks[i].ToInternalReplicationTaskInfo()\n\t\ts.NoError(err)\n\t\ts.Equal(task, t)\n\t}\n\ts.Equal(nextPageToken, replicationTasksResponse.NextPageToken)\n}\n\nfunc (s *dlqHandlerSuite) TestReadMessagesWithAckLevel_GetReplicationTasksFromDLQFailed() {\n\terrorMessage := \"GetReplicationTasksFromDLQFailed\"\n\tctx := context.Background()\n\tlastMessageID := int64(123)\n\tpageSize := 12\n\tpageToken := []byte(\"token\")\n\n\treq := &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         defaultBeginningMessageID + 1,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}\n\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", ctx, req).Return(nil, errors.New(errorMessage)).Times(1)\n\n\t_, _, _, err := s.messageHandler.readMessagesWithAckLevel(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\n\ts.Error(err)\n\ts.Equal(err, errors.New(errorMessage))\n}\n\nfunc (s *dlqHandlerSuite) TestReadMessagesWithAckLevel_InvalidCluster() {\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, mock.Anything).Return(nil, nil).Times(1)\n\n\ts.mockShard.Resource.ClientBean = client.NewMockBean(s.controller)\n\ts.mockShard.Resource.ClientBean.EXPECT().GetRemoteAdminClient(\"invalidCluster\").Return(nil, errors.New(\"invalidCluster\")).Times(1)\n\n\t_, _, _, err := s.messageHandler.readMessagesWithAckLevel(context.Background(), \"invalidCluster\", 123, 12, []byte(\"token\"))\n\n\ts.Error(err)\n\ts.ErrorContains(err, \"invalidCluster\")\n}\n\nfunc (s *dlqHandlerSuite) TestReadMessagesWithAckLevel_GetDLQReplicationMessagesFailed() {\n\terrorMessage := \"GetDLQReplicationMessagesFailed\"\n\tctx := context.Background()\n\tlastMessageID := int64(123)\n\tpageSize := 12\n\tpageToken := []byte(\"token\")\n\n\treq := &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         defaultBeginningMessageID + 1,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}\n\n\treplicationTasksResponse := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.FailoverMarkerTask{\n\t\t\t\tDomainID: \"domainID\",\n\t\t\t},\n\t\t},\n\t}\n\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", ctx, req).Return(replicationTasksResponse, nil).Times(1)\n\n\ts.adminClient.EXPECT().\n\t\tGetDLQReplicationMessages(ctx, gomock.Any()).\n\t\tReturn(nil, errors.New(errorMessage)).Times(1)\n\n\t_, _, _, err := s.messageHandler.readMessagesWithAckLevel(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\n\ts.Error(err)\n\ts.Equal(err, errors.New(errorMessage))\n}\n\nfunc (s *dlqHandlerSuite) TestPurgeMessages() {\n\ttests := []struct {\n\t\tname string\n\t\terr  error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\terr:  errors.New(\"error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\tlastMessageID := int64(1)\n\t\t\ts.executionManager.On(\"RangeDeleteReplicationTaskFromDLQ\", mock.Anything,\n\t\t\t\t&persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\t\t\tSourceClusterName:    s.sourceCluster,\n\t\t\t\t\tInclusiveBeginTaskID: 0,\n\t\t\t\t\tExclusiveEndTaskID:   lastMessageID + 1,\n\t\t\t\t}).Return(&persistence.RangeDeleteReplicationTaskFromDLQResponse{TasksCompleted: persistence.UnknownNumRowsAffected}, tc.err).Times(1)\n\n\t\t\terr := s.messageHandler.PurgeMessages(context.Background(), s.sourceCluster, lastMessageID)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *dlqHandlerSuite) TestMergeMessages_OK() {\n\tctx := context.Background()\n\tlastMessageID := int64(2)\n\tpageSize := 1\n\tvar pageToken []byte\n\n\tresp := &persistence.GetHistoryTasksResponse{\n\t\tTasks: []persistence.Task{\n\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   uuid.New(),\n\t\t\t\t\tWorkflowID: uuid.New(),\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\t&persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   uuid.New(),\n\t\t\t\t\tWorkflowID: uuid.New(),\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID: 2,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         0,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}).Return(resp, nil).Times(1)\n\n\ts.mockClientBean.EXPECT().GetRemoteAdminClient(s.sourceCluster).Return(s.adminClient, nil).AnyTimes()\n\treplicationTask := &types.ReplicationTask{\n\t\tTaskType:     types.ReplicationTaskTypeHistory.Ptr(),\n\t\tSourceTaskID: 1,\n\t}\n\ts.adminClient.EXPECT().\n\t\tGetDLQReplicationMessages(ctx, gomock.Any()).\n\t\tReturn(&types.GetDLQReplicationMessagesResponse{\n\t\t\tReplicationTasks: []*types.ReplicationTask{replicationTask},\n\t\t}, nil)\n\ts.executionManager.On(\"RangeDeleteReplicationTaskFromDLQ\", mock.Anything,\n\t\t&persistence.RangeDeleteReplicationTaskFromDLQRequest{\n\t\t\tSourceClusterName:    s.sourceCluster,\n\t\t\tInclusiveBeginTaskID: 0,\n\t\t\tExclusiveEndTaskID:   lastMessageID + 1,\n\t\t}).Return(&persistence.RangeDeleteReplicationTaskFromDLQResponse{TasksCompleted: persistence.UnknownNumRowsAffected}, nil).Times(1)\n\n\ttoken, err := s.messageHandler.MergeMessages(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\ts.NoError(err)\n\ts.Nil(token)\n\ts.Equal(1, len(s.taskExecutor.executedTasks))\n}\n\nfunc (s *dlqHandlerSuite) TestMergeMessages_InvalidCluster() {\n\t_, err := s.messageHandler.MergeMessages(context.Background(), \"invalid\", 1, 1, nil)\n\ts.Error(err)\n\ts.Equal(errInvalidCluster, err)\n}\n\nfunc (s *dlqHandlerSuite) TestMergeMessages_GetReplicationTasksFromDLQFailed() {\n\terrorMessage := \"GetReplicationTasksFromDLQFailed\"\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, mock.Anything).Return(nil, errors.New(errorMessage)).Times(1)\n\t_, err := s.messageHandler.MergeMessages(context.Background(), s.sourceCluster, 1, 1, nil)\n\ts.Error(err)\n\ts.Equal(err, errors.New(errorMessage))\n}\n\nfunc (s *dlqHandlerSuite) TestMergeMessages_RangeDeleteReplicationTaskFromDLQFailed() {\n\terrorMessage := \"RangeDeleteReplicationTaskFromDLQFailed\"\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, mock.Anything).Return(&persistence.GetHistoryTasksResponse{}, nil).Times(1)\n\ts.executionManager.On(\"RangeDeleteReplicationTaskFromDLQ\", mock.Anything, mock.Anything).Return(nil, errors.New(errorMessage)).Times(1)\n\t_, err := s.messageHandler.MergeMessages(context.Background(), s.sourceCluster, 1, 1, nil)\n\ts.Error(err)\n\ts.Equal(err, errors.New(errorMessage))\n}\n\nfunc (s *dlqHandlerSuite) TestMergeMessages_executeFailed() {\n\terrorMessage := \"executeFailed\"\n\ts.taskExecutors[s.sourceCluster] = &fakeTaskExecutor{err: errors.New(errorMessage)}\n\n\tctx := context.Background()\n\tlastMessageID := int64(2)\n\tpageSize := 1\n\tvar pageToken []byte\n\n\ts.executionManager.On(\"GetReplicationTasksFromDLQ\", mock.Anything, &persistence.GetReplicationTasksFromDLQRequest{\n\t\tSourceClusterName: s.sourceCluster,\n\t\tReadLevel:         0,\n\t\tMaxReadLevel:      lastMessageID + 1,\n\t\tBatchSize:         pageSize,\n\t\tNextPageToken:     pageToken,\n\t}).Return(&persistence.GetHistoryTasksResponse{Tasks: []persistence.Task{&persistence.HistoryReplicationTask{TaskData: persistence.TaskData{TaskID: 1}}}}, nil).Times(1)\n\n\ts.adminClient.EXPECT().GetDLQReplicationMessages(ctx, gomock.Any()).\n\t\tReturn(&types.GetDLQReplicationMessagesResponse{ReplicationTasks: []*types.ReplicationTask{{SourceTaskID: 1}}}, nil)\n\n\t_, err := s.messageHandler.MergeMessages(ctx, s.sourceCluster, lastMessageID, pageSize, pageToken)\n\n\ts.Error(err)\n\ts.Equal(err, errors.New(errorMessage))\n}\n\ntype fakeTaskExecutor struct {\n\tscope metrics.ScopeIdx\n\terr   error\n\n\texecutedTasks []*types.ReplicationTask\n}\n\nfunc (e *fakeTaskExecutor) execute(replicationTask *types.ReplicationTask, _ bool) (metrics.ScopeIdx, error) {\n\te.executedTasks = append(e.executedTasks, replicationTask)\n\treturn e.scope, e.err\n}\n"
  },
  {
    "path": "service/history/replication/dynamic_task_batch_sizer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"strconv\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rangeiter\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\n// DynamicTaskBatchSizer is responsible for the batch size used to retrieve ReplicationTasks by TaskAckManager\n// It adjusts the task batch size based on the error and the getTasksResult,\n// and use the following rules:\n//\n//  1. If there is an error, decrease the task batch size.\n//     In case of an increased load to the database, we should reduce the load to the database\n//     Emitted metric with tag reason:\"error\"\n//\n//  2. If the task batch size is shrunk, decrease the task batch size.\n//     The payload size of messages is too big, so we should decrease\n//     the number of messages to be sure that future messages will not be shrunk\n//     Emitted metric with tag reason:\"shrunk\"\n//\n//  3. If the read level of a passive cluster has not been changed and there are no fetched tasks,\n//     not change the task batch size. There is no need to change because the replication is not stuck,\n//     there just are no new tasks\n//     Metric is not emitted\n//\n//  4. If the read level of a passive cluster has not been changed and if there are fetched tasks,\n//     and number of previously fetched tasks is not zero, decrease the task batch size.\n//     The replication is stuck on the passive side\n//     Emitted metric with tag reason:\"possible_stuck\"\n//\n//  5. If the read level of a passive cluster has not been changed and if there are fetched tasks,\n//     and number of previously fetched tasks is zero, not change the task batch size.\n//     The replication is not stuck, and there are new tasks to be replicated\n//     Metric is not emitted\n//\n//  6. If the read level of a passive cluster has been changed and if there are more tasks in db,\n//     increase the task batch size. We should retrieve the maximum possible value at the next time,\n//     as there are more tasks to be replicated\n//     Emitted metric with tag reason:\"more_tasks\"\n//\n//  7. If the read level of a passive cluster has been changed and if there are no more tasks in db,\n//     not change the size. The existing size is already enough, and there are no more tasks to be replicated\n//     Metric is not emitted\ntype DynamicTaskBatchSizer interface {\n\tanalyse(err error, state *getTasksResult)\n\tvalue() int\n}\n\n// dynamicTaskBatchSizerImpl is the implementation of DynamicTaskBatchSizer\ntype dynamicTaskBatchSizerImpl struct {\n\t// isFetchedTasks indicates that there are fetched tasks in the last GetTasks call\n\tisFetchedTasks atomic.Bool\n\titer           rangeiter.Iterator[int]\n\tlogger         log.Logger\n\tscope          metrics.Scope\n}\n\n// NewDynamicTaskBatchSizer creates a new dynamicTaskBatchSizerImpl\nfunc NewDynamicTaskBatchSizer(shardID int, logger log.Logger, config *config.Config, metricsClient metrics.Client) DynamicTaskBatchSizer {\n\tlogger = logger.WithTags(tag.ComponentReplicationDynamicTaskBatchSizer)\n\treturn &dynamicTaskBatchSizerImpl{\n\t\tlogger: logger,\n\t\tscope: metricsClient.Scope(\n\t\t\tmetrics.ReplicatorQueueProcessorScope,\n\t\t\tmetrics.InstanceTag(strconv.Itoa(shardID)),\n\t\t),\n\t\titer: rangeiter.NewDynamicConfigLinearIterator(\n\t\t\tfunc() int { return config.ReplicatorProcessorMinTaskBatchSize(shardID) },\n\t\t\tfunc() int { return config.ReplicatorProcessorMaxTaskBatchSize(shardID) },\n\t\t\tfunc() int { return config.ReplicatorProcessorBatchSizeStepCount(shardID) },\n\t\t\tlogger,\n\t\t),\n\t}\n}\n\nfunc (d *dynamicTaskBatchSizerImpl) analyse(err error, state *getTasksResult) {\n\tswitch {\n\tcase err != nil:\n\t\td.decrease(\"error\")\n\n\tcase state.isShrunk:\n\t\td.decrease(\"shrunk\")\n\n\tcase state.previousReadTaskID == state.lastReadTaskID &&\n\t\tlen(state.taskInfos) > 0 && d.isFetchedTasks.Load():\n\t\td.decrease(\"possible_stuck\")\n\n\tcase state.msgs.HasMore:\n\t\td.increase(\"more_tasks\")\n\t}\n\n\t// update isFetchedTasks\n\tif state == nil {\n\t\td.isFetchedTasks.Store(false)\n\t\treturn\n\t}\n\n\td.isFetchedTasks.Store(len(state.taskInfos) != 0)\n}\n\nfunc (d *dynamicTaskBatchSizerImpl) value() int {\n\treturn d.iter.Value()\n}\n\nfunc (d *dynamicTaskBatchSizerImpl) decrease(reason string) {\n\toldVal, newVal := d.iter.Value(), d.iter.Previous()\n\n\tif oldVal != newVal {\n\t\td.emitMetric(reason, \"decrease\")\n\t}\n\td.logger.Debug(\"Decrease task batch size\", tag.Reason(reason), tag.ReplicationTaskBatchSize(newVal))\n}\n\nfunc (d *dynamicTaskBatchSizerImpl) increase(reason string) {\n\toldVal, newVal := d.iter.Value(), d.iter.Next()\n\n\tif oldVal != newVal {\n\t\td.emitMetric(reason, \"increase\")\n\t}\n\td.logger.Debug(\"Increase task batch size\", tag.Reason(reason), tag.ReplicationTaskBatchSize(newVal))\n}\n\nfunc (d *dynamicTaskBatchSizerImpl) emitMetric(reason, decision string) {\n\td.scope.Tagged(\n\t\tmetrics.ReasonTag(reason),\n\t\tmetrics.DecisionTag(decision),\n\t).IncCounter(metrics.ReplicationDynamicTaskBatchSizerDecision)\n}\n"
  },
  {
    "path": "service/history/replication/dynamic_task_batch_sizer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\n// TestDynamicTaskBatchSizer checks that batch size is changed dynamically based on the results of GetTasks\nfunc TestDynamicTaskBatchSizer(t *testing.T) {\n\tvar (\n\t\t// range [0, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000]\n\t\t// start point: 500\n\t\tmin       = 0\n\t\tmax       = 1000\n\t\tstepCount = 11\n\n\t\tsizer = NewDynamicTaskBatchSizer(0, testlogger.New(t), &config.Config{\n\t\t\tReplicatorProcessorMinTaskBatchSize:   func(_ int) int { return min },\n\t\t\tReplicatorProcessorMaxTaskBatchSize:   func(_ int) int { return max },\n\t\t\tReplicatorProcessorBatchSizeStepCount: func(_ int) int { return stepCount },\n\t\t}, metrics.NewNoopMetricsClient()).(*dynamicTaskBatchSizerImpl)\n\t)\n\n\tt.Run(\"initial batch size\", func(t *testing.T) {\n\t\tassert.Equal(t, 500, sizer.value())\n\t})\n\n\tt.Run(\"an error occurred 3 times\", func(t *testing.T) {\n\t\terr := errors.New(\"error\")\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 400, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 300, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 200, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has changed 2 times and there are tasks in db\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 0,\n\t\t\tlastReadTaskID:     10,\n\t\t\ttaskInfos:          make([]persistence.Task, 10),\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: true,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 300, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 400, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has changed, there are tasks in db, but shrunk\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 0,\n\t\t\tlastReadTaskID:     10,\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: true,\n\t\t\t},\n\t\t\ttaskInfos: make([]persistence.Task, 10),\n\t\t\tisShrunk:  true,\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 300, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has not changed, replication tasks are returned\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 10,\n\t\t\tlastReadTaskID:     10,\n\t\t\ttaskInfos:          make([]persistence.Task, 10),\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: true,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 200, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has not changed, replication tasks are not returned\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 10,\n\t\t\tlastReadTaskID:     10,\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: false,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 200, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has changed 2 times and there are no tasks in db\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 0,\n\t\t\tlastReadTaskID:     10,\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: false,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 200, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 200, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"an error occurred 3 times\", func(t *testing.T) {\n\t\terr := errors.New(\"error\")\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 100, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 0, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\n\t\tsizer.analyse(err, nil)\n\t\tassert.Equal(t, 0, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"max has been changed to 500\", func(t *testing.T) {\n\t\t// range [0, 50, 100, 150, 200, 250, 300, 350, 400, 450, 500]\n\t\t// start point: 250\n\t\tmax = 500\n\n\t\tassert.Equal(t, 250, sizer.value())\n\t\tassert.False(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"shrunk 10 times\", func(t *testing.T) {\n\t\tvar want = []int{\n\t\t\t200, 150, 100, 50, 0,\n\t\t\t0, 0, 0, 0, 0,\n\t\t}\n\n\t\tfor i := 0; i < len(want); i++ {\n\t\t\tstate := &getTasksResult{\n\t\t\t\ttaskInfos: make([]persistence.Task, 10),\n\t\t\t\tisShrunk:  true,\n\t\t\t}\n\n\t\t\tsizer.analyse(nil, state)\n\t\t\tassert.Equal(t, want[i], sizer.value())\n\t\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t\t}\n\t})\n\n\tt.Run(\"read level has changed and there are tasks in db\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 0,\n\t\t\tlastReadTaskID:     10,\n\t\t\ttaskInfos:          make([]persistence.Task, 10),\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: true,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 50, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has changed and there are no tasks in db\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 0,\n\t\t\tlastReadTaskID:     10,\n\t\t\ttaskInfos:          make([]persistence.Task, 5),\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: false,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 50, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n\n\tt.Run(\"read level has not changed, there are returned tasks, previously tasks were returned\", func(t *testing.T) {\n\t\tstate := &getTasksResult{\n\t\t\tpreviousReadTaskID: 10,\n\t\t\tlastReadTaskID:     10,\n\t\t\ttaskInfos:          make([]persistence.Task, 5),\n\t\t\tmsgs: &types.ReplicationMessages{\n\t\t\t\tHasMore: false,\n\t\t\t},\n\t\t}\n\n\t\tsizer.analyse(nil, state)\n\t\tassert.Equal(t, 0, sizer.value())\n\t\tassert.True(t, sizer.isFetchedTasks.Load())\n\t})\n}\n"
  },
  {
    "path": "service/history/replication/metrics_emitter.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nconst (\n\tmetricsEmissionInterval = time.Minute\n)\n\ntype (\n\t// MetricsEmitterImpl is responsible for emitting source side replication metrics occasionally.\n\tMetricsEmitterImpl struct {\n\t\tshardID        int\n\t\tcurrentCluster string\n\t\tremoteClusters map[string]config.ClusterInformation\n\t\tshardData      metricsEmitterShardData\n\t\treader         taskReader\n\t\tscope          metrics.Scope\n\t\tlogger         log.Logger\n\t\tstatus         int32\n\t\tinterval       time.Duration\n\t\tctx            context.Context\n\t\tcancelCtx      context.CancelFunc\n\t\twg             sync.WaitGroup\n\t}\n\n\t// metricsEmitterShardData is for testing.\n\tmetricsEmitterShardData interface {\n\t\tGetLogger() log.Logger\n\t\tGetClusterMetadata() cluster.Metadata\n\t\tGetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey\n\t\tGetTimeSource() clock.TimeSource\n\t}\n)\n\n// NewMetricsEmitter creates a new metrics emitter, which starts a goroutine to emit replication metrics occasionally.\nfunc NewMetricsEmitter(\n\tshardID int,\n\tshardData metricsEmitterShardData,\n\treader taskReader,\n\tmetricsClient metrics.Client,\n) *MetricsEmitterImpl {\n\tcurrentCluster := shardData.GetClusterMetadata().GetCurrentClusterName()\n\tremoteClusters := shardData.GetClusterMetadata().GetRemoteClusterInfo()\n\n\tscope := metricsClient.Scope(\n\t\tmetrics.ReplicationMetricEmitterScope,\n\t\tmetrics.ActiveClusterTag(currentCluster),\n\t\tmetrics.InstanceTag(strconv.Itoa(shardID)),\n\t)\n\tlogger := shardData.GetLogger().WithTags(\n\t\ttag.ClusterName(currentCluster),\n\t\ttag.ShardID(shardID))\n\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &MetricsEmitterImpl{\n\t\tshardID:        shardID,\n\t\tcurrentCluster: currentCluster,\n\t\tremoteClusters: remoteClusters,\n\t\tstatus:         common.DaemonStatusInitialized,\n\t\tshardData:      shardData,\n\t\treader:         reader,\n\t\tscope:          scope,\n\t\tinterval:       metricsEmissionInterval,\n\t\tlogger:         logger,\n\t\tctx:            ctx,\n\t\tcancelCtx:      cancel,\n\t}\n}\n\nfunc (m *MetricsEmitterImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&m.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tm.wg.Add(1)\n\tgo m.emitMetricsLoop()\n\tm.logger.Info(\"ReplicationMetricsEmitter started.\")\n}\n\nfunc (m *MetricsEmitterImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&m.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tm.logger.Info(\"ReplicationMetricsEmitter shutting down.\")\n\tm.cancelCtx()\n\tif !common.AwaitWaitGroup(&m.wg, 5*time.Second) {\n\t\tm.logger.Warn(\"ReplicationMetricsEmitter timed out on shutdown.\")\n\t}\n}\n\nfunc (m *MetricsEmitterImpl) emitMetricsLoop() {\n\tdefer m.wg.Done()\n\n\tticker := time.NewTicker(m.interval)\n\tdefer ticker.Stop()\n\tdefer func() { log.CapturePanic(recover(), m.logger, nil) }()\n\n\tfor {\n\t\tselect {\n\t\tcase <-m.ctx.Done():\n\t\t\treturn\n\t\tcase <-ticker.C:\n\t\t\tm.emitMetrics()\n\t\t}\n\t}\n}\n\nfunc (m *MetricsEmitterImpl) emitMetrics() {\n\tfor remoteClusterName := range m.remoteClusters {\n\t\tlogger := m.logger.WithTags(tag.RemoteCluster(remoteClusterName))\n\t\tscope := m.scope.Tagged(metrics.TargetClusterTag(remoteClusterName))\n\n\t\treplicationLatency, err := m.determineReplicationLatency(remoteClusterName)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\n\t\tscope.UpdateGauge(metrics.ReplicationLatency, float64(replicationLatency.Nanoseconds()))\n\t\tlogger.Debug(fmt.Sprintf(\"ReplicationLatency metric emitted: %v\", float64(replicationLatency.Nanoseconds())))\n\t}\n}\n\nfunc (m *MetricsEmitterImpl) determineReplicationLatency(remoteClusterName string) (time.Duration, error) {\n\tlogger := m.logger.WithTags(tag.RemoteCluster(remoteClusterName))\n\tlastReadTaskID := m.shardData.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, remoteClusterName).GetTaskID()\n\n\ttasks, _, err := m.reader.Read(m.ctx, lastReadTaskID, lastReadTaskID+1, 1)\n\tif err != nil {\n\t\tlogger.Error(fmt.Sprintf(\n\t\t\t\"Error reading when determining replication latency, lastReadTaskID=%v\", lastReadTaskID),\n\t\t\ttag.Error(err))\n\t\treturn 0, err\n\t}\n\tlogger.Debug(\"Number of tasks retrieved\", tag.Number(int64(len(tasks))))\n\n\tvar replicationLatency time.Duration\n\tif len(tasks) > 0 {\n\t\tcreationTime := tasks[0].GetVisibilityTimestamp()\n\t\treplicationLatency = m.shardData.GetTimeSource().Now().Sub(creationTime)\n\t}\n\n\treturn replicationLatency, nil\n}\n"
  },
  {
    "path": "service/history/replication/metrics_emitter_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nvar (\n\tcluster1 = \"cluster1\"\n\tcluster2 = \"cluster2\"\n\tcluster3 = \"cluster3\"\n)\n\nfunc TestMetricsEmitterStartStop(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\n\ttimeSource := clock.NewMockedTimeSource()\n\tmetadata := newClusterMetadata(t)\n\ttestShardData := newTestShardData(timeSource, metadata)\n\n\tmetricsEmitter := NewMetricsEmitter(1, testShardData, fakeTaskReader{}, metrics.NewNoopMetricsClient())\n\tmetricsEmitter.interval = 5 * time.Millisecond\n\tmetricsEmitter.Start()\n\ttime.Sleep(20 * time.Millisecond) // let the metrics emitter run a few times\n\tmetricsEmitter.Stop()\n}\n\nfunc TestMetricsEmitter(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\tmetadata := newClusterMetadata(t)\n\ttestShardData := newTestShardData(timeSource, metadata)\n\n\ttask1 := &persistence.HistoryReplicationTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              1,\n\t\t\tVisibilityTimestamp: timeSource.Now().Add(-time.Hour),\n\t\t},\n\t}\n\ttask2 := &persistence.HistoryReplicationTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              2,\n\t\t\tVisibilityTimestamp: timeSource.Now().Add(-time.Minute),\n\t\t},\n\t}\n\treader := fakeTaskReader{task1, task2}\n\n\tmetricsEmitter := NewMetricsEmitter(1, testShardData, reader, metrics.NewNoopMetricsClient())\n\tlatency, err := metricsEmitter.determineReplicationLatency(cluster2)\n\tassert.NoError(t, err)\n\tassert.Equal(t, time.Hour, latency)\n\n\t// Move replication level up for cluster2 and our latency shortens\n\ttestShardData.clusterReplicationLevel[cluster2] = persistence.NewImmediateTaskKey(2)\n\tlatency, err = metricsEmitter.determineReplicationLatency(cluster2)\n\tassert.NoError(t, err)\n\tassert.Equal(t, time.Minute, latency)\n\n\t// Move replication level up for cluster2 and we no longer have latency\n\ttestShardData.clusterReplicationLevel[cluster2] = persistence.NewImmediateTaskKey(3)\n\tlatency, err = metricsEmitter.determineReplicationLatency(cluster2)\n\tassert.NoError(t, err)\n\tassert.Equal(t, time.Duration(0), latency)\n\n\t// Cluster3 will still have latency\n\tlatency, err = metricsEmitter.determineReplicationLatency(cluster3)\n\tassert.NoError(t, err)\n\tassert.Equal(t, time.Hour, latency)\n}\n\ntype testShardData struct {\n\tlogger                  log.Logger\n\tclusterReplicationLevel map[string]persistence.HistoryTaskKey\n\ttimeSource              clock.TimeSource\n\tmetadata                cluster.Metadata\n}\n\nfunc newTestShardData(timeSource clock.TimeSource, metadata cluster.Metadata) testShardData {\n\tremotes := metadata.GetRemoteClusterInfo()\n\tclusterReplicationLevels := make(map[string]persistence.HistoryTaskKey, len(remotes))\n\tfor remote := range remotes {\n\t\tclusterReplicationLevels[remote] = persistence.NewImmediateTaskKey(1)\n\t}\n\treturn testShardData{\n\t\tlogger:                  log.NewNoop(),\n\t\ttimeSource:              timeSource,\n\t\tmetadata:                metadata,\n\t\tclusterReplicationLevel: clusterReplicationLevels,\n\t}\n}\n\nfunc (t testShardData) GetLogger() log.Logger {\n\treturn t.logger\n}\n\nfunc (t testShardData) GetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\treturn t.clusterReplicationLevel[cluster]\n}\n\nfunc (t testShardData) GetTimeSource() clock.TimeSource {\n\treturn t.timeSource\n}\n\nfunc (t testShardData) GetClusterMetadata() cluster.Metadata {\n\treturn t.metadata\n}\n\nfunc newClusterMetadata(t *testing.T) cluster.Metadata {\n\treturn cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tPrimaryClusterName: cluster1,\n\t\t\tCurrentClusterName: cluster1,\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\tcluster1: {Enabled: true},\n\t\t\t\tcluster2: {Enabled: true},\n\t\t\t\tcluster3: {Enabled: true},\n\t\t\t},\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\ttestlogger.New(t),\n\t)\n}\n"
  },
  {
    "path": "service/history/replication/task_ack_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype (\n\t// TaskAckManager is the ack manager for replication tasks\n\tTaskAckManager struct {\n\t\tackLevels ackLevelStore\n\n\t\tscope  metrics.Scope\n\t\tlogger log.Logger\n\n\t\treader taskReader\n\t\tstore  *TaskStore\n\n\t\t// replicationMessagesSizeFn is the function to calculate the size of types.ReplicationMessages\n\t\treplicationMessagesSizeFn types.ReplicationMessagesSizeFn\n\n\t\t// maxReplicationMessagesSize is the max size of types.ReplicationMessages\n\t\t// that can be sent in a single RPC call\n\t\tmaxReplicationMessagesSize int\n\n\t\tdynamicTaskBatchSizer DynamicTaskBatchSizer\n\t\ttimeSource            clock.TimeSource\n\t}\n\n\tackLevelStore interface {\n\t\tUpdateIfNeededAndGetQueueMaxReadLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey\n\t\tGetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey\n\t\tUpdateQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string, ackLevel persistence.HistoryTaskKey) error\n\t}\n\ttaskReader interface {\n\t\tRead(ctx context.Context, readLevel int64, maxReadLevel int64, batchSize int) ([]persistence.Task, bool, error)\n\t}\n)\n\n// NewTaskAckManager initializes a new replication task ack manager\nfunc NewTaskAckManager(\n\tshardID int,\n\tackLevels ackLevelStore,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\treader taskReader,\n\tstore *TaskStore,\n\ttimeSource clock.TimeSource,\n\tconfig *config.Config,\n\treplicationMessagesSizeFn types.ReplicationMessagesSizeFn,\n\tdynamicTaskBatchSizer DynamicTaskBatchSizer,\n) TaskAckManager {\n\n\treturn TaskAckManager{\n\t\tackLevels: ackLevels,\n\t\tscope: metricsClient.Scope(\n\t\t\tmetrics.ReplicatorQueueProcessorScope,\n\t\t\tmetrics.InstanceTag(strconv.Itoa(shardID)),\n\t\t),\n\t\tlogger:     logger.WithTags(tag.ComponentReplicationAckManager),\n\t\treader:     reader,\n\t\tstore:      store,\n\t\ttimeSource: timeSource,\n\n\t\tmaxReplicationMessagesSize: config.MaxResponseSize,\n\t\treplicationMessagesSizeFn:  replicationMessagesSizeFn,\n\t\tdynamicTaskBatchSizer:      dynamicTaskBatchSizer,\n\t}\n}\n\nfunc (t *TaskAckManager) GetTasks(ctx context.Context, pollingCluster string, lastReadTaskID int64) (_ *types.ReplicationMessages, err error) {\n\tresult, err := t.getTasks(ctx, pollingCluster, lastReadTaskID)\n\tt.dynamicTaskBatchSizer.analyse(err, result)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn result.msgs, nil\n}\n\n// getTasksResult contains the result of a TaskAckManager.getTasks\n// It is used to adjust the task batch size by DynamicTaskBatchSizer\ntype getTasksResult struct {\n\tpreviousReadTaskID int64\n\tlastReadTaskID     int64\n\tmsgs               *types.ReplicationMessages\n\ttaskInfos          []persistence.Task\n\tisShrunk           bool\n}\n\nfunc (t *TaskAckManager) getTasks(ctx context.Context, pollingCluster string, lastReadTaskID int64) (*getTasksResult, error) {\n\tvar oldestUnprocessedTaskID int64\n\tvar oldestUnprocessedTaskTimestamp int64\n\tpreviousReadTaskID := t.ackLevels.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, pollingCluster).GetTaskID()\n\n\tif lastReadTaskID == constants.EmptyMessageID {\n\t\tlastReadTaskID = previousReadTaskID\n\t}\n\n\ttaskGeneratedStart := t.timeSource.Now()\n\ttaskGeneratedTimer := t.scope.StartTimer(metrics.TaskLatency)\n\tdefer taskGeneratedTimer.Stop()\n\tdefer func() {\n\t\tt.scope.ExponentialHistogram(metrics.ExponentialTaskLatency, t.timeSource.Since(taskGeneratedStart))\n\t}()\n\n\tbatchSize := t.dynamicTaskBatchSizer.value()\n\tt.scope.UpdateGauge(metrics.ReplicationTasksBatchSize, float64(batchSize))\n\n\ttaskInfos, hasMore, err := t.reader.Read(ctx, lastReadTaskID, t.ackLevels.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryReplication, pollingCluster).GetTaskID(), batchSize)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttasksFetched := len(taskInfos)\n\tt.scope.RecordTimer(metrics.ReplicationTasksFetched, time.Duration(tasksFetched))\n\tt.scope.RecordHistogramValue(metrics.ReplicationTasksFetchedHistogram, float64(tasksFetched))\n\n\t// Happy path assumption - we will push all tasks to replication tasks.\n\tmsgs := &types.ReplicationMessages{\n\t\tReplicationTasks:       make([]*types.ReplicationTask, 0, len(taskInfos)),\n\t\tLastRetrievedMessageID: lastReadTaskID,\n\t\tHasMore:                hasMore,\n\t}\n\n\tif len(taskInfos) > 0 {\n\t\t// it does not matter if we can process task or not, but we need to know what was the oldest task information we have read.\n\t\t// tasks must be ordered by taskID/time.\n\t\toldestUnprocessedTaskID = taskInfos[0].GetTaskID()\n\t\toldestUnprocessedTaskTimestamp = taskInfos[0].GetVisibilityTimestamp().UnixNano()\n\t} else {\n\t\t// if there are no replication tasks, use the max read level as the oldest unprocessed task\n\t\t// this means there are no tasks to process, so the lag is 0\n\t\toldestUnprocessedTaskID = t.ackLevels.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryReplication, pollingCluster).GetTaskID()\n\t\toldestUnprocessedTaskTimestamp = t.timeSource.Now().UnixNano()\n\t}\n\n\tlagRaw := int(t.ackLevels.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryReplication, pollingCluster).GetTaskID() - oldestUnprocessedTaskID)\n\tt.scope.RecordTimer(metrics.ReplicationTasksLagRaw, time.Duration(lagRaw))\n\tt.scope.RecordHistogramValue(metrics.ReplicationTasksLagRawHistogram, float64(lagRaw))\n\tt.scope.RecordHistogramDuration(metrics.ReplicationTasksDelay, time.Duration(oldestUnprocessedTaskTimestamp-t.timeSource.Now().UnixNano()))\n\n\t// hydrate the tasks\n\tfor _, info := range taskInfos {\n\t\ttask, err := t.store.Get(ctx, pollingCluster, info)\n\t\tif err != nil {\n\t\t\tif errors.As(err, new(*types.BadRequestError)) ||\n\t\t\t\terrors.As(err, new(*types.InternalDataInconsistencyError)) ||\n\t\t\t\terrors.As(err, new(*types.EntityNotExistsError)) {\n\t\t\t\tt.logger.Warn(\"Failed to get replication task.\", tag.Error(err))\n\t\t\t} else {\n\t\t\t\tt.logger.Error(\"Failed to get replication task. Return what we have so far.\", tag.Error(err))\n\t\t\t\tmsgs.HasMore = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tmsgs.LastRetrievedMessageID = info.GetTaskID()\n\t\tif task != nil {\n\t\t\tmsgs.ReplicationTasks = append(msgs.ReplicationTasks, task)\n\t\t}\n\t}\n\n\t// Sometimes the total size of replication tasks can be larger than the max response size\n\t// It caused the replication lag until history.replicatorTaskBatchSize is not adjusted to a smaller value\n\t// To prevent the lag and manual actions, we stop adding more tasks to the batch if the total size exceeds the limit\n\tisShrunk, err := t.shrinkMessagesBySize(msgs)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treplicationLag := int(t.ackLevels.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryReplication, pollingCluster).GetTaskID() - msgs.LastRetrievedMessageID)\n\tt.scope.RecordTimer(metrics.ReplicationTasksLag, time.Duration(replicationLag))\n\tt.scope.RecordHistogramValue(metrics.ReplicationTasksLagHistogram, float64(replicationLag))\n\n\ttasksReturned := len(msgs.ReplicationTasks)\n\tt.scope.RecordTimer(metrics.ReplicationTasksReturned, time.Duration(tasksReturned))\n\tt.scope.RecordHistogramValue(metrics.ReplicationTasksReturnedHistogram, float64(tasksReturned))\n\n\ttasksReturnedDiff := len(taskInfos) - len(msgs.ReplicationTasks)\n\tt.scope.RecordTimer(metrics.ReplicationTasksReturnedDiff, time.Duration(tasksReturnedDiff))\n\tt.scope.RecordHistogramValue(metrics.ReplicationTasksReturnedDiffHistogram, float64(tasksReturnedDiff))\n\n\tt.ackLevel(pollingCluster, lastReadTaskID)\n\n\tt.logger.Debug(\n\t\t\"Get replication tasks\",\n\t\ttag.SourceCluster(pollingCluster),\n\t\ttag.ShardReplicationAck(msgs.LastRetrievedMessageID),\n\t\ttag.ReadLevel(msgs.LastRetrievedMessageID),\n\t)\n\n\treturn &getTasksResult{\n\t\tpreviousReadTaskID: previousReadTaskID,\n\t\tlastReadTaskID:     lastReadTaskID,\n\t\tmsgs:               msgs,\n\t\ttaskInfos:          taskInfos,\n\t\tisShrunk:           isShrunk,\n\t}, nil\n}\n\n// ackLevel updates the ack level for the given cluster\nfunc (t *TaskAckManager) ackLevel(pollingCluster string, lastReadTaskID int64) {\n\tif err := t.ackLevels.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, pollingCluster, persistence.NewImmediateTaskKey(lastReadTaskID)); err != nil {\n\t\tt.logger.Error(\"error updating replication level for shard\", tag.Error(err), tag.OperationFailed)\n\t}\n\n\tif err := t.store.Ack(pollingCluster, lastReadTaskID); err != nil {\n\t\tt.logger.Error(\"error updating replication level for hydrated task store\", tag.Error(err), tag.OperationFailed)\n\t}\n}\n\n// shrinkMessagesBySize shrinks the replication messages by removing the last replication task until the total size is allowed\nfunc (t *TaskAckManager) shrinkMessagesBySize(msgs *types.ReplicationMessages) (bool, error) {\n\t// if there are no replication tasks, do nothing\n\tif len(msgs.ReplicationTasks) == 0 {\n\t\treturn false, nil\n\t}\n\n\tmaxSize := t.maxReplicationMessagesSize\n\tisShrunk := false\n\n\tfor {\n\t\ttotalSize := t.replicationMessagesSizeFn(msgs)\n\n\t\t// if the total size is allowed, return the replication messages\n\t\tif totalSize < maxSize {\n\t\t\treturn isShrunk, nil\n\t\t}\n\n\t\tlastTask := msgs.ReplicationTasks[len(msgs.ReplicationTasks)-1]\n\t\tt.logger.Warn(\"Replication messages size is too large. Shrinking the messages by removing the last replication task\",\n\t\t\ttag.ReplicationMessagesTotalSize(totalSize),\n\t\t\ttag.ReplicationMessagesMaxSize(maxSize),\n\t\t\ttag.ReplicationTaskID(lastTask.SourceTaskID),\n\t\t\ttag.ReplicationTaskCreationTime(lastTask.CreationTime),\n\t\t)\n\n\t\t// change HasMore to true to indicate that there are more tasks to be fetched\n\t\tmsgs.HasMore = true\n\n\t\t// remove the last replication task\n\t\tmsgs.ReplicationTasks = msgs.ReplicationTasks[:len(msgs.ReplicationTasks)-1]\n\n\t\t// set isShrunk to true to indicate that the replication messages have been shrunk\n\t\tisShrunk = true\n\n\t\t// should never happen, but just in case\n\t\t// if there are no more replication tasks, return an error\n\t\tif len(msgs.ReplicationTasks) == 0 {\n\t\t\treturn isShrunk, fmt.Errorf(\"replication messages size is too large and cannot be shrunk anymore, shard will be stuck until the message size is reduced or max size is increased\")\n\n\t\t}\n\n\t\t// update the last retrieved message ID to the new last task ID\n\t\tmsgs.LastRetrievedMessageID = msgs.ReplicationTasks[len(msgs.ReplicationTasks)-1].SourceTaskID\n\t}\n}\n"
  },
  {
    "path": "service/history/replication/task_ack_manager_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\nvar (\n\ttestTask11 = persistence.HistoryReplicationTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID: testDomainID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID: 11,\n\t\t},\n\t}\n\ttestTask12 = persistence.SyncActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID: testDomainID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID: 12,\n\t\t},\n\t}\n\ttestTask13 = persistence.HistoryReplicationTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID: testDomainID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID: 13,\n\t\t},\n\t}\n\ttestTask14 = persistence.FailoverMarkerTask{\n\t\tDomainID: testDomainID,\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID: 14,\n\t\t},\n\t}\n\ttestHydratedTask11 = types.ReplicationTask{SourceTaskID: 11, HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{DomainID: testDomainID}}\n\ttestHydratedTask12 = types.ReplicationTask{SourceTaskID: 12, SyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{DomainID: testDomainID}}\n\ttestHydratedTask13 = types.ReplicationTask{SourceTaskID: 13, HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{DomainID: testDomainID}}\n\ttestHydratedTask14 = types.ReplicationTask{SourceTaskID: 14, FailoverMarkerAttributes: &types.FailoverMarkerAttributes{DomainID: testDomainID}}\n\n\ttestHydratedTaskErrorRecoverable    = types.ReplicationTask{SourceTaskID: -100}\n\ttestHydratedTaskErrorNonRecoverable = types.ReplicationTask{SourceTaskID: -200}\n\n\ttestClusterA = \"cluster-A\"\n\ttestClusterB = \"cluster-B\"\n\ttestClusterC = \"cluster-C\"\n\n\ttestDomainName = \"test-domain-name\"\n\n\ttestDomain = cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testDomainID, Name: testDomainName},\n\t\t&persistence.DomainConfig{},\n\t\ttrue,\n\t\t&persistence.DomainReplicationConfig{Clusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testClusterA},\n\t\t}},\n\t\t0,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\n\ttestConfig = &config.Config{\n\t\tMaxResponseSize: testMaxResponseSize,\n\t}\n)\n\nconst (\n\ttestMaxResponseSize = 4 * 1024 * 1024 // 4MB\n)\n\nfunc TestTaskAckManager_GetTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tackLevels      *fakeAckLevelStore\n\t\tdomains        domainCache\n\t\treader         taskReader\n\t\thydrator       taskHydrator\n\t\tpollingCluster string\n\t\tlastReadLevel  int64\n\t\tbatchSize      fakeDynamicTaskBatchSizer\n\t\texpectResult   *types.ReplicationMessages\n\t\texpectErr      string\n\t\texpectAckLevel int64\n\t\tconfig         *config.Config\n\t}{\n\t\t{\n\t\t\tname: \"main flow - no replication tasks\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains:        fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:         fakeTaskReader{},\n\t\t\thydrator:       fakeTaskHydrator{},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{},\n\t\t\t\tLastRetrievedMessageID: 5,\n\t\t\t\tHasMore:                false,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"main flow - continues on recoverable error\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:  fakeTaskReader{&testTask11, &testTask12, &testTask13, &testTask14},\n\t\t\thydrator: fakeTaskHydrator{\n\t\t\t\ttestTask11.TaskID: testHydratedTask11,\n\t\t\t\ttestTask12.TaskID: testHydratedTask12,\n\t\t\t\ttestTask13.TaskID: testHydratedTaskErrorRecoverable, // Will continue hydrating beyond this point\n\t\t\t\ttestTask14.TaskID: testHydratedTask14,\n\t\t\t},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11, &testHydratedTask12, &testHydratedTask14},\n\t\t\t\tLastRetrievedMessageID: 14,\n\t\t\t\tHasMore:                false,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"main flow - stops at non recoverable error\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:  fakeTaskReader{&testTask11, &testTask12, &testTask13, &testTask14},\n\t\t\thydrator: fakeTaskHydrator{\n\t\t\t\ttestTask11.TaskID: testHydratedTask11,\n\t\t\t\ttestTask12.TaskID: testHydratedTask12,\n\t\t\t\ttestTask13.TaskID: testHydratedTaskErrorNonRecoverable, // Will stop hydrating beyond this point\n\t\t\t\ttestTask14.TaskID: testHydratedTask14,\n\t\t\t},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11, &testHydratedTask12},\n\t\t\t\tLastRetrievedMessageID: 12,\n\t\t\t\tHasMore:                true,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"main flow - stops at second task, batch size is 2\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:  fakeTaskReader{&testTask11, &testTask12, &testTask13, &testTask14},\n\t\t\thydrator: fakeTaskHydrator{\n\t\t\t\ttestTask11.TaskID: testHydratedTask11,\n\t\t\t\ttestTask12.TaskID: testHydratedTask12, // Will stop hydrating beyond this point\n\t\t\t\ttestTask13.TaskID: testHydratedTask13,\n\t\t\t\ttestTask14.TaskID: testHydratedTask14,\n\t\t\t},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      2,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11, &testHydratedTask12},\n\t\t\t\tLastRetrievedMessageID: 12,\n\t\t\t\tHasMore:                true,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"main flow - stops at a message exceeded max response size\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:  fakeTaskReader{&testTask11, &testTask12, &testTask13, &testTask14},\n\t\t\thydrator: fakeTaskHydrator{\n\t\t\t\ttestTask11.TaskID: testHydratedTask11,\n\t\t\t\ttestTask12.TaskID: testHydratedTask12,\n\t\t\t\ttestTask13.TaskID: testHydratedTask13,\n\t\t\t\ttestTask14.TaskID: testHydratedTask14, // Will stop adding tasks beyond this point\n\t\t\t},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11, &testHydratedTask12, &testHydratedTask13},\n\t\t\t\tLastRetrievedMessageID: 13,\n\t\t\t\tHasMore:                true,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tbatchSize:      10,\n\t\t\tconfig: &config.Config{\n\t\t\t\tMaxResponseSize: proto.FromReplicationMessages(&types.ReplicationMessages{\n\t\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11, &testHydratedTask12, &testHydratedTask13},\n\t\t\t\t\tLastRetrievedMessageID: 13,\n\t\t\t\t\tHasMore:                true,\n\t\t\t\t}).Size() + 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"main flow - fail at a message exceeded max response size\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:  fakeTaskReader{&testTask11, &testTask12, &testTask13, &testTask14},\n\t\t\thydrator: fakeTaskHydrator{\n\t\t\t\ttestTask11.TaskID: testHydratedTask11,\n\t\t\t\ttestTask12.TaskID: testHydratedTask12,\n\t\t\t\ttestTask13.TaskID: testHydratedTask13,\n\t\t\t\ttestTask14.TaskID: testHydratedTask14, // Will stop adding tasks beyond this point\n\t\t\t},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult:   nil,\n\t\t\texpectErr:      \"replication messages size is too large and cannot be shrunk anymore, shard will be stuck until the message size is reduced or max size is increased\",\n\t\t\texpectAckLevel: 2,\n\t\t\tconfig: &config.Config{\n\t\t\t\tMaxResponseSize: 0,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"skips tasks for domains non belonging to polling cluster\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains:        fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:         fakeTaskReader{&testTask11},\n\t\t\thydrator:       fakeTaskHydrator{testTask11.TaskID: testHydratedTask11},\n\t\t\tpollingCluster: testClusterB,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{},\n\t\t\t\tLastRetrievedMessageID: 11,\n\t\t\t\tHasMore:                false,\n\t\t\t},\n\t\t\texpectAckLevel: 5,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"uses remote ack level for first fetch (empty task ID)\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(12)},\n\t\t\t},\n\t\t\tdomains:        fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:         fakeTaskReader{&testTask11, &testTask12},\n\t\t\thydrator:       fakeTaskHydrator{testTask12.TaskID: testHydratedTask12},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  -1,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask12},\n\t\t\t\tLastRetrievedMessageID: 12,\n\t\t\t\tHasMore:                false,\n\t\t\t},\n\t\t\texpectAckLevel: 12,\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to read replication tasks - return error\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\treader:         (fakeTaskReader)(nil),\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectErr:      \"error reading replication tasks\",\n\t\t\tconfig:         testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get domain - stops\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t},\n\t\t\tdomains:        fakeDomainCache{},\n\t\t\treader:         fakeTaskReader{&testTask11},\n\t\t\thydrator:       fakeTaskHydrator{},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{},\n\t\t\t\tLastRetrievedMessageID: 5,\n\t\t\t\tHasMore:                true,\n\t\t\t},\n\t\t\tconfig: testConfig,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to update ack level - no error, return response anyway\",\n\t\t\tackLevels: &fakeAckLevelStore{\n\t\t\t\treadLevel: 200,\n\t\t\t\tremote:    map[string]persistence.HistoryTaskKey{testClusterA: persistence.NewImmediateTaskKey(2)},\n\t\t\t\tupdateErr: errors.New(\"error update ack level\"),\n\t\t\t},\n\t\t\tdomains:        fakeDomainCache{testDomainID: testDomain},\n\t\t\treader:         fakeTaskReader{&testTask11},\n\t\t\thydrator:       fakeTaskHydrator{testTask11.TaskID: testHydratedTask11},\n\t\t\tpollingCluster: testClusterA,\n\t\t\tlastReadLevel:  5,\n\t\t\tbatchSize:      10,\n\t\t\texpectResult: &types.ReplicationMessages{\n\t\t\t\tReplicationTasks:       []*types.ReplicationTask{&testHydratedTask11},\n\t\t\t\tLastRetrievedMessageID: 11,\n\t\t\t\tHasMore:                false,\n\t\t\t},\n\t\t\tconfig: testConfig,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttaskStore := createTestTaskStore(t, tt.domains, tt.hydrator)\n\t\t\tackManager := NewTaskAckManager(\n\t\t\t\ttestShardID,\n\t\t\t\ttt.ackLevels,\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t\ttt.reader,\n\t\t\t\ttaskStore,\n\t\t\t\tclock.NewMockedTimeSource(),\n\t\t\t\ttt.config,\n\t\t\t\tproto.ReplicationMessagesSize,\n\t\t\t\ttt.batchSize,\n\t\t\t)\n\t\t\tresult, err := ackManager.GetTasks(context.Background(), tt.pollingCluster, tt.lastReadLevel)\n\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectResult, result)\n\t\t\t}\n\n\t\t\tif tt.expectAckLevel != 0 {\n\t\t\t\tassert.Equal(t, tt.expectAckLevel, tt.ackLevels.remote[tt.pollingCluster].GetTaskID())\n\t\t\t}\n\t\t})\n\t}\n}\n\ntype fakeAckLevelStore struct {\n\tremote    map[string]persistence.HistoryTaskKey\n\treadLevel int64\n\tupdateErr error\n}\n\nfunc (s *fakeAckLevelStore) UpdateIfNeededAndGetQueueMaxReadLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\treturn persistence.NewImmediateTaskKey(s.readLevel)\n}\nfunc (s *fakeAckLevelStore) GetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\treturn s.remote[cluster]\n}\nfunc (s *fakeAckLevelStore) UpdateQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string, lastTaskID persistence.HistoryTaskKey) error {\n\ts.remote[cluster] = lastTaskID\n\treturn s.updateErr\n}\n\ntype fakeTaskReader []persistence.Task\n\nfunc (r fakeTaskReader) Read(ctx context.Context, readLevel int64, maxReadLevel int64, batchSize int) ([]persistence.Task, bool, error) {\n\tif r == nil {\n\t\treturn nil, false, errors.New(\"error reading replication tasks\")\n\t}\n\n\thasMore := false\n\tvar result []persistence.Task\n\tfor _, task := range r {\n\t\tif task.GetTaskID() < readLevel {\n\t\t\tcontinue\n\t\t}\n\t\tif task.GetTaskID() >= maxReadLevel {\n\t\t\thasMore = true\n\t\t\tbreak\n\t\t}\n\t\tresult = append(result, task)\n\t}\n\n\tif len(result) > batchSize {\n\t\treturn result[:batchSize], true, nil\n\t}\n\n\treturn result, hasMore, nil\n}\n\ntype fakeTaskHydrator map[int64]types.ReplicationTask\n\nfunc (h fakeTaskHydrator) Hydrate(ctx context.Context, task persistence.Task) (*types.ReplicationTask, error) {\n\tif hydratedTask, ok := h[task.GetTaskID()]; ok {\n\t\tif hydratedTask == testHydratedTaskErrorNonRecoverable {\n\t\t\treturn nil, errors.New(\"error hydrating task\")\n\t\t}\n\t\tif hydratedTask == testHydratedTaskErrorRecoverable {\n\t\t\treturn nil, &types.EntityNotExistsError{}\n\t\t}\n\t\treturn &hydratedTask, nil\n\t}\n\tpanic(\"fix the test, should not reach this\")\n}\n\ntype fakeDynamicTaskBatchSizer int\n\nfunc (s fakeDynamicTaskBatchSizer) analyse(_ error, _ *getTasksResult) {}\nfunc (s fakeDynamicTaskBatchSizer) value() int                         { return int(s) }\n"
  },
  {
    "path": "service/history/replication/task_executor.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination task_executor_mock.go -self_package github.com/uber/cadence/service/history/replication\n\ntype (\n\t// TaskExecutor is the executor for replication task\n\tTaskExecutor interface {\n\t\texecute(replicationTask *types.ReplicationTask, forceApply bool) (metrics.ScopeIdx, error)\n\t}\n\n\ttaskExecutorImpl struct {\n\t\tcurrentCluster  string\n\t\tsourceCluster   string\n\t\tshard           shard.Context\n\t\tdomainCache     cache.DomainCache\n\t\thistoryResender ndc.HistoryResender\n\t\thistoryEngine   engine.Engine\n\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t}\n)\n\nvar _ TaskExecutor = (*taskExecutorImpl)(nil)\n\n// NewTaskExecutor creates an replication task executor\n// The executor uses by 1) DLQ replication task handler 2) history replication task processor\nfunc NewTaskExecutor(\n\tsourceCluster string,\n\tshard shard.Context,\n\tdomainCache cache.DomainCache,\n\thistoryResender ndc.HistoryResender,\n\thistoryEngine engine.Engine,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) TaskExecutor {\n\treturn &taskExecutorImpl{\n\t\tcurrentCluster:  shard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tsourceCluster:   sourceCluster,\n\t\tshard:           shard,\n\t\tdomainCache:     domainCache,\n\t\thistoryResender: historyResender,\n\t\thistoryEngine:   historyEngine,\n\t\tmetricsClient:   metricsClient,\n\t\tlogger:          logger,\n\t}\n}\n\nfunc (e *taskExecutorImpl) execute(\n\treplicationTask *types.ReplicationTask,\n\tforceApply bool,\n) (metrics.ScopeIdx, error) {\n\n\tvar err error\n\tvar scope metrics.ScopeIdx\n\tswitch replicationTask.GetTaskType() {\n\tcase types.ReplicationTaskTypeSyncActivity:\n\t\tscope = metrics.SyncActivityTaskScope\n\t\terr = e.handleActivityTask(replicationTask, forceApply)\n\tcase types.ReplicationTaskTypeHistoryV2:\n\t\tscope = metrics.HistoryReplicationV2TaskScope\n\t\terr = e.handleHistoryReplicationTaskV2(replicationTask, forceApply)\n\tcase types.ReplicationTaskTypeFailoverMarker:\n\t\tscope = metrics.FailoverMarkerScope\n\t\terr = e.handleFailoverReplicationTask(replicationTask)\n\tdefault:\n\t\te.logger.Error(\"Unknown task type.\")\n\t\tscope = metrics.ReplicatorScope\n\t\terr = ErrUnknownReplicationTask\n\t}\n\n\treturn scope, err\n}\n\nfunc (e *taskExecutorImpl) handleActivityTask(\n\ttask *types.ReplicationTask,\n\tforceApply bool,\n) (err error) {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tattr := task.SyncActivityTaskAttributes\n\t\t\te.logger.Error(\n\t\t\t\t\"handleActivityTask encountered panic.\",\n\t\t\t\ttag.WorkflowDomainID(attr.GetDomainID()),\n\t\t\t\ttag.WorkflowID(attr.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(attr.GetRunID()),\n\t\t\t\ttag.Value(r),\n\t\t\t)\n\t\t\tpanic(r)\n\t\t}\n\t}()\n\n\tattr := task.SyncActivityTaskAttributes\n\tdoContinue, err := e.filterTask(attr.GetDomainID(), forceApply)\n\tif err != nil || !doContinue {\n\t\treturn err\n\t}\n\n\treplicationStopWatch := e.metricsClient.StartTimer(metrics.SyncActivityTaskScope, metrics.CadenceLatency)\n\tdefer replicationStopWatch.Stop()\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:           attr.DomainID,\n\t\tWorkflowID:         attr.WorkflowID,\n\t\tRunID:              attr.RunID,\n\t\tVersion:            attr.Version,\n\t\tScheduledID:        attr.ScheduledID,\n\t\tScheduledTime:      attr.ScheduledTime,\n\t\tStartedID:          attr.StartedID,\n\t\tStartedTime:        attr.StartedTime,\n\t\tLastHeartbeatTime:  attr.LastHeartbeatTime,\n\t\tDetails:            attr.Details,\n\t\tAttempt:            attr.Attempt,\n\t\tLastFailureReason:  attr.LastFailureReason,\n\t\tLastFailureDetails: attr.LastFailureDetails,\n\t\tLastWorkerIdentity: attr.LastWorkerIdentity,\n\t\tVersionHistory:     attr.GetVersionHistory(),\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), replicationTimeout)\n\tdefer cancel()\n\n\tvar syncActivityAction func() error\n\t// Check if the number of shards between clusters are equal. If not, redirect the request.\n\tif e.shard.GetShardID() != common.WorkflowIDToHistoryShard(attr.WorkflowID, e.shard.GetConfig().NumberOfShards) {\n\t\tsyncActivityAction = func() error {\n\t\t\treturn e.shard.GetService().GetClientBean().GetHistoryClient().SyncActivity(ctx, request)\n\t\t}\n\t} else {\n\t\tsyncActivityAction = func() error {\n\t\t\treturn e.historyEngine.SyncActivity(ctx, request)\n\t\t}\n\t}\n\n\terr = syncActivityAction()\n\tretryErr, ok := toRetryTaskV2Error(err)\n\tif !ok {\n\t\treturn err\n\t}\n\t// Handle resend error\n\te.metricsClient.IncCounter(metrics.HistoryRereplicationByActivityReplicationScope, metrics.CadenceClientRequests)\n\tstopwatch := e.metricsClient.StartTimer(metrics.HistoryRereplicationByActivityReplicationScope, metrics.CadenceClientLatency)\n\tdefer stopwatch.Stop()\n\n\tresendErr := e.historyResender.SendSingleWorkflowHistory(\n\t\te.sourceCluster,\n\t\tretryErr.GetDomainID(),\n\t\tretryErr.GetWorkflowID(),\n\t\tretryErr.GetRunID(),\n\t\tretryErr.StartEventID,\n\t\tretryErr.StartEventVersion,\n\t\tretryErr.EndEventID,\n\t\tretryErr.EndEventVersion,\n\t)\n\tswitch {\n\tcase resendErr == nil:\n\t\tbreak\n\tcase resendErr == ndc.ErrSkipTask:\n\t\te.logger.Error(\n\t\t\t\"skip replication sync activity task\",\n\t\t\ttag.WorkflowDomainID(retryErr.GetDomainID()),\n\t\t\ttag.WorkflowID(retryErr.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(retryErr.GetRunID()),\n\t\t)\n\t\treturn nil\n\tdefault:\n\t\te.logger.Error(\n\t\t\t\"error resend history for sync activity\",\n\t\t\ttag.WorkflowDomainID(retryErr.GetDomainID()),\n\t\t\ttag.WorkflowID(retryErr.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(retryErr.GetRunID()),\n\t\t\ttag.Error(resendErr),\n\t\t)\n\t\t// should return the replication error, not the resending error\n\t\treturn err\n\t}\n\t// should try again after back fill the history\n\treturn syncActivityAction()\n}\n\nfunc (e *taskExecutorImpl) handleHistoryReplicationTaskV2(\n\ttask *types.ReplicationTask,\n\tforceApply bool,\n) (err error) {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tattr := task.HistoryTaskV2Attributes\n\t\t\te.logger.Error(\n\t\t\t\t\"handleHistoryReplicationTaskV2 encountered panic.\",\n\t\t\t\ttag.WorkflowDomainID(attr.GetDomainID()),\n\t\t\t\ttag.WorkflowID(attr.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(attr.GetRunID()),\n\t\t\t\ttag.Value(r),\n\t\t\t)\n\t\t\tpanic(r)\n\t\t}\n\t}()\n\n\tattr := task.HistoryTaskV2Attributes\n\tdoContinue, err := e.filterTask(attr.GetDomainID(), forceApply)\n\tif err != nil || !doContinue {\n\t\treturn err\n\t}\n\n\treplicationStopWatch := e.metricsClient.StartTimer(metrics.HistoryReplicationV2TaskScope, metrics.CadenceLatency)\n\tdefer replicationStopWatch.Stop()\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: attr.DomainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: attr.WorkflowID,\n\t\t\tRunID:      attr.RunID,\n\t\t},\n\t\tVersionHistoryItems: attr.VersionHistoryItems,\n\t\tEvents:              attr.Events,\n\t\t// new run events does not need version history since there is no prior events\n\t\tNewRunEvents: attr.NewRunEvents,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), replicationTimeout)\n\tdefer cancel()\n\n\tvar historyReplicationAction func() error\n\t// Check if the number of shards between clusters are equal. If not, redirect the request.\n\tif e.shard.GetShardID() != common.WorkflowIDToHistoryShard(attr.WorkflowID, e.shard.GetConfig().NumberOfShards) {\n\t\thistoryReplicationAction = func() error {\n\t\t\treturn e.shard.GetService().GetClientBean().GetHistoryClient().ReplicateEventsV2(ctx, request)\n\t\t}\n\t} else {\n\t\thistoryReplicationAction = func() error {\n\t\t\treturn e.historyEngine.ReplicateEventsV2(ctx, request)\n\t\t}\n\t}\n\n\terr = historyReplicationAction()\n\tretryErr, ok := toRetryTaskV2Error(err)\n\tif !ok {\n\t\treturn err\n\t}\n\te.metricsClient.IncCounter(metrics.HistoryRereplicationByHistoryReplicationScope, metrics.CadenceClientRequests)\n\tresendStopWatch := e.metricsClient.StartTimer(metrics.HistoryRereplicationByHistoryReplicationScope, metrics.CadenceClientLatency)\n\tdefer resendStopWatch.Stop()\n\n\tresendErr := e.historyResender.SendSingleWorkflowHistory(\n\t\te.sourceCluster,\n\t\tretryErr.GetDomainID(),\n\t\tretryErr.GetWorkflowID(),\n\t\tretryErr.GetRunID(),\n\t\tretryErr.StartEventID,\n\t\tretryErr.StartEventVersion,\n\t\tretryErr.EndEventID,\n\t\tretryErr.EndEventVersion,\n\t)\n\tswitch {\n\tcase resendErr == nil:\n\t\tbreak\n\tcase resendErr == ndc.ErrSkipTask:\n\t\te.logger.Error(\n\t\t\t\"skip replication history task\",\n\t\t\ttag.WorkflowDomainID(retryErr.GetDomainID()),\n\t\t\ttag.WorkflowID(retryErr.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(retryErr.GetRunID()),\n\t\t)\n\t\treturn nil\n\tdefault:\n\t\te.logger.Error(\n\t\t\t\"error resend history for history event v2\",\n\t\t\ttag.WorkflowDomainID(retryErr.GetDomainID()),\n\t\t\ttag.WorkflowID(retryErr.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(retryErr.GetRunID()),\n\t\t\ttag.Error(resendErr),\n\t\t)\n\t\t// should return the replication error, not the resending error\n\t\treturn err\n\t}\n\n\treturn historyReplicationAction()\n}\n\nfunc (e *taskExecutorImpl) handleFailoverReplicationTask(\n\ttask *types.ReplicationTask,\n) error {\n\tfailoverAttributes := task.GetFailoverMarkerAttributes()\n\tfailoverAttributes.CreationTime = task.CreationTime\n\treturn e.shard.AddingPendingFailoverMarker(failoverAttributes)\n}\n\nfunc (e *taskExecutorImpl) filterTask(\n\tdomainID string,\n\tforceApply bool,\n) (bool, error) {\n\tif forceApply {\n\t\treturn true, nil\n\t}\n\tdomainEntry, err := e.domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tshouldProcessTask := false\n\tfor _, targetCluster := range domainEntry.GetReplicationConfig().Clusters {\n\t\tif e.currentCluster == targetCluster.ClusterName {\n\t\t\tshouldProcessTask = true\n\t\t\tbreak\n\t\t}\n\t}\n\treturn shouldProcessTask, nil\n}\n\nfunc toRetryTaskV2Error(err error) (*types.RetryTaskV2Error, bool) {\n\tretError, ok := err.(*types.RetryTaskV2Error)\n\treturn retError, ok\n}\n"
  },
  {
    "path": "service/history/replication/task_executor_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: task_executor.go\n//\n// Generated by this command:\n//\n//\tmockgen -package replication -source task_executor.go -destination task_executor_mock.go -self_package github.com/uber/cadence/service/history/replication\n//\n\n// Package replication is a generated GoMock package.\npackage replication\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockTaskExecutor is a mock of TaskExecutor interface.\ntype MockTaskExecutor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskExecutorMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskExecutorMockRecorder is the mock recorder for MockTaskExecutor.\ntype MockTaskExecutorMockRecorder struct {\n\tmock *MockTaskExecutor\n}\n\n// NewMockTaskExecutor creates a new mock instance.\nfunc NewMockTaskExecutor(ctrl *gomock.Controller) *MockTaskExecutor {\n\tmock := &MockTaskExecutor{ctrl: ctrl}\n\tmock.recorder = &MockTaskExecutorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskExecutor) EXPECT() *MockTaskExecutorMockRecorder {\n\treturn m.recorder\n}\n\n// execute mocks base method.\nfunc (m *MockTaskExecutor) execute(replicationTask *types.ReplicationTask, forceApply bool) (metrics.ScopeIdx, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"execute\", replicationTask, forceApply)\n\tret0, _ := ret[0].(metrics.ScopeIdx)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// execute indicates an expected call of execute.\nfunc (mr *MockTaskExecutorMockRecorder) execute(replicationTask, forceApply any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"execute\", reflect.TypeOf((*MockTaskExecutor)(nil).execute), replicationTask, forceApply)\n}\n"
  },
  {
    "path": "service/history/replication/task_executor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\thistoryClient \"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\ttaskExecutorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t\tcontroller *gomock.Controller\n\n\t\tsourceCluster      string\n\t\tmockShard          *shard.TestContext\n\t\tmockEngine         *engine.MockEngine\n\t\thistoryClient      *historyClient.MockClient\n\t\tconfig             *config.Config\n\t\tmockDomainCache    *cache.MockDomainCache\n\t\tmockClientBean     *client.MockBean\n\t\tadminClient        *admin.MockClient\n\t\texecutionManager   *mocks.ExecutionManager\n\t\tnDCHistoryResender *ndc.MockHistoryResender\n\n\t\ttaskHandler *taskExecutorImpl\n\t}\n)\n\nfunc TestTaskExecutorSuite(t *testing.T) {\n\ts := new(taskExecutorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *taskExecutorSuite) SetupSuite() {\n\n}\n\nfunc (s *taskExecutorSuite) TearDownSuite() {\n\n}\n\nfunc (s *taskExecutorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.config = config.NewForTestByShardNumber(2)\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:                0,\n\t\t\tRangeID:                1,\n\t\t\tReplicationAckLevel:    0,\n\t\t\tReplicationDLQAckLevel: map[string]int64{\"test\": -1},\n\t\t},\n\t\ts.config,\n\t)\n\n\ts.sourceCluster = \"source-cluster\"\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockClientBean = s.mockShard.Resource.ClientBean\n\ts.adminClient = s.mockShard.Resource.RemoteAdminClient\n\ts.executionManager = s.mockShard.Resource.ExecutionMgr\n\ts.nDCHistoryResender = ndc.NewMockHistoryResender(s.controller)\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.historyClient = s.mockShard.Resource.HistoryClient\n\n\ts.taskHandler = NewTaskExecutor(\n\t\ts.sourceCluster,\n\t\ts.mockShard,\n\t\ts.mockDomainCache,\n\t\ts.nDCHistoryResender,\n\t\ts.mockEngine,\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\ts.mockShard.GetLogger(),\n\t).(*taskExecutorImpl)\n}\n\nfunc (s *taskExecutorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *taskExecutorSuite) TestConvertRetryTaskV2Error_OK() {\n\terr := &types.RetryTaskV2Error{}\n\t_, ok := toRetryTaskV2Error(err)\n\ts.True(ok)\n}\n\nfunc (s *taskExecutorSuite) TestConvertRetryTaskV2Error_NotOK() {\n\terr := &types.BadRequestError{}\n\t_, ok := toRetryTaskV2Error(err)\n\ts.False(ok)\n}\n\nfunc (s *taskExecutorSuite) TestFilterTask() {\n\tdomainID := uuid.New()\n\ts.mockDomainCache.EXPECT().\n\t\tGetDomainByID(domainID).\n\t\tReturn(cache.NewGlobalDomainCacheEntryForTest(\n\t\t\tnil,\n\t\t\tnil,\n\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t\t{\n\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t},\n\t\t\t\t}},\n\t\t\t0,\n\t\t), nil)\n\tok, err := s.taskHandler.filterTask(domainID, false)\n\ts.NoError(err)\n\ts.True(ok)\n}\n\nfunc (s *taskExecutorSuite) TestFilterTask_Error() {\n\tdomainID := uuid.New()\n\ts.mockDomainCache.EXPECT().\n\t\tGetDomainByID(domainID).\n\t\tReturn(nil, fmt.Errorf(\"test\"))\n\tok, err := s.taskHandler.filterTask(domainID, false)\n\ts.Error(err)\n\ts.False(ok)\n}\n\nfunc (s *taskExecutorSuite) TestFilterTask_EnforceApply() {\n\tdomainID := uuid.New()\n\tok, err := s.taskHandler.filterTask(domainID, true)\n\ts.NoError(err)\n\ts.True(ok)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_SyncActivityReplicationTask_SameShardID() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_SyncActivityReplicationTask_SameShardID_RetryTaskV2Error_Success() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(&types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(nil)\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_SyncActivityReplicationTask_SameShardID_RetryTaskV2Error_SkipTask() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(&types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(ndc.ErrSkipTask)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_SyncActivityReplicationTask_SameShardID_RetryTaskV2Error_Error() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\tretryErr := &types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(retryErr).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(fmt.Errorf(\"some error\"))\n\t_, err := s.taskHandler.execute(task, true)\n\ts.Error(err)\n\ts.ErrorIs(err, retryErr)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_HistoryV2ReplicationTask_SameShardID() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_HistoryV2ReplicationTask_SameShardID_RetryTaskErr_Success() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\tretryErr := &types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(retryErr).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(nil)\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_HistoryV2ReplicationTask_SameShardID_RetryTaskErr_SkipTask() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\tretryErr := &types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(retryErr).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(ndc.ErrSkipTask)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcessTask_HistoryV2ReplicationTask_SameShardID_RetryTaskErr_Error() {\n\tdomainID := uuid.New()\n\tworkflowID := \"6d89f939-e6a4-4c26-a0ed-626ce27bcc9c\" // belong to shard 0\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\tretryErr := &types.RetryTaskV2Error{\n\t\tDomainID:          \"test-domain-id\",\n\t\tWorkflowID:        \"test-wf-id\",\n\t\tRunID:             \"test-run-id\",\n\t\tStartEventID:      common.Ptr(int64(11)),\n\t\tStartEventVersion: common.Ptr(int64(100)),\n\t\tEndEventID:        common.Ptr(int64(19)),\n\t\tEndEventVersion:   common.Ptr(int64(102)),\n\t}\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(retryErr).Times(1)\n\ts.nDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.sourceCluster,\n\t\t\"test-domain-id\",\n\t\t\"test-wf-id\",\n\t\t\"test-run-id\",\n\t\tcommon.Ptr(int64(11)),\n\t\tcommon.Ptr(int64(100)),\n\t\tcommon.Ptr(int64(19)),\n\t\tcommon.Ptr(int64(102))).\n\t\tReturn(fmt.Errorf(\"some error\"))\n\t_, err := s.taskHandler.execute(task, true)\n\ts.Error(err)\n\ts.ErrorIs(err, retryErr)\n}\n\nfunc (s *taskExecutorSuite) TestProcess_HistoryV2ReplicationTask_DifferentShardID() {\n\tdomainID := uuid.New()\n\tworkflowID := \"abc\" // belong to shard 1\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.ReplicateEventsV2Request{\n\t\tDomainUUID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\ts.mockEngine.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil).Times(0)\n\ts.historyClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcess_SyncActivityReplicationTask_DifferentShardID() {\n\tdomainID := uuid.New()\n\tworkflowID := \"abc\" // belong to shard 1\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\trequest := &types.SyncActivityRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t}\n\n\ts.mockEngine.EXPECT().SyncActivity(gomock.Any(), request).Return(nil).Times(0)\n\ts.historyClient.EXPECT().SyncActivity(gomock.Any(), request).Return(nil).Times(1)\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcess_FailoverReplicationTask() {\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeFailoverMarker.Ptr(),\n\t\tFailoverMarkerAttributes: &types.FailoverMarkerAttributes{\n\t\t\tDomainID:        \"test-domain-id\",\n\t\t\tFailoverVersion: 101,\n\t\t\tCreationTime:    common.Ptr(int64(111)),\n\t\t},\n\t\tCreationTime: common.Ptr(int64(222)),\n\t}\n\ts.mockShard.MockAddingPendingFailoverMarker = func(marker *types.FailoverMarkerAttributes) error {\n\t\ts.Equal(&types.FailoverMarkerAttributes{\n\t\t\tDomainID:        \"test-domain-id\",\n\t\t\tFailoverVersion: 101,\n\t\t\tCreationTime:    common.Ptr(int64(222)),\n\t\t}, marker)\n\t\treturn nil\n\t}\n\n\t_, err := s.taskHandler.execute(task, true)\n\ts.NoError(err)\n}\n\nfunc (s *taskExecutorSuite) TestProcess_UnknownTask() {\n\ttask := &types.ReplicationTask{\n\t\tTaskType: common.Ptr(types.ReplicationTaskType(-100)),\n\t}\n\t_, err := s.taskHandler.execute(task, true)\n\ts.Error(err)\n\ts.ErrorIs(err, ErrUnknownReplicationTask)\n}\n"
  },
  {
    "path": "service/history/replication/task_fetcher.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination task_fetcher_mock.go -self_package github.com/uber/cadence/service/history/replication\n\n// TODO: reuse the interface and implementation defined in history/task package\nconst (\n\tfetchTaskRequestTimeout = 60 * time.Second\n\trequestChanBufferSize   = 1000\n)\n\ntype (\n\t// TaskFetcher is responsible for fetching replication messages from remote DC.\n\tTaskFetcher interface {\n\t\tcommon.Daemon\n\n\t\tGetSourceCluster() string\n\t\tGetRequestChan(shardID int) chan<- *request\n\t\tGetRateLimiter() quotas.Limiter\n\t}\n\n\t// TaskFetchers is a group of fetchers, one per source DC.\n\tTaskFetchers interface {\n\t\tcommon.Daemon\n\n\t\tGetFetchers() []TaskFetcher\n\t}\n\n\t// taskFetcherImpl is the implementation of fetching replication messages.\n\ttaskFetcherImpl struct {\n\t\tstatus         int32\n\t\tcurrentCluster string\n\t\tsourceCluster  string\n\t\tconfig         *config.Config\n\t\tlogger         log.Logger\n\t\tmetricsScope   metrics.Scope\n\t\tremotePeer     admin.Client\n\t\trateLimiter    quotas.Limiter\n\t\ttimeSource     clock.TimeSource\n\t\trequestChan    []chan *request\n\t\tctx            context.Context\n\t\tcancelCtx      context.CancelFunc\n\t\twg             sync.WaitGroup\n\n\t\tfetchAndDistributeTasksFn func(map[int32]*request) error\n\t}\n\n\t// taskFetchersImpl is a group of fetchers, one per source DC.\n\ttaskFetchersImpl struct {\n\t\tstatus   int32\n\t\tlogger   log.Logger\n\t\tfetchers []TaskFetcher\n\t}\n)\n\nvar _ TaskFetcher = (*taskFetcherImpl)(nil)\nvar _ TaskFetchers = (*taskFetchersImpl)(nil)\n\n// NewTaskFetchers creates an instance of ReplicationTaskFetchers with given configs.\nfunc NewTaskFetchers(\n\tlogger log.Logger,\n\tconfig *config.Config,\n\tclusterMetadata cluster.Metadata,\n\tclientBean client.Bean,\n\tmetricsClient metrics.Client,\n) (TaskFetchers, error) {\n\tcurrentCluster := clusterMetadata.GetCurrentClusterName()\n\tvar fetchers []TaskFetcher\n\tfor clusterName := range clusterMetadata.GetRemoteClusterInfo() {\n\t\tremoteFrontendClient, err := clientBean.GetRemoteAdminClient(clusterName)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tfetcher := newReplicationTaskFetcher(\n\t\t\tlogger,\n\t\t\tclusterName,\n\t\t\tcurrentCluster,\n\t\t\tconfig,\n\t\t\tremoteFrontendClient,\n\t\t\tmetricsClient,\n\t\t)\n\t\tfetchers = append(fetchers, fetcher)\n\t}\n\n\treturn &taskFetchersImpl{\n\t\tfetchers: fetchers,\n\t\tstatus:   common.DaemonStatusInitialized,\n\t\tlogger:   logger,\n\t}, nil\n}\n\n// Start starts the fetchers\nfunc (f *taskFetchersImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tfor _, fetcher := range f.fetchers {\n\t\tfetcher.Start()\n\t}\n\tf.logger.Info(\"Replication task fetchers started.\", tag.Counter(len(f.fetchers)))\n}\n\n// Stop stops the fetchers\nfunc (f *taskFetchersImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tfor _, fetcher := range f.fetchers {\n\t\tfetcher.Stop()\n\t}\n\tf.logger.Info(\"Replication task fetchers stopped.\", tag.Counter(len(f.fetchers)))\n}\n\n// GetFetchers returns all the fetchers\nfunc (f *taskFetchersImpl) GetFetchers() []TaskFetcher {\n\treturn f.fetchers\n}\n\n// newReplicationTaskFetcher creates a new fetcher.\nfunc newReplicationTaskFetcher(\n\tlogger log.Logger,\n\tsourceCluster string,\n\tcurrentCluster string,\n\tconfig *config.Config,\n\tsourceFrontend admin.Client,\n\tmetricsClient metrics.Client,\n) TaskFetcher {\n\tctx, cancel := context.WithCancel(context.Background())\n\n\trequestChan := make([]chan *request, config.ReplicationTaskFetcherParallelism())\n\tfor i := 0; i < config.ReplicationTaskFetcherParallelism(); i++ {\n\t\trequestChan[i] = make(chan *request, requestChanBufferSize)\n\t}\n\n\tfetcher := &taskFetcherImpl{\n\t\tstatus:         common.DaemonStatusInitialized,\n\t\tconfig:         config,\n\t\tlogger:         logger.WithTags(tag.ClusterName(sourceCluster)),\n\t\tmetricsScope:   metricsClient.Scope(metrics.ReplicationTaskFetcherScope, metrics.TargetClusterTag(sourceCluster)),\n\t\tremotePeer:     sourceFrontend,\n\t\tcurrentCluster: currentCluster,\n\t\tsourceCluster:  sourceCluster,\n\t\trateLimiter:    quotas.NewDynamicRateLimiter(config.ReplicationTaskProcessorHostQPS.AsFloat64()),\n\t\ttimeSource:     clock.NewRealTimeSource(),\n\t\trequestChan:    requestChan,\n\t\tctx:            ctx,\n\t\tcancelCtx:      cancel,\n\t}\n\tfetcher.fetchAndDistributeTasksFn = fetcher.fetchAndDistributeTasks\n\treturn fetcher\n}\n\n// Start starts the fetcher\nfunc (f *taskFetcherImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\t// NOTE: ReplicationTaskFetcherParallelism > 1 is now supported. Each fetcher goroutine handles a subset of shards\n\t// (distributed via shardID % parallelism) and runs its own fetch cycle independently.\n\tfor i := 0; i < f.config.ReplicationTaskFetcherParallelism(); i++ {\n\t\ti := i\n\t\tf.wg.Add(1)\n\t\tgo f.fetchTasks(i)\n\t}\n\tf.logger.Info(\"Replication task fetcher started.\", tag.Counter(f.config.ReplicationTaskFetcherParallelism()))\n}\n\n// Stop stops the fetcher\nfunc (f *taskFetcherImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&f.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tf.cancelCtx()\n\tif !common.AwaitWaitGroup(&f.wg, 10*time.Second) {\n\t\tf.logger.Warn(\"Replication task fetcher timed out on shutdown.\")\n\t} else {\n\t\tf.logger.Info(\"Replication task fetcher graceful shutdown completed.\")\n\t}\n}\n\n// fetchTasks collects getReplicationTasks request from shards and send out aggregated request to source frontend.\nfunc (f *taskFetcherImpl) fetchTasks(chanIdx int) {\n\tdefer f.wg.Done()\n\n\ttimer := f.timeSource.NewTimer(backoff.JitDuration(\n\t\tf.config.ReplicationTaskFetcherAggregationInterval(),\n\t\tf.config.ReplicationTaskFetcherTimerJitterCoefficient(),\n\t))\n\tdefer timer.Stop()\n\n\trequestByShard := make(map[int32]*request)\n\tfor {\n\t\tselect {\n\t\tcase request := <-f.requestChan[chanIdx]:\n\t\t\t// Here we only add the request to map. We will wait until timer fires to send the request to remote.\n\t\t\tif req, ok := requestByShard[request.token.GetShardID()]; ok && req != request {\n\t\t\t\t// since this replication task fetcher is per host and replication task processor is per shard\n\t\t\t\t// during shard movement, duplicated requests can appear, if shard moved from this host to this host.\n\t\t\t\tf.logger.Info(\"Get replication task request already exist for shard.\", tag.ShardID(int(request.token.GetShardID())))\n\t\t\t\tclose(req.respChan)\n\t\t\t}\n\t\t\trequestByShard[request.token.GetShardID()] = request\n\n\t\tcase <-timer.Chan():\n\t\t\t// When timer fires, we collect all the requests we have so far and attempt to send them to remote.\n\t\t\terr := f.fetchAndDistributeTasksFn(requestByShard)\n\t\t\tif err != nil {\n\t\t\t\tif _, ok := err.(*types.ServiceBusyError); ok {\n\t\t\t\t\t// slow down replication when source cluster is busy\n\t\t\t\t\ttimer.Reset(f.config.ReplicationTaskFetcherServiceBusyWait())\n\t\t\t\t} else {\n\t\t\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\t\t\tf.config.ReplicationTaskFetcherErrorRetryWait(),\n\t\t\t\t\t\tf.config.ReplicationTaskFetcherTimerJitterCoefficient(),\n\t\t\t\t\t))\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\t\tf.config.ReplicationTaskFetcherAggregationInterval(),\n\t\t\t\t\tf.config.ReplicationTaskFetcherTimerJitterCoefficient(),\n\t\t\t\t))\n\t\t\t}\n\t\tcase <-f.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (f *taskFetcherImpl) fetchAndDistributeTasks(requestByShard map[int32]*request) error {\n\tstartTime := f.timeSource.Now()\n\tdefer func() {\n\t\tfetchLatency := f.timeSource.Now().Sub(startTime)\n\t\tf.metricsScope.ExponentialHistogram(metrics.ExponentialReplicationTaskFetchLatency, fetchLatency)\n\t}()\n\n\tif len(requestByShard) == 0 {\n\t\t// We don't receive tasks from previous fetch so processors are all sleeping.\n\t\tf.logger.Debug(\"Skip fetching as no processor is asking for tasks.\")\n\t\treturn nil\n\t}\n\n\tmessagesByShard, err := f.getMessages(requestByShard)\n\tif err != nil {\n\t\tif _, ok := err.(*types.ServiceBusyError); !ok {\n\t\t\tf.logger.Error(\"Failed to get replication tasks\", tag.Error(err))\n\t\t} else {\n\t\t\tf.logger.Debug(\"Failed to get replication tasks because service busy\")\n\t\t}\n\n\t\treturn err\n\t}\n\n\ttotalTasks := 0\n\tfor _, messages := range messagesByShard {\n\t\ttotalTasks += len(messages.ReplicationTasks)\n\t}\n\tf.metricsScope.RecordHistogramValue(metrics.ReplicationTasksFetchedSize, float64(totalTasks))\n\n\tf.logger.Debug(\"Successfully fetched replication tasks.\", tag.Counter(len(messagesByShard)))\n\tfor shardID, tasks := range messagesByShard {\n\t\trequest := requestByShard[shardID]\n\t\trequest.respChan <- tasks\n\t\tclose(request.respChan)\n\t\tdelete(requestByShard, shardID)\n\t}\n\n\treturn nil\n}\n\nfunc (f *taskFetcherImpl) getMessages(requestByShard map[int32]*request) (map[int32]*types.ReplicationMessages, error) {\n\tvar tokens []*types.ReplicationToken\n\tfor _, request := range requestByShard {\n\t\ttokens = append(tokens, request.token)\n\t}\n\n\tctx, cancel := context.WithTimeout(f.ctx, fetchTaskRequestTimeout)\n\tdefer cancel()\n\n\trequest := &types.GetReplicationMessagesRequest{\n\t\tTokens:      tokens,\n\t\tClusterName: f.currentCluster,\n\t}\n\tresponse, err := f.remotePeer.GetReplicationMessages(ctx, request)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn response.GetMessagesByShard(), nil\n}\n\n// GetSourceCluster returns the source cluster for the fetcher\nfunc (f *taskFetcherImpl) GetSourceCluster() string {\n\treturn f.sourceCluster\n}\n\n// GetRequestChan returns the request chan for the fetcher\nfunc (f *taskFetcherImpl) GetRequestChan(shardID int) chan<- *request {\n\tchanIdx := shardID % len(f.requestChan)\n\n\treturn f.requestChan[chanIdx]\n}\n\n// GetRateLimiter returns the host level rate limiter for the fetcher\nfunc (f *taskFetcherImpl) GetRateLimiter() quotas.Limiter {\n\treturn f.rateLimiter\n}\n"
  },
  {
    "path": "service/history/replication/task_fetcher_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: task_fetcher.go\n//\n// Generated by this command:\n//\n//\tmockgen -package replication -source task_fetcher.go -destination task_fetcher_mock.go -self_package github.com/uber/cadence/service/history/replication\n//\n\n// Package replication is a generated GoMock package.\npackage replication\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tquotas \"github.com/uber/cadence/common/quotas\"\n)\n\n// MockTaskFetcher is a mock of TaskFetcher interface.\ntype MockTaskFetcher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskFetcherMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskFetcherMockRecorder is the mock recorder for MockTaskFetcher.\ntype MockTaskFetcherMockRecorder struct {\n\tmock *MockTaskFetcher\n}\n\n// NewMockTaskFetcher creates a new mock instance.\nfunc NewMockTaskFetcher(ctrl *gomock.Controller) *MockTaskFetcher {\n\tmock := &MockTaskFetcher{ctrl: ctrl}\n\tmock.recorder = &MockTaskFetcherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskFetcher) EXPECT() *MockTaskFetcherMockRecorder {\n\treturn m.recorder\n}\n\n// GetRateLimiter mocks base method.\nfunc (m *MockTaskFetcher) GetRateLimiter() quotas.Limiter {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRateLimiter\")\n\tret0, _ := ret[0].(quotas.Limiter)\n\treturn ret0\n}\n\n// GetRateLimiter indicates an expected call of GetRateLimiter.\nfunc (mr *MockTaskFetcherMockRecorder) GetRateLimiter() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRateLimiter\", reflect.TypeOf((*MockTaskFetcher)(nil).GetRateLimiter))\n}\n\n// GetRequestChan mocks base method.\nfunc (m *MockTaskFetcher) GetRequestChan(shardID int) chan<- *request {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRequestChan\", shardID)\n\tret0, _ := ret[0].(chan<- *request)\n\treturn ret0\n}\n\n// GetRequestChan indicates an expected call of GetRequestChan.\nfunc (mr *MockTaskFetcherMockRecorder) GetRequestChan(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRequestChan\", reflect.TypeOf((*MockTaskFetcher)(nil).GetRequestChan), shardID)\n}\n\n// GetSourceCluster mocks base method.\nfunc (m *MockTaskFetcher) GetSourceCluster() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSourceCluster\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetSourceCluster indicates an expected call of GetSourceCluster.\nfunc (mr *MockTaskFetcherMockRecorder) GetSourceCluster() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSourceCluster\", reflect.TypeOf((*MockTaskFetcher)(nil).GetSourceCluster))\n}\n\n// Start mocks base method.\nfunc (m *MockTaskFetcher) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockTaskFetcherMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockTaskFetcher)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockTaskFetcher) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockTaskFetcherMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockTaskFetcher)(nil).Stop))\n}\n\n// MockTaskFetchers is a mock of TaskFetchers interface.\ntype MockTaskFetchers struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskFetchersMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskFetchersMockRecorder is the mock recorder for MockTaskFetchers.\ntype MockTaskFetchersMockRecorder struct {\n\tmock *MockTaskFetchers\n}\n\n// NewMockTaskFetchers creates a new mock instance.\nfunc NewMockTaskFetchers(ctrl *gomock.Controller) *MockTaskFetchers {\n\tmock := &MockTaskFetchers{ctrl: ctrl}\n\tmock.recorder = &MockTaskFetchersMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskFetchers) EXPECT() *MockTaskFetchersMockRecorder {\n\treturn m.recorder\n}\n\n// GetFetchers mocks base method.\nfunc (m *MockTaskFetchers) GetFetchers() []TaskFetcher {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFetchers\")\n\tret0, _ := ret[0].([]TaskFetcher)\n\treturn ret0\n}\n\n// GetFetchers indicates an expected call of GetFetchers.\nfunc (mr *MockTaskFetchersMockRecorder) GetFetchers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFetchers\", reflect.TypeOf((*MockTaskFetchers)(nil).GetFetchers))\n}\n\n// Start mocks base method.\nfunc (m *MockTaskFetchers) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockTaskFetchersMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockTaskFetchers)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockTaskFetchers) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockTaskFetchersMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockTaskFetchers)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/replication/task_fetcher_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype (\n\ttaskFetcherSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t\tcontroller *gomock.Controller\n\n\t\tmockResource   *resource.Test\n\t\tconfig         *config.Config\n\t\tfrontendClient *admin.MockClient\n\t\ttaskFetcher    *taskFetcherImpl\n\t}\n)\n\nfunc TestReplicationTaskFetcherSuite(t *testing.T) {\n\ts := new(taskFetcherSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *taskFetcherSuite) SetupSuite() {\n\n}\n\nfunc (s *taskFetcherSuite) TearDownSuite() {\n\n}\n\nfunc (s *taskFetcherSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.History)\n\ts.frontendClient = s.mockResource.RemoteAdminClient\n\tlogger := testlogger.New(s.T())\n\ts.config = config.NewForTest()\n\ts.config.ReplicationTaskFetcherTimerJitterCoefficient = dynamicproperties.GetFloatPropertyFn(0.0) // set jitter to 0 for test\n\n\ts.taskFetcher = newReplicationTaskFetcher(\n\t\tlogger,\n\t\t\"standby\",\n\t\t\"active\",\n\t\ts.config,\n\t\ts.frontendClient,\n\t\tmetrics.NewNoopMetricsClient(),\n\t).(*taskFetcherImpl)\n}\n\nfunc (s *taskFetcherSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n}\n\nfunc (s *taskFetcherSuite) TestGetMessages() {\n\trequestByShard := make(map[int32]*request)\n\ttoken := &types.ReplicationToken{\n\t\tShardID:                0,\n\t\tLastProcessedMessageID: 1,\n\t\tLastRetrievedMessageID: 2,\n\t}\n\trequestByShard[0] = &request{\n\t\ttoken: token,\n\t}\n\treplicationMessageRequest := &types.GetReplicationMessagesRequest{\n\t\tTokens: []*types.ReplicationToken{\n\t\t\ttoken,\n\t\t},\n\t\tClusterName: \"active\",\n\t}\n\tmessageByShared := make(map[int32]*types.ReplicationMessages)\n\tmessageByShared[0] = &types.ReplicationMessages{}\n\texpectedResponse := &types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: messageByShared,\n\t}\n\ts.frontendClient.EXPECT().GetReplicationMessages(gomock.Any(), replicationMessageRequest).Return(expectedResponse, nil)\n\tresponse, err := s.taskFetcher.getMessages(requestByShard)\n\ts.NoError(err)\n\ts.Equal(messageByShared, response)\n}\n\nfunc (s *taskFetcherSuite) TestFetchAndDistributeTasks() {\n\trequestByShard := make(map[int32]*request)\n\ttoken := &types.ReplicationToken{\n\t\tShardID:                0,\n\t\tLastProcessedMessageID: 1,\n\t\tLastRetrievedMessageID: 2,\n\t}\n\trespChan := make(chan *types.ReplicationMessages, 1)\n\trequestByShard[0] = &request{\n\t\ttoken:    token,\n\t\trespChan: respChan,\n\t}\n\treplicationMessageRequest := &types.GetReplicationMessagesRequest{\n\t\tTokens: []*types.ReplicationToken{\n\t\t\ttoken,\n\t\t},\n\t\tClusterName: \"active\",\n\t}\n\tmessageByShared := make(map[int32]*types.ReplicationMessages)\n\tmessageByShared[0] = &types.ReplicationMessages{}\n\texpectedResponse := &types.GetReplicationMessagesResponse{\n\t\tMessagesByShard: messageByShared,\n\t}\n\ts.frontendClient.EXPECT().GetReplicationMessages(gomock.Any(), replicationMessageRequest).Return(expectedResponse, nil)\n\terr := s.taskFetcher.fetchAndDistributeTasks(requestByShard)\n\ts.NoError(err)\n\trespToken := <-respChan\n\ts.Equal(messageByShared[0], respToken)\n}\n\nfunc (s *taskFetcherSuite) TestLifecycle() {\n\tdefer goleak.VerifyNone(s.T())\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\ts.taskFetcher.timeSource = mockTimeSource\n\trespChan0 := make(chan *types.ReplicationMessages, 1)\n\trespChan1 := make(chan *types.ReplicationMessages, 1)\n\trespChan2 := make(chan *types.ReplicationMessages, 1)\n\trespChan3 := make(chan *types.ReplicationMessages, 1)\n\treq0 := &request{\n\t\ttoken: &types.ReplicationToken{\n\t\t\tShardID:                0,\n\t\t\tLastRetrievedMessageID: 100,\n\t\t\tLastProcessedMessageID: 10,\n\t\t},\n\t\trespChan: respChan0,\n\t}\n\treq1 := &request{\n\t\ttoken: &types.ReplicationToken{\n\t\t\tShardID:                1,\n\t\t\tLastRetrievedMessageID: 10,\n\t\t\tLastProcessedMessageID: 1,\n\t\t},\n\t\trespChan: respChan1,\n\t}\n\treq2 := &request{\n\t\ttoken: &types.ReplicationToken{\n\t\t\tShardID:                0,\n\t\t\tLastRetrievedMessageID: 10,\n\t\t\tLastProcessedMessageID: 1,\n\t\t},\n\t\trespChan: respChan2,\n\t}\n\treq3 := &request{\n\t\ttoken: &types.ReplicationToken{\n\t\t\tShardID:                1,\n\t\t\tLastRetrievedMessageID: 11,\n\t\t\tLastProcessedMessageID: 2,\n\t\t},\n\t\trespChan: respChan3,\n\t}\n\tfetchAndDistributeTasksFnCall := 0\n\tfetchAndDistributeTasksSyncChan := []chan struct{}{make(chan struct{}), make(chan struct{}), make(chan struct{}), make(chan struct{})}\n\ts.taskFetcher.fetchAndDistributeTasksFn = func(requestByShard map[int32]*request) error {\n\t\tdefer func() {\n\t\t\tfetchAndDistributeTasksFnCall++\n\t\t\tclose(fetchAndDistributeTasksSyncChan[fetchAndDistributeTasksFnCall-1])\n\t\t}()\n\t\tif fetchAndDistributeTasksFnCall == 0 {\n\t\t\ts.Equal(map[int32]*request{1: req1, 0: req2}, requestByShard)\n\t\t\treturn &types.ServiceBusyError{}\n\t\t} else if fetchAndDistributeTasksFnCall == 1 {\n\t\t\ts.Equal(map[int32]*request{1: req3, 0: req2}, requestByShard)\n\t\t\treturn &types.InternalServiceError{}\n\t\t} else if fetchAndDistributeTasksFnCall == 2 {\n\t\t\ts.Equal(map[int32]*request{1: req3, 0: req2}, requestByShard)\n\t\t\tfor shard := range requestByShard {\n\t\t\t\tdelete(requestByShard, shard)\n\t\t\t}\n\t\t\treturn nil\n\t\t} else if fetchAndDistributeTasksFnCall == 3 {\n\t\t\ts.Equal(map[int32]*request{}, requestByShard)\n\t\t\treturn nil\n\t\t}\n\t\treturn nil\n\t}\n\ts.taskFetcher.Start()\n\tdefer s.taskFetcher.Stop()\n\n\trequestChan := s.taskFetcher.GetRequestChan(0)\n\t// send 3 replication requests to the fetcher\n\trequestChan <- req0\n\trequestChan <- req1\n\trequestChan <- req2\n\t_, open := <-respChan0 // block until duplicate replication task is read from fetcher's request channel\n\ts.False(open)\n\n\t// process the existing replication requests and return service busy error\n\ts.Equal(0, fetchAndDistributeTasksFnCall)\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(s.config.ReplicationTaskFetcherAggregationInterval())\n\t_, open = <-fetchAndDistributeTasksSyncChan[0] // block until fetchAndDistributeTasksFn is called\n\ts.False(open)\n\ts.Equal(1, fetchAndDistributeTasksFnCall)\n\n\t// send a new duplicate replication request to fetcher\n\trequestChan <- req3\n\t_, open = <-respChan1 // block until duplicate replication task is read from fetcher's request channel\n\ts.False(open)\n\n\t// process the existing replication requests and return non-service busy error\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(s.config.ReplicationTaskFetcherServiceBusyWait())\n\t_, open = <-fetchAndDistributeTasksSyncChan[1] // block until fetchAndDistributeTasksFn is called\n\ts.False(open)\n\ts.Equal(2, fetchAndDistributeTasksFnCall)\n\n\t// process the existing replication requests and return success\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(s.config.ReplicationTaskFetcherErrorRetryWait())\n\t_, open = <-fetchAndDistributeTasksSyncChan[2] // block until fetchAndDistributeTasksFn is called\n\ts.False(open)\n\ts.Equal(3, fetchAndDistributeTasksFnCall)\n\n\t// process empty requests and return success\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(s.config.ReplicationTaskFetcherAggregationInterval())\n\t_, open = <-fetchAndDistributeTasksSyncChan[3] // block until fetchAndDistributeTasksFn is called\n\ts.False(open)\n\ts.Equal(4, fetchAndDistributeTasksFnCall)\n}\n\nfunc TestTaskFetchers(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tctrl := gomock.NewController(t)\n\tmockBean := client.NewMockBean(ctrl)\n\tmockAdminClient := admin.NewMockClient(ctrl)\n\tlogger := testlogger.New(t)\n\tcfg := config.NewForTest()\n\n\tmockBean.EXPECT().GetRemoteAdminClient(cluster.TestAlternativeClusterName).Return(mockAdminClient, nil)\n\tfetchers, err := NewTaskFetchers(logger, cfg, cluster.TestActiveClusterMetadata, mockBean, metrics.NewNoopMetricsClient())\n\tassert.NoError(t, err)\n\tassert.NotNil(t, fetchers)\n\tassert.Len(t, fetchers.GetFetchers(), len(cluster.TestActiveClusterMetadata.GetRemoteClusterInfo()))\n\n\tfetchers.Start()\n\tfetchers.Stop()\n}\n\nfunc TestTaskFetcherParallelism(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tlogger := testlogger.New(t)\n\tcfg := config.NewForTest()\n\tparallelism := 4\n\tcfg.ReplicationTaskFetcherParallelism = dynamicproperties.GetIntPropertyFn(parallelism)\n\n\tctrl := gomock.NewController(t)\n\tmockAdminClient := admin.NewMockClient(ctrl)\n\n\tfetcher := newReplicationTaskFetcher(\n\t\tlogger,\n\t\t\"standby\",\n\t\t\"active\",\n\t\tcfg,\n\t\tmockAdminClient,\n\t\tmetrics.NewNoopMetricsClient(),\n\t).(*taskFetcherImpl)\n\n\t// Test 1: Verify correct number of channels created\n\tassert.Equal(t, parallelism, len(fetcher.requestChan), \"Should create 4 request channels\")\n\n\t// Test 2: Verify shard-to-channel mapping\n\tchan0 := fetcher.GetRequestChan(0)\n\tchan1 := fetcher.GetRequestChan(1)\n\tchan4 := fetcher.GetRequestChan(4) // 4 % 4 = 0, should be same as chan0\n\tchan5 := fetcher.GetRequestChan(5) // 5 % 4 = 1, should be same as chan1\n\n\tassert.Equal(t, chan0, chan4, \"Shards 0 and 4 should map to same channel (0 % 4 == 4 % 4)\")\n\tassert.Equal(t, chan1, chan5, \"Shards 1 and 5 should map to same channel (1 % 4 == 5 % 4)\")\n\tassert.NotEqual(t, chan0, chan1, \"Different channels should be different\")\n\n\t// Test 3: Start fetcher and verify WaitGroup is properly incremented\n\tfetcher.Start()\n\n\t// The WaitGroup counter should now be 4 (one per goroutine)\n\t// We can verify this by calling Stop() which waits on the WaitGroup\n\t// If it hangs or times out, the goroutines weren't started correctly\n\tdone := make(chan bool)\n\tgo func() {\n\t\tfetcher.Stop()\n\t\tdone <- true\n\t}()\n\n\tselect {\n\tcase <-done:\n\t\t// Success - all goroutines exited cleanly\n\tcase <-time.After(11 * time.Second):\n\t\tt.Fatal(\"Stop() timed out - goroutines may not have been started correctly\")\n\t}\n}\n"
  },
  {
    "path": "service/history/replication/task_hydrator.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nvar errUnknownReplicationTask = errors.New(\"unknown replication task\")\n\n// TaskHydrator will enrich replication task with additional information from mutable state and history events.\n// Mutable state and history providers can be either in-memory or persistence based implementations;\n// depending whether we have available data already or need to load it.\ntype TaskHydrator struct {\n\tmsProvider mutableStateProvider\n\thistory    historyProvider\n}\n\ntype (\n\thistoryProvider interface {\n\t\tGetEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error)\n\t\tGetNextRunEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error)\n\t}\n\n\tmutableStateProvider interface {\n\t\tGetMutableState(ctx context.Context, domainID, workflowID, runID string) (mutableState, execution.ReleaseFunc, error)\n\t}\n\n\tmutableState interface {\n\t\tIsWorkflowExecutionRunning() bool\n\t\tGetActivityInfo(int64) (*persistence.ActivityInfo, bool)\n\t\tGetVersionHistories() *persistence.VersionHistories\n\t}\n)\n\n// NewImmediateTaskHydrator will enrich replication tasks with additional information that is immediately available.\nfunc NewImmediateTaskHydrator(isRunning bool, vh *persistence.VersionHistories, activities map[int64]*persistence.ActivityInfo, blob, nextBlob *persistence.DataBlob) TaskHydrator {\n\treturn TaskHydrator{\n\t\thistory:    immediateHistoryProvider{blob: blob, nextBlob: nextBlob},\n\t\tmsProvider: immediateMutableStateProvider{immediateMutableState{isRunning, activities, vh}},\n\t}\n}\n\n// NewDeferredTaskHydrator will enrich replication tasks with additional information that is not available on hand,\n// but is rather loaded in a deferred way later from a database and cache.\nfunc NewDeferredTaskHydrator(shardID int, historyManager persistence.HistoryManager, executionCache execution.Cache, domains domainCache) TaskHydrator {\n\treturn TaskHydrator{\n\t\thistory:    historyLoader{shardID, historyManager, domains},\n\t\tmsProvider: mutableStateLoader{executionCache},\n\t}\n}\n\n// Hydrate will enrich replication task with additional information from mutable state and history events.\nfunc (h TaskHydrator) Hydrate(ctx context.Context, task persistence.Task) (retTask *types.ReplicationTask, retErr error) {\n\tswitch t := task.(type) {\n\tcase *persistence.FailoverMarkerTask:\n\t\treturn hydrateFailoverMarkerTask(t), nil\n\t}\n\n\tms, release, err := h.msProvider.GetMutableState(ctx, task.GetDomainID(), task.GetWorkflowID(), task.GetRunID())\n\tdefer func() {\n\t\tif release != nil {\n\t\t\trelease(retErr)\n\t\t}\n\t}()\n\n\tif common.IsEntityNotExistsError(err) {\n\t\treturn nil, nil\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tswitch t := task.(type) {\n\tcase *persistence.SyncActivityTask:\n\t\treturn hydrateSyncActivityTask(t, ms)\n\tcase *persistence.HistoryReplicationTask:\n\t\tversionHistories := ms.GetVersionHistories()\n\t\tif versionHistories != nil {\n\t\t\t// Create a copy to release workflow lock early, as hydration will make a DB call, which may take a while\n\t\t\tversionHistories = versionHistories.Duplicate()\n\t\t}\n\t\trelease(nil)\n\t\treturn hydrateHistoryReplicationTask(ctx, t, versionHistories, h.history)\n\tdefault:\n\t\treturn nil, errUnknownReplicationTask\n\t}\n}\n\nfunc hydrateFailoverMarkerTask(t *persistence.FailoverMarkerTask) *types.ReplicationTask {\n\treturn &types.ReplicationTask{\n\t\tTaskType:     types.ReplicationTaskTypeFailoverMarker.Ptr(),\n\t\tSourceTaskID: t.TaskID,\n\t\tFailoverMarkerAttributes: &types.FailoverMarkerAttributes{\n\t\t\tDomainID:        t.DomainID,\n\t\t\tFailoverVersion: t.Version,\n\t\t},\n\t\tCreationTime: common.Ptr(t.VisibilityTimestamp.UnixNano()),\n\t}\n}\n\nfunc hydrateSyncActivityTask(task *persistence.SyncActivityTask, ms mutableState) (*types.ReplicationTask, error) {\n\tif !ms.IsWorkflowExecutionRunning() {\n\t\t// workflow already finished, no need to process the replication task\n\t\treturn nil, nil\n\t}\n\n\tactivityInfo, ok := ms.GetActivityInfo(task.ScheduledID)\n\tif !ok {\n\t\treturn nil, nil\n\t}\n\n\tvar startedTime *int64\n\tif activityInfo.StartedID != constants.EmptyEventID {\n\t\tstartedTime = timeToUnixNano(activityInfo.StartedTime)\n\t}\n\n\t// Version history uses when replicate the sync activity task\n\tvar versionHistory *types.VersionHistory\n\tif versionHistories := ms.GetVersionHistories(); versionHistories != nil {\n\t\tcurrentVersionHistory, err := versionHistories.GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tversionHistory = currentVersionHistory.ToInternalType()\n\t}\n\n\treturn &types.ReplicationTask{\n\t\tTaskType:     types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSourceTaskID: task.TaskID,\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:           task.DomainID,\n\t\t\tWorkflowID:         task.WorkflowID,\n\t\t\tRunID:              task.RunID,\n\t\t\tVersion:            activityInfo.Version,\n\t\t\tScheduledID:        activityInfo.ScheduleID,\n\t\t\tScheduledTime:      timeToUnixNano(activityInfo.ScheduledTime),\n\t\t\tStartedID:          activityInfo.StartedID,\n\t\t\tStartedTime:        startedTime,\n\t\t\tLastHeartbeatTime:  timeToUnixNano(activityInfo.LastHeartBeatUpdatedTime),\n\t\t\tDetails:            activityInfo.Details,\n\t\t\tAttempt:            activityInfo.Attempt,\n\t\t\tLastFailureReason:  common.Ptr(activityInfo.LastFailureReason),\n\t\t\tLastWorkerIdentity: activityInfo.LastWorkerIdentity,\n\t\t\tLastFailureDetails: activityInfo.LastFailureDetails,\n\t\t\tVersionHistory:     versionHistory,\n\t\t},\n\t\tCreationTime: common.Ptr(task.VisibilityTimestamp.UnixNano()),\n\t}, nil\n}\n\nfunc hydrateHistoryReplicationTask(ctx context.Context, task *persistence.HistoryReplicationTask, versionHistories *persistence.VersionHistories, history historyProvider) (*types.ReplicationTask, error) {\n\tif versionHistories == nil {\n\t\treturn nil, nil\n\t}\n\n\t_, versionHistory, err := versionHistories.FindFirstVersionHistoryByItem(persistence.NewVersionHistoryItem(task.FirstEventID, task.Version))\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// BranchToken will not set in get dlq replication message request\n\tif len(task.BranchToken) == 0 {\n\t\ttask.BranchToken = versionHistory.GetBranchToken()\n\t}\n\n\teventsBlob, err := history.GetEventBlob(ctx, task)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tnewRunEventsBlob, err := history.GetNextRunEventBlob(ctx, task)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &types.ReplicationTask{\n\t\tTaskType:     types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tSourceTaskID: task.TaskID,\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:            task.DomainID,\n\t\t\tWorkflowID:          task.WorkflowID,\n\t\t\tRunID:               task.RunID,\n\t\t\tVersionHistoryItems: versionHistory.ToInternalType().Items,\n\t\t\tEvents:              eventsBlob,\n\t\t\tNewRunEvents:        newRunEventsBlob,\n\t\t},\n\t\tCreationTime: common.Ptr(task.VisibilityTimestamp.UnixNano()),\n\t}, nil\n}\n\n// historyLoader loads history event blobs on demand from a database\ntype historyLoader struct {\n\tshardID int\n\thistory persistence.HistoryManager\n\tdomains domainCache\n}\n\nfunc (h historyLoader) GetEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\treturn h.getEventsBlob(ctx, task.DomainID, task.BranchToken, task.FirstEventID, task.NextEventID)\n}\n\nfunc (h historyLoader) GetNextRunEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\tif len(task.NewRunBranchToken) == 0 {\n\t\treturn nil, nil\n\t}\n\t// only get the first batch\n\treturn h.getEventsBlob(ctx, task.DomainID, task.NewRunBranchToken, constants.FirstEventID, constants.FirstEventID+1)\n}\n\nfunc (h historyLoader) getEventsBlob(ctx context.Context, domainID string, branchToken []byte, minEventID, maxEventID int64) (*types.DataBlob, error) {\n\tdomain, err := h.domains.GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresp, err := h.history.ReadRawHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken: branchToken,\n\t\tMinEventID:  minEventID,\n\t\tMaxEventID:  maxEventID,\n\t\tPageSize:    2, // Load more than one to check for data inconsistency errors\n\t\tShardID:     &h.shardID,\n\t\tDomainName:  domain.GetInfo().Name,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif len(resp.HistoryEventBlobs) != 1 {\n\t\treturn nil, &types.InternalDataInconsistencyError{Message: \"replication hydrator encountered more than 1 NDC raw event batch\"}\n\t}\n\n\treturn resp.HistoryEventBlobs[0].ToInternal(), nil\n}\n\n// mutableStateLoader uses workflow execution cache to load mutable state\ntype mutableStateLoader struct {\n\tcache execution.Cache\n}\n\nfunc (l mutableStateLoader) GetMutableState(ctx context.Context, domainID, workflowID, runID string) (mutableState, execution.ReleaseFunc, error) {\n\twfContext, release, err := l.cache.GetOrCreateWorkflowExecution(ctx, domainID, types.WorkflowExecution{WorkflowID: workflowID, RunID: runID})\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\trelease(err)\n\t\treturn nil, nil, err\n\t}\n\n\treturn mutableState, release, nil\n}\n\nfunc timeToUnixNano(t time.Time) *int64 {\n\treturn common.Int64Ptr(t.UnixNano())\n}\n\ntype immediateHistoryProvider struct {\n\tblob     *persistence.DataBlob\n\tnextBlob *persistence.DataBlob\n}\n\nfunc (h immediateHistoryProvider) GetEventBlob(_ context.Context, _ *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\tif h.blob == nil {\n\t\treturn nil, errors.New(\"history blob not set\")\n\t}\n\treturn h.blob.ToInternal(), nil\n}\n\nfunc (h immediateHistoryProvider) GetNextRunEventBlob(_ context.Context, _ *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\tif h.nextBlob == nil {\n\t\treturn nil, nil // Expected and common\n\t}\n\treturn h.nextBlob.ToInternal(), nil\n}\n\ntype immediateMutableStateProvider struct {\n\tms immediateMutableState\n}\n\nfunc (r immediateMutableStateProvider) GetMutableState(_ context.Context, _, _, _ string) (mutableState, execution.ReleaseFunc, error) {\n\treturn r.ms, execution.NoopReleaseFn, nil\n}\n\ntype immediateMutableState struct {\n\tisRunning        bool\n\tactivities       map[int64]*persistence.ActivityInfo\n\tversionHistories *persistence.VersionHistories\n}\n\nfunc (ms immediateMutableState) IsWorkflowExecutionRunning() bool {\n\treturn ms.isRunning\n}\nfunc (ms immediateMutableState) GetActivityInfo(id int64) (*persistence.ActivityInfo, bool) {\n\tinfo, ok := ms.activities[id]\n\treturn info, ok\n}\nfunc (ms immediateMutableState) GetVersionHistories() *persistence.VersionHistories {\n\treturn ms.versionHistories\n}\n"
  },
  {
    "path": "service/history/replication/task_hydrator_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\ttestShardID           = 0\n\ttestDomainID          = \"11111111-1111-1111-1111-111111111111\"\n\ttestWorkflowID        = \"workflow-id\"\n\ttestRunID             = \"22222222-2222-2222-2222-222222222222\"\n\ttestTaskID            = 111\n\ttestCreationTime      = int64(333)\n\ttestFirstEventID      = 6\n\ttestNextEventID       = 8\n\ttestVersion           = 456\n\ttestScheduleID        = int64(10)\n\ttestStartedID         = int64(11)\n\ttestLastFailureReason = \"failure-reason\"\n\ttestWorkerIdentity    = \"worker-identity\"\n\ttestAttempt           = 42\n)\n\nvar (\n\ttestBranchToken               = []byte{91, 92, 93}\n\ttestBranchTokenNewRun         = []byte{94, 95, 96}\n\ttestBranchTokenVersionHistory = []byte{97, 98, 99}\n\ttestDataBlob                  = &types.DataBlob{Data: []byte{1, 2, 3}, EncodingType: types.EncodingTypeJSON.Ptr()}\n\ttestDataBlobNewRun            = &types.DataBlob{Data: []byte{4, 5, 6}, EncodingType: types.EncodingTypeJSON.Ptr()}\n\ttestDataBlobVersionHistory    = &types.DataBlob{Data: []byte{7, 8, 9}, EncodingType: types.EncodingTypeJSON.Ptr()}\n\ttestDetails                   = []byte{100, 101, 102}\n\ttestLastFailureDetails        = []byte{103, 104, 105}\n\ttestScheduleTime              = time.Now()\n\ttestStartedTime               = time.Now()\n\ttestHeartbeatTime             = time.Now()\n\ttestWorkflowIdentifier        = definition.NewWorkflowIdentifier(testDomainID, testWorkflowID, testRunID)\n)\n\nfunc TestNewDeferredTaskHydrator(t *testing.T) {\n\th := NewDeferredTaskHydrator(0, nil, nil, nil)\n\trequire.NotNil(t, h)\n\tassert.IsType(t, historyLoader{}, h.history)\n\tassert.IsType(t, mutableStateLoader{}, h.msProvider)\n}\n\nfunc TestTaskHydrator_UnknownTask(t *testing.T) {\n\ttask := &persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   testWorkflowIdentifier.DomainID,\n\t\t\tWorkflowID: testWorkflowIdentifier.WorkflowID,\n\t\t\tRunID:      testWorkflowIdentifier.RunID,\n\t\t},\n\t}\n\tth := TaskHydrator{msProvider: &fakeMutableStateProvider{\n\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\ttestWorkflowIdentifier: &fakeMutableState{},\n\t\t},\n\t}}\n\tresult, err := th.Hydrate(context.Background(), task)\n\tassert.Equal(t, errUnknownReplicationTask, err)\n\tassert.Nil(t, result)\n\tassert.True(t, th.msProvider.(*fakeMutableStateProvider).released)\n}\n\nfunc TestTaskHydrator_HydrateFailoverMarkerTask(t *testing.T) {\n\ttask := &persistence.FailoverMarkerTask{\n\t\tDomainID: testDomainID,\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              testTaskID,\n\t\t\tVersion:             testVersion,\n\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t},\n\t}\n\n\texpected := types.ReplicationTask{\n\t\tTaskType:     types.ReplicationTaskTypeFailoverMarker.Ptr(),\n\t\tSourceTaskID: testTaskID,\n\t\tFailoverMarkerAttributes: &types.FailoverMarkerAttributes{\n\t\t\tDomainID:        testDomainID,\n\t\t\tFailoverVersion: testVersion,\n\t\t},\n\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t}\n\n\tth := TaskHydrator{}\n\tactual, err := th.Hydrate(context.Background(), task)\n\tassert.NoError(t, err)\n\tassert.Equal(t, &expected, actual)\n}\n\nfunc TestTaskHydrator_HydrateSyncActivityTask(t *testing.T) {\n\ttask := &persistence.SyncActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   testDomainID,\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              testTaskID,\n\t\t\tVersion:             testVersion,\n\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t},\n\t\tScheduledID: testScheduleID,\n\t}\n\n\tversionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: testBranchTokenVersionHistory,\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{EventID: testFirstEventID, Version: testVersion},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tactivityInfo := persistence.ActivityInfo{\n\t\tVersion:                  testVersion,\n\t\tScheduleID:               testScheduleID,\n\t\tScheduledTime:            testScheduleTime,\n\t\tStartedID:                testStartedID,\n\t\tStartedTime:              testStartedTime,\n\t\tDomainID:                 testDomainID,\n\t\tLastHeartBeatUpdatedTime: testHeartbeatTime,\n\t\tDetails:                  testDetails,\n\t\tAttempt:                  testAttempt,\n\t\tLastFailureReason:        testLastFailureReason,\n\t\tLastFailureDetails:       testLastFailureDetails,\n\t\tLastWorkerIdentity:       testWorkerIdentity,\n\t}\n\n\ttests := []struct {\n\t\tname       string\n\t\ttask       persistence.Task\n\t\tmsProvider mutableStateProvider\n\t\texpectTask *types.ReplicationTask\n\t\texpectErr  string\n\t}{\n\t\t{\n\t\t\tname: \"hydrates sync activity task\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{\n\t\t\t\t\t\tisWorkflowExecutionRunning: true,\n\t\t\t\t\t\tversionHistories:           versionHistories,\n\t\t\t\t\t\tactivityInfos:              map[int64]persistence.ActivityInfo{testScheduleID: activityInfo},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:           testDomainID,\n\t\t\t\t\tWorkflowID:         testWorkflowID,\n\t\t\t\t\tRunID:              testRunID,\n\t\t\t\t\tVersion:            testVersion,\n\t\t\t\t\tScheduledID:        testScheduleID,\n\t\t\t\t\tScheduledTime:      common.Int64Ptr(testScheduleTime.UnixNano()),\n\t\t\t\t\tStartedID:          testStartedID,\n\t\t\t\t\tStartedTime:        common.Int64Ptr(testStartedTime.UnixNano()),\n\t\t\t\t\tLastHeartbeatTime:  common.Int64Ptr(testHeartbeatTime.UnixNano()),\n\t\t\t\t\tDetails:            testDetails,\n\t\t\t\t\tAttempt:            testAttempt,\n\t\t\t\t\tLastFailureReason:  common.StringPtr(testLastFailureReason),\n\t\t\t\t\tLastWorkerIdentity: testWorkerIdentity,\n\t\t\t\t\tLastFailureDetails: testLastFailureDetails,\n\t\t\t\t\tVersionHistory: &types.VersionHistory{\n\t\t\t\t\t\tItems:       []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\t\tBranchToken: testBranchTokenVersionHistory,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"workflow is not running - return nil, no error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{\n\t\t\t\t\t\tisWorkflowExecutionRunning: false,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"no activity info - return nil, no error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{\n\t\t\t\t\t\tisWorkflowExecutionRunning: true,\n\t\t\t\t\t\tactivityInfos:              map[int64]persistence.ActivityInfo{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"bad version histories - return error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{\n\t\t\t\t\t\tisWorkflowExecutionRunning: true,\n\t\t\t\t\t\tversionHistories:           &persistence.VersionHistories{},\n\t\t\t\t\t\tactivityInfos:              map[int64]persistence.ActivityInfo{testScheduleID: activityInfo},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: \"getting branch index: 0, available branch count: 0\",\n\t\t},\n\t\t{\n\t\t\tname: \"workflow does not exist - return nil, no error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{},\n\t\t\t},\n\t\t\texpectTask: nil,\n\t\t},\n\t\t{\n\t\t\tname:       \"error loading mutable state\",\n\t\t\ttask:       task,\n\t\t\tmsProvider: &fakeMutableStateProvider{},\n\t\t\texpectErr:  \"error loading mutable state\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tth := TaskHydrator{msProvider: tt.msProvider}\n\t\t\tactualTask, err := th.Hydrate(context.Background(), tt.task)\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectTask, actualTask)\n\t\t\t}\n\t\t\tassert.True(t, th.msProvider.(*fakeMutableStateProvider).released)\n\t\t})\n\t}\n}\n\nfunc TestTaskHydrator_HydrateHistoryReplicationTask(t *testing.T) {\n\ttask := &persistence.HistoryReplicationTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              testTaskID,\n\t\t\tVersion:             testVersion,\n\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t},\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   testDomainID,\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tFirstEventID:      testFirstEventID,\n\t\tNextEventID:       testNextEventID,\n\t\tBranchToken:       testBranchToken,\n\t\tNewRunBranchToken: testBranchTokenNewRun,\n\t}\n\ttaskWithoutBranchToken := &persistence.HistoryReplicationTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              testTaskID,\n\t\t\tVersion:             testVersion,\n\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t},\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   testDomainID,\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tFirstEventID:      testFirstEventID,\n\t\tNextEventID:       testNextEventID,\n\t\tNewRunBranchToken: testBranchTokenNewRun,\n\t}\n\n\tversionHistories := persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: testBranchTokenVersionHistory,\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{EventID: testFirstEventID, Version: testVersion},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname       string\n\t\ttask       *persistence.HistoryReplicationTask\n\t\tmsProvider mutableStateProvider\n\t\thistory    historyProvider\n\t\texpectTask *types.ReplicationTask\n\t\texpectErr  string\n\t}{\n\t\t{\n\t\t\tname: \"hydrates history with given branch token\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: &versionHistories},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: &fakeHistoryProvider{\n\t\t\t\tblobs: []historyBlob{\n\t\t\t\t\t{branch: testBranchToken, blob: testDataBlob},\n\t\t\t\t\t{branch: testBranchTokenNewRun, blob: testDataBlobNewRun},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\t\tDomainID:            testDomainID,\n\t\t\t\t\tWorkflowID:          testWorkflowID,\n\t\t\t\t\tRunID:               testRunID,\n\t\t\t\t\tVersionHistoryItems: []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\tEvents:              testDataBlob,\n\t\t\t\t\tNewRunEvents:        testDataBlobNewRun,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"hydrates history with branch token from version histories\",\n\t\t\ttask: taskWithoutBranchToken,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: &versionHistories},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: &fakeHistoryProvider{\n\t\t\t\tblobs: []historyBlob{\n\t\t\t\t\t{branch: testBranchTokenVersionHistory, blob: testDataBlobVersionHistory},\n\t\t\t\t\t{branch: testBranchTokenNewRun, blob: testDataBlobNewRun},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\t\tDomainID:            testDomainID,\n\t\t\t\t\tWorkflowID:          testWorkflowID,\n\t\t\t\t\tRunID:               testRunID,\n\t\t\t\t\tVersionHistoryItems: []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\tEvents:              testDataBlobVersionHistory,\n\t\t\t\t\tNewRunEvents:        testDataBlobNewRun,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"no version histories - return nil, no error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: nil},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectTask: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"bad version histories - return error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: &persistence.VersionHistories{}},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: \"version histories does not contains given item.\",\n\t\t},\n\t\t{\n\t\t\tname: \"workflow does not exist - return nil, no error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{},\n\t\t\t},\n\t\t\texpectTask: nil,\n\t\t},\n\t\t{\n\t\t\tname:       \"error loading mutable state\",\n\t\t\ttask:       task,\n\t\t\tmsProvider: &fakeMutableStateProvider{},\n\t\t\texpectErr:  \"error loading mutable state\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed reading event blob - return error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: &versionHistories},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: &fakeHistoryProvider{\n\t\t\t\tblobs: []historyBlob{\n\t\t\t\t\t{branch: testBranchTokenNewRun, blob: testDataBlobNewRun},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: \"failed reading history\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed reading event blob for new run - return error\",\n\t\t\ttask: task,\n\t\t\tmsProvider: &fakeMutableStateProvider{\n\t\t\t\tworkflows: map[definition.WorkflowIdentifier]mutableState{\n\t\t\t\t\ttestWorkflowIdentifier: &fakeMutableState{versionHistories: &versionHistories},\n\t\t\t\t},\n\t\t\t},\n\t\t\thistory: &fakeHistoryProvider{\n\t\t\t\tblobs: []historyBlob{\n\t\t\t\t\t{branch: testBranchToken, blob: testDataBlob},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: \"failed reading history\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tth := TaskHydrator{msProvider: tt.msProvider, history: tt.history}\n\t\t\tactualTask, err := th.Hydrate(context.Background(), tt.task)\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectTask, actualTask)\n\t\t\t}\n\t\t\tassert.True(t, th.msProvider.(*fakeMutableStateProvider).released)\n\t\t})\n\t}\n}\n\nfunc TestHistoryLoader_GetEventBlob(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttask           *persistence.HistoryReplicationTask\n\t\tdomains        fakeDomainCache\n\t\tmockHistory    func(hm *mocks.HistoryV2Manager)\n\t\texpectDataBlob *types.DataBlob\n\t\texpectErr      string\n\t}{\n\t\t{\n\t\t\tname: \"loads data blob\",\n\t\t\ttask: &persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID: testDomainID,\n\t\t\t\t},\n\t\t\t\tBranchToken:  testBranchToken,\n\t\t\t\tFirstEventID: 10,\n\t\t\t\tNextEventID:  11,\n\t\t\t},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\tmockHistory: func(hm *mocks.HistoryV2Manager) {\n\t\t\t\thm.On(\"ReadRawHistoryBranch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\t\t\t\tBranchToken: testBranchToken,\n\t\t\t\t\tMinEventID:  10,\n\t\t\t\t\tMaxEventID:  11,\n\t\t\t\t\tPageSize:    2,\n\t\t\t\t\tShardID:     common.IntPtr(testShardID),\n\t\t\t\t\tDomainName:  testDomainName,\n\t\t\t\t}).Return(&persistence.ReadRawHistoryBranchResponse{\n\t\t\t\t\tHistoryEventBlobs: []*persistence.DataBlob{{Encoding: constants.EncodingTypeJSON, Data: testDataBlob.Data}},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectDataBlob: testDataBlob,\n\t\t},\n\t\t{\n\t\t\tname:        \"failed to get domain name\",\n\t\t\ttask:        &persistence.HistoryReplicationTask{WorkflowIdentifier: persistence.WorkflowIdentifier{DomainID: testDomainID}},\n\t\t\tdomains:     fakeDomainCache{},\n\t\t\tmockHistory: func(hm *mocks.HistoryV2Manager) {},\n\t\t\texpectErr:   \"domain does not exist\",\n\t\t},\n\t\t{\n\t\t\tname:    \"load failure\",\n\t\t\ttask:    &persistence.HistoryReplicationTask{WorkflowIdentifier: persistence.WorkflowIdentifier{DomainID: testDomainID}},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\tmockHistory: func(hm *mocks.HistoryV2Manager) {\n\t\t\t\thm.On(\"ReadRawHistoryBranch\", mock.Anything, mock.Anything).Return(nil, errors.New(\"load failure\"))\n\t\t\t},\n\t\t\texpectErr: \"load failure\",\n\t\t},\n\t\t{\n\t\t\tname:    \"response must contain exactly one blob\",\n\t\t\ttask:    &persistence.HistoryReplicationTask{WorkflowIdentifier: persistence.WorkflowIdentifier{DomainID: testDomainID}},\n\t\t\tdomains: fakeDomainCache{testDomainID: testDomain},\n\t\t\tmockHistory: func(hm *mocks.HistoryV2Manager) {\n\t\t\t\thm.On(\"ReadRawHistoryBranch\", mock.Anything, mock.Anything).Return(&persistence.ReadRawHistoryBranchResponse{\n\t\t\t\t\tHistoryEventBlobs: []*persistence.DataBlob{{}, {}}, // two blobs\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectErr: \"replication hydrator encountered more than 1 NDC raw event batch\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\thm := &mocks.HistoryV2Manager{}\n\t\t\ttt.mockHistory(hm)\n\t\t\tloader := historyLoader{shardID: testShardID, history: hm, domains: tt.domains}\n\t\t\tdataBlob, err := loader.GetEventBlob(context.Background(), tt.task)\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectDataBlob, dataBlob)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestHistoryLoader_GetNextRunEventBlob(t *testing.T) {\n\thm := &mocks.HistoryV2Manager{}\n\tloader := historyLoader{shardID: testShardID, history: hm, domains: fakeDomainCache{testDomainID: testDomain}}\n\n\tdataBlob, err := loader.GetNextRunEventBlob(context.Background(), &persistence.HistoryReplicationTask{NewRunBranchToken: nil})\n\tassert.NoError(t, err)\n\tassert.Nil(t, dataBlob)\n\n\thm.On(\"ReadRawHistoryBranch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken: testBranchTokenNewRun,\n\t\tMinEventID:  1,\n\t\tMaxEventID:  2,\n\t\tPageSize:    2,\n\t\tShardID:     common.IntPtr(testShardID),\n\t\tDomainName:  testDomainName,\n\t}).Return(&persistence.ReadRawHistoryBranchResponse{\n\t\tHistoryEventBlobs: []*persistence.DataBlob{{Encoding: constants.EncodingTypeJSON, Data: testDataBlob.Data}},\n\t}, nil)\n\tdataBlob, err = loader.GetNextRunEventBlob(context.Background(), &persistence.HistoryReplicationTask{WorkflowIdentifier: persistence.WorkflowIdentifier{DomainID: testDomainID}, NewRunBranchToken: testBranchTokenNewRun})\n\tassert.NoError(t, err)\n\tassert.Equal(t, testDataBlob, dataBlob)\n}\n\nfunc TestMutableStateLoader_GetMutableState(t *testing.T) {\n\tctx := context.Background()\n\tcontroller := gomock.NewController(t)\n\ttestShardContext := shard.NewTestContext(\n\t\tt,\n\t\tcontroller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:                 testShardID,\n\t\t\tRangeID:                 1,\n\t\t\tTransferAckLevel:        0,\n\t\t\tClusterReplicationLevel: make(map[string]int64),\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\tdomainCache := testShardContext.Resource.DomainCache\n\texecutionCache := execution.NewCache(testShardContext)\n\texpectedMS := execution.NewMockMutableState(controller)\n\tmsLoader := mutableStateLoader{executionCache}\n\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(testDomainName, nil).AnyTimes()\n\texec, release, err := executionCache.GetOrCreateWorkflowExecution(ctx, testDomainID, types.WorkflowExecution{WorkflowID: testWorkflowID, RunID: testRunID})\n\trequire.NoError(t, err)\n\t// Try getting mutable state while it is still locked, will result in an error\n\tcontextWithTimeout, cancel := context.WithTimeout(ctx, time.Millisecond)\n\tdefer cancel()\n\t_, _, err = msLoader.GetMutableState(contextWithTimeout, testDomainID, testWorkflowID, testRunID)\n\tassert.EqualError(t, err, \"context deadline exceeded\")\n\trelease(nil)\n\n\t// Error while trying to load mutable state will be returned\n\tdomainCache.EXPECT().GetDomainByID(\"non-existing-domain\").Return(nil, errors.New(\"does not exist\"))\n\t_, _, err = msLoader.GetMutableState(ctx, \"non-existing-domain\", testWorkflowID, testRunID)\n\tassert.EqualError(t, err, \"does not exist\")\n\n\t// Happy path\n\tdomainCache.EXPECT().GetDomainByID(testDomainID).Return(&cache.DomainCacheEntry{}, nil)\n\texpectedMS.EXPECT().StartTransaction(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil)\n\texec.SetWorkflowExecution(expectedMS)\n\tms, release, err := msLoader.GetMutableState(ctx, testDomainID, testWorkflowID, testRunID)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expectedMS, ms)\n\tassert.NotNil(t, release)\n\trelease(nil)\n}\n\nfunc TestImmediateTaskHydrator(t *testing.T) {\n\tactivityInfo := persistence.ActivityInfo{\n\t\tVersion:                  testVersion,\n\t\tScheduleID:               testScheduleID,\n\t\tScheduledTime:            testScheduleTime,\n\t\tStartedID:                testStartedID,\n\t\tStartedTime:              testStartedTime,\n\t\tDomainID:                 testDomainID,\n\t\tLastHeartBeatUpdatedTime: testHeartbeatTime,\n\t\tDetails:                  testDetails,\n\t\tAttempt:                  testAttempt,\n\t\tLastFailureReason:        testLastFailureReason,\n\t\tLastFailureDetails:       testLastFailureDetails,\n\t\tLastWorkerIdentity:       testWorkerIdentity,\n\t}\n\tversionHistories := &persistence.VersionHistories{\n\t\tCurrentVersionHistoryIndex: 0,\n\t\tHistories: []*persistence.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: testBranchTokenVersionHistory,\n\t\t\t\tItems: []*persistence.VersionHistoryItem{\n\t\t\t\t\t{EventID: testFirstEventID, Version: testVersion},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname             string\n\t\tversionHistories *persistence.VersionHistories\n\t\tactivities       map[int64]*persistence.ActivityInfo\n\t\tblob             *persistence.DataBlob\n\t\tnextRunBlob      *persistence.DataBlob\n\t\ttask             persistence.Task\n\t\texpectResult     *types.ReplicationTask\n\t\texpectErr        string\n\t}{\n\t\t{\n\t\t\tname:             \"sync activity task - happy path\",\n\t\t\tversionHistories: versionHistories,\n\t\t\tactivities:       map[int64]*persistence.ActivityInfo{testScheduleID: &activityInfo},\n\t\t\ttask: &persistence.SyncActivityTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID:              testTaskID,\n\t\t\t\t\tVersion:             testVersion,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t\t\t},\n\t\t\t\tScheduledID: testScheduleID,\n\t\t\t},\n\t\t\texpectResult: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:           testDomainID,\n\t\t\t\t\tWorkflowID:         testWorkflowID,\n\t\t\t\t\tRunID:              testRunID,\n\t\t\t\t\tVersion:            testVersion,\n\t\t\t\t\tScheduledID:        testScheduleID,\n\t\t\t\t\tScheduledTime:      common.Int64Ptr(testScheduleTime.UnixNano()),\n\t\t\t\t\tStartedID:          testStartedID,\n\t\t\t\t\tStartedTime:        common.Int64Ptr(testStartedTime.UnixNano()),\n\t\t\t\t\tLastHeartbeatTime:  common.Int64Ptr(testHeartbeatTime.UnixNano()),\n\t\t\t\t\tDetails:            testDetails,\n\t\t\t\t\tAttempt:            testAttempt,\n\t\t\t\t\tLastFailureReason:  common.StringPtr(testLastFailureReason),\n\t\t\t\t\tLastWorkerIdentity: testWorkerIdentity,\n\t\t\t\t\tLastFailureDetails: testLastFailureDetails,\n\t\t\t\t\tVersionHistory: &types.VersionHistory{\n\t\t\t\t\t\tItems:       []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\t\tBranchToken: testBranchTokenVersionHistory,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:             \"sync activity task - missing activity info\",\n\t\t\tversionHistories: versionHistories,\n\t\t\tactivities:       map[int64]*persistence.ActivityInfo{},\n\t\t\ttask: &persistence.SyncActivityTask{\n\t\t\t\tScheduledID: testScheduleID,\n\t\t\t},\n\t\t\texpectResult: nil,\n\t\t},\n\t\t{\n\t\t\tname:             \"history task - happy path\",\n\t\t\tversionHistories: versionHistories,\n\t\t\tblob:             persistence.NewDataBlobFromInternal(testDataBlob),\n\t\t\tnextRunBlob:      persistence.NewDataBlobFromInternal(testDataBlobNewRun),\n\t\t\ttask: &persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID:              testTaskID,\n\t\t\t\t\tVersion:             testVersion,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t\t\t},\n\t\t\t\tFirstEventID:      testFirstEventID,\n\t\t\t\tNextEventID:       testNextEventID,\n\t\t\t\tBranchToken:       testBranchToken,\n\t\t\t\tNewRunBranchToken: testBranchTokenNewRun,\n\t\t\t},\n\t\t\texpectResult: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\t\tDomainID:            testDomainID,\n\t\t\t\t\tWorkflowID:          testWorkflowID,\n\t\t\t\t\tRunID:               testRunID,\n\t\t\t\t\tVersionHistoryItems: []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\tEvents:              testDataBlob,\n\t\t\t\t\tNewRunEvents:        testDataBlobNewRun,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:             \"history task - no next run\",\n\t\t\tversionHistories: versionHistories,\n\t\t\tblob:             persistence.NewDataBlobFromInternal(testDataBlob),\n\t\t\ttask: &persistence.HistoryReplicationTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID:              testTaskID,\n\t\t\t\t\tVersion:             testVersion,\n\t\t\t\t\tVisibilityTimestamp: time.Unix(0, testCreationTime),\n\t\t\t\t},\n\t\t\t\tFirstEventID: testFirstEventID,\n\t\t\t\tNextEventID:  testNextEventID,\n\t\t\t\tBranchToken:  testBranchToken,\n\t\t\t},\n\t\t\texpectResult: &types.ReplicationTask{\n\t\t\t\tTaskType:     types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t\tCreationTime: common.Int64Ptr(testCreationTime),\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\t\tDomainID:            testDomainID,\n\t\t\t\t\tWorkflowID:          testWorkflowID,\n\t\t\t\t\tRunID:               testRunID,\n\t\t\t\t\tVersionHistoryItems: []*types.VersionHistoryItem{{EventID: testFirstEventID, Version: testVersion}},\n\t\t\t\t\tEvents:              testDataBlob,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:             \"history task - missing data blob\",\n\t\t\tversionHistories: versionHistories,\n\t\t\ttask: &persistence.HistoryReplicationTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion: testVersion,\n\t\t\t\t},\n\t\t\t\tFirstEventID: testFirstEventID,\n\t\t\t\tBranchToken:  testBranchToken,\n\t\t\t},\n\t\t\texpectErr: \"history blob not set\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\th := NewImmediateTaskHydrator(true, tt.versionHistories, tt.activities, tt.blob, tt.nextRunBlob)\n\t\t\tresult, err := h.Hydrate(context.Background(), tt.task)\n\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectResult, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\ntype fakeMutableStateProvider struct {\n\tworkflows map[definition.WorkflowIdentifier]mutableState\n\treleased  bool\n}\n\nfunc (msp *fakeMutableStateProvider) GetMutableState(ctx context.Context, domainID, workflowID, runID string) (mutableState, execution.ReleaseFunc, error) {\n\treleaseFn := func(error) {\n\t\tmsp.released = true\n\t}\n\n\tif msp.workflows == nil {\n\t\treturn nil, releaseFn, errors.New(\"error loading mutable state\")\n\t}\n\n\tms, ok := msp.workflows[definition.NewWorkflowIdentifier(domainID, workflowID, runID)]\n\tif !ok {\n\t\treturn nil, releaseFn, &types.EntityNotExistsError{}\n\t}\n\treturn ms, releaseFn, nil\n}\n\ntype fakeMutableState struct {\n\tisWorkflowExecutionRunning bool\n\tversionHistories           *persistence.VersionHistories\n\tactivityInfos              map[int64]persistence.ActivityInfo\n}\n\nfunc (ms fakeMutableState) IsWorkflowExecutionRunning() bool {\n\treturn ms.isWorkflowExecutionRunning\n}\nfunc (ms fakeMutableState) GetActivityInfo(scheduleID int64) (*persistence.ActivityInfo, bool) {\n\tai, ok := ms.activityInfos[scheduleID]\n\treturn &ai, ok\n}\nfunc (ms fakeMutableState) GetVersionHistories() *persistence.VersionHistories {\n\treturn ms.versionHistories\n}\n\ntype historyBlob struct {\n\tbranch []byte\n\tblob   *types.DataBlob\n}\n\ntype fakeHistoryProvider struct {\n\tblobs []historyBlob\n}\n\nfunc (h fakeHistoryProvider) GetEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\treturn h.getBlob(task.BranchToken)\n}\nfunc (h fakeHistoryProvider) GetNextRunEventBlob(ctx context.Context, task *persistence.HistoryReplicationTask) (*types.DataBlob, error) {\n\treturn h.getBlob(task.NewRunBranchToken)\n}\nfunc (h fakeHistoryProvider) getBlob(branch []byte) (*types.DataBlob, error) {\n\tfor _, b := range h.blobs {\n\t\tif bytes.Equal(b.branch, branch) {\n\t\t\treturn b.blob, nil\n\t\t}\n\t}\n\treturn nil, errors.New(\"failed reading history\")\n}\n"
  },
  {
    "path": "service/history/replication/task_processor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/reconciliation\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nconst (\n\tdropSyncShardTaskTimeThreshold = 10 * time.Minute\n\treplicationTimeout             = 30 * time.Second\n\tdlqErrorRetryWait              = time.Second\n\tdlqMetricsEmitTimerInterval    = 5 * time.Minute\n\tdlqMetricsEmitTimerCoefficient = 0.05\n)\n\nvar (\n\t// ErrUnknownReplicationTask is the error to indicate unknown replication task type\n\tErrUnknownReplicationTask = &types.BadRequestError{Message: \"unknown replication task\"}\n)\n\ntype (\n\t// TaskProcessor is responsible for processing replication tasks for a shard.\n\tTaskProcessor interface {\n\t\tcommon.Daemon\n\t}\n\n\t// taskProcessorImpl is responsible for processing replication tasks for a shard.\n\ttaskProcessorImpl struct {\n\t\tcurrentCluster    string\n\t\tsourceCluster     string\n\t\tstatus            int32\n\t\tshard             shard.Context\n\t\thistoryEngine     engine.Engine\n\t\thistorySerializer persistence.PayloadSerializer\n\t\tconfig            *config.Config\n\t\tmetricsClient     metrics.Client\n\t\tlogger            log.Logger\n\t\ttaskExecutor      TaskExecutor\n\t\thostRateLimiter   quotas.Limiter\n\t\tshardRateLimiter  quotas.Limiter\n\n\t\ttaskRetryPolicy backoff.RetryPolicy\n\t\tdlqRetryPolicy  backoff.RetryPolicy\n\t\tnoTaskRetrier   backoff.Retrier\n\n\t\tlastProcessedMessageID int64\n\t\tlastRetrievedMessageID int64\n\n\t\trequestChan   chan<- *request\n\t\tsyncShardChan chan *types.SyncShardStatus\n\t\tdone          chan struct{}\n\t\twg            sync.WaitGroup\n\t}\n\n\trequest struct {\n\t\ttoken    *types.ReplicationToken\n\t\trespChan chan<- *types.ReplicationMessages\n\t}\n)\n\nvar _ TaskProcessor = (*taskProcessorImpl)(nil)\n\n// NewTaskProcessor creates a new replication task processor.\nfunc NewTaskProcessor(\n\tshard shard.Context,\n\thistoryEngine engine.Engine,\n\tconfig *config.Config,\n\tmetricsClient metrics.Client,\n\ttaskFetcher TaskFetcher,\n\ttaskExecutor TaskExecutor,\n\tclock clock.TimeSource,\n) TaskProcessor {\n\tshardID := shard.GetShardID()\n\tsourceCluster := taskFetcher.GetSourceCluster()\n\tfirstRetryPolicy := backoff.NewExponentialRetryPolicy(config.ReplicationTaskProcessorErrorRetryWait(shardID))\n\tfirstRetryPolicy.SetMaximumAttempts(config.ReplicationTaskProcessorErrorRetryMaxAttempts(shardID))\n\tsecondRetryPolicy := backoff.NewExponentialRetryPolicy(config.ReplicationTaskProcessorErrorSecondRetryWait(shardID))\n\tsecondRetryPolicy.SetMaximumInterval(config.ReplicationTaskProcessorErrorSecondRetryMaxWait(shardID))\n\tsecondRetryPolicy.SetExpirationInterval(config.ReplicationTaskProcessorErrorSecondRetryExpiration(shardID))\n\ttaskRetryPolicy := backoff.NewMultiPhasesRetryPolicy(firstRetryPolicy, secondRetryPolicy)\n\n\tdlqRetryPolicy := backoff.NewExponentialRetryPolicy(dlqErrorRetryWait)\n\tdlqRetryPolicy.SetExpirationInterval(backoff.NoInterval)\n\n\tnoTaskBackoffPolicy := backoff.NewExponentialRetryPolicy(config.ReplicationTaskProcessorNoTaskRetryWait(shardID))\n\tnoTaskBackoffPolicy.SetBackoffCoefficient(1)\n\tnoTaskBackoffPolicy.SetExpirationInterval(backoff.NoInterval)\n\tnoTaskRetrier := backoff.NewRetrier(noTaskBackoffPolicy, clock)\n\treturn &taskProcessorImpl{\n\t\tcurrentCluster:         shard.GetClusterMetadata().GetCurrentClusterName(),\n\t\tsourceCluster:          sourceCluster,\n\t\tstatus:                 common.DaemonStatusInitialized,\n\t\tshard:                  shard,\n\t\thistoryEngine:          historyEngine,\n\t\thistorySerializer:      persistence.NewPayloadSerializer(),\n\t\tconfig:                 config,\n\t\tmetricsClient:          metricsClient,\n\t\tlogger:                 shard.GetLogger().WithTags(tag.SourceCluster(sourceCluster), tag.ShardID(shardID)),\n\t\ttaskExecutor:           taskExecutor,\n\t\thostRateLimiter:        taskFetcher.GetRateLimiter(),\n\t\tshardRateLimiter:       quotas.NewDynamicRateLimiter(config.ReplicationTaskProcessorShardQPS.AsFloat64()),\n\t\ttaskRetryPolicy:        taskRetryPolicy,\n\t\tdlqRetryPolicy:         dlqRetryPolicy,\n\t\tnoTaskRetrier:          noTaskRetrier,\n\t\trequestChan:            taskFetcher.GetRequestChan(shardID),\n\t\tsyncShardChan:          make(chan *types.SyncShardStatus, 1),\n\t\tdone:                   make(chan struct{}),\n\t\tlastProcessedMessageID: constants.EmptyMessageID,\n\t\tlastRetrievedMessageID: constants.EmptyMessageID,\n\t}\n}\n\n// Start starts the processor\nfunc (p *taskProcessorImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.wg.Add(3)\n\tgo p.processorLoop()\n\tgo p.syncShardStatusLoop()\n\tgo p.cleanupReplicationTaskLoop()\n\tp.logger.Info(\"ReplicationTaskProcessor started.\")\n}\n\n// Stop stops the processor\nfunc (p *taskProcessorImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tp.logger.Debug(\"ReplicationTaskProcessor shutting down.\")\n\tclose(p.done)\n\n\tif !common.AwaitWaitGroup(&p.wg, 10*time.Second) {\n\t\tp.logger.Warn(\"ReplicationTaskProcessor timed out on shutdown.\")\n\t} else {\n\t\tp.logger.Info(\"ReplicationTaskProcessor shutdown completed\")\n\t}\n}\n\nfunc (p *taskProcessorImpl) processorLoop() {\n\tdefer func() {\n\t\tp.logger.Debug(\"Closing replication task processor.\", tag.ReadLevel(p.lastRetrievedMessageID))\n\t\tp.wg.Done()\n\t}()\n\nLoop:\n\tfor {\n\t\trespChan := make(chan *types.ReplicationMessages, 1)\n\t\t// TODO: when we support prefetching, LastRetrievedMessageID can be different than LastProcessedMessageID\n\n\t\tif p.isShuttingDown() {\n\t\t\treturn\n\t\t}\n\n\t\tselect {\n\t\tcase <-p.done:\n\t\t\t// shard is closing\n\t\t\treturn\n\t\tcase p.requestChan <- &request{\n\t\t\ttoken: &types.ReplicationToken{\n\t\t\t\tShardID:                int32(p.shard.GetShardID()),\n\t\t\t\tLastRetrievedMessageID: p.lastRetrievedMessageID,\n\t\t\t\tLastProcessedMessageID: p.lastProcessedMessageID,\n\t\t\t},\n\t\t\trespChan: respChan,\n\t\t}:\n\t\t\t// signal sent, continue to process replication messages\n\t\t}\n\n\t\tselect {\n\t\tcase response, ok := <-respChan:\n\t\t\tif !ok {\n\t\t\t\tp.logger.Debug(\"Fetch replication messages chan closed.\")\n\t\t\t\tcontinue Loop\n\t\t\t}\n\n\t\t\tp.logger.Debug(\"Got fetch replication messages response.\",\n\t\t\t\ttag.ReadLevel(response.GetLastRetrievedMessageID()),\n\t\t\t\ttag.Bool(response.GetHasMore()),\n\t\t\t\ttag.Counter(len(response.GetReplicationTasks())),\n\t\t\t)\n\n\t\t\tp.taskProcessingStartWait()\n\t\t\tp.processResponse(response)\n\t\tcase <-p.done:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *taskProcessorImpl) cleanupReplicationTaskLoop() {\n\tdefer p.wg.Done()\n\n\tshardID := p.shard.GetShardID()\n\ttimer := time.NewTimer(backoff.JitDuration(\n\t\tp.config.ReplicationTaskProcessorCleanupInterval(shardID),\n\t\tp.config.ReplicationTaskProcessorCleanupJitterCoefficient(shardID),\n\t))\n\tdefer timer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.done:\n\t\t\treturn\n\t\tcase <-timer.C:\n\t\t\tif err := p.cleanupAckedReplicationTasks(); err != nil {\n\t\t\t\tp.logger.Error(\"Failed to clean up replication messages.\", tag.Error(err))\n\t\t\t\tp.metricsClient.Scope(metrics.ReplicationTaskCleanupScope).IncCounter(metrics.ReplicationTaskCleanupFailure)\n\t\t\t}\n\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\tp.config.ReplicationTaskProcessorCleanupInterval(shardID),\n\t\t\t\tp.config.ReplicationTaskProcessorCleanupJitterCoefficient(shardID),\n\t\t\t))\n\t\t}\n\t}\n}\n\nfunc (p *taskProcessorImpl) cleanupAckedReplicationTasks() error {\n\tminAckLevel := int64(math.MaxInt64)\n\tfor clusterName := range p.shard.GetClusterMetadata().GetRemoteClusterInfo() {\n\t\tackLevel := p.shard.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, clusterName).GetTaskID()\n\t\tif ackLevel < minAckLevel {\n\t\t\tminAckLevel = ackLevel\n\t\t}\n\t}\n\tp.logger.Debug(\"Cleaning up replication task queue.\", tag.ReadLevel(minAckLevel))\n\tp.metricsClient.Scope(metrics.ReplicationTaskCleanupScope).IncCounter(metrics.ReplicationTaskCleanupCount)\n\tmaxReadLevel := p.shard.UpdateIfNeededAndGetQueueMaxReadLevel(\n\t\tpersistence.HistoryTaskCategoryReplication,\n\t\tp.currentCluster,\n\t).GetTaskID()\n\tlagCount := int(maxReadLevel - minAckLevel)\n\tscope := p.metricsClient.Scope(metrics.ReplicationTaskFetcherScope,\n\t\tmetrics.TargetClusterTag(p.currentCluster),\n\t)\n\tscope.RecordTimer(metrics.ReplicationTasksLag, time.Duration(lagCount))\n\tscope.RecordHistogramValue(metrics.ReplicationTasksLagHistogram, float64(lagCount))\n\tfor {\n\t\tpageSize := p.config.ReplicatorTaskDeleteBatchSize()\n\t\tresp, err := p.shard.GetExecutionManager().RangeCompleteHistoryTask(\n\t\t\tcontext.Background(),\n\t\t\t&persistence.RangeCompleteHistoryTaskRequest{\n\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(minAckLevel + 1),\n\t\t\t\tPageSize:            pageSize,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif !persistence.HasMoreRowsToDelete(resp.TasksCompleted, pageSize) {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (p *taskProcessorImpl) processResponse(response *types.ReplicationMessages) {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tp.logger.Error(\"processResponse encountered panic.\", tag.Value(r))\n\t\t\tpanic(r)\n\t\t}\n\t}()\n\n\tselect {\n\tcase p.syncShardChan <- response.GetSyncShardStatus():\n\tdefault:\n\t}\n\n\tscope := p.metricsClient.Scope(metrics.ReplicationTaskFetcherScope, metrics.TargetClusterTag(p.sourceCluster))\n\tbatchRequestStartTime := time.Now()\n\tctx := context.Background()\n\tfor _, replicationTask := range response.ReplicationTasks {\n\t\t// TODO: move to MultiStageRateLimiter\n\t\t_ = p.hostRateLimiter.Wait(ctx)\n\t\t_ = p.shardRateLimiter.Wait(ctx)\n\t\terr := p.processSingleTask(replicationTask)\n\t\tif err != nil {\n\t\t\t// Encounter error and skip updating ack levels\n\t\t\t// TODO: Does this behavior make sense? If ack levels are not updated the whole batch will have to be re-fetched via processor loop.\n\t\t\t// Potential improvements:\n\t\t\t// 1. Update ack levels for the tasks that were processed successfully.\n\t\t\t// 2. Emit logs/metrics for these cases that we give up on updating ack levels.\n\t\t\treturn\n\t\t}\n\t}\n\n\t// Note here we check replication tasks instead of hasMore. The expectation is that in a steady state\n\t// we will receive replication tasks but hasMore is false (meaning that we are always catching up).\n\t// So hasMore might not be a good indicator for additional wait.\n\tif len(response.ReplicationTasks) == 0 {\n\t\tbackoffDuration := p.noTaskRetrier.NextBackOff()\n\t\ttime.Sleep(backoffDuration)\n\t} else {\n\t\tappliedLatency := time.Since(batchRequestStartTime)\n\t\tscope.RecordTimer(metrics.ReplicationTasksAppliedLatency, appliedLatency)\n\t\tscope.ExponentialHistogram(metrics.ReplicationTasksAppliedLatencyHistogram, appliedLatency)\n\t}\n\n\tif p.isShuttingDown() {\n\t\t// avoid updating ack-levels if there's a shutdown as well remembering that GetReplication messages is a *write* api\n\t\t// (keeping track of consumer offsets), so we have to be careful what data it's sent\n\t\treturn\n\t}\n\n\tp.lastProcessedMessageID = response.GetLastRetrievedMessageID()\n\tp.lastRetrievedMessageID = response.GetLastRetrievedMessageID()\n\tscope.UpdateGauge(metrics.LastRetrievedMessageID, float64(p.lastRetrievedMessageID))\n\tp.noTaskRetrier.Reset()\n}\n\nfunc (p *taskProcessorImpl) syncShardStatusLoop() {\n\tdefer p.wg.Done()\n\n\ttimer := time.NewTimer(backoff.JitDuration(\n\t\tp.config.ShardSyncMinInterval(),\n\t\tp.config.ShardSyncTimerJitterCoefficient(),\n\t))\n\tdefer timer.Stop()\n\n\tvar syncShardTask *types.SyncShardStatus\n\tfor {\n\t\tselect {\n\t\tcase syncShardRequest := <-p.syncShardChan:\n\t\t\tsyncShardTask = syncShardRequest\n\t\tcase <-timer.C:\n\t\t\tif err := p.handleSyncShardStatus(\n\t\t\t\tsyncShardTask,\n\t\t\t); err != nil {\n\t\t\t\tp.logger.Error(\"failed to sync shard status\", tag.Error(err))\n\t\t\t\tp.metricsClient.Scope(metrics.HistorySyncShardStatusScope).IncCounter(metrics.SyncShardFromRemoteFailure)\n\t\t\t}\n\t\t\ttimer.Reset(backoff.JitDuration(\n\t\t\t\tp.config.ShardSyncMinInterval(),\n\t\t\t\tp.config.ShardSyncTimerJitterCoefficient(),\n\t\t\t))\n\t\tcase <-p.done:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *taskProcessorImpl) handleSyncShardStatus(status *types.SyncShardStatus) error {\n\tif status == nil ||\n\t\tp.shard.GetTimeSource().Now().Sub(time.Unix(0, status.GetTimestamp())) > dropSyncShardTaskTimeThreshold {\n\t\treturn nil\n\t}\n\tp.metricsClient.Scope(metrics.HistorySyncShardStatusScope).IncCounter(metrics.SyncShardFromRemoteCounter)\n\tctx, cancel := context.WithTimeout(context.Background(), replicationTimeout)\n\tdefer cancel()\n\treturn p.historyEngine.SyncShardStatus(ctx, &types.SyncShardStatusRequest{\n\t\tSourceCluster: p.sourceCluster,\n\t\tShardID:       int64(p.shard.GetShardID()),\n\t\tTimestamp:     status.Timestamp,\n\t})\n}\n\nfunc (p *taskProcessorImpl) processSingleTask(replicationTask *types.ReplicationTask) error {\n\tretryTransientError := func(ctx context.Context) error {\n\t\tthrottleRetry := backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(p.taskRetryPolicy),\n\t\t\tbackoff.WithRetryableError(isTransientRetryableError),\n\t\t)\n\t\treturn throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\t\tselect {\n\t\t\tcase <-p.done:\n\t\t\t\t// if the processor is stopping, skip the task\n\t\t\t\t// the ack level will not update and the new shard owner will retry the task.\n\t\t\t\treturn nil\n\t\t\tdefault:\n\t\t\t\treturn p.processTaskOnce(replicationTask)\n\t\t\t}\n\t\t})\n\t}\n\n\t// Handle service busy error\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreateReplicationServiceBusyRetryPolicy()),\n\t\tbackoff.WithRetryableError(common.IsServiceBusyError),\n\t)\n\terr := throttleRetry.Do(context.Background(), retryTransientError)\n\n\tswitch {\n\tcase err == nil:\n\t\treturn nil\n\tcase common.IsServiceBusyError(err):\n\t\treturn err\n\tcase err == execution.ErrMissingVersionHistories:\n\t\t// skip the workflow without version histories\n\t\tp.logger.Warn(\"Encounter workflow without version histories\")\n\t\treturn nil\n\tdefault:\n\t\t// handle error\n\t}\n\n\t// handle error to DLQ\n\tselect {\n\tcase <-p.done:\n\t\tp.logger.Warn(\"Skip adding new messages to DLQ.\", tag.Error(err))\n\t\treturn err\n\tdefault:\n\t\trequest, err2 := p.generateDLQRequest(replicationTask)\n\t\tif err2 != nil {\n\t\t\tp.logger.Error(\"Failed to generate DLQ replication task.\", tag.Error(err2))\n\t\t\t// We cannot deserialize the task. Dropping it.\n\t\t\treturn nil\n\t\t}\n\t\tp.logger.Error(\"Failed to apply replication task after retry. Putting task into DLQ.\",\n\t\t\ttag.WorkflowDomainID(request.TaskInfo.GetDomainID()),\n\t\t\ttag.WorkflowID(request.TaskInfo.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(request.TaskInfo.GetRunID()),\n\t\t\ttag.TaskID(request.TaskInfo.GetTaskID()),\n\t\t\ttag.TaskType(request.TaskInfo.GetTaskType()),\n\t\t\ttag.Error(err),\n\t\t)\n\t\t// TODO: uncomment this when the execution fixer workflow is ready\n\t\t// if err = p.triggerDataInconsistencyScan(replicationTask); err != nil {\n\t\t//\tp.logger.Warn(\"Failed to trigger data scan\", tag.Error(err))\n\t\t//\tp.metricsClient.IncCounter(metrics.ReplicationDLQStatsScope, metrics.ReplicationDLQValidationFailed)\n\t\t// }\n\t\treturn p.putReplicationTaskToDLQ(request)\n\t}\n}\n\nfunc (p *taskProcessorImpl) processTaskOnce(replicationTask *types.ReplicationTask) error {\n\tts := p.shard.GetTimeSource()\n\tstartTime := ts.Now()\n\tscope, err := p.taskExecutor.execute(replicationTask, false)\n\n\tif err != nil {\n\t\tp.updateFailureMetric(scope, err, p.shard.GetShardID())\n\t} else {\n\t\tnow := ts.Now()\n\t\tmScope := p.metricsClient.Scope(scope, metrics.TargetClusterTag(p.sourceCluster))\n\t\tdomainID := replicationTask.HistoryTaskV2Attributes.GetDomainID()\n\t\tvar domainName string\n\t\tif domainID != \"\" {\n\t\t\tcachedName, errorDomainName := p.shard.GetDomainCache().GetDomainName(domainID)\n\t\t\tif errorDomainName != nil {\n\t\t\t\treturn errorDomainName\n\t\t\t}\n\t\t\tdomainName = cachedName\n\t\t}\n\t\tmScope = mScope.Tagged(metrics.DomainTag(domainName)) // use consistent tags so Prometheus does not break\n\n\t\t// emit single task processing latency\n\t\tmScope.ExponentialHistogram(metrics.ExponentialTaskProcessingLatency, now.Sub(startTime))\n\t\t// emit latency from task generated to task received\n\t\tmScope.ExponentialHistogram(\n\t\t\tmetrics.ExponentialReplicationTaskLatency,\n\t\t\tnow.Sub(time.Unix(0, replicationTask.GetCreationTime())),\n\t\t)\n\n\t\t// emit single task processing latency\n\t\tmScope.RecordTimer(metrics.TaskProcessingLatency, now.Sub(startTime))\n\t\te2eLatency := now.Sub(time.Unix(0, replicationTask.GetCreationTime()))\n\t\t// emit latency from task generated to task received\n\t\tmScope.RecordTimer(metrics.ReplicationTaskLatency, e2eLatency)\n\n\t\t// if latency is not switched off, and exceeds threshold, emit structured log\n\t\tif p.config.ReplicationTaskProcessorLatencyLogThreshold() > 0 && e2eLatency >= p.config.ReplicationTaskProcessorLatencyLogThreshold() {\n\t\t\tattr := replicationTask.GetHistoryTaskV2Attributes()\n\t\t\tvar workflowID, runID string\n\t\t\tif attr != nil {\n\t\t\t\tworkflowID = attr.GetWorkflowID()\n\t\t\t\trunID = attr.GetRunID()\n\t\t\t}\n\t\t\tp.logger.Warn(\"high replication latency\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\ttag.WorkflowRunID(runID),\n\t\t\t\ttag.ShardID(p.shard.GetShardID()),\n\t\t\t\ttag.SourceCluster(p.sourceCluster),\n\t\t\t)\n\t\t}\n\n\t\t// emit the number of replication tasks\n\t\tmScope.IncCounter(metrics.ReplicationTasksAppliedPerDomain)\n\t\tshardScope := p.metricsClient.Scope(scope, metrics.TargetClusterTag(p.sourceCluster), metrics.InstanceTag(strconv.Itoa(p.shard.GetShardID())))\n\t\tshardScope.IncCounter(metrics.ReplicationTasksApplied)\n\t}\n\n\treturn err\n}\n\nfunc (p *taskProcessorImpl) putReplicationTaskToDLQ(request *persistence.PutReplicationTaskToDLQRequest) error {\n\tp.metricsClient.Scope(\n\t\tmetrics.ReplicationDLQStatsScope,\n\t\tmetrics.TargetClusterTag(p.sourceCluster),\n\t\tmetrics.InstanceTag(strconv.Itoa(p.shard.GetShardID())),\n\t).UpdateGauge(\n\t\tmetrics.ReplicationDLQMaxLevelGauge,\n\t\tfloat64(request.TaskInfo.GetTaskID()),\n\t)\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(p.dlqRetryPolicy),\n\t\tbackoff.WithRetryableError(p.shouldRetryDLQ),\n\t)\n\t// The following is guaranteed to success or retry forever until processor is shutdown.\n\treturn throttleRetry.Do(context.Background(), func(ctx context.Context) error {\n\t\terr := p.shard.GetExecutionManager().PutReplicationTaskToDLQ(ctx, request)\n\t\tif err != nil {\n\t\t\tp.logger.Error(\"Failed to put replication task to DLQ.\", tag.Error(err))\n\t\t\tp.metricsClient.IncCounter(metrics.ReplicationTaskFetcherScope, metrics.ReplicationDLQFailed)\n\t\t}\n\t\treturn err\n\t})\n}\n\nfunc (p *taskProcessorImpl) generateDLQRequest(\n\treplicationTask *types.ReplicationTask,\n) (*persistence.PutReplicationTaskToDLQRequest, error) {\n\tswitch *replicationTask.TaskType {\n\tcase types.ReplicationTaskTypeSyncActivity:\n\t\ttaskAttributes := replicationTask.GetSyncActivityTaskAttributes()\n\t\tdomainName, err := p.shard.GetDomainCache().GetDomainName(taskAttributes.GetDomainID())\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn &persistence.PutReplicationTaskToDLQRequest{\n\t\t\tSourceClusterName: p.sourceCluster,\n\t\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\t\tDomainID:    taskAttributes.GetDomainID(),\n\t\t\t\tWorkflowID:  taskAttributes.GetWorkflowID(),\n\t\t\t\tRunID:       taskAttributes.GetRunID(),\n\t\t\t\tTaskID:      replicationTask.GetSourceTaskID(),\n\t\t\t\tTaskType:    persistence.ReplicationTaskTypeSyncActivity,\n\t\t\t\tScheduledID: taskAttributes.GetScheduledID(),\n\t\t\t},\n\t\t\tDomainName: domainName,\n\t\t}, nil\n\n\tcase types.ReplicationTaskTypeHistoryV2:\n\t\ttaskAttributes := replicationTask.GetHistoryTaskV2Attributes()\n\t\tdomainName, err := p.shard.GetDomainCache().GetDomainName(taskAttributes.GetDomainID())\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\teventsDataBlob := persistence.NewDataBlobFromInternal(taskAttributes.GetEvents())\n\t\tevents, err := p.historySerializer.DeserializeBatchEvents(eventsDataBlob)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif len(events) == 0 {\n\t\t\tp.logger.Error(\"Empty events in a batch\")\n\t\t\treturn nil, fmt.Errorf(\"corrupted history event batch, empty events\")\n\t\t}\n\n\t\treturn &persistence.PutReplicationTaskToDLQRequest{\n\t\t\tSourceClusterName: p.sourceCluster,\n\t\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\t\tDomainID:     taskAttributes.GetDomainID(),\n\t\t\t\tWorkflowID:   taskAttributes.GetWorkflowID(),\n\t\t\t\tRunID:        taskAttributes.GetRunID(),\n\t\t\t\tTaskID:       replicationTask.GetSourceTaskID(),\n\t\t\t\tTaskType:     persistence.ReplicationTaskTypeHistory,\n\t\t\t\tFirstEventID: events[0].ID,\n\t\t\t\tNextEventID:  events[len(events)-1].ID + 1,\n\t\t\t\tVersion:      events[0].Version,\n\t\t\t},\n\t\t\tDomainName: domainName,\n\t\t}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unknown replication task type\")\n\t}\n}\n\nfunc (p *taskProcessorImpl) triggerDataInconsistencyScan(replicationTask *types.ReplicationTask) error {\n\tvar failoverVersion int64\n\tvar domainID string\n\tvar workflowID string\n\tvar runID string\n\tswitch {\n\tcase replicationTask.GetHistoryTaskV2Attributes() != nil:\n\t\tattr := replicationTask.GetHistoryTaskV2Attributes()\n\t\tversionHistoryItems := attr.GetVersionHistoryItems()\n\t\tif len(versionHistoryItems) == 0 {\n\t\t\treturn errors.New(\"failed to trigger data scan due to invalid version history\")\n\t\t}\n\t\t// version history items in same batch should be the same\n\t\tfailoverVersion = versionHistoryItems[0].GetVersion()\n\t\tdomainID = attr.GetDomainID()\n\t\tworkflowID = attr.GetWorkflowID()\n\t\trunID = attr.GetRunID()\n\tcase replicationTask.GetSyncActivityTaskAttributes() != nil:\n\t\tattr := replicationTask.GetSyncActivityTaskAttributes()\n\t\tfailoverVersion = replicationTask.GetSyncActivityTaskAttributes().Version\n\t\tdomainID = attr.GetDomainID()\n\t\tworkflowID = attr.GetWorkflowID()\n\t\trunID = attr.GetRunID()\n\tdefault:\n\t\treturn nil\n\t}\n\tclusterName, err := p.shard.GetClusterMetadata().ClusterNameForFailoverVersion(failoverVersion)\n\tif err != nil {\n\t\treturn err\n\t}\n\tclient, err := p.shard.GetService().GetClientBean().GetRemoteFrontendClient(clusterName)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfixExecution := entity.Execution{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tShardID:    p.shard.GetShardID(),\n\t}\n\tfixExecutionInput, err := json.Marshal(fixExecution)\n\tif err != nil {\n\t\treturn err\n\t}\n\t// Assume the workflow is corrupted, rely on invariant to validate it\n\t_, err = client.SignalWithStartWorkflowExecution(context.Background(), &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              constants.SystemLocalDomainName,\n\t\tWorkflowID:                          reconciliation.CheckDataCorruptionWorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: reconciliation.CheckDataCorruptionWorkflowType},\n\t\tTaskList:                            &types.TaskList{Name: reconciliation.CheckDataCorruptionWorkflowTaskList},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(reconciliation.CheckDataCorruptionWorkflowTimeoutInSeconds),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(reconciliation.CheckDataCorruptionWorkflowTaskTimeoutInSeconds),\n\t\tIdentity:                            \"cadence-history-replication\",\n\t\tRequestID:                           uuid.New(),\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\tSignalName:                          reconciliation.CheckDataCorruptionWorkflowSignalName,\n\t\tSignalInput:                         fixExecutionInput,\n\t})\n\treturn err\n}\n\nfunc isTransientRetryableError(err error) bool {\n\tswitch err.(type) {\n\tcase *types.BadRequestError:\n\t\treturn false\n\tcase *types.ServiceBusyError:\n\t\treturn false\n\tdefault:\n\t\treturn true\n\t}\n}\n\nfunc (p *taskProcessorImpl) shouldRetryDLQ(err error) bool {\n\tif err == nil {\n\t\treturn false\n\t}\n\n\tselect {\n\tcase <-p.done:\n\t\tp.logger.Debug(\"ReplicationTaskProcessor shutting down.\")\n\t\treturn false\n\tdefault:\n\t\treturn true\n\t}\n}\n\nfunc (p *taskProcessorImpl) updateFailureMetric(scope metrics.ScopeIdx, err error, shardID int) {\n\t// Always update failure counter for all replicator errors\n\tshardScope := p.metricsClient.Scope(scope, metrics.InstanceTag(strconv.Itoa(shardID)))\n\tshardScope.IncCounter(metrics.ReplicatorFailures)\n\n\t// Also update counter to distinguish between type of failures\n\tswitch err := err.(type) {\n\tcase *types.ShardOwnershipLostError:\n\t\tshardScope.IncCounter(metrics.CadenceErrShardOwnershipLostCounter)\n\tcase *types.BadRequestError:\n\t\tshardScope.IncCounter(metrics.CadenceErrBadRequestCounter)\n\tcase *types.DomainNotActiveError:\n\t\tshardScope.IncCounter(metrics.CadenceErrDomainNotActiveCounter)\n\tcase *types.WorkflowExecutionAlreadyStartedError:\n\t\tshardScope.IncCounter(metrics.CadenceErrExecutionAlreadyStartedCounter)\n\tcase *types.EntityNotExistsError:\n\t\tshardScope.IncCounter(metrics.CadenceErrEntityNotExistsCounter)\n\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\tshardScope.IncCounter(metrics.CadenceErrWorkflowExecutionAlreadyCompletedCounter)\n\tcase *types.LimitExceededError:\n\t\tshardScope.IncCounter(metrics.CadenceErrLimitExceededCounter)\n\tcase *yarpcerrors.Status:\n\t\tif err.Code() == yarpcerrors.CodeDeadlineExceeded {\n\t\t\tshardScope.IncCounter(metrics.CadenceErrContextTimeoutCounter)\n\t\t}\n\t}\n}\n\nfunc (p *taskProcessorImpl) taskProcessingStartWait() {\n\tshardID := p.shard.GetShardID()\n\ttime.Sleep(backoff.JitDuration(\n\t\tp.config.ReplicationTaskProcessorStartWait(shardID),\n\t\tp.config.ReplicationTaskProcessorStartWaitJitterCoefficient(shardID),\n\t))\n}\n\nfunc (p *taskProcessorImpl) isShuttingDown() bool {\n\tselect {\n\tcase <-p.done:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "service/history/replication/task_processor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/reconciliation\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\ttaskProcessorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockEngine         *engine.MockEngine\n\t\tconfig             *config.Config\n\t\tmockDomainCache    *cache.MockDomainCache\n\t\tmockClientBean     *client.MockBean\n\t\tmockFrontendClient *frontend.MockClient\n\t\tadminClient        *admin.MockClient\n\t\texecutionManager   *mocks.ExecutionManager\n\t\trequestChan        chan *request\n\t\ttaskFetcher        *fakeTaskFetcher\n\t\ttaskExecutor       *MockTaskExecutor\n\t\ttaskProcessor      *taskProcessorImpl\n\t\tclock              clock.MockedTimeSource\n\t}\n)\n\ntype fakeTaskFetcher struct {\n\tsourceCluster string\n\trequestChan   chan *request\n\trateLimiter   quotas.Limiter\n}\n\nfunc (f fakeTaskFetcher) Start() {}\nfunc (f fakeTaskFetcher) Stop()  {}\nfunc (f fakeTaskFetcher) GetSourceCluster() string {\n\treturn f.sourceCluster\n}\nfunc (f fakeTaskFetcher) GetRequestChan(shardID int) chan<- *request {\n\treturn f.requestChan\n}\nfunc (f fakeTaskFetcher) GetRateLimiter() quotas.Limiter {\n\treturn f.rateLimiter\n}\n\nfunc TestTaskProcessorSuite(t *testing.T) {\n\ts := new(taskProcessorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *taskProcessorSuite) SetupSuite() {\n\n}\n\nfunc (s *taskProcessorSuite) TearDownSuite() {\n\n}\n\nfunc (s *taskProcessorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:                 0,\n\t\t\tRangeID:                 1,\n\t\t\tTransferAckLevel:        0,\n\t\t\tClusterReplicationLevel: map[string]int64{cluster.TestAlternativeClusterName: 350},\n\t\t},\n\t\ts.config,\n\t)\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockClientBean = s.mockShard.Resource.ClientBean\n\ts.mockFrontendClient = s.mockShard.Resource.RemoteFrontendClient\n\ts.adminClient = s.mockShard.Resource.RemoteAdminClient\n\ts.executionManager = s.mockShard.Resource.ExecutionMgr\n\ts.clock = clock.NewMockedTimeSource()\n\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.config = config.NewForTest()\n\ts.config.ReplicationTaskProcessorNoTaskRetryWait = dynamicproperties.GetDurationPropertyFnFilteredByShardID(1 * time.Millisecond)\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.requestChan = make(chan *request, 10)\n\n\ts.taskFetcher = &fakeTaskFetcher{\n\t\tsourceCluster: \"standby\",\n\t\trequestChan:   s.requestChan,\n\t\trateLimiter:   quotas.NewDynamicRateLimiter(func() float64 { return 100 }),\n\t}\n\n\ts.taskExecutor = NewMockTaskExecutor(s.controller)\n\n\ts.taskProcessor = NewTaskProcessor(\n\t\ts.mockShard,\n\t\ts.mockEngine,\n\t\ts.config,\n\t\tmetricsClient,\n\t\ts.taskFetcher,\n\t\ts.taskExecutor,\n\t\ts.clock,\n\t).(*taskProcessorImpl)\n}\n\nfunc (s *taskProcessorSuite) TearDownTest() {\n\ts.mockShard.Finish(s.T())\n\tgoleak.VerifyNone(s.T())\n}\n\nfunc (s *taskProcessorSuite) TestStartStop() {\n\ts.taskProcessor.Start()\n\ts.taskProcessor.Stop()\n}\n\nfunc (s *taskProcessorSuite) TestProcessResponse_NoTask() {\n\tresponse := &types.ReplicationMessages{\n\t\tLastRetrievedMessageID: 100,\n\t}\n\n\ts.taskProcessor.processResponse(response)\n\ts.Equal(int64(100), s.taskProcessor.lastProcessedMessageID)\n\ts.Equal(int64(100), s.taskProcessor.lastRetrievedMessageID)\n}\n\nfunc (s *taskProcessorSuite) TestProcessorLoop_RequestChanPopulated() {\n\t// start the process loop so it poppulates requestChan\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.processorLoop()\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n\n\t// check the request\n\trequestMessage := <-s.requestChan\n\n\ts.Equal(int32(0), requestMessage.token.GetShardID())\n\ts.Equal(int64(-1), requestMessage.token.GetLastProcessedMessageID())\n\ts.Equal(int64(-1), requestMessage.token.GetLastRetrievedMessageID())\n\ts.NotNil(requestMessage.respChan)\n}\n\nfunc (s *taskProcessorSuite) TestProcessorLoop_RespChanClosed() {\n\t// start the process loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.processorLoop()\n\tdefer close(s.taskProcessor.done)\n\n\t// act like taskFetcher here and populate respChan of the request\n\trequestMessage := <-s.requestChan\n\tclose(requestMessage.respChan)\n\n\t// loop should have continued by now. validate by checking the new request\n\tselect {\n\tcase <-s.requestChan:\n\t// expected\n\tcase <-time.After(50 * time.Millisecond):\n\t\ts.Fail(\"new request not sent to requestChan\")\n\t}\n}\n\nfunc (s *taskProcessorSuite) TestProcessorLoop_TaskExecuteSuccess() {\n\t// taskExecutor will fail to execute the task\n\t// returning a non-retriable task to keep mocking simpler\n\ts.taskExecutor.EXPECT().execute(gomock.Any(), false).Return(metrics.ScopeIdx(0), nil).Times(1)\n\n\t// domain name will be fetched\n\ts.mockDomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).AnyTimes()\n\n\t// start the process loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.processorLoop()\n\n\t// act like taskFetcher here and populate respChan of the request\n\trequestMessage := <-s.requestChan\n\trequestMessage.respChan <- &types.ReplicationMessages{\n\t\tLastRetrievedMessageID: 100,\n\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t{\n\t\t\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:    testDomainID,\n\t\t\t\t\tWorkflowID:  testWorkflowID,\n\t\t\t\t\tRunID:       testRunID,\n\t\t\t\t\tScheduledID: testScheduleID,\n\t\t\t\t},\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t},\n\t\t},\n\t}\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n}\n\nfunc (s *taskProcessorSuite) TestProcessorLoop_TaskExecuteFailed_PutDLQSuccess() {\n\t// taskExecutor will fail to execute the task\n\t// returning a non-retriable task to keep mocking simpler\n\ts.taskExecutor.EXPECT().execute(gomock.Any(), false).Return(metrics.ScopeIdx(0), &types.BadRequestError{}).Times(1)\n\n\t// domain name will be fetched\n\ts.mockDomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).AnyTimes()\n\n\t// task will be put into dlq\n\tdlqReq := &persistence.PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: \"standby\", // TODO move to a constant\n\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\tDomainID:    testDomainID,\n\t\t\tWorkflowID:  testWorkflowID,\n\t\t\tRunID:       testRunID,\n\t\t\tTaskID:      testTaskID,\n\t\t\tTaskType:    persistence.ReplicationTaskTypeSyncActivity,\n\t\t\tScheduledID: testScheduleID,\n\t\t},\n\t\tDomainName: testDomainName,\n\t}\n\ts.mockShard.Resource.ExecutionMgr.On(\"PutReplicationTaskToDLQ\", mock.Anything, dlqReq).Return(nil).Times(1)\n\n\t// start the process loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.processorLoop()\n\n\t// act like taskFetcher here and populate respChan of the request\n\trequestMessage := <-s.requestChan\n\trequestMessage.respChan <- &types.ReplicationMessages{\n\t\tLastRetrievedMessageID: 100,\n\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t{\n\t\t\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:    testDomainID,\n\t\t\t\t\tWorkflowID:  testWorkflowID,\n\t\t\t\t\tRunID:       testRunID,\n\t\t\t\t\tScheduledID: testScheduleID,\n\t\t\t\t},\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t},\n\t\t},\n\t}\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n}\n\nfunc (s *taskProcessorSuite) TestProcessorLoop_TaskExecuteFailed_PutDLQFailed() {\n\t// taskExecutor will fail to execute the task\n\t// returning a non-retriable task to keep mocking simpler\n\ts.taskExecutor.EXPECT().execute(gomock.Any(), false).Return(metrics.ScopeIdx(0), &types.BadRequestError{}).Times(1)\n\n\t// domain name will be fetched\n\ts.mockDomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).AnyTimes()\n\n\t// task will be put into dlq and will fail. It will be attempted 3 times. (first call + 2 retries based on policy overriden below)\n\tdqlRetryPolicy := backoff.NewExponentialRetryPolicy(time.Millisecond)\n\tdqlRetryPolicy.SetMaximumAttempts(2)\n\ts.taskProcessor.dlqRetryPolicy = dqlRetryPolicy\n\tdlqReq := &persistence.PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: \"standby\", // TODO move to a constant\n\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\tDomainID:    testDomainID,\n\t\t\tWorkflowID:  testWorkflowID,\n\t\t\tRunID:       testRunID,\n\t\t\tTaskID:      testTaskID,\n\t\t\tTaskType:    persistence.ReplicationTaskTypeSyncActivity,\n\t\t\tScheduledID: testScheduleID,\n\t\t},\n\t\tDomainName: testDomainName,\n\t}\n\ts.mockShard.Resource.ExecutionMgr.\n\t\tOn(\"PutReplicationTaskToDLQ\", mock.Anything, dlqReq).\n\t\tReturn(errors.New(\"failed to put to dlq\")).\n\t\tTimes(3)\n\n\t// start the process loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.processorLoop()\n\n\t// act like taskFetcher here and populate respChan of the request\n\trequestMessage := <-s.requestChan\n\trequestMessage.respChan <- &types.ReplicationMessages{\n\t\tLastRetrievedMessageID: 100,\n\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t{\n\t\t\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:    testDomainID,\n\t\t\t\t\tWorkflowID:  testWorkflowID,\n\t\t\t\t\tRunID:       testRunID,\n\t\t\t\t\tScheduledID: testScheduleID,\n\t\t\t\t},\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t},\n\t\t},\n\t}\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n}\n\nfunc (s *taskProcessorSuite) TestHandleSyncShardStatus() {\n\tnow := time.Now()\n\ts.mockEngine.EXPECT().SyncShardStatus(gomock.Any(), &types.SyncShardStatusRequest{\n\t\tSourceCluster: \"standby\",\n\t\tShardID:       0,\n\t\tTimestamp:     common.Int64Ptr(now.UnixNano()),\n\t}).Return(nil).Times(1)\n\n\terr := s.taskProcessor.handleSyncShardStatus(&types.SyncShardStatus{\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t})\n\ts.NoError(err)\n}\n\nfunc (s *taskProcessorSuite) TestPutReplicationTaskToDLQ_SyncActivityReplicationTask() {\n\trequest := &persistence.PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: \"standby\",\n\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\tDomainID:   uuid.New(),\n\t\t\tWorkflowID: uuid.New(),\n\t\t\tRunID:      uuid.New(),\n\t\t\tTaskType:   persistence.ReplicationTaskTypeSyncActivity,\n\t\t},\n\t\tDomainName: uuid.New(),\n\t}\n\ts.executionManager.On(\"PutReplicationTaskToDLQ\", mock.Anything, request).Return(nil)\n\terr := s.taskProcessor.putReplicationTaskToDLQ(request)\n\ts.NoError(err)\n}\n\nfunc (s *taskProcessorSuite) TestPutReplicationTaskToDLQ_HistoryV2ReplicationTask() {\n\trequest := &persistence.PutReplicationTaskToDLQRequest{\n\t\tSourceClusterName: \"standby\",\n\t\tTaskInfo: &persistence.ReplicationTaskInfo{\n\t\t\tDomainID:     uuid.New(),\n\t\t\tWorkflowID:   uuid.New(),\n\t\t\tRunID:        uuid.New(),\n\t\t\tTaskType:     persistence.ReplicationTaskTypeHistory,\n\t\t\tFirstEventID: 1,\n\t\t\tNextEventID:  2,\n\t\t\tVersion:      1,\n\t\t},\n\t\tDomainName: uuid.New(),\n\t}\n\ts.executionManager.On(\"PutReplicationTaskToDLQ\", mock.Anything, request).Return(nil)\n\terr := s.taskProcessor.putReplicationTaskToDLQ(request)\n\ts.NoError(err)\n}\n\nfunc (s *taskProcessorSuite) TestGenerateDLQRequest_ReplicationTaskTypeHistoryV2() {\n\tdomainID := uuid.New()\n\tworkflowID := uuid.New()\n\trunID := uuid.New()\n\tevents := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:      1,\n\t\t\tVersion: 1,\n\t\t},\n\t}\n\tserializer := s.mockShard.GetPayloadSerializer()\n\tdata, err := serializer.SerializeBatchEvents(events, constants.EncodingTypeThriftRW)\n\ts.NoError(err)\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tEvents: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\tData:         data.Data,\n\t\t\t},\n\t\t},\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\trequest, err := s.taskProcessor.generateDLQRequest(task)\n\ts.NoError(err)\n\ts.Equal(\"standby\", request.SourceClusterName)\n\ts.Equal(int64(1), request.TaskInfo.FirstEventID)\n\ts.Equal(int64(2), request.TaskInfo.NextEventID)\n\ts.Equal(int64(1), request.TaskInfo.GetVersion())\n\ts.Equal(domainID, request.TaskInfo.GetDomainID())\n\ts.Equal(workflowID, request.TaskInfo.GetWorkflowID())\n\ts.Equal(runID, request.TaskInfo.GetRunID())\n\ts.Equal(persistence.ReplicationTaskTypeHistory, request.TaskInfo.GetTaskType())\n}\n\nfunc (s *taskProcessorSuite) TestGenerateDLQRequest_ReplicationTaskTypeSyncActivity() {\n\tdomainID := uuid.New()\n\tworkflowID := uuid.New()\n\trunID := uuid.New()\n\tdomainName := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:    domainID,\n\t\t\tWorkflowID:  workflowID,\n\t\t\tRunID:       runID,\n\t\t\tScheduledID: 1,\n\t\t},\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(domainID).Return(domainName, nil).AnyTimes()\n\trequest, err := s.taskProcessor.generateDLQRequest(task)\n\ts.NoError(err)\n\ts.Equal(\"standby\", request.SourceClusterName)\n\ts.Equal(int64(1), request.TaskInfo.ScheduledID)\n\ts.Equal(domainID, request.TaskInfo.GetDomainID())\n\ts.Equal(workflowID, request.TaskInfo.GetWorkflowID())\n\ts.Equal(runID, request.TaskInfo.GetRunID())\n\ts.Equal(persistence.ReplicationTaskTypeSyncActivity, request.TaskInfo.GetTaskType())\n}\n\nfunc (s *taskProcessorSuite) TestGenerateDLQRequest_InvalidTaskType() {\n\tdomainID := uuid.New()\n\tworkflowID := uuid.New()\n\trunID := uuid.New()\n\tevents := []*types.HistoryEvent{\n\t\t{\n\t\t\tID:      1,\n\t\t\tVersion: 1,\n\t\t},\n\t}\n\tserializer := s.mockShard.GetPayloadSerializer()\n\tdata, err := serializer.SerializeBatchEvents(events, constants.EncodingTypeThriftRW)\n\ts.NoError(err)\n\ttaskType := types.ReplicationTaskType(-1)\n\ttask := &types.ReplicationTask{\n\t\tTaskType: &taskType,\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   domainID,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t\tEvents: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\tData:         data.Data,\n\t\t\t},\n\t\t},\n\t}\n\n\t_, err = s.taskProcessor.generateDLQRequest(task)\n\ts.ErrorContains(err, \"unknown replication task type\")\n}\n\nfunc (s *taskProcessorSuite) TestTriggerDataInconsistencyScan_Success() {\n\tdomainID := uuid.New()\n\tworkflowID := uuid.New()\n\trunID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\tDomainID:    domainID,\n\t\t\tWorkflowID:  workflowID,\n\t\t\tRunID:       runID,\n\t\t\tScheduledID: 1,\n\t\t\tVersion:     100,\n\t\t},\n\t}\n\tfixExecution := entity.Execution{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: workflowID,\n\t\tRunID:      runID,\n\t\tShardID:    s.mockShard.GetShardID(),\n\t}\n\tjsArray, err := json.Marshal(fixExecution)\n\ts.NoError(err)\n\ts.mockFrontendClient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, request *types.SignalWithStartWorkflowExecutionRequest, option ...yarpc.CallOption) {\n\t\t\ts.Equal(constants.SystemLocalDomainName, request.GetDomain())\n\t\t\ts.Equal(reconciliation.CheckDataCorruptionWorkflowID, request.GetWorkflowID())\n\t\t\ts.Equal(reconciliation.CheckDataCorruptionWorkflowType, request.GetWorkflowType().GetName())\n\t\t\ts.Equal(reconciliation.CheckDataCorruptionWorkflowTaskList, request.GetTaskList().GetName())\n\t\t\ts.Equal(types.WorkflowIDReusePolicyAllowDuplicate.String(), request.GetWorkflowIDReusePolicy().String())\n\t\t\ts.Equal(reconciliation.CheckDataCorruptionWorkflowSignalName, request.GetSignalName())\n\t\t\ts.Equal(jsArray, request.GetSignalInput())\n\t\t}).Return(&types.StartWorkflowExecutionResponse{}, nil)\n\n\terr = s.taskProcessor.triggerDataInconsistencyScan(task)\n\ts.NoError(err)\n}\n\nfunc (s *taskProcessorSuite) TestCleanupReplicationTaskLoop() {\n\treq := &persistence.RangeCompleteHistoryTaskRequest{\n\t\t// this is min ack level of remote clusters. there's only one remote cluster in this test \"standby\".\n\t\t// its replication ack level is set to 350 in SetupTest(), and since the max key is exclusive, set the task id to 351\n\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(351),\n\t\tPageSize:            50, // this comes from test config\n\t}\n\ts.executionManager.On(\"RangeCompleteHistoryTask\", mock.Anything, req).Return(&persistence.RangeCompleteHistoryTaskResponse{\n\t\tTasksCompleted: 50, // if this number equals to page size the loop continues\n\t}, nil).Times(1)\n\ts.executionManager.On(\"RangeCompleteHistoryTask\", mock.Anything, req).Return(&persistence.RangeCompleteHistoryTaskResponse{\n\t\tTasksCompleted: 15, // if this number is different than page size the loop breaks\n\t}, nil)\n\n\t// start the cleanup loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.cleanupReplicationTaskLoop()\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n\n\t// wait until goroutine terminates\n\ts.taskProcessor.wg.Wait()\n}\n\nfunc (s *taskProcessorSuite) TestSyncShardStatusLoop_WithoutSyncShardTask() {\n\t// start the sync shard loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.syncShardStatusLoop()\n\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\tclose(s.taskProcessor.done)\n\n\t// wait until goroutine terminates\n\ts.taskProcessor.wg.Wait()\n}\n\nfunc (s *taskProcessorSuite) TestSyncShardStatusLoop_WithSyncShardTask() {\n\tnow := time.Now()\n\ts.taskProcessor.syncShardChan <- &types.SyncShardStatus{\n\t\tTimestamp: common.Int64Ptr(now.UnixNano()),\n\t}\n\ts.mockEngine.EXPECT().SyncShardStatus(gomock.Any(), &types.SyncShardStatusRequest{\n\t\tSourceCluster: \"standby\",\n\t\tShardID:       0,\n\t\tTimestamp:     common.Int64Ptr(now.UnixNano()),\n\t}).DoAndReturn(func(ctx context.Context, request *types.SyncShardStatusRequest) error {\n\t\tclose(s.taskProcessor.done)\n\t\treturn nil\n\t}).Times(1)\n\n\t// start the sync shard loop\n\ts.taskProcessor.wg.Add(1)\n\tgo s.taskProcessor.syncShardStatusLoop()\n\n\t// wait until goroutine terminates\n\ts.taskProcessor.wg.Wait()\n}\n\nfunc (s *taskProcessorSuite) TestShouldRetryDLQ() {\n\ts.False(s.taskProcessor.shouldRetryDLQ(nil))\n\ts.True(s.taskProcessor.shouldRetryDLQ(&types.InternalServiceError{}))\n\ts.True(s.taskProcessor.shouldRetryDLQ(fmt.Errorf(\"error before done channel closed should be retried\")))\n\tclose(s.taskProcessor.done)\n\ts.False(s.taskProcessor.shouldRetryDLQ(&types.ServiceBusyError{}))\n\ts.False(s.taskProcessor.shouldRetryDLQ(fmt.Errorf(\"error after done channel closed should NOT be retried\")))\n}\n\nfunc TestProcessorLoop_TaskExecuteFailed_ShardChangeErr(t *testing.T) {\n\n\tctrl := gomock.NewController(t)\n\tconfig := config.NewForTest()\n\tmockShard := shard.NewTestContext(\n\t\tt,\n\t\tctrl,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:                 0,\n\t\t\tRangeID:                 1,\n\t\t\tTransferAckLevel:        0,\n\t\t\tClusterReplicationLevel: map[string]int64{cluster.TestAlternativeClusterName: 350},\n\t\t},\n\t\tconfig,\n\t)\n\n\tmockDomainCache := mockShard.Resource.DomainCache\n\n\trequestChan := make(chan *request, 10)\n\n\ttaskFetcher := &fakeTaskFetcher{\n\t\tsourceCluster: \"standby\",\n\t\trequestChan:   requestChan,\n\t\t// ensure that the fetcher always nearly-immediately fetches\n\t\trateLimiter: quotas.NewDynamicRateLimiter(func() float64 { return 100000 }),\n\t}\n\n\tmockEngine := engine.NewMockEngine(ctrl)\n\tmetricsClient := metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\n\ttaskExecutor := NewMockTaskExecutor(ctrl)\n\n\ttaskProcessor := NewTaskProcessor(\n\t\tmockShard,\n\t\tmockEngine,\n\t\tconfig,\n\t\tmetricsClient,\n\t\ttaskFetcher,\n\t\ttaskExecutor,\n\t\tclock.NewMockedTimeSource(),\n\t).(*taskProcessorImpl)\n\n\t// start the process loop\n\ttaskProcessor.wg.Add(1)\n\tgo taskProcessor.processorLoop()\n\n\ttaskExecutor.EXPECT().execute(gomock.Any(), false).\n\t\tDoAndReturn(func(*types.ReplicationTask, bool) (any, any) {\n\t\t\t// take a minute like a real RPC call, enough time for it probably be doing it's thing\n\t\t\t// then to fail, return the shard error (as if the host is closing down) and then\n\t\t\t// trigger the Stop function as if the shard is properly closing.\n\t\t\ttime.Sleep(time.Millisecond * 50)\n\t\t\treturn 0, &persistence.ShardOwnershipLostError{Msg: \"some shard err\"}\n\t\t}).AnyTimes()\n\n\t// domain name will be fetched\n\tmockDomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).AnyTimes()\n\n\t// act like taskFetcher here and populate respChan of the request\n\trequestMessage := <-requestChan\n\trequestMessage.respChan <- &types.ReplicationMessages{\n\t\tLastRetrievedMessageID: int64(105),\n\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t{\n\t\t\t\tTaskType: types.ReplicationTaskTypeSyncActivity.Ptr(),\n\t\t\t\tSyncActivityTaskAttributes: &types.SyncActivityTaskAttributes{\n\t\t\t\t\tDomainID:    testDomainID,\n\t\t\t\t\tWorkflowID:  testWorkflowID,\n\t\t\t\t\tRunID:       testRunID,\n\t\t\t\t\tScheduledID: testScheduleID,\n\t\t\t\t},\n\t\t\t\tSourceTaskID: testTaskID,\n\t\t\t},\n\t\t},\n\t}\n\tif len(taskProcessor.requestChan) != 0 {\n\t\tt.Error(\"there shoudn't have been any data sent\")\n\t}\n\ttime.Sleep(50 * time.Millisecond)\n\n\tclose(taskProcessor.done)\n\t// wait a bit and terminate the loop\n\ttime.Sleep(50 * time.Millisecond)\n\n\ttaskProcessor.wg.Wait()\n\t// this is a rather complicated test, and this is the main assertion:\n\t// that *if* there's some shutdown logic thats going on, and in addition to that\n\t// there's replication tasks that can't be processed because there's shard stealing\n\t// going on, then we should expect that these in-memory offsets aren't changed and,\n\t// more importantly, aren't sent until they're successfully processed by a shard.\n\tassert.Equal(t, int64(-1), taskProcessor.lastProcessedMessageID)\n\tassert.Equal(t, int64(-1), taskProcessor.lastRetrievedMessageID)\n}\n\nfunc TestIsShuttingDown(t *testing.T) {\n\ttaskProcessor := taskProcessorImpl{\n\t\tdone: make(chan struct{}),\n\t}\n\tassert.False(t, taskProcessor.isShuttingDown())\n\tclose(taskProcessor.done)\n\tassert.True(t, taskProcessor.isShuttingDown())\n}\n\nfunc (s *taskProcessorSuite) TestProcessTaskOnce_OverThreshold_WithNoopLogger() {\n\ts.taskExecutor.EXPECT().execute(gomock.Any(), false).Return(metrics.ScopeIdx(0), nil).Times(1)\n\t// Domain cache is called for tagging\n\ts.mockDomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).AnyTimes()\n\n\t// Use noop logger (no capturing, just to exercise the path)\n\ts.taskProcessor.logger = log.NewNoop()\n\n\t// Make e2e latency > threshold (e.g., 26 minutes)\n\tcreation := time.Now().Add(-26 * time.Minute).UnixNano()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\tDomainID:   testDomainID,\n\t\t\tWorkflowID: testWorkflowID,\n\t\t\tRunID:      testRunID,\n\t\t},\n\t\tCreationTime: common.Int64Ptr(creation),\n\t}\n\n\terr := s.taskProcessor.processTaskOnce(task)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "service/history/replication/task_reader.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\n// TaskReader will read replication tasks from database\ntype TaskReader struct {\n\tshardID          int\n\texecutionManager persistence.ExecutionManager\n}\n\n// NewTaskReader creates new TaskReader\nfunc NewTaskReader(shardID int, executionManager persistence.ExecutionManager) *TaskReader {\n\treturn &TaskReader{\n\t\tshardID:          shardID,\n\t\texecutionManager: executionManager,\n\t}\n}\n\n// Read reads and returns replications tasks from readLevel to maxReadLevel\nfunc (r *TaskReader) Read(ctx context.Context, readLevel int64, maxReadLevel int64, batchSize int) ([]persistence.Task, bool, error) {\n\t// Check if it is even possible to return any results.\n\t// If not return early with empty response. Do not hit persistence.\n\tif readLevel >= maxReadLevel {\n\t\treturn nil, false, nil\n\t}\n\n\tresponse, err := r.executionManager.GetHistoryTasks(ctx, &persistence.GetHistoryTasksRequest{\n\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(readLevel + 1),\n\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(maxReadLevel + 1),\n\t\tPageSize:            batchSize,\n\t})\n\tif err != nil {\n\t\treturn nil, false, err\n\t}\n\n\thasMore := response.NextPageToken != nil\n\treturn response.Tasks, hasMore, nil\n}\n"
  },
  {
    "path": "service/history/replication/task_reader_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nconst (\n\ttestBatchSize = 50\n)\n\nvar (\n\ttestTime = time.Now()\n\n\ttestReplicationTasks = []persistence.Task{\n\t\t&persistence.HistoryReplicationTask{TaskData: persistence.TaskData{TaskID: 50, VisibilityTimestamp: testTime.Add(-1 * time.Second)}},\n\t\t&persistence.HistoryReplicationTask{TaskData: persistence.TaskData{TaskID: 51, VisibilityTimestamp: testTime.Add(-2 * time.Second)}},\n\t}\n)\n\nfunc TestTaskReader(t *testing.T) {\n\ttests := []struct {\n\t\tname              string\n\t\tprepareExecutions func(m *persistence.MockExecutionManager)\n\t\treadLevel         int64\n\t\tmaxReadLevel      int64\n\t\texpectResponse    []persistence.Task\n\t\texpectErr         string\n\t}{\n\t\t{\n\t\t\tname:         \"read replication tasks - first read will use default batch size\",\n\t\t\treadLevel:    50,\n\t\t\tmaxReadLevel: 100,\n\t\t\tprepareExecutions: func(m *persistence.MockExecutionManager) {\n\t\t\t\tm.EXPECT().GetHistoryTasks(gomock.Any(), &persistence.GetHistoryTasksRequest{\n\t\t\t\t\tTaskCategory:        persistence.HistoryTaskCategoryReplication,\n\t\t\t\t\tInclusiveMinTaskKey: persistence.NewImmediateTaskKey(51),\n\t\t\t\t\tExclusiveMaxTaskKey: persistence.NewImmediateTaskKey(101),\n\t\t\t\t\tPageSize:            testBatchSize,\n\t\t\t\t}).Return(&persistence.GetHistoryTasksResponse{Tasks: testReplicationTasks}, nil)\n\t\t\t},\n\t\t\texpectResponse: testReplicationTasks,\n\t\t},\n\t\t{\n\t\t\tname:         \"do not hit persistence when no task will be returned\",\n\t\t\treadLevel:    50,\n\t\t\tmaxReadLevel: 50,\n\t\t\tprepareExecutions: func(m *persistence.MockExecutionManager) {\n\t\t\t\tm.EXPECT().GetHistoryTasks(gomock.Any(), gomock.Any()).Times(0)\n\t\t\t},\n\t\t\texpectResponse: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tem := persistence.NewMockExecutionManager(ctrl)\n\t\t\ttt.prepareExecutions(em)\n\n\t\t\treader := NewTaskReader(testShardID, em)\n\t\t\tresponse, _, err := reader.Read(context.Background(), tt.readLevel, tt.maxReadLevel, testBatchSize)\n\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectResponse, response)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/replication/task_store.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\n// ErrUnknownCluster is returned when given cluster is not defined in cluster metadata\nvar ErrUnknownCluster = errors.New(\"unknown cluster\")\n\n// TaskStore is a component that hydrates and caches replication messages so that they can be reused across several polling source clusters.\n// It also exposes public Put method. This allows pre-store already hydrated messages at the end of successful transaction, saving a DB call to fetch history events.\n//\n// TaskStore uses a separate cache per each source cluster allowing messages to be fetched at different rates.\n// Once a cache becomes full it will not accept further messages for that cluster. Later those messages be fetched from DB and hydrated again.\n// A cache stores only a pointer to the message. It is hydrates once and shared across caches. Cluster acknowledging the message will remove it from that corresponding cache.\n// Once all clusters acknowledge it, no more references will be held, and GC will eventually pick it up.\ntype TaskStore struct {\n\tclusters      map[string]cache.AckCache[*types.ReplicationTask]\n\tdomains       domainCache\n\thydrator      taskHydrator\n\trateLimiter   quotas.Limiter\n\tthrottleRetry *backoff.ThrottleRetry\n\ttimeSource    clock.TimeSource\n\n\tscope  metrics.Scope\n\tlogger log.Logger\n\n\tlastLogTime time.Time\n}\n\ntype (\n\tdomainCache interface {\n\t\tGetDomainByID(id string) (*cache.DomainCacheEntry, error)\n\t}\n\ttaskHydrator interface {\n\t\tHydrate(ctx context.Context, task persistence.Task) (*types.ReplicationTask, error)\n\t}\n)\n\n// NewTaskStore create new instance of TaskStore\nfunc NewTaskStore(\n\tconfig *config.Config,\n\tclusterMetadata cluster.Metadata,\n\tdomains domainCache,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\thydrator taskHydrator,\n\tbudgetManager cache.Manager,\n\tshardID int,\n\ttimeSource clock.TimeSource,\n) *TaskStore {\n\tcacheName := fmt.Sprintf(\"replication-cache-%d\", shardID)\n\n\tclusters := map[string]cache.AckCache[*types.ReplicationTask]{}\n\tfor clusterName := range clusterMetadata.GetRemoteClusterInfo() {\n\t\tclusters[clusterName] = cache.NewBoundedAckCache[*types.ReplicationTask](config.ReplicatorCacheCapacity, config.ReplicatorCacheMaxSize, logger, budgetManager, cacheName)\n\t}\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(100 * time.Millisecond)\n\tretryPolicy.SetMaximumAttempts(config.ReplicatorReadTaskMaxRetryCount())\n\tretryPolicy.SetBackoffCoefficient(1)\n\n\treturn &TaskStore{\n\t\tclusters: clusters,\n\t\tdomains:  domains,\n\t\thydrator: hydrator,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t\t),\n\t\tscope:       metricsClient.Scope(metrics.ReplicatorCacheManagerScope),\n\t\tlogger:      logger.WithTags(tag.ComponentReplicationCacheManager),\n\t\trateLimiter: quotas.NewDynamicRateLimiter(config.ReplicationTaskGenerationQPS.AsFloat64()),\n\t\ttimeSource:  timeSource,\n\t}\n}\n\n// Get will return a hydrated replication message for a given cluster based on raw task info.\n// It will either return it immediately from cache or hydrate it, store in cache and then return.\n//\n// Returned task may be nil. This may be due domain not existing in a given cluster or replication message is not longer relevant.\n// Either case is valid and such replication message should be ignored and not returned in the response.\nfunc (m *TaskStore) Get(ctx context.Context, cluster string, info persistence.Task) (*types.ReplicationTask, error) {\n\tcacheForTargetCluster, ok := m.clusters[cluster]\n\tif !ok {\n\t\treturn nil, ErrUnknownCluster\n\t}\n\n\tdomain, err := m.domains.GetDomainByID(info.GetDomainID())\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"resolving domain: %w\", err)\n\t}\n\n\t// Domain does not exist in this cluster, do not replicate the task\n\tif !domain.HasReplicationCluster(cluster) {\n\t\treturn nil, nil\n\t}\n\n\tscope := m.scope.Tagged(metrics.SourceClusterTag(cluster))\n\n\tscope.IncCounter(metrics.CacheRequests)\n\t// Keep timer (backwards compatible), dual-emit exponential histogram for migration.\n\tcacheLatencyStart := m.timeSource.Now()\n\tsw := scope.StartTimer(metrics.CacheLatency)\n\tdefer sw.Stop()\n\tdefer func() {\n\t\tscope.ExponentialHistogram(metrics.ExponentialCacheLatency, m.timeSource.Since(cacheLatencyStart))\n\t}()\n\n\ttask := cacheForTargetCluster.Get(info.GetTaskID())\n\n\tif task != nil {\n\t\tscope.IncCounter(metrics.CacheHitCounter)\n\t\treturn task, nil\n\t}\n\n\tm.scope.IncCounter(metrics.CacheMissCounter)\n\n\t// Rate limit to not kill the database\n\tm.rateLimiter.Wait(ctx)\n\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\ttask, err = m.hydrator.Hydrate(ctx, info)\n\t\treturn err\n\t}\n\n\terr = m.throttleRetry.Do(ctx, op)\n\n\tif err != nil {\n\t\tm.scope.IncCounter(metrics.CacheFailures)\n\t\treturn nil, err\n\t}\n\n\tm.Put(task)\n\n\treturn task, nil\n}\n\n// Put will try to store hydrated replication to all cluster caches.\n// Tasks may not be relevant, as domain is not enabled in some clusters. Ignore task for that cluster.\n// Some clusters may already have full cache. Ignore the task, it will be fetched and hydrated again later.\n// Some clusters may have already acknowledged such task. Ignore task, it is no longer relevant for such cluster.\nfunc (m *TaskStore) Put(task *types.ReplicationTask) {\n\t// Do not store nil tasks\n\tif task == nil {\n\t\treturn\n\t}\n\n\tdomain, err := m.getDomain(task)\n\tif err != nil {\n\t\tm.logger.Error(\"failed to resolve domain\", tag.Error(err))\n\t\treturn\n\t}\n\n\tfor targetCluster, cacheByCluster := range m.clusters {\n\t\tif domain != nil && !domain.HasReplicationCluster(targetCluster) {\n\t\t\tcontinue\n\t\t}\n\n\t\tscope := m.scope.Tagged(metrics.SourceClusterTag(targetCluster))\n\n\t\terr = cacheByCluster.Put(task, task.ByteSize())\n\t\tswitch {\n\t\tcase errors.Is(err, cache.ErrAckCacheFull):\n\t\t\tscope.IncCounter(metrics.CacheFullCounter)\n\n\t\t\t// This will help debug which shard is full. Logger already has ShardID tag attached.\n\t\t\t// Log only once a minute to not flood the logs.\n\t\t\tif m.timeSource.Since(m.lastLogTime) > time.Minute {\n\t\t\t\tm.logger.Warn(\"Replication cache is full\")\n\t\t\t\tm.lastLogTime = m.timeSource.Now()\n\t\t\t}\n\t\tcase errors.Is(err, cache.ErrAlreadyAcked):\n\t\t\t// No action, this is expected.\n\t\t\t// Some cluster(s) may be already past this, due to different fetch rates.\n\t\t}\n\n\t\tcount := cacheByCluster.Count()\n\t\tscope.RecordTimer(metrics.CacheSize, time.Duration(count))\n\t\tscope.RecordHistogramValue(metrics.CacheSizeHistogram, float64(count))\n\t}\n}\n\n// Ack will acknowledge replication message for a given cluster.\n// This will result in all messages removed from the cache up to a given lastTaskID.\nfunc (m *TaskStore) Ack(cluster string, lastTaskID int64) error {\n\tcache, ok := m.clusters[cluster]\n\tif !ok {\n\t\treturn ErrUnknownCluster\n\t}\n\n\t_, _ = cache.Ack(lastTaskID)\n\n\tscope := m.scope.Tagged(metrics.SourceClusterTag(cluster))\n\tcount := cache.Count()\n\tscope.RecordTimer(metrics.CacheSize, time.Duration(count))\n\tscope.RecordHistogramValue(metrics.CacheSizeHistogram, float64(count))\n\n\treturn nil\n}\n\nfunc (m *TaskStore) getDomain(task *types.ReplicationTask) (*cache.DomainCacheEntry, error) {\n\tif domainID := task.GetHistoryTaskV2Attributes().GetDomainID(); domainID != \"\" {\n\t\treturn m.domains.GetDomainByID(domainID)\n\t}\n\tif domainID := task.GetSyncActivityTaskAttributes().GetDomainID(); domainID != \"\" {\n\t\treturn m.domains.GetDomainByID(domainID)\n\t}\n\tif domainID := task.GetFailoverMarkerAttributes().GetDomainID(); domainID != \"\" {\n\t\treturn m.domains.GetDomainByID(domainID)\n\t}\n\n\treturn nil, nil\n}\n"
  },
  {
    "path": "service/history/replication/task_store_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2022 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replication\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\thconfig \"github.com/uber/cadence/service/history/config\"\n)\n\nfunc TestTaskStore(t *testing.T) {\n\tctx := context.Background()\n\n\tt.Run(\"Get error on unknown cluster\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, nil, nil)\n\t\t_, err := ts.Get(ctx, \"unknown cluster\", &testTask11)\n\t\tassert.Equal(t, ErrUnknownCluster, err)\n\t})\n\n\tt.Run(\"Get error resolving domain\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{}, nil)\n\t\t_, err := ts.Get(ctx, testClusterA, &testTask11)\n\t\tassert.EqualError(t, err, \"resolving domain: domain does not exist\")\n\t})\n\n\tt.Run(\"Get skips task for domains non belonging to polling cluster\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\ttask, err := ts.Get(ctx, testClusterB, &testTask11)\n\t\tassert.NoError(t, err)\n\t\tassert.Nil(t, task)\n\t})\n\n\tt.Run(\"Get returns cached replication task\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\tts.Put(&testHydratedTask11)\n\t\ttask, err := ts.Get(ctx, testClusterA, &testTask11)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, &testHydratedTask11, task)\n\t})\n\n\tt.Run(\"Get returns non-cached replication task by hydrating it\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, fakeTaskHydrator{testTask11.TaskID: testHydratedTask11})\n\t\ttask, err := ts.Get(ctx, testClusterA, &testTask11)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, &testHydratedTask11, task)\n\t})\n\n\tt.Run(\"Get fails to hydrate replication task\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, fakeTaskHydrator{testTask11.TaskID: testHydratedTaskErrorNonRecoverable})\n\t\ttask, err := ts.Get(ctx, testClusterA, &testTask11)\n\t\tassert.EqualError(t, err, \"error hydrating task\")\n\t\tassert.Nil(t, task)\n\t})\n\n\tt.Run(\"Put does not store nil task\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, nil, nil)\n\t\tts.Put(nil)\n\t\tfor _, cache := range ts.clusters {\n\t\t\tassert.Zero(t, cache.Size())\n\t\t}\n\t})\n\n\tt.Run(\"Put error resolving domain - does not store task\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{}, nil)\n\t\tts.Put(&testHydratedTask11)\n\t\tts.Put(&testHydratedTask12)\n\t\tts.Put(&testHydratedTask14)\n\t\tfor _, cache := range ts.clusters {\n\t\t\tassert.Zero(t, cache.Size())\n\t\t}\n\t})\n\n\tt.Run(\"Put hydrated task into appropriate cache\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\tts.Put(&testHydratedTask11)\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\tassert.Equal(t, 1, ts.clusters[cluster.ClusterName].Count())\n\t\t}\n\t})\n\n\tt.Run(\"Put hydrated task without domain info - will put it to all caches\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\tts.Put(&types.ReplicationTask{SourceTaskID: 123})\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\tassert.Equal(t, 1, ts.clusters[cluster.ClusterName].Count())\n\t\t}\n\t})\n\n\tt.Run(\"Put full cache error\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\tts.Put(&testHydratedTask11)\n\t\tts.Put(&testHydratedTask12)\n\t\tts.Put(&testHydratedTask14)\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\tassert.Equal(t, 3, ts.clusters[cluster.ClusterName].Count())\n\t\t}\n\t})\n\n\tt.Run(\"Put will not store acked task\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, fakeDomainCache{testDomainID: testDomain}, nil)\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\tts.Ack(cluster.ClusterName, testHydratedTask11.SourceTaskID)\n\t\t\tts.Put(&testHydratedTask11)\n\t\t\tassert.Equal(t, 0, ts.clusters[cluster.ClusterName].Count())\n\t\t}\n\t})\n\n\tt.Run(\"Ack error on unknown cluster\", func(t *testing.T) {\n\t\tts := createTestTaskStore(t, nil, nil)\n\t\terr := ts.Ack(\"unknown cluster\", 0)\n\t\tassert.Equal(t, ErrUnknownCluster, err)\n\t})\n}\n\nfunc TestTaskStoreWithBudgetManager(t *testing.T) {\n\tctx := context.Background()\n\n\tt.Run(\"Budget manager enforces capacity limits\", func(t *testing.T) {\n\t\tmaxCount := 5\n\n\t\tbudgetManager := cache.NewBudgetManager(\n\t\t\t\"test-replication-cache\",\n\t\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\t\tdynamicproperties.GetIntPropertyFn(maxCount),\n\t\t\tcache.AdmissionOptimistic,\n\t\t\t0,\n\t\t\tmetrics.NewNoopMetricsClient().Scope(metrics.ReplicatorCacheManagerScope),\n\t\t\ttestlogger.New(t),\n\t\t\tdynamicproperties.GetFloatPropertyFn(1.0),\n\t\t)\n\t\tdefer budgetManager.Stop()\n\n\t\tts := createTestTaskStoreWithBudgetManager(t, fakeDomainCache{testDomainID: testDomain}, nil, budgetManager, 1)\n\n\t\tfor i := int64(1); i <= 10; i++ {\n\t\t\ttask := &types.ReplicationTask{\n\t\t\t\tSourceTaskID:            i,\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{DomainID: testDomainID},\n\t\t\t}\n\t\t\tts.Put(task)\n\t\t}\n\n\t\ttotalCached := 0\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\ttotalCached += ts.clusters[cluster.ClusterName].Count()\n\t\t}\n\t\tassert.Equal(t, int64(totalCached), budgetManager.UsedCount())\n\t\tassert.Equal(t, int64(maxCount), budgetManager.UsedCount())\n\t})\n\n\tt.Run(\"Nil budget manager works without errors\", func(t *testing.T) {\n\t\tts := createTestTaskStoreWithBudgetManager(t, fakeDomainCache{testDomainID: testDomain}, nil, nil, 2)\n\n\t\tts.Put(&testHydratedTask11)\n\t\ttask, err := ts.Get(ctx, testClusterA, &testTask11)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, &testHydratedTask11, task)\n\t})\n\n\tt.Run(\"Multiple shards get separate cache IDs\", func(t *testing.T) {\n\t\tbudgetManager := cache.NewBudgetManager(\n\t\t\t\"test-replication-cache\",\n\t\t\tdynamicproperties.GetIntPropertyFn(2000),\n\t\t\tdynamicproperties.GetIntPropertyFn(100),\n\t\t\tcache.AdmissionOptimistic,\n\t\t\t0,\n\t\t\tmetrics.NewNoopMetricsClient().Scope(metrics.ReplicatorCacheManagerScope),\n\t\t\ttestlogger.New(t),\n\t\t\tdynamicproperties.GetFloatPropertyFn(1.0),\n\t\t)\n\t\tdefer budgetManager.Stop()\n\n\t\tts1 := createTestTaskStoreWithBudgetManager(t, fakeDomainCache{testDomainID: testDomain}, nil, budgetManager, 1)\n\t\tts2 := createTestTaskStoreWithBudgetManager(t, fakeDomainCache{testDomainID: testDomain}, nil, budgetManager, 2)\n\n\t\tts1.Put(&testHydratedTask11)\n\t\tts2.Put(&testHydratedTask12)\n\n\t\tfor _, cluster := range testDomain.GetReplicationConfig().Clusters {\n\t\t\tassert.Equal(t, 1, ts1.clusters[cluster.ClusterName].Count())\n\t\t\tassert.Equal(t, 1, ts2.clusters[cluster.ClusterName].Count())\n\t\t}\n\t})\n}\n\nfunc createTestTaskStore(t *testing.T, domains domainCache, hydrator taskHydrator) *TaskStore {\n\treturn createTestTaskStoreWithBudgetManager(t, domains, hydrator, nil, 0)\n}\n\nfunc createTestTaskStoreWithBudgetManager(t *testing.T, domains domainCache, hydrator taskHydrator, budgetManager cache.Manager, shardID int) *TaskStore {\n\tcfg := hconfig.Config{\n\t\tReplicatorCacheCapacity:         dynamicproperties.GetIntPropertyFn(100),\n\t\tReplicationTaskGenerationQPS:    dynamicproperties.GetFloatPropertyFn(0),\n\t\tReplicatorReadTaskMaxRetryCount: dynamicproperties.GetIntPropertyFn(1),\n\t\tReplicatorCacheMaxSize:          dynamicproperties.GetIntPropertyFn(20000),\n\t}\n\n\tclusterMetadata := cluster.NewMetadata(\n\t\tconfig.ClusterGroupMetadata{\n\t\t\tFailoverVersionIncrement: 0,\n\t\t\tPrimaryClusterName:       testClusterC,\n\t\t\tCurrentClusterName:       testClusterC,\n\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\ttestClusterA: {Enabled: true},\n\t\t\t\ttestClusterB: {Enabled: true},\n\t\t\t\ttestClusterC: {Enabled: true},\n\t\t\t},\n\t\t},\n\t\tfunc(d string) bool { return false },\n\t\tmetrics.NewNoopMetricsClient(),\n\t\ttestlogger.New(t),\n\t)\n\n\treturn NewTaskStore(\n\t\t&cfg,\n\t\tclusterMetadata,\n\t\tdomains,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t\thydrator,\n\t\tbudgetManager,\n\t\tshardID,\n\t\tclock.NewRealTimeSource(),\n\t)\n}\n\ntype fakeDomainCache map[string]*cache.DomainCacheEntry\n\nfunc (cache fakeDomainCache) GetDomainByID(id string) (*cache.DomainCacheEntry, error) {\n\tif entry, ok := cache[id]; ok {\n\t\treturn entry, nil\n\t}\n\treturn nil, types.EntityNotExistsError{Message: \"domain does not exist\"}\n}\n"
  },
  {
    "path": "service/history/reset/resetter.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination resetter_mock.go\n\npackage reset\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpersistenceutils \"github.com/uber/cadence/common/persistence/persistence-utils\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\t// WorkflowResetter is the new NDC compatible workflow reset component\n\tWorkflowResetter interface {\n\t\tResetWorkflow(\n\t\t\tctx context.Context,\n\t\t\tdomainID string,\n\t\t\tworkflowID string,\n\t\t\tbaseRunID string,\n\t\t\tbaseBranchToken []byte,\n\t\t\tbaseRebuildLastEventID int64,\n\t\t\tbaseRebuildLastEventVersion int64,\n\t\t\tbaseNextEventID int64,\n\t\t\tresetRunID string,\n\t\t\tresetRequestID string,\n\t\t\tcurrentWorkflow execution.Workflow,\n\t\t\tresetReason string,\n\t\t\tadditionalReapplyEvents []*types.HistoryEvent,\n\t\t\tskipSignalReapply bool,\n\t\t) error\n\t}\n\n\tworkflowResetterImpl struct {\n\t\tshard                shard.Context\n\t\tdomainCache          cache.DomainCache\n\t\tclusterMetadata      cluster.Metadata\n\t\tactiveClusterManager activecluster.Manager\n\t\thistoryV2Mgr         persistence.HistoryManager\n\t\texecutionCache       execution.Cache\n\t\tnewStateRebuilder    nDCStateRebuilderProvider\n\t\tlogger               log.Logger\n\t}\n\n\tnDCStateRebuilderProvider func() execution.StateRebuilder\n)\n\nvar _ WorkflowResetter = (*workflowResetterImpl)(nil)\n\n// NewWorkflowResetter creates a workflow resetter\nfunc NewWorkflowResetter(\n\tshard shard.Context,\n\texecutionCache execution.Cache,\n\tlogger log.Logger,\n) WorkflowResetter {\n\treturn &workflowResetterImpl{\n\t\tshard:                shard,\n\t\tdomainCache:          shard.GetDomainCache(),\n\t\tclusterMetadata:      shard.GetClusterMetadata(),\n\t\tactiveClusterManager: shard.GetActiveClusterManager(),\n\t\thistoryV2Mgr:         shard.GetHistoryManager(),\n\t\texecutionCache:       executionCache,\n\t\tnewStateRebuilder: func() execution.StateRebuilder {\n\t\t\treturn execution.NewStateRebuilder(shard, logger)\n\t\t},\n\t\tlogger: logger,\n\t}\n}\n\nfunc (r *workflowResetterImpl) ResetWorkflow(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\tbaseRunID string,\n\tbaseBranchToken []byte,\n\tbaseRebuildLastEventID int64,\n\tbaseRebuildLastEventVersion int64,\n\tbaseNextEventID int64,\n\tresetRunID string,\n\tresetRequestID string,\n\tcurrentWorkflow execution.Workflow,\n\tresetReason string,\n\tadditionalReapplyEvents []*types.HistoryEvent,\n\tskipSignalReapply bool,\n) (retError error) {\n\tactiveClusterSelectionPolicy := currentWorkflow.GetMutableState().GetExecutionInfo().ActiveClusterSelectionPolicy\n\tactiveClusterInfo, err := r.activeClusterManager.GetActiveClusterInfoByClusterAttribute(ctx, domainID, activeClusterSelectionPolicy.GetClusterAttribute())\n\tif err != nil {\n\t\treturn err\n\t}\n\tresetWorkflowVersion := activeClusterInfo.FailoverVersion\n\n\tcurrentMutableState := currentWorkflow.GetMutableState()\n\tcurrentWorkflowTerminated := false\n\tif currentMutableState.IsWorkflowExecutionRunning() {\n\t\tif err := r.terminateWorkflow(\n\t\t\tcurrentMutableState,\n\t\t\tresetReason,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tresetWorkflowVersion = currentMutableState.GetCurrentVersion()\n\t\tcurrentWorkflowTerminated = true\n\t}\n\n\tresetWorkflow, err := r.prepareResetWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tbaseRunID,\n\t\tbaseBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tbaseNextEventID,\n\t\tresetRunID,\n\t\tresetRequestID,\n\t\tresetWorkflowVersion,\n\t\tresetReason,\n\t\tadditionalReapplyEvents,\n\t\tskipSignalReapply,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer resetWorkflow.GetReleaseFn()(retError)\n\n\treturn r.persistToDB(\n\t\tctx,\n\t\tcurrentWorkflowTerminated,\n\t\tcurrentWorkflow,\n\t\tresetWorkflow,\n\t)\n}\n\nfunc (r *workflowResetterImpl) prepareResetWorkflow(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\tbaseRunID string,\n\tbaseBranchToken []byte,\n\tbaseRebuildLastEventID int64,\n\tbaseRebuildLastEventVersion int64,\n\tbaseNextEventID int64,\n\tresetRunID string,\n\tresetRequestID string,\n\tresetWorkflowVersion int64,\n\tresetReason string,\n\tadditionalReapplyEvents []*types.HistoryEvent,\n\tskipSignalReapply bool,\n) (execution.Workflow, error) {\n\n\tresetWorkflow, err := r.replayResetWorkflow(\n\t\tctx,\n\t\tdomainID,\n\t\tworkflowID,\n\t\tbaseRunID,\n\t\tbaseBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tresetRunID,\n\t\tresetRequestID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresetMutableState := resetWorkflow.GetMutableState()\n\n\tbaseLastEventVersion := resetMutableState.GetCurrentVersion()\n\tif baseLastEventVersion > resetWorkflowVersion {\n\t\treturn nil, &types.InternalServiceError{\n\t\t\tMessage: \"workflowResetter encounter version mismatch.\",\n\t\t}\n\t}\n\tif err := resetMutableState.UpdateCurrentVersion(\n\t\tresetWorkflowVersion,\n\t\tfalse,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tresetMutableState, err = r.closePendingDecisionTask(\n\t\tresetMutableState,\n\t\tbaseRunID,\n\t\tresetRunID,\n\t\tbaseLastEventVersion,\n\t\tresetReason,\n\t\tresetRequestID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := r.failInflightActivity(resetMutableState.GetExecutionInfo().StartTimestamp, resetMutableState, resetReason); err != nil {\n\t\treturn nil, err\n\t}\n\n\t// TODO right now only signals are eligible for reapply, so we can directly skip the whole reapply process\n\t// for the sake of performance. In the future, if there are other events that need to be reapplied, remove this check\n\t// For example, we may want to re-apply activity/timer results for https://github.com/uber/cadence/issues/2934\n\tif !skipSignalReapply {\n\t\tif err := r.reapplyResetAndContinueAsNewWorkflowEvents(\n\t\t\tctx,\n\t\t\tresetMutableState,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tbaseRunID,\n\t\t\tbaseBranchToken,\n\t\t\tbaseRebuildLastEventID+1,\n\t\t\tbaseNextEventID,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t}\n\n\t// NOTE: this is reapplying events that are passing into the API that we shouldn't skip\n\tif err := r.reapplyEvents(resetMutableState, additionalReapplyEvents); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := execution.ScheduleDecision(resetMutableState); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resetWorkflow, nil\n}\n\nfunc (r *workflowResetterImpl) persistToDB(\n\tctx context.Context,\n\tcurrentWorkflowTerminated bool,\n\tcurrentWorkflow execution.Workflow,\n\tresetWorkflow execution.Workflow,\n) error {\n\n\tif currentWorkflowTerminated {\n\t\treturn currentWorkflow.GetContext().UpdateWorkflowExecutionWithNewAsActive(\n\t\t\tctx,\n\t\t\tr.shard.GetTimeSource().Now(),\n\t\t\tresetWorkflow.GetContext(),\n\t\t\tresetWorkflow.GetMutableState(),\n\t\t)\n\t}\n\n\tcurrentMutableState := currentWorkflow.GetMutableState()\n\tcurrentRunID := currentMutableState.GetExecutionInfo().RunID\n\tcurrentLastWriteVersion, err := currentMutableState.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tnow := r.shard.GetTimeSource().Now()\n\tresetWorkflowSnapshot, resetWorkflowEventsSeq, err := resetWorkflow.GetMutableState().CloseTransactionAsSnapshot(\n\t\tnow,\n\t\texecution.TransactionPolicyActive,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif len(resetWorkflowEventsSeq) != 1 {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"there should be EXACTLY one batch of events for reset\",\n\t\t}\n\t}\n\n\t// reset workflow with decision task failed or timed out\n\tresetWorkflowHistory, err := resetWorkflow.GetContext().PersistNonStartWorkflowBatchEvents(ctx, resetWorkflowEventsSeq[0])\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn resetWorkflow.GetContext().CreateWorkflowExecution(\n\t\tctx,\n\t\tresetWorkflowSnapshot,\n\t\tresetWorkflowHistory,\n\t\tpersistence.CreateWorkflowModeContinueAsNew,\n\t\tcurrentRunID,\n\t\tcurrentLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeNew,\n\t)\n}\n\nfunc (r *workflowResetterImpl) replayResetWorkflow(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\tbaseRunID string,\n\tbaseBranchToken []byte,\n\tbaseRebuildLastEventID int64,\n\tbaseRebuildLastEventVersion int64,\n\tresetRunID string,\n\tresetRequestID string,\n) (execution.Workflow, error) {\n\n\tresetContext := execution.NewContext(\n\t\tdomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      resetRunID,\n\t\t},\n\t\tr.shard,\n\t\tr.shard.GetExecutionManager(),\n\t\tr.logger,\n\t)\n\n\tresetBranchTokenFn := func() ([]byte, error) {\n\t\tresetBranchToken, err := r.forkAndGenerateBranchToken(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tbaseBranchToken,\n\t\t\tbaseRebuildLastEventID+1,\n\t\t\tresetRunID,\n\t\t)\n\t\treturn resetBranchToken, err\n\t}\n\n\tresetMutableState, resetHistorySize, err := r.newStateRebuilder().Rebuild(\n\t\tctx,\n\t\tr.shard.GetTimeSource().Now(),\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tbaseRunID,\n\t\t),\n\t\tbaseBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\tdomainID,\n\t\t\tworkflowID,\n\t\t\tresetRunID,\n\t\t),\n\t\tresetBranchTokenFn,\n\t\tresetRequestID,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tresetContext.SetHistorySize(resetHistorySize)\n\treturn execution.NewWorkflow(\n\t\tctx,\n\t\tr.clusterMetadata,\n\t\tresetContext,\n\t\tresetMutableState,\n\t\texecution.NoopReleaseFn,\n\t\tr.logger,\n\t), nil\n}\n\nfunc (r *workflowResetterImpl) failInflightActivity(\n\tnow time.Time,\n\tmutableState execution.MutableState,\n\tterminateReason string,\n) error {\n\n\tfor _, ai := range mutableState.GetPendingActivityInfos() {\n\t\tswitch ai.StartedID {\n\t\tcase constants.EmptyEventID:\n\t\t\t// activity not started, noop\n\t\t\t// override the activity time now\n\t\t\tai.ScheduledTime = now\n\t\t\tai.TimerTaskStatus = execution.TimerTaskStatusNone\n\t\t\tif err := mutableState.UpdateActivity(ai); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\tcase constants.TransientEventID:\n\t\t\t// activity is started (with retry policy)\n\t\t\t// should not encounter this case when rebuilding mutable state\n\t\t\treturn &types.InternalServiceError{\n\t\t\t\tMessage: \"workflowResetter encounter transient activity\",\n\t\t\t}\n\t\tdefault:\n\t\t\tif _, err := mutableState.AddActivityTaskFailedEvent(\n\t\t\t\tai.ScheduleID,\n\t\t\t\tai.StartedID,\n\t\t\t\t&types.RespondActivityTaskFailedRequest{\n\t\t\t\t\tReason:   common.StringPtr(terminateReason),\n\t\t\t\t\tDetails:  ai.Details,\n\t\t\t\t\tIdentity: ai.StartedIdentity,\n\t\t\t\t},\n\t\t\t); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (r *workflowResetterImpl) forkAndGenerateBranchToken(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\tforkBranchToken []byte,\n\tforkNodeID int64,\n\tresetRunID string,\n) ([]byte, error) {\n\t// fork a new history branch\n\tshardID := r.shard.GetShardID()\n\tdomainName, err := r.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp, err := r.historyV2Mgr.ForkHistoryBranch(ctx, &persistence.ForkHistoryBranchRequest{\n\t\tForkBranchToken: forkBranchToken,\n\t\tForkNodeID:      forkNodeID,\n\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(domainID, workflowID, resetRunID),\n\t\tShardID:         common.IntPtr(shardID),\n\t\tDomainName:      domainName,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn resp.NewBranchToken, nil\n}\n\nfunc (r *workflowResetterImpl) terminateWorkflow(\n\tmutableState execution.MutableState,\n\tterminateReason string,\n) error {\n\n\teventBatchFirstEventID := mutableState.GetNextEventID()\n\treturn execution.TerminateWorkflow(\n\t\tmutableState,\n\t\teventBatchFirstEventID,\n\t\tterminateReason,\n\t\tnil,\n\t\texecution.IdentityHistoryService,\n\t)\n}\n\nfunc (r *workflowResetterImpl) reapplyResetAndContinueAsNewWorkflowEvents(\n\tctx context.Context,\n\tresetMutableState execution.MutableState,\n\tdomainID string,\n\tworkflowID string,\n\tbaseRunID string,\n\tbaseBranchToken []byte,\n\tbaseRebuildNextEventID int64,\n\tbaseNextEventID int64,\n) error {\n\n\t// TODO change this logic to fetching all workflow [baseWorkflow, currentWorkflow]\n\t//  from visibility for better coverage of events eligible for re-application.\n\n\tvar nextRunID string\n\tvar err error\n\n\t// first special handling the remaining events for base workflow\n\tif nextRunID, err = r.reapplyWorkflowEvents(\n\t\tctx,\n\t\tresetMutableState,\n\t\tbaseRebuildNextEventID,\n\t\tbaseNextEventID,\n\t\tbaseBranchToken,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tgetNextEventIDBranchToken := func(runID string) (nextEventID int64, branchToken []byte, retError error) {\n\t\tcontext, release, err := r.executionCache.GetOrCreateWorkflowExecution(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\ttypes.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn 0, nil, err\n\t\t}\n\t\tdefer func() { release(retError) }()\n\n\t\tmutableState, err := context.LoadWorkflowExecution(ctx)\n\t\tif err != nil {\n\t\t\t// no matter what error happen, we need to retry\n\t\t\treturn 0, nil, err\n\t\t}\n\n\t\tnextEventID = mutableState.GetNextEventID()\n\t\tbranchToken, err = mutableState.GetCurrentBranchToken()\n\t\tif err != nil {\n\t\t\treturn 0, nil, err\n\t\t}\n\t\treturn nextEventID, branchToken, nil\n\t}\n\n\t// second for remaining continue as new workflow, reapply eligible events\n\tfor len(nextRunID) != 0 {\n\t\tnextWorkflowNextEventID, nextWorkflowBranchToken, err := getNextEventIDBranchToken(nextRunID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif nextRunID, err = r.reapplyWorkflowEvents(\n\t\t\tctx,\n\t\t\tresetMutableState,\n\t\t\tconstants.FirstEventID,\n\t\t\tnextWorkflowNextEventID,\n\t\t\tnextWorkflowBranchToken,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (r *workflowResetterImpl) reapplyWorkflowEvents(\n\tctx context.Context,\n\tmutableState execution.MutableState,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tbranchToken []byte,\n) (string, error) {\n\n\t// TODO change this logic to fetching all workflow [baseWorkflow, currentWorkflow]\n\t//  from visibility for better coverage of events eligible for re-application.\n\t//  after the above change, this API do not have to return the continue as new run ID\n\n\tif firstEventID == nextEventID {\n\t\t// This means the workflow reset to a pending decision task\n\t\t// and the decision task is the latest event in the workflow.\n\t\treturn \"\", nil\n\t}\n\tdomainID := mutableState.GetExecutionInfo().DomainID\n\titer := collection.NewPagingIterator(r.getPaginationFn(\n\t\tctx,\n\t\tfirstEventID,\n\t\tnextEventID,\n\t\tbranchToken,\n\t\tdomainID,\n\t))\n\n\tvar nextRunID string\n\tvar lastEvents []*types.HistoryEvent\n\n\tfor iter.HasNext() {\n\t\tbatch, err := iter.Next()\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\tlastEvents = batch.(*types.History).Events\n\t\tif err := r.reapplyEvents(mutableState, lastEvents); err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t}\n\n\tif len(lastEvents) > 0 {\n\t\tlastEvent := lastEvents[len(lastEvents)-1]\n\t\tif lastEvent.GetEventType() == types.EventTypeWorkflowExecutionContinuedAsNew {\n\t\t\tnextRunID = lastEvent.GetWorkflowExecutionContinuedAsNewEventAttributes().GetNewExecutionRunID()\n\t\t}\n\t}\n\treturn nextRunID, nil\n}\n\nfunc (r *workflowResetterImpl) reapplyEvents(\n\tmutableState execution.MutableState,\n\tevents []*types.HistoryEvent,\n) error {\n\n\tfor _, event := range events {\n\t\tswitch event.GetEventType() {\n\t\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\t\tattr := event.GetWorkflowExecutionSignaledEventAttributes()\n\t\t\tif _, err := mutableState.AddWorkflowExecutionSignaled(\n\t\t\t\tattr.GetSignalName(),\n\t\t\t\tattr.GetInput(),\n\t\t\t\tattr.GetIdentity(),\n\t\t\t\t\"\", // Do not set requestID for requests reapplied, because they have already been applied previously\n\t\t\t); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\tdefault:\n\t\t\t// events other than signal will be ignored\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (r *workflowResetterImpl) getPaginationFn(\n\tctx context.Context,\n\tfirstEventID int64,\n\tnextEventID int64,\n\tbranchToken []byte,\n\tdomainID string,\n) collection.PaginationFn {\n\n\treturn func(paginationToken []byte) ([]interface{}, []byte, error) {\n\n\t\t_, historyBatches, token, _, err := persistenceutils.PaginateHistory(\n\t\t\tctx,\n\t\t\tr.historyV2Mgr,\n\t\t\ttrue,\n\t\t\tbranchToken,\n\t\t\tfirstEventID,\n\t\t\tnextEventID,\n\t\t\tpaginationToken,\n\t\t\texecution.NDCDefaultPageSize,\n\t\t\tcommon.IntPtr(r.shard.GetShardID()),\n\t\t\tdomainID,\n\t\t\tr.domainCache,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\n\t\tvar paginateItems []interface{}\n\t\tfor _, history := range historyBatches {\n\t\t\tpaginateItems = append(paginateItems, history)\n\t\t}\n\t\treturn paginateItems, token, nil\n\t}\n}\n\nfunc (r *workflowResetterImpl) closePendingDecisionTask(\n\tresetMutableState execution.MutableState,\n\tbaseRunID string,\n\tresetRunID string,\n\tbaseLastEventVersion int64,\n\tresetReason string,\n\tresetRequestID string,\n) (execution.MutableState, error) {\n\n\tif len(resetMutableState.GetPendingChildExecutionInfos()) > 0 {\n\t\treturn nil, &types.BadRequestError{\n\t\t\tMessage: \"Can not reset workflow with pending child workflows\",\n\t\t}\n\t}\n\n\tif decision, ok := resetMutableState.GetInFlightDecision(); ok {\n\t\t// reset workflow has decision task start\n\t\t_, err := resetMutableState.AddDecisionTaskFailedEvent(\n\t\t\tdecision.ScheduleID,\n\t\t\tdecision.StartedID,\n\t\t\ttypes.DecisionTaskFailedCauseResetWorkflow,\n\t\t\tnil,\n\t\t\texecution.IdentityHistoryService,\n\t\t\tresetReason,\n\t\t\t\"\",\n\t\t\tbaseRunID,\n\t\t\tresetRunID,\n\t\t\tbaseLastEventVersion,\n\t\t\tresetRequestID,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else if decision, ok := resetMutableState.GetPendingDecision(); ok {\n\t\tif ok {\n\t\t\t// reset workflow has decision task schedule\n\t\t\t_, err := resetMutableState.AddDecisionTaskResetTimeoutEvent(\n\t\t\t\tdecision.ScheduleID,\n\t\t\t\tbaseRunID,\n\t\t\t\tresetRunID,\n\t\t\t\tbaseLastEventVersion,\n\t\t\t\tresetReason,\n\t\t\t\tresetRequestID,\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t}\n\treturn resetMutableState, nil\n}\n"
  },
  {
    "path": "service/history/reset/resetter_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: resetter.go\n//\n// Generated by this command:\n//\n//\tmockgen -package reset -source resetter.go -destination resetter_mock.go\n//\n\n// Package reset is a generated GoMock package.\npackage reset\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n\texecution \"github.com/uber/cadence/service/history/execution\"\n)\n\n// MockWorkflowResetter is a mock of WorkflowResetter interface.\ntype MockWorkflowResetter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWorkflowResetterMockRecorder\n\tisgomock struct{}\n}\n\n// MockWorkflowResetterMockRecorder is the mock recorder for MockWorkflowResetter.\ntype MockWorkflowResetterMockRecorder struct {\n\tmock *MockWorkflowResetter\n}\n\n// NewMockWorkflowResetter creates a new mock instance.\nfunc NewMockWorkflowResetter(ctrl *gomock.Controller) *MockWorkflowResetter {\n\tmock := &MockWorkflowResetter{ctrl: ctrl}\n\tmock.recorder = &MockWorkflowResetterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWorkflowResetter) EXPECT() *MockWorkflowResetterMockRecorder {\n\treturn m.recorder\n}\n\n// ResetWorkflow mocks base method.\nfunc (m *MockWorkflowResetter) ResetWorkflow(ctx context.Context, domainID, workflowID, baseRunID string, baseBranchToken []byte, baseRebuildLastEventID, baseRebuildLastEventVersion, baseNextEventID int64, resetRunID, resetRequestID string, currentWorkflow execution.Workflow, resetReason string, additionalReapplyEvents []*types.HistoryEvent, skipSignalReapply bool) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ResetWorkflow\", ctx, domainID, workflowID, baseRunID, baseBranchToken, baseRebuildLastEventID, baseRebuildLastEventVersion, baseNextEventID, resetRunID, resetRequestID, currentWorkflow, resetReason, additionalReapplyEvents, skipSignalReapply)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ResetWorkflow indicates an expected call of ResetWorkflow.\nfunc (mr *MockWorkflowResetterMockRecorder) ResetWorkflow(ctx, domainID, workflowID, baseRunID, baseBranchToken, baseRebuildLastEventID, baseRebuildLastEventVersion, baseNextEventID, resetRunID, resetRequestID, currentWorkflow, resetReason, additionalReapplyEvents, skipSignalReapply any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ResetWorkflow\", reflect.TypeOf((*MockWorkflowResetter)(nil).ResetWorkflow), ctx, domainID, workflowID, baseRunID, baseBranchToken, baseRebuildLastEventID, baseRebuildLastEventVersion, baseNextEventID, resetRunID, resetRequestID, currentWorkflow, resetReason, additionalReapplyEvents, skipSignalReapply)\n}\n"
  },
  {
    "path": "service/history/reset/resetter_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage reset\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/collection\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tworkflowResetterSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockStateRebuilder *execution.MockStateRebuilder\n\n\t\tmockHistoryV2Mgr *mocks.HistoryV2Manager\n\n\t\tlogger       log.Logger\n\t\tdomainID     string\n\t\tworkflowID   string\n\t\tbaseRunID    string\n\t\tcurrentRunID string\n\t\tresetRunID   string\n\n\t\tworkflowResetter *workflowResetterImpl\n\t}\n)\n\nfunc TestWorkflowResetterSuite(t *testing.T) {\n\ts := new(workflowResetterSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *workflowResetterSuite) SetupSuite() {\n}\n\nfunc (s *workflowResetterSuite) TearDownSuite() {\n}\n\nfunc (s *workflowResetterSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.controller = gomock.NewController(s.T())\n\ts.mockStateRebuilder = execution.NewMockStateRebuilder(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\n\ts.workflowResetter = NewWorkflowResetter(\n\t\ts.mockShard,\n\t\texecution.NewCache(s.mockShard),\n\t\ts.logger,\n\t).(*workflowResetterImpl)\n\ts.workflowResetter.newStateRebuilder = func() execution.StateRebuilder {\n\t\treturn s.mockStateRebuilder\n\t}\n\n\ts.domainID = constants.TestDomainID\n\ts.workflowID = \"some random workflow ID\"\n\ts.baseRunID = uuid.New()\n\ts.currentRunID = uuid.New()\n\ts.resetRunID = uuid.New()\n}\n\nfunc (s *workflowResetterSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *workflowResetterSuite) TestPersistToDB_CurrentTerminated() {\n\tcurrentWorkflowTerminated := true\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentReleaseCalled := false\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\tresetWorkflow := execution.NewMockWorkflow(s.controller)\n\tresetReleaseCalled := false\n\tresetContext := execution.NewMockContext(s.controller)\n\tresetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { resetReleaseCalled = true }\n\tresetWorkflow.EXPECT().GetContext().Return(resetContext).AnyTimes()\n\tresetWorkflow.EXPECT().GetMutableState().Return(resetMutableState).AnyTimes()\n\tresetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tcurrentContext.EXPECT().UpdateWorkflowExecutionWithNewAsActive(\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t\tresetContext,\n\t\tresetMutableState,\n\t).Return(nil).Times(1)\n\n\terr := s.workflowResetter.persistToDB(context.Background(), currentWorkflowTerminated, currentWorkflow, resetWorkflow)\n\ts.NoError(err)\n\t// persistToDB function is not charged of releasing locks\n\ts.False(currentReleaseCalled)\n\ts.False(resetReleaseCalled)\n}\n\nfunc (s *workflowResetterSuite) TestPersistToDB_CurrentNotTerminated() {\n\tcurrentWorkflowTerminated := false\n\tcurrentLastWriteVersion := int64(1234)\n\n\tcurrentWorkflow := execution.NewMockWorkflow(s.controller)\n\tcurrentReleaseCalled := false\n\tcurrentContext := execution.NewMockContext(s.controller)\n\tcurrentMutableState := execution.NewMockMutableState(s.controller)\n\tvar currentReleaseFn execution.ReleaseFunc = func(error) { currentReleaseCalled = true }\n\tcurrentWorkflow.EXPECT().GetContext().Return(currentContext).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetMutableState().Return(currentMutableState).AnyTimes()\n\tcurrentWorkflow.EXPECT().GetReleaseFn().Return(currentReleaseFn).AnyTimes()\n\n\tcurrentMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\tRunID: s.currentRunID,\n\t}).AnyTimes()\n\tcurrentMutableState.EXPECT().GetLastWriteVersion().Return(currentLastWriteVersion, nil).AnyTimes()\n\n\tresetWorkflow := execution.NewMockWorkflow(s.controller)\n\tresetReleaseCalled := false\n\tresetContext := execution.NewMockContext(s.controller)\n\tresetMutableState := execution.NewMockMutableState(s.controller)\n\tvar targetReleaseFn execution.ReleaseFunc = func(error) { resetReleaseCalled = true }\n\tresetWorkflow.EXPECT().GetContext().Return(resetContext).AnyTimes()\n\tresetWorkflow.EXPECT().GetMutableState().Return(resetMutableState).AnyTimes()\n\tresetWorkflow.EXPECT().GetReleaseFn().Return(targetReleaseFn).AnyTimes()\n\n\tresetSnapshot := &persistence.WorkflowSnapshot{}\n\tresetEventsSeq := []*persistence.WorkflowEvents{{\n\t\tDomainID:    s.domainID,\n\t\tWorkflowID:  s.workflowID,\n\t\tRunID:       s.resetRunID,\n\t\tBranchToken: []byte(\"some random reset branch token\"),\n\t\tEvents: []*types.HistoryEvent{{\n\t\t\tID: 123,\n\t\t}},\n\t}}\n\tresetEvents := events.PersistedBlob{DataBlob: persistence.DataBlob{Data: make([]byte, 4321)}}\n\tresetMutableState.EXPECT().CloseTransactionAsSnapshot(\n\t\tgomock.Any(),\n\t\texecution.TransactionPolicyActive,\n\t).Return(resetSnapshot, resetEventsSeq, nil).Times(1)\n\tresetContext.EXPECT().PersistNonStartWorkflowBatchEvents(gomock.Any(), resetEventsSeq[0]).Return(resetEvents, nil).Times(1)\n\tresetContext.EXPECT().CreateWorkflowExecution(\n\t\tgomock.Any(),\n\t\tresetSnapshot,\n\t\tresetEvents,\n\t\tpersistence.CreateWorkflowModeContinueAsNew,\n\t\ts.currentRunID,\n\t\tcurrentLastWriteVersion,\n\t\tpersistence.CreateWorkflowRequestModeNew,\n\t).Return(nil).Times(1)\n\n\terr := s.workflowResetter.persistToDB(context.Background(), currentWorkflowTerminated, currentWorkflow, resetWorkflow)\n\ts.NoError(err)\n\t// persistToDB function is not charged of releasing locks\n\ts.False(currentReleaseCalled)\n\ts.False(resetReleaseCalled)\n}\n\nfunc (s *workflowResetterSuite) TestReplayResetWorkflow() {\n\tctx := context.Background()\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\tbaseRebuildLastEventID := int64(1233)\n\tbaseRebuildLastEventVersion := int64(12)\n\tbaseNodeID := baseRebuildLastEventID + 1\n\n\tresetBranchToken := []byte(\"some random reset branch token\")\n\tresetRequestID := uuid.New()\n\tresetHistorySize := int64(4411)\n\tresetMutableState := execution.NewMockMutableState(s.controller)\n\tdomainName := uuid.New()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ForkHistoryBranch\", mock.Anything, &persistence.ForkHistoryBranchRequest{\n\t\tForkBranchToken: baseBranchToken,\n\t\tForkNodeID:      baseNodeID,\n\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(s.domainID, s.workflowID, s.resetRunID),\n\t\tShardID:         common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:      domainName,\n\t}).Return(&persistence.ForkHistoryBranchResponse{NewBranchToken: resetBranchToken}, nil).Times(1)\n\n\ts.mockStateRebuilder.EXPECT().Rebuild(\n\t\tctx,\n\t\tgomock.Any(),\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\ts.domainID,\n\t\t\ts.workflowID,\n\t\t\ts.baseRunID,\n\t\t),\n\t\tbaseBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tdefinition.NewWorkflowIdentifier(\n\t\t\ts.domainID,\n\t\t\ts.workflowID,\n\t\t\ts.resetRunID,\n\t\t),\n\t\tgomock.Any(),\n\t\tresetRequestID,\n\t).DoAndReturn(func(ctx context.Context, now time.Time, baseWorkflowIdentifier definition.WorkflowIdentifier, baseBranchToken []byte, baseRebuildLastEventID int64, baseRebuildLastEventVersion int64, targetWorkflowIdentifier definition.WorkflowIdentifier, targetBranchFn func() ([]byte, error), requestID string) (execution.MutableState, int64, error) {\n\t\ttargetBranchToken, err := targetBranchFn()\n\t\ts.NoError(err)\n\t\ts.Equal(resetBranchToken, targetBranchToken)\n\n\t\treturn resetMutableState, resetHistorySize, nil\n\t}).Times(1)\n\n\tresetWorkflow, err := s.workflowResetter.replayResetWorkflow(\n\t\tctx,\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.baseRunID,\n\t\tbaseBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\ts.resetRunID,\n\t\tresetRequestID,\n\t)\n\ts.NoError(err)\n\ts.Equal(resetHistorySize, resetWorkflow.GetContext().GetHistorySize())\n\ts.Equal(resetMutableState, resetWorkflow.GetMutableState())\n}\n\nfunc (s *workflowResetterSuite) TestFailInflightActivity() {\n\tnow := time.Now().UTC()\n\tterminateReason := \"some random termination reason\"\n\n\tmutableState := execution.NewMockMutableState(s.controller)\n\n\tactivity1 := &persistence.ActivityInfo{\n\t\tVersion:         12,\n\t\tScheduleID:      123,\n\t\tScheduledTime:   now.Add(-10 * time.Second),\n\t\tStartedID:       124,\n\t\tDetails:         []byte(\"some random activity 1 details\"),\n\t\tStartedIdentity: \"some random activity 1 started identity\",\n\t}\n\tactivity2 := &persistence.ActivityInfo{\n\t\tVersion:         12,\n\t\tScheduleID:      456,\n\t\tScheduledTime:   now.Add(-10 * time.Second),\n\t\tStartedID:       commonconstants.EmptyEventID,\n\t\tTimerTaskStatus: execution.TimerTaskStatusCreatedScheduleToStart,\n\t}\n\tmutableState.EXPECT().GetPendingActivityInfos().Return(map[int64]*persistence.ActivityInfo{\n\t\tactivity1.ScheduleID: activity1,\n\t\tactivity2.ScheduleID: activity2,\n\t}).AnyTimes()\n\n\tmutableState.EXPECT().AddActivityTaskFailedEvent(\n\t\tactivity1.ScheduleID,\n\t\tactivity1.StartedID,\n\t\t&types.RespondActivityTaskFailedRequest{\n\t\t\tReason:   common.StringPtr(terminateReason),\n\t\t\tDetails:  activity1.Details,\n\t\t\tIdentity: activity1.StartedIdentity,\n\t\t},\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\n\tmutableState.EXPECT().UpdateActivity(&persistence.ActivityInfo{\n\t\tVersion:         activity2.Version,\n\t\tScheduleID:      activity2.ScheduleID,\n\t\tScheduledTime:   now,\n\t\tStartedID:       activity2.StartedID,\n\t\tTimerTaskStatus: execution.TimerTaskStatusNone,\n\t}).Return(nil).Times(1)\n\n\terr := s.workflowResetter.failInflightActivity(now, mutableState, terminateReason)\n\ts.NoError(err)\n}\n\nfunc (s *workflowResetterSuite) TestGenerateBranchToken() {\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\tbaseNodeID := int64(1234)\n\n\tresetBranchToken := []byte(\"some random reset branch token\")\n\tdomainName := uuid.New()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ForkHistoryBranch\", mock.Anything, &persistence.ForkHistoryBranchRequest{\n\t\tForkBranchToken: baseBranchToken,\n\t\tForkNodeID:      baseNodeID,\n\t\tInfo:            persistence.BuildHistoryGarbageCleanupInfo(s.domainID, s.workflowID, s.resetRunID),\n\t\tShardID:         common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:      domainName,\n\t}).Return(&persistence.ForkHistoryBranchResponse{NewBranchToken: resetBranchToken}, nil).Times(1)\n\n\tnewBranchToken, err := s.workflowResetter.forkAndGenerateBranchToken(\n\t\tcontext.Background(), s.domainID, s.workflowID, baseBranchToken, baseNodeID, s.resetRunID,\n\t)\n\ts.NoError(err)\n\ts.Equal(resetBranchToken, newBranchToken)\n}\n\nfunc (s *workflowResetterSuite) TestTerminateWorkflow() {\n\tdecision := &execution.DecisionInfo{\n\t\tVersion:    123,\n\t\tScheduleID: 1234,\n\t\tStartedID:  5678,\n\t}\n\tnextEventID := int64(666)\n\tterminateReason := \"some random terminate reason\"\n\n\tmutableState := execution.NewMockMutableState(s.controller)\n\n\tmutableState.EXPECT().GetNextEventID().Return(nextEventID).AnyTimes()\n\tmutableState.EXPECT().GetInFlightDecision().Return(decision, true).Times(1)\n\tmutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\tdecision.ScheduleID,\n\t\tdecision.StartedID,\n\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t([]byte)(nil),\n\t\texecution.IdentityHistoryService,\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\t\"\",\n\t\tint64(0),\n\t\t\"\",\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\tmutableState.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\tmutableState.EXPECT().AddWorkflowExecutionTerminatedEvent(\n\t\tnextEventID,\n\t\tterminateReason,\n\t\t([]byte)(nil),\n\t\texecution.IdentityHistoryService,\n\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\n\terr := s.workflowResetter.terminateWorkflow(mutableState, terminateReason)\n\ts.NoError(err)\n}\n\nfunc (s *workflowResetterSuite) TestReapplyContinueAsNewWorkflowEvents() {\n\tctx := context.Background()\n\tbaseFirstEventID := int64(124)\n\tbaseNextEventID := int64(456)\n\tbaseBranchToken := []byte(\"some random base branch token\")\n\n\tnewRunID := uuid.New()\n\tnewFirstEventID := commonconstants.FirstEventID\n\tnewNextEventID := int64(6)\n\tnewBranchToken := []byte(\"some random new branch token\")\n\n\tdomainName := \"test-domain\"\n\n\tbaseEvent1 := &types.HistoryEvent{\n\t\tID:                                   124,\n\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tbaseEvent2 := &types.HistoryEvent{\n\t\tID:                                 125,\n\t\tEventType:                          types.EventTypeDecisionTaskStarted.Ptr(),\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t}\n\tbaseEvent3 := &types.HistoryEvent{\n\t\tID:                                   126,\n\t\tEventType:                            types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tbaseEvent4 := &types.HistoryEvent{\n\t\tID:        127,\n\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tNewExecutionRunID: newRunID,\n\t\t},\n\t}\n\n\tnewEvent1 := &types.HistoryEvent{\n\t\tID:                                      1,\n\t\tEventType:                               types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t}\n\tnewEvent2 := &types.HistoryEvent{\n\t\tID:                                   2,\n\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tnewEvent3 := &types.HistoryEvent{\n\t\tID:                                 3,\n\t\tEventType:                          types.EventTypeDecisionTaskStarted.Ptr(),\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t}\n\tnewEvent4 := &types.HistoryEvent{\n\t\tID:                                   4,\n\t\tEventType:                            types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tnewEvent5 := &types.HistoryEvent{\n\t\tID:                                     5,\n\t\tEventType:                              types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{},\n\t}\n\n\tbaseEvents := []*types.HistoryEvent{baseEvent1, baseEvent2, baseEvent3, baseEvent4}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   baseBranchToken,\n\t\tMinEventID:    baseFirstEventID,\n\t\tMaxEventID:    baseNextEventID,\n\t\tPageSize:      execution.NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       []*types.History{{Events: baseEvents}},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\n\tnewEvents := []*types.HistoryEvent{newEvent1, newEvent2, newEvent3, newEvent4, newEvent5}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   newBranchToken,\n\t\tMinEventID:    newFirstEventID,\n\t\tMaxEventID:    newNextEventID,\n\t\tPageSize:      execution.NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       []*types.History{{Events: newEvents}},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\tresetMutableState := execution.NewMockMutableState(s.controller)\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\tresetContext := execution.NewMockContext(s.controller)\n\tresetContext.EXPECT().Lock(gomock.Any()).Return(nil).AnyTimes()\n\tresetContext.EXPECT().Unlock().AnyTimes()\n\tresetContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(resetMutableState, nil).AnyTimes()\n\tresetContext.EXPECT().ByteSize().Return(uint64(1)).AnyTimes()\n\tresetMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\tresetMutableState.EXPECT().GetNextEventID().Return(newNextEventID).AnyTimes()\n\tresetMutableState.EXPECT().GetCurrentBranchToken().Return(newBranchToken, nil).AnyTimes()\n\tresetContextCacheKey := definition.NewWorkflowIdentifier(s.domainID, s.workflowID, newRunID)\n\t_, _ = s.workflowResetter.executionCache.PutIfNotExist(resetContextCacheKey, resetContext)\n\n\terr := s.workflowResetter.reapplyResetAndContinueAsNewWorkflowEvents(\n\t\tctx,\n\t\tresetMutableState,\n\t\ts.domainID,\n\t\ts.workflowID,\n\t\ts.baseRunID,\n\t\tbaseBranchToken,\n\t\tbaseFirstEventID,\n\t\tbaseNextEventID,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *workflowResetterSuite) TestReapplyWorkflowEvents() {\n\tfirstEventID := commonconstants.FirstEventID\n\tnextEventID := int64(6)\n\tbranchToken := []byte(\"some random branch token\")\n\tdomainName := \"test-domain\"\n\n\tnewRunID := uuid.New()\n\tevent1 := &types.HistoryEvent{\n\t\tID:                                      1,\n\t\tEventType:                               types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                   2,\n\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tevent3 := &types.HistoryEvent{\n\t\tID:                                 3,\n\t\tEventType:                          types.EventTypeDecisionTaskStarted.Ptr(),\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t}\n\tevent4 := &types.HistoryEvent{\n\t\tID:                                   4,\n\t\tEventType:                            types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tevent5 := &types.HistoryEvent{\n\t\tID:        5,\n\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\tNewExecutionRunID: newRunID,\n\t\t},\n\t}\n\tevents := []*types.HistoryEvent{event1, event2, event3, event4, event5}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      execution.NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       []*types.History{{Events: events}},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\tmutableState := execution.NewMockMutableState(s.controller)\n\tmutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{}).AnyTimes()\n\tnextRunID, err := s.workflowResetter.reapplyWorkflowEvents(\n\t\tcontext.Background(),\n\t\tmutableState,\n\t\tfirstEventID,\n\t\tnextEventID,\n\t\tbranchToken,\n\t)\n\ts.NoError(err)\n\ts.Equal(newRunID, nextRunID)\n}\n\nfunc (s *workflowResetterSuite) TestClosePendingDecisionTask() {\n\tsourceMutableState := execution.NewMockMutableState(s.controller)\n\tbaseRunID := uuid.New()\n\tnewRunID := uuid.New()\n\tbaseForkEventVerison := int64(10)\n\treason := \"test\"\n\tdecisionScheduleEventID := int64(2)\n\tdecisionStartEventID := decisionScheduleEventID + 1\n\tresetRequestID := \"fe4a2833-f761-4cfe-91f2-6cd34c5e987a\"\n\n\t// The workflow has decision schedule and decision start\n\tsourceMutableState.EXPECT().GetInFlightDecision().Return(&execution.DecisionInfo{\n\t\tScheduleID: decisionScheduleEventID,\n\t\tStartedID:  decisionStartEventID,\n\t}, true).Times(1)\n\tsourceMutableState.EXPECT().GetPendingChildExecutionInfos().Return(make(map[int64]*persistence.ChildExecutionInfo)).Times(1)\n\tsourceMutableState.EXPECT().AddDecisionTaskFailedEvent(\n\t\tdecisionScheduleEventID,\n\t\tdecisionStartEventID,\n\t\ttypes.DecisionTaskFailedCauseResetWorkflow,\n\t\tnil,\n\t\texecution.IdentityHistoryService,\n\t\treason,\n\t\t\"\",\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tbaseForkEventVerison,\n\t\tresetRequestID,\n\t).Return(nil, nil).Times(1)\n\n\t_, err := s.workflowResetter.closePendingDecisionTask(\n\t\tsourceMutableState,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tbaseForkEventVerison,\n\t\treason,\n\t\tresetRequestID,\n\t)\n\ts.NoError(err)\n\n\t// The workflow has only decision schedule\n\tsourceMutableState.EXPECT().GetInFlightDecision().Return(nil, false).Times(1)\n\tsourceMutableState.EXPECT().GetPendingDecision().Return(&execution.DecisionInfo{ScheduleID: decisionScheduleEventID}, true).Times(1)\n\tsourceMutableState.EXPECT().GetPendingChildExecutionInfos().Return(make(map[int64]*persistence.ChildExecutionInfo)).Times(1)\n\tsourceMutableState.EXPECT().AddDecisionTaskResetTimeoutEvent(\n\t\tdecisionScheduleEventID,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tbaseForkEventVerison,\n\t\treason,\n\t\tresetRequestID,\n\t).Return(nil, nil).Times(1)\n\t_, err = s.workflowResetter.closePendingDecisionTask(\n\t\tsourceMutableState,\n\t\tbaseRunID,\n\t\tnewRunID,\n\t\tbaseForkEventVerison,\n\t\treason,\n\t\tresetRequestID,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *workflowResetterSuite) TestReapplyEvents() {\n\n\tevent1 := &types.HistoryEvent{\n\t\tID:        101,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tSignalName: \"some random signal name\",\n\t\t\tInput:      []byte(\"some random signal input\"),\n\t\t\tIdentity:   \"some random signal identity\",\n\t\t\tRequestID:  \"a255a38a-1e5b-47a1-a7fc-243566eed78e\",\n\t\t},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                   102,\n\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tevent3 := &types.HistoryEvent{\n\t\tID:        103,\n\t\tEventType: types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{\n\t\t\tSignalName: \"another random signal name\",\n\t\t\tInput:      []byte(\"another random signal input\"),\n\t\t\tIdentity:   \"another random signal identity\",\n\t\t\tRequestID:  \"b4d446a7-c277-4cf7-93b4-0dc304f05346\",\n\t\t},\n\t}\n\tevents := []*types.HistoryEvent{event1, event2, event3}\n\n\tmutableState := execution.NewMockMutableState(s.controller)\n\n\tfor _, event := range events {\n\t\tif event.GetEventType() == types.EventTypeWorkflowExecutionSignaled {\n\t\t\tattr := event.GetWorkflowExecutionSignaledEventAttributes()\n\t\t\tmutableState.EXPECT().AddWorkflowExecutionSignaled(\n\t\t\t\tattr.GetSignalName(),\n\t\t\t\tattr.GetInput(),\n\t\t\t\tattr.GetIdentity(),\n\t\t\t\t\"\",\n\t\t\t).Return(&types.HistoryEvent{}, nil).Times(1)\n\t\t}\n\t}\n\n\terr := s.workflowResetter.reapplyEvents(mutableState, events)\n\ts.NoError(err)\n}\n\nfunc (s *workflowResetterSuite) TestPagination() {\n\tfirstEventID := commonconstants.FirstEventID\n\tnextEventID := int64(101)\n\tbranchToken := []byte(\"some random branch token\")\n\tdomainName := \"some random domain name\"\n\n\tevent1 := &types.HistoryEvent{\n\t\tID:                                      1,\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t}\n\tevent2 := &types.HistoryEvent{\n\t\tID:                                   2,\n\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t}\n\tevent3 := &types.HistoryEvent{\n\t\tID:                                 3,\n\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t}\n\tevent4 := &types.HistoryEvent{\n\t\tID:                                   4,\n\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t}\n\tevent5 := &types.HistoryEvent{\n\t\tID:                                   5,\n\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t}\n\thistory1 := []*types.History{{[]*types.HistoryEvent{event1, event2, event3}}}\n\thistory2 := []*types.History{{[]*types.HistoryEvent{event4, event5}}}\n\thistory := append(history1, history2...)\n\tpageToken := []byte(\"some random token\")\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      execution.NDCDefaultPageSize,\n\t\tNextPageToken: nil,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history1,\n\t\tNextPageToken: pageToken,\n\t\tSize:          12345,\n\t}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranchByBatch\", mock.Anything, &persistence.ReadHistoryBranchRequest{\n\t\tBranchToken:   branchToken,\n\t\tMinEventID:    firstEventID,\n\t\tMaxEventID:    nextEventID,\n\t\tPageSize:      execution.NDCDefaultPageSize,\n\t\tNextPageToken: pageToken,\n\t\tShardID:       common.IntPtr(s.mockShard.GetShardID()),\n\t\tDomainName:    domainName,\n\t}).Return(&persistence.ReadHistoryBranchByBatchResponse{\n\t\tHistory:       history2,\n\t\tNextPageToken: nil,\n\t\tSize:          67890,\n\t}, nil).Once()\n\n\tpaginationFn := s.workflowResetter.getPaginationFn(context.Background(), firstEventID, nextEventID, branchToken, s.domainID)\n\titer := collection.NewPagingIterator(paginationFn)\n\n\tresult := []*types.History{}\n\tfor iter.HasNext() {\n\t\titem, err := iter.Next()\n\t\ts.NoError(err)\n\t\tresult = append(result, item.(*types.History))\n\t}\n\n\ts.Equal(history, result)\n}\n"
  },
  {
    "path": "service/history/resource/resource.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination resource_mock.go -self_package github.com/uber/cadence/service/history/resource\n\npackage resource\n\nimport (\n\t\"fmt\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/quotas/permember\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\n// Resource is the interface which expose common history resources\ntype Resource interface {\n\tresource.Resource\n\tGetEventCache() events.Cache\n\tGetRatelimiterAlgorithm() algorithm.RequestWeighted\n\tGetArchiverClient() archiver.Client\n}\n\ntype resourceImpl struct {\n\tstatus int32\n\n\tresource.Resource\n\teventCache         events.Cache\n\tratelimitAlgorithm algorithm.RequestWeighted\n\tarchiverClient     archiver.Client\n}\n\n// Start starts all resources\nfunc (h *resourceImpl) Start() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&h.status,\n\t\tcommon.DaemonStatusInitialized,\n\t\tcommon.DaemonStatusStarted,\n\t) {\n\t\treturn\n\t}\n\n\th.Resource.Start()\n\th.GetLogger().Info(\"history resource started\", tag.LifeCycleStarted)\n}\n\n// Stop stops all resources\nfunc (h *resourceImpl) Stop() {\n\n\tif !atomic.CompareAndSwapInt32(\n\t\t&h.status,\n\t\tcommon.DaemonStatusStarted,\n\t\tcommon.DaemonStatusStopped,\n\t) {\n\t\treturn\n\t}\n\n\th.Resource.Stop()\n\th.GetLogger().Info(\"history resource stopped\", tag.LifeCycleStopped)\n}\n\n// GetEventCache return event cache\nfunc (h *resourceImpl) GetEventCache() events.Cache {\n\treturn h.eventCache\n}\nfunc (h *resourceImpl) GetRatelimiterAlgorithm() algorithm.RequestWeighted {\n\treturn h.ratelimitAlgorithm\n}\n\nfunc (h *resourceImpl) GetArchiverClient() archiver.Client {\n\treturn h.archiverClient\n}\n\n// New create a new resource containing common history dependencies\nfunc New(\n\tparams *resource.Params,\n\tserviceName string,\n\tconfig *config.Config,\n) (historyResource Resource, retError error) {\n\tserviceResource, err := resource.New(\n\t\tparams,\n\t\tserviceName,\n\t\t&service.Config{\n\t\t\tPersistenceMaxQPS:       config.PersistenceMaxQPS,\n\t\t\tPersistenceGlobalMaxQPS: config.PersistenceGlobalMaxQPS,\n\t\t\tThrottledLoggerMaxRPS:   config.ThrottledLogRPS,\n\n\t\t\tReadVisibilityStoreName:         nil, // history service never read,\n\t\t\tWriteVisibilityStoreName:        config.WriteVisibilityStoreName,\n\t\t\tEnableLogCustomerQueryParameter: nil, // log customer parameter will be done in front-end\n\n\t\t\tEnableDBVisibilitySampling:                  config.EnableVisibilitySampling,\n\t\t\tEnableReadDBVisibilityFromClosedExecutionV2: nil, // history service never read,\n\t\t\tDBVisibilityListMaxQPS:                      nil, // history service never read,\n\t\t\tWriteDBVisibilityOpenMaxQPS:                 config.VisibilityOpenMaxQPS,\n\t\t\tWriteDBVisibilityClosedMaxQPS:               config.VisibilityClosedMaxQPS,\n\n\t\t\tESVisibilityListMaxQPS:   nil,                          // history service never read,\n\t\t\tESIndexMaxResultWindow:   nil,                          // history service never read,\n\t\t\tValidSearchAttributes:    config.ValidSearchAttributes, // history service never read, (Pinot need this to initialize pinotQueryValidator)\n\t\t\tIsErrorRetryableFunction: common.IsServiceTransientError,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\teventCache := events.NewGlobalCache(\n\t\tconfig.EventsCacheGlobalInitialCount(),\n\t\tconfig.EventsCacheGlobalMaxCount(),\n\t\tconfig.EventsCacheTTL(),\n\t\tserviceResource.GetHistoryManager(),\n\t\tparams.Logger,\n\t\tparams.MetricsClient,\n\t\tconfig.EnableSizeBasedHistoryEventCache,\n\t\tconfig.EventsCacheMaxSize,\n\t\tserviceResource.GetDomainCache(),\n\t)\n\tratelimitAlgorithm, err := algorithm.New(\n\t\tparams.MetricsClient,\n\t\tparams.Logger,\n\t\talgorithm.Config{\n\t\t\tNewDataWeight:  config.GlobalRatelimiterNewDataWeight,\n\t\t\tUpdateInterval: config.GlobalRatelimiterUpdateInterval,\n\t\t\tDecayAfter:     config.GlobalRatelimiterDecayAfter,\n\t\t\tGcAfter:        config.GlobalRatelimiterGCAfter,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid ratelimit algorithm config: %w\", err)\n\t}\n\n\tarchivalClient := archiver.NewClient(\n\t\tparams.MetricsClient,\n\t\tparams.Logger,\n\t\tparams.PublicClient,\n\t\tconfig.NumArchiveSystemWorkflows,\n\t\tquotas.NewDynamicRateLimiter(config.ArchiveRequestRPS.AsFloat64()),\n\t\tquotas.NewDynamicRateLimiter(func() float64 {\n\t\t\treturn permember.PerMember(\n\t\t\t\tservice.History,\n\t\t\t\tfloat64(config.ArchiveInlineHistoryGlobalRPS()),\n\t\t\t\tfloat64(config.ArchiveInlineHistoryRPS()),\n\t\t\t\tparams.MembershipResolver,\n\t\t\t)\n\t\t}),\n\t\tquotas.NewDynamicRateLimiter(func() float64 {\n\t\t\treturn permember.PerMember(\n\t\t\t\tservice.History,\n\t\t\t\tfloat64(config.ArchiveInlineVisibilityGlobalRPS()),\n\t\t\t\tfloat64(config.ArchiveInlineVisibilityRPS()),\n\t\t\t\tparams.MembershipResolver,\n\t\t\t)\n\t\t}),\n\t\tparams.ArchiverProvider,\n\t\tconfig.AllowArchivingIncompleteHistory,\n\t)\n\thistoryResource = &resourceImpl{\n\t\tResource:           serviceResource,\n\t\teventCache:         eventCache,\n\t\tratelimitAlgorithm: ratelimitAlgorithm,\n\t\tarchiverClient:     archivalClient,\n\t}\n\treturn\n}\n"
  },
  {
    "path": "service/history/resource/resource_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: resource.go\n//\n// Generated by this command:\n//\n//\tmockgen -package resource -source resource.go -destination resource_mock.go -self_package github.com/uber/cadence/service/history/resource\n//\n\n// Package resource is a generated GoMock package.\npackage resource\n\nimport (\n\treflect \"reflect\"\n\n\ttally \"github.com/uber-go/tally\"\n\tworkflowserviceclient \"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tclient \"github.com/uber/cadence/client\"\n\tadmin \"github.com/uber/cadence/client/admin\"\n\tfrontend \"github.com/uber/cadence/client/frontend\"\n\thistory \"github.com/uber/cadence/client/history\"\n\tmatching \"github.com/uber/cadence/client/matching\"\n\tactivecluster \"github.com/uber/cadence/common/activecluster\"\n\tarchiver \"github.com/uber/cadence/common/archiver\"\n\tprovider \"github.com/uber/cadence/common/archiver/provider\"\n\tqueue \"github.com/uber/cadence/common/asyncworkflow/queue\"\n\tblobstore \"github.com/uber/cadence/common/blobstore\"\n\tcache \"github.com/uber/cadence/common/cache\"\n\tclock \"github.com/uber/cadence/common/clock\"\n\tcluster \"github.com/uber/cadence/common/cluster\"\n\tdomain \"github.com/uber/cadence/common/domain\"\n\tconfigstore \"github.com/uber/cadence/common/dynamicconfig/configstore\"\n\tisolationgroup \"github.com/uber/cadence/common/isolationgroup\"\n\tlog \"github.com/uber/cadence/common/log\"\n\tmembership \"github.com/uber/cadence/common/membership\"\n\tmessaging \"github.com/uber/cadence/common/messaging\"\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\tclient0 \"github.com/uber/cadence/common/persistence/client\"\n\talgorithm \"github.com/uber/cadence/common/quotas/global/algorithm\"\n\trpc \"github.com/uber/cadence/common/quotas/global/rpc\"\n\tevents \"github.com/uber/cadence/service/history/events\"\n\texecutorclient \"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n\tarchiver0 \"github.com/uber/cadence/service/worker/archiver\"\n)\n\n// MockResource is a mock of Resource interface.\ntype MockResource struct {\n\tctrl     *gomock.Controller\n\trecorder *MockResourceMockRecorder\n\tisgomock struct{}\n}\n\n// MockResourceMockRecorder is the mock recorder for MockResource.\ntype MockResourceMockRecorder struct {\n\tmock *MockResource\n}\n\n// NewMockResource creates a new mock instance.\nfunc NewMockResource(ctrl *gomock.Controller) *MockResource {\n\tmock := &MockResource{ctrl: ctrl}\n\tmock.recorder = &MockResourceMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockResource) EXPECT() *MockResourceMockRecorder {\n\treturn m.recorder\n}\n\n// GetActiveClusterManager mocks base method.\nfunc (m *MockResource) GetActiveClusterManager() activecluster.Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterManager\")\n\tret0, _ := ret[0].(activecluster.Manager)\n\treturn ret0\n}\n\n// GetActiveClusterManager indicates an expected call of GetActiveClusterManager.\nfunc (mr *MockResourceMockRecorder) GetActiveClusterManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterManager\", reflect.TypeOf((*MockResource)(nil).GetActiveClusterManager))\n}\n\n// GetArchivalMetadata mocks base method.\nfunc (m *MockResource) GetArchivalMetadata() archiver.ArchivalMetadata {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetArchivalMetadata\")\n\tret0, _ := ret[0].(archiver.ArchivalMetadata)\n\treturn ret0\n}\n\n// GetArchivalMetadata indicates an expected call of GetArchivalMetadata.\nfunc (mr *MockResourceMockRecorder) GetArchivalMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetArchivalMetadata\", reflect.TypeOf((*MockResource)(nil).GetArchivalMetadata))\n}\n\n// GetArchiverClient mocks base method.\nfunc (m *MockResource) GetArchiverClient() archiver0.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetArchiverClient\")\n\tret0, _ := ret[0].(archiver0.Client)\n\treturn ret0\n}\n\n// GetArchiverClient indicates an expected call of GetArchiverClient.\nfunc (mr *MockResourceMockRecorder) GetArchiverClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetArchiverClient\", reflect.TypeOf((*MockResource)(nil).GetArchiverClient))\n}\n\n// GetArchiverProvider mocks base method.\nfunc (m *MockResource) GetArchiverProvider() provider.ArchiverProvider {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetArchiverProvider\")\n\tret0, _ := ret[0].(provider.ArchiverProvider)\n\treturn ret0\n}\n\n// GetArchiverProvider indicates an expected call of GetArchiverProvider.\nfunc (mr *MockResourceMockRecorder) GetArchiverProvider() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetArchiverProvider\", reflect.TypeOf((*MockResource)(nil).GetArchiverProvider))\n}\n\n// GetAsyncWorkflowQueueProvider mocks base method.\nfunc (m *MockResource) GetAsyncWorkflowQueueProvider() queue.Provider {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAsyncWorkflowQueueProvider\")\n\tret0, _ := ret[0].(queue.Provider)\n\treturn ret0\n}\n\n// GetAsyncWorkflowQueueProvider indicates an expected call of GetAsyncWorkflowQueueProvider.\nfunc (mr *MockResourceMockRecorder) GetAsyncWorkflowQueueProvider() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAsyncWorkflowQueueProvider\", reflect.TypeOf((*MockResource)(nil).GetAsyncWorkflowQueueProvider))\n}\n\n// GetBlobstoreClient mocks base method.\nfunc (m *MockResource) GetBlobstoreClient() blobstore.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetBlobstoreClient\")\n\tret0, _ := ret[0].(blobstore.Client)\n\treturn ret0\n}\n\n// GetBlobstoreClient indicates an expected call of GetBlobstoreClient.\nfunc (mr *MockResourceMockRecorder) GetBlobstoreClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetBlobstoreClient\", reflect.TypeOf((*MockResource)(nil).GetBlobstoreClient))\n}\n\n// GetClientBean mocks base method.\nfunc (m *MockResource) GetClientBean() client.Bean {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClientBean\")\n\tret0, _ := ret[0].(client.Bean)\n\treturn ret0\n}\n\n// GetClientBean indicates an expected call of GetClientBean.\nfunc (mr *MockResourceMockRecorder) GetClientBean() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClientBean\", reflect.TypeOf((*MockResource)(nil).GetClientBean))\n}\n\n// GetClusterMetadata mocks base method.\nfunc (m *MockResource) GetClusterMetadata() cluster.Metadata {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClusterMetadata\")\n\tret0, _ := ret[0].(cluster.Metadata)\n\treturn ret0\n}\n\n// GetClusterMetadata indicates an expected call of GetClusterMetadata.\nfunc (mr *MockResourceMockRecorder) GetClusterMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClusterMetadata\", reflect.TypeOf((*MockResource)(nil).GetClusterMetadata))\n}\n\n// GetDispatcher mocks base method.\nfunc (m *MockResource) GetDispatcher() *yarpc.Dispatcher {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDispatcher\")\n\tret0, _ := ret[0].(*yarpc.Dispatcher)\n\treturn ret0\n}\n\n// GetDispatcher indicates an expected call of GetDispatcher.\nfunc (mr *MockResourceMockRecorder) GetDispatcher() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDispatcher\", reflect.TypeOf((*MockResource)(nil).GetDispatcher))\n}\n\n// GetDomainAuditManager mocks base method.\nfunc (m *MockResource) GetDomainAuditManager() persistence.DomainAuditManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainAuditManager\")\n\tret0, _ := ret[0].(persistence.DomainAuditManager)\n\treturn ret0\n}\n\n// GetDomainAuditManager indicates an expected call of GetDomainAuditManager.\nfunc (mr *MockResourceMockRecorder) GetDomainAuditManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainAuditManager\", reflect.TypeOf((*MockResource)(nil).GetDomainAuditManager))\n}\n\n// GetDomainCache mocks base method.\nfunc (m *MockResource) GetDomainCache() cache.DomainCache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainCache\")\n\tret0, _ := ret[0].(cache.DomainCache)\n\treturn ret0\n}\n\n// GetDomainCache indicates an expected call of GetDomainCache.\nfunc (mr *MockResourceMockRecorder) GetDomainCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainCache\", reflect.TypeOf((*MockResource)(nil).GetDomainCache))\n}\n\n// GetDomainManager mocks base method.\nfunc (m *MockResource) GetDomainManager() persistence.DomainManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainManager\")\n\tret0, _ := ret[0].(persistence.DomainManager)\n\treturn ret0\n}\n\n// GetDomainManager indicates an expected call of GetDomainManager.\nfunc (mr *MockResourceMockRecorder) GetDomainManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainManager\", reflect.TypeOf((*MockResource)(nil).GetDomainManager))\n}\n\n// GetDomainMetricsScopeCache mocks base method.\nfunc (m *MockResource) GetDomainMetricsScopeCache() cache.DomainMetricsScopeCache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainMetricsScopeCache\")\n\tret0, _ := ret[0].(cache.DomainMetricsScopeCache)\n\treturn ret0\n}\n\n// GetDomainMetricsScopeCache indicates an expected call of GetDomainMetricsScopeCache.\nfunc (mr *MockResourceMockRecorder) GetDomainMetricsScopeCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainMetricsScopeCache\", reflect.TypeOf((*MockResource)(nil).GetDomainMetricsScopeCache))\n}\n\n// GetDomainReplicationQueue mocks base method.\nfunc (m *MockResource) GetDomainReplicationQueue() domain.ReplicationQueue {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainReplicationQueue\")\n\tret0, _ := ret[0].(domain.ReplicationQueue)\n\treturn ret0\n}\n\n// GetDomainReplicationQueue indicates an expected call of GetDomainReplicationQueue.\nfunc (mr *MockResourceMockRecorder) GetDomainReplicationQueue() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainReplicationQueue\", reflect.TypeOf((*MockResource)(nil).GetDomainReplicationQueue))\n}\n\n// GetEventCache mocks base method.\nfunc (m *MockResource) GetEventCache() events.Cache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEventCache\")\n\tret0, _ := ret[0].(events.Cache)\n\treturn ret0\n}\n\n// GetEventCache indicates an expected call of GetEventCache.\nfunc (mr *MockResourceMockRecorder) GetEventCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEventCache\", reflect.TypeOf((*MockResource)(nil).GetEventCache))\n}\n\n// GetExecutionManager mocks base method.\nfunc (m *MockResource) GetExecutionManager(arg0 int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionManager\", arg0)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetExecutionManager indicates an expected call of GetExecutionManager.\nfunc (mr *MockResourceMockRecorder) GetExecutionManager(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionManager\", reflect.TypeOf((*MockResource)(nil).GetExecutionManager), arg0)\n}\n\n// GetFrontendClient mocks base method.\nfunc (m *MockResource) GetFrontendClient() frontend.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFrontendClient\")\n\tret0, _ := ret[0].(frontend.Client)\n\treturn ret0\n}\n\n// GetFrontendClient indicates an expected call of GetFrontendClient.\nfunc (mr *MockResourceMockRecorder) GetFrontendClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFrontendClient\", reflect.TypeOf((*MockResource)(nil).GetFrontendClient))\n}\n\n// GetFrontendRawClient mocks base method.\nfunc (m *MockResource) GetFrontendRawClient() frontend.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetFrontendRawClient\")\n\tret0, _ := ret[0].(frontend.Client)\n\treturn ret0\n}\n\n// GetFrontendRawClient indicates an expected call of GetFrontendRawClient.\nfunc (mr *MockResourceMockRecorder) GetFrontendRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetFrontendRawClient\", reflect.TypeOf((*MockResource)(nil).GetFrontendRawClient))\n}\n\n// GetHistoryClient mocks base method.\nfunc (m *MockResource) GetHistoryClient() history.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryClient\")\n\tret0, _ := ret[0].(history.Client)\n\treturn ret0\n}\n\n// GetHistoryClient indicates an expected call of GetHistoryClient.\nfunc (mr *MockResourceMockRecorder) GetHistoryClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryClient\", reflect.TypeOf((*MockResource)(nil).GetHistoryClient))\n}\n\n// GetHistoryManager mocks base method.\nfunc (m *MockResource) GetHistoryManager() persistence.HistoryManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryManager\")\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\treturn ret0\n}\n\n// GetHistoryManager indicates an expected call of GetHistoryManager.\nfunc (mr *MockResourceMockRecorder) GetHistoryManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryManager\", reflect.TypeOf((*MockResource)(nil).GetHistoryManager))\n}\n\n// GetHistoryRawClient mocks base method.\nfunc (m *MockResource) GetHistoryRawClient() history.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryRawClient\")\n\tret0, _ := ret[0].(history.Client)\n\treturn ret0\n}\n\n// GetHistoryRawClient indicates an expected call of GetHistoryRawClient.\nfunc (mr *MockResourceMockRecorder) GetHistoryRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryRawClient\", reflect.TypeOf((*MockResource)(nil).GetHistoryRawClient))\n}\n\n// GetHostInfo mocks base method.\nfunc (m *MockResource) GetHostInfo() membership.HostInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHostInfo\")\n\tret0, _ := ret[0].(membership.HostInfo)\n\treturn ret0\n}\n\n// GetHostInfo indicates an expected call of GetHostInfo.\nfunc (mr *MockResourceMockRecorder) GetHostInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHostInfo\", reflect.TypeOf((*MockResource)(nil).GetHostInfo))\n}\n\n// GetHostName mocks base method.\nfunc (m *MockResource) GetHostName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHostName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetHostName indicates an expected call of GetHostName.\nfunc (mr *MockResourceMockRecorder) GetHostName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHostName\", reflect.TypeOf((*MockResource)(nil).GetHostName))\n}\n\n// GetIsolationGroupState mocks base method.\nfunc (m *MockResource) GetIsolationGroupState() isolationgroup.State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIsolationGroupState\")\n\tret0, _ := ret[0].(isolationgroup.State)\n\treturn ret0\n}\n\n// GetIsolationGroupState indicates an expected call of GetIsolationGroupState.\nfunc (mr *MockResourceMockRecorder) GetIsolationGroupState() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIsolationGroupState\", reflect.TypeOf((*MockResource)(nil).GetIsolationGroupState))\n}\n\n// GetIsolationGroupStore mocks base method.\nfunc (m *MockResource) GetIsolationGroupStore() configstore.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetIsolationGroupStore\")\n\tret0, _ := ret[0].(configstore.Client)\n\treturn ret0\n}\n\n// GetIsolationGroupStore indicates an expected call of GetIsolationGroupStore.\nfunc (mr *MockResourceMockRecorder) GetIsolationGroupStore() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetIsolationGroupStore\", reflect.TypeOf((*MockResource)(nil).GetIsolationGroupStore))\n}\n\n// GetLogger mocks base method.\nfunc (m *MockResource) GetLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetLogger indicates an expected call of GetLogger.\nfunc (mr *MockResourceMockRecorder) GetLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLogger\", reflect.TypeOf((*MockResource)(nil).GetLogger))\n}\n\n// GetMatchingClient mocks base method.\nfunc (m *MockResource) GetMatchingClient() matching.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMatchingClient\")\n\tret0, _ := ret[0].(matching.Client)\n\treturn ret0\n}\n\n// GetMatchingClient indicates an expected call of GetMatchingClient.\nfunc (mr *MockResourceMockRecorder) GetMatchingClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMatchingClient\", reflect.TypeOf((*MockResource)(nil).GetMatchingClient))\n}\n\n// GetMatchingRawClient mocks base method.\nfunc (m *MockResource) GetMatchingRawClient() matching.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMatchingRawClient\")\n\tret0, _ := ret[0].(matching.Client)\n\treturn ret0\n}\n\n// GetMatchingRawClient indicates an expected call of GetMatchingRawClient.\nfunc (mr *MockResourceMockRecorder) GetMatchingRawClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMatchingRawClient\", reflect.TypeOf((*MockResource)(nil).GetMatchingRawClient))\n}\n\n// GetMembershipResolver mocks base method.\nfunc (m *MockResource) GetMembershipResolver() membership.Resolver {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMembershipResolver\")\n\tret0, _ := ret[0].(membership.Resolver)\n\treturn ret0\n}\n\n// GetMembershipResolver indicates an expected call of GetMembershipResolver.\nfunc (mr *MockResourceMockRecorder) GetMembershipResolver() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMembershipResolver\", reflect.TypeOf((*MockResource)(nil).GetMembershipResolver))\n}\n\n// GetMessagingClient mocks base method.\nfunc (m *MockResource) GetMessagingClient() messaging.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMessagingClient\")\n\tret0, _ := ret[0].(messaging.Client)\n\treturn ret0\n}\n\n// GetMessagingClient indicates an expected call of GetMessagingClient.\nfunc (mr *MockResourceMockRecorder) GetMessagingClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMessagingClient\", reflect.TypeOf((*MockResource)(nil).GetMessagingClient))\n}\n\n// GetMetricsClient mocks base method.\nfunc (m *MockResource) GetMetricsClient() metrics.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsClient\")\n\tret0, _ := ret[0].(metrics.Client)\n\treturn ret0\n}\n\n// GetMetricsClient indicates an expected call of GetMetricsClient.\nfunc (mr *MockResourceMockRecorder) GetMetricsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsClient\", reflect.TypeOf((*MockResource)(nil).GetMetricsClient))\n}\n\n// GetMetricsScope mocks base method.\nfunc (m *MockResource) GetMetricsScope() tally.Scope {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsScope\")\n\tret0, _ := ret[0].(tally.Scope)\n\treturn ret0\n}\n\n// GetMetricsScope indicates an expected call of GetMetricsScope.\nfunc (mr *MockResourceMockRecorder) GetMetricsScope() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsScope\", reflect.TypeOf((*MockResource)(nil).GetMetricsScope))\n}\n\n// GetPayloadSerializer mocks base method.\nfunc (m *MockResource) GetPayloadSerializer() persistence.PayloadSerializer {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPayloadSerializer\")\n\tret0, _ := ret[0].(persistence.PayloadSerializer)\n\treturn ret0\n}\n\n// GetPayloadSerializer indicates an expected call of GetPayloadSerializer.\nfunc (mr *MockResourceMockRecorder) GetPayloadSerializer() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPayloadSerializer\", reflect.TypeOf((*MockResource)(nil).GetPayloadSerializer))\n}\n\n// GetPersistenceBean mocks base method.\nfunc (m *MockResource) GetPersistenceBean() client0.Bean {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetPersistenceBean\")\n\tret0, _ := ret[0].(client0.Bean)\n\treturn ret0\n}\n\n// GetPersistenceBean indicates an expected call of GetPersistenceBean.\nfunc (mr *MockResourceMockRecorder) GetPersistenceBean() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetPersistenceBean\", reflect.TypeOf((*MockResource)(nil).GetPersistenceBean))\n}\n\n// GetRatelimiterAggregatorsClient mocks base method.\nfunc (m *MockResource) GetRatelimiterAggregatorsClient() rpc.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRatelimiterAggregatorsClient\")\n\tret0, _ := ret[0].(rpc.Client)\n\treturn ret0\n}\n\n// GetRatelimiterAggregatorsClient indicates an expected call of GetRatelimiterAggregatorsClient.\nfunc (mr *MockResourceMockRecorder) GetRatelimiterAggregatorsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRatelimiterAggregatorsClient\", reflect.TypeOf((*MockResource)(nil).GetRatelimiterAggregatorsClient))\n}\n\n// GetRatelimiterAlgorithm mocks base method.\nfunc (m *MockResource) GetRatelimiterAlgorithm() algorithm.RequestWeighted {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRatelimiterAlgorithm\")\n\tret0, _ := ret[0].(algorithm.RequestWeighted)\n\treturn ret0\n}\n\n// GetRatelimiterAlgorithm indicates an expected call of GetRatelimiterAlgorithm.\nfunc (mr *MockResourceMockRecorder) GetRatelimiterAlgorithm() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRatelimiterAlgorithm\", reflect.TypeOf((*MockResource)(nil).GetRatelimiterAlgorithm))\n}\n\n// GetRemoteAdminClient mocks base method.\nfunc (m *MockResource) GetRemoteAdminClient(cluster string) (admin.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteAdminClient\", cluster)\n\tret0, _ := ret[0].(admin.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteAdminClient indicates an expected call of GetRemoteAdminClient.\nfunc (mr *MockResourceMockRecorder) GetRemoteAdminClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteAdminClient\", reflect.TypeOf((*MockResource)(nil).GetRemoteAdminClient), cluster)\n}\n\n// GetRemoteFrontendClient mocks base method.\nfunc (m *MockResource) GetRemoteFrontendClient(cluster string) (frontend.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRemoteFrontendClient\", cluster)\n\tret0, _ := ret[0].(frontend.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetRemoteFrontendClient indicates an expected call of GetRemoteFrontendClient.\nfunc (mr *MockResourceMockRecorder) GetRemoteFrontendClient(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRemoteFrontendClient\", reflect.TypeOf((*MockResource)(nil).GetRemoteFrontendClient), cluster)\n}\n\n// GetSDKClient mocks base method.\nfunc (m *MockResource) GetSDKClient() workflowserviceclient.Interface {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSDKClient\")\n\tret0, _ := ret[0].(workflowserviceclient.Interface)\n\treturn ret0\n}\n\n// GetSDKClient indicates an expected call of GetSDKClient.\nfunc (mr *MockResourceMockRecorder) GetSDKClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSDKClient\", reflect.TypeOf((*MockResource)(nil).GetSDKClient))\n}\n\n// GetServiceName mocks base method.\nfunc (m *MockResource) GetServiceName() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetServiceName\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetServiceName indicates an expected call of GetServiceName.\nfunc (mr *MockResourceMockRecorder) GetServiceName() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetServiceName\", reflect.TypeOf((*MockResource)(nil).GetServiceName))\n}\n\n// GetShardDistributorExecutorClient mocks base method.\nfunc (m *MockResource) GetShardDistributorExecutorClient() executorclient.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardDistributorExecutorClient\")\n\tret0, _ := ret[0].(executorclient.Client)\n\treturn ret0\n}\n\n// GetShardDistributorExecutorClient indicates an expected call of GetShardDistributorExecutorClient.\nfunc (mr *MockResourceMockRecorder) GetShardDistributorExecutorClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardDistributorExecutorClient\", reflect.TypeOf((*MockResource)(nil).GetShardDistributorExecutorClient))\n}\n\n// GetShardManager mocks base method.\nfunc (m *MockResource) GetShardManager() persistence.ShardManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardManager\")\n\tret0, _ := ret[0].(persistence.ShardManager)\n\treturn ret0\n}\n\n// GetShardManager indicates an expected call of GetShardManager.\nfunc (mr *MockResourceMockRecorder) GetShardManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardManager\", reflect.TypeOf((*MockResource)(nil).GetShardManager))\n}\n\n// GetTaskManager mocks base method.\nfunc (m *MockResource) GetTaskManager() persistence.TaskManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskManager\")\n\tret0, _ := ret[0].(persistence.TaskManager)\n\treturn ret0\n}\n\n// GetTaskManager indicates an expected call of GetTaskManager.\nfunc (mr *MockResourceMockRecorder) GetTaskManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskManager\", reflect.TypeOf((*MockResource)(nil).GetTaskManager))\n}\n\n// GetThrottledLogger mocks base method.\nfunc (m *MockResource) GetThrottledLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetThrottledLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetThrottledLogger indicates an expected call of GetThrottledLogger.\nfunc (mr *MockResourceMockRecorder) GetThrottledLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetThrottledLogger\", reflect.TypeOf((*MockResource)(nil).GetThrottledLogger))\n}\n\n// GetTimeSource mocks base method.\nfunc (m *MockResource) GetTimeSource() clock.TimeSource {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTimeSource\")\n\tret0, _ := ret[0].(clock.TimeSource)\n\treturn ret0\n}\n\n// GetTimeSource indicates an expected call of GetTimeSource.\nfunc (mr *MockResourceMockRecorder) GetTimeSource() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTimeSource\", reflect.TypeOf((*MockResource)(nil).GetTimeSource))\n}\n\n// GetVisibilityManager mocks base method.\nfunc (m *MockResource) GetVisibilityManager() persistence.VisibilityManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityManager\")\n\tret0, _ := ret[0].(persistence.VisibilityManager)\n\treturn ret0\n}\n\n// GetVisibilityManager indicates an expected call of GetVisibilityManager.\nfunc (mr *MockResourceMockRecorder) GetVisibilityManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityManager\", reflect.TypeOf((*MockResource)(nil).GetVisibilityManager))\n}\n\n// Start mocks base method.\nfunc (m *MockResource) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockResourceMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockResource)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockResource) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockResourceMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockResource)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/resource/resource_test_utils.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage resource\n\nimport (\n\t\"testing\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas/global/algorithm\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\t// Test is the test implementation used for testing\n\tTest struct {\n\t\t*resource.Test\n\t\tEventCache           *events.MockCache\n\t\tratelimiterAlgorithm algorithm.RequestWeighted\n\t\tarchiverClient       archiver.Client\n\t}\n)\n\nvar _ Resource = (*Test)(nil)\n\n// NewTest returns a new test resource instance\nfunc NewTest(\n\tt *testing.T,\n\tcontroller *gomock.Controller,\n\tserviceMetricsIndex metrics.ServiceIdx,\n) *Test {\n\treturn &Test{\n\t\tTest:           resource.NewTest(t, controller, serviceMetricsIndex),\n\t\tEventCache:     events.NewMockCache(controller),\n\t\tarchiverClient: archiver.NewMockClient(controller),\n\t}\n}\n\n// GetEventCache for testing\nfunc (s *Test) GetEventCache() events.Cache {\n\treturn s.EventCache\n}\n\nfunc (s *Test) GetRatelimiterAlgorithm() algorithm.RequestWeighted {\n\treturn s.ratelimiterAlgorithm\n}\n\nfunc (s *Test) GetArchiverClient() archiver.Client {\n\treturn s.archiverClient\n}\n"
  },
  {
    "path": "service/history/service.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\tcommonResource \"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/handler\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/history/wrappers/grpc\"\n\t\"github.com/uber/cadence/service/history/wrappers/ratelimited\"\n\t\"github.com/uber/cadence/service/history/wrappers/thrift\"\n)\n\nconst (\n\tworkflowIDCacheTTL      = 1 * time.Second\n\tworkflowIDCacheMaxCount = 10_000\n)\n\n// Service represents the cadence-history service\ntype Service struct {\n\tresource.Resource\n\n\tstatus  int32\n\thandler handler.Handler\n\tstopC   chan struct{}\n\tparams  *commonResource.Params\n\tconfig  *config.Config\n}\n\n// NewService builds a new cadence-history service\nfunc NewService(\n\tparams *commonResource.Params,\n) (resource.Resource, error) {\n\tserviceConfig := config.New(\n\t\tdynamicconfig.NewCollection(\n\t\t\tparams.DynamicConfig,\n\t\t\tparams.Logger,\n\t\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t\t),\n\t\tparams.PersistenceConfig.NumHistoryShards,\n\t\tparams.RPCFactory.GetMaxMessageSize(),\n\t\tparams.PersistenceConfig.IsAdvancedVisibilityConfigExist(),\n\t\tparams.HostName)\n\n\tserviceResource, err := resource.New(\n\t\tparams,\n\t\tservice.History,\n\t\tserviceConfig,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Service{\n\t\tResource: serviceResource,\n\t\tstatus:   common.DaemonStatusInitialized,\n\t\tstopC:    make(chan struct{}),\n\t\tparams:   params,\n\t\tconfig:   serviceConfig,\n\t}, nil\n}\n\n// Start starts the service\nfunc (s *Service) Start() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tlogger := s.GetLogger()\n\tlogger.Info(\"elastic search config\", tag.ESConfig(s.params.ESConfig))\n\tlogger.Info(\"history starting\")\n\n\twfIDCache := workflowcache.New(workflowcache.Params{\n\t\tTTL:                    workflowIDCacheTTL,\n\t\tExternalLimiterFactory: quotas.NewSimpleDynamicRateLimiterFactory(s.config.WorkflowIDExternalRPS),\n\t\tInternalLimiterFactory: quotas.NewSimpleDynamicRateLimiterFactory(s.config.WorkflowIDInternalRPS),\n\t\tMaxCount:               workflowIDCacheMaxCount,\n\t\tDomainCache:            s.Resource.GetDomainCache(),\n\t\tLogger:                 s.Resource.GetLogger(),\n\t\tMetricsClient:          s.Resource.GetMetricsClient(),\n\t})\n\n\trawHandler := handler.NewHandler(s.Resource, s.config, wfIDCache)\n\ts.handler = ratelimited.NewHistoryHandler(\n\t\trawHandler,\n\t\twfIDCache,\n\t\ts.Resource.GetLogger(),\n\t)\n\n\tthriftHandler := thrift.NewThriftHandler(s.handler)\n\tthriftHandler.Register(s.GetDispatcher())\n\n\tgrpcHandler := grpc.NewGRPCHandler(s.handler)\n\tgrpcHandler.Register(s.GetDispatcher())\n\n\t// must start resource first\n\ts.Resource.Start()\n\ts.handler.Start()\n\n\tlogger.Info(\"history started\")\n\n\t<-s.stopC\n}\n\n// Stop stops the service\nfunc (s *Service) Stop() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\t// initiate graceful shutdown :\n\t// 1. remove self from the membership ring\n\t// 2. wait for other members to discover we are going down\n\t// 3. stop acquiring new shards (periodically or based on other membership changes)\n\t// 4. wait for shard ownership to transfer (and inflight requests to drain) while still accepting new requests\n\t// 5. Reject all requests arriving at rpc handler to avoid taking on more work except for RespondXXXCompleted and\n\t//    RecordXXStarted APIs - for these APIs, most of the work is already one and rejecting at last stage is\n\t//    probably not that desirable. If the shard is closed, these requests will fail anyways.\n\t// 6. wait for grace period\n\t// 7. force stop the whole world and return\n\n\tconst gossipPropagationDelay = 400 * time.Millisecond\n\tconst gracePeriod = 2 * time.Second\n\n\tremainingTime := s.config.ShutdownDrainDuration()\n\n\ts.GetLogger().Info(\"ShutdownHandler: Evicting self from membership ring\")\n\ts.GetMembershipResolver().EvictSelf()\n\n\ts.GetLogger().Info(\"ShutdownHandler: Waiting for others to discover I am unhealthy\")\n\tremainingTime = common.SleepWithMinDuration(gossipPropagationDelay, remainingTime)\n\n\tremainingTime = s.handler.PrepareToStop(remainingTime)\n\t_ = common.SleepWithMinDuration(gracePeriod, remainingTime)\n\n\tclose(s.stopC)\n\n\ts.handler.Stop()\n\ts.Resource.Stop()\n\n\ts.GetLogger().Info(\"history stopped\")\n}\n"
  },
  {
    "path": "service/history/shard/context.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination context_mock.go -package shard github.com/uber/cadence/history/shard/context Context\n\npackage shard\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/history/simulation\"\n)\n\ntype (\n\t// Context represents a history engine shard\n\tContext interface {\n\t\tGetShardID() int\n\t\tGetService() resource.Resource\n\t\tGetExecutionManager() persistence.ExecutionManager\n\t\tGetHistoryManager() persistence.HistoryManager\n\t\tGetDomainCache() cache.DomainCache\n\t\tGetActiveClusterManager() activecluster.Manager\n\t\tGetClusterMetadata() cluster.Metadata\n\t\tGetConfig() *config.Config\n\t\tGetEventsCache() events.Cache\n\t\tGetLogger() log.Logger\n\t\tGetThrottledLogger() log.Logger\n\t\tGetMetricsClient() metrics.Client\n\t\tGetTimeSource() clock.TimeSource\n\t\tPreviousShardOwnerWasDifferent() bool\n\t\tGetReplicationBudgetManager() cache.Manager\n\n\t\tGetEngine() engine.Engine\n\t\tSetEngine(engine.Engine)\n\n\t\tGenerateTaskID() (int64, error)\n\t\tGenerateTaskIDs(number int) ([]int64, error)\n\n\t\tUpdateIfNeededAndGetQueueMaxReadLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey\n\n\t\tSetCurrentTime(cluster string, currentTime time.Time)\n\t\tGetCurrentTime(cluster string) time.Time\n\t\tGetLastUpdatedTime() time.Time\n\n\t\tGetQueueAckLevel(category persistence.HistoryTaskCategory) persistence.HistoryTaskKey\n\t\tUpdateQueueAckLevel(category persistence.HistoryTaskCategory, ackLevel persistence.HistoryTaskKey) error\n\t\tGetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey\n\t\tUpdateQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string, ackLevel persistence.HistoryTaskKey) error\n\t\tGetQueueState(category persistence.HistoryTaskCategory) (*types.QueueState, error)\n\t\tUpdateQueueState(category persistence.HistoryTaskCategory, state *types.QueueState) error\n\n\t\tGetTransferProcessingQueueStates(cluster string) []*types.ProcessingQueueState\n\t\tUpdateTransferProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error\n\n\t\tGetTimerProcessingQueueStates(cluster string) []*types.ProcessingQueueState\n\t\tUpdateTimerProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error\n\n\t\tUpdateFailoverLevel(category persistence.HistoryTaskCategory, failoverID string, level persistence.FailoverLevel) error\n\t\tDeleteFailoverLevel(category persistence.HistoryTaskCategory, failoverID string) error\n\t\tGetAllFailoverLevels(category persistence.HistoryTaskCategory) map[string]persistence.FailoverLevel\n\n\t\tGetDomainNotificationVersion() int64\n\t\tUpdateDomainNotificationVersion(domainNotificationVersion int64) error\n\n\t\tGetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error)\n\t\tCreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error)\n\t\tUpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error)\n\t\tConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (*persistence.ConflictResolveWorkflowExecutionResponse, error)\n\t\tAppendHistoryV2Events(ctx context.Context, request *persistence.AppendHistoryNodesRequest, domainID string, execution types.WorkflowExecution) (*persistence.AppendHistoryNodesResponse, error)\n\n\t\tReplicateFailoverMarkers(ctx context.Context, markers []*persistence.FailoverMarkerTask) error\n\t\tAddingPendingFailoverMarker(*types.FailoverMarkerAttributes) error\n\t\tValidateAndUpdateFailoverMarkers() ([]*types.FailoverMarkerAttributes, error)\n\t}\n\n\tcontextImpl struct {\n\t\tresource.Resource\n\n\t\tshardItem                *historyShardsItem\n\t\tshardID                  int\n\t\trangeID                  int64\n\t\texecutionManager         persistence.ExecutionManager\n\t\tactiveClusterManager     activecluster.Manager\n\t\teventsCache              events.Cache\n\t\tcloseCallback            func(int, *historyShardsItem)\n\t\tclosedAt                 atomic.Pointer[time.Time]\n\t\tconfig                   *config.Config\n\t\tpersistenceConfig        *persistence.DynamicConfiguration\n\t\tlogger                   log.Logger\n\t\tthrottledLogger          log.Logger\n\t\tengine                   engine.Engine\n\t\treplicationBudgetManager cache.Manager\n\n\t\tsync.RWMutex\n\t\tlastUpdated                  time.Time\n\t\tshardInfo                    *persistence.ShardInfo\n\t\ttaskSequenceNumber           int64\n\t\tmaxTaskSequenceNumber        int64\n\t\timmediateTaskMaxReadLevel    int64\n\t\tscheduledTaskMaxReadLevelMap map[string]time.Time                                                     // cluster -> timerMaxReadLevel\n\t\tfailoverLevels               map[persistence.HistoryTaskCategory]map[string]persistence.FailoverLevel // category -> uuid -> FailoverLevel\n\n\t\t// exist only in memory\n\t\tremoteClusterCurrentTime map[string]time.Time\n\n\t\t// true if previous owner was different from the acquirer's identity.\n\t\tpreviousShardOwnerWasDifferent bool\n\t}\n)\n\nvar _ Context = (*contextImpl)(nil)\n\ntype ErrShardClosed struct {\n\tMsg      string\n\tClosedAt time.Time\n}\n\nvar _ error = (*ErrShardClosed)(nil)\n\nfunc (e *ErrShardClosed) Error() string {\n\treturn e.Msg\n}\n\nconst (\n\tTimeBeforeShardClosedIsError = 10 * time.Second\n)\n\nconst (\n\t// transfer/cross cluster diff/lag is in terms of taskID, which is calculated based on shard rangeID\n\t// on shard movement, taskID will increase by around 1 million\n\tlogWarnTransferLevelDiff    = 3000000 // 3 million\n\tlogWarnCrossClusterLevelLag = 3000000 // 3 million\n\tlogWarnTimerLevelDiff       = time.Duration(30 * time.Minute)\n\thistorySizeLogThreshold     = 10 * 1024 * 1024\n\tminContextTimeout           = 1 * time.Second\n\tactiveClusterLookupTimeout  = 1 * time.Second\n)\n\nfunc (s *contextImpl) GetShardID() int {\n\treturn s.shardID\n}\n\nfunc (s *contextImpl) GetService() resource.Resource {\n\treturn s.Resource\n}\n\nfunc (s *contextImpl) GetExecutionManager() persistence.ExecutionManager {\n\treturn s.executionManager\n}\n\nfunc (s *contextImpl) GetEngine() engine.Engine {\n\treturn s.engine\n}\n\nfunc (s *contextImpl) SetEngine(engine engine.Engine) {\n\ts.engine = engine\n}\n\nfunc (s *contextImpl) GetActiveClusterManager() activecluster.Manager {\n\treturn s.activeClusterManager\n}\n\nfunc (s *contextImpl) GenerateTaskID() (int64, error) {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\treturn s.generateTaskIDLocked()\n}\n\nfunc (s *contextImpl) GenerateTaskIDs(number int) ([]int64, error) {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tresult := []int64{}\n\tfor i := 0; i < number; i++ {\n\t\tid, err := s.generateTaskIDLocked()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult = append(result, id)\n\t}\n\treturn result, nil\n}\n\nfunc (s *contextImpl) UpdateIfNeededAndGetQueueMaxReadLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\tswitch category.Type() {\n\tcase persistence.HistoryTaskCategoryTypeImmediate:\n\t\treturn s.getImmediateTaskMaxReadLevel()\n\tcase persistence.HistoryTaskCategoryTypeScheduled:\n\t\treturn s.updateScheduledTaskMaxReadLevel(cluster)\n\tdefault:\n\t\ts.logger.Fatal(\"unknown history task category\", tag.Dynamic(\"category-type\", category.Type()))\n\t}\n\treturn persistence.HistoryTaskKey{}\n}\n\nfunc (s *contextImpl) getImmediateTaskMaxReadLevel() persistence.HistoryTaskKey {\n\ts.RLock()\n\tdefer s.RUnlock()\n\treturn persistence.NewImmediateTaskKey(s.immediateTaskMaxReadLevel)\n}\n\nfunc (s *contextImpl) updateScheduledTaskMaxReadLevel(cluster string) persistence.HistoryTaskKey {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tcurrentTime := s.GetTimeSource().Now()\n\tif cluster != \"\" && cluster != s.GetClusterMetadata().GetCurrentClusterName() {\n\t\tcurrentTime = s.remoteClusterCurrentTime[cluster]\n\t}\n\n\tnewMaxReadLevel := currentTime.Add(s.config.TimerProcessorMaxTimeShift()).Truncate(persistence.DBTimestampMinPrecision)\n\tif newMaxReadLevel.After(s.scheduledTaskMaxReadLevelMap[cluster]) {\n\t\ts.scheduledTaskMaxReadLevelMap[cluster] = newMaxReadLevel\n\t}\n\treturn persistence.NewHistoryTaskKey(s.scheduledTaskMaxReadLevelMap[cluster], 0)\n}\n\nfunc (s *contextImpl) GetQueueAckLevel(category persistence.HistoryTaskCategory) persistence.HistoryTaskKey {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.getQueueAckLevelLocked(category)\n}\n\nfunc (s *contextImpl) getQueueAckLevelLocked(category persistence.HistoryTaskCategory) persistence.HistoryTaskKey {\n\tswitch category {\n\tcase persistence.HistoryTaskCategoryTransfer:\n\t\treturn persistence.NewImmediateTaskKey(s.shardInfo.TransferAckLevel)\n\tcase persistence.HistoryTaskCategoryTimer:\n\t\treturn persistence.NewHistoryTaskKey(s.shardInfo.TimerAckLevel, 0)\n\tcase persistence.HistoryTaskCategoryReplication:\n\t\treturn persistence.NewImmediateTaskKey(s.shardInfo.ReplicationAckLevel)\n\tdefault:\n\t\treturn persistence.HistoryTaskKey{}\n\t}\n}\n\nfunc (s *contextImpl) UpdateQueueAckLevel(category persistence.HistoryTaskCategory, ackLevel persistence.HistoryTaskKey) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tswitch category {\n\tcase persistence.HistoryTaskCategoryTransfer:\n\t\ts.shardInfo.TransferAckLevel = ackLevel.GetTaskID()\n\t\t// for forward compatibility\n\t\ts.shardInfo.QueueStates[int32(persistence.HistoryTaskCategoryIDTransfer)] = &types.QueueState{\n\t\t\tExclusiveMaxReadLevel: &types.TaskKey{TaskID: ackLevel.GetTaskID() + 1},\n\t\t}\n\tcase persistence.HistoryTaskCategoryTimer:\n\t\ts.shardInfo.TimerAckLevel = ackLevel.GetScheduledTime()\n\t\t// for forward compatibility\n\t\ts.shardInfo.QueueStates[int32(persistence.HistoryTaskCategoryIDTimer)] = &types.QueueState{\n\t\t\tExclusiveMaxReadLevel: &types.TaskKey{ScheduledTimeNano: ackLevel.GetScheduledTime().UnixNano()},\n\t\t}\n\tcase persistence.HistoryTaskCategoryReplication:\n\t\ts.shardInfo.ReplicationAckLevel = ackLevel.GetTaskID()\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown history task category: %v\", category)\n\t}\n\ts.shardInfo.StolenSinceRenew = 0\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) GetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\tswitch category {\n\tcase persistence.HistoryTaskCategoryTransfer:\n\t\t// if we can find corresponding ack level\n\t\tif ackLevel, ok := s.shardInfo.ClusterTransferAckLevel[cluster]; ok {\n\t\t\treturn persistence.NewImmediateTaskKey(ackLevel)\n\t\t}\n\t\t// otherwise, default to existing ack level, which belongs to local cluster\n\t\t// this can happen if you add more cluster\n\t\treturn persistence.NewImmediateTaskKey(s.shardInfo.TransferAckLevel)\n\tcase persistence.HistoryTaskCategoryTimer:\n\t\t// if we can find corresponding ack level\n\t\tif ackLevel, ok := s.shardInfo.ClusterTimerAckLevel[cluster]; ok {\n\t\t\treturn persistence.NewHistoryTaskKey(ackLevel, 0)\n\t\t}\n\t\t// otherwise, default to existing ack level, which belongs to local cluster\n\t\t// this can happen if you add more cluster\n\t\treturn persistence.NewHistoryTaskKey(s.shardInfo.TimerAckLevel, 0)\n\tcase persistence.HistoryTaskCategoryReplication:\n\t\t// if we can find corresponding replication level\n\t\tif replicationLevel, ok := s.shardInfo.ClusterReplicationLevel[cluster]; ok {\n\t\t\treturn persistence.NewImmediateTaskKey(replicationLevel)\n\t\t}\n\t\t// New cluster always starts from -1\n\t\treturn persistence.NewImmediateTaskKey(-1)\n\tdefault:\n\t\treturn persistence.HistoryTaskKey{}\n\t}\n}\n\nfunc (s *contextImpl) UpdateQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string, ackLevel persistence.HistoryTaskKey) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tswitch category {\n\tcase persistence.HistoryTaskCategoryTransfer:\n\t\ts.shardInfo.ClusterTransferAckLevel[cluster] = ackLevel.GetTaskID()\n\tcase persistence.HistoryTaskCategoryTimer:\n\t\ts.shardInfo.ClusterTimerAckLevel[cluster] = ackLevel.GetScheduledTime()\n\tcase persistence.HistoryTaskCategoryReplication:\n\t\ts.shardInfo.ClusterReplicationLevel[cluster] = ackLevel.GetTaskID()\n\tdefault:\n\t\treturn fmt.Errorf(\"unknown history task category: %v\", category)\n\t}\n\ts.shardInfo.StolenSinceRenew = 0\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) GetQueueState(category persistence.HistoryTaskCategory) (*types.QueueState, error) {\n\ts.RLock()\n\tdefer s.RUnlock()\n\tqueueState, ok := s.shardInfo.QueueStates[int32(category.ID())]\n\tif !ok {\n\t\tswitch category {\n\t\tcase persistence.HistoryTaskCategoryTransfer:\n\t\t\tqueueState = &types.QueueState{\n\t\t\t\tExclusiveMaxReadLevel: &types.TaskKey{TaskID: s.shardInfo.TransferAckLevel + 1},\n\t\t\t}\n\t\tcase persistence.HistoryTaskCategoryTimer:\n\t\t\tqueueState = &types.QueueState{\n\t\t\t\tExclusiveMaxReadLevel: &types.TaskKey{ScheduledTimeNano: s.shardInfo.TimerAckLevel.UnixNano()},\n\t\t\t}\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"unknown history task category: %v\", category)\n\t\t}\n\t}\n\treturn queueState, nil\n}\n\nfunc (s *contextImpl) UpdateQueueState(category persistence.HistoryTaskCategory, state *types.QueueState) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.shardInfo.QueueStates[int32(category.ID())] = state\n\n\t// for backward compatibility\n\t// we must make sure that there is no task being missed when converting the queue state\n\t// it's ok to have tasks being processed multiple times\n\t// for immediate tasks the exclusive level should be converted back to inclusive level\n\tswitch category {\n\tcase persistence.HistoryTaskCategoryTransfer:\n\t\tackLevel := state.ExclusiveMaxReadLevel.TaskID\n\t\tfor _, virtualQueueState := range state.VirtualQueueStates {\n\t\t\tfor _, virtualSliceState := range virtualQueueState.VirtualSliceStates {\n\t\t\t\tackLevel = min(ackLevel, virtualSliceState.TaskRange.InclusiveMin.TaskID)\n\t\t\t}\n\t\t}\n\t\t// exclusive to inclusive\n\t\tackLevel = ackLevel - 1\n\t\ts.shardInfo.TransferAckLevel = ackLevel\n\t\tfor clusterName := range s.GetClusterMetadata().GetEnabledClusterInfo() {\n\t\t\ts.shardInfo.ClusterTransferAckLevel[clusterName] = ackLevel\n\t\t\tif s.shardInfo.TransferProcessingQueueStates.StatesByCluster == nil {\n\t\t\t\ts.shardInfo.TransferProcessingQueueStates.StatesByCluster = make(map[string][]*types.ProcessingQueueState)\n\t\t\t}\n\t\t\ts.shardInfo.TransferProcessingQueueStates.StatesByCluster[clusterName] = []*types.ProcessingQueueState{\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\t\t\tAckLevel: common.Int64Ptr(ackLevel),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\tcase persistence.HistoryTaskCategoryTimer:\n\t\tackLevel := state.ExclusiveMaxReadLevel.ScheduledTimeNano\n\t\tfor _, virtualQueueState := range state.VirtualQueueStates {\n\t\t\tfor _, virtualSliceState := range virtualQueueState.VirtualSliceStates {\n\t\t\t\tackLevel = min(ackLevel, virtualSliceState.TaskRange.InclusiveMin.ScheduledTimeNano)\n\t\t\t}\n\t\t}\n\t\ts.shardInfo.TimerAckLevel = time.Unix(0, ackLevel)\n\t\tfor clusterName := range s.GetClusterMetadata().GetEnabledClusterInfo() {\n\t\t\ts.shardInfo.ClusterTimerAckLevel[clusterName] = time.Unix(0, ackLevel)\n\t\t\tif s.shardInfo.TimerProcessingQueueStates.StatesByCluster == nil {\n\t\t\t\ts.shardInfo.TimerProcessingQueueStates.StatesByCluster = make(map[string][]*types.ProcessingQueueState)\n\t\t\t}\n\t\t\ts.shardInfo.TimerProcessingQueueStates.StatesByCluster[clusterName] = []*types.ProcessingQueueState{\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\t\t\tAckLevel: common.Int64Ptr(ackLevel),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\t\t\tReverseMatch: true,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\tcase persistence.HistoryTaskCategoryReplication:\n\t\treturn fmt.Errorf(\"replication queue state is not supported\")\n\t}\n\n\ts.shardInfo.StolenSinceRenew = 0\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) GetTransferProcessingQueueStates(cluster string) []*types.ProcessingQueueState {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\t// if we can find corresponding processing queue states\n\tif states, ok := s.shardInfo.TransferProcessingQueueStates.StatesByCluster[cluster]; ok {\n\t\treturn states\n\t}\n\n\t// check if we can find corresponding ack level\n\tvar ackLevel int64\n\tvar ok bool\n\tif ackLevel, ok = s.shardInfo.ClusterTransferAckLevel[cluster]; !ok {\n\t\t// otherwise, default to existing ack level, which belongs to local cluster\n\t\t// this can happen if you add more cluster\n\t\tackLevel = s.shardInfo.TransferAckLevel\n\t}\n\n\t// otherwise, create default queue state based on existing ack level,\n\t// which belongs to local cluster. this can happen if you add more cluster\n\treturn []*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(ackLevel),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc (s *contextImpl) UpdateTransferProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tif len(states) == 0 {\n\t\treturn errors.New(\"empty transfer processing queue states\")\n\t}\n\n\tif s.shardInfo.TransferProcessingQueueStates.StatesByCluster == nil {\n\t\ts.shardInfo.TransferProcessingQueueStates.StatesByCluster = make(map[string][]*types.ProcessingQueueState)\n\t}\n\ts.shardInfo.TransferProcessingQueueStates.StatesByCluster[cluster] = states\n\n\t// for backward compatibility\n\tackLevel := states[0].GetAckLevel()\n\tfor _, state := range states {\n\t\tackLevel = min(ackLevel, state.GetAckLevel())\n\t}\n\ts.shardInfo.ClusterTransferAckLevel[cluster] = ackLevel\n\n\ts.shardInfo.StolenSinceRenew = 0\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) GetTimerProcessingQueueStates(cluster string) []*types.ProcessingQueueState {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\t// if we can find corresponding processing queue states\n\tif states, ok := s.shardInfo.TimerProcessingQueueStates.StatesByCluster[cluster]; ok {\n\t\treturn states\n\t}\n\n\t// check if we can find corresponding ack level\n\tvar ackLevel time.Time\n\tvar ok bool\n\tif ackLevel, ok = s.shardInfo.ClusterTimerAckLevel[cluster]; !ok {\n\t\t// otherwise, default to existing ack level, which belongs to local cluster\n\t\t// this can happen if you add more cluster\n\t\tackLevel = s.shardInfo.TimerAckLevel\n\t}\n\n\t// otherwise, create default queue state based on existing ack level,\n\t// which belongs to local cluster. this can happen if you add more cluster\n\treturn []*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(ackLevel.UnixNano()),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc (s *contextImpl) UpdateTimerProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tif len(states) == 0 {\n\t\treturn errors.New(\"empty transfer processing queue states\")\n\t}\n\n\tif s.shardInfo.TimerProcessingQueueStates.StatesByCluster == nil {\n\t\ts.shardInfo.TimerProcessingQueueStates.StatesByCluster = make(map[string][]*types.ProcessingQueueState)\n\t}\n\ts.shardInfo.TimerProcessingQueueStates.StatesByCluster[cluster] = states\n\n\t// for backward compatibility\n\tackLevel := states[0].GetAckLevel()\n\tfor _, state := range states {\n\t\tackLevel = min(ackLevel, state.GetAckLevel())\n\t}\n\ts.shardInfo.ClusterTimerAckLevel[cluster] = time.Unix(0, ackLevel)\n\n\ts.shardInfo.StolenSinceRenew = 0\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) UpdateFailoverLevel(category persistence.HistoryTaskCategory, failoverID string, level persistence.FailoverLevel) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tif _, ok := s.failoverLevels[category]; !ok {\n\t\ts.failoverLevels[category] = make(map[string]persistence.FailoverLevel)\n\t}\n\ts.failoverLevels[category][failoverID] = level\n\treturn nil\n}\n\nfunc (s *contextImpl) DeleteFailoverLevel(category persistence.HistoryTaskCategory, failoverID string) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tif levels, ok := s.failoverLevels[category]; ok {\n\t\tif level, ok := levels[failoverID]; ok {\n\t\t\tdelete(levels, failoverID)\n\t\t\tswitch category {\n\t\t\tcase persistence.HistoryTaskCategoryTransfer:\n\t\t\t\ts.GetMetricsClient().RecordTimer(metrics.ShardInfoScope, metrics.ShardInfoTransferFailoverLatencyTimer, time.Since(level.StartTime))\n\t\t\tcase persistence.HistoryTaskCategoryTimer:\n\t\t\t\ts.GetMetricsClient().RecordTimer(metrics.ShardInfoScope, metrics.ShardInfoTimerFailoverLatencyTimer, time.Since(level.StartTime))\n\t\t\t}\n\t\t\treturn nil\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (s *contextImpl) GetAllFailoverLevels(category persistence.HistoryTaskCategory) map[string]persistence.FailoverLevel {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\tret := map[string]persistence.FailoverLevel{}\n\tfor k, v := range s.failoverLevels[category] {\n\t\tret[k] = v\n\t}\n\treturn ret\n}\n\nfunc (s *contextImpl) GetDomainNotificationVersion() int64 {\n\ts.RLock()\n\tdefer s.RUnlock()\n\n\treturn s.shardInfo.DomainNotificationVersion\n}\n\nfunc (s *contextImpl) UpdateDomainNotificationVersion(domainNotificationVersion int64) error {\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.shardInfo.DomainNotificationVersion = domainNotificationVersion\n\treturn s.updateShardInfoLocked()\n}\n\nfunc (s *contextImpl) GetWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.GetWorkflowExecutionRequest,\n) (*persistence.GetWorkflowExecutionResponse, error) {\n\trequest.RangeID = atomic.LoadInt64(&s.rangeID) // This is to make sure read is not blocked by write, s.rangeID is synced with s.shardInfo.RangeID\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn s.executionManager.GetWorkflowExecution(ctx, request)\n}\n\nfunc (s *contextImpl) CreateWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.CreateWorkflowExecutionRequest,\n) (*persistence.CreateWorkflowExecutionResponse, error) {\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tctx, cancel, err := s.ensureMinContextTimeout(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cancel != nil {\n\t\tdefer cancel()\n\t}\n\n\tdomainID := request.NewWorkflowSnapshot.ExecutionInfo.DomainID\n\tworkflowID := request.NewWorkflowSnapshot.ExecutionInfo.WorkflowID\n\n\t// do not try to get domain cache within shard lock\n\tdomainEntry, err := s.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\timmediateTaskMaxReadLevel := int64(0)\n\tif err := s.allocateTaskIDsLocked(\n\t\tdomainEntry,\n\t\tworkflowID,\n\t\trequest.NewWorkflowSnapshot.TasksByCategory,\n\t\t&immediateTaskMaxReadLevel,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\tcurrentRangeID := s.getRangeID()\n\trequest.RangeID = currentRangeID\n\n\tresponse, err := s.executionManager.CreateWorkflowExecution(ctx, request)\n\tswitch err.(type) {\n\tcase nil:\n\t\t// Update MaxReadLevel if write to DB succeeds\n\t\ts.updateMaxReadLevelLocked(immediateTaskMaxReadLevel)\n\t\ts.logCreateWorkflowExecutionEvents(request)\n\t\treturn response, nil\n\tcase *types.WorkflowExecutionAlreadyStartedError,\n\t\t*persistence.WorkflowExecutionAlreadyStartedError,\n\t\t*persistence.CurrentWorkflowConditionFailedError,\n\t\t*persistence.DuplicateRequestError,\n\t\t*types.ServiceBusyError:\n\t\t// No special handling required for these errors\n\t\t// We know write to DB fails if these errors are returned\n\t\treturn nil, err\n\tcase *persistence.ShardOwnershipLostError:\n\t\t{\n\t\t\t// Shard is stolen, trigger shutdown of history engine\n\t\t\ts.logger.Warn(\n\t\t\t\t\"Closing shard: CreateWorkflowExecution failed due to stolen shard.\",\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\ts.closeShard()\n\t\t\treturn nil, err\n\t\t}\n\tdefault:\n\t\t{\n\t\t\t// We have no idea if the write failed or will eventually make it to\n\t\t\t// persistence. Increment RangeID to guarantee that subsequent reads\n\t\t\t// will either see that write, or know for certain that it failed.\n\t\t\t// This allows the callers to reliably check the outcome by performing\n\t\t\t// a read.\n\t\t\terr1 := s.renewRangeLocked(false)\n\t\t\tif err1 != nil {\n\t\t\t\t// At this point we have no choice but to unload the shard, so that it\n\t\t\t\t// gets a new RangeID when it's reloaded.\n\t\t\t\ts.logger.Warn(\n\t\t\t\t\t\"Closing shard: CreateWorkflowExecution failed due to unknown error.\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\ts.closeShard()\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t}\n}\n\nfunc (s *contextImpl) getDefaultEncoding(domainName string) constants.EncodingType {\n\treturn constants.EncodingType(s.config.EventEncodingType(domainName))\n}\n\nfunc (s *contextImpl) UpdateWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.UpdateWorkflowExecutionRequest,\n) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\tctx, cancel, err := s.ensureMinContextTimeout(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cancel != nil {\n\t\tdefer cancel()\n\t}\n\n\tdomainID := request.UpdateWorkflowMutation.ExecutionInfo.DomainID\n\tworkflowID := request.UpdateWorkflowMutation.ExecutionInfo.WorkflowID\n\n\t// do not try to get domain cache within shard lock\n\tdomainEntry, err := s.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\trequest.Encoding = s.getDefaultEncoding(domainEntry.GetInfo().Name)\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\timmediateTaskMaxReadLevel := int64(0)\n\tif err := s.allocateTaskIDsLocked(\n\t\tdomainEntry,\n\t\tworkflowID,\n\t\trequest.UpdateWorkflowMutation.TasksByCategory,\n\t\t&immediateTaskMaxReadLevel,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\tif request.NewWorkflowSnapshot != nil {\n\t\tif err := s.allocateTaskIDsLocked(\n\t\t\tdomainEntry,\n\t\t\tworkflowID,\n\t\t\trequest.NewWorkflowSnapshot.TasksByCategory,\n\t\t\t&immediateTaskMaxReadLevel,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\tcurrentRangeID := s.getRangeID()\n\trequest.RangeID = currentRangeID\n\n\tresp, err := s.executionManager.UpdateWorkflowExecution(ctx, request)\n\tswitch err.(type) {\n\tcase nil:\n\t\t// Update MaxReadLevel if write to DB succeeds\n\t\ts.updateMaxReadLevelLocked(immediateTaskMaxReadLevel)\n\t\ts.logUpdateWorkflowExecutionEvents(request)\n\t\treturn resp, nil\n\tcase *persistence.ConditionFailedError,\n\t\t*persistence.DuplicateRequestError,\n\t\t*types.ServiceBusyError:\n\t\t// No special handling required for these errors\n\t\t// We know write to DB fails if these errors are returned\n\t\treturn nil, err\n\tcase *persistence.ShardOwnershipLostError:\n\t\t{\n\t\t\t// Shard is stolen, trigger shutdown of history engine\n\t\t\ts.logger.Warn(\n\t\t\t\t\"Closing shard: UpdateWorkflowExecution failed due to stolen shard.\",\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\ts.closeShard()\n\t\t\treturn nil, err\n\t\t}\n\tdefault:\n\t\t{\n\t\t\t// We have no idea if the write failed or will eventually make it to\n\t\t\t// persistence. Increment RangeID to guarantee that subsequent reads\n\t\t\t// will either see that write, or know for certain that it failed.\n\t\t\t// This allows the callers to reliably check the outcome by performing\n\t\t\t// a read.\n\t\t\terr1 := s.renewRangeLocked(false)\n\t\t\tif err1 != nil {\n\t\t\t\t// At this point we have no choice but to unload the shard, so that it\n\t\t\t\t// gets a new RangeID when it's reloaded.\n\t\t\t\ts.logger.Warn(\n\t\t\t\t\t\"Closing shard: UpdateWorkflowExecution failed due to unknown error.\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\ts.closeShard()\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t}\n}\n\nfunc (s *contextImpl) ConflictResolveWorkflowExecution(\n\tctx context.Context,\n\trequest *persistence.ConflictResolveWorkflowExecutionRequest,\n) (*persistence.ConflictResolveWorkflowExecutionResponse, error) {\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tctx, cancel, err := s.ensureMinContextTimeout(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif cancel != nil {\n\t\tdefer cancel()\n\t}\n\n\tdomainID := request.ResetWorkflowSnapshot.ExecutionInfo.DomainID\n\tworkflowID := request.ResetWorkflowSnapshot.ExecutionInfo.WorkflowID\n\n\t// do not try to get domain cache within shard lock\n\tdomainEntry, err := s.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trequest.Encoding = s.getDefaultEncoding(domainEntry.GetInfo().Name)\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\timmediateTaskMaxReadLevel := int64(0)\n\tif request.CurrentWorkflowMutation != nil {\n\t\tif err := s.allocateTaskIDsLocked(\n\t\t\tdomainEntry,\n\t\t\tworkflowID,\n\t\t\trequest.CurrentWorkflowMutation.TasksByCategory,\n\t\t\t&immediateTaskMaxReadLevel,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tif err := s.allocateTaskIDsLocked(\n\t\tdomainEntry,\n\t\tworkflowID,\n\t\trequest.ResetWorkflowSnapshot.TasksByCategory,\n\t\t&immediateTaskMaxReadLevel,\n\t); err != nil {\n\t\treturn nil, err\n\t}\n\tif request.NewWorkflowSnapshot != nil {\n\t\tif err := s.allocateTaskIDsLocked(\n\t\t\tdomainEntry,\n\t\t\tworkflowID,\n\t\t\trequest.NewWorkflowSnapshot.TasksByCategory,\n\t\t\t&immediateTaskMaxReadLevel,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\tcurrentRangeID := s.getRangeID()\n\trequest.RangeID = currentRangeID\n\tresp, err := s.executionManager.ConflictResolveWorkflowExecution(ctx, request)\n\tswitch err.(type) {\n\tcase nil:\n\t\t// Update MaxReadLevel if write to DB succeeds\n\t\ts.updateMaxReadLevelLocked(immediateTaskMaxReadLevel)\n\t\ts.logConflictResolveWorkflowExecutionEvents(request)\n\t\treturn resp, nil\n\tcase *persistence.ConditionFailedError,\n\t\t*types.ServiceBusyError:\n\t\t// No special handling required for these errors\n\t\t// We know write to DB fails if these errors are returned\n\t\treturn nil, err\n\tcase *persistence.ShardOwnershipLostError:\n\t\t{\n\t\t\t// RangeID might have been renewed by the same host while this update was in flight\n\t\t\t// Retry the operation if we still have the shard ownership\n\t\t\t// Shard is stolen, trigger shutdown of history engine\n\t\t\ts.logger.Warn(\n\t\t\t\t\"Closing shard: ConflictResolveWorkflowExecution failed due to stolen shard.\",\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\ts.closeShard()\n\t\t\treturn nil, err\n\t\t}\n\tdefault:\n\t\t{\n\t\t\t// We have no idea if the write failed or will eventually make it to\n\t\t\t// persistence. Increment RangeID to guarantee that subsequent reads\n\t\t\t// will either see that write, or know for certain that it failed.\n\t\t\t// This allows the callers to reliably check the outcome by performing\n\t\t\t// a read.\n\t\t\terr1 := s.renewRangeLocked(false)\n\t\t\tif err1 != nil {\n\t\t\t\t// At this point we have no choice but to unload the shard, so that it\n\t\t\t\t// gets a new RangeID when it's reloaded.\n\t\t\t\ts.logger.Warn(\n\t\t\t\t\t\"Closing shard: ConflictResolveWorkflowExecution failed due to unknown error.\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\ts.closeShard()\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t}\n}\n\nfunc (s *contextImpl) ensureMinContextTimeout(\n\tparent context.Context,\n) (context.Context, context.CancelFunc, error) {\n\tif err := parent.Err(); err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tdeadline, ok := parent.Deadline()\n\tif !ok || deadline.Sub(s.GetTimeSource().Now()) >= minContextTimeout {\n\t\treturn parent, nil, nil\n\t}\n\n\tchildCtx, cancel := context.WithTimeout(context.Background(), minContextTimeout)\n\treturn childCtx, cancel, nil\n}\n\nfunc (s *contextImpl) AppendHistoryV2Events(\n\tctx context.Context,\n\trequest *persistence.AppendHistoryNodesRequest,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n) (*persistence.AppendHistoryNodesResponse, error) {\n\tif err := s.closedError(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomainName, err := s.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// NOTE: do not use generateNextTransferTaskIDLocked since\n\t// generateNextTransferTaskIDLocked is not guarded by lock\n\ttransactionID, err := s.GenerateTaskID()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trequest.Encoding = s.getDefaultEncoding(domainName)\n\trequest.ShardID = common.IntPtr(s.shardID)\n\trequest.TransactionID = transactionID\n\n\tsize := 0\n\tdefer func() {\n\t\ts.GetMetricsClient().Scope(metrics.SessionSizeStatsScope, metrics.DomainTag(domainName)).\n\t\t\tRecordTimer(metrics.HistorySize, time.Duration(size))\n\t\tif size >= historySizeLogThreshold {\n\t\t\ts.throttledLogger.Warn(\"history size threshold breached\",\n\t\t\t\ttag.WorkflowID(execution.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(execution.GetRunID()),\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.WorkflowHistorySizeBytes(size))\n\t\t}\n\t}()\n\tresp, err0 := s.GetHistoryManager().AppendHistoryNodes(ctx, request)\n\tif resp != nil {\n\t\tsize = len(resp.DataBlob.Data)\n\t}\n\treturn resp, err0\n}\n\nfunc (s *contextImpl) GetConfig() *config.Config {\n\treturn s.config\n}\n\nfunc (s *contextImpl) PreviousShardOwnerWasDifferent() bool {\n\treturn s.previousShardOwnerWasDifferent\n}\n\nfunc (s *contextImpl) GetEventsCache() events.Cache {\n\t// the shard needs to be restarted to release the shard cache once global mode is on.\n\tif s.config.EventsCacheGlobalEnable() {\n\t\treturn s.GetEventCache()\n\t}\n\treturn s.eventsCache\n}\n\nfunc (s *contextImpl) GetLogger() log.Logger {\n\treturn s.logger\n}\n\nfunc (s *contextImpl) GetThrottledLogger() log.Logger {\n\treturn s.throttledLogger\n}\n\nfunc (s *contextImpl) GetReplicationBudgetManager() cache.Manager {\n\treturn s.replicationBudgetManager\n}\n\nfunc (s *contextImpl) getRangeID() int64 {\n\treturn s.shardInfo.RangeID\n}\n\nfunc (s *contextImpl) closedError() error {\n\tclosedAt := s.closedAt.Load()\n\tif closedAt == nil {\n\t\treturn nil\n\t}\n\n\treturn &ErrShardClosed{\n\t\tMsg:      \"shard closed\",\n\t\tClosedAt: *closedAt,\n\t}\n}\n\nfunc (s *contextImpl) closeShard() {\n\tif !s.closedAt.CompareAndSwap(nil, common.TimePtr(time.Now())) {\n\t\treturn\n\t}\n\n\ts.logger.Info(\"Shard context closeShard called\")\n\tgo func() {\n\t\ts.closeCallback(s.shardID, s.shardItem)\n\t}()\n\n\t// fails any writes that may start after this point.\n\ts.shardInfo.RangeID = -1\n\tatomic.StoreInt64(&s.rangeID, s.shardInfo.RangeID)\n}\n\nfunc (s *contextImpl) generateTaskIDLocked() (int64, error) {\n\tif err := s.updateRangeIfNeededLocked(); err != nil {\n\t\treturn -1, err\n\t}\n\n\ttaskID := s.taskSequenceNumber\n\ts.taskSequenceNumber++\n\n\treturn taskID, nil\n}\n\nfunc (s *contextImpl) updateRangeIfNeededLocked() error {\n\tif s.taskSequenceNumber < s.maxTaskSequenceNumber {\n\t\treturn nil\n\t}\n\n\treturn s.renewRangeLocked(false)\n}\n\nfunc (s *contextImpl) renewRangeLocked(isStealing bool) error {\n\tupdatedShardInfo := s.shardInfo.ToNilSafeCopy()\n\tupdatedShardInfo.RangeID++\n\tif isStealing {\n\t\tupdatedShardInfo.StolenSinceRenew++\n\t}\n\n\tvar err error\n\tif err := s.closedError(); err != nil {\n\t\treturn err\n\t}\n\terr = s.GetShardManager().UpdateShard(context.Background(), &persistence.UpdateShardRequest{\n\t\tShardInfo:       updatedShardInfo,\n\t\tPreviousRangeID: s.shardInfo.RangeID})\n\tswitch err.(type) {\n\tcase nil:\n\tcase *persistence.ShardOwnershipLostError:\n\t\t// Shard is stolen, trigger history engine shutdown\n\t\ts.logger.Warn(\n\t\t\t\"Closing shard: renewRangeLocked failed due to stolen shard.\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\ts.closeShard()\n\tdefault:\n\t\ts.logger.Warn(\"UpdateShard failed with an unknown error.\",\n\t\t\ttag.Error(err),\n\t\t\ttag.ShardRangeID(updatedShardInfo.RangeID),\n\t\t\ttag.PreviousShardRangeID(s.shardInfo.RangeID))\n\t}\n\tif err != nil {\n\t\t// Failure in updating shard to grab new RangeID\n\t\ts.logger.Error(\"renewRangeLocked failed.\",\n\t\t\ttag.StoreOperationUpdateShard,\n\t\t\ttag.Error(err),\n\t\t\ttag.ShardRangeID(updatedShardInfo.RangeID),\n\t\t\ttag.PreviousShardRangeID(s.shardInfo.RangeID))\n\t\treturn err\n\t}\n\n\t// Range is successfully updated in cassandra now update shard context to reflect new range\n\ts.taskSequenceNumber = updatedShardInfo.RangeID << s.config.RangeSizeBits\n\ts.maxTaskSequenceNumber = (updatedShardInfo.RangeID + 1) << s.config.RangeSizeBits\n\ts.immediateTaskMaxReadLevel = s.taskSequenceNumber - 1\n\tatomic.StoreInt64(&s.rangeID, updatedShardInfo.RangeID)\n\ts.shardInfo = updatedShardInfo\n\n\ts.logger.Info(\"Range updated for shardID\",\n\t\ttag.ShardRangeID(s.shardInfo.RangeID),\n\t\ttag.Number(s.taskSequenceNumber),\n\t\ttag.NextNumber(s.maxTaskSequenceNumber))\n\treturn nil\n}\n\nfunc (s *contextImpl) updateMaxReadLevelLocked(rl int64) {\n\tif rl > s.immediateTaskMaxReadLevel {\n\t\ts.logger.Debug(fmt.Sprintf(\"Updating MaxReadLevel: %v\", rl))\n\t\ts.immediateTaskMaxReadLevel = rl\n\t}\n}\n\nfunc (s *contextImpl) updateShardInfoLocked() error {\n\treturn s.persistShardInfoLocked(false)\n}\n\nfunc (s *contextImpl) forceUpdateShardInfoLocked() error {\n\treturn s.persistShardInfoLocked(true)\n}\n\nfunc (s *contextImpl) persistShardInfoLocked(\n\tisForced bool,\n) error {\n\n\tif err := s.closedError(); err != nil {\n\t\treturn err\n\t}\n\n\tvar err error\n\tnow := clock.NewRealTimeSource().Now()\n\tif !isForced && s.lastUpdated.Add(s.config.ShardUpdateMinInterval()).After(now) {\n\t\treturn nil\n\t}\n\tupdatedShardInfo := s.shardInfo.ToNilSafeCopy()\n\ts.emitShardInfoMetricsLogsLocked()\n\n\terr = s.GetShardManager().UpdateShard(context.Background(), &persistence.UpdateShardRequest{\n\t\tShardInfo:       updatedShardInfo,\n\t\tPreviousRangeID: s.shardInfo.RangeID,\n\t})\n\n\tif err != nil {\n\t\t// Shard is stolen, trigger history engine shutdown\n\t\tif _, ok := err.(*persistence.ShardOwnershipLostError); ok {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"Closing shard: updateShardInfoLocked failed due to stolen shard.\",\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\ts.closeShard()\n\t\t}\n\t} else {\n\t\ts.lastUpdated = now\n\t}\n\n\treturn err\n}\n\nfunc (s *contextImpl) emitShardInfoMetricsLogsLocked() {\n\tcurrentCluster := s.GetClusterMetadata().GetCurrentClusterName()\n\tclusterInfo := s.GetClusterMetadata().GetAllClusterInfo()\n\n\tminTransferLevel := s.shardInfo.ClusterTransferAckLevel[currentCluster]\n\tmaxTransferLevel := s.shardInfo.ClusterTransferAckLevel[currentCluster]\n\tfor clusterName, v := range s.shardInfo.ClusterTransferAckLevel {\n\t\tif !clusterInfo[clusterName].Enabled {\n\t\t\tcontinue\n\t\t}\n\n\t\tif v < minTransferLevel {\n\t\t\tminTransferLevel = v\n\t\t}\n\t\tif v > maxTransferLevel {\n\t\t\tmaxTransferLevel = v\n\t\t}\n\t}\n\tdiffTransferLevel := maxTransferLevel - minTransferLevel\n\n\tminTimerLevel := s.shardInfo.ClusterTimerAckLevel[currentCluster]\n\tmaxTimerLevel := s.shardInfo.ClusterTimerAckLevel[currentCluster]\n\tfor clusterName, v := range s.shardInfo.ClusterTimerAckLevel {\n\t\tif !clusterInfo[clusterName].Enabled {\n\t\t\tcontinue\n\t\t}\n\n\t\tif v.Before(minTimerLevel) {\n\t\t\tminTimerLevel = v\n\t\t}\n\t\tif v.After(maxTimerLevel) {\n\t\t\tmaxTimerLevel = v\n\t\t}\n\t}\n\tdiffTimerLevel := maxTimerLevel.Sub(minTimerLevel)\n\n\treplicationLag := s.immediateTaskMaxReadLevel - s.shardInfo.ReplicationAckLevel\n\ttransferLag := s.immediateTaskMaxReadLevel - s.shardInfo.TransferAckLevel\n\ttimerLag := time.Since(s.shardInfo.TimerAckLevel)\n\n\ttransferFailoverInProgress := len(s.failoverLevels[persistence.HistoryTaskCategoryTransfer])\n\ttimerFailoverInProgress := len(s.failoverLevels[persistence.HistoryTaskCategoryTimer])\n\n\tif s.config.EmitShardDiffLog() &&\n\t\t(logWarnTransferLevelDiff < diffTransferLevel ||\n\t\t\tlogWarnTimerLevelDiff < diffTimerLevel ||\n\t\t\tlogWarnTransferLevelDiff < transferLag ||\n\t\t\tlogWarnTimerLevelDiff < timerLag) {\n\n\t\tlogger := s.logger.WithTags(\n\t\t\ttag.ShardTime(s.remoteClusterCurrentTime),\n\t\t\ttag.ShardReplicationAck(s.shardInfo.ReplicationAckLevel),\n\t\t\ttag.ShardTimerAcks(s.shardInfo.ClusterTimerAckLevel),\n\t\t\ttag.ShardTransferAcks(s.shardInfo.ClusterTransferAckLevel),\n\t\t)\n\n\t\tlogger.Warn(\"Shard ack levels diff exceeds warn threshold.\")\n\t}\n\n\tmetricsScope := s.GetMetricsClient().Scope(metrics.ShardInfoScope)\n\tmetricsScope.RecordTimer(metrics.ShardInfoTransferDiffTimer, time.Duration(diffTransferLevel))\n\tmetricsScope.RecordTimer(metrics.ShardInfoTimerDiffTimer, diffTimerLevel)\n\n\tmetricsScope.RecordTimer(metrics.ShardInfoReplicationLagTimer, time.Duration(replicationLag))\n\tmetricsScope.RecordTimer(metrics.ShardInfoTransferLagTimer, time.Duration(transferLag))\n\tmetricsScope.RecordTimer(metrics.ShardInfoTimerLagTimer, timerLag)\n\n\tmetricsScope.RecordTimer(metrics.ShardInfoTransferFailoverInProgressTimer, time.Duration(transferFailoverInProgress))\n\tmetricsScope.RecordTimer(metrics.ShardInfoTimerFailoverInProgressTimer, time.Duration(timerFailoverInProgress))\n}\n\nfunc (s *contextImpl) allocateTaskIDsLocked(\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowID string,\n\ttasksByCategory map[persistence.HistoryTaskCategory][]persistence.Task,\n\timmediateTaskMaxReadLevel *int64,\n) error {\n\tvar err error\n\tvar replicationTasks []persistence.Task\n\tfor c, tasks := range tasksByCategory {\n\t\tswitch c.Type() {\n\t\tcase persistence.HistoryTaskCategoryTypeImmediate:\n\t\t\tif c.ID() == persistence.HistoryTaskCategoryIDReplication {\n\t\t\t\treplicationTasks = tasks\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\terr = s.allocateTransferIDsLocked(tasks, immediateTaskMaxReadLevel)\n\t\tcase persistence.HistoryTaskCategoryTypeScheduled:\n\t\t\terr = s.allocateTimerIDsLocked(domainEntry, workflowID, tasks)\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t// Ensure that task IDs for replication tasks are generated last.\n\t// This allows optimizing replication by checking whether there no potential tasks to read.\n\treturn s.allocateTransferIDsLocked(\n\t\treplicationTasks,\n\t\timmediateTaskMaxReadLevel,\n\t)\n}\n\nfunc (s *contextImpl) allocateTransferIDsLocked(\n\ttasks []persistence.Task,\n\timmediateTaskMaxReadLevel *int64,\n) error {\n\tnow := s.GetTimeSource().Now()\n\n\tfor _, task := range tasks {\n\t\tid, err := s.generateTaskIDLocked()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ts.logger.Debug(fmt.Sprintf(\"Assigning task ID: %v\", id))\n\t\ttask.SetTaskID(id)\n\t\t// only set task visibility timestamp if it's not set\n\t\tif task.GetVisibilityTimestamp().IsZero() {\n\t\t\ttask.SetVisibilityTimestamp(now)\n\t\t}\n\t\t*immediateTaskMaxReadLevel = id\n\t}\n\treturn nil\n}\n\n// NOTE: allocateTimerIDsLocked should always been called after assigning taskID for transferTasks when assigning taskID together,\n// because Cadence Indexer assume timer taskID of deleteWorkflowExecution is larger than transfer taskID of closeWorkflowExecution\n// for a given workflow.\nfunc (s *contextImpl) allocateTimerIDsLocked(\n\tdomainEntry *cache.DomainCacheEntry,\n\tworkflowID string,\n\ttimerTasks []persistence.Task,\n) error {\n\tnow := s.GetTimeSource().Now().Truncate(persistence.DBTimestampMinPrecision)\n\t// assign IDs for the timer tasks. They need to be assigned under shard lock.\n\tcluster := s.GetClusterMetadata().GetCurrentClusterName()\n\tfor _, task := range timerTasks {\n\t\tts := task.GetVisibilityTimestamp().Truncate(persistence.DBTimestampMinPrecision)\n\t\t// always use current cluster's max read level for queue v2, and this is safe for rollback,\n\t\t// because if we go back to queue v1, the standby queue and active queue will start from the same ack level to read tasks\n\t\tif task.GetVersion() != constants.EmptyVersion && !s.GetConfig().EnableTimerQueueV2(s.shardID) {\n\t\t\t// cannot use version to determine the corresponding cluster for timer task\n\t\t\t// this is because during failover, timer task should be created as active\n\t\t\t// or otherwise, failover + active processing logic may not pick up the task.\n\t\t\tcluster = domainEntry.GetReplicationConfig().ActiveClusterName\n\n\t\t\tif domainEntry.GetReplicationConfig().IsActiveActive() {\n\t\t\t\t// Note: This doesn't work for initial backoff timer task because the workflow's active-cluster-selection-policy row is not stored yet.\n\t\t\t\t// Therefore GetActiveClusterInfoByWorkflow returns current cluster (fallback logic in activecluster manager)\n\t\t\t\t// Queue v2 doesn't use this logic and it must be enabled to properly handle initial backoff timer task for active-active domains.\n\t\t\t\t// Leaving this code block instead of rejecting the whole id allocation request.\n\t\t\t\t// Active-active domains should not be used in Cadence clusters that don't have queue v2 enabled.\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), activeClusterLookupTimeout)\n\t\t\t\tlookupRes, err := s.GetActiveClusterManager().GetActiveClusterInfoByWorkflow(ctx, task.GetDomainID(), task.GetWorkflowID(), task.GetRunID())\n\t\t\t\tcancel()\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tcluster = lookupRes.ActiveClusterName\n\t\t\t}\n\t\t}\n\n\t\treadCursorTS := s.scheduledTaskMaxReadLevelMap[cluster]\n\t\t// make sure scheduled task timestamp is higher than\n\t\t// 1. max read level, so that queue processor can read the task back.\n\t\t// 2. current time. Otherwise the task timestamp is in the past and causes aritical load latency in queue processor metrics.\n\t\t// Above cases can happen if shard move and new host have a time SKU,\n\t\t// or there is db write delay, or we are simply (re-)generating tasks for an old workflow.\n\t\tif ts.Before(readCursorTS) {\n\t\t\t// This can happen if shard move and new host have a time SKU, or there is db write delay.\n\t\t\t// We generate a new timer ID using timerMaxReadLevel.\n\t\t\ts.logger.Warn(\"New timer generated is less than read level\",\n\t\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\ttag.Timestamp(ts),\n\t\t\t\ttag.CursorTimestamp(readCursorTS),\n\t\t\t\ttag.ClusterName(cluster),\n\t\t\t\ttag.ValueShardAllocateTimerBeforeRead)\n\t\t\tts = readCursorTS.Add(persistence.DBTimestampMinPrecision)\n\t\t}\n\t\tif ts.Before(now) {\n\t\t\ts.logger.Warn(\"New timer generated is in the past\",\n\t\t\t\ttag.WorkflowDomainID(domainEntry.GetInfo().ID),\n\t\t\t\ttag.WorkflowID(workflowID),\n\t\t\t\ttag.Timestamp(ts),\n\t\t\t\ttag.ValueShardAllocateTimerBeforeRead)\n\t\t\tts = now.Add(persistence.DBTimestampMinPrecision)\n\t\t}\n\t\ttask.SetVisibilityTimestamp(ts)\n\n\t\tseqNum, err := s.generateTaskIDLocked()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttask.SetTaskID(seqNum)\n\t\tvisibilityTs := task.GetVisibilityTimestamp()\n\t\ts.logger.Debug(fmt.Sprintf(\"Assigning new timer (timestamp: %v, seq: %v)) ackLeveL: %v\",\n\t\t\tvisibilityTs, task.GetTaskID(), s.shardInfo.TimerAckLevel))\n\t}\n\treturn nil\n}\n\nfunc (s *contextImpl) SetCurrentTime(cluster string, currentTime time.Time) {\n\ts.Lock()\n\tdefer s.Unlock()\n\tif cluster != s.GetClusterMetadata().GetCurrentClusterName() {\n\t\tprevTime := s.remoteClusterCurrentTime[cluster]\n\t\tif prevTime.Before(currentTime) {\n\t\t\ts.remoteClusterCurrentTime[cluster] = currentTime\n\t\t}\n\t} else {\n\t\tpanic(\"Cannot set current time for current cluster\")\n\t}\n}\n\nfunc (s *contextImpl) GetCurrentTime(cluster string) time.Time {\n\ts.RLock()\n\tdefer s.RUnlock()\n\tif cluster != s.GetClusterMetadata().GetCurrentClusterName() {\n\t\treturn s.remoteClusterCurrentTime[cluster]\n\t}\n\treturn s.GetTimeSource().Now()\n}\n\nfunc (s *contextImpl) GetLastUpdatedTime() time.Time {\n\ts.RLock()\n\tdefer s.RUnlock()\n\treturn s.lastUpdated\n}\n\nfunc (s *contextImpl) ReplicateFailoverMarkers(\n\tctx context.Context,\n\tmarkers []*persistence.FailoverMarkerTask,\n) error {\n\tif err := s.closedError(); err != nil {\n\t\treturn err\n\t}\n\n\ttasks := make([]persistence.Task, 0, len(markers))\n\tfor _, marker := range markers {\n\t\ttasks = append(tasks, marker)\n\t}\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\timmediateTaskMaxReadLevel := int64(0)\n\tif err := s.allocateTransferIDsLocked(\n\t\ttasks,\n\t\t&immediateTaskMaxReadLevel,\n\t); err != nil {\n\t\treturn err\n\t}\n\n\tvar err error\n\tif err := s.closedError(); err != nil {\n\t\treturn err\n\t}\n\terr = s.executionManager.CreateFailoverMarkerTasks(\n\t\tctx,\n\t\t&persistence.CreateFailoverMarkersRequest{\n\t\t\tRangeID: s.getRangeID(),\n\t\t\tMarkers: markers,\n\t\t},\n\t)\n\tswitch err.(type) {\n\tcase nil:\n\t\t// Update MaxReadLevel if write to DB succeeds\n\t\ts.updateMaxReadLevelLocked(immediateTaskMaxReadLevel)\n\tcase *persistence.ShardOwnershipLostError:\n\t\t// do not retry on ShardOwnershipLostError\n\t\ts.logger.Warn(\n\t\t\t\"Closing shard: ReplicateFailoverMarkers failed due to stolen shard.\",\n\t\t\ttag.Error(err),\n\t\t)\n\t\ts.closeShard()\n\tdefault:\n\t\ts.logger.Error(\n\t\t\t\"Failed to insert the failover marker into replication queue.\",\n\t\t\ttag.Error(err),\n\t\t)\n\t}\n\treturn err\n}\n\nfunc (s *contextImpl) AddingPendingFailoverMarker(\n\tmarker *types.FailoverMarkerAttributes,\n) error {\n\n\tdomainEntry, err := s.GetDomainCache().GetDomainByID(marker.GetDomainID())\n\tif err != nil {\n\t\treturn err\n\t}\n\t// domain is active, the marker is expired\n\tisActive := domainEntry.IsActiveIn(s.GetClusterMetadata().GetCurrentClusterName())\n\tif isActive || domainEntry.GetFailoverVersion() > marker.GetFailoverVersion() {\n\t\ts.logger.Info(\"Skipped out-of-date failover marker\", tag.WorkflowDomainName(domainEntry.GetInfo().Name))\n\t\treturn nil\n\t}\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\ts.shardInfo.PendingFailoverMarkers = append(s.shardInfo.PendingFailoverMarkers, marker)\n\treturn s.forceUpdateShardInfoLocked()\n}\n\nfunc (s *contextImpl) ValidateAndUpdateFailoverMarkers() ([]*types.FailoverMarkerAttributes, error) {\n\n\tcompletedFailoverMarkers := make(map[*types.FailoverMarkerAttributes]struct{})\n\tvar pendingMarkers []*types.FailoverMarkerAttributes\n\n\ts.RLock()\n\t// Get a copy of pending markers while holding read lock\n\tpendingMarkers = make([]*types.FailoverMarkerAttributes, len(s.shardInfo.PendingFailoverMarkers))\n\tcopy(pendingMarkers, s.shardInfo.PendingFailoverMarkers)\n\n\tfor _, marker := range s.shardInfo.PendingFailoverMarkers {\n\t\tdomainEntry, err := s.GetDomainCache().GetDomainByID(marker.GetDomainID())\n\t\tif err != nil {\n\t\t\ts.RUnlock()\n\t\t\treturn nil, err\n\t\t}\n\n\t\tisActive := domainEntry.IsActiveIn(s.GetClusterMetadata().GetCurrentClusterName())\n\t\tdomainStatus := domainEntry.GetInfo().Status\n\n\t\t// Drop failover markers if domain is already active in the currentCluster\n\t\t// or domain have been failed over\n\t\t// or domain is deprecated\n\t\tif isActive || domainEntry.GetFailoverVersion() > marker.GetFailoverVersion() || domainStatus == persistence.DomainStatusDeprecated {\n\t\t\tcompletedFailoverMarkers[marker] = struct{}{}\n\t\t}\n\t}\n\ts.RUnlock()\n\n\tif len(completedFailoverMarkers) == 0 {\n\t\t// No markers to clean up, return the copy\n\t\treturn pendingMarkers, nil\n\t}\n\n\t// clean up all pending failover tasks\n\ts.Lock()\n\tdefer s.Unlock()\n\n\t// Re-read the current state since it might have changed\n\tcurrentPendingMarkers := s.shardInfo.PendingFailoverMarkers\n\tremainingMarkers := make([]*types.FailoverMarkerAttributes, 0, len(currentPendingMarkers))\n\n\tfor _, marker := range currentPendingMarkers {\n\t\tif _, ok := completedFailoverMarkers[marker]; !ok {\n\t\t\tremainingMarkers = append(remainingMarkers, marker)\n\t\t}\n\t}\n\n\ts.shardInfo.PendingFailoverMarkers = remainingMarkers\n\tif err := s.updateShardInfoLocked(); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn s.shardInfo.PendingFailoverMarkers, nil\n}\n\nfunc acquireShard(\n\tshardItem *historyShardsItem,\n\tcloseCallback func(int, *historyShardsItem),\n) (Context, error) {\n\n\tvar shardInfo *persistence.ShardInfo\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(50 * time.Millisecond)\n\tretryPolicy.SetMaximumInterval(time.Second)\n\tretryPolicy.SetExpirationInterval(5 * time.Second)\n\n\tretryPredicate := func(err error) bool {\n\t\tif persistence.IsTransientError(err) {\n\t\t\treturn true\n\t\t}\n\t\t_, ok := err.(*persistence.ShardAlreadyExistError)\n\t\treturn ok\n\t}\n\n\tgetShard := func(ctx context.Context) error {\n\t\tresp, err := shardItem.GetShardManager().GetShard(ctx, &persistence.GetShardRequest{\n\t\t\tShardID: shardItem.shardID,\n\t\t})\n\t\tif err == nil {\n\t\t\tshardInfo = resp.ShardInfo\n\t\t\treturn nil\n\t\t}\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\n\t\t// EntityNotExistsError error\n\t\tshardInfo = &persistence.ShardInfo{\n\t\t\tShardID:          shardItem.shardID,\n\t\t\tRangeID:          0,\n\t\t\tTransferAckLevel: 0,\n\t\t}\n\t\treturn shardItem.GetShardManager().CreateShard(ctx, &persistence.CreateShardRequest{ShardInfo: shardInfo})\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(retryPredicate),\n\t)\n\terr := throttleRetry.Do(context.Background(), getShard)\n\tif err != nil {\n\t\tshardItem.logger.Error(\"Fail to acquire shard.\", tag.Error(err))\n\t\treturn nil, err\n\t}\n\n\tupdatedShardInfo := shardInfo.ToNilSafeCopy()\n\townershipChanged := shardInfo.Owner != shardItem.GetHostInfo().Identity()\n\tupdatedShardInfo.Owner = shardItem.GetHostInfo().Identity()\n\n\t// initialize the cluster current time to be the same as ack level\n\tremoteClusterCurrentTime := make(map[string]time.Time)\n\t// TODO: get this information from QueueState once TimerAckLevel field is deprecated\n\tscheduledTaskMaxReadLevelMap := make(map[string]time.Time)\n\tfor clusterName := range shardItem.GetClusterMetadata().GetEnabledClusterInfo() {\n\t\tif clusterName != shardItem.GetClusterMetadata().GetCurrentClusterName() {\n\t\t\tif currentTime, ok := shardInfo.ClusterTimerAckLevel[clusterName]; ok {\n\t\t\t\tremoteClusterCurrentTime[clusterName] = currentTime\n\t\t\t\tscheduledTaskMaxReadLevelMap[clusterName] = currentTime\n\t\t\t} else {\n\t\t\t\tremoteClusterCurrentTime[clusterName] = shardInfo.TimerAckLevel\n\t\t\t\tscheduledTaskMaxReadLevelMap[clusterName] = shardInfo.TimerAckLevel\n\t\t\t}\n\t\t} else { // active cluster\n\t\t\tscheduledTaskMaxReadLevelMap[clusterName] = shardInfo.TimerAckLevel\n\t\t}\n\n\t\tscheduledTaskMaxReadLevelMap[clusterName] = scheduledTaskMaxReadLevelMap[clusterName].Truncate(persistence.DBTimestampMinPrecision)\n\t}\n\n\texecutionMgr, err := shardItem.GetExecutionManager(shardItem.shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcontext := &contextImpl{\n\t\tResource:                       shardItem.Resource,\n\t\tshardItem:                      shardItem,\n\t\tshardID:                        shardItem.shardID,\n\t\texecutionManager:               executionMgr,\n\t\tactiveClusterManager:           shardItem.GetActiveClusterManager(),\n\t\tshardInfo:                      updatedShardInfo,\n\t\tcloseCallback:                  closeCallback,\n\t\tconfig:                         shardItem.config,\n\t\tremoteClusterCurrentTime:       remoteClusterCurrentTime,\n\t\tscheduledTaskMaxReadLevelMap:   scheduledTaskMaxReadLevelMap, // use ack to init read level\n\t\tfailoverLevels:                 make(map[persistence.HistoryTaskCategory]map[string]persistence.FailoverLevel),\n\t\tlogger:                         shardItem.logger,\n\t\tthrottledLogger:                shardItem.throttledLogger,\n\t\tpreviousShardOwnerWasDifferent: ownershipChanged,\n\t\treplicationBudgetManager:       shardItem.replicationBudgetManager,\n\t}\n\n\t// TODO remove once migrated to global event cache\n\tcontext.eventsCache = events.NewCache(\n\t\tcontext.shardID,\n\t\tcontext.Resource.GetHistoryManager(),\n\t\tcontext.config,\n\t\tcontext.logger,\n\t\tcontext.Resource.GetMetricsClient(),\n\t\tshardItem.GetDomainCache(),\n\t)\n\n\tcontext.logger.Debug(fmt.Sprintf(\"Global event cache mode: %v\", context.config.EventsCacheGlobalEnable()))\n\n\terr1 := context.renewRangeLocked(true)\n\tif err1 != nil {\n\t\treturn nil, err1\n\t}\n\n\treturn context, nil\n}\n\nfunc (s *contextImpl) getEventsFromWorkflowSnapshot(snapshot *persistence.WorkflowSnapshot) []simulation.E {\n\tif snapshot == nil {\n\t\treturn nil\n\t}\n\tvar events []simulation.E\n\tfor category, tasks := range snapshot.TasksByCategory {\n\t\tfor _, task := range tasks {\n\t\t\tevents = append(events, simulation.E{\n\t\t\t\tEventName:  simulation.EventNameCreateHistoryTask,\n\t\t\t\tHost:       s.config.HostName,\n\t\t\t\tShardID:    s.shardID,\n\t\t\t\tDomainID:   task.GetDomainID(),\n\t\t\t\tWorkflowID: task.GetWorkflowID(),\n\t\t\t\tRunID:      task.GetRunID(),\n\t\t\t\tPayload: map[string]any{\n\t\t\t\t\t\"task_category\": category.Name(),\n\t\t\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\treturn events\n}\n\nfunc (s *contextImpl) getEventsFromWorkflowMutation(mutation *persistence.WorkflowMutation) []simulation.E {\n\tif mutation == nil {\n\t\treturn nil\n\t}\n\tvar events []simulation.E\n\tfor category, tasks := range mutation.TasksByCategory {\n\t\tfor _, task := range tasks {\n\t\t\tevents = append(events, simulation.E{\n\t\t\t\tEventName:  simulation.EventNameCreateHistoryTask,\n\t\t\t\tHost:       s.config.HostName,\n\t\t\t\tShardID:    s.shardID,\n\t\t\t\tDomainID:   task.GetDomainID(),\n\t\t\t\tWorkflowID: task.GetWorkflowID(),\n\t\t\t\tRunID:      task.GetRunID(),\n\t\t\t\tPayload: map[string]any{\n\t\t\t\t\t\"task_category\": category.Name(),\n\t\t\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\treturn events\n}\n\nfunc (s *contextImpl) logCreateWorkflowExecutionEvents(request *persistence.CreateWorkflowExecutionRequest) {\n\tif !simulation.Enabled() {\n\t\treturn\n\t}\n\tevents := s.getEventsFromWorkflowSnapshot(&request.NewWorkflowSnapshot)\n\tsimulation.LogEvents(events...)\n}\n\nfunc (s *contextImpl) logUpdateWorkflowExecutionEvents(request *persistence.UpdateWorkflowExecutionRequest) {\n\tif !simulation.Enabled() {\n\t\treturn\n\t}\n\tevents := s.getEventsFromWorkflowMutation(&request.UpdateWorkflowMutation)\n\tsimulation.LogEvents(events...)\n\tevents = s.getEventsFromWorkflowSnapshot(request.NewWorkflowSnapshot)\n\tsimulation.LogEvents(events...)\n}\n\nfunc (s *contextImpl) logConflictResolveWorkflowExecutionEvents(request *persistence.ConflictResolveWorkflowExecutionRequest) {\n\tif !simulation.Enabled() {\n\t\treturn\n\t}\n\tevents := s.getEventsFromWorkflowMutation(request.CurrentWorkflowMutation)\n\tsimulation.LogEvents(events...)\n\tevents = s.getEventsFromWorkflowSnapshot(&request.ResetWorkflowSnapshot)\n\tsimulation.LogEvents(events...)\n\tevents = s.getEventsFromWorkflowSnapshot(request.NewWorkflowSnapshot)\n\tsimulation.LogEvents(events...)\n}\n"
  },
  {
    "path": "service/history/shard/context_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: context.go\n//\n// Generated by this command:\n//\n//\tmockgen -package shard -source context.go -destination context_mock.go -package shard github.com/uber/cadence/history/shard/context Context\n//\n\n// Package shard is a generated GoMock package.\npackage shard\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tactivecluster \"github.com/uber/cadence/common/activecluster\"\n\tcache \"github.com/uber/cadence/common/cache\"\n\tclock \"github.com/uber/cadence/common/clock\"\n\tcluster \"github.com/uber/cadence/common/cluster\"\n\tlog \"github.com/uber/cadence/common/log\"\n\tmetrics \"github.com/uber/cadence/common/metrics\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttypes \"github.com/uber/cadence/common/types\"\n\tconfig \"github.com/uber/cadence/service/history/config\"\n\tengine \"github.com/uber/cadence/service/history/engine\"\n\tevents \"github.com/uber/cadence/service/history/events\"\n\tresource \"github.com/uber/cadence/service/history/resource\"\n)\n\n// MockContext is a mock of Context interface.\ntype MockContext struct {\n\tctrl     *gomock.Controller\n\trecorder *MockContextMockRecorder\n\tisgomock struct{}\n}\n\n// MockContextMockRecorder is the mock recorder for MockContext.\ntype MockContextMockRecorder struct {\n\tmock *MockContext\n}\n\n// NewMockContext creates a new mock instance.\nfunc NewMockContext(ctrl *gomock.Controller) *MockContext {\n\tmock := &MockContext{ctrl: ctrl}\n\tmock.recorder = &MockContextMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockContext) EXPECT() *MockContextMockRecorder {\n\treturn m.recorder\n}\n\n// AddingPendingFailoverMarker mocks base method.\nfunc (m *MockContext) AddingPendingFailoverMarker(arg0 *types.FailoverMarkerAttributes) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddingPendingFailoverMarker\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AddingPendingFailoverMarker indicates an expected call of AddingPendingFailoverMarker.\nfunc (mr *MockContextMockRecorder) AddingPendingFailoverMarker(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddingPendingFailoverMarker\", reflect.TypeOf((*MockContext)(nil).AddingPendingFailoverMarker), arg0)\n}\n\n// AppendHistoryV2Events mocks base method.\nfunc (m *MockContext) AppendHistoryV2Events(ctx context.Context, request *persistence.AppendHistoryNodesRequest, domainID string, execution types.WorkflowExecution) (*persistence.AppendHistoryNodesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AppendHistoryV2Events\", ctx, request, domainID, execution)\n\tret0, _ := ret[0].(*persistence.AppendHistoryNodesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AppendHistoryV2Events indicates an expected call of AppendHistoryV2Events.\nfunc (mr *MockContextMockRecorder) AppendHistoryV2Events(ctx, request, domainID, execution any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AppendHistoryV2Events\", reflect.TypeOf((*MockContext)(nil).AppendHistoryV2Events), ctx, request, domainID, execution)\n}\n\n// ConflictResolveWorkflowExecution mocks base method.\nfunc (m *MockContext) ConflictResolveWorkflowExecution(ctx context.Context, request *persistence.ConflictResolveWorkflowExecutionRequest) (*persistence.ConflictResolveWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ConflictResolveWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*persistence.ConflictResolveWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ConflictResolveWorkflowExecution indicates an expected call of ConflictResolveWorkflowExecution.\nfunc (mr *MockContextMockRecorder) ConflictResolveWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ConflictResolveWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).ConflictResolveWorkflowExecution), ctx, request)\n}\n\n// CreateWorkflowExecution mocks base method.\nfunc (m *MockContext) CreateWorkflowExecution(ctx context.Context, request *persistence.CreateWorkflowExecutionRequest) (*persistence.CreateWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*persistence.CreateWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateWorkflowExecution indicates an expected call of CreateWorkflowExecution.\nfunc (mr *MockContextMockRecorder) CreateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).CreateWorkflowExecution), ctx, request)\n}\n\n// DeleteFailoverLevel mocks base method.\nfunc (m *MockContext) DeleteFailoverLevel(category persistence.HistoryTaskCategory, failoverID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteFailoverLevel\", category, failoverID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteFailoverLevel indicates an expected call of DeleteFailoverLevel.\nfunc (mr *MockContextMockRecorder) DeleteFailoverLevel(category, failoverID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteFailoverLevel\", reflect.TypeOf((*MockContext)(nil).DeleteFailoverLevel), category, failoverID)\n}\n\n// GenerateTaskID mocks base method.\nfunc (m *MockContext) GenerateTaskID() (int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateTaskID\")\n\tret0, _ := ret[0].(int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GenerateTaskID indicates an expected call of GenerateTaskID.\nfunc (mr *MockContextMockRecorder) GenerateTaskID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateTaskID\", reflect.TypeOf((*MockContext)(nil).GenerateTaskID))\n}\n\n// GenerateTaskIDs mocks base method.\nfunc (m *MockContext) GenerateTaskIDs(number int) ([]int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GenerateTaskIDs\", number)\n\tret0, _ := ret[0].([]int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GenerateTaskIDs indicates an expected call of GenerateTaskIDs.\nfunc (mr *MockContextMockRecorder) GenerateTaskIDs(number any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GenerateTaskIDs\", reflect.TypeOf((*MockContext)(nil).GenerateTaskIDs), number)\n}\n\n// GetActiveClusterManager mocks base method.\nfunc (m *MockContext) GetActiveClusterManager() activecluster.Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetActiveClusterManager\")\n\tret0, _ := ret[0].(activecluster.Manager)\n\treturn ret0\n}\n\n// GetActiveClusterManager indicates an expected call of GetActiveClusterManager.\nfunc (mr *MockContextMockRecorder) GetActiveClusterManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetActiveClusterManager\", reflect.TypeOf((*MockContext)(nil).GetActiveClusterManager))\n}\n\n// GetAllFailoverLevels mocks base method.\nfunc (m *MockContext) GetAllFailoverLevels(category persistence.HistoryTaskCategory) map[string]persistence.FailoverLevel {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllFailoverLevels\", category)\n\tret0, _ := ret[0].(map[string]persistence.FailoverLevel)\n\treturn ret0\n}\n\n// GetAllFailoverLevels indicates an expected call of GetAllFailoverLevels.\nfunc (mr *MockContextMockRecorder) GetAllFailoverLevels(category any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllFailoverLevels\", reflect.TypeOf((*MockContext)(nil).GetAllFailoverLevels), category)\n}\n\n// GetClusterMetadata mocks base method.\nfunc (m *MockContext) GetClusterMetadata() cluster.Metadata {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetClusterMetadata\")\n\tret0, _ := ret[0].(cluster.Metadata)\n\treturn ret0\n}\n\n// GetClusterMetadata indicates an expected call of GetClusterMetadata.\nfunc (mr *MockContextMockRecorder) GetClusterMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetClusterMetadata\", reflect.TypeOf((*MockContext)(nil).GetClusterMetadata))\n}\n\n// GetConfig mocks base method.\nfunc (m *MockContext) GetConfig() *config.Config {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetConfig\")\n\tret0, _ := ret[0].(*config.Config)\n\treturn ret0\n}\n\n// GetConfig indicates an expected call of GetConfig.\nfunc (mr *MockContextMockRecorder) GetConfig() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetConfig\", reflect.TypeOf((*MockContext)(nil).GetConfig))\n}\n\n// GetCurrentTime mocks base method.\nfunc (m *MockContext) GetCurrentTime(cluster string) time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCurrentTime\", cluster)\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// GetCurrentTime indicates an expected call of GetCurrentTime.\nfunc (mr *MockContextMockRecorder) GetCurrentTime(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCurrentTime\", reflect.TypeOf((*MockContext)(nil).GetCurrentTime), cluster)\n}\n\n// GetDomainCache mocks base method.\nfunc (m *MockContext) GetDomainCache() cache.DomainCache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainCache\")\n\tret0, _ := ret[0].(cache.DomainCache)\n\treturn ret0\n}\n\n// GetDomainCache indicates an expected call of GetDomainCache.\nfunc (mr *MockContextMockRecorder) GetDomainCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainCache\", reflect.TypeOf((*MockContext)(nil).GetDomainCache))\n}\n\n// GetDomainNotificationVersion mocks base method.\nfunc (m *MockContext) GetDomainNotificationVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainNotificationVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetDomainNotificationVersion indicates an expected call of GetDomainNotificationVersion.\nfunc (mr *MockContextMockRecorder) GetDomainNotificationVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainNotificationVersion\", reflect.TypeOf((*MockContext)(nil).GetDomainNotificationVersion))\n}\n\n// GetEngine mocks base method.\nfunc (m *MockContext) GetEngine() engine.Engine {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEngine\")\n\tret0, _ := ret[0].(engine.Engine)\n\treturn ret0\n}\n\n// GetEngine indicates an expected call of GetEngine.\nfunc (mr *MockContextMockRecorder) GetEngine() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEngine\", reflect.TypeOf((*MockContext)(nil).GetEngine))\n}\n\n// GetEventsCache mocks base method.\nfunc (m *MockContext) GetEventsCache() events.Cache {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEventsCache\")\n\tret0, _ := ret[0].(events.Cache)\n\treturn ret0\n}\n\n// GetEventsCache indicates an expected call of GetEventsCache.\nfunc (mr *MockContextMockRecorder) GetEventsCache() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEventsCache\", reflect.TypeOf((*MockContext)(nil).GetEventsCache))\n}\n\n// GetExecutionManager mocks base method.\nfunc (m *MockContext) GetExecutionManager() persistence.ExecutionManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutionManager\")\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\treturn ret0\n}\n\n// GetExecutionManager indicates an expected call of GetExecutionManager.\nfunc (mr *MockContextMockRecorder) GetExecutionManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutionManager\", reflect.TypeOf((*MockContext)(nil).GetExecutionManager))\n}\n\n// GetHistoryManager mocks base method.\nfunc (m *MockContext) GetHistoryManager() persistence.HistoryManager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHistoryManager\")\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\treturn ret0\n}\n\n// GetHistoryManager indicates an expected call of GetHistoryManager.\nfunc (mr *MockContextMockRecorder) GetHistoryManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHistoryManager\", reflect.TypeOf((*MockContext)(nil).GetHistoryManager))\n}\n\n// GetLastUpdatedTime mocks base method.\nfunc (m *MockContext) GetLastUpdatedTime() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLastUpdatedTime\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// GetLastUpdatedTime indicates an expected call of GetLastUpdatedTime.\nfunc (mr *MockContextMockRecorder) GetLastUpdatedTime() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLastUpdatedTime\", reflect.TypeOf((*MockContext)(nil).GetLastUpdatedTime))\n}\n\n// GetLogger mocks base method.\nfunc (m *MockContext) GetLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetLogger indicates an expected call of GetLogger.\nfunc (mr *MockContextMockRecorder) GetLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetLogger\", reflect.TypeOf((*MockContext)(nil).GetLogger))\n}\n\n// GetMetricsClient mocks base method.\nfunc (m *MockContext) GetMetricsClient() metrics.Client {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetricsClient\")\n\tret0, _ := ret[0].(metrics.Client)\n\treturn ret0\n}\n\n// GetMetricsClient indicates an expected call of GetMetricsClient.\nfunc (mr *MockContextMockRecorder) GetMetricsClient() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetricsClient\", reflect.TypeOf((*MockContext)(nil).GetMetricsClient))\n}\n\n// GetQueueAckLevel mocks base method.\nfunc (m *MockContext) GetQueueAckLevel(category persistence.HistoryTaskCategory) persistence.HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueAckLevel\", category)\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\treturn ret0\n}\n\n// GetQueueAckLevel indicates an expected call of GetQueueAckLevel.\nfunc (mr *MockContextMockRecorder) GetQueueAckLevel(category any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueAckLevel\", reflect.TypeOf((*MockContext)(nil).GetQueueAckLevel), category)\n}\n\n// GetQueueClusterAckLevel mocks base method.\nfunc (m *MockContext) GetQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueClusterAckLevel\", category, cluster)\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\treturn ret0\n}\n\n// GetQueueClusterAckLevel indicates an expected call of GetQueueClusterAckLevel.\nfunc (mr *MockContextMockRecorder) GetQueueClusterAckLevel(category, cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueClusterAckLevel\", reflect.TypeOf((*MockContext)(nil).GetQueueClusterAckLevel), category, cluster)\n}\n\n// GetQueueState mocks base method.\nfunc (m *MockContext) GetQueueState(category persistence.HistoryTaskCategory) (*types.QueueState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueState\", category)\n\tret0, _ := ret[0].(*types.QueueState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetQueueState indicates an expected call of GetQueueState.\nfunc (mr *MockContextMockRecorder) GetQueueState(category any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueState\", reflect.TypeOf((*MockContext)(nil).GetQueueState), category)\n}\n\n// GetReplicationBudgetManager mocks base method.\nfunc (m *MockContext) GetReplicationBudgetManager() cache.Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetReplicationBudgetManager\")\n\tret0, _ := ret[0].(cache.Manager)\n\treturn ret0\n}\n\n// GetReplicationBudgetManager indicates an expected call of GetReplicationBudgetManager.\nfunc (mr *MockContextMockRecorder) GetReplicationBudgetManager() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetReplicationBudgetManager\", reflect.TypeOf((*MockContext)(nil).GetReplicationBudgetManager))\n}\n\n// GetService mocks base method.\nfunc (m *MockContext) GetService() resource.Resource {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetService\")\n\tret0, _ := ret[0].(resource.Resource)\n\treturn ret0\n}\n\n// GetService indicates an expected call of GetService.\nfunc (mr *MockContextMockRecorder) GetService() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetService\", reflect.TypeOf((*MockContext)(nil).GetService))\n}\n\n// GetShardID mocks base method.\nfunc (m *MockContext) GetShardID() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardID\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetShardID indicates an expected call of GetShardID.\nfunc (mr *MockContextMockRecorder) GetShardID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardID\", reflect.TypeOf((*MockContext)(nil).GetShardID))\n}\n\n// GetThrottledLogger mocks base method.\nfunc (m *MockContext) GetThrottledLogger() log.Logger {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetThrottledLogger\")\n\tret0, _ := ret[0].(log.Logger)\n\treturn ret0\n}\n\n// GetThrottledLogger indicates an expected call of GetThrottledLogger.\nfunc (mr *MockContextMockRecorder) GetThrottledLogger() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetThrottledLogger\", reflect.TypeOf((*MockContext)(nil).GetThrottledLogger))\n}\n\n// GetTimeSource mocks base method.\nfunc (m *MockContext) GetTimeSource() clock.TimeSource {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTimeSource\")\n\tret0, _ := ret[0].(clock.TimeSource)\n\treturn ret0\n}\n\n// GetTimeSource indicates an expected call of GetTimeSource.\nfunc (mr *MockContextMockRecorder) GetTimeSource() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTimeSource\", reflect.TypeOf((*MockContext)(nil).GetTimeSource))\n}\n\n// GetTimerProcessingQueueStates mocks base method.\nfunc (m *MockContext) GetTimerProcessingQueueStates(cluster string) []*types.ProcessingQueueState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTimerProcessingQueueStates\", cluster)\n\tret0, _ := ret[0].([]*types.ProcessingQueueState)\n\treturn ret0\n}\n\n// GetTimerProcessingQueueStates indicates an expected call of GetTimerProcessingQueueStates.\nfunc (mr *MockContextMockRecorder) GetTimerProcessingQueueStates(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTimerProcessingQueueStates\", reflect.TypeOf((*MockContext)(nil).GetTimerProcessingQueueStates), cluster)\n}\n\n// GetTransferProcessingQueueStates mocks base method.\nfunc (m *MockContext) GetTransferProcessingQueueStates(cluster string) []*types.ProcessingQueueState {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTransferProcessingQueueStates\", cluster)\n\tret0, _ := ret[0].([]*types.ProcessingQueueState)\n\treturn ret0\n}\n\n// GetTransferProcessingQueueStates indicates an expected call of GetTransferProcessingQueueStates.\nfunc (mr *MockContextMockRecorder) GetTransferProcessingQueueStates(cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTransferProcessingQueueStates\", reflect.TypeOf((*MockContext)(nil).GetTransferProcessingQueueStates), cluster)\n}\n\n// GetWorkflowExecution mocks base method.\nfunc (m *MockContext) GetWorkflowExecution(ctx context.Context, request *persistence.GetWorkflowExecutionRequest) (*persistence.GetWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*persistence.GetWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetWorkflowExecution indicates an expected call of GetWorkflowExecution.\nfunc (mr *MockContextMockRecorder) GetWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).GetWorkflowExecution), ctx, request)\n}\n\n// PreviousShardOwnerWasDifferent mocks base method.\nfunc (m *MockContext) PreviousShardOwnerWasDifferent() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PreviousShardOwnerWasDifferent\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// PreviousShardOwnerWasDifferent indicates an expected call of PreviousShardOwnerWasDifferent.\nfunc (mr *MockContextMockRecorder) PreviousShardOwnerWasDifferent() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PreviousShardOwnerWasDifferent\", reflect.TypeOf((*MockContext)(nil).PreviousShardOwnerWasDifferent))\n}\n\n// ReplicateFailoverMarkers mocks base method.\nfunc (m *MockContext) ReplicateFailoverMarkers(ctx context.Context, markers []*persistence.FailoverMarkerTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReplicateFailoverMarkers\", ctx, markers)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReplicateFailoverMarkers indicates an expected call of ReplicateFailoverMarkers.\nfunc (mr *MockContextMockRecorder) ReplicateFailoverMarkers(ctx, markers any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReplicateFailoverMarkers\", reflect.TypeOf((*MockContext)(nil).ReplicateFailoverMarkers), ctx, markers)\n}\n\n// SetCurrentTime mocks base method.\nfunc (m *MockContext) SetCurrentTime(cluster string, currentTime time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetCurrentTime\", cluster, currentTime)\n}\n\n// SetCurrentTime indicates an expected call of SetCurrentTime.\nfunc (mr *MockContextMockRecorder) SetCurrentTime(cluster, currentTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetCurrentTime\", reflect.TypeOf((*MockContext)(nil).SetCurrentTime), cluster, currentTime)\n}\n\n// SetEngine mocks base method.\nfunc (m *MockContext) SetEngine(arg0 engine.Engine) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetEngine\", arg0)\n}\n\n// SetEngine indicates an expected call of SetEngine.\nfunc (mr *MockContextMockRecorder) SetEngine(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetEngine\", reflect.TypeOf((*MockContext)(nil).SetEngine), arg0)\n}\n\n// UpdateDomainNotificationVersion mocks base method.\nfunc (m *MockContext) UpdateDomainNotificationVersion(domainNotificationVersion int64) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateDomainNotificationVersion\", domainNotificationVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateDomainNotificationVersion indicates an expected call of UpdateDomainNotificationVersion.\nfunc (mr *MockContextMockRecorder) UpdateDomainNotificationVersion(domainNotificationVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateDomainNotificationVersion\", reflect.TypeOf((*MockContext)(nil).UpdateDomainNotificationVersion), domainNotificationVersion)\n}\n\n// UpdateFailoverLevel mocks base method.\nfunc (m *MockContext) UpdateFailoverLevel(category persistence.HistoryTaskCategory, failoverID string, level persistence.FailoverLevel) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateFailoverLevel\", category, failoverID, level)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateFailoverLevel indicates an expected call of UpdateFailoverLevel.\nfunc (mr *MockContextMockRecorder) UpdateFailoverLevel(category, failoverID, level any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateFailoverLevel\", reflect.TypeOf((*MockContext)(nil).UpdateFailoverLevel), category, failoverID, level)\n}\n\n// UpdateIfNeededAndGetQueueMaxReadLevel mocks base method.\nfunc (m *MockContext) UpdateIfNeededAndGetQueueMaxReadLevel(category persistence.HistoryTaskCategory, cluster string) persistence.HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateIfNeededAndGetQueueMaxReadLevel\", category, cluster)\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\treturn ret0\n}\n\n// UpdateIfNeededAndGetQueueMaxReadLevel indicates an expected call of UpdateIfNeededAndGetQueueMaxReadLevel.\nfunc (mr *MockContextMockRecorder) UpdateIfNeededAndGetQueueMaxReadLevel(category, cluster any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateIfNeededAndGetQueueMaxReadLevel\", reflect.TypeOf((*MockContext)(nil).UpdateIfNeededAndGetQueueMaxReadLevel), category, cluster)\n}\n\n// UpdateQueueAckLevel mocks base method.\nfunc (m *MockContext) UpdateQueueAckLevel(category persistence.HistoryTaskCategory, ackLevel persistence.HistoryTaskKey) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueAckLevel\", category, ackLevel)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueAckLevel indicates an expected call of UpdateQueueAckLevel.\nfunc (mr *MockContextMockRecorder) UpdateQueueAckLevel(category, ackLevel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueAckLevel\", reflect.TypeOf((*MockContext)(nil).UpdateQueueAckLevel), category, ackLevel)\n}\n\n// UpdateQueueClusterAckLevel mocks base method.\nfunc (m *MockContext) UpdateQueueClusterAckLevel(category persistence.HistoryTaskCategory, cluster string, ackLevel persistence.HistoryTaskKey) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueClusterAckLevel\", category, cluster, ackLevel)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueClusterAckLevel indicates an expected call of UpdateQueueClusterAckLevel.\nfunc (mr *MockContextMockRecorder) UpdateQueueClusterAckLevel(category, cluster, ackLevel any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueClusterAckLevel\", reflect.TypeOf((*MockContext)(nil).UpdateQueueClusterAckLevel), category, cluster, ackLevel)\n}\n\n// UpdateQueueState mocks base method.\nfunc (m *MockContext) UpdateQueueState(category persistence.HistoryTaskCategory, state *types.QueueState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateQueueState\", category, state)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateQueueState indicates an expected call of UpdateQueueState.\nfunc (mr *MockContextMockRecorder) UpdateQueueState(category, state any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateQueueState\", reflect.TypeOf((*MockContext)(nil).UpdateQueueState), category, state)\n}\n\n// UpdateTimerProcessingQueueStates mocks base method.\nfunc (m *MockContext) UpdateTimerProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTimerProcessingQueueStates\", cluster, states)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTimerProcessingQueueStates indicates an expected call of UpdateTimerProcessingQueueStates.\nfunc (mr *MockContextMockRecorder) UpdateTimerProcessingQueueStates(cluster, states any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTimerProcessingQueueStates\", reflect.TypeOf((*MockContext)(nil).UpdateTimerProcessingQueueStates), cluster, states)\n}\n\n// UpdateTransferProcessingQueueStates mocks base method.\nfunc (m *MockContext) UpdateTransferProcessingQueueStates(cluster string, states []*types.ProcessingQueueState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTransferProcessingQueueStates\", cluster, states)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTransferProcessingQueueStates indicates an expected call of UpdateTransferProcessingQueueStates.\nfunc (mr *MockContextMockRecorder) UpdateTransferProcessingQueueStates(cluster, states any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTransferProcessingQueueStates\", reflect.TypeOf((*MockContext)(nil).UpdateTransferProcessingQueueStates), cluster, states)\n}\n\n// UpdateWorkflowExecution mocks base method.\nfunc (m *MockContext) UpdateWorkflowExecution(ctx context.Context, request *persistence.UpdateWorkflowExecutionRequest) (*persistence.UpdateWorkflowExecutionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateWorkflowExecution\", ctx, request)\n\tret0, _ := ret[0].(*persistence.UpdateWorkflowExecutionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateWorkflowExecution indicates an expected call of UpdateWorkflowExecution.\nfunc (mr *MockContextMockRecorder) UpdateWorkflowExecution(ctx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateWorkflowExecution\", reflect.TypeOf((*MockContext)(nil).UpdateWorkflowExecution), ctx, request)\n}\n\n// ValidateAndUpdateFailoverMarkers mocks base method.\nfunc (m *MockContext) ValidateAndUpdateFailoverMarkers() ([]*types.FailoverMarkerAttributes, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ValidateAndUpdateFailoverMarkers\")\n\tret0, _ := ret[0].([]*types.FailoverMarkerAttributes)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ValidateAndUpdateFailoverMarkers indicates an expected call of ValidateAndUpdateFailoverMarkers.\nfunc (mr *MockContextMockRecorder) ValidateAndUpdateFailoverMarkers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ValidateAndUpdateFailoverMarkers\", reflect.TypeOf((*MockContext)(nil).ValidateAndUpdateFailoverMarkers))\n}\n"
  },
  {
    "path": "service/history/shard/context_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shard\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"math\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/resource\"\n)\n\nconst (\n\ttestShardID                   = 123\n\ttestRangeID                   = 1\n\ttestTransferMaxReadLevel      = 10\n\ttestMaxTransferSequenceNumber = 100\n\ttestCluster                   = \"test-cluster\"\n\ttestDomain                    = \"test-domain\"\n\ttestDomainID                  = \"test-domain-id\"\n\ttestWorkflowID                = \"test-workflow-id\"\n)\n\ntype (\n\tcontextTestSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller       *gomock.Controller\n\t\tmockResource     *resource.Test\n\t\tmockShardManager *mocks.ShardManager\n\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\n\t\tcontext *contextImpl\n\t}\n)\n\nfunc TestContextSuite(t *testing.T) {\n\ts := new(contextTestSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *contextTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.History)\n\ts.mockShardManager = s.mockResource.ShardMgr\n\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.logger = testlogger.New(s.T())\n\n\ts.context = s.newContext()\n}\n\nfunc (s *contextTestSuite) newContext() *contextImpl {\n\teventsCache := events.NewMockCache(s.controller)\n\tconfig := config.NewForTest()\n\tshardInfo := &persistence.ShardInfo{\n\t\tShardID: testShardID,\n\t\tRangeID: testRangeID,\n\t\t// the following fields will be initialized\n\t\t// when acquiring the shard if they are nil\n\t\tClusterTransferAckLevel: make(map[string]int64),\n\t\tClusterTimerAckLevel:    make(map[string]time.Time),\n\t\tClusterReplicationLevel: make(map[string]int64),\n\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t},\n\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t},\n\t\tQueueStates: make(map[int32]*types.QueueState),\n\t}\n\tcontext := &contextImpl{\n\t\tResource:                     s.mockResource,\n\t\tshardID:                      shardInfo.ShardID,\n\t\trangeID:                      shardInfo.RangeID,\n\t\tshardInfo:                    shardInfo,\n\t\texecutionManager:             s.mockResource.ExecutionMgr,\n\t\tactiveClusterManager:         s.mockResource.ActiveClusterMgr,\n\t\tcloseCallback:                func(i int, item *historyShardsItem) {},\n\t\tconfig:                       config,\n\t\tlogger:                       s.logger,\n\t\tthrottledLogger:              s.logger,\n\t\ttaskSequenceNumber:           1,\n\t\timmediateTaskMaxReadLevel:    testTransferMaxReadLevel,\n\t\tmaxTaskSequenceNumber:        testMaxTransferSequenceNumber,\n\t\tscheduledTaskMaxReadLevelMap: make(map[string]time.Time),\n\t\tremoteClusterCurrentTime:     make(map[string]time.Time),\n\t\tfailoverLevels:               make(map[persistence.HistoryTaskCategory]map[string]persistence.FailoverLevel),\n\t\teventsCache:                  eventsCache,\n\t}\n\n\ts.Require().True(testMaxTransferSequenceNumber < (1<<context.config.RangeSizeBits), \"bad config value\")\n\n\treturn context\n}\n\nfunc (s *contextTestSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\n// test various setters and getters\nfunc (s *contextTestSuite) TestAccessorMethods() {\n\ts.Assert().EqualValues(testShardID, s.context.GetShardID())\n\ts.Assert().Equal(s.mockResource, s.context.GetService())\n\ts.Assert().Equal(s.mockResource.ExecutionMgr, s.context.GetExecutionManager())\n\ts.Assert().EqualValues(testTransferMaxReadLevel, s.context.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTransfer, cluster.TestCurrentClusterName).GetTaskID())\n\ts.Assert().Equal(s.logger, s.context.GetLogger())\n\ts.Assert().Equal(s.logger, s.context.GetThrottledLogger())\n\n\tmockEngine := engine.NewMockEngine(s.controller)\n\ts.context.SetEngine(mockEngine)\n\ts.Assert().Equal(mockEngine, s.context.GetEngine())\n}\n\nfunc (s *contextTestSuite) TestTransferQueueState() {\n\ts.context.shardInfo.TransferAckLevel = 5\n\tqueueState, err := s.context.GetQueueState(persistence.HistoryTaskCategoryTransfer)\n\ts.NoError(err)\n\ts.EqualValues(6, queueState.ExclusiveMaxReadLevel.TaskID)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\terr = s.context.UpdateQueueState(persistence.HistoryTaskCategoryTransfer, &types.QueueState{\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t0: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{TaskID: 7},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{TaskID: 10},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tExclusiveMaxReadLevel: &types.TaskKey{TaskID: 10},\n\t})\n\ts.NoError(err)\n\tqueueState, err = s.context.GetQueueState(persistence.HistoryTaskCategoryTransfer)\n\ts.NoError(err)\n\ts.EqualValues(10, queueState.ExclusiveMaxReadLevel.TaskID)\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n\n\ts.EqualValues(6, s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTransfer).GetTaskID())\n\ts.EqualValues(6, s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, cluster.TestCurrentClusterName).GetTaskID())\n\ts.Equal([]*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(6),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}, s.context.shardInfo.TransferProcessingQueueStates.StatesByCluster[cluster.TestCurrentClusterName])\n}\n\nfunc (s *contextTestSuite) TestTimerQueueState() {\n\tnow := time.Now()\n\ts.context.shardInfo.TimerAckLevel = now\n\tqueueState, err := s.context.GetQueueState(persistence.HistoryTaskCategoryTimer)\n\ts.NoError(err)\n\ts.Equal(now.UnixNano(), queueState.ExclusiveMaxReadLevel.ScheduledTimeNano)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\terr = s.context.UpdateQueueState(persistence.HistoryTaskCategoryTimer, &types.QueueState{\n\t\tVirtualQueueStates: map[int64]*types.VirtualQueueState{\n\t\t\t0: {\n\t\t\t\tVirtualSliceStates: []*types.VirtualSliceState{\n\t\t\t\t\t{\n\t\t\t\t\t\tTaskRange: &types.TaskRange{\n\t\t\t\t\t\t\tInclusiveMin: &types.TaskKey{ScheduledTimeNano: now.Add(time.Second).UnixNano()},\n\t\t\t\t\t\t\tExclusiveMax: &types.TaskKey{ScheduledTimeNano: now.Add(time.Second * 2).UnixNano()},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tExclusiveMaxReadLevel: &types.TaskKey{ScheduledTimeNano: now.Add(time.Second * 2).UnixNano()},\n\t})\n\ts.NoError(err)\n\tqueueState, err = s.context.GetQueueState(persistence.HistoryTaskCategoryTimer)\n\ts.NoError(err)\n\ts.Equal(now.Add(time.Second*2).UnixNano(), queueState.ExclusiveMaxReadLevel.ScheduledTimeNano)\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n\n\ts.EqualValues(now.Add(time.Second).UnixNano(), s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTimer).GetScheduledTime().UnixNano())\n\ts.EqualValues(now.Add(time.Second).UnixNano(), s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, cluster.TestCurrentClusterName).GetScheduledTime().UnixNano())\n\ts.Equal([]*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(now.Add(time.Second).UnixNano()),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}, s.context.shardInfo.TimerProcessingQueueStates.StatesByCluster[cluster.TestCurrentClusterName])\n}\n\nfunc (s *contextTestSuite) TestTransferAckLevel() {\n\t// validate default value returned\n\ts.context.shardInfo.TransferAckLevel = 5\n\ts.Assert().EqualValues(5, s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTransfer).GetTaskID())\n\n\t// update and validate it's returned\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\ts.context.UpdateQueueAckLevel(persistence.HistoryTaskCategoryTransfer, persistence.NewImmediateTaskKey(20))\n\ts.Assert().EqualValues(20, s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTransfer).GetTaskID())\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n}\n\nfunc (s *contextTestSuite) TestClusterTransferAckLevel() {\n\t// update and validate cluster transfer ack level\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\ts.context.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, cluster.TestCurrentClusterName, persistence.NewImmediateTaskKey(5))\n\ts.Assert().EqualValues(5, s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, cluster.TestCurrentClusterName).GetTaskID())\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n\n\t// get cluster transfer ack level for non existing cluster\n\ts.context.shardInfo.TransferAckLevel = 10\n\ts.Assert().EqualValues(10, s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, \"non-existing-cluster\").GetTaskID())\n}\n\nfunc (s *contextTestSuite) TestTimerAckLevel() {\n\t// validate default value returned\n\tnow := time.Now()\n\ts.context.shardInfo.TimerAckLevel = now\n\ts.Assert().Equal(now.UnixNano(), s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTimer).GetScheduledTime().UnixNano())\n\n\t// update and validate it's returned\n\tnewTime := time.Now()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\ts.context.UpdateQueueAckLevel(persistence.HistoryTaskCategoryTimer, persistence.NewHistoryTaskKey(newTime, 0))\n\ts.Assert().EqualValues(newTime.UnixNano(), s.context.GetQueueAckLevel(persistence.HistoryTaskCategoryTimer).GetScheduledTime().UnixNano())\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n}\n\nfunc (s *contextTestSuite) TestClusterTimerAckLevel() {\n\t// update and validate cluster timer ack level\n\tnow := time.Now()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\ts.context.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, cluster.TestCurrentClusterName, persistence.NewHistoryTaskKey(now, 0))\n\ts.Assert().EqualValues(now.UnixNano(), s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, cluster.TestCurrentClusterName).GetScheduledTime().UnixNano())\n\ts.Assert().Equal(0, s.context.shardInfo.StolenSinceRenew)\n\n\t// get cluster timer ack level for non existing cluster\n\ts.context.shardInfo.TimerAckLevel = now\n\ts.Assert().EqualValues(now.UnixNano(), s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, \"non-existing-cluster\").GetScheduledTime().UnixNano())\n}\n\nfunc (s *contextTestSuite) TestUpdateTransferFailoverLevel() {\n\tfailoverLevel1 := persistence.FailoverLevel{\n\t\tStartTime:    time.Now(),\n\t\tMinLevel:     persistence.NewImmediateTaskKey(1),\n\t\tCurrentLevel: persistence.NewImmediateTaskKey(10),\n\t\tMaxLevel:     persistence.NewImmediateTaskKey(100),\n\t\tDomainIDs:    map[string]struct{}{\"testDomainID\": {}},\n\t}\n\tfailoverLevel2 := persistence.FailoverLevel{\n\t\tStartTime:    time.Now(),\n\t\tMinLevel:     persistence.NewImmediateTaskKey(2),\n\t\tCurrentLevel: persistence.NewImmediateTaskKey(20),\n\t\tMaxLevel:     persistence.NewImmediateTaskKey(200),\n\t\tDomainIDs:    map[string]struct{}{\"testDomainID2\": {}},\n\t}\n\n\terr := s.context.UpdateFailoverLevel(persistence.HistoryTaskCategoryTransfer, \"id1\", failoverLevel1)\n\ts.NoError(err)\n\terr = s.context.UpdateFailoverLevel(persistence.HistoryTaskCategoryTransfer, \"id2\", failoverLevel2)\n\ts.NoError(err)\n\n\tgotLevels := s.context.GetAllFailoverLevels(persistence.HistoryTaskCategoryTransfer)\n\ts.Len(gotLevels, 2)\n\tassert.Equal(s.T(), failoverLevel1, gotLevels[\"id1\"])\n\tassert.Equal(s.T(), failoverLevel2, gotLevels[\"id2\"])\n\n\terr = s.context.DeleteFailoverLevel(persistence.HistoryTaskCategoryTransfer, \"id1\")\n\ts.NoError(err)\n\tgotLevels = s.context.GetAllFailoverLevels(persistence.HistoryTaskCategoryTransfer)\n\ts.Len(gotLevels, 1)\n\tassert.Equal(s.T(), failoverLevel2, gotLevels[\"id2\"])\n}\n\nfunc (s *contextTestSuite) TestUpdateTimerFailoverLevel() {\n\tt := time.Now()\n\tfailoverLevel1 := persistence.FailoverLevel{\n\t\tStartTime:    t,\n\t\tMinLevel:     persistence.NewHistoryTaskKey(t.Add(time.Minute), 0),\n\t\tCurrentLevel: persistence.NewHistoryTaskKey(t.Add(time.Minute*2), 0),\n\t\tMaxLevel:     persistence.NewHistoryTaskKey(t.Add(time.Minute*3), 0),\n\t\tDomainIDs:    map[string]struct{}{\"testDomainID\": {}},\n\t}\n\tfailoverLevel2 := persistence.FailoverLevel{\n\t\tStartTime:    t,\n\t\tMinLevel:     persistence.NewHistoryTaskKey(t.Add(time.Minute*2), 0),\n\t\tCurrentLevel: persistence.NewHistoryTaskKey(t.Add(time.Minute*4), 0),\n\t\tMaxLevel:     persistence.NewHistoryTaskKey(t.Add(time.Minute*6), 0),\n\t\tDomainIDs:    map[string]struct{}{\"testDomainID2\": {}},\n\t}\n\n\terr := s.context.UpdateFailoverLevel(persistence.HistoryTaskCategoryTimer, \"id1\", failoverLevel1)\n\ts.NoError(err)\n\terr = s.context.UpdateFailoverLevel(persistence.HistoryTaskCategoryTimer, \"id2\", failoverLevel2)\n\ts.NoError(err)\n\n\tgotLevels := s.context.GetAllFailoverLevels(persistence.HistoryTaskCategoryTimer)\n\ts.Len(gotLevels, 2)\n\tassert.Equal(s.T(), failoverLevel1, gotLevels[\"id1\"])\n\tassert.Equal(s.T(), failoverLevel2, gotLevels[\"id2\"])\n\n\terr = s.context.DeleteFailoverLevel(persistence.HistoryTaskCategoryTimer, \"id1\")\n\ts.NoError(err)\n\tgotLevels = s.context.GetAllFailoverLevels(persistence.HistoryTaskCategoryTimer)\n\ts.Len(gotLevels, 1)\n\tassert.Equal(s.T(), failoverLevel2, gotLevels[\"id2\"])\n}\n\nfunc (s *contextTestSuite) TestDomainNotificationVersion() {\n\t// test initial value\n\ts.EqualValues(0, s.context.GetDomainNotificationVersion())\n\n\t// test updated value\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\terr := s.context.UpdateDomainNotificationVersion(10)\n\ts.NoError(err)\n\ts.EqualValues(10, s.context.GetDomainNotificationVersion())\n}\n\nfunc (s *contextTestSuite) TestTimerMaxReadLevel() {\n\t// this test requires mocked time since we're comparing timestamps\n\ts.mockResource.TimeSource = clock.NewMockedTimeSource()\n\n\t// get current cluster's level\n\tgotLevel := s.context.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, cluster.TestCurrentClusterName)\n\twantLevel := persistence.NewHistoryTaskKey(s.mockResource.TimeSource.Now().Add(s.context.config.TimerProcessorMaxTimeShift()).Truncate(persistence.DBTimestampMinPrecision), 0)\n\ts.Equal(wantLevel, gotLevel)\n\n\t// get remote cluster's level\n\tremoteCluster := \"remote-cluster\"\n\tnow := time.Now()\n\ts.context.SetCurrentTime(remoteCluster, now)\n\tgotLevel = s.context.UpdateIfNeededAndGetQueueMaxReadLevel(persistence.HistoryTaskCategoryTimer, remoteCluster)\n\twantLevel = persistence.NewHistoryTaskKey(now.Add(s.context.config.TimerProcessorMaxTimeShift()).Truncate(persistence.DBTimestampMinPrecision), 0)\n\ts.Equal(wantLevel, gotLevel)\n}\n\nfunc (s *contextTestSuite) TestGenerateTransferTaskID() {\n\ttaskID, err := s.context.GenerateTaskID()\n\ts.Require().NoError(err)\n\ts.Assert().Equal(int64(1), taskID)\n\n\ttaskID, err = s.context.GenerateTaskID()\n\ts.Require().NoError(err)\n\ts.Assert().Equal(int64(2), taskID)\n}\n\nfunc (s *contextTestSuite) TestGenerateTransferTaskIDs() {\n\texpectedTaskIDs := []int64{1, 2, 3, 4}\n\n\ttaskIDs, err := s.context.GenerateTaskIDs(4)\n\ts.Require().NoError(err)\n\ts.Assert().Equal(expectedTaskIDs, taskIDs)\n}\n\nfunc (s *contextTestSuite) TestGenerateTransferTaskID_RenewsRange() {\n\t// we acquire task IDs until testMaxTransferSequenceNumber, then next generation should involve\n\t// renewing range\n\t_, err := s.context.GenerateTaskIDs(testMaxTransferSequenceNumber - 1)\n\ts.Require().NoError(err)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\n\ttaskID, err := s.context.GenerateTaskID()\n\ts.Require().NoError(err)\n\n\tnewRangeID := testRangeID + 1\n\ts.Assert().EqualValues(newRangeID, s.context.getRangeID(), \"RangeID should be incremented when renewing range\")\n\texpectedTransferTaskID := newRangeID << s.context.config.RangeSizeBits\n\n\ts.Assert().EqualValues(expectedTransferTaskID, taskID)\n}\n\nfunc (s *contextTestSuite) TestRenewRangeLockedRetriesExceeded() {\n\tsomeError := errors.New(\"some error\")\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(someError)\n\n\terr := s.context.renewRangeLocked(false)\n\ts.Error(err)\n}\n\nfunc (s *contextTestSuite) TestUpdateClusterReplicationLevel_Succeeds() {\n\tlastTaskID := int64(123)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Once().Return(nil)\n\terr := s.context.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, testCluster, persistence.NewImmediateTaskKey(lastTaskID))\n\ts.Require().NoError(err)\n\n\ts.Equal(lastTaskID, s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, testCluster).GetTaskID())\n}\n\nfunc (s *contextTestSuite) TestUpdateClusterReplicationLevel_FailsWhenUpdateShardFail() {\n\townershipLostError := &persistence.ShardOwnershipLostError{ShardID: testShardID, Msg: \"testing ownership lost\"}\n\tshardClosed := false\n\tcloseCallbackCalled := make(chan bool)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(ownershipLostError)\n\ts.context.closeCallback = func(int, *historyShardsItem) {\n\t\tshardClosed = true\n\t\tcloseCallbackCalled <- true\n\t}\n\n\terr := s.context.UpdateQueueClusterAckLevel(persistence.HistoryTaskCategoryReplication, testCluster, persistence.NewImmediateTaskKey(123))\n\n\tselect {\n\tcase <-closeCallbackCalled:\n\t\tbreak\n\tcase <-time.NewTimer(time.Second).C:\n\t\ts.T().Fatal(\"close callback is still not called\")\n\t}\n\n\ts.Require().ErrorContains(err, ownershipLostError.Msg)\n\ts.True(shardClosed, \"the shard should have been closed on ShardOwnershipLostError\")\n}\n\nfunc (s *contextTestSuite) TestReplicateFailoverMarkers() {\n\tcases := []struct {\n\t\tname    string\n\t\tmarkers []*persistence.FailoverMarkerTask\n\t\terr     error\n\t\tasserts func()\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tmarkers: []*persistence.FailoverMarkerTask{{\n\t\t\t\tTaskData: persistence.TaskData{},\n\t\t\t\tDomainID: testDomainID,\n\t\t\t}},\n\t\t\tasserts: func() {\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Shard ownership lost error\",\n\t\t\terr:  &persistence.ShardOwnershipLostError{},\n\t\t\tasserts: func() {\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error\",\n\t\t\terr:  assert.AnError,\n\t\t\tasserts: func() {\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Need setup the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\ts.mockResource.ExecutionMgr.On(\"CreateFailoverMarkerTasks\", mock.Anything, mock.Anything).Once().Return(tc.err)\n\n\t\t\terr := s.context.ReplicateFailoverMarkers(context.Background(), tc.markers)\n\t\t\ts.Equal(tc.err, err)\n\t\t})\n\t}\n}\n\nfunc (s *contextTestSuite) TestCreateWorkflowExecution() {\n\tcases := []struct {\n\t\tname            string\n\t\terr             error\n\t\tdomainLookupErr error\n\t\tresponse        *persistence.CreateWorkflowExecutionResponse\n\t\tsetup           func()\n\t\tasserts         func(*persistence.CreateWorkflowExecutionResponse, error)\n\t}{\n\t\t{\n\t\t\tname:     \"Success\",\n\t\t\tresponse: &persistence.CreateWorkflowExecutionResponse{},\n\t\t\tasserts: func(response *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"No special handling\",\n\t\t\terr:  &types.WorkflowExecutionAlreadyStartedError{},\n\t\t\tasserts: func(resp *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &types.WorkflowExecutionAlreadyStartedError{})\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Shard ownership lost error\",\n\t\t\terr:  &persistence.ShardOwnershipLostError{},\n\t\t\tasserts: func(resp *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &persistence.ShardOwnershipLostError{})\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard succeed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard failed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(assert.AnError)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"Domain lookup failed\",\n\t\t\tdomainLookupErr: assert.AnError,\n\t\t\tasserts: func(resp *persistence.CreateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.ErrorIs(err, assert.AnError)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Need setup the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\tctx := context.Background()\n\t\t\trequest := &persistence.CreateWorkflowExecutionRequest{\n\t\t\t\tDomainName: testDomain,\n\t\t\t\tNewWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tdomainCacheEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t\t\t&persistence.DomainConfig{Retention: 7},\n\t\t\t\ttestCluster,\n\t\t\t)\n\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntry, tc.domainLookupErr)\n\t\t\tif tc.setup != nil {\n\t\t\t\ttc.setup()\n\t\t\t}\n\n\t\t\ts.mockResource.ExecutionMgr.On(\"CreateWorkflowExecution\", ctx, mock.Anything).Once().Return(tc.response, tc.err)\n\n\t\t\tresp, err := s.context.CreateWorkflowExecution(ctx, request)\n\t\t\ttc.asserts(resp, err)\n\t\t})\n\t}\n}\n\nfunc (s *contextTestSuite) TestUpdateWorkflowExecution() {\n\tcases := []struct {\n\t\tname            string\n\t\terr             error\n\t\tdomainLookupErr error\n\t\tresponse        *persistence.UpdateWorkflowExecutionResponse\n\t\tsetup           func()\n\t\tasserts         func(*persistence.UpdateWorkflowExecutionResponse, error)\n\t}{\n\t\t{\n\t\t\tname:     \"Success\",\n\t\t\tresponse: &persistence.UpdateWorkflowExecutionResponse{},\n\t\t\tasserts: func(response *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"No special handling\",\n\t\t\terr:  &types.ServiceBusyError{},\n\t\t\tasserts: func(resp *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &types.ServiceBusyError{})\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Shard ownership lost error\",\n\t\t\terr:  &persistence.ShardOwnershipLostError{},\n\t\t\tasserts: func(resp *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &persistence.ShardOwnershipLostError{})\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard succeed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard failed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(assert.AnError)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"Domain lookup failed\",\n\t\t\tdomainLookupErr: assert.AnError,\n\t\t\tasserts: func(resp *persistence.UpdateWorkflowExecutionResponse, err error) {\n\t\t\t\ts.ErrorIs(err, assert.AnError)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Need setup the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\tctx := context.Background()\n\t\t\trequest := &persistence.UpdateWorkflowExecutionRequest{\n\t\t\t\tRangeID: 123,\n\t\t\t\tMode:    persistence.UpdateWorkflowModeUpdateCurrent,\n\t\t\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot: &persistence.WorkflowSnapshot{},\n\t\t\t\tDomainName:          testDomain,\n\t\t\t}\n\n\t\t\tdomainCacheEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t\t\t&persistence.DomainConfig{Retention: 7},\n\t\t\t\ttestCluster,\n\t\t\t)\n\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntry, tc.domainLookupErr)\n\t\t\tif tc.setup != nil {\n\t\t\t\ttc.setup()\n\t\t\t}\n\n\t\t\ts.mockResource.ExecutionMgr.On(\"UpdateWorkflowExecution\", ctx, mock.Anything).Once().Return(tc.response, tc.err)\n\n\t\t\tresp, err := s.context.UpdateWorkflowExecution(ctx, request)\n\t\t\ttc.asserts(resp, err)\n\t\t})\n\t}\n}\n\nfunc (s *contextTestSuite) TestConflictResolveWorkflowExecution() {\n\tcases := []struct {\n\t\tname            string\n\t\terr             error\n\t\tdomainLookupErr error\n\t\tresponse        *persistence.ConflictResolveWorkflowExecutionResponse\n\t\tsetup           func()\n\t\tasserts         func(*persistence.ConflictResolveWorkflowExecutionResponse, error)\n\t}{\n\t\t{\n\t\t\tname:     \"Success\",\n\t\t\tresponse: &persistence.ConflictResolveWorkflowExecutionResponse{},\n\t\t\tasserts: func(response *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"No special handling\",\n\t\t\terr:  &types.ServiceBusyError{},\n\t\t\tasserts: func(resp *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &types.ServiceBusyError{})\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Shard ownership lost error\",\n\t\t\terr:  &persistence.ShardOwnershipLostError{},\n\t\t\tasserts: func(resp *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(err, &persistence.ShardOwnershipLostError{})\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard succeed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.NoError(s.context.closedError())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Other error - update shard failed\",\n\t\t\terr:  assert.AnError,\n\t\t\tsetup: func() {\n\t\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(assert.AnError)\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.Equal(assert.AnError, err)\n\t\t\t\ts.ErrorContains(s.context.closedError(), \"shard closed\")\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"Domain lookup failed\",\n\t\t\tdomainLookupErr: assert.AnError,\n\t\t\tasserts: func(resp *persistence.ConflictResolveWorkflowExecutionResponse, err error) {\n\t\t\t\ts.ErrorIs(err, assert.AnError)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Need setup the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\tctx := context.Background()\n\t\t\trequest := &persistence.ConflictResolveWorkflowExecutionRequest{\n\t\t\t\tResetWorkflowSnapshot: persistence.WorkflowSnapshot{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tNewWorkflowSnapshot:     &persistence.WorkflowSnapshot{},\n\t\t\t\tCurrentWorkflowMutation: &persistence.WorkflowMutation{},\n\t\t\t\tDomainName:              testDomain,\n\t\t\t}\n\n\t\t\tdomainCacheEntry := cache.NewLocalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t\t\t&persistence.DomainConfig{Retention: 7},\n\t\t\t\ttestCluster,\n\t\t\t)\n\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntry, tc.domainLookupErr)\n\t\t\tif tc.setup != nil {\n\t\t\t\ttc.setup()\n\t\t\t}\n\n\t\t\ts.mockResource.ExecutionMgr.On(\"ConflictResolveWorkflowExecution\", ctx, mock.Anything).Once().Return(tc.response, tc.err)\n\n\t\t\tresp, err := s.context.ConflictResolveWorkflowExecution(ctx, request)\n\t\t\ttc.asserts(resp, err)\n\t\t})\n\t}\n}\n\nfunc (s *contextTestSuite) TestAppendHistoryV2Events() {\n\tcases := []struct {\n\t\tname            string\n\t\terr             error\n\t\tdomainLookupErr error\n\t\tresponse        *persistence.AppendHistoryNodesResponse\n\t\tsetup           func()\n\t\tasserts         func(*persistence.AppendHistoryNodesResponse, error)\n\t}{\n\t\t{\n\t\t\tname:     \"Success\",\n\t\t\tresponse: &persistence.AppendHistoryNodesResponse{},\n\t\t\tasserts: func(response *persistence.AppendHistoryNodesResponse, err error) {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(response)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"Domain lookup failed\",\n\t\t\tdomainLookupErr: assert.AnError,\n\t\t\tasserts: func(resp *persistence.AppendHistoryNodesResponse, err error) {\n\t\t\t\ts.ErrorIs(err, assert.AnError)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"History too big\",\n\t\t\tresponse: &persistence.AppendHistoryNodesResponse{\n\t\t\t\tDataBlob: persistence.DataBlob{\n\t\t\t\t\tData: make([]byte, historySizeLogThreshold+1),\n\t\t\t\t},\n\t\t\t},\n\t\t\tasserts: func(resp *persistence.AppendHistoryNodesResponse, err error) {\n\t\t\t\t// We do not err on history too big here, we just log it\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(resp)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\ts.Run(tc.name, func() {\n\t\t\t// Need setup the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\tctx := context.Background()\n\t\t\trequest := &persistence.AppendHistoryNodesRequest{}\n\t\t\tworkflowExecution := types.WorkflowExecution{\n\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\tRunID:      testWorkflowID,\n\t\t\t}\n\n\t\t\ts.mockResource.DomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomain, tc.domainLookupErr)\n\t\t\tif tc.setup != nil {\n\t\t\t\ttc.setup()\n\t\t\t}\n\n\t\t\ts.mockResource.HistoryMgr.On(\"AppendHistoryNodes\", ctx, mock.Anything).Once().Return(tc.response, tc.err)\n\n\t\t\tresp, err := s.context.AppendHistoryV2Events(ctx, request, testDomainID, workflowExecution)\n\t\t\ttc.asserts(resp, err)\n\t\t})\n\t}\n}\n\nfunc (s *contextTestSuite) TestValidateAndUpdateFailoverMarkers() {\n\t// This test verifies that failover markers are processed when a domain becomes active\n\tdomainFailoverVersion := 100\n\tdomainCacheEntryInactiveCluster := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t&persistence.DomainConfig{Retention: 7},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName, // active is TestCurrentClusterName\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tint64(domainFailoverVersion),\n\t)\n\tdomainCacheEntryActiveCluster := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t&persistence.DomainConfig{Retention: 7},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName, // active cluster\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tint64(domainFailoverVersion),\n\t)\n\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntryInactiveCluster, nil)\n\n\tfailoverMarker := types.FailoverMarkerAttributes{\n\t\tDomainID:        testDomainID,\n\t\tFailoverVersion: 101,\n\t}\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\ts.NoError(s.context.AddingPendingFailoverMarker(&failoverMarker))\n\ts.Require().Len(s.context.shardInfo.PendingFailoverMarkers, 1, \"we should have one failover marker saved since the cluster is not active\")\n\n\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntryActiveCluster, nil)\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\n\tpendingFailoverMarkers, err := s.context.ValidateAndUpdateFailoverMarkers()\n\ts.NoError(err)\n\ts.Empty(pendingFailoverMarkers, \"all pending failover tasks should be cleaned up\")\n}\n\nfunc (s *contextTestSuite) TestValidateAndUpdateFailoverMarkers_DeprecatedDomain() {\n\t// This test verifies that failover markers are dropped (not processed) when a domain is deprecated\n\tdomainFailoverVersion := 100\n\tdomainCacheEntryInactiveCluster := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t&persistence.DomainConfig{Retention: 7},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName, // active is TestCurrentClusterName\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tint64(domainFailoverVersion),\n\t)\n\tdomainCacheEntryActiveCluster := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: testDomainID},\n\t\t&persistence.DomainConfig{Retention: 7},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName, // active cluster\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\tint64(domainFailoverVersion),\n\t)\n\tdomainCacheEntryInactiveCluster.GetInfo().Status = persistence.DomainStatusDeprecated\n\tdomainCacheEntryActiveCluster.GetInfo().Status = persistence.DomainStatusDeprecated\n\n\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntryInactiveCluster, nil).Times(2)\n\n\tfailoverMarker := types.FailoverMarkerAttributes{\n\t\tDomainID:        testDomainID,\n\t\tFailoverVersion: int64(domainFailoverVersion),\n\t}\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\n\t// adding failover marker\n\tdomainFailoverVersion++\n\ts.NoError(s.context.AddingPendingFailoverMarker(&failoverMarker))\n\ts.Require().Len(s.context.shardInfo.PendingFailoverMarkers, 1, \"we should have one failover marker saved since the cluster is not active\")\n\n\t// adding more failover markers\n\tdomainFailoverVersion++\n\ts.NoError(s.context.AddingPendingFailoverMarker(&failoverMarker))\n\n\ts.mockResource.DomainCache.EXPECT().GetDomainByID(testDomainID).Return(domainCacheEntryActiveCluster, nil).Times(2)\n\n\tpendingFailoverMarkers, err := s.context.ValidateAndUpdateFailoverMarkers()\n\ts.NoError(err)\n\ts.Empty(pendingFailoverMarkers, \"pending failover markers should be dropped when the domain is deprecated\")\n\ts.Empty(s.context.shardInfo.PendingFailoverMarkers, \"pending failover markers should be dropped from shard info when domain is deprecated\")\n}\n\nfunc (s *contextTestSuite) TestGetAndUpdateProcessingQueueStates() {\n\tclusterName := cluster.TestCurrentClusterName\n\tvar initialQueueStates [][]*types.ProcessingQueueState\n\tinitialQueueStates = append(initialQueueStates, s.context.GetTransferProcessingQueueStates(clusterName))\n\tinitialQueueStates = append(initialQueueStates, s.context.GetTimerProcessingQueueStates(clusterName))\n\tfor _, queueStates := range initialQueueStates {\n\t\ts.Len(queueStates, 1)\n\t\ts.Zero(queueStates[0].GetLevel())\n\t\tackLevel := queueStates[0].GetAckLevel()\n\t\tif ackLevel != 0 {\n\t\t\t// for timer queue\n\t\t\ts.Equal(time.Time{}.UnixNano(), ackLevel)\n\t\t}\n\t\ts.Equal(int64(math.MaxInt64), queueStates[0].GetMaxLevel())\n\t\ts.Equal(&types.DomainFilter{ReverseMatch: true}, queueStates[0].GetDomainFilter())\n\t}\n\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(nil)\n\tupdatedTransferQueueStates := []*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(123),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}\n\tupdatedTimerQueueStates := []*types.ProcessingQueueState{\n\t\t{\n\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\tAckLevel: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\tMaxLevel: common.Int64Ptr(math.MaxInt64),\n\t\t\tDomainFilter: &types.DomainFilter{\n\t\t\t\tReverseMatch: true,\n\t\t\t},\n\t\t},\n\t}\n\terr := s.context.UpdateTransferProcessingQueueStates(clusterName, updatedTransferQueueStates)\n\ts.NoError(err)\n\terr = s.context.UpdateTimerProcessingQueueStates(clusterName, updatedTimerQueueStates)\n\ts.NoError(err)\n\n\ts.Equal(updatedTransferQueueStates, s.context.GetTransferProcessingQueueStates(clusterName))\n\ts.Equal(updatedTimerQueueStates, s.context.GetTimerProcessingQueueStates(clusterName))\n\n\t// check if cluster ack level for transfer and timer is backfilled for backward compatibility\n\ts.Equal(updatedTransferQueueStates[0].GetAckLevel(), s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTransfer, clusterName).GetTaskID())\n\ts.Equal(time.Unix(0, updatedTimerQueueStates[0].GetAckLevel()), s.context.GetQueueClusterAckLevel(persistence.HistoryTaskCategoryTimer, clusterName).GetScheduledTime())\n}\n\nfunc TestGetWorkflowExecution(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\trequest        *persistence.GetWorkflowExecutionRequest\n\t\tmockSetup      func(*mocks.ExecutionManager)\n\t\texpectedResult *persistence.GetWorkflowExecutionResponse\n\t\texpectedError  error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tDomainID:  \"testDomain\",\n\t\t\t\tExecution: types.WorkflowExecution{WorkflowID: \"testWorkflowID\", RunID: \"testRunID\"},\n\t\t\t},\n\t\t\tmockSetup: func(mgr *mocks.ExecutionManager) {\n\t\t\t\tmgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:   \"testDomain\",\n\t\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: &persistence.GetWorkflowExecutionResponse{\n\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\tDomainID:   \"testDomain\",\n\t\t\t\t\t\tWorkflowID: \"testWorkflowID\",\n\t\t\t\t\t\tRunID:      \"testRunID\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error\",\n\t\t\trequest: &persistence.GetWorkflowExecutionRequest{\n\t\t\t\tDomainID:  \"testDomain\",\n\t\t\t\tExecution: types.WorkflowExecution{WorkflowID: \"testWorkflowID\", RunID: \"testRunID\"},\n\t\t\t},\n\t\t\tmockSetup: func(mgr *mocks.ExecutionManager) {\n\t\t\t\tmgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedError:  errors.New(\"some random error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tmockExecutionMgr := &mocks.ExecutionManager{}\n\t\tshardContext := &contextImpl{\n\t\t\texecutionManager: mockExecutionMgr,\n\t\t\tshardInfo: &persistence.ShardInfo{\n\t\t\t\tRangeID: 12,\n\t\t\t},\n\t\t}\n\t\ttc.mockSetup(mockExecutionMgr)\n\n\t\tresult, err := shardContext.GetWorkflowExecution(context.Background(), tc.request)\n\t\tassert.Equal(t, tc.expectedResult, result)\n\t\tassert.Equal(t, tc.expectedError, err)\n\t}\n}\n\nfunc TestCloseShard(t *testing.T) {\n\tcloseCallback := make(chan struct{})\n\n\tshardContext := &contextImpl{\n\t\tshardInfo: &persistence.ShardInfo{RangeID: 12},\n\t\tcloseCallback: func(i int, item *historyShardsItem) {\n\t\t\tclose(closeCallback)\n\t\t},\n\t\tlogger: log.NewNoop(),\n\t}\n\tshardContext.closeShard()\n\n\tselect {\n\tcase <-closeCallback:\n\tcase <-time.After(time.Second):\n\t\tassert.Fail(t, \"closeCallback not called\")\n\t}\n\n\tassert.WithinDuration(t, time.Now(), *shardContext.closedAt.Load(), time.Second)\n\tassert.Equal(t, int64(-1), shardContext.shardInfo.RangeID)\n}\n\nfunc TestCloseShard_AlreadyClosed(t *testing.T) {\n\tcloseTime := time.Unix(123, 456)\n\n\tshardContext := &contextImpl{\n\t\tcloseCallback: func(i int, item *historyShardsItem) {\n\t\t\tassert.Fail(t, \"closeCallback should not be called\")\n\t\t},\n\t}\n\tshardContext.closedAt.Store(&closeTime)\n\tshardContext.closeShard()\n\tassert.Equal(t, closeTime, *shardContext.closedAt.Load())\n}\n\nfunc TestShardClosedGuard(t *testing.T) {\n\tshardContext := &contextImpl{\n\t\tshardInfo: &persistence.ShardInfo{},\n\t}\n\n\ttestCases := []struct {\n\t\tname string\n\t\tcall func() error\n\t}{\n\t\t{\n\t\t\tname: \"GetWorkflowExecution\",\n\t\t\tcall: func() error {\n\t\t\t\t_, err := shardContext.GetWorkflowExecution(\n\t\t\t\t\tcontext.Background(),\n\t\t\t\t\t&persistence.GetWorkflowExecutionRequest{RangeID: 0},\n\t\t\t\t)\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CreateWorkflowExecution\",\n\t\t\tcall: func() error {\n\t\t\t\t_, err := shardContext.CreateWorkflowExecution(context.Background(), nil)\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateWorkflowExecution\",\n\t\t\tcall: func() error {\n\t\t\t\t_, err := shardContext.UpdateWorkflowExecution(context.Background(), nil)\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ConflictResolveWorkflowExecution\",\n\t\t\tcall: func() error {\n\t\t\t\t_, err := shardContext.ConflictResolveWorkflowExecution(context.Background(), nil)\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"AppendHistoryV2Events\",\n\t\t\tcall: func() error {\n\t\t\t\t_, err := shardContext.AppendHistoryV2Events(context.Background(), nil, \"\", types.WorkflowExecution{})\n\t\t\t\treturn err\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"renewRangeLocked\",\n\t\t\tcall: func() error {\n\t\t\t\treturn shardContext.renewRangeLocked(false)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"persistShardInfoLocked\",\n\t\t\tcall: func() error {\n\t\t\t\treturn shardContext.persistShardInfoLocked(false)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ReplicateFailoverMarkers\",\n\t\t\tcall: func() error {\n\t\t\t\treturn shardContext.ReplicateFailoverMarkers(context.Background(), nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tclosedAt := time.Unix(123, 456)\n\n\t\t\tshardContext.closedAt.Store(&closedAt)\n\t\t\terr := tc.call()\n\t\t\tvar shardClosedErr *ErrShardClosed\n\t\t\tassert.ErrorAs(t, err, &shardClosedErr)\n\t\t\tassert.Equal(t, closedAt, shardClosedErr.ClosedAt)\n\t\t\tassert.ErrorContains(t, err, \"shard closed\")\n\t\t})\n\t}\n}\n\n// setupAllocateTimerIDsTest creates common test setup for allocateTimerIDsLocked tests\nfunc (s *contextTestSuite) setupAllocateTimerIDsTest() *cache.DomainCacheEntry {\n\t// Create basic domain cache entry with sensible defaults\n\tdomainInfo := &persistence.DomainInfo{ID: testDomainID}\n\tdomainConfig := &persistence.DomainConfig{}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: testCluster,\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testCluster},\n\t\t},\n\t}\n\n\treturn cache.NewDomainCacheEntryForTest(\n\t\tdomainInfo,\n\t\tdomainConfig,\n\t\tfalse, // domainIsActiveActive - default to false\n\t\treplicationConfig,\n\t\t456, // failover version\n\t\tnil, // failover end time\n\t\t123, // failover notification version\n\t\t0,   // previous failover version\n\t\t1,   // notification version\n\t)\n}\n\ntype createMockTimerTaskParams struct {\n\tVersion    int64\n\tTimestamp  time.Time\n\tDomainID   string\n\tWorkflowID string\n\tRunID      string\n}\n\nfunc (s *contextTestSuite) createMockTimerTask(params createMockTimerTaskParams) *persistence.MockTask {\n\tmockTask := persistence.NewMockTask(s.controller)\n\n\t// Use variables to track changes made by the allocateTimerIDsLocked function\n\tvar taskID int64\n\tvar visibilityTimestamp = params.Timestamp\n\n\tmockTask.EXPECT().GetDomainID().Return(params.DomainID).AnyTimes()\n\tmockTask.EXPECT().GetWorkflowID().Return(params.WorkflowID).AnyTimes()\n\tmockTask.EXPECT().GetRunID().Return(params.RunID).AnyTimes()\n\tmockTask.EXPECT().GetVersion().Return(params.Version).AnyTimes()\n\tmockTask.EXPECT().GetTaskCategory().Return(persistence.HistoryTaskCategoryTimer).AnyTimes()\n\tmockTask.EXPECT().GetTaskType().Return(123).AnyTimes()\n\tmockTask.EXPECT().ByteSize().Return(uint64(100)).AnyTimes()\n\n\t// Mock GetTaskID to return the current task ID\n\tmockTask.EXPECT().GetTaskID().DoAndReturn(func() int64 {\n\t\treturn taskID\n\t}).AnyTimes()\n\n\t// Mock SetTaskID to update the task ID variable\n\tmockTask.EXPECT().SetTaskID(gomock.Any()).DoAndReturn(func(id int64) {\n\t\ttaskID = id\n\t}).AnyTimes()\n\n\t// Mock GetVisibilityTimestamp to return the current timestamp\n\tmockTask.EXPECT().GetVisibilityTimestamp().DoAndReturn(func() time.Time {\n\t\treturn visibilityTimestamp\n\t}).AnyTimes()\n\n\t// Mock SetVisibilityTimestamp to update the timestamp variable\n\tmockTask.EXPECT().SetVisibilityTimestamp(gomock.Any()).DoAndReturn(func(ts time.Time) {\n\t\tvisibilityTimestamp = ts\n\t}).AnyTimes()\n\n\t// Mock GetTaskKey to return the current key based on current values\n\tmockTask.EXPECT().GetTaskKey().DoAndReturn(func() persistence.HistoryTaskKey {\n\t\treturn persistence.NewHistoryTaskKey(visibilityTimestamp, taskID)\n\t}).AnyTimes()\n\n\treturn mockTask\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenNoTasksProvidedReturnsSuccessfully() {\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{})\n\n\ts.NoError(err)\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTaskHasEmptyVersionAllocatesTaskID() {\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\toriginalTaskID := task.GetTaskID()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\ts.NotEqual(originalTaskID, task.GetTaskID(), \"Task ID should have been updated\")\n\ts.True(task.GetTaskID() > 0, \"Task ID should be positive\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTaskHasVersionAllocatesNewTaskID() {\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\t// Enable timer queue v2\n\ts.context.config.EnableTimerQueueV2 = func(int) bool {\n\t\treturn true\n\t}\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\toriginalTaskID := task.GetTaskID()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\ts.NotEqual(originalTaskID, task.GetTaskID(), \"Task ID should have been updated\")\n\ts.True(task.GetTaskID() > 0, \"Task ID should be positive\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTimerQueueV2DisabledUsesReplicationConfigClusterName() {\n\tdomainInfo := &persistence.DomainInfo{ID: testDomainID}\n\tdomainConfig := &persistence.DomainConfig{}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active-cluster\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testCluster},\n\t\t},\n\t}\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\tdomainInfo, domainConfig, false, replicationConfig, 456, nil, 123, 0, 1,\n\t)\n\n\t// Disable timer queue v2\n\ts.context.config.EnableTimerQueueV2 = func(int) bool {\n\t\treturn false\n\t}\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\toriginalTaskID := task.GetTaskID()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\ts.NotEqual(originalTaskID, task.GetTaskID(), \"Task ID should have been updated\")\n\ts.True(task.GetTaskID() > 0, \"Task ID should be positive\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenDomainIsActiveActiveUsesClusterManagerLookup() {\n\n\t// Create active-active domain cache entry\n\tdomainInfo := &persistence.DomainInfo{ID: testDomainID}\n\tdomainConfig := &persistence.DomainConfig{}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active-cluster\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testCluster},\n\t\t},\n\t\tActiveClusters: &types.ActiveClusters{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\t\t\tFailoverVersion:   456,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\tdomainInfo, domainConfig, true, replicationConfig, 456, nil, 123, 0, 1,\n\t)\n\n\t// Disable timer queue v2\n\ts.context.config.EnableTimerQueueV2 = func(int) bool {\n\t\treturn false\n\t}\n\n\t// Setup active cluster manager mock\n\ts.mockResource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\tgomock.Any(), testDomainID, testWorkflowID, gomock.Any(),\n\t).Return(&types.ActiveClusterInfo{\n\t\tActiveClusterName: \"looked-up-cluster\",\n\t}, nil).Times(1)\n\n\t// Create task with non-empty version to trigger the lookup logic\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    123,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\toriginalTaskID := task.GetTaskID()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\ts.NotEqual(originalTaskID, task.GetTaskID(), \"Task ID should have been updated\")\n\ts.True(task.GetTaskID() > 0, \"Task ID should be positive\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTaskTimestampBeforeReadCursorAdjustsTimestamp() {\n\ts.mockResource.TimeSource = clock.NewMockedTimeSourceAt(time.Now())\n\ttestTimeNow := s.mockResource.TimeSource.Now()\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\t// Set up scheduled task max read level map with read cursor ahead of task timestamp\n\t// Use the actual current cluster name from cluster metadata\n\tcurrentCluster := s.context.GetClusterMetadata().GetCurrentClusterName()\n\treadCursor := testTimeNow.Add(time.Second)\n\ts.context.scheduledTaskMaxReadLevelMap[currentCluster] = readCursor\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  readCursor.Add(-time.Second), // before read cursor\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\n\t// Verify timestamp was adjusted to be after read cursor\n\ts.True(task.GetVisibilityTimestamp().After(readCursor),\n\t\t\"Task timestamp should be adjusted to be after read cursor\")\n\n\t// Verify it's the expected adjusted time (readCursor + DBTimestampMinPrecision)\n\texpectedTime := readCursor.Add(persistence.DBTimestampMinPrecision)\n\tactualTime := task.GetVisibilityTimestamp()\n\ts.Equal(expectedTime.Truncate(persistence.DBTimestampMinPrecision),\n\t\tactualTime.Truncate(persistence.DBTimestampMinPrecision),\n\t\t\"Adjusted timestamp should match expected adjusted time\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTaskTimestampBeforeNow() {\n\ts.mockResource.TimeSource = clock.NewMockedTimeSourceAt(time.Now())\n\ttestTimeNow := s.mockResource.TimeSource.Now()\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\t// Set up scheduled task max read level map with read cursor ahead of task timestamp\n\t// Use the actual current cluster name from cluster metadata\n\tcurrentCluster := s.context.GetClusterMetadata().GetCurrentClusterName()\n\treadCursor := testTimeNow.Add(-2 * time.Second) // read cursor is in the past\n\ts.context.scheduledTaskMaxReadLevelMap[currentCluster] = readCursor\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  readCursor.Add(-time.Second), // before now but after read cursor\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.NoError(err)\n\n\t// Verify timestamp was adjusted to be after read cursor\n\ts.True(task.GetVisibilityTimestamp().After(readCursor),\n\t\t\"Task timestamp should be adjusted to be after read cursor\")\n\n\t// Verify it's the expected adjusted time (readCursor + DBTimestampMinPrecision)\n\texpectedTime := testTimeNow.Add(persistence.DBTimestampMinPrecision)\n\tactualTime := task.GetVisibilityTimestamp()\n\ts.Equal(expectedTime.Truncate(persistence.DBTimestampMinPrecision),\n\t\tactualTime.Truncate(persistence.DBTimestampMinPrecision),\n\t\t\"Adjusted timestamp should match expected adjusted time\")\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenClusterManagerLookupFailsReturnsError() {\n\t// Create active-active domain cache entry\n\tdomainInfo := &persistence.DomainInfo{ID: testDomainID}\n\tdomainConfig := &persistence.DomainConfig{}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active-cluster\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testCluster},\n\t\t},\n\t\tActiveClusters: &types.ActiveClusters{\n\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\"region\": {\n\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\tActiveClusterName: \"active-cluster\",\n\t\t\t\t\t\t\tFailoverVersion:   456,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\tdomainInfo, domainConfig, true, replicationConfig, 456, nil, 123, 0, 1,\n\t)\n\n\t// Disable timer queue v2\n\ts.context.config.EnableTimerQueueV2 = func(int) bool {\n\t\treturn false\n\t}\n\n\t// Setup active cluster manager mock to return error\n\ts.mockResource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(\n\t\tgomock.Any(), testDomainID, testWorkflowID, gomock.Any(),\n\t).Return(nil, assert.AnError).Times(1)\n\n\t// Create task with non-empty version to trigger the lookup logic\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    123,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.Error(err)\n\ts.Equal(assert.AnError, err)\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenTaskIDGenerationFailsReturnsError() {\n\tdomainCacheEntry := s.setupAllocateTimerIDsTest()\n\n\t// Force task sequence number to exceed max to trigger error\n\toriginalTaskSequenceNumber := s.context.taskSequenceNumber\n\ts.context.taskSequenceNumber = s.context.maxTaskSequenceNumber + 1\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, mock.Anything).Return(assert.AnError)\n\tdefer func() {\n\t\ts.context.taskSequenceNumber = originalTaskSequenceNumber\n\t}()\n\n\ttask := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id\",\n\t})\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task})\n\n\ts.Error(err)\n\ts.Equal(assert.AnError, err)\n}\n\nfunc (s *contextTestSuite) TestAllocateTimerIDsLocked_WhenMultipleTasksProvidedAllocatesAllTaskIDs() {\n\n\t// Create domain cache entry for non-active-active domain\n\tdomainInfo := &persistence.DomainInfo{ID: testDomainID}\n\tdomainConfig := &persistence.DomainConfig{}\n\treplicationConfig := &persistence.DomainReplicationConfig{\n\t\tActiveClusterName: \"active-cluster\",\n\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t{ClusterName: testCluster},\n\t\t},\n\t}\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\tdomainInfo, domainConfig, false, replicationConfig, 456, nil, 123, 0, 1,\n\t)\n\n\t// Disable timer queue v2\n\ts.context.config.EnableTimerQueueV2 = func(int) bool {\n\t\treturn false\n\t}\n\n\ttask1 := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tVersion:    constants.EmptyVersion,\n\t\tTimestamp:  time.Now().Add(time.Hour),\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id-1\",\n\t})\n\ttask2 := s.createMockTimerTask(createMockTimerTaskParams{\n\t\tDomainID:   testDomainID,\n\t\tWorkflowID: testWorkflowID,\n\t\tRunID:      \"test-run-id-2\",\n\t\tVersion:    456,\n\t\tTimestamp:  time.Now().Add(2 * time.Hour),\n\t})\n\n\toriginalTaskID1 := task1.GetTaskID()\n\toriginalTaskID2 := task2.GetTaskID()\n\n\terr := s.context.allocateTimerIDsLocked(domainCacheEntry, testWorkflowID, []persistence.Task{task1, task2})\n\n\ts.NoError(err)\n\ts.NotEqual(originalTaskID1, task1.GetTaskID(), \"Task 1 ID should have been updated\")\n\ts.NotEqual(originalTaskID2, task2.GetTaskID(), \"Task 2 ID should have been updated\")\n\ts.True(task1.GetTaskID() > 0, \"Task 1 ID should be positive\")\n\ts.True(task2.GetTaskID() > 0, \"Task 2 ID should be positive\")\n}\n"
  },
  {
    "path": "service/history/shard/context_test_utils.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage shard\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/resource\"\n)\n\n// TestContext is a test implementation for shard Context interface\ntype TestContext struct {\n\t*contextImpl\n\n\tResource                        *resource.Test\n\tMockEventsCache                 *events.MockCache\n\tMockAddingPendingFailoverMarker func(*types.FailoverMarkerAttributes) error\n}\n\nvar _ Context = (*TestContext)(nil)\n\n// NewTestContext create a new shardContext for test\nfunc NewTestContext(\n\tt *testing.T,\n\tctrl *gomock.Controller,\n\tshardInfo *persistence.ShardInfo,\n\tconfig *config.Config,\n) *TestContext {\n\tresource := resource.NewTest(t, ctrl, metrics.History)\n\teventsCache := events.NewMockCache(ctrl)\n\tshardInfo = shardInfo.ToNilSafeCopy()\n\tshardInfo.ClusterTransferAckLevel = map[string]int64{resource.ClusterMetadata.GetCurrentClusterName(): 3, \"standby\": 2}\n\n\tshard := &contextImpl{\n\t\tResource:                     resource,\n\t\tshardID:                      shardInfo.ShardID,\n\t\trangeID:                      shardInfo.RangeID,\n\t\tshardInfo:                    shardInfo,\n\t\texecutionManager:             resource.ExecutionMgr,\n\t\tactiveClusterManager:         resource.ActiveClusterMgr,\n\t\tconfig:                       config,\n\t\tlogger:                       resource.GetLogger(),\n\t\tthrottledLogger:              resource.GetThrottledLogger(),\n\t\ttaskSequenceNumber:           1,\n\t\timmediateTaskMaxReadLevel:    0,\n\t\tmaxTaskSequenceNumber:        100000,\n\t\tscheduledTaskMaxReadLevelMap: make(map[string]time.Time),\n\t\tfailoverLevels:               make(map[persistence.HistoryTaskCategory]map[string]persistence.FailoverLevel),\n\t\tremoteClusterCurrentTime:     make(map[string]time.Time),\n\t\teventsCache:                  eventsCache,\n\t}\n\treturn &TestContext{\n\t\tcontextImpl:     shard,\n\t\tResource:        resource,\n\t\tMockEventsCache: eventsCache,\n\t}\n}\n\n// ShardInfo is a test hook for getting shard info\nfunc (s *TestContext) ShardInfo() *persistence.ShardInfo {\n\treturn s.shardInfo\n}\n\n// SetEventsCache is a test hook for setting events cache\nfunc (s *TestContext) SetEventsCache(\n\teventsCache events.Cache,\n) {\n\ts.eventsCache = eventsCache\n\ts.MockEventsCache = nil\n}\n\n// Finish checks whether expectations are met\nfunc (s *TestContext) Finish(\n\tt mock.TestingT,\n) {\n\ts.Resource.Finish(t)\n}\n\nfunc (s *TestContext) AddingPendingFailoverMarker(marker *types.FailoverMarkerAttributes) error {\n\tif s.MockAddingPendingFailoverMarker != nil {\n\t\treturn s.MockAddingPendingFailoverMarker(marker)\n\t}\n\treturn s.contextImpl.AddingPendingFailoverMarker(marker)\n}\n"
  },
  {
    "path": "service/history/shard/controller.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination controller_mock.go -self_package github.com/uber/cadence/service/history/shard\n\npackage shard\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\tworkflow \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/lookup\"\n\t\"github.com/uber/cadence/service/history/resource\"\n)\n\nconst (\n\tshardControllerMembershipUpdateListenerName = \"ShardController\"\n)\n\nvar (\n\terrShardIDOutOfBoundary = &workflow.BadRequestError{Message: \"shard ID is out of boundary\"}\n)\n\ntype (\n\t// EngineFactory is used to create an instance of sharded history engine\n\tEngineFactory interface {\n\t\tCreateEngine(Context) engine.Engine\n\t}\n\n\t// Controller controls history service shards\n\tController interface {\n\t\tcommon.Daemon\n\n\t\t// PrepareToStop starts the graceful shutdown process for controller\n\t\tPrepareToStop()\n\n\t\tGetEngine(workflowID string) (engine.Engine, error)\n\t\tGetEngineForShard(shardID int) (engine.Engine, error)\n\t\tRemoveEngineForShard(shardID int)\n\n\t\t// Following methods describes the current status of the controller\n\t\t// TODO: consider converting to a unified describe method\n\t\tStatus() int32\n\t\tNumShards() int\n\t\tShardIDs() []int32\n\t}\n\n\tshardIDSnapshot struct {\n\t\tshardIDs  []int32\n\t\tnumShards int\n\t}\n\n\tcontroller struct {\n\t\tresource.Resource\n\n\t\tmembershipUpdateCh       chan *membership.ChangedEvent\n\t\tengineFactory            EngineFactory\n\t\tstatus                   int32\n\t\tshuttingDown             int32\n\t\tshutdownWG               sync.WaitGroup\n\t\tshutdownCh               chan struct{}\n\t\tlogger                   log.Logger\n\t\tthrottledLogger          log.Logger\n\t\tconfig                   *config.Config\n\t\tmetricsScope             metrics.Scope\n\t\treplicationBudgetManager cache.Manager\n\n\t\tsync.RWMutex\n\t\thistoryShards   map[int]*historyShardsItem\n\t\tshardIDSnapshot atomic.Pointer[shardIDSnapshot]\n\t}\n\n\thistoryShardsItemStatus int\n\n\thistoryShardsItem struct {\n\t\tresource.Resource\n\n\t\tshardID                  int\n\t\tconfig                   *config.Config\n\t\tlogger                   log.Logger\n\t\tthrottledLogger          log.Logger\n\t\tengineFactory            EngineFactory\n\t\treplicationBudgetManager cache.Manager\n\n\t\tsync.RWMutex\n\t\tstatus historyShardsItemStatus\n\t\tengine engine.Engine\n\t}\n)\n\nconst (\n\thistoryShardsItemStatusInitialized = iota\n\thistoryShardsItemStatusStarted\n\thistoryShardsItemStatusStopped\n)\n\n// NewShardController creates a new shard controller\nfunc NewShardController(\n\tresource resource.Resource,\n\tfactory EngineFactory,\n\tconfig *config.Config,\n\treplicationBudgetManager cache.Manager,\n) Controller {\n\thostAddress := resource.GetHostInfo().GetAddress()\n\treturn &controller{\n\t\tResource:                 resource,\n\t\tstatus:                   common.DaemonStatusInitialized,\n\t\tmembershipUpdateCh:       make(chan *membership.ChangedEvent, 10),\n\t\tengineFactory:            factory,\n\t\thistoryShards:            make(map[int]*historyShardsItem),\n\t\tshutdownCh:               make(chan struct{}),\n\t\tlogger:                   resource.GetLogger().WithTags(tag.ComponentShardController, tag.Address(hostAddress)),\n\t\tthrottledLogger:          resource.GetThrottledLogger().WithTags(tag.ComponentShardController, tag.Address(hostAddress)),\n\t\tconfig:                   config,\n\t\tmetricsScope:             resource.GetMetricsClient().Scope(metrics.HistoryShardControllerScope),\n\t\treplicationBudgetManager: replicationBudgetManager,\n\t}\n}\n\nfunc newHistoryShardsItem(\n\tresource resource.Resource,\n\tshardID int,\n\tfactory EngineFactory,\n\tconfig *config.Config,\n\treplicationBudgetManager cache.Manager,\n) (*historyShardsItem, error) {\n\n\thostAddress := resource.GetHostInfo().GetAddress()\n\treturn &historyShardsItem{\n\t\tResource:                 resource,\n\t\tshardID:                  shardID,\n\t\tstatus:                   historyShardsItemStatusInitialized,\n\t\tengineFactory:            factory,\n\t\tconfig:                   config,\n\t\tlogger:                   resource.GetLogger().WithTags(tag.ShardID(shardID), tag.Address(hostAddress)),\n\t\tthrottledLogger:          resource.GetThrottledLogger().WithTags(tag.ShardID(shardID), tag.Address(hostAddress)),\n\t\treplicationBudgetManager: replicationBudgetManager,\n\t}, nil\n}\n\nfunc (c *controller) Start() {\n\tif !atomic.CompareAndSwapInt32(&c.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tc.acquireShards()\n\tc.shutdownWG.Add(1)\n\tgo c.shardManagementPump()\n\n\terr := c.GetMembershipResolver().Subscribe(service.History, shardControllerMembershipUpdateListenerName, c.membershipUpdateCh)\n\tif err != nil {\n\t\tc.logger.Error(\"subscribing to membership resolver\", tag.Error(err))\n\t}\n\n\tc.logger.Info(\"Shard controller state changed\", tag.LifeCycleStarted)\n}\n\nfunc (c *controller) Stop() {\n\tif !atomic.CompareAndSwapInt32(&c.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tc.logger.Info(\"Stopping shard controller\", tag.ComponentShardController)\n\tdefer c.logger.Info(\"Stopped shard controller\", tag.ComponentShardController)\n\n\tc.PrepareToStop()\n\n\tif err := c.GetMembershipResolver().Unsubscribe(service.History, shardControllerMembershipUpdateListenerName); err != nil {\n\t\tc.logger.Error(\"unsubscribing from membership resolver\", tag.Error(err), tag.OperationFailed)\n\t}\n\tclose(c.shutdownCh)\n\n\tif success := common.AwaitWaitGroup(&c.shutdownWG, time.Minute); !success {\n\t\tc.logger.Warn(\"\", tag.LifeCycleStopTimedout)\n\t}\n\n\tc.logger.Info(\"Shard controller state changed\", tag.LifeCycleStopped)\n}\n\nfunc (c *controller) PrepareToStop() {\n\tatomic.StoreInt32(&c.shuttingDown, 1)\n}\n\nfunc (c *controller) GetEngine(workflowID string) (engine.Engine, error) {\n\tshardID := c.config.GetShardID(workflowID)\n\treturn c.GetEngineForShard(shardID)\n}\n\nfunc (c *controller) GetEngineForShard(shardID int) (engine.Engine, error) {\n\tsw := c.metricsScope.StartTimer(metrics.GetEngineForShardLatency)\n\tdefer sw.Stop()\n\titem, err := c.getOrCreateHistoryShardItem(shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn item.getOrCreateEngine(c.shardClosedCallback)\n}\n\nfunc (c *controller) RemoveEngineForShard(shardID int) {\n\tc.removeEngineForShard(shardID, nil)\n}\n\nfunc (c *controller) Status() int32 {\n\treturn atomic.LoadInt32(&c.status)\n}\n\nfunc (c *controller) NumShards() int {\n\ts := c.shardIDSnapshot.Load()\n\tif s == nil {\n\t\treturn 0\n\t}\n\treturn s.numShards\n}\n\nfunc (c *controller) ShardIDs() []int32 {\n\ts := c.shardIDSnapshot.Load()\n\tif s == nil {\n\t\treturn []int32{}\n\t}\n\treturn s.shardIDs\n}\n\nfunc (c *controller) removeEngineForShard(shardID int, shardItem *historyShardsItem) {\n\tsw := c.metricsScope.StartTimer(metrics.RemoveEngineForShardLatency)\n\tdefer sw.Stop()\n\tc.logger.Info(\"removeEngineForShard called\", tag.ShardID(shardID))\n\tdefer c.logger.Info(\"removeEngineForShard completed\", tag.ShardID(shardID))\n\n\tcurrentShardItem, err := c.removeHistoryShardItem(shardID, shardItem)\n\tif err != nil {\n\t\tc.logger.Error(\"Failed to remove history shard item\", tag.Error(err), tag.ShardID(shardID))\n\t}\n\tif shardItem != nil {\n\t\t// if shardItem is not nil, then currentShardItem either equals to shardItem or is nil\n\t\t// in both cases, we need to stop the engine in shardItem\n\t\tshardItem.stopEngine()\n\t\treturn\n\t}\n\n\t// if shardItem is nil, then stop the engine for the current shardItem, if exists\n\tif currentShardItem != nil {\n\t\tcurrentShardItem.stopEngine()\n\t}\n}\n\nfunc (c *controller) shardClosedCallback(shardID int, shardItem *historyShardsItem) {\n\tc.metricsScope.IncCounter(metrics.ShardClosedCounter)\n\tc.logger.Info(\"Shard controller state changed\", tag.LifeCycleStopping, tag.ComponentShard, tag.ShardID(shardID), tag.Reason(\"shardClosedCallback\"))\n\tc.removeEngineForShard(shardID, shardItem)\n}\n\nfunc (c *controller) getOrCreateHistoryShardItem(shardID int) (*historyShardsItem, error) {\n\tif shardID >= c.config.NumberOfShards || shardID < 0 { // zero based shard ID\n\t\tc.logger.Error(fmt.Sprintf(\"Received shard ID: %v is larger than supported shard number %v\",\n\t\t\tshardID,\n\t\t\tc.config.NumberOfShards,\n\t\t),\n\t\t)\n\t\treturn nil, errShardIDOutOfBoundary\n\t}\n\n\tc.RLock()\n\tif item, ok := c.historyShards[shardID]; ok {\n\t\tif item.isValid() {\n\t\t\tc.RUnlock()\n\t\t\treturn item, nil\n\t\t}\n\t\t// if item not valid then process to create a new one\n\t}\n\tc.RUnlock()\n\n\tc.logger.Info(\"Creating new history shard item\", tag.ShardID(shardID))\n\tdefer c.logger.Info(\"Created new history shard item\", tag.ShardID(shardID))\n\n\tc.Lock()\n\tdefer c.Unlock()\n\n\tif item, ok := c.historyShards[shardID]; ok {\n\t\tif item.isValid() {\n\t\t\treturn item, nil\n\t\t}\n\t\t// if item not valid then process to create a new one\n\t}\n\n\tif c.isShuttingDown() || atomic.LoadInt32(&c.status) == common.DaemonStatusStopped {\n\t\treturn nil, fmt.Errorf(\"controller for host '%v' shutting down\", c.GetHostInfo().Identity())\n\t}\n\tinfo, err := lookup.HistoryServerByShardID(c.GetMembershipResolver(), shardID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tshardBelongsToCurrentHost := info.Identity() == c.GetHostInfo().Identity()\n\tc.logger.Info(\"Shard belongs to current host?\",\n\t\ttag.ShardID(shardID),\n\t\ttag.Value(shardBelongsToCurrentHost),\n\t\ttag.Dynamic(\"shard-owner\", info.Identity()),\n\t\ttag.Dynamic(\"current-host\", c.GetHostInfo().Identity()),\n\t)\n\n\tif shardBelongsToCurrentHost {\n\t\tshardItem, err := newHistoryShardsItem(\n\t\t\tc.Resource,\n\t\t\tshardID,\n\t\t\tc.engineFactory,\n\t\t\tc.config,\n\t\t\tc.replicationBudgetManager,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tc.historyShards[shardID] = shardItem\n\t\tc.updateShardIDSnapshotLocked()\n\t\tc.metricsScope.IncCounter(metrics.ShardItemCreatedCounter)\n\n\t\tshardItem.logger.Info(\"Shard item state changed\", tag.LifeCycleStarted, tag.ComponentShardItem)\n\t\treturn shardItem, nil\n\t}\n\n\t// for backwards compatibility, always return tchannel port\n\treturn nil, CreateShardOwnershipLostError(c.GetHostInfo(), info)\n}\n\nfunc (c *controller) updateShardIDSnapshotLocked() {\n\tshardIDs := make([]int32, 0, len(c.historyShards))\n\tfor shardID := range c.historyShards {\n\t\tshardIDs = append(shardIDs, int32(shardID))\n\t}\n\tsnapshot := &shardIDSnapshot{\n\t\tshardIDs:  shardIDs,\n\t\tnumShards: len(shardIDs),\n\t}\n\tc.shardIDSnapshot.Store(snapshot)\n}\n\nfunc (c *controller) removeHistoryShardItem(shardID int, shardItem *historyShardsItem) (*historyShardsItem, error) {\n\tc.Lock()\n\tdefer c.Unlock()\n\n\tcurrentShardItem, ok := c.historyShards[shardID]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"no item found to remove for shard: %v\", shardID)\n\t}\n\tif shardItem != nil && currentShardItem != shardItem {\n\t\t// the shardItem comparison is a defensive check to make sure we are deleting\n\t\t// what we intend to delete.\n\t\treturn nil, fmt.Errorf(\"current shardItem doesn't match the one we intend to delete for shard: %v\", shardID)\n\t}\n\n\tdelete(c.historyShards, shardID)\n\tc.updateShardIDSnapshotLocked()\n\tc.metricsScope.IncCounter(metrics.ShardItemRemovedCounter)\n\tcurrentShardItem.logger.Info(\"Shard item state changed\", tag.LifeCycleStopped, tag.ComponentShardItem, tag.Number(int64(len(c.historyShards))))\n\treturn currentShardItem, nil\n}\n\n// shardManagementPump is the main event loop for\n// controller. It is responsible for acquiring /\n// releasing shards in response to any event that can\n// change the shard ownership. These events are\n//\n//\ta. Ring membership change\n//\tb. Periodic ticker\n//\tc. ShardOwnershipLostError and subsequent ShardClosedEvents from engine\nfunc (c *controller) shardManagementPump() {\n\tdefer c.shutdownWG.Done()\n\n\tacquireTicker := time.NewTicker(c.config.AcquireShardInterval())\n\tdefer acquireTicker.Stop()\n\n\tfor {\n\n\t\tselect {\n\t\tcase <-c.shutdownCh:\n\t\t\tc.doShutdown()\n\t\t\treturn\n\t\tcase <-acquireTicker.C:\n\t\t\tc.acquireShards()\n\t\tcase changedEvent := <-c.membershipUpdateCh:\n\t\t\tc.metricsScope.IncCounter(metrics.MembershipChangedCounter)\n\n\t\t\tc.logger.Info(\"Ring membership changed\", tag.ValueRingMembershipChangedEvent,\n\t\t\t\ttag.NumberProcessed(len(changedEvent.HostsAdded)),\n\t\t\t\ttag.NumberDeleted(len(changedEvent.HostsRemoved)),\n\t\t\t\ttag.Number(int64(len(changedEvent.HostsUpdated))))\n\t\t\tc.acquireShards()\n\t\t}\n\t}\n}\n\nfunc (c *controller) acquireShards() {\n\tc.logger.Info(\"Acquiring shards\", tag.ComponentShardController, tag.Number(int64(c.NumShards())))\n\tdefer c.logger.Info(\"Acquired shards\", tag.ComponentShardController, tag.Number(int64(c.NumShards())))\n\n\tc.metricsScope.IncCounter(metrics.AcquireShardsCounter)\n\tsw := c.metricsScope.StartTimer(metrics.AcquireShardsLatency)\n\tdefer sw.Stop()\n\n\tnumShards := c.config.NumberOfShards\n\tshardActionCh := make(chan int, numShards)\n\t// Submit all tasks to the channel.\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\tshardActionCh <- shardID // must be non-blocking as there is no other coordination with shutdown\n\t}\n\tclose(shardActionCh)\n\n\tconcurrency := max(c.config.AcquireShardConcurrency(), 1)\n\tvar wg sync.WaitGroup\n\twg.Add(concurrency)\n\t// Spawn workers that would do lookup and add/remove shards concurrently.\n\tfor i := 0; i < concurrency; i++ {\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor shardID := range shardActionCh {\n\t\t\t\tif c.isShuttingDown() {\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tinfo, err := lookup.HistoryServerByShardID(c.GetMembershipResolver(), shardID)\n\t\t\t\tif err != nil {\n\t\t\t\t\tc.logger.Error(\"Error looking up host for shardID\", tag.Error(err), tag.OperationFailed, tag.ShardID(shardID))\n\t\t\t\t} else {\n\t\t\t\t\tif info.Identity() == c.GetHostInfo().Identity() {\n\t\t\t\t\t\t_, err1 := c.GetEngineForShard(shardID)\n\t\t\t\t\t\tif err1 != nil {\n\t\t\t\t\t\t\tc.metricsScope.IncCounter(metrics.GetEngineForShardErrorCounter)\n\t\t\t\t\t\t\tc.logger.Error(\"Unable to create history shard engine\", tag.Error(err1), tag.OperationFailed, tag.ShardID(shardID))\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\t// Wait until all shards are processed.\n\twg.Wait()\n\n\tc.metricsScope.UpdateGauge(metrics.NumShardsGauge, float64(c.NumShards()))\n}\n\nfunc (c *controller) doShutdown() {\n\tc.logger.Info(\"Shard controller state changed\", tag.LifeCycleStopping, tag.Reason(\"shutdown\"))\n\tc.Lock()\n\tdefer c.Unlock()\n\tfor _, item := range c.historyShards {\n\t\titem.stopEngine()\n\t}\n\tc.historyShards = nil\n\tc.updateShardIDSnapshotLocked()\n}\n\nfunc (c *controller) isShuttingDown() bool {\n\treturn atomic.LoadInt32(&c.shuttingDown) != 0\n}\n\nfunc (i *historyShardsItem) getOrCreateEngine(\n\tcloseCallback func(int, *historyShardsItem),\n) (engine.Engine, error) {\n\ti.RLock()\n\tif i.status == historyShardsItemStatusStarted {\n\t\tdefer i.RUnlock()\n\t\treturn i.engine, nil\n\t}\n\ti.RUnlock()\n\n\ti.Lock()\n\tdefer i.Unlock()\n\tswitch i.status {\n\tcase historyShardsItemStatusInitialized:\n\t\ti.logger.Info(\"Shard engine state changed\", tag.LifeCycleStarting, tag.ComponentShardEngine)\n\t\tcontext, err := acquireShard(i, closeCallback)\n\t\tif err != nil {\n\t\t\t// invalidate the shardItem so that the same shardItem won't be\n\t\t\t// used to create another shardContext\n\t\t\ti.logger.Info(\"Shard engine state changed\", tag.LifeCycleStopped, tag.ComponentShardEngine)\n\t\t\ti.status = historyShardsItemStatusStopped\n\t\t\treturn nil, err\n\t\t}\n\t\tif context.PreviousShardOwnerWasDifferent() {\n\t\t\ti.GetMetricsClient().RecordTimer(metrics.ShardInfoScope, metrics.ShardItemAcquisitionLatency,\n\t\t\t\tcontext.GetCurrentTime(i.GetClusterMetadata().GetCurrentClusterName()).Sub(context.GetLastUpdatedTime()))\n\t\t}\n\t\ti.engine = i.engineFactory.CreateEngine(context)\n\t\ti.engine.Start()\n\t\ti.logger.Info(\"Shard engine state changed\", tag.LifeCycleStarted, tag.ComponentShardEngine)\n\t\ti.status = historyShardsItemStatusStarted\n\t\treturn i.engine, nil\n\tcase historyShardsItemStatusStarted:\n\t\treturn i.engine, nil\n\tcase historyShardsItemStatusStopped:\n\t\treturn nil, fmt.Errorf(\"shard %v for host '%v' is shut down\", i.shardID, i.GetHostInfo().Identity())\n\tdefault:\n\t\tpanic(i.logInvalidStatus())\n\t}\n}\n\nfunc (i *historyShardsItem) stopEngine() {\n\ti.Lock()\n\tdefer i.Unlock()\n\n\ti.logger.Info(\"Shard item stopEngine called\", tag.ComponentShardEngine, tag.Dynamic(\"status\", i.status))\n\n\tswitch i.status {\n\tcase historyShardsItemStatusInitialized:\n\t\ti.status = historyShardsItemStatusStopped\n\tcase historyShardsItemStatusStarted:\n\t\ti.logger.Info(\"Shard engine state changed\", tag.LifeCycleStopping, tag.ComponentShardEngine)\n\t\ti.engine.Stop()\n\t\ti.engine = nil\n\t\ti.logger.Info(\"Shard engine state changed\", tag.LifeCycleStopped, tag.ComponentShardEngine)\n\t\ti.status = historyShardsItemStatusStopped\n\tcase historyShardsItemStatusStopped:\n\t\t// no op\n\tdefault:\n\t\tpanic(i.logInvalidStatus())\n\t}\n}\n\nfunc (i *historyShardsItem) isValid() bool {\n\ti.RLock()\n\tdefer i.RUnlock()\n\n\tswitch i.status {\n\tcase historyShardsItemStatusInitialized, historyShardsItemStatusStarted:\n\t\treturn true\n\tcase historyShardsItemStatusStopped:\n\t\treturn false\n\tdefault:\n\t\tpanic(i.logInvalidStatus())\n\t}\n}\n\nfunc (i *historyShardsItem) logInvalidStatus() string {\n\tmsg := fmt.Sprintf(\"Host '%v' encounter invalid status %v for shard item for shardID '%v'.\",\n\t\ti.GetHostInfo().Identity(), i.status, i.shardID)\n\ti.logger.Error(msg)\n\treturn msg\n}\n\n// IsShardOwnershiptLostError checks if a given error is shard ownership lost error\nfunc IsShardOwnershiptLostError(err error) bool {\n\tswitch err.(type) {\n\tcase *persistence.ShardOwnershipLostError:\n\t\treturn true\n\t}\n\n\treturn false\n}\n\n// CreateShardOwnershipLostError creates a new shard ownership lost error\nfunc CreateShardOwnershipLostError(\n\tcurrentHost membership.HostInfo,\n\townerHost membership.HostInfo,\n) *types.ShardOwnershipLostError {\n\taddress, err := ownerHost.GetNamedAddress(membership.PortTchannel)\n\tif err != nil {\n\t\taddress = ownerHost.Identity()\n\t}\n\treturn &types.ShardOwnershipLostError{\n\t\tMessage: fmt.Sprintf(\"Shard is not owned by host: %v\", currentHost.Identity()),\n\t\tOwner:   address,\n\t}\n}\n"
  },
  {
    "path": "service/history/shard/controller_benchmark_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shard\n\nimport (\n\t\"sync\"\n\t\"testing\"\n)\n\n// BenchmarkController_ShardIDs-96         52588224               293.9 ns/op            66 B/op          0 allocs/op\n// go test -bench=. --benchtime=10s --benchmem\n// goos: linux\n// goarch: amd64\n// pkg: github.com/uber/cadence/service/history/shard\n// cpu: AMD EPYC 7B13\n// With the old approach, the benchmark result is:\n// BenchmarkController_ShardIDs-96            39314            324629 ns/op          272333 B/op         19 allocs/op\nfunc BenchmarkController_ShardIDs(b *testing.B) {\n\tnumShards := 16384\n\thistoryShards := make(map[int]*historyShardsItem)\n\tfor i := 0; i < numShards; i++ {\n\t\thistoryShards[i] = &historyShardsItem{shardID: i}\n\t}\n\tshardController := &controller{\n\t\thistoryShards: historyShards,\n\t}\n\tvar wg sync.WaitGroup\n\tfor i := 0; i < b.N; i++ {\n\t\tif i%1000 == 0 { // update is much much less frequent than read\n\t\t\twg.Add(1)\n\t\t\tgo func() {\n\t\t\t\tdefer wg.Done()\n\t\t\t\tshardController.Lock()\n\t\t\t\tshardController.updateShardIDSnapshotLocked()\n\t\t\t\tshardController.Unlock()\n\t\t\t}()\n\t\t}\n\t\tshardController.ShardIDs()\n\t\tshardController.NumShards()\n\t}\n\twg.Wait()\n}\n"
  },
  {
    "path": "service/history/shard/controller_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: controller.go\n//\n// Generated by this command:\n//\n//\tmockgen -package shard -source controller.go -destination controller_mock.go -self_package github.com/uber/cadence/service/history/shard\n//\n\n// Package shard is a generated GoMock package.\npackage shard\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tengine \"github.com/uber/cadence/service/history/engine\"\n)\n\n// MockEngineFactory is a mock of EngineFactory interface.\ntype MockEngineFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEngineFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockEngineFactoryMockRecorder is the mock recorder for MockEngineFactory.\ntype MockEngineFactoryMockRecorder struct {\n\tmock *MockEngineFactory\n}\n\n// NewMockEngineFactory creates a new mock instance.\nfunc NewMockEngineFactory(ctrl *gomock.Controller) *MockEngineFactory {\n\tmock := &MockEngineFactory{ctrl: ctrl}\n\tmock.recorder = &MockEngineFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEngineFactory) EXPECT() *MockEngineFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// CreateEngine mocks base method.\nfunc (m *MockEngineFactory) CreateEngine(arg0 Context) engine.Engine {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateEngine\", arg0)\n\tret0, _ := ret[0].(engine.Engine)\n\treturn ret0\n}\n\n// CreateEngine indicates an expected call of CreateEngine.\nfunc (mr *MockEngineFactoryMockRecorder) CreateEngine(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateEngine\", reflect.TypeOf((*MockEngineFactory)(nil).CreateEngine), arg0)\n}\n\n// MockController is a mock of Controller interface.\ntype MockController struct {\n\tctrl     *gomock.Controller\n\trecorder *MockControllerMockRecorder\n\tisgomock struct{}\n}\n\n// MockControllerMockRecorder is the mock recorder for MockController.\ntype MockControllerMockRecorder struct {\n\tmock *MockController\n}\n\n// NewMockController creates a new mock instance.\nfunc NewMockController(ctrl *gomock.Controller) *MockController {\n\tmock := &MockController{ctrl: ctrl}\n\tmock.recorder = &MockControllerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockController) EXPECT() *MockControllerMockRecorder {\n\treturn m.recorder\n}\n\n// GetEngine mocks base method.\nfunc (m *MockController) GetEngine(workflowID string) (engine.Engine, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEngine\", workflowID)\n\tret0, _ := ret[0].(engine.Engine)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetEngine indicates an expected call of GetEngine.\nfunc (mr *MockControllerMockRecorder) GetEngine(workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEngine\", reflect.TypeOf((*MockController)(nil).GetEngine), workflowID)\n}\n\n// GetEngineForShard mocks base method.\nfunc (m *MockController) GetEngineForShard(shardID int) (engine.Engine, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetEngineForShard\", shardID)\n\tret0, _ := ret[0].(engine.Engine)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetEngineForShard indicates an expected call of GetEngineForShard.\nfunc (mr *MockControllerMockRecorder) GetEngineForShard(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetEngineForShard\", reflect.TypeOf((*MockController)(nil).GetEngineForShard), shardID)\n}\n\n// NumShards mocks base method.\nfunc (m *MockController) NumShards() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NumShards\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// NumShards indicates an expected call of NumShards.\nfunc (mr *MockControllerMockRecorder) NumShards() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NumShards\", reflect.TypeOf((*MockController)(nil).NumShards))\n}\n\n// PrepareToStop mocks base method.\nfunc (m *MockController) PrepareToStop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"PrepareToStop\")\n}\n\n// PrepareToStop indicates an expected call of PrepareToStop.\nfunc (mr *MockControllerMockRecorder) PrepareToStop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PrepareToStop\", reflect.TypeOf((*MockController)(nil).PrepareToStop))\n}\n\n// RemoveEngineForShard mocks base method.\nfunc (m *MockController) RemoveEngineForShard(shardID int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RemoveEngineForShard\", shardID)\n}\n\n// RemoveEngineForShard indicates an expected call of RemoveEngineForShard.\nfunc (mr *MockControllerMockRecorder) RemoveEngineForShard(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveEngineForShard\", reflect.TypeOf((*MockController)(nil).RemoveEngineForShard), shardID)\n}\n\n// ShardIDs mocks base method.\nfunc (m *MockController) ShardIDs() []int32 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ShardIDs\")\n\tret0, _ := ret[0].([]int32)\n\treturn ret0\n}\n\n// ShardIDs indicates an expected call of ShardIDs.\nfunc (mr *MockControllerMockRecorder) ShardIDs() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ShardIDs\", reflect.TypeOf((*MockController)(nil).ShardIDs))\n}\n\n// Start mocks base method.\nfunc (m *MockController) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockControllerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockController)(nil).Start))\n}\n\n// Status mocks base method.\nfunc (m *MockController) Status() int32 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Status\")\n\tret0, _ := ret[0].(int32)\n\treturn ret0\n}\n\n// Status indicates an expected call of Status.\nfunc (mr *MockControllerMockRecorder) Status() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Status\", reflect.TypeOf((*MockController)(nil).Status))\n}\n\n// Stop mocks base method.\nfunc (m *MockController) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockControllerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockController)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/shard/controller_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage shard\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/resource\"\n)\n\ntype (\n\tcontrollerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller             *gomock.Controller\n\t\tmockResource           *resource.Test\n\t\tmockHistoryEngine      *engine.MockEngine\n\t\tmockMembershipResolver *membership.MockResolver\n\n\t\thostInfo          membership.HostInfo\n\t\tmockShardManager  *mmocks.ShardManager\n\t\tmockEngineFactory *MockEngineFactory\n\n\t\tconfig          *config.Config\n\t\tlogger          log.Logger\n\t\tshardController *controller\n\t}\n)\n\nfunc TestControllerSuite(t *testing.T) {\n\ts := new(controllerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *controllerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.History)\n\ts.mockEngineFactory = NewMockEngineFactory(s.controller)\n\n\ts.mockHistoryEngine = engine.NewMockEngine(s.controller)\n\n\ts.mockShardManager = s.mockResource.ShardMgr\n\ts.mockMembershipResolver = s.mockResource.MembershipResolver\n\ts.hostInfo = s.mockResource.GetHostInfo()\n\n\ts.logger = s.mockResource.Logger\n\ts.config = config.NewForTest()\n\n\ts.shardController = NewShardController(s.mockResource, s.mockEngineFactory, s.config, nil).(*controller)\n}\n\nfunc (s *controllerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n}\n\nfunc (s *controllerSuite) TestAcquireShardSuccess() {\n\tnumShards := 10\n\ts.config.NumberOfShards = numShards\n\n\treplicationAck := int64(201)\n\tcurrentClusterTransferAck := int64(210)\n\talternativeClusterTransferAck := int64(320)\n\tcurrentClusterTimerAck := time.Now().Add(-100 * time.Second)\n\talternativeClusterTimerAck := time.Now().Add(-200 * time.Second)\n\n\tmyShards := []int{}\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\thostID := shardID % 4\n\t\tif hostID == 0 {\n\t\t\tmyShards = append(myShards, shardID)\n\t\t\ts.mockHistoryEngine.EXPECT().Start().Return().Times(1)\n\t\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(2)\n\t\t\ts.mockEngineFactory.EXPECT().CreateEngine(gomock.Any()).Return(s.mockHistoryEngine).Times(1)\n\t\t\ts.mockShardManager.On(\"GetShard\", mock.Anything, &persistence.GetShardRequest{ShardID: shardID}).Return(\n\t\t\t\t&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\t\tRangeID:             5,\n\t\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Once()\n\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, &persistence.UpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\tRangeID:             6,\n\t\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t},\n\t\t\t\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t\t},\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t\t\t\t\tQueueStates:             map[int32]*types.QueueState{},\n\t\t\t\t},\n\t\t\t\tPreviousRangeID: 5,\n\t\t\t}).Return(nil).Once()\n\t\t} else {\n\t\t\townerHost := fmt.Sprintf(\"test-acquire-shard-host-%v\", hostID)\n\t\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(membership.NewHostInfo(ownerHost), nil).Times(1)\n\t\t}\n\t}\n\n\ts.shardController.acquireShards()\n\tcount := 0\n\tfor _, shardID := range myShards {\n\t\ts.NotNil(s.shardController.GetEngineForShard(shardID))\n\t\tcount++\n\t}\n\ts.Equal(3, count)\n\ts.Equal(3, s.shardController.NumShards())\n\ts.ElementsMatch([]int32{0, 4, 8}, s.shardController.ShardIDs())\n}\n\nfunc (s *controllerSuite) TestAcquireShardsConcurrently() {\n\tnumShards := 10\n\ts.config.NumberOfShards = numShards\n\ts.config.AcquireShardConcurrency = func(opts ...dynamicproperties.FilterOption) int {\n\t\treturn 10\n\t}\n\n\treplicationAck := int64(201)\n\tcurrentClusterTransferAck := int64(210)\n\talternativeClusterTransferAck := int64(320)\n\tcurrentClusterTimerAck := time.Now().Add(-100 * time.Second)\n\talternativeClusterTimerAck := time.Now().Add(-200 * time.Second)\n\n\tvar myShards []int\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\thostID := shardID % 4\n\t\tif hostID == 0 {\n\t\t\tmyShards = append(myShards, shardID)\n\t\t\ts.mockHistoryEngine.EXPECT().Start().Return().Times(1)\n\t\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(2)\n\t\t\ts.mockEngineFactory.EXPECT().CreateEngine(gomock.Any()).Return(s.mockHistoryEngine).Times(1)\n\t\t\ts.mockShardManager.On(\"GetShard\", mock.Anything, &persistence.GetShardRequest{ShardID: shardID}).Return(\n\t\t\t\t&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\t\tRangeID:             5,\n\t\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t\t},\n\t\t\t\t}, nil).Once()\n\t\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, &persistence.UpdateShardRequest{\n\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\tRangeID:             6,\n\t\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t},\n\t\t\t\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t\t},\n\t\t\t\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t\t},\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t\t\t\t\tQueueStates:             map[int32]*types.QueueState{},\n\t\t\t\t},\n\t\t\t\tPreviousRangeID: 5,\n\t\t\t}).Return(nil).Once()\n\t\t} else {\n\t\t\townerHost := fmt.Sprintf(\"test-acquire-shard-host-%v\", hostID)\n\t\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(membership.NewHostInfo(ownerHost), nil).Times(1)\n\t\t}\n\t}\n\n\ts.shardController.acquireShards()\n\tcount := 0\n\tfor _, shardID := range myShards {\n\t\ts.NotNil(s.shardController.GetEngineForShard(shardID))\n\t\tcount++\n\t}\n\ts.Equal(3, count)\n\ts.Equal(3, s.shardController.NumShards())\n\ts.ElementsMatch([]int32{0, 4, 8}, s.shardController.ShardIDs())\n}\n\nfunc (s *controllerSuite) TestAcquireShardLookupFailure() {\n\tnumShards := 2\n\ts.config.NumberOfShards = numShards\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(membership.HostInfo{}, errors.New(\"ring failure\")).Times(1)\n\t}\n\n\ts.shardController.acquireShards()\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(membership.HostInfo{}, errors.New(\"ring failure\")).Times(1)\n\t\ts.Nil(s.shardController.GetEngineForShard(shardID))\n\t}\n\ts.Equal(0, s.shardController.NumShards())\n\ts.Empty(s.shardController.ShardIDs())\n}\n\nfunc (s *controllerSuite) TestAcquireShardRenewSuccess() {\n\tnumShards := 2\n\ts.config.NumberOfShards = numShards\n\n\treplicationAck := int64(201)\n\tcurrentClusterTransferAck := int64(210)\n\talternativeClusterTransferAck := int64(320)\n\tcurrentClusterTimerAck := time.Now().Add(-100 * time.Second)\n\talternativeClusterTimerAck := time.Now().Add(-200 * time.Second)\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockHistoryEngine.EXPECT().Start().Return().Times(1)\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(2)\n\t\ts.mockEngineFactory.EXPECT().CreateEngine(gomock.Any()).Return(s.mockHistoryEngine).Times(1)\n\t\ts.mockShardManager.On(\"GetShard\", mock.Anything, &persistence.GetShardRequest{ShardID: shardID}).Return(\n\t\t\t&persistence.GetShardResponse{\n\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\tRangeID:             5,\n\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t},\n\t\t\t}, nil).Once()\n\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, &persistence.UpdateShardRequest{\n\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\tShardID:             shardID,\n\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\tRangeID:             6,\n\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t},\n\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t},\n\t\t\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t},\n\t\t\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t},\n\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t\t\t\tQueueStates:             map[int32]*types.QueueState{},\n\t\t\t},\n\t\t\tPreviousRangeID: 5,\n\t\t}).Return(nil).Once()\n\t}\n\n\ts.shardController.acquireShards()\n\n\ts.Equal(2, s.shardController.NumShards())\n\ts.ElementsMatch([]int32{0, 1}, s.shardController.ShardIDs())\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(1)\n\t}\n\ts.shardController.acquireShards()\n\n\ts.Equal(2, s.shardController.NumShards())\n\ts.ElementsMatch([]int32{0, 1}, s.shardController.ShardIDs())\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.NotNil(s.shardController.GetEngineForShard(shardID))\n\t}\n}\n\nfunc (s *controllerSuite) TestAcquireShardRenewLookupFailed() {\n\tnumShards := 2\n\ts.config.NumberOfShards = numShards\n\n\treplicationAck := int64(201)\n\tcurrentClusterTransferAck := int64(210)\n\talternativeClusterTransferAck := int64(320)\n\tcurrentClusterTimerAck := time.Now().Add(-100 * time.Second)\n\talternativeClusterTimerAck := time.Now().Add(-200 * time.Second)\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockHistoryEngine.EXPECT().Start().Return().Times(1)\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(2)\n\t\ts.mockEngineFactory.EXPECT().CreateEngine(gomock.Any()).Return(s.mockHistoryEngine).Times(1)\n\t\ts.mockShardManager.On(\"GetShard\", mock.Anything, &persistence.GetShardRequest{ShardID: shardID}).Return(\n\t\t\t&persistence.GetShardResponse{\n\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\tShardID:             shardID,\n\t\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\t\tRangeID:             5,\n\t\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t\t},\n\t\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\t},\n\t\t\t}, nil).Once()\n\t\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, &persistence.UpdateShardRequest{\n\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\tShardID:             shardID,\n\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\tRangeID:             6,\n\t\t\t\tStolenSinceRenew:    1,\n\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t},\n\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t},\n\t\t\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t},\n\t\t\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t\t},\n\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t\t\t\tQueueStates:             map[int32]*types.QueueState{},\n\t\t\t},\n\t\t\tPreviousRangeID: 5,\n\t\t}).Return(nil).Once()\n\t}\n\n\ts.shardController.acquireShards()\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(membership.HostInfo{}, errors.New(\"ring failure\")).Times(1)\n\t}\n\ts.shardController.acquireShards()\n\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\ts.NotNil(s.shardController.GetEngineForShard(shardID))\n\t}\n}\n\nfunc (s *controllerSuite) TestHistoryEngineClosed() {\n\tnumShards := 4\n\ts.config.NumberOfShards = numShards\n\ts.shardController = NewShardController(s.mockResource, s.mockEngineFactory, s.config, nil).(*controller)\n\thistoryEngines := make(map[int]*engine.MockEngine)\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\tmockEngine := engine.NewMockEngine(s.controller)\n\t\thistoryEngines[shardID] = mockEngine\n\t\ts.setupMocksForAcquireShard(shardID, mockEngine, 5, 6)\n\t}\n\n\ts.mockMembershipResolver.EXPECT().Subscribe(service.History, shardControllerMembershipUpdateListenerName,\n\t\tgomock.Any()).Return(nil).AnyTimes()\n\ts.shardController.Start()\n\tvar workerWG sync.WaitGroup\n\tfor w := 0; w < 10; w++ {\n\t\tworkerWG.Add(1)\n\t\tgo func() {\n\t\t\tfor attempt := 0; attempt < 10; attempt++ {\n\t\t\t\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\t\t\t\tengine, err := s.shardController.GetEngineForShard(shardID)\n\t\t\t\t\ts.Nil(err)\n\t\t\t\t\ts.NotNil(engine)\n\t\t\t\t}\n\t\t\t}\n\t\t\tworkerWG.Done()\n\t\t}()\n\t}\n\n\tworkerWG.Wait()\n\n\tdifferentHostInfo := membership.NewHostInfo(\"another-host\")\n\tfor shardID := 0; shardID < 2; shardID++ {\n\t\tmockEngine := historyEngines[shardID]\n\t\tmockEngine.EXPECT().Stop().Return().Times(1)\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(differentHostInfo, nil).AnyTimes()\n\t\ts.shardController.shardClosedCallback(shardID, nil)\n\t}\n\n\tfor w := 0; w < 10; w++ {\n\t\tworkerWG.Add(1)\n\t\tgo func() {\n\t\t\tfor attempt := 0; attempt < 10; attempt++ {\n\t\t\t\tfor shardID := 2; shardID < numShards; shardID++ {\n\t\t\t\t\tengine, err := s.shardController.GetEngineForShard(shardID)\n\t\t\t\t\ts.Nil(err)\n\t\t\t\t\ts.NotNil(engine)\n\t\t\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t\t\t}\n\t\t\t}\n\t\t\tworkerWG.Done()\n\t\t}()\n\t}\n\n\tfor w := 0; w < 10; w++ {\n\t\tworkerWG.Add(1)\n\t\tgo func() {\n\t\t\tshardLost := false\n\t\t\tfor attempt := 0; !shardLost && attempt < 10; attempt++ {\n\t\t\t\tfor shardID := 0; shardID < 2; shardID++ {\n\t\t\t\t\t_, err := s.shardController.GetEngineForShard(shardID)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\ts.logger.Error(\"ShardLost\", tag.Error(err))\n\t\t\t\t\t\tshardLost = true\n\t\t\t\t\t}\n\t\t\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\ts.True(shardLost)\n\t\t\tworkerWG.Done()\n\t\t}()\n\t}\n\n\tworkerWG.Wait()\n\n\ts.mockMembershipResolver.EXPECT().Unsubscribe(service.History, shardControllerMembershipUpdateListenerName).Return(nil).AnyTimes()\n\tfor shardID := 2; shardID < numShards; shardID++ {\n\t\tmockEngine := historyEngines[shardID]\n\t\tmockEngine.EXPECT().Stop().Return().Times(1)\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).AnyTimes()\n\t}\n\ts.shardController.Stop()\n\n\ts.Equal(0, s.shardController.NumShards())\n\ts.Empty(s.shardController.ShardIDs())\n}\n\nfunc (s *controllerSuite) TestShardControllerClosed() {\n\tnumShards := 4\n\ts.config.NumberOfShards = numShards\n\ts.shardController = NewShardController(s.mockResource, s.mockEngineFactory, s.config, nil).(*controller)\n\thistoryEngines := make(map[int]*engine.MockEngine)\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\tmockEngine := engine.NewMockEngine(s.controller)\n\t\thistoryEngines[shardID] = mockEngine\n\t\ts.setupMocksForAcquireShard(shardID, mockEngine, 5, 6)\n\t}\n\n\ts.mockMembershipResolver.EXPECT().Subscribe(service.History, shardControllerMembershipUpdateListenerName, gomock.Any()).Return(nil).AnyTimes()\n\ts.shardController.Start()\n\n\tvar workerWG sync.WaitGroup\n\tfor w := 0; w < 10; w++ {\n\t\tworkerWG.Add(1)\n\t\tgo func() {\n\t\t\tshardLost := false\n\t\t\tfor attempt := 0; !shardLost && attempt < 10; attempt++ {\n\t\t\t\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\t\t\t\t_, err := s.shardController.GetEngineForShard(shardID)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\ts.logger.Error(\"ShardLost\", tag.Error(err))\n\t\t\t\t\t\tshardLost = true\n\t\t\t\t\t}\n\t\t\t\t\ttime.Sleep(20 * time.Millisecond)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\ts.True(shardLost)\n\t\t\tworkerWG.Done()\n\t\t}()\n\t}\n\n\ts.mockMembershipResolver.EXPECT().Unsubscribe(service.History, shardControllerMembershipUpdateListenerName).Return(nil).AnyTimes()\n\tfor shardID := 0; shardID < numShards; shardID++ {\n\t\tmockEngine := historyEngines[shardID]\n\t\tmockEngine.EXPECT().Stop().Times(1)\n\t\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).AnyTimes()\n\t}\n\ts.shardController.Stop()\n\tworkerWG.Wait()\n\n\ts.Equal(0, s.shardController.NumShards())\n\ts.Empty(s.shardController.ShardIDs())\n}\n\nfunc (s *controllerSuite) TestGetOrCreateHistoryShardItem_InvalidShardID_Error() {\n\ts.config.NumberOfShards = 4\n\ts.shardController = NewShardController(s.mockResource, s.mockEngineFactory, s.config, nil).(*controller)\n\n\teng, err := s.shardController.GetEngineForShard(-1)\n\ts.Nil(eng)\n\ts.Error(err)\n\n\teng, err = s.shardController.GetEngineForShard(s.config.NumberOfShards)\n\ts.Nil(eng)\n\ts.Error(err)\n}\n\nfunc (s *controllerSuite) setupMocksForAcquireShard(shardID int, mockEngine *engine.MockEngine, currentRangeID,\n\tnewRangeID int64) {\n\n\treplicationAck := int64(201)\n\tcurrentClusterTransferAck := int64(210)\n\talternativeClusterTransferAck := int64(320)\n\tcurrentClusterTimerAck := time.Now().Add(-100 * time.Second)\n\talternativeClusterTimerAck := time.Now().Add(-200 * time.Second)\n\n\t// s.mockResource.ExecutionMgr.On(\"Close\").Return()\n\tmockEngine.EXPECT().Start().Times(1)\n\ts.mockMembershipResolver.EXPECT().Lookup(service.History, string(rune(shardID))).Return(s.hostInfo, nil).Times(2)\n\ts.mockEngineFactory.EXPECT().CreateEngine(gomock.Any()).Return(mockEngine).Times(1)\n\ts.mockShardManager.On(\"GetShard\", mock.Anything, &persistence.GetShardRequest{ShardID: shardID}).Return(\n\t\t&persistence.GetShardResponse{\n\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\tShardID:             shardID,\n\t\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\t\tRangeID:             currentRangeID,\n\t\t\t\tReplicationAckLevel: replicationAck,\n\t\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t\t},\n\t\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t\t},\n\t\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\t},\n\t\t}, nil).Once()\n\ts.mockShardManager.On(\"UpdateShard\", mock.Anything, &persistence.UpdateShardRequest{\n\t\tShardInfo: &persistence.ShardInfo{\n\t\t\tShardID:             shardID,\n\t\t\tOwner:               s.hostInfo.Identity(),\n\t\t\tRangeID:             newRangeID,\n\t\t\tStolenSinceRenew:    1,\n\t\t\tReplicationAckLevel: replicationAck,\n\t\t\tTransferAckLevel:    currentClusterTransferAck,\n\t\t\tTimerAckLevel:       currentClusterTimerAck,\n\t\t\tClusterTransferAckLevel: map[string]int64{\n\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTransferAck,\n\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTransferAck,\n\t\t\t},\n\t\t\tClusterTimerAckLevel: map[string]time.Time{\n\t\t\t\tcluster.TestCurrentClusterName:     currentClusterTimerAck,\n\t\t\t\tcluster.TestAlternativeClusterName: alternativeClusterTimerAck,\n\t\t\t},\n\t\t\tTransferProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t},\n\t\t\tTimerProcessingQueueStates: &types.ProcessingQueueStates{\n\t\t\t\tStatesByCluster: make(map[string][]*types.ProcessingQueueState),\n\t\t\t},\n\t\t\tClusterReplicationLevel: map[string]int64{},\n\t\t\tReplicationDLQAckLevel:  map[string]int64{},\n\t\t\tQueueStates:             map[int32]*types.QueueState{},\n\t\t},\n\t\tPreviousRangeID: currentRangeID,\n\t}).Return(nil).Once()\n}\n"
  },
  {
    "path": "service/history/simulation/event.go",
    "content": "package simulation\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os\"\n\t\"time\"\n)\n\nvar enabled = false\n\nfunc init() {\n\tenabled = os.Getenv(\"HISTORY_LOG_EVENTS\") == \"true\"\n}\n\nconst (\n\tEventNameCreateHistoryTask  = \"Create History Task\"\n\tEventNameExecuteHistoryTask = \"Execute History Task\"\n)\n\ntype E struct {\n\tShardID    int\n\tDomainID   string\n\tWorkflowID string\n\tRunID      string\n\n\tEventTime time.Time\n\n\t// EventName describes the event. It is used to query events in simulations so don't change existing event names.\n\tEventName string\n\tHost      string\n\tPayload   map[string]any\n}\n\nfunc Enabled() bool {\n\treturn enabled\n}\n\nfunc LogEvents(events ...E) {\n\tif !enabled {\n\t\treturn\n\t}\n\tfor _, e := range events {\n\t\te.EventTime = time.Now()\n\t\tdata, err := json.Marshal(e)\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"failed to marshal event: %v\", err)\n\t\t}\n\n\t\tfmt.Printf(\"History New Event: %s\\n\", data)\n\t}\n}\n"
  },
  {
    "path": "service/history/task/constants.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport \"time\"\n\nconst (\n\tloadDomainEntryForTaskRetryDelay = 100 * time.Millisecond\n\n\tactiveTaskResubmitMaxAttempts = 10\n\n\tdefaultTaskEventLoggerSize = 100\n\n\tstickyTaskMaxRetryCount = 100\n\n\t// noPriority is the value returned if no priority is ever assigned to the task\n\tnoPriority = -1\n)\n"
  },
  {
    "path": "service/history/task/event_logger.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype (\n\teventLogger interface {\n\t\tAddEvent(eventMsg string, details ...interface{})\n\t\tFlushEvents(msg string) int\n\t}\n\n\tevent struct {\n\t\ttimestamp time.Time\n\t\tmessage   string\n\t\tdetails   []interface{}\n\t}\n\n\teventLoggerImpl struct {\n\t\tlogger     log.Logger\n\t\ttimeSource clock.TimeSource\n\n\t\tevents       []event\n\t\tnextEventIdx int\n\t\tmaxSize      int\n\t}\n)\n\nfunc newEventLogger(\n\tlogger log.Logger,\n\ttimeSource clock.TimeSource,\n\tmaxSize int,\n) eventLogger {\n\treturn &eventLoggerImpl{\n\t\tlogger:     logger,\n\t\ttimeSource: timeSource,\n\n\t\tevents:       make([]event, maxSize),\n\t\tnextEventIdx: 0,\n\t\tmaxSize:      maxSize,\n\t}\n}\n\nfunc (e *eventLoggerImpl) AddEvent(\n\teventMsg string,\n\tdetails ...interface{},\n) {\n\te.events[e.nextEventIdx] = event{\n\t\ttimestamp: e.timeSource.Now(),\n\t\tmessage:   eventMsg,\n\t\tdetails:   details,\n\t}\n\n\te.nextEventIdx = (e.nextEventIdx + 1) % e.maxSize\n}\n\nfunc (e *eventLoggerImpl) FlushEvents(\n\tmsg string,\n) int {\n\tvar builder strings.Builder\n\teventIdx := e.nextEventIdx\n\teventsFlushed := 0\n\n\tfor i := 0; i != len(e.events); i++ {\n\t\t// dump all events in reverse order\n\t\teventIdx--\n\t\tif eventIdx < 0 {\n\t\t\teventIdx = len(e.events) - 1\n\t\t}\n\n\t\tevent := e.events[eventIdx]\n\t\tif event.timestamp.IsZero() {\n\t\t\tbreak\n\t\t}\n\n\t\teventsFlushed++\n\t\tbuilder.WriteString(fmt.Sprintf(\"%v, %s, %v\\n\", event.timestamp, event.message, event.details))\n\t}\n\n\te.logger.Info(msg, tag.Key(\"events\"), tag.Value(builder.String()))\n\n\te.events = make([]event, e.maxSize)\n\n\treturn eventsFlushed\n}\n"
  },
  {
    "path": "service/history/task/event_logger_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n)\n\ntype (\n\teventLoggerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tmockLogger *log.MockLogger\n\n\t\teventLogger *eventLoggerImpl\n\t}\n)\n\nfunc TestEventLoggerSuite(t *testing.T) {\n\ts := new(eventLoggerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *eventLoggerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.mockLogger = log.NewMockLogger(gomock.NewController(s.T()))\n\n\ts.eventLogger = newEventLogger(\n\t\ts.mockLogger,\n\t\tclock.NewRealTimeSource(),\n\t\tdefaultTaskEventLoggerSize,\n\t).(*eventLoggerImpl)\n}\n\nfunc (s *eventLoggerSuite) TestAddEvent() {\n\tfor i := 0; i != defaultTaskEventLoggerSize*2; i++ {\n\t\ts.eventLogger.AddEvent(\"some random event\", i)\n\t\ts.Equal((i+1)%defaultTaskEventLoggerSize, s.eventLogger.nextEventIdx)\n\t}\n\n\tfor i := 0; i != defaultTaskEventLoggerSize; i++ {\n\t\t// check if old events got overwritten\n\t\ts.Equal(i+defaultTaskEventLoggerSize, s.eventLogger.events[i].details[0])\n\t}\n\n\ts.Len(s.eventLogger.events, defaultTaskEventLoggerSize)\n}\n\nfunc (s *eventLoggerSuite) TestFlushEvents() {\n\tfor _, numEvents := range []int{0, defaultTaskEventLoggerSize / 2, defaultTaskEventLoggerSize, defaultTaskEventLoggerSize * 2} {\n\t\tfor i := 0; i != numEvents; i++ {\n\t\t\ts.eventLogger.AddEvent(\"some random event\")\n\t\t}\n\n\t\texpectedEventsFlushed := min(numEvents, defaultTaskEventLoggerSize)\n\t\ts.mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\n\t\ts.Equal(expectedEventsFlushed, s.eventLogger.FlushEvents(\"some random message\"))\n\t}\n}\n"
  },
  {
    "path": "service/history/task/executor_wrapper.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"runtime/debug\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\ntype (\n\texecutorWrapper struct {\n\t\tcurrentClusterName string\n\t\tactiveClusterMgr   activecluster.Manager\n\t\tactiveExecutor     Executor\n\t\tstandbyExecutor    Executor\n\t\tlogger             log.Logger\n\t}\n)\n\nfunc NewExecutorWrapper(\n\tcurrentClusterName string,\n\tactiveClusterMgr activecluster.Manager,\n\tactiveExecutor Executor,\n\tstandbyExecutor Executor,\n\tlogger log.Logger,\n) Executor {\n\treturn &executorWrapper{\n\t\tcurrentClusterName: currentClusterName,\n\t\tactiveClusterMgr:   activeClusterMgr,\n\t\tactiveExecutor:     activeExecutor,\n\t\tstandbyExecutor:    standbyExecutor,\n\t\tlogger:             logger,\n\t}\n}\n\nfunc (e *executorWrapper) Stop() {\n\te.activeExecutor.Stop()\n\te.standbyExecutor.Stop()\n}\n\nfunc (e *executorWrapper) Execute(task Task) (ExecuteResponse, error) {\n\tif e.isActiveTask(task) {\n\t\treturn e.activeExecutor.Execute(task)\n\t}\n\n\treturn e.standbyExecutor.Execute(task)\n}\n\nfunc (e *executorWrapper) isActiveTask(\n\ttask Task,\n) bool {\n\tdomainID := task.GetDomainID()\n\twfID := task.GetWorkflowID()\n\trID := task.GetRunID()\n\n\tactiveClusterInfo, err := e.activeClusterMgr.GetActiveClusterInfoByWorkflow(context.Background(), domainID, wfID, rID)\n\tif err != nil {\n\t\te.logger.Warn(\"Failed to get active cluster info, process task as active.\", tag.WorkflowDomainID(domainID), tag.WorkflowID(wfID), tag.WorkflowRunID(rID), tag.Error(err))\n\t\treturn true\n\t}\n\n\tif activeClusterInfo.ActiveClusterName != e.currentClusterName {\n\t\tif e.logger.DebugOn() {\n\t\t\ttaskJSON, _ := json.Marshal(task)\n\t\t\te.logger.Debug(\"Process task as standby.\",\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.Dynamic(\"task\", string(taskJSON)),\n\t\t\t\ttag.Dynamic(\"taskType\", task.GetTaskType()),\n\t\t\t\ttag.ClusterName(activeClusterInfo.ActiveClusterName),\n\t\t\t\ttag.Dynamic(\"stack\", string(debug.Stack())),\n\t\t\t)\n\t\t}\n\t\treturn false\n\t}\n\tif e.logger.DebugOn() {\n\t\ttaskJSON, _ := json.Marshal(task)\n\t\te.logger.Debug(\"Process task as active.\",\n\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\ttag.Dynamic(\"task\", string(taskJSON)),\n\t\t\ttag.Dynamic(\"taskType\", task.GetTaskType()),\n\t\t\ttag.ClusterName(activeClusterInfo.ActiveClusterName),\n\t\t\ttag.Dynamic(\"stack\", string(debug.Stack())),\n\t\t)\n\t}\n\treturn true\n}\n"
  },
  {
    "path": "service/history/task/executor_wrapper_test.go",
    "content": "package task\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestExecutorWrapper_IsActiveTask(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcurrentCluster string\n\t\tdomainID       string\n\t\tworkflowID     string\n\t\trunID          string\n\t\tdomainError    error\n\t\tisActiveActive bool\n\t\tactiveCluster  string\n\t\tlookupError    error\n\t\texpectedResult bool\n\t}{\n\t\t{\n\t\t\tname:           \"Active-Active domain - current cluster is active\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tdomainID:       \"domain1\",\n\t\t\tworkflowID:     \"workflow1\",\n\t\t\trunID:          \"run1\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveCluster:  \"cluster1\",\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"Active-Active domain - current cluster is not active\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tdomainID:       \"domain1\",\n\t\t\tworkflowID:     \"workflow1\",\n\t\t\trunID:          \"run1\",\n\t\t\tisActiveActive: true,\n\t\t\tactiveCluster:  \"cluster2\",\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"Active-Active domain - lookup error\",\n\t\t\tcurrentCluster: \"cluster1\",\n\t\t\tdomainID:       \"domain1\",\n\t\t\tworkflowID:     \"workflow1\",\n\t\t\trunID:          \"run1\",\n\t\t\tisActiveActive: true,\n\t\t\tlookupError:    assert.AnError,\n\t\t\texpectedResult: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\t// Setup mocks\n\t\t\tmockTask := NewMockTask(ctrl)\n\t\t\tmockTask.EXPECT().GetDomainID().Return(tt.domainID).AnyTimes()\n\t\t\tmockTask.EXPECT().GetWorkflowID().Return(tt.workflowID).AnyTimes()\n\t\t\tmockTask.EXPECT().GetRunID().Return(tt.runID).AnyTimes()\n\t\t\tmockTask.EXPECT().GetInfo().Return(&persistence.DecisionTask{}).AnyTimes()\n\t\t\tmockTask.EXPECT().GetTaskType().Return(0).AnyTimes() // called by debug log\n\n\t\t\tmockActiveClusterMgr := activecluster.NewMockManager(ctrl)\n\t\t\tif tt.lookupError == nil {\n\t\t\t\tmockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), tt.domainID, tt.workflowID, tt.runID).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{ActiveClusterName: tt.activeCluster}, nil)\n\t\t\t} else {\n\t\t\t\tmockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), tt.domainID, tt.workflowID, tt.runID).\n\t\t\t\t\tReturn(nil, tt.lookupError)\n\t\t\t}\n\n\t\t\tmockLogger := testlogger.New(t)\n\n\t\t\t// Create executor wrapper\n\t\t\twrapper := NewExecutorWrapper(\n\t\t\t\ttt.currentCluster,\n\t\t\t\tmockActiveClusterMgr,\n\t\t\t\tNewMockExecutor(ctrl),\n\t\t\t\tNewMockExecutor(ctrl),\n\t\t\t\tmockLogger,\n\t\t\t)\n\n\t\t\t// Execute test\n\t\t\tresult := wrapper.(*executorWrapper).isActiveTask(mockTask)\n\n\t\t\t// Verify result\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/interface.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go -self_package github.com/uber/cadence/service/history/task\n\npackage task\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/future\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\t// Task is the interface for all tasks generated by history service\n\tTask interface {\n\t\ttask.PriorityTask\n\t\tpersistence.Task\n\t\tGetQueueType() QueueType\n\t\tGetShard() shard.Context\n\t\tGetAttempt() int\n\t\tGetInfo() persistence.Task\n\t\tSetInitialSubmitTime(time.Time)\n\t}\n\n\t// CrossClusterTask is the interface for processing cross cluster task in the source cluster\n\tCrossClusterTask interface {\n\t\tTask\n\n\t\tIsValid() bool\n\t\tIsReadyForPoll() bool\n\t\tGetCrossClusterRequest() (*types.CrossClusterTaskRequest, error)\n\t\tRecordResponse(*types.CrossClusterTaskResponse) error\n\t}\n\n\t// Key identifies a Task and defines a total order among tasks\n\tKey interface {\n\t\tLess(Key) bool\n\t}\n\n\t// Executor contains the execution logic for Task\n\tExecutor interface {\n\t\tExecute(task Task) (ExecuteResponse, error)\n\t\tStop()\n\t}\n\n\t// Filter filters Task\n\tFilter func(task persistence.Task) (bool, error)\n\n\t// Initializer initializes a Task based on the Info\n\tInitializer func(persistence.Task) Task\n\n\t// PriorityAssigner assigns priority to Tasks\n\tPriorityAssigner interface {\n\t\tAssign(Task) error\n\t}\n\n\t// Processor is the worker pool for processing Tasks\n\tProcessor interface {\n\t\tcommon.Daemon\n\t\tSubmit(Task) error\n\t\tTrySubmit(Task) (bool, error)\n\t}\n\n\t// Redispatcher buffers tasks and periodically redispatch them to Processor\n\t// redispatch can also be triggered immediately by calling the Redispatch method\n\tRedispatcher interface {\n\t\tcommon.Daemon\n\t\tAddTask(Task)\n\t\tRedispatch(targetSize int)\n\t\tRedispatchTask(Task, time.Time)\n\t\tSize() int\n\t}\n\n\tRescheduler interface {\n\t\tcommon.Daemon\n\t\tRescheduleTask(Task, time.Time)\n\t\tRescheduleDomains(domainIDs map[string]struct{})\n\t\tSize() int\n\t}\n\t// Fetcher is a host level component for aggregating task fetch requests\n\t// from all shards on the host and perform one fetching operation for\n\t// aggregated requests.\n\tFetcher interface {\n\t\tcommon.Daemon\n\t\tGetSourceCluster() string\n\t\tFetch(shardID int, fetchParams ...interface{}) future.Future\n\t}\n\n\t// Fetchers is a group of Fetchers, one for each source cluster\n\tFetchers []Fetcher\n\n\t// QueueType is the type of task queue\n\tQueueType int\n\n\tExecuteResponse struct {\n\t\tScope        metrics.Scope\n\t\tIsActiveTask bool\n\t}\n)\n\nconst (\n\t// QueueTypeActiveTransfer is the queue type for active transfer queue processor (TODO: remove this when history queue v1 is deprecated)\n\tQueueTypeActiveTransfer QueueType = iota + 1\n\t// QueueTypeStandbyTransfer is the queue type for standby transfer queue processor (TODO: remove this when history queue v1 is deprecated)\n\tQueueTypeStandbyTransfer\n\t// QueueTypeActiveTimer is the queue type for active timer queue processor (TODO: remove this when history queue v1 is deprecated)\n\tQueueTypeActiveTimer\n\t// QueueTypeStandbyTimer is the queue type for standby timer queue processor (TODO: remove this when history queue v1 is deprecated)\n\tQueueTypeStandbyTimer\n\t// QueueTypeReplication is the queue type for replication queue processor\n\tQueueTypeReplication\n\t// QueueTypeTransfer is the queue type for transfer queue processor\n\tQueueTypeTransfer\n\t// QueueTypeTimer is the queue type for timer queue processor\n\tQueueTypeTimer\n)\n"
  },
  {
    "path": "service/history/task/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -package task -source interface.go -destination interface_mock.go -self_package github.com/uber/cadence/service/history/task\n//\n\n// Package task is a generated GoMock package.\npackage task\n\nimport (\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tfuture \"github.com/uber/cadence/common/future\"\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\ttask \"github.com/uber/cadence/common/task\"\n\ttypes \"github.com/uber/cadence/common/types\"\n\tshard \"github.com/uber/cadence/service/history/shard\"\n)\n\n// MockTask is a mock of Task interface.\ntype MockTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskMockRecorder is the mock recorder for MockTask.\ntype MockTaskMockRecorder struct {\n\tmock *MockTask\n}\n\n// NewMockTask creates a new mock instance.\nfunc NewMockTask(ctrl *gomock.Controller) *MockTask {\n\tmock := &MockTask{ctrl: ctrl}\n\tmock.recorder = &MockTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTask) EXPECT() *MockTaskMockRecorder {\n\treturn m.recorder\n}\n\n// Ack mocks base method.\nfunc (m *MockTask) Ack() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Ack\")\n}\n\n// Ack indicates an expected call of Ack.\nfunc (mr *MockTaskMockRecorder) Ack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ack\", reflect.TypeOf((*MockTask)(nil).Ack))\n}\n\n// ByteSize mocks base method.\nfunc (m *MockTask) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockTaskMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockTask)(nil).ByteSize))\n}\n\n// Cancel mocks base method.\nfunc (m *MockTask) Cancel() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Cancel\")\n}\n\n// Cancel indicates an expected call of Cancel.\nfunc (mr *MockTaskMockRecorder) Cancel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Cancel\", reflect.TypeOf((*MockTask)(nil).Cancel))\n}\n\n// Execute mocks base method.\nfunc (m *MockTask) Execute() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockTaskMockRecorder) Execute() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockTask)(nil).Execute))\n}\n\n// GetAttempt mocks base method.\nfunc (m *MockTask) GetAttempt() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAttempt\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetAttempt indicates an expected call of GetAttempt.\nfunc (mr *MockTaskMockRecorder) GetAttempt() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAttempt\", reflect.TypeOf((*MockTask)(nil).GetAttempt))\n}\n\n// GetDomainID mocks base method.\nfunc (m *MockTask) GetDomainID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainID indicates an expected call of GetDomainID.\nfunc (mr *MockTaskMockRecorder) GetDomainID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainID\", reflect.TypeOf((*MockTask)(nil).GetDomainID))\n}\n\n// GetInfo mocks base method.\nfunc (m *MockTask) GetInfo() persistence.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetInfo\")\n\tret0, _ := ret[0].(persistence.Task)\n\treturn ret0\n}\n\n// GetInfo indicates an expected call of GetInfo.\nfunc (mr *MockTaskMockRecorder) GetInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetInfo\", reflect.TypeOf((*MockTask)(nil).GetInfo))\n}\n\n// GetOriginalTaskList mocks base method.\nfunc (m *MockTask) GetOriginalTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetOriginalTaskList indicates an expected call of GetOriginalTaskList.\nfunc (mr *MockTaskMockRecorder) GetOriginalTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskList\", reflect.TypeOf((*MockTask)(nil).GetOriginalTaskList))\n}\n\n// GetOriginalTaskListKind mocks base method.\nfunc (m *MockTask) GetOriginalTaskListKind() types.TaskListKind {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskListKind\")\n\tret0, _ := ret[0].(types.TaskListKind)\n\treturn ret0\n}\n\n// GetOriginalTaskListKind indicates an expected call of GetOriginalTaskListKind.\nfunc (mr *MockTaskMockRecorder) GetOriginalTaskListKind() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskListKind\", reflect.TypeOf((*MockTask)(nil).GetOriginalTaskListKind))\n}\n\n// GetQueueType mocks base method.\nfunc (m *MockTask) GetQueueType() QueueType {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueType\")\n\tret0, _ := ret[0].(QueueType)\n\treturn ret0\n}\n\n// GetQueueType indicates an expected call of GetQueueType.\nfunc (mr *MockTaskMockRecorder) GetQueueType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueType\", reflect.TypeOf((*MockTask)(nil).GetQueueType))\n}\n\n// GetRunID mocks base method.\nfunc (m *MockTask) GetRunID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRunID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetRunID indicates an expected call of GetRunID.\nfunc (mr *MockTaskMockRecorder) GetRunID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRunID\", reflect.TypeOf((*MockTask)(nil).GetRunID))\n}\n\n// GetShard mocks base method.\nfunc (m *MockTask) GetShard() shard.Context {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShard\")\n\tret0, _ := ret[0].(shard.Context)\n\treturn ret0\n}\n\n// GetShard indicates an expected call of GetShard.\nfunc (mr *MockTaskMockRecorder) GetShard() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShard\", reflect.TypeOf((*MockTask)(nil).GetShard))\n}\n\n// GetTaskCategory mocks base method.\nfunc (m *MockTask) GetTaskCategory() persistence.HistoryTaskCategory {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskCategory\")\n\tret0, _ := ret[0].(persistence.HistoryTaskCategory)\n\treturn ret0\n}\n\n// GetTaskCategory indicates an expected call of GetTaskCategory.\nfunc (mr *MockTaskMockRecorder) GetTaskCategory() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskCategory\", reflect.TypeOf((*MockTask)(nil).GetTaskCategory))\n}\n\n// GetTaskID mocks base method.\nfunc (m *MockTask) GetTaskID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetTaskID indicates an expected call of GetTaskID.\nfunc (mr *MockTaskMockRecorder) GetTaskID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskID\", reflect.TypeOf((*MockTask)(nil).GetTaskID))\n}\n\n// GetTaskKey mocks base method.\nfunc (m *MockTask) GetTaskKey() persistence.HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskKey\")\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\treturn ret0\n}\n\n// GetTaskKey indicates an expected call of GetTaskKey.\nfunc (mr *MockTaskMockRecorder) GetTaskKey() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskKey\", reflect.TypeOf((*MockTask)(nil).GetTaskKey))\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockTask) GetTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockTaskMockRecorder) GetTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockTask)(nil).GetTaskList))\n}\n\n// GetTaskType mocks base method.\nfunc (m *MockTask) GetTaskType() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskType\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetTaskType indicates an expected call of GetTaskType.\nfunc (mr *MockTaskMockRecorder) GetTaskType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskType\", reflect.TypeOf((*MockTask)(nil).GetTaskType))\n}\n\n// GetVersion mocks base method.\nfunc (m *MockTask) GetVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetVersion indicates an expected call of GetVersion.\nfunc (mr *MockTaskMockRecorder) GetVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVersion\", reflect.TypeOf((*MockTask)(nil).GetVersion))\n}\n\n// GetVisibilityTimestamp mocks base method.\nfunc (m *MockTask) GetVisibilityTimestamp() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityTimestamp\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// GetVisibilityTimestamp indicates an expected call of GetVisibilityTimestamp.\nfunc (mr *MockTaskMockRecorder) GetVisibilityTimestamp() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityTimestamp\", reflect.TypeOf((*MockTask)(nil).GetVisibilityTimestamp))\n}\n\n// GetWorkflowID mocks base method.\nfunc (m *MockTask) GetWorkflowID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetWorkflowID indicates an expected call of GetWorkflowID.\nfunc (mr *MockTaskMockRecorder) GetWorkflowID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowID\", reflect.TypeOf((*MockTask)(nil).GetWorkflowID))\n}\n\n// HandleErr mocks base method.\nfunc (m *MockTask) HandleErr(err error) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleErr\", err)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// HandleErr indicates an expected call of HandleErr.\nfunc (mr *MockTaskMockRecorder) HandleErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleErr\", reflect.TypeOf((*MockTask)(nil).HandleErr), err)\n}\n\n// Nack mocks base method.\nfunc (m *MockTask) Nack(err error) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Nack\", err)\n}\n\n// Nack indicates an expected call of Nack.\nfunc (mr *MockTaskMockRecorder) Nack(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Nack\", reflect.TypeOf((*MockTask)(nil).Nack), err)\n}\n\n// Priority mocks base method.\nfunc (m *MockTask) Priority() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Priority\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Priority indicates an expected call of Priority.\nfunc (mr *MockTaskMockRecorder) Priority() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Priority\", reflect.TypeOf((*MockTask)(nil).Priority))\n}\n\n// RetryErr mocks base method.\nfunc (m *MockTask) RetryErr(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RetryErr\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// RetryErr indicates an expected call of RetryErr.\nfunc (mr *MockTaskMockRecorder) RetryErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RetryErr\", reflect.TypeOf((*MockTask)(nil).RetryErr), err)\n}\n\n// SetInitialSubmitTime mocks base method.\nfunc (m *MockTask) SetInitialSubmitTime(arg0 time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetInitialSubmitTime\", arg0)\n}\n\n// SetInitialSubmitTime indicates an expected call of SetInitialSubmitTime.\nfunc (mr *MockTaskMockRecorder) SetInitialSubmitTime(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetInitialSubmitTime\", reflect.TypeOf((*MockTask)(nil).SetInitialSubmitTime), arg0)\n}\n\n// SetPriority mocks base method.\nfunc (m *MockTask) SetPriority(arg0 int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetPriority\", arg0)\n}\n\n// SetPriority indicates an expected call of SetPriority.\nfunc (mr *MockTaskMockRecorder) SetPriority(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetPriority\", reflect.TypeOf((*MockTask)(nil).SetPriority), arg0)\n}\n\n// SetTaskID mocks base method.\nfunc (m *MockTask) SetTaskID(id int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetTaskID\", id)\n}\n\n// SetTaskID indicates an expected call of SetTaskID.\nfunc (mr *MockTaskMockRecorder) SetTaskID(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetTaskID\", reflect.TypeOf((*MockTask)(nil).SetTaskID), id)\n}\n\n// SetVersion mocks base method.\nfunc (m *MockTask) SetVersion(version int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVersion\", version)\n}\n\n// SetVersion indicates an expected call of SetVersion.\nfunc (mr *MockTaskMockRecorder) SetVersion(version any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVersion\", reflect.TypeOf((*MockTask)(nil).SetVersion), version)\n}\n\n// SetVisibilityTimestamp mocks base method.\nfunc (m *MockTask) SetVisibilityTimestamp(timestamp time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVisibilityTimestamp\", timestamp)\n}\n\n// SetVisibilityTimestamp indicates an expected call of SetVisibilityTimestamp.\nfunc (mr *MockTaskMockRecorder) SetVisibilityTimestamp(timestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVisibilityTimestamp\", reflect.TypeOf((*MockTask)(nil).SetVisibilityTimestamp), timestamp)\n}\n\n// State mocks base method.\nfunc (m *MockTask) State() task.State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"State\")\n\tret0, _ := ret[0].(task.State)\n\treturn ret0\n}\n\n// State indicates an expected call of State.\nfunc (mr *MockTaskMockRecorder) State() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"State\", reflect.TypeOf((*MockTask)(nil).State))\n}\n\n// ToInternalReplicationTaskInfo mocks base method.\nfunc (m *MockTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToInternalReplicationTaskInfo\")\n\tret0, _ := ret[0].(*types.ReplicationTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToInternalReplicationTaskInfo indicates an expected call of ToInternalReplicationTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToInternalReplicationTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToInternalReplicationTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToInternalReplicationTaskInfo))\n}\n\n// ToTimerTaskInfo mocks base method.\nfunc (m *MockTask) ToTimerTaskInfo() (*persistence.TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTimerTaskInfo\")\n\tret0, _ := ret[0].(*persistence.TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTimerTaskInfo indicates an expected call of ToTimerTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToTimerTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTimerTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToTimerTaskInfo))\n}\n\n// ToTransferTaskInfo mocks base method.\nfunc (m *MockTask) ToTransferTaskInfo() (*persistence.TransferTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTransferTaskInfo\")\n\tret0, _ := ret[0].(*persistence.TransferTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTransferTaskInfo indicates an expected call of ToTransferTaskInfo.\nfunc (mr *MockTaskMockRecorder) ToTransferTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTransferTaskInfo\", reflect.TypeOf((*MockTask)(nil).ToTransferTaskInfo))\n}\n\n// MockCrossClusterTask is a mock of CrossClusterTask interface.\ntype MockCrossClusterTask struct {\n\tctrl     *gomock.Controller\n\trecorder *MockCrossClusterTaskMockRecorder\n\tisgomock struct{}\n}\n\n// MockCrossClusterTaskMockRecorder is the mock recorder for MockCrossClusterTask.\ntype MockCrossClusterTaskMockRecorder struct {\n\tmock *MockCrossClusterTask\n}\n\n// NewMockCrossClusterTask creates a new mock instance.\nfunc NewMockCrossClusterTask(ctrl *gomock.Controller) *MockCrossClusterTask {\n\tmock := &MockCrossClusterTask{ctrl: ctrl}\n\tmock.recorder = &MockCrossClusterTaskMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockCrossClusterTask) EXPECT() *MockCrossClusterTaskMockRecorder {\n\treturn m.recorder\n}\n\n// Ack mocks base method.\nfunc (m *MockCrossClusterTask) Ack() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Ack\")\n}\n\n// Ack indicates an expected call of Ack.\nfunc (mr *MockCrossClusterTaskMockRecorder) Ack() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ack\", reflect.TypeOf((*MockCrossClusterTask)(nil).Ack))\n}\n\n// ByteSize mocks base method.\nfunc (m *MockCrossClusterTask) ByteSize() uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ByteSize\")\n\tret0, _ := ret[0].(uint64)\n\treturn ret0\n}\n\n// ByteSize indicates an expected call of ByteSize.\nfunc (mr *MockCrossClusterTaskMockRecorder) ByteSize() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ByteSize\", reflect.TypeOf((*MockCrossClusterTask)(nil).ByteSize))\n}\n\n// Cancel mocks base method.\nfunc (m *MockCrossClusterTask) Cancel() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Cancel\")\n}\n\n// Cancel indicates an expected call of Cancel.\nfunc (mr *MockCrossClusterTaskMockRecorder) Cancel() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Cancel\", reflect.TypeOf((*MockCrossClusterTask)(nil).Cancel))\n}\n\n// Execute mocks base method.\nfunc (m *MockCrossClusterTask) Execute() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockCrossClusterTaskMockRecorder) Execute() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockCrossClusterTask)(nil).Execute))\n}\n\n// GetAttempt mocks base method.\nfunc (m *MockCrossClusterTask) GetAttempt() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAttempt\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetAttempt indicates an expected call of GetAttempt.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetAttempt() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAttempt\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetAttempt))\n}\n\n// GetCrossClusterRequest mocks base method.\nfunc (m *MockCrossClusterTask) GetCrossClusterRequest() (*types.CrossClusterTaskRequest, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetCrossClusterRequest\")\n\tret0, _ := ret[0].(*types.CrossClusterTaskRequest)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetCrossClusterRequest indicates an expected call of GetCrossClusterRequest.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetCrossClusterRequest() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetCrossClusterRequest\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetCrossClusterRequest))\n}\n\n// GetDomainID mocks base method.\nfunc (m *MockCrossClusterTask) GetDomainID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetDomainID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetDomainID indicates an expected call of GetDomainID.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetDomainID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetDomainID\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetDomainID))\n}\n\n// GetInfo mocks base method.\nfunc (m *MockCrossClusterTask) GetInfo() persistence.Task {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetInfo\")\n\tret0, _ := ret[0].(persistence.Task)\n\treturn ret0\n}\n\n// GetInfo indicates an expected call of GetInfo.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetInfo\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetInfo))\n}\n\n// GetOriginalTaskList mocks base method.\nfunc (m *MockCrossClusterTask) GetOriginalTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetOriginalTaskList indicates an expected call of GetOriginalTaskList.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetOriginalTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskList\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetOriginalTaskList))\n}\n\n// GetOriginalTaskListKind mocks base method.\nfunc (m *MockCrossClusterTask) GetOriginalTaskListKind() types.TaskListKind {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetOriginalTaskListKind\")\n\tret0, _ := ret[0].(types.TaskListKind)\n\treturn ret0\n}\n\n// GetOriginalTaskListKind indicates an expected call of GetOriginalTaskListKind.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetOriginalTaskListKind() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetOriginalTaskListKind\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetOriginalTaskListKind))\n}\n\n// GetQueueType mocks base method.\nfunc (m *MockCrossClusterTask) GetQueueType() QueueType {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetQueueType\")\n\tret0, _ := ret[0].(QueueType)\n\treturn ret0\n}\n\n// GetQueueType indicates an expected call of GetQueueType.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetQueueType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetQueueType\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetQueueType))\n}\n\n// GetRunID mocks base method.\nfunc (m *MockCrossClusterTask) GetRunID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetRunID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetRunID indicates an expected call of GetRunID.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetRunID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetRunID\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetRunID))\n}\n\n// GetShard mocks base method.\nfunc (m *MockCrossClusterTask) GetShard() shard.Context {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShard\")\n\tret0, _ := ret[0].(shard.Context)\n\treturn ret0\n}\n\n// GetShard indicates an expected call of GetShard.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetShard() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShard\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetShard))\n}\n\n// GetTaskCategory mocks base method.\nfunc (m *MockCrossClusterTask) GetTaskCategory() persistence.HistoryTaskCategory {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskCategory\")\n\tret0, _ := ret[0].(persistence.HistoryTaskCategory)\n\treturn ret0\n}\n\n// GetTaskCategory indicates an expected call of GetTaskCategory.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetTaskCategory() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskCategory\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetTaskCategory))\n}\n\n// GetTaskID mocks base method.\nfunc (m *MockCrossClusterTask) GetTaskID() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskID\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetTaskID indicates an expected call of GetTaskID.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetTaskID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskID\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetTaskID))\n}\n\n// GetTaskKey mocks base method.\nfunc (m *MockCrossClusterTask) GetTaskKey() persistence.HistoryTaskKey {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskKey\")\n\tret0, _ := ret[0].(persistence.HistoryTaskKey)\n\treturn ret0\n}\n\n// GetTaskKey indicates an expected call of GetTaskKey.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetTaskKey() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskKey\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetTaskKey))\n}\n\n// GetTaskList mocks base method.\nfunc (m *MockCrossClusterTask) GetTaskList() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskList\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetTaskList indicates an expected call of GetTaskList.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetTaskList() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskList\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetTaskList))\n}\n\n// GetTaskType mocks base method.\nfunc (m *MockCrossClusterTask) GetTaskType() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskType\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// GetTaskType indicates an expected call of GetTaskType.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetTaskType() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskType\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetTaskType))\n}\n\n// GetVersion mocks base method.\nfunc (m *MockCrossClusterTask) GetVersion() int64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVersion\")\n\tret0, _ := ret[0].(int64)\n\treturn ret0\n}\n\n// GetVersion indicates an expected call of GetVersion.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVersion\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetVersion))\n}\n\n// GetVisibilityTimestamp mocks base method.\nfunc (m *MockCrossClusterTask) GetVisibilityTimestamp() time.Time {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetVisibilityTimestamp\")\n\tret0, _ := ret[0].(time.Time)\n\treturn ret0\n}\n\n// GetVisibilityTimestamp indicates an expected call of GetVisibilityTimestamp.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetVisibilityTimestamp() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetVisibilityTimestamp\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetVisibilityTimestamp))\n}\n\n// GetWorkflowID mocks base method.\nfunc (m *MockCrossClusterTask) GetWorkflowID() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetWorkflowID\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetWorkflowID indicates an expected call of GetWorkflowID.\nfunc (mr *MockCrossClusterTaskMockRecorder) GetWorkflowID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetWorkflowID\", reflect.TypeOf((*MockCrossClusterTask)(nil).GetWorkflowID))\n}\n\n// HandleErr mocks base method.\nfunc (m *MockCrossClusterTask) HandleErr(err error) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HandleErr\", err)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// HandleErr indicates an expected call of HandleErr.\nfunc (mr *MockCrossClusterTaskMockRecorder) HandleErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HandleErr\", reflect.TypeOf((*MockCrossClusterTask)(nil).HandleErr), err)\n}\n\n// IsReadyForPoll mocks base method.\nfunc (m *MockCrossClusterTask) IsReadyForPoll() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsReadyForPoll\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsReadyForPoll indicates an expected call of IsReadyForPoll.\nfunc (mr *MockCrossClusterTaskMockRecorder) IsReadyForPoll() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsReadyForPoll\", reflect.TypeOf((*MockCrossClusterTask)(nil).IsReadyForPoll))\n}\n\n// IsValid mocks base method.\nfunc (m *MockCrossClusterTask) IsValid() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsValid\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsValid indicates an expected call of IsValid.\nfunc (mr *MockCrossClusterTaskMockRecorder) IsValid() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsValid\", reflect.TypeOf((*MockCrossClusterTask)(nil).IsValid))\n}\n\n// Nack mocks base method.\nfunc (m *MockCrossClusterTask) Nack(err error) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Nack\", err)\n}\n\n// Nack indicates an expected call of Nack.\nfunc (mr *MockCrossClusterTaskMockRecorder) Nack(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Nack\", reflect.TypeOf((*MockCrossClusterTask)(nil).Nack), err)\n}\n\n// Priority mocks base method.\nfunc (m *MockCrossClusterTask) Priority() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Priority\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Priority indicates an expected call of Priority.\nfunc (mr *MockCrossClusterTaskMockRecorder) Priority() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Priority\", reflect.TypeOf((*MockCrossClusterTask)(nil).Priority))\n}\n\n// RecordResponse mocks base method.\nfunc (m *MockCrossClusterTask) RecordResponse(arg0 *types.CrossClusterTaskResponse) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordResponse\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordResponse indicates an expected call of RecordResponse.\nfunc (mr *MockCrossClusterTaskMockRecorder) RecordResponse(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordResponse\", reflect.TypeOf((*MockCrossClusterTask)(nil).RecordResponse), arg0)\n}\n\n// RetryErr mocks base method.\nfunc (m *MockCrossClusterTask) RetryErr(err error) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RetryErr\", err)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// RetryErr indicates an expected call of RetryErr.\nfunc (mr *MockCrossClusterTaskMockRecorder) RetryErr(err any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RetryErr\", reflect.TypeOf((*MockCrossClusterTask)(nil).RetryErr), err)\n}\n\n// SetInitialSubmitTime mocks base method.\nfunc (m *MockCrossClusterTask) SetInitialSubmitTime(arg0 time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetInitialSubmitTime\", arg0)\n}\n\n// SetInitialSubmitTime indicates an expected call of SetInitialSubmitTime.\nfunc (mr *MockCrossClusterTaskMockRecorder) SetInitialSubmitTime(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetInitialSubmitTime\", reflect.TypeOf((*MockCrossClusterTask)(nil).SetInitialSubmitTime), arg0)\n}\n\n// SetPriority mocks base method.\nfunc (m *MockCrossClusterTask) SetPriority(arg0 int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetPriority\", arg0)\n}\n\n// SetPriority indicates an expected call of SetPriority.\nfunc (mr *MockCrossClusterTaskMockRecorder) SetPriority(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetPriority\", reflect.TypeOf((*MockCrossClusterTask)(nil).SetPriority), arg0)\n}\n\n// SetTaskID mocks base method.\nfunc (m *MockCrossClusterTask) SetTaskID(id int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetTaskID\", id)\n}\n\n// SetTaskID indicates an expected call of SetTaskID.\nfunc (mr *MockCrossClusterTaskMockRecorder) SetTaskID(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetTaskID\", reflect.TypeOf((*MockCrossClusterTask)(nil).SetTaskID), id)\n}\n\n// SetVersion mocks base method.\nfunc (m *MockCrossClusterTask) SetVersion(version int64) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVersion\", version)\n}\n\n// SetVersion indicates an expected call of SetVersion.\nfunc (mr *MockCrossClusterTaskMockRecorder) SetVersion(version any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVersion\", reflect.TypeOf((*MockCrossClusterTask)(nil).SetVersion), version)\n}\n\n// SetVisibilityTimestamp mocks base method.\nfunc (m *MockCrossClusterTask) SetVisibilityTimestamp(timestamp time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetVisibilityTimestamp\", timestamp)\n}\n\n// SetVisibilityTimestamp indicates an expected call of SetVisibilityTimestamp.\nfunc (mr *MockCrossClusterTaskMockRecorder) SetVisibilityTimestamp(timestamp any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetVisibilityTimestamp\", reflect.TypeOf((*MockCrossClusterTask)(nil).SetVisibilityTimestamp), timestamp)\n}\n\n// State mocks base method.\nfunc (m *MockCrossClusterTask) State() task.State {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"State\")\n\tret0, _ := ret[0].(task.State)\n\treturn ret0\n}\n\n// State indicates an expected call of State.\nfunc (mr *MockCrossClusterTaskMockRecorder) State() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"State\", reflect.TypeOf((*MockCrossClusterTask)(nil).State))\n}\n\n// ToInternalReplicationTaskInfo mocks base method.\nfunc (m *MockCrossClusterTask) ToInternalReplicationTaskInfo() (*types.ReplicationTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToInternalReplicationTaskInfo\")\n\tret0, _ := ret[0].(*types.ReplicationTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToInternalReplicationTaskInfo indicates an expected call of ToInternalReplicationTaskInfo.\nfunc (mr *MockCrossClusterTaskMockRecorder) ToInternalReplicationTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToInternalReplicationTaskInfo\", reflect.TypeOf((*MockCrossClusterTask)(nil).ToInternalReplicationTaskInfo))\n}\n\n// ToTimerTaskInfo mocks base method.\nfunc (m *MockCrossClusterTask) ToTimerTaskInfo() (*persistence.TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTimerTaskInfo\")\n\tret0, _ := ret[0].(*persistence.TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTimerTaskInfo indicates an expected call of ToTimerTaskInfo.\nfunc (mr *MockCrossClusterTaskMockRecorder) ToTimerTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTimerTaskInfo\", reflect.TypeOf((*MockCrossClusterTask)(nil).ToTimerTaskInfo))\n}\n\n// ToTransferTaskInfo mocks base method.\nfunc (m *MockCrossClusterTask) ToTransferTaskInfo() (*persistence.TransferTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ToTransferTaskInfo\")\n\tret0, _ := ret[0].(*persistence.TransferTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ToTransferTaskInfo indicates an expected call of ToTransferTaskInfo.\nfunc (mr *MockCrossClusterTaskMockRecorder) ToTransferTaskInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ToTransferTaskInfo\", reflect.TypeOf((*MockCrossClusterTask)(nil).ToTransferTaskInfo))\n}\n\n// MockKey is a mock of Key interface.\ntype MockKey struct {\n\tctrl     *gomock.Controller\n\trecorder *MockKeyMockRecorder\n\tisgomock struct{}\n}\n\n// MockKeyMockRecorder is the mock recorder for MockKey.\ntype MockKeyMockRecorder struct {\n\tmock *MockKey\n}\n\n// NewMockKey creates a new mock instance.\nfunc NewMockKey(ctrl *gomock.Controller) *MockKey {\n\tmock := &MockKey{ctrl: ctrl}\n\tmock.recorder = &MockKeyMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockKey) EXPECT() *MockKeyMockRecorder {\n\treturn m.recorder\n}\n\n// Less mocks base method.\nfunc (m *MockKey) Less(arg0 Key) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Less\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Less indicates an expected call of Less.\nfunc (mr *MockKeyMockRecorder) Less(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Less\", reflect.TypeOf((*MockKey)(nil).Less), arg0)\n}\n\n// MockExecutor is a mock of Executor interface.\ntype MockExecutor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutorMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutorMockRecorder is the mock recorder for MockExecutor.\ntype MockExecutorMockRecorder struct {\n\tmock *MockExecutor\n}\n\n// NewMockExecutor creates a new mock instance.\nfunc NewMockExecutor(ctrl *gomock.Controller) *MockExecutor {\n\tmock := &MockExecutor{ctrl: ctrl}\n\tmock.recorder = &MockExecutorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutor) EXPECT() *MockExecutorMockRecorder {\n\treturn m.recorder\n}\n\n// Execute mocks base method.\nfunc (m *MockExecutor) Execute(task Task) (ExecuteResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Execute\", task)\n\tret0, _ := ret[0].(ExecuteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Execute indicates an expected call of Execute.\nfunc (mr *MockExecutorMockRecorder) Execute(task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Execute\", reflect.TypeOf((*MockExecutor)(nil).Execute), task)\n}\n\n// Stop mocks base method.\nfunc (m *MockExecutor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockExecutorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockExecutor)(nil).Stop))\n}\n\n// MockPriorityAssigner is a mock of PriorityAssigner interface.\ntype MockPriorityAssigner struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPriorityAssignerMockRecorder\n\tisgomock struct{}\n}\n\n// MockPriorityAssignerMockRecorder is the mock recorder for MockPriorityAssigner.\ntype MockPriorityAssignerMockRecorder struct {\n\tmock *MockPriorityAssigner\n}\n\n// NewMockPriorityAssigner creates a new mock instance.\nfunc NewMockPriorityAssigner(ctrl *gomock.Controller) *MockPriorityAssigner {\n\tmock := &MockPriorityAssigner{ctrl: ctrl}\n\tmock.recorder = &MockPriorityAssignerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPriorityAssigner) EXPECT() *MockPriorityAssignerMockRecorder {\n\treturn m.recorder\n}\n\n// Assign mocks base method.\nfunc (m *MockPriorityAssigner) Assign(arg0 Task) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Assign\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Assign indicates an expected call of Assign.\nfunc (mr *MockPriorityAssignerMockRecorder) Assign(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Assign\", reflect.TypeOf((*MockPriorityAssigner)(nil).Assign), arg0)\n}\n\n// MockProcessor is a mock of Processor interface.\ntype MockProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessorMockRecorder is the mock recorder for MockProcessor.\ntype MockProcessorMockRecorder struct {\n\tmock *MockProcessor\n}\n\n// NewMockProcessor creates a new mock instance.\nfunc NewMockProcessor(ctrl *gomock.Controller) *MockProcessor {\n\tmock := &MockProcessor{ctrl: ctrl}\n\tmock.recorder = &MockProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessor) EXPECT() *MockProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// Start mocks base method.\nfunc (m *MockProcessor) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockProcessorMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockProcessor)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockProcessor)(nil).Stop))\n}\n\n// Submit mocks base method.\nfunc (m *MockProcessor) Submit(arg0 Task) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Submit\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Submit indicates an expected call of Submit.\nfunc (mr *MockProcessorMockRecorder) Submit(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Submit\", reflect.TypeOf((*MockProcessor)(nil).Submit), arg0)\n}\n\n// TrySubmit mocks base method.\nfunc (m *MockProcessor) TrySubmit(arg0 Task) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TrySubmit\", arg0)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TrySubmit indicates an expected call of TrySubmit.\nfunc (mr *MockProcessorMockRecorder) TrySubmit(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TrySubmit\", reflect.TypeOf((*MockProcessor)(nil).TrySubmit), arg0)\n}\n\n// MockRedispatcher is a mock of Redispatcher interface.\ntype MockRedispatcher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockRedispatcherMockRecorder\n\tisgomock struct{}\n}\n\n// MockRedispatcherMockRecorder is the mock recorder for MockRedispatcher.\ntype MockRedispatcherMockRecorder struct {\n\tmock *MockRedispatcher\n}\n\n// NewMockRedispatcher creates a new mock instance.\nfunc NewMockRedispatcher(ctrl *gomock.Controller) *MockRedispatcher {\n\tmock := &MockRedispatcher{ctrl: ctrl}\n\tmock.recorder = &MockRedispatcherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRedispatcher) EXPECT() *MockRedispatcherMockRecorder {\n\treturn m.recorder\n}\n\n// AddTask mocks base method.\nfunc (m *MockRedispatcher) AddTask(arg0 Task) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"AddTask\", arg0)\n}\n\n// AddTask indicates an expected call of AddTask.\nfunc (mr *MockRedispatcherMockRecorder) AddTask(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTask\", reflect.TypeOf((*MockRedispatcher)(nil).AddTask), arg0)\n}\n\n// Redispatch mocks base method.\nfunc (m *MockRedispatcher) Redispatch(targetSize int) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Redispatch\", targetSize)\n}\n\n// Redispatch indicates an expected call of Redispatch.\nfunc (mr *MockRedispatcherMockRecorder) Redispatch(targetSize any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Redispatch\", reflect.TypeOf((*MockRedispatcher)(nil).Redispatch), targetSize)\n}\n\n// RedispatchTask mocks base method.\nfunc (m *MockRedispatcher) RedispatchTask(arg0 Task, arg1 time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RedispatchTask\", arg0, arg1)\n}\n\n// RedispatchTask indicates an expected call of RedispatchTask.\nfunc (mr *MockRedispatcherMockRecorder) RedispatchTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RedispatchTask\", reflect.TypeOf((*MockRedispatcher)(nil).RedispatchTask), arg0, arg1)\n}\n\n// Size mocks base method.\nfunc (m *MockRedispatcher) Size() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Size\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Size indicates an expected call of Size.\nfunc (mr *MockRedispatcherMockRecorder) Size() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Size\", reflect.TypeOf((*MockRedispatcher)(nil).Size))\n}\n\n// Start mocks base method.\nfunc (m *MockRedispatcher) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockRedispatcherMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockRedispatcher)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockRedispatcher) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockRedispatcherMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockRedispatcher)(nil).Stop))\n}\n\n// MockRescheduler is a mock of Rescheduler interface.\ntype MockRescheduler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockReschedulerMockRecorder\n\tisgomock struct{}\n}\n\n// MockReschedulerMockRecorder is the mock recorder for MockRescheduler.\ntype MockReschedulerMockRecorder struct {\n\tmock *MockRescheduler\n}\n\n// NewMockRescheduler creates a new mock instance.\nfunc NewMockRescheduler(ctrl *gomock.Controller) *MockRescheduler {\n\tmock := &MockRescheduler{ctrl: ctrl}\n\tmock.recorder = &MockReschedulerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRescheduler) EXPECT() *MockReschedulerMockRecorder {\n\treturn m.recorder\n}\n\n// RescheduleDomains mocks base method.\nfunc (m *MockRescheduler) RescheduleDomains(domainIDs map[string]struct{}) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RescheduleDomains\", domainIDs)\n}\n\n// RescheduleDomains indicates an expected call of RescheduleDomains.\nfunc (mr *MockReschedulerMockRecorder) RescheduleDomains(domainIDs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RescheduleDomains\", reflect.TypeOf((*MockRescheduler)(nil).RescheduleDomains), domainIDs)\n}\n\n// RescheduleTask mocks base method.\nfunc (m *MockRescheduler) RescheduleTask(arg0 Task, arg1 time.Time) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RescheduleTask\", arg0, arg1)\n}\n\n// RescheduleTask indicates an expected call of RescheduleTask.\nfunc (mr *MockReschedulerMockRecorder) RescheduleTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RescheduleTask\", reflect.TypeOf((*MockRescheduler)(nil).RescheduleTask), arg0, arg1)\n}\n\n// Size mocks base method.\nfunc (m *MockRescheduler) Size() int {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Size\")\n\tret0, _ := ret[0].(int)\n\treturn ret0\n}\n\n// Size indicates an expected call of Size.\nfunc (mr *MockReschedulerMockRecorder) Size() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Size\", reflect.TypeOf((*MockRescheduler)(nil).Size))\n}\n\n// Start mocks base method.\nfunc (m *MockRescheduler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockReschedulerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockRescheduler)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockRescheduler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockReschedulerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockRescheduler)(nil).Stop))\n}\n\n// MockFetcher is a mock of Fetcher interface.\ntype MockFetcher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFetcherMockRecorder\n\tisgomock struct{}\n}\n\n// MockFetcherMockRecorder is the mock recorder for MockFetcher.\ntype MockFetcherMockRecorder struct {\n\tmock *MockFetcher\n}\n\n// NewMockFetcher creates a new mock instance.\nfunc NewMockFetcher(ctrl *gomock.Controller) *MockFetcher {\n\tmock := &MockFetcher{ctrl: ctrl}\n\tmock.recorder = &MockFetcherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFetcher) EXPECT() *MockFetcherMockRecorder {\n\treturn m.recorder\n}\n\n// Fetch mocks base method.\nfunc (m *MockFetcher) Fetch(shardID int, fetchParams ...any) future.Future {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{shardID}\n\tfor _, a := range fetchParams {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Fetch\", varargs...)\n\tret0, _ := ret[0].(future.Future)\n\treturn ret0\n}\n\n// Fetch indicates an expected call of Fetch.\nfunc (mr *MockFetcherMockRecorder) Fetch(shardID any, fetchParams ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{shardID}, fetchParams...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Fetch\", reflect.TypeOf((*MockFetcher)(nil).Fetch), varargs...)\n}\n\n// GetSourceCluster mocks base method.\nfunc (m *MockFetcher) GetSourceCluster() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetSourceCluster\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetSourceCluster indicates an expected call of GetSourceCluster.\nfunc (mr *MockFetcherMockRecorder) GetSourceCluster() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetSourceCluster\", reflect.TypeOf((*MockFetcher)(nil).GetSourceCluster))\n}\n\n// Start mocks base method.\nfunc (m *MockFetcher) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockFetcherMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockFetcher)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockFetcher) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockFetcherMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockFetcher)(nil).Stop))\n}\n"
  },
  {
    "path": "service/history/task/priority_assigner.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\tdynamicquotas \"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\nvar (\n\thighTaskPriority    = constants.GetTaskPriority(constants.HighPriorityClass, constants.DefaultPrioritySubclass)\n\tdefaultTaskPriority = constants.GetTaskPriority(constants.DefaultPriorityClass, constants.DefaultPrioritySubclass)\n\tlowTaskPriority     = constants.GetTaskPriority(constants.LowPriorityClass, constants.DefaultPrioritySubclass)\n)\n\ntype priorityAssignerImpl struct {\n\tsync.RWMutex\n\n\tcurrentClusterName string\n\tdomainCache        cache.DomainCache\n\tconfig             *config.Config\n\tlogger             log.Logger\n\tscope              metrics.Scope\n\trateLimiters       *quotas.Collection\n\tactiveClusterMgr   activecluster.Manager\n}\n\n// NewPriorityAssigner creates a new task priority assigner\nfunc NewPriorityAssigner(\n\tcurrentClusterName string,\n\tdomainCache cache.DomainCache,\n\tactiveClusterMgr activecluster.Manager,\n\tlogger log.Logger,\n\tmetricClient metrics.Client,\n\tconfig *config.Config,\n) PriorityAssigner {\n\treturn &priorityAssignerImpl{\n\t\tcurrentClusterName: currentClusterName,\n\t\tdomainCache:        domainCache,\n\t\tactiveClusterMgr:   activeClusterMgr,\n\t\tconfig:             config,\n\t\tlogger:             logger,\n\t\tscope:              metricClient.Scope(metrics.TaskPriorityAssignerScope),\n\t\trateLimiters: quotas.NewCollection(dynamicquotas.NewSimpleDynamicRateLimiterFactory(\n\t\t\tconfig.TaskProcessRPS,\n\t\t)),\n\t}\n}\n\nfunc (a *priorityAssignerImpl) Assign(queueTask Task) error {\n\tif priority := queueTask.Priority(); priority != noPriority {\n\t\tif priority != lowTaskPriority && queueTask.GetAttempt() > a.config.TaskCriticalRetryCount() {\n\t\t\t// automatically lower the priority if task attempt exceeds certain threshold\n\t\t\tqueueTask.SetPriority(lowTaskPriority)\n\t\t}\n\t\treturn nil\n\t}\n\n\tqueueType := queueTask.GetQueueType()\n\n\tif queueType == QueueTypeReplication {\n\t\tqueueTask.SetPriority(lowTaskPriority)\n\t\treturn nil\n\t}\n\n\t// timer, transfer or cross cluster task, first check if task is active or not and if domain is active or not\n\tisActiveTask := queueType == QueueTypeActiveTimer || queueType == QueueTypeActiveTransfer\n\tdomainName, isActiveDomain, err := a.getDomainInfo(queueTask.GetDomainID(), queueTask.GetWorkflowID(), queueTask.GetRunID())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// there are four cases here:\n\t// 1. active task for active domain\n\t// 2. active task for standby domain\n\t// 3. standby task for active domain\n\t// 4. standby task for standby domain\n\n\tif !isActiveTask && !isActiveDomain {\n\t\t// only assign low priority to tasks in the fourth case\n\t\tqueueTask.SetPriority(lowTaskPriority)\n\t\treturn nil\n\t}\n\n\t// for case 1 we should give the task a high priority\n\t// for case 2 and 3 the task will be a no-op in most cases, also give it a high priority so that\n\t// it can be quickly verified/acked and won't prevent the ack level in the processor from advancing\n\t// (especially for active processor)\n\tif !a.rateLimiters.For(domainName).Allow() {\n\t\tqueueTask.SetPriority(defaultTaskPriority)\n\t\ttaggedScope := a.scope.Tagged(metrics.DomainTag(domainName))\n\t\tswitch queueType {\n\t\tcase QueueTypeActiveTransfer, QueueTypeStandbyTransfer:\n\t\t\ttaggedScope.IncCounter(metrics.TransferTaskThrottledCounter)\n\t\tcase QueueTypeActiveTimer, QueueTypeStandbyTimer:\n\t\t\ttaggedScope.IncCounter(metrics.TimerTaskThrottledCounter)\n\t\t}\n\t\treturn nil\n\t}\n\n\tqueueTask.SetPriority(highTaskPriority)\n\treturn nil\n}\n\n// getDomainInfo returns three pieces of information:\n//  1. domain name\n//  2. if domain is active\n//  3. error, if any\nfunc (a *priorityAssignerImpl) getDomainInfo(domainID, wfID, rID string) (string, bool, error) {\n\tdomainEntry, err := a.domainCache.GetDomainByID(domainID)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\ta.logger.Warn(\"Cannot find domain\", tag.WorkflowDomainID(domainID))\n\t\t\treturn \"\", false, err\n\t\t}\n\t\t// it is possible that the domain is deleted\n\t\t// we should treat that domain as active\n\t\ta.logger.Warn(\"Cannot find domain, treat as active \", tag.WorkflowDomainID(domainID))\n\t\treturn \"\", true, nil\n\t}\n\n\tactiveClusterInfo, err := a.activeClusterMgr.GetActiveClusterInfoByWorkflow(context.Background(), domainID, wfID, rID)\n\tif err != nil {\n\t\ta.logger.Warn(\"Failed to get active cluster info\", tag.WorkflowDomainID(domainID), tag.Error(err))\n\t\treturn \"\", true, nil\n\t}\n\n\tif activeClusterInfo.ActiveClusterName != a.currentClusterName {\n\t\treturn domainEntry.GetInfo().Name, false, nil\n\t}\n\treturn domainEntry.GetInfo().Name, true, nil\n}\n"
  },
  {
    "path": "service/history/task/priority_assigner_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\ntype (\n\ttaskPriorityAssignerSuite struct {\n\t\t*require.Assertions\n\t\tsuite.Suite\n\n\t\tcontroller           *gomock.Controller\n\t\tmockDomainCache      *cache.MockDomainCache\n\t\tmockActiveClusterMgr *activecluster.MockManager\n\n\t\tconfig             *config.Config\n\t\tpriorityAssigner   *priorityAssignerImpl\n\t\ttestTaskProcessRPS int\n\t}\n)\n\nfunc TestTaskPriorityAssignerSuite(t *testing.T) {\n\ts := new(taskPriorityAssignerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *taskPriorityAssignerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockActiveClusterMgr = activecluster.NewMockManager(s.controller)\n\n\ts.testTaskProcessRPS = 10\n\tclient := dynamicconfig.NewInMemoryClient()\n\terr := client.UpdateValue(dynamicproperties.TaskProcessRPS, s.testTaskProcessRPS)\n\ts.NoError(err)\n\tdc := dynamicconfig.NewCollection(client, log.NewNoop())\n\ts.config = config.NewForTest()\n\ts.config.TaskProcessRPS = dc.GetIntPropertyFilteredByDomain(dynamicproperties.TaskProcessRPS)\n\n\ts.priorityAssigner = NewPriorityAssigner(\n\t\tcluster.TestCurrentClusterName,\n\t\ts.mockDomainCache,\n\t\ts.mockActiveClusterMgr,\n\t\tlog.NewNoop(),\n\t\tmetrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}),\n\t\ts.config,\n\t).(*priorityAssignerImpl)\n}\n\nfunc (s *taskPriorityAssignerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *taskPriorityAssignerSuite) TestGetDomainInfo_Success_Active() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\tdomainName, isActive, err := s.priorityAssigner.getDomainInfo(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\ts.NoError(err)\n\ts.Equal(constants.TestDomainName, domainName)\n\ts.True(isActive)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestGetDomainInfo_Success_Passive() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestAlternativeClusterName}, nil)\n\n\tdomainName, isActive, err := s.priorityAssigner.getDomainInfo(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\ts.NoError(err)\n\ts.Equal(constants.TestDomainName, domainName)\n\ts.False(isActive)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestGetDomainInfo_Fail_DomainNotExist() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(\n\t\tnil,\n\t\t&types.EntityNotExistsError{Message: \"domain not exist\"},\n\t)\n\n\tdomainName, isActive, err := s.priorityAssigner.getDomainInfo(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\ts.NoError(err)\n\ts.Empty(domainName)\n\ts.True(isActive)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestGetDomainInfo_Fail_UnknownError() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(\n\t\tnil,\n\t\terrors.New(\"some random error\"),\n\t)\n\n\tdomainName, isActive, err := s.priorityAssigner.getDomainInfo(constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID)\n\ts.Error(err)\n\ts.Empty(domainName)\n\ts.False(isActive)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_ReplicationTask() {\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeReplication).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.LowPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_StandbyTask_StandbyDomain() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestAlternativeClusterName}, nil)\n\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeStandbyTransfer).AnyTimes()\n\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.LowPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_StandbyTask_ActiveDomain() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeStandbyTransfer).AnyTimes()\n\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.HighPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_ActiveTask_StandbyDomain() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestAlternativeClusterName}, nil)\n\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeActiveTimer).AnyTimes()\n\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.HighPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_ActiveTransferTask_ActiveDomain() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeActiveTransfer).AnyTimes()\n\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.HighPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_ActiveTimerTask_ActiveDomain() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil)\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil)\n\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().GetQueueType().Return(QueueTypeActiveTimer).AnyTimes()\n\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.HighPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_ThrottledTask() {\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\ts.mockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestCurrentClusterName}, nil).AnyTimes()\n\n\tfor i := 0; i != s.testTaskProcessRPS*2; i++ {\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().GetQueueType().Return(QueueTypeActiveTimer).AnyTimes()\n\t\tmockTask.EXPECT().GetDomainID().Return(constants.TestDomainID).Times(1)\n\t\tmockTask.EXPECT().GetWorkflowID().Return(constants.TestWorkflowID).Times(1)\n\t\tmockTask.EXPECT().GetRunID().Return(constants.TestRunID).Times(1)\n\t\tmockTask.EXPECT().Priority().Return(noPriority).Times(1)\n\t\tif i < s.testTaskProcessRPS {\n\t\t\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.HighPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\t\t} else {\n\t\t\tmockTask.EXPECT().SetPriority(commonconstants.GetTaskPriority(commonconstants.DefaultPriorityClass, commonconstants.DefaultPrioritySubclass)).Times(1)\n\t\t}\n\n\t\terr := s.priorityAssigner.Assign(mockTask)\n\t\ts.NoError(err)\n\t}\n}\n\nfunc (s *taskPriorityAssignerSuite) TestAssign_AlreadyAssigned() {\n\tpriority := 5\n\n\t// case 1: task attempt less than critical retry count\n\tmockTask := NewMockTask(s.controller)\n\tmockTask.EXPECT().Priority().Return(priority).Times(1)\n\tmockTask.EXPECT().GetAttempt().Return(s.config.TaskCriticalRetryCount() - 1).Times(1)\n\terr := s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n\n\t// case 2: task attempt higher than critical retry count\n\tmockTask.EXPECT().Priority().Return(priority).Times(1)\n\tmockTask.EXPECT().GetAttempt().Return(s.config.TaskCriticalRetryCount() + 1).Times(1)\n\tmockTask.EXPECT().SetPriority(lowTaskPriority).Times(1)\n\terr = s.priorityAssigner.Assign(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *taskPriorityAssignerSuite) TestGetTaskPriority() {\n\ttestCases := []struct {\n\t\tclass            int\n\t\tsubClass         int\n\t\texpectedPriority int\n\t}{\n\t\t{\n\t\t\tclass:            commonconstants.HighPriorityClass,\n\t\t\tsubClass:         commonconstants.DefaultPrioritySubclass,\n\t\t\texpectedPriority: 1,\n\t\t},\n\t\t{\n\t\t\tclass:            commonconstants.DefaultPriorityClass,\n\t\t\tsubClass:         commonconstants.LowPrioritySubclass,\n\t\t\texpectedPriority: 10,\n\t\t},\n\t\t{\n\t\t\tclass:            commonconstants.LowPriorityClass,\n\t\t\tsubClass:         commonconstants.HighPrioritySubclass,\n\t\t\texpectedPriority: 16,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.expectedPriority, commonconstants.GetTaskPriority(tc.class, tc.subClass))\n\t}\n}\n"
  },
  {
    "path": "service/history/task/processor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype DomainPriorityKey struct {\n\tDomainID string\n\tPriority int\n}\n\ntype processorImpl struct {\n\tsync.RWMutex\n\n\tpriorityAssigner PriorityAssigner\n\ttaskProcessor    task.Processor\n\tscheduler        task.Scheduler[Task]\n\n\tstatus        int32\n\tlogger        log.Logger\n\tmetricsClient metrics.Client\n\ttimeSource    clock.TimeSource\n}\n\nvar (\n\terrTaskProcessorNotRunning = errors.New(\"queue task processor is not running\")\n)\n\nconst (\n\tephemeralTaskListGroupKey = \"__ephemeral__\"\n)\n\n// NewProcessor creates a new task processor\nfunc NewProcessor(\n\tpriorityAssigner PriorityAssigner,\n\tconfig *config.Config,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n\tdomainCache cache.DomainCache,\n) (Processor, error) {\n\ttaskProcessor := task.NewParallelTaskProcessor(\n\t\tlogger,\n\t\tmetricsClient,\n\t\t&task.ParallelTaskProcessorOptions{\n\t\t\tQueueSize:   1,\n\t\t\tWorkerCount: config.TaskSchedulerWorkerCount,\n\t\t\tRetryPolicy: common.CreateTaskProcessingRetryPolicy(),\n\t\t},\n\t)\n\tvar scheduler task.Scheduler[Task]\n\tvar err error\n\tif config.EnableHierarchicalWeightedRoundRobinTaskScheduler() {\n\t\ttaskToWeightedKeysFn := func(t Task) []task.WeightedKey[any] {\n\t\t\tvar domainID, taskList string\n\t\t\tdomainID = t.GetDomainID()\n\t\t\ttaskList = t.GetOriginalTaskList()\n\t\t\tif t.GetOriginalTaskListKind() == types.TaskListKindEphemeral {\n\t\t\t\ttaskList = ephemeralTaskListGroupKey\n\t\t\t}\n\t\t\tkey := DomainPriorityKey{\n\t\t\t\tDomainID: domainID,\n\t\t\t\tPriority: t.Priority(),\n\t\t\t}\n\t\t\tdomainName, err := domainCache.GetDomainName(domainID)\n\t\t\tif err != nil {\n\t\t\t\tlogger.Error(\"failed to get domain name from cache\", tag.Error(err))\n\t\t\t\tdomainName = \"\"\n\t\t\t}\n\t\t\tif !config.EnableTaskListAwareTaskSchedulerByDomain(domainName) || t.Priority() != highTaskPriority {\n\t\t\t\treturn []task.WeightedKey[any]{\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:    key,\n\t\t\t\t\t\tWeight: getDomainPriorityWeight(logger, config, domainCache, key),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn []task.WeightedKey[any]{\n\t\t\t\t{\n\t\t\t\t\tKey:    key,\n\t\t\t\t\tWeight: getDomainPriorityWeight(logger, config, domainCache, key),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tKey:    taskList,\n\t\t\t\t\tWeight: 1,\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\t\tscheduler, err = task.NewHierarchicalWeightedRoundRobinTaskScheduler(\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\ttimeSource,\n\t\t\ttaskProcessor,\n\t\t\t&task.HierarchicalWeightedRoundRobinTaskPoolOptions[any, Task]{\n\t\t\t\tBufferSize:           config.TaskSchedulerQueueSize(),\n\t\t\t\tTaskToWeightedKeysFn: taskToWeightedKeysFn,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t} else {\n\t\ttaskToChannelKeyFn := func(t Task) DomainPriorityKey {\n\t\t\tvar domainID string\n\t\t\tdomainID = t.GetDomainID()\n\t\t\treturn DomainPriorityKey{\n\t\t\t\tDomainID: domainID,\n\t\t\t\tPriority: t.Priority(),\n\t\t\t}\n\t\t}\n\t\tchannelKeyToWeightFn := func(k DomainPriorityKey) int {\n\t\t\treturn getDomainPriorityWeight(logger, config, domainCache, k)\n\t\t}\n\t\tscheduler, err = task.NewWeightedRoundRobinTaskScheduler(\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\ttimeSource,\n\t\t\ttaskProcessor,\n\t\t\t&task.WeightedRoundRobinTaskSchedulerOptions[DomainPriorityKey, Task]{\n\t\t\t\tQueueSize:            config.TaskSchedulerQueueSize(),\n\t\t\t\tDispatcherCount:      config.TaskSchedulerDispatcherCount(),\n\t\t\t\tTaskToChannelKeyFn:   taskToChannelKeyFn,\n\t\t\t\tChannelKeyToWeightFn: channelKeyToWeightFn,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn &processorImpl{\n\t\tpriorityAssigner: priorityAssigner,\n\t\ttaskProcessor:    taskProcessor,\n\t\tscheduler:        scheduler,\n\t\tstatus:           common.DaemonStatusInitialized,\n\t\tlogger:           logger,\n\t\tmetricsClient:    metricsClient,\n\t\ttimeSource:       timeSource,\n\t}, nil\n}\n\nfunc (p *processorImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.taskProcessor.Start()\n\tp.scheduler.Start()\n\n\tp.logger.Info(\"Queue task processor started.\")\n}\n\nfunc (p *processorImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tp.scheduler.Stop()\n\tp.taskProcessor.Stop()\n\n\tp.logger.Info(\"Queue task processor stopped.\")\n}\n\nfunc (p *processorImpl) Submit(task Task) error {\n\tif err := p.priorityAssigner.Assign(task); err != nil {\n\t\treturn err\n\t}\n\treturn p.scheduler.Submit(task)\n}\n\nfunc (p *processorImpl) TrySubmit(task Task) (bool, error) {\n\tif err := p.priorityAssigner.Assign(task); err != nil {\n\t\treturn false, err\n\t}\n\treturn p.scheduler.TrySubmit(task)\n}\n\nfunc getDomainPriorityWeight(\n\tlogger log.Logger,\n\tconfig *config.Config,\n\tdomainCache cache.DomainCache,\n\tk DomainPriorityKey,\n) int {\n\tvar weights map[int]int\n\tdomainName, err := domainCache.GetDomainName(k.DomainID)\n\tif err != nil {\n\t\tlogger.Error(\"failed to get domain name from cache, use default round robin weights\", tag.Error(err))\n\t\tweights = dynamicproperties.DefaultTaskSchedulerRoundRobinWeights\n\t} else {\n\t\tweights, err = dynamicproperties.ConvertDynamicConfigMapPropertyToIntMap(config.TaskSchedulerDomainRoundRobinWeights(domainName))\n\t\tif err != nil {\n\t\t\tlogger.Error(\"failed to convert dynamic config map to int map, use default round robin weights\", tag.Error(err))\n\t\t\tweights = dynamicproperties.DefaultTaskSchedulerRoundRobinWeights\n\t\t}\n\t}\n\tweight, ok := weights[k.Priority]\n\tif !ok {\n\t\tlogger.Error(\"weights not found for task priority, default to 1\", tag.Dynamic(\"priority\", k.Priority), tag.Dynamic(\"weights\", weights))\n\t\tweight = 1\n\t}\n\treturn weight\n}\n"
  },
  {
    "path": "service/history/task/processor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/config\"\n)\n\ntype (\n\tqueueTaskProcessorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockDomainCache      *cache.MockDomainCache\n\t\tmockPriorityAssigner *MockPriorityAssigner\n\n\t\ttimeSource    clock.TimeSource\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\n\t\tprocessor *processorImpl\n\t}\n)\n\nfunc TestQueueTaskProcessorSuite(t *testing.T) {\n\ts := new(queueTaskProcessorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *queueTaskProcessorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockPriorityAssigner = NewMockPriorityAssigner(s.controller)\n\n\ts.timeSource = clock.NewRealTimeSource()\n\ts.metricsClient = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{})\n\ts.logger = testlogger.New(s.Suite.T())\n\n\ts.processor = s.newTestQueueTaskProcessor()\n}\n\nfunc (s *queueTaskProcessorSuite) TearDownTest() {}\n\nfunc (s *queueTaskProcessorSuite) TestStartStop() {\n\tmockScheduler := task.NewMockScheduler[Task](s.controller)\n\tmockScheduler.EXPECT().Start().Times(1)\n\tmockScheduler.EXPECT().Stop().Times(1)\n\ts.processor.scheduler = mockScheduler\n\n\ts.processor.Start()\n\ts.processor.Stop()\n}\n\nfunc (s *queueTaskProcessorSuite) TestSubmit() {\n\tmockTask := NewMockTask(s.controller)\n\ts.mockPriorityAssigner.EXPECT().Assign(NewMockTaskMatcher(mockTask)).Return(nil).Times(1)\n\n\tmockScheduler := task.NewMockScheduler[Task](s.controller)\n\tmockScheduler.EXPECT().Submit(NewMockTaskMatcher(mockTask)).Return(nil).Times(1)\n\n\ts.processor.scheduler = mockScheduler\n\n\terr := s.processor.Submit(mockTask)\n\ts.NoError(err)\n}\n\nfunc (s *queueTaskProcessorSuite) TestTrySubmit_AssignPriorityFailed() {\n\tmockTask := NewMockTask(s.controller)\n\n\terrAssignPriority := errors.New(\"some randome error\")\n\ts.mockPriorityAssigner.EXPECT().Assign(NewMockTaskMatcher(mockTask)).Return(errAssignPriority).Times(1)\n\n\tsubmitted, err := s.processor.TrySubmit(mockTask)\n\ts.Equal(errAssignPriority, err)\n\ts.False(submitted)\n}\n\nfunc (s *queueTaskProcessorSuite) TestTrySubmit_Fail() {\n\tmockTask := NewMockTask(s.controller)\n\ts.mockPriorityAssigner.EXPECT().Assign(NewMockTaskMatcher(mockTask)).Return(nil).Times(1)\n\n\terrTrySubmit := errors.New(\"some randome error\")\n\tmockScheduler := task.NewMockScheduler[Task](s.controller)\n\tmockScheduler.EXPECT().TrySubmit(NewMockTaskMatcher(mockTask)).Return(false, errTrySubmit).Times(1)\n\n\ts.processor.scheduler = mockScheduler\n\n\tsubmitted, err := s.processor.TrySubmit(mockTask)\n\ts.Equal(errTrySubmit, err)\n\ts.False(submitted)\n}\n\nfunc (s *queueTaskProcessorSuite) TestNewSchedulerOptions_UnknownSchedulerType() {\n\toptions, err := task.NewSchedulerOptions[int](0, 100, dynamicproperties.GetIntPropertyFn(10), 1, func(task.PriorityTask) int { return 1 }, func(int) int { return 1 })\n\ts.Error(err)\n\ts.Nil(options)\n}\n\nfunc (s *queueTaskProcessorSuite) newTestQueueTaskProcessor() *processorImpl {\n\tconfig := config.NewForTest()\n\tprocessor, err := NewProcessor(\n\t\ts.mockPriorityAssigner,\n\t\tconfig,\n\t\ts.logger,\n\t\ts.metricsClient,\n\t\ts.timeSource,\n\t\ts.mockDomainCache,\n\t)\n\ts.NoError(err)\n\treturn processor.(*processorImpl)\n}\n\nfunc TestGetDomainPriorityWeight(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*cache.MockDomainCache, dynamicconfig.Client)\n\t\texpected  int\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, client dynamicconfig.Client) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain-name\", nil).Times(1)\n\t\t\t\tclient.UpdateValue(dynamicproperties.TaskSchedulerDomainRoundRobinWeights, map[string]interface{}{\"1\": 10})\n\t\t\t},\n\t\t\texpected: 10,\n\t\t},\n\t\t{\n\t\t\tname: \"domain cache error, use default\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, client dynamicconfig.Client) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"\", errors.New(\"test-error\")).Times(1)\n\t\t\t\tclient.UpdateValue(dynamicproperties.TaskSchedulerDomainRoundRobinWeights, map[string]interface{}{\"1\": 10})\n\t\t\t},\n\t\t\texpected: 500,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid map value, use default\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, client dynamicconfig.Client) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain-name\", nil).Times(1)\n\t\t\t\tclient.UpdateValue(dynamicproperties.TaskSchedulerDomainRoundRobinWeights, map[string]interface{}{\"1\": \"invalid\"})\n\t\t\t},\n\t\t\texpected: 500,\n\t\t},\n\t\t{\n\t\t\tname: \"unspecified priority\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, client dynamicconfig.Client) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain-name\", nil).Times(1)\n\t\t\t\tclient.UpdateValue(dynamicproperties.TaskSchedulerDomainRoundRobinWeights, map[string]interface{}{\"2\": 10})\n\t\t\t},\n\t\t\texpected: 1,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\tconfig := config.New(\n\t\t\t\tdynamicconfig.NewCollection(\n\t\t\t\t\tclient,\n\t\t\t\t\ttestlogger.New(t),\n\t\t\t\t),\n\t\t\t\t1024,\n\t\t\t\t1024,\n\t\t\t\tfalse,\n\t\t\t\t\"hostname\",\n\t\t\t)\n\t\t\ttc.mockSetup(mockDomainCache, client)\n\n\t\t\tweight := getDomainPriorityWeight(testlogger.New(t), config, mockDomainCache, DomainPriorityKey{DomainID: \"test-domain-id\", Priority: 1})\n\t\t\trequire.Equal(t, tc.expected, weight)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/rate_limited_processor.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n)\n\ntype rateLimitedProcessor struct {\n\tbaseProcessor Processor\n\trateLimiter   RateLimiter\n\tcancelCtx     context.Context\n\tcancelFn      context.CancelFunc\n\tstatus        int32\n}\n\nfunc NewRateLimitedProcessor(\n\tbaseProcessor Processor,\n\trateLimiter RateLimiter,\n) Processor {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &rateLimitedProcessor{\n\t\tbaseProcessor: baseProcessor,\n\t\trateLimiter:   rateLimiter,\n\t\tcancelCtx:     ctx,\n\t\tcancelFn:      cancel,\n\t\tstatus:        common.DaemonStatusInitialized,\n\t}\n}\n\nfunc (p *rateLimitedProcessor) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.baseProcessor.Start()\n}\n\nfunc (p *rateLimitedProcessor) Stop() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tp.cancelFn()\n\tp.baseProcessor.Stop()\n}\n\nfunc (p *rateLimitedProcessor) Submit(t Task) error {\n\tif err := p.rateLimiter.Wait(p.cancelCtx, t); err != nil {\n\t\treturn err\n\t}\n\treturn p.baseProcessor.Submit(t)\n}\n\nfunc (p *rateLimitedProcessor) TrySubmit(t Task) (bool, error) {\n\tif ok := p.rateLimiter.Allow(t); !ok {\n\t\treturn false, nil\n\t}\n\treturn p.baseProcessor.TrySubmit(t)\n}\n"
  },
  {
    "path": "service/history/task/rate_limited_processor_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\ntype rateLimitedProcessorMockDeps struct {\n\tmockProcessor   *MockProcessor\n\tmockRateLimiter *MockRateLimiter\n}\n\nfunc setupMocksForRateLimitedProcessor(t *testing.T) (*rateLimitedProcessor, *rateLimitedProcessorMockDeps) {\n\tctrl := gomock.NewController(t)\n\n\tdeps := &rateLimitedProcessorMockDeps{\n\t\tmockProcessor:   NewMockProcessor(ctrl),\n\t\tmockRateLimiter: NewMockRateLimiter(ctrl),\n\t}\n\n\tprocessor := NewRateLimitedProcessor(deps.mockProcessor, deps.mockRateLimiter)\n\trp, ok := processor.(*rateLimitedProcessor)\n\trequire.True(t, ok)\n\treturn rp, deps\n}\n\nfunc TestRateLimitedProcessorLifecycle(t *testing.T) {\n\trp, deps := setupMocksForRateLimitedProcessor(t)\n\n\tdeps.mockProcessor.EXPECT().Start().Times(1)\n\trp.Start()\n\n\tdeps.mockProcessor.EXPECT().Stop().Times(1)\n\trp.Stop()\n}\n\nfunc TestRateLimitedProcessorSubmit(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\ttask          Task\n\t\tsetupMocks    func(*rateLimitedProcessorMockDeps)\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Wait(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\tdeps.mockProcessor.EXPECT().Submit(gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"rate limiter error\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Wait(gomock.Any(), gomock.Any()).Return(errors.New(\"rate limited\")).Times(1)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"rate limited\",\n\t\t},\n\t\t{\n\t\t\tname: \"processor error\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Wait(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\tdeps.mockProcessor.EXPECT().Submit(gomock.Any()).Return(errors.New(\"processor error\")).Times(1)\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"processor error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\trp, deps := setupMocksForRateLimitedProcessor(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\terr := rp.Submit(tc.task)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRateLimitedProcessorTrySubmit(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\ttask          Task\n\t\tsetupMocks    func(*rateLimitedProcessorMockDeps)\n\t\texpected      bool\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Allow(gomock.Any()).Return(true)\n\t\t\t\tdeps.mockProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil)\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"rate limited\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Allow(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error\",\n\t\t\ttask: &noopTask{},\n\t\t\tsetupMocks: func(deps *rateLimitedProcessorMockDeps) {\n\t\t\t\tdeps.mockRateLimiter.EXPECT().Allow(gomock.Any()).Return(true)\n\t\t\t\tdeps.mockProcessor.EXPECT().TrySubmit(gomock.Any()).Return(false, errors.New(\"submit error\"))\n\t\t\t},\n\t\t\texpected:      false,\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"submit error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\trp, deps := setupMocksForRateLimitedProcessor(t)\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tsubmitted, err := rp.TrySubmit(tc.task)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, submitted)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/redispatcher.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tctask \"github.com/uber/cadence/common/task\"\n)\n\nconst (\n\tdefaultBufferSize = 200\n)\n\ntype (\n\tredispatchNotification struct {\n\t\ttargetSize int\n\t\tdoneCh     chan struct{}\n\t}\n\n\t// RedispatcherOptions configs redispatch interval\n\tRedispatcherOptions struct {\n\t\tTaskRedispatchInterval dynamicproperties.DurationPropertyFn\n\t}\n\tredispatcherImpl struct {\n\t\tsync.Mutex\n\n\t\ttaskProcessor Processor\n\t\ttimeSource    clock.TimeSource\n\t\tlogger        log.Logger\n\t\tmetricsScope  metrics.Scope\n\n\t\tstatus        int32\n\t\tshutdownCh    chan struct{}\n\t\tshutdownWG    sync.WaitGroup\n\t\tredispatchCh  chan redispatchNotification\n\t\ttimerGate     clock.TimerGate\n\t\tbackoffPolicy backoff.RetryPolicy\n\t\tpqMap         map[int]collection.Queue[redispatchTask] // priority -> redispatch queue\n\t\ttaskChFull    map[int]bool                             // priority -> if taskCh is full\n\t}\n\n\tredispatchTask struct {\n\t\ttask           Task\n\t\tredispatchTime time.Time\n\t}\n)\n\n// NewRedispatcher creates a new task Redispatcher\nfunc NewRedispatcher(\n\ttaskProcessor Processor,\n\ttimeSource clock.TimeSource,\n\toptions *RedispatcherOptions,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) Redispatcher {\n\tbackoffPolicy := backoff.NewExponentialRetryPolicy(options.TaskRedispatchInterval())\n\tbackoffPolicy.SetBackoffCoefficient(redispatchBackoffCoefficient)\n\tbackoffPolicy.SetMaximumInterval(redispatchMaxBackoffInternval)\n\tbackoffPolicy.SetExpirationInterval(backoff.NoInterval)\n\n\treturn &redispatcherImpl{\n\t\ttaskProcessor: taskProcessor,\n\t\ttimeSource:    timeSource,\n\t\tlogger:        logger,\n\t\tmetricsScope:  metricsScope,\n\t\tstatus:        common.DaemonStatusInitialized,\n\t\tshutdownCh:    make(chan struct{}),\n\t\tredispatchCh:  make(chan redispatchNotification, 1),\n\t\ttimerGate:     clock.NewTimerGate(timeSource),\n\t\tbackoffPolicy: backoffPolicy,\n\t\tpqMap:         make(map[int]collection.Queue[redispatchTask]),\n\t\ttaskChFull:    make(map[int]bool),\n\t}\n}\n\nfunc (r *redispatcherImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tr.shutdownWG.Add(1)\n\tgo r.redispatchLoop()\n\n\tr.logger.Info(\"Task redispatcher started.\", tag.LifeCycleStarted)\n}\n\nfunc (r *redispatcherImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(r.shutdownCh)\n\tr.timerGate.Stop()\n\tif success := common.AwaitWaitGroup(&r.shutdownWG, time.Minute); !success {\n\t\tr.logger.Warn(\"Task redispatcher timedout on shutdown.\", tag.LifeCycleStopTimedout)\n\t}\n\n\tr.logger.Info(\"Task redispatcher stopped.\", tag.LifeCycleStopped)\n}\n\nfunc (r *redispatcherImpl) AddTask(task Task) {\n\tpriority := task.Priority()\n\tattempt := task.GetAttempt()\n\n\tr.Lock()\n\tpq := r.getOrCreatePQLocked(priority)\n\tt := r.getRedispatchTime(attempt)\n\tpq.Add(redispatchTask{\n\t\ttask:           task,\n\t\tredispatchTime: t,\n\t})\n\tr.Unlock()\n\n\tr.timerGate.Update(t)\n}\n\nfunc (r *redispatcherImpl) RedispatchTask(task Task, t time.Time) {\n\tpriority := task.Priority()\n\n\tr.Lock()\n\tpq := r.getOrCreatePQLocked(priority)\n\tpq.Add(redispatchTask{\n\t\ttask:           task,\n\t\tredispatchTime: t,\n\t})\n\tr.Unlock()\n\n\tr.timerGate.Update(t)\n}\n\n// TODO: review this method, it doesn't seem to redispatch the tasks immediately\nfunc (r *redispatcherImpl) Redispatch(targetSize int) {\n\tdoneCh := make(chan struct{})\n\tntf := redispatchNotification{\n\t\ttargetSize: targetSize,\n\t\tdoneCh:     doneCh,\n\t}\n\n\tselect {\n\tcase r.redispatchCh <- ntf:\n\t\t// block until the redispatch is done\n\t\t<-doneCh\n\tcase <-r.shutdownCh:\n\t\tclose(doneCh)\n\t\treturn\n\t}\n}\n\nfunc (r *redispatcherImpl) Size() int {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\treturn r.sizeLocked()\n}\n\nfunc (r *redispatcherImpl) redispatchLoop() {\n\tdefer r.shutdownWG.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-r.shutdownCh:\n\t\t\treturn\n\t\tcase <-r.timerGate.Chan():\n\t\t\tr.redispatchTasks(redispatchNotification{})\n\t\tcase notification := <-r.redispatchCh:\n\t\t\tr.redispatchTasks(notification)\n\t\t}\n\t}\n}\n\nfunc (r *redispatcherImpl) redispatchTasks(notification redispatchNotification) {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\tdefer func() {\n\t\tif notification.doneCh != nil {\n\t\t\tclose(notification.doneCh)\n\t\t}\n\t}()\n\n\tif r.isStopped() {\n\t\treturn\n\t}\n\n\tqueueSize := r.sizeLocked()\n\tr.metricsScope.RecordTimer(metrics.TaskRedispatchQueuePendingTasksTimer, time.Duration(queueSize))\n\n\t// add some buffer here as new tasks may be added\n\ttargetRedispatched := queueSize + defaultBufferSize - notification.targetSize\n\tif targetRedispatched <= 0 {\n\t\t// target size has already been met, no need to redispatch\n\t\treturn\n\t}\n\n\ttotalRedispatched := 0\n\tnow := r.timeSource.Now()\n\tfor priority := range r.pqMap {\n\t\tr.taskChFull[priority] = false\n\t}\n\n\tfor priority, pq := range r.pqMap {\n\t\t// Note the third condition regarding taskChFull is not 100% accurate\n\t\t// since task may get a new, lower priority upon redispatch, and\n\t\t// the taskCh for the new priority may not be full.\n\t\t// But the current estimation should be good enough as task with\n\t\t// lower priority should be executed after high priority ones,\n\t\t// so it's ok to leave them in the queue\n\t\tfor !pq.IsEmpty() && totalRedispatched < targetRedispatched && !r.taskChFull[priority] {\n\t\t\titem, _ := pq.Peek() // error is impossible because we've checked that the queue is not empty\n\t\t\tif item.redispatchTime.After(now) {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tif item.task.State() != ctask.TaskStatePending {\n\t\t\t\tpq.Remove()\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif item.task.GetAttempt() == 0 {\n\t\t\t\titem.task.SetInitialSubmitTime(now)\n\t\t\t}\n\t\t\tsubmitted, err := r.taskProcessor.TrySubmit(item.task)\n\t\t\tif err != nil {\n\t\t\t\tif r.isStopped() {\n\t\t\t\t\t// if error is due to shard shutdown\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\t// otherwise it might be error from domain cache etc, add\n\t\t\t\t// the task to redispatch queue so that it can be retried\n\t\t\t\tr.logger.Error(\"Failed to redispatch task\", tag.Error(err))\n\t\t\t}\n\t\t\tpq.Remove()\n\t\t\tnewPriority := item.task.Priority()\n\t\t\tif err != nil || !submitted {\n\t\t\t\t// failed to submit, enqueue again\n\t\t\t\titem.redispatchTime = r.timeSource.Now().Add(redispatchFailureBackoffInterval)\n\t\t\t\tr.getOrCreatePQLocked(newPriority).Add(item)\n\t\t\t}\n\t\t\tif err == nil && !submitted {\n\t\t\t\t// task chan is full for the new priority\n\t\t\t\tr.taskChFull[newPriority] = true\n\t\t\t}\n\t\t\tif submitted {\n\t\t\t\ttotalRedispatched++\n\t\t\t}\n\t\t}\n\t\tif !pq.IsEmpty() {\n\t\t\titem, _ := pq.Peek()\n\t\t\tr.timerGate.Update(item.redispatchTime)\n\t\t}\n\t\tif r.isStopped() {\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (r *redispatcherImpl) sizeLocked() int {\n\tsize := 0\n\tfor _, queue := range r.pqMap {\n\t\tsize += queue.Len()\n\t}\n\treturn size\n}\n\nfunc (r *redispatcherImpl) isStopped() bool {\n\treturn atomic.LoadInt32(&r.status) == common.DaemonStatusStopped\n}\n\nfunc (r *redispatcherImpl) getRedispatchTime(attempt int) time.Time {\n\t// note that elapsedTime (the first parameter) is not relevant when\n\t// the retry policy has not expiration intervaly(0, attempt)))\n\treturn r.timeSource.Now().Add(r.backoffPolicy.ComputeNextDelay(0, attempt))\n}\n\nfunc (r *redispatcherImpl) getOrCreatePQLocked(priority int) collection.Queue[redispatchTask] {\n\tif pq, ok := r.pqMap[priority]; ok {\n\t\treturn pq\n\t}\n\n\tpq := collection.NewPriorityQueue(redispatchTaskCompareLess)\n\tr.pqMap[priority] = pq\n\treturn pq\n}\n\nfunc redispatchTaskCompareLess(\n\tthis redispatchTask,\n\tthat redispatchTask,\n) bool {\n\treturn this.redispatchTime.Before(that.redispatchTime)\n}\n"
  },
  {
    "path": "service/history/task/redispatcher_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"math/rand\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tctask \"github.com/uber/cadence/common/task\"\n)\n\ntype (\n\tredispatcherSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller     *gomock.Controller\n\t\tmockProcessor  *MockProcessor\n\t\tmockTimeSource clock.MockedTimeSource\n\t\toptions        *RedispatcherOptions\n\n\t\tmetricsScope metrics.Scope\n\t\tlogger       log.Logger\n\n\t\tredispatcher *redispatcherImpl\n\t}\n)\n\nfunc TestRedispatcherSuite(t *testing.T) {\n\ts := new(redispatcherSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *redispatcherSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockProcessor = NewMockProcessor(s.controller)\n\ts.mockTimeSource = clock.NewMockedTimeSource()\n\ts.options = &RedispatcherOptions{\n\t\tTaskRedispatchInterval: dynamicproperties.GetDurationPropertyFn(time.Millisecond * 50),\n\t}\n\n\ts.metricsScope = metrics.NewClient(tally.NoopScope, metrics.History, metrics.HistogramMigration{}).Scope(0)\n\ts.logger = testlogger.New(s.T())\n\n\ts.redispatcher = NewRedispatcher(\n\t\ts.mockProcessor,\n\t\ts.mockTimeSource,\n\t\ts.options,\n\t\ts.logger,\n\t\ts.metricsScope,\n\t).(*redispatcherImpl)\n}\n\nfunc (s *redispatcherSuite) TearDownTest() {\n\ts.redispatcher.Stop()\n}\n\nfunc (s *redispatcherSuite) TestRedispatch_ProcessorShutDown() {\n\tnumTasks := 5\n\n\tsuccessfullyRedispatched := 3\n\tstopDoneCh := make(chan struct{})\n\ts.mockProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).Times(successfullyRedispatched)\n\ts.mockProcessor.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(_ interface{}) (bool, error) {\n\t\tgo func() {\n\t\t\ts.redispatcher.Stop()\n\t\t\tclose(stopDoneCh)\n\t\t}()\n\n\t\t<-s.redispatcher.shutdownCh\n\t\treturn false, errors.New(\"processor shutdown\")\n\t}).Times(1)\n\n\tfor i := 0; i < numTasks; i++ {\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(5)).AnyTimes()\n\t\tmockTask.EXPECT().GetAttempt().Return(0).Times(1)\n\t\tmockTask.EXPECT().GetAttempt().Return(0).MaxTimes(1)\n\t\tmockTask.EXPECT().SetInitialSubmitTime(gomock.Any()).MaxTimes(1)\n\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).MaxTimes(1)\n\t\ts.redispatcher.AddTask(mockTask)\n\t}\n\n\ts.Equal(numTasks, s.redispatcher.Size())\n\ts.mockTimeSource.Advance(2 * s.options.TaskRedispatchInterval())\n\ts.redispatcher.Start()\n\t<-s.redispatcher.shutdownCh\n\t<-stopDoneCh\n\n\ts.Equal(numTasks-successfullyRedispatched, s.redispatcher.Size())\n}\n\nfunc (s *redispatcherSuite) TestRedispatch_WithTargetSize() {\n\tnumTasks := defaultBufferSize + 20\n\ttargetSize := defaultBufferSize + 10\n\n\ti := 0\n\tfor ; i < numTasks+defaultBufferSize-targetSize; i++ {\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(5)).AnyTimes()\n\t\tmockTask.EXPECT().GetAttempt().Return(0).Times(2)\n\t\tmockTask.EXPECT().SetInitialSubmitTime(gomock.Any()).Times(1)\n\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).MaxTimes(1)\n\t\ts.redispatcher.AddTask(mockTask)\n\t}\n\n\tfor ; i < numTasks; i++ {\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(5)).AnyTimes()\n\t\tmockTask.EXPECT().GetAttempt().Return(1000).Times(1) // make sure these tasks are not dispatched\n\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).MaxTimes(1)\n\t\ts.redispatcher.AddTask(mockTask)\n\t}\n\n\ts.redispatcher.Start()\n\ts.redispatcher.timerGate.Stop()\n\ts.mockProcessor.EXPECT().TrySubmit(gomock.Any()).Return(true, nil).Times(numTasks + defaultBufferSize - targetSize)\n\ts.mockTimeSource.Advance(2 * s.options.TaskRedispatchInterval()) // the time should be enough to let tasks with attempt 0 be dispatched\n\ts.redispatcher.Redispatch(targetSize)\n\n\t// implementation can choose to redispatch more tasks than needed\n\tsz := s.redispatcher.Size()\n\ts.Equal(targetSize-defaultBufferSize, sz)\n}\n\nfunc (s *redispatcherSuite) TestRedispatch_Backoff() {\n\tnumTasks := 50\n\tnumLowAttemptTasks := 0\n\tnumHighAttemptTasks := 0\n\tfor i := 0; i < numTasks; i++ {\n\t\tattempt := 100\n\t\tif rand.Intn(2) == 0 {\n\t\t\tnumLowAttemptTasks++\n\t\t\tattempt = 0\n\t\t} else {\n\t\t\tnumHighAttemptTasks++\n\t\t}\n\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(5)).AnyTimes()\n\t\tif attempt == 0 {\n\t\t\tmockTask.EXPECT().GetAttempt().Return(attempt).Times(2)\n\t\t\tmockTask.EXPECT().SetInitialSubmitTime(gomock.Any()).Times(1)\n\t\t} else {\n\t\t\tmockTask.EXPECT().GetAttempt().Return(attempt).Times(1)\n\t\t}\n\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).MaxTimes(1)\n\t\ts.redispatcher.AddTask(mockTask)\n\t\ts.mockProcessor.EXPECT().TrySubmit(NewMockTaskMatcher(mockTask)).Return(true, nil).MaxTimes(1)\n\t}\n\n\ts.redispatcher.Start()\n\ts.redispatcher.timerGate.Stop()\n\ts.mockTimeSource.Advance(2 * s.options.TaskRedispatchInterval())\n\ts.redispatcher.Redispatch(0)\n\n\ts.Equal(numHighAttemptTasks, s.redispatcher.Size())\n}\n\nfunc (s *redispatcherSuite) TestRedispatch_Random() {\n\tnumTasks := 100\n\tdispatched := 0\n\n\tfor i := 0; i != numTasks; i++ {\n\t\tsubmitted := false\n\t\tattempt := 100\n\t\tif rand.Intn(2) == 0 {\n\t\t\tsubmitted = true\n\t\t\tif rand.Intn(2) == 0 {\n\t\t\t\tdispatched++\n\t\t\t\tattempt = 0\n\t\t\t}\n\t\t}\n\n\t\tmockTask := NewMockTask(s.controller)\n\t\tmockTask.EXPECT().Priority().Return(rand.Intn(5)).AnyTimes()\n\t\tif attempt == 0 {\n\t\t\tmockTask.EXPECT().GetAttempt().Return(attempt).Times(2)\n\t\t\tmockTask.EXPECT().SetInitialSubmitTime(gomock.Any()).Times(1)\n\t\t} else {\n\t\t\tmockTask.EXPECT().GetAttempt().Return(attempt).Times(1)\n\t\t}\n\t\tmockTask.EXPECT().State().Return(ctask.TaskStatePending).MaxTimes(1)\n\t\ts.redispatcher.AddTask(mockTask)\n\t\ts.mockProcessor.EXPECT().TrySubmit(NewMockTaskMatcher(mockTask)).Return(submitted, nil).MaxTimes(1)\n\t}\n\n\ts.redispatcher.Start()\n\ts.redispatcher.timerGate.Stop()\n\ts.mockTimeSource.Advance(2 * s.options.TaskRedispatchInterval())\n\ts.redispatcher.Redispatch(0)\n\n\t// implementation can choose to stop redispatch for a certain priority when previous submit has failed\n\ts.True(s.redispatcher.Size() >= numTasks-dispatched)\n}\n"
  },
  {
    "path": "service/history/task/rescheduler.go",
    "content": "package task\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tctask \"github.com/uber/cadence/common/task\"\n)\n\nconst (\n\ttaskChanFullBackoff                  = 2 * time.Second\n\ttaskChanFullBackoffJitterCoefficient = 0.5\n\n\treschedulerPQCleanupDuration          = 3 * time.Minute\n\treschedulerPQCleanupJitterCoefficient = 0.15\n)\n\ntype (\n\trescheduledTask struct {\n\t\ttask           Task\n\t\trescheduleTime time.Time\n\t}\n\n\treschedulerImpl struct {\n\t\tscheduler    Processor\n\t\ttimeSource   clock.TimeSource\n\t\tlogger       log.Logger\n\t\tmetricsScope metrics.Scope\n\n\t\tstatus     int32\n\t\tctx        context.Context\n\t\tcancel     context.CancelFunc\n\t\tshutdownWG sync.WaitGroup\n\n\t\ttimerGate clock.TimerGate\n\n\t\tsync.Mutex\n\t\tpqMap          map[DomainPriorityKey]collection.Queue[rescheduledTask]\n\t\tnumExecutables int\n\t}\n)\n\nfunc NewRescheduler(\n\tscheduler Processor,\n\ttimeSource clock.TimeSource,\n\tlogger log.Logger,\n\tmetricsScope metrics.Scope,\n) Rescheduler {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &reschedulerImpl{\n\t\tscheduler:    scheduler,\n\t\ttimeSource:   timeSource,\n\t\tlogger:       logger,\n\t\tmetricsScope: metricsScope,\n\n\t\tstatus: common.DaemonStatusInitialized,\n\t\tctx:    ctx,\n\t\tcancel: cancel,\n\n\t\ttimerGate: clock.NewTimerGate(timeSource),\n\n\t\tpqMap: make(map[DomainPriorityKey]collection.Queue[rescheduledTask]),\n\t}\n}\n\nfunc (r *reschedulerImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tr.shutdownWG.Add(1)\n\tgo r.rescheduleLoop()\n\n\tr.logger.Info(\"Task rescheduler started.\", tag.LifeCycleStarted)\n}\n\nfunc (r *reschedulerImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&r.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tr.cancel()\n\tr.timerGate.Stop()\n\n\tif success := common.AwaitWaitGroup(&r.shutdownWG, time.Minute); !success {\n\t\tr.logger.Warn(\"Task rescheduler timedout on shutdown.\", tag.LifeCycleStopTimedout)\n\t}\n\n\tr.logger.Info(\"Task rescheduler stopped.\", tag.LifeCycleStopped)\n}\n\nfunc (r *reschedulerImpl) RescheduleTask(\n\ttask Task,\n\trescheduleTime time.Time,\n) {\n\tr.Lock()\n\tpq := r.getOrCreatePQLocked(DomainPriorityKey{\n\t\tDomainID: task.GetDomainID(),\n\t\tPriority: task.Priority(),\n\t})\n\tpq.Add(rescheduledTask{\n\t\ttask:           task,\n\t\trescheduleTime: rescheduleTime,\n\t})\n\tr.numExecutables++\n\tr.Unlock()\n\tr.timerGate.Update(rescheduleTime)\n\n\tif r.isStopped() {\n\t\tr.drain()\n\t}\n}\n\nfunc (r *reschedulerImpl) RescheduleDomains(\n\tdomainIDs map[string]struct{},\n) {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\tnow := r.timeSource.Now()\n\tupdatedRescheduleTime := false\n\tfor key, pq := range r.pqMap {\n\t\tif _, ok := domainIDs[key.DomainID]; !ok {\n\t\t\tcontinue\n\t\t}\n\n\t\tupdatedRescheduleTime = true\n\t\t// set reschedule time for all tasks in this pq to be now\n\t\titems := make([]rescheduledTask, 0, pq.Len())\n\t\tfor !pq.IsEmpty() {\n\t\t\trescheduled, _ := pq.Remove()\n\t\t\t// scheduled queue pre-fetches tasks,\n\t\t\t// so we need to make sure the reschedule time is not before the task scheduled time\n\t\t\tscheduleTime := rescheduled.task.GetTaskKey().GetScheduledTime()\n\t\t\tif now.Before(scheduleTime) {\n\t\t\t\trescheduled.rescheduleTime = scheduleTime\n\t\t\t} else {\n\t\t\t\trescheduled.rescheduleTime = now\n\t\t\t}\n\t\t\titems = append(items, rescheduled)\n\t\t}\n\t\tr.pqMap[key] = r.newPriorityQueue(items)\n\t}\n\n\t// then update timer gate to trigger the actual reschedule\n\tif updatedRescheduleTime {\n\t\tr.timerGate.Update(now)\n\t}\n}\n\nfunc (r *reschedulerImpl) Size() int {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\treturn r.numExecutables\n}\n\nfunc (r *reschedulerImpl) rescheduleLoop() {\n\tdefer r.shutdownWG.Done()\n\n\tcleanupTimer := r.timeSource.NewTimer(backoff.JitDuration(\n\t\treschedulerPQCleanupDuration,\n\t\treschedulerPQCleanupJitterCoefficient,\n\t))\n\tdefer cleanupTimer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-r.ctx.Done():\n\t\t\tr.drain()\n\t\t\treturn\n\t\tcase <-r.timerGate.Chan():\n\t\t\tr.reschedule()\n\t\tcase <-cleanupTimer.Chan():\n\t\t\tr.cleanupPQ()\n\t\t\tcleanupTimer.Reset(backoff.JitDuration(\n\t\t\t\treschedulerPQCleanupDuration,\n\t\t\t\treschedulerPQCleanupJitterCoefficient,\n\t\t\t))\n\t\t}\n\t}\n\n}\n\nfunc (r *reschedulerImpl) reschedule() {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\tr.metricsScope.UpdateGauge(metrics.ReschedulerTaskCountGauge, float64(r.numExecutables))\n\tnow := r.timeSource.Now()\n\tfor _, pq := range r.pqMap {\n\t\tfor !pq.IsEmpty() {\n\t\t\trescheduled, _ := pq.Peek()\n\n\t\t\tif rescheduleTime := rescheduled.rescheduleTime; now.Before(rescheduleTime) {\n\t\t\t\tr.timerGate.Update(rescheduleTime)\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\ttask := rescheduled.task\n\t\t\tif task.State() != ctask.TaskStatePending {\n\t\t\t\tpq.Remove()\n\t\t\t\tr.numExecutables--\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tif task.GetAttempt() == 0 {\n\t\t\t\ttask.SetInitialSubmitTime(now)\n\t\t\t}\n\t\t\tsubmitted, err := r.scheduler.TrySubmit(task)\n\t\t\tif err != nil {\n\t\t\t\tif r.isStopped() {\n\t\t\t\t\t// if error is due to shard shutdown\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\t// otherwise it might be error from domain cache etc, add\n\t\t\t\t// the task to reschedule queue so that it can be retried\n\t\t\t\tr.logger.Error(\"Failed to reschedule task\", tag.Error(err))\n\t\t\t}\n\t\t\tif !submitted {\n\t\t\t\tr.timerGate.Update(now.Add(backoff.JitDuration(taskChanFullBackoff, taskChanFullBackoffJitterCoefficient)))\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\tpq.Remove()\n\t\t\tr.numExecutables--\n\t\t}\n\t}\n}\n\nfunc (r *reschedulerImpl) cleanupPQ() {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\tfor key, pq := range r.pqMap {\n\t\tif pq.IsEmpty() {\n\t\t\tdelete(r.pqMap, key)\n\t\t}\n\t}\n}\n\nfunc (r *reschedulerImpl) drain() {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\tfor key, pq := range r.pqMap {\n\t\tfor !pq.IsEmpty() {\n\t\t\tpq.Remove()\n\t\t}\n\t\tdelete(r.pqMap, key)\n\t}\n\n\tr.numExecutables = 0\n}\n\nfunc (r *reschedulerImpl) isStopped() bool {\n\treturn atomic.LoadInt32(&r.status) == common.DaemonStatusStopped\n}\n\nfunc (r *reschedulerImpl) getOrCreatePQLocked(\n\tkey DomainPriorityKey,\n) collection.Queue[rescheduledTask] {\n\tif pq, ok := r.pqMap[key]; ok {\n\t\treturn pq\n\t}\n\n\tpq := r.newPriorityQueue(nil)\n\tr.pqMap[key] = pq\n\treturn pq\n}\n\nfunc (r *reschedulerImpl) newPriorityQueue(\n\titems []rescheduledTask,\n) collection.Queue[rescheduledTask] {\n\treturn collection.NewPriorityQueue(rescheduledTaskCompareLess, items...)\n}\n\nfunc rescheduledTaskCompareLess(\n\tthis rescheduledTask,\n\tthat rescheduledTask,\n) bool {\n\treturn this.rescheduleTime.Before(that.rescheduleTime)\n}\n"
  },
  {
    "path": "service/history/task/rescheduler_test.go",
    "content": "package task\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tctask \"github.com/uber/cadence/common/task\"\n)\n\nfunc newNoopTransferTaskFromDomainID(domainID string) *noopTask {\n\treturn &noopTask{\n\t\tTask: &persistence.DecisionTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: domainID,\n\t\t\t},\n\t\t},\n\t\tstate: ctask.TaskStatePending,\n\t}\n}\n\nfunc newNoopTimerTaskFromDomainID(domainID string, scheduledTime time.Time) *noopTask {\n\treturn &noopTask{\n\t\tTask: &persistence.UserTimerTask{\n\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\tDomainID: domainID,\n\t\t\t},\n\t\t\tTaskData: persistence.TaskData{\n\t\t\t\tVisibilityTimestamp: scheduledTime,\n\t\t\t},\n\t\t},\n\t\tstate: ctask.TaskStatePending,\n\t}\n}\n\nfunc TestReschedulerLifeCycle(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockScheduler := NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tnow := mockTimeSource.Now()\n\n\trescheduler := NewRescheduler(mockScheduler, mockTimeSource, logger, scope)\n\n\trescheduler.Start()\n\n\tfor i := 0; i < 10; i++ {\n\t\tvar task Task\n\t\tif i%2 == 0 {\n\t\t\ttask = newNoopTransferTaskFromDomainID(\"domainID\")\n\t\t} else {\n\t\t\ttask = newNoopTimerTaskFromDomainID(\"domainID\", now)\n\t\t}\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second*time.Duration(i+1)))\n\t\tassert.Equal(t, rescheduler.Size(), i+1)\n\t}\n\n\ttaskCh := make(chan Task, 10)\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(1)\n\tmockTimeSource.Advance(time.Second)\n\t<-taskCh\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(4)\n\tmockTimeSource.Advance(time.Second * 4)\n\tfor i := 0; i < 4; i++ {\n\t\t<-taskCh\n\t}\n\n\tassert.Equal(t, 5, rescheduler.Size())\n\trescheduler.Stop()\n\tassert.Equal(t, 0, rescheduler.Size())\n}\n\nfunc TestReschedulerRescheduleDomains(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockScheduler := NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tnow := mockTimeSource.Now()\n\n\trescheduler := NewRescheduler(mockScheduler, mockTimeSource, logger, scope)\n\n\trescheduler.Start()\n\n\tfor i := 0; i < 10; i++ {\n\t\tvar task Task\n\t\ttask = newNoopTransferTaskFromDomainID(\"X\")\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second*time.Duration(i+1)))\n\t\tassert.Equal(t, rescheduler.Size(), i+1)\n\t}\n\n\tfor i := 0; i < 10; i++ {\n\t\tvar task Task\n\t\ttask = newNoopTransferTaskFromDomainID(\"Y\")\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second*100))\n\t\tassert.Equal(t, rescheduler.Size(), i+11)\n\t}\n\n\tfor i := 0; i < 10; i++ {\n\t\tvar scheduledTime time.Time\n\t\tif i%2 == 0 {\n\t\t\tscheduledTime = now.Add(time.Second * 1000)\n\t\t} else {\n\t\t\tscheduledTime = now\n\t\t}\n\t\ttask := newNoopTimerTaskFromDomainID(\"Z\", scheduledTime)\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second*1000))\n\t\tassert.Equal(t, rescheduler.Size(), i+21)\n\t}\n\n\ttaskCh := make(chan Task, 30)\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(10)\n\t// advance time to reschedule tasks from domain X\n\tmockTimeSource.Advance(time.Second * 10)\n\tfor i := 0; i < 10; i++ {\n\t\ttask := <-taskCh\n\t\tassert.Equal(t, \"X\", task.GetDomainID())\n\t}\n\tassert.Equal(t, 20, rescheduler.Size())\n\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(10)\n\t// reschedule tasks from domain Y immediately even though their reschedule time is 100s in the future\n\trescheduler.RescheduleDomains(map[string]struct{}{\"Y\": {}})\n\tfor i := 0; i < 10; i++ {\n\t\ttask := <-taskCh\n\t\tassert.Equal(t, \"Y\", task.GetDomainID())\n\t}\n\tassert.Equal(t, 10, rescheduler.Size())\n\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(5)\n\t// reschedule tasks from domain Z immediately and verify that tasks with scheduled time in the future are not rescheduled\n\trescheduler.RescheduleDomains(map[string]struct{}{\"Z\": {}})\n\tfor i := 0; i < 5; i++ {\n\t\ttask := <-taskCh\n\t\tassert.Equal(t, \"Z\", task.GetDomainID())\n\t\tassert.False(t, now.After(task.GetTaskKey().GetScheduledTime()))\n\t}\n\tassert.Equal(t, 5, rescheduler.Size())\n\trescheduler.Stop()\n\tassert.Equal(t, 0, rescheduler.Size())\n}\n\nfunc TestReschedulerIgnoreTaskWithStateNotPending(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockScheduler := NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tnow := mockTimeSource.Now()\n\trescheduler := NewRescheduler(mockScheduler, mockTimeSource, logger, scope)\n\n\trescheduler.Start()\n\n\tfor i := 0; i < 10; i++ {\n\t\tvar task *noopTask\n\t\tif i%2 == 0 {\n\t\t\ttask = newNoopTransferTaskFromDomainID(\"domainID\")\n\t\t} else {\n\t\t\ttask = newNoopTransferTaskFromDomainID(\"domainID\")\n\t\t\ttask.state = ctask.TaskStateCanceled\n\t\t}\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second))\n\t\tassert.Equal(t, rescheduler.Size(), i+1)\n\t}\n\n\ttaskCh := make(chan Task, 5)\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(5)\n\tmockTimeSource.Advance(time.Second)\n\tfor i := 0; i < 5; i++ {\n\t\ttask := <-taskCh\n\t\tassert.Equal(t, ctask.TaskStatePending, task.State())\n\t}\n\tassert.Equal(t, 0, rescheduler.Size())\n\trescheduler.Stop()\n}\n\nfunc TestReschedulerProcessorThrottled(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockScheduler := NewMockProcessor(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tnow := mockTimeSource.Now()\n\n\trescheduler := NewRescheduler(mockScheduler, mockTimeSource, logger, scope)\n\n\trescheduler.Start()\n\n\tfor i := 0; i < 10; i++ {\n\t\ttask := newNoopTransferTaskFromDomainID(\"domainID\")\n\t\trescheduler.RescheduleTask(task, now.Add(time.Second))\n\t\tassert.Equal(t, rescheduler.Size(), i+1)\n\t}\n\n\ttaskCh := make(chan Task, 10)\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).DoAndReturn(func(task Task) (bool, error) {\n\t\ttaskCh <- task\n\t\treturn true, nil\n\t}).Times(4)\n\tmockScheduler.EXPECT().TrySubmit(gomock.Any()).Return(false, nil).Times(1)\n\tmockTimeSource.Advance(time.Second)\n\tfor i := 0; i < 4; i++ {\n\t\t<-taskCh\n\t}\n\tassert.Equal(t, 6, rescheduler.Size())\n\trescheduler.Stop()\n\tassert.Equal(t, 0, rescheduler.Size())\n}\n"
  },
  {
    "path": "service/history/task/standby_task_util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\tstandbyActionFn     func(context.Context, execution.Context, execution.MutableState) (interface{}, error)\n\tstandbyPostActionFn func(context.Context, persistence.Task, interface{}, log.Logger) error\n\n\tstandbyCurrentTimeFn func(persistence.Task) (time.Time, error)\n)\n\nvar (\n\terrDomainBecomesActive = errors.New(\"domain becomes active when processing task as standby\")\n)\n\nfunc standbyTaskPostActionNoOp(\n\tctx context.Context,\n\ttaskInfo persistence.Task,\n\tpostActionInfo interface{},\n\tlogger log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\t// return error so task processing logic will retry\n\tlogger.Debug(\"standbyTaskPostActionNoOp return redispatch error so task processing logic will retry\",\n\t\ttag.WorkflowID(taskInfo.GetWorkflowID()),\n\t\ttag.WorkflowRunID(taskInfo.GetRunID()),\n\t\ttag.WorkflowDomainID(taskInfo.GetDomainID()),\n\t\ttag.TaskID(taskInfo.GetTaskID()),\n\t\ttag.TaskType(taskInfo.GetTaskType()),\n\t\ttag.FailoverVersion(taskInfo.GetVersion()),\n\t\ttag.Timestamp(taskInfo.GetVisibilityTimestamp()))\n\treturn &redispatchError{Reason: fmt.Sprintf(\"post action is %T\", postActionInfo)}\n}\n\nfunc standbyTaskPostActionTaskDiscarded(\n\tctx context.Context,\n\ttask persistence.Task,\n\tpostActionInfo interface{},\n\tlogger log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\tlogger.Error(\"Discarding standby task due to task being pending for too long.\",\n\t\ttag.WorkflowID(task.GetWorkflowID()),\n\t\ttag.WorkflowRunID(task.GetRunID()),\n\t\ttag.WorkflowDomainID(task.GetDomainID()),\n\t\ttag.TaskID(task.GetTaskID()),\n\t\ttag.TaskType(task.GetTaskType()),\n\t\ttag.FailoverVersion(task.GetVersion()),\n\t\ttag.Timestamp(task.GetVisibilityTimestamp()))\n\treturn ErrTaskDiscarded\n}\n\ntype (\n\thistoryResendInfo struct {\n\t\t// used by NDC\n\t\tlastEventID      *int64\n\t\tlastEventVersion *int64\n\t}\n\n\tpushActivityToMatchingInfo struct {\n\t\tactivityScheduleToStartTimeout int32\n\t\ttasklist                       types.TaskList\n\t\tpartitionConfig                map[string]string\n\t}\n\n\tpushDecisionToMatchingInfo struct {\n\t\tdecisionScheduleToStartTimeout int32\n\t\ttasklist                       types.TaskList\n\t\tpartitionConfig                map[string]string\n\t}\n)\n\nfunc newPushActivityToMatchingInfo(\n\tactivityScheduleToStartTimeout int32,\n\ttasklist types.TaskList,\n\tpartitionConfig map[string]string,\n) *pushActivityToMatchingInfo {\n\n\treturn &pushActivityToMatchingInfo{\n\t\tactivityScheduleToStartTimeout: activityScheduleToStartTimeout,\n\t\ttasklist:                       tasklist,\n\t\tpartitionConfig:                partitionConfig,\n\t}\n}\n\nfunc newPushDecisionToMatchingInfo(\n\tdecisionScheduleToStartTimeout int32,\n\ttasklist types.TaskList,\n\tpartitionConfig map[string]string,\n) *pushDecisionToMatchingInfo {\n\n\treturn &pushDecisionToMatchingInfo{\n\t\tdecisionScheduleToStartTimeout: decisionScheduleToStartTimeout,\n\t\ttasklist:                       tasklist,\n\t\tpartitionConfig:                partitionConfig,\n\t}\n}\n\nfunc getHistoryResendInfo(\n\tmutableState execution.MutableState,\n) (*historyResendInfo, error) {\n\n\tversionHistories := mutableState.GetVersionHistories()\n\tif versionHistories == nil {\n\t\treturn nil, execution.ErrMissingVersionHistories\n\t}\n\tcurrentBranch, err := versionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlastItem, err := currentBranch.GetLastItem()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &historyResendInfo{\n\t\tlastEventID:      common.Int64Ptr(lastItem.EventID),\n\t\tlastEventVersion: common.Int64Ptr(lastItem.Version),\n\t}, nil\n}\n\nfunc getStandbyPostActionFn(\n\tlogger log.Logger,\n\ttaskInfo persistence.Task,\n\tstandbyNow standbyCurrentTimeFn,\n\tstandbyTaskMissingEventsResendDelay time.Duration,\n\tstandbyTaskMissingEventsDiscardDelay time.Duration,\n\tfetchHistoryStandbyPostActionFn standbyPostActionFn,\n\tdiscardTaskStandbyPostActionFn standbyPostActionFn,\n) standbyPostActionFn {\n\n\ttaskTime := taskInfo.GetVisibilityTimestamp()\n\tresendTime := taskTime.Add(standbyTaskMissingEventsResendDelay)\n\tdiscardTime := taskTime.Add(standbyTaskMissingEventsDiscardDelay)\n\n\ttags := []tag.Tag{\n\t\ttag.WorkflowID(taskInfo.GetWorkflowID()),\n\t\ttag.WorkflowRunID(taskInfo.GetRunID()),\n\t\ttag.WorkflowDomainID(taskInfo.GetDomainID()),\n\t\ttag.TaskID(taskInfo.GetTaskID()),\n\t\ttag.TaskType(int(taskInfo.GetTaskType())),\n\t\ttag.Timestamp(taskInfo.GetVisibilityTimestamp()),\n\t}\n\n\tnow, err := standbyNow(taskInfo)\n\tif err != nil {\n\t\ttags = append(tags, tag.Error(err))\n\t\tlogger.Error(\"getStandbyPostActionFn error getting current time, fallback to standbyTaskPostActionNoOp\", tags...)\n\t\treturn standbyTaskPostActionNoOp\n\t}\n\n\t// now < task start time + StandbyTaskMissingEventsResendDelay\n\tif now.Before(resendTime) {\n\t\tlogger.Debug(\"getStandbyPostActionFn returning standbyTaskPostActionNoOp because now < task start time + StandbyTaskMissingEventsResendDelay\", tags...)\n\t\treturn standbyTaskPostActionNoOp\n\t}\n\n\t// task start time + StandbyTaskMissingEventsResendDelay <= now < task start time + StandbyTaskMissingEventsResendDelay\n\tif now.Before(discardTime) {\n\t\tlogger.Debug(\"getStandbyPostActionFn returning fetchHistoryStandbyPostActionFn because task start time + StandbyTaskMissingEventsResendDelay <= now < task start time + StandbyTaskMissingEventsResendDelay\", tags...)\n\t\treturn fetchHistoryStandbyPostActionFn\n\t}\n\n\t// task start time + StandbyTaskMissingEventsResendDelay <= now\n\tlogger.Debug(\"getStandbyPostActionFn returning discardTaskStandbyPostActionFn because task start time + StandbyTaskMissingEventsResendDelay <= now\", tags...)\n\treturn discardTaskStandbyPostActionFn\n}\n\nfunc getRemoteClusterName(\n\tctx context.Context,\n\tcurrentCluster string,\n\tactiveClusterMgr activecluster.Manager,\n\ttaskInfo persistence.Task,\n) (string, error) {\n\tactiveClusterInfo, err := activeClusterMgr.GetActiveClusterInfoByWorkflow(ctx, taskInfo.GetDomainID(), taskInfo.GetWorkflowID(), taskInfo.GetRunID())\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tif activeClusterInfo.ActiveClusterName == currentCluster {\n\t\treturn \"\", errDomainBecomesActive\n\t}\n\treturn activeClusterInfo.ActiveClusterName, nil\n}\n"
  },
  {
    "path": "service/history/task/standby_task_util_test.go",
    "content": "package task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGetRemoteClusterName(t *testing.T) {\n\ttestDomainID := \"test-domain-id\"\n\ttestWorkflowID := \"test-workflow-id\"\n\ttestRunID := \"test-run-id\"\n\tcurrentCluster := \"cluster-A\"\n\tremoteCluster := \"cluster-B\"\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func(*gomock.Controller) activecluster.Manager\n\t\ttaskInfo       persistence.Task\n\t\texpectedResult string\n\t\texpectedError  error\n\t}{\n\t\t{\n\t\t\tname: \"active-active domain with lookup error\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) activecluster.Manager {\n\t\t\t\tmockActiveClusterMgr := activecluster.NewMockManager(ctrl)\n\n\t\t\t\tmockActiveClusterMgr.EXPECT().\n\t\t\t\t\tGetActiveClusterInfoByWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).\n\t\t\t\t\tReturn(nil, errors.New(\"lookup error\"))\n\n\t\t\t\treturn mockActiveClusterMgr\n\t\t\t},\n\t\t\ttaskInfo: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: \"\",\n\t\t\texpectedError:  errors.New(\"lookup error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain becomes active\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) activecluster.Manager {\n\t\t\t\tmockActiveClusterMgr := activecluster.NewMockManager(ctrl)\n\n\t\t\t\tmockActiveClusterMgr.EXPECT().\n\t\t\t\t\tGetActiveClusterInfoByWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: currentCluster,\n\t\t\t\t\t}, nil)\n\n\t\t\t\treturn mockActiveClusterMgr\n\t\t\t},\n\t\t\ttaskInfo: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: \"\",\n\t\t\texpectedError:  errors.New(\"domain becomes active when processing task as standby\"),\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain successful lookup\",\n\t\t\tsetupMocks: func(ctrl *gomock.Controller) activecluster.Manager {\n\t\t\t\tmockActiveClusterMgr := activecluster.NewMockManager(ctrl)\n\n\t\t\t\tmockActiveClusterMgr.EXPECT().\n\t\t\t\t\tGetActiveClusterInfoByWorkflow(gomock.Any(), testDomainID, testWorkflowID, testRunID).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: remoteCluster,\n\t\t\t\t\t}, nil)\n\n\t\t\t\treturn mockActiveClusterMgr\n\t\t\t},\n\t\t\ttaskInfo: &persistence.DecisionTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   testDomainID,\n\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedResult: remoteCluster,\n\t\t\texpectedError:  nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockActiveClusterMgr := tt.setupMocks(ctrl)\n\n\t\t\tresult, err := getRemoteClusterName(\n\t\t\t\tcontext.Background(),\n\t\t\t\tcurrentCluster,\n\t\t\t\tmockActiveClusterMgr,\n\t\t\t\ttt.taskInfo,\n\t\t\t)\n\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedResult, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/task.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tctask \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n// redispatchError is the error indicating that the timer / transfer task should be redispatched and retried.\ntype redispatchError struct {\n\tReason string\n}\n\n// Error explains why this task should be redispatched\nfunc (r *redispatchError) Error() string {\n\treturn fmt.Sprintf(\"Redispatch reason: %q\", r.Reason)\n}\n\nfunc isRedispatchErr(err error) bool {\n\tvar redispatchErr *redispatchError\n\treturn errors.As(err, &redispatchErr)\n}\n\nconst (\n\tactiveTaskRedispatchInitialInterval  = 5 * time.Second\n\tstandbyTaskRedispatchInitialInterval = 30 * time.Second\n\tredispatchBackoffCoefficient         = 1.05\n\tredispatchMaxBackoffInternval        = 2 * time.Minute\n\tredispatchFailureBackoffInterval     = 2 * time.Second\n)\n\nvar (\n\t// ErrTaskDiscarded is the error indicating that the timer / transfer task is pending for too long and discarded.\n\tErrTaskDiscarded = errors.New(\"passive task pending for too long\")\n\t// ErrTaskPendingActive is the error indicating that the task should be re-dispatched\n\tErrTaskPendingActive = errors.New(\"redispatch the task while the domain is pending-active\")\n\n\tactiveTaskRedispatchPolicy  = createTaskRedispatchPolicy(activeTaskRedispatchInitialInterval)\n\tstandbyTaskRedispatchPolicy = createTaskRedispatchPolicy(standbyTaskRedispatchInitialInterval)\n)\n\ntype (\n\ttaskImpl struct {\n\t\tsync.Mutex\n\t\tpersistence.Task\n\n\t\tshard                    shard.Context\n\t\tstate                    ctask.State\n\t\tpriority                 int\n\t\tattempt                  int\n\t\ttimeSource               clock.TimeSource\n\t\tinitialSubmitTime        time.Time\n\t\tlogger                   log.Logger\n\t\teventLogger              eventLogger\n\t\tscope                    metrics.Scope // initialized when processing task to make the initialization parallel\n\t\ttaskExecutor             Executor\n\t\ttaskProcessor            Processor\n\t\tredispatcher             Redispatcher\n\t\trescheduler              Rescheduler\n\t\tcriticalRetryCount       dynamicproperties.IntPropertyFn\n\t\tisPreviousExecutorActive bool\n\n\t\t// TODO: following three fields should be removed after new task lifecycle is implemented\n\t\ttaskFilter        Filter\n\t\tqueueType         QueueType\n\t\tshouldProcessTask bool\n\t}\n)\n\nfunc NewHistoryTask(\n\tshard shard.Context,\n\ttaskInfo persistence.Task,\n\tqueueType QueueType,\n\tlogger log.Logger,\n\ttaskFilter Filter,\n\ttaskExecutor Executor,\n\ttaskProcessor Processor,\n\tredispatcher Redispatcher,\n\tcriticalRetryCount dynamicproperties.IntPropertyFn,\n) Task {\n\ttimeSource := shard.GetTimeSource()\n\tvar eventLogger eventLogger\n\tif shard.GetConfig().EnableDebugMode &&\n\t\t(queueType == QueueTypeActiveTimer || queueType == QueueTypeActiveTransfer) &&\n\t\tshard.GetConfig().EnableTaskInfoLogByDomainID(taskInfo.GetDomainID()) {\n\t\teventLogger = newEventLogger(logger, timeSource, defaultTaskEventLoggerSize)\n\t\teventLogger.AddEvent(\"Created task\")\n\t}\n\n\treturn &taskImpl{\n\t\tTask:               taskInfo,\n\t\tshard:              shard,\n\t\tstate:              ctask.TaskStatePending,\n\t\tpriority:           noPriority,\n\t\tqueueType:          queueType,\n\t\tscope:              metrics.NoopScope,\n\t\tlogger:             logger,\n\t\teventLogger:        eventLogger,\n\t\tattempt:            0,\n\t\tinitialSubmitTime:  timeSource.Now(),\n\t\ttimeSource:         timeSource,\n\t\tcriticalRetryCount: criticalRetryCount,\n\t\tredispatcher:       redispatcher,\n\t\ttaskFilter:         taskFilter,\n\t\ttaskExecutor:       taskExecutor,\n\t\ttaskProcessor:      taskProcessor,\n\t}\n}\n\nfunc NewHistoryTaskV2(\n\tshard shard.Context,\n\ttaskInfo persistence.Task,\n\tqueueType QueueType,\n\tlogger log.Logger,\n\ttaskExecutor Executor,\n\ttaskProcessor Processor,\n\trescheduler Rescheduler,\n\tcriticalRetryCount dynamicproperties.IntPropertyFn,\n) Task {\n\ttimeSource := shard.GetTimeSource()\n\tvar eventLogger eventLogger\n\tif shard.GetConfig().EnableDebugMode &&\n\t\t(queueType == QueueTypeActiveTimer || queueType == QueueTypeActiveTransfer) &&\n\t\tshard.GetConfig().EnableTaskInfoLogByDomainID(taskInfo.GetDomainID()) {\n\t\teventLogger = newEventLogger(logger, timeSource, defaultTaskEventLoggerSize)\n\t\teventLogger.AddEvent(\"Created task\")\n\t}\n\n\treturn &taskImpl{\n\t\tTask:               taskInfo,\n\t\tshard:              shard,\n\t\tstate:              ctask.TaskStatePending,\n\t\tpriority:           noPriority,\n\t\tqueueType:          queueType,\n\t\tscope:              metrics.NoopScope,\n\t\tlogger:             logger,\n\t\teventLogger:        eventLogger,\n\t\tattempt:            0,\n\t\tinitialSubmitTime:  timeSource.Now(),\n\t\ttimeSource:         timeSource,\n\t\tcriticalRetryCount: criticalRetryCount,\n\t\trescheduler:        rescheduler,\n\t\ttaskFilter:         func(task persistence.Task) (bool, error) { return true, nil },\n\t\ttaskExecutor:       taskExecutor,\n\t\ttaskProcessor:      taskProcessor,\n\t}\n}\n\nfunc (t *taskImpl) Execute() error {\n\tif t.State() != ctask.TaskStatePending {\n\t\treturn nil\n\t}\n\tscheduleLatency := t.timeSource.Now().Sub(t.initialSubmitTime)\n\tvar err error\n\tt.shouldProcessTask, err = t.taskFilter(t.Task)\n\tif err != nil {\n\t\tlogEvent(t.eventLogger, \"TaskFilter execution failed\", err)\n\t\ttime.Sleep(loadDomainEntryForTaskRetryDelay)\n\t\treturn err\n\t}\n\tlogEvent(t.eventLogger, \"Executing task\", t.shouldProcessTask)\n\tif !t.shouldProcessTask {\n\t\treturn nil\n\t}\n\n\texecutionStartTime := t.timeSource.Now()\n\ttaskListTaggedScope := metrics.NoopScope\n\tdefer func() {\n\t\tt.scope.IncCounter(metrics.TaskRequestsPerDomain)\n\t\tprocessingLatency := time.Since(executionStartTime)\n\t\tt.scope.RecordTimer(metrics.TaskProcessingLatencyPerDomain, processingLatency)\n\t\tt.scope.ExponentialHistogram(metrics.ExponentialTaskProcessingLatencyPerDomain, processingLatency)\n\n\t\ttaskListTaggedScope.IncCounter(metrics.TaskRequestsPerTaskList)\n\t\ttaskListTaggedScope.ExponentialHistogram(metrics.ExponentialTaskProcessingLatencyPerTaskList, processingLatency)\n\t}()\n\texecuteResponse, err := t.taskExecutor.Execute(t)\n\tt.scope = executeResponse.Scope\n\ttaskListTaggedScope = t.scope.Tagged(common.GetTaskListTag(t.GetOriginalTaskList(), t.GetOriginalTaskListKind()))\n\tif t.GetAttempt() == 0 {\n\t\ttaskListTaggedScope.ExponentialHistogram(metrics.ExponentialTaskScheduleLatencyPerTaskList, scheduleLatency)\n\t\t// TODO: replace with ExponentialHistogram\n\t\t// domain level metrics for the duration between task being submitted to task scheduler and being executed\n\t\tt.scope.RecordHistogramDuration(metrics.TaskScheduleLatencyPerDomain, scheduleLatency)\n\t}\n\tif t.isPreviousExecutorActive != executeResponse.IsActiveTask {\n\t\tt.resetAttempt()\n\t}\n\tt.isPreviousExecutorActive = executeResponse.IsActiveTask\n\treturn err\n}\n\nfunc (t *taskImpl) resetAttempt() {\n\tt.Lock()\n\tdefer t.Unlock()\n\tt.attempt = 0\n}\n\nfunc (t *taskImpl) HandleErr(err error) (retErr error) {\n\tlogger := t.logger.Helper()\n\n\tdefer func() {\n\t\tif retErr != nil {\n\t\t\tlogEvent(t.eventLogger, \"Failed to handle error\", retErr)\n\n\t\t\tt.Lock()\n\t\t\tdefer t.Unlock()\n\n\t\t\tt.attempt++\n\t\t\tif t.attempt > t.criticalRetryCount() {\n\t\t\t\tt.scope.RecordTimer(metrics.TaskAttemptTimerPerDomain, time.Duration(t.attempt))\n\t\t\t\tt.scope.IntExponentialHistogram(metrics.ExponentialTaskAttemptCountsPerDomain, t.attempt)\n\t\t\t\tlogger.Error(\"Critical error processing task, retrying.\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t\ttag.OperationCritical,\n\t\t\t\t\ttag.TaskType(t.GetTaskType()),\n\t\t\t\t\ttag.AttemptCount(t.attempt),\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\t}()\n\n\tif err == nil {\n\t\treturn nil\n\t}\n\n\tlogEvent(t.eventLogger, \"Handling task processing error\", err)\n\n\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\treturn nil\n\t} else if _, ok := err.(*types.WorkflowExecutionAlreadyCompletedError); ok {\n\t\treturn nil\n\t}\n\n\tif _, ok := t.Task.(*persistence.CloseExecutionTask); ok &&\n\t\terr == execution.ErrMissingWorkflowStartEvent &&\n\t\tt.shard.GetConfig().EnableDropStuckTaskByDomainID(t.GetDomainID()) { // use domainID here to avoid accessing domainCache\n\t\tt.scope.IncCounter(metrics.TransferTaskMissingEventCounterPerDomain)\n\t\tlogger.Error(\"Drop close execution transfer task due to corrupted workflow history\", tag.Error(err), tag.LifeCycleProcessingFailed)\n\t\treturn nil\n\t}\n\n\tif err == errWorkflowBusy {\n\t\tt.scope.IncCounter(metrics.TaskWorkflowBusyPerDomain)\n\t\treturn err\n\t}\n\n\tif err == errWorkflowRateLimited {\n\t\t// metrics are emitted within the rate limiter\n\t\treturn err\n\t}\n\n\t// If the shard were recently closed we just return an error, so we retry in a bit.\n\tvar errShardClosed *shard.ErrShardClosed\n\tif errors.As(err, &errShardClosed) && time.Since(errShardClosed.ClosedAt) < shard.TimeBeforeShardClosedIsError {\n\t\treturn err\n\t}\n\n\t// If the task list is not owned by the host we connected to we just return an error, so we retry in a bit\n\t// with the new membership information.\n\tvar taskListNotOwnedByHostError *cadence_errors.TaskListNotOwnedByHostError\n\tif errors.As(err, &taskListNotOwnedByHostError) {\n\t\tt.scope.IncCounter(metrics.TaskListNotOwnedByHostCounterPerDomain)\n\t\treturn err\n\t}\n\n\t// this is a transient error\n\tif isRedispatchErr(err) {\n\t\tt.scope.IncCounter(metrics.TaskStandbyRetryCounterPerDomain)\n\t\treturn err\n\t}\n\n\t// this is a transient error during graceful failover\n\tif err == ErrTaskPendingActive {\n\t\tt.scope.IncCounter(metrics.TaskPendingActiveCounterPerDomain)\n\t\treturn err\n\t}\n\n\tif err == ErrTaskDiscarded {\n\t\tt.scope.IncCounter(metrics.TaskDiscardedPerDomain)\n\t\terr = nil\n\t}\n\n\tif err == execution.ErrMissingVersionHistories {\n\t\tt.logger.Error(\"Encounter 2DC workflow during task processing.\")\n\t\tt.scope.IncCounter(metrics.TaskUnsupportedPerDomain)\n\t\terr = nil\n\t}\n\n\t// using a fairly long timeout here because domain updates is an async process\n\t// which could take a fair while to be processed by the domain queue, the DB updated\n\t// the domain cache refeshed and then updated here.\n\tvar e *types.DomainNotActiveError\n\tif errors.As(err, &e) || errors.Is(err, types.DomainNotActiveError{}) {\n\t\tif t.timeSource.Now().Sub(t.initialSubmitTime) > 5*cache.DomainCacheRefreshInterval {\n\t\t\tt.scope.IncCounter(metrics.TaskNotActiveCounterPerDomain)\n\t\t\t// If the domain is *still* not active, drop after a while.\n\t\t\treturn nil\n\t\t}\n\n\t\treturn err\n\t}\n\n\tt.scope.IncCounter(metrics.TaskFailuresPerDomain)\n\n\tif _, ok := err.(*persistence.CurrentWorkflowConditionFailedError); ok {\n\t\tlogger.Error(\"More than 2 workflow are running.\", tag.Error(err), tag.LifeCycleProcessingFailed)\n\t\treturn nil\n\t}\n\n\tattempt := t.GetAttempt()\n\tif attempt > stickyTaskMaxRetryCount && common.IsStickyTaskConditionError(err) {\n\t\t// sticky task could end up into endless loop in rare cases and\n\t\t// cause worker to keep getting decision timeout unless restart.\n\t\t// return nil here to break the endless loop\n\t\treturn nil\n\t}\n\n\tlogger.Error(\"Fail to process task\", tag.Error(err), tag.LifeCycleProcessingFailed, tag.AttemptCount(attempt))\n\treturn err\n}\n\nfunc (t *taskImpl) RetryErr(err error) bool {\n\tif t.State() != ctask.TaskStatePending {\n\t\treturn false\n\t}\n\n\tvar errShardClosed *shard.ErrShardClosed\n\tif errors.As(err, &errShardClosed) || err == errWorkflowBusy || isRedispatchErr(err) || err == ErrTaskPendingActive || common.IsContextTimeoutError(err) {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\nfunc (t *taskImpl) Ack() {\n\tlogEvent(t.eventLogger, \"Acked task\")\n\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tif t.state != ctask.TaskStatePending {\n\t\treturn\n\t}\n\n\tt.state = ctask.TaskStateAcked\n\tif t.shouldProcessTask {\n\t\t// Record attempt count as duration so timer mean ≈ average attempt count.\n\t\tt.scope.RecordTimer(metrics.TaskAttemptTimerPerDomain, time.Duration(t.attempt))\n\n\t\tlatency := time.Since(t.initialSubmitTime)\n\t\tqueueLatency := time.Since(t.GetVisibilityTimestamp())\n\t\t// Use IntExponentialHistogram with Mid1To16k buckets (1–64k) for attempt counts\n\t\tt.scope.IntExponentialHistogram(metrics.ExponentialTaskAttemptCountsPerDomain, t.attempt)\n\t\tt.scope.RecordTimer(metrics.TaskLatencyPerDomain, latency)\n\t\tt.scope.ExponentialHistogram(metrics.ExponentialTaskLatencyPerDomain, latency)\n\t\tt.scope.RecordTimer(metrics.TaskQueueLatencyPerDomain, queueLatency)\n\t\tt.scope.ExponentialHistogram(metrics.ExponentialTaskQueueLatencyPerDomain, queueLatency)\n\n\t\ttaskListTaggedScope := t.scope.Tagged(common.GetTaskListTag(t.GetOriginalTaskList(), t.GetOriginalTaskListKind()))\n\t\ttaskListTaggedScope.ExponentialHistogram(metrics.ExponentialTaskLatencyPerTaskList, latency)\n\t\ttaskListTaggedScope.ExponentialHistogram(metrics.ExponentialTaskQueueLatencyPerTaskList, queueLatency)\n\t}\n\n\tif t.eventLogger != nil && t.shouldProcessTask && t.attempt != 0 {\n\t\t// only dump events when the task should be processed and has been retried\n\t\tt.eventLogger.FlushEvents(\"Task processing events\")\n\t}\n}\n\nfunc (t *taskImpl) Nack(err error) {\n\tif t.State() != ctask.TaskStatePending {\n\t\treturn\n\t}\n\n\tlogEvent(t.eventLogger, \"Nacked task\")\n\n\tif t.shouldResubmitOnNack(err) {\n\t\tif submitted, _ := t.taskProcessor.TrySubmit(t); submitted {\n\t\t\treturn\n\t\t}\n\t}\n\n\tif t.rescheduler != nil {\n\t\tt.rescheduler.RescheduleTask(t, t.timeSource.Now().Add(t.backoffDuration(t.GetAttempt())))\n\t} else if t.redispatcher != nil {\n\t\tt.redispatcher.RedispatchTask(t, t.timeSource.Now().Add(t.backoffDuration(t.GetAttempt())))\n\t}\n}\n\nfunc (t *taskImpl) Cancel() {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tif t.state == ctask.TaskStatePending {\n\t\tt.state = ctask.TaskStateCanceled\n\t}\n}\n\nfunc (t *taskImpl) State() ctask.State {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\treturn t.state\n}\n\nfunc (t *taskImpl) Priority() int {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\treturn t.priority\n}\n\nfunc (t *taskImpl) SetPriority(priority int) {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tt.priority = priority\n}\n\nfunc (t *taskImpl) GetShard() shard.Context {\n\treturn t.shard\n}\n\nfunc (t *taskImpl) GetAttempt() int {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\treturn t.attempt\n}\n\nfunc (t *taskImpl) GetInfo() persistence.Task {\n\treturn t.Task\n}\n\nfunc (t *taskImpl) GetQueueType() QueueType {\n\treturn t.queueType\n}\n\nfunc (t *taskImpl) SetInitialSubmitTime(submitTime time.Time) {\n\tt.Lock()\n\tdefer t.Unlock()\n\n\tt.initialSubmitTime = submitTime\n}\n\nfunc (t *taskImpl) shouldResubmitOnNack(err error) bool {\n\t// TODO: for now only resubmit active task on Nack()\n\t// we can also consider resubmit standby tasks that fails due to certain error types\n\t// this may require change the Nack() interface to Nack(error)\n\tif errors.Is(err, errDomainBecomesActive) {\n\t\treturn true\n\t}\n\treturn t.GetAttempt() < activeTaskResubmitMaxAttempts &&\n\t\t(t.queueType == QueueTypeActiveTransfer || t.queueType == QueueTypeActiveTimer || ((t.queueType == QueueTypeTransfer || t.queueType == QueueTypeTimer) && t.isPreviousExecutorActive))\n}\n\nfunc (t *taskImpl) backoffDuration(attempt int) time.Duration {\n\t// TODO: we might need to consider using the same backoff policy for active and standby tasks,\n\t// but we keep them separate for now so that we can compare metrics between history queue v1 and v2\n\tif t.isPreviousExecutorActive {\n\t\treturn activeTaskRedispatchPolicy.ComputeNextDelay(0, attempt)\n\t}\n\treturn standbyTaskRedispatchPolicy.ComputeNextDelay(0, attempt)\n}\n\nfunc logEvent(\n\teventLogger eventLogger,\n\tmsg string,\n\tdetail ...interface{},\n) {\n\tif eventLogger != nil {\n\t\teventLogger.AddEvent(msg, detail...)\n\t}\n}\n\n// getOrCreateDomainTaggedScope returns cached domain-tagged metrics scope if exists\n// otherwise, it creates a new domain-tagged scope, cache and return the scope\nfunc getOrCreateDomainTaggedScope(\n\tshard shard.Context,\n\tscopeIdx metrics.ScopeIdx,\n\tdomainID string,\n\tlogger log.Logger,\n) metrics.Scope {\n\tscopeCache := shard.GetService().GetDomainMetricsScopeCache()\n\tscope, ok := scopeCache.Get(domainID, scopeIdx)\n\tif !ok {\n\t\tdomainTag, err := getDomainTagByID(shard.GetDomainCache(), domainID)\n\t\tscope = shard.GetMetricsClient().Scope(scopeIdx, domainTag)\n\t\tif err == nil {\n\t\t\tscopeCache.Put(domainID, scopeIdx, scope)\n\t\t} else {\n\t\t\tlogger.Error(\"Unable to get domainName\", tag.Error(err))\n\t\t}\n\t}\n\treturn scope\n}\n\nfunc getDomainTagByID(\n\tdomainCache cache.DomainCache,\n\tdomainID string,\n) (metrics.Tag, error) {\n\tdomainName, err := domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn metrics.DomainUnknownTag(), err\n\t}\n\treturn metrics.DomainTag(domainName), nil\n}\n\nfunc createTaskRedispatchPolicy(initialInterval time.Duration) backoff.RetryPolicy {\n\tbackoffPolicy := backoff.NewExponentialRetryPolicy(initialInterval)\n\tbackoffPolicy.SetBackoffCoefficient(redispatchBackoffCoefficient)\n\tbackoffPolicy.SetMaximumInterval(redispatchMaxBackoffInternval)\n\tbackoffPolicy.SetExpirationInterval(backoff.NoInterval)\n\treturn backoffPolicy\n}\n"
  },
  {
    "path": "service/history/task/task_rate_limiter.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination task_rate_limiter_mock.go github.com/uber/cadence/service/history/task RateLimiter\n\npackage task\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tdynamicquotas \"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tRateLimiter interface {\n\t\tAllow(Task) bool\n\t\tWait(context.Context, Task) error\n\t}\n\n\ttaskRateLimiterImpl struct {\n\t\tlogger           log.Logger\n\t\tmetricsScope     metrics.Scope\n\t\tdomainCache      cache.DomainCache\n\t\tlimiters         quotas.ICollection\n\t\tenabled          dynamicproperties.BoolPropertyFn\n\t\tenableShadowMode dynamicproperties.BoolPropertyFnWithDomainFilter\n\t}\n)\n\nfunc NewRateLimiter(\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdomainCache cache.DomainCache,\n\tconfig *config.Config,\n\tcontroller shard.Controller,\n) RateLimiter {\n\trps := func(domain string) int {\n\t\ttotalShards := float64(config.NumberOfShards)\n\t\ttotalRPS := float64(config.TaskSchedulerGlobalDomainRPS(domain))\n\t\tnumShards := float64(controller.NumShards())\n\t\treturn int(totalRPS * numShards / totalShards)\n\t}\n\tlimiterFactory := dynamicquotas.NewSimpleDynamicRateLimiterFactory(rps)\n\treturn &taskRateLimiterImpl{\n\t\tlogger:           logger,\n\t\tmetricsScope:     metricsClient.Scope(metrics.TaskSchedulerRateLimiterScope),\n\t\tdomainCache:      domainCache,\n\t\tenabled:          config.TaskSchedulerEnableRateLimiter,\n\t\tenableShadowMode: config.TaskSchedulerEnableRateLimiterShadowMode,\n\t\tlimiters:         quotas.NewCollection(limiterFactory),\n\t}\n}\n\nfunc (r *taskRateLimiterImpl) Allow(t Task) bool {\n\tif !r.enabled() {\n\t\treturn true\n\t}\n\tlimiter, scope, shadow := r.getLimiter(t)\n\tallow := limiter.Allow()\n\tif allow {\n\t\tscope.IncCounter(metrics.TaskSchedulerAllowedCounterPerDomain)\n\t\treturn true\n\t}\n\tscope.IncCounter(metrics.TaskSchedulerThrottledCounterPerDomain)\n\treturn shadow\n}\n\nfunc (r *taskRateLimiterImpl) Wait(ctx context.Context, t Task) error {\n\tif !r.enabled() {\n\t\treturn nil\n\t}\n\tlimiter, _, _ := r.getLimiter(t)\n\t// wait has kinda complicated semantics and we haven't really used it and it's not super well understood\n\t// the current interface of rate limiter doesn't support shadow mode, so we are not supporting shadow mode for Wait method now\n\t// Besides, this code path is not hit in production right now\n\treturn limiter.Wait(ctx)\n}\n\nfunc (r *taskRateLimiterImpl) getLimiter(t Task) (quotas.Limiter, metrics.Scope, bool) {\n\tdomainID := t.GetDomainID()\n\tdomainName, err := r.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\tr.logger.Warn(\"failed to get domain name from domain cache\", tag.WorkflowDomainID(domainID), tag.Error(err))\n\t}\n\treturn r.limiters.For(domainName), r.metricsScope.Tagged(metrics.DomainTag(domainName)), r.enableShadowMode(domainName)\n}\n"
  },
  {
    "path": "service/history/task/task_rate_limiter_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/task (interfaces: RateLimiter)\n//\n// Generated by this command:\n//\n//\tmockgen -package task -destination task_rate_limiter_mock.go github.com/uber/cadence/service/history/task RateLimiter\n//\n\n// Package task is a generated GoMock package.\npackage task\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockRateLimiter is a mock of RateLimiter interface.\ntype MockRateLimiter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockRateLimiterMockRecorder\n\tisgomock struct{}\n}\n\n// MockRateLimiterMockRecorder is the mock recorder for MockRateLimiter.\ntype MockRateLimiterMockRecorder struct {\n\tmock *MockRateLimiter\n}\n\n// NewMockRateLimiter creates a new mock instance.\nfunc NewMockRateLimiter(ctrl *gomock.Controller) *MockRateLimiter {\n\tmock := &MockRateLimiter{ctrl: ctrl}\n\tmock.recorder = &MockRateLimiterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockRateLimiter) EXPECT() *MockRateLimiterMockRecorder {\n\treturn m.recorder\n}\n\n// Allow mocks base method.\nfunc (m *MockRateLimiter) Allow(arg0 Task) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Allow\", arg0)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Allow indicates an expected call of Allow.\nfunc (mr *MockRateLimiterMockRecorder) Allow(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Allow\", reflect.TypeOf((*MockRateLimiter)(nil).Allow), arg0)\n}\n\n// Wait mocks base method.\nfunc (m *MockRateLimiter) Wait(arg0 context.Context, arg1 Task) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Wait\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Wait indicates an expected call of Wait.\nfunc (mr *MockRateLimiterMockRecorder) Wait(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Wait\", reflect.TypeOf((*MockRateLimiter)(nil).Wait), arg0, arg1)\n}\n"
  },
  {
    "path": "service/history/task/task_rate_limiter_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\tctask \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tnoopTask struct {\n\t\tsync.RWMutex\n\t\tpersistence.Task\n\t\tqueueType QueueType\n\t\tshard     shard.Context\n\t\tattempt   int\n\t\tpriority  int\n\t\tstate     ctask.State\n\t}\n)\n\nfunc (s *noopTask) Execute() error {\n\treturn nil\n}\n\nfunc (s *noopTask) HandleErr(err error) error {\n\treturn nil\n}\n\nfunc (s *noopTask) RetryErr(err error) bool {\n\treturn false\n}\n\nfunc (s *noopTask) Ack() {\n\ts.Lock()\n\tdefer s.Unlock()\n\tif s.state != ctask.TaskStatePending {\n\t\treturn\n\t}\n\ts.state = ctask.TaskStateAcked\n}\n\nfunc (s *noopTask) Nack(err error) {\n\ts.Lock()\n\tdefer s.Unlock()\n\tif s.state != ctask.TaskStatePending {\n\t\treturn\n\t}\n}\n\nfunc (s *noopTask) Cancel() {\n\ts.Lock()\n\tdefer s.Unlock()\n\tif s.state == ctask.TaskStatePending {\n\t\ts.state = ctask.TaskStateCanceled\n\t}\n}\n\nfunc (s *noopTask) State() ctask.State {\n\ts.RLock()\n\tdefer s.RUnlock()\n\treturn s.state\n}\n\nfunc (s *noopTask) Priority() int {\n\ts.RLock()\n\tdefer s.RUnlock()\n\treturn s.priority\n}\n\nfunc (s *noopTask) SetPriority(p int) {\n\ts.priority = p\n}\n\nfunc (s *noopTask) GetShard() shard.Context {\n\treturn s.shard\n}\n\nfunc (s *noopTask) GetQueueType() QueueType {\n\treturn s.queueType\n}\n\nfunc (s *noopTask) GetAttempt() int {\n\treturn s.attempt\n}\n\nfunc (s *noopTask) GetInfo() persistence.Task {\n\treturn s.Task\n}\n\nfunc (s *noopTask) SetInitialSubmitTime(submitTime time.Time) {\n}\n\ntype taskRateLimiterMockDeps struct {\n\tctrl                *gomock.Controller\n\tmockDomainCache     *cache.MockDomainCache\n\tmockShardController *shard.MockController\n\tmockICollection     *quotas.MockICollection\n\tdynamicClient       dynamicconfig.Client\n}\n\nfunc setupMocksForTaskRateLimiter(t *testing.T, mockQuotasCollection bool) (*taskRateLimiterImpl, *taskRateLimiterMockDeps) {\n\tctrl := gomock.NewController(t)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockShardController := shard.NewMockController(ctrl)\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\n\tdeps := &taskRateLimiterMockDeps{\n\t\tctrl:                ctrl,\n\t\tmockDomainCache:     mockDomainCache,\n\t\tmockShardController: mockShardController,\n\t\tdynamicClient:       dynamicClient,\n\t}\n\n\tconfig := config.New(\n\t\tdynamicconfig.NewCollection(\n\t\t\tdynamicClient,\n\t\t\ttestlogger.New(t),\n\t\t),\n\t\t16,\n\t\t1024,\n\t\tfalse,\n\t\t\"hostname\",\n\t)\n\n\trateLimiter := NewRateLimiter(\n\t\ttestlogger.New(t),\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tdeps.mockDomainCache,\n\t\tconfig,\n\t\tdeps.mockShardController,\n\t)\n\tr, ok := rateLimiter.(*taskRateLimiterImpl)\n\trequire.True(t, ok, \"rate limiter type assertion failure\")\n\tif mockQuotasCollection {\n\t\tdeps.mockICollection = quotas.NewMockICollection(ctrl)\n\t\tr.limiters = deps.mockICollection\n\t}\n\treturn r, deps\n}\n\nfunc TestRateLimiterRPS(t *testing.T) {\n\tr, deps := setupMocksForTaskRateLimiter(t, false)\n\n\tdeps.mockShardController.EXPECT().NumShards().Return(8).AnyTimes()\n\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\n\tl := r.limiters.For(\"test-domain\").Limit()\n\tassert.Equal(t, 50, int(l))\n}\n\nfunc TestRateLimiterAllow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname       string\n\t\ttask       Task\n\t\tsetupMocks func(*taskRateLimiterMockDeps)\n\t\texpected   bool\n\t}{\n\t\t{\n\t\t\tname: \"Not rate limited\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain\", nil)\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"test-domain\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Allow().Return(true)\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Not rate limited - domain cache error ignored\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"\", errors.New(\"cache error\"))\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Allow().Return(true)\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Rate limited - shadow mode\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, true))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain\", nil)\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"test-domain\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Allow().Return(false)\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname: \"Rate limited\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain\", nil)\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"test-domain\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Allow().Return(false)\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tr, deps := setupMocksForTaskRateLimiter(t, true)\n\n\t\t\ttc.setupMocks(deps)\n\n\t\t\tallow := r.Allow(tc.task)\n\t\t\tassert.Equal(t, tc.expected, allow)\n\t\t})\n\t}\n}\n\nfunc TestRateLimiterWait(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\ttask        Task\n\t\tsetupMocks  func(*taskRateLimiterMockDeps)\n\t\texpectErr   bool\n\t\texpectedErr string\n\t}{\n\t\t{\n\t\t\tname: \"Not rate limited\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain\", nil)\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"test-domain\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Wait(gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Not rate limited - domain cache error ignored\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"\", errors.New(\"cache error\"))\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Wait(gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Rate limited - error\",\n\t\t\ttask: &noopTask{\n\t\t\t\tTask: &persistence.DecisionTask{\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID: \"test-domain-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func(deps *taskRateLimiterMockDeps) {\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerGlobalDomainRPS, 100))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiter, true))\n\t\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.TaskSchedulerEnableRateLimiterShadowMode, false))\n\t\t\t\tdeps.mockDomainCache.EXPECT().GetDomainName(\"test-domain-id\").Return(\"test-domain\", nil)\n\t\t\t\tlimiter := quotas.NewMockLimiter(deps.ctrl)\n\t\t\t\tdeps.mockICollection.EXPECT().For(\"test-domain\").Return(limiter)\n\t\t\t\tlimiter.EXPECT().Wait(gomock.Any()).Return(errors.New(\"wait error\"))\n\t\t\t},\n\t\t\texpectErr:   true,\n\t\t\texpectedErr: \"wait error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tr, deps := setupMocksForTaskRateLimiter(t, true)\n\n\t\t\ttc.setupMocks(deps)\n\n\t\t\terr := r.Wait(context.Background(), tc.task)\n\t\t\tif tc.expectErr {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedErr)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/task_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tctask \"github.com/uber/cadence/common/task\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\ttaskSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller           *gomock.Controller\n\t\tmockShard            *shard.TestContext\n\t\tmockTaskExecutor     *MockExecutor\n\t\tmockTaskProcessor    *MockProcessor\n\t\tmockTaskRedispatcher *MockRedispatcher\n\t\tmockTaskInfo         *persistence.MockTask\n\n\t\tlogger        log.Logger\n\t\tmaxRetryCount dynamicproperties.IntPropertyFn\n\t}\n)\n\nfunc TestTaskSuite(t *testing.T) {\n\ts := new(taskSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *taskSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID: 10,\n\t\t\tRangeID: 1,\n\t\t},\n\t\tconfig.NewForTest(),\n\t)\n\ts.mockTaskExecutor = NewMockExecutor(s.controller)\n\ts.mockTaskProcessor = NewMockProcessor(s.controller)\n\ts.mockTaskRedispatcher = NewMockRedispatcher(s.controller)\n\ts.mockTaskInfo = persistence.NewMockTask(s.controller)\n\ts.mockTaskInfo.EXPECT().GetDomainID().Return(constants.TestDomainID).AnyTimes()\n\ts.mockTaskInfo.EXPECT().GetOriginalTaskList().Return(\"test-task-list\").AnyTimes()\n\ts.mockTaskInfo.EXPECT().GetOriginalTaskListKind().Return(types.TaskListKindNormal).AnyTimes()\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\n\ts.logger = testlogger.New(s.Suite.T())\n\ts.maxRetryCount = dynamicproperties.GetIntPropertyFn(10)\n}\n\nfunc (s *taskSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *taskSuite) TestExecute_TaskFilterErr() {\n\ttaskFilterErr := errors.New(\"some random error\")\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn false, taskFilterErr\n\t})\n\terr := taskBase.Execute()\n\ts.Equal(taskFilterErr, err)\n}\n\nfunc (s *taskSuite) TestExecute_ExecutionErr() {\n\ttask := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\texecutionErr := errors.New(\"some random error\")\n\ts.mockTaskExecutor.EXPECT().Execute(task).Return(ExecuteResponse{\n\t\tScope:        metrics.NoopScope,\n\t\tIsActiveTask: true,\n\t}, executionErr).Times(1)\n\n\terr := task.Execute()\n\ts.Equal(executionErr, err)\n}\n\nfunc (s *taskSuite) TestExecute_Success() {\n\ttask := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ts.mockTaskExecutor.EXPECT().Execute(task).Return(ExecuteResponse{\n\t\tScope:        metrics.NoopScope,\n\t\tIsActiveTask: true,\n\t}, nil).Times(1)\n\n\terr := task.Execute()\n\ts.NoError(err)\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrEntityNotExists() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := &types.EntityNotExistsError{}\n\ts.NoError(taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrTaskRetry() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := &redispatchError{Reason: \"random-reason\"}\n\ts.Equal(err, taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrTaskDiscarded() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := ErrTaskDiscarded\n\ts.NoError(taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrTargetDomainNotActive() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := &types.DomainNotActiveError{}\n\n\t// we should always return the target domain not active error\n\t// no matter that the submit time is\n\ttaskBase.initialSubmitTime = time.Now().Add(-cache.DomainCacheRefreshInterval*time.Duration(5) - time.Second)\n\ts.Equal(nil, taskBase.HandleErr(err), \"should drop errors after a reasonable time\")\n\n\ttaskBase.initialSubmitTime = time.Now()\n\ts.Equal(err, taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrDomainNotActive() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := &types.DomainNotActiveError{}\n\n\ttaskBase.initialSubmitTime = time.Now().Add(-cache.DomainCacheRefreshInterval*time.Duration(5) - time.Second)\n\ts.NoError(taskBase.HandleErr(err))\n\n\ttaskBase.initialSubmitTime = time.Now()\n\ts.Equal(err, taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrWorkflowRateLimited() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ttaskBase.initialSubmitTime = time.Now()\n\ts.Equal(errWorkflowRateLimited, taskBase.HandleErr(errWorkflowRateLimited))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrShardRecentlyClosed() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ttaskBase.initialSubmitTime = time.Now()\n\n\tshardClosedError := &shard.ErrShardClosed{\n\t\tMsg: \"shard closed\",\n\t\t// The shard was closed within the TimeBeforeShardClosedIsError interval\n\t\tClosedAt: time.Now().Add(-shard.TimeBeforeShardClosedIsError / 2),\n\t}\n\n\ts.Equal(shardClosedError, taskBase.HandleErr(shardClosedError))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrTaskListNotOwnedByHost() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ttaskBase.initialSubmitTime = time.Now()\n\n\ttaskListNotOwnedByHost := &cadence_errors.TaskListNotOwnedByHostError{\n\t\tOwnedByIdentity: \"HostNameOwnedBy\",\n\t\tMyIdentity:      \"HostNameMe\",\n\t\tTasklistName:    \"TaskListName\",\n\t}\n\n\ts.Equal(taskListNotOwnedByHost, taskBase.HandleErr(taskListNotOwnedByHost))\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrCurrentWorkflowConditionFailed() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\terr := &persistence.CurrentWorkflowConditionFailedError{}\n\ts.NoError(taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestHandleErr_UnknownErr() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\t// Need to mock a return value for function GetTaskType\n\t// If don't do, there'll be an error when code goes into the defer function:\n\t// Unexpected call: because: there are no expected calls of the method \"GetTaskType\" for that receiver\n\t// But can't put it in the setup function since it may cause other errors\n\ts.mockTaskInfo.EXPECT().GetTaskType().Return(123)\n\n\t// make sure it will go into the defer function.\n\t// in this case, make it 0 < attempt < stickyTaskMaxRetryCount\n\ttaskBase.attempt = 10\n\n\terr := errors.New(\"some random error\")\n\ts.Equal(err, taskBase.HandleErr(err))\n}\n\nfunc (s *taskSuite) TestTaskCancel() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ttaskBase.Cancel()\n\ts.Equal(ctask.TaskStateCanceled, taskBase.State())\n\n\ts.NoError(taskBase.Execute())\n\n\ttaskBase.Ack()\n\ts.Equal(ctask.TaskStateCanceled, taskBase.State())\n\n\ttaskBase.Nack(nil)\n\ts.Equal(ctask.TaskStateCanceled, taskBase.State())\n\n\ts.False(taskBase.RetryErr(errors.New(\"some random error\")))\n}\n\nfunc (s *taskSuite) TestTaskState() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ts.Equal(ctask.TaskStatePending, taskBase.State())\n\n\ttaskBase.Ack()\n\ts.Equal(ctask.TaskStateAcked, taskBase.State())\n\n\ttaskBase.Nack(nil)\n\ts.Equal(ctask.TaskStateAcked, taskBase.State())\n}\n\nfunc (s *taskSuite) TestTaskPriority() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\tpriority := 10\n\ttaskBase.SetPriority(priority)\n\ts.Equal(priority, taskBase.Priority())\n}\n\nfunc (s *taskSuite) TestTaskNack_ResubmitSucceeded() {\n\ttask := s.newTestTask(\n\t\tfunc(task persistence.Task) (bool, error) {\n\t\t\treturn true, nil\n\t\t},\n\t)\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(task).Return(true, nil).Times(1)\n\n\ttask.Nack(nil)\n\ts.Equal(ctask.TaskStatePending, task.State())\n}\n\nfunc (s *taskSuite) TestTaskNack_DomainBecomesActive() {\n\ttask := s.newTestTask(\n\t\tfunc(task persistence.Task) (bool, error) {\n\t\t\treturn true, nil\n\t\t},\n\t)\n\ttask.queueType = QueueTypeTransfer\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(task).Return(true, nil).Times(1)\n\n\ttask.Nack(errDomainBecomesActive)\n\ts.Equal(ctask.TaskStatePending, task.State())\n}\n\nfunc (s *taskSuite) TestTaskNack_ResubmitFailed() {\n\ttask := s.newTestTask(\n\t\tfunc(task persistence.Task) (bool, error) {\n\t\t\treturn true, nil\n\t\t},\n\t)\n\n\ts.mockTaskProcessor.EXPECT().TrySubmit(task).Return(false, errTaskProcessorNotRunning).Times(1)\n\ts.mockTaskRedispatcher.EXPECT().RedispatchTask(task, gomock.Any()).Times(1)\n\n\ttask.Nack(nil)\n\ts.Equal(ctask.TaskStatePending, task.State())\n}\n\nfunc (s *taskSuite) TestHandleErr_ErrMaxAttempts() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ttaskBase.criticalRetryCount = func(i ...dynamicproperties.FilterOption) int { return 0 }\n\ts.mockTaskInfo.EXPECT().GetTaskType().Return(0)\n\tassert.NotPanics(s.T(), func() {\n\t\ttaskBase.HandleErr(errors.New(\"err\"))\n\t})\n}\n\nfunc (s *taskSuite) TestRetryErr() {\n\ttaskBase := s.newTestTask(func(task persistence.Task) (bool, error) {\n\t\treturn true, nil\n\t})\n\n\ts.Equal(false, taskBase.RetryErr(&shard.ErrShardClosed{}))\n\ts.Equal(false, taskBase.RetryErr(errWorkflowBusy))\n\ts.Equal(false, taskBase.RetryErr(ErrTaskPendingActive))\n\ts.Equal(false, taskBase.RetryErr(context.DeadlineExceeded))\n\ts.Equal(false, taskBase.RetryErr(&redispatchError{Reason: \"random-reason\"}))\n\t// rate limited errors are retried\n\ts.Equal(true, taskBase.RetryErr(errWorkflowRateLimited))\n}\n\nfunc (s *taskSuite) newTestTask(\n\ttaskFilter Filter,\n) *taskImpl {\n\ttaskBase := NewHistoryTask(\n\t\ts.mockShard,\n\t\ts.mockTaskInfo,\n\t\tQueueTypeActiveTransfer,\n\t\ts.logger,\n\t\ttaskFilter,\n\t\ts.mockTaskExecutor,\n\t\ts.mockTaskProcessor,\n\t\ts.mockTaskRedispatcher,\n\t\ts.maxRetryCount,\n\t).(*taskImpl)\n\ttaskBase.scope = s.mockShard.GetMetricsClient().Scope(0)\n\treturn taskBase\n}\n"
  },
  {
    "path": "service/history/task/task_util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\ntype (\n\tmockTaskMatcher struct {\n\t\ttask *MockTask\n\t}\n)\n\n// InitializeLoggerForTask creates a new logger with additional tags for task info\nfunc InitializeLoggerForTask(\n\tshardID int,\n\ttask persistence.Task,\n\tlogger log.Logger,\n) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.ShardID(shardID),\n\t\ttag.TaskID(task.GetTaskID()),\n\t\ttag.TaskVisibilityTimestamp(task.GetVisibilityTimestamp().UnixNano()),\n\t\ttag.FailoverVersion(task.GetVersion()),\n\t\ttag.TaskType(task.GetTaskType()),\n\t\ttag.WorkflowDomainID(task.GetDomainID()),\n\t\ttag.WorkflowID(task.GetWorkflowID()),\n\t\ttag.WorkflowRunID(task.GetRunID()),\n\t)\n}\n\n// GetTransferTaskMetricsScope returns the metrics scope index for transfer task\nfunc GetTransferTaskMetricsScope(\n\ttaskType int,\n\tisActive bool,\n) metrics.ScopeIdx {\n\tswitch taskType {\n\tcase persistence.TransferTaskTypeActivityTask:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskActivityScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskActivityScope\n\tcase persistence.TransferTaskTypeDecisionTask:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskDecisionScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskDecisionScope\n\tcase persistence.TransferTaskTypeCloseExecution:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskCloseExecutionScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskCloseExecutionScope\n\tcase persistence.TransferTaskTypeCancelExecution:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskCancelExecutionScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskCancelExecutionScope\n\tcase persistence.TransferTaskTypeSignalExecution:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskSignalExecutionScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskSignalExecutionScope\n\tcase persistence.TransferTaskTypeStartChildExecution:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskStartChildExecutionScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskStartChildExecutionScope\n\tcase persistence.TransferTaskTypeRecordWorkflowStarted:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskRecordWorkflowStartedScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskRecordWorkflowStartedScope\n\tcase persistence.TransferTaskTypeResetWorkflow:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskResetWorkflowScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskResetWorkflowScope\n\tcase persistence.TransferTaskTypeUpsertWorkflowSearchAttributes:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskUpsertWorkflowSearchAttributesScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskUpsertWorkflowSearchAttributesScope\n\tcase persistence.TransferTaskTypeRecordWorkflowClosed:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskRecordWorkflowClosedScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskRecordWorkflowClosedScope\n\tcase persistence.TransferTaskTypeRecordChildExecutionCompleted:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskRecordChildExecutionCompletedScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskRecordChildExecutionCompletedScope\n\tcase persistence.TransferTaskTypeApplyParentClosePolicy:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveTaskApplyParentClosePolicyScope\n\t\t}\n\t\treturn metrics.TransferStandbyTaskApplyParentClosePolicyScope\n\tdefault:\n\t\tif isActive {\n\t\t\treturn metrics.TransferActiveQueueProcessorScope\n\t\t}\n\t\treturn metrics.TransferStandbyQueueProcessorScope\n\t}\n}\n\n// GetTimerTaskMetricScope returns the metrics scope index for timer task\nfunc GetTimerTaskMetricScope(\n\ttaskType int,\n\tisActive bool,\n) metrics.ScopeIdx {\n\tswitch taskType {\n\tcase persistence.TaskTypeDecisionTimeout:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskDecisionTimeoutScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskDecisionTimeoutScope\n\tcase persistence.TaskTypeActivityTimeout:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskActivityTimeoutScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskActivityTimeoutScope\n\tcase persistence.TaskTypeUserTimer:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskUserTimerScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskUserTimerScope\n\tcase persistence.TaskTypeWorkflowTimeout:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskWorkflowTimeoutScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskWorkflowTimeoutScope\n\tcase persistence.TaskTypeDeleteHistoryEvent:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskDeleteHistoryEventScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskDeleteHistoryEventScope\n\tcase persistence.TaskTypeActivityRetryTimer:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskActivityRetryTimerScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskActivityRetryTimerScope\n\tcase persistence.TaskTypeWorkflowBackoffTimer:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveTaskWorkflowBackoffTimerScope\n\t\t}\n\t\treturn metrics.TimerStandbyTaskWorkflowBackoffTimerScope\n\tdefault:\n\t\tif isActive {\n\t\t\treturn metrics.TimerActiveQueueProcessorScope\n\t\t}\n\t\treturn metrics.TimerStandbyQueueProcessorScope\n\t}\n}\n\n// verifyTaskVersion, will return true if failover version check is successful\nfunc verifyTaskVersion(\n\tshard shard.Context,\n\tlogger log.Logger,\n\tdomainID string,\n\tversion int64,\n\ttaskVersion int64,\n\ttask persistence.Task,\n) (bool, error) {\n\n\t// the first return value is whether this task is valid for further processing\n\tdomainEntry, err := shard.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\tlogger.Debug(fmt.Sprintf(\"Cannot find domainID: %v, err: %v.\", domainID, err))\n\t\treturn false, err\n\t}\n\tif !domainEntry.IsGlobalDomain() {\n\t\tlogger.Debug(fmt.Sprintf(\"DomainID: %v is not active, task: %v version check pass\", domainID, task))\n\t\treturn true, nil\n\t} else if version != taskVersion {\n\t\tlogger.Debug(fmt.Sprintf(\"DomainID: %v is active, task: %v version != target version: %v.\", domainID, task, version))\n\t\treturn false, nil\n\t}\n\tlogger.Debug(fmt.Sprintf(\"DomainID: %v is active, task: %v version == target version: %v.\", domainID, task, version))\n\treturn true, nil\n}\n\n// load mutable state, if mutable state's next event ID <= task event ID, will attempt to refresh\n// if still mutable state's next event ID <= task event ID, will return nil, nil\nfunc loadMutableState(\n\tctx context.Context,\n\twfContext execution.Context,\n\ttask persistence.Task,\n\tmetricsScope metrics.Scope,\n\tlogger log.Logger,\n\teventID int64,\n) (execution.MutableState, error) {\n\tmsBuilder, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t// this could happen if this is a duplicate processing of the task, and the execution has already completed.\n\t\t\treturn nil, nil\n\t\t}\n\t\treturn nil, err\n\t}\n\texecutionInfo := msBuilder.GetExecutionInfo()\n\n\t// check to see if cache needs to be refreshed as we could potentially have stale workflow execution\n\t// the exception is decision consistently fail\n\t// there will be no event generated, thus making the decision schedule ID == next event ID\n\tisDecisionRetry := (task.GetTaskType() == persistence.TaskTypeDecisionTimeout || task.GetTaskType() == persistence.TransferTaskTypeDecisionTask) &&\n\t\texecutionInfo.DecisionScheduleID == eventID &&\n\t\texecutionInfo.DecisionAttempt > 0\n\n\tif eventID >= msBuilder.GetNextEventID() && !isDecisionRetry {\n\t\tmetricsScope.IncCounter(metrics.StaleMutableStateCounter)\n\t\twfContext.Clear()\n\n\t\tmsBuilder, err = wfContext.LoadWorkflowExecution(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\t// after refresh, still mutable state's next event ID <= task ID\n\t\tif eventID >= msBuilder.GetNextEventID() {\n\t\t\tdomainName := msBuilder.GetDomainEntry().GetInfo().Name\n\t\t\tmetricsScope.Tagged(metrics.DomainTag(domainName)).IncCounter(metrics.DataInconsistentCounter)\n\t\t\tlogger.Error(\"Task Processor: task event ID >= MS NextEventID, skip.\",\n\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\ttag.WorkflowDomainID(task.GetDomainID()),\n\t\t\t\ttag.WorkflowID(task.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(task.GetRunID()),\n\t\t\t\ttag.TaskType(task.GetTaskType()),\n\t\t\t\ttag.TaskID(task.GetTaskID()),\n\t\t\t\ttag.WorkflowEventID(eventID),\n\t\t\t\ttag.WorkflowNextEventID(msBuilder.GetNextEventID()),\n\t\t\t)\n\t\t\treturn nil, nil\n\t\t}\n\t}\n\treturn msBuilder, nil\n}\n\nfunc timeoutWorkflow(\n\tmutableState execution.MutableState,\n\teventBatchFirstEventID int64,\n) error {\n\n\tif decision, ok := mutableState.GetInFlightDecision(); ok {\n\t\tif err := execution.FailDecision(\n\t\t\tmutableState,\n\t\t\tdecision,\n\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t_, err := mutableState.AddTimeoutWorkflowEvent(\n\t\teventBatchFirstEventID,\n\t)\n\treturn err\n}\n\nfunc retryWorkflow(\n\tctx context.Context,\n\tmutableState execution.MutableState,\n\teventBatchFirstEventID int64,\n\tparentDomainName string,\n\tcontinueAsNewAttributes *types.ContinueAsNewWorkflowExecutionDecisionAttributes,\n) (execution.MutableState, error) {\n\n\tif decision, ok := mutableState.GetInFlightDecision(); ok {\n\t\tif err := execution.FailDecision(\n\t\t\tmutableState,\n\t\t\tdecision,\n\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\t_, newMutableState, err := mutableState.AddContinueAsNewEvent(\n\t\tctx,\n\t\teventBatchFirstEventID,\n\t\tconstants.EmptyEventID,\n\t\tparentDomainName,\n\t\tcontinueAsNewAttributes,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newMutableState, nil\n}\n\nfunc getWorkflowExecution(\n\ttaskInfo persistence.Task,\n) types.WorkflowExecution {\n\n\treturn types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.GetWorkflowID(),\n\t\tRunID:      taskInfo.GetRunID(),\n\t}\n}\n\nfunc shouldPushToMatching(\n\tctx context.Context,\n\tshard shard.Context,\n\ttaskInfo persistence.Task,\n) (bool, error) {\n\tdomainEntry, err := shard.GetDomainCache().GetDomainByID(taskInfo.GetDomainID())\n\tif err != nil {\n\t\treturn false, err\n\t}\n\n\tif !domainEntry.GetReplicationConfig().IsActiveActive() {\n\t\t// Preserve the behavior of active-standby and local domains. Always push to matching\n\t\treturn true, nil\n\t}\n\n\t// For active-active domains, only push to matching if the workflow is active in current cluster\n\t// We may revisit this logic in the future. Current idea is to not pollute tasklists with passive workflows of active-active domains\n\t// because they would cause head-of-line blocking in the tasklist. Passive task completion logic doesn't apply to active-active domains.\n\tactiveClusterInfo, err := shard.GetActiveClusterManager().GetActiveClusterInfoByWorkflow(ctx, taskInfo.GetDomainID(), taskInfo.GetWorkflowID(), taskInfo.GetRunID())\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tif activeClusterInfo.ActiveClusterName != shard.GetClusterMetadata().GetCurrentClusterName() {\n\t\treturn false, nil\n\t}\n\n\treturn true, nil\n}\n\n// NewMockTaskMatcher creates a gomock matcher for mock Task\nfunc NewMockTaskMatcher(mockTask *MockTask) gomock.Matcher {\n\treturn &mockTaskMatcher{\n\t\ttask: mockTask,\n\t}\n}\n\nfunc (m *mockTaskMatcher) Matches(x interface{}) bool {\n\ttaskPtr, ok := x.(*MockTask)\n\tif !ok {\n\t\treturn false\n\t}\n\treturn taskPtr == m.task\n}\n\nfunc (m *mockTaskMatcher) String() string {\n\treturn fmt.Sprintf(\"is equal to %v\", m.task)\n}\n"
  },
  {
    "path": "service/history/task/task_util_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestInitializeLoggerForTask(t *testing.T) {\n\ttimeSource := clock.NewMockedTimeSource()\n\n\ttestCases := []struct {\n\t\tname string\n\t\ttask persistence.Task\n\t}{\n\t\t{\n\t\t\tname: \"UserTimerTask\",\n\t\t\ttask: &persistence.UserTimerTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tVisibilityTimestamp: timeSource.Now(),\n\t\t\t\t\tVersion:             10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tlogger := InitializeLoggerForTask(1, tc.task, log.NewLogger(zap.NewNop()))\n\t\t\tassert.NotNil(t, logger)\n\t\t})\n\t}\n}\n\nfunc TestGetTransferTaskMetricsScope(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\ttaskType      int\n\t\tisActive      bool\n\t\texpectedScope metrics.ScopeIdx\n\t}{\n\t\t{\n\t\t\tname:          \"TransferTaskTypeActivityTask - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeActivityTask,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskActivityScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeActivityTask - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeActivityTask,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskActivityScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeDecisionTask - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeDecisionTask,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskDecisionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeDecisionTask - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeDecisionTask,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskDecisionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeCloseExecution - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeCloseExecution,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskCloseExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeCloseExecution - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeCloseExecution,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskCloseExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeCancelExecution - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeCancelExecution,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskCancelExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeCancelExecution - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeCancelExecution,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskCancelExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeSignalExecution - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeSignalExecution,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskSignalExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeSignalExecution - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeSignalExecution,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskSignalExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeStartChildExecution - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeStartChildExecution,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskStartChildExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeStartChildExecution - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeStartChildExecution,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskStartChildExecutionScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordWorkflowStarted - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordWorkflowStarted,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskRecordWorkflowStartedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordWorkflowStarted - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordWorkflowStarted,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskRecordWorkflowStartedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeResetWorkflow - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeResetWorkflow,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskResetWorkflowScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeResetWorkflow - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeResetWorkflow,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskResetWorkflowScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeUpsertWorkflowSearchAttributes - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeUpsertWorkflowSearchAttributes,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskUpsertWorkflowSearchAttributesScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeUpsertWorkflowSearchAttributes - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeUpsertWorkflowSearchAttributes,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskUpsertWorkflowSearchAttributesScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordWorkflowClosed - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordWorkflowClosed,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskRecordWorkflowClosedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordWorkflowClosed - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordWorkflowClosed,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskRecordWorkflowClosedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordChildExecutionCompleted - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordChildExecutionCompleted,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskRecordChildExecutionCompletedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeRecordChildExecutionCompleted - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeRecordChildExecutionCompleted,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskRecordChildExecutionCompletedScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeApplyParentClosePolicy - active\",\n\t\t\ttaskType:      persistence.TransferTaskTypeApplyParentClosePolicy,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveTaskApplyParentClosePolicyScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskTypeApplyParentClosePolicy - standby\",\n\t\t\ttaskType:      persistence.TransferTaskTypeApplyParentClosePolicy,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyTaskApplyParentClosePolicyScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskType not caught - active\",\n\t\t\ttaskType:      -100,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TransferActiveQueueProcessorScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TransferTaskType not caught - standby\",\n\t\t\ttaskType:      -100,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TransferStandbyQueueProcessorScope,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tscope := GetTransferTaskMetricsScope(tc.taskType, tc.isActive)\n\t\t\tassert.Equal(t, tc.expectedScope, scope)\n\t\t})\n\t}\n}\n\nfunc TestGetTimerTaskMetricScope(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\ttaskType      int\n\t\tisActive      bool\n\t\texpectedScope metrics.ScopeIdx\n\t}{\n\t\t{\n\t\t\tname:          \"TimerTaskTypeDecisionTimeout - active\",\n\t\t\ttaskType:      persistence.TaskTypeDecisionTimeout,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskDecisionTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeDecisionTimeout - standby\",\n\t\t\ttaskType:      persistence.TaskTypeDecisionTimeout,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskDecisionTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeActivityTimeout - active\",\n\t\t\ttaskType:      persistence.TaskTypeActivityTimeout,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskActivityTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeActivityTimeout - standby\",\n\t\t\ttaskType:      persistence.TaskTypeActivityTimeout,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskActivityTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeUserTimer - active\",\n\t\t\ttaskType:      persistence.TaskTypeUserTimer,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskUserTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeUserTimer - standby\",\n\t\t\ttaskType:      persistence.TaskTypeUserTimer,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskUserTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeWorkflowTimeout - active\",\n\t\t\ttaskType:      persistence.TaskTypeWorkflowTimeout,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskWorkflowTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeWorkflowTimeout - standby\",\n\t\t\ttaskType:      persistence.TaskTypeWorkflowTimeout,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskWorkflowTimeoutScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeActivityRetryTimer - active\",\n\t\t\ttaskType:      persistence.TaskTypeActivityRetryTimer,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskActivityRetryTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeActivityRetryTimer - standby\",\n\t\t\ttaskType:      persistence.TaskTypeActivityRetryTimer,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskActivityRetryTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeWorkflowBackoffTimer - active\",\n\t\t\ttaskType:      persistence.TaskTypeWorkflowBackoffTimer,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskWorkflowBackoffTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeWorkflowBackoffTimer - standby\",\n\t\t\ttaskType:      persistence.TaskTypeWorkflowBackoffTimer,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskWorkflowBackoffTimerScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeDeleteHistoryEvent - active\",\n\t\t\ttaskType:      persistence.TaskTypeDeleteHistoryEvent,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveTaskDeleteHistoryEventScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskTypeDeleteHistoryEvent - standby\",\n\t\t\ttaskType:      persistence.TaskTypeDeleteHistoryEvent,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyTaskDeleteHistoryEventScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskType not caught - active\",\n\t\t\ttaskType:      -100,\n\t\t\tisActive:      true,\n\t\t\texpectedScope: metrics.TimerActiveQueueProcessorScope,\n\t\t},\n\t\t{\n\t\t\tname:          \"TimerTaskType not caught - standby\",\n\t\t\ttaskType:      -100,\n\t\t\tisActive:      false,\n\t\t\texpectedScope: metrics.TimerStandbyQueueProcessorScope,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tscope := GetTimerTaskMetricScope(tc.taskType, tc.isActive)\n\t\t\tassert.Equal(t, tc.expectedScope, scope)\n\t\t})\n\t}\n}\n\nfunc Test_verifyTaskVersion(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(*shard.MockContext, *cache.MockDomainCache)\n\t\tversion   int64\n\t\terr       error\n\t\tresponse  bool\n\t}{\n\t\t{\n\t\t\tname: \"error - could not get domain entry\",\n\t\t\tsetupMock: func(s *shard.MockContext, c *cache.MockDomainCache) {\n\t\t\t\tc.EXPECT().GetDomainByID(constants.TestDomainID).Return(nil, errors.New(\"test error\")).Times(1)\n\t\t\t\ts.EXPECT().GetDomainCache().Return(c).Times(1)\n\t\t\t},\n\t\t\terr:      errors.New(\"test error\"),\n\t\t\tresponse: false,\n\t\t},\n\t\t{\n\t\t\tname: \"true - domain is not global\",\n\t\t\tsetupMock: func(s *shard.MockContext, c *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &persistence.DomainConfig{}, false, nil, 0, nil, 0, 0, 0)\n\t\t\t\tc.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainResponse, nil).Times(1)\n\t\t\t\ts.EXPECT().GetDomainCache().Return(c).Times(1)\n\t\t\t},\n\t\t\terr:      nil,\n\t\t\tresponse: true,\n\t\t},\n\t\t{\n\t\t\tname: \"false - version is different from task version\",\n\t\t\tsetupMock: func(s *shard.MockContext, c *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\t\t\t\tc.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainResponse, nil).Times(1)\n\t\t\t\ts.EXPECT().GetDomainCache().Return(c).Times(1)\n\t\t\t},\n\t\t\tversion:  6,\n\t\t\terr:      nil,\n\t\t\tresponse: false,\n\t\t},\n\t\t{\n\t\t\tname: \"true - version is same as task version\",\n\t\t\tsetupMock: func(s *shard.MockContext, c *cache.MockDomainCache) {\n\t\t\t\tdomainResponse := cache.NewDomainCacheEntryForTest(&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName}, &persistence.DomainConfig{}, true, nil, 0, nil, 0, 0, 0)\n\t\t\t\tc.EXPECT().GetDomainByID(constants.TestDomainID).Return(domainResponse, nil).Times(1)\n\t\t\t\ts.EXPECT().GetDomainCache().Return(c).Times(1)\n\t\t\t},\n\t\t\tversion:  50,\n\t\t\terr:      nil,\n\t\t\tresponse: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\ts := shard.NewMockContext(ctrl)\n\t\t\tl := log.NewNoop()\n\t\t\ttask := &persistence.UserTimerTask{\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion: constants.TestVersion,\n\t\t\t\t},\n\t\t\t}\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\ttc.setupMock(s, mockDomainCache)\n\t\t\tok, err := verifyTaskVersion(s, l, constants.TestDomainID, tc.version, task.Version, task)\n\t\t\tassert.Equal(t, tc.err, err)\n\t\t\tassert.Equal(t, tc.response, ok)\n\t\t})\n\t}\n}\n\nfunc Test_loadMutableState(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                   string\n\t\tsetupMock              func(*execution.MockContext, *execution.MockMutableState)\n\t\ttask                   persistence.Task\n\t\teventID                int64\n\t\terr                    error\n\t\tisMutableStateReturned bool\n\t}{\n\t\t{\n\t\t\tname: \"error - failed to load workflow execution - entity does not exist\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, &types.EntityNotExistsError{}).Times(1)\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"error - failed to load workflow execution - another error\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, errors.New(\"other-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"other-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"error - transfer task scheduleID > next eventID and !isDecisionRetry - error loading workflow execution\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tw.EXPECT().Clear().Times(1),\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, errors.New(\"some-other-error\")).Times(1),\n\t\t\t\t)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tScheduleID: 11,\n\t\t\t},\n\t\t\teventID: 11,\n\t\t\terr:     errors.New(\"some-other-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"no mutable state returned - transfer task scheduleID > next eventID after refresh\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tw.EXPECT().Clear().Times(1),\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tm.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t)\n\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tScheduleID: 11,\n\t\t\t},\n\t\t\teventID: 11,\n\t\t\terr:     nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - transfer task scheduleID < next eventID\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1)\n\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1)\n\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1)\n\t\t\t},\n\t\t\ttask: &persistence.DecisionTask{\n\t\t\t\tScheduleID: 3,\n\t\t\t},\n\t\t\teventID:                3,\n\t\t\tisMutableStateReturned: true,\n\t\t},\n\t\t{\n\t\t\tname: \"error - timer task eventID > next eventID and !isDecisionRetry - error loading workflow execution\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tw.EXPECT().Clear().Times(1),\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(nil, errors.New(\"some-other-error\")).Times(1),\n\t\t\t\t)\n\n\t\t\t},\n\t\t\ttask:    &persistence.DecisionTimeoutTask{},\n\t\t\teventID: 11,\n\t\t\terr:     errors.New(\"some-other-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"no mutable state returned - timer task eventID > next eventID after refresh\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tgomock.InOrder(\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tw.EXPECT().Clear().Times(1),\n\t\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t\tm.EXPECT().GetDomainEntry().Return(constants.TestGlobalDomainEntry).Times(1),\n\t\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1),\n\t\t\t\t)\n\t\t\t},\n\t\t\ttask:    &persistence.DecisionTimeoutTask{},\n\t\t\teventID: 11,\n\t\t\terr:     nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - timer task eventID < next eventID\",\n\t\t\tsetupMock: func(w *execution.MockContext, m *execution.MockMutableState) {\n\t\t\t\tw.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(m, nil).Times(1)\n\t\t\t\tm.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{NextEventID: int64(5), DecisionAttempt: 1, DecisionScheduleID: 1}).Times(1)\n\t\t\t\tm.EXPECT().GetNextEventID().Return(int64(10)).Times(1)\n\t\t\t},\n\t\t\ttask:                   &persistence.DecisionTimeoutTask{},\n\t\t\teventID:                3,\n\t\t\tisMutableStateReturned: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tw := execution.NewMockContext(ctrl)\n\t\t\tm := execution.NewMockMutableState(ctrl)\n\t\t\tmetricsScope := metrics.NoopScope\n\t\t\tl := log.NewNoop()\n\n\t\t\ttc.setupMock(w, m)\n\t\t\tms, err := loadMutableState(context.Background(), w, tc.task, metricsScope, l, tc.eventID)\n\t\t\tassert.Equal(t, tc.err, err)\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Nil(t, ms)\n\t\t\t} else if tc.isMutableStateReturned {\n\t\t\t\tassert.Equal(t, m, ms)\n\t\t\t} else {\n\t\t\t\tassert.Nil(t, ms)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_timeoutWorkflow(t *testing.T) {\n\teventBatchFirstEventID := int64(2)\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(*execution.MockMutableState)\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"error - execution failDecision error\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, errors.New(\"failDecision-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"failDecision-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"failDecision and timeout added to mutable state\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddTimeoutWorkflowEvent(eventBatchFirstEventID).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"error - AddTimeoutWorkflowEvent error\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddTimeoutWorkflowEvent(eventBatchFirstEventID).Return(nil, errors.New(\"timeoutWorkflow-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"timeoutWorkflow-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"success - timeoutWorkflow event added\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddTimeoutWorkflowEvent(eventBatchFirstEventID).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tm := execution.NewMockMutableState(ctrl)\n\n\t\t\ttc.setupMock(m)\n\t\t\terr := timeoutWorkflow(m, eventBatchFirstEventID)\n\t\t\tassert.Equal(t, tc.err, err)\n\t\t})\n\t}\n}\n\nfunc Test_retryWorkflow(t *testing.T) {\n\teventBatchFirstEventID := int64(2)\n\tparentDomainName := \"parent-domain-name\"\n\tcontinueAsNewAttributes := &types.ContinueAsNewWorkflowExecutionDecisionAttributes{}\n\n\ttestCases := []struct {\n\t\tname      string\n\t\tsetupMock func(*execution.MockMutableState)\n\t\terr       error\n\t}{\n\t\t{\n\t\t\tname: \"error - execution failDecision error\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, errors.New(\"failDecision-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"failDecision-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"failDecision and continueAsNew added to mutable state\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddContinueAsNewEvent(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\teventBatchFirstEventID,\n\t\t\t\t\tcommonconstants.EmptyEventID,\n\t\t\t\t\tparentDomainName,\n\t\t\t\t\tcontinueAsNewAttributes,\n\t\t\t\t).Return(nil, m, nil).Times(1)\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"error - AddContinueAsNewEvent error\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddContinueAsNewEvent(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\teventBatchFirstEventID,\n\t\t\t\t\tcommonconstants.EmptyEventID,\n\t\t\t\t\tparentDomainName,\n\t\t\t\t\tcontinueAsNewAttributes,\n\t\t\t\t).Return(nil, nil, errors.New(\"continueAsNew-error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"continueAsNew-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"success - continueAsNew event added\",\n\t\t\tsetupMock: func(m *execution.MockMutableState) {\n\t\t\t\tdecisionInfo := &execution.DecisionInfo{\n\t\t\t\t\tScheduleID: 1,\n\t\t\t\t\tStartedID:  2,\n\t\t\t\t}\n\t\t\t\tm.EXPECT().GetInFlightDecision().Return(decisionInfo, true).Times(1)\n\t\t\t\tm.EXPECT().AddDecisionTaskFailedEvent(\n\t\t\t\t\tdecisionInfo.ScheduleID,\n\t\t\t\t\tdecisionInfo.StartedID,\n\t\t\t\t\ttypes.DecisionTaskFailedCauseForceCloseDecision,\n\t\t\t\t\tnil, execution.IdentityHistoryService,\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\t\"\",\n\t\t\t\t\tint64(0),\n\t\t\t\t\t\"\").Return(nil, nil).Times(1)\n\t\t\t\tm.EXPECT().FlushBufferedEvents().Return(nil).Times(1)\n\t\t\t\tm.EXPECT().AddContinueAsNewEvent(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\teventBatchFirstEventID,\n\t\t\t\t\tcommonconstants.EmptyEventID,\n\t\t\t\t\tparentDomainName,\n\t\t\t\t\tcontinueAsNewAttributes,\n\t\t\t\t).Return(nil, m, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tm := execution.NewMockMutableState(ctrl)\n\t\t\tctx := context.Background()\n\n\t\t\ttc.setupMock(m)\n\t\t\tresp, err := retryWorkflow(ctx, m, eventBatchFirstEventID, \"parent-domain-name\", continueAsNewAttributes)\n\t\t\tassert.Equal(t, tc.err, err)\n\t\t\tif tc.err == nil {\n\t\t\t\tassert.Equal(t, m, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_mocks(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockTask := NewMockTask(ctrl)\n\tresp := NewMockTaskMatcher(mockTask)\n\tassert.NotNil(t, resp)\n\tassert.Equal(t, &mockTaskMatcher{task: mockTask}, resp)\n\n\tmatches := resp.Matches(mockTask)\n\tassert.True(t, matches)\n\n\tmatches = resp.Matches(nil)\n\tassert.False(t, matches)\n\n\tassert.Equal(t, fmt.Sprintf(\"is equal to %v\", mockTask), resp.String())\n}\n\nfunc TestShouldPushToMatching(t *testing.T) {\n\tdomainID := \"domainID\"\n\twfid := \"wfid\"\n\trid := \"rid\"\n\tcurrentClusterName := \"currentCluster\"\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMock      func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager)\n\t\texpectedResult bool\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"failed to get domain\",\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(nil, errors.New(\"failed to get domain\"))\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  \"failed to get domain\",\n\t\t},\n\t\t{\n\t\t\tname: \"not an active-active domain so returns true\",\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(getDomainCacheEntry(true, false), nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain and workflow is active in current cluster so returns true\",\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(getDomainCacheEntry(true, true), nil)\n\t\t\t\tmockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, wfid, rid).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: currentClusterName,\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain and workflow is not active in current cluster so returns false\",\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(getDomainCacheEntry(true, true), nil)\n\t\t\t\tmockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, wfid, rid).\n\t\t\t\t\tReturn(&types.ActiveClusterInfo{\n\t\t\t\t\t\tActiveClusterName: \"otherCluster\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"active-active domain - failed to lookup workflow\",\n\t\t\tsetupMock: func(mockDomainCache *cache.MockDomainCache, mockActiveClusterMgr *activecluster.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(getDomainCacheEntry(true, true), nil)\n\t\t\t\tmockActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), domainID, wfid, rid).\n\t\t\t\t\tReturn(nil, errors.New(\"failed to lookup workflow\"))\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  \"failed to lookup workflow\",\n\t\t},\n\t}\n\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tshard := shard.NewMockContext(ctrl)\n\n\t\t\tdomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\tshard.EXPECT().GetClusterMetadata().Return(cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tCurrentClusterName: currentClusterName,\n\t\t\t\t},\n\t\t\t\tfunc(d string) bool { return false },\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tlog.NewNoop(),\n\t\t\t)).AnyTimes()\n\n\t\t\tshard.EXPECT().GetDomainCache().Return(domainCache)\n\n\t\t\tactiveClusterMgr := activecluster.NewMockManager(ctrl)\n\t\t\tshard.EXPECT().GetActiveClusterManager().Return(activeClusterMgr).AnyTimes()\n\n\t\t\ttest.setupMock(domainCache, activeClusterMgr)\n\n\t\t\tmockTask := NewMockTask(ctrl)\n\t\t\tmockTask.EXPECT().GetDomainID().Return(domainID).AnyTimes()\n\t\t\tmockTask.EXPECT().GetWorkflowID().Return(wfid).AnyTimes()\n\t\t\tmockTask.EXPECT().GetRunID().Return(rid).AnyTimes()\n\n\t\t\tresult, err := shouldPushToMatching(context.Background(), shard, mockTask)\n\t\t\tassert.Equal(t, test.expectedResult, result)\n\t\t\tif test.expectedError != \"\" {\n\t\t\t\tassert.EqualError(t, err, test.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc getDomainCacheEntry(isGlobal, isActiveActive bool) *cache.DomainCacheEntry {\n\tactiveClusters := &types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\"region\": {\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tif !isActiveActive {\n\t\tactiveClusters = nil\n\t}\n\treturn cache.NewDomainCacheEntryForTest(\n\t\tnil,\n\t\tnil,\n\t\tisGlobal,\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusters: activeClusters,\n\t\t},\n\t\t1,\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t1,\n\t)\n}\n"
  },
  {
    "path": "service/history/task/timer_active_task_executor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/simulation\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nconst (\n\tscanWorkflowTimeout = 30 * time.Second\n)\n\nvar (\n\tnormalDecisionTypeTag = metrics.DecisionTypeTag(\"normal\")\n\tstickyDecisionTypeTag = metrics.DecisionTypeTag(\"sticky\")\n)\n\ntype (\n\ttimerActiveTaskExecutor struct {\n\t\t*timerTaskExecutorBase\n\t}\n)\n\n// NewTimerActiveTaskExecutor creates a new task executor for active timer task\nfunc NewTimerActiveTaskExecutor(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconfig *config.Config,\n) Executor {\n\treturn &timerActiveTaskExecutor{\n\t\ttimerTaskExecutorBase: newTimerTaskExecutorBase(\n\t\t\tshard,\n\t\t\tarchiverClient,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\tconfig,\n\t\t),\n\t}\n}\n\nfunc (t *timerActiveTaskExecutor) Execute(task Task) (ExecuteResponse, error) {\n\tsimulation.LogEvents(simulation.E{\n\t\tEventName:  simulation.EventNameExecuteHistoryTask,\n\t\tHost:       t.shard.GetConfig().HostName,\n\t\tShardID:    t.shard.GetShardID(),\n\t\tDomainID:   task.GetDomainID(),\n\t\tWorkflowID: task.GetWorkflowID(),\n\t\tRunID:      task.GetRunID(),\n\t\tPayload: map[string]any{\n\t\t\t\"task_category\": persistence.HistoryTaskCategoryTimer.Name(),\n\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t},\n\t})\n\tscope := getOrCreateDomainTaggedScope(t.shard, GetTimerTaskMetricScope(task.GetTaskType(), true), task.GetDomainID(), t.logger)\n\texecuteResponse := ExecuteResponse{\n\t\tScope:        scope,\n\t\tIsActiveTask: true,\n\t}\n\tswitch timerTask := task.GetInfo().(type) {\n\tcase *persistence.UserTimerTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeUserTimerTimeoutTask(ctx, timerTask)\n\tcase *persistence.ActivityTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeActivityTimeoutTask(ctx, timerTask)\n\tcase *persistence.DecisionTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeDecisionTimeoutTask(ctx, timerTask)\n\tcase *persistence.WorkflowTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeWorkflowTimeoutTask(ctx, timerTask)\n\tcase *persistence.ActivityRetryTimerTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeActivityRetryTimerTask(ctx, timerTask)\n\tcase *persistence.WorkflowBackoffTimerTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeWorkflowBackoffTimerTask(ctx, timerTask)\n\tcase *persistence.DeleteHistoryEventTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, time.Duration(t.config.DeleteHistoryEventContextTimeout())*time.Second)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeDeleteHistoryEventTask(ctx, timerTask)\n\tdefault:\n\t\treturn executeResponse, errUnknownTimerTask\n\t}\n}\n\nfunc (t *timerActiveTaskExecutor) executeUserTimerTimeoutTask(\n\tctx context.Context,\n\ttask *persistence.UserTimerTask,\n) (retError error) {\n\tt.logger.Debug(\"Processing user timer\",\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t\ttag.TaskID(task.TaskID),\n\t)\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, task.EventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\treferenceTime := t.shard.GetTimeSource().Now()\n\tresurrectionCheckMinDelay := t.config.ResurrectionCheckMinDelay(mutableState.GetDomainEntry().GetInfo().Name)\n\tupdateMutableState := false\n\tdebugLog := t.logger.Debug\n\tif t.config.EnableDebugMode && t.config.EnableTimerDebugLogByDomainID(task.DomainID) {\n\t\tdebugLog = t.logger.Info\n\t}\n\n\t// initialized when a timer with delay >= resurrectionCheckMinDelay\n\t// is encountered, so that we don't need to scan history multiple times\n\t// where there're multiple timers with high delay\n\tvar resurrectedTimer map[string]struct{}\n\tscanWorkflowCtx, cancel := context.WithTimeout(t.ctx, scanWorkflowTimeout)\n\tdefer cancel()\n\n\tsortedUserTimers := timerSequence.LoadAndSortUserTimers()\n\tdebugLog(\"Sorted user timers\",\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t\ttag.Counter(len(sortedUserTimers)),\n\t)\n\nLoop:\n\tfor _, timerSequenceID := range sortedUserTimers {\n\t\ttimerInfo, ok := mutableState.GetUserTimerInfoByEventID(timerSequenceID.EventID)\n\t\tif !ok {\n\t\t\terrString := fmt.Sprintf(\"failed to find in user timer event ID: %v\", timerSequenceID.EventID)\n\t\t\tt.logger.Error(errString)\n\t\t\treturn &types.InternalServiceError{Message: errString}\n\t\t}\n\n\t\tdelay, expired := timerSequence.IsExpired(referenceTime, timerSequenceID)\n\t\tdebugLog(\"Processing user timer sequence id\",\n\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\ttag.TaskID(task.TaskID),\n\t\t\ttag.WorkflowTimerID(timerInfo.TimerID),\n\t\t\ttag.WorkflowScheduleID(timerInfo.StartedID),\n\t\t\ttag.Dynamic(\"timer-sequence-id\", timerSequenceID),\n\t\t\ttag.Dynamic(\"timer-info\", timerInfo),\n\t\t\ttag.Dynamic(\"delay\", delay),\n\t\t\ttag.Dynamic(\"expired\", expired),\n\t\t)\n\n\t\tif !expired {\n\t\t\t// timer sequence IDs are sorted, once there is one timer\n\t\t\t// sequence ID not expired, all after that wil not expired\n\t\t\tbreak Loop\n\t\t}\n\n\t\tif delay >= resurrectionCheckMinDelay || resurrectedTimer != nil {\n\t\t\tif resurrectedTimer == nil {\n\t\t\t\t// overwrite the context here as scan history may take a long time to complete\n\t\t\t\t// ctx will also be used by other operations like updateWorkflow\n\t\t\t\tctx = scanWorkflowCtx\n\t\t\t\tresurrectedTimer, err = execution.GetResurrectedTimers(ctx, t.shard, mutableState)\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.logger.Error(\"Timer resurrection check failed\", tag.Error(err))\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif _, ok := resurrectedTimer[timerInfo.TimerID]; ok {\n\t\t\t\t// found timer resurrection\n\t\t\t\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\t\t\t\tt.metricsClient.Scope(metrics.TimerQueueProcessorScope, metrics.DomainTag(domainName)).IncCounter(metrics.TimerResurrectionCounter)\n\t\t\t\tt.logger.Warn(\"Encounter resurrected timer, skip\",\n\t\t\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\t\ttag.TaskID(task.TaskID),\n\t\t\t\t\ttag.WorkflowTimerID(timerInfo.TimerID),\n\t\t\t\t\ttag.WorkflowScheduleID(timerInfo.StartedID), // timerStartedEvent is basically scheduled event\n\t\t\t\t)\n\n\t\t\t\t// remove resurrected timer from mutable state\n\t\t\t\tif err := mutableState.DeleteUserTimer(timerInfo.TimerID); err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tupdateMutableState = true\n\t\t\t\tcontinue Loop\n\t\t\t}\n\t\t}\n\n\t\tif _, err := mutableState.AddTimerFiredEvent(timerInfo.TimerID); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tupdateMutableState = true\n\n\t\tdebugLog(\"User timer fired\",\n\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\ttag.TaskID(task.TaskID),\n\t\t\ttag.WorkflowTimerID(timerInfo.TimerID),\n\t\t\ttag.WorkflowScheduleID(timerInfo.StartedID),\n\t\t\ttag.WorkflowNextEventID(mutableState.GetNextEventID()),\n\t\t)\n\n\t}\n\n\tif !updateMutableState {\n\t\treturn nil\n\t}\n\n\treturn t.updateWorkflowExecution(ctx, wfContext, mutableState, updateMutableState)\n}\n\nfunc (t *timerActiveTaskExecutor) executeActivityTimeoutTask(\n\tctx context.Context,\n\ttask *persistence.ActivityTimeoutTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to find domainID: %v, err: %v\", task.DomainID, err)\n\t}\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, task.EventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\twfType := mutableState.GetWorkflowType()\n\tif wfType == nil {\n\t\treturn fmt.Errorf(\"unable to find workflow type, task %v\", task)\n\t}\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\treferenceTime := t.shard.GetTimeSource().Now()\n\tresurrectionCheckMinDelay := t.config.ResurrectionCheckMinDelay(mutableState.GetDomainEntry().GetInfo().Name)\n\tupdateMutableState := false\n\tscheduleDecision := false\n\n\t// initialized when an activity timer with delay >= resurrectionCheckMinDelay\n\t// is encountered, so that we don't need to scan history multiple times\n\t// where there're multiple timers with high delay\n\tvar resurrectedActivity map[int64]struct{}\n\tscanWorkflowCtx, cancel := context.WithTimeout(t.ctx, scanWorkflowTimeout)\n\tdefer cancel()\n\n\t// need to clear activity heartbeat timer task mask for new activity timer task creation\n\t// NOTE: LastHeartbeatTimeoutVisibilityInSeconds is for deduping heartbeat timer creation as it's possible\n\t// one heartbeat task was persisted multiple times with different taskIDs due to the retry logic\n\t// for updating workflow execution. In that case, only one new heartbeat timeout task should be\n\t// created.\n\tisHeartBeatTask := task.TimeoutType == int(types.TimeoutTypeHeartbeat)\n\tactivityInfo, ok := mutableState.GetActivityInfo(task.EventID)\n\tif isHeartBeatTask && ok && activityInfo.LastHeartbeatTimeoutVisibilityInSeconds <= task.VisibilityTimestamp.Unix() {\n\t\tactivityInfo.TimerTaskStatus = activityInfo.TimerTaskStatus &^ execution.TimerTaskStatusCreatedHeartbeat\n\t\tif err := mutableState.UpdateActivity(activityInfo); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tupdateMutableState = true\n\t}\n\nLoop:\n\tfor _, timerSequenceID := range timerSequence.LoadAndSortActivityTimers() {\n\t\tactivityInfo, ok := mutableState.GetActivityInfo(timerSequenceID.EventID)\n\t\tif !ok || timerSequenceID.Attempt < activityInfo.Attempt {\n\t\t\t// handle 2 cases:\n\t\t\t// 1. !ok\n\t\t\t//  this case can happen since each activity can have 4 timers\n\t\t\t//  and one of those 4 timers may have fired in this loop\n\t\t\t// 2. timerSequenceID.attempt < activityInfo.Attempt\n\t\t\t//  retry could update activity attempt, should not timeouts new attempt\n\t\t\t// 3. it's a resurrected activity and has already been deleted in this loop\n\t\t\tcontinue Loop\n\t\t}\n\n\t\tdelay, expired := timerSequence.IsExpired(referenceTime, timerSequenceID)\n\t\tif !expired {\n\t\t\t// timer sequence IDs are sorted, once there is one timer\n\t\t\t// sequence ID not expired, all after that wil not expired\n\t\t\tbreak Loop\n\t\t}\n\n\t\tif delay >= resurrectionCheckMinDelay || resurrectedActivity != nil {\n\t\t\tif resurrectedActivity == nil {\n\t\t\t\t// overwrite the context here as scan history may take a long time to complete\n\t\t\t\t// ctx will also be used by other operations like updateWorkflow\n\t\t\t\tctx = scanWorkflowCtx\n\t\t\t\tresurrectedActivity, err = execution.GetResurrectedActivities(ctx, t.shard, mutableState)\n\t\t\t\tif err != nil {\n\t\t\t\t\tt.logger.Error(\"Activity resurrection check failed\", tag.Error(err))\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif _, ok := resurrectedActivity[activityInfo.ScheduleID]; ok {\n\t\t\t\t// found activity resurrection\n\t\t\t\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\t\t\t\tt.metricsClient.Scope(metrics.TimerQueueProcessorScope, metrics.DomainTag(domainName)).IncCounter(metrics.ActivityResurrectionCounter)\n\t\t\t\tt.logger.Warn(\"Encounter resurrected activity, skip\",\n\t\t\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\t\ttag.TaskType(task.GetTaskType()),\n\t\t\t\t\ttag.TaskID(task.TaskID),\n\t\t\t\t\ttag.WorkflowActivityID(activityInfo.ActivityID),\n\t\t\t\t\ttag.WorkflowScheduleID(activityInfo.ScheduleID),\n\t\t\t\t)\n\n\t\t\t\t// remove resurrected activity from mutable state\n\t\t\t\tif err := mutableState.DeleteActivity(activityInfo.ScheduleID); err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\tupdateMutableState = true\n\t\t\t\tcontinue Loop\n\t\t\t}\n\t\t}\n\n\t\t// check if it's possible that the timeout is due to activity task lost\n\t\tif timerSequenceID.TimerType == execution.TimerTypeScheduleToStart {\n\t\t\tdomainName, err := t.shard.GetDomainCache().GetDomainName(mutableState.GetExecutionInfo().DomainID)\n\t\t\tif err == nil && activityInfo.ScheduleToStartTimeout >= int32(t.config.ActivityMaxScheduleToStartTimeoutForRetry(domainName).Seconds()) {\n\t\t\t\t// note that we ignore the race condition for the dynamic config value change here as it's only for metric and logging purpose.\n\t\t\t\t// theoratically the check only applies to activities with retry policy\n\t\t\t\t// however for activities without retry policy, we also want to check the potential task lost and emit the metric\n\t\t\t\t// so reuse the same config value as a threshold so that the metric only got emitted if the activity has been started after a long time.\n\t\t\t\tt.metricsClient.Scope(metrics.TimerActiveTaskActivityTimeoutScope, metrics.DomainTag(domainName)).IncCounter(metrics.ActivityLostCounter)\n\t\t\t\tt.logger.Warn(\"Potentially activity task lost\",\n\t\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\t\ttag.WorkflowScheduleID(activityInfo.ScheduleID),\n\t\t\t\t)\n\t\t\t}\n\t\t}\n\n\t\tif ok, err := mutableState.RetryActivity(\n\t\t\tactivityInfo,\n\t\t\texecution.TimerTypeToReason(timerSequenceID.TimerType),\n\t\t\tnil,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t} else if ok {\n\t\t\tupdateMutableState = true\n\t\t\tcontinue Loop\n\t\t}\n\n\t\tt.emitTimeoutMetricScopeWithDomainTag(\n\t\t\tmutableState.GetExecutionInfo().DomainID,\n\t\t\tmetrics.TimerActiveTaskActivityTimeoutScope,\n\t\t\ttimerSequenceID.TimerType,\n\t\t\tmetrics.WorkflowTypeTag(wfType.GetName()),\n\t\t)\n\n\t\tt.logger.Info(\"Activity timed out\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.WorkflowDomainID(task.GetDomainID()),\n\t\t\ttag.WorkflowID(task.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(task.GetRunID()),\n\t\t\ttag.ScheduleAttempt(task.Attempt),\n\t\t\ttag.FailoverVersion(task.GetVersion()),\n\t\t\ttag.ActivityTimeoutType(shared.TimeoutType(timerSequenceID.TimerType)),\n\t\t)\n\n\t\tif _, err := mutableState.AddActivityTaskTimedOutEvent(\n\t\t\tactivityInfo.ScheduleID,\n\t\t\tactivityInfo.StartedID,\n\t\t\texecution.TimerTypeToInternal(timerSequenceID.TimerType),\n\t\t\tactivityInfo.Details,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tupdateMutableState = true\n\t\tscheduleDecision = true\n\t}\n\n\tif !updateMutableState {\n\t\treturn nil\n\t}\n\treturn t.updateWorkflowExecution(ctx, wfContext, mutableState, scheduleDecision)\n}\n\nfunc (t *timerActiveTaskExecutor) executeDecisionTimeoutTask(\n\tctx context.Context,\n\ttask *persistence.DecisionTimeoutTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to find domainID: %v, err: %v\", task.DomainID, err)\n\t}\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, task.EventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\twfType := mutableState.GetWorkflowType()\n\tif wfType == nil {\n\t\treturn fmt.Errorf(\"unable to find workflow type, task %v\", task)\n\t}\n\n\tscheduleID := task.EventID\n\tdecision, ok := mutableState.GetDecisionInfo(scheduleID)\n\tif !ok {\n\t\tt.logger.Debug(\"Potentially duplicate\", tag.TaskID(task.TaskID), tag.WorkflowScheduleID(scheduleID), tag.TaskType(persistence.TaskTypeDecisionTimeout))\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, decision.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tif decision.Attempt != task.ScheduleAttempt {\n\t\treturn nil\n\t}\n\n\tscheduleDecision := false\n\tisStickyDecision := mutableState.GetExecutionInfo().StickyTaskList != \"\"\n\tdecisionTypeTag := normalDecisionTypeTag\n\tif isStickyDecision {\n\t\tdecisionTypeTag = stickyDecisionTypeTag\n\t}\n\ttags := []metrics.Tag{metrics.WorkflowTypeTag(wfType.GetName()), decisionTypeTag}\n\tswitch execution.TimerTypeFromInternal(types.TimeoutType(task.TimeoutType)) {\n\tcase execution.TimerTypeStartToClose:\n\t\tt.emitTimeoutMetricScopeWithDomainTag(\n\t\t\tmutableState.GetExecutionInfo().DomainID,\n\t\t\tmetrics.TimerActiveTaskDecisionTimeoutScope,\n\t\t\texecution.TimerTypeStartToClose,\n\t\t\ttags...,\n\t\t)\n\t\tif _, err := mutableState.AddDecisionTaskTimedOutEvent(\n\t\t\tdecision.ScheduleID,\n\t\t\tdecision.StartedID,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tmaxAttempts := t.config.DecisionRetryMaxAttempts(domainName)\n\t\tenforceDecisionTaskAttempts := t.config.EnforceDecisionTaskAttempts(domainName)\n\t\tif enforceDecisionTaskAttempts && maxAttempts > 0 && mutableState.GetExecutionInfo().DecisionAttempt > int64(maxAttempts) {\n\t\t\tmessage := \"Decision attempt exceeds limit. Last decision attempt failed due to start-to-close timeout.\"\n\t\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\t\tt.logger.Error(message,\n\t\t\t\ttag.WorkflowDomainID(executionInfo.DomainID),\n\t\t\t\ttag.WorkflowID(executionInfo.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(executionInfo.RunID))\n\t\t\tt.metricsClient.IncCounter(metrics.TimerActiveTaskDecisionTimeoutScope, metrics.DecisionRetriesExceededCounter)\n\n\t\t\tif _, err := mutableState.AddWorkflowExecutionTerminatedEvent(\n\t\t\t\tmutableState.GetNextEventID(),\n\t\t\t\tcommon.FailureReasonDecisionAttemptsExceedsLimit,\n\t\t\t\t[]byte(message),\n\t\t\t\texecution.IdentityHistoryService,\n\t\t\t); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\tscheduleDecision = true\n\t\t}\n\n\tcase execution.TimerTypeScheduleToStart:\n\t\tif decision.StartedID != constants.EmptyEventID {\n\t\t\t// decision has already started\n\t\t\treturn nil\n\t\t}\n\n\t\tif !isStickyDecision {\n\t\t\tt.logger.Warn(\"Potential lost normal decision task\",\n\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\ttag.WorkflowDomainID(task.GetDomainID()),\n\t\t\t\ttag.WorkflowID(task.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(task.GetRunID()),\n\t\t\t\ttag.WorkflowScheduleID(scheduleID),\n\t\t\t\ttag.ScheduleAttempt(task.ScheduleAttempt),\n\t\t\t\ttag.FailoverVersion(task.GetVersion()),\n\t\t\t)\n\t\t}\n\n\t\tt.emitTimeoutMetricScopeWithDomainTag(\n\t\t\tmutableState.GetExecutionInfo().DomainID,\n\t\t\tmetrics.TimerActiveTaskDecisionTimeoutScope,\n\t\t\texecution.TimerTypeScheduleToStart,\n\t\t\ttags...,\n\t\t)\n\t\t_, err := mutableState.AddDecisionTaskScheduleToStartTimeoutEvent(scheduleID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tscheduleDecision = true\n\t}\n\n\treturn t.updateWorkflowExecution(ctx, wfContext, mutableState, scheduleDecision)\n}\n\nfunc (t *timerActiveTaskExecutor) executeWorkflowBackoffTimerTask(\n\tctx context.Context,\n\ttask *persistence.WorkflowBackoffTimerTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tif task.TimeoutType == persistence.WorkflowBackoffTimeoutTypeRetry {\n\t\tt.metricsClient.IncCounter(metrics.TimerActiveTaskWorkflowBackoffTimerScope, metrics.WorkflowRetryBackoffTimerCount)\n\t} else {\n\t\tt.metricsClient.IncCounter(metrics.TimerActiveTaskWorkflowBackoffTimerScope, metrics.WorkflowCronBackoffTimerCount)\n\t}\n\n\tif mutableState.HasProcessedOrPendingDecision() {\n\t\t// already has decision task\n\t\treturn nil\n\t}\n\n\t// Check if this is a cron backoff for the first decision task\n\t// When the first decision is scheduled after the backoff, we need to trigger a visibility update\n\t// to change ExecutionStatus from PENDING to STARTED\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tisCronBackoff := task.TimeoutType == persistence.WorkflowBackoffTimeoutTypeCron\n\tisFirstDecision := executionInfo.DecisionScheduleID == constants.EmptyEventID\n\n\tif isCronBackoff && isFirstDecision {\n\t\t// Only create upsert task if advanced visibility is enabled\n\t\t// For basic DB visibility, the ExecutionStatus is already set to STARTED at workflow start\n\t\t// to avoid showing misleading PENDING status for running workflows.\n\t\t// With advanced visibility, we can update PENDING -> STARTED via upsert.\n\t\tif common.IsAdvancedVisibilityWritingEnabled(\n\t\t\tt.config.WriteVisibilityStoreName(),\n\t\t\tt.config.IsAdvancedVisConfigExist,\n\t\t) {\n\t\t\t// Add UpsertWorkflowSearchAttributes task to trigger visibility update\n\t\t\t// This will update ExecutionStatus from PENDING to STARTED in visibility\n\t\t\t// The status is calculated dynamically in the upsert handler based on decision task state\n\t\t\tmutableState.AddTransferTasks(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\tDomainID:   executionInfo.DomainID,\n\t\t\t\t\tWorkflowID: executionInfo.WorkflowID,\n\t\t\t\t\tRunID:      executionInfo.RunID,\n\t\t\t\t},\n\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\tVersion: mutableState.GetCurrentVersion(),\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\n\t// schedule first decision task\n\treturn t.updateWorkflowExecution(ctx, wfContext, mutableState, true)\n}\n\nfunc (t *timerActiveTaskExecutor) executeActivityRetryTimerTask(\n\tctx context.Context,\n\ttask *persistence.ActivityRetryTimerTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, task.EventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\t// generate activity task\n\tscheduledID := task.EventID\n\tactivityInfo, ok := mutableState.GetActivityInfo(scheduledID)\n\tif !ok || task.Attempt < int64(activityInfo.Attempt) || activityInfo.StartedID != constants.EmptyEventID {\n\t\tif ok {\n\t\t\tt.logger.Info(\"Duplicate activity retry timer task\",\n\t\t\t\ttag.WorkflowID(mutableState.GetExecutionInfo().WorkflowID),\n\t\t\t\ttag.WorkflowRunID(mutableState.GetExecutionInfo().RunID),\n\t\t\t\ttag.WorkflowDomainID(mutableState.GetExecutionInfo().DomainID),\n\t\t\t\ttag.WorkflowScheduleID(activityInfo.ScheduleID),\n\t\t\t\ttag.Attempt(activityInfo.Attempt),\n\t\t\t\ttag.FailoverVersion(activityInfo.Version),\n\t\t\t\ttag.TimerTaskStatus(activityInfo.TimerTaskStatus),\n\t\t\t\ttag.ScheduleAttempt(task.Attempt))\n\t\t}\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, activityInfo.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tdomainID := task.DomainID\n\ttargetDomainID := domainID\n\tif activityInfo.DomainID != \"\" {\n\t\ttargetDomainID = activityInfo.DomainID\n\t} else {\n\t\t// TODO remove this block after Mar, 1th, 2020\n\t\t//  previously, DomainID in activity info is not used, so need to get\n\t\t//  schedule event from DB checking whether activity to be scheduled\n\t\t//  belongs to this domain\n\t\tscheduledEvent, err := mutableState.GetActivityScheduledEvent(ctx, scheduledID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif scheduledEvent.ActivityTaskScheduledEventAttributes.GetDomain() != \"\" {\n\t\t\tdomainEntry, err := t.shard.GetDomainCache().GetDomain(scheduledEvent.ActivityTaskScheduledEventAttributes.GetDomain())\n\t\t\tif err != nil {\n\t\t\t\treturn &types.InternalServiceError{Message: \"unable to re-schedule activity across domain.\"}\n\t\t\t}\n\t\t\ttargetDomainID = domainEntry.GetInfo().ID\n\t\t}\n\t}\n\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: task.WorkflowID,\n\t\tRunID:      task.RunID}\n\ttaskList := &types.TaskList{\n\t\tName: activityInfo.TaskList,\n\t}\n\tscheduleToStartTimeout := activityInfo.ScheduleToStartTimeout\n\n\trelease(nil) // release earlier as we don't need the lock anymore\n\n\tshouldPush, err := shouldPushToMatching(ctx, t.shard, task)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !shouldPush {\n\t\treturn nil\n\t}\n\n\t_, err = t.shard.GetService().GetMatchingClient().AddActivityTask(ctx, &types.AddActivityTaskRequest{\n\t\tDomainUUID:                    targetDomainID,\n\t\tSourceDomainUUID:              domainID,\n\t\tExecution:                     &execution,\n\t\tTaskList:                      taskList,\n\t\tScheduleID:                    scheduledID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(scheduleToStartTimeout),\n\t\tPartitionConfig:               mutableState.GetExecutionInfo().PartitionConfig,\n\t})\n\treturn err\n}\n\nfunc (t *timerActiveTaskExecutor) executeWorkflowTimeoutTask(\n\tctx context.Context,\n\ttask *persistence.WorkflowTimeoutTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tstartVersion, err := mutableState.GetStartVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\tok, err := verifyTaskVersion(t.shard, t.logger, task.DomainID, startVersion, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\teventBatchFirstEventID := mutableState.GetNextEventID()\n\n\ttimeoutReason := execution.TimerTypeToReason(execution.TimerTypeStartToClose)\n\tbackoffInterval := mutableState.GetRetryBackoffDuration(timeoutReason)\n\tcontinueAsNewInitiator := types.ContinueAsNewInitiatorRetryPolicy\n\tif backoffInterval == backoff.NoBackoff {\n\t\t// check if a cron backoff is needed\n\t\tbackoffInterval, err = mutableState.GetCronBackoffDuration(ctx)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tcontinueAsNewInitiator = types.ContinueAsNewInitiatorCronSchedule\n\t}\n\t// ignore event id\n\tisCanceled, _ := mutableState.IsCancelRequested()\n\tif isCanceled || backoffInterval == backoff.NoBackoff {\n\t\tif err := timeoutWorkflow(mutableState, eventBatchFirstEventID); err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\t// We apply the update to execution using optimistic concurrency.  If it fails due to a conflict than reload\n\t\t// the history and try the operation again.\n\t\treturn t.updateWorkflowExecution(ctx, wfContext, mutableState, false)\n\t}\n\n\t// workflow timeout, but a retry or cron is needed, so we do continue as new to retry or cron\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tstartAttributes := startEvent.WorkflowExecutionStartedEventAttributes\n\tcontinueAsNewAttributes := &types.ContinueAsNewWorkflowExecutionDecisionAttributes{\n\t\tWorkflowType:                        startAttributes.WorkflowType,\n\t\tTaskList:                            startAttributes.TaskList,\n\t\tInput:                               startAttributes.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: startAttributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      startAttributes.TaskStartToCloseTimeoutSeconds,\n\t\tBackoffStartIntervalInSeconds:       common.Int32Ptr(int32(backoffInterval.Seconds())),\n\t\tRetryPolicy:                         startAttributes.RetryPolicy,\n\t\tInitiator:                           continueAsNewInitiator.Ptr(),\n\t\tFailureReason:                       common.StringPtr(timeoutReason),\n\t\tCronSchedule:                        mutableState.GetExecutionInfo().CronSchedule,\n\t\tHeader:                              startAttributes.Header,\n\t\tMemo:                                startAttributes.Memo,\n\t\tSearchAttributes:                    startAttributes.SearchAttributes,\n\t\tJitterStartSeconds:                  startAttributes.JitterStartSeconds,\n\t\tCronOverlapPolicy:                   startAttributes.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy:        startAttributes.ActiveClusterSelectionPolicy,\n\t}\n\tnewMutableState, err := retryWorkflow(\n\t\tctx,\n\t\tmutableState,\n\t\teventBatchFirstEventID,\n\t\tstartAttributes.GetParentWorkflowDomain(),\n\t\tcontinueAsNewAttributes,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tnewExecutionInfo := newMutableState.GetExecutionInfo()\n\treturn wfContext.UpdateWorkflowExecutionWithNewAsActive(\n\t\tctx,\n\t\tt.shard.GetTimeSource().Now(),\n\t\texecution.NewContext(\n\t\t\tnewExecutionInfo.DomainID,\n\t\t\ttypes.WorkflowExecution{\n\t\t\t\tWorkflowID: newExecutionInfo.WorkflowID,\n\t\t\t\tRunID:      newExecutionInfo.RunID,\n\t\t\t},\n\t\t\tt.shard,\n\t\t\tt.shard.GetExecutionManager(),\n\t\t\tt.logger,\n\t\t),\n\t\tnewMutableState,\n\t)\n}\n\nfunc (t *timerActiveTaskExecutor) updateWorkflowExecution(\n\tctx context.Context,\n\twfContext execution.Context,\n\tmutableState execution.MutableState,\n\tscheduleNewDecision bool,\n) error {\n\tvar err error\n\tif scheduleNewDecision {\n\t\t// Schedule a new decision.\n\t\terr = execution.ScheduleDecision(mutableState)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tnow := t.shard.GetTimeSource().Now()\n\tt.logger.Debugf(\"timerActiveTaskExecutor.updateWorkflowExecution calling UpdateWorkflowExecutionAsActive for wfID %s\",\n\t\tmutableState.GetExecutionInfo().WorkflowID,\n\t)\n\terr = wfContext.UpdateWorkflowExecutionAsActive(ctx, now)\n\tif err != nil {\n\t\t// if is shard ownership error, the shard context will stop the entire history engine\n\t\t// we don't need to explicitly stop the queue processor here\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc (t *timerActiveTaskExecutor) emitTimeoutMetricScopeWithDomainTag(\n\tdomainID string,\n\tscope metrics.ScopeIdx,\n\ttimerType execution.TimerType,\n\ttags ...metrics.Tag,\n) {\n\tdomainTag, err := getDomainTagByID(t.shard.GetDomainCache(), domainID)\n\tif err != nil {\n\t\treturn\n\t}\n\ttags = append(tags, domainTag)\n\n\tmetricsScope := t.metricsClient.Scope(scope, tags...)\n\tswitch timerType {\n\tcase execution.TimerTypeScheduleToStart:\n\t\tmetricsScope.IncCounter(metrics.ScheduleToStartTimeoutCounter)\n\tcase execution.TimerTypeScheduleToClose:\n\t\tmetricsScope.IncCounter(metrics.ScheduleToCloseTimeoutCounter)\n\tcase execution.TimerTypeStartToClose:\n\t\tmetricsScope.IncCounter(metrics.StartToCloseTimeoutCounter)\n\tcase execution.TimerTypeHeartbeat:\n\t\tmetricsScope.IncCounter(metrics.HeartbeatTimeoutCounter)\n\t}\n}\n"
  },
  {
    "path": "service/history/task/timer_active_task_executor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n)\n\ntype (\n\ttimerActiveTaskExecutorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockEngine         *engine.MockEngine\n\t\tmockDomainCache    *cache.MockDomainCache\n\t\tmockMatchingClient *matching.MockClient\n\n\t\tmockExecutionMgr *mocks.ExecutionManager\n\t\tmockHistoryV2Mgr *mocks.HistoryV2Manager\n\n\t\texecutionCache execution.Cache\n\t\tlogger         log.Logger\n\t\tdomainID       string\n\t\tdomain         string\n\t\tdomainEntry    *cache.DomainCacheEntry\n\t\tversion        int64\n\n\t\ttimeSource              clock.MockedTimeSource\n\t\ttimerActiveTaskExecutor *timerActiveTaskExecutor\n\t}\n)\n\nfunc TestTimerActiveTaskExecutorSuite(t *testing.T) {\n\ts := new(timerActiveTaskExecutorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) SetupSuite() {\n\n}\n\nfunc (s *timerActiveTaskExecutorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.domainID = constants.TestDomainID\n\ts.domain = constants.TestDomainName\n\ts.domainEntry = constants.TestGlobalDomainEntry\n\ts.version = s.domainEntry.GetFailoverVersion()\n\n\ts.timeSource = clock.NewMockedTimeSource()\n\n\ts.controller = gomock.NewController(s.T())\n\n\tconfig := config.NewForTest()\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig,\n\t)\n\ts.mockShard.SetEventsCache(events.NewCache(\n\t\ts.mockShard.GetShardID(),\n\t\ts.mockShard.GetHistoryManager(),\n\t\ts.mockShard.GetConfig(),\n\t\ts.mockShard.GetLogger(),\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.mockShard.GetDomainCache(),\n\t))\n\ts.mockShard.Resource.TimeSource = s.timeSource\n\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.mockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTransferTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTimerTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewReplicationTasks(gomock.Any()).AnyTimes()\n\ts.mockShard.SetEngine(s.mockEngine)\n\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockMatchingClient = s.mockShard.Resource.MatchingClient\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\t// ack manager will use the domain information\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestGlobalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestGlobalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\ts.executionCache = execution.NewCache(s.mockShard)\n\ts.timerActiveTaskExecutor = NewTimerActiveTaskExecutor(\n\t\ts.mockShard,\n\t\tnil,\n\t\ts.executionCache,\n\t\ts.logger,\n\t\ts.mockShard.GetMetricsClient(),\n\t\tconfig,\n\t).(*timerActiveTaskExecutor)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessUserTimerTimeout_Fire() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerID := \"timer\"\n\ttimerTimeout := 2 * time.Second\n\tevent, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID, int64(timerTimeout.Seconds()))\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\t_, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetUserTimerInfo(timerID)\n\ts.False(ok)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessUserTimerTimeout_Noop() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerID := \"timer\"\n\ttimerTimeout := 2 * time.Second\n\tevent, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID, int64(timerTimeout.Seconds()))\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tevent = test.AddTimerFiredEvent(mutableState, timerID)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessUserTimerTimeout_Resurrected() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// schedule two timers\n\ttimerID1 := \"timer1\"\n\ttimerTimeout1 := 2 * time.Second\n\tstartEvent1, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID1, int64(timerTimeout1.Seconds()))\n\ttimerID2 := \"timer2\"\n\ttimerTimeout2 := 5 * time.Second\n\tstartEvent2, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID2, int64(timerTimeout2.Seconds()))\n\n\t// fire timer 1\n\tfiredEvent1 := test.AddTimerFiredEvent(mutableState, timerID1)\n\tmutableState.FlushBufferedEvents()\n\t// there should be a decision scheduled event after timer1 is fired\n\t// omitted here to make the test easier to read\n\n\t// create timer task for timer2\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks() // remove existing timer task for timerID1\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, firedEvent1.ID, firedEvent1.Version)\n\ts.NoError(err)\n\t// add resurrected timer info for timer1\n\tpersistenceMutableState.TimerInfos[timerID1] = &persistence.TimerInfo{\n\t\tVersion:    startEvent1.Version,\n\t\tTimerID:    timerID1,\n\t\tExpiryTime: time.Unix(0, startEvent1.GetTimestamp()).Add(timerTimeout1),\n\t\tStartedID:  startEvent1.ID,\n\t\tTaskStatus: execution.TimerTaskStatusNone,\n\t}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tnextPageToken := []byte{1, 2, 3}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.ReadHistoryBranchRequest) bool {\n\t\treturn req.MinEventID == 1 && req.NextPageToken == nil\n\t})).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{startEvent1, startEvent2},\n\t\tNextPageToken: nextPageToken,\n\t}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.ReadHistoryBranchRequest) bool {\n\t\treturn req.MinEventID == 1 && bytes.Equal(req.NextPageToken, nextPageToken)\n\t})).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{firedEvent1},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\t// only timer2 should be fired\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.MatchedBy(func(req *persistence.AppendHistoryNodesRequest) bool {\n\t\tnumTimerFiredEvents := 0\n\t\tfor _, event := range req.Events {\n\t\t\tif event.GetEventType() == types.EventTypeTimerFired {\n\t\t\t\tnumTimerFiredEvents++\n\t\t\t}\n\t\t}\n\t\treturn numTimerFiredEvents == 1\n\t})).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t// both timerInfo should be deleted\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn len(req.UpdateWorkflowMutation.DeleteTimerInfos) == 2\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timerActiveTaskExecutor.config.ResurrectionCheckMinDelay = dynamicproperties.GetDurationPropertyFnFilteredByDomain(timerTimeout2 - timerTimeout1)\n\ts.timeSource.Advance(timerTimeout2)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_NoRetryPolicy_Fire() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\t_, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetActivityInfo(scheduledEvent.ID)\n\ts.False(ok)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_NoRetryPolicy_Noop() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tidentity := \"identity\"\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, identity)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tcompleteEvent := test.AddActivityTaskCompletedEvent(mutableState, scheduledEvent.ID, startedEvent.ID, []byte(nil), identity)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, completeEvent.ID, completeEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_RetryPolicy_Retry_StartToClose() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, \"identity\")\n\ts.Nil(startedEvent)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tactivityInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetActivityInfo(scheduledEvent.ID)\n\ts.True(ok)\n\ts.Equal(scheduledEvent.ID, activityInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, activityInfo.StartedID)\n\t// only a schedule to start timer will be created, apart from the retry timer\n\ts.Equal(int32(execution.TimerTaskStatusCreatedScheduleToStart), activityInfo.TimerTaskStatus)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_RetryPolicy_Retry_ScheduleToStart() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tactivityInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetActivityInfo(scheduledEvent.ID)\n\ts.True(ok)\n\ts.Equal(scheduledEvent.ID, activityInfo.ScheduleID)\n\ts.Equal(commonconstants.EmptyEventID, activityInfo.StartedID)\n\t// only a schedule to start timer will be created, apart from the retry timer\n\ts.Equal(int32(execution.TimerTaskStatusCreatedScheduleToStart), activityInfo.TimerTaskStatus)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_RetryPolicy_Noop() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tidentity := \"identity\"\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, identity)\n\ts.Nil(startedEvent)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tcompleteEvent := test.AddActivityTaskCompletedEvent(mutableState, scheduledEvent.ID, commonconstants.TransientEventID, []byte(nil), identity)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, completeEvent.ID, completeEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.timeSource.Advance(2 * timerTimeout)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_Heartbeat_Noop() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tidentity := \"identity\"\n\ttimerTimeout := 2 * time.Second\n\theartbeatTimerTimeout := time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(heartbeatTimerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, identity)\n\ts.Nil(startedEvent)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ts.Equal(int(execution.TimerTypeHeartbeat), task.(*persistence.ActivityTimeoutTask).TimeoutType)\n\ttask.SetVisibilityTimestamp(task.GetVisibilityTimestamp().Add(-time.Second))\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestProcessActivityTimeout_Resurrected() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tidentity := \"identity\"\n\ttimerTimeout1 := 2 * time.Second\n\ttimerTimeout2 := 5 * time.Second\n\t// schedule 2 activities\n\tscheduledEvent1, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity1\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout1.Seconds()),\n\t\tint32(timerTimeout1.Seconds()),\n\t\tint32(timerTimeout1.Seconds()),\n\t\tint32(timerTimeout1.Seconds()),\n\t)\n\tscheduledEvent2, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity2\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout2.Seconds()),\n\t\tint32(timerTimeout2.Seconds()),\n\t\tint32(timerTimeout2.Seconds()),\n\t\tint32(timerTimeout2.Seconds()),\n\t)\n\n\tstartedEvent1 := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent1.ID, identity)\n\tcompleteEvent1 := test.AddActivityTaskCompletedEvent(mutableState, scheduledEvent1.ID, startedEvent1.ID, []byte(nil), identity)\n\tmutableState.FlushBufferedEvents()\n\t// there should be a decision scheduled event after activity1 is completed\n\t// omitted here to make the test easier to read\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, completeEvent1.ID, completeEvent1.Version)\n\ts.NoError(err)\n\t// add resurrected activity info for activity1\n\tpersistenceMutableState.ActivityInfos[scheduledEvent1.ID] = &persistence.ActivityInfo{\n\t\tVersion:                  scheduledEvent1.Version,\n\t\tScheduleID:               scheduledEvent1.ID,\n\t\tScheduledEventBatchID:    scheduledEvent1.ID,\n\t\tScheduledTime:            time.Unix(0, scheduledEvent1.GetTimestamp()),\n\t\tStartedID:                commonconstants.EmptyEventID,\n\t\tStartedTime:              time.Time{},\n\t\tActivityID:               \"activity1\",\n\t\tDomainID:                 s.domainID,\n\t\tScheduleToStartTimeout:   int32(timerTimeout1.Seconds()),\n\t\tScheduleToCloseTimeout:   int32(timerTimeout1.Seconds()),\n\t\tStartToCloseTimeout:      int32(timerTimeout1.Seconds()),\n\t\tHeartbeatTimeout:         int32(timerTimeout1.Seconds()),\n\t\tCancelRequested:          false,\n\t\tCancelRequestID:          commonconstants.EmptyEventID,\n\t\tLastHeartBeatUpdatedTime: time.Time{},\n\t\tTimerTaskStatus:          execution.TimerTaskStatusNone,\n\t\tTaskList:                 mutableState.GetExecutionInfo().TaskList,\n\t}\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tnextPageToken := []byte{1, 2, 3}\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.ReadHistoryBranchRequest) bool {\n\t\treturn req.MinEventID == 1 && req.NextPageToken == nil\n\t})).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{scheduledEvent1, scheduledEvent2, startedEvent1, completeEvent1},\n\t\tNextPageToken: nextPageToken,\n\t}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"ReadHistoryBranch\", mock.Anything, mock.MatchedBy(func(req *persistence.ReadHistoryBranchRequest) bool {\n\t\treturn req.MinEventID == 1 && bytes.Equal(req.NextPageToken, nextPageToken)\n\t})).Return(&persistence.ReadHistoryBranchResponse{\n\t\tHistoryEvents: []*types.HistoryEvent{scheduledEvent1, scheduledEvent2, startedEvent1, completeEvent1},\n\t\tNextPageToken: nil,\n\t}, nil).Once()\n\t// only activity timer for activity2 should be fired\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.MatchedBy(func(req *persistence.AppendHistoryNodesRequest) bool {\n\t\tnumActivityTimeoutEvents := 0\n\t\tfor _, event := range req.Events {\n\t\t\tif event.GetEventType() == types.EventTypeActivityTaskTimedOut {\n\t\t\t\tnumActivityTimeoutEvents++\n\t\t\t}\n\t\t}\n\t\treturn numActivityTimeoutEvents == 1\n\t})).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t// both activityInfo should be deleted\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn len(req.UpdateWorkflowMutation.DeleteActivityInfos) == 2\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.timerActiveTaskExecutor.config.ResurrectionCheckMinDelay = dynamicproperties.GetDurationPropertyFnFilteredByDomain(timerTimeout2 - timerTimeout1)\n\ts.timeSource.Advance(timerTimeout2)\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionScheduleToStartTimeout_NormalDecision() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeScheduleToStart),\n\t\tEventID:     di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.DecisionAttempt == 1 &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.DecisionScheduleID == 4 &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.NextEventID == 4 && // transient decision\n\t\t\tlen(req.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]) == 1 // another schedule to start timer\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionScheduleToStartTimeout_TransientDecision() {\n\ts.mockShard.GetConfig().NormalDecisionScheduleToStartMaxAttempts = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdecisionAttempt := int64(1)\n\tmutableState.GetExecutionInfo().DecisionAttempt = decisionAttempt // fake a transient decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType:     int(types.TimeoutTypeScheduleToStart),\n\t\tEventID:         di.ScheduleID,\n\t\tScheduleAttempt: decisionAttempt,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.DecisionAttempt == 2 &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.DecisionScheduleID == 2 &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.NextEventID == 2 && // transient decision\n\t\t\tlen(req.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]) == 0 // since the max attempt is 1 at the beginning of the test\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionScheduleToStartTimeout_StickyDecision() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.StickyTaskList = \"sticky-tasklist\"\n\texecutionInfo.StickyScheduleToStartTimeout = 1\n\texecutionInfo.LastUpdatedTimestamp = s.timeSource.Now()\n\n\t// schedule a second (sticky) decision task\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeScheduleToStart),\n\t\tEventID:     di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\texecutionInfo := req.UpdateWorkflowMutation.ExecutionInfo\n\t\treturn executionInfo.DecisionAttempt == 0 && // attempt is not increased when convert to normal decision\n\t\t\texecutionInfo.DecisionScheduleID == 7 &&\n\t\t\texecutionInfo.NextEventID == 8 && // normal decision\n\t\t\texecutionInfo.StickyTaskList == \"\" && // stickyness should be cleared\n\t\t\texecutionInfo.StickyScheduleToStartTimeout == 0 && // stickyness should be cleared\n\t\t\tlen(req.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]) == 1 // schedule to start timer\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionStartToCloseTimeout_Fire() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\tEventID:     di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tdecisionInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetPendingDecision()\n\ts.True(ok)\n\ts.True(decisionInfo.ScheduleID != commonconstants.EmptyEventID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(int64(1), decisionInfo.Attempt)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionStartToCloseTimeout_ExceedsMaxAttempts() {\n\ts.mockShard.GetConfig().DecisionRetryMaxAttempts = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\ts.mockShard.GetConfig().EnforceDecisionTaskAttempts = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\n\tdecisionAttempt := int64(1)\n\tmutableState.GetExecutionInfo().DecisionAttempt = decisionAttempt\n\tdi.Attempt = decisionAttempt\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType:     int(types.TimeoutTypeStartToClose),\n\t\tEventID:         di.ScheduleID,\n\t\tScheduleAttempt: decisionAttempt,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.State == persistence.WorkflowStateCompleted &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.CloseStatus == persistence.WorkflowCloseStatusTerminated\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tms := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID())\n\ts.False(ms.IsWorkflowExecutionRunning())\n\t_, closeStatus := ms.GetWorkflowStateCloseStatus()\n\ts.Equal(closeStatus, persistence.WorkflowCloseStatusTerminated)\n\tcompletionEvent, err := ms.GetCompletionEvent(context.Background())\n\ts.NoError(err)\n\ts.Equal(completionEvent.WorkflowExecutionTerminatedEventAttributes.Reason, common.FailureReasonDecisionAttemptsExceedsLimit)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionStartToCloseTimeout_BelowMaxAttempts() {\n\ts.mockShard.GetConfig().DecisionRetryMaxAttempts = dynamicproperties.GetIntPropertyFilteredByDomain(5)\n\ts.mockShard.GetConfig().EnforceDecisionTaskAttempts = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\n\tdecisionAttempt := int64(1)\n\tmutableState.GetExecutionInfo().DecisionAttempt = decisionAttempt\n\tdi.Attempt = decisionAttempt\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType:     int(types.TimeoutTypeStartToClose),\n\t\tEventID:         di.ScheduleID,\n\t\tScheduleAttempt: decisionAttempt,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.State == persistence.WorkflowStateRunning &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.DecisionAttempt == 2\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tdecisionInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetPendingDecision()\n\ts.True(ok)\n\ts.Equal(int64(2), decisionInfo.Attempt)\n\ts.True(s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).IsWorkflowExecutionRunning())\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionStartToCloseTimeout_AttemptsNotEnforced() {\n\ts.mockShard.GetConfig().DecisionRetryMaxAttempts = dynamicproperties.GetIntPropertyFilteredByDomain(1)\n\ts.mockShard.GetConfig().EnforceDecisionTaskAttempts = dynamicproperties.GetBoolPropertyFnFilteredByDomain(false)\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\n\tdecisionAttempt := int64(1)\n\tmutableState.GetExecutionInfo().DecisionAttempt = decisionAttempt\n\tdi.Attempt = decisionAttempt\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType:     int(types.TimeoutTypeStartToClose),\n\t\tEventID:         di.ScheduleID,\n\t\tScheduleAttempt: decisionAttempt,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(req *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\treturn req.UpdateWorkflowMutation.ExecutionInfo.State == persistence.WorkflowStateRunning &&\n\t\t\treq.UpdateWorkflowMutation.ExecutionInfo.DecisionAttempt == 2\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tdecisionInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetPendingDecision()\n\ts.True(ok)\n\ts.Equal(int64(2), decisionInfo.Attempt)\n\ts.True(s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).IsWorkflowExecutionRunning())\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestDecisionStartToCloseTimeout_Noop() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\tEventID:     di.ScheduleID - 1, // no corresponding decision for this scheduleID\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestWorkflowBackoffTimer_Fire() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowBackoffTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: persistence.WorkflowBackoffTimeoutTypeRetry,\n\t})\n\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tdecisionInfo, ok := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetPendingDecision()\n\ts.True(ok)\n\ts.True(decisionInfo.ScheduleID != commonconstants.EmptyEventID)\n\ts.Equal(commonconstants.EmptyEventID, decisionInfo.StartedID)\n\ts.Equal(int64(0), decisionInfo.Attempt)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestWorkflowBackoffTimer_Noop() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowBackoffTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: persistence.WorkflowBackoffTimeoutTypeRetry,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestActivityRetryTimer_Fire() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, activityInfo := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\tactivityInfo.Attempt = 1\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.ActivityRetryTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tEventID: activityInfo.ScheduleID,\n\t\tAttempt: int64(activityInfo.Attempt),\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(\n\t\tgomock.Any(),\n\t\t&types.AddActivityTaskRequest{\n\t\t\tDomainUUID:       activityInfo.DomainID,\n\t\t\tSourceDomainUUID: activityInfo.DomainID,\n\t\t\tExecution:        &workflowExecution,\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: activityInfo.TaskList,\n\t\t\t},\n\t\t\tScheduleID:                    activityInfo.ScheduleID,\n\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(activityInfo.ScheduleToStartTimeout),\n\t\t\tPartitionConfig:               mutableState.GetExecutionInfo().PartitionConfig,\n\t\t},\n\t).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestActivityRetryTimer_Noop() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, activityInfo := test.AddActivityTaskScheduledEventWithRetry(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\t&types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tBackoffCoefficient:          1.2,\n\t\t\tMaximumIntervalInSeconds:    5,\n\t\t\tMaximumAttempts:             5,\n\t\t\tNonRetriableErrorReasons:    []string{\"（╯' - ')╯ ┻━┻ \"},\n\t\t\tExpirationIntervalInSeconds: 999,\n\t\t},\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, \"identity\")\n\ts.Nil(startedEvent)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.ActivityRetryTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tEventID: activityInfo.ScheduleID,\n\t\tAttempt: int64(activityInfo.Attempt),\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestWorkflowTimeout_Fire() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\trunning := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).IsWorkflowExecutionRunning()\n\ts.False(running)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestWorkflowTimeout_ContinueAsNew_Retry() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// need to override the workflow retry policy\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.HasRetryPolicy = true\n\texecutionInfo.ExpirationTime = s.timeSource.Now().Add(1000 * time.Second)\n\texecutionInfo.MaximumAttempts = 10\n\texecutionInfo.InitialInterval = 1\n\texecutionInfo.MaximumInterval = 1\n\texecutionInfo.BackoffCoefficient = 1\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t// one for current workflow, one for new\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Times(2)\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), s.domainID, gomock.Any()).Return(&types.ActiveClusterInfo{}, nil).AnyTimes()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tstate, closeStatus := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetWorkflowStateCloseStatus()\n\ts.Equal(persistence.WorkflowStateCompleted, state)\n\ts.Equal(persistence.WorkflowCloseStatusContinuedAsNew, closeStatus)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestWorkflowTimeout_ContinueAsNew_Cron() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.StartTimestamp = s.timeSource.Now()\n\texecutionInfo.CronSchedule = \"* * * * *\"\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t// one for current workflow, one for new\n\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Times(2)\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByClusterAttribute(gomock.Any(), s.domainID, gomock.Any()).Return(&types.ActiveClusterInfo{}, nil).AnyTimes()\n\n\t_, err = s.timerActiveTaskExecutor.Execute(timerTask)\n\ts.NoError(err)\n\n\tstate, closeStatus := s.getMutableStateFromCache(s.domainID, workflowExecution.GetWorkflowID(), workflowExecution.GetRunID()).GetWorkflowStateCloseStatus()\n\ts.Equal(persistence.WorkflowStateCompleted, state)\n\ts.Equal(persistence.WorkflowCloseStatusContinuedAsNew, closeStatus)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) getMutableStateFromCache(\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) execution.MutableState {\n\n\treturn s.executionCache.Get(\n\t\tdefinition.NewWorkflowIdentifier(domainID, workflowID, runID),\n\t).(execution.Context).GetWorkflowExecution()\n}\n\nfunc (s *timerActiveTaskExecutorSuite) newTimerTaskFromInfo(\n\ttask persistence.Task,\n) Task {\n\treturn NewHistoryTask(s.mockShard, task, QueueTypeActiveTimer, s.logger, nil, nil, nil, nil, nil)\n}\n\nfunc (s *timerActiveTaskExecutorSuite) TestActiveTaskTimeout() {\n\tdeleteHistoryEventTask := s.newTimerTaskFromInfo(&persistence.DeleteHistoryEventTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      \"runID\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(100),\n\t\t},\n\t})\n\ts.timerActiveTaskExecutor.Execute(deleteHistoryEventTask)\n}\n"
  },
  {
    "path": "service/history/task/timer_standby_task_executor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/simulation\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nvar (\n\terrUnexpectedTask   = errors.New(\"unexpected task\")\n\terrUnknownTimerTask = errors.New(\"unknown timer task\")\n)\n\ntype (\n\ttimerStandbyTaskExecutor struct {\n\t\t*timerTaskExecutorBase\n\n\t\tclusterName            string\n\t\thistoryResender        ndc.HistoryResender\n\t\tgetRemoteClusterNameFn func(context.Context, persistence.Task) (string, error)\n\t}\n)\n\n// NewTimerStandbyTaskExecutor creates a new task executor for standby timer task\nfunc NewTimerStandbyTaskExecutor(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\thistoryResender ndc.HistoryResender,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tclusterName string,\n\tconfig *config.Config,\n) Executor {\n\treturn &timerStandbyTaskExecutor{\n\t\ttimerTaskExecutorBase: newTimerTaskExecutorBase(\n\t\t\tshard,\n\t\t\tarchiverClient,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t\tmetricsClient,\n\t\t\tconfig,\n\t\t),\n\t\tclusterName:     clusterName,\n\t\thistoryResender: historyResender,\n\t\tgetRemoteClusterNameFn: func(ctx context.Context, taskInfo persistence.Task) (string, error) {\n\t\t\tif shard.GetConfig().EnableTimerQueueV2(shard.GetShardID()) {\n\t\t\t\treturn getRemoteClusterName(ctx, shard.GetClusterMetadata().GetCurrentClusterName(), shard.GetActiveClusterManager(), taskInfo)\n\t\t\t}\n\t\t\treturn clusterName, nil\n\t\t},\n\t}\n}\n\nfunc (t *timerStandbyTaskExecutor) Execute(task Task) (ExecuteResponse, error) {\n\tsimulation.LogEvents(simulation.E{\n\t\tEventName:  simulation.EventNameExecuteHistoryTask,\n\t\tHost:       t.shard.GetConfig().HostName,\n\t\tShardID:    t.shard.GetShardID(),\n\t\tDomainID:   task.GetDomainID(),\n\t\tWorkflowID: task.GetWorkflowID(),\n\t\tRunID:      task.GetRunID(),\n\t\tPayload: map[string]any{\n\t\t\t\"task_category\": persistence.HistoryTaskCategoryTimer.Name(),\n\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t},\n\t})\n\tscope := getOrCreateDomainTaggedScope(t.shard, GetTimerTaskMetricScope(task.GetTaskType(), false), task.GetDomainID(), t.logger)\n\texecuteResponse := ExecuteResponse{\n\t\tScope:        scope,\n\t\tIsActiveTask: false,\n\t}\n\tswitch timerTask := task.GetInfo().(type) {\n\tcase *persistence.UserTimerTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeUserTimerTimeoutTask(ctx, timerTask)\n\tcase *persistence.ActivityTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeActivityTimeoutTask(ctx, timerTask)\n\tcase *persistence.DecisionTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeDecisionTimeoutTask(ctx, timerTask)\n\tcase *persistence.WorkflowTimeoutTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeWorkflowTimeoutTask(ctx, timerTask)\n\tcase *persistence.ActivityRetryTimerTask:\n\t\t// retry backoff timer should not get created on passive cluster\n\t\t// TODO: add error logs\n\t\treturn executeResponse, nil\n\tcase *persistence.WorkflowBackoffTimerTask:\n\t\tctx, cancel := context.WithTimeout(t.ctx, taskDefaultTimeout)\n\t\tdefer cancel()\n\t\treturn executeResponse, t.executeWorkflowBackoffTimerTask(ctx, timerTask)\n\tcase *persistence.DeleteHistoryEventTask:\n\t\t// special timeout for delete history event\n\t\tdeleteHistoryEventContext, deleteHistoryEventCancel := context.WithTimeout(t.ctx, time.Duration(t.config.DeleteHistoryEventContextTimeout())*time.Second)\n\t\tdefer deleteHistoryEventCancel()\n\t\treturn executeResponse, t.executeDeleteHistoryEventTask(deleteHistoryEventContext, timerTask)\n\tdefault:\n\t\treturn executeResponse, errUnknownTimerTask\n\t}\n}\n\nfunc (t *timerStandbyTaskExecutor) executeUserTimerTimeoutTask(\n\tctx context.Context,\n\ttimerTask *persistence.UserTimerTask,\n) error {\n\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\ttimerSequence := execution.NewTimerSequence(mutableState)\n\n\tLoop:\n\t\tfor _, timerSequenceID := range timerSequence.LoadAndSortUserTimers() {\n\t\t\t_, ok := mutableState.GetUserTimerInfoByEventID(timerSequenceID.EventID)\n\t\t\tif !ok {\n\t\t\t\terrString := fmt.Sprintf(\"failed to find in user timer event ID: %v\", timerSequenceID.EventID)\n\t\t\t\tt.logger.Error(errString)\n\t\t\t\treturn nil, &types.InternalServiceError{Message: errString}\n\t\t\t}\n\n\t\t\tif _, isExpired := timerSequence.IsExpired(\n\t\t\t\ttimerTask.VisibilityTimestamp,\n\t\t\t\ttimerSequenceID,\n\t\t\t); isExpired {\n\t\t\t\treturn getHistoryResendInfo(mutableState)\n\t\t\t}\n\t\t\t// since the user timer are already sorted, so if there is one timer which will not expired\n\t\t\t// all user timer after this timer will not expired\n\t\t\tbreak Loop //nolint:staticcheck\n\t\t}\n\t\t// if there is no user timer expired, then we are good\n\t\treturn nil, nil\n\t}\n\n\treturn t.processTimer(\n\t\tctx,\n\t\ttimerTask,\n\t\ttimerTask.EventID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttimerTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *timerStandbyTaskExecutor) executeActivityTimeoutTask(\n\tctx context.Context,\n\ttimerTask *persistence.ActivityTimeoutTask,\n) error {\n\n\t// activity heartbeat timer task is a special snowflake.\n\t// normal activity timer task on the passive side will be generated by events related to activity in history replicator,\n\t// and the standby timer processor will only need to verify whether the timer task can be safely throw away.\n\t//\n\t// activity heartbeat timer task cannot be handled in the way mentioned above.\n\t// the reason is, there is no event driving the creation of new activity heartbeat timer.\n\t// although there will be an task syncing activity from remote, the task is not an event,\n\t// and cannot attempt to recreate a new activity timer task.\n\t//\n\t// the overall solution is to attempt to generate a new activity timer task whenever the\n\t// task passed in is safe to be throw away.\n\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\ttimerSequence := execution.NewTimerSequence(mutableState)\n\t\tupdateMutableState := false\n\n\tLoop:\n\t\tfor _, timerSequenceID := range timerSequence.LoadAndSortActivityTimers() {\n\t\t\t_, ok := mutableState.GetActivityInfo(timerSequenceID.EventID)\n\t\t\tif !ok {\n\t\t\t\terrString := fmt.Sprintf(\"failed to find in memory activity timer: %v\", timerSequenceID.EventID)\n\t\t\t\tt.logger.Error(errString)\n\t\t\t\treturn nil, &types.InternalServiceError{Message: errString}\n\t\t\t}\n\n\t\t\tif _, isExpired := timerSequence.IsExpired(\n\t\t\t\ttimerTask.VisibilityTimestamp,\n\t\t\t\ttimerSequenceID,\n\t\t\t); isExpired {\n\t\t\t\treturn getHistoryResendInfo(mutableState)\n\t\t\t}\n\t\t\t// since the activity timer are already sorted, so if there is one timer which will not expired\n\t\t\t// all activity timer after this timer will not expired\n\t\t\tbreak Loop //nolint:staticcheck\n\t\t}\n\n\t\t// for reason to update mutable state & generate a new activity task,\n\t\t// see comments at the beginning of this function.\n\t\t// NOTE: this is the only place in the standby logic where mutable state can be updated\n\n\t\t// need to clear the activity heartbeat timer task marks\n\t\tlastWriteVersion, err := mutableState.GetLastWriteVersion()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\t// NOTE: LastHeartbeatTimeoutVisibilityInSeconds is for deduping heartbeat timer creation as it's possible\n\t\t// one heartbeat task was persisted multiple times with different taskIDs due to the retry logic\n\t\t// for updating workflow execution. In that case, only one new heartbeat timeout task should be\n\t\t// created.\n\t\tisHeartBeatTask := timerTask.TimeoutType == int(types.TimeoutTypeHeartbeat)\n\t\tactivityInfo, ok := mutableState.GetActivityInfo(timerTask.EventID)\n\t\tif isHeartBeatTask && ok && activityInfo.LastHeartbeatTimeoutVisibilityInSeconds <= timerTask.VisibilityTimestamp.Unix() {\n\t\t\tactivityInfo.TimerTaskStatus = activityInfo.TimerTaskStatus &^ execution.TimerTaskStatusCreatedHeartbeat\n\t\t\tif err := mutableState.UpdateActivity(activityInfo); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tupdateMutableState = true\n\t\t}\n\n\t\t// passive logic need to explicitly call create timer\n\t\tmodified, err := timerSequence.CreateNextActivityTimer()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tupdateMutableState = updateMutableState || modified\n\n\t\tif !updateMutableState {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tnow := t.getStandbyClusterTime()\n\t\t// we need to handcraft some of the variables\n\t\t// since the job being done here is update the activity and possibly write a timer task to DB\n\t\t// also need to reset the current version.\n\t\tt.logger.Debugf(\"executeActivityTimeoutTask calling UpdateCurrentVersion for domain %s, wfID %v, lastWriteVersion %v\",\n\t\t\ttimerTask.DomainID, timerTask.WorkflowID, lastWriteVersion)\n\t\tif err := mutableState.UpdateCurrentVersion(lastWriteVersion, true); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\terr = wfContext.UpdateWorkflowExecutionAsPassive(ctx, now)\n\t\treturn nil, err\n\t}\n\n\treturn t.processTimer(\n\t\tctx,\n\t\ttimerTask,\n\t\ttimerTask.EventID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttimerTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *timerStandbyTaskExecutor) executeDecisionTimeoutTask(\n\tctx context.Context,\n\ttimerTask *persistence.DecisionTimeoutTask,\n) error {\n\n\t// decision schedule to start timer won't be generated for sticky decision,\n\t// since sticky is cleared when applying events on passive.\n\t// for normal decision, we don't know if a schedule to start timeout timer\n\t// is generated or not since it's based on a dynamicconfig. On passive cluster,\n\t// a timer task will be generated based on passive cluster's config, however, it\n\t// may not match the active cluster.\n\t// so we simply ignore the schedule to start timer here as the decision task will be\n\t// pushed to matching without any timeout if's not started, and the workflow\n\t// can continue execution after failover.\n\tif timerTask.TimeoutType == int(types.TimeoutTypeScheduleToStart) {\n\t\treturn nil\n\t}\n\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tdecision, isPending := mutableState.GetDecisionInfo(timerTask.EventID)\n\t\tif !isPending {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, timerTask.DomainID, decision.Version, timerTask.Version, timerTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTimer(\n\t\tctx,\n\t\ttimerTask,\n\t\ttimerTask.EventID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttimerTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *timerStandbyTaskExecutor) executeWorkflowBackoffTimerTask(\n\tctx context.Context,\n\ttimerTask *persistence.WorkflowBackoffTimerTask,\n) error {\n\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tif mutableState.HasProcessedOrPendingDecision() {\n\t\t\t// if there is one decision already been processed\n\t\t\t// or has pending decision, meaning workflow has already running\n\t\t\treturn nil, nil\n\t\t}\n\n\t\t// Note: do not need to verify task version here\n\t\t// logic can only go here if mutable state build's next event ID is 2\n\t\t// meaning history only contains workflow started event.\n\t\t// we can do the checking of task version vs workflow started version\n\t\t// however, workflow started version is immutable\n\n\t\t// active cluster will add first decision task after backoff timeout.\n\t\t// standby cluster should just call ack manager to retry this task\n\t\t// since we are stilling waiting for the first DecisionScheduledEvent to be replicated from active side.\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTimer(\n\t\tctx,\n\t\ttimerTask,\n\t\t0,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttimerTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *timerStandbyTaskExecutor) executeWorkflowTimeoutTask(\n\tctx context.Context,\n\ttimerTask *persistence.WorkflowTimeoutTask,\n) error {\n\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\t// we do not need to notify new timer to base, since if there is no new event being replicated\n\t\t// checking again if the timer can be completed is meaningless\n\n\t\tstartVersion, err := mutableState.GetStartVersion()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, timerTask.DomainID, startVersion, timerTask.Version, timerTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTimer(\n\t\tctx,\n\t\ttimerTask,\n\t\t0,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttimerTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *timerStandbyTaskExecutor) getStandbyClusterTime() time.Time {\n\t// time of remote cluster in the shard is delayed by \"StandbyClusterDelay\"\n\t// so to get the current accurate remote cluster time, need to add it back\n\treturn t.shard.GetCurrentTime(t.clusterName).Add(t.shard.GetConfig().StandbyClusterDelay())\n}\n\nfunc (t *timerStandbyTaskExecutor) processTimer(\n\tctx context.Context,\n\ttimerTask persistence.Task,\n\teventID int64,\n\tactionFn standbyActionFn,\n\tpostActionFn standbyPostActionFn,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttimerTask.GetDomainID(),\n\t\tgetWorkflowExecution(timerTask),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() {\n\t\tif isRedispatchErr(retError) {\n\t\t\trelease(nil)\n\t\t} else {\n\t\t\trelease(retError)\n\t\t}\n\t}()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, timerTask, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, eventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil {\n\t\treturn nil\n\t}\n\n\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t// TODO: Check if workflow timeout timer comes to this point and then discarded.\n\t\t// workflow already finished, no need to process the timer\n\t\treturn nil\n\t}\n\n\thistoryResendInfo, err := actionFn(ctx, wfContext, mutableState)\n\tif err != nil {\n\t\tif t.logger.DebugOn() {\n\t\t\tt.logger.Debug(\"processTimer got error from actionFn\",\n\t\t\t\ttag.Error(err),\n\t\t\t\ttag.WorkflowID(timerTask.GetWorkflowID()),\n\t\t\t\ttag.WorkflowRunID(timerTask.GetRunID()),\n\t\t\t\ttag.WorkflowDomainID(timerTask.GetDomainID()),\n\t\t\t\ttag.TaskID(timerTask.GetTaskID()),\n\t\t\t\ttag.TaskType(int(timerTask.GetTaskType())),\n\t\t\t\ttag.Timestamp(timerTask.GetVisibilityTimestamp()),\n\t\t\t)\n\t\t}\n\t\treturn err\n\t}\n\n\tif t.logger.DebugOn() {\n\t\tt.logger.Debug(\"processTimer got historyResendInfo from actionFn\",\n\t\t\ttag.WorkflowID(timerTask.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(timerTask.GetRunID()),\n\t\t\ttag.WorkflowDomainID(timerTask.GetDomainID()),\n\t\t\ttag.TaskID(timerTask.GetTaskID()),\n\t\t\ttag.TaskType(int(timerTask.GetTaskType())),\n\t\t\ttag.Timestamp(timerTask.GetVisibilityTimestamp()),\n\t\t)\n\t}\n\n\trelease(nil)\n\treturn postActionFn(ctx, timerTask, historyResendInfo, t.logger)\n}\n\nfunc (t *timerStandbyTaskExecutor) fetchHistoryFromRemote(\n\tctx context.Context,\n\ttaskInfo persistence.Task,\n\tpostActionInfo interface{},\n\t_ log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\tresendInfo := postActionInfo.(*historyResendInfo)\n\n\tt.metricsClient.IncCounter(metrics.HistoryRereplicationByTimerTaskScope, metrics.CadenceClientRequests)\n\tstopwatch := t.metricsClient.StartTimer(metrics.HistoryRereplicationByTimerTaskScope, metrics.CadenceClientLatency)\n\tdefer stopwatch.Stop()\n\n\tremoteClusterName, err := t.getRemoteClusterNameFn(ctx, taskInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resendInfo.lastEventID != nil && resendInfo.lastEventVersion != nil {\n\t\t// note history resender doesn't take in a context parameter, there's a separate dynamicconfig for\n\t\t// controlling the timeout for resending history.\n\t\terr = t.historyResender.SendSingleWorkflowHistory(\n\t\t\tremoteClusterName,\n\t\t\ttaskInfo.GetDomainID(),\n\t\t\ttaskInfo.GetWorkflowID(),\n\t\t\ttaskInfo.GetRunID(),\n\t\t\tresendInfo.lastEventID,\n\t\t\tresendInfo.lastEventVersion,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\t} else {\n\t\terr = &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"incomplete historyResendInfo: %v\", resendInfo),\n\t\t}\n\t}\n\n\tif err != nil {\n\t\tt.logger.Error(\"Error re-replicating history from remote.\",\n\t\t\ttag.ShardID(t.shard.GetShardID()),\n\t\t\ttag.WorkflowDomainID(taskInfo.GetDomainID()),\n\t\t\ttag.WorkflowID(taskInfo.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(taskInfo.GetRunID()),\n\t\t\ttag.SourceCluster(remoteClusterName),\n\t\t\ttag.Error(err),\n\t\t)\n\t} else if t.logger.DebugOn() {\n\t\tt.logger.Debug(\"Successfully re-replicated history from remote.\",\n\t\t\ttag.WorkflowID(taskInfo.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(taskInfo.GetRunID()),\n\t\t\ttag.WorkflowDomainID(taskInfo.GetDomainID()),\n\t\t\ttag.TaskID(taskInfo.GetTaskID()),\n\t\t\ttag.TaskType(int(taskInfo.GetTaskType())),\n\t\t\ttag.SourceCluster(remoteClusterName),\n\t\t)\n\t}\n\n\t// return error so task processing logic will retry\n\treturn &redispatchError{Reason: \"fetchHistoryFromRemote\"}\n}\n\nfunc (t *timerStandbyTaskExecutor) getCurrentTime(taskInfo persistence.Task) (time.Time, error) {\n\t// the schedule time of a standby task is the time from the active cluster of the task,\n\t// for history queue v2, t.clusterName is the current cluster, so we should get remote cluster name of the task\n\t// However, this only has an effect when the time skew between the current cluster and the remote cluster is large enough (probably more than 1 minute),\n\t// the impact is that the standby task may be discarded too early\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tdefer cancel()\n\tsourceClusterName, err := t.getRemoteClusterNameFn(ctx, taskInfo)\n\tif err != nil {\n\t\treturn time.Time{}, err\n\t}\n\treturn t.shard.GetCurrentTime(sourceClusterName), nil\n}\n"
  },
  {
    "path": "service/history/task/timer_standby_task_executor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n)\n\ntype (\n\ttimerStandbyTaskExecutorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller             *gomock.Controller\n\t\tmockShard              *shard.TestContext\n\t\tmockEngine             *engine.MockEngine\n\t\tmockDomainCache        *cache.MockDomainCache\n\t\tmockNDCHistoryResender *ndc.MockHistoryResender\n\n\t\tmockExecutionMgr *mocks.ExecutionManager\n\n\t\tlogger               log.Logger\n\t\tdomainID             string\n\t\tdomainEntry          *cache.DomainCacheEntry\n\t\tversion              int64\n\t\tclusterName          string\n\t\ttimeSource           clock.MockedTimeSource\n\t\tfetchHistoryDuration time.Duration\n\t\tdiscardDuration      time.Duration\n\n\t\ttimerStandbyTaskExecutor *timerStandbyTaskExecutor\n\t}\n)\n\nfunc TestTimerStandbyTaskExecutorSuite(t *testing.T) {\n\ts := new(timerStandbyTaskExecutorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) SetupSuite() {\n\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\tconfig := config.NewForTest()\n\ts.domainID = constants.TestDomainID\n\ts.domainEntry = constants.TestGlobalDomainEntry\n\ts.version = s.domainEntry.GetFailoverVersion()\n\ts.clusterName = cluster.TestAlternativeClusterName\n\ts.timeSource = clock.NewMockedTimeSource()\n\ts.fetchHistoryDuration = config.StandbyTaskMissingEventsResendDelay() +\n\t\t(config.StandbyTaskMissingEventsDiscardDelay()-config.StandbyTaskMissingEventsResendDelay())/2\n\ts.discardDuration = config.StandbyTaskMissingEventsDiscardDelay() * 2\n\n\ts.controller = gomock.NewController(s.T())\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\tconfig,\n\t)\n\ts.mockShard.SetEventsCache(events.NewCache(\n\t\ts.mockShard.GetShardID(),\n\t\ts.mockShard.GetHistoryManager(),\n\t\ts.mockShard.GetConfig(),\n\t\ts.mockShard.GetLogger(),\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.mockShard.GetDomainCache(),\n\t))\n\ts.mockShard.Resource.TimeSource = s.timeSource\n\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.mockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTransferTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTimerTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewReplicationTasks(gomock.Any()).AnyTimes()\n\ts.mockShard.SetEngine(s.mockEngine)\n\ts.mockNDCHistoryResender = ndc.NewMockHistoryResender(s.controller)\n\n\t// ack manager will use the domain information\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(constants.TestDomainName, nil).AnyTimes()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestGlobalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestGlobalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\ts.timerStandbyTaskExecutor = NewTimerStandbyTaskExecutor(\n\t\ts.mockShard,\n\t\tnil,\n\t\texecution.NewCache(s.mockShard),\n\t\ts.mockNDCHistoryResender,\n\t\ts.logger,\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.clusterName,\n\t\tconfig,\n\t).(*timerStandbyTaskExecutor)\n\ts.timerStandbyTaskExecutor.getRemoteClusterNameFn = func(ctx context.Context, taskInfo persistence.Task) (string, error) {\n\t\treturn s.clusterName, nil\n\t}\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessUserTimerTimeout_Pending() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerID := \"timer\"\n\ttimerTimeout := 2 * time.Second\n\tevent, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID, int64(timerTimeout.Seconds()))\n\tnextEventID := event.ID\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttimerTask.GetDomainID(),\n\t\ttimerTask.GetWorkflowID(),\n\t\ttimerTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.discardDuration))\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessUserTimerTimeout_Success() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerID := \"timer\"\n\ttimerTimeout := 2 * time.Second\n\tevent, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID, int64(timerTimeout.Seconds()))\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tevent = test.AddTimerFiredEvent(mutableState, timerID)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessUserTimerTimeout_Multiple() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerID1 := \"timer-1\"\n\ttimerTimeout1 := 2 * time.Second\n\tevent, _ := test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID1, int64(timerTimeout1.Seconds()))\n\n\ttimerID2 := \"timer-2\"\n\ttimerTimeout2 := 50 * time.Second\n\t_, _ = test.AddTimerStartedEvent(mutableState, decisionCompletionID, timerID2, int64(timerTimeout2.Seconds()))\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextUserTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tevent = test.AddTimerFiredEvent(mutableState, timerID1)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessActivityTimeout_Pending() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t)\n\tnextEventID := scheduledEvent.ID\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, scheduledEvent.ID, scheduledEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttimerTask.GetDomainID(),\n\t\ttimerTask.GetWorkflowID(),\n\t\ttimerTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.discardDuration))\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessActivityTimeout_Success() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tidentity := \"identity\"\n\ttimerTimeout := 2 * time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, identity)\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tcompleteEvent := test.AddActivityTaskCompletedEvent(mutableState, scheduledEvent.ID, startedEvent.ID, []byte(nil), identity)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, completeEvent.ID, completeEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessActivityTimeout_Heartbeat_Noop() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTimeout := 2 * time.Second\n\theartbeatTimerTimeout := time.Second\n\tscheduledEvent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity\",\n\t\t\"activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte(nil),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(timerTimeout.Seconds()),\n\t\tint32(heartbeatTimerTimeout.Seconds()),\n\t)\n\tstartedEvent := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent.ID, \"identity\")\n\tmutableState.FlushBufferedEvents()\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttask := mutableState.GetTimerTasks()[0]\n\ts.Equal(int(execution.TimerTypeHeartbeat), task.(*persistence.ActivityTimeoutTask).TimeoutType)\n\ttask.SetVisibilityTimestamp(task.GetVisibilityTimestamp().Add(-5 * time.Second))\n\ttimerTask := s.newTimerTaskFromInfo(task)\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessActivityTimeout_Multiple_CanUpdate() {\n\n\t_, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\tidentity := \"identity\"\n\ttasklist := \"tasklist\"\n\tactivityID1 := \"activity 1\"\n\tactivityType1 := \"activity type 1\"\n\ttimerTimeout1 := 2 * time.Second\n\tscheduledEvent1, _ := test.AddActivityTaskScheduledEvent(mutableState, decisionCompletionID, activityID1, activityType1, tasklist, []byte(nil),\n\t\tint32(timerTimeout1.Seconds()), int32(timerTimeout1.Seconds()), int32(timerTimeout1.Seconds()), int32(timerTimeout1.Seconds()))\n\tstartedEvent1 := test.AddActivityTaskStartedEvent(mutableState, scheduledEvent1.ID, identity)\n\n\tactivityID2 := \"activity 2\"\n\tactivityType2 := \"activity type 2\"\n\ttimerTimeout2 := 20 * time.Second\n\tscheduledEvent2, _ := test.AddActivityTaskScheduledEvent(mutableState, decisionCompletionID, activityID2, activityType2, tasklist, []byte(nil),\n\t\tint32(timerTimeout2.Seconds()), int32(timerTimeout2.Seconds()), int32(timerTimeout2.Seconds()), int32(timerTimeout2.Seconds()))\n\ttest.AddActivityTaskStartedEvent(mutableState, scheduledEvent2.ID, identity)\n\tactivityInfo2 := mutableState.GetPendingActivityInfos()[scheduledEvent2.ID]\n\tactivityInfo2.TimerTaskStatus |= execution.TimerTaskStatusCreatedHeartbeat\n\tactivityInfo2.LastHeartBeatUpdatedTime = time.Now()\n\n\ttimerSequence := execution.NewTimerSequence(mutableState)\n\tmutableState.DeleteTimerTasks()\n\tmodified, err := timerSequence.CreateNextActivityTimer()\n\ts.NoError(err)\n\ts.True(modified)\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.ActivityTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: mutableState.GetExecutionInfo().WorkflowID,\n\t\t\tRunID:      mutableState.GetExecutionInfo().RunID,\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: activityInfo2.LastHeartBeatUpdatedTime.Add(-5 * time.Second),\n\t\t},\n\t\tEventID: scheduledEvent2.ID,\n\t})\n\n\tcompleteEvent1 := test.AddActivityTaskCompletedEvent(mutableState, scheduledEvent1.ID, startedEvent1.ID, []byte(nil), identity)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, completeEvent1.ID, completeEvent1.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.MatchedBy(func(input *persistence.UpdateWorkflowExecutionRequest) bool {\n\t\ts.Equal(1, len(input.UpdateWorkflowMutation.TasksByCategory[persistence.HistoryTaskCategoryTimer]))\n\t\ts.Equal(1, len(input.UpdateWorkflowMutation.UpsertActivityInfos))\n\t\tmutableState.GetExecutionInfo().LastUpdatedTimestamp = input.UpdateWorkflowMutation.ExecutionInfo.LastUpdatedTimestamp\n\t\tinput.RangeID = 0\n\t\tinput.UpdateWorkflowMutation.ExecutionInfo.LastEventTaskID = 0\n\t\tmutableState.GetExecutionInfo().LastEventTaskID = 0\n\t\tmutableState.GetExecutionInfo().DecisionOriginalScheduledTimestamp = input.UpdateWorkflowMutation.ExecutionInfo.DecisionOriginalScheduledTimestamp\n\t\ts.Equal(&persistence.UpdateWorkflowExecutionRequest{\n\t\t\tUpdateWorkflowMutation: persistence.WorkflowMutation{\n\t\t\t\tExecutionInfo:             mutableState.GetExecutionInfo(),\n\t\t\t\tExecutionStats:            &persistence.ExecutionStats{},\n\t\t\t\tTasksByCategory:           input.UpdateWorkflowMutation.TasksByCategory,\n\t\t\t\tCondition:                 mutableState.GetNextEventID(),\n\t\t\t\tUpsertActivityInfos:       input.UpdateWorkflowMutation.UpsertActivityInfos,\n\t\t\t\tDeleteActivityInfos:       []int64{},\n\t\t\t\tUpsertTimerInfos:          []*persistence.TimerInfo{},\n\t\t\t\tDeleteTimerInfos:          []string{},\n\t\t\t\tUpsertChildExecutionInfos: []*persistence.ChildExecutionInfo{},\n\t\t\t\tDeleteChildExecutionInfos: []int64{},\n\t\t\t\tUpsertRequestCancelInfos:  []*persistence.RequestCancelInfo{},\n\t\t\t\tDeleteRequestCancelInfos:  []int64{},\n\t\t\t\tUpsertSignalInfos:         []*persistence.SignalInfo{},\n\t\t\t\tDeleteSignalInfos:         []int64{},\n\t\t\t\tUpsertSignalRequestedIDs:  []string{},\n\t\t\t\tDeleteSignalRequestedIDs:  []string{},\n\t\t\t\tNewBufferedEvents:         nil,\n\t\t\t\tClearBufferedEvents:       false,\n\t\t\t\tVersionHistories:          mutableState.GetVersionHistories(),\n\t\t\t\tWorkflowRequests:          []*persistence.WorkflowRequest{},\n\t\t\t},\n\t\t\tNewWorkflowSnapshot: nil,\n\t\t\tWorkflowRequestMode: persistence.CreateWorkflowRequestModeReplicated,\n\t\t\tEncoding:            commonconstants.EncodingType(s.mockShard.GetConfig().EventEncodingType(s.domainID)),\n\t\t\tDomainName:          constants.TestDomainName,\n\t\t}, input)\n\t\treturn true\n\t})).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessDecisionTimeout_Pending() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\tstartedEvent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\tnextEventID := startedEvent.ID\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\tEventID:     di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startedEvent.ID, startedEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttimerTask.GetDomainID(),\n\t\ttimerTask.GetWorkflowID(),\n\t\ttimerTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.discardDuration))\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessDecisionTimeout_ScheduleToStartTimer() {\n\n\texecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tdecisionScheduleID := int64(16384)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: execution.GetWorkflowID(),\n\t\t\tRunID:      execution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeScheduleToStart),\n\t\tEventID:     decisionScheduleID,\n\t})\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err := s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(nil, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessDecisionTimeout_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.DecisionTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t\tTimeoutType: int(types.TimeoutTypeStartToClose),\n\t\tEventID:     mutableState.GetPreviousStartedEventID() - 1,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessWorkflowBackoffTimer_Pending() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\tmutableState.FlushBufferedEvents()\n\tnextEventID := mutableState.GetNextEventID() - 1\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowBackoffTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, mutableState.GetNextEventID()-1, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, time.Now().Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttimerTask.GetDomainID(),\n\t\ttimerTask.GetWorkflowID(),\n\t\ttimerTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, time.Now().Add(s.discardDuration))\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessWorkflowBackoffTimer_Success() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowBackoffTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessWorkflowTimeout_Pending() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\tmutableState.FlushBufferedEvents()\n\tnextEventID := decisionCompletionID\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttimerTask.GetDomainID(),\n\t\ttimerTask.GetWorkflowID(),\n\t\ttimerTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now().Add(s.discardDuration))\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessWorkflowTimeout_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.WorkflowTimeoutTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestProcessRetryTimeout() {\n\n\tworkflowExecution, _, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttimerTask := s.newTimerTaskFromInfo(&persistence.ActivityRetryTimerTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tTaskID:              int64(100),\n\t\t\tVisibilityTimestamp: s.timeSource.Now(),\n\t\t},\n\t})\n\n\ts.mockShard.SetCurrentTime(s.clusterName, s.timeSource.Now())\n\t_, err = s.timerStandbyTaskExecutor.Execute(timerTask)\n\ts.Nil(err)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) newTimerTaskFromInfo(\n\ttask persistence.Task,\n) Task {\n\treturn NewHistoryTask(s.mockShard, task, QueueTypeStandbyTimer, s.logger, nil, nil, nil, nil, nil)\n}\n\nfunc (s *timerStandbyTaskExecutorSuite) TestTransferTaskTimeout() {\n\tdeleteHistoryEventTask := s.newTimerTaskFromInfo(&persistence.DeleteHistoryEventTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: \"workflowID\",\n\t\t\tRunID:      \"runID\",\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(100),\n\t\t},\n\t})\n\ts.timerStandbyTaskExecutor.Execute(deleteHistoryEventTask)\n}\n"
  },
  {
    "path": "service/history/task/timer_task_executor_base.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nvar (\n\ttaskRetryPolicy = common.CreateTaskProcessingRetryPolicy()\n)\n\ntype (\n\ttimerTaskExecutorBase struct {\n\t\tshard          shard.Context\n\t\tarchiverClient archiver.Client\n\t\texecutionCache execution.Cache\n\t\tlogger         log.Logger\n\t\tmetricsClient  metrics.Client\n\t\tconfig         *config.Config\n\t\tthrottleRetry  *backoff.ThrottleRetry\n\t\tctx            context.Context\n\t\tcancelFn       context.CancelFunc\n\t}\n)\n\nfunc newTimerTaskExecutorBase(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconfig *config.Config,\n) *timerTaskExecutorBase {\n\tctx, cancelFn := context.WithCancel(context.Background())\n\treturn &timerTaskExecutorBase{\n\t\tshard:          shard,\n\t\tarchiverClient: archiverClient,\n\t\texecutionCache: executionCache,\n\t\tlogger:         logger,\n\t\tmetricsClient:  metricsClient,\n\t\tconfig:         config,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t\t),\n\t\tctx:      ctx,\n\t\tcancelFn: cancelFn,\n\t}\n}\n\nfunc (t *timerTaskExecutorBase) executeDeleteHistoryEventTask(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TimerQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil {\n\t\tt.logger.Debug(\"could not load mutable state while attempting to clean up workflow\",\n\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t)\n\t\tt.metricsClient.IncCounter(metrics.HistoryProcessDeleteHistoryEventScope, metrics.TimerProcessingDeletionTimerNoopDueToMutableStateNotLoading)\n\t\treturn nil\n\t}\n\tif mutableState.IsWorkflowExecutionRunning() {\n\t\tt.logger.Warn(\"could not clean up workflow, it was running\",\n\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t)\n\t\tt.metricsClient.IncCounter(metrics.HistoryProcessDeleteHistoryEventScope, metrics.TimerProcessingDeletionTimerNoopDueToMutableStateNotLoading)\n\t\treturn nil\n\t}\n\n\tlastWriteVersion, err := mutableState.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\tok, err := verifyTaskVersion(t.shard, t.logger, task.DomainID, lastWriteVersion, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tdomainCacheEntry, err := t.shard.GetDomainCache().GetDomainByID(task.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tclusterConfiguredForHistoryArchival := t.shard.GetService().GetArchivalMetadata().GetHistoryConfig().ClusterConfiguredForArchival()\n\tdomainConfiguredForHistoryArchival := domainCacheEntry.GetConfig().HistoryArchivalStatus == types.ArchivalStatusEnabled\n\tarchiveHistory := clusterConfiguredForHistoryArchival && domainConfiguredForHistoryArchival\n\n\t// TODO: @ycyang once archival backfill is in place cluster:paused && domain:enabled should be a nop rather than a delete\n\tif archiveHistory {\n\t\tt.metricsClient.IncCounter(metrics.HistoryProcessDeleteHistoryEventScope, metrics.WorkflowCleanupArchiveCount)\n\t\treturn t.archiveWorkflow(ctx, task, wfContext, mutableState, domainCacheEntry)\n\t}\n\n\tt.metricsClient.IncCounter(metrics.HistoryProcessDeleteHistoryEventScope, metrics.WorkflowCleanupDeleteCount)\n\treturn t.deleteWorkflow(ctx, task, wfContext, mutableState)\n}\n\nfunc (t *timerTaskExecutorBase) deleteWorkflow(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n\tcontext execution.Context,\n\tmsBuilder execution.MutableState,\n) error {\n\n\tif err := t.deleteWorkflowHistory(ctx, task, msBuilder); err != nil {\n\t\treturn err\n\t}\n\n\tif err := t.deleteWorkflowVisibility(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\tif err := t.deleteCurrentWorkflowExecution(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\tif err := t.deleteActiveClusterSelectionPolicy(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\t// it must be the last one due to the nature of workflow execution deletion\n\tif err := t.deleteWorkflowExecution(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\t// calling clear here to force accesses of mutable state to read database\n\t// if this is not called then callers will get mutable state even though its been removed from database\n\tcontext.Clear()\n\treturn nil\n}\n\nfunc (t *timerTaskExecutorBase) archiveWorkflow(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n\tworkflowContext execution.Context,\n\tmsBuilder execution.MutableState,\n\tdomainCacheEntry *cache.DomainCacheEntry,\n) error {\n\tbranchToken, err := msBuilder.GetCurrentBranchToken()\n\tif err != nil {\n\t\treturn err\n\t}\n\tcloseFailoverVersion, err := msBuilder.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treq := &archiver.ClientRequest{\n\t\tArchiveRequest: &archiver.ArchiveRequest{\n\t\t\tDomainID:             task.DomainID,\n\t\t\tWorkflowID:           task.WorkflowID,\n\t\t\tRunID:                task.RunID,\n\t\t\tDomainName:           domainCacheEntry.GetInfo().Name,\n\t\t\tShardID:              t.shard.GetShardID(),\n\t\t\tTargets:              []archiver.ArchivalTarget{archiver.ArchiveTargetHistory},\n\t\t\tURI:                  domainCacheEntry.GetConfig().HistoryArchivalURI,\n\t\t\tNextEventID:          msBuilder.GetNextEventID(),\n\t\t\tBranchToken:          branchToken,\n\t\t\tCloseFailoverVersion: closeFailoverVersion,\n\t\t},\n\t\tCallerService:        service.History,\n\t\tAttemptArchiveInline: false, // archive in workflow by default\n\t}\n\texecutionStats, err := workflowContext.LoadExecutionStats(ctx)\n\tif err == nil && executionStats.HistorySize < int64(t.config.TimerProcessorHistoryArchivalSizeLimit()) {\n\t\treq.AttemptArchiveInline = true\n\t}\n\n\tarchiveCtx, cancel := context.WithTimeout(ctx, t.config.TimerProcessorArchivalTimeLimit())\n\tdefer cancel()\n\tresp, err := t.archiverClient.Archive(archiveCtx, req)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// delete workflow history if history archival is not needed or history as been archived inline\n\tif resp.HistoryArchivedInline {\n\t\tt.metricsClient.IncCounter(metrics.HistoryProcessDeleteHistoryEventScope, metrics.WorkflowCleanupDeleteHistoryInlineCount)\n\t\tif err := t.deleteWorkflowHistory(ctx, task, msBuilder); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\t// delete visibility record here regardless if it's been archived inline or not\n\t// since the entire record is included as part of the archive request.\n\tif err := t.deleteWorkflowVisibility(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\tif err := t.deleteActiveClusterSelectionPolicy(ctx, task); err != nil {\n\t\treturn err\n\t}\n\n\tif err := t.deleteCurrentWorkflowExecution(ctx, task); err != nil {\n\t\treturn err\n\t}\n\tif err := t.deleteWorkflowExecution(ctx, task); err != nil {\n\t\treturn err\n\t}\n\t// calling clear here to force accesses of mutable state to read database\n\t// if this is not called then callers will get mutable state even though its been removed from database\n\tworkflowContext.Clear()\n\treturn nil\n}\n\nfunc (t *timerTaskExecutorBase) deleteWorkflowExecution(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n) error {\n\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context) error {\n\t\treturn t.shard.GetExecutionManager().DeleteWorkflowExecution(ctx, &persistence.DeleteWorkflowExecutionRequest{\n\t\t\tDomainID:   task.DomainID,\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t\tDomainName: domainName,\n\t\t})\n\t}\n\treturn t.throttleRetry.Do(ctx, op)\n}\n\nfunc (t *timerTaskExecutorBase) deleteCurrentWorkflowExecution(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n) error {\n\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\top := func(ctx context.Context) error {\n\t\treturn t.shard.GetExecutionManager().DeleteCurrentWorkflowExecution(ctx, &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\t\tDomainID:   task.DomainID,\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t\tDomainName: domainName,\n\t\t})\n\t}\n\treturn t.throttleRetry.Do(ctx, op)\n}\n\nfunc (t *timerTaskExecutorBase) deleteActiveClusterSelectionPolicy(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n) error {\n\top := func(ctx context.Context) error {\n\t\treturn t.shard.GetExecutionManager().DeleteActiveClusterSelectionPolicy(ctx, task.DomainID, task.WorkflowID, task.RunID)\n\t}\n\treturn t.throttleRetry.Do(ctx, op)\n}\n\nfunc (t *timerTaskExecutorBase) deleteWorkflowHistory(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n\tmsBuilder execution.MutableState,\n) error {\n\n\top := func(ctx context.Context) error {\n\t\tbranchToken, err := msBuilder.GetCurrentBranchToken()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn t.shard.GetHistoryManager().DeleteHistoryBranch(ctx, &persistence.DeleteHistoryBranchRequest{\n\t\t\tBranchToken: branchToken,\n\t\t\tShardID:     common.IntPtr(t.shard.GetShardID()),\n\t\t\tDomainName:  domainName,\n\t\t})\n\n\t}\n\treturn t.throttleRetry.Do(ctx, op)\n}\n\nfunc (t *timerTaskExecutorBase) deleteWorkflowVisibility(\n\tctx context.Context,\n\ttask *persistence.DeleteHistoryEventTask,\n) error {\n\n\tdomain, errorDomainName := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif errorDomainName != nil {\n\t\treturn errorDomainName\n\t}\n\top := func(ctx context.Context) error {\n\t\trequest := &persistence.VisibilityDeleteWorkflowExecutionRequest{\n\t\t\tDomainID:   task.DomainID,\n\t\t\tDomain:     domain,\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t\tTaskID:     task.TaskID,\n\t\t}\n\t\t// TODO: expose GetVisibilityManager method on shardContext interface\n\t\treturn t.shard.GetService().GetVisibilityManager().DeleteWorkflowExecution(ctx, request) // delete from db\n\t}\n\treturn t.throttleRetry.Do(ctx, op)\n}\n\nfunc (t *timerTaskExecutorBase) Stop() {\n\tt.logger.Info(\"Stopping timerTaskExecutorBase\")\n\tt.cancelFn()\n}\n"
  },
  {
    "path": "service/history/task/timer_task_executor_base_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\texecutioncache \"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\ttimerQueueTaskExecutorBaseSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller                   *gomock.Controller\n\t\tmockShard                    *shard.TestContext\n\t\tmockWorkflowExecutionContext *execution.MockContext\n\t\tmockMutableState             *execution.MockMutableState\n\n\t\tmockExecutionManager  *mocks.ExecutionManager\n\t\tmockVisibilityManager *mocks.VisibilityManager\n\t\tmockHistoryV2Manager  *mocks.HistoryV2Manager\n\t\tmockArchivalClient    *archiver.MockClient\n\n\t\ttimerQueueTaskExecutorBase *timerTaskExecutorBase\n\t}\n)\n\nfunc TestTimerQueueTaskExecutorBaseSuite(t *testing.T) {\n\ts := new(timerQueueTaskExecutorBaseSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) SetupSuite() {\n\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) TearDownSuite() {\n\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockWorkflowExecutionContext = execution.NewMockContext(s.controller)\n\ts.mockMutableState = execution.NewMockMutableState(s.controller)\n\n\ttestConfig := config.NewForTest()\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ttestConfig,\n\t)\n\n\ts.mockExecutionManager = s.mockShard.Resource.ExecutionMgr\n\ts.mockVisibilityManager = s.mockShard.Resource.VisibilityMgr\n\ts.mockHistoryV2Manager = s.mockShard.Resource.HistoryMgr\n\ts.mockArchivalClient = archiver.NewMockClient(s.controller)\n\n\tlogger := s.mockShard.GetLogger()\n\n\ts.timerQueueTaskExecutorBase = newTimerTaskExecutorBase(\n\t\ts.mockShard,\n\t\ts.mockArchivalClient,\n\t\tnil,\n\t\tlogger,\n\t\ts.mockShard.GetMetricsClient(),\n\t\ttestConfig,\n\t)\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n\ts.timerQueueTaskExecutorBase.Stop()\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) TestDeleteWorkflow_NoErr() {\n\ttask := &persistence.DeleteHistoryEventTask{\n\t\tTaskData: persistence.TaskData{\n\t\t\tTaskID:              12345,\n\t\t\tVisibilityTimestamp: time.Now(),\n\t\t},\n\t}\n\texecutionInfo := types.WorkflowExecution{\n\t\tWorkflowID: task.WorkflowID,\n\t\tRunID:      task.RunID,\n\t}\n\twfContext := execution.NewContext(task.DomainID, executionInfo, s.mockShard, s.mockExecutionManager, log.NewNoop())\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"Sample\", nil).AnyTimes()\n\n\ts.mockExecutionManager.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionManager.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionManager.On(\"DeleteActiveClusterSelectionPolicy\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockHistoryV2Manager.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockVisibilityManager.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte{1, 2, 3}, nil).Times(1)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(int64(1234), nil).AnyTimes()\n\n\terr := s.timerQueueTaskExecutorBase.deleteWorkflow(context.Background(), task, wfContext, s.mockMutableState)\n\ts.NoError(err)\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) TestArchiveHistory_NoErr_InlineArchivalFailed() {\n\ts.mockWorkflowExecutionContext.EXPECT().LoadExecutionStats(gomock.Any()).Return(&persistence.ExecutionStats{\n\t\tHistorySize: 1024,\n\t}, nil).Times(1)\n\ts.mockWorkflowExecutionContext.EXPECT().Clear().Times(1)\n\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"Sample\", nil)\n\n\ts.mockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte{1, 2, 3}, nil).Times(1)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(int64(1234), nil).Times(1)\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(int64(101)).Times(1)\n\ts.mockShard.Resource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"Sample\", nil).AnyTimes()\n\ts.mockExecutionManager.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionManager.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockExecutionManager.On(\"DeleteActiveClusterSelectionPolicy\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockVisibilityManager.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\n\ts.mockArchivalClient.EXPECT().Archive(gomock.Any(), gomock.Cond(func(req *archiver.ClientRequest) bool {\n\t\treturn req.CallerService == service.History && req.AttemptArchiveInline && req.ArchiveRequest.Targets[0] == archiver.ArchiveTargetHistory\n\t})).Return(&archiver.ClientResponse{\n\t\tHistoryArchivedInline: false,\n\t}, nil)\n\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{},\n\t\t&persistence.DomainConfig{},\n\t\tfalse,\n\t\tnil,\n\t\t0,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\terr := s.timerQueueTaskExecutorBase.archiveWorkflow(\n\t\tcontext.Background(),\n\t\t&persistence.DeleteHistoryEventTask{},\n\t\ts.mockWorkflowExecutionContext,\n\t\ts.mockMutableState,\n\t\tdomainCacheEntry,\n\t)\n\ts.NoError(err)\n}\n\nfunc (s *timerQueueTaskExecutorBaseSuite) TestArchiveHistory_SendSignalErr() {\n\ts.mockWorkflowExecutionContext.EXPECT().LoadExecutionStats(gomock.Any()).Return(&persistence.ExecutionStats{\n\t\tHistorySize: 1024 * 1024 * 1024,\n\t}, nil).Times(1)\n\n\ts.mockMutableState.EXPECT().GetCurrentBranchToken().Return([]byte{1, 2, 3}, nil).Times(1)\n\ts.mockMutableState.EXPECT().GetLastWriteVersion().Return(int64(1234), nil).Times(1)\n\ts.mockMutableState.EXPECT().GetNextEventID().Return(int64(101)).Times(1)\n\n\ts.mockArchivalClient.EXPECT().Archive(gomock.Any(), gomock.Cond(func(req *archiver.ClientRequest) bool {\n\t\treturn req.CallerService == service.History && !req.AttemptArchiveInline && req.ArchiveRequest.Targets[0] == archiver.ArchiveTargetHistory\n\t})).Return(nil, errors.New(\"failed to send signal\"))\n\n\tdomainCacheEntry := cache.NewDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{},\n\t\t&persistence.DomainConfig{},\n\t\tfalse,\n\t\tnil,\n\t\t0,\n\t\tnil,\n\t\t0,\n\t\t0,\n\t\t0,\n\t)\n\terr := s.timerQueueTaskExecutorBase.archiveWorkflow(\n\t\tcontext.Background(),\n\t\t&persistence.DeleteHistoryEventTask{},\n\t\ts.mockWorkflowExecutionContext,\n\t\ts.mockMutableState, domainCacheEntry,\n\t)\n\ts.Error(err)\n}\n\nfunc TestExecuteDeleteHistoryEventTask(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMocks    func(*gomock.Controller) (*timerTaskExecutorBase, *persistence.DeleteHistoryEventTask)\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"mutable was not able to be loaded\",\n\t\t\tsetupMocks: func(controller *gomock.Controller) (*timerTaskExecutorBase, *persistence.DeleteHistoryEventTask) {\n\n\t\t\t\tmockShard := shard.NewTestContext(\n\t\t\t\t\tt,\n\t\t\t\t\tcontroller,\n\t\t\t\t\t&persistence.ShardInfo{\n\t\t\t\t\t\tShardID:          0,\n\t\t\t\t\t\tRangeID:          1,\n\t\t\t\t\t\tTransferAckLevel: 0,\n\t\t\t\t\t},\n\t\t\t\t\tconfig.NewForTest(),\n\t\t\t\t)\n\n\t\t\t\tmockShard.Resource.DomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\tfalse,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t), nil)\n\n\t\t\t\ttimerTask := &persistence.DeleteHistoryEventTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              123,\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t},\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"test-domain\",\n\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t\tRunID:      \"run\",\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\twfContext := execution.NewContext(\n\t\t\t\t\ttimerTask.DomainID,\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: timerTask.WorkflowID,\n\t\t\t\t\t\tRunID:      timerTask.RunID,\n\t\t\t\t\t},\n\t\t\t\t\tmockShard,\n\t\t\t\t\tmockShard.Resource.ExecutionMgr,\n\t\t\t\t\tmockShard.GetLogger(),\n\t\t\t\t)\n\n\t\t\t\texecutionCache := executioncache.NewMockCache(controller)\n\t\t\t\texecutionCache.EXPECT().GetOrCreateWorkflowExecutionWithTimeout(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(wfContext, func(error) {}, nil)\n\n\t\t\t\tmockExecutionMgr := mockShard.Resource.ExecutionMgr\n\t\t\t\tmockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(nil, &types.EntityNotExistsError{}).Once()\n\n\t\t\t\treturn newTimerTaskExecutorBase(\n\t\t\t\t\tmockShard,\n\t\t\t\t\t&archiver.MockClient{},\n\t\t\t\t\texecutionCache,\n\t\t\t\t\tmockShard.GetLogger(),\n\t\t\t\t\tmockShard.GetMetricsClient(),\n\t\t\t\t\tmockShard.GetConfig(),\n\t\t\t\t), timerTask\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"mutable showed that the workflow was still running \",\n\t\t\tsetupMocks: func(controller *gomock.Controller) (*timerTaskExecutorBase, *persistence.DeleteHistoryEventTask) {\n\n\t\t\t\tmockShard := shard.NewTestContext(\n\t\t\t\t\tt,\n\t\t\t\t\tcontroller,\n\t\t\t\t\t&persistence.ShardInfo{\n\t\t\t\t\t\tShardID:          0,\n\t\t\t\t\t\tRangeID:          1,\n\t\t\t\t\t\tTransferAckLevel: 0,\n\t\t\t\t\t},\n\t\t\t\t\tconfig.NewForTest(),\n\t\t\t\t)\n\n\t\t\t\tmockShard.Resource.DomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{},\n\t\t\t\t\t&persistence.DomainConfig{},\n\t\t\t\t\tfalse,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\tnil,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t\t0,\n\t\t\t\t), nil).AnyTimes()\n\t\t\t\tmockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), \"domain\", \"wf\", \"run\").Return(&types.ActiveClusterInfo{}, nil).AnyTimes()\n\n\t\t\t\ttimerTask := &persistence.DeleteHistoryEventTask{\n\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\tTaskID:              123,\n\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t},\n\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\tDomainID:   \"domain\",\n\t\t\t\t\t\tWorkflowID: \"wf\",\n\t\t\t\t\t\tRunID:      \"run\",\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\twfContext := execution.NewContext(\n\t\t\t\t\ttimerTask.DomainID,\n\t\t\t\t\ttypes.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: timerTask.WorkflowID,\n\t\t\t\t\t\tRunID:      timerTask.RunID,\n\t\t\t\t\t},\n\t\t\t\t\tmockShard,\n\t\t\t\t\tmockShard.Resource.ExecutionMgr,\n\t\t\t\t\tmockShard.GetLogger(),\n\t\t\t\t)\n\n\t\t\t\texecutionCache := executioncache.NewMockCache(controller)\n\t\t\t\texecutionCache.EXPECT().GetOrCreateWorkflowExecutionWithTimeout(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(wfContext, func(error) {}, nil)\n\n\t\t\t\tmockExecutionMgr := mockShard.Resource.ExecutionMgr\n\t\t\t\tmockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\tState: &persistence.WorkflowMutableState{\n\t\t\t\t\t\tExecutionStats: &persistence.ExecutionStats{},\n\t\t\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\t\t\tDomainID:    \"domain\",\n\t\t\t\t\t\t\tWorkflowID:  \"wf\",\n\t\t\t\t\t\t\tRunID:       \"run\",\n\t\t\t\t\t\t\tCloseStatus: 0,\n\t\t\t\t\t\t\tState:       persistence.WorkflowStateRunning,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\treturn newTimerTaskExecutorBase(\n\t\t\t\t\tmockShard,\n\t\t\t\t\t&archiver.MockClient{},\n\t\t\t\t\texecutionCache,\n\t\t\t\t\tmockShard.GetLogger(),\n\t\t\t\t\tmockShard.GetMetricsClient(),\n\t\t\t\t\tmockShard.GetConfig(),\n\t\t\t\t), timerTask\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tdefer controller.Finish()\n\n\t\t\texecutor, timerTask := tt.setupMocks(controller)\n\t\t\terr := executor.executeDeleteHistoryEventTask(context.Background(), timerTask)\n\t\t\tif tt.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.Equal(t, tt.expectedError, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/task/transfer_active_task_executor.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2021 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/simulation\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n\t\"github.com/uber/cadence/service/worker/parentclosepolicy\"\n)\n\nconst (\n\tresetWorkflowTimeout = 30 * time.Second\n)\n\nvar (\n\t// ErrMissingRequestCancelInfo indicates missing request cancel info\n\tErrMissingRequestCancelInfo = &types.InternalServiceError{Message: \"unable to get request cancel info\"}\n\t// ErrMissingSignalInfo indicates missing signal external\n\tErrMissingSignalInfo = &types.InternalServiceError{Message: \"unable to get signal info\"}\n)\n\nvar (\n\terrUnknownTransferTask = errors.New(\"unknown transfer task\")\n\terrWorkflowBusy        = errors.New(\"unable to get workflow execution lock within specified timeout\")\n\terrWorkflowRateLimited = errors.New(\"workflow is being rate limited for making too many requests\")\n)\n\ntype (\n\ttransferActiveTaskExecutor struct {\n\t\t*transferTaskExecutorBase\n\n\t\thistoryClient           history.Client\n\t\tparentClosePolicyClient parentclosepolicy.Client\n\t\tworkflowResetter        reset.WorkflowResetter\n\t\twfIDCache               workflowcache.WFCache\n\t}\n\n\tgeneratorF = func(taskGenerator execution.MutableStateTaskGenerator) error\n)\n\n// NewTransferActiveTaskExecutor creates a new task executor for active transfer task\nfunc NewTransferActiveTaskExecutor(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\tworkflowResetter reset.WorkflowResetter,\n\tlogger log.Logger,\n\tconfig *config.Config,\n\twfIDCache workflowcache.WFCache,\n) Executor {\n\n\treturn &transferActiveTaskExecutor{\n\t\ttransferTaskExecutorBase: newTransferTaskExecutorBase(\n\t\t\tshard,\n\t\t\tarchiverClient,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t\tconfig,\n\t\t),\n\t\thistoryClient: shard.GetService().GetHistoryClient(),\n\t\tparentClosePolicyClient: parentclosepolicy.NewClient(\n\t\t\tshard.GetMetricsClient(),\n\t\t\tshard.GetLogger(),\n\t\t\tshard.GetService().GetSDKClient(),\n\t\t\tconfig.NumParentClosePolicySystemWorkflows(),\n\t\t),\n\t\tworkflowResetter: workflowResetter,\n\t\twfIDCache:        wfIDCache,\n\t}\n}\n\nfunc (t *transferActiveTaskExecutor) Execute(task Task) (ExecuteResponse, error) {\n\tsimulation.LogEvents(simulation.E{\n\t\tEventName:  simulation.EventNameExecuteHistoryTask,\n\t\tHost:       t.shard.GetConfig().HostName,\n\t\tShardID:    t.shard.GetShardID(),\n\t\tDomainID:   task.GetDomainID(),\n\t\tWorkflowID: task.GetWorkflowID(),\n\t\tRunID:      task.GetRunID(),\n\t\tPayload: map[string]any{\n\t\t\t\"task_category\": persistence.HistoryTaskCategoryTransfer.Name(),\n\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t},\n\t})\n\tscope := getOrCreateDomainTaggedScope(t.shard, GetTransferTaskMetricsScope(task.GetTaskType(), true), task.GetDomainID(), t.logger)\n\texecuteResponse := ExecuteResponse{\n\t\tScope:        scope,\n\t\tIsActiveTask: true,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), taskDefaultTimeout)\n\tdefer cancel()\n\n\tswitch transferTask := task.GetInfo().(type) {\n\tcase *persistence.ActivityTask:\n\t\treturn executeResponse, t.processActivityTask(ctx, transferTask)\n\tcase *persistence.DecisionTask:\n\t\treturn executeResponse, t.processDecisionTask(ctx, transferTask)\n\tcase *persistence.CloseExecutionTask:\n\t\treturn executeResponse, t.processCloseExecution(ctx, transferTask)\n\tcase *persistence.RecordWorkflowClosedTask:\n\t\treturn executeResponse, t.processRecordWorkflowClosed(ctx, transferTask)\n\tcase *persistence.RecordChildExecutionCompletedTask:\n\t\treturn executeResponse, t.processRecordChildExecutionCompleted(ctx, transferTask)\n\tcase *persistence.CancelExecutionTask:\n\t\treturn executeResponse, t.processCancelExecution(ctx, transferTask)\n\tcase *persistence.SignalExecutionTask:\n\t\treturn executeResponse, t.processSignalExecution(ctx, transferTask)\n\tcase *persistence.StartChildExecutionTask:\n\t\treturn executeResponse, t.processStartChildExecution(ctx, transferTask)\n\tcase *persistence.RecordWorkflowStartedTask:\n\t\treturn executeResponse, t.processRecordWorkflowStarted(ctx, transferTask)\n\tcase *persistence.ResetWorkflowTask:\n\t\treturn executeResponse, t.processResetWorkflow(ctx, transferTask)\n\tcase *persistence.UpsertWorkflowSearchAttributesTask:\n\t\treturn executeResponse, t.processUpsertWorkflowSearchAttributes(ctx, transferTask)\n\tdefault:\n\t\treturn executeResponse, errUnknownTransferTask\n\t}\n}\n\n// Empty func for now\nfunc (t *transferActiveTaskExecutor) Stop() {}\n\nfunc (t *transferActiveTaskExecutor) processActivityTask(\n\tctx context.Context,\n\ttask *persistence.ActivityTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, task.ScheduleID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\tai, ok := mutableState.GetActivityInfo(task.ScheduleID)\n\tif !ok {\n\t\tt.logger.Debug(\"Potentially duplicate \", tag.TaskID(task.TaskID), tag.WorkflowScheduleID(task.ScheduleID), tag.TaskType(persistence.TransferTaskTypeActivityTask))\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, ai.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\ttimeout := min(ai.ScheduleToStartTimeout, constants.MaxTaskTimeout)\n\n\ttaskList := types.TaskList{\n\t\tName: ai.TaskList,\n\t\tKind: ai.TaskListKind.Ptr(),\n\t}\n\tif taskList.Name == \"\" {\n\t\ttaskList.Name = task.TaskList\n\t}\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(nil)\n\n\t// Rate limiting task processing requests\n\tif !t.allowTask(task) {\n\t\treturn errWorkflowRateLimited\n\t}\n\n\tpushActivityInfo := &pushActivityToMatchingInfo{\n\t\tactivityScheduleToStartTimeout: timeout,\n\t\ttasklist:                       taskList,\n\t\tpartitionConfig:                mutableState.GetExecutionInfo().PartitionConfig,\n\t}\n\terr = t.pushActivity(ctx, task, pushActivityInfo)\n\tif err == nil {\n\t\tscope := common.NewPerTaskListScope(domainName, taskList.Name, taskList.GetKind(), t.metricsClient, metrics.TransferActiveTaskActivityScope)\n\t\tscope.RecordTimer(metrics.ScheduleToStartHistoryQueueLatencyPerTaskList, time.Since(task.GetVisibilityTimestamp()))\n\t}\n\treturn err\n}\n\nfunc (t *transferActiveTaskExecutor) processDecisionTask(\n\tctx context.Context,\n\ttask *persistence.DecisionTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, task.ScheduleID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tdecision, found := mutableState.GetDecisionInfo(task.ScheduleID)\n\tif !found {\n\t\tt.logger.Debug(\"Potentially duplicate \", tag.TaskID(task.TaskID), tag.WorkflowScheduleID(task.ScheduleID), tag.TaskType(persistence.TransferTaskTypeDecisionTask))\n\t\treturn nil\n\t}\n\tok, err := verifyTaskVersion(t.shard, t.logger, task.DomainID, decision.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tworkflowTimeout := executionInfo.WorkflowTimeout\n\tdecisionTimeout := min(workflowTimeout, constants.MaxTaskTimeout)\n\n\t// NOTE: previously this section check whether mutable state has enabled\n\t// sticky decision, if so convert the decision to a sticky decision.\n\t// that logic has a bug which timer task for that sticky decision is not generated\n\t// the correct logic should check whether the decision task is a sticky decision\n\t// task or not.\n\ttaskList := types.TaskList{\n\t\tName: task.TaskList,\n\t\tKind: executionInfo.TaskListKind.Ptr(),\n\t}\n\tif mutableState.GetExecutionInfo().TaskList != task.TaskList {\n\t\t// this decision is an sticky decision\n\t\t// there shall already be an timer set\n\t\ttaskList.Kind = types.TaskListKindSticky.Ptr()\n\t\tdecisionTimeout = executionInfo.StickyScheduleToStartTimeout\n\t}\n\t// TODO: for normal decision, we don't know if there's a scheduleToStart\n\t// timeout timer task associated with the decision since it's determined\n\t// when creating the decision and the result is not persisted in mutable\n\t// state.\n\t// If we calculated the timeout again here, the timeout may be different,\n\t// or even lost the decision if there's originally no timeout timer task\n\t// for the decision. Using MaxTaskTimeout here for now so at least no\n\t// decision will be lost.\n\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(nil)\n\n\t// Rate limiting task processing requests\n\tif !t.allowTask(task) {\n\t\treturn errWorkflowRateLimited\n\t}\n\n\terr = t.pushDecision(ctx, task, &pushDecisionToMatchingInfo{\n\t\tdecisionScheduleToStartTimeout: decisionTimeout,\n\t\ttasklist:                       taskList,\n\t\tpartitionConfig:                mutableState.GetExecutionInfo().PartitionConfig,\n\t})\n\tif _, ok := err.(*types.StickyWorkerUnavailableError); ok {\n\t\t// sticky worker is unavailable, switch to non-sticky task list\n\t\ttaskList = types.TaskList{\n\t\t\tName: mutableState.GetExecutionInfo().TaskList,\n\t\t}\n\n\t\t// Continue to use sticky schedule_to_start timeout as TTL for the matching task. Because the schedule_to_start\n\t\t// timeout timer task is already created which will timeout this task if no worker pick it up in 5s anyway.\n\t\t// There is no need to reset sticky, because if this task is picked by new worker, the new worker will reset\n\t\t// the sticky queue to a new one. However, if worker is completely down, that schedule_to_start timeout task\n\t\t// will re-create a new non-sticky task and reset sticky.\n\t\terr = t.pushDecision(ctx, task, &pushDecisionToMatchingInfo{\n\t\t\tdecisionScheduleToStartTimeout: decisionTimeout,\n\t\t\ttasklist:                       taskList,\n\t\t\tpartitionConfig:                mutableState.GetExecutionInfo().PartitionConfig,\n\t\t})\n\t}\n\tif err == nil {\n\t\tscope := common.NewPerTaskListScope(domainName, taskList.Name, taskList.GetKind(), t.metricsClient, metrics.TransferActiveTaskDecisionScope)\n\t\tscope.RecordTimer(metrics.ScheduleToStartHistoryQueueLatencyPerTaskList, time.Since(task.GetVisibilityTimestamp()))\n\t}\n\treturn err\n}\n\nfunc (t *transferActiveTaskExecutor) allowTask(task persistence.Task) bool {\n\treturn t.wfIDCache.AllowInternal(task.GetDomainID(), task.GetWorkflowID())\n}\n\nfunc (t *transferActiveTaskExecutor) processCloseExecution(\n\tctx context.Context,\n\ttask *persistence.CloseExecutionTask,\n) error {\n\treturn t.processCloseExecutionTaskHelper(ctx, task, true, true, true)\n}\n\nfunc (t *transferActiveTaskExecutor) processRecordWorkflowClosed(\n\tctx context.Context,\n\ttask *persistence.RecordWorkflowClosedTask,\n) error {\n\treturn t.processCloseExecutionTaskHelper(ctx, task, true, false, false)\n}\n\nfunc (t *transferActiveTaskExecutor) processRecordChildExecutionCompleted(\n\tctx context.Context,\n\ttask *persistence.RecordChildExecutionCompletedTask,\n) error {\n\treturn t.processCloseExecutionTaskHelper(ctx, task, false, true, false)\n}\n\n// TODO: this helper function performs three operations:\n// 1. publish workflow closed visibility record\n// 2. if has parent workflow, reply to the parent workflow\n// 3. if has child workflow(s), apply parent close policy\n// ideally we should separate them into 3 functions, but it is complicated\n// but the fact that we want to release mutable state lock as early as possible\n// we should see if there's a better way to organize the code\nfunc (t *transferActiveTaskExecutor) processCloseExecutionTaskHelper(\n\tctx context.Context,\n\ttask persistence.Task,\n\trecordWorkflowClosed bool,\n\treplyToParentWorkflowIfApplicable bool,\n\tapplyParentClosePolicy bool,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.GetDomainID(),\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tlastWriteVersion, err := mutableState.GetLastWriteVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\tok, err := verifyTaskVersion(t.shard, t.logger, task.GetDomainID(), lastWriteVersion, task.GetVersion(), task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(task.GetDomainID())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tcompletionEvent, err := mutableState.GetCompletionEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\twfCloseTime := completionEvent.GetTimestamp()\n\n\tparentDomainID := executionInfo.ParentDomainID\n\tparentWorkflowID := executionInfo.ParentWorkflowID\n\tparentRunID := executionInfo.ParentRunID\n\tinitiatedID := executionInfo.InitiatedID\n\n\tworkflowTypeName := executionInfo.WorkflowTypeName\n\tworkflowCloseTimestamp := wfCloseTime\n\tworkflowCloseStatus := persistence.ToInternalWorkflowExecutionCloseStatus(executionInfo.CloseStatus)\n\tworkflowHistoryLength := mutableState.GetNextEventID() - 1\n\tisCron := len(executionInfo.CronSchedule) > 0\n\tcronSchedule := executionInfo.CronSchedule\n\tnumClusters := (int16)(len(domainEntry.GetReplicationConfig().Clusters))\n\tupdateTimestamp := t.shard.GetTimeSource().Now()\n\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tworkflowStartTimestamp := startEvent.GetTimestamp()\n\tworkflowExecutionTimestamp := getWorkflowExecutionTimestamp(mutableState, startEvent)\n\tvisibilityMemo := getWorkflowMemo(executionInfo.Memo)\n\tsearchAttr := executionInfo.SearchAttributes\n\theaders := getWorkflowHeaders(startEvent)\n\tdomainName := mutableState.GetDomainEntry().GetInfo().Name\n\tchildren := mutableState.GetPendingChildExecutionInfos()\n\n\texecutionStatus := getWorkflowExecutionStatus(mutableState, completionEvent)\n\n\t// Calculate ScheduledExecutionTime\n\tscheduledExecutionTimestamp := startEvent.GetTimestamp()\n\tif startEvent.WorkflowExecutionStartedEventAttributes != nil &&\n\t\tstartEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds() > 0 {\n\t\tscheduledExecutionTimestamp = startEvent.GetTimestamp() +\n\t\t\tint64(startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds())*int64(time.Second)\n\t}\n\n\t// we've gathered all necessary information from mutable state.\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(nil)\n\n\t// publish workflow closed visibility records\n\tif recordWorkflowClosed {\n\t\tif err := t.recordWorkflowClosed(\n\t\t\tctx,\n\t\t\ttask.GetDomainID(),\n\t\t\ttask.GetWorkflowID(),\n\t\t\ttask.GetRunID(),\n\t\t\tworkflowTypeName,\n\t\t\tworkflowStartTimestamp,\n\t\t\tworkflowExecutionTimestamp.UnixNano(),\n\t\t\tworkflowCloseTimestamp,\n\t\t\t*workflowCloseStatus,\n\t\t\tworkflowHistoryLength,\n\t\t\ttask.GetTaskID(),\n\t\t\tvisibilityMemo,\n\t\t\texecutionInfo.TaskList,\n\t\t\tisCron,\n\t\t\tcronSchedule,\n\t\t\tnumClusters,\n\t\t\tupdateTimestamp.UnixNano(),\n\t\t\tsearchAttr,\n\t\t\theaders,\n\t\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\t\texecutionStatus,\n\t\t\tscheduledExecutionTimestamp,\n\t\t); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\t// Communicate the result to parent execution if this is Child Workflow execution\n\t// and parent domain is in the same cluster\n\treplyToParentWorkflow := replyToParentWorkflowIfApplicable && mutableState.HasParentExecution() && executionInfo.CloseStatus != persistence.WorkflowCloseStatusContinuedAsNew\n\tif replyToParentWorkflow {\n\t\trecordChildCompletionCtx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\t\tdefer cancel()\n\t\terr := t.historyClient.RecordChildExecutionCompleted(recordChildCompletionCtx, &types.RecordChildExecutionCompletedRequest{\n\t\t\tDomainUUID: parentDomainID,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: parentWorkflowID,\n\t\t\t\tRunID:      parentRunID,\n\t\t\t},\n\t\t\tInitiatedID: initiatedID,\n\t\t\tCompletedExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: task.GetWorkflowID(),\n\t\t\t\tRunID:      task.GetRunID(),\n\t\t\t},\n\t\t\tCompletionEvent: completionEvent,\n\t\t})\n\n\t\t// Check to see if the error is non-transient, in which case reset the error and continue with processing\n\t\tswitch err.(type) {\n\t\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError:\n\t\t\terr = nil\n\t\t}\n\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif applyParentClosePolicy {\n\n\t\tparentExecution := types.WorkflowExecution{\n\t\t\tWorkflowID: task.GetWorkflowID(),\n\t\t\tRunID:      task.GetRunID(),\n\t\t}\n\n\t\terr := t.processParentClosePolicy(ctx, task.GetDomainID(), domainName, &parentExecution, children)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc (t *transferActiveTaskExecutor) processCancelExecution(\n\tctx context.Context,\n\ttask *persistence.CancelExecutionTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, task.InitiatedID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tinitiatedEventID := task.InitiatedID\n\trequestCancelInfo, ok := mutableState.GetRequestCancelInfo(initiatedEventID)\n\tif !ok {\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, requestCancelInfo.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\ttargetDomainEntry, err := t.shard.GetDomainCache().GetDomainByID(task.TargetDomainID)\n\tif err != nil {\n\t\t// TODO: handle the case where target domain does not exist\n\t\treturn err\n\t}\n\n\ttargetDomainName := targetDomainEntry.GetInfo().Name\n\n\t// handle workflow cancel itself\n\tif task.DomainID == task.TargetDomainID && task.WorkflowID == task.TargetWorkflowID {\n\t\t// it does not matter if the run ID is a mismatch\n\t\terr = requestCancelExternalExecutionFailed(\n\t\t\tctx,\n\t\t\tt.logger,\n\t\t\ttask,\n\t\t\twfContext,\n\t\t\ttargetDomainName,\n\t\t\ttask.TargetWorkflowID,\n\t\t\ttask.TargetRunID,\n\t\t\tt.shard.GetTimeSource().Now(),\n\t\t)\n\t\treturn err\n\t}\n\n\tif err = requestCancelExternalExecutionWithRetry(\n\t\tctx,\n\t\tt.historyClient,\n\t\ttask,\n\t\ttargetDomainName,\n\t\trequestCancelInfo.CancelRequestID,\n\t); err != nil {\n\t\tt.logger.Error(\"Failed to cancel external workflow execution\",\n\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\t\ttag.TargetWorkflowID(task.TargetWorkflowID),\n\t\t\ttag.TargetWorkflowRunID(task.TargetRunID),\n\t\t\ttag.Error(err))\n\n\t\t// Check to see if the error is non-transient, in which case add RequestCancelFailed\n\t\t// event and complete transfer task by setting the err = nil\n\t\tif common.IsServiceTransientError(err) || common.IsContextTimeoutError(err) {\n\t\t\t// for retryable error just return\n\t\t\treturn err\n\t\t}\n\t\treturn requestCancelExternalExecutionFailed(\n\t\t\tctx,\n\t\t\tt.logger,\n\t\t\ttask,\n\t\t\twfContext,\n\t\t\ttargetDomainName,\n\t\t\ttask.TargetWorkflowID,\n\t\t\ttask.TargetRunID,\n\t\t\tt.shard.GetTimeSource().Now(),\n\t\t)\n\t}\n\n\tt.logger.Debug(\"RequestCancel successfully recorded to external workflow execution\",\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\ttag.TargetWorkflowID(task.TargetWorkflowID),\n\t\ttag.TargetWorkflowRunID(task.TargetRunID))\n\n\t// Record ExternalWorkflowExecutionCancelRequested in source execution\n\treturn requestCancelExternalExecutionCompleted(\n\t\tctx,\n\t\tt.logger,\n\t\ttask,\n\t\twfContext,\n\t\ttargetDomainName,\n\t\ttask.TargetWorkflowID,\n\t\ttask.TargetRunID,\n\t\tt.shard.GetTimeSource().Now(),\n\t)\n}\n\nfunc (t *transferActiveTaskExecutor) processSignalExecution(\n\tctx context.Context,\n\ttask *persistence.SignalExecutionTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, task.InitiatedID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\tinitiatedEventID := task.InitiatedID\n\tsignalInfo, ok := mutableState.GetSignalInfo(initiatedEventID)\n\tif !ok {\n\t\t// TODO: here we should also RemoveSignalMutableState from target workflow\n\t\t// Otherwise, target SignalRequestID still can leak if shard restart after signalExternalExecutionCompleted\n\t\t// To do that, probably need to add the SignalRequestID in transfer\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, signalInfo.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\ttargetDomainEntry, err := t.shard.GetDomainCache().GetDomainByID(task.TargetDomainID)\n\tif err != nil {\n\t\t// TODO: handle the case where target domain does not exist\n\t\treturn err\n\t}\n\n\ttargetDomainName := targetDomainEntry.GetInfo().Name\n\n\t// handle workflow signal itself\n\tif task.DomainID == task.TargetDomainID && task.WorkflowID == task.TargetWorkflowID {\n\t\t// it does not matter if the run ID is a mismatch\n\t\treturn signalExternalExecutionFailed(\n\t\t\tctx,\n\t\t\tt.logger,\n\t\t\ttask,\n\t\t\twfContext,\n\t\t\ttargetDomainName,\n\t\t\ttask.TargetWorkflowID,\n\t\t\ttask.TargetRunID,\n\t\t\tsignalInfo.Control,\n\t\t\tt.shard.GetTimeSource().Now(),\n\t\t)\n\t}\n\n\tif err = signalExternalExecutionWithRetry(\n\t\tctx,\n\t\tt.historyClient,\n\t\ttask,\n\t\ttargetDomainName,\n\t\tsignalInfo,\n\t); err != nil {\n\t\tif !common.IsExpectedError(err) {\n\t\t\tt.logger.Error(\"Failed to signal external workflow execution\",\n\t\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\t\t\ttag.TargetWorkflowID(task.TargetWorkflowID),\n\t\t\t\ttag.TargetWorkflowRunID(task.TargetRunID),\n\t\t\t\ttag.Error(err))\n\t\t}\n\n\t\t// Check to see if the error is non-transient, in which case add SignalFailed\n\t\t// event and complete transfer task by setting the err = nil\n\t\tif common.IsServiceTransientError(err) || common.IsContextTimeoutError(err) {\n\t\t\t// for retryable error just return\n\t\t\treturn err\n\t\t}\n\t\treturn signalExternalExecutionFailed(\n\t\t\tctx,\n\t\t\tt.logger,\n\t\t\ttask,\n\t\t\twfContext,\n\t\t\ttargetDomainName,\n\t\t\ttask.TargetWorkflowID,\n\t\t\ttask.TargetRunID,\n\t\t\tsignalInfo.Control,\n\t\t\tt.shard.GetTimeSource().Now(),\n\t\t)\n\t}\n\n\tt.logger.Debug(\"Signal successfully recorded to external workflow execution\",\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\ttag.TargetWorkflowID(task.TargetWorkflowID),\n\t\ttag.TargetWorkflowRunID(task.TargetRunID))\n\n\terr = signalExternalExecutionCompleted(\n\t\tctx,\n\t\tt.logger,\n\t\ttask,\n\t\twfContext,\n\t\ttargetDomainName,\n\t\ttask.TargetWorkflowID,\n\t\ttask.TargetRunID,\n\t\tsignalInfo.Control,\n\t\tt.shard.GetTimeSource().Now(),\n\t)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(retError)\n\n\t// remove signalRequestedID from target workflow, after Signal detail is removed from source workflow\n\treturn removeSignalMutableStateWithRetry(ctx, t.historyClient, task, signalInfo.SignalRequestID)\n}\n\nfunc (t *transferActiveTaskExecutor) processStartChildExecution(\n\tctx context.Context,\n\ttask *persistence.StartChildExecutionTask,\n) (retError error) {\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, task.InitiatedID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil {\n\t\treturn nil\n\t}\n\n\tinitiatedEventID := task.InitiatedID\n\tchildInfo, ok := mutableState.GetChildExecutionInfo(initiatedEventID)\n\tif !ok {\n\t\treturn nil\n\t}\n\tok, err = verifyTaskVersion(t.shard, t.logger, task.DomainID, childInfo.Version, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\tworkflowRunning := mutableState.IsWorkflowExecutionRunning()\n\tchildStarted := childInfo.StartedID != constants.EmptyEventID\n\n\tif !workflowRunning && (!childStarted || childInfo.ParentClosePolicy != types.ParentClosePolicyAbandon) {\n\t\t// three cases here:\n\t\t// case 1: workflow not running, child started, close policy is not abandon\n\t\t// case 2 & 3: workflow not running, child not started, close policy is or is not abandon\n\t\treturn nil\n\t}\n\n\t// Get target domain name\n\tvar targetDomainName string\n\tvar targetDomainEntry *cache.DomainCacheEntry\n\tif targetDomainEntry, err = t.shard.GetDomainCache().GetDomainByID(task.TargetDomainID); err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\t\t// TODO: handle the case where target domain does not exist\n\n\t\t// it is possible that the domain got deleted. Use domainID instead as this is only needed for the history event\n\t\ttargetDomainName = task.TargetDomainID\n\t} else {\n\n\t\ttargetDomainName = targetDomainEntry.GetInfo().Name\n\t}\n\n\t// ChildExecution already started, just create DecisionTask and complete transfer task\n\t// if parent already closed, since child workflow started event already written to history,\n\t// still schedule the decision if the parent close policy is Abandon.\n\t// If parent close policy cancel, a decision will be scheduled when processing that close policy.\n\tif childStarted {\n\t\t// NOTE: do not access anything related mutable state after this lock release\n\t\t// release the context lock since we no longer need mutable state builder and\n\t\t// the rest of logic is making RPC call, which takes time.\n\t\trelease(nil)\n\t\treturn createFirstDecisionTask(\n\t\t\tctx,\n\t\t\tt.historyClient,\n\t\t\ttask.TargetDomainID,\n\t\t\t&types.WorkflowExecution{\n\t\t\t\tWorkflowID: childInfo.StartedWorkflowID,\n\t\t\t\tRunID:      childInfo.StartedRunID,\n\t\t\t})\n\t}\n\t// remaining 2 cases:\n\t// workflow running, child not started, close policy is or is not abandon\n\n\tinitiatedEvent, err := mutableState.GetChildExecutionInitiatedEvent(ctx, initiatedEventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tattributes := initiatedEvent.StartChildWorkflowExecutionInitiatedEventAttributes\n\tchildRunID, err := startWorkflowWithRetry(\n\t\tctx,\n\t\tt.historyClient,\n\t\tt.shard.GetTimeSource(),\n\t\tt.shard.GetDomainCache(),\n\t\ttask,\n\t\ttargetDomainName,\n\t\tchildInfo.CreateRequestID,\n\t\tattributes,\n\t\tmutableState.GetExecutionInfo().PartitionConfig,\n\t\tmutableState.GetExecutionInfo().ActiveClusterSelectionPolicy,\n\t)\n\tif err != nil {\n\n\t\t// Check to see if the error is non-transient, in which case add StartChildWorkflowExecutionFailed\n\t\t// event and complete transfer task by setting the err = nil\n\t\tswitch err.(type) {\n\t\t// TODO: we should also handle domain not exist error here\n\t\t// but we probably need to introduce a new error type for DomainNotExists,\n\t\t// for now when getting an EntityNotExists error, we can't tell if it's domain or workflow.\n\t\tcase *types.WorkflowExecutionAlreadyStartedError:\n\t\t\tt.logger.Info(\"workflow has already started\",\n\t\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\t\t\ttag.TargetWorkflowID(attributes.WorkflowID),\n\t\t\t\ttag.WorkflowRequestID(childInfo.CreateRequestID),\n\t\t\t)\n\t\t\terr = recordStartChildExecutionFailed(ctx, t.logger, task, wfContext, attributes, t.shard.GetTimeSource().Now())\n\t\tdefault:\n\t\t\tt.logger.Error(\"Failed to start child workflow execution\",\n\t\t\t\ttag.WorkflowDomainID(task.DomainID),\n\t\t\t\ttag.WorkflowID(task.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(task.RunID),\n\t\t\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\t\t\ttag.TargetWorkflowID(attributes.WorkflowID),\n\t\t\t\ttag.Error(err))\n\t\t}\n\t\treturn err\n\t}\n\n\tt.logger.Debug(\"Child Execution started successfully\",\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t\ttag.TargetWorkflowDomainID(task.TargetDomainID),\n\t\ttag.TargetWorkflowID(attributes.WorkflowID),\n\t\ttag.TargetWorkflowRunID(childRunID))\n\n\t// Child execution is successfully started, record ChildExecutionStartedEvent in parent execution\n\terr = recordChildExecutionStarted(ctx, t.logger, task, wfContext, attributes, childRunID, t.shard.GetTimeSource().Now())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// NOTE: do not access anything related mutable state after this lock release\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(nil)\n\t// Finally create first decision task for Child execution so it is really started\n\t// entity not exist error is checked and ignored in HandleErr() method in task.go\n\treturn createFirstDecisionTask(\n\t\tctx,\n\t\tt.historyClient,\n\t\ttask.TargetDomainID,\n\t\t&types.WorkflowExecution{\n\t\t\tWorkflowID: task.TargetWorkflowID,\n\t\t\tRunID:      childRunID,\n\t\t})\n}\n\nfunc (t *transferActiveTaskExecutor) processRecordWorkflowStarted(\n\tctx context.Context,\n\ttask *persistence.RecordWorkflowStartedTask,\n) (retError error) {\n\n\treturn t.processRecordWorkflowStartedOrUpsertHelper(ctx, task, true)\n}\n\nfunc (t *transferActiveTaskExecutor) processUpsertWorkflowSearchAttributes(\n\tctx context.Context,\n\ttask *persistence.UpsertWorkflowSearchAttributesTask,\n) (retError error) {\n\n\treturn t.processRecordWorkflowStartedOrUpsertHelper(ctx, task, false)\n}\n\nfunc (t *transferActiveTaskExecutor) processRecordWorkflowStartedOrUpsertHelper(\n\tctx context.Context,\n\ttask persistence.Task,\n\trecordStart bool,\n) (retError error) {\n\n\tworkflowStartedScope := getOrCreateDomainTaggedScope(t.shard, metrics.TransferActiveTaskRecordWorkflowStartedScope, task.GetDomainID(), t.logger)\n\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.GetDomainID(),\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { release(retError) }()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif mutableState == nil || !mutableState.IsWorkflowExecutionRunning() {\n\t\treturn nil\n\t}\n\n\t// verify task version for RecordWorkflowStarted.\n\t// upsert doesn't require verifyTask, because it is just a sync of mutableState.\n\tif recordStart {\n\t\tstartVersion, err := mutableState.GetStartVersion()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, task.GetDomainID(), startVersion, task.GetVersion(), task)\n\t\tif err != nil || !ok {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(task.GetDomainID())\n\tif err != nil {\n\t\treturn err\n\t}\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tworkflowTimeout := executionInfo.WorkflowTimeout\n\twfTypeName := executionInfo.WorkflowTypeName\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tstartTimestamp := startEvent.GetTimestamp()\n\texecutionTimestamp := getWorkflowExecutionTimestamp(mutableState, startEvent)\n\tvisibilityMemo := getWorkflowMemo(executionInfo.Memo)\n\tsearchAttr := copySearchAttributes(executionInfo.SearchAttributes)\n\theaders := getWorkflowHeaders(startEvent)\n\tisCron := len(executionInfo.CronSchedule) > 0\n\tcronSchedule := \"\"\n\tif isCron {\n\t\tcronSchedule = executionInfo.CronSchedule\n\t}\n\tnumClusters := (int16)(len(domainEntry.GetReplicationConfig().Clusters))\n\tupdateTimestamp := t.shard.GetTimeSource().Now()\n\n\tisAdvancedVisibilityEnabled := common.IsAdvancedVisibilityWritingEnabled(\n\t\tt.config.WriteVisibilityStoreName(),\n\t\tt.config.IsAdvancedVisConfigExist,\n\t)\n\texecutionStatus, scheduledExecutionTimestamp := determineExecutionStatusForVisibility(startEvent, mutableState, isAdvancedVisibilityEnabled)\n\n\t// release the context lock since we no longer need mutable state builder and\n\t// the rest of logic is making RPC call, which takes time.\n\trelease(nil)\n\n\tif recordStart {\n\t\tworkflowStartedScope.IncCounter(metrics.WorkflowStartedCount)\n\t\treturn t.recordWorkflowStarted(\n\t\t\tctx,\n\t\t\ttask.GetDomainID(),\n\t\t\ttask.GetWorkflowID(),\n\t\t\ttask.GetRunID(),\n\t\t\twfTypeName,\n\t\t\tstartTimestamp,\n\t\t\texecutionTimestamp.UnixNano(),\n\t\t\tworkflowTimeout,\n\t\t\ttask.GetTaskID(),\n\t\t\texecutionInfo.TaskList,\n\t\t\tisCron,\n\t\t\tnumClusters,\n\t\t\tvisibilityMemo,\n\t\t\tupdateTimestamp.UnixNano(),\n\t\t\tsearchAttr,\n\t\t\theaders,\n\t\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\t\tcronSchedule,\n\t\t\texecutionStatus,\n\t\t\tscheduledExecutionTimestamp,\n\t\t)\n\t}\n\treturn t.upsertWorkflowExecution(\n\t\tctx,\n\t\ttask.GetDomainID(),\n\t\ttask.GetWorkflowID(),\n\t\ttask.GetRunID(),\n\t\twfTypeName,\n\t\tstartTimestamp,\n\t\texecutionTimestamp.UnixNano(),\n\t\tworkflowTimeout,\n\t\ttask.GetTaskID(),\n\t\texecutionInfo.TaskList,\n\t\tvisibilityMemo,\n\t\tisCron,\n\t\tnumClusters,\n\t\tupdateTimestamp.UnixNano(),\n\t\tsearchAttr,\n\t\theaders,\n\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\tcronSchedule,\n\t\texecutionStatus,\n\t\tscheduledExecutionTimestamp,\n\t)\n}\n\nfunc (t *transferActiveTaskExecutor) processResetWorkflow(\n\tctx context.Context,\n\ttask *persistence.ResetWorkflowTask,\n) (retError error) {\n\n\tcurrentContext, currentRelease, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttask.DomainID,\n\t\tgetWorkflowExecution(task),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() { currentRelease(retError) }()\n\n\tcurrentMutableState, err := loadMutableState(ctx, currentContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, 0)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif currentMutableState == nil {\n\t\treturn nil\n\t}\n\n\tlogger := t.logger.WithTags(\n\t\ttag.WorkflowDomainID(task.DomainID),\n\t\ttag.WorkflowID(task.WorkflowID),\n\t\ttag.WorkflowRunID(task.RunID),\n\t)\n\tdomainName, err := t.shard.GetDomainCache().GetDomainName(task.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !currentMutableState.IsWorkflowExecutionRunning() {\n\t\t// it means this this might not be current anymore, we need to check\n\t\tvar resp *persistence.GetCurrentExecutionResponse\n\t\tresp, err = t.shard.GetExecutionManager().GetCurrentExecution(ctx, &persistence.GetCurrentExecutionRequest{\n\t\t\tDomainID:   task.DomainID,\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tDomainName: domainName,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif resp.RunID != task.RunID {\n\t\t\tlogger.Warn(\"Auto-Reset is skipped, because current run is stale.\")\n\t\t\treturn nil\n\t\t}\n\t}\n\t// TODO: current reset doesn't allow childWFs, in the future we will release this restriction\n\tif len(currentMutableState.GetPendingChildExecutionInfos()) > 0 {\n\t\tlogger.Warn(\"Auto-Reset is skipped, because current run has pending child executions.\")\n\t\treturn nil\n\t}\n\n\tcurrentStartVersion, err := currentMutableState.GetStartVersion()\n\tif err != nil {\n\t\treturn err\n\t}\n\tok, err := verifyTaskVersion(t.shard, t.logger, task.DomainID, currentStartVersion, task.Version, task)\n\tif err != nil || !ok {\n\t\treturn err\n\t}\n\n\texecutionInfo := currentMutableState.GetExecutionInfo()\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(executionInfo.DomainID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tlogger = logger.WithTags(tag.WorkflowDomainName(domainEntry.GetInfo().Name))\n\n\treason, resetPoint := execution.FindAutoResetPoint(t.shard.GetTimeSource(), &domainEntry.GetConfig().BadBinaries, executionInfo.AutoResetPoints)\n\tif resetPoint == nil {\n\t\tlogger.Warn(\"Auto-Reset is skipped, because reset point is not found.\")\n\t\treturn nil\n\t}\n\tlogger = logger.WithTags(\n\t\ttag.WorkflowResetBaseRunID(resetPoint.GetRunID()),\n\t\ttag.WorkflowBinaryChecksum(resetPoint.GetBinaryChecksum()),\n\t\ttag.WorkflowEventID(resetPoint.GetFirstDecisionCompletedID()),\n\t)\n\n\tvar baseContext execution.Context\n\tvar baseMutableState execution.MutableState\n\tvar baseRelease execution.ReleaseFunc\n\tif resetPoint.GetRunID() == executionInfo.RunID {\n\t\tbaseContext = currentContext\n\t\tbaseMutableState = currentMutableState\n\t\tbaseRelease = currentRelease\n\t} else {\n\t\tbaseExecution := types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      resetPoint.GetRunID(),\n\t\t}\n\t\tbaseContext, baseRelease, err = t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\t\ttask.DomainID,\n\t\t\tbaseExecution,\n\t\t\ttaskGetExecutionContextTimeout,\n\t\t)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tdefer func() { baseRelease(retError) }()\n\t\tbaseMutableState, err = loadMutableState(ctx, baseContext, task, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, 0)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tif baseMutableState == nil {\n\t\t\treturn nil\n\t\t}\n\t}\n\n\t// reset workflow needs to go through the history so it may take a long time.\n\t// as a result it's not subject to the taskDefaultTimeout. Otherwise the task\n\t// may got stuck if the workflow history is large.\n\treturn t.resetWorkflow(\n\t\ttask,\n\t\tdomainEntry.GetInfo().Name,\n\t\treason,\n\t\tresetPoint,\n\t\tbaseContext,\n\t\tbaseMutableState,\n\t\tcurrentContext,\n\t\tcurrentMutableState,\n\t\tlogger,\n\t)\n}\n\nfunc recordChildExecutionStarted(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.StartChildExecutionTask,\n\twfContext execution.Context,\n\tinitiatedAttributes *types.StartChildWorkflowExecutionInitiatedEventAttributes,\n\trunID string,\n\tnow time.Time,\n) error {\n\n\treturn updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tdomain := initiatedAttributes.Domain\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\tci, ok := mutableState.GetChildExecutionInfo(initiatedEventID)\n\t\t\tif !ok || ci.StartedID != constants.EmptyEventID {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Pending child execution not found.\"}\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddChildWorkflowExecutionStartedEvent(\n\t\t\t\tdomain,\n\t\t\t\t&types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: task.TargetWorkflowID,\n\t\t\t\t\tRunID:      runID,\n\t\t\t\t},\n\t\t\t\tinitiatedAttributes.WorkflowType,\n\t\t\t\tinitiatedEventID,\n\t\t\t\tinitiatedAttributes.Header,\n\t\t\t)\n\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n}\n\nfunc recordStartChildExecutionFailed(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.StartChildExecutionTask,\n\twfContext execution.Context,\n\tinitiatedAttributes *types.StartChildWorkflowExecutionInitiatedEventAttributes,\n\tnow time.Time,\n) error {\n\n\treturn updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\tci, ok := mutableState.GetChildExecutionInfo(initiatedEventID)\n\t\t\tif !ok || ci.StartedID != constants.EmptyEventID {\n\t\t\t\treturn &types.EntityNotExistsError{Message: \"Pending child execution not found.\"}\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddStartChildWorkflowExecutionFailedEvent(initiatedEventID,\n\t\t\t\ttypes.ChildWorkflowExecutionFailedCauseWorkflowAlreadyRunning, initiatedAttributes)\n\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n}\n\n// createFirstDecisionTask is used by StartChildExecution transfer task to create the first decision task for\n// child execution.\nfunc createFirstDecisionTask(\n\tctx context.Context,\n\thistoryClient history.Client,\n\tdomainID string,\n\texecution *types.WorkflowExecution,\n) error {\n\n\tscheduleDecisionCtx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\terr := historyClient.ScheduleDecisionTask(scheduleDecisionCtx, &types.ScheduleDecisionTaskRequest{\n\t\tDomainUUID:        domainID,\n\t\tWorkflowExecution: execution,\n\t\tIsFirstDecision:   true,\n\t})\n\n\tif err != nil {\n\t\tswitch err.(type) {\n\t\t// Maybe child workflow execution already timedout or terminated\n\t\t// Safe to discard the error and complete this transfer task\n\t\t// cross cluster task need to catch entity not exist error\n\t\t// as the target domain may failover before first decision is scheduled.\n\t\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\t\treturn nil\n\t\t}\n\t}\n\n\treturn err\n}\n\nfunc requestCancelExternalExecutionCompleted(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.CancelExecutionTask,\n\twfContext execution.Context,\n\ttargetDomain string,\n\ttargetWorkflowID string,\n\ttargetRunID string,\n\tnow time.Time,\n) error {\n\n\terr := updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.WorkflowExecutionAlreadyCompletedError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\t_, ok := mutableState.GetRequestCancelInfo(initiatedEventID)\n\t\t\tif !ok {\n\t\t\t\treturn ErrMissingRequestCancelInfo\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddExternalWorkflowExecutionCancelRequested(\n\t\t\t\tinitiatedEventID,\n\t\t\t\ttargetDomain,\n\t\t\t\ttargetWorkflowID,\n\t\t\t\ttargetRunID,\n\t\t\t)\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n\n\tswitch err.(type) {\n\t// this could happen if this is a duplicate processing of the task,\n\t// or the execution has already completed.\n\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError:\n\t\treturn nil\n\t}\n\treturn err\n}\n\nfunc signalExternalExecutionCompleted(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.SignalExecutionTask,\n\twfContext execution.Context,\n\ttargetDomain string,\n\ttargetWorkflowID string,\n\ttargetRunID string,\n\tcontrol []byte,\n\tnow time.Time,\n) error {\n\n\terr := updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.WorkflowExecutionAlreadyCompletedError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\t_, ok := mutableState.GetSignalInfo(initiatedEventID)\n\t\t\tif !ok {\n\t\t\t\treturn ErrMissingSignalInfo\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddExternalWorkflowExecutionSignaled(\n\t\t\t\tinitiatedEventID,\n\t\t\t\ttargetDomain,\n\t\t\t\ttargetWorkflowID,\n\t\t\t\ttargetRunID,\n\t\t\t\tcontrol,\n\t\t\t)\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n\n\tswitch err.(type) {\n\t// this could happen if this is a duplicate processing of the task,\n\t// or the execution has already completed.\n\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError:\n\t\treturn nil\n\t}\n\n\treturn err\n}\n\nfunc requestCancelExternalExecutionFailed(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.CancelExecutionTask,\n\twfContext execution.Context,\n\ttargetDomain string,\n\ttargetWorkflowID string,\n\ttargetRunID string,\n\tnow time.Time,\n) error {\n\n\terr := updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.WorkflowExecutionAlreadyCompletedError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\t_, ok := mutableState.GetRequestCancelInfo(initiatedEventID)\n\t\t\tif !ok {\n\t\t\t\treturn ErrMissingRequestCancelInfo\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddRequestCancelExternalWorkflowExecutionFailedEvent(\n\t\t\t\tconstants.EmptyEventID,\n\t\t\t\tinitiatedEventID,\n\t\t\t\ttargetDomain,\n\t\t\t\ttargetWorkflowID,\n\t\t\t\ttargetRunID,\n\t\t\t\ttypes.CancelExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution,\n\t\t\t)\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n\n\tswitch err.(type) {\n\t// this could happen if this is a duplicate processing of the task,\n\t// or the execution has already completed.\n\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError:\n\t\treturn nil\n\t}\n\treturn err\n}\n\nfunc signalExternalExecutionFailed(\n\tctx context.Context,\n\tlogger log.Logger,\n\ttask *persistence.SignalExecutionTask,\n\twfContext execution.Context,\n\ttargetDomain string,\n\ttargetWorkflowID string,\n\ttargetRunID string,\n\tcontrol []byte,\n\tnow time.Time,\n) error {\n\n\terr := updateWorkflowExecution(ctx, logger, wfContext, true,\n\t\tfunc(ctx context.Context, mutableState execution.MutableState) error {\n\t\t\tif !mutableState.IsWorkflowExecutionRunning() {\n\t\t\t\treturn &types.WorkflowExecutionAlreadyCompletedError{Message: \"Workflow execution already completed.\"}\n\t\t\t}\n\n\t\t\tinitiatedEventID := task.InitiatedID\n\t\t\t_, ok := mutableState.GetSignalInfo(initiatedEventID)\n\t\t\tif !ok {\n\t\t\t\treturn ErrMissingSignalInfo\n\t\t\t}\n\n\t\t\t_, err := mutableState.AddSignalExternalWorkflowExecutionFailedEvent(\n\t\t\t\tconstants.EmptyEventID,\n\t\t\t\tinitiatedEventID,\n\t\t\t\ttargetDomain,\n\t\t\t\ttargetWorkflowID,\n\t\t\t\ttargetRunID,\n\t\t\t\tcontrol,\n\t\t\t\ttypes.SignalExternalWorkflowExecutionFailedCauseUnknownExternalWorkflowExecution,\n\t\t\t)\n\t\t\treturn err\n\t\t},\n\t\tnow,\n\t)\n\n\tswitch err.(type) {\n\t// this could happen if this is a duplicate processing of the task,\n\t// or the execution has already completed.\n\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError:\n\t\treturn nil\n\t}\n\n\treturn err\n}\n\nfunc updateWorkflowExecution(\n\tctx context.Context,\n\tlogger log.Logger,\n\twfContext execution.Context,\n\tcreateDecisionTask bool,\n\taction func(ctx context.Context, builder execution.MutableState) error,\n\tnow time.Time,\n) error {\n\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif err := action(ctx, mutableState); err != nil {\n\t\treturn err\n\t}\n\n\tif createDecisionTask {\n\t\t// Create a transfer task to schedule a decision task\n\t\terr := execution.ScheduleDecision(mutableState)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tlogger.Debugf(\"transferActiveTaskExecutor.updateWorkflowExecution calling UpdateWorkflowExecutionAsActive for wfID %s\",\n\t\tmutableState.GetExecutionInfo().WorkflowID,\n\t)\n\treturn wfContext.UpdateWorkflowExecutionAsActive(ctx, now)\n}\n\nfunc requestCancelExternalExecutionWithRetry(\n\tctx context.Context,\n\thistoryClient history.Client,\n\ttask *persistence.CancelExecutionTask,\n\ttargetDomain string,\n\tcancelRequestID string,\n) error {\n\n\trequest := &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: task.TargetDomainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tDomain: targetDomain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: task.TargetWorkflowID,\n\t\t\t\tRunID:      task.TargetRunID,\n\t\t\t},\n\t\t\tIdentity: execution.IdentityHistoryService,\n\t\t\t// Use the same request ID to dedupe RequestCancelWorkflowExecution calls\n\t\t\tRequestID: cancelRequestID,\n\t\t},\n\t\tExternalInitiatedEventID: common.Int64Ptr(task.InitiatedID),\n\t\tExternalWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t},\n\t\tChildWorkflowOnly: task.TargetChildWorkflowOnly,\n\t}\n\n\trequestCancelCtx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\top := func(ctx context.Context) error {\n\t\treturn historyClient.RequestCancelWorkflowExecution(ctx, request)\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\terr := throttleRetry.Do(requestCancelCtx, op)\n\tswitch err.(type) {\n\tcase *types.CancellationAlreadyRequestedError:\n\t\t// err is CancellationAlreadyRequestedError\n\t\t// this could happen if target workflow cancellation is already requested\n\t\t// mark as success\n\t\terr = nil\n\t}\n\treturn err\n}\n\nfunc signalExternalExecutionWithRetry(\n\tctx context.Context,\n\thistoryClient history.Client,\n\ttask *persistence.SignalExecutionTask,\n\ttargetDomain string,\n\tsignalInfo *persistence.SignalInfo,\n) error {\n\n\trequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: task.TargetDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain: targetDomain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: task.TargetWorkflowID,\n\t\t\t\tRunID:      task.TargetRunID,\n\t\t\t},\n\t\t\tIdentity:   execution.IdentityHistoryService,\n\t\t\tSignalName: signalInfo.SignalName,\n\t\t\tInput:      signalInfo.Input,\n\t\t\t// Use same request ID to deduplicate SignalWorkflowExecution calls\n\t\t\tRequestID: signalInfo.SignalRequestID,\n\t\t\tControl:   signalInfo.Control,\n\t\t},\n\t\tExternalWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t},\n\t\tChildWorkflowOnly: task.TargetChildWorkflowOnly,\n\t}\n\n\tsignalCtx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\top := func(ctx context.Context) error {\n\t\treturn historyClient.SignalWorkflowExecution(ctx, request)\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\treturn throttleRetry.Do(signalCtx, op)\n}\n\nfunc removeSignalMutableStateWithRetry(\n\tctx context.Context,\n\thistoryClient history.Client,\n\ttask *persistence.SignalExecutionTask,\n\tsignalRequestID string,\n) error {\n\tctx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\n\tremoveSignalRequest := &types.RemoveSignalMutableStateRequest{\n\t\tDomainUUID: task.TargetDomainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.TargetWorkflowID,\n\t\t\tRunID:      task.TargetRunID,\n\t\t},\n\t\tRequestID: signalRequestID,\n\t}\n\n\top := func(ctx context.Context) error {\n\t\treturn historyClient.RemoveSignalMutableState(ctx, removeSignalRequest)\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\tswitch err.(type) {\n\tcase *types.EntityNotExistsError:\n\t\t// it's safe to discard entity not exists error here\n\t\t// as there's nothing to remove.\n\t\treturn nil\n\t}\n\treturn err\n}\n\nfunc startWorkflowWithRetry(\n\tctx context.Context,\n\thistoryClient history.Client,\n\ttimeSource clock.TimeSource,\n\tdomainCache cache.DomainCache,\n\ttask *persistence.StartChildExecutionTask,\n\ttargetDomain string,\n\trequestID string,\n\tattributes *types.StartChildWorkflowExecutionInitiatedEventAttributes,\n\tpartitionConfig map[string]string,\n\tactiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n) (string, error) {\n\n\t// Get parent domain name\n\tdomainName, err := domainCache.GetDomainName(task.DomainID)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn \"\", err\n\t\t}\n\t\t// it is possible that the domain got deleted. Use domainID instead as this is only needed for the history event\n\t\tdomainName = task.DomainID\n\t}\n\n\tfrontendStartReq := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              targetDomain,\n\t\tWorkflowID:                          attributes.WorkflowID,\n\t\tWorkflowType:                        attributes.WorkflowType,\n\t\tTaskList:                            attributes.TaskList,\n\t\tInput:                               attributes.Input,\n\t\tHeader:                              attributes.Header,\n\t\tExecutionStartToCloseTimeoutSeconds: attributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      attributes.TaskStartToCloseTimeoutSeconds,\n\t\t// Use the same request ID to dedupe StartWorkflowExecution calls\n\t\tRequestID:                    requestID,\n\t\tWorkflowIDReusePolicy:        attributes.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                  attributes.RetryPolicy,\n\t\tCronSchedule:                 attributes.CronSchedule,\n\t\tCronOverlapPolicy:            attributes.CronOverlapPolicy,\n\t\tMemo:                         attributes.Memo,\n\t\tSearchAttributes:             attributes.SearchAttributes,\n\t\tDelayStartSeconds:            attributes.DelayStartSeconds,\n\t\tJitterStartSeconds:           attributes.JitterStartSeconds,\n\t\tFirstRunAtTimeStamp:          attributes.FirstRunAtTimestamp,\n\t\tActiveClusterSelectionPolicy: activeClusterSelectionPolicy,\n\t}\n\n\thistoryStartReq, err := common.CreateHistoryStartWorkflowRequest(task.TargetDomainID, frontendStartReq, timeSource.Now(), partitionConfig)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\thistoryStartReq.ParentExecutionInfo = &types.ParentExecutionInfo{\n\t\tDomainUUID: task.DomainID,\n\t\tDomain:     domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t},\n\t\tInitiatedID: task.InitiatedID,\n\t}\n\n\tstartWorkflowCtx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\tvar response *types.StartWorkflowExecutionResponse\n\top := func(ctx context.Context) error {\n\t\tresponse, err = historyClient.StartWorkflowExecution(ctx, historyStartReq)\n\t\treturn err\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t)\n\tif err := throttleRetry.Do(startWorkflowCtx, op); err != nil {\n\t\treturn \"\", err\n\t}\n\treturn response.GetRunID(), nil\n}\n\nfunc (t *transferActiveTaskExecutor) resetWorkflow(\n\ttask *persistence.ResetWorkflowTask,\n\tdomain string,\n\treason string,\n\tresetPoint *types.ResetPointInfo,\n\tbaseContext execution.Context,\n\tbaseMutableState execution.MutableState,\n\tcurrentContext execution.Context,\n\tcurrentMutableState execution.MutableState,\n\tlogger log.Logger,\n) error {\n\n\tvar err error\n\tresetCtx, cancel := context.WithTimeout(context.Background(), resetWorkflowTimeout)\n\tdefer cancel()\n\n\tdomainID := task.DomainID\n\tWorkflowID := task.WorkflowID\n\tbaseRunID := baseMutableState.GetExecutionInfo().RunID\n\tresetRunID := uuid.New()\n\tbaseRebuildLastEventID := resetPoint.GetFirstDecisionCompletedID() - 1\n\tbaseVersionHistories := baseMutableState.GetVersionHistories()\n\tif baseVersionHistories == nil {\n\t\treturn execution.ErrMissingVersionHistories\n\t}\n\tbaseCurrentVersionHistory, err := baseVersionHistories.GetCurrentVersionHistory()\n\tif err != nil {\n\t\treturn err\n\t}\n\tbaseRebuildLastEventVersion, err := baseCurrentVersionHistory.GetEventVersion(baseRebuildLastEventID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tbaseCurrentBranchToken := baseCurrentVersionHistory.GetBranchToken()\n\tbaseNextEventID := baseMutableState.GetNextEventID()\n\n\terr = t.workflowResetter.ResetWorkflow(\n\t\tresetCtx,\n\t\tdomainID,\n\t\tWorkflowID,\n\t\tbaseRunID,\n\t\tbaseCurrentBranchToken,\n\t\tbaseRebuildLastEventID,\n\t\tbaseRebuildLastEventVersion,\n\t\tbaseNextEventID,\n\t\tresetRunID,\n\t\tuuid.New(),\n\t\texecution.NewWorkflow(\n\t\t\tresetCtx,\n\t\t\tt.shard.GetClusterMetadata(),\n\t\t\tcurrentContext,\n\t\t\tcurrentMutableState,\n\t\t\texecution.NoopReleaseFn, // this is fine since caller will defer on release\n\t\t\tlogger,\n\t\t),\n\t\treason,\n\t\tnil,\n\t\tfalse,\n\t)\n\n\tswitch err.(type) {\n\tcase nil:\n\t\treturn nil\n\n\tcase *types.BadRequestError:\n\t\t// This means the reset point is corrupted and not retry able.\n\t\t// There must be a bug in our system that we must fix.(for example, history is not the same in active/passive)\n\t\tt.metricsClient.IncCounter(metrics.TransferQueueProcessorScope, metrics.AutoResetPointCorruptionCounter)\n\t\tlogger.Error(\"Auto-Reset workflow failed and not retryable. The reset point is corrupted.\", tag.Error(err))\n\t\treturn nil\n\n\tdefault:\n\t\t// log this error and retry\n\t\tlogger.Error(\"Auto-Reset workflow failed\", tag.Error(err))\n\t\treturn err\n\t}\n}\n\nfunc (t *transferActiveTaskExecutor) processParentClosePolicy(\n\tctx context.Context,\n\tdomainID string,\n\tdomainName string,\n\tparentExecution *types.WorkflowExecution,\n\tchildInfos map[int64]*persistence.ChildExecutionInfo,\n) error {\n\n\tif len(childInfos) == 0 {\n\t\treturn nil\n\t}\n\n\tscope := t.metricsClient.Scope(metrics.TransferActiveTaskCloseExecutionScope)\n\n\tif t.shard.GetConfig().EnableParentClosePolicyWorker() &&\n\t\tlen(childInfos) >= t.shard.GetConfig().ParentClosePolicyThreshold(domainName) {\n\n\t\tbatchSize := t.shard.GetConfig().ParentClosePolicyBatchSize(domainName)\n\t\texecutions := make([]parentclosepolicy.RequestDetail, 0, min(len(childInfos), batchSize))\n\t\tcount := 0\n\t\tfor _, childInfo := range childInfos {\n\t\t\tcount++\n\t\t\tif childInfo.ParentClosePolicy == types.ParentClosePolicyAbandon {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\texecutions = append(executions, parentclosepolicy.RequestDetail{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tDomainName: domainName,\n\t\t\t\tWorkflowID: childInfo.StartedWorkflowID,\n\t\t\t\tRunID:      childInfo.StartedRunID,\n\t\t\t\tPolicy:     childInfo.ParentClosePolicy,\n\t\t\t})\n\n\t\t\tif len(executions) == batchSize {\n\t\t\t\terr := t.parentClosePolicyClient.SendParentClosePolicyRequest(ctx, parentclosepolicy.Request{\n\t\t\t\t\tDomainName: domainName,\n\t\t\t\t\tExecutions: executions,\n\t\t\t\t})\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\texecutions = make([]parentclosepolicy.RequestDetail, 0, min(len(childInfos)-count, batchSize))\n\t\t\t}\n\t\t}\n\n\t\tif len(executions) == 0 {\n\t\t\treturn nil\n\t\t}\n\n\t\treturn t.parentClosePolicyClient.SendParentClosePolicyRequest(ctx, parentclosepolicy.Request{\n\t\t\tDomainName: domainName,\n\t\t\tExecutions: executions,\n\t\t})\n\t}\n\n\tfor _, childInfo := range childInfos {\n\t\tif err := t.applyParentClosePolicy(\n\t\t\tctx,\n\t\t\tdomainID,\n\t\t\tdomainName,\n\t\t\tparentExecution,\n\t\t\tchildInfo,\n\t\t); err != nil {\n\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.EntityNotExistsError,\n\t\t\t\t*types.WorkflowExecutionAlreadyCompletedError,\n\t\t\t\t*types.CancellationAlreadyRequestedError:\n\t\t\t\t// expected error, no-op\n\t\t\t\tbreak\n\t\t\tdefault:\n\t\t\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorFailures)\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorSuccess)\n\t}\n\treturn nil\n}\n\nfunc (t *transferActiveTaskExecutor) applyParentClosePolicy(\n\tctx context.Context,\n\tdomainID string,\n\tdomainName string,\n\tparentWorkflowExecution *types.WorkflowExecution,\n\tchildInfo *persistence.ChildExecutionInfo,\n) error {\n\n\tswitch childInfo.ParentClosePolicy {\n\tcase types.ParentClosePolicyAbandon:\n\t\t// noop\n\t\treturn nil\n\n\tcase types.ParentClosePolicyTerminate:\n\t\treturn t.historyClient.TerminateWorkflowExecution(ctx, &types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: childInfo.StartedWorkflowID,\n\t\t\t\t},\n\t\t\t\tReason:   \"by parent close policy\",\n\t\t\t\tIdentity: execution.IdentityHistoryService,\n\t\t\t\t// Include StartedRunID as FirstExecutionRunID on the request to allow child to be terminated across runs.\n\t\t\t\t// If the child does continue as new it still propagates the RunID of first execution.\n\t\t\t\tFirstExecutionRunID: childInfo.StartedRunID,\n\t\t\t},\n\t\t\tExternalWorkflowExecution: parentWorkflowExecution,\n\t\t\tChildWorkflowOnly:         true,\n\t\t})\n\n\tcase types.ParentClosePolicyRequestCancel:\n\t\treturn t.historyClient.RequestCancelWorkflowExecution(ctx, &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomain: domainName,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: childInfo.StartedWorkflowID,\n\t\t\t\t},\n\t\t\t\tIdentity: execution.IdentityHistoryService,\n\t\t\t\t// Include StartedRunID as FirstExecutionRunID on the request to allow child to be canceled across runs.\n\t\t\t\t// If the child does continue as new it still propagates the RunID of first execution.\n\t\t\t\tFirstExecutionRunID: childInfo.StartedRunID,\n\t\t\t},\n\t\t\tExternalWorkflowExecution: parentWorkflowExecution,\n\t\t\tChildWorkflowOnly:         true,\n\t\t})\n\n\tdefault:\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"unknown parent close policy: %v\", childInfo.ParentClosePolicy),\n\t\t}\n\t}\n}\n\nfunc filterPendingChildExecutions(\n\ttargetDomainIDs map[string]struct{},\n\tchildren map[int64]*persistence.ChildExecutionInfo,\n\tdomainCache cache.DomainCache,\n\tparentDomainEntry *cache.DomainCacheEntry,\n) (map[int64]*persistence.ChildExecutionInfo, error) {\n\tif len(targetDomainIDs) == 0 {\n\t\treturn children, nil\n\t}\n\n\tfilteredChildren := make(map[int64]*persistence.ChildExecutionInfo, len(children))\n\tfor initiatedID, child := range children {\n\t\tdomainID, err := execution.GetChildExecutionDomainID(child, domainCache, parentDomainEntry)\n\t\tif err != nil {\n\t\t\tif common.IsEntityNotExistsError(err) {\n\t\t\t\t// target domain deleted, ignore the child\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treturn nil, err\n\t\t}\n\t\tif _, ok := targetDomainIDs[domainID]; ok {\n\t\t\tfilteredChildren[initiatedID] = child\n\t\t}\n\t}\n\n\treturn filteredChildren, nil\n}\n\nfunc getWorkflowExecutionStatus(mutableState execution.MutableState, closeEvent *types.HistoryEvent) types.WorkflowExecutionStatus {\n\t// Extract close status from the close event\n\tvar closeStatus types.WorkflowExecutionCloseStatus\n\n\tswitch closeEvent.GetEventType() {\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusCompleted\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusFailed\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusCanceled\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusTerminated\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusTimedOut\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusContinuedAsNew\n\n\t\t// For cron workflows that continue as new:\n\t\t// - If FailureReason is present and non-empty, the workflow failed\n\t\t// - If FailureReason is nil/empty, the workflow completed successfully\n\t\tif attr := closeEvent.WorkflowExecutionContinuedAsNewEventAttributes; attr != nil {\n\t\t\tfailureReason := attr.GetFailureReason()\n\t\t\tif failureReason != \"\" {\n\t\t\t\t// Workflow failed - check if it's a timeout or generic failure\n\t\t\t\tif strings.Contains(strings.ToLower(failureReason), \"timeout\") ||\n\t\t\t\t\tstrings.Contains(strings.ToLower(failureReason), \"timed out\") {\n\t\t\t\t\tcloseStatus = types.WorkflowExecutionCloseStatusTimedOut\n\t\t\t\t} else {\n\t\t\t\t\tcloseStatus = types.WorkflowExecutionCloseStatusFailed\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\t// No failure reason means successful completion\n\t\t\t\tcloseStatus = types.WorkflowExecutionCloseStatusCompleted\n\t\t\t}\n\t\t}\n\tdefault:\n\t\t// Unknown close event, default to completed\n\t\tcloseStatus = types.WorkflowExecutionCloseStatusCompleted\n\t}\n\n\t// Map close status to execution status\n\tswitch closeStatus {\n\tcase types.WorkflowExecutionCloseStatusCompleted:\n\t\treturn types.WorkflowExecutionStatusCompleted\n\tcase types.WorkflowExecutionCloseStatusFailed:\n\t\treturn types.WorkflowExecutionStatusFailed\n\tcase types.WorkflowExecutionCloseStatusCanceled:\n\t\treturn types.WorkflowExecutionStatusCanceled\n\tcase types.WorkflowExecutionCloseStatusTerminated:\n\t\treturn types.WorkflowExecutionStatusTerminated\n\tcase types.WorkflowExecutionCloseStatusContinuedAsNew:\n\t\treturn types.WorkflowExecutionStatusContinuedAsNew\n\tcase types.WorkflowExecutionCloseStatusTimedOut:\n\t\treturn types.WorkflowExecutionStatusTimedOut\n\tdefault:\n\t\treturn types.WorkflowExecutionStatusCompleted\n\t}\n}\n"
  },
  {
    "path": "service/history/task/transfer_active_task_executor_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\thclient \"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/engine\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/reset\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n\twarchiver \"github.com/uber/cadence/service/worker/archiver\"\n\t\"github.com/uber/cadence/service/worker/parentclosepolicy\"\n)\n\ntype (\n\ttransferActiveTaskExecutorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller         *gomock.Controller\n\t\tmockShard          *shard.TestContext\n\t\tmockEngine         *engine.MockEngine\n\t\tmockDomainCache    *cache.MockDomainCache\n\t\tmockWFCache        *workflowcache.MockWFCache\n\t\tmockHistoryClient  *hclient.MockClient\n\t\tmockMatchingClient *matching.MockClient\n\n\t\tmockVisibilityMgr           *mocks.VisibilityManager\n\t\tmockExecutionMgr            *mocks.ExecutionManager\n\t\tmockHistoryV2Mgr            *mocks.HistoryV2Manager\n\t\tmockArchivalClient          *warchiver.MockClient\n\t\tmockArchivalMetadata        *archiver.MockArchivalMetadata\n\t\tmockArchiverProvider        *provider.MockArchiverProvider\n\t\tmockParentClosePolicyClient *parentclosepolicy.MockClient\n\n\t\tlogger                     log.Logger\n\t\tdomainID                   string\n\t\tdomainName                 string\n\t\tdomainEntry                *cache.DomainCacheEntry\n\t\tversion                    int64\n\t\ttimeSource                 clock.MockedTimeSource\n\t\ttransferActiveTaskExecutor *transferActiveTaskExecutor\n\t}\n)\n\nfunc TestTransferActiveTaskExecutorSuite(t *testing.T) {\n\ts := new(transferActiveTaskExecutorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) SetupSuite() {\n\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TearDownSuite() {\n\n}\n\nfunc (s *transferActiveTaskExecutorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.domainID = constants.TestDomainID\n\ts.domainName = constants.TestDomainName\n\ts.domainEntry = constants.TestGlobalDomainEntry\n\ts.version = s.domainEntry.GetFailoverVersion()\n\ts.timeSource = clock.NewMockedTimeSource()\n\n\ts.controller = gomock.NewController(s.T())\n\n\ttestConfig := config.NewForTest()\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tShardID:          0,\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ttestConfig,\n\t)\n\ts.mockShard.SetEventsCache(events.NewCache(\n\t\ts.mockShard.GetShardID(),\n\t\ts.mockShard.GetHistoryManager(),\n\t\ts.mockShard.GetConfig(),\n\t\ts.mockShard.GetLogger(),\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.mockShard.GetDomainCache(),\n\t))\n\ts.mockShard.Resource.TimeSource = s.timeSource\n\n\ts.mockEngine = engine.NewMockEngine(s.controller)\n\ts.mockEngine.EXPECT().NotifyNewHistoryEvent(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTransferTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewTimerTasks(gomock.Any()).AnyTimes()\n\ts.mockEngine.EXPECT().NotifyNewReplicationTasks(gomock.Any()).AnyTimes()\n\ts.mockShard.SetEngine(s.mockEngine)\n\n\ts.mockParentClosePolicyClient = parentclosepolicy.NewMockClient(s.controller)\n\ts.mockArchivalClient = warchiver.NewMockClient(s.controller)\n\ts.mockMatchingClient = s.mockShard.Resource.MatchingClient\n\ts.mockHistoryClient = s.mockShard.Resource.HistoryClient\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockHistoryV2Mgr = s.mockShard.Resource.HistoryMgr\n\ts.mockVisibilityMgr = s.mockShard.Resource.VisibilityMgr\n\ts.mockArchivalMetadata = s.mockShard.Resource.ArchivalMetadata\n\ts.mockArchiverProvider = s.mockShard.Resource.ArchiverProvider\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockWFCache = workflowcache.NewMockWFCache(s.controller)\n\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestRateLimitedDomainName).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(s.domainName).Return(s.domainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestRateLimitedDomainID).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainByID(s.domainID).Return(s.domainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(s.domainName).Return(s.domainID, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(s.domainID).Return(s.domainName, nil).AnyTimes()\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestRateLimitedDomainID).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestRateLimitedDomainID).Return(constants.TestRateLimitedDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestRateLimitedDomainName).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestGlobalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestGlobalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(testActiveClusterInfo, nil).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\ts.transferActiveTaskExecutor = NewTransferActiveTaskExecutor(\n\t\ts.mockShard,\n\t\ts.mockArchivalClient,\n\t\texecution.NewCache(s.mockShard),\n\t\tnil,\n\t\ts.logger,\n\t\ttestConfig,\n\t\ts.mockWFCache,\n\t).(*transferActiveTaskExecutor)\n\ts.transferActiveTaskExecutor.parentClosePolicyClient = s.mockParentClosePolicyClient\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TearDownTest() {\n\ts.transferActiveTaskExecutor.Stop()\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessActivityTask_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, ai := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\tmutableState.FlushBufferedEvents()\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID: constants.TestDomainID,\n\t\tTaskList:       mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID:     event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTask, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessActivityTask_EphemeralTaskListKind() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, ai, _, _, _, _ := mutableState.AddActivityTaskScheduledEvent(nil, decisionCompletionID, &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    \"activity-1\",\n\t\tActivityType:                  &types.ActivityType{Name: \"some random activity type\"},\n\t\tTaskList:                      &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList, Kind: types.TaskListKindEphemeral.Ptr()},\n\t\tInput:                         []byte{},\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(1),\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t}, false,\n\t)\n\tmutableState.FlushBufferedEvents()\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID: constants.TestDomainID,\n\t\tTaskList:       \"ignored\",\n\t\tScheduleID:     event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTask, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessActivityTask_Ratelimits() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, constants.TestDomainID)\n\ts.NoError(err)\n\n\tevent, ai := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\tmutableState.FlushBufferedEvents()\n\n\ttransferTaskInRatelimitedDomain := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   constants.TestRateLimitedDomainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID: constants.TestRateLimitedDomainID,\n\t\tTaskList:       mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID:     event.ID,\n\t})\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID: constants.TestDomainID,\n\t\tTaskList:       mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID:     event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t// expected calls to the matching service if task processing is allowed\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTaskInRatelimitedDomain, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTask, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\t// RPS still below allowed value so the task can be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestRateLimitedDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTaskInRatelimitedDomain)\n\ts.Nil(err)\n\n\t// RPS more than allowed limit so the task cannot be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestRateLimitedDomainID, constants.TestWorkflowID).Return(false).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTaskInRatelimitedDomain)\n\ts.Error(err)\n\ts.Equal(\"workflow is being rate limited for making too many requests\", err.Error())\n\n\t// RPS still below allowed value so the task can be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n\n\t// RPS more than allowed limit so the task cannot be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(false).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Error(err)\n\ts.Equal(\"workflow is being rate limited for making too many requests\", err.Error())\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessActivityTask_Duplication() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, ai := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\n\tevent = test.AddActivityTaskStartedEvent(mutableState, event.ID, \"\")\n\tai.StartedID = event.ID\n\tevent = test.AddActivityTaskCompletedEvent(mutableState, ai.ScheduleID, ai.StartedID, nil, \"\")\n\tmutableState.FlushBufferedEvents()\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID: constants.TestDomainID,\n\t\tTaskList:       mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID:     event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_FirstDecision() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_NonFirstDecision() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// make another round of decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_Ratelimits() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// make another round of decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\trateLimitedTransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   constants.TestRateLimitedDomainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t// expected calls to the matching service if task processing is allowed\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(rateLimitedTransferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t// RPS still below allowed value so the task can be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestRateLimitedDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(rateLimitedTransferTask)\n\ts.Nil(err)\n\n\t// RPS more than allowed limit so the task cannot be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestRateLimitedDomainID, constants.TestWorkflowID).Return(false).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(rateLimitedTransferTask)\n\ts.Error(err)\n\ts.Equal(\"workflow is being rate limited for making too many requests\", err.Error())\n\n\t// RPS still below allowed value so the task can be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n\n\t// RPS more than allowed limit so the task cannot be executed\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(false).Times(1)\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Error(err)\n\ts.Equal(\"workflow is being rate limited for making too many requests\", err.Error())\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_Sticky_NonFirstDecision() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// set the sticky tasklist attr\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.StickyTaskList = \"some random sticky task list\"\n\texecutionInfo.StickyScheduleToStartTimeout = int32(233)\n\n\t// make another round of decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   executionInfo.StickyTaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_DecisionNotSticky_MutableStateSticky() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// set the sticky tasklist attr\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.StickyTaskList = \"some random sticky task list\"\n\texecutionInfo.StickyScheduleToStartTimeout = int32(233)\n\n\t// make another round of decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_Duplication() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(4096),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: mutableState.GetPreviousStartedEventID() - 1,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, mutableState.GetNextEventID()-1, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_StickyWorkerUnavailable() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\t// set the sticky tasklist attr\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.StickyTaskList = \"some random sticky task list\"\n\texecutionInfo.StickyScheduleToStartTimeout = int32(233)\n\n\t// make another round of decision\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   executionInfo.StickyTaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\n\taddDecisionTaskRequest := createAddDecisionTaskRequest(transferTask, mutableState)\n\n\t// Create a deep copy of the expected modified request\n\tmodifiedRequest := *addDecisionTaskRequest\n\tmodifiedRequest.TaskList = &types.TaskList{\n\t\tName: mutableState.GetExecutionInfo().TaskList,\n\t}\n\n\tgomock.InOrder(\n\t\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), addDecisionTaskRequest).Return(nil, &types.StickyWorkerUnavailableError{}).Times(1),\n\t\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), gomock.Eq(&modifiedRequest)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1),\n\t)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.NoError(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessDecisionTask_EphemeralTaskListKind() {\n\tworkflowExecution, mutableState, err := test.StartWorkflowWithTaskList(&types.TaskList{Name: \"ephemmy\", Kind: types.TaskListKindEphemeral.Ptr()}, s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockWFCache.EXPECT().AllowInternal(constants.TestDomainID, constants.TestWorkflowID).Return(true).Times(1)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_HasParent_Success() {\n\ts.testProcessCloseExecutionWithParent(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t) {\n\t\t\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\t\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\t\t},\n\t\tfalse,\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_HasParent_Failure() {\n\ts.testProcessCloseExecutionWithParent(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t) {\n\t\t\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\t\t\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\t\t},\n\t\ttrue,\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessCloseExecutionWithParent(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t),\n\tfailRecordChild bool,\n) {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tparentInitiatedID := int64(3222)\n\tparentExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random parent workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\texecutionInfo.ParentDomainID = targetDomainID\n\texecutionInfo.ParentWorkflowID = parentExecution.WorkflowID\n\texecutionInfo.ParentRunID = parentExecution.RunID\n\texecutionInfo.InitiatedID = parentInitiatedID\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tif targetDomainID == constants.TestDomainID {\n\t\tvar recordChildErr error\n\t\tif failRecordChild {\n\t\t\trecordChildErr = &types.DomainNotActiveError{}\n\t\t}\n\t\ts.mockHistoryClient.EXPECT().RecordChildExecutionCompleted(gomock.Any(), &types.RecordChildExecutionCompletedRequest{\n\t\t\tDomainUUID:         targetDomainID,\n\t\t\tWorkflowExecution:  &parentExecution,\n\t\t\tInitiatedID:        parentInitiatedID,\n\t\t\tCompletedExecution: &workflowExecution,\n\t\t\tCompletionEvent:    event,\n\t\t}).Return(recordChildErr).Times(1)\n\t}\n\n\tsetupMockFn(mutableState, workflowExecution, parentExecution)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\tif failRecordChild {\n\t\ts.Equal(&types.DomainNotActiveError{}, err)\n\t} else {\n\t\ts.NoError(err)\n\t}\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_NoParent() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.ActiveClusterSelectionPolicy = &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewArchivalConfig(\"enabled\", dynamicproperties.GetStringPropertyFn(\"enabled\"), true, dynamicproperties.GetBoolPropertyFn(true), \"disabled\", \"random URI\"))\n\ts.mockArchivalClient.EXPECT().Archive(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t// switch on context header in viz\n\ts.mockShard.GetConfig().EnableContextHeaderInVisibility = func(domain string) bool {\n\t\treturn true\n\t}\n\ts.mockShard.GetConfig().ValidSearchAttributes = func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\treturn map[string]interface{}{\n\t\t\t\"Header_context_key\": struct{}{},\n\t\t}\n\t}\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_NoParent_HasFewChildren() {\n\ts.testProcessCloseExecutionNoParentHasFewChildren(\n\t\tmap[string]string{\n\t\t\t\"child_abandon\":   s.domainName,\n\t\t\t\"child_terminate\": s.domainName,\n\t\t\t\"child_cancel\":    s.domainName,\n\t\t},\n\t\tfunc() {\n\t\t\ts.expectCancelRequest(s.domainName)\n\t\t\ts.expectTerminateRequest(s.domainName)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestApplyParentPolicy_SameClusterChild_TargetNotActive() {\n\ts.testProcessCloseExecutionNoParentHasFewChildrenWithError(\n\t\tmap[string]string{\n\t\t\t\"child_terminate\": s.domainName,\n\t\t\t\"child_cancel\":    s.domainName,\n\t\t},\n\t\tfunc() {\n\t\t\ts.mockHistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(&types.DomainNotActiveError{}).MaxTimes(1)\n\t\t\ts.mockHistoryClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(&types.DomainNotActiveError{}).MaxTimes(1)\n\t\t},\n\t\t&types.DomainNotActiveError{},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) expectCancelRequest(childDomainName string) {\n\tchildDomainID, err := s.mockDomainCache.GetDomainID(childDomainName)\n\ts.NoError(err)\n\ts.mockHistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(\n\t\t\t_ context.Context,\n\t\t\trequest *types.HistoryRequestCancelWorkflowExecutionRequest,\n\t\t\toption ...yarpc.CallOption,\n\t\t) error {\n\t\t\ts.Equal(childDomainID, request.DomainUUID)\n\t\t\ts.Equal(childDomainName, request.CancelRequest.Domain)\n\t\t\ts.True(request.GetChildWorkflowOnly())\n\t\t\terrs := []error{nil, &types.CancellationAlreadyRequestedError{}, &types.EntityNotExistsError{}}\n\t\t\treturn errs[rand.Intn(len(errs))]\n\t\t},\n\t).Times(1)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) expectTerminateRequest(childDomainName string) {\n\tchildDomainID, err := s.mockDomainCache.GetDomainID(childDomainName)\n\ts.NoError(err)\n\ts.mockHistoryClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(\n\t\t\t_ context.Context,\n\t\t\trequest *types.HistoryTerminateWorkflowExecutionRequest,\n\t\t\toption ...yarpc.CallOption,\n\t\t) error {\n\t\t\ts.Equal(childDomainID, request.DomainUUID)\n\t\t\ts.Equal(childDomainName, request.TerminateRequest.Domain)\n\t\t\terrs := []error{nil, &types.EntityNotExistsError{}}\n\t\t\treturn errs[rand.Intn(len(errs))]\n\t\t},\n\t).Times(1)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessCloseExecutionNoParentHasFewChildren(\n\tchildrenDomainNames map[string]string,\n\tsetupMockFn func(),\n) {\n\ts.testProcessCloseExecutionNoParentHasFewChildrenWithError(childrenDomainNames, setupMockFn, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessCloseExecutionNoParentHasFewChildrenWithError(\n\tchildrenDomainNames map[string]string,\n\tsetupMockFn func(),\n\texpectedErr error,\n) {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tparentClosePolicy1 := types.ParentClosePolicyAbandon\n\tparentClosePolicy2 := types.ParentClosePolicyTerminate\n\tparentClosePolicy3 := types.ParentClosePolicyRequestCancel\n\n\t_, _, err = mutableState.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletionID, uuid.New(), &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:     childrenDomainNames[\"child_abandon\"],\n\t\tWorkflowID: \"child workflow1\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"child workflow type\",\n\t\t},\n\t\tTaskList:          &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList},\n\t\tInput:             []byte(\"random input\"),\n\t\tParentClosePolicy: &parentClosePolicy1,\n\t})\n\ts.Nil(err)\n\t_, _, err = mutableState.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletionID, uuid.New(), &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:     childrenDomainNames[\"child_terminate\"],\n\t\tWorkflowID: \"child workflow2\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"child workflow type\",\n\t\t},\n\t\tTaskList:          &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList},\n\t\tInput:             []byte(\"random input\"),\n\t\tParentClosePolicy: &parentClosePolicy2,\n\t})\n\ts.Nil(err)\n\t_, _, err = mutableState.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletionID, uuid.New(), &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\tDomain:     childrenDomainNames[\"child_cancel\"],\n\t\tWorkflowID: \"child workflow3\",\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: \"child workflow type\",\n\t\t},\n\t\tTaskList:          &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList},\n\t\tInput:             []byte(\"random input\"),\n\t\tParentClosePolicy: &parentClosePolicy3,\n\t})\n\ts.Nil(err)\n\ts.NoError(mutableState.FlushBufferedEvents())\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\tsetupMockFn()\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_NoParent_HasManyChildren() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tnumChildWorkflows := 500\n\tfor i := 0; i < numChildWorkflows; i++ {\n\t\t_, _, err = mutableState.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletionID, uuid.New(), &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:     s.domainName,\n\t\t\tWorkflowID: \"child workflow\" + strconv.Itoa(i),\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"child workflow type\",\n\t\t\t},\n\t\t\tTaskList:          &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList},\n\t\t\tInput:             []byte(\"random input\"),\n\t\t\tParentClosePolicy: types.ParentClosePolicyTerminate.Ptr(),\n\t\t})\n\t\ts.Nil(err)\n\t}\n\n\ts.NoError(mutableState.FlushBufferedEvents())\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\ts.mockParentClosePolicyClient.EXPECT().SendParentClosePolicyRequest(gomock.Any(), gomock.Cond(\n\t\tfunc(request parentclosepolicy.Request) bool {\n\t\t\tif len(request.Executions) != s.mockShard.GetConfig().ParentClosePolicyBatchSize(constants.TestDomainName) {\n\t\t\t\treturn false\n\t\t\t}\n\t\t\tfor _, executions := range request.Executions {\n\t\t\t\tif executions.DomainName != constants.TestDomainName {\n\t\t\t\t\treturn false\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn true\n\t\t},\n\t)).Return(nil).Times(numChildWorkflows / s.mockShard.GetConfig().ParentClosePolicyBatchSize(constants.TestDomainName))\n\ts.mockParentClosePolicyClient.EXPECT().SendParentClosePolicyRequest(gomock.Any(), gomock.Cond(\n\t\tfunc(request parentclosepolicy.Request) bool {\n\t\t\tif len(request.Executions) != numChildWorkflows%s.mockShard.GetConfig().ParentClosePolicyBatchSize(constants.TestDomainName) {\n\t\t\t\treturn false\n\t\t\t}\n\t\t\tfor _, executions := range request.Executions {\n\t\t\t\tif executions.DomainName != constants.TestDomainName {\n\t\t\t\t\treturn false\n\t\t\t\t}\n\t\t\t}\n\t\t\treturn true\n\t\t},\n\t)).Return(nil).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCloseExecution_NoParent_HasManyAbandonedChildren() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tfor i := 0; i < 10; i++ {\n\t\t_, _, err = mutableState.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletionID, uuid.New(), &types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tWorkflowID: \"child workflow\" + strconv.Itoa(i),\n\t\t\tWorkflowType: &types.WorkflowType{\n\t\t\t\tName: \"child workflow type\",\n\t\t\t},\n\t\t\tTaskList:          &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList},\n\t\t\tInput:             []byte(\"random input\"),\n\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon.Ptr(),\n\t\t})\n\t\ts.Nil(err)\n\t}\n\n\ts.NoError(mutableState.FlushBufferedEvents())\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCancelExecution_Success() {\n\ts.testProcessCancelExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\tcancelRequest := createTestRequestCancelWorkflowExecutionRequest(constants.TestDomainName, transferTask.GetInfo().(*persistence.CancelExecutionTask), requestCancelInfo.CancelRequestID)\n\t\t\ts.mockHistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), cancelRequest).Return(nil).Times(1)\n\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCancelExecution_Failure() {\n\ts.testProcessCancelExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\tcancelRequest := createTestRequestCancelWorkflowExecutionRequest(constants.TestDomainName, transferTask.GetInfo().(*persistence.CancelExecutionTask), requestCancelInfo.CancelRequestID)\n\t\t\ts.mockHistoryClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), cancelRequest).Return(&types.EntityNotExistsError{}).Times(1)\n\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCancelExecution_Duplication() {\n\ts.testProcessCancelExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t\t) {\n\t\t\tevent = test.AddCancelRequestedEvent(mutableState, event.ID, constants.TestDomainID, targetExecution.GetWorkflowID(), targetExecution.GetRunID())\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessCancelExecution(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tcancelInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t),\n) {\n\ts.testProcessCancelExecutionWithError(targetDomainID, setupMockFn, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessCancelExecutionWithError(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tcancelInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t),\n\texpectedErr error,\n) {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, rci := test.AddRequestCancelInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CancelExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID:   targetDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tsetupMockFn(mutableState, workflowExecution, targetExecution, event, transferTask, rci)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessCancelExecution_WorkflowCancellingItself() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, rci := test.AddRequestCancelInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\tworkflowExecution.GetWorkflowID(),\n\t\tworkflowExecution.GetRunID(),\n\t)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CancelExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID:   s.domainID,\n\t\tTargetWorkflowID: workflowExecution.GetWorkflowID(),\n\t\tTargetRunID:      workflowExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tsetupMockFn := func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tevent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\trequestCancelInfo *persistence.RequestCancelInfo,\n\t) {\n\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\ts.NoError(err)\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t}\n\tsetupMockFn(mutableState, workflowExecution, workflowExecution, event, transferTask, rci)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.NoError(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessSignalExecution_Success() {\n\ts.testProcessSignalExecution(\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tsignalInfo *persistence.SignalInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\tsignalRequest := createTestSignalWorkflowExecutionRequest(constants.TestDomainName, transferTask.GetInfo().(*persistence.SignalExecutionTask), signalInfo)\n\t\t\ts.mockHistoryClient.EXPECT().SignalWorkflowExecution(gomock.Any(), signalRequest).Return(nil).Times(1)\n\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\n\t\t\ttaskInfo := transferTask.GetInfo().(*persistence.SignalExecutionTask)\n\t\t\ts.mockHistoryClient.EXPECT().RemoveSignalMutableState(gomock.Any(), &types.RemoveSignalMutableStateRequest{\n\t\t\t\tDomainUUID: taskInfo.TargetDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: taskInfo.TargetWorkflowID,\n\t\t\t\t\tRunID:      taskInfo.TargetRunID,\n\t\t\t\t},\n\t\t\t\tRequestID: signalInfo.SignalRequestID,\n\t\t\t}).Return(nil).Times(1)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessSignalExecution_Failure() {\n\tfor name, c := range map[string]struct {\n\t\tErr          error\n\t\tExpectedLogs []string\n\t}{\n\t\t\"GenericErr\": {\n\t\t\tErr:          assert.AnError,\n\t\t\tExpectedLogs: []string{\"Failed to signal external workflow execution\"},\n\t\t},\n\t\t\"NotExistsErr\": {\n\t\t\tErr:          &types.EntityNotExistsError{},\n\t\t\tExpectedLogs: nil,\n\t\t},\n\t\t\"WorkflowExecutionAlreadyCompleted\": {\n\t\t\tErr:          &types.WorkflowExecutionAlreadyCompletedError{},\n\t\t\tExpectedLogs: nil,\n\t\t},\n\t} {\n\t\ts.Run(name, func() {\n\t\t\t// Need set up the suite manually, since we are in a subtest\n\t\t\ts.SetupTest()\n\t\t\ts.testProcessSignalExecutionWithErrorAndLogs(\n\t\t\t\tconstants.TestDomainID,\n\t\t\t\tfunc(\n\t\t\t\t\tmutableState execution.MutableState,\n\t\t\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\t\t\tevent *types.HistoryEvent,\n\t\t\t\t\ttransferTask Task,\n\t\t\t\t\tsignalInfo *persistence.SignalInfo,\n\t\t\t\t) {\n\t\t\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\t\t\ts.NoError(err)\n\t\t\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\t\t\tsignalRequest := createTestSignalWorkflowExecutionRequest(constants.TestDomainName, transferTask.GetInfo().(*persistence.SignalExecutionTask), signalInfo)\n\t\t\t\t\ts.mockHistoryClient.EXPECT().SignalWorkflowExecution(gomock.Any(), signalRequest).Return(c.Err).Times(1)\n\t\t\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t\t\t},\n\t\t\t\tnil,\n\t\t\t\tc.ExpectedLogs,\n\t\t\t)\n\t\t})\n\t}\n\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessSignalExecution_Duplication() {\n\ts.testProcessSignalExecution(\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tsignalInfo *persistence.SignalInfo,\n\t\t) {\n\t\t\tevent = test.AddSignaledEvent(mutableState, event.ID, constants.TestDomainName, targetExecution.GetWorkflowID(), targetExecution.GetRunID(), nil)\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessSignalExecution_WorkflowSignalingItself() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, signalInfo := test.AddRequestSignalInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\tworkflowExecution.GetWorkflowID(),\n\t\tworkflowExecution.GetRunID(),\n\t\t\"some random signal name\",\n\t\t[]byte(\"some random signal input\"),\n\t\t[]byte(\"some random signal control\"),\n\t)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.SignalExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID:   s.domainID,\n\t\tTargetWorkflowID: workflowExecution.GetWorkflowID(),\n\t\tTargetRunID:      workflowExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\t// Make sure we can observe the logs\n\tobservedZapCore, _ := observer.New(zap.InfoLevel)\n\ts.transferActiveTaskExecutor.logger = log.NewLogger(zap.New(observedZapCore))\n\n\tsetupMockFn := func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tevent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\tsignalInfo *persistence.SignalInfo,\n\t) {\n\t\tmutableState.FlushBufferedEvents()\n\n\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\ts.NoError(err)\n\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t}\n\n\tsetupMockFn(mutableState, workflowExecution, workflowExecution, event, transferTask, signalInfo)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.NoError(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessSignalExecution(\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tsignalInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\tsignalInfo *persistence.SignalInfo,\n\t),\n) {\n\ts.testProcessSignalExecutionWithErrorAndLogs(constants.TestDomainID, setupMockFn, nil, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessSignalExecutionWithErrorAndLogs(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tsignalInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\tsignalInfo *persistence.SignalInfo,\n\t),\n\texpectedErr error,\n\texpectedLogs []string,\n) {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, signalInfo := test.AddRequestSignalInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t\t\"some random signal name\",\n\t\t[]byte(\"some random signal input\"),\n\t\t[]byte(\"some random signal control\"),\n\t)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.SignalExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID:   targetDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\t// Make sure we can observe the logs\n\tobservedZapCore, observedLogs := observer.New(zap.InfoLevel)\n\ts.transferActiveTaskExecutor.logger = log.NewLogger(zap.New(observedZapCore))\n\n\tsetupMockFn(mutableState, workflowExecution, targetExecution, event, transferTask, signalInfo)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Equal(expectedErr, err)\n\n\tassert.Equal(s.T(), len(expectedLogs), observedLogs.Len(), \"expected %v logs, but got %v logs\", len(expectedLogs), observedLogs.Len())\n\tfor _, expectedLog := range expectedLogs {\n\t\ts.True(observedLogs.FilterMessage(expectedLog).Len() > 0, \"expected log missing: %v\", expectedLog)\n\t}\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_Success() {\n\ts.testProcessStartChildExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ttaskInfo := transferTask.GetInfo().(*persistence.StartChildExecutionTask)\n\t\t\tevent, err = mutableState.GetChildExecutionInitiatedEvent(context.Background(), taskInfo.InitiatedID)\n\t\t\ts.NoError(err)\n\t\t\thistoryReq, err := createTestChildWorkflowExecutionRequest(\n\t\t\t\ts.domainName,\n\t\t\t\tconstants.TestDomainName,\n\t\t\t\ttaskInfo,\n\t\t\t\tevent.StartChildWorkflowExecutionInitiatedEventAttributes,\n\t\t\t\tchildInfo.CreateRequestID,\n\t\t\t\ts.mockShard.GetTimeSource().Now(),\n\t\t\t\tmutableState.GetExecutionInfo().PartitionConfig,\n\t\t\t\tmutableState.GetExecutionInfo().ActiveClusterSelectionPolicy,\n\t\t\t)\n\t\t\trequire.NoError(s.T(), err)\n\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), historyReq).Return(&types.StartWorkflowExecutionResponse{RunID: childExecution.RunID}, nil).Times(1)\n\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t\ts.mockHistoryClient.EXPECT().ScheduleDecisionTask(gomock.Any(), &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: childExecution.WorkflowID,\n\t\t\t\t\tRunID:      childExecution.RunID,\n\t\t\t\t},\n\t\t\t\tIsFirstDecision: true,\n\t\t\t}).Return(nil).Times(1)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_Failure() {\n\ts.testProcessStartChildExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ttaskInfo := transferTask.GetInfo().(*persistence.StartChildExecutionTask)\n\t\t\tevent, err = mutableState.GetChildExecutionInitiatedEvent(context.Background(), taskInfo.InitiatedID)\n\t\t\ts.NoError(err)\n\t\t\thistoryReq, err := createTestChildWorkflowExecutionRequest(\n\t\t\t\ts.domainName,\n\t\t\t\tconstants.TestDomainName,\n\t\t\t\ttaskInfo,\n\t\t\t\tevent.StartChildWorkflowExecutionInitiatedEventAttributes,\n\t\t\t\tchildInfo.CreateRequestID,\n\t\t\t\ts.mockShard.GetTimeSource().Now(),\n\t\t\t\tmutableState.GetExecutionInfo().PartitionConfig,\n\t\t\t\tmutableState.GetExecutionInfo().ActiveClusterSelectionPolicy,\n\t\t\t)\n\t\t\trequire.NoError(s.T(), err)\n\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), historyReq).Return(nil, &types.WorkflowExecutionAlreadyStartedError{}).Times(1)\n\t\t\ts.mockHistoryV2Mgr.On(\"AppendHistoryNodes\", mock.Anything, mock.Anything).Return(&persistence.AppendHistoryNodesResponse{}, nil).Once()\n\t\t\ts.mockExecutionMgr.On(\"UpdateWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.UpdateWorkflowExecutionResponse{MutableStateUpdateSessionStats: &persistence.MutableStateUpdateSessionStats{}}, nil).Once()\n\t\t},\n\t)\n}\n\n// This test was originally written for the Cross-cluster use-case where the target domain is not active.\n// However, it remains a valid test for the scenario where there's a race between parent and child in transfer\n// tasks.\n// ie: When a failover has occurred, and the parent workflow has spawned a child, this has been picked up by a\n// host which has not yet updated it's domain-cache to include the new information that the domain has failed over\n// and incorrectly thinks that the domain is not active.\n// In this case the correct behaviour would be to return the domainNotActive error and ensure that it's retried on the\n// host running the child workflow.\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_TargetNotActive() {\n\ts.testProcessStartChildExecutionWithError(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ttaskInfo := transferTask.GetInfo().(*persistence.StartChildExecutionTask)\n\t\t\t_, err = mutableState.GetChildExecutionInitiatedEvent(context.Background(), taskInfo.InitiatedID)\n\t\t\ts.NoError(err)\n\t\t\ts.mockHistoryClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.DomainNotActiveError{\n\t\t\t\tMessage: \"domain not active error 123\",\n\t\t\t}).Times(1)\n\t\t},\n\t\t&types.DomainNotActiveError{\n\t\t\tMessage: \"domain not active error 123\",\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_Success_Dup() {\n\ts.testProcessStartChildExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tstartEvent := test.AddChildWorkflowExecutionStartedEvent(mutableState, event.ID, constants.TestDomainID, childExecution.WorkflowID, childExecution.RunID, childInfo.WorkflowTypeName)\n\t\t\tchildInfo.StartedID = startEvent.ID\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ts.mockHistoryClient.EXPECT().ScheduleDecisionTask(gomock.Any(), &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: childExecution.WorkflowID,\n\t\t\t\t\tRunID:      childExecution.RunID,\n\t\t\t\t},\n\t\t\t\tIsFirstDecision: true,\n\t\t\t}).Return(nil).Times(1)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_Dup_TargetNotActive() {\n\ts.testProcessStartChildExecutionWithError(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tstartEvent := test.AddChildWorkflowExecutionStartedEvent(mutableState, event.ID, constants.TestDomainID, childExecution.WorkflowID, childExecution.RunID, childInfo.WorkflowTypeName)\n\t\t\tchildInfo.StartedID = startEvent.ID\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ts.mockHistoryClient.EXPECT().ScheduleDecisionTask(gomock.Any(), gomock.Any()).Return(&types.DomainNotActiveError{}).Times(1)\n\t\t},\n\t\t&types.DomainNotActiveError{},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_Duplication() {\n\ts.testProcessStartChildExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tstartEvent := test.AddChildWorkflowExecutionStartedEvent(mutableState, event.ID, constants.TestDomainID, childExecution.GetWorkflowID(), childExecution.GetRunID(), childInfo.WorkflowTypeName)\n\t\t\tchildInfo.StartedID = startEvent.ID\n\t\t\tstartEvent = test.AddChildWorkflowExecutionCompletedEvent(mutableState, childInfo.InitiatedID, &childExecution, &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\tResult:                       []byte(\"some random child workflow execution result\"),\n\t\t\t\tDecisionTaskCompletedEventID: transferTask.GetInfo().(*persistence.StartChildExecutionTask).InitiatedID,\n\t\t\t})\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessStartChildExecution_StartedAbandonChild_ParentClosed() {\n\ts.testProcessStartChildExecution(\n\t\tconstants.TestDomainID,\n\t\tfunc(\n\t\t\tmutableState execution.MutableState,\n\t\t\tworkflowExecution, childExecution types.WorkflowExecution,\n\t\t\tevent *types.HistoryEvent,\n\t\t\ttransferTask Task,\n\t\t\tchildInfo *persistence.ChildExecutionInfo,\n\t\t) {\n\t\t\tevent = test.AddChildWorkflowExecutionStartedEvent(mutableState, event.ID, constants.TestDomainID, childExecution.WorkflowID, childExecution.RunID, childInfo.WorkflowTypeName)\n\t\t\tchildInfo.StartedID = event.ID\n\t\t\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\t\t\tevent = test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, \"some random identity\")\n\t\t\tevent = test.AddDecisionTaskCompletedEvent(mutableState, di.ScheduleID, event.ID, nil, \"some random identity\")\n\t\t\tevent = test.AddCompleteWorkflowEvent(mutableState, event.ID, nil)\n\t\t\tmutableState.FlushBufferedEvents()\n\n\t\t\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\t\t\ts.NoError(err)\n\t\t\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\t\t\ts.mockHistoryClient.EXPECT().ScheduleDecisionTask(gomock.Any(), &types.ScheduleDecisionTaskRequest{\n\t\t\t\tDomainUUID: constants.TestDomainID,\n\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: childExecution.WorkflowID,\n\t\t\t\t\tRunID:      childExecution.RunID,\n\t\t\t\t},\n\t\t\t\tIsFirstDecision: true,\n\t\t\t}).Return(nil).Times(1)\n\t\t},\n\t)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessStartChildExecution(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tchildInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\tchildInfo *persistence.ChildExecutionInfo,\n\t),\n) {\n\ts.testProcessStartChildExecutionWithError(targetDomainID, setupMockFn, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessStartChildExecutionWithError(\n\ttargetDomainID string,\n\tsetupMockFn func(\n\t\tmutableState execution.MutableState,\n\t\tworkflowExecution, targetExecution types.WorkflowExecution,\n\t\tchildInitEvent *types.HistoryEvent,\n\t\ttransferTask Task,\n\t\tchildInfo *persistence.ChildExecutionInfo,\n\t),\n\texpectedErr error,\n) {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\tchildExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random child workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, ci := test.AddStartChildWorkflowExecutionInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\tchildExecution.WorkflowID,\n\t\t\"some random child workflow type\",\n\t\t\"some random child task list\",\n\t\tnil,\n\t\t1,\n\t\t1,\n\t\t&types.RetryPolicy{\n\t\t\tExpirationIntervalInSeconds: 100,\n\t\t\tMaximumAttempts:             3,\n\t\t\tInitialIntervalInSeconds:    1,\n\t\t\tMaximumIntervalInSeconds:    2,\n\t\t\tBackoffCoefficient:          1,\n\t\t},\n\t)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.StartChildExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t\tTargetDomainID:   targetDomainID,\n\t\tTargetWorkflowID: childExecution.WorkflowID,\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tsetupMockFn(mutableState, workflowExecution, childExecution, event, transferTask, ci)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Equal(expectedErr, err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessRecordWorkflowStartedTask() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.CronSchedule = \"@every 5s\"\n\texecutionInfo.ActiveClusterSelectionPolicy = &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\tstartEvent.WorkflowExecutionStartedEventAttributes.FirstDecisionTaskBackoffSeconds = common.Int32Ptr(5)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tif s.mockShard.GetConfig().EnableRecordWorkflowExecutionUninitialized(s.domainName) {\n\t\ts.mockVisibilityMgr.On(\n\t\t\t\"RecordWorkflowExecutionUninitialized\",\n\t\t\tmock.Anything,\n\t\t\tcreateRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState, s.mockShard.GetTimeSource().Now(), 1234),\n\t\t).Once().Return(nil)\n\t}\n\ts.mockVisibilityMgr.On(\n\t\t\"RecordWorkflowExecutionStarted\",\n\t\tmock.Anything,\n\t\tcreateRecordWorkflowExecutionStartedRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\tfalse),\n\t).Once().Return(nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessRecordWorkflowStartedTaskWithContextHeader() {\n\t// switch on context header in viz\n\ts.mockShard.GetConfig().EnableContextHeaderInVisibility = func(domain string) bool { return true }\n\ts.mockShard.GetConfig().ValidSearchAttributes = func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\treturn map[string]interface{}{\n\t\t\t\"Header_context_key\": struct{}{},\n\t\t\t\"123456\":             struct{}{}, // unsanitizable key\n\t\t}\n\t}\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.CronSchedule = \"@every 5s\"\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\tstartEvent.WorkflowExecutionStartedEventAttributes.FirstDecisionTaskBackoffSeconds = common.Int32Ptr(5)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tif s.mockShard.GetConfig().EnableRecordWorkflowExecutionUninitialized(s.domainName) {\n\t\ts.mockVisibilityMgr.On(\n\t\t\t\"RecordWorkflowExecutionUninitialized\",\n\t\t\tmock.Anything,\n\t\t\tcreateRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState, s.mockShard.GetTimeSource().Now(), 1234),\n\t\t).Once().Return(nil)\n\t}\n\ts.mockVisibilityMgr.On(\n\t\t\"RecordWorkflowExecutionStarted\",\n\t\tmock.Anything,\n\t\tcreateRecordWorkflowExecutionStartedRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\ttrue),\n\t).Once().Return(nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessUpsertWorkflowSearchAttributes() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.ActiveClusterSelectionPolicy = &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\ts.mockVisibilityMgr.On(\n\t\t\"UpsertWorkflowExecution\",\n\t\tmock.Anything,\n\t\tcreateUpsertWorkflowSearchAttributesRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\tfalse),\n\t).Once().Return(nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessUpsertWorkflowSearchAttributesWithContextHeader() {\n\t// switch on context header in viz\n\ts.mockShard.GetConfig().EnableContextHeaderInVisibility = func(domain string) bool { return true }\n\ts.mockShard.GetConfig().ValidSearchAttributes = func(opts ...dynamicproperties.FilterOption) map[string]interface{} {\n\t\treturn map[string]interface{}{\n\t\t\t\"Header_context_key\": struct{}{},\n\t\t}\n\t}\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\ts.mockVisibilityMgr.On(\n\t\t\"UpsertWorkflowExecution\",\n\t\tmock.Anything,\n\t\tcreateUpsertWorkflowSearchAttributesRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\ttrue),\n\t).Once().Return(nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_ResetPointNil() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ResetWorkflowTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_WorkflowNotRunning() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ResetWorkflowTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tmutableState.GetExecutionInfo().State = persistence.WorkflowStateCompleted\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, &persistence.GetCurrentExecutionRequest{DomainID: s.domainID, WorkflowID: workflowExecution.GetWorkflowID(), DomainName: s.domainName}).\n\t\tReturn(&persistence.GetCurrentExecutionResponse{RunID: \"runID\"}, nil)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_Success() {\n\ts.testProcessResetWorkflowWithError(true, nil, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_DifferentRunID() {\n\ts.testProcessResetWorkflowWithError(false, nil, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_CorruptedResetPoint() {\n\ts.testProcessResetWorkflowWithError(true, &types.BadRequestError{}, nil)\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestProcessResetWorkflow_OtherError() {\n\ts.testProcessResetWorkflowWithError(true, errors.New(\"some random error\"), errors.New(\"some random error\"))\n}\n\nfunc (s *transferActiveTaskExecutorSuite) testProcessResetWorkflowWithError(sameRunID bool, resetError error, returnErr error) {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\ts.domainEntry.GetConfig().BadBinaries = types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\"test-binary-checksum\": {\n\t\t\t\tReason:          \"test-reason\",\n\t\t\t\tOperator:        \"test-operator\",\n\t\t\t\tCreatedTimeNano: common.Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t},\n\t}\n\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ResetWorkflowTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion: s.version,\n\t\t\tTaskID:  int64(59),\n\t\t},\n\t})\n\n\tfirstDecisionCompletedID := int64(2)\n\n\trunID := workflowExecution.GetRunID()\n\tif !sameRunID {\n\t\trunID = uuid.New()\n\t}\n\n\tresetPoints := &types.ResetPoints{\n\t\tPoints: []*types.ResetPointInfo{\n\t\t\t{\n\t\t\t\tBinaryChecksum:           \"test-binary-checksum\",\n\t\t\t\tRunID:                    runID,\n\t\t\t\tFirstDecisionCompletedID: firstDecisionCompletedID,\n\t\t\t\tResettable:               true,\n\t\t\t},\n\t\t},\n\t}\n\n\tmutableState.GetExecutionInfo().AutoResetPoints = resetPoints\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\tworkflowResetter := reset.NewMockWorkflowResetter(s.controller)\n\ts.transferActiveTaskExecutor.workflowResetter = workflowResetter\n\tversionHistories := mutableState.GetVersionHistories()\n\tcurrentVersionHistory, err := versionHistories.GetCurrentVersionHistory()\n\ts.NoError(err)\n\n\tcurrentBranchToken := currentVersionHistory.GetBranchToken()\n\trebuildLastEventVersion, err := currentVersionHistory.GetEventVersion(firstDecisionCompletedID)\n\ts.NoError(err)\n\n\tworkflowResetter.EXPECT().ResetWorkflow(\n\t\tgomock.Any(),\n\t\ts.domainID,\n\t\tworkflowExecution.GetWorkflowID(),\n\t\tworkflowExecution.GetRunID(),\n\t\tcurrentBranchToken,\n\t\tfirstDecisionCompletedID-1,\n\t\trebuildLastEventVersion,\n\t\tmutableState.GetNextEventID(),\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t\tgomock.Any(),\n\t\t\"test-reason\",\n\t\tnil,\n\t\tfalse).Return(resetError).Times(1)\n\n\t_, err = s.transferActiveTaskExecutor.Execute(transferTask)\n\n\tif returnErr != nil {\n\t\ts.Equal(returnErr, err)\n\t} else {\n\t\ts.Nil(err)\n\t}\n}\n\nfunc (s *transferActiveTaskExecutorSuite) TestCopySearchAttributes() {\n\tvar input map[string][]byte\n\ts.Nil(copySearchAttributes(input))\n\n\tkey := \"key\"\n\tval := []byte{'1', '2', '3'}\n\tinput = map[string][]byte{\n\t\tkey: val,\n\t}\n\tresult := copySearchAttributes(input)\n\ts.Equal(input, result)\n\tresult[key][0] = '0'\n\ts.Equal(byte('1'), val[0])\n}\n\nfunc (s *transferActiveTaskExecutorSuite) newTransferTaskFromInfo(\n\ttask persistence.Task,\n) Task {\n\treturn NewHistoryTask(s.mockShard, task, QueueTypeActiveTransfer, s.logger, nil, nil, nil, nil, nil)\n}\n\nfunc createAddActivityTaskRequest(\n\ttransferTask Task,\n\tai *persistence.ActivityInfo,\n\tpartitionConfig map[string]string,\n) *types.AddActivityTaskRequest {\n\n\ttaskInfo := transferTask.GetInfo().(*persistence.ActivityTask)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\ttaskList := &types.TaskList{Name: ai.TaskList, Kind: ai.TaskListKind.Ptr()}\n\n\treturn &types.AddActivityTaskRequest{\n\t\tDomainUUID:                    taskInfo.TargetDomainID,\n\t\tSourceDomainUUID:              taskInfo.DomainID,\n\t\tExecution:                     &workflowExecution,\n\t\tTaskList:                      taskList,\n\t\tScheduleID:                    taskInfo.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(ai.ScheduleToStartTimeout),\n\t\tPartitionConfig:               partitionConfig,\n\t}\n}\n\nfunc createAddDecisionTaskRequest(\n\ttransferTask Task,\n\tmutableState execution.MutableState,\n) *types.AddDecisionTaskRequest {\n\n\ttaskInfo := transferTask.GetInfo().(*persistence.DecisionTask)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\ttaskList := &types.TaskList{Name: taskInfo.TaskList, Kind: mutableState.GetExecutionInfo().TaskListKind.Ptr()}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\ttimeout := executionInfo.WorkflowTimeout\n\tif mutableState.GetExecutionInfo().TaskList != taskInfo.TaskList {\n\t\ttaskListStickyKind := types.TaskListKindSticky\n\t\ttaskList.Kind = &taskListStickyKind\n\t\ttimeout = executionInfo.StickyScheduleToStartTimeout\n\t}\n\n\treturn &types.AddDecisionTaskRequest{\n\t\tDomainUUID:                    taskInfo.DomainID,\n\t\tExecution:                     &workflowExecution,\n\t\tTaskList:                      taskList,\n\t\tScheduleID:                    taskInfo.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(timeout),\n\t\tPartitionConfig:               executionInfo.PartitionConfig,\n\t}\n}\n\nfunc createRecordWorkflowExecutionStartedRequest(\n\tt *testing.T,\n\tdomainName string,\n\tstartEvent *types.HistoryEvent,\n\ttransferTask Task,\n\tmutableState execution.MutableState,\n\tnumClusters int16,\n\tupdateTime time.Time,\n\tenableContextHeaderInVisibility bool,\n) *persistence.RecordWorkflowExecutionStartedRequest {\n\ttaskInfo := transferTask.GetInfo().(*persistence.RecordWorkflowStartedTask)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tbackoffSeconds := startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds()\n\texecutionTimestamp := int64(0)\n\tif backoffSeconds != 0 {\n\t\texecutionTimestamp = startEvent.GetTimestamp() + int64(backoffSeconds)*int64(time.Second)\n\t}\n\tvar searchAttributes map[string][]byte\n\tif enableContextHeaderInVisibility {\n\t\tcontextValueJSONString, err := json.Marshal(\"contextValue\")\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\t\tsearchAttributes = map[string][]byte{\n\t\t\t\"Header_context_key\": contextValueJSONString,\n\t\t}\n\t}\n\texecutionStatus, scheduledExecutionTimestamp := determineExecutionStatusForVisibility(startEvent, mutableState, false)\n\n\treturn &persistence.RecordWorkflowExecutionStartedRequest{\n\t\tDomain:                      domainName,\n\t\tDomainUUID:                  taskInfo.DomainID,\n\t\tExecution:                   workflowExecution,\n\t\tWorkflowTypeName:            executionInfo.WorkflowTypeName,\n\t\tStartTimestamp:              startEvent.GetTimestamp(),\n\t\tExecutionTimestamp:          executionTimestamp,\n\t\tWorkflowTimeout:             int64(executionInfo.WorkflowTimeout),\n\t\tTaskID:                      taskInfo.TaskID,\n\t\tTaskList:                    executionInfo.TaskList,\n\t\tIsCron:                      len(executionInfo.CronSchedule) > 0,\n\t\tNumClusters:                 numClusters,\n\t\tClusterAttributeScope:       executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetScope(),\n\t\tClusterAttributeName:        executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetName(),\n\t\tUpdateTimestamp:             updateTime.UnixNano(),\n\t\tSearchAttributes:            searchAttributes,\n\t\tExecutionStatus:             executionStatus,\n\t\tCronSchedule:                executionInfo.CronSchedule,\n\t\tScheduledExecutionTimestamp: scheduledExecutionTimestamp,\n\t}\n}\n\nfunc createRecordWorkflowExecutionClosedRequest(\n\tt *testing.T,\n\tdomainName string,\n\tstartEvent *types.HistoryEvent,\n\ttransferTask Task,\n\tmutableState execution.MutableState,\n\tnumClusters int16,\n\tupdateTime time.Time,\n\tcloseTimestamp *int64,\n\tenableContextHeaderInVisibility bool,\n) *persistence.RecordWorkflowExecutionClosedRequest {\n\ttaskInfo := transferTask.GetInfo().(*persistence.CloseExecutionTask)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tbackoffSeconds := startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds()\n\texecutionTimestamp := int64(0)\n\tif backoffSeconds != 0 {\n\t\texecutionTimestamp = startEvent.GetTimestamp() + int64(backoffSeconds)*int64(time.Second)\n\t}\n\tvar searchAttributes map[string][]byte\n\tif enableContextHeaderInVisibility {\n\t\tcontextValueJSONString, err := json.Marshal(\"contextValue\")\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\t\tsearchAttributes = map[string][]byte{\n\t\t\t\"Header_context_key\": contextValueJSONString,\n\t\t}\n\t}\n\tscheduledExecutionTimestamp := int64(0)\n\tif executionTimestamp != 0 {\n\t\tscheduledExecutionTimestamp = executionTimestamp\n\t}\n\n\t// Get completion event to determine execution status\n\t// For closed workflows, ExecutionStatus should reflect the close status (COMPLETED, FAILED, etc.)\n\t// not the running status (PENDING, STARTED)\n\tcompletionEvent, err := mutableState.GetCompletionEvent(context.Background())\n\tif err != nil {\n\t\tt.Fatal(err)\n\t}\n\texecutionStatus := getTestWorkflowExecutionStatus(completionEvent)\n\n\treturn &persistence.RecordWorkflowExecutionClosedRequest{\n\t\tDomain:                      domainName,\n\t\tDomainUUID:                  taskInfo.DomainID,\n\t\tExecution:                   workflowExecution,\n\t\tHistoryLength:               mutableState.GetNextEventID() - 1,\n\t\tWorkflowTypeName:            executionInfo.WorkflowTypeName,\n\t\tStartTimestamp:              startEvent.GetTimestamp(),\n\t\tExecutionTimestamp:          executionTimestamp,\n\t\tTaskID:                      taskInfo.TaskID,\n\t\tTaskList:                    executionInfo.TaskList,\n\t\tIsCron:                      len(executionInfo.CronSchedule) > 0,\n\t\tNumClusters:                 numClusters,\n\t\tClusterAttributeScope:       executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetScope(),\n\t\tClusterAttributeName:        executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetName(),\n\t\tUpdateTimestamp:             updateTime.UnixNano(),\n\t\tCloseTimestamp:              *closeTimestamp,\n\t\tRetentionSeconds:            int64(mutableState.GetDomainEntry().GetRetentionDays(taskInfo.GetWorkflowID()) * 24 * 3600),\n\t\tSearchAttributes:            searchAttributes,\n\t\tExecutionStatus:             executionStatus,\n\t\tCronSchedule:                executionInfo.CronSchedule,\n\t\tScheduledExecutionTimestamp: scheduledExecutionTimestamp,\n\t}\n}\n\n// getTestWorkflowExecutionStatus determines execution status from close event\n// This mirrors the logic in getWorkflowExecutionStatus in transfer_active_task_executor.go\nfunc getTestWorkflowExecutionStatus(closeEvent *types.HistoryEvent) types.WorkflowExecutionStatus {\n\tswitch closeEvent.GetEventType() {\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\treturn types.WorkflowExecutionStatusCompleted\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\treturn types.WorkflowExecutionStatusFailed\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\treturn types.WorkflowExecutionStatusCanceled\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\treturn types.WorkflowExecutionStatusTerminated\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\treturn types.WorkflowExecutionStatusTimedOut\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\t// For simplicity in tests, return ContinuedAsNew\n\t\t// Production code has more complex logic for cron workflows\n\t\treturn types.WorkflowExecutionStatusContinuedAsNew\n\tdefault:\n\t\treturn types.WorkflowExecutionStatusCompleted\n\t}\n}\n\nfunc createTestRequestCancelWorkflowExecutionRequest(\n\ttargetDomainName string,\n\ttaskInfo *persistence.CancelExecutionTask,\n\trequestID string,\n) *types.HistoryRequestCancelWorkflowExecutionRequest {\n\n\tsourceExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.TargetWorkflowID,\n\t\tRunID:      taskInfo.TargetRunID,\n\t}\n\n\treturn &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\tDomainUUID: taskInfo.TargetDomainID,\n\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\tDomain:            targetDomainName,\n\t\t\tWorkflowExecution: &targetExecution,\n\t\t\tIdentity:          execution.IdentityHistoryService,\n\t\t\t// Use the same request ID to dedupe RequestCancelWorkflowExecution calls\n\t\t\tRequestID: requestID,\n\t\t},\n\t\tExternalInitiatedEventID:  common.Int64Ptr(taskInfo.InitiatedID),\n\t\tExternalWorkflowExecution: &sourceExecution,\n\t\tChildWorkflowOnly:         taskInfo.TargetChildWorkflowOnly,\n\t}\n}\n\nfunc createTestSignalWorkflowExecutionRequest(\n\ttargetDomainName string,\n\ttaskInfo *persistence.SignalExecutionTask,\n\tsi *persistence.SignalInfo,\n) *types.HistorySignalWorkflowExecutionRequest {\n\n\tsourceExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.TargetWorkflowID,\n\t\tRunID:      taskInfo.TargetRunID,\n\t}\n\n\treturn &types.HistorySignalWorkflowExecutionRequest{\n\t\tDomainUUID: taskInfo.TargetDomainID,\n\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\tDomain:            targetDomainName,\n\t\t\tWorkflowExecution: &targetExecution,\n\t\t\tIdentity:          execution.IdentityHistoryService,\n\t\t\tSignalName:        si.SignalName,\n\t\t\tInput:             si.Input,\n\t\t\tRequestID:         si.SignalRequestID,\n\t\t\tControl:           si.Control,\n\t\t},\n\t\tExternalWorkflowExecution: &sourceExecution,\n\t\tChildWorkflowOnly:         taskInfo.TargetChildWorkflowOnly,\n\t}\n}\n\nfunc createTestChildWorkflowExecutionRequest(\n\tdomainName string,\n\tchildDomainName string,\n\ttaskInfo *persistence.StartChildExecutionTask,\n\tattributes *types.StartChildWorkflowExecutionInitiatedEventAttributes,\n\trequestID string,\n\tnow time.Time,\n\tpartitionConfig map[string]string,\n\tactiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n) (*types.HistoryStartWorkflowExecutionRequest, error) {\n\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\tfrontendStartReq := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              childDomainName,\n\t\tWorkflowID:                          attributes.WorkflowID,\n\t\tWorkflowType:                        attributes.WorkflowType,\n\t\tTaskList:                            attributes.TaskList,\n\t\tInput:                               attributes.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: attributes.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      attributes.TaskStartToCloseTimeoutSeconds,\n\t\t// Use the same request ID to dedupe StartWorkflowExecution calls\n\t\tRequestID:                    requestID,\n\t\tWorkflowIDReusePolicy:        attributes.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                  attributes.RetryPolicy,\n\t\tActiveClusterSelectionPolicy: activeClusterSelectionPolicy,\n\t}\n\n\tparentInfo := &types.ParentExecutionInfo{\n\t\tDomainUUID:  taskInfo.DomainID,\n\t\tDomain:      domainName,\n\t\tExecution:   &workflowExecution,\n\t\tInitiatedID: taskInfo.InitiatedID,\n\t}\n\n\thistoryStartReq, err := common.CreateHistoryStartWorkflowRequest(\n\t\ttaskInfo.TargetDomainID, frontendStartReq, now, partitionConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\thistoryStartReq.ParentExecutionInfo = parentInfo\n\treturn historyStartReq, nil\n}\n\nfunc createUpsertWorkflowSearchAttributesRequest(\n\tt *testing.T,\n\tdomainName string,\n\tstartEvent *types.HistoryEvent,\n\ttransferTask Task,\n\tmutableState execution.MutableState,\n\tnumClusters int16,\n\tupdateTime time.Time,\n\tenableContextHeaderInVisibility bool,\n) *persistence.UpsertWorkflowExecutionRequest {\n\n\ttaskInfo := transferTask.GetInfo().(*persistence.UpsertWorkflowSearchAttributesTask)\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: taskInfo.WorkflowID,\n\t\tRunID:      taskInfo.RunID,\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tbackoffSeconds := startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds()\n\texecutionTimestamp := int64(0)\n\tif backoffSeconds != 0 {\n\t\texecutionTimestamp = startEvent.GetTimestamp() + int64(backoffSeconds)*int64(time.Second)\n\t}\n\tvar searchAttributes map[string][]byte\n\tif enableContextHeaderInVisibility {\n\t\tcontextValueJSONString, err := json.Marshal(\"contextValue\")\n\t\tif err != nil {\n\t\t\tt.Fatal(err)\n\t\t}\n\t\tsearchAttributes = map[string][]byte{\n\t\t\t\"Header_context_key\": contextValueJSONString,\n\t\t}\n\t}\n\n\texecutionStatus, scheduledExecutionTimestamp := determineExecutionStatus(startEvent, mutableState)\n\n\treturn &persistence.UpsertWorkflowExecutionRequest{\n\t\tDomain:                      domainName,\n\t\tDomainUUID:                  taskInfo.DomainID,\n\t\tExecution:                   workflowExecution,\n\t\tWorkflowTypeName:            executionInfo.WorkflowTypeName,\n\t\tStartTimestamp:              startEvent.GetTimestamp(),\n\t\tExecutionTimestamp:          executionTimestamp,\n\t\tWorkflowTimeout:             int64(executionInfo.WorkflowTimeout),\n\t\tTaskID:                      taskInfo.TaskID,\n\t\tTaskList:                    executionInfo.TaskList,\n\t\tIsCron:                      len(executionInfo.CronSchedule) > 0,\n\t\tNumClusters:                 numClusters,\n\t\tClusterAttributeScope:       executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetScope(),\n\t\tClusterAttributeName:        executionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute().GetName(),\n\t\tUpdateTimestamp:             updateTime.UnixNano(),\n\t\tSearchAttributes:            searchAttributes,\n\t\tExecutionStatus:             executionStatus,\n\t\tCronSchedule:                executionInfo.CronSchedule,\n\t\tScheduledExecutionTimestamp: scheduledExecutionTimestamp,\n\t}\n}\n\nfunc createRecordWorkflowExecutionUninitializedRequest(\n\ttransferTask Task,\n\tmutableState execution.MutableState,\n\tupdateTime time.Time,\n\tshardID int64,\n) *persistence.RecordWorkflowExecutionUninitializedRequest {\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: transferTask.GetWorkflowID(),\n\t\tRunID:      transferTask.GetRunID(),\n\t}\n\texecutionInfo := mutableState.GetExecutionInfo()\n\treturn &persistence.RecordWorkflowExecutionUninitializedRequest{\n\t\tDomainUUID:       transferTask.GetDomainID(),\n\t\tExecution:        workflowExecution,\n\t\tWorkflowTypeName: executionInfo.WorkflowTypeName,\n\t\tUpdateTimestamp:  updateTime.UnixNano(),\n\t\tShardID:          shardID,\n\t}\n}\n"
  },
  {
    "path": "service/history/task/transfer_standby_task_executor.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/history/simulation\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\ttransferStandbyTaskExecutor struct {\n\t\t*transferTaskExecutorBase\n\n\t\tclusterName            string\n\t\thistoryResender        ndc.HistoryResender\n\t\tgetRemoteClusterNameFn func(context.Context, persistence.Task) (string, error)\n\t}\n)\n\n// NewTransferStandbyTaskExecutor creates a new task executor for standby transfer task\nfunc NewTransferStandbyTaskExecutor(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\thistoryResender ndc.HistoryResender,\n\tlogger log.Logger,\n\tclusterName string,\n\tconfig *config.Config,\n) Executor {\n\treturn &transferStandbyTaskExecutor{\n\t\ttransferTaskExecutorBase: newTransferTaskExecutorBase(\n\t\t\tshard,\n\t\t\tarchiverClient,\n\t\t\texecutionCache,\n\t\t\tlogger,\n\t\t\tconfig,\n\t\t),\n\t\tclusterName:     clusterName,\n\t\thistoryResender: historyResender,\n\t\tgetRemoteClusterNameFn: func(ctx context.Context, taskInfo persistence.Task) (string, error) {\n\t\t\tif shard.GetConfig().EnableTransferQueueV2(shard.GetShardID()) {\n\t\t\t\treturn getRemoteClusterName(ctx, shard.GetClusterMetadata().GetCurrentClusterName(), shard.GetActiveClusterManager(), taskInfo)\n\t\t\t}\n\t\t\treturn clusterName, nil\n\t\t},\n\t}\n}\n\nfunc (t *transferStandbyTaskExecutor) Execute(task Task) (ExecuteResponse, error) {\n\tsimulation.LogEvents(simulation.E{\n\t\tEventName:  simulation.EventNameExecuteHistoryTask,\n\t\tHost:       t.shard.GetConfig().HostName,\n\t\tShardID:    t.shard.GetShardID(),\n\t\tDomainID:   task.GetDomainID(),\n\t\tWorkflowID: task.GetWorkflowID(),\n\t\tRunID:      task.GetRunID(),\n\t\tPayload: map[string]any{\n\t\t\t\"task_category\": persistence.HistoryTaskCategoryTransfer.Name(),\n\t\t\t\"task_type\":     task.GetTaskType(),\n\t\t\t\"task_key\":      task.GetTaskKey(),\n\t\t},\n\t})\n\tscope := getOrCreateDomainTaggedScope(t.shard, GetTransferTaskMetricsScope(task.GetTaskType(), false), task.GetDomainID(), t.logger)\n\texecuteResponse := ExecuteResponse{\n\t\tScope:        scope,\n\t\tIsActiveTask: false,\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), taskDefaultTimeout)\n\tdefer cancel()\n\n\tswitch transferTask := task.GetInfo().(type) {\n\tcase *persistence.ActivityTask:\n\t\treturn executeResponse, t.processActivityTask(ctx, transferTask)\n\tcase *persistence.DecisionTask:\n\t\treturn executeResponse, t.processDecisionTask(ctx, transferTask)\n\tcase *persistence.CloseExecutionTask:\n\t\treturn executeResponse, t.processCloseExecution(ctx, transferTask)\n\tcase *persistence.RecordWorkflowClosedTask:\n\t\treturn executeResponse, t.processCloseExecution(ctx, &persistence.CloseExecutionTask{\n\t\t\tWorkflowIdentifier: transferTask.WorkflowIdentifier,\n\t\t\tTaskData:           transferTask.TaskData,\n\t\t})\n\tcase *persistence.RecordChildExecutionCompletedTask:\n\t\t// no action needed for standby\n\t\t// check the comment in t.processCloseExecution()\n\t\treturn executeResponse, nil\n\tcase *persistence.CancelExecutionTask:\n\t\treturn executeResponse, t.processCancelExecution(ctx, transferTask)\n\tcase *persistence.SignalExecutionTask:\n\t\treturn executeResponse, t.processSignalExecution(ctx, transferTask)\n\tcase *persistence.StartChildExecutionTask:\n\t\treturn executeResponse, t.processStartChildExecution(ctx, transferTask)\n\tcase *persistence.RecordWorkflowStartedTask:\n\t\treturn executeResponse, t.processRecordWorkflowStarted(ctx, transferTask)\n\tcase *persistence.ResetWorkflowTask:\n\t\t// no reset needed for standby\n\t\t// TODO: add error logs\n\t\treturn executeResponse, nil\n\tcase *persistence.UpsertWorkflowSearchAttributesTask:\n\t\treturn executeResponse, t.processUpsertWorkflowSearchAttributes(ctx, transferTask)\n\tdefault:\n\t\treturn executeResponse, errUnknownTransferTask\n\t}\n}\n\n// Empty func for now\nfunc (t *transferStandbyTaskExecutor) Stop() {}\n\nfunc (t *transferStandbyTaskExecutor) processActivityTask(\n\tctx context.Context,\n\ttransferTask *persistence.ActivityTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tactivityInfo, ok := mutableState.GetActivityInfo(transferTask.ScheduleID)\n\t\tif !ok {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, activityInfo.Version, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttaskList := types.TaskList{\n\t\t\tName: activityInfo.TaskList,\n\t\t\tKind: activityInfo.TaskListKind.Ptr(),\n\t\t}\n\t\tif taskList.Name == \"\" {\n\t\t\ttaskList.Name = transferTask.TaskList\n\t\t}\n\n\t\tif activityInfo.StartedID == constants.EmptyEventID {\n\t\t\treturn newPushActivityToMatchingInfo(\n\t\t\t\tactivityInfo.ScheduleToStartTimeout,\n\t\t\t\ttaskList,\n\t\t\t\tmutableState.GetExecutionInfo().PartitionConfig,\n\t\t\t), nil\n\t\t}\n\n\t\treturn nil, nil\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\ttransferTask.ScheduleID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttransferTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.pushActivity,\n\t\t\tt.pushActivity,\n\t\t),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processDecisionTask(\n\tctx context.Context,\n\ttransferTask *persistence.DecisionTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tdecisionInfo, ok := mutableState.GetDecisionInfo(transferTask.ScheduleID)\n\t\tif !ok {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\tworkflowTimeout := executionInfo.WorkflowTimeout\n\t\tdecisionTimeout := min(workflowTimeout, constants.MaxTaskTimeout)\n\t\tif executionInfo.TaskList != transferTask.TaskList {\n\t\t\t// Experimental: try to push sticky task as regular task with sticky timeout as TTL.\n\t\t\t// workflow might be sticky before namespace become standby\n\t\t\t// there shall already be a schedule_to_start timer created\n\t\t\tdecisionTimeout = executionInfo.StickyScheduleToStartTimeout\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, decisionInfo.Version, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif decisionInfo.StartedID == constants.EmptyEventID {\n\t\t\treturn newPushDecisionToMatchingInfo(\n\t\t\t\tdecisionTimeout,\n\t\t\t\ttypes.TaskList{Name: executionInfo.TaskList, Kind: executionInfo.TaskListKind.Ptr()}, // at standby, always use non-sticky tasklist\n\t\t\t\tmutableState.GetExecutionInfo().PartitionConfig,\n\t\t\t), nil\n\t\t}\n\n\t\treturn nil, nil\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\ttransferTask.ScheduleID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttransferTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.pushDecision,\n\t\t\tt.pushDecision,\n\t\t),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processCloseExecution(\n\tctx context.Context,\n\ttransferTask *persistence.CloseExecutionTask,\n) error {\n\n\tprocessTaskIfClosed := true\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tif mutableState.IsWorkflowExecutionRunning() {\n\t\t\t// this can happen if workflow is reset.\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tcompletionEvent, err := mutableState.GetCompletionEvent(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\twfCloseTime := completionEvent.GetTimestamp()\n\n\t\texecutionInfo := mutableState.GetExecutionInfo()\n\t\tworkflowTypeName := executionInfo.WorkflowTypeName\n\t\tworkflowCloseTimestamp := wfCloseTime\n\t\tworkflowCloseStatus := persistence.ToInternalWorkflowExecutionCloseStatus(executionInfo.CloseStatus)\n\t\tworkflowHistoryLength := mutableState.GetNextEventID() - 1\n\t\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tworkflowStartTimestamp := startEvent.GetTimestamp()\n\t\tworkflowExecutionTimestamp := getWorkflowExecutionTimestamp(mutableState, startEvent)\n\t\tvisibilityMemo := getWorkflowMemo(executionInfo.Memo)\n\t\tsearchAttr := executionInfo.SearchAttributes\n\t\theaders := getWorkflowHeaders(startEvent)\n\t\tisCron := len(executionInfo.CronSchedule) > 0\n\t\tcronSchedule := \"\"\n\t\tif isCron {\n\t\t\tcronSchedule = executionInfo.CronSchedule\n\t\t}\n\t\tupdateTimestamp := t.shard.GetTimeSource().Now()\n\n\t\tlastWriteVersion, err := mutableState.GetLastWriteVersion()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, lastWriteVersion, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(transferTask.DomainID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tnumClusters := (int16)(len(domainEntry.GetReplicationConfig().Clusters))\n\n\t\texecutionStatus := getWorkflowExecutionStatus(mutableState, completionEvent)\n\n\t\t// Calculate ScheduledExecutionTime\n\t\tscheduledExecutionTimestamp := startEvent.GetTimestamp()\n\t\tif startEvent.WorkflowExecutionStartedEventAttributes != nil &&\n\t\t\tstartEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds() > 0 {\n\t\t\tscheduledExecutionTimestamp = startEvent.GetTimestamp() +\n\t\t\t\tint64(startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds())*int64(time.Second)\n\t\t}\n\n\t\t// DO NOT REPLY TO PARENT\n\t\t// since event replication should be done by active cluster\n\t\treturn nil, t.recordWorkflowClosed(\n\t\t\tctx,\n\t\t\ttransferTask.DomainID,\n\t\t\ttransferTask.WorkflowID,\n\t\t\ttransferTask.RunID,\n\t\t\tworkflowTypeName,\n\t\t\tworkflowStartTimestamp,\n\t\t\tworkflowExecutionTimestamp.UnixNano(),\n\t\t\tworkflowCloseTimestamp,\n\t\t\t*workflowCloseStatus,\n\t\t\tworkflowHistoryLength,\n\t\t\ttransferTask.GetTaskID(),\n\t\t\tvisibilityMemo,\n\t\t\texecutionInfo.TaskList,\n\t\t\tisCron,\n\t\t\tcronSchedule,\n\t\t\tnumClusters,\n\t\t\tupdateTimestamp.UnixNano(),\n\t\t\tsearchAttr,\n\t\t\theaders,\n\t\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\t\texecutionStatus,\n\t\t\tscheduledExecutionTimestamp,\n\t\t)\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\t0,\n\t\tactionFn,\n\t\tstandbyTaskPostActionNoOp,\n\t) // no op post action, since the entire workflow is finished\n}\n\nfunc (t *transferStandbyTaskExecutor) processCancelExecution(\n\tctx context.Context,\n\ttransferTask *persistence.CancelExecutionTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\trequestCancelInfo, ok := mutableState.GetRequestCancelInfo(transferTask.InitiatedID)\n\t\tif !ok {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, requestCancelInfo.Version, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\ttransferTask.InitiatedID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttransferTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processSignalExecution(\n\tctx context.Context,\n\ttransferTask *persistence.SignalExecutionTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tsignalInfo, ok := mutableState.GetSignalInfo(transferTask.InitiatedID)\n\t\tif !ok {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, signalInfo.Version, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\ttransferTask.InitiatedID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttransferTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processStartChildExecution(\n\tctx context.Context,\n\ttransferTask *persistence.StartChildExecutionTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\tactionFn := func(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\n\t\tchildWorkflowInfo, ok := mutableState.GetChildExecutionInfo(transferTask.InitiatedID)\n\t\tif !ok {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.DomainID, childWorkflowInfo.Version, transferTask.Version, transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif childWorkflowInfo.StartedID != constants.EmptyEventID {\n\t\t\treturn nil, nil\n\t\t}\n\n\t\treturn getHistoryResendInfo(mutableState)\n\t}\n\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\ttransferTask.InitiatedID,\n\t\tactionFn,\n\t\tgetStandbyPostActionFn(\n\t\t\tt.logger,\n\t\t\ttransferTask,\n\t\t\tt.getCurrentTime,\n\t\t\tt.config.StandbyTaskMissingEventsResendDelay(),\n\t\t\tt.config.StandbyTaskMissingEventsDiscardDelay(),\n\t\t\tt.fetchHistoryFromRemote,\n\t\t\tstandbyTaskPostActionTaskDiscarded,\n\t\t),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processRecordWorkflowStarted(\n\tctx context.Context,\n\ttransferTask *persistence.RecordWorkflowStartedTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\t0,\n\t\tfunc(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\t\t\treturn nil, t.processRecordWorkflowStartedOrUpsertHelper(ctx, transferTask, mutableState, true)\n\t\t},\n\t\tstandbyTaskPostActionNoOp,\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processUpsertWorkflowSearchAttributes(\n\tctx context.Context,\n\ttransferTask *persistence.UpsertWorkflowSearchAttributesTask,\n) error {\n\n\tprocessTaskIfClosed := false\n\treturn t.processTransfer(\n\t\tctx,\n\t\tprocessTaskIfClosed,\n\t\ttransferTask,\n\t\t0,\n\t\tfunc(ctx context.Context, wfContext execution.Context, mutableState execution.MutableState) (interface{}, error) {\n\t\t\treturn nil, t.processRecordWorkflowStartedOrUpsertHelper(ctx, transferTask, mutableState, false)\n\t\t},\n\t\tstandbyTaskPostActionNoOp,\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) processRecordWorkflowStartedOrUpsertHelper(\n\tctx context.Context,\n\ttransferTask persistence.Task,\n\tmutableState execution.MutableState,\n\tisRecordStart bool,\n) error {\n\n\tworkflowStartedScope := getOrCreateDomainTaggedScope(t.shard, metrics.TransferStandbyTaskRecordWorkflowStartedScope, transferTask.GetDomainID(), t.logger)\n\n\t// verify task version for RecordWorkflowStarted.\n\t// upsert doesn't require verifyTask, because it is just a sync of mutableState.\n\tif isRecordStart {\n\t\tstartVersion, err := mutableState.GetStartVersion()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tok, err := verifyTaskVersion(t.shard, t.logger, transferTask.GetDomainID(), startVersion, transferTask.GetVersion(), transferTask)\n\t\tif err != nil || !ok {\n\t\t\treturn err\n\t\t}\n\t}\n\n\texecutionInfo := mutableState.GetExecutionInfo()\n\tworkflowTimeout := executionInfo.WorkflowTimeout\n\twfTypeName := executionInfo.WorkflowTypeName\n\tstartEvent, err := mutableState.GetStartEvent(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tstartTimestamp := startEvent.GetTimestamp()\n\texecutionTimestamp := getWorkflowExecutionTimestamp(mutableState, startEvent)\n\tvisibilityMemo := getWorkflowMemo(executionInfo.Memo)\n\tisCron := len(executionInfo.CronSchedule) > 0\n\tcronSchedule := \"\"\n\tif isCron {\n\t\tcronSchedule = executionInfo.CronSchedule\n\t}\n\tupdateTimestamp := t.shard.GetTimeSource().Now()\n\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(transferTask.GetDomainID())\n\tif err != nil {\n\t\treturn err\n\t}\n\tnumClusters := (int16)(len(domainEntry.GetReplicationConfig().Clusters))\n\n\tsearchAttr := copySearchAttributes(executionInfo.SearchAttributes)\n\theaders := getWorkflowHeaders(startEvent)\n\n\tisAdvancedVisibilityEnabled := common.IsAdvancedVisibilityWritingEnabled(\n\t\tt.config.WriteVisibilityStoreName(),\n\t\tt.config.IsAdvancedVisConfigExist,\n\t)\n\texecutionStatus, scheduledExecutionTimestamp := determineExecutionStatusForVisibility(startEvent, mutableState, isAdvancedVisibilityEnabled)\n\n\tif isRecordStart {\n\t\tworkflowStartedScope.IncCounter(metrics.WorkflowStartedCount)\n\t\treturn t.recordWorkflowStarted(\n\t\t\tctx,\n\t\t\ttransferTask.GetDomainID(),\n\t\t\ttransferTask.GetWorkflowID(),\n\t\t\ttransferTask.GetRunID(),\n\t\t\twfTypeName,\n\t\t\tstartTimestamp,\n\t\t\texecutionTimestamp.UnixNano(),\n\t\t\tworkflowTimeout,\n\t\t\ttransferTask.GetTaskID(),\n\t\t\texecutionInfo.TaskList,\n\t\t\tisCron,\n\t\t\tnumClusters,\n\t\t\tvisibilityMemo,\n\t\t\tupdateTimestamp.UnixNano(),\n\t\t\tsearchAttr,\n\t\t\theaders,\n\t\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\t\tcronSchedule,\n\t\t\texecutionStatus,\n\t\t\tscheduledExecutionTimestamp,\n\t\t)\n\t}\n\treturn t.upsertWorkflowExecution(\n\t\tctx,\n\t\ttransferTask.GetDomainID(),\n\t\ttransferTask.GetWorkflowID(),\n\t\ttransferTask.GetRunID(),\n\t\twfTypeName,\n\t\tstartTimestamp,\n\t\texecutionTimestamp.UnixNano(),\n\t\tworkflowTimeout,\n\t\ttransferTask.GetTaskID(),\n\t\texecutionInfo.TaskList,\n\t\tvisibilityMemo,\n\t\tisCron,\n\t\tnumClusters,\n\t\tupdateTimestamp.UnixNano(),\n\t\tsearchAttr,\n\t\theaders,\n\t\texecutionInfo.ActiveClusterSelectionPolicy.GetClusterAttribute(),\n\t\tcronSchedule,\n\t\texecutionStatus,\n\t\tscheduledExecutionTimestamp,\n\t)\n\n}\n\nfunc (t *transferStandbyTaskExecutor) processTransfer(\n\tctx context.Context,\n\tprocessTaskIfClosed bool,\n\ttransferTask persistence.Task,\n\teventID int64,\n\tactionFn standbyActionFn,\n\tpostActionFn standbyPostActionFn,\n) (retError error) {\n\twfContext, release, err := t.executionCache.GetOrCreateWorkflowExecutionWithTimeout(\n\t\ttransferTask.GetDomainID(),\n\t\tgetWorkflowExecution(transferTask),\n\t\ttaskGetExecutionContextTimeout,\n\t)\n\tif err != nil {\n\t\tif err == context.DeadlineExceeded {\n\t\t\treturn errWorkflowBusy\n\t\t}\n\t\treturn err\n\t}\n\tdefer func() {\n\t\tif isRedispatchErr(err) {\n\t\t\trelease(nil)\n\t\t} else {\n\t\t\trelease(retError)\n\t\t}\n\t}()\n\n\tmutableState, err := loadMutableState(ctx, wfContext, transferTask, t.metricsClient.Scope(metrics.TransferQueueProcessorScope), t.logger, eventID)\n\tif err != nil || mutableState == nil {\n\t\treturn err\n\t}\n\n\tif !mutableState.IsWorkflowExecutionRunning() && !processTaskIfClosed {\n\t\t// workflow already finished, no need to process the timer\n\t\treturn nil\n\t}\n\n\thistoryResendInfo, err := actionFn(ctx, wfContext, mutableState)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\trelease(nil)\n\treturn postActionFn(ctx, transferTask, historyResendInfo, t.logger)\n}\n\nfunc (t *transferStandbyTaskExecutor) pushActivity(\n\tctx context.Context,\n\ttask persistence.Task,\n\tpostActionInfo interface{},\n\tlogger log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\treturn t.transferTaskExecutorBase.pushActivity(\n\t\tctx,\n\t\ttask.(*persistence.ActivityTask),\n\t\tpostActionInfo.(*pushActivityToMatchingInfo),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) pushDecision(\n\tctx context.Context,\n\ttask persistence.Task,\n\tpostActionInfo interface{},\n\tlogger log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\treturn t.transferTaskExecutorBase.pushDecision(\n\t\tctx,\n\t\ttask.(*persistence.DecisionTask),\n\t\tpostActionInfo.(*pushDecisionToMatchingInfo),\n\t)\n}\n\nfunc (t *transferStandbyTaskExecutor) fetchHistoryFromRemote(\n\tctx context.Context,\n\ttaskInfo persistence.Task,\n\tpostActionInfo interface{},\n\t_ log.Logger,\n) error {\n\n\tif postActionInfo == nil {\n\t\treturn nil\n\t}\n\n\tresendInfo := postActionInfo.(*historyResendInfo)\n\n\tt.metricsClient.IncCounter(metrics.HistoryRereplicationByTransferTaskScope, metrics.CadenceClientRequests)\n\tstopwatch := t.metricsClient.StartTimer(metrics.HistoryRereplicationByTransferTaskScope, metrics.CadenceClientLatency)\n\tdefer stopwatch.Stop()\n\n\tremoteClusterName, err := t.getRemoteClusterNameFn(ctx, taskInfo)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif resendInfo.lastEventID != nil && resendInfo.lastEventVersion != nil {\n\t\t// note history resender doesn't take in a context parameter, there's a separate dynamicconfig for\n\t\t// controlling the timeout for resending history.\n\t\terr = t.historyResender.SendSingleWorkflowHistory(\n\t\t\tremoteClusterName,\n\t\t\ttaskInfo.GetDomainID(),\n\t\t\ttaskInfo.GetWorkflowID(),\n\t\t\ttaskInfo.GetRunID(),\n\t\t\tresendInfo.lastEventID,\n\t\t\tresendInfo.lastEventVersion,\n\t\t\tnil,\n\t\t\tnil,\n\t\t)\n\t} else {\n\t\terr = &types.InternalServiceError{\n\t\t\tMessage: fmt.Sprintf(\"incomplete historyResendInfo: %v\", resendInfo),\n\t\t}\n\t}\n\n\tif err != nil {\n\t\tt.logger.Error(\"Error re-replicating history from remote.\",\n\t\t\ttag.ShardID(t.shard.GetShardID()),\n\t\t\ttag.WorkflowDomainID(taskInfo.GetDomainID()),\n\t\t\ttag.WorkflowID(taskInfo.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(taskInfo.GetRunID()),\n\t\t\ttag.SourceCluster(remoteClusterName),\n\t\t\ttag.Error(err),\n\t\t)\n\t}\n\n\t// return error so task processing logic will retry\n\treturn &redispatchError{Reason: \"fetchHistoryFromRemote\"}\n}\n\nfunc (t *transferStandbyTaskExecutor) getCurrentTime(taskInfo persistence.Task) (time.Time, error) {\n\t// for standby task, we can always use timesource.Now, because they're supposed to be processed immediately after creation,\n\t// this is what're doing for queue v2, for queue v1, I just don't bother to change the behavior\n\treturn t.shard.GetCurrentTime(t.clusterName), nil\n}\n"
  },
  {
    "path": "service/history/task/transfer_standby_task_executor_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tdc \"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/ndc\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/events\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\ttest \"github.com/uber/cadence/service/history/testing\"\n\twarchiver \"github.com/uber/cadence/service/worker/archiver\"\n)\n\ntype (\n\ttransferStandbyTaskExecutorSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller             *gomock.Controller\n\t\tmockShard              *shard.TestContext\n\t\tmockDomainCache        *cache.MockDomainCache\n\t\tmockNDCHistoryResender *ndc.MockHistoryResender\n\t\tmockMatchingClient     *matching.MockClient\n\n\t\tmockVisibilityMgr    *mocks.VisibilityManager\n\t\tmockExecutionMgr     *mocks.ExecutionManager\n\t\tmockArchivalClient   *warchiver.MockClient\n\t\tmockArchivalMetadata *archiver.MockArchivalMetadata\n\t\tmockArchiverProvider *provider.MockArchiverProvider\n\n\t\tlogger      log.Logger\n\t\tdomainID    string\n\t\tdomainName  string\n\t\tdomainEntry *cache.DomainCacheEntry\n\t\tversion     int64\n\t\tclusterName string\n\n\t\ttimeSource           clock.MockedTimeSource\n\t\tfetchHistoryDuration time.Duration\n\t\tdiscardDuration      time.Duration\n\n\t\ttransferStandbyTaskExecutor *transferStandbyTaskExecutor\n\t}\n)\n\nfunc TestTransferStandbyTaskExecutorSuite(t *testing.T) {\n\ts := new(transferStandbyTaskExecutorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) SetupSuite() {\n\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ttestConfig := config.NewForTest()\n\ts.domainID = constants.TestDomainID\n\ts.domainName = constants.TestDomainName\n\ts.domainEntry = constants.TestGlobalDomainEntry\n\ts.version = s.domainEntry.GetFailoverVersion()\n\n\ts.timeSource = clock.NewMockedTimeSource()\n\ts.fetchHistoryDuration = testConfig.StandbyTaskMissingEventsResendDelay() +\n\t\t(testConfig.StandbyTaskMissingEventsDiscardDelay()-testConfig.StandbyTaskMissingEventsResendDelay())/2\n\ts.discardDuration = testConfig.StandbyTaskMissingEventsDiscardDelay() * 2\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockNDCHistoryResender = ndc.NewMockHistoryResender(s.controller)\n\n\ts.mockShard = shard.NewTestContext(\n\t\ts.T(),\n\t\ts.controller,\n\t\t&persistence.ShardInfo{\n\t\t\tRangeID:          1,\n\t\t\tTransferAckLevel: 0,\n\t\t},\n\t\ttestConfig,\n\t)\n\ts.mockShard.SetEventsCache(events.NewCache(\n\t\ts.mockShard.GetShardID(),\n\t\ts.mockShard.GetHistoryManager(),\n\t\ts.mockShard.GetConfig(),\n\t\ts.mockShard.GetLogger(),\n\t\ts.mockShard.GetMetricsClient(),\n\t\ts.mockShard.GetDomainCache(),\n\t))\n\ts.mockShard.Resource.TimeSource = s.timeSource\n\n\ts.mockMatchingClient = s.mockShard.Resource.MatchingClient\n\ts.mockExecutionMgr = s.mockShard.Resource.ExecutionMgr\n\ts.mockVisibilityMgr = s.mockShard.Resource.VisibilityMgr\n\ts.mockArchivalClient = warchiver.NewMockClient(s.controller)\n\ts.mockArchivalMetadata = s.mockShard.Resource.ArchivalMetadata\n\ts.mockArchiverProvider = s.mockShard.Resource.ArchiverProvider\n\ts.mockDomainCache = s.mockShard.Resource.DomainCache\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestDomainID).Return(constants.TestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestDomainName).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(constants.TestDomainName).Return(constants.TestDomainID, nil).AnyTimes()\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestRateLimitedDomainID).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestRateLimitedDomainID).Return(constants.TestRateLimitedDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestRateLimitedDomainName).Return(constants.TestRateLimitedDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(constants.TestRateLimitedDomainName).Return(constants.TestRateLimitedDomainID, nil).AnyTimes()\n\n\ts.mockDomainCache.EXPECT().GetDomainByID(constants.TestActiveActiveDomainID).Return(constants.TestActiveActiveDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(constants.TestActiveActiveDomainID).Return(constants.TestActiveActiveDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(constants.TestActiveActiveDomainName).Return(constants.TestActiveActiveDomainEntry, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainID(constants.TestActiveActiveDomainName).Return(constants.TestActiveActiveDomainID, nil).AnyTimes()\n\n\ttestActiveClusterInfo := &types.ActiveClusterInfo{\n\t\tActiveClusterName: constants.TestGlobalDomainEntry.GetReplicationConfig().ActiveClusterName,\n\t\tFailoverVersion:   constants.TestGlobalDomainEntry.GetFailoverVersion(),\n\t}\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(testActiveClusterInfo, nil).AnyTimes()\n\ts.mockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestActiveActiveDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: cluster.TestAlternativeClusterName}, nil).AnyTimes()\n\n\ts.logger = s.mockShard.GetLogger()\n\ts.clusterName = cluster.TestAlternativeClusterName\n\ts.transferStandbyTaskExecutor = NewTransferStandbyTaskExecutor(\n\t\ts.mockShard,\n\t\ts.mockArchivalClient,\n\t\texecution.NewCache(s.mockShard),\n\t\ts.mockNDCHistoryResender,\n\t\ts.logger,\n\t\ts.clusterName,\n\t\ttestConfig,\n\t).(*transferStandbyTaskExecutor)\n\ts.transferStandbyTaskExecutor.getRemoteClusterNameFn = func(ctx context.Context, taskInfo persistence.Task) (string, error) {\n\t\treturn s.clusterName, nil\n\t}\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockShard.Finish(s.T())\n\ts.transferStandbyTaskExecutor.Stop()\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessActivityTask_Pending() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   \"unused\",\n\t\tScheduleID: event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessActivityTask_Pending_ActiveActiveDomain() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, constants.TestActiveActiveDomainID)\n\ts.NoError(err)\n\n\tevent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   constants.TestActiveActiveDomainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessActivityTask_Pending_PushToMatching() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, ai := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTask, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessActivityTask_Pending_TaskListKindEphemeral() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, ai, _, _, _, _ := mutableState.AddActivityTaskScheduledEvent(nil, decisionCompletionID, &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    \"activity-1\",\n\t\tActivityType:                  &types.ActivityType{Name: \"some random activity type\"},\n\t\tTaskList:                      &types.TaskList{Name: mutableState.GetExecutionInfo().TaskList, Kind: types.TaskListKindEphemeral.Ptr()},\n\t\tInput:                         []byte{},\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(1),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(1),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(1),\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(1),\n\t}, false,\n\t)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddActivityTask(gomock.Any(), createAddActivityTaskRequest(transferTask, ai, mutableState.GetExecutionInfo().PartitionConfig)).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessActivityTask_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent, _ := test.AddActivityTaskScheduledEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\t\"activity-1\",\n\t\t\"some random activity type\",\n\t\tmutableState.GetExecutionInfo().TaskList,\n\t\t[]byte{}, 1, 1, 1, 1,\n\t)\n\tevent = test.AddActivityTaskStartedEvent(mutableState, event.ID, \"\")\n\tmutableState.FlushBufferedEvents()\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.ActivityTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Pending() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Pending_ActiveActiveDomain() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, constants.TestActiveActiveDomainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   constants.TestActiveActiveDomainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Pending_PushToMatching() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Pending_TaskListKindEphemeral() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflowWithTaskList(&types.TaskList{Name: \"ephemy\", Kind: types.TaskListKindEphemeral.Ptr()}, s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, di.ScheduleID, di.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockMatchingClient.EXPECT().AddDecisionTask(gomock.Any(), createAddDecisionTaskRequest(transferTask, mutableState)).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Success_FirstDecision() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tevent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\tdi.StartedID = event.ID\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessDecisionTask_Success_NonFirstDecision() {\n\n\tworkflowExecution, mutableState, _, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tdi := test.AddDecisionTaskScheduledEvent(mutableState)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.DecisionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTaskList:   mutableState.GetExecutionInfo().TaskList,\n\t\tScheduleID: di.ScheduleID,\n\t})\n\n\tevent := test.AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\tdi.StartedID = event.ID\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessCloseExecution() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CloseExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessRecordWorkflowClosedTask() {\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tevent := test.AddCompleteWorkflowEvent(mutableState, decisionCompletionID, nil)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.RecordWorkflowClosedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\ts.mockVisibilityMgr.On(\"RecordWorkflowExecutionClosed\", mock.Anything, mock.Anything).Return(nil).Once()\n\ts.mockArchivalMetadata.On(\"GetVisibilityConfig\").Return(archiver.NewDisabledArchvialConfig())\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessCancelExecution_Pending() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, _ := test.AddRequestCancelInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t)\n\tnextEventID := event.ID\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CancelExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttransferTask.GetDomainID(),\n\t\ttransferTask.GetWorkflowID(),\n\t\ttransferTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.discardDuration))\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessCancelExecution_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, _ := test.AddRequestCancelInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t)\n\tevent = test.AddCancelRequestedEvent(mutableState, event.ID, constants.TestDomainID, targetExecution.GetWorkflowID(), targetExecution.GetRunID())\n\tmutableState.FlushBufferedEvents()\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.CancelExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessSignalExecution_Pending() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, _ := test.AddRequestSignalInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t\t\"some random signal name\", nil, nil,\n\t)\n\tnextEventID := event.ID\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.SignalExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttransferTask.GetDomainID(),\n\t\ttransferTask.GetWorkflowID(),\n\t\ttransferTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.discardDuration))\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessSignalExecution_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\ttargetExecution := types.WorkflowExecution{\n\t\tWorkflowID: \"some random target workflow ID\",\n\t\tRunID:      uuid.New(),\n\t}\n\n\tevent, _ := test.AddRequestSignalInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\ttargetExecution.GetWorkflowID(),\n\t\ttargetExecution.GetRunID(),\n\t\t\"some random signal name\", nil, nil,\n\t)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.SignalExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: targetExecution.GetWorkflowID(),\n\t\tTargetRunID:      targetExecution.GetRunID(),\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tevent = test.AddSignaledEvent(mutableState, event.ID, constants.TestDomainName, targetExecution.GetWorkflowID(), targetExecution.GetRunID(), nil)\n\tmutableState.FlushBufferedEvents()\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessStartChildExecution_Pending() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\tchildWorkflowID := \"some random child workflow ID\"\n\n\tevent, _ := test.AddStartChildWorkflowExecutionInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\tchildWorkflowID,\n\t\t\"some random child workflow type\",\n\t\t\"some random child task list\",\n\t\tnil, 1, 1, nil,\n\t)\n\tnextEventID := event.ID\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.StartChildExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: childWorkflowID,\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.fetchHistoryDuration))\n\ts.mockNDCHistoryResender.EXPECT().SendSingleWorkflowHistory(\n\t\ts.clusterName,\n\t\ttransferTask.GetDomainID(),\n\t\ttransferTask.GetWorkflowID(),\n\t\ttransferTask.GetRunID(),\n\t\tcommon.Int64Ptr(nextEventID),\n\t\tcommon.Int64Ptr(s.version),\n\t\tnil,\n\t\tnil,\n\t).Return(nil).Times(1)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.True(isRedispatchErr(err))\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now.Add(s.discardDuration))\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Equal(ErrTaskDiscarded, err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessStartChildExecution_Success() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tchildWorkflowID := \"some random child workflow ID\"\n\tchildWorkflowType := \"some random child workflow type\"\n\tevent, childInfo := test.AddStartChildWorkflowExecutionInitiatedEvent(\n\t\tmutableState,\n\t\tdecisionCompletionID,\n\t\tuuid.New(),\n\t\tconstants.TestDomainName,\n\t\tchildWorkflowID,\n\t\tchildWorkflowType,\n\t\t\"some random child task list\",\n\t\tnil, 1, 1, nil,\n\t)\n\n\tevent = test.AddChildWorkflowExecutionStartedEvent(mutableState, event.ID, constants.TestDomainName, childWorkflowID, uuid.New(), childWorkflowType)\n\tmutableState.FlushBufferedEvents()\n\tchildInfo.StartedID = event.ID\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.StartChildExecutionTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t\tTargetDomainID:   constants.TestDomainID,\n\t\tTargetWorkflowID: childWorkflowID,\n\t\tInitiatedID:      event.ID,\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, event.ID, event.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessRecordWorkflowStartedTask() {\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.CronSchedule = \"@every 5s\"\n\texecutionInfo.ActiveClusterSelectionPolicy = &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\tstartEvent.WorkflowExecutionStartedEventAttributes.FirstDecisionTaskBackoffSeconds = common.Int32Ptr(5)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tif s.mockShard.GetConfig().EnableRecordWorkflowExecutionUninitialized(s.domainName) {\n\t\ts.mockVisibilityMgr.On(\n\t\t\t\"RecordWorkflowExecutionUninitialized\",\n\t\t\tmock.Anything,\n\t\t\tcreateRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState, s.mockShard.GetTimeSource().Now(), 1234),\n\t\t).Once().Return(nil)\n\t}\n\ts.mockVisibilityMgr.On(\n\t\t\"RecordWorkflowExecutionStarted\",\n\t\tmock.Anything,\n\t\tcreateRecordWorkflowExecutionStartedRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\tfalse),\n\t).Return(nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessRecordWorkflowStartedTaskWithContextHeader() {\n\t// switch on context header in viz\n\ts.mockShard.GetConfig().EnableContextHeaderInVisibility = func(domain string) bool {\n\t\treturn true\n\t}\n\ts.mockShard.GetConfig().ValidSearchAttributes = func(opts ...dc.FilterOption) map[string]interface{} {\n\t\treturn map[string]interface{}{\n\t\t\t\"Header_context_key\": struct{}{},\n\t\t}\n\t}\n\n\tworkflowExecution, mutableState, err := test.StartWorkflow(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.CronSchedule = \"@every 5s\"\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\tstartEvent.WorkflowExecutionStartedEventAttributes.FirstDecisionTaskBackoffSeconds = common.Int32Ptr(5)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.RecordWorkflowStartedTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, startEvent.ID, startEvent.Version)\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tif s.mockShard.GetConfig().EnableRecordWorkflowExecutionUninitialized(s.domainName) {\n\t\ts.mockVisibilityMgr.On(\n\t\t\t\"RecordWorkflowExecutionUninitialized\",\n\t\t\tmock.Anything,\n\t\t\tcreateRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState, s.mockShard.GetTimeSource().Now(), 1234),\n\t\t).Once().Return(nil)\n\t}\n\ts.mockVisibilityMgr.On(\n\t\t\"RecordWorkflowExecutionStarted\",\n\t\tmock.Anything,\n\t\tcreateRecordWorkflowExecutionStartedRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\ttrue),\n\t).Return(nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessUpsertWorkflowSearchAttributesTask() {\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\texecutionInfo := mutableState.GetExecutionInfo()\n\texecutionInfo.ActiveClusterSelectionPolicy = &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: \"region\",\n\t\t\tName:  \"us-west\",\n\t\t},\n\t}\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\ts.mockVisibilityMgr.On(\n\t\t\"UpsertWorkflowExecution\",\n\t\tmock.Anything,\n\t\tcreateUpsertWorkflowSearchAttributesRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\tfalse),\n\t).Return(nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) TestProcessUpsertWorkflowSearchAttributesTaskWithContextHeader() {\n\t// switch on context header in viz\n\ts.mockShard.GetConfig().EnableContextHeaderInVisibility = func(domain string) bool {\n\t\treturn true\n\t}\n\ts.mockShard.GetConfig().ValidSearchAttributes = func(opts ...dc.FilterOption) map[string]interface{} {\n\t\treturn map[string]interface{}{\n\t\t\t\"Header_context_key\": struct{}{},\n\t\t}\n\t}\n\n\tworkflowExecution, mutableState, decisionCompletionID, err := test.SetupWorkflowWithCompletedDecision(s.T(), s.mockShard, s.domainID)\n\ts.NoError(err)\n\n\tnow := time.Now()\n\ttransferTask := s.newTransferTaskFromInfo(&persistence.UpsertWorkflowSearchAttributesTask{\n\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\tDomainID:   s.domainID,\n\t\t\tWorkflowID: workflowExecution.GetWorkflowID(),\n\t\t\tRunID:      workflowExecution.GetRunID(),\n\t\t},\n\t\tTaskData: persistence.TaskData{\n\t\t\tVersion:             s.version,\n\t\t\tVisibilityTimestamp: now,\n\t\t\tTaskID:              int64(59),\n\t\t},\n\t})\n\n\tpersistenceMutableState, err := test.CreatePersistenceMutableState(s.T(), mutableState, decisionCompletionID, mutableState.GetCurrentVersion())\n\ts.NoError(err)\n\ts.mockExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)\n\tstartEvent, err := mutableState.GetStartEvent(context.Background())\n\ts.NoError(err)\n\ts.mockVisibilityMgr.On(\n\t\t\"UpsertWorkflowExecution\",\n\t\tmock.Anything,\n\t\tcreateUpsertWorkflowSearchAttributesRequest(\n\t\t\ts.T(),\n\t\t\ts.domainName, startEvent, transferTask, mutableState, 2, s.mockShard.GetTimeSource().Now(),\n\t\t\ttrue),\n\t).Return(nil).Once()\n\n\ts.mockShard.SetCurrentTime(s.clusterName, now)\n\t_, err = s.transferStandbyTaskExecutor.Execute(transferTask)\n\ts.Nil(err)\n}\n\nfunc (s *transferStandbyTaskExecutorSuite) newTransferTaskFromInfo(\n\ttask persistence.Task,\n) Task {\n\treturn NewHistoryTask(s.mockShard, task, QueueTypeStandbyTransfer, s.logger, nil, nil, nil, nil, nil)\n}\n"
  },
  {
    "path": "service/history/task/transfer_task_executor_base.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/visibility\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n)\n\nconst (\n\ttaskDefaultTimeout             = 3 * time.Second\n\ttaskGetExecutionContextTimeout = 500 * time.Millisecond\n\ttaskRPCCallTimeout             = 2 * time.Second\n\tsecondsInDay                   = int32(24 * time.Hour / time.Second)\n\tdefaultDomainName              = \"defaultDomainName\"\n)\n\ntype (\n\ttransferTaskExecutorBase struct {\n\t\tshard          shard.Context\n\t\tarchiverClient archiver.Client\n\t\texecutionCache execution.Cache\n\t\tlogger         log.Logger\n\t\tmetricsClient  metrics.Client\n\t\tmatchingClient matching.Client\n\t\tvisibilityMgr  persistence.VisibilityManager\n\t\tconfig         *config.Config\n\t\tthrottleRetry  *backoff.ThrottleRetry\n\t}\n)\n\nfunc newTransferTaskExecutorBase(\n\tshard shard.Context,\n\tarchiverClient archiver.Client,\n\texecutionCache execution.Cache,\n\tlogger log.Logger,\n\tconfig *config.Config,\n) *transferTaskExecutorBase {\n\treturn &transferTaskExecutorBase{\n\t\tshard:          shard,\n\t\tarchiverClient: archiverClient,\n\t\texecutionCache: executionCache,\n\t\tlogger:         logger,\n\t\tmetricsClient:  shard.GetMetricsClient(),\n\t\tmatchingClient: shard.GetService().GetMatchingClient(),\n\t\tvisibilityMgr:  shard.GetService().GetVisibilityManager(),\n\t\tconfig:         config,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(taskRetryPolicy),\n\t\t\tbackoff.WithRetryableError(common.IsServiceTransientError),\n\t\t),\n\t}\n}\n\nfunc (t *transferTaskExecutorBase) pushActivity(\n\tctx context.Context,\n\ttask *persistence.ActivityTask,\n\tpushActivityInfo *pushActivityToMatchingInfo,\n) error {\n\n\tctx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\n\tif task.GetTaskType() != persistence.TransferTaskTypeActivityTask {\n\t\tt.logger.Fatal(\"Cannot process non activity task\", tag.TaskType(task.GetTaskType()))\n\t}\n\n\tshouldPush, err := shouldPushToMatching(ctx, t.shard, task)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !shouldPush {\n\t\t// The domain is an active-active domain and the task is pending\n\t\t// For the first few minutes, we will retry the task and then discard it if it is still pending.\n\t\tif t.shard.GetTimeSource().Now().Before(task.GetVisibilityTimestamp().Add(t.config.StandbyTaskMissingEventsDiscardDelay())) {\n\t\t\terr = standbyTaskPostActionNoOp(ctx, task, pushActivityInfo, t.logger)\n\t\t\treturn err\n\t\t}\n\t\treturn standbyTaskPostActionTaskDiscarded(ctx, task, pushActivityInfo, t.logger)\n\t}\n\n\tactivityScheduleToStartTimeout := min(pushActivityInfo.activityScheduleToStartTimeout, constants.MaxTaskTimeout)\n\ttaskList := &pushActivityInfo.tasklist\n\tpartitionConfig := pushActivityInfo.partitionConfig\n\t_, err = t.matchingClient.AddActivityTask(ctx, &types.AddActivityTaskRequest{\n\t\tDomainUUID:       task.TargetDomainID,\n\t\tSourceDomainUUID: task.DomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t},\n\t\tTaskList:                      taskList,\n\t\tScheduleID:                    task.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(activityScheduleToStartTimeout),\n\t\tPartitionConfig:               partitionConfig,\n\t})\n\treturn err\n}\n\nfunc (t *transferTaskExecutorBase) pushDecision(\n\tctx context.Context,\n\ttask *persistence.DecisionTask,\n\tpushDecisionInfo *pushDecisionToMatchingInfo,\n) error {\n\n\tctx, cancel := context.WithTimeout(ctx, taskRPCCallTimeout)\n\tdefer cancel()\n\n\tif task.GetTaskType() != persistence.TransferTaskTypeDecisionTask {\n\t\tt.logger.Fatal(\"Cannot process non decision task\", tag.TaskType(task.GetTaskType()))\n\t}\n\n\tshouldPush, err := shouldPushToMatching(ctx, t.shard, task)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !shouldPush {\n\t\t// The domain is an active-active domain and the task is pending\n\t\t// For the first few minutes, we will retry the task and then discard it if it is still pending.\n\t\tif t.shard.GetTimeSource().Now().Before(task.GetVisibilityTimestamp().Add(t.config.StandbyTaskMissingEventsDiscardDelay())) {\n\t\t\treturn standbyTaskPostActionNoOp(ctx, task, pushDecisionInfo, t.logger)\n\t\t}\n\t\treturn standbyTaskPostActionTaskDiscarded(ctx, task, pushDecisionInfo, t.logger)\n\t}\n\n\tdecisionScheduleToStartTimeout := min(pushDecisionInfo.decisionScheduleToStartTimeout, constants.MaxTaskTimeout)\n\ttasklist := &pushDecisionInfo.tasklist\n\tpartitionConfig := pushDecisionInfo.partitionConfig\n\t_, err = t.matchingClient.AddDecisionTask(ctx, &types.AddDecisionTaskRequest{\n\t\tDomainUUID: task.DomainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: task.WorkflowID,\n\t\t\tRunID:      task.RunID,\n\t\t},\n\t\tTaskList:                      tasklist,\n\t\tScheduleID:                    task.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(decisionScheduleToStartTimeout),\n\t\tPartitionConfig:               partitionConfig,\n\t})\n\treturn err\n}\n\nfunc (t *transferTaskExecutorBase) recordWorkflowStarted(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tworkflowTypeName string,\n\tstartTimeUnixNano int64,\n\texecutionTimeUnixNano int64,\n\tworkflowTimeout int32,\n\ttaskID int64,\n\ttaskList string,\n\tisCron bool,\n\tnumClusters int16,\n\tvisibilityMemo *types.Memo,\n\tupdateTimeUnixNano int64,\n\timmutableSearchAttributes map[string][]byte,\n\theaders map[string][]byte,\n\tclusterAttribute *types.ClusterAttribute,\n\tcronSchedule string,\n\texecutionStatus types.WorkflowExecutionStatus,\n\tscheduledExecutionTimeUnixNano int64,\n) error {\n\n\tdomain := defaultDomainName\n\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\t} else {\n\t\tdomain = domainEntry.GetInfo().Name\n\t\t// if sampled for longer retention is enabled, only record those sampled events\n\t\tif domainEntry.IsSampledForLongerRetentionEnabled(workflowID) &&\n\t\t\t!domainEntry.IsSampledForLongerRetention(workflowID) {\n\t\t\treturn nil\n\t\t}\n\t}\n\n\t// headers are appended into search attributes if enabled\n\tsearchAttributes := copySearchAttributes(immutableSearchAttributes)\n\tif t.config.EnableContextHeaderInVisibility(domainEntry.GetInfo().Name) {\n\t\t// fail open, if error occurs, just log it; successfully appended headers will be stored\n\t\tif searchAttributes, err = appendContextHeaderToSearchAttributes(searchAttributes, headers, t.config.ValidSearchAttributes(), t.config.SearchAttributesHiddenValueKeys()); err != nil {\n\t\t\tt.logger.Error(\"failed to add headers to search attributes\", tag.Error(err))\n\t\t}\n\t}\n\n\trequest := &persistence.RecordWorkflowExecutionStartedRequest{\n\t\tDomainUUID: domainID,\n\t\tDomain:     domain,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tWorkflowTypeName:            workflowTypeName,\n\t\tStartTimestamp:              startTimeUnixNano,\n\t\tExecutionTimestamp:          executionTimeUnixNano,\n\t\tWorkflowTimeout:             int64(workflowTimeout),\n\t\tTaskID:                      taskID,\n\t\tMemo:                        visibilityMemo,\n\t\tTaskList:                    taskList,\n\t\tIsCron:                      isCron,\n\t\tCronSchedule:                cronSchedule,\n\t\tNumClusters:                 numClusters,\n\t\tClusterAttributeScope:       clusterAttribute.GetScope(),\n\t\tClusterAttributeName:        clusterAttribute.GetName(),\n\t\tUpdateTimestamp:             updateTimeUnixNano,\n\t\tSearchAttributes:            searchAttributes,\n\t\tShardID:                     int16(t.shard.GetShardID()),\n\t\tExecutionStatus:             executionStatus,\n\t\tScheduledExecutionTimestamp: scheduledExecutionTimeUnixNano,\n\t}\n\n\tif t.config.EnableRecordWorkflowExecutionUninitialized(domain) {\n\t\tuninitializedRequest := &persistence.RecordWorkflowExecutionUninitializedRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tDomain:     domain,\n\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tWorkflowTypeName: workflowTypeName,\n\t\t\tUpdateTimestamp:  updateTimeUnixNano,\n\t\t\tShardID:          int64(t.shard.GetShardID()),\n\t\t}\n\t\tif err := t.visibilityMgr.RecordWorkflowExecutionUninitialized(ctx, uninitializedRequest); err != nil {\n\t\t\tt.logger.Error(\"Failed to record uninitialized workflow execution\", tag.Error(err))\n\t\t}\n\t}\n\n\treturn t.visibilityMgr.RecordWorkflowExecutionStarted(ctx, request)\n}\n\nfunc (t *transferTaskExecutorBase) upsertWorkflowExecution(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tworkflowTypeName string,\n\tstartTimeUnixNano int64,\n\texecutionTimeUnixNano int64,\n\tworkflowTimeout int32,\n\ttaskID int64,\n\ttaskList string,\n\tvisibilityMemo *types.Memo,\n\tisCron bool,\n\tnumClusters int16,\n\tupdateTimeUnixNano int64,\n\timmutableSearchAttributes map[string][]byte,\n\theaders map[string][]byte,\n\tclusterAttribute *types.ClusterAttribute,\n\tcronSchedule string,\n\texecutionStatus types.WorkflowExecutionStatus,\n\tscheduledExecutionTimeUnixNano int64,\n) error {\n\n\tdomain, err := t.shard.GetDomainCache().GetDomainName(domainID)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn err\n\t\t}\n\t\tdomain = defaultDomainName\n\t}\n\n\t// headers are appended into search attributes if enabled\n\tsearchAttributes := copySearchAttributes(immutableSearchAttributes)\n\tif t.config.EnableContextHeaderInVisibility(domain) {\n\t\t// fail open, if error occurs, just log it; successfully appended headers will be stored\n\t\tif searchAttributes, err = appendContextHeaderToSearchAttributes(searchAttributes, headers, t.config.ValidSearchAttributes(), t.config.SearchAttributesHiddenValueKeys()); err != nil {\n\t\t\tt.logger.Error(\"failed to add headers to search attributes\", tag.Error(err))\n\t\t}\n\t}\n\n\trequest := &persistence.UpsertWorkflowExecutionRequest{\n\t\tDomainUUID: domainID,\n\t\tDomain:     domain,\n\t\tExecution: types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tWorkflowTypeName:            workflowTypeName,\n\t\tStartTimestamp:              startTimeUnixNano,\n\t\tExecutionTimestamp:          executionTimeUnixNano,\n\t\tWorkflowTimeout:             int64(workflowTimeout),\n\t\tTaskID:                      taskID,\n\t\tMemo:                        visibilityMemo,\n\t\tTaskList:                    taskList,\n\t\tIsCron:                      isCron,\n\t\tNumClusters:                 numClusters,\n\t\tClusterAttributeScope:       clusterAttribute.GetScope(),\n\t\tClusterAttributeName:        clusterAttribute.GetName(),\n\t\tSearchAttributes:            searchAttributes,\n\t\tUpdateTimestamp:             updateTimeUnixNano,\n\t\tShardID:                     int64(t.shard.GetShardID()),\n\t\tExecutionStatus:             executionStatus,\n\t\tCronSchedule:                cronSchedule,\n\t\tScheduledExecutionTimestamp: scheduledExecutionTimeUnixNano,\n\t}\n\n\treturn t.visibilityMgr.UpsertWorkflowExecution(ctx, request)\n}\n\nfunc (t *transferTaskExecutorBase) recordWorkflowClosed(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n\tworkflowTypeName string,\n\tstartTimeUnixNano int64,\n\texecutionTimeUnixNano int64,\n\tendTimeUnixNano int64,\n\tcloseStatus types.WorkflowExecutionCloseStatus,\n\thistoryLength int64,\n\ttaskID int64,\n\tvisibilityMemo *types.Memo,\n\ttaskList string,\n\tisCron bool,\n\tcronSchedule string,\n\tnumClusters int16,\n\tupdateTimeUnixNano int64,\n\timmutableSearchAttributes map[string][]byte,\n\theaders map[string][]byte,\n\tclusterAttribute *types.ClusterAttribute,\n\texecutionStatus types.WorkflowExecutionStatus,\n\tscheduledExecutionTimeUnixNano int64,\n) error {\n\n\t// Record closing in visibility store\n\tretentionSeconds := int64(0)\n\tdomain := defaultDomainName\n\trecordWorkflowClose := true\n\tarchiveVisibility := false\n\n\tdomainEntry, err := t.shard.GetDomainCache().GetDomainByID(domainID)\n\tif err != nil && !isWorkflowNotExistError(err) {\n\t\treturn err\n\t}\n\n\tif err == nil {\n\t\t// retention in domain config is in days, convert to seconds\n\t\tretentionSeconds = int64(domainEntry.GetRetentionDays(workflowID)) * int64(secondsInDay)\n\t\tdomain = domainEntry.GetInfo().Name\n\t\t// if sampled for longer retention is enabled, only record those sampled events\n\t\tif domainEntry.IsSampledForLongerRetentionEnabled(workflowID) &&\n\t\t\t!domainEntry.IsSampledForLongerRetention(workflowID) {\n\t\t\trecordWorkflowClose = false\n\t\t}\n\n\t\tclusterConfiguredForVisibilityArchival := t.shard.GetService().GetArchivalMetadata().GetVisibilityConfig().ClusterConfiguredForArchival()\n\t\tdomainConfiguredForVisibilityArchival := domainEntry.GetConfig().VisibilityArchivalStatus == types.ArchivalStatusEnabled\n\t\tarchiveVisibility = clusterConfiguredForVisibilityArchival && domainConfiguredForVisibilityArchival\n\t}\n\n\t// headers are appended into search attributes if enabled\n\tsearchAttributes := copySearchAttributes(immutableSearchAttributes)\n\tif t.config.EnableContextHeaderInVisibility(domainEntry.GetInfo().Name) {\n\t\t// fail open, if error occurs, just log it; successfully appended headers will be stored\n\t\tif searchAttributes, err = appendContextHeaderToSearchAttributes(searchAttributes, headers, t.config.ValidSearchAttributes(), t.config.SearchAttributesHiddenValueKeys()); err != nil {\n\t\t\tt.logger.Error(\"failed to add headers to search attributes\", tag.Error(err))\n\t\t}\n\t}\n\n\tif recordWorkflowClose {\n\t\tif err := t.visibilityMgr.RecordWorkflowExecutionClosed(ctx, &persistence.RecordWorkflowExecutionClosedRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tDomain:     domain,\n\t\t\tExecution: types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tWorkflowTypeName:            workflowTypeName,\n\t\t\tStartTimestamp:              startTimeUnixNano,\n\t\t\tExecutionTimestamp:          executionTimeUnixNano,\n\t\t\tCloseTimestamp:              endTimeUnixNano,\n\t\t\tStatus:                      closeStatus,\n\t\t\tHistoryLength:               historyLength,\n\t\t\tRetentionSeconds:            retentionSeconds,\n\t\t\tTaskID:                      taskID,\n\t\t\tMemo:                        visibilityMemo,\n\t\t\tTaskList:                    taskList,\n\t\t\tSearchAttributes:            searchAttributes,\n\t\t\tIsCron:                      isCron,\n\t\t\tCronSchedule:                cronSchedule,\n\t\t\tClusterAttributeScope:       clusterAttribute.GetScope(),\n\t\t\tClusterAttributeName:        clusterAttribute.GetName(),\n\t\t\tUpdateTimestamp:             updateTimeUnixNano,\n\t\t\tNumClusters:                 numClusters,\n\t\t\tShardID:                     int16(t.shard.GetShardID()),\n\t\t\tExecutionStatus:             executionStatus,\n\t\t\tScheduledExecutionTimestamp: scheduledExecutionTimeUnixNano,\n\t\t}); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif archiveVisibility {\n\t\tarchiveCtx, cancel := context.WithTimeout(ctx, t.config.TransferProcessorVisibilityArchivalTimeLimit())\n\t\tdefer cancel()\n\t\t_, err := t.archiverClient.Archive(archiveCtx, &archiver.ClientRequest{\n\t\t\tArchiveRequest: &archiver.ArchiveRequest{\n\t\t\t\tDomainID:           domainID,\n\t\t\t\tDomainName:         domain,\n\t\t\t\tWorkflowID:         workflowID,\n\t\t\t\tRunID:              runID,\n\t\t\t\tWorkflowTypeName:   workflowTypeName,\n\t\t\t\tStartTimestamp:     startTimeUnixNano,\n\t\t\t\tExecutionTimestamp: executionTimeUnixNano,\n\t\t\t\tCloseTimestamp:     endTimeUnixNano,\n\t\t\t\tCloseStatus:        closeStatus,\n\t\t\t\tHistoryLength:      historyLength,\n\t\t\t\tMemo:               visibilityMemo,\n\t\t\t\tSearchAttributes:   searchAttributes,\n\t\t\t\tVisibilityURI:      domainEntry.GetConfig().VisibilityArchivalURI,\n\t\t\t\tURI:                domainEntry.GetConfig().HistoryArchivalURI,\n\t\t\t\tTargets:            []archiver.ArchivalTarget{archiver.ArchiveTargetVisibility},\n\t\t\t},\n\t\t\tCallerService:        service.History,\n\t\t\tAttemptArchiveInline: true, // archive visibility inline by default\n\t\t})\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// Argument startEvent is to save additional call of msBuilder.GetStartEvent\nfunc getWorkflowExecutionTimestamp(\n\tmsBuilder execution.MutableState,\n\tstartEvent *types.HistoryEvent,\n) time.Time {\n\t// Use value 0 to represent workflows that don't need backoff. Since ES doesn't support\n\t// comparison between two field, we need a value to differentiate them from cron workflows\n\t// or later runs of a workflow that needs retry.\n\texecutionTimestamp := time.Unix(0, 0)\n\tif startEvent == nil {\n\t\treturn executionTimestamp\n\t}\n\n\tif backoffSeconds := startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds(); backoffSeconds != 0 {\n\t\tstartTimestamp := time.Unix(0, startEvent.GetTimestamp())\n\t\texecutionTimestamp = startTimestamp.Add(time.Duration(backoffSeconds) * time.Second)\n\t}\n\treturn executionTimestamp\n}\n\nfunc getWorkflowMemo(\n\tmemo map[string][]byte,\n) *types.Memo {\n\n\tif memo == nil {\n\t\treturn nil\n\t}\n\treturn &types.Memo{Fields: memo}\n}\n\n// context headers are appended to search attributes if in allow list; return errors when all context key is processed\nfunc appendContextHeaderToSearchAttributes(attr, context map[string][]byte, allowedKeys map[string]interface{}, hiddenValueKeys map[string]interface{}) (map[string][]byte, error) {\n\t// sanity check\n\tif attr == nil {\n\t\tattr = make(map[string][]byte)\n\t}\n\tvar errGroup error\n\tfor k, v := range context {\n\t\tsanitizedKey, err := visibility.SanitizeSearchAttributeKey(k)\n\t\tif err != nil { // This could never happen\n\t\t\tmultierr.Append(errGroup, fmt.Errorf(\"fail to sanitize context key %s: %w\", k, err))\n\t\t\tcontinue\n\t\t}\n\t\tkey := fmt.Sprintf(definition.HeaderFormat, sanitizedKey)\n\t\tif _, ok := attr[key]; ok { // skip if key already exists\n\t\t\tcontinue\n\t\t}\n\t\tif _, allowed := allowedKeys[key]; !allowed { // skip if not allowed\n\t\t\tcontinue\n\t\t}\n\t\t// context header are raw string bytes, need to be json encoded to be stored in search attributes\n\t\t// ignore error as it can't happen to err on json encoding string\n\t\t// context header can contain sensitive information, so we need to hide the value if it is in the hiddenValueKeys\n\t\tvar val string\n\t\tif shouldRedactContextHeader(key, hiddenValueKeys) {\n\t\t\tval = \"***redacted***\" // Hide the actualvalue\n\t\t} else {\n\t\t\tval = string(v) // Convert []byte to string safely\n\t\t}\n\t\tdata, _ := json.Marshal(val)\n\t\tattr[key] = data\n\t}\n\treturn attr, errGroup\n}\n\nfunc getWorkflowHeaders(startEvent *types.HistoryEvent) map[string][]byte {\n\tattr := startEvent.GetWorkflowExecutionStartedEventAttributes()\n\tif attr == nil || attr.Header == nil {\n\t\treturn nil\n\t}\n\theaders := make(map[string][]byte, len(attr.Header.Fields))\n\tfor k, v := range attr.Header.Fields {\n\t\tval := make([]byte, len(v))\n\t\tcopy(val, v)\n\t\theaders[k] = val\n\t}\n\treturn headers\n}\n\nfunc copySearchAttributes(\n\tinput map[string][]byte,\n) map[string][]byte {\n\n\tif input == nil {\n\t\treturn nil\n\t}\n\n\tresult := make(map[string][]byte)\n\tfor k, v := range input {\n\t\tval := make([]byte, len(v))\n\t\tcopy(val, v)\n\t\tresult[k] = val\n\t}\n\treturn result\n}\n\nfunc isWorkflowNotExistError(err error) bool {\n\t_, ok := err.(*types.EntityNotExistsError)\n\treturn ok\n}\n\nfunc shouldRedactContextHeader(key string, hiddenValueKeys map[string]interface{}) bool {\n\tif hiddenValueKeys == nil {\n\t\treturn false\n\t}\n\tif val, exists := hiddenValueKeys[key]; exists {\n\t\tswitch v := val.(type) {\n\t\tcase bool:\n\t\t\treturn v\n\t\tcase string:\n\t\t\treturn strings.ToLower(v) == \"true\"\n\t\tcase int:\n\t\t\treturn v > 0 // 1 means true, 0 means false\n\t\t}\n\t}\n\treturn false\n}\n\n// determineExecutionStatus determines whether a workflow execution is PENDING or STARTED\n// based on whether it has a first decision task backoff and if the decision task has been scheduled.\n//\n// A workflow is PENDING if:\n// - It has a firstDecisionTaskBackoffSeconds > 0, AND\n// - No decision task has been scheduled/processed yet\n//\n// Otherwise, it's STARTED.\n//\n// Returns:\n// - executionStatus: either WorkflowExecutionStatusPending or WorkflowExecutionStatusStarted\n// - scheduledExecutionTimestamp: startTime + firstDecisionTaskBackoffSeconds (in nanoseconds)\nfunc determineExecutionStatus(\n\tstartEvent *types.HistoryEvent,\n\tmutableState execution.MutableState,\n) (executionStatus types.WorkflowExecutionStatus, scheduledExecutionTimestamp int64) {\n\texecutionStatus = types.WorkflowExecutionStatusStarted\n\tbackoffSeconds := int32(0)\n\n\tif startEvent.WorkflowExecutionStartedEventAttributes != nil {\n\t\tbackoffSeconds = startEvent.WorkflowExecutionStartedEventAttributes.GetFirstDecisionTaskBackoffSeconds()\n\t}\n\n\tif backoffSeconds > 0 {\n\t\thasPending := mutableState.HasPendingDecision()\n\t\thasInFlight := mutableState.HasInFlightDecision()\n\t\thasProcessed := mutableState.HasProcessedOrPendingDecision()\n\n\t\t// Check if the first decision task has been scheduled yet\n\t\t// If there's no decision info, the workflow is still pending\n\t\tif !hasPending && !hasInFlight && !hasProcessed {\n\t\t\texecutionStatus = types.WorkflowExecutionStatusPending\n\t\t}\n\t}\n\n\t// startTime + firstDecisionTaskBackoffSeconds\n\tscheduledExecutionTimestamp = startEvent.GetTimestamp() +\n\t\tint64(backoffSeconds)*int64(time.Second)\n\n\treturn executionStatus, scheduledExecutionTimestamp\n}\n\nfunc determineExecutionStatusForVisibility(\n\tstartEvent *types.HistoryEvent,\n\tmutableState execution.MutableState,\n\tisAdvancedVisibilityEnabled bool,\n) (executionStatus types.WorkflowExecutionStatus, scheduledExecutionTimestamp int64) {\n\texecutionStatus, scheduledExecutionTimestamp = determineExecutionStatus(startEvent, mutableState)\n\n\t// For basic DB visibility, always use STARTED status\n\t// Basic visibility cannot be updated mid-execution, so showing PENDING would be\n\t// misleading as the workflow would appear to be waiting even when it's running.\n\t// With advanced visibility, we can update PENDING -> STARTED via upsert.\n\tif !isAdvancedVisibilityEnabled && executionStatus == types.WorkflowExecutionStatusPending {\n\t\texecutionStatus = types.WorkflowExecutionStatusStarted\n\t}\n\n\treturn executionStatus, scheduledExecutionTimestamp\n}\n"
  },
  {
    "path": "service/history/task/transfer_task_executor_base_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage task\n\nimport (\n\t\"testing\"\n)\n\nfunc TestShouldRedactContextHeader(t *testing.T) {\n\tcases := []struct {\n\t\tkey             string\n\t\thiddenValueKeys map[string]interface{}\n\t\texpected        bool\n\t}{\n\t\t{\n\t\t\tkey: \"key1\",\n\t\t\thiddenValueKeys: map[string]interface{}{\n\t\t\t\t\"key1\": true,\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tkey: \"key2\",\n\t\t\thiddenValueKeys: map[string]interface{}{\n\t\t\t\t\"key1\": true,\n\t\t\t},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tkey:             \"key3\",\n\t\t\thiddenValueKeys: map[string]interface{}{},\n\t\t\texpected:        false,\n\t\t},\n\t\t{\n\t\t\tkey:             \"key4\",\n\t\t\thiddenValueKeys: nil,\n\t\t\texpected:        false,\n\t\t},\n\t\t{\n\t\t\tkey: \"key5\",\n\t\t\thiddenValueKeys: map[string]interface{}{\n\t\t\t\t\"key5\": \"true\",\n\t\t\t},\n\t\t\texpected: true,\n\t\t},\n\t}\n\n\tfor _, c := range cases {\n\t\tresult := shouldRedactContextHeader(c.key, c.hiddenValueKeys)\n\t\tif result != c.expected {\n\t\t\tt.Errorf(\"shouldRedactContextHeader(%s, %v) = %t; expected %t\", c.key, c.hiddenValueKeys, result, c.expected)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/history/templates/ratelimited.tmpl",
    "content": "import (\n    \"context\"\n\n    \"github.com/uber/cadence/common\"\n    \"github.com/uber/cadence/common/cache\"\n    \"github.com/uber/cadence/common/constants\"\n    \"github.com/uber/cadence/common/quotas\"\n    \"github.com/uber/cadence/common/types\"\n    \"github.com/uber/cadence/service/history\"\n    \"github.com/uber/cadence/common/log\"\n)\n\n{{ $ratelimitTypeMap := dict \"StartWorkflowExecution\" (\n    dict\n        \"ratelimit\" \"ratelimitTypeUserPerID\"\n        \"workflowID\" \"StartRequest.GetWorkflowID()\"\n    )\n}}\n{{ $ratelimitTypeMap := set $ratelimitTypeMap \"SignalWithStartWorkflowExecution\" (\n    dict\n        \"ratelimit\" \"ratelimitTypeUserPerID\"\n        \"workflowID\" \"SignalWithStartRequest.GetWorkflowID()\"\n    )\n}}\n{{ $ratelimitTypeMap := set $ratelimitTypeMap \"SignalWorkflowExecution\" (\n    dict\n        \"ratelimit\" \"ratelimitTypeUserPerID\"\n        \"workflowID\" \"SignalRequest.GetWorkflowExecution().GetWorkflowID()\"\n    )\n}}\n{{ $ratelimitTypeMap := set $ratelimitTypeMap \"DescribeWorkflowExecution\" (\n    dict\n        \"ratelimit\" \"ratelimitTypeUserPerID\"\n        \"workflowID\" \"Request.GetExecution().GetWorkflowID()\"\n    )\n}}\n\n{{ $interfaceName := .Interface.Name }}\n{{ $handlerName := (index .Vars \"handler\") }}\n{{ $decorator := (printf \"%s%s\" (down $handlerName) $interfaceName) }}\n{{ $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with rate limiter.\ntype {{$decorator}} struct {\n    wrapped                        {{.Interface.Type}}\n    workflowIDCache                workflowcache.WFCache\n    logger                         log.Logger\n    allowFunc                      func (domainID string, workflowID string) bool\n}\n\n// New{{$Decorator}} creates a new instance of {{$interfaceName}} with ratelimiter.\nfunc New{{$Decorator}}(\n    wrapped {{.Interface.Type}},\n    workflowIDCache workflowcache.WFCache,\n    logger log.Logger,\n) {{.Interface.Type}} {\n    wrapper := &{{$decorator}}{\n        wrapped: wrapped,\n        workflowIDCache: workflowIDCache,\n        logger: logger,\n    }\n    wrapper.allowFunc = wrapper.allowWfID\n\n    return wrapper\n}\n\n{{range $method := .Interface.Methods}}\nfunc (h *{{$decorator}}) {{$method.Declaration}} {\n    {{- if hasKey $ratelimitTypeMap $method.Name }}\n        {{- if eq\n            ( get ( get $ratelimitTypeMap $method.Name ) \"ratelimit\" )\n            \"ratelimitTypeUserPerID\"\n        }}\n\n            {{ $workflowID := ( get ( get $ratelimitTypeMap $method.Name ) \"workflowID\" ) }}\n            {{ $req := (index $method.Params 1).Name }}\n\n            if {{ $req }} == nil {\n                err = validate.ErrRequestNotSet\n                return\n            }\n            {{- $domainID := printf \"%s.GetDomainUUID()\" $req }}\n            {{- $workflowID := printf \"%s.%s\" $req ( $workflowID ) }}\n\n            if {{$domainID}} == \"\" {\n                err = validate.ErrDomainNotSet\n                return\n            }\n\n            if {{$workflowID}} == \"\" {\n                err = validate.ErrWorkflowIDNotSet\n                return\n            }\n\n            if !h.allowFunc({{$domainID}}, {{$workflowID}}) {\n                err = &types.ServiceBusyError{\n                    Message: \"Too many requests for the workflow ID\",\n                    Reason: constants.WorkflowIDRateLimitReason,\n                }\n                return\n            }\n\n        {{- end}}\n    {{- end}}\n    {{$method.Pass \"h.wrapped.\"}}\n}\n{{end}}\n"
  },
  {
    "path": "service/history/testing/events_util.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage testing\n\nimport (\n\t\"github.com/uber/cadence/common\"\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\n// AddWorkflowExecutionStartedEventWithParent adds WorkflowExecutionStarted event with parent workflow info\nfunc AddWorkflowExecutionStartedEventWithParent(\n\tbuilder execution.MutableState,\n\tworkflowExecution types.WorkflowExecution,\n\tworkflowType string,\n\ttaskList string,\n\tinput []byte,\n\texecutionStartToCloseTimeout,\n\ttaskStartToCloseTimeout int32,\n\tparentInfo *types.ParentExecutionInfo,\n\tidentity string,\n\tactiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n) *types.HistoryEvent {\n\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tWorkflowID:                          workflowExecution.WorkflowID,\n\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\tTaskList:                            &types.TaskList{Name: taskList},\n\t\tInput:                               input,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(executionStartToCloseTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(taskStartToCloseTimeout),\n\t\tIdentity:                            identity,\n\t\tActiveClusterSelectionPolicy:        activeClusterSelectionPolicy,\n\t}\n\n\tevent, _ := builder.AddWorkflowExecutionStartedEvent(\n\t\tworkflowExecution,\n\t\t&types.HistoryStartWorkflowExecutionRequest{\n\t\t\tDomainUUID:          constants.TestDomainID,\n\t\t\tStartRequest:        startRequest,\n\t\t\tParentExecutionInfo: parentInfo,\n\t\t},\n\t)\n\n\treturn event\n}\n\n// AddWorkflowExecutionStartedEvent adds WorkflowExecutionStarted event\nfunc AddWorkflowExecutionStartedEvent(\n\tbuilder execution.MutableState,\n\tworkflowExecution types.WorkflowExecution,\n\tworkflowType string,\n\ttaskList string,\n\tinput []byte,\n\texecutionStartToCloseTimeout int32,\n\ttaskStartToCloseTimeout int32,\n\tidentity string,\n\tactiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy,\n) *types.HistoryEvent {\n\treturn AddWorkflowExecutionStartedEventWithParent(builder, workflowExecution, workflowType, taskList, input,\n\t\texecutionStartToCloseTimeout, taskStartToCloseTimeout, nil, identity, activeClusterSelectionPolicy)\n}\n\n// AddDecisionTaskScheduledEvent adds DecisionTaskScheduled event\nfunc AddDecisionTaskScheduledEvent(\n\tbuilder execution.MutableState,\n) *execution.DecisionInfo {\n\tdi, _ := builder.AddDecisionTaskScheduledEvent(false)\n\treturn di\n}\n\n// AddDecisionTaskStartedEvent adds DecisionTaskStarted event\nfunc AddDecisionTaskStartedEvent(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\ttaskList string,\n\tidentity string,\n) *types.HistoryEvent {\n\treturn AddDecisionTaskStartedEventWithRequestID(builder, scheduleID, constants.TestRunID, taskList, identity)\n}\n\n// AddDecisionTaskStartedEventWithRequestID adds DecisionTaskStarted event with requestID\nfunc AddDecisionTaskStartedEventWithRequestID(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\trequestID string,\n\ttaskList string,\n\tidentity string,\n) *types.HistoryEvent {\n\tevent, _, _ := builder.AddDecisionTaskStartedEvent(scheduleID, requestID, &types.PollForDecisionTaskRequest{\n\t\tTaskList: &types.TaskList{Name: taskList},\n\t\tIdentity: identity,\n\t})\n\n\treturn event\n}\n\n// AddDecisionTaskCompletedEvent adds DecisionTaskCompleted event\nfunc AddDecisionTaskCompletedEvent(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\tstartedID int64,\n\tcontext []byte,\n\tidentity string,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddDecisionTaskCompletedEvent(scheduleID, startedID, &types.RespondDecisionTaskCompletedRequest{\n\t\tExecutionContext: context,\n\t\tIdentity:         identity,\n\t}, commonconstants.DefaultHistoryMaxAutoResetPoints)\n\n\tbuilder.FlushBufferedEvents() //nolint:errcheck\n\n\treturn event\n}\n\n// AddActivityTaskScheduledEvent adds ActivityTaskScheduled event\nfunc AddActivityTaskScheduledEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedID int64,\n\tactivityID string,\n\tactivityType string,\n\ttaskList string,\n\tinput []byte,\n\tscheduleToCloseTimeout int32,\n\tscheduleToStartTimeout int32,\n\tstartToCloseTimeout int32,\n\theartbeatTimeout int32,\n) (*types.HistoryEvent,\n\t*persistence.ActivityInfo) {\n\n\tevent, ai, _, _, _, _ := builder.AddActivityTaskScheduledEvent(nil, decisionCompletedID, &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    activityID,\n\t\tActivityType:                  &types.ActivityType{Name: activityType},\n\t\tTaskList:                      &types.TaskList{Name: taskList, Kind: types.TaskListKindNormal.Ptr()},\n\t\tInput:                         input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(scheduleToCloseTimeout),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(scheduleToStartTimeout),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(startToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(heartbeatTimeout),\n\t}, false,\n\t)\n\n\treturn event, ai\n}\n\n// AddActivityTaskScheduledEventWithRetry adds ActivityTaskScheduled event with retry policy\nfunc AddActivityTaskScheduledEventWithRetry(\n\tbuilder execution.MutableState,\n\tdecisionCompletedID int64,\n\tactivityID string,\n\tactivityType string,\n\ttaskList string,\n\tinput []byte,\n\tscheduleToCloseTimeout int32,\n\tscheduleToStartTimeout int32,\n\tstartToCloseTimeout int32,\n\theartbeatTimeout int32,\n\tretryPolicy *types.RetryPolicy,\n) (*types.HistoryEvent, *persistence.ActivityInfo) {\n\n\tevent, ai, _, _, _, _ := builder.AddActivityTaskScheduledEvent(nil, decisionCompletedID, &types.ScheduleActivityTaskDecisionAttributes{\n\t\tActivityID:                    activityID,\n\t\tActivityType:                  &types.ActivityType{Name: activityType},\n\t\tTaskList:                      &types.TaskList{Name: taskList},\n\t\tInput:                         input,\n\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(scheduleToCloseTimeout),\n\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(scheduleToStartTimeout),\n\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(startToCloseTimeout),\n\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(heartbeatTimeout),\n\t\tRetryPolicy:                   retryPolicy,\n\t}, false,\n\t)\n\n\treturn event, ai\n}\n\n// AddActivityTaskStartedEvent adds ActivityTaskStarted event\nfunc AddActivityTaskStartedEvent(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\tidentity string,\n) *types.HistoryEvent {\n\tai, _ := builder.GetActivityInfo(scheduleID)\n\tevent, _ := builder.AddActivityTaskStartedEvent(ai, scheduleID, constants.TestRunID, identity)\n\treturn event\n}\n\n// AddActivityTaskCompletedEvent adds ActivityTaskCompleted event\nfunc AddActivityTaskCompletedEvent(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\tstartedID int64,\n\tresult []byte,\n\tidentity string,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddActivityTaskCompletedEvent(scheduleID, startedID, &types.RespondActivityTaskCompletedRequest{\n\t\tResult:   result,\n\t\tIdentity: identity,\n\t})\n\n\treturn event\n}\n\n// AddActivityTaskFailedEvent adds ActivityTaskFailed event\nfunc AddActivityTaskFailedEvent(\n\tbuilder execution.MutableState,\n\tscheduleID int64,\n\tstartedID int64,\n\treason string,\n\tdetails []byte,\n\tidentity string,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddActivityTaskFailedEvent(scheduleID, startedID, &types.RespondActivityTaskFailedRequest{\n\t\tReason:   common.StringPtr(reason),\n\t\tDetails:  details,\n\t\tIdentity: identity,\n\t})\n\n\treturn event\n}\n\n// AddTimerStartedEvent adds TimerStarted event\nfunc AddTimerStartedEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedEventID int64,\n\ttimerID string,\n\ttimeOut int64,\n) (*types.HistoryEvent, *persistence.TimerInfo) {\n\tevent, ti, _ := builder.AddTimerStartedEvent(decisionCompletedEventID,\n\t\t&types.StartTimerDecisionAttributes{\n\t\t\tTimerID:                   timerID,\n\t\t\tStartToFireTimeoutSeconds: common.Int64Ptr(timeOut),\n\t\t})\n\treturn event, ti\n}\n\n// AddTimerFiredEvent adds TimerFired event\nfunc AddTimerFiredEvent(\n\tmutableState execution.MutableState,\n\ttimerID string,\n) *types.HistoryEvent {\n\tevent, _ := mutableState.AddTimerFiredEvent(timerID)\n\treturn event\n}\n\n// AddRequestCancelInitiatedEvent adds RequestCancelExternalWorkflowExecutionInitiated event\nfunc AddRequestCancelInitiatedEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedEventID int64,\n\tcancelRequestID string,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n) (*types.HistoryEvent, *persistence.RequestCancelInfo) {\n\tevent, rci, _ := builder.AddRequestCancelExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID,\n\t\tcancelRequestID, &types.RequestCancelExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:     domain,\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t})\n\n\treturn event, rci\n}\n\n// AddCancelRequestedEvent adds ExternalWorkflowExecutionCancelRequested event\nfunc AddCancelRequestedEvent(\n\tbuilder execution.MutableState,\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddExternalWorkflowExecutionCancelRequested(initiatedID, domain, workflowID, runID)\n\treturn event\n}\n\n// AddRequestSignalInitiatedEvent adds SignalExternalWorkflowExecutionInitiated event\nfunc AddRequestSignalInitiatedEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedEventID int64,\n\tsignalRequestID string,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tsignalName string,\n\tinput []byte,\n\tcontrol []byte,\n) (*types.HistoryEvent, *persistence.SignalInfo) {\n\tevent, si, _ := builder.AddSignalExternalWorkflowExecutionInitiatedEvent(decisionCompletedEventID, signalRequestID,\n\t\t&types.SignalExternalWorkflowExecutionDecisionAttributes{\n\t\t\tDomain: domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tSignalName: signalName,\n\t\t\tInput:      input,\n\t\t\tControl:    control,\n\t\t})\n\n\treturn event, si\n}\n\n// AddSignaledEvent adds ExternalWorkflowExecutionSignaled event\nfunc AddSignaledEvent(\n\tbuilder execution.MutableState,\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tcontrol []byte,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddExternalWorkflowExecutionSignaled(initiatedID, domain, workflowID, runID, control)\n\treturn event\n}\n\n// AddStartChildWorkflowExecutionInitiatedEvent adds ChildWorkflowExecutionInitiated event\nfunc AddStartChildWorkflowExecutionInitiatedEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedID int64,\n\tcreateRequestID string,\n\tdomain string,\n\tworkflowID string,\n\tworkflowType string,\n\ttasklist string,\n\tinput []byte,\n\texecutionStartToCloseTimeout int32,\n\ttaskStartToCloseTimeout int32,\n\tretryPolicy *types.RetryPolicy,\n) (*types.HistoryEvent,\n\t*persistence.ChildExecutionInfo) {\n\n\tevent, cei, _ := builder.AddStartChildWorkflowExecutionInitiatedEvent(decisionCompletedID, createRequestID,\n\t\t&types.StartChildWorkflowExecutionDecisionAttributes{\n\t\t\tDomain:                              domain,\n\t\t\tWorkflowID:                          workflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: tasklist},\n\t\t\tInput:                               input,\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(executionStartToCloseTimeout),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(taskStartToCloseTimeout),\n\t\t\tControl:                             nil,\n\t\t\tRetryPolicy:                         retryPolicy,\n\t\t})\n\treturn event, cei\n}\n\n// AddChildWorkflowExecutionStartedEvent adds ChildWorkflowExecutionStarted event\nfunc AddChildWorkflowExecutionStartedEvent(\n\tbuilder execution.MutableState,\n\tinitiatedID int64,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tworkflowType string,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddChildWorkflowExecutionStartedEvent(\n\t\tdomain,\n\t\t&types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\t&types.WorkflowType{Name: workflowType},\n\t\tinitiatedID,\n\t\t&types.Header{},\n\t)\n\treturn event\n}\n\n// AddChildWorkflowExecutionCompletedEvent adds ChildWorkflowExecutionCompleted event\nfunc AddChildWorkflowExecutionCompletedEvent(\n\tbuilder execution.MutableState,\n\tinitiatedID int64,\n\tchildExecution *types.WorkflowExecution,\n\tattributes *types.WorkflowExecutionCompletedEventAttributes,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddChildWorkflowExecutionCompletedEvent(initiatedID, childExecution, attributes)\n\treturn event\n}\n\n// AddCompleteWorkflowEvent adds WorkflowExecutionCompleted event\nfunc AddCompleteWorkflowEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedEventID int64,\n\tresult []byte,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddCompletedWorkflowEvent(decisionCompletedEventID, &types.CompleteWorkflowExecutionDecisionAttributes{\n\t\tResult: result,\n\t})\n\treturn event\n}\n\n// AddFailWorkflowEvent adds WorkflowExecutionFailed event\nfunc AddFailWorkflowEvent(\n\tbuilder execution.MutableState,\n\tdecisionCompletedEventID int64,\n\treason string,\n\tdetails []byte,\n) *types.HistoryEvent {\n\tevent, _ := builder.AddFailWorkflowEvent(decisionCompletedEventID, &types.FailWorkflowExecutionDecisionAttributes{\n\t\tReason:  &reason,\n\t\tDetails: details,\n\t})\n\treturn event\n}\n"
  },
  {
    "path": "service/history/testing/workflow_util.go",
    "content": "// Copyright (c) 2021 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testing\n\nimport (\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\n// StartWorkflow setup a workflow for testing purpose\nfunc StartWorkflow(\n\tt *testing.T,\n\tmockShard *shard.TestContext,\n\tsourceDomainID string,\n) (types.WorkflowExecution, execution.MutableState, error) {\n\treturn StartWorkflowWithTaskList(&types.TaskList{\n\t\tName: \"some random task list\",\n\t}, mockShard, sourceDomainID)\n}\n\nfunc StartWorkflowWithTaskList(\n\ttl *types.TaskList,\n\tmockShard *shard.TestContext,\n\tsourceDomainID string,\n) (types.WorkflowExecution, execution.MutableState, error) {\n\tworkflowExecution := types.WorkflowExecution{\n\t\tWorkflowID: constants.TestWorkflowID,\n\t\tRunID:      constants.TestRunID,\n\t}\n\tworkflowType := \"some random workflow type\"\n\n\tentry, err := mockShard.GetDomainCache().GetDomainByID(sourceDomainID)\n\tif err != nil {\n\t\treturn types.WorkflowExecution{}, nil, err\n\t}\n\tversion := entry.GetFailoverVersion()\n\tmutableState := execution.NewMutableStateBuilderWithVersionHistoriesWithEventV2(\n\t\tmockShard,\n\t\tmockShard.GetLogger(),\n\t\tversion,\n\t\tworkflowExecution.GetRunID(),\n\t\tentry,\n\t)\n\t_, err = mutableState.AddWorkflowExecutionStartedEvent(\n\t\tworkflowExecution,\n\t\t&types.HistoryStartWorkflowExecutionRequest{\n\t\t\tDomainUUID: sourceDomainID,\n\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{\n\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: workflowType},\n\t\t\t\tTaskList:                            tl,\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(2),\n\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(1),\n\t\t\t\tHeader: &types.Header{Fields: map[string][]byte{\n\t\t\t\t\t\"context-key\":         []byte(\"contextValue\"),\n\t\t\t\t\t\"123456\":              []byte(\"123456\"), // unsanitizable key\n\t\t\t\t\t\"invalid-context-key\": []byte(\"invalidContextValue\"),\n\t\t\t\t}},\n\t\t\t},\n\t\t\tPartitionConfig: map[string]string{\"userid\": uuid.New()},\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn types.WorkflowExecution{}, nil, err\n\t}\n\n\treturn workflowExecution, mutableState, nil\n}\n\n// SetupWorkflowWithCompletedDecision setup a workflow with a completed decision task for testing purpose\nfunc SetupWorkflowWithCompletedDecision(\n\tt *testing.T,\n\tmockShard *shard.TestContext,\n\tsourceDomainID string,\n) (types.WorkflowExecution, execution.MutableState, int64, error) {\n\tworkflowExecution, mutableState, err := StartWorkflow(t, mockShard, sourceDomainID)\n\tif err != nil {\n\t\treturn types.WorkflowExecution{}, nil, 0, err\n\t}\n\n\tdi := AddDecisionTaskScheduledEvent(mutableState)\n\tevent := AddDecisionTaskStartedEvent(mutableState, di.ScheduleID, mutableState.GetExecutionInfo().TaskList, uuid.New())\n\tdi.StartedID = event.ID\n\tevent = AddDecisionTaskCompletedEvent(mutableState, di.ScheduleID, di.StartedID, nil, \"some random identity\")\n\n\treturn workflowExecution, mutableState, event.ID, nil\n}\n\n// CreatePersistenceMutableState generated a persistence representation of the mutable state\n// a based on the in memory version\nfunc CreatePersistenceMutableState(\n\tt *testing.T,\n\tms execution.MutableState,\n\tlastEventID int64,\n\tlastEventVersion int64,\n) (*persistence.WorkflowMutableState, error) {\n\n\tif ms.GetVersionHistories() != nil {\n\t\tcurrentVersionHistory, err := ms.GetVersionHistories().GetCurrentVersionHistory()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\terr = currentVersionHistory.AddOrUpdateItem(persistence.NewVersionHistoryItem(\n\t\t\tlastEventID,\n\t\t\tlastEventVersion,\n\t\t))\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn execution.CreatePersistenceMutableState(t, ms), nil\n}\n"
  },
  {
    "path": "service/history/workflow/context.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage workflow\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\ntype (\n\t// Context is an helper interface on top of execution.Context\n\tContext interface {\n\t\tGetContext() execution.Context\n\t\tGetMutableState() execution.MutableState\n\t\tReloadMutableState(ctx context.Context) (execution.MutableState, error)\n\t\tGetReleaseFn() execution.ReleaseFunc\n\t\tGetWorkflowID() string\n\t\tGetRunID() string\n\t}\n\n\tcontextImpl struct {\n\t\tcontext      execution.Context\n\t\tmutableState execution.MutableState\n\t\treleaseFn    execution.ReleaseFunc\n\t}\n)\n\n// NewContext creates a new helper instance on top of execution.Context\nfunc NewContext(\n\tcontext execution.Context,\n\treleaseFn execution.ReleaseFunc,\n\tmutableState execution.MutableState,\n) Context {\n\n\treturn &contextImpl{\n\t\tcontext:      context,\n\t\treleaseFn:    releaseFn,\n\t\tmutableState: mutableState,\n\t}\n}\n\nfunc (w *contextImpl) GetContext() execution.Context {\n\treturn w.context\n}\n\nfunc (w *contextImpl) GetMutableState() execution.MutableState {\n\treturn w.mutableState\n}\n\nfunc (w *contextImpl) ReloadMutableState(ctx context.Context) (execution.MutableState, error) {\n\tmutableState, err := w.GetContext().LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tw.mutableState = mutableState\n\treturn mutableState, nil\n}\n\nfunc (w *contextImpl) GetReleaseFn() execution.ReleaseFunc {\n\treturn w.releaseFn\n}\n\nfunc (w *contextImpl) GetWorkflowID() string {\n\treturn w.GetContext().GetExecution().GetWorkflowID()\n}\n\nfunc (w *contextImpl) GetRunID() string {\n\treturn w.GetContext().GetExecution().GetRunID()\n}\n"
  },
  {
    "path": "service/history/workflow/errors.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage workflow\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar (\n\t// ErrStaleState is the error returned during state update indicating that cached mutable state could be stale\n\tErrStaleState = errors.New(\"cache mutable state could potentially be stale\")\n\t// ErrMaxAttemptsExceeded is exported temporarily for integration test\n\tErrMaxAttemptsExceeded = errors.New(\"maximum attempts exceeded to update history\")\n\t// ErrActivityTaskNotFound is the error to indicate activity task could be duplicate and activity already completed\n\tErrActivityTaskNotFound = &types.EntityNotExistsError{Message: \"activity task not found\"}\n\t// ErrNotExists is the error to indicate workflow doesn't exist\n\tErrNotExists = &types.EntityNotExistsError{Message: \"workflow execution already completed\"}\n\t// ErrAlreadyCompleted is the error to indicate workflow execution already completed\n\tErrAlreadyCompleted = &types.WorkflowExecutionAlreadyCompletedError{Message: \"workflow execution already completed\"}\n\t// ErrParentMismatch is the error to parent execution is given and mismatch\n\tErrParentMismatch = &types.EntityNotExistsError{Message: \"workflow parent does not match\"}\n\t// ErrDeserializingToken is the error to indicate task token is invalid\n\tErrDeserializingToken = &types.BadRequestError{Message: \"error deserializing task token\"}\n\t// ErrSerializingToken is the error to indicate task token can not be serialized\n\tErrSerializingToken = &types.BadRequestError{Message: \"error serializing task token\"}\n\t// ErrSignalOverSize is the error to indicate signal input size is > 256K\n\tErrSignalOverSize = &types.BadRequestError{Message: \"signal input size is over 256K\"}\n\t// ErrCancellationAlreadyRequested is the error indicating cancellation for target workflow is already requested\n\tErrCancellationAlreadyRequested = &types.CancellationAlreadyRequestedError{Message: \"cancellation already requested for this workflow execution\"}\n\t// ErrSignalsLimitExceeded is the error indicating limit reached for maximum number of signal events\n\tErrSignalsLimitExceeded = &types.LimitExceededError{Message: \"exceeded workflow execution limit for signal events\"}\n\t// ErrQueryEnteredInvalidState is error indicating query entered invalid state\n\tErrQueryEnteredInvalidState = &types.BadRequestError{Message: \"query entered invalid state, this should be impossible\"}\n\t// ErrQueryWorkflowBeforeFirstDecision is error indicating that query was attempted before first decision task completed\n\tErrQueryWorkflowBeforeFirstDecision = &types.QueryFailedError{Message: \"workflow must handle at least one decision task before it can be queried\"}\n\t// ErrConsistentQueryNotEnabled is error indicating that consistent query was requested but either cluster or domain does not enable consistent query\n\tErrConsistentQueryNotEnabled = &types.BadRequestError{Message: \"cluster or domain does not enable strongly consistent query but strongly consistent query was requested\"}\n\t// ErrConsistentQueryBufferExceeded is error indicating that too many consistent queries have been buffered and until buffered queries are finished new consistent queries cannot be buffered\n\tErrConsistentQueryBufferExceeded = &types.InternalServiceError{Message: \"consistent query buffer is full, cannot accept new consistent queries\"}\n\t// ErrConcurrentStartRequest is error indicating there is an outstanding start workflow request. The incoming request fails to acquires the lock before the outstanding request finishes.\n\tErrConcurrentStartRequest = &types.ServiceBusyError{Message: \"an outstanding start workflow request is in-progress. Failed to acquire the resource.\"}\n)\n"
  },
  {
    "path": "service/history/workflow/util.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage workflow\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n)\n\nvar ConditionalRetryCount = 5\n\ntype (\n\tUpdateAction struct {\n\t\tNoop           bool\n\t\tCreateDecision bool\n\t}\n\n\tUpdateActionFunc func(execution.Context, execution.MutableState) (*UpdateAction, error)\n)\n\nvar (\n\tUpdateWithNewDecision = &UpdateAction{\n\t\tCreateDecision: true,\n\t}\n\tUpdateWithoutDecision = &UpdateAction{\n\t\tCreateDecision: false,\n\t}\n)\n\nfunc LoadOnce(\n\tctx context.Context,\n\tcache execution.Cache,\n\tdomainID string,\n\tworkflowID string,\n\trunID string,\n) (Context, error) {\n\n\twfContext, release, err := cache.GetOrCreateWorkflowExecution(\n\t\tctx,\n\t\tdomainID,\n\t\ttypes.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmutableState, err := wfContext.LoadWorkflowExecution(ctx)\n\tif err != nil {\n\t\trelease(err)\n\t\treturn nil, err\n\t}\n\n\treturn NewContext(wfContext, release, mutableState), nil\n}\n\nfunc Load(\n\tctx context.Context,\n\tcache execution.Cache,\n\texecutionManager persistence.ExecutionManager,\n\tdomainID string,\n\tdomainName string,\n\tworkflowID string,\n\trunID string,\n) (Context, error) {\n\n\tif runID != \"\" {\n\t\treturn LoadOnce(ctx, cache, domainID, workflowID, runID)\n\t}\n\n\tfor attempt := 0; attempt < ConditionalRetryCount; attempt++ {\n\n\t\tworkflowContext, err := LoadOnce(ctx, cache, domainID, workflowID, \"\")\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif workflowContext.GetMutableState().IsWorkflowExecutionRunning() {\n\t\t\treturn workflowContext, nil\n\t\t}\n\n\t\t// workflow not running, need to check current record\n\t\tresp, err := executionManager.GetCurrentExecution(\n\t\t\tctx,\n\t\t\t&persistence.GetCurrentExecutionRequest{\n\t\t\t\tDomainID:   domainID,\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tDomainName: domainName,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\tworkflowContext.GetReleaseFn()(err)\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif resp.RunID == workflowContext.GetRunID() {\n\t\t\treturn workflowContext, nil\n\t\t}\n\t\tworkflowContext.GetReleaseFn()(nil)\n\t}\n\n\treturn nil, &types.InternalServiceError{Message: \"unable to locate current workflow execution\"}\n}\n\n// /////////////////  Util function for updating workflows ///////////////////\n\n// UpdateWithActionFunc updates the given workflow execution.\n// If runID is empty, it only tries to load the current workflow once.\n// If the update should always be applied to the current run, use UpdateCurrentWithActionFunc instead.\nfunc UpdateWithActionFunc(\n\tctx context.Context,\n\tlogger log.Logger,\n\tcache execution.Cache,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\tnow time.Time,\n\taction UpdateActionFunc,\n) (retError error) {\n\n\tworkflowContext, err := LoadOnce(ctx, cache, domainID, execution.GetWorkflowID(), execution.GetRunID())\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer func() { workflowContext.GetReleaseFn()(retError) }()\n\n\treturn updateHelper(ctx, logger, workflowContext, now, action)\n}\n\n// UpdateCurrentWithActionFunc updates the given workflow execution or current execution if runID is empty.\n// It's the same as UpdateWithActionFunc if runID is not empty.\n// This function is suitable for the case when the change should always be applied to the current execution.\nfunc UpdateCurrentWithActionFunc(\n\tctx context.Context,\n\tlogger log.Logger,\n\tcache execution.Cache,\n\texecutionManager persistence.ExecutionManager,\n\tdomainID string,\n\tdomainCache cache.DomainCache,\n\texecution types.WorkflowExecution,\n\tnow time.Time,\n\taction UpdateActionFunc,\n) (retError error) {\n\n\tdomainName, err := domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil\n\t}\n\tworkflowContext, err := Load(ctx, cache, executionManager, domainID, domainName, execution.GetWorkflowID(), execution.GetRunID())\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer func() { workflowContext.GetReleaseFn()(retError) }()\n\n\treturn updateHelper(ctx, logger, workflowContext, now, action)\n}\n\n// TODO: deprecate and use UpdateWithActionFunc\nfunc UpdateWithAction(\n\tctx context.Context,\n\tlogger log.Logger,\n\tcache execution.Cache,\n\tdomainID string,\n\texecution types.WorkflowExecution,\n\tcreateDecisionTask bool,\n\tnow time.Time,\n\taction func(wfContext execution.Context, mutableState execution.MutableState) error,\n) error {\n\n\treturn UpdateWithActionFunc(\n\t\tctx,\n\t\tlogger,\n\t\tcache,\n\t\tdomainID,\n\t\texecution,\n\t\tnow,\n\t\tgetUpdateActionFunc(createDecisionTask, action),\n\t)\n}\n\nfunc getUpdateActionFunc(\n\tcreateDecisionTask bool,\n\taction func(wfContext execution.Context, mutableState execution.MutableState) error,\n) UpdateActionFunc {\n\n\treturn func(wfContext execution.Context, mutableState execution.MutableState) (*UpdateAction, error) {\n\t\terr := action(wfContext, mutableState)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tpostActions := &UpdateAction{\n\t\t\tCreateDecision: createDecisionTask,\n\t\t}\n\t\treturn postActions, nil\n\t}\n}\n\nfunc updateHelper(\n\tctx context.Context,\n\tlogger log.Logger,\n\tworkflowContext Context,\n\tnow time.Time,\n\taction UpdateActionFunc,\n) (retError error) {\n\nUpdateHistoryLoop:\n\tfor attempt := 0; attempt < ConditionalRetryCount; attempt++ {\n\t\twfContext := workflowContext.GetContext()\n\t\tmutableState := workflowContext.GetMutableState()\n\n\t\t// conduct caller action\n\t\tpostActions, err := action(wfContext, mutableState)\n\t\tif err != nil {\n\t\t\tif err == ErrStaleState {\n\t\t\t\t// Handler detected that cached workflow mutable could potentially be stale\n\t\t\t\t// Reload workflow execution history\n\t\t\t\tworkflowContext.GetContext().Clear()\n\t\t\t\tif attempt != ConditionalRetryCount-1 {\n\t\t\t\t\t_, err = workflowContext.ReloadMutableState(ctx)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn err\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tcontinue UpdateHistoryLoop\n\t\t\t}\n\n\t\t\t// Returned error back to the caller\n\t\t\treturn err\n\t\t}\n\t\tif postActions.Noop {\n\t\t\treturn nil\n\t\t}\n\n\t\tif postActions.CreateDecision {\n\t\t\t// Create a transfer task to schedule a decision task\n\t\t\tif !mutableState.HasPendingDecision() {\n\t\t\t\t_, err := mutableState.AddDecisionTaskScheduledEvent(false)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn &types.InternalServiceError{Message: \"Failed to add decision scheduled event.\"}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tlogger.Debug(\"updateHelper calling UpdateWorkflowExecutionAsActive\", tag.WorkflowID(mutableState.GetExecutionInfo().WorkflowID))\n\t\terr = workflowContext.GetContext().UpdateWorkflowExecutionAsActive(ctx, now)\n\t\tif _, ok := err.(*persistence.DuplicateRequestError); ok {\n\t\t\treturn nil\n\t\t}\n\t\tif execution.IsConflictError(err) {\n\t\t\tif attempt != ConditionalRetryCount-1 {\n\t\t\t\t_, err = workflowContext.ReloadMutableState(ctx)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t}\n\t\t\tcontinue UpdateHistoryLoop\n\t\t}\n\t\treturn err\n\t}\n\treturn ErrMaxAttemptsExceeded\n}\n"
  },
  {
    "path": "service/history/workflow/util_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage workflow\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\tcommonconstants \"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/config\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/service/history/shard\"\n)\n\nfunc TestUpdateHelper(t *testing.T) {\n\ttestCases := []struct {\n\t\tmsg         string\n\t\tmockSetupFn func(*execution.MockContext, *execution.MockMutableState)\n\t\tactionFn    UpdateActionFunc\n\t}{\n\t\t{\n\t\t\tmsg: \"stale mutable state\",\n\t\t\tmockSetupFn: func(mockContext *execution.MockContext, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockContext.EXPECT().Clear().Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(commonconstants.FirstEventID).Times(1)\n\t\t\t\tmockMutableState.EXPECT().GetNextEventID().Return(commonconstants.FirstEventID + 1).Times(1)\n\t\t\t},\n\t\t\tactionFn: func(context execution.Context, mutableState execution.MutableState) (*UpdateAction, error) {\n\t\t\t\tif mutableState.GetNextEventID() == commonconstants.FirstEventID {\n\t\t\t\t\treturn nil, ErrStaleState\n\t\t\t\t}\n\t\t\t\treturn &UpdateAction{Noop: true}, nil\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg: \"schedule new decision\",\n\t\t\tmockSetupFn: func(mockContext *execution.MockContext, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockMutableState.EXPECT().HasPendingDecision().Return(false).Times(1)\n\t\t\t\tmockMutableState.EXPECT().AddDecisionTaskScheduledEvent(gomock.Any()).Return(&execution.DecisionInfo{}, nil).Times(1)\n\t\t\t\tmockContext.EXPECT().UpdateWorkflowExecutionAsActive(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tactionFn: func(context execution.Context, mutableState execution.MutableState) (*UpdateAction, error) {\n\t\t\t\treturn UpdateWithNewDecision, nil\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg: \"update workflow conflict\",\n\t\t\tmockSetupFn: func(mockContext *execution.MockContext, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockContext.EXPECT().UpdateWorkflowExecutionAsActive(gomock.Any(), gomock.Any()).Return(execution.NewConflictError(t, assert.AnError)).Times(ConditionalRetryCount - 1)\n\t\t\t\tmockContext.EXPECT().UpdateWorkflowExecutionAsActive(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t\tactionFn: func(context execution.Context, mutableState execution.MutableState) (*UpdateAction, error) {\n\t\t\t\treturn UpdateWithoutDecision, nil\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg: \"duplicate request\",\n\t\t\tmockSetupFn: func(mockContext *execution.MockContext, mockMutableState *execution.MockMutableState) {\n\t\t\t\tmockContext.EXPECT().UpdateWorkflowExecutionAsActive(gomock.Any(), gomock.Any()).Return(&persistence.DuplicateRequestError{})\n\t\t\t},\n\t\t\tactionFn: func(context execution.Context, mutableState execution.MutableState) (*UpdateAction, error) {\n\t\t\t\treturn UpdateWithoutDecision, nil\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.msg, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tmockMutableState := execution.NewMockMutableState(controller)\n\t\t\tmockMutableState.EXPECT().GetExecutionInfo().Return(&persistence.WorkflowExecutionInfo{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t\t}).AnyTimes()\n\t\t\tmockContext := execution.NewMockContext(controller)\n\t\t\tmockContext.EXPECT().LoadWorkflowExecution(gomock.Any()).Return(mockMutableState, nil).AnyTimes()\n\t\t\tworkflowContext := NewContext(mockContext, nil, mockMutableState)\n\n\t\t\ttc.mockSetupFn(mockContext, mockMutableState)\n\t\t\terr := updateHelper(context.Background(), testlogger.New(t), workflowContext, time.Now(), tc.actionFn)\n\t\t\trequire.NoError(t, err)\n\t\t})\n\t}\n\n}\n\nfunc TestWorkflowLoad(t *testing.T) {\n\tpersistenceMS := &persistence.WorkflowMutableState{\n\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   constants.TestDomainID,\n\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\tRunID:      constants.TestRunID,\n\t\t\tState:      persistence.WorkflowStateRunning,\n\t\t},\n\t\tExecutionStats: &persistence.ExecutionStats{},\n\t}\n\n\ttestCases := []struct {\n\t\tmsg         string\n\t\trunID       string\n\t\tmockSetupFn func(*shard.TestContext)\n\t}{\n\t\t{\n\t\t\tmsg:   \"runID not empty\",\n\t\t\trunID: constants.TestRunID,\n\t\t\tmockSetupFn: func(mockShard *shard.TestContext) {\n\t\t\t\tmockShard.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: persistenceMS,\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t).Times(1)\n\t\t\t\tmockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tmsg:   \"current run closed\",\n\t\t\trunID: \"\",\n\t\t\tmockSetupFn: func(mockShard *shard.TestContext) {\n\t\t\t\tpersistenceMS.ExecutionInfo.State = persistence.WorkflowStateCompleted\n\t\t\t\tmockShard.Resource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&persistence.GetWorkflowExecutionResponse{\n\t\t\t\t\t\tState: persistenceMS,\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t).Times(1)\n\t\t\t\tmockShard.Resource.ActiveClusterMgr.EXPECT().GetActiveClusterInfoByWorkflow(gomock.Any(), constants.TestDomainID, constants.TestWorkflowID, constants.TestRunID).Return(&types.ActiveClusterInfo{ActiveClusterName: \"test-active-cluster\"}, nil).AnyTimes()\n\t\t\t\tmockShard.Resource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&persistence.GetCurrentExecutionResponse{\n\t\t\t\t\t\tRunID: constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t\tnil,\n\t\t\t\t).Times(2)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.msg, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tmockShard := shard.NewTestContext(\n\t\t\t\tt,\n\t\t\t\tcontroller,\n\t\t\t\t&persistence.ShardInfo{\n\t\t\t\t\tShardID: 10,\n\t\t\t\t\tRangeID: 1,\n\t\t\t\t},\n\t\t\t\tconfig.NewForTest(),\n\t\t\t)\n\n\t\t\tmockDomainCache := mockShard.Resource.DomainCache\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestLocalDomainEntry.GetInfo().ID).Return(constants.TestLocalDomainEntry, nil)\n\t\t\tmockDomainCache.EXPECT().GetDomainName(constants.TestLocalDomainEntry.GetInfo().ID).Return(constants.TestLocalDomainEntry.GetInfo().Name, nil).AnyTimes()\n\n\t\t\ttc.mockSetupFn(mockShard)\n\n\t\t\tworkflowContext, err := Load(\n\t\t\t\tcontext.Background(),\n\t\t\t\texecution.NewCache(mockShard),\n\t\t\t\tmockShard.Resource.ExecutionMgr,\n\t\t\t\tconstants.TestDomainID,\n\t\t\t\tconstants.TestDomainName,\n\t\t\t\tconstants.TestWorkflowID,\n\t\t\t\ttc.runID,\n\t\t\t)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, constants.TestWorkflowID, workflowContext.GetWorkflowID())\n\t\t\trequire.Equal(t, constants.TestRunID, workflowContext.GetRunID())\n\t\t\tworkflowContext.GetReleaseFn()(nil)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/workflowcache/cache.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package=$GOPACKAGE -destination=cache_mock.go github.com/uber/cadence/service/history/workflowcache WFCache\n\npackage workflowcache\n\nimport (\n\t\"errors\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\nvar (\n\terrDomainName = errors.New(\"failed to get domain name from domainID\")\n)\n\n// WFCache is a per workflow cache used for workflow specific in memory data\ntype WFCache interface {\n\tAllowExternal(domainID string, workflowID string) bool\n\tAllowInternal(domainID string, workflowID string) bool\n}\n\ntype wfCache struct {\n\tlru                    cache.Cache\n\texternalLimiterFactory quotas.LimiterFactory\n\tinternalLimiterFactory quotas.LimiterFactory\n\tdomainCache            cache.DomainCache\n\tmetricsClient          metrics.Client\n\tlogger                 log.Logger\n\ttimeSource             clock.TimeSource\n\n\t// we use functions to get cache items, and the current time, so we can mock it in unit tests\n\tgetCacheItemFn func(domainName string, workflowID string) (*cacheValue, error)\n}\n\ntype cacheKey struct {\n\tdomainName string\n\tworkflowID string\n}\n\ntype cacheValue struct {\n\texternalRateLimiter quotas.Limiter\n\tinternalRateLimiter quotas.Limiter\n\texternalCountMetric workflowIDCountMetric\n\tinternalCountMetric workflowIDCountMetric\n}\n\n// Params is the parameters for a new WFCache\ntype Params struct {\n\tTTL                    time.Duration\n\tMaxCount               int\n\tExternalLimiterFactory quotas.LimiterFactory\n\tInternalLimiterFactory quotas.LimiterFactory\n\tDomainCache            cache.DomainCache\n\tMetricsClient          metrics.Client\n\tLogger                 log.Logger\n}\n\n// New creates a new WFCache\nfunc New(params Params) WFCache {\n\tcache := &wfCache{\n\t\tlru: cache.New(&cache.Options{\n\t\t\tTTL:           params.TTL,\n\t\t\tPin:           false,\n\t\t\tMaxCount:      params.MaxCount,\n\t\t\tActivelyEvict: true,\n\t\t\tMetricsScope:  params.MetricsClient.Scope(metrics.HistoryWorkflowCacheScope),\n\t\t\tLogger:        params.Logger,\n\t\t}),\n\t\texternalLimiterFactory: params.ExternalLimiterFactory,\n\t\tinternalLimiterFactory: params.InternalLimiterFactory,\n\t\tdomainCache:            params.DomainCache,\n\t\tmetricsClient:          params.MetricsClient,\n\t\ttimeSource:             clock.NewRealTimeSource(),\n\t\tlogger:                 params.Logger,\n\t}\n\t// We set getCacheItemFn to cache.getCacheItem so that we can mock it in unit tests\n\tcache.getCacheItemFn = cache.getCacheItem\n\n\treturn cache\n}\n\ntype rateLimitType int\n\nconst (\n\texternal rateLimitType = iota\n\tinternal\n)\n\nfunc (c *wfCache) allow(domainID string, workflowID string, rateLimitType rateLimitType) bool {\n\tdomainName, err := c.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\tc.logError(domainID, workflowID, errDomainName)\n\t\t// The cache is not enabled if the domain does not exist or there is an error getting it (fail open)\n\t\treturn true\n\t}\n\n\tc.metricsClient.\n\t\tScope(metrics.HistoryClientWfIDCacheScope).\n\t\tUpdateGauge(metrics.WorkflowIDCacheSizeGauge, float64(c.lru.Size()))\n\n\t// Locking is not needed because both getCacheItem and the rate limiter are thread safe\n\tvalue, err := c.getCacheItemFn(domainName, workflowID)\n\tif err != nil {\n\t\tc.logError(domainID, workflowID, err)\n\t\t// If we can't get the cache item, we should allow the request through\n\t\treturn true\n\t}\n\n\tswitch rateLimitType {\n\tcase external:\n\t\tvalue.externalCountMetric.updatePerDomainMaxWFRequestCount(domainName, c.timeSource, c.metricsClient, metrics.WorkflowIDCacheRequestsExternalMaxRequestsPerSecondsTimer)\n\t\tif !value.externalRateLimiter.Allow() {\n\t\t\tc.emitRateLimitMetrics(\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\tdomainName,\n\t\t\t\t\"external\",\n\t\t\t\tmetrics.WorkflowIDCacheRequestsExternalRatelimitedCounter,\n\t\t\t)\n\t\t\treturn false\n\t\t}\n\t\treturn true\n\tcase internal:\n\t\tvalue.internalCountMetric.updatePerDomainMaxWFRequestCount(domainName, c.timeSource, c.metricsClient, metrics.WorkflowIDCacheRequestsInternalMaxRequestsPerSecondsTimer)\n\t\tif !value.internalRateLimiter.Allow() {\n\t\t\tc.emitRateLimitMetrics(\n\t\t\t\tdomainID,\n\t\t\t\tworkflowID,\n\t\t\t\tdomainName,\n\t\t\t\t\"internal\",\n\t\t\t\tmetrics.WorkflowIDCacheRequestsInternalRatelimitedCounter,\n\t\t\t)\n\t\t\treturn false\n\t\t}\n\t\treturn true\n\tdefault:\n\t\t// This should never happen, and we fail open\n\t\tc.logError(domainID, workflowID, errors.New(\"unknown rate limit type\"))\n\t\treturn true\n\t}\n}\n\nfunc (c *wfCache) emitRateLimitMetrics(\n\tdomainID string,\n\tworkflowID string,\n\tdomainName string,\n\tcallType string,\n\tmetric metrics.MetricIdx,\n) {\n\tc.metricsClient.Scope(\n\t\tmetrics.HistoryClientWfIDCacheScope,\n\t\tmetrics.DomainTag(domainName),\n\t).IncCounter(metric)\n\tc.logger.Info(\n\t\t\"Rate limiting workflowID\",\n\t\ttag.RequestType(callType),\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowDomainName(domainName),\n\t\ttag.WorkflowID(workflowID),\n\t)\n}\n\n// AllowExternal returns true if the rate limiter for this domain/workflow allows an external request\nfunc (c *wfCache) AllowExternal(domainID string, workflowID string) bool {\n\treturn c.allow(domainID, workflowID, external)\n}\n\n// AllowInternal returns true if the rate limiter for this domain/workflow allows an internal request\nfunc (c *wfCache) AllowInternal(domainID string, workflowID string) bool {\n\treturn c.allow(domainID, workflowID, internal)\n}\n\nfunc (c *wfCache) getCacheItem(domainName string, workflowID string) (*cacheValue, error) {\n\t// The underlying lru cache is thread safe, so there is no need to lock\n\tkey := cacheKey{\n\t\tdomainName: domainName,\n\t\tworkflowID: workflowID,\n\t}\n\n\tvalue, ok := c.lru.Get(key).(*cacheValue)\n\n\tif ok {\n\t\treturn value, nil\n\t}\n\n\tvalue = &cacheValue{\n\t\texternalRateLimiter: c.externalLimiterFactory.GetLimiter(domainName),\n\t\tinternalRateLimiter: c.internalLimiterFactory.GetLimiter(domainName),\n\t}\n\t// PutIfNotExist is thread safe, and will either return the value that was already in the cache or the value we just created\n\t// another thread might have inserted a value between the Get and PutIfNotExist, but that is ok\n\t// it should never return an error as we do not use Pin\n\tvalueInterface, err := c.lru.PutIfNotExist(key, value)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvalue, ok = valueInterface.(*cacheValue)\n\n\t// This should never happen, either the value was already in the cache or we just inserted it\n\tif !ok {\n\t\treturn nil, errors.New(\"Failed to insert new value into cache\")\n\t}\n\n\treturn value, err\n}\n\nfunc (c *wfCache) logError(domainID string, workflowID string, err error) {\n\tc.logger.Error(\"Unexpected error from workflow cache\",\n\t\ttag.Error(err),\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowID(workflowID),\n\t\ttag.WorkflowIDCacheSize(c.lru.Size()),\n\t)\n}\n"
  },
  {
    "path": "service/history/workflowcache/cache_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/history/workflowcache (interfaces: WFCache)\n//\n// Generated by this command:\n//\n//\tmockgen -package=workflowcache -destination=cache_mock.go github.com/uber/cadence/service/history/workflowcache WFCache\n//\n\n// Package workflowcache is a generated GoMock package.\npackage workflowcache\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockWFCache is a mock of WFCache interface.\ntype MockWFCache struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWFCacheMockRecorder\n\tisgomock struct{}\n}\n\n// MockWFCacheMockRecorder is the mock recorder for MockWFCache.\ntype MockWFCacheMockRecorder struct {\n\tmock *MockWFCache\n}\n\n// NewMockWFCache creates a new mock instance.\nfunc NewMockWFCache(ctrl *gomock.Controller) *MockWFCache {\n\tmock := &MockWFCache{ctrl: ctrl}\n\tmock.recorder = &MockWFCacheMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWFCache) EXPECT() *MockWFCacheMockRecorder {\n\treturn m.recorder\n}\n\n// AllowExternal mocks base method.\nfunc (m *MockWFCache) AllowExternal(domainID, workflowID string) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AllowExternal\", domainID, workflowID)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// AllowExternal indicates an expected call of AllowExternal.\nfunc (mr *MockWFCacheMockRecorder) AllowExternal(domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AllowExternal\", reflect.TypeOf((*MockWFCache)(nil).AllowExternal), domainID, workflowID)\n}\n\n// AllowInternal mocks base method.\nfunc (m *MockWFCache) AllowInternal(domainID, workflowID string) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AllowInternal\", domainID, workflowID)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// AllowInternal indicates an expected call of AllowInternal.\nfunc (mr *MockWFCacheMockRecorder) AllowInternal(domainID, workflowID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AllowInternal\", reflect.TypeOf((*MockWFCache)(nil).AllowInternal), domainID, workflowID)\n}\n"
  },
  {
    "path": "service/history/workflowcache/cache_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage workflowcache\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n)\n\nconst (\n\ttestDomainID    = \"B59344B2-4166-462D-9CBD-22B25D2A7B1B\"\n\ttestDomainName  = \"testDomainName\"\n\ttestWorkflowID  = \"8ED9219B-36A2-4FD0-B9EA-6298A0F2ED1A\"\n\ttestWorkflowID2 = \"F6E31C3D-3E54-4530-BDBE-68AEBA475473\"\n)\n\n// TestWfCache_AllowSingleWorkflow tests that the cache will use the correct rate limiter for internal and external requests.\nfunc TestWfCache_AllowSingleWorkflow(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tdomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).Times(4)\n\n\t// The external rate limiter will allow the first request, but not the second.\n\texternalLimiter := quotas.NewMockLimiter(ctrl)\n\texternalLimiter.EXPECT().Allow().Return(true).Times(1)\n\texternalLimiter.EXPECT().Allow().Return(false).Times(1)\n\n\texternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\texternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(externalLimiter).Times(1)\n\n\t// The internal rate limiter will allow the second request, but not the first.\n\tinternalLimiter := quotas.NewMockLimiter(ctrl)\n\tinternalLimiter.EXPECT().Allow().Return(false).Times(1)\n\tinternalLimiter.EXPECT().Allow().Return(true).Times(1)\n\n\tinternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\tinternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(internalLimiter).Times(1)\n\n\twfCache := New(Params{\n\t\t// The cache TTL is set to 1 minute, so all requests will hit the cache\n\t\tTTL:                    time.Minute,\n\t\tMaxCount:               1_000,\n\t\tExternalLimiterFactory: externalLimiterFactory,\n\t\tInternalLimiterFactory: internalLimiterFactory,\n\t\tLogger:                 log.NewNoop(),\n\t\tDomainCache:            domainCache,\n\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t})\n\n\tassert.True(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.False(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\n\tassert.False(t, wfCache.AllowInternal(testDomainID, testWorkflowID))\n\tassert.True(t, wfCache.AllowInternal(testDomainID, testWorkflowID))\n}\n\n// TestWfCache_AllowMultipleWorkflow tests that the cache will use the correct rate limiter for different workflows.\nfunc TestWfCache_AllowMultipleWorkflow(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tdomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).Times(4)\n\n\t// The external rate limiter for wf1 will allow the first request, but not the second.\n\texternalLimiterWf1 := quotas.NewMockLimiter(ctrl)\n\texternalLimiterWf1.EXPECT().Allow().Return(true).Times(1)\n\texternalLimiterWf1.EXPECT().Allow().Return(false).Times(1)\n\n\t// The external rate limiter for wf2 will allow the second request, but not the first.\n\texternalLimiterWf2 := quotas.NewMockLimiter(ctrl)\n\texternalLimiterWf2.EXPECT().Allow().Return(false).Times(1)\n\texternalLimiterWf2.EXPECT().Allow().Return(true).Times(1)\n\n\texternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\texternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(externalLimiterWf1).Times(1)\n\texternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(externalLimiterWf2).Times(1)\n\n\t// We do not expect calls to the internal rate limiters, but they will still be created.\n\tinternalLimiterWf1 := quotas.NewMockLimiter(ctrl)\n\tinternalLimiterWf2 := quotas.NewMockLimiter(ctrl)\n\n\tinternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\tinternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(internalLimiterWf1).Times(1)\n\tinternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(internalLimiterWf2).Times(1)\n\n\twfCache := New(Params{\n\t\tTTL:                    time.Minute,\n\t\tMaxCount:               1_000,\n\t\tExternalLimiterFactory: externalLimiterFactory,\n\t\tInternalLimiterFactory: internalLimiterFactory,\n\t\tLogger:                 log.NewNoop(),\n\t\tDomainCache:            domainCache,\n\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t})\n\n\tassert.True(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.False(t, wfCache.AllowExternal(testDomainID, testWorkflowID2))\n\n\tassert.False(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.True(t, wfCache.AllowExternal(testDomainID, testWorkflowID2))\n}\n\n// TestWfCache_AllowInternalError tests that the cache will allow internal requests through if there is an error getting the rate limiter.\nfunc TestWfCache_AllowError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tdomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).Times(2)\n\n\t// Setup the mock logger\n\tlogger := log.NewMockLogger(ctrl)\n\n\tlogger.EXPECT().Error(\n\t\t\"Unexpected error from workflow cache\",\n\t\t[]tag.Tag{\n\t\t\ttag.Error(assert.AnError),\n\t\t\ttag.WorkflowDomainID(testDomainID),\n\t\t\ttag.WorkflowID(testWorkflowID),\n\t\t\ttag.WorkflowIDCacheSize(0),\n\t\t},\n\t).Times(2)\n\tlogger.EXPECT().Info(\n\t\t\"LRU cache initialized\",\n\t\t[]tag.Tag{\n\t\t\ttag.Value(map[string]interface{}{\n\t\t\t\t\"isSizeBased\": false,\n\t\t\t\t\"maxCount\":    1000,\n\t\t\t\t\"maxSize\":     1073741824,\n\t\t\t}),\n\t\t}).Times(1)\n\n\t// Setup the cache, we do not need the factories, as we will mock the getCacheItemFn\n\twfCache := New(Params{\n\t\tTTL:                    time.Minute,\n\t\tMaxCount:               1_000,\n\t\tExternalLimiterFactory: nil,\n\t\tInternalLimiterFactory: nil,\n\t\tLogger:                 logger,\n\t\tDomainCache:            domainCache,\n\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t}).(*wfCache)\n\n\t// We set getCacheItemFn to a function that will return an error so that we can test the error logic\n\twfCache.getCacheItemFn = func(domainName string, workflowID string) (*cacheValue, error) {\n\t\treturn nil, assert.AnError\n\t}\n\n\t// We fail open\n\tassert.True(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.True(t, wfCache.AllowInternal(testDomainID, testWorkflowID))\n}\n\n// TestWfCache_AllowDomainCacheError tests that the cache will allow requests through if there is an error getting the domain name.\nfunc TestWfCache_AllowDomainCacheError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tdomainCache.EXPECT().GetDomainName(testDomainID).Return(\"\", assert.AnError).Times(2)\n\n\t// Setup the mock logger\n\tlogger := log.NewMockLogger(ctrl)\n\n\tlogger.EXPECT().Error(\n\t\t\"Unexpected error from workflow cache\",\n\t\t[]tag.Tag{\n\t\t\ttag.Error(errDomainName),\n\t\t\ttag.WorkflowDomainID(testDomainID),\n\t\t\ttag.WorkflowID(testWorkflowID),\n\t\t\ttag.WorkflowIDCacheSize(0),\n\t\t},\n\t).Times(2)\n\tlogger.EXPECT().Info(\n\t\t\"LRU cache initialized\",\n\t\t[]tag.Tag{\n\t\t\ttag.Value(map[string]interface{}{\n\t\t\t\t\"isSizeBased\": false,\n\t\t\t\t\"maxCount\":    1000,\n\t\t\t\t\"maxSize\":     1073741824,\n\t\t\t}),\n\t\t}).Times(1)\n\n\t// Setup the cache, we do not need the factories, as we will mock the getCacheItemFn\n\twfCache := New(Params{\n\t\tTTL:                    time.Minute,\n\t\tMaxCount:               1_000,\n\t\tExternalLimiterFactory: nil,\n\t\tInternalLimiterFactory: nil,\n\t\tLogger:                 logger,\n\t\tDomainCache:            domainCache,\n\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t})\n\n\t// We fail open\n\tassert.True(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.True(t, wfCache.AllowInternal(testDomainID, testWorkflowID))\n}\n\nfunc TestWfCache_RejectLog(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tdomainCache := cache.NewMockDomainCache(ctrl)\n\tdomainCache.EXPECT().GetDomainName(testDomainID).Return(testDomainName, nil).Times(2)\n\n\t// The external rate limiter will reject\n\texternalLimiter := quotas.NewMockLimiter(ctrl)\n\texternalLimiter.EXPECT().Allow().Return(false).Times(1)\n\n\texternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\texternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(externalLimiter).Times(1)\n\n\t// The internal rate limiter will reject\n\tinternalLimiter := quotas.NewMockLimiter(ctrl)\n\tinternalLimiter.EXPECT().Allow().Return(false).Times(1)\n\n\tinternalLimiterFactory := quotas.NewMockLimiterFactory(ctrl)\n\tinternalLimiterFactory.EXPECT().GetLimiter(testDomainName).Return(internalLimiter).Times(1)\n\n\t// Setup the mock logger\n\tlogger := log.NewMockLogger(ctrl)\n\n\tlogger.EXPECT().Info(\n\t\t\"LRU cache initialized\",\n\t\t[]tag.Tag{\n\t\t\ttag.Value(map[string]interface{}{\n\t\t\t\t\"isSizeBased\": false,\n\t\t\t\t\"maxCount\":    1000,\n\t\t\t\t\"maxSize\":     1073741824,\n\t\t\t}),\n\t\t}).Times(1)\n\n\tlogger.EXPECT().Warn(\"Cache is strictly count-based because value *workflowcache.cacheValue does not implement sizable\").Times(1)\n\n\texpectRatelimitLog(logger, \"external\")\n\texpectRatelimitLog(logger, \"internal\")\n\n\twfCache := New(Params{\n\t\tTTL:                    time.Minute,\n\t\tMaxCount:               1_000,\n\t\tExternalLimiterFactory: externalLimiterFactory,\n\t\tInternalLimiterFactory: internalLimiterFactory,\n\t\tLogger:                 logger,\n\t\tDomainCache:            domainCache,\n\t\tMetricsClient:          metrics.NewNoopMetricsClient(),\n\t})\n\n\tassert.False(t, wfCache.AllowExternal(testDomainID, testWorkflowID))\n\tassert.False(t, wfCache.AllowInternal(testDomainID, testWorkflowID))\n}\n\nfunc expectRatelimitLog(logger *log.MockLogger, requestType string) {\n\tlogger.EXPECT().Info(\n\t\t\"Rate limiting workflowID\",\n\t\t[]tag.Tag{\n\t\t\ttag.RequestType(requestType),\n\t\t\ttag.WorkflowDomainID(testDomainID),\n\t\t\ttag.WorkflowDomainName(testDomainName),\n\t\t\ttag.WorkflowID(testWorkflowID),\n\t\t},\n\t).Times(1)\n}\n"
  },
  {
    "path": "service/history/workflowcache/metrics.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage workflowcache\n\nimport (\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// workflowIDCountMetric holds the count of requests for a given second, for a domain/workflowID pair\n// This is used to emit the max count of requests for a given domain\n// Ideally we would just emit the count of requests for a given domain/workflowID pair, but this is not\n// possible, due to the high cardinality of workflowIDs\ntype workflowIDCountMetric struct {\n\tsync.Mutex\n\n\tstartingSecond time.Time\n\tcount          int\n}\n\nfunc (cm *workflowIDCountMetric) reset(now time.Time) {\n\tcm.startingSecond = now\n\tcm.count = 0\n}\n\nfunc (cm *workflowIDCountMetric) updatePerDomainMaxWFRequestCount(\n\tdomainName string,\n\ttimeSource clock.TimeSource,\n\tmetricsClient metrics.Client,\n\tmetric metrics.MetricIdx,\n) {\n\tcm.Lock()\n\tdefer cm.Unlock()\n\n\tif timeSource.Since(cm.startingSecond) > time.Second {\n\t\tcm.reset(timeSource.Now().UTC())\n\t}\n\tcm.count++\n\n\t// We can just use the upper of the metric, so it is not an issue to emit all the counts\n\tmetricsClient.Scope(metrics.HistoryClientWfIDCacheScope, metrics.DomainTag(domainName)).\n\t\tRecordTimer(metric, time.Duration(cm.count))\n}\n"
  },
  {
    "path": "service/history/workflowcache/metrics_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage workflowcache\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestUpdatePerDomainMaxWFRequestCount(t *testing.T) {\n\tdomainName := \"some domain name\"\n\tmetric := metrics.WorkflowIDCacheRequestsInternalMaxRequestsPerSecondsTimer\n\n\tcases := []struct {\n\t\tname                             string\n\t\tupdatePerDomainMaxWFRequestCount func(metricsClient metrics.Client, source clock.MockedTimeSource)\n\t\texpecetMetrics                   []time.Duration\n\t}{\n\t\t{\n\t\t\tname: \"Single workflowID\",\n\t\t\tupdatePerDomainMaxWFRequestCount: func(metricsClient metrics.Client, timeSource clock.MockedTimeSource) {\n\t\t\t\tworkflowID1 := &workflowIDCountMetric{}\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 1\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 2\n\t\t\t},\n\t\t\texpecetMetrics: []time.Duration{1, 2},\n\t\t},\n\t\t{\n\t\t\tname: \"Separate workflowIDs\",\n\t\t\tupdatePerDomainMaxWFRequestCount: func(metricsClient metrics.Client, timeSource clock.MockedTimeSource) {\n\t\t\t\tworkflowID1 := &workflowIDCountMetric{}\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 1\n\n\t\t\t\tworkflowID2 := &workflowIDCountMetric{}\n\t\t\t\tworkflowID2.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 1\n\t\t\t\tworkflowID2.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 2\n\t\t\t\tworkflowID2.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 3\n\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 2\n\n\t\t\t},\n\t\t\texpecetMetrics: []time.Duration{1, 1, 2, 3, 2},\n\t\t},\n\t\t{\n\t\t\tname: \"Reset\",\n\t\t\tupdatePerDomainMaxWFRequestCount: func(metricsClient metrics.Client, timeSource clock.MockedTimeSource) {\n\t\t\t\tworkflowID1 := &workflowIDCountMetric{}\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 1\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 2\n\n\t\t\t\ttimeSource.Advance(1100 * time.Millisecond)\n\t\t\t\tworkflowID1.updatePerDomainMaxWFRequestCount(domainName, timeSource, metricsClient, metric) // Emits 1\n\t\t\t},\n\t\t\texpecetMetrics: []time.Duration{1, 2, 1},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttestScope := tally.NewTestScope(\"\", make(map[string]string))\n\t\t\ttimeSource := clock.NewMockedTimeSourceAt(time.Unix(123, 456))\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.History, metrics.HistogramMigration{})\n\n\t\t\ttc.updatePerDomainMaxWFRequestCount(metricsClient, timeSource)\n\n\t\t\t// We expect the domain tag to be set to \"all\" and \"some domain name\", we don't know the order, so use a set\n\t\t\texpectedDomainTags := map[string]struct{}{\"all\": {}, \"some domain name\": {}}\n\t\t\tactualDomainTags := map[string]struct{}{}\n\n\t\t\ttimers := testScope.Snapshot().Timers()\n\t\t\tassert.Equal(t, 2, len(timers))\n\t\t\tfor _, v := range timers {\n\t\t\t\tactualDomainTags[v.Tags()[\"domain\"]] = struct{}{}\n\t\t\t\tassert.Equal(t, tc.expecetMetrics, v.Values())\n\t\t\t}\n\n\t\t\tassert.Equal(t, expectedDomainTags, actualDomainTags)\n\t\t})\n\t}\n\n}\n"
  },
  {
    "path": "service/history/wrappers/grpc/grpc_handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage grpc\n\nimport (\n\t\"context\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g GRPCHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(historyv1.BuildHistoryAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildMetaAPIYARPCProcedures(g))\n}\n\nfunc (g GRPCHandler) Health(ctx context.Context, _ *apiv1.HealthRequest) (*apiv1.HealthResponse, error) {\n\tresponse, err := g.h.Health(ctx)\n\treturn proto.FromHealthResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/history/wrappers/grpc/grpc_handler_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\thistoryv1 \"github.com/uber/cadence/.gen/proto/history/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/history/handler\"\n)\n\ntype GRPCHandler struct {\n\th handler.Handler\n}\n\nfunc NewGRPCHandler(h handler.Handler) GRPCHandler {\n\treturn GRPCHandler{h}\n}\n\nfunc (g GRPCHandler) CloseShard(ctx context.Context, request *historyv1.CloseShardRequest) (*historyv1.CloseShardResponse, error) {\n\terr := g.h.CloseShard(ctx, proto.ToHistoryCloseShardRequest(request))\n\treturn &historyv1.CloseShardResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) CountDLQMessages(ctx context.Context, request *historyv1.CountDLQMessagesRequest) (*historyv1.CountDLQMessagesResponse, error) {\n\tresponse, err := g.h.CountDLQMessages(ctx, proto.ToHistoryCountDLQMessagesRequest(request))\n\treturn proto.FromHistoryCountDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) DescribeHistoryHost(ctx context.Context, request *historyv1.DescribeHistoryHostRequest) (*historyv1.DescribeHistoryHostResponse, error) {\n\tresponse, err := g.h.DescribeHistoryHost(ctx, proto.ToHistoryDescribeHistoryHostRequest(request))\n\treturn proto.FromHistoryDescribeHistoryHostResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) DescribeMutableState(ctx context.Context, request *historyv1.DescribeMutableStateRequest) (*historyv1.DescribeMutableStateResponse, error) {\n\tresponse, err := g.h.DescribeMutableState(ctx, proto.ToHistoryDescribeMutableStateRequest(request))\n\treturn proto.FromHistoryDescribeMutableStateResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) DescribeQueue(ctx context.Context, request *historyv1.DescribeQueueRequest) (*historyv1.DescribeQueueResponse, error) {\n\tresponse, err := g.h.DescribeQueue(ctx, proto.ToHistoryDescribeQueueRequest(request))\n\treturn proto.FromHistoryDescribeQueueResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) DescribeWorkflowExecution(ctx context.Context, request *historyv1.DescribeWorkflowExecutionRequest) (*historyv1.DescribeWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, proto.ToHistoryDescribeWorkflowExecutionRequest(request))\n\treturn proto.FromHistoryDescribeWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetCrossClusterTasks(ctx context.Context, request *historyv1.GetCrossClusterTasksRequest) (*historyv1.GetCrossClusterTasksResponse, error) {\n\tresponse, err := g.h.GetCrossClusterTasks(ctx, proto.ToHistoryGetCrossClusterTasksRequest(request))\n\treturn proto.FromHistoryGetCrossClusterTasksResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetDLQReplicationMessages(ctx context.Context, request *historyv1.GetDLQReplicationMessagesRequest) (*historyv1.GetDLQReplicationMessagesResponse, error) {\n\tresponse, err := g.h.GetDLQReplicationMessages(ctx, proto.ToHistoryGetDLQReplicationMessagesRequest(request))\n\treturn proto.FromHistoryGetDLQReplicationMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetFailoverInfo(ctx context.Context, request *historyv1.GetFailoverInfoRequest) (*historyv1.GetFailoverInfoResponse, error) {\n\tresponse, err := g.h.GetFailoverInfo(ctx, proto.ToHistoryGetFailoverInfoRequest(request))\n\treturn proto.FromHistoryGetFailoverInfoResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetMutableState(ctx context.Context, request *historyv1.GetMutableStateRequest) (*historyv1.GetMutableStateResponse, error) {\n\tresponse, err := g.h.GetMutableState(ctx, proto.ToHistoryGetMutableStateRequest(request))\n\treturn proto.FromHistoryGetMutableStateResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetReplicationMessages(ctx context.Context, request *historyv1.GetReplicationMessagesRequest) (*historyv1.GetReplicationMessagesResponse, error) {\n\tresponse, err := g.h.GetReplicationMessages(ctx, proto.ToHistoryGetReplicationMessagesRequest(request))\n\treturn proto.FromHistoryGetReplicationMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) MergeDLQMessages(ctx context.Context, request *historyv1.MergeDLQMessagesRequest) (*historyv1.MergeDLQMessagesResponse, error) {\n\tresponse, err := g.h.MergeDLQMessages(ctx, proto.ToHistoryMergeDLQMessagesRequest(request))\n\treturn proto.FromHistoryMergeDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) NotifyFailoverMarkers(ctx context.Context, request *historyv1.NotifyFailoverMarkersRequest) (*historyv1.NotifyFailoverMarkersResponse, error) {\n\terr := g.h.NotifyFailoverMarkers(ctx, proto.ToHistoryNotifyFailoverMarkersRequest(request))\n\treturn &historyv1.NotifyFailoverMarkersResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) PollMutableState(ctx context.Context, request *historyv1.PollMutableStateRequest) (*historyv1.PollMutableStateResponse, error) {\n\tresponse, err := g.h.PollMutableState(ctx, proto.ToHistoryPollMutableStateRequest(request))\n\treturn proto.FromHistoryPollMutableStateResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) PurgeDLQMessages(ctx context.Context, request *historyv1.PurgeDLQMessagesRequest) (*historyv1.PurgeDLQMessagesResponse, error) {\n\terr := g.h.PurgeDLQMessages(ctx, proto.ToHistoryPurgeDLQMessagesRequest(request))\n\treturn &historyv1.PurgeDLQMessagesResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) QueryWorkflow(ctx context.Context, request *historyv1.QueryWorkflowRequest) (*historyv1.QueryWorkflowResponse, error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, proto.ToHistoryQueryWorkflowRequest(request))\n\treturn proto.FromHistoryQueryWorkflowResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RatelimitUpdate(ctx context.Context, request *historyv1.RatelimitUpdateRequest) (*historyv1.RatelimitUpdateResponse, error) {\n\tresponse, err := g.h.RatelimitUpdate(ctx, proto.ToHistoryRatelimitUpdateRequest(request))\n\treturn proto.FromHistoryRatelimitUpdateResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ReadDLQMessages(ctx context.Context, request *historyv1.ReadDLQMessagesRequest) (*historyv1.ReadDLQMessagesResponse, error) {\n\tresponse, err := g.h.ReadDLQMessages(ctx, proto.ToHistoryReadDLQMessagesRequest(request))\n\treturn proto.FromHistoryReadDLQMessagesResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ReapplyEvents(ctx context.Context, request *historyv1.ReapplyEventsRequest) (*historyv1.ReapplyEventsResponse, error) {\n\terr := g.h.ReapplyEvents(ctx, proto.ToHistoryReapplyEventsRequest(request))\n\treturn &historyv1.ReapplyEventsResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RecordActivityTaskHeartbeat(ctx context.Context, request *historyv1.RecordActivityTaskHeartbeatRequest) (*historyv1.RecordActivityTaskHeartbeatResponse, error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeat(ctx, proto.ToHistoryRecordActivityTaskHeartbeatRequest(request))\n\treturn proto.FromHistoryRecordActivityTaskHeartbeatResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RecordActivityTaskStarted(ctx context.Context, request *historyv1.RecordActivityTaskStartedRequest) (*historyv1.RecordActivityTaskStartedResponse, error) {\n\tresponse, err := g.h.RecordActivityTaskStarted(ctx, proto.ToHistoryRecordActivityTaskStartedRequest(request))\n\treturn proto.FromHistoryRecordActivityTaskStartedResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RecordChildExecutionCompleted(ctx context.Context, request *historyv1.RecordChildExecutionCompletedRequest) (*historyv1.RecordChildExecutionCompletedResponse, error) {\n\terr := g.h.RecordChildExecutionCompleted(ctx, proto.ToHistoryRecordChildExecutionCompletedRequest(request))\n\treturn &historyv1.RecordChildExecutionCompletedResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RecordDecisionTaskStarted(ctx context.Context, request *historyv1.RecordDecisionTaskStartedRequest) (*historyv1.RecordDecisionTaskStartedResponse, error) {\n\tresponse, err := g.h.RecordDecisionTaskStarted(ctx, proto.ToHistoryRecordDecisionTaskStartedRequest(request))\n\treturn proto.FromHistoryRecordDecisionTaskStartedResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RefreshWorkflowTasks(ctx context.Context, request *historyv1.RefreshWorkflowTasksRequest) (*historyv1.RefreshWorkflowTasksResponse, error) {\n\terr := g.h.RefreshWorkflowTasks(ctx, proto.ToHistoryRefreshWorkflowTasksRequest(request))\n\treturn &historyv1.RefreshWorkflowTasksResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RemoveSignalMutableState(ctx context.Context, request *historyv1.RemoveSignalMutableStateRequest) (*historyv1.RemoveSignalMutableStateResponse, error) {\n\terr := g.h.RemoveSignalMutableState(ctx, proto.ToHistoryRemoveSignalMutableStateRequest(request))\n\treturn &historyv1.RemoveSignalMutableStateResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RemoveTask(ctx context.Context, request *historyv1.RemoveTaskRequest) (*historyv1.RemoveTaskResponse, error) {\n\terr := g.h.RemoveTask(ctx, proto.ToHistoryRemoveTaskRequest(request))\n\treturn &historyv1.RemoveTaskResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ReplicateEventsV2(ctx context.Context, request *historyv1.ReplicateEventsV2Request) (*historyv1.ReplicateEventsV2Response, error) {\n\terr := g.h.ReplicateEventsV2(ctx, proto.ToHistoryReplicateEventsV2Request(request))\n\treturn &historyv1.ReplicateEventsV2Response{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RequestCancelWorkflowExecution(ctx context.Context, request *historyv1.RequestCancelWorkflowExecutionRequest) (*historyv1.RequestCancelWorkflowExecutionResponse, error) {\n\terr := g.h.RequestCancelWorkflowExecution(ctx, proto.ToHistoryRequestCancelWorkflowExecutionRequest(request))\n\treturn &historyv1.RequestCancelWorkflowExecutionResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ResetQueue(ctx context.Context, request *historyv1.ResetQueueRequest) (*historyv1.ResetQueueResponse, error) {\n\terr := g.h.ResetQueue(ctx, proto.ToHistoryResetQueueRequest(request))\n\treturn &historyv1.ResetQueueResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ResetStickyTaskList(ctx context.Context, request *historyv1.ResetStickyTaskListRequest) (*historyv1.ResetStickyTaskListResponse, error) {\n\tresponse, err := g.h.ResetStickyTaskList(ctx, proto.ToHistoryResetStickyTaskListRequest(request))\n\treturn proto.FromHistoryResetStickyTaskListResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ResetWorkflowExecution(ctx context.Context, request *historyv1.ResetWorkflowExecutionRequest) (*historyv1.ResetWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.ResetWorkflowExecution(ctx, proto.ToHistoryResetWorkflowExecutionRequest(request))\n\treturn proto.FromHistoryResetWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondActivityTaskCanceled(ctx context.Context, request *historyv1.RespondActivityTaskCanceledRequest) (*historyv1.RespondActivityTaskCanceledResponse, error) {\n\terr := g.h.RespondActivityTaskCanceled(ctx, proto.ToHistoryRespondActivityTaskCanceledRequest(request))\n\treturn &historyv1.RespondActivityTaskCanceledResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondActivityTaskCompleted(ctx context.Context, request *historyv1.RespondActivityTaskCompletedRequest) (*historyv1.RespondActivityTaskCompletedResponse, error) {\n\terr := g.h.RespondActivityTaskCompleted(ctx, proto.ToHistoryRespondActivityTaskCompletedRequest(request))\n\treturn &historyv1.RespondActivityTaskCompletedResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondActivityTaskFailed(ctx context.Context, request *historyv1.RespondActivityTaskFailedRequest) (*historyv1.RespondActivityTaskFailedResponse, error) {\n\terr := g.h.RespondActivityTaskFailed(ctx, proto.ToHistoryRespondActivityTaskFailedRequest(request))\n\treturn &historyv1.RespondActivityTaskFailedResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondCrossClusterTasksCompleted(ctx context.Context, request *historyv1.RespondCrossClusterTasksCompletedRequest) (*historyv1.RespondCrossClusterTasksCompletedResponse, error) {\n\tresponse, err := g.h.RespondCrossClusterTasksCompleted(ctx, proto.ToHistoryRespondCrossClusterTasksCompletedRequest(request))\n\treturn proto.FromHistoryRespondCrossClusterTasksCompletedResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondDecisionTaskCompleted(ctx context.Context, request *historyv1.RespondDecisionTaskCompletedRequest) (*historyv1.RespondDecisionTaskCompletedResponse, error) {\n\tresponse, err := g.h.RespondDecisionTaskCompleted(ctx, proto.ToHistoryRespondDecisionTaskCompletedRequest(request))\n\treturn proto.FromHistoryRespondDecisionTaskCompletedResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondDecisionTaskFailed(ctx context.Context, request *historyv1.RespondDecisionTaskFailedRequest) (*historyv1.RespondDecisionTaskFailedResponse, error) {\n\terr := g.h.RespondDecisionTaskFailed(ctx, proto.ToHistoryRespondDecisionTaskFailedRequest(request))\n\treturn &historyv1.RespondDecisionTaskFailedResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ScheduleDecisionTask(ctx context.Context, request *historyv1.ScheduleDecisionTaskRequest) (*historyv1.ScheduleDecisionTaskResponse, error) {\n\terr := g.h.ScheduleDecisionTask(ctx, proto.ToHistoryScheduleDecisionTaskRequest(request))\n\treturn &historyv1.ScheduleDecisionTaskResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) SignalWithStartWorkflowExecution(ctx context.Context, request *historyv1.SignalWithStartWorkflowExecutionRequest) (*historyv1.SignalWithStartWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecution(ctx, proto.ToHistorySignalWithStartWorkflowExecutionRequest(request))\n\treturn proto.FromHistorySignalWithStartWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) SignalWorkflowExecution(ctx context.Context, request *historyv1.SignalWorkflowExecutionRequest) (*historyv1.SignalWorkflowExecutionResponse, error) {\n\terr := g.h.SignalWorkflowExecution(ctx, proto.ToHistorySignalWorkflowExecutionRequest(request))\n\treturn &historyv1.SignalWorkflowExecutionResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) StartWorkflowExecution(ctx context.Context, request *historyv1.StartWorkflowExecutionRequest) (*historyv1.StartWorkflowExecutionResponse, error) {\n\tresponse, err := g.h.StartWorkflowExecution(ctx, proto.ToHistoryStartWorkflowExecutionRequest(request))\n\treturn proto.FromHistoryStartWorkflowExecutionResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) SyncActivity(ctx context.Context, request *historyv1.SyncActivityRequest) (*historyv1.SyncActivityResponse, error) {\n\terr := g.h.SyncActivity(ctx, proto.ToHistorySyncActivityRequest(request))\n\treturn &historyv1.SyncActivityResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) SyncShardStatus(ctx context.Context, request *historyv1.SyncShardStatusRequest) (*historyv1.SyncShardStatusResponse, error) {\n\terr := g.h.SyncShardStatus(ctx, proto.ToHistorySyncShardStatusRequest(request))\n\treturn &historyv1.SyncShardStatusResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) TerminateWorkflowExecution(ctx context.Context, request *historyv1.TerminateWorkflowExecutionRequest) (*historyv1.TerminateWorkflowExecutionResponse, error) {\n\terr := g.h.TerminateWorkflowExecution(ctx, proto.ToHistoryTerminateWorkflowExecutionRequest(request))\n\treturn &historyv1.TerminateWorkflowExecutionResponse{}, proto.FromError(err)\n}\n"
  },
  {
    "path": "service/history/wrappers/ratelimited/handler_generated.go",
    "content": "package ratelimited\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/ratelimited.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\n//go:generate gowrap gen -p github.com/uber/cadence/service/history/handler -i Handler -t ../../templates/ratelimited.tmpl -o handler_generated.go -v handler=History -l \"\"\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/frontend/validate\"\n\t\"github.com/uber/cadence/service/history/handler\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n)\n\n// historyHandler implements handler.Handler interface instrumented with rate limiter.\ntype historyHandler struct {\n\twrapped         handler.Handler\n\tworkflowIDCache workflowcache.WFCache\n\tlogger          log.Logger\n\tallowFunc       func(domainID string, workflowID string) bool\n}\n\n// NewHistoryHandler creates a new instance of Handler with ratelimiter.\nfunc NewHistoryHandler(\n\twrapped handler.Handler,\n\tworkflowIDCache workflowcache.WFCache,\n\tlogger log.Logger,\n) handler.Handler {\n\twrapper := &historyHandler{\n\t\twrapped:         wrapped,\n\t\tworkflowIDCache: workflowIDCache,\n\t\tlogger:          logger,\n\t}\n\twrapper.allowFunc = wrapper.allowWfID\n\n\treturn wrapper\n}\n\nfunc (h *historyHandler) CloseShard(ctx context.Context, cp1 *types.CloseShardRequest) (err error) {\n\treturn h.wrapped.CloseShard(ctx, cp1)\n}\n\nfunc (h *historyHandler) CountDLQMessages(ctx context.Context, cp1 *types.CountDLQMessagesRequest) (hp1 *types.HistoryCountDLQMessagesResponse, err error) {\n\treturn h.wrapped.CountDLQMessages(ctx, cp1)\n}\n\nfunc (h *historyHandler) DescribeHistoryHost(ctx context.Context, dp1 *types.DescribeHistoryHostRequest) (dp2 *types.DescribeHistoryHostResponse, err error) {\n\treturn h.wrapped.DescribeHistoryHost(ctx, dp1)\n}\n\nfunc (h *historyHandler) DescribeMutableState(ctx context.Context, dp1 *types.DescribeMutableStateRequest) (dp2 *types.DescribeMutableStateResponse, err error) {\n\treturn h.wrapped.DescribeMutableState(ctx, dp1)\n}\n\nfunc (h *historyHandler) DescribeQueue(ctx context.Context, dp1 *types.DescribeQueueRequest) (dp2 *types.DescribeQueueResponse, err error) {\n\treturn h.wrapped.DescribeQueue(ctx, dp1)\n}\n\nfunc (h *historyHandler) DescribeWorkflowExecution(ctx context.Context, hp1 *types.HistoryDescribeWorkflowExecutionRequest) (dp1 *types.DescribeWorkflowExecutionResponse, err error) {\n\n\tif hp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\n\tif hp1.GetDomainUUID() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\n\tif hp1.Request.GetExecution().GetWorkflowID() == \"\" {\n\t\terr = validate.ErrWorkflowIDNotSet\n\t\treturn\n\t}\n\n\tif !h.allowFunc(hp1.GetDomainUUID(), hp1.Request.GetExecution().GetWorkflowID()) {\n\t\terr = &types.ServiceBusyError{\n\t\t\tMessage: \"Too many requests for the workflow ID\",\n\t\t\tReason:  constants.WorkflowIDRateLimitReason,\n\t\t}\n\t\treturn\n\t}\n\treturn h.wrapped.DescribeWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) GetCrossClusterTasks(ctx context.Context, gp1 *types.GetCrossClusterTasksRequest) (gp2 *types.GetCrossClusterTasksResponse, err error) {\n\treturn h.wrapped.GetCrossClusterTasks(ctx, gp1)\n}\n\nfunc (h *historyHandler) GetDLQReplicationMessages(ctx context.Context, gp1 *types.GetDLQReplicationMessagesRequest) (gp2 *types.GetDLQReplicationMessagesResponse, err error) {\n\treturn h.wrapped.GetDLQReplicationMessages(ctx, gp1)\n}\n\nfunc (h *historyHandler) GetFailoverInfo(ctx context.Context, gp1 *types.GetFailoverInfoRequest) (gp2 *types.GetFailoverInfoResponse, err error) {\n\treturn h.wrapped.GetFailoverInfo(ctx, gp1)\n}\n\nfunc (h *historyHandler) GetMutableState(ctx context.Context, gp1 *types.GetMutableStateRequest) (gp2 *types.GetMutableStateResponse, err error) {\n\treturn h.wrapped.GetMutableState(ctx, gp1)\n}\n\nfunc (h *historyHandler) GetReplicationMessages(ctx context.Context, gp1 *types.GetReplicationMessagesRequest) (gp2 *types.GetReplicationMessagesResponse, err error) {\n\treturn h.wrapped.GetReplicationMessages(ctx, gp1)\n}\n\nfunc (h *historyHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn h.wrapped.Health(ctx)\n}\n\nfunc (h *historyHandler) MergeDLQMessages(ctx context.Context, mp1 *types.MergeDLQMessagesRequest) (mp2 *types.MergeDLQMessagesResponse, err error) {\n\treturn h.wrapped.MergeDLQMessages(ctx, mp1)\n}\n\nfunc (h *historyHandler) NotifyFailoverMarkers(ctx context.Context, np1 *types.NotifyFailoverMarkersRequest) (err error) {\n\treturn h.wrapped.NotifyFailoverMarkers(ctx, np1)\n}\n\nfunc (h *historyHandler) PollMutableState(ctx context.Context, pp1 *types.PollMutableStateRequest) (pp2 *types.PollMutableStateResponse, err error) {\n\treturn h.wrapped.PollMutableState(ctx, pp1)\n}\n\nfunc (h *historyHandler) PrepareToStop(d1 time.Duration) (d2 time.Duration) {\n\treturn h.wrapped.PrepareToStop(d1)\n}\n\nfunc (h *historyHandler) PurgeDLQMessages(ctx context.Context, pp1 *types.PurgeDLQMessagesRequest) (err error) {\n\treturn h.wrapped.PurgeDLQMessages(ctx, pp1)\n}\n\nfunc (h *historyHandler) QueryWorkflow(ctx context.Context, hp1 *types.HistoryQueryWorkflowRequest) (hp2 *types.HistoryQueryWorkflowResponse, err error) {\n\treturn h.wrapped.QueryWorkflow(ctx, hp1)\n}\n\nfunc (h *historyHandler) RatelimitUpdate(ctx context.Context, rp1 *types.RatelimitUpdateRequest) (rp2 *types.RatelimitUpdateResponse, err error) {\n\treturn h.wrapped.RatelimitUpdate(ctx, rp1)\n}\n\nfunc (h *historyHandler) ReadDLQMessages(ctx context.Context, rp1 *types.ReadDLQMessagesRequest) (rp2 *types.ReadDLQMessagesResponse, err error) {\n\treturn h.wrapped.ReadDLQMessages(ctx, rp1)\n}\n\nfunc (h *historyHandler) ReapplyEvents(ctx context.Context, hp1 *types.HistoryReapplyEventsRequest) (err error) {\n\treturn h.wrapped.ReapplyEvents(ctx, hp1)\n}\n\nfunc (h *historyHandler) RecordActivityTaskHeartbeat(ctx context.Context, hp1 *types.HistoryRecordActivityTaskHeartbeatRequest) (rp1 *types.RecordActivityTaskHeartbeatResponse, err error) {\n\treturn h.wrapped.RecordActivityTaskHeartbeat(ctx, hp1)\n}\n\nfunc (h *historyHandler) RecordActivityTaskStarted(ctx context.Context, rp1 *types.RecordActivityTaskStartedRequest) (rp2 *types.RecordActivityTaskStartedResponse, err error) {\n\treturn h.wrapped.RecordActivityTaskStarted(ctx, rp1)\n}\n\nfunc (h *historyHandler) RecordChildExecutionCompleted(ctx context.Context, rp1 *types.RecordChildExecutionCompletedRequest) (err error) {\n\treturn h.wrapped.RecordChildExecutionCompleted(ctx, rp1)\n}\n\nfunc (h *historyHandler) RecordDecisionTaskStarted(ctx context.Context, rp1 *types.RecordDecisionTaskStartedRequest) (rp2 *types.RecordDecisionTaskStartedResponse, err error) {\n\treturn h.wrapped.RecordDecisionTaskStarted(ctx, rp1)\n}\n\nfunc (h *historyHandler) RefreshWorkflowTasks(ctx context.Context, hp1 *types.HistoryRefreshWorkflowTasksRequest) (err error) {\n\treturn h.wrapped.RefreshWorkflowTasks(ctx, hp1)\n}\n\nfunc (h *historyHandler) RemoveSignalMutableState(ctx context.Context, rp1 *types.RemoveSignalMutableStateRequest) (err error) {\n\treturn h.wrapped.RemoveSignalMutableState(ctx, rp1)\n}\n\nfunc (h *historyHandler) RemoveTask(ctx context.Context, rp1 *types.RemoveTaskRequest) (err error) {\n\treturn h.wrapped.RemoveTask(ctx, rp1)\n}\n\nfunc (h *historyHandler) ReplicateEventsV2(ctx context.Context, rp1 *types.ReplicateEventsV2Request) (err error) {\n\treturn h.wrapped.ReplicateEventsV2(ctx, rp1)\n}\n\nfunc (h *historyHandler) RequestCancelWorkflowExecution(ctx context.Context, hp1 *types.HistoryRequestCancelWorkflowExecutionRequest) (err error) {\n\treturn h.wrapped.RequestCancelWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) ResetQueue(ctx context.Context, rp1 *types.ResetQueueRequest) (err error) {\n\treturn h.wrapped.ResetQueue(ctx, rp1)\n}\n\nfunc (h *historyHandler) ResetStickyTaskList(ctx context.Context, hp1 *types.HistoryResetStickyTaskListRequest) (hp2 *types.HistoryResetStickyTaskListResponse, err error) {\n\treturn h.wrapped.ResetStickyTaskList(ctx, hp1)\n}\n\nfunc (h *historyHandler) ResetWorkflowExecution(ctx context.Context, hp1 *types.HistoryResetWorkflowExecutionRequest) (rp1 *types.ResetWorkflowExecutionResponse, err error) {\n\treturn h.wrapped.ResetWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) RespondActivityTaskCanceled(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCanceledRequest) (err error) {\n\treturn h.wrapped.RespondActivityTaskCanceled(ctx, hp1)\n}\n\nfunc (h *historyHandler) RespondActivityTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondActivityTaskCompletedRequest) (err error) {\n\treturn h.wrapped.RespondActivityTaskCompleted(ctx, hp1)\n}\n\nfunc (h *historyHandler) RespondActivityTaskFailed(ctx context.Context, hp1 *types.HistoryRespondActivityTaskFailedRequest) (err error) {\n\treturn h.wrapped.RespondActivityTaskFailed(ctx, hp1)\n}\n\nfunc (h *historyHandler) RespondCrossClusterTasksCompleted(ctx context.Context, rp1 *types.RespondCrossClusterTasksCompletedRequest) (rp2 *types.RespondCrossClusterTasksCompletedResponse, err error) {\n\treturn h.wrapped.RespondCrossClusterTasksCompleted(ctx, rp1)\n}\n\nfunc (h *historyHandler) RespondDecisionTaskCompleted(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskCompletedRequest) (hp2 *types.HistoryRespondDecisionTaskCompletedResponse, err error) {\n\treturn h.wrapped.RespondDecisionTaskCompleted(ctx, hp1)\n}\n\nfunc (h *historyHandler) RespondDecisionTaskFailed(ctx context.Context, hp1 *types.HistoryRespondDecisionTaskFailedRequest) (err error) {\n\treturn h.wrapped.RespondDecisionTaskFailed(ctx, hp1)\n}\n\nfunc (h *historyHandler) ScheduleDecisionTask(ctx context.Context, sp1 *types.ScheduleDecisionTaskRequest) (err error) {\n\treturn h.wrapped.ScheduleDecisionTask(ctx, sp1)\n}\n\nfunc (h *historyHandler) SignalWithStartWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWithStartWorkflowExecutionRequest) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\n\tif hp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\n\tif hp1.GetDomainUUID() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\n\tif hp1.SignalWithStartRequest.GetWorkflowID() == \"\" {\n\t\terr = validate.ErrWorkflowIDNotSet\n\t\treturn\n\t}\n\n\tif !h.allowFunc(hp1.GetDomainUUID(), hp1.SignalWithStartRequest.GetWorkflowID()) {\n\t\terr = &types.ServiceBusyError{\n\t\t\tMessage: \"Too many requests for the workflow ID\",\n\t\t\tReason:  constants.WorkflowIDRateLimitReason,\n\t\t}\n\t\treturn\n\t}\n\treturn h.wrapped.SignalWithStartWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) SignalWorkflowExecution(ctx context.Context, hp1 *types.HistorySignalWorkflowExecutionRequest) (err error) {\n\n\tif hp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\n\tif hp1.GetDomainUUID() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\n\tif hp1.SignalRequest.GetWorkflowExecution().GetWorkflowID() == \"\" {\n\t\terr = validate.ErrWorkflowIDNotSet\n\t\treturn\n\t}\n\n\tif !h.allowFunc(hp1.GetDomainUUID(), hp1.SignalRequest.GetWorkflowExecution().GetWorkflowID()) {\n\t\terr = &types.ServiceBusyError{\n\t\t\tMessage: \"Too many requests for the workflow ID\",\n\t\t\tReason:  constants.WorkflowIDRateLimitReason,\n\t\t}\n\t\treturn\n\t}\n\treturn h.wrapped.SignalWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) Start() {\n\th.wrapped.Start()\n\treturn\n}\n\nfunc (h *historyHandler) StartWorkflowExecution(ctx context.Context, hp1 *types.HistoryStartWorkflowExecutionRequest) (sp1 *types.StartWorkflowExecutionResponse, err error) {\n\n\tif hp1 == nil {\n\t\terr = validate.ErrRequestNotSet\n\t\treturn\n\t}\n\n\tif hp1.GetDomainUUID() == \"\" {\n\t\terr = validate.ErrDomainNotSet\n\t\treturn\n\t}\n\n\tif hp1.StartRequest.GetWorkflowID() == \"\" {\n\t\terr = validate.ErrWorkflowIDNotSet\n\t\treturn\n\t}\n\n\tif !h.allowFunc(hp1.GetDomainUUID(), hp1.StartRequest.GetWorkflowID()) {\n\t\terr = &types.ServiceBusyError{\n\t\t\tMessage: \"Too many requests for the workflow ID\",\n\t\t\tReason:  constants.WorkflowIDRateLimitReason,\n\t\t}\n\t\treturn\n\t}\n\treturn h.wrapped.StartWorkflowExecution(ctx, hp1)\n}\n\nfunc (h *historyHandler) Stop() {\n\th.wrapped.Stop()\n\treturn\n}\n\nfunc (h *historyHandler) SyncActivity(ctx context.Context, sp1 *types.SyncActivityRequest) (err error) {\n\treturn h.wrapped.SyncActivity(ctx, sp1)\n}\n\nfunc (h *historyHandler) SyncShardStatus(ctx context.Context, sp1 *types.SyncShardStatusRequest) (err error) {\n\treturn h.wrapped.SyncShardStatus(ctx, sp1)\n}\n\nfunc (h *historyHandler) TerminateWorkflowExecution(ctx context.Context, hp1 *types.HistoryTerminateWorkflowExecutionRequest) (err error) {\n\treturn h.wrapped.TerminateWorkflowExecution(ctx, hp1)\n}\n"
  },
  {
    "path": "service/history/wrappers/ratelimited/handler_generated_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/handler\"\n)\n\nconst (\n\ttestDomainID   = \"test-domain-id\"\n\ttestWorkflowID = \"test-workflow-id\"\n\ttestDomainName = \"test-domain-name\"\n)\n\nfunc TestRatelimitedEndpoints_Table(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\n\thandlerMock := handler.NewMockHandler(controller)\n\n\twrapper := NewHistoryHandler(handlerMock, nil, log.NewNoop())\n\n\t// We define the calls that should be ratelimited\n\tlimitedCalls := []struct {\n\t\tname string\n\t\t// Defines how to call the wrapper function (correct request type, and call)\n\t\tcallWrapper func() (interface{}, error)\n\t\t// Defines the expected call to the wrapped handler (what to call if the call is not ratelimited)\n\t\texpectCallToEndpoint func()\n\t}{\n\t\t{\n\t\t\tname: \"StartWorkflowExecution\",\n\t\t\tcallWrapper: func() (interface{}, error) {\n\t\t\t\tstartRequest := &types.HistoryStartWorkflowExecutionRequest{\n\t\t\t\t\tDomainUUID:   testDomainID,\n\t\t\t\t\tStartRequest: &types.StartWorkflowExecutionRequest{WorkflowID: testWorkflowID},\n\t\t\t\t}\n\t\t\t\treturn wrapper.StartWorkflowExecution(context.Background(), startRequest)\n\t\t\t},\n\t\t\texpectCallToEndpoint: func() {\n\t\t\t\thandlerMock.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalWithStartWorkflowExecution\",\n\t\t\tcallWrapper: func() (interface{}, error) {\n\t\t\t\tsignalWithStartRequest := &types.HistorySignalWithStartWorkflowExecutionRequest{\n\t\t\t\t\tDomainUUID:             testDomainID,\n\t\t\t\t\tSignalWithStartRequest: &types.SignalWithStartWorkflowExecutionRequest{WorkflowID: testWorkflowID},\n\t\t\t\t}\n\n\t\t\t\treturn wrapper.SignalWithStartWorkflowExecution(context.Background(), signalWithStartRequest)\n\t\t\t},\n\t\t\texpectCallToEndpoint: func() {\n\t\t\t\thandlerMock.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalWorkflowExecution\",\n\t\t\tcallWrapper: func() (interface{}, error) {\n\t\t\t\tsignalRequest := &types.HistorySignalWorkflowExecutionRequest{\n\t\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\t\tSignalRequest: &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{WorkflowID: testWorkflowID},\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\treturn nil, wrapper.SignalWorkflowExecution(context.Background(), signalRequest)\n\t\t\t},\n\t\t\texpectCallToEndpoint: func() {\n\t\t\t\thandlerMock.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeWorkflowExecution\",\n\t\t\tcallWrapper: func() (interface{}, error) {\n\t\t\t\tdescribeRequest := &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomainUUID: testDomainID,\n\t\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{WorkflowID: testWorkflowID},\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\treturn wrapper.DescribeWorkflowExecution(context.Background(), describeRequest)\n\t\t\t},\n\t\t\texpectCallToEndpoint: func() {\n\t\t\t\thandlerMock.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, endpoint := range limitedCalls {\n\t\tt.Run(fmt.Sprintf(\"%s, %s\", endpoint.name, \"not limited\"), func(t *testing.T) {\n\t\t\twrapper.(*historyHandler).allowFunc = func(string, string) bool { return true }\n\t\t\tendpoint.expectCallToEndpoint()\n\t\t\t_, err := endpoint.callWrapper()\n\t\t\tassert.NoError(t, err)\n\t\t})\n\n\t\tt.Run(fmt.Sprintf(\"%s, %s\", endpoint.name, \"limited\"), func(t *testing.T) {\n\t\t\twrapper.(*historyHandler).allowFunc = func(string, string) bool { return false }\n\t\t\t_, err := endpoint.callWrapper()\n\t\t\tvar sbErr *types.ServiceBusyError\n\t\t\tassert.ErrorAs(t, err, &sbErr)\n\t\t\tassert.ErrorContains(t, err, \"Too many requests for the workflow ID\")\n\t\t\tassert.Equal(t, constants.WorkflowIDRateLimitReason, sbErr.Reason)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/wrappers/ratelimited/ratelimit.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nfunc (h *historyHandler) allowWfID(domainUUID, workflowID string) bool {\n\treturn h.workflowIDCache.AllowExternal(domainUUID, workflowID)\n}\n"
  },
  {
    "path": "service/history/wrappers/ratelimited/ratelimit_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage ratelimited\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/history/workflowcache\"\n)\n\nfunc TestAllowWfID(t *testing.T) {\n\ttests := []struct {\n\t\tworkflowIDCacheAllow bool\n\t\texpected             bool\n\t}{\n\t\t{\n\t\t\tworkflowIDCacheAllow: true,\n\t\t\texpected:             true,\n\t\t},\n\t\t{\n\t\t\tworkflowIDCacheAllow: false,\n\t\t\texpected:             false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(fmt.Sprintf(\"workflowIDCacheAllow: %t\", tt.workflowIDCacheAllow), func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tworkflowIDCacheMock := workflowcache.NewMockWFCache(ctrl)\n\t\t\tworkflowIDCacheMock.EXPECT().AllowExternal(testDomainID, testWorkflowID).Return(tt.workflowIDCacheAllow).Times(1)\n\n\t\t\th := &historyHandler{\n\t\t\t\tworkflowIDCache: workflowIDCacheMock,\n\t\t\t\tlogger:          log.NewNoop(),\n\t\t\t}\n\n\t\t\tgot := h.allowWfID(testDomainID, testWorkflowID)\n\n\t\t\tassert.Equal(t, tt.expected, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/history/wrappers/thrift/thrift_handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/health\"\n\t\"github.com/uber/cadence/.gen/go/health/metaserver\"\n\t\"github.com/uber/cadence/.gen/go/history/historyserviceserver\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/history/handler\"\n)\n\ntype ThriftHandler struct {\n\th handler.Handler\n}\n\nfunc NewThriftHandler(h handler.Handler) ThriftHandler {\n\treturn ThriftHandler{h}\n}\n\nfunc (t ThriftHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(historyserviceserver.New(&t))\n\tdispatcher.Register(metaserver.New(&t))\n}\n\nfunc (t ThriftHandler) Health(ctx context.Context) (*health.HealthStatus, error) {\n\tresponse, err := t.h.Health(ctx)\n\treturn thrift.FromHealthStatus(response), thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/history/wrappers/thrift/thrift_handler_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g ThriftHandler) CloseShard(ctx context.Context, Request *shared.CloseShardRequest) (err error) {\n\terr = g.h.CloseShard(ctx, thrift.ToHistoryCloseShardRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) DescribeHistoryHost(ctx context.Context, Request *shared.DescribeHistoryHostRequest) (dp1 *shared.DescribeHistoryHostResponse, err error) {\n\tresponse, err := g.h.DescribeHistoryHost(ctx, thrift.ToHistoryDescribeHistoryHostRequest(Request))\n\treturn thrift.FromHistoryDescribeHistoryHostResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) DescribeMutableState(ctx context.Context, Request *history.DescribeMutableStateRequest) (dp1 *history.DescribeMutableStateResponse, err error) {\n\tresponse, err := g.h.DescribeMutableState(ctx, thrift.ToHistoryDescribeMutableStateRequest(Request))\n\treturn thrift.FromHistoryDescribeMutableStateResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) DescribeQueue(ctx context.Context, Request *shared.DescribeQueueRequest) (dp1 *shared.DescribeQueueResponse, err error) {\n\tresponse, err := g.h.DescribeQueue(ctx, thrift.ToHistoryDescribeQueueRequest(Request))\n\treturn thrift.FromHistoryDescribeQueueResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) DescribeWorkflowExecution(ctx context.Context, DescribeRequest *history.DescribeWorkflowExecutionRequest) (dp1 *shared.DescribeWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.DescribeWorkflowExecution(ctx, thrift.ToHistoryDescribeWorkflowExecutionRequest(DescribeRequest))\n\treturn thrift.FromHistoryDescribeWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetCrossClusterTasks(ctx context.Context, Request *shared.GetCrossClusterTasksRequest) (gp1 *shared.GetCrossClusterTasksResponse, err error) {\n\tresponse, err := g.h.GetCrossClusterTasks(ctx, thrift.ToHistoryGetCrossClusterTasksRequest(Request))\n\treturn thrift.FromHistoryGetCrossClusterTasksResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetDLQReplicationMessages(ctx context.Context, Request *replicator.GetDLQReplicationMessagesRequest) (gp1 *replicator.GetDLQReplicationMessagesResponse, err error) {\n\tresponse, err := g.h.GetDLQReplicationMessages(ctx, thrift.ToHistoryGetDLQReplicationMessagesRequest(Request))\n\treturn thrift.FromHistoryGetDLQReplicationMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetFailoverInfo(ctx context.Context, Request *history.GetFailoverInfoRequest) (gp1 *history.GetFailoverInfoResponse, err error) {\n\tresponse, err := g.h.GetFailoverInfo(ctx, thrift.ToHistoryGetFailoverInfoRequest(Request))\n\treturn thrift.FromHistoryGetFailoverInfoResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetMutableState(ctx context.Context, GetRequest *history.GetMutableStateRequest) (gp1 *history.GetMutableStateResponse, err error) {\n\tresponse, err := g.h.GetMutableState(ctx, thrift.ToHistoryGetMutableStateRequest(GetRequest))\n\treturn thrift.FromHistoryGetMutableStateResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetReplicationMessages(ctx context.Context, Request *replicator.GetReplicationMessagesRequest) (gp1 *replicator.GetReplicationMessagesResponse, err error) {\n\tresponse, err := g.h.GetReplicationMessages(ctx, thrift.ToHistoryGetReplicationMessagesRequest(Request))\n\treturn thrift.FromHistoryGetReplicationMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) MergeDLQMessages(ctx context.Context, Request *replicator.MergeDLQMessagesRequest) (mp1 *replicator.MergeDLQMessagesResponse, err error) {\n\tresponse, err := g.h.MergeDLQMessages(ctx, thrift.ToHistoryMergeDLQMessagesRequest(Request))\n\treturn thrift.FromHistoryMergeDLQMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) NotifyFailoverMarkers(ctx context.Context, Request *history.NotifyFailoverMarkersRequest) (err error) {\n\terr = g.h.NotifyFailoverMarkers(ctx, thrift.ToHistoryNotifyFailoverMarkersRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) PollMutableState(ctx context.Context, PollRequest *history.PollMutableStateRequest) (pp1 *history.PollMutableStateResponse, err error) {\n\tresponse, err := g.h.PollMutableState(ctx, thrift.ToHistoryPollMutableStateRequest(PollRequest))\n\treturn thrift.FromHistoryPollMutableStateResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) PurgeDLQMessages(ctx context.Context, Request *replicator.PurgeDLQMessagesRequest) (err error) {\n\terr = g.h.PurgeDLQMessages(ctx, thrift.ToHistoryPurgeDLQMessagesRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) QueryWorkflow(ctx context.Context, QueryRequest *history.QueryWorkflowRequest) (qp1 *history.QueryWorkflowResponse, err error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, thrift.ToHistoryQueryWorkflowRequest(QueryRequest))\n\treturn thrift.FromHistoryQueryWorkflowResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RatelimitUpdate(ctx context.Context, Request *history.RatelimitUpdateRequest) (rp1 *history.RatelimitUpdateResponse, err error) {\n\tresponse, err := g.h.RatelimitUpdate(ctx, thrift.ToHistoryRatelimitUpdateRequest(Request))\n\treturn thrift.FromHistoryRatelimitUpdateResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ReadDLQMessages(ctx context.Context, Request *replicator.ReadDLQMessagesRequest) (rp1 *replicator.ReadDLQMessagesResponse, err error) {\n\tresponse, err := g.h.ReadDLQMessages(ctx, thrift.ToHistoryReadDLQMessagesRequest(Request))\n\treturn thrift.FromHistoryReadDLQMessagesResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ReapplyEvents(ctx context.Context, ReapplyEventsRequest *history.ReapplyEventsRequest) (err error) {\n\terr = g.h.ReapplyEvents(ctx, thrift.ToHistoryReapplyEventsRequest(ReapplyEventsRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RecordActivityTaskHeartbeat(ctx context.Context, HeartbeatRequest *history.RecordActivityTaskHeartbeatRequest) (rp1 *shared.RecordActivityTaskHeartbeatResponse, err error) {\n\tresponse, err := g.h.RecordActivityTaskHeartbeat(ctx, thrift.ToHistoryRecordActivityTaskHeartbeatRequest(HeartbeatRequest))\n\treturn thrift.FromHistoryRecordActivityTaskHeartbeatResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RecordActivityTaskStarted(ctx context.Context, AddRequest *history.RecordActivityTaskStartedRequest) (rp1 *history.RecordActivityTaskStartedResponse, err error) {\n\tresponse, err := g.h.RecordActivityTaskStarted(ctx, thrift.ToHistoryRecordActivityTaskStartedRequest(AddRequest))\n\treturn thrift.FromHistoryRecordActivityTaskStartedResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RecordChildExecutionCompleted(ctx context.Context, CompletionRequest *history.RecordChildExecutionCompletedRequest) (err error) {\n\terr = g.h.RecordChildExecutionCompleted(ctx, thrift.ToHistoryRecordChildExecutionCompletedRequest(CompletionRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RecordDecisionTaskStarted(ctx context.Context, AddRequest *history.RecordDecisionTaskStartedRequest) (rp1 *history.RecordDecisionTaskStartedResponse, err error) {\n\tresponse, err := g.h.RecordDecisionTaskStarted(ctx, thrift.ToHistoryRecordDecisionTaskStartedRequest(AddRequest))\n\treturn thrift.FromHistoryRecordDecisionTaskStartedResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RefreshWorkflowTasks(ctx context.Context, Request *history.RefreshWorkflowTasksRequest) (err error) {\n\terr = g.h.RefreshWorkflowTasks(ctx, thrift.ToHistoryRefreshWorkflowTasksRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RemoveSignalMutableState(ctx context.Context, RemoveRequest *history.RemoveSignalMutableStateRequest) (err error) {\n\terr = g.h.RemoveSignalMutableState(ctx, thrift.ToHistoryRemoveSignalMutableStateRequest(RemoveRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RemoveTask(ctx context.Context, Request *shared.RemoveTaskRequest) (err error) {\n\terr = g.h.RemoveTask(ctx, thrift.ToHistoryRemoveTaskRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ReplicateEventsV2(ctx context.Context, ReplicateV2Request *history.ReplicateEventsV2Request) (err error) {\n\terr = g.h.ReplicateEventsV2(ctx, thrift.ToHistoryReplicateEventsV2Request(ReplicateV2Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RequestCancelWorkflowExecution(ctx context.Context, CancelRequest *history.RequestCancelWorkflowExecutionRequest) (err error) {\n\terr = g.h.RequestCancelWorkflowExecution(ctx, thrift.ToHistoryRequestCancelWorkflowExecutionRequest(CancelRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ResetQueue(ctx context.Context, Request *shared.ResetQueueRequest) (err error) {\n\terr = g.h.ResetQueue(ctx, thrift.ToHistoryResetQueueRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ResetStickyTaskList(ctx context.Context, ResetRequest *history.ResetStickyTaskListRequest) (rp1 *history.ResetStickyTaskListResponse, err error) {\n\tresponse, err := g.h.ResetStickyTaskList(ctx, thrift.ToHistoryResetStickyTaskListRequest(ResetRequest))\n\treturn thrift.FromHistoryResetStickyTaskListResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ResetWorkflowExecution(ctx context.Context, ResetRequest *history.ResetWorkflowExecutionRequest) (rp1 *shared.ResetWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.ResetWorkflowExecution(ctx, thrift.ToHistoryResetWorkflowExecutionRequest(ResetRequest))\n\treturn thrift.FromHistoryResetWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondActivityTaskCanceled(ctx context.Context, CanceledRequest *history.RespondActivityTaskCanceledRequest) (err error) {\n\terr = g.h.RespondActivityTaskCanceled(ctx, thrift.ToHistoryRespondActivityTaskCanceledRequest(CanceledRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondActivityTaskCompleted(ctx context.Context, CompleteRequest *history.RespondActivityTaskCompletedRequest) (err error) {\n\terr = g.h.RespondActivityTaskCompleted(ctx, thrift.ToHistoryRespondActivityTaskCompletedRequest(CompleteRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondActivityTaskFailed(ctx context.Context, FailRequest *history.RespondActivityTaskFailedRequest) (err error) {\n\terr = g.h.RespondActivityTaskFailed(ctx, thrift.ToHistoryRespondActivityTaskFailedRequest(FailRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondCrossClusterTasksCompleted(ctx context.Context, Request *shared.RespondCrossClusterTasksCompletedRequest) (rp1 *shared.RespondCrossClusterTasksCompletedResponse, err error) {\n\tresponse, err := g.h.RespondCrossClusterTasksCompleted(ctx, thrift.ToHistoryRespondCrossClusterTasksCompletedRequest(Request))\n\treturn thrift.FromHistoryRespondCrossClusterTasksCompletedResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondDecisionTaskCompleted(ctx context.Context, CompleteRequest *history.RespondDecisionTaskCompletedRequest) (rp1 *history.RespondDecisionTaskCompletedResponse, err error) {\n\tresponse, err := g.h.RespondDecisionTaskCompleted(ctx, thrift.ToHistoryRespondDecisionTaskCompletedRequest(CompleteRequest))\n\treturn thrift.FromHistoryRespondDecisionTaskCompletedResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondDecisionTaskFailed(ctx context.Context, FailedRequest *history.RespondDecisionTaskFailedRequest) (err error) {\n\terr = g.h.RespondDecisionTaskFailed(ctx, thrift.ToHistoryRespondDecisionTaskFailedRequest(FailedRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ScheduleDecisionTask(ctx context.Context, ScheduleRequest *history.ScheduleDecisionTaskRequest) (err error) {\n\terr = g.h.ScheduleDecisionTask(ctx, thrift.ToHistoryScheduleDecisionTaskRequest(ScheduleRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) SignalWithStartWorkflowExecution(ctx context.Context, SignalWithStartRequest *history.SignalWithStartWorkflowExecutionRequest) (sp1 *shared.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.SignalWithStartWorkflowExecution(ctx, thrift.ToHistorySignalWithStartWorkflowExecutionRequest(SignalWithStartRequest))\n\treturn thrift.FromHistorySignalWithStartWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) SignalWorkflowExecution(ctx context.Context, SignalRequest *history.SignalWorkflowExecutionRequest) (err error) {\n\terr = g.h.SignalWorkflowExecution(ctx, thrift.ToHistorySignalWorkflowExecutionRequest(SignalRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) StartWorkflowExecution(ctx context.Context, StartRequest *history.StartWorkflowExecutionRequest) (sp1 *shared.StartWorkflowExecutionResponse, err error) {\n\tresponse, err := g.h.StartWorkflowExecution(ctx, thrift.ToHistoryStartWorkflowExecutionRequest(StartRequest))\n\treturn thrift.FromHistoryStartWorkflowExecutionResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) SyncActivity(ctx context.Context, SyncActivityRequest *history.SyncActivityRequest) (err error) {\n\terr = g.h.SyncActivity(ctx, thrift.ToHistorySyncActivityRequest(SyncActivityRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) SyncShardStatus(ctx context.Context, SyncShardStatusRequest *history.SyncShardStatusRequest) (err error) {\n\terr = g.h.SyncShardStatus(ctx, thrift.ToHistorySyncShardStatusRequest(SyncShardStatusRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) TerminateWorkflowExecution(ctx context.Context, TerminateRequest *history.TerminateWorkflowExecutionRequest) (err error) {\n\terr = g.h.TerminateWorkflowExecution(ctx, thrift.ToHistoryTerminateWorkflowExecutionRequest(TerminateRequest))\n\treturn thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/history/wrappers/thrift/thrift_handler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/health\"\n\thist \"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/handler\"\n)\n\nfunc TestThriftHandler(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\th := handler.NewMockHandler(ctrl)\n\tth := NewThriftHandler(h)\n\tctx := context.Background()\n\tinternalErr := &types.InternalServiceError{Message: \"test\"}\n\texpectedErr := &shared.InternalServiceError{Message: \"test\"}\n\n\tt.Run(\"Health\", func(t *testing.T) {\n\t\th.EXPECT().Health(ctx).Return(&types.HealthStatus{}, internalErr).Times(1)\n\t\tresp, err := th.Health(ctx)\n\t\tassert.Equal(t, health.HealthStatus{Msg: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"CloseShard\", func(t *testing.T) {\n\t\th.EXPECT().CloseShard(ctx, &types.CloseShardRequest{}).Return(internalErr).Times(1)\n\t\terr := th.CloseShard(ctx, &shared.CloseShardRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeHistoryHost\", func(t *testing.T) {\n\t\th.EXPECT().DescribeHistoryHost(ctx, &types.DescribeHistoryHostRequest{}).Return(&types.DescribeHistoryHostResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeHistoryHost(ctx, &shared.DescribeHistoryHostRequest{})\n\t\tassert.Equal(t, shared.DescribeHistoryHostResponse{NumberOfShards: common.Int32Ptr(0), ShardControllerStatus: common.StringPtr(\"\"), Address: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeMutableState\", func(t *testing.T) {\n\t\th.EXPECT().DescribeMutableState(ctx, &types.DescribeMutableStateRequest{}).Return(&types.DescribeMutableStateResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeMutableState(ctx, &hist.DescribeMutableStateRequest{})\n\t\tassert.Equal(t, hist.DescribeMutableStateResponse{MutableStateInCache: common.StringPtr(\"\"), MutableStateInDatabase: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeQueue\", func(t *testing.T) {\n\t\th.EXPECT().DescribeQueue(ctx, &types.DescribeQueueRequest{}).Return(&types.DescribeQueueResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeQueue(ctx, &shared.DescribeQueueRequest{})\n\t\tassert.Equal(t, shared.DescribeQueueResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().DescribeWorkflowExecution(ctx, &types.HistoryDescribeWorkflowExecutionRequest{}).Return(&types.DescribeWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeWorkflowExecution(ctx, &hist.DescribeWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.DescribeWorkflowExecutionResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetCrossClusterTasks\", func(t *testing.T) {\n\t\th.EXPECT().GetCrossClusterTasks(ctx, &types.GetCrossClusterTasksRequest{}).Return(&types.GetCrossClusterTasksResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetCrossClusterTasks(ctx, &shared.GetCrossClusterTasksRequest{})\n\t\tassert.Equal(t, shared.GetCrossClusterTasksResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetDLQReplicationMessages\", func(t *testing.T) {\n\t\th.EXPECT().GetDLQReplicationMessages(ctx, &types.GetDLQReplicationMessagesRequest{}).Return(&types.GetDLQReplicationMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetDLQReplicationMessages(ctx, &replicator.GetDLQReplicationMessagesRequest{})\n\t\tassert.Equal(t, replicator.GetDLQReplicationMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetMutableState\", func(t *testing.T) {\n\t\th.EXPECT().GetMutableState(ctx, &types.GetMutableStateRequest{}).Return(&types.GetMutableStateResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetMutableState(ctx, &hist.GetMutableStateRequest{})\n\t\tassert.Equal(t, hist.GetMutableStateResponse{\n\t\t\tIsWorkflowRunning:       common.BoolPtr(false),\n\t\t\tNextEventId:             common.Int64Ptr(0),\n\t\t\tLastFirstEventId:        common.Int64Ptr(0),\n\t\t\tClientLibraryVersion:    common.StringPtr(\"\"),\n\t\t\tClientFeatureVersion:    common.StringPtr(\"\"),\n\t\t\tClientImpl:              common.StringPtr(\"\"),\n\t\t\tEventStoreVersion:       common.Int32Ptr(0),\n\t\t\tIsStickyTaskListEnabled: common.BoolPtr(false),\n\t\t\tHistorySize:             common.Int64Ptr(0),\n\t\t}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"GetReplicationMessages\", func(t *testing.T) {\n\t\th.EXPECT().GetReplicationMessages(ctx, &types.GetReplicationMessagesRequest{}).Return(&types.GetReplicationMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.GetReplicationMessages(ctx, &replicator.GetReplicationMessagesRequest{})\n\t\tassert.Equal(t, replicator.GetReplicationMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"MergeDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().MergeDLQMessages(ctx, &types.MergeDLQMessagesRequest{}).Return(&types.MergeDLQMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.MergeDLQMessages(ctx, &replicator.MergeDLQMessagesRequest{})\n\t\tassert.Equal(t, replicator.MergeDLQMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"NotifyFailoverMarkers\", func(t *testing.T) {\n\t\th.EXPECT().NotifyFailoverMarkers(ctx, &types.NotifyFailoverMarkersRequest{}).Return(internalErr).Times(1)\n\t\terr := th.NotifyFailoverMarkers(ctx, &hist.NotifyFailoverMarkersRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PollMutableState\", func(t *testing.T) {\n\t\th.EXPECT().PollMutableState(ctx, &types.PollMutableStateRequest{}).Return(&types.PollMutableStateResponse{}, internalErr).Times(1)\n\t\tresp, err := th.PollMutableState(ctx, &hist.PollMutableStateRequest{})\n\t\tassert.Equal(t, hist.PollMutableStateResponse{\n\t\t\tNextEventId:          common.Int64Ptr(0),\n\t\t\tLastFirstEventId:     common.Int64Ptr(0),\n\t\t\tClientLibraryVersion: common.StringPtr(\"\"),\n\t\t\tClientFeatureVersion: common.StringPtr(\"\"),\n\t\t\tClientImpl:           common.StringPtr(\"\"),\n\t\t}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PurgeDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().PurgeDLQMessages(ctx, &types.PurgeDLQMessagesRequest{}).Return(internalErr).Times(1)\n\t\terr := th.PurgeDLQMessages(ctx, &replicator.PurgeDLQMessagesRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"QueryWorkflow\", func(t *testing.T) {\n\t\th.EXPECT().QueryWorkflow(ctx, &types.HistoryQueryWorkflowRequest{}).Return(&types.HistoryQueryWorkflowResponse{}, internalErr).Times(1)\n\t\tresp, err := th.QueryWorkflow(ctx, &hist.QueryWorkflowRequest{})\n\t\tassert.Equal(t, hist.QueryWorkflowResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ReadDLQMessages\", func(t *testing.T) {\n\t\th.EXPECT().ReadDLQMessages(ctx, &types.ReadDLQMessagesRequest{}).Return(&types.ReadDLQMessagesResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ReadDLQMessages(ctx, &replicator.ReadDLQMessagesRequest{})\n\t\tassert.Equal(t, replicator.ReadDLQMessagesResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ReapplyEvents\", func(t *testing.T) {\n\t\th.EXPECT().ReapplyEvents(ctx, &types.HistoryReapplyEventsRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ReapplyEvents(ctx, &hist.ReapplyEventsRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordActivityTaskHeartbeat\", func(t *testing.T) {\n\t\th.EXPECT().RecordActivityTaskHeartbeat(ctx, &types.HistoryRecordActivityTaskHeartbeatRequest{}).Return(&types.RecordActivityTaskHeartbeatResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RecordActivityTaskHeartbeat(ctx, &hist.RecordActivityTaskHeartbeatRequest{})\n\t\tassert.Equal(t, shared.RecordActivityTaskHeartbeatResponse{CancelRequested: common.BoolPtr(false)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordActivityTaskStarted\", func(t *testing.T) {\n\t\th.EXPECT().RecordActivityTaskStarted(ctx, &types.RecordActivityTaskStartedRequest{}).Return(&types.RecordActivityTaskStartedResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RecordActivityTaskStarted(ctx, &hist.RecordActivityTaskStartedRequest{})\n\t\tassert.Equal(t, hist.RecordActivityTaskStartedResponse{WorkflowDomain: common.StringPtr(\"\"), Attempt: common.Int64Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordChildExecutionCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RecordChildExecutionCompleted(ctx, &types.RecordChildExecutionCompletedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RecordChildExecutionCompleted(ctx, &hist.RecordChildExecutionCompletedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RecordDecisionTaskStarted\", func(t *testing.T) {\n\t\th.EXPECT().RecordDecisionTaskStarted(ctx, &types.RecordDecisionTaskStartedRequest{}).Return(&types.RecordDecisionTaskStartedResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RecordDecisionTaskStarted(ctx, &hist.RecordDecisionTaskStartedRequest{})\n\t\tassert.Equal(t, hist.RecordDecisionTaskStartedResponse{\n\t\t\tScheduledEventId:       common.Int64Ptr(0),\n\t\t\tStartedEventId:         common.Int64Ptr(0),\n\t\t\tNextEventId:            common.Int64Ptr(0),\n\t\t\tAttempt:                common.Int64Ptr(0),\n\t\t\tStickyExecutionEnabled: common.BoolPtr(false),\n\t\t\tEventStoreVersion:      common.Int32Ptr(0),\n\t\t\tHistorySize:            common.Int64Ptr(0),\n\t\t}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RefreshWorkflowTasks\", func(t *testing.T) {\n\t\th.EXPECT().RefreshWorkflowTasks(ctx, &types.HistoryRefreshWorkflowTasksRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RefreshWorkflowTasks(ctx, &hist.RefreshWorkflowTasksRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RemoveSignalMutableState\", func(t *testing.T) {\n\t\th.EXPECT().RemoveSignalMutableState(ctx, &types.RemoveSignalMutableStateRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RemoveSignalMutableState(ctx, &hist.RemoveSignalMutableStateRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RemoveTask\", func(t *testing.T) {\n\t\th.EXPECT().RemoveTask(ctx, &types.RemoveTaskRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RemoveTask(ctx, &shared.RemoveTaskRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ReplicateEventsV2\", func(t *testing.T) {\n\t\th.EXPECT().ReplicateEventsV2(ctx, &types.ReplicateEventsV2Request{}).Return(internalErr).Times(1)\n\t\terr := th.ReplicateEventsV2(ctx, &hist.ReplicateEventsV2Request{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RequestCancelWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().RequestCancelWorkflowExecution(ctx, &types.HistoryRequestCancelWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RequestCancelWorkflowExecution(ctx, &hist.RequestCancelWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetQueue\", func(t *testing.T) {\n\t\th.EXPECT().ResetQueue(ctx, &types.ResetQueueRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ResetQueue(ctx, &shared.ResetQueueRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetStickyTaskList\", func(t *testing.T) {\n\t\th.EXPECT().ResetStickyTaskList(ctx, &types.HistoryResetStickyTaskListRequest{}).Return(&types.HistoryResetStickyTaskListResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ResetStickyTaskList(ctx, &hist.ResetStickyTaskListRequest{})\n\t\tassert.Equal(t, hist.ResetStickyTaskListResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ResetWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().ResetWorkflowExecution(ctx, &types.HistoryResetWorkflowExecutionRequest{}).Return(&types.ResetWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ResetWorkflowExecution(ctx, &hist.ResetWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.ResetWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCanceled\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCanceled(ctx, &types.HistoryRespondActivityTaskCanceledRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCanceled(ctx, &hist.RespondActivityTaskCanceledRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskCompleted(ctx, &types.HistoryRespondActivityTaskCompletedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskCompleted(ctx, &hist.RespondActivityTaskCompletedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondActivityTaskFailed\", func(t *testing.T) {\n\t\th.EXPECT().RespondActivityTaskFailed(ctx, &types.HistoryRespondActivityTaskFailedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondActivityTaskFailed(ctx, &hist.RespondActivityTaskFailedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondCrossClusterTasksCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondCrossClusterTasksCompleted(ctx, &types.RespondCrossClusterTasksCompletedRequest{}).Return(&types.RespondCrossClusterTasksCompletedResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RespondCrossClusterTasksCompleted(ctx, &shared.RespondCrossClusterTasksCompletedRequest{})\n\t\tassert.Equal(t, shared.RespondCrossClusterTasksCompletedResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondDecisionTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondDecisionTaskCompleted(ctx, &types.HistoryRespondDecisionTaskCompletedRequest{}).Return(&types.HistoryRespondDecisionTaskCompletedResponse{}, internalErr).Times(1)\n\t\tresp, err := th.RespondDecisionTaskCompleted(ctx, &hist.RespondDecisionTaskCompletedRequest{})\n\t\tassert.Equal(t, hist.RespondDecisionTaskCompletedResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondDecisionTaskFailed\", func(t *testing.T) {\n\t\th.EXPECT().RespondDecisionTaskFailed(ctx, &types.HistoryRespondDecisionTaskFailedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondDecisionTaskFailed(ctx, &hist.RespondDecisionTaskFailedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ScheduleDecisionTask\", func(t *testing.T) {\n\t\th.EXPECT().ScheduleDecisionTask(ctx, &types.ScheduleDecisionTaskRequest{}).Return(internalErr).Times(1)\n\t\terr := th.ScheduleDecisionTask(ctx, &hist.ScheduleDecisionTaskRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SignalWithStartWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().SignalWithStartWorkflowExecution(ctx, &types.HistorySignalWithStartWorkflowExecutionRequest{}).Return(&types.StartWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.SignalWithStartWorkflowExecution(ctx, &hist.SignalWithStartWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.StartWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SignalWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().SignalWorkflowExecution(ctx, &types.HistorySignalWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.SignalWorkflowExecution(ctx, &hist.SignalWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"StartWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().StartWorkflowExecution(ctx, &types.HistoryStartWorkflowExecutionRequest{}).Return(&types.StartWorkflowExecutionResponse{}, internalErr).Times(1)\n\t\tresp, err := th.StartWorkflowExecution(ctx, &hist.StartWorkflowExecutionRequest{})\n\t\tassert.Equal(t, shared.StartWorkflowExecutionResponse{RunId: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SyncActivity\", func(t *testing.T) {\n\t\th.EXPECT().SyncActivity(ctx, &types.SyncActivityRequest{}).Return(internalErr).Times(1)\n\t\terr := th.SyncActivity(ctx, &hist.SyncActivityRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"SyncShardStatus\", func(t *testing.T) {\n\t\th.EXPECT().SyncShardStatus(ctx, &types.SyncShardStatusRequest{}).Return(internalErr).Times(1)\n\t\terr := th.SyncShardStatus(ctx, &hist.SyncShardStatusRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"TerminateWorkflowExecution\", func(t *testing.T) {\n\t\th.EXPECT().TerminateWorkflowExecution(ctx, &types.HistoryTerminateWorkflowExecutionRequest{}).Return(internalErr).Times(1)\n\t\terr := th.TerminateWorkflowExecution(ctx, &hist.TerminateWorkflowExecutionRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n}\n"
  },
  {
    "path": "service/matching/config/config.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage config\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n)\n\ntype (\n\t// Config represents configuration for cadence-matching service\n\tConfig struct {\n\t\tPersistenceMaxQPS       dynamicproperties.IntPropertyFn\n\t\tPersistenceGlobalMaxQPS dynamicproperties.IntPropertyFn\n\t\tEnableSyncMatch         dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tUserRPS                 dynamicproperties.IntPropertyFn\n\t\tWorkerRPS               dynamicproperties.IntPropertyFn\n\t\tDomainUserRPS           dynamicproperties.IntPropertyFnWithDomainFilter\n\t\tDomainWorkerRPS         dynamicproperties.IntPropertyFnWithDomainFilter\n\t\tShutdownDrainDuration   dynamicproperties.DurationPropertyFn\n\n\t\t// taskListManager configuration\n\t\tRangeSize                                 int64\n\t\tReadRangeSize                             dynamicproperties.IntPropertyFn\n\t\tEnableReturnAllTaskListKinds              dynamicproperties.BoolPropertyFn\n\t\tGetTasksBatchSize                         dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tUpdateAckInterval                         dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tIdleTasklistCheckInterval                 dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tMaxTasklistIdleTime                       dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tNumTasklistWritePartitions                dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tNumTasklistReadPartitions                 dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tForwarderMaxOutstandingPolls              dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tForwarderMaxOutstandingTasks              dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tForwarderMaxRatePerSecond                 dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tForwarderMaxChildrenPerNode               dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tAppendTaskTimeout                         dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tAsyncTaskDispatchTimeout                  dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tLocalPollWaitTime                         dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tLocalTaskWaitTime                         dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tTaskIsolationDuration                     dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tTaskIsolationPollerWindow                 dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tEnableGetNumberOfPartitionsFromCache      dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tPartitionUpscaleRPS                       dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tPartitionDownscaleFactor                  dynamicproperties.FloatPropertyFnWithTaskListInfoFilters\n\t\tPartitionUpscaleSustainedDuration         dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tPartitionDownscaleSustainedDuration       dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tAdaptiveScalerUpdateInterval              dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tEnableAdaptiveScaler                      dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tEnablePartitionEmptyCheck                 dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tEnableStandbyTaskCompletion               dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tEnableClientAutoConfig                    dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tQPSTrackerInterval                        dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tOverrideTaskListRPS                       dynamicproperties.FloatPropertyFnWithTaskListInfoFilters\n\t\tEnablePartitionIsolationGroupAssignment   dynamicproperties.BoolPropertyFnWithTaskListInfoFilters\n\t\tIsolationGroupUpscaleSustainedDuration    dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tIsolationGroupDownscaleSustainedDuration  dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tIsolationGroupHasPollersSustainedDuration dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tIsolationGroupNoPollersSustainedDuration  dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tIsolationGroupsPerPartition               dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\n\t\t// Time to hold a poll request before returning an empty response if there are no tasks\n\t\tLongPollExpirationInterval dynamicproperties.DurationPropertyFnWithTaskListInfoFilters\n\t\tMinTaskThrottlingBurstSize dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tMaxTaskDeleteBatchSize     dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\n\t\t// taskWriter configuration\n\t\tOutstandingTaskAppendsThreshold dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\t\tMaxTaskBatchSize                dynamicproperties.IntPropertyFnWithTaskListInfoFilters\n\n\t\tThrottledLogRPS dynamicproperties.IntPropertyFn\n\n\t\t// debugging configuration\n\t\tEnableDebugMode             bool // note that this value is initialized once on service start\n\t\tEnableTaskInfoLogByDomainID dynamicproperties.BoolPropertyFnWithDomainIDFilter\n\n\t\tActivityTaskSyncMatchWaitTime dynamicproperties.DurationPropertyFnWithDomainFilter\n\n\t\t// isolation configuration\n\t\tEnableTasklistIsolation dynamicproperties.BoolPropertyFnWithDomainFilter\n\t\tAllIsolationGroups      func() []string\n\t\t// hostname info\n\t\tHostName string\n\t\t// RPCConfig contains RPC configuration including ports and bindOnLocalHost\n\t\tRPCConfig config.RPC\n\t\t// rate limiter configuration\n\t\tTaskDispatchRPS    float64\n\t\tTaskDispatchRPSTTL time.Duration\n\t\t// task gc configuration\n\t\tMaxTimeBetweenTaskDeletes time.Duration\n\n\t\tEnableTasklistOwnershipGuard               dynamicproperties.BoolPropertyFn\n\t\tExcludeShortLivedTaskListsFromShardManager dynamicproperties.BoolPropertyFn\n\t\tPercentageOnboardedToShardManager          dynamicproperties.IntPropertyFn\n\t}\n\n\tForwarderConfig struct {\n\t\tForwarderMaxOutstandingPolls func() int\n\t\tForwarderMaxOutstandingTasks func() int\n\t\tForwarderMaxRatePerSecond    func() int\n\t\tForwarderMaxChildrenPerNode  func() int\n\t}\n\n\tTaskListConfig struct {\n\t\tForwarderConfig\n\t\tEnableSyncMatch func() bool\n\t\t// Time to hold a poll request before returning an empty response if there are no tasks\n\t\tLongPollExpirationInterval                func() time.Duration\n\t\tRangeSize                                 int64\n\t\tReadRangeSize                             dynamicproperties.IntPropertyFn\n\t\tActivityTaskSyncMatchWaitTime             dynamicproperties.DurationPropertyFnWithDomainFilter\n\t\tGetTasksBatchSize                         func() int\n\t\tUpdateAckInterval                         func() time.Duration\n\t\tIdleTasklistCheckInterval                 func() time.Duration\n\t\tMaxTasklistIdleTime                       func() time.Duration\n\t\tMinTaskThrottlingBurstSize                func() int\n\t\tMaxTaskDeleteBatchSize                    func() int\n\t\tAppendTaskTimeout                         func() time.Duration\n\t\tAsyncTaskDispatchTimeout                  func() time.Duration\n\t\tLocalPollWaitTime                         func() time.Duration\n\t\tLocalTaskWaitTime                         func() time.Duration\n\t\tPartitionUpscaleRPS                       func() int\n\t\tPartitionDownscaleFactor                  func() float64\n\t\tPartitionUpscaleSustainedDuration         func() time.Duration\n\t\tPartitionDownscaleSustainedDuration       func() time.Duration\n\t\tAdaptiveScalerUpdateInterval              func() time.Duration\n\t\tQPSTrackerInterval                        func() time.Duration\n\t\tOverrideTaskListRPS                       func() float64\n\t\tEnablePartitionIsolationGroupAssignment   func() bool\n\t\tIsolationGroupUpscaleSustainedDuration    func() time.Duration\n\t\tIsolationGroupDownscaleSustainedDuration  func() time.Duration\n\t\tIsolationGroupHasPollersSustainedDuration func() time.Duration\n\t\tIsolationGroupNoPollersSustainedDuration  func() time.Duration\n\t\tIsolationGroupsPerPartition               func() int\n\t\t// taskWriter configuration\n\t\tOutstandingTaskAppendsThreshold      func() int\n\t\tMaxTaskBatchSize                     func() int\n\t\tNumWritePartitions                   func() int\n\t\tNumReadPartitions                    func() int\n\t\tEnableGetNumberOfPartitionsFromCache func() bool\n\t\tEnableAdaptiveScaler                 func() bool\n\t\tEnablePartitionEmptyCheck            func() bool\n\t\t// isolation configuration\n\t\tEnableTasklistIsolation func() bool\n\t\t// A function which returns all the isolation groups\n\t\tAllIsolationGroups        func() []string\n\t\tTaskIsolationDuration     func() time.Duration\n\t\tTaskIsolationPollerWindow func() time.Duration\n\t\t// hostname\n\t\tHostName string\n\t\t// rate limiter configuration\n\t\tTaskDispatchRPS    float64\n\t\tTaskDispatchRPSTTL time.Duration\n\t\t// task gc configuration\n\t\tMaxTimeBetweenTaskDeletes time.Duration\n\t\t// standby task completion configuration\n\t\tEnableStandbyTaskCompletion func() bool\n\t\tEnableClientAutoConfig      func() bool\n\t}\n)\n\n// NewConfig returns new service config with default values\nfunc NewConfig(dc *dynamicconfig.Collection, hostName string, rpcConfig config.RPC, getIsolationGroups func() []string) *Config {\n\treturn &Config{\n\t\tPersistenceMaxQPS:                          dc.GetIntProperty(dynamicproperties.MatchingPersistenceMaxQPS),\n\t\tPersistenceGlobalMaxQPS:                    dc.GetIntProperty(dynamicproperties.MatchingPersistenceGlobalMaxQPS),\n\t\tEnableSyncMatch:                            dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableSyncMatch),\n\t\tUserRPS:                                    dc.GetIntProperty(dynamicproperties.MatchingUserRPS),\n\t\tWorkerRPS:                                  dc.GetIntProperty(dynamicproperties.MatchingWorkerRPS),\n\t\tDomainUserRPS:                              dc.GetIntPropertyFilteredByDomain(dynamicproperties.MatchingDomainUserRPS),\n\t\tDomainWorkerRPS:                            dc.GetIntPropertyFilteredByDomain(dynamicproperties.MatchingDomainWorkerRPS),\n\t\tRangeSize:                                  100000,\n\t\tReadRangeSize:                              dc.GetIntProperty(dynamicproperties.MatchingReadRangeSize),\n\t\tGetTasksBatchSize:                          dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingGetTasksBatchSize),\n\t\tUpdateAckInterval:                          dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingUpdateAckInterval),\n\t\tIdleTasklistCheckInterval:                  dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIdleTasklistCheckInterval),\n\t\tMaxTasklistIdleTime:                        dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MaxTasklistIdleTime),\n\t\tLongPollExpirationInterval:                 dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingLongPollExpirationInterval),\n\t\tMinTaskThrottlingBurstSize:                 dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingMinTaskThrottlingBurstSize),\n\t\tMaxTaskDeleteBatchSize:                     dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingMaxTaskDeleteBatchSize),\n\t\tOutstandingTaskAppendsThreshold:            dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingOutstandingTaskAppendsThreshold),\n\t\tMaxTaskBatchSize:                           dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingMaxTaskBatchSize),\n\t\tThrottledLogRPS:                            dc.GetIntProperty(dynamicproperties.MatchingThrottledLogRPS),\n\t\tNumTasklistWritePartitions:                 dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingNumTasklistWritePartitions),\n\t\tNumTasklistReadPartitions:                  dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingNumTasklistReadPartitions),\n\t\tForwarderMaxOutstandingPolls:               dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingForwarderMaxOutstandingPolls),\n\t\tForwarderMaxOutstandingTasks:               dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingForwarderMaxOutstandingTasks),\n\t\tForwarderMaxRatePerSecond:                  dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingForwarderMaxRatePerSecond),\n\t\tForwarderMaxChildrenPerNode:                dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingForwarderMaxChildrenPerNode),\n\t\tEnableGetNumberOfPartitionsFromCache:       dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache),\n\t\tShutdownDrainDuration:                      dc.GetDurationProperty(dynamicproperties.MatchingShutdownDrainDuration),\n\t\tEnableDebugMode:                            dc.GetBoolProperty(dynamicproperties.EnableDebugMode)(),\n\t\tEnableTaskInfoLogByDomainID:                dc.GetBoolPropertyFilteredByDomainID(dynamicproperties.MatchingEnableTaskInfoLogByDomainID),\n\t\tActivityTaskSyncMatchWaitTime:              dc.GetDurationPropertyFilteredByDomain(dynamicproperties.MatchingActivityTaskSyncMatchWaitTime),\n\t\tEnableTasklistIsolation:                    dc.GetBoolPropertyFilteredByDomain(dynamicproperties.EnableTasklistIsolation),\n\t\tAppendTaskTimeout:                          dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.AppendTaskTimeout),\n\t\tAsyncTaskDispatchTimeout:                   dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.AsyncTaskDispatchTimeout),\n\t\tEnableTasklistOwnershipGuard:               dc.GetBoolProperty(dynamicproperties.MatchingEnableTasklistGuardAgainstOwnershipShardLoss),\n\t\tLocalPollWaitTime:                          dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.LocalPollWaitTime),\n\t\tLocalTaskWaitTime:                          dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.LocalTaskWaitTime),\n\t\tPartitionUpscaleRPS:                        dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingPartitionUpscaleRPS),\n\t\tPartitionDownscaleFactor:                   dc.GetFloat64PropertyFilteredByTaskListInfo(dynamicproperties.MatchingPartitionDownscaleFactor),\n\t\tPartitionUpscaleSustainedDuration:          dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingPartitionUpscaleSustainedDuration),\n\t\tPartitionDownscaleSustainedDuration:        dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingPartitionDownscaleSustainedDuration),\n\t\tAdaptiveScalerUpdateInterval:               dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingAdaptiveScalerUpdateInterval),\n\t\tEnableAdaptiveScaler:                       dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableAdaptiveScaler),\n\t\tEnablePartitionEmptyCheck:                  dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnablePartitionEmptyCheck),\n\t\tQPSTrackerInterval:                         dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingQPSTrackerInterval),\n\t\tOverrideTaskListRPS:                        dc.GetFloat64PropertyFilteredByTaskListInfo(dynamicproperties.MatchingOverrideTaskListRPS),\n\t\tEnablePartitionIsolationGroupAssignment:    dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.EnablePartitionIsolationGroupAssignment),\n\t\tIsolationGroupUpscaleSustainedDuration:     dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIsolationGroupUpscaleSustainedDuration),\n\t\tIsolationGroupDownscaleSustainedDuration:   dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIsolationGroupDownscaleSustainedDuration),\n\t\tIsolationGroupHasPollersSustainedDuration:  dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIsolationGroupHasPollersSustainedDuration),\n\t\tIsolationGroupNoPollersSustainedDuration:   dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIsolationGroupNoPollersSustainedDuration),\n\t\tIsolationGroupsPerPartition:                dc.GetIntPropertyFilteredByTaskListInfo(dynamicproperties.MatchingIsolationGroupsPerPartition),\n\t\tTaskIsolationDuration:                      dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.TaskIsolationDuration),\n\t\tTaskIsolationPollerWindow:                  dc.GetDurationPropertyFilteredByTaskListInfo(dynamicproperties.TaskIsolationPollerWindow),\n\t\tHostName:                                   hostName,\n\t\tRPCConfig:                                  rpcConfig,\n\t\tTaskDispatchRPS:                            100000.0,\n\t\tTaskDispatchRPSTTL:                         time.Minute,\n\t\tMaxTimeBetweenTaskDeletes:                  time.Second,\n\t\tAllIsolationGroups:                         getIsolationGroups,\n\t\tEnableStandbyTaskCompletion:                dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableStandbyTaskCompletion),\n\t\tEnableClientAutoConfig:                     dc.GetBoolPropertyFilteredByTaskListInfo(dynamicproperties.MatchingEnableClientAutoConfig),\n\t\tEnableReturnAllTaskListKinds:               dc.GetBoolProperty(dynamicproperties.MatchingEnableReturnAllTaskListKinds),\n\t\tExcludeShortLivedTaskListsFromShardManager: dc.GetBoolProperty(dynamicproperties.MatchingExcludeShortLivedTaskListsFromShardManager),\n\t\tPercentageOnboardedToShardManager:          dc.GetIntProperty(dynamicproperties.MatchingPercentageOnboardedToShardManager),\n\t}\n}\n"
  },
  {
    "path": "service/matching/config/config_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype configTestCase struct {\n\tkey   dynamicproperties.Key\n\tvalue interface{}\n}\n\nfunc TestNewConfig(t *testing.T) {\n\thostname := \"hostname\"\n\tfields := map[string]configTestCase{\n\t\t\"PersistenceMaxQPS\":                          {dynamicproperties.MatchingPersistenceMaxQPS, 1},\n\t\t\"PersistenceGlobalMaxQPS\":                    {dynamicproperties.MatchingPersistenceGlobalMaxQPS, 2},\n\t\t\"EnableSyncMatch\":                            {dynamicproperties.MatchingEnableSyncMatch, true},\n\t\t\"UserRPS\":                                    {dynamicproperties.MatchingUserRPS, 3},\n\t\t\"WorkerRPS\":                                  {dynamicproperties.MatchingWorkerRPS, 4},\n\t\t\"DomainUserRPS\":                              {dynamicproperties.MatchingDomainUserRPS, 5},\n\t\t\"DomainWorkerRPS\":                            {dynamicproperties.MatchingDomainWorkerRPS, 6},\n\t\t\"RangeSize\":                                  {nil, int64(100000)},\n\t\t\"ReadRangeSize\":                              {dynamicproperties.MatchingReadRangeSize, 50000},\n\t\t\"GetTasksBatchSize\":                          {dynamicproperties.MatchingGetTasksBatchSize, 7},\n\t\t\"UpdateAckInterval\":                          {dynamicproperties.MatchingUpdateAckInterval, time.Duration(8)},\n\t\t\"IdleTasklistCheckInterval\":                  {dynamicproperties.MatchingIdleTasklistCheckInterval, time.Duration(9)},\n\t\t\"MaxTasklistIdleTime\":                        {dynamicproperties.MaxTasklistIdleTime, time.Duration(10)},\n\t\t\"LongPollExpirationInterval\":                 {dynamicproperties.MatchingLongPollExpirationInterval, time.Duration(11)},\n\t\t\"MinTaskThrottlingBurstSize\":                 {dynamicproperties.MatchingMinTaskThrottlingBurstSize, 12},\n\t\t\"MaxTaskDeleteBatchSize\":                     {dynamicproperties.MatchingMaxTaskDeleteBatchSize, 13},\n\t\t\"OutstandingTaskAppendsThreshold\":            {dynamicproperties.MatchingOutstandingTaskAppendsThreshold, 14},\n\t\t\"MaxTaskBatchSize\":                           {dynamicproperties.MatchingMaxTaskBatchSize, 15},\n\t\t\"ThrottledLogRPS\":                            {dynamicproperties.MatchingThrottledLogRPS, 16},\n\t\t\"NumTasklistWritePartitions\":                 {dynamicproperties.MatchingNumTasklistWritePartitions, 17},\n\t\t\"NumTasklistReadPartitions\":                  {dynamicproperties.MatchingNumTasklistReadPartitions, 18},\n\t\t\"ForwarderMaxOutstandingPolls\":               {dynamicproperties.MatchingForwarderMaxOutstandingPolls, 19},\n\t\t\"ForwarderMaxOutstandingTasks\":               {dynamicproperties.MatchingForwarderMaxOutstandingTasks, 20},\n\t\t\"ForwarderMaxRatePerSecond\":                  {dynamicproperties.MatchingForwarderMaxRatePerSecond, 21},\n\t\t\"ForwarderMaxChildrenPerNode\":                {dynamicproperties.MatchingForwarderMaxChildrenPerNode, 22},\n\t\t\"ShutdownDrainDuration\":                      {dynamicproperties.MatchingShutdownDrainDuration, time.Duration(23)},\n\t\t\"EnableDebugMode\":                            {dynamicproperties.EnableDebugMode, false},\n\t\t\"EnableTaskInfoLogByDomainID\":                {dynamicproperties.MatchingEnableTaskInfoLogByDomainID, true},\n\t\t\"ActivityTaskSyncMatchWaitTime\":              {dynamicproperties.MatchingActivityTaskSyncMatchWaitTime, time.Duration(24)},\n\t\t\"EnableTasklistIsolation\":                    {dynamicproperties.EnableTasklistIsolation, false},\n\t\t\"AsyncTaskDispatchTimeout\":                   {dynamicproperties.AsyncTaskDispatchTimeout, time.Duration(25)},\n\t\t\"LocalPollWaitTime\":                          {dynamicproperties.LocalPollWaitTime, time.Duration(10)},\n\t\t\"LocalTaskWaitTime\":                          {dynamicproperties.LocalTaskWaitTime, time.Duration(10)},\n\t\t\"HostName\":                                   {nil, hostname},\n\t\t\"RPCConfig\":                                  {nil, config.RPC{}},\n\t\t\"TaskDispatchRPS\":                            {nil, 100000.0},\n\t\t\"TaskDispatchRPSTTL\":                         {nil, time.Minute},\n\t\t\"MaxTimeBetweenTaskDeletes\":                  {nil, time.Second},\n\t\t\"AllIsolationGroups\":                         {nil, []string{\"zone-1\", \"zone-2\"}},\n\t\t\"EnableTasklistOwnershipGuard\":               {dynamicproperties.MatchingEnableTasklistGuardAgainstOwnershipShardLoss, false},\n\t\t\"EnableGetNumberOfPartitionsFromCache\":       {dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache, false},\n\t\t\"EnablePartitionEmptyCheck\":                  {dynamicproperties.MatchingEnablePartitionEmptyCheck, true},\n\t\t\"PartitionUpscaleRPS\":                        {dynamicproperties.MatchingPartitionUpscaleRPS, 30},\n\t\t\"PartitionDownscaleFactor\":                   {dynamicproperties.MatchingPartitionDownscaleFactor, 31.0},\n\t\t\"PartitionUpscaleSustainedDuration\":          {dynamicproperties.MatchingPartitionUpscaleSustainedDuration, time.Duration(32)},\n\t\t\"PartitionDownscaleSustainedDuration\":        {dynamicproperties.MatchingPartitionDownscaleSustainedDuration, time.Duration(33)},\n\t\t\"AdaptiveScalerUpdateInterval\":               {dynamicproperties.MatchingAdaptiveScalerUpdateInterval, time.Duration(34)},\n\t\t\"EnableAdaptiveScaler\":                       {dynamicproperties.MatchingEnableAdaptiveScaler, true},\n\t\t\"QPSTrackerInterval\":                         {dynamicproperties.MatchingQPSTrackerInterval, 5 * time.Second},\n\t\t\"OverrideTaskListRPS\":                        {dynamicproperties.MatchingOverrideTaskListRPS, 1500.0},\n\t\t\"EnableStandbyTaskCompletion\":                {dynamicproperties.MatchingEnableStandbyTaskCompletion, false},\n\t\t\"EnableClientAutoConfig\":                     {dynamicproperties.MatchingEnableClientAutoConfig, false},\n\t\t\"TaskIsolationDuration\":                      {dynamicproperties.TaskIsolationDuration, time.Duration(35)},\n\t\t\"TaskIsolationPollerWindow\":                  {dynamicproperties.TaskIsolationPollerWindow, time.Duration(36)},\n\t\t\"EnablePartitionIsolationGroupAssignment\":    {dynamicproperties.EnablePartitionIsolationGroupAssignment, true},\n\t\t\"IsolationGroupUpscaleSustainedDuration\":     {dynamicproperties.MatchingIsolationGroupUpscaleSustainedDuration, time.Duration(37)},\n\t\t\"IsolationGroupDownscaleSustainedDuration\":   {dynamicproperties.MatchingIsolationGroupDownscaleSustainedDuration, time.Duration(38)},\n\t\t\"IsolationGroupHasPollersSustainedDuration\":  {dynamicproperties.MatchingIsolationGroupHasPollersSustainedDuration, time.Duration(39)},\n\t\t\"IsolationGroupNoPollersSustainedDuration\":   {dynamicproperties.MatchingIsolationGroupNoPollersSustainedDuration, time.Duration(40)},\n\t\t\"IsolationGroupsPerPartition\":                {dynamicproperties.MatchingIsolationGroupsPerPartition, 41},\n\t\t\"EnableReturnAllTaskListKinds\":               {dynamicproperties.MatchingEnableReturnAllTaskListKinds, true},\n\t\t\"AppendTaskTimeout\":                          {dynamicproperties.AppendTaskTimeout, time.Duration(42)},\n\t\t\"ExcludeShortLivedTaskListsFromShardManager\": {dynamicproperties.MatchingExcludeShortLivedTaskListsFromShardManager, false},\n\t\t\"PercentageOnboardedToShardManager\":          {dynamicproperties.MatchingPercentageOnboardedToShardManager, 0},\n\t}\n\tclient := dynamicconfig.NewInMemoryClient()\n\tfor fieldName, expected := range fields {\n\t\tif expected.key != nil {\n\t\t\terr := client.UpdateValue(expected.key, expected.value)\n\t\t\tif err != nil {\n\t\t\t\tt.Errorf(\"Failed to update config for %s: %s\", fieldName, err)\n\t\t\t}\n\t\t}\n\t}\n\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\n\tconfig := NewConfig(dc, hostname, config.RPC{}, isolationGroupsHelper)\n\n\tassertFieldsMatch(t, *config, fields)\n}\n\nfunc assertFieldsMatch(t *testing.T, config interface{}, fields map[string]configTestCase) {\n\tconfigType := reflect.ValueOf(config)\n\n\tfor i := 0; i < configType.NumField(); i++ {\n\t\tf := configType.Field(i)\n\t\tfieldName := configType.Type().Field(i).Name\n\n\t\tif expected, ok := fields[fieldName]; ok {\n\t\t\tactual := getValue(&f)\n\t\t\tif f.Kind() == reflect.Slice {\n\t\t\t\tassert.ElementsMatch(t, expected.value, actual, \"Incorrect value for field: %s\", fieldName)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, expected.value, actual, \"Incorrect value for field: %s\", fieldName)\n\t\t\t}\n\n\t\t} else {\n\t\t\tt.Errorf(\"Unknown property on Config: %s\", fieldName)\n\t\t}\n\t}\n}\n\nfunc getValue(f *reflect.Value) interface{} {\n\tswitch f.Kind() {\n\tcase reflect.Func:\n\t\tswitch fn := f.Interface().(type) {\n\t\tcase dynamicproperties.IntPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.IntPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.IntPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.BoolPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFnWithDomainIDFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.BoolPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.DurationPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.DurationPropertyFnWithDomainFilter:\n\t\t\treturn fn(\"domain\")\n\t\tcase dynamicproperties.DurationPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase dynamicproperties.FloatPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.MapPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.StringPropertyFn:\n\t\t\treturn fn()\n\t\tcase dynamicproperties.FloatPropertyFnWithTaskListInfoFilters:\n\t\t\treturn fn(\"domain\", \"tasklist\", int(types.TaskListTypeDecision))\n\t\tcase func() []string:\n\t\t\treturn fn()\n\t\tdefault:\n\t\t\tpanic(\"Unable to handle type: \" + f.Type().Name())\n\t\t}\n\tdefault:\n\t\treturn f.Interface()\n\t}\n}\n\nfunc isolationGroupsHelper() []string {\n\treturn []string{\"zone-1\", \"zone-2\"}\n}\n\nfunc TestGetTasksBatchSizeValidation(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tconfigValue int\n\t\texpected    int\n\t}{\n\t\t{\n\t\t\tname:        \"valid positive batch size\",\n\t\t\tconfigValue: 1000,\n\t\t\texpected:    1000,\n\t\t},\n\t\t{\n\t\t\tname:        \"valid small batch size\",\n\t\t\tconfigValue: 1,\n\t\t\texpected:    1,\n\t\t},\n\t\t{\n\t\t\tname:        \"zero batch size returns zero (validation happens at usage site)\",\n\t\t\tconfigValue: 0,\n\t\t\texpected:    0,\n\t\t},\n\t\t{\n\t\t\tname:        \"negative batch size returns negative (validation happens at usage site)\",\n\t\t\tconfigValue: -5,\n\t\t\texpected:    -5,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Create a mock dynamic config client\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\terr := client.UpdateValue(dynamicproperties.MatchingGetTasksBatchSize, tt.configValue)\n\t\t\tassert.NoError(t, err)\n\n\t\t\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\t\t\tconfig := NewConfig(dc, \"test-host\", config.RPC{}, isolationGroupsHelper)\n\n\t\t\t// Test that GetTasksBatchSize returns the raw value (validation happens at usage site)\n\t\t\tresult := config.GetTasksBatchSize(\"test-domain\", \"test-tasklist\", int(types.TaskListTypeDecision))\n\t\t\tassert.Equal(t, tt.expected, result, \"GetTasksBatchSize should return raw config value\")\n\n\t\t\t// Test with different task list types\n\t\t\tresult = config.GetTasksBatchSize(\"test-domain\", \"test-tasklist\", int(types.TaskListTypeActivity))\n\t\t\tassert.Equal(t, tt.expected, result, \"GetTasksBatchSize should return raw config value for activity task list\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/event/logger.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage event\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar enabled = false\n\nfunc init() {\n\tenabled = os.Getenv(\"MATCHING_LOG_EVENTS\") == \"true\"\n}\n\ntype E struct {\n\tpersistence.TaskInfo\n\tTaskListName string\n\tTaskListKind *types.TaskListKind\n\tTaskListType int // persistence.TaskListTypeDecision or persistence.TaskListTypeActivity\n\n\tEventTime time.Time\n\n\t// EventName describes the event. It is used to query events in simulations so don't change existing event names.\n\tEventName string\n\tHost      string\n\tPayload   map[string]any\n}\n\nfunc Log(events ...E) {\n\tif !enabled {\n\t\treturn\n\t}\n\tfor _, e := range events {\n\t\te.EventTime = time.Now()\n\t\tdata, err := json.Marshal(e)\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"failed to marshal event: %v\", err)\n\t\t}\n\n\t\tfmt.Printf(\"Matching New Event: %s\\n\", data)\n\t}\n}\n"
  },
  {
    "path": "service/matching/handler/context.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype handlerContext struct {\n\tcontext.Context\n\tscope  metrics.Scope\n\tlogger log.Logger\n}\n\nfunc newHandlerContext(\n\tctx context.Context,\n\tdomainName string,\n\ttaskList *types.TaskList,\n\tmetricsClient metrics.Client,\n\tmetricsScope metrics.ScopeIdx,\n\tlogger log.Logger,\n) *handlerContext {\n\treturn &handlerContext{\n\t\tContext: ctx,\n\t\tscope:   common.NewPerTaskListScope(domainName, taskList.GetName(), taskList.GetKind(), metricsClient, metricsScope).Tagged(metrics.GetContextTags(ctx)...),\n\t\tlogger:  logger.WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(taskList.GetName())),\n\t}\n}\n\n// startProfiling initiates recording of request metrics\nfunc (reqCtx *handlerContext) startProfiling(wg *sync.WaitGroup) metrics.Stopwatch {\n\twg.Wait()\n\tsw := reqCtx.scope.StartTimer(metrics.CadenceLatencyPerTaskList)\n\treqCtx.scope.IncCounter(metrics.CadenceRequestsPerTaskList)\n\treturn sw\n}\n\nfunc (reqCtx *handlerContext) handleErr(err error) error {\n\tif err == nil {\n\t\treturn nil\n\t}\n\n\tlogger := reqCtx.logger.Helper()\n\n\tswitch {\n\tcase errors.As(err, new(*types.InternalServiceError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceFailuresPerTaskList)\n\t\tlogger.Error(\"Internal service error\", tag.Error(err))\n\t\treturn err\n\tcase errors.As(err, new(*types.BadRequestError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrBadRequestPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.EntityNotExistsError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrEntityNotExistsPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.WorkflowExecutionAlreadyStartedError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrExecutionAlreadyStartedPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.DomainAlreadyExistsError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrDomainAlreadyExistsPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.QueryFailedError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrQueryFailedPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.LimitExceededError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrLimitExceededPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.ServiceBusyError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrServiceBusyPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.DomainNotActiveError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrDomainNotActivePerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.RemoteSyncMatchedError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrRemoteSyncMatchFailedPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.StickyWorkerUnavailableError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrStickyWorkerUnavailablePerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*types.ReadOnlyPartitionError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrReadOnlyPartitionPerTaskListCounter)\n\t\treturn err\n\tcase errors.As(err, new(*cadence_errors.TaskListNotOwnedByHostError)):\n\t\treqCtx.scope.IncCounter(metrics.CadenceErrTaskListNotOwnedByHostPerTaskListCounter)\n\t\treturn err\n\tdefault:\n\t\treqCtx.scope.IncCounter(metrics.CadenceFailuresPerTaskList)\n\t\tlogger.Error(\"Uncategorized error\", tag.Error(err))\n\t\treturn &types.InternalServiceError{Message: err.Error()}\n\t}\n}\n"
  },
  {
    "path": "service/matching/handler/context_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestHandleErrNil(t *testing.T) {\n\treqCtx := &handlerContext{\n\t\tscope:  metrics.NoopScope,\n\t\tlogger: log.NewNoop(),\n\t}\n\n\terr := reqCtx.handleErr(nil)\n\tassert.NoError(t, err)\n}\n\nfunc TestHandleErrKnowErrors(t *testing.T) {\n\ttestCases := []struct {\n\t\tname    string\n\t\terr     error\n\t\twantErr error\n\t}{\n\t\t{\n\t\t\tname: \"InternalServiceError\",\n\t\t\terr:  &types.InternalServiceError{},\n\t\t},\n\t\t{\n\t\t\tname: \"BadRequestError\",\n\t\t\terr:  &types.BadRequestError{},\n\t\t},\n\t\t{\n\t\t\tname: \"EntityNotExistsError\",\n\t\t\terr:  &types.EntityNotExistsError{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionAlreadyStartedError\",\n\t\t\terr:  &types.WorkflowExecutionAlreadyStartedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"DomainAlreadyExistsError\",\n\t\t\terr:  &types.DomainAlreadyExistsError{},\n\t\t},\n\t\t{\n\t\t\tname: \"QueryFailedError\",\n\t\t\terr:  &types.QueryFailedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"LimitExceededError\",\n\t\t\terr:  &types.LimitExceededError{},\n\t\t},\n\t\t{\n\t\t\tname: \"ServiceBusyError\",\n\t\t\terr:  &types.ServiceBusyError{},\n\t\t},\n\t\t{\n\t\t\tname: \"DomainNotActiveError\",\n\t\t\terr:  &types.DomainNotActiveError{},\n\t\t},\n\t\t{\n\t\t\tname: \"RemoteSyncMatchedError\",\n\t\t\terr:  &types.RemoteSyncMatchedError{},\n\t\t},\n\t\t{\n\t\t\tname: \"StickyWorkerUnavailableError\",\n\t\t\terr:  &types.StickyWorkerUnavailableError{},\n\t\t},\n\t\t{\n\t\t\tname: \"ReadOnlyPartitionError\",\n\t\t\terr:  &types.ReadOnlyPartitionError{},\n\t\t},\n\t\t{\n\t\t\tname: \"TaskListNotOwnedByHostError\",\n\t\t\terr:  &cadence_errors.TaskListNotOwnedByHostError{},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\treqCtx := &handlerContext{\n\t\t\tscope:  metrics.NoopScope,\n\t\t\tlogger: log.NewNoop(),\n\t\t}\n\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\thandledErr := reqCtx.handleErr(tc.err)\n\t\t\trequire.Equal(t, tc.err, handledErr)\n\n\t\t\twrappedErr := fmt.Errorf(\"wrapped: %w\", tc.err)\n\t\t\twrappedHandledErr := reqCtx.handleErr(wrappedErr)\n\t\t\trequire.Equal(t, wrappedErr, wrappedHandledErr)\n\t\t})\n\t}\n}\n\nfunc TestHandleErrUncategorizedError(t *testing.T) {\n\treqCtx := &handlerContext{\n\t\tscope:  metrics.NoopScope,\n\t\tlogger: log.NewNoop(),\n\t}\n\n\terr := reqCtx.handleErr(assert.AnError)\n\tassert.ErrorAs(t, err, new(*types.InternalServiceError))\n}\n"
  },
  {
    "path": "service/matching/handler/engine.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nconst (\n\t// If sticky poller is not seem in last 10s, we treat it as sticky worker unavailable\n\t// This seems aggressive, but the default sticky schedule_to_start timeout is 5s, so 10s seems reasonable.\n\t_stickyPollerUnavailableWindow = 10 * time.Second\n\n\t// _defaultSDReportTTL is the default TTL for shard status reports from matching executor to shard distributor.\n\t// This controls how frequently the executor reports its shard load/status to the distributor.\n\t_defaultSDReportTTL = 1 * time.Minute\n\t// _recordTaskStartedTimeout is the maximum time allowed for RecordDecisionTaskStarted or RecordActivityTaskStarted\n\t// Any time we spend attempting to start an individual task is blocking that poller from starting a different task.\n\t// If a task is taking too long we'd rather try other tasks to maintain higher throughput.\n\t_recordTaskStartedTimeout = time.Second\n)\n\n// Implements matching.Engine\n// TODO: Switch implementation from lock/channel based to a partitioned agent\n// to simplify code and reduce possibility of synchronization errors.\ntype (\n\tqueryResult struct {\n\t\tworkerResponse *types.MatchingRespondQueryTaskCompletedRequest\n\t\tinternalError  error\n\t}\n\n\t// lockableQueryTaskMap maps query TaskID (which is a UUID generated in QueryWorkflow() call) to a channel\n\t// that QueryWorkflow() will block on. The channel is unblocked either by worker sending response through\n\t// RespondQueryTaskCompleted() or through an internal service error causing cadence to be unable to dispatch\n\t// query task to workflow worker.\n\tlockableQueryTaskMap struct {\n\t\tsync.RWMutex\n\t\tqueryTaskMap map[string]chan *queryResult\n\t}\n\n\tmatchingEngineImpl struct {\n\t\ttaskListCreationLock           sync.Mutex\n\t\ttaskListRegistry               tasklist.TaskListRegistry\n\t\tshutdownCompletion             *sync.WaitGroup\n\t\tshutdown                       chan struct{}\n\t\ttaskManager                    persistence.TaskManager\n\t\tclusterMetadata                cluster.Metadata\n\t\thistoryService                 history.Client\n\t\tmatchingClient                 matching.Client\n\t\ttokenSerializer                common.TaskTokenSerializer\n\t\tlogger                         log.Logger\n\t\tmetricsClient                  metrics.Client\n\t\tmetricsScope                   tally.Scope\n\t\texecutor                       executorclient.Executor[tasklist.ShardProcessor]\n\t\ttaskListsFactory               *tasklist.ShardProcessorFactory\n\t\tconfig                         *config.Config\n\t\tlockableQueryTaskMap           lockableQueryTaskMap\n\t\tdomainCache                    cache.DomainCache\n\t\tversionChecker                 client.VersionChecker\n\t\tmembershipResolver             membership.Resolver\n\t\tisolationState                 isolationgroup.State\n\t\ttimeSource                     clock.TimeSource\n\t\tfailoverNotificationVersion    int64\n\t\tShardDistributorMatchingConfig clientcommon.Config\n\t\tdrainObserver                  clientcommon.DrainSignalObserver\n\t}\n)\n\nvar (\n\trecordTaskStartedRetryPolicy = common.CreateRecordTaskStartedRetryPolicy()\n\n\terrPumpClosed = errors.New(\"task list pump closed its channel\")\n\n\t_stickyPollerUnavailableError = &types.StickyWorkerUnavailableError{Message: \"sticky worker is unavailable, please use non-sticky task list.\"}\n)\n\nvar _ Engine = (*matchingEngineImpl)(nil) // Asserts that interface is indeed implemented\n\n// NewEngine creates an instance of matching engine\nfunc NewEngine(\n\ttaskManager persistence.TaskManager,\n\tclusterMetadata cluster.Metadata,\n\thistoryService history.Client,\n\tmatchingClient matching.Client,\n\tconfig *config.Config,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tmetricsScope tally.Scope,\n\tdomainCache cache.DomainCache,\n\tresolver membership.Resolver,\n\tisolationState isolationgroup.State,\n\ttimeSource clock.TimeSource,\n\tshardDistributorClient executorclient.Client,\n\tShardDistributorMatchingConfig clientcommon.Config,\n\tdrainObserver clientcommon.DrainSignalObserver,\n) Engine {\n\te := &matchingEngineImpl{\n\t\ttaskListRegistry:               tasklist.NewTaskListRegistry(metricsClient),\n\t\tshutdown:                       make(chan struct{}),\n\t\tshutdownCompletion:             &sync.WaitGroup{},\n\t\ttaskManager:                    taskManager,\n\t\tclusterMetadata:                clusterMetadata,\n\t\thistoryService:                 historyService,\n\t\ttokenSerializer:                common.NewJSONTaskTokenSerializer(),\n\t\tlogger:                         logger.WithTags(tag.ComponentMatchingEngine),\n\t\tmetricsClient:                  metricsClient,\n\t\tmetricsScope:                   metricsScope,\n\t\tmatchingClient:                 matchingClient,\n\t\tconfig:                         config,\n\t\tlockableQueryTaskMap:           lockableQueryTaskMap{queryTaskMap: make(map[string]chan *queryResult)},\n\t\tdomainCache:                    domainCache,\n\t\tversionChecker:                 client.NewVersionChecker(),\n\t\tmembershipResolver:             resolver,\n\t\tisolationState:                 isolationState,\n\t\ttimeSource:                     timeSource,\n\t\tShardDistributorMatchingConfig: ShardDistributorMatchingConfig,\n\t\tdrainObserver:                  drainObserver,\n\t}\n\n\te.setupExecutor(shardDistributorClient)\n\te.shutdownCompletion.Add(1)\n\tgo e.runMembershipChangeLoop()\n\n\treturn e\n}\n\nfunc (e *matchingEngineImpl) Start() {\n\te.executor.Start(context.Background())\n\te.registerDomainFailoverCallback()\n}\n\nfunc (e *matchingEngineImpl) Stop() {\n\tclose(e.shutdown)\n\te.executor.Stop()\n\t// Executes Stop() on each task list outside of lock\n\tfor _, l := range e.taskListRegistry.AllManagers() {\n\t\tl.Stop()\n\t}\n\te.unregisterDomainFailoverCallback()\n\te.shutdownCompletion.Wait()\n}\n\nfunc (e *matchingEngineImpl) setupExecutor(shardDistributorExecutorClient executorclient.Client) {\n\t// If no shard-distributor namespaces are configured, use a no-op executor so that\n\t// the matching service falls back to local hash-ring assignment entirely.\n\tif len(e.ShardDistributorMatchingConfig.Namespaces) == 0 {\n\t\te.logger.Info(\"No shard-distributor-matching namespaces configured, using no-op executor\")\n\t\te.executor = executorclient.NewNoopExecutor[tasklist.ShardProcessor]()\n\n\t\ttaskListFactory := &tasklist.ShardProcessorFactory{\n\t\t\tTaskListsRegistry: e.taskListRegistry,\n\t\t\tReportTTL:         _defaultSDReportTTL,\n\t\t\tTimeSource:        e.timeSource,\n\t\t}\n\t\te.taskListsFactory = taskListFactory\n\t\treturn\n\t}\n\n\tcfg, reportTTL := e.getValidatedShardDistributorConfig()\n\n\ttaskListFactory := &tasklist.ShardProcessorFactory{\n\t\tTaskListsRegistry: e.taskListRegistry,\n\t\tReportTTL:         reportTTL,\n\t\tTimeSource:        e.timeSource,\n\t}\n\te.taskListsFactory = taskListFactory\n\n\t// Get the IP address to advertise to external services\n\t// This respects bindOnLocalHost config (127.0.0.1 for local dev, external IP for production)\n\thostIP, err := rpc.GetListenIP(e.config.RPCConfig)\n\tif err != nil {\n\t\te.logger.Fatal(\"Failed to get listen IP\", tag.Error(err))\n\t}\n\n\tparams := executorclient.Params[tasklist.ShardProcessor]{\n\t\tExecutorClient:        shardDistributorExecutorClient,\n\t\tMetricsScope:          e.metricsScope,\n\t\tLogger:                e.logger,\n\t\tShardProcessorFactory: taskListFactory,\n\t\tConfig:                cfg,\n\t\tTimeSource:            e.timeSource,\n\t\tMetadata: map[string]string{\n\t\t\t\"tchannel\": fmt.Sprintf(\"%d\", e.config.RPCConfig.Port),\n\t\t\t\"grpc\":     fmt.Sprintf(\"%d\", e.config.RPCConfig.GRPCPort),\n\t\t\t\"hostIP\":   hostIP.String(),\n\t\t},\n\t\tDrainObserver: e.drainObserver,\n\t}\n\texecutor, err := executorclient.NewExecutor[tasklist.ShardProcessor](params)\n\tif err != nil {\n\t\te.logger.Fatal(\"Failed to create new executor\", tag.Error(err))\n\t}\n\n\te.executor = executor\n}\n\nfunc (e *matchingEngineImpl) getValidatedShardDistributorConfig() (clientcommon.Config, time.Duration) {\n\tcfg := e.ShardDistributorMatchingConfig\n\n\tif len(cfg.Namespaces) > 1 {\n\t\te.logger.Fatal(\"matching service does not support multiple namespaces\", tag.Value(cfg.Namespaces))\n\t}\n\n\t// Get TTLReport from config, default if not configured\n\treportTTL := _defaultSDReportTTL\n\tif len(cfg.Namespaces) == 1 && cfg.Namespaces[0].TTLReport != 0 {\n\t\treportTTL = cfg.Namespaces[0].TTLReport\n\t}\n\treturn cfg, reportTTL\n}\n\nfunc (e *matchingEngineImpl) String() string {\n\t// Executes taskList.String() on each task list outside of lock\n\tbuf := new(bytes.Buffer)\n\n\tfor i, tl := range e.taskListRegistry.AllManagers() {\n\t\tif i >= 1000 {\n\t\t\tbreak\n\t\t}\n\t\tfmt.Fprintf(buf, \"\\n%s\", tl.String())\n\t}\n\treturn buf.String()\n}\n\n// Returns taskListManager for a task list. If not already cached gets new range from DB and\n// if successful creates one.\nfunc (e *matchingEngineImpl) getOrCreateTaskListManager(ctx context.Context, taskList *tasklist.Identifier, taskListKind types.TaskListKind) (tasklist.Manager, error) {\n\t// The first check is an optimization so almost all requests will have a task list manager\n\t// and return avoiding the write lock\n\tresult, ok := e.taskListRegistry.ManagerByTaskListIdentifier(*taskList)\n\texcludedFromShardDistributor := e.isExcludedFromShardDistributor(taskList.GetName())\n\n\t// Task lists excluded from the ShardDistributor (short-lived task lists with UUIDs) bypass\n\t// the executor/shard-processor entirely and always use local hash-ring assignment.\n\tif excludedFromShardDistributor && ok {\n\t\treturn result, nil\n\t}\n\tif !excludedFromShardDistributor {\n\t\tsp, _ := e.executor.GetShardProcess(ctx, taskList.GetName())\n\t\tif sp != nil && ok {\n\t\t\treturn result, nil\n\t\t}\n\t}\n\n\terr := e.errIfShardOwnershipLost(ctx, taskList)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// If it gets here, write lock and check again in case a task list is created between the two locks\n\te.taskListCreationLock.Lock()\n\tif result, ok := e.taskListRegistry.ManagerByTaskListIdentifier(*taskList); ok {\n\t\te.taskListCreationLock.Unlock()\n\t\treturn result, nil\n\t}\n\n\t// common tagged logger\n\tlogger := e.logger.WithTags(\n\t\ttag.WorkflowTaskListName(taskList.GetName()),\n\t\ttag.WorkflowTaskListType(taskList.GetType()),\n\t\ttag.WorkflowDomainID(taskList.GetDomainID()),\n\t)\n\n\tlogger.Info(\"Task list manager state changed\", tag.LifeCycleStarting)\n\tparams := tasklist.ManagerParams{\n\t\tDomainCache:     e.domainCache,\n\t\tLogger:          e.logger,\n\t\tMetricsClient:   e.metricsClient,\n\t\tTaskManager:     e.taskManager,\n\t\tClusterMetadata: e.clusterMetadata,\n\t\tIsolationState:  e.isolationState,\n\t\tMatchingClient:  e.matchingClient,\n\t\tRegistry:        e.taskListRegistry,\n\t\tTaskList:        taskList,\n\t\tTaskListKind:    taskListKind,\n\t\tCfg:             e.config,\n\t\tTimeSource:      e.timeSource,\n\t\tCreateTime:      e.timeSource.Now(),\n\t\tHistoryService:  e.historyService,\n\t}\n\tmgr, err := tasklist.NewManager(params)\n\tif err != nil {\n\t\te.taskListCreationLock.Unlock()\n\t\tlogger.Info(\"Task list manager state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t\treturn nil, err\n\t}\n\n\te.taskListRegistry.Register(*taskList, mgr)\n\te.taskListCreationLock.Unlock()\n\n\terr = mgr.Start(context.Background())\n\tif err != nil {\n\t\tlogger.Info(\"Task list manager state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t\treturn nil, err\n\t}\n\n\tlogger.Info(\"Task list manager state changed\", tag.LifeCycleStarted)\n\tevent.Log(event.E{\n\t\tTaskListName: taskList.GetName(),\n\t\tTaskListKind: &taskListKind,\n\t\tTaskListType: taskList.GetType(),\n\t\tTaskInfo: persistence.TaskInfo{\n\t\t\tDomainID: taskList.GetDomainID(),\n\t\t},\n\t\tEventName: \"TaskListManager Started\",\n\t\tHost:      e.config.HostName,\n\t})\n\treturn mgr, nil\n}\n\n// AddDecisionTask either delivers task directly to waiting poller or save it into task list persistence.\nfunc (e *matchingEngineImpl) AddDecisionTask(\n\thCtx *handlerContext,\n\trequest *types.AddDecisionTaskRequest,\n) (*types.AddDecisionTaskResponse, error) {\n\tstartT := time.Now()\n\tdomainID := request.GetDomainUUID()\n\ttaskListName := request.GetTaskList().GetName()\n\ttaskListKind := request.GetTaskList().GetKind()\n\ttaskListType := persistence.TaskListTypeDecision\n\n\tevent.Log(event.E{\n\t\tTaskListName: taskListName,\n\t\tTaskListKind: &taskListKind,\n\t\tTaskListType: taskListType,\n\t\tTaskInfo: persistence.TaskInfo{\n\t\t\tDomainID:        domainID,\n\t\t\tScheduleID:      request.GetScheduleID(),\n\t\t\tPartitionConfig: request.GetPartitionConfig(),\n\t\t},\n\t\tEventName: \"Received AddDecisionTask\",\n\t\tHost:      e.config.HostName,\n\t\tPayload: map[string]any{\n\t\t\t\"RequestForwardedFrom\": request.GetForwardedFrom(),\n\t\t},\n\t})\n\te.emitInfoOrDebugLog(\n\t\tdomainID,\n\t\t\"Received AddDecisionTask\",\n\t\ttag.WorkflowTaskListName(taskListName),\n\t\ttag.WorkflowID(request.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.Execution.GetRunID()),\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowTaskListType(taskListType),\n\t\ttag.WorkflowScheduleID(request.GetScheduleID()),\n\t\ttag.WorkflowTaskListKind(int32(request.GetTaskList().GetKind())),\n\t)\n\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// get the domainName\n\tdomainName, err := e.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Only emit traffic metrics if the tasklist is not sticky and is not forwarded\n\tif int32(request.GetTaskList().GetKind()) == 0 && request.ForwardedFrom == \"\" {\n\t\te.metricsClient.Scope(metrics.MatchingAddTaskScope).Tagged(metrics.DomainTag(domainName),\n\t\t\tmetrics.TaskListTag(taskListName), metrics.TaskListTypeTag(\"decision_task\"),\n\t\t\tmetrics.MatchingHostTag(e.config.HostName)).IncCounter(metrics.CadenceTasklistRequests)\n\t\te.emitInfoOrDebugLog(domainID, \"Emitting tasklist counter on decision task\",\n\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\ttag.Dynamic(\"taskListBaseName\", taskListID.GetRoot()))\n\t}\n\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif taskListKind == types.TaskListKindSticky {\n\t\t// check if the sticky worker is still available, if not, fail this request early\n\t\tif !tlMgr.HasPollerAfter(e.timeSource.Now().Add(-_stickyPollerUnavailableWindow)) {\n\t\t\treturn nil, _stickyPollerUnavailableError\n\t\t}\n\t}\n\n\ttaskInfo := &persistence.TaskInfo{\n\t\tDomainID:                      domainID,\n\t\tRunID:                         request.Execution.GetRunID(),\n\t\tWorkflowID:                    request.Execution.GetWorkflowID(),\n\t\tScheduleID:                    request.GetScheduleID(),\n\t\tScheduleToStartTimeoutSeconds: request.GetScheduleToStartTimeoutSeconds(),\n\t\tCreatedTime:                   e.timeSource.Now(),\n\t\tPartitionConfig:               request.GetPartitionConfig(),\n\t}\n\n\tsyncMatched, err := tlMgr.AddTask(hCtx.Context, tasklist.AddTaskParams{\n\t\tTaskInfo:      taskInfo,\n\t\tSource:        request.GetSource(),\n\t\tForwardedFrom: request.GetForwardedFrom(),\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif syncMatched {\n\t\thCtx.scope.RecordTimer(metrics.SyncMatchLatencyPerTaskList, time.Since(startT))\n\t}\n\treturn &types.AddDecisionTaskResponse{\n\t\tPartitionConfig: tlMgr.TaskListPartitionConfig(),\n\t}, nil\n}\n\n// AddActivityTask either delivers task directly to waiting poller or save it into task list persistence.\nfunc (e *matchingEngineImpl) AddActivityTask(\n\thCtx *handlerContext,\n\trequest *types.AddActivityTaskRequest,\n) (*types.AddActivityTaskResponse, error) {\n\tstartT := time.Now()\n\tdomainID := request.GetDomainUUID()\n\ttaskListName := request.GetTaskList().GetName()\n\ttaskListKind := request.GetTaskList().GetKind()\n\ttaskListType := persistence.TaskListTypeActivity\n\n\te.emitInfoOrDebugLog(\n\t\tdomainID,\n\t\t\"Received AddActivityTask\",\n\t\ttag.WorkflowTaskListName(taskListName),\n\t\ttag.WorkflowID(request.Execution.GetWorkflowID()),\n\t\ttag.WorkflowRunID(request.Execution.GetRunID()),\n\t\ttag.WorkflowDomainID(domainID),\n\t\ttag.WorkflowTaskListType(taskListType),\n\t\ttag.WorkflowScheduleID(request.GetScheduleID()),\n\t\ttag.WorkflowTaskListKind(int32(request.GetTaskList().GetKind())),\n\t)\n\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// get the domainName\n\tdomainName, err := e.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// Only emit traffic metrics if the tasklist is not sticky and is not forwarded\n\tif request.GetTaskList().GetKind() == types.TaskListKindNormal && request.ForwardedFrom == \"\" {\n\t\te.metricsClient.Scope(metrics.MatchingAddTaskScope).Tagged(metrics.DomainTag(domainName),\n\t\t\tmetrics.TaskListTag(taskListName), metrics.TaskListTypeTag(\"activity_task\"),\n\t\t\tmetrics.MatchingHostTag(e.config.HostName)).IncCounter(metrics.CadenceTasklistRequests)\n\t\te.emitInfoOrDebugLog(domainID, \"Emitting tasklist counter on activity task\",\n\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\ttag.Dynamic(\"taskListBaseName\", taskListID.GetRoot()))\n\t}\n\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttaskInfo := &persistence.TaskInfo{\n\t\tDomainID:                      request.GetSourceDomainUUID(),\n\t\tRunID:                         request.Execution.GetRunID(),\n\t\tWorkflowID:                    request.Execution.GetWorkflowID(),\n\t\tScheduleID:                    request.GetScheduleID(),\n\t\tScheduleToStartTimeoutSeconds: request.GetScheduleToStartTimeoutSeconds(),\n\t\tCreatedTime:                   e.timeSource.Now(),\n\t\tPartitionConfig:               request.GetPartitionConfig(),\n\t}\n\n\tsyncMatched, err := tlMgr.AddTask(hCtx.Context, tasklist.AddTaskParams{\n\t\tTaskInfo:                 taskInfo,\n\t\tSource:                   request.GetSource(),\n\t\tForwardedFrom:            request.GetForwardedFrom(),\n\t\tActivityTaskDispatchInfo: request.ActivityTaskDispatchInfo,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif syncMatched {\n\t\thCtx.scope.RecordTimer(metrics.SyncMatchLatencyPerTaskList, time.Since(startT))\n\t}\n\treturn &types.AddActivityTaskResponse{\n\t\tPartitionConfig: tlMgr.TaskListPartitionConfig(),\n\t}, nil\n}\n\n// PollForDecisionTask tries to get the decision task using exponential backoff.\nfunc (e *matchingEngineImpl) PollForDecisionTask(\n\thCtx *handlerContext,\n\treq *types.MatchingPollForDecisionTaskRequest,\n) (*types.MatchingPollForDecisionTaskResponse, error) {\n\tdomainID := req.GetDomainUUID()\n\tpollerID := req.GetPollerID()\n\trequest := req.PollRequest\n\ttaskListName := request.GetTaskList().GetName()\n\ttaskListKind := request.GetTaskList().GetKind()\n\te.logger.Debug(\"Received PollForDecisionTask for taskList\",\n\t\ttag.WorkflowTaskListName(taskListName),\n\t\ttag.WorkflowDomainID(domainID),\n\t)\n\tevent.Log(event.E{\n\t\tTaskListName: taskListName,\n\t\tTaskListKind: &taskListKind,\n\t\tTaskListType: persistence.TaskListTypeDecision,\n\t\tTaskInfo: persistence.TaskInfo{\n\t\t\tDomainID: domainID,\n\t\t},\n\t\tEventName: \"Received PollForDecisionTask\",\n\t\tHost:      e.config.HostName,\n\t\tPayload: map[string]any{\n\t\t\t\"RequestForwardedFrom\": req.GetForwardedFrom(),\n\t\t\t\"IsolationGroup\":       req.GetIsolationGroup(),\n\t\t},\n\t})\npollLoop:\n\tfor {\n\t\tif err := common.IsValidContext(hCtx.Context); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, persistence.TaskListTypeDecision)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"couldn't create new decision tasklist %w\", err)\n\t\t}\n\n\t\t// Add frontend generated pollerID to context so tasklistMgr can support cancellation of\n\t\t// long-poll when frontend calls CancelOutstandingPoll API\n\t\tpollerCtx := tasklist.ContextWithPollerID(hCtx.Context, pollerID)\n\t\tpollerCtx = tasklist.ContextWithIdentity(pollerCtx, request.GetIdentity())\n\t\tpollerCtx = tasklist.ContextWithIsolationGroup(pollerCtx, req.GetIsolationGroup())\n\t\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"couldn't load tasklist manager: %w\", err)\n\t\t}\n\t\tstartT := time.Now() // Record the start time\n\t\ttask, err := tlMgr.GetTask(pollerCtx, nil)\n\t\tif err != nil {\n\t\t\t// TODO: Is empty poll the best reply for errPumpClosed?\n\t\t\tif errors.Is(err, tasklist.ErrNoTasks) || errors.Is(err, errPumpClosed) {\n\t\t\t\te.logger.Debug(\"no decision tasks\",\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\tevent.Log(event.E{\n\t\t\t\t\tTaskListName: taskListName,\n\t\t\t\t\tTaskListKind: &taskListKind,\n\t\t\t\t\tTaskListType: persistence.TaskListTypeDecision,\n\t\t\t\t\tTaskInfo: persistence.TaskInfo{\n\t\t\t\t\t\tDomainID: domainID,\n\t\t\t\t\t},\n\t\t\t\t\tEventName: \"PollForDecisionTask returned no tasks\",\n\t\t\t\t\tHost:      e.config.HostName,\n\t\t\t\t\tPayload: map[string]any{\n\t\t\t\t\t\t\"RequestForwardedFrom\": req.GetForwardedFrom(),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tdomainName, _ := e.domainCache.GetDomainName(domainID)\n\t\t\t\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\t\tPartitionConfig:   tlMgr.TaskListPartitionConfig(),\n\t\t\t\t\tLoadBalancerHints: tlMgr.LoadBalancerHints(),\n\t\t\t\t\tAutoConfigHint: &types.AutoConfigHint{\n\t\t\t\t\t\tEnableAutoConfig:   e.config.EnableClientAutoConfig(domainName, taskListName, persistence.TaskListTypeDecision),\n\t\t\t\t\t\tPollerWaitTimeInMs: time.Since(startT).Milliseconds(),\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t}\n\t\t\treturn nil, fmt.Errorf(\"couldn't get task: %w\", err)\n\t\t}\n\n\t\tif task.IsStarted() {\n\t\t\tevent.Log(event.E{\n\t\t\t\tTaskListName: taskListName,\n\t\t\t\tTaskListKind: &taskListKind,\n\t\t\t\tTaskListType: persistence.TaskListTypeDecision,\n\t\t\t\tTaskInfo:     task.Info(),\n\t\t\t\tEventName:    \"PollForDecisionTask returning already started task\",\n\t\t\t\tHost:         e.config.HostName,\n\t\t\t})\n\t\t\tresp := task.PollForDecisionResponse()\n\t\t\tresp.PartitionConfig = tlMgr.TaskListPartitionConfig()\n\t\t\tresp.LoadBalancerHints = tlMgr.LoadBalancerHints()\n\t\t\tresp.AutoConfigHint = task.AutoConfigHint\n\t\t\treturn resp, nil\n\t\t\t// TODO: Maybe add history expose here?\n\t\t}\n\n\t\te.emitForwardedFromStats(hCtx.scope, task.IsForwarded(), req.GetForwardedFrom())\n\t\tif task.IsQuery() {\n\t\t\ttask.Finish(nil) // this only means query task sync match succeed.\n\n\t\t\t// for query task, we don't need to update history to record decision task started. but we need to know\n\t\t\t// the NextEventID so front end knows what are the history events to load for this decision task.\n\t\t\tmutableStateResp, err := e.historyService.GetMutableState(hCtx.Context, &types.GetMutableStateRequest{\n\t\t\t\tDomainUUID: req.DomainUUID,\n\t\t\t\tExecution:  task.WorkflowExecution(),\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\t// will notify query client that the query task failed\n\t\t\t\te.deliverQueryResult(task.Query.TaskID, &queryResult{internalError: err}) //nolint:errcheck\n\t\t\t\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\t\tPartitionConfig:   tlMgr.TaskListPartitionConfig(),\n\t\t\t\t\tLoadBalancerHints: tlMgr.LoadBalancerHints(),\n\t\t\t\t}, nil\n\t\t\t}\n\n\t\t\tisStickyEnabled := false\n\t\t\tsupportsSticky := e.versionChecker.SupportsStickyQuery(mutableStateResp.GetClientImpl(), mutableStateResp.GetClientFeatureVersion()) == nil\n\t\t\tif len(mutableStateResp.StickyTaskList.GetName()) != 0 && supportsSticky {\n\t\t\t\tisStickyEnabled = true\n\t\t\t}\n\t\t\tresp := &types.RecordDecisionTaskStartedResponse{\n\t\t\t\tPreviousStartedEventID:    mutableStateResp.PreviousStartedEventID,\n\t\t\t\tNextEventID:               mutableStateResp.NextEventID,\n\t\t\t\tWorkflowType:              mutableStateResp.WorkflowType,\n\t\t\t\tStickyExecutionEnabled:    isStickyEnabled,\n\t\t\t\tWorkflowExecutionTaskList: mutableStateResp.TaskList,\n\t\t\t\tBranchToken:               mutableStateResp.CurrentBranchToken,\n\t\t\t\tHistorySize:               mutableStateResp.HistorySize,\n\t\t\t}\n\t\t\treturn e.createPollForDecisionTaskResponse(task, resp, hCtx.scope, tlMgr.TaskListPartitionConfig(), tlMgr.LoadBalancerHints()), nil\n\t\t}\n\n\t\te.emitTaskIsolationMetrics(hCtx.scope, task.Event.PartitionConfig, req.GetIsolationGroup())\n\t\tresp, err := e.recordDecisionTaskStarted(hCtx.Context, request, task)\n\n\t\tif err != nil {\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.DomainNotActiveError:\n\t\t\t\te.emitInfoOrDebugLog(\n\t\t\t\t\ttask.Event.DomainID,\n\t\t\t\t\t\"Decision task dropped because domain is not active\",\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowScheduleID(task.Event.ScheduleID),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t)\n\t\t\t\t// NOTE: There is a risk that if the domain is failed over back immediately. To prevent the decision task from being stuck, we must let\n\t\t\t\t// history service to regenerate the decision transfer task before dropping the task in matching\n\t\t\t\tif err := e.refreshWorkflowTasks(hCtx.Context, task.Event.DomainID, task.WorkflowExecution()); err != nil {\n\t\t\t\t\ttask.Finish(err)\n\t\t\t\t} else {\n\t\t\t\t\ttask.Finish(nil)\n\t\t\t\t}\n\t\t\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError, *types.EventAlreadyStartedError:\n\t\t\t\tdomainName, _ := e.domainCache.GetDomainName(domainID)\n\t\t\t\thCtx.scope.\n\t\t\t\t\tTagged(metrics.DomainTag(domainName)).\n\t\t\t\t\tTagged(metrics.TaskListTag(taskListName)).\n\t\t\t\t\tIncCounter(metrics.PollDecisionTaskAlreadyStartedCounterPerTaskList)\n\n\t\t\t\te.emitInfoOrDebugLog(\n\t\t\t\t\ttask.Event.DomainID,\n\t\t\t\t\t\"Duplicated decision task\",\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowScheduleID(task.Event.ScheduleID),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\ttask.Finish(nil)\n\t\t\tdefault:\n\t\t\t\te.logger.Error(\"unknown error recording task started\",\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowScheduleID(task.Event.ScheduleID),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t\ttask.Finish(err)\n\t\t\t}\n\n\t\t\tcontinue pollLoop\n\t\t}\n\n\t\ttask.Finish(nil)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: taskListName,\n\t\t\tTaskListKind: &taskListKind,\n\t\t\tTaskListType: persistence.TaskListTypeDecision,\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"PollForDecisionTask returning task\",\n\t\t\tHost:         e.config.HostName,\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":      task.IsForwarded(),\n\t\t\t\t\"RequestForwardedFrom\": req.GetForwardedFrom(),\n\t\t\t\t\"Latency\":              time.Since(task.Info().CreatedTime).Milliseconds(),\n\t\t\t\t\"IsolationGroup\":       req.GetIsolationGroup(),\n\t\t\t},\n\t\t})\n\n\t\treturn e.createPollForDecisionTaskResponse(task, resp, hCtx.scope, tlMgr.TaskListPartitionConfig(), tlMgr.LoadBalancerHints()), nil\n\t}\n}\n\n// pollForActivityTaskOperation takes one task from the task manager, update workflow execution history, mark task as\n// completed and return it to user. If a task from task manager is already started, return an empty response, without\n// error. Timeouts handled by the timer queue.\nfunc (e *matchingEngineImpl) PollForActivityTask(\n\thCtx *handlerContext,\n\treq *types.MatchingPollForActivityTaskRequest,\n) (*types.MatchingPollForActivityTaskResponse, error) {\n\tdomainID := req.GetDomainUUID()\n\tpollerID := req.GetPollerID()\n\trequest := req.PollRequest\n\ttaskListName := request.GetTaskList().GetName()\n\te.logger.Debug(\"Received PollForActivityTask\",\n\t\ttag.WorkflowTaskListName(taskListName),\n\t\ttag.WorkflowDomainID(domainID),\n\t)\n\npollLoop:\n\tfor {\n\t\terr := common.IsValidContext(hCtx.Context)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, persistence.TaskListTypeActivity)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tvar maxDispatch *float64\n\t\tif request.TaskListMetadata != nil {\n\t\t\tmaxDispatch = request.TaskListMetadata.MaxTasksPerSecond\n\t\t}\n\t\t// Add frontend generated pollerID to context so tasklistMgr can support cancellation of\n\t\t// long-poll when frontend calls CancelOutstandingPoll API\n\t\tpollerCtx := tasklist.ContextWithPollerID(hCtx.Context, pollerID)\n\t\tpollerCtx = tasklist.ContextWithIdentity(pollerCtx, request.GetIdentity())\n\t\tpollerCtx = tasklist.ContextWithIsolationGroup(pollerCtx, req.GetIsolationGroup())\n\t\ttaskListKind := request.TaskList.GetKind()\n\t\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"couldn't load tasklist manager: %w\", err)\n\t\t}\n\t\tstartT := time.Now() // Record the start time\n\t\ttask, err := tlMgr.GetTask(pollerCtx, maxDispatch)\n\t\tif err != nil {\n\t\t\t// TODO: Is empty poll the best reply for errPumpClosed?\n\t\t\tif errors.Is(err, tasklist.ErrNoTasks) || errors.Is(err, errPumpClosed) {\n\t\t\t\tdomainName, _ := e.domainCache.GetDomainName(domainID)\n\t\t\t\treturn &types.MatchingPollForActivityTaskResponse{\n\t\t\t\t\tPartitionConfig:   tlMgr.TaskListPartitionConfig(),\n\t\t\t\t\tLoadBalancerHints: tlMgr.LoadBalancerHints(),\n\t\t\t\t\tAutoConfigHint: &types.AutoConfigHint{\n\t\t\t\t\t\tEnableAutoConfig:   e.config.EnableClientAutoConfig(domainName, taskListName, persistence.TaskListTypeDecision),\n\t\t\t\t\t\tPollerWaitTimeInMs: time.Since(startT).Milliseconds(),\n\t\t\t\t\t},\n\t\t\t\t}, nil\n\t\t\t}\n\t\t\te.logger.Error(\"Received unexpected err while getting task\",\n\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\treturn nil, err\n\t\t}\n\n\t\tif task.IsStarted() {\n\t\t\t// task received from remote is already started. So, simply forward the response\n\t\t\tresp := task.PollForActivityResponse()\n\t\t\tresp.PartitionConfig = tlMgr.TaskListPartitionConfig()\n\t\t\tresp.LoadBalancerHints = tlMgr.LoadBalancerHints()\n\t\t\tresp.AutoConfigHint = task.AutoConfigHint\n\t\t\treturn resp, nil\n\t\t}\n\t\te.emitForwardedFromStats(hCtx.scope, task.IsForwarded(), req.GetForwardedFrom())\n\t\te.emitTaskIsolationMetrics(hCtx.scope, task.Event.PartitionConfig, req.GetIsolationGroup())\n\t\tif task.ActivityTaskDispatchInfo != nil {\n\t\t\ttask.Finish(nil)\n\t\t\treturn e.createSyncMatchPollForActivityTaskResponse(task, task.ActivityTaskDispatchInfo, tlMgr.TaskListPartitionConfig(), tlMgr.LoadBalancerHints()), nil\n\t\t}\n\n\t\tresp, err := e.recordActivityTaskStarted(hCtx.Context, request, task)\n\t\tif err != nil {\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.DomainNotActiveError:\n\t\t\t\te.emitInfoOrDebugLog(\n\t\t\t\t\ttask.Event.DomainID,\n\t\t\t\t\t\"Decision task dropped because domain is not active\",\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowScheduleID(task.Event.ScheduleID),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t)\n\t\t\t\t// NOTE: There is a risk that if the domain is failed over back immediately. To prevent the decision task from being stuck, we must let\n\t\t\t\t// history service to regenerate the decision transfer task before dropping the task in matching\n\t\t\t\tif err := e.refreshWorkflowTasks(hCtx.Context, task.Event.DomainID, task.WorkflowExecution()); err != nil {\n\t\t\t\t\ttask.Finish(err)\n\t\t\t\t} else {\n\t\t\t\t\ttask.Finish(nil)\n\t\t\t\t}\n\t\t\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError, *types.EventAlreadyStartedError:\n\t\t\t\tdomainName, _ := e.domainCache.GetDomainName(domainID)\n\n\t\t\t\thCtx.scope.\n\t\t\t\t\tTagged(metrics.DomainTag(domainName)).\n\t\t\t\t\tTagged(metrics.TaskListTag(taskListName)).\n\t\t\t\t\tIncCounter(metrics.PollActivityTaskAlreadyStartedCounterPerTaskList)\n\n\t\t\t\te.emitInfoOrDebugLog(\n\t\t\t\t\ttask.Event.DomainID,\n\t\t\t\t\t\"Duplicated activity task\",\n\t\t\t\t\ttag.WorkflowDomainID(domainID),\n\t\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\t\t\ttag.WorkflowScheduleID(task.Event.ScheduleID),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t)\n\t\t\t\ttask.Finish(nil)\n\t\t\tdefault:\n\t\t\t\ttask.Finish(err)\n\t\t\t}\n\n\t\t\tcontinue pollLoop\n\t\t}\n\t\ttask.Finish(nil)\n\t\treturn e.createPollForActivityTaskResponse(task, resp, hCtx.scope, tlMgr.TaskListPartitionConfig(), tlMgr.LoadBalancerHints()), nil\n\t}\n}\n\nfunc (e *matchingEngineImpl) createSyncMatchPollForActivityTaskResponse(\n\ttask *tasklist.InternalTask,\n\tactivityTaskDispatchInfo *types.ActivityTaskDispatchInfo,\n\tpartitionConfig *types.TaskListPartitionConfig,\n\tloadBalancerHints *types.LoadBalancerHints,\n) *types.MatchingPollForActivityTaskResponse {\n\n\tscheduledEvent := activityTaskDispatchInfo.ScheduledEvent\n\tattributes := scheduledEvent.ActivityTaskScheduledEventAttributes\n\tresponse := &types.MatchingPollForActivityTaskResponse{}\n\tresponse.ActivityID = attributes.ActivityID\n\tresponse.ActivityType = attributes.ActivityType\n\tresponse.Header = attributes.Header\n\tresponse.Input = attributes.Input\n\tresponse.WorkflowExecution = task.WorkflowExecution()\n\tresponse.ScheduledTimestampOfThisAttempt = activityTaskDispatchInfo.ScheduledTimestampOfThisAttempt\n\tresponse.ScheduledTimestamp = scheduledEvent.Timestamp\n\tresponse.ScheduleToCloseTimeoutSeconds = attributes.ScheduleToCloseTimeoutSeconds\n\tresponse.StartedTimestamp = activityTaskDispatchInfo.StartedTimestamp\n\tresponse.StartToCloseTimeoutSeconds = attributes.StartToCloseTimeoutSeconds\n\tresponse.HeartbeatTimeoutSeconds = attributes.HeartbeatTimeoutSeconds\n\n\ttoken := &common.TaskToken{\n\t\tDomainID:        task.Event.DomainID,\n\t\tWorkflowID:      task.Event.WorkflowID,\n\t\tWorkflowType:    activityTaskDispatchInfo.WorkflowType.GetName(),\n\t\tRunID:           task.Event.RunID,\n\t\tScheduleID:      task.Event.ScheduleID,\n\t\tScheduleAttempt: common.Int64Default(activityTaskDispatchInfo.Attempt),\n\t\tActivityID:      attributes.GetActivityID(),\n\t\tActivityType:    attributes.GetActivityType().GetName(),\n\t}\n\n\tresponse.TaskToken, _ = e.tokenSerializer.Serialize(token)\n\tresponse.Attempt = int32(token.ScheduleAttempt)\n\tresponse.HeartbeatDetails = activityTaskDispatchInfo.HeartbeatDetails\n\tresponse.WorkflowType = activityTaskDispatchInfo.WorkflowType\n\tresponse.WorkflowDomain = activityTaskDispatchInfo.WorkflowDomain\n\tresponse.PartitionConfig = partitionConfig\n\tresponse.LoadBalancerHints = loadBalancerHints\n\tresponse.AutoConfigHint = task.AutoConfigHint\n\treturn response\n}\n\n// QueryWorkflow creates a DecisionTask with query data, send it through sync match channel, wait for that DecisionTask\n// to be processed by worker, and then return the query result.\nfunc (e *matchingEngineImpl) QueryWorkflow(\n\thCtx *handlerContext,\n\tqueryRequest *types.MatchingQueryWorkflowRequest,\n) (*types.MatchingQueryWorkflowResponse, error) {\n\tdomainID := queryRequest.GetDomainUUID()\n\ttaskListName := queryRequest.GetTaskList().GetName()\n\ttaskListKind := queryRequest.GetTaskList().GetKind()\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, persistence.TaskListTypeDecision)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif taskListKind == types.TaskListKindSticky {\n\t\t// check if the sticky worker is still available, if not, fail this request early\n\t\tif !tlMgr.HasPollerAfter(e.timeSource.Now().Add(-_stickyPollerUnavailableWindow)) {\n\t\t\treturn nil, _stickyPollerUnavailableError\n\t\t}\n\t}\n\n\ttaskID := uuid.New()\n\tqueryResultCh := make(chan *queryResult, 1)\n\te.lockableQueryTaskMap.put(taskID, queryResultCh)\n\tdefer e.lockableQueryTaskMap.delete(taskID)\n\n\tresp, err := tlMgr.DispatchQueryTask(hCtx.Context, taskID, queryRequest)\n\t// if get response or error it means that query task was handled by forwarding to another matching host\n\t// this remote host's result can be returned directly\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif resp != nil {\n\t\tresp.PartitionConfig = tlMgr.TaskListPartitionConfig()\n\t\treturn resp, nil\n\t}\n\n\t// if get here it means that dispatch of query task has occurred locally\n\t// must wait on result channel to get query result\n\tresp, err = e.waitForQueryResult(hCtx, queryRequest.GetQueryRequest().GetQueryConsistencyLevel() == types.QueryConsistencyLevelStrong, queryResultCh)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp.PartitionConfig = tlMgr.TaskListPartitionConfig()\n\treturn resp, nil\n}\n\nfunc (e *matchingEngineImpl) waitForQueryResult(hCtx *handlerContext, isStrongConsistencyQuery bool, queryResultCh <-chan *queryResult) (*types.MatchingQueryWorkflowResponse, error) {\n\tselect {\n\tcase result := <-queryResultCh:\n\t\tif result.internalError != nil {\n\t\t\treturn nil, result.internalError\n\t\t}\n\t\tworkerResponse := result.workerResponse\n\t\t// if query was intended as consistent query check to see if worker supports consistent query\n\t\tif isStrongConsistencyQuery {\n\t\t\tif err := e.versionChecker.SupportsConsistentQuery(\n\t\t\t\tworkerResponse.GetCompletedRequest().GetWorkerVersionInfo().GetImpl(),\n\t\t\t\tworkerResponse.GetCompletedRequest().GetWorkerVersionInfo().GetFeatureVersion()); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\tswitch workerResponse.GetCompletedRequest().GetCompletedType() {\n\t\tcase types.QueryTaskCompletedTypeCompleted:\n\t\t\treturn &types.MatchingQueryWorkflowResponse{QueryResult: workerResponse.GetCompletedRequest().GetQueryResult()}, nil\n\t\tcase types.QueryTaskCompletedTypeFailed:\n\t\t\treturn nil, &types.QueryFailedError{Message: workerResponse.GetCompletedRequest().GetErrorMessage()}\n\t\tdefault:\n\t\t\treturn nil, &types.InternalServiceError{Message: \"unknown query completed type\"}\n\t\t}\n\tcase <-hCtx.Done():\n\t\treturn nil, hCtx.Err()\n\t}\n}\n\nfunc (e *matchingEngineImpl) RespondQueryTaskCompleted(hCtx *handlerContext, request *types.MatchingRespondQueryTaskCompletedRequest) error {\n\tif err := e.deliverQueryResult(request.GetTaskID(), &queryResult{workerResponse: request}); err != nil {\n\t\thCtx.scope.IncCounter(metrics.RespondQueryTaskFailedPerTaskListCounter)\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) deliverQueryResult(taskID string, queryResult *queryResult) error {\n\tqueryResultCh, ok := e.lockableQueryTaskMap.get(taskID)\n\tif !ok {\n\t\treturn &types.InternalServiceError{Message: \"query task not found, or already expired\"}\n\t}\n\tqueryResultCh <- queryResult\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) CancelOutstandingPoll(\n\thCtx *handlerContext,\n\trequest *types.CancelOutstandingPollRequest,\n) error {\n\tdomainID := request.GetDomainUUID()\n\ttaskListType := int(request.GetTaskListType())\n\ttaskListName := request.GetTaskList().GetName()\n\ttaskListKind := request.GetTaskList().GetKind()\n\tpollerID := request.GetPollerID()\n\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttlMgr.CancelPoller(pollerID)\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) DescribeTaskList(\n\thCtx *handlerContext,\n\trequest *types.MatchingDescribeTaskListRequest,\n) (*types.DescribeTaskListResponse, error) {\n\tdomainID := request.GetDomainUUID()\n\ttaskListType := persistence.TaskListTypeDecision\n\tif request.DescRequest.GetTaskListType() == types.TaskListTypeActivity {\n\t\ttaskListType = persistence.TaskListTypeActivity\n\t}\n\ttaskListName := request.GetDescRequest().GetTaskList().GetName()\n\ttaskListKind := request.GetDescRequest().GetTaskList().GetKind()\n\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn tlMgr.DescribeTaskList(request.DescRequest.GetIncludeTaskListStatus()), nil\n}\n\nfunc (e *matchingEngineImpl) ListTaskListPartitions(\n\thCtx *handlerContext,\n\trequest *types.MatchingListTaskListPartitionsRequest,\n) (*types.ListTaskListPartitionsResponse, error) {\n\tactivityTaskListInfo, err := e.listTaskListPartitions(request, persistence.TaskListTypeActivity)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdecisionTaskListInfo, err := e.listTaskListPartitions(request, persistence.TaskListTypeDecision)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresp := &types.ListTaskListPartitionsResponse{\n\t\tActivityTaskListPartitions: activityTaskListInfo,\n\t\tDecisionTaskListPartitions: decisionTaskListInfo,\n\t}\n\n\treturn resp, nil\n}\n\nfunc (e *matchingEngineImpl) listTaskListPartitions(\n\trequest *types.MatchingListTaskListPartitionsRequest,\n\ttaskListType int,\n) ([]*types.TaskListPartitionMetadata, error) {\n\tpartitions, err := e.getAllPartitions(\n\t\trequest,\n\t\ttaskListType,\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar partitionHostInfo []*types.TaskListPartitionMetadata\n\tfor _, partition := range partitions {\n\t\thost, _ := e.getHostInfo(partition)\n\t\tpartitionHostInfo = append(partitionHostInfo,\n\t\t\t&types.TaskListPartitionMetadata{\n\t\t\t\tKey:           partition,\n\t\t\t\tOwnerHostName: host,\n\t\t\t})\n\t}\n\treturn partitionHostInfo, nil\n}\n\nfunc (e *matchingEngineImpl) getTaskListsByDomainAndKind(domainID string, taskListKind *types.TaskListKind) *types.GetTaskListsByDomainResponse {\n\tdecisionTaskListMap := make(map[string]*types.DescribeTaskListResponse)\n\tactivityTaskListMap := make(map[string]*types.DescribeTaskListResponse)\n\n\tfor _, tlm := range e.taskListRegistry.ManagersByDomainID(domainID) {\n\t\tif taskListKind == nil || tlm.GetTaskListKind() == *taskListKind {\n\t\t\ttl := tlm.TaskListID()\n\t\t\tif types.TaskListType(tl.GetType()) == types.TaskListTypeDecision {\n\t\t\t\tdecisionTaskListMap[tl.GetRoot()] = tlm.DescribeTaskList(false)\n\t\t\t} else {\n\t\t\t\tactivityTaskListMap[tl.GetRoot()] = tlm.DescribeTaskList(false)\n\t\t\t}\n\t\t}\n\t}\n\treturn &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: decisionTaskListMap,\n\t\tActivityTaskListMap: activityTaskListMap,\n\t}\n}\n\nfunc (e *matchingEngineImpl) GetTaskListsByDomain(\n\thCtx *handlerContext,\n\trequest *types.GetTaskListsByDomainRequest,\n) (*types.GetTaskListsByDomainResponse, error) {\n\tdomainID, err := e.domainCache.GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttlKind := types.TaskListKindNormal.Ptr()\n\tif e.config.EnableReturnAllTaskListKinds() {\n\t\ttlKind = nil\n\t}\n\n\treturn e.getTaskListsByDomainAndKind(domainID, tlKind), nil\n}\n\nfunc (e *matchingEngineImpl) UpdateTaskListPartitionConfig(\n\thCtx *handlerContext,\n\trequest *types.MatchingUpdateTaskListPartitionConfigRequest,\n) (*types.MatchingUpdateTaskListPartitionConfigResponse, error) {\n\tdomainID := request.DomainUUID\n\ttaskListName := request.TaskList.GetName()\n\ttaskListKind := request.TaskList.GetKind()\n\ttaskListType := persistence.TaskListTypeDecision\n\tif request.GetTaskListType() == types.TaskListTypeActivity {\n\t\ttaskListType = persistence.TaskListTypeActivity\n\t}\n\tdomainName, err := e.domainCache.GetDomainName(domainID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif e.config.EnableAdaptiveScaler(domainName, taskListName, taskListType) {\n\t\treturn nil, &types.BadRequestError{Message: \"Manual update is not allowed because adaptive scaler is enabled.\"}\n\t}\n\tif taskListKind != types.TaskListKindNormal {\n\t\treturn nil, &types.BadRequestError{Message: \"Only normal tasklist's partition config can be updated.\"}\n\t}\n\tif request.PartitionConfig == nil {\n\t\treturn nil, &types.BadRequestError{Message: \"Task list partition config is not set in the request.\"}\n\t}\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif !taskListID.IsRoot() {\n\t\treturn nil, &types.BadRequestError{Message: \"Only root partition's partition config can be updated.\"}\n\t}\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\terr = tlMgr.UpdateTaskListPartitionConfig(hCtx.Context, request.PartitionConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.MatchingUpdateTaskListPartitionConfigResponse{}, nil\n}\n\nfunc (e *matchingEngineImpl) RefreshTaskListPartitionConfig(\n\thCtx *handlerContext,\n\trequest *types.MatchingRefreshTaskListPartitionConfigRequest,\n) (*types.MatchingRefreshTaskListPartitionConfigResponse, error) {\n\tdomainID := request.DomainUUID\n\ttaskListName := request.TaskList.GetName()\n\ttaskListKind := request.TaskList.GetKind()\n\ttaskListType := persistence.TaskListTypeDecision\n\tif request.GetTaskListType() == types.TaskListTypeActivity {\n\t\ttaskListType = persistence.TaskListTypeActivity\n\t}\n\tif taskListKind != types.TaskListKindNormal {\n\t\treturn nil, &types.BadRequestError{Message: \"Only normal tasklist's partition config can be updated.\"}\n\t}\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif taskListID.IsRoot() && request.PartitionConfig != nil {\n\t\treturn nil, &types.BadRequestError{Message: \"PartitionConfig must be nil for root partition.\"}\n\t}\n\ttlMgr, err := e.getOrCreateTaskListManager(hCtx.Context, taskListID, taskListKind)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\terr = tlMgr.RefreshTaskListPartitionConfig(hCtx.Context, request.PartitionConfig)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.MatchingRefreshTaskListPartitionConfigResponse{}, nil\n}\n\nfunc (e *matchingEngineImpl) getHostInfo(partitionKey string) (string, error) {\n\thost, err := e.membershipResolver.Lookup(service.Matching, partitionKey)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn host.GetAddress(), nil\n}\n\nfunc (e *matchingEngineImpl) getAllPartitions(\n\trequest *types.MatchingListTaskListPartitionsRequest,\n\ttaskListType int,\n) ([]string, error) {\n\tdomainID, err := e.domainCache.GetDomainID(request.GetDomain())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttaskList := request.GetTaskList()\n\ttaskListID, err := tasklist.NewIdentifier(domainID, taskList.GetName(), taskListType)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\trootPartition := taskListID.GetRoot()\n\tpartitionKeys := []string{rootPartition}\n\tn := e.config.NumTasklistWritePartitions(request.GetDomain(), rootPartition, taskListType)\n\tfor i := 1; i < n; i++ {\n\t\tpartitionKeys = append(partitionKeys, fmt.Sprintf(\"%v%v/%v\", constants.ReservedTaskListPrefix, rootPartition, i))\n\t}\n\treturn partitionKeys, nil\n}\n\nfunc (e *matchingEngineImpl) unloadTaskList(tlMgr tasklist.Manager) {\n\tunregistered := e.taskListRegistry.Unregister(tlMgr)\n\tif unregistered {\n\t\ttlMgr.Stop()\n\t}\n}\n\n// Populate the decision task response based on context and scheduled/started events.\nfunc (e *matchingEngineImpl) createPollForDecisionTaskResponse(\n\ttask *tasklist.InternalTask,\n\thistoryResponse *types.RecordDecisionTaskStartedResponse,\n\tscope metrics.Scope,\n\tpartitionConfig *types.TaskListPartitionConfig,\n\tloadBalancerHints *types.LoadBalancerHints,\n) *types.MatchingPollForDecisionTaskResponse {\n\n\tvar token []byte\n\tif task.IsQuery() {\n\t\t// for a query task\n\t\tqueryRequest := task.Query.Request\n\t\texecution := task.WorkflowExecution()\n\t\ttaskToken := &common.QueryTaskToken{\n\t\t\tDomainID:   queryRequest.DomainUUID,\n\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\tRunID:      execution.RunID,\n\t\t\tTaskList:   queryRequest.TaskList.Name,\n\t\t\tTaskID:     task.Query.TaskID,\n\t\t}\n\t\ttoken, _ = e.tokenSerializer.SerializeQueryTaskToken(taskToken)\n\t} else {\n\t\ttaskToken := &common.TaskToken{\n\t\t\tDomainID:        task.Event.DomainID,\n\t\t\tWorkflowID:      task.Event.WorkflowID,\n\t\t\tRunID:           task.Event.RunID,\n\t\t\tScheduleID:      historyResponse.GetScheduledEventID(),\n\t\t\tScheduleAttempt: historyResponse.GetAttempt(),\n\t\t}\n\t\ttoken, _ = e.tokenSerializer.Serialize(taskToken)\n\t\tif task.ResponseC == nil {\n\t\t\tscope.RecordTimer(metrics.AsyncMatchLatencyPerTaskList, time.Since(task.Event.CreatedTime))\n\t\t}\n\t}\n\n\tresponse := common.CreateMatchingPollForDecisionTaskResponse(historyResponse, task.WorkflowExecution(), token)\n\tif task.Query != nil {\n\t\tresponse.Query = task.Query.Request.QueryRequest.Query\n\t}\n\tresponse.BacklogCountHint = task.BacklogCountHint\n\tresponse.PartitionConfig = partitionConfig\n\tresponse.LoadBalancerHints = loadBalancerHints\n\tresponse.AutoConfigHint = task.AutoConfigHint\n\treturn response\n}\n\n// Populate the activity task response based on context and scheduled/started events.\nfunc (e *matchingEngineImpl) createPollForActivityTaskResponse(\n\ttask *tasklist.InternalTask,\n\thistoryResponse *types.RecordActivityTaskStartedResponse,\n\tscope metrics.Scope,\n\tpartitionConfig *types.TaskListPartitionConfig,\n\tloadBalancerHints *types.LoadBalancerHints,\n) *types.MatchingPollForActivityTaskResponse {\n\n\tscheduledEvent := historyResponse.ScheduledEvent\n\tif scheduledEvent.ActivityTaskScheduledEventAttributes == nil {\n\t\tpanic(\"GetActivityTaskScheduledEventAttributes is not set\")\n\t}\n\tattributes := scheduledEvent.ActivityTaskScheduledEventAttributes\n\tif attributes.ActivityID == \"\" {\n\t\tpanic(\"ActivityTaskScheduledEventAttributes.ActivityID is not set\")\n\t}\n\tif task.ResponseC == nil {\n\t\tscope.RecordTimer(metrics.AsyncMatchLatencyPerTaskList, time.Since(task.Event.CreatedTime))\n\t}\n\n\tresponse := &types.MatchingPollForActivityTaskResponse{}\n\tresponse.ActivityID = attributes.ActivityID\n\tresponse.ActivityType = attributes.ActivityType\n\tresponse.Header = attributes.Header\n\tresponse.Input = attributes.Input\n\tresponse.WorkflowExecution = task.WorkflowExecution()\n\tresponse.ScheduledTimestampOfThisAttempt = historyResponse.ScheduledTimestampOfThisAttempt\n\tresponse.ScheduledTimestamp = scheduledEvent.Timestamp\n\tresponse.ScheduleToCloseTimeoutSeconds = attributes.ScheduleToCloseTimeoutSeconds\n\tresponse.StartedTimestamp = historyResponse.StartedTimestamp\n\tresponse.StartToCloseTimeoutSeconds = attributes.StartToCloseTimeoutSeconds\n\tresponse.HeartbeatTimeoutSeconds = attributes.HeartbeatTimeoutSeconds\n\n\ttoken := &common.TaskToken{\n\t\tDomainID:        task.Event.DomainID,\n\t\tWorkflowID:      task.Event.WorkflowID,\n\t\tWorkflowType:    historyResponse.WorkflowType.GetName(),\n\t\tRunID:           task.Event.RunID,\n\t\tScheduleID:      task.Event.ScheduleID,\n\t\tScheduleAttempt: historyResponse.GetAttempt(),\n\t\tActivityID:      attributes.GetActivityID(),\n\t\tActivityType:    attributes.GetActivityType().GetName(),\n\t}\n\n\tresponse.TaskToken, _ = e.tokenSerializer.Serialize(token)\n\tresponse.Attempt = int32(token.ScheduleAttempt)\n\tresponse.HeartbeatDetails = historyResponse.HeartbeatDetails\n\tresponse.WorkflowType = historyResponse.WorkflowType\n\tresponse.WorkflowDomain = historyResponse.WorkflowDomain\n\tresponse.PartitionConfig = partitionConfig\n\tresponse.LoadBalancerHints = loadBalancerHints\n\tresponse.AutoConfigHint = task.AutoConfigHint\n\treturn response\n}\n\nfunc (e *matchingEngineImpl) recordDecisionTaskStarted(\n\tctx context.Context,\n\tpollReq *types.PollForDecisionTaskRequest,\n\ttask *tasklist.InternalTask,\n) (*types.RecordDecisionTaskStartedResponse, error) {\n\trequest := &types.RecordDecisionTaskStartedRequest{\n\t\tDomainUUID:        task.Event.DomainID,\n\t\tWorkflowExecution: task.WorkflowExecution(),\n\t\tScheduleID:        task.Event.ScheduleID,\n\t\tTaskID:            task.Event.TaskID,\n\t\tRequestID:         uuid.New(),\n\t\tPollRequest:       pollReq,\n\t}\n\tvar resp *types.RecordDecisionTaskStartedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = e.historyService.RecordDecisionTaskStarted(ctx, request)\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(recordTaskStartedRetryPolicy),\n\t\tbackoff.WithRetryableError(isMatchingRetryableError),\n\t\tbackoff.WithOperationTimeout(_recordTaskStartedTimeout),\n\t\tbackoff.WithContextExpiration(),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (e *matchingEngineImpl) recordActivityTaskStarted(\n\tctx context.Context,\n\tpollReq *types.PollForActivityTaskRequest,\n\ttask *tasklist.InternalTask,\n) (*types.RecordActivityTaskStartedResponse, error) {\n\trequest := &types.RecordActivityTaskStartedRequest{\n\t\tDomainUUID:        task.Event.DomainID,\n\t\tWorkflowExecution: task.WorkflowExecution(),\n\t\tScheduleID:        task.Event.ScheduleID,\n\t\tTaskID:            task.Event.TaskID,\n\t\tRequestID:         uuid.New(),\n\t\tPollRequest:       pollReq,\n\t}\n\tvar resp *types.RecordActivityTaskStartedResponse\n\top := func(ctx context.Context) error {\n\t\tvar err error\n\t\tresp, err = e.historyService.RecordActivityTaskStarted(ctx, request)\n\t\treturn err\n\t}\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(recordTaskStartedRetryPolicy),\n\t\tbackoff.WithRetryableError(isMatchingRetryableError),\n\t\tbackoff.WithOperationTimeout(_recordTaskStartedTimeout),\n\t\tbackoff.WithContextExpiration(),\n\t)\n\terr := throttleRetry.Do(ctx, op)\n\treturn resp, err\n}\n\nfunc (e *matchingEngineImpl) refreshWorkflowTasks(\n\tctx context.Context,\n\tdomainID string,\n\tworkflowExecution *types.WorkflowExecution,\n) error {\n\trequest := &types.HistoryRefreshWorkflowTasksRequest{\n\t\tDomainUIID: domainID,\n\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\tExecution: workflowExecution,\n\t\t},\n\t}\n\terr := e.historyService.RefreshWorkflowTasks(ctx, request)\n\tif err != nil {\n\t\tvar e *types.EntityNotExistsError\n\t\tif errors.As(err, &e) {\n\t\t\treturn nil\n\t\t}\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) emitForwardedFromStats(\n\tscope metrics.Scope,\n\tisTaskForwarded bool,\n\tpollForwardedFrom string,\n) {\n\tisPollForwarded := len(pollForwardedFrom) > 0\n\tswitch {\n\tcase isTaskForwarded && isPollForwarded:\n\t\tscope.IncCounter(metrics.RemoteToRemoteMatchPerTaskListCounter)\n\tcase isTaskForwarded:\n\t\tscope.IncCounter(metrics.RemoteToLocalMatchPerTaskListCounter)\n\tcase isPollForwarded:\n\t\tscope.IncCounter(metrics.LocalToRemoteMatchPerTaskListCounter)\n\tdefault:\n\t\tscope.IncCounter(metrics.LocalToLocalMatchPerTaskListCounter)\n\t}\n}\n\nfunc (e *matchingEngineImpl) emitTaskIsolationMetrics(\n\tscope metrics.Scope,\n\tpartitionConfig map[string]string,\n\tpollerIsolationGroup string,\n) {\n\tif currentGroup, ok := partitionConfig[isolationgroup.GroupKey]; ok {\n\t\toriginalGroup, ok := partitionConfig[isolationgroup.OriginalGroupKey]\n\t\tif !ok {\n\t\t\toriginalGroup = currentGroup\n\t\t}\n\t\tscope.Tagged(metrics.IsolationGroupTag(originalGroup), metrics.PollerIsolationGroupTag(pollerIsolationGroup)).IncCounter(metrics.IsolationTaskMatchPerTaskListCounter)\n\t\tif originalGroup == pollerIsolationGroup {\n\t\t\tscope.Tagged(metrics.IsolationGroupTag(originalGroup)).IncCounter(metrics.IsolationSuccessPerTaskListCounter)\n\t\t}\n\t}\n}\n\nfunc (e *matchingEngineImpl) emitInfoOrDebugLog(\n\tdomainID string,\n\tmsg string,\n\ttags ...tag.Tag,\n) {\n\tif e.config.EnableDebugMode && e.config.EnableTaskInfoLogByDomainID(domainID) {\n\t\te.logger.Info(msg, tags...)\n\t} else {\n\t\te.logger.Debug(msg, tags...)\n\t}\n}\n\nfunc (e *matchingEngineImpl) errIfShardOwnershipLost(ctx context.Context, taskList *tasklist.Identifier) error {\n\tif !e.config.EnableTasklistOwnershipGuard() {\n\t\treturn nil\n\t}\n\n\tself, err := e.membershipResolver.WhoAmI()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to lookup self im membership: %w\", err)\n\t}\n\n\tnewNotOwnedByHostError := func(newOwner string) error {\n\t\treturn cadence_errors.NewTaskListNotOwnedByHostError(\n\t\t\tnewOwner,\n\t\t\tself.Identity(),\n\t\t\ttaskList.GetName(),\n\t\t)\n\t}\n\n\tif e.isShuttingDown() {\n\t\te.logger.Warn(\"request to get tasklist is being rejected because engine is shutting down\",\n\t\t\ttag.WorkflowDomainID(taskList.GetDomainID()),\n\t\t\ttag.WorkflowTaskListType(taskList.GetType()),\n\t\t\ttag.WorkflowTaskListName(taskList.GetName()),\n\t\t)\n\n\t\treturn newNotOwnedByHostError(\"not known\")\n\t}\n\n\t// Task lists excluded from the ShardDistributor bypass the executor entirely and rely on\n\t// the local hash-ring for ownership, so skip the SD-based ownership check for them.\n\tif !e.isExcludedFromShardDistributor(taskList.GetName()) {\n\t\t// We have a shard-processor shared by all the task lists with the same name.\n\t\t// For now there is no 1:1 mapping between shards and tasklists. (#tasklists >= #shards)\n\t\tsp, err := e.executor.GetShardProcess(ctx, taskList.GetName())\n\t\tif err != nil {\n\t\t\tif errors.Is(err, executorclient.ErrShardProcessNotFound) {\n\t\t\t\t// The shard is not assigned to this host – treat it as an ownership loss,\n\t\t\t\t// not an internal error.\n\t\t\t\treturn newNotOwnedByHostError(\"not known\")\n\t\t\t}\n\t\t\treturn fmt.Errorf(\"failed to lookup ownership in SD: %w\", err)\n\t\t}\n\t\tif sp == nil {\n\t\t\treturn newNotOwnedByHostError(\"not known\")\n\t\t}\n\t\treturn nil\n\t}\n\n\t// Defensive check to make sure we actually own the task list\n\t//   If we try to create a task list manager for a task list that is not owned by us, return an error\n\t//   The new task list manager will steal the task list from the current owner, which should only happen if\n\t//   the task list is owned by the current host.\n\ttaskListOwner, err := e.membershipResolver.Lookup(service.Matching, taskList.GetName())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to lookup task list owner: %w\", err)\n\t}\n\tif taskListOwner.Identity() != self.Identity() {\n\t\te.logger.Warn(\"Request to get tasklist is being rejected because engine does not own this shard\",\n\t\t\ttag.WorkflowDomainID(taskList.GetDomainID()),\n\t\t\ttag.WorkflowTaskListType(taskList.GetType()),\n\t\t\ttag.WorkflowTaskListName(taskList.GetName()),\n\t\t)\n\t\treturn newNotOwnedByHostError(taskListOwner.Identity())\n\t}\n\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) isShuttingDown() bool {\n\tselect {\n\tcase <-e.shutdown:\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\n// isExcludedFromShardDistributor returns true if the task list should bypass the\n// ShardDistributor and executor, and instead rely on local hash-ring assignment.\n// This applies to short-lived task lists (e.g. sticky or bits task lists whose names\n// contain a UUID) when the corresponding feature flag is enabled.\nfunc (e *matchingEngineImpl) isExcludedFromShardDistributor(taskListName string) bool {\n\texcludeTaskList := membership.TaskListExcludedFromShardDistributor(taskListName, uint64(e.config.PercentageOnboardedToShardManager()), e.config.ExcludeShortLivedTaskListsFromShardManager())\n\treturn excludeTaskList\n}\n\nfunc (e *matchingEngineImpl) domainChangeCallback(nextDomains []*cache.DomainCacheEntry) {\n\tnewFailoverNotificationVersion := e.failoverNotificationVersion\n\n\tfor _, domain := range nextDomains {\n\t\tif domain.GetFailoverNotificationVersion() > newFailoverNotificationVersion {\n\t\t\tnewFailoverNotificationVersion = domain.GetFailoverNotificationVersion()\n\t\t}\n\n\t\tif !isDomainEligibleToDisconnectPollers(domain, e.failoverNotificationVersion) {\n\t\t\tcontinue\n\t\t}\n\n\t\ttaskListNormal := types.TaskListKindNormal\n\n\t\tresp := e.getTaskListsByDomainAndKind(domain.GetInfo().ID, &taskListNormal)\n\n\t\tfor taskListName := range resp.DecisionTaskListMap {\n\t\t\te.disconnectTaskListPollersAfterDomainFailover(taskListName, domain, persistence.TaskListTypeDecision, taskListNormal)\n\t\t}\n\n\t\tfor taskListName := range resp.ActivityTaskListMap {\n\t\t\te.disconnectTaskListPollersAfterDomainFailover(taskListName, domain, persistence.TaskListTypeActivity, taskListNormal)\n\t\t}\n\n\t\ttaskListSticky := types.TaskListKindSticky\n\n\t\tresp = e.getTaskListsByDomainAndKind(domain.GetInfo().ID, &taskListSticky)\n\n\t\tfor taskListName := range resp.DecisionTaskListMap {\n\t\t\te.disconnectTaskListPollersAfterDomainFailover(taskListName, domain, persistence.TaskListTypeDecision, taskListSticky)\n\t\t}\n\t}\n\te.failoverNotificationVersion = newFailoverNotificationVersion\n}\n\nfunc (e *matchingEngineImpl) registerDomainFailoverCallback() {\n\tcatchUpFn := func(domainCache cache.DomainCache, _ cache.PrepareCallbackFn, _ cache.CallbackFn) {\n\t\tfor _, domain := range domainCache.GetAllDomain() {\n\t\t\tif domain.GetFailoverNotificationVersion() > e.failoverNotificationVersion {\n\t\t\t\te.failoverNotificationVersion = domain.GetFailoverNotificationVersion()\n\t\t\t}\n\t\t}\n\t}\n\n\te.domainCache.RegisterDomainChangeCallback(\n\t\tservice.Matching,\n\t\tcatchUpFn,\n\t\tfunc() {},\n\t\te.domainChangeCallback)\n}\n\nfunc (e *matchingEngineImpl) unregisterDomainFailoverCallback() {\n\te.domainCache.UnregisterDomainChangeCallback(service.Matching)\n}\n\nfunc (e *matchingEngineImpl) disconnectTaskListPollersAfterDomainFailover(taskListName string, domain *cache.DomainCacheEntry, taskType int, taskListKind types.TaskListKind) {\n\ttaskList, err := tasklist.NewIdentifier(domain.GetInfo().ID, taskListName, taskType)\n\tif err != nil {\n\t\treturn\n\t}\n\ttlMgr, err := e.getOrCreateTaskListManager(context.Background(), taskList, taskListKind)\n\tif err != nil {\n\t\te.logger.Error(\"Couldn't load tasklist manager\", tag.Error(err))\n\t\treturn\n\t}\n\n\terr = tlMgr.ReleaseBlockedPollers()\n\tif err != nil {\n\t\te.logger.Error(\"Couldn't disconnect tasklist pollers after domain failover\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowDomainID(domain.GetInfo().ID),\n\t\t\ttag.WorkflowDomainName(domain.GetInfo().Name),\n\t\t\ttag.WorkflowTaskListName(taskListName),\n\t\t\ttag.WorkflowTaskListType(taskType),\n\t\t)\n\t\treturn\n\t}\n}\n\nfunc (m *lockableQueryTaskMap) put(key string, value chan *queryResult) {\n\tm.Lock()\n\tdefer m.Unlock()\n\tm.queryTaskMap[key] = value\n}\n\nfunc (m *lockableQueryTaskMap) get(key string) (chan *queryResult, bool) {\n\tm.RLock()\n\tdefer m.RUnlock()\n\tresult, ok := m.queryTaskMap[key]\n\treturn result, ok\n}\n\nfunc (m *lockableQueryTaskMap) delete(key string) {\n\tm.Lock()\n\tdefer m.Unlock()\n\tdelete(m.queryTaskMap, key)\n}\n\nfunc isMatchingRetryableError(err error) bool {\n\tswitch err.(type) {\n\tcase *types.EntityNotExistsError, *types.WorkflowExecutionAlreadyCompletedError, *types.EventAlreadyStartedError, *types.DomainNotActiveError:\n\t\treturn false\n\t}\n\treturn true\n}\n\nfunc isDomainEligibleToDisconnectPollers(domain *cache.DomainCacheEntry, currentVersion int64) bool {\n\treturn domain.IsGlobalDomain() &&\n\t\tdomain.GetReplicationConfig() != nil &&\n\t\t!domain.GetReplicationConfig().IsActiveActive() &&\n\t\tdomain.GetFailoverNotificationVersion() > currentVersion\n}\n"
  },
  {
    "path": "service/matching/handler/engine_integration_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"strings\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/activecluster\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/isolationgroup/defaultisolationgroupstate\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n\tsdconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\ntype (\n\tmatchingEngineSuite struct {\n\t\tsuite.Suite\n\t\tcontroller                     *gomock.Controller\n\t\tmockHistoryClient              *history.MockClient\n\t\tmockMatchingClient             *matching.MockClient\n\t\tmockDomainCache                *cache.MockDomainCache\n\t\tmockMembershipResolver         *membership.MockResolver\n\t\tmockIsolationStore             *dynamicconfig.MockClient\n\t\tmockShardExecutorClient        *executorclient.MockClient\n\t\tmockExecutor                   *executorclient.MockExecutor[tasklist.ShardProcessor]\n\t\tShardDistributorMatchingConfig clientcommon.Config\n\n\t\tmatchingEngine       *matchingEngineImpl\n\t\ttaskManager          *tasklist.TestTaskManager\n\t\tisolationState       isolationgroup.State\n\t\tmockExecutionManager *mocks.ExecutionManager\n\t\tmockTimeSource       clock.MockedTimeSource\n\t\tlogger               log.Logger\n\t\thandlerContext       *handlerContext\n\t\tmockActiveClusterMgr activecluster.MockManager\n\t\tsync.Mutex\n\t}\n)\n\nconst (\n\t_minBurst              = 10000\n\tmatchingTestDomainName = \"matching-test\"\n\tmatchingTestTaskList   = \"matching-test-tasklist\"\n\n\treturnEmptyTaskTimeBudget       = time.Second\n\t_defaultTaskDispatchRPS         = 100000.0\n\tdefaultTaskBufferIsolationGroup = \"\"\n)\n\nvar errRemoteSyncMatchFailed = &types.RemoteSyncMatchedError{Message: \"remote sync match failed\"}\n\nfunc TestMatchingEngineSuite(t *testing.T) {\n\ts := new(matchingEngineSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *matchingEngineSuite) SetupSuite() {\n\t// http.Handle(\"/test/tasks\", http.HandlerFunc(s.TasksHandler))\n}\n\n// Renders content of taskManager and matchingEngine when called at http://localhost:6060/test/tasks\n// Uncomment HTTP server initialization in SetupSuite method to enable.\nfunc (s *matchingEngineSuite) TasksHandler(w http.ResponseWriter, r *http.Request) {\n\ts.Lock()\n\tdefer s.Unlock()\n\tw.Header().Set(\"Content-Type\", \"text/plain; charset=utf-8\")\n\tfmt.Fprintf(w, \"%v\\n\", s.taskManager)\n\tfmt.Fprintf(w, \"%v\\n\", s.matchingEngine)\n}\n\nfunc (s *matchingEngineSuite) TearDownSuite() {\n}\n\nfunc (s *matchingEngineSuite) SetupTest() {\n\ts.Lock()\n\tdefer s.Unlock()\n\ts.logger = testlogger.New(s.Suite.T()).WithTags(tag.Dynamic(\"test-name\", s.T().Name()))\n\ttlKindNormal := types.TaskListKindNormal\n\ts.mockExecutionManager = &mocks.ExecutionManager{}\n\ts.controller = gomock.NewController(s.T())\n\ts.mockHistoryClient = history.NewMockClient(s.controller)\n\ts.mockMatchingClient = matching.NewMockClient(s.controller)\n\ts.mockTimeSource = clock.NewMockedTimeSourceAt(time.Now())\n\ts.taskManager = tasklist.NewTestTaskManager(s.T(), s.logger, s.mockTimeSource)\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(matchingTestDomainName), nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.CreateDomainCacheEntry(matchingTestDomainName), nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(matchingTestDomainName, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().RegisterDomainChangeCallback(service.Matching, gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()\n\ts.mockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).AnyTimes()\n\ts.mockMembershipResolver = membership.NewMockResolver(s.controller)\n\ts.mockMembershipResolver.EXPECT().Lookup(gomock.Any(), gomock.Any()).Return(membership.HostInfo{}, nil).AnyTimes()\n\ts.mockMembershipResolver.EXPECT().WhoAmI().Return(membership.HostInfo{}, nil).AnyTimes()\n\ts.mockMembershipResolver.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any()).AnyTimes()\n\ts.mockIsolationStore = dynamicconfig.NewMockClient(s.controller)\n\tdcClient := dynamicconfig.NewInMemoryClient()\n\ts.NoError(dcClient.UpdateValue(dynamicproperties.EnableTasklistIsolation, true))\n\tdc := dynamicconfig.NewCollection(dcClient, s.logger)\n\ts.isolationState, _ = defaultisolationgroupstate.NewDefaultIsolationGroupStateWatcherWithConfigStoreClient(s.logger,\n\t\tdc,\n\t\ts.mockDomainCache,\n\t\ts.mockIsolationStore,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tgetIsolationGroupsHelper)\n\ts.handlerContext = newHandlerContext(\n\t\tcontext.Background(),\n\t\tmatchingTestDomainName,\n\t\t&types.TaskList{Name: matchingTestTaskList, Kind: &tlKindNormal},\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\tmetrics.MatchingTaskListMgrScope,\n\t\ttestlogger.New(s.Suite.T()),\n\t)\n\ts.mockActiveClusterMgr = *activecluster.NewMockManager(s.controller)\n\n\ts.matchingEngine = s.newMatchingEngine(defaultTestConfig(), s.taskManager)\n\ts.mockExecutor = s.matchingEngine.executor.(*executorclient.MockExecutor[tasklist.ShardProcessor])\n\ts.matchingEngine.Start()\n}\n\nfunc (s *matchingEngineSuite) TearDownTest() {\n\ts.mockExecutionManager.AssertExpectations(s.T())\n\ts.matchingEngine.Stop()\n\ts.controller.Finish()\n}\n\nfunc (s *matchingEngineSuite) newMatchingEngine(\n\tconfig *config.Config, taskMgr persistence.TaskManager,\n) *matchingEngineImpl {\n\te := NewEngine(\n\t\ttaskMgr,\n\t\tcluster.GetTestClusterMetadata(true),\n\t\ts.mockHistoryClient,\n\t\ts.mockMatchingClient,\n\t\tconfig,\n\t\ts.logger,\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\ttally.NoopScope,\n\t\ts.mockDomainCache,\n\t\ts.mockMembershipResolver,\n\t\ts.isolationState,\n\t\ts.mockTimeSource,\n\t\ts.mockShardExecutorClient,\n\t\tdefaultSDExecutorConfig(),\n\t\tnil,\n\t).(*matchingEngineImpl)\n\t// Replace the real executor with a mock that behaves as a fully onboarded SD executor.\n\tmockExec := executorclient.NewMockExecutor[tasklist.ShardProcessor](s.controller)\n\tmockExec.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(&tasklist.MockShardProcessor{}, nil).AnyTimes()\n\tmockExec.EXPECT().Start(gomock.Any()).AnyTimes()\n\tmockExec.EXPECT().Stop().AnyTimes()\n\te.executor = mockExec\n\treturn e\n}\n\nfunc (s *matchingEngineSuite) TestPollForActivityTasksEmptyResult() {\n\ts.PollForTasksEmptyResultTest(context.Background(), persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestPollForDecisionTasksEmptyResult() {\n\ts.PollForTasksEmptyResultTest(context.Background(), persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) TestPollForActivityTasksEmptyResultWithShortContext() {\n\tshortContextTimeout := returnEmptyTaskTimeBudget + 10*time.Millisecond\n\tcallContext, cancel := context.WithTimeout(context.Background(), shortContextTimeout)\n\tdefer cancel()\n\ts.PollForTasksEmptyResultTest(callContext, persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestPollForDecisionTasksEmptyResultWithShortContext() {\n\tshortContextTimeout := returnEmptyTaskTimeBudget + 10*time.Millisecond\n\tcallContext, cancel := context.WithTimeout(context.Background(), shortContextTimeout)\n\tdefer cancel()\n\ts.PollForTasksEmptyResultTest(callContext, persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) TestOnlyUnloadMatchingInstance() {\n\ttaskListID := tasklist.NewTestTaskListID(\n\t\ts.T(),\n\t\tuuid.New(),\n\t\t\"makeToast\",\n\t\tpersistence.TaskListTypeActivity)\n\ttlKind := types.TaskListKindNormal\n\ttlm, err := s.matchingEngine.getOrCreateTaskListManager(context.Background(), taskListID, tlKind)\n\ts.Require().NoError(err)\n\tparams := tasklist.ManagerParams{\n\t\tDomainCache:     s.matchingEngine.domainCache,\n\t\tLogger:          s.matchingEngine.logger,\n\t\tMetricsClient:   s.matchingEngine.metricsClient,\n\t\tTaskManager:     s.matchingEngine.taskManager,\n\t\tClusterMetadata: s.matchingEngine.clusterMetadata,\n\t\tIsolationState:  s.matchingEngine.isolationState,\n\t\tMatchingClient:  s.matchingEngine.matchingClient,\n\t\tRegistry:        s.matchingEngine.taskListRegistry,\n\t\tTaskList:        taskListID, // same taskListID as above\n\t\tTaskListKind:    tlKind,\n\t\tCfg:             s.matchingEngine.config,\n\t\tTimeSource:      s.matchingEngine.timeSource,\n\t\tCreateTime:      s.matchingEngine.timeSource.Now(),\n\t\tHistoryService:  s.matchingEngine.historyService,\n\t}\n\ttlm2, err := tasklist.NewManager(params)\n\ts.Require().NoError(err)\n\n\t// try to unload a different tlm instance with the same taskListID\n\ts.matchingEngine.unloadTaskList(tlm2)\n\n\tgot, err := s.matchingEngine.getOrCreateTaskListManager(context.Background(), taskListID, tlKind)\n\ts.Require().NoError(err)\n\ts.Require().Same(tlm, got,\n\t\t\"Unload call with non-matching taskListManager should not cause unload\")\n\n\t// this time unload the right tlm\n\ts.matchingEngine.unloadTaskList(tlm)\n\n\tgot, err = s.matchingEngine.getOrCreateTaskListManager(context.Background(), taskListID, tlKind)\n\ts.Require().NoError(err)\n\ts.Require().NotSame(tlm, got,\n\t\t\"Unload call with matching incarnation should have caused unload\")\n}\n\nfunc (s *matchingEngineSuite) TestPollForDecisionTasks() {\n\ts.PollForDecisionTasksResultTest()\n}\n\nfunc (s *matchingEngineSuite) PollForDecisionTasksResultTest() {\n\ttaskType := persistence.TaskListTypeDecision\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\ttlKind := types.TaskListKindNormal\n\tstickyTl := \"makeStickyToast\"\n\tstickyTlKind := types.TaskListKindSticky\n\tidentity := \"selfDrivingToaster\"\n\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stickyTl\n\tstickyTaskList.Kind = &stickyTlKind\n\n\ts.matchingEngine.config.RangeSize = 2 // to test that range is not updated without tasks\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(1)\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\n\trunID := \"run1\"\n\tworkflowID := \"workflow1\"\n\tworkflowType := types.WorkflowType{\n\t\tName: \"workflow\",\n\t}\n\texecution := types.WorkflowExecution{RunID: runID, WorkflowID: workflowID}\n\tscheduleID := int64(0)\n\n\t// History service is using mock\n\ts.mockHistoryClient.EXPECT().RecordDecisionTaskStarted(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(ctx context.Context, taskRequest *types.RecordDecisionTaskStartedRequest, option ...yarpc.CallOption) (*types.RecordDecisionTaskStartedResponse, error) {\n\t\t\ts.logger.Debug(\"Mock Received RecordDecisionTaskStartedRequest\")\n\t\t\ttaskListKindNormal := types.TaskListKindNormal\n\t\t\tresponse := &types.RecordDecisionTaskStartedResponse{}\n\t\t\tresponse.WorkflowType = &workflowType\n\t\t\tresponse.PreviousStartedEventID = common.Int64Ptr(scheduleID)\n\t\t\tresponse.ScheduledEventID = scheduleID + 1\n\t\t\tresponse.Attempt = 0\n\t\t\tresponse.StickyExecutionEnabled = true\n\t\t\tresponse.WorkflowExecutionTaskList = &types.TaskList{\n\t\t\t\tName: tl,\n\t\t\t\tKind: &taskListKindNormal,\n\t\t\t}\n\t\t\treturn response, nil\n\t\t}).AnyTimes()\n\n\taddRequest := &addTaskRequest{\n\t\tTaskType:                      taskType,\n\t\tDomainUUID:                    domainID,\n\t\tExecution:                     &execution,\n\t\tTaskList:                      stickyTaskList,\n\t\tScheduleToStartTimeoutSeconds: 1,\n\t}\n\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\ts.Error(err)\n\ts.Contains(err.Error(), \"sticky worker is unavailable\")\n\t// poll the sticky tasklist, should get no result\n\tpollReq := &pollTaskRequest{\n\t\tTaskType:   taskType,\n\t\tDomainUUID: domainID,\n\t\tTaskList:   stickyTaskList,\n\t\tIdentity:   identity,\n\t}\n\tresp, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\ts.NoError(err)\n\ts.NotNil(resp.AutoConfigHint)\n\tresp.AutoConfigHint = nil\n\ts.Equal(&pollTaskResponse{}, resp)\n\t// add task to sticky tasklist again, this time it should pass\n\t_, err = addTask(s.matchingEngine, s.handlerContext, addRequest)\n\ts.NoError(err)\n\n\tresp, err = pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\n\texpectedResp := &pollTaskResponse{\n\t\tTaskToken:              resp.TaskToken,\n\t\tWorkflowExecution:      &execution,\n\t\tWorkflowType:           &workflowType,\n\t\tPreviousStartedEventID: common.Int64Ptr(scheduleID),\n\t\tAttempt:                0,\n\t\tBacklogCountHint:       1,\n\t\tStickyExecutionEnabled: true,\n\t\tWorkflowExecutionTaskList: &types.TaskList{\n\t\t\tName: tl,\n\t\t\tKind: &tlKind,\n\t\t},\n\t\tAutoConfigHint: &types.AutoConfigHint{\n\t\t\tEnableAutoConfig:   false,\n\t\t\tPollerWaitTimeInMs: 0,\n\t\t},\n\t}\n\n\ts.Nil(err)\n\ts.Equal(expectedResp, resp)\n}\n\nfunc (s *matchingEngineSuite) PollForTasksEmptyResultTest(callContext context.Context, taskType int) {\n\ts.matchingEngine.config.RangeSize = 2 // to test that range is not updated without tasks\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(1)\n\tif _, ok := callContext.Deadline(); !ok {\n\t\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\t}\n\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\tidentity := \"selfDrivingToaster\"\n\n\ttaskList := &types.TaskList{Name: tl}\n\tvar taskListType types.TaskListType\n\ttlID := tasklist.NewTestTaskListID(s.T(), domainID, tl, taskType)\n\ts.handlerContext.Context = callContext\n\tconst pollCount = 10\n\tfor i := 0; i < pollCount; i++ {\n\t\tpollReq := &pollTaskRequest{\n\t\t\tTaskType:   taskType,\n\t\t\tDomainUUID: domainID,\n\t\t\tTaskList:   taskList,\n\t\t\tIdentity:   identity,\n\t\t}\n\t\tpollResp, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\t\ts.NotNil(pollResp.AutoConfigHint)\n\t\tpollResp.AutoConfigHint = nil // poller wait time is not a fixed value, exclude it from comparison\n\t\ts.NoError(err)\n\t\ts.Equal(&pollTaskResponse{}, pollResp)\n\n\t\tif taskType == persistence.TaskListTypeActivity {\n\t\t\ttaskListType = types.TaskListTypeActivity\n\t\t} else {\n\t\t\ttaskListType = types.TaskListTypeDecision\n\t\t}\n\t\tselect {\n\t\tcase <-callContext.Done():\n\t\t\ts.FailNow(\"Call context has expired.\")\n\t\tdefault:\n\t\t}\n\t\t// check the poller information\n\t\ts.handlerContext.Context = context.Background()\n\t\tdescResp, err := s.matchingEngine.DescribeTaskList(s.handlerContext, &types.MatchingDescribeTaskListRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\t\tTaskList:              taskList,\n\t\t\t\tTaskListType:          &taskListType,\n\t\t\t\tIncludeTaskListStatus: false,\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\t\ts.Equal(1, len(descResp.Pollers))\n\t\ts.Equal(identity, descResp.Pollers[0].GetIdentity())\n\t\ts.NotEmpty(descResp.Pollers[0].GetLastAccessTime())\n\t\ts.Nil(descResp.GetTaskListStatus())\n\t}\n\ts.EqualValues(1, s.taskManager.GetRangeID(tlID))\n}\n\nfunc (s *matchingEngineSuite) TestQueryWorkflow() {\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\ttlKind := types.TaskListKindNormal\n\tstickyTl := \"makeStickyToast\"\n\tstickyTlKind := types.TaskListKindSticky\n\tidentity := \"selfDrivingToaster\"\n\ttaskList := &types.TaskList{\n\t\tName: tl,\n\t\tKind: &tlKind,\n\t}\n\tstickyTaskList := &types.TaskList{}\n\tstickyTaskList.Name = stickyTl\n\tstickyTaskList.Kind = &stickyTlKind\n\n\ts.matchingEngine.config.RangeSize = 2 // to test that range is not updated without tasks\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(1)\n\n\trunID := \"run1\"\n\tworkflowID := \"workflow1\"\n\tworkflowType := types.WorkflowType{\n\t\tName: \"workflow\",\n\t}\n\texecution := types.WorkflowExecution{RunID: runID, WorkflowID: workflowID}\n\n\t// History service is using mock\n\ts.mockHistoryClient.EXPECT().GetMutableState(gomock.Any(), &types.GetMutableStateRequest{\n\t\tDomainUUID: domainID,\n\t\tExecution:  &execution,\n\t}).Return(&types.GetMutableStateResponse{\n\t\tPreviousStartedEventID: common.Int64Ptr(123),\n\t\tNextEventID:            345,\n\t\tWorkflowType:           &workflowType,\n\t\tTaskList:               taskList,\n\t\tCurrentBranchToken:     []byte(`branch token`),\n\t\tHistorySize:            999,\n\t\tClientImpl:             \"uber-go\",\n\t\tClientFeatureVersion:   \"1.0.0\",\n\t\tStickyTaskList:         stickyTaskList,\n\t}, nil)\n\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\tresp, err := s.matchingEngine.PollForDecisionTask(s.handlerContext, &types.MatchingPollForDecisionTaskRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\tTaskList: taskList,\n\t\t\t\tIdentity: identity,\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\t\ts.NotNil(resp.TaskToken)\n\t\ttoken, err := s.matchingEngine.tokenSerializer.DeserializeQueryTaskToken(resp.TaskToken)\n\t\ts.NoError(err)\n\t\ts.Equal(domainID, token.DomainID)\n\t\ts.Equal(workflowID, token.WorkflowID)\n\t\ts.Equal(runID, token.RunID)\n\t\ts.Equal(tl, token.TaskList)\n\t\ts.True(resp.StickyExecutionEnabled)\n\t\terr = s.matchingEngine.RespondQueryTaskCompleted(s.handlerContext, &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\tDomainUUID: domainID,\n\t\t\tTaskList:   taskList,\n\t\t\tTaskID:     token.TaskID,\n\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\tTaskToken:     []byte(``),\n\t\t\t\tCompletedType: types.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\t\t\tQueryResult:   []byte(`result`),\n\t\t\t\tWorkerVersionInfo: &types.WorkerVersionInfo{\n\t\t\t\t\tImpl:           \"uber-go\",\n\t\t\t\t\tFeatureVersion: \"1.5.0\",\n\t\t\t\t},\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\t}()\n\ttime.Sleep(10 * time.Millisecond) // wait for poller to start\n\tresp, err := s.matchingEngine.QueryWorkflow(s.handlerContext, &types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID: domainID,\n\t\tTaskList:   taskList,\n\t\tQueryRequest: &types.QueryWorkflowRequest{\n\t\t\tDomain:                \"domain\",\n\t\t\tExecution:             &execution,\n\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t},\n\t})\n\twg.Wait()\n\ts.NoError(err)\n\ts.Equal(&types.MatchingQueryWorkflowResponse{\n\t\tQueryResult: []byte(`result`),\n\t}, resp)\n}\n\nfunc (s *matchingEngineSuite) TestAddActivityTasks() {\n\ts.AddTasksTest(persistence.TaskListTypeActivity, false)\n}\n\nfunc (s *matchingEngineSuite) TestAddDecisionTasks() {\n\ts.AddTasksTest(persistence.TaskListTypeDecision, false)\n}\n\nfunc (s *matchingEngineSuite) TestAddActivityTasksForwarded() {\n\ts.AddTasksTest(persistence.TaskListTypeActivity, true)\n}\n\nfunc (s *matchingEngineSuite) TestAddDecisionTasksForwarded() {\n\ts.AddTasksTest(persistence.TaskListTypeDecision, true)\n}\n\nfunc (s *matchingEngineSuite) AddTasksTest(taskType int, isForwarded bool) {\n\ts.matchingEngine.config.RangeSize = 300 // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(150)\n\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\tforwardedFrom := \"/__cadence_sys/makeToast/1\"\n\n\ttaskList := &types.TaskList{Name: tl}\n\n\tconst taskCount = 111\n\n\trunID := \"run1\"\n\tworkflowID := \"workflow1\"\n\texecution := types.WorkflowExecution{RunID: runID, WorkflowID: workflowID}\n\n\tfor i := int64(0); i < taskCount; i++ {\n\t\tscheduleID := i * 3\n\t\taddRequest := &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    domainID,\n\t\t\tExecution:                     &execution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      taskList,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t}\n\t\tif isForwarded {\n\t\t\taddRequest.ForwardedFrom = forwardedFrom\n\t\t}\n\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\n\t\tswitch isForwarded {\n\t\tcase false:\n\t\t\ts.NoError(err)\n\t\tcase true:\n\t\t\ts.Equal(errRemoteSyncMatchFailed, err)\n\t\t}\n\t}\n\n\tswitch isForwarded {\n\tcase false:\n\t\ts.EqualValues(taskCount, s.taskManager.GetTaskCount(tasklist.NewTestTaskListID(s.T(), domainID, tl, taskType)))\n\tcase true:\n\t\ts.EqualValues(0, s.taskManager.GetTaskCount(tasklist.NewTestTaskListID(s.T(), domainID, tl, taskType)))\n\t}\n}\n\nfunc (s *matchingEngineSuite) TestAddAndPollDecisionTasks() {\n\ts.AddAndPollTasks(persistence.TaskListTypeDecision, false)\n}\n\nfunc (s *matchingEngineSuite) TestAddAndPollActivityTasks() {\n\ts.AddAndPollTasks(persistence.TaskListTypeDecision, false)\n}\n\nfunc (s *matchingEngineSuite) TestAddAndPollDecisionTasksIsolation() {\n\ts.AddAndPollTasks(persistence.TaskListTypeDecision, true)\n}\n\nfunc (s *matchingEngineSuite) TestAddAndPollActivityTasksIsolation() {\n\ts.AddAndPollTasks(persistence.TaskListTypeDecision, true)\n}\n\nfunc (s *matchingEngineSuite) AddAndPollTasks(taskType int, enableIsolation bool) {\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(enableIsolation)\n\n\tisolationGroups := s.matchingEngine.config.AllIsolationGroups()\n\n\tconst taskCount = 6\n\tconst initialRangeID = 102\n\t// TODO: Understand why publish is low when rangeSize is 3\n\tconst rangeSize = 30\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\n\ts.setupGetDrainStatus()\n\n\tfor i := int64(0); i < taskCount; i++ {\n\t\tscheduleID := i * 3\n\t\taddRequest := &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      testParam.TaskList,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\tPartitionConfig:               map[string]string{isolationgroup.GroupKey: isolationGroups[int(i)%len(isolationGroups)]},\n\t\t}\n\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\t\ts.NoError(err)\n\t}\n\ts.EqualValues(taskCount, s.taskManager.GetTaskCount(testParam.TaskListID))\n\n\ts.setupRecordTaskStartedMock(taskType, testParam, false)\n\n\tfor i := int64(0); i < taskCount; {\n\t\tscheduleID := i * 3\n\t\tpollReq := &pollTaskRequest{\n\t\t\tTaskType:       taskType,\n\t\t\tDomainUUID:     testParam.DomainID,\n\t\t\tTaskList:       testParam.TaskList,\n\t\t\tIdentity:       testParam.Identity,\n\t\t\tIsolationGroup: isolationGroups[int(i)%len(isolationGroups)],\n\t\t}\n\t\tresult, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\t\ts.NoError(err)\n\t\ts.NotNil(result)\n\t\tif isEmptyToken(result.TaskToken) {\n\t\t\ts.logger.Debug(\"empty poll returned\")\n\t\t\tcontinue\n\t\t}\n\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t\ti++\n\t}\n\ts.EqualValues(0, s.taskManager.GetTaskCount(testParam.TaskListID))\n\texpectedRange := getExpectedRange(initialRangeID, taskCount, rangeSize)\n\t// Due to conflicts some ids are skipped and more real ranges are used.\n\ts.True(expectedRange <= s.taskManager.GetRangeID(testParam.TaskListID))\n}\n\nfunc (s *matchingEngineSuite) TestSyncMatchActivityTasks() {\n\ts.SyncMatchTasks(persistence.TaskListTypeActivity, false)\n}\n\nfunc (s *matchingEngineSuite) TestSyncMatchDecisionTasks() {\n\ts.SyncMatchTasks(persistence.TaskListTypeDecision, false)\n}\n\nfunc (s *matchingEngineSuite) TestSyncMatchActivityTasksIsolation() {\n\ts.SyncMatchTasks(persistence.TaskListTypeActivity, true)\n}\n\nfunc (s *matchingEngineSuite) TestSyncMatchDecisionTasksIsolation() {\n\ts.SyncMatchTasks(persistence.TaskListTypeDecision, true)\n}\n\nfunc (s *matchingEngineSuite) SyncMatchTasks(taskType int, enableIsolation bool) {\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(enableIsolation)\n\tconst taskCount = 10\n\tconst initialRangeID = 102\n\t// TODO: Understand why publish is low when rangeSize is 3\n\tconst rangeSize = 30\n\tvar throttledTaskCount int64\n\tif taskType == persistence.TaskListTypeActivity {\n\t\tthrottledTaskCount = 3\n\t}\n\tisolationGroups := s.matchingEngine.config.AllIsolationGroups\n\n\t// Set a short long poll expiration so we don't have to wait too long for 0 throttling cases\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(200 * time.Millisecond)\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\ts.matchingEngine.config.TaskDispatchRPSTTL = time.Nanosecond\n\ts.matchingEngine.config.MinTaskThrottlingBurstSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(_minBurst)\n\t// So we can get snapshots\n\tscope := tally.NewTestScope(\"test\", nil)\n\ts.matchingEngine.metricsClient = metrics.NewClient(scope, metrics.Matching, metrics.HistogramMigration{})\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\n\ts.setupGetDrainStatus()\n\ts.setupRecordTaskStartedMock(taskType, testParam, false)\n\n\tpollFunc := func(maxDispatch float64, isolationGroup string) (*pollTaskResponse, error) {\n\t\tpollReq := &pollTaskRequest{\n\t\t\tTaskType:         taskType,\n\t\t\tDomainUUID:       testParam.DomainID,\n\t\t\tTaskList:         testParam.TaskList,\n\t\t\tIdentity:         testParam.Identity,\n\t\t\tTaskListMetadata: &types.TaskListMetadata{MaxTasksPerSecond: &maxDispatch},\n\t\t\tIsolationGroup:   isolationGroup,\n\t\t}\n\t\treturn pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\t}\n\n\tfor i := int64(0); i < taskCount; i++ {\n\t\tscheduleID := i * 3\n\t\tgroup := isolationGroups()[int(i)%len(isolationGroups())]\n\t\tvar wg sync.WaitGroup\n\t\tvar result *pollTaskResponse\n\t\tvar pollErr error\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tresult, pollErr = pollFunc(_defaultTaskDispatchRPS, group)\n\t\t}()\n\t\ttime.Sleep(20 * time.Millisecond) // Wait for a short period of time to let the poller start so that sync match will happen\n\t\taddRequest := &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      testParam.TaskList,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\tPartitionConfig:               map[string]string{isolationgroup.GroupKey: group},\n\t\t}\n\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\t\twg.Wait()\n\t\ts.NoError(err)\n\t\ts.NoError(pollErr)\n\t\ts.NotNil(result)\n\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t}\n\t// expect more than half of the tasks get sync matches\n\ts.True(s.taskManager.GetCreateTaskCount(testParam.TaskListID) < taskCount/2)\n\n\t// Set the dispatch RPS to 0, to verify that poller will not get any task and task will be persisted into database\n\t// Revert the dispatch RPS and verify that poller will get the task\n\tfor i := int64(0); i < throttledTaskCount; i++ {\n\t\tscheduleID := i * 3\n\t\tgroup := isolationGroups()[int(i)%len(isolationGroups())]\n\t\tvar pollerDone sync.WaitGroup\n\t\tvar result *pollTaskResponse\n\t\tvar pollErr error\n\t\tpollerDone.Add(1)\n\t\tgo func() {\n\t\t\tdefer pollerDone.Done()\n\t\t\tresult, pollErr = pollFunc(0.0, group)\n\t\t}()\n\t\ttime.Sleep(20 * time.Millisecond) // Wait for a short period of time to let the poller start so that sync match will happen\n\t\taddRequest := &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      testParam.TaskList,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\tPartitionConfig:               map[string]string{isolationgroup.GroupKey: group},\n\t\t}\n\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\t\tpollerDone.Wait()\n\t\ts.NoError(err)\n\t\ts.NoError(pollErr)\n\t\ts.NotNil(result)\n\t\t// when ratelimit is set to zero, poller is expected to return empty result\n\t\t// reset ratelimit, poll again and make sure task is returned this time\n\t\ts.True(isEmptyToken(result.TaskToken))\n\t\t// If we don't increment the mockTime then the RateLimiter will never accept a higher RPS\n\t\ts.mockTimeSource.Advance(time.Nanosecond)\n\t\tresult, pollErr = pollFunc(_defaultTaskDispatchRPS, group)\n\t\ts.NoError(err)\n\t\ts.NoError(pollErr)\n\t\ts.NotNil(result)\n\t\ts.False(isEmptyToken(result.TaskToken))\n\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t}\n\ts.True(int(throttledTaskCount) <= s.taskManager.GetCreateTaskCount(testParam.TaskListID))\n\ts.EqualValues(0, s.taskManager.GetTaskCount(testParam.TaskListID))\n\texpectedRange := getExpectedRange(initialRangeID, int(taskCount+throttledTaskCount), rangeSize)\n\t// Due to conflicts some ids are skipped and more real ranges are used.\n\ts.True(expectedRange <= s.taskManager.GetRangeID(testParam.TaskListID))\n\n\tif throttledTaskCount > 0 {\n\t\tsyncCtr := scope.Snapshot().Counters()[\"test.sync_throttle_count_per_tl+domain=\"+matchingTestDomainName+\",operation=TaskListMgr,tasklist=\"+testParam.TaskList.Name+\",tasklistType=activity\"]\n\t\ts.EqualValues(throttledTaskCount, int(syncCtr.Value()))\n\t}\n\n\t// check the poller information\n\tdescResp, err := s.matchingEngine.DescribeTaskList(s.handlerContext, &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID: testParam.DomainID,\n\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\tTaskList:              testParam.TaskList,\n\t\t\tTaskListType:          testParam.TaskListType,\n\t\t\tIncludeTaskListStatus: true,\n\t\t},\n\t})\n\ts.NoError(err)\n\ts.Equal(1, len(descResp.Pollers))\n\ts.Equal(testParam.Identity, descResp.Pollers[0].GetIdentity())\n\ts.NotEmpty(descResp.Pollers[0].GetLastAccessTime())\n\ts.Equal(_defaultTaskDispatchRPS, descResp.Pollers[0].GetRatePerSecond())\n\ts.NotNil(descResp.GetTaskListStatus())\n\ts.True(descResp.GetTaskListStatus().GetRatePerSecond() >= (_defaultTaskDispatchRPS - 1))\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollActivities() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeActivity, 20, 100, false, false)\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollActivitiesWithZeroDispatch() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeActivity, 20, 100, true, false)\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollDecisions() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeDecision, 20, 100, false, false)\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollActivitiesIsolation() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeActivity, 20, 100, false, true)\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollActivitiesWithZeroDispatchIsolation() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeActivity, 20, 100, true, true)\n}\n\nfunc (s *matchingEngineSuite) TestConcurrentAddAndPollDecisionsIsolation() {\n\ts.ConcurrentAddAndPollTasks(persistence.TaskListTypeDecision, 20, 100, false, true)\n}\n\nfunc (s *matchingEngineSuite) ConcurrentAddAndPollTasks(taskType int, workerCount int, taskCount int64, throttled, enableIsolation bool) {\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(enableIsolation)\n\tisolationGroups := s.matchingEngine.config.AllIsolationGroups\n\tdispatchLimitFn := func(wc int, attempt int) float64 {\n\t\treturn _defaultTaskDispatchRPS\n\t}\n\tif throttled {\n\t\tdispatchLimitFn = func(wc int, attempt int) float64 {\n\t\t\tif attempt%50 == 0 && wc%5 == 0 { // Gets triggered atleast 20 times\n\t\t\t\treturn 0\n\t\t\t}\n\t\t\treturn _defaultTaskDispatchRPS\n\t\t}\n\t}\n\tscope := tally.NewTestScope(\"test\", nil)\n\ts.matchingEngine.metricsClient = metrics.NewClient(scope, metrics.Matching, metrics.HistogramMigration{})\n\n\tconst initialRangeID = 0\n\tconst rangeSize = 3\n\tvar scheduleID int64 = 123\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ttlKind := types.TaskListKindNormal\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\t// The TaskListLimiter uses the MockTimeSource, which will be incremented by 1ns after each poll\n\ts.matchingEngine.config.TaskDispatchRPSTTL = 50 * time.Nanosecond\n\ts.matchingEngine.config.MinTaskThrottlingBurstSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(_minBurst)\n\t// Make the pollers time out relatively quickly if there are no tasks\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\n\ts.setupGetDrainStatus()\n\n\tvar wg sync.WaitGroup\n\twg.Add(2 * workerCount)\n\n\tfor p := 0; p < workerCount; p++ {\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor i := int64(0); i < taskCount; i++ {\n\t\t\t\tgroup := isolationGroups()[int(i)%len(isolationGroups())] // let each worker to generate tasks for all isolation groups\n\t\t\t\taddRequest := &addTaskRequest{\n\t\t\t\t\tTaskType:                      taskType,\n\t\t\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\t\t\tScheduleID:                    scheduleID,\n\t\t\t\t\tTaskList:                      testParam.TaskList,\n\t\t\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\t\t\tPartitionConfig:               map[string]string{isolationgroup.GroupKey: group},\n\t\t\t\t}\n\t\t\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\t\t\t\tif err != nil {\n\t\t\t\t\ts.logger.Info(\"Failure in AddActivityTask\", tag.Error(err))\n\t\t\t\t\ti--\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\n\ts.setupRecordTaskStartedMock(taskType, testParam, false)\n\n\tfor p := 0; p < workerCount; p++ {\n\t\tgo func(wNum int) {\n\t\t\tdefer wg.Done()\n\t\t\tattempt := 0\n\t\t\tfor i := int64(0); i < taskCount; {\n\t\t\t\tattempt++\n\t\t\t\tmaxDispatch := dispatchLimitFn(wNum, attempt)\n\t\t\t\tgroup := isolationGroups()[int(wNum)%len(isolationGroups())] // let each worker only polls from one isolation group\n\t\t\t\tpollReq := &pollTaskRequest{\n\t\t\t\t\tTaskType:         taskType,\n\t\t\t\t\tDomainUUID:       testParam.DomainID,\n\t\t\t\t\tTaskList:         testParam.TaskList,\n\t\t\t\t\tIdentity:         testParam.Identity,\n\t\t\t\t\tTaskListMetadata: &types.TaskListMetadata{MaxTasksPerSecond: &maxDispatch},\n\t\t\t\t\tIsolationGroup:   group,\n\t\t\t\t}\n\t\t\t\tresult, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\t\t\t\ts.mockTimeSource.Advance(time.Nanosecond)\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(result)\n\t\t\t\tif isEmptyToken(result.TaskToken) {\n\t\t\t\t\ts.logger.Debug(\"empty poll returned\")\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t\t\t\ti++\n\t\t\t}\n\t\t}(p)\n\t}\n\twg.Wait()\n\ttotalTasks := int(taskCount) * workerCount\n\tpersisted := s.taskManager.GetCreateTaskCount(testParam.TaskListID)\n\ts.True(persisted < totalTasks)\n\texpectedRange := getExpectedRange(initialRangeID, persisted, rangeSize)\n\t// Due to conflicts some ids are skipped and more real ranges are used.\n\ts.True(expectedRange <= s.taskManager.GetRangeID(testParam.TaskListID))\n\tmgr, err := s.matchingEngine.getOrCreateTaskListManager(context.Background(), testParam.TaskListID, tlKind)\n\ts.NoError(err)\n\t// stop the tasklist manager to force the acked tasks to be deleted\n\tmgr.Stop()\n\ts.EqualValues(0, s.taskManager.GetTaskCount(testParam.TaskListID))\n\n\tsyncCtr := scope.Snapshot().Counters()[\"test.sync_throttle_count_per_tl+domain=\"+matchingTestDomainName+\",operation=TaskListMgr,tasklist=\"+testParam.TaskList.Name+\",tasklistType=activity\"]\n\tbufCtr := scope.Snapshot().Counters()[\"test.buffer_throttle_count_per_tl+domain=\"+matchingTestDomainName+\",operation=TaskListMgr,tasklist=\"+testParam.TaskList.Name+\",tasklistType=activity\"]\n\ttotal := int64(0)\n\tif syncCtr != nil {\n\t\ttotal += syncCtr.Value()\n\t}\n\tif bufCtr != nil {\n\t\ttotal += bufCtr.Value()\n\t}\n\tif throttled {\n\t\t// atleast once from 0 dispatch poll, and until TTL is hit at which time throttle limit is reset\n\t\t// hard to predict exactly how many times, since the atomic.Value load might not have updated.\n\t\ts.True(total >= 1)\n\t} else {\n\t\ts.EqualValues(0, total)\n\t}\n}\n\nfunc (s *matchingEngineSuite) TestPollActivityWithExpiredContext() {\n\ts.PollWithExpiredContext(persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestPollDecisionWithExpiredContext() {\n\ts.PollWithExpiredContext(persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) PollWithExpiredContext(taskType int) {\n\tidentity := \"nobody\"\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\n\ttaskList := &types.TaskList{Name: tl}\n\n\t// Try with cancelled context\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tcancel()\n\ts.handlerContext.Context = ctx\n\tpollReq := &pollTaskRequest{\n\t\tTaskType:   taskType,\n\t\tDomainUUID: domainID,\n\t\tTaskList:   taskList,\n\t\tIdentity:   identity,\n\t}\n\t_, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\ts.Equal(ctx.Err(), err)\n\n\t// Try with expired context\n\tctx, cancel = context.WithTimeout(context.Background(), time.Second)\n\tdefer cancel()\n\ts.handlerContext.Context = ctx\n\tresp, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\ts.Nil(err)\n\ts.NotNil(resp.AutoConfigHint)\n\tresp.AutoConfigHint = nil\n\ts.Equal(&pollTaskResponse{}, resp)\n}\n\nfunc (s *matchingEngineSuite) TestMultipleEnginesActivitiesRangeStealing() {\n\ts.MultipleEnginesTasksRangeStealing(persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestMultipleEnginesDecisionsRangeStealing() {\n\ts.MultipleEnginesTasksRangeStealing(persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) MultipleEnginesTasksRangeStealing(taskType int) {\n\ts.T().Cleanup(func() { goleak.VerifyNone(s.T()) })\n\t// Add N tasks to engine1 and then N tasks to engine2. Then poll all tasks from engine2. Engine1 should be closed\n\tconst N = 10\n\tconst initialRangeID = 0\n\tconst rangeSize = 5\n\tvar scheduleID int64 = 123\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\n\tengine1 := s.newMatchingEngine(defaultTestConfig(), s.taskManager)\n\tengine1.config.RangeSize = rangeSize\n\tengine1.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\tengine1.Start()\n\tdefer engine1.Stop()\n\n\tcreateAddTaskReq := func() *addTaskRequest {\n\t\treturn &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      testParam.TaskList,\n\t\t\tScheduleToStartTimeoutSeconds: 600,\n\t\t}\n\t}\n\n\t// First add tasks to engine1\n\tfor i := 0; i < N; i++ {\n\t\t_, err := addTask(engine1, s.handlerContext, createAddTaskReq())\n\t\ts.Require().NoError(err)\n\t}\n\n\tengine2 := s.newMatchingEngine(defaultTestConfig(), s.taskManager)\n\tengine2.config.RangeSize = rangeSize\n\tengine2.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\tengine2.Start()\n\tdefer engine2.Stop()\n\n\t// Then add tasks to engine2. It should be able to steal lease and process tasks\n\tfor i := 0; i < N; i++ {\n\t\t_, err := addTask(engine2, s.handlerContext, createAddTaskReq())\n\t\ts.Require().NoError(err)\n\t}\n\n\t_, err := addTask(engine1, s.handlerContext, createAddTaskReq())\n\t// Adding another task to engine1 should fail because it will not have the lease\n\ts.Require().ErrorContains(err, \"task list shutting down\")\n\n\ts.EqualValues(2*N, s.taskManager.GetCreateTaskCount(testParam.TaskListID))\n\n\ts.setupRecordTaskStartedMock(taskType, testParam, true)\n\n\t// Poll all tasks from engine2.\n\tfor i := 0; i < 2*N; i++ {\n\t\tpollReq := &pollTaskRequest{\n\t\t\tTaskType:   taskType,\n\t\t\tDomainUUID: testParam.DomainID,\n\t\t\tTaskList:   testParam.TaskList,\n\t\t\tIdentity:   testParam.Identity,\n\t\t}\n\t\tresult, err := pollTask(engine2, s.handlerContext, pollReq)\n\t\ts.Require().NoError(err)\n\t\ts.NotNil(result)\n\t\tif isEmptyToken(result.TaskToken) {\n\t\t\ts.Fail(\"empty poll returned\")\n\t\t}\n\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t}\n\n\ts.EqualValues(0, s.taskManager.GetTaskCount(testParam.TaskListID))\n\ttotalTasks := 2 * N\n\tpersisted := s.taskManager.GetCreateTaskCount(testParam.TaskListID)\n\t// No sync matching as all messages are added first\n\ts.EqualValues(totalTasks, persisted)\n\texpectedRange := getExpectedRange(initialRangeID, persisted, rangeSize)\n\t// Due to conflicts some ids are skipped and more real ranges are used.\n\ts.True(expectedRange <= s.taskManager.GetRangeID(testParam.TaskListID))\n}\n\nfunc (s *matchingEngineSuite) TestAddTaskAfterStartFailure() {\n\ttaskType := persistence.TaskListTypeActivity\n\trunID := \"run1\"\n\tworkflowID := \"workflow1\"\n\tworkflowExecution := types.WorkflowExecution{RunID: runID, WorkflowID: workflowID}\n\n\tdomainID := \"domainId\"\n\ttl := \"makeToast\"\n\ttlID := tasklist.NewTestTaskListID(s.T(), domainID, tl, taskType)\n\ttlKind := types.TaskListKindNormal\n\n\ttaskList := &types.TaskList{Name: tl}\n\n\tscheduleID := int64(0)\n\taddRequest := &addTaskRequest{\n\t\tTaskType:                      taskType,\n\t\tDomainUUID:                    domainID,\n\t\tExecution:                     &workflowExecution,\n\t\tScheduleID:                    scheduleID,\n\t\tTaskList:                      taskList,\n\t\tScheduleToStartTimeoutSeconds: 1,\n\t}\n\n\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\ts.NoError(err)\n\ts.EqualValues(1, s.taskManager.GetTaskCount(tlID))\n\n\ttlMgr, err := s.matchingEngine.getOrCreateTaskListManager(context.Background(), tlID, tlKind)\n\ts.NoError(err)\n\tctx, err := tlMgr.GetTask(context.Background(), nil)\n\ts.NoError(err)\n\n\tctx.Finish(errors.New(\"test error\"))\n\ts.EqualValues(1, s.taskManager.GetTaskCount(tlID))\n\tctx2, err := tlMgr.GetTask(context.Background(), nil)\n\ts.NoError(err)\n\n\ts.NotEqual(ctx.Event.TaskID, ctx2.Event.TaskID)\n\ts.Equal(ctx.Event.WorkflowID, ctx2.Event.WorkflowID)\n\ts.Equal(ctx.Event.RunID, ctx2.Event.RunID)\n\ts.Equal(ctx.Event.ScheduleID, ctx2.Event.ScheduleID)\n\n\tctx2.Finish(nil)\n\ts.EqualValues(0, s.taskManager.GetTaskCount(tlID))\n}\n\nfunc (s *matchingEngineSuite) TestUnloadActivityTasklistOnIsolationConfigChange() {\n\ts.UnloadTasklistOnIsolationConfigChange(persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestUnloadDecisionTasklistOnIsolationConfigChange() {\n\ts.UnloadTasklistOnIsolationConfigChange(persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) UnloadTasklistOnIsolationConfigChange(taskType int) {\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(50 * time.Millisecond)\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(false)\n\n\tconst taskCount = 1000\n\tconst initialRangeID = 102\n\t// TODO: Understand why publish is low when rangeSize is 3\n\tconst rangeSize = 30\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\n\taddRequest := &addTaskRequest{\n\t\tTaskType:                      taskType,\n\t\tDomainUUID:                    testParam.DomainID,\n\t\tExecution:                     testParam.WorkflowExecution,\n\t\tScheduleID:                    333,\n\t\tTaskList:                      testParam.TaskList,\n\t\tScheduleToStartTimeoutSeconds: 1,\n\t}\n\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\ts.NoError(err)\n\n\t// enable isolation and verify that poller should not get any task\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(true)\n\ts.setupGetDrainStatus()\n\ts.setupRecordTaskStartedMock(taskType, testParam, false)\n\n\tpollReq := &pollTaskRequest{\n\t\tTaskType:   taskType,\n\t\tDomainUUID: testParam.DomainID,\n\t\tTaskList:   testParam.TaskList,\n\t\tIdentity:   testParam.Identity,\n\t}\n\tresult, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\ts.NoError(err)\n\ts.NotNil(result.AutoConfigHint)\n\tresult.AutoConfigHint = nil\n\ts.Equal(&pollTaskResponse{}, result)\n\n\tresult, err = pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\ts.NoError(err)\n\ts.NotNil(result)\n\ts.assertPollTaskResponse(taskType, testParam, 333, result)\n\n\t// disable isolation again and verify add tasklist should fail\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(false)\n\t_, err = addTask(s.matchingEngine, s.handlerContext, addRequest)\n\ts.Error(err)\n\ts.Contains(err.Error(), \"task list shutting down\")\n}\n\nfunc (s *matchingEngineSuite) TestDrainActivityBacklogNoPollersIsolationGroup() {\n\ts.DrainBacklogNoPollersIsolationGroup(persistence.TaskListTypeActivity)\n}\n\nfunc (s *matchingEngineSuite) TestDrainDecisionBacklogNoPollersIsolationGroup() {\n\ts.DrainBacklogNoPollersIsolationGroup(persistence.TaskListTypeDecision)\n}\n\nfunc (s *matchingEngineSuite) DrainBacklogNoPollersIsolationGroup(taskType int) {\n\ts.matchingEngine.config.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\ts.matchingEngine.config.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomainID(true)\n\ts.matchingEngine.config.AsyncTaskDispatchTimeout = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\n\tisolationGroups := s.matchingEngine.config.AllIsolationGroups\n\n\tconst taskCount = 1000\n\tconst initialRangeID = 102\n\t// TODO: Understand why publish is low when rangeSize is 3\n\tconst rangeSize = 30\n\t// use a const scheduleID because we don't know the order of task polled\n\tconst scheduleID = 11111\n\n\ttestParam := newTestParam(s.T(), taskType)\n\ts.taskManager.SetRangeID(testParam.TaskListID, initialRangeID)\n\ts.matchingEngine.config.RangeSize = rangeSize // override to low number for the test\n\ts.matchingEngine.config.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\t_, err := s.matchingEngine.getOrCreateTaskListManager(context.Background(), testParam.TaskListID, testParam.TaskList.GetKind())\n\ts.NoError(err)\n\t// advance the time a bit more than warmup time of new tasklist after the creation of tasklist manager, which is 1 minute\n\ts.mockTimeSource.Advance(time.Minute)\n\ts.mockTimeSource.Advance(time.Second)\n\n\ts.setupGetDrainStatus()\n\n\tfor i := int64(0); i < taskCount; i++ {\n\t\taddRequest := &addTaskRequest{\n\t\t\tTaskType:                      taskType,\n\t\t\tDomainUUID:                    testParam.DomainID,\n\t\t\tExecution:                     testParam.WorkflowExecution,\n\t\t\tScheduleID:                    scheduleID,\n\t\t\tTaskList:                      testParam.TaskList,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\tPartitionConfig:               map[string]string{isolationgroup.GroupKey: isolationGroups()[int(i)%len(isolationGroups())]},\n\t\t}\n\t\t_, err := addTask(s.matchingEngine, s.handlerContext, addRequest)\n\t\ts.NoError(err)\n\t}\n\ts.EqualValues(taskCount, s.taskManager.GetTaskCount(testParam.TaskListID))\n\n\ts.setupRecordTaskStartedMock(taskType, testParam, false)\n\n\tfor i := int64(0); i < taskCount; {\n\t\tpollReq := &pollTaskRequest{\n\t\t\tTaskType:       taskType,\n\t\t\tDomainUUID:     testParam.DomainID,\n\t\t\tTaskList:       testParam.TaskList,\n\t\t\tIdentity:       testParam.Identity,\n\t\t\tIsolationGroup: isolationGroups()[0],\n\t\t}\n\t\tresult, err := pollTask(s.matchingEngine, s.handlerContext, pollReq)\n\t\ts.NoError(err)\n\t\ts.NotNil(result)\n\t\tif isEmptyToken(result.TaskToken) {\n\t\t\ts.logger.Debug(\"empty poll returned\")\n\t\t\tcontinue\n\t\t}\n\t\ts.assertPollTaskResponse(taskType, testParam, scheduleID, result)\n\t\ti++\n\t}\n\ts.EqualValues(0, s.taskManager.GetTaskCount(testParam.TaskListID))\n\texpectedRange := getExpectedRange(initialRangeID, taskCount, rangeSize)\n\t// Due to conflicts some ids are skipped and more real ranges are used.\n\ts.True(expectedRange <= s.taskManager.GetRangeID(testParam.TaskListID))\n}\n\nfunc (s *matchingEngineSuite) setupRecordTaskStartedMock(taskType int, param *testParam, checkDuplicate bool) {\n\tstartedTasks := make(map[int64]bool)\n\tif taskType == persistence.TaskListTypeActivity {\n\t\ts.mockHistoryClient.EXPECT().RecordActivityTaskStarted(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\tfunc(ctx context.Context, taskRequest *types.RecordActivityTaskStartedRequest, option ...yarpc.CallOption) (*types.RecordActivityTaskStartedResponse, error) {\n\t\t\t\ts.logger.Debug(fmt.Sprintf(\"Mock Received RecordActivityTaskStartedRequest, taskID: %v\", taskRequest.TaskID))\n\t\t\t\tif checkDuplicate {\n\t\t\t\t\tif _, ok := startedTasks[taskRequest.TaskID]; ok {\n\t\t\t\t\t\treturn nil, &types.EventAlreadyStartedError{Message: \"already started\"}\n\t\t\t\t\t}\n\t\t\t\t\tstartedTasks[taskRequest.TaskID] = true\n\t\t\t\t}\n\t\t\t\treturn &types.RecordActivityTaskStartedResponse{\n\t\t\t\t\tScheduledEvent: newActivityTaskScheduledEvent(taskRequest.ScheduleID, 0,\n\t\t\t\t\t\t&types.ScheduleActivityTaskDecisionAttributes{\n\t\t\t\t\t\t\tActivityID:                    param.ActivityID,\n\t\t\t\t\t\t\tTaskList:                      param.TaskList,\n\t\t\t\t\t\t\tActivityType:                  param.ActivityType,\n\t\t\t\t\t\t\tInput:                         param.ActivityInput,\n\t\t\t\t\t\t\tHeader:                        param.ActivityHeader,\n\t\t\t\t\t\t\tScheduleToCloseTimeoutSeconds: common.Int32Ptr(100),\n\t\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(50),\n\t\t\t\t\t\t\tStartToCloseTimeoutSeconds:    common.Int32Ptr(50),\n\t\t\t\t\t\t\tHeartbeatTimeoutSeconds:       common.Int32Ptr(10),\n\t\t\t\t\t\t}),\n\t\t\t\t}, nil\n\t\t\t}).AnyTimes()\n\t} else {\n\t\ts.mockHistoryClient.EXPECT().RecordDecisionTaskStarted(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\tfunc(ctx context.Context, taskRequest *types.RecordDecisionTaskStartedRequest, option ...yarpc.CallOption) (*types.RecordDecisionTaskStartedResponse, error) {\n\t\t\t\ts.logger.Debug(fmt.Sprintf(\"Mock Received RecordDecisionTaskStartedRequest, taskID: %v\", taskRequest.TaskID))\n\t\t\t\tif checkDuplicate {\n\t\t\t\t\tif _, ok := startedTasks[taskRequest.TaskID]; ok {\n\t\t\t\t\t\treturn nil, &types.EventAlreadyStartedError{Message: \"already started\"}\n\t\t\t\t\t}\n\t\t\t\t\tstartedTasks[taskRequest.TaskID] = true\n\t\t\t\t}\n\t\t\t\treturn &types.RecordDecisionTaskStartedResponse{\n\t\t\t\t\tPreviousStartedEventID: &param.StartedEventID,\n\t\t\t\t\tStartedEventID:         param.StartedEventID,\n\t\t\t\t\tScheduledEventID:       taskRequest.ScheduleID,\n\t\t\t\t\tWorkflowType:           param.WorkflowType,\n\t\t\t\t}, nil\n\t\t\t}).AnyTimes()\n\t}\n}\n\nfunc (s *matchingEngineSuite) setupGetDrainStatus() {\n\ts.mockIsolationStore.EXPECT().GetListValue(dynamicproperties.DefaultIsolationGroupConfigStoreManagerGlobalMapping, nil).Return(nil, nil).AnyTimes()\n}\n\nfunc (s *matchingEngineSuite) awaitCondition(cond func() bool, timeout time.Duration) bool {\n\texpiry := time.Now().Add(timeout)\n\tfor !cond() {\n\t\ttime.Sleep(time.Millisecond * 5)\n\t\tif time.Now().After(expiry) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc (s *matchingEngineSuite) assertPollTaskResponse(taskType int, param *testParam, scheduleID int64, actual *pollTaskResponse) {\n\tif taskType == persistence.TaskListTypeActivity {\n\t\ttoken := &common.TaskToken{\n\t\t\tDomainID:     param.DomainID,\n\t\t\tWorkflowID:   param.WorkflowExecution.WorkflowID,\n\t\t\tRunID:        param.WorkflowExecution.RunID,\n\t\t\tScheduleID:   scheduleID,\n\t\t\tActivityID:   param.ActivityID,\n\t\t\tActivityType: param.ActivityType.Name,\n\t\t}\n\t\ts.EqualValues(token, actual.TaskToken)\n\t\ts.EqualValues(param.ActivityID, actual.ActivityID)\n\t\ts.EqualValues(param.ActivityType, actual.ActivityType)\n\t\ts.EqualValues(param.ActivityInput, actual.Input)\n\t\ts.EqualValues(param.ActivityHeader, actual.Header)\n\t\ts.EqualValues(param.WorkflowExecution, actual.WorkflowExecution)\n\t} else {\n\t\ttoken := &common.TaskToken{\n\t\t\tDomainID:   param.DomainID,\n\t\t\tWorkflowID: param.WorkflowExecution.WorkflowID,\n\t\t\tRunID:      param.WorkflowExecution.RunID,\n\t\t\tScheduleID: scheduleID,\n\t\t}\n\t\ts.EqualValues(token, actual.TaskToken)\n\t\ts.EqualValues(param.WorkflowExecution, actual.WorkflowExecution)\n\t\ts.EqualValues(param.WorkflowType, actual.WorkflowType)\n\t\ts.EqualValues(param.StartedEventID, actual.StartedEventID)\n\t}\n}\n\nfunc (s *matchingEngineSuite) TestConfigDefaultHostName() {\n\tconfigEmpty := config.Config{}\n\ts.NotEqualValues(s.matchingEngine.config.HostName, configEmpty.HostName)\n\ts.EqualValues(configEmpty.HostName, \"\")\n}\n\nfunc newActivityTaskScheduledEvent(eventID int64, decisionTaskCompletedEventID int64,\n\tscheduleAttributes *types.ScheduleActivityTaskDecisionAttributes) *types.HistoryEvent {\n\thistoryEvent := newHistoryEvent(eventID, types.EventTypeActivityTaskScheduled)\n\tattributes := &types.ActivityTaskScheduledEventAttributes{}\n\tattributes.ActivityID = scheduleAttributes.ActivityID\n\tattributes.ActivityType = scheduleAttributes.ActivityType\n\tattributes.TaskList = scheduleAttributes.TaskList\n\tattributes.Input = scheduleAttributes.Input\n\tattributes.Header = scheduleAttributes.Header\n\tattributes.ScheduleToCloseTimeoutSeconds = scheduleAttributes.ScheduleToCloseTimeoutSeconds\n\tattributes.ScheduleToStartTimeoutSeconds = scheduleAttributes.ScheduleToStartTimeoutSeconds\n\tattributes.StartToCloseTimeoutSeconds = scheduleAttributes.StartToCloseTimeoutSeconds\n\tattributes.HeartbeatTimeoutSeconds = scheduleAttributes.HeartbeatTimeoutSeconds\n\tattributes.DecisionTaskCompletedEventID = decisionTaskCompletedEventID\n\thistoryEvent.ActivityTaskScheduledEventAttributes = attributes\n\n\treturn historyEvent\n}\n\nfunc newHistoryEvent(eventID int64, eventType types.EventType) *types.HistoryEvent {\n\tts := common.Int64Ptr(time.Now().UnixNano())\n\thistoryEvent := &types.HistoryEvent{}\n\thistoryEvent.ID = eventID\n\thistoryEvent.Timestamp = ts\n\thistoryEvent.EventType = &eventType\n\n\treturn historyEvent\n}\n\nfunc validateTimeRange(t time.Time, expectedDuration time.Duration) bool {\n\tcurrentTime := time.Now()\n\tdiff := time.Duration(currentTime.UnixNano() - t.UnixNano())\n\tif diff > expectedDuration {\n\t\tfmt.Printf(\"Current time: %v, Application time: %v, Difference: %v \\n\", currentTime, t, diff)\n\t\treturn false\n\t}\n\treturn true\n}\n\nfunc defaultTestConfig() *config.Config {\n\tconfig := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, getIsolationGroupsHelper)\n\tconfig.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(100 * time.Millisecond)\n\tconfig.MaxTaskDeleteBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(1)\n\tconfig.ReadRangeSize = dynamicproperties.GetIntPropertyFn(50000)\n\tconfig.GetTasksBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(10)\n\tconfig.AsyncTaskDispatchTimeout = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\tconfig.MaxTimeBetweenTaskDeletes = time.Duration(0)\n\tconfig.EnableTasklistOwnershipGuard = func(opts ...dynamicproperties.FilterOption) bool { return true }\n\treturn config\n}\n\nfunc defaultSDExecutorConfig() clientcommon.Config {\n\treturn clientcommon.Config{\n\t\tNamespaces: []clientcommon.NamespaceConfig{{\n\t\t\tNamespace:         \"cadence-matching\",\n\t\t\tHeartBeatInterval: 1 * time.Second,\n\t\t\tMigrationMode:     sdconfig.MigrationModeONBOARDED,\n\t\t\tTTLShard:          5 * time.Minute,\n\t\t\tTTLReport:         1 * time.Minute,\n\t\t}},\n\t}\n}\n\nfunc getExpectedRange(initialRangeID, taskCount, rangeSize int) int64 {\n\texpectedRange := int64(initialRangeID + taskCount/rangeSize)\n\tif taskCount%rangeSize > 0 {\n\t\texpectedRange++\n\t}\n\treturn expectedRange\n}\n\ntype testParam struct {\n\tDomainID          string\n\tWorkflowExecution *types.WorkflowExecution\n\tTaskList          *types.TaskList\n\tTaskListID        *tasklist.Identifier\n\tTaskListType      *types.TaskListType\n\tIdentity          string\n\tActivityID        string\n\tActivityType      *types.ActivityType\n\tActivityInput     []byte\n\tActivityHeader    *types.Header\n\tWorkflowType      *types.WorkflowType\n\tStartedEventID    int64\n\tScheduledEventID  int64\n}\n\nfunc newTestParam(t *testing.T, taskType int) *testParam {\n\tdomainID := uuid.New()\n\ttaskList := &types.TaskList{\n\t\tName: strings.ReplaceAll(uuid.New(), \"-\", \"\"), // metric tags are sanitized\n\t}\n\ttlID := tasklist.NewTestTaskListID(t, domainID, taskList.Name, taskType)\n\ttaskListType := types.TaskListType(taskType)\n\treturn &testParam{\n\t\tDomainID: domainID,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: uuid.New(),\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tTaskList:     taskList,\n\t\tTaskListID:   tlID,\n\t\tTaskListType: &taskListType,\n\t\tIdentity:     uuid.New(),\n\t\tActivityID:   uuid.New(),\n\t\tActivityType: &types.ActivityType{\n\t\t\tName: uuid.New(),\n\t\t},\n\t\tActivityInput:    []byte(uuid.New()),\n\t\tActivityHeader:   &types.Header{Fields: map[string][]byte{\"tracing\": []byte(\"tracing data\")}},\n\t\tWorkflowType:     &types.WorkflowType{Name: uuid.New()},\n\t\tScheduledEventID: 1412,\n\t}\n}\n\ntype addTaskRequest struct {\n\tTaskType                      int\n\tDomainUUID                    string\n\tExecution                     *types.WorkflowExecution\n\tTaskList                      *types.TaskList\n\tScheduleID                    int64\n\tScheduleToStartTimeoutSeconds int32\n\tSource                        *types.TaskSource\n\tForwardedFrom                 string\n\tPartitionConfig               map[string]string\n}\n\ntype addTaskResponse struct {\n\tPartitionConfig *types.TaskListPartitionConfig\n}\n\nfunc addTask(engine *matchingEngineImpl, hCtx *handlerContext, request *addTaskRequest) (*addTaskResponse, error) {\n\tif request.TaskType == persistence.TaskListTypeActivity {\n\t\tresp, err := engine.AddActivityTask(hCtx, &types.AddActivityTaskRequest{\n\t\t\tSourceDomainUUID:              request.DomainUUID,\n\t\t\tDomainUUID:                    request.DomainUUID,\n\t\t\tExecution:                     request.Execution,\n\t\t\tTaskList:                      request.TaskList,\n\t\t\tScheduleID:                    request.ScheduleID,\n\t\t\tScheduleToStartTimeoutSeconds: &request.ScheduleToStartTimeoutSeconds,\n\t\t\tSource:                        request.Source,\n\t\t\tForwardedFrom:                 request.ForwardedFrom,\n\t\t\tPartitionConfig:               request.PartitionConfig,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn &addTaskResponse{\n\t\t\tPartitionConfig: resp.PartitionConfig,\n\t\t}, nil\n\t}\n\tresp, err := engine.AddDecisionTask(hCtx, &types.AddDecisionTaskRequest{\n\t\tDomainUUID:                    request.DomainUUID,\n\t\tExecution:                     request.Execution,\n\t\tTaskList:                      request.TaskList,\n\t\tScheduleID:                    request.ScheduleID,\n\t\tScheduleToStartTimeoutSeconds: &request.ScheduleToStartTimeoutSeconds,\n\t\tSource:                        request.Source,\n\t\tForwardedFrom:                 request.ForwardedFrom,\n\t\tPartitionConfig:               request.PartitionConfig,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &addTaskResponse{\n\t\tPartitionConfig: resp.PartitionConfig,\n\t}, nil\n}\n\ntype pollTaskRequest struct {\n\tTaskType         int\n\tDomainUUID       string\n\tPollerID         string\n\tTaskList         *types.TaskList\n\tIdentity         string\n\tForwardedFrom    string\n\tIsolationGroup   string\n\tTaskListMetadata *types.TaskListMetadata\n\tBinaryChecksum   string\n}\n\ntype pollTaskResponse struct {\n\tTaskToken                       *common.TaskToken\n\tWorkflowExecution               *types.WorkflowExecution\n\tActivityID                      string\n\tActivityType                    *types.ActivityType\n\tInput                           []byte\n\tScheduledTimestamp              *int64\n\tScheduleToCloseTimeoutSeconds   *int32\n\tStartedTimestamp                *int64\n\tStartToCloseTimeoutSeconds      *int32\n\tHeartbeatTimeoutSeconds         *int32\n\tScheduledTimestampOfThisAttempt *int64\n\tHeartbeatDetails                []byte\n\tWorkflowType                    *types.WorkflowType\n\tWorkflowDomain                  string\n\tHeader                          *types.Header\n\tPreviousStartedEventID          *int64\n\tStartedEventID                  int64\n\tAttempt                         int64\n\tNextEventID                     int64\n\tBacklogCountHint                int64\n\tStickyExecutionEnabled          bool\n\tQuery                           *types.WorkflowQuery\n\tDecisionInfo                    *types.TransientDecisionInfo\n\tWorkflowExecutionTaskList       *types.TaskList\n\tEventStoreVersion               int32\n\tBranchToken                     []byte\n\tQueries                         map[string]*types.WorkflowQuery\n\tAutoConfigHint                  *types.AutoConfigHint\n}\n\nfunc pollTask(engine *matchingEngineImpl, hCtx *handlerContext, request *pollTaskRequest) (*pollTaskResponse, error) {\n\tif request.TaskType == persistence.TaskListTypeActivity {\n\t\tresp, err := engine.PollForActivityTask(hCtx, &types.MatchingPollForActivityTaskRequest{\n\t\t\tDomainUUID: request.DomainUUID,\n\t\t\tPollerID:   request.PollerID,\n\t\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\t\tTaskList:         request.TaskList,\n\t\t\t\tIdentity:         request.Identity,\n\t\t\t\tTaskListMetadata: request.TaskListMetadata,\n\t\t\t},\n\t\t\tIsolationGroup: request.IsolationGroup,\n\t\t\tForwardedFrom:  request.ForwardedFrom,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tvar token *common.TaskToken\n\t\tif len(resp.TaskToken) > 0 {\n\t\t\ttoken, err = engine.tokenSerializer.Deserialize(resp.TaskToken)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\treturn &pollTaskResponse{\n\t\t\tTaskToken:                       token,\n\t\t\tWorkflowExecution:               resp.WorkflowExecution,\n\t\t\tActivityID:                      resp.ActivityID,\n\t\t\tActivityType:                    resp.ActivityType,\n\t\t\tInput:                           resp.Input,\n\t\t\tScheduledTimestamp:              resp.ScheduledTimestamp,\n\t\t\tScheduleToCloseTimeoutSeconds:   resp.ScheduleToCloseTimeoutSeconds,\n\t\t\tStartedTimestamp:                resp.StartedTimestamp,\n\t\t\tStartToCloseTimeoutSeconds:      resp.StartToCloseTimeoutSeconds,\n\t\t\tHeartbeatTimeoutSeconds:         resp.HeartbeatTimeoutSeconds,\n\t\t\tAttempt:                         int64(resp.Attempt),\n\t\t\tScheduledTimestampOfThisAttempt: resp.ScheduledTimestampOfThisAttempt,\n\t\t\tHeartbeatDetails:                resp.HeartbeatDetails,\n\t\t\tWorkflowType:                    resp.WorkflowType,\n\t\t\tWorkflowDomain:                  resp.WorkflowDomain,\n\t\t\tHeader:                          resp.Header,\n\t\t\tAutoConfigHint:                  resp.AutoConfigHint,\n\t\t}, nil\n\t}\n\tresp, err := engine.PollForDecisionTask(hCtx, &types.MatchingPollForDecisionTaskRequest{\n\t\tDomainUUID: request.DomainUUID,\n\t\tPollerID:   request.PollerID,\n\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\tTaskList:       request.TaskList,\n\t\t\tIdentity:       request.Identity,\n\t\t\tBinaryChecksum: request.BinaryChecksum,\n\t\t},\n\t\tIsolationGroup: request.IsolationGroup,\n\t\tForwardedFrom:  request.ForwardedFrom,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar token *common.TaskToken\n\tif len(resp.TaskToken) > 0 {\n\t\ttoken, err = engine.tokenSerializer.Deserialize(resp.TaskToken)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn &pollTaskResponse{\n\t\tTaskToken:                 token,\n\t\tWorkflowExecution:         resp.WorkflowExecution,\n\t\tWorkflowType:              resp.WorkflowType,\n\t\tPreviousStartedEventID:    resp.PreviousStartedEventID,\n\t\tStartedEventID:            resp.StartedEventID,\n\t\tAttempt:                   resp.Attempt,\n\t\tNextEventID:               resp.NextEventID,\n\t\tBacklogCountHint:          resp.BacklogCountHint,\n\t\tStickyExecutionEnabled:    resp.StickyExecutionEnabled,\n\t\tQuery:                     resp.Query,\n\t\tDecisionInfo:              resp.DecisionInfo,\n\t\tWorkflowExecutionTaskList: resp.WorkflowExecutionTaskList,\n\t\tEventStoreVersion:         resp.EventStoreVersion,\n\t\tBranchToken:               resp.BranchToken,\n\t\tScheduledTimestamp:        resp.ScheduledTimestamp,\n\t\tStartedTimestamp:          resp.StartedTimestamp,\n\t\tQueries:                   resp.Queries,\n\t\tAutoConfigHint:            resp.AutoConfigHint,\n\t}, nil\n}\n\nfunc isEmptyToken(token *common.TaskToken) bool {\n\treturn token == nil || *token == common.TaskToken{}\n}\n\nfunc getIsolationGroupsHelper() []string {\n\treturn []string{\"zone-a\", \"zone-b\"}\n}\n"
  },
  {
    "path": "service/matching/handler/engine_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tcommonerrors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc mustNewIdentifier(t *testing.T, domainID, taskListName string, taskListType int) *tasklist.Identifier {\n\tt.Helper()\n\tid, err := tasklist.NewIdentifier(domainID, taskListName, taskListType)\n\trequire.NoError(t, err)\n\treturn id\n}\n\n// newMockManagerWithTaskListID returns a MockManager with TaskListID() stubbed to return id (AnyTimes).\nfunc newMockManagerWithTaskListID(ctrl *gomock.Controller, id *tasklist.Identifier) *tasklist.MockManager {\n\tmgr := tasklist.NewMockManager(ctrl)\n\tmgr.EXPECT().TaskListID().Return(id).AnyTimes()\n\treturn mgr\n}\n\nfunc TestGetTaskListsByDomain(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\tmockSetup      func(*cache.MockDomainCache, map[tasklist.Identifier]*tasklist.MockManager, map[tasklist.Identifier]*tasklist.MockManager)\n\t\treturnAllKinds bool\n\t\twantErr        bool\n\t\twant           *types.GetTaskListsByDomainResponse\n\t}{\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockTaskListManagers map[tasklist.Identifier]*tasklist.MockManager, mockStickyManagers map[tasklist.Identifier]*tasklist.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"\", errors.New(\"cache failure\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockTaskListManagers map[tasklist.Identifier]*tasklist.MockManager, mockStickyManagers map[tasklist.Identifier]*tasklist.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"test-domain-id\", nil)\n\t\t\t\tfor id, mockManager := range mockTaskListManagers {\n\t\t\t\t\tif id.GetDomainID() == \"test-domain-id\" {\n\t\t\t\t\t\tmockManager.EXPECT().GetTaskListKind().Return(types.TaskListKindNormal)\n\t\t\t\t\t\tmockManager.EXPECT().DescribeTaskList(false).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tIdentity: fmt.Sprintf(\"test-poller-%s\", id.GetRoot()),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfor id, mockManager := range mockStickyManagers {\n\t\t\t\t\tif id.GetDomainID() == \"test-domain-id\" {\n\t\t\t\t\t\tmockManager.EXPECT().GetTaskListKind().Return(types.TaskListKindSticky)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &types.GetTaskListsByDomainResponse{\n\t\t\t\tDecisionTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\t\t\"decision0\": {\n\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tIdentity: \"test-poller-decision0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\t\t\"activity0\": {\n\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tIdentity: \"test-poller-activity0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"success - all kinds\",\n\t\t\treturnAllKinds: true,\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockTaskListManagers map[tasklist.Identifier]*tasklist.MockManager, mockStickyManagers map[tasklist.Identifier]*tasklist.MockManager) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"test-domain-id\", nil)\n\t\t\t\tfor id, mockManager := range mockTaskListManagers {\n\t\t\t\t\tif id.GetDomainID() == \"test-domain-id\" {\n\t\t\t\t\t\tmockManager.EXPECT().DescribeTaskList(false).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tIdentity: fmt.Sprintf(\"test-poller-%s\", id.GetRoot()),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tfor id, mockManager := range mockStickyManagers {\n\t\t\t\t\tif id.GetDomainID() == \"test-domain-id\" {\n\t\t\t\t\t\tmockManager.EXPECT().DescribeTaskList(false).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tIdentity: fmt.Sprintf(\"sticky-poller-%s\", id.GetRoot()),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t})\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &types.GetTaskListsByDomainResponse{\n\t\t\t\tDecisionTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\t\t\"decision0\": {\n\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tIdentity: \"test-poller-decision0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t\"sticky0\": {\n\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tIdentity: \"sticky-poller-sticky0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\t\t\"activity0\": {\n\t\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tIdentity: \"test-poller-activity0\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tdecisionTasklistID := mustNewIdentifier(t, \"test-domain-id\", \"decision0\", 0)\n\t\t\tactivityTasklistID := mustNewIdentifier(t, \"test-domain-id\", \"activity0\", 1)\n\t\t\totherDomainTasklistID := mustNewIdentifier(t, \"other-domain-id\", \"other0\", 0)\n\t\t\tstickyTasklistID := mustNewIdentifier(t, \"test-domain-id\", \"sticky0\", 0)\n\t\t\tmockDecisionTaskListManager := newMockManagerWithTaskListID(mockCtrl, decisionTasklistID)\n\t\t\tmockActivityTaskListManager := newMockManagerWithTaskListID(mockCtrl, activityTasklistID)\n\t\t\tmockOtherDomainTaskListManager := newMockManagerWithTaskListID(mockCtrl, otherDomainTasklistID)\n\t\t\tmockStickyManager := newMockManagerWithTaskListID(mockCtrl, stickyTasklistID)\n\t\t\tmockTaskListManagers := map[tasklist.Identifier]*tasklist.MockManager{\n\t\t\t\t*decisionTasklistID:    mockDecisionTaskListManager,\n\t\t\t\t*activityTasklistID:    mockActivityTaskListManager,\n\t\t\t\t*otherDomainTasklistID: mockOtherDomainTaskListManager,\n\t\t\t}\n\t\t\tmockStickyManagers := map[tasklist.Identifier]*tasklist.MockManager{\n\t\t\t\t*stickyTasklistID: mockStickyManager,\n\t\t\t}\n\t\t\ttc.mockSetup(mockDomainCache, mockTaskListManagers, mockStickyManagers)\n\n\t\t\ttaskListRegistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\tdomainCache:      mockDomainCache,\n\t\t\t\ttaskListRegistry: taskListRegistry,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tEnableReturnAllTaskListKinds: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\t\treturn tc.returnAllKinds\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t\ttaskListRegistry.Register(*decisionTasklistID, mockDecisionTaskListManager)\n\t\t\ttaskListRegistry.Register(*activityTasklistID, mockActivityTaskListManager)\n\t\t\ttaskListRegistry.Register(*otherDomainTasklistID, mockOtherDomainTaskListManager)\n\t\t\ttaskListRegistry.Register(*stickyTasklistID, mockStickyManager)\n\t\t\tresp, err := engine.GetTaskListsByDomain(nil, &types.GetTaskListsByDomainRequest{Domain: \"test-domain\"})\n\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListTaskListPartitions(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *types.MatchingListTaskListPartitionsRequest\n\t\tmockSetup func(*cache.MockDomainCache, *membership.MockResolver)\n\t\twantErr   bool\n\t\twant      *types.ListTaskListPartitionsResponse\n\t}{\n\t\t{\n\t\t\tname: \"domain cache error\",\n\t\t\treq: &types.MatchingListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockResolver *membership.MockResolver) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"\", errors.New(\"cache failure\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid tasklist name\",\n\t\t\treq: &types.MatchingListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/invalid-tasklist-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockResolver *membership.MockResolver) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"test-domain-id\", nil)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.MatchingListTaskListPartitionsRequest{\n\t\t\t\tDomain: \"test-domain\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockDomainCache *cache.MockDomainCache, mockResolver *membership.MockResolver) {\n\t\t\t\t// activity tasklist\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"test-domain-id\", nil)\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"test-tasklist\").Return(membership.NewHostInfo(\"addr2\"), nil)\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"/__cadence_sys/test-tasklist/1\").Return(membership.HostInfo{}, errors.New(\"some error\"))\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"/__cadence_sys/test-tasklist/2\").Return(membership.NewHostInfo(\"addr3\"), nil)\n\t\t\t\t// decision tasklist\n\t\t\t\tmockDomainCache.EXPECT().GetDomainID(\"test-domain\").Return(\"test-domain-id\", nil)\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"test-tasklist\").Return(membership.NewHostInfo(\"addr0\"), nil)\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"/__cadence_sys/test-tasklist/1\").Return(membership.HostInfo{}, errors.New(\"some error\"))\n\t\t\t\tmockResolver.EXPECT().Lookup(gomock.Any(), \"/__cadence_sys/test-tasklist/2\").Return(membership.NewHostInfo(\"addr1\"), nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &types.ListTaskListPartitionsResponse{\n\t\t\t\tDecisionTaskListPartitions: []*types.TaskListPartitionMetadata{\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"test-tasklist\",\n\t\t\t\t\t\tOwnerHostName: \"addr0\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"/__cadence_sys/test-tasklist/1\",\n\t\t\t\t\t\tOwnerHostName: \"\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"/__cadence_sys/test-tasklist/2\",\n\t\t\t\t\t\tOwnerHostName: \"addr1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tActivityTaskListPartitions: []*types.TaskListPartitionMetadata{\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"test-tasklist\",\n\t\t\t\t\t\tOwnerHostName: \"addr2\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"/__cadence_sys/test-tasklist/1\",\n\t\t\t\t\t\tOwnerHostName: \"\",\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tKey:           \"/__cadence_sys/test-tasklist/2\",\n\t\t\t\t\t\tOwnerHostName: \"addr3\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockResolver := membership.NewMockResolver(mockCtrl)\n\t\t\ttc.mockSetup(mockDomainCache, mockResolver)\n\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\tdomainCache:        mockDomainCache,\n\t\t\t\tmembershipResolver: mockResolver,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tNumTasklistWritePartitions: dynamicproperties.GetIntPropertyFilteredByTaskListInfo(3),\n\t\t\t\t},\n\t\t\t}\n\t\t\tresp, err := engine.ListTaskListPartitions(nil, tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCancelOutstandingPoll(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *types.CancelOutstandingPollRequest\n\t\tmockSetup func(mockCtrl *gomock.Controller, processor *tasklist.MockManager, executor *executorclient.MockExecutor[tasklist.ShardProcessor])\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tname: \"invalid tasklist name\",\n\t\t\treq: &types.CancelOutstandingPollRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/invalid-tasklist-name\",\n\t\t\t\t},\n\t\t\t\tPollerID: \"test-poller-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockCtrl *gomock.Controller, mockManager *tasklist.MockManager, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.CancelOutstandingPollRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tPollerID: \"test-poller-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockCtrl *gomock.Controller, mockManager *tasklist.MockManager, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().CancelPoller(\"test-poller-id\")\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\ttasklistID := mustNewIdentifier(t, \"test-domain-id\", \"test-tasklist\", 0)\n\t\t\tmockManager := newMockManagerWithTaskListID(mockCtrl, tasklistID)\n\t\t\texecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\t\t\ttc.mockSetup(mockCtrl, mockManager, executor)\n\t\t\ttaskListRegistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\ttaskListRegistry: taskListRegistry,\n\t\t\t\texecutor:         executor,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return false },\n\t\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t},\n\t\t\t}\n\t\t\ttaskListRegistry.Register(*tasklistID, mockManager)\n\t\t\thCtx := &handlerContext{Context: context.Background()}\n\t\t\terr := engine.CancelOutstandingPoll(hCtx, tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestErrIfShardOwnershipLost(t *testing.T) {\n\ttaskListID := mustNewIdentifier(t, \"test-domain-id\", \"test-tasklist\", 0)\n\n\tnewEngine := func(t *testing.T) (*matchingEngineImpl, *executorclient.MockExecutor[tasklist.ShardProcessor], *membership.MockResolver) {\n\t\tt.Helper()\n\t\tctrl := gomock.NewController(t)\n\t\texecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](ctrl)\n\t\tresolver := membership.NewMockResolver(ctrl)\n\t\tresolver.EXPECT().WhoAmI().Return(membership.NewDetailedHostInfo(\"self\", \"self\", nil), nil).AnyTimes()\n\n\t\tengine := &matchingEngineImpl{\n\t\t\texecutor:           executor,\n\t\t\tmembershipResolver: resolver,\n\t\t\tconfig: &config.Config{\n\t\t\t\tEnableTasklistOwnershipGuard:               func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return false },\n\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t},\n\t\t\tshutdown: make(chan struct{}),\n\t\t\tlogger:   log.NewNoop(),\n\t\t}\n\t\treturn engine, executor, resolver\n\t}\n\n\tassertTypedOwnershipErr := func(t *testing.T, err error, ownedBy, me string) {\n\t\tt.Helper()\n\t\trequire.Error(t, err)\n\t\tvar ownershipErr *commonerrors.TaskListNotOwnedByHostError\n\t\trequire.ErrorAs(t, err, &ownershipErr)\n\t\tassert.Equal(t, ownedBy, ownershipErr.OwnedByIdentity)\n\t\tassert.Equal(t, me, ownershipErr.MyIdentity)\n\t}\n\n\tt.Run(\"ownership guard disabled\", func(t *testing.T) {\n\t\tengine, _, _ := newEngine(t)\n\t\tengine.config.EnableTasklistOwnershipGuard = func(opts ...dynamicproperties.FilterOption) bool { return false }\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"not excluded from sd with shard process error\", func(t *testing.T) {\n\t\tengine, executor, _ := newEngine(t)\n\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"sd lookup failed\"))\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\trequire.Error(t, err)\n\t\tassert.ErrorContains(t, err, \"failed to lookup ownership in SD\")\n\t})\n\n\tt.Run(\"not excluded from sd with shard process not found returns ownership error\", func(t *testing.T) {\n\t\tengine, executor, _ := newEngine(t)\n\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, executorclient.ErrShardProcessNotFound)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\tassertTypedOwnershipErr(t, err, \"not known\", \"self\")\n\t})\n\n\tt.Run(\"not excluded from sd and shard no longer owned\", func(t *testing.T) {\n\t\tengine, executor, _ := newEngine(t)\n\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, nil)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\tassertTypedOwnershipErr(t, err, \"not known\", \"self\")\n\t})\n\n\tt.Run(\"not excluded from sd and shard is still owned by this host\", func(t *testing.T) {\n\t\tengine, executor, _ := newEngine(t)\n\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).\n\t\t\tReturn(&tasklist.MockShardProcessor{}, nil). // not nil being returned because the shard is still owned by this host\n\t\t\tAnyTimes()\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"matching engine is shutting down\", func(t *testing.T) {\n\t\tengine, _, _ := newEngine(t)\n\t\tclose(engine.shutdown)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), taskListID)\n\t\tassertTypedOwnershipErr(t, err, \"not known\", \"self\")\n\t})\n\n\tt.Run(\"excluded from sd, ringpop and owner has changed\", func(t *testing.T) {\n\t\tengine, _, resolver := newEngine(t)\n\t\tengine.config.ExcludeShortLivedTaskListsFromShardManager = func(opts ...dynamicproperties.FilterOption) bool { return true }\n\t\texcludedID := mustNewIdentifier(t, \"test-domain-id\", \"tasklist-550e8400-e29b-41d4-a716-446655440000\", 0)\n\t\tresolver.EXPECT().Lookup(service.Matching, excludedID.GetName()).Return(membership.NewDetailedHostInfo(\"owner\", \"owner\", nil), nil)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), excludedID)\n\t\tassertTypedOwnershipErr(t, err, \"owner\", \"self\")\n\t})\n\n\tt.Run(\"excluded from sd, ringpop and owner is the same\", func(t *testing.T) {\n\t\tengine, _, resolver := newEngine(t)\n\t\tengine.config.ExcludeShortLivedTaskListsFromShardManager = func(opts ...dynamicproperties.FilterOption) bool { return true }\n\t\texcludedID := mustNewIdentifier(t, \"test-domain-id\", \"tasklist-550e8400-e29b-41d4-a716-446655440000\", 0)\n\t\tresolver.EXPECT().Lookup(service.Matching, excludedID.GetName()).Return(membership.NewDetailedHostInfo(\"self\", \"self\", nil), nil)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), excludedID)\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"non-excluded tasklist with uuid-like name and flag disabled still uses executor\", func(t *testing.T) {\n\t\tengine, executor, _ := newEngine(t)\n\t\tengine.config.ExcludeShortLivedTaskListsFromShardManager = func(opts ...dynamicproperties.FilterOption) bool { return false }\n\t\tuuidID := mustNewIdentifier(t, \"test-domain-id\", \"tasklist-550e8400-e29b-41d4-a716-446655440000\", 0)\n\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(&tasklist.MockShardProcessor{}, nil)\n\n\t\terr := engine.errIfShardOwnershipLost(context.Background(), uuidID)\n\t\trequire.NoError(t, err)\n\t})\n}\n\nfunc TestIsExcludedFromShardDistributor(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\ttaskListName string\n\t\tflagEnabled  bool\n\t\twant         bool\n\t}{\n\t\t{\n\t\t\tname:         \"flag disabled, uuid name\",\n\t\t\ttaskListName: \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tflagEnabled:  false,\n\t\t\twant:         false,\n\t\t},\n\t\t{\n\t\t\tname:         \"flag enabled, no uuid in name\",\n\t\t\ttaskListName: \"my-regular-tasklist\",\n\t\t\tflagEnabled:  true,\n\t\t\twant:         false,\n\t\t},\n\t\t{\n\t\t\tname:         \"flag enabled, uuid in name\",\n\t\t\ttaskListName: \"tasklist-550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tflagEnabled:  true,\n\t\t\twant:         true,\n\t\t},\n\t\t{\n\t\t\tname:         \"flag enabled, uuid-only name\",\n\t\t\ttaskListName: \"550e8400-e29b-41d4-a716-446655440000\",\n\t\t\tflagEnabled:  true,\n\t\t\twant:         true,\n\t\t},\n\t\t{\n\t\t\tname:         \"flag disabled, no uuid in name\",\n\t\t\ttaskListName: \"my-regular-tasklist\",\n\t\t\tflagEnabled:  false,\n\t\t\twant:         false,\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return tc.flagEnabled },\n\t\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t},\n\t\t\t}\n\t\t\tgot := engine.isExcludedFromShardDistributor(tc.taskListName)\n\t\t\tassert.Equal(t, tc.want, got)\n\t\t})\n\t}\n}\n\nfunc TestRespondQueryTaskCompleted(t *testing.T) {\n\ttestCases := []struct {\n\t\tname         string\n\t\treq          *types.MatchingRespondQueryTaskCompletedRequest\n\t\tqueryTaskMap map[string]chan *queryResult\n\t\twantErr      bool\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskID: \"id-0\",\n\t\t\t},\n\t\t\tqueryTaskMap: map[string]chan *queryResult{\n\t\t\t\t\"id-0\": make(chan *queryResult, 1),\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname: \"query task not found\",\n\t\t\treq: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskID: \"id-0\",\n\t\t\t},\n\t\t\tqueryTaskMap: map[string]chan *queryResult{},\n\t\t\twantErr:      true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\tlockableQueryTaskMap: lockableQueryTaskMap{\n\t\t\t\t\tqueryTaskMap: tc.queryTaskMap,\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := engine.RespondQueryTaskCompleted(&handlerContext{scope: metrics.NewNoopMetricsClient().Scope(0)}, tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestQueryWorkflow(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\treq       *types.MatchingQueryWorkflowRequest\n\t\thCtx      *handlerContext\n\t\tmockSetup func(*tasklist.MockManager, *lockableQueryTaskMap, *gomock.Controller, *executorclient.MockExecutor[tasklist.ShardProcessor])\n\t\twantErr   bool\n\t\twant      *types.MatchingQueryWorkflowResponse\n\t}{\n\t\t{\n\t\t\tname: \"invalid tasklist name\",\n\t\t\treq: &types.MatchingQueryWorkflowRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/invalid-tasklist-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, queryResultMap *lockableQueryTaskMap, mockCtrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"sticky worker unavailable\",\n\t\t\treq: &types.MatchingQueryWorkflowRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, queryResultMap *lockableQueryTaskMap, mockCtrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().HasPollerAfter(gomock.Any()).Return(false)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"failed to dispatch query task\",\n\t\t\treq: &types.MatchingQueryWorkflowRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, queryResultMap *lockableQueryTaskMap, mockCtrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().DispatchQueryTask(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.MatchingQueryWorkflowRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: func() context.Context {\n\t\t\t\t\treturn context.Background()\n\t\t\t\t}(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, queryResultMap *lockableQueryTaskMap, mockCtrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().DispatchQueryTask(gomock.Any(), gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, taskID string, request *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error) {\n\t\t\t\t\tqueryResChan, ok := queryResultMap.get(taskID)\n\t\t\t\t\tif !ok {\n\t\t\t\t\t\treturn nil, errors.New(\"cannot find query result channel by taskID\")\n\t\t\t\t\t}\n\t\t\t\t\tqueryResChan <- &queryResult{workerResponse: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\t\t\tTaskID: taskID,\n\t\t\t\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\t\t\t\tQueryResult: []byte(\"some result\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}}\n\t\t\t\t\treturn nil, nil\n\t\t\t\t})\n\t\t\t\tmockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &types.MatchingQueryWorkflowResponse{\n\t\t\t\tQueryResult: []byte(\"some result\"),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\ttasklistID := mustNewIdentifier(t, \"test-domain-id\", \"test-tasklist\", 0)\n\t\t\tmockManager := newMockManagerWithTaskListID(mockCtrl, tasklistID)\n\t\t\texecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\t\t\ttaskListRegistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\ttaskListRegistry:     taskListRegistry,\n\t\t\t\ttimeSource:           clock.NewRealTimeSource(),\n\t\t\t\tlockableQueryTaskMap: lockableQueryTaskMap{queryTaskMap: make(map[string]chan *queryResult)},\n\t\t\t\texecutor:             executor,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return false },\n\t\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t},\n\t\t\t}\n\t\t\ttaskListRegistry.Register(*tasklistID, mockManager)\n\t\t\ttc.mockSetup(mockManager, &engine.lockableQueryTaskMap, mockCtrl, executor)\n\t\t\tresp, err := engine.QueryWorkflow(tc.hCtx, tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestWaitForQueryResult(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tresult    *queryResult\n\t\tmockSetup func(*client.MockVersionChecker)\n\t\twantErr   bool\n\t\tassertErr func(*testing.T, error)\n\t\twant      *types.MatchingQueryWorkflowResponse\n\t}{\n\t\t{\n\t\t\tname: \"internal error\",\n\t\t\tresult: &queryResult{\n\t\t\t\tinternalError: errors.New(\"some error\"),\n\t\t\t},\n\t\t\tmockSetup: func(mockVersionChecker *client.MockVersionChecker) {},\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"strong consistency query not supported\",\n\t\t\tresult: &queryResult{\n\t\t\t\tworkerResponse: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\t\t\tWorkerVersionInfo: &types.WorkerVersionInfo{\n\t\t\t\t\t\t\tImpl:           \"uber-go\",\n\t\t\t\t\t\t\tFeatureVersion: \"1.0.0\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockVersionChecker *client.MockVersionChecker) {\n\t\t\t\tmockVersionChecker.EXPECT().SupportsConsistentQuery(\"uber-go\", \"1.0.0\").Return(errors.New(\"version error\"))\n\t\t\t},\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, \"version error\", err.Error())\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - query task completed\",\n\t\t\tresult: &queryResult{\n\t\t\t\tworkerResponse: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\t\t\tWorkerVersionInfo: &types.WorkerVersionInfo{\n\t\t\t\t\t\t\tImpl:           \"uber-go\",\n\t\t\t\t\t\t\tFeatureVersion: \"1.0.0\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tCompletedType: types.QueryTaskCompletedTypeCompleted.Ptr(),\n\t\t\t\t\t\tQueryResult:   []byte(\"some result\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockVersionChecker *client.MockVersionChecker) {\n\t\t\t\tmockVersionChecker.EXPECT().SupportsConsistentQuery(\"uber-go\", \"1.0.0\").Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t\twant: &types.MatchingQueryWorkflowResponse{\n\t\t\t\tQueryResult: []byte(\"some result\"),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"query task failed\",\n\t\t\tresult: &queryResult{\n\t\t\t\tworkerResponse: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\t\t\tWorkerVersionInfo: &types.WorkerVersionInfo{\n\t\t\t\t\t\t\tImpl:           \"uber-go\",\n\t\t\t\t\t\t\tFeatureVersion: \"1.0.0\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tCompletedType: types.QueryTaskCompletedTypeFailed.Ptr(),\n\t\t\t\t\t\tErrorMessage:  \"query failed\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockVersionChecker *client.MockVersionChecker) {\n\t\t\t\tmockVersionChecker.EXPECT().SupportsConsistentQuery(\"uber-go\", \"1.0.0\").Return(nil)\n\t\t\t},\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar e *types.QueryFailedError\n\t\t\t\tassert.ErrorAs(t, err, &e)\n\t\t\t\tassert.Equal(t, \"query failed\", e.Message)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"unknown query result\",\n\t\t\tresult: &queryResult{\n\t\t\t\tworkerResponse: &types.MatchingRespondQueryTaskCompletedRequest{\n\t\t\t\t\tCompletedRequest: &types.RespondQueryTaskCompletedRequest{\n\t\t\t\t\t\tWorkerVersionInfo: &types.WorkerVersionInfo{\n\t\t\t\t\t\t\tImpl:           \"uber-go\",\n\t\t\t\t\t\t\tFeatureVersion: \"1.0.0\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tCompletedType: types.QueryTaskCompletedType(100).Ptr(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockSetup: func(mockVersionChecker *client.MockVersionChecker) {\n\t\t\t\tmockVersionChecker.EXPECT().SupportsConsistentQuery(\"uber-go\", \"1.0.0\").Return(nil)\n\t\t\t},\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar e *types.InternalServiceError\n\t\t\t\tassert.ErrorAs(t, err, &e)\n\t\t\t\tassert.Equal(t, \"unknown query completed type\", e.Message)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockVersionChecker := client.NewMockVersionChecker(mockCtrl)\n\t\t\ttc.mockSetup(mockVersionChecker)\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\tversionChecker: mockVersionChecker,\n\t\t\t}\n\t\t\thCtx := &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t}\n\t\t\tch := make(chan *queryResult, 1)\n\t\t\tch <- tc.result\n\t\t\tresp, err := engine.waitForQueryResult(hCtx, true, ch)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestIsShuttingDown(t *testing.T) {\n\twg := sync.WaitGroup{}\n\twg.Add(0)\n\tmockCtrl := gomock.NewController(t)\n\tmockDomainCache := cache.NewMockDomainCache(gomock.NewController(t))\n\tmockDomainCache.EXPECT().RegisterDomainChangeCallback(service.Matching, gomock.Any(), gomock.Any(), gomock.Any()).Times(1)\n\tmockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).Times(1)\n\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\tmockExecutor.EXPECT().Start(gomock.Any())\n\tmockExecutor.EXPECT().Stop()\n\te := matchingEngineImpl{\n\t\tdomainCache:        mockDomainCache,\n\t\tshutdownCompletion: &wg,\n\t\tshutdown:           make(chan struct{}),\n\t\texecutor:           mockExecutor,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t}\n\te.Start()\n\tassert.False(t, e.isShuttingDown())\n\te.Stop()\n\tassert.True(t, e.isShuttingDown())\n}\n\nfunc TestGetTasklistsNotOwned(t *testing.T) {\n\n\tctrl := gomock.NewController(t)\n\tresolver := membership.NewMockResolver(ctrl)\n\n\tresolver.EXPECT().WhoAmI().Return(membership.NewDetailedHostInfo(\"self\", \"host123\", nil), nil)\n\n\ttl1 := mustNewIdentifier(t, \"\", \"tl1\", 0)\n\ttl2 := mustNewIdentifier(t, \"\", \"tl2\", 0)\n\ttl3 := mustNewIdentifier(t, \"\", \"tl3\", 0)\n\n\ttl1m := newMockManagerWithTaskListID(ctrl, tl1)\n\ttl2m := newMockManagerWithTaskListID(ctrl, tl2)\n\ttl3m := newMockManagerWithTaskListID(ctrl, tl3)\n\n\tresolver.EXPECT().Lookup(service.Matching, tl1.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host123\", nil), nil)\n\tresolver.EXPECT().Lookup(service.Matching, tl2.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host456\", nil), nil)\n\tresolver.EXPECT().Lookup(service.Matching, tl3.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host123\", nil), nil)\n\n\te := matchingEngineImpl{\n\t\tshutdown:           make(chan struct{}),\n\t\tmembershipResolver: resolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig: &config.Config{\n\t\t\tEnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t},\n\t\tlogger: log.NewNoop(),\n\t}\n\te.taskListRegistry.Register(*tl1, tl1m)\n\te.taskListRegistry.Register(*tl2, tl2m)\n\te.taskListRegistry.Register(*tl3, tl3m)\n\n\ttls, err := e.getNonOwnedTasklistsLocked()\n\tassert.NoError(t, err)\n\n\tassert.Equal(t, []tasklist.Manager{tl2m}, tls)\n}\n\nfunc TestShutDownTasklistsNotOwned(t *testing.T) {\n\n\tctrl := gomock.NewController(t)\n\tresolver := membership.NewMockResolver(ctrl)\n\n\tresolver.EXPECT().WhoAmI().Return(membership.NewDetailedHostInfo(\"self\", \"host123\", nil), nil)\n\n\ttl1 := mustNewIdentifier(t, \"\", \"tl1\", 0)\n\ttl2 := mustNewIdentifier(t, \"\", \"tl2\", 0)\n\ttl3 := mustNewIdentifier(t, \"\", \"tl3\", 0)\n\n\ttl1m := newMockManagerWithTaskListID(ctrl, tl1)\n\ttl2m := newMockManagerWithTaskListID(ctrl, tl2)\n\ttl3m := newMockManagerWithTaskListID(ctrl, tl3)\n\n\tresolver.EXPECT().Lookup(service.Matching, tl1.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host123\", nil), nil)\n\tresolver.EXPECT().Lookup(service.Matching, tl2.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host456\", nil), nil)\n\tresolver.EXPECT().Lookup(service.Matching, tl3.GetName()).Return(membership.NewDetailedHostInfo(\"\", \"host123\", nil), nil)\n\n\te := matchingEngineImpl{\n\t\tshutdown:           make(chan struct{}),\n\t\tmembershipResolver: resolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig: &config.Config{\n\t\t\tEnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t},\n\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\tlogger:        log.NewNoop(),\n\t}\n\te.taskListRegistry.Register(*tl1, tl1m)\n\te.taskListRegistry.Register(*tl2, tl2m)\n\te.taskListRegistry.Register(*tl3, tl3m)\n\n\twg := sync.WaitGroup{}\n\n\twg.Add(1)\n\n\ttl2m.EXPECT().String().AnyTimes()\n\n\ttl2m.EXPECT().Stop().Do(func() {\n\t\twg.Done()\n\t})\n\n\terr := e.shutDownNonOwnedTasklists()\n\twg.Wait()\n\n\tassert.NoError(t, err)\n}\n\nfunc TestUpdateTaskListPartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                 string\n\t\treq                  *types.MatchingUpdateTaskListPartitionConfigRequest\n\t\tenableAdaptiveScaler bool\n\t\thCtx                 *handlerContext\n\t\tmockSetup            func(*tasklist.MockManager, *gomock.Controller, *executorclient.MockExecutor[tasklist.ShardProcessor])\n\t\texpectError          bool\n\t\texpectedError        string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(ctrl), nil)\n\t\t\t\tmockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"tasklist manager error\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\texecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(ctrl), nil)\n\t\t\t\tmockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(errors.New(\"tasklist manager error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"tasklist manager error\",\n\t\t},\n\t\t{\n\t\t\tname: \"non root partition error\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/test-tasklist/1\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only root partition's partition config can be updated.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid tasklist name\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"invalid partitioned task list name /__cadence_sys/test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname: \"nil partition config\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Task list partition config is not set in the request.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid tasklist kind\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only normal tasklist's partition config can be updated.\",\n\t\t},\n\t\t{\n\t\t\tname: \"manual update not allowed\",\n\t\t\treq: &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\tenableAdaptiveScaler: true,\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, executor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Manual update is not allowed because adaptive scaler is enabled.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil)\n\t\t\ttasklistID := mustNewIdentifier(t, \"test-domain-id\", \"test-tasklist\", 1)\n\t\t\tmockManager := newMockManagerWithTaskListID(mockCtrl, tasklistID)\n\t\t\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\t\t\ttc.mockSetup(mockManager, mockCtrl, mockExecutor)\n\t\t\ttaskListRegistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\ttaskListRegistry: taskListRegistry,\n\t\t\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\t\t\tdomainCache:      mockDomainCache,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tEnableAdaptiveScaler:                       dynamicproperties.GetBoolPropertyFilteredByTaskListInfo(tc.enableAdaptiveScaler),\n\t\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return false },\n\t\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t},\n\t\t\t\texecutor: mockExecutor,\n\t\t\t}\n\t\t\ttaskListRegistry.Register(*tasklistID, mockManager)\n\t\t\t_, err := engine.UpdateTaskListPartitionConfig(tc.hCtx, tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshTaskListPartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\treq           *types.MatchingRefreshTaskListPartitionConfigRequest\n\t\thCtx          *handlerContext\n\t\tmockSetup     func(*tasklist.MockManager, *gomock.Controller, *executorclient.MockExecutor[tasklist.ShardProcessor])\n\t\texpectError   bool\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\treq: &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/test-tasklist/1\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, mockCtrl *gomock.Controller, mockExecutor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\tmockExecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"tasklist manager error\",\n\t\t\treq: &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/test-tasklist/1\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, mockCtrl *gomock.Controller, mockExecutor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t\tmockExecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil)\n\t\t\t\tmockManager.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(errors.New(\"tasklist manager error\"))\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"tasklist manager error\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid tasklist name\",\n\t\t\treq: &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"/__cadence_sys/test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, mockExecutor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"invalid partitioned task list name /__cadence_sys/test-tasklist\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid tasklist kind\",\n\t\t\treq: &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t\tKind: types.TaskListKindSticky.Ptr(),\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, mockCtrl *gomock.Controller, mockExecutor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"Only normal tasklist's partition config can be updated.\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid request for root partition\",\n\t\t\treq: &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID: \"test-domain-id\",\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: \"test-tasklist\",\n\t\t\t\t},\n\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\thCtx: &handlerContext{\n\t\t\t\tContext: context.Background(),\n\t\t\t},\n\t\t\tmockSetup: func(mockManager *tasklist.MockManager, ctrl *gomock.Controller, mockExecutor *executorclient.MockExecutor[tasklist.ShardProcessor]) {\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"PartitionConfig must be nil for root partition.\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\ttasklistID := mustNewIdentifier(t, \"test-domain-id\", \"test-tasklist\", 1)\n\t\t\ttasklistID2 := mustNewIdentifier(t, \"test-domain-id\", \"/__cadence_sys/test-tasklist/1\", 1)\n\t\t\tmockManager := newMockManagerWithTaskListID(mockCtrl, tasklistID)\n\t\t\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\t\t\ttc.mockSetup(mockManager, mockCtrl, mockExecutor)\n\t\t\ttaskListRegistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\ttaskListRegistry: taskListRegistry,\n\t\t\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\t\t\texecutor:         mockExecutor,\n\t\t\t\tconfig: &config.Config{\n\t\t\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return false },\n\t\t\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t\t\t},\n\t\t\t}\n\t\t\ttaskListRegistry.Register(*tasklistID, mockManager)\n\t\t\ttaskListRegistry.Register(*tasklistID2, mockManager)\n\t\t\t_, err := engine.RefreshTaskListPartitionConfig(tc.hCtx, tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_domainChangeCallback(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tmockDomainCache := cache.NewMockDomainCache(mockCtrl)\n\n\tclusters := []string{\"cluster0\", \"cluster1\"}\n\n\ttlGlobalDecision1 := mustNewIdentifier(t, \"global-domain-1-id\", \"global-domain-1\", persistence.TaskListTypeDecision)\n\ttlGlobalActivity1 := mustNewIdentifier(t, \"global-domain-1-id\", \"global-domain-1\", persistence.TaskListTypeActivity)\n\ttlGlobalDecision2 := mustNewIdentifier(t, \"global-domain-2-id\", \"global-domain-2\", persistence.TaskListTypeDecision)\n\ttlGlobalActivity2 := mustNewIdentifier(t, \"global-domain-2-id\", \"global-domain-2\", persistence.TaskListTypeActivity)\n\ttlGlobalSticky2 := mustNewIdentifier(t, \"global-domain-2-id\", \"sticky-global-domain-2\", persistence.TaskListTypeDecision)\n\ttlGlobalActivity3 := mustNewIdentifier(t, \"global-domain-3-id\", \"global-domain-3\", persistence.TaskListTypeActivity)\n\ttlGlobalDecision3 := mustNewIdentifier(t, \"global-domain-3-id\", \"global-domain-3\", persistence.TaskListTypeDecision)\n\ttlGlobalSticky3 := mustNewIdentifier(t, \"global-domain-3-id\", \"sticky-global-domain-3\", persistence.TaskListTypeDecision)\n\ttlLocalDecision1 := mustNewIdentifier(t, \"local-domain-1-id\", \"local-domain-1\", persistence.TaskListTypeDecision)\n\ttlLocalActivity1 := mustNewIdentifier(t, \"local-domain-1-id\", \"local-domain-1\", persistence.TaskListTypeActivity)\n\ttlActiveActiveDecision1 := mustNewIdentifier(t, \"active-active-domain-1-id\", \"active-active-domain-1\", persistence.TaskListTypeDecision)\n\ttlActiveActiveActivity1 := mustNewIdentifier(t, \"active-active-domain-1-id\", \"active-active-domain-1\", persistence.TaskListTypeActivity)\n\n\tnewNormalManager := func(id *tasklist.Identifier) *tasklist.MockManager {\n\t\tmgr := newMockManagerWithTaskListID(mockCtrl, id)\n\t\tmgr.EXPECT().GetTaskListKind().Return(types.TaskListKindNormal).AnyTimes()\n\t\tmgr.EXPECT().DescribeTaskList(gomock.Any()).Return(&types.DescribeTaskListResponse{}).AnyTimes()\n\t\treturn mgr\n\t}\n\tnewStickyManager := func(id *tasklist.Identifier) *tasklist.MockManager {\n\t\tmgr := newMockManagerWithTaskListID(mockCtrl, id)\n\t\tmgr.EXPECT().GetTaskListKind().Return(types.TaskListKindSticky).AnyTimes()\n\t\tmgr.EXPECT().DescribeTaskList(gomock.Any()).Return(&types.DescribeTaskListResponse{}).AnyTimes()\n\t\treturn mgr\n\t}\n\n\tmockGlobalDecision1 := newNormalManager(tlGlobalDecision1)\n\tmockGlobalActivity1 := newNormalManager(tlGlobalActivity1)\n\tmockGlobalDecision2 := newNormalManager(tlGlobalDecision2)\n\tmockGlobalActivity2 := newNormalManager(tlGlobalActivity2)\n\tmockGlobalSticky2 := newStickyManager(tlGlobalSticky2)\n\tmockGlobalDecision3 := newNormalManager(tlGlobalDecision3)\n\tmockGlobalActivity3 := newNormalManager(tlGlobalActivity3)\n\tmockGlobalSticky3 := newStickyManager(tlGlobalSticky3)\n\tmockLocalDecision1 := newNormalManager(tlLocalDecision1)\n\tmockLocalActivity1 := newNormalManager(tlLocalActivity1)\n\tmockActiveActiveDecision1 := newNormalManager(tlActiveActiveDecision1)\n\tmockActiveActiveActivity1 := newNormalManager(tlActiveActiveActivity1)\n\n\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](mockCtrl)\n\tmockExecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(tasklist.NewMockShardProcessor(mockCtrl), nil).AnyTimes()\n\n\tengine := &matchingEngineImpl{\n\t\tdomainCache:                 mockDomainCache,\n\t\tfailoverNotificationVersion: 1,\n\t\tconfig:                      defaultTestConfig(),\n\t\tlogger:                      log.NewNoop(),\n\t\ttaskListRegistry:            tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\texecutor:                    mockExecutor,\n\t}\n\tengine.taskListRegistry.Register(*tlGlobalDecision1, mockGlobalDecision1)\n\tengine.taskListRegistry.Register(*tlGlobalActivity1, mockGlobalActivity1)\n\tengine.taskListRegistry.Register(*tlGlobalDecision2, mockGlobalDecision2)\n\tengine.taskListRegistry.Register(*tlGlobalActivity2, mockGlobalActivity2)\n\tengine.taskListRegistry.Register(*tlGlobalSticky2, mockGlobalSticky2)\n\tengine.taskListRegistry.Register(*tlGlobalDecision3, mockGlobalDecision3)\n\tengine.taskListRegistry.Register(*tlGlobalActivity3, mockGlobalActivity3)\n\tengine.taskListRegistry.Register(*tlGlobalSticky3, mockGlobalSticky3)\n\tengine.taskListRegistry.Register(*tlLocalDecision1, mockLocalDecision1)\n\tengine.taskListRegistry.Register(*tlLocalActivity1, mockLocalActivity1)\n\tengine.taskListRegistry.Register(*tlActiveActiveDecision1, mockActiveActiveDecision1)\n\tengine.taskListRegistry.Register(*tlActiveActiveActivity1, mockActiveActiveActivity1)\n\n\t// Eligible for failover handling is defined by isDomainEligibleToDisconnectPollers.\n\tmockGlobalDecision1.EXPECT().ReleaseBlockedPollers().Times(0)       // global-domain-1 has failover version 0 (<= current 1), so not eligible.\n\tmockGlobalActivity1.EXPECT().ReleaseBlockedPollers().Times(0)       // global-domain-1 has failover version 0 (<= current 1), so not eligible.\n\tmockGlobalDecision2.EXPECT().ReleaseBlockedPollers().Times(1)       // global-domain-2 is global, non-active-active, and version 4 > 1.\n\tmockGlobalActivity2.EXPECT().ReleaseBlockedPollers().Times(1)       // global-domain-2 is global, non-active-active, and version 4 > 1.\n\tmockGlobalSticky2.EXPECT().ReleaseBlockedPollers().Times(1)         // sticky task list under eligible global-domain-2.\n\tmockGlobalDecision3.EXPECT().ReleaseBlockedPollers().Times(1)       // global-domain-3 is eligible.\n\tmockGlobalActivity3.EXPECT().ReleaseBlockedPollers().Times(1)       // global-domain-3 is eligible.\n\tmockGlobalSticky3.EXPECT().ReleaseBlockedPollers().Times(1)         // sticky task list under eligible global-domain-3.\n\tmockLocalDecision1.EXPECT().ReleaseBlockedPollers().Times(0)        // local domains are not eligible.\n\tmockLocalActivity1.EXPECT().ReleaseBlockedPollers().Times(0)        // local domains are not eligible.\n\tmockActiveActiveDecision1.EXPECT().ReleaseBlockedPollers().Times(0) // active-active domains are not eligible.\n\tmockActiveActiveActivity1.EXPECT().ReleaseBlockedPollers().Times(0) // active-active domains are not eligible.\n\n\tdomains := []*cache.DomainCacheEntry{\n\t\tcache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"global-domain-1\", ID: \"global-domain-1-id\"},\n\t\t\tnil,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: clusters[0], Clusters: []*persistence.ClusterReplicationConfig{{ClusterName: \"cluster0\"}, {ClusterName: \"cluster1\"}}},\n\t\t\t0,\n\t\t\tnil,\n\t\t\t0,\n\t\t\t0,\n\t\t\t6,\n\t\t),\n\t\tcache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"global-domain-2\", ID: \"global-domain-2-id\"},\n\t\t\tnil,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: clusters[1], Clusters: []*persistence.ClusterReplicationConfig{{ClusterName: \"cluster0\"}, {ClusterName: \"cluster1\"}}},\n\t\t\t0,\n\t\t\tnil,\n\t\t\t4,\n\t\t\t0,\n\t\t\t4,\n\t\t),\n\t\tcache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"global-domain-3\", ID: \"global-domain-3-id\"},\n\t\t\tnil,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: clusters[1], Clusters: []*persistence.ClusterReplicationConfig{{ClusterName: \"cluster0\"}, {ClusterName: \"cluster1\"}}},\n\t\t\t0,\n\t\t\tnil,\n\t\t\t5,\n\t\t\t0,\n\t\t\t5,\n\t\t),\n\t\tcache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"local-domain-1\", ID: \"local-domain-1-id\"},\n\t\t\tnil,\n\t\t\tfalse,\n\t\t\tnil,\n\t\t\t0,\n\t\t\tnil,\n\t\t\t0,\n\t\t\t0,\n\t\t\t3,\n\t\t),\n\t\tcache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"active-active-domain-1\", ID: \"active-active-domain-1-id\"},\n\t\t\tnil,\n\t\t\ttrue,\n\t\t\t&persistence.DomainReplicationConfig{ActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {\n\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\"us-west\": {\n\t\t\t\t\t\t\t\tActiveClusterName: \"cluster0\",\n\t\t\t\t\t\t\t\tFailoverVersion:   1,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}},\n\t\t\t0,\n\t\t\tnil,\n\t\t\t0,\n\t\t\t0,\n\t\t\t2,\n\t\t),\n\t}\n\n\tengine.domainChangeCallback(domains)\n\n\tassert.Equal(t, int64(5), engine.failoverNotificationVersion, \"5 is the highest failover notification version in the fixtures (global-domain-3)\")\n}\n\nfunc Test_registerDomainFailoverCallback(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t// Capture the registered catchUpFn\n\tvar registeredCatchUpFn func(cache.DomainCache, cache.PrepareCallbackFn, cache.CallbackFn)\n\tmockDomainCache.EXPECT().RegisterDomainChangeCallback(\n\t\tservice.Matching, // id of the callback\n\t\tgomock.Any(),     // catchUpFn\n\t\tgomock.Any(),     // lockTaskProcessingForDomainUpdate\n\t\tgomock.Any(),     // domainChangeCB\n\t).Do(func(_ string, catchUpFn, _, _ interface{}) {\n\t\tif fn, ok := catchUpFn.(cache.CatchUpFn); ok {\n\t\t\tregisteredCatchUpFn = fn\n\t\t} else {\n\t\t\tt.Fatalf(\"Failed to convert catchUpFn to cache.CatchUpFn: got type %T\", catchUpFn)\n\t\t}\n\t}).Times(1)\n\n\tengine := &matchingEngineImpl{\n\t\tdomainCache:                 mockDomainCache,\n\t\tfailoverNotificationVersion: 0,\n\t\tconfig:                      defaultTestConfig(),\n\t\tlogger:                      log.NewNoop(),\n\t\ttaskListRegistry:            tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t}\n\n\tengine.registerDomainFailoverCallback()\n\n\tt.Run(\"catchUpFn - No failoverNotificationVersion updates\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\t\"uuid-domain1\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain1\", Name: \"domain1\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t1,\n\t\t\t),\n\t\t\t\"uuid-domain2\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain2\", Name: \"domain2\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t0,\n\t\t\t\t0,\n\t\t\t\t4,\n\t\t\t),\n\t\t})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func([]*cache.DomainCacheEntry) { callbackCalled = true }\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.False(t, prepareCalled, \"prepareCallback should not be called\")\n\t\t\tassert.False(t, callbackCalled, \"callback should not be called\")\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\n\t\tassert.Equal(t, int64(0), engine.failoverNotificationVersion)\n\t})\n\n\tt.Run(\"catchUpFn - No failoverNotificationVersion updates\", func(t *testing.T) {\n\t\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\t\"uuid-domain1\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain1\", Name: \"domain1\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t3,\n\t\t\t\t0,\n\t\t\t\t3,\n\t\t\t),\n\t\t\t\"uuid-domain2\": cache.NewDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: \"uuid-domain2\", Name: \"domain2\"},\n\t\t\t\tnil,\n\t\t\t\ttrue,\n\t\t\t\t&persistence.DomainReplicationConfig{ActiveClusterName: \"A\"},\n\t\t\t\t0,\n\t\t\t\tnil,\n\t\t\t\t2,\n\t\t\t\t0,\n\t\t\t\t4,\n\t\t\t),\n\t\t})\n\n\t\tprepareCalled := false\n\t\tcallbackCalled := false\n\t\tprepare := func() { prepareCalled = true }\n\t\tcallback := func([]*cache.DomainCacheEntry) { callbackCalled = true }\n\n\t\tif registeredCatchUpFn != nil {\n\t\t\tregisteredCatchUpFn(mockDomainCache, prepare, callback)\n\t\t\tassert.False(t, prepareCalled, \"prepareCallback should not be called\")\n\t\t\tassert.False(t, callbackCalled, \"callback should not be called\")\n\t\t} else {\n\t\t\tassert.Fail(t, \"catchUpFn was not registered\")\n\t\t}\n\n\t\tassert.Equal(t, int64(3), engine.failoverNotificationVersion)\n\t})\n\n}\n\nfunc TestRefreshWorkflowTasks(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tctx               context.Context\n\t\tdomainID          string\n\t\tworkflowExecution *types.WorkflowExecution\n\t\tmockSetup         func(*history.MockClient)\n\t\twantErr           bool\n\t\tassertErr         func(*testing.T, error)\n\t}{\n\t\t{\n\t\t\tname:     \"success\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\t\tDomainUIID: \"test-domain-id\",\n\t\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"entity not exists error - returns nil\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&types.EntityNotExistsError{Message: \"workflow not found\"})\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"internal service error\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&types.InternalServiceError{Message: \"internal error\"})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar serviceErr *types.InternalServiceError\n\t\t\t\tassert.ErrorAs(t, err, &serviceErr)\n\t\t\t\tassert.Equal(t, \"internal error\", serviceErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"generic error propagated\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(errors.New(\"some error\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, \"some error\", err.Error())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"workflow execution already completed error\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&types.WorkflowExecutionAlreadyCompletedError{Message: \"workflow already completed\"})\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tvar completedErr *types.WorkflowExecutionAlreadyCompletedError\n\t\t\t\tassert.ErrorAs(t, err, &completedErr)\n\t\t\t\tassert.Equal(t, \"workflow already completed\", completedErr.Message)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"context canceled\",\n\t\t\tctx:      func() context.Context { ctx, cancel := context.WithCancel(context.Background()); cancel(); return ctx }(),\n\t\t\tdomainID: \"test-domain-id\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(context.Canceled)\n\t\t\t},\n\t\t\twantErr: true,\n\t\t\tassertErr: func(t *testing.T, err error) {\n\t\t\t\tassert.Equal(t, context.Canceled, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:     \"empty domain id\",\n\t\t\tctx:      context.Background(),\n\t\t\tdomainID: \"\",\n\t\t\tworkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\t\tDomainUIID: \"\",\n\t\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t\t{\n\t\t\tname:              \"nil workflow execution\",\n\t\t\tctx:               context.Background(),\n\t\t\tdomainID:          \"test-domain-id\",\n\t\t\tworkflowExecution: nil,\n\t\t\tmockSetup: func(mockHistoryService *history.MockClient) {\n\t\t\t\tmockHistoryService.EXPECT().RefreshWorkflowTasks(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&types.HistoryRefreshWorkflowTasksRequest{\n\t\t\t\t\t\tDomainUIID: \"test-domain-id\",\n\t\t\t\t\t\tRequest: &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\t\t\tExecution: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t).Return(nil)\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockHistoryService := history.NewMockClient(mockCtrl)\n\t\t\ttc.mockSetup(mockHistoryService)\n\n\t\t\tengine := &matchingEngineImpl{\n\t\t\t\thistoryService: mockHistoryService,\n\t\t\t}\n\n\t\t\terr := engine.refreshWorkflowTasks(tc.ctx, tc.domainID, tc.workflowExecution)\n\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tif tc.assertErr != nil {\n\t\t\t\t\ttc.assertErr(t, err)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestSetupExecutorWithEmptyConfig(t *testing.T) {\n\t// When no shard-distributor namespaces are configured, setupExecutor must\n\t// succeed and install a no-op executor so the matching service falls back\n\t// to local hash-ring assignment.\n\tregistry := tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\tengine := &matchingEngineImpl{\n\t\ttaskListRegistry:               registry,\n\t\tlogger:                         log.NewNoop(),\n\t\ttimeSource:                     clock.NewRealTimeSource(),\n\t\tShardDistributorMatchingConfig: clientcommon.Config{}, // empty – no namespaces\n\t\tconfig:                         &config.Config{},\n\t}\n\n\t// Must not panic or fatal; executor must be set afterward.\n\tengine.setupExecutor(nil)\n\n\trequire.NotNil(t, engine.executor)\n\trequire.NotNil(t, engine.taskListsFactory)\n\n\t// The no-op executor reports itself as not onboarded to SD.\n\tassert.False(t, engine.executor.IsOnboardedToSD())\n}\n"
  },
  {
    "path": "service/matching/handler/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\tdynamicquotas \"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\ntype (\n\t// handlerImpl is an implementation for matching service independent of wire protocol\n\thandlerImpl struct {\n\t\tengine            Engine\n\t\tmetricsClient     metrics.Client\n\t\tstartWG           sync.WaitGroup\n\t\tuserRateLimiter   quotas.Policy\n\t\tworkerRateLimiter quotas.Policy\n\t\tlogger            log.Logger\n\t\tthrottledLogger   log.Logger\n\t\tdomainCache       cache.DomainCache\n\t}\n)\n\nvar (\n\terrMatchingHostThrottle = &types.ServiceBusyError{Message: \"Matching host rps exceeded\"}\n)\n\n// NewHandler creates a thrift handler for the matching service\nfunc NewHandler(\n\tengine Engine,\n\tconfig *config.Config,\n\tdomainCache cache.DomainCache,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tthrottledLogger log.Logger,\n) Handler {\n\thandler := &handlerImpl{\n\t\tmetricsClient: metricsClient,\n\t\tuserRateLimiter: quotas.NewMultiStageRateLimiter(\n\t\t\tquotas.NewDynamicRateLimiter(config.UserRPS.AsFloat64()),\n\t\t\tquotas.NewCollection(dynamicquotas.NewFallbackDynamicRateLimiterFactory(\n\t\t\t\tconfig.DomainUserRPS,\n\t\t\t\tconfig.UserRPS,\n\t\t\t)),\n\t\t),\n\t\tworkerRateLimiter: quotas.NewMultiStageRateLimiter(\n\t\t\tquotas.NewDynamicRateLimiter(config.WorkerRPS.AsFloat64()),\n\t\t\tquotas.NewCollection(dynamicquotas.NewFallbackDynamicRateLimiterFactory(\n\t\t\t\tconfig.DomainWorkerRPS,\n\t\t\t\tconfig.WorkerRPS,\n\t\t\t)),\n\t\t),\n\t\tengine:          engine,\n\t\tlogger:          logger,\n\t\tthrottledLogger: throttledLogger,\n\t\tdomainCache:     domainCache,\n\t}\n\t// prevent us from trying to serve requests before matching engine is started and ready\n\thandler.startWG.Add(1)\n\treturn handler\n}\n\n// Start starts the handler\nfunc (h *handlerImpl) Start() {\n\th.engine.Start()\n\th.startWG.Done()\n}\n\n// Stop stops the handler\nfunc (h *handlerImpl) Stop() {\n\th.engine.Stop()\n}\n\n// Health is for health check\nfunc (h *handlerImpl) Health(ctx context.Context) (*types.HealthStatus, error) {\n\th.startWG.Wait()\n\th.logger.Debug(\"Matching service health check endpoint reached.\")\n\ths := &types.HealthStatus{Ok: true, Msg: \"matching good\"}\n\treturn hs, nil\n}\n\nfunc (h *handlerImpl) newHandlerContext(\n\tctx context.Context,\n\tdomainName string,\n\ttaskList *types.TaskList,\n\tscope metrics.ScopeIdx,\n) *handlerContext {\n\treturn newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\ttaskList,\n\t\th.metricsClient,\n\t\tscope,\n\t\th.logger,\n\t)\n}\n\n// AddActivityTask - adds an activity task.\nfunc (h *handlerImpl) AddActivityTask(\n\tctx context.Context,\n\trequest *types.AddActivityTaskRequest,\n) (resp *types.AddActivityTaskResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetTaskList(),\n\t\tmetrics.MatchingAddActivityTaskScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif request.GetForwardedFrom() != \"\" {\n\t\thCtx.scope.IncCounter(metrics.ForwardedPerTaskListCounter)\n\t}\n\n\tif ok := h.workerRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresp, err := h.engine.AddActivityTask(hCtx, request)\n\treturn resp, hCtx.handleErr(err)\n}\n\n// AddDecisionTask - adds a decision task.\nfunc (h *handlerImpl) AddDecisionTask(\n\tctx context.Context,\n\trequest *types.AddDecisionTaskRequest,\n) (resp *types.AddDecisionTaskResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetTaskList(),\n\t\tmetrics.MatchingAddDecisionTaskScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif request.GetForwardedFrom() != \"\" {\n\t\thCtx.scope.IncCounter(metrics.ForwardedPerTaskListCounter)\n\t}\n\n\tif ok := h.workerRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresp, err := h.engine.AddDecisionTask(hCtx, request)\n\treturn resp, hCtx.handleErr(err)\n}\n\n// PollForActivityTask - long poll for an activity task.\nfunc (h *handlerImpl) PollForActivityTask(\n\tctx context.Context,\n\trequest *types.MatchingPollForActivityTaskRequest,\n) (resp *types.MatchingPollForActivityTaskResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetPollRequest().GetTaskList(),\n\t\tmetrics.MatchingPollForActivityTaskScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif request.GetForwardedFrom() != \"\" {\n\t\thCtx.scope.IncCounter(metrics.ForwardedPerTaskListCounter)\n\t}\n\n\tif ok := h.workerRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tif _, err := common.ValidateLongPollContextTimeoutIsSet(ctx,\n\t\t\"PollForActivityTask\",\n\t\th.throttledLogger,\n\t); err != nil {\n\t\treturn nil, hCtx.handleErr(err)\n\t}\n\n\tresponse, err := h.engine.PollForActivityTask(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\n// PollForDecisionTask - long poll for a decision task.\nfunc (h *handlerImpl) PollForDecisionTask(\n\tctx context.Context,\n\trequest *types.MatchingPollForDecisionTaskRequest,\n) (resp *types.MatchingPollForDecisionTaskResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetPollRequest().GetTaskList(),\n\t\tmetrics.MatchingPollForDecisionTaskScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif request.GetForwardedFrom() != \"\" {\n\t\thCtx.scope.IncCounter(metrics.ForwardedPerTaskListCounter)\n\t}\n\n\tif ok := h.workerRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tif _, err := common.ValidateLongPollContextTimeoutIsSet(\n\t\tctx,\n\t\t\"PollForDecisionTask\",\n\t\th.throttledLogger,\n\t); err != nil {\n\t\treturn nil, hCtx.handleErr(err)\n\t}\n\n\tresponse, err := h.engine.PollForDecisionTask(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\n// QueryWorkflow queries a given workflow synchronously and return the query result.\nfunc (h *handlerImpl) QueryWorkflow(\n\tctx context.Context,\n\trequest *types.MatchingQueryWorkflowRequest,\n) (resp *types.MatchingQueryWorkflowResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetTaskList(),\n\t\tmetrics.MatchingQueryWorkflowScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif request.GetForwardedFrom() != \"\" {\n\t\thCtx.scope.IncCounter(metrics.ForwardedPerTaskListCounter)\n\t}\n\n\tif ok := h.userRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresponse, err := h.engine.QueryWorkflow(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\n// RespondQueryTaskCompleted responds a query task completed\nfunc (h *handlerImpl) RespondQueryTaskCompleted(\n\tctx context.Context,\n\trequest *types.MatchingRespondQueryTaskCompletedRequest,\n) (retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetTaskList(),\n\t\tmetrics.MatchingRespondQueryTaskCompletedScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\t// Count the request in the RPS, but we still accept it even if RPS is exceeded\n\th.workerRateLimiter.Allow(quotas.Info{Domain: domainName})\n\n\terr := h.engine.RespondQueryTaskCompleted(hCtx, request)\n\treturn hCtx.handleErr(err)\n}\n\n// CancelOutstandingPoll is used to cancel outstanding pollers\nfunc (h *handlerImpl) CancelOutstandingPoll(ctx context.Context,\n\trequest *types.CancelOutstandingPollRequest) (retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetTaskList(),\n\t\tmetrics.MatchingCancelOutstandingPollScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\t// Count the request in the RPS, but we still accept it even if RPS is exceeded\n\th.workerRateLimiter.Allow(quotas.Info{Domain: domainName})\n\n\terr := h.engine.CancelOutstandingPoll(hCtx, request)\n\treturn hCtx.handleErr(err)\n}\n\n// DescribeTaskList returns information about the target tasklist, right now this API returns the\n// pollers which polled this tasklist in last few minutes. If includeTaskListStatus field is true,\n// it will also return status of tasklist's ackManager (readLevel, ackLevel, backlogCountHint and taskIDBlock).\nfunc (h *handlerImpl) DescribeTaskList(\n\tctx context.Context,\n\trequest *types.MatchingDescribeTaskListRequest,\n) (resp *types.DescribeTaskListResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.GetDomainUUID())\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.GetDescRequest().GetTaskList(),\n\t\tmetrics.MatchingDescribeTaskListScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif ok := h.userRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresponse, err := h.engine.DescribeTaskList(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\n// ListTaskListPartitions returns information about partitions for a taskList\nfunc (h *handlerImpl) ListTaskListPartitions(\n\tctx context.Context,\n\trequest *types.MatchingListTaskListPartitionsRequest,\n) (resp *types.ListTaskListPartitionsResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\thCtx := newHandlerContext(\n\t\tctx,\n\t\trequest.GetDomain(),\n\t\trequest.GetTaskList(),\n\t\th.metricsClient,\n\t\tmetrics.MatchingListTaskListPartitionsScope,\n\t\th.logger,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif ok := h.userRateLimiter.Allow(quotas.Info{Domain: request.GetDomain()}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresponse, err := h.engine.ListTaskListPartitions(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\n// GetTaskListsByDomain returns information about partitions for a taskList\nfunc (h *handlerImpl) GetTaskListsByDomain(\n\tctx context.Context,\n\trequest *types.GetTaskListsByDomainRequest,\n) (resp *types.GetTaskListsByDomainResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\thCtx := newHandlerContext(\n\t\tctx,\n\t\trequest.GetDomain(),\n\t\tnil,\n\t\th.metricsClient,\n\t\tmetrics.MatchingGetTaskListsByDomainScope,\n\t\th.logger,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif ok := h.userRateLimiter.Allow(quotas.Info{Domain: request.GetDomain()}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresponse, err := h.engine.GetTaskListsByDomain(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\nfunc (h *handlerImpl) UpdateTaskListPartitionConfig(\n\tctx context.Context,\n\trequest *types.MatchingUpdateTaskListPartitionConfigRequest,\n) (resp *types.MatchingUpdateTaskListPartitionConfigResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.DomainUUID)\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.TaskList,\n\t\tmetrics.MatchingUpdateTaskListPartitionConfigScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\tif ok := h.userRateLimiter.Allow(quotas.Info{Domain: domainName}); !ok {\n\t\treturn nil, hCtx.handleErr(errMatchingHostThrottle)\n\t}\n\n\tresponse, err := h.engine.UpdateTaskListPartitionConfig(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\nfunc (h *handlerImpl) RefreshTaskListPartitionConfig(\n\tctx context.Context,\n\trequest *types.MatchingRefreshTaskListPartitionConfigRequest,\n) (resp *types.MatchingRefreshTaskListPartitionConfigResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tdomainName := h.domainName(request.DomainUUID)\n\thCtx := h.newHandlerContext(\n\t\tctx,\n\t\tdomainName,\n\t\trequest.TaskList,\n\t\tmetrics.MatchingRefreshTaskListPartitionConfigScope,\n\t)\n\n\tsw := hCtx.startProfiling(&h.startWG)\n\tdefer sw.Stop()\n\n\t// Count the request in the RPS, but we still accept it even if RPS is exceeded\n\th.userRateLimiter.Allow(quotas.Info{Domain: domainName})\n\n\tresponse, err := h.engine.RefreshTaskListPartitionConfig(hCtx, request)\n\treturn response, hCtx.handleErr(err)\n}\n\nfunc (h *handlerImpl) domainName(id string) string {\n\tdomainName, err := h.domainCache.GetDomainName(id)\n\tif err != nil {\n\t\treturn \"\"\n\t}\n\treturn domainName\n}\n"
  },
  {
    "path": "service/matching/handler/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\tdynamicquotas \"github.com/uber/cadence/common/dynamicconfig/quotas\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nconst (\n\ttestDomain = \"test-domain\"\n)\n\ntype (\n\thandlerSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\tcontroller      *gomock.Controller\n\t\tmockResource    *resource.Test\n\t\tmockEngine      *MockEngine\n\t\tmockDomainCache *cache.MockDomainCache\n\t\tmockLimiter     *quotas.MockLimiter\n\t\thandler         *handlerImpl\n\n\t\ttestDomain string\n\t}\n)\n\nfunc TestHandlerSuite(t *testing.T) {\n\ts := new(handlerSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *handlerSuite) SetupSuite() {}\n\nfunc (s *handlerSuite) TearDownSuite() {}\n\nfunc (s *handlerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Matching)\n\ts.mockEngine = NewMockEngine(s.controller)\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.mockLimiter = quotas.NewMockLimiter(s.controller)\n\n\t// Create a handler with a mock limiter\n\ts.handler = &handlerImpl{\n\t\tengine:        s.mockEngine,\n\t\tmetricsClient: s.mockResource.MetricsClient,\n\t\tstartWG:       sync.WaitGroup{},\n\t\tuserRateLimiter: quotas.NewMultiStageRateLimiter(\n\t\t\ts.mockLimiter,\n\t\t\tquotas.NewCollection(dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int { return 10 })),\n\t\t),\n\t\tworkerRateLimiter: quotas.NewMultiStageRateLimiter(\n\t\t\ts.mockLimiter,\n\t\t\tquotas.NewCollection(dynamicquotas.NewSimpleDynamicRateLimiterFactory(func(domain string) int { return 10 })),\n\t\t),\n\t\tlogger:          s.mockResource.GetLogger(),\n\t\tthrottledLogger: s.mockResource.GetThrottledLogger(),\n\t\tdomainCache:     s.mockDomainCache,\n\t}\n\n\ts.testDomain = testDomain\n}\n\nfunc (s *handlerSuite) TearDownTest() {\n\ts.controller.Finish()\n\ts.mockResource.Finish(s.T())\n}\n\nfunc (s *handlerSuite) getHandler(config *config.Config) Handler {\n\treturn NewHandler(s.mockEngine, config, s.mockDomainCache, s.mockResource.MetricsClient, s.mockResource.GetLogger(), s.mockResource.GetThrottledLogger())\n}\n\nfunc (s *handlerSuite) TestNewHandler() {\n\tcfg := config.NewConfig(dynamicconfig.NewCollection(dynamicconfig.NewInMemoryClient(), s.mockResource.Logger), \"matching-test\", commonConfig.RPC{}, getIsolationGroupsHelper)\n\thandler := s.getHandler(cfg)\n\ts.NotNil(handler)\n}\n\nfunc (s *handlerSuite) TestStart() {\n\tdefer goleak.VerifyNone(s.T())\n\n\tcfg := config.NewConfig(dynamicconfig.NewCollection(dynamicconfig.NewInMemoryClient(), s.mockResource.Logger), \"matching-test\", commonConfig.RPC{}, getIsolationGroupsHelper)\n\thandler := s.getHandler(cfg)\n\n\ts.mockEngine.EXPECT().Start().Times(1)\n\n\thandler.Start()\n}\n\nfunc (s *handlerSuite) TestStop() {\n\tdefer goleak.VerifyNone(s.T())\n\n\tcfg := config.NewConfig(dynamicconfig.NewCollection(dynamicconfig.NewInMemoryClient(), s.mockResource.Logger), \"matching-test\", commonConfig.RPC{}, getIsolationGroupsHelper)\n\thandler := s.getHandler(cfg)\n\n\ts.mockEngine.EXPECT().Start().Times(1)\n\ts.mockEngine.EXPECT().Stop().Times(1)\n\n\thandler.Start()\n\thandler.Stop()\n}\n\nfunc (s *handlerSuite) TestHealth() {\n\tresp, err := s.handler.Health(context.Background())\n\n\ts.NoError(err)\n\ts.Equal(&types.HealthStatus{Ok: true, Msg: \"matching good\"}, resp)\n}\n\nfunc (s *handlerSuite) TestNewHandlerContext() {\n\thandlerCtx := s.handler.newHandlerContext(context.Background(), testDomain, &types.TaskList{Name: \"test-task-list\"}, 0)\n\n\ts.NotNil(handlerCtx)\n\ts.IsType(&handlerContext{}, handlerCtx)\n}\n\nfunc (s *handlerSuite) TestAddActivityTask() {\n\trequest := types.AddActivityTaskRequest{\n\t\tDomainUUID:    \"test-domain-id\",\n\t\tTaskList:      &types.TaskList{Name: \"test-task-list\"},\n\t\tForwardedFrom: \"forwarded-from\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\twant       *types.AddActivityTaskResponse\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().AddActivityTask(gomock.Any(), &request).Return(&types.AddActivityTaskResponse{\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.AddActivityTaskResponse{\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion:         1,\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - AddActivityTask failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1) // Ensure Allow() returns true\n\t\t\t\ts.mockEngine.EXPECT().AddActivityTask(gomock.Any(), &request).Return(nil, errors.New(\"add-activity-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"add-activity-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.AddActivityTask(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.Equal(tc.want, resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestAddDecisionTask() {\n\trequest := types.AddDecisionTaskRequest{\n\t\tDomainUUID:    \"test-domain-id\",\n\t\tTaskList:      &types.TaskList{Name: \"test-task-list\"},\n\t\tForwardedFrom: \"forwarded-from\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\twant       *types.AddDecisionTaskResponse\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().AddDecisionTask(gomock.Any(), &request).Return(&types.AddDecisionTaskResponse{\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.AddDecisionTaskResponse{\n\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion:         1,\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - AddDecisionTask failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1) // Ensure Allow() returns true\n\t\t\t\ts.mockEngine.EXPECT().AddDecisionTask(gomock.Any(), &request).Return(nil, errors.New(\"add-decision-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"add-decision-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.AddDecisionTask(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.Equal(tc.want, resp)\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestPollForActivityTask() {\n\trequest := types.MatchingPollForActivityTaskRequest{\n\t\tDomainUUID:    \"test-domain-id\",\n\t\tForwardedFrom: \"forwarded-from\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\tgetCtx     func() (context.Context, context.CancelFunc)\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollForActivityTask(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.MatchingPollForActivityTaskResponse{TaskToken: []byte(\"task-token\")}, nil).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) {\n\t\t\t\tctx, cancel := context.WithDeadline(context.Background(), time.Now())\n\t\t\t\treturn ctx, cancel\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) { return context.Background(), nil },\n\t\t\terr:    &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - LongPollContextTimeout not set\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) { return context.Background(), nil },\n\t\t\terr:    common.ErrContextTimeoutNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - PollForActivityTask failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollForActivityTask(gomock.Any(), &request).Return(nil, errors.New(\"poll-activity-error\")).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) {\n\t\t\t\tctx, cancel := context.WithDeadline(context.Background(), time.Now())\n\t\t\t\treturn ctx, cancel\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"poll-activity-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tctx, cancel := tc.getCtx()\n\t\t\tif cancel != nil {\n\t\t\t\tdefer cancel()\n\t\t\t}\n\n\t\t\tresp, err := s.handler.PollForActivityTask(ctx, &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.MatchingPollForActivityTaskResponse{TaskToken: []byte(\"task-token\")}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestPollForDecisionTask() {\n\trequest := types.MatchingPollForDecisionTaskRequest{\n\t\tDomainUUID:    \"test-domain-id\",\n\t\tForwardedFrom: \"forwarded-from\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\tgetCtx     func() (context.Context, context.CancelFunc)\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollForDecisionTask(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.MatchingPollForDecisionTaskResponse{TaskToken: []byte(\"task-token\")}, nil).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) {\n\t\t\t\tctx, cancel := context.WithDeadline(context.Background(), time.Now())\n\t\t\t\treturn ctx, cancel\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) { return context.Background(), nil },\n\t\t\terr:    &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - LongPollContextTimeout not set\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) { return context.Background(), nil },\n\t\t\terr:    common.ErrContextTimeoutNotSet,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - PollForDecisionTask failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().PollForDecisionTask(gomock.Any(), &request).Return(nil, errors.New(\"poll-decision-error\")).Times(1)\n\t\t\t},\n\t\t\tgetCtx: func() (context.Context, context.CancelFunc) {\n\t\t\t\tctx, cancel := context.WithDeadline(context.Background(), time.Now())\n\t\t\t\treturn ctx, cancel\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"poll-decision-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tctx, cancel := tc.getCtx()\n\t\t\tif cancel != nil {\n\t\t\t\tdefer cancel()\n\t\t\t}\n\n\t\t\tresp, err := s.handler.PollForDecisionTask(ctx, &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.MatchingPollForDecisionTaskResponse{TaskToken: []byte(\"task-token\")}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestQueryWorkflow() {\n\trequest := types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID:    \"test-domain-id\",\n\t\tForwardedFrom: \"forwarded-from\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().QueryWorkflow(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.MatchingQueryWorkflowResponse{QueryResult: []byte(\"query-result\")}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - QueryWorkflow failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().QueryWorkflow(gomock.Any(), &request).Return(nil, errors.New(\"query-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"query-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.QueryWorkflow(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.MatchingQueryWorkflowResponse{QueryResult: []byte(\"query-result\")}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestRespondQueryTaskCompleted() {\n\trequest := types.MatchingRespondQueryTaskCompletedRequest{\n\t\tDomainUUID: \"test-domain-id\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockEngine.EXPECT().RespondQueryTaskCompleted(gomock.Any(), &request).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - RespondQueryTaskCompleted failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockEngine.EXPECT().RespondQueryTaskCompleted(gomock.Any(), &request).Return(errors.New(\"respond-query-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"respond-query-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\n\t\t\terr := s.handler.RespondQueryTaskCompleted(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestCancelOutstandingPoll() {\n\trequest := types.CancelOutstandingPollRequest{\n\t\tDomainUUID: \"test-domain-id\",\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockEngine.EXPECT().CancelOutstandingPoll(gomock.Any(), &request).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - CancelOutstandingPoll failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockEngine.EXPECT().CancelOutstandingPoll(gomock.Any(), &request).Return(errors.New(\"cancel-poll-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"cancel-poll-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\n\t\t\terr := s.handler.CancelOutstandingPoll(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestDescribeTaskList() {\n\trequest := types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID: \"test-domain-id\",\n\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\tTaskList: &types.TaskList{Name: \"test-task-list\"},\n\t\t},\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeTaskList(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{{}}}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - DescribeTaskList failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().DescribeTaskList(gomock.Any(), &request).\n\t\t\t\t\tReturn(nil, errors.New(\"describe-tasklist-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"describe-tasklist-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.DescribeTaskList(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{{}}}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestListTaskListPartitions() {\n\trequest := types.MatchingListTaskListPartitionsRequest{\n\t\tTaskList: &types.TaskList{Name: \"test-task-list\"},\n\t\tDomain:   s.testDomain,\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ListTaskListPartitions(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.ListTaskListPartitionsResponse{ActivityTaskListPartitions: []*types.TaskListPartitionMetadata{\n\t\t\t\t\t\t{Key: \"test-key\", OwnerHostName: \"test-host\"}}}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - ListTaskListPartitions failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().ListTaskListPartitions(gomock.Any(), &request).\n\t\t\t\t\tReturn(nil, errors.New(\"list-tasklist-partitions-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"list-tasklist-partitions-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\tresp, err := s.handler.ListTaskListPartitions(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.ListTaskListPartitionsResponse{ActivityTaskListPartitions: []*types.TaskListPartitionMetadata{\n\t\t\t\t\t{Key: \"test-key\", OwnerHostName: \"test-host\"}}}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestGetTaskListsByDomain() {\n\trequest := types.GetTaskListsByDomainRequest{\n\t\tDomain: s.testDomain,\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetTaskListsByDomain(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.GetTaskListsByDomainResponse{\n\t\t\t\t\t\tDecisionTaskListMap: map[string]*types.DescribeTaskListResponse{\"test-decision-task-list\": {}},\n\t\t\t\t\t\tActivityTaskListMap: map[string]*types.DescribeTaskListResponse{\"test-activity-task-list\": {}},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetTaskListsByDomain failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().GetTaskListsByDomain(gomock.Any(), &request).\n\t\t\t\t\tReturn(nil, errors.New(\"get-tasklists-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"get-tasklists-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\tresp, err := s.handler.GetTaskListsByDomain(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(&types.GetTaskListsByDomainResponse{\n\t\t\t\t\tDecisionTaskListMap: map[string]*types.DescribeTaskListResponse{\"test-decision-task-list\": {}},\n\t\t\t\t\tActivityTaskListMap: map[string]*types.DescribeTaskListResponse{\"test-activity-task-list\": {}},\n\t\t\t\t}, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestDomainName() {\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\tresp       string\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomain).Return(s.testDomain, nil).Times(1)\n\t\t\t},\n\t\t\tresp: s.testDomain,\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - GetDomainName failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockDomainCache.EXPECT().GetDomainName(s.testDomain).Return(\"\", errors.New(\"get-domain-error\")).Times(1)\n\t\t\t},\n\t\t\tresp: \"\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\n\t\t\tresp := s.handler.domainName(s.testDomain)\n\n\t\t\ts.Equal(tc.resp, resp)\n\t\t})\n\n\t}\n}\nfunc (s *handlerSuite) TestRefreshTaskListPartitionConfig() {\n\trequest := types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\tDomainUUID: \"test-domain-id\",\n\t\tTaskList:   &types.TaskList{Name: \"test-task-list\"},\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\twant       *types.MatchingRefreshTaskListPartitionConfigResponse\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.MatchingRefreshTaskListPartitionConfigResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - RefreshTaskListPartitionConfig failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &request).\n\t\t\t\t\tReturn(nil, errors.New(\"refresh-tasklist-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"refresh-tasklist-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.RefreshTaskListPartitionConfig(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *handlerSuite) TestUpdateTaskListPartitionConfig() {\n\trequest := types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\tDomainUUID: \"test-domain-id\",\n\t\tTaskList:   &types.TaskList{Name: \"test-task-list\"},\n\t}\n\n\ttestCases := []struct {\n\t\tname       string\n\t\tsetupMocks func()\n\t\twant       *types.MatchingUpdateTaskListPartitionConfigResponse\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"Success case\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &request).\n\t\t\t\t\tReturn(&types.MatchingUpdateTaskListPartitionConfigResponse{}, nil).Times(1)\n\t\t\t},\n\t\t\twant: &types.MatchingUpdateTaskListPartitionConfigResponse{},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - rate limiter not allowed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(false).Times(1)\n\t\t\t},\n\t\t\terr: &types.ServiceBusyError{Message: \"Matching host rps exceeded\"},\n\t\t},\n\t\t{\n\t\t\tname: \"Error case - UpdateTaskListPartitionConfig failed\",\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockLimiter.EXPECT().Allow().Return(true).Times(1)\n\t\t\t\ts.mockEngine.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &request).\n\t\t\t\t\tReturn(nil, errors.New(\"update-tasklist-error\")).Times(1)\n\t\t\t},\n\t\t\terr: &types.InternalServiceError{Message: \"update-tasklist-error\"},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks()\n\t\t\ts.mockDomainCache.EXPECT().GetDomainName(request.DomainUUID).Return(s.testDomain, nil).Times(1)\n\n\t\t\tresp, err := s.handler.UpdateTaskListPartitionConfig(context.Background(), &request)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(tc.want, resp)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc partitions(num int) map[int]*types.TaskListPartition {\n\tresult := make(map[int]*types.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tresult[i] = &types.TaskListPartition{}\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "service/matching/handler/interfaces.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go -package handler github.com/uber/cadence/service/matching/handler Handler\n//go:generate gowrap gen -g -p . -i Handler -t ../../templates/grpc.tmpl -o ../wrappers/grpc/grpc_handler_generated.go -v handler=GRPC -v package=matchingv1 -v path=github.com/uber/cadence/.gen/proto/matching/v1 -v prefix=Matching\n//go:generate gowrap gen -g -p ../../../.gen/go/matching/matchingserviceserver -i Interface -t ../../templates/thrift.tmpl -o ../wrappers/thrift/thrift_handler_generated.go -v handler=Thrift -v prefix=Matching\n\npackage handler\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// Engine exposes interfaces for clients to poll for activity and decision tasks.\n\tEngine interface {\n\t\tcommon.Daemon\n\n\t\tAddDecisionTask(hCtx *handlerContext, request *types.AddDecisionTaskRequest) (*types.AddDecisionTaskResponse, error)\n\t\tAddActivityTask(hCtx *handlerContext, request *types.AddActivityTaskRequest) (*types.AddActivityTaskResponse, error)\n\t\tPollForDecisionTask(hCtx *handlerContext, request *types.MatchingPollForDecisionTaskRequest) (*types.MatchingPollForDecisionTaskResponse, error)\n\t\tPollForActivityTask(hCtx *handlerContext, request *types.MatchingPollForActivityTaskRequest) (*types.MatchingPollForActivityTaskResponse, error)\n\t\tQueryWorkflow(hCtx *handlerContext, request *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error)\n\t\tRespondQueryTaskCompleted(hCtx *handlerContext, request *types.MatchingRespondQueryTaskCompletedRequest) error\n\t\tCancelOutstandingPoll(hCtx *handlerContext, request *types.CancelOutstandingPollRequest) error\n\t\tDescribeTaskList(hCtx *handlerContext, request *types.MatchingDescribeTaskListRequest) (*types.DescribeTaskListResponse, error)\n\t\tListTaskListPartitions(hCtx *handlerContext, request *types.MatchingListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error)\n\t\tGetTaskListsByDomain(hCtx *handlerContext, request *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error)\n\t\tUpdateTaskListPartitionConfig(hCtx *handlerContext, request *types.MatchingUpdateTaskListPartitionConfigRequest) (*types.MatchingUpdateTaskListPartitionConfigResponse, error)\n\t\tRefreshTaskListPartitionConfig(hCtx *handlerContext, request *types.MatchingRefreshTaskListPartitionConfigRequest) (*types.MatchingRefreshTaskListPartitionConfigResponse, error)\n\t}\n\n\t// Handler interface for matching service\n\tHandler interface {\n\t\tcommon.Daemon\n\n\t\tHealth(context.Context) (*types.HealthStatus, error)\n\t\tAddActivityTask(context.Context, *types.AddActivityTaskRequest) (*types.AddActivityTaskResponse, error)\n\t\tAddDecisionTask(context.Context, *types.AddDecisionTaskRequest) (*types.AddDecisionTaskResponse, error)\n\t\tCancelOutstandingPoll(context.Context, *types.CancelOutstandingPollRequest) error\n\t\tDescribeTaskList(context.Context, *types.MatchingDescribeTaskListRequest) (*types.DescribeTaskListResponse, error)\n\t\tListTaskListPartitions(context.Context, *types.MatchingListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error)\n\t\tGetTaskListsByDomain(context.Context, *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error)\n\t\tPollForActivityTask(context.Context, *types.MatchingPollForActivityTaskRequest) (*types.MatchingPollForActivityTaskResponse, error)\n\t\tPollForDecisionTask(context.Context, *types.MatchingPollForDecisionTaskRequest) (*types.MatchingPollForDecisionTaskResponse, error)\n\t\tQueryWorkflow(context.Context, *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error)\n\t\tRespondQueryTaskCompleted(context.Context, *types.MatchingRespondQueryTaskCompletedRequest) error\n\t\tUpdateTaskListPartitionConfig(context.Context, *types.MatchingUpdateTaskListPartitionConfigRequest) (*types.MatchingUpdateTaskListPartitionConfigResponse, error)\n\t\tRefreshTaskListPartitionConfig(context.Context, *types.MatchingRefreshTaskListPartitionConfigRequest) (*types.MatchingRefreshTaskListPartitionConfigResponse, error)\n\t}\n)\n"
  },
  {
    "path": "service/matching/handler/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package handler -source interfaces.go -destination interfaces_mock.go -package handler github.com/uber/cadence/service/matching/handler Handler\n//\n\n// Package handler is a generated GoMock package.\npackage handler\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockEngine is a mock of Engine interface.\ntype MockEngine struct {\n\tctrl     *gomock.Controller\n\trecorder *MockEngineMockRecorder\n\tisgomock struct{}\n}\n\n// MockEngineMockRecorder is the mock recorder for MockEngine.\ntype MockEngineMockRecorder struct {\n\tmock *MockEngine\n}\n\n// NewMockEngine creates a new mock instance.\nfunc NewMockEngine(ctrl *gomock.Controller) *MockEngine {\n\tmock := &MockEngine{ctrl: ctrl}\n\tmock.recorder = &MockEngineMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockEngine) EXPECT() *MockEngineMockRecorder {\n\treturn m.recorder\n}\n\n// AddActivityTask mocks base method.\nfunc (m *MockEngine) AddActivityTask(hCtx *handlerContext, request *types.AddActivityTaskRequest) (*types.AddActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTask\", hCtx, request)\n\tret0, _ := ret[0].(*types.AddActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTask indicates an expected call of AddActivityTask.\nfunc (mr *MockEngineMockRecorder) AddActivityTask(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTask\", reflect.TypeOf((*MockEngine)(nil).AddActivityTask), hCtx, request)\n}\n\n// AddDecisionTask mocks base method.\nfunc (m *MockEngine) AddDecisionTask(hCtx *handlerContext, request *types.AddDecisionTaskRequest) (*types.AddDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTask\", hCtx, request)\n\tret0, _ := ret[0].(*types.AddDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTask indicates an expected call of AddDecisionTask.\nfunc (mr *MockEngineMockRecorder) AddDecisionTask(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTask\", reflect.TypeOf((*MockEngine)(nil).AddDecisionTask), hCtx, request)\n}\n\n// CancelOutstandingPoll mocks base method.\nfunc (m *MockEngine) CancelOutstandingPoll(hCtx *handlerContext, request *types.CancelOutstandingPollRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CancelOutstandingPoll\", hCtx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CancelOutstandingPoll indicates an expected call of CancelOutstandingPoll.\nfunc (mr *MockEngineMockRecorder) CancelOutstandingPoll(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CancelOutstandingPoll\", reflect.TypeOf((*MockEngine)(nil).CancelOutstandingPoll), hCtx, request)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockEngine) DescribeTaskList(hCtx *handlerContext, request *types.MatchingDescribeTaskListRequest) (*types.DescribeTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", hCtx, request)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockEngineMockRecorder) DescribeTaskList(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockEngine)(nil).DescribeTaskList), hCtx, request)\n}\n\n// GetTaskListsByDomain mocks base method.\nfunc (m *MockEngine) GetTaskListsByDomain(hCtx *handlerContext, request *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", hCtx, request)\n\tret0, _ := ret[0].(*types.GetTaskListsByDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListsByDomain indicates an expected call of GetTaskListsByDomain.\nfunc (mr *MockEngineMockRecorder) GetTaskListsByDomain(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListsByDomain\", reflect.TypeOf((*MockEngine)(nil).GetTaskListsByDomain), hCtx, request)\n}\n\n// ListTaskListPartitions mocks base method.\nfunc (m *MockEngine) ListTaskListPartitions(hCtx *handlerContext, request *types.MatchingListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", hCtx, request)\n\tret0, _ := ret[0].(*types.ListTaskListPartitionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskListPartitions indicates an expected call of ListTaskListPartitions.\nfunc (mr *MockEngineMockRecorder) ListTaskListPartitions(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskListPartitions\", reflect.TypeOf((*MockEngine)(nil).ListTaskListPartitions), hCtx, request)\n}\n\n// PollForActivityTask mocks base method.\nfunc (m *MockEngine) PollForActivityTask(hCtx *handlerContext, request *types.MatchingPollForActivityTaskRequest) (*types.MatchingPollForActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", hCtx, request)\n\tret0, _ := ret[0].(*types.MatchingPollForActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForActivityTask indicates an expected call of PollForActivityTask.\nfunc (mr *MockEngineMockRecorder) PollForActivityTask(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForActivityTask\", reflect.TypeOf((*MockEngine)(nil).PollForActivityTask), hCtx, request)\n}\n\n// PollForDecisionTask mocks base method.\nfunc (m *MockEngine) PollForDecisionTask(hCtx *handlerContext, request *types.MatchingPollForDecisionTaskRequest) (*types.MatchingPollForDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", hCtx, request)\n\tret0, _ := ret[0].(*types.MatchingPollForDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForDecisionTask indicates an expected call of PollForDecisionTask.\nfunc (mr *MockEngineMockRecorder) PollForDecisionTask(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForDecisionTask\", reflect.TypeOf((*MockEngine)(nil).PollForDecisionTask), hCtx, request)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockEngine) QueryWorkflow(hCtx *handlerContext, request *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", hCtx, request)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockEngineMockRecorder) QueryWorkflow(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockEngine)(nil).QueryWorkflow), hCtx, request)\n}\n\n// RefreshTaskListPartitionConfig mocks base method.\nfunc (m *MockEngine) RefreshTaskListPartitionConfig(hCtx *handlerContext, request *types.MatchingRefreshTaskListPartitionConfigRequest) (*types.MatchingRefreshTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshTaskListPartitionConfig\", hCtx, request)\n\tret0, _ := ret[0].(*types.MatchingRefreshTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RefreshTaskListPartitionConfig indicates an expected call of RefreshTaskListPartitionConfig.\nfunc (mr *MockEngineMockRecorder) RefreshTaskListPartitionConfig(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshTaskListPartitionConfig\", reflect.TypeOf((*MockEngine)(nil).RefreshTaskListPartitionConfig), hCtx, request)\n}\n\n// RespondQueryTaskCompleted mocks base method.\nfunc (m *MockEngine) RespondQueryTaskCompleted(hCtx *handlerContext, request *types.MatchingRespondQueryTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", hCtx, request)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondQueryTaskCompleted indicates an expected call of RespondQueryTaskCompleted.\nfunc (mr *MockEngineMockRecorder) RespondQueryTaskCompleted(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondQueryTaskCompleted\", reflect.TypeOf((*MockEngine)(nil).RespondQueryTaskCompleted), hCtx, request)\n}\n\n// Start mocks base method.\nfunc (m *MockEngine) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockEngineMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockEngine)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockEngine) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockEngineMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockEngine)(nil).Stop))\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockEngine) UpdateTaskListPartitionConfig(hCtx *handlerContext, request *types.MatchingUpdateTaskListPartitionConfigRequest) (*types.MatchingUpdateTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", hCtx, request)\n\tret0, _ := ret[0].(*types.MatchingUpdateTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockEngineMockRecorder) UpdateTaskListPartitionConfig(hCtx, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockEngine)(nil).UpdateTaskListPartitionConfig), hCtx, request)\n}\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// AddActivityTask mocks base method.\nfunc (m *MockHandler) AddActivityTask(arg0 context.Context, arg1 *types.AddActivityTaskRequest) (*types.AddActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddActivityTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.AddActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddActivityTask indicates an expected call of AddActivityTask.\nfunc (mr *MockHandlerMockRecorder) AddActivityTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddActivityTask\", reflect.TypeOf((*MockHandler)(nil).AddActivityTask), arg0, arg1)\n}\n\n// AddDecisionTask mocks base method.\nfunc (m *MockHandler) AddDecisionTask(arg0 context.Context, arg1 *types.AddDecisionTaskRequest) (*types.AddDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddDecisionTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.AddDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddDecisionTask indicates an expected call of AddDecisionTask.\nfunc (mr *MockHandlerMockRecorder) AddDecisionTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddDecisionTask\", reflect.TypeOf((*MockHandler)(nil).AddDecisionTask), arg0, arg1)\n}\n\n// CancelOutstandingPoll mocks base method.\nfunc (m *MockHandler) CancelOutstandingPoll(arg0 context.Context, arg1 *types.CancelOutstandingPollRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CancelOutstandingPoll\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CancelOutstandingPoll indicates an expected call of CancelOutstandingPoll.\nfunc (mr *MockHandlerMockRecorder) CancelOutstandingPoll(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CancelOutstandingPoll\", reflect.TypeOf((*MockHandler)(nil).CancelOutstandingPoll), arg0, arg1)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockHandler) DescribeTaskList(arg0 context.Context, arg1 *types.MatchingDescribeTaskListRequest) (*types.DescribeTaskListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", arg0, arg1)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockHandlerMockRecorder) DescribeTaskList(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockHandler)(nil).DescribeTaskList), arg0, arg1)\n}\n\n// GetTaskListsByDomain mocks base method.\nfunc (m *MockHandler) GetTaskListsByDomain(arg0 context.Context, arg1 *types.GetTaskListsByDomainRequest) (*types.GetTaskListsByDomainResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListsByDomain\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetTaskListsByDomainResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTaskListsByDomain indicates an expected call of GetTaskListsByDomain.\nfunc (mr *MockHandlerMockRecorder) GetTaskListsByDomain(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListsByDomain\", reflect.TypeOf((*MockHandler)(nil).GetTaskListsByDomain), arg0, arg1)\n}\n\n// Health mocks base method.\nfunc (m *MockHandler) Health(arg0 context.Context) (*types.HealthStatus, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Health\", arg0)\n\tret0, _ := ret[0].(*types.HealthStatus)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Health indicates an expected call of Health.\nfunc (mr *MockHandlerMockRecorder) Health(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Health\", reflect.TypeOf((*MockHandler)(nil).Health), arg0)\n}\n\n// ListTaskListPartitions mocks base method.\nfunc (m *MockHandler) ListTaskListPartitions(arg0 context.Context, arg1 *types.MatchingListTaskListPartitionsRequest) (*types.ListTaskListPartitionsResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ListTaskListPartitions\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ListTaskListPartitionsResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ListTaskListPartitions indicates an expected call of ListTaskListPartitions.\nfunc (mr *MockHandlerMockRecorder) ListTaskListPartitions(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ListTaskListPartitions\", reflect.TypeOf((*MockHandler)(nil).ListTaskListPartitions), arg0, arg1)\n}\n\n// PollForActivityTask mocks base method.\nfunc (m *MockHandler) PollForActivityTask(arg0 context.Context, arg1 *types.MatchingPollForActivityTaskRequest) (*types.MatchingPollForActivityTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForActivityTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MatchingPollForActivityTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForActivityTask indicates an expected call of PollForActivityTask.\nfunc (mr *MockHandlerMockRecorder) PollForActivityTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForActivityTask\", reflect.TypeOf((*MockHandler)(nil).PollForActivityTask), arg0, arg1)\n}\n\n// PollForDecisionTask mocks base method.\nfunc (m *MockHandler) PollForDecisionTask(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest) (*types.MatchingPollForDecisionTaskResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForDecisionTask\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MatchingPollForDecisionTaskResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForDecisionTask indicates an expected call of PollForDecisionTask.\nfunc (mr *MockHandlerMockRecorder) PollForDecisionTask(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForDecisionTask\", reflect.TypeOf((*MockHandler)(nil).PollForDecisionTask), arg0, arg1)\n}\n\n// QueryWorkflow mocks base method.\nfunc (m *MockHandler) QueryWorkflow(arg0 context.Context, arg1 *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueryWorkflow\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// QueryWorkflow indicates an expected call of QueryWorkflow.\nfunc (mr *MockHandlerMockRecorder) QueryWorkflow(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueryWorkflow\", reflect.TypeOf((*MockHandler)(nil).QueryWorkflow), arg0, arg1)\n}\n\n// RefreshTaskListPartitionConfig mocks base method.\nfunc (m *MockHandler) RefreshTaskListPartitionConfig(arg0 context.Context, arg1 *types.MatchingRefreshTaskListPartitionConfigRequest) (*types.MatchingRefreshTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshTaskListPartitionConfig\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MatchingRefreshTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RefreshTaskListPartitionConfig indicates an expected call of RefreshTaskListPartitionConfig.\nfunc (mr *MockHandlerMockRecorder) RefreshTaskListPartitionConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshTaskListPartitionConfig\", reflect.TypeOf((*MockHandler)(nil).RefreshTaskListPartitionConfig), arg0, arg1)\n}\n\n// RespondQueryTaskCompleted mocks base method.\nfunc (m *MockHandler) RespondQueryTaskCompleted(arg0 context.Context, arg1 *types.MatchingRespondQueryTaskCompletedRequest) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RespondQueryTaskCompleted\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RespondQueryTaskCompleted indicates an expected call of RespondQueryTaskCompleted.\nfunc (mr *MockHandlerMockRecorder) RespondQueryTaskCompleted(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RespondQueryTaskCompleted\", reflect.TypeOf((*MockHandler)(nil).RespondQueryTaskCompleted), arg0, arg1)\n}\n\n// Start mocks base method.\nfunc (m *MockHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockHandler)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockHandler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockHandlerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockHandler)(nil).Stop))\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockHandler) UpdateTaskListPartitionConfig(arg0 context.Context, arg1 *types.MatchingUpdateTaskListPartitionConfigRequest) (*types.MatchingUpdateTaskListPartitionConfigResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", arg0, arg1)\n\tret0, _ := ret[0].(*types.MatchingUpdateTaskListPartitionConfigResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockHandlerMockRecorder) UpdateTaskListPartitionConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockHandler)(nil).UpdateTaskListPartitionConfig), arg0, arg1)\n}\n"
  },
  {
    "path": "service/matching/handler/membership.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n)\n\nconst subscriptionBufferSize = 1000\n\n// Because there's a bunch of conditions under which matching may be holding a tasklist\n// reader daemon and other live procesess but when it doesn't (according to the rest of the hashring)\n// own the tasklist anymore, this listener watches for membership changes and purges anything disused\n// in the hashring on membership changes.\n//\n// Combined with the guard on tasklist instantiation, it should prevent incorrect or poorly timed\n// creating of tasklist ownership and database shard thrashing between hosts while they figure out\n// which host is the real owner of the tasklist.\n//\n// This is not the main shutdown process, its just an optimization.\nfunc (e *matchingEngineImpl) runMembershipChangeLoop() {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\te.logger.Error(\"matching membership watcher changes caused a panic, recovering\", tag.Dynamic(\"recovered-panic\", r))\n\t\t}\n\t}()\n\n\tdefer e.shutdownCompletion.Done()\n\n\tif !e.config.EnableTasklistOwnershipGuard() {\n\t\treturn\n\t}\n\n\tlistener := make(chan *membership.ChangedEvent, subscriptionBufferSize)\n\tif err := e.membershipResolver.Subscribe(service.Matching, \"matching-engine\", listener); err != nil {\n\t\te.logger.Error(\"Failed to subscribe to membership updates\")\n\t\treturn\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase event := <-listener:\n\t\t\terr := e.shutDownNonOwnedTasklists()\n\t\t\tif err != nil {\n\t\t\t\te.logger.Error(\"Error while trying to determine if tasklists have been shutdown\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t\ttag.MembershipChangeEvent(event),\n\t\t\t\t)\n\t\t\t}\n\t\tcase <-e.shutdown:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (e *matchingEngineImpl) shutDownNonOwnedTasklists() error {\n\tif !e.config.EnableTasklistOwnershipGuard() {\n\t\treturn nil\n\t}\n\tnoLongerOwned, err := e.getNonOwnedTasklistsLocked()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttasklistsShutdownWG := sync.WaitGroup{}\n\n\tfor _, tl := range noLongerOwned {\n\t\t// for each of the tasklists that are no longer owned, kick off the\n\t\t// process of stopping them. The stopping process is IO heavy and\n\t\t// can take a while, so do them in parallel to efficiently unload tasklists not owned\n\t\ttasklistsShutdownWG.Add(1)\n\t\tgo func(tl tasklist.Manager) {\n\n\t\t\tdefer func() {\n\t\t\t\tif r := recover(); r != nil {\n\t\t\t\t\te.logger.Error(\"panic occurred while trying to shut down tasklist\", tag.Dynamic(\"recovered-panic\", r))\n\t\t\t\t}\n\t\t\t}()\n\t\t\tdefer tasklistsShutdownWG.Done()\n\n\t\t\te.logger.Info(\"shutting down tasklist preemptively because they are no longer owned by this host\",\n\t\t\t\ttag.WorkflowTaskListType(tl.TaskListID().GetType()),\n\t\t\t\ttag.WorkflowTaskListName(tl.TaskListID().GetName()),\n\t\t\t\ttag.WorkflowDomainID(tl.TaskListID().GetDomainID()),\n\t\t\t\ttag.Dynamic(\"tasklist-debug-info\", tl.String()),\n\t\t\t)\n\n\t\t\te.unloadTaskList(tl)\n\t\t}(tl)\n\t}\n\n\ttasklistsShutdownWG.Wait()\n\n\treturn nil\n}\n\nfunc (e *matchingEngineImpl) getNonOwnedTasklistsLocked() ([]tasklist.Manager, error) {\n\tif !e.config.EnableTasklistOwnershipGuard() {\n\t\treturn nil, nil\n\t}\n\n\tvar toShutDown []tasklist.Manager\n\n\ttaskLists := e.taskListRegistry.AllManagers()\n\n\tself, err := e.membershipResolver.WhoAmI()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to lookup self im membership: %w\", err)\n\t}\n\n\tfor _, tl := range taskLists {\n\t\ttaskListOwner, err := e.membershipResolver.Lookup(service.Matching, tl.TaskListID().GetName())\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to lookup task list owner: %w\", err)\n\t\t}\n\n\t\tif taskListOwner.Identity() != self.Identity() {\n\t\t\ttoShutDown = append(toShutDown, tl)\n\t\t}\n\t}\n\n\te.logger.Info(\"Got list of non-owned-tasklists\",\n\t\ttag.Dynamic(\"tasklist-debug-info\", toShutDown),\n\t)\n\treturn toShutDown, nil\n}\n"
  },
  {
    "path": "service/matching/handler/membership_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tcadence_errors \"github.com/uber/cadence/common/errors\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/tasklist\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc TestGetTaskListManager_OwnerShip(t *testing.T) {\n\n\ttestCases := []struct {\n\t\tname                 string\n\t\tlookUpResult         string\n\t\tlookUpErr            error\n\t\twhoAmIResult         string\n\t\twhoAmIErr            error\n\t\ttasklistGuardEnabled bool\n\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname:                 \"Not owned by current host\",\n\t\t\tlookUpResult:         \"A\",\n\t\t\twhoAmIResult:         \"B\",\n\t\t\ttasklistGuardEnabled: true,\n\n\t\t\texpectedError: new(cadence_errors.TaskListNotOwnedByHostError),\n\t\t},\n\t\t{\n\t\t\tname:                 \"LookupError\",\n\t\t\tlookUpErr:            assert.AnError,\n\t\t\ttasklistGuardEnabled: true,\n\t\t\texpectedError:        assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname:                 \"WhoAmIError\",\n\t\t\twhoAmIErr:            assert.AnError,\n\t\t\ttasklistGuardEnabled: true,\n\t\t\texpectedError:        assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname:                 \"when feature is not enabled, expect previous behaviour to continue\",\n\t\t\tlookUpResult:         \"A\",\n\t\t\twhoAmIResult:         \"B\",\n\t\t\ttasklistGuardEnabled: false,\n\n\t\t\texpectedError: nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\n\t\tt.Run(tc.name, func(t *testing.T) {\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := log.NewNoop()\n\n\t\t\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\t\t\ttaskManager := tasklist.NewTestTaskManager(t, logger, mockTimeSource)\n\t\t\tmockHistoryClient := history.NewMockClient(ctrl)\n\t\t\tmockMatchingClient := matching.NewMockClient(ctrl)\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tresolverMock := membership.NewMockResolver(ctrl)\n\t\t\tresolverMock.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any()).AnyTimes()\n\t\t\tmockShardDistributorExecutorClient := executorclient.NewMockClient(ctrl)\n\n\t\t\t// this is only if the call goes through\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(matchingTestDomainName), nil).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.CreateDomainCacheEntry(matchingTestDomainName), nil).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(matchingTestDomainName, nil).AnyTimes()\n\n\t\t\tconfig := defaultTestConfig()\n\t\t\ttaskListEnabled := tc.tasklistGuardEnabled\n\t\t\tconfig.EnableTasklistOwnershipGuard = func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\treturn taskListEnabled\n\t\t\t}\n\t\t\t// Exclude all task lists from the ShardDistributor so that errIfShardOwnershipLost\n\t\t\t// exercises the ringpop (hash-ring) ownership path that these test cases are actually\n\t\t\t// testing. With PercentageOnboardedToShardManager=0, no task list name is below the\n\t\t\t// percentage threshold, so every name is considered excluded regardless of whether it\n\t\t\t// contains a UUID.\n\t\t\tconfig.ExcludeShortLivedTaskListsFromShardManager = func(opts ...dynamicproperties.FilterOption) bool { return true }\n\t\t\tconfig.PercentageOnboardedToShardManager = func(opts ...dynamicproperties.FilterOption) int { return 0 }\n\n\t\t\tmatchingEngine := NewEngine(\n\t\t\t\ttaskManager,\n\t\t\t\tcluster.GetTestClusterMetadata(true),\n\t\t\t\tmockHistoryClient,\n\t\t\t\tmockMatchingClient,\n\t\t\t\tconfig,\n\t\t\t\tlogger,\n\t\t\t\tmetrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\t\t\ttally.NoopScope,\n\t\t\t\tmockDomainCache,\n\t\t\t\tresolverMock,\n\t\t\t\tisolationgroup.NewMockState(ctrl),\n\t\t\t\tmockTimeSource,\n\t\t\t\tmockShardDistributorExecutorClient,\n\t\t\t\tdefaultSDExecutorConfig(),\n\t\t\t\tnil,\n\t\t\t).(*matchingEngineImpl)\n\n\t\t\t// All task lists are excluded from the ShardDistributor, so GetShardProcess is\n\t\t\t// never called. Only Start/Stop are needed for lifecycle management.\n\t\t\tmockExec := executorclient.NewMockExecutor[tasklist.ShardProcessor](ctrl)\n\t\t\tmockExec.EXPECT().Start(gomock.Any()).AnyTimes()\n\t\t\tmockExec.EXPECT().Stop().AnyTimes()\n\t\t\tmatchingEngine.executor = mockExec\n\n\t\t\tresolverMock.EXPECT().Lookup(gomock.Any(), gomock.Any()).Return(\n\t\t\t\tmembership.NewDetailedHostInfo(\"\", tc.lookUpResult, make(membership.PortMap)), tc.lookUpErr,\n\t\t\t).AnyTimes()\n\t\t\tresolverMock.EXPECT().WhoAmI().Return(\n\t\t\t\tmembership.NewDetailedHostInfo(\"\", tc.whoAmIResult, make(membership.PortMap)), tc.whoAmIErr,\n\t\t\t).AnyTimes()\n\n\t\t\t_, err := matchingEngine.getOrCreateTaskListManager(context.Background(),\n\t\t\t\ttasklist.NewTestTaskListID(t, \"domain\", \"tasklist\", persistence.TaskListTypeActivity),\n\t\t\t\ttypes.TaskListKindNormal,\n\t\t\t)\n\t\t\tif tc.expectedError != nil {\n\t\t\t\tassert.ErrorAs(t, err, &tc.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMembershipSubscriptionRecoversAfterPanic(t *testing.T) {\n\tassert.NotPanics(t, func() {\n\t\tctrl := gomock.NewController(t)\n\n\t\tr := resource.NewTest(t, ctrl, 0)\n\t\tr.MembershipResolver.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any()).DoAndReturn(func(_, _, _ any) {\n\t\t\tpanic(\"a panic has occurred\")\n\t\t})\n\n\t\tengine := matchingEngineImpl{\n\t\t\tmembershipResolver: r.MembershipResolver,\n\t\t\tconfig: &config.Config{\n\t\t\t\tEnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\t},\n\t\t\tlogger:   log.NewNoop(),\n\t\t\tshutdown: make(chan struct{}),\n\t\t}\n\n\t\tengine.runMembershipChangeLoop()\n\t})\n}\n\nfunc TestSubscriptionAndShutdown(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](ctrl)\n\tmockExecutor.EXPECT().Stop()\n\n\tshutdownWG := sync.WaitGroup{}\n\tshutdownWG.Add(1)\n\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\tengine := matchingEngineImpl{\n\t\tshutdownCompletion: &shutdownWG,\n\t\tmembershipResolver: mockResolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig:             &config.Config{EnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true }},\n\t\tshutdown:           make(chan struct{}),\n\t\tlogger:             log.NewNoop(),\n\t\tdomainCache:        mockDomainCache,\n\t\texecutor:           mockExecutor,\n\t}\n\n\tmockResolver.EXPECT().WhoAmI().Return(membership.NewDetailedHostInfo(\"host2\", \"host2\", nil), nil).AnyTimes()\n\tmockResolver.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any())\n\tmockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).Times(1)\n\n\tgo engine.runMembershipChangeLoop()\n\n\tengine.Stop()\n\tassert.True(t, common.AwaitWaitGroup(&shutdownWG, 10*time.Second), \"runMembershipChangeLoop has to be shut down\")\n}\n\nfunc TestSubscriptionAndErrorReturned(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](ctrl)\n\tmockExecutor.EXPECT().Stop()\n\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\tshutdownWG := sync.WaitGroup{}\n\tshutdownWG.Add(1)\n\n\tmembershipChangeHandledWG := sync.WaitGroup{}\n\tmembershipChangeHandledWG.Add(1)\n\n\tengine := matchingEngineImpl{\n\t\tshutdownCompletion: &shutdownWG,\n\t\tmembershipResolver: mockResolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig:             &config.Config{EnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true }},\n\t\tshutdown:           make(chan struct{}),\n\t\tlogger:             log.NewNoop(),\n\t\tdomainCache:        mockDomainCache,\n\t\texecutor:           mockExecutor,\n\t}\n\n\t// this should trigger the error case on a membership event\n\t// unfortunately, this is purely for code-coverage, no checks are involved\n\tmockResolver.EXPECT().WhoAmI().DoAndReturn(func() (membership.HostInfo, error) {\n\t\tmembershipChangeHandledWG.Done()\n\t\treturn membership.HostInfo{}, errors.New(\"failure\")\n\t}).MinTimes(1)\n\n\tmockResolver.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any()).Do(\n\t\tfunc(service string, name string, inc chan<- *membership.ChangedEvent) {\n\t\t\tm := membership.ChangedEvent{\n\t\t\t\tHostsAdded:   nil,\n\t\t\t\tHostsUpdated: nil,\n\t\t\t\tHostsRemoved: []string{\"host123\"},\n\t\t\t}\n\t\t\tinc <- &m\n\t\t})\n\n\tmockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).Times(1)\n\n\tgo engine.runMembershipChangeLoop()\n\n\tassert.True(t,\n\t\tcommon.AwaitWaitGroup(&membershipChangeHandledWG, 10*time.Second),\n\t\t\"membership event is not handled\",\n\t)\n\n\tengine.Stop()\n\tassert.True(t,\n\t\tcommon.AwaitWaitGroup(&shutdownWG, 10*time.Second),\n\t\t\"runMembershipChangeLoop has to be shut down\",\n\t)\n}\n\nfunc TestSubscribeToMembershipChangesQuitsIfSubscribeFails(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\tlogger, logs := testlogger.NewObserved(t)\n\n\tshutdownWG := sync.WaitGroup{}\n\tshutdownWG.Add(1)\n\n\tengine := matchingEngineImpl{\n\t\tshutdownCompletion: &shutdownWG,\n\t\tmembershipResolver: mockResolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig:             &config.Config{EnableTasklistOwnershipGuard: func(opts ...dynamicproperties.FilterOption) bool { return true }},\n\t\tshutdown:           make(chan struct{}),\n\t\tlogger:             logger,\n\t\tdomainCache:        mockDomainCache,\n\t}\n\n\tmockResolver.EXPECT().Subscribe(service.Matching, \"matching-engine\", gomock.Any()).\n\t\tReturn(errors.New(\"failed to subscribe\"))\n\n\tmockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).AnyTimes()\n\n\tgo engine.runMembershipChangeLoop()\n\t// we do not stop `engine` here - it has to shut down after failing to Subscribe\n\n\tassert.True(\n\t\tt,\n\t\tcommon.AwaitWaitGroup(&shutdownWG, 10*time.Second),\n\t\t\"runMembershipChangeLoop should immediately shut down because of critical error\",\n\t)\n\n\t// check we emitted error-message\n\tfilteredLogs := logs.FilterMessage(\"Failed to subscribe to membership updates\")\n\tassert.Equal(t, 1, filteredLogs.Len(), \"error-message should be produced\")\n}\n\nfunc TestGetTasklistManagerShutdownScenario(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockResolver := membership.NewMockResolver(ctrl)\n\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\tself := membership.NewDetailedHostInfo(\"self\", \"self\", nil)\n\n\tmockResolver.EXPECT().WhoAmI().Return(self, nil).AnyTimes()\n\tmockDomainCache.EXPECT().UnregisterDomainChangeCallback(service.Matching).Times(1)\n\n\tmockExecutor := executorclient.NewMockExecutor[tasklist.ShardProcessor](ctrl)\n\tmockExecutor.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tmockExecutor.EXPECT().Stop()\n\n\tshutdownWG := sync.WaitGroup{}\n\tshutdownWG.Add(0)\n\n\tengine := matchingEngineImpl{\n\t\tshutdownCompletion: &shutdownWG,\n\t\tmembershipResolver: mockResolver,\n\t\ttaskListRegistry:   tasklist.NewTaskListRegistry(metrics.NewNoopMetricsClient()),\n\t\tconfig: &config.Config{\n\t\t\tEnableTasklistOwnershipGuard:               func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\tExcludeShortLivedTaskListsFromShardManager: func(opts ...dynamicproperties.FilterOption) bool { return true },\n\t\t\tPercentageOnboardedToShardManager:          func(opts ...dynamicproperties.FilterOption) int { return 100 },\n\t\t},\n\t\tshutdown:    make(chan struct{}),\n\t\tlogger:      log.NewNoop(),\n\t\tdomainCache: mockDomainCache,\n\t\texecutor:    mockExecutor,\n\t}\n\n\t// set this engine to be shutting down to trigger the tasklistGetTasklistByID guard\n\tengine.Stop()\n\n\ttl, _ := tasklist.NewIdentifier(\"domainid\", \"tl\", 0)\n\tres, err := engine.getOrCreateTaskListManager(context.Background(), tl, types.TaskListKindNormal)\n\tassertErr := &cadence_errors.TaskListNotOwnedByHostError{}\n\tassert.ErrorAs(t, err, &assertErr)\n\tassert.Nil(t, res)\n}\n"
  },
  {
    "path": "service/matching/liveness/liveness.go",
    "content": "// Modifications Copyright (c) 2020 Uber Technologies Inc.\n\n// Copyright (c) 2020 Temporal Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage liveness\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\tLiveness struct {\n\t\tstatus     int32\n\t\ttimeSource clock.TimeSource\n\t\tttl        time.Duration\n\n\t\t// stopCh is used to signal the liveness to stop\n\t\tstopCh chan struct{}\n\t\t// wg is used to wait for the liveness to stop\n\t\twg sync.WaitGroup\n\n\t\t// broadcast shutdown functions\n\t\tbroadcastShutdownFn func()\n\n\t\tlastEventTimeNano int64\n\t}\n)\n\nvar _ common.Daemon = (*Liveness)(nil)\n\n// NewLiveness creates a Liveness daemon that calls the broadcastShutdownFn if it does not receive MarkAlive() within ttl\n// NOTE: livesness needs to be stopped explicitly to avoid go routine leak\nfunc NewLiveness(timeSource clock.TimeSource, ttl time.Duration, broadcastShutdownFn func()) *Liveness {\n\treturn &Liveness{\n\t\tstatus:              common.DaemonStatusInitialized,\n\t\ttimeSource:          timeSource,\n\t\tttl:                 ttl,\n\t\tstopCh:              make(chan struct{}),\n\t\tbroadcastShutdownFn: broadcastShutdownFn,\n\t\tlastEventTimeNano:   timeSource.Now().UnixNano(),\n\t}\n}\n\nfunc (l *Liveness) Start() {\n\tif !atomic.CompareAndSwapInt32(&l.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tl.wg.Add(1)\n\tcheckTimer := l.timeSource.NewTicker(l.ttl / 2)\n\tgo l.eventLoop(checkTimer)\n}\n\n// Stop ONLY shuts down liveness does not block on broadcastShutdownFn\nfunc (l *Liveness) Stop() {\n\tif !atomic.CompareAndSwapInt32(&l.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\tclose(l.stopCh)\n\tl.wg.Wait()\n}\n\nfunc (l *Liveness) eventLoop(ticker clock.Ticker) {\n\tdefer l.wg.Done()\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tif !l.IsAlive() {\n\t\t\t\tgo l.broadcastShutdownFn() // do not block shutdown\n\t\t\t\treturn\n\t\t\t}\n\n\t\tcase <-l.stopCh:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (l *Liveness) IsAlive() bool {\n\tnow := l.timeSource.Now().UnixNano()\n\tlastUpdate := atomic.LoadInt64(&l.lastEventTimeNano)\n\treturn now-lastUpdate < l.ttl.Nanoseconds()\n}\n\nfunc (l *Liveness) MarkAlive() {\n\tnow := l.timeSource.Now().UnixNano()\n\tatomic.StoreInt64(&l.lastEventTimeNano, now)\n}\n"
  },
  {
    "path": "service/matching/liveness/liveness_test.go",
    "content": "// Modifications Copyright (c) 2020 Uber Technologies Inc.\n\n// Copyright (c) 2020 Temporal Technologies, Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage liveness\n\nimport (\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\ntype (\n\tlivenessSuite struct {\n\t\tsuite.Suite\n\t\t*require.Assertions\n\n\t\ttimeSource   clock.MockedTimeSource\n\t\tttl          time.Duration\n\t\tshutdownFlag int32\n\t\tliveness     *Liveness\n\t}\n)\n\nfunc TestLivenessSuite(t *testing.T) {\n\ts := new(livenessSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *livenessSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.ttl = 500 * time.Millisecond\n\ts.timeSource = clock.NewMockedTimeSource()\n\ts.shutdownFlag = 0\n\ts.liveness = NewLiveness(s.timeSource, s.ttl, func() {\n\t\tatomic.CompareAndSwapInt32(&s.shutdownFlag, 0, 1)\n\t\ts.liveness.Stop()\n\t})\n}\n\nfunc (s *livenessSuite) TestIsAlive_No() {\n\ts.timeSource.Advance(s.ttl)\n\talive := s.liveness.IsAlive()\n\ts.False(alive)\n}\n\nfunc (s *livenessSuite) TestIsAlive_Yes() {\n\ts.timeSource.Advance(s.ttl / 2)\n\talive := s.liveness.IsAlive()\n\ts.True(alive)\n}\n\nfunc (s *livenessSuite) TestMarkAlive_Noop() {\n\tlastEventTime := s.liveness.lastEventTimeNano\n\t// not advanding time so markAlive will be a noop\n\ts.liveness.MarkAlive()\n\ts.Equal(lastEventTime, s.liveness.lastEventTimeNano)\n}\n\nfunc (s *livenessSuite) TestMarkAlive_Updated() {\n\ts.timeSource.Advance(time.Duration(1))\n\ts.liveness.MarkAlive()\n\ts.Equal(s.timeSource.Now().UnixNano(), s.liveness.lastEventTimeNano)\n}\n\nfunc (s *livenessSuite) TestEventLoop_Noop() {\n\ts.liveness.Start()\n\tdefer s.liveness.Stop()\n\n\t// advance time ttl/2 and mark alive\n\ts.timeSource.Advance(s.ttl / 2)\n\ts.liveness.MarkAlive()\n\ts.True(s.liveness.IsAlive())\n\n\t// advance time ttl/2 more and validate still alive\n\ts.timeSource.Advance(s.ttl / 2)\n\ttime.Sleep(100 * time.Millisecond) // give event loop time to run\n\ts.True(s.liveness.IsAlive())\n\ts.Equal(int32(0), atomic.LoadInt32(&s.shutdownFlag))\n}\n\nfunc (s *livenessSuite) TestEventLoop_Shutdown() {\n\ts.liveness.Start()\n\tdefer s.liveness.Stop()\n\n\ts.timeSource.Advance(s.ttl)\n\t<-s.liveness.stopCh\n\ts.Equal(int32(1), atomic.LoadInt32(&s.shutdownFlag))\n}\n"
  },
  {
    "path": "service/matching/poller/manager.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage poller\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tpollerHistoryInitSize    = 0\n\tpollerHistoryInitMaxSize = 5000\n\tpollerHistoryTTL         = 5 * time.Minute\n)\n\ntype (\n\tInfo struct {\n\t\tIdentity       string\n\t\tRatePerSecond  float64\n\t\tIsolationGroup string\n\t}\n\n\tManager interface {\n\t\tStartPoll(pollerID string, cancelFunc context.CancelFunc, info *Info)\n\t\tEndPoll(pollerID string)\n\t\tCancelPoll(pollerID string) bool\n\t\tHasPollerFromIsolationGroupAfter(isolationGroup string, after time.Time) bool\n\t\tHasPollerAfter(after time.Time) bool\n\t\tGetCount() int\n\t\tGetCountByIsolationGroup(after time.Time) map[string]int\n\t\tListInfo() []*types.PollerInfo\n\t}\n\n\thistoricalPoller struct {\n\t\tinfo        *Info\n\t\toutstanding bool\n\t}\n\n\toutstandingPoller struct {\n\t\tinfo   *Info\n\t\tcancel context.CancelFunc\n\t}\n\n\tmanager struct {\n\t\t// identity -> historicalPoller\n\t\thistoryCache cache.Cache\n\t\ttimeSource   clock.TimeSource\n\n\t\tlock                     sync.RWMutex\n\t\tmostRecentPollEnd        time.Time\n\t\tmostRecentPollEndByGroup map[string]time.Time\n\t\toutstandingCountByGroup  map[string]int\n\t\t// pollerID -> outstandingPoller\n\t\toutstanding map[string]outstandingPoller\n\n\t\t// OnHistoryUpdatedFunc is a function called when the historyCache was updated\n\t\tonHistoryUpdatedFunc HistoryUpdatedFunc\n\t}\n\n\t// HistoryUpdatedFunc is a type for notifying applications when the poller historyCache was updated\n\tHistoryUpdatedFunc func()\n)\n\nfunc NewPollerManager(historyUpdatedFunc HistoryUpdatedFunc, timeSource clock.TimeSource) Manager {\n\topts := &cache.Options{\n\t\tInitialCapacity: pollerHistoryInitSize,\n\t\tTTL:             pollerHistoryTTL,\n\t\tPin:             false,\n\t\tMaxCount:        pollerHistoryInitMaxSize,\n\t\tTimeSource:      timeSource,\n\t}\n\n\treturn &manager{\n\t\thistoryCache:             cache.New(opts),\n\t\ttimeSource:               timeSource,\n\t\tonHistoryUpdatedFunc:     historyUpdatedFunc,\n\t\tmostRecentPollEndByGroup: make(map[string]time.Time),\n\t\toutstandingCountByGroup:  make(map[string]int),\n\t\toutstanding:              make(map[string]outstandingPoller),\n\t}\n}\n\nfunc (m *manager) StartPoll(pollerID string, cancelFunc context.CancelFunc, info *Info) {\n\tif info.Identity != \"\" {\n\t\tm.historyCache.Put(info.Identity, &historicalPoller{\n\t\t\tinfo: info,\n\t\t\t// If there's no PollerID then we'll never have a subsequent EndPoll. Treat it like it isn't outstanding\n\t\t\t// so that we don't keep returning it forever\n\t\t\t// It doesn't seem like there's a possible code path where this happens\n\t\t\toutstanding: pollerID != \"\",\n\t\t})\n\t\tif m.onHistoryUpdatedFunc != nil {\n\t\t\tm.onHistoryUpdatedFunc()\n\t\t}\n\t}\n\tif pollerID != \"\" {\n\t\tm.lock.Lock()\n\t\tdefer m.lock.Unlock()\n\t\tm.outstanding[pollerID] = outstandingPoller{\n\t\t\tinfo:   info,\n\t\t\tcancel: cancelFunc,\n\t\t}\n\t\tif info.IsolationGroup != \"\" {\n\t\t\tm.outstandingCountByGroup[info.IsolationGroup]++\n\t\t}\n\t}\n}\n\nfunc (m *manager) EndPoll(pollerID string) {\n\tpoller, ok := m.tryRemovePoller(pollerID)\n\tif ok && poller.info.Identity != \"\" {\n\t\t// Refresh the cache to update the timestamp and clear outstanding value\n\t\tm.historyCache.Put(poller.info.Identity, &historicalPoller{\n\t\t\tinfo:        poller.info,\n\t\t\toutstanding: false,\n\t\t})\n\t\tif m.onHistoryUpdatedFunc != nil {\n\t\t\tm.onHistoryUpdatedFunc()\n\t\t}\n\t}\n}\n\nfunc (m *manager) tryRemovePoller(pollerID string) (outstandingPoller, bool) {\n\tm.lock.Lock()\n\tdefer m.lock.Unlock()\n\tnow := m.timeSource.Now()\n\tpoller, ok := m.outstanding[pollerID]\n\tif ok {\n\t\tdelete(m.outstanding, pollerID)\n\t\tif poller.info.IsolationGroup != \"\" {\n\t\t\tm.mostRecentPollEndByGroup[poller.info.IsolationGroup] = now\n\t\t\tm.outstandingCountByGroup[poller.info.IsolationGroup]--\n\t\t}\n\t}\n\t// reset the mostRecentPollEnd even if we didn't find the poller. They might not have specified a PollerID.\n\t// It doesn't seem possible outside of tests, but there's no harm in being safe\n\tm.mostRecentPollEnd = now\n\treturn poller, ok\n}\n\nfunc (m *manager) CancelPoll(pollerID string) bool {\n\tpoller, ok := m.tryGetPoller(pollerID)\n\tif ok {\n\t\tpoller.cancel()\n\t}\n\treturn ok\n}\n\nfunc (m *manager) tryGetPoller(pollerID string) (outstandingPoller, bool) {\n\tm.lock.RLock()\n\tdefer m.lock.RUnlock()\n\tpoller, ok := m.outstanding[pollerID]\n\treturn poller, ok\n}\n\nfunc (m *manager) HasPollerFromIsolationGroupAfter(isolationGroup string, earliest time.Time) bool {\n\tm.lock.RLock()\n\tdefer m.lock.RUnlock()\n\treturn m.outstandingCountByGroup[isolationGroup] > 0 || m.mostRecentPollEndByGroup[isolationGroup].After(earliest)\n}\n\nfunc (m *manager) HasPollerAfter(earliestAccessTime time.Time) bool {\n\tm.lock.RLock()\n\tdefer m.lock.RUnlock()\n\treturn len(m.outstanding) > 0 || m.mostRecentPollEnd.After(earliestAccessTime)\n}\n\nfunc (m *manager) GetCount() int {\n\treturn m.historyCache.Size()\n}\n\nfunc (m *manager) GetCountByIsolationGroup(after time.Time) map[string]int {\n\tgroupSet := make(map[string]int)\n\n\tm.forEachPoller(after, func(identity string, info *Info, lastAccessTime time.Time) {\n\t\tif info.IsolationGroup != \"\" {\n\t\t\tgroupSet[info.IsolationGroup]++\n\t\t}\n\t})\n\n\treturn groupSet\n}\n\nfunc (m *manager) ListInfo() []*types.PollerInfo {\n\tvar result []*types.PollerInfo\n\t// optimistic size get, it can change before Iterator call.\n\tsize := m.historyCache.Size()\n\tresult = make([]*types.PollerInfo, 0, size)\n\n\tm.forEachPoller(time.Time{}, func(identity string, info *Info, lastAccessTime time.Time) {\n\t\tresult = append(result, &types.PollerInfo{\n\t\t\tIdentity:       identity,\n\t\t\tLastAccessTime: common.Int64Ptr(lastAccessTime.UnixNano()),\n\t\t\tRatePerSecond:  info.RatePerSecond,\n\t\t})\n\t})\n\n\treturn result\n}\n\nfunc (m *manager) forEachPoller(after time.Time, callback func(string, *Info, time.Time)) {\n\tite := m.historyCache.Iterator()\n\tdefer ite.Close()\n\n\tfor ite.HasNext() {\n\t\tentry := ite.Next()\n\t\tkey := entry.Key().(string)\n\t\tvalue := entry.Value().(*historicalPoller)\n\t\tlastAccessTime := entry.CreateTime()\n\t\tif after.IsZero() || value.outstanding || after.Before(lastAccessTime) {\n\t\t\tcallback(key, value.info, lastAccessTime)\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/matching/poller/manager_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage poller\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar NoopFunc = func() {}\n\nfunc TestManager_HistoryCallback(t *testing.T) {\n\tmockTime := clock.NewMockedTimeSource()\n\tcounter := 0\n\tm := NewPollerManager(func() {\n\t\tcounter++\n\t}, mockTime)\n\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"a\"})\n\n\tassert.Equal(t, 1, counter)\n\n\tm.EndPoll(\"a\")\n\tassert.Equal(t, 2, counter)\n\n\t// Require identity\n\tcounter = 0\n\tm.StartPoll(\"b\", NoopFunc, &Info{})\n\n\tassert.Equal(t, 0, counter)\n}\n\nfunc TestManager_CancelPoll(t *testing.T) {\n\tt.Run(\"happy path\", func(t *testing.T) {\n\t\tmockTime := clock.NewMockedTimeSource()\n\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\tcounter := 0\n\t\tm.StartPoll(\"a\", func() {\n\t\t\tcounter++\n\t\t}, &Info{})\n\n\t\tres := m.CancelPoll(\"a\")\n\n\t\tassert.Equal(t, true, res)\n\t\tassert.Equal(t, 1, counter)\n\t})\n\tt.Run(\"repeated\", func(t *testing.T) {\n\t\tmockTime := clock.NewMockedTimeSource()\n\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\tcounter := 0\n\t\tm.StartPoll(\"a\", func() {\n\t\t\tcounter++\n\t\t}, &Info{})\n\n\t\t// Cancel doesn't remove it from the outstanding\n\t\tres := m.CancelPoll(\"a\")\n\t\tres2 := m.CancelPoll(\"a\")\n\n\t\tassert.Equal(t, true, res)\n\t\tassert.Equal(t, true, res2)\n\t\tassert.Equal(t, 2, counter)\n\t})\n\tt.Run(\"unknown\", func(t *testing.T) {\n\t\tmockTime := clock.NewMockedTimeSource()\n\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\tcounter := 0\n\t\tm.StartPoll(\"b\", func() {\n\t\t\tcounter++\n\t\t}, &Info{})\n\n\t\tres := m.CancelPoll(\"a\")\n\n\t\tassert.Equal(t, false, res)\n\t\tassert.Equal(t, 0, counter)\n\t})\n}\n\nfunc TestManager_HasPollerFromIsolationGroupAfter(t *testing.T) {\n\tstartTime := time.Date(2024, time.October, 28, 0, 0, 0, 0, time.UTC)\n\tgroup := \"the group\"\n\tcases := []struct {\n\t\tname   string\n\t\tfn     func(mockTime clock.MockedTimeSource, m Manager)\n\t\tafter  time.Time\n\t\tresult bool\n\t}{\n\t\t{\n\t\t\tname: \"outstanding poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{IsolationGroup: group})\n\t\t\t},\n\t\t\tafter:  startTime,\n\t\t\tresult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"recent poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{IsolationGroup: group})\n\t\t\t\tmockTime.Advance(time.Second)\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime,\n\t\t\tresult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"boundary condition poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{IsolationGroup: group})\n\t\t\t\tmockTime.Advance(time.Nanosecond)\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\t// Needs to after this time, not at this time\n\t\t\tafter:  startTime.Add(time.Nanosecond),\n\t\t\tresult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"expired poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{IsolationGroup: group})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime.Add(time.Minute),\n\t\t\tresult: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"never polled\",\n\t\t\tfn:     func(mockTime clock.MockedTimeSource, m Manager) {},\n\t\t\tafter:  startTime,\n\t\t\tresult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"unrelated poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{IsolationGroup: \"other\"})\n\t\t\t\tmockTime.Advance(time.Second)\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime,\n\t\t\tresult: false,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockTime := clock.NewMockedTimeSourceAt(startTime)\n\t\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\t\ttc.fn(mockTime, m)\n\t\t\tassert.Equal(t, tc.result, m.HasPollerFromIsolationGroupAfter(group, tc.after))\n\t\t})\n\t}\n}\n\nfunc TestManager_HasPollerAfter(t *testing.T) {\n\tstartTime := time.Date(2024, time.October, 28, 0, 0, 0, 0, time.UTC)\n\tcases := []struct {\n\t\tname   string\n\t\tfn     func(mockTime clock.MockedTimeSource, m Manager)\n\t\tafter  time.Time\n\t\tresult bool\n\t}{\n\t\t{\n\t\t\tname: \"outstanding poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{})\n\t\t\t},\n\t\t\tafter:  startTime,\n\t\t\tresult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"recent poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{})\n\t\t\t\tmockTime.Advance(time.Second)\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime,\n\t\t\tresult: true,\n\t\t},\n\t\t{\n\t\t\tname: \"boundary condition poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{})\n\t\t\t\tmockTime.Advance(time.Nanosecond)\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\t// Needs to after this time, not at this time\n\t\t\tafter:  startTime.Add(time.Nanosecond),\n\t\t\tresult: false,\n\t\t},\n\t\t{\n\t\t\tname: \"expired poller\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime.Add(time.Minute),\n\t\t\tresult: false,\n\t\t},\n\t\t{\n\t\t\tname:   \"never polled\",\n\t\t\tfn:     func(mockTime clock.MockedTimeSource, m Manager) {},\n\t\t\tafter:  startTime,\n\t\t\tresult: false,\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockTime := clock.NewMockedTimeSourceAt(startTime)\n\t\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\t\ttc.fn(mockTime, m)\n\t\t\tassert.Equal(t, tc.result, m.HasPollerAfter(tc.after))\n\t\t})\n\t}\n}\n\nfunc TestManager_GetCount(t *testing.T) {\n\tmockTime := clock.NewMockedTimeSource()\n\tm := NewPollerManager(NoopFunc, mockTime)\n\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"aIdent\"})\n\tm.EndPoll(\"a\")\n\tm.StartPoll(\"b\", NoopFunc, &Info{Identity: \"bIdent\"})\n\tmockTime.Advance(4 * time.Minute) // t = 4m\n\tm.EndPoll(\"b\")\n\tmockTime.Advance(2 * time.Minute) // t = 6m\n\tm.StartPoll(\"c\", NoopFunc, &Info{Identity: \"cIdent\"})\n\tm.EndPoll(\"c\")\n\tm.StartPoll(\"d\", NoopFunc, &Info{Identity: \"dIdent\"})\n\n\t// Since the cache doesn't actively evict, we still see a in the results\n\tassert.Equal(t, 4, m.GetCount())\n}\n\nfunc TestManager_ListInfo(t *testing.T) {\n\tstartTime := time.Date(2024, time.October, 28, 0, 0, 0, 0, time.UTC)\n\tcases := []struct {\n\t\tname   string\n\t\tfn     func(mockTime clock.MockedTimeSource, m Manager)\n\t\tresult []*types.PollerInfo\n\t}{\n\t\t{\n\t\t\tname: \"happy path\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"aIdent\"})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t\tm.StartPoll(\"b\", NoopFunc, &Info{Identity: \"bIdent\"})\n\t\t\t\tmockTime.Advance(time.Minute) // t = 1m\n\t\t\t\tm.EndPoll(\"b\")\n\t\t\t\tmockTime.Advance(time.Minute) // t = 2m\n\t\t\t\tm.StartPoll(\"c\", NoopFunc, &Info{Identity: \"cIdent\", RatePerSecond: 1.0})\n\t\t\t\tm.EndPoll(\"c\")\n\t\t\t\tm.StartPoll(\"d\", NoopFunc, &Info{Identity: \"dIdent\"})\n\t\t\t},\n\t\t\tresult: []*types.PollerInfo{\n\t\t\t\t{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(startTime.Add(2 * time.Minute).UnixNano()),\n\t\t\t\t\tIdentity:       \"dIdent\",\n\t\t\t\t\tRatePerSecond:  0,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(startTime.Add(2 * time.Minute).UnixNano()),\n\t\t\t\t\tIdentity:       \"cIdent\",\n\t\t\t\t\tRatePerSecond:  1.0,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(startTime.Add(time.Minute).UnixNano()),\n\t\t\t\t\tIdentity:       \"bIdent\",\n\t\t\t\t\tRatePerSecond:  0,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(startTime.UnixNano()),\n\t\t\t\t\tIdentity:       \"aIdent\",\n\t\t\t\t\tRatePerSecond:  0,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"exclude pollers with no identity\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"\"})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tresult: []*types.PollerInfo{},\n\t\t},\n\t\t{\n\t\t\tname: \"include pollers with no pollerID\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"\", NoopFunc, &Info{Identity: \"aIdent\"})\n\t\t\t},\n\t\t\tresult: []*types.PollerInfo{\n\t\t\t\t{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(startTime.UnixNano()),\n\t\t\t\t\tIdentity:       \"aIdent\",\n\t\t\t\t\tRatePerSecond:  0,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockTime := clock.NewMockedTimeSourceAt(startTime)\n\t\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\t\ttc.fn(mockTime, m)\n\t\t\tassert.Equal(t, tc.result, m.ListInfo())\n\t\t})\n\t}\n}\n\nfunc TestManager_GetCountByIsolationGroup(t *testing.T) {\n\tstartTime := time.Date(2024, time.October, 28, 0, 0, 0, 0, time.UTC)\n\tcases := []struct {\n\t\tname   string\n\t\tfn     func(mockTime clock.MockedTimeSource, m Manager)\n\t\tafter  time.Time\n\t\tresult map[string]int\n\t}{\n\t\t{\n\t\t\tname: \"happy path\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"aIdent\", IsolationGroup: \"groupA\"})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t\tm.StartPoll(\"b\", NoopFunc, &Info{Identity: \"bIdent\", IsolationGroup: \"groupA\"})\n\t\t\t\tmockTime.Advance(time.Minute) // t = 1m\n\t\t\t\tm.EndPoll(\"b\")\n\t\t\t\tmockTime.Advance(time.Minute) // t = 2m\n\t\t\t\tm.StartPoll(\"c\", NoopFunc, &Info{Identity: \"cIdent\", RatePerSecond: 1.0, IsolationGroup: \"groupB\"})\n\t\t\t\tm.EndPoll(\"c\")\n\t\t\t\tm.StartPoll(\"d\", NoopFunc, &Info{Identity: \"dIdent\"})\n\t\t\t},\n\t\t\tresult: map[string]int{\n\t\t\t\t\"groupA\": 2,\n\t\t\t\t\"groupB\": 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"some expired\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"aIdent\", IsolationGroup: \"groupA\"})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t\tm.StartPoll(\"b\", NoopFunc, &Info{Identity: \"bIdent\", IsolationGroup: \"groupA\"})\n\t\t\t\tmockTime.Advance(time.Minute) // t = 1m\n\t\t\t\tm.EndPoll(\"b\")\n\t\t\t\tmockTime.Advance(time.Minute) // t = 2m\n\t\t\t\tm.StartPoll(\"c\", NoopFunc, &Info{Identity: \"cIdent\", RatePerSecond: 1.0, IsolationGroup: \"groupB\"})\n\t\t\t\tm.EndPoll(\"c\")\n\t\t\t\tm.StartPoll(\"d\", NoopFunc, &Info{Identity: \"dIdent\"})\n\t\t\t},\n\t\t\tafter: startTime.Add(time.Minute - time.Nanosecond),\n\t\t\tresult: map[string]int{\n\t\t\t\t\"groupA\": 1,\n\t\t\t\t\"groupB\": 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"exclude pollers with no identity\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"a\", NoopFunc, &Info{Identity: \"\"})\n\t\t\t\tm.EndPoll(\"a\")\n\t\t\t},\n\t\t\tafter:  startTime.Add(-time.Nanosecond),\n\t\t\tresult: map[string]int{},\n\t\t},\n\t\t{\n\t\t\tname: \"include pollers with no pollerID\",\n\t\t\tfn: func(mockTime clock.MockedTimeSource, m Manager) {\n\t\t\t\tm.StartPoll(\"\", NoopFunc, &Info{Identity: \"aIdent\", IsolationGroup: \"groupA\"})\n\t\t\t},\n\t\t\tafter: startTime.Add(-time.Nanosecond),\n\t\t\tresult: map[string]int{\n\t\t\t\t\"groupA\": 1,\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockTime := clock.NewMockedTimeSourceAt(startTime)\n\t\t\tm := NewPollerManager(NoopFunc, mockTime)\n\t\t\ttc.fn(mockTime, m)\n\t\t\tassert.Equal(t, tc.result, m.GetCountByIsolationGroup(tc.after))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/service.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage matching\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/handler\"\n\t\"github.com/uber/cadence/service/matching/wrappers/grpc\"\n\t\"github.com/uber/cadence/service/matching/wrappers/thrift\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n)\n\n// Service represents the cadence-matching service\ntype Service struct {\n\tresource.Resource\n\n\tstatus                         int32\n\thandler                        handler.Handler\n\tstopC                          chan struct{}\n\tconfig                         *config.Config\n\tShardDistributorMatchingConfig clientcommon.Config\n\tdrainObserver                  clientcommon.DrainSignalObserver\n}\n\n// NewService builds a new cadence-matching service\nfunc NewService(\n\tparams *resource.Params,\n) (resource.Resource, error) {\n\n\tserviceConfig := config.NewConfig(\n\t\tdynamicconfig.NewCollection(\n\t\t\tparams.DynamicConfig,\n\t\t\tparams.Logger,\n\t\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t\t),\n\t\tparams.HostName,\n\t\tparams.RPCConfig,\n\t\tparams.GetIsolationGroups,\n\t)\n\n\tserviceResource, err := resource.New(\n\t\tparams,\n\t\tservice.Matching,\n\t\t&service.Config{\n\t\t\tPersistenceMaxQPS:        serviceConfig.PersistenceMaxQPS,\n\t\t\tPersistenceGlobalMaxQPS:  serviceConfig.PersistenceGlobalMaxQPS,\n\t\t\tThrottledLoggerMaxRPS:    serviceConfig.ThrottledLogRPS,\n\t\t\tIsErrorRetryableFunction: common.IsServiceTransientError,\n\t\t\t// matching doesn't need visibility config as it never read or write visibility\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Service{\n\t\tResource:                       serviceResource,\n\t\tstatus:                         common.DaemonStatusInitialized,\n\t\tconfig:                         serviceConfig,\n\t\tstopC:                          make(chan struct{}),\n\t\tShardDistributorMatchingConfig: params.ShardDistributorMatchingConfig,\n\t\tdrainObserver:                  params.DrainObserver,\n\t}, nil\n}\n\n// Start starts the service\nfunc (s *Service) Start() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tlogger := s.GetLogger()\n\tlogger.Info(\"matching starting\")\n\n\tengine := handler.NewEngine(\n\t\ts.GetTaskManager(),\n\t\ts.GetClusterMetadata(),\n\t\ts.GetHistoryClient(),\n\t\ts.GetMatchingRawClient(), // Use non retry client inside matching\n\t\ts.config,\n\t\ts.GetLogger(),\n\t\ts.GetMetricsClient(),\n\t\ts.GetMetricsScope(),\n\t\ts.GetDomainCache(),\n\t\ts.GetMembershipResolver(),\n\t\ts.GetIsolationGroupState(),\n\t\ts.GetTimeSource(),\n\t\ts.GetShardDistributorExecutorClient(),\n\t\ts.ShardDistributorMatchingConfig,\n\t\ts.drainObserver,\n\t)\n\n\ts.handler = handler.NewHandler(engine, s.config, s.GetDomainCache(), s.GetMetricsClient(), s.GetLogger(), s.GetThrottledLogger())\n\n\tthriftHandler := thrift.NewThriftHandler(s.handler)\n\tthriftHandler.Register(s.GetDispatcher())\n\n\tgrpcHandler := grpc.NewGRPCHandler(s.handler)\n\tgrpcHandler.Register(s.GetDispatcher())\n\n\t// must start base service first\n\ts.Resource.Start()\n\ts.handler.Start()\n\n\tlogger.Info(\"matching started\")\n\n\t<-s.stopC\n}\n\n// Stop stops the service\nfunc (s *Service) Stop() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\t// remove self from membership ring and wait for traffic to drain\n\ts.GetLogger().Info(\"ShutdownHandler: Evicting self from membership ring\")\n\ts.GetMembershipResolver().EvictSelf()\n\ts.GetLogger().Info(\"ShutdownHandler: Waiting for others to discover I am unhealthy\")\n\ttime.Sleep(s.config.ShutdownDrainDuration())\n\n\tclose(s.stopC)\n\n\ts.handler.Stop()\n\ts.Resource.Stop()\n\n\ts.GetLogger().Info(\"matching stopped\")\n}\n"
  },
  {
    "path": "service/matching/tasklist/adaptive_scaler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\ntype (\n\tAdaptiveScaler interface {\n\t\tcommon.Daemon\n\t}\n\n\tadaptiveScalerImpl struct {\n\t\ttaskListID     *Identifier\n\t\ttlMgr          Manager\n\t\tconfig         *config.TaskListConfig\n\t\ttimeSource     clock.TimeSource\n\t\tlogger         log.Logger\n\t\tscope          metrics.Scope\n\t\tmatchingClient matching.Client\n\n\t\ttaskListType *types.TaskListType\n\t\tstatus       int32\n\t\twg           sync.WaitGroup\n\t\tctx          context.Context\n\t\tcancel       func()\n\t\toverLoad     clock.Sustain\n\t\tunderLoad    clock.Sustain\n\t\tisolation    *isolationBalancer\n\t\tbaseEvent    event.E\n\t}\n\n\taggregatePartitionMetrics struct {\n\t\ttotalQPS                   float64\n\t\tqpsByIsolationGroup        map[string]float64\n\t\thasPollersByIsolationGroup map[string]bool\n\t\tbyPartition                map[int]*partitionMetrics\n\t\tisIsolationEnabled         bool\n\t}\n\n\tpartitionMetrics struct {\n\t\tqps      float64\n\t\treadOnly bool\n\t\tempty    bool\n\t}\n)\n\nfunc NewAdaptiveScaler(\n\ttaskListID *Identifier,\n\ttlMgr Manager,\n\tconfig *config.TaskListConfig,\n\ttimeSource clock.TimeSource,\n\tlogger log.Logger,\n\tscope metrics.Scope,\n\tmatchingClient matching.Client,\n\tbaseEvent event.E,\n) AdaptiveScaler {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &adaptiveScalerImpl{\n\t\ttaskListID:     taskListID,\n\t\ttlMgr:          tlMgr,\n\t\tconfig:         config,\n\t\ttimeSource:     timeSource,\n\t\tlogger:         logger.WithTags(tag.ComponentTaskListAdaptiveScaler),\n\t\tscope:          scope,\n\t\tmatchingClient: matchingClient,\n\t\ttaskListType:   getTaskListType(taskListID.GetType()),\n\t\tctx:            ctx,\n\t\tcancel:         cancel,\n\t\tisolation:      newIsolationBalancer(timeSource, scope, config),\n\t\toverLoad:       clock.NewSustain(timeSource, config.PartitionUpscaleSustainedDuration),\n\t\tunderLoad:      clock.NewSustain(timeSource, config.PartitionDownscaleSustainedDuration),\n\t\tbaseEvent:      baseEvent,\n\t}\n}\n\nfunc (a *adaptiveScalerImpl) Start() {\n\tif !atomic.CompareAndSwapInt32(&a.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\ta.logger.Info(\"adaptive task list scaler state changed\", tag.LifeCycleStarted)\n\ta.wg.Add(1)\n\tgo a.runPeriodicLoop()\n}\n\nfunc (a *adaptiveScalerImpl) Stop() {\n\tif !atomic.CompareAndSwapInt32(&a.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\ta.cancel()\n\ta.wg.Wait()\n\ta.logger.Info(\"adaptive task list scaler state changed\", tag.LifeCycleStopped)\n}\n\nfunc (a *adaptiveScalerImpl) runPeriodicLoop() {\n\tdefer a.wg.Done()\n\ttimer := a.timeSource.NewTimer(a.config.AdaptiveScalerUpdateInterval())\n\tdefer timer.Stop()\n\tfor {\n\t\tselect {\n\t\tcase <-a.ctx.Done():\n\t\t\treturn\n\t\tcase <-timer.Chan():\n\t\t\ta.run()\n\t\t\ttimer.Reset(a.config.AdaptiveScalerUpdateInterval())\n\t\t}\n\t}\n}\n\nfunc (a *adaptiveScalerImpl) run() {\n\tif !a.config.EnableAdaptiveScaler() || !a.config.EnableGetNumberOfPartitionsFromCache() {\n\t\treturn\n\t}\n\n\tpartitionConfig := a.getPartitionConfig()\n\tm, err := a.collectPartitionMetrics(partitionConfig)\n\t// TODO: Handle this better. Maybe we should allow scaling up but not down if our data is incomplete?\n\tif err != nil {\n\t\ta.underLoad.Reset()\n\t\ta.overLoad.Reset()\n\t\ta.isolation.reset()\n\t\ta.logger.Error(\"Failed to collect partition metrics\", tag.Error(err))\n\t\treturn\n\t}\n\t// adjust the number of write partitions based on qps\n\tnumWritePartitions := a.calculateWritePartitionCount(m.totalQPS, len(partitionConfig.WritePartitions))\n\twritePartitions, writeChanged := a.adjustWritePartitions(partitionConfig.WritePartitions, numWritePartitions)\n\n\tisolationChanged := false\n\tif m.isIsolationEnabled {\n\t\twritePartitions, isolationChanged = a.isolation.adjustWritePartitions(m, writePartitions)\n\t\tif isolationChanged {\n\t\t\ta.scope.IncCounter(metrics.IsolationRebalance)\n\t\t}\n\t} else {\n\t\ta.isolation.reset()\n\t}\n\n\t// adjustReadPartitions will copy over any changes to the writePartitions, so it needs to happen after we\n\t// potentially change the isolation group assignments\n\treadPartitions, readChanged := a.adjustReadPartitions(m, partitionConfig.ReadPartitions, writePartitions)\n\n\te := a.baseEvent\n\te.EventName = \"AdaptiveScalerCalculationResult\"\n\te.Payload = map[string]any{\n\t\t\"NumReadPartitions\":  len(readPartitions),\n\t\t\"NumWritePartitions\": len(writePartitions),\n\t\t\"ReadChanged\":        readChanged,\n\t\t\"WriteChanged\":       writeChanged,\n\t\t\"IsolationChanged\":   isolationChanged,\n\t\t\"QPS\":                m.totalQPS,\n\t}\n\tevent.Log(e)\n\n\tif !writeChanged && !readChanged && !isolationChanged {\n\t\treturn\n\t}\n\tnewConfig := &types.TaskListPartitionConfig{\n\t\tReadPartitions:  readPartitions,\n\t\tWritePartitions: writePartitions,\n\t}\n\ta.logger.Info(\"adaptive scaler is updating number of partitions\",\n\t\ttag.CurrentQPS(m.totalQPS),\n\t\ttag.NumReadPartitions(len(readPartitions)),\n\t\ttag.NumWritePartitions(len(writePartitions)),\n\t\ttag.ReadChanged(readChanged),\n\t\ttag.WriteChanged(writeChanged),\n\t\ttag.IsolationChanged(isolationChanged),\n\t\ttag.Dynamic(\"old-partition-config\", partitionConfig),\n\t\ttag.Dynamic(\"new-partition-config\", newConfig),\n\t)\n\ta.scope.IncCounter(metrics.CadenceRequests)\n\terr = a.tlMgr.UpdateTaskListPartitionConfig(a.ctx, newConfig)\n\tif err != nil {\n\t\ta.logger.Error(\"failed to update task list partition config\", tag.Error(err))\n\t\ta.scope.IncCounter(metrics.CadenceFailures)\n\t}\n}\n\nfunc (a *adaptiveScalerImpl) getPartitionConfig() *types.TaskListPartitionConfig {\n\tpartitionConfig := a.tlMgr.TaskListPartitionConfig()\n\tif partitionConfig == nil {\n\t\tpartitionConfig = &types.TaskListPartitionConfig{\n\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t},\n\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t},\n\t\t}\n\t}\n\treturn partitionConfig\n}\n\nfunc (a *adaptiveScalerImpl) calculateWritePartitionCount(qps float64, numWritePartitions int) int {\n\tupscaleRps := float64(a.config.PartitionUpscaleRPS())\n\tpartitions := float64(numWritePartitions)\n\tdownscaleFactor := a.config.PartitionDownscaleFactor()\n\tupscaleThreshold := partitions * upscaleRps\n\tdownscaleThreshold := (partitions - 1) * upscaleRps * downscaleFactor\n\ta.scope.UpdateGauge(metrics.EstimatedAddTaskQPSGauge, qps)\n\ta.scope.UpdateGauge(metrics.TaskListPartitionUpscaleThresholdGauge, upscaleThreshold)\n\ta.scope.UpdateGauge(metrics.TaskListPartitionDownscaleThresholdGauge, downscaleThreshold)\n\n\tresult := numWritePartitions\n\tif a.overLoad.CheckAndReset(qps > upscaleThreshold) {\n\t\tresult = getNumberOfPartitions(qps, upscaleRps)\n\t\ta.scope.IncCounter(metrics.PartitionUpscale)\n\t\ta.logger.Info(\"adjust write partitions\", tag.CurrentQPS(qps), tag.PartitionUpscaleThreshold(upscaleThreshold), tag.PartitionDownscaleThreshold(downscaleThreshold), tag.PartitionDownscaleFactor(downscaleFactor), tag.CurrentNumWritePartitions(numWritePartitions), tag.NumWritePartitions(result))\n\t}\n\tif a.underLoad.CheckAndReset(qps < downscaleThreshold) {\n\t\tresult = getNumberOfPartitions(qps, upscaleRps)\n\t\ta.scope.IncCounter(metrics.PartitionDownscale)\n\t\ta.logger.Info(\"adjust write partitions\", tag.CurrentQPS(qps), tag.PartitionUpscaleThreshold(upscaleThreshold), tag.PartitionDownscaleThreshold(downscaleThreshold), tag.PartitionDownscaleFactor(downscaleFactor), tag.CurrentNumWritePartitions(numWritePartitions), tag.NumWritePartitions(result))\n\n\t}\n\treturn result\n}\n\nfunc (a *adaptiveScalerImpl) adjustWritePartitions(writePartitions map[int]*types.TaskListPartition, targetWritePartitions int) (map[int]*types.TaskListPartition, bool) {\n\tif len(writePartitions) == targetWritePartitions {\n\t\treturn writePartitions, false\n\t}\n\tresult := make(map[int]*types.TaskListPartition, targetWritePartitions)\n\n\tfor i := 0; i < targetWritePartitions; i++ {\n\t\tif p, ok := writePartitions[i]; ok {\n\t\t\tresult[i] = p\n\t\t} else {\n\t\t\tresult[i] = &types.TaskListPartition{}\n\t\t}\n\t}\n\treturn result, true\n}\n\nfunc (a *adaptiveScalerImpl) adjustReadPartitions(m *aggregatePartitionMetrics, oldReadPartitions map[int]*types.TaskListPartition, newWritePartitions map[int]*types.TaskListPartition) (map[int]*types.TaskListPartition, bool) {\n\tresult := make(map[int]*types.TaskListPartition, len(newWritePartitions))\n\tchanged := false\n\tfor id, p := range oldReadPartitions {\n\t\tresult[id] = p\n\t}\n\tfor id, p := range newWritePartitions {\n\t\tif _, ok := result[id]; !ok {\n\t\t\tchanged = true\n\t\t}\n\t\tresult[id] = p\n\t}\n\n\tfor i := len(result) - 1; i >= len(newWritePartitions); i-- {\n\t\tp, ok := m.byPartition[i]\n\t\tif !ok {\n\t\t\tresp, err := a.describePartition(i)\n\t\t\tif err != nil {\n\t\t\t\ta.logger.Warn(\"failed to get partition metrics\", tag.WorkflowTaskListName(a.taskListID.GetPartition(i)), tag.Error(err))\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tp = a.toPartitionMetrics(i, resp)\n\t\t}\n\t\tif p.readOnly && p.empty {\n\t\t\tchanged = true\n\t\t\tdelete(result, i)\n\t\t\ta.scope.IncCounter(metrics.PartitionDrained)\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\tif changed {\n\t\ta.logger.Info(\"adjust read partitions\", tag.NumReadPartitions(len(result)), tag.NumWritePartitions(len(newWritePartitions)))\n\t}\n\treturn result, changed\n}\n\nfunc (a *adaptiveScalerImpl) collectPartitionMetrics(config *types.TaskListPartitionConfig) (*aggregatePartitionMetrics, error) {\n\tif a.config.EnablePartitionIsolationGroupAssignment() {\n\t\treturn a.fetchMetricsFromPartitions(config)\n\t}\n\treturn a.assumeEvenQPS(config)\n}\n\nfunc (a *adaptiveScalerImpl) assumeEvenQPS(config *types.TaskListPartitionConfig) (*aggregatePartitionMetrics, error) {\n\tresp := a.tlMgr.DescribeTaskList(true)\n\n\ttotalQPS := resp.TaskListStatus.NewTasksPerSecond * float64(len(config.WritePartitions))\n\n\treturn &aggregatePartitionMetrics{\n\t\ttotalQPS:           totalQPS,\n\t\tisIsolationEnabled: false,\n\t}, nil\n}\n\nfunc (a *adaptiveScalerImpl) fetchMetricsFromPartitions(config *types.TaskListPartitionConfig) (*aggregatePartitionMetrics, error) {\n\tvar mutex sync.Mutex\n\tresults := make(map[int]*types.DescribeTaskListResponse, len(config.ReadPartitions))\n\tg := &errgroup.Group{}\n\tfor p := range config.ReadPartitions {\n\t\tpartitionID := p\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), a.logger, &e) }()\n\n\t\t\tresult, e := a.describePartition(partitionID)\n\t\t\tif e != nil {\n\t\t\t\ta.logger.Warn(\"failed to get partition metrics\", tag.WorkflowTaskListName(a.taskListID.GetPartition(partitionID)), tag.Error(e))\n\t\t\t}\n\t\t\tif result != nil {\n\t\t\t\tmutex.Lock()\n\t\t\t\tdefer mutex.Unlock()\n\t\t\t\tresults[partitionID] = result\n\t\t\t}\n\t\t\treturn e\n\t\t})\n\t}\n\terr := g.Wait()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn a.toAggregateMetrics(results), err\n}\n\nfunc (a *adaptiveScalerImpl) describePartition(partitionID int) (*types.DescribeTaskListResponse, error) {\n\tif partitionID == 0 {\n\t\treturn a.tlMgr.DescribeTaskList(true), nil\n\t}\n\treturn a.matchingClient.DescribeTaskList(a.ctx, &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID: a.taskListID.GetDomainID(),\n\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\tTaskListType: a.taskListType,\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: a.taskListID.GetPartition(partitionID),\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tIncludeTaskListStatus: true,\n\t\t},\n\t})\n}\n\nfunc (a *adaptiveScalerImpl) toAggregateMetrics(partitions map[int]*types.DescribeTaskListResponse) *aggregatePartitionMetrics {\n\ttotal := 0.0\n\tbyIsolationGroup := make(map[string]float64)\n\thasPollersByIsolationGroup := make(map[string]bool)\n\tbyPartition := make(map[int]*partitionMetrics, len(partitions))\n\tfor id, p := range partitions {\n\t\tfor ig, groupMetrics := range p.TaskListStatus.IsolationGroupMetrics {\n\t\t\tbyIsolationGroup[ig] += groupMetrics.NewTasksPerSecond\n\t\t\thasPollersByIsolationGroup[ig] = hasPollersByIsolationGroup[ig] || groupMetrics.PollerCount > 0\n\t\t}\n\t\ttotal += p.TaskListStatus.NewTasksPerSecond\n\n\t\tbyPartition[id] = a.toPartitionMetrics(id, p)\n\t}\n\treturn &aggregatePartitionMetrics{\n\t\ttotalQPS:                   total,\n\t\tqpsByIsolationGroup:        byIsolationGroup,\n\t\thasPollersByIsolationGroup: hasPollersByIsolationGroup,\n\t\tbyPartition:                byPartition,\n\t\tisIsolationEnabled:         true,\n\t}\n}\n\nfunc (a *adaptiveScalerImpl) toPartitionMetrics(id int, p *types.DescribeTaskListResponse) *partitionMetrics {\n\thasWritePartition := true\n\tif p.PartitionConfig != nil {\n\t\t_, hasWritePartition = p.PartitionConfig.WritePartitions[id]\n\t}\n\tvar empty bool\n\tif a.config.EnablePartitionEmptyCheck() {\n\t\tempty = p.TaskListStatus.Empty\n\t} else {\n\t\tempty = p.TaskListStatus.BacklogCountHint == 0\n\t}\n\treturn &partitionMetrics{\n\t\tqps:      p.TaskListStatus.NewTasksPerSecond,\n\t\tempty:    empty,\n\t\treadOnly: !hasWritePartition,\n\t}\n}\n\nfunc getTaskListType(taskListType int) *types.TaskListType {\n\tif taskListType == persistence.TaskListTypeDecision {\n\t\treturn types.TaskListTypeDecision.Ptr()\n\t} else if taskListType == persistence.TaskListTypeActivity {\n\t\treturn types.TaskListTypeActivity.Ptr()\n\t}\n\treturn nil\n}\n\nfunc getNumberOfPartitions(qps float64, upscaleQPS float64) int {\n\tp := int(math.Ceil(qps / upscaleQPS))\n\tif p <= 0 {\n\t\tp = 1\n\t}\n\treturn p\n}\n"
  },
  {
    "path": "service/matching/tasklist/adaptive_scaler_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/stats\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\ntype mockAdaptiveScalerDeps struct {\n\tid                 *Identifier\n\tmockManager        *MockManager\n\tmockQPSTracker     *stats.MockQPSTracker\n\tmockTimeSource     clock.MockedTimeSource\n\tmockMatchingClient *matching.MockClient\n\tdynamicClient      dynamicconfig.Client\n\n\tconfig *config.TaskListConfig\n\tlogger log.Logger\n\tscope  metrics.Scope\n}\n\nfunc setupMocksForAdaptiveScaler(t *testing.T, taskListID *Identifier) (*adaptiveScalerImpl, *mockAdaptiveScalerDeps) {\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\tscope := metrics.NoopScope\n\tmockManager := NewMockManager(ctrl)\n\tmockQPSTracker := stats.NewMockQPSTracker(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSourceAt(time.Now())\n\tmockMatchingClient := matching.NewMockClient(ctrl)\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\tcfg := newTaskListConfig(taskListID, config.NewConfig(dynamicconfig.NewCollection(dynamicClient, logger), \"test-host\", commonConfig.RPC{}, func() []string { return nil }), \"test-domain\")\n\n\tdeps := &mockAdaptiveScalerDeps{\n\t\tid:                 taskListID,\n\t\tmockManager:        mockManager,\n\t\tmockQPSTracker:     mockQPSTracker,\n\t\tmockTimeSource:     mockTimeSource,\n\t\tmockMatchingClient: mockMatchingClient,\n\t\tdynamicClient:      dynamicClient,\n\t\tconfig:             cfg,\n\t}\n\n\tscaler := NewAdaptiveScaler(taskListID, mockManager, cfg, mockTimeSource, logger, scope, mockMatchingClient, event.E{}).(*adaptiveScalerImpl)\n\treturn scaler, deps\n}\n\nfunc TestAdaptiveScalerLifecycle(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttaskListID, err := NewIdentifier(\"test-domain-id\", \"test-task-list\", 0)\n\trequire.NoError(t, err)\n\tscaler, _ := setupMocksForAdaptiveScaler(t, taskListID)\n\n\t// test idempotency\n\tassert.NotPanics(t, scaler.Start)\n\tassert.NotPanics(t, scaler.Start)\n\tassert.NotPanics(t, scaler.Stop)\n\tassert.NotPanics(t, scaler.Stop)\n}\n\nfunc TestAdaptiveScalerRun(t *testing.T) {\n\ttestCases := []struct {\n\t\tname      string\n\t\tmockSetup func(*mockAdaptiveScalerDeps)\n\t\tcycles    int\n\t}{\n\t\t{\n\t\t\tname: \"no op\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t},\n\t\t\tcycles: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"overload start\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 300))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t},\n\t\t\tcycles: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"overload sustained\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\t// overload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 300))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\n\t\t\t\t// overload passing sustained period\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 300))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"overload fluctuate\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\t// overload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 300))\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t\t// load back to normal\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 100))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t\t// overload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 300))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t\t// load back to normal\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 100))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t},\n\t\t\tcycles: 4,\n\t\t},\n\t\t{\n\t\t\tname: \"underload start\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(10, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(10),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\t\t\t},\n\t\t\tcycles: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"underload sustained\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(10, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(10),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(10, 0))\n\t\t\t\t// Partition 9 will be checked if it is drained, but it won't have received the update yet\n\t\t\t\tmockDescribeTaskList(deps, 9, withPartitionsAndQPS(10, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(10),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"underload sustained then drain - require empty\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionEmptyCheck = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\t// Start of Cycle 1\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\n\t\t\t\t// Start of Cycle 2\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 0))\n\t\t\t\t// Partition 2 will be checked but won't be drained because it hasn't received the update yet\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndQPS(3, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t}).Return(nil)\n\n\t\t\t\t// Start of cycle 3\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndBacklog(3, 1, 0))\n\t\t\t\t// 2 will be checked and drained because Empty == true, BacklogCountHint is ignored\n\t\t\t\tmockDescribeTaskList(deps, 2, &types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:        nil,\n\t\t\t\t\tTaskListStatus: &types.TaskListStatus{NewTasksPerSecond: 0, BacklogCountHint: 1000, Empty: true},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\t// 1 will be checked and won't be drained because Empty == false, even though BacklogCountHint == 0\n\t\t\t\tmockDescribeTaskList(deps, 1, &types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:        nil,\n\t\t\t\t\tTaskListStatus: &types.TaskListStatus{NewTasksPerSecond: 0, BacklogCountHint: 0, Empty: false},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t})\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 3,\n\t\t},\n\t\t{\n\t\t\tname: \"underload sustained then drain\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(10, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(10),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(10, 0))\n\t\t\t\t// Partition 9 will be checked if it is drained, but it won't have received the update yet\n\t\t\t\tmockDescribeTaskList(deps, 9, withPartitionsAndQPS(10, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(10),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t}).Return(nil)\n\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 9, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 8, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 7, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 6, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 5, withPartitionsAndBacklog(10, 1, 0))\n\t\t\t\tmockDescribeTaskList(deps, 4, withPartitionsAndBacklog(10, 1, 1))\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\tReadPartitions:  partitions(5),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 3,\n\t\t},\n\t\t{\n\t\t\tname: \"overload but no fluctuation\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\t// overload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 210))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\n\t\t\t\t// overload passing sustained period\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(1, 210))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(nil)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t}).Return(nil)\n\n\t\t\t\t// not overload with 1 partition, but avoid fluctuation, so don't scale down\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(2, 190))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t})\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(2, 190))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t})\n\t\t\t},\n\t\t\tcycles: 4,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - aggregate metrics to scale up\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\t// overload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(2, 1))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(2, 400))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t})\n\n\t\t\t\t// overload passing sustained period\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(2, 1))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(2, 400))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - aggregate metrics to scale down\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\t// underload start\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 200))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(3, 49))\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndQPS(3, 50))\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\n\t\t\t\t// underload passing sustained period\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 200))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(3, 49))\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndQPS(3, 50))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t}).Return(nil)\n\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(2, 200))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(2, 99))\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndBacklog(3, 2, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 3,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - scale group up\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\tpartitionConfig := &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tpartition0Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"a\": 101,\n\t\t\t\t\t\"b\": 20,\n\t\t\t\t})\n\t\t\t\tpartition1Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"c\": 20,\n\t\t\t\t\t\"d\": 20,\n\t\t\t\t})\n\t\t\t\t// overload start for a\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\n\t\t\t\t// overload sustained\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\n\t\t\t},\n\t\t\tcycles: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - scale group down\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\tpartitionConfig := &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tpartition0Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\"b\": 50,\n\t\t\t\t})\n\t\t\t\tpartition1Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"c\": 50,\n\t\t\t\t\t\"d\": 50,\n\t\t\t\t})\n\t\t\t\t// overload start for a\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\n\t\t\t\t// overload sustained\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - scale partitions down\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\n\t\t\t\tpartitionConfig := &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion: 1,\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tpartition0Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\"b\": 20,\n\t\t\t\t})\n\t\t\t\tpartition1Resp := withConfigAndQPS(partitionConfig, map[string]float64{\n\t\t\t\t\t\"c\": 20,\n\t\t\t\t\t\"d\": 20,\n\t\t\t\t})\n\t\t\t\t// overload start for a\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\n\t\t\t\t// overload sustained\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, partition1Resp)\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(partitionConfig)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\n\t\t\t\tdrainConfig := &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\t// Drain partition 1\n\t\t\t\tmockDescribeTaskList(deps, 0, partition0Resp)\n\t\t\t\tmockDescribeTaskList(deps, 1, withConfigAndQPS(drainConfig, nil))\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(drainConfig)\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 3,\n\t\t},\n\t\t{\n\t\t\tname: \"isolation - error calling DescribeTaskList results in no-op\",\n\t\t\tmockSetup: func(deps *mockAdaptiveScalerDeps) {\n\t\t\t\tdeps.config.EnablePartitionIsolationGroupAssignment = func() bool {\n\t\t\t\t\treturn true\n\t\t\t\t}\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskListWithErr(deps, 2, context.DeadlineExceeded)\n\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\n\t\t\t\t// underload would normally pass sustain period, but the error resets it\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndQPS(3, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\n\t\t\t\t// Now we can scale down\n\t\t\t\tmockDescribeTaskList(deps, 0, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskList(deps, 1, withPartitionsAndQPS(3, 0))\n\t\t\t\tmockDescribeTaskList(deps, 2, withPartitionsAndQPS(3, 0))\n\t\t\t\tdeps.mockManager.EXPECT().TaskListPartitionConfig().Return(&types.TaskListPartitionConfig{\n\t\t\t\t\tWritePartitions: partitions(3),\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t})\n\t\t\t\tdeps.mockManager.EXPECT().UpdateTaskListPartitionConfig(gomock.Any(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\tcycles: 3,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttaskListID, err := NewIdentifier(\"test-domain-id\", \"test-task-list\", 0)\n\t\t\trequire.NoError(t, err)\n\t\t\tscaler, deps := setupMocksForAdaptiveScaler(t, taskListID)\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingEnableAdaptiveScaler, true))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache, true))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionUpscaleRPS, 200))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionDownscaleFactor, 0.75))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionUpscaleSustainedDuration, time.Second))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionDownscaleSustainedDuration, time.Second))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupsPerPartition, 2))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupUpscaleSustainedDuration, time.Second))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupDownscaleSustainedDuration, time.Second))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupHasPollersSustainedDuration, time.Second))\n\t\t\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupNoPollersSustainedDuration, time.Second))\n\t\t\ttc.mockSetup(deps)\n\n\t\t\tfor i := 0; i < tc.cycles; i++ {\n\t\t\t\tscaler.run()\n\t\t\t\tdeps.mockTimeSource.Advance(time.Second)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc withConfigAndQPS(config *types.TaskListPartitionConfig, qpsByGroup map[string]float64) *types.DescribeTaskListResponse {\n\tisolationMetrics := make(map[string]*types.IsolationGroupMetrics)\n\ttotal := float64(0)\n\tfor group, qps := range qpsByGroup {\n\t\tisolationMetrics[group] = &types.IsolationGroupMetrics{\n\t\t\tNewTasksPerSecond: qps,\n\t\t\tPollerCount:       1,\n\t\t}\n\t\ttotal += qps\n\t}\n\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers: []*types.PollerInfo{},\n\t\tTaskListStatus: &types.TaskListStatus{\n\t\t\tBacklogCountHint:      0,\n\t\t\tReadLevel:             0,\n\t\t\tAckLevel:              0,\n\t\t\tRatePerSecond:         0,\n\t\t\tTaskIDBlock:           nil,\n\t\t\tIsolationGroupMetrics: isolationMetrics,\n\t\t\tNewTasksPerSecond:     total,\n\t\t},\n\t\tPartitionConfig: config,\n\t}\n}\n\nfunc withPartitionsAndQPS(numPartitions int, qps float64) *types.DescribeTaskListResponse {\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers:        nil,\n\t\tTaskListStatus: &types.TaskListStatus{NewTasksPerSecond: qps},\n\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\tReadPartitions:  partitions(numPartitions),\n\t\t\tWritePartitions: partitions(numPartitions),\n\t\t},\n\t}\n}\n\nfunc withPartitionsAndBacklog(numRead, numWrite int, backlog int64) *types.DescribeTaskListResponse {\n\treturn &types.DescribeTaskListResponse{\n\t\tPollers:        nil,\n\t\tTaskListStatus: &types.TaskListStatus{NewTasksPerSecond: 0, BacklogCountHint: backlog},\n\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\tReadPartitions:  partitions(numRead),\n\t\t\tWritePartitions: partitions(numWrite),\n\t\t},\n\t}\n}\n\nfunc mockDescribeTaskList(mocks *mockAdaptiveScalerDeps, partitionID int, resp *types.DescribeTaskListResponse) {\n\tif partitionID == 0 {\n\t\tmocks.mockManager.EXPECT().DescribeTaskList(true).Return(resp)\n\t} else {\n\t\tmocks.mockMatchingClient.EXPECT().DescribeTaskList(gomock.Any(), &types.MatchingDescribeTaskListRequest{\n\t\t\tDomainUUID: mocks.id.domainID,\n\t\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: mocks.id.GetPartition(partitionID),\n\t\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t\t},\n\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t},\n\t\t}).Return(resp, nil)\n\t}\n}\n\nfunc mockDescribeTaskListWithErr(mocks *mockAdaptiveScalerDeps, partitionID int, err error) {\n\tmocks.mockMatchingClient.EXPECT().DescribeTaskList(gomock.Any(), &types.MatchingDescribeTaskListRequest{\n\t\tDomainUUID: mocks.id.domainID,\n\t\tDescRequest: &types.DescribeTaskListRequest{\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: mocks.id.GetPartition(partitionID),\n\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t},\n\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\tIncludeTaskListStatus: true,\n\t\t},\n\t}).Return(nil, err)\n}\n"
  },
  {
    "path": "service/matching/tasklist/db.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\ttaskListDB struct {\n\t\tsync.RWMutex\n\t\tdomainID        string\n\t\tdomainName      string\n\t\ttaskListName    string\n\t\ttaskListKind    int\n\t\ttaskType        int\n\t\trangeID         int64\n\t\tbacklogCount    int64\n\t\tackLevel        int64\n\t\tpartitionConfig *persistence.TaskListPartitionConfig\n\t\tstore           persistence.TaskManager\n\t\tlogger          log.Logger\n\t}\n\ttaskListState struct {\n\t\trangeID  int64\n\t\tackLevel int64\n\t}\n)\n\n// newTaskListDB returns an instance of an object that represents\n// persistence view of a taskList. All mutations / reads to taskLists\n// wrt persistence go through this object.\n//\n// This class will serialize writes to persistence that do condition updates. There are\n// two reasons for doing this:\n//   - To work around known Cassandra issue where concurrent LWT to the same partition cause timeout errors\n//   - To provide the guarantee that there is only writer who updates taskList in persistence at any given point in time\n//     This guarantee makes some of the other code simpler and there is no impact to perf because updates to tasklist are\n//     spread out and happen in background routines\nfunc newTaskListDB(store persistence.TaskManager, domainID string, domainName string, name string, taskType int, kind int, logger log.Logger) *taskListDB {\n\treturn &taskListDB{\n\t\tdomainID:     domainID,\n\t\tdomainName:   domainName,\n\t\ttaskListName: name,\n\t\ttaskListKind: kind,\n\t\ttaskType:     taskType,\n\t\tstore:        store,\n\t\tlogger:       logger,\n\t}\n}\n\n// RangeID returns the current persistence view of rangeID\nfunc (db *taskListDB) RangeID() int64 {\n\tdb.RLock()\n\tdefer db.RUnlock()\n\treturn db.rangeID\n}\n\n// BacklogCount returns the current backlog size\nfunc (db *taskListDB) BacklogCount() int64 {\n\treturn atomic.LoadInt64(&db.backlogCount)\n}\n\nfunc (db *taskListDB) PartitionConfig() *persistence.TaskListPartitionConfig {\n\tdb.RLock()\n\tdefer db.RUnlock()\n\treturn db.partitionConfig\n}\n\n// RenewLease renews the lease on a tasklist. If there is no previous lease,\n// this method will attempt to steal tasklist from current owner\nfunc (db *taskListDB) RenewLease() (taskListState, error) {\n\tdb.Lock()\n\tdefer db.Unlock()\n\tresp, err := db.store.LeaseTaskList(context.Background(), &persistence.LeaseTaskListRequest{\n\t\tDomainID:     db.domainID,\n\t\tTaskList:     db.taskListName,\n\t\tTaskType:     db.taskType,\n\t\tTaskListKind: db.taskListKind,\n\t\tRangeID:      atomic.LoadInt64(&db.rangeID),\n\t\tDomainName:   db.domainName,\n\t})\n\tif err != nil {\n\t\treturn taskListState{}, err\n\t}\n\tdb.rangeID = resp.TaskListInfo.RangeID\n\tdb.ackLevel = resp.TaskListInfo.AckLevel\n\tdb.partitionConfig = resp.TaskListInfo.AdaptivePartitionConfig\n\treturn taskListState{rangeID: db.rangeID, ackLevel: resp.TaskListInfo.AckLevel}, nil\n}\n\n// UpdateState updates the taskList state with the given value\nfunc (db *taskListDB) UpdateState(ackLevel int64) error {\n\tdb.Lock()\n\tdefer db.Unlock()\n\t_, err := db.store.UpdateTaskList(context.Background(), &persistence.UpdateTaskListRequest{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID:                db.domainID,\n\t\t\tName:                    db.taskListName,\n\t\t\tTaskType:                db.taskType,\n\t\t\tAckLevel:                ackLevel,\n\t\t\tRangeID:                 db.rangeID,\n\t\t\tKind:                    db.taskListKind,\n\t\t\tAdaptivePartitionConfig: db.partitionConfig,\n\t\t},\n\t\tDomainName: db.domainName,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\tdb.ackLevel = ackLevel\n\treturn nil\n}\n\nfunc (db *taskListDB) UpdateTaskListPartitionConfig(partitionConfig *persistence.TaskListPartitionConfig) error {\n\tdb.Lock()\n\tdefer db.Unlock()\n\t_, err := db.store.UpdateTaskList(context.Background(), &persistence.UpdateTaskListRequest{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID:                db.domainID,\n\t\t\tName:                    db.taskListName,\n\t\t\tTaskType:                db.taskType,\n\t\t\tAckLevel:                db.ackLevel,\n\t\t\tRangeID:                 db.rangeID,\n\t\t\tKind:                    db.taskListKind,\n\t\t\tAdaptivePartitionConfig: partitionConfig,\n\t\t},\n\t\tDomainName: db.domainName,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\tdb.partitionConfig = partitionConfig\n\treturn nil\n}\n\n// CreateTasks creates a batch of given tasks for this task list\nfunc (db *taskListDB) CreateTasks(tasks []*persistence.CreateTaskInfo) (*persistence.CreateTasksResponse, error) {\n\tdb.Lock()\n\tdefer db.Unlock()\n\treturn db.store.CreateTasks(context.Background(), &persistence.CreateTasksRequest{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID: db.domainID,\n\t\t\tName:     db.taskListName,\n\t\t\tTaskType: db.taskType,\n\t\t\tRangeID:  db.rangeID,\n\t\t},\n\t\tTasks:      tasks,\n\t\tDomainName: db.domainName,\n\t})\n}\n\n// GetTasks returns a batch of tasks between the given range\nfunc (db *taskListDB) GetTasks(minTaskID int64, maxTaskID int64, batchSize int) (*persistence.GetTasksResponse, error) {\n\treturn db.store.GetTasks(context.Background(), &persistence.GetTasksRequest{\n\t\tDomainID:     db.domainID,\n\t\tTaskList:     db.taskListName,\n\t\tTaskType:     db.taskType,\n\t\tBatchSize:    batchSize,\n\t\tReadLevel:    minTaskID,  // exclusive\n\t\tMaxReadLevel: &maxTaskID, // inclusive\n\t\tDomainName:   db.domainName,\n\t})\n}\n\n// CompleteTasksLessThan deletes of tasks less than the given taskID. Limit is\n// the upper bound of number of tasks that can be deleted by this method. It may\n// or may not be honored\nfunc (db *taskListDB) CompleteTasksLessThan(taskID int64, limit int) (int, error) {\n\tresp, err := db.store.CompleteTasksLessThan(context.Background(), &persistence.CompleteTasksLessThanRequest{\n\t\tDomainID:     db.domainID,\n\t\tTaskListName: db.taskListName,\n\t\tTaskType:     db.taskType,\n\t\tTaskID:       taskID,\n\t\tLimit:        limit,\n\t\tDomainName:   db.domainName,\n\t})\n\tif err != nil {\n\t\tdb.logger.Error(\"Persistent store operation failure\",\n\t\t\ttag.StoreOperationCompleteTasksLessThan,\n\t\t\ttag.Error(err),\n\t\t\ttag.TaskID(taskID),\n\t\t\ttag.TaskType(db.taskType),\n\t\t\ttag.WorkflowTaskListName(db.taskListName))\n\t\treturn 0, err\n\t}\n\treturn resp.TasksCompleted, nil\n}\n\n// GetTaskListSize gets the backlog size of a tasklist\nfunc (db *taskListDB) GetTaskListSize(ackLevel int64) (int64, error) {\n\tresp, err := db.store.GetTaskListSize(context.Background(), &persistence.GetTaskListSizeRequest{\n\t\tDomainID:     db.domainID,\n\t\tDomainName:   db.domainName,\n\t\tTaskListName: db.taskListName,\n\t\tTaskListType: db.taskType,\n\t\tAckLevel:     ackLevel,\n\t})\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\tatomic.StoreInt64(&db.backlogCount, resp.Size)\n\treturn resp.Size, nil\n}\n\nfunc (db *taskListDB) GetTaskListInfo(taskListName string) (*persistence.TaskListInfo, error) {\n\tresp, err := db.store.GetTaskList(context.Background(), &persistence.GetTaskListRequest{\n\t\tDomainID:   db.domainID,\n\t\tDomainName: db.domainName,\n\t\tTaskList:   taskListName,\n\t\tTaskType:   db.taskType,\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn resp.TaskListInfo, nil\n}\n"
  },
  {
    "path": "service/matching/tasklist/forwarder.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\ntype (\n\t// forwarderImpl is the type that contains state pertaining to\n\t// the api call forwarder component\n\tforwarderImpl struct {\n\t\tscope        metrics.Scope\n\t\tcfg          *config.ForwarderConfig\n\t\ttaskListID   *Identifier\n\t\ttaskListKind types.TaskListKind\n\t\tclient       matching.Client\n\n\t\t// token channels that vend tokens necessary to make\n\t\t// API calls exposed by forwarder. Tokens are used\n\t\t// to enforce maxOutstanding forwarded calls from this\n\t\t// instance. And channels are used so that the caller\n\t\t// can use them in a select{} block along with other\n\t\t// conditions\n\t\taddReqToken  atomic.Value\n\t\tpollReqToken atomic.Value\n\n\t\t// cached values of maxOutstanding dynamic config values.\n\t\t// these are used to detect changes\n\t\toutstandingTasksLimit int32\n\t\toutstandingPollsLimit int32\n\n\t\t// todo: implement a rate limiter that automatically\n\t\t// adjusts rate based on ServiceBusy errors from API calls\n\t\tlimiter quotas.Limiter\n\t}\n\t// ForwarderReqToken is the token that must be acquired before\n\t// making forwarder API calls. This type contains the state\n\t// for the token itself\n\tForwarderReqToken struct {\n\t\tch chan *ForwarderReqToken\n\t}\n)\n\nvar (\n\tErrNoParent            = errors.New(\"cannot find parent task list for forwarding\")\n\tErrTaskListKind        = errors.New(\"forwarding is not supported on sticky task list\")\n\tErrInvalidTaskListType = errors.New(\"unrecognized task list type\")\n\tErrForwarderSlowDown   = errors.New(\"tasklist forwarding throttle limit exceeded\")\n)\n\n// noopForwarderTokenC refers to a token channel that blocks forever\nvar noopForwarderTokenC <-chan *ForwarderReqToken = make(chan *ForwarderReqToken)\n\n// newForwarder returns an instance of Forwarder object which\n// can be used to forward api request calls from a task list\n// child partition to a task list parent partition. The returned\n// forwarder is tied to a single task list. All of the exposed\n// methods can return the following errors:\n// Returns following errors:\n//   - errNoParent: If this task list doesn't have a parent to forward to\n//   - errTaskListKind: If the task list is a sticky task list. Sticky task lists are never partitioned\n//   - errForwarderSlowDown: When the rate limit is exceeded\n//   - errInvalidTaskType: If the task list type is invalid\nfunc newForwarder(\n\tcfg *config.ForwarderConfig,\n\ttaskListID *Identifier,\n\tkind types.TaskListKind,\n\tclient matching.Client,\n\tscope metrics.Scope,\n) Forwarder {\n\trpsFunc := func() float64 { return float64(cfg.ForwarderMaxRatePerSecond()) }\n\tfwdr := &forwarderImpl{\n\t\tcfg:                   cfg,\n\t\tclient:                client,\n\t\ttaskListID:            taskListID,\n\t\ttaskListKind:          kind,\n\t\toutstandingTasksLimit: int32(cfg.ForwarderMaxOutstandingTasks()),\n\t\toutstandingPollsLimit: int32(cfg.ForwarderMaxOutstandingPolls()),\n\t\tlimiter:               quotas.NewDynamicRateLimiter(rpsFunc),\n\t\tscope:                 scope,\n\t}\n\tfwdr.addReqToken.Store(newForwarderReqToken(int(fwdr.outstandingTasksLimit)))\n\tfwdr.pollReqToken.Store(newForwarderReqToken(int(fwdr.outstandingPollsLimit)))\n\treturn fwdr\n}\n\n// ForwardTask forwards an activity or decision task to the parent task list partition if it exist\nfunc (fwdr *forwarderImpl) ForwardTask(ctx context.Context, task *InternalTask) error {\n\tif fwdr.taskListKind == types.TaskListKindSticky {\n\t\treturn ErrTaskListKind\n\t}\n\n\tname := fwdr.taskListID.Parent(fwdr.cfg.ForwarderMaxChildrenPerNode())\n\tif name == \"\" {\n\t\treturn ErrNoParent\n\t}\n\n\tif !fwdr.limiter.Allow() {\n\t\treturn ErrForwarderSlowDown\n\t}\n\n\tvar err error\n\n\tsw := fwdr.scope.StartTimer(metrics.ForwardTaskLatencyPerTaskList)\n\tdefer sw.Stop()\n\tswitch fwdr.taskListID.GetType() {\n\tcase persistence.TaskListTypeDecision:\n\t\t_, err = fwdr.client.AddDecisionTask(ctx, &types.AddDecisionTaskRequest{\n\t\t\tDomainUUID: task.Event.DomainID,\n\t\t\tExecution:  task.WorkflowExecution(),\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: name,\n\t\t\t\tKind: &fwdr.taskListKind,\n\t\t\t},\n\t\t\tScheduleID:                    task.Event.ScheduleID,\n\t\t\tScheduleToStartTimeoutSeconds: &task.Event.ScheduleToStartTimeoutSeconds,\n\t\t\tSource:                        &task.source,\n\t\t\tForwardedFrom:                 fwdr.taskListID.GetName(),\n\t\t\tPartitionConfig:               task.Event.PartitionConfig,\n\t\t})\n\tcase persistence.TaskListTypeActivity:\n\t\t_, err = fwdr.client.AddActivityTask(ctx, &types.AddActivityTaskRequest{\n\t\t\tDomainUUID:       fwdr.taskListID.GetDomainID(),\n\t\t\tSourceDomainUUID: task.Event.DomainID,\n\t\t\tExecution:        task.WorkflowExecution(),\n\t\t\tTaskList: &types.TaskList{\n\t\t\t\tName: name,\n\t\t\t\tKind: &fwdr.taskListKind,\n\t\t\t},\n\t\t\tScheduleID:                    task.Event.ScheduleID,\n\t\t\tScheduleToStartTimeoutSeconds: &task.Event.ScheduleToStartTimeoutSeconds,\n\t\t\tSource:                        &task.source,\n\t\t\tForwardedFrom:                 fwdr.taskListID.GetName(),\n\t\t\tPartitionConfig:               task.Event.PartitionConfig,\n\t\t})\n\tdefault:\n\t\treturn ErrInvalidTaskListType\n\t}\n\n\treturn fwdr.handleErr(err)\n}\n\n// ForwardQueryTask forwards a query task to parent task list partition, if it exist\nfunc (fwdr *forwarderImpl) ForwardQueryTask(\n\tctx context.Context,\n\ttask *InternalTask,\n) (*types.MatchingQueryWorkflowResponse, error) {\n\n\tif fwdr.taskListKind == types.TaskListKindSticky {\n\t\treturn nil, ErrTaskListKind\n\t}\n\n\tname := fwdr.taskListID.Parent(fwdr.cfg.ForwarderMaxChildrenPerNode())\n\tif name == \"\" {\n\t\treturn nil, ErrNoParent\n\t}\n\n\tsw := fwdr.scope.StartTimer(metrics.ForwardQueryLatencyPerTaskList)\n\tdefer sw.Stop()\n\tresp, err := fwdr.client.QueryWorkflow(ctx, &types.MatchingQueryWorkflowRequest{\n\t\tDomainUUID: task.Query.Request.DomainUUID,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: name,\n\t\t\tKind: &fwdr.taskListKind,\n\t\t},\n\t\tQueryRequest:  task.Query.Request.QueryRequest,\n\t\tForwardedFrom: fwdr.taskListID.GetName(),\n\t})\n\n\treturn resp, fwdr.handleErr(err)\n}\n\n// ForwardPoll forwards a poll request to parent task list partition if it exist\nfunc (fwdr *forwarderImpl) ForwardPoll(ctx context.Context) (*InternalTask, error) {\n\tif fwdr.taskListKind == types.TaskListKindSticky {\n\t\treturn nil, ErrTaskListKind\n\t}\n\n\tname := fwdr.taskListID.Parent(fwdr.cfg.ForwarderMaxChildrenPerNode())\n\tif name == \"\" {\n\t\treturn nil, ErrNoParent\n\t}\n\n\tsw := fwdr.scope.StartTimer(metrics.ForwardPollLatencyPerTaskList)\n\tdefer sw.Stop()\n\tpollerID := PollerIDFromContext(ctx)\n\tidentity := IdentityFromContext(ctx)\n\tisolationGroup := IsolationGroupFromContext(ctx)\n\n\tswitch fwdr.taskListID.GetType() {\n\tcase persistence.TaskListTypeDecision:\n\t\tresp, err := fwdr.client.PollForDecisionTask(ctx, &types.MatchingPollForDecisionTaskRequest{\n\t\t\tDomainUUID: fwdr.taskListID.GetDomainID(),\n\t\t\tPollerID:   pollerID,\n\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: name,\n\t\t\t\t\tKind: &fwdr.taskListKind,\n\t\t\t\t},\n\t\t\t\tIdentity: identity,\n\t\t\t},\n\t\t\tForwardedFrom:  fwdr.taskListID.GetName(),\n\t\t\tIsolationGroup: isolationGroup,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, fwdr.handleErr(err)\n\t\t}\n\t\treturn newInternalStartedTask(&startedTaskInfo{decisionTaskInfo: resp}), nil\n\tcase persistence.TaskListTypeActivity:\n\t\tresp, err := fwdr.client.PollForActivityTask(ctx, &types.MatchingPollForActivityTaskRequest{\n\t\t\tDomainUUID: fwdr.taskListID.GetDomainID(),\n\t\t\tPollerID:   pollerID,\n\t\t\tPollRequest: &types.PollForActivityTaskRequest{\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: name,\n\t\t\t\t\tKind: &fwdr.taskListKind,\n\t\t\t\t},\n\t\t\t\tIdentity: identity,\n\t\t\t},\n\t\t\tForwardedFrom:  fwdr.taskListID.GetName(),\n\t\t\tIsolationGroup: isolationGroup,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, fwdr.handleErr(err)\n\t\t}\n\t\treturn newInternalStartedTask(&startedTaskInfo{activityTaskInfo: resp}), nil\n\t}\n\n\treturn nil, ErrInvalidTaskListType\n}\n\n// AddReqTokenC returns a channel that can be used to wait for a token\n// that's necessary before making a ForwardTask or ForwardQueryTask API call.\n// After the API call is invoked, token.release() must be invoked\nfunc (fwdr *forwarderImpl) AddReqTokenC() <-chan *ForwarderReqToken {\n\tfwdr.refreshTokenC(&fwdr.addReqToken, &fwdr.outstandingTasksLimit, int32(fwdr.cfg.ForwarderMaxOutstandingTasks()))\n\treturn fwdr.addReqToken.Load().(*ForwarderReqToken).ch\n}\n\n// PollReqTokenC returns a channel that can be used to wait for a token\n// that's necessary before making a ForwardPoll API call. After the API\n// call is invoked, token.release() must be invoked\nfunc (fwdr *forwarderImpl) PollReqTokenC() <-chan *ForwarderReqToken {\n\tfwdr.refreshTokenC(&fwdr.pollReqToken, &fwdr.outstandingPollsLimit, int32(fwdr.cfg.ForwarderMaxOutstandingPolls()))\n\treturn fwdr.pollReqToken.Load().(*ForwarderReqToken).ch\n}\n\nfunc (fwdr *forwarderImpl) refreshTokenC(value *atomic.Value, curr *int32, maxLimit int32) {\n\tcurrLimit := atomic.LoadInt32(curr)\n\tif currLimit != maxLimit {\n\t\tif atomic.CompareAndSwapInt32(curr, currLimit, maxLimit) {\n\t\t\tvalue.Store(newForwarderReqToken(int(maxLimit)))\n\t\t}\n\t}\n}\n\nfunc (fwdr *forwarderImpl) handleErr(err error) error {\n\tif _, ok := err.(*types.ServiceBusyError); ok {\n\t\treturn ErrForwarderSlowDown\n\t}\n\treturn err\n}\n\nfunc newForwarderReqToken(maxOutstanding int) *ForwarderReqToken {\n\treqToken := &ForwarderReqToken{ch: make(chan *ForwarderReqToken, maxOutstanding)}\n\tfor i := 0; i < maxOutstanding; i++ {\n\t\treqToken.ch <- reqToken\n\t}\n\treturn reqToken\n}\n\nfunc (token *ForwarderReqToken) release() {\n\ttoken.ch <- token\n}\n"
  },
  {
    "path": "service/matching/tasklist/forwarder_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\ntype ForwarderTestSuite struct {\n\tsuite.Suite\n\tcontroller *gomock.Controller\n\tclient     *matching.MockClient\n\tfwdr       *forwarderImpl\n\tcfg        *config.ForwarderConfig\n\ttaskList   *Identifier\n}\n\nfunc TestForwarderSuite(t *testing.T) {\n\tsuite.Run(t, new(ForwarderTestSuite))\n}\n\nfunc (t *ForwarderTestSuite) SetupTest() {\n\tt.controller = gomock.NewController(t.T())\n\tt.client = matching.NewMockClient(t.controller)\n\tt.cfg = &config.ForwarderConfig{\n\t\tForwarderMaxOutstandingPolls: func() int { return 1 },\n\t\tForwarderMaxRatePerSecond:    func() int { return 2 },\n\t\tForwarderMaxChildrenPerNode:  func() int { return 20 },\n\t\tForwarderMaxOutstandingTasks: func() int { return 1 },\n\t}\n\tid, err := NewIdentifier(\"fwdr\", \"tl0\", persistence.TaskListTypeDecision)\n\tt.NoError(err)\n\tt.taskList = id\n\tt.fwdr = newForwarder(t.cfg, t.taskList, types.TaskListKindNormal, t.client, metrics.NoopScope).(*forwarderImpl)\n}\n\nfunc (t *ForwarderTestSuite) TearDownTest() {\n\tt.controller.Finish()\n}\n\nfunc (t *ForwarderTestSuite) TestForwardTaskError() {\n\ttask := newInternalTask(&persistence.TaskInfo{}, nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\tt.Equal(ErrNoParent, t.fwdr.ForwardTask(context.Background(), task))\n\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\tt.fwdr.taskListKind = types.TaskListKindSticky\n\tt.Equal(ErrTaskListKind, t.fwdr.ForwardTask(context.Background(), task))\n}\n\nfunc (t *ForwarderTestSuite) TestForwardDecisionTask() {\n\tt.usingTasklistPartition(persistence.TaskListTypeDecision)\n\n\tvar request *types.AddDecisionTaskRequest\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\trequest = arg1\n\t\t},\n\t).Return(&types.AddDecisionTaskResponse{}, nil).Times(1)\n\n\ttaskInfo := t.newTaskInfo()\n\ttask := newInternalTask(taskInfo, nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\tt.NoError(t.fwdr.ForwardTask(context.Background(), task))\n\tt.NotNil(request)\n\tt.Equal(t.taskList.Parent(20), request.TaskList.GetName())\n\tt.Equal(t.fwdr.taskListKind, request.TaskList.GetKind())\n\tt.Equal(taskInfo.DomainID, request.GetDomainUUID())\n\tt.Equal(taskInfo.WorkflowID, request.GetExecution().GetWorkflowID())\n\tt.Equal(taskInfo.RunID, request.GetExecution().GetRunID())\n\tt.Equal(taskInfo.ScheduleID, request.GetScheduleID())\n\tt.Equal(taskInfo.ScheduleToStartTimeoutSeconds, request.GetScheduleToStartTimeoutSeconds())\n\tt.Equal(t.taskList.name, request.GetForwardedFrom())\n}\n\nfunc (t *ForwarderTestSuite) TestForwardActivityTask() {\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\n\tvar request *types.AddActivityTaskRequest\n\tt.client.EXPECT().AddActivityTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddActivityTaskRequest, option ...yarpc.CallOption) {\n\t\t\trequest = arg1\n\t\t},\n\t).Return(&types.AddActivityTaskResponse{}, nil).Times(1)\n\n\ttaskInfo := t.newTaskInfo()\n\ttask := newInternalTask(taskInfo, nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\tt.NoError(t.fwdr.ForwardTask(context.Background(), task))\n\tt.NotNil(request)\n\tt.Equal(t.taskList.Parent(20), request.TaskList.GetName())\n\tt.Equal(t.fwdr.taskListKind, request.TaskList.GetKind())\n\tt.Equal(t.taskList.domainID, request.GetDomainUUID())\n\tt.Equal(taskInfo.DomainID, request.GetSourceDomainUUID())\n\tt.Equal(taskInfo.WorkflowID, request.GetExecution().GetWorkflowID())\n\tt.Equal(taskInfo.RunID, request.GetExecution().GetRunID())\n\tt.Equal(taskInfo.ScheduleID, request.GetScheduleID())\n\tt.Equal(taskInfo.ScheduleToStartTimeoutSeconds, request.GetScheduleToStartTimeoutSeconds())\n\tt.Equal(t.taskList.name, request.GetForwardedFrom())\n}\n\nfunc (t *ForwarderTestSuite) TestForwardTaskRateExceeded() {\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\n\trps := 2\n\tt.client.EXPECT().AddActivityTask(gomock.Any(), gomock.Any()).Return(&types.AddActivityTaskResponse{}, nil).Times(rps)\n\ttaskInfo := t.newTaskInfo()\n\ttask := newInternalTask(taskInfo, nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\tfor i := 0; i < rps; i++ {\n\t\tt.NoError(t.fwdr.ForwardTask(context.Background(), task))\n\t}\n\tt.Equal(ErrForwarderSlowDown, t.fwdr.ForwardTask(context.Background(), task))\n}\n\nfunc (t *ForwarderTestSuite) TestForwardQueryTaskError() {\n\ttask := newInternalQueryTask(\"id1\", &types.MatchingQueryWorkflowRequest{})\n\t_, err := t.fwdr.ForwardQueryTask(context.Background(), task)\n\tt.Equal(ErrNoParent, err)\n\n\tt.usingTasklistPartition(persistence.TaskListTypeDecision)\n\tt.fwdr.taskListKind = types.TaskListKindSticky\n\t_, err = t.fwdr.ForwardQueryTask(context.Background(), task)\n\tt.Equal(ErrTaskListKind, err)\n}\n\nfunc (t *ForwarderTestSuite) TestForwardQueryTask() {\n\tt.usingTasklistPartition(persistence.TaskListTypeDecision)\n\ttask := newInternalQueryTask(\"id1\", &types.MatchingQueryWorkflowRequest{})\n\tresp := &types.MatchingQueryWorkflowResponse{}\n\tvar request *types.MatchingQueryWorkflowRequest\n\tt.client.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingQueryWorkflowRequest, option ...yarpc.CallOption) {\n\t\t\trequest = arg1\n\t\t},\n\t).Return(resp, nil).Times(1)\n\n\tgotResp, err := t.fwdr.ForwardQueryTask(context.Background(), task)\n\tt.NoError(err)\n\tt.Equal(t.taskList.Parent(20), request.TaskList.GetName())\n\tt.Equal(t.fwdr.taskListKind, request.TaskList.GetKind())\n\tt.True(task.Query.Request.QueryRequest == request.QueryRequest)\n\tt.Equal(resp, gotResp)\n}\n\nfunc (t *ForwarderTestSuite) TestForwardQueryTaskRateNotEnforced() {\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\ttask := newInternalQueryTask(\"id1\", &types.MatchingQueryWorkflowRequest{})\n\tresp := &types.MatchingQueryWorkflowResponse{}\n\trps := 2\n\tt.client.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(resp, nil).Times(rps + 1)\n\tfor i := 0; i < rps; i++ {\n\t\t_, err := t.fwdr.ForwardQueryTask(context.Background(), task)\n\t\tt.NoError(err)\n\t}\n\t_, err := t.fwdr.ForwardQueryTask(context.Background(), task)\n\tt.NoError(err) // no rateliming should be enforced for query task\n}\n\nfunc (t *ForwarderTestSuite) TestForwardPollError() {\n\t_, err := t.fwdr.ForwardPoll(context.Background())\n\tt.Equal(ErrNoParent, err)\n\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\tt.fwdr.taskListKind = types.TaskListKindSticky\n\t_, err = t.fwdr.ForwardPoll(context.Background())\n\tt.Equal(ErrTaskListKind, err)\n\n}\n\nfunc (t *ForwarderTestSuite) TestForwardPollForDecision() {\n\tt.usingTasklistPartition(persistence.TaskListTypeDecision)\n\n\tpollerID := uuid.New()\n\tctx := ContextWithPollerID(context.Background(), pollerID)\n\tctx = ContextWithIdentity(ctx, \"id1\")\n\tresp := &types.MatchingPollForDecisionTaskResponse{}\n\n\tvar request *types.MatchingPollForDecisionTaskRequest\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\trequest = arg1\n\t\t},\n\t).Return(resp, nil).Times(1)\n\n\ttask, err := t.fwdr.ForwardPoll(ctx)\n\tt.NoError(err)\n\tt.NotNil(task)\n\tt.NotNil(request)\n\tt.Equal(pollerID, request.GetPollerID())\n\tt.Equal(t.taskList.domainID, request.GetDomainUUID())\n\tt.Equal(\"id1\", request.GetPollRequest().GetIdentity())\n\tt.Equal(t.taskList.Parent(20), request.GetPollRequest().GetTaskList().GetName())\n\tt.Equal(t.fwdr.taskListKind, request.GetPollRequest().GetTaskList().GetKind())\n\tt.Equal(resp, task.PollForDecisionResponse())\n\tt.Nil(task.PollForActivityResponse())\n}\n\nfunc (t *ForwarderTestSuite) TestForwardPollForActivity() {\n\tt.usingTasklistPartition(persistence.TaskListTypeActivity)\n\n\tpollerID := uuid.New()\n\tctx := ContextWithPollerID(context.Background(), pollerID)\n\tctx = ContextWithIdentity(ctx, \"id1\")\n\tresp := &types.MatchingPollForActivityTaskResponse{}\n\n\tvar request *types.MatchingPollForActivityTaskRequest\n\tt.client.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForActivityTaskRequest, option ...yarpc.CallOption) {\n\t\t\trequest = arg1\n\t\t},\n\t).Return(resp, nil).Times(1)\n\n\ttask, err := t.fwdr.ForwardPoll(ctx)\n\tt.NoError(err)\n\tt.NotNil(task)\n\tt.NotNil(request)\n\tt.Equal(pollerID, request.GetPollerID())\n\tt.Equal(t.taskList.domainID, request.GetDomainUUID())\n\tt.Equal(\"id1\", request.GetPollRequest().GetIdentity())\n\tt.Equal(t.taskList.Parent(20), request.GetPollRequest().GetTaskList().GetName())\n\tt.Equal(t.fwdr.taskListKind, request.GetPollRequest().GetTaskList().GetKind())\n\tt.Equal(resp, task.PollForActivityResponse())\n\tt.Nil(task.PollForDecisionResponse())\n}\n\nfunc (t *ForwarderTestSuite) TestMaxOutstandingConcurrency() {\n\tconcurrency := 50\n\ttestCases := []struct {\n\t\tname          string\n\t\tmustLeakToken bool\n\t\toutput        int32\n\t}{\n\t\t{\"contention\", false, int32(concurrency)},\n\t\t{\"token_leak\", true, 1},\n\t}\n\n\tvar adds int32\n\tvar polls int32\n\tvar wg sync.WaitGroup\n\n\tfor _, tc := range testCases {\n\t\tadds = 0\n\t\tpolls = 0\n\t\tt.Run(tc.name, func() {\n\t\t\tfor i := 0; i < concurrency; i++ {\n\t\t\t\twg.Add(1)\n\t\t\t\tgo func() {\n\t\t\t\t\tselect {\n\t\t\t\t\tcase token := <-t.fwdr.AddReqTokenC():\n\t\t\t\t\t\tif !tc.mustLeakToken {\n\t\t\t\t\t\t\ttoken.release()\n\t\t\t\t\t\t}\n\t\t\t\t\t\tatomic.AddInt32(&adds, 1)\n\t\t\t\t\tcase <-time.After(time.Millisecond * 100):\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\n\t\t\t\t\tselect {\n\t\t\t\t\tcase token := <-t.fwdr.PollReqTokenC():\n\t\t\t\t\t\tif !tc.mustLeakToken {\n\t\t\t\t\t\t\ttoken.release()\n\t\t\t\t\t\t}\n\t\t\t\t\t\tatomic.AddInt32(&polls, 1)\n\t\t\t\t\tcase <-time.After(time.Millisecond * 100):\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\n\t\t\t\t\twg.Done()\n\t\t\t\t}()\n\t\t\t}\n\t\t\tt.True(common.AwaitWaitGroup(&wg, time.Second))\n\t\t\tt.Equal(tc.output, adds)\n\t\t\tt.Equal(tc.output, polls)\n\t\t})\n\t}\n}\n\nfunc (t *ForwarderTestSuite) TestMaxOutstandingConfigUpdate() {\n\tmaxOutstandingTasks := int32(1)\n\tmaxOutstandingPolls := int32(1)\n\tt.fwdr.cfg.ForwarderMaxOutstandingTasks = func() int { return int(atomic.LoadInt32(&maxOutstandingTasks)) }\n\tt.fwdr.cfg.ForwarderMaxOutstandingPolls = func() int { return int(atomic.LoadInt32(&maxOutstandingPolls)) }\n\n\tstartC := make(chan struct{})\n\tdoneWG := sync.WaitGroup{}\n\tfor i := 0; i < 10; i++ {\n\t\tdoneWG.Add(1)\n\t\tgo func() {\n\t\t\t<-startC\n\t\t\ttoken1 := <-t.fwdr.AddReqTokenC()\n\t\t\ttoken1.release()\n\t\t\ttoken2 := <-t.fwdr.PollReqTokenC()\n\t\t\ttoken2.release()\n\t\t\tdoneWG.Done()\n\t\t}()\n\t}\n\n\tmaxOutstandingTasks = 10\n\tmaxOutstandingPolls = 10\n\tclose(startC)\n\tt.True(common.AwaitWaitGroup(&doneWG, time.Second))\n\n\tt.Equal(10, cap(t.fwdr.addReqToken.Load().(*ForwarderReqToken).ch))\n\tt.Equal(10, cap(t.fwdr.pollReqToken.Load().(*ForwarderReqToken).ch))\n}\n\nfunc (t *ForwarderTestSuite) usingTasklistPartition(taskType int) {\n\tt.taskList = NewTestTaskListID(t.T(), \"fwdr\", constants.ReservedTaskListPrefix+\"tl0/1\", taskType)\n\tt.fwdr.taskListID = t.taskList\n}\n\nfunc (t *ForwarderTestSuite) newTaskInfo() *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:                      uuid.New(),\n\t\tWorkflowID:                    uuid.New(),\n\t\tRunID:                         uuid.New(),\n\t\tTaskID:                        rand.Int63(),\n\t\tScheduleID:                    rand.Int63(),\n\t\tScheduleToStartTimeoutSeconds: rand.Int31(),\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/identifier.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"bytes\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// Identifier is the key that uniquely identifies a task list\n\tIdentifier struct {\n\t\tqualifiedTaskListName\n\t\tdomainID string\n\t\ttaskType int\n\t}\n\t// qualifiedTaskListName refers to the fully qualified task list name\n\tqualifiedTaskListName struct {\n\t\tname      string // internal name of the tasks list\n\t\tbaseName  string // original name of the task list as specified by user\n\t\tpartition int    // partitionID of task list\n\t}\n)\n\n// newTaskListName returns a fully qualified task list name.\n// Fully qualified names contain additional metadata about task list\n// derived from their given name. The additional metadata only makes sense\n// when a task list has more than one partition. When there is more than\n// one partition for a user specified task list, each of the\n// individual partitions have an internal name of the form\n//\n//\t/__cadence_sys/[original-name]/[partitionID]\n//\n// The name of the root partition is always the same as the user specified name. Rest of\n// the partitions follow the naming convention above. In addition, the task lists partitions\n// logically form a N-ary tree where N is configurable dynamically. The tree formation is an\n// optimization to allow for partitioned task lists to dispatch tasks with low latency when\n// throughput is low - See https://github.com/uber/cadence/issues/2098\n//\n// Returns error if the given name is non-compliant with the required format\n// for task list names\nfunc newTaskListName(name string) (qualifiedTaskListName, error) {\n\ttn := qualifiedTaskListName{\n\t\tname:     name,\n\t\tbaseName: name,\n\t}\n\tif err := tn.init(); err != nil {\n\t\treturn qualifiedTaskListName{}, err\n\t}\n\treturn tn, nil\n}\n\n// IsRoot returns true if this task list is a root partition\nfunc (tn *qualifiedTaskListName) IsRoot() bool {\n\treturn tn.partition == 0\n}\n\nfunc (tn *qualifiedTaskListName) Partition() int {\n\treturn tn.partition\n}\n\n// GetRoot returns the root name for a task list\nfunc (tn *qualifiedTaskListName) GetRoot() string {\n\treturn tn.baseName\n}\n\n// GetName returns the name of the task list\nfunc (tn *qualifiedTaskListName) GetName() string {\n\treturn tn.name\n}\n\n// Parent returns the name of the parent task list\n// input:\n//\n//\tdegree: Number of children at each level of the tree\n//\n// Returns empty string if this task list is the root\nfunc (tn *qualifiedTaskListName) Parent(degree int) string {\n\tif tn.IsRoot() || degree == 0 {\n\t\treturn \"\"\n\t}\n\tpid := (tn.partition+degree-1)/degree - 1\n\treturn tn.GetPartition(pid)\n}\n\nfunc (tn *qualifiedTaskListName) GetPartition(partition int) string {\n\tif partition == 0 {\n\t\treturn tn.baseName\n\t}\n\treturn fmt.Sprintf(\"%v%v/%v\", constants.ReservedTaskListPrefix, tn.baseName, partition)\n}\n\nfunc (tn *qualifiedTaskListName) init() error {\n\tif !strings.HasPrefix(tn.name, constants.ReservedTaskListPrefix) {\n\t\treturn nil\n\t}\n\n\tsuffixOff := strings.LastIndex(tn.name, \"/\")\n\tif suffixOff <= len(constants.ReservedTaskListPrefix) {\n\t\treturn fmt.Errorf(\"invalid partitioned task list name %v\", tn.name)\n\t}\n\n\tp, err := strconv.Atoi(tn.name[suffixOff+1:])\n\tif err != nil || p <= 0 {\n\t\treturn fmt.Errorf(\"invalid partitioned task list name %v\", tn.name)\n\t}\n\n\ttn.partition = p\n\ttn.baseName = tn.name[len(constants.ReservedTaskListPrefix):suffixOff]\n\treturn nil\n}\n\n// NewIdentifier returns identifier which uniquely identifies as task list\nfunc NewIdentifier(\n\tdomainID string,\n\ttaskListName string,\n\ttaskType int,\n) (*Identifier, error) {\n\tname, err := newTaskListName(taskListName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Identifier{\n\t\tqualifiedTaskListName: name,\n\t\tdomainID:              domainID,\n\t\ttaskType:              taskType,\n\t}, nil\n}\n\n// GetDomainID returns the domain ID of the task list\nfunc (tid *Identifier) GetDomainID() string {\n\treturn tid.domainID\n}\n\n// GetType returns the task type of the task list\nfunc (tid *Identifier) GetType() int {\n\treturn tid.taskType\n}\n\nfunc (tid *Identifier) String() string {\n\tvar b bytes.Buffer\n\tb.WriteString(\"[\")\n\tb.WriteString(\"name=\")\n\tb.WriteString(tid.name)\n\tb.WriteString(\"type=\")\n\tif tid.taskType == persistence.TaskListTypeActivity {\n\t\tb.WriteString(\"activity\")\n\t} else {\n\t\tb.WriteString(\"decision\")\n\t}\n\tb.WriteString(\"]\")\n\treturn b.String()\n}\n"
  },
  {
    "path": "service/matching/tasklist/identifier_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"fmt\"\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestValidTaskListNames(t *testing.T) {\n\ttestCases := []struct {\n\t\tinput     string\n\t\tbaseName  string\n\t\tpartition int\n\t}{\n\t\t{\"0\", \"0\", 0},\n\t\t{\"list0\", \"list0\", 0},\n\t\t{\"/list0\", \"/list0\", 0},\n\t\t{\"/list0/\", \"/list0/\", 0},\n\t\t{\"__cadence_sys/list0\", \"__cadence_sys/list0\", 0},\n\t\t{\"__cadence_sys/list0/\", \"__cadence_sys/list0/\", 0},\n\t\t{\"/__cadence_sys_list0\", \"/__cadence_sys_list0\", 0},\n\t\t{\"/__cadence_sys/list0/1\", \"list0\", 1},\n\t\t{\"/__cadence_sys//list0//41\", \"/list0/\", 41},\n\t\t{\"/__cadence_sys//__cadence_sys/sys/0/41\", \"/__cadence_sys/sys/0\", 41},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.input, func(t *testing.T) {\n\t\t\ttn, err := newTaskListName(tc.input)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, tc.partition, tn.partition)\n\t\t\trequire.Equal(t, tc.partition == 0, tn.IsRoot())\n\t\t\trequire.Equal(t, tc.baseName, tn.baseName)\n\t\t\trequire.Equal(t, tc.baseName, tn.GetRoot())\n\t\t\trequire.Equal(t, tc.input, tn.name)\n\t\t\trequire.Equal(t, tc.input, tn.GetName())\n\n\t\t\t// NewIdentifier should validate taskListName as well\n\t\t\tid, err := NewIdentifier(\"domain-name\", tc.input, persistence.TaskListTypeActivity)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, \"domain-name\", id.GetDomainID())\n\t\t\trequire.Equal(t, persistence.TaskListTypeActivity, id.GetType())\n\t\t})\n\t}\n}\n\nfunc TestTaskListParentName(t *testing.T) {\n\ttestCases := []struct {\n\t\tname   string\n\t\tdegree int\n\t\toutput string\n\t}{\n\t\t/* unexpected input */\n\t\t{\"list0\", 0, \"\"},\n\t\t/* 1-ary tree */\n\t\t{\"list0\", 1, \"\"},\n\t\t{\"/__cadence_sys/list0/1\", 1, \"list0\"},\n\t\t{\"/__cadence_sys/list0/2\", 1, \"/__cadence_sys/list0/1\"},\n\t\t/* 2-ary tree */\n\t\t{\"list0\", 2, \"\"},\n\t\t{\"/__cadence_sys/list0/1\", 2, \"list0\"},\n\t\t{\"/__cadence_sys/list0/2\", 2, \"list0\"},\n\t\t{\"/__cadence_sys/list0/3\", 2, \"/__cadence_sys/list0/1\"},\n\t\t{\"/__cadence_sys/list0/4\", 2, \"/__cadence_sys/list0/1\"},\n\t\t{\"/__cadence_sys/list0/5\", 2, \"/__cadence_sys/list0/2\"},\n\t\t{\"/__cadence_sys/list0/6\", 2, \"/__cadence_sys/list0/2\"},\n\t\t/* 3-ary tree */\n\t\t{\"/__cadence_sys/list0/1\", 3, \"list0\"},\n\t\t{\"/__cadence_sys/list0/2\", 3, \"list0\"},\n\t\t{\"/__cadence_sys/list0/3\", 3, \"list0\"},\n\t\t{\"/__cadence_sys/list0/4\", 3, \"/__cadence_sys/list0/1\"},\n\t\t{\"/__cadence_sys/list0/5\", 3, \"/__cadence_sys/list0/1\"},\n\t\t{\"/__cadence_sys/list0/6\", 3, \"/__cadence_sys/list0/1\"},\n\t\t{\"/__cadence_sys/list0/7\", 3, \"/__cadence_sys/list0/2\"},\n\t\t{\"/__cadence_sys/list0/10\", 3, \"/__cadence_sys/list0/3\"},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name+\"#\"+strconv.Itoa(tc.degree), func(t *testing.T) {\n\t\t\ttn, err := newTaskListName(tc.name)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, tc.output, tn.Parent(tc.degree))\n\t\t})\n\t}\n}\n\nfunc TestInvalidTasklistNames(t *testing.T) {\n\tinputs := []string{\n\t\t\"/__cadence_sys/\",\n\t\t\"/__cadence_sys/0\",\n\t\t\"/__cadence_sys//1\",\n\t\t\"/__cadence_sys//0\",\n\t\t\"/__cadence_sys/list0\",\n\t\t\"/__cadence_sys/list0/0\",\n\t\t\"/__cadence_sys/list0/-1\",\n\t}\n\tfor _, name := range inputs {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t_, err := newTaskListName(name)\n\t\t\trequire.Error(t, err)\n\n\t\t\t// NewIdentifier should validate taskListName as well\n\t\t\t_, err = NewIdentifier(\"domain-name\", name, persistence.TaskListTypeActivity)\n\t\t\trequire.Error(t, err)\n\t\t})\n\t}\n}\n\nfunc TestTaskListIDToString(t *testing.T) {\n\tid, err := NewIdentifier(\"test-domain\", \"/tasklist/\", persistence.TaskListTypeActivity)\n\trequire.NoError(t, err)\n\trequire.Equal(t, \"[name=/tasklist/type=activity]\", fmt.Sprint(id))\n\n\tid, err = NewIdentifier(\"test-domain\", \"/tasklist/\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\trequire.Equal(t, \"[name=/tasklist/type=decision]\", fmt.Sprint(id))\n}\n"
  },
  {
    "path": "service/matching/tasklist/interfaces.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist Manager\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist TaskListRegistry\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist TaskMatcher\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist Forwarder\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist TaskCompleter\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist ShardProcessor\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype (\n\t// TaskListRegistry is a registry of task list managers\n\t// it tracks all task list managers and provides a way to get them by identifier, domain ID, or task list name\n\tTaskListRegistry interface {\n\t\t// Register registers a manager for a given identifier.\n\t\t// we can override the manager for the same identifier if it is already registered\n\t\t// this case should be handled by the caller\n\t\tRegister(id Identifier, mgr Manager)\n\n\t\t// Unregister unregisters a manager for a given identifier.\n\t\t// it returns true if the manager was unregistered, false if it was not found\n\t\tUnregister(mgr Manager) bool\n\n\t\t// AllManagers returns a list of all managers.\n\t\tAllManagers() []Manager\n\n\t\tManagersByDomainID(domainID string) []Manager\n\t\tManagersByTaskListName(name string) []Manager\n\t\t// ManagerByTaskListIdentifier returns a manager for a given identifier.\n\t\t// it returns the manager and true if it was found, false if it was not found\n\t\tManagerByTaskListIdentifier(id Identifier) (Manager, bool)\n\t}\n\n\tManager interface {\n\t\tStart(ctx context.Context) error\n\t\tStop()\n\t\t// AddTask adds a task to the task list. This method will first attempt a synchronous\n\t\t// match with a poller. When that fails, task will be written to database and later\n\t\t// asynchronously matched with a poller\n\t\tAddTask(ctx context.Context, params AddTaskParams) (syncMatch bool, err error)\n\t\t// GetTask blocks waiting for a task Returns error when context deadline is exceeded\n\t\t// maxDispatchPerSecond is the max rate at which tasks are allowed to be dispatched\n\t\t// from this task list to pollers\n\t\tGetTask(ctx context.Context, maxDispatchPerSecond *float64) (*InternalTask, error)\n\t\t// DispatchTask dispatches a task to a poller. When there are no pollers to pick\n\t\t// up the task, this method will return error. Task will not be persisted to db\n\t\tDispatchTask(ctx context.Context, task *InternalTask) error\n\t\t// DispatchQueryTask will dispatch query to local or remote poller. If forwarded then result or error is returned,\n\t\t// if dispatched to local poller then nil and nil is returned.\n\t\tDispatchQueryTask(ctx context.Context, taskID string, request *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error)\n\t\tCancelPoller(pollerID string)\n\t\tGetAllPollerInfo() []*types.PollerInfo\n\t\tHasPollerAfter(accessTime time.Time) bool\n\t\t// DescribeTaskList returns information about the target tasklist\n\t\tDescribeTaskList(includeTaskListStatus bool) *types.DescribeTaskListResponse\n\t\tString() string\n\t\tGetTaskListKind() types.TaskListKind\n\t\tTaskListID() *Identifier\n\t\tTaskListPartitionConfig() *types.TaskListPartitionConfig\n\t\tUpdateTaskListPartitionConfig(context.Context, *types.TaskListPartitionConfig) error\n\t\tRefreshTaskListPartitionConfig(context.Context, *types.TaskListPartitionConfig) error\n\t\tLoadBalancerHints() *types.LoadBalancerHints\n\t\tQueriesPerSecond() float64\n\t\tReleaseBlockedPollers() error\n\t}\n\n\tTaskMatcher interface {\n\t\tDisconnectBlockedPollers()\n\t\tOffer(ctx context.Context, task *InternalTask) (bool, error)\n\t\tOfferOrTimeout(ctx context.Context, startT time.Time, task *InternalTask) (bool, error)\n\t\tOfferQuery(ctx context.Context, task *InternalTask) (*types.MatchingQueryWorkflowResponse, error)\n\t\tMustOffer(ctx context.Context, task *InternalTask) error\n\t\tPoll(ctx context.Context, isolationGroup string) (*InternalTask, error)\n\t\tPollForQuery(ctx context.Context) (*InternalTask, error)\n\t\tRefreshCancelContext()\n\t}\n\n\tForwarder interface {\n\t\tForwardTask(ctx context.Context, task *InternalTask) error\n\t\tForwardQueryTask(ctx context.Context, task *InternalTask) (*types.MatchingQueryWorkflowResponse, error)\n\t\tForwardPoll(ctx context.Context) (*InternalTask, error)\n\t\tAddReqTokenC() <-chan *ForwarderReqToken\n\t\tPollReqTokenC() <-chan *ForwarderReqToken\n\t}\n\n\tTaskCompleter interface {\n\t\tCompleteTaskIfStarted(ctx context.Context, task *InternalTask) error\n\t}\n\n\tShardProcessor interface {\n\t\tStart(ctx context.Context) error\n\t\tStop()\n\t\tGetShardReport() executorclient.ShardReport\n\t\tSetShardStatus(types.ShardStatus)\n\t}\n)\n"
  },
  {
    "path": "service/matching/tasklist/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package tasklist -source interfaces.go -destination interfaces_mock.go github.com/uber/cadence/service/matching/tasklist ShardProcessor\n//\n\n// Package tasklist is a generated GoMock package.\npackage tasklist\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\ttime \"time\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n\texecutorclient \"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// MockTaskListRegistry is a mock of TaskListRegistry interface.\ntype MockTaskListRegistry struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskListRegistryMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskListRegistryMockRecorder is the mock recorder for MockTaskListRegistry.\ntype MockTaskListRegistryMockRecorder struct {\n\tmock *MockTaskListRegistry\n}\n\n// NewMockTaskListRegistry creates a new mock instance.\nfunc NewMockTaskListRegistry(ctrl *gomock.Controller) *MockTaskListRegistry {\n\tmock := &MockTaskListRegistry{ctrl: ctrl}\n\tmock.recorder = &MockTaskListRegistryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskListRegistry) EXPECT() *MockTaskListRegistryMockRecorder {\n\treturn m.recorder\n}\n\n// AllManagers mocks base method.\nfunc (m *MockTaskListRegistry) AllManagers() []Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AllManagers\")\n\tret0, _ := ret[0].([]Manager)\n\treturn ret0\n}\n\n// AllManagers indicates an expected call of AllManagers.\nfunc (mr *MockTaskListRegistryMockRecorder) AllManagers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AllManagers\", reflect.TypeOf((*MockTaskListRegistry)(nil).AllManagers))\n}\n\n// ManagerByTaskListIdentifier mocks base method.\nfunc (m *MockTaskListRegistry) ManagerByTaskListIdentifier(id Identifier) (Manager, bool) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ManagerByTaskListIdentifier\", id)\n\tret0, _ := ret[0].(Manager)\n\tret1, _ := ret[1].(bool)\n\treturn ret0, ret1\n}\n\n// ManagerByTaskListIdentifier indicates an expected call of ManagerByTaskListIdentifier.\nfunc (mr *MockTaskListRegistryMockRecorder) ManagerByTaskListIdentifier(id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ManagerByTaskListIdentifier\", reflect.TypeOf((*MockTaskListRegistry)(nil).ManagerByTaskListIdentifier), id)\n}\n\n// ManagersByDomainID mocks base method.\nfunc (m *MockTaskListRegistry) ManagersByDomainID(domainID string) []Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ManagersByDomainID\", domainID)\n\tret0, _ := ret[0].([]Manager)\n\treturn ret0\n}\n\n// ManagersByDomainID indicates an expected call of ManagersByDomainID.\nfunc (mr *MockTaskListRegistryMockRecorder) ManagersByDomainID(domainID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ManagersByDomainID\", reflect.TypeOf((*MockTaskListRegistry)(nil).ManagersByDomainID), domainID)\n}\n\n// ManagersByTaskListName mocks base method.\nfunc (m *MockTaskListRegistry) ManagersByTaskListName(name string) []Manager {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ManagersByTaskListName\", name)\n\tret0, _ := ret[0].([]Manager)\n\treturn ret0\n}\n\n// ManagersByTaskListName indicates an expected call of ManagersByTaskListName.\nfunc (mr *MockTaskListRegistryMockRecorder) ManagersByTaskListName(name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ManagersByTaskListName\", reflect.TypeOf((*MockTaskListRegistry)(nil).ManagersByTaskListName), name)\n}\n\n// Register mocks base method.\nfunc (m *MockTaskListRegistry) Register(id Identifier, mgr Manager) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Register\", id, mgr)\n}\n\n// Register indicates an expected call of Register.\nfunc (mr *MockTaskListRegistryMockRecorder) Register(id, mgr any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Register\", reflect.TypeOf((*MockTaskListRegistry)(nil).Register), id, mgr)\n}\n\n// Unregister mocks base method.\nfunc (m *MockTaskListRegistry) Unregister(mgr Manager) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Unregister\", mgr)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// Unregister indicates an expected call of Unregister.\nfunc (mr *MockTaskListRegistryMockRecorder) Unregister(mgr any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Unregister\", reflect.TypeOf((*MockTaskListRegistry)(nil).Unregister), mgr)\n}\n\n// MockManager is a mock of Manager interface.\ntype MockManager struct {\n\tctrl     *gomock.Controller\n\trecorder *MockManagerMockRecorder\n\tisgomock struct{}\n}\n\n// MockManagerMockRecorder is the mock recorder for MockManager.\ntype MockManagerMockRecorder struct {\n\tmock *MockManager\n}\n\n// NewMockManager creates a new mock instance.\nfunc NewMockManager(ctrl *gomock.Controller) *MockManager {\n\tmock := &MockManager{ctrl: ctrl}\n\tmock.recorder = &MockManagerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockManager) EXPECT() *MockManagerMockRecorder {\n\treturn m.recorder\n}\n\n// AddTask mocks base method.\nfunc (m *MockManager) AddTask(ctx context.Context, params AddTaskParams) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddTask\", ctx, params)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AddTask indicates an expected call of AddTask.\nfunc (mr *MockManagerMockRecorder) AddTask(ctx, params any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddTask\", reflect.TypeOf((*MockManager)(nil).AddTask), ctx, params)\n}\n\n// CancelPoller mocks base method.\nfunc (m *MockManager) CancelPoller(pollerID string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"CancelPoller\", pollerID)\n}\n\n// CancelPoller indicates an expected call of CancelPoller.\nfunc (mr *MockManagerMockRecorder) CancelPoller(pollerID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CancelPoller\", reflect.TypeOf((*MockManager)(nil).CancelPoller), pollerID)\n}\n\n// DescribeTaskList mocks base method.\nfunc (m *MockManager) DescribeTaskList(includeTaskListStatus bool) *types.DescribeTaskListResponse {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DescribeTaskList\", includeTaskListStatus)\n\tret0, _ := ret[0].(*types.DescribeTaskListResponse)\n\treturn ret0\n}\n\n// DescribeTaskList indicates an expected call of DescribeTaskList.\nfunc (mr *MockManagerMockRecorder) DescribeTaskList(includeTaskListStatus any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DescribeTaskList\", reflect.TypeOf((*MockManager)(nil).DescribeTaskList), includeTaskListStatus)\n}\n\n// DispatchQueryTask mocks base method.\nfunc (m *MockManager) DispatchQueryTask(ctx context.Context, taskID string, request *types.MatchingQueryWorkflowRequest) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DispatchQueryTask\", ctx, taskID, request)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// DispatchQueryTask indicates an expected call of DispatchQueryTask.\nfunc (mr *MockManagerMockRecorder) DispatchQueryTask(ctx, taskID, request any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DispatchQueryTask\", reflect.TypeOf((*MockManager)(nil).DispatchQueryTask), ctx, taskID, request)\n}\n\n// DispatchTask mocks base method.\nfunc (m *MockManager) DispatchTask(ctx context.Context, task *InternalTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DispatchTask\", ctx, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DispatchTask indicates an expected call of DispatchTask.\nfunc (mr *MockManagerMockRecorder) DispatchTask(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DispatchTask\", reflect.TypeOf((*MockManager)(nil).DispatchTask), ctx, task)\n}\n\n// GetAllPollerInfo mocks base method.\nfunc (m *MockManager) GetAllPollerInfo() []*types.PollerInfo {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetAllPollerInfo\")\n\tret0, _ := ret[0].([]*types.PollerInfo)\n\treturn ret0\n}\n\n// GetAllPollerInfo indicates an expected call of GetAllPollerInfo.\nfunc (mr *MockManagerMockRecorder) GetAllPollerInfo() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetAllPollerInfo\", reflect.TypeOf((*MockManager)(nil).GetAllPollerInfo))\n}\n\n// GetTask mocks base method.\nfunc (m *MockManager) GetTask(ctx context.Context, maxDispatchPerSecond *float64) (*InternalTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTask\", ctx, maxDispatchPerSecond)\n\tret0, _ := ret[0].(*InternalTask)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetTask indicates an expected call of GetTask.\nfunc (mr *MockManagerMockRecorder) GetTask(ctx, maxDispatchPerSecond any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTask\", reflect.TypeOf((*MockManager)(nil).GetTask), ctx, maxDispatchPerSecond)\n}\n\n// GetTaskListKind mocks base method.\nfunc (m *MockManager) GetTaskListKind() types.TaskListKind {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetTaskListKind\")\n\tret0, _ := ret[0].(types.TaskListKind)\n\treturn ret0\n}\n\n// GetTaskListKind indicates an expected call of GetTaskListKind.\nfunc (mr *MockManagerMockRecorder) GetTaskListKind() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetTaskListKind\", reflect.TypeOf((*MockManager)(nil).GetTaskListKind))\n}\n\n// HasPollerAfter mocks base method.\nfunc (m *MockManager) HasPollerAfter(accessTime time.Time) bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HasPollerAfter\", accessTime)\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// HasPollerAfter indicates an expected call of HasPollerAfter.\nfunc (mr *MockManagerMockRecorder) HasPollerAfter(accessTime any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HasPollerAfter\", reflect.TypeOf((*MockManager)(nil).HasPollerAfter), accessTime)\n}\n\n// LoadBalancerHints mocks base method.\nfunc (m *MockManager) LoadBalancerHints() *types.LoadBalancerHints {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"LoadBalancerHints\")\n\tret0, _ := ret[0].(*types.LoadBalancerHints)\n\treturn ret0\n}\n\n// LoadBalancerHints indicates an expected call of LoadBalancerHints.\nfunc (mr *MockManagerMockRecorder) LoadBalancerHints() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"LoadBalancerHints\", reflect.TypeOf((*MockManager)(nil).LoadBalancerHints))\n}\n\n// QueriesPerSecond mocks base method.\nfunc (m *MockManager) QueriesPerSecond() float64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"QueriesPerSecond\")\n\tret0, _ := ret[0].(float64)\n\treturn ret0\n}\n\n// QueriesPerSecond indicates an expected call of QueriesPerSecond.\nfunc (mr *MockManagerMockRecorder) QueriesPerSecond() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"QueriesPerSecond\", reflect.TypeOf((*MockManager)(nil).QueriesPerSecond))\n}\n\n// RefreshTaskListPartitionConfig mocks base method.\nfunc (m *MockManager) RefreshTaskListPartitionConfig(arg0 context.Context, arg1 *types.TaskListPartitionConfig) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RefreshTaskListPartitionConfig\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RefreshTaskListPartitionConfig indicates an expected call of RefreshTaskListPartitionConfig.\nfunc (mr *MockManagerMockRecorder) RefreshTaskListPartitionConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshTaskListPartitionConfig\", reflect.TypeOf((*MockManager)(nil).RefreshTaskListPartitionConfig), arg0, arg1)\n}\n\n// ReleaseBlockedPollers mocks base method.\nfunc (m *MockManager) ReleaseBlockedPollers() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReleaseBlockedPollers\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ReleaseBlockedPollers indicates an expected call of ReleaseBlockedPollers.\nfunc (mr *MockManagerMockRecorder) ReleaseBlockedPollers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReleaseBlockedPollers\", reflect.TypeOf((*MockManager)(nil).ReleaseBlockedPollers))\n}\n\n// Start mocks base method.\nfunc (m *MockManager) Start(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockManagerMockRecorder) Start(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockManager)(nil).Start), ctx)\n}\n\n// Stop mocks base method.\nfunc (m *MockManager) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockManagerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockManager)(nil).Stop))\n}\n\n// String mocks base method.\nfunc (m *MockManager) String() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"String\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// String indicates an expected call of String.\nfunc (mr *MockManagerMockRecorder) String() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"String\", reflect.TypeOf((*MockManager)(nil).String))\n}\n\n// TaskListID mocks base method.\nfunc (m *MockManager) TaskListID() *Identifier {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskListID\")\n\tret0, _ := ret[0].(*Identifier)\n\treturn ret0\n}\n\n// TaskListID indicates an expected call of TaskListID.\nfunc (mr *MockManagerMockRecorder) TaskListID() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskListID\", reflect.TypeOf((*MockManager)(nil).TaskListID))\n}\n\n// TaskListPartitionConfig mocks base method.\nfunc (m *MockManager) TaskListPartitionConfig() *types.TaskListPartitionConfig {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"TaskListPartitionConfig\")\n\tret0, _ := ret[0].(*types.TaskListPartitionConfig)\n\treturn ret0\n}\n\n// TaskListPartitionConfig indicates an expected call of TaskListPartitionConfig.\nfunc (mr *MockManagerMockRecorder) TaskListPartitionConfig() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TaskListPartitionConfig\", reflect.TypeOf((*MockManager)(nil).TaskListPartitionConfig))\n}\n\n// UpdateTaskListPartitionConfig mocks base method.\nfunc (m *MockManager) UpdateTaskListPartitionConfig(arg0 context.Context, arg1 *types.TaskListPartitionConfig) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateTaskListPartitionConfig\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateTaskListPartitionConfig indicates an expected call of UpdateTaskListPartitionConfig.\nfunc (mr *MockManagerMockRecorder) UpdateTaskListPartitionConfig(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateTaskListPartitionConfig\", reflect.TypeOf((*MockManager)(nil).UpdateTaskListPartitionConfig), arg0, arg1)\n}\n\n// MockTaskMatcher is a mock of TaskMatcher interface.\ntype MockTaskMatcher struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskMatcherMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskMatcherMockRecorder is the mock recorder for MockTaskMatcher.\ntype MockTaskMatcherMockRecorder struct {\n\tmock *MockTaskMatcher\n}\n\n// NewMockTaskMatcher creates a new mock instance.\nfunc NewMockTaskMatcher(ctrl *gomock.Controller) *MockTaskMatcher {\n\tmock := &MockTaskMatcher{ctrl: ctrl}\n\tmock.recorder = &MockTaskMatcherMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskMatcher) EXPECT() *MockTaskMatcherMockRecorder {\n\treturn m.recorder\n}\n\n// DisconnectBlockedPollers mocks base method.\nfunc (m *MockTaskMatcher) DisconnectBlockedPollers() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"DisconnectBlockedPollers\")\n}\n\n// DisconnectBlockedPollers indicates an expected call of DisconnectBlockedPollers.\nfunc (mr *MockTaskMatcherMockRecorder) DisconnectBlockedPollers() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DisconnectBlockedPollers\", reflect.TypeOf((*MockTaskMatcher)(nil).DisconnectBlockedPollers))\n}\n\n// MustOffer mocks base method.\nfunc (m *MockTaskMatcher) MustOffer(ctx context.Context, task *InternalTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MustOffer\", ctx, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// MustOffer indicates an expected call of MustOffer.\nfunc (mr *MockTaskMatcherMockRecorder) MustOffer(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MustOffer\", reflect.TypeOf((*MockTaskMatcher)(nil).MustOffer), ctx, task)\n}\n\n// Offer mocks base method.\nfunc (m *MockTaskMatcher) Offer(ctx context.Context, task *InternalTask) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Offer\", ctx, task)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Offer indicates an expected call of Offer.\nfunc (mr *MockTaskMatcherMockRecorder) Offer(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Offer\", reflect.TypeOf((*MockTaskMatcher)(nil).Offer), ctx, task)\n}\n\n// OfferOrTimeout mocks base method.\nfunc (m *MockTaskMatcher) OfferOrTimeout(ctx context.Context, startT time.Time, task *InternalTask) (bool, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"OfferOrTimeout\", ctx, startT, task)\n\tret0, _ := ret[0].(bool)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// OfferOrTimeout indicates an expected call of OfferOrTimeout.\nfunc (mr *MockTaskMatcherMockRecorder) OfferOrTimeout(ctx, startT, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"OfferOrTimeout\", reflect.TypeOf((*MockTaskMatcher)(nil).OfferOrTimeout), ctx, startT, task)\n}\n\n// OfferQuery mocks base method.\nfunc (m *MockTaskMatcher) OfferQuery(ctx context.Context, task *InternalTask) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"OfferQuery\", ctx, task)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// OfferQuery indicates an expected call of OfferQuery.\nfunc (mr *MockTaskMatcherMockRecorder) OfferQuery(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"OfferQuery\", reflect.TypeOf((*MockTaskMatcher)(nil).OfferQuery), ctx, task)\n}\n\n// Poll mocks base method.\nfunc (m *MockTaskMatcher) Poll(ctx context.Context, isolationGroup string) (*InternalTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Poll\", ctx, isolationGroup)\n\tret0, _ := ret[0].(*InternalTask)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Poll indicates an expected call of Poll.\nfunc (mr *MockTaskMatcherMockRecorder) Poll(ctx, isolationGroup any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Poll\", reflect.TypeOf((*MockTaskMatcher)(nil).Poll), ctx, isolationGroup)\n}\n\n// PollForQuery mocks base method.\nfunc (m *MockTaskMatcher) PollForQuery(ctx context.Context) (*InternalTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollForQuery\", ctx)\n\tret0, _ := ret[0].(*InternalTask)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// PollForQuery indicates an expected call of PollForQuery.\nfunc (mr *MockTaskMatcherMockRecorder) PollForQuery(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollForQuery\", reflect.TypeOf((*MockTaskMatcher)(nil).PollForQuery), ctx)\n}\n\n// RefreshCancelContext mocks base method.\nfunc (m *MockTaskMatcher) RefreshCancelContext() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RefreshCancelContext\")\n}\n\n// RefreshCancelContext indicates an expected call of RefreshCancelContext.\nfunc (mr *MockTaskMatcherMockRecorder) RefreshCancelContext() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RefreshCancelContext\", reflect.TypeOf((*MockTaskMatcher)(nil).RefreshCancelContext))\n}\n\n// MockForwarder is a mock of Forwarder interface.\ntype MockForwarder struct {\n\tctrl     *gomock.Controller\n\trecorder *MockForwarderMockRecorder\n\tisgomock struct{}\n}\n\n// MockForwarderMockRecorder is the mock recorder for MockForwarder.\ntype MockForwarderMockRecorder struct {\n\tmock *MockForwarder\n}\n\n// NewMockForwarder creates a new mock instance.\nfunc NewMockForwarder(ctrl *gomock.Controller) *MockForwarder {\n\tmock := &MockForwarder{ctrl: ctrl}\n\tmock.recorder = &MockForwarderMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockForwarder) EXPECT() *MockForwarderMockRecorder {\n\treturn m.recorder\n}\n\n// AddReqTokenC mocks base method.\nfunc (m *MockForwarder) AddReqTokenC() <-chan *ForwarderReqToken {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AddReqTokenC\")\n\tret0, _ := ret[0].(<-chan *ForwarderReqToken)\n\treturn ret0\n}\n\n// AddReqTokenC indicates an expected call of AddReqTokenC.\nfunc (mr *MockForwarderMockRecorder) AddReqTokenC() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AddReqTokenC\", reflect.TypeOf((*MockForwarder)(nil).AddReqTokenC))\n}\n\n// ForwardPoll mocks base method.\nfunc (m *MockForwarder) ForwardPoll(ctx context.Context) (*InternalTask, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ForwardPoll\", ctx)\n\tret0, _ := ret[0].(*InternalTask)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ForwardPoll indicates an expected call of ForwardPoll.\nfunc (mr *MockForwarderMockRecorder) ForwardPoll(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ForwardPoll\", reflect.TypeOf((*MockForwarder)(nil).ForwardPoll), ctx)\n}\n\n// ForwardQueryTask mocks base method.\nfunc (m *MockForwarder) ForwardQueryTask(ctx context.Context, task *InternalTask) (*types.MatchingQueryWorkflowResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ForwardQueryTask\", ctx, task)\n\tret0, _ := ret[0].(*types.MatchingQueryWorkflowResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ForwardQueryTask indicates an expected call of ForwardQueryTask.\nfunc (mr *MockForwarderMockRecorder) ForwardQueryTask(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ForwardQueryTask\", reflect.TypeOf((*MockForwarder)(nil).ForwardQueryTask), ctx, task)\n}\n\n// ForwardTask mocks base method.\nfunc (m *MockForwarder) ForwardTask(ctx context.Context, task *InternalTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ForwardTask\", ctx, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ForwardTask indicates an expected call of ForwardTask.\nfunc (mr *MockForwarderMockRecorder) ForwardTask(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ForwardTask\", reflect.TypeOf((*MockForwarder)(nil).ForwardTask), ctx, task)\n}\n\n// PollReqTokenC mocks base method.\nfunc (m *MockForwarder) PollReqTokenC() <-chan *ForwarderReqToken {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"PollReqTokenC\")\n\tret0, _ := ret[0].(<-chan *ForwarderReqToken)\n\treturn ret0\n}\n\n// PollReqTokenC indicates an expected call of PollReqTokenC.\nfunc (mr *MockForwarderMockRecorder) PollReqTokenC() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"PollReqTokenC\", reflect.TypeOf((*MockForwarder)(nil).PollReqTokenC))\n}\n\n// MockTaskCompleter is a mock of TaskCompleter interface.\ntype MockTaskCompleter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTaskCompleterMockRecorder\n\tisgomock struct{}\n}\n\n// MockTaskCompleterMockRecorder is the mock recorder for MockTaskCompleter.\ntype MockTaskCompleterMockRecorder struct {\n\tmock *MockTaskCompleter\n}\n\n// NewMockTaskCompleter creates a new mock instance.\nfunc NewMockTaskCompleter(ctrl *gomock.Controller) *MockTaskCompleter {\n\tmock := &MockTaskCompleter{ctrl: ctrl}\n\tmock.recorder = &MockTaskCompleterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTaskCompleter) EXPECT() *MockTaskCompleterMockRecorder {\n\treturn m.recorder\n}\n\n// CompleteTaskIfStarted mocks base method.\nfunc (m *MockTaskCompleter) CompleteTaskIfStarted(ctx context.Context, task *InternalTask) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CompleteTaskIfStarted\", ctx, task)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CompleteTaskIfStarted indicates an expected call of CompleteTaskIfStarted.\nfunc (mr *MockTaskCompleterMockRecorder) CompleteTaskIfStarted(ctx, task any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CompleteTaskIfStarted\", reflect.TypeOf((*MockTaskCompleter)(nil).CompleteTaskIfStarted), ctx, task)\n}\n\n// MockShardProcessor is a mock of ShardProcessor interface.\ntype MockShardProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardProcessorMockRecorder is the mock recorder for MockShardProcessor.\ntype MockShardProcessorMockRecorder struct {\n\tmock *MockShardProcessor\n}\n\n// NewMockShardProcessor creates a new mock instance.\nfunc NewMockShardProcessor(ctrl *gomock.Controller) *MockShardProcessor {\n\tmock := &MockShardProcessor{ctrl: ctrl}\n\tmock.recorder = &MockShardProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardProcessor) EXPECT() *MockShardProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// GetShardReport mocks base method.\nfunc (m *MockShardProcessor) GetShardReport() executorclient.ShardReport {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardReport\")\n\tret0, _ := ret[0].(executorclient.ShardReport)\n\treturn ret0\n}\n\n// GetShardReport indicates an expected call of GetShardReport.\nfunc (mr *MockShardProcessorMockRecorder) GetShardReport() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardReport\", reflect.TypeOf((*MockShardProcessor)(nil).GetShardReport))\n}\n\n// SetShardStatus mocks base method.\nfunc (m *MockShardProcessor) SetShardStatus(arg0 types.ShardStatus) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetShardStatus\", arg0)\n}\n\n// SetShardStatus indicates an expected call of SetShardStatus.\nfunc (mr *MockShardProcessorMockRecorder) SetShardStatus(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetShardStatus\", reflect.TypeOf((*MockShardProcessor)(nil).SetShardStatus), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockShardProcessor) Start(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockShardProcessorMockRecorder) Start(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockShardProcessor)(nil).Start), ctx)\n}\n\n// Stop mocks base method.\nfunc (m *MockShardProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockShardProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockShardProcessor)(nil).Stop))\n}\n"
  },
  {
    "path": "service/matching/tasklist/isolation_balancer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"math\"\n\t\"slices\"\n\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\ntype (\n\tisolationBalancer struct {\n\t\ttimeSource clock.TimeSource\n\t\tscope      metrics.Scope\n\t\tconfig     *config.TaskListConfig\n\t\tgroupState map[string]*isolationGroupState\n\t}\n\n\tisolationGroupState struct {\n\t\tunderLoad             clock.Sustain\n\t\toverLoad              clock.Sustain\n\t\tnoPollers             clock.Sustain\n\t\thasPollers            clock.Sustain\n\t\tcanAssignToPartitions bool\n\t}\n)\n\nfunc newIsolationBalancer(timeSource clock.TimeSource, scope metrics.Scope, config *config.TaskListConfig) *isolationBalancer {\n\treturn &isolationBalancer{\n\t\ttimeSource: timeSource,\n\t\tscope:      scope,\n\t\tconfig:     config,\n\t\tgroupState: make(map[string]*isolationGroupState),\n\t}\n}\n\nfunc (i *isolationBalancer) adjustWritePartitions(metrics *aggregatePartitionMetrics, writePartitions map[int]*types.TaskListPartition) (map[int]*types.TaskListPartition, bool) {\n\ti.refreshGroups(metrics, writePartitions)\n\tpartitionCount := i.getPartitionsPerGroup(writePartitions)\n\n\tpartitionGoal := i.calculatePartitionGoal(metrics, partitionCount, len(writePartitions))\n\n\t// If there's a single partition we can't actually change anything, but we still want to run the above calculations\n\t// so that we record the correct sustain values. That way when a new partition is added we can immediately respond\n\t// with the correct number of pollers\n\tif len(writePartitions) == 1 {\n\t\t// If we resized to 1 partition, remove all assignments from that partition and clear all group status\n\t\tif len(writePartitions[0].IsolationGroups) != 0 {\n\t\t\ti.reset()\n\t\t\treturn map[int]*types.TaskListPartition{\n\t\t\t\t0: {},\n\t\t\t}, true\n\t\t}\n\t\treturn writePartitions, false\n\t}\n\n\treturn i.applyGroupChanges(metrics, partitionCount, partitionGoal, writePartitions)\n}\n\nfunc (i *isolationBalancer) reset() {\n\tmaps.Clear(i.groupState)\n}\n\nfunc (i *isolationBalancer) refreshGroups(metrics *aggregatePartitionMetrics, writePartitions map[int]*types.TaskListPartition) {\n\tpolledOrAssigned := make(map[string]bool)\n\t// Track it for resizing/removal\n\tfor _, p := range writePartitions {\n\t\tfor _, group := range p.IsolationGroups {\n\t\t\tpolledOrAssigned[group] = true\n\t\t\tif _, ok := i.groupState[group]; !ok {\n\t\t\t\ti.groupState[group] = i.createIsolationGroupState(true)\n\t\t\t}\n\t\t}\n\t}\n\t// Track it as we might assign it if they maintain pollers\n\tfor group := range metrics.hasPollersByIsolationGroup {\n\t\tpolledOrAssigned[group] = true\n\t\tif _, ok := i.groupState[group]; !ok {\n\t\t\ti.groupState[group] = i.createIsolationGroupState(false)\n\t\t}\n\t}\n\tfor group := range i.groupState {\n\t\t// No pollers and no write partitions means we don't need to track it\n\t\tif !polledOrAssigned[group] {\n\t\t\tdelete(i.groupState, group)\n\t\t\tcontinue\n\t\t}\n\t}\n}\n\nfunc (i *isolationBalancer) createIsolationGroupState(assignToPartitions bool) *isolationGroupState {\n\treturn &isolationGroupState{\n\t\tunderLoad:             clock.NewSustain(i.timeSource, i.config.IsolationGroupDownscaleSustainedDuration),\n\t\toverLoad:              clock.NewSustain(i.timeSource, i.config.IsolationGroupUpscaleSustainedDuration),\n\t\tnoPollers:             clock.NewSustain(i.timeSource, i.config.IsolationGroupNoPollersSustainedDuration),\n\t\thasPollers:            clock.NewSustain(i.timeSource, i.config.IsolationGroupHasPollersSustainedDuration),\n\t\tcanAssignToPartitions: assignToPartitions,\n\t}\n}\n\nfunc (i *isolationBalancer) getPartitionsPerGroup(writePartitions map[int]*types.TaskListPartition) map[string]int {\n\tpartitionCount := make(map[string]int)\n\tfor _, p := range writePartitions {\n\t\tfor _, group := range p.IsolationGroups {\n\t\t\tpartitionCount[group]++\n\t\t}\n\t}\n\treturn partitionCount\n}\n\nfunc (i *isolationBalancer) calculatePartitionGoal(aggregateMetrics *aggregatePartitionMetrics, partitionCountByGroup map[string]int, writePartitionCount int) map[string]int {\n\tpartitionGoal := make(map[string]int)\n\tassignableGroups := 0\n\t// Identify groups that have switched to assignable and count them. This impacts the minimum partitions per group\n\tfor group, state := range i.groupState {\n\t\tgroupHasPollers := aggregateMetrics.hasPollersByIsolationGroup[group]\n\t\tsustainedPollers := state.hasPollers.CheckAndReset(groupHasPollers)\n\t\tsustainedNoPollers := state.noPollers.CheckAndReset(!groupHasPollers)\n\t\tif state.canAssignToPartitions {\n\t\t\tif sustainedNoPollers {\n\t\t\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).IncCounter(metrics.IsolationGroupStoppedPolling)\n\t\t\t\tstate.canAssignToPartitions = false\n\t\t\t} else {\n\t\t\t\tassignableGroups++\n\t\t\t}\n\t\t} else if sustainedPollers {\n\t\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).IncCounter(metrics.IsolationGroupStartedPolling)\n\t\t\tstate.canAssignToPartitions = true\n\t\t\tstate.overLoad.Reset()\n\t\t\tstate.underLoad.Reset()\n\t\t\tassignableGroups++\n\t\t}\n\t}\n\n\tminPartitions := i.getMinPartitionsPerGroup(writePartitionCount, assignableGroups)\n\n\tfor group, state := range i.groupState {\n\t\t// Remove all assignments from inactive groups\n\t\tif !state.canAssignToPartitions {\n\t\t\tpartitionGoal[group] = 0\n\t\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).UpdateGauge(metrics.IsolationGroupPartitionsGauge, 0)\n\t\t\tcontinue\n\t\t}\n\t\tcurrentPartitions := partitionCountByGroup[group]\n\t\tgroupQPS := aggregateMetrics.qpsByIsolationGroup[group]\n\n\t\tupscaleRps := float64(i.config.PartitionUpscaleRPS()) / float64(i.config.IsolationGroupsPerPartition())\n\t\tdownscaleFactor := i.config.PartitionDownscaleFactor()\n\t\tupscaleThreshold := float64(max(currentPartitions, 1)) * upscaleRps\n\t\tdownscaleThreshold := float64(max(currentPartitions-1, 0)) * upscaleRps * downscaleFactor\n\n\t\tsustainedOverLoad := state.overLoad.Check(groupQPS > upscaleThreshold)\n\t\tsustainedUnderLoad := state.underLoad.Check(groupQPS < downscaleThreshold)\n\n\t\tidealPartitions := getNumberOfPartitions(groupQPS, upscaleRps)\n\t\t// partitions must be >= minPartitions and <= writePartitionCount\n\t\ttargetPartitions := min(max(minPartitions, idealPartitions), writePartitionCount)\n\n\t\t// We may have a sustained overload but already be at the max number of partitions. Only reset the sustain\n\t\t// if we can actually make changes. This allows for instantly scaling up if the number of write partitions\n\t\t// changes\n\t\t// If currentPartitions == 0 then we just scaled up from not having pollers. Allow for more than the minimum\n\t\tif (sustainedOverLoad || sustainedUnderLoad || currentPartitions == 0) && targetPartitions != currentPartitions {\n\t\t\tif sustainedOverLoad {\n\t\t\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).IncCounter(metrics.IsolationGroupUpscale)\n\t\t\t} else if sustainedUnderLoad {\n\t\t\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).IncCounter(metrics.IsolationGroupDownscale)\n\t\t\t}\n\t\t\tstate.overLoad.Reset()\n\t\t\tstate.underLoad.Reset()\n\t\t\tpartitionGoal[group] = targetPartitions\n\t\t} else {\n\t\t\tpartitionGoal[group] = max(currentPartitions, minPartitions)\n\t\t}\n\t\ti.scope.Tagged(metrics.IsolationGroupTag(group)).UpdateGauge(metrics.IsolationGroupPartitionsGauge, float64(partitionGoal[group]))\n\t}\n\n\treturn partitionGoal\n}\n\nfunc (i *isolationBalancer) applyGroupChanges(m *aggregatePartitionMetrics, partitionCount, partitionGoal map[string]int, partitions map[int]*types.TaskListPartition) (map[int]*types.TaskListPartition, bool) {\n\tgroupSizePerPartition := make(map[string]float64)\n\tvar toRemove []string\n\tvar toAdd []string\n\tassignableGroups := 0\n\tfor group, state := range i.groupState {\n\t\tif state.canAssignToPartitions {\n\t\t\tassignableGroups++\n\t\t}\n\t\tchange := partitionGoal[group] - partitionCount[group]\n\t\tgoalPartitions := partitionGoal[group]\n\t\tif goalPartitions > 0 {\n\t\t\tgroupSizePerPartition[group] = m.qpsByIsolationGroup[group] / float64(goalPartitions)\n\t\t}\n\t\tfor j := 0; j < change; j++ {\n\t\t\ttoAdd = append(toAdd, group)\n\t\t}\n\t\tfor j := change; j < 0; j++ {\n\t\t\ttoRemove = append(toRemove, group)\n\t\t}\n\t}\n\t// partition id to set of groups\n\tpartitionGroups := make(map[int]map[string]any)\n\tfor id, partition := range partitions {\n\t\tset := make(map[string]any)\n\t\tfor _, group := range partition.IsolationGroups {\n\t\t\tset[group] = true\n\t\t}\n\t\tpartitionGroups[id] = set\n\t}\n\tremovePartitions(toRemove, groupSizePerPartition, partitionGroups)\n\taddPartitions(toAdd, groupSizePerPartition, partitionGroups)\n\tminimumGroups := i.getMinGroupsPerPartition(assignableGroups)\n\tmovedGroupBetweenPartitions := ensureMinimumGroupsPerPartition(minimumGroups, groupSizePerPartition, partitionGroups)\n\tresult := make(map[int]*types.TaskListPartition)\n\tfor id, groups := range partitionGroups {\n\t\tif len(groups) == 0 {\n\t\t\tresult[id] = &types.TaskListPartition{}\n\t\t} else {\n\t\t\tasSlice := maps.Keys(groups)\n\t\t\t// Sort for the sake of stability\n\t\t\tslices.Sort(asSlice)\n\t\t\tresult[id] = &types.TaskListPartition{IsolationGroups: asSlice}\n\t\t}\n\t}\n\treturn result, movedGroupBetweenPartitions || len(toRemove) > 0 || len(toAdd) > 0\n}\n\nfunc (i *isolationBalancer) getMinPartitionsPerGroup(writePartitionCount, assignableGroups int) int {\n\tif assignableGroups == 0 {\n\t\treturn 0\n\t}\n\treturn int(math.Ceil(float64(writePartitionCount*i.getMinGroupsPerPartition(assignableGroups)) / float64(assignableGroups)))\n}\n\nfunc (i *isolationBalancer) getMinGroupsPerPartition(assignableGroups int) int {\n\treturn min(i.config.IsolationGroupsPerPartition(), assignableGroups)\n}\n\nfunc removePartitions(toRemove []string, groupSizePerPartition map[string]float64, groupsByPartitionID map[int]map[string]any) {\n\t// Use a greedy approach to remove the specified groups (which may include duplicates) from the largest partition\n\t// Similarly to adding partitions we sort the input by size first\n\tslices.SortFunc(toRemove, byValueDescending(groupSizePerPartition))\n\tfor _, group := range toRemove {\n\t\tpartitionID, ok := findMaxWithGroup(group, groupSizePerPartition, groupsByPartitionID)\n\t\tif ok {\n\t\t\tdelete(groupsByPartitionID[partitionID], group)\n\t\t}\n\t}\n}\n\nfunc addPartitions(toAdd []string, groupSizePerPartition map[string]float64, groupsByPartitionID map[int]map[string]any) {\n\t// Use greedy number partitioning to add the group to the smallest partition where it isn't already present. This\n\t// is an approximate solution. Sorting the input by size in a descending manner should result in a more even\n\t// distribution.\n\tslices.SortFunc(toAdd, byValueDescending(groupSizePerPartition))\n\tfor _, group := range toAdd {\n\t\tpartitionID, ok := findMinWithoutGroup(group, groupSizePerPartition, groupsByPartitionID)\n\t\tif ok {\n\t\t\tgroupsByPartitionID[partitionID][group] = true\n\t\t}\n\t}\n}\n\nfunc ensureMinimumGroupsPerPartition(minGroups int, groupSizePerPartition map[string]float64, groupsByPartitionID map[int]map[string]any) bool {\n\t// Every partition needs to be assigned groups that have pollers, otherwise we're unable to process tasks on those\n\t// partitions.\n\t//\n\t// Sort the partitions by the number of isolation groups assigned to them, and then use a two-pointers approach\n\t// to iteratively take groups from partitions that have the most and move them to a partition that has the least.\n\t// calculatePartitionGoal enforces that there are at least \"getMinPartitionsPerGroup\" instances of each group,\n\t// which in turn ensures that every partition can be assigned \"getMinGroupsPerPartition\" by only moving groups\n\t// around.\n\t//\n\t// The specific groups that we move between partitions isn't particularly significant, since the size of each group\n\t// within a partition has an inherent upper-bound before we scale up. Ideally we keep partitions relatively close\n\t// in size, so we use a simple heuristic of minimizing the difference in total size between the two partitions that\n\t// we're considering.\n\tchanged := false\n\tpartitionIDsByGroupCount := maps.Keys(groupsByPartitionID)\n\tslices.SortFunc(partitionIDsByGroupCount, func(a, b int) int {\n\t\taLen := len(groupsByPartitionID[a])\n\t\tbLen := len(groupsByPartitionID[b])\n\t\tif aLen < bLen {\n\t\t\treturn -1\n\t\t} else if aLen == bLen {\n\t\t\t// Fall back to size ascending. Larger is later to prioritize moving groups from those partitions\n\t\t\taSize := getPartitionSize(groupSizePerPartition, groupsByPartitionID[a])\n\t\t\tbSize := getPartitionSize(groupSizePerPartition, groupsByPartitionID[b])\n\t\t\tif aSize < bSize {\n\t\t\t\treturn -1\n\t\t\t} else if aSize == bSize {\n\t\t\t\treturn 0\n\t\t\t} else {\n\t\t\t\treturn 1\n\t\t\t}\n\t\t} else {\n\t\t\treturn 1\n\t\t}\n\t})\n\tlo, hi := 0, len(partitionIDsByGroupCount)-1\n\tfor lo < hi {\n\t\tloPartition := groupsByPartitionID[partitionIDsByGroupCount[lo]]\n\t\thiPartition := groupsByPartitionID[partitionIDsByGroupCount[hi]]\n\t\tif len(loPartition) >= minGroups {\n\t\t\tlo++\n\t\t\tcontinue\n\t\t}\n\t\tif len(hiPartition) <= minGroups {\n\t\t\thi--\n\t\t\tcontinue\n\t\t}\n\t\tchanged = true\n\t\tneeded := minGroups - len(loPartition)\n\t\tavailable := len(hiPartition) - minGroups\n\t\tmovePartitions(min(needed, available), hiPartition, loPartition, groupSizePerPartition)\n\t}\n\treturn changed\n}\n\nfunc movePartitions(amount int, from, to map[string]any, groupSizePerPartition map[string]float64) {\n\tfromSize := getPartitionSize(groupSizePerPartition, from)\n\ttoSize := getPartitionSize(groupSizePerPartition, to)\n\t// This is non-optimal when amount > 1 but it's good enough\n\t// Solving it ideally is number partitioning problem, exactly what we're trying to solve across all partitions\n\tfor range amount {\n\t\tbest := math.MaxFloat64\n\t\tbestGroup := \"\"\n\t\tfor group := range from {\n\t\t\tif _, ok := to[group]; !ok {\n\t\t\t\tgroupSize := groupSizePerPartition[group]\n\t\t\t\tdiff := math.Abs(fromSize - toSize - (2 * groupSize))\n\t\t\t\tif diff < best {\n\t\t\t\t\tbest = diff\n\t\t\t\t\tbestGroup = group\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tdelete(from, bestGroup)\n\t\tfromSize -= groupSizePerPartition[bestGroup]\n\t\ttoSize += groupSizePerPartition[bestGroup]\n\t\tto[bestGroup] = true\n\t}\n}\n\nfunc findMaxWithGroup(group string, groupSizePerPartition map[string]float64, groupsByPartitionID map[int]map[string]any) (int, bool) {\n\tmaxSize := -math.MaxFloat64\n\tmaxPartition := -1\n\t// iterate in order of partitionID to make decisions deterministic\n\tfor partitionID := 0; partitionID < len(groupsByPartitionID); partitionID++ {\n\t\tset := groupsByPartitionID[partitionID]\n\t\tif _, ok := set[group]; ok {\n\t\t\tpartitionSize := getPartitionSize(groupSizePerPartition, set)\n\t\t\tif maxSize < partitionSize {\n\t\t\t\tmaxSize = partitionSize\n\t\t\t\tmaxPartition = partitionID\n\t\t\t}\n\t\t}\n\t}\n\treturn maxPartition, maxPartition != -1\n}\n\nfunc findMinWithoutGroup(group string, groupSizePerPartition map[string]float64, groupsByPartitionID map[int]map[string]any) (int, bool) {\n\tminSize := math.MaxFloat64\n\tminPartition := -1\n\t// iterate in order of partitionID to make decisions deterministic\n\tfor partitionID := 0; partitionID < len(groupsByPartitionID); partitionID++ {\n\t\tset := groupsByPartitionID[partitionID]\n\t\tif _, ok := set[group]; !ok {\n\t\t\tpartitionSize := getPartitionSize(groupSizePerPartition, set)\n\t\t\tif partitionSize < minSize {\n\t\t\t\tminSize = partitionSize\n\t\t\t\tminPartition = partitionID\n\t\t\t}\n\t\t}\n\t}\n\treturn minPartition, minPartition != -1\n}\n\nfunc getPartitionSize(groupSizePerPartition map[string]float64, groupSet map[string]any) float64 {\n\ttotal := float64(0)\n\tfor group := range groupSet {\n\t\ttotal += groupSizePerPartition[group]\n\t}\n\treturn total\n}\n\nfunc byValueDescending[T comparable](m map[T]float64) func(a, b T) int {\n\treturn func(a, b T) int {\n\t\taVal := m[a]\n\t\tbVal := m[b]\n\t\tif aVal == bVal {\n\t\t\treturn 0\n\t\t} else if aVal > bVal {\n\t\t\treturn -1\n\t\t} else {\n\t\t\treturn 1\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/isolation_balancer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nconst cycleInterval = time.Second * 15\n\ntype testCycle struct {\n\t// If not specified, defaults to the previous cycle's value\n\tmetrics *aggregatePartitionMetrics\n\t// If not specified, defaults to the previous cycle's value\n\tpartitions map[int]*types.TaskListPartition\n\t// If not specified then there should be no change this cycle. The returned value should be equal to partitions\n\t// and changed should be false\n\texpected map[int]*types.TaskListPartition\n\n\tcallback func(dynamicClient dynamicconfig.Client)\n}\n\nfunc TestAdjustWritePartitions(t *testing.T) {\n\tcases := []struct {\n\t\tname               string\n\t\tgroupsPerPartition int\n\t\tcycles             []*testCycle\n\t}{\n\t\t{\n\t\t\tname: \"initial balance\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"initial balance - from one partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"initial balance - from one partition skewed\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 300,\n\t\t\t\t\t\t\t\"b\": 75,\n\t\t\t\t\t\t\t\"c\": 2,\n\t\t\t\t\t\t\t\"d\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"one partition minimum\",\n\t\t\tgroupsPerPartition: 1,\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"require pollers\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"fewer groups than partitions\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t\t2: {},\n\t\t\t\t\t\t3: {},\n\t\t\t\t\t\t4: {},\n\t\t\t\t\t\t5: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t\t4: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t\t5: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"fewer partitions than groups\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"fewer active groups than IsolationGroupsPerPartition\",\n\t\t\tgroupsPerPartition: 4,\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"single partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"single group\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"pollers added\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"pollers removed\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\ttotalQPS: 1000,\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 101,\n\t\t\t\t\t\t\t\"b\": 100,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale up - single partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 101,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale up - multiple partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 201,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}}, // 202\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"g\"}}, // 104\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}}, // 108\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}}, // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}},      // 68\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"g\"}}, // 171\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"c\", \"f\"}}, // 175\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}},      // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale up - multiple groups\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 102,\n\t\t\t\t\t\t\t\"b\": 101,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}}, // 103\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"g\"}}, // 106\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}}, // 108\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}}, // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t// Kind of a weird case as 0 and 1 (which have the groups that need to be split)\n\t\t\t\t\t\t// also happen to be the smallest groups.\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"h\"}}, // 103\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"g\"}}, // 106\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}},      // 108\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}},      // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"immediate scale up on new partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 303,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"g\", \"h\"}}, // 206\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"f\"}},      // 209\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"d\", \"e\"}},      // 213\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{}, // If we had the space \"a\" would have scaled up on this cycle\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"g\", \"h\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"f\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"d\", \"e\"}},\n\t\t\t\t\t\t// adaptive scaler gave us another partition\n\t\t\t\t\t\t3: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"g\", \"h\"}}, // 81.75\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"c\", \"f\"}}, // 183.75\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"d\", \"e\"}}, // 187.75\n\t\t\t\t\t\t// a is scaled up and b is taken from group 0 because it has the most isolation groups\n\t\t\t\t\t\t// and minimizes the difference in size between the two groups (93 for b vs 96 for g vs 103 for h)\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"a\", \"b\"}}, // 174.75\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"immediate scale up on new partition - multiple\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 303,\n\t\t\t\t\t\t\t\"b\": 303,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},                // 300\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"d\"}},                // 299\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"b\", \"e\", \"f\", \"g\", \"h\"}}, // 233\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{}, // If we had the space \"a\" and \"b\" would have scaled up on this cycle\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"d\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"b\", \"e\", \"f\", \"g\", \"h\"}},\n\t\t\t\t\t\t// adaptive scaler gave us another partition\n\t\t\t\t\t\t3: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"d\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"b\", \"e\", \"f\", \"g\", \"h\"}},\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale down - single partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 75,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"c\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"d\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale down - multiple partition\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}},      // 25\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"g\"}}, // 129\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"c\", \"f\"}}, // 132\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}},      // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}}, // 25\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"g\"}}, // 105\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}}, // 108\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}}, // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale down - multiple groups\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\t\t\"b\": 73,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"h\"}}, // ~49\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"g\"}}, // ~54\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}},      // 108\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}},      // 112\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"h\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"g\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"c\", \"f\"}},\n\t\t\t\t\t\t3: {IsolationGroups: []string{\"d\", \"e\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale down - single partition left\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\t\t\"b\": 73,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t\t\"e\": 15,\n\t\t\t\t\t\t\t\"f\": 10,\n\t\t\t\t\t\t\t\"g\": 5,\n\t\t\t\t\t\t\t\"h\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t\t\"e\": true,\n\t\t\t\t\t\t\t\"f\": true,\n\t\t\t\t\t\t\t\"g\": true,\n\t\t\t\t\t\t\t\"h\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"h\"}},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"scale up then down\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 101,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 199,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 74,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t\t\"c\": 98,\n\t\t\t\t\t\t\t\"d\": 97,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t\"c\": true,\n\t\t\t\t\t\t\t\"d\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"d\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\", \"c\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"poller added then removed\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 100,\n\t\t\t\t\t\t\t\"b\": 99,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"ensure minimum partitions\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 2,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\"}},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"increase minimum partitions\",\n\t\t\tgroupsPerPartition: 1,\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 2,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcallback: func(dynamicClient dynamicconfig.Client) {\n\t\t\t\t\t\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupsPerPartition, 2))\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:               \"decrease minimum partitions\",\n\t\t\tgroupsPerPartition: 2,\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 2,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"b\"}},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tcallback: func(dynamicClient dynamicconfig.Client) {\n\t\t\t\t\t\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupsPerPartition, 1))\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{},\n\t\t\t\t{\n\t\t\t\t\t// Has to go through a normal scale down operation\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new partition - increase partitionsPerGroup\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 2,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t2: {},\n\t\t\t\t\t},\n\t\t\t\t\t// adaptive scaler gave us a new partition and even though we don't think these groups need more\n\t\t\t\t\t// partitions, we need to ensure every partition is assigned groups\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"b\"}},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new partition - move groups\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 99,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t\t\"c\": 2,\n\t\t\t\t\t\t\t\"d\": 3,\n\t\t\t\t\t\t\t\"e\": 4,\n\t\t\t\t\t\t\t\"f\": 5,\n\t\t\t\t\t\t\t\"g\": 6,\n\t\t\t\t\t\t\t\"h\": 7,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t\t// Even though the others aren't polled, they're still active because they're assigned\n\t\t\t\t\t\t\t// to a partition and we haven't been missing pollers long enough to remove them\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},                     // 100\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\", \"e\", \"f\", \"g\", \"h\"}}, // 27\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},                     // 100\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\", \"e\", \"f\", \"g\", \"h\"}}, // 27\n\t\t\t\t\t\t2: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\"}},           // 100\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"c\", \"d\", \"e\", \"f\"}}, // 13\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"g\", \"h\"}},           // 14\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new partition - move groups from multiple partitions\",\n\t\t\tcycles: []*testCycle{\n\t\t\t\t{\n\t\t\t\t\tmetrics: &aggregatePartitionMetrics{\n\t\t\t\t\t\tqpsByIsolationGroup: map[string]float64{\n\t\t\t\t\t\t\t\"a\": 99,\n\t\t\t\t\t\t\t\"b\": 1,\n\t\t\t\t\t\t\t\"c\": 2,\n\t\t\t\t\t\t\t\"d\": 3,\n\t\t\t\t\t\t\t\"e\": 4,\n\t\t\t\t\t\t\t\"f\": 5,\n\t\t\t\t\t\t},\n\t\t\t\t\t\thasPollersByIsolationGroup: map[string]bool{\n\t\t\t\t\t\t\t\"a\": true,\n\t\t\t\t\t\t\t\"b\": true,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}}, // 103\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"d\", \"e\", \"f\"}}, // 12\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tpartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"a\", \"b\", \"c\"}}, // 103\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"d\", \"e\", \"f\"}}, // 12\n\t\t\t\t\t\t2: {},\n\t\t\t\t\t},\n\t\t\t\t\texpected: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: {IsolationGroups: []string{\"b\", \"c\"}}, // 3\n\t\t\t\t\t\t1: {IsolationGroups: []string{\"e\", \"f\"}}, // 9\n\t\t\t\t\t\t2: {IsolationGroups: []string{\"a\", \"d\"}}, // 102\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcycles := tc.cycles\n\t\t\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\t\t\ttcConfig := isolationConfig(t, dynamicClient, tc.groupsPerPartition)\n\t\t\tmockTime := clock.NewMockedTimeSource()\n\t\t\tbalancer := newIsolationBalancer(mockTime, metrics.NoopScope, tcConfig)\n\t\t\tvar prevMetrics *aggregatePartitionMetrics\n\t\t\tvar prevPartitions map[int]*types.TaskListPartition\n\t\t\tfor i, cycle := range cycles {\n\t\t\t\tmockTime.Advance(cycleInterval)\n\t\t\t\tcycleMetrics := cycle.metrics\n\t\t\t\tif cycleMetrics == nil {\n\t\t\t\t\tcycleMetrics = prevMetrics\n\t\t\t\t}\n\t\t\t\tcyclePartitions := cycle.partitions\n\t\t\t\tif cyclePartitions == nil {\n\t\t\t\t\tcyclePartitions = prevPartitions\n\t\t\t\t}\n\t\t\t\tif len(cyclePartitions) == 0 || cycleMetrics == nil {\n\t\t\t\t\tassert.Fail(t, \"invalid cycle configuration at cycle %d\", i)\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tif cycle.callback != nil {\n\t\t\t\t\tcycle.callback(dynamicClient)\n\t\t\t\t}\n\t\t\t\tactual, actualChanged := balancer.adjustWritePartitions(cycleMetrics, cyclePartitions)\n\t\t\t\tif cycle.expected != nil {\n\t\t\t\t\tassert.True(t, actualChanged, \"cycle %d - expected change\", i)\n\t\t\t\t\tassert.Equal(t, cycle.expected, actual, \"cycle %d - did not match expected\", i)\n\t\t\t\t} else {\n\t\t\t\t\tassert.False(t, actualChanged, \"cycle %d - expected no change\", i)\n\t\t\t\t\tassert.Equal(t, cyclePartitions, actual, \"cycle %d - was unexpectedly changed\", i)\n\t\t\t\t}\n\t\t\t\tprevMetrics = cycleMetrics\n\t\t\t\tprevPartitions = cyclePartitions\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc isolationConfig(t *testing.T, dynamicClient dynamicconfig.Client, groupsPerPartition int) *config.TaskListConfig {\n\tif groupsPerPartition == 0 {\n\t\tgroupsPerPartition = 2\n\t}\n\ttaskListID, err := NewIdentifier(\"test-domain-id\", \"test-task-list\", 0)\n\trequire.NoError(t, err)\n\tlogger := testlogger.New(t)\n\tcfg := newTaskListConfig(taskListID, config.NewConfig(dynamicconfig.NewCollection(dynamicClient, logger), \"test-host\", commonConfig.RPC{}, func() []string { return nil }), \"test-domain\")\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionUpscaleRPS, 200))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingPartitionDownscaleFactor, 0.75))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupsPerPartition, groupsPerPartition))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupUpscaleSustainedDuration, cycleInterval*4))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupDownscaleSustainedDuration, cycleInterval*4))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupHasPollersSustainedDuration, cycleInterval*4))\n\trequire.NoError(t, dynamicClient.UpdateValue(dynamicproperties.MatchingIsolationGroupNoPollersSustainedDuration, cycleInterval*4))\n\n\treturn cfg\n}\n"
  },
  {
    "path": "service/matching/tasklist/matcher.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/ctxutils\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\n// taskMatcherImpl matches a task producer with a task consumer\n// Producers are usually rpc calls from history or taskReader\n// that drains backlog from db. Consumers are the task list pollers\ntype taskMatcherImpl struct {\n\tlog log.Logger\n\t// synchronous task channel to match producer/consumer for any isolation group\n\t// tasks having no isolation requirement are added to this channel\n\t// and pollers from all isolation groups read from this channel\n\ttaskC chan *InternalTask\n\t// synchronos task channels to match producer/consumer for a certain isolation group\n\t// the key is the name of the isolation group\n\tisolatedTaskC map[string]chan *InternalTask\n\t// synchronous task channel to match query task - the reason to have\n\t// separate channel for this is because there are cases when consumers\n\t// are interested in queryTasks but not others. Example is when domain is\n\t// not active in a cluster\n\tqueryTaskC chan *InternalTask\n\t// ratelimiter that limits the rate at which tasks can be dispatched to consumers\n\tlimiter quotas.Limiter\n\n\tfwdr   Forwarder\n\tscope  metrics.Scope // domain metric scope\n\tconfig *config.TaskListConfig\n\n\tcancelCtx  context.Context // used to cancel long polling\n\tcancelFunc context.CancelFunc\n\tcancelLock sync.Mutex\n\n\ttasklist     *Identifier\n\ttasklistKind types.TaskListKind\n\n\tnumReadPartitionsFn func(*config.TaskListConfig) int\n}\n\n// ErrTasklistThrottled implies a tasklist was throttled\nvar ErrTasklistThrottled = errors.New(\"tasklist limit exceeded\")\n\n// newTaskMatcher returns a task matcher instance. The returned instance can be\n// used by task producers and consumers to find a match. Both sync matches and non-sync\n// matches should use this implementation\nfunc newTaskMatcher(\n\tconfig *config.TaskListConfig,\n\tfwdr Forwarder,\n\tscope metrics.Scope,\n\tisolationGroups []string,\n\tlog log.Logger,\n\ttasklist *Identifier,\n\ttasklistKind types.TaskListKind,\n\tlimiter quotas.Limiter,\n) TaskMatcher {\n\tisolatedTaskC := make(map[string]chan *InternalTask)\n\tfor _, g := range isolationGroups {\n\t\tisolatedTaskC[g] = make(chan *InternalTask)\n\t}\n\n\tcancelCtx, cancelFunc := context.WithCancel(context.Background())\n\n\tmatcher := &taskMatcherImpl{\n\t\tlog:           log,\n\t\tscope:         scope,\n\t\tfwdr:          fwdr,\n\t\ttaskC:         make(chan *InternalTask),\n\t\tisolatedTaskC: isolatedTaskC,\n\t\tqueryTaskC:    make(chan *InternalTask),\n\t\tconfig:        config,\n\t\ttasklist:      tasklist,\n\t\ttasklistKind:  tasklistKind,\n\t\tlimiter:       limiter,\n\t\tcancelCtx:     cancelCtx,\n\t\tcancelFunc:    cancelFunc,\n\t}\n\n\treturn matcher\n}\n\n// DisconnectBlockedPollers gradually disconnects pollers which are blocked on long polling\nfunc (tm *taskMatcherImpl) DisconnectBlockedPollers() {\n\ttm.cancelFunc()\n}\n\n// Offer offers a task to a potential consumer (poller)\n// If the task is successfully matched with a consumer, this\n// method will return true and no error. If the task is matched\n// but consumer returned error, then this method will return\n// true and error message. This method should not be used for query\n// task. This method should ONLY be used for sync match.\n//\n// When a local poller is not available and forwarding to a parent\n// task list partition is possible, this method will attempt forwarding\n// to the parent partition.\n//\n// Cases when this method will block:\n//\n// Ratelimit:\n// When a ratelimit token is not available, this method might block\n// waiting for a token until the provided context timeout. Rate limits are\n// not enforced for forwarded tasks from child partition.\n//\n// Forwarded tasks that originated from db backlog:\n// When this method is called with a task that is forwarded from a\n// remote partition and if (1) this task list is root (2) task\n// was from db backlog - this method will block until context timeout\n// trying to match with a poller. The caller is expected to set the\n// correct context timeout.\n//\n// returns error when:\n//   - ratelimit is exceeded (does not apply to query task)\n//   - context deadline is exceeded\n//   - task is matched and consumer returns error in response channel\nfunc (tm *taskMatcherImpl) Offer(ctx context.Context, task *InternalTask) (bool, error) {\n\tstartT := time.Now()\n\tif !task.IsForwarded() {\n\t\terr := tm.ratelimit(ctx)\n\t\tif err != nil {\n\t\t\ttm.scope.IncCounter(metrics.SyncThrottlePerTaskListCounter)\n\t\t\treturn false, err\n\t\t}\n\t}\n\te := event.E{\n\t\tTaskListName: tm.tasklist.GetName(),\n\t\tTaskListType: tm.tasklist.GetType(),\n\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t}\n\tlocalWaitTime := tm.config.LocalTaskWaitTime()\n\tif localWaitTime > 0 {\n\t\tchildCtx, cancel := context.WithTimeout(ctx, localWaitTime)\n\t\tselect {\n\t\tcase tm.getTaskC(task) <- task: // poller picked up the task\n\t\t\tcancel()\n\t\t\tif task.ResponseC != nil {\n\t\t\t\t// if there is a response channel, block until resp is received\n\t\t\t\t// and return error if the response contains error\n\t\t\t\terr := <-task.ResponseC\n\t\t\t\ttm.scope.RecordTimer(metrics.SyncMatchLocalPollLatencyPerTaskList, time.Since(startT))\n\t\t\t\tif err == nil {\n\t\t\t\t\te.EventName = \"Offer task due to local wait\"\n\t\t\t\t\te.Payload = map[string]any{\n\t\t\t\t\t\t\"TaskIsForwarded\": task.IsForwarded(),\n\t\t\t\t\t}\n\t\t\t\t\tevent.Log(e)\n\t\t\t\t}\n\t\t\t\treturn true, err\n\t\t\t}\n\t\t\treturn false, nil\n\t\tcase <-childCtx.Done():\n\t\t\tcancel()\n\t\t}\n\t}\n\tselect {\n\tcase tm.getTaskC(task) <- task: // poller picked up the task\n\t\tif task.ResponseC != nil {\n\t\t\t// if there is a response channel, block until resp is received\n\t\t\t// and return error if the response contains error\n\t\t\terr := <-task.ResponseC\n\t\t\ttm.scope.RecordTimer(metrics.SyncMatchLocalPollLatencyPerTaskList, time.Since(startT))\n\t\t\treturn true, err\n\t\t}\n\t\treturn false, nil\n\tdefault:\n\t\t// no poller waiting for tasks, try forwarding this task to the\n\t\t// root partition if possible\n\t\tselect {\n\t\tcase token := <-tm.fwdrAddReqTokenC():\n\t\t\te.EventName = \"Attempting to Forward Task\"\n\t\t\tevent.Log(e)\n\t\t\terr := tm.fwdr.ForwardTask(ctx, task)\n\t\t\ttoken.release()\n\t\t\tif err == nil {\n\t\t\t\t// task was remotely sync matched on the parent partition\n\t\t\t\ttm.scope.RecordTimer(metrics.SyncMatchForwardPollLatencyPerTaskList, time.Since(startT))\n\t\t\t\treturn true, nil\n\t\t\t}\n\t\t\tif errors.Is(err, ErrForwarderSlowDown) {\n\t\t\t\ttm.scope.IncCounter(metrics.SyncMatchForwardTaskThrottleErrorPerTasklist)\n\t\t\t}\n\t\tdefault:\n\t\t\tif !tm.isForwardingAllowed() && // we are the root partition and forwarding is not possible\n\t\t\t\ttask.source == types.TaskSourceDbBacklog && // task was from backlog (stored in db)\n\t\t\t\ttask.IsForwarded() { // task came from a child partition\n\t\t\t\t// a forwarded backlog task from a child partition, block trying\n\t\t\t\t// to match with a poller until ctx timeout\n\t\t\t\treturn tm.OfferOrTimeout(ctx, startT, task)\n\t\t\t}\n\t\t}\n\n\t\treturn false, nil\n\t}\n}\n\n// OfferOrTimeout offers a task to a poller and blocks until a poller picks up the task or context timeouts\nfunc (tm *taskMatcherImpl) OfferOrTimeout(ctx context.Context, startT time.Time, task *InternalTask) (bool, error) {\n\tif !task.IsForwarded() {\n\t\terr := tm.ratelimit(ctx)\n\t\tif err != nil {\n\t\t\t// If context was canceled/timed out, return without error (consistent with original behavior)\n\t\t\tif err == ctx.Err() {\n\t\t\t\treturn false, nil\n\t\t\t}\n\t\t\ttm.scope.IncCounter(metrics.SyncThrottlePerTaskListCounter)\n\t\t\treturn false, err\n\t\t}\n\t}\n\tselect {\n\tcase tm.getTaskC(task) <- task: // poller picked up the task\n\t\tif task.ResponseC != nil {\n\t\t\tselect {\n\t\t\tcase err := <-task.ResponseC:\n\t\t\t\ttm.scope.RecordTimer(metrics.SyncMatchLocalPollLatencyPerTaskList, time.Since(startT))\n\t\t\t\treturn true, err\n\t\t\tcase <-ctx.Done():\n\t\t\t\treturn false, nil\n\t\t\t}\n\t\t}\n\t\treturn task.ActivityTaskDispatchInfo != nil, nil\n\tcase <-ctx.Done():\n\t\treturn false, nil\n\t}\n}\n\n// OfferQuery will either match task to local poller or will forward query task.\n// Local match is always attempted before forwarding is attempted. If local match occurs\n// response and error are both nil, if forwarding occurs then response or error is returned.\nfunc (tm *taskMatcherImpl) OfferQuery(ctx context.Context, task *InternalTask) (*types.MatchingQueryWorkflowResponse, error) {\n\tselect {\n\tcase tm.queryTaskC <- task:\n\t\t<-task.ResponseC\n\t\treturn nil, nil\n\tdefault:\n\t}\n\n\tfwdrTokenC := tm.fwdrAddReqTokenC()\n\n\tfor {\n\t\tselect {\n\t\tcase tm.queryTaskC <- task:\n\t\t\t<-task.ResponseC\n\t\t\treturn nil, nil\n\t\tcase token := <-fwdrTokenC:\n\t\t\tresp, err := tm.fwdr.ForwardQueryTask(ctx, task)\n\t\t\ttoken.release()\n\t\t\tif err == nil {\n\t\t\t\treturn resp, nil\n\t\t\t}\n\t\t\tif err == ErrForwarderSlowDown {\n\t\t\t\t// if we are rate limited, try only local match for the\n\t\t\t\t// remainder of the context timeout left\n\t\t\t\tfwdrTokenC = noopForwarderTokenC\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treturn nil, err\n\t\tcase <-ctx.Done():\n\t\t\treturn nil, ctx.Err()\n\t\t}\n\t}\n}\n\n// MustOffer blocks until a consumer is found to handle this task\n// Returns error only when context is canceled, expired or the ratelimit is set to zero (allow nothing)\nfunc (tm *taskMatcherImpl) MustOffer(ctx context.Context, task *InternalTask) error {\n\te := event.E{\n\t\tTaskListName: tm.tasklist.GetName(),\n\t\tTaskListType: tm.tasklist.GetType(),\n\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\tTaskInfo:     task.Info(),\n\t}\n\tif err := tm.ratelimit(ctx); err != nil {\n\t\te.EventName = \"Throttled While Dispatching\"\n\t\tevent.Log(e)\n\t\treturn fmt.Errorf(\"rate limit error dispatching: %w\", err)\n\t}\n\n\tstartT := time.Now()\n\t// attempt a match with local poller first. When that\n\t// doesn't succeed, try both local match and remote match\n\ttaskC := tm.getTaskC(task)\n\tlocalWaitTime := tm.config.LocalTaskWaitTime()\n\tchildCtx, cancel := context.WithTimeout(ctx, localWaitTime)\n\tselect {\n\tcase taskC <- task: // poller picked up the task\n\t\tcancel()\n\t\ttm.scope.IncCounter(metrics.AsyncMatchLocalPollCounterPerTaskList)\n\t\ttm.scope.RecordTimer(metrics.AsyncMatchLocalPollLatencyPerTaskList, time.Since(startT))\n\t\te.EventName = \"Dispatched to Local Poller\"\n\t\tevent.Log(e)\n\t\treturn nil\n\tcase <-ctx.Done():\n\t\tcancel()\n\t\te.EventName = \"Context Done While Dispatching to Local Poller\"\n\t\tevent.Log(e)\n\t\treturn fmt.Errorf(\"context done when trying to forward local task: %w\", ctx.Err())\n\tcase <-childCtx.Done():\n\t\tcancel()\n\t}\n\n\tattempt := 0\nforLoop:\n\tfor {\n\t\tif err := ctx.Err(); err != nil {\n\t\t\te.EventName = \"Context Done While Dispatching to Local or Forwarding\"\n\t\t\tevent.Log(e)\n\t\t\treturn fmt.Errorf(\"failed to offer task: %w\", ctx.Err())\n\t\t}\n\t\tselect {\n\t\tcase taskC <- task: // poller picked up the task\n\t\t\te.EventName = \"Dispatched to Local Poller\"\n\t\t\tevent.Log(e)\n\t\t\ttm.scope.IncCounter(metrics.AsyncMatchLocalPollCounterPerTaskList)\n\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchLocalPollAttemptPerTaskList, time.Duration(attempt))\n\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchLocalPollLatencyPerTaskList, time.Since(startT))\n\t\t\treturn nil\n\t\tcase token := <-tm.fwdrAddReqTokenC():\n\t\t\te.EventName = \"Attempting to Forward Task\"\n\t\t\tevent.Log(e)\n\t\t\tchildCtx, cancel := context.WithTimeout(ctx, time.Second*2)\n\t\t\terr := tm.fwdr.ForwardTask(childCtx, task)\n\t\t\ttoken.release()\n\t\t\tif err != nil {\n\t\t\t\tif errors.Is(err, ErrForwarderSlowDown) {\n\t\t\t\t\ttm.scope.IncCounter(metrics.AsyncMatchForwardTaskThrottleErrorPerTasklist)\n\t\t\t\t}\n\t\t\t\te.EventName = \"Task Forwarding Failed\"\n\t\t\t\te.Payload = map[string]any{\"error\": err.Error()}\n\t\t\t\tevent.Log(e)\n\t\t\t\te.Payload = nil\n\t\t\t\ttm.log.Debug(\"failed to forward task\",\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t\ttag.TaskID(task.Event.TaskID),\n\t\t\t\t)\n\t\t\t\t// forwarder returns error only when the call is rate limited. To\n\t\t\t\t// avoid a busy loop on such rate limiting events, we only attempt to make\n\t\t\t\t// the next forwarded call after this childCtx expires. Till then, we block\n\t\t\t\t// hoping for a local poller match\n\t\t\t\tselect {\n\t\t\t\tcase taskC <- task: // poller picked up the task\n\t\t\t\t\te.EventName = \"Dispatched to Local Poller (after failed forward)\"\n\t\t\t\t\tevent.Log(e)\n\t\t\t\t\tcancel()\n\t\t\t\t\ttm.scope.IncCounter(metrics.AsyncMatchLocalPollAfterForwardFailedCounterPerTaskList)\n\t\t\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchLocalPollAfterForwardFailedAttemptPerTaskList, time.Duration(attempt))\n\t\t\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchLocalPollAfterForwardFailedLatencyPerTaskList, time.Since(startT))\n\t\t\t\t\treturn nil\n\t\t\t\tcase <-childCtx.Done():\n\t\t\t\t\tattempt++\n\t\t\t\t\tcancel()\n\t\t\t\t\tcontinue forLoop\n\t\t\t\t}\n\t\t\t}\n\t\t\tcancel()\n\n\t\t\te.EventName = \"Task Forwarded\"\n\t\t\tevent.Log(e)\n\t\t\ttm.scope.IncCounter(metrics.AsyncMatchForwardPollCounterPerTaskList)\n\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchForwardPollAttemptPerTaskList, time.Duration(attempt))\n\t\t\ttm.scope.RecordTimer(metrics.AsyncMatchForwardPollLatencyPerTaskList, time.Since(startT))\n\n\t\t\t// at this point, we forwarded the task to a parent partition which\n\t\t\t// in turn dispatched the task to a poller. Make sure we delete the\n\t\t\t// task from the database\n\t\t\ttask.Finish(nil)\n\t\t\treturn nil\n\t\tcase <-ctx.Done():\n\t\t\te.EventName = \"Context Done While Dispatching to Local or Forwarding\"\n\t\t\tevent.Log(e)\n\t\t\treturn fmt.Errorf(\"failed to offer task: %w\", ctx.Err())\n\t\t}\n\t}\n}\n\n// Poll blocks until a task is found or context deadline is exceeded\n// On success, the returned task could be a query task or a regular task\n// Returns ErrNoTasks when context deadline is exceeded\n// Returns ErrMatcherClosed when matching is closed\nfunc (tm *taskMatcherImpl) Poll(ctx context.Context, isolationGroup string) (*InternalTask, error) {\n\tstartT := time.Now()\n\tisolatedTaskC, ok := tm.isolatedTaskC[isolationGroup]\n\tif !ok && isolationGroup != \"\" {\n\t\t// fallback to default isolation group instead of making poller crash if the isolation group is invalid\n\t\tisolatedTaskC = tm.taskC\n\t\ttm.scope.IncCounter(metrics.PollerInvalidIsolationGroupCounter)\n\t}\n\n\t// we want cancellation of taskMatcher to be treated as cancellation of client context\n\t// original context (ctx) won't be affected\n\tctxWithCancelPropagation, stopFn := ctxutils.WithPropagatedContextCancel(ctx, tm.cancelCtx)\n\tdefer stopFn()\n\n\tvar task *InternalTask\n\tvar err error\n\tdefer func() {\n\t\tif task != nil {\n\t\t\ttask.AutoConfigHint = &types.AutoConfigHint{\n\t\t\t\tEnableAutoConfig:   tm.config.EnableClientAutoConfig(),\n\t\t\t\tPollerWaitTimeInMs: time.Since(startT).Milliseconds(),\n\t\t\t}\n\t\t}\n\t}()\n\n\t// try local match first without blocking until context timeout\n\tif task, err = tm.pollNonBlocking(ctxWithCancelPropagation, isolatedTaskC, tm.taskC, tm.queryTaskC); err == nil {\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchLatencyPerTaskList, time.Since(startT))\n\t\treturn task, nil\n\t}\n\t// there is no local poller available to pickup this task. Now block waiting\n\t// either for a local poller or a forwarding token to be available. When a\n\t// forwarding token becomes available, send this poll to a parent partition\n\ttm.log.Debug(\"falling back to non-local polling\",\n\t\ttag.IsolationGroup(isolationGroup),\n\t\ttag.Dynamic(\"isolated channel\", len(isolatedTaskC)),\n\t\ttag.Dynamic(\"fallback channel\", len(tm.taskC)),\n\t)\n\tevent.Log(event.E{\n\t\tTaskListName: tm.tasklist.GetName(),\n\t\tTaskListType: tm.tasklist.GetType(),\n\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\tEventName:    \"Matcher Falling Back to Non-Local Polling\",\n\t})\n\ttask, err = tm.pollOrForward(ctxWithCancelPropagation, startT, isolationGroup, isolatedTaskC, tm.taskC, tm.queryTaskC)\n\treturn task, err\n}\n\n// PollForQuery blocks until a *query* task is found or context deadline is exceeded\n// Returns ErrNoTasks when context deadline is exceeded\n// Returns ErrMatcherClosed when matching is closed\nfunc (tm *taskMatcherImpl) PollForQuery(ctx context.Context) (*InternalTask, error) {\n\tstartT := time.Now()\n\t// try local match first without blocking until context timeout\n\tif task, err := tm.pollNonBlocking(ctx, nil, nil, tm.queryTaskC); err == nil {\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchLatencyPerTaskList, time.Since(startT))\n\t\treturn task, nil\n\t}\n\n\tctxWithCancelPropagation, stopFn := ctxutils.WithPropagatedContextCancel(ctx, tm.cancelCtx)\n\tdefer stopFn()\n\n\t// there is no local poller available to pickup this task. Now block waiting\n\t// either for a local poller or a forwarding token to be available. When a\n\t// forwarding token becomes available, send this poll to a parent partition\n\treturn tm.pollOrForward(ctxWithCancelPropagation, startT, \"\", nil, nil, tm.queryTaskC)\n}\n\nfunc (tm *taskMatcherImpl) RefreshCancelContext() {\n\ttm.cancelLock.Lock()\n\tdefer tm.cancelLock.Unlock()\n\ttm.cancelCtx, tm.cancelFunc = context.WithCancel(context.Background())\n}\n\nfunc (tm *taskMatcherImpl) pollOrForward(\n\tctx context.Context,\n\tstartT time.Time,\n\tisolationGroup string,\n\tisolatedTaskC <-chan *InternalTask,\n\ttaskC <-chan *InternalTask,\n\tqueryTaskC <-chan *InternalTask,\n) (*InternalTask, error) {\n\tselect {\n\tcase task := <-isolatedTaskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchLatencyPerTaskList, time.Since(startT))\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task (pollOrForward)\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": true,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-taskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchLatencyPerTaskList, time.Since(startT))\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task (pollOrForward)\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": false,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-queryTaskC:\n\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\treturn task, nil\n\tcase <-ctx.Done():\n\t\ttm.scope.IncCounter(metrics.PollTimeoutPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tEventName:    \"Poll Timeout\",\n\t\t})\n\t\treturn nil, ErrNoTasks\n\tcase token := <-tm.fwdrPollReqTokenC():\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tEventName:    \"Attempting to Forward Poll\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"IsolationGroup\": isolationGroup,\n\t\t\t},\n\t\t})\n\t\tif task, err := tm.fwdr.ForwardPoll(ctx); err == nil {\n\t\t\ttoken.release()\n\t\t\ttm.scope.RecordTimer(metrics.PollForwardMatchLatencyPerTaskList, time.Since(startT))\n\t\t\tevent.Log(event.E{\n\t\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\t\tEventName:    \"Forwarded Poll returned task\",\n\t\t\t})\n\t\t\treturn task, nil\n\t\t}\n\t\ttoken.release()\n\t\treturn tm.poll(ctx, startT, isolatedTaskC, taskC, queryTaskC)\n\t}\n}\n\nfunc (tm *taskMatcherImpl) poll(\n\tctx context.Context,\n\tstartT time.Time,\n\tisolatedTaskC <-chan *InternalTask,\n\ttaskC <-chan *InternalTask,\n\tqueryTaskC <-chan *InternalTask,\n) (*InternalTask, error) {\n\tselect {\n\tcase task := <-isolatedTaskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchAfterForwardFailedLatencyPerTaskList, time.Since(startT))\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task (poll)\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": true,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-taskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.RecordTimer(metrics.PollLocalMatchAfterForwardFailedLatencyPerTaskList, time.Since(startT))\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task (poll)\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": false,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-queryTaskC:\n\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\treturn task, nil\n\tcase <-ctx.Done():\n\t\ttm.scope.IncCounter(metrics.PollTimeoutPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tEventName:    \"Poll Timeout\",\n\t\t})\n\t\treturn nil, ErrNoTasks\n\t}\n}\n\nfunc (tm *taskMatcherImpl) pollLocalWait(\n\tctx context.Context,\n\tisolatedTaskC <-chan *InternalTask,\n\ttaskC <-chan *InternalTask,\n\tqueryTaskC <-chan *InternalTask,\n) (*InternalTask, error) {\n\tselect {\n\tcase task := <-isolatedTaskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task Nonblocking\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": true,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-taskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task Nonblocking\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": false,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-queryTaskC:\n\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\treturn task, nil\n\tcase <-ctx.Done():\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tEventName:    \"Matcher Found No Tasks Nonblocking\",\n\t\t})\n\t\treturn nil, ErrNoTasks\n\t}\n}\n\nfunc (tm *taskMatcherImpl) pollNonBlocking(\n\tctx context.Context,\n\tisolatedTaskC <-chan *InternalTask,\n\ttaskC <-chan *InternalTask,\n\tqueryTaskC <-chan *InternalTask,\n) (*InternalTask, error) {\n\twaitTime := tm.config.LocalPollWaitTime()\n\tif waitTime > 0 {\n\t\tchildCtx, cancel := context.WithTimeout(ctx, waitTime)\n\t\tdefer cancel()\n\t\treturn tm.pollLocalWait(childCtx, isolatedTaskC, taskC, queryTaskC)\n\t}\n\tselect {\n\tcase task := <-isolatedTaskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task Nonblocking\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": true,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-taskC:\n\t\tif task.ResponseC != nil {\n\t\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\t}\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tTaskInfo:     task.Info(),\n\t\t\tEventName:    \"Matched Task Nonblocking\",\n\t\t\tPayload: map[string]any{\n\t\t\t\t\"TaskIsForwarded\":   task.IsForwarded(),\n\t\t\t\t\"SyncMatched\":       task.ResponseC != nil,\n\t\t\t\t\"FromIsolatedTaskC\": false,\n\t\t\t\t\"IsolationGroup\":    task.isolationGroup,\n\t\t\t},\n\t\t})\n\t\treturn task, nil\n\tcase task := <-queryTaskC:\n\t\ttm.scope.IncCounter(metrics.PollSuccessWithSyncPerTaskListCounter)\n\t\ttm.scope.IncCounter(metrics.PollSuccessPerTaskListCounter)\n\t\treturn task, nil\n\tdefault:\n\t\tevent.Log(event.E{\n\t\t\tTaskListName: tm.tasklist.GetName(),\n\t\t\tTaskListType: tm.tasklist.GetType(),\n\t\t\tTaskListKind: tm.tasklistKind.Ptr(),\n\t\t\tEventName:    \"Matcher Found No Tasks Nonblocking\",\n\t\t})\n\t\treturn nil, ErrNoTasks\n\t}\n}\n\nfunc (tm *taskMatcherImpl) fwdrPollReqTokenC() <-chan *ForwarderReqToken {\n\tif tm.fwdr == nil {\n\t\treturn noopForwarderTokenC\n\t}\n\treturn tm.fwdr.PollReqTokenC()\n}\n\nfunc (tm *taskMatcherImpl) fwdrAddReqTokenC() <-chan *ForwarderReqToken {\n\tif tm.fwdr == nil {\n\t\treturn noopForwarderTokenC\n\t}\n\treturn tm.fwdr.AddReqTokenC()\n}\n\nfunc (tm *taskMatcherImpl) ratelimit(ctx context.Context) error {\n\terr := tm.limiter.Wait(ctx)\n\tif errors.Is(err, clock.ErrCannotWait) {\n\t\t// \"err != ctx.Err()\" may also be correct, as that would mean \"gave up due to context\".\n\t\t//\n\t\t// in this branch: either the request would wait longer than ctx's timeout,\n\t\t// or the limiter's config does not allow any operations at all (burst 0).\n\t\t// in either case, this is returned immediately.\n\t\treturn ErrTasklistThrottled\n\t}\n\treturn err // nil if success, non-nil if canceled\n}\n\nfunc (tm *taskMatcherImpl) isForwardingAllowed() bool {\n\treturn tm.fwdr != nil\n}\n\nfunc (tm *taskMatcherImpl) getTaskC(task *InternalTask) chan<- *InternalTask {\n\ttaskC := tm.taskC\n\tif isolatedTaskC, ok := tm.isolatedTaskC[task.isolationGroup]; ok && task.isolationGroup != \"\" {\n\t\ttaskC = isolatedTaskC\n\t}\n\treturn taskC\n}\n"
  },
  {
    "path": "service/matching/tasklist/matcher_test.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nconst testDomainName = \"test-domain\"\n\ntype MatcherTestSuite struct {\n\tsuite.Suite\n\tcontroller      *gomock.Controller\n\tclient          *matching.MockClient\n\tfwdr            Forwarder\n\tcfg             *config.TaskListConfig\n\ttaskList        *Identifier\n\tmatcher         *taskMatcherImpl // matcher for child partition\n\trootMatcher     TaskMatcher      // matcher for parent partition\n\tisolationGroups []string\n}\n\nfunc TestMatcherSuite(t *testing.T) {\n\tsuite.Run(t, new(MatcherTestSuite))\n}\n\nfunc (t *MatcherTestSuite) SetupTest() {\n\tt.controller = gomock.NewController(t.T())\n\tt.client = matching.NewMockClient(t.controller)\n\tcfg := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, func() []string { return nil })\n\tcfg.TaskDispatchRPSTTL = 0\n\tt.taskList = NewTestTaskListID(t.T(), uuid.New(), constants.ReservedTaskListPrefix+\"tl0/1\", persistence.TaskListTypeDecision)\n\ttlCfg := newTaskListConfig(t.taskList, cfg, testDomainName)\n\n\ttlCfg.ForwarderConfig = config.ForwarderConfig{\n\t\tForwarderMaxOutstandingPolls: func() int { return 1 },\n\t\tForwarderMaxOutstandingTasks: func() int { return 1 },\n\t\tForwarderMaxRatePerSecond:    func() int { return 2 },\n\t\tForwarderMaxChildrenPerNode:  func() int { return 20 },\n\t}\n\tt.cfg = tlCfg\n\tt.isolationGroups = []string{\"dca1\", \"dca2\"}\n\tt.fwdr = newForwarder(&t.cfg.ForwarderConfig, t.taskList, types.TaskListKindNormal, t.client, metrics.NoopScope)\n\tt.matcher = newTaskMatcher(tlCfg, t.fwdr, metrics.NoopScope, []string{\"dca1\", \"dca2\"}, log.NewNoop(), t.taskList, types.TaskListKindNormal, clock.NewRatelimiter(rate.Limit(100), 100)).(*taskMatcherImpl)\n\n\trootTaskList := NewTestTaskListID(t.T(), t.taskList.GetDomainID(), t.taskList.Parent(20), persistence.TaskListTypeDecision)\n\trootTasklistCfg := newTaskListConfig(rootTaskList, cfg, testDomainName)\n\tt.rootMatcher = newTaskMatcher(rootTasklistCfg, nil, metrics.NoopScope, []string{\"dca1\", \"dca2\"}, log.NewNoop(), t.taskList, types.TaskListKindNormal, clock.NewRatelimiter(rate.Limit(100), 100)).(*taskMatcherImpl)\n}\n\nfunc (t *MatcherTestSuite) TearDownTest() {\n\tt.controller.Finish()\n}\n\nfunc (t *MatcherTestSuite) TestLocalSyncMatch() {\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatch, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n\tt.True(syncMatch)\n}\n\nfunc (t *MatcherTestSuite) TestIsolationLocalSyncMatch() {\n\tconst isolationGroup = \"dca1\"\n\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, isolationGroup)\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"dca1\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatch, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n\tt.True(syncMatch)\n}\n\nfunc (t *MatcherTestSuite) TestRemoteSyncMatch() {\n\tt.testRemoteSyncMatch(types.TaskSourceHistory, \"\")\n}\n\nfunc (t *MatcherTestSuite) TestIsolationRemoteSyncMatch() {\n\tt.testRemoteSyncMatch(types.TaskSourceHistory, \"dca1\")\n}\n\nfunc (t *MatcherTestSuite) TestRemoteSyncMatchBlocking() {\n\tt.testRemoteSyncMatch(types.TaskSourceDbBacklog, \"\")\n}\n\nfunc (t *MatcherTestSuite) TestIsolationRemoteSyncMatchBlocking() {\n\tt.testRemoteSyncMatch(types.TaskSourceDbBacklog, \"dca1\")\n}\n\nfunc (t *MatcherTestSuite) testRemoteSyncMatch(taskSource types.TaskSource, isolationGroup string) {\n\tpollSigC := make(chan struct{})\n\n\tbgctx, bgcancel := context.WithTimeout(context.Background(), time.Second)\n\tgo func() {\n\t\t<-pollSigC\n\t\tif taskSource == types.TaskSourceDbBacklog {\n\t\t\t// when task is from dbBacklog, sync match SHOULD block\n\t\t\t// so lets delay polling by a bit to verify that\n\t\t\ttime.Sleep(time.Millisecond * 10)\n\t\t}\n\t\ttask, err := t.matcher.Poll(bgctx, isolationGroup)\n\t\tbgcancel()\n\t\tif err == nil && !task.IsStarted() {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t}()\n\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\ttask, err := t.rootMatcher.Poll(arg0, isolationGroup)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\ttask.Finish(nil)\n\t\t\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\tWorkflowExecution: task.WorkflowExecution(),\n\t\t\t}, nil\n\t\t},\n\t).AnyTimes()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, taskSource, \"\", true, nil, isolationGroup)\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\n\tvar err error\n\tvar remoteSyncMatch bool\n\tvar req *types.AddDecisionTaskRequest\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t\ttask.forwardedFrom = req.GetForwardedFrom()\n\t\t\tclose(pollSigC)\n\t\t\tif taskSource != types.TaskSourceDbBacklog {\n\t\t\t\t// when task is not from backlog, wait a bit for poller to arrive first\n\t\t\t\t// when task is from backlog, offer blocks, so we don't need to do this\n\t\t\t\ttime.Sleep(time.Millisecond)\n\t\t\t\t// Sleep for local poll wait time so that poller will be forwarded to root partition\n\t\t\t\ttime.Sleep(t.matcher.config.LocalPollWaitTime())\n\t\t\t}\n\t\t\tremoteSyncMatch, err = t.rootMatcher.Offer(ctx, task)\n\t\t},\n\t).Return(&types.AddDecisionTaskResponse{}, nil)\n\n\t_, err0 := t.matcher.Offer(ctx, task)\n\tt.NoError(err0)\n\tcancel()\n\t<-bgctx.Done() // wait for async work to finish\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.True(remoteSyncMatch)\n\tt.Equal(t.taskList.name, req.GetForwardedFrom())\n\tt.Equal(t.taskList.Parent(20), req.GetTaskList().GetName())\n}\n\nfunc (t *MatcherTestSuite) TestSyncMatchFailure() {\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\n\tvar req *types.AddDecisionTaskRequest\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t},\n\t).Return(nil, &types.ServiceBusyError{})\n\n\tsyncMatch, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.False(syncMatch)\n}\n\nfunc (t *MatcherTestSuite) TestRateLimitHandling() {\n\tscope := mocks.Scope{}\n\tscope.On(\"IncCounter\", metrics.SyncMatchForwardTaskThrottleErrorPerTasklist)\n\tscope.On(\"RecordTimer\", mock.Anything, mock.Anything)\n\tt.matcher.scope = &scope\n\tfor i := 0; i < 5; i++ {\n\t\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Return(&types.AddDecisionTaskResponse{}, nil).AnyTimes()\n\t\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t\t_, err := t.matcher.Offer(ctx, task)\n\t\tcancel()\n\t\tassert.NoError(t.T(), err)\n\t}\n}\n\nfunc (t *MatcherTestSuite) TestIsolationSyncMatchFailure() {\n\tconst isolationGroup = \"dca2\"\n\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, isolationGroup)\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"dca1\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatch, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n\tt.False(syncMatch)\n}\n\nfunc (t *MatcherTestSuite) TestQueryLocalSyncMatch() {\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.PollForQuery(ctx)\n\t\tif err == nil && task.IsQuery() {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tresp, err := t.matcher.OfferQuery(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n\tt.Nil(resp)\n}\n\nfunc (t *MatcherTestSuite) TestQueryRemoteSyncMatch() {\n\tready, wait := ensureAsyncAfterReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.PollForQuery(ctx)\n\t\tif err == nil && task.IsQuery() {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tvar remotePollResp *types.MatchingPollForDecisionTaskResponse\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\ttask, err := t.rootMatcher.PollForQuery(arg0)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t} else if task.IsQuery() {\n\t\t\t\ttask.Finish(nil)\n\t\t\t\tres := &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\t\tQuery: &types.WorkflowQuery{},\n\t\t\t\t}\n\t\t\t\tremotePollResp = res\n\t\t\t\treturn res, nil\n\t\t\t}\n\t\t\treturn nil, nil\n\t\t},\n\t).AnyTimes()\n\n\ttask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\n\tvar req *types.MatchingQueryWorkflowRequest\n\tt.client.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingQueryWorkflowRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t\ttask.forwardedFrom = req.GetForwardedFrom()\n\t\t\tready()\n\t\t\tt.rootMatcher.OfferQuery(ctx, task)\n\t\t},\n\t).Return(&types.MatchingQueryWorkflowResponse{QueryResult: []byte(\"answer\")}, nil)\n\n\tresult, err := t.matcher.OfferQuery(ctx, task)\n\tcancel()\n\twait()\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.NotNil(result)\n\tt.NotNil(remotePollResp.Query)\n\tt.Equal(\"answer\", string(result.QueryResult))\n\tt.Equal(t.taskList.name, req.GetForwardedFrom())\n\tt.Equal(t.taskList.Parent(20), req.GetTaskList().GetName())\n}\n\nfunc (t *MatcherTestSuite) TestQueryRemoteSyncMatchError() {\n\t<-t.fwdr.PollReqTokenC()\n\n\tmatched := false\n\tready, wait := ensureAsyncAfterReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.PollForQuery(ctx)\n\t\tif err == nil && task.IsQuery() {\n\t\t\tmatched = true\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\n\tvar req *types.MatchingQueryWorkflowRequest\n\tt.client.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingQueryWorkflowRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t\tready()\n\t\t},\n\t).Return(nil, &types.ServiceBusyError{})\n\n\tresult, err := t.matcher.OfferQuery(ctx, task)\n\tcancel()\n\twait()\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.Nil(result)\n\tt.True(matched)\n}\n\nfunc (t *MatcherTestSuite) TestMustOfferLocalMatch() {\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\terr := t.matcher.MustOffer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n}\n\nfunc (t *MatcherTestSuite) TestIsolationMustOfferLocalMatch() {\n\t// force disable remote forwarding\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"dca1\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"dca1\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\terr := t.matcher.MustOffer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n}\n\nfunc (t *MatcherTestSuite) TestMustOfferRemoteMatch() {\n\tpollSigC := make(chan struct{})\n\tforwardPollSigC := make(chan struct{})\n\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\tclose(forwardPollSigC)\n\t\t\t<-pollSigC\n\t\t\ttime.Sleep(time.Millisecond * 500) // delay poll to verify that offer blocks on parent\n\t\t\ttask, err := t.rootMatcher.Poll(arg0, \"\")\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\ttask.Finish(nil)\n\t\t\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\tWorkflowExecution: task.WorkflowExecution(),\n\t\t\t}, nil\n\t\t},\n\t).AnyTimes()\n\n\ttaskCompleted := false\n\tcompletionFunc := func(*persistence.TaskInfo, error) {\n\t\ttaskCompleted = true\n\t}\n\n\ttask := newInternalTask(t.newTaskInfo(), completionFunc, types.TaskSourceDbBacklog, \"\", false, nil, \"\")\n\tctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)\n\n\tvar err error\n\tvar remoteSyncMatch bool\n\tvar req *types.AddDecisionTaskRequest\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{}).Times(1)\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t\ttask := newInternalTask(task.Event.TaskInfo, nil, types.TaskSourceDbBacklog, req.GetForwardedFrom(), true, nil, \"\")\n\t\t\tclose(pollSigC)\n\t\t\tremoteSyncMatch, err = t.rootMatcher.Offer(ctx, task)\n\t\t},\n\t).Return(&types.AddDecisionTaskResponse{}, nil)\n\n\t// Poll needs to happen before MustOffer, or else it goes into the non-blocking path.\n\tvar pollResultMu sync.Mutex\n\tvar polledTask *InternalTask\n\tvar pollErr error\n\twait := ensureAsyncReady(time.Second*3, func(ctx context.Context) {\n\t\tpollResultMu.Lock()\n\t\tdefer pollResultMu.Unlock()\n\t\tpolledTask, pollErr = t.matcher.Poll(ctx, \"\")\n\t})\n\t<-forwardPollSigC\n\n\tt.NoError(t.matcher.MustOffer(ctx, task))\n\tcancel()\n\twait()\n\tpollResultMu.Lock()\n\tdefer pollResultMu.Unlock()\n\tt.NoError(pollErr)\n\tt.NotNil(polledTask)\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.True(remoteSyncMatch)\n\tt.True(taskCompleted)\n\tt.Equal(t.taskList.name, req.GetForwardedFrom())\n\tt.Equal(t.taskList.Parent(20), req.GetTaskList().GetName())\n}\n\nfunc (t *MatcherTestSuite) TestMustOfferRemoteRateLimit() {\n\tscope := mocks.Scope{}\n\tscope.On(\"IncCounter\", metrics.AsyncMatchForwardTaskThrottleErrorPerTasklist)\n\tscope.On(\"RecordTimer\", mock.Anything, mock.Anything)\n\tt.matcher.scope = &scope\n\tcompletionFunc := func(*persistence.TaskInfo, error) {}\n\tfor i := 0; i < 5; i++ {\n\t\tscope.On(\"IncCounter\", metrics.AsyncMatchForwardPollCounterPerTaskList)\n\t\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Return(&types.AddDecisionTaskResponse{}, nil)\n\t\ttask := newInternalTask(t.newTaskInfo(), completionFunc, types.TaskSourceDbBacklog, \"\", false, nil, \"\")\n\t\tctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)\n\t\tt.NoError(t.matcher.MustOffer(ctx, task))\n\t\tcancel()\n\t}\n}\n\nfunc (t *MatcherTestSuite) TestIsolationMustOfferRemoteMatch() {\n\tpollSigC := make(chan struct{})\n\tforwardPollSigC := make(chan struct{})\n\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\tclose(forwardPollSigC)\n\t\t\t<-pollSigC\n\t\t\ttime.Sleep(time.Millisecond * 500) // delay poll to verify that offer blocks on parent\n\t\t\ttask, err := t.rootMatcher.Poll(arg0, \"dca1\")\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\ttask.Finish(nil)\n\t\t\treturn &types.MatchingPollForDecisionTaskResponse{\n\t\t\t\tWorkflowExecution: task.WorkflowExecution(),\n\t\t\t}, nil\n\t\t},\n\t).AnyTimes()\n\n\ttaskCompleted := false\n\tcompletionFunc := func(*persistence.TaskInfo, error) {\n\t\ttaskCompleted = true\n\t}\n\n\ttask := newInternalTask(t.newTaskInfo(), completionFunc, types.TaskSourceDbBacklog, \"\", false, nil, \"dca1\")\n\tctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)\n\n\tvar err error\n\tvar remoteSyncMatch bool\n\tvar req *types.AddDecisionTaskRequest\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Return(nil, &types.ServiceBusyError{}).Times(1)\n\tt.client.EXPECT().AddDecisionTask(gomock.Any(), gomock.Any()).Do(\n\t\tfunc(arg0 context.Context, arg1 *types.AddDecisionTaskRequest, option ...yarpc.CallOption) {\n\t\t\treq = arg1\n\t\t\ttask := newInternalTask(task.Event.TaskInfo, nil, types.TaskSourceDbBacklog, req.GetForwardedFrom(), true, nil, \"dca1\")\n\t\t\tclose(pollSigC)\n\t\t\tremoteSyncMatch, err = t.rootMatcher.Offer(arg0, task)\n\t\t},\n\t).Return(&types.AddDecisionTaskResponse{}, nil)\n\n\t// Poll needs to happen before MustOffer, or else it goes into the non-blocking path.\n\tvar pollResultMu sync.Mutex\n\tvar polledTask *InternalTask\n\tvar pollErr error\n\twait := ensureAsyncReady(time.Second*3, func(ctx context.Context) {\n\t\tpollResultMu.Lock()\n\t\tdefer pollResultMu.Unlock()\n\t\tpolledTask, pollErr = t.matcher.Poll(ctx, \"dca1\")\n\t})\n\t<-forwardPollSigC\n\n\tt.NoError(t.matcher.MustOffer(ctx, task))\n\tcancel()\n\twait()\n\tpollResultMu.Lock()\n\tdefer pollResultMu.Unlock()\n\tt.NoError(pollErr)\n\tt.NotNil(polledTask)\n\tt.NotNil(req)\n\tt.NoError(err)\n\tt.True(remoteSyncMatch)\n\tt.True(taskCompleted)\n\tt.Equal(t.taskList.name, req.GetForwardedFrom())\n\tt.Equal(t.taskList.Parent(20), req.GetTaskList().GetName())\n}\n\nfunc (t *MatcherTestSuite) TestPollersDisconnectedAfterDisconnectBlockedPollers() {\n\tdefer goleak.VerifyNone(t.T())\n\tt.disableRemoteForwarding()\n\tt.matcher.DisconnectBlockedPollers()\n\n\tlongPollingCtx, cancel := context.WithTimeout(context.Background(), 1*time.Second)\n\tdefer cancel()\n\n\ttask, err := t.matcher.Poll(longPollingCtx, \"\")\n\tt.ErrorIs(err, ErrNoTasks, \"closed matcher should result in no tasks\")\n\tt.Nil(task)\n\tt.NoError(longPollingCtx.Err(), \"the parent context was not cancelled, the child context was cancelled\")\n}\n\nfunc (t *MatcherTestSuite) TestPollersConnectedAfterDisconnectBlockedPollersSetCancelContext() {\n\tdefer goleak.VerifyNone(t.T())\n\tt.disableRemoteForwarding()\n\tt.matcher.DisconnectBlockedPollers()\n\tt.matcher.RefreshCancelContext()\n\n\tlongPollingCtx, cancel := context.WithTimeout(context.Background(), 1*time.Second)\n\tdefer cancel()\n\n\ttask, err := t.matcher.Poll(longPollingCtx, \"\")\n\tt.ErrorIs(err, ErrNoTasks, \"no tasks to be matched to poller\")\n\tt.Nil(task)\n\tt.ErrorIs(longPollingCtx.Err(), context.DeadlineExceeded, \"the child context wasn't cancelled, the parent context timed out\")\n}\n\nfunc (t *MatcherTestSuite) TestRemotePoll() {\n\tpollToken := <-t.fwdr.PollReqTokenC()\n\n\tvar req *types.MatchingPollForDecisionTaskRequest\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\treq = arg1\n\t\t\treturn nil, nil\n\t\t},\n\t)\n\n\tready, wait := ensureAsyncAfterReady(0, func(_ context.Context) {\n\t\tpollToken.release()\n\t})\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tready()\n\ttask, err := t.matcher.Poll(ctx, \"\")\n\tcancel()\n\twait()\n\tt.NoError(err)\n\tt.NotNil(req)\n\tt.NotNil(task)\n\tt.True(task.IsStarted())\n}\n\nfunc (t *MatcherTestSuite) TestRemotePollForQuery() {\n\tpollToken := <-t.fwdr.PollReqTokenC()\n\n\tvar req *types.MatchingPollForDecisionTaskRequest\n\tt.client.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(arg0 context.Context, arg1 *types.MatchingPollForDecisionTaskRequest, option ...yarpc.CallOption) (*types.MatchingPollForDecisionTaskResponse, error) {\n\t\t\treq = arg1\n\t\t\treturn nil, nil\n\t\t},\n\t)\n\n\tready, wait := ensureAsyncAfterReady(0, func(_ context.Context) {\n\t\tpollToken.release()\n\t})\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tdefer cancel()\n\tready()\n\ttask, err := t.matcher.PollForQuery(ctx)\n\twait()\n\tt.NoError(err)\n\tt.NotNil(req)\n\tt.NotNil(task)\n\tt.True(task.IsStarted())\n}\n\nfunc (t *MatcherTestSuite) TestIsolationPollFailure() {\n\tt.disableRemoteForwarding()\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\ttask, err := t.matcher.Poll(ctx, \"invalid-group\")\n\tcancel()\n\tt.Error(err)\n\tt.Nil(task)\n}\n\nfunc (t *MatcherTestSuite) TestOffer_RateLimited() {\n\tt.matcher.limiter = clock.NewRatelimiter(0, 0)\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tctx := context.Background()\n\n\tmatched, err := t.matcher.Offer(ctx, task)\n\n\tt.ErrorIs(err, ErrTasklistThrottled)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_RateLimited() {\n\tt.matcher.limiter = clock.NewRatelimiter(0, 0)\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, &types.ActivityTaskDispatchInfo{}, \"\")\n\n\tctx := context.Background()\n\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\n\tt.ErrorIs(err, ErrTasklistThrottled)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_ForwardedNotRateLimited() {\n\t// Forwarded tasks should not be rate limited\n\tt.matcher.limiter = clock.NewRatelimiter(0, 0)\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"forwarded-from\", false, &types.ActivityTaskDispatchInfo{}, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond)\n\tdefer cancel()\n\n\t// Should timeout instead of being rate limited since forwarded tasks bypass rate limiting\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\n\tt.NoError(err)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOffer_NoTimeoutSyncMatchedNoError() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.matcher.config.LocalTaskWaitTime = func() time.Duration { return 0 }\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatched, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\n\tt.NoError(err)\n\tt.True(syncMatched)\n}\n\nfunc (t *MatcherTestSuite) TestOffer_NoTimeoutSyncMatchedError() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.matcher.config.LocalTaskWaitTime = func() time.Duration { return 0 }\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\ttask.ResponseC <- errShutdown\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatched, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\n\tt.Error(err)\n\tt.True(syncMatched)\n}\n\nfunc (t *MatcherTestSuite) TestOffer_NoTimeoutAsyncMatchedNoError() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.matcher.config.LocalTaskWaitTime = func() time.Duration { return 0 }\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatched, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\n\tt.NoError(err)\n\tt.False(syncMatched)\n}\n\nfunc (t *MatcherTestSuite) TestOffer_AsyncMatchedNoError() {\n\tdefer goleak.VerifyNone(t.T())\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tsyncMatch, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\n\tt.NoError(err)\n\tt.False(syncMatch)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_SyncMatchTimedOut() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttime.Sleep(time.Millisecond * 100)\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\tdefer cancel()\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\twait()\n\n\tt.NoError(err)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_AsyncMatchNotMatched() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\tcancel()\n\twait()\n\n\tt.NoError(err)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_AsyncMatchMatched() {\n\tdefer goleak.VerifyNone(t.T())\n\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, &types.ActivityTaskDispatchInfo{}, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\tcancel()\n\twait()\n\n\tt.NoError(err)\n\tt.True(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferOrTimeout_TimedOut() {\n\tt.disableRemoteForwarding()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, &types.ActivityTaskDispatchInfo{}, \"\")\n\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\tcancel()\n\n\tmatched, err := t.matcher.OfferOrTimeout(ctx, time.Now(), task)\n\n\tt.NoError(err)\n\tt.False(matched)\n}\n\nfunc (t *MatcherTestSuite) TestOfferQuery_ForwardError() {\n\tctx := context.Background()\n\ttask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\tfn := func() <-chan *ForwarderReqToken {\n\t\tc := make(chan *ForwarderReqToken, 1)\n\t\tc <- &ForwarderReqToken{\n\t\t\tch: make(chan *ForwarderReqToken, 1),\n\t\t}\n\t\treturn c\n\t}\n\n\tmockForwarder.EXPECT().AddReqTokenC().Return(fn()).Times(1)\n\tmockForwarder.EXPECT().ForwardQueryTask(ctx, task).Return(nil, ErrNoParent).Times(1)\n\n\tretTask, err := t.matcher.OfferQuery(ctx, task)\n\n\tt.ErrorIs(err, ErrNoParent)\n\tt.Nil(retTask)\n}\n\nfunc (t *MatcherTestSuite) TestOfferQuery_ContextExpired() {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\tcancel()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\ttask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\n\tmockForwarder.EXPECT().AddReqTokenC().Times(1)\n\n\tretTask, err := t.matcher.OfferQuery(ctx, task)\n\n\tt.ErrorIs(err, context.Canceled)\n\tt.Nil(retTask)\n}\n\nfunc (t *MatcherTestSuite) TestMustOffer_ContextExpiredFirstAttempt() {\n\tctx, cancel := context.WithTimeout(context.Background(), t.matcher.config.LocalTaskWaitTime())\n\tdefer cancel()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\terr := t.matcher.MustOffer(ctx, task)\n\n\tt.ErrorIs(err, context.DeadlineExceeded)\n}\n\nfunc (t *MatcherTestSuite) TestMustOffer_ContextExpiredAfterFirstAttempt() {\n\tctx, cancel := context.WithTimeout(context.Background(), 2*t.matcher.config.LocalTaskWaitTime())\n\tdefer cancel()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\tmockForwarder.EXPECT().AddReqTokenC().Times(1)\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\terr := t.matcher.MustOffer(ctx, task)\n\n\tt.ErrorIs(err, context.DeadlineExceeded)\n}\n\nfunc (t *MatcherTestSuite) TestMustOffer_LocalMatchAfterChildCtxExpired() {\n\tctx := context.Background()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\tt.disableRemoteForwarding()\n\n\tmockForwarder.EXPECT().AddReqTokenC().AnyTimes()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tgo func() {\n\t\t// Waits for the child context to expire; by default, it is 10 ms.\n\t\t// Forces no forwarding and the context background will never expire, therefore guaranteeing that the poller\n\t\t// will pick up the task after the first attempt.\n\t\ttime.Sleep(200 * time.Millisecond)\n\t\tretTask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\tretTask.Finish(nil)\n\t\t}\n\t}()\n\n\terr := t.matcher.MustOffer(ctx, task)\n\n\tt.NoError(err)\n}\n\nfunc (t *MatcherTestSuite) TestMustOffer_LocalMatchAfterForwardError() {\n\tctx := context.Background()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\tforwardToken := &ForwarderReqToken{\n\t\tch: make(chan *ForwarderReqToken, 1),\n\t}\n\n\tfn := func() <-chan *ForwarderReqToken {\n\t\tc := make(chan *ForwarderReqToken, 1)\n\t\tc <- forwardToken\n\t\treturn c\n\t}\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tmockForwarder.EXPECT().AddReqTokenC().Return(fn()).Times(1)\n\tmockForwarder.EXPECT().ForwardTask(gomock.Any(), task).Return(ErrNoParent).Times(1)\n\n\tgo func() {\n\t\t<-forwardToken.ch\n\t\tretTask, err := t.matcher.Poll(ctx, \"\")\n\t\tif err == nil {\n\t\t\tretTask.Finish(nil)\n\t\t}\n\t}()\n\n\terr := t.matcher.MustOffer(ctx, task)\n\n\tt.NoError(err)\n}\n\nfunc (t *MatcherTestSuite) TestMustOffer_ContextExpiredAfterForwardError() {\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\tforwardToken := &ForwarderReqToken{\n\t\tch: make(chan *ForwarderReqToken, 1),\n\t}\n\n\t// Have 1 token ready and waiting\n\tc := make(chan *ForwarderReqToken, 1)\n\tc <- forwardToken\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", false, nil, \"\")\n\n\tmockForwarder.EXPECT().AddReqTokenC().Return(c).Times(1)\n\tmockForwarder.EXPECT().ForwardTask(gomock.Any(), task).Return(ErrNoParent).Times(1)\n\n\tgo func() {\n\t\t// Wait for the token to be returned\n\t\t<-forwardToken.ch\n\t\t// using cancel here to simulate the context being expired\n\t\tcancel()\n\t}()\n\n\terr := t.matcher.MustOffer(ctx, task)\n\tt.ErrorIs(err, context.Canceled)\n\tt.ErrorContains(err, \"failed to offer task:\")\n}\n\nfunc (t *MatcherTestSuite) Test_pollOrForward_PollIsolatedTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tisolationGroup := \"dca1\"\n\tisolatedTaskC := make(chan *InternalTask, 1)\n\n\tt.disableRemoteForwarding()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\t// Mock the fwdrPollReqTokenC method to return a controlled channel\n\tmockTokenC := make(chan *ForwarderReqToken)\n\tmockForwarder.EXPECT().PollReqTokenC().Return(mockTokenC).AnyTimes()\n\n\t// Test pollOrForward for isolated task - poll\n\tisolatedTask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, isolationGroup)\n\tisolatedTaskC <- isolatedTask\n\tretTask, err := t.matcher.pollOrForward(ctx, startT, isolationGroup, isolatedTaskC, nil, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(isolatedTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollOrForward_PollTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\ttaskC := make(chan *InternalTask, 1)\n\n\tt.disableRemoteForwarding()\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\t// Mock the fwdrPollReqTokenC method to return a controlled channel\n\tmockTokenC := make(chan *ForwarderReqToken)\n\tmockForwarder.EXPECT().PollReqTokenC().Return(mockTokenC).AnyTimes()\n\n\t// Test pollOrForward for regular task - poll\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\ttaskC <- task\n\tretTask, err := t.matcher.pollOrForward(ctx, startT, \"\", nil, taskC, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(task, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollOrForward_PollQueryTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tqueryTaskC := make(chan *InternalTask, 1)\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\t// Mock the fwdrPollReqTokenC method to return a controlled channel\n\tmockTokenC := make(chan *ForwarderReqToken)\n\tmockForwarder.EXPECT().PollReqTokenC().Return(mockTokenC).AnyTimes()\n\n\t// Test pollOrForward for query task - poll\n\tqueryTask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tqueryTaskC <- queryTask\n\tretTask, err := t.matcher.pollOrForward(ctx, startT, \"\", nil, nil, queryTaskC)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(queryTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollOrForward_ForwardTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tisolationGroup := \"dca1\"\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\t// Mock the fwdrPollReqTokenC method to return a controlled channel\n\tmockTokenC := make(chan *ForwarderReqToken, 1)\n\tforwardToken := &ForwarderReqToken{\n\t\tch: make(chan *ForwarderReqToken, 1),\n\t}\n\tmockTokenC <- forwardToken\n\tmockForwarder.EXPECT().PollReqTokenC().Return(mockTokenC).AnyTimes()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\tmockForwarder.EXPECT().ForwardPoll(ctx).Return(task, nil).Times(1)\n\n\tretTask, err := t.matcher.pollOrForward(ctx, startT, isolationGroup, nil, nil, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(task, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollOrForward_ForwardTaskThenPoll() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tisolationGroup := \"dca1\"\n\ttaskC := make(chan *InternalTask, 1)\n\n\tmockForwarder := NewMockForwarder(t.controller)\n\tt.matcher.fwdr = mockForwarder\n\n\t// Mock the fwdrPollReqTokenC method to return a controlled channel\n\tmockTokenC := make(chan *ForwarderReqToken, 1)\n\tforwardToken := &ForwarderReqToken{\n\t\tch: make(chan *ForwarderReqToken, 1),\n\t}\n\tmockTokenC <- forwardToken\n\tmockForwarder.EXPECT().PollReqTokenC().Return(mockTokenC).AnyTimes()\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\tmockForwarder.EXPECT().ForwardPoll(ctx).Return(nil, ErrNoParent).Times(1)\n\n\t// Add the task after the forwarderReqToken is released\n\t// It's not a race condition because the PollReqTokenC is mocked and does not use the isolatedCh\n\tgo func() {\n\t\tselect {\n\t\tcase <-forwardToken.ch:\n\t\t\ttaskC <- task\n\t\t}\n\t}()\n\n\tretTask, err := t.matcher.pollOrForward(ctx, startT, isolationGroup, nil, taskC, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(task, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_poll_IsolatedTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tisolationGroup := \"dca1\"\n\tisolatedTaskC := make(chan *InternalTask, 1)\n\n\tisolatedTask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, isolationGroup)\n\tisolatedTaskC <- isolatedTask\n\tretTask, err := t.matcher.poll(ctx, startT, isolatedTaskC, nil, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(isolatedTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_poll_Task() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\ttaskC := make(chan *InternalTask, 1)\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\ttaskC <- task\n\tretTask, err := t.matcher.poll(ctx, startT, nil, taskC, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(task, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_poll_QueryTask() {\n\tctx := context.Background()\n\tstartT := time.Now()\n\tqueryTaskC := make(chan *InternalTask, 1)\n\n\tqueryTask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tqueryTaskC <- queryTask\n\tretTask, err := t.matcher.poll(ctx, startT, nil, nil, queryTaskC)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(queryTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_poll_ContextExpired() {\n\tctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)\n\tcancel()\n\n\tstartT := time.Now()\n\n\tretTask, err := t.matcher.poll(ctx, startT, nil, nil, nil)\n\n\tt.ErrorIs(err, ErrNoTasks)\n\tt.Nil(retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollNonBlocking_IsolatedTask() {\n\tt.matcher.config.LocalPollWaitTime = func() time.Duration { return 0 }\n\n\tctx := context.Background()\n\tisolationGroup := \"dca1\"\n\tisolatedTaskC := make(chan *InternalTask, 1)\n\n\tisolatedTask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, isolationGroup)\n\tisolatedTaskC <- isolatedTask\n\tretTask, err := t.matcher.pollNonBlocking(ctx, isolatedTaskC, nil, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(isolatedTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollNonBlocking_Task() {\n\tt.matcher.config.LocalPollWaitTime = func() time.Duration { return 0 }\n\n\tctx := context.Background()\n\ttaskC := make(chan *InternalTask, 1)\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\ttaskC <- task\n\tretTask, err := t.matcher.pollNonBlocking(ctx, nil, taskC, nil)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(task, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollNonBlocking_QueryTask() {\n\tt.matcher.config.LocalPollWaitTime = func() time.Duration { return 0 }\n\n\tctx := context.Background()\n\tqueryTaskC := make(chan *InternalTask, 1)\n\n\tqueryTask := newInternalQueryTask(uuid.New(), &types.MatchingQueryWorkflowRequest{})\n\tqueryTaskC <- queryTask\n\tretTask, err := t.matcher.pollNonBlocking(ctx, nil, nil, queryTaskC)\n\tt.NoError(err)\n\tt.NotNil(retTask)\n\tt.Equal(queryTask, retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_pollNonBlocking_NoTasks() {\n\tt.matcher.config.LocalPollWaitTime = func() time.Duration { return 0 }\n\n\tctx := context.Background()\n\n\tretTask, err := t.matcher.pollNonBlocking(ctx, nil, nil, nil)\n\n\tt.ErrorIs(err, ErrNoTasks)\n\tt.Nil(retTask)\n}\n\nfunc (t *MatcherTestSuite) Test_fwdrPollReqTokenC() {\n\tt.matcher.fwdr = nil\n\tt.Equal(noopForwarderTokenC, t.matcher.fwdrPollReqTokenC())\n}\n\nfunc (t *MatcherTestSuite) disableRemoteForwarding() {\n\t// force disable remote forwarding\n\t<-t.fwdr.AddReqTokenC()\n\t<-t.fwdr.PollReqTokenC()\n}\n\nfunc (t *MatcherTestSuite) newDomainCache() cache.DomainCache {\n\tdomainName := testDomainName\n\tdc := cache.NewMockDomainCache(t.controller)\n\tdc.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\treturn dc\n}\n\nfunc (t *MatcherTestSuite) newTaskInfo() *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:                      uuid.New(),\n\t\tWorkflowID:                    uuid.New(),\n\t\tRunID:                         uuid.New(),\n\t\tTaskID:                        rand.Int63(),\n\t\tScheduleID:                    rand.Int63(),\n\t\tScheduleToStartTimeoutSeconds: rand.Int31(),\n\t}\n}\n\nfunc TestRatelimitBehavior(t *testing.T) {\n\t// NOT t.Parallel() to avoid noise from cpu-heavy tests\n\n\tconst (\n\t\tgranularity = 100 * time.Millisecond\n\t\tprecision   = float64(granularity / 10) // for elapsed-time delta checks\n\t)\n\n\t// code prior to ~june 2024, kept around until new behavior is thoroughly validated,\n\t// largely to make regression tests easier to build if we encounter problems\n\torig := func(ctx context.Context, limiter *rate.Limiter) (*rate.Reservation, error) {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn nil, ctx.Err()\n\t\tdefault:\n\t\t}\n\n\t\tdeadline, ok := ctx.Deadline()\n\t\tif !ok {\n\t\t\tif err := limiter.Wait(ctx); err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\treturn nil, nil\n\t\t}\n\n\t\trsv := limiter.Reserve()\n\t\t// If we have to wait too long for reservation, give up and return\n\t\tif !rsv.OK() || rsv.Delay() > time.Until(deadline) {\n\t\t\tif rsv.OK() { // if we were indeed given a reservation, return it before we bail out\n\t\t\t\trsv.Cancel()\n\t\t\t}\n\t\t\treturn nil, ErrTasklistThrottled\n\t\t}\n\n\t\ttime.Sleep(rsv.Delay())\n\t\treturn rsv, nil\n\t}\n\ttests := map[string]struct {\n\t\trun         func(allowable func(ctx context.Context) (*rate.Reservation, error)) error\n\t\terr         error\n\t\tduration    time.Duration\n\t\tbeginTokens int\n\t\tendTokens   int\n\t}{\n\t\t\"no deadline returns immediately if available\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\t_, err := allowable(context.Background())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\tduration:    0,\n\t\t\tbeginTokens: 1,\n\t\t},\n\t\t\"no deadline waits until available\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\t_, err := allowable(context.Background())\n\t\t\t\treturn err\n\t\t\t},\n\t\t\tduration: granularity,\n\t\t},\n\t\t\"canceling no deadline while waiting returns ctx err\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tdefer cancel()\n\t\t\t\tgo func() {\n\t\t\t\t\ttime.Sleep(granularity / 2)\n\t\t\t\t\tcancel()\n\t\t\t\t}()\n\t\t\t\t_, err := allowable(ctx)\n\t\t\t\treturn err\n\t\t\t},\n\t\t\terr:      context.Canceled,\n\t\t\tduration: granularity / 2,\n\t\t},\n\t\t\"returns immediately if canceled\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tcancel()\n\t\t\t\t_, err := allowable(ctx)\n\t\t\t\treturn err\n\t\t\t},\n\t\t\terr:         context.Canceled,\n\t\t\tduration:    0,\n\t\t\tbeginTokens: 1,\n\t\t\tendTokens:   1,\n\t\t},\n\t\t\"sufficient deadline returns immediately if available\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), granularity*2)\n\t\t\t\tdefer cancel()\n\t\t\t\t_, err := allowable(ctx)\n\t\t\t\treturn err\n\t\t\t},\n\t\t\tduration:    0,\n\t\t\tbeginTokens: 1,\n\t\t},\n\t\t\"sufficient deadline waits\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), granularity*2)\n\t\t\t\tdefer cancel()\n\t\t\t\t_, err := allowable(ctx)\n\t\t\t\treturn err\n\t\t\t},\n\t\t\tduration: granularity,\n\t\t},\n\t\t\"insufficient deadline stops immediately\": {\n\t\t\trun: func(allowable func(ctx context.Context) (*rate.Reservation, error)) error {\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), granularity/2)\n\t\t\t\tdefer cancel()\n\t\t\t\t_, err := allowable(ctx)\n\t\t\t\treturn err\n\t\t\t},\n\t\t\terr:         ErrTasklistThrottled,\n\t\t\tduration:    0,\n\t\t\tbeginTokens: 0,\n\t\t},\n\t}\n\tfor name, test := range tests {\n\t\ttest := test // closure copy\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tcheck := func(t *testing.T, allow func() bool, allowable func(ctx context.Context) (*rate.Reservation, error)) {\n\t\t\t\tfor allow() {\n\t\t\t\t\t// drain tokens to start\n\t\t\t\t}\n\t\t\t\tif test.beginTokens > 0 {\n\t\t\t\t\ttime.Sleep(time.Duration(test.beginTokens) * granularity)\n\t\t\t\t}\n\t\t\t\tstart := time.Now()\n\t\t\t\terr := test.run(allowable)\n\t\t\t\tdur := time.Since(start).Round(time.Millisecond)\n\n\t\t\t\tif test.err != nil {\n\t\t\t\t\tassert.ErrorIs(t, err, test.err, \"wrong err returned, got %v\", err)\n\t\t\t\t} else {\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t}\n\t\t\t\tassert.InDeltaf(t, test.duration, dur, precision, \"duration should be within %v of %v, was %v\", time.Duration(precision), test.duration, dur)\n\t\t\t\tavailableTokens := 0\n\t\t\t\tfor allow() {\n\t\t\t\t\tavailableTokens++\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, test.endTokens, availableTokens, \"should have %v tokens available after the test runs\", test.endTokens)\n\n\t\t\t}\n\t\t\tt.Run(\"orig\", func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tperSecond := time.Second / granularity\n\t\t\t\tlimiter := rate.NewLimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\t\tcheck(t, limiter.Allow, func(ctx context.Context) (*rate.Reservation, error) {\n\t\t\t\t\treturn orig(ctx, limiter)\n\t\t\t\t})\n\t\t\t})\n\t\t\tt.Run(\"current\", func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tperSecond := float64(time.Second / granularity)\n\t\t\t\tlimiter := clock.NewRatelimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\t\tcheck(t, limiter.Allow, func(ctx context.Context) (*rate.Reservation, error) {\n\t\t\t\t\treturn nil, (&taskMatcherImpl{limiter: limiter}).ratelimit(ctx)\n\t\t\t\t})\n\t\t\t})\n\t\t})\n\t}\n\n\tt.Run(\"known misleading behavior\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\trun := func(t *testing.T, limit *rate.Limiter) time.Duration {\n\t\t\t// more than enough time to wait\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), granularity*2)\n\t\t\tdefer cancel()\n\n\t\t\tstart := time.Now()\n\t\t\tt.Logf(\"tokens before wait: %0.5f\", limit.Tokens())\n\t\t\tres, err := orig(ctx, limit)\n\t\t\tt.Logf(\"tokens after wait: %0.5f\", limit.Tokens())\n\t\t\telapsed := time.Since(start).Round(time.Millisecond)\n\n\t\t\trequire.NoError(t, err, \"should have succeeded\")\n\n\t\t\ttokensBefore := limit.Tokens()\n\t\t\tt.Logf(\"tokens before cancel: %0.5f\", tokensBefore)\n\t\t\tres.Cancel()\n\t\t\ttokensAfter := limit.Tokens()\n\t\t\tt.Logf(\"tokens after cancel: %0.5f\", tokensAfter)\n\n\t\t\tassert.Lessf(t, math.Abs(tokensBefore-tokensAfter), 0.1, \"canceling a delay-passed reservation does not return any tokens.  had %0.5f, cancel(), now have %0.5f\", tokensBefore, tokensAfter)\n\t\t\tassert.Lessf(t, tokensAfter, 0.1, \"near zero tokens should remain\")\n\n\t\t\treturn elapsed // varies per test\n\t\t}\n\t\tt.Run(\"orig returned tokens are un-cancel-able if available immediately\", func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tperSecond := time.Second / granularity\n\t\t\tlimit := rate.NewLimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\tfor limit.Allow() {\n\t\t\t\t// drain tokens to start\n\t\t\t}\n\n\t\t\ttime.Sleep(granularity) // restore one full token\n\t\t\telapsed := run(t, limit)\n\t\t\trequire.Lessf(t, elapsed, granularity/10, \"should have returned basically immediately\")\n\t\t})\n\t\tt.Run(\"orig returned tokens are un-cancel-able if it had to wait\", func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tperSecond := time.Second / granularity\n\t\t\tlimit := rate.NewLimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\tfor limit.Allow() {\n\t\t\t\t// drain tokens to start\n\t\t\t}\n\t\t\ttime.Sleep(granularity / 2) // restore only half of one token so it waits\n\t\t\telapsed := run(t, limit)\n\t\t\trequire.InDeltaf(t, elapsed, granularity/2, precision, \"should have waited %v for the remaining half token\", granularity/2)\n\t\t})\n\t})\n\n\tt.Run(\"known different behavior\", func(t *testing.T) {\n\t\tt.Parallel()\n\t\tt.Run(\"canceling while waiting with a deadline\", func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tcancelWhileWaiting := func(cb func(context.Context) error) (time.Duration, error) {\n\t\t\t\t// sufficient deadline, but canceled after half a token recovers\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), granularity*2)\n\t\t\t\tdefer cancel()\n\t\t\t\tgo func() {\n\t\t\t\t\ttime.Sleep(granularity / 2)\n\t\t\t\t\tcancel()\n\t\t\t\t}()\n\n\t\t\t\tstart := time.Now()\n\t\t\t\terr := cb(ctx)\n\t\t\t\telapsed := time.Since(start)\n\t\t\t\treturn elapsed, err\n\t\t\t}\n\t\t\tt.Run(\"orig does not stop\", func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tperSecond := time.Second / granularity\n\t\t\t\tlimit := rate.NewLimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\t\tfor limit.Allow() {\n\t\t\t\t\t// drain tokens to start\n\t\t\t\t}\n\n\t\t\t\telapsed, err := cancelWhileWaiting(func(ctx context.Context) error {\n\t\t\t\t\t_, err := orig(ctx, limit)\n\t\t\t\t\treturn err\n\t\t\t\t})\n\n\t\t\t\tassert.NoError(t, err, \"continues waiting and returns successfully despite being canceled\")\n\t\t\t\tassert.InDeltaf(t, granularity, elapsed, precision, \"waits until a full token would recover\")\n\t\t\t\tassert.False(t, limit.Allow(), \"consumes the token that was recovered\")\n\t\t\t\ttime.Sleep(granularity / 2)\n\t\t\t\tassert.False(t, limit.Allow(), \"did not have 0.5 tokens available after waiting\") // should be 0.5 now though\n\t\t\t})\n\t\t\tt.Run(\"new code stops quickly and returns unused tokens\", func(t *testing.T) {\n\t\t\t\tt.Parallel()\n\t\t\t\tperSecond := float64(time.Second / granularity)\n\t\t\t\tlimit := clock.NewRatelimiter(rate.Limit(perSecond), int(perSecond))\n\t\t\t\tfor limit.Allow() {\n\t\t\t\t\t// drain tokens to start\n\t\t\t\t}\n\n\t\t\t\telapsed, err := cancelWhileWaiting(func(ctx context.Context) error {\n\t\t\t\t\treturn (&taskMatcherImpl{limiter: limit}).ratelimit(ctx)\n\t\t\t\t})\n\n\t\t\t\tassert.ErrorIs(t, err, context.Canceled, \"gives up and returns context error\")\n\t\t\t\tassert.InDeltaf(t, granularity/2, elapsed, precision, \"waits only until canceled\")\n\t\t\t\tassert.False(t, limit.Allow(), \"does not yet have a full token, only 0.5\")\n\t\t\t\ttime.Sleep(granularity / 2)\n\t\t\t\t// note this is false in the original code\n\t\t\t\tassert.True(t, limit.Allow(), \"0.5 -> 1 tokens recovered shows 0.5 were available after waiting\")\n\t\t\t})\n\t\t})\n\t})\n}\n\n// Try to ensure a blocking callback in a goroutine is not running until the thing immediately\n// after `ready()` has blocked, so tests can ensure that the callback contents happen last.\n//\n// Try to delay calling `ready()` until *immediately* before the blocking call for best results.\n//\n// This is a best-effort technique, as there is no way to reliably synchronize this kind of thing\n// without exposing internal latches or having a more sophisticated locking library than Go offers.\n// In case of flakiness, increase the time.Sleep and hope for the best.\n//\n// Note that adding fmt.Println() calls touches synchronization code (for I/O), so it may change behavior.\nfunc ensureAsyncAfterReady(ctxTimeout time.Duration, cb func(ctx context.Context)) (ready func(), wait func()) {\n\tctx, cancel := context.WithTimeout(context.Background(), ctxTimeout)\n\tready = func() {\n\t\tgo func() {\n\t\t\tdefer cancel()\n\n\t\t\t// since `go func()` is non-blocking, the ready()-ing goroutine should generally continue,\n\t\t\t// and read whatever blocking point is relevant before this goroutine runs.\n\t\t\t// in many cases this sleep is unnecessary (especially with -cpu=1), but it does help.\n\t\t\ttime.Sleep(1 * time.Millisecond)\n\n\t\t\tcb(ctx)\n\n\t\t}()\n\t}\n\twait = func() {\n\t\t<-ctx.Done()\n\t}\n\treturn ready, wait\n}\n\n// Try to ensure a blocking callback is actively blocked in a goroutine before returning, so tests can\n// ensure that the callback contents happen first.\n// Do NOT access shared variables in the callback without mutex, as it may cause data races.\n//\n// This is a best-effort technique, as there is no way to reliably synchronize this kind of thing\n// without exposing internal latches or having a more sophisticated locking library than Go offers.\n// In case of flakiness, increase the time.Sleep and hope for the best.\n//\n// Note that adding fmt.Println() calls touches synchronization code (for I/O), so it may change behavior.\nfunc ensureAsyncReady(ctxTimeout time.Duration, cb func(ctx context.Context)) (wait func()) {\n\trunning := make(chan struct{})\n\tclosed := make(chan struct{})\n\tctx, cancel := context.WithTimeout(context.Background(), ctxTimeout)\n\tgo func() {\n\t\t// Defers are stacked. The last one added is the first one to run.\n\t\t// We want to cancel the context which will make the callback return because it's a Poll,\n\t\t// and then close the closed channel.\n\t\t// This way the returned wait function will block until the callback has returned.\n\t\tdefer close(closed)\n\t\tdefer cancel()\n\n\t\tclose(running)\n\t\tcb(ctx)\n\t}()\n\t<-running // ensure the goroutine is alive\n\n\t// `close(running)` is non-blocking, so it should generally begin polling before yielding control to other goroutines,\n\t// but there is still a race to reach whatever blocking sync point exists between the code being tested.\n\t// In many cases this sleep is completely unnecessary (especially with -cpu=1), but it does help.\n\ttime.Sleep(1 * time.Millisecond)\n\n\treturn func() {\n\t\t<-closed\n\t}\n}\n\nfunc (t *MatcherTestSuite) Test_LocalSyncMatch_AutoConfigHint() {\n\tt.disableRemoteForwarding()\n\n\twait := ensureAsyncReady(time.Second, func(ctx context.Context) {\n\t\ttask, err := t.matcher.Poll(ctx, \"\")\n\t\tt.NotNil(task.AutoConfigHint)\n\t\tt.Equal(false, task.AutoConfigHint.EnableAutoConfig) // disabled by default\n\t\tif err == nil {\n\t\t\ttask.Finish(nil)\n\t\t}\n\t})\n\n\ttask := newInternalTask(t.newTaskInfo(), nil, types.TaskSourceHistory, \"\", true, nil, \"\")\n\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t_, err := t.matcher.Offer(ctx, task)\n\tcancel()\n\twait()\n\tt.NoError(err)\n}\n"
  },
  {
    "path": "service/matching/tasklist/shard_processor.go",
    "content": "package tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\ntype ShardProcessorParams struct {\n\tShardID           string\n\tTaskListsRegistry TaskListRegistry\n\tReportTTL         time.Duration\n\tTimeSource        clock.TimeSource\n}\n\ntype shardProcessorImpl struct {\n\tshardID           string\n\ttaskListsRegistry TaskListRegistry\n\tStatus            atomic.Int32\n\treportLock        sync.RWMutex\n\tshardReport       executorclient.ShardReport\n\treportTime        time.Time\n\treportTTL         time.Duration\n\ttimeSource        clock.TimeSource\n}\n\nfunc NewShardProcessor(params ShardProcessorParams) (ShardProcessor, error) {\n\terr := validateSPParams(params)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tshardprocessor := &shardProcessorImpl{\n\t\tshardID:           params.ShardID,\n\t\ttaskListsRegistry: params.TaskListsRegistry,\n\t\tshardReport:       executorclient.ShardReport{},\n\t\treportTime:        params.TimeSource.Now(),\n\t\treportTTL:         params.ReportTTL,\n\t\ttimeSource:        params.TimeSource,\n\t}\n\tshardprocessor.SetShardStatus(types.ShardStatusREADY)\n\tshardprocessor.shardReport = executorclient.ShardReport{\n\t\tShardLoad: 0,\n\t\tStatus:    types.ShardStatusREADY,\n\t}\n\treturn shardprocessor, nil\n}\n\n// Start is now not doing anything since the shard lifecycle management is still handled by the logic in matching.\n// In the future the executor client will start the shard processor when a shard is assigned.\n// Ideally we want to have a tasklist mapping to a shard process, but in order to do that we need a major refactoring\n// of tasklists partitions and reloading processes\nfunc (sp *shardProcessorImpl) Start(ctx context.Context) error {\n\treturn nil\n}\n\n// Stop is stopping the tasklist when a shard is not assigned to this executor anymore.\nfunc (sp *shardProcessorImpl) Stop() {\n\ttoShutDown := sp.taskListsRegistry.ManagersByTaskListName(sp.shardID)\n\tfor _, tlMgr := range toShutDown {\n\t\ttlMgr.Stop()\n\t}\n}\n\nfunc (sp *shardProcessorImpl) GetShardReport() executorclient.ShardReport {\n\tsp.reportLock.Lock()\n\tdefer sp.reportLock.Unlock()\n\tload := sp.shardReport.ShardLoad\n\n\tif sp.reportTime.Add(sp.reportTTL).Before(sp.timeSource.Now()) {\n\t\tload = sp.getShardLoad()\n\t}\n\tsp.shardReport = executorclient.ShardReport{\n\t\tShardLoad: load,\n\t\tStatus:    types.ShardStatus(sp.Status.Load()),\n\t}\n\treturn sp.shardReport\n}\n\nfunc (sp *shardProcessorImpl) SetShardStatus(status types.ShardStatus) {\n\tsp.Status.Store(int32(status))\n}\n\nfunc (sp *shardProcessorImpl) getShardLoad() float64 {\n\tvar load float64\n\n\t// We assign a shard only based on the task list name\n\t// so task lists of different task type (decisions/activities), of different kind (normal, sticky, ephemeral) or partitions\n\t// will be assigned all to the same matching instance (executor)\n\t// we need to sum the rps for each of the tasklist to calculate the load.\n\tfor _, tlMgr := range sp.taskListsRegistry.ManagersByTaskListName(sp.shardID) {\n\t\tqps := tlMgr.QueriesPerSecond()\n\t\tload = load + qps\n\t}\n\treturn load\n}\n\nfunc validateSPParams(params ShardProcessorParams) error {\n\tif params.ShardID == \"\" {\n\t\treturn errors.New(\"ShardID must be specified\")\n\t}\n\tif params.TaskListsRegistry == nil {\n\t\treturn errors.New(\"TaskListsRegistry must be specified\")\n\t}\n\tif params.TimeSource == nil {\n\t\treturn errors.New(\"TimeSource must be specified\")\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/matching/tasklist/shard_processor_factory.go",
    "content": "package tasklist\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n)\n\n// ShardProcessorFactory is a generic factory for creating ShardProcessor instances.\ntype ShardProcessorFactory struct {\n\tTaskListsRegistry TaskListRegistry\n\tReportTTL         time.Duration\n\tTimeSource        clock.TimeSource\n}\n\nfunc (spf ShardProcessorFactory) NewShardProcessor(shardID string) (ShardProcessor, error) {\n\n\tparams := ShardProcessorParams{\n\t\tShardID:           shardID,\n\t\tTaskListsRegistry: spf.TaskListsRegistry,\n\t\tReportTTL:         spf.ReportTTL,\n\t\tTimeSource:        spf.TimeSource,\n\t}\n\treturn NewShardProcessor(params)\n}\n"
  },
  {
    "path": "service/matching/tasklist/shard_processor_test.go",
    "content": "package tasklist\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc mustNewIdentifier(domainID, name string, taskType int) *Identifier {\n\tid, err := NewIdentifier(domainID, name, taskType)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn id\n}\n\nvar testIdentifier = mustNewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\ntype shardProcessorTestData struct {\n\tmockRegistry   *MockTaskListRegistry\n\tshardProcessor ShardProcessor\n}\n\nfunc newShardProcessorTestData(t *testing.T, taskListID *Identifier) shardProcessorTestData {\n\tctrl := gomock.NewController(t)\n\n\tmockRegistry := NewMockTaskListRegistry(ctrl)\n\tmockRegistry.EXPECT().ManagersByTaskListName(taskListID.GetName()).Return([]Manager{}).AnyTimes()\n\n\tparams := ShardProcessorParams{\n\t\tShardID:           taskListID.GetName(),\n\t\tTaskListsRegistry: mockRegistry,\n\t\tReportTTL:         1 * time.Millisecond,\n\t\tTimeSource:        clock.NewRealTimeSource(),\n\t}\n\n\tshardProcessor, err := NewShardProcessor(params)\n\trequire.NoError(t, err)\n\treturn shardProcessorTestData{\n\t\tmockRegistry:   mockRegistry,\n\t\tshardProcessor: shardProcessor,\n\t}\n}\n\nfunc TestNewShardProcessorFailsWithEmptyParams(t *testing.T) {\n\tparams := ShardProcessorParams{}\n\tsp, err := NewShardProcessor(params)\n\trequire.Nil(t, sp)\n\trequire.Error(t, err)\n}\n\nfunc TestGetShardReport(t *testing.T) {\n\ttd := newShardProcessorTestData(t, testIdentifier)\n\n\tshardReport := td.shardProcessor.GetShardReport()\n\trequire.NotNil(t, shardReport)\n\trequire.Equal(t, float64(0), shardReport.ShardLoad)\n\trequire.Equal(t, types.ShardStatusREADY, shardReport.Status)\n}\n\nfunc TestSetShardStatus(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttd := newShardProcessorTestData(t, testIdentifier)\n\n\ttd.shardProcessor.SetShardStatus(types.ShardStatusREADY)\n\tshardReport := td.shardProcessor.GetShardReport()\n\trequire.NotNil(t, shardReport)\n\trequire.Equal(t, float64(0), shardReport.ShardLoad)\n\trequire.Equal(t, types.ShardStatusREADY, shardReport.Status)\n}\n"
  },
  {
    "path": "service/matching/tasklist/task.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// TODO: review the usage of InternalTask and provide a better abstraction\ntype (\n\t// genericTaskInfo contains the info for an activity or decision task\n\tgenericTaskInfo struct {\n\t\t*persistence.TaskInfo\n\t\tcompletionFunc func(*persistence.TaskInfo, error)\n\t}\n\t// queryTaskInfo contains the info for a query task\n\tqueryTaskInfo struct {\n\t\tTaskID  string\n\t\tRequest *types.MatchingQueryWorkflowRequest\n\t}\n\t// startedTaskInfo contains info for any task received from\n\t// another matching host. This type of task is already marked as started\n\tstartedTaskInfo struct {\n\t\tdecisionTaskInfo *types.MatchingPollForDecisionTaskResponse\n\t\tactivityTaskInfo *types.MatchingPollForActivityTaskResponse\n\t}\n\t// InternalTask represents an activity, decision, query or started (received from another host).\n\t// this struct is more like a union and only one of [ query, event, forwarded ] is\n\t// non-nil for any given task\n\tInternalTask struct {\n\t\tEvent                    *genericTaskInfo // non-nil for activity or decision task that's locally generated\n\t\tQuery                    *queryTaskInfo   // non-nil for a query task that's locally sync matched\n\t\tstarted                  *startedTaskInfo // non-nil for a task received from a parent partition which is already started\n\t\tdomainName               string\n\t\tsource                   types.TaskSource\n\t\tforwardedFrom            string     // name of the child partition this task is forwarded from (empty if not forwarded)\n\t\tisolationGroup           string     // isolation group of this task (empty if it can be polled by workers from any isolation group)\n\t\tResponseC                chan error // non-nil only where there is a caller waiting for response (sync-match)\n\t\tBacklogCountHint         int64\n\t\tActivityTaskDispatchInfo *types.ActivityTaskDispatchInfo\n\t\tAutoConfigHint           *types.AutoConfigHint // worker auto-scaler hint, which includes enable auto config flag and poller wait time on the matching engine\n\t}\n)\n\nfunc newInternalTask(\n\tinfo *persistence.TaskInfo,\n\tcompletionFunc func(*persistence.TaskInfo, error),\n\tsource types.TaskSource,\n\tforwardedFrom string,\n\tforSyncMatch bool,\n\tactivityTaskDispatchInfo *types.ActivityTaskDispatchInfo,\n\tisolationGroup string,\n) *InternalTask {\n\ttask := &InternalTask{\n\t\tEvent: &genericTaskInfo{\n\t\t\tTaskInfo:       info,\n\t\t\tcompletionFunc: completionFunc,\n\t\t},\n\t\tsource:                   source,\n\t\tforwardedFrom:            forwardedFrom,\n\t\tisolationGroup:           isolationGroup,\n\t\tActivityTaskDispatchInfo: activityTaskDispatchInfo,\n\t}\n\tif forSyncMatch {\n\t\ttask.ResponseC = make(chan error, 1)\n\t}\n\n\t// Rewrite the partitionConfig to match how we're dispatching it\n\t// OriginalIsolationGroup is populated here and isn't written to the DB. If it's already\n\t// present then it's a forwarded task and we should respect it.\n\tif configIsolationGroup, ok := task.Event.PartitionConfig[isolationgroup.GroupKey]; ok {\n\t\tpartitionConfig := make(map[string]string, 3)\n\t\tif originalIsolationGroup, ok := task.Event.PartitionConfig[isolationgroup.OriginalGroupKey]; ok {\n\t\t\tpartitionConfig[isolationgroup.OriginalGroupKey] = originalIsolationGroup\n\t\t} else {\n\t\t\tpartitionConfig[isolationgroup.OriginalGroupKey] = configIsolationGroup\n\t\t}\n\t\tpartitionConfig[isolationgroup.GroupKey] = isolationGroup\n\t\tpartitionConfig[isolationgroup.WorkflowIDKey] = task.Event.PartitionConfig[isolationgroup.WorkflowIDKey]\n\t\ttask.Event.PartitionConfig = partitionConfig\n\t}\n\treturn task\n}\n\nfunc newInternalQueryTask(\n\ttaskID string,\n\trequest *types.MatchingQueryWorkflowRequest,\n) *InternalTask {\n\treturn &InternalTask{\n\t\tQuery: &queryTaskInfo{\n\t\t\tTaskID:  taskID,\n\t\t\tRequest: request,\n\t\t},\n\t\tforwardedFrom: request.GetForwardedFrom(),\n\t\tResponseC:     make(chan error, 1),\n\t}\n}\n\nfunc newInternalStartedTask(info *startedTaskInfo) *InternalTask {\n\treturn &InternalTask{started: info}\n}\n\n// isQuery returns true if the underlying task is a query task\nfunc (task *InternalTask) IsQuery() bool {\n\treturn task.Query != nil\n}\n\n// isStarted is true when this task is already marked as started\nfunc (task *InternalTask) IsStarted() bool {\n\treturn task.started != nil\n}\n\n// isForwarded returns true if the underlying task is forwarded by a remote matching host\n// forwarded tasks are already marked as started in history\nfunc (task *InternalTask) IsForwarded() bool {\n\treturn task.forwardedFrom != \"\"\n}\n\nfunc (task *InternalTask) IsSyncMatch() bool {\n\treturn task.ResponseC != nil\n}\n\nfunc (task *InternalTask) Info() persistence.TaskInfo {\n\tif task == nil || task.Event == nil || task.Event.TaskInfo == nil {\n\t\treturn persistence.TaskInfo{}\n\t}\n\n\treturn *task.Event.TaskInfo\n}\n\nfunc (task *InternalTask) WorkflowExecution() *types.WorkflowExecution {\n\tswitch {\n\tcase task.Event != nil:\n\t\treturn &types.WorkflowExecution{WorkflowID: task.Event.WorkflowID, RunID: task.Event.RunID}\n\tcase task.Query != nil:\n\t\treturn task.Query.Request.GetQueryRequest().GetExecution()\n\tcase task.started != nil && task.started.decisionTaskInfo != nil:\n\t\treturn task.started.decisionTaskInfo.WorkflowExecution\n\tcase task.started != nil && task.started.activityTaskInfo != nil:\n\t\treturn task.started.activityTaskInfo.WorkflowExecution\n\t}\n\treturn &types.WorkflowExecution{}\n}\n\n// pollForDecisionResponse returns the poll response for a decision task that is\n// already marked as started. This method should only be called when isStarted() is true\nfunc (task *InternalTask) PollForDecisionResponse() *types.MatchingPollForDecisionTaskResponse {\n\tif task.IsStarted() {\n\t\treturn task.started.decisionTaskInfo\n\t}\n\treturn nil\n}\n\n// pollForActivityResponse returns the poll response for an activity task that is\n// already marked as started. This method should only be called when isStarted() is true\nfunc (task *InternalTask) PollForActivityResponse() *types.MatchingPollForActivityTaskResponse {\n\tif task.IsStarted() {\n\t\treturn task.started.activityTaskInfo\n\t}\n\treturn nil\n}\n\n// finish marks a task as finished. Should be called after a poller picks up a task\n// and marks it as started. If the task is unable to marked as started, then this\n// method should be called with a non-nil error argument.\nfunc (task *InternalTask) Finish(err error) {\n\tswitch {\n\tcase task.ResponseC != nil:\n\t\ttask.ResponseC <- err\n\tcase task.Event.completionFunc != nil:\n\t\ttask.Event.completionFunc(task.Event.TaskInfo, err)\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_completer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\ttaskCompleterImpl struct {\n\t\tdomainCache     cache.DomainCache\n\t\ttaskListID      *Identifier\n\t\tclusterMetadata cluster.Metadata\n\t\thistoryService  history.Client\n\t\tscope           metrics.Scope\n\t\tlogger          log.Logger\n\t\tthrottleRetry   *backoff.ThrottleRetry\n\t\ttimeSource      clock.TimeSource\n\t}\n)\n\n// DomainIsActiveInThisClusterError type\ntype DomainIsActiveInThisClusterError struct {\n\tMessage string\n}\n\n// Implement the Error method for DomainIsActiveInThisClusterError\nfunc (e *DomainIsActiveInThisClusterError) Error() string {\n\treturn e.Message\n}\n\nvar (\n\terrWorkflowExecutionInfoIsNil           = errors.New(\"workflow execution info is nil\")\n\terrTaskTypeNotSupported                 = errors.New(\"task type not supported\")\n\terrTaskNotStarted                       = errors.New(\"task not started\")\n\terrWaitTimeNotReachedForEntityNotExists = errors.New(\"wait time not reached for workflow EntityNotExistsError\")\n\terrDomainIsActive                       = &DomainIsActiveInThisClusterError{Message: \"domain is active\"}\n\thistoryServiceOperationRetryPolicy      = common.CreateTaskCompleterRetryPolicy()\n)\n\nfunc newTaskCompleter(tlMgr *taskListManagerImpl, retryPolicy backoff.RetryPolicy) TaskCompleter {\n\treturn &taskCompleterImpl{\n\t\tdomainCache:     tlMgr.domainCache,\n\t\thistoryService:  tlMgr.historyService,\n\t\ttaskListID:      tlMgr.taskListID,\n\t\tclusterMetadata: tlMgr.clusterMetadata,\n\t\tscope:           tlMgr.scope,\n\t\tlogger:          tlMgr.logger,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\t\tbackoff.WithRetryableError(isRetryableError),\n\t\t),\n\t\ttimeSource: tlMgr.timeSource,\n\t}\n}\n\nfunc (tc *taskCompleterImpl) CompleteTaskIfStarted(ctx context.Context, task *InternalTask) error {\n\top := func(ctx context.Context) (err error) {\n\t\tdomainEntry, err := tc.domainCache.GetDomainByID(task.Event.TaskInfo.DomainID)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unable to fetch domain from cache: %w\", err)\n\t\t}\n\n\t\tif domainEntry.IsActiveIn(tc.clusterMetadata.GetCurrentClusterName()) {\n\t\t\ttc.logger.Debug(\"Domain is active in the current cluster, completing task\",\n\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t)\n\t\t\treturn errDomainIsActive\n\t\t}\n\n\t\treq := &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\tDomainUUID: task.Event.TaskInfo.DomainID,\n\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\tDomain: task.domainName,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: task.Event.WorkflowID,\n\t\t\t\t\tRunID:      task.Event.RunID,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tworkflowExecutionResponse, err := tc.historyService.DescribeWorkflowExecution(ctx, req)\n\n\t\tif errors.As(err, new(*types.EntityNotExistsError)) {\n\t\t\ttc.logger.Warn(\"Workflow execution not found while attempting to complete task on standby cluster\", tag.WorkflowID(task.Event.WorkflowID), tag.WorkflowRunID(task.Event.RunID))\n\n\t\t\t// this is a guard to prevent some race condition or quorum issue where the workflow is not found in the database\n\t\t\tif tc.timeSource.Since(task.Event.CreatedTime) > 24*time.Hour {\n\t\t\t\ttask.Finish(nil)\n\n\t\t\t\ttc.scope.IncCounter(metrics.StandbyClusterTasksCompletedCounterPerTaskList)\n\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\treturn errWaitTimeNotReachedForEntityNotExists\n\t\t} else if err != nil {\n\t\t\treturn fmt.Errorf(\"unable to fetch workflow execution from the history service: %w\", err)\n\t\t}\n\n\t\tif workflowExecutionResponse.WorkflowExecutionInfo == nil {\n\t\t\treturn errWorkflowExecutionInfoIsNil\n\t\t}\n\n\t\tisTaskStarted, err := tc.isTaskStarted(task, workflowExecutionResponse)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"unable to determine if task has started: %w\", err)\n\t\t}\n\n\t\tif isTaskStarted {\n\t\t\ttask.Finish(nil)\n\n\t\t\ttc.scope.IncCounter(metrics.StandbyClusterTasksCompletedCounterPerTaskList)\n\n\t\t\treturn nil\n\t\t}\n\n\t\ttc.scope.IncCounter(metrics.StandbyClusterTasksNotStartedCounterPerTaskList)\n\n\t\treturn errTaskNotStarted\n\t}\n\n\terr := tc.throttleRetry.Do(ctx, op)\n\n\tif err != nil && !errors.Is(err, errDomainIsActive) && !errors.Is(err, errTaskNotStarted) {\n\t\ttc.scope.IncCounter(metrics.StandbyClusterTasksCompletionFailurePerTaskList)\n\t\ttc.logger.Error(\"Error completing task on domain's standby cluster\", tag.Error(err), tag.WorkflowID(task.Event.WorkflowID), tag.WorkflowRunID(task.Event.RunID), tag.MatchingTaskID(task.Event.TaskID))\n\t}\n\n\treturn err\n}\n\nfunc (tc *taskCompleterImpl) isTaskStarted(task *InternalTask, workflowExecutionResponse *types.DescribeWorkflowExecutionResponse) (bool, error) {\n\tif workflowExecutionResponse.WorkflowExecutionInfo.CloseStatus != nil {\n\t\treturn true, nil\n\t}\n\n\t// taskType can only be Activity or Decision, leaving the default case for future proofing\n\tswitch tc.taskListID.taskType {\n\tcase persistence.TaskListTypeActivity:\n\t\treturn tc.isActivityTaskStarted(task, workflowExecutionResponse), nil\n\tcase persistence.TaskListTypeDecision:\n\t\treturn tc.isDecisionTaskStarted(task, workflowExecutionResponse), nil\n\tdefault:\n\t\treturn false, errTaskTypeNotSupported\n\t}\n}\n\nfunc (tc *taskCompleterImpl) isDecisionTaskStarted(task *InternalTask, workflowExecutionResponse *types.DescribeWorkflowExecutionResponse) bool {\n\t// if there is no pending decision task, that means that this task has been already started\n\tif workflowExecutionResponse.PendingDecision == nil {\n\t\treturn true\n\t}\n\n\t// if the scheduleID is different from the pending decision scheduleID or the state is started, then the decision task with the task's scheduleID has already been started\n\tif task.Event.ScheduleID < workflowExecutionResponse.PendingDecision.ScheduleID || (task.Event.ScheduleID == workflowExecutionResponse.PendingDecision.ScheduleID && *workflowExecutionResponse.PendingDecision.State == types.PendingDecisionStateStarted) {\n\t\treturn true\n\t}\n\n\ttc.logger.Debug(\"Decision task not started.\",\n\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\ttag.MatchingTaskID(task.Event.TaskID),\n\t\ttag.MatchingTaskScheduleID(task.Event.ScheduleID),\n\t\ttag.WorkflowScheduleID(workflowExecutionResponse.PendingDecision.ScheduleID),\n\t\ttag.DecisionTaskState(int32(*workflowExecutionResponse.PendingDecision.State)),\n\t)\n\n\treturn false\n}\n\nfunc (tc *taskCompleterImpl) isActivityTaskStarted(task *InternalTask, workflowExecutionResponse *types.DescribeWorkflowExecutionResponse) bool {\n\t// if the scheduleID is different from all pending tasks' scheduleID or the pending activity has PendingActivityStateStarted, then the activity task with the task's scheduleID has already been started\n\tfor _, pendingActivity := range workflowExecutionResponse.PendingActivities {\n\t\tif task.Event.ScheduleID == pendingActivity.ScheduleID {\n\t\t\tif *pendingActivity.State == types.PendingActivityStateStarted {\n\t\t\t\treturn true\n\t\t\t}\n\n\t\t\ttc.logger.Debug(\"Activity task not started.\",\n\t\t\t\ttag.WorkflowID(task.Event.WorkflowID),\n\t\t\t\ttag.WorkflowRunID(task.Event.RunID),\n\t\t\t\ttag.MatchingTaskID(task.Event.TaskID),\n\t\t\t\ttag.MatchingTaskScheduleID(task.Event.ScheduleID),\n\t\t\t\ttag.WorkflowScheduleID(pendingActivity.ScheduleID),\n\t\t\t\ttag.ActivityTaskState(int32(*pendingActivity.State)),\n\t\t\t)\n\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc isRetryableError(err error) bool {\n\t// entityNotExistsError is returned when the workflow is not found in the database\n\tvar domainIsActiveInThisClusterError *DomainIsActiveInThisClusterError\n\tswitch {\n\tcase errors.As(err, &domainIsActiveInThisClusterError):\n\t\treturn false\n\t}\n\treturn true\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_completer_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\nvar retryPolicyMaxAttempts = 3\n\nfunc createTestTaskCompleter(controller *gomock.Controller, taskType int) *taskCompleterImpl {\n\tmockDomainCache := cache.NewMockDomainCache(controller)\n\tmockHistoryService := history.NewMockClient(controller)\n\n\tretryPolicy := backoff.NewExponentialRetryPolicy(100 * time.Millisecond)\n\tretryPolicy.SetMaximumAttempts(retryPolicyMaxAttempts)\n\n\ttlMgr := &taskListManagerImpl{\n\t\tdomainCache:     mockDomainCache,\n\t\thistoryService:  mockHistoryService,\n\t\ttaskListID:      &Identifier{domainID: constants.TestDomainID, taskType: taskType},\n\t\tclusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\tscope:           metrics.NoopScope,\n\t\tlogger:          log.NewNoop(),\n\t}\n\ttc := newTaskCompleter(tlMgr, retryPolicy)\n\n\treturn tc.(*taskCompleterImpl)\n}\n\nfunc TestCompleteTaskIfStarted(t *testing.T) {\n\tctx := context.Background()\n\tcreatedAt := time.Date(2020, 8, 1, 0, 0, 0, 0, time.UTC)\n\n\ttestCases := []struct {\n\t\tname           string\n\t\tsetupMock      func(*types.HistoryDescribeWorkflowExecutionRequest, *cache.MockDomainCache, *history.MockClient, clock.MockedTimeSource)\n\t\ttask           func(chan bool) *InternalTask\n\t\ttaskType       int\n\t\tisTaskComplete bool\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname: \"error - could not get domain by ID from cache\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(nil, errors.New(\"error-getting-domain-by-id\")).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errors.New(\"error-getting-domain-by-id\"),\n\t\t},\n\t\t{\n\t\t\tname: \"error - domain is active\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalDomainEntry, nil).Times(1)\n\t\t\t},\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errDomainIsActive,\n\t\t},\n\t\t{\n\t\t\tname: \"error - could not fetch workflow execution from history service\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(nil, errors.New(\"error-describing-workflow-execution\")).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errors.New(\"error-describing-workflow-execution\"),\n\t\t},\n\t\t{\n\t\t\tname: \"error - no WorkflowExecutionInfo in workflow execution response\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errWorkflowExecutionInfoIsNil,\n\t\t},\n\t\t{\n\t\t\tname: \"error - task type not supported\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\ttaskType:       999,\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errTaskTypeNotSupported,\n\t\t},\n\t\t{\n\t\t\tname: \"error - decision task not started - scheduleID greater than PendingDecision scheduleID\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\t\t\tScheduleID: 2,\n\t\t\t\t\t\tState:      types.PendingDecisionStateScheduled.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errTaskNotStarted,\n\t\t},\n\t\t{\n\t\t\tname: \"error - decision task not started - scheduleID equal to PendingDecision scheduleID but PendingDecision state is not PendingDecisionStateStarted\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\t\t\tScheduleID: 3,\n\t\t\t\t\t\tState:      types.PendingDecisionStateScheduled.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errTaskNotStarted,\n\t\t},\n\t\t{\n\t\t\tname: \"error - activity task not started - activity matching scheduleID is in PendingActivities but its state is not PendingActivityStateStarted\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tScheduleID: 3,\n\t\t\t\t\t\t\tState:      types.PendingActivityStateScheduled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeActivity,\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errTaskNotStarted,\n\t\t},\n\t\t{\n\t\t\tname: \"error - workflow not found and task created less than 24 hours before completion attempt\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(retryPolicyMaxAttempts + 1)\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(nil, &types.EntityNotExistsError{}).Times(retryPolicyMaxAttempts + 1)\n\t\t\t},\n\t\t\tisTaskComplete: false,\n\t\t\terr:            errWaitTimeNotReachedForEntityNotExists,\n\t\t},\n\t\t{\n\t\t\tname: \"complete task - workflow not found and task created more than 24 hours before completion attempt\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(nil, &types.EntityNotExistsError{}).Times(1)\n\t\t\t\ttimeSource.Advance(time.Hour*24 + time.Second)\n\t\t\t},\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete task - workflow closed\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete decision task - no pending decision\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete decision task - scheduleID is less than PendingDecision scheduleID\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  2,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\t\t\tScheduleID: 3,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete decision task - scheduleID is equal to PendingDecision scheduleID and PendingDecision state is PendingDecisionStateStarted\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\t\t\tScheduleID: 3,\n\t\t\t\t\t\tState:      types.PendingDecisionStateStarted.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeDecision,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete activity task - no activity matching scheduleID in PendingActivities\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tScheduleID: 2,\n\t\t\t\t\t\t\tState:      types.PendingActivityStateScheduled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tScheduleID: 4,\n\t\t\t\t\t\t\tState:      types.PendingActivityStateScheduled.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeActivity,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t\t{\n\t\t\tname: \"complete activity task - activity matching scheduleID is in PendingActivities and its state is PendingActivityStateStarted\",\n\t\t\ttask: func(isComplete chan bool) *InternalTask {\n\t\t\t\treturn &InternalTask{\n\t\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\t\t\tScheduleID:  3,\n\t\t\t\t\t\t\tCreatedTime: createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tcompletionFunc: func(_ *persistence.TaskInfo, _ error) { isComplete <- true },\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t},\n\t\t\tsetupMock: func(req *types.HistoryDescribeWorkflowExecutionRequest, mockDomainCache *cache.MockDomainCache, mockHistoryService *history.MockClient, timeSource clock.MockedTimeSource) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(constants.TestGlobalStandbyDomainEntry, nil).Times(1)\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tScheduleID: 3,\n\t\t\t\t\t\t\tState:      types.PendingActivityStateStarted.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockHistoryService.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil).Times(1)\n\t\t\t},\n\t\t\ttaskType:       persistence.TaskListTypeActivity,\n\t\t\tisTaskComplete: true,\n\t\t\terr:            nil,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\ttCmp := createTestTaskCompleter(ctrl, tc.taskType)\n\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tisTaskComplete := make(chan bool, 1)\n\n\t\t\ttask := tc.task(isTaskComplete)\n\n\t\t\treq := &types.HistoryDescribeWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: task.Event.TaskInfo.DomainID,\n\t\t\t\tRequest: &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: task.domainName,\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: task.Event.WorkflowID,\n\t\t\t\t\t\tRunID:      task.Event.RunID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tmockedTimeSource := clock.NewMockedTimeSourceAt(createdAt)\n\t\t\ttCmp.timeSource = mockedTimeSource\n\n\t\t\ttc.setupMock(req, tCmp.domainCache.(*cache.MockDomainCache), tCmp.historyService.(*history.MockClient), mockedTimeSource)\n\n\t\t\terr := tCmp.CompleteTaskIfStarted(ctx, task)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tif errors.Unwrap(err) != nil {\n\t\t\t\t\tassert.ErrorContains(t, errors.Unwrap(err), tc.err.Error())\n\t\t\t\t} else {\n\t\t\t\t\tassert.ErrorContains(t, err, tc.err.Error())\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\tvar val bool\n\t\t\tselect {\n\t\t\tcase val = <-isTaskComplete:\n\t\t\tdefault:\n\t\t\t}\n\n\t\t\tassert.Equal(t, tc.isTaskComplete, val)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_gc.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\ntype taskGC struct {\n\tlock           int64\n\tdb             *taskListDB\n\tackLevel       int64\n\tlastDeleteTime time.Time\n\tconfig         *config.TaskListConfig\n}\n\n// newTaskGC returns an instance of a task garbage collector object\n// taskGC internally maintains a delete cursor and attempts to delete\n// a batch of tasks everytime Run() method is called.\n//\n// In order for the taskGC to actually delete tasks when Run() is called, one of\n// two conditions must be met\n//   - Size Threshold: More than MaxDeleteBatchSize tasks are waiting to be deleted (rough estimation)\n//   - Time Threshold: Time since previous delete was attempted exceeds maxTimeBetweenTaskDeletes\n//\n// Finally, the Run() method is safe to be called from multiple threads. The underlying\n// implementation will make sure only one caller executes Run() and others simply bail out\nfunc newTaskGC(db *taskListDB, config *config.TaskListConfig) *taskGC {\n\treturn &taskGC{db: db, config: config}\n}\n\n// Run deletes a batch of completed tasks, if its possible to do so\n// Only attempts deletion if size or time thresholds are met\nfunc (tgc *taskGC) Run(ackLevel int64) {\n\ttgc.tryDeleteNextBatch(ackLevel, false)\n}\n\n// RunNow deletes a batch of completed tasks if its possible to do so\n// This method attempts deletions without waiting for size/time threshold to be met\nfunc (tgc *taskGC) RunNow(ackLevel int64) {\n\ttgc.tryDeleteNextBatch(ackLevel, true)\n}\n\nfunc (tgc *taskGC) tryDeleteNextBatch(ackLevel int64, ignoreTimeCond bool) {\n\tif !tgc.tryLock() {\n\t\treturn\n\t}\n\tdefer tgc.unlock()\n\tbatchSize := tgc.config.MaxTaskDeleteBatchSize()\n\tif !tgc.checkPrecond(ackLevel, batchSize, ignoreTimeCond) {\n\t\treturn\n\t}\n\ttgc.lastDeleteTime = time.Now()\n\tfor {\n\t\tn, err := tgc.db.CompleteTasksLessThan(ackLevel, batchSize)\n\t\tif err != nil {\n\t\t\tbreak\n\t\t}\n\t\tif n < batchSize {\n\t\t\ttgc.ackLevel = ackLevel\n\t\t\tbreak\n\t\t}\n\t}\n}\n\nfunc (tgc *taskGC) checkPrecond(ackLevel int64, batchSize int, ignoreTimeCond bool) bool {\n\tbacklog := ackLevel - tgc.ackLevel\n\tif backlog >= int64(batchSize) {\n\t\treturn true\n\t}\n\treturn backlog > 0 && (ignoreTimeCond || time.Since(tgc.lastDeleteTime) > tgc.config.MaxTimeBetweenTaskDeletes)\n}\n\nfunc (tgc *taskGC) tryLock() bool {\n\treturn atomic.CompareAndSwapInt64(&tgc.lock, 0, 1)\n}\n\nfunc (tgc *taskGC) unlock() {\n\tatomic.StoreInt64(&tgc.lock, 0)\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_limiter.go",
    "content": "package tasklist\n\nimport (\n\t\"context\"\n\t\"math\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/atomic\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nconst precision = 0.001\n\ntype taskListLimiter struct {\n\tbacking           clock.Ratelimiter\n\ttimeSource        clock.TimeSource\n\tscope             metrics.Scope\n\tttl               time.Duration\n\tminBurst          int\n\tlock              sync.Mutex\n\tvalue             atomic.Float64\n\tlastReceivedValue atomic.Time\n\tlastUpdate        atomic.Time\n\tcountPartitions   func() int\n\tpartitions        int\n}\n\nfunc newTaskListLimiter(timeSource clock.TimeSource, scope metrics.Scope, config *config.TaskListConfig, numPartitions func() int) *taskListLimiter {\n\tl := &taskListLimiter{\n\t\ttimeSource:      timeSource,\n\t\tscope:           scope,\n\t\tttl:             config.TaskDispatchRPSTTL,\n\t\tcountPartitions: numPartitions,\n\t\tminBurst:        config.MinTaskThrottlingBurstSize(),\n\t}\n\tl.value.Store(config.TaskDispatchRPS)\n\tl.partitions = numPartitions()\n\tnow := timeSource.Now()\n\tl.lastUpdate.Store(now)\n\tl.lastReceivedValue.Store(now)\n\tl.backing = clock.NewRatelimiter(l.getLimitAndBurst())\n\treturn l\n}\n\nfunc (l *taskListLimiter) Allow() bool {\n\treturn l.backing.Allow()\n}\n\nfunc (l *taskListLimiter) Wait(ctx context.Context) error {\n\treturn l.backing.Wait(ctx)\n}\n\nfunc (l *taskListLimiter) Reserve() clock.Reservation {\n\treturn l.backing.Reserve()\n}\n\nfunc (l *taskListLimiter) Limit() rate.Limit {\n\treturn l.backing.Limit()\n}\n\nfunc (l *taskListLimiter) ReportLimit(rps float64) {\n\tnow := l.timeSource.Now()\n\t// Optimistically reject it without locking if it's >= current and within the TTL\n\tif now.Sub(l.lastUpdate.Load()) < l.ttl {\n\t\tcurrent := l.value.Load()\n\t\tif rps > current {\n\t\t\treturn\n\t\t}\n\t\t// If it's roughly equal to the current value, track the timestamp\n\t\t// We'll maintain this value in the next update unless something lower\n\t\t// comes along\n\t\tif math.Abs(current-rps) < precision {\n\t\t\tl.lastReceivedValue.Store(now)\n\t\t\treturn\n\t\t}\n\t\t// else if rps < current, we try to update\n\t}\n\tl.tryUpdate(rps)\n}\n\nfunc (l *taskListLimiter) tryUpdate(rps float64) {\n\tl.lock.Lock()\n\tdefer l.lock.Unlock()\n\tnow := l.timeSource.Now()\n\tcurrent := l.value.Load()\n\tlastUpdated := l.lastUpdate.Load()\n\tttlElapsed := now.Sub(lastUpdated) >= l.ttl\n\tchanged := false\n\n\t// Take the lower value, or if the ttl expired and haven't received the current low value within the TTL, take the\n\t// new value\n\tif rps < current || (ttlElapsed && !l.lastReceivedValue.Load().After(now.Add(-l.ttl))) {\n\t\tl.lastUpdate.Store(now)\n\t\tl.value.Store(rps)\n\t\tl.lastReceivedValue.Store(now)\n\t\tl.partitions = l.countPartitions()\n\t\tchanged = true\n\t\tl.scope.UpdateGauge(metrics.RateLimitPerTaskListGauge, rps)\n\t} else if ttlElapsed {\n\t\t// If the TTL elapsed, recalculate the partition count in case it changed\n\t\tl.lastUpdate.Store(now)\n\t\tnewPartitions := l.countPartitions()\n\t\tif newPartitions != l.partitions {\n\t\t\tl.partitions = newPartitions\n\t\t\tchanged = true\n\t\t}\n\t\tl.scope.UpdateGauge(metrics.RateLimitPerTaskListGauge, current)\n\t}\n\n\tif changed {\n\t\tl.backing.SetLimitAndBurst(l.getLimitAndBurst())\n\t}\n}\n\nfunc (l *taskListLimiter) getLimitAndBurst() (rate.Limit, int) {\n\trps := l.value.Load()\n\trps = rps / float64(l.partitions)\n\tburst := max(int(math.Ceil(rps)), l.minBurst)\n\tif rps == 0 {\n\t\tburst = 0\n\t}\n\treturn rate.Limit(rps), burst\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_limiter_test.go",
    "content": "package tasklist\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nfunc TestTaskListLimiter(t *testing.T) {\n\tconst ttl = time.Second\n\tconst defaultRps = 100\n\tcases := []struct {\n\t\tname     string\n\t\tfn       func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int)\n\t\texpected rate.Limit\n\t}{\n\t\t{\n\t\t\tname:     \"no update\",\n\t\t\tfn:       func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {},\n\t\t\texpected: rate.Limit(defaultRps),\n\t\t},\n\t\t{\n\t\t\tname: \"take lower\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tlimiter.ReportLimit(1)\n\t\t\t},\n\t\t\texpected: rate.Limit(1),\n\t\t},\n\t\t{\n\t\t\tname: \"take lowest\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tlimiter.ReportLimit(50)\n\t\t\t\tlimiter.ReportLimit(10)\n\t\t\t},\n\t\t\texpected: rate.Limit(10),\n\t\t},\n\t\t{\n\t\t\tname: \"ignore higher\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tlimiter.ReportLimit(defaultRps + 1)\n\t\t\t},\n\t\t\texpected: rate.Limit(defaultRps),\n\t\t},\n\t\t{\n\t\t\tname: \"take higher after ttl\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(ttl)\n\t\t\t\tlimiter.ReportLimit(101)\n\t\t\t},\n\t\t\texpected: rate.Limit(101),\n\t\t},\n\t\t{\n\t\t\tname: \"keep lower if reported within ttl\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(time.Millisecond)\n\t\t\t\tlimiter.ReportLimit(defaultRps)\n\t\t\t\tmockClock.Advance(ttl - time.Nanosecond)\n\t\t\t\tlimiter.ReportLimit(defaultRps + 1)\n\t\t\t},\n\t\t\texpected: rate.Limit(defaultRps),\n\t\t},\n\t\t{\n\t\t\tname: \"taker higher if lower not recently reported\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(time.Millisecond)\n\t\t\t\tlimiter.ReportLimit(defaultRps)\n\t\t\t\tmockClock.Advance(ttl + time.Nanosecond)\n\t\t\t\tlimiter.ReportLimit(defaultRps + 1)\n\t\t\t},\n\t\t\texpected: rate.Limit(defaultRps + 1),\n\t\t},\n\t\t{\n\t\t\tname: \"recalculate partitions on ttl expiration - lower limit\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(ttl)\n\t\t\t\t*numPartitions = 2\n\t\t\t\tlimiter.ReportLimit(50)\n\t\t\t},\n\t\t\texpected: rate.Limit(25),\n\t\t},\n\t\t{\n\t\t\tname: \"recalculate partitions on ttl expiration - equal limit\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(ttl)\n\t\t\t\t*numPartitions = 2\n\t\t\t\tlimiter.ReportLimit(defaultRps)\n\t\t\t},\n\t\t\texpected: rate.Limit(50),\n\t\t},\n\t\t{\n\t\t\tname: \"recalculate partitions on ttl expiration - higher limit\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\tmockClock.Advance(ttl)\n\t\t\t\t*numPartitions = 2\n\t\t\t\tlimiter.ReportLimit(defaultRps + 2)\n\t\t\t},\n\t\t\texpected: rate.Limit(51),\n\t\t},\n\t\t{\n\t\t\tname: \"recalculate partitions on early updates\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\t*numPartitions = 2\n\t\t\t\tlimiter.ReportLimit(50)\n\t\t\t},\n\t\t\texpected: rate.Limit(25),\n\t\t},\n\t\t{\n\t\t\tname: \"recalculate partitions even when keeping lower limit\",\n\t\t\tfn: func(mockClock clock.MockedTimeSource, limiter *taskListLimiter, numPartitions *int) {\n\t\t\t\t// lastUpdate time\n\t\t\t\tlimiter.ReportLimit(50)\n\t\t\t\tmockClock.Advance(time.Nanosecond)\n\t\t\t\tlimiter.ReportLimit(50)\n\t\t\t\tmockClock.Advance(ttl - time.Nanosecond)\n\t\t\t\t*numPartitions = 2\n\t\t\t\t// 100 won't be applied because we received 50 within the TTL, but\n\t\t\t\t// it's been TTL since we last did an update so recalculate the partitions\n\t\t\t\tlimiter.ReportLimit(100)\n\t\t\t},\n\t\t\texpected: rate.Limit(25),\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmockClock := clock.NewMockedTimeSource()\n\t\t\tscope := metrics.NoopScope\n\t\t\tnumPartitions := 1\n\t\t\ttlConfig := &config.TaskListConfig{\n\t\t\t\tTaskDispatchRPSTTL: ttl,\n\t\t\t\tTaskDispatchRPS:    defaultRps,\n\t\t\t\tMinTaskThrottlingBurstSize: func() int {\n\t\t\t\t\treturn 1\n\t\t\t\t},\n\t\t\t\tOverrideTaskListRPS: func() float64 {\n\t\t\t\t\treturn 0\n\t\t\t\t},\n\t\t\t}\n\t\t\tlimiter := newTaskListLimiter(mockClock, scope, tlConfig, func() int {\n\t\t\t\treturn numPartitions\n\t\t\t})\n\t\t\ttc.fn(mockClock, limiter, &numPartitions)\n\t\t\tassert.Equal(t, tc.expected, limiter.Limit())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_manager.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"slices\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"golang.org/x/sync/errgroup\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/stats\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n\t\"github.com/uber/cadence/service/matching/liveness\"\n\t\"github.com/uber/cadence/service/matching/poller\"\n)\n\nconst (\n\t// Time budget for empty task to propagate through the function stack and be returned to\n\t// pollForActivityTask or pollForDecisionTask handler.\n\treturnEmptyTaskTimeBudget = time.Second\n\tnoIsolationTimeout        = time.Duration(0)\n\tminimumIsolationDuration  = time.Millisecond * 50\n\n\tnotifyPartitionConfigTimeout = 5 * time.Second\n)\n\nvar (\n\t// ErrNoTasks is exported temporarily for integration test\n\tErrNoTasks                        = errors.New(\"no tasks\")\n\tpersistenceOperationRetryPolicy   = common.CreatePersistenceRetryPolicy()\n\ttaskListActivityTypeTag           = metrics.TaskListTypeTag(\"activity\")\n\ttaskListDecisionTypeTag           = metrics.TaskListTypeTag(\"decision\")\n\tIsolationLeakCauseError           = metrics.IsolationLeakCause(\"error\")\n\tIsolationLeakCauseGroupDrained    = metrics.IsolationLeakCause(\"group_drained\")\n\tIsolationLeakCauseGroupUnknown    = metrics.IsolationLeakCause(\"group_unknown\")\n\tIsolationLeakCauseNoRecentPollers = metrics.IsolationLeakCause(\"no_recent_pollers\")\n\tIsolationLeakCausePartitionChange = metrics.IsolationLeakCause(\"partition_change\")\n\tIsolationLeakCauseExpired         = metrics.IsolationLeakCause(\"expired\")\n\n\tdefaultPartition = &types.TaskListPartition{}\n)\n\ntype (\n\tpollerIDCtxKey       struct{}\n\tidentityCtxKey       struct{}\n\tisolationGroupCtxKey struct{}\n\n\tManagerParams struct {\n\t\tDomainCache     cache.DomainCache\n\t\tLogger          log.Logger\n\t\tMetricsClient   metrics.Client\n\t\tTaskManager     persistence.TaskManager\n\t\tClusterMetadata cluster.Metadata\n\t\tIsolationState  isolationgroup.State\n\t\tMatchingClient  matching.Client\n\t\tRegistry        TaskListRegistry\n\t\tTaskList        *Identifier\n\t\tTaskListKind    types.TaskListKind\n\t\tCfg             *config.Config\n\t\tTimeSource      clock.TimeSource\n\t\tCreateTime      time.Time\n\t\tHistoryService  history.Client\n\t}\n\n\tAddTaskParams struct {\n\t\tTaskInfo                 *persistence.TaskInfo\n\t\tSource                   types.TaskSource\n\t\tForwardedFrom            string\n\t\tActivityTaskDispatchInfo *types.ActivityTaskDispatchInfo\n\t}\n\n\t// Single task list in memory state\n\ttaskListManagerImpl struct {\n\t\tcreateTime      time.Time\n\t\tenableIsolation bool\n\t\ttaskListID      *Identifier\n\t\ttaskListKind    types.TaskListKind // sticky taskList has different process in persistence\n\t\tconfig          *config.TaskListConfig\n\t\tdb              *taskListDB\n\t\ttaskWriter      *taskWriter\n\t\ttaskReader      *taskReader // reads tasks from db and async matches it with poller\n\t\tliveness        *liveness.Liveness\n\t\ttaskGC          *taskGC\n\t\ttaskAckManager  messaging.AckManager // tracks ackLevel for delivered messages\n\t\tmatcher         TaskMatcher          // for matching a task producer with a poller\n\t\tlimiter         *taskListLimiter\n\t\tclusterMetadata cluster.Metadata\n\t\tdomainCache     cache.DomainCache\n\t\tisolationState  isolationgroup.State\n\t\tisolationGroups []string\n\t\tlogger          log.Logger\n\t\tscope           metrics.Scope\n\t\ttimeSource      clock.TimeSource\n\t\tmatchingClient  matching.Client\n\t\tdomainName      string\n\t\t// pollers stores poller which poll from this tasklist in last few minutes\n\t\tpollers       poller.Manager\n\t\tstartWG       sync.WaitGroup // ensures that background processes do not start until setup is ready\n\t\tstopWG        sync.WaitGroup\n\t\tstopped       int32\n\t\tstoppedLock   sync.RWMutex\n\t\tregistry      TaskListRegistry\n\t\tthrottleRetry *backoff.ThrottleRetry\n\n\t\tqpsTracker     stats.QPSTrackerGroup\n\t\tadaptiveScaler AdaptiveScaler\n\n\t\tpartitionConfigLock sync.RWMutex\n\t\tpartitionConfig     *types.TaskListPartitionConfig\n\t\thistoryService      history.Client\n\t\ttaskCompleter       TaskCompleter\n\t}\n)\n\nconst (\n\t// maxSyncMatchWaitTime is the max amount of time that we are willing to wait for a sync match to happen\n\tmaxSyncMatchWaitTime = 200 * time.Millisecond\n)\n\nvar errRemoteSyncMatchFailed = &types.RemoteSyncMatchedError{Message: \"remote sync match failed\"}\n\nfunc NewManager(p ManagerParams) (Manager, error) {\n\terr := validateParams(p)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainName, err := p.DomainCache.GetDomainName(p.TaskList.GetDomainID())\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\ttaskListConfig := newTaskListConfig(p.TaskList, p.Cfg, domainName)\n\n\tscope := common.NewPerTaskListScope(domainName, p.TaskList.GetName(), p.TaskListKind, p.MetricsClient, metrics.MatchingTaskListMgrScope).\n\t\tTagged(getTaskListTypeTag(p.TaskList.GetType()))\n\tdb := newTaskListDB(p.TaskManager, p.TaskList.GetDomainID(), domainName, p.TaskList.GetName(), p.TaskList.GetType(), int(p.TaskListKind), p.Logger)\n\tvar isolationGroups []string\n\tif p.TaskListKind != types.TaskListKindSticky && taskListConfig.EnableTasklistIsolation() {\n\t\tisolationGroups = slices.Clone(p.Cfg.AllIsolationGroups())\n\t\tslices.Sort(isolationGroups)\n\t}\n\n\ttlMgr := &taskListManagerImpl{\n\t\tcreateTime:      p.CreateTime,\n\t\tenableIsolation: taskListConfig.EnableTasklistIsolation(),\n\t\tdomainCache:     p.DomainCache,\n\t\tclusterMetadata: p.ClusterMetadata,\n\t\tisolationState:  p.IsolationState,\n\t\tisolationGroups: isolationGroups,\n\t\ttaskListID:      p.TaskList,\n\t\ttaskListKind:    p.TaskListKind,\n\t\tlogger:          p.Logger.WithTags(tag.WorkflowDomainName(domainName), tag.WorkflowTaskListName(p.TaskList.GetName()), tag.WorkflowTaskListType(p.TaskList.GetType())),\n\t\tdb:              db,\n\t\ttaskAckManager:  messaging.NewAckManager(p.Logger),\n\t\ttaskGC:          newTaskGC(db, taskListConfig),\n\t\tconfig:          taskListConfig,\n\t\tmatchingClient:  p.MatchingClient,\n\t\tdomainName:      domainName,\n\t\tscope:           scope,\n\t\ttimeSource:      p.TimeSource,\n\t\tregistry:        p.Registry,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(persistenceOperationRetryPolicy),\n\t\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t\t),\n\t\thistoryService: p.HistoryService,\n\t}\n\n\ttlMgr.pollers = poller.NewPollerManager(func() {\n\t\tscope.UpdateGauge(metrics.PollerPerTaskListCounter,\n\t\t\tfloat64(tlMgr.pollers.GetCount()))\n\t}, p.TimeSource)\n\n\tlivenessInterval := taskListConfig.IdleTasklistCheckInterval()\n\ttlMgr.liveness = liveness.NewLiveness(p.TimeSource, livenessInterval, func() {\n\t\ttlMgr.logger.Info(\"Task list manager stopping because no recent events\", tag.Dynamic(\"interval\", livenessInterval))\n\t\ttlMgr.Stop()\n\t})\n\n\tbaseEvent := event.E{\n\t\tTaskListName: p.TaskList.GetName(),\n\t\tTaskListKind: &p.TaskListKind,\n\t\tTaskListType: p.TaskList.GetType(),\n\t}\n\n\ttlMgr.qpsTracker = stats.NewEmaFixedWindowQPSTracker(p.TimeSource, 0.5, taskListConfig.QPSTrackerInterval(), baseEvent)\n\tif p.TaskList.IsRoot() && p.TaskListKind == types.TaskListKindNormal {\n\t\tadaptiveScalerScope := common.NewPerTaskListScope(domainName, p.TaskList.GetName(), p.TaskListKind, p.MetricsClient, metrics.MatchingAdaptiveScalerScope).\n\t\t\tTagged(getTaskListTypeTag(p.TaskList.GetType()))\n\t\ttlMgr.adaptiveScaler = NewAdaptiveScaler(p.TaskList, tlMgr, taskListConfig, p.TimeSource, tlMgr.logger, adaptiveScalerScope, p.MatchingClient, baseEvent)\n\t}\n\n\tvar fwdr Forwarder\n\tif tlMgr.isFowardingAllowed(p.TaskList, p.TaskListKind) {\n\t\tfwdr = newForwarder(&taskListConfig.ForwarderConfig, p.TaskList, p.TaskListKind, p.MatchingClient, scope)\n\t}\n\tnumReadPartitionsFn := func() int {\n\t\tif taskListConfig.EnableGetNumberOfPartitionsFromCache() {\n\t\t\tpartitionConfig := tlMgr.TaskListPartitionConfig()\n\t\t\tr := 1\n\t\t\tif partitionConfig != nil {\n\t\t\t\tr = len(partitionConfig.ReadPartitions)\n\t\t\t}\n\t\t\treturn r\n\t\t}\n\t\treturn taskListConfig.NumReadPartitions()\n\t}\n\ttlMgr.limiter = newTaskListLimiter(p.TimeSource, tlMgr.scope, taskListConfig, numReadPartitionsFn)\n\ttlMgr.matcher = newTaskMatcher(taskListConfig, fwdr, tlMgr.scope, isolationGroups, tlMgr.logger, p.TaskList, p.TaskListKind, tlMgr.limiter).(*taskMatcherImpl)\n\ttlMgr.taskWriter = newTaskWriter(tlMgr)\n\ttlMgr.taskReader = newTaskReader(tlMgr, isolationGroups)\n\ttlMgr.taskCompleter = newTaskCompleter(tlMgr, historyServiceOperationRetryPolicy)\n\ttlMgr.startWG.Add(1)\n\treturn tlMgr, nil\n}\n\n// Starts reading pump for the given task list.\n// The pump fills up taskBuffer from persistence.\nfunc (c *taskListManagerImpl) Start(ctx context.Context) error {\n\tdefer c.startWG.Done()\n\n\tif !c.taskListID.IsRoot() && c.taskListKind == types.TaskListKindNormal {\n\t\tvar info *persistence.TaskListInfo\n\t\terr := c.throttleRetry.Do(context.Background(), func(ctx context.Context) error {\n\t\t\tvar err error\n\t\t\tinfo, err = c.db.GetTaskListInfo(c.taskListID.GetRoot())\n\t\t\treturn err\n\t\t})\n\t\tif err != nil {\n\t\t\t// This is an edge case, and only occur before we fully migrate partition config to database for a tasklist.\n\t\t\t// Currently, if a task list is configured with multiple partitions in dynamicconfig before the creation of the task list,\n\t\t\t// a non-root partition can receive a request before the root partition and when it task list manager tries to read partition config from the root partition it will get this error.\n\t\t\t// For example, if in the dynamicconfig we set all task list to have 2 partitions by default, all the non-root partitions of newly created task lists will get this error.\n\t\t\t// This will not happen once we fully migrate partition config to database. Because in that case, root partition will always be created before non-root partitions.\n\t\t\tvar e *types.EntityNotExistsError\n\t\t\tif !errors.As(err, &e) {\n\t\t\t\tc.Stop()\n\t\t\t\treturn err\n\t\t\t}\n\t\t} else {\n\t\t\tc.partitionConfig = info.AdaptivePartitionConfig.ToInternalType()\n\t\t}\n\t}\n\tif err := c.taskWriter.Start(); err != nil {\n\t\tc.Stop()\n\t\treturn err\n\t}\n\tif c.taskListID.IsRoot() && c.taskListKind == types.TaskListKindNormal {\n\t\tc.partitionConfig = c.db.PartitionConfig().ToInternalType()\n\t\tc.logger.Info(\"get task list partition config from db\", tag.Dynamic(\"root-partition\", c.taskListID.GetRoot()), tag.Dynamic(\"task-list-partition-config\", c.partitionConfig))\n\t\tif c.partitionConfig != nil {\n\t\t\tstartConfig := c.partitionConfig\n\t\t\t// push update notification to all non-root partitions on start\n\t\t\tc.stopWG.Add(1)\n\t\t\tgo func() {\n\t\t\t\tdefer c.stopWG.Done()\n\t\t\t\tctx, cancel := context.WithTimeout(context.Background(), notifyPartitionConfigTimeout)\n\t\t\t\tdefer cancel()\n\t\t\t\tc.notifyPartitionConfig(ctx, nil, startConfig)\n\t\t\t}()\n\t\t}\n\t}\n\tc.liveness.Start()\n\tc.taskReader.Start()\n\tc.qpsTracker.Start()\n\tif c.adaptiveScaler != nil {\n\t\tc.adaptiveScaler.Start()\n\t}\n\n\treturn nil\n}\n\n// Stop stops task list manager and calls Stop on all background child objects\nfunc (c *taskListManagerImpl) Stop() {\n\tc.stoppedLock.Lock()\n\tdefer c.stoppedLock.Unlock()\n\tif !atomic.CompareAndSwapInt32(&c.stopped, 0, 1) {\n\t\treturn\n\t}\n\n\t// Notify parent registry to unregister this manager\n\tc.registry.Unregister(c)\n\n\tif c.adaptiveScaler != nil {\n\t\tc.adaptiveScaler.Stop()\n\t}\n\tc.qpsTracker.Stop()\n\tc.liveness.Stop()\n\tc.taskWriter.Stop()\n\tc.taskReader.Stop()\n\tc.matcher.DisconnectBlockedPollers()\n\tc.stopWG.Wait()\n\tc.logger.Info(\"Task list manager state changed\", tag.LifeCycleStopped)\n}\n\nfunc (c *taskListManagerImpl) handleErr(err error) error {\n\tvar e *persistence.ConditionFailedError\n\tif errors.As(err, &e) {\n\t\t// This indicates the task list may have moved to another host.\n\t\tc.scope.IncCounter(metrics.ConditionFailedErrorPerTaskListCounter)\n\t\tc.logger.Info(\"Stopping task list due to persistence condition failure.\", tag.Error(err))\n\t\tc.Stop()\n\t\tif c.taskListKind == types.TaskListKindSticky {\n\t\t\t// TODO: we don't see this error in our logs, we might be able to remove this error\n\t\t\terr = &types.InternalServiceError{Message: constants.StickyTaskConditionFailedErrorMsg}\n\t\t}\n\t}\n\treturn err\n}\n\nfunc (c *taskListManagerImpl) TaskListPartitionConfig() *types.TaskListPartitionConfig {\n\tc.partitionConfigLock.RLock()\n\tdefer c.partitionConfigLock.RUnlock()\n\n\tscope := c.scope.Tagged(metrics.TaskListRootPartitionTag(c.taskListID.GetRoot()))\n\tif c.partitionConfig == nil {\n\t\t// if partition config is nil, read/write partition count is considered 1. Emit those metrics for continuity\n\t\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumReadGauge, 1)\n\t\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumWriteGauge, 1)\n\t\treturn nil\n\t}\n\n\tconfig := *c.partitionConfig\n\tc.logger.Debug(\"current partition config\", tag.Dynamic(\"root-partition\", c.taskListID.GetRoot()), tag.Dynamic(\"config\", config))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumReadGauge, float64(len(config.ReadPartitions)))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigNumWriteGauge, float64(len(config.WritePartitions)))\n\tscope.UpdateGauge(metrics.TaskListPartitionConfigVersionGauge, float64(config.Version))\n\treturn &config\n}\n\nfunc (c *taskListManagerImpl) PartitionReadConfig() (*types.TaskListPartition, bool) {\n\ttlConfig := c.TaskListPartitionConfig()\n\t// no config indicates 1 partition\n\tif tlConfig == nil {\n\t\treturn defaultPartition, true\n\t}\n\tpartition, ok := tlConfig.ReadPartitions[c.taskListID.Partition()]\n\treturn partition, ok\n}\n\nfunc (c *taskListManagerImpl) PartitionWriteConfig() (*types.TaskListPartition, bool) {\n\ttlConfig := c.TaskListPartitionConfig()\n\t// no config indicates 1 partition\n\tif tlConfig == nil {\n\t\treturn defaultPartition, true\n\t}\n\tpartition, ok := tlConfig.WritePartitions[c.taskListID.Partition()]\n\treturn partition, ok\n}\n\nfunc (c *taskListManagerImpl) LoadBalancerHints() *types.LoadBalancerHints {\n\tc.startWG.Wait()\n\tif c.timeSource.Now().Sub(c.createTime) < time.Second*10 {\n\t\treturn nil\n\t}\n\treturn &types.LoadBalancerHints{\n\t\tBacklogCount:  c.taskAckManager.GetBacklogCount(),\n\t\tRatePerSecond: c.qpsTracker.QPS(),\n\t}\n}\n\nfunc (c *taskListManagerImpl) QueriesPerSecond() float64 {\n\treturn c.qpsTracker.QPS()\n}\n\nfunc isTaskListPartitionConfigEqual(a types.TaskListPartitionConfig, b types.TaskListPartitionConfig) bool {\n\ta.Version = b.Version\n\treturn reflect.DeepEqual(a, b)\n}\n\nfunc (c *taskListManagerImpl) RefreshTaskListPartitionConfig(ctx context.Context, config *types.TaskListPartitionConfig) error {\n\tc.startWG.Wait()\n\tif config == nil {\n\t\t// if config is nil, we'll reload it from database\n\t\tvar info *persistence.TaskListInfo\n\t\terr := c.throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\t\tvar err error\n\t\t\tinfo, err = c.db.GetTaskListInfo(c.taskListID.GetRoot())\n\t\t\treturn err\n\t\t})\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tconfig = info.AdaptivePartitionConfig.ToInternalType()\n\t\tc.partitionConfigLock.Lock()\n\t\tc.partitionConfig = config\n\t\tc.partitionConfigLock.Unlock()\n\t\treturn nil\n\t}\n\tc.partitionConfigLock.Lock()\n\tdefer c.partitionConfigLock.Unlock()\n\tif c.partitionConfig == nil || c.partitionConfig.Version < config.Version {\n\t\tc.partitionConfig = config\n\t}\n\treturn nil\n}\n\n// UpdateTaskListPartitionConfig updates the task list partition config. It is called by adaptive scaler component on the root partition.\n// Root tasklist manager will update the partition config in the database and notify all non-root partitions.\nfunc (c *taskListManagerImpl) UpdateTaskListPartitionConfig(ctx context.Context, config *types.TaskListPartitionConfig) error {\n\tc.startWG.Wait()\n\toldConfig, newConfig, err := c.updatePartitionConfig(ctx, config)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif newConfig != nil {\n\t\t// push update notification to all non-root partitions\n\t\tc.notifyPartitionConfig(ctx, oldConfig, newConfig)\n\t}\n\treturn nil\n}\n\nfunc (c *taskListManagerImpl) updatePartitionConfig(ctx context.Context, newConfig *types.TaskListPartitionConfig) (*types.TaskListPartitionConfig, *types.TaskListPartitionConfig, error) {\n\terr := validatePartitionConfig(newConfig)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tvar version int64\n\tc.partitionConfigLock.Lock()\n\tdefer c.partitionConfigLock.Unlock()\n\toldConfig := c.partitionConfig\n\tif oldConfig != nil {\n\t\tif isTaskListPartitionConfigEqual(*oldConfig, *newConfig) {\n\t\t\treturn nil, nil, nil\n\t\t}\n\t\tversion = oldConfig.Version\n\t} else {\n\t\tif len(newConfig.ReadPartitions) == 1 && len(newConfig.WritePartitions) == 1 {\n\t\t\treturn nil, nil, nil\n\t\t}\n\t}\n\terr = c.throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\treturn c.db.UpdateTaskListPartitionConfig(toPersistenceConfig(version+1, newConfig))\n\t})\n\tif err != nil {\n\t\t// We're not sure whether the update was persisted or not,\n\t\t// Stop the tasklist manager and let it be reloaded\n\t\tc.scope.IncCounter(metrics.TaskListPartitionUpdateFailedCounter)\n\t\tc.Stop()\n\t\treturn nil, nil, err\n\t}\n\tc.partitionConfig = c.db.PartitionConfig().ToInternalType()\n\treturn oldConfig, c.partitionConfig, nil\n}\n\nfunc (c *taskListManagerImpl) notifyPartitionConfig(ctx context.Context, oldConfig, newConfig *types.TaskListPartitionConfig) {\n\ttaskListType := types.TaskListTypeDecision.Ptr()\n\tif c.taskListID.GetType() == persistence.TaskListTypeActivity {\n\t\ttaskListType = types.TaskListTypeActivity.Ptr()\n\t}\n\ttoNotify := make(map[int]any)\n\tif oldConfig != nil {\n\t\tfor id := range oldConfig.ReadPartitions {\n\t\t\tif id != 0 {\n\t\t\t\ttoNotify[id] = true\n\t\t\t}\n\t\t}\n\t}\n\tfor id := range newConfig.ReadPartitions {\n\t\tif id != 0 {\n\t\t\ttoNotify[id] = true\n\t\t}\n\t}\n\tg := &errgroup.Group{}\n\tfor p := range toNotify {\n\t\ttaskListName := c.taskListID.GetPartition(p)\n\t\tg.Go(func() (e error) {\n\t\t\tdefer func() { log.CapturePanic(recover(), c.logger, &e) }()\n\n\t\t\t_, e = c.matchingClient.RefreshTaskListPartitionConfig(ctx, &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\tDomainUUID:      c.taskListID.GetDomainID(),\n\t\t\t\tTaskList:        &types.TaskList{Name: taskListName, Kind: &c.taskListKind},\n\t\t\t\tTaskListType:    taskListType,\n\t\t\t\tPartitionConfig: newConfig,\n\t\t\t})\n\t\t\tif e != nil {\n\t\t\t\tc.logger.Error(\"failed to notify partition\", tag.Error(e), tag.Dynamic(\"task-list-partition-name\", taskListName))\n\t\t\t}\n\t\t\treturn e\n\t\t})\n\t}\n\terr := g.Wait()\n\tif err != nil {\n\t\tc.logger.Error(\"failed to notify all partitions\", tag.Error(err))\n\t}\n}\n\n// AddTask adds a task to the task list. This method will first attempt a synchronous\n// match with a poller. When there are no pollers or if rate limit is exceeded, task will\n// be written to database and later asynchronously matched with a poller.\n// It returns whether the sync match succeeded, along with any error that occurred.\nfunc (c *taskListManagerImpl) AddTask(ctx context.Context, params AddTaskParams) (bool, error) {\n\tc.startWG.Wait()\n\n\tif c.shouldReload() {\n\t\tc.Stop()\n\t\treturn false, errShutdown\n\t}\n\tif c.config.EnableGetNumberOfPartitionsFromCache() {\n\t\t_, ok := c.PartitionWriteConfig()\n\t\tif !ok {\n\t\t\treturn false, &types.ReadOnlyPartitionError{Message: \"Current partition is drained.\"}\n\t\t}\n\t}\n\tif params.ForwardedFrom == \"\" {\n\t\t// request sent by history service\n\t\tc.liveness.MarkAlive()\n\t\tif isolationGroup, ok := params.TaskInfo.PartitionConfig[isolationgroup.GroupKey]; ok {\n\t\t\tc.qpsTracker.ReportGroup(isolationGroup, 1)\n\t\t} else {\n\t\t\tc.qpsTracker.ReportCounter(1)\n\t\t}\n\t\tc.scope.UpdateGauge(metrics.EstimatedAddTaskQPSGauge, c.qpsTracker.QPS())\n\t}\n\n\t// Sync match flow\n\tvar syncMatch bool\n\te := event.E{\n\t\tTaskListName: c.taskListID.GetName(),\n\t\tTaskListKind: &c.taskListKind,\n\t\tTaskListType: c.taskListID.GetType(),\n\t\tTaskInfo:     *params.TaskInfo,\n\t}\n\n\tdomainEntry, err := c.domainCache.GetDomainByID(params.TaskInfo.DomainID)\n\tif err != nil {\n\t\t// If we cannot fetch the domain entry from the cache, we cannot proceed. Sync match fails.\n\t\treturn false, err\n\t}\n\n\t// Check if the task was forwarded from another partition\n\tisForwarded := params.ForwardedFrom != \"\"\n\tif !domainEntry.IsActiveIn(c.clusterMetadata.GetCurrentClusterName()) {\n\t\t// standby task, only persist when task is not forwarded from child partition\n\t\tsyncMatch = false\n\t\tif isForwarded {\n\t\t\treturn syncMatch, errRemoteSyncMatchFailed\n\t\t}\n\n\t\t// Persist the standby task, but the sync match still fails.\n\t\t// Return the false syncMatch flag along with any error\n\t\t_, err = c.taskWriter.appendTask(ctx, params.TaskInfo)\n\t\tif err == nil {\n\t\t\t// Signal the task reader only if appendTask succeeded\n\t\t\tc.taskReader.Signal()\n\t\t}\n\t\treturn syncMatch, err\n\t}\n\n\tisolationGroup, _ := c.getIsolationGroupForTask(ctx, params.TaskInfo)\n\t// active task, try sync match first\n\tsyncMatch, err = c.trySyncMatch(ctx, params, isolationGroup)\n\tif syncMatch {\n\t\te.EventName = \"SyncMatched so not persisted\"\n\t\tevent.Log(e)\n\t\treturn syncMatch, err\n\t}\n\tif params.ActivityTaskDispatchInfo != nil {\n\t\treturn syncMatch, errRemoteSyncMatchFailed\n\t}\n\n\tif isForwarded {\n\t\t// forwarded from child partition - only do sync match\n\t\t// child partition will persist the task when sync match fails\n\t\te.EventName = \"Could not SyncMatched Forwarded Task so not persisted\"\n\t\tevent.Log(e)\n\t\treturn syncMatch, errRemoteSyncMatchFailed\n\t}\n\n\te.EventName = \"Task Sent to Writer\"\n\tevent.Log(e)\n\tif _, err := c.taskWriter.appendTask(ctx, params.TaskInfo); err != nil {\n\t\treturn syncMatch, err\n\t}\n\tc.taskReader.Signal()\n\treturn syncMatch, nil\n}\n\n// DispatchTask dispatches a task to a poller on the active side. When there are no pollers to pick\n// up the task or if the rate limit is exceeded, this method will return error. Task\n// *will not* be persisted to db. On the passive side, dispatches the task to the taskCompleter; it will attempt\n// to complete the task if it has already been started.\nfunc (c *taskListManagerImpl) DispatchTask(ctx context.Context, task *InternalTask) error {\n\t// check if this is the active cluster for the domain\n\tdomainEntry, err := c.domainCache.GetDomainByID(task.Event.TaskInfo.DomainID)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to fetch domain from cache: %w\", err)\n\t}\n\n\tif domainEntry.IsActiveIn(c.clusterMetadata.GetCurrentClusterName()) {\n\t\tc.logger.Debug(\"Domain is active in the current cluster, dispatching task\",\n\t\t\ttag.WorkflowDomainID(task.Event.TaskInfo.DomainID),\n\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\ttag.WorkflowID(task.Event.TaskInfo.WorkflowID),\n\t\t\ttag.WorkflowRunID(task.Event.TaskInfo.RunID),\n\t\t)\n\t\treturn c.matcher.MustOffer(ctx, task)\n\t}\n\n\t// optional configuration to enable cleanup of tasks, in the standby cluster, that have already been started\n\tif c.config.EnableStandbyTaskCompletion() && !domainEntry.GetReplicationConfig().IsActiveActive() {\n\t\tif err := c.taskCompleter.CompleteTaskIfStarted(ctx, task); err != nil {\n\t\t\tif errors.Is(err, errDomainIsActive) {\n\t\t\t\treturn c.matcher.MustOffer(ctx, task)\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\treturn nil\n\t}\n\n\treturn c.matcher.MustOffer(ctx, task)\n}\n\n// DispatchQueryTask will dispatch query to local or remote poller. If forwarded then result or error is returned,\n// if dispatched to local poller then nil and nil is returned.\nfunc (c *taskListManagerImpl) DispatchQueryTask(\n\tctx context.Context,\n\ttaskID string,\n\trequest *types.MatchingQueryWorkflowRequest,\n) (*types.MatchingQueryWorkflowResponse, error) {\n\tc.startWG.Wait()\n\ttask := newInternalQueryTask(taskID, request)\n\treturn c.matcher.OfferQuery(ctx, task)\n}\n\n// GetTask blocks waiting for a task.\n// Returns error when context deadline is exceeded\n// maxDispatchPerSecond is the max rate at which tasks are allowed\n// to be dispatched from this task list to pollers\nfunc (c *taskListManagerImpl) GetTask(\n\tctx context.Context,\n\tmaxDispatchPerSecond *float64,\n) (*InternalTask, error) {\n\tif c.shouldReload() {\n\t\tc.Stop()\n\t\treturn nil, ErrNoTasks\n\t}\n\tc.liveness.MarkAlive()\n\t// TODO: consider return early if QPS and backlog count are both 0,\n\t// since there is no task to be returned\n\ttask, err := c.getTask(ctx, maxDispatchPerSecond)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"couldn't get task: %w\", err)\n\t}\n\ttask.domainName = c.domainName\n\ttask.BacklogCountHint = c.taskAckManager.GetBacklogCount()\n\treturn task, nil\n}\n\nfunc (c *taskListManagerImpl) getTask(ctx context.Context, maxDispatchPerSecond *float64) (*InternalTask, error) {\n\tc.emitMisconfiguredPartitionMetrics()\n\t// We need to set a shorter timeout than the original ctx; otherwise, by the time ctx deadline is\n\t// reached, instead of emptyTask, context timeout error is returned to the frontend by the rpc stack,\n\t// which counts against our SLO. By shortening the timeout by a very small amount, the emptyTask can be\n\t// returned to the handler before a context timeout error is generated.\n\tchildCtx, cancel := c.newChildContext(ctx, c.config.LongPollExpirationInterval(), returnEmptyTaskTimeBudget)\n\tdefer cancel()\n\n\tisolationGroup := IsolationGroupFromContext(ctx)\n\tpollerID := PollerIDFromContext(ctx)\n\tidentity := IdentityFromContext(ctx)\n\trps := -1.0\n\trpsOverride := c.config.OverrideTaskListRPS()\n\tif rpsOverride > 0 {\n\t\trps = rpsOverride\n\t} else if maxDispatchPerSecond != nil {\n\t\trps = *maxDispatchPerSecond\n\t}\n\tif rps >= 0 {\n\t\tc.limiter.ReportLimit(rps)\n\t} else {\n\t\trps = c.config.TaskDispatchRPS\n\t}\n\tc.pollers.StartPoll(pollerID, cancel, &poller.Info{\n\t\tIdentity:       identity,\n\t\tIsolationGroup: isolationGroup,\n\t\tRatePerSecond:  rps,\n\t})\n\tdefer c.pollers.EndPoll(pollerID)\n\n\tdomainEntry, err := c.domainCache.GetDomainByID(c.taskListID.GetDomainID())\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"unable to fetch domain from cache: %w\", err)\n\t}\n\n\tif !domainEntry.IsActiveIn(c.clusterMetadata.GetCurrentClusterName()) {\n\t\treturn c.matcher.PollForQuery(childCtx)\n\t}\n\n\tif c.isIsolationMatcherEnabled() {\n\t\treturn c.matcher.Poll(childCtx, isolationGroup)\n\t}\n\treturn c.matcher.Poll(childCtx, \"\")\n}\n\n// GetAllPollerInfo returns all pollers that polled from this tasklist in last few minutes\nfunc (c *taskListManagerImpl) GetAllPollerInfo() []*types.PollerInfo {\n\treturn c.pollers.ListInfo()\n}\n\n// HasPollerAfter checks if there is any poller after a timestamp\nfunc (c *taskListManagerImpl) HasPollerAfter(accessTime time.Time) bool {\n\treturn c.pollers.HasPollerAfter(accessTime)\n}\n\nfunc (c *taskListManagerImpl) CancelPoller(pollerID string) {\n\tif c.pollers.CancelPoll(pollerID) {\n\t\tc.logger.Info(\"canceled outstanding poller\", tag.WorkflowDomainName(c.domainName))\n\t}\n}\n\n// DescribeTaskList returns information about the target tasklist, right now this API returns the\n// pollers which polled this tasklist in last few minutes and status of tasklist's ackManager\n// (readLevel, ackLevel, backlogCountHint and taskIDBlock).\nfunc (c *taskListManagerImpl) DescribeTaskList(includeTaskListStatus bool) *types.DescribeTaskListResponse {\n\tresponse := &types.DescribeTaskListResponse{\n\t\tPollers: c.GetAllPollerInfo(),\n\t\tTaskList: &types.TaskList{\n\t\t\tName: c.taskListID.GetName(),\n\t\t\tKind: &c.taskListKind,\n\t\t},\n\t}\n\tresponse.PartitionConfig = c.TaskListPartitionConfig()\n\tif !includeTaskListStatus {\n\t\treturn response\n\t}\n\n\tidBlock := rangeIDToTaskIDBlock(c.db.RangeID(), c.config.RangeSize)\n\tisolationGroups := c.config.AllIsolationGroups()\n\tpollerCounts := c.pollers.GetCountByIsolationGroup(c.timeSource.Now().Add(-1 * c.config.TaskIsolationPollerWindow()))\n\tisolationGroupMetrics := make(map[string]*types.IsolationGroupMetrics, len(isolationGroups))\n\tfor _, group := range isolationGroups {\n\t\tisolationGroupMetrics[group] = &types.IsolationGroupMetrics{\n\t\t\tNewTasksPerSecond: c.qpsTracker.GroupQPS(group),\n\t\t\tPollerCount:       int64(pollerCounts[group]),\n\t\t}\n\t}\n\tresponse.TaskListStatus = &types.TaskListStatus{\n\t\tReadLevel:        c.taskAckManager.GetReadLevel(),\n\t\tAckLevel:         c.taskAckManager.GetAckLevel(),\n\t\tBacklogCountHint: c.taskAckManager.GetBacklogCount(),\n\t\tRatePerSecond:    float64(c.limiter.Limit()),\n\t\tTaskIDBlock: &types.TaskIDBlock{\n\t\t\tStartID: idBlock.start,\n\t\t\tEndID:   idBlock.end,\n\t\t},\n\t\tIsolationGroupMetrics: isolationGroupMetrics,\n\t\tNewTasksPerSecond:     c.qpsTracker.QPS(),\n\t\tEmpty:                 c.taskAckManager.GetAckLevel() == c.taskWriter.GetMaxReadLevel(),\n\t}\n\n\treturn response\n}\n\nfunc (c *taskListManagerImpl) ReleaseBlockedPollers() error {\n\tc.stoppedLock.RLock()\n\tdefer c.stoppedLock.RUnlock()\n\n\tif atomic.LoadInt32(&c.stopped) == 1 {\n\t\tc.logger.Info(\"Task list manager is already stopped\")\n\t\treturn errShutdown\n\t}\n\n\tc.matcher.DisconnectBlockedPollers()\n\tc.matcher.RefreshCancelContext()\n\n\treturn nil\n}\n\nfunc (c *taskListManagerImpl) String() string {\n\tbuf := new(bytes.Buffer)\n\tif c.taskListID.GetType() == persistence.TaskListTypeActivity {\n\t\tbuf.WriteString(\"Activity\")\n\t} else {\n\t\tbuf.WriteString(\"Decision\")\n\t}\n\trangeID := c.db.RangeID()\n\tfmt.Fprintf(buf, \" task list %v\\n\", c.taskListID.GetName())\n\tfmt.Fprintf(buf, \"RangeID=%v\\n\", rangeID)\n\tfmt.Fprintf(buf, \"TaskIDBlock=%+v\\n\", rangeIDToTaskIDBlock(rangeID, c.config.RangeSize))\n\tfmt.Fprintf(buf, \"AckLevel=%v\\n\", c.taskAckManager.GetAckLevel())\n\tfmt.Fprintf(buf, \"MaxReadLevel=%v\\n\", c.taskAckManager.GetReadLevel())\n\n\treturn buf.String()\n}\n\nfunc (c *taskListManagerImpl) GetTaskListKind() types.TaskListKind {\n\treturn c.taskListKind\n}\n\nfunc (c *taskListManagerImpl) TaskListID() *Identifier {\n\treturn c.taskListID\n}\n\nfunc (c *taskListManagerImpl) trySyncMatch(ctx context.Context, params AddTaskParams, isolationGroup string) (bool, error) {\n\ttask := newInternalTask(params.TaskInfo, nil, params.Source, params.ForwardedFrom, true, params.ActivityTaskDispatchInfo, isolationGroup)\n\tchildCtx := ctx\n\tcancel := func() {}\n\twaitTime := maxSyncMatchWaitTime\n\tif params.ActivityTaskDispatchInfo != nil {\n\t\twaitTime = c.config.ActivityTaskSyncMatchWaitTime(params.ActivityTaskDispatchInfo.WorkflowDomain)\n\t}\n\tif !task.IsForwarded() {\n\t\t// when task is forwarded from another matching host, we trust the context as is\n\t\t// otherwise, we override to limit the amount of time we can block on sync match\n\t\tchildCtx, cancel = c.newChildContext(ctx, waitTime, time.Second)\n\t}\n\tvar matched bool\n\tvar err error\n\tif params.ActivityTaskDispatchInfo != nil {\n\t\tmatched, err = c.matcher.OfferOrTimeout(childCtx, c.timeSource.Now(), task)\n\t} else {\n\t\tmatched, err = c.matcher.Offer(childCtx, task)\n\t}\n\tcancel()\n\treturn matched, err\n}\n\n// newChildContext creates a child context with desired timeout.\n// if tailroom is non-zero, then child context timeout will be\n// the minOf(parentCtx.Deadline()-tailroom, timeout). Use this\n// method to create child context when childContext cannot use\n// all of parent's deadline but instead there is a need to leave\n// some time for parent to do some post-work\nfunc (c *taskListManagerImpl) newChildContext(\n\tparent context.Context,\n\ttimeout time.Duration,\n\ttailroom time.Duration,\n) (context.Context, context.CancelFunc) {\n\tselect {\n\tcase <-parent.Done():\n\t\treturn parent, func() {}\n\tdefault:\n\t}\n\tdeadline, ok := parent.Deadline()\n\tif !ok {\n\t\treturn context.WithTimeout(parent, timeout)\n\t}\n\tremaining := time.Until(deadline) - tailroom\n\tif remaining < timeout {\n\t\ttimeout = time.Duration(max(0, int64(remaining)))\n\t}\n\treturn context.WithTimeout(parent, timeout)\n}\n\nfunc (c *taskListManagerImpl) isFowardingAllowed(taskList *Identifier, kind types.TaskListKind) bool {\n\treturn !taskList.IsRoot() && kind != types.TaskListKindSticky\n}\n\nfunc (c *taskListManagerImpl) isIsolationMatcherEnabled() bool {\n\treturn c.taskListKind != types.TaskListKindSticky && c.enableIsolation\n}\n\nfunc (c *taskListManagerImpl) shouldReload() bool {\n\treturn c.config.EnableTasklistIsolation() != c.enableIsolation\n}\n\nfunc (c *taskListManagerImpl) getIsolationGroupForTask(ctx context.Context, taskInfo *persistence.TaskInfo) (string, time.Duration) {\n\tif !c.enableIsolation || len(taskInfo.PartitionConfig) == 0 || c.taskListKind == types.TaskListKindSticky {\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\tgroup := taskInfo.PartitionConfig[isolationgroup.GroupKey]\n\n\tif group == defaultTaskBufferIsolationGroup {\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\n\tisDrained, err := c.isolationState.IsDrained(ctx, c.domainName, group)\n\tif err != nil {\n\t\t// if we're unable to get the isolation group, log the error and fallback to no isolation\n\t\tc.logger.Error(\"Failed to determine whether isolation group is drained\", tag.IsolationGroup(group), tag.WorkflowID(taskInfo.WorkflowID), tag.WorkflowRunID(taskInfo.RunID), tag.TaskID(taskInfo.TaskID), tag.Error(err))\n\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCauseError).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\tif isDrained {\n\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCauseGroupDrained).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\tif _, ok := slices.BinarySearch(c.isolationGroups, group); !ok {\n\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCauseGroupUnknown).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\tif !c.pollers.HasPollerFromIsolationGroupAfter(group, c.timeSource.Now().Add(-1*c.config.TaskIsolationPollerWindow())) {\n\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCauseNoRecentPollers).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\tpartition, ok := c.PartitionReadConfig()\n\tif !ok || (len(partition.IsolationGroups) != 0 && !slices.Contains(partition.IsolationGroups, group)) {\n\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCausePartitionChange).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t}\n\n\ttotalTaskIsolationDuration := c.config.TaskIsolationDuration()\n\ttaskIsolationDuration := noIsolationTimeout\n\tif totalTaskIsolationDuration != noIsolationTimeout {\n\t\ttaskLatency := c.timeSource.Now().Sub(taskInfo.CreatedTime)\n\t\tif taskLatency < (totalTaskIsolationDuration - minimumIsolationDuration) {\n\t\t\ttaskIsolationDuration = totalTaskIsolationDuration - taskLatency\n\t\t} else {\n\t\t\tc.logger.Debug(\"Leaking task due to taskIsolationDuration expired\", tag.IsolationGroup(group), tag.IsolationDuration(taskIsolationDuration), tag.TaskLatency(taskLatency))\n\t\t\tc.scope.Tagged(metrics.IsolationGroupTag(group), IsolationLeakCauseExpired).IncCounter(metrics.TaskIsolationLeakPerTaskList)\n\t\t\treturn defaultTaskBufferIsolationGroup, noIsolationTimeout\n\t\t}\n\t}\n\treturn group, taskIsolationDuration\n\n}\n\nfunc (c *taskListManagerImpl) emitMisconfiguredPartitionMetrics() {\n\tif !c.taskListID.IsRoot() || c.taskListKind == types.TaskListKindSticky {\n\t\t// only emit the metric in root partition of non-sticky tasklist\n\t\treturn\n\t}\n\tif c.config.NumReadPartitions() != c.config.NumWritePartitions() {\n\t\tc.scope.UpdateGauge(metrics.TaskListReadWritePartitionMismatchGauge, 1)\n\t}\n\tpollerCount := c.pollers.GetCount()\n\tif pollerCount < c.config.NumReadPartitions() || pollerCount < c.config.NumWritePartitions() {\n\t\tc.scope.Tagged(metrics.IsolationEnabledTag(c.enableIsolation)).UpdateGauge(metrics.TaskListPollerPartitionMismatchGauge, 1)\n\t}\n}\n\nfunc getTaskListTypeTag(taskListType int) metrics.Tag {\n\tswitch taskListType {\n\tcase persistence.TaskListTypeActivity:\n\t\treturn taskListActivityTypeTag\n\tcase persistence.TaskListTypeDecision:\n\t\treturn taskListDecisionTypeTag\n\tdefault:\n\t\treturn metrics.TaskListTypeTag(\"\")\n\t}\n}\n\nfunc createServiceBusyError(msg string) *types.ServiceBusyError {\n\treturn &types.ServiceBusyError{Message: msg}\n}\n\nfunc rangeIDToTaskIDBlock(rangeID, rangeSize int64) taskIDBlock {\n\treturn taskIDBlock{\n\t\tstart: (rangeID-1)*rangeSize + 1,\n\t\tend:   rangeID * rangeSize,\n\t}\n}\n\nfunc toPersistenceConfig(version int64, config *types.TaskListPartitionConfig) *persistence.TaskListPartitionConfig {\n\tread := make(map[int]*persistence.TaskListPartition, len(config.ReadPartitions))\n\twrite := make(map[int]*persistence.TaskListPartition, len(config.WritePartitions))\n\tfor id, p := range config.ReadPartitions {\n\t\tread[id] = &persistence.TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\tfor id, p := range config.WritePartitions {\n\t\twrite[id] = &persistence.TaskListPartition{IsolationGroups: p.IsolationGroups}\n\t}\n\treturn &persistence.TaskListPartitionConfig{\n\t\tVersion:         version,\n\t\tReadPartitions:  read,\n\t\tWritePartitions: write,\n\t}\n}\n\nfunc validatePartitionConfig(config *types.TaskListPartitionConfig) error {\n\tif len(config.ReadPartitions) < 1 {\n\t\treturn &types.BadRequestError{Message: \"read partitions < 1\"}\n\t}\n\tif len(config.WritePartitions) < 1 {\n\t\treturn &types.BadRequestError{Message: \"write partitions < 1\"}\n\t}\n\tif len(config.ReadPartitions) < len(config.WritePartitions) {\n\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"read partitions (%d) < write partitions (%d)\", len(config.ReadPartitions), len(config.WritePartitions))}\n\t}\n\tif _, ok := config.ReadPartitions[0]; !ok {\n\t\treturn &types.BadRequestError{Message: \"the root partition must always be in read partitions\"}\n\t}\n\tif _, ok := config.WritePartitions[0]; !ok {\n\t\treturn &types.BadRequestError{Message: \"the root partition must always be in write partitions\"}\n\t}\n\tfor id := range config.WritePartitions {\n\t\tif _, ok := config.ReadPartitions[id]; !ok {\n\t\t\treturn &types.BadRequestError{Message: fmt.Sprintf(\"partition %d included in write but not read\", id)}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc newTaskListConfig(id *Identifier, cfg *config.Config, domainName string) *config.TaskListConfig {\n\ttaskListName := id.GetName()\n\ttaskType := id.GetType()\n\treturn &config.TaskListConfig{\n\t\tRangeSize:          cfg.RangeSize,\n\t\tReadRangeSize:      cfg.ReadRangeSize,\n\t\tAllIsolationGroups: cfg.AllIsolationGroups,\n\t\tEnableTasklistIsolation: func() bool {\n\t\t\treturn cfg.EnableTasklistIsolation(domainName)\n\t\t},\n\t\tActivityTaskSyncMatchWaitTime: cfg.ActivityTaskSyncMatchWaitTime,\n\t\tGetTasksBatchSize: func() int {\n\t\t\treturn cfg.GetTasksBatchSize(domainName, taskListName, taskType)\n\t\t},\n\t\tUpdateAckInterval: func() time.Duration {\n\t\t\treturn cfg.UpdateAckInterval(domainName, taskListName, taskType)\n\t\t},\n\t\tIdleTasklistCheckInterval: func() time.Duration {\n\t\t\treturn cfg.IdleTasklistCheckInterval(domainName, taskListName, taskType)\n\t\t},\n\t\tMaxTasklistIdleTime: func() time.Duration {\n\t\t\treturn cfg.MaxTasklistIdleTime(domainName, taskListName, taskType)\n\t\t},\n\t\tMinTaskThrottlingBurstSize: func() int {\n\t\t\treturn cfg.MinTaskThrottlingBurstSize(domainName, taskListName, taskType)\n\t\t},\n\t\tEnableSyncMatch: func() bool {\n\t\t\treturn cfg.EnableSyncMatch(domainName, taskListName, taskType)\n\t\t},\n\t\tLongPollExpirationInterval: func() time.Duration {\n\t\t\treturn cfg.LongPollExpirationInterval(domainName, taskListName, taskType)\n\t\t},\n\t\tMaxTaskDeleteBatchSize: func() int {\n\t\t\treturn cfg.MaxTaskDeleteBatchSize(domainName, taskListName, taskType)\n\t\t},\n\t\tOutstandingTaskAppendsThreshold: func() int {\n\t\t\treturn cfg.OutstandingTaskAppendsThreshold(domainName, taskListName, taskType)\n\t\t},\n\t\tMaxTaskBatchSize: func() int {\n\t\t\treturn cfg.MaxTaskBatchSize(domainName, taskListName, taskType)\n\t\t},\n\t\tNumWritePartitions: func() int {\n\t\t\treturn max(1, cfg.NumTasklistWritePartitions(domainName, taskListName, taskType))\n\t\t},\n\t\tNumReadPartitions: func() int {\n\t\t\treturn max(1, cfg.NumTasklistReadPartitions(domainName, taskListName, taskType))\n\t\t},\n\t\tEnableGetNumberOfPartitionsFromCache: func() bool {\n\t\t\treturn cfg.EnableGetNumberOfPartitionsFromCache(domainName, id.GetRoot(), taskType)\n\t\t},\n\t\tAppendTaskTimeout: func() time.Duration {\n\t\t\treturn cfg.AppendTaskTimeout(domainName, taskListName, taskType)\n\t\t},\n\t\tAsyncTaskDispatchTimeout: func() time.Duration {\n\t\t\treturn cfg.AsyncTaskDispatchTimeout(domainName, taskListName, taskType)\n\t\t},\n\t\tLocalPollWaitTime: func() time.Duration {\n\t\t\treturn cfg.LocalPollWaitTime(domainName, taskListName, taskType)\n\t\t},\n\t\tLocalTaskWaitTime: func() time.Duration {\n\t\t\treturn cfg.LocalTaskWaitTime(domainName, taskListName, taskType)\n\t\t},\n\t\tPartitionUpscaleRPS: func() int {\n\t\t\treturn cfg.PartitionUpscaleRPS(domainName, taskListName, taskType)\n\t\t},\n\t\tPartitionDownscaleFactor: func() float64 {\n\t\t\treturn cfg.PartitionDownscaleFactor(domainName, taskListName, taskType)\n\t\t},\n\t\tPartitionUpscaleSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.PartitionUpscaleSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tPartitionDownscaleSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.PartitionDownscaleSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tAdaptiveScalerUpdateInterval: func() time.Duration {\n\t\t\treturn cfg.AdaptiveScalerUpdateInterval(domainName, taskListName, taskType)\n\t\t},\n\t\tEnablePartitionIsolationGroupAssignment: func() bool {\n\t\t\treturn cfg.EnablePartitionIsolationGroupAssignment(domainName, taskListName, taskType)\n\t\t},\n\t\tIsolationGroupUpscaleSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.IsolationGroupUpscaleSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tIsolationGroupDownscaleSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.IsolationGroupDownscaleSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tIsolationGroupHasPollersSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.IsolationGroupHasPollersSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tIsolationGroupNoPollersSustainedDuration: func() time.Duration {\n\t\t\treturn cfg.IsolationGroupNoPollersSustainedDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tIsolationGroupsPerPartition: func() int {\n\t\t\treturn cfg.IsolationGroupsPerPartition(domainName, taskListName, taskType)\n\t\t},\n\t\tQPSTrackerInterval: func() time.Duration {\n\t\t\treturn cfg.QPSTrackerInterval(domainName, taskListName, taskType)\n\t\t},\n\t\tOverrideTaskListRPS: func() float64 {\n\t\t\treturn cfg.OverrideTaskListRPS(domainName, taskListName, taskType)\n\t\t},\n\t\tEnableAdaptiveScaler: func() bool {\n\t\t\treturn cfg.EnableAdaptiveScaler(domainName, taskListName, taskType)\n\t\t},\n\t\tEnablePartitionEmptyCheck: func() bool {\n\t\t\treturn cfg.EnablePartitionEmptyCheck(domainName, taskListName, taskType)\n\t\t},\n\t\tTaskIsolationDuration: func() time.Duration {\n\t\t\treturn cfg.TaskIsolationDuration(domainName, taskListName, taskType)\n\t\t},\n\t\tTaskIsolationPollerWindow: func() time.Duration {\n\t\t\treturn cfg.TaskIsolationPollerWindow(domainName, taskListName, taskType)\n\t\t},\n\t\tForwarderConfig: config.ForwarderConfig{\n\t\t\tForwarderMaxOutstandingPolls: func() int {\n\t\t\t\treturn cfg.ForwarderMaxOutstandingPolls(domainName, taskListName, taskType)\n\t\t\t},\n\t\t\tForwarderMaxOutstandingTasks: func() int {\n\t\t\t\treturn cfg.ForwarderMaxOutstandingTasks(domainName, taskListName, taskType)\n\t\t\t},\n\t\t\tForwarderMaxRatePerSecond: func() int {\n\t\t\t\treturn cfg.ForwarderMaxRatePerSecond(domainName, taskListName, taskType)\n\t\t\t},\n\t\t\tForwarderMaxChildrenPerNode: func() int {\n\t\t\t\treturn max(1, cfg.ForwarderMaxChildrenPerNode(domainName, taskListName, taskType))\n\t\t\t},\n\t\t},\n\t\tHostName:                  cfg.HostName,\n\t\tTaskDispatchRPS:           cfg.TaskDispatchRPS,\n\t\tTaskDispatchRPSTTL:        cfg.TaskDispatchRPSTTL,\n\t\tMaxTimeBetweenTaskDeletes: cfg.MaxTimeBetweenTaskDeletes,\n\t\tEnableStandbyTaskCompletion: func() bool {\n\t\t\treturn cfg.EnableStandbyTaskCompletion(domainName, taskListName, taskType)\n\t\t},\n\t\tEnableClientAutoConfig: func() bool {\n\t\t\treturn cfg.EnableClientAutoConfig(domainName, taskListName, taskType)\n\t\t},\n\t}\n}\n\nfunc IdentityFromContext(ctx context.Context) string {\n\tval, ok := ctx.Value(identityCtxKey{}).(string)\n\tif !ok {\n\t\treturn \"\"\n\t}\n\treturn val\n}\n\nfunc ContextWithIdentity(ctx context.Context, identity string) context.Context {\n\treturn context.WithValue(ctx, identityCtxKey{}, identity)\n}\n\nfunc PollerIDFromContext(ctx context.Context) string {\n\tval, ok := ctx.Value(pollerIDCtxKey{}).(string)\n\tif !ok {\n\t\treturn \"\"\n\t}\n\treturn val\n}\n\nfunc ContextWithPollerID(ctx context.Context, pollerID string) context.Context {\n\treturn context.WithValue(ctx, pollerIDCtxKey{}, pollerID)\n}\n\nfunc IsolationGroupFromContext(ctx context.Context) string {\n\tval, ok := ctx.Value(isolationGroupCtxKey{}).(string)\n\tif !ok {\n\t\treturn \"\"\n\t}\n\treturn val\n}\n\nfunc ContextWithIsolationGroup(ctx context.Context, isolationGroup string) context.Context {\n\treturn context.WithValue(ctx, isolationGroupCtxKey{}, isolationGroup)\n}\n\nfunc validateParams(p ManagerParams) (err error) {\n\tif p.DomainCache == nil {\n\t\treturn errors.New(\"ManagerParams.DomainCache is required\")\n\t}\n\tif p.Logger == nil {\n\t\treturn errors.New(\"ManagerParams.Logger is required\")\n\t}\n\tif p.MetricsClient == nil {\n\t\treturn errors.New(\"ManagerParams.MetricsClient is required\")\n\t}\n\tif p.TaskManager == nil {\n\t\treturn errors.New(\"ManagerParams.TaskManager is required\")\n\t}\n\tif p.IsolationState == nil {\n\t\treturn errors.New(\"ManagerParams.IsolationState is required\")\n\t}\n\tif p.MatchingClient == nil {\n\t\treturn errors.New(\"ManagerParams.MatchingClient is required\")\n\t}\n\tif p.Registry == nil {\n\t\treturn errors.New(\"ManagerParams.Registry is required\")\n\t}\n\tif p.TaskList == nil {\n\t\treturn errors.New(\"ManagerParams.TaskList is required\")\n\t}\n\tif p.Cfg == nil {\n\t\treturn errors.New(\"ManagerParams.Cfg is required\")\n\t}\n\tif p.TimeSource == nil {\n\t\treturn errors.New(\"ManagerParams.TimeSource is required\")\n\t}\n\tif p.HistoryService == nil {\n\t\treturn errors.New(\"ManagerParams.HistoryService is required\")\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_manager_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/sync/errgroup\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/client/matching\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/stats\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/poller\"\n)\n\ntype mockDeps struct {\n\tmockDomainCache    *cache.MockDomainCache\n\tmockTaskManager    *persistence.MockTaskManager\n\tmockIsolationState *isolationgroup.MockState\n\tmockMatchingClient *matching.MockClient\n\tmockTimeSource     clock.MockedTimeSource\n\tdynamicClient      dynamicconfig.Client\n}\n\nvar (\n\ttestIsolationGroups = []string{\"datacenterA\", \"datacenterB\"}\n)\n\nfunc setupMocksForTaskListManager(t *testing.T, taskListID *Identifier, taskListKind types.TaskListKind) (*taskListManagerImpl, *mockDeps) {\n\tctrl := gomock.NewController(t)\n\tdynamicClient := dynamicconfig.NewInMemoryClient()\n\tlogger := testlogger.New(t)\n\tmetricsClient := metrics.NewNoopMetricsClient()\n\tclusterMetadata := cluster.GetTestClusterMetadata(true)\n\tdeps := &mockDeps{\n\t\tmockDomainCache:    cache.NewMockDomainCache(ctrl),\n\t\tmockTaskManager:    persistence.NewMockTaskManager(ctrl),\n\t\tmockIsolationState: isolationgroup.NewMockState(ctrl),\n\t\tmockMatchingClient: matching.NewMockClient(ctrl),\n\t\tmockTimeSource:     clock.NewMockedTimeSource(),\n\t\tdynamicClient:      dynamicClient,\n\t}\n\tdeps.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domainName\", nil).Times(1)\n\tconfig := config.NewConfig(dynamicconfig.NewCollection(dynamicClient, logger), \"hostname\", commonConfig.RPC{}, getIsolationgroupsHelper)\n\tmockHistoryService := history.NewMockClient(ctrl)\n\tmockRegistry := NewMockTaskListRegistry(ctrl)\n\tmockRegistry.EXPECT().Unregister(gomock.Any()).AnyTimes()\n\tparams := ManagerParams{\n\t\tDomainCache:     deps.mockDomainCache,\n\t\tLogger:          logger,\n\t\tMetricsClient:   metricsClient,\n\t\tTaskManager:     deps.mockTaskManager,\n\t\tClusterMetadata: clusterMetadata,\n\t\tIsolationState:  deps.mockIsolationState,\n\t\tMatchingClient:  deps.mockMatchingClient,\n\t\tRegistry:        mockRegistry,\n\t\tTaskList:        taskListID,\n\t\tTaskListKind:    taskListKind,\n\t\tCfg:             config,\n\t\tTimeSource:      deps.mockTimeSource,\n\t\tCreateTime:      deps.mockTimeSource.Now(),\n\t\tHistoryService:  mockHistoryService,\n\t}\n\ttlm, err := NewManager(params)\n\trequire.NoError(t, err)\n\treturn tlm.(*taskListManagerImpl), deps\n}\n\nfunc defaultTestConfig() *config.Config {\n\tconfig := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, getIsolationgroupsHelper)\n\tconfig.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(100 * time.Millisecond)\n\tconfig.MaxTaskDeleteBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(1)\n\tconfig.AllIsolationGroups = getIsolationgroupsHelper\n\tconfig.GetTasksBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(10)\n\tconfig.AsyncTaskDispatchTimeout = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\tconfig.LocalTaskWaitTime = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(time.Millisecond)\n\treturn config\n}\n\nfunc TestDeliverBufferTasks(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\ttests := []func(tlm *taskListManagerImpl){\n\t\tfunc(tlm *taskListManagerImpl) { tlm.taskReader.cancelFunc() },\n\t\tfunc(tlm *taskListManagerImpl) {\n\t\t\ttlm.limiter.ReportLimit(0.1)\n\t\t\ttlm.taskReader.taskBuffers[defaultTaskBufferIsolationGroup] <- &persistence.TaskInfo{}\n\t\t\terr := tlm.matcher.(*taskMatcherImpl).ratelimit(context.Background()) // consume the token\n\t\t\tassert.NoError(t, err)\n\t\t\ttlm.taskReader.cancelFunc()\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\ttlm := createTestTaskListManager(t, logger, controller)\n\t\tvar wg sync.WaitGroup\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\ttlm.taskReader.dispatchBufferedTasks(defaultTaskBufferIsolationGroup)\n\t\t}()\n\t\ttest(tlm)\n\t\t// dispatchBufferedTasks should stop after invocation of the test function\n\t\twg.Wait()\n\t}\n}\n\nfunc TestTaskListString(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttlm := createTestTaskListManager(t, logger, controller)\n\tgot := tlm.String()\n\twant := \"Activity task list tl\\nRangeID=0\\nTaskIDBlock={start:-99999 end:0}\\nAckLevel=-1\\nMaxReadLevel=-1\\n\"\n\tassert.Equal(t, want, got)\n}\n\nfunc TestDeliverBufferTasks_NoPollers(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\ttlm := createTestTaskListManager(t, logger, controller)\n\ttlm.taskReader.taskBuffers[defaultTaskBufferIsolationGroup] <- &persistence.TaskInfo{}\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tgo func() {\n\t\ttlm.taskReader.dispatchBufferedTasks(\"\")\n\t\twg.Done()\n\t}()\n\ttime.Sleep(100 * time.Millisecond) // let go routine run first and block on tasksForPoll\n\ttlm.taskReader.cancelFunc()\n\twg.Wait()\n}\n\nfunc TestReadLevelForAllExpiredTasksInBatch(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\ttlm := createTestTaskListManager(t, logger, controller)\n\ttlm.db.rangeID = int64(1)\n\ttlm.taskAckManager.SetAckLevel(0)\n\ttlm.taskAckManager.SetReadLevel(0)\n\trequire.Equal(t, int64(0), tlm.taskAckManager.GetAckLevel())\n\trequire.Equal(t, int64(0), tlm.taskAckManager.GetReadLevel())\n\n\t// Add all expired tasks\n\ttasks := []*persistence.TaskInfo{\n\t\t{\n\t\t\tTaskID:      11,\n\t\t\tExpiry:      time.Now().Add(-time.Minute),\n\t\t\tCreatedTime: time.Now().Add(-time.Hour),\n\t\t},\n\t\t{\n\t\t\tTaskID:      12,\n\t\t\tExpiry:      time.Now().Add(-time.Minute),\n\t\t\tCreatedTime: time.Now().Add(-time.Hour),\n\t\t},\n\t}\n\n\trequire.True(t, tlm.taskReader.addTasksToBuffer(tasks))\n\trequire.Equal(t, int64(0), tlm.taskAckManager.GetAckLevel())\n\trequire.Equal(t, int64(12), tlm.taskAckManager.GetReadLevel())\n\n\t// Now add a mix of valid and expired tasks\n\trequire.True(t, tlm.taskReader.addTasksToBuffer([]*persistence.TaskInfo{\n\t\t{\n\t\t\tTaskID:      13,\n\t\t\tExpiry:      time.Now().Add(-time.Minute),\n\t\t\tCreatedTime: time.Now().Add(-time.Hour),\n\t\t},\n\t\t{\n\t\t\tTaskID:      14,\n\t\t\tExpiry:      time.Now().Add(time.Hour),\n\t\t\tCreatedTime: time.Now().Add(time.Minute),\n\t\t},\n\t}))\n\trequire.Equal(t, int64(0), tlm.taskAckManager.GetAckLevel())\n\trequire.Equal(t, int64(14), tlm.taskAckManager.GetReadLevel())\n}\n\nfunc createTestTaskListManager(t *testing.T, logger log.Logger, controller *gomock.Controller) *taskListManagerImpl {\n\treturn createTestTaskListManagerWithConfig(t, logger, controller, defaultTestConfig(), clock.NewMockedTimeSource())\n}\n\nfunc createTestTaskListManagerWithConfig(t *testing.T, logger log.Logger, controller *gomock.Controller, cfg *config.Config, timeSource clock.TimeSource) *taskListManagerImpl {\n\ttm := NewTestTaskManager(t, logger, timeSource)\n\tmockIsolationState := isolationgroup.NewMockState(controller)\n\tmockIsolationState.EXPECT().IsDrained(gomock.Any(), \"domainName\", gomock.Any()).Return(false, nil).AnyTimes()\n\tmockDomainCache := cache.NewMockDomainCache(controller)\n\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(\"domainName\"), nil).AnyTimes()\n\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domainName\", nil).AnyTimes()\n\tmockMatchingClient := matching.NewMockClient(controller)\n\tmockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tmockHistoryService := history.NewMockClient(controller)\n\tmockRegistry := NewMockTaskListRegistry(controller)\n\tmockRegistry.EXPECT().Unregister(gomock.Any()).AnyTimes()\n\ttl := \"tl\"\n\tdID := \"domain\"\n\ttlID, err := NewIdentifier(dID, tl, persistence.TaskListTypeActivity)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\tparams := ManagerParams{\n\t\tDomainCache:     mockDomainCache,\n\t\tLogger:          logger,\n\t\tMetricsClient:   metrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\tTaskManager:     tm,\n\t\tClusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\tIsolationState:  mockIsolationState,\n\t\tMatchingClient:  mockMatchingClient,\n\t\tRegistry:        mockRegistry,\n\t\tTaskList:        tlID,\n\t\tTaskListKind:    types.TaskListKindNormal,\n\t\tCfg:             cfg,\n\t\tTimeSource:      timeSource,\n\t\tCreateTime:      timeSource.Now(),\n\t\tHistoryService:  mockHistoryService,\n\t}\n\ttlMgr, err := NewManager(params)\n\tif err != nil {\n\t\tlogger.Fatal(\"error when createTestTaskListManager\", tag.Error(err))\n\t}\n\treturn tlMgr.(*taskListManagerImpl)\n}\n\nfunc TestTaskListManagerRegistryNotification(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockRegistry := NewMockTaskListRegistry(ctrl)\n\n\t// Create a minimal task list manager for testing\n\ttaskListID := NewTestTaskListID(t, uuid.New(), \"test-tasklist\", 0)\n\ttlm, deps := setupMocksForTaskListManager(t, taskListID, types.TaskListKindNormal)\n\n\t// Set up expectations for Start/Stop\n\tdeps.mockTaskManager.EXPECT().LeaseTaskList(gomock.Any(), gomock.Any()).Return(&persistence.LeaseTaskListResponse{TaskListInfo: &persistence.TaskListInfo{}}, nil).AnyTimes()\n\tdeps.mockTaskManager.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any()).Return(&persistence.UpdateTaskListResponse{}, nil).AnyTimes()\n\n\t// Replace the registry with our mock\n\ttlm.registry = mockRegistry\n\n\t// Expect Unregister to be called exactly once with the manager instance\n\tmockRegistry.EXPECT().Unregister(tlm).Times(1)\n\n\t// Start the manager\n\terr := tlm.Start(context.Background())\n\trequire.NoError(t, err)\n\n\t// Stop should call Unregister\n\ttlm.Stop()\n\t// Verify the manager stopped\n\trequire.Equal(t, int32(1), tlm.stopped)\n\n\t// Second call should be a no-op\n\ttlm.Stop()\n\trequire.Equal(t, int32(1), tlm.stopped)\n}\n\nfunc TestDescribeTaskList(t *testing.T) {\n\t// Magic values hardcoded in matching/config.go. Not much of a config :(\n\tdefaultRps := 100000.0\n\tdefaultRangeSize := 100000\n\tstartedID := int64(1)\n\tfirstIDBlock := &types.TaskIDBlock{\n\t\tStartID: startedID,\n\t\tEndID:   int64(defaultRangeSize),\n\t}\n\n\tcases := []struct {\n\t\tname           string\n\t\tincludeStatus  bool\n\t\tpollers        map[string]poller.Info\n\t\tallowance      func(ctrl *gomock.Controller, impl *taskListManagerImpl)\n\t\texpectedStatus *types.TaskListStatus\n\t\texpectedConfig *types.TaskListPartitionConfig\n\t}{\n\t\t{\n\t\t\tname: \"no status, pollers, or config\",\n\t\t},\n\t\t{\n\t\t\tname: \"no status, with config\",\n\t\t\tallowance: func(_ *gomock.Controller, impl *taskListManagerImpl) {\n\t\t\t\terr := impl.RefreshTaskListPartitionConfig(context.Background(), &types.TaskListPartitionConfig{\n\t\t\t\t\tVersion:         1,\n\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\tWritePartitions: partitions(2),\n\t\t\t\t})\n\t\t\t\trequire.NoError(t, err)\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"no status, with pollers\",\n\t\t\tpollers: map[string]poller.Info{\n\t\t\t\t\"pollerID\": {\n\t\t\t\t\tIdentity:       \"pollerIdentity\",\n\t\t\t\t\tRatePerSecond:  1.0,\n\t\t\t\t\tIsolationGroup: \"a\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"with status\",\n\t\t\tincludeStatus: true,\n\t\t\texpectedStatus: &types.TaskListStatus{\n\t\t\t\tRatePerSecond: defaultRps,\n\t\t\t\tTaskIDBlock:   firstIDBlock,\n\t\t\t\tIsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{\n\t\t\t\t\t\"datacenterA\": {},\n\t\t\t\t\t\"datacenterB\": {},\n\t\t\t\t},\n\t\t\t\tEmpty: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"with status, tasks completed\",\n\t\t\tincludeStatus: true,\n\t\t\tallowance: func(_ *gomock.Controller, tlm *taskListManagerImpl) {\n\t\t\t\tfor i := startedID; i < 11; i++ {\n\t\t\t\t\terr := tlm.taskAckManager.ReadItem(i)\n\t\t\t\t\trequire.NoError(t, err)\n\t\t\t\t}\n\t\t\t\tfor i := startedID; i < 5; i++ {\n\t\t\t\t\ttlm.taskAckManager.AckItem(i)\n\t\t\t\t}\n\t\t\t},\n\t\t\texpectedStatus: &types.TaskListStatus{\n\t\t\t\tBacklogCountHint: 6,\n\t\t\t\tReadLevel:        10,\n\t\t\t\tAckLevel:         4,\n\t\t\t\tRatePerSecond:    defaultRps,\n\t\t\t\tTaskIDBlock:      firstIDBlock,\n\t\t\t\tIsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{\n\t\t\t\t\t\"datacenterA\": {},\n\t\t\t\t\t\"datacenterB\": {},\n\t\t\t\t},\n\t\t\t\tEmpty: false,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"with status, pollers and metrics\",\n\t\t\tincludeStatus: true,\n\t\t\tpollers: map[string]poller.Info{\n\t\t\t\t\"a-1\": {\n\t\t\t\t\tIdentity:       \"a1Identity\",\n\t\t\t\t\tRatePerSecond:  1.0,\n\t\t\t\t\tIsolationGroup: \"datacenterA\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tallowance: func(ctrl *gomock.Controller, impl *taskListManagerImpl) {\n\t\t\t\tmockQPS := stats.NewMockQPSTrackerGroup(ctrl)\n\t\t\t\tmockQPS.EXPECT().GroupQPS(\"datacenterA\").Return(float64(75.0))\n\t\t\t\tmockQPS.EXPECT().GroupQPS(\"datacenterB\").Return(float64(25.0))\n\t\t\t\tmockQPS.EXPECT().QPS().Return(float64(100.0))\n\t\t\t\timpl.qpsTracker = mockQPS\n\t\t\t\timpl.limiter.ReportLimit(1.0)\n\t\t\t},\n\t\t\texpectedStatus: &types.TaskListStatus{\n\t\t\t\tRatePerSecond:     1.0, // From poller\n\t\t\t\tTaskIDBlock:       firstIDBlock,\n\t\t\t\tNewTasksPerSecond: 100,\n\t\t\t\tIsolationGroupMetrics: map[string]*types.IsolationGroupMetrics{\n\t\t\t\t\t\"datacenterA\": {\n\t\t\t\t\t\tPollerCount:       1,\n\t\t\t\t\t\tNewTasksPerSecond: 75.0,\n\t\t\t\t\t},\n\t\t\t\t\t\"datacenterB\": {\n\t\t\t\t\t\tPollerCount:       0,\n\t\t\t\t\t\tNewTasksPerSecond: 25.0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tEmpty: true,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\texpectedTl := &types.TaskList{Name: \"tl\", Kind: types.TaskListKindNormal.Ptr()}\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttlm := createTestTaskListManager(t, logger, controller)\n\t\t\ttlm.db.rangeID = int64(1)\n\t\t\ttlm.taskAckManager.SetAckLevel(0)\n\t\t\ttlm.startWG.Done()\n\n\t\t\texpectedPollers := make([]*types.PollerInfo, 0, len(tc.pollers))\n\t\t\tfor id, info := range tc.pollers {\n\t\t\t\ttlm.pollers.StartPoll(id, func() {}, &info)\n\t\t\t\texpectedPollers = append(expectedPollers, &types.PollerInfo{\n\t\t\t\t\tLastAccessTime: common.Int64Ptr(tlm.timeSource.Now().UnixNano()),\n\t\t\t\t\tIdentity:       info.Identity,\n\t\t\t\t\tRatePerSecond:  info.RatePerSecond,\n\t\t\t\t})\n\t\t\t}\n\t\t\tif tc.allowance != nil {\n\t\t\t\ttc.allowance(controller, tlm)\n\t\t\t}\n\t\t\tresult := tlm.DescribeTaskList(tc.includeStatus)\n\t\t\tassert.Equal(t, expectedTl, result.TaskList)\n\t\t\tassert.Equal(t, tc.expectedStatus, result.TaskListStatus)\n\t\t\tassert.Equal(t, tc.expectedConfig, result.PartitionConfig)\n\t\t\tassert.ElementsMatch(t, expectedPollers, result.Pollers)\n\t\t})\n\t}\n}\n\nfunc TestQueriesPerSecond(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tmockSetup   func(ctrl *gomock.Controller, tlm *taskListManagerImpl)\n\t\texpectedQPS float64\n\t}{\n\t\t{\n\t\t\tname: \"returns QPS from tracker\",\n\t\t\tmockSetup: func(ctrl *gomock.Controller, tlm *taskListManagerImpl) {\n\t\t\t\tmockQPS := stats.NewMockQPSTrackerGroup(ctrl)\n\t\t\t\tmockQPS.EXPECT().QPS().Return(float64(42.5))\n\t\t\t\ttlm.qpsTracker = mockQPS\n\t\t\t},\n\t\t\texpectedQPS: 42.5,\n\t\t},\n\t\t{\n\t\t\tname: \"returns zero QPS\",\n\t\t\tmockSetup: func(ctrl *gomock.Controller, tlm *taskListManagerImpl) {\n\t\t\t\tmockQPS := stats.NewMockQPSTrackerGroup(ctrl)\n\t\t\t\tmockQPS.EXPECT().QPS().Return(float64(0))\n\t\t\t\ttlm.qpsTracker = mockQPS\n\t\t\t},\n\t\t\texpectedQPS: 0,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttlm := createTestTaskListManager(t, logger, ctrl)\n\n\t\t\ttc.mockSetup(ctrl, tlm)\n\n\t\t\tactualQPS := tlm.QueriesPerSecond()\n\t\t\tassert.Equal(t, tc.expectedQPS, actualQPS)\n\t\t})\n\t}\n}\n\nfunc TestCheckIdleTaskList(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tcfg := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, getIsolationgroupsHelper)\n\tcfg.IdleTasklistCheckInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\n\tt.Run(\"Idle task-list\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), ctrl, cfg, clock.NewRealTimeSource())\n\t\trequire.NoError(t, tlm.Start(context.Background()))\n\n\t\trequire.EqualValues(t, 0, atomic.LoadInt32(&tlm.stopped), \"idle check interval had not passed yet\")\n\t\ttime.Sleep(20 * time.Millisecond)\n\t\trequire.EqualValues(t, 1, atomic.LoadInt32(&tlm.stopped), \"idle check interval should have pass\")\n\t})\n\n\tt.Run(\"Active poll-er\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), ctrl, cfg, clock.NewRealTimeSource())\n\t\trequire.NoError(t, tlm.Start(context.Background()))\n\n\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t\t_, _ = tlm.GetTask(ctx, nil)\n\t\tcancel()\n\n\t\t// task list manager should have been stopped,\n\t\t// but GetTask extends auto-stop until the next check-idle-task-list-interval\n\t\ttime.Sleep(6 * time.Millisecond)\n\t\trequire.EqualValues(t, 0, atomic.LoadInt32(&tlm.stopped))\n\n\t\ttime.Sleep(20 * time.Millisecond)\n\t\trequire.EqualValues(t, 1, atomic.LoadInt32(&tlm.stopped), \"idle check interval should have pass\")\n\t})\n\n\tt.Run(\"Active adding task\", func(t *testing.T) {\n\t\tdomainID := uuid.New()\n\t\tworkflowID := uuid.New()\n\t\trunID := uuid.New()\n\n\t\taddTaskParam := AddTaskParams{\n\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\tDomainID:                      domainID,\n\t\t\t\tWorkflowID:                    workflowID,\n\t\t\t\tRunID:                         runID,\n\t\t\t\tScheduleID:                    2,\n\t\t\t\tScheduleToStartTimeoutSeconds: 5,\n\t\t\t\tCreatedTime:                   time.Now(),\n\t\t\t},\n\t\t}\n\n\t\tctrl := gomock.NewController(t)\n\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), ctrl, cfg, clock.NewRealTimeSource())\n\t\trequire.NoError(t, tlm.Start(context.Background()))\n\n\t\ttime.Sleep(8 * time.Millisecond)\n\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second)\n\t\t_, err := tlm.AddTask(ctx, addTaskParam)\n\t\trequire.NoError(t, err)\n\t\tcancel()\n\n\t\t// task list manager should have been stopped,\n\t\t// but AddTask extends auto-stop until the next check-idle-task-list-interval\n\t\ttime.Sleep(6 * time.Millisecond)\n\t\trequire.EqualValues(t, 0, atomic.LoadInt32(&tlm.stopped))\n\n\t\ttime.Sleep(20 * time.Millisecond)\n\t\trequire.EqualValues(t, 1, atomic.LoadInt32(&tlm.stopped), \"idle check interval should have pass\")\n\t})\n}\n\nfunc TestAddTaskStandby(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\tcfg := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, getIsolationgroupsHelper)\n\tcfg.IdleTasklistCheckInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(10 * time.Millisecond)\n\n\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, cfg, clock.NewMockedTimeSource())\n\trequire.NoError(t, tlm.Start(context.Background()))\n\n\t// stop taskWriter so that we can check if there's any call to it\n\t// otherwise the task persist process is async and hard to test\n\ttlm.taskWriter.Stop()\n\n\tdomainID := uuid.New()\n\tworkflowID := \"some random workflowID\"\n\trunID := \"some random runID\"\n\n\taddTaskParam := AddTaskParams{\n\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\tDomainID:                      domainID,\n\t\t\tWorkflowID:                    workflowID,\n\t\t\tRunID:                         runID,\n\t\t\tScheduleID:                    2,\n\t\t\tScheduleToStartTimeoutSeconds: 5,\n\t\t\tCreatedTime:                   time.Now(),\n\t\t},\n\t}\n\n\ttestStandbyDomainEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: domainID, Name: \"some random domain name\"},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\tmockDomainCache := tlm.domainCache.(*cache.MockDomainCache)\n\tmockDomainCache.EXPECT().GetDomainByID(domainID).Return(testStandbyDomainEntry, nil).AnyTimes()\n\n\tsyncMatch, err := tlm.AddTask(context.Background(), addTaskParam)\n\trequire.Equal(t, errShutdown, err) // task writer was stopped above\n\trequire.False(t, syncMatch)\n\n\taddTaskParam.ForwardedFrom = \"from child partition\"\n\tsyncMatch, err = tlm.AddTask(context.Background(), addTaskParam)\n\trequire.Error(t, err) // should not persist the task\n\trequire.False(t, syncMatch)\n}\n\n// return a client side tasklist throttle error from the rate limiter.\n// The expected behaviour is to retry\nfunc TestRateLimitErrorsFromTasklistDispatch(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\tconfig := defaultTestConfig()\n\tconfig.EnableTasklistIsolation = func(domain string) bool { return true }\n\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, config, clock.NewMockedTimeSource())\n\n\ttlm.taskReader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\treturn ErrTasklistThrottled\n\t}\n\ttlm.taskReader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\treturn \"datacenterA\", -1\n\t}\n\n\tbreakDispatcher, breakRetryLoop := tlm.taskReader.dispatchSingleTaskFromBuffer(&persistence.TaskInfo{})\n\tassert.False(t, breakDispatcher)\n\tassert.False(t, breakRetryLoop)\n}\n\n// This is a bit of a strange unit-test as it's\n// ensuring that invalid behaviour is handled defensively.\n// It *should never be the case* that the isolation group tries to\n// dispatch to a buffer that isn't there, however, if it does, we want to just\n// log this, emit a metric and fallback to the default isolation group.\nfunc TestMisconfiguredZoneDoesNotBlock(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\tconfig := defaultTestConfig()\n\tconfig.EnableTasklistIsolation = func(domain string) bool { return true }\n\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, config, clock.NewMockedTimeSource())\n\tinvalidIsolationGroup := \"invalid\"\n\tdispatched := make(map[string]int)\n\n\t// record dispatched isolation group\n\ttlm.taskReader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\tdispatched[task.isolationGroup] = dispatched[task.isolationGroup] + 1\n\t\treturn nil\n\t}\n\ttlm.taskReader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\treturn invalidIsolationGroup, -1\n\t}\n\n\tmaxBufferSize := config.GetTasksBatchSize(\"\", \"\", 0) - 1\n\n\tfor i := 0; i < maxBufferSize; i++ {\n\t\tbreakDispatcher, breakRetryLoop := tlm.taskReader.dispatchSingleTaskFromBuffer(&persistence.TaskInfo{})\n\t\tassert.False(t, breakDispatcher, \"dispatch isn't shutting down\")\n\t\tassert.True(t, breakRetryLoop, \"should be able to successfully dispatch all these tasks to the default isolation group\")\n\t}\n\t// We should see them all being redirected\n\tassert.Equal(t, dispatched[\"\"], maxBufferSize)\n\t// we should see none in the returned isolation group\n\tassert.Equal(t, dispatched[invalidIsolationGroup], 0)\n\n\t// ok, and here we try and ensure that this *does not block\n\t// and instead complains and live-retries\n\tbreakDispatcher, breakRetryLoop := tlm.taskReader.dispatchSingleTaskFromBuffer(&persistence.TaskInfo{})\n\tassert.False(t, breakDispatcher, \"dispatch isn't shutting down\")\n\tassert.True(t, breakRetryLoop, \"task should be dispatched to default channel\")\n}\n\nfunc TestGetIsolationGroupForTask(t *testing.T) {\n\tdefaultAvailableIsolationGroups := []string{\n\t\t\"a\", \"b\", \"c\",\n\t}\n\ttaskIsolationPollerWindow := time.Second * 10\n\ttestCases := []struct {\n\t\tname                     string\n\t\ttaskIsolationGroup       string\n\t\ttaskIsolationDuration    time.Duration\n\t\ttaskLatency              time.Duration\n\t\tavailableIsolationGroups []string\n\t\trecentPollers            []string\n\t\texpectedGroup            string\n\t\texpectedDuration         time.Duration\n\t\tdrainedGroups            map[string]bool\n\t\tisolationStateErr        error\n\t\tdisableTaskIsolation     bool\n\t\tpartitionConfig          *types.TaskListPartition\n\t}{\n\t\t{\n\t\t\tname:                     \"success - recent poller allows group\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"a\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t},\n\t\t{\n\t\t\tname:                     \"success - with isolation duration\",\n\t\t\ttaskIsolationGroup:       \"b\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"b\",\n\t\t\texpectedDuration:         time.Second,\n\t\t\trecentPollers:            []string{\"b\"},\n\t\t},\n\t\t{\n\t\t\tname:                     \"success - low task latency\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\ttaskLatency:              time.Millisecond * 300,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"a\",\n\t\t\texpectedDuration:         time.Second - (time.Millisecond * 300),\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t},\n\t\t{\n\t\t\tname:                     \"success - other group drained\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"a\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\tdrainedGroups: map[string]bool{\n\t\t\t\t\"b\": true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                     \"success - right partition\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"a\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\tpartitionConfig:          &types.TaskListPartition{IsolationGroups: []string{\"a\", \"b\", \"c\", \"d\"}},\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - no recent pollers\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            nil,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - no matching recent poller\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"b\"},\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - unknown group\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: []string{\"b\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - unknown group even with recent poller\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: []string{\"b\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - task latency\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\ttaskLatency:              time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - task latency close to taskIsolationDuration\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\ttaskLatency:              time.Second - minimumIsolationDuration,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - group drained\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\tdrainedGroups: map[string]bool{\n\t\t\t\t\"a\": true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - state error\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\tisolationStateErr:        errors.New(\">:(\"),\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - task isolation disabled\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\ttaskIsolationDuration:    time.Second,\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\tdisableTaskIsolation:     true,\n\t\t},\n\t\t{\n\t\t\tname:                     \"leak - wrong partition\",\n\t\t\ttaskIsolationGroup:       \"a\",\n\t\t\tavailableIsolationGroups: defaultAvailableIsolationGroups,\n\t\t\texpectedGroup:            \"\",\n\t\t\texpectedDuration:         0,\n\t\t\trecentPollers:            []string{\"a\"},\n\t\t\tpartitionConfig:          &types.TaskListPartition{IsolationGroups: []string{\"b\", \"c\", \"d\"}},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\n\t\t\tconfig := defaultTestConfig()\n\t\t\tconfig.EnableTasklistIsolation = func(domain string) bool { return !tc.disableTaskIsolation }\n\t\t\tconfig.TaskIsolationDuration = func(domain string, taskList string, taskType int) time.Duration {\n\t\t\t\treturn tc.taskIsolationDuration\n\t\t\t}\n\t\t\tconfig.TaskIsolationPollerWindow = func(domain string, taskList string, taskType int) time.Duration {\n\t\t\t\treturn taskIsolationPollerWindow\n\t\t\t}\n\t\t\tconfig.AllIsolationGroups = func() []string {\n\t\t\t\treturn tc.availableIsolationGroups\n\t\t\t}\n\t\t\tmockClock := clock.NewMockedTimeSource()\n\t\t\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, config, mockClock)\n\n\t\t\tif tc.partitionConfig != nil {\n\t\t\t\ttlm.startWG.Done()\n\t\t\t\t// Add a partition and update the root to the specified config.\n\t\t\t\t// If we're not adding a new partition it is treated as a no-op since the default is 1 partition\n\t\t\t\terr := tlm.UpdateTaskListPartitionConfig(context.Background(), &types.TaskListPartitionConfig{\n\t\t\t\t\tReadPartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: tc.partitionConfig,\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t\tWritePartitions: map[int]*types.TaskListPartition{\n\t\t\t\t\t\t0: tc.partitionConfig,\n\t\t\t\t\t\t1: {},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\n\t\t\tmockIsolationGroupState := isolationgroup.NewMockState(controller)\n\t\t\tif tc.isolationStateErr != nil {\n\t\t\t\tmockIsolationGroupState.EXPECT().IsDrained(gomock.Any(), \"domainName\", gomock.Any()).Return(false, tc.isolationStateErr).AnyTimes()\n\t\t\t} else {\n\t\t\t\tmockIsolationGroupState.EXPECT().IsDrained(gomock.Any(), \"domainName\", gomock.Any()).DoAndReturn(func(ctx context.Context, domainName, group string) (bool, error) {\n\t\t\t\t\treturn tc.drainedGroups[group], nil\n\t\t\t\t}).AnyTimes()\n\t\t\t}\n\t\t\ttlm.isolationState = mockIsolationGroupState\n\n\t\t\tfor i, pollerGroup := range tc.recentPollers {\n\t\t\t\ttlm.pollers.StartPoll(strconv.Itoa(i), func() {}, &poller.Info{Identity: pollerGroup, IsolationGroup: pollerGroup})\n\t\t\t}\n\n\t\t\ttaskInfo := &persistence.TaskInfo{\n\t\t\t\tDomainID:                      \"domainId\",\n\t\t\t\tRunID:                         \"run1\",\n\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\tScheduleID:                    5,\n\t\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t\t\tPartitionConfig: map[string]string{\n\t\t\t\t\tisolationgroup.GroupKey:      tc.taskIsolationGroup,\n\t\t\t\t\tisolationgroup.WorkflowIDKey: \"workflow1\",\n\t\t\t\t},\n\t\t\t\tCreatedTime: mockClock.Now().Add(-1 * tc.taskLatency),\n\t\t\t}\n\n\t\t\tactual, actualDuration := tlm.getIsolationGroupForTask(context.Background(), taskInfo)\n\n\t\t\tassert.Equal(t, tc.expectedGroup, actual)\n\t\t\tassert.Equal(t, tc.expectedDuration, actualDuration)\n\t\t})\n\t}\n}\n\nfunc TestTaskWriterShutdown(t *testing.T) {\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttlm := createTestTaskListManager(t, logger, controller)\n\terr := tlm.Start(context.Background())\n\tassert.NoError(t, err)\n\n\t// stop the task writer explicitly\n\ttlm.taskWriter.Stop()\n\n\t// now attempt to add a task\n\taddParams := AddTaskParams{\n\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\tDomainID:                      \"domainId\",\n\t\t\tRunID:                         \"run1\",\n\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\tScheduleID:                    5,\n\t\t\tScheduleToStartTimeoutSeconds: 1,\n\t\t},\n\t}\n\t_, err = tlm.AddTask(context.Background(), addParams)\n\tassert.Error(t, err)\n\n\t// test race\n\ttlm.taskWriter.stopped = 0\n\t_, err = tlm.AddTask(context.Background(), addParams)\n\tassert.Error(t, err)\n\ttlm.taskWriter.stopped = 1 // reset it back to old value\n\ttlm.Stop()\n}\n\nfunc TestTaskListManagerGetTaskBatch(t *testing.T) {\n\tconst taskCount = 1200\n\tconst rangeSize = 10\n\tcontroller := gomock.NewController(t)\n\tmockIsolationState := isolationgroup.NewMockState(controller)\n\tmockIsolationState.EXPECT().IsDrained(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil).AnyTimes()\n\tmockDomainCache := cache.NewMockDomainCache(controller)\n\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(\"domainName\"), nil).AnyTimes()\n\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domainName\", nil).AnyTimes()\n\tmockHistoryService := history.NewMockClient(controller)\n\tlogger := testlogger.New(t)\n\ttimeSource := clock.NewRealTimeSource()\n\ttm := NewTestTaskManager(t, logger, timeSource)\n\ttaskListID := NewTestTaskListID(t, \"domainId\", \"tl\", 0)\n\tcfg := defaultTestConfig()\n\tcfg.RangeSize = rangeSize\n\tcfg.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\tmockRegistry := NewMockTaskListRegistry(controller)\n\tmockRegistry.EXPECT().Unregister(gomock.Any()).AnyTimes()\n\tparams := ManagerParams{\n\t\tDomainCache:     mockDomainCache,\n\t\tLogger:          logger,\n\t\tMetricsClient:   metrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\tTaskManager:     tm,\n\t\tClusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\tIsolationState:  mockIsolationState,\n\t\tMatchingClient:  matching.NewMockClient(controller),\n\t\tRegistry:        mockRegistry,\n\t\tTaskList:        taskListID,\n\t\tTaskListKind:    types.TaskListKindNormal,\n\t\tCfg:             cfg,\n\t\tTimeSource:      timeSource,\n\t\tCreateTime:      timeSource.Now(),\n\t\tHistoryService:  mockHistoryService,\n\t}\n\ttlMgr, err := NewManager(params)\n\tassert.NoError(t, err)\n\ttlm := tlMgr.(*taskListManagerImpl)\n\terr = tlm.Start(context.Background())\n\tassert.NoError(t, err)\n\n\t// add taskCount tasks\n\tfor i := int64(0); i < taskCount; i++ {\n\t\tscheduleID := i * 3\n\t\taddParams := AddTaskParams{\n\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\tDomainID:                      \"domainId\",\n\t\t\t\tRunID:                         \"run1\",\n\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\tScheduleID:                    scheduleID,\n\t\t\t\tScheduleToStartTimeoutSeconds: 100,\n\t\t\t},\n\t\t}\n\t\t_, err = tlm.AddTask(context.Background(), addParams)\n\t\tassert.NoError(t, err)\n\t}\n\tassert.Equal(t, taskCount, tm.GetTaskCount(taskListID))\n\n\t// wait until all tasks are read by the task pump and enqeued into the in-memory buffer\n\t// at the end of this step, ackManager readLevel will also be equal to the buffer size\n\texpectedBufSize := min(cap(tlm.taskReader.taskBuffers[defaultTaskBufferIsolationGroup]), taskCount)\n\tassert.True(t, awaitCondition(func() bool {\n\t\treturn len(tlm.taskReader.taskBuffers[defaultTaskBufferIsolationGroup]) == expectedBufSize\n\t}, 10*time.Second))\n\n\t// stop all goroutines that read / write tasks in the background\n\t// remainder of this test works with the in-memory buffer\n\ttlm.Stop()\n\n\t// SetReadLevel should NEVER be called without updating ackManager.outstandingTasks\n\t// This is only for unit test purpose\n\ttlm.taskAckManager.SetReadLevel(tlm.taskWriter.GetMaxReadLevel())\n\ttasks, readLevel, isReadBatchDone, err := tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.NoError(t, err)\n\tassert.Equal(t, 0, len(tasks))\n\tassert.Equal(t, readLevel, tlm.taskWriter.GetMaxReadLevel())\n\tassert.True(t, isReadBatchDone)\n\n\ttlm.taskAckManager.SetReadLevel(0)\n\ttasks, readLevel, isReadBatchDone, err = tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.NoError(t, err)\n\tassert.Equal(t, rangeSize/2, len(tasks))\n\tassert.Equal(t, rangeSize/2, int(readLevel))\n\tassert.True(t, isReadBatchDone)\n\n\t// reset the ackManager readLevel to the buffer size and consume\n\t// the in-memory tasks by calling Poll API - assert ackMgr state\n\t// at the end\n\ttlm.taskAckManager.SetReadLevel(int64(expectedBufSize))\n\n\t// complete rangeSize events\n\tnewParams := ManagerParams{\n\t\tDomainCache:     mockDomainCache,\n\t\tLogger:          logger,\n\t\tMetricsClient:   metrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\tTaskManager:     tm,\n\t\tClusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\tIsolationState:  mockIsolationState,\n\t\tMatchingClient:  matching.NewMockClient(controller),\n\t\tRegistry:        mockRegistry,\n\t\tTaskList:        taskListID,\n\t\tTaskListKind:    types.TaskListKindNormal,\n\t\tCfg:             cfg,\n\t\tTimeSource:      timeSource,\n\t\tCreateTime:      timeSource.Now(),\n\t\tHistoryService:  mockHistoryService,\n\t}\n\ttlMgr, err = NewManager(newParams)\n\tassert.NoError(t, err)\n\ttlm = tlMgr.(*taskListManagerImpl)\n\terr = tlm.Start(context.Background())\n\tassert.NoError(t, err)\n\tfor i := int64(0); i < rangeSize; i++ {\n\t\ttask, err := tlm.GetTask(context.Background(), nil)\n\t\tif errors.Is(err, ErrNoTasks) {\n\t\t\tcontinue\n\t\t}\n\t\tassert.NotNil(t, task.AutoConfigHint)\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, task)\n\t\ttask.Finish(nil)\n\t}\n\tassert.Equal(t, taskCount-rangeSize, tm.GetTaskCount(taskListID))\n\ttasks, _, isReadBatchDone, err = tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.NoError(t, err)\n\tassert.True(t, 0 < len(tasks) && len(tasks) <= rangeSize)\n\tassert.True(t, isReadBatchDone)\n\ttlm.Stop()\n}\n\nfunc TestTaskListReaderPumpAdvancesAckLevelAfterEmptyReads(t *testing.T) {\n\tconst taskCount = 5\n\tconst rangeSize = 10\n\tconst nLeaseRenewals = 15\n\n\tcontroller := gomock.NewController(t)\n\tmockIsolationState := isolationgroup.NewMockState(controller)\n\tmockIsolationState.EXPECT().IsDrained(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil).AnyTimes()\n\tmockDomainCache := cache.NewMockDomainCache(controller)\n\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(\"domainName\"), nil).AnyTimes()\n\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domainName\", nil).AnyTimes()\n\tmockHistoryService := history.NewMockClient(controller)\n\n\tlogger := testlogger.New(t)\n\ttimeSource := clock.NewRealTimeSource()\n\ttm := NewTestTaskManager(t, logger, timeSource)\n\ttaskListID := NewTestTaskListID(t, \"domainId\", \"tl\", 0)\n\tcfg := defaultTestConfig()\n\tcfg.RangeSize = rangeSize\n\tcfg.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\n\tmockRegistry := NewMockTaskListRegistry(controller)\n\tmockRegistry.EXPECT().Unregister(gomock.Any()).AnyTimes()\n\tparams := ManagerParams{\n\t\tDomainCache:     mockDomainCache,\n\t\tLogger:          logger,\n\t\tMetricsClient:   metrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\tTaskManager:     tm,\n\t\tClusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\tIsolationState:  mockIsolationState,\n\t\tMatchingClient:  matching.NewMockClient(controller),\n\t\tRegistry:        mockRegistry,\n\t\tTaskList:        taskListID,\n\t\tTaskListKind:    types.TaskListKindNormal,\n\t\tCfg:             cfg,\n\t\tTimeSource:      timeSource,\n\t\tCreateTime:      timeSource.Now(),\n\t\tHistoryService:  mockHistoryService,\n\t}\n\ttlMgr, err := NewManager(params)\n\trequire.NoError(t, err)\n\ttlm := tlMgr.(*taskListManagerImpl)\n\n\t// simulate lease renewal multiple times without writing any tasks\n\tfor i := 0; i < nLeaseRenewals; i++ {\n\t\ttlm.taskWriter.renewLeaseWithRetry()\n\t}\n\n\terr = tlm.Start(context.Background()) // this call will also renew lease\n\trequire.NoError(t, err)\n\tdefer tlm.Stop()\n\n\t// we expect AckLevel to advance and skip all the previously leased ranges\n\texpectedAckLevel := int64(rangeSize) * nLeaseRenewals\n\n\t// wait until task pump will read batches of empty ranges\n\tassert.True(t, awaitCondition(func() bool {\n\t\treturn tlm.taskAckManager.GetAckLevel() == expectedAckLevel\n\t}, 10*time.Second))\n\n\tassert.Equal(\n\t\tt,\n\t\texpectedAckLevel,\n\t\ttlm.taskAckManager.GetAckLevel(),\n\t\t\"we should ack ranges of all the previously acquired leases\",\n\t)\n\n\tassert.Equal(\n\t\tt,\n\t\ttlm.taskWriter.GetMaxReadLevel(),\n\t\ttlm.taskAckManager.GetAckLevel(),\n\t\t\"we should have been acked everything possible\",\n\t)\n\n\tmaxReadLevelBeforeAddingTasks := tlm.taskWriter.GetMaxReadLevel()\n\n\t// verify new task writes go beyond the MaxReadLevel/AckLevel\n\tfor i := int64(0); i < taskCount; i++ {\n\t\taddParams := AddTaskParams{\n\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\tDomainID:   \"domainId\",\n\t\t\t\tRunID:      \"run1\",\n\t\t\t\tWorkflowID: \"workflow1\",\n\t\t\t\tScheduleID: i,\n\t\t\t},\n\t\t}\n\t\t_, err = tlm.AddTask(context.Background(), addParams)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, maxReadLevelBeforeAddingTasks+i+1, tlm.taskWriter.GetMaxReadLevel())\n\t}\n}\n\nfunc TestTaskListManagerGetTaskBatch_ReadBatchDone(t *testing.T) {\n\tconst rangeSize = 10\n\tconst maxReadLevel = int64(120)\n\tconfig := defaultTestConfig()\n\tconfig.RangeSize = rangeSize\n\tconfig.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\tcontroller := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, config, clock.NewMockedTimeSource())\n\n\ttlm.taskAckManager.SetReadLevel(0)\n\tatomic.StoreInt64(&tlm.taskWriter.maxReadLevel, maxReadLevel)\n\ttasks, readLevel, isReadBatchDone, err := tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.Empty(t, tasks)\n\tassert.Equal(t, int64(rangeSize/2*10), readLevel)\n\tassert.False(t, isReadBatchDone)\n\tassert.NoError(t, err)\n\n\ttlm.taskAckManager.SetReadLevel(readLevel)\n\tatomic.StoreInt64(&tlm.taskWriter.maxReadLevel, maxReadLevel)\n\ttasks, readLevel, isReadBatchDone, err = tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.Empty(t, tasks)\n\tassert.Equal(t, 2*int64(rangeSize/2*10), readLevel)\n\tassert.False(t, isReadBatchDone)\n\tassert.NoError(t, err)\n\n\ttlm.taskAckManager.SetReadLevel(readLevel)\n\ttasks, readLevel, isReadBatchDone, err = tlm.taskReader.getTaskBatch(tlm.taskAckManager.GetReadLevel(), tlm.taskWriter.GetMaxReadLevel())\n\tassert.Empty(t, tasks)\n\tassert.Equal(t, maxReadLevel, readLevel)\n\tassert.True(t, isReadBatchDone)\n\tassert.NoError(t, err)\n}\n\nfunc awaitCondition(cond func() bool, timeout time.Duration) bool {\n\texpiry := time.Now().Add(timeout)\n\tfor !cond() {\n\t\ttime.Sleep(time.Millisecond * 5)\n\t\tif time.Now().After(expiry) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n\nfunc TestTaskExpiryAndCompletion(t *testing.T) {\n\tconst taskCount = 20\n\tconst rangeSize = 10\n\n\ttestCases := []struct {\n\t\tname               string\n\t\tbatchSize          int\n\t\tmaxTimeBtwnDeletes time.Duration\n\t}{\n\t\t{\"test taskGC deleting due to size threshold\", 2, time.Minute},\n\t\t{\"test taskGC deleting due to time condition\", 100, time.Nanosecond},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(\"\", func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tmockIsolationState := isolationgroup.NewMockState(controller)\n\t\t\tmockIsolationState.EXPECT().IsDrained(gomock.Any(), gomock.Any(), gomock.Any()).Return(false, nil).AnyTimes()\n\t\t\tmockDomainCache := cache.NewMockDomainCache(controller)\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(cache.CreateDomainCacheEntry(\"domainName\"), nil).AnyTimes()\n\t\t\tmockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"domainName\", nil).AnyTimes()\n\t\t\tmockHistoryService := history.NewMockClient(controller)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttimeSource := clock.NewRealTimeSource()\n\t\t\ttm := NewTestTaskManager(t, logger, timeSource)\n\t\t\ttaskListID := NewTestTaskListID(t, \"domainId\", \"tl\", 0)\n\t\t\tcfg := defaultTestConfig()\n\t\t\tcfg.RangeSize = rangeSize\n\t\t\tcfg.ReadRangeSize = dynamicproperties.GetIntPropertyFn(rangeSize / 2)\n\t\t\tcfg.MaxTaskDeleteBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(tc.batchSize)\n\t\t\tcfg.MaxTimeBetweenTaskDeletes = tc.maxTimeBtwnDeletes\n\t\t\t// set idle timer check to a really small value to assert that we don't accidentally drop tasks while blocking\n\t\t\t// on enqueuing a task to task buffer\n\t\t\tcfg.IdleTasklistCheckInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(20 * time.Millisecond)\n\t\t\tmockRegistry := NewMockTaskListRegistry(controller)\n\t\t\tmockRegistry.EXPECT().Unregister(gomock.Any()).AnyTimes()\n\t\t\tparams := ManagerParams{\n\t\t\t\tDomainCache:     mockDomainCache,\n\t\t\t\tLogger:          logger,\n\t\t\t\tMetricsClient:   metrics.NewClient(tally.NoopScope, metrics.Matching, metrics.HistogramMigration{}),\n\t\t\t\tTaskManager:     tm,\n\t\t\t\tClusterMetadata: cluster.GetTestClusterMetadata(true),\n\t\t\t\tIsolationState:  mockIsolationState,\n\t\t\t\tMatchingClient:  matching.NewMockClient(controller),\n\t\t\t\tRegistry:        mockRegistry,\n\t\t\t\tTaskList:        taskListID,\n\t\t\t\tTaskListKind:    types.TaskListKindNormal,\n\t\t\t\tCfg:             cfg,\n\t\t\t\tTimeSource:      timeSource,\n\t\t\t\tCreateTime:      timeSource.Now(),\n\t\t\t\tHistoryService:  mockHistoryService,\n\t\t\t}\n\t\t\ttlMgr, err := NewManager(params)\n\t\t\tassert.NoError(t, err)\n\t\t\ttlm := tlMgr.(*taskListManagerImpl)\n\t\t\terr = tlm.Start(context.Background())\n\t\t\tassert.NoError(t, err)\n\t\t\tfor i := int64(0); i < taskCount; i++ {\n\t\t\t\tscheduleID := i * 3\n\t\t\t\taddParams := AddTaskParams{\n\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\tDomainID:                      \"domainId\",\n\t\t\t\t\t\tRunID:                         \"run1\",\n\t\t\t\t\t\tWorkflowID:                    \"workflow1\",\n\t\t\t\t\t\tScheduleID:                    scheduleID,\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: 100,\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tif i%2 == 0 {\n\t\t\t\t\t// simulates creating a task whose scheduledToStartTimeout is already expired\n\t\t\t\t\taddParams.TaskInfo.ScheduleToStartTimeoutSeconds = -5\n\t\t\t\t}\n\t\t\t\t_, err = tlm.AddTask(context.Background(), addParams)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, taskCount, tm.GetTaskCount(taskListID))\n\n\t\t\t// wait until all tasks are loaded by into in-memory buffers by task list manager\n\t\t\t// the buffer size should be one less than expected because dispatcher will dequeue the head\n\t\t\tassert.True(t, awaitCondition(func() bool {\n\t\t\t\treturn len(tlm.taskReader.taskBuffers[defaultTaskBufferIsolationGroup]) >= (taskCount/2 - 1)\n\t\t\t}, time.Second))\n\n\t\t\tremaining := taskCount\n\t\t\tfor i := 0; i < 2; i++ {\n\t\t\t\t// verify that (1) expired tasks are not returned in poll result (2) taskCleaner deletes tasks correctly\n\t\t\t\tfor i := int64(0); i < taskCount/4; i++ {\n\t\t\t\t\ttask, err := tlm.GetTask(context.Background(), nil)\n\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\tassert.NotNil(t, task)\n\t\t\t\t\ttask.Finish(nil)\n\t\t\t\t}\n\t\t\t\tremaining -= taskCount / 2\n\t\t\t\t// since every other task is expired, we expect half the tasks to be deleted\n\t\t\t\t// after poll consumed 1/4th of what is available\n\t\t\t\tassert.Equal(t, remaining, tm.GetTaskCount(taskListID))\n\t\t\t}\n\t\t\ttlm.Stop()\n\t\t})\n\t}\n}\n\nfunc TestTaskListManagerImpl_HasPollerAfter(t *testing.T) {\n\tfor name, tc := range map[string]struct {\n\t\toutstandingPollers []string\n\t\tprepareManager     func(*taskListManagerImpl)\n\t}{\n\t\t\"has_outstanding_pollers\": {\n\t\t\tprepareManager: func(tlm *taskListManagerImpl) {\n\t\t\t\ttlm.pollers.StartPoll(\"poller1\", func() {}, &poller.Info{Identity: \"foo\"})\n\t\t\t},\n\t\t},\n\t\t\"no_outstanding_pollers\": {\n\t\t\tprepareManager: func(tlm *taskListManagerImpl) {\n\t\t\t\ttlm.pollers.StartPoll(\"poller1\", func() {}, &poller.Info{Identity: \"foo\"})\n\t\t\t\ttlm.pollers.EndPoll(\"poller1\")\n\t\t\t},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttlm := createTestTaskListManager(t, logger, controller)\n\t\t\terr := tlm.Start(context.Background())\n\t\t\tassert.NoError(t, err)\n\n\t\t\tif tc.prepareManager != nil {\n\t\t\t\ttc.prepareManager(tlm)\n\t\t\t}\n\n\t\t\tassert.True(t, tlm.HasPollerAfter(time.Time{}))\n\t\t})\n\t}\n}\n\nfunc getIsolationgroupsHelper() []string {\n\treturn testIsolationGroups\n}\n\nfunc TestRefreshTaskListPartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\treq            *types.TaskListPartitionConfig\n\t\toriginalConfig *types.TaskListPartitionConfig\n\t\tsetupMocks     func(*mockDeps)\n\t\texpectedConfig *types.TaskListPartitionConfig\n\t\texpectError    bool\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"success - refresh from request\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         2,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\tsetupMocks: func(m *mockDeps) {},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         2,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - ignore older version\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         2,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         3,\n\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t\tsetupMocks: func(m *mockDeps) {},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         3,\n\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - refresh from database\",\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         3,\n\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockTaskManager.EXPECT().GetTaskList(gomock.Any(), &persistence.GetTaskListRequest{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tDomainName: \"domainName\",\n\t\t\t\t\tTaskList:   \"tl\",\n\t\t\t\t\tTaskType:   persistence.TaskListTypeDecision,\n\t\t\t\t}).Return(&persistence.GetTaskListResponse{\n\t\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\t\tVersion:         4,\n\t\t\t\t\t\t\tReadPartitions:  persistencePartitions(10),\n\t\t\t\t\t\t\tWritePartitions: persistencePartitions(10),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         4,\n\t\t\t\tReadPartitions:  partitions(10),\n\t\t\t\tWritePartitions: partitions(10),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to refresh from database\",\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         3,\n\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockTaskManager.EXPECT().GetTaskList(gomock.Any(), &persistence.GetTaskListRequest{\n\t\t\t\t\tDomainID:   \"domain-id\",\n\t\t\t\t\tDomainName: \"domainName\",\n\t\t\t\t\tTaskList:   \"tl\",\n\t\t\t\t\tTaskType:   persistence.TaskListTypeDecision,\n\t\t\t\t}).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         3,\n\t\t\t\tReadPartitions:  partitions(2),\n\t\t\t\tWritePartitions: partitions(2),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"some error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttlID, err := NewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\t\t\trequire.NoError(t, err)\n\t\t\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\t\t\ttc.setupMocks(deps)\n\t\t\ttlm.partitionConfig = tc.originalConfig\n\t\t\ttlm.startWG.Done()\n\n\t\t\terr = tlm.RefreshTaskListPartitionConfig(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t\tassert.Equal(t, tc.expectedConfig, tlm.TaskListPartitionConfig())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedConfig, tlm.TaskListPartitionConfig())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestUpdateTaskListPartitionConfig(t *testing.T) {\n\ttestCases := []struct {\n\t\tname           string\n\t\treq            *types.TaskListPartitionConfig\n\t\toriginalConfig *types.TaskListPartitionConfig\n\t\tsetupMocks     func(*mockDeps)\n\t\texpectedConfig *types.TaskListPartitionConfig\n\t\texpectError    bool\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"success - no op\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\tsetupMocks: func(m *mockDeps) {},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - no op, nil pointer\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  partitions(1),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t\toriginalConfig: nil,\n\t\t\tsetupMocks:     func(m *mockDeps) {},\n\t\t\texpectedConfig: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"success - update\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockTaskManager.EXPECT().UpdateTaskList(gomock.Any(), &persistence.UpdateTaskListRequest{\n\t\t\t\t\tDomainName: \"domainName\",\n\t\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\t\tDomainID: \"domain-id\",\n\t\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\t\tRangeID:  0,\n\t\t\t\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\t\tReadPartitions:  persistencePartitions(3),\n\t\t\t\t\t\t\tWritePartitions: persistencePartitions(1),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(&persistence.UpdateTaskListResponse{}, nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:   \"domain-id\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"/__cadence_sys/tl/1\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:   \"domain-id\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"/__cadence_sys/tl/2\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil)\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         2,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success - push failures are ignored\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockTaskManager.EXPECT().UpdateTaskList(gomock.Any(), &persistence.UpdateTaskListRequest{\n\t\t\t\t\tDomainName: \"domainName\",\n\t\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\t\tDomainID: \"domain-id\",\n\t\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\t\tRangeID:  0,\n\t\t\t\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\t\tReadPartitions:  persistencePartitions(3),\n\t\t\t\t\t\t\tWritePartitions: persistencePartitions(1),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(&persistence.UpdateTaskListResponse{}, nil)\n\t\t\t\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:   \"domain-id\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"/__cadence_sys/tl/1\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\t\t\t\tDomainUUID:   \"domain-id\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"/__cadence_sys/tl/2\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\t\t\tWritePartitions: partitions(1),\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"matching client error\"))\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         2,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to update\",\n\t\t\treq: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(1),\n\t\t\t},\n\t\t\toriginalConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\tsetupMocks: func(deps *mockDeps) {\n\t\t\t\tdeps.mockTaskManager.EXPECT().UpdateTaskList(gomock.Any(), &persistence.UpdateTaskListRequest{\n\t\t\t\t\tDomainName: \"domainName\",\n\t\t\t\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\t\t\t\tDomainID: \"domain-id\",\n\t\t\t\t\t\tName:     \"tl\",\n\t\t\t\t\t\tAckLevel: 0,\n\t\t\t\t\t\tRangeID:  0,\n\t\t\t\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\t\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\t\t\t\tVersion:         2,\n\t\t\t\t\t\t\tReadPartitions:  persistencePartitions(3),\n\t\t\t\t\t\t\tWritePartitions: persistencePartitions(1),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil, errors.New(\"some error\"))\n\t\t\t},\n\t\t\texpectedConfig: &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  partitions(3),\n\t\t\t\tWritePartitions: partitions(3),\n\t\t\t},\n\t\t\texpectError:   true,\n\t\t\texpectedError: \"some error\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttlID, err := NewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\t\t\trequire.NoError(t, err)\n\t\t\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\t\t\ttc.setupMocks(deps)\n\t\t\ttlm.partitionConfig = tc.originalConfig\n\t\t\ttlm.startWG.Done()\n\n\t\t\terr = tlm.UpdateTaskListPartitionConfig(context.Background(), tc.req)\n\t\t\tif tc.expectError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError)\n\t\t\t\tassert.Equal(t, int32(1), tlm.stopped)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tc.expectedConfig, tlm.TaskListPartitionConfig())\n\t\t})\n\t}\n}\n\nfunc TestRefreshTaskListPartitionConfigConcurrency(t *testing.T) {\n\ttlID, err := NewIdentifier(\"domain-id\", \"/__cadence_sys/tl/1\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\ttlm, _ := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\ttlm.startWG.Done()\n\n\tvar g errgroup.Group\n\tfor i := 0; i < 100; i++ {\n\t\tv := i\n\t\tg.Go(func() error {\n\t\t\treturn tlm.RefreshTaskListPartitionConfig(context.Background(), &types.TaskListPartitionConfig{Version: int64(v), ReadPartitions: partitions(v), WritePartitions: partitions(v)})\n\t\t})\n\t}\n\trequire.NoError(t, g.Wait())\n\tassert.Equal(t, int64(99), tlm.TaskListPartitionConfig().Version)\n}\n\nfunc TestUpdateTaskListPartitionConfigConcurrency(t *testing.T) {\n\ttlID, err := NewIdentifier(\"domain-id\", \"/__cadence_sys/tl/1\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\tdeps.mockTaskManager.EXPECT().UpdateTaskList(gomock.Any(), gomock.Any()).Return(&persistence.UpdateTaskListResponse{}, nil).AnyTimes()\n\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), gomock.Any()).Return(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil).AnyTimes()\n\ttlm.startWG.Done()\n\n\tvar g errgroup.Group\n\tfor i := 2; i < 102; i++ {\n\t\tv := i\n\t\tg.Go(func() error {\n\t\t\treturn tlm.UpdateTaskListPartitionConfig(context.Background(), &types.TaskListPartitionConfig{ReadPartitions: partitions(v), WritePartitions: partitions(v)})\n\t\t})\n\t}\n\trequire.NoError(t, g.Wait())\n\tassert.Equal(t, int64(100), tlm.TaskListPartitionConfig().Version)\n}\n\nfunc TestManagerStart_RootPartition(t *testing.T) {\n\ttlID, err := NewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\tdeps.mockTaskManager.EXPECT().LeaseTaskList(gomock.Any(), &persistence.LeaseTaskListRequest{\n\t\tDomainID:   \"domain-id\",\n\t\tDomainName: \"domainName\",\n\t\tTaskList:   \"tl\",\n\t\tTaskType:   persistence.TaskListTypeDecision,\n\t}).Return(&persistence.LeaseTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID: \"domain-id\",\n\t\t\tName:     \"tl\",\n\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\tAckLevel: 0,\n\t\t\tRangeID:  0,\n\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  persistencePartitions(2),\n\t\t\t\tWritePartitions: persistencePartitions(2),\n\t\t\t},\n\t\t},\n\t}, nil)\n\tdeps.mockMatchingClient.EXPECT().RefreshTaskListPartitionConfig(gomock.Any(), &types.MatchingRefreshTaskListPartitionConfigRequest{\n\t\tDomainUUID:   \"domain-id\",\n\t\tTaskList:     &types.TaskList{Name: \"/__cadence_sys/tl/1\", Kind: types.TaskListKindNormal.Ptr()},\n\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\tVersion:         1,\n\t\t\tReadPartitions:  partitions(2),\n\t\t\tWritePartitions: partitions(2),\n\t\t},\n\t}).Return(&types.MatchingRefreshTaskListPartitionConfigResponse{}, nil)\n\tassert.NoError(t, tlm.Start(context.Background()))\n\tassert.Equal(t, &types.TaskListPartitionConfig{Version: 1, ReadPartitions: partitions(2), WritePartitions: partitions(2)}, tlm.TaskListPartitionConfig())\n\ttlm.stopWG.Wait()\n}\n\nfunc TestManagerStart_NonRootPartition(t *testing.T) {\n\ttlID, err := NewIdentifier(\"domain-id\", \"/__cadence_sys/tl/1\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\tdeps.mockTaskManager.EXPECT().GetTaskList(gomock.Any(), &persistence.GetTaskListRequest{\n\t\tDomainID:   \"domain-id\",\n\t\tDomainName: \"domainName\",\n\t\tTaskList:   \"tl\",\n\t\tTaskType:   persistence.TaskListTypeDecision,\n\t}).Return(&persistence.GetTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID: \"domain-id\",\n\t\t\tName:     \"tl\",\n\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\tAckLevel: 0,\n\t\t\tRangeID:  0,\n\t\t\tAdaptivePartitionConfig: &persistence.TaskListPartitionConfig{\n\t\t\t\tVersion:         1,\n\t\t\t\tReadPartitions:  persistencePartitions(3),\n\t\t\t\tWritePartitions: persistencePartitions(3),\n\t\t\t},\n\t\t},\n\t}, nil)\n\tdeps.mockTaskManager.EXPECT().LeaseTaskList(gomock.Any(), &persistence.LeaseTaskListRequest{\n\t\tDomainID:   \"domain-id\",\n\t\tDomainName: \"domainName\",\n\t\tTaskList:   \"/__cadence_sys/tl/1\",\n\t\tTaskType:   persistence.TaskListTypeDecision,\n\t}).Return(&persistence.LeaseTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tDomainID: \"domain-id\",\n\t\t\tName:     \"/__cadence_sys/tl/1\",\n\t\t\tKind:     persistence.TaskListKindNormal,\n\t\t\tAckLevel: 0,\n\t\t\tRangeID:  0,\n\t\t},\n\t}, nil)\n\tassert.NoError(t, tlm.Start(context.Background()))\n\tassert.Equal(t, &types.TaskListPartitionConfig{\n\t\tVersion:         1,\n\t\tReadPartitions:  partitions(3),\n\t\tWritePartitions: partitions(3),\n\t}, tlm.TaskListPartitionConfig())\n}\n\nfunc TestDispatchTask(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                        string\n\t\tmockSetup                   func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask)\n\t\tenableStandByTaskCompletion bool\n\t\tactiveClusterName           string\n\t\terr                         error\n\t}{\n\t\t{\n\t\t\tname: \"active cluster - disabled StandByTaskCompletion - task sent to MustOffer and no error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(nil).Times(1)\n\t\t\t},\n\t\t\tactiveClusterName: cluster.TestCurrentClusterName,\n\t\t\terr:               nil,\n\t\t},\n\t\t{\n\t\t\tname: \"active cluster - disabled StandByTaskCompletion - task sent to MustOffer and error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(errors.New(\"no-task-completion-must-offer-error\")).Times(1)\n\t\t\t},\n\t\t\tactiveClusterName: cluster.TestCurrentClusterName,\n\t\t\terr:               errors.New(\"no-task-completion-must-offer-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"active cluster - enabled StandByTaskCompletion - task sent to MustOffer and no error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(nil).Times(1)\n\t\t\t},\n\t\t\tenableStandByTaskCompletion: true,\n\t\t\tactiveClusterName:           cluster.TestCurrentClusterName,\n\t\t\terr:                         nil,\n\t\t},\n\t\t{\n\t\t\tname: \"active cluster - enabled StandByTaskCompletion - task sent to MustOffer and error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(errors.New(\"task-completion-must-offer-error\")).Times(1)\n\t\t\t},\n\t\t\tenableStandByTaskCompletion: true,\n\t\t\tactiveClusterName:           cluster.TestCurrentClusterName,\n\t\t\terr:                         errors.New(\"task-completion-must-offer-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"standby cluster - disabled StandByTaskCompletion - task sent to MustOffer and no error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(nil).Times(1)\n\t\t\t},\n\t\t\tactiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\terr:               nil,\n\t\t},\n\t\t{\n\t\t\tname: \"standby cluster - disabled StandByTaskCompletion - task sent to MustOffer and error returned\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\tmatcher.EXPECT().MustOffer(ctx, task).Return(errors.New(\"no-task-completion-must-offer-error\")).Times(1)\n\t\t\t},\n\t\t\tactiveClusterName: cluster.TestAlternativeClusterName,\n\t\t\terr:               errors.New(\"no-task-completion-must-offer-error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"standby cluster - enabled StandByTaskCompletion - task completed\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\ttaskCompleter.EXPECT().CompleteTaskIfStarted(ctx, task).Return(nil).Times(1)\n\t\t\t},\n\t\t\tenableStandByTaskCompletion: true,\n\t\t\tactiveClusterName:           cluster.TestAlternativeClusterName,\n\t\t\terr:                         nil,\n\t\t},\n\t\t{\n\t\t\tname: \"standby cluster - enabled StandByTaskCompletion - task completion error\",\n\t\t\tmockSetup: func(matcher *MockTaskMatcher, taskCompleter *MockTaskCompleter, ctx context.Context, task *InternalTask) {\n\t\t\t\ttaskCompleter.EXPECT().CompleteTaskIfStarted(ctx, task).Return(errTaskNotStarted).Times(1)\n\t\t\t},\n\t\t\tenableStandByTaskCompletion: true,\n\t\t\tactiveClusterName:           cluster.TestAlternativeClusterName,\n\t\t\terr:                         errTaskNotStarted,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\t\t\ttlm := createTestTaskListManager(t, logger, controller)\n\n\t\t\ttask := &InternalTask{\n\t\t\t\tEvent: &genericTaskInfo{\n\t\t\t\t\tTaskInfo: &persistence.TaskInfo{\n\t\t\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\n\t\t\ttaskMatcher := NewMockTaskMatcher(controller)\n\t\t\ttaskCompleter := NewMockTaskCompleter(controller)\n\t\t\ttlm.matcher = taskMatcher\n\t\t\ttlm.taskCompleter = taskCompleter\n\t\t\ttlm.config.EnableStandbyTaskCompletion = func() bool {\n\t\t\t\treturn tc.enableStandByTaskCompletion\n\t\t\t}\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(controller)\n\t\t\tcacheEntry := cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t\t&persistence.DomainInfo{ID: constants.TestDomainID, Name: constants.TestDomainName},\n\t\t\t\t&persistence.DomainConfig{Retention: 1},\n\t\t\t\t&persistence.DomainReplicationConfig{\n\t\t\t\t\tActiveClusterName: tc.activeClusterName,\n\t\t\t\t\tClusters:          []*persistence.ClusterReplicationConfig{{ClusterName: cluster.TestCurrentClusterName}, {ClusterName: cluster.TestAlternativeClusterName}},\n\t\t\t\t},\n\t\t\t\t1,\n\t\t\t)\n\n\t\t\tmockDomainCache.EXPECT().GetDomainByID(constants.TestDomainID).Return(cacheEntry, nil).AnyTimes()\n\t\t\ttlm.domainCache = mockDomainCache\n\n\t\t\tctx := context.Background()\n\t\t\ttc.mockSetup(taskMatcher, taskCompleter, ctx, task)\n\n\t\t\terr := tlm.DispatchTask(ctx, task)\n\n\t\t\tif tc.err != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, tc.err, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGetNumPartitions(t *testing.T) {\n\ttlID, err := NewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\ttlm, deps := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\trequire.NoError(t, deps.dynamicClient.UpdateValue(dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache, true))\n\tassert.NotPanics(t, func() { tlm.limiter.ReportLimit(float64(100)) })\n}\n\nfunc TestDisconnectBlockedPollers(t *testing.T) {\n\ttests := []struct {\n\t\tname                 string\n\t\tnewActiveClusterName *string\n\t\tmockSetup            func(mockMatcher *MockTaskMatcher)\n\t\tstopped              int32\n\t\texpectedErr          error\n\t}{\n\t\t{\n\t\t\tname:                 \"disconnect blocked pollers and refresh cancel context\",\n\t\t\tnewActiveClusterName: common.StringPtr(\"new-active-cluster\"),\n\t\t\tmockSetup: func(mockMatcher *MockTaskMatcher) {\n\t\t\t\tmockMatcher.EXPECT().DisconnectBlockedPollers().Times(1)\n\t\t\t\tmockMatcher.EXPECT().RefreshCancelContext().Times(1)\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t{\n\t\t\tname:                 \"tasklist manager is shutting down, noop\",\n\t\t\tnewActiveClusterName: common.StringPtr(\"new-active-cluster\"),\n\t\t\tmockSetup:            func(mockMatcher *MockTaskMatcher) {},\n\t\t\tstopped:              int32(1),\n\t\t\texpectedErr:          errShutdown,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttlID, err := NewIdentifier(\"domain-id\", \"tl\", persistence.TaskListTypeDecision)\n\t\t\trequire.NoError(t, err)\n\n\t\t\ttlm, _ := setupMocksForTaskListManager(t, tlID, types.TaskListKindNormal)\n\n\t\t\tmockMatcher := NewMockTaskMatcher(gomock.NewController(t))\n\t\t\ttlm.matcher = mockMatcher\n\n\t\t\ttc.mockSetup(mockMatcher)\n\n\t\t\ttlm.stopped = tc.stopped\n\n\t\t\terr = tlm.ReleaseBlockedPollers()\n\n\t\t\tassert.Equal(t, tc.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc partitions(num int) map[int]*types.TaskListPartition {\n\tresult := make(map[int]*types.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tresult[i] = &types.TaskListPartition{}\n\t}\n\treturn result\n}\n\nfunc persistencePartitions(num int) map[int]*persistence.TaskListPartition {\n\tresult := make(map[int]*persistence.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tresult[i] = &persistence.TaskListPartition{}\n\t}\n\treturn result\n}\n\nfunc TestTaskListUsesOverrideRPS(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                 string\n\t\toverrideRPS          float64\n\t\tmaxDispatchPerSecond *float64\n\t\texpectedRPS          float64\n\t}{\n\t\t{\n\t\t\tname:                 \"Uses maxDispatchPerSecond when overrideRPS is zero\",\n\t\t\toverrideRPS:          0.0,\n\t\t\tmaxDispatchPerSecond: common.Float64Ptr(75.0),\n\t\t\texpectedRPS:          75.0,\n\t\t},\n\t\t{\n\t\t\tname:                 \"Uses maxDispatchPerSecond when overrideRPS is negative\",\n\t\t\toverrideRPS:          -1.0,\n\t\t\tmaxDispatchPerSecond: common.Float64Ptr(60.0),\n\t\t\texpectedRPS:          60.0,\n\t\t},\n\t\t{\n\t\t\tname:                 \"OverrideRPS takes precedence when higher than maxDispatchPerSecond\",\n\t\t\toverrideRPS:          200.0,\n\t\t\tmaxDispatchPerSecond: common.Float64Ptr(50.0),\n\t\t\texpectedRPS:          200.0,\n\t\t},\n\t\t{\n\t\t\tname:                 \"OverrideRPS takes precedence when lower than maxDispatchPerSecond\",\n\t\t\toverrideRPS:          25.0,\n\t\t\tmaxDispatchPerSecond: common.Float64Ptr(100.0),\n\t\t\texpectedRPS:          25.0,\n\t\t},\n\t\t{\n\t\t\tname:                 \"No maxDispatchPerSecond and no overrideRPS\",\n\t\t\toverrideRPS:          0.0,\n\t\t\tmaxDispatchPerSecond: nil,\n\t\t\texpectedRPS:          100000.0,\n\t\t},\n\t\t{\n\t\t\tname:                 \"OverrideRPS only, with no maxDispatchPerSecond\",\n\t\t\toverrideRPS:          150.0,\n\t\t\tmaxDispatchPerSecond: nil,\n\t\t\texpectedRPS:          150.0,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\tlogger := testlogger.New(t)\n\n\t\t\tcfg := defaultTestConfig()\n\t\t\tcfg.OverrideTaskListRPS = func(domain, taskList string, taskType int) float64 {\n\t\t\t\treturn tc.overrideRPS\n\t\t\t}\n\n\t\t\ttlm := createTestTaskListManagerWithConfig(t, logger, controller, cfg, clock.NewMockedTimeSource())\n\n\t\t\t_, _ = tlm.GetTask(context.Background(), tc.maxDispatchPerSecond)\n\n\t\t\tassert.Equal(t, rate.Limit(tc.expectedRPS), tlm.limiter.Limit())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_registry.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"sync\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype taskListRegistryImpl struct {\n\tsync.RWMutex\n\ttaskLists map[Identifier]Manager\n\n\t// task lists indexed by domain ID and task list name\n\t// to quickly get all task lists for a domain or a task list name\n\t// they are always in sync with the taskLists map\n\ttaskListsByDomainID     map[string]map[Identifier]Manager\n\ttaskListsByTaskListName map[string]map[Identifier]Manager\n\n\tmetricsClient metrics.Client\n}\n\nfunc NewTaskListRegistry(metricsClient metrics.Client) TaskListRegistry {\n\treturn &taskListRegistryImpl{\n\t\ttaskLists:               make(map[Identifier]Manager),\n\t\ttaskListsByDomainID:     make(map[string]map[Identifier]Manager),\n\t\ttaskListsByTaskListName: make(map[string]map[Identifier]Manager),\n\t\tmetricsClient:           metricsClient,\n\t}\n}\n\nfunc (r *taskListRegistryImpl) Register(id Identifier, mgr Manager) {\n\tr.Lock()\n\tdefer r.Unlock()\n\n\t// we can override the manager for the same identifier if it is already registered\n\t// this case should be handled by the caller\n\tr.taskLists[id] = mgr\n\n\tif _, ok := r.taskListsByDomainID[id.GetDomainID()]; !ok {\n\t\tr.taskListsByDomainID[id.GetDomainID()] = make(map[Identifier]Manager)\n\t}\n\tif _, ok := r.taskListsByTaskListName[id.GetName()]; !ok {\n\t\tr.taskListsByTaskListName[id.GetName()] = make(map[Identifier]Manager)\n\t}\n\n\tr.taskListsByDomainID[id.GetDomainID()][id] = mgr\n\tr.taskListsByTaskListName[id.GetName()][id] = mgr\n\tr.updateMetricsLocked()\n}\n\nfunc (r *taskListRegistryImpl) Unregister(mgr Manager) bool {\n\tid := mgr.TaskListID()\n\tr.Lock()\n\tdefer r.Unlock()\n\n\t// we need to make sure we still hold the given `mgr` or we already replaced with a new one.\n\tcurrentTlMgr, ok := r.taskLists[*id]\n\tif ok && currentTlMgr == mgr {\n\t\tdelete(r.taskLists, *id)\n\n\t\tdelete(r.taskListsByDomainID[id.GetDomainID()], *id)\n\t\tif len(r.taskListsByDomainID[id.GetDomainID()]) == 0 {\n\t\t\tdelete(r.taskListsByDomainID, id.GetDomainID())\n\t\t}\n\n\t\tdelete(r.taskListsByTaskListName[id.GetName()], *id)\n\t\tif len(r.taskListsByTaskListName[id.GetName()]) == 0 {\n\t\t\tdelete(r.taskListsByTaskListName, id.GetName())\n\t\t}\n\n\t\tr.updateMetricsLocked()\n\t\treturn true\n\t}\n\n\treturn false\n}\n\nfunc (r *taskListRegistryImpl) ManagerByTaskListIdentifier(id Identifier) (Manager, bool) {\n\tr.RLock()\n\tdefer r.RUnlock()\n\n\ttlMgr, ok := r.taskLists[id]\n\treturn tlMgr, ok\n}\n\nfunc (r *taskListRegistryImpl) AllManagers() []Manager {\n\tr.RLock()\n\tdefer r.RUnlock()\n\n\tres := make([]Manager, 0, len(r.taskLists))\n\tfor _, tlMgr := range r.taskLists {\n\t\tres = append(res, tlMgr)\n\t}\n\treturn res\n}\n\nfunc (r *taskListRegistryImpl) ManagersByDomainID(domainID string) []Manager {\n\tr.RLock()\n\tdefer r.RUnlock()\n\n\tvar res []Manager\n\tfor _, tlm := range r.taskListsByDomainID[domainID] {\n\t\tres = append(res, tlm)\n\t}\n\treturn res\n}\n\nfunc (r *taskListRegistryImpl) ManagersByTaskListName(name string) []Manager {\n\tr.RLock()\n\tdefer r.RUnlock()\n\n\tvar res []Manager\n\tfor _, tlm := range r.taskListsByTaskListName[name] {\n\t\tres = append(res, tlm)\n\t}\n\treturn res\n}\n\nfunc (r *taskListRegistryImpl) updateMetricsLocked() {\n\tr.metricsClient.Scope(metrics.MatchingTaskListMgrScope).UpdateGauge(\n\t\tmetrics.TaskListManagersGauge,\n\t\tfloat64(len(r.taskLists)),\n\t)\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_list_registry_test.go",
    "content": "package tasklist\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\tmetricsmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc mustNewIdentifierForTest(t *testing.T, domainID, taskListName string) *Identifier {\n\tt.Helper()\n\tid, err := NewIdentifier(domainID, taskListName, persistence.TaskListTypeDecision)\n\trequire.NoError(t, err)\n\treturn id\n}\n\nfunc newMockManagerWithID(t *testing.T, ctrl *gomock.Controller, id *Identifier) *MockManager {\n\tt.Helper()\n\tmgr := NewMockManager(ctrl)\n\tmgr.EXPECT().TaskListID().Return(id).AnyTimes()\n\treturn mgr\n}\n\nfunc TestTaskListRegistry_RegisterLookupAndUnregister(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmetricsClient := metricsmocks.Client{}\n\tmetricsScope := metricsmocks.Scope{}\n\tmetricsClient.On(\"Scope\", metrics.MatchingTaskListMgrScope).Return(&metricsScope)\n\tregistry := NewTaskListRegistry(&metricsClient)\n\n\tid := mustNewIdentifierForTest(t, \"domain-a\", \"task-list-a\")\n\n\tinitialMgr := newMockManagerWithID(t, ctrl, id)\n\tupdatedMgr := newMockManagerWithID(t, ctrl, id)\n\n\tmetricsScope.On(\"UpdateGauge\", metrics.TaskListManagersGauge, float64(1)).Once()\n\tregistry.Register(*id, initialMgr)\n\tgot, ok := registry.ManagerByTaskListIdentifier(*id)\n\trequire.True(t, ok)\n\tassert.Equal(t, initialMgr, got)\n\n\t// Re-register with the same identifier should replace the manager.\n\tmetricsScope.On(\"UpdateGauge\", metrics.TaskListManagersGauge, float64(1)).Once()\n\tregistry.Register(*id, updatedMgr)\n\tgot, ok = registry.ManagerByTaskListIdentifier(*id)\n\trequire.True(t, ok)\n\tassert.Equal(t, updatedMgr, got)\n\n\t// Unregister should not remove a replaced/stale manager.\n\tassert.False(t, registry.Unregister(initialMgr))\n\tgot, ok = registry.ManagerByTaskListIdentifier(*id)\n\trequire.True(t, ok)\n\tassert.Equal(t, updatedMgr, got)\n\n\t// Unregistering the current manager should remove the entry.\n\tmetricsScope.On(\"UpdateGauge\", metrics.TaskListManagersGauge, float64(0)).Once()\n\tassert.True(t, registry.Unregister(updatedMgr))\n\t_, ok = registry.ManagerByTaskListIdentifier(*id)\n\tassert.False(t, ok)\n\n\timpl := registry.(*taskListRegistryImpl)\n\tassert.Empty(t, impl.taskListsByDomainID, \"should be empty because there are no other task lists for this domain\")\n\tassert.Empty(t, impl.taskListsByTaskListName, \"should be empty because there are no other task lists for this task list name\")\n\n\tmetricsClient.AssertExpectations(t)\n\tmetricsScope.AssertExpectations(t)\n}\n\nfunc TestTaskListRegistry_Filters(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tregistry := NewTaskListRegistry(metrics.NewNoopMetricsClient())\n\n\tdomainA1 := mustNewIdentifierForTest(t, \"domain-a\", \"shared-name\")\n\tdomainA2 := mustNewIdentifierForTest(t, \"domain-a\", \"other-name\")\n\tdomainB1 := mustNewIdentifierForTest(t, \"domain-b\", \"shared-name\")\n\n\tmgrA1 := newMockManagerWithID(t, ctrl, domainA1)\n\tmgrA2 := newMockManagerWithID(t, ctrl, domainA2)\n\tmgrB1 := newMockManagerWithID(t, ctrl, domainB1)\n\n\tregistry.Register(*domainA1, mgrA1)\n\tregistry.Register(*domainA2, mgrA2)\n\tregistry.Register(*domainB1, mgrB1)\n\n\tt.Run(\"all managers\", func(t *testing.T) {\n\t\tassert.ElementsMatch(t, []Manager{mgrA1, mgrA2, mgrB1}, registry.AllManagers())\n\t})\n\n\tt.Run(\"managers by domain\", func(t *testing.T) {\n\t\tassert.ElementsMatch(t, []Manager{mgrA1, mgrA2}, registry.ManagersByDomainID(\"domain-a\"))\n\t\tassert.ElementsMatch(t, []Manager{mgrB1}, registry.ManagersByDomainID(\"domain-b\"))\n\t\tassert.Empty(t, registry.ManagersByDomainID(\"missing-domain\"))\n\t})\n\n\tt.Run(\"managers by task list name\", func(t *testing.T) {\n\t\tassert.ElementsMatch(\n\t\t\tt,\n\t\t\t[]Manager{mgrA1, mgrB1},\n\t\t\tregistry.ManagersByTaskListName(\"shared-name\"),\n\t\t)\n\t\tassert.ElementsMatch(t, []Manager{mgrA2}, registry.ManagersByTaskListName(\"other-name\"))\n\t\tassert.Empty(t, registry.ManagersByTaskListName(\"missing-name\"))\n\t})\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_reader.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"runtime\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\nvar epochStartTime = time.Unix(0, 0)\n\nconst (\n\tdefaultTaskBufferIsolationGroup = \"\" // a task buffer which is not using an isolation group\n)\n\ntype (\n\ttaskReader struct {\n\n\t\t// taskBuffers: This is the in-memory queue of tasks for dispatch\n\t\t// that are enqueued for pollers to pickup. It's written to by\n\t\t// - getTasksPump - the primary means of loading async matching tasks\n\t\t// - task dispatch redirection - when a task is redirected from another isolation group\n\t\ttaskBuffers     map[string]chan *persistence.TaskInfo\n\t\tnotifyC         chan struct{} // Used as signal to notify pump of new tasks\n\t\ttlMgr           *taskListManagerImpl\n\t\ttaskListID      *Identifier\n\t\tconfig          *config.TaskListConfig\n\t\tdb              *taskListDB\n\t\ttaskWriter      *taskWriter\n\t\ttaskGC          *taskGC\n\t\ttaskAckManager  messaging.AckManager\n\t\tdomainCache     cache.DomainCache\n\t\tclusterMetadata cluster.Metadata\n\t\ttimeSource      clock.TimeSource\n\t\t// The cancel objects are to cancel the ratelimiter Wait in dispatchBufferedTasks. The ideal\n\t\t// approach is to use request-scoped contexts and use a unique one for each call to Wait. However\n\t\t// in order to cancel it on shutdown, we need a new goroutine for each call that would wait on\n\t\t// the shutdown channel. To optimize on efficiency, we instead create one and tag it on the struct\n\t\t// so the cancel can be called directly on shutdown.\n\t\tcancelCtx                context.Context\n\t\tcancelFunc               context.CancelFunc\n\t\tstopped                  int64 // set to 1 if the reader is stopped or is shutting down\n\t\tlogger                   log.Logger\n\t\tscope                    metrics.Scope\n\t\tthrottleRetry            *backoff.ThrottleRetry\n\t\thandleErr                func(error) error\n\t\tonFatalErr               func()\n\t\tdispatchTask             func(context.Context, *InternalTask) error\n\t\tgetIsolationGroupForTask func(context.Context, *persistence.TaskInfo) (string, time.Duration)\n\t\trateLimit                func() rate.Limit\n\n\t\t// stopWg is used to wait for all dispatchers to stop.\n\t\tstopWg sync.WaitGroup\n\t}\n)\n\nfunc newTaskReader(tlMgr *taskListManagerImpl, isolationGroups []string) *taskReader {\n\tctx, cancel := context.WithCancel(context.Background())\n\ttaskBuffers := make(map[string]chan *persistence.TaskInfo)\n\n\t// Validate batch size to prevent system failures\n\tbatchSize := tlMgr.config.GetTasksBatchSize()\n\tif batchSize <= 0 {\n\t\tfallback := dynamicproperties.IntKeys[dynamicproperties.MatchingGetTasksBatchSize].DefaultValue\n\t\ttlMgr.logger.Warn(\"matching.getTasksBatchSize is set to invalid value, using default value\",\n\t\t\ttag.Dynamic(\"invalidBatchSize\", batchSize),\n\t\t\ttag.Dynamic(\"correctedBatchSize\", fallback))\n\t\tbatchSize = fallback\n\t}\n\n\ttaskBuffers[defaultTaskBufferIsolationGroup] = make(chan *persistence.TaskInfo, batchSize-1)\n\tfor _, g := range isolationGroups {\n\t\ttaskBuffers[g] = make(chan *persistence.TaskInfo, batchSize-1)\n\t}\n\treturn &taskReader{\n\t\ttlMgr:          tlMgr,\n\t\ttaskListID:     tlMgr.taskListID,\n\t\tconfig:         tlMgr.config,\n\t\tdb:             tlMgr.db,\n\t\ttaskWriter:     tlMgr.taskWriter,\n\t\ttaskGC:         tlMgr.taskGC,\n\t\ttaskAckManager: tlMgr.taskAckManager,\n\t\tcancelCtx:      ctx,\n\t\tcancelFunc:     cancel,\n\t\tnotifyC:        make(chan struct{}, 1),\n\t\t// we always dequeue the head of the buffer and try to dispatch it to a poller\n\t\t// so allocate one less than desired target buffer size\n\t\ttaskBuffers:              taskBuffers,\n\t\tdomainCache:              tlMgr.domainCache,\n\t\tclusterMetadata:          tlMgr.clusterMetadata,\n\t\ttimeSource:               tlMgr.timeSource,\n\t\tlogger:                   tlMgr.logger,\n\t\tscope:                    tlMgr.scope,\n\t\thandleErr:                tlMgr.handleErr,\n\t\tonFatalErr:               tlMgr.Stop,\n\t\tdispatchTask:             tlMgr.DispatchTask,\n\t\tgetIsolationGroupForTask: tlMgr.getIsolationGroupForTask,\n\t\trateLimit:                tlMgr.limiter.Limit,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(persistenceOperationRetryPolicy),\n\t\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t\t),\n\t}\n}\n\nfunc (tr *taskReader) Start() {\n\ttr.Signal()\n\tfor g := range tr.taskBuffers {\n\t\tg := g\n\t\ttr.stopWg.Add(1)\n\t\tgo func() {\n\t\t\tdefer tr.stopWg.Done()\n\t\t\ttr.dispatchBufferedTasks(g)\n\t\t}()\n\t}\n\ttr.stopWg.Add(1)\n\tgo func() {\n\t\tdefer tr.stopWg.Done()\n\t\ttr.getTasksPump()\n\t}()\n}\n\nfunc (tr *taskReader) Stop() {\n\tif atomic.CompareAndSwapInt64(&tr.stopped, 0, 1) {\n\t\ttr.cancelFunc()\n\t\tif err := tr.persistAckLevel(); err != nil {\n\t\t\ttr.logger.Error(\"Persistent store operation failure\",\n\t\t\t\ttag.StoreOperationUpdateTaskList,\n\t\t\t\ttag.Error(err))\n\t\t}\n\t\ttr.taskGC.RunNow(tr.taskAckManager.GetAckLevel())\n\t\ttr.stopWg.Wait()\n\t}\n}\n\nfunc (tr *taskReader) Signal() {\n\tvar event struct{}\n\tselect {\n\tcase tr.notifyC <- event:\n\tdefault: // channel already has an event, don't block\n\t}\n}\n\nfunc (tr *taskReader) dispatchBufferedTasks(isolationGroup string) {\ndispatchLoop:\n\tfor {\n\t\tselect {\n\t\tcase taskInfo, ok := <-tr.taskBuffers[isolationGroup]:\n\t\t\tif !ok { // Task list getTasks pump is shutdown\n\t\t\t\tbreak dispatchLoop\n\t\t\t}\n\t\t\tevent.Log(event.E{\n\t\t\t\tTaskListName: tr.taskListID.GetName(),\n\t\t\t\tTaskListType: tr.taskListID.GetType(),\n\t\t\t\tTaskListKind: &tr.tlMgr.taskListKind,\n\t\t\t\tTaskInfo:     *taskInfo,\n\t\t\t\tEventName:    \"Attempting to Dispatch Buffered Task\",\n\t\t\t})\n\t\t\tbreakDispatchLoop := tr.dispatchSingleTaskFromBufferWithRetries(taskInfo)\n\t\t\tif breakDispatchLoop {\n\t\t\t\t// shutting down\n\t\t\t\tbreak dispatchLoop\n\t\t\t}\n\t\tcase <-tr.cancelCtx.Done():\n\t\t\tbreak dispatchLoop\n\t\t}\n\t}\n}\n\nfunc (tr *taskReader) getTasksPump() {\n\tupdateAckTimer := tr.timeSource.NewTimer(tr.config.UpdateAckInterval())\n\tdefer updateAckTimer.Stop()\ngetTasksPumpLoop:\n\tfor {\n\t\ttr.scope.UpdateGauge(metrics.TaskBacklogPerTaskListGauge, float64(tr.taskAckManager.GetBacklogCount()))\n\t\tselect {\n\t\tcase <-tr.cancelCtx.Done():\n\t\t\tbreak getTasksPumpLoop\n\t\tcase <-tr.notifyC:\n\t\t\t{\n\t\t\t\tinitialReadLevel := tr.taskAckManager.GetReadLevel()\n\t\t\t\tmaxReadLevel := tr.taskWriter.GetMaxReadLevel()\n\n\t\t\t\ttasks, readLevel, isReadBatchDone, err := tr.getTaskBatch(initialReadLevel, maxReadLevel)\n\t\t\t\tif err != nil {\n\t\t\t\t\ttr.Signal() // re-enqueue the event\n\t\t\t\t\t// TODO: Should we ever stop retrying on db errors?\n\t\t\t\t\tcontinue getTasksPumpLoop\n\t\t\t\t}\n\n\t\t\t\tif len(tasks) == 0 {\n\t\t\t\t\ttr.taskAckManager.SetReadLevel(readLevel)\n\n\t\t\t\t\tif tr.taskAckManager.GetAckLevel() == initialReadLevel {\n\t\t\t\t\t\t// Even though we didn't handle any tasks, we want to advance the ack-level\n\t\t\t\t\t\t// in order to avoid needless querying database the next time.\n\t\t\t\t\t\t// This is safe since we started reading exactly from the current AckLevel and read no tasks\n\t\t\t\t\t\ttr.taskAckManager.SetAckLevel(readLevel)\n\t\t\t\t\t}\n\n\t\t\t\t\tif !isReadBatchDone {\n\t\t\t\t\t\ttr.Signal()\n\t\t\t\t\t}\n\t\t\t\t\tcontinue getTasksPumpLoop\n\t\t\t\t}\n\n\t\t\t\tif !tr.addTasksToBuffer(tasks) {\n\t\t\t\t\tbreak getTasksPumpLoop\n\t\t\t\t}\n\t\t\t\t// There maybe more tasks. We yield now, but signal pump to check again later.\n\t\t\t\ttr.Signal()\n\t\t\t}\n\t\tcase <-updateAckTimer.Chan():\n\t\t\t{\n\t\t\t\tackLevel := tr.taskAckManager.GetAckLevel()\n\t\t\t\tif size, err := tr.db.GetTaskListSize(ackLevel); err == nil {\n\t\t\t\t\ttr.scope.UpdateGauge(metrics.TaskCountPerTaskListGauge, float64(size))\n\t\t\t\t}\n\t\t\t\tif err := tr.handleErr(tr.persistAckLevel()); err != nil {\n\t\t\t\t\ttr.logger.Error(\"Persistent store operation failure\",\n\t\t\t\t\t\ttag.StoreOperationUpdateTaskList,\n\t\t\t\t\t\ttag.Error(err))\n\t\t\t\t\t// keep going as saving ack is not critical\n\t\t\t\t}\n\t\t\t\ttr.Signal() // periodically signal pump to check persistence for tasks\n\t\t\t\tupdateAckTimer.Reset(tr.config.UpdateAckInterval())\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (tr *taskReader) getTaskBatchWithRange(readLevel int64, maxReadLevel int64) ([]*persistence.TaskInfo, error) {\n\tvar response *persistence.GetTasksResponse\n\n\t// Validate batch size to prevent requesting 0 tasks\n\tbatchSize := tr.config.GetTasksBatchSize()\n\tif batchSize <= 0 {\n\t\tfallback := dynamicproperties.IntKeys[dynamicproperties.MatchingGetTasksBatchSize].DefaultValue\n\t\ttr.logger.Warn(\"matching.getTasksBatchSize is set to invalid value, using default value\",\n\t\t\ttag.Dynamic(\"invalidBatchSize\", batchSize),\n\t\t\ttag.Dynamic(\"correctedBatchSize\", fallback))\n\t\tbatchSize = fallback\n\t}\n\n\top := func(ctx context.Context) (err error) {\n\t\tresponse, err = tr.db.GetTasks(readLevel, maxReadLevel, batchSize)\n\t\treturn\n\t}\n\terr := tr.throttleRetry.Do(context.Background(), op)\n\tif err != nil {\n\t\ttr.logger.Error(\"Persistent store operation failure\",\n\t\t\ttag.StoreOperationGetTasks,\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowTaskListName(tr.taskListID.GetName()),\n\t\t\ttag.WorkflowTaskListType(tr.taskListID.GetType()))\n\t\treturn nil, err\n\t}\n\treturn response.Tasks, nil\n}\n\n// Returns a batch of tasks from persistence starting form current read level.\n// Also return a number that can be used to update readLevel\n// Also return a bool to indicate whether read is finished\nfunc (tr *taskReader) getTaskBatch(readLevel, maxReadLevel int64) ([]*persistence.TaskInfo, int64, bool, error) {\n\tvar tasks []*persistence.TaskInfo\n\n\t// counter i is used to break and let caller check whether tasklist is still alive and need resume read.\n\tfor i := 0; i < 10 && readLevel < maxReadLevel; i++ {\n\t\tupper := readLevel + int64(tr.config.ReadRangeSize())\n\t\tif upper > maxReadLevel {\n\t\t\tupper = maxReadLevel\n\t\t}\n\t\ttasks, err := tr.getTaskBatchWithRange(readLevel, upper)\n\t\tif err != nil {\n\t\t\treturn nil, readLevel, true, err\n\t\t}\n\t\t// return as long as it grabs any tasks\n\t\tif len(tasks) > 0 {\n\t\t\treturn tasks, upper, true, nil\n\t\t}\n\t\treadLevel = upper\n\t}\n\treturn tasks, readLevel, readLevel == maxReadLevel, nil // caller will update readLevel when no task grabbed\n}\n\nfunc (tr *taskReader) isTaskExpired(t *persistence.TaskInfo) bool {\n\treturn !t.Expiry.IsZero() && t.Expiry.After(epochStartTime) && tr.timeSource.Now().After(t.Expiry)\n}\n\nfunc (tr *taskReader) addTasksToBuffer(tasks []*persistence.TaskInfo) bool {\n\tfor _, t := range tasks {\n\t\tif !tr.addSingleTaskToBuffer(t) {\n\t\t\treturn false // we are shutting down the task list\n\t\t}\n\t}\n\treturn true\n}\n\nfunc (tr *taskReader) addSingleTaskToBuffer(task *persistence.TaskInfo) bool {\n\tif tr.isTaskExpired(task) {\n\t\ttr.scope.IncCounter(metrics.ExpiredTasksPerTaskListCounter)\n\t\t// Also increment readLevel for expired tasks otherwise it could result in\n\t\t// looping over the same tasks if all tasks read in the batch are expired\n\t\ttr.taskAckManager.SetReadLevel(task.TaskID)\n\t\treturn true\n\t}\n\terr := tr.taskAckManager.ReadItem(task.TaskID)\n\tif err != nil {\n\t\ttr.logger.Fatal(\"critical bug when adding item to ackManager\", tag.Error(err))\n\t}\n\t// Ignore the isolation duration as we're just putting it into a buffer to be dispatched later.\n\tisolationGroup, _ := tr.getIsolationGroupForTask(tr.cancelCtx, task)\n\tbuffer, ok := tr.taskBuffers[isolationGroup]\n\tif !ok {\n\t\tbuffer = tr.taskBuffers[defaultTaskBufferIsolationGroup]\n\t}\n\tselect {\n\tcase buffer <- task:\n\t\treturn true\n\tcase <-tr.cancelCtx.Done():\n\t\treturn false\n\t}\n}\n\nfunc (tr *taskReader) persistAckLevel() error {\n\tackLevel := tr.taskAckManager.GetAckLevel()\n\tif ackLevel >= 0 {\n\t\tmaxReadLevel := tr.taskWriter.GetMaxReadLevel()\n\t\t// note: this metrics is only an estimation for the lag. taskID in DB may not be continuous,\n\t\t// especially when task list ownership changes.\n\t\ttr.scope.UpdateGauge(metrics.TaskLagPerTaskListGauge, float64(maxReadLevel-ackLevel))\n\n\t\treturn tr.db.UpdateState(ackLevel)\n\t}\n\treturn nil\n}\n\n// completeTask marks a task as processed. Only tasks created by taskReader (i.e. backlog from db) reach\n// here. As part of completion:\n//   - task is deleted from the database when err is nil\n//   - new task is created and current task is deleted when err is not nil\nfunc (tr *taskReader) completeTask(task *persistence.TaskInfo, err error) {\n\tif err != nil {\n\t\t// failed to start the task.\n\t\t// We cannot just remove it from persistence because then it will be lost.\n\t\t// We handle this by writing the task back to persistence with a higher taskID.\n\t\t// This will allow subsequent tasks to make progress, and hopefully by the time this task is picked-up\n\t\t// again the underlying reason for failing to start will be resolved.\n\t\t// Note that RecordTaskStarted only fails after retrying for a long time, so a single task will not be\n\t\t// re-written to persistence frequently.\n\t\top := func(ctx context.Context) error {\n\t\t\t_, err := tr.taskWriter.appendTask(ctx, task)\n\t\t\treturn err\n\t\t}\n\t\terr = tr.throttleRetry.Do(context.Background(), op)\n\t\tif err != nil {\n\t\t\t// OK, we also failed to write to persistence.\n\t\t\t// This should only happen in very extreme cases where persistence is completely down.\n\t\t\t// We still can't lose the old task so we just unload the entire task list\n\t\t\ttr.logger.Error(\"Failed to complete task\", tag.Error(err))\n\t\t\ttr.onFatalErr()\n\t\t\treturn\n\t\t}\n\t\ttr.Signal()\n\t}\n\tackLevel := tr.taskAckManager.AckItem(task.TaskID)\n\ttr.taskGC.Run(ackLevel)\n}\n\nfunc (tr *taskReader) newDispatchContext(isolationGroup string, isolationDuration time.Duration) (context.Context, context.CancelFunc) {\n\trps := float64(tr.rateLimit())\n\tif isolationGroup != \"\" || rps > 1e-7 { // 1e-7 is a random number chosen to avoid overflow, normally user don't set such a low rps\n\t\ttimeout := tr.getDispatchTimeout(rps, isolationDuration)\n\t\tdomainEntry, err := tr.domainCache.GetDomainByID(tr.taskListID.GetDomainID())\n\t\tif err != nil {\n\t\t\t// we don't know if the domain is active in the current cluster, assume it is active and set the timeout\n\t\t\treturn context.WithTimeout(tr.cancelCtx, timeout)\n\t\t}\n\t\tif domainEntry.IsActiveIn(tr.clusterMetadata.GetCurrentClusterName()) {\n\t\t\ttr.logger.Debug(\"domain is active in the current cluster, setting timeout\",\n\t\t\t\ttag.WorkflowDomainName(domainEntry.GetInfo().Name),\n\t\t\t)\n\t\t\t// if the domain is active in the current cluster, set the timeout\n\t\t\treturn context.WithTimeout(tr.cancelCtx, timeout)\n\t\t}\n\t}\n\treturn tr.cancelCtx, func() {}\n}\n\nfunc (tr *taskReader) getDispatchTimeout(rps float64, isolationDuration time.Duration) time.Duration {\n\t// this is the minimum timeout required to dispatch a task, if the timeout value is smaller than this\n\t// async task dispatch can be completely throttled, which could happen when ratePerSecond is pretty low\n\tminTimeout := time.Duration(float64(len(tr.taskBuffers))/rps) * time.Second\n\t// timeout = max (min(asyncDispatchTimeout, isolationDuration), minTimeout)\n\ttimeout := tr.config.AsyncTaskDispatchTimeout()\n\tif timeout > isolationDuration && isolationDuration != noIsolationTimeout {\n\t\ttimeout = isolationDuration\n\t}\n\tif timeout < minTimeout {\n\t\ttimeout = minTimeout\n\t}\n\treturn timeout\n}\n\nfunc (tr *taskReader) dispatchSingleTaskFromBufferWithRetries(taskInfo *persistence.TaskInfo) (breakDispatchLoop bool) {\n\t// retry loop for dispatching a single task\n\tfor {\n\t\tbreakDispatchLoop, breakRetryLoop := tr.dispatchSingleTaskFromBuffer(taskInfo)\n\t\tif breakRetryLoop {\n\t\t\treturn breakDispatchLoop\n\t\t}\n\t}\n}\n\nfunc (tr *taskReader) dispatchSingleTaskFromBuffer(taskInfo *persistence.TaskInfo) (breakDispatchLoop bool, breakRetries bool) {\n\te := event.E{\n\t\tTaskListName: tr.taskListID.GetName(),\n\t\tTaskListType: tr.taskListID.GetType(),\n\t\tTaskListKind: &tr.tlMgr.taskListKind,\n\t\tTaskInfo:     *taskInfo,\n\t}\n\tif tr.isTaskExpired(taskInfo) {\n\t\te.EventName = \"Task Expired\"\n\t\tevent.Log(e)\n\t\ttr.scope.IncCounter(metrics.ExpiredTasksPerTaskListCounter)\n\t\ttr.taskAckManager.AckItem(taskInfo.TaskID)\n\t\treturn false, true\n\t}\n\tisolationGroup, isolationDuration := tr.getIsolationGroupForTask(tr.cancelCtx, taskInfo)\n\t_, isolationGroupIsKnown := tr.taskBuffers[isolationGroup]\n\tif !isolationGroupIsKnown {\n\t\tisolationGroup = defaultTaskBufferIsolationGroup\n\t\tisolationDuration = noIsolationTimeout\n\t}\n\ttask := newInternalTask(taskInfo, tr.completeTask, types.TaskSourceDbBacklog, \"\", false, nil, isolationGroup)\n\tdispatchCtx, cancel := tr.newDispatchContext(isolationGroup, isolationDuration)\n\ttimerScope := tr.scope.StartTimer(metrics.AsyncMatchLatencyPerTaskList)\n\terr := tr.dispatchTask(dispatchCtx, task)\n\ttimerScope.Stop()\n\tcancel()\n\n\tif err == nil {\n\t\te.EventName = \"Dispatched Buffered Task\"\n\t\tevent.Log(e)\n\t\treturn false, true\n\t}\n\n\tif errors.Is(err, context.Canceled) {\n\t\te.EventName = \"Dispatch Failed because Context Cancelled\"\n\t\tevent.Log(e)\n\t\ttr.logger.Info(\"Tasklist manager context is cancelled, shutting down\")\n\t\treturn true, true\n\t}\n\n\tif errors.Is(err, context.DeadlineExceeded) {\n\t\t// it only happens when isolation is enabled and there is no pollers from the given isolation group\n\t\t// if this happens, we don't want to block the task dispatching, because there might be pollers from\n\t\t// other isolation groups, we just simply continue and dispatch the task to a new isolation group which\n\t\t// has pollers\n\t\ttr.logger.Warn(\"Async task dispatch timed out\",\n\t\t\ttag.IsolationGroup(isolationGroup),\n\t\t\ttag.WorkflowRunID(taskInfo.RunID),\n\t\t\ttag.WorkflowID(taskInfo.WorkflowID),\n\t\t\ttag.TaskID(taskInfo.TaskID),\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowDomainID(taskInfo.DomainID),\n\t\t)\n\t\te.EventName = \"Dispatch Timed Out\"\n\t\tevent.Log(e)\n\t\ttr.scope.IncCounter(metrics.AsyncMatchDispatchTimeoutCounterPerTaskList)\n\t\treturn false, false\n\t}\n\n\tif errors.Is(err, ErrTasklistThrottled) {\n\t\te.EventName = \"Dispatch failed because throttled. Will retry dispatch\"\n\t\tevent.Log(e)\n\t\ttr.scope.IncCounter(metrics.BufferThrottlePerTaskListCounter)\n\t\truntime.Gosched()\n\t\treturn false, false\n\t}\n\n\tif errors.Is(err, errTaskNotStarted) {\n\t\te.EventName = \"Dispatch failed on completing task on the passive side because task not started.\"\n\t\tevent.Log(e)\n\t\treturn false, false\n\t}\n\n\tif errors.Is(err, errWaitTimeNotReachedForEntityNotExists) {\n\t\te.EventName = \"Dispatch failed on completing task on the passive side because workflow could not be found and not enough time has passed to complete the task. Will retry dispatch\"\n\t\tevent.Log(e)\n\t\treturn false, false\n\t}\n\n\te.EventName = \"Dispatch failed because of unknown error. Will retry dispatch\"\n\te.Payload = map[string]any{\n\t\t\"error\": err,\n\t}\n\tevent.Log(e)\n\te.Payload = nil\n\n\ttr.scope.IncCounter(metrics.BufferUnknownTaskDispatchError)\n\ttr.logger.Error(\"unknown error while dispatching task\",\n\t\ttag.Error(err),\n\t\ttag.IsolationGroup(isolationGroup),\n\t\ttag.WorkflowRunID(taskInfo.RunID),\n\t\ttag.WorkflowID(taskInfo.WorkflowID),\n\t\ttag.TaskID(taskInfo.TaskID),\n\t\ttag.WorkflowDomainID(taskInfo.DomainID),\n\t)\n\treturn false, false\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_reader_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\tcommonConfig \"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/matching/config\"\n)\n\nconst defaultIsolationGroup = \"a\"\nconst defaultAsyncDispatchTimeout = 3 * time.Second\n\nvar defaultIsolationGroups = []string{\n\t\"a\",\n\t\"b\",\n\t\"c\",\n\t\"d\",\n}\n\nfunc TestDispatchSingleTaskFromBuffer(t *testing.T) {\n\ttestCases := []struct {\n\t\tname          string\n\t\tallowances    func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource)\n\t\tttl           int\n\t\tbreakDispatch bool\n\t\tbreakRetries  bool\n\t}{\n\t\t{\n\t\t\tname: \"success - no isolation\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn \"\", -1\n\t\t\t\t}\n\t\t\t\tmarkCalled := requireCallbackInvocation(t, \"expected task to be dispatched\")\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\tassert.Equal(t, \"\", task.isolationGroup)\n\t\t\t\t\tmarkCalled()\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - isolation\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\tmarkCalled := requireCallbackInvocation(t, \"expected task to be dispatched\")\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\tassert.Equal(t, defaultIsolationGroup, task.isolationGroup)\n\t\t\t\t\tmarkCalled()\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - unknown isolation group\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn \"mystery group\", -1\n\t\t\t\t}\n\t\t\t\tmarkCalled := requireCallbackInvocation(t, \"expected task to be dispatched\")\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\tassert.Equal(t, \"\", task.isolationGroup)\n\t\t\t\t\tmarkCalled()\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"success - skip expired tasks\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\tt.Fatal(\"task must not be dispatched\")\n\t\t\t\t\treturn nil\n\t\t\t\t}\n\t\t\t},\n\t\t\tttl:           -2,\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - context cancelled, should stop\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn context.Canceled\n\t\t\t\t}\n\n\t\t\t},\n\t\t\tbreakDispatch: true,\n\t\t\tbreakRetries:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - Deadline Exceeded, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn context.DeadlineExceeded\n\t\t\t\t}\n\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - throttled, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn ErrTasklistThrottled\n\t\t\t\t}\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - unknown, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn errors.New(\"wow\")\n\t\t\t\t}\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - task not started and not expired, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn errTaskNotStarted\n\t\t\t\t}\n\t\t\t},\n\t\t\tttl:           100,\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - task not started and expired, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\tmockTime.Advance(time.Hour)\n\t\t\t\t\treturn errTaskNotStarted\n\t\t\t\t}\n\t\t\t},\n\t\t\tttl:           1,\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - time not reached to complete task without workflow execution, should retry\",\n\t\t\tallowances: func(t *testing.T, reader *taskReader, mockTime clock.MockedTimeSource) {\n\t\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\t\treturn defaultIsolationGroup, -1\n\t\t\t\t}\n\t\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\t\treturn errWaitTimeNotReachedForEntityNotExists\n\t\t\t\t}\n\t\t\t},\n\t\t\tbreakDispatch: false,\n\t\t\tbreakRetries:  false,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\ttimeSource := clock.NewMockedTimeSource()\n\t\t\tc := defaultConfig()\n\t\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), controller, c, timeSource)\n\t\t\treader := tlm.taskReader\n\t\t\ttc.allowances(t, reader, timeSource)\n\t\t\ttaskInfo := newTask(timeSource)\n\t\t\ttaskInfo.Expiry = timeSource.Now().Add(time.Duration(tc.ttl) * time.Second)\n\n\t\t\tbreakDispatch, breakRetries := reader.dispatchSingleTaskFromBuffer(taskInfo)\n\t\t\tassert.Equal(t, tc.breakDispatch, breakDispatch)\n\t\t\tassert.Equal(t, tc.breakRetries, breakRetries)\n\t\t})\n\t}\n}\n\nfunc TestGetDispatchTimeout(t *testing.T) {\n\ttestCases := []struct {\n\t\tname              string\n\t\tisolationDuration time.Duration\n\t\tdispatchRps       float64\n\t\texpected          time.Duration\n\t}{\n\t\t{\n\t\t\tname:              \"default - async dispatch timeout\",\n\t\t\tisolationDuration: noIsolationTimeout,\n\t\t\tdispatchRps:       1000,\n\t\t\texpected:          defaultAsyncDispatchTimeout,\n\t\t},\n\t\t{\n\t\t\tname:              \"isolation duration below async dispatch timeout\",\n\t\t\tisolationDuration: 1 * time.Second,\n\t\t\tdispatchRps:       1000,\n\t\t\texpected:          1 * time.Second,\n\t\t},\n\t\t{\n\t\t\tname:              \"isolation duration above async dispatch timeout\",\n\t\t\tisolationDuration: 5 * time.Second,\n\t\t\tdispatchRps:       1000,\n\t\t\texpected:          defaultAsyncDispatchTimeout,\n\t\t},\n\t\t{\n\t\t\tname:              \"no isolation - low dispatch rps extends timeout\",\n\t\t\tisolationDuration: noIsolationTimeout,\n\t\t\tdispatchRps:       0.1,\n\t\t\t// rate is divided by 4 isolation groups (plus default buffer) and only one task gets dispatched per 10 seconds\n\t\t\texpected: 50 * time.Second,\n\t\t},\n\t\t{\n\t\t\tname:              \"with isolation - low dispatch rps extends timeout\",\n\t\t\tisolationDuration: time.Second,\n\t\t\tdispatchRps:       0.1,\n\t\t\t// rate is divided by 4 isolation groups (plus default buffer) and only one task gets dispatched per 10 seconds\n\t\t\t// This means taskIsolationDuration is extended, and we don't leak tasks as quickly if the\n\t\t\t// task list has a very low RPS\n\t\t\texpected: 50 * time.Second,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\ttimeSource := clock.NewMockedTimeSource()\n\t\t\tc := defaultConfig()\n\t\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), controller, c, timeSource)\n\t\t\treader := tlm.taskReader\n\n\t\t\tactual := reader.getDispatchTimeout(tc.dispatchRps, tc.isolationDuration)\n\t\t\tassert.Equal(t, tc.expected, actual)\n\n\t\t})\n\t}\n}\n\nfunc TestTaskPump(t *testing.T) {\n\tcases := []struct {\n\t\tname       string\n\t\ttasks      []*persistence.TaskInfo\n\t\tassertions func(t *testing.T, tasks []*InternalTask)\n\t}{\n\t\t{\n\t\t\tname: \"single task\",\n\t\t\ttasks: []*persistence.TaskInfo{\n\t\t\t\ttaskWithIsolationGroup(\"a\"),\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, tasks []*InternalTask) {\n\t\t\t\tassert.Equal(t, 1, len(tasks))\n\t\t\t\tassert.Equal(t, \"a\", tasks[0].isolationGroup)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"unknown isolation group\",\n\t\t\ttasks: []*persistence.TaskInfo{\n\t\t\t\ttaskWithIsolationGroup(\"unknown\"),\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, tasks []*InternalTask) {\n\t\t\t\tassert.Equal(t, 1, len(tasks))\n\t\t\t\tassert.Equal(t, \"\", tasks[0].isolationGroup)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcontroller := gomock.NewController(t)\n\t\t\ttimeSource := clock.NewMockedTimeSource()\n\t\t\tc := defaultConfig()\n\t\t\ttlm := createTestTaskListManagerWithConfig(t, testlogger.New(t), controller, c, timeSource)\n\t\t\treader := tlm.taskReader\n\n\t\t\tvar lock sync.Mutex\n\t\t\treceived := make([]*InternalTask, 0)\n\t\t\tdone := make(chan struct{})\n\n\t\t\treader.dispatchTask = func(ctx context.Context, task *InternalTask) error {\n\t\t\t\tlock.Lock()\n\t\t\t\tdefer lock.Unlock()\n\n\t\t\t\treceived = append(received, task)\n\t\t\t\tif len(received) == len(tc.tasks) {\n\t\t\t\t\tclose(done)\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treader.getIsolationGroupForTask = func(ctx context.Context, info *persistence.TaskInfo) (string, time.Duration) {\n\t\t\t\treturn info.PartitionConfig[\"isolation-group\"], -1\n\t\t\t}\n\n\t\t\tfor i, taskInfo := range tc.tasks {\n\t\t\t\ttaskInfo.CreatedTime = timeSource.Now()\n\t\t\t\ttaskInfo.Expiry = timeSource.Now().Add(10 * time.Second)\n\t\t\t\t_, err := tlm.db.CreateTasks([]*persistence.CreateTaskInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tData:   taskInfo,\n\t\t\t\t\t\tTaskID: int64(1 + i),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t\t// This is how the reader knows the range that's valid to read\n\t\t\ttlm.taskWriter.maxReadLevel = int64(len(tc.tasks))\n\n\t\t\tctx, cancel := context.WithTimeout(context.Background(), time.Second*10)\n\t\t\tdefer cancel()\n\t\t\treader.Start()\n\n\t\t\tselect {\n\t\t\tcase <-ctx.Done():\n\t\t\t\tt.Fatalf(\"timed out waiting for tasks to be dispatched\")\n\t\t\tcase <-done:\n\t\t\t}\n\t\t\treader.Stop()\n\n\t\t\ttc.assertions(t, received)\n\t\t})\n\t}\n}\n\nfunc defaultConfig() *config.Config {\n\tconfig := config.NewConfig(dynamicconfig.NewNopCollection(), \"some random hostname\", commonConfig.RPC{}, func() []string {\n\t\treturn defaultIsolationGroups\n\t})\n\tconfig.EnableTasklistIsolation = dynamicproperties.GetBoolPropertyFnFilteredByDomain(true)\n\tconfig.LongPollExpirationInterval = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(100 * time.Millisecond)\n\tconfig.MaxTaskDeleteBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(1)\n\tconfig.GetTasksBatchSize = dynamicproperties.GetIntPropertyFilteredByTaskListInfo(10)\n\tconfig.AsyncTaskDispatchTimeout = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(defaultAsyncDispatchTimeout)\n\tconfig.LocalTaskWaitTime = dynamicproperties.GetDurationPropertyFnFilteredByTaskListInfo(time.Millisecond)\n\treturn config\n}\n\nfunc taskWithIsolationGroup(ig string) *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:                      \"domain-id\",\n\t\tWorkflowID:                    \"workflow-id\",\n\t\tRunID:                         \"run-id\",\n\t\tTaskID:                        1,\n\t\tScheduleID:                    2,\n\t\tScheduleToStartTimeoutSeconds: 10,\n\t\tPartitionConfig: map[string]string{\n\t\t\t\"isolation-group\": ig,\n\t\t},\n\t}\n}\n\nfunc newTask(timeSource clock.TimeSource) *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:                      \"domain-id\",\n\t\tWorkflowID:                    \"workflow-id\",\n\t\tRunID:                         \"run-id\",\n\t\tTaskID:                        1,\n\t\tScheduleID:                    2,\n\t\tScheduleToStartTimeoutSeconds: 10,\n\t\tExpiry:                        timeSource.Now().Add(10 * time.Second),\n\t\tCreatedTime:                   timeSource.Now(),\n\t\tPartitionConfig: map[string]string{\n\t\t\t\"isolation-group\": defaultIsolationGroup,\n\t\t},\n\t}\n}\n\nfunc requireCallbackInvocation(t *testing.T, msg string) func() {\n\tcalled := false\n\tt.Cleanup(func() {\n\t\tif !called {\n\t\t\tt.Fatal(msg)\n\t\t}\n\t})\n\n\treturn func() {\n\t\tcalled = true\n\t}\n}\n\nfunc TestTaskReaderBatchSizeValidation(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tconfigValue    int\n\t\texpectedBuffer int\n\t}{\n\t\t{\n\t\t\tname:           \"valid positive batch size\",\n\t\t\tconfigValue:    1000,\n\t\t\texpectedBuffer: 999, // buffer size is batchSize - 1\n\t\t},\n\t\t{\n\t\t\tname:           \"valid small batch size\",\n\t\t\tconfigValue:    1,\n\t\t\texpectedBuffer: 0, // buffer size is batchSize - 1\n\t\t},\n\t\t{\n\t\t\tname:           \"zero batch size should be corrected to default (1000)\",\n\t\t\tconfigValue:    0,\n\t\t\texpectedBuffer: 999, // corrected to 1000, so buffer is 999\n\t\t},\n\t\t{\n\t\t\tname:           \"negative batch size should be corrected to default (1000)\",\n\t\t\tconfigValue:    -5,\n\t\t\texpectedBuffer: 999, // corrected to 1000, so buffer is 999\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tbatchSize := tt.configValue\n\t\t\tif batchSize <= 0 {\n\t\t\t\t// This is the validation logic from newTaskReader - use default value (1000)\n\t\t\t\tbatchSize = 1000\n\t\t\t}\n\n\t\t\t// Test buffer creation with validated batch size\n\t\t\texpectedCapacity := batchSize - 1\n\t\t\tif expectedCapacity < 0 {\n\t\t\t\texpectedCapacity = 0\n\t\t\t}\n\n\t\t\t// Simulate the buffer creation logic\n\t\t\tbuffer := make(chan *persistence.TaskInfo, expectedCapacity)\n\t\t\tactualCapacity := cap(buffer)\n\n\t\t\tassert.Equal(t, tt.expectedBuffer, actualCapacity, \"Task buffer should have correct capacity after validation\")\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewInternalTask(t *testing.T) {\n\tcases := []struct {\n\t\tname                    string\n\t\tpartitionConfig         map[string]string\n\t\tsource                  types.TaskSource\n\t\tforwardedFrom           string\n\t\tforSyncMatch            bool\n\t\tisolationGroup          string\n\t\texpectedPartitionConfig map[string]string\n\t\tadditionalAssertions    func(t *testing.T, task *InternalTask)\n\t}{\n\t\t{\n\t\t\tname:         \"sync match\",\n\t\t\tsource:       types.TaskSourceHistory,\n\t\t\tforSyncMatch: true,\n\t\t\tadditionalAssertions: func(t *testing.T, task *InternalTask) {\n\t\t\t\t// Only initialized for sync match\n\t\t\t\tassert.NotNil(t, task.ResponseC)\n\t\t\t\tassert.True(t, task.IsSyncMatch())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:         \"async match\",\n\t\t\tsource:       types.TaskSourceDbBacklog,\n\t\t\tforSyncMatch: false,\n\t\t\tadditionalAssertions: func(t *testing.T, task *InternalTask) {\n\t\t\t\t// Only initialized for sync match\n\t\t\t\tassert.Nil(t, task.ResponseC)\n\t\t\t\tassert.False(t, task.IsSyncMatch())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"forwarded from history\",\n\t\t\tsource:        types.TaskSourceDbBacklog,\n\t\t\tforSyncMatch:  true,\n\t\t\tforwardedFrom: \"elsewhere\",\n\t\t\tadditionalAssertions: func(t *testing.T, task *InternalTask) {\n\t\t\t\tassert.True(t, task.IsForwarded())\n\t\t\t\tassert.True(t, task.IsSyncMatch())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"forwarded from backlog\",\n\t\t\tsource:        types.TaskSourceDbBacklog,\n\t\t\tforSyncMatch:  true,\n\t\t\tforwardedFrom: \"elsewhere\",\n\t\t\tadditionalAssertions: func(t *testing.T, task *InternalTask) {\n\t\t\t\tassert.True(t, task.IsForwarded())\n\t\t\t\t// Still technically sync match, just on a different host\n\t\t\t\tassert.True(t, task.IsSyncMatch())\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"tasklist isolation\",\n\t\t\tsource:         types.TaskSourceDbBacklog,\n\t\t\tisolationGroup: \"a\",\n\t\t\tpartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.GroupKey:      \"a\",\n\t\t\t\tisolationgroup.WorkflowIDKey: \"workflowID\",\n\t\t\t},\n\t\t\texpectedPartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.OriginalGroupKey: \"a\",\n\t\t\t\tisolationgroup.GroupKey:         \"a\",\n\t\t\t\tisolationgroup.WorkflowIDKey:    \"workflowID\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"tasklist isolation - leaked\",\n\t\t\tsource:         types.TaskSourceDbBacklog,\n\t\t\tisolationGroup: \"\",\n\t\t\tpartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.GroupKey:      \"a\",\n\t\t\t\tisolationgroup.WorkflowIDKey: \"workflowID\",\n\t\t\t},\n\t\t\texpectedPartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.OriginalGroupKey: \"a\",\n\t\t\t\tisolationgroup.GroupKey:         \"\",\n\t\t\t\tisolationgroup.WorkflowIDKey:    \"workflowID\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"tasklist isolation - forwarded\",\n\t\t\tsource:         types.TaskSourceDbBacklog,\n\t\t\tisolationGroup: \"\",\n\t\t\tpartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.OriginalGroupKey: \"a\",\n\t\t\t\tisolationgroup.GroupKey:         \"\",\n\t\t\t\tisolationgroup.WorkflowIDKey:    \"workflowID\",\n\t\t\t},\n\t\t\texpectedPartitionConfig: map[string]string{\n\t\t\t\tisolationgroup.OriginalGroupKey: \"a\",\n\t\t\t\tisolationgroup.GroupKey:         \"\",\n\t\t\t\tisolationgroup.WorkflowIDKey:    \"workflowID\",\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcompletionFunc := func(_ *persistence.TaskInfo, _ error) {}\n\t\t\ttaskInfo := defaultTaskInfo(tc.partitionConfig)\n\t\t\tactivityDispatchInfo := &types.ActivityTaskDispatchInfo{WorkflowDomain: \"domain\"}\n\t\t\ttask := newInternalTask(taskInfo, completionFunc, tc.source, tc.forwardedFrom, tc.forSyncMatch, activityDispatchInfo, tc.isolationGroup)\n\t\t\tassert.Equal(t, defaultTaskInfo(tc.expectedPartitionConfig), task.Event.TaskInfo)\n\t\t\tassert.NotNil(t, task.Event.completionFunc)\n\t\t\tassert.Equal(t, tc.source, task.source)\n\t\t\tassert.Equal(t, tc.forwardedFrom, task.forwardedFrom)\n\t\t\tassert.Equal(t, tc.isolationGroup, task.isolationGroup)\n\t\t\tassert.Equal(t, activityDispatchInfo, task.ActivityTaskDispatchInfo)\n\t\t\tif tc.additionalAssertions != nil {\n\t\t\t\ttc.additionalAssertions(t, task)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc defaultTaskInfo(partitionConfig map[string]string) *persistence.TaskInfo {\n\treturn &persistence.TaskInfo{\n\t\tDomainID:                      \"DomainID\",\n\t\tWorkflowID:                    \"WorkflowID\",\n\t\tRunID:                         \"RunID\",\n\t\tTaskID:                        1,\n\t\tScheduleID:                    2,\n\t\tScheduleToStartTimeoutSeconds: 3,\n\t\tExpiry:                        time.UnixMicro(4),\n\t\tCreatedTime:                   time.UnixMicro(5),\n\t\tPartitionConfig:               partitionConfig,\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/task_writer.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/config\"\n\t\"github.com/uber/cadence/service/matching/event\"\n)\n\ntype (\n\twriteTaskResponse struct {\n\t\terr                 error\n\t\tpersistenceResponse *persistence.CreateTasksResponse\n\t}\n\n\twriteTaskRequest struct {\n\t\ttaskInfo   *persistence.TaskInfo\n\t\tresponseCh chan<- *writeTaskResponse\n\t}\n\n\ttaskIDBlock struct {\n\t\tstart int64\n\t\tend   int64\n\t}\n\n\t// taskWriter writes tasks sequentially to persistence\n\ttaskWriter struct {\n\t\tdb             *taskListDB\n\t\tconfig         *config.TaskListConfig\n\t\ttaskListID     *Identifier\n\t\ttaskAckManager messaging.AckManager\n\t\tappendCh       chan *writeTaskRequest\n\t\ttaskIDBlock    taskIDBlock\n\t\tmaxReadLevel   int64\n\t\tstopped        int64 // set to 1 if the writer is stopped or is shutting down\n\t\tlogger         log.Logger\n\t\tscope          metrics.Scope\n\t\tstopCh         chan struct{} // shutdown signal for all routines in this class\n\t\tthrottleRetry  *backoff.ThrottleRetry\n\t\thandleErr      func(error) error\n\t\tonFatalErr     func()\n\t}\n)\n\n// errShutdown indicates that the task list is shutting down\nvar errShutdown = errors.New(\"task list shutting down\")\n\nfunc newTaskWriter(tlMgr *taskListManagerImpl) *taskWriter {\n\treturn &taskWriter{\n\t\tdb:             tlMgr.db,\n\t\tconfig:         tlMgr.config,\n\t\ttaskListID:     tlMgr.taskListID,\n\t\ttaskAckManager: tlMgr.taskAckManager,\n\t\tstopCh:         make(chan struct{}),\n\t\tappendCh:       make(chan *writeTaskRequest, tlMgr.config.OutstandingTaskAppendsThreshold()),\n\t\tlogger:         tlMgr.logger,\n\t\tscope:          tlMgr.scope,\n\t\thandleErr:      tlMgr.handleErr,\n\t\tonFatalErr:     tlMgr.Stop,\n\t\tthrottleRetry: backoff.NewThrottleRetry(\n\t\t\tbackoff.WithRetryPolicy(persistenceOperationRetryPolicy),\n\t\t\tbackoff.WithRetryableError(persistence.IsTransientError),\n\t\t),\n\t}\n}\n\nfunc (w *taskWriter) Start() error {\n\t// Make sure to grab the range first before starting task writer, as it needs the range to initialize maxReadLevel\n\tstate, err := w.renewLeaseWithRetry()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tw.taskAckManager.SetAckLevel(state.ackLevel)\n\tblock := rangeIDToTaskIDBlock(state.rangeID, w.config.RangeSize)\n\tw.taskIDBlock = block\n\tw.maxReadLevel = block.start - 1\n\tgo w.taskWriterLoop()\n\treturn nil\n}\n\n// Stop stops the taskWriter\nfunc (w *taskWriter) Stop() {\n\tif atomic.CompareAndSwapInt64(&w.stopped, 0, 1) {\n\t\tclose(w.stopCh)\n\t}\n}\n\nfunc (w *taskWriter) isStopped() bool {\n\treturn atomic.LoadInt64(&w.stopped) == 1\n}\n\nfunc (w *taskWriter) appendTask(ctx context.Context, taskInfo *persistence.TaskInfo) (*persistence.CreateTasksResponse, error) {\n\t// Make context to respect the append task timeout.\n\ttCtx, cancel := context.WithTimeout(ctx, w.config.AppendTaskTimeout())\n\tdefer cancel()\n\n\t// If the writer has already stopped, it returns an error\n\tif w.isStopped() {\n\t\treturn nil, errShutdown\n\t}\n\n\t// Create a result channel for this request.\n\t// Use buffer size 1 so the writer can send one response even if this call is already timed out.\n\t// Without a buffer, the writer can block forever when no receiver is waiting.\n\tch := make(chan *writeTaskResponse, 1)\n\n\treq := &writeTaskRequest{\n\t\ttaskInfo:   taskInfo,\n\t\tresponseCh: ch,\n\t}\n\n\tselect {\n\tcase w.appendCh <- req: // If enqueue succeed\n\t\tselect { // Wait for the task result after the enqueue\n\t\tcase r := <-ch: // If the writer sends the task result after the DB I/O task\n\t\t\treturn r.persistenceResponse, r.err\n\t\tcase <-tCtx.Done(): // If the writer didn't send the task result within a timeout threshold\n\t\t\treturn nil, fmt.Errorf(\"failed to receive a success response from the channel within a timeout threshold: %w\", tCtx.Err())\n\t\tcase <-w.stopCh:\n\t\t\t// if we are shutting down, this request will never make\n\t\t\t// it to cassandra, just bail out and fail this request\n\t\t\treturn nil, errShutdown\n\t\t}\n\tcase <-tCtx.Done(): // If a channel is full and timed out waiting to enqueue\n\t\treturn nil, createServiceBusyError(\"task queue is full and timed out waiting to enqueue\")\n\t}\n}\n\nfunc (w *taskWriter) GetMaxReadLevel() int64 {\n\treturn atomic.LoadInt64(&w.maxReadLevel)\n}\n\nfunc (w *taskWriter) allocTaskIDs(count int) ([]int64, error) {\n\tresult := make([]int64, count)\n\tfor i := 0; i < count; i++ {\n\t\tif w.taskIDBlock.start > w.taskIDBlock.end {\n\t\t\t// we ran out of current allocation block\n\t\t\tnewBlock, err := w.allocTaskIDBlock(w.taskIDBlock.end)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tw.taskIDBlock = newBlock\n\t\t}\n\t\tresult[i] = w.taskIDBlock.start\n\t\tw.taskIDBlock.start++\n\t}\n\treturn result, nil\n}\n\nfunc (w *taskWriter) allocTaskIDBlock(prevBlockEnd int64) (taskIDBlock, error) {\n\tcurrBlock := rangeIDToTaskIDBlock(w.db.RangeID(), w.config.RangeSize)\n\tif currBlock.end != prevBlockEnd {\n\t\treturn taskIDBlock{},\n\t\t\tfmt.Errorf(\"allocTaskIDBlock: invalid state: prevBlockEnd:%v != currTaskIDBlock:%+v\", prevBlockEnd, currBlock)\n\t}\n\tstate, err := w.renewLeaseWithRetry()\n\tif err != nil {\n\t\treturn taskIDBlock{}, err\n\t}\n\treturn rangeIDToTaskIDBlock(state.rangeID, w.config.RangeSize), nil\n}\n\nfunc (w *taskWriter) renewLeaseWithRetry() (taskListState, error) {\n\tvar newState taskListState\n\top := func(ctx context.Context) (err error) {\n\t\tnewState, err = w.db.RenewLease()\n\t\treturn\n\t}\n\tw.scope.IncCounter(metrics.LeaseRequestPerTaskListCounter)\n\terr := w.throttleRetry.Do(context.Background(), op)\n\tif err != nil {\n\t\tw.scope.IncCounter(metrics.LeaseFailurePerTaskListCounter)\n\t\tw.onFatalErr()\n\t\treturn newState, err\n\t}\n\treturn newState, nil\n}\n\nfunc (w *taskWriter) taskWriterLoop() {\nwriterLoop:\n\tfor {\n\t\tselect {\n\t\tcase request := <-w.appendCh:\n\t\t\t{\n\t\t\t\tif w.isStopped() {\n\t\t\t\t\tbreak writerLoop\n\t\t\t\t}\n\n\t\t\t\t// read a batch of requests from the channel\n\t\t\t\treqs := []*writeTaskRequest{request}\n\t\t\t\treqs = w.getWriteBatch(reqs)\n\t\t\t\tbatchSize := len(reqs)\n\n\t\t\t\tmaxReadLevel := int64(0)\n\n\t\t\t\ttaskIDs, err := w.allocTaskIDs(batchSize)\n\t\t\t\tif err != nil {\n\t\t\t\t\tw.logger.Error(\"error allocating task ids\",\n\t\t\t\t\t\ttag.Error(err),\n\t\t\t\t\t)\n\t\t\t\t\tw.sendWriteResponse(reqs, err, nil)\n\t\t\t\t\tcontinue writerLoop\n\t\t\t\t}\n\n\t\t\t\ttasks := []*persistence.CreateTaskInfo{}\n\t\t\t\tevents := []event.E{}\n\t\t\t\tfor i, req := range reqs {\n\t\t\t\t\ttasks = append(tasks, &persistence.CreateTaskInfo{\n\t\t\t\t\t\tTaskID: taskIDs[i],\n\t\t\t\t\t\tData:   req.taskInfo,\n\t\t\t\t\t})\n\t\t\t\t\tkind := types.TaskListKind(w.db.taskListKind)\n\t\t\t\t\tevents = append(events, event.E{\n\t\t\t\t\t\tTaskListName: w.db.taskListName,\n\t\t\t\t\t\tTaskListKind: &kind,\n\t\t\t\t\t\tTaskListType: w.db.taskType,\n\t\t\t\t\t\tTaskInfo:     *req.taskInfo,\n\t\t\t\t\t\tEventName:    \"Task Written to DB\",\n\t\t\t\t\t})\n\t\t\t\t\tmaxReadLevel = taskIDs[i]\n\t\t\t\t}\n\n\t\t\t\tresp, err := w.db.CreateTasks(tasks)\n\t\t\t\terr = w.handleErr(err)\n\t\t\t\tif err != nil {\n\t\t\t\t\tw.logger.Error(\"Persistent store operation failure\",\n\t\t\t\t\t\ttag.StoreOperationCreateTasks,\n\t\t\t\t\t\ttag.Error(err),\n\t\t\t\t\t\ttag.Number(taskIDs[0]),\n\t\t\t\t\t\ttag.NextNumber(taskIDs[batchSize-1]),\n\t\t\t\t\t)\n\t\t\t\t} else {\n\t\t\t\t\tevent.Log(events...)\n\t\t\t\t}\n\t\t\t\t// Update the maxReadLevel after the writes are completed.\n\t\t\t\tif maxReadLevel > 0 {\n\t\t\t\t\tatomic.StoreInt64(&w.maxReadLevel, maxReadLevel)\n\t\t\t\t}\n\n\t\t\t\tw.sendWriteResponse(reqs, err, resp)\n\t\t\t}\n\t\tcase <-w.stopCh:\n\t\t\t// we don't close the appendCh here\n\t\t\t// because that can cause \"send on closed channel\" panic on the appendTask()\n\t\t\tbreak writerLoop\n\t\t}\n\t}\n}\n\nfunc (w *taskWriter) getWriteBatch(reqs []*writeTaskRequest) []*writeTaskRequest {\nreadLoop:\n\tfor i := 0; i < w.config.MaxTaskBatchSize(); i++ {\n\t\tselect {\n\t\tcase req := <-w.appendCh:\n\t\t\treqs = append(reqs, req)\n\t\tdefault: // channel is empty, don't block\n\t\t\tbreak readLoop\n\t\t}\n\t}\n\treturn reqs\n}\n\nfunc (w *taskWriter) sendWriteResponse(reqs []*writeTaskRequest,\n\terr error, persistenceResponse *persistence.CreateTasksResponse) {\n\tfor _, req := range reqs {\n\t\tresp := &writeTaskResponse{\n\t\t\terr:                 err,\n\t\t\tpersistenceResponse: persistenceResponse,\n\t\t}\n\n\t\t// appendTask() listens stopCh and terminates early without reading from responseCh.\n\t\t// Therefore we need to listen stopCh here to avoid getting stuck while pushing response to responseCh.\n\t\tselect {\n\t\tcase <-w.stopCh:\n\t\tcase req.responseCh <- resp:\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/matching/tasklist/testing.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/davecgh/go-spew/spew\"\n\t\"github.com/emirpasic/gods/maps/treemap\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nvar _ persistence.TaskManager = (*TestTaskManager)(nil) // Asserts that interface is indeed implemented\n\ntype (\n\tTestTaskManager struct {\n\t\tsync.Mutex\n\t\tt          *testing.T\n\t\ttaskLists  map[Identifier]*testTaskListManager\n\t\tlogger     log.Logger\n\t\ttimeSource clock.TimeSource\n\t}\n\n\ttestTaskListManager struct {\n\t\tsync.RWMutex\n\t\trangeID                 int64\n\t\tackLevel                int64\n\t\tkind                    int\n\t\tcreateTaskCount         int\n\t\ttasks                   *treemap.Map\n\t\tadaptivePartitionConfig *persistence.TaskListPartitionConfig\n\t}\n)\n\nfunc newTestTaskListManager() *testTaskListManager {\n\treturn &testTaskListManager{tasks: treemap.NewWith(int64Comparator)}\n}\n\nfunc NewTestTaskManager(t *testing.T, logger log.Logger, timeSource clock.TimeSource) *TestTaskManager {\n\treturn &TestTaskManager{\n\t\tt:          t,\n\t\ttaskLists:  make(map[Identifier]*testTaskListManager),\n\t\tlogger:     logger,\n\t\ttimeSource: timeSource,\n\t}\n}\n\nfunc (m *TestTaskManager) GetName() string {\n\treturn \"test\"\n}\n\nfunc (m *TestTaskManager) Close() {\n}\n\n// LeaseTaskList provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) LeaseTaskList(\n\t_ context.Context,\n\trequest *persistence.LeaseTaskListRequest,\n) (*persistence.LeaseTaskListResponse, error) {\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, request.DomainID, request.TaskList, request.TaskType))\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\tif request.RangeID > 0 && request.RangeID != tlm.rangeID {\n\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\tMsg: fmt.Sprintf(\"leaseTaskList:renew failed: taskList:%v, taskListType:%v, haveRangeID:%v, gotRangeID:%v\",\n\t\t\t\trequest.TaskList, request.TaskType, request.RangeID, tlm.rangeID),\n\t\t}\n\t}\n\ttlm.rangeID++\n\ttlm.kind = request.TaskListKind\n\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.LeaseTaskList rangeID=%v\", tlm.rangeID))\n\n\treturn &persistence.LeaseTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tAckLevel:                tlm.ackLevel,\n\t\t\tDomainID:                request.DomainID,\n\t\t\tName:                    request.TaskList,\n\t\t\tTaskType:                request.TaskType,\n\t\t\tRangeID:                 tlm.rangeID,\n\t\t\tKind:                    tlm.kind,\n\t\t\tAdaptivePartitionConfig: tlm.adaptivePartitionConfig,\n\t\t},\n\t}, nil\n}\n\nfunc (m *TestTaskManager) GetTaskList(\n\t_ context.Context,\n\trequest *persistence.GetTaskListRequest,\n) (*persistence.GetTaskListResponse, error) {\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, request.DomainID, request.TaskList, request.TaskType))\n\ttlm.RLock()\n\tdefer tlm.RUnlock()\n\treturn &persistence.GetTaskListResponse{\n\t\tTaskListInfo: &persistence.TaskListInfo{\n\t\t\tAckLevel:                tlm.ackLevel,\n\t\t\tDomainID:                request.DomainID,\n\t\t\tName:                    request.TaskList,\n\t\t\tTaskType:                request.TaskType,\n\t\t\tRangeID:                 tlm.rangeID,\n\t\t\tKind:                    tlm.kind,\n\t\t\tAdaptivePartitionConfig: tlm.adaptivePartitionConfig,\n\t\t},\n\t}, nil\n}\n\n// UpdateTaskList provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) UpdateTaskList(\n\t_ context.Context,\n\trequest *persistence.UpdateTaskListRequest,\n) (*persistence.UpdateTaskListResponse, error) {\n\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.UpdateTaskList taskListInfo=%v, ackLevel=%v\", request.TaskListInfo, request.TaskListInfo.AckLevel))\n\n\ttli := request.TaskListInfo\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, tli.DomainID, tli.Name, tli.TaskType))\n\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\tif tlm.rangeID != tli.RangeID {\n\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\tMsg: fmt.Sprintf(\"Failed to update task list: name=%v, type=%v, expected rangeID=%v, input rangeID=%v\", tli.Name, tli.TaskType, tlm.rangeID, tli.RangeID),\n\t\t}\n\t}\n\ttlm.ackLevel = tli.AckLevel\n\treturn &persistence.UpdateTaskListResponse{}, nil\n}\n\n// CompleteTask provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) CompleteTask(\n\t_ context.Context,\n\trequest *persistence.CompleteTaskRequest,\n) error {\n\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.CompleteTask taskID=%v, ackLevel=%v\", request.TaskID, request.TaskList.AckLevel))\n\tif request.TaskID <= 0 {\n\t\tpanic(fmt.Errorf(\"Invalid taskID=%v\", request.TaskID))\n\t}\n\n\ttli := request.TaskList\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, tli.DomainID, tli.Name, tli.TaskType))\n\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\n\ttlm.tasks.Remove(request.TaskID)\n\treturn nil\n}\n\n// CompleteTasksLessThan provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) CompleteTasksLessThan(\n\t_ context.Context,\n\trequest *persistence.CompleteTasksLessThanRequest,\n) (*persistence.CompleteTasksLessThanResponse, error) {\n\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.CompleteTasksLessThan taskID=%v\", request.TaskID))\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, request.DomainID, request.TaskListName, request.TaskType))\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\trowsDeleted := 0\n\tkeys := tlm.tasks.Keys()\n\tfor _, key := range keys {\n\t\tid := key.(int64)\n\t\tif id <= request.TaskID {\n\t\t\ttlm.tasks.Remove(id)\n\t\t\trowsDeleted++\n\t\t}\n\t}\n\treturn &persistence.CompleteTasksLessThanResponse{TasksCompleted: rowsDeleted}, nil\n}\n\n// ListTaskList provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) ListTaskList(\n\t_ context.Context,\n\trequest *persistence.ListTaskListRequest,\n) (*persistence.ListTaskListResponse, error) {\n\treturn nil, fmt.Errorf(\"unsupported operation\")\n}\n\n// DeleteTaskList provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) DeleteTaskList(\n\t_ context.Context,\n\trequest *persistence.DeleteTaskListRequest,\n) error {\n\tm.Lock()\n\tdefer m.Unlock()\n\tkey := NewTestTaskListID(m.t, request.DomainID, request.TaskListName, request.TaskListType)\n\tdelete(m.taskLists, *key)\n\treturn nil\n}\n\n// CreateTask provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) CreateTasks(\n\t_ context.Context,\n\trequest *persistence.CreateTasksRequest,\n) (*persistence.CreateTasksResponse, error) {\n\tdomainID := request.TaskListInfo.DomainID\n\ttaskList := request.TaskListInfo.Name\n\ttaskType := request.TaskListInfo.TaskType\n\trangeID := request.TaskListInfo.RangeID\n\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, domainID, taskList, taskType))\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\n\t// First validate the entire batch\n\tfor _, task := range request.Tasks {\n\t\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.CreateTask taskID=%v, rangeID=%v\", task.TaskID, rangeID))\n\t\tif task.TaskID <= 0 {\n\t\t\tpanic(fmt.Errorf(\"Invalid taskID=%v\", task.TaskID))\n\t\t}\n\n\t\tif tlm.rangeID != rangeID {\n\t\t\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.CreateTask ConditionFailedError taskID=%v, rangeID: %v, db rangeID: %v\",\n\t\t\t\ttask.TaskID, rangeID, tlm.rangeID))\n\n\t\t\treturn nil, &persistence.ConditionFailedError{\n\t\t\t\tMsg: fmt.Sprintf(\"testTaskManager.CreateTask failed. TaskList: %v, taskType: %v, rangeID: %v, db rangeID: %v\",\n\t\t\t\t\ttaskList, taskType, rangeID, tlm.rangeID),\n\t\t\t}\n\t\t}\n\t\t_, ok := tlm.tasks.Get(task.TaskID)\n\t\tif ok {\n\t\t\tpanic(fmt.Sprintf(\"Duplicated TaskID %v\", task.TaskID))\n\t\t}\n\t}\n\n\t// Then insert all tasks if no errors\n\tfor _, task := range request.Tasks {\n\t\tscheduleID := task.Data.ScheduleID\n\t\tinfo := &persistence.TaskInfo{\n\t\t\tDomainID:        domainID,\n\t\t\tWorkflowID:      task.Data.WorkflowID,\n\t\t\tRunID:           task.Data.RunID,\n\t\t\tScheduleID:      scheduleID,\n\t\t\tTaskID:          task.TaskID,\n\t\t\tPartitionConfig: task.Data.PartitionConfig,\n\t\t}\n\t\tif task.Data.ScheduleToStartTimeoutSeconds != 0 {\n\t\t\tinfo.Expiry = m.timeSource.Now().Add(time.Duration(task.Data.ScheduleToStartTimeoutSeconds) * time.Second)\n\t\t}\n\t\ttlm.tasks.Put(task.TaskID, info)\n\t\ttlm.createTaskCount++\n\t}\n\n\treturn &persistence.CreateTasksResponse{}, nil\n}\n\n// GetTasks provides a mock function with given fields: ctx, request\nfunc (m *TestTaskManager) GetTasks(\n\t_ context.Context,\n\trequest *persistence.GetTasksRequest,\n) (*persistence.GetTasksResponse, error) {\n\tm.logger.Debug(fmt.Sprintf(\"testTaskManager.GetTasks readLevel=%v, maxReadLevel=%v\", request.ReadLevel, *request.MaxReadLevel))\n\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, request.DomainID, request.TaskList, request.TaskType))\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\tvar tasks []*persistence.TaskInfo\n\n\tit := tlm.tasks.Iterator()\n\tfor it.Next() {\n\t\ttaskID := it.Key().(int64)\n\t\tif taskID <= request.ReadLevel {\n\t\t\tcontinue\n\t\t}\n\t\tif taskID > *request.MaxReadLevel {\n\t\t\tbreak\n\t\t}\n\t\ttasks = append(tasks, it.Value().(*persistence.TaskInfo))\n\t}\n\treturn &persistence.GetTasksResponse{\n\t\tTasks: tasks,\n\t}, nil\n}\n\nfunc (m *TestTaskManager) GetTaskListSize(_ context.Context, request *persistence.GetTaskListSizeRequest) (*persistence.GetTaskListSizeResponse, error) {\n\ttlm := m.getTaskListManager(NewTestTaskListID(m.t, request.DomainID, request.TaskListName, request.TaskListType))\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\tcount := int64(0)\n\tit := tlm.tasks.Iterator()\n\tfor it.Next() {\n\t\ttaskID := it.Key().(int64)\n\t\tif taskID <= request.AckLevel {\n\t\t\tcontinue\n\t\t}\n\t\tcount++\n\t}\n\treturn &persistence.GetTaskListSizeResponse{Size: count}, nil\n}\n\nfunc (m *TestTaskManager) GetOrphanTasks(_ context.Context, request *persistence.GetOrphanTasksRequest) (*persistence.GetOrphanTasksResponse, error) {\n\treturn &persistence.GetOrphanTasksResponse{}, nil\n}\n\n// getTaskCount returns number of tasks in a task list\nfunc (m *TestTaskManager) GetTaskCount(taskList *Identifier) int {\n\ttlm := m.getTaskListManager(taskList)\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\treturn tlm.tasks.Size()\n}\n\n// getCreateTaskCount returns how many times CreateTask was called\nfunc (m *TestTaskManager) GetCreateTaskCount(taskList *Identifier) int {\n\ttlm := m.getTaskListManager(taskList)\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\treturn tlm.createTaskCount\n}\n\nfunc (m *TestTaskManager) SetRangeID(taskList *Identifier, rangeID int64) {\n\ttlm := m.getTaskListManager(taskList)\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\ttlm.rangeID = rangeID\n}\n\nfunc (m *TestTaskManager) GetRangeID(taskList *Identifier) int64 {\n\ttlm := m.getTaskListManager(taskList)\n\ttlm.Lock()\n\tdefer tlm.Unlock()\n\treturn tlm.rangeID\n}\n\nfunc (m *TestTaskManager) getTaskListManager(id *Identifier) *testTaskListManager {\n\tm.Lock()\n\tdefer m.Unlock()\n\tresult, ok := m.taskLists[*id]\n\tif ok {\n\t\treturn result\n\t}\n\tresult = newTestTaskListManager()\n\tm.taskLists[*id] = result\n\treturn result\n}\n\nfunc (m *TestTaskManager) String() string {\n\tm.Lock()\n\tdefer m.Unlock()\n\tvar result string\n\tfor id, tl := range m.taskLists {\n\t\ttl.Lock()\n\t\tif id.taskType == persistence.TaskListTypeActivity {\n\t\t\tresult += \"Activity\"\n\t\t} else {\n\t\t\tresult += \"Decision\"\n\t\t}\n\t\tresult += \" task list \" + id.name\n\t\tresult += \"\\n\"\n\t\tresult += fmt.Sprintf(\"AckLevel=%v\\n\", tl.ackLevel)\n\t\tresult += fmt.Sprintf(\"CreateTaskCount=%v\\n\", tl.createTaskCount)\n\t\tresult += fmt.Sprintf(\"RangeID=%v\\n\", tl.rangeID)\n\t\tresult += \"Tasks=\\n\"\n\t\tfor _, t := range tl.tasks.Values() {\n\t\t\tresult += spew.Sdump(t)\n\t\t\tresult += \"\\n\"\n\n\t\t}\n\t\ttl.Unlock()\n\t}\n\treturn result\n}\n\nfunc NewTestTaskListID(t *testing.T, domainID string, taskListName string, taskType int) *Identifier {\n\tid, err := NewIdentifier(domainID, taskListName, taskType)\n\tif err != nil {\n\t\tt.Fatal(\"failed to create task list identifier for test\", err)\n\t}\n\treturn id\n}\n\nfunc int64Comparator(a, b interface{}) int {\n\taAsserted := a.(int64)\n\tbAsserted := b.(int64)\n\tswitch {\n\tcase aAsserted > bAsserted:\n\t\treturn 1\n\tcase aAsserted < bAsserted:\n\t\treturn -1\n\tdefault:\n\t\treturn 0\n\t}\n}\n"
  },
  {
    "path": "service/matching/wrappers/grpc/grpc_handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage grpc\n\nimport (\n\t\"context\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g GRPCHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(matchingv1.BuildMatchingAPIYARPCProcedures(g))\n\tdispatcher.Register(apiv1.BuildMetaAPIYARPCProcedures(g))\n}\n\nfunc (g GRPCHandler) Health(ctx context.Context, _ *apiv1.HealthRequest) (*apiv1.HealthResponse, error) {\n\tresponse, err := g.h.Health(ctx)\n\treturn proto.FromHealthResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/matching/wrappers/grpc/grpc_handler_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tmatchingv1 \"github.com/uber/cadence/.gen/proto/matching/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/matching/handler\"\n)\n\ntype GRPCHandler struct {\n\th handler.Handler\n}\n\nfunc NewGRPCHandler(h handler.Handler) GRPCHandler {\n\treturn GRPCHandler{h}\n}\n\nfunc (g GRPCHandler) AddActivityTask(ctx context.Context, request *matchingv1.AddActivityTaskRequest) (*matchingv1.AddActivityTaskResponse, error) {\n\tresponse, err := g.h.AddActivityTask(ctx, proto.ToMatchingAddActivityTaskRequest(request))\n\treturn proto.FromMatchingAddActivityTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) AddDecisionTask(ctx context.Context, request *matchingv1.AddDecisionTaskRequest) (*matchingv1.AddDecisionTaskResponse, error) {\n\tresponse, err := g.h.AddDecisionTask(ctx, proto.ToMatchingAddDecisionTaskRequest(request))\n\treturn proto.FromMatchingAddDecisionTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) CancelOutstandingPoll(ctx context.Context, request *matchingv1.CancelOutstandingPollRequest) (*matchingv1.CancelOutstandingPollResponse, error) {\n\terr := g.h.CancelOutstandingPoll(ctx, proto.ToMatchingCancelOutstandingPollRequest(request))\n\treturn &matchingv1.CancelOutstandingPollResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) DescribeTaskList(ctx context.Context, request *matchingv1.DescribeTaskListRequest) (*matchingv1.DescribeTaskListResponse, error) {\n\tresponse, err := g.h.DescribeTaskList(ctx, proto.ToMatchingDescribeTaskListRequest(request))\n\treturn proto.FromMatchingDescribeTaskListResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) GetTaskListsByDomain(ctx context.Context, request *matchingv1.GetTaskListsByDomainRequest) (*matchingv1.GetTaskListsByDomainResponse, error) {\n\tresponse, err := g.h.GetTaskListsByDomain(ctx, proto.ToMatchingGetTaskListsByDomainRequest(request))\n\treturn proto.FromMatchingGetTaskListsByDomainResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) ListTaskListPartitions(ctx context.Context, request *matchingv1.ListTaskListPartitionsRequest) (*matchingv1.ListTaskListPartitionsResponse, error) {\n\tresponse, err := g.h.ListTaskListPartitions(ctx, proto.ToMatchingListTaskListPartitionsRequest(request))\n\treturn proto.FromMatchingListTaskListPartitionsResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) PollForActivityTask(ctx context.Context, request *matchingv1.PollForActivityTaskRequest) (*matchingv1.PollForActivityTaskResponse, error) {\n\tresponse, err := g.h.PollForActivityTask(ctx, proto.ToMatchingPollForActivityTaskRequest(request))\n\treturn proto.FromMatchingPollForActivityTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) PollForDecisionTask(ctx context.Context, request *matchingv1.PollForDecisionTaskRequest) (*matchingv1.PollForDecisionTaskResponse, error) {\n\tresponse, err := g.h.PollForDecisionTask(ctx, proto.ToMatchingPollForDecisionTaskRequest(request))\n\treturn proto.FromMatchingPollForDecisionTaskResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) QueryWorkflow(ctx context.Context, request *matchingv1.QueryWorkflowRequest) (*matchingv1.QueryWorkflowResponse, error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, proto.ToMatchingQueryWorkflowRequest(request))\n\treturn proto.FromMatchingQueryWorkflowResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RefreshTaskListPartitionConfig(ctx context.Context, request *matchingv1.RefreshTaskListPartitionConfigRequest) (*matchingv1.RefreshTaskListPartitionConfigResponse, error) {\n\tresponse, err := g.h.RefreshTaskListPartitionConfig(ctx, proto.ToMatchingRefreshTaskListPartitionConfigRequest(request))\n\treturn proto.FromMatchingRefreshTaskListPartitionConfigResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) RespondQueryTaskCompleted(ctx context.Context, request *matchingv1.RespondQueryTaskCompletedRequest) (*matchingv1.RespondQueryTaskCompletedResponse, error) {\n\terr := g.h.RespondQueryTaskCompleted(ctx, proto.ToMatchingRespondQueryTaskCompletedRequest(request))\n\treturn &matchingv1.RespondQueryTaskCompletedResponse{}, proto.FromError(err)\n}\n\nfunc (g GRPCHandler) UpdateTaskListPartitionConfig(ctx context.Context, request *matchingv1.UpdateTaskListPartitionConfigRequest) (*matchingv1.UpdateTaskListPartitionConfigResponse, error) {\n\tresponse, err := g.h.UpdateTaskListPartitionConfig(ctx, proto.ToMatchingUpdateTaskListPartitionConfigRequest(request))\n\treturn proto.FromMatchingUpdateTaskListPartitionConfigResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/matching/wrappers/thrift/thrift_handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/health\"\n\t\"github.com/uber/cadence/.gen/go/health/metaserver\"\n\t\"github.com/uber/cadence/.gen/go/matching/matchingserviceserver\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/service/matching/handler\"\n)\n\ntype ThriftHandler struct {\n\th handler.Handler\n}\n\nfunc NewThriftHandler(h handler.Handler) ThriftHandler {\n\treturn ThriftHandler{h}\n}\n\nfunc (t ThriftHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(matchingserviceserver.New(&t))\n\tdispatcher.Register(metaserver.New(&t))\n}\n\nfunc (t ThriftHandler) Health(ctx context.Context) (*health.HealthStatus, error) {\n\tresponse, err := t.h.Health(ctx)\n\treturn thrift.FromHealthStatus(response), thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/matching/wrappers/thrift/thrift_handler_generated.go",
    "content": "package thrift\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/thrift.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/.gen/go/matching\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\nfunc (g ThriftHandler) AddActivityTask(ctx context.Context, AddRequest *matching.AddActivityTaskRequest) (err error) {\n\t_, err = g.h.AddActivityTask(ctx, thrift.ToMatchingAddActivityTaskRequest(AddRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) AddDecisionTask(ctx context.Context, AddRequest *matching.AddDecisionTaskRequest) (err error) {\n\t_, err = g.h.AddDecisionTask(ctx, thrift.ToMatchingAddDecisionTaskRequest(AddRequest))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) CancelOutstandingPoll(ctx context.Context, Request *matching.CancelOutstandingPollRequest) (err error) {\n\terr = g.h.CancelOutstandingPoll(ctx, thrift.ToMatchingCancelOutstandingPollRequest(Request))\n\treturn thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) DescribeTaskList(ctx context.Context, Request *matching.DescribeTaskListRequest) (dp1 *shared.DescribeTaskListResponse, err error) {\n\tresponse, err := g.h.DescribeTaskList(ctx, thrift.ToMatchingDescribeTaskListRequest(Request))\n\treturn thrift.FromMatchingDescribeTaskListResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) GetTaskListsByDomain(ctx context.Context, Request *shared.GetTaskListsByDomainRequest) (gp1 *shared.GetTaskListsByDomainResponse, err error) {\n\tresponse, err := g.h.GetTaskListsByDomain(ctx, thrift.ToMatchingGetTaskListsByDomainRequest(Request))\n\treturn thrift.FromMatchingGetTaskListsByDomainResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) ListTaskListPartitions(ctx context.Context, Request *matching.ListTaskListPartitionsRequest) (lp1 *shared.ListTaskListPartitionsResponse, err error) {\n\tresponse, err := g.h.ListTaskListPartitions(ctx, thrift.ToMatchingListTaskListPartitionsRequest(Request))\n\treturn thrift.FromMatchingListTaskListPartitionsResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) PollForActivityTask(ctx context.Context, PollRequest *matching.PollForActivityTaskRequest) (pp1 *shared.PollForActivityTaskResponse, err error) {\n\tresponse, err := g.h.PollForActivityTask(ctx, thrift.ToMatchingPollForActivityTaskRequest(PollRequest))\n\treturn thrift.FromMatchingPollForActivityTaskResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) PollForDecisionTask(ctx context.Context, PollRequest *matching.PollForDecisionTaskRequest) (pp1 *matching.PollForDecisionTaskResponse, err error) {\n\tresponse, err := g.h.PollForDecisionTask(ctx, thrift.ToMatchingPollForDecisionTaskRequest(PollRequest))\n\treturn thrift.FromMatchingPollForDecisionTaskResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) QueryWorkflow(ctx context.Context, QueryRequest *matching.QueryWorkflowRequest) (qp1 *shared.QueryWorkflowResponse, err error) {\n\tresponse, err := g.h.QueryWorkflow(ctx, thrift.ToMatchingQueryWorkflowRequest(QueryRequest))\n\treturn thrift.FromMatchingQueryWorkflowResponse(response), thrift.FromError(err)\n}\n\nfunc (g ThriftHandler) RespondQueryTaskCompleted(ctx context.Context, Request *matching.RespondQueryTaskCompletedRequest) (err error) {\n\terr = g.h.RespondQueryTaskCompleted(ctx, thrift.ToMatchingRespondQueryTaskCompletedRequest(Request))\n\treturn thrift.FromError(err)\n}\n"
  },
  {
    "path": "service/matching/wrappers/thrift/thrift_handler_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage thrift\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/health\"\n\tm \"github.com/uber/cadence/.gen/go/matching\"\n\ts \"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/matching/handler\"\n)\n\nfunc TestThriftHandler(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\th := handler.NewMockHandler(ctrl)\n\tth := NewThriftHandler(h)\n\tctx := context.Background()\n\tinternalErr := &types.InternalServiceError{Message: \"test\"}\n\texpectedErr := &s.InternalServiceError{Message: \"test\"}\n\n\tt.Run(\"Health\", func(t *testing.T) {\n\t\th.EXPECT().Health(ctx).Return(&types.HealthStatus{}, internalErr).Times(1)\n\t\tresp, err := th.Health(ctx)\n\t\tassert.Equal(t, health.HealthStatus{Msg: common.StringPtr(\"\")}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"AddActivityTask\", func(t *testing.T) {\n\t\th.EXPECT().AddActivityTask(ctx, &types.AddActivityTaskRequest{}).Return(nil, internalErr).Times(1)\n\t\terr := th.AddActivityTask(ctx, &m.AddActivityTaskRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"AddDecisionTask\", func(t *testing.T) {\n\t\th.EXPECT().AddDecisionTask(ctx, &types.AddDecisionTaskRequest{}).Return(nil, internalErr).Times(1)\n\t\terr := th.AddDecisionTask(ctx, &m.AddDecisionTaskRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"CancelOutstandingPoll\", func(t *testing.T) {\n\t\th.EXPECT().CancelOutstandingPoll(ctx, &types.CancelOutstandingPollRequest{}).Return(internalErr).Times(1)\n\t\terr := th.CancelOutstandingPoll(ctx, &m.CancelOutstandingPollRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"DescribeTaskList\", func(t *testing.T) {\n\t\th.EXPECT().DescribeTaskList(ctx, &types.MatchingDescribeTaskListRequest{}).Return(&types.DescribeTaskListResponse{}, internalErr).Times(1)\n\t\tresp, err := th.DescribeTaskList(ctx, &m.DescribeTaskListRequest{})\n\t\tassert.Equal(t, s.DescribeTaskListResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"ListTaskListPartitions\", func(t *testing.T) {\n\t\th.EXPECT().ListTaskListPartitions(ctx, &types.MatchingListTaskListPartitionsRequest{}).Return(&types.ListTaskListPartitionsResponse{}, internalErr).Times(1)\n\t\tresp, err := th.ListTaskListPartitions(ctx, &m.ListTaskListPartitionsRequest{})\n\t\tassert.Equal(t, s.ListTaskListPartitionsResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PollForActivityTask\", func(t *testing.T) {\n\t\th.EXPECT().PollForActivityTask(ctx, &types.MatchingPollForActivityTaskRequest{}).Return(&types.MatchingPollForActivityTaskResponse{}, internalErr).Times(1)\n\t\tresp, err := th.PollForActivityTask(ctx, &m.PollForActivityTaskRequest{})\n\t\tassert.Equal(t, s.PollForActivityTaskResponse{WorkflowDomain: common.StringPtr(\"\"), ActivityId: common.StringPtr(\"\"), Attempt: common.Int32Ptr(0)}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"PollForDecisionTask\", func(t *testing.T) {\n\t\th.EXPECT().PollForDecisionTask(ctx, &types.MatchingPollForDecisionTaskRequest{}).Return(&types.MatchingPollForDecisionTaskResponse{}, internalErr).Times(1)\n\t\tresp, err := th.PollForDecisionTask(ctx, &m.PollForDecisionTaskRequest{})\n\t\tassert.Equal(t, m.PollForDecisionTaskResponse{\n\t\t\tStartedEventId:         common.Int64Ptr(0),\n\t\t\tNextEventId:            common.Int64Ptr(0),\n\t\t\tAttempt:                common.Int64Ptr(0),\n\t\t\tStickyExecutionEnabled: common.BoolPtr(false),\n\t\t\tBacklogCountHint:       common.Int64Ptr(0),\n\t\t\tEventStoreVersion:      common.Int32Ptr(0),\n\t\t\tTotalHistoryBytes:      common.Int64Ptr(0),\n\t\t}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"QueryWorkflow\", func(t *testing.T) {\n\t\th.EXPECT().QueryWorkflow(ctx, &types.MatchingQueryWorkflowRequest{}).Return(&types.MatchingQueryWorkflowResponse{}, internalErr).Times(1)\n\t\tresp, err := th.QueryWorkflow(ctx, &m.QueryWorkflowRequest{})\n\t\tassert.Equal(t, s.QueryWorkflowResponse{}, *resp)\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n\tt.Run(\"RespondQueryTaskCompleted\", func(t *testing.T) {\n\t\th.EXPECT().RespondQueryTaskCompleted(ctx, &types.MatchingRespondQueryTaskCompletedRequest{}).Return(internalErr).Times(1)\n\t\terr := th.RespondQueryTaskCompleted(ctx, &m.RespondQueryTaskCompletedRequest{})\n\t\tassert.Equal(t, expectedErr, err)\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/config/config.go",
    "content": "package config\n\n// Config is the configuration for the shard distributor canary\ntype Config struct {\n\tCanary CanaryConfig `yaml:\"canary\"`\n}\n\ntype CanaryConfig struct {\n\t// NumFixedExecutors is the number of executors of fixed namespace\n\t// Values more than 1 will create multiple executors processing the same fixed namespace\n\t// Default: 1\n\tNumFixedExecutors int `yaml:\"numFixedExecutors\"`\n\n\t// NumEphemeralExecutors is the number of executors of ephemeral namespace\n\t// Values more than 1 will create multiple executors processing the same ephemeral namespace\n\t// Default: 1\n\tNumEphemeralExecutors int `yaml:\"numEphemeralExecutors\"`\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/executors/executors.go",
    "content": "package executors\n\nimport (\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/externalshardassignment\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nconst (\n\t// LocalPassthroughNamespace namespace used to test the local passthrough\n\tLocalPassthroughNamespace = \"test-local-passthrough\"\n\t// LocalPassthroughShadowNamespace namespace used to test the local passthrough shadow\n\tLocalPassthroughShadowNamespace = \"test-local-passthrough-shadow\"\n\t// DistributedPassthroughNamespace namespace used to test the distributed passthrough\n\tDistributedPassthroughNamespace = \"test-distributed-passthrough\"\n\t// ExternalAssignmentNamespace namespace used to test the external assignments\n\tExternalAssignmentNamespace = \"test-external-assignment\"\n)\n\ntype ExecutorResult struct {\n\tfx.Out\n\tExecutor executorclient.Executor[*processor.ShardProcessor] `group:\"executor-fixed-proc\"`\n}\n\ntype ExecutorEphemeralResult struct {\n\tfx.Out\n\tExecutor executorclient.Executor[*processorephemeral.ShardProcessor] `group:\"executor-ephemeral-proc\"`\n}\n\ntype ExecutorsResult struct {\n\tfx.Out\n\tExecutors []executorclient.Executor[*processor.ShardProcessor] `group:\"executor-fixed-proc,flatten\"`\n}\n\ntype ExecutorsEphemeralResult struct {\n\tfx.Out\n\tExecutors []executorclient.Executor[*processorephemeral.ShardProcessor] `group:\"executor-ephemeral-proc,flatten\"`\n}\n\nfunc NewExecutorsWithFixedNamespace(params executorclient.Params[*processor.ShardProcessor], namespace string, numExecutors int) (ExecutorsResult, error) {\n\tvar result ExecutorsResult\n\n\tif numExecutors <= 0 {\n\t\tnumExecutors = 1\n\t}\n\n\tfor i := 0; i < numExecutors; i++ {\n\t\texecutor, err := executorclient.NewExecutorWithNamespace(params, namespace)\n\t\tif err != nil {\n\t\t\treturn ExecutorsResult{}, err\n\t\t}\n\t\tresult.Executors = append(result.Executors, executor)\n\t}\n\n\treturn result, nil\n}\n\nfunc NewExecutorWithFixedNamespace(params executorclient.Params[*processor.ShardProcessor], namespace string) (ExecutorResult, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, namespace)\n\treturn ExecutorResult{Executor: executor}, err\n}\n\nfunc NewExecutorsWithEphemeralNamespace(params executorclient.Params[*processorephemeral.ShardProcessor], namespace string, numExecutors int) (ExecutorsEphemeralResult, error) {\n\tvar result ExecutorsEphemeralResult\n\n\tif numExecutors <= 0 {\n\t\tnumExecutors = 1\n\t}\n\n\tfor i := 0; i < numExecutors; i++ {\n\t\texecutor, err := executorclient.NewExecutorWithNamespace(params, namespace)\n\t\tif err != nil {\n\t\t\treturn ExecutorsEphemeralResult{}, err\n\t\t}\n\t\tresult.Executors = append(result.Executors, executor)\n\t}\n\n\treturn result, nil\n}\n\nfunc NewExecutorWithEphemeralNamespace(params executorclient.Params[*processorephemeral.ShardProcessor], namespace string) (ExecutorEphemeralResult, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, namespace)\n\treturn ExecutorEphemeralResult{Executor: executor}, err\n}\n\nfunc NewExecutorLocalPassthroughNamespace(params executorclient.Params[*processor.ShardProcessor]) (ExecutorResult, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, LocalPassthroughNamespace)\n\treturn ExecutorResult{Executor: executor}, err\n}\nfunc NewExecutorLocalPassthroughShadowNamespace(params executorclient.Params[*processorephemeral.ShardProcessor]) (ExecutorEphemeralResult, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, LocalPassthroughShadowNamespace)\n\treturn ExecutorEphemeralResult{Executor: executor}, err\n}\nfunc NewExecutorDistributedPassthroughNamespace(params executorclient.Params[*processor.ShardProcessor]) (ExecutorResult, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, DistributedPassthroughNamespace)\n\treturn ExecutorResult{Executor: executor}, err\n}\n\nfunc NewExecutorExternalAssignmentNamespace(params executorclient.Params[*processorephemeral.ShardProcessor], shardDistributorClient sharddistributor.Client, namespace string) (ExecutorEphemeralResult, *externalshardassignment.ShardAssigner, error) {\n\texecutor, err := executorclient.NewExecutorWithNamespace(params, namespace)\n\tassigner := externalshardassignment.NewShardAssigner(externalshardassignment.ShardAssignerParams{\n\t\tLogger:           params.Logger,\n\t\tTimeSource:       params.TimeSource,\n\t\tShardDistributor: shardDistributorClient,\n\t\tExecutorclient:   executor,\n\t}, namespace)\n\n\treturn ExecutorEphemeralResult{Executor: executor}, assigner, err\n}\n\ntype ExecutorsParams struct {\n\tfx.In\n\tLc                 fx.Lifecycle\n\tExecutorsFixed     []executorclient.Executor[*processor.ShardProcessor]          `group:\"executor-fixed-proc\"`\n\tExecutorsephemeral []executorclient.Executor[*processorephemeral.ShardProcessor] `group:\"executor-ephemeral-proc\"`\n}\n\nfunc NewExecutorsModule(params ExecutorsParams) {\n\tfor _, e := range params.ExecutorsFixed {\n\t\tparams.Lc.Append(fx.StartStopHook(e.Start, e.Stop))\n\t}\n\tfor _, e := range params.Executorsephemeral {\n\t\tparams.Lc.Append(fx.StartStopHook(e.Start, e.Stop))\n\t}\n}\n\nfunc Module(fixedNamespace, ephemeralNamespace, externalAssignmentNamespace string) fx.Option {\n\treturn fx.Module(\"Executors\",\n\t\t// Executor that is used for testing a namespace with fixed shards\n\t\tfx.Provide(func(cfg config.Config, params executorclient.Params[*processor.ShardProcessor]) (ExecutorsResult, error) {\n\t\t\treturn NewExecutorsWithFixedNamespace(params, fixedNamespace, cfg.Canary.NumFixedExecutors)\n\t\t}),\n\n\t\t// Executor that is used for testing a namespaces with ephemeral shards\n\t\tfx.Provide(func(cfg config.Config, params executorclient.Params[*processorephemeral.ShardProcessor]) (ExecutorsEphemeralResult, error) {\n\t\t\treturn NewExecutorsWithEphemeralNamespace(params, ephemeralNamespace, cfg.Canary.NumEphemeralExecutors)\n\t\t}),\n\n\t\t// Executor used for testing a namespace where the shards are assigned externally and reflected in the state of the SD\n\t\t// this is reproducing the behaviour that matching service is going to have during the DistributedPassthrough mode\n\t\tfx.Module(\"Executor-with-external-assignment\",\n\t\t\tfx.Provide(func(params executorclient.Params[*processorephemeral.ShardProcessor], shardDistributorClient sharddistributor.Client) (ExecutorEphemeralResult, *externalshardassignment.ShardAssigner, error) {\n\t\t\t\treturn NewExecutorExternalAssignmentNamespace(params, shardDistributorClient, externalAssignmentNamespace)\n\t\t\t}),\n\t\t\tfx.Invoke(func(lifecycle fx.Lifecycle, shardAssigner *externalshardassignment.ShardAssigner) {\n\t\t\t\tlifecycle.Append(fx.StartStopHook(shardAssigner.Start, shardAssigner.Stop))\n\t\t\t}),\n\t\t),\n\t\tfx.Invoke(NewExecutorsModule),\n\t)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/executors/executors_test.go",
    "content": "package executors\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// mockLifecycle is a simple mock implementation of fx.Lifecycle for testing\ntype mockLifecycle struct {\n\thookCount int\n}\n\nfunc (m *mockLifecycle) Append(hook fx.Hook) {\n\tm.hookCount++\n}\n\nfunc TestNewExecutorsFixedNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := []struct {\n\t\tname        string\n\t\tparams      executorclient.Params[*processor.ShardProcessor]\n\t\tnewExecutor func(params executorclient.Params[*processor.ShardProcessor]) (ExecutorResult, error)\n\t}{\n\t\t{\n\t\t\tname:   \"TestNewExecutorWithFixedNamespace\",\n\t\t\tparams: createMockParams[*processor.ShardProcessor](ctrl, \"shard-distributor-canary\"),\n\t\t\tnewExecutor: func(params executorclient.Params[*processor.ShardProcessor]) (ExecutorResult, error) {\n\t\t\t\treturn NewExecutorWithFixedNamespace(params, \"shard-distributor-canary\")\n\t\t\t}},\n\t\t{\n\t\t\tname:        \"TestNewExecutorLocalPassthroughNamespace\",\n\t\t\tparams:      createMockParams[*processor.ShardProcessor](ctrl, LocalPassthroughNamespace),\n\t\t\tnewExecutor: NewExecutorLocalPassthroughNamespace,\n\t\t},\n\t\t{\n\t\t\tname:        \"TestNewExecutorDistributedPassthroughNamespace\",\n\t\t\tparams:      createMockParams[*processor.ShardProcessor](ctrl, DistributedPassthroughNamespace),\n\t\t\tnewExecutor: NewExecutorDistributedPassthroughNamespace,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := tt.newExecutor(tt.params)\n\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.NotNil(t, result.Executor)\n\t\t})\n\t}\n}\n\nfunc TestNewExecutorsEphemeralNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := []struct {\n\t\tname        string\n\t\tparams      executorclient.Params[*processorephemeral.ShardProcessor]\n\t\tnewExecutor func(params executorclient.Params[*processorephemeral.ShardProcessor]) (ExecutorEphemeralResult, error)\n\t}{\n\t\t{\n\t\t\tname:   \"TestNewExecutorWithEphemeralNamespace\",\n\t\t\tparams: createMockParams[*processorephemeral.ShardProcessor](ctrl, \"shard-distributor-canary-ephemeral\"),\n\t\t\tnewExecutor: func(params executorclient.Params[*processorephemeral.ShardProcessor]) (ExecutorEphemeralResult, error) {\n\t\t\t\treturn NewExecutorWithEphemeralNamespace(params, \"shard-distributor-canary-ephemeral\")\n\t\t\t}},\n\t\t{\n\t\t\tname:        \"TestNewExecutorLocalPassthroughShadowNamespace\",\n\t\t\tparams:      createMockParams[*processorephemeral.ShardProcessor](ctrl, LocalPassthroughShadowNamespace),\n\t\t\tnewExecutor: NewExecutorLocalPassthroughShadowNamespace,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := tt.newExecutor(tt.params)\n\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.NotNil(t, result.Executor)\n\t\t})\n\t}\n}\n\nfunc TestNewExecutorExternalAssignmentNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributor.NewMockClient(ctrl)\n\n\tparams := createMockParams[*processorephemeral.ShardProcessor](ctrl, \"test-external-assignment\")\n\n\tresult, assigner, err := NewExecutorExternalAssignmentNamespace(params, mockShardDistributorClient, \"test-external-assignment\")\n\n\trequire.NoError(t, err)\n\trequire.NotNil(t, result.Executor)\n\trequire.NotNil(t, assigner)\n\t// Verify that the assigner is properly configured and can start/stop\n\tassigner.Start()\n\tdefer assigner.Stop()\n}\n\nfunc TestNewExecutor_InvalidConfig(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockShardProcessorFactory := executorclient.NewMockShardProcessorFactory[*processor.ShardProcessor](ctrl)\n\n\ttests := []struct {\n\t\tname        string\n\t\tparams      executorclient.Params[*processor.ShardProcessor]\n\t\tnewExecutor func(params executorclient.Params[*processor.ShardProcessor]) (ExecutorResult, error)\n\t\terrorString string\n\t}{\n\t\t{\n\t\t\tname: \"No namespaces configured\",\n\t\t\tparams: executorclient.Params[*processor.ShardProcessor]{\n\t\t\t\tMetricsScope:          tally.NoopScope,\n\t\t\t\tLogger:                log.NewNoop(),\n\t\t\t\tShardProcessorFactory: mockShardProcessorFactory,\n\t\t\t\tConfig: clientcommon.Config{\n\t\t\t\t\tNamespaces: []clientcommon.NamespaceConfig{},\n\t\t\t\t},\n\t\t\t\tTimeSource: clock.NewMockedTimeSource(),\n\t\t\t},\n\t\t\terrorString: \"at least one namespace must be configured\",\n\t\t},\n\t\t{\n\t\t\tname: \"No valid namespace\",\n\t\t\tparams: executorclient.Params[*processor.ShardProcessor]{\n\t\t\t\tMetricsScope:          tally.NoopScope,\n\t\t\t\tLogger:                log.NewNoop(),\n\t\t\t\tShardProcessorFactory: mockShardProcessorFactory,\n\t\t\t\tConfig: clientcommon.Config{\n\t\t\t\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tNamespace:         \"wrong-namespace\",\n\t\t\t\t\t\t\tHeartBeatInterval: 5 * time.Second,\n\t\t\t\t\t\t\tMigrationMode:     \"onboarded\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTimeSource: clock.NewMockedTimeSource(),\n\t\t\t},\n\t\t\terrorString: \"namespace shard-distributor-canary not found in config\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t_, err := NewExecutorWithFixedNamespace(tt.params, \"shard-distributor-canary\")\n\n\t\t\trequire.Error(t, err)\n\t\t\tassert.Contains(t, err.Error(), tt.errorString)\n\t\t})\n\t}\n}\n\nfunc TestNewExecutorsModule(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\t// Create a mock lifecycle\n\ttests := []struct {\n\t\tname               string\n\t\tparams             ExecutorsParams\n\t\texpectedInvocation int\n\t}{\n\t\t{\n\t\t\tname: \"multiple executors\",\n\t\t\tparams: ExecutorsParams{\n\t\t\t\tExecutorsFixed: []executorclient.Executor[*processor.ShardProcessor]{\n\t\t\t\t\texecutorclient.NewMockExecutor[*processor.ShardProcessor](ctrl),\n\t\t\t\t\texecutorclient.NewMockExecutor[*processor.ShardProcessor](ctrl),\n\t\t\t\t},\n\t\t\t\tExecutorsephemeral: []executorclient.Executor[*processorephemeral.ShardProcessor]{\n\t\t\t\t\texecutorclient.NewMockExecutor[*processorephemeral.ShardProcessor](ctrl),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedInvocation: 3,\n\t\t},\n\t\t{\n\t\t\tname: \"no executors\",\n\t\t\tparams: ExecutorsParams{\n\t\t\t\tExecutorsFixed:     []executorclient.Executor[*processor.ShardProcessor]{},\n\t\t\t\tExecutorsephemeral: []executorclient.Executor[*processorephemeral.ShardProcessor]{},\n\t\t\t},\n\t\t\texpectedInvocation: 0,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockLifecycle := &mockLifecycle{}\n\t\t\ttt.params.Lc = mockLifecycle\n\t\t\t// Call NewExecutorsModule - it should not panic or error\n\t\t\t// The function doesn't return anything, so we just verify it executes successfully\n\t\t\trequire.NotPanics(t, func() {\n\t\t\t\tNewExecutorsModule(tt.params)\n\t\t\t})\n\t\t\t// Verify that lifecycle hooks were registered for all executors\n\t\t\tassert.Equal(t, tt.expectedInvocation, mockLifecycle.hookCount)\n\t\t})\n\t}\n}\n\n// Helper functions to create mock parameters\nfunc createMockParams[SP executorclient.ShardProcessor](\n\tctrl *gomock.Controller,\n\tnamespace string,\n) executorclient.Params[SP] {\n\tmockShardProcessorFactory := executorclient.NewMockShardProcessorFactory[SP](ctrl)\n\n\treturn executorclient.Params[SP]{\n\t\tMetricsScope:          tally.NoopScope,\n\t\tLogger:                log.NewNoop(),\n\t\tShardProcessorFactory: mockShardProcessorFactory,\n\t\tConfig: clientcommon.Config{\n\t\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t\t{\n\t\t\t\t\tNamespace:         namespace,\n\t\t\t\t\tHeartBeatInterval: 5 * time.Second,\n\t\t\t\t\tMigrationMode:     \"onboarded\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tTimeSource: clock.NewMockedTimeSource(),\n\t}\n}\n\nfunc TestNewExecutorsWithFixedNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := []struct {\n\t\tname         string\n\t\tnamespace    string\n\t\tnumExecutors int\n\t\texpected     int\n\t}{\n\t\t{\n\t\t\tname:         \"zero executors defaults to one\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 0,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"negative executors defaults to one\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: -1,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"one executor\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 1,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"two executors\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 2,\n\t\t\texpected:     2,\n\t\t},\n\t\t{\n\t\t\tname:         \"five executors\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 5,\n\t\t\texpected:     5,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tparams := createMockParams[*processor.ShardProcessor](ctrl, tt.namespace)\n\t\t\tresult, err := NewExecutorsWithFixedNamespace(params, tt.namespace, tt.numExecutors)\n\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Len(t, result.Executors, tt.expected)\n\t\t\tfor _, executor := range result.Executors {\n\t\t\t\tassert.NotNil(t, executor)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNewExecutorsWithEphemeralNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := []struct {\n\t\tname         string\n\t\tnamespace    string\n\t\tnumExecutors int\n\t\texpected     int\n\t}{\n\t\t{\n\t\t\tname:         \"zero executors defaults to one\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 0,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"negative executors defaults to one\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: -1,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"one executor\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 1,\n\t\t\texpected:     1,\n\t\t},\n\t\t{\n\t\t\tname:         \"two executors\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 2,\n\t\t\texpected:     2,\n\t\t},\n\t\t{\n\t\t\tname:         \"five executors\",\n\t\t\tnamespace:    \"test-namespace\",\n\t\t\tnumExecutors: 5,\n\t\t\texpected:     5,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tparams := createMockParams[*processorephemeral.ShardProcessor](ctrl, tt.namespace)\n\t\t\tresult, err := NewExecutorsWithEphemeralNamespace(params, tt.namespace, tt.numExecutors)\n\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Len(t, result.Executors, tt.expected)\n\t\t\tfor _, executor := range result.Executors {\n\t\t\t\tassert.NotNil(t, executor)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/externalshardassignment/shardassigner.go",
    "content": "package externalshardassignment\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"strconv\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nconst (\n\tshardAssignmentInterval = 4 * time.Second\n\tminimumAssignedShards   = 3\n)\n\n// ShardAssigneer assigns shards to the executor for canary testing\ntype ShardAssigner struct {\n\tlogger          log.Logger\n\ttimeSource      clock.TimeSource\n\tshardProcessors map[string]*processorephemeral.ShardProcessor\n\texecutorclient  executorclient.Executor[*processorephemeral.ShardProcessor]\n\tstopChan        chan struct{}\n\tgoRoutineWg     sync.WaitGroup\n\tnamespace       string\n}\n\n// ShardAssignerParams contains the dependencies needed to create a ShardParam\ntype ShardAssignerParams struct {\n\tLogger           log.Logger\n\tTimeSource       clock.TimeSource\n\tShardDistributor sharddistributor.Client\n\tExecutorclient   executorclient.Executor[*processorephemeral.ShardProcessor]\n}\n\n// NewShardCreator creates a new ShardCreator instance with the given parameters and namespace\nfunc NewShardAssigner(params ShardAssignerParams, namespace string) *ShardAssigner {\n\tsp := make(map[string]*processorephemeral.ShardProcessor)\n\treturn &ShardAssigner{\n\t\tlogger:          params.Logger,\n\t\ttimeSource:      params.TimeSource,\n\t\tshardProcessors: sp,\n\t\texecutorclient:  params.Executorclient,\n\t\tstopChan:        make(chan struct{}),\n\t\tgoRoutineWg:     sync.WaitGroup{},\n\t\tnamespace:       namespace,\n\t}\n}\n\n// Start begins the shard creation process in a background goroutine\nfunc (s *ShardAssigner) Start() {\n\ts.goRoutineWg.Add(1)\n\tgo s.process(context.Background())\n\n\ts.logger.Info(\"Shard assigner started\")\n}\n\n// Stop stops the shard creation process and waits for the goroutine to finish\nfunc (s *ShardAssigner) Stop() {\n\tclose(s.stopChan)\n\ts.goRoutineWg.Wait()\n}\n\n// ShardCreatorModule creates an fx module for the shard creator with the given namespace\nfunc ShardAssignerModule(namespace string) fx.Option {\n\treturn fx.Module(\"shard-assigner\",\n\t\tfx.Provide(func(params ShardAssignerParams) *ShardAssigner {\n\n\t\t\treturn NewShardAssigner(params, namespace)\n\t\t}),\n\t\tfx.Invoke(func(lifecycle fx.Lifecycle, shardAssigner *ShardAssigner) {\n\t\t\tlifecycle.Append(fx.StartStopHook(shardAssigner.Start, shardAssigner.Stop))\n\t\t}),\n\t)\n}\n\nfunc (s *ShardAssigner) process(ctx context.Context) {\n\tdefer s.goRoutineWg.Done()\n\n\tticker := s.timeSource.NewTicker(shardAssignmentInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn\n\t\tcase <-s.stopChan:\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tif len(s.shardProcessors) > minimumAssignedShards {\n\t\t\t\tvar shardToRemove string\n\t\t\t\tshardToRemoveIndex := rand.Intn(len(s.shardProcessors))\n\t\t\t\tfor shardID := range s.shardProcessors {\n\t\t\t\t\tif shardToRemoveIndex == 0 {\n\t\t\t\t\t\tshardToRemove = shardID\n\t\t\t\t\t\tbreak\n\t\t\t\t\t}\n\t\t\t\t\tshardToRemoveIndex--\n\t\t\t\t}\n\t\t\t\terr := s.executorclient.RemoveShardsFromLocalLogic([]string{shardToRemove})\n\t\t\t\tif err != nil {\n\t\t\t\t\ts.logger.Error(\"Failed to remove shards\", tag.Error(err))\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tdelete(s.shardProcessors, shardToRemove)\n\t\t\t\ts.logger.Info(\"Removed a shard from external source\", tag.ShardKey(shardToRemove))\n\n\t\t\t}\n\n\t\t\t// Simulate the assignment of new shards\n\t\t\tnewAssignedShard := uuid.New().String()\n\t\t\ts.logger.Info(\"Assign a new shard from external source\", tag.ShardKey(newAssignedShard))\n\t\t\tshardAssignment := map[string]*types.ShardAssignment{\n\t\t\t\tnewAssignedShard: {\n\t\t\t\t\tStatus: types.AssignmentStatusREADY,\n\t\t\t\t},\n\t\t\t}\n\t\t\terr := s.executorclient.AssignShardsFromLocalLogic(context.Background(), shardAssignment)\n\t\t\tif err != nil {\n\t\t\t\ts.logger.Error(\"Failed to assign shard from external source\", tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tsp, err := s.executorclient.GetShardProcess(ctx, newAssignedShard)\n\t\t\tif err != nil {\n\t\t\t\ts.logger.Error(\"failed to get shard assigned\", tag.ShardKey(newAssignedShard), tag.Error(err))\n\t\t\t} else {\n\t\t\t\ts.logger.Info(\"shard assigned\", tag.ShardStatus(string(sp.GetShardReport().Status)), tag.ShardLoad(strconv.FormatFloat(sp.GetShardReport().ShardLoad, 'f', -1, 64)))\n\t\t\t}\n\t\t\ts.shardProcessors[newAssignedShard] = sp\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/externalshardassignment/shardassigner_test.go",
    "content": "package externalshardassignment\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc TestSShardAssigner_Lifecycle(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\n\tlogger := log.NewNoop()\n\ttimeSource := clock.NewMockedTimeSource()\n\tctrl := gomock.NewController(t)\n\texecutorclientmock := executorclient.NewMockExecutor[*processorephemeral.ShardProcessor](ctrl)\n\n\tnamespace := \"test-namespace\"\n\n\texecutorclientmock.EXPECT().AssignShardsFromLocalLogic(gomock.Any(), gomock.Any())\n\texecutorclientmock.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\texecutorclientmock.EXPECT().AssignShardsFromLocalLogic(gomock.Any(), gomock.Any())\n\texecutorclientmock.EXPECT().GetShardProcess(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\n\tparams := ShardAssignerParams{\n\t\tLogger:         logger,\n\t\tTimeSource:     timeSource,\n\t\tExecutorclient: executorclientmock,\n\t}\n\n\tassigner := NewShardAssigner(params, namespace)\n\tassigner.Start()\n\n\t// Wait for the goroutine to start\n\ttimeSource.BlockUntil(1)\n\n\t// Trigger shard creation that will fail\n\ttimeSource.Advance(shardAssignmentInterval + 100*time.Millisecond)\n\ttime.Sleep(10 * time.Millisecond) // Allow processing\n\n\t// Trigger another shard creation to ensure processing continues after error\n\ttimeSource.Advance(shardAssignmentInterval + 100*time.Millisecond)\n\ttime.Sleep(10 * time.Millisecond) // Allow processing\n\n\tassigner.Stop()\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/factory/factory.go",
    "content": "package factory\n\nimport (\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// ShardProcessorFactory is a generic factory for creating ShardProcessor instances.\ntype ShardProcessorFactory[T executorclient.ShardProcessor] struct {\n\tlogger      *zap.Logger\n\ttimeSource  clock.TimeSource\n\tconstructor func(shardID string, timeSource clock.TimeSource, logger *zap.Logger) T\n}\n\n// NewShardProcessor creates a new ShardProcessor.\nfunc (s *ShardProcessorFactory[T]) NewShardProcessor(shardID string) (T, error) {\n\treturn s.constructor(shardID, s.timeSource, s.logger), nil\n}\n\n// Params are the parameters for creating a ShardProcessorFactory.\ntype Params struct {\n\tfx.In\n\n\tLogger     *zap.Logger\n\tTimeSource clock.TimeSource\n}\n\n// NewShardProcessorFactory creates a new ShardProcessorFactory with a constructor function.\nfunc NewShardProcessorFactory[T executorclient.ShardProcessor](\n\tparams Params,\n\tconstructor func(shardID string, timeSource clock.TimeSource, logger *zap.Logger) T,\n) executorclient.ShardProcessorFactory[T] {\n\treturn &ShardProcessorFactory[T]{\n\t\tlogger:      params.Logger,\n\t\ttimeSource:  params.TimeSource,\n\t\tconstructor: constructor,\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/factory/factory_test.go",
    "content": "package factory\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/zap/zaptest\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc TestNewShardProcessorFactory(t *testing.T) {\n\tlogger := zaptest.NewLogger(t)\n\ttimeSource := clock.NewRealTimeSource()\n\n\tparams := Params{\n\t\tLogger:     logger,\n\t\tTimeSource: timeSource,\n\t}\n\n\tfactory := NewShardProcessorFactory(params, processor.NewShardProcessor)\n\n\t// Test that the factory implements the correct interface\n\tvar _ executorclient.ShardProcessorFactory[*processor.ShardProcessor] = factory\n\n\t// Test creating a processor\n\tprocessor, err := factory.NewShardProcessor(\"test-shard-1\")\n\trequire.NoError(t, err)\n\tassert.NotNil(t, processor)\n}\n\nfunc TestShardProcessorFactory_NewShardProcessor(t *testing.T) {\n\tlogger := zaptest.NewLogger(t)\n\ttimeSource := clock.NewRealTimeSource()\n\n\tfactory := &ShardProcessorFactory[*processor.ShardProcessor]{\n\t\tlogger:      logger,\n\t\ttimeSource:  timeSource,\n\t\tconstructor: processor.NewShardProcessor,\n\t}\n\n\t// Test creating multiple processors\n\tprocessor1, err := factory.NewShardProcessor(\"shard-1\")\n\trequire.NoError(t, err)\n\n\tprocessor2, err := factory.NewShardProcessor(\"shard-2\")\n\trequire.NoError(t, err)\n\n\t// Ensure they are different instances\n\tassert.NotEqual(t, processor1, processor2)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/handler/ping_handler.go",
    "content": "package handler\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// PingHandler handles ping requests to verify executor ownership of shards\ntype PingHandler struct {\n\tlogger             *zap.Logger\n\texecutorsFixed     map[string]executorclient.Executor[*processor.ShardProcessor]          // namespace -> executor\n\texecutorsEphemeral map[string]executorclient.Executor[*processorephemeral.ShardProcessor] // namespace -> executor\n}\n\n// Params are the parameters for creating a PingHandler\ntype Params struct {\n\tfx.In\n\n\tLogger *zap.Logger\n\n\tExecutorsFixed     []executorclient.Executor[*processor.ShardProcessor]          `group:\"executor-fixed-proc\"`\n\tExecutorsEphemeral []executorclient.Executor[*processorephemeral.ShardProcessor] `group:\"executor-ephemeral-proc\"`\n}\n\n// NewPingHandler creates a new PingHandler\nfunc NewPingHandler(params Params) *PingHandler {\n\t// Create maps of executors for quick lookup\n\texecutorsFixed := make(map[string]executorclient.Executor[*processor.ShardProcessor])\n\tfor _, executor := range params.ExecutorsFixed {\n\t\texecutorsFixed[executor.GetNamespace()] = executor\n\t}\n\texecutorsEphemeral := make(map[string]executorclient.Executor[*processorephemeral.ShardProcessor])\n\tfor _, executor := range params.ExecutorsEphemeral {\n\t\texecutorsEphemeral[executor.GetNamespace()] = executor\n\t}\n\n\t// Return the handler\n\treturn &PingHandler{\n\t\tlogger:             params.Logger,\n\t\texecutorsFixed:     executorsFixed,\n\t\texecutorsEphemeral: executorsEphemeral,\n\t}\n}\n\n// Ping handles ping requests to check shard ownership\nfunc (h *PingHandler) Ping(ctx context.Context, request *sharddistributorv1.PingRequest) (*sharddistributorv1.PingResponse, error) {\n\th.logger.Info(\"Received ping request\",\n\t\tzap.String(\"shard_key\", request.GetShardKey()),\n\t\tzap.String(\"namespace\", request.GetNamespace()))\n\n\tnamespace := request.GetNamespace()\n\n\t// Check fixed executors\n\tif executor, found := h.executorsFixed[namespace]; found {\n\t\treturn checkOwnerShipAndLog(ctx, executor, request, h), nil\n\t}\n\n\t// Check ephemeral executors\n\tif executor, found := h.executorsEphemeral[namespace]; found {\n\t\treturn checkOwnerShipAndLog(ctx, executor, request, h), nil\n\t}\n\n\t// Namespace not found\n\th.logger.Warn(\"Namespace executor not found\",\n\t\tzap.String(\"namespace\", namespace))\n\n\treturn &sharddistributorv1.PingResponse{\n\t\tExecutorId: \"\",\n\t\tOwnsShard:  false,\n\t\tShardKey:   request.GetShardKey(),\n\t}, nil\n}\n\nfunc checkOwnerShipAndLog[T executorclient.ShardProcessor](ctx context.Context, executor executorclient.Executor[T], request *sharddistributorv1.PingRequest, h *PingHandler) *sharddistributorv1.PingResponse {\n\t// We just check that we have a processor for the shard\n\t_, err := executor.GetShardProcess(ctx, request.GetShardKey())\n\townshard := err == nil\n\n\tmetadata := executor.GetMetadata()\n\texecutorID := getExecutorID(metadata)\n\n\tresponse := &sharddistributorv1.PingResponse{\n\t\tExecutorId: executorID,\n\t\tOwnsShard:  ownshard,\n\t\tShardKey:   request.GetShardKey(),\n\t}\n\n\th.logger.Info(\"Responding to ping\",\n\t\tzap.String(\"shard_key\", request.GetShardKey()),\n\t\tzap.Bool(\"owns_shard\", ownshard),\n\t\tzap.String(\"executor_id\", executorID))\n\n\treturn response\n}\n\nfunc getExecutorID(metadata map[string]string) string {\n\tif addr, ok := metadata[\"grpc_address\"]; ok && addr != \"\" {\n\t\treturn addr\n\t}\n\treturn \"\"\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/handler/ping_handler_test.go",
    "content": "package handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc TestPingHandler_Ping(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\tnamespace string\n\t\tshardKey  string\n\t\tsetup     func(*gomock.Controller) ([]executorclient.Executor[*processor.ShardProcessor], []executorclient.Executor[*processorephemeral.ShardProcessor])\n\t\twantID    string\n\t\twantOwns  bool\n\t}{\n\t\t{\n\t\t\tname:      \"fixed executor owns shard\",\n\t\t\tnamespace: \"ns1\",\n\t\t\tshardKey:  \"shard-1\",\n\t\t\tsetup: func(ctrl *gomock.Controller) ([]executorclient.Executor[*processor.ShardProcessor], []executorclient.Executor[*processorephemeral.ShardProcessor]) {\n\t\t\t\texec := executorclient.NewMockExecutor[*processor.ShardProcessor](ctrl)\n\t\t\t\texec.EXPECT().GetNamespace().Return(\"ns1\").AnyTimes()\n\t\t\t\texec.EXPECT().GetShardProcess(gomock.Any(), \"shard-1\").Return(&processor.ShardProcessor{}, nil)\n\t\t\t\texec.EXPECT().GetMetadata().Return(map[string]string{\"grpc_address\": \"127.0.0.1:7953\"})\n\t\t\t\treturn []executorclient.Executor[*processor.ShardProcessor]{exec}, nil\n\t\t\t},\n\t\t\twantID:   \"127.0.0.1:7953\",\n\t\t\twantOwns: true,\n\t\t},\n\t\t{\n\t\t\tname:      \"fixed executor does not own shard\",\n\t\t\tnamespace: \"ns1\",\n\t\t\tshardKey:  \"shard-2\",\n\t\t\tsetup: func(ctrl *gomock.Controller) ([]executorclient.Executor[*processor.ShardProcessor], []executorclient.Executor[*processorephemeral.ShardProcessor]) {\n\t\t\t\texec := executorclient.NewMockExecutor[*processor.ShardProcessor](ctrl)\n\t\t\t\texec.EXPECT().GetNamespace().Return(\"ns1\").AnyTimes()\n\t\t\t\texec.EXPECT().GetShardProcess(gomock.Any(), \"shard-2\").Return(nil, errors.New(\"not found\"))\n\t\t\t\texec.EXPECT().GetMetadata().Return(map[string]string{\"grpc_address\": \"127.0.0.1:7954\"})\n\t\t\t\treturn []executorclient.Executor[*processor.ShardProcessor]{exec}, nil\n\t\t\t},\n\t\t\twantID:   \"127.0.0.1:7954\",\n\t\t\twantOwns: false,\n\t\t},\n\t\t{\n\t\t\tname:      \"ephemeral executor owns shard\",\n\t\t\tnamespace: \"ns2\",\n\t\t\tshardKey:  \"shard-3\",\n\t\t\tsetup: func(ctrl *gomock.Controller) ([]executorclient.Executor[*processor.ShardProcessor], []executorclient.Executor[*processorephemeral.ShardProcessor]) {\n\t\t\t\texec := executorclient.NewMockExecutor[*processorephemeral.ShardProcessor](ctrl)\n\t\t\t\texec.EXPECT().GetNamespace().Return(\"ns2\").AnyTimes()\n\t\t\t\texec.EXPECT().GetShardProcess(gomock.Any(), \"shard-3\").Return(&processorephemeral.ShardProcessor{}, nil)\n\t\t\t\texec.EXPECT().GetMetadata().Return(map[string]string{\"grpc_address\": \"127.0.0.1:7955\"})\n\t\t\t\treturn nil, []executorclient.Executor[*processorephemeral.ShardProcessor]{exec}\n\t\t\t},\n\t\t\twantID:   \"127.0.0.1:7955\",\n\t\t\twantOwns: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tfixed, ephemeral := tt.setup(ctrl)\n\t\t\thandler := NewPingHandler(Params{\n\t\t\t\tLogger:             zap.NewNop(),\n\t\t\t\tExecutorsFixed:     fixed,\n\t\t\t\tExecutorsEphemeral: ephemeral,\n\t\t\t})\n\n\t\t\tresp, err := handler.Ping(context.Background(), &sharddistributorv1.PingRequest{\n\t\t\t\tNamespace: tt.namespace,\n\t\t\t\tShardKey:  tt.shardKey,\n\t\t\t})\n\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tt.wantID, resp.ExecutorId)\n\t\t\tassert.Equal(t, tt.wantOwns, resp.OwnsShard)\n\t\t\tassert.Equal(t, tt.shardKey, resp.ShardKey)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/module.go",
    "content": "package canary\n\nimport (\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/executors\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/factory\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/handler\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/pinger\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processor\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/processorephemeral\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/sharddistributorclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/sharddistributorexecutorclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n)\n\ntype NamespacesNames struct {\n\tfx.In\n\tFixedNamespace              string\n\tEphemeralNamespace          string\n\tExternalAssignmentNamespace string\n\tSharddistributorServiceName string\n\n\tConfig config.Config\n}\n\nfunc Module(namespacesNames NamespacesNames) fx.Option {\n\treturn fx.Module(\"shard-distributor-canary\", opts(namespacesNames))\n}\n\nfunc opts(names NamespacesNames) fx.Option {\n\treturn fx.Options(\n\t\tfx.Supply(names.Config),\n\n\t\tfx.Provide(sharddistributorv1.NewFxShardDistributorExecutorAPIYARPCClient(names.SharddistributorServiceName)),\n\t\tfx.Provide(sharddistributorv1.NewFxShardDistributorAPIYARPCClient(names.SharddistributorServiceName)),\n\n\t\tfx.Provide(sharddistributorclient.NewShardDistributorClient),\n\t\tfx.Provide(sharddistributorexecutorclient.NewShardDistributorExecutorClient),\n\n\t\t// Modules for the shard distributor canary\n\t\tfx.Provide(\n\t\t\tfunc(params factory.Params) executorclient.ShardProcessorFactory[*processor.ShardProcessor] {\n\t\t\t\treturn factory.NewShardProcessorFactory(params, processor.NewShardProcessor)\n\t\t\t},\n\t\t\tfunc(params factory.Params) executorclient.ShardProcessorFactory[*processorephemeral.ShardProcessor] {\n\t\t\t\treturn factory.NewShardProcessorFactory(params, processorephemeral.NewShardProcessor)\n\t\t\t},\n\t\t),\n\n\t\t// Simple way to instantiate executor if only one namespace is used\n\t\t// executorclient.ModuleWithNamespace[*processor.ShardProcessor](names.FixedNamespace),\n\t\t// executorclient.ModuleWithNamespace[*processorephemeral.ShardProcessor](names.EphemeralNamespace),\n\n\t\t// Instantiate executors for multiple namespaces\n\t\texecutors.Module(names.FixedNamespace, names.EphemeralNamespace, names.ExternalAssignmentNamespace),\n\n\t\tprocessorephemeral.ShardCreatorModule([]string{names.EphemeralNamespace}),\n\n\t\tspectatorclient.Module(),\n\t\tfx.Provide(spectatorclient.NewSpectatorPeerChooser),\n\t\tfx.Invoke(func(chooser spectatorclient.SpectatorPeerChooserInterface, lc fx.Lifecycle) {\n\t\t\tlc.Append(fx.StartStopHook(chooser.Start, chooser.Stop))\n\t\t}),\n\n\t\t// Create canary client using the dispatcher's client config\n\t\tfx.Provide(func(dispatcher *yarpc.Dispatcher) sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient {\n\t\t\tconfig := dispatcher.ClientConfig(\"shard-distributor-canary\")\n\t\t\treturn sharddistributorv1.NewShardDistributorExecutorCanaryAPIYARPCClient(config)\n\t\t}),\n\n\t\tfx.Provide(func(params pinger.Params) *pinger.Pinger {\n\t\t\treturn pinger.NewPinger(params, names.FixedNamespace, 32)\n\t\t}),\n\t\tfx.Invoke(func(p *pinger.Pinger, lc fx.Lifecycle) {\n\t\t\tlc.Append(fx.StartStopHook(p.Start, p.Stop))\n\t\t}),\n\n\t\t// Register canary ping handler to receive ping requests from other executors\n\t\tfx.Provide(handler.NewPingHandler),\n\t\tfx.Provide(fx.Annotate(\n\t\t\tfunc(h *handler.PingHandler) sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCServer {\n\t\t\t\treturn h\n\t\t\t},\n\t\t)),\n\t\tfx.Provide(sharddistributorv1.NewFxShardDistributorExecutorCanaryAPIYARPCProcedures()),\n\n\t\t// There is a circular dependency between the spectator client and the peer chooser, since\n\t\t// the yarpc dispatcher needs the peer chooser and the peer chooser needs the spectators, which needs the yarpc dispatcher.\n\t\t// To break the circular dependency, we set the spectators on the peer chooser here.\n\t\tfx.Invoke(func(chooser spectatorclient.SpectatorPeerChooserInterface, spectators *spectatorclient.Spectators) {\n\t\t\tchooser.SetSpectators(spectators)\n\t\t}),\n\t)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/module_test.go",
    "content": "package canary\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/golang/mock/gomock\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\tubergomock \"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport/transporttest\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/yarpctest\"\n\t\"go.uber.org/zap/zaptest\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\nfunc TestModule(t *testing.T) {\n\t// Create mocks\n\tctrl := gomock.NewController(t)\n\tuberCtrl := ubergomock.NewController(t)\n\tmockLogger := log.NewNoop()\n\n\tmockClientConfig := transporttest.NewMockClientConfig(ctrl)\n\ttransport := grpc.NewTransport()\n\toutbound := transport.NewOutbound(yarpctest.NewFakePeerList())\n\n\tmockClientConfig.EXPECT().Caller().Return(\"test-executor\").AnyTimes()\n\tmockClientConfig.EXPECT().Service().Return(\"shard-distributor\").AnyTimes()\n\tmockClientConfig.EXPECT().GetUnaryOutbound().Return(outbound).AnyTimes()\n\n\tmockClientConfigProvider := transporttest.NewMockClientConfigProvider(ctrl)\n\tmockClientConfigProvider.EXPECT().ClientConfig(\"cadence-shard-distributor\").Return(mockClientConfig).AnyTimes()\n\n\t// Create executor yarpc client mock\n\tmockYARPCClient := executorclient.NewMockShardDistributorExecutorAPIYARPCClient(uberCtrl)\n\tmockYARPCClient.EXPECT().\n\t\tHeartbeat(ubergomock.Any(), ubergomock.Any(), ubergomock.Any()).\n\t\tReturn(&sharddistributorv1.HeartbeatResponse{}, nil).\n\t\tAnyTimes()\n\n\tconfig := clientcommon.Config{\n\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t{Namespace: \"shard-distributor-canary\", HeartBeatInterval: 5 * time.Second, MigrationMode: \"onboarded\"},\n\t\t\t{Namespace: \"shard-distributor-canary-ephemeral\", HeartBeatInterval: 5 * time.Second, MigrationMode: \"onboarded\"},\n\t\t\t{Namespace: \"test-local-passthrough\", HeartBeatInterval: 1 * time.Second, MigrationMode: \"local_pass\"},\n\t\t\t{Namespace: \"test-local-passthrough-shadow\", HeartBeatInterval: 1 * time.Second, MigrationMode: \"local_pass_shadow\"},\n\t\t\t{Namespace: \"test-distributed-passthrough\", HeartBeatInterval: 1 * time.Second, MigrationMode: \"distributed_pass\"},\n\t\t\t{Namespace: \"test-external-assignment\", HeartBeatInterval: 1 * time.Second, MigrationMode: \"distributed_pass\"},\n\t\t},\n\t}\n\n\t// Create a mock dispatcher with the required outbound\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: \"test-canary\",\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\"shard-distributor-canary\": {\n\t\t\t\tUnary: outbound,\n\t\t\t},\n\t\t},\n\t})\n\n\t// Create a test app with the library, check that it starts and stops\n\tfxtest.New(t,\n\t\tfx.Supply(\n\t\t\tfx.Annotate(tally.NoopScope, fx.As(new(tally.Scope))),\n\t\t\tfx.Annotate(clock.NewMockedTimeSource(), fx.As(new(clock.TimeSource))),\n\t\t\tfx.Annotate(mockLogger, fx.As(new(log.Logger))),\n\t\t\tfx.Annotate(mockClientConfigProvider, fx.As(new(yarpc.ClientConfig))),\n\t\t\tfx.Annotate(transport, fx.As(new(peer.Transport))),\n\t\t\tzaptest.NewLogger(t),\n\t\t\tconfig,\n\t\t\tdispatcher,\n\t\t),\n\t\t// Replacing the real YARPC client with mock to handle the draining heartbeat\n\t\tfx.Decorate(func() sharddistributorv1.ShardDistributorExecutorAPIYARPCClient {\n\t\t\treturn mockYARPCClient\n\t\t}),\n\t\tModule(NamespacesNames{\n\t\t\tFixedNamespace:              \"shard-distributor-canary\",\n\t\t\tEphemeralNamespace:          \"shard-distributor-canary-ephemeral\",\n\t\t\tExternalAssignmentNamespace: \"test-external-assignment\",\n\t\t\tSharddistributorServiceName: \"cadence-shard-distributor\",\n\t\t}),\n\t).RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/pinger/canary_client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/.gen/proto/sharddistributor/v1 (interfaces: ShardDistributorExecutorCanaryAPIYARPCClient)\n//\n// Generated by this command:\n//\n//\tmockgen -package pinger -destination canary_client_mock.go github.com/uber/cadence/.gen/proto/sharddistributor/v1 ShardDistributorExecutorCanaryAPIYARPCClient\n//\n\n// Package pinger is a generated GoMock package.\npackage pinger\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n)\n\n// MockShardDistributorExecutorCanaryAPIYARPCClient is a mock of ShardDistributorExecutorCanaryAPIYARPCClient interface.\ntype MockShardDistributorExecutorCanaryAPIYARPCClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder is the mock recorder for MockShardDistributorExecutorCanaryAPIYARPCClient.\ntype MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder struct {\n\tmock *MockShardDistributorExecutorCanaryAPIYARPCClient\n}\n\n// NewMockShardDistributorExecutorCanaryAPIYARPCClient creates a new mock instance.\nfunc NewMockShardDistributorExecutorCanaryAPIYARPCClient(ctrl *gomock.Controller) *MockShardDistributorExecutorCanaryAPIYARPCClient {\n\tmock := &MockShardDistributorExecutorCanaryAPIYARPCClient{ctrl: ctrl}\n\tmock.recorder = &MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardDistributorExecutorCanaryAPIYARPCClient) EXPECT() *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder {\n\treturn m.recorder\n}\n\n// Ping mocks base method.\nfunc (m *MockShardDistributorExecutorCanaryAPIYARPCClient) Ping(arg0 context.Context, arg1 *sharddistributorv1.PingRequest, arg2 ...yarpc.CallOption) (*sharddistributorv1.PingResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Ping\", varargs...)\n\tret0, _ := ret[0].(*sharddistributorv1.PingResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Ping indicates an expected call of Ping.\nfunc (mr *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder) Ping(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ping\", reflect.TypeOf((*MockShardDistributorExecutorCanaryAPIYARPCClient)(nil).Ping), varargs...)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/pinger/pingAndLog.go",
    "content": "package pinger\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient\"\n)\n\nconst (\n\tpingTimeout = 5 * time.Second\n)\n\nfunc PingShard(ctx context.Context, canaryClient sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient, logger *zap.Logger, namespace, shardKey string) {\n\trequest := &sharddistributorv1.PingRequest{\n\t\tShardKey:  shardKey,\n\t\tNamespace: namespace,\n\t}\n\n\tctx, cancel := context.WithTimeout(ctx, pingTimeout)\n\tdefer cancel()\n\n\tresponse, err := canaryClient.Ping(ctx, request, yarpc.WithShardKey(shardKey), yarpc.WithHeader(spectatorclient.NamespaceHeader, namespace))\n\tif err != nil {\n\t\tlogger.Error(\"Failed to ping shard\", zap.String(\"namespace\", namespace), zap.String(\"shard_key\", shardKey), zap.Error(err))\n\t\treturn\n\t}\n\n\t// Verify response\n\tif !response.GetOwnsShard() {\n\t\tlogger.Warn(\"Executor does not own shard\", zap.String(\"namespace\", namespace), zap.String(\"shard_key\", shardKey), zap.String(\"executor_id\", response.GetExecutorId()))\n\t\treturn\n\t}\n\n\tlogger.Info(\"Successfully pinged shard owner\", zap.String(\"namespace\", namespace), zap.String(\"shard_key\", shardKey), zap.String(\"executor_id\", response.GetExecutorId()))\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/pinger/pinger.go",
    "content": "package pinger\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination canary_client_mock.go github.com/uber/cadence/.gen/proto/sharddistributor/v1 ShardDistributorExecutorCanaryAPIYARPCClient\n\nconst (\n\tpingInterval    = 1 * time.Second\n\tpingJitterCoeff = 0.1 // 10% jitter\n)\n\n// Pinger periodically pings shard owners in the fixed namespace\ntype Pinger struct {\n\tlogger       *zap.Logger\n\ttimeSource   clock.TimeSource\n\tcanaryClient sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient\n\tnamespace    string\n\tnumShards    int\n\tctx          context.Context\n\tcancel       context.CancelFunc\n\twg           sync.WaitGroup\n}\n\n// Params are the parameters for creating a Pinger\ntype Params struct {\n\tfx.In\n\n\tLogger       *zap.Logger\n\tTimeSource   clock.TimeSource\n\tCanaryClient sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient\n}\n\n// NewPinger creates a new Pinger for the fixed namespace\nfunc NewPinger(params Params, namespace string, numShards int) *Pinger {\n\treturn &Pinger{\n\t\tlogger:       params.Logger,\n\t\ttimeSource:   params.TimeSource,\n\t\tcanaryClient: params.CanaryClient,\n\t\tnamespace:    namespace,\n\t\tnumShards:    numShards,\n\t}\n}\n\n// Start begins the periodic ping loop\nfunc (p *Pinger) Start(ctx context.Context) {\n\tp.logger.Info(\"Starting canary pinger\", zap.String(\"namespace\", p.namespace), zap.Int(\"num_shards\", p.numShards))\n\tp.ctx, p.cancel = context.WithCancel(context.WithoutCancel(ctx))\n\tp.wg.Add(1)\n\tgo p.pingLoop()\n}\n\n// Stop stops the ping loop\nfunc (p *Pinger) Stop() {\n\tif p.cancel != nil {\n\t\tp.cancel()\n\t}\n\tp.wg.Wait()\n}\n\nfunc (p *Pinger) pingLoop() {\n\tdefer p.wg.Done()\n\n\tticker := p.timeSource.NewTicker(backoff.JitDuration(pingInterval, pingJitterCoeff))\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.ctx.Done():\n\t\t\tp.logger.Info(\"Pinger context done, stopping\")\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tp.pingRandomShard()\n\t\t\tticker.Reset(backoff.JitDuration(pingInterval, pingJitterCoeff))\n\t\t}\n\t}\n}\n\n// Pings a random shard in the namespace and logs the results\nfunc (p *Pinger) pingRandomShard() {\n\tshardNum := rand.Intn(p.numShards)\n\tshardKey := fmt.Sprintf(\"%d\", shardNum)\n\n\tPingShard(p.ctx, p.canaryClient, p.logger, p.namespace, shardKey)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/pinger/pinger_test.go",
    "content": "package pinger\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zaptest/observer\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nfunc TestPingerStartStop(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockClient := NewMockShardDistributorExecutorCanaryAPIYARPCClient(ctrl)\n\n\tpinger := NewPinger(Params{\n\t\tLogger:       zap.NewNop(),\n\t\tTimeSource:   clock.NewRealTimeSource(),\n\t\tCanaryClient: mockClient,\n\t}, \"test-ns\", 10)\n\n\tpinger.Start(context.Background())\n\tpinger.Stop()\n}\n\nfunc TestPingerPingRandomShard(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tcases := []struct {\n\t\tname            string\n\t\tsetupClientMock func(*MockShardDistributorExecutorCanaryAPIYARPCClient)\n\t\texpectedLog     string\n\t}{\n\t\t{\n\t\t\tname: \"owns shard\",\n\t\t\tsetupClientMock: func(mockClient *MockShardDistributorExecutorCanaryAPIYARPCClient) {\n\t\t\t\tmockClient.EXPECT().Ping(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&sharddistributorv1.PingResponse{\n\t\t\t\t\t\tOwnsShard:  true,\n\t\t\t\t\t\tExecutorId: \"127.0.0.1:7953\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedLog: \"Successfully pinged shard owner\",\n\t\t},\n\t\t{\n\t\t\tname: \"does not own shard\",\n\t\t\tsetupClientMock: func(mockClient *MockShardDistributorExecutorCanaryAPIYARPCClient) {\n\t\t\t\tmockClient.EXPECT().\n\t\t\t\t\tPing(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&sharddistributorv1.PingResponse{\n\t\t\t\t\t\tOwnsShard:  false,\n\t\t\t\t\t\tExecutorId: \"127.0.0.1:7953\",\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedLog: \"Executor does not own shard\",\n\t\t},\n\t\t{\n\t\t\tname: \"RPC error\",\n\t\t\tsetupClientMock: func(mockClient *MockShardDistributorExecutorCanaryAPIYARPCClient) {\n\t\t\t\tmockClient.EXPECT().\n\t\t\t\t\tPing(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"network error\"))\n\t\t\t},\n\t\t\texpectedLog: \"Failed to ping shard\",\n\t\t},\n\t}\n\n\tfor _, tt := range cases {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockClient := NewMockShardDistributorExecutorCanaryAPIYARPCClient(ctrl)\n\t\t\tzapCore, logs := observer.New(zap.InfoLevel)\n\t\t\tlogger := zap.New(zapCore)\n\n\t\t\tpinger := NewPinger(Params{\n\t\t\t\tLogger:       logger,\n\t\t\t\tTimeSource:   clock.NewRealTimeSource(),\n\t\t\t\tCanaryClient: mockClient,\n\t\t\t}, \"test-ns\", 10)\n\t\t\tpinger.ctx = context.Background()\n\n\t\t\ttt.setupClientMock(mockClient)\n\n\t\t\tpinger.pingRandomShard()\n\n\t\t\tassert.Equal(t, 1, logs.FilterMessage(tt.expectedLog).Len())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processor/shardprocessor.go",
    "content": "package processor\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// This is a small shard processor, the only thing it currently does it\n// count the number of steps it has processed and log that information.\nconst (\n\tprocessInterval = 10 * time.Second\n)\n\n// NewShardProcessor creates a new ShardProcessor.\nfunc NewShardProcessor(shardID string, timeSource clock.TimeSource, logger *zap.Logger) *ShardProcessor {\n\tp := &ShardProcessor{\n\t\tshardID:    shardID,\n\t\tshardLoad:  shardLoadFromID(shardID),\n\t\ttimeSource: timeSource,\n\t\tlogger:     logger,\n\t\tstopChan:   make(chan struct{}),\n\t}\n\tp.status.Store(int32(types.ShardStatusREADY))\n\treturn p\n}\n\n// ShardProcessor is a processor for a shard.\ntype ShardProcessor struct {\n\tshardID      string\n\tshardLoad    float64\n\ttimeSource   clock.TimeSource\n\tlogger       *zap.Logger\n\tstopChan     chan struct{}\n\tgoRoutineWg  sync.WaitGroup\n\tprocessSteps int\n\n\tstatus atomic.Int32\n}\n\nvar _ executorclient.ShardProcessor = (*ShardProcessor)(nil)\n\n// GetShardReport implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) GetShardReport() executorclient.ShardReport {\n\treturn executorclient.ShardReport{\n\t\tShardLoad: p.shardLoad,                        // We return a load from shardID\n\t\tStatus:    types.ShardStatus(p.status.Load()), // Report the shard as ready since it's actively processing\n\t}\n}\n\n// Start implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) Start(_ context.Context) error {\n\tp.logger.Info(\"Starting shard processor\", zap.String(\"shardID\", p.shardID))\n\tp.goRoutineWg.Add(1)\n\tgo p.process()\n\treturn nil\n}\n\n// Stop implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) Stop() {\n\tp.logger.Info(\"Stopping shard processor\", zap.String(\"shardID\", p.shardID))\n\tclose(p.stopChan)\n\tp.goRoutineWg.Wait()\n}\n\nfunc (p *ShardProcessor) SetShardStatus(status types.ShardStatus) {\n\tp.status.Store(int32(status))\n}\n\nfunc (p *ShardProcessor) process() {\n\tdefer p.goRoutineWg.Done()\n\n\tticker := p.timeSource.NewTicker(processInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.stopChan:\n\t\t\tp.logger.Info(\"Stopping shard processor\", zap.String(\"shardID\", p.shardID), zap.Int(\"steps\", p.processSteps))\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tp.processSteps++\n\t\t\tp.logger.Info(\"Processing shard\", zap.String(\"shardID\", p.shardID), zap.Int(\"steps\", p.processSteps))\n\t\t}\n\t}\n}\n\n// shardLoadFromID returns a shard load based on the shard ID.\n// If the shard ID is not a valid integer, it returns 1.0.\nfunc shardLoadFromID(shardID string) float64 {\n\tif parsed, err := strconv.Atoi(shardID); err == nil && parsed > 0 {\n\t\treturn float64(parsed)\n\t}\n\treturn 1.0\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processor/shardprocessor_test.go",
    "content": "package processor\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/zap/zaptest\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewShardProcessor(t *testing.T) {\n\tshardID := \"test-shard-123\"\n\ttimeSource := clock.NewRealTimeSource()\n\tlogger := zaptest.NewLogger(t)\n\n\tprocessor := NewShardProcessor(shardID, timeSource, logger)\n\n\trequire.NotNil(t, processor)\n\tassert.Equal(t, shardID, processor.shardID)\n\tassert.Equal(t, timeSource, processor.timeSource)\n\tassert.Equal(t, logger, processor.logger)\n\tassert.NotNil(t, processor.stopChan)\n}\n\nfunc TestShardProcessor_GetShardReport(t *testing.T) {\n\tprocessor := NewShardProcessor(\"test-shard\", clock.NewRealTimeSource(), zaptest.NewLogger(t))\n\n\treport := processor.GetShardReport()\n\t// the simple implementation just returns 1.0 for load and READY status\n\tassert.Equal(t, 1.0, report.ShardLoad)\n\tassert.Equal(t, types.ShardStatusREADY, report.Status)\n}\n\nfunc TestShardProcessor_Start_Process_Stop(t *testing.T) {\n\t// Verify that after stopping the processor, there are no goroutines left\n\tgoleak.VerifyNone(t)\n\n\tlogger := zaptest.NewLogger(t)\n\tclock := clock.NewMockedTimeSource()\n\tprocessor := NewShardProcessor(\"test-shard\", clock, logger)\n\n\tctx := context.Background()\n\tprocessor.Start(ctx)\n\n\t// The processor will block on time\n\tclock.BlockUntil(1)\n\n\t// Let time pass until the ticker is triggered\n\tclock.Advance(processInterval + 1*time.Second)\n\n\t// Sleep for a bit to schedule the go-routine\n\ttime.Sleep(10 * time.Millisecond)\n\n\t// Stop the processor\n\tprocessor.Stop()\n\n\t// Assert that the processor has processed at least once\n\tassert.Greater(t, processor.processSteps, 0)\n}\n\nfunc Test_shardLoadFromID(t *testing.T) {\n\ttests := []struct {\n\t\tshardID string\n\t\twant    float64\n\t}{\n\t\t{\"0\", 1.0},\n\t\t{\"shard-1\", 1.0},\n\t\t{\"42\", 42.0},\n\t\t{\"999\", 999.0},\n\t\t{\"\", 1.0},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.shardID, func(t *testing.T) {\n\t\t\tgot := shardLoadFromID(tt.shardID)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processorephemeral/canary_client_mock_test.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/.gen/proto/sharddistributor/v1 (interfaces: ShardDistributorExecutorCanaryAPIYARPCClient)\n//\n// Generated by this command:\n//\n//\tmockgen -package processorephemeral -destination canary_client_mock_test.go github.com/uber/cadence/.gen/proto/sharddistributor/v1 ShardDistributorExecutorCanaryAPIYARPCClient\n//\n\n// Package processorephemeral is a generated GoMock package.\npackage processorephemeral\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n)\n\n// MockShardDistributorExecutorCanaryAPIYARPCClient is a mock of ShardDistributorExecutorCanaryAPIYARPCClient interface.\ntype MockShardDistributorExecutorCanaryAPIYARPCClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder is the mock recorder for MockShardDistributorExecutorCanaryAPIYARPCClient.\ntype MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder struct {\n\tmock *MockShardDistributorExecutorCanaryAPIYARPCClient\n}\n\n// NewMockShardDistributorExecutorCanaryAPIYARPCClient creates a new mock instance.\nfunc NewMockShardDistributorExecutorCanaryAPIYARPCClient(ctrl *gomock.Controller) *MockShardDistributorExecutorCanaryAPIYARPCClient {\n\tmock := &MockShardDistributorExecutorCanaryAPIYARPCClient{ctrl: ctrl}\n\tmock.recorder = &MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardDistributorExecutorCanaryAPIYARPCClient) EXPECT() *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder {\n\treturn m.recorder\n}\n\n// Ping mocks base method.\nfunc (m *MockShardDistributorExecutorCanaryAPIYARPCClient) Ping(arg0 context.Context, arg1 *sharddistributorv1.PingRequest, arg2 ...yarpc.CallOption) (*sharddistributorv1.PingResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Ping\", varargs...)\n\tret0, _ := ret[0].(*sharddistributorv1.PingResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Ping indicates an expected call of Ping.\nfunc (mr *MockShardDistributorExecutorCanaryAPIYARPCClientMockRecorder) Ping(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Ping\", reflect.TypeOf((*MockShardDistributorExecutorCanaryAPIYARPCClient)(nil).Ping), varargs...)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processorephemeral/shardcreator.go",
    "content": "package processorephemeral\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/zap\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/service/sharddistributor/canary/pinger\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -destination canary_client_mock_test.go github.com/uber/cadence/.gen/proto/sharddistributor/v1 ShardDistributorExecutorCanaryAPIYARPCClient\n\nconst (\n\tshardCreationInterval = 1 * time.Second\n)\n\n// ShardCreator creates shards at regular intervals for ephemeral canary testing\ntype ShardCreator struct {\n\tlogger       *zap.Logger\n\ttimeSource   clock.TimeSource\n\tcanaryClient sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient\n\tnamespaces   []string\n\n\tstopChan    chan struct{}\n\tgoRoutineWg sync.WaitGroup\n}\n\n// ShardCreatorParams contains the dependencies needed to create a ShardCreator\ntype ShardCreatorParams struct {\n\tfx.In\n\n\tLogger       *zap.Logger\n\tTimeSource   clock.TimeSource\n\tCanaryClient sharddistributorv1.ShardDistributorExecutorCanaryAPIYARPCClient\n}\n\n// NewShardCreator creates a new ShardCreator instance with the given parameters and namespace\nfunc NewShardCreator(params ShardCreatorParams, namespaces []string) *ShardCreator {\n\treturn &ShardCreator{\n\t\tlogger:       params.Logger,\n\t\ttimeSource:   params.TimeSource,\n\t\tcanaryClient: params.CanaryClient,\n\t\tstopChan:     make(chan struct{}),\n\t\tgoRoutineWg:  sync.WaitGroup{},\n\t\tnamespaces:   namespaces,\n\t}\n}\n\n// Start begins the shard creation process in a background goroutine\nfunc (s *ShardCreator) Start() {\n\ts.goRoutineWg.Add(1)\n\tgo s.process(context.Background())\n\ts.logger.Info(\"Shard creator started\")\n}\n\n// Stop stops the shard creation process and waits for the goroutine to finish\nfunc (s *ShardCreator) Stop() {\n\tclose(s.stopChan)\n\ts.goRoutineWg.Wait()\n\ts.logger.Info(\"Shard creator stopped\")\n}\n\n// ShardCreatorModule creates an fx module for the shard creator with the given namespace\nfunc ShardCreatorModule(namespace []string) fx.Option {\n\treturn fx.Module(\"shard-creator\",\n\t\tfx.Provide(func(params ShardCreatorParams) *ShardCreator {\n\t\t\treturn NewShardCreator(params, namespace)\n\t\t}),\n\t\tfx.Invoke(func(lifecycle fx.Lifecycle, shardCreator *ShardCreator) {\n\t\t\tlifecycle.Append(fx.StartStopHook(shardCreator.Start, shardCreator.Stop))\n\t\t}),\n\t)\n}\n\nfunc (s *ShardCreator) process(ctx context.Context) {\n\tdefer s.goRoutineWg.Done()\n\n\tticker := s.timeSource.NewTicker(shardCreationInterval)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn\n\t\tcase <-s.stopChan:\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tfor _, namespace := range s.namespaces {\n\t\t\t\tshardKey := uuid.New().String()\n\t\t\t\ts.logger.Info(\"Creating shard\", zap.String(\"shardKey\", shardKey), zap.String(\"namespace\", namespace))\n\n\t\t\t\tpinger.PingShard(ctx, s.canaryClient, s.logger, namespace, shardKey)\n\t\t\t}\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processorephemeral/shardcreator_test.go",
    "content": "package processorephemeral\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap/zaptest\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/clock\"\n)\n\nfunc TestShardCreator_PingsShards(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\n\tlogger := zaptest.NewLogger(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\tctrl := gomock.NewController(t)\n\n\tnamespace := \"test-namespace\"\n\tmockCanaryClient := NewMockShardDistributorExecutorCanaryAPIYARPCClient(ctrl)\n\n\t// Ping happens after successful GetShardOwner\n\tmockCanaryClient.EXPECT().\n\t\tPing(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx interface{}, req *sharddistributorv1.PingRequest, opts ...interface{}) (*sharddistributorv1.PingResponse, error) {\n\t\t\tassert.NotEmpty(t, req.ShardKey)\n\t\t\tassert.Equal(t, namespace, req.Namespace)\n\t\t\treturn &sharddistributorv1.PingResponse{\n\t\t\t\tOwnsShard:  true,\n\t\t\t\tExecutorId: \"executor-1\",\n\t\t\t}, nil\n\t\t})\n\n\tparams := ShardCreatorParams{\n\t\tLogger:       logger,\n\t\tTimeSource:   timeSource,\n\t\tCanaryClient: mockCanaryClient,\n\t}\n\n\tcreator := NewShardCreator(params, []string{namespace})\n\tcreator.Start()\n\n\t// Wait for the goroutine to start and do it's ping\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(shardCreationInterval + 100*time.Millisecond)\n\ttime.Sleep(10 * time.Millisecond)\n\n\tcreator.Stop()\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processorephemeral/shardprocessor.go",
    "content": "package processorephemeral\n\nimport (\n\t\"context\"\n\t\"math/rand\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// This is a small shard processor, the only thing it currently does it\n// count the number of steps it has processed and log that information.\nconst (\n\tprocessInterval = 10 * time.Second\n\n\t// We create a new shard every second. For each of them we have a chance of them to be done of 1/60 every second.\n\t// This means the average time to complete a shard is 60 seconds.\n\t// It also means we in average have 60 shards per instance running at any given time.\n\tstopInterval             = 1 * time.Second\n\tshardProcessorDoneChance = 60\n)\n\n// NewShardProcessor creates a new ShardProcessor.\nfunc NewShardProcessor(shardID string, timeSource clock.TimeSource, logger *zap.Logger) *ShardProcessor {\n\tp := &ShardProcessor{\n\t\tshardID:    shardID,\n\t\ttimeSource: timeSource,\n\t\tlogger:     logger,\n\t\tstopChan:   make(chan struct{}),\n\t}\n\tp.SetShardStatus(types.ShardStatusREADY)\n\treturn p\n}\n\n// ShardProcessor is a processor for a shard.\ntype ShardProcessor struct {\n\tshardID      string\n\ttimeSource   clock.TimeSource\n\tlogger       *zap.Logger\n\tstopChan     chan struct{}\n\tgoRoutineWg  sync.WaitGroup\n\tprocessSteps int\n\n\tstatus atomic.Int32\n}\n\nvar _ executorclient.ShardProcessor = (*ShardProcessor)(nil)\n\n// GetShardReport implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) GetShardReport() executorclient.ShardReport {\n\treturn executorclient.ShardReport{\n\t\tShardLoad: 1.0,                                // We return 1.0 for all shards for now.\n\t\tStatus:    types.ShardStatus(p.status.Load()), // Report the status of the shard\n\t}\n}\n\n// Start implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) Start(_ context.Context) error {\n\tp.logger.Info(\"Starting shard processor\", zap.String(\"shardID\", p.shardID))\n\tp.goRoutineWg.Add(1)\n\tgo p.process()\n\treturn nil\n}\n\n// Stop implements executorclient.ShardProcessor.\nfunc (p *ShardProcessor) Stop() {\n\tclose(p.stopChan)\n\tp.goRoutineWg.Wait()\n}\n\nfunc (p *ShardProcessor) SetShardStatus(status types.ShardStatus) {\n\tp.status.Store(int32(status))\n}\n\nfunc (p *ShardProcessor) process() {\n\tdefer p.goRoutineWg.Done()\n\n\tticker := p.timeSource.NewTicker(processInterval)\n\tdefer ticker.Stop()\n\n\tstopTicker := p.timeSource.NewTicker(stopInterval)\n\tdefer stopTicker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-p.stopChan:\n\t\t\tp.logger.Info(\"Stopping shard processor\", zap.String(\"shardID\", p.shardID), zap.Int(\"steps\", p.processSteps), zap.String(\"status\", types.ShardStatus(p.status.Load()).String()))\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\tp.logger.Info(\"Processing shard\", zap.String(\"shardID\", p.shardID), zap.Int(\"steps\", p.processSteps), zap.String(\"status\", types.ShardStatus(p.status.Load()).String()))\n\t\tcase <-stopTicker.Chan():\n\t\t\tp.processSteps++\n\t\t\tif rand.Intn(shardProcessorDoneChance) == 0 {\n\t\t\t\tp.logger.Info(\"Setting shard processor to done\", zap.String(\"shardID\", p.shardID), zap.Int(\"steps\", p.processSteps), zap.String(\"status\", types.ShardStatus(p.status.Load()).String()))\n\t\t\t\tp.SetShardStatus(types.ShardStatusDONE)\n\t\t\t}\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/processorephemeral/shardprocessor_test.go",
    "content": "package processorephemeral\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/zap/zaptest\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewShardProcessor(t *testing.T) {\n\tshardID := \"test-shard-123\"\n\ttimeSource := clock.NewRealTimeSource()\n\tlogger := zaptest.NewLogger(t)\n\n\tprocessor := NewShardProcessor(shardID, timeSource, logger)\n\n\trequire.NotNil(t, processor)\n\tassert.Equal(t, shardID, processor.shardID)\n\tassert.Equal(t, timeSource, processor.timeSource)\n\tassert.Equal(t, logger, processor.logger)\n\tassert.NotNil(t, processor.stopChan)\n}\n\nfunc TestShardProcessor_GetShardReport(t *testing.T) {\n\tprocessor := NewShardProcessor(\"test-shard\", clock.NewRealTimeSource(), zaptest.NewLogger(t))\n\n\treport := processor.GetShardReport()\n\t// the simple implementation just returns 1.0 for load and READY status\n\tassert.Equal(t, 1.0, report.ShardLoad)\n\tassert.Equal(t, types.ShardStatusREADY, report.Status)\n}\n\nfunc TestShardProcessor_Start_Process_Stop(t *testing.T) {\n\t// Verify that after stopping the processor, there are no goroutines left\n\tgoleak.VerifyNone(t)\n\n\tlogger := zaptest.NewLogger(t)\n\tclock := clock.NewMockedTimeSource()\n\tprocessor := NewShardProcessor(\"test-shard\", clock, logger)\n\n\tctx := context.Background()\n\tprocessor.Start(ctx)\n\n\t// The processor will block on time\n\tclock.BlockUntil(1)\n\n\t// Let time pass until the ticker is triggered\n\tclock.Advance(processInterval + 1*time.Second)\n\n\t// Sleep for a bit to schedule the go-routine\n\ttime.Sleep(10 * time.Millisecond)\n\n\t// Stop the processor\n\tprocessor.Stop()\n\n\t// Assert that the processor has processed at least once\n\tassert.Greater(t, processor.processSteps, 0)\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/sharddistributorclient/shardDistributorClient.go",
    "content": "package sharddistributorclient\n\nimport (\n\t\"go.uber.org/fx\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\ttimeoutwrapper \"github.com/uber/cadence/client/wrappers/timeout\"\n)\n\n// Params contains the dependencies needed to create a shard distributor client\ntype Params struct {\n\tfx.In\n\n\tYarpcClient sharddistributorv1.ShardDistributorAPIYARPCClient\n}\n\n// NewShardDistributorClient creates a new shard distributor client with GRPC and timeout wrappers\nfunc NewShardDistributorClient(p Params) (sharddistributor.Client, error) {\n\tshardDistributorExecutorClient := grpc.NewShardDistributorClient(p.YarpcClient)\n\tshardDistributorExecutorClient = timeoutwrapper.NewShardDistributorClient(shardDistributorExecutorClient, timeoutwrapper.ShardDistributorExecutorDefaultTimeout)\n\treturn shardDistributorExecutorClient, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/canary/sharddistributorexecutorclient/shardDistributorExecutorClient.go",
    "content": "package sharddistributorexecutorclient\n\nimport (\n\t\"go.uber.org/fx\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\ttimeoutwrapper \"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient\"\n)\n\n// Params contains the dependencies needed to create a shard distributor client\ntype Params struct {\n\tfx.In\n\n\tYarpcClient sharddistributorv1.ShardDistributorExecutorAPIYARPCClient\n}\n\n// NewShardDistributorExecutorClient creates a new shard distributor executor client with GRPC and timeout wrappers\nfunc NewShardDistributorExecutorClient(p Params) (executorclient.Client, error) {\n\tshardDistributorExecutorClient := grpc.NewShardDistributorExecutorClient(p.YarpcClient)\n\tshardDistributorExecutorClient = timeoutwrapper.NewShardDistributorExecutorClient(shardDistributorExecutorClient, timeoutwrapper.ShardDistributorExecutorDefaultTimeout)\n\treturn shardDistributorExecutorClient, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/client/clientcommon/config.go",
    "content": "package clientcommon\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n\tsdconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\nconst defaultPeerTTL = 2 * time.Minute\n\n// NamespaceConfig represents configuration for a single namespace\ntype NamespaceConfig struct {\n\tNamespace         string        `yaml:\"namespace\"`\n\tHeartBeatInterval time.Duration `yaml:\"heartbeat_interval\"`\n\tMigrationMode     string        `yaml:\"migration_mode\"`\n\tTTLShard          time.Duration `yaml:\"ttl_shard\"`  // time after which shards are stopped if they are not used\n\tTTLReport         time.Duration `yaml:\"ttl_report\"` // time after which the shard report status (including load) needs to be updated\n}\n\n// GetMigrationMode converts the string migration mode to types.MigrationMode using the shared configMode map\nfunc (nc *NamespaceConfig) GetMigrationMode() types.MigrationMode {\n\tmode := strings.ToLower(strings.TrimSpace(nc.MigrationMode))\n\tif migrationMode, ok := sdconfig.MigrationMode[mode]; ok {\n\t\treturn migrationMode\n\t}\n\t// Default to INVALID if not specified or unrecognized\n\treturn types.MigrationModeINVALID\n}\n\n// Config represents configuration for multiple namespaces\ntype Config struct {\n\tNamespaces []NamespaceConfig `yaml:\"namespaces\"`\n\tPeerTTL    time.Duration     `yaml:\"peer_ttl\"` // TTL for idle peers; default 2m if zero\n}\n\n// GetConfigForNamespace returns the config for a specific namespace\nfunc (c *Config) GetConfigForNamespace(namespace string) (*NamespaceConfig, error) {\n\tfor _, ns := range c.Namespaces {\n\t\tif ns.Namespace == namespace {\n\t\t\treturn &ns, nil\n\t\t}\n\t}\n\treturn nil, fmt.Errorf(\"namespace %s not found in config\", namespace)\n}\n\n// GetSingleConfig returns the config if there's exactly one namespace, otherwise returns an error\nfunc (c *Config) GetSingleConfig() (*NamespaceConfig, error) {\n\tif len(c.Namespaces) == 0 {\n\t\treturn nil, fmt.Errorf(\"no namespaces configured\")\n\t}\n\tif len(c.Namespaces) > 1 {\n\t\treturn nil, fmt.Errorf(\"multiple namespaces configured (%d), must specify which namespace to use\", len(c.Namespaces))\n\t}\n\treturn &c.Namespaces[0], nil\n}\n\n// GetPeerTTL returns the configured peer TTL, or the default 2m if not set.\nfunc (c *Config) GetPeerTTL() time.Duration {\n\tif c.PeerTTL <= 0 {\n\t\treturn defaultPeerTTL\n\t}\n\treturn c.PeerTTL\n}\n\n// Validate validates the configuration\nfunc (c *Config) Validate() error {\n\tif len(c.Namespaces) == 0 {\n\t\treturn fmt.Errorf(\"at least one namespace must be configured\")\n\t}\n\n\tseenNamespaces := make(map[string]bool)\n\tfor i, ns := range c.Namespaces {\n\t\tif ns.Namespace == \"\" {\n\t\t\treturn fmt.Errorf(\"namespace %d: namespace name cannot be empty\", i)\n\t\t}\n\t\tif ns.HeartBeatInterval <= 0 {\n\t\t\treturn fmt.Errorf(\"namespace %d (%s): heartbeat_interval must be greater than 0\", i, ns.Namespace)\n\t\t}\n\t\tif seenNamespaces[ns.Namespace] {\n\t\t\treturn fmt.Errorf(\"duplicate namespace: %s\", ns.Namespace)\n\t\t}\n\t\tseenNamespaces[ns.Namespace] = true\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/sharddistributor/client/clientcommon/config_test.go",
    "content": "package clientcommon\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNamespaceConfig_GetMigrationMode(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tmigrationMode string\n\t\texpected      types.MigrationMode\n\t}{\n\t\t{\n\t\t\tname:          \"local_passthrough\",\n\t\t\tmigrationMode: \"local_pass\",\n\t\t\texpected:      types.MigrationModeLOCALPASSTHROUGH,\n\t\t},\n\t\t{\n\t\t\tname:          \"local_passthrough_shadow\",\n\t\t\tmigrationMode: \"local_pass_shadow\",\n\t\t\texpected:      types.MigrationModeLOCALPASSTHROUGHSHADOW,\n\t\t},\n\t\t{\n\t\t\tname:          \"distributed_passthrough\",\n\t\t\tmigrationMode: \"distributed_pass\",\n\t\t\texpected:      types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t\t},\n\t\t{\n\t\t\tname:          \"onboarded\",\n\t\t\tmigrationMode: \"onboarded\",\n\t\t\texpected:      types.MigrationModeONBOARDED,\n\t\t},\n\t\t{\n\t\t\tname:          \"invalid\",\n\t\t\tmigrationMode: \"invalid\",\n\t\t\texpected:      types.MigrationModeINVALID,\n\t\t},\n\t\t{\n\t\t\tname:          \"empty string\",\n\t\t\tmigrationMode: \"\",\n\t\t\texpected:      types.MigrationModeINVALID,\n\t\t},\n\t\t{\n\t\t\tname:          \"unknown mode\",\n\t\t\tmigrationMode: \"unknown_mode\",\n\t\t\texpected:      types.MigrationModeINVALID,\n\t\t},\n\t\t{\n\t\t\tname:          \"case insensitive - uppercase\",\n\t\t\tmigrationMode: \"ONBOARDED\",\n\t\t\texpected:      types.MigrationModeONBOARDED,\n\t\t},\n\t\t{\n\t\t\tname:          \"case insensitive - mixed case\",\n\t\t\tmigrationMode: \"Local_Pass\",\n\t\t\texpected:      types.MigrationModeLOCALPASSTHROUGH,\n\t\t},\n\t\t{\n\t\t\tname:          \"whitespace trimming\",\n\t\t\tmigrationMode: \"  onboarded  \",\n\t\t\texpected:      types.MigrationModeONBOARDED,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tconfig := &NamespaceConfig{\n\t\t\t\tMigrationMode: tt.migrationMode,\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expected, config.GetMigrationMode())\n\t\t})\n\t}\n}\n\nfunc TestConfig_GetPeerTTL(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tpeerTTL  time.Duration\n\t\texpected time.Duration\n\t}{\n\t\t{\n\t\t\tname:     \"zero uses default\",\n\t\t\tpeerTTL:  0,\n\t\t\texpected: 2 * time.Minute,\n\t\t},\n\t\t{\n\t\t\tname:     \"non-zero returned as-is\",\n\t\t\tpeerTTL:  5 * time.Minute,\n\t\t\texpected: 5 * time.Minute,\n\t\t},\n\t\t{\n\t\t\tname:     \"negative uses default\",\n\t\t\tpeerTTL:  -1 * time.Second,\n\t\t\texpected: 2 * time.Minute,\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tcfg := Config{PeerTTL: tc.peerTTL}\n\t\t\tassert.Equal(t, tc.expected, cfg.GetPeerTTL())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/client/clientcommon/constants.go",
    "content": "package clientcommon\n\nconst (\n\tGrpcAddressMetadataKey = \"grpc_address\"\n)\n"
  },
  {
    "path": "service/sharddistributor/client/clientcommon/drain_observer.go",
    "content": "package clientcommon\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination drain_observer_mock.go . DrainSignalObserver\n\n// DrainSignalObserver observes infrastructure drain signals.\n// Drain is reversible: if the instance reappears in discovery,\n// Undrain() fires, allowing the consumer to resume operations.\n//\n// Implementations use close-to-broadcast semantics: the returned channel is\n// closed when the event occurs, so all goroutines selecting on it wake up.\n// After each close, a fresh channel is created for the next cycle.\ntype DrainSignalObserver interface {\n\t// Drain returns a channel closed when the instance is\n\t// removed from service discovery.\n\tDrain() <-chan struct{}\n\n\t// Undrain returns a channel closed when the instance is\n\t// added back to service discovery after a drain.\n\tUndrain() <-chan struct{}\n}\n"
  },
  {
    "path": "service/sharddistributor/client/clientcommon/drain_observer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: drain_observer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package clientcommon -source drain_observer.go -destination drain_observer_mock.go . DrainSignalObserver\n//\n\n// Package clientcommon is a generated GoMock package.\npackage clientcommon\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockDrainSignalObserver is a mock of DrainSignalObserver interface.\ntype MockDrainSignalObserver struct {\n\tctrl     *gomock.Controller\n\trecorder *MockDrainSignalObserverMockRecorder\n\tisgomock struct{}\n}\n\n// MockDrainSignalObserverMockRecorder is the mock recorder for MockDrainSignalObserver.\ntype MockDrainSignalObserverMockRecorder struct {\n\tmock *MockDrainSignalObserver\n}\n\n// NewMockDrainSignalObserver creates a new mock instance.\nfunc NewMockDrainSignalObserver(ctrl *gomock.Controller) *MockDrainSignalObserver {\n\tmock := &MockDrainSignalObserver{ctrl: ctrl}\n\tmock.recorder = &MockDrainSignalObserverMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockDrainSignalObserver) EXPECT() *MockDrainSignalObserverMockRecorder {\n\treturn m.recorder\n}\n\n// Drain mocks base method.\nfunc (m *MockDrainSignalObserver) Drain() <-chan struct{} {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Drain\")\n\tret0, _ := ret[0].(<-chan struct{})\n\treturn ret0\n}\n\n// Drain indicates an expected call of Drain.\nfunc (mr *MockDrainSignalObserverMockRecorder) Drain() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Drain\", reflect.TypeOf((*MockDrainSignalObserver)(nil).Drain))\n}\n\n// Undrain mocks base method.\nfunc (m *MockDrainSignalObserver) Undrain() <-chan struct{} {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Undrain\")\n\tret0, _ := ret[0].(<-chan struct{})\n\treturn ret0\n}\n\n// Undrain indicates an expected call of Undrain.\nfunc (mr *MockDrainSignalObserverMockRecorder) Undrain() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Undrain\", reflect.TypeOf((*MockDrainSignalObserver)(nil).Undrain))\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/client.go",
    "content": "package executorclient\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"os\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc\"\n\n\ttimeoutwrapper \"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient/metricsconstants\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go . ShardProcessorFactory,ShardProcessor,Executor,Client\n\n// ErrShardProcessNotFound is returned by GetShardProcess when this host is not\n// assigned the requested shard. Callers that interpret shard ownership should\n// treat this as an ownership-loss signal rather than an internal error.\nvar ErrShardProcessNotFound = errors.New(\"shard process not found\")\n\ntype Client interface {\n\tHeartbeat(context.Context, *types.ExecutorHeartbeatRequest, ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error)\n}\n\ntype ExecutorMetadata map[string]string\n\ntype ShardReport struct {\n\tShardLoad float64\n\tStatus    types.ShardStatus\n}\n\ntype ShardProcessor interface {\n\tStart(ctx context.Context) error\n\tStop()\n\tGetShardReport() ShardReport\n\tSetShardStatus(types.ShardStatus)\n}\n\ntype ShardProcessorFactory[SP ShardProcessor] interface {\n\tNewShardProcessor(shardID string) (SP, error)\n}\n\ntype Executor[SP ShardProcessor] interface {\n\tStart(ctx context.Context)\n\tStop()\n\n\tGetShardProcess(ctx context.Context, shardID string) (SP, error)\n\n\t// Get the namespace this executor is responsible for\n\tGetNamespace() string\n\n\t// Set metadata for the executor\n\tSetMetadata(metadata map[string]string)\n\t// Get the current metadata of the executor\n\tGetMetadata() map[string]string\n\n\t// AssignShardsFromLocalLogic is used for the migration during local-passthrough, local-passthrough-shadow, distributed-passthrough\n\tAssignShardsFromLocalLogic(ctx context.Context, shardAssignment map[string]*types.ShardAssignment) error\n\t// RemoveShardsFromLocalLogic is used for the migration during local-passthrough, local-passthrough-shadow, distributed-passthrough\n\tRemoveShardsFromLocalLogic(shardIDs []string) error\n\n\t// IsOnboardedToSD is returning true if the executor relies on SD for distribution\n\tIsOnboardedToSD() bool\n}\n\ntype Params[SP ShardProcessor] struct {\n\tfx.In\n\n\tExecutorClient        Client\n\tMetricsScope          tally.Scope\n\tLogger                log.Logger\n\tShardProcessorFactory ShardProcessorFactory[SP]\n\tConfig                clientcommon.Config\n\tTimeSource            clock.TimeSource\n\tMetadata              ExecutorMetadata                 `optional:\"true\"`\n\tDrainObserver         clientcommon.DrainSignalObserver `optional:\"true\"`\n}\n\n// NewExecutorWithNamespace creates an executor for a specific namespace\nfunc NewExecutorWithNamespace[SP ShardProcessor](params Params[SP], namespace string) (Executor[SP], error) {\n\t// Validate the config first\n\tif err := params.Config.Validate(); err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid config: %w\", err)\n\t}\n\n\t// Get config for the specified namespace\n\tnamespaceConfig, err := params.Config.GetConfigForNamespace(namespace)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get config for namespace %s: %w\", namespace, err)\n\t}\n\n\treturn newExecutorWithConfig(params, namespaceConfig)\n}\n\n// NewExecutor creates an executor using auto-selection (single namespace only)\nfunc NewExecutor[SP ShardProcessor](params Params[SP]) (Executor[SP], error) {\n\t// Validate the config first\n\tif err := params.Config.Validate(); err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid config: %w\", err)\n\t}\n\n\t// Auto-select if there's only one namespace\n\tnamespaceConfig, err := params.Config.GetSingleConfig()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"auto-select namespace: %w\", err)\n\t}\n\n\treturn newExecutorWithConfig(params, namespaceConfig)\n}\n\nfunc newExecutorWithConfig[SP ShardProcessor](params Params[SP], namespaceConfig *clientcommon.NamespaceConfig) (Executor[SP], error) {\n\tshardDistributorClient, err := createShardDistributorExecutorClient(params.ExecutorClient, params.MetricsScope, params.Logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"create shard distributor executor client: %w\", err)\n\t}\n\n\t// TODO: get executor ID from environment\n\texecutorID := uuid.New().String()\n\n\thostname, err := os.Hostname()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get hostname: %w\", err)\n\t}\n\n\tmetricsScope := params.MetricsScope.Tagged(map[string]string{\n\t\tmetrics.OperationTagName: metricsconstants.ShardDistributorExecutorOperationTagName,\n\t\t\"namespace\":              namespaceConfig.Namespace,\n\t})\n\n\thostMetricsScope := metricsScope.Tagged(map[string]string{\n\t\t\"host\": hostname,\n\t})\n\n\texecutor := &executorImpl[SP]{\n\t\tlogger:                 params.Logger,\n\t\tshardDistributorClient: shardDistributorClient,\n\t\tshardProcessorFactory:  params.ShardProcessorFactory,\n\t\theartBeatInterval:      namespaceConfig.HeartBeatInterval,\n\t\tttlShard:               namespaceConfig.TTLShard,\n\t\tnamespace:              namespaceConfig.Namespace,\n\t\texecutorID:             executorID,\n\t\ttimeSource:             params.TimeSource,\n\t\tstopC:                  make(chan struct{}),\n\t\tmetrics:                metricsScope,\n\t\thostMetrics:            hostMetricsScope,\n\t\tmetadata: syncExecutorMetadata{\n\t\t\tdata: params.Metadata,\n\t\t},\n\t\tdrainObserver: params.DrainObserver,\n\t}\n\texecutor.setMigrationMode(namespaceConfig.GetMigrationMode())\n\n\treturn executor, nil\n}\n\nfunc createShardDistributorExecutorClient(client Client, metricsScope tally.Scope, logger log.Logger) (Client, error) {\n\n\tshardDistributorExecutorClient := timeoutwrapper.NewShardDistributorExecutorClient(client, timeoutwrapper.ShardDistributorExecutorDefaultTimeout)\n\n\tif metricsScope != nil {\n\t\tshardDistributorExecutorClient = NewMeteredShardDistributorExecutorClient(shardDistributorExecutorClient, metricsScope)\n\t}\n\n\treturn shardDistributorExecutorClient, nil\n}\n\nfunc Module[SP ShardProcessor]() fx.Option {\n\treturn fx.Module(\"shard-distributor-executor-client\",\n\t\tfx.Provide(NewExecutor[SP]),\n\t\tfx.Invoke(func(executor Executor[SP], lc fx.Lifecycle) {\n\t\t\tlc.Append(fx.StartStopHook(executor.Start, executor.Stop))\n\t\t}),\n\t)\n}\n\n// ModuleWithNamespace creates an executor module for a specific namespace\nfunc ModuleWithNamespace[SP ShardProcessor](namespace string) fx.Option {\n\treturn fx.Module(fmt.Sprintf(\"shard-distributor-executor-client-%s\", namespace),\n\t\tfx.Provide(func(params Params[SP]) (Executor[SP], error) {\n\t\t\treturn NewExecutorWithNamespace(params, namespace)\n\t\t}),\n\t\tfx.Invoke(func(executor Executor[SP], lc fx.Lifecycle) {\n\t\t\tlc.Append(fx.StartStopHook(executor.Start, executor.Stop))\n\t\t}),\n\t)\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/client_test.go",
    "content": "package executorclient\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n)\n\nfunc TestModule(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockLogger := log.NewNoop()\n\n\tmockShardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\tshardDistributorExecutorClient := NewMockClient(ctrl)\n\tshardDistributorExecutorClient.EXPECT().\n\t\tHeartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(&types.ExecutorHeartbeatResponse{}, nil).\n\t\tAnyTimes()\n\t// Example config\n\tconfig := clientcommon.Config{\n\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t{\n\t\t\t\tNamespace:         \"test-namespace\",\n\t\t\t\tHeartBeatInterval: 5 * time.Second,\n\t\t\t},\n\t\t},\n\t}\n\n\t// Create a test app with the library, check that it starts and stops\n\tfxtest.New(t,\n\t\tfx.Provide(func() Client {\n\t\t\treturn shardDistributorExecutorClient\n\t\t}),\n\t\tfx.Supply(\n\t\t\tfx.Annotate(tally.NoopScope, fx.As(new(tally.Scope))),\n\t\t\tfx.Annotate(mockLogger, fx.As(new(log.Logger))),\n\t\t\tfx.Annotate(mockShardProcessorFactory, fx.As(new(ShardProcessorFactory[*MockShardProcessor]))),\n\t\t\tfx.Annotate(clock.NewMockedTimeSource(), fx.As(new(clock.TimeSource))),\n\t\t\tconfig,\n\t\t),\n\t\tModule[*MockShardProcessor](),\n\t).RequireStart().RequireStop()\n}\n\n// Create distinct mock processor types for testing multiple namespaces\ntype MockShardProcessor1 struct {\n\t*MockShardProcessor\n}\n\ntype MockShardProcessor2 struct {\n\t*MockShardProcessor\n}\n\nfunc TestModuleWithNamespace(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockLogger := log.NewNoop()\n\n\tshardDistributorExecutorClient := NewMockClient(ctrl)\n\tshardDistributorExecutorClient.EXPECT().\n\t\tHeartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(&types.ExecutorHeartbeatResponse{}, nil).\n\t\tAnyTimes()\n\n\tmockFactory1 := NewMockShardProcessorFactory[*MockShardProcessor1](ctrl)\n\tmockFactory2 := NewMockShardProcessorFactory[*MockShardProcessor2](ctrl)\n\n\t// Multi-namespace config\n\tconfig := clientcommon.Config{\n\t\tNamespaces: []clientcommon.NamespaceConfig{\n\t\t\t{\n\t\t\t\tNamespace:         \"namespace1\",\n\t\t\t\tHeartBeatInterval: 5 * time.Second,\n\t\t\t},\n\t\t\t{\n\t\t\t\tNamespace:         \"namespace2\",\n\t\t\t\tHeartBeatInterval: 10 * time.Second,\n\t\t\t},\n\t\t},\n\t}\n\n\t// Create a test app with two namespace-specific modules using different processor types\n\tfxtest.New(t,\n\t\tfx.Provide(func() Client {\n\t\t\treturn shardDistributorExecutorClient\n\t\t}),\n\t\tfx.Supply(\n\t\t\tfx.Annotate(tally.NoopScope, fx.As(new(tally.Scope))),\n\t\t\tfx.Annotate(mockLogger, fx.As(new(log.Logger))),\n\t\t\tfx.Annotate(clock.NewMockedTimeSource(), fx.As(new(clock.TimeSource))),\n\t\t\tfx.Annotate(mockFactory1, fx.As(new(ShardProcessorFactory[*MockShardProcessor1]))),\n\t\t\tfx.Annotate(mockFactory2, fx.As(new(ShardProcessorFactory[*MockShardProcessor2]))),\n\t\t\tconfig,\n\t\t),\n\t\t// Two namespace-specific modules with different processor types\n\t\tModuleWithNamespace[*MockShardProcessor1](\"namespace1\"),\n\t\tModuleWithNamespace[*MockShardProcessor2](\"namespace2\"),\n\t).RequireStart().RequireStop()\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/clientimpl.go",
    "content": "package executorclient\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient/metricsconstants\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient/syncgeneric\"\n)\n\nvar (\n\t// ErrLocalPassthroughMode indicates that the heartbeat loop should stop due to local passthrough mode\n\tErrLocalPassthroughMode = errors.New(\"local passthrough mode: stopping heartbeat loop\")\n\t// ErrAssignmentDivergenceLocalShard indicates that the local shard is not reported back from the heartbeat\n\tErrAssignmentDivergenceLocalShard = errors.New(\"assignment divergence: local shard not in heartbeat or not ready\")\n\t// ErrAssignmentDivergenceHeartbeatShard indicates that the shard in the heartbeat is not present in the local assignment\n\tErrAssignmentDivergenceHeartbeatShard = errors.New(\"assignment divergence: heartbeat shard not in local\")\n)\n\ntype processorState int32\n\nconst (\n\tprocessorStateStarting processorState = iota\n\tprocessorStateStarted\n\tprocessorStateStopping\n)\n\nconst (\n\theartbeatJitterCoeff     = 0.1 // 10% jitter\n\tdrainingHeartbeatTimeout = 5 * time.Second\n)\n\ntype managedProcessor[SP ShardProcessor] struct {\n\tprocessor SP\n\tstate     atomic.Int32\n}\n\ntype syncExecutorMetadata struct {\n\tsync.RWMutex\n\n\tdata map[string]string\n}\n\nfunc (m *syncExecutorMetadata) Set(metadata map[string]string) {\n\tm.Lock()\n\tdefer m.Unlock()\n\n\tm.data = metadata\n}\n\nfunc (m *syncExecutorMetadata) Get() map[string]string {\n\tm.RLock()\n\tdefer m.RUnlock()\n\n\t// Copy the map\n\tresult := make(map[string]string, len(m.data))\n\tfor k, v := range m.data {\n\t\tresult[k] = v\n\t}\n\n\treturn result\n}\n\nfunc (mp *managedProcessor[SP]) setState(state processorState) {\n\tmp.state.Store(int32(state))\n}\n\nfunc (mp *managedProcessor[SP]) getState() processorState {\n\treturn processorState(mp.state.Load())\n}\n\nfunc newManagedProcessor[SP ShardProcessor](processor SP, state processorState) *managedProcessor[SP] {\n\tmanaged := &managedProcessor[SP]{\n\t\tprocessor: processor,\n\t\tstate:     atomic.Int32{},\n\t}\n\n\tmanaged.setState(state)\n\treturn managed\n}\n\ntype executorImpl[SP ShardProcessor] struct {\n\tlogger                 log.Logger\n\tshardDistributorClient sharddistributorexecutor.Client\n\tshardProcessorFactory  ShardProcessorFactory[SP]\n\tnamespace              string\n\tstopC                  chan struct{}\n\theartBeatInterval      time.Duration\n\tttlShard               time.Duration\n\tmanagedProcessors      syncgeneric.Map[string, *managedProcessor[SP]]\n\tprocessorsToLastUse    syncgeneric.Map[string, time.Time]\n\texecutorID             string\n\ttimeSource             clock.TimeSource\n\tprocessLoopWG          sync.WaitGroup\n\tassignmentMutex        sync.Mutex\n\tmetrics                tally.Scope\n\thostMetrics            tally.Scope\n\tmigrationMode          atomic.Int32\n\tmetadata               syncExecutorMetadata\n\tdrainObserver          clientcommon.DrainSignalObserver\n}\n\nfunc (e *executorImpl[SP]) setMigrationMode(mode types.MigrationMode) {\n\te.migrationMode.Store(int32(mode))\n}\n\nfunc (e *executorImpl[SP]) getMigrationMode() types.MigrationMode {\n\treturn types.MigrationMode(e.migrationMode.Load())\n}\n\nfunc (e *executorImpl[SP]) Start(ctx context.Context) {\n\te.logger.Info(\"starting shard distributor executor\", tag.ShardNamespace(e.namespace))\n\te.processLoopWG.Add(2)\n\tgo func() {\n\t\tdefer e.processLoopWG.Done()\n\t\te.heartbeatloop(context.WithoutCancel(ctx))\n\t}()\n\tgo func() {\n\t\tdefer e.processLoopWG.Done()\n\t\te.shardCleanUpLoop(context.WithoutCancel(ctx))\n\t}()\n}\n\nfunc (e *executorImpl[SP]) Stop() {\n\te.logger.Info(\"stopping shard distributor executor\", tag.ShardNamespace(e.namespace))\n\tclose(e.stopC)\n\te.processLoopWG.Wait()\n}\n\nfunc (e *executorImpl[SP]) GetShardProcess(ctx context.Context, shardID string) (SP, error) {\n\te.processorsToLastUse.Store(shardID, e.timeSource.Now())\n\n\tshardProcess, ok := e.managedProcessors.Load(shardID)\n\tif !ok {\n\t\tif e.getMigrationMode() == types.MigrationModeLOCALPASSTHROUGH {\n\t\t\t// Fail immediately if we are in LOCAL_PASSTHROUGH mode\n\t\t\tvar zero SP\n\t\t\treturn zero, fmt.Errorf(\"%w for shard ID: %s\", ErrShardProcessNotFound, shardID)\n\t\t}\n\n\t\t// Do a heartbeat and check again\n\t\terr := e.heartbeatAndUpdateAssignment(ctx)\n\t\tif err != nil {\n\t\t\tvar zero SP\n\t\t\treturn zero, fmt.Errorf(\"heartbeat and assign shards: %w\", err)\n\t\t}\n\n\t\t// Check again if the shard process is found\n\t\tshardProcess, ok = e.managedProcessors.Load(shardID)\n\t\tif !ok {\n\t\t\tvar zero SP\n\t\t\treturn zero, fmt.Errorf(\"%w for shard ID: %s\", ErrShardProcessNotFound, shardID)\n\t\t}\n\t}\n\n\treturn shardProcess.processor, nil\n}\n\nfunc (e *executorImpl[SP]) IsOnboardedToSD() bool {\n\treturn e.getMigrationMode() == types.MigrationModeONBOARDED\n}\n\nfunc (e *executorImpl[SP]) AssignShardsFromLocalLogic(ctx context.Context, shardAssignment map[string]*types.ShardAssignment) error {\n\te.assignmentMutex.Lock()\n\tdefer e.assignmentMutex.Unlock()\n\tif e.getMigrationMode() == types.MigrationModeONBOARDED {\n\t\treturn fmt.Errorf(\"migration mode is onborded, no local assignemnt allowed\")\n\t}\n\te.logger.Info(\"Executing external shard assignment\")\n\te.addNewShards(ctx, shardAssignment)\n\treturn nil\n}\n\nfunc (e *executorImpl[SP]) RemoveShardsFromLocalLogic(shardIDs []string) error {\n\tif e.getMigrationMode() == types.MigrationModeONBOARDED {\n\t\treturn fmt.Errorf(\"migration mode is onborded, no local assignemnt allowed\")\n\t}\n\n\treturn e.removeShards(shardIDs)\n}\n\nfunc (e *executorImpl[SP]) removeShards(shardIDs []string) error {\n\te.assignmentMutex.Lock()\n\tdefer e.assignmentMutex.Unlock()\n\te.logger.Info(\"Executing external shard deletion assignment\")\n\te.deleteShards(shardIDs)\n\treturn nil\n}\n\n// drainChannel returns the drain signal channel, or nil if no observer is configured.\nfunc (e *executorImpl[SP]) drainChannel() <-chan struct{} {\n\tif e.drainObserver != nil {\n\t\treturn e.drainObserver.Drain()\n\t}\n\treturn nil\n}\n\nfunc (e *executorImpl[SP]) heartbeatloop(ctx context.Context) {\n\t// Check if initial migration mode is LOCAL_PASSTHROUGH - if so, skip heartbeating entirely\n\tif e.getMigrationMode() == types.MigrationModeLOCALPASSTHROUGH {\n\t\te.logger.Info(\"initial migration mode is local passthrough, skipping heartbeat loop\")\n\t\treturn\n\t}\n\n\theartBeatTimer := e.timeSource.NewTimer(backoff.JitDuration(e.heartBeatInterval, heartbeatJitterCoeff))\n\tdefer heartBeatTimer.Stop()\n\n\tdrainCh := e.drainChannel()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\te.logger.Info(\"shard distributor executor context done, stopping\")\n\t\t\te.stopShardProcessors()\n\t\t\te.sendDrainingHeartbeat()\n\t\t\treturn\n\t\tcase <-e.stopC:\n\t\t\te.logger.Info(\"shard distributor executor stopped\")\n\t\t\te.stopShardProcessors()\n\t\t\te.sendDrainingHeartbeat()\n\t\t\treturn\n\t\tcase <-drainCh:\n\t\t\te.logger.Info(\"drain signal received, stopping shard processors\")\n\t\t\te.stopShardProcessors()\n\t\t\te.sendDrainingHeartbeat()\n\n\t\t\tif !e.waitForUndrain(ctx) {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\te.logger.Info(\"undrain signal received, resuming heartbeat\")\n\t\t\tdrainCh = e.drainObserver.Drain()\n\t\t\theartBeatTimer.Reset(backoff.JitDuration(e.heartBeatInterval, heartbeatJitterCoeff))\n\t\tcase <-heartBeatTimer.Chan():\n\t\t\theartBeatTimer.Reset(backoff.JitDuration(e.heartBeatInterval, heartbeatJitterCoeff))\n\t\t\terr := e.heartbeatAndUpdateAssignment(ctx)\n\t\t\tif errors.Is(err, ErrLocalPassthroughMode) {\n\t\t\t\te.logger.Info(\"local passthrough mode: stopping heartbeat loop\")\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\te.logger.Error(\"failed to heartbeat and assign shards\", tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t}\n\t}\n}\n\n// waitForUndrain blocks until the undrain signal fires or the executor is stopped.\n// Returns true if undrained (caller should resume), false if stopped.\nfunc (e *executorImpl[SP]) waitForUndrain(ctx context.Context) bool {\n\tif e.drainObserver == nil {\n\t\treturn false\n\t}\n\n\tundrainCh := e.drainObserver.Undrain()\n\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn false\n\tcase <-e.stopC:\n\t\treturn false\n\tcase <-undrainCh:\n\t\treturn true\n\t}\n}\n\nfunc (e *executorImpl[SP]) heartbeatAndUpdateAssignment(ctx context.Context) error {\n\tif !e.assignmentMutex.TryLock() {\n\t\te.logger.Error(\"still doing assignment, skipping heartbeat\")\n\t\te.metrics.Counter(metricsconstants.ShardDistributorExecutorHeartbeatSkipped).Inc(1)\n\t\treturn nil\n\t}\n\tdefer e.assignmentMutex.Unlock()\n\tshardAssignment, err := e.heartbeatAndHandleMigrationMode(ctx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif shardAssignment != nil {\n\t\te.updateShardAssignmentMetered(ctx, shardAssignment)\n\t}\n\treturn nil\n}\n\nfunc (e *executorImpl[SP]) heartbeatAndHandleMigrationMode(ctx context.Context) (shardAssignment map[string]*types.ShardAssignment, err error) {\n\tshardAssignment, migrationMode, err := e.heartbeat(ctx)\n\tif err != nil {\n\t\t// TODO: should we stop the executor, and drop all the shards?\n\t\treturn nil, fmt.Errorf(\"failed to heartbeat: %w\", err)\n\t}\n\n\t// Handle migration mode logic\n\tswitch migrationMode {\n\tcase types.MigrationModeLOCALPASSTHROUGH:\n\t\t// LOCAL_PASSTHROUGH: statically assigned, stop heartbeating\n\t\treturn nil, ErrLocalPassthroughMode\n\n\tcase types.MigrationModeLOCALPASSTHROUGHSHADOW:\n\t\t// LOCAL_PASSTHROUGH_SHADOW: check response but don't apply it\n\t\terr = e.compareAssignments(shardAssignment)\n\t\treturn nil, err\n\n\tcase types.MigrationModeDISTRIBUTEDPASSTHROUGH:\n\t\t// DISTRIBUTED_PASSTHROUGH: validate then apply the assignment\n\t\terr = e.compareAssignments(shardAssignment)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\treturn shardAssignment, nil\n\t\t// Continue with applying the assignment from heartbeat\n\n\tcase types.MigrationModeONBOARDED:\n\t\t// ONBOARDED: normal flow, apply the assignment from heartbeat\n\t\treturn shardAssignment, nil\n\t\t// Continue with normal assignment logic below\n\n\tdefault:\n\t\te.logger.Warn(\"unknown migration mode, skipping assignment\",\n\t\t\ttag.ShardNamespace(e.namespace), tag.Dynamic(\"migration-mode\", migrationMode))\n\t\treturn nil, nil\n\t}\n}\n\nfunc (e *executorImpl[SP]) updateShardAssignmentMetered(ctx context.Context, shardAssignment map[string]*types.ShardAssignment) {\n\tstartTime := e.timeSource.Now()\n\tdefer e.metrics.\n\t\tHistogram(metricsconstants.ShardDistributorExecutorAssignLoopLatency, metricsconstants.ShardDistributorExecutorAssignLoopLatencyBuckets).\n\t\tRecordDuration(e.timeSource.Since(startTime))\n\n\te.updateShardAssignment(ctx, shardAssignment)\n}\n\nfunc (e *executorImpl[SP]) heartbeat(ctx context.Context) (shardAssignments map[string]*types.ShardAssignment, migrationMode types.MigrationMode, err error) {\n\treturn e.sendHeartbeat(ctx, types.ExecutorStatusACTIVE)\n}\n\nfunc (e *executorImpl[SP]) sendHeartbeat(ctx context.Context, status types.ExecutorStatus) (map[string]*types.ShardAssignment, types.MigrationMode, error) {\n\t// Fill in the shard status reports\n\tshardStatusReports := make(map[string]*types.ShardStatusReport)\n\te.managedProcessors.Range(func(shardID string, managedProcessor *managedProcessor[SP]) bool {\n\t\tif managedProcessor.getState() == processorStateStarted {\n\t\t\tshardStatus := managedProcessor.processor.GetShardReport()\n\n\t\t\tshardStatusReports[shardID] = &types.ShardStatusReport{\n\t\t\t\tShardLoad: shardStatus.ShardLoad,\n\t\t\t\tStatus:    shardStatus.Status,\n\t\t\t}\n\t\t}\n\t\treturn true\n\t})\n\n\te.hostMetrics.Gauge(metricsconstants.ShardDistributorExecutorOwnedShards).Update(float64(len(shardStatusReports)))\n\n\t// Create the request\n\trequest := &types.ExecutorHeartbeatRequest{\n\t\tNamespace:          e.namespace,\n\t\tExecutorID:         e.executorID,\n\t\tStatus:             status,\n\t\tShardStatusReports: shardStatusReports,\n\t\tMetadata:           e.metadata.Get(),\n\t}\n\n\t// Send the request\n\tresponse, err := e.shardDistributorClient.Heartbeat(ctx, request)\n\tif err != nil {\n\t\treturn nil, types.MigrationModeINVALID, fmt.Errorf(\"send heartbeat: %w\", err)\n\t}\n\n\tpreviousMode := e.getMigrationMode()\n\tcurrentMode := response.MigrationMode\n\tif previousMode != currentMode {\n\t\te.logger.Info(\"migration mode transition\",\n\t\t\ttag.Dynamic(\"previous\", previousMode),\n\t\t\ttag.Dynamic(\"current\", currentMode),\n\t\t\ttag.ShardNamespace(e.namespace),\n\t\t\ttag.ShardExecutor(e.executorID))\n\t\te.setMigrationMode(currentMode)\n\t}\n\n\treturn response.ShardAssignments, response.MigrationMode, nil\n}\n\nfunc (e *executorImpl[SP]) sendDrainingHeartbeat() {\n\tctx, cancel := context.WithTimeout(context.Background(), drainingHeartbeatTimeout)\n\tdefer cancel()\n\n\t_, _, err := e.sendHeartbeat(ctx, types.ExecutorStatusDRAINING)\n\tif err != nil {\n\t\te.logger.Error(\"failed to send draining heartbeat\", tag.Error(err))\n\t}\n}\n\nfunc (e *executorImpl[SP]) updateShardAssignment(ctx context.Context, shardAssignments map[string]*types.ShardAssignment) {\n\twg := sync.WaitGroup{}\n\n\t// Stop shard processing for shards not assigned to this executor\n\te.managedProcessors.Range(func(shardID string, managedProcessor *managedProcessor[SP]) bool {\n\t\tif assignment, ok := shardAssignments[shardID]; !ok || assignment.Status != types.AssignmentStatusREADY {\n\t\t\twg.Add(1)\n\t\t\tgo func(shardID string) {\n\t\t\t\tdefer wg.Done()\n\t\t\t\te.stopManagerProcessor(shardID)\n\t\t\t}(shardID)\n\t\t}\n\t\treturn true\n\t})\n\n\t// Start shard processing for shards assigned to this executor\n\tfor shardID, assignment := range shardAssignments {\n\t\tif assignment.Status == types.AssignmentStatusREADY {\n\t\t\twg.Add(1)\n\t\t\tgo func(shardID string) {\n\t\t\t\tdefer wg.Done()\n\t\t\t\te.addManagerProcessor(ctx, shardID)\n\t\t\t}(shardID)\n\t\t}\n\t}\n\n\twg.Wait()\n}\n\nfunc (e *executorImpl[SP]) addNewShards(ctx context.Context, shardAssignments map[string]*types.ShardAssignment) {\n\twg := sync.WaitGroup{}\n\n\tfor shardID, assignment := range shardAssignments {\n\t\tif assignment.Status == types.AssignmentStatusREADY {\n\t\t\twg.Add(1)\n\t\t\tgo func(shardID string) {\n\t\t\t\tdefer wg.Done()\n\t\t\t\te.addManagerProcessor(ctx, shardID)\n\t\t\t}(shardID)\n\t\t}\n\t}\n\n\twg.Wait()\n}\n\nfunc (e *executorImpl[SP]) deleteShards(shardIDs []string) {\n\twg := sync.WaitGroup{}\n\tfor _, shardID := range shardIDs {\n\t\twg.Add(1)\n\t\tgo func(shardID string) {\n\t\t\tdefer wg.Done()\n\t\t\te.stopManagerProcessor(shardID)\n\t\t}(shardID)\n\t}\n\twg.Wait()\n}\n\nfunc (e *executorImpl[SP]) stopShardProcessors() {\n\twg := sync.WaitGroup{}\n\n\te.managedProcessors.Range(func(shardID string, managedProcessor *managedProcessor[SP]) bool {\n\t\twg.Add(1)\n\t\tgo func(shardID string) {\n\t\t\tdefer wg.Done()\n\t\t\te.stopManagerProcessor(shardID)\n\t\t}(shardID)\n\t\treturn true\n\t})\n\n\twg.Wait()\n}\n\nfunc (e *executorImpl[SP]) addManagerProcessor(ctx context.Context, shardID string) {\n\tif _, ok := e.managedProcessors.Load(shardID); !ok {\n\t\te.metrics.Counter(metricsconstants.ShardDistributorExecutorShardsStarted).Inc(1)\n\t\tprocessor, err := e.shardProcessorFactory.NewShardProcessor(shardID)\n\t\tif err != nil {\n\t\t\te.logger.Error(\"failed to create shard processor\", tag.Error(err))\n\t\t\te.metrics.Counter(metricsconstants.ShardDistributorExecutorProcessorCreationFailures).Inc(1)\n\t\t\treturn\n\t\t}\n\t\tmanagedProcessor := newManagedProcessor(processor, processorStateStarting)\n\t\te.managedProcessors.Store(shardID, managedProcessor)\n\n\t\tprocessor.Start(ctx)\n\n\t\tmanagedProcessor.setState(processorStateStarted)\n\n\t}\n}\nfunc (e *executorImpl[SP]) stopManagerProcessor(shardID string) {\n\tmanagedProcessor, ok := e.managedProcessors.Load(shardID)\n\t// If the processor do not exist for the shard, or it is already stopping, skip it\n\tif !ok || managedProcessor.getState() == processorStateStopping {\n\t\treturn\n\t}\n\te.metrics.Counter(metricsconstants.ShardDistributorExecutorShardsStopped).Inc(1)\n\tmanagedProcessor.setState(processorStateStopping)\n\tmanagedProcessor.processor.Stop()\n\te.managedProcessors.Delete(shardID)\n}\n\nfunc (e *executorImpl[SP]) shardCleanUpLoop(ctx context.Context) {\n\t// We don't run the loop for invalid durations\n\tif e.ttlShard <= 0 {\n\n\t\treturn\n\t}\n\tshardCleanUpTimer := e.timeSource.NewTimer(backoff.JitDuration(e.ttlShard, heartbeatJitterCoeff))\n\tdefer shardCleanUpTimer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn\n\t\tcase <-e.stopC:\n\t\t\treturn\n\t\tcase <-shardCleanUpTimer.Chan():\n\t\t\te.processorsToLastUse.Range(func(shardID string, time time.Time) bool {\n\t\t\t\tif time.Add(e.ttlShard).Before(e.timeSource.Now()) {\n\t\t\t\t\tif e.getMigrationMode() == types.MigrationModeONBOARDED {\n\t\t\t\t\t\tmp, ok := e.managedProcessors.Load(shardID)\n\t\t\t\t\t\tif ok {\n\t\t\t\t\t\t\tmp.processor.SetShardStatus(types.ShardStatusDONE)\n\t\t\t\t\t\t}\n\t\t\t\t\t} else {\n\t\t\t\t\t\te.deleteShards([]string{shardID})\n\t\t\t\t\t}\n\t\t\t\t\te.processorsToLastUse.Delete(shardID)\n\t\t\t\t}\n\t\t\t\treturn true\n\t\t\t})\n\t\t}\n\t}\n}\n\n// compareAssignments compares the local assignments with the heartbeat response assignments\n// return error if the assignment are not the same and emits convergence or divergence metrics\nfunc (e *executorImpl[SP]) compareAssignments(heartbeatAssignments map[string]*types.ShardAssignment) error {\n\t// Get current local assignments\n\tlocalAssignments := make(map[string]bool)\n\te.managedProcessors.Range(func(shardID string, managedProcessor *managedProcessor[SP]) bool {\n\t\tif managedProcessor.getState() == processorStateStarted {\n\t\t\tlocalAssignments[shardID] = true\n\t\t}\n\t\treturn true\n\t})\n\n\t// Check if all local assignments are in heartbeat assignments with READY status\n\tfor shardID := range localAssignments {\n\t\tassignment, exists := heartbeatAssignments[shardID]\n\t\tif !exists || assignment.Status != types.AssignmentStatusREADY {\n\t\t\te.logger.Warn(\"assignment divergence: local shard not in heartbeat or not ready\",\n\t\t\t\ttag.Dynamic(\"shard-id\", shardID))\n\t\t\te.emitMetricsConvergence(false)\n\t\t\treturn ErrAssignmentDivergenceLocalShard\n\t\t}\n\t}\n\n\t// Check if all heartbeat READY assignments are in local assignments\n\tfor shardID, assignment := range heartbeatAssignments {\n\t\tif assignment.Status == types.AssignmentStatusREADY {\n\t\t\tif !localAssignments[shardID] {\n\t\t\t\te.logger.Warn(\"assignment divergence: heartbeat shard not in local\",\n\t\t\t\t\ttag.Dynamic(\"shard-id\", shardID))\n\t\t\t\te.emitMetricsConvergence(false)\n\t\t\t\treturn ErrAssignmentDivergenceHeartbeatShard\n\t\t\t}\n\t\t}\n\t}\n\n\te.emitMetricsConvergence(true)\n\treturn nil\n}\n\nfunc (e *executorImpl[SP]) emitMetricsConvergence(converged bool) {\n\tif converged {\n\t\te.metrics.Counter(metricsconstants.ShardDistributorExecutorAssignmentConvergence).Inc(1)\n\t} else {\n\t\te.metrics.Counter(metricsconstants.ShardDistributorExecutorAssignmentDivergence).Inc(1)\n\t}\n}\n\nfunc (e *executorImpl[SP]) GetNamespace() string {\n\treturn e.namespace\n}\n\nfunc (e *executorImpl[SP]) SetMetadata(metadata map[string]string) {\n\te.metadata.Set(metadata)\n}\n\nfunc (e *executorImpl[SP]) GetMetadata() map[string]string {\n\treturn e.metadata.Get()\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/clientimpl_test.go",
    "content": "package executorclient\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient/syncgeneric\"\n)\n\n// closeDrainObserver is a test helper that implements DrainSignalObserver\n// with close-to-broadcast semantics, matching the real implementation.\ntype closeDrainObserver struct {\n\tmu        sync.Mutex\n\tdrainCh   chan struct{}\n\tundrainCh chan struct{}\n}\n\nfunc newCloseDrainObserver() *closeDrainObserver {\n\treturn &closeDrainObserver{\n\t\tdrainCh:   make(chan struct{}),\n\t\tundrainCh: make(chan struct{}),\n\t}\n}\n\nfunc (o *closeDrainObserver) Drain() <-chan struct{} {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\treturn o.drainCh\n}\n\nfunc (o *closeDrainObserver) Undrain() <-chan struct{} {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\treturn o.undrainCh\n}\n\nfunc (o *closeDrainObserver) SignalDrain() {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\tclose(o.drainCh)\n\to.undrainCh = make(chan struct{})\n}\n\nfunc (o *closeDrainObserver) SignalUndrain() {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\tclose(o.undrainCh)\n\to.drainCh = make(chan struct{})\n}\n\nfunc expectDrainingHeartbeat(t *testing.T, mockClient *sharddistributorexecutor.MockClient) {\n\tmockClient.EXPECT().\n\t\tHeartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\tassert.Equal(t, types.ExecutorStatusDRAINING, req.Status)\n\t\t\treturn &types.ExecutorHeartbeatResponse{}, nil\n\t\t})\n}\n\nfunc newTestExecutor(\n\tclient sharddistributorexecutor.Client,\n\tfactory ShardProcessorFactory[*MockShardProcessor],\n\ttimeSource clock.TimeSource,\n) *executorImpl[*MockShardProcessor] {\n\tif timeSource == nil {\n\t\ttimeSource = clock.NewMockedTimeSource()\n\t}\n\treturn &executorImpl[*MockShardProcessor]{\n\t\tlogger:                 log.NewNoop(),\n\t\tmetrics:                tally.NoopScope,\n\t\thostMetrics:            tally.NoopScope,\n\t\tshardDistributorClient: client,\n\t\tshardProcessorFactory:  factory,\n\t\tnamespace:              \"test-namespace\",\n\t\tstopC:                  make(chan struct{}),\n\t\theartBeatInterval:      10 * time.Second,\n\t\tmanagedProcessors:      syncgeneric.Map[string, *managedProcessor[*MockShardProcessor]]{},\n\t\texecutorID:             \"test-executor-id\",\n\t\ttimeSource:             timeSource,\n\t}\n}\n\nfunc TestHeartBeatLoop(t *testing.T) {\n\t// Ensure that there are no goroutines leaked\n\tdefer goleak.VerifyNone(t)\n\n\t// Create mocks\n\tctrl := gomock.NewController(t)\n\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\n\t// We expect nothing is assigned to the executor, and we assign two shards to it\n\tmockShardDistributorClient.EXPECT().Heartbeat(gomock.Any(),\n\t\t&types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:          \"test-namespace\",\n\t\t\tExecutorID:         \"test-executor-id\",\n\t\t\tStatus:             types.ExecutorStatusACTIVE,\n\t\t\tShardStatusReports: make(map[string]*types.ShardStatusReport),\n\t\t\tMetadata:           make(map[string]string),\n\t\t}, gomock.Any()).\n\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\tMigrationMode: types.MigrationModeONBOARDED,\n\t\t}, nil)\n\texpectDrainingHeartbeat(t, mockShardDistributorClient)\n\n\t// The two shards are assigned to the executor, so we expect them to be created, started and stopped\n\tmockShardProcessor1 := NewMockShardProcessor(ctrl)\n\tmockShardProcessor1.EXPECT().Start(gomock.Any())\n\tmockShardProcessor1.EXPECT().Stop()\n\n\tmockShardProcessor2 := NewMockShardProcessor(ctrl)\n\tmockShardProcessor2.EXPECT().Start(gomock.Any())\n\tmockShardProcessor2.EXPECT().Stop()\n\n\tmockShardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\tmockShardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(mockShardProcessor1, nil)\n\tmockShardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(mockShardProcessor2, nil)\n\n\t// We use a mock time source to control the heartbeat loop\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\t// Create the executor\n\texecutor := newTestExecutor(mockShardDistributorClient, mockShardProcessorFactory, mockTimeSource)\n\n\t// Start the executor, and defer stopping it\n\texecutor.Start(context.Background())\n\tdefer executor.Stop()\n\n\t// Make sure the heartbeat loop has done an iteration and assigned the shards to the executor\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(15 * time.Second)\n\ttime.Sleep(10 * time.Millisecond) // Force the heartbeatloop goroutine to run\n\tmockTimeSource.BlockUntil(1)\n\n\t// Assert that the two shards are assigned to the executor\n\tprocessor1, err := executor.GetShardProcess(context.Background(), \"test-shard-id1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, mockShardProcessor1, processor1)\n\n\tprocessor2, err := executor.GetShardProcess(context.Background(), \"test-shard-id2\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, mockShardProcessor2, processor2)\n\n\t// Check that non-owned-shard-id is not in the local cache, we check directly since we don't want to trigger a heartbeat\n\t_, ok := executor.managedProcessors.Load(\"non-owned-shard-id\")\n\tassert.False(t, ok)\n}\n\nfunc TestHeartbeat(t *testing.T) {\n\t// Setup mocks\n\tctrl := gomock.NewController(t)\n\n\t// We have two shards assigned to the executor, and we expect a third shard to be assigned to it\n\tshardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\tshardDistributorClient.EXPECT().Heartbeat(gomock.Any(),\n\t\t&types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  \"test-namespace\",\n\t\t\tExecutorID: \"test-executor-id\",\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t\tShardStatusReports: map[string]*types.ShardStatusReport{\n\t\t\t\t\"test-shard-id1\": {Status: types.ShardStatusREADY, ShardLoad: 0.123},\n\t\t\t\t\"test-shard-id2\": {Status: types.ShardStatusREADY, ShardLoad: 0.456},\n\t\t\t},\n\t\t\tMetadata: make(map[string]string),\n\t\t}, gomock.Any()).Return(&types.ExecutorHeartbeatResponse{\n\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\"test-shard-id3\": {Status: types.AssignmentStatusREADY},\n\t\t},\n\t\tMigrationMode: types.MigrationModeONBOARDED,\n\t}, nil)\n\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock1.EXPECT().GetShardReport().Return(ShardReport{ShardLoad: 0.123, Status: types.ShardStatusREADY})\n\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock2.EXPECT().GetShardReport().Return(ShardReport{ShardLoad: 0.456, Status: types.ShardStatusREADY})\n\n\t// Create the executor\n\texecutor := newTestExecutor(shardDistributorClient, nil, nil)\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t// Do the call to heartbeat\n\tshardAssignments, _, err := executor.heartbeat(context.Background())\n\n\t// Assert that we now have 3 shards in the assignment\n\tassert.NoError(t, err)\n\tassert.Equal(t, 3, len(shardAssignments))\n\tassert.Equal(t, types.AssignmentStatusREADY, shardAssignments[\"test-shard-id1\"].Status)\n\tassert.Equal(t, types.AssignmentStatusREADY, shardAssignments[\"test-shard-id2\"].Status)\n\tassert.Equal(t, types.AssignmentStatusREADY, shardAssignments[\"test-shard-id3\"].Status)\n}\n\nfunc TestHeartBeatLoop_ShardAssignmentChange(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Setup mocks\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock3 := NewMockShardProcessor(ctrl)\n\n\tshardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\tshardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(shardProcessorMock3, nil)\n\n\t// Create the executor currently has shards 1 and 2 assigned to it\n\texecutor := newTestExecutor(nil, shardProcessorFactory, nil)\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t// We expect to get a new assignment with shards 2 and 3 assigned to it\n\tnewAssignment := map[string]*types.ShardAssignment{\n\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t\t\"test-shard-id3\": {Status: types.AssignmentStatusREADY},\n\t}\n\n\t// With the new assignment, shardProcessorMock1 should be stopped and shardProcessorMock3 should be started\n\tshardProcessorMock1.EXPECT().Stop()\n\tshardProcessorMock3.EXPECT().Start(gomock.Any())\n\n\t// Update the shard assignment\n\texecutor.updateShardAssignment(context.Background(), newAssignment)\n\ttime.Sleep(10 * time.Millisecond) // Force the updateShardAssignment goroutines to run\n\n\t// Assert that we now have the 2 shards in the assignment\n\tprocessor2, err := executor.GetShardProcess(context.Background(), \"test-shard-id2\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, shardProcessorMock2, processor2)\n\n\tprocessor3, err := executor.GetShardProcess(context.Background(), \"test-shard-id3\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, shardProcessorMock3, processor3)\n\n\t// Check that we do not have shard \"test-shard-id1\" in the local cache\n\t// we lookup directly since we don't want to trigger a heartbeat\n\t_, ok := executor.managedProcessors.Load(\"test-shard-id1\")\n\tassert.False(t, ok)\n}\n\nfunc TestAssignShardsFromLocalLogic(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\ttests := []struct {\n\t\tname   string\n\t\tparams map[string]*types.ShardAssignment\n\t\tsetup  func() *executorImpl[*MockShardProcessor]\n\t\tassert func(err error, executor *executorImpl[*MockShardProcessor])\n\t}{\n\t\t{\n\t\t\tname:   \"AssignShardsFromLocalLogic fails if the namespace is onboarded\",\n\t\t\tparams: map[string]*types.ShardAssignment{},\n\t\t\tsetup: func() *executorImpl[*MockShardProcessor] {\n\t\t\t\texecutor := newTestExecutor(nil, nil, nil)\n\t\t\t\texecutor.setMigrationMode(types.MigrationModeONBOARDED)\n\t\t\t\treturn executor\n\t\t\t},\n\t\t\tassert: func(err error, executor *executorImpl[*MockShardProcessor]) {},\n\t\t},\n\t\t{\n\t\t\tname: \"AssignShardsFromLocalLogic succeed with only logs if it is not possible to create a new shard processor\",\n\t\t\tparams: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\"test-shard-id3\": {Status: types.AssignmentStatusREADY}},\n\t\t\tsetup: func() *executorImpl[*MockShardProcessor] {\n\t\t\t\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\t\t\t\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\n\t\t\t\tshardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\n\t\t\t\t// Setup mocks\n\t\t\t\tshardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(nil, assert.AnError)\n\n\t\t\t\texecutor := newTestExecutor(nil, shardProcessorFactory, nil)\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t\t\t\treturn executor\n\t\t\t},\n\t\t\tassert: func(err error, executor *executorImpl[*MockShardProcessor]) {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"AssignShardsFromLocalLogic succeed \",\n\t\t\tparams: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\"test-shard-id3\": {Status: types.AssignmentStatusREADY}},\n\t\t\tsetup: func() *executorImpl[*MockShardProcessor] {\n\t\t\t\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\t\t\t\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\t\t\t\tshardProcessorMock3 := NewMockShardProcessor(ctrl)\n\n\t\t\t\tshardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\n\t\t\t\tshardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(shardProcessorMock3, nil)\n\n\t\t\t\texecutor := newTestExecutor(nil, shardProcessorFactory, nil)\n\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t\t\t\t// With the new assignment, shardProcessorMock3 should be started\n\t\t\t\tshardProcessorMock3.EXPECT().Start(gomock.Any())\n\t\t\t\tshardProcessorMock1.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY})\n\t\t\t\tshardProcessorMock2.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY})\n\t\t\t\tshardProcessorMock3.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY})\n\t\t\t\treturn executor\n\t\t\t},\n\t\t\tassert: func(_ error, executor *executorImpl[*MockShardProcessor]) {\n\t\t\t\t// Assert that we now have the 3 shards in the assignment\n\t\t\t\tprocessor1, err := executor.GetShardProcess(context.Background(), \"test-shard-id1\")\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, types.ShardStatusREADY, processor1.GetShardReport().Status)\n\n\t\t\t\tprocessor2, err := executor.GetShardProcess(context.Background(), \"test-shard-id2\")\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, types.ShardStatusREADY, processor2.GetShardReport().Status)\n\n\t\t\t\tprocessor3, err := executor.GetShardProcess(context.Background(), \"test-shard-id3\")\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, types.ShardStatusREADY, processor3.GetShardReport().Status)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\texecutor := test.setup()\n\t\t\terr := executor.AssignShardsFromLocalLogic(context.Background(), test.params)\n\t\t\ttime.Sleep(10 * time.Millisecond) // Force the updateShardAssignment goroutines to run\n\t\t\ttest.assert(err, executor)\n\t\t})\n\t}\n}\n\nfunc TestRemoveShardsFromLocalLogic(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\ttests := []struct {\n\t\tname   string\n\t\tparams []string\n\t\tsetup  func() *executorImpl[*MockShardProcessor]\n\t\tassert func(err error, executor *executorImpl[*MockShardProcessor])\n\t}{\n\t\t{\n\t\t\tname:   \"RemoveShardsFromLocalLogic fails if the namespace is onboarded\",\n\t\t\tparams: []string{},\n\t\t\tsetup: func() *executorImpl[*MockShardProcessor] {\n\t\t\t\texecutor := newTestExecutor(nil, nil, nil)\n\t\t\t\texecutor.setMigrationMode(types.MigrationModeONBOARDED)\n\t\t\t\treturn executor\n\t\t\t},\n\t\t\tassert: func(err error, executor *executorImpl[*MockShardProcessor]) {},\n\t\t},\n\t\t{\n\t\t\tname: \"RemoveShardsFromLocalLogic succeed \",\n\t\t\tparams: []string{\n\t\t\t\t\"test-shard-id2\",\n\t\t\t\t\"test-shard-id3\"},\n\t\t\tsetup: func() *executorImpl[*MockShardProcessor] {\n\t\t\t\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\t\t\t\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\t\t\t\texecutor := newTestExecutor(nil, nil, nil)\n\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\t\t\t\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t\t\t\t// With the new assignment, shardProcessorMock2 should be stopped\n\t\t\t\tshardProcessorMock2.EXPECT().Stop()\n\t\t\t\tshardProcessorMock1.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY})\n\t\t\t\treturn executor\n\t\t\t},\n\t\t\tassert: func(_ error, executor *executorImpl[*MockShardProcessor]) {\n\t\t\t\t// Assert that we now have the 1 shard in the assignment\n\t\t\t\tprocessor1, err := executor.GetShardProcess(context.Background(), \"test-shard-id1\")\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, types.ShardStatusREADY, processor1.GetShardReport().Status)\n\n\t\t\t\t// Check that we do not have shard \"test-shard-id2\" in the local cache\n\t\t\t\t// we lookup directly since we don't want to trigger a heartbeat\n\t\t\t\t_, ok := executor.managedProcessors.Load(\"test-shard-id2\")\n\t\t\t\tassert.False(t, ok)\n\t\t\t},\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\tt.Run(test.name, func(t *testing.T) {\n\t\t\texecutor := test.setup()\n\t\t\terr := executor.RemoveShardsFromLocalLogic(test.params)\n\t\t\ttime.Sleep(10 * time.Millisecond) // Force the updateShardAssignment goroutines to run\n\t\t\ttest.assert(err, executor)\n\t\t})\n\t}\n}\n\nfunc TestHeartbeat_WithMigrationMode(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Test that heartbeat returns migration mode correctly\n\tshardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\tshardDistributorClient.EXPECT().Heartbeat(gomock.Any(),\n\t\t&types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:          \"test-namespace\",\n\t\t\tExecutorID:         \"test-executor-id\",\n\t\t\tStatus:             types.ExecutorStatusACTIVE,\n\t\t\tShardStatusReports: map[string]*types.ShardStatusReport{},\n\t\t\tMetadata:           make(map[string]string),\n\t\t}, gomock.Any()).Return(&types.ExecutorHeartbeatResponse{\n\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t},\n\t\tMigrationMode: types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t}, nil)\n\n\texecutor := newTestExecutor(shardDistributorClient, nil, nil)\n\texecutor.setMigrationMode(types.MigrationModeINVALID)\n\n\tshardAssignments, migrationMode, err := executor.heartbeat(context.Background())\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, 1, len(shardAssignments))\n\tassert.Equal(t, types.MigrationModeDISTRIBUTEDPASSTHROUGH, migrationMode)\n\tassert.Equal(t, types.MigrationModeDISTRIBUTEDPASSTHROUGH, executor.getMigrationMode())\n}\n\nfunc TestHeartbeat_MigrationModeTransition(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tshardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\tshardDistributorClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).Return(&types.ExecutorHeartbeatResponse{\n\t\tShardAssignments: map[string]*types.ShardAssignment{},\n\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t}, nil)\n\n\texecutor := newTestExecutor(shardDistributorClient, nil, nil)\n\texecutor.setMigrationMode(types.MigrationModeDISTRIBUTEDPASSTHROUGH)\n\n\t_, migrationMode, err := executor.heartbeat(context.Background())\n\n\tassert.NoError(t, err)\n\tassert.Equal(t, types.MigrationModeONBOARDED, migrationMode)\n\tassert.Equal(t, types.MigrationModeONBOARDED, executor.getMigrationMode())\n}\n\nfunc TestHeartbeatLoop_LocalPassthrough_SkipsHeartbeat(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\n\t// No heartbeat should be called for LOCAL_PASSTHROUGH mode\n\tmockShardDistributorClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).Times(0)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\texecutor := newTestExecutor(mockShardDistributorClient, nil, mockTimeSource)\n\texecutor.setMigrationMode(types.MigrationModeLOCALPASSTHROUGH)\n\n\texecutor.Start(context.Background())\n\tdefer executor.Stop()\n\n\t// Give some time for the heartbeat loop to potentially run (it shouldn't)\n\ttime.Sleep(10 * time.Millisecond)\n}\n\nfunc TestHeartbeatLoop_LocalPassthroughShadow_SkipsAssignment(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\n\t// Heartbeat should be called but assignment should not be applied\n\tmockShardDistributorClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\tMigrationMode: types.MigrationModeLOCALPASSTHROUGHSHADOW,\n\t\t}, nil)\n\texpectDrainingHeartbeat(t, mockShardDistributorClient)\n\n\tmockShardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\t// No shard processor should be created\n\tmockShardProcessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Times(0)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\texecutor := newTestExecutor(mockShardDistributorClient, mockShardProcessorFactory, mockTimeSource)\n\texecutor.setMigrationMode(types.MigrationModeONBOARDED)\n\n\texecutor.Start(context.Background())\n\tdefer executor.Stop()\n\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(15 * time.Second)\n\ttime.Sleep(10 * time.Millisecond)\n\tmockTimeSource.BlockUntil(1)\n\n\t// Assert that no shards were assigned, we check directly since we don't want to trigger a heartbeat\n\t_, ok := executor.managedProcessors.Load(\"test-shard-id1\")\n\tassert.False(t, ok)\n}\n\nfunc TestHeartbeatLoop_DistributedPassthrough_AppliesAssignment(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\n\tmockShardProcessor := NewMockShardProcessor(ctrl)\n\tmockShardProcessor.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY}).AnyTimes()\n\tmockShardProcessor.EXPECT().Stop()\n\n\tmockShardDistributorClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\tMigrationMode: types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t\t}, nil)\n\texpectDrainingHeartbeat(t, mockShardDistributorClient)\n\n\tmockShardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\texecutor := newTestExecutor(mockShardDistributorClient, mockShardProcessorFactory, mockTimeSource)\n\texecutor.setMigrationMode(types.MigrationModeONBOARDED)\n\t// Pre-populate the executor with the shard to ensure convergence\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(mockShardProcessor, processorStateStarted))\n\n\texecutor.Start(context.Background())\n\tdefer executor.Stop()\n\n\tmockTimeSource.BlockUntil(1)\n\tmockTimeSource.Advance(15 * time.Second)\n\ttime.Sleep(10 * time.Millisecond)\n\tmockTimeSource.BlockUntil(1)\n\n\t// Assert that the shard was assigned\n\tprocessor, err := executor.GetShardProcess(context.Background(), \"test-shard-id1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, mockShardProcessor, processor)\n}\n\nfunc TestHeartbeatLoop_StopSignalSendsDrainingHeartbeat(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\texpectDrainingHeartbeat(t, mockShardDistributorClient)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\texecutor := newTestExecutor(mockShardDistributorClient, nil, mockTimeSource)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\tdone := make(chan struct{})\n\tgo func() {\n\t\texecutor.heartbeatloop(ctx)\n\t\tclose(done)\n\t}()\n\n\tclose(executor.stopC)\n\t<-done\n}\n\nfunc TestHeartbeatLoop_ContextCancelSendsDrainingHeartbeat(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockShardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\texpectDrainingHeartbeat(t, mockShardDistributorClient)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\texecutor := newTestExecutor(mockShardDistributorClient, nil, mockTimeSource)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tcancel() // cancelling the context\n\n\texecutor.heartbeatloop(ctx)\n}\n\nfunc TestCompareAssignments_Converged(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\texecutor := newTestExecutor(nil, nil, nil)\n\texecutor.metrics = testScope\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\theartbeatAssignments := map[string]*types.ShardAssignment{\n\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t}\n\n\terr := executor.compareAssignments(heartbeatAssignments)\n\n\t// Verify no error is returned when assignments converge\n\tassert.NoError(t, err)\n\n\t// Verify convergence metric was emitted\n\tsnapshot := testScope.Snapshot()\n\tassert.Equal(t, int64(1), snapshot.Counters()[\"test.shard_distributor_executor_assignment_convergence+\"].Value())\n}\n\nfunc TestCompareAssignments_Diverged_MissingShard(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\tshardProcessorMock2 := NewMockShardProcessor(ctrl)\n\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\texecutor := newTestExecutor(nil, nil, nil)\n\texecutor.metrics = testScope\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\texecutor.managedProcessors.Store(\"test-shard-id2\", newManagedProcessor(shardProcessorMock2, processorStateStarted))\n\n\t// Heartbeat only has shard 1, local has both 1 and 2\n\theartbeatAssignments := map[string]*types.ShardAssignment{\n\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t}\n\n\terr := executor.compareAssignments(heartbeatAssignments)\n\n\t// Verify error is returned when local shard is missing from heartbeat\n\tassert.ErrorIs(t, err, ErrAssignmentDivergenceLocalShard)\n\n\t// Verify divergence metric was emitted\n\tsnapshot := testScope.Snapshot()\n\tassert.Equal(t, int64(1), snapshot.Counters()[\"test.shard_distributor_executor_assignment_divergence+\"].Value())\n}\n\nfunc TestCompareAssignments_Diverged_ExtraShard(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\texecutor := newTestExecutor(nil, nil, nil)\n\texecutor.metrics = testScope\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\n\t// Heartbeat has shards 1 and 2, local only has 1\n\theartbeatAssignments := map[string]*types.ShardAssignment{\n\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\"test-shard-id2\": {Status: types.AssignmentStatusREADY},\n\t}\n\n\terr := executor.compareAssignments(heartbeatAssignments)\n\n\t// Verify error is returned when heartbeat has extra shard not in local\n\tassert.ErrorIs(t, err, ErrAssignmentDivergenceHeartbeatShard)\n\n\t// Verify divergence metric was emitted\n\tsnapshot := testScope.Snapshot()\n\tassert.Equal(t, int64(1), snapshot.Counters()[\"test.shard_distributor_executor_assignment_divergence+\"].Value())\n}\n\nfunc TestCompareAssignments_Diverged_WrongStatus(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tshardProcessorMock1 := NewMockShardProcessor(ctrl)\n\n\ttestScope := tally.NewTestScope(\"test\", nil)\n\texecutor := newTestExecutor(nil, nil, nil)\n\texecutor.metrics = testScope\n\n\texecutor.managedProcessors.Store(\"test-shard-id1\", newManagedProcessor(shardProcessorMock1, processorStateStarted))\n\n\t// Heartbeat has shard 1 but with INVALID status\n\theartbeatAssignments := map[string]*types.ShardAssignment{\n\t\t\"test-shard-id1\": {Status: types.AssignmentStatusINVALID},\n\t}\n\n\terr := executor.compareAssignments(heartbeatAssignments)\n\n\t// Verify error is returned when local shard has wrong status in heartbeat\n\tassert.ErrorIs(t, err, ErrAssignmentDivergenceLocalShard)\n\n\t// Verify divergence metric was emitted\n\tsnapshot := testScope.Snapshot()\n\tassert.Equal(t, int64(1), snapshot.Counters()[\"test.shard_distributor_executor_assignment_divergence+\"].Value())\n}\n\nfunc TestGetShardProcess_NonOwnedShard_Fails(t *testing.T) {\n\tcases := map[string]struct {\n\t\tmigrationMode             types.MigrationMode\n\t\texpectedError             error\n\t\tshardsInCache             []string\n\t\tshardsReturnedOnHeartbeat map[string]*types.ShardAssignment\n\t\theartbeatCallsExpected    int\n\t\theartBeatError            error\n\t\tsetupMocks                func(shardProcessorFactory *MockShardProcessorFactory[*MockShardProcessor], shardProcessor *MockShardProcessor)\n\t}{\n\t\t\"empty cache local passthrough\": {\n\t\t\tmigrationMode:          types.MigrationModeLOCALPASSTHROUGH,\n\t\t\texpectedError:          ErrShardProcessNotFound,\n\t\t\tshardsInCache:          []string{},\n\t\t\theartbeatCallsExpected: 0,\n\t\t},\n\t\t\"shard found\": {\n\t\t\tmigrationMode:          types.MigrationModeONBOARDED,\n\t\t\tshardsInCache:          []string{\"test-shard-id1\"},\n\t\t\theartbeatCallsExpected: 0,\n\t\t},\n\t\t\"shard found on heartbeat\": {\n\t\t\tmigrationMode: types.MigrationModeONBOARDED,\n\t\t\tshardsInCache: []string{},\n\t\t\tshardsReturnedOnHeartbeat: map[string]*types.ShardAssignment{\n\t\t\t\t\"test-shard-id1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\theartbeatCallsExpected: 1,\n\t\t\tsetupMocks: func(processorFactory *MockShardProcessorFactory[*MockShardProcessor], processor *MockShardProcessor) {\n\t\t\t\tprocessorFactory.EXPECT().NewShardProcessor(gomock.Any()).Return(processor, nil)\n\t\t\t\tprocessor.EXPECT().Start(gomock.Any())\n\t\t\t},\n\t\t},\n\t\t\"shard not found on heartbeat\": {\n\t\t\tmigrationMode:             types.MigrationModeONBOARDED,\n\t\t\tshardsInCache:             []string{},\n\t\t\tshardsReturnedOnHeartbeat: map[string]*types.ShardAssignment{},\n\t\t\theartbeatCallsExpected:    1,\n\t\t\texpectedError:             ErrShardProcessNotFound,\n\t\t},\n\t\t\"heartbeat error\": {\n\t\t\tmigrationMode:          types.MigrationModeONBOARDED,\n\t\t\tshardsInCache:          []string{},\n\t\t\theartbeatCallsExpected: 1,\n\t\t\texpectedError:          fmt.Errorf(\"heartbeat and assign shards\"),\n\t\t\theartBeatError:         assert.AnError,\n\t\t},\n\t}\n\n\tfor name, tc := range cases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tshardDistributorClient := sharddistributorexecutor.NewMockClient(ctrl)\n\t\t\tshardDistributorClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\t\t\tShardAssignments: tc.shardsReturnedOnHeartbeat,\n\t\t\t\t\tMigrationMode:    tc.migrationMode,\n\t\t\t\t}, tc.heartBeatError).Times(tc.heartbeatCallsExpected)\n\n\t\t\tshardProcessorFactory := NewMockShardProcessorFactory[*MockShardProcessor](ctrl)\n\t\t\tif tc.setupMocks != nil {\n\t\t\t\ttc.setupMocks(shardProcessorFactory, NewMockShardProcessor(ctrl))\n\t\t\t}\n\n\t\t\texecutor := newTestExecutor(shardDistributorClient, shardProcessorFactory, clock.NewMockedTimeSource())\n\t\t\texecutor.setMigrationMode(tc.migrationMode)\n\n\t\t\tfor _, shardID := range tc.shardsInCache {\n\t\t\t\texecutor.managedProcessors.Store(shardID, newManagedProcessor(NewMockShardProcessor(ctrl), processorStateStarted))\n\t\t\t}\n\n\t\t\t_, err := executor.GetShardProcess(context.Background(), \"test-shard-id1\")\n\t\t\tif tc.expectedError != nil {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedError.Error())\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestExecutorMetadata_SetAndGet(t *testing.T) {\n\tmetadata := &syncExecutorMetadata{\n\t\tdata: make(map[string]string),\n\t}\n\n\t// Set some metadata\n\ttestData := map[string]string{\n\t\t\"key1\": \"value1\",\n\t\t\"key2\": \"value2\",\n\t}\n\tmetadata.Set(testData)\n\n\t// Get the metadata\n\tresult := metadata.Get()\n\n\tassert.Equal(t, testData, result)\n}\n\nfunc TestExecutorMetadata_DefensiveCopy(t *testing.T) {\n\tmetadata := &syncExecutorMetadata{\n\t\tdata: map[string]string{\n\t\t\t\"key1\": \"value1\",\n\t\t},\n\t}\n\n\t// Get the metadata\n\tresult := metadata.Get()\n\n\t// Modify the returned map\n\tresult[\"key1\"] = \"modified\"\n\tresult[\"key2\"] = \"new\"\n\n\t// Original metadata should be unchanged\n\toriginal := metadata.Get()\n\tassert.Equal(t, \"value1\", original[\"key1\"])\n\tassert.NotContains(t, original, \"key2\")\n}\n\nfunc TestExecutorMetadata_ConcurrentAccess(t *testing.T) {\n\tmetadata := &syncExecutorMetadata{\n\t\tdata: make(map[string]string),\n\t}\n\n\tconst numGoroutines = 100\n\tconst numOperations = 100\n\n\tvar wg sync.WaitGroup\n\twg.Add(numGoroutines * 2)\n\n\t// Concurrent writers\n\tfor i := 0; i < numGoroutines; i++ {\n\t\tgo func(id int) {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < numOperations; j++ {\n\t\t\t\tmetadata.Set(map[string]string{\n\t\t\t\t\tfmt.Sprintf(\"key-%d\", id): fmt.Sprintf(\"value-%d-%d\", id, j),\n\t\t\t\t})\n\t\t\t}\n\t\t}(i)\n\t}\n\n\t// Concurrent readers\n\tfor i := 0; i < numGoroutines; i++ {\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tfor j := 0; j < numOperations; j++ {\n\t\t\t\t_ = metadata.Get()\n\t\t\t}\n\t\t}()\n\t}\n\n\twg.Wait()\n\n\t// Should not panic and should have some data\n\tresult := metadata.Get()\n\tassert.NotNil(t, result)\n}\n\nfunc TestExecutorMetadata_NilHandling(t *testing.T) {\n\tmetadata := &syncExecutorMetadata{\n\t\tdata: nil,\n\t}\n\n\t// Get should return empty map, not nil\n\tresult := metadata.Get()\n\tassert.NotNil(t, result)\n\tassert.Empty(t, result)\n\n\t// Set with nil should work\n\tmetadata.Set(nil)\n\tresult = metadata.Get()\n\tassert.NotNil(t, result)\n\tassert.Empty(t, result)\n}\n\nfunc TestExecutorMetadata_EmptyMap(t *testing.T) {\n\tmetadata := &syncExecutorMetadata{\n\t\tdata: make(map[string]string),\n\t}\n\n\tresult := metadata.Get()\n\tassert.NotNil(t, result)\n\tassert.Empty(t, result)\n\n\t// Set empty map\n\tmetadata.Set(map[string]string{})\n\tresult = metadata.Get()\n\tassert.NotNil(t, result)\n\tassert.Empty(t, result)\n}\n\nfunc TestShardCleanupLoop(t *testing.T) {\n\ttests := []struct {\n\t\tname                  string\n\t\tttlShard              time.Duration\n\t\tmigrationMode         types.MigrationMode\n\t\tsetupProcessors       func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor\n\t\tadvanceTime           time.Duration\n\t\texpectedShardsDeleted []string\n\t\texpectedShardsKept    []string\n\t}{\n\t\t{\n\t\t\tname:     \"cleanup loop does not run when ttlShard is zero\",\n\t\t\tttlShard: 0,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           10 * time.Second,\n\t\t\texpectedShardsDeleted: []string{},\n\t\t\texpectedShardsKept:    []string{\"shard-1\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"cleanup loop does not run when ttlShard is negative\",\n\t\t\tttlShard: -1 * time.Second,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           10 * time.Second,\n\t\t\texpectedShardsDeleted: []string{},\n\t\t\texpectedShardsKept:    []string{\"shard-1\"},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup removes expired shard in non-onboarded mode\",\n\t\t\tttlShard:      5 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeLOCALPASSTHROUGH,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor.EXPECT().Stop()\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           6 * time.Second,\n\t\t\texpectedShardsDeleted: []string{\"shard-1\"},\n\t\t\texpectedShardsKept:    []string{},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup does not remove non-expired shard\",\n\t\t\tttlShard:      10 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeLOCALPASSTHROUGH,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           5 * time.Second,\n\t\t\texpectedShardsDeleted: []string{},\n\t\t\texpectedShardsKept:    []string{\"shard-1\"},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup sets shard status to DONE and removes from processorsToLastUse in onboarded mode\",\n\t\t\tttlShard:      5 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeONBOARDED,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor.EXPECT().SetShardStatus(types.ShardStatusDONE)\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           6 * time.Second,\n\t\t\texpectedShardsDeleted: []string{}, // Shard is removed from processorsToLastUse but kept in managedProcessors\n\t\t\texpectedShardsKept:    []string{\"shard-1\"},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup removes multiple expired shards in non-onboarded mode\",\n\t\t\tttlShard:      5 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeLOCALPASSTHROUGH,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor1 := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor1.EXPECT().Stop()\n\t\t\t\tprocessor2 := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor2.EXPECT().Stop()\n\t\t\t\tprocessor3 := NewMockShardProcessor(ctrl)\n\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor1, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\n\t\t\t\texecutor.managedProcessors.Store(\"shard-2\", newManagedProcessor(processor2, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-2\", executor.timeSource.Now())\n\n\t\t\t\texecutor.managedProcessors.Store(\"shard-3\", newManagedProcessor(processor3, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-3\", executor.timeSource.Now().Add(3*time.Second))\n\n\t\t\t\treturn []*MockShardProcessor{processor1, processor2, processor3}\n\t\t\t},\n\t\t\tadvanceTime:           6 * time.Second,\n\t\t\texpectedShardsDeleted: []string{\"shard-1\", \"shard-2\"},\n\t\t\texpectedShardsKept:    []string{\"shard-3\"},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup in distributed passthrough mode removes expired shards\",\n\t\t\tttlShard:      5 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor.EXPECT().Stop()\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{processor}\n\t\t\t},\n\t\t\tadvanceTime:           6 * time.Second,\n\t\t\texpectedShardsDeleted: []string{\"shard-1\"},\n\t\t\texpectedShardsKept:    []string{},\n\t\t},\n\t\t{\n\t\t\tname:          \"cleanup removes shard from processorsToLastUse map even if not in managedProcessors\",\n\t\t\tttlShard:      5 * time.Second,\n\t\t\tmigrationMode: types.MigrationModeLOCALPASSTHROUGH,\n\t\t\tsetupProcessors: func(executor *executorImpl[*MockShardProcessor], ctrl *gomock.Controller) []*MockShardProcessor {\n\t\t\t\t// Add to processorsToLastUse but not to managedProcessors\n\t\t\t\texecutor.processorsToLastUse.Store(\"shard-orphan\", executor.timeSource.Now())\n\t\t\t\treturn []*MockShardProcessor{}\n\t\t\t},\n\t\t\tadvanceTime:           6 * time.Second,\n\t\t\texpectedShardsDeleted: []string{},\n\t\t\texpectedShardsKept:    []string{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tdefer goleak.VerifyNone(t)\n\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\texecutor := newTestExecutor(nil, nil, mockTimeSource)\n\t\t\texecutor.ttlShard = tt.ttlShard\n\t\t\tif tt.migrationMode != types.MigrationModeINVALID {\n\t\t\t\texecutor.setMigrationMode(tt.migrationMode)\n\t\t\t}\n\n\t\t\t// Setup processors\n\t\t\ttt.setupProcessors(executor, ctrl)\n\n\t\t\t// Start the executor\n\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\tdefer cancel()\n\n\t\t\tvar wg sync.WaitGroup\n\t\t\twg.Add(1)\n\t\t\tgo func() {\n\t\t\t\tdefer wg.Done()\n\t\t\t\texecutor.shardCleanUpLoop(ctx)\n\t\t\t}()\n\n\t\t\t// For invalid ttlShard, the loop returns immediately\n\t\t\tif tt.ttlShard <= 0 {\n\t\t\t\tcancel()\n\t\t\t\twg.Wait()\n\t\t\t} else {\n\t\t\t\t// Wait for timer to be set\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\n\t\t\t\t// Advance time to trigger cleanup\n\t\t\t\tmockTimeSource.Advance(tt.advanceTime)\n\n\t\t\t\t// Stop the loop before waiting for next timer\n\t\t\t\tcancel()\n\t\t\t\twg.Wait()\n\t\t\t}\n\n\t\t\t// Verify expected shards are deleted\n\t\t\tfor _, shardID := range tt.expectedShardsDeleted {\n\t\t\t\t_, ok := executor.managedProcessors.Load(shardID)\n\t\t\t\tassert.False(t, ok, \"shard %s should be deleted\", shardID)\n\n\t\t\t\t_, ok = executor.processorsToLastUse.Load(shardID)\n\t\t\t\tassert.False(t, ok, \"shard %s should be removed from processorsToLastUse\", shardID)\n\t\t\t}\n\n\t\t\t// Verify expected shards are kept\n\t\t\tfor _, shardID := range tt.expectedShardsKept {\n\t\t\t\t_, ok := executor.managedProcessors.Load(shardID)\n\t\t\t\tassert.True(t, ok, \"shard %s should be kept\", shardID)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardCleanupLoop_StopsOnContextCancel(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\texecutor := newTestExecutor(nil, nil, mockTimeSource)\n\texecutor.ttlShard = 10 * time.Second\n\n\tprocessor := NewMockShardProcessor(ctrl)\n\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tdone := make(chan struct{})\n\tgo func() {\n\t\texecutor.shardCleanUpLoop(ctx)\n\t\tclose(done)\n\t}()\n\n\t// Wait for timer\n\tmockTimeSource.BlockUntil(1)\n\n\t// Cancel context\n\tcancel()\n\n\t// Wait for loop to exit\n\t<-done\n}\n\nfunc TestShardCleanupLoop_StopsOnStopChannel(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\n\tmockTimeSource := clock.NewMockedTimeSource()\n\texecutor := newTestExecutor(nil, nil, mockTimeSource)\n\texecutor.ttlShard = 10 * time.Second\n\n\tprocessor := NewMockShardProcessor(ctrl)\n\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\texecutor.processorsToLastUse.Store(\"shard-1\", executor.timeSource.Now())\n\n\tctx := context.Background()\n\n\tdone := make(chan struct{})\n\tgo func() {\n\t\texecutor.shardCleanUpLoop(ctx)\n\t\tclose(done)\n\t}()\n\n\t// Wait for timer\n\tmockTimeSource.BlockUntil(1)\n\n\t// Close stop channel\n\tclose(executor.stopC)\n\n\t// Wait for loop to exit\n\t<-done\n}\n\nfunc TestHeartbeatLoop_DrainSignal(t *testing.T) {\n\ttests := []struct {\n\t\tname  string\n\t\tsetup func(\n\t\t\tt *testing.T,\n\t\t\tctrl *gomock.Controller,\n\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\tobserver *closeDrainObserver,\n\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t)\n\t}{\n\t\t{\n\t\t\tname: \"drain stops processors and sends draining heartbeat\",\n\t\t\tsetup: func(\n\t\t\t\tt *testing.T,\n\t\t\t\tctrl *gomock.Controller,\n\t\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\t\tobserver *closeDrainObserver,\n\t\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t\t) {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY}).AnyTimes()\n\t\t\t\tprocessor.EXPECT().Stop()\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\n\t\t\t\tmockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\t\t\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\"shard-1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMigrationMode: types.MigrationModeONBOARDED,\n\t\t\t\t\t}, nil)\n\t\t\t\texpectDrainingHeartbeat(t, mockClient)\n\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tdefer cancel()\n\n\t\t\t\tdone := make(chan struct{})\n\t\t\t\tgo func() {\n\t\t\t\t\texecutor.heartbeatloop(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t}()\n\n\t\t\t\t// Trigger a heartbeat\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\t\t\t\tmockTimeSource.Advance(15 * time.Second)\n\t\t\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\t\t\t// Signal drain\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\t_, ok := executor.managedProcessors.Load(\"shard-1\")\n\t\t\t\tassert.False(t, ok, \"shard processor should be stopped after drain\")\n\n\t\t\t\tcancel()\n\t\t\t\t<-done\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"drain then undrain resumes heartbeating\",\n\t\t\tsetup: func(\n\t\t\t\tt *testing.T,\n\t\t\t\tctrl *gomock.Controller,\n\t\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\t\tobserver *closeDrainObserver,\n\t\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t\t) {\n\t\t\t\tprocessor := NewMockShardProcessor(ctrl)\n\t\t\t\tprocessor.EXPECT().GetShardReport().Return(ShardReport{Status: types.ShardStatusREADY}).AnyTimes()\n\t\t\t\tprocessor.EXPECT().Stop()\n\t\t\t\texecutor.managedProcessors.Store(\"shard-1\", newManagedProcessor(processor, processorStateStarted))\n\n\t\t\t\t// Phase 1: active heartbeat, then draining heartbeat on drain\n\t\t\t\t// Phase 2: active heartbeat after undrain, then draining heartbeat on stop\n\t\t\t\tactiveHB := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\t\t\t\tShardAssignments: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\"shard-1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tMigrationMode: types.MigrationModeONBOARDED,\n\t\t\t\t\t}, nil)\n\t\t\t\tdrainingHB := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusDRAINING, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{}, nil\n\t\t\t\t\t}).After(activeHB)\n\t\t\t\tresumedHB := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusACTIVE, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{\n\t\t\t\t\t\t\tShardAssignments: map[string]*types.ShardAssignment{},\n\t\t\t\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).After(drainingHB)\n\t\t\t\tmockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusDRAINING, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{}, nil\n\t\t\t\t\t}).After(resumedHB)\n\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tdefer cancel()\n\n\t\t\t\tdone := make(chan struct{})\n\t\t\t\tgo func() {\n\t\t\t\t\texecutor.heartbeatloop(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t}()\n\n\t\t\t\t// Phase 1: heartbeat, then drain\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\t\t\t\tmockTimeSource.Advance(15 * time.Second)\n\t\t\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\t// Phase 2: undrain resumes, heartbeat again\n\t\t\t\tobserver.SignalUndrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\t\t\t\tmockTimeSource.Advance(15 * time.Second)\n\t\t\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\t\t\tcancel()\n\t\t\t\t<-done\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"drain -> undrain -> drain -> context cancelled\",\n\t\t\tsetup: func(\n\t\t\t\tt *testing.T,\n\t\t\t\tctrl *gomock.Controller,\n\t\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\t\tobserver *closeDrainObserver,\n\t\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t\t) {\n\t\t\t\t// Heartbeat sequence:\n\t\t\t\t// 1. ACTIVE (initial heartbeat)\n\t\t\t\t// 2. DRAINING (drain #1)\n\t\t\t\t// 3. ACTIVE (after undrain #1)\n\t\t\t\t// 4. DRAINING (drain #2)\n\t\t\t\t// 5. DRAINING (context cancelled while waiting for undrain #2)\n\t\t\t\tactiveHB1 := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.ExecutorHeartbeatResponse{\n\t\t\t\t\t\tShardAssignments: map[string]*types.ShardAssignment{},\n\t\t\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t\t\t}, nil)\n\t\t\t\tdrainingHB1 := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusDRAINING, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{}, nil\n\t\t\t\t\t}).After(activeHB1)\n\t\t\t\tactiveHB2 := mockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusACTIVE, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{\n\t\t\t\t\t\t\tShardAssignments: map[string]*types.ShardAssignment{},\n\t\t\t\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).After(drainingHB1)\n\t\t\t\tmockClient.EXPECT().Heartbeat(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, req *types.ExecutorHeartbeatRequest, _ ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, types.ExecutorStatusDRAINING, req.Status)\n\t\t\t\t\t\treturn &types.ExecutorHeartbeatResponse{}, nil\n\t\t\t\t\t}).After(activeHB2)\n\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tdefer cancel()\n\n\t\t\t\tdone := make(chan struct{})\n\t\t\t\tgo func() {\n\t\t\t\t\texecutor.heartbeatloop(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t}()\n\n\t\t\t\t// Initial heartbeat\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\t\t\t\tmockTimeSource.Advance(15 * time.Second)\n\t\t\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\t\t\t// Drain #1\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\t// Undrain #1 - resumes heartbeating\n\t\t\t\tobserver.SignalUndrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\t\t\t\tmockTimeSource.Advance(15 * time.Second)\n\t\t\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\t\t\t// Drain #2\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\t// Cancel while waiting for undrain #2\n\t\t\t\tcancel()\n\t\t\t\t<-done\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"drain then executor stops\",\n\t\t\tsetup: func(\n\t\t\t\tt *testing.T,\n\t\t\t\tctrl *gomock.Controller,\n\t\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\t\tobserver *closeDrainObserver,\n\t\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t\t) {\n\t\t\t\texpectDrainingHeartbeat(t, mockClient)\n\n\t\t\t\tdone := make(chan struct{})\n\t\t\t\tgo func() {\n\t\t\t\t\texecutor.heartbeatloop(context.Background())\n\t\t\t\t\tclose(done)\n\t\t\t\t}()\n\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\tclose(executor.stopC)\n\t\t\t\t<-done\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"drain then context cancelled\",\n\t\t\tsetup: func(\n\t\t\t\tt *testing.T,\n\t\t\t\tctrl *gomock.Controller,\n\t\t\t\texecutor *executorImpl[*MockShardProcessor],\n\t\t\t\tobserver *closeDrainObserver,\n\t\t\t\tmockTimeSource clock.MockedTimeSource,\n\t\t\t\tmockClient *sharddistributorexecutor.MockClient,\n\t\t\t) {\n\t\t\t\texpectDrainingHeartbeat(t, mockClient)\n\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tdefer cancel()\n\n\t\t\t\tdone := make(chan struct{})\n\t\t\t\tgo func() {\n\t\t\t\t\texecutor.heartbeatloop(ctx)\n\t\t\t\t\tclose(done)\n\t\t\t\t}()\n\n\t\t\t\tmockTimeSource.BlockUntil(1)\n\n\t\t\t\tobserver.SignalDrain()\n\t\t\t\ttime.Sleep(50 * time.Millisecond)\n\n\t\t\t\tcancel()\n\t\t\t\t<-done\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tdefer goleak.VerifyNone(t)\n\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockClient := sharddistributorexecutor.NewMockClient(ctrl)\n\t\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\t\tobserver := newCloseDrainObserver()\n\n\t\t\texecutor := newTestExecutor(mockClient, nil, mockTimeSource)\n\t\t\texecutor.drainObserver = observer\n\n\t\t\ttt.setup(t, ctrl, executor, observer, mockTimeSource, mockClient)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: client.go\n//\n// Generated by this command:\n//\n//\tmockgen -package executorclient -source client.go -destination interface_mock.go . ShardProcessorFactory,ShardProcessor,Executor,Client\n//\n\n// Package executorclient is a generated GoMock package.\npackage executorclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// Heartbeat mocks base method.\nfunc (m *MockClient) Heartbeat(arg0 context.Context, arg1 *types.ExecutorHeartbeatRequest, arg2 ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Heartbeat\", varargs...)\n\tret0, _ := ret[0].(*types.ExecutorHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Heartbeat indicates an expected call of Heartbeat.\nfunc (mr *MockClientMockRecorder) Heartbeat(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Heartbeat\", reflect.TypeOf((*MockClient)(nil).Heartbeat), varargs...)\n}\n\n// MockShardProcessor is a mock of ShardProcessor interface.\ntype MockShardProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardProcessorMockRecorder is the mock recorder for MockShardProcessor.\ntype MockShardProcessorMockRecorder struct {\n\tmock *MockShardProcessor\n}\n\n// NewMockShardProcessor creates a new mock instance.\nfunc NewMockShardProcessor(ctrl *gomock.Controller) *MockShardProcessor {\n\tmock := &MockShardProcessor{ctrl: ctrl}\n\tmock.recorder = &MockShardProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardProcessor) EXPECT() *MockShardProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// GetShardReport mocks base method.\nfunc (m *MockShardProcessor) GetShardReport() ShardReport {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardReport\")\n\tret0, _ := ret[0].(ShardReport)\n\treturn ret0\n}\n\n// GetShardReport indicates an expected call of GetShardReport.\nfunc (mr *MockShardProcessorMockRecorder) GetShardReport() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardReport\", reflect.TypeOf((*MockShardProcessor)(nil).GetShardReport))\n}\n\n// SetShardStatus mocks base method.\nfunc (m *MockShardProcessor) SetShardStatus(arg0 types.ShardStatus) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetShardStatus\", arg0)\n}\n\n// SetShardStatus indicates an expected call of SetShardStatus.\nfunc (mr *MockShardProcessorMockRecorder) SetShardStatus(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetShardStatus\", reflect.TypeOf((*MockShardProcessor)(nil).SetShardStatus), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockShardProcessor) Start(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockShardProcessorMockRecorder) Start(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockShardProcessor)(nil).Start), ctx)\n}\n\n// Stop mocks base method.\nfunc (m *MockShardProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockShardProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockShardProcessor)(nil).Stop))\n}\n\n// MockShardProcessorFactory is a mock of ShardProcessorFactory interface.\ntype MockShardProcessorFactory[SP ShardProcessor] struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardProcessorFactoryMockRecorder[SP]\n\tisgomock struct{}\n}\n\n// MockShardProcessorFactoryMockRecorder is the mock recorder for MockShardProcessorFactory.\ntype MockShardProcessorFactoryMockRecorder[SP ShardProcessor] struct {\n\tmock *MockShardProcessorFactory[SP]\n}\n\n// NewMockShardProcessorFactory creates a new mock instance.\nfunc NewMockShardProcessorFactory[SP ShardProcessor](ctrl *gomock.Controller) *MockShardProcessorFactory[SP] {\n\tmock := &MockShardProcessorFactory[SP]{ctrl: ctrl}\n\tmock.recorder = &MockShardProcessorFactoryMockRecorder[SP]{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardProcessorFactory[SP]) EXPECT() *MockShardProcessorFactoryMockRecorder[SP] {\n\treturn m.recorder\n}\n\n// NewShardProcessor mocks base method.\nfunc (m *MockShardProcessorFactory[SP]) NewShardProcessor(shardID string) (SP, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"NewShardProcessor\", shardID)\n\tret0, _ := ret[0].(SP)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// NewShardProcessor indicates an expected call of NewShardProcessor.\nfunc (mr *MockShardProcessorFactoryMockRecorder[SP]) NewShardProcessor(shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"NewShardProcessor\", reflect.TypeOf((*MockShardProcessorFactory[SP])(nil).NewShardProcessor), shardID)\n}\n\n// MockExecutor is a mock of Executor interface.\ntype MockExecutor[SP ShardProcessor] struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutorMockRecorder[SP]\n\tisgomock struct{}\n}\n\n// MockExecutorMockRecorder is the mock recorder for MockExecutor.\ntype MockExecutorMockRecorder[SP ShardProcessor] struct {\n\tmock *MockExecutor[SP]\n}\n\n// NewMockExecutor creates a new mock instance.\nfunc NewMockExecutor[SP ShardProcessor](ctrl *gomock.Controller) *MockExecutor[SP] {\n\tmock := &MockExecutor[SP]{ctrl: ctrl}\n\tmock.recorder = &MockExecutorMockRecorder[SP]{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutor[SP]) EXPECT() *MockExecutorMockRecorder[SP] {\n\treturn m.recorder\n}\n\n// AssignShardsFromLocalLogic mocks base method.\nfunc (m *MockExecutor[SP]) AssignShardsFromLocalLogic(ctx context.Context, shardAssignment map[string]*types.ShardAssignment) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AssignShardsFromLocalLogic\", ctx, shardAssignment)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AssignShardsFromLocalLogic indicates an expected call of AssignShardsFromLocalLogic.\nfunc (mr *MockExecutorMockRecorder[SP]) AssignShardsFromLocalLogic(ctx, shardAssignment any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AssignShardsFromLocalLogic\", reflect.TypeOf((*MockExecutor[SP])(nil).AssignShardsFromLocalLogic), ctx, shardAssignment)\n}\n\n// GetMetadata mocks base method.\nfunc (m *MockExecutor[SP]) GetMetadata() map[string]string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetMetadata\")\n\tret0, _ := ret[0].(map[string]string)\n\treturn ret0\n}\n\n// GetMetadata indicates an expected call of GetMetadata.\nfunc (mr *MockExecutorMockRecorder[SP]) GetMetadata() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetMetadata\", reflect.TypeOf((*MockExecutor[SP])(nil).GetMetadata))\n}\n\n// GetNamespace mocks base method.\nfunc (m *MockExecutor[SP]) GetNamespace() string {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetNamespace\")\n\tret0, _ := ret[0].(string)\n\treturn ret0\n}\n\n// GetNamespace indicates an expected call of GetNamespace.\nfunc (mr *MockExecutorMockRecorder[SP]) GetNamespace() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetNamespace\", reflect.TypeOf((*MockExecutor[SP])(nil).GetNamespace))\n}\n\n// GetShardProcess mocks base method.\nfunc (m *MockExecutor[SP]) GetShardProcess(ctx context.Context, shardID string) (SP, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardProcess\", ctx, shardID)\n\tret0, _ := ret[0].(SP)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShardProcess indicates an expected call of GetShardProcess.\nfunc (mr *MockExecutorMockRecorder[SP]) GetShardProcess(ctx, shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardProcess\", reflect.TypeOf((*MockExecutor[SP])(nil).GetShardProcess), ctx, shardID)\n}\n\n// IsOnboardedToSD mocks base method.\nfunc (m *MockExecutor[SP]) IsOnboardedToSD() bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"IsOnboardedToSD\")\n\tret0, _ := ret[0].(bool)\n\treturn ret0\n}\n\n// IsOnboardedToSD indicates an expected call of IsOnboardedToSD.\nfunc (mr *MockExecutorMockRecorder[SP]) IsOnboardedToSD() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"IsOnboardedToSD\", reflect.TypeOf((*MockExecutor[SP])(nil).IsOnboardedToSD))\n}\n\n// RemoveShardsFromLocalLogic mocks base method.\nfunc (m *MockExecutor[SP]) RemoveShardsFromLocalLogic(shardIDs []string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RemoveShardsFromLocalLogic\", shardIDs)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RemoveShardsFromLocalLogic indicates an expected call of RemoveShardsFromLocalLogic.\nfunc (mr *MockExecutorMockRecorder[SP]) RemoveShardsFromLocalLogic(shardIDs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RemoveShardsFromLocalLogic\", reflect.TypeOf((*MockExecutor[SP])(nil).RemoveShardsFromLocalLogic), shardIDs)\n}\n\n// SetMetadata mocks base method.\nfunc (m *MockExecutor[SP]) SetMetadata(metadata map[string]string) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"SetMetadata\", metadata)\n}\n\n// SetMetadata indicates an expected call of SetMetadata.\nfunc (mr *MockExecutorMockRecorder[SP]) SetMetadata(metadata any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SetMetadata\", reflect.TypeOf((*MockExecutor[SP])(nil).SetMetadata), metadata)\n}\n\n// Start mocks base method.\nfunc (m *MockExecutor[SP]) Start(ctx context.Context) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\", ctx)\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockExecutorMockRecorder[SP]) Start(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockExecutor[SP])(nil).Start), ctx)\n}\n\n// Stop mocks base method.\nfunc (m *MockExecutor[SP]) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockExecutorMockRecorder[SP]) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockExecutor[SP])(nil).Stop))\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/meteredClientDecorater.go",
    "content": "package executorclient\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributorexecutor\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/executorclient/metricsconstants\"\n)\n\n// TODO: consider using gowrap to generate this code\ntype meteredShardDistributorExecutorClient struct {\n\tclient       sharddistributorexecutor.Client\n\tmetricsScope tally.Scope\n}\n\n// NewShardDistributorExecutorClient creates a new instance of sharddistributorexecutorClient with retry policy\nfunc NewMeteredShardDistributorExecutorClient(client sharddistributorexecutor.Client, metricsScope tally.Scope) sharddistributorexecutor.Client {\n\treturn &meteredShardDistributorExecutorClient{\n\t\tclient:       client,\n\t\tmetricsScope: metricsScope,\n\t}\n}\n\nfunc (c *meteredShardDistributorExecutorClient) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest, p1 ...yarpc.CallOption) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tvar scope tally.Scope\n\tscope = c.metricsScope.Tagged(map[string]string{\n\t\tmetrics.OperationTagName: metricsconstants.ShardDistributorExecutorHeartbeatOperationTagName,\n\t})\n\n\tscope.Counter(metricsconstants.ShardDistributorExecutorClientRequests).Inc(1)\n\n\tsw := scope.Timer(metricsconstants.ShardDistributorExecutorClientLatency).Start()\n\tep2, err = c.client.Heartbeat(ctx, ep1, p1...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.Counter(metricsconstants.ShardDistributorExecutorClientFailures).Inc(1)\n\t}\n\treturn ep2, err\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/metricsconstants/metrics.go",
    "content": "package metricsconstants\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n)\n\nconst (\n\t// Operation tag names for ShardDistributorExecutor metrics\n\tShardDistributorExecutorOperationTagName          = \"ShardDistributorExecutor\"\n\tShardDistributorExecutorHeartbeatOperationTagName = \"ShardDistributorExecutorHeartbeat\"\n\n\t// Counter metrics\n\tShardDistributorExecutorHeartbeatSkipped          = \"shard_distributor_executor_heartbeat_skipped\"\n\tShardDistributorExecutorShardsStarted             = \"shard_distributor_executor_shards_started\"\n\tShardDistributorExecutorShardsStopped             = \"shard_distributor_executor_shards_stopped\"\n\tShardDistributorExecutorProcessorCreationFailures = \"shard_distributor_executor_processor_creation_failures\"\n\tShardDistributorExecutorClientRequests            = \"shard_distributor_executor_client_requests\"\n\tShardDistributorExecutorClientFailures            = \"shard_distributor_executor_client_failures\"\n\tShardDistributorExecutorAssignmentDivergence      = \"shard_distributor_executor_assignment_divergence\"\n\tShardDistributorExecutorAssignmentConvergence     = \"shard_distributor_executor_assignment_convergence\"\n\n\t// Gauge metrics\n\tShardDistributorExecutorOwnedShards = \"shard_distributor_executor_owned_shards\"\n\n\t// Histogram/Timer metrics\n\tShardDistributorExecutorAssignLoopLatency = \"shard_distributor_executor_assign_loop_latency\"\n\tShardDistributorExecutorClientLatency     = \"shard_distributor_executor_client_latency\"\n)\n\nvar (\n\t// Histogram buckets for ShardDistributorExecutor metrics\n\tShardDistributorExecutorAssignLoopLatencyBuckets = tally.DurationBuckets([]time.Duration{\n\t\t0,\n\t\t1 * time.Millisecond,\n\t\t5 * time.Millisecond,\n\t\t10 * time.Millisecond,\n\t\t25 * time.Millisecond,\n\t\t50 * time.Millisecond,\n\t\t100 * time.Millisecond,\n\t\t200 * time.Millisecond,\n\t\t500 * time.Millisecond,\n\t\t1 * time.Second,\n\t\t2 * time.Second,\n\t\t5 * time.Second,\n\t\t10 * time.Second,\n\t\t20 * time.Second,\n\t\t60 * time.Second,\n\t})\n)\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/noop.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage executorclient\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// noopExecutor is an Executor implementation used when no shard-distributor\n// config is provided. It acts as if all task lists are locally assigned\n// (local-passthrough behaviour): it never contacts the shard distributor,\n// never manages shard processors, and reports itself as not onboarded to SD.\ntype noopExecutor[SP ShardProcessor] struct{}\n\n// NewNoopExecutor returns an Executor that is a no-op. Use this when the\n// shard-distributor config is absent so that the rest of the matching engine\n// can operate using the local hash-ring exclusively.\nfunc NewNoopExecutor[SP ShardProcessor]() Executor[SP] {\n\treturn &noopExecutor[SP]{}\n}\n\nfunc (e *noopExecutor[SP]) Start(_ context.Context)         {}\nfunc (e *noopExecutor[SP]) Stop()                           {}\nfunc (e *noopExecutor[SP]) GetNamespace() string            { return \"\" }\nfunc (e *noopExecutor[SP]) IsOnboardedToSD() bool           { return false }\nfunc (e *noopExecutor[SP]) SetMetadata(_ map[string]string) {}\nfunc (e *noopExecutor[SP]) GetMetadata() map[string]string  { return nil }\n\n// GetShardProcess always returns ErrShardProcessNotFound so that callers\n// fall back to local hash-ring ownership checks.\nfunc (e *noopExecutor[SP]) GetShardProcess(_ context.Context, _ string) (SP, error) {\n\tvar zero SP\n\treturn zero, ErrShardProcessNotFound\n}\n\n// AssignShardsFromLocalLogic is a no-op: without SD there is no shard\n// assignment to manage.\nfunc (e *noopExecutor[SP]) AssignShardsFromLocalLogic(_ context.Context, _ map[string]*types.ShardAssignment) error {\n\treturn nil\n}\n\n// RemoveShardsFromLocalLogic is a no-op: without SD there is no shard\n// assignment to manage.\nfunc (e *noopExecutor[SP]) RemoveShardsFromLocalLogic(_ []string) error {\n\treturn nil\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/noop_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage executorclient\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNoopExecutor(t *testing.T) {\n\texec := NewNoopExecutor[*MockShardProcessor]()\n\n\tt.Run(\"Start and Stop are no-ops\", func(t *testing.T) {\n\t\t// should not panic\n\t\texec.Start(context.Background())\n\t\texec.Stop()\n\t})\n\n\tt.Run(\"GetNamespace returns empty string\", func(t *testing.T) {\n\t\tassert.Equal(t, \"\", exec.GetNamespace())\n\t})\n\n\tt.Run(\"IsOnboardedToSD returns false\", func(t *testing.T) {\n\t\tassert.False(t, exec.IsOnboardedToSD())\n\t})\n\n\tt.Run(\"SetMetadata and GetMetadata are no-ops\", func(t *testing.T) {\n\t\texec.SetMetadata(map[string]string{\"key\": \"value\"})\n\t\tassert.Nil(t, exec.GetMetadata())\n\t})\n\n\tt.Run(\"GetShardProcess returns ErrShardProcessNotFound\", func(t *testing.T) {\n\t\tsp, err := exec.GetShardProcess(context.Background(), \"any-shard\")\n\t\tassert.Nil(t, sp)\n\t\tassert.ErrorIs(t, err, ErrShardProcessNotFound)\n\t})\n\n\tt.Run(\"AssignShardsFromLocalLogic is a no-op\", func(t *testing.T) {\n\t\terr := exec.AssignShardsFromLocalLogic(context.Background(), map[string]*types.ShardAssignment{\n\t\t\t\"shard-1\": {},\n\t\t})\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"RemoveShardsFromLocalLogic is a no-op\", func(t *testing.T) {\n\t\terr := exec.RemoveShardsFromLocalLogic([]string{\"shard-1\"})\n\t\tassert.NoError(t, err)\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/syncgeneric/map.go",
    "content": "package syncgeneric\n\nimport \"sync\"\n\n// Map is a wrapper around sync.Map that provides a more convenient, generic,\n// and type-safe API.\ntype Map[K comparable, V any] struct {\n\tcontents sync.Map\n}\n\n// Load returns the value stored in the map for a key, or nil if no\n// value is present.\n// The ok result indicates whether value was found in the map.\nfunc (m *Map[K, V]) Load(key K) (V, bool) {\n\tval, ok := m.contents.Load(key)\n\tif !ok {\n\t\tvar zero V\n\t\treturn zero, false\n\t}\n\treturn val.(V), true\n}\n\n// Store sets the value for a key.\nfunc (m *Map[K, V]) Store(key K, value V) {\n\tm.contents.Store(key, value)\n}\n\n// Delete deletes the value for a key.\nfunc (m *Map[K, V]) Delete(key K) {\n\tm.contents.Delete(key)\n}\n\n// Range calls f sequentially for each key and value present in the map.\n// If f returns false, range stops the iteration.\n//\n// Range does not necessarily correspond to any consistent snapshot of the Map's\n// contents: no key will be visited more than once, but if the value for any key\n// is stored or deleted concurrently (including by f), Range may reflect any\n// mapping for that key from any point during the Range call. Range does not\n// block other methods on the receiver; even f itself may call any method on m.\n//\n// Range may be O(N) with the number of elements in the map even if f returns\n// false after a constant number of calls.\nfunc (m *Map[K, V]) Range(f func(key K, value V) bool) {\n\tm.contents.Range(func(key, value interface{}) bool {\n\t\treturn f(key.(K), value.(V))\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/syncgeneric/map_test.go",
    "content": "package syncgeneric\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\ntype testKey struct {\n\tstr string\n\ti   int\n}\n\ntype testValue struct {\n\tm map[string]string\n}\n\nvar (\n\ttestKey1   = &testKey{str: \"key1\", i: 1}\n\ttestKey2   = &testKey{str: \"key2\", i: 2}\n\ttestValue1 = &testValue{m: map[string]string{\"key1\": \"value1\"}}\n\ttestValue2 = &testValue{m: map[string]string{\"key2\": \"value2\"}}\n)\n\nfunc testMap(t *testing.T) *Map[*testKey, *testValue] {\n\n\t// Create the type-safe map\n\tm := &Map[*testKey, *testValue]{}\n\n\t// Insert some data into the underlying sync.Map\n\tm.contents.Store(testKey1, testValue1)\n\tm.contents.Store(testKey2, testValue2)\n\tloaded, ok := m.contents.Load(testKey1)\n\trequire.True(t, ok)\n\trequire.Equal(t, testValue1, loaded)\n\tloaded, ok = m.contents.Load(testKey2)\n\trequire.True(t, ok)\n\trequire.Equal(t, testValue2, loaded)\n\n\treturn m\n}\n\nfunc TestStoreAndLoad(t *testing.T) {\n\tcases := map[string]struct {\n\t\tkey   *testKey\n\t\tvalue *testValue\n\t}{\n\t\t\"new key\":      {key: &testKey{str: \"key3\", i: 1}, value: &testValue{m: map[string]string{\"key3\": \"value3\"}}},\n\t\t\"existing key\": {key: testKey1, value: testValue1},\n\t\t\"nil key\":      {key: nil, value: &testValue{m: map[string]string{\"key1\": \"value1-updated\"}}},\n\t\t\"nil value\":    {key: &testKey{str: \"key1\", i: 1}, value: nil},\n\t}\n\n\tfor name, c := range cases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tm := testMap(t)\n\t\t\tm.Store(c.key, c.value)\n\t\t\tloaded, ok := m.Load(c.key)\n\t\t\tassert.True(t, ok)\n\t\t\tassert.Equal(t, c.value, loaded)\n\t\t})\n\t}\n}\n\nfunc TestLoad(t *testing.T) {\n\tt.Run(\"load existing key\", func(t *testing.T) {\n\t\tm := testMap(t)\n\t\tloaded, ok := m.Load(testKey1)\n\t\tassert.True(t, ok)\n\t\tassert.Equal(t, testValue1, loaded)\n\t})\n\tt.Run(\"load non-existing key\", func(t *testing.T) {\n\t\tm := testMap(t)\n\t\tloaded, ok := m.Load(&testKey{str: \"key3\", i: 1})\n\t\tassert.False(t, ok)\n\t\tassert.Nil(t, loaded)\n\t})\n\tt.Run(\"load non-existing key with zero value\", func(t *testing.T) {\n\t\tm := Map[int, string]{}\n\t\tloaded, ok := m.Load(0)\n\t\tassert.False(t, ok)\n\t\t// Should return the zero value of the type\n\t\tassert.Equal(t, \"\", loaded)\n\t})\n\tt.Run(\"load non-existing key with other zero value\", func(t *testing.T) {\n\t\ttype s struct {\n\t\t\t_ string\n\t\t\t_ int\n\t\t\t_ *int\n\t\t}\n\t\tm := Map[int, s]{}\n\t\tloaded, ok := m.Load(0)\n\t\tassert.False(t, ok)\n\t\tassert.Equal(t, s{}, loaded)\n\t})\n}\n\nfunc TestDelete(t *testing.T) {\n\tm := testMap(t)\n\t// testKey1 is in the map\n\tloaded, ok := m.Load(testKey1)\n\tassert.True(t, ok)\n\tassert.Equal(t, testValue1, loaded)\n\n\t// delete testKey1\n\tm.Delete(testKey1)\n\n\t// testKey1 is no longer in the map\n\tloaded, ok = m.Load(testKey1)\n\tassert.False(t, ok)\n\tassert.Nil(t, loaded)\n}\n\nfunc TestRange(t *testing.T) {\n\tm := testMap(t)\n\n\t// We have 2 keys in the map\n\tcount := 0\n\tm.Range(func(key *testKey, value *testValue) bool {\n\t\tcount++\n\t\treturn true\n\t})\n\tassert.Equal(t, 2, count)\n}\n"
  },
  {
    "path": "service/sharddistributor/client/executorclient/yarpc_client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/.gen/proto/sharddistributor/v1 (interfaces: ShardDistributorExecutorAPIYARPCClient)\n//\n// Generated by this command:\n//\n//\tmockgen -package executorclient -destination yarpc_client_mock.go github.com/uber/cadence/.gen/proto/sharddistributor/v1 ShardDistributorExecutorAPIYARPCClient\n//\n\n// Package executorclient is a generated GoMock package.\npackage executorclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\tyarpc \"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n)\n\n// MockShardDistributorExecutorAPIYARPCClient is a mock of ShardDistributorExecutorAPIYARPCClient interface.\ntype MockShardDistributorExecutorAPIYARPCClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockShardDistributorExecutorAPIYARPCClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockShardDistributorExecutorAPIYARPCClientMockRecorder is the mock recorder for MockShardDistributorExecutorAPIYARPCClient.\ntype MockShardDistributorExecutorAPIYARPCClientMockRecorder struct {\n\tmock *MockShardDistributorExecutorAPIYARPCClient\n}\n\n// NewMockShardDistributorExecutorAPIYARPCClient creates a new mock instance.\nfunc NewMockShardDistributorExecutorAPIYARPCClient(ctrl *gomock.Controller) *MockShardDistributorExecutorAPIYARPCClient {\n\tmock := &MockShardDistributorExecutorAPIYARPCClient{ctrl: ctrl}\n\tmock.recorder = &MockShardDistributorExecutorAPIYARPCClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockShardDistributorExecutorAPIYARPCClient) EXPECT() *MockShardDistributorExecutorAPIYARPCClientMockRecorder {\n\treturn m.recorder\n}\n\n// Heartbeat mocks base method.\nfunc (m *MockShardDistributorExecutorAPIYARPCClient) Heartbeat(arg0 context.Context, arg1 *sharddistributorv1.HeartbeatRequest, arg2 ...yarpc.CallOption) (*sharddistributorv1.HeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{arg0, arg1}\n\tfor _, a := range arg2 {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Heartbeat\", varargs...)\n\tret0, _ := ret[0].(*sharddistributorv1.HeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Heartbeat indicates an expected call of Heartbeat.\nfunc (mr *MockShardDistributorExecutorAPIYARPCClientMockRecorder) Heartbeat(arg0, arg1 any, arg2 ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{arg0, arg1}, arg2...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Heartbeat\", reflect.TypeOf((*MockShardDistributorExecutorAPIYARPCClient)(nil).Heartbeat), varargs...)\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/client.go",
    "content": "package spectatorclient\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\tcsync \"github.com/uber/cadence/service/sharddistributor/client/spectatorclient/sync\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interface_mock.go . Spectator\n\n// EnabledFunc is a function that returns true if the spectator is enabled\n// This is used to disable the spectator in the case of a migration\ntype EnabledFunc func() bool\n\ntype Spectators struct {\n\tspectators map[string]Spectator\n}\n\nfunc (s *Spectators) ForNamespace(namespace string) (Spectator, error) {\n\tspectator, ok := s.spectators[namespace]\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"spectator not found for namespace %s\", namespace)\n\t}\n\treturn spectator, nil\n}\n\nfunc (s *Spectators) Start(ctx context.Context) error {\n\tfor namespace, spectator := range s.spectators {\n\t\tif err := spectator.Start(ctx); err != nil {\n\t\t\treturn fmt.Errorf(\"start spectator for namespace %s: %w\", namespace, err)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (s *Spectators) Stop() {\n\tfor _, spectator := range s.spectators {\n\t\tspectator.Stop()\n\t}\n}\n\nfunc NewSpectators(params Params) (*Spectators, error) {\n\tspectators := make(map[string]Spectator)\n\tfor _, namespace := range params.Config.Namespaces {\n\t\tspectator, err := NewSpectatorWithNamespace(params, namespace.Namespace)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"create spectator for namespace %s: %w\", namespace.Namespace, err)\n\t\t}\n\n\t\tspectators[namespace.Namespace] = spectator\n\t}\n\treturn &Spectators{spectators: spectators}, nil\n}\n\ntype Spectator interface {\n\tStart(ctx context.Context) error\n\tStop()\n\n\t// GetShardOwner returns the owner of a shard\n\tGetShardOwner(ctx context.Context, shardKey string) (*ShardOwner, error)\n}\n\ntype Params struct {\n\tfx.In\n\n\tClient       sharddistributor.Client\n\tMetricsScope tally.Scope\n\tLogger       log.Logger\n\tConfig       clientcommon.Config\n\tTimeSource   clock.TimeSource\n\n\tEnabled EnabledFunc `optional:\"true\"`\n}\n\n// NewSpectatorWithNamespace creates a spectator for a specific namespace\nfunc NewSpectatorWithNamespace(params Params, namespace string) (Spectator, error) {\n\treturn newSpectatorImpl(params, namespace)\n}\n\n// NewSpectator creates a spectator for the single namespace in config\nfunc NewSpectator(params Params) (Spectator, error) {\n\tcfg, err := params.Config.GetSingleConfig()\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn newSpectatorImpl(params, cfg.Namespace)\n}\n\nfunc newSpectatorImpl(params Params, namespace string) (Spectator, error) {\n\t// Get config for the specified namespace\n\tnamespaceConfig, err := params.Config.GetConfigForNamespace(namespace)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get config for namespace %s: %w\", namespace, err)\n\t}\n\n\treturn newSpectatorWithConfig(params, namespaceConfig)\n}\n\nfunc newSpectatorWithConfig(params Params, namespaceConfig *clientcommon.NamespaceConfig) (Spectator, error) {\n\tenabled := params.Enabled\n\tif enabled == nil {\n\t\tenabled = func() bool { return true }\n\t}\n\n\timpl := &spectatorImpl{\n\t\tnamespace:        namespaceConfig.Namespace,\n\t\tconfig:           *namespaceConfig,\n\t\tclient:           params.Client,\n\t\tlogger:           params.Logger,\n\t\tscope:            params.MetricsScope,\n\t\ttimeSource:       params.TimeSource,\n\t\tfirstStateSignal: csync.NewResettableSignal(),\n\t\tenabled:          enabled,\n\t}\n\n\treturn impl, nil\n}\n\n// Module creates a spectator module using auto-selection (single namespace only)\nfunc Module() fx.Option {\n\treturn fx.Module(\"shard-distributor-spectator-client\",\n\t\tfx.Provide(NewSpectators),\n\t\tfx.Invoke(func(spectators *Spectators, lc fx.Lifecycle) {\n\t\t\tlc.Append(fx.StartStopHook(spectators.Start, spectators.Stop))\n\t\t}),\n\t)\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/clientimpl.go",
    "content": "package spectatorclient\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\tcsync \"github.com/uber/cadence/service/sharddistributor/client/spectatorclient/sync\"\n)\n\n// stateFn represents a state in the election state machine.\n// Each state is a function that blocks until a transition occurs\n// and returns the next state function, or nil to stop.\n// Note this is a recursive type definition.\ntype stateFn func(ctx context.Context) stateFn\n\nconst (\n\tstreamRetryInterval    = 1 * time.Second\n\tstreamRetryJitterCoeff = 0.1 // 10% jitter (900ms - 1100ms)\n)\n\n// ShardOwner contains information about the executor that owns a shard\ntype ShardOwner struct {\n\tExecutorID string\n\tMetadata   map[string]string\n}\n\ntype spectatorImpl struct {\n\tnamespace  string\n\tenabled    EnabledFunc\n\tconfig     clientcommon.NamespaceConfig\n\tclient     sharddistributor.Client\n\tscope      tally.Scope\n\tlogger     log.Logger\n\ttimeSource clock.TimeSource\n\tstream     sharddistributor.WatchNamespaceStateClient\n\n\tcancel context.CancelFunc\n\tstopWG sync.WaitGroup\n\n\t// State storage with lock for thread-safe access\n\t// Map from shard ID to shard owner (executor ID + metadata)\n\tstateMu      sync.RWMutex\n\tshardToOwner map[string]*ShardOwner\n\n\t// Signal to notify when first state is received\n\tfirstStateSignal *csync.ResettableSignal\n}\n\nfunc (s *spectatorImpl) Start(ctx context.Context) error {\n\t// Create a cancellable context for the lifetime of the spectator\n\t// Use context.WithoutCancel to inherit values but not cancellation from fx lifecycle ctx\n\tctx, s.cancel = context.WithCancel(context.WithoutCancel(ctx))\n\n\ts.stopWG.Add(1)\n\tgo func() {\n\t\tdefer s.stopWG.Done()\n\t\ts.watchLoop(ctx)\n\t}()\n\n\treturn nil\n}\n\nfunc (s *spectatorImpl) Stop() {\n\tif s.cancel != nil {\n\t\ts.cancel()\n\t}\n\t// Close the firstStateSignal to unblock any goroutines waiting for first state\n\ts.firstStateSignal.Done()\n\ts.stopWG.Wait()\n}\n\nfunc (s *spectatorImpl) watchLoop(ctx context.Context) {\n\tdefer s.logger.Info(\"Shutting down, stopping watch loop\", tag.ShardNamespace(s.namespace))\n\ts.logger.Info(\"Starting watch loop for namespace\", tag.ShardNamespace(s.namespace))\n\n\tvar state stateFn\n\tif s.enabled() {\n\t\tstate = s.connectState\n\t} else {\n\t\tstate = s.disabledState\n\t}\n\n\tfor state != nil {\n\t\tstate = state(ctx)\n\t}\n}\n\nfunc (s *spectatorImpl) connectState(ctx context.Context) stateFn {\n\tdefer s.logger.Info(\"Exiting connect state\", tag.ShardNamespace(s.namespace))\n\ts.logger.Info(\"Starting connect state for namespace\", tag.ShardNamespace(s.namespace))\n\n\tif !s.enabled() {\n\t\treturn s.disabledState\n\t}\n\n\tstream, err := s.client.WatchNamespaceState(ctx, &types.WatchNamespaceStateRequest{\n\t\tNamespace: s.namespace,\n\t})\n\n\tif err != nil {\n\t\tif ctx.Err() != nil {\n\t\t\treturn nil\n\t\t}\n\n\t\ts.logger.Error(\"Failed to create stream, retrying\", tag.Error(err), tag.ShardNamespace(s.namespace))\n\t\tif err := s.timeSource.SleepWithContext(ctx, backoff.JitDuration(streamRetryInterval, streamRetryJitterCoeff)); err != nil {\n\t\t\treturn nil\n\t\t}\n\t\treturn s.connectState\n\t}\n\n\ts.stream = stream\n\n\treturn s.enabledState\n}\n\nfunc (s *spectatorImpl) enabledState(ctx context.Context) stateFn {\n\tdefer s.logger.Info(\"Exiting enabled state\", tag.ShardNamespace(s.namespace))\n\tdefer func() {\n\t\tif err := s.stream.CloseSend(); err != nil {\n\t\t\ts.logger.Warn(\"Failed to close stream\", tag.Error(err), tag.ShardNamespace(s.namespace))\n\t\t}\n\t}()\n\n\ts.logger.Info(\"Starting enabled state for namespace\", tag.ShardNamespace(s.namespace))\n\n\tfor {\n\t\tif !s.enabled() {\n\t\t\treturn s.disabledState\n\t\t}\n\n\t\tresponse, err := s.stream.Recv()\n\t\tif err != nil {\n\t\t\tif ctx.Err() != nil {\n\t\t\t\ts.logger.Info(\"Recv interrupted by client shutdown\", tag.ShardNamespace(s.namespace))\n\t\t\t\treturn nil\n\t\t\t}\n\n\t\t\ts.logger.Warn(\"Stream error (server issue), will reconnect\", tag.Error(err), tag.ShardNamespace(s.namespace))\n\t\t\tif err := s.timeSource.SleepWithContext(ctx, backoff.JitDuration(streamRetryInterval, streamRetryJitterCoeff)); err != nil {\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn s.connectState\n\n\t\t}\n\n\t\t// Process the response\n\t\ts.handleResponse(response)\n\t}\n}\n\nfunc (s *spectatorImpl) disabledState(ctx context.Context) stateFn {\n\tdefer s.logger.Info(\"Exiting disabled state\", tag.ShardNamespace(s.namespace))\n\ts.logger.Info(\"Starting disabled state for namespace\", tag.ShardNamespace(s.namespace))\n\t// We reset the first state signal to ensure we wait for the first state to be received when we re-enable.\n\ts.firstStateSignal.Reset()\n\n\tfor {\n\t\t// Sleep for a short period of time before checking again.\n\t\t// If the context is cancelled, we return nil to exit the loop.\n\t\tif err := s.timeSource.SleepWithContext(ctx, backoff.JitDuration(streamRetryInterval, streamRetryJitterCoeff)); err != nil {\n\t\t\treturn nil\n\t\t}\n\t\tif s.enabled() {\n\t\t\treturn s.connectState\n\t\t}\n\t}\n}\n\nfunc (s *spectatorImpl) handleResponse(response *types.WatchNamespaceStateResponse) {\n\t// Build inverted map: shard ID -> shard owner (executor ID + metadata)\n\tshardToOwner := make(map[string]*ShardOwner)\n\tfor _, executor := range response.Executors {\n\t\towner := &ShardOwner{\n\t\t\tExecutorID: executor.ExecutorID,\n\t\t\tMetadata:   executor.Metadata,\n\t\t}\n\t\tfor _, shard := range executor.AssignedShards {\n\t\t\tshardToOwner[shard.ShardKey] = owner\n\t\t}\n\t}\n\n\ts.stateMu.Lock()\n\ts.shardToOwner = shardToOwner\n\ts.stateMu.Unlock()\n\n\t// Signal that first state has been received - this function is free to call\n\t// multiple times.\n\ts.firstStateSignal.Done()\n\n\ts.logger.Debug(\"Received namespace state update\",\n\t\ttag.ShardNamespace(s.namespace),\n\t\ttag.Counter(len(response.Executors)))\n}\n\n// GetShardOwner returns the full owner information including metadata for a given shard.\n// It first waits for the initial state to be received, then checks the cache.\n// If not found in cache, it falls back to querying the shard distributor directly.\nfunc (s *spectatorImpl) GetShardOwner(ctx context.Context, shardKey string) (*ShardOwner, error) {\n\t// Wait for first state to be received to avoid flooding shard distributor on startup\n\tif err := s.firstStateSignal.Wait(ctx); err != nil {\n\t\treturn nil, fmt.Errorf(\"wait for first state: %w\", err)\n\t}\n\n\t// Check cache first\n\ts.stateMu.RLock()\n\towner := s.shardToOwner[shardKey]\n\ts.stateMu.RUnlock()\n\n\tif owner != nil {\n\t\treturn owner, nil\n\t}\n\n\t// Cache miss - fall back to RPC call\n\ts.logger.Debug(\"Shard not found in cache, querying shard distributor\",\n\t\ttag.ShardKey(shardKey),\n\t\ttag.ShardNamespace(s.namespace))\n\n\tresponse, err := s.client.GetShardOwner(ctx, &types.GetShardOwnerRequest{\n\t\tNamespace: s.namespace,\n\t\tShardKey:  shardKey,\n\t})\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get shard owner from shard distributor: %w\", err)\n\t}\n\n\treturn &ShardOwner{\n\t\tExecutorID: response.Owner,\n\t\tMetadata:   response.Metadata,\n\t}, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/clientimpl_test.go",
    "content": "package spectatorclient\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\tcsync \"github.com/uber/cadence/service/sharddistributor/client/spectatorclient/sync\"\n)\n\nfunc TestWatchLoopBasicFlow(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockClient := sharddistributor.NewMockClient(ctrl)\n\tmockStream := sharddistributor.NewMockWatchNamespaceStateClient(ctrl)\n\n\t// Create a context to control when the mock stream should unblock\n\tstreamCtx, cancelStream := context.WithCancel(context.Background())\n\n\tspectator := &spectatorImpl{\n\t\tnamespace:        \"test-ns\",\n\t\tclient:           mockClient,\n\t\tlogger:           log.NewNoop(),\n\t\tscope:            tally.NoopScope,\n\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\tfirstStateSignal: csync.NewResettableSignal(),\n\t\tenabled:          func() bool { return true },\n\t}\n\n\t// Expect stream creation\n\tmockClient.EXPECT().\n\t\tWatchNamespaceState(gomock.Any(), &types.WatchNamespaceStateRequest{Namespace: \"test-ns\"}).\n\t\tReturn(mockStream, nil)\n\n\t// First Recv returns state\n\tmockStream.EXPECT().Recv().Return(&types.WatchNamespaceStateResponse{\n\t\tExecutors: []*types.ExecutorShardAssignment{\n\t\t\t{\n\t\t\t\tExecutorID: \"executor-1\",\n\t\t\t\tMetadata: map[string]string{\n\t\t\t\t\t\"grpc_address\": \"127.0.0.1:7953\",\n\t\t\t\t},\n\t\t\t\tAssignedShards: []*types.Shard{\n\t\t\t\t\t{ShardKey: \"shard-1\"},\n\t\t\t\t\t{ShardKey: \"shard-2\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil)\n\n\t// Second Recv blocks until shutdown\n\tmockStream.EXPECT().Recv().DoAndReturn(func(...interface{}) (*types.WatchNamespaceStateResponse, error) {\n\t\t// Wait for context to be done\n\t\t<-streamCtx.Done()\n\t\treturn nil, streamCtx.Err()\n\t})\n\n\tmockStream.EXPECT().CloseSend().Return(nil)\n\n\tctx := context.Background()\n\terr := spectator.Start(ctx)\n\trequire.NoError(t, err)\n\tdefer func() {\n\t\tcancelStream()\n\t\tspectator.Stop()\n\t}()\n\n\t// Wait for first state\n\trequire.NoError(t, spectator.firstStateSignal.Wait(context.Background()))\n\n\t// Query shard owner\n\towner, err := spectator.GetShardOwner(context.Background(), \"shard-1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-1\", owner.ExecutorID)\n\tassert.Equal(t, \"127.0.0.1:7953\", owner.Metadata[\"grpc_address\"])\n\n\towner, err = spectator.GetShardOwner(context.Background(), \"shard-2\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-1\", owner.ExecutorID)\n}\n\nfunc TestGetShardOwner_CacheMiss_FallbackToRPC(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockClient := sharddistributor.NewMockClient(ctrl)\n\tmockStream := sharddistributor.NewMockWatchNamespaceStateClient(ctrl)\n\n\t// Create a context to control when the mock stream should unblock\n\tstreamCtx, cancelStream := context.WithCancel(context.Background())\n\n\tspectator := &spectatorImpl{\n\t\tnamespace:        \"test-ns\",\n\t\tclient:           mockClient,\n\t\tlogger:           log.NewNoop(),\n\t\tscope:            tally.NoopScope,\n\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\tfirstStateSignal: csync.NewResettableSignal(),\n\t\tenabled:          func() bool { return true },\n\t}\n\n\t// Setup stream\n\tmockClient.EXPECT().\n\t\tWatchNamespaceState(gomock.Any(), gomock.Any()).\n\t\tReturn(mockStream, nil)\n\n\t// First Recv returns state\n\tmockStream.EXPECT().Recv().Return(&types.WatchNamespaceStateResponse{\n\t\tExecutors: []*types.ExecutorShardAssignment{\n\t\t\t{\n\t\t\t\tExecutorID: \"executor-1\",\n\t\t\t\tMetadata: map[string]string{\n\t\t\t\t\t\"grpc_address\": \"127.0.0.1:7953\",\n\t\t\t\t},\n\t\t\t\tAssignedShards: []*types.Shard{{ShardKey: \"shard-1\"}},\n\t\t\t},\n\t\t},\n\t}, nil)\n\n\t// Second Recv blocks until shutdown\n\tmockStream.EXPECT().Recv().AnyTimes().DoAndReturn(func(...interface{}) (*types.WatchNamespaceStateResponse, error) {\n\t\t// Wait for context to be done\n\t\t<-streamCtx.Done()\n\t\treturn nil, streamCtx.Err()\n\t})\n\n\tmockStream.EXPECT().CloseSend().Return(nil)\n\n\t// Expect RPC fallback for unknown shard\n\tmockClient.EXPECT().\n\t\tGetShardOwner(gomock.Any(), &types.GetShardOwnerRequest{\n\t\t\tNamespace: \"test-ns\",\n\t\t\tShardKey:  \"unknown-shard\",\n\t\t}).\n\t\tReturn(&types.GetShardOwnerResponse{\n\t\t\tOwner: \"executor-2\",\n\t\t\tMetadata: map[string]string{\n\t\t\t\t\"grpc_address\": \"127.0.0.1:7954\",\n\t\t\t},\n\t\t}, nil)\n\n\tspectator.Start(context.Background())\n\tdefer func() {\n\t\tcancelStream()\n\t\tspectator.Stop()\n\t}()\n\n\trequire.NoError(t, spectator.firstStateSignal.Wait(context.Background()))\n\n\t// Cache hit\n\towner, err := spectator.GetShardOwner(context.Background(), \"shard-1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-1\", owner.ExecutorID)\n\n\t// Cache miss - should trigger RPC\n\towner, err = spectator.GetShardOwner(context.Background(), \"unknown-shard\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-2\", owner.ExecutorID)\n\tassert.Equal(t, \"127.0.0.1:7954\", owner.Metadata[\"grpc_address\"])\n}\n\nfunc TestStreamReconnection(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockClient := sharddistributor.NewMockClient(ctrl)\n\tmockStream1 := sharddistributor.NewMockWatchNamespaceStateClient(ctrl)\n\tmockStream2 := sharddistributor.NewMockWatchNamespaceStateClient(ctrl)\n\tmockTimeSource := clock.NewMockedTimeSource()\n\n\t// Create a context to control when the mock stream should unblock\n\tstreamCtx, cancelStream := context.WithCancel(context.Background())\n\n\tspectator := &spectatorImpl{\n\t\tnamespace:        \"test-ns\",\n\t\tclient:           mockClient,\n\t\tlogger:           log.NewNoop(),\n\t\tscope:            tally.NoopScope,\n\t\ttimeSource:       mockTimeSource,\n\t\tfirstStateSignal: csync.NewResettableSignal(),\n\t\tenabled:          func() bool { return true },\n\t}\n\n\t// First stream fails immediately\n\tmockClient.EXPECT().\n\t\tWatchNamespaceState(gomock.Any(), gomock.Any()).\n\t\tReturn(mockStream1, nil)\n\n\tmockStream1.EXPECT().Recv().Return(nil, errors.New(\"network error\"))\n\tmockStream1.EXPECT().CloseSend().Return(nil)\n\n\t// Second stream succeeds\n\tmockClient.EXPECT().\n\t\tWatchNamespaceState(gomock.Any(), gomock.Any()).\n\t\tReturn(mockStream2, nil)\n\n\t// First Recv returns state\n\tmockStream2.EXPECT().Recv().Return(&types.WatchNamespaceStateResponse{\n\t\tExecutors: []*types.ExecutorShardAssignment{{ExecutorID: \"executor-1\"}},\n\t}, nil)\n\n\t// Second Recv blocks until shutdown\n\tmockStream2.EXPECT().Recv().AnyTimes().DoAndReturn(func(...interface{}) (*types.WatchNamespaceStateResponse, error) {\n\t\t// Wait for context to be done\n\t\t<-streamCtx.Done()\n\t\treturn nil, errors.New(\"shutdown\")\n\t})\n\n\tmockStream2.EXPECT().CloseSend().Return(nil)\n\n\tspectator.Start(context.Background())\n\tdefer func() {\n\t\tcancelStream()\n\t\tspectator.Stop()\n\t}()\n\n\t// Wait for the goroutine to be blocked in Sleep, then advance time\n\tmockTimeSource.BlockUntil(1) // Wait for 1 goroutine to be blocked in Sleep\n\tmockTimeSource.Advance(2 * time.Second)\n\n\trequire.NoError(t, spectator.firstStateSignal.Wait(context.Background()))\n}\n\nfunc TestGetShardOwner_TimeoutBeforeFirstState(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tmockClient := sharddistributor.NewMockClient(ctrl)\n\n\tspectator := &spectatorImpl{\n\t\tnamespace:        \"test-ns\",\n\t\tclient:           mockClient,\n\t\tlogger:           log.NewNoop(),\n\t\tscope:            tally.NoopScope,\n\t\ttimeSource:       clock.NewRealTimeSource(),\n\t\tfirstStateSignal: csync.NewResettableSignal(),\n\t\tenabled:          func() bool { return true },\n\t}\n\n\t// Create a context with a short timeout\n\tctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond)\n\tdefer cancel()\n\n\t// Try to get shard owner before first state is received\n\t// Should timeout and return an error\n\t_, err := spectator.GetShardOwner(ctx, \"shard-1\")\n\tassert.Error(t, err)\n\tassert.Contains(t, err.Error(), \"wait for first state\")\n}\n\nfunc TestWatchLoopDisabled(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tstateSignal := csync.NewResettableSignal()\n\ttimeSource := clock.NewMockedTimeSource()\n\n\tspectator := &spectatorImpl{\n\t\tfirstStateSignal: stateSignal,\n\t\ttimeSource:       timeSource,\n\t\tlogger:           log.NewNoop(),\n\t\tenabled:          func() bool { return false },\n\t}\n\n\terr := spectator.Start(context.Background())\n\tassert.NoError(t, err)\n\n\t// Disabled state enters a sleep loop, verify it sleeps periodically\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(1200 * time.Millisecond)\n\n\ttimeSource.BlockUntil(1)\n\ttimeSource.Advance(1200 * time.Millisecond)\n\n\t// Stop exits cleanly and calls Done() on the signal\n\tspectator.Stop()\n\n\t// After Stop(), Done() has been called so Wait returns nil\n\terr = stateSignal.Wait(context.Background())\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/interface_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: client.go\n//\n// Generated by this command:\n//\n//\tmockgen -package spectatorclient -source client.go -destination interface_mock.go . Spectator\n//\n\n// Package spectatorclient is a generated GoMock package.\npackage spectatorclient\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockSpectator is a mock of Spectator interface.\ntype MockSpectator struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSpectatorMockRecorder\n\tisgomock struct{}\n}\n\n// MockSpectatorMockRecorder is the mock recorder for MockSpectator.\ntype MockSpectatorMockRecorder struct {\n\tmock *MockSpectator\n}\n\n// NewMockSpectator creates a new mock instance.\nfunc NewMockSpectator(ctrl *gomock.Controller) *MockSpectator {\n\tmock := &MockSpectator{ctrl: ctrl}\n\tmock.recorder = &MockSpectatorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSpectator) EXPECT() *MockSpectatorMockRecorder {\n\treturn m.recorder\n}\n\n// GetShardOwner mocks base method.\nfunc (m *MockSpectator) GetShardOwner(ctx context.Context, shardKey string) (*ShardOwner, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardOwner\", ctx, shardKey)\n\tret0, _ := ret[0].(*ShardOwner)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShardOwner indicates an expected call of GetShardOwner.\nfunc (mr *MockSpectatorMockRecorder) GetShardOwner(ctx, shardKey any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardOwner\", reflect.TypeOf((*MockSpectator)(nil).GetShardOwner), ctx, shardKey)\n}\n\n// Start mocks base method.\nfunc (m *MockSpectator) Start(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockSpectatorMockRecorder) Start(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockSpectator)(nil).Start), ctx)\n}\n\n// Stop mocks base method.\nfunc (m *MockSpectator) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockSpectatorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockSpectator)(nil).Stop))\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/metricsconstants/metrics.go",
    "content": "package metricsconstants\n\nconst (\n\t// Operation tag names for ShardDistributorSpectator metrics\n\tShardDistributorSpectatorOperationTagName                    = \"ShardDistributorSpectator\"\n\tShardDistributorSpectatorGetShardOwnerOperationTagName       = \"ShardDistributorSpectatorGetShardOwner\"\n\tShardDistributorSpectatorWatchNamespaceStateOperationTagName = \"ShardDistributorSpectatorWatchNamespaceState\"\n\n\t// Counter metrics\n\tShardDistributorSpectatorClientRequests = \"shard_distributor_spectator_client_requests\"\n\tShardDistributorSpectatorClientFailures = \"shard_distributor_spectator_client_failures\"\n\n\t// Timer metrics\n\tShardDistributorSpectatorClientLatency = \"shard_distributor_spectator_client_latency\"\n)\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/peer_chooser.go",
    "content": "package spectatorclient\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/yarpc/api/peer\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer/hostport\"\n\t\"go.uber.org/yarpc/yarpcerrors\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n)\n\nconst (\n\tNamespaceHeader = \"x-shard-distributor-namespace\"\n)\n\ntype trackedPeer struct {\n\tpeer     peer.Peer\n\tlastUsed time.Time\n}\n\n// SpectatorPeerChooserInterface extends peer.Chooser with SetSpectators method\ntype SpectatorPeerChooserInterface interface {\n\tpeer.Chooser\n\tSetSpectators(spectators *Spectators)\n}\n\n// SpectatorPeerChooser is a peer.Chooser that uses the Spectator to route requests\n// to the correct executor based on shard ownership.\n// This is the shard distributor equivalent of Cadence's RingpopPeerChooser.\n//\n// Flow:\n//  1. Client calls RPC with yarpc.WithShardKey(\"shard-key\")\n//  2. Choose() is called with req.ShardKey = \"shard-key\"\n//  3. Query Spectator for shard owner\n//  4. Extract grpc_address from owner metadata\n//  5. Create/reuse peer for that address\n//  6. Return peer to YARPC for connection\ntype SpectatorPeerChooser struct {\n\tspectators *Spectators\n\ttransport  peer.Transport\n\tlogger     log.Logger\n\tnamespace  string\n\n\tpeersMutex sync.RWMutex\n\tpeers      map[string]*trackedPeer // grpc_address -> peer\n\ttimeSource clock.TimeSource\n\tpeerTTL    time.Duration\n\tstopCh     chan struct{}\n\tstopWG     sync.WaitGroup\n}\n\ntype SpectatorPeerChooserParams struct {\n\tfx.In\n\tTransport  peer.Transport\n\tLogger     log.Logger\n\tConfig     clientcommon.Config\n\tTimeSource clock.TimeSource\n}\n\n// NewSpectatorPeerChooser creates a new peer chooser that routes based on shard distributor ownership\nfunc NewSpectatorPeerChooser(\n\tparams SpectatorPeerChooserParams,\n) SpectatorPeerChooserInterface {\n\treturn &SpectatorPeerChooser{\n\t\ttransport:  params.Transport,\n\t\tlogger:     params.Logger,\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\tpeerTTL:    params.Config.GetPeerTTL(),\n\t\ttimeSource: params.TimeSource,\n\t}\n}\n\n// Start satisfies the peer.Chooser interface\nfunc (c *SpectatorPeerChooser) Start() error {\n\tc.logger.Info(\"Starting shard distributor peer chooser\", tag.ShardNamespace(c.namespace))\n\tc.startEvictionLoop()\n\treturn nil\n}\n\n// Stop satisfies the peer.Chooser interface\nfunc (c *SpectatorPeerChooser) Stop() error {\n\tc.logger.Info(\"Stopping shard distributor peer chooser\", tag.ShardNamespace(c.namespace))\n\n\tif c.stopCh != nil {\n\t\tclose(c.stopCh)\n\t\tc.stopWG.Wait()\n\t\tc.stopCh = nil\n\t}\n\n\tc.peersMutex.Lock()\n\tdefer c.peersMutex.Unlock()\n\n\tfor addr, tp := range c.peers {\n\t\tif err := c.transport.ReleasePeer(tp.peer, &noOpSubscriber{}); err != nil {\n\t\t\tc.logger.Error(\"Failed to release peer\", tag.Error(err), tag.Address(addr))\n\t\t}\n\t}\n\tc.peers = make(map[string]*trackedPeer)\n\n\treturn nil\n}\n\n// IsRunning satisfies the peer.Chooser interface\nfunc (c *SpectatorPeerChooser) IsRunning() bool {\n\treturn true\n}\n\n// Choose returns a peer for the given shard key by:\n// 0. Looking up the spectator for the namespace using the x-shard-distributor-namespace header\n// 1. Looking up the shard owner via the Spectator\n// 2. Extracting the grpc_address from the owner's metadata\n// 3. Creating/reusing a peer for that address\n//\n// The ShardKey in the request is the shard key (e.g., shard ID)\n// The function returns\n// peer: the peer to use for the request\n// onFinish: a function to call when the request is finished (currently no-op)\n// err: the error if the request failed\nfunc (c *SpectatorPeerChooser) Choose(ctx context.Context, req *transport.Request) (peer peer.Peer, onFinish func(error), err error) {\n\tif req.ShardKey == \"\" {\n\t\treturn nil, nil, yarpcerrors.InvalidArgumentErrorf(\"chooser requires ShardKey to be non-empty\")\n\t}\n\n\t// Get the spectator for the namespace\n\tnamespace, ok := req.Headers.Get(NamespaceHeader)\n\tif !ok || namespace == \"\" {\n\t\treturn nil, nil, yarpcerrors.InvalidArgumentErrorf(\"chooser requires x-shard-distributor-namespace header to be non-empty\")\n\t}\n\n\tspectator, err := c.spectators.ForNamespace(namespace)\n\tif err != nil {\n\t\treturn nil, nil, yarpcerrors.InvalidArgumentErrorf(\"get spectator for namespace %s: %w\", namespace, err)\n\t}\n\n\t// Query spectator for shard owner\n\towner, err := spectator.GetShardOwner(ctx, req.ShardKey)\n\tif err != nil {\n\t\treturn nil, nil, yarpcerrors.UnavailableErrorf(\"get shard owner for key %s: %v\", req.ShardKey, err)\n\t}\n\n\t// Extract GRPC address from owner metadata\n\tgrpcAddress, ok := owner.Metadata[clientcommon.GrpcAddressMetadataKey]\n\tif !ok || grpcAddress == \"\" {\n\t\treturn nil, nil, yarpcerrors.InternalErrorf(\"no grpc_address in metadata for executor %s owning shard %s\", owner.ExecutorID, req.ShardKey)\n\t}\n\n\t// Get peer for this address\n\tp, err := c.getOrCreatePeer(grpcAddress)\n\tif err != nil {\n\t\treturn nil, nil, yarpcerrors.InternalErrorf(\"get or create peer for address %s: %v\", grpcAddress, err)\n\t}\n\n\treturn p, func(error) {}, nil\n}\n\nfunc (c *SpectatorPeerChooser) SetSpectators(spectators *Spectators) {\n\tc.spectators = spectators\n}\n\nfunc (c *SpectatorPeerChooser) getOrCreatePeer(grpcAddress string) (peer.Peer, error) {\n\tc.peersMutex.Lock()\n\tdefer c.peersMutex.Unlock()\n\n\tif tp, ok := c.peers[grpcAddress]; ok {\n\t\ttp.lastUsed = c.timeSource.Now()\n\t\treturn tp.peer, nil\n\t}\n\n\tp, err := c.transport.RetainPeer(hostport.Identify(grpcAddress), &noOpSubscriber{})\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"retain peer: %w\", err)\n\t}\n\n\ttp := &trackedPeer{peer: p, lastUsed: c.timeSource.Now()}\n\tc.peers[grpcAddress] = tp\n\treturn p, nil\n}\n\nfunc (c *SpectatorPeerChooser) evictStalePeers() {\n\tnow := c.timeSource.Now()\n\n\tc.peersMutex.Lock()\n\tdefer c.peersMutex.Unlock()\n\n\tfor addr, tp := range c.peers {\n\t\tif now.Sub(tp.lastUsed) > c.peerTTL {\n\t\t\tif err := c.transport.ReleasePeer(tp.peer, &noOpSubscriber{}); err != nil {\n\t\t\t\tc.logger.Error(\"Failed to release stale peer\", tag.Error(err), tag.Address(addr))\n\t\t\t}\n\t\t\tdelete(c.peers, addr)\n\t\t}\n\t}\n}\n\nfunc (c *SpectatorPeerChooser) startEvictionLoop() {\n\tif c.stopCh != nil {\n\t\treturn // already started\n\t}\n\tc.stopCh = make(chan struct{})\n\tc.stopWG.Add(1)\n\tgo func() {\n\t\tdefer c.stopWG.Done()\n\t\t// Tick at half the TTL so a peer is evicted within at most 1.5x the TTL\n\t\t// after its last use (worst case: eviction check just missed, then TTL passes,\n\t\t// then next tick fires at TTL/2 later).\n\t\tticker := c.timeSource.NewTicker(c.peerTTL / 2)\n\t\tdefer ticker.Stop()\n\t\tfor {\n\t\t\tselect {\n\t\t\tcase <-ticker.Chan():\n\t\t\t\tc.evictStalePeers()\n\t\t\tcase <-c.stopCh:\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}()\n}\n\n// noOpSubscriber is a no-op implementation of peer.Subscriber\ntype noOpSubscriber struct{}\n\nfunc (*noOpSubscriber) NotifyStatusChanged(peer.Identifier) {}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/peer_chooser_test.go",
    "content": "package spectatorclient\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer/hostport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n)\n\nfunc TestSpectatorPeerChooser_Choose_MissingShardKey(t *testing.T) {\n\tchooser := &SpectatorPeerChooser{\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t}\n\n\treq := &transport.Request{\n\t\tShardKey: \"\",\n\t\tHeaders:  transport.NewHeaders(),\n\t}\n\n\tp, onFinish, err := chooser.Choose(context.Background(), req)\n\n\tassert.Error(t, err)\n\tassert.Nil(t, p)\n\tassert.Nil(t, onFinish)\n\tassert.Contains(t, err.Error(), \"ShardKey\")\n}\n\nfunc TestSpectatorPeerChooser_Choose_MissingNamespaceHeader(t *testing.T) {\n\tchooser := &SpectatorPeerChooser{\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t}\n\n\treq := &transport.Request{\n\t\tShardKey: \"shard-1\",\n\t\tHeaders:  transport.NewHeaders(),\n\t}\n\n\tp, onFinish, err := chooser.Choose(context.Background(), req)\n\n\tassert.Error(t, err)\n\tassert.Nil(t, p)\n\tassert.Nil(t, onFinish)\n\tassert.Contains(t, err.Error(), \"x-shard-distributor-namespace\")\n}\n\nfunc TestSpectatorPeerChooser_Choose_SpectatorNotFound(t *testing.T) {\n\tchooser := &SpectatorPeerChooser{\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t\tspectators: &Spectators{spectators: make(map[string]Spectator)},\n\t}\n\n\treq := &transport.Request{\n\t\tShardKey: \"shard-1\",\n\t\tHeaders:  transport.NewHeaders().With(NamespaceHeader, \"unknown-namespace\"),\n\t}\n\n\tp, onFinish, err := chooser.Choose(context.Background(), req)\n\n\tassert.Error(t, err)\n\tassert.Nil(t, p)\n\tassert.Nil(t, onFinish)\n\tassert.Contains(t, err.Error(), \"spectator not found\")\n}\n\nfunc TestSpectatorPeerChooser_StartStop(t *testing.T) {\n\tchooser := &SpectatorPeerChooser{\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t\tpeerTTL:    time.Minute,\n\t}\n\n\terr := chooser.Start()\n\trequire.NoError(t, err)\n\n\tassert.True(t, chooser.IsRunning())\n\n\terr = chooser.Stop()\n\tassert.NoError(t, err)\n}\n\nfunc TestSpectatorPeerChooser_SetSpectators(t *testing.T) {\n\tchooser := &SpectatorPeerChooser{\n\t\tlogger: testlogger.New(t),\n\t}\n\n\tspectators := &Spectators{spectators: make(map[string]Spectator)}\n\tchooser.SetSpectators(spectators)\n\n\tassert.Equal(t, spectators, chooser.spectators)\n}\n\nfunc TestSpectatorPeerChooser_Choose_Success(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockSpectator := NewMockSpectator(ctrl)\n\tpeerTransport := grpc.NewTransport()\n\trequire.NoError(t, peerTransport.Start())\n\tdefer peerTransport.Stop()\n\n\tchooser := &SpectatorPeerChooser{\n\t\ttransport:  peerTransport,\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t\tspectators: &Spectators{\n\t\t\tspectators: map[string]Spectator{\n\t\t\t\t\"test-namespace\": mockSpectator,\n\t\t\t},\n\t\t},\n\t}\n\n\tctx := context.Background()\n\treq := &transport.Request{\n\t\tShardKey: \"shard-1\",\n\t\tHeaders:  transport.NewHeaders().With(NamespaceHeader, \"test-namespace\"),\n\t}\n\n\t// Mock spectator to return shard owner with grpc_address\n\tmockSpectator.EXPECT().\n\t\tGetShardOwner(ctx, \"shard-1\").\n\t\tReturn(&ShardOwner{\n\t\t\tExecutorID: \"executor-1\",\n\t\t\tMetadata: map[string]string{\n\t\t\t\tclientcommon.GrpcAddressMetadataKey: \"127.0.0.1:7953\",\n\t\t\t},\n\t\t}, nil)\n\n\t// Execute\n\tp, onFinish, err := chooser.Choose(ctx, req)\n\n\t// Assert\n\tassert.NoError(t, err)\n\tassert.NotNil(t, p)\n\tassert.NotNil(t, onFinish)\n\tassert.Equal(t, \"127.0.0.1:7953\", p.Identifier())\n\tassert.Len(t, chooser.peers, 1)\n\tassert.Equal(t, \"127.0.0.1:7953\", chooser.peers[\"127.0.0.1:7953\"].peer.Identifier())\n}\n\nfunc TestSpectatorPeerChooser_Choose_ReusesPeer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockSpectator := NewMockSpectator(ctrl)\n\tpeerTransport := grpc.NewTransport()\n\trequire.NoError(t, peerTransport.Start())\n\tdefer peerTransport.Stop()\n\n\tchooser := &SpectatorPeerChooser{\n\t\ttransport:  peerTransport,\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t\tspectators: &Spectators{\n\t\t\tspectators: map[string]Spectator{\n\t\t\t\t\"test-namespace\": mockSpectator,\n\t\t\t},\n\t\t},\n\t}\n\n\treq := &transport.Request{\n\t\tShardKey: \"shard-1\",\n\t\tHeaders:  transport.NewHeaders().With(NamespaceHeader, \"test-namespace\"),\n\t}\n\n\t// First call creates the peer\n\tmockSpectator.EXPECT().\n\t\tGetShardOwner(gomock.Any(), \"shard-1\").\n\t\tReturn(&ShardOwner{\n\t\t\tExecutorID: \"executor-1\",\n\t\t\tMetadata: map[string]string{\n\t\t\t\tclientcommon.GrpcAddressMetadataKey: \"127.0.0.1:7953\",\n\t\t\t},\n\t\t}, nil).Times(2)\n\n\tfirstPeer, _, err := chooser.Choose(context.Background(), req)\n\trequire.NoError(t, err)\n\n\t// Second call should reuse the same peer\n\tsecondPeer, _, err := chooser.Choose(context.Background(), req)\n\n\t// Assert - should reuse existing peer\n\tassert.NoError(t, err)\n\tassert.Equal(t, firstPeer, secondPeer)\n\tassert.Len(t, chooser.peers, 1)\n}\n\nfunc TestSpectatorPeerChooser_Choose_TracksLastUsed(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockSpectator := NewMockSpectator(ctrl)\n\tpeerTransport := grpc.NewTransport()\n\trequire.NoError(t, peerTransport.Start())\n\tdefer peerTransport.Stop()\n\n\tmockClock := clock.NewMockedTimeSource()\n\n\tchooser := &SpectatorPeerChooser{\n\t\ttransport:  peerTransport,\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: mockClock,\n\t\tspectators: &Spectators{\n\t\t\tspectators: map[string]Spectator{\"ns\": mockSpectator},\n\t\t},\n\t}\n\n\treq := &transport.Request{\n\t\tShardKey: \"shard-1\",\n\t\tHeaders:  transport.NewHeaders().With(NamespaceHeader, \"ns\"),\n\t}\n\n\t// First call — creates peer, sets lastUsed to t0\n\tmockSpectator.EXPECT().\n\t\tGetShardOwner(gomock.Any(), \"shard-1\").\n\t\tReturn(&ShardOwner{\n\t\t\tExecutorID: \"exec-1\",\n\t\t\tMetadata:   map[string]string{clientcommon.GrpcAddressMetadataKey: \"127.0.0.1:7953\"},\n\t\t}, nil).Times(2)\n\n\t_, _, err := chooser.Choose(context.Background(), req)\n\trequire.NoError(t, err)\n\tfirstLastUsed := chooser.peers[\"127.0.0.1:7953\"].lastUsed\n\n\t// Advance the clock\n\tmockClock.Advance(30 * time.Second)\n\n\t// Second call — reuses peer, should update lastUsed to t0+30s\n\t_, _, err = chooser.Choose(context.Background(), req)\n\trequire.NoError(t, err)\n\tsecondLastUsed := chooser.peers[\"127.0.0.1:7953\"].lastUsed\n\n\t// lastUsed should have advanced\n\tassert.True(t, secondLastUsed.After(firstLastUsed), \"lastUsed should be updated on reuse\")\n}\n\nfunc TestSpectatorPeerChooser_StartStop_WithTTL(t *testing.T) {\n\tpeerTransport := grpc.NewTransport()\n\trequire.NoError(t, peerTransport.Start())\n\tdefer peerTransport.Stop()\n\n\tchooser := &SpectatorPeerChooser{\n\t\ttransport:  peerTransport,\n\t\tlogger:     testlogger.New(t),\n\t\tpeers:      make(map[string]*trackedPeer),\n\t\ttimeSource: clock.NewRealTimeSource(),\n\t\tpeerTTL:    100 * time.Millisecond,\n\t}\n\n\trequire.NoError(t, chooser.Start())\n\tassert.NotNil(t, chooser.stopCh, \"eviction loop should be started\")\n\n\trequire.NoError(t, chooser.Stop())\n\t// After Stop, the goroutine must have exited (stopWG.Wait() returned)\n\t// Verify idempotency: a second Stop should not panic\n\trequire.NoError(t, chooser.Stop())\n}\n\nfunc TestSpectatorPeerChooser_EvictStalePeers(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tpeerTTL       time.Duration\n\t\tadvanceBy     time.Duration\n\t\texpectEvicted bool\n\t}{\n\t\t{\n\t\t\tname:          \"peer within TTL is kept\",\n\t\t\tpeerTTL:       2 * time.Minute,\n\t\t\tadvanceBy:     1 * time.Minute,\n\t\t\texpectEvicted: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"peer exactly at TTL boundary is kept\",\n\t\t\tpeerTTL:       2 * time.Minute,\n\t\t\tadvanceBy:     2 * time.Minute,\n\t\t\texpectEvicted: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"peer past TTL is evicted\",\n\t\t\tpeerTTL:       2 * time.Minute,\n\t\t\tadvanceBy:     2*time.Minute + time.Millisecond,\n\t\t\texpectEvicted: true,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tpeerTransport := grpc.NewTransport()\n\t\t\trequire.NoError(t, peerTransport.Start())\n\t\t\tdefer peerTransport.Stop()\n\n\t\t\tmockClock := clock.NewMockedTimeSource()\n\n\t\t\tchooser := &SpectatorPeerChooser{\n\t\t\t\ttransport:  peerTransport,\n\t\t\t\tlogger:     testlogger.New(t),\n\t\t\t\tpeers:      make(map[string]*trackedPeer),\n\t\t\t\ttimeSource: mockClock,\n\t\t\t\tpeerTTL:    tc.peerTTL,\n\t\t\t}\n\n\t\t\t// Manually insert a tracked peer with lastUsed = now\n\t\t\tp, err := peerTransport.RetainPeer(hostport.Identify(\"127.0.0.1:7953\"), &noOpSubscriber{})\n\t\t\trequire.NoError(t, err)\n\t\t\tchooser.peers[\"127.0.0.1:7953\"] = &trackedPeer{\n\t\t\t\tpeer:     p,\n\t\t\t\tlastUsed: mockClock.Now(),\n\t\t\t}\n\n\t\t\t// Advance clock\n\t\t\tmockClock.Advance(tc.advanceBy)\n\n\t\t\t// Run eviction directly\n\t\t\tchooser.evictStalePeers()\n\n\t\t\tif tc.expectEvicted {\n\t\t\t\tassert.Empty(t, chooser.peers)\n\t\t\t} else {\n\t\t\t\tassert.Len(t, chooser.peers, 1)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/sync/resettable_signal.go",
    "content": "package sync\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n)\n\n// ErrReset is returned by Wait() when Reset() is called while goroutines are waiting\nvar ErrReset = errors.New(\"signal was reset\")\n\n// resettableSignal is a synchronization primitive that allows waiting for a one-time\n// signal with context support, and can be reset to wait for a new signal.\n// Similar to sync.WaitGroup but for single completion events with reset capability.\ntype ResettableSignal struct {\n\tmu      sync.Mutex\n\tdoneCh  chan struct{}\n\tresetCh chan struct{}\n\tdone    bool\n}\n\n// NewResettableSignal creates a new resettable signal in waiting state\nfunc NewResettableSignal() *ResettableSignal {\n\treturn &ResettableSignal{\n\t\tdoneCh:  make(chan struct{}),\n\t\tresetCh: make(chan struct{}),\n\t}\n}\n\n// Done signals that the event has completed. Safe to call multiple times (idempotent).\nfunc (s *ResettableSignal) Done() {\n\ts.mu.Lock()\n\tdefer s.mu.Unlock()\n\n\tif !s.done {\n\t\ts.done = true\n\t\tclose(s.doneCh)\n\t}\n}\n\n// Wait blocks until either Done() is called, the context is cancelled, or Reset() is called.\n// Returns:\n//   - nil if Done() was called\n//   - ctx.Err() if context was cancelled\n//   - ErrReset if Reset() was called while waiting\nfunc (s *ResettableSignal) Wait(ctx context.Context) error {\n\ts.mu.Lock()\n\tdoneCh := s.doneCh\n\tresetCh := s.resetCh\n\tdone := s.done\n\ts.mu.Unlock()\n\n\t// Fast path: already done\n\tif done {\n\t\treturn nil\n\t}\n\n\tselect {\n\tcase <-doneCh:\n\t\treturn nil\n\tcase <-resetCh:\n\t\treturn ErrReset\n\tcase <-ctx.Done():\n\t\treturn ctx.Err()\n\t}\n}\n\n// Reset resets the signal to waiting state. Any goroutines currently blocked in Wait()\n// will immediately be unblocked with ErrReset.\nfunc (s *ResettableSignal) Reset() {\n\ts.mu.Lock()\n\tdefer s.mu.Unlock()\n\n\t// Close reset channel to unblock any waiters (they'll get ErrReset)\n\t// Only close if not already done (to avoid closing a closed channel)\n\tif !s.done {\n\t\tclose(s.resetCh)\n\t}\n\n\t// Create new channels and reset done flag\n\ts.doneCh = make(chan struct{})\n\ts.resetCh = make(chan struct{})\n\ts.done = false\n}\n"
  },
  {
    "path": "service/sharddistributor/client/spectatorclient/sync/resettable_signal_test.go",
    "content": "package sync\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestResettableSignal_DoneWait(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\tsignal.Done()\n\n\tctx := context.Background()\n\terr := signal.Wait(ctx)\n\tassert.NoError(t, err)\n}\n\nfunc TestResettableSignal_WaitDone(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\tdone := make(chan error)\n\tgo func() {\n\t\tdone <- signal.Wait(context.Background())\n\t}()\n\n\t// Give goroutine time to start waiting\n\ttime.Sleep(10 * time.Millisecond)\n\n\tsignal.Done()\n\n\tselect {\n\tcase err := <-done:\n\t\tassert.NoError(t, err)\n\tcase <-time.After(1 * time.Second):\n\t\tt.Fatal(\"Wait did not complete after Done\")\n\t}\n}\n\nfunc TestResettableSignal_ContextCancellation(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tdone := make(chan error)\n\tgo func() {\n\t\tdone <- signal.Wait(ctx)\n\t}()\n\n\ttime.Sleep(10 * time.Millisecond)\n\n\tcancel()\n\n\tselect {\n\tcase err := <-done:\n\t\tassert.Error(t, err)\n\t\tassert.Equal(t, context.Canceled, err)\n\tcase <-time.After(1 * time.Second):\n\t\tt.Fatal(\"Wait did not complete after context cancellation\")\n\t}\n}\n\nfunc TestResettableSignal_ResetWhileWaiting(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\tdone := make(chan error)\n\tgo func() {\n\t\tdone <- signal.Wait(context.Background())\n\t}()\n\n\ttime.Sleep(10 * time.Millisecond)\n\n\tsignal.Reset()\n\n\tselect {\n\tcase err := <-done:\n\t\tassert.Error(t, err)\n\t\tassert.True(t, errors.Is(err, ErrReset), \"expected ErrReset, got %v\", err)\n\tcase <-time.After(1 * time.Second):\n\t\tt.Fatal(\"Wait did not complete after Reset\")\n\t}\n}\n\nfunc TestResettableSignal_MultipleWaitersReset(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\tconst numWaiters = 5\n\tresults := make([]chan error, numWaiters)\n\tfor i := 0; i < numWaiters; i++ {\n\t\ti := i // capture the loop variable, the linter insists\n\t\tresults[i] = make(chan error)\n\t\tgo func() {\n\t\t\tresults[i] <- signal.Wait(context.Background())\n\t\t}()\n\t}\n\n\t// Give goroutines time to start waiting\n\ttime.Sleep(10 * time.Millisecond)\n\n\tsignal.Reset()\n\n\tfor i, ch := range results {\n\t\tselect {\n\t\tcase err := <-ch:\n\t\t\tassert.Error(t, err, \"waiter %d should get error\", i)\n\t\t\tassert.True(t, errors.Is(err, ErrReset), \"waiter %d: expected ErrReset, got %v\", i, err)\n\t\tcase <-time.After(1 * time.Second):\n\t\t\tt.Fatalf(\"Waiter %d did not complete after Reset\", i)\n\t\t}\n\t}\n}\n\nfunc TestResettableSignal_ResetAfterDone(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\t// Terminate and reset the signal\n\tsignal.Done()\n\tsignal.Reset()\n\n\t// Now signal should be waiting again\n\tctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)\n\tdefer cancel()\n\n\terr := signal.Wait(ctx)\n\tassert.Error(t, err)\n\tassert.ErrorIs(t, err, context.DeadlineExceeded)\n}\n\nfunc TestResettableSignal_ResetThenDoneThenWait(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\t// Do a full cycle\n\tsignal.Done()\n\tsignal.Reset()\n\tsignal.Done()\n\n\terr := signal.Wait(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestResettableSignal_IdempotentDone(t *testing.T) {\n\tsignal := NewResettableSignal()\n\n\t// Call Done multiple times\n\tsignal.Done()\n\tsignal.Done()\n\tsignal.Done()\n\n\terr := signal.Wait(context.Background())\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "service/sharddistributor/config/config.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage config\n\nimport (\n\t\"time\"\n\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// Config represents configuration for shard manager service\n\tConfig struct {\n\t\tLoadBalancingMode dynamicproperties.StringPropertyFnWithNamespaceFilters\n\t\tMigrationMode     dynamicproperties.StringPropertyFnWithNamespaceFilters\n\t\tMaxEtcdTxnOps     dynamicproperties.IntPropertyFn\n\n\t\tLoadBalancingNaive LoadBalancingNaiveConfig\n\t}\n\n\tLoadBalancingNaiveConfig struct {\n\t\tMaxDeviation dynamicproperties.Float64PropertyFnWithNamespaceFilters\n\t}\n\n\tStaticConfig struct {\n\t\t// ShardDistribution is the configuration for leader election mechanism that is used by Shard distributor to handle shard distribution per namespace.\n\t\tShardDistribution ShardDistribution `yaml:\"shardDistribution\"`\n\t}\n\n\t// ShardDistribution is a configuration for leader election running.\n\tShardDistribution struct {\n\t\tLeaderStore Store         `yaml:\"leaderStore\"`\n\t\tElection    Election      `yaml:\"election\"`\n\t\tNamespaces  []Namespace   `yaml:\"namespaces\"`\n\t\tProcess     LeaderProcess `yaml:\"process\"`\n\t\tStore       Store         `yaml:\"store\"`\n\t}\n\n\t// Store is a generic container for any storage configuration that should be parsed by the implementation.\n\tStore struct {\n\t\tStorageParams *YamlNode `yaml:\"storageParams\"`\n\t}\n\n\tNamespace struct {\n\t\tName string `yaml:\"name\"`\n\t\tType string `yaml:\"type\"` // The field is a string since it is shared between global config Supported values: fixed|ephemeral.\n\t\tMode string `yaml:\"mode\"` // TODO: this should be an ENUM with possible modes: enabled, read_only, proxy, disabled\n\t\t// ShardNum is defined for fixed namespace.\n\t\tShardNum int64 `yaml:\"shardNum\"`\n\t}\n\n\tElection struct {\n\t\tLeaderPeriod           time.Duration `yaml:\"leaderPeriod\"`           // Time to hold leadership before resigning\n\t\tMaxRandomDelay         time.Duration `yaml:\"maxRandomDelay\"`         // Maximum random delay before campaigning\n\t\tFailedElectionCooldown time.Duration `yaml:\"failedElectionCooldown\"` // wait between election attempts with unhandled errors\n\t}\n\n\tLeaderProcess struct {\n\t\t// Period is the maximum duration between shard rebalance operations\n\t\t// Default: 1 second\n\t\tPeriod time.Duration `yaml:\"period\"`\n\n\t\t// RebalanceCooldown is the minimum duration between shard rebalance operations\n\t\t// Default: 250ms\n\t\tRebalanceCooldown time.Duration `yaml:\"rebalanceCooldown\"`\n\n\t\t// Timeout is the maximum duration of a single shard rebalance operation\n\t\t// Default: 1 second\n\t\tTimeout time.Duration `yaml:\"timeout\"`\n\n\t\t// HeartbeatTTL is the duration after which, if no heartbeat is received from an executor,\n\t\t// the executor is considered stale and its shards are eligible for redistribution.\n\t\t// Default: 10 seconds\n\t\tHeartbeatTTL time.Duration `yaml:\"heartbeatTTL\"`\n\t}\n\n\t// YamlNode is a lazy-unmarshaler, because *yaml.Node only exists in gopkg.in/yaml.v3, not v2,\n\t// and go.uber.org/config currently uses only v2.\n\tYamlNode struct {\n\t\tunmarshal func(out any) error\n\t}\n)\n\nconst (\n\tNamespaceTypeFixed     = \"fixed\"\n\tNamespaceTypeEphemeral = \"ephemeral\"\n)\n\nconst (\n\tMigrationModeINVALID                = \"invalid\"\n\tMigrationModeLOCALPASSTHROUGH       = \"local_pass\"\n\tMigrationModeLOCALPASSTHROUGHSHADOW = \"local_pass_shadow\"\n\tMigrationModeDISTRIBUTEDPASSTHROUGH = \"distributed_pass\"\n\tMigrationModeONBOARDED              = \"onboarded\"\n)\n\n// MigrationMode maps string migration mode values to types.MigrationMode\nvar MigrationMode = map[string]types.MigrationMode{\n\tMigrationModeINVALID:                types.MigrationModeINVALID,\n\tMigrationModeLOCALPASSTHROUGH:       types.MigrationModeLOCALPASSTHROUGH,\n\tMigrationModeLOCALPASSTHROUGHSHADOW: types.MigrationModeLOCALPASSTHROUGHSHADOW,\n\tMigrationModeDISTRIBUTEDPASSTHROUGH: types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\tMigrationModeONBOARDED:              types.MigrationModeONBOARDED,\n}\n\n// NewConfig returns a new instance of Config\nfunc NewConfig(dc *dynamicconfig.Collection) *Config {\n\treturn &Config{\n\t\tLoadBalancingMode: dc.GetStringPropertyFilteredByNamespace(dynamicproperties.ShardDistributorLoadBalancingMode),\n\t\tMigrationMode:     dc.GetStringPropertyFilteredByNamespace(dynamicproperties.ShardDistributorMigrationMode),\n\t\tMaxEtcdTxnOps:     dc.GetIntProperty(dynamicproperties.ShardDistributorMaxEtcdTxnOps),\n\n\t\tLoadBalancingNaive: LoadBalancingNaiveConfig{\n\t\t\tMaxDeviation: dc.GetFloat64PropertyFilteredByNamespace(dynamicproperties.ShardDistributorLoadBalancingNaiveMaxDeviation),\n\t\t},\n\t}\n}\n\n// GetMigrationMode gets the migration mode for a given namespace\n// If the mode is not set, it defaults to MigrationModeINVALID\nfunc (c *Config) GetMigrationMode(namespace string) types.MigrationMode {\n\tmode, ok := MigrationMode[c.MigrationMode(namespace)]\n\tif !ok {\n\t\treturn MigrationMode[MigrationModeINVALID]\n\t}\n\treturn mode\n}\n\nconst (\n\tLoadBalancingModeINVALID = \"invalid\"\n\tLoadBalancingModeNAIVE   = \"naive\"\n\tLoadBalancingModeGREEDY  = \"greedy\"\n)\n\n// LoadBalancingMode maps string migration mode values to types.LoadBalancingMode\nvar LoadBalancingMode = map[string]types.LoadBalancingMode{\n\tLoadBalancingModeINVALID: types.LoadBalancingModeINVALID,\n\tLoadBalancingModeNAIVE:   types.LoadBalancingModeNAIVE,\n\tLoadBalancingModeGREEDY:  types.LoadBalancingModeGREEDY,\n}\n\n// GetLoadBalancingMode gets the load balancing mode for a given namespace\n// If the mode is invalid, it returns types.LoadBalancingModeINVALID\nfunc (c *Config) GetLoadBalancingMode(namespace string) types.LoadBalancingMode {\n\tmode, ok := LoadBalancingMode[c.LoadBalancingMode(namespace)]\n\tif !ok {\n\t\treturn types.LoadBalancingModeINVALID\n\t}\n\n\treturn mode\n}\n\nfunc (s *ShardDistribution) GetMigrationMode(namespace string) types.MigrationMode {\n\tfor _, ns := range s.Namespaces {\n\t\tif ns.Name == namespace {\n\t\t\treturn MigrationMode[ns.Mode]\n\t\t}\n\t}\n\t// TODO in the dynamic configuration I will setup a default value\n\treturn MigrationMode[MigrationModeONBOARDED]\n}\n\nvar _ yaml.Unmarshaler = (*YamlNode)(nil)\n\nfunc (y *YamlNode) UnmarshalYAML(unmarshal func(interface{}) error) error {\n\ty.unmarshal = unmarshal\n\treturn nil\n}\n\nfunc (y *YamlNode) Decode(out any) error {\n\tif y == nil {\n\t\treturn nil\n\t}\n\treturn y.unmarshal(out)\n}\n"
  },
  {
    "path": "service/sharddistributor/config/config_test.go",
    "content": "package config\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNewDynamicConfigCreatesInstanceWithProperties(t *testing.T) {\n\tdc := dynamicconfig.NewNopCollection()\n\n\tconfig := NewConfig(dc)\n\n\tassert.NotNil(t, config)\n\tassert.NotNil(t, config.LoadBalancingMode)\n\tassert.NotNil(t, config.MigrationMode)\n}\n\nfunc TestGetMigrationMode(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tconfigValue  string\n\t\texpectedMode types.MigrationMode\n\t}{\n\t\t{\n\t\t\tname:         \"LocalPassthrough\",\n\t\t\tconfigValue:  MigrationModeLOCALPASSTHROUGH,\n\t\t\texpectedMode: types.MigrationModeLOCALPASSTHROUGH,\n\t\t},\n\t\t{\n\t\t\tname:         \"LocalPassthroughShadow\",\n\t\t\tconfigValue:  MigrationModeLOCALPASSTHROUGHSHADOW,\n\t\t\texpectedMode: types.MigrationModeLOCALPASSTHROUGHSHADOW,\n\t\t},\n\t\t{\n\t\t\tname:         \"DistributedPassthrough\",\n\t\t\tconfigValue:  MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t\t\texpectedMode: types.MigrationModeDISTRIBUTEDPASSTHROUGH,\n\t\t},\n\t\t{\n\t\t\tname:         \"Onboarded\",\n\t\t\tconfigValue:  MigrationModeONBOARDED,\n\t\t\texpectedMode: types.MigrationModeONBOARDED,\n\t\t},\n\t\t{\n\t\t\tname:         \"Empty\",\n\t\t\tconfigValue:  \"\",\n\t\t\texpectedMode: types.MigrationModeINVALID,\n\t\t},\n\t\t{\n\t\t\tname:         \"Invalid\",\n\t\t\tconfigValue:  MigrationModeINVALID,\n\t\t\texpectedMode: types.MigrationModeINVALID,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\terr := client.UpdateValue(dynamicproperties.ShardDistributorMigrationMode, tt.configValue)\n\t\t\trequire.NoError(t, err)\n\t\t\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\t\t\tconfig := NewConfig(dc)\n\n\t\t\tmode := config.GetMigrationMode(\"test-namespace\")\n\t\t\tassert.Equal(t, tt.expectedMode, mode)\n\t\t})\n\t}\n}\n\nfunc TestGetLoadBalancingMode(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tconfigValue  string\n\t\texpectedMode types.LoadBalancingMode\n\t}{\n\t\t{\n\t\t\tname:         \"Naive\",\n\t\t\tconfigValue:  \"naive\",\n\t\t\texpectedMode: types.LoadBalancingModeNAIVE,\n\t\t},\n\t\t{\n\t\t\tname:         \"Greedy\",\n\t\t\tconfigValue:  \"greedy\",\n\t\t\texpectedMode: types.LoadBalancingModeGREEDY,\n\t\t},\n\t\t{\n\t\t\tname:         \"Invalid\",\n\t\t\tconfigValue:  \"invalid\",\n\t\t\texpectedMode: types.LoadBalancingModeINVALID,\n\t\t},\n\t\t{\n\t\t\tname:         \"Empty\",\n\t\t\tconfigValue:  \"\",\n\t\t\texpectedMode: types.LoadBalancingModeINVALID,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tclient := dynamicconfig.NewInMemoryClient()\n\t\t\terr := client.UpdateValue(dynamicproperties.ShardDistributorLoadBalancingMode, tt.configValue)\n\t\t\trequire.NoError(t, err)\n\t\t\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\t\t\tconfig := NewConfig(dc)\n\n\t\t\tmode := config.GetLoadBalancingMode(\"test-namespace\")\n\t\t\tassert.Equal(t, tt.expectedMode, mode)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/config/configtest/config.go",
    "content": "package configtest\n\nimport (\n\t\"testing\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\ntype ConfigEntry struct {\n\tKey   dynamicproperties.Key\n\tValue interface{}\n}\n\nfunc NewTestMigrationConfig(t *testing.T, configEntries ...ConfigEntry) *config.Config {\n\tclient := dynamicconfig.NewInMemoryClient()\n\tfor _, entry := range configEntries {\n\t\terr := client.UpdateValue(entry.Key, entry.Value)\n\t\tif err != nil {\n\t\t\tt.Errorf(\"Failed to update config \")\n\t\t}\n\t}\n\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\tmigrationConfig := config.NewConfig(dc)\n\treturn migrationConfig\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/batcher.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t_requestsChannelLimit = 1024\n\t_intervalJitterCoeff  = 0.1 // 10% jitter\n)\n\n// ephemeralAssignmentBatchFn is a function that assigns a batch of shards within a namespace\n// and returns a map of shardKey -> GetShardOwnerResponse for each successfully assigned shard.\n// Keys absent from the result map indicate an error for that specific shard.\ntype ephemeralAssignmentBatchFn func(ctx context.Context, namespace string, shardKeys []string) (map[string]*types.GetShardOwnerResponse, error)\n\n// batchRequest is a single caller's request submitted to the shardBatcher.\ntype batchRequest struct {\n\tnamespace string\n\tshardKey  string\n\t// respChan is a buffered channel (cap 1) so the batcher goroutine never blocks\n\t// when writing the result back.\n\trespChan chan batchResponse\n}\n\ntype batchResponse struct {\n\tresp *types.GetShardOwnerResponse\n\terr  error\n}\n\n// shardBatcher collects GetShardOwner calls for ephemeral namespaces over a\n// configurable time window and processes them in a single batch.\n//\n// Usage:\n//\n//\tb := newShardBatcher(100*time.Millisecond, processFn)\n//\tb.Start()\n//\tdefer b.Stop()\n//\tresp, err := b.Submit(ctx, &types.GetShardOwnerRequest{Namespace: namespace, ShardKey: shardKey})\ntype shardBatcher struct {\n\ttimeSource   clock.TimeSource\n\tinterval     time.Duration\n\tprocessBatch ephemeralAssignmentBatchFn\n\n\t// requestChan is shared across all goroutines; requests are keyed by namespace\n\t// inside the loop so a single goroutine can handle multiple namespaces.\n\trequestChan chan *batchRequest\n\n\tctx    context.Context\n\tcancel context.CancelFunc\n\twg     sync.WaitGroup\n}\n\nfunc newShardBatcher(timeSource clock.TimeSource, interval time.Duration, processBatch ephemeralAssignmentBatchFn) *shardBatcher {\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &shardBatcher{\n\t\ttimeSource:   timeSource,\n\t\tinterval:     interval,\n\t\tprocessBatch: processBatch,\n\t\trequestChan:  make(chan *batchRequest, _requestsChannelLimit),\n\t\tctx:          ctx,\n\t\tcancel:       cancel,\n\t}\n}\n\n// Start launches the background batching loop.\nfunc (b *shardBatcher) Start() {\n\tb.wg.Add(1)\n\tgo b.loop()\n}\n\n// Stop signals the batching loop to shut down and waits for it to finish.\n// Any requests that have already been submitted but not yet flushed will be\n// drained and processed before the loop exits.\nfunc (b *shardBatcher) Stop() {\n\tb.cancel()\n\tb.wg.Wait()\n}\n\n// Submit enqueues a single GetShardOwner request and blocks until the batcher\n// has processed the batch that contains it, or until ctx is cancelled.\nfunc (b *shardBatcher) Submit(ctx context.Context, request *types.GetShardOwnerRequest) (*types.GetShardOwnerResponse, error) {\n\treq := &batchRequest{\n\t\tnamespace: request.Namespace,\n\t\tshardKey:  request.ShardKey,\n\t\trespChan:  make(chan batchResponse, 1),\n\t}\n\n\tselect {\n\tcase b.requestChan <- req:\n\tcase <-ctx.Done():\n\t\treturn nil, ctx.Err()\n\tcase <-b.ctx.Done():\n\t\treturn nil, b.ctx.Err()\n\t}\n\n\tselect {\n\tcase res := <-req.respChan:\n\t\treturn res.resp, res.err\n\tcase <-ctx.Done():\n\t\treturn nil, ctx.Err()\n\tcase <-b.ctx.Done():\n\t\treturn nil, b.ctx.Err()\n\t}\n}\n\nfunc (b *shardBatcher) loop() {\n\tdefer b.wg.Done()\n\n\tticker := b.timeSource.NewTicker(backoff.JitDuration(b.interval, _intervalJitterCoeff))\n\tdefer ticker.Stop()\n\n\t// pending maps namespace -> list of batchRequests accumulated since last flush.\n\tpending := make(map[string][]*batchRequest)\n\n\tfor {\n\t\tselect {\n\t\tcase req := <-b.requestChan:\n\t\t\tpending[req.namespace] = append(pending[req.namespace], req)\n\n\t\tcase <-ticker.Chan():\n\t\t\tb.flush(pending)\n\t\t\tpending = make(map[string][]*batchRequest)\n\n\t\tcase <-b.ctx.Done():\n\t\t\t// drainAndCancel sends a cancellation response to all requests that arrived\n\t\t\t// after the last tick but before the context was cancelled.\n\t\t\tb.drainAndCancel(pending)\n\t\t\treturn\n\t\t}\n\t}\n}\n\n// flush processes all accumulated requests namespace by namespace.\nfunc (b *shardBatcher) flush(pending map[string][]*batchRequest) {\n\tfor namespace, reqs := range pending {\n\t\tif len(reqs) == 0 {\n\t\t\tcontinue\n\t\t}\n\n\t\tshardKeys := make([]string, len(reqs))\n\t\tfor i, r := range reqs {\n\t\t\tshardKeys[i] = r.shardKey\n\t\t}\n\n\t\t// Use a background context so individual caller cancellations do not\n\t\t// abort the whole batch; callers will surface their own context error\n\t\t// via the select in Submit.\n\t\tctx, cancel := context.WithTimeout(context.Background(), 5*b.interval)\n\t\tresults, err := b.processBatch(ctx, namespace, shardKeys)\n\t\tcancel()\n\n\t\tfor _, req := range reqs {\n\t\t\tvar res batchResponse\n\t\t\tif err != nil {\n\t\t\t\tres = batchResponse{err: err}\n\t\t\t} else {\n\t\t\t\tres = batchResponse{resp: results[req.shardKey]}\n\t\t\t\tif res.resp == nil {\n\t\t\t\t\t// processBatch is expected to always include an entry for\n\t\t\t\t\t// every key it was given; a missing entry is an internal error.\n\t\t\t\t\tres = batchResponse{err: &types.InternalServiceError{\n\t\t\t\t\t\tMessage: \"batch processor returned no result for shard key: \" + req.shardKey,\n\t\t\t\t\t}}\n\t\t\t\t}\n\t\t\t}\n\t\t\t// Non-blocking write: respChan has capacity 1 and each req has\n\t\t\t// exactly one writer (this loop) and one reader (Submit).\n\t\t\treq.respChan <- res\n\t\t}\n\t}\n}\n\nfunc (b *shardBatcher) drainAndCancel(pending map[string][]*batchRequest) {\n\t// First flush whatever was already accumulated.\n\tb.flush(pending)\n\n\t// Then drain the channel itself.\n\tfor {\n\t\tselect {\n\t\tcase req := <-b.requestChan:\n\t\t\treq.respChan <- batchResponse{err: b.ctx.Err()}\n\t\tdefault:\n\t\t\treturn\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/batcher_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// processFnFromMap returns an ephemeralAssignmentBatchFn that resolves shard keys from a\n// fixed map, returning an empty map (and no error) for any key not present.\nfunc processFnFromMap(results map[string]*types.GetShardOwnerResponse) ephemeralAssignmentBatchFn {\n\treturn func(_ context.Context, _ string, shardKeys []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\tout := make(map[string]*types.GetShardOwnerResponse, len(shardKeys))\n\t\tfor _, k := range shardKeys {\n\t\t\tif v, ok := results[k]; ok {\n\t\t\t\tout[k] = v\n\t\t\t}\n\t\t}\n\t\treturn out, nil\n\t}\n}\n\n// advanceUntilDone repeatedly advances the mocked clock by step until done is closed.\n// Each advance fires one ticker cycle, draining whatever requests have been enqueued\n// since the previous tick. This avoids races between goroutine scheduling and a\n// single Advance call.\nfunc advanceUntilDone(ts clock.MockedTimeSource, done <-chan struct{}, step time.Duration) {\n\tfor {\n\t\tselect {\n\t\tcase <-done:\n\t\t\treturn\n\t\tdefault:\n\t\t\tts.Advance(step)\n\t\t}\n\t}\n}\n\nfunc TestShardBatcher_Submit(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttests := []struct {\n\t\tname      string\n\t\tbatchFn   ephemeralAssignmentBatchFn\n\t\tnamespace string\n\t\tshardKey  string\n\t\t// ctxFn builds the context used for the Submit call; defaults to context.Background().\n\t\tctxFn      func() context.Context\n\t\twantErr    bool\n\t\twantErrIs  error\n\t\twantErrMsg string\n\t\twantOwner  string\n\t}{\n\t\t{\n\t\t\tname: \"single request resolved from map\",\n\t\t\tbatchFn: processFnFromMap(map[string]*types.GetShardOwnerResponse{\n\t\t\t\t\"shard-1\": {Owner: \"exec-1\", Namespace: \"ns\"},\n\t\t\t}),\n\t\t\tnamespace: \"ns\",\n\t\t\tshardKey:  \"shard-1\",\n\t\t\twantOwner: \"exec-1\",\n\t\t},\n\t\t{\n\t\t\tname: \"batch function returns error - propagated to caller\",\n\t\t\tbatchFn: func(_ context.Context, _ string, _ []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\t\t\treturn nil, errors.New(\"storage unavailable\")\n\t\t\t},\n\t\t\tnamespace:  \"ns\",\n\t\t\tshardKey:   \"shard-1\",\n\t\t\twantErr:    true,\n\t\t\twantErrMsg: \"storage unavailable\",\n\t\t},\n\t\t{\n\t\t\tname: \"key absent from batch result returns internal error\",\n\t\t\tbatchFn: func(_ context.Context, _ string, _ []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\t\t\treturn map[string]*types.GetShardOwnerResponse{}, nil\n\t\t\t},\n\t\t\tnamespace:  \"ns\",\n\t\t\tshardKey:   \"shard-missing\",\n\t\t\twantErr:    true,\n\t\t\twantErrMsg: \"shard-missing\",\n\t\t},\n\t\t{\n\t\t\tname:      \"context cancelled before submit\",\n\t\t\tbatchFn:   processFnFromMap(nil),\n\t\t\tnamespace: \"ns\",\n\t\t\tshardKey:  \"shard-1\",\n\t\t\tctxFn: func() context.Context {\n\t\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\t\tcancel()\n\t\t\t\treturn ctx\n\t\t\t},\n\t\t\twantErr:   true,\n\t\t\twantErrIs: context.Canceled,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tts := clock.NewMockedTimeSource()\n\t\t\tb := newShardBatcher(ts, 10*time.Millisecond, tc.batchFn)\n\t\t\tb.Start()\n\t\t\tdefer b.Stop()\n\n\t\t\tctx := context.Background()\n\t\t\tif tc.ctxFn != nil {\n\t\t\t\tctx = tc.ctxFn()\n\t\t\t}\n\n\t\t\ttype result struct {\n\t\t\t\tresp *types.GetShardOwnerResponse\n\t\t\t\terr  error\n\t\t\t}\n\t\t\tdone := make(chan struct{})\n\t\t\tch := make(chan result, 1)\n\t\t\tgo func() {\n\t\t\t\tresp, err := b.Submit(ctx, &types.GetShardOwnerRequest{Namespace: tc.namespace, ShardKey: tc.shardKey})\n\t\t\t\tch <- result{resp, err}\n\t\t\t\tclose(done)\n\t\t\t}()\n\n\t\t\t// Keep ticking until the goroutine finishes. Advancing by 2x the interval\n\t\t\t// per step accounts for jitter and ensures the request is flushed even if\n\t\t\t// it races with the first tick.\n\t\t\tts.BlockUntil(1)\n\t\t\tadvanceUntilDone(ts, done, 20*time.Millisecond)\n\n\t\t\tgot := <-ch\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, got.err)\n\t\t\t\tif tc.wantErrIs != nil {\n\t\t\t\t\tassert.ErrorIs(t, got.err, tc.wantErrIs)\n\t\t\t\t}\n\t\t\t\tif tc.wantErrMsg != \"\" {\n\t\t\t\t\tassert.ErrorContains(t, got.err, tc.wantErrMsg)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\trequire.NoError(t, got.err)\n\t\t\tif tc.wantOwner != \"\" {\n\t\t\t\tassert.Equal(t, tc.wantOwner, got.resp.Owner)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardBatcher_MultipleNamespacesIsolated(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttests := []struct {\n\t\tname     string\n\t\tresults  map[string]*types.GetShardOwnerResponse\n\t\trequests []struct {\n\t\t\tnamespace string\n\t\t\tshardKey  string\n\t\t\twantOwner string\n\t\t}\n\t}{\n\t\t{\n\t\t\tname: \"two namespaces resolved independently\",\n\t\t\tresults: map[string]*types.GetShardOwnerResponse{\n\t\t\t\t\"shard-a\": {Owner: \"exec-ns1\", Namespace: \"ns1\"},\n\t\t\t\t\"shard-b\": {Owner: \"exec-ns2\", Namespace: \"ns2\"},\n\t\t\t},\n\t\t\trequests: []struct {\n\t\t\t\tnamespace string\n\t\t\t\tshardKey  string\n\t\t\t\twantOwner string\n\t\t\t}{\n\t\t\t\t{namespace: \"ns1\", shardKey: \"shard-a\", wantOwner: \"exec-ns1\"},\n\t\t\t\t{namespace: \"ns2\", shardKey: \"shard-b\", wantOwner: \"exec-ns2\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tts := clock.NewMockedTimeSource()\n\t\t\tb := newShardBatcher(ts, 10*time.Millisecond, processFnFromMap(tc.results))\n\t\t\tb.Start()\n\t\t\tdefer b.Stop()\n\n\t\t\ttype result struct {\n\t\t\t\towner string\n\t\t\t\terr   error\n\t\t\t}\n\t\t\tgot := make([]result, len(tc.requests))\n\t\t\tvar wg sync.WaitGroup\n\t\t\tfor i, req := range tc.requests {\n\t\t\t\twg.Add(1)\n\t\t\t\tgo func(i int, namespace, shardKey string) {\n\t\t\t\t\tdefer wg.Done()\n\t\t\t\t\tresp, err := b.Submit(context.Background(), &types.GetShardOwnerRequest{Namespace: namespace, ShardKey: shardKey})\n\t\t\t\t\tif err == nil {\n\t\t\t\t\t\tgot[i] = result{owner: resp.Owner}\n\t\t\t\t\t} else {\n\t\t\t\t\t\tgot[i] = result{err: err}\n\t\t\t\t\t}\n\t\t\t\t}(i, req.namespace, req.shardKey)\n\t\t\t}\n\n\t\t\tdone := make(chan struct{})\n\t\t\tgo func() { wg.Wait(); close(done) }()\n\n\t\t\tts.BlockUntil(1)\n\t\t\tadvanceUntilDone(ts, done, 20*time.Millisecond)\n\n\t\t\tfor i, req := range tc.requests {\n\t\t\t\trequire.NoError(t, got[i].err)\n\t\t\t\tassert.Equal(t, req.wantOwner, got[i].owner)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardBatcher_ErrorPropagatedToAllCallers(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttests := []struct {\n\t\tname       string\n\t\tbatchErr   error\n\t\tnumCallers int\n\t}{\n\t\t{\n\t\t\tname:       \"error propagated to all concurrent callers\",\n\t\t\tbatchErr:   errors.New(\"storage unavailable\"),\n\t\t\tnumCallers: 5,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tbatchFn := func(_ context.Context, _ string, _ []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\t\t\treturn nil, tc.batchErr\n\t\t\t}\n\n\t\t\tts := clock.NewMockedTimeSource()\n\t\t\tb := newShardBatcher(ts, 10*time.Millisecond, batchFn)\n\t\t\tb.Start()\n\t\t\tdefer b.Stop()\n\n\t\t\terrs := make([]error, tc.numCallers)\n\t\t\tvar wg sync.WaitGroup\n\t\t\tfor i := range tc.numCallers {\n\t\t\t\twg.Add(1)\n\t\t\t\tgo func(i int) {\n\t\t\t\t\tdefer wg.Done()\n\t\t\t\t\t_, errs[i] = b.Submit(context.Background(), &types.GetShardOwnerRequest{Namespace: \"ns\", ShardKey: \"shard\"})\n\t\t\t\t}(i)\n\t\t\t}\n\n\t\t\tdone := make(chan struct{})\n\t\t\tgo func() { wg.Wait(); close(done) }()\n\n\t\t\tts.BlockUntil(1)\n\t\t\tadvanceUntilDone(ts, done, 20*time.Millisecond)\n\n\t\t\tfor i, err := range errs {\n\t\t\t\tassert.ErrorContains(t, err, tc.batchErr.Error(), \"caller %d should receive the batch error\", i)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardBatcher_ConcurrentRequestsBatchedTogether(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttests := []struct {\n\t\tname      string\n\t\tnumShards int\n\t\tinterval  time.Duration\n\t}{\n\t\t{\n\t\t\tname:      \"20 concurrent shards collapsed into fewer batch calls\",\n\t\t\tnumShards: 20,\n\t\t\tinterval:  50 * time.Millisecond,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresults := make(map[string]*types.GetShardOwnerResponse, tc.numShards)\n\t\t\tfor i := range tc.numShards {\n\t\t\t\tkey := \"shard-\" + string(rune('A'+i))\n\t\t\t\tresults[key] = &types.GetShardOwnerResponse{Owner: \"exec-1\", Namespace: \"ns\"}\n\t\t\t}\n\n\t\t\t// maxBatchSize tracks the largest single batch seen. Batching is confirmed\n\t\t\t// when at least one flush call receives more than one shard key.\n\t\t\tvar mu sync.Mutex\n\t\t\tmaxBatchSize := 0\n\t\t\tbatchFn := func(_ context.Context, _ string, shardKeys []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\t\t\tmu.Lock()\n\t\t\t\tif len(shardKeys) > maxBatchSize {\n\t\t\t\t\tmaxBatchSize = len(shardKeys)\n\t\t\t\t}\n\t\t\t\tmu.Unlock()\n\t\t\t\tout := make(map[string]*types.GetShardOwnerResponse, len(shardKeys))\n\t\t\t\tfor _, k := range shardKeys {\n\t\t\t\t\tif v, ok := results[k]; ok {\n\t\t\t\t\t\tout[k] = v\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\treturn out, nil\n\t\t\t}\n\n\t\t\tts := clock.NewMockedTimeSource()\n\t\t\tb := newShardBatcher(ts, tc.interval, batchFn)\n\t\t\tb.Start()\n\t\t\tdefer b.Stop()\n\n\t\t\tvar wg sync.WaitGroup\n\t\t\tfor i := range tc.numShards {\n\t\t\t\twg.Add(1)\n\t\t\t\tgo func(i int) {\n\t\t\t\t\tdefer wg.Done()\n\t\t\t\t\tkey := \"shard-\" + string(rune('A'+i))\n\t\t\t\t\tresp, err := b.Submit(context.Background(), &types.GetShardOwnerRequest{Namespace: \"ns\", ShardKey: key})\n\t\t\t\t\trequire.NoError(t, err)\n\t\t\t\t\tassert.Equal(t, \"exec-1\", resp.Owner)\n\t\t\t\t}(i)\n\t\t\t}\n\n\t\t\tdone := make(chan struct{})\n\t\t\tgo func() { wg.Wait(); close(done) }()\n\n\t\t\t// Keep ticking until all callers finish. Each advance drains whatever\n\t\t\t// has been enqueued since the previous tick.\n\t\t\tts.BlockUntil(1)\n\t\t\tadvanceUntilDone(ts, done, 2*tc.interval)\n\n\t\t\tmu.Lock()\n\t\t\tdefer mu.Unlock()\n\t\t\tassert.Greater(t, maxBatchSize, 1, \"expected at least one batch call to receive more than one shard\")\n\t\t})\n\t}\n}\n\nfunc TestShardBatcher_StopDrainsAndCancelsRemainingRequests(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\ttests := []struct {\n\t\tname        string\n\t\tstopTimeout time.Duration\n\t}{\n\t\t{\n\t\t\tname:        \"in-flight request resolves after Stop\",\n\t\t\tstopTimeout: 500 * time.Millisecond,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tblock := make(chan struct{})\n\t\t\tbatchFn := func(_ context.Context, _ string, _ []string) (map[string]*types.GetShardOwnerResponse, error) {\n\t\t\t\t<-block\n\t\t\t\treturn nil, nil\n\t\t\t}\n\n\t\t\tts := clock.NewMockedTimeSource()\n\t\t\tb := newShardBatcher(ts, 5*time.Millisecond, batchFn)\n\t\t\tb.Start()\n\n\t\t\terrCh := make(chan error, 1)\n\t\t\tdone := make(chan struct{})\n\t\t\tgo func() {\n\t\t\t\t_, err := b.Submit(context.Background(), &types.GetShardOwnerRequest{Namespace: \"ns\", ShardKey: \"shard-1\"})\n\t\t\t\terrCh <- err\n\t\t\t\tclose(done)\n\t\t\t}()\n\n\t\t\t// Keep ticking until the flush is triggered and the goroutine unblocks.\n\t\t\tts.BlockUntil(1)\n\t\t\tgo advanceUntilDone(ts, done, 10*time.Millisecond)\n\n\t\t\t// Unblock the batchFn and stop the batcher.\n\t\t\tclose(block)\n\t\t\tb.Stop()\n\n\t\t\tselect {\n\t\t\tcase <-errCh:\n\t\t\t\t// ok — caller received either a result or a cancellation error\n\t\t\tcase <-time.After(tc.stopTimeout):\n\t\t\t\tt.Fatal(\"Submit did not return after Stop\")\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/ephemeral_assigner.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n// assignEphemeralBatch is the ephemeralAssignmentBatchFn wired into the shardBatcher.\n// It processes a whole batch of unassigned shard keys for a single ephemeral\n// namespace using two storage operations:\n//  1. GetState     — read current namespace state once for the whole batch.\n//  2. AssignShards — write all new assignments atomically in one operation.\n//\n// After the write, GetExecutor is called once per unique chosen executor (not\n// per shard) to fetch metadata for the response, since metadata is stored\n// separately in the shard cache and is not returned by GetState.\n//\n// Within the batch the least-loaded ACTIVE executor is chosen per shard, with\n// the in-batch running count updated after each pick so load is spread evenly.\nfunc (h *handlerImpl) assignEphemeralBatch(ctx context.Context, namespace string, shardKeys []string) (map[string]*types.GetShardOwnerResponse, error) {\n\tstate, err := h.storage.GetState(ctx, namespace)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"get namespace state: %v\", err)}\n\t}\n\n\tassignedCounts, err := buildAssignedCounts(state)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tchosenExecutors, err := pickExecutors(namespace, shardKeys, assignedCounts)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tmergeAssignments(state, chosenExecutors)\n\n\tif err := h.storage.AssignShards(ctx, namespace, store.AssignShardsRequest{NewState: state}, store.NopGuard()); err != nil {\n\t\tif errors.Is(err, store.ErrVersionConflict) {\n\t\t\t// Return the version-conflict sentinel unwrapped so callers can\n\t\t\t// detect it with errors.Is and decide whether to retry.\n\t\t\treturn nil, fmt.Errorf(\"assign ephemeral shards: %w\", err)\n\t\t}\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"assign ephemeral shards: %v\", err)}\n\t}\n\n\texecutorOwners, err := h.fetchExecutorMetadata(ctx, namespace, chosenExecutors)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn buildResults(namespace, shardKeys, chosenExecutors, executorOwners), nil\n}\n\n// buildAssignedCounts returns a map of executorID -> current shard count for\n// all ACTIVE executors in the given namespace state.\nfunc buildAssignedCounts(state *store.NamespaceState) (map[string]int, error) {\n\tcounts := make(map[string]int, len(state.ShardAssignments))\n\tfor executorID, assignment := range state.ShardAssignments {\n\t\texecutorState, ok := state.Executors[executorID]\n\t\tif !ok || executorState.Status != types.ExecutorStatusACTIVE {\n\t\t\tcontinue\n\t\t}\n\t\tcounts[executorID] = len(assignment.AssignedShards)\n\t}\n\treturn counts, nil\n}\n\n// pickExecutors assigns each shard key to the least-loaded active executor,\n// updating the in-batch running count after every pick to spread load evenly.\n// It returns a map of shardKey -> chosen executorID.\nfunc pickExecutors(namespace string, shardKeys []string, assignedCounts map[string]int) (map[string]string, error) {\n\tchosenExecutors := make(map[string]string, len(shardKeys))\n\tfor _, shardKey := range shardKeys {\n\t\tchosenExecutor := \"\"\n\t\tminCount := math.MaxInt\n\t\tfor executorID, count := range assignedCounts {\n\t\t\tif count < minCount {\n\t\t\t\tminCount = count\n\t\t\t\tchosenExecutor = executorID\n\t\t\t}\n\t\t}\n\t\tif chosenExecutor == \"\" {\n\t\t\treturn nil, &types.InternalServiceError{Message: \"no active executors available for namespace: \" + namespace}\n\t\t}\n\t\tchosenExecutors[shardKey] = chosenExecutor\n\t\tassignedCounts[chosenExecutor]++\n\t}\n\treturn chosenExecutors, nil\n}\n\n// mergeAssignments folds the chosen shard→executor assignments back into state.\n// The AssignedShards maps are copied to avoid mutating the object returned by\n// GetState.\nfunc mergeAssignments(state *store.NamespaceState, chosenExecutors map[string]string) {\n\tfor executorID, shardsForExecutor := range invertMap(chosenExecutors) {\n\t\texisting := state.ShardAssignments[executorID]\n\t\tnewShards := make(map[string]*types.ShardAssignment, len(existing.AssignedShards)+len(shardsForExecutor))\n\t\tfor k, v := range existing.AssignedShards {\n\t\t\tnewShards[k] = v\n\t\t}\n\t\tfor _, shardKey := range shardsForExecutor {\n\t\t\tnewShards[shardKey] = &types.ShardAssignment{Status: types.AssignmentStatusREADY}\n\t\t}\n\t\texisting.AssignedShards = newShards\n\t\tstate.ShardAssignments[executorID] = existing\n\t}\n}\n\n// fetchExecutorMetadata calls GetExecutor once per unique chosen executor to\n// retrieve metadata. Metadata is stored separately from HeartbeatState and is\n// not returned by GetState.\nfunc (h *handlerImpl) fetchExecutorMetadata(ctx context.Context, namespace string, chosenExecutors map[string]string) (map[string]*store.ShardOwner, error) {\n\texecutorOwners := make(map[string]*store.ShardOwner, len(chosenExecutors))\n\tfor _, executorID := range chosenExecutors {\n\t\tif _, already := executorOwners[executorID]; already {\n\t\t\tcontinue\n\t\t}\n\t\towner, err := h.storage.GetExecutor(ctx, namespace, executorID)\n\t\tif err != nil {\n\t\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"get executor %q: %v\", executorID, err)}\n\t\t}\n\t\texecutorOwners[executorID] = owner\n\t}\n\treturn executorOwners, nil\n}\n\n// buildResults constructs the shardKey -> GetShardOwnerResponse map from the\n// chosen executors and their fetched metadata.\nfunc buildResults(namespace string, shardKeys []string, chosenExecutors map[string]string, executorOwners map[string]*store.ShardOwner) map[string]*types.GetShardOwnerResponse {\n\tresults := make(map[string]*types.GetShardOwnerResponse, len(shardKeys))\n\tfor _, shardKey := range shardKeys {\n\t\texecutorID := chosenExecutors[shardKey]\n\t\towner := executorOwners[executorID]\n\t\tresults[shardKey] = &types.GetShardOwnerResponse{\n\t\t\tOwner:     owner.ExecutorID,\n\t\t\tNamespace: namespace,\n\t\t\tMetadata:  owner.Metadata,\n\t\t}\n\t}\n\treturn results\n}\n\n// invertMap turns map[shardKey]executorID into map[executorID][]shardKey.\nfunc invertMap(m map[string]string) map[string][]string {\n\tout := make(map[string][]string)\n\tfor shardKey, executorID := range m {\n\t\tout[executorID] = append(out[executorID], shardKey)\n\t}\n\treturn out\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/ephemeral_assigner_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nfunc TestAssignEphemeralBatch(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tnamespace      string\n\t\tshardKeys      []string\n\t\tsetupMocks     func(mockStore *store.MockStore)\n\t\texpectedOwners map[string]string // shardKey -> expected owner\n\t\texpectedError  bool\n\t\texpectedErrMsg string\n\t}{\n\t\t{\n\t\t\tname:      \"AssignsToLeastLoadedExecutor\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"NON-EXISTING-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors: map[string]store.HeartbeatState{\n\t\t\t\t\t\t\"owner1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\t\t\"owner2\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\t},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\t\t\"owner1\": {\n\t\t\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\t\"shard1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t\t\"shard2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t\t\"shard3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"owner2\": {\n\t\t\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\t\"shard4\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\t// owner2 has the fewest shards; expect a single batch AssignShards call.\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockStore.EXPECT().GetExecutor(gomock.Any(), _testNamespaceEphemeral, \"owner2\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner2\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwners: map[string]string{\"NON-EXISTING-SHARD\": \"owner2\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"SkipDrainingExecutor\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"NON-EXISTING-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors: map[string]store.HeartbeatState{\n\t\t\t\t\t\t\"owner1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\t\t// owner2 is DRAINING, should be skipped even though it has fewer shards\n\t\t\t\t\t\t\"owner2\": {Status: types.ExecutorStatusDRAINING},\n\t\t\t\t\t},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\t\t\"owner1\": {\n\t\t\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\t\"shard1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t\t\"shard2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t\t\"shard3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"owner2\": {\n\t\t\t\t\t\t\t// owner2 has fewer shards but is DRAINING, so should be skipped\n\t\t\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\t\t\"shard4\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t\t// owner1 should be selected; single batch write\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockStore.EXPECT().GetExecutor(gomock.Any(), _testNamespaceEphemeral, \"owner1\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwners: map[string]string{\"NON-EXISTING-SHARD\": \"owner1\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"GetStateFailure\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"NON-EXISTING-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(nil, errors.New(\"get state failure\"))\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"get state failure\",\n\t\t},\n\t\t{\n\t\t\t// When two batches race and the first wins, the second gets\n\t\t\t// ErrVersionConflict from AssignShards. This is surfaced as an\n\t\t\t// internal error; the caller is expected to retry GetShardOwner,\n\t\t\t// which will then find the shard already assigned in the cache.\n\t\t\tname:      \"VersionConflict\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"CONCURRENT-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}}}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(store.ErrVersionConflict)\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"version conflict\",\n\t\t},\n\t\t{\n\t\t\tname:      \"NoActiveExecutors\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"NON-EXISTING-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\t// All executors are DRAINING — none are eligible for assignment.\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors: map[string]store.HeartbeatState{\n\t\t\t\t\t\t\"owner1\": {Status: types.ExecutorStatusDRAINING},\n\t\t\t\t\t\t\"owner2\": {Status: types.ExecutorStatusDRAINING},\n\t\t\t\t\t},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\t\t\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}},\n\t\t\t\t\t\t\"owner2\": {AssignedShards: map[string]*types.ShardAssignment{}},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"no active executors available for namespace\",\n\t\t},\n\t\t{\n\t\t\tname:      \"AssignShardsFailure\",\n\t\t\tnamespace: _testNamespaceEphemeral,\n\t\t\tshardKeys: []string{\"NON-EXISTING-SHARD\"},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}}}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(errors.New(\"assign shards failure\"))\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"assign shards failure\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockStorage := store.NewMockStore(ctrl)\n\n\t\t\th := &handlerImpl{\n\t\t\t\tlogger:  testlogger.New(t),\n\t\t\t\tstorage: mockStorage,\n\t\t\t}\n\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(mockStorage)\n\t\t\t}\n\n\t\t\tresults, err := h.assignEphemeralBatch(context.Background(), tt.namespace, tt.shardKeys)\n\t\t\tif tt.expectedError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.Contains(t, err.Error(), tt.expectedErrMsg)\n\t\t\t\trequire.Nil(t, results)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\trequire.Len(t, results, len(tt.expectedOwners))\n\t\t\t\tfor shardKey, expectedOwner := range tt.expectedOwners {\n\t\t\t\t\trequire.Equal(t, expectedOwner, results[shardKey].Owner)\n\t\t\t\t\trequire.Equal(t, tt.namespace, results[shardKey].Namespace)\n\t\t\t\t\trequire.Equal(t, map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"}, results[shardKey].Metadata)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/executor.go",
    "content": "package handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t_maxMetadataKeys      = 32\n\t_maxMetadataKeyLength = 128\n\t_maxMetadataValueSize = 512 * 1024 // 512KB\n)\n\ntype executor struct {\n\tlogger               log.Logger\n\ttimeSource           clock.TimeSource\n\tstorage              store.Store\n\tshardDistributionCfg config.ShardDistribution\n\tcfg                  *config.Config\n\tmetricsClient        metrics.Client\n}\n\nfunc NewExecutorHandler(\n\tlogger log.Logger,\n\tstorage store.Store,\n\ttimeSource clock.TimeSource,\n\tshardDistributionCfg config.ShardDistribution,\n\tcfg *config.Config,\n\tmetricsClient metrics.Client,\n) Executor {\n\treturn &executor{\n\t\tlogger:               logger,\n\t\ttimeSource:           timeSource,\n\t\tstorage:              storage,\n\t\tshardDistributionCfg: shardDistributionCfg,\n\t\tcfg:                  cfg,\n\t\tmetricsClient:        metricsClient,\n\t}\n}\n\nfunc (h *executor) Heartbeat(ctx context.Context, request *types.ExecutorHeartbeatRequest) (*types.ExecutorHeartbeatResponse, error) {\n\tpreviousHeartbeat, assignedShards, err := h.storage.GetHeartbeat(ctx, request.Namespace, request.ExecutorID)\n\t// We ignore Executor not found errors, since it just means that this executor heartbeat the first time.\n\tif err != nil && !errors.Is(err, store.ErrExecutorNotFound) {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"failed to get heartbeat: %v\", err)}\n\t}\n\n\theartbeatTime := h.timeSource.Now().UTC()\n\tmode := h.cfg.GetMigrationMode(request.Namespace)\n\tshardAssignedInBackground := true\n\n\tswitch mode {\n\tcase types.MigrationModeINVALID:\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"namespace's migration mode is invalid: %v\", err)}\n\tcase types.MigrationModeLOCALPASSTHROUGH:\n\t\th.logger.Info(\"Migration mode is local passthrough, no calls to heartbeat should be allowed\", tag.ShardNamespace(request.Namespace), tag.ShardExecutor(request.ExecutorID))\n\t\treturn _convertResponse(nil, mode), nil\n\t// From SD perspective the behaviour is the same\n\tcase types.MigrationModeLOCALPASSTHROUGHSHADOW, types.MigrationModeDISTRIBUTEDPASSTHROUGH:\n\t\tassignedShards, err = h.assignShardsInCurrentHeartbeat(ctx, request, assignedShards)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tshardAssignedInBackground = false\n\t}\n\n\tnewHeartbeat := store.HeartbeatState{\n\t\tLastHeartbeat:  heartbeatTime,\n\t\tStatus:         request.Status,\n\t\tReportedShards: request.ShardStatusReports,\n\t\tMetadata:       request.GetMetadata(),\n\t}\n\n\tif err := validateMetadata(newHeartbeat.Metadata); err != nil {\n\t\treturn nil, types.BadRequestError{Message: fmt.Sprintf(\"invalid metadata: %s\", err)}\n\t}\n\n\terr = h.storage.RecordHeartbeat(ctx, request.Namespace, request.ExecutorID, newHeartbeat)\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"failed to record heartbeat: %v\", err)}\n\t}\n\n\t// emit shard assignment metrics only if shards are assigned in the background\n\t// shard assignment in heartbeat doesn't involve any assignment changes happening in the background\n\t// thus there was no shard handover and no assignment distribution latency\n\t// to measure, so don't need to emit metrics in that case\n\tif shardAssignedInBackground {\n\t\th.emitShardAssignmentMetrics(request.Namespace, heartbeatTime, previousHeartbeat, assignedShards)\n\t}\n\n\treturn _convertResponse(assignedShards, mode), nil\n}\n\n// emitShardAssignmentMetrics emits the following metrics for newly assigned shards:\n// - ShardAssignmentDistributionLatency: time taken since the shard was assigned to heartbeat time\n// - ShardHandoverLatency: time taken since the previous executor's last heartbeat to heartbeat time\nfunc (h *executor) emitShardAssignmentMetrics(namespace string, heartbeatTime time.Time, previousHeartbeat *store.HeartbeatState, assignedState *store.AssignedState) {\n\t// find newly assigned shards, if there are none, no handovers happened\n\tnewAssignedShardIDs := filterNewlyAssignedShardIDs(previousHeartbeat, assignedState)\n\tif len(newAssignedShardIDs) == 0 {\n\t\t// no need to emit ShardDistributorShardAssignmentDistributionLatency due to no handovers\n\t\treturn\n\t}\n\n\tmetricsScope := h.metricsClient.Scope(metrics.ShardDistributorHeartbeatScope).\n\t\tTagged(metrics.NamespaceTag(namespace))\n\n\tdistributionLatency := heartbeatTime.Sub(assignedState.LastUpdated)\n\tmetricsScope.RecordHistogramDuration(metrics.ShardDistributorShardAssignmentDistributionLatency, distributionLatency)\n\n\tfor _, shardID := range newAssignedShardIDs {\n\t\thandoverStats, ok := assignedState.ShardHandoverStats[shardID]\n\t\tif !ok {\n\t\t\t// no handover stats for this shard, means it was never handed over before\n\t\t\t// so no handover latency metric to emit\n\t\t\tcontinue\n\t\t}\n\n\t\thandoverLatency := heartbeatTime.Sub(handoverStats.PreviousExecutorLastHeartbeatTime)\n\t\tmetricsScope.Tagged(metrics.HandoverTypeTag(handoverStats.HandoverType.String())).\n\t\t\tRecordHistogramDuration(metrics.ShardDistributorShardHandoverLatency, handoverLatency)\n\n\t}\n}\n\n// assignShardsInCurrentHeartbeat is used during the migration phase to assign the shards to the executors according to what is reported during the heartbeat\nfunc (h *executor) assignShardsInCurrentHeartbeat(ctx context.Context, request *types.ExecutorHeartbeatRequest, assignedShards *store.AssignedState) (*store.AssignedState, error) {\n\tmodRevision := int64(0)\n\tif assignedShards != nil {\n\t\tmodRevision = assignedShards.ModRevision\n\t}\n\tnewState := store.AssignedState{\n\t\tAssignedShards: make(map[string]*types.ShardAssignment),\n\t\tLastUpdated:    h.timeSource.Now().UTC(),\n\t\tModRevision:    modRevision,\n\t}\n\n\tfor shard := range request.GetShardStatusReports() {\n\t\tnewState.AssignedShards[shard] = &types.ShardAssignment{\n\t\t\tStatus: types.AssignmentStatusREADY,\n\t\t}\n\t}\n\tassignShardsRequest := store.AssignShardsRequest{\n\t\tNewState: &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\trequest.GetExecutorID(): newState,\n\t\t\t},\n\t\t},\n\t}\n\terr := h.storage.AssignShards(ctx, request.GetNamespace(), assignShardsRequest, store.NopGuard())\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"failed to assign shards in the current heartbeat: %v\", err)}\n\t}\n\treturn &newState, nil\n}\n\nfunc _convertResponse(shards *store.AssignedState, mode types.MigrationMode) *types.ExecutorHeartbeatResponse {\n\tres := &types.ExecutorHeartbeatResponse{}\n\tres.MigrationMode = mode\n\tif shards == nil {\n\t\treturn res\n\t}\n\tres.ShardAssignments = shards.AssignedShards\n\treturn res\n}\n\nfunc validateMetadata(metadata map[string]string) error {\n\tif len(metadata) > _maxMetadataKeys {\n\t\treturn fmt.Errorf(\"metadata has %d keys, which exceeds the maximum of %d\", len(metadata), _maxMetadataKeys)\n\t}\n\n\tfor key, value := range metadata {\n\t\tif len(key) > _maxMetadataKeyLength {\n\t\t\treturn fmt.Errorf(\"metadata key %q has length %d, which exceeds the maximum of %d\", key, len(key), _maxMetadataKeyLength)\n\t\t}\n\n\t\tif len(value) > _maxMetadataValueSize {\n\t\t\treturn fmt.Errorf(\"metadata value for key %q has size %d bytes, which exceeds the maximum of %d bytes\", key, len(value), _maxMetadataValueSize)\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc filterNewlyAssignedShardIDs(previousHeartbeat *store.HeartbeatState, assignedState *store.AssignedState) []string {\n\t// if assignedState is nil, no shards are assigned\n\tif assignedState == nil {\n\t\treturn nil\n\t}\n\n\tvar newAssignedShardIDs []string\n\tfor assignedShardID := range assignedState.AssignedShards {\n\t\tif previousHeartbeat == nil || !shardInReportedShards(previousHeartbeat.ReportedShards, assignedShardID) {\n\t\t\tnewAssignedShardIDs = append(newAssignedShardIDs, assignedShardID)\n\t\t}\n\t}\n\n\treturn newAssignedShardIDs\n}\n\nfunc shardInReportedShards(reportedShards map[string]*types.ShardStatusReport, shardID string) bool {\n\t_, ok := reportedShards[shardID]\n\treturn ok\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/executor_test.go",
    "content": "package handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmetricmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nfunc TestHeartbeat(t *testing.T) {\n\tctx := context.Background()\n\tnamespace := \"test-namespace\"\n\texecutorID := \"test-executor\"\n\tnow := time.Now().UTC()\n\n\t// Test Case 1: First Heartbeat\n\tt.Run(\"FirstHeartbeat\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSourceAt(now)\n\t\tshardDistributionCfg := config.ShardDistribution{}\n\t\tcfg := newConfig(t, []configEntry{})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(nil, nil, store.ErrExecutorNotFound)\n\t\tmockStore.EXPECT().RecordHeartbeat(gomock.Any(), namespace, executorID, store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t})\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.NoError(t, err)\n\t})\n\n\t// Test Case 2: Subsequent heartbeat records a new heartbeat\n\tt.Run(\"SubsequentHeartbeatRecords\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSourceAt(now)\n\t\tshardDistributionCfg := config.ShardDistribution{}\n\t\tcfg := newConfig(t, []configEntry{})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, nil, nil)\n\t\tmockStore.EXPECT().RecordHeartbeat(gomock.Any(), namespace, executorID, store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t})\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.NoError(t, err)\n\t})\n\n\t// Test Case 3: Status Change (with update)\n\tt.Run(\"StatusChange\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSourceAt(now)\n\t\tshardDistributionCfg := config.ShardDistribution{}\n\t\tcfg := newConfig(t, []configEntry{})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusDRAINING, // Status changed\n\t\t}\n\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, nil, nil)\n\t\tmockStore.EXPECT().RecordHeartbeat(gomock.Any(), namespace, executorID, store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\t})\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.NoError(t, err)\n\t})\n\n\t// Test Case 4: Storage Error\n\tt.Run(\"StorageError\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\tshardDistributionCfg := config.ShardDistribution{}\n\t\tcfg := newConfig(t, []configEntry{})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\texpectedErr := errors.New(\"storage is down\")\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(nil, nil, expectedErr)\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.Error(t, err)\n\t\trequire.Contains(t, err.Error(), expectedErr.Error())\n\t})\n\n\t// Test Case 5: Heartbeat with executor associated invalid migration mode\n\tt.Run(\"MigrationModeInvald\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\tshardDistributionCfg := config.ShardDistribution{\n\t\t\tNamespaces: []config.Namespace{{Name: namespace, Mode: config.MigrationModeINVALID}},\n\t\t}\n\t\tcfg := newConfig(t, []configEntry{{dynamicproperties.ShardDistributorMigrationMode, config.MigrationModeINVALID}})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t}\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\texpectedErr := errors.New(\"migration mode is invalid\")\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, nil, nil)\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.Error(t, err)\n\t\trequire.Contains(t, err.Error(), expectedErr.Error())\n\t})\n\n\t// Test Case 6: Heartbeat with executor associated with local passthrough mode\n\tt.Run(\"MigrationModeLocalPassthrough\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\tshardDistributionCfg := config.ShardDistribution{\n\t\t\tNamespaces: []config.Namespace{{Name: namespace, Mode: config.MigrationModeLOCALPASSTHROUGH}},\n\t\t}\n\t\tcfg := newConfig(t, []configEntry{{dynamicproperties.ShardDistributorMigrationMode, config.MigrationModeLOCALPASSTHROUGH}})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t}\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, nil, nil)\n\n\t\tresp, err := handler.Heartbeat(ctx, req)\n\t\trequire.NoError(t, err)\n\t\trequire.Equal(t, types.MigrationModeLOCALPASSTHROUGH, resp.MigrationMode)\n\t})\n\n\t// Test Case 7: Heartbeat with executor associated with local passthrough shadow\n\tt.Run(\"MigrationModeLocalPassthroughWithAssignmentChanges\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\tshardDistributionCfg := config.ShardDistribution{\n\t\t\tNamespaces: []config.Namespace{{Name: namespace, Mode: config.MigrationModeLOCALPASSTHROUGHSHADOW}},\n\t\t}\n\t\tcfg := newConfig(t, []configEntry{{dynamicproperties.ShardDistributorMigrationMode, config.MigrationModeLOCALPASSTHROUGHSHADOW}})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t\tShardStatusReports: map[string]*types.ShardStatusReport{\n\t\t\t\t\"shard0\": {Status: types.ShardStatusREADY, ShardLoad: 1.0},\n\t\t\t},\n\t\t}\n\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\"shard1\": {Status: types.ShardStatusREADY, ShardLoad: 1.0},\n\t\t\t},\n\t\t}\n\n\t\tassignedState := store.AssignedState{\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"shard1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, &assignedState, nil)\n\t\tmockStore.EXPECT().AssignShards(gomock.Any(), namespace, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\t\tfunc(ctx context.Context, namespace string, request store.AssignShardsRequest, guard store.GuardFunc) error {\n\t\t\t\t// Expect to Assign the shard in the request\n\t\t\t\texpectedRequest := store.AssignShardsRequest{\n\t\t\t\t\tNewState: &store.NamespaceState{\n\t\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\t\t\texecutorID: {AssignedShards: map[string]*types.ShardAssignment{\"shard0\": {Status: types.AssignmentStatusREADY}}},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\trequire.Equal(t, expectedRequest.NewState.ShardAssignments[executorID].AssignedShards, request.NewState.ShardAssignments[executorID].AssignedShards)\n\t\t\t\treturn nil\n\t\t\t},\n\t\t)\n\n\t\tmockStore.EXPECT().RecordHeartbeat(gomock.Any(), namespace, executorID, gomock.AssignableToTypeOf(store.HeartbeatState{})).DoAndReturn(\n\t\t\tfunc(_ context.Context, _ string, _ string, hb store.HeartbeatState) error {\n\t\t\t\t// Validate status and reported shards, ignore exact timestamp\n\t\t\t\trequire.Equal(t, types.ExecutorStatusACTIVE, hb.Status)\n\t\t\t\trequire.Contains(t, hb.ReportedShards, \"shard0\")\n\t\t\t\treturn nil\n\t\t\t},\n\t\t)\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.NoError(t, err)\n\t},\n\t)\n\n\t// Test Case 8: Heartbeat with executor associated with distributed passthrough\n\tt.Run(\"MigrationModeDISTRIBUTEDPASSTHROUGHAssignmentFailure\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSource()\n\t\tshardDistributionCfg := config.ShardDistribution{\n\t\t\tNamespaces: []config.Namespace{{Name: namespace, Mode: config.MigrationModeLOCALPASSTHROUGHSHADOW}},\n\t\t}\n\t\tcfg := newConfig(t, []configEntry{{dynamicproperties.ShardDistributorMigrationMode, config.MigrationModeLOCALPASSTHROUGHSHADOW}})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t\tShardStatusReports: map[string]*types.ShardStatusReport{\n\t\t\t\t\"shard0\": {Status: types.ShardStatusREADY, ShardLoad: 1.0},\n\t\t\t},\n\t\t}\n\n\t\tpreviousHeartbeat := store.HeartbeatState{\n\t\t\tLastHeartbeat: now,\n\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\"shard1\": {Status: types.ShardStatusREADY, ShardLoad: 1.0},\n\t\t\t},\n\t\t}\n\n\t\tassignedState := store.AssignedState{\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"shard1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(&previousHeartbeat, &assignedState, nil)\n\t\texpectedErr := errors.New(\"assignemnt failed\")\n\t\tmockStore.EXPECT().AssignShards(gomock.Any(), namespace, gomock.Any(), gomock.Any()).Return(expectedErr)\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.Error(t, err)\n\t\trequire.Contains(t, err.Error(), expectedErr.Error())\n\t})\n\n\t// Test Case 9: Heartbeat with metadata validation failure - too many keys\n\tt.Run(\"MetadataValidationTooManyKeys\", func(t *testing.T) {\n\t\tctrl := gomock.NewController(t)\n\t\tmockStore := store.NewMockStore(ctrl)\n\t\tmockTimeSource := clock.NewMockedTimeSourceAt(now)\n\t\tshardDistributionCfg := config.ShardDistribution{}\n\t\tcfg := newConfig(t, []configEntry{})\n\t\thandler := NewExecutorHandler(testlogger.New(t), mockStore, mockTimeSource, shardDistributionCfg, cfg, metrics.NoopClient)\n\n\t\t// Create metadata with more than max allowed keys\n\t\tmetadata := make(map[string]string)\n\t\tfor i := 0; i < _maxMetadataKeys+1; i++ {\n\t\t\tmetadata[string(rune('a'+i))] = \"value\"\n\t\t}\n\n\t\treq := &types.ExecutorHeartbeatRequest{\n\t\t\tNamespace:  namespace,\n\t\t\tExecutorID: executorID,\n\t\t\tStatus:     types.ExecutorStatusACTIVE,\n\t\t\tMetadata:   metadata,\n\t\t}\n\n\t\tmockStore.EXPECT().GetHeartbeat(gomock.Any(), namespace, executorID).Return(nil, nil, store.ErrExecutorNotFound)\n\n\t\t_, err := handler.Heartbeat(ctx, req)\n\t\trequire.Error(t, err)\n\t\trequire.Contains(t, err.Error(), \"invalid metadata: metadata has 33 keys, which exceeds the maximum of 32\")\n\t})\n\n}\n\nfunc TestValidateMetadata(t *testing.T) {\n\t// Helper function to generate metadata with N keys\n\tmakeMetadataWithKeys := func(n int) map[string]string {\n\t\tmetadata := make(map[string]string)\n\t\tfor i := 0; i < n; i++ {\n\t\t\tmetadata[string(rune('a'+i))] = \"value\"\n\t\t}\n\t\treturn metadata\n\t}\n\n\ttestCases := []struct {\n\t\tname           string\n\t\tmetadata       map[string]string\n\t\texpectError    bool\n\t\terrorSubstring string\n\t}{\n\t\t{\n\t\t\tname: \"ValidMetadata\",\n\t\t\tmetadata: map[string]string{\n\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t\"key2\": \"value2\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"EmptyMetadata\",\n\t\t\tmetadata:    map[string]string{},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"NilMetadata\",\n\t\t\tmetadata:    nil,\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:           \"TooManyKeys\",\n\t\t\tmetadata:       makeMetadataWithKeys(_maxMetadataKeys + 1),\n\t\t\texpectError:    true,\n\t\t\terrorSubstring: \"exceeds the maximum of 32\",\n\t\t},\n\t\t{\n\t\t\tname:        \"ExactlyMaxKeys\",\n\t\t\tmetadata:    makeMetadataWithKeys(_maxMetadataKeys),\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"KeyTooLong\",\n\t\t\tmetadata: map[string]string{\n\t\t\t\tstring(make([]byte, _maxMetadataKeyLength+1)): \"value\",\n\t\t\t},\n\t\t\texpectError:    true,\n\t\t\terrorSubstring: \"exceeds the maximum of 128\",\n\t\t},\n\t\t{\n\t\t\tname: \"KeyExactlyMaxLength\",\n\t\t\tmetadata: map[string]string{\n\t\t\t\tstring(make([]byte, _maxMetadataKeyLength)): \"value\",\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"ValueTooLarge\",\n\t\t\tmetadata: map[string]string{\n\t\t\t\t\"key\": string(make([]byte, _maxMetadataValueSize+1)),\n\t\t\t},\n\t\t\texpectError:    true,\n\t\t\terrorSubstring: \"exceeds the maximum of 524288 bytes\",\n\t\t},\n\t\t{\n\t\t\tname: \"ValueExactlyMaxSize\",\n\t\t\tmetadata: map[string]string{\n\t\t\t\t\"key\": string(make([]byte, _maxMetadataValueSize)),\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"MultipleValidationErrors\",\n\t\t\tmetadata: func() map[string]string {\n\t\t\t\tmetadata := makeMetadataWithKeys(_maxMetadataKeys + 1)\n\t\t\t\tlongKey := string(make([]byte, _maxMetadataKeyLength+1))\n\t\t\t\tmetadata[longKey] = \"value\"\n\t\t\t\treturn metadata\n\t\t\t}(),\n\t\t\texpectError:    true,\n\t\t\terrorSubstring: \"exceeds the maximum of 32\", // First validation error\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\terr := validateMetadata(tc.metadata)\n\t\t\tif tc.expectError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.Contains(t, err.Error(), tc.errorSubstring)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConvertResponse(t *testing.T) {\n\ttestCases := []struct {\n\t\tname         string\n\t\tinput        *store.AssignedState\n\t\texpectedResp *types.ExecutorHeartbeatResponse\n\t}{\n\t\t{\n\t\t\tname:  \"Nil input\",\n\t\t\tinput: nil,\n\t\t\texpectedResp: &types.ExecutorHeartbeatResponse{\n\t\t\t\tShardAssignments: make(map[string]*types.ShardAssignment),\n\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Empty input\",\n\t\t\tinput: &store.AssignedState{\n\t\t\t\tAssignedShards: make(map[string]*types.ShardAssignment),\n\t\t\t},\n\t\t\texpectedResp: &types.ExecutorHeartbeatResponse{\n\t\t\t\tShardAssignments: make(map[string]*types.ShardAssignment),\n\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Populated input\",\n\t\t\tinput: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t},\n\t\t\texpectedResp: &types.ExecutorHeartbeatResponse{\n\t\t\t\tShardAssignments: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t\tMigrationMode:    types.MigrationModeONBOARDED,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\t// In Go, you can't initialize a map in a struct to nil directly,\n\t\t\t// so we handle the nil case for ShardAssignments separately for comparison.\n\t\t\tif tc.expectedResp.ShardAssignments == nil {\n\t\t\t\ttc.expectedResp.ShardAssignments = make(map[string]*types.ShardAssignment)\n\t\t\t}\n\t\t\tres := _convertResponse(tc.input, types.MigrationModeONBOARDED)\n\n\t\t\t// Ensure ShardAssignments is not nil for comparison purposes\n\t\t\tif res.ShardAssignments == nil {\n\t\t\t\tres.ShardAssignments = make(map[string]*types.ShardAssignment)\n\t\t\t}\n\t\t\trequire.Equal(t, tc.expectedResp, res)\n\t\t})\n\t}\n}\n\ntype configEntry struct {\n\tkey   dynamicproperties.Key\n\tvalue interface{}\n}\n\nfunc newConfig(t *testing.T, configEntries []configEntry) *config.Config {\n\tclient := dynamicconfig.NewInMemoryClient()\n\tfor _, entry := range configEntries {\n\t\terr := client.UpdateValue(entry.key, entry.value)\n\t\tif err != nil {\n\t\t\tt.Errorf(\"Failed to update config \")\n\t\t}\n\t}\n\tdc := dynamicconfig.NewCollection(client, testlogger.New(t))\n\treturn config.NewConfig(dc)\n}\n\nfunc TestFilterNewlyAssignedShardIDs(t *testing.T) {\n\ttype testCase struct {\n\t\tname     string\n\t\tprevious *store.HeartbeatState\n\t\tassigned *store.AssignedState\n\t\texpected []string\n\t}\n\ttests := []testCase{\n\t\t{\n\t\t\tname:     \"nil previousHeartbeat returns all assigned\",\n\t\t\tprevious: nil,\n\t\t\tassigned: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t\t\"shard2\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []string{\"shard1\", \"shard2\"},\n\t\t},\n\t\t{\n\t\t\tname: \"no new assigned shards\",\n\t\t\tprevious: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t\t\"shard2\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassigned: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t\t\"shard2\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"some new assigned shards\",\n\t\t\tprevious: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassigned: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t\t\"shard2\": {},\n\t\t\t\t\t\"shard3\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: []string{\"shard2\", \"shard3\"},\n\t\t},\n\t\t{\n\t\t\tname: \"empty assigned returns empty\",\n\t\t\tprevious: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassigned: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{},\n\t\t\t},\n\t\t\texpected: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"nil assignedState returns nil\",\n\t\t\tprevious: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard1\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassigned: nil,\n\t\t\texpected: nil,\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := filterNewlyAssignedShardIDs(tt.previous, tt.assigned)\n\t\t\trequire.ElementsMatch(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestEmitShardAssignmentMetrics(t *testing.T) {\n\theartbeatTime := time.Now().UTC()\n\tnamespace := \"test-namespace\"\n\tshardID := \"shard-1\"\n\temptyHeartbeatState := &store.HeartbeatState{ReportedShards: map[string]*types.ShardStatusReport{}}\n\n\ttype expectHandoverMetric struct {\n\t\tLatency      time.Duration\n\t\tHandoverType types.HandoverType\n\t}\n\n\ttype testCase struct {\n\t\tname              string\n\t\tpreviousHeartbeat *store.HeartbeatState\n\t\tassignedState     *store.AssignedState\n\n\t\texpectedDistributionLatency *time.Duration\n\t\texpectedHandoverLatencies   []*expectHandoverMetric\n\t}\n\n\ttestCases := []testCase{\n\t\t{\n\t\t\tname:                        \"no new assigned shards\",\n\t\t\tpreviousHeartbeat:           &store.HeartbeatState{ReportedShards: map[string]*types.ShardStatusReport{shardID: {}}},\n\t\t\tassignedState:               &store.AssignedState{AssignedShards: makeReadyAssignedShards(shardID)},\n\t\t\texpectedDistributionLatency: nil,\n\t\t\texpectedHandoverLatencies:   nil,\n\t\t},\n\t\t{\n\t\t\tname:              \"newly assigned shard with handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(shardID),\n\t\t\t\tLastUpdated:    heartbeatTime.Add(-10 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\tshardID: {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-20 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(10 * time.Second),\n\t\t\texpectedHandoverLatencies:   []*expectHandoverMetric{{Latency: 20 * time.Second, HandoverType: types.HandoverTypeGRACEFUL}},\n\t\t},\n\t\t{\n\t\t\tname:              \"one assigned shard with no handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards:     makeReadyAssignedShards(shardID),\n\t\t\t\tLastUpdated:        heartbeatTime.Add(-5 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(5 * time.Second),\n\t\t\texpectedHandoverLatencies:   nil,\n\t\t},\n\t\t{\n\t\t\tname:              \"one assigned shard with nil handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards:     makeReadyAssignedShards(shardID),\n\t\t\t\tLastUpdated:        heartbeatTime.Add(-5 * time.Second),\n\t\t\t\tShardHandoverStats: nil,\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(5 * time.Second),\n\t\t\texpectedHandoverLatencies:   nil,\n\t\t},\n\t\t{\n\t\t\tname:              \"multiple newly assigned shards with handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t\tLastUpdated:    heartbeatTime.Add(-15 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"shard-1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-25 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t\t\"shard-2\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-30 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeEMERGENCY,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(15 * time.Second),\n\t\t\texpectedHandoverLatencies: []*expectHandoverMetric{\n\t\t\t\t{Latency: 30 * time.Second, HandoverType: types.HandoverTypeGRACEFUL},\n\t\t\t\t{Latency: 25 * time.Second, HandoverType: types.HandoverTypeEMERGENCY},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:              \"multiple newly assigned shards with some handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t\tLastUpdated:    heartbeatTime.Add(-15 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"shard-1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-25 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(15 * time.Second),\n\t\t\texpectedHandoverLatencies:   []*expectHandoverMetric{{Latency: 25 * time.Second, HandoverType: types.HandoverTypeGRACEFUL}},\n\t\t},\n\t\t{\n\t\t\tname:              \"multiple newly assigned shards without handover stats\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t\tLastUpdated:    heartbeatTime.Add(-15 * time.Second),\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(15 * time.Second),\n\t\t\texpectedHandoverLatencies:   nil,\n\t\t},\n\t\t{\n\t\t\tname:              \"nil handover stats with new assigned shard\",\n\t\t\tpreviousHeartbeat: emptyHeartbeatState,\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards:     makeReadyAssignedShards(shardID),\n\t\t\t\tLastUpdated:        heartbeatTime.Add(-5 * time.Second),\n\t\t\t\tShardHandoverStats: nil,\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(5 * time.Second),\n\t\t\texpectedHandoverLatencies:   nil,\n\t\t},\n\t\t{\n\t\t\tname: \"newly assigned shard with previous heartbeat containing reported shards\",\n\t\t\tpreviousHeartbeat: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard-2\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: makeReadyAssignedShards(\"shard-1\", \"shard-2\"),\n\t\t\t\tLastUpdated:    heartbeatTime.Add(-8 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"shard-1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-18 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(8 * time.Second),\n\t\t\texpectedHandoverLatencies:   []*expectHandoverMetric{{Latency: 18 * time.Second, HandoverType: types.HandoverTypeGRACEFUL}},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple new assigned shards, previous heartbeat contains some\",\n\t\t\tpreviousHeartbeat: &store.HeartbeatState{\n\t\t\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\t\t\"shard-2\": {},\n\t\t\t\t},\n\t\t\t},\n\t\t\tassignedState: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard-1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\"shard-2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\"shard-3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: heartbeatTime.Add(-12 * time.Second),\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"shard-1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-22 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t\t\"shard-3\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: heartbeatTime.Add(-30 * time.Second),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeEMERGENCY,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDistributionLatency: common.Ptr(12 * time.Second),\n\t\t\texpectedHandoverLatencies: []*expectHandoverMetric{\n\t\t\t\t{Latency: 22 * time.Second, HandoverType: types.HandoverTypeGRACEFUL},\n\t\t\t\t{Latency: 30 * time.Second, HandoverType: types.HandoverTypeEMERGENCY},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmetricsClient := &metricmocks.Client{}\n\t\t\tmetricsScope := &metricmocks.Scope{}\n\n\t\t\tif tc.expectedDistributionLatency != nil {\n\t\t\t\tmetricsClient.On(\"Scope\", metrics.ShardDistributorHeartbeatScope).Return(metricsScope).Once()\n\t\t\t\tmetricsScope.On(\"Tagged\", metrics.NamespaceTag(namespace)).Return(metricsScope).Once()\n\t\t\t\tmetricsScope.On(\"RecordHistogramDuration\", metrics.ShardDistributorShardAssignmentDistributionLatency, *tc.expectedDistributionLatency).Once()\n\t\t\t}\n\n\t\t\tif tc.expectedHandoverLatencies != nil {\n\t\t\t\tfor _, expected := range tc.expectedHandoverLatencies {\n\t\t\t\t\tmetricsScope.On(\"Tagged\", metrics.HandoverTypeTag(expected.HandoverType.String())).Return(metricsScope)\n\t\t\t\t\tmetricsScope.On(\"RecordHistogramDuration\", metrics.ShardDistributorShardHandoverLatency, expected.Latency).Once()\n\t\t\t\t}\n\t\t\t}\n\n\t\t\texec := &executor{metricsClient: metricsClient, logger: testlogger.New(t)}\n\t\t\texec.emitShardAssignmentMetrics(namespace, heartbeatTime, tc.previousHeartbeat, tc.assignedState)\n\n\t\t\tmetricsClient.AssertExpectations(t)\n\t\t\tmetricsScope.AssertExpectations(t)\n\t\t})\n\t}\n}\n\n// makeReadyAssignedShards is a helper function to create a map of shard assignments with READY status.\nfunc makeReadyAssignedShards(shardIDs ...string) map[string]*types.ShardAssignment {\n\treturn makeAssignedShards(types.AssignmentStatusREADY, shardIDs...)\n}\n\n// makeAssignedShards is a helper function to create a map of shard assignments with the given status.\nfunc makeAssignedShards(status types.AssignmentStatus, shardIDs ...string) map[string]*types.ShardAssignment {\n\tassignedShards := make(map[string]*types.ShardAssignment)\n\tfor _, shardID := range shardIDs {\n\t\tassignedShards[shardID] = &types.ShardAssignment{Status: status}\n\t}\n\treturn assignedShards\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"slices\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t// ephemeralBatchInterval is the time window over which GetShardOwner calls for\n\t// ephemeral namespaces are collected before being processed as a single batch.\n\tephemeralBatchInterval = 100 * time.Millisecond\n\n\t// versionConflictRetryInitialInterval is the starting backoff for retries\n\t// triggered when a concurrent shard assignment causes a version conflict.\n\tversionConflictRetryInitialInterval = 50 * time.Millisecond\n\t// versionConflictRetryMaxInterval caps the per-attempt sleep.\n\tversionConflictRetryMaxInterval = 1 * time.Second\n\t// versionConflictRetryMaxAttempts is the maximum number of retry attempts\n\t// before the error is surfaced to the caller.\n\tversionConflictRetryMaxAttempts = 3\n)\n\nfunc NewHandler(\n\tlogger log.Logger,\n\ttimeSource clock.TimeSource,\n\tshardDistributionCfg config.ShardDistribution,\n\tstorage store.Store,\n) Handler {\n\thandler := &handlerImpl{\n\t\tlogger:               logger,\n\t\tshardDistributionCfg: shardDistributionCfg,\n\t\tstorage:              storage,\n\t}\n\n\thandler.batcher = newShardBatcher(timeSource, ephemeralBatchInterval, handler.assignEphemeralBatch)\n\n\t// prevent us from trying to serve requests before shard distributor is started and ready\n\thandler.startWG.Add(1)\n\treturn handler\n}\n\ntype handlerImpl struct {\n\tlogger log.Logger\n\n\tstartWG sync.WaitGroup\n\n\tstorage              store.Store\n\tshardDistributionCfg config.ShardDistribution\n\n\tbatcher *shardBatcher\n}\n\nfunc (h *handlerImpl) Start() {\n\th.batcher.Start()\n\th.startWG.Done()\n}\n\nfunc (h *handlerImpl) Stop() {\n\th.batcher.Stop()\n}\n\nfunc (h *handlerImpl) Health(ctx context.Context) (*types.HealthStatus, error) {\n\th.startWG.Wait()\n\th.logger.Debug(\"Shard Distributor service health check endpoint reached.\")\n\ths := &types.HealthStatus{Ok: true, Msg: \"shard distributor good\"}\n\treturn hs, nil\n}\n\nfunc (h *handlerImpl) GetShardOwner(ctx context.Context, request *types.GetShardOwnerRequest) (resp *types.GetShardOwnerResponse, retError error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &retError) }()\n\n\tnamespaceIdx := slices.IndexFunc(h.shardDistributionCfg.Namespaces, func(namespace config.Namespace) bool {\n\t\treturn namespace.Name == request.Namespace\n\t})\n\tif namespaceIdx == -1 {\n\t\treturn nil, &types.NamespaceNotFoundError{\n\t\t\tNamespace: request.Namespace,\n\t\t}\n\t}\n\n\tshardOwner, err := h.storage.GetShardOwner(ctx, request.Namespace, request.ShardKey)\n\tif errors.Is(err, store.ErrShardNotFound) {\n\t\tif h.shardDistributionCfg.Namespaces[namespaceIdx].Type == config.NamespaceTypeEphemeral {\n\t\t\treturn h.getOrAssignEphemeralShard(ctx, request)\n\t\t}\n\n\t\treturn nil, &types.ShardNotFoundError{\n\t\t\tNamespace: request.Namespace,\n\t\t\tShardKey:  request.ShardKey,\n\t\t}\n\t}\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"failed to get shard owner: %v\", err)}\n\t}\n\n\treturn &types.GetShardOwnerResponse{\n\t\tOwner:     shardOwner.ExecutorID,\n\t\tMetadata:  shardOwner.Metadata,\n\t\tNamespace: request.Namespace,\n\t}, nil\n}\n\n// getOrAssignEphemeralShard assigns an ephemeral shard that does not yet exist\n// in storage. It submits the request to the batcher and, on a version conflict\n// (concurrent assignment by another goroutine), retries with exponential backoff.\n// Each retry re-reads storage first: if the concurrent writer already committed\n// the assignment we return it immediately without re-submitting to the batcher.\nfunc (h *handlerImpl) getOrAssignEphemeralShard(ctx context.Context, request *types.GetShardOwnerRequest) (*types.GetShardOwnerResponse, error) {\n\tretryPolicy := backoff.NewExponentialRetryPolicy(versionConflictRetryInitialInterval)\n\tretryPolicy.SetMaximumInterval(versionConflictRetryMaxInterval)\n\tretryPolicy.SetMaximumAttempts(versionConflictRetryMaxAttempts)\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(func(err error) bool {\n\t\t\treturn errors.Is(err, store.ErrVersionConflict)\n\t\t}),\n\t)\n\n\tvar resp *types.GetShardOwnerResponse\n\tisRetry := false\n\terr := throttleRetry.Do(ctx, func(ctx context.Context) error {\n\t\tif isRetry {\n\t\t\t// A concurrent batch won the race. Re-read storage first: if the\n\t\t\t// winner already committed our shard's assignment we can return\n\t\t\t// immediately without re-submitting to the batcher.\n\t\t\towner, err := h.storage.GetShardOwner(ctx, request.Namespace, request.ShardKey)\n\t\t\tif err != nil && !errors.Is(err, store.ErrShardNotFound) {\n\t\t\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"failed to get shard owner: %v\", err)}\n\t\t\t}\n\t\t\tif err == nil {\n\t\t\t\tresp = &types.GetShardOwnerResponse{\n\t\t\t\t\tOwner:     owner.ExecutorID,\n\t\t\t\t\tMetadata:  owner.Metadata,\n\t\t\t\t\tNamespace: request.Namespace,\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t}\n\t\t}\n\t\tisRetry = true\n\n\t\t// Submit to the batcher to assign the shard.\n\t\tvar err error\n\t\tresp, err = h.batcher.Submit(ctx, request)\n\t\treturn err\n\t})\n\n\tif err != nil {\n\t\treturn nil, &types.InternalServiceError{Message: fmt.Sprintf(\"failed to assign ephemeral shard: %v\", err)}\n\t}\n\treturn resp, nil\n}\n\nfunc (h *handlerImpl) WatchNamespaceState(request *types.WatchNamespaceStateRequest, server WatchNamespaceStateServer) error {\n\th.startWG.Wait()\n\n\t// Subscribe to state changes from storage\n\tassignmentChangesChan, unSubscribe, err := h.storage.SubscribeToAssignmentChanges(server.Context(), request.Namespace)\n\tdefer unSubscribe()\n\tif err != nil {\n\t\treturn &types.InternalServiceError{Message: fmt.Sprintf(\"failed to subscribe to namespace state: %v\", err)}\n\t}\n\n\t// Stream subsequent updates\n\tfor {\n\t\tselect {\n\t\tcase <-server.Context().Done():\n\t\t\treturn server.Context().Err()\n\t\tcase assignmentChanges, ok := <-assignmentChangesChan:\n\t\t\tif !ok {\n\t\t\t\treturn fmt.Errorf(\"unexpected close of updates channel\")\n\t\t\t}\n\t\t\tresponse := &types.WatchNamespaceStateResponse{\n\t\t\t\tExecutors: make([]*types.ExecutorShardAssignment, 0, len(assignmentChanges)),\n\t\t\t}\n\t\t\tfor executor, shardIDs := range assignmentChanges {\n\t\t\t\tresponse.Executors = append(response.Executors, &types.ExecutorShardAssignment{\n\t\t\t\t\tExecutorID:     executor.ExecutorID,\n\t\t\t\t\tAssignedShards: WrapShards(shardIDs),\n\t\t\t\t\tMetadata:       executor.Metadata,\n\t\t\t\t})\n\t\t\t}\n\n\t\t\terr = server.Send(response)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"send response: %w\", err)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc WrapShards(shardIDs []string) []*types.Shard {\n\tshards := make([]*types.Shard, 0, len(shardIDs))\n\tfor _, shardID := range shardIDs {\n\t\tshards = append(shards, &types.Shard{ShardKey: shardID})\n\t}\n\treturn shards\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/handler_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t_testNamespaceFixed     = \"test-fixed\"\n\t_testNamespaceEphemeral = \"test-ephemeral\"\n)\n\n// newTestHandler creates a handlerImpl wired with a real shardBatcher backed by\n// the provided store mock. The batcher is started and the handler is returned\n// ready to use; callers should call Stop() when done.\nfunc newTestHandler(t *testing.T, cfg config.ShardDistribution, mockStore *store.MockStore) *handlerImpl {\n\tt.Helper()\n\thandler := &handlerImpl{\n\t\tlogger:               testlogger.New(t),\n\t\tshardDistributionCfg: cfg,\n\t\tstorage:              mockStore,\n\t}\n\thandler.batcher = newShardBatcher(clock.NewRealTimeSource(), 10*time.Millisecond, handler.assignEphemeralBatch)\n\thandler.batcher.Start()\n\tt.Cleanup(handler.batcher.Stop)\n\treturn handler\n}\n\nfunc TestGetShardOwner(t *testing.T) {\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{\n\t\t\t\tName:     _testNamespaceFixed,\n\t\t\t\tType:     config.NamespaceTypeFixed,\n\t\t\t\tShardNum: 32,\n\t\t\t},\n\t\t\t{\n\t\t\t\tName: _testNamespaceEphemeral,\n\t\t\t\tType: config.NamespaceTypeEphemeral,\n\t\t\t},\n\t\t},\n\t}\n\n\ttests := []struct {\n\t\tname           string\n\t\trequest        *types.GetShardOwnerRequest\n\t\tsetupMocks     func(mockStore *store.MockStore)\n\t\texpectedOwner  string\n\t\texpectedError  bool\n\t\texpectedErrMsg string\n\t}{\n\t\t{\n\t\t\tname: \"InvalidNamespace\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: \"namespace not found invalidNamespace\",\n\t\t\t\tShardKey:  \"1\",\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"namespace not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"LookupError\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceFixed,\n\t\t\t\tShardKey:  \"1\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceFixed, \"1\").Return(nil, errors.New(\"lookup error\"))\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"lookup error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Existing_Success_Fixed\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceFixed,\n\t\t\t\tShardKey:  \"123\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceFixed, \"123\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwner: \"owner1\",\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"ShardNotFound_Fixed\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceFixed,\n\t\t\t\tShardKey:  \"NON-EXISTING-SHARD\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceFixed, \"NON-EXISTING-SHARD\").Return(nil, store.ErrShardNotFound)\n\t\t\t},\n\t\t\texpectedError:  true,\n\t\t\texpectedErrMsg: \"shard not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"Existing_Success_Ephemeral\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceEphemeral,\n\t\t\t\tShardKey:  \"123\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"123\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwner: \"owner1\",\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\t// ShardNotFound for an ephemeral namespace routes to the batcher, which\n\t\t\t// calls assignEphemeralBatch. This case validates the routing only;\n\t\t\t// detailed assignment behaviour is covered in TestAssignEphemeralBatch.\n\t\t\tname: \"ShardNotFound_Ephemeral_RoutesToBatcher\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceEphemeral,\n\t\t\t\tShardKey:  \"NON-EXISTING-SHARD\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"NON-EXISTING-SHARD\").Return(nil, store.ErrShardNotFound)\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}},\n\t\t\t\t}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockStore.EXPECT().GetExecutor(gomock.Any(), _testNamespaceEphemeral, \"owner1\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwner: \"owner1\",\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\t// A version conflict from AssignShards causes the batcher to return\n\t\t\t// ErrVersionConflict. getOrAssignEphemeralShard re-reads storage on\n\t\t\t// retry; here the concurrent winner has already written the assignment\n\t\t\t// so the second GetShardOwner call succeeds and no second batcher\n\t\t\t// submission is required.\n\t\t\tname: \"Ephemeral_VersionConflict_ResolvedByStorageRead\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceEphemeral,\n\t\t\t\tShardKey:  \"NON-EXISTING-SHARD\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\t// Initial lookup — shard absent.\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"NON-EXISTING-SHARD\").\n\t\t\t\t\tReturn(nil, store.ErrShardNotFound)\n\n\t\t\t\t// Batcher fires: GetState + AssignShards returns a version conflict.\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}},\n\t\t\t\t}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(store.ErrVersionConflict)\n\n\t\t\t\t// Retry: re-read finds the shard already assigned by the concurrent winner.\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"NON-EXISTING-SHARD\").\n\t\t\t\t\tReturn(&store.ShardOwner{\n\t\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwner: \"owner1\",\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\t// A version conflict from AssignShards is retried; on the retry\n\t\t\t// GetShardOwner still returns ErrShardNotFound, so the batcher is\n\t\t\t// re-submitted and this time AssignShards succeeds.\n\t\t\tname: \"Ephemeral_VersionConflict_RetriedAndSucceeds\",\n\t\t\trequest: &types.GetShardOwnerRequest{\n\t\t\t\tNamespace: _testNamespaceEphemeral,\n\t\t\t\tShardKey:  \"NON-EXISTING-SHARD\",\n\t\t\t},\n\t\t\tsetupMocks: func(mockStore *store.MockStore) {\n\t\t\t\t// Initial lookup — shard absent.\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"NON-EXISTING-SHARD\").\n\t\t\t\t\tReturn(nil, store.ErrShardNotFound)\n\n\t\t\t\t// First batcher attempt: version conflict.\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}},\n\t\t\t\t}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(fmt.Errorf(\"assign ephemeral shards: %w\", store.ErrVersionConflict))\n\n\t\t\t\t// Retry re-read — shard still absent, so batcher is submitted again.\n\t\t\t\tmockStore.EXPECT().GetShardOwner(gomock.Any(), _testNamespaceEphemeral, \"NON-EXISTING-SHARD\").\n\t\t\t\t\tReturn(nil, store.ErrShardNotFound)\n\n\t\t\t\t// Second batcher attempt: succeeds.\n\t\t\t\tmockStore.EXPECT().GetState(gomock.Any(), _testNamespaceEphemeral).Return(&store.NamespaceState{\n\t\t\t\t\tExecutors:        map[string]store.HeartbeatState{\"owner1\": {Status: types.ExecutorStatusACTIVE}},\n\t\t\t\t\tShardAssignments: map[string]store.AssignedState{\"owner1\": {AssignedShards: map[string]*types.ShardAssignment{}}},\n\t\t\t\t}, nil)\n\t\t\t\tmockStore.EXPECT().AssignShards(gomock.Any(), _testNamespaceEphemeral, gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t\tmockStore.EXPECT().GetExecutor(gomock.Any(), _testNamespaceEphemeral, \"owner1\").Return(&store.ShardOwner{\n\t\t\t\t\tExecutorID: \"owner1\",\n\t\t\t\t\tMetadata:   map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedOwner: \"owner1\",\n\t\t\texpectedError: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockStorage := store.NewMockStore(ctrl)\n\n\t\t\thandler := newTestHandler(t, cfg, mockStorage)\n\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(mockStorage)\n\t\t\t}\n\t\t\tresp, err := handler.GetShardOwner(context.Background(), tt.request)\n\t\t\tif tt.expectedError {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\trequire.Contains(t, err.Error(), tt.expectedErrMsg)\n\t\t\t\trequire.Nil(t, resp)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\trequire.Equal(t, tt.expectedOwner, resp.Owner)\n\t\t\t\trequire.Equal(t, tt.request.Namespace, resp.Namespace)\n\t\t\t\texpectedMetadata := map[string]string{\"ip\": \"127.0.0.1\", \"port\": \"1234\"}\n\t\t\t\trequire.Equal(t, expectedMetadata, resp.Metadata)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestWatchNamespaceState(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\tmockStorage := store.NewMockStore(ctrl)\n\tmockServer := NewMockWatchNamespaceStateServer(ctrl)\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-ns\", Type: config.NamespaceTypeFixed, ShardNum: 2},\n\t\t},\n\t}\n\n\thandler := &handlerImpl{\n\t\tlogger:               logger,\n\t\tshardDistributionCfg: cfg,\n\t\tstorage:              mockStorage,\n\t\tstartWG:              sync.WaitGroup{},\n\t}\n\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tupdatesChan := make(chan map[*store.ShardOwner][]string, 1)\n\tunsubscribe := func() { close(updatesChan) }\n\n\tmockServer.EXPECT().Context().Return(ctx).AnyTimes()\n\tmockStorage.EXPECT().SubscribeToAssignmentChanges(gomock.Any(), \"test-ns\").Return(updatesChan, unsubscribe, nil)\n\n\t// Expect update send\n\tmockServer.EXPECT().Send(gomock.Any()).DoAndReturn(func(resp *types.WatchNamespaceStateResponse) error {\n\t\trequire.Len(t, resp.Executors, 1)\n\t\trequire.Equal(t, \"executor-1\", resp.Executors[0].ExecutorID)\n\t\treturn nil\n\t})\n\n\t// Send update, then cancel\n\tgo func() {\n\t\ttime.Sleep(10 * time.Millisecond)\n\t\tupdatesChan <- map[*store.ShardOwner][]string{\n\t\t\t{ExecutorID: \"executor-1\", Metadata: map[string]string{}}: {\"shard-1\"},\n\t\t}\n\t\tcancel()\n\t}()\n\n\terr := handler.WatchNamespaceState(&types.WatchNamespaceStateRequest{Namespace: \"test-ns\"}, mockServer)\n\trequire.Error(t, err)\n\trequire.ErrorIs(t, err, context.Canceled)\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/interfaces.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage handler\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination interfaces_mock.go\n//go:generate gowrap gen -g -p . -i Handler -t ../../templates/grpc.tmpl -o ../wrappers/grpc/grpc_handler_generated.go -v handler=GRPC -v package=sharddistributorv1 -v path=github.com/uber/cadence/.gen/proto/sharddistributor/v1 -v prefix=ShardDistributor\n//go:generate gowrap gen -g -p . -i Executor -t ../../templates/grpc.tmpl -o ../wrappers/grpc/grpc_executor_generated.go -v handler=ExecutorGRPC -v package=sharddistributorv1 -v path=github.com/uber/cadence/.gen/proto/sharddistributor/v1 -v prefix=ShardDistributorExecutor\n//go:generate gowrap gen -g -p . -i Handler -t ../templates/metered.tmpl -o ../wrappers/metered/api_generated.go -v handler=Metrics\n//go:generate gowrap gen -g -p . -i Executor -t ../templates/metered.tmpl -o ../wrappers/metered/executor_generated.go -v handler=ExecutorMetrics\n\n// Handler is the interface for shard distributor handler\ntype Handler interface {\n\tcommon.Daemon\n\n\tHealth(context.Context) (*types.HealthStatus, error)\n\n\tGetShardOwner(context.Context, *types.GetShardOwnerRequest) (*types.GetShardOwnerResponse, error)\n\n\tWatchNamespaceState(*types.WatchNamespaceStateRequest, WatchNamespaceStateServer) error\n}\n\ntype Executor interface {\n\tHeartbeat(context.Context, *types.ExecutorHeartbeatRequest) (*types.ExecutorHeartbeatResponse, error)\n}\n\ntype WatchNamespaceStateServer interface {\n\tContext() context.Context\n\tSend(*types.WatchNamespaceStateResponse) error\n}\n"
  },
  {
    "path": "service/sharddistributor/handler/interfaces_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: interfaces.go\n//\n// Generated by this command:\n//\n//\tmockgen -package handler -source interfaces.go -destination interfaces_mock.go\n//\n\n// Package handler is a generated GoMock package.\npackage handler\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\ttypes \"github.com/uber/cadence/common/types\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// GetShardOwner mocks base method.\nfunc (m *MockHandler) GetShardOwner(arg0 context.Context, arg1 *types.GetShardOwnerRequest) (*types.GetShardOwnerResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardOwner\", arg0, arg1)\n\tret0, _ := ret[0].(*types.GetShardOwnerResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShardOwner indicates an expected call of GetShardOwner.\nfunc (mr *MockHandlerMockRecorder) GetShardOwner(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardOwner\", reflect.TypeOf((*MockHandler)(nil).GetShardOwner), arg0, arg1)\n}\n\n// Health mocks base method.\nfunc (m *MockHandler) Health(arg0 context.Context) (*types.HealthStatus, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Health\", arg0)\n\tret0, _ := ret[0].(*types.HealthStatus)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Health indicates an expected call of Health.\nfunc (mr *MockHandlerMockRecorder) Health(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Health\", reflect.TypeOf((*MockHandler)(nil).Health), arg0)\n}\n\n// Start mocks base method.\nfunc (m *MockHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockHandler)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockHandler) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockHandlerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockHandler)(nil).Stop))\n}\n\n// WatchNamespaceState mocks base method.\nfunc (m *MockHandler) WatchNamespaceState(arg0 *types.WatchNamespaceStateRequest, arg1 WatchNamespaceStateServer) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WatchNamespaceState\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// WatchNamespaceState indicates an expected call of WatchNamespaceState.\nfunc (mr *MockHandlerMockRecorder) WatchNamespaceState(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WatchNamespaceState\", reflect.TypeOf((*MockHandler)(nil).WatchNamespaceState), arg0, arg1)\n}\n\n// MockExecutor is a mock of Executor interface.\ntype MockExecutor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockExecutorMockRecorder\n\tisgomock struct{}\n}\n\n// MockExecutorMockRecorder is the mock recorder for MockExecutor.\ntype MockExecutorMockRecorder struct {\n\tmock *MockExecutor\n}\n\n// NewMockExecutor creates a new mock instance.\nfunc NewMockExecutor(ctrl *gomock.Controller) *MockExecutor {\n\tmock := &MockExecutor{ctrl: ctrl}\n\tmock.recorder = &MockExecutorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockExecutor) EXPECT() *MockExecutorMockRecorder {\n\treturn m.recorder\n}\n\n// Heartbeat mocks base method.\nfunc (m *MockExecutor) Heartbeat(arg0 context.Context, arg1 *types.ExecutorHeartbeatRequest) (*types.ExecutorHeartbeatResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Heartbeat\", arg0, arg1)\n\tret0, _ := ret[0].(*types.ExecutorHeartbeatResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Heartbeat indicates an expected call of Heartbeat.\nfunc (mr *MockExecutorMockRecorder) Heartbeat(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Heartbeat\", reflect.TypeOf((*MockExecutor)(nil).Heartbeat), arg0, arg1)\n}\n\n// MockWatchNamespaceStateServer is a mock of WatchNamespaceStateServer interface.\ntype MockWatchNamespaceStateServer struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWatchNamespaceStateServerMockRecorder\n\tisgomock struct{}\n}\n\n// MockWatchNamespaceStateServerMockRecorder is the mock recorder for MockWatchNamespaceStateServer.\ntype MockWatchNamespaceStateServerMockRecorder struct {\n\tmock *MockWatchNamespaceStateServer\n}\n\n// NewMockWatchNamespaceStateServer creates a new mock instance.\nfunc NewMockWatchNamespaceStateServer(ctrl *gomock.Controller) *MockWatchNamespaceStateServer {\n\tmock := &MockWatchNamespaceStateServer{ctrl: ctrl}\n\tmock.recorder = &MockWatchNamespaceStateServerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWatchNamespaceStateServer) EXPECT() *MockWatchNamespaceStateServerMockRecorder {\n\treturn m.recorder\n}\n\n// Context mocks base method.\nfunc (m *MockWatchNamespaceStateServer) Context() context.Context {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Context\")\n\tret0, _ := ret[0].(context.Context)\n\treturn ret0\n}\n\n// Context indicates an expected call of Context.\nfunc (mr *MockWatchNamespaceStateServerMockRecorder) Context() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Context\", reflect.TypeOf((*MockWatchNamespaceStateServer)(nil).Context))\n}\n\n// Send mocks base method.\nfunc (m *MockWatchNamespaceStateServer) Send(arg0 *types.WatchNamespaceStateResponse) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Send\", arg0)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Send indicates an expected call of Send.\nfunc (mr *MockWatchNamespaceStateServerMockRecorder) Send(arg0 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Send\", reflect.TypeOf((*MockWatchNamespaceStateServer)(nil).Send), arg0)\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/election/election.go",
    "content": "package election\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/multierr\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/process\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=election_mock.go Factory,Elector\n\nvar errSelfResign = fmt.Errorf(\"self-resigned\")\n\n// Module provides election factory for fx app.\nvar Module = fx.Module(\n\t\"leader-election\",\n\tfx.Provide(NewElectionFactory),\n)\n\ntype ProcessFunc func(ctx context.Context) error\n\n// Elector handles leader election for a specific namespace\ntype Elector interface {\n\tRun(ctx context.Context) <-chan bool\n}\n\n// Factory creates elector instances\ntype Factory interface {\n\tCreateElector(ctx context.Context, namespaceCfg config.Namespace) (Elector, error)\n}\n\ntype electionFactory struct {\n\thostname       string\n\tcfg            config.Election\n\tleaderStore    store.Elector\n\tstore          store.Store\n\tlogger         log.Logger\n\tserviceID      string\n\tclock          clock.TimeSource\n\tprocessFactory process.Factory\n}\n\ntype elector struct {\n\thostname       string\n\tnamespace      config.Namespace\n\tleaderStore    store.Elector\n\tstore          store.Store\n\tlogger         log.Logger\n\tcfg            config.Election\n\tleaderStarted  time.Time\n\tclock          clock.TimeSource\n\tprocessFactory process.Factory\n}\n\ntype FactoryParams struct {\n\tfx.In\n\n\tHostName       string `name:\"hostname\"`\n\tCfg            config.ShardDistribution\n\tLeaderStore    store.Elector\n\tLogger         log.Logger\n\tClock          clock.TimeSource\n\tProcessFactory process.Factory\n\tStore          store.Store\n}\n\n// NewElectionFactory creates a new election factory\nfunc NewElectionFactory(p FactoryParams) Factory {\n\treturn &electionFactory{\n\t\tcfg:            p.Cfg.Election,\n\t\tleaderStore:    p.LeaderStore,\n\t\tstore:          p.Store,\n\t\tlogger:         p.Logger,\n\t\tclock:          p.Clock,\n\t\thostname:       p.HostName,\n\t\tprocessFactory: p.ProcessFactory,\n\t}\n}\n\n// CreateElector creates a new elector for the given namespace\nfunc (f *electionFactory) CreateElector(ctx context.Context, namespaceCfg config.Namespace) (Elector, error) {\n\treturn &elector{\n\t\tnamespace:      namespaceCfg,\n\t\tleaderStore:    f.leaderStore,\n\t\tstore:          f.store,\n\t\tlogger:         f.logger.WithTags(tag.ComponentLeaderElection, tag.ShardNamespace(namespaceCfg.Name)),\n\t\tcfg:            f.cfg,\n\t\tclock:          f.clock,\n\t\thostname:       f.hostname,\n\t\tprocessFactory: f.processFactory,\n\t}, nil\n}\n\n// Run starts the leader election process it returns a channel that will return the value if the current instance becomes the leader or resigns from leadership.\nfunc (e *elector) Run(ctx context.Context) <-chan bool {\n\tleaderCh := make(chan bool, 1)\n\n\t// Create a child context that we can explicitly cancel when errors occur\n\trunCtx, cancelRun := context.WithCancel(ctx)\n\n\tgo func() {\n\t\tdefer close(leaderCh)\n\t\tdefer cancelRun() // Ensure child context is canceled on exit\n\t\tdefer func() {\n\t\t\te.logger.Info(\"Leader election process exiting\")\n\t\t\tif r := recover(); r != nil {\n\t\t\t\te.logger.Error(\"Panic in election process\", tag.Value(r))\n\t\t\t}\n\t\t}()\n\n\t\tfor {\n\t\t\tif err := e.runElection(runCtx, leaderCh); err != nil {\n\t\t\t\t// Check if parent context is already canceled\n\t\t\t\tif runCtx.Err() != nil {\n\t\t\t\t\te.logger.Info(\"Context canceled, stopping election loop\", tag.Error(runCtx.Err()))\n\t\t\t\t\treturn\n\t\t\t\t}\n\n\t\t\t\t// Self resign, immediately retry, otherwise, wait\n\t\t\t\tif !errors.Is(err, errSelfResign) {\n\t\t\t\t\te.logger.Error(\"Error in election, retrying\", tag.Error(err))\n\n\t\t\t\t\tselect {\n\t\t\t\t\tcase <-runCtx.Done():\n\t\t\t\t\t\te.logger.Info(\"Context canceled, stopping election loop, exit immediately\", tag.Error(runCtx.Err()))\n\t\t\t\t\t\treturn // Context was canceled, exit immediately\n\t\t\t\t\tcase <-e.clock.After(e.cfg.FailedElectionCooldown):\n\t\t\t\t\t\te.logger.Info(\"Cooldown period ended, retrying election\")\n\t\t\t\t\t\t// Continue after cooldown\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tif runCtx.Err() != nil {\n\t\t\t\te.logger.Info(\"Context canceled, stopping election loop\", tag.Error(runCtx.Err()))\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}()\n\n\treturn leaderCh\n}\n\n// runElection runs a single election attempt\nfunc (e *elector) runElection(ctx context.Context, leaderCh chan<- bool) (err error) {\n\te.logger.Info(\"Run election\")\n\t// Add random delay before campaigning to spread load across instances\n\tdelay := time.Duration(rand.Intn(int(e.cfg.MaxRandomDelay)))\n\n\te.logger.Debug(\"Adding random delay before campaigning\", tag.ElectionDelay(delay))\n\n\tselect {\n\tcase <-e.clock.After(delay):\n\t\te.logger.Debug(\"Random delay before campaigning completed\")\n\t\t// Continue after delay\n\tcase <-ctx.Done():\n\t\treturn fmt.Errorf(\"context cancelled during pre-campaign delay: %w\", ctx.Err())\n\t}\n\n\te.logger.Info(\"Creating election\")\n\telection, err := e.leaderStore.CreateElection(ctx, e.namespace.Name)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"create session: %w\", err)\n\t}\n\n\tvar leaderProcess process.Processor\n\n\tdefer func() {\n\t\tresignErr := e.resign(election, leaderProcess)\n\t\tif resignErr != nil {\n\t\t\tif err == nil {\n\t\t\t\terr = resignErr\n\t\t\t} else {\n\t\t\t\t// Something already went wrong, so the process is most likely not started or stopped.\n\t\t\t\t// It should be safe to report leadership lost.\n\t\t\t\te.logger.Error(\"Error resigning leader\", tag.Error(resignErr))\n\t\t\t}\n\t\t}\n\n\t\t// We are no longer a leader and OnResign is called - notify the manager that leader elected process\n\t\tleaderCh <- false\n\t}()\n\n\te.logger.Info(\"Starting campaign for leader\")\n\t// Campaign to become leader\n\tif err := election.Campaign(ctx, e.hostname); err != nil {\n\t\treturn fmt.Errorf(\"failed to campaign: %w\", err)\n\t}\n\n\tleaderProcess = e.processFactory.CreateProcessor(e.namespace, e.store, election)\n\n\te.logger.Debug(\"Run leader process\")\n\terr = leaderProcess.Run(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"onLeader: %w\", err)\n\t}\n\n\t// Successfully became leader\n\te.leaderStarted = e.clock.Now()\n\tleaderCh <- true\n\n\te.logger.Info(\"Became leader\")\n\n\t// Start a timer to voluntarily resign after the leadership period\n\tleaderTimer := e.clock.NewTimer(e.cfg.LeaderPeriod)\n\tdefer leaderTimer.Stop()\n\n\t// Watch for session expiration, context cancellation, or timer expiration\n\tselect {\n\tcase <-ctx.Done():\n\t\te.logger.Info(\"Context cancelled while leader\")\n\t\treturn nil\n\n\tcase <-election.Done():\n\t\te.logger.Info(\"Session expired while leader\")\n\t\treturn fmt.Errorf(\"session expired\")\n\n\tcase <-leaderTimer.Chan():\n\t\te.logger.Info(\"Leadership period ended, voluntarily resigning\")\n\t\treturn errSelfResign\n\t}\n}\n\nfunc (e *elector) resign(election store.Election, processor process.Processor) error {\n\tctx, cancel := e.clock.ContextWithTimeout(context.Background(), 3*time.Second)\n\tdefer cancel()\n\n\tvar resignErr error\n\n\tif processor != nil {\n\t\t// First try to terminate the processor to stop leader processing\n\t\tif err := processor.Terminate(ctx); err != nil {\n\t\t\tresignErr = fmt.Errorf(\"terminate processor: %w\", err)\n\t\t}\n\t}\n\n\t// Then try to resign leadership\n\tif err := election.Resign(ctx); err != nil {\n\t\tresignErr = multierr.Append(resignErr, fmt.Errorf(\"resign election: %w\", err))\n\t}\n\n\t// Finally, try to clean up the election\n\t// to be sure all resources are released\n\tif err := election.Cleanup(ctx); err != nil {\n\t\tresignErr = multierr.Append(resignErr, fmt.Errorf(\"cleanup election: %w\", err))\n\t}\n\n\treturn resignErr\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/election/election_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: election.go\n//\n// Generated by this command:\n//\n//\tmockgen -package election -source election.go -destination=election_mock.go Factory,Elector\n//\n\n// Package election is a generated GoMock package.\npackage election\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\n// MockElector is a mock of Elector interface.\ntype MockElector struct {\n\tctrl     *gomock.Controller\n\trecorder *MockElectorMockRecorder\n\tisgomock struct{}\n}\n\n// MockElectorMockRecorder is the mock recorder for MockElector.\ntype MockElectorMockRecorder struct {\n\tmock *MockElector\n}\n\n// NewMockElector creates a new mock instance.\nfunc NewMockElector(ctrl *gomock.Controller) *MockElector {\n\tmock := &MockElector{ctrl: ctrl}\n\tmock.recorder = &MockElectorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockElector) EXPECT() *MockElectorMockRecorder {\n\treturn m.recorder\n}\n\n// Run mocks base method.\nfunc (m *MockElector) Run(ctx context.Context) <-chan bool {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Run\", ctx)\n\tret0, _ := ret[0].(<-chan bool)\n\treturn ret0\n}\n\n// Run indicates an expected call of Run.\nfunc (mr *MockElectorMockRecorder) Run(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Run\", reflect.TypeOf((*MockElector)(nil).Run), ctx)\n}\n\n// MockFactory is a mock of Factory interface.\ntype MockFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockFactoryMockRecorder is the mock recorder for MockFactory.\ntype MockFactoryMockRecorder struct {\n\tmock *MockFactory\n}\n\n// NewMockFactory creates a new mock instance.\nfunc NewMockFactory(ctrl *gomock.Controller) *MockFactory {\n\tmock := &MockFactory{ctrl: ctrl}\n\tmock.recorder = &MockFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFactory) EXPECT() *MockFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// CreateElector mocks base method.\nfunc (m *MockFactory) CreateElector(ctx context.Context, namespaceCfg config.Namespace) (Elector, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateElector\", ctx, namespaceCfg)\n\tret0, _ := ret[0].(Elector)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateElector indicates an expected call of CreateElector.\nfunc (mr *MockFactoryMockRecorder) CreateElector(ctx, namespaceCfg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateElector\", reflect.TypeOf((*MockFactory)(nil).CreateElector), ctx, namespaceCfg)\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/election/election_test.go",
    "content": "package election\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/process\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t_testHost = \"localhost\"\n)\n\nvar (\n\t_testNamespace = config.Namespace{Name: \"test-namespace\"}\n)\n\nvar (\n\t_testLeaderPeriod           = time.Minute\n\t_testMaxRandomDelay         = time.Second\n\t_testFailedElectionCooldown = 10 * time.Second\n)\n\nfunc TestElector_Run(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\n\telection := store.NewMockElection(ctrl)\n\telection.EXPECT().Campaign(gomock.Any(), _testHost).Return(nil)\n\telection.EXPECT().Done().Return(make(chan struct{}))\n\n\tfinished := make(chan struct{})\n\t// once test is done cleanup will be called\n\telection.EXPECT().Resign(gomock.Any()).DoAndReturn(func(_ context.Context) error {\n\t\tclose(finished)\n\t\treturn nil\n\t})\n\telection.EXPECT().Cleanup(gomock.Any()).Return(nil)\n\n\tleaderStore := store.NewMockElector(ctrl)\n\tleaderStore.EXPECT().CreateElection(gomock.Any(), _testNamespace.Name).Return(election, nil)\n\n\tshardStore := store.NewMockStore(ctrl)\n\n\tprocessFactory := process.NewMockFactory(ctrl)\n\tprocessRunner := process.NewMockProcessor(ctrl)\n\tprocessFactory.EXPECT().CreateProcessor(_testNamespace, shardStore, election).Return(processRunner)\n\n\tfactory := NewElectionFactory(FactoryParams{\n\t\tHostName: _testHost,\n\t\tCfg: config.ShardDistribution{\n\t\t\tElection: config.Election{\n\t\t\t\tLeaderPeriod:           _testLeaderPeriod,\n\t\t\t\tMaxRandomDelay:         _testMaxRandomDelay,\n\t\t\t\tFailedElectionCooldown: _testFailedElectionCooldown,\n\t\t\t},\n\t\t},\n\t\tLeaderStore:    leaderStore,\n\t\tStore:          shardStore,\n\t\tLogger:         logger,\n\t\tClock:          timeSource,\n\t\tProcessFactory: processFactory,\n\t})\n\n\tel, err := factory.CreateElector(context.Background(), _testNamespace)\n\trequire.NoError(t, err)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\t// Track callback executions\n\tonLeaderCalled := false\n\tonResignCalled := false\n\n\tprocessRunner.EXPECT().Run(gomock.Any()).DoAndReturn(func(_ context.Context) error {\n\t\tonLeaderCalled = true\n\t\treturn nil\n\t})\n\n\tprocessRunner.EXPECT().Terminate(gomock.Any()).DoAndReturn(func(ctx context.Context) error {\n\t\tonResignCalled = true\n\t\treturn nil\n\t})\n\n\tgo func() {\n\t\t// Wait until run will stop on timer\n\t\ttimeSource.BlockUntil(1)\n\t\t// Advance the time to kick in the election.\n\t\ttimeSource.Advance(_testMaxRandomDelay)\n\t}()\n\n\tleaderChan := el.Run(ctx)\n\tassert.True(t, <-leaderChan)\n\tassert.True(t, onLeaderCalled, \"OnLeader callback should have been called\")\n\tassert.False(t, onResignCalled, \"OnResign callback should not have been called\")\n\tcancel()\n\t<-finished\n}\n\nfunc TestElector_Run_Resign(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\tt.Run(\"context_canceled\", func(t *testing.T) {\n\t\tleaderChan, p := prepareRun(t, nil, nil)\n\t\tp.election.EXPECT().Resign(gomock.Any()).Return(nil)\n\t\tp.cancel()\n\t\tassert.False(t, <-leaderChan)\n\t\t// Wait for the goroutine to exit\n\t\tfor range leaderChan {\n\t\t}\n\t})\n\tt.Run(\"session_expired\", func(t *testing.T) {\n\t\tleaderChan, p := prepareRun(t, nil, nil)\n\t\tp.election.EXPECT().Resign(gomock.Any()).Return(nil)\n\t\tclose(p.electionCh)\n\t\tassert.False(t, <-leaderChan)\n\t\tp.cancel()\n\t\t// Wait for the goroutine to exit\n\t\tfor range leaderChan {\n\t\t}\n\t})\n\tt.Run(\"leader_resign\", func(t *testing.T) {\n\t\t// Verify onResign is called before resignation\n\t\tonResignCalled := false\n\t\tleaderChan, p := prepareRun(t, nil, func(ctx context.Context) error {\n\t\t\tonResignCalled = true\n\t\t\treturn nil\n\t\t})\n\n\t\t// We should be blocked on the timer.\n\t\tp.timeSource.BlockUntil(1)\n\n\t\tp.election.EXPECT().Resign(gomock.Any()).DoAndReturn(func(ctx context.Context) error {\n\t\t\tassert.True(t, onResignCalled, \"OnResign callback should be called before Resign\")\n\t\t\treturn nil\n\t\t})\n\n\t\tp.timeSource.Advance(_testLeaderPeriod + 1)\n\t\tp.timeSource.BlockUntil(1)\n\t\tassert.False(t, <-leaderChan)\n\t\tp.cancel()\n\t\t// Wait for the goroutine to exit\n\t\tfor range leaderChan {\n\t\t}\n\t})\n\n\tt.Run(\"onResign_error\", func(t *testing.T) {\n\t\t// Set onResign to return an error\n\t\tonResignCalled := false\n\t\tresignErr := errors.New(\"resign error\")\n\t\tonResign := func(ctx context.Context) error {\n\t\t\tonResignCalled = true\n\t\t\treturn resignErr\n\t\t}\n\n\t\tleaderChan, p := prepareRun(t, nil, onResign)\n\t\tp.election.EXPECT().Resign(gomock.Any()).Return(nil)\n\t\t// We should be blocked on the timer.\n\t\tp.timeSource.BlockUntil(1)\n\n\t\t// The resign function on election should not be called if onResign returns an error\n\t\tp.timeSource.Advance(_testLeaderPeriod + 1)\n\t\tp.timeSource.BlockUntil(1)\n\t\tassert.False(t, <-leaderChan)\n\t\tassert.True(t, onResignCalled, \"OnResign callback should have been called\")\n\n\t\tp.cancel()\n\t\t// Wait for the goroutine to exit\n\t\tfor range leaderChan {\n\t\t}\n\t})\n\tt.Run(\"OnResign_and_resign_error\", func(t *testing.T) {\n\t\t// Set onResign to return an error\n\t\tonResignCalled := false\n\t\tresignErr := errors.New(\"resign error\")\n\t\tonResign := func(ctx context.Context) error {\n\t\t\tonResignCalled = true\n\t\t\treturn resignErr\n\t\t}\n\n\t\tleaderChan, p := prepareRun(t, nil, onResign)\n\t\tp.election.EXPECT().Resign(gomock.Any()).Return(fmt.Errorf(\"failed to resign\"))\n\t\t// We should be blocked on the timer.\n\t\tp.timeSource.BlockUntil(1)\n\n\t\t// The resign function on election should not be called if onResign returns an error\n\t\tp.timeSource.Advance(_testLeaderPeriod + 1)\n\t\tp.timeSource.BlockUntil(1)\n\t\tassert.False(t, <-leaderChan)\n\t\tassert.True(t, onResignCalled, \"OnResign callback should have been called\")\n\n\t\tp.cancel()\n\t\t// Wait for the goroutine to exit\n\t\tfor range leaderChan {\n\t\t}\n\t})\n}\n\ntype runParams struct {\n\tctx        context.Context\n\tcancel     context.CancelFunc\n\ttimeSource clock.MockedTimeSource\n\telectionCh chan struct{}\n\telection   *store.MockElection\n\tonLeader   ProcessFunc\n\tonResign   ProcessFunc\n}\n\nfunc prepareRun(t *testing.T, onLeader, onResign ProcessFunc) (<-chan bool, runParams) {\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\n\telectionCh := make(chan struct{})\n\n\telection := store.NewMockElection(ctrl)\n\telection.EXPECT().Campaign(gomock.Any(), _testHost).Return(nil)\n\telection.EXPECT().Cleanup(gomock.Any()).Return(nil)\n\telection.EXPECT().Done().Return(electionCh)\n\n\tleaderStore := store.NewMockElector(ctrl)\n\tleaderStore.EXPECT().CreateElection(gomock.Any(), _testNamespace.Name).Return(election, nil)\n\n\tshardStore := store.NewMockStore(ctrl)\n\n\tprocessFactory := process.NewMockFactory(ctrl)\n\tprocessRunner := process.NewMockProcessor(ctrl)\n\tprocessFactory.EXPECT().CreateProcessor(_testNamespace, shardStore, election).Return(processRunner)\n\n\tfactory := NewElectionFactory(FactoryParams{\n\t\tHostName: _testHost,\n\t\tCfg: config.ShardDistribution{\n\t\t\tElection: config.Election{\n\t\t\t\tLeaderPeriod:           _testLeaderPeriod,\n\t\t\t\tMaxRandomDelay:         _testMaxRandomDelay,\n\t\t\t\tFailedElectionCooldown: _testFailedElectionCooldown,\n\t\t\t},\n\t\t},\n\t\tLeaderStore:    leaderStore,\n\t\tStore:          shardStore,\n\t\tLogger:         logger,\n\t\tClock:          timeSource,\n\t\tProcessFactory: processFactory,\n\t})\n\n\telector, err := factory.CreateElector(context.Background(), _testNamespace)\n\trequire.NoError(t, err)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\n\t// Default callbacks\n\tif onLeader == nil {\n\t\tonLeader = func(ctx context.Context) error {\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tif onResign == nil {\n\t\tonResign = func(ctx context.Context) error {\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tprocessRunner.EXPECT().Run(gomock.Any()).DoAndReturn(onLeader)\n\tprocessRunner.EXPECT().Terminate(gomock.Any()).DoAndReturn(onResign)\n\n\tgo func() {\n\t\t// Wait until run will stop on timer\n\t\ttimeSource.BlockUntil(1)\n\t\t// Advance the time to kick in the election.\n\t\ttimeSource.Advance(_testMaxRandomDelay)\n\t}()\n\n\tleaderChan := elector.Run(ctx)\n\tassert.True(t, <-leaderChan)\n\n\treturn leaderChan, runParams{\n\t\tctx:        ctx,\n\t\tcancel:     cancel,\n\t\ttimeSource: timeSource,\n\t\telectionCh: electionCh,\n\t\telection:   election,\n\t\tonLeader:   onLeader,\n\t\tonResign:   onResign,\n\t}\n}\n\nfunc TestOnLeader_Error(t *testing.T) {\n\tgoleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\ttimeSource := clock.NewMockedTimeSource()\n\n\telection := store.NewMockElection(ctrl)\n\telection.EXPECT().Campaign(gomock.Any(), _testHost).Return(nil)\n\t// Expect resignation after onLeader failure\n\telection.EXPECT().Resign(gomock.Any()).Return(nil)\n\telection.EXPECT().Cleanup(gomock.Any()).Return(nil)\n\n\tleaderStore := store.NewMockElector(ctrl)\n\tleaderStore.EXPECT().CreateElection(gomock.Any(), _testNamespace.Name).Return(election, nil)\n\n\tshardStore := store.NewMockStore(ctrl)\n\n\tprocessFactory := process.NewMockFactory(ctrl)\n\tprocessRunner := process.NewMockProcessor(ctrl)\n\tprocessFactory.EXPECT().CreateProcessor(_testNamespace, shardStore, election).Return(processRunner)\n\n\t// Create elector directly for test control\n\tel := &elector{\n\t\tnamespace:   _testNamespace,\n\t\tleaderStore: leaderStore,\n\t\tstore:       shardStore,\n\t\tlogger:      logger,\n\t\tcfg: config.Election{\n\t\t\tLeaderPeriod:           _testLeaderPeriod,\n\t\t\tMaxRandomDelay:         _testMaxRandomDelay,\n\t\t\tFailedElectionCooldown: _testFailedElectionCooldown,\n\t\t},\n\t\tclock:          timeSource,\n\t\thostname:       _testHost,\n\t\tprocessFactory: processFactory,\n\t}\n\n\t// Create a cancelable context for the test\n\tctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)\n\tdefer cancel() // Ensure context is canceled at the end of the test\n\n\t// Make onLeader return an error\n\tonLeaderErr := errors.New(\"leader error\")\n\tprocessRunner.EXPECT().Run(gomock.Any()).Return(onLeaderErr)\n\tprocessRunner.EXPECT().Terminate(gomock.Any()).Return(nil)\n\n\tgo func() {\n\t\t// Wait until run will stop on timer\n\t\ttimeSource.BlockUntil(1)\n\t\t// Advance the time to kick in the election.\n\t\ttimeSource.Advance(_testMaxRandomDelay)\n\t}()\n\n\t// Run the test\n\tleaderCh := make(chan bool, 1)\n\terr := el.runElection(ctx, leaderCh)\n\n\t// Error should contain our onLeader error\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"onLeader\")\n\tassert.Contains(t, err.Error(), \"leader error\")\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/namespace/manager.go",
    "content": "package namespace\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/clientcommon\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/election\"\n)\n\n// Module provides namespace manager component for an fx app.\nvar Module = fx.Module(\n\t\"namespace-manager\",\n\tfx.Invoke(NewManager),\n)\n\n// stateFn is a recursive function type representing a state in the election\n// state machine.\n// Each state function blocks until a transition occurs and returns the next state function,\n// or nil to stop the machine.\ntype stateFn func(ctx context.Context) stateFn\n\ntype Manager struct {\n\tcfg             config.ShardDistribution\n\tlogger          log.Logger\n\telectionFactory election.Factory\n\tdrainObserver   clientcommon.DrainSignalObserver\n\tnamespaces      map[string]*namespaceHandler\n\tctx             context.Context\n\tcancel          context.CancelFunc\n}\n\ntype namespaceHandler struct {\n\tlogger          log.Logger\n\telectionFactory election.Factory\n\tnamespaceCfg    config.Namespace\n\tdrainObserver   clientcommon.DrainSignalObserver\n\tcleanupWg       sync.WaitGroup\n}\n\ntype ManagerParams struct {\n\tfx.In\n\n\tCfg             config.ShardDistribution\n\tLogger          log.Logger\n\tElectionFactory election.Factory\n\tLifecycle       fx.Lifecycle\n\tDrainObserver   clientcommon.DrainSignalObserver `optional:\"true\"`\n}\n\n// NewManager creates a new namespace manager\nfunc NewManager(p ManagerParams) *Manager {\n\tmanager := &Manager{\n\t\tcfg:             p.Cfg,\n\t\tlogger:          p.Logger.WithTags(tag.ComponentNamespaceManager),\n\t\telectionFactory: p.ElectionFactory,\n\t\tdrainObserver:   p.DrainObserver,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\tp.Lifecycle.Append(fx.StartStopHook(manager.Start, manager.Stop))\n\n\treturn manager\n}\n\n// Start initializes the namespace manager and starts handling all namespaces\nfunc (m *Manager) Start(ctx context.Context) error {\n\tm.ctx, m.cancel = context.WithCancel(context.Background())\n\n\tfor _, ns := range m.cfg.Namespaces {\n\t\tm.logger.Info(\"Starting namespace handler\", tag.ShardNamespace(ns.Name))\n\t\tif err := m.handleNamespace(ns); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// Stop gracefully stops all namespace handlers.\n// Cancels the manager context which cascades to all handler contexts,\n// then waits for all election goroutines to finish.\nfunc (m *Manager) Stop(ctx context.Context) error {\n\tif m.cancel == nil {\n\t\treturn fmt.Errorf(\"manager was not running\")\n\t}\n\n\tm.cancel()\n\n\tfor ns, handler := range m.namespaces {\n\t\tm.logger.Info(\"Waiting for namespace handler to stop\", tag.ShardNamespace(ns))\n\t\thandler.cleanupWg.Wait()\n\t}\n\n\treturn nil\n}\n\n// handleNamespace sets up a namespace handler and starts its election goroutine.\nfunc (m *Manager) handleNamespace(namespaceCfg config.Namespace) error {\n\tif _, exists := m.namespaces[namespaceCfg.Name]; exists {\n\t\treturn fmt.Errorf(\"namespace %s already running\", namespaceCfg.Name)\n\t}\n\n\thandler := &namespaceHandler{\n\t\tlogger:          m.logger.WithTags(tag.ShardNamespace(namespaceCfg.Name)),\n\t\telectionFactory: m.electionFactory,\n\t\tnamespaceCfg:    namespaceCfg,\n\t\tdrainObserver:   m.drainObserver,\n\t}\n\n\tm.namespaces[namespaceCfg.Name] = handler\n\thandler.cleanupWg.Add(1)\n\n\tgo handler.runElection(m.ctx)\n\n\treturn nil\n}\n\n// runElection drives the election state machine for a namespace.\n// It starts in the campaigning state and follows state transitions\n// until a state returns nil (stop).\nfunc (h *namespaceHandler) runElection(ctx context.Context) {\n\tdefer h.cleanupWg.Done()\n\n\tfor state := h.campaigning; state != nil; {\n\t\tstate = state(ctx)\n\t}\n}\n\nfunc (h *namespaceHandler) drainChannel() <-chan struct{} {\n\tif h.drainObserver != nil {\n\t\treturn h.drainObserver.Drain()\n\t}\n\treturn nil\n}\n\nfunc (h *namespaceHandler) startElection(ctx context.Context) (<-chan bool, context.CancelFunc, error) {\n\telectorCtx, cancel := context.WithCancel(ctx)\n\telector, err := h.electionFactory.CreateElector(electorCtx, h.namespaceCfg)\n\tif err != nil {\n\t\tcancel()\n\t\treturn nil, nil, err\n\t}\n\treturn elector.Run(electorCtx), cancel, nil\n}\n\n// campaigning creates an elector and participates in leader election.\n// Transitions: h.idle on drain, h.campaigning on recoverable error, nil on stop.\nfunc (h *namespaceHandler) campaigning(ctx context.Context) stateFn {\n\th.logger.Info(\"Entering campaigning state\")\n\n\tdrainCh := h.drainChannel()\n\n\tselect {\n\tcase <-drainCh:\n\t\th.logger.Info(\"Drain signal detected before election start\")\n\t\treturn h.idle\n\tdefault:\n\t}\n\n\tleaderCh, cancel, err := h.startElection(ctx)\n\tif err != nil {\n\t\th.logger.Error(\"Failed to create elector\", tag.Error(err))\n\t\treturn nil\n\t}\n\tdefer cancel()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn nil\n\t\tcase <-drainCh:\n\t\t\th.logger.Info(\"Drain signal received, resigning from election\")\n\t\t\treturn h.idle\n\t\tcase isLeader, ok := <-leaderCh:\n\t\t\tif !ok {\n\t\t\t\th.logger.Error(\"Election channel closed unexpectedly\")\n\t\t\t\treturn h.campaigning\n\t\t\t}\n\t\t\tif isLeader {\n\t\t\t\th.logger.Info(\"Became leader for namespace\")\n\t\t\t} else {\n\t\t\t\th.logger.Info(\"Lost leadership for namespace\")\n\t\t\t}\n\t\t}\n\t}\n}\n\n// idle waits for an undrain signal to resume campaigning.\n// Transitions: h.campaigning on undrain, nil on stop.\nfunc (h *namespaceHandler) idle(ctx context.Context) stateFn {\n\th.logger.Info(\"Entering idle state (drained)\")\n\n\tvar undrainCh <-chan struct{}\n\tif h.drainObserver != nil {\n\t\tundrainCh = h.drainObserver.Undrain()\n\t}\n\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn nil\n\tcase <-undrainCh:\n\t\th.logger.Info(\"Undrain signal received, resuming election\")\n\t\treturn h.campaigning\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/namespace/manager_test.go",
    "content": "package namespace\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/election\"\n)\n\n// mockElectorRun returns a DoAndReturn function that simulates an elector:\n// it returns a leaderCh and closes it when the context is cancelled.\nfunc mockElectorRun(leaderCh chan bool) func(ctx context.Context) <-chan bool {\n\treturn func(ctx context.Context) <-chan bool {\n\t\tgo func() {\n\t\t\t<-ctx.Done()\n\t\t\tclose(leaderCh)\n\t\t}()\n\t\treturn (<-chan bool)(leaderCh)\n\t}\n}\n\n// closeDrainObserver is a test helper that simulates the close-and-recreate\n// semantics of the real DrainSignalObserver. It wraps a mock and manages\n// the channel lifecycle.\ntype closeDrainObserver struct {\n\tmu        sync.Mutex\n\tdrainCh   chan struct{}\n\tundrainCh chan struct{}\n}\n\nfunc newCloseDrainObserver() *closeDrainObserver {\n\treturn &closeDrainObserver{\n\t\tdrainCh:   make(chan struct{}),\n\t\tundrainCh: make(chan struct{}),\n\t}\n}\n\nfunc (o *closeDrainObserver) Drain() <-chan struct{} {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\treturn o.drainCh\n}\n\nfunc (o *closeDrainObserver) Undrain() <-chan struct{} {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\treturn o.undrainCh\n}\n\nfunc (o *closeDrainObserver) SignalDrain() {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\tclose(o.drainCh)\n\to.undrainCh = make(chan struct{})\n}\n\nfunc (o *closeDrainObserver) SignalUndrain() {\n\to.mu.Lock()\n\tdefer o.mu.Unlock()\n\tclose(o.undrainCh)\n\to.drainCh = make(chan struct{})\n}\n\nfunc TestNewManager(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := NewManager(ManagerParams{\n\t\tCfg:             cfg,\n\t\tLogger:          logger,\n\t\tElectionFactory: electionFactory,\n\t\tLifecycle:       fxtest.NewLifecycle(t),\n\t})\n\n\tassert.NotNil(t, manager)\n\tassert.Equal(t, cfg, manager.cfg)\n\tassert.Equal(t, 0, len(manager.namespaces))\n}\n\nfunc TestStartManager(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\ttime.Sleep(10 * time.Millisecond)\n\n\tassert.NoError(t, err)\n\tassert.NotNil(t, manager.ctx)\n\tassert.NotNil(t, manager.cancel)\n\tassert.Equal(t, 1, len(manager.namespaces))\n\tassert.Contains(t, manager.namespaces, \"test-namespace\")\n\n\t// Cleanup\n\tmanager.cancel()\n\tmanager.namespaces[\"test-namespace\"].cleanupWg.Wait()\n}\n\nfunc TestStartManagerWithElectorError(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\texpectedErr := errors.New(\"elector creation failed\")\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), config.Namespace{Name: \"test-namespace\"}).Return(nil, expectedErr)\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\tassert.NoError(t, err)\n\n\t// The goroutine exits on elector creation error\n\thandler := manager.namespaces[\"test-namespace\"]\n\thandler.cleanupWg.Wait()\n\n\t// Cleanup\n\tmanager.cancel()\n}\n\nfunc TestStopManager(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\t_ = manager.Start(context.Background())\n\ttime.Sleep(10 * time.Millisecond)\n\n\terr := manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestHandleNamespaceAlreadyExists(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\n\tmanager := &Manager{\n\t\tcfg:             config.ShardDistribution{},\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\tmanager.ctx, manager.cancel = context.WithCancel(context.Background())\n\tdefer manager.cancel()\n\n\tmanager.namespaces[\"test-namespace\"] = &namespaceHandler{}\n\n\terr := manager.handleNamespace(config.Namespace{Name: \"test-namespace\"})\n\tassert.ErrorContains(t, err, \"namespace test-namespace already running\")\n}\n\nfunc TestRunElection_LeadershipEvents(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\trequire.NoError(t, err)\n\n\tleaderCh <- true\n\ttime.Sleep(10 * time.Millisecond)\n\n\tleaderCh <- false\n\ttime.Sleep(10 * time.Millisecond)\n\n\terr = manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestDrainSignal_TriggersResign(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tobserver := newCloseDrainObserver()\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tdrainObserver:   observer,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\trequire.NoError(t, err)\n\n\t// Wait for the elector to be running\n\tleaderCh <- true\n\ttime.Sleep(10 * time.Millisecond)\n\n\t// Close drain channel — all handlers see it\n\tobserver.SignalDrain()\n\ttime.Sleep(50 * time.Millisecond)\n\n\t// Handler should be in an idle state\n\terr = manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestDrainSignal_NilDrainObserver(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\trequire.NoError(t, err)\n\n\tassert.Nil(t, manager.drainObserver)\n\n\terr = manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestDrainSignal_ManagerStopsBeforeDrain(t *testing.T) {\n\tlogger := testlogger.New(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\telector := election.NewMockElector(ctrl)\n\n\tleaderCh := make(chan bool)\n\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector, nil)\n\telector.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh))\n\n\tobserver := newCloseDrainObserver()\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tdrainObserver:   observer,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\trequire.NoError(t, err)\n\n\t// Stop before drain fires\n\terr = manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n\nfunc TestDrainThenUndrain_ResumesElection(t *testing.T) {\n\tlogger, logs := testlogger.NewObserved(t)\n\tctrl := gomock.NewController(t)\n\telectionFactory := election.NewMockFactory(ctrl)\n\n\telector1 := election.NewMockElector(ctrl)\n\tleaderCh1 := make(chan bool)\n\telector2 := election.NewMockElector(ctrl)\n\tleaderCh2 := make(chan bool)\n\n\tgomock.InOrder(\n\t\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector1, nil),\n\t\telectionFactory.EXPECT().CreateElector(gomock.Any(), gomock.Any()).Return(elector2, nil),\n\t)\n\telector1.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh1))\n\telector2.EXPECT().Run(gomock.Any()).DoAndReturn(mockElectorRun(leaderCh2))\n\n\tobserver := newCloseDrainObserver()\n\n\tcfg := config.ShardDistribution{\n\t\tNamespaces: []config.Namespace{\n\t\t\t{Name: \"test-namespace\"},\n\t\t},\n\t}\n\n\tmanager := &Manager{\n\t\tcfg:             cfg,\n\t\tlogger:          logger,\n\t\telectionFactory: electionFactory,\n\t\tdrainObserver:   observer,\n\t\tnamespaces:      make(map[string]*namespaceHandler),\n\t}\n\n\terr := manager.Start(context.Background())\n\trequire.NoError(t, err)\n\n\t// Phase 1: elector1 running, verify it becomes leader\n\tleaderCh1 <- true\n\ttime.Sleep(10 * time.Millisecond)\n\tassert.Equal(t, 1, logs.FilterMessage(\"Became leader for namespace\").Len(), \"expected leader elected in phase 1\")\n\n\t// Drain - elector1 resigns\n\tobserver.SignalDrain()\n\ttime.Sleep(50 * time.Millisecond)\n\tassert.Equal(t, 1, logs.FilterMessage(\"Drain signal received, resigning from election\").Len())\n\n\t// Undrain - elector2 created, campaign again\n\tobserver.SignalUndrain()\n\ttime.Sleep(50 * time.Millisecond)\n\tassert.Equal(t, 1, logs.FilterMessage(\"Undrain signal received, resuming election\").Len())\n\n\t// Phase 2: verify elector2 is running and becomes leader after undrain\n\tleaderCh2 <- true\n\ttime.Sleep(10 * time.Millisecond)\n\tassert.Equal(t, 2, logs.FilterMessage(\"Became leader for namespace\").Len(), \"expected leader elected in both phases\")\n\n\terr = manager.Stop(context.Background())\n\tassert.NoError(t, err)\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/process/process_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: processor.go\n//\n// Generated by this command:\n//\n//\tmockgen -package process -source processor.go -destination=process_mock.go Factory,Processor\n//\n\n// Package process is a generated GoMock package.\npackage process\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tconfig \"github.com/uber/cadence/service/sharddistributor/config\"\n\tstore \"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n// MockProcessor is a mock of Processor interface.\ntype MockProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockProcessorMockRecorder is the mock recorder for MockProcessor.\ntype MockProcessorMockRecorder struct {\n\tmock *MockProcessor\n}\n\n// NewMockProcessor creates a new mock instance.\nfunc NewMockProcessor(ctrl *gomock.Controller) *MockProcessor {\n\tmock := &MockProcessor{ctrl: ctrl}\n\tmock.recorder = &MockProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockProcessor) EXPECT() *MockProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// Run mocks base method.\nfunc (m *MockProcessor) Run(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Run\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Run indicates an expected call of Run.\nfunc (mr *MockProcessorMockRecorder) Run(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Run\", reflect.TypeOf((*MockProcessor)(nil).Run), ctx)\n}\n\n// Terminate mocks base method.\nfunc (m *MockProcessor) Terminate(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Terminate\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Terminate indicates an expected call of Terminate.\nfunc (mr *MockProcessorMockRecorder) Terminate(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Terminate\", reflect.TypeOf((*MockProcessor)(nil).Terminate), ctx)\n}\n\n// MockFactory is a mock of Factory interface.\ntype MockFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockFactoryMockRecorder is the mock recorder for MockFactory.\ntype MockFactoryMockRecorder struct {\n\tmock *MockFactory\n}\n\n// NewMockFactory creates a new mock instance.\nfunc NewMockFactory(ctrl *gomock.Controller) *MockFactory {\n\tmock := &MockFactory{ctrl: ctrl}\n\tmock.recorder = &MockFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockFactory) EXPECT() *MockFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// CreateProcessor mocks base method.\nfunc (m *MockFactory) CreateProcessor(cfg config.Namespace, storage store.Store, election store.Election) Processor {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateProcessor\", cfg, storage, election)\n\tret0, _ := ret[0].(Processor)\n\treturn ret0\n}\n\n// CreateProcessor indicates an expected call of CreateProcessor.\nfunc (mr *MockFactoryMockRecorder) CreateProcessor(cfg, storage, election any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateProcessor\", reflect.TypeOf((*MockFactory)(nil).CreateProcessor), cfg, storage, election)\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/process/processor.go",
    "content": "package process\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"maps\"\n\t\"math\"\n\t\"math/rand\"\n\t\"slices\"\n\t\"sort\"\n\t\"strconv\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=process_mock.go Factory,Processor\n\n// Module provides processor factory for fx app.\nvar Module = fx.Module(\n\t\"leader-process\",\n\tfx.Provide(NewProcessorFactory),\n)\n\n// Processor represents a process that runs when the instance is the leader\ntype Processor interface {\n\tRun(ctx context.Context) error\n\tTerminate(ctx context.Context) error\n}\n\n// Factory creates processor instances\ntype Factory interface {\n\t// CreateProcessor creates a new processor, it takes the generic store\n\t// and the election object which provides the transactional guard.\n\tCreateProcessor(cfg config.Namespace, storage store.Store, election store.Election) Processor\n}\n\nconst (\n\t_defaultPeriod       = time.Second\n\t_defaultHeartbeatTTL = 10 * time.Second\n\t_defaultTimeout      = 1 * time.Second\n\t_defaultCooldown     = 250 * time.Millisecond\n)\n\ntype processorFactory struct {\n\tlogger        log.Logger\n\ttimeSource    clock.TimeSource\n\tcfg           config.LeaderProcess\n\tmetricsClient metrics.Client\n\tsdConfig      *config.Config\n}\n\ntype namespaceProcessor struct {\n\tnamespaceCfg  config.Namespace\n\tlogger        log.Logger\n\tmetricsClient metrics.Client\n\ttimeSource    clock.TimeSource\n\trunning       bool\n\tcancel        context.CancelFunc\n\tsdConfig      *config.Config\n\tcfg           config.LeaderProcess\n\twg            sync.WaitGroup\n\tshardStore    store.Store\n\telection      store.Election\n}\n\n// NewProcessorFactory creates a new processor factory\nfunc NewProcessorFactory(\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\ttimeSource clock.TimeSource,\n\tcfg config.ShardDistribution,\n\tsdConfig *config.Config,\n) Factory {\n\tif cfg.Process.Period == 0 {\n\t\tcfg.Process.Period = _defaultPeriod\n\t}\n\tif cfg.Process.HeartbeatTTL == 0 {\n\t\tcfg.Process.HeartbeatTTL = _defaultHeartbeatTTL\n\t}\n\tif cfg.Process.Timeout == 0 {\n\t\tcfg.Process.Timeout = _defaultTimeout\n\t}\n\tif cfg.Process.RebalanceCooldown == 0 {\n\t\tcfg.Process.RebalanceCooldown = _defaultCooldown\n\t}\n\n\treturn &processorFactory{\n\t\tlogger:        logger,\n\t\ttimeSource:    timeSource,\n\t\tcfg:           cfg.Process,\n\t\tmetricsClient: metricsClient,\n\t\tsdConfig:      sdConfig,\n\t}\n}\n\n// CreateProcessor creates a new processor for the given namespace\nfunc (f *processorFactory) CreateProcessor(cfg config.Namespace, shardStore store.Store, election store.Election) Processor {\n\treturn &namespaceProcessor{\n\t\tnamespaceCfg:  cfg,\n\t\tlogger:        f.logger.WithTags(tag.ComponentLeaderProcessor, tag.ShardNamespace(cfg.Name)),\n\t\ttimeSource:    f.timeSource,\n\t\tcfg:           f.cfg,\n\t\tshardStore:    shardStore,\n\t\telection:      election, // Store the election object\n\t\tmetricsClient: f.metricsClient,\n\t\tsdConfig:      f.sdConfig,\n\t}\n}\n\n// Run begins processing for this namespace\nfunc (p *namespaceProcessor) Run(ctx context.Context) error {\n\tif p.running {\n\t\treturn fmt.Errorf(\"processor is already running\")\n\t}\n\n\tpCtx, cancel := context.WithCancel(ctx)\n\tp.cancel = cancel\n\tp.running = true\n\n\tp.logger.Info(\"Starting\")\n\n\tp.wg.Add(1)\n\t// Start the process in a goroutine\n\tgo p.runProcess(pCtx)\n\n\treturn nil\n}\n\n// Terminate halts processing for this namespace\nfunc (p *namespaceProcessor) Terminate(ctx context.Context) error {\n\tif !p.running {\n\t\treturn fmt.Errorf(\"processor has not been started\")\n\t}\n\n\tp.logger.Info(\"Stopping\")\n\n\tif p.cancel != nil {\n\t\tp.cancel()\n\t\tp.cancel = nil\n\t}\n\n\tp.running = false\n\n\t// Ensure that the process has stopped.\n\tp.wg.Wait()\n\n\treturn nil\n}\n\n// runProcess launches and manages the processing loops.\nfunc (p *namespaceProcessor) runProcess(ctx context.Context) {\n\tdefer p.wg.Done()\n\n\tvar loopWg sync.WaitGroup\n\tloopWg.Add(2) // We have two loops to manage.\n\n\t// Launch the assignment and executor cleanup process in its own goroutine.\n\tgo func() {\n\t\tdefer loopWg.Done()\n\t\tp.runRebalancingLoop(ctx)\n\t}()\n\n\t// Launch the shard stats cleanup process in its own goroutine.\n\tgo func() {\n\t\tdefer loopWg.Done()\n\t\tp.runShardStatsCleanupLoop(ctx)\n\t}()\n\n\t// Wait for both loops to exit.\n\tloopWg.Wait()\n}\n\n// runRebalancingLoop handles shard assignment and redistribution.\nfunc (p *namespaceProcessor) runRebalancingLoop(ctx context.Context) {\n\t// Buffered channel to allow one pending rebalance trigger.\n\ttriggerChan := make(chan string, 1)\n\n\t// Perform an initial rebalance on startup.\n\terr := p.rebalanceShards(ctx)\n\tif err != nil {\n\t\tp.logger.Error(\"initial rebalance failed\", tag.Error(err))\n\t}\n\n\tif err := p.runRebalanceTriggeringLoop(ctx, triggerChan); err != nil {\n\t\tp.logger.Error(\"failed to start rebalance triggering loop\", tag.Error(err))\n\t\treturn\n\t}\n\n\tnextRebalanceAllowedAt := p.timeSource.Now()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tp.logger.Info(\"Rebalancing loop cancelled\")\n\t\t\treturn\n\n\t\tcase triggerReason := <-triggerChan:\n\t\t\t// If an update comes in before the cooldown has expired,\n\t\t\t// we wait until the cooldown has passed since the last rebalance before processing it.\n\t\t\t// This ensures that we don't rebalance too frequently in response to a flurry of updates\n\t\t\tp.timeSource.Sleep(nextRebalanceAllowedAt.Sub(p.timeSource.Now()))\n\t\t\tnextRebalanceAllowedAt = p.timeSource.Now().Add(p.cfg.RebalanceCooldown)\n\n\t\t\tp.logger.Info(\"Rebalancing triggered\", tag.Dynamic(\"triggerReason\", triggerReason))\n\t\t\tif err := p.rebalanceShards(ctx); err != nil {\n\t\t\t\tp.logger.Error(\"rebalance failed\", tag.Error(err))\n\n\t\t\t\t// If rebalance fails, we want to trigger another rebalance ASAP,\n\t\t\t\t// but with a cooldown to avoid rebalance storms if the underlying issue is persistent.\n\t\t\t\tselect {\n\t\t\t\tcase triggerChan <- \"Previous rebalance failed\":\n\t\t\t\tdefault:\n\t\t\t\t\t// If the channel is full, we skip sending the update to avoid blocking the loop.\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n\n// runRebalanceTriggeringLoop monitors for state changes and periodic triggers to initiate rebalancing.\n// it doesn't block Subscribe calls to avoid a growing backlog of updates.\nfunc (p *namespaceProcessor) runRebalanceTriggeringLoop(ctx context.Context, triggerChan chan<- string) error {\n\tupdateChan, err := p.shardStore.SubscribeToExecutorStatusChanges(ctx, p.namespaceCfg.Name)\n\tif err != nil {\n\t\tp.logger.Error(\"Failed to subscribe to state changes, stopping rebalancing loop.\", tag.Error(err))\n\t\treturn err\n\t}\n\n\tgo p.rebalanceTriggeringLoop(ctx, updateChan, triggerChan)\n\treturn nil\n}\n\nfunc (p *namespaceProcessor) rebalanceTriggeringLoop(ctx context.Context, updateChan <-chan int64, triggerChan chan<- string) {\n\tticker := p.timeSource.NewTicker(p.cfg.Period)\n\tdefer ticker.Stop()\n\n\ttryTriggerRebalancing := func(reason string) {\n\t\tselect {\n\t\tcase triggerChan <- reason:\n\t\tdefault:\n\t\t\tp.logger.Info(\"Rebalance already pending, skipping trigger attempt\", tag.Dynamic(\"reason\", reason))\n\t\t}\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tp.logger.Info(\"Rebalance triggering loop cancelled\")\n\t\t\treturn\n\n\t\tcase <-ticker.Chan():\n\t\t\ttryTriggerRebalancing(\"Periodic reconciliation triggered\")\n\n\t\tcase _, ok := <-updateChan:\n\t\t\tif !ok {\n\t\t\t\tp.logger.Info(\"Update channel closed, stopping rebalance triggering loop\")\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\ttryTriggerRebalancing(\"State change detected\")\n\t\t}\n\t}\n}\n\n// runShardStatsCleanupLoop periodically removes stale shard statistics.\nfunc (p *namespaceProcessor) runShardStatsCleanupLoop(ctx context.Context) {\n\tticker := p.timeSource.NewTicker(p.cfg.HeartbeatTTL)\n\tdefer ticker.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tp.logger.Info(\"Shard stats cleanup loop cancelled.\")\n\t\t\treturn\n\t\tcase <-ticker.Chan():\n\t\t\t// Only perform shard stats cleanup in GREEDY load balancing mode\n\t\t\t// TODO: refactor this to not have this loop for non-GREEDY modes\n\t\t\tif p.sdConfig.GetLoadBalancingMode(p.namespaceCfg.Name) != types.LoadBalancingModeGREEDY {\n\t\t\t\tp.logger.Debug(\"Load balancing mode is not GREEDY, skipping shard stats cleanup.\", tag.ShardNamespace(p.namespaceCfg.Name))\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tp.logger.Info(\"Periodic shard stats cleanup triggered.\")\n\t\t\tnamespaceState, err := p.shardStore.GetState(ctx, p.namespaceCfg.Name)\n\t\t\tif err != nil {\n\t\t\t\tp.logger.Error(\"Failed to get state for shard stats cleanup\", tag.Error(err))\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tstaleShardStats := p.identifyStaleShardStats(namespaceState)\n\t\t\tif len(staleShardStats) == 0 {\n\t\t\t\t// No stale shard stats to delete\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif err := p.shardStore.DeleteShardStats(ctx, p.namespaceCfg.Name, staleShardStats, p.election.Guard()); err != nil {\n\t\t\t\tp.logger.Error(\"Failed to delete stale shard stats\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t}\n}\n\n// identifyStaleExecutors returns a list of executors who have not reported a heartbeat recently.\nfunc (p *namespaceProcessor) identifyStaleExecutors(namespaceState *store.NamespaceState) map[string]int64 {\n\texpiredExecutors := make(map[string]int64)\n\tnow := p.timeSource.Now().UTC()\n\n\tfor executorID, state := range namespaceState.Executors {\n\t\tif now.Sub(state.LastHeartbeat) > p.cfg.HeartbeatTTL {\n\t\t\tp.logger.Info(\"Executor has not reported a heartbeat recently\", tag.ShardExecutor(executorID), tag.ShardNamespace(p.namespaceCfg.Name), tag.Value(state.LastHeartbeat))\n\t\t\texpiredExecutors[executorID] = namespaceState.ShardAssignments[executorID].ModRevision\n\t\t}\n\t}\n\n\treturn expiredExecutors\n}\n\n// identifyStaleShardStats returns a list of shard statistics that are no longer relevant.\nfunc (p *namespaceProcessor) identifyStaleShardStats(namespaceState *store.NamespaceState) []string {\n\tactiveShards := make(map[string]struct{})\n\tnow := p.timeSource.Now().UTC()\n\n\t// 1. build set of active executors\n\n\t// add all assigned shards from executors that are ACTIVE and not stale\n\tfor executorID, assignedState := range namespaceState.ShardAssignments {\n\t\texecutor, exists := namespaceState.Executors[executorID]\n\t\tif !exists {\n\t\t\tcontinue\n\t\t}\n\n\t\tisActive := executor.Status == types.ExecutorStatusACTIVE\n\t\tisNotStale := now.Sub(executor.LastHeartbeat) <= p.cfg.HeartbeatTTL\n\t\tif isActive && isNotStale {\n\t\t\tfor shardID := range assignedState.AssignedShards {\n\t\t\t\tactiveShards[shardID] = struct{}{}\n\t\t\t}\n\t\t}\n\t}\n\n\t// add all shards in ReportedShards where the status is not DONE\n\tfor _, heartbeatState := range namespaceState.Executors {\n\t\tfor shardID, shardStatusReport := range heartbeatState.ReportedShards {\n\t\t\tif shardStatusReport.Status != types.ShardStatusDONE {\n\t\t\t\tactiveShards[shardID] = struct{}{}\n\t\t\t}\n\t\t}\n\t}\n\n\t// 2. build set of stale shard stats\n\n\t// append all shard stats that are not in the active shards set\n\tvar staleShardStats []string\n\tfor shardID, stats := range namespaceState.ShardStats {\n\t\tif _, ok := activeShards[shardID]; ok {\n\t\t\tcontinue\n\t\t}\n\t\trecentUpdate := !stats.LastUpdateTime.IsZero() && now.Sub(stats.LastUpdateTime) <= p.cfg.HeartbeatTTL\n\t\trecentMove := !stats.LastMoveTime.IsZero() && now.Sub(stats.LastMoveTime) <= p.cfg.HeartbeatTTL\n\t\tif recentUpdate || recentMove {\n\t\t\t// Preserve stats that have been updated recently to allow cooldown/load history to\n\t\t\t// survive executor churn. These shards are likely awaiting reassignment,\n\t\t\t// so we don't want to delete them.\n\t\t\tcontinue\n\t\t}\n\t\tstaleShardStats = append(staleShardStats, shardID)\n\t}\n\n\treturn staleShardStats\n}\n\n// rebalanceShards is the core logic for distributing shards among active executors.\nfunc (p *namespaceProcessor) rebalanceShards(ctx context.Context) (err error) {\n\tmetricsLoopScope := p.metricsClient.Scope(\n\t\tmetrics.ShardDistributorAssignLoopScope,\n\t\tmetrics.NamespaceTag(p.namespaceCfg.Name),\n\t\tmetrics.NamespaceTypeTag(p.namespaceCfg.Type),\n\t)\n\n\tmetricsLoopScope.AddCounter(metrics.ShardDistributorAssignLoopAttempts, 1)\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tmetricsLoopScope.AddCounter(metrics.ShardDistributorAssignLoopFail, 1)\n\t\t} else {\n\t\t\tmetricsLoopScope.AddCounter(metrics.ShardDistributorAssignLoopSuccess, 1)\n\t\t}\n\t}()\n\n\tstart := p.timeSource.Now()\n\tdefer func() {\n\t\tmetricsLoopScope.RecordHistogramDuration(metrics.ShardDistributorAssignLoopShardRebalanceLatency, p.timeSource.Now().Sub(start))\n\t}()\n\n\tctx, cancel := context.WithTimeout(ctx, p.cfg.Timeout)\n\tdefer cancel()\n\n\treturn p.rebalanceShardsImpl(ctx, metricsLoopScope)\n}\n\nfunc (p *namespaceProcessor) rebalanceShardsImpl(ctx context.Context, metricsLoopScope metrics.Scope) (err error) {\n\tnamespaceState, err := p.shardStore.GetState(ctx, p.namespaceCfg.Name)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"get state: %w\", err)\n\t}\n\n\t// Identify stale executors that need to be removed\n\tstaleExecutors := p.identifyStaleExecutors(namespaceState)\n\tif len(staleExecutors) > 0 {\n\t\tp.logger.Info(\"Identified stale executors for removal\", tag.ShardExecutors(slices.Collect(maps.Keys(staleExecutors))))\n\t}\n\n\tactiveExecutors := p.getActiveExecutors(namespaceState, staleExecutors)\n\tif len(activeExecutors) == 0 {\n\t\tp.logger.Error(\"No active executors found. Cannot assign shards.\")\n\n\t\t// Cleanup stale executors even if no active executors remain\n\t\tif len(staleExecutors) > 0 {\n\t\t\tp.logger.Info(\"Cleaning up stale executors (no active executors)\", tag.ShardExecutors(slices.Collect(maps.Keys(staleExecutors))))\n\t\t\tif err := p.shardStore.DeleteExecutors(ctx, p.namespaceCfg.Name, slices.Collect(maps.Keys(staleExecutors)), p.election.Guard()); err != nil {\n\t\t\t\tp.logger.Error(\"Failed to delete stale executors\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t}\n\tp.logger.Info(\"Active executors\", tag.ShardExecutors(activeExecutors))\n\n\tdeletedShards := p.findDeletedShards(namespaceState)\n\tshardsToReassign, currentAssignments := p.findShardsToReassign(activeExecutors, namespaceState, deletedShards, staleExecutors)\n\n\tmetricsLoopScope.UpdateGauge(metrics.ShardDistributorAssignLoopNumRebalancedShards, float64(len(shardsToReassign)))\n\n\t// If there are deleted shards or stale executors, the distribution has changed.\n\tassignedToEmptyExecutors := assignShardsToEmptyExecutors(currentAssignments)\n\tupdatedAssignments := p.updateAssignments(shardsToReassign, activeExecutors, currentAssignments)\n\tisRebalancedByShardLoad := p.rebalanceByShardLoad(calcShardLoad(namespaceState), currentAssignments)\n\n\tdistributionChanged := len(deletedShards) > 0 || len(staleExecutors) > 0 || assignedToEmptyExecutors || updatedAssignments || isRebalancedByShardLoad\n\tif !distributionChanged {\n\t\tp.logger.Info(\"No changes to distribution detected. Skipping rebalance.\")\n\t\treturn nil\n\t}\n\n\tnewState := p.getNewAssignmentsState(namespaceState, currentAssignments)\n\n\tp.emitExecutorMetric(namespaceState, metricsLoopScope)\n\tp.emitOldestExecutorHeartbeatLag(namespaceState, metricsLoopScope)\n\n\tif p.sdConfig.GetMigrationMode(p.namespaceCfg.Name) != types.MigrationModeONBOARDED {\n\t\tp.logger.Info(\"Running rebalancing in shadow mode\", tag.Dynamic(\"old_assignments\", namespaceState.ShardAssignments), tag.Dynamic(\"new_assignments\", newState))\n\t\tp.emitActiveShardMetric(namespaceState.ShardAssignments, metricsLoopScope)\n\n\t\tif len(staleExecutors) > 0 {\n\t\t\tp.logger.Info(\"Cleaning up stale executors in shadow mode\", tag.ShardExecutors(slices.Collect(maps.Keys(staleExecutors))))\n\t\t\tif err := p.shardStore.DeleteExecutors(ctx, p.namespaceCfg.Name, slices.Collect(maps.Keys(staleExecutors)), p.election.Guard()); err != nil {\n\t\t\t\tp.logger.Error(\"Failed to delete stale executors in shadow mode\", tag.Error(err))\n\t\t\t\t// Non-blocking: stale executors in shadow mode will be cleaned up the next cycle\n\t\t\t}\n\t\t}\n\t\treturn nil\n\t}\n\n\tnamespaceState.ShardAssignments = newState\n\tp.logger.Info(\"Applying new shard distribution.\")\n\n\t// Use the leader guard for the assign and delete operation.\n\terr = p.shardStore.AssignShards(ctx, p.namespaceCfg.Name, store.AssignShardsRequest{\n\t\tNewState:          namespaceState,\n\t\tExecutorsToDelete: staleExecutors,\n\t}, p.election.Guard())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"assign shards: %w\", err)\n\t}\n\n\tp.emitActiveShardMetric(namespaceState.ShardAssignments, metricsLoopScope)\n\treturn nil\n}\n\nfunc (p *namespaceProcessor) emitActiveShardMetric(shardAssignments map[string]store.AssignedState, metricsLoopScope metrics.Scope) {\n\ttotalActiveShards := 0\n\tfor _, assignedState := range shardAssignments {\n\t\ttotalActiveShards += len(assignedState.AssignedShards)\n\t}\n\tmetricsLoopScope.UpdateGauge(metrics.ShardDistributorActiveShards, float64(totalActiveShards))\n}\n\nfunc (p *namespaceProcessor) emitExecutorMetric(namespaceState *store.NamespaceState, metricsLoopScope metrics.Scope) {\n\tfor status, count := range namespaceState.CountExecutorsByStatus() {\n\t\tmetricsLoopScope.Tagged(metrics.ExecutorStatusTag(status.String())).UpdateGauge(metrics.ShardDistributorTotalExecutors, float64(count))\n\t}\n}\n\nfunc (p *namespaceProcessor) emitOldestExecutorHeartbeatLag(namespaceState *store.NamespaceState, metricsLoopScope metrics.Scope) {\n\tif len(namespaceState.Executors) == 0 {\n\t\treturn\n\t}\n\n\tvar oldestHeartbeat time.Time\n\tfor _, executor := range namespaceState.Executors {\n\t\tif oldestHeartbeat.IsZero() || executor.LastHeartbeat.Before(oldestHeartbeat) {\n\t\t\toldestHeartbeat = executor.LastHeartbeat\n\t\t}\n\t}\n\n\tlag := p.timeSource.Now().Sub(oldestHeartbeat)\n\tmetricsLoopScope.UpdateGauge(metrics.ShardDistributorOldestExecutorHeartbeatLag, float64(lag.Milliseconds()))\n}\n\nfunc (p *namespaceProcessor) findDeletedShards(namespaceState *store.NamespaceState) map[string]store.ShardState {\n\tdeletedShards := make(map[string]store.ShardState)\n\n\tfor executorID, executor := range namespaceState.Executors {\n\t\tfor shardID, shardState := range executor.ReportedShards {\n\t\t\tif shardState.Status == types.ShardStatusDONE {\n\t\t\t\tdeletedShards[shardID] = store.ShardState{\n\t\t\t\t\tExecutorID: executorID,\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\treturn deletedShards\n}\n\nfunc (p *namespaceProcessor) findShardsToReassign(\n\tactiveExecutors []string,\n\tnamespaceState *store.NamespaceState,\n\tdeletedShards map[string]store.ShardState,\n\tstaleExecutors map[string]int64,\n) ([]string, map[string][]string) {\n\tallShards := make(map[string]struct{})\n\tfor _, shardID := range getShards(p.namespaceCfg, namespaceState, deletedShards) {\n\t\tallShards[shardID] = struct{}{}\n\t}\n\n\tshardsToReassign := make([]string, 0)\n\tcurrentAssignments := make(map[string][]string)\n\n\tfor _, executorID := range activeExecutors {\n\t\tcurrentAssignments[executorID] = []string{}\n\t}\n\n\tfor executorID, state := range namespaceState.ShardAssignments {\n\t\tisActive := namespaceState.Executors[executorID].Status == types.ExecutorStatusACTIVE\n\t\t_, isStale := staleExecutors[executorID]\n\n\t\tfor shardID := range state.AssignedShards {\n\t\t\tif _, ok := allShards[shardID]; ok {\n\t\t\t\tdelete(allShards, shardID)\n\t\t\t\t// If executor is active AND not stale, keep the assignment\n\t\t\t\tif isActive && !isStale {\n\t\t\t\t\tcurrentAssignments[executorID] = append(currentAssignments[executorID], shardID)\n\t\t\t\t} else {\n\t\t\t\t\t// Otherwise, reassign the shard (executor is either inactive or stale)\n\t\t\t\t\tshardsToReassign = append(shardsToReassign, shardID)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\tfor shardID := range allShards {\n\t\tshardsToReassign = append(shardsToReassign, shardID)\n\t}\n\treturn shardsToReassign, currentAssignments\n}\n\nfunc (*namespaceProcessor) updateAssignments(shardsToReassign []string, activeExecutors []string, currentAssignments map[string][]string) (distributionChanged bool) {\n\tif len(shardsToReassign) == 0 {\n\t\treturn false\n\t}\n\n\ti := rand.Intn(len(activeExecutors))\n\tfor _, shardID := range shardsToReassign {\n\t\texecutorID := activeExecutors[i%len(activeExecutors)]\n\t\tcurrentAssignments[executorID] = append(currentAssignments[executorID], shardID)\n\t\ti++\n\t}\n\n\treturn true\n}\n\n// calcShardLoad returns a map of shardID to its load based on the latest reported shard loads from executors\nfunc calcShardLoad(namespaceState *store.NamespaceState) map[string]float64 {\n\tshardLoad := make(map[string]float64)\n\tfor _, state := range namespaceState.Executors {\n\t\tfor shardID, report := range state.ReportedShards {\n\t\t\tshardLoad[shardID] = report.ShardLoad\n\t\t}\n\t}\n\treturn shardLoad\n}\n\n// rebalanceByShardLoad does a rebalance if a difference between hottest and coldest executors' loads is more than maxDeviation\n// in this case the hottest shard will be moved to the coldest executor\nfunc (p *namespaceProcessor) rebalanceByShardLoad(shardLoad map[string]float64, currentAssignments map[string][]string) (distributedChanged bool) {\n\t// no rebalance if there are no more than 1 executor\n\tif len(currentAssignments) < 2 {\n\t\treturn false\n\t}\n\n\tvar (\n\t\thottestExecutorLoad = float64(0)\n\t\thottestExecutorID   = \"\"\n\n\t\thottestShardID   = \"\"\n\t\thottestShardLoad = float64(0)\n\n\t\tcoldestExecutorLoad = math.MaxFloat64\n\t\tcoldestExecutorID   = \"\"\n\t)\n\n\t// finding loads of hottest, coldest executors and hottest shard\n\texecutorLoad := make(map[string]float64)\n\tfor executorID, shardIDs := range currentAssignments {\n\t\tfor _, shardID := range shardIDs {\n\t\t\texecutorLoad[executorID] += shardLoad[shardID]\n\t\t}\n\n\t\tif executorLoad[executorID] <= coldestExecutorLoad {\n\t\t\tcoldestExecutorLoad = executorLoad[executorID]\n\t\t\tcoldestExecutorID = executorID\n\t\t}\n\n\t\tif executorLoad[executorID] >= hottestExecutorLoad {\n\t\t\thottestExecutorLoad = executorLoad[executorID]\n\t\t\thottestExecutorID = executorID\n\n\t\t\tvar maxShardLoad = float64(0)\n\t\t\tfor _, shardID := range shardIDs {\n\t\t\t\tif shardLoad[shardID] >= maxShardLoad {\n\t\t\t\t\thottestShardID = shardID\n\t\t\t\t\tmaxShardLoad = shardLoad[shardID]\n\t\t\t\t}\n\t\t\t}\n\t\t\thottestShardLoad = maxShardLoad\n\t\t}\n\t}\n\n\t// no rebalance if a deviation between coldest and hottest executors less than maxDeviation\n\tif hottestExecutorLoad/coldestExecutorLoad < p.sdConfig.LoadBalancingNaive.MaxDeviation(p.namespaceCfg.Name) {\n\t\treturn false\n\t}\n\n\t// no rebalance if coldest executor becomes a hottest\n\tif coldestExecutorLoad+hottestShardLoad >= hottestExecutorLoad {\n\t\treturn false\n\t}\n\n\t// remove the hottest Shard from the hottest executor\n\t// put it to the coldest executor\n\tfor i, shardID := range currentAssignments[hottestExecutorID] {\n\t\tif shardID == hottestShardID {\n\t\t\tcurrentAssignments[hottestExecutorID] = append(currentAssignments[hottestExecutorID][:i], currentAssignments[hottestExecutorID][i+1:]...)\n\t\t}\n\t}\n\tcurrentAssignments[coldestExecutorID] = append(currentAssignments[coldestExecutorID], hottestShardID)\n\n\treturn true\n}\n\nfunc (p *namespaceProcessor) getNewAssignmentsState(namespaceState *store.NamespaceState, currentAssignments map[string][]string) map[string]store.AssignedState {\n\tnewState := make(map[string]store.AssignedState, len(currentAssignments))\n\n\tfor executorID, shards := range currentAssignments {\n\t\tassignedShardsMap := make(map[string]*types.ShardAssignment)\n\n\t\tfor _, shardID := range shards {\n\t\t\tassignedShardsMap[shardID] = &types.ShardAssignment{Status: types.AssignmentStatusREADY}\n\t\t}\n\n\t\tmodRevision := int64(0) // Should be 0 if we have not seen it yet\n\t\tif namespaceAssignments, ok := namespaceState.ShardAssignments[executorID]; ok {\n\t\t\tmodRevision = namespaceAssignments.ModRevision\n\t\t}\n\n\t\tnewState[executorID] = store.AssignedState{\n\t\t\tAssignedShards:     assignedShardsMap,\n\t\t\tLastUpdated:        p.timeSource.Now().UTC(),\n\t\t\tModRevision:        modRevision,\n\t\t\tShardHandoverStats: p.addHandoverStatsToExecutorAssignedState(namespaceState, executorID, shards),\n\t\t}\n\t}\n\n\treturn newState\n}\n\nfunc (p *namespaceProcessor) addHandoverStatsToExecutorAssignedState(\n\tnamespaceState *store.NamespaceState,\n\texecutorID string, shardIDs []string,\n) map[string]store.ShardHandoverStats {\n\tvar newStats = make(map[string]store.ShardHandoverStats)\n\n\t// Prepare handover stats for each shard\n\tfor _, shardID := range shardIDs {\n\t\thandoverStats := p.newHandoverStats(namespaceState, shardID, executorID)\n\n\t\t// If there is no handover (first assignment), we skip adding handover stats\n\t\tif handoverStats != nil {\n\t\t\tnewStats[shardID] = *handoverStats\n\t\t}\n\t}\n\n\treturn newStats\n}\n\n// newHandoverStats creates shard handover statistics if a handover occurred.\nfunc (p *namespaceProcessor) newHandoverStats(\n\tnamespaceState *store.NamespaceState,\n\tshardID string,\n\tnewExecutorID string,\n) *store.ShardHandoverStats {\n\tlogger := p.logger.WithTags(\n\t\ttag.ShardNamespace(p.namespaceCfg.Name),\n\t\ttag.ShardKey(shardID),\n\t\ttag.ShardExecutor(newExecutorID),\n\t)\n\n\t// Fetch previous shard owners from cache\n\tprevExecutor, err := p.shardStore.GetShardOwner(context.Background(), p.namespaceCfg.Name, shardID)\n\tif err != nil && !errors.Is(err, store.ErrShardNotFound) {\n\t\tlogger.Warn(\"failed to get shard owner for shard statistic\", tag.Error(err))\n\t\treturn nil\n\t}\n\t// previous executor is not found in cache\n\t// meaning this is the first assignment of the shard\n\t// so we skip updating handover stats\n\tif prevExecutor == nil {\n\t\treturn nil\n\t}\n\n\t// No change in assignment\n\t// meaning no handover occurred\n\t// skip updating handover stats\n\tif prevExecutor.ExecutorID == newExecutorID {\n\t\treturn nil\n\t}\n\n\t// previous executor heartbeat is not found in namespace state\n\t// meaning the executor has already been cleaned up\n\t// skip updating handover stats\n\tprevExecutorHeartbeat, ok := namespaceState.Executors[prevExecutor.ExecutorID]\n\tif !ok {\n\t\tlogger.Info(\"previous executor heartbeat not found, skipping handover stats\")\n\t\treturn nil\n\t}\n\n\thandoverType := types.HandoverTypeEMERGENCY\n\n\t// Consider it a graceful handover if the previous executor was in DRAINING or DRAINED status\n\t// otherwise, it's an emergency handover\n\tif prevExecutorHeartbeat.Status == types.ExecutorStatusDRAINING || prevExecutorHeartbeat.Status == types.ExecutorStatusDRAINED {\n\t\thandoverType = types.HandoverTypeGRACEFUL\n\t}\n\n\treturn &store.ShardHandoverStats{\n\t\tHandoverType:                      handoverType,\n\t\tPreviousExecutorLastHeartbeatTime: prevExecutorHeartbeat.LastHeartbeat,\n\t}\n}\n\nfunc (*namespaceProcessor) getActiveExecutors(namespaceState *store.NamespaceState, staleExecutors map[string]int64) []string {\n\tvar activeExecutors []string\n\tfor id, state := range namespaceState.Executors {\n\t\t// Executor must be ACTIVE and not stale\n\t\tif state.Status == types.ExecutorStatusACTIVE {\n\t\t\tif _, ok := staleExecutors[id]; !ok {\n\t\t\t\tactiveExecutors = append(activeExecutors, id)\n\t\t\t}\n\t\t}\n\t}\n\n\tsort.Strings(activeExecutors)\n\treturn activeExecutors\n}\n\nfunc assignShardsToEmptyExecutors(currentAssignments map[string][]string) bool {\n\temptyExecutors := make([]string, 0)\n\texecutorsWithShards := make([]string, 0)\n\tminShardsCurrentlyAssigned := 0\n\n\t// Ensure the iteration is deterministic.\n\texecutors := make([]string, 0, len(currentAssignments))\n\tfor executorID := range currentAssignments {\n\t\texecutors = append(executors, executorID)\n\t}\n\tslices.Sort(executors)\n\n\tfor _, executorID := range executors {\n\t\tif len(currentAssignments[executorID]) == 0 {\n\t\t\temptyExecutors = append(emptyExecutors, executorID)\n\t\t} else {\n\t\t\texecutorsWithShards = append(executorsWithShards, executorID)\n\t\t\tif minShardsCurrentlyAssigned == 0 || len(currentAssignments[executorID]) < minShardsCurrentlyAssigned {\n\t\t\t\tminShardsCurrentlyAssigned = len(currentAssignments[executorID])\n\t\t\t}\n\t\t}\n\t}\n\n\t// If there are no empty executors or no executors with shards, we don't need to do anything.\n\tif len(emptyExecutors) == 0 || len(executorsWithShards) == 0 {\n\t\treturn false\n\t}\n\n\t// We calculate the number of shards to assign each of the empty executors. The idea is to assume all current executors have\n\t// the same number of shards `minShardsCurrentlyAssigned`. We use the minimum so when steeling we don't have to worry about\n\t// steeling more shards that the executors have.\n\t// We then calculate the total number of assumed shards `minShardsCurrentlyAssigned * len(executorsWithShards)` and divide it by the\n\t// number of current executors. This gives us the number of shards per executor, thus the number of shards to assign to each of the\n\t// empty executors.\n\tnumShardsToAssignEmptyExecutors := minShardsCurrentlyAssigned * len(executorsWithShards) / len(currentAssignments)\n\n\tstealRound := 0\n\tfor i := 0; i < numShardsToAssignEmptyExecutors; i++ {\n\t\tfor _, emptyExecutor := range emptyExecutors {\n\t\t\texecutorToSteelFrom := executorsWithShards[stealRound%len(executorsWithShards)]\n\t\t\tstealRound++\n\n\t\t\tstolenShard := currentAssignments[executorToSteelFrom][0]\n\n\t\t\tcurrentAssignments[executorToSteelFrom] = currentAssignments[executorToSteelFrom][1:]\n\t\t\tcurrentAssignments[emptyExecutor] = append(currentAssignments[emptyExecutor], stolenShard)\n\t\t}\n\t}\n\n\treturn true\n}\n\nfunc getShards(cfg config.Namespace, namespaceState *store.NamespaceState, deletedShards map[string]store.ShardState) []string {\n\tif cfg.Type == config.NamespaceTypeFixed {\n\t\treturn makeShards(cfg.ShardNum)\n\t} else if cfg.Type == config.NamespaceTypeEphemeral {\n\t\tshards := make([]string, 0)\n\t\tfor _, state := range namespaceState.ShardAssignments {\n\t\t\tfor shardID := range state.AssignedShards {\n\t\t\t\tif _, ok := deletedShards[shardID]; !ok {\n\t\t\t\t\tshards = append(shards, shardID)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn shards\n\t}\n\treturn nil\n}\n\nfunc makeShards(num int64) []string {\n\tshards := make([]string, num)\n\tfor i := range num {\n\t\tshards[i] = strconv.FormatInt(i, 10)\n\t}\n\treturn shards\n}\n"
  },
  {
    "path": "service/sharddistributor/leader/process/processor_test.go",
    "content": "package process\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"slices\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmetricmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/config/configtest\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\ntype testDependencies struct {\n\tctrl       *gomock.Controller\n\tstore      *store.MockStore\n\telection   *store.MockElection\n\ttimeSource clock.MockedTimeSource\n\tfactory    Factory\n\tcfg        config.Namespace\n\tsdConfig   *config.Config\n}\n\nfunc setupProcessorTest(t *testing.T, namespaceType string) *testDependencies {\n\tmigrationConfig := configtest.NewTestMigrationConfig(t,\n\t\tconfigtest.ConfigEntry{\n\t\t\tKey:   dynamicproperties.ShardDistributorMigrationMode,\n\t\t\tValue: config.MigrationModeONBOARDED})\n\treturn setupProcessorTestWithMigrationConfig(t, namespaceType, migrationConfig)\n}\n\nfunc setupProcessorTestWithMigrationConfig(t *testing.T, namespaceType string, migrationConfig *config.Config) *testDependencies {\n\tctrl := gomock.NewController(t)\n\tmockedClock := clock.NewMockedTimeSource()\n\tdeps := &testDependencies{\n\t\tctrl:       ctrl,\n\t\tstore:      store.NewMockStore(ctrl),\n\t\telection:   store.NewMockElection(ctrl),\n\t\ttimeSource: mockedClock,\n\t\tcfg:        config.Namespace{Name: \"test-ns\", ShardNum: 2, Type: namespaceType, Mode: config.MigrationModeONBOARDED},\n\t}\n\tdeps.sdConfig = &config.Config{\n\t\tLoadBalancingMode: func(namespace string) string {\n\t\t\treturn config.LoadBalancingModeNAIVE\n\t\t},\n\t\tLoadBalancingNaive: config.LoadBalancingNaiveConfig{\n\t\t\tMaxDeviation: func(namespace string) float64 {\n\t\t\t\treturn 2.0\n\t\t\t},\n\t\t},\n\t\tMigrationMode: migrationConfig.MigrationMode,\n\t}\n\n\tdeps.factory = NewProcessorFactory(\n\t\ttestlogger.New(t),\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tmockedClock,\n\t\tconfig.ShardDistribution{\n\t\t\tProcess: config.LeaderProcess{\n\t\t\t\tPeriod:       time.Second,\n\t\t\t\tHeartbeatTTL: time.Second,\n\t\t\t},\n\t\t},\n\t\tdeps.sdConfig,\n\t)\n\treturn deps\n}\n\nfunc TestRunAndTerminate(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election)\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{}, nil).AnyTimes()\n\tmocks.store.EXPECT().SubscribeToExecutorStatusChanges(gomock.Any(), mocks.cfg.Name).Return(make(chan int64), nil).AnyTimes()\n\n\terr := processor.Run(ctx)\n\trequire.NoError(t, err)\n\n\terr = processor.Run(ctx)\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"processor is already running\")\n\n\terr = processor.Terminate(context.Background())\n\trequire.NoError(t, err)\n\n\terr = processor.Terminate(context.Background())\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"processor has not been started\")\n\n\tcancel()\n}\n\nfunc TestRebalanceShards_InitialDistribution(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\tstate := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t}\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{Executors: state}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(nil, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(nil, nil)\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, _ string, request store.AssignShardsRequest, _ store.GuardFunc) error {\n\t\t\tassert.Len(t, request.NewState.ShardAssignments, 2)\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].AssignedShards, 1)\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-2\"].AssignedShards, 1)\n\t\t\tassert.Lenf(t, request.NewState.ShardAssignments[\"exec-1\"].ShardHandoverStats, 0, \"no handover stats should be present on initial assignment\")\n\t\t\tassert.Lenf(t, request.NewState.ShardAssignments[\"exec-2\"].ShardHandoverStats, 0, \"no handover stats should be present on initial assignment\")\n\t\t\treturn nil\n\t\t},\n\t)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_ExecutorRemoved(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t\"exec-2\": {Status: types.ExecutorStatusDRAINING, LastHeartbeat: now},\n\t}\n\tassignments := map[string]store.AssignedState{\n\t\t\"exec-2\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t},\n\t}\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors:        heartbeats,\n\t\tShardAssignments: assignments,\n\t}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(&store.ShardOwner{ExecutorID: \"exec-2\"}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(&store.ShardOwner{ExecutorID: \"exec-1\"}, nil)\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, _ string, request store.AssignShardsRequest, _ store.GuardFunc) error {\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].AssignedShards, 2)\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-2\"].AssignedShards, 0)\n\t\t\tassert.Lenf(t, request.NewState.ShardAssignments[\"exec-1\"].ShardHandoverStats, 1, \"only shard 0 should have handover stats\")\n\t\t\tassert.Lenf(t, request.NewState.ShardAssignments[\"exec-2\"].ShardHandoverStats, 0, \"no handover stats should be present for drained executor\")\n\t\t\treturn nil\n\t\t},\n\t)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_ExecutorStale(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-2 * time.Second)},\n\t}\n\tassignments := map[string]store.AssignedState{\n\t\t\"exec-1\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\tModRevision: 1,\n\t\t},\n\t\t\"exec-2\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t\tModRevision: 1,\n\t\t},\n\t}\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors:        heartbeats,\n\t\tShardAssignments: assignments,\n\t}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(&store.ShardOwner{ExecutorID: \"exec-1\"}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(&store.ShardOwner{ExecutorID: \"exec-2\"}, nil)\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, _ string, request store.AssignShardsRequest, _ store.GuardFunc) error {\n\t\t\tassert.Len(t, request.NewState.ShardAssignments, 1)\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].AssignedShards, 2)\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].ShardHandoverStats, 1, \"only shard 1 should have handover stats\")\n\t\t\tassert.Equal(t, request.ExecutorsToDelete, map[string]int64{\"exec-2\": 1})\n\t\t\treturn nil\n\t\t},\n\t)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_NoActiveExecutors(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\tstate := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusDRAINING, LastHeartbeat: now},\n\t}\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{Executors: state}, nil)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_NoActiveExecutors_WithStaleExecutors(t *testing.T) {\n\tt.Run(\"one stale executor\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\t\texecutorStates := map[string]store.HeartbeatState{\n\t\t\t\"exec-1\": {Status: types.ExecutorStatusDRAINING, LastHeartbeat: now},\n\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-10 * time.Minute)},\n\t\t}\n\t\texpectedStaleExecutorIDs := []string{\"exec-2\"}\n\n\t\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\t\tExecutors: executorStates,\n\t\t}, nil)\n\n\t\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\t\tmocks.store.EXPECT().DeleteExecutors(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).\n\t\t\tDoAndReturn(func(_ context.Context, namespace string, executorIDs []string, _ store.GuardFunc) error {\n\t\t\t\tassert.ElementsMatch(t, expectedStaleExecutorIDs, executorIDs)\n\t\t\t\tassert.Equal(t, mocks.cfg.Name, namespace)\n\t\t\t\treturn nil\n\t\t\t})\n\n\t\terr := processor.rebalanceShards(context.Background())\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"all stale executor\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\t\texecutorStates := map[string]store.HeartbeatState{\n\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-10 * time.Minute)},\n\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-10 * time.Minute)},\n\t\t}\n\t\texpectedStaleExecutorIDs := []string{\"exec-1\", \"exec-2\"}\n\n\t\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\t\tExecutors: executorStates,\n\t\t}, nil)\n\n\t\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\t\tmocks.store.EXPECT().DeleteExecutors(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).\n\t\t\tDoAndReturn(func(_ context.Context, namespace string, executorIDs []string, _ store.GuardFunc) error {\n\t\t\t\tassert.ElementsMatch(t, expectedStaleExecutorIDs, executorIDs)\n\t\t\t\tassert.Equal(t, mocks.cfg.Name, namespace)\n\t\t\t\treturn nil\n\t\t\t})\n\n\t\terr := processor.rebalanceShards(context.Background())\n\t\trequire.NoError(t, err)\n\t})\n}\n\nfunc TestCleanupStaleExecutors(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\tnow := mocks.timeSource.Now()\n\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-active\": {LastHeartbeat: now},\n\t\t\"exec-stale\":  {LastHeartbeat: now.Add(-2 * time.Second)},\n\t}\n\n\tnamespaceState := &store.NamespaceState{Executors: heartbeats}\n\n\tstaleExecutors := processor.identifyStaleExecutors(namespaceState)\n\tassert.Equal(t, map[string]int64{\"exec-stale\": 0}, staleExecutors)\n}\n\nfunc TestCleanupStaleShardStats(t *testing.T) {\n\tt.Run(\"stale shard stats are deleted\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now().UTC()\n\n\t\theartbeats := map[string]store.HeartbeatState{\n\t\t\t\"exec-active\": {LastHeartbeat: now, Status: types.ExecutorStatusACTIVE},\n\t\t\t\"exec-stale\":  {LastHeartbeat: now.Add(-2 * time.Second)},\n\t\t}\n\n\t\tassignments := map[string]store.AssignedState{\n\t\t\t\"exec-active\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard-1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\"shard-2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"exec-stale\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"shard-3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tshardStats := map[string]store.ShardStatistics{\n\t\t\t\"shard-1\": {SmoothedLoad: 1.0, LastUpdateTime: now, LastMoveTime: now},\n\t\t\t\"shard-2\": {SmoothedLoad: 2.0, LastUpdateTime: now, LastMoveTime: now},\n\t\t\t\"shard-3\": {SmoothedLoad: 3.0, LastUpdateTime: now.Add(-2 * time.Second), LastMoveTime: now.Add(-2 * time.Second)},\n\t\t}\n\n\t\tnamespaceState := &store.NamespaceState{\n\t\t\tExecutors:        heartbeats,\n\t\t\tShardAssignments: assignments,\n\t\t\tShardStats:       shardStats,\n\t\t}\n\n\t\tstaleShardStats := processor.identifyStaleShardStats(namespaceState)\n\t\tassert.Equal(t, []string{\"shard-3\"}, staleShardStats)\n\t})\n\n\tt.Run(\"recent shard stats are preserved\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\n\t\texpiredExecutor := now.Add(-2 * time.Second)\n\t\tnamespaceState := &store.NamespaceState{\n\t\t\tExecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"exec-stale\": {LastHeartbeat: expiredExecutor},\n\t\t\t},\n\t\t\tShardAssignments: map[string]store.AssignedState{},\n\t\t\tShardStats: map[string]store.ShardStatistics{\n\t\t\t\t\"shard-1\": {SmoothedLoad: 5.0, LastUpdateTime: now, LastMoveTime: now},\n\t\t\t},\n\t\t}\n\n\t\tstaleShardStats := processor.identifyStaleShardStats(namespaceState)\n\t\tassert.Empty(t, staleShardStats)\n\t})\n\n}\n\nfunc TestRebalance_StoreErrors(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\texpectedErr := errors.New(\"store is down\")\n\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(nil, expectedErr)\n\terr := processor.rebalanceShards(context.Background())\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), expectedErr.Error())\n\n\tnow := mocks.timeSource.Now()\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors: map[string]store.HeartbeatState{\"e\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now}},\n\t}, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(nil, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(nil, nil)\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).Return(expectedErr)\n\terr = processor.rebalanceShards(context.Background())\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), expectedErr.Error())\n}\n\nfunc TestRunLoop_SubscriptionError(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\texpectedErr := errors.New(\"subscription failed\")\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{}, nil)\n\tmocks.store.EXPECT().SubscribeToExecutorStatusChanges(gomock.Any(), mocks.cfg.Name).Return(nil, expectedErr)\n\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\tprocessor.runRebalancingLoop(context.Background())\n\t}()\n\twg.Wait()\n}\n\nfunc TestRunLoop_ContextCancellation(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\tctx, cancel := context.WithCancel(context.Background())\n\n\t// Setup for the initial call to rebalanceShards and the subscription\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{}, nil)\n\tmocks.store.EXPECT().SubscribeToExecutorStatusChanges(gomock.Any(), mocks.cfg.Name).Return(make(chan int64), nil)\n\n\tprocessor.wg.Add(1)\n\t// Run the process in a separate goroutine to avoid blocking the test\n\tgo processor.runProcess(ctx)\n\n\t// Wait for the two loops (rebalance and cleanup) to create their tickers\n\tmocks.timeSource.BlockUntil(2)\n\n\t// Now, cancel the context to signal the loops to stop\n\tcancel()\n\n\t// Wait for the main process loop to exit gracefully\n\tprocessor.wg.Wait()\n}\n\nfunc TestRebalanceShards_WithUnassignedShardsButMigrationModeNotOnboarded(t *testing.T) {\n\tmigrationConfig := configtest.NewTestMigrationConfig(t, configtest.ConfigEntry{\n\t\tKey:   dynamicproperties.ShardDistributorMigrationMode,\n\t\tValue: config.MigrationModeDISTRIBUTEDPASSTHROUGH})\n\tmocks := setupProcessorTestWithMigrationConfig(t, config.NamespaceTypeFixed, migrationConfig)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t}\n\t// Note: shard \"1\" is missing from assignments\n\tassignments := map[string]store.AssignedState{\n\t\t\"exec-1\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t},\n\t}\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(nil, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(nil, nil)\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors:        heartbeats,\n\t\tShardAssignments: assignments,\n\t}, nil)\n\t// These are the expected calls in case of onboarding, with the assignment of new shards\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard()).Times(0)\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, _ string, request store.AssignShardsRequest, _ store.GuardFunc) error {\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].AssignedShards, 2, \"Both shards should now be assigned to exec-1\")\n\t\t\treturn nil\n\t\t},\n\t).Times(0)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_ShadowModeWithStaleExecutors(t *testing.T) {\n\tt.Run(\"stale executors are deleted in shadow mode\", func(t *testing.T) {\n\t\tmigrationConfig := configtest.NewTestMigrationConfig(t, configtest.ConfigEntry{\n\t\t\tKey:   dynamicproperties.ShardDistributorMigrationMode,\n\t\t\tValue: config.MigrationModeDISTRIBUTEDPASSTHROUGH})\n\t\tmocks := setupProcessorTestWithMigrationConfig(t, config.NamespaceTypeFixed, migrationConfig)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\t\theartbeats := map[string]store.HeartbeatState{\n\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-10 * time.Second)},\n\t\t}\n\t\tassignments := map[string]store.AssignedState{\n\t\t\t\"exec-1\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tModRevision: 1,\n\t\t\t},\n\t\t\t\"exec-2\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tModRevision: 1,\n\t\t\t},\n\t\t}\n\t\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\t\tExecutors:        heartbeats,\n\t\t\tShardAssignments: assignments,\n\t\t}, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(&store.ShardOwner{ExecutorID: \"exec-1\"}, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(&store.ShardOwner{ExecutorID: \"exec-2\"}, nil)\n\t\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\t\tmocks.store.EXPECT().DeleteExecutors(gomock.Any(), mocks.cfg.Name, []string{\"exec-2\"}, gomock.Any()).Return(nil)\n\n\t\terr := processor.rebalanceShards(context.Background())\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"delete executors error is non-blocking in shadow mode\", func(t *testing.T) {\n\t\tmigrationConfig := configtest.NewTestMigrationConfig(t, configtest.ConfigEntry{\n\t\t\tKey:   dynamicproperties.ShardDistributorMigrationMode,\n\t\t\tValue: config.MigrationModeDISTRIBUTEDPASSTHROUGH})\n\t\tmocks := setupProcessorTestWithMigrationConfig(t, config.NamespaceTypeFixed, migrationConfig)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\t\theartbeats := map[string]store.HeartbeatState{\n\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now.Add(-10 * time.Second)},\n\t\t}\n\t\tassignments := map[string]store.AssignedState{\n\t\t\t\"exec-1\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tModRevision: 1,\n\t\t\t},\n\t\t\t\"exec-2\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tModRevision: 1,\n\t\t\t},\n\t\t}\n\t\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\t\tExecutors:        heartbeats,\n\t\t\tShardAssignments: assignments,\n\t\t}, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(&store.ShardOwner{ExecutorID: \"exec-1\"}, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(&store.ShardOwner{ExecutorID: \"exec-2\"}, nil)\n\t\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\t\tmocks.store.EXPECT().DeleteExecutors(gomock.Any(), mocks.cfg.Name, []string{\"exec-2\"}, gomock.Any()).Return(errors.New(\"transaction failed\"))\n\n\t\terr := processor.rebalanceShards(context.Background())\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"no stale executors - delete not called in shadow mode\", func(t *testing.T) {\n\t\tmigrationConfig := configtest.NewTestMigrationConfig(t, configtest.ConfigEntry{\n\t\t\tKey:   dynamicproperties.ShardDistributorMigrationMode,\n\t\t\tValue: config.MigrationModeDISTRIBUTEDPASSTHROUGH})\n\t\tmocks := setupProcessorTestWithMigrationConfig(t, config.NamespaceTypeFixed, migrationConfig)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tnow := mocks.timeSource.Now()\n\t\theartbeats := map[string]store.HeartbeatState{\n\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t\t}\n\t\tassignments := map[string]store.AssignedState{\n\t\t\t\"exec-1\": {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\t\tExecutors:        heartbeats,\n\t\t\tShardAssignments: assignments,\n\t\t}, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(nil, nil)\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(nil, nil)\n\t\t// DeleteExecutors should not be called when there are no stale executors, thus Times(0)\n\t\tmocks.store.EXPECT().DeleteExecutors(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(0)\n\n\t\terr := processor.rebalanceShards(context.Background())\n\t\trequire.NoError(t, err)\n\t})\n}\n\nfunc TestRebalanceShards_NoShardsToReassign(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t}\n\tassignments := map[string]store.AssignedState{\n\t\t\"exec-1\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t},\n\t}\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors:        heartbeats,\n\t\tShardAssignments: assignments,\n\t}, nil)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestRebalanceShards_WithUnassignedShards(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := mocks.timeSource.Now()\n\theartbeats := map[string]store.HeartbeatState{\n\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE, LastHeartbeat: now},\n\t}\n\t// Note: shard \"1\" is missing from assignments\n\tassignments := map[string]store.AssignedState{\n\t\t\"exec-1\": {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"0\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t},\n\t}\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"0\").Return(nil, nil)\n\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, \"1\").Return(nil, nil)\n\tmocks.store.EXPECT().GetState(gomock.Any(), mocks.cfg.Name).Return(&store.NamespaceState{\n\t\tExecutors:        heartbeats,\n\t\tShardAssignments: assignments,\n\t}, nil)\n\tmocks.election.EXPECT().Guard().Return(store.NopGuard())\n\tmocks.store.EXPECT().AssignShards(gomock.Any(), mocks.cfg.Name, gomock.Any(), gomock.Any()).DoAndReturn(\n\t\tfunc(_ context.Context, _ string, request store.AssignShardsRequest, _ store.GuardFunc) error {\n\t\t\tassert.Len(t, request.NewState.ShardAssignments[\"exec-1\"].AssignedShards, 2, \"Both shards should now be assigned to exec-1\")\n\t\t\treturn nil\n\t\t},\n\t)\n\n\terr := processor.rebalanceShards(context.Background())\n\trequire.NoError(t, err)\n}\n\nfunc TestGetShards_Utility(t *testing.T) {\n\tt.Run(\"Fixed type\", func(t *testing.T) {\n\t\tcfg := config.Namespace{Type: config.NamespaceTypeFixed, ShardNum: 5}\n\t\tshards := getShards(cfg, nil, nil)\n\t\tassert.Equal(t, []string{\"0\", \"1\", \"2\", \"3\", \"4\"}, shards)\n\t})\n\n\tt.Run(\"Ephemeral type\", func(t *testing.T) {\n\t\tcfg := config.Namespace{Type: config.NamespaceTypeEphemeral}\n\t\tnsState := &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\"executor1\": {\n\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\"s0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"executor2\": {\n\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\"s3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s4\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tshards := getShards(cfg, nsState, nil)\n\t\tslices.Sort(shards)\n\t\tassert.Equal(t, []string{\"s0\", \"s1\", \"s2\", \"s3\", \"s4\"}, shards)\n\t})\n\n\tt.Run(\"Ephemeral type with deleted shards\", func(t *testing.T) {\n\t\tcfg := config.Namespace{Type: config.NamespaceTypeEphemeral}\n\t\tnsState := &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\t\"executor1\": {\n\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\"s0\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s2\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t\"executor2\": {\n\t\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\t\"s3\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t\t\"s4\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tdeletedShards := map[string]store.ShardState{\n\t\t\t\"s0\": {},\n\t\t\t\"s1\": {},\n\t\t}\n\t\tshards := getShards(cfg, nsState, deletedShards)\n\t\tslices.Sort(shards)\n\t\tassert.Equal(t, []string{\"s2\", \"s3\", \"s4\"}, shards)\n\t})\n\n\t// Unknown type\n\tt.Run(\"Other type\", func(t *testing.T) {\n\t\tcfg := config.Namespace{Type: \"other\"}\n\t\tshards := getShards(cfg, nil, nil)\n\t\tassert.Nil(t, shards)\n\t})\n}\n\nfunc TestAssignShardsToEmptyExecutors(t *testing.T) {\n\tcases := []struct {\n\t\tname                       string\n\t\tinputAssignments           map[string][]string\n\t\texpectedAssignments        map[string][]string\n\t\texpectedDistributonChanged bool\n\t}{\n\t\t{\n\t\t\tname:                       \"no executors\",\n\t\t\tinputAssignments:           map[string][]string{},\n\t\t\texpectedAssignments:        map[string][]string{},\n\t\t\texpectedDistributonChanged: false,\n\t\t},\n\t\t{\n\t\t\tname: \"no empty executors\",\n\t\t\tinputAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-3\", \"shard-4\", \"shard-5\", \"shard-6\"},\n\t\t\t\t\"exec-2\": {\"shard-7\", \"shard-8\"},\n\t\t\t},\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-3\", \"shard-4\", \"shard-5\", \"shard-6\"},\n\t\t\t\t\"exec-2\": {\"shard-7\", \"shard-8\"},\n\t\t\t},\n\t\t\texpectedDistributonChanged: false,\n\t\t},\n\t\t{\n\t\t\tname: \"empty executor\",\n\t\t\tinputAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-3\", \"shard-4\", \"shard-5\", \"shard-6\"},\n\t\t\t\t\"exec-2\": {\"shard-7\", \"shard-8\", \"shard-9\", \"shard-10\"},\n\t\t\t\t\"exec-3\": {},\n\t\t\t},\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-2\", \"shard-3\", \"shard-4\", \"shard-5\", \"shard-6\"},\n\t\t\t\t\"exec-2\": {\"shard-8\", \"shard-9\", \"shard-10\"},\n\t\t\t\t\"exec-3\": {\"shard-1\", \"shard-7\"},\n\t\t\t},\n\t\t\texpectedDistributonChanged: true,\n\t\t},\n\t\t{\n\t\t\tname:                       \"all empty executors\",\n\t\t\tinputAssignments:           map[string][]string{\"exec-1\": {}, \"exec-2\": {}, \"exec-3\": {}},\n\t\t\texpectedAssignments:        map[string][]string{\"exec-1\": {}, \"exec-2\": {}, \"exec-3\": {}},\n\t\t\texpectedDistributonChanged: false,\n\t\t},\n\t\t{\n\t\t\tname: \"multiple empty executors\",\n\t\t\tinputAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-3\", \"shard-4\", \"shard-5\", \"shard-6\", \"shard-7\", \"shard-8\", \"shard-9\", \"shard-10\"},\n\t\t\t\t\"exec-2\": {\"shard-11\", \"shard-12\", \"shard-13\", \"shard-14\", \"shard-15\", \"shard-16\", \"shard-17\"},\n\t\t\t\t\"exec-3\": {\"shard-18\", \"shard-19\", \"shard-20\", \"shard-21\", \"shard-22\", \"shard-23\", \"shard-24\"},\n\t\t\t\t\"exec-4\": {},\n\t\t\t\t\"exec-5\": {},\n\t\t\t},\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-4\", \"shard-5\", \"shard-6\", \"shard-7\", \"shard-8\", \"shard-9\", \"shard-10\"},\n\t\t\t\t\"exec-2\": {\"shard-14\", \"shard-15\", \"shard-16\", \"shard-17\"},\n\t\t\t\t\"exec-3\": {\"shard-20\", \"shard-21\", \"shard-22\", \"shard-23\", \"shard-24\"},\n\t\t\t\t\"exec-4\": {\"shard-1\", \"shard-18\", \"shard-12\", \"shard-3\"},\n\t\t\t\t\"exec-5\": {\"shard-11\", \"shard-2\", \"shard-19\", \"shard-13\"},\n\t\t\t},\n\t\t\texpectedDistributonChanged: true,\n\t\t},\n\t}\n\n\tfor _, c := range cases {\n\t\tt.Run(c.name, func(t *testing.T) {\n\t\t\tactualDistributionChanged := assignShardsToEmptyExecutors(c.inputAssignments)\n\n\t\t\tassert.Equal(t, c.expectedAssignments, c.inputAssignments)\n\t\t\tassert.Equal(t, c.expectedDistributonChanged, actualDistributionChanged)\n\t\t})\n\t}\n}\n\nfunc TestNewHandoverStats(t *testing.T) {\n\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\tdefer mocks.ctrl.Finish()\n\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\tnow := time.Now().UTC()\n\tshardID := \"shard-1\"\n\tnewExecutorID := \"exec-new\"\n\n\ttype testCase struct {\n\t\tname             string\n\t\tgetOwner         *store.ShardOwner\n\t\tgetOwnerErr      error\n\t\texecutors        map[string]store.HeartbeatState\n\t\texpectShardStats *store.ShardHandoverStats // nil means expect nil result\n\t}\n\n\ttestCases := []testCase{\n\t\t{\n\t\t\tname:             \"error other than shard not found -> stat without handover\",\n\t\t\tgetOwner:         nil,\n\t\t\tgetOwnerErr:      errors.New(\"random error\"),\n\t\t\texecutors:        map[string]store.HeartbeatState{},\n\t\t\texpectShardStats: nil,\n\t\t},\n\t\t{\n\t\t\tname:             \"ErrShardNotFound -> stat without handover\",\n\t\t\tgetOwner:         nil,\n\t\t\tgetOwnerErr:      store.ErrShardNotFound,\n\t\t\texecutors:        map[string]store.HeartbeatState{},\n\t\t\texpectShardStats: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"same executor as previous -> nil\",\n\t\t\tgetOwner:    &store.ShardOwner{ExecutorID: newExecutorID},\n\t\t\tgetOwnerErr: nil,\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\tnewExecutorID: {\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second)},\n\t\t\t},\n\t\t\texpectShardStats: nil,\n\t\t},\n\t\t{\n\t\t\tname:             \"prev executor different but heartbeat missing -> no handover\",\n\t\t\tgetOwner:         &store.ShardOwner{ExecutorID: \"old-exec\"},\n\t\t\tgetOwnerErr:      nil,\n\t\t\texecutors:        map[string]store.HeartbeatState{},\n\t\t\texpectShardStats: nil,\n\t\t},\n\t\t{\n\t\t\tname:        \"prev executor ACTIVE -> emergency handover\",\n\t\t\tgetOwner:    &store.ShardOwner{ExecutorID: \"old-active\"},\n\t\t\tgetOwnerErr: nil,\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"old-active\": {\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectShardStats: &store.ShardHandoverStats{\n\t\t\t\tHandoverType:                      types.HandoverTypeEMERGENCY,\n\t\t\t\tPreviousExecutorLastHeartbeatTime: now.Add(-10 * time.Second),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"prev executor DRAINING -> graceful handover\",\n\t\t\tgetOwner:    &store.ShardOwner{ExecutorID: \"old-draining\"},\n\t\t\tgetOwnerErr: nil,\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"old-draining\": {\n\t\t\t\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectShardStats: &store.ShardHandoverStats{\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\tPreviousExecutorLastHeartbeatTime: now.Add(-10 * time.Second),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"prev executor DRAINED -> graceful handover\",\n\t\t\tgetOwner:    &store.ShardOwner{ExecutorID: \"old-drained\"},\n\t\t\tgetOwnerErr: nil,\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"old-drained\": {\n\t\t\t\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectShardStats: &store.ShardHandoverStats{\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\tPreviousExecutorLastHeartbeatTime: now.Add(-10 * time.Second),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, shardID).Return(tc.getOwner, tc.getOwnerErr)\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tstat := processor.newHandoverStats(&store.NamespaceState{Executors: tc.executors}, shardID, newExecutorID)\n\t\t\tif tc.expectShardStats == nil {\n\t\t\t\trequire.Nil(t, stat)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, stat)\n\t\t\trequire.Equal(t, tc.expectShardStats, stat)\n\t\t})\n\t}\n}\nfunc TestAddHandoverStatsToExecutorAssignedState(t *testing.T) {\n\n\tnow := time.Now().UTC()\n\texecutorID := \"exec-1\"\n\tshardIDs := []string{\"shard-1\", \"shard-2\"}\n\n\tfor name, tc := range map[string]struct {\n\t\tname      string\n\t\texecutors map[string]store.HeartbeatState\n\n\t\tgetOwners    map[string]*store.ShardOwner\n\t\tgetOwnerErrs map[string]error\n\n\t\texpected map[string]store.ShardHandoverStats\n\t}{\n\t\t\"no previous owner for both shards\": {\n\t\t\tgetOwners:    map[string]*store.ShardOwner{\"shard-1\": nil, \"shard-2\": nil},\n\t\t\tgetOwnerErrs: map[string]error{\"shard-1\": store.ErrShardNotFound, \"shard-2\": store.ErrShardNotFound},\n\t\t\texecutors:    map[string]store.HeartbeatState{},\n\t\t\texpected:     map[string]store.ShardHandoverStats{},\n\t\t},\n\t\t\"emergency handover for shard-1, no handover for shard-2\": {\n\t\t\tgetOwners: map[string]*store.ShardOwner{\n\t\t\t\t\"shard-1\": {ExecutorID: \"old-active\"},\n\t\t\t\t\"shard-2\": nil,\n\t\t\t},\n\t\t\tgetOwnerErrs: map[string]error{\n\t\t\t\t\"shard-1\": nil,\n\t\t\t\t\"shard-2\": store.ErrShardNotFound,\n\t\t\t},\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"old-active\": {\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]store.ShardHandoverStats{\n\t\t\t\t\"shard-1\": {\n\t\t\t\t\tHandoverType:                      types.HandoverTypeEMERGENCY,\n\t\t\t\t\tPreviousExecutorLastHeartbeatTime: now.Add(-10 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"graceful handover for shard-1\": {\n\t\t\tgetOwners: map[string]*store.ShardOwner{\n\t\t\t\t\"shard-1\": {ExecutorID: \"old-draining\"},\n\t\t\t\t\"shard-2\": nil,\n\t\t\t},\n\t\t\tgetOwnerErrs: map[string]error{\n\t\t\t\t\"shard-1\": nil,\n\t\t\t},\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"old-draining\": {\n\t\t\t\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\t\t\t\tLastHeartbeat: now.Add(-20 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]store.ShardHandoverStats{\n\t\t\t\t\"shard-1\": {\n\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\tPreviousExecutorLastHeartbeatTime: now.Add(-20 * time.Second),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"same executor as previous, no handover\": {\n\t\t\tgetOwners: map[string]*store.ShardOwner{\n\t\t\t\t\"shard-1\": {ExecutorID: executorID},\n\t\t\t\t\"shard-2\": nil,\n\t\t\t},\n\t\t\tgetOwnerErrs: map[string]error{\n\t\t\t\t\"shard-1\": nil,\n\t\t\t},\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\texecutorID: {\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]store.ShardHandoverStats{},\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\t\tdefer mocks.ctrl.Finish()\n\t\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\t\tfor _, shardID := range shardIDs {\n\t\t\t\tmocks.store.EXPECT().GetShardOwner(gomock.Any(), mocks.cfg.Name, shardID).Return(tc.getOwners[shardID], tc.getOwnerErrs[shardID]).AnyTimes()\n\t\t\t}\n\t\t\tnamespaceState := &store.NamespaceState{\n\t\t\t\tExecutors: tc.executors,\n\t\t\t}\n\t\t\tstats := processor.addHandoverStatsToExecutorAssignedState(namespaceState, executorID, shardIDs)\n\t\t\tassert.Equal(t, tc.expected, stats)\n\t\t})\n\t}\n}\n\nfunc TestRebalanceByShardLoad(t *testing.T) {\n\tcases := []struct {\n\t\tname                       string\n\t\tshardLoad                  map[string]float64\n\t\tcurrentAssignments         map[string][]string\n\t\tmaxDeviation               float64\n\t\texpectedDistributionChange bool\n\t\texpectedAssignments        map[string][]string\n\t}{\n\t\t{\n\t\t\tname:                       \"single executor - no rebalance\",\n\t\t\tshardLoad:                  map[string]float64{\"shard-1\": 10.0},\n\t\t\tcurrentAssignments:         map[string][]string{\"exec-1\": {\"shard-1\"}},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments:        map[string][]string{\"exec-1\": {\"shard-1\"}},\n\t\t},\n\t\t{\n\t\t\tname: \"balanced load - no rebalance needed\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 10.0,\n\t\t\t\t\"shard-2\": 10.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 10.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 10.0\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"deviation below threshold - no rebalance\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 10.0,\n\t\t\t\t\"shard-2\": 15.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 15.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 15.0\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple shards - hottest moved\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 5.0,\n\t\t\t\t\"shard-2\": 30.0,\n\t\t\t\t\"shard-3\": 20.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},            // 5.0\n\t\t\t\t\"exec-2\": {\"shard-2\", \"shard-3\"}, // 50.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: true,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\"}, // 35.0\n\t\t\t\t\"exec-2\": {\"shard-3\"},            // 20.0\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"coldest would become hottest - no rebalance\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 10.0,\n\t\t\t\t\"shard-2\": 100.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 100.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple shards per executor\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 5.0, \"shard-2\": 5.0,\n\t\t\t\t\"shard-3\": 40.0, \"shard-4\": 30.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-3\", \"shard-4\"}, // 70.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: true,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-3\"}, // 50.0\n\t\t\t\t\"exec-2\": {\"shard-4\"},                       // 30.0\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"zero load shards - no rebalance\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 0.0,\n\t\t\t\t\"shard-2\": 50.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 0.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 50.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"new shard load - equal shards - no rebalance\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-2\": 0.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"}, // 0.0\n\t\t\t\t\"exec-2\": {\"shard-2\"}, // 50.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"four executors - balanced load\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 10.0,\n\t\t\t\t\"shard-2\": 10.0,\n\t\t\t\t\"shard-3\": 10.0,\n\t\t\t\t\"shard-4\": 10.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t\t\"exec-3\": {\"shard-3\"},\n\t\t\t\t\"exec-4\": {\"shard-4\"},\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t\t\"exec-3\": {\"shard-3\"},\n\t\t\t\t\"exec-4\": {\"shard-4\"},\n\t\t\t},\n\t\t}, {\n\t\t\tname: \"four executors - one overloaded - no rebalance\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 10.0,\n\t\t\t\t\"shard-2\": 10.0,\n\t\t\t\t\"shard-3\": 10.0,\n\t\t\t\t\"shard-4\": 50.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t\t\"exec-3\": {\"shard-3\"},\n\t\t\t\t\"exec-4\": {\"shard-4\"},\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: false,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\"},\n\t\t\t\t\"exec-2\": {\"shard-2\"},\n\t\t\t\t\"exec-3\": {\"shard-3\"},\n\t\t\t\t\"exec-4\": {\"shard-4\"},\n\t\t\t},\n\t\t}, {\n\t\t\tname: \"four executors - uneven distribution - stale executor\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 15.0,\n\t\t\t\t\"shard-2\": 15.0,\n\t\t\t\t\"shard-3\": 15.0,\n\t\t\t\t\"shard-4\": 15.0,\n\t\t\t\t\"shard-5\": 40.0,\n\t\t\t\t\"shard-6\": 40.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\", \"shard-5\"}, // 70.0\n\t\t\t\t\"exec-2\": {\"shard-3\", \"shard-4\"},            // 30.0\n\t\t\t\t\"exec-3\": {},                                // 0.0\n\t\t\t\t\"exec-4\": {\"shard-6\"},                       // 40.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: true,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\"}, // 30.0\n\t\t\t\t\"exec-2\": {\"shard-3\", \"shard-4\"}, // 30.0\n\t\t\t\t\"exec-3\": {\"shard-5\"},            // 40.0\n\t\t\t\t\"exec-4\": {\"shard-6\"},            // 40.0\n\t\t\t},\n\t\t}, {\n\t\t\tname: \"four executors - mixed load with multiple shards\",\n\t\t\tshardLoad: map[string]float64{\n\t\t\t\t\"shard-1\": 5.0, \"shard-2\": 5.0,\n\t\t\t\t\"shard-3\": 5.0, \"shard-4\": 2.0,\n\t\t\t\t\"shard-5\": 25.0, \"shard-6\": 25.0,\n\t\t\t\t\"shard-7\": 15.0, \"shard-8\": 15.0,\n\t\t\t},\n\t\t\tcurrentAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\"}, // 10.0\n\t\t\t\t\"exec-2\": {\"shard-3\", \"shard-4\"}, // 7.0\n\t\t\t\t\"exec-3\": {\"shard-5\", \"shard-6\"}, // 50.0\n\t\t\t\t\"exec-4\": {\"shard-7\", \"shard-8\"}, // 30.0\n\t\t\t},\n\t\t\tmaxDeviation:               2.0,\n\t\t\texpectedDistributionChange: true,\n\t\t\texpectedAssignments: map[string][]string{\n\t\t\t\t\"exec-1\": {\"shard-1\", \"shard-2\"},            // 10.0\n\t\t\t\t\"exec-2\": {\"shard-3\", \"shard-4\", \"shard-6\"}, // 32.0\n\t\t\t\t\"exec-3\": {\"shard-5\"},                       // 25.0\n\t\t\t\t\"exec-4\": {\"shard-7\", \"shard-8\"},            // 30.0\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\t\tmocks.cfg.Name = tc.name\n\t\t\tmocks.sdConfig.LoadBalancingNaive.MaxDeviation = func(namespace string) float64 {\n\t\t\t\treturn tc.maxDeviation\n\t\t\t}\n\t\t\tdefer mocks.ctrl.Finish()\n\t\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\t\tdistributionChanged := processor.rebalanceByShardLoad(tc.shardLoad, tc.currentAssignments)\n\n\t\t\tassert.Equal(t, tc.expectedDistributionChange, distributionChanged, \"distribution change mismatch\")\n\t\t\tassert.Equal(t, tc.expectedAssignments, tc.currentAssignments, \"final assignments mismatch\")\n\t\t})\n\t}\n}\n\nfunc TestEmitExecutorMetric(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\texecutors      map[string]store.HeartbeatState\n\t\texpectedCounts map[types.ExecutorStatus]int\n\t}{\n\t\t{\n\t\t\tname:           \"empty executors\",\n\t\t\texecutors:      map[string]store.HeartbeatState{},\n\t\t\texpectedCounts: map[types.ExecutorStatus]int{},\n\t\t},\n\t\t{\n\t\t\tname: \"single active executor\",\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t},\n\t\t\texpectedCounts: map[types.ExecutorStatus]int{\n\t\t\t\ttypes.ExecutorStatusACTIVE: 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple executors\",\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\"exec-3\": {Status: types.ExecutorStatusDRAINING},\n\t\t\t\t\"exec-4\": {Status: types.ExecutorStatusDRAINED},\n\t\t\t},\n\t\t\texpectedCounts: map[types.ExecutorStatus]int{\n\t\t\t\ttypes.ExecutorStatusACTIVE:   2,\n\t\t\t\ttypes.ExecutorStatusDRAINING: 1,\n\t\t\t\ttypes.ExecutorStatusDRAINED:  1,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\t\tdefer mocks.ctrl.Finish()\n\t\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\t\tnamespaceState := &store.NamespaceState{\n\t\t\t\tExecutors: tt.executors,\n\t\t\t}\n\n\t\t\tmetricsScope := &metricmocks.Scope{}\n\n\t\t\tfor status, count := range tt.expectedCounts {\n\t\t\t\ttaggedScope := &metricmocks.Scope{}\n\t\t\t\tmetricsScope.On(\"Tagged\", metrics.ExecutorStatusTag(status.String())).Return(taggedScope).Once()\n\t\t\t\ttaggedScope.On(\"UpdateGauge\", metrics.ShardDistributorTotalExecutors, float64(count)).Once()\n\t\t\t}\n\n\t\t\tprocessor.emitExecutorMetric(namespaceState, metricsScope)\n\n\t\t\tmetricsScope.AssertExpectations(t)\n\t\t})\n\t}\n}\n\nfunc TestEmitOldestExecutorHeartbeatLag(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\texecutors   map[string]store.HeartbeatState\n\t\texpectedLag *float64\n\t}{\n\t\t{\n\t\t\tname:        \"empty executors\",\n\t\t\texecutors:   map[string]store.HeartbeatState{},\n\t\t\texpectedLag: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"single executor\",\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t},\n\t\t\texpectedLag: common.Float64Ptr(5000),\n\t\t},\n\t\t{\n\t\t\tname: \"multiple executors\",\n\t\t\texecutors: map[string]store.HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},   // 5 seconds\n\t\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE},   // 10 seconds (oldest)\n\t\t\t\t\"exec-3\": {Status: types.ExecutorStatusDRAINING}, // 3 seconds\n\t\t\t},\n\t\t\texpectedLag: common.Float64Ptr(10000), // 10 seconds\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\t\tdefer mocks.ctrl.Finish()\n\t\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\t\tnow := mocks.timeSource.Now()\n\n\t\t\tif tt.name == \"single executor\" {\n\t\t\t\ttt.executors[\"exec-1\"] = store.HeartbeatState{\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-5 * time.Second),\n\t\t\t\t}\n\t\t\t} else if tt.name == \"multiple executors\" {\n\t\t\t\ttt.executors[\"exec-1\"] = store.HeartbeatState{\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-5 * time.Second),\n\t\t\t\t}\n\t\t\t\ttt.executors[\"exec-2\"] = store.HeartbeatState{\n\t\t\t\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\t\t\t\tLastHeartbeat: now.Add(-10 * time.Second), // oldest\n\t\t\t\t}\n\t\t\t\ttt.executors[\"exec-3\"] = store.HeartbeatState{\n\t\t\t\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\t\t\t\tLastHeartbeat: now.Add(-3 * time.Second),\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tnamespaceState := &store.NamespaceState{\n\t\t\t\tExecutors: tt.executors,\n\t\t\t}\n\n\t\t\tmetricsScope := &metricmocks.Scope{}\n\n\t\t\tif tt.expectedLag != nil {\n\t\t\t\tmetricsScope.On(\"UpdateGauge\", metrics.ShardDistributorOldestExecutorHeartbeatLag, *tt.expectedLag).Once()\n\t\t\t}\n\n\t\t\tprocessor.emitOldestExecutorHeartbeatLag(namespaceState, metricsScope)\n\n\t\t\tmetricsScope.AssertExpectations(t)\n\t\t})\n\t}\n}\n\nfunc TestRunRebalanceTriggeringLoop(t *testing.T) {\n\tt.Run(\"no events from subscribe, trigger from ticker\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tdefer cancel()\n\n\t\tupdateChan := make(chan int64)\n\t\ttriggerChan := make(chan string, 1)\n\n\t\tgo processor.rebalanceTriggeringLoop(ctx, updateChan, triggerChan)\n\n\t\t// Wait for ticker to be created\n\t\tmocks.timeSource.BlockUntil(1)\n\n\t\t// Advance time to trigger the ticker\n\t\tmocks.timeSource.Advance(processor.cfg.Period)\n\n\t\t// Expect trigger from periodic reconciliation\n\t\tselect {\n\t\tcase reason := <-triggerChan:\n\t\t\tassert.Equal(t, \"Periodic reconciliation triggered\", reason)\n\t\tcase <-time.After(time.Second):\n\t\t\tt.Fatal(\"expected trigger from ticker, but timed out\")\n\t\t}\n\n\t\tcancel()\n\t})\n\n\tt.Run(\"events from subscribe before period, trigger from state change\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tdefer cancel()\n\n\t\tupdateChan := make(chan int64, 1)\n\t\ttriggerChan := make(chan string, 1)\n\n\t\tgo processor.rebalanceTriggeringLoop(ctx, updateChan, triggerChan)\n\n\t\t// Wait for ticker to be created\n\t\tmocks.timeSource.BlockUntil(1)\n\n\t\t// Send a state change event before the ticker fires\n\t\tupdateChan <- 1\n\n\t\t// Expect trigger from state change\n\t\tselect {\n\t\tcase reason := <-triggerChan:\n\t\t\tassert.Equal(t, \"State change detected\", reason)\n\t\tcase <-time.After(time.Second):\n\t\t\tt.Fatal(\"expected trigger from state change, but timed out\")\n\t\t}\n\n\t\tcancel()\n\t})\n\n\tt.Run(\"triggerChan full, multiple subscribe events, loop not stuck\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tctx, cancel := context.WithCancel(context.Background())\n\t\tdefer cancel()\n\n\t\t// Use unbuffered channel for updates to ensure they are processed one at a time\n\t\tupdateChan := make(chan int64)\n\t\ttriggerChan := make(chan string, 1)\n\n\t\tgo processor.rebalanceTriggeringLoop(ctx, updateChan, triggerChan)\n\n\t\t// Wait for ticker to be created\n\t\tmocks.timeSource.BlockUntil(1)\n\n\t\t// Don't read from triggerChan yet to keep it full\n\t\t// Send multiple state change events\n\t\tfor i := int64(0); i <= 10; i++ {\n\t\t\tselect {\n\t\t\tcase updateChan <- i:\n\t\t\tcase <-time.After(time.Second):\n\t\t\t\t// Expect that the loop is not stuck\n\t\t\t\tt.Fatalf(\"failed to send update %d, channel blocked\", i)\n\t\t\t}\n\t\t}\n\n\t\t// Expect trigger from state change\n\t\tselect {\n\t\tcase reason := <-triggerChan:\n\t\t\tassert.Equal(t, \"State change detected\", reason)\n\t\tcase <-time.After(time.Second):\n\t\t\tt.Fatal(\"expected trigger from state change, but timed out\")\n\t\t}\n\n\t\tcancel()\n\t})\n\n\tt.Run(\"update channel closed stops loop\", func(t *testing.T) {\n\t\tmocks := setupProcessorTest(t, config.NamespaceTypeFixed)\n\t\tdefer mocks.ctrl.Finish()\n\t\tprocessor := mocks.factory.CreateProcessor(mocks.cfg, mocks.store, mocks.election).(*namespaceProcessor)\n\n\t\tctx := context.Background()\n\n\t\tupdateChan := make(chan int64)\n\t\ttriggerChan := make(chan string, 1)\n\n\t\tvar wg sync.WaitGroup\n\t\twg.Add(1)\n\t\tgo func() {\n\t\t\tdefer wg.Done()\n\t\t\tprocessor.rebalanceTriggeringLoop(ctx, updateChan, triggerChan)\n\t\t}()\n\n\t\t// Wait for ticker to be created\n\t\tmocks.timeSource.BlockUntil(1)\n\n\t\t// Close update channel\n\t\tclose(updateChan)\n\n\t\t// Wait for loop to exit\n\t\tdone := make(chan struct{})\n\t\tgo func() {\n\t\t\twg.Wait()\n\t\t\tclose(done)\n\t\t}()\n\n\t\tselect {\n\t\tcase <-done:\n\t\t\t// Loop exited as expected\n\t\tcase <-time.After(time.Second):\n\t\t\tt.Fatal(\"loop did not exit after updateChan closed\")\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/sharddistributorfx/fx_test.go",
    "content": "package sharddistributorfx\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/fx\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nfunc TestFxServiceStartStop(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\ttestDispatcher := yarpc.NewDispatcher(yarpc.Config{Name: \"test\"})\n\tctrl := gomock.NewController(t)\n\tapp := fxtest.New(t,\n\t\ttestlogger.Module(t),\n\t\tfx.Provide(\n\t\t\tfunc() metrics.Client { return metrics.NewNoopMetricsClient() },\n\t\t\tfunc() rpc.Factory {\n\t\t\t\tfactory := rpc.NewMockFactory(ctrl)\n\t\t\t\tfactory.EXPECT().GetDispatcher().Return(testDispatcher)\n\t\t\t\treturn factory\n\t\t\t},\n\t\t\tfunc() *dynamicconfig.Collection {\n\t\t\t\treturn dynamicconfig.NewNopCollection()\n\t\t\t},\n\t\t\tfx.Annotated{Target: func() string { return \"testHost\" }, Name: \"hostname\"},\n\t\t\tfunc() store.Elector {\n\t\t\t\treturn store.NewMockElector(ctrl)\n\t\t\t},\n\t\t\tfunc() store.Store {\n\t\t\t\treturn store.NewMockStore(ctrl)\n\t\t\t},\n\t\t\tfunc() config.ShardDistribution {\n\t\t\t\treturn config.ShardDistribution{}\n\t\t\t},\n\t\t\tfunc() clock.TimeSource {\n\t\t\t\treturn clock.NewMockedTimeSource()\n\t\t\t},\n\t\t),\n\t\tModule)\n\tapp.RequireStart().RequireStop()\n\t// API should be registered inside dispatcher.\n\tassert.True(t, len(testDispatcher.Introspect().Procedures) > 3)\n}\n"
  },
  {
    "path": "service/sharddistributor/sharddistributorfx/sharddistributorfx.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sharddistributorfx\n\nimport (\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/rpc\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/election\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/namespace\"\n\t\"github.com/uber/cadence/service/sharddistributor/leader/process\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\tmeteredStore \"github.com/uber/cadence/service/sharddistributor/store/wrappers/metered\"\n\t\"github.com/uber/cadence/service/sharddistributor/wrappers/grpc\"\n\t\"github.com/uber/cadence/service/sharddistributor/wrappers/metered\"\n)\n\nvar Module = fx.Module(\"sharddistributor\",\n\tnamespace.Module,\n\telection.Module,\n\tprocess.Module,\n\tfx.Provide(config.NewConfig),\n\tfx.Decorate(func(s store.Store, metricsClient metrics.Client, logger log.Logger, timeSource clock.TimeSource) store.Store {\n\t\treturn meteredStore.NewStore(s, metricsClient, logger, timeSource)\n\t}),\n\tfx.Invoke(registerHandlers))\n\ntype registerHandlersParams struct {\n\tfx.In\n\n\tShardDistributionCfg config.ShardDistribution\n\n\tLogger        log.Logger\n\tMetricsClient metrics.Client\n\tRPCFactory    rpc.Factory\n\tConfig        *config.Config\n\n\tTimeSource clock.TimeSource\n\tStore      store.Store\n\n\tLifecycle fx.Lifecycle\n}\n\nfunc registerHandlers(params registerHandlersParams) error {\n\tdispatcher := params.RPCFactory.GetDispatcher()\n\n\trawHandler := handler.NewHandler(params.Logger, params.TimeSource, params.ShardDistributionCfg, params.Store)\n\twrappedHandler := metered.NewMetricsHandler(rawHandler, params.Logger, params.MetricsClient)\n\n\texecutorHandler := handler.NewExecutorHandler(params.Logger, params.Store, params.TimeSource, params.ShardDistributionCfg, params.Config, params.MetricsClient)\n\twrappedExecutor := metered.NewExecutorMetricsExecutor(executorHandler, params.Logger, params.MetricsClient)\n\n\tgrpcHandler := grpc.NewGRPCHandler(wrappedHandler)\n\tgrpcHandler.Register(dispatcher)\n\n\texecutorGRPCHander := grpc.NewExecutorGRPCExecutor(wrappedExecutor)\n\texecutorGRPCHander.Register(dispatcher)\n\n\tparams.Lifecycle.Append(fx.StartStopHook(rawHandler.Start, rawHandler.Stop))\n\n\treturn nil\n}\n"
  },
  {
    "path": "service/sharddistributor/sharddistributorspectatorclient/factory/meteredClientDecorater.go",
    "content": "package factory\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/client/spectatorclient/metricsconstants\"\n)\n\n// TODO: consider using gowrap to generate this code\ntype meteredShardDistributorClient struct {\n\tclient       sharddistributor.Client\n\tmetricsScope tally.Scope\n}\n\n// NewMeteredShardDistributorClient creates a new instance of metered shard distributor client\nfunc NewMeteredShardDistributorClient(client sharddistributor.Client, metricsScope tally.Scope) sharddistributor.Client {\n\treturn &meteredShardDistributorClient{\n\t\tclient:       client,\n\t\tmetricsScope: metricsScope,\n\t}\n}\n\nfunc (c *meteredShardDistributorClient) GetShardOwner(ctx context.Context, request *types.GetShardOwnerRequest, opts ...yarpc.CallOption) (*types.GetShardOwnerResponse, error) {\n\tscope := c.metricsScope.Tagged(map[string]string{\n\t\tmetrics.OperationTagName: metricsconstants.ShardDistributorSpectatorGetShardOwnerOperationTagName,\n\t})\n\n\tscope.Counter(metricsconstants.ShardDistributorSpectatorClientRequests).Inc(1)\n\n\tsw := scope.Timer(metricsconstants.ShardDistributorSpectatorClientLatency).Start()\n\tresponse, err := c.client.GetShardOwner(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.Counter(metricsconstants.ShardDistributorSpectatorClientFailures).Inc(1)\n\t}\n\treturn response, err\n}\n\nfunc (c *meteredShardDistributorClient) WatchNamespaceState(ctx context.Context, request *types.WatchNamespaceStateRequest, opts ...yarpc.CallOption) (sharddistributor.WatchNamespaceStateClient, error) {\n\tscope := c.metricsScope.Tagged(map[string]string{\n\t\tmetrics.OperationTagName: metricsconstants.ShardDistributorSpectatorWatchNamespaceStateOperationTagName,\n\t})\n\n\tscope.Counter(metricsconstants.ShardDistributorSpectatorClientRequests).Inc(1)\n\n\tsw := scope.Timer(metricsconstants.ShardDistributorSpectatorClientLatency).Start()\n\tstream, err := c.client.WatchNamespaceState(ctx, request, opts...)\n\tsw.Stop()\n\n\tif err != nil {\n\t\tscope.Counter(metricsconstants.ShardDistributorSpectatorClientFailures).Inc(1)\n\t}\n\treturn stream, err\n}\n"
  },
  {
    "path": "service/sharddistributor/sharddistributorspectatorclient/factory/sharddistributorspectatorclient.go",
    "content": "package factory\n\nimport (\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/fx\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/client/sharddistributor\"\n\t\"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/client/wrappers/retryable\"\n\t\"github.com/uber/cadence/client/wrappers/timeout\"\n\t\"github.com/uber/cadence/common\"\n)\n\ntype Params struct {\n\tfx.In\n\n\tYarpcClient  sharddistributorv1.ShardDistributorAPIYARPCClient\n\tMetricsScope tally.Scope\n}\n\nfunc NewShardDistributorSpectatorClient(params Params) (sharddistributor.Client, error) {\n\t// Wrap the YARPC client with GRPC wrapper\n\tclient := grpc.NewShardDistributorClient(params.YarpcClient)\n\n\t// Add timeout wrapper\n\tclient = timeout.NewShardDistributorClient(client, timeout.ShardDistributorDefaultTimeout)\n\n\t// Add metered wrapper\n\tif params.MetricsScope != nil {\n\t\tclient = NewMeteredShardDistributorClient(client, params.MetricsScope)\n\t}\n\n\t// Add retry wrapper\n\tclient = retryable.NewShardDistributorClient(\n\t\tclient,\n\t\tcommon.CreateShardDistributorServiceRetryPolicy(),\n\t\tcommon.IsServiceTransientError,\n\t)\n\n\treturn client, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdclient/client.go",
    "content": "package etcdclient\n\nimport clientv3 \"go.etcd.io/etcd/client/v3\"\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=client_mock.go Client\n\n// Client is an interface that groups the etcd clientv3 interfaces\ntype Client interface {\n\tclientv3.Cluster\n\tclientv3.KV\n\tclientv3.Lease\n\tclientv3.Watcher\n\tclientv3.Auth\n\tclientv3.Maintenance\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdclient/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: client.go\n//\n// Generated by this command:\n//\n//\tmockgen -package etcdclient -source client.go -destination=client_mock.go Client\n//\n\n// Package etcdclient is a generated GoMock package.\npackage etcdclient\n\nimport (\n\tcontext \"context\"\n\tio \"io\"\n\treflect \"reflect\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// AlarmDisarm mocks base method.\nfunc (m_2 *MockClient) AlarmDisarm(ctx context.Context, m *clientv3.AlarmMember) (*clientv3.AlarmResponse, error) {\n\tm_2.ctrl.T.Helper()\n\tret := m_2.ctrl.Call(m_2, \"AlarmDisarm\", ctx, m)\n\tret0, _ := ret[0].(*clientv3.AlarmResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AlarmDisarm indicates an expected call of AlarmDisarm.\nfunc (mr *MockClientMockRecorder) AlarmDisarm(ctx, m any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AlarmDisarm\", reflect.TypeOf((*MockClient)(nil).AlarmDisarm), ctx, m)\n}\n\n// AlarmList mocks base method.\nfunc (m *MockClient) AlarmList(ctx context.Context) (*clientv3.AlarmResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AlarmList\", ctx)\n\tret0, _ := ret[0].(*clientv3.AlarmResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AlarmList indicates an expected call of AlarmList.\nfunc (mr *MockClientMockRecorder) AlarmList(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AlarmList\", reflect.TypeOf((*MockClient)(nil).AlarmList), ctx)\n}\n\n// AuthDisable mocks base method.\nfunc (m *MockClient) AuthDisable(ctx context.Context) (*clientv3.AuthDisableResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AuthDisable\", ctx)\n\tret0, _ := ret[0].(*clientv3.AuthDisableResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AuthDisable indicates an expected call of AuthDisable.\nfunc (mr *MockClientMockRecorder) AuthDisable(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AuthDisable\", reflect.TypeOf((*MockClient)(nil).AuthDisable), ctx)\n}\n\n// AuthEnable mocks base method.\nfunc (m *MockClient) AuthEnable(ctx context.Context) (*clientv3.AuthEnableResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AuthEnable\", ctx)\n\tret0, _ := ret[0].(*clientv3.AuthEnableResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AuthEnable indicates an expected call of AuthEnable.\nfunc (mr *MockClientMockRecorder) AuthEnable(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AuthEnable\", reflect.TypeOf((*MockClient)(nil).AuthEnable), ctx)\n}\n\n// AuthStatus mocks base method.\nfunc (m *MockClient) AuthStatus(ctx context.Context) (*clientv3.AuthStatusResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AuthStatus\", ctx)\n\tret0, _ := ret[0].(*clientv3.AuthStatusResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// AuthStatus indicates an expected call of AuthStatus.\nfunc (mr *MockClientMockRecorder) AuthStatus(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AuthStatus\", reflect.TypeOf((*MockClient)(nil).AuthStatus), ctx)\n}\n\n// Authenticate mocks base method.\nfunc (m *MockClient) Authenticate(ctx context.Context, name, password string) (*clientv3.AuthenticateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Authenticate\", ctx, name, password)\n\tret0, _ := ret[0].(*clientv3.AuthenticateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Authenticate indicates an expected call of Authenticate.\nfunc (mr *MockClientMockRecorder) Authenticate(ctx, name, password any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Authenticate\", reflect.TypeOf((*MockClient)(nil).Authenticate), ctx, name, password)\n}\n\n// Close mocks base method.\nfunc (m *MockClient) Close() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Close\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockClientMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockClient)(nil).Close))\n}\n\n// Compact mocks base method.\nfunc (m *MockClient) Compact(ctx context.Context, rev int64, opts ...clientv3.CompactOption) (*clientv3.CompactResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, rev}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Compact\", varargs...)\n\tret0, _ := ret[0].(*clientv3.CompactResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Compact indicates an expected call of Compact.\nfunc (mr *MockClientMockRecorder) Compact(ctx, rev any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, rev}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Compact\", reflect.TypeOf((*MockClient)(nil).Compact), varargs...)\n}\n\n// Defragment mocks base method.\nfunc (m *MockClient) Defragment(ctx context.Context, endpoint string) (*clientv3.DefragmentResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Defragment\", ctx, endpoint)\n\tret0, _ := ret[0].(*clientv3.DefragmentResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Defragment indicates an expected call of Defragment.\nfunc (mr *MockClientMockRecorder) Defragment(ctx, endpoint any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Defragment\", reflect.TypeOf((*MockClient)(nil).Defragment), ctx, endpoint)\n}\n\n// Delete mocks base method.\nfunc (m *MockClient) Delete(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.DeleteResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, key}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Delete\", varargs...)\n\tret0, _ := ret[0].(*clientv3.DeleteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Delete indicates an expected call of Delete.\nfunc (mr *MockClientMockRecorder) Delete(ctx, key any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, key}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Delete\", reflect.TypeOf((*MockClient)(nil).Delete), varargs...)\n}\n\n// Do mocks base method.\nfunc (m *MockClient) Do(ctx context.Context, op clientv3.Op) (clientv3.OpResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Do\", ctx, op)\n\tret0, _ := ret[0].(clientv3.OpResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Do indicates an expected call of Do.\nfunc (mr *MockClientMockRecorder) Do(ctx, op any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Do\", reflect.TypeOf((*MockClient)(nil).Do), ctx, op)\n}\n\n// Get mocks base method.\nfunc (m *MockClient) Get(ctx context.Context, key string, opts ...clientv3.OpOption) (*clientv3.GetResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, key}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Get\", varargs...)\n\tret0, _ := ret[0].(*clientv3.GetResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Get indicates an expected call of Get.\nfunc (mr *MockClientMockRecorder) Get(ctx, key any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, key}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Get\", reflect.TypeOf((*MockClient)(nil).Get), varargs...)\n}\n\n// Grant mocks base method.\nfunc (m *MockClient) Grant(ctx context.Context, ttl int64) (*clientv3.LeaseGrantResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Grant\", ctx, ttl)\n\tret0, _ := ret[0].(*clientv3.LeaseGrantResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Grant indicates an expected call of Grant.\nfunc (mr *MockClientMockRecorder) Grant(ctx, ttl any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Grant\", reflect.TypeOf((*MockClient)(nil).Grant), ctx, ttl)\n}\n\n// HashKV mocks base method.\nfunc (m *MockClient) HashKV(ctx context.Context, endpoint string, rev int64) (*clientv3.HashKVResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"HashKV\", ctx, endpoint, rev)\n\tret0, _ := ret[0].(*clientv3.HashKVResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// HashKV indicates an expected call of HashKV.\nfunc (mr *MockClientMockRecorder) HashKV(ctx, endpoint, rev any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"HashKV\", reflect.TypeOf((*MockClient)(nil).HashKV), ctx, endpoint, rev)\n}\n\n// KeepAlive mocks base method.\nfunc (m *MockClient) KeepAlive(ctx context.Context, id clientv3.LeaseID) (<-chan *clientv3.LeaseKeepAliveResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"KeepAlive\", ctx, id)\n\tret0, _ := ret[0].(<-chan *clientv3.LeaseKeepAliveResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// KeepAlive indicates an expected call of KeepAlive.\nfunc (mr *MockClientMockRecorder) KeepAlive(ctx, id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"KeepAlive\", reflect.TypeOf((*MockClient)(nil).KeepAlive), ctx, id)\n}\n\n// KeepAliveOnce mocks base method.\nfunc (m *MockClient) KeepAliveOnce(ctx context.Context, id clientv3.LeaseID) (*clientv3.LeaseKeepAliveResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"KeepAliveOnce\", ctx, id)\n\tret0, _ := ret[0].(*clientv3.LeaseKeepAliveResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// KeepAliveOnce indicates an expected call of KeepAliveOnce.\nfunc (mr *MockClientMockRecorder) KeepAliveOnce(ctx, id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"KeepAliveOnce\", reflect.TypeOf((*MockClient)(nil).KeepAliveOnce), ctx, id)\n}\n\n// Leases mocks base method.\nfunc (m *MockClient) Leases(ctx context.Context) (*clientv3.LeaseLeasesResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Leases\", ctx)\n\tret0, _ := ret[0].(*clientv3.LeaseLeasesResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Leases indicates an expected call of Leases.\nfunc (mr *MockClientMockRecorder) Leases(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Leases\", reflect.TypeOf((*MockClient)(nil).Leases), ctx)\n}\n\n// MemberAdd mocks base method.\nfunc (m *MockClient) MemberAdd(ctx context.Context, peerAddrs []string) (*clientv3.MemberAddResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberAdd\", ctx, peerAddrs)\n\tret0, _ := ret[0].(*clientv3.MemberAddResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberAdd indicates an expected call of MemberAdd.\nfunc (mr *MockClientMockRecorder) MemberAdd(ctx, peerAddrs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberAdd\", reflect.TypeOf((*MockClient)(nil).MemberAdd), ctx, peerAddrs)\n}\n\n// MemberAddAsLearner mocks base method.\nfunc (m *MockClient) MemberAddAsLearner(ctx context.Context, peerAddrs []string) (*clientv3.MemberAddResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberAddAsLearner\", ctx, peerAddrs)\n\tret0, _ := ret[0].(*clientv3.MemberAddResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberAddAsLearner indicates an expected call of MemberAddAsLearner.\nfunc (mr *MockClientMockRecorder) MemberAddAsLearner(ctx, peerAddrs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberAddAsLearner\", reflect.TypeOf((*MockClient)(nil).MemberAddAsLearner), ctx, peerAddrs)\n}\n\n// MemberList mocks base method.\nfunc (m *MockClient) MemberList(ctx context.Context) (*clientv3.MemberListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberList\", ctx)\n\tret0, _ := ret[0].(*clientv3.MemberListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberList indicates an expected call of MemberList.\nfunc (mr *MockClientMockRecorder) MemberList(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberList\", reflect.TypeOf((*MockClient)(nil).MemberList), ctx)\n}\n\n// MemberPromote mocks base method.\nfunc (m *MockClient) MemberPromote(ctx context.Context, id uint64) (*clientv3.MemberPromoteResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberPromote\", ctx, id)\n\tret0, _ := ret[0].(*clientv3.MemberPromoteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberPromote indicates an expected call of MemberPromote.\nfunc (mr *MockClientMockRecorder) MemberPromote(ctx, id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberPromote\", reflect.TypeOf((*MockClient)(nil).MemberPromote), ctx, id)\n}\n\n// MemberRemove mocks base method.\nfunc (m *MockClient) MemberRemove(ctx context.Context, id uint64) (*clientv3.MemberRemoveResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberRemove\", ctx, id)\n\tret0, _ := ret[0].(*clientv3.MemberRemoveResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberRemove indicates an expected call of MemberRemove.\nfunc (mr *MockClientMockRecorder) MemberRemove(ctx, id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberRemove\", reflect.TypeOf((*MockClient)(nil).MemberRemove), ctx, id)\n}\n\n// MemberUpdate mocks base method.\nfunc (m *MockClient) MemberUpdate(ctx context.Context, id uint64, peerAddrs []string) (*clientv3.MemberUpdateResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MemberUpdate\", ctx, id, peerAddrs)\n\tret0, _ := ret[0].(*clientv3.MemberUpdateResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MemberUpdate indicates an expected call of MemberUpdate.\nfunc (mr *MockClientMockRecorder) MemberUpdate(ctx, id, peerAddrs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MemberUpdate\", reflect.TypeOf((*MockClient)(nil).MemberUpdate), ctx, id, peerAddrs)\n}\n\n// MoveLeader mocks base method.\nfunc (m *MockClient) MoveLeader(ctx context.Context, transfereeID uint64) (*clientv3.MoveLeaderResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"MoveLeader\", ctx, transfereeID)\n\tret0, _ := ret[0].(*clientv3.MoveLeaderResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// MoveLeader indicates an expected call of MoveLeader.\nfunc (mr *MockClientMockRecorder) MoveLeader(ctx, transfereeID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"MoveLeader\", reflect.TypeOf((*MockClient)(nil).MoveLeader), ctx, transfereeID)\n}\n\n// Put mocks base method.\nfunc (m *MockClient) Put(ctx context.Context, key, val string, opts ...clientv3.OpOption) (*clientv3.PutResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, key, val}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Put\", varargs...)\n\tret0, _ := ret[0].(*clientv3.PutResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Put indicates an expected call of Put.\nfunc (mr *MockClientMockRecorder) Put(ctx, key, val any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, key, val}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Put\", reflect.TypeOf((*MockClient)(nil).Put), varargs...)\n}\n\n// RequestProgress mocks base method.\nfunc (m *MockClient) RequestProgress(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RequestProgress\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RequestProgress indicates an expected call of RequestProgress.\nfunc (mr *MockClientMockRecorder) RequestProgress(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RequestProgress\", reflect.TypeOf((*MockClient)(nil).RequestProgress), ctx)\n}\n\n// Revoke mocks base method.\nfunc (m *MockClient) Revoke(ctx context.Context, id clientv3.LeaseID) (*clientv3.LeaseRevokeResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Revoke\", ctx, id)\n\tret0, _ := ret[0].(*clientv3.LeaseRevokeResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Revoke indicates an expected call of Revoke.\nfunc (mr *MockClientMockRecorder) Revoke(ctx, id any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Revoke\", reflect.TypeOf((*MockClient)(nil).Revoke), ctx, id)\n}\n\n// RoleAdd mocks base method.\nfunc (m *MockClient) RoleAdd(ctx context.Context, name string) (*clientv3.AuthRoleAddResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleAdd\", ctx, name)\n\tret0, _ := ret[0].(*clientv3.AuthRoleAddResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleAdd indicates an expected call of RoleAdd.\nfunc (mr *MockClientMockRecorder) RoleAdd(ctx, name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleAdd\", reflect.TypeOf((*MockClient)(nil).RoleAdd), ctx, name)\n}\n\n// RoleDelete mocks base method.\nfunc (m *MockClient) RoleDelete(ctx context.Context, role string) (*clientv3.AuthRoleDeleteResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleDelete\", ctx, role)\n\tret0, _ := ret[0].(*clientv3.AuthRoleDeleteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleDelete indicates an expected call of RoleDelete.\nfunc (mr *MockClientMockRecorder) RoleDelete(ctx, role any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleDelete\", reflect.TypeOf((*MockClient)(nil).RoleDelete), ctx, role)\n}\n\n// RoleGet mocks base method.\nfunc (m *MockClient) RoleGet(ctx context.Context, role string) (*clientv3.AuthRoleGetResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleGet\", ctx, role)\n\tret0, _ := ret[0].(*clientv3.AuthRoleGetResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleGet indicates an expected call of RoleGet.\nfunc (mr *MockClientMockRecorder) RoleGet(ctx, role any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleGet\", reflect.TypeOf((*MockClient)(nil).RoleGet), ctx, role)\n}\n\n// RoleGrantPermission mocks base method.\nfunc (m *MockClient) RoleGrantPermission(ctx context.Context, name, key, rangeEnd string, permType clientv3.PermissionType) (*clientv3.AuthRoleGrantPermissionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleGrantPermission\", ctx, name, key, rangeEnd, permType)\n\tret0, _ := ret[0].(*clientv3.AuthRoleGrantPermissionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleGrantPermission indicates an expected call of RoleGrantPermission.\nfunc (mr *MockClientMockRecorder) RoleGrantPermission(ctx, name, key, rangeEnd, permType any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleGrantPermission\", reflect.TypeOf((*MockClient)(nil).RoleGrantPermission), ctx, name, key, rangeEnd, permType)\n}\n\n// RoleList mocks base method.\nfunc (m *MockClient) RoleList(ctx context.Context) (*clientv3.AuthRoleListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleList\", ctx)\n\tret0, _ := ret[0].(*clientv3.AuthRoleListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleList indicates an expected call of RoleList.\nfunc (mr *MockClientMockRecorder) RoleList(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleList\", reflect.TypeOf((*MockClient)(nil).RoleList), ctx)\n}\n\n// RoleRevokePermission mocks base method.\nfunc (m *MockClient) RoleRevokePermission(ctx context.Context, role, key, rangeEnd string) (*clientv3.AuthRoleRevokePermissionResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RoleRevokePermission\", ctx, role, key, rangeEnd)\n\tret0, _ := ret[0].(*clientv3.AuthRoleRevokePermissionResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// RoleRevokePermission indicates an expected call of RoleRevokePermission.\nfunc (mr *MockClientMockRecorder) RoleRevokePermission(ctx, role, key, rangeEnd any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RoleRevokePermission\", reflect.TypeOf((*MockClient)(nil).RoleRevokePermission), ctx, role, key, rangeEnd)\n}\n\n// Snapshot mocks base method.\nfunc (m *MockClient) Snapshot(ctx context.Context) (io.ReadCloser, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Snapshot\", ctx)\n\tret0, _ := ret[0].(io.ReadCloser)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Snapshot indicates an expected call of Snapshot.\nfunc (mr *MockClientMockRecorder) Snapshot(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Snapshot\", reflect.TypeOf((*MockClient)(nil).Snapshot), ctx)\n}\n\n// Status mocks base method.\nfunc (m *MockClient) Status(ctx context.Context, endpoint string) (*clientv3.StatusResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Status\", ctx, endpoint)\n\tret0, _ := ret[0].(*clientv3.StatusResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Status indicates an expected call of Status.\nfunc (mr *MockClientMockRecorder) Status(ctx, endpoint any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Status\", reflect.TypeOf((*MockClient)(nil).Status), ctx, endpoint)\n}\n\n// TimeToLive mocks base method.\nfunc (m *MockClient) TimeToLive(ctx context.Context, id clientv3.LeaseID, opts ...clientv3.LeaseOption) (*clientv3.LeaseTimeToLiveResponse, error) {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, id}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"TimeToLive\", varargs...)\n\tret0, _ := ret[0].(*clientv3.LeaseTimeToLiveResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// TimeToLive indicates an expected call of TimeToLive.\nfunc (mr *MockClientMockRecorder) TimeToLive(ctx, id any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, id}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"TimeToLive\", reflect.TypeOf((*MockClient)(nil).TimeToLive), varargs...)\n}\n\n// Txn mocks base method.\nfunc (m *MockClient) Txn(ctx context.Context) clientv3.Txn {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Txn\", ctx)\n\tret0, _ := ret[0].(clientv3.Txn)\n\treturn ret0\n}\n\n// Txn indicates an expected call of Txn.\nfunc (mr *MockClientMockRecorder) Txn(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Txn\", reflect.TypeOf((*MockClient)(nil).Txn), ctx)\n}\n\n// UserAdd mocks base method.\nfunc (m *MockClient) UserAdd(ctx context.Context, name, password string) (*clientv3.AuthUserAddResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserAdd\", ctx, name, password)\n\tret0, _ := ret[0].(*clientv3.AuthUserAddResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserAdd indicates an expected call of UserAdd.\nfunc (mr *MockClientMockRecorder) UserAdd(ctx, name, password any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserAdd\", reflect.TypeOf((*MockClient)(nil).UserAdd), ctx, name, password)\n}\n\n// UserAddWithOptions mocks base method.\nfunc (m *MockClient) UserAddWithOptions(ctx context.Context, name, password string, opt *clientv3.UserAddOptions) (*clientv3.AuthUserAddResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserAddWithOptions\", ctx, name, password, opt)\n\tret0, _ := ret[0].(*clientv3.AuthUserAddResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserAddWithOptions indicates an expected call of UserAddWithOptions.\nfunc (mr *MockClientMockRecorder) UserAddWithOptions(ctx, name, password, opt any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserAddWithOptions\", reflect.TypeOf((*MockClient)(nil).UserAddWithOptions), ctx, name, password, opt)\n}\n\n// UserChangePassword mocks base method.\nfunc (m *MockClient) UserChangePassword(ctx context.Context, name, password string) (*clientv3.AuthUserChangePasswordResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserChangePassword\", ctx, name, password)\n\tret0, _ := ret[0].(*clientv3.AuthUserChangePasswordResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserChangePassword indicates an expected call of UserChangePassword.\nfunc (mr *MockClientMockRecorder) UserChangePassword(ctx, name, password any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserChangePassword\", reflect.TypeOf((*MockClient)(nil).UserChangePassword), ctx, name, password)\n}\n\n// UserDelete mocks base method.\nfunc (m *MockClient) UserDelete(ctx context.Context, name string) (*clientv3.AuthUserDeleteResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserDelete\", ctx, name)\n\tret0, _ := ret[0].(*clientv3.AuthUserDeleteResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserDelete indicates an expected call of UserDelete.\nfunc (mr *MockClientMockRecorder) UserDelete(ctx, name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserDelete\", reflect.TypeOf((*MockClient)(nil).UserDelete), ctx, name)\n}\n\n// UserGet mocks base method.\nfunc (m *MockClient) UserGet(ctx context.Context, name string) (*clientv3.AuthUserGetResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserGet\", ctx, name)\n\tret0, _ := ret[0].(*clientv3.AuthUserGetResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserGet indicates an expected call of UserGet.\nfunc (mr *MockClientMockRecorder) UserGet(ctx, name any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserGet\", reflect.TypeOf((*MockClient)(nil).UserGet), ctx, name)\n}\n\n// UserGrantRole mocks base method.\nfunc (m *MockClient) UserGrantRole(ctx context.Context, user, role string) (*clientv3.AuthUserGrantRoleResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserGrantRole\", ctx, user, role)\n\tret0, _ := ret[0].(*clientv3.AuthUserGrantRoleResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserGrantRole indicates an expected call of UserGrantRole.\nfunc (mr *MockClientMockRecorder) UserGrantRole(ctx, user, role any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserGrantRole\", reflect.TypeOf((*MockClient)(nil).UserGrantRole), ctx, user, role)\n}\n\n// UserList mocks base method.\nfunc (m *MockClient) UserList(ctx context.Context) (*clientv3.AuthUserListResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserList\", ctx)\n\tret0, _ := ret[0].(*clientv3.AuthUserListResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserList indicates an expected call of UserList.\nfunc (mr *MockClientMockRecorder) UserList(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserList\", reflect.TypeOf((*MockClient)(nil).UserList), ctx)\n}\n\n// UserRevokeRole mocks base method.\nfunc (m *MockClient) UserRevokeRole(ctx context.Context, name, role string) (*clientv3.AuthUserRevokeRoleResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UserRevokeRole\", ctx, name, role)\n\tret0, _ := ret[0].(*clientv3.AuthUserRevokeRoleResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// UserRevokeRole indicates an expected call of UserRevokeRole.\nfunc (mr *MockClientMockRecorder) UserRevokeRole(ctx, name, role any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UserRevokeRole\", reflect.TypeOf((*MockClient)(nil).UserRevokeRole), ctx, name, role)\n}\n\n// Watch mocks base method.\nfunc (m *MockClient) Watch(ctx context.Context, key string, opts ...clientv3.OpOption) clientv3.WatchChan {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{ctx, key}\n\tfor _, a := range opts {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"Watch\", varargs...)\n\tret0, _ := ret[0].(clientv3.WatchChan)\n\treturn ret0\n}\n\n// Watch indicates an expected call of Watch.\nfunc (mr *MockClientMockRecorder) Watch(ctx, key any, opts ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{ctx, key}, opts...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Watch\", reflect.TypeOf((*MockClient)(nil).Watch), varargs...)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdkeys/etcdkeys.go",
    "content": "package etcdkeys\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\n// BuildNamespacePrefix constructs the etcd key prefix for a given namespace.\n// result: <prefix>/<namespace>\nfunc BuildNamespacePrefix(prefix, namespace string) string {\n\treturn fmt.Sprintf(\"%s/%s\", prefix, namespace)\n}\n\n// BuildExecutorsPrefix constructs the etcd key prefix for executors within a given namespace.\n// result: <prefix>/<namespace>/executors/\nfunc BuildExecutorsPrefix(prefix, namespace string) string {\n\treturn fmt.Sprintf(\"%s/executors/\", BuildNamespacePrefix(prefix, namespace))\n}\n\n// BuildExecutorIDPrefix constructs the etcd key prefix for a specific executor within a namespace.\n// result: <prefix>/<namespace>/executors/<executorID>/\nfunc BuildExecutorIDPrefix(prefix, namespace, executorID string) string {\n\treturn fmt.Sprintf(\"%s%s/\", BuildExecutorsPrefix(prefix, namespace), executorID)\n}\n\n// ExecutorKeyType represents the allowed executor-level key types in etcd.\n// Use BuildExecutorKey to construct keys of these types.\ntype ExecutorKeyType string\n\nconst (\n\tExecutorHeartbeatKey       ExecutorKeyType = \"heartbeat\"\n\tExecutorStatusKey          ExecutorKeyType = \"status\"\n\tExecutorReportedShardsKey  ExecutorKeyType = \"reported_shards\"\n\tExecutorAssignedStateKey   ExecutorKeyType = \"assigned_state\"\n\tExecutorMetadataKey        ExecutorKeyType = \"metadata\"\n\tExecutorShardStatisticsKey ExecutorKeyType = \"statistics\"\n)\n\n// validExecutorKeyTypes defines the set of valid executor key types.\nvar validExecutorKeyTypes = map[ExecutorKeyType]struct{}{\n\tExecutorHeartbeatKey:       {},\n\tExecutorStatusKey:          {},\n\tExecutorReportedShardsKey:  {},\n\tExecutorAssignedStateKey:   {},\n\tExecutorMetadataKey:        {},\n\tExecutorShardStatisticsKey: {},\n}\n\n// IsValidExecutorKeyType checks if the provided key type is valid.\nfunc IsValidExecutorKeyType(keyType ExecutorKeyType) bool {\n\t_, exist := validExecutorKeyTypes[keyType]\n\treturn exist\n}\n\n// BuildExecutorKey constructs the etcd key for a specific executor and key type.\n// result: <prefix>/<namespace>/executors/<executorID>/<keyType>\nfunc BuildExecutorKey(prefix, namespace, executorID string, keyType ExecutorKeyType) string {\n\treturn fmt.Sprintf(\"%s%s\", BuildExecutorIDPrefix(prefix, namespace, executorID), keyType)\n}\n\n// ParseExecutorKey parses an etcd key and extracts the executor ID and key type.\n// It returns an error if the key does not conform to the expected format.\n// Expected format of key: <prefix>/<namespace>/executors/<executorID>/<keyType>\nfunc ParseExecutorKey(prefix, namespace, key string) (executorID string, keyType ExecutorKeyType, err error) {\n\tprefix = BuildExecutorsPrefix(prefix, namespace)\n\tif !strings.HasPrefix(key, prefix) {\n\t\treturn \"\", \"\", fmt.Errorf(\"key '%s' does not have expected prefix '%s'\", key, prefix)\n\t}\n\tremainder := strings.TrimPrefix(key, prefix)\n\tparts := strings.Split(remainder, \"/\")\n\tif len(parts) < 2 {\n\t\treturn \"\", \"\", fmt.Errorf(\"unexpected key format: %s\", key)\n\t}\n\t// For metadata keys, the format is: executorID/metadata/metadataKey\n\t// For other keys, the format is: executorID/keyType\n\t// We return executorID and the first keyType (e.g., \"metadata\")\n\tif len(parts) > 2 && ExecutorKeyType(parts[1]) == ExecutorMetadataKey {\n\t\t// This is a metadata key, return \"metadata\" as the keyType\n\t\treturn parts[0], ExecutorMetadataKey, nil\n\t}\n\tif len(parts) != 2 {\n\t\treturn \"\", \"\", fmt.Errorf(\"unexpected key format: %s\", key)\n\t}\n\tif !IsValidExecutorKeyType(ExecutorKeyType(parts[1])) {\n\t\treturn \"\", \"\", fmt.Errorf(\"invalid executor key type: %s\", parts[1])\n\t}\n\treturn parts[0], ExecutorKeyType(parts[1]), nil\n}\n\n// BuildMetadataKey constructs the etcd key for a specific metadata entry of an executor.\n// result: <prefix>/<namespace>/executors/<executorID>/metadata/<metadataKey>\nfunc BuildMetadataKey(prefix string, namespace, executorID, metadataKey string) string {\n\treturn fmt.Sprintf(\"%s/%s\", BuildExecutorKey(prefix, namespace, executorID, ExecutorMetadataKey), metadataKey)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdkeys/etcdkeys_test.go",
    "content": "package etcdkeys\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestBuildNamespacePrefix(t *testing.T) {\n\tgot := BuildNamespacePrefix(\"/cadence\", \"test-ns\")\n\tassert.Equal(t, \"/cadence/test-ns\", got)\n}\n\nfunc TestBuildExecutorsPrefix(t *testing.T) {\n\tgot := BuildExecutorsPrefix(\"/cadence\", \"test-ns\")\n\tassert.Equal(t, \"/cadence/test-ns/executors/\", got)\n}\n\nfunc TestBuildExecutorKey(t *testing.T) {\n\tgot := BuildExecutorKey(\"/cadence\", \"test-ns\", \"exec-1\", \"heartbeat\")\n\tassert.Equal(t, \"/cadence/test-ns/executors/exec-1/heartbeat\", got)\n}\n\nfunc TestParseExecutorKey(t *testing.T) {\n\t// Valid key\n\texecutorID, keyType, err := ParseExecutorKey(\"/cadence\", \"test-ns\", \"/cadence/test-ns/executors/exec-1/heartbeat\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"exec-1\", executorID)\n\tassert.Equal(t, ExecutorHeartbeatKey, keyType)\n\n\t// Prefix missing\n\t_, _, err = ParseExecutorKey(\"/cadence\", \"test-ns\", \"/wrong/prefix\")\n\tassert.ErrorContains(t, err, \"key '/wrong/prefix' does not have expected prefix '/cadence/test-ns/executors/'\")\n\n\t// Unexpected key format\n\t_, _, err = ParseExecutorKey(\"/cadence\", \"test-ns\", \"/cadence/test-ns/executors/exec-1/heartbeat/extra\")\n\tassert.ErrorContains(t, err, \"unexpected key format: /cadence/test-ns/executors/exec-1/heartbeat/extra\")\n}\n\nfunc TestBuildMetadataKey(t *testing.T) {\n\tgot := BuildMetadataKey(\"/cadence\", \"test-ns\", \"exec-1\", \"my-metadata-key\")\n\tassert.Equal(t, \"/cadence/test-ns/executors/exec-1/metadata/my-metadata-key\", got)\n}\n\nfunc TestParseExecutorKey_MetadataKey(t *testing.T) {\n\t// Test that ParseExecutorKey correctly identifies metadata keys\n\t// and that we can extract the metadata key name from the full key\n\tmetadataKey := BuildMetadataKey(\"/cadence\", \"test-ns\", \"exec-1\", \"hostname\")\n\n\texecutorID, keyType, err := ParseExecutorKey(\"/cadence\", \"test-ns\", metadataKey)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"exec-1\", executorID)\n\tassert.Equal(t, ExecutorMetadataKey, keyType)\n}\n\nfunc TestParseExecutorKey_InvalidKeyType(t *testing.T) {\n\tkey := BuildExecutorIDPrefix(\"/cadence\", \"test-ns\", \"exec-1\") + \"invalid_type\"\n\t_, _, err := ParseExecutorKey(\"/cadence\", \"test-ns\", key)\n\tassert.ErrorContains(t, err, \"invalid executor key type: invalid_type\")\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdtypes/state.go",
    "content": "package etcdtypes\n\nimport (\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\ntype AssignedState struct {\n\tAssignedShards     map[string]*types.ShardAssignment `json:\"assigned_shards\"`\n\tShardHandoverStats map[string]ShardHandoverStats     `json:\"shard_handover_stats,omitempty\"`\n\tLastUpdated        Time                              `json:\"last_updated\"`\n\t// ModRevision is the etcd mod revision for this record. It is not serialized.\n\tModRevision int64 `json:\"-\"`\n}\n\n// ToAssignedState converts the current AssignedState to store.AssignedState.\nfunc (s *AssignedState) ToAssignedState() *store.AssignedState {\n\tif s == nil {\n\t\treturn nil\n\t}\n\n\treturn &store.AssignedState{\n\t\tAssignedShards:     s.AssignedShards,\n\t\tShardHandoverStats: convertMap(s.ShardHandoverStats, ToShardHandoverStats),\n\t\tLastUpdated:        s.LastUpdated.ToTime(),\n\t\tModRevision:        s.ModRevision,\n\t}\n}\n\n// FromAssignedState creates an AssignedState from a store.AssignedState.\nfunc FromAssignedState(src *store.AssignedState) *AssignedState {\n\tif src == nil {\n\t\treturn nil\n\t}\n\n\treturn &AssignedState{\n\t\tAssignedShards:     src.AssignedShards,\n\t\tLastUpdated:        Time(src.LastUpdated),\n\t\tShardHandoverStats: convertMap(src.ShardHandoverStats, FromShardHandoverStats),\n\t\tModRevision:        src.ModRevision,\n\t}\n}\n\ntype ShardHandoverStats struct {\n\tPreviousExecutorLastHeartbeatTime Time               `json:\"previous_executor_last_heartbeat_time\"`\n\tHandoverType                      types.HandoverType `json:\"handover_type\"`\n}\n\n// FromShardHandoverStats creates an ShardHandoverStats from a store.ShardHandoverStats.\nfunc FromShardHandoverStats(src *store.ShardHandoverStats) *ShardHandoverStats {\n\tif src == nil {\n\t\treturn nil\n\t}\n\n\treturn &ShardHandoverStats{\n\t\tPreviousExecutorLastHeartbeatTime: Time(src.PreviousExecutorLastHeartbeatTime),\n\t\tHandoverType:                      src.HandoverType,\n\t}\n}\n\n// ToShardHandoverStats converts the current ShardHandoverStats to store.ShardHandoverStats.\nfunc ToShardHandoverStats(src *ShardHandoverStats) *store.ShardHandoverStats {\n\tif src == nil {\n\t\treturn nil\n\t}\n\n\treturn &store.ShardHandoverStats{\n\t\tPreviousExecutorLastHeartbeatTime: src.PreviousExecutorLastHeartbeatTime.ToTime(),\n\t\tHandoverType:                      src.HandoverType,\n\t}\n}\n\ntype ShardStatistics struct {\n\tSmoothedLoad   float64 `json:\"smoothed_load\"`\n\tLastUpdateTime Time    `json:\"last_update_time\"`\n\tLastMoveTime   Time    `json:\"last_move_time\"`\n}\n\n// ToShardStatistics converts the current ShardStatistics to store.ShardStatistics.\nfunc (s *ShardStatistics) ToShardStatistics() *store.ShardStatistics {\n\tif s == nil {\n\t\treturn nil\n\t}\n\n\treturn &store.ShardStatistics{\n\t\tSmoothedLoad:   s.SmoothedLoad,\n\t\tLastUpdateTime: s.LastUpdateTime.ToTime(),\n\t\tLastMoveTime:   s.LastMoveTime.ToTime(),\n\t}\n}\n\n// FromShardStatistics creates a ShardStatistics from a store.ShardStatistics.\nfunc FromShardStatistics(src *store.ShardStatistics) *ShardStatistics {\n\tif src == nil {\n\t\treturn nil\n\t}\n\n\treturn &ShardStatistics{\n\t\tSmoothedLoad:   src.SmoothedLoad,\n\t\tLastUpdateTime: Time(src.LastUpdateTime),\n\t\tLastMoveTime:   Time(src.LastMoveTime),\n\t}\n}\n\n// ConvertMap converts a map[K]SrcType to map[K]DstType using a provided converter function.\nfunc convertMap[K comparable, SrcType any, DstType any](src map[K]SrcType, converter func(*SrcType) *DstType) map[K]DstType {\n\tif src == nil {\n\t\treturn nil\n\t}\n\tdst := make(map[K]DstType, len(src))\n\tfor k, v := range src {\n\t\tdst[k] = *converter(&v)\n\t}\n\treturn dst\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdtypes/state_test.go",
    "content": "package etcdtypes\n\nimport (\n\t\"encoding/json\"\n\t\"reflect\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nfunc TestAssignedState_FieldNumberMatched(t *testing.T) {\n\trequire.Equal(t,\n\t\treflect.TypeOf(AssignedState{}).NumField(),\n\t\treflect.TypeOf(store.AssignedState{}).NumField(),\n\t\t\"AssignedState field count mismatch with store.AssignedState; ensure conversion is updated\",\n\t)\n}\n\nfunc TestAssignedState_ToAssignedState(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *AssignedState\n\t\texpect *store.AssignedState\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t\texpect: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t},\n\t\t\"success with map\": {\n\t\t\tinput: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tShardHandoverStats: map[string]ShardHandoverStats{\n\t\t\t\t\t\"1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t\texpect: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tLastUpdated: time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := c.input.ToAssignedState()\n\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.Equal(t, len(c.input.AssignedShards), len(got.AssignedShards))\n\t\t\tfor k := range c.input.AssignedShards {\n\t\t\t\trequire.Equal(t, c.input.AssignedShards[k].Status, got.AssignedShards[k].Status)\n\t\t\t}\n\t\t\trequire.Equal(t, time.Time(c.input.LastUpdated).UnixNano(), got.LastUpdated.UnixNano())\n\t\t\trequire.Equal(t, c.input.ModRevision, got.ModRevision)\n\t\t})\n\t}\n}\nfunc TestAssignedState_FromAssignedState(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *store.AssignedState\n\t\texpect *AssignedState\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"9\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC),\n\t\t\t\tModRevision: 77,\n\t\t\t},\n\t\t\texpect: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"9\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC)),\n\t\t\t\tModRevision: 77,\n\t\t\t},\n\t\t},\n\t\t\"with map\": {\n\t\t\tinput: &store.AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"9\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tShardHandoverStats: map[string]store.ShardHandoverStats{\n\t\t\t\t\t\"9\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tLastUpdated: time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC),\n\t\t\t\tModRevision: 77,\n\t\t\t},\n\t\t\texpect: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"9\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tShardHandoverStats: map[string]ShardHandoverStats{\n\t\t\t\t\t\"9\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC)),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 13, 0, 0, 987654321, time.UTC)),\n\t\t\t\tModRevision: 77,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := FromAssignedState(c.input)\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.Equal(t, len(c.input.AssignedShards), len(got.AssignedShards))\n\t\t\tfor k := range c.input.AssignedShards {\n\t\t\t\trequire.Equal(t, c.input.AssignedShards[k].Status, got.AssignedShards[k].Status)\n\t\t\t}\n\t\t\trequire.Equal(t, c.input.LastUpdated.UnixNano(), time.Time(got.LastUpdated).UnixNano())\n\t\t\trequire.Equal(t, c.input.ModRevision, got.ModRevision)\n\t\t})\n\t}\n}\n\nfunc TestAssignedState_JSONMarshalling(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput   *AssignedState\n\t\tjsonStr string\n\t}{\n\t\t\"simple\": {\n\t\t\tinput: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t\tjsonStr: `{\"assigned_shards\":{\"1\":{\"status\":\"AssignmentStatusREADY\"}},\"last_updated\":\"2025-11-18T12:00:00.123456789Z\"}`,\n\t\t},\n\t\t\"with map\": {\n\t\t\tinput: &AssignedState{\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\t\"1\": {Status: types.AssignmentStatusREADY},\n\t\t\t\t},\n\t\t\t\tShardHandoverStats: map[string]ShardHandoverStats{\n\t\t\t\t\t\"1\": {\n\t\t\t\t\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tLastUpdated: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\t\tModRevision: 42,\n\t\t\t},\n\t\t\tjsonStr: `{\"assigned_shards\":{\"1\":{\"status\":\"AssignmentStatusREADY\"}},\"shard_handover_stats\":{\"1\":{\"previous_executor_last_heartbeat_time\":\"2025-11-18T12:00:00.123456789Z\",\"handover_type\":\"HandoverTypeGRACEFUL\"}},\"last_updated\":\"2025-11-18T12:00:00.123456789Z\"}`,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Marshal to JSON\n\t\t\tb, err := json.Marshal(tc.input)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.JSONEq(t, tc.jsonStr, string(b))\n\n\t\t\t// Unmarshal from JSON\n\t\t\tvar unmarshalled AssignedState\n\t\t\terr = json.Unmarshal([]byte(tc.jsonStr), &unmarshalled)\n\t\t\trequire.NoError(t, err)\n\t\t\trequire.Equal(t, tc.input.AssignedShards[\"1\"].Status, unmarshalled.AssignedShards[\"1\"].Status)\n\t\t\trequire.Equal(t, time.Time(tc.input.LastUpdated).UnixNano(), time.Time(unmarshalled.LastUpdated).UnixNano())\n\t\t\tif tc.input.ShardHandoverStats != nil {\n\t\t\t\trequire.NotNil(t, unmarshalled.ShardHandoverStats)\n\t\t\t\trequire.Equal(t, tc.input.ShardHandoverStats[\"1\"].HandoverType, unmarshalled.ShardHandoverStats[\"1\"].HandoverType)\n\t\t\t\trequire.Equal(t, time.Time(tc.input.ShardHandoverStats[\"1\"].PreviousExecutorLastHeartbeatTime).UnixNano(),\n\t\t\t\t\ttime.Time(unmarshalled.ShardHandoverStats[\"1\"].PreviousExecutorLastHeartbeatTime).UnixNano())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestShardStatistics_FieldNumberMatched(t *testing.T) {\n\trequire.Equal(t,\n\t\treflect.TypeOf(ShardStatistics{}).NumField(),\n\t\treflect.TypeOf(store.ShardStatistics{}).NumField(),\n\t\t\"ShardStatistics field count mismatch with store.ShardStatistics; ensure conversion is updated\",\n\t)\n}\n\nfunc TestShardStatistics_ToShardStatistics(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *ShardStatistics\n\t\texpect *store.ShardStatistics\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &ShardStatistics{\n\t\t\t\tSmoothedLoad:   12.34,\n\t\t\t\tLastUpdateTime: Time(time.Date(2025, 11, 18, 14, 0, 0, 111111111, time.UTC)),\n\t\t\t\tLastMoveTime:   Time(time.Date(2025, 11, 18, 15, 0, 0, 222222222, time.UTC)),\n\t\t\t},\n\t\t\texpect: &store.ShardStatistics{\n\t\t\t\tSmoothedLoad:   12.34,\n\t\t\t\tLastUpdateTime: time.Date(2025, 11, 18, 14, 0, 0, 111111111, time.UTC),\n\t\t\t\tLastMoveTime:   time.Date(2025, 11, 18, 15, 0, 0, 222222222, time.UTC),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := c.input.ToShardStatistics()\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.Equal(t, c.input.SmoothedLoad, got.SmoothedLoad)\n\t\t\trequire.Equal(t, time.Time(c.input.LastUpdateTime).UnixNano(), got.LastUpdateTime.UnixNano())\n\t\t\trequire.Equal(t, time.Time(c.input.LastMoveTime).UnixNano(), got.LastMoveTime.UnixNano())\n\t\t})\n\t}\n}\n\nfunc TestShardStatistics_FromShardStatistics(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *store.ShardStatistics\n\t\texpect *ShardStatistics\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &store.ShardStatistics{\n\t\t\t\tSmoothedLoad:   99.01,\n\t\t\t\tLastUpdateTime: time.Date(2025, 11, 18, 16, 0, 0, 333333333, time.UTC),\n\t\t\t\tLastMoveTime:   time.Date(2025, 11, 18, 17, 0, 0, 444444444, time.UTC),\n\t\t\t},\n\t\t\texpect: &ShardStatistics{\n\t\t\t\tSmoothedLoad:   99.01,\n\t\t\t\tLastUpdateTime: Time(time.Date(2025, 11, 18, 16, 0, 0, 333333333, time.UTC)),\n\t\t\t\tLastMoveTime:   Time(time.Date(2025, 11, 18, 17, 0, 0, 444444444, time.UTC)),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := FromShardStatistics(c.input)\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.InDelta(t, c.input.SmoothedLoad, got.SmoothedLoad, 0.0000001)\n\t\t\trequire.Equal(t, c.input.LastUpdateTime.UnixNano(), time.Time(got.LastUpdateTime).UnixNano())\n\t\t\trequire.Equal(t, c.input.LastMoveTime.UnixNano(), time.Time(got.LastMoveTime).UnixNano())\n\t\t})\n\t}\n}\n\nfunc TestShardStatistics_JSONMarshalling(t *testing.T) {\n\tconst jsonStr = `{\"smoothed_load\":12.34,\"last_update_time\":\"2025-11-18T14:00:00.111111111Z\",\"last_move_time\":\"2025-11-18T15:00:00.222222222Z\"}`\n\n\tstate := &ShardStatistics{\n\t\tSmoothedLoad:   12.34,\n\t\tLastUpdateTime: Time(time.Date(2025, 11, 18, 14, 0, 0, 111111111, time.UTC)),\n\t\tLastMoveTime:   Time(time.Date(2025, 11, 18, 15, 0, 0, 222222222, time.UTC)),\n\t}\n\n\t// Marshal to JSON\n\tb, err := json.Marshal(state)\n\trequire.NoError(t, err)\n\trequire.JSONEq(t, jsonStr, string(b))\n\n\t// Unmarshal from JSON\n\tvar unmarshalled ShardStatistics\n\terr = json.Unmarshal([]byte(jsonStr), &unmarshalled)\n\trequire.NoError(t, err)\n\trequire.InDelta(t, state.SmoothedLoad, unmarshalled.SmoothedLoad, 0.0000001)\n\trequire.Equal(t, time.Time(state.LastUpdateTime).UnixNano(), time.Time(unmarshalled.LastUpdateTime).UnixNano())\n\trequire.Equal(t, time.Time(state.LastMoveTime).UnixNano(), time.Time(unmarshalled.LastMoveTime).UnixNano())\n}\n\nfunc TestShardHandoverStats_FieldNumberMatched(t *testing.T) {\n\trequire.Equal(t,\n\t\treflect.TypeOf(ShardHandoverStats{}).NumField(),\n\t\treflect.TypeOf(store.ShardHandoverStats{}).NumField(),\n\t\t\"ShardHandoverStats field count mismatch with store.ShardHandoverStats; ensure conversion is updated\",\n\t)\n}\n\nfunc TestShardHandoverStats_ToShardHandoverStats(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *ShardHandoverStats\n\t\texpect *store.ShardHandoverStats\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &ShardHandoverStats{\n\t\t\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 18, 0, 0, 555555555, time.UTC)),\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t},\n\t\t\texpect: &store.ShardHandoverStats{\n\t\t\t\tPreviousExecutorLastHeartbeatTime: time.Date(2025, 11, 18, 18, 0, 0, 555555555, time.UTC),\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := ToShardHandoverStats(c.input)\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.Equal(t, c.input.HandoverType, got.HandoverType)\n\t\t\trequire.Equal(t, time.Time(c.input.PreviousExecutorLastHeartbeatTime).UnixNano(), got.PreviousExecutorLastHeartbeatTime.UnixNano())\n\t\t})\n\t}\n}\n\nfunc TestShardHandoverStats_FromShardHandoverStats(t *testing.T) {\n\ttests := map[string]struct {\n\t\tinput  *store.ShardHandoverStats\n\t\texpect *ShardHandoverStats\n\t}{\n\t\t\"nil\": {\n\t\t\tinput:  nil,\n\t\t\texpect: nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tinput: &store.ShardHandoverStats{\n\t\t\t\tPreviousExecutorLastHeartbeatTime: time.Date(2025, 11, 18, 19, 0, 0, 666666666, time.UTC),\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t},\n\t\t\texpect: &ShardHandoverStats{\n\t\t\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 19, 0, 0, 666666666, time.UTC)),\n\t\t\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, c := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tgot := FromShardHandoverStats(c.input)\n\t\t\tif c.expect == nil {\n\t\t\t\trequire.Nil(t, got)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NotNil(t, got)\n\t\t\trequire.Equal(t, c.input.HandoverType, got.HandoverType)\n\t\t\trequire.Equal(t, c.input.PreviousExecutorLastHeartbeatTime.UnixNano(), time.Time(got.PreviousExecutorLastHeartbeatTime).UnixNano())\n\t\t})\n\t}\n}\n\nfunc TestShardHandoverStats_JSONMarshalling(t *testing.T) {\n\tconst jsonStr = `{\"previous_executor_last_heartbeat_time\":\"2025-11-18T20:00:00.777777777Z\",\"handover_type\":\"HandoverTypeGRACEFUL\"}`\n\n\tstats := &ShardHandoverStats{\n\t\tPreviousExecutorLastHeartbeatTime: Time(time.Date(2025, 11, 18, 20, 0, 0, 777777777, time.UTC)),\n\t\tHandoverType:                      types.HandoverTypeGRACEFUL,\n\t}\n\n\t// Marshal to JSON\n\tb, err := json.Marshal(stats)\n\trequire.NoError(t, err)\n\trequire.JSONEq(t, jsonStr, string(b))\n\n\t// Unmarshal from JSON\n\tvar unmarshalled ShardHandoverStats\n\terr = json.Unmarshal([]byte(jsonStr), &unmarshalled)\n\trequire.NoError(t, err)\n\trequire.Equal(t, stats.HandoverType, unmarshalled.HandoverType)\n\trequire.Equal(t, time.Time(stats.PreviousExecutorLastHeartbeatTime).UnixNano(), time.Time(unmarshalled.PreviousExecutorLastHeartbeatTime).UnixNano())\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdtypes/time.go",
    "content": "package etcdtypes\n\nimport \"time\"\n\n// Time is a wrapper around time that implements JSON marshalling/unmarshalling\n// in time.RFC3339Nano format to keep precision when storing in etcd.\n// Convert to UTC before storing/parsing to ensure consistency.\ntype Time time.Time\n\n// ToTimePtr converts *time.Time to *Time.\nfunc ToTimePtr(t *time.Time) *Time {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttt := Time(*t)\n\treturn &tt\n}\n\n// ToTime converts Time back to time.Time.\nfunc (t Time) ToTime() time.Time {\n\treturn time.Time(t)\n}\n\n// ToTimePtr converts Time back to *time.Time.\nfunc (t *Time) ToTimePtr() *time.Time {\n\tif t == nil {\n\t\treturn nil\n\t}\n\ttt := time.Time(*t)\n\treturn &tt\n}\n\n// MarshalJSON implements the json.Marshaler interface.\n// It encodes the time in time.RFC3339Nano format.\nfunc (t Time) MarshalJSON() ([]byte, error) {\n\ts := FormatTime(time.Time(t))\n\treturn []byte(`\"` + s + `\"`), nil\n}\n\n// UnmarshalJSON implements the json.Unmarshaler interface.\n// It decodes the time from time.RFC3339Nano format.\nfunc (t *Time) UnmarshalJSON(data []byte) error {\n\tstr := string(data)\n\tif len(str) >= 2 && str[0] == '\"' && str[len(str)-1] == '\"' {\n\t\tstr = str[1 : len(str)-1]\n\t}\n\tparsed, err := ParseTime(str)\n\tif err != nil {\n\t\treturn err\n\t}\n\t*t = Time(parsed)\n\treturn nil\n}\n\n// ParseTime parses a string in time.RFC3339Nano format and returns a time.Time in UTC.\nfunc ParseTime(s string) (time.Time, error) {\n\tparsed, err := time.ParseInLocation(time.RFC3339Nano, s, time.UTC)\n\tif err != nil {\n\t\treturn time.Time{}, err\n\t}\n\treturn parsed.UTC(), nil\n}\n\n// FormatTime converts time.Time to UTC and\n// formats time.Time to a string in time.RFC3339Nano format.\nfunc FormatTime(t time.Time) string {\n\treturn t.UTC().Format(time.RFC3339Nano)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/etcdtypes/time_test.go",
    "content": "package etcdtypes\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\n// TestTimeToTime verifies ToTime returns the original time.Time value.\nfunc TestTimeToTime(t *testing.T) {\n\tnow := time.Now()\n\trequire.Equal(t, now, Time(now).ToTime())\n}\n\nfunc TestTimeToTimePtr_NilInput_ReturnsNil(t *testing.T) {\n\tvar input *time.Time\n\tresult := ToTimePtr(input)\n\trequire.Nil(t, result)\n}\n\nfunc TestTimeToTimePtr(t *testing.T) {\n\tnow := time.Now()\n\tresult := ToTimePtr(&now)\n\trequire.NotNil(t, result)\n\trequire.Equal(t, Time(now), *result)\n}\n\nfunc TestTimeToTimePtr_Nil(t *testing.T) {\n\tresult := ToTimePtr(nil)\n\trequire.Nil(t, result)\n}\n\nfunc TestTimeMarshalJSON(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput        Time\n\t\texpectOutput string\n\t\texpectErr    string\n\t}{\n\t\t\"zero time\": {\n\t\t\tinput:        Time(time.Time{}),\n\t\t\texpectOutput: `\"0001-01-01T00:00:00Z\"`,\n\t\t\texpectErr:    \"\",\n\t\t},\n\t\t\"utc time with nanos\": {\n\t\t\tinput:        Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\texpectOutput: `\"2025-11-18T12:00:00.123456789Z\"`,\n\t\t\texpectErr:    \"\",\n\t\t},\n\t\t\"non-UTC time\": {\n\t\t\tinput:        Time(time.Date(2025, 11, 18, 1, 2, 3, 456789012, time.FixedZone(\"X\", -2*3600))),\n\t\t\texpectOutput: `\"2025-11-18T03:02:03.456789012Z\"`,\n\t\t\texpectErr:    \"\",\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\toutput, err := c.input.MarshalJSON()\n\t\t\trequire.Equal(t, c.expectOutput, string(output))\n\t\t\tif c.expectErr != \"\" {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), c.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTimeUnmarshalJSON(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput      []byte\n\t\texpectTime Time\n\t\texpectErr  string\n\t}{\n\t\t\"zero time\": {\n\t\t\tinput:      []byte(`\"0001-01-01T00:00:00Z\"`),\n\t\t\texpectTime: Time(time.Time{}),\n\t\t\texpectErr:  \"\",\n\t\t},\n\t\t\"valid RFC3339Nano\": {\n\t\t\tinput:      []byte(`\"2025-11-18T12:00:00.123456789Z\"`),\n\t\t\texpectTime: Time(time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC)),\n\t\t\texpectErr:  \"\",\n\t\t},\n\t\t\"non-UTC time\": {\n\t\t\tinput:      []byte(`\"2025-11-17T23:02:03.456789012Z\"`),\n\t\t\texpectTime: Time(time.Date(2025, 11, 17, 23, 2, 3, 456789012, time.UTC)),\n\t\t\texpectErr:  \"\",\n\t\t},\n\t\t\"invalid format\": {\n\t\t\tinput:      []byte(`\"not-a-time\"`),\n\t\t\texpectTime: Time{},\n\t\t\texpectErr:  \"cannot parse\",\n\t\t},\n\t\t\"empty string\": {\n\t\t\tinput:      []byte(`\"\"`),\n\t\t\texpectTime: Time{},\n\t\t\texpectErr:  \"cannot parse\",\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tvar tim Time\n\t\t\terr := json.Unmarshal(c.input, &tim)\n\t\t\trequire.Equal(t, c.expectTime, tim)\n\t\t\tif c.expectErr != \"\" {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), c.expectErr)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// TestParseTimeSuccess checks valid parsing scenarios.\nfunc TestParseTimeSuccess(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput    string\n\t\texpected time.Time\n\t}{\n\t\t\"with nanos\": {\n\t\t\tinput:    \"2025-11-18T12:00:00.123456789Z\",\n\t\t\texpected: time.Date(2025, 11, 18, 12, 0, 0, 123456789, time.UTC),\n\t\t},\n\t\t\"without nanos\": {\n\t\t\tinput:    \"2000-01-01T00:00:00Z\",\n\t\t\texpected: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t},\n\t\t\"non-UTC time\": {\n\t\t\tinput:    \"2025-11-18T01:02:03.456789012-02:00\",\n\t\t\texpected: time.Date(2025, 11, 18, 3, 2, 3, 456789012, time.UTC),\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tparsed, err := ParseTime(c.input)\n\t\t\trequire.NoError(t, err, \"unexpected error for input %q\", c.input)\n\t\t\tassert.Equal(t, c.expected, parsed)\n\t\t\tassert.Equal(t, c.expected.UnixNano(), parsed.UnixNano(), \"ParseTime mismatch got %v want %v\", parsed, c.expected)\n\t\t\tassert.Equal(t, time.UTC, parsed.Location(), \"ParseTime location mismatch got %v want %v\", parsed.Location(), time.UTC)\n\t\t})\n\t}\n}\n\nfunc TestParseTime_Failures(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput string\n\t}{\n\t\t\"not-a-time\":    {input: \"not-a-time\"},\n\t\t\"empty string\":  {input: \"\"},\n\t\t\"invalid month\": {input: \"2025-13-01T00:00:00Z\"},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t_, err := ParseTime(c.input)\n\t\t\tassert.Error(t, err, \"[%s] expected error for input %q\", name, c.input)\n\t\t})\n\t}\n}\n\nfunc TestFormatTime(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput    time.Time\n\t\texpected string\n\t}{\n\t\t\"non-UTC with nanos\": {\n\t\t\tinput:    time.Date(2025, 11, 17, 21, 2, 3, 456789012, time.FixedZone(\"X\", -2*3600)),\n\t\t\texpected: \"2025-11-17T23:02:03.456789012Z\",\n\t\t},\n\t\t\"non-UTC without nanos\": {\n\t\t\tinput:    time.Date(2000, 1, 1, 0, 0, 0, 0, time.FixedZone(\"Y\", +5*3600)),\n\t\t\texpected: \"1999-12-31T19:00:00Z\",\n\t\t},\n\t\t\"zero time\": {\n\t\t\tinput:    time.Time{},\n\t\t\texpected: \"0001-01-01T00:00:00Z\",\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tstr := FormatTime(c.input)\n\t\t\tassert.Equal(t, c.expected, str)\n\t\t})\n\t}\n}\n\n// TestTime_JSONMarshalling ensures Marshal and Unmarshal retains nanosecond precision.\nfunc TestTime_JSONMarshalling(t *testing.T) {\n\tfor name, c := range map[string]struct {\n\t\tinput time.Time\n\t}{\n\t\t\"zero time\": {\n\t\t\tinput: time.Time{},\n\t\t},\n\t\t\"start of 2025\": {\n\t\t\tinput: time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t},\n\t\t\"now with nanos\": {\n\t\t\tinput: time.Now().UTC(),\n\t\t},\n\t\t\"max nanos\": {\n\t\t\tinput: time.Date(2025, 6, 30, 23, 59, 59, 999999999, time.UTC),\n\t\t},\n\t} {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\twrapped := Time(c.input)\n\t\t\tb, err := json.Marshal(wrapped)\n\t\t\trequire.NoError(t, err, \"marshal error\")\n\n\t\t\tvar back Time\n\t\t\trequire.NoError(t, json.Unmarshal(b, &back), \"unmarshal error\")\n\t\t\tassert.Equal(t, c.input.UTC().UnixNano(), time.Time(back).UnixNano(), \"round-trip lost precision: got %v want %v\", time.Time(back), c.input.UTC())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/client.go",
    "content": "package executorstore\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n)\n\ntype ClientParams struct {\n\tfx.In\n\n\t// If provided, this client will be used instead of creating a new one\n\tClient    *clientv3.Client `optional:\"true\"`\n\tCfg       ETCDConfig\n\tLifecycle fx.Lifecycle\n}\n\ntype ClientOutput struct {\n\tfx.Out\n\n\tClient etcdclient.Client `name:\"executorstore\"`\n}\n\n// NewClient creates a new etcd client or uses the provided one\nfunc NewClient(p ClientParams) (ClientOutput, error) {\n\tif p.Client != nil {\n\t\treturn ClientOutput{Client: p.Client}, nil\n\t}\n\n\tetcdClient, err := clientv3.New(clientv3.Config{\n\t\tEndpoints:   p.Cfg.Endpoints,\n\t\tDialTimeout: p.Cfg.DialTimeout,\n\t})\n\tif err != nil {\n\t\treturn ClientOutput{}, err\n\t}\n\n\tp.Lifecycle.Append(fx.StopHook(etcdClient.Close))\n\treturn ClientOutput{Client: etcdClient}, nil\n}\n\ntype ETCDConfig struct {\n\tEndpoints   []string      `yaml:\"endpoints\"`\n\tDialTimeout time.Duration `yaml:\"dialTimeout\"`\n\tPrefix      string        `yaml:\"prefix\"`\n\tCompression string        `yaml:\"compression\"`\n}\n\n// NewETCDConfig parses ETCDConfig from ShardDistribution config\nfunc NewETCDConfig(cfg config.ShardDistribution) (ETCDConfig, error) {\n\tvar etcdCfg ETCDConfig\n\tif err := cfg.Store.StorageParams.Decode(&etcdCfg); err != nil {\n\t\treturn etcdCfg, fmt.Errorf(\"bad config for etcd store: %w\", err)\n\t}\n\n\treturn etcdCfg, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/client_test.go",
    "content": "package executorstore\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n)\n\nfunc TestNewClient_WithProvidedClient(t *testing.T) {\n\tmockClient := &clientv3.Client{}\n\tparams := ClientParams{\n\t\tClient: mockClient,\n\t}\n\n\toutput, err := NewClient(params)\n\n\trequire.NoError(t, err)\n\trequire.Equal(t, mockClient, output.Client)\n}\n\nfunc TestNewClient_WithInvalidConfig(t *testing.T) {\n\tparams := ClientParams{\n\t\tCfg: ETCDConfig{\n\t\t\tEndpoints:   []string{},\n\t\t\tDialTimeout: 5 * time.Second,\n\t\t},\n\t\tLifecycle: fxtest.NewLifecycle(t),\n\t}\n\n\toutput, err := NewClient(params)\n\n\trequire.Error(t, err)\n\trequire.Nil(t, output.Client)\n}\n\nfunc TestNewETCDConfig_WithValidConfig(t *testing.T) {\n\tetcdCfg := ETCDConfig{\n\t\tEndpoints:   []string{\"127.0.0.1:2379\"},\n\t\tDialTimeout: 5 * time.Second,\n\t\tPrefix:      \"/prefix\",\n\t\tCompression: \"none\",\n\t}\n\n\tencoded, err := yaml.Marshal(etcdCfg)\n\trequire.NoError(t, err)\n\n\tdecoded := &config.YamlNode{}\n\terr = yaml.Unmarshal(encoded, decoded)\n\trequire.NoError(t, err)\n\n\tsdConfig := config.ShardDistribution{\n\t\tStore: config.Store{\n\t\t\tStorageParams: decoded,\n\t\t},\n\t}\n\n\tresultCfg, err := NewETCDConfig(sdConfig)\n\trequire.NoError(t, err)\n\trequire.Equal(t, etcdCfg, resultCfg)\n}\n\nfunc TestNewETCDConfig_WithInvalidConfig(t *testing.T) {\n\tencoded, err := yaml.Marshal(\"\")\n\trequire.NoError(t, err)\n\n\tdecoded := &config.YamlNode{}\n\terr = yaml.Unmarshal(encoded, decoded)\n\trequire.NoError(t, err)\n\n\tsdConfig := config.ShardDistribution{\n\t\tStore: config.Store{\n\t\t\tStorageParams: decoded,\n\t\t},\n\t}\n\n\t_, err = NewETCDConfig(sdConfig)\n\trequire.Error(t, err)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/common/compression.go",
    "content": "package common\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"strings\"\n\n\t\"github.com/golang/snappy\"\n)\n\nvar (\n\t// _snappyHeader is a magic prefix prepended to compressed data to distinguish it from uncompressed data\n\t_snappyHeader = []byte{0xff, 0x06, 0x00, 0x00, 's', 'N', 'a', 'P', 'p', 'Y'}\n)\n\nconst (\n\t// CompressionSnappy indicates snappy compression should be applied\n\tCompressionSnappy = \"snappy\"\n)\n\ntype compressionMode int\n\nconst (\n\tcompressionNone compressionMode = iota\n\tcompressionSnappy\n)\n\n// RecordWriter handles serialization of data for etcd, applying compression when configured.\ntype RecordWriter struct {\n\tmode compressionMode\n}\n\n// NewRecordWriter constructs a RecordWriter based on the configured compression type.\nfunc NewRecordWriter(compressionType string) (*RecordWriter, error) {\n\tswitch strings.ToLower(compressionType) {\n\tcase \"\", \"none\":\n\t\treturn &RecordWriter{mode: compressionNone}, nil\n\tcase CompressionSnappy:\n\t\treturn &RecordWriter{mode: compressionSnappy}, nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"unsupported compression type: %s\", compressionType)\n\t}\n}\n\n// Write serializes data using the configured compression mode.\nfunc (w *RecordWriter) Write(data []byte) ([]byte, error) {\n\tif w.mode == compressionNone {\n\t\treturn data, nil\n\t}\n\n\tvar buf bytes.Buffer\n\twriter := snappy.NewBufferedWriter(&buf)\n\n\tif _, err := writer.Write(data); err != nil {\n\t\treturn nil, err\n\t}\n\tif err := writer.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn buf.Bytes(), nil\n}\n\n// Decompress decodes snappy-compressed data\n// If the snappy header is present, it will successfully decompress it or return an error\n// If the snappy header is absent, it treats data as uncompressed and returns it as-is\nfunc Decompress(data []byte) ([]byte, error) {\n\tif len(data) == 0 {\n\t\treturn data, nil\n\t}\n\n\tif !bytes.HasPrefix(data, _snappyHeader) {\n\t\treturn data, nil\n\t}\n\n\tr := snappy.NewReader(bytes.NewReader(data))\n\tdecompressed, err := io.ReadAll(r)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn decompressed, nil\n}\n\n// DecompressAndUnmarshal decompresses data and unmarshals it into the target\nfunc DecompressAndUnmarshal(data []byte, target interface{}) error {\n\tdecompressed, err := Decompress(data)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"decompress: %w\", err)\n\t}\n\n\tif err := json.Unmarshal(decompressed, target); err != nil {\n\t\treturn fmt.Errorf(\"unmarshal: %w\", err)\n\t}\n\n\treturn nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/common/compression_test.go",
    "content": "package common\n\nimport (\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n)\n\nfunc TestRecordWriter(t *testing.T) {\n\toriginal := []byte(\"test-data\")\n\n\tt.Run(\"no compression when empty\", func(t *testing.T) {\n\t\twriter, err := NewRecordWriter(\"\")\n\t\trequire.NoError(t, err)\n\n\t\tout, err := writer.Write(original)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, original, out)\n\t})\n\n\tt.Run(\"no compression when none\", func(t *testing.T) {\n\t\twriter, err := NewRecordWriter(\"none\")\n\t\trequire.NoError(t, err)\n\n\t\tout, err := writer.Write(original)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, original, out)\n\t})\n\n\tt.Run(\"snappy compression\", func(t *testing.T) {\n\t\twriter, err := NewRecordWriter(CompressionSnappy)\n\t\trequire.NoError(t, err)\n\n\t\tout, err := writer.Write(original)\n\t\trequire.NoError(t, err)\n\t\trequire.NotNil(t, out)\n\t\tassert.NotEqual(t, original, out)\n\n\t\tdecompressed, err := Decompress(out)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, original, decompressed)\n\t})\n\n\tt.Run(\"unsupported compression\", func(t *testing.T) {\n\t\twriter, err := NewRecordWriter(\"unsupported\")\n\t\trequire.Error(t, err)\n\t\tassert.Nil(t, writer)\n\t})\n}\n\nfunc TestDecompress(t *testing.T) {\n\tt.Run(\"Empty data\", func(t *testing.T) {\n\t\tdecompressed, err := Decompress([]byte{})\n\t\trequire.NoError(t, err)\n\t\tassert.Empty(t, decompressed)\n\t})\n\n\tt.Run(\"Nil data\", func(t *testing.T) {\n\t\tdecompressed, err := Decompress(nil)\n\t\trequire.NoError(t, err)\n\t\tassert.Nil(t, decompressed)\n\t})\n\n\tt.Run(\"Uncompressed data\", func(t *testing.T) {\n\t\tuncompressed := []byte(`{\"status\":\"ACTIVE\"}`)\n\n\t\tresult, err := Decompress(uncompressed)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, uncompressed, result, \"Uncompressed data is returned as-is\")\n\n\t\tvar status map[string]string\n\t\terr = json.Unmarshal(result, &status)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, \"ACTIVE\", status[\"status\"])\n\t})\n\n\tt.Run(\"Compressed data\", func(t *testing.T) {\n\t\toriginal := []byte(`{\"status\":\"DRAINING\"}`)\n\t\twriter, err := NewRecordWriter(CompressionSnappy)\n\t\trequire.NoError(t, err)\n\n\t\tcompressed, err := writer.Write(original)\n\t\trequire.NoError(t, err)\n\n\t\tresult, err := Decompress(compressed)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, original, result)\n\n\t\tvar status map[string]string\n\t\terr = json.Unmarshal(result, &status)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, \"DRAINING\", status[\"status\"])\n\t})\n}\n\nfunc TestDecompressAndUnmarshal(t *testing.T) {\n\ttype testData struct {\n\t\tStatus string   `json:\"status\"`\n\t\tShards []string `json:\"shards\"`\n\t}\n\n\tt.Run(\"Uncompressed data\", func(t *testing.T) {\n\t\tdata := []byte(`{\"status\":\"ACTIVE\",\"shards\":[\"shard1\",\"shard2\"]}`)\n\n\t\tvar result testData\n\t\terr := DecompressAndUnmarshal(data, &result)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, \"ACTIVE\", result.Status)\n\t\tassert.Equal(t, []string{\"shard1\", \"shard2\"}, result.Shards)\n\t})\n\n\tt.Run(\"Compressed data\", func(t *testing.T) {\n\t\toriginal := testData{\n\t\t\tStatus: \"DRAINING\",\n\t\t\tShards: []string{\"shard3\", \"shard4\"},\n\t\t}\n\t\toriginalJSON, _ := json.Marshal(original)\n\t\twriter, err := NewRecordWriter(CompressionSnappy)\n\t\trequire.NoError(t, err)\n\n\t\tcompressed, err := writer.Write(originalJSON)\n\t\trequire.NoError(t, err)\n\n\t\tvar result testData\n\t\terr = DecompressAndUnmarshal(compressed, &result)\n\t\trequire.NoError(t, err)\n\t\tassert.Equal(t, original.Status, result.Status)\n\t\tassert.Equal(t, original.Shards, result.Shards)\n\t})\n\n\tt.Run(\"Invalid JSON in uncompressed data\", func(t *testing.T) {\n\t\tinvalidJSON := []byte(`{invalid json}`)\n\n\t\tvar result testData\n\t\terr := DecompressAndUnmarshal(invalidJSON, &result)\n\t\trequire.Error(t, err)\n\t\tassert.Contains(t, err.Error(), \"unmarshal\")\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/etcdstore.go",
    "content": "package executorstore\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=executorstore_mock.go ExecutorStore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdkeys\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdtypes\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore/common\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore/shardcache\"\n)\n\nconst (\n\t// guardOpOverhead is the number of transaction slots consumed by the leadership guard's If condition.\n\tguardOpOverhead = 1\n)\n\nvar (\n\t_executorStatusRunningJSON = fmt.Sprintf(`\"%s\"`, types.ExecutorStatusACTIVE)\n)\n\ntype executorStoreImpl struct {\n\tclient        etcdclient.Client\n\tprefix        string\n\tlogger        log.Logger\n\tshardCache    *shardcache.ShardToExecutorCache\n\ttimeSource    clock.TimeSource\n\trecordWriter  *common.RecordWriter\n\tcfg           *config.Config\n\tmetricsClient metrics.Client\n}\n\n// shardStatisticsUpdate holds the staged statistics for a shard so we can write them\n// to etcd after the main AssignShards transaction commits.\ntype shardStatisticsUpdate struct {\n\texecutorID string\n\tstats      map[string]etcdtypes.ShardStatistics\n}\n\n// ExecutorStoreParams defines the dependencies for the etcd store, for use with fx.\ntype ExecutorStoreParams struct {\n\tfx.In\n\n\tClient        etcdclient.Client `name:\"executorstore\"`\n\tETCDConfig    ETCDConfig\n\tLifecycle     fx.Lifecycle\n\tLogger        log.Logger\n\tTimeSource    clock.TimeSource\n\tConfig        *config.Config\n\tMetricsClient metrics.Client\n}\n\n// NewStore creates a new etcd-backed store and provides it to the fx application.\nfunc NewStore(p ExecutorStoreParams) (store.Store, error) {\n\tshardCache := shardcache.NewShardToExecutorCache(p.ETCDConfig.Prefix, p.Client, p.Logger, p.TimeSource, p.MetricsClient)\n\n\ttimeSource := p.TimeSource\n\tif timeSource == nil {\n\t\ttimeSource = clock.NewRealTimeSource()\n\t}\n\n\trecordWriter, err := common.NewRecordWriter(p.ETCDConfig.Compression)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"create record writer: %w\", err)\n\t}\n\n\tstore := &executorStoreImpl{\n\t\tclient:        p.Client,\n\t\tprefix:        p.ETCDConfig.Prefix,\n\t\tlogger:        p.Logger,\n\t\tshardCache:    shardCache,\n\t\ttimeSource:    timeSource,\n\t\trecordWriter:  recordWriter,\n\t\tcfg:           p.Config,\n\t\tmetricsClient: p.MetricsClient,\n\t}\n\n\tp.Lifecycle.Append(fx.StartStopHook(store.Start, store.Stop))\n\n\treturn store, nil\n}\n\nfunc (s *executorStoreImpl) Start() {\n\ts.shardCache.Start()\n}\n\nfunc (s *executorStoreImpl) Stop() {\n\ts.shardCache.Stop()\n}\n\n// --- HeartbeatStore Implementation ---\n\nfunc (s *executorStoreImpl) RecordHeartbeat(ctx context.Context, namespace, executorID string, request store.HeartbeatState) error {\n\theartbeatKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorHeartbeatKey)\n\tstateKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorStatusKey)\n\treportedShardsKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorReportedShardsKey)\n\n\treportedShardsData, err := json.Marshal(request.ReportedShards)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"marshal reported shards: %w\", err)\n\t}\n\n\tjsonState, err := json.Marshal(request.Status)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"marshal assinged state: %w\", err)\n\t}\n\n\t// Compress data before writing to etcd\n\tcompressedReportedShards, err := s.recordWriter.Write(reportedShardsData)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"compress reported shards: %w\", err)\n\t}\n\n\tcompressedState, err := s.recordWriter.Write(jsonState)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"compress assigned state: %w\", err)\n\t}\n\n\t// Build all operations including metadata\n\tops := []clientv3.Op{\n\t\tclientv3.OpPut(heartbeatKey, etcdtypes.FormatTime(request.LastHeartbeat)),\n\t\tclientv3.OpPut(stateKey, string(compressedState)),\n\t\tclientv3.OpPut(reportedShardsKey, string(compressedReportedShards)),\n\t}\n\tfor key, value := range request.Metadata {\n\t\tmetadataKey := etcdkeys.BuildMetadataKey(s.prefix, namespace, executorID, key)\n\t\tops = append(ops, clientv3.OpPut(metadataKey, value))\n\t}\n\n\t// Atomically update both the timestamp and the state.\n\t_, err = s.client.Txn(ctx).Then(ops...).Commit()\n\n\tif err != nil {\n\t\treturn fmt.Errorf(\"record heartbeat: %w\", err)\n\t}\n\treturn nil\n}\n\n// GetHeartbeat retrieves the last known heartbeat state for a single executor.\nfunc (s *executorStoreImpl) GetHeartbeat(ctx context.Context, namespace string, executorID string) (*store.HeartbeatState, *store.AssignedState, error) {\n\t// The prefix for all keys related to a single executor.\n\texecutorIDPrefix := etcdkeys.BuildExecutorIDPrefix(s.prefix, namespace, executorID)\n\tresp, err := s.client.Get(ctx, executorIDPrefix, clientv3.WithPrefix())\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"etcd get failed for executor %s: %w\", executorID, err)\n\t}\n\n\tif resp.Count == 0 {\n\t\treturn nil, nil, store.ErrExecutorNotFound\n\t}\n\n\theartbeatState := &store.HeartbeatState{}\n\tassignedState := &etcdtypes.AssignedState{}\n\tfound := false\n\n\tfor _, kv := range resp.Kvs {\n\t\tkey := string(kv.Key)\n\t\tvalue := string(kv.Value)\n\t\t_, keyType, keyErr := etcdkeys.ParseExecutorKey(s.prefix, namespace, key)\n\t\tif keyErr != nil {\n\t\t\tcontinue // Ignore unexpected keys\n\t\t}\n\n\t\tfound = true // We found at least one valid key part for the executor.\n\t\tswitch keyType {\n\t\tcase etcdkeys.ExecutorHeartbeatKey:\n\t\t\theartbeatState.LastHeartbeat, err = etcdtypes.ParseTime(value)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, nil, fmt.Errorf(\"parse heartbeat timestamp: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorStatusKey:\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &heartbeatState.Status); err != nil {\n\t\t\t\treturn nil, nil, fmt.Errorf(\"parse executor status: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorReportedShardsKey:\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &heartbeatState.ReportedShards); err != nil {\n\t\t\t\treturn nil, nil, fmt.Errorf(\"parse reported shards: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorAssignedStateKey:\n\t\t\tassignedState.ModRevision = kv.ModRevision\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &assignedState); err != nil {\n\t\t\t\treturn nil, nil, fmt.Errorf(\"parse assigned shards: %w\", err)\n\t\t\t}\n\t\t}\n\t}\n\n\tif !found {\n\t\t// This case is unlikely if resp.Count > 0, but is a good safeguard.\n\t\treturn nil, nil, store.ErrExecutorNotFound\n\t}\n\n\treturn heartbeatState, assignedState.ToAssignedState(), nil\n}\n\n// --- ShardStore Implementation ---\n\nfunc (s *executorStoreImpl) GetState(ctx context.Context, namespace string) (*store.NamespaceState, error) {\n\theartbeatStates := make(map[string]store.HeartbeatState)\n\tassignedStates := make(map[string]store.AssignedState)\n\tshardStats := make(map[string]store.ShardStatistics)\n\n\texecutorPrefix := etcdkeys.BuildExecutorsPrefix(s.prefix, namespace)\n\tresp, err := s.client.Get(ctx, executorPrefix, clientv3.WithPrefix())\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get executor data: %w\", err)\n\t}\n\n\tfor _, kv := range resp.Kvs {\n\t\tkey := string(kv.Key)\n\t\tvalue := string(kv.Value)\n\t\texecutorID, keyType, keyErr := etcdkeys.ParseExecutorKey(s.prefix, namespace, key)\n\t\tif keyErr != nil {\n\t\t\tcontinue\n\t\t}\n\t\theartbeat := heartbeatStates[executorID]\n\t\tassigned := assignedStates[executorID]\n\n\t\tswitch keyType {\n\t\tcase etcdkeys.ExecutorHeartbeatKey:\n\t\t\theartbeat.LastHeartbeat, err = etcdtypes.ParseTime(value)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"parse heartbeat timestamp: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorStatusKey:\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &heartbeat.Status); err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"parse executor status: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorReportedShardsKey:\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &heartbeat.ReportedShards); err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"parse reported shards: %w\", err)\n\t\t\t}\n\t\tcase etcdkeys.ExecutorAssignedStateKey:\n\t\t\tvar assignedRaw etcdtypes.AssignedState\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &assignedRaw); err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"parse assigned shards: %w, %s\", err, value)\n\t\t\t}\n\t\t\tassignedRaw.ModRevision = kv.ModRevision\n\t\t\tassigned = *assignedRaw.ToAssignedState()\n\t\tcase etcdkeys.ExecutorShardStatisticsKey:\n\t\t\t// Only load shard statistics if the load balancing mode requires it\n\t\t\t// TODO: refactor this code to not have a dependency on dynamic config in the store layer\n\t\t\tif s.cfg.GetLoadBalancingMode(namespace) == types.LoadBalancingModeGREEDY {\n\t\t\t\texecutorShardStats := make(map[string]etcdtypes.ShardStatistics)\n\t\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &executorShardStats); err != nil {\n\t\t\t\t\treturn nil, fmt.Errorf(\"parse executor shard statistics: %w, %s\", err, value)\n\t\t\t\t}\n\t\t\t\tfor shardID, stat := range executorShardStats {\n\t\t\t\t\tshardStats[shardID] = *stat.ToShardStatistics()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\theartbeatStates[executorID] = heartbeat\n\t\tassignedStates[executorID] = assigned\n\t}\n\n\treturn &store.NamespaceState{\n\t\tExecutors:        heartbeatStates,\n\t\tShardStats:       shardStats,\n\t\tShardAssignments: assignedStates,\n\t}, nil\n}\n\nfunc (s *executorStoreImpl) SubscribeToAssignmentChanges(ctx context.Context, namespace string) (<-chan map[*store.ShardOwner][]string, func(), error) {\n\treturn s.shardCache.Subscribe(ctx, namespace)\n}\n\nfunc (s *executorStoreImpl) SubscribeToExecutorStatusChanges(ctx context.Context, namespace string) (<-chan int64, error) {\n\trevisionChan := make(chan int64, 1)\n\n\tgo func() {\n\t\tdefer close(revisionChan)\n\n\t\tscope := s.metricsClient.Scope(metrics.ShardDistributorWatchScope).\n\t\t\tTagged(metrics.NamespaceTag(namespace)).\n\t\t\tTagged(metrics.ShardDistributorWatchTypeTag(\"rebalance\"))\n\n\t\twatchChan := s.client.Watch(ctx,\n\t\t\tetcdkeys.BuildExecutorsPrefix(s.prefix, namespace),\n\t\t\tclientv3.WithPrefix(),\n\t\t\tclientv3.WithPrevKV(),\n\t\t)\n\n\t\tfor watchResp := range watchChan {\n\t\t\tif err := watchResp.Err(); err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\t// Track watch metrics\n\t\t\tsw := scope.StartTimer(metrics.ShardDistributorWatchProcessingLatency)\n\t\t\tscope.AddCounter(metrics.ShardDistributorWatchEventsReceived, int64(len(watchResp.Events)))\n\n\t\t\tif !s.hasExecutorStatusChanged(watchResp, namespace) {\n\t\t\t\tsw.Stop()\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\t// If the channel is full, it means the previous revision hasn't been processed yet.\n\t\t\t// Pop the old revision to make room for the new one, ensuring we always have the latest revision.\n\t\t\tselect {\n\t\t\tcase <-revisionChan:\n\t\t\tdefault:\n\t\t\t}\n\n\t\t\trevisionChan <- watchResp.Header.Revision\n\t\t\tsw.Stop()\n\t\t}\n\t}()\n\n\treturn revisionChan, nil\n}\n\n// hasExecutorStatusChanged checks if any of the events in the watch response correspond to changes in executor status.\nfunc (s *executorStoreImpl) hasExecutorStatusChanged(watchResp clientv3.WatchResponse, namespace string) bool {\n\tfor _, event := range watchResp.Events {\n\t\t_, keyType, err := etcdkeys.ParseExecutorKey(s.prefix, namespace, string(event.Kv.Key))\n\t\tif err != nil {\n\t\t\ts.logger.Warn(\"Received watch event with unrecognized key format\", tag.Key(string(event.Kv.Key)))\n\t\t\tcontinue\n\t\t}\n\n\t\t// Only consider changes to the ExecutorStatusKey as significant for triggering a revision update.\n\t\tif keyType != etcdkeys.ExecutorStatusKey {\n\t\t\tcontinue\n\t\t}\n\n\t\t// If the previous value is the same as the new value, it means the status didn't actually change\n\t\tif event.PrevKv != nil && string(event.PrevKv.Value) == string(event.Kv.Value) {\n\t\t\tcontinue\n\t\t}\n\n\t\treturn true\n\t}\n\n\treturn false\n}\n\nfunc (s *executorStoreImpl) AssignShards(ctx context.Context, namespace string, request store.AssignShardsRequest, guard store.GuardFunc) (err error) {\n\tvar ops []clientv3.Op\n\tvar opsElse []clientv3.Op\n\tvar comparisons []clientv3.Cmp\n\tcomparisonMaps := make(map[string]int64)\n\n\t// TODO: Should be extracted to a higher level so that statistics updates are prepared\n\tif s.cfg.GetLoadBalancingMode(namespace) == types.LoadBalancingModeGREEDY {\n\t\tstatsUpdates, errUpdate := s.prepareShardStatisticsUpdates(ctx, namespace, request.NewState.ShardAssignments)\n\t\tif errUpdate != nil {\n\t\t\treturn fmt.Errorf(\"prepare shard statistics: %w\", err)\n\t\t}\n\n\t\tdefer func() {\n\t\t\t// Apply the shard statistics updates after the main transaction commits.\n\t\t\t// Only apply if there was no error in the main transaction.\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\t\t\ts.applyShardStatisticsUpdates(ctx, namespace, statsUpdates)\n\t\t}()\n\t}\n\n\t// 1. Prepare operations to delete stale executors and add comparisons to ensure they haven't been modified\n\tfor executorID, expectedModRevision := range request.ExecutorsToDelete {\n\t\t// Build the assigned state key to check for concurrent modifications\n\t\texecutorStateKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorAssignedStateKey)\n\n\t\t// Add a comparison to ensure the executor's assigned state hasn't changed\n\t\t// This prevents deleting an executor that just received a shard assignment\n\t\tcomparisons = append(comparisons, clientv3.Compare(clientv3.ModRevision(executorStateKey), \"=\", expectedModRevision))\n\t\tcomparisonMaps[executorStateKey] = expectedModRevision\n\n\t\t// Delete all keys for this executor\n\t\texecutorPrefix := etcdkeys.BuildExecutorIDPrefix(s.prefix, namespace, executorID)\n\t\tops = append(ops, clientv3.OpDelete(executorPrefix, clientv3.WithPrefix()))\n\t\topsElse = append(opsElse, clientv3.OpGet(executorStateKey))\n\t}\n\n\t// 2. Prepare operations to update executor states and shard ownership,\n\t// and comparisons to check for concurrent modifications.\n\tfor executorID, state := range request.NewState.ShardAssignments {\n\t\t// Update the executor's assigned_state key.\n\t\texecutorStateKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorAssignedStateKey)\n\t\tvalue, err := json.Marshal(etcdtypes.FromAssignedState(&state))\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"marshal assigned shards for executor %s: %w\", executorID, err)\n\t\t}\n\n\t\tcompressedValue, err := s.recordWriter.Write(value)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"compress assigned shards for executor %s: %w\", executorID, err)\n\t\t}\n\t\tops = append(ops, clientv3.OpPut(executorStateKey, string(compressedValue)))\n\n\t\tcomparisons = append(comparisons, clientv3.Compare(clientv3.ModRevision(executorStateKey), \"=\", state.ModRevision))\n\t\tcomparisonMaps[executorStateKey] = state.ModRevision\n\t\topsElse = append(opsElse, clientv3.OpGet(executorStateKey))\n\t}\n\n\tif len(ops) == 0 {\n\t\treturn nil\n\t}\n\n\t// 3. Apply the guard function to get the base transaction, which may already have an 'If' condition for leadership.\n\tnativeTxn := s.client.Txn(ctx)\n\tguardedTxn, err := guard(nativeTxn)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"apply transaction guard: %w\", err)\n\t}\n\tetcdGuardedTxn, ok := guardedTxn.(clientv3.Txn)\n\tif !ok {\n\t\treturn fmt.Errorf(\"guard function returned invalid transaction type\")\n\t}\n\n\t// 4. Create a nested transaction operation. This allows us to add our own 'If' (comparisons)\n\t// and 'Then' (ops) logic that will only execute if the outer guard's 'If' condition passes.\n\t// we catch what is the state in the else operations so we can identify which part of the condition failed\n\tnestedTxnOp := clientv3.OpTxn(\n\t\tcomparisons, // Our IF conditions\n\t\tops,         // Our THEN operations\n\t\topsElse,     // Our ELSE operations\n\t)\n\n\t// 5. Add the nested transaction to the guarded transaction's THEN clause and commit.\n\tetcdGuardedTxn = etcdGuardedTxn.Then(nestedTxnOp)\n\ttxnResp, err := etcdGuardedTxn.Commit()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"commit shard assignments transaction: %w\", err)\n\t}\n\n\t// 6. Check the results of both the outer and nested transactions.\n\tif !txnResp.Succeeded {\n\t\t// This means the guard's condition (e.g., leadership) failed.\n\t\treturn fmt.Errorf(\"%w: transaction failed, leadership may have changed\", store.ErrVersionConflict)\n\t}\n\n\t// The guard's condition passed. Now check if our nested transaction succeeded.\n\t// Since we only have one Op in our 'Then', we check the first response.\n\tif len(txnResp.Responses) == 0 {\n\t\treturn fmt.Errorf(\"unexpected empty response from transaction\")\n\t}\n\n\tnestedResp := txnResp.Responses[0].GetResponseTxn()\n\tif !nestedResp.Succeeded {\n\t\t// This means our revision checks failed.\n\t\tfailingRevisionString := \"\"\n\t\tfor _, keyValue := range nestedResp.Responses[0].GetResponseRange().Kvs {\n\t\t\texpectedValue, ok := comparisonMaps[string(keyValue.Key)]\n\t\t\tif !ok || expectedValue != keyValue.ModRevision {\n\t\t\t\tfailingRevisionString = failingRevisionString + fmt.Sprintf(\"{ key: %s, expected:%v, actual: %v }\", string(keyValue.Key), expectedValue, keyValue.ModRevision)\n\t\t\t}\n\t\t}\n\t\treturn fmt.Errorf(\"%w: transaction failed, a shard may have been concurrently assigned, %v\", store.ErrVersionConflict, failingRevisionString)\n\t}\n\n\treturn nil\n}\n\nfunc (s *executorStoreImpl) AssignShard(ctx context.Context, namespace, shardID, executorID string) error {\n\tassignedState := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorAssignedStateKey)\n\tstatusKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorStatusKey)\n\n\t// Use a read-modify-write loop to handle concurrent updates safely.\n\tfor {\n\t\tvar comparisons []clientv3.Cmp\n\t\tvar ops []clientv3.Op\n\n\t\t// 1. Get the current assigned state of the executor\n\t\tresp, err := s.client.Get(ctx, assignedState)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"get executor assigned state: %w\", err)\n\t\t}\n\n\t\tvar state etcdtypes.AssignedState\n\t\tmodRevision := int64(0) // A revision of 0 means the key doesn't exist yet.\n\n\t\tif len(resp.Kvs) > 0 {\n\t\t\t// If the executor already has shards, load its state.\n\t\t\tkv := resp.Kvs[0]\n\t\t\tmodRevision = kv.ModRevision\n\t\t\tif err := common.DecompressAndUnmarshal(kv.Value, &state); err != nil {\n\t\t\t\treturn fmt.Errorf(\"parse assigned state: %w\", err)\n\t\t\t}\n\t\t} else {\n\t\t\t// If this is the first shard, initialize the state map.\n\t\t\tstate.AssignedShards = make(map[string]*types.ShardAssignment)\n\t\t}\n\n\t\t// 2. Get the executor state.\n\t\tstatusResp, err := s.client.Get(ctx, statusKey)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"get executor status: %w\", err)\n\t\t}\n\t\tif len(statusResp.Kvs) == 0 {\n\t\t\treturn store.ErrExecutorNotFound\n\t\t}\n\t\tstatusValue := string(statusResp.Kvs[0].Value)\n\t\tdecompressedStatusValue, err := common.Decompress(statusResp.Kvs[0].Value)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"decompress executor status: %w\", err)\n\t\t}\n\n\t\tif string(decompressedStatusValue) != _executorStatusRunningJSON {\n\t\t\treturn fmt.Errorf(\"%w: executor status is %s\", store.ErrVersionConflict, statusValue)\n\t\t}\n\t\tstatusModRev := statusResp.Kvs[0].ModRevision\n\n\t\t// 3. Modify the state in memory, adding the new shard if it's not already there.\n\t\tif _, alreadyAssigned := state.AssignedShards[shardID]; !alreadyAssigned {\n\t\t\tstate.AssignedShards[shardID] = &types.ShardAssignment{Status: types.AssignmentStatusREADY}\n\t\t}\n\n\t\t// Update the last updated timestamp.\n\t\tnow := s.timeSource.Now().UTC()\n\t\tstate.LastUpdated = etcdtypes.Time(now)\n\n\t\t// Compress new state value\n\t\tnewStateValue, err := json.Marshal(state)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"marshal new assigned state: %w\", err)\n\t\t}\n\t\tcompressedStateValue, err := s.recordWriter.Write(newStateValue)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"compress new assigned state: %w\", err)\n\t\t}\n\n\t\tops = append(ops, clientv3.OpPut(assignedState, string(compressedStateValue)))\n\n\t\t// 4. Prepare and commit the transaction with four atomic checks.\n\t\t// a) Check that the executor's status ACTIVE has not been changed.\n\t\tcomparisons = append(comparisons, clientv3.Compare(clientv3.ModRevision(statusKey), \"=\", statusModRev))\n\t\t// b) Check that the assigned_state has not been modified concurrently.\n\t\tcomparisons = append(comparisons, clientv3.Compare(clientv3.ModRevision(assignedState), \"=\", modRevision))\n\t\t// c) Check that the cache is up to date.\n\t\tcmp, err := s.shardCache.GetExecutorModRevisionCmp(namespace)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"get executor mod revision cmp: %w\", err)\n\t\t}\n\t\tcomparisons = append(comparisons, cmp...)\n\n\t\t// We check the shard cache to see if the shard is already assigned to an executor.\n\t\tshardOwner, err := s.shardCache.GetShardOwner(ctx, namespace, shardID)\n\t\tif err != nil && !errors.Is(err, store.ErrShardNotFound) {\n\t\t\treturn fmt.Errorf(\"checking shard owner: %w\", err)\n\t\t}\n\t\tif err == nil {\n\t\t\treturn &store.ErrShardAlreadyAssigned{ShardID: shardID, AssignedTo: shardOwner.ExecutorID, Metadata: shardOwner.Metadata}\n\t\t}\n\n\t\t// TODO: Extract to higher level so that statistics updates are prepared\n\t\tif s.cfg.GetLoadBalancingMode(namespace) == types.LoadBalancingModeGREEDY {\n\t\t\texecutorStatsKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorShardStatisticsKey)\n\n\t\t\tstatsResp, err := s.client.Get(ctx, executorStatsKey)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"get shard statistics: %w\", err)\n\t\t\t}\n\n\t\t\texecutorShardStats := make(map[string]etcdtypes.ShardStatistics)\n\t\t\tif len(statsResp.Kvs) > 0 {\n\t\t\t\tif err := common.DecompressAndUnmarshal(statsResp.Kvs[0].Value, &executorShardStats); err != nil {\n\t\t\t\t\treturn fmt.Errorf(\"parse shard statistics: %w\", err)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tshardStats, ok := executorShardStats[shardID]\n\t\t\tif !ok {\n\t\t\t\tshardStats.SmoothedLoad = 0\n\t\t\t\tshardStats.LastUpdateTime = etcdtypes.Time(now)\n\t\t\t}\n\t\t\tshardStats.LastMoveTime = etcdtypes.Time(now)\n\t\t\texecutorShardStats[shardID] = shardStats\n\n\t\t\tnewStatsValue, err := json.Marshal(executorShardStats)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"marshal new shard statistics: %w\", err)\n\t\t\t}\n\t\t\tcompressedStatsValue, err := s.recordWriter.Write(newStatsValue)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"compress new shard statistics: %w\", err)\n\t\t\t}\n\t\t\tops = append(ops, clientv3.OpPut(executorStatsKey, string(compressedStatsValue)))\n\t\t}\n\n\t\ttxnResp, err := s.client.Txn(ctx).\n\t\t\tIf(comparisons...).\n\t\t\tThen(ops...).\n\t\t\tCommit()\n\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"assign shard transaction: %w\", err)\n\t\t}\n\n\t\tif txnResp.Succeeded {\n\t\t\treturn nil\n\t\t}\n\n\t\t// If the transaction failed, another process interfered.\n\t\t// Provide a specific error if the status check failed.\n\t\tcurrentStatusResp, err := s.client.Get(ctx, statusKey)\n\t\tif err != nil || len(currentStatusResp.Kvs) == 0 {\n\t\t\treturn store.ErrExecutorNotFound\n\t\t}\n\t\tdecompressedStatus, err := common.Decompress(currentStatusResp.Kvs[0].Value)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"decompress executor status %w\", err)\n\t\t}\n\t\tif string(decompressedStatus) != _executorStatusRunningJSON {\n\t\t\treturn fmt.Errorf(`%w: executor status is %s\"`, store.ErrVersionConflict, currentStatusResp.Kvs[0].Value)\n\t\t}\n\n\t\ts.logger.Info(\"Assign shard transaction failed due to a conflict. Retrying...\", tag.ShardNamespace(namespace), tag.ShardKey(shardID), tag.ShardExecutor(executorID))\n\t\t// Otherwise, it was a revision mismatch. Loop to retry the operation.\n\t}\n}\n\n// commitGuardedOps commits the given operations in batches to stay within etcd's per-transaction operation limit.\n// Each batch creates a new guarded transaction. If any batch fails, the function returns immediately\n// with the error\nfunc (s *executorStoreImpl) commitGuardedOps(ctx context.Context, ops []clientv3.Op, guard store.GuardFunc) error {\n\tif len(ops) == 0 {\n\t\treturn nil\n\t}\n\tmaxOpsPerTxn := s.cfg.MaxEtcdTxnOps() - guardOpOverhead\n\tif maxOpsPerTxn < 1 {\n\t\tmaxOpsPerTxn = 1\n\t}\n\n\tnumBatches := (len(ops) + maxOpsPerTxn - 1) / maxOpsPerTxn\n\tbatchSize := (len(ops) + numBatches - 1) / numBatches\n\n\tfor i := 0; i < len(ops); i += batchSize {\n\t\tend := i + batchSize\n\t\tif end > len(ops) {\n\t\t\tend = len(ops)\n\t\t}\n\n\t\tnativeTxn := s.client.Txn(ctx)\n\t\tguardedTxn, err := guard(nativeTxn)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"apply transaction guard: %w\", err)\n\t\t}\n\t\tetcdGuardedTxn, ok := guardedTxn.(clientv3.Txn)\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"guard function returned invalid transaction type\")\n\t\t}\n\n\t\tetcdGuardedTxn = etcdGuardedTxn.Then(ops[i:end]...)\n\t\tresp, err := etcdGuardedTxn.Commit()\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"commit batch: %w\", err)\n\t\t}\n\t\tif !resp.Succeeded {\n\t\t\treturn fmt.Errorf(\"transaction failed, leadership may have changed\")\n\t\t}\n\t}\n\treturn nil\n}\n\n// DeleteExecutors deletes the given executors from the store. It does not delete the shards owned by the executors, this\n// should be handled by the namespace processor loop as we want to reassign, not delete the shards.\nfunc (s *executorStoreImpl) DeleteExecutors(ctx context.Context, namespace string, executorIDs []string, guard store.GuardFunc) error {\n\tif len(executorIDs) == 0 {\n\t\treturn nil\n\t}\n\tops := make([]clientv3.Op, 0, len(executorIDs))\n\n\tfor _, executorID := range executorIDs {\n\t\texecutorIDPrefix := etcdkeys.BuildExecutorIDPrefix(s.prefix, namespace, executorID)\n\t\tops = append(ops, clientv3.OpDelete(executorIDPrefix, clientv3.WithPrefix()))\n\t}\n\n\tif err := s.commitGuardedOps(ctx, ops, guard); err != nil {\n\t\treturn fmt.Errorf(\"delete executors: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc (s *executorStoreImpl) DeleteAssignedStates(ctx context.Context, namespace string, executorIDs []string, guard store.GuardFunc) error {\n\tif len(executorIDs) == 0 {\n\t\treturn nil\n\t}\n\tops := make([]clientv3.Op, 0, len(executorIDs))\n\n\tfor _, executorID := range executorIDs {\n\t\texecutorIDPrefix := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorAssignedStateKey)\n\t\tops = append(ops, clientv3.OpDelete(executorIDPrefix, clientv3.WithPrefix()))\n\t}\n\n\tif err := s.commitGuardedOps(ctx, ops, guard); err != nil {\n\t\treturn fmt.Errorf(\"delete assigned states: %w\", err)\n\t}\n\treturn nil\n}\n\n// DeleteShardStats deletes shard statistics for the given shard IDs.\n// If the operation fails (e.g. due to leadership loss), it returns immediately.\n// Partial deletions are acceptable as the periodic cleanup loop will retry remaining keys.\nfunc (s *executorStoreImpl) DeleteShardStats(ctx context.Context, namespace string, shardIDs []string, guard store.GuardFunc) error {\n\tif len(shardIDs) == 0 {\n\t\treturn nil\n\t}\n\n\t// Build a lookup for shard IDs to delete.\n\ttoDelete := make(map[string]struct{}, len(shardIDs))\n\tfor _, shardID := range shardIDs {\n\t\ttoDelete[shardID] = struct{}{}\n\t}\n\n\t// Fetch all statistics at the executor level for this namespace.\n\texecutorPrefix := etcdkeys.BuildExecutorsPrefix(s.prefix, namespace)\n\tresp, err := s.client.Get(ctx, executorPrefix, clientv3.WithPrefix())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"get executor data for shard stats deletion: %w\", err)\n\t}\n\n\tops := make([]clientv3.Op, 0)\n\n\tfor _, kv := range resp.Kvs {\n\t\tkey := string(kv.Key)\n\t\texecutorID, keyType, keyErr := etcdkeys.ParseExecutorKey(s.prefix, namespace, key)\n\t\tif keyErr != nil || keyType != etcdkeys.ExecutorShardStatisticsKey {\n\t\t\tcontinue\n\t\t}\n\n\t\texecutorStats := make(map[string]etcdtypes.ShardStatistics)\n\t\tif err := common.DecompressAndUnmarshal(kv.Value, &executorStats); err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to parse executor shard statistics during cleanup\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tcontinue\n\t\t}\n\n\t\tchanged := false\n\t\tfor shardID := range executorStats {\n\t\t\tif _, ok := toDelete[shardID]; ok {\n\t\t\t\tdelete(executorStats, shardID)\n\t\t\t\tchanged = true\n\t\t\t}\n\t\t}\n\n\t\tif !changed {\n\t\t\tcontinue\n\t\t}\n\n\t\tstatsKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorShardStatisticsKey)\n\t\tif len(executorStats) == 0 {\n\t\t\tops = append(ops, clientv3.OpDelete(statsKey))\n\t\t\tcontinue\n\t\t}\n\n\t\tpayload, err := json.Marshal(executorStats)\n\t\tif err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to marshal executor shard statistics during cleanup\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tcontinue\n\t\t}\n\n\t\tcompressedPayload, err := s.recordWriter.Write(payload)\n\t\tif err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to compress executor shard statistics during cleanup\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tcontinue\n\t\t}\n\n\t\tops = append(ops, clientv3.OpPut(statsKey, string(compressedPayload)))\n\t}\n\n\tif err := s.commitGuardedOps(ctx, ops, guard); err != nil {\n\t\treturn fmt.Errorf(\"delete shard stats: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc (s *executorStoreImpl) GetShardOwner(ctx context.Context, namespace, shardID string) (*store.ShardOwner, error) {\n\treturn s.shardCache.GetShardOwner(ctx, namespace, shardID)\n}\n\nfunc (s *executorStoreImpl) GetExecutor(ctx context.Context, namespace string, executorID string) (*store.ShardOwner, error) {\n\treturn s.shardCache.GetExecutor(ctx, namespace, executorID)\n}\n\nfunc (s *executorStoreImpl) prepareShardStatisticsUpdates(ctx context.Context, namespace string, newAssignments map[string]store.AssignedState) ([]shardStatisticsUpdate, error) {\n\texecutorStatsCache := make(map[string]map[string]etcdtypes.ShardStatistics)\n\tchangedExecutors := make(map[string]struct{})\n\n\tfor executorID, state := range newAssignments {\n\t\tfor shardID := range state.AssignedShards {\n\t\t\tnow := s.timeSource.Now().UTC()\n\n\t\t\toldOwner, err := s.shardCache.GetShardOwner(ctx, namespace, shardID)\n\t\t\tif err != nil && !errors.Is(err, store.ErrShardNotFound) {\n\t\t\t\treturn nil, fmt.Errorf(\"lookup cached shard owner: %w\", err)\n\t\t\t}\n\n\t\t\tif err == nil && oldOwner.ExecutorID == executorID {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tvar stats etcdtypes.ShardStatistics\n\n\t\t\tif err == nil {\n\t\t\t\toldStats, err := s.getOrLoadExecutorShardStatistics(ctx, namespace, oldOwner.ExecutorID, executorStatsCache)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\n\t\t\t\tif existing, ok := oldStats[shardID]; ok {\n\t\t\t\t\tstats = existing\n\t\t\t\t}\n\n\t\t\t\tdelete(oldStats, shardID)\n\t\t\t\tchangedExecutors[oldOwner.ExecutorID] = struct{}{}\n\t\t\t} else {\n\t\t\t\tstats.SmoothedLoad = 0\n\t\t\t\tstats.LastUpdateTime = etcdtypes.Time(now)\n\t\t\t}\n\n\t\t\tstats.LastMoveTime = etcdtypes.Time(now)\n\n\t\t\tnewStats, err := s.getOrLoadExecutorShardStatistics(ctx, namespace, executorID, executorStatsCache)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\n\t\t\tnewStats[shardID] = stats\n\t\t\tchangedExecutors[executorID] = struct{}{}\n\t\t}\n\t}\n\n\tupdates := make([]shardStatisticsUpdate, 0, len(changedExecutors))\n\tfor executorID := range changedExecutors {\n\t\tstats := executorStatsCache[executorID]\n\t\tupdates = append(updates, shardStatisticsUpdate{\n\t\t\texecutorID: executorID,\n\t\t\tstats:      stats,\n\t\t})\n\t}\n\n\treturn updates, nil\n}\n\n// applyShardStatisticsUpdates updates shard statistics.\n// Is intentionally made tolerant of failures since the data is telemetry only.\nfunc (s *executorStoreImpl) applyShardStatisticsUpdates(ctx context.Context, namespace string, updates []shardStatisticsUpdate) {\n\tfor _, update := range updates {\n\t\tstatsKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, update.executorID, etcdkeys.ExecutorShardStatisticsKey)\n\n\t\tif len(update.stats) == 0 {\n\t\t\tif _, err := s.client.Delete(ctx, statsKey); err != nil {\n\t\t\t\ts.logger.Warn(\n\t\t\t\t\t\"failed to delete executor shard statistics\",\n\t\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\t\ttag.ShardExecutor(update.executorID),\n\t\t\t\t\ttag.Error(err),\n\t\t\t\t)\n\t\t\t}\n\t\t\tcontinue\n\t\t}\n\n\t\tpayload, err := json.Marshal(update.stats)\n\t\tif err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to marshal shard statistics after assignment\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(update.executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tcontinue\n\t\t}\n\n\t\tcompressedPayload, err := s.recordWriter.Write(payload)\n\t\tif err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to compress shard statistics after assignment\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(update.executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tcontinue\n\t\t}\n\n\t\tif _, err := s.client.Put(ctx, statsKey, string(compressedPayload)); err != nil {\n\t\t\ts.logger.Warn(\n\t\t\t\t\"failed to update shard statistics\",\n\t\t\t\ttag.ShardNamespace(namespace),\n\t\t\t\ttag.ShardExecutor(update.executorID),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t}\n\t}\n}\n\n// getExecutorShardStatistics returns the shard statistics for the given executor from etcd.\nfunc (s *executorStoreImpl) getExecutorShardStatistics(ctx context.Context, namespace, executorID string) (map[string]etcdtypes.ShardStatistics, error) {\n\tstatsKey := etcdkeys.BuildExecutorKey(s.prefix, namespace, executorID, etcdkeys.ExecutorShardStatisticsKey)\n\tresp, err := s.client.Get(ctx, statsKey)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get executor shard statistics: %w\", err)\n\t}\n\n\tstats := make(map[string]etcdtypes.ShardStatistics)\n\tif len(resp.Kvs) == 0 {\n\t\treturn stats, nil\n\t}\n\n\tif err := common.DecompressAndUnmarshal(resp.Kvs[0].Value, &stats); err != nil {\n\t\treturn nil, fmt.Errorf(\"parse executor shard statistics: %w\", err)\n\t}\n\n\treturn stats, nil\n}\n\n// getOrLoadExecutorShardStatistics returns the shard statistics for the given executor.\n// If the statistics are not cached, it will fetch them from etcd.\nfunc (s *executorStoreImpl) getOrLoadExecutorShardStatistics(\n\tctx context.Context,\n\tnamespace, executorID string,\n\tcache map[string]map[string]etcdtypes.ShardStatistics,\n) (map[string]etcdtypes.ShardStatistics, error) {\n\t// Load from cache if available.\n\tif stats, ok := cache[executorID]; ok {\n\t\treturn stats, nil\n\t}\n\n\t// Otherwise, load from etcd.\n\tstats, err := s.getExecutorShardStatistics(ctx, namespace, executorID)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcache[executorID] = stats\n\treturn stats, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/etcdstore_test.go",
    "content": "package executorstore\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"go.uber.org/mock/gomock\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdkeys\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdtypes\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore/common\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/leaderstore\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/testhelper\"\n)\n\n// TestRecordHeartbeat verifies that an executor's heartbeat is correctly stored.\nfunc TestRecordHeartbeat(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnow := time.Now().UTC()\n\n\texecutorID := \"executor-TestRecordHeartbeat\"\n\treq := store.HeartbeatState{\n\t\tLastHeartbeat: now,\n\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\"shard-TestRecordHeartbeat\": {Status: types.ShardStatusREADY},\n\t\t},\n\t\tMetadata: map[string]string{\n\t\t\t\"key-1\": \"value-1\",\n\t\t\t\"key-2\": \"value-2\",\n\t\t},\n\t}\n\n\terr := executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, req)\n\trequire.NoError(t, err)\n\n\t// Verify directly in etcd\n\theartbeatKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorHeartbeatKey)\n\tstateKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorStatusKey)\n\treportedShardsKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorReportedShardsKey)\n\tmetadataKey1 := etcdkeys.BuildMetadataKey(tc.EtcdPrefix, tc.Namespace, executorID, \"key-1\")\n\tmetadataKey2 := etcdkeys.BuildMetadataKey(tc.EtcdPrefix, tc.Namespace, executorID, \"key-2\")\n\n\tresp, err := tc.Client.Get(ctx, heartbeatKey)\n\trequire.NoError(t, err)\n\tassert.Equal(t, int64(1), resp.Count, \"Heartbeat key should exist\")\n\tassert.Equal(t, etcdtypes.FormatTime(now), string(resp.Kvs[0].Value))\n\n\tresp, err = tc.Client.Get(ctx, stateKey)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), resp.Count, \"State key should exist\")\n\tdecompressedState, err := common.Decompress(resp.Kvs[0].Value)\n\trequire.NoError(t, err)\n\tassert.Equal(t, stringStatus(types.ExecutorStatusACTIVE), string(decompressedState))\n\n\tresp, err = tc.Client.Get(ctx, reportedShardsKey)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), resp.Count, \"Reported shards key should exist\")\n\n\tdecompressedReportedShards, err := common.Decompress(resp.Kvs[0].Value)\n\trequire.NoError(t, err)\n\tvar reportedShards map[string]*types.ShardStatusReport\n\terr = json.Unmarshal(decompressedReportedShards, &reportedShards)\n\trequire.NoError(t, err)\n\trequire.Len(t, reportedShards, 1)\n\tassert.Equal(t, types.ShardStatusREADY, reportedShards[\"shard-TestRecordHeartbeat\"].Status)\n\n\tresp, err = tc.Client.Get(ctx, metadataKey1)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), resp.Count, \"Metadata key 1 should exist\")\n\tassert.Equal(t, \"value-1\", string(resp.Kvs[0].Value))\n\n\tresp, err = tc.Client.Get(ctx, metadataKey2)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), resp.Count, \"Metadata key 2 should exist\")\n\tassert.Equal(t, \"value-2\", string(resp.Kvs[0].Value))\n}\n\nfunc TestRecordHeartbeat_NoCompression(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\n\tvar etcdCfg struct {\n\t\tEndpoints   []string      `yaml:\"endpoints\"`\n\t\tDialTimeout time.Duration `yaml:\"dialTimeout\"`\n\t\tPrefix      string        `yaml:\"prefix\"`\n\t\tCompression string        `yaml:\"compression\"`\n\t}\n\trequire.NoError(t, tc.LeaderCfg.Store.StorageParams.Decode(&etcdCfg))\n\tetcdCfg.Compression = \"none\"\n\n\tencodedCfg, err := yaml.Marshal(etcdCfg)\n\trequire.NoError(t, err)\n\n\tvar yamlNode *config.YamlNode\n\trequire.NoError(t, yaml.Unmarshal(encodedCfg, &yamlNode))\n\ttc.LeaderCfg.Store.StorageParams = yamlNode\n\ttc.LeaderCfg.LeaderStore.StorageParams = yamlNode\n\ttc.Compression = \"none\"\n\n\texecutorStore := createStore(t, tc)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\texecutorID := \"executor-no-compression\"\n\treq := store.HeartbeatState{\n\t\tLastHeartbeat: time.Now().UTC(),\n\t\tStatus:        types.ExecutorStatusACTIVE,\n\t\tReportedShards: map[string]*types.ShardStatusReport{\n\t\t\t\"shard-no-compression\": {Status: types.ShardStatusREADY},\n\t\t},\n\t}\n\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, req))\n\n\tstateKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorStatusKey)\n\trequire.NoError(t, err)\n\treportedShardsKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorReportedShardsKey)\n\trequire.NoError(t, err)\n\n\tstateResp, err := tc.Client.Get(ctx, stateKey)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), stateResp.Count)\n\tstatusJSON, err := json.Marshal(req.Status)\n\trequire.NoError(t, err)\n\tassert.Equal(t, string(statusJSON), string(stateResp.Kvs[0].Value))\n\n\treportedResp, err := tc.Client.Get(ctx, reportedShardsKey)\n\trequire.NoError(t, err)\n\trequire.Equal(t, int64(1), reportedResp.Count)\n\treportedJSON, err := json.Marshal(req.ReportedShards)\n\trequire.NoError(t, err)\n\tassert.Equal(t, string(reportedJSON), string(reportedResp.Kvs[0].Value))\n}\n\nfunc TestGetHeartbeat(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnow := time.Now().UTC()\n\n\texecutorID := \"executor-get\"\n\treq := store.HeartbeatState{\n\t\tStatus:        types.ExecutorStatusDRAINING,\n\t\tLastHeartbeat: now,\n\t}\n\n\t// 1. Record a heartbeat\n\terr := executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, req)\n\trequire.NoError(t, err)\n\n\t// Assign shards to one executor\n\tassignState := map[string]store.AssignedState{\n\t\texecutorID: {\n\t\t\tAssignedShards: map[string]*types.ShardAssignment{\n\t\t\t\t\"shard-1\": {Status: types.AssignmentStatusREADY},\n\t\t\t},\n\t\t},\n\t}\n\trequire.NoError(t, executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{\n\t\tNewState: &store.NamespaceState{\n\t\t\tShardAssignments: assignState,\n\t\t},\n\t}, store.NopGuard()))\n\n\t// 2. Get the heartbeat back\n\thb, assignedFromDB, err := executorStore.GetHeartbeat(ctx, tc.Namespace, executorID)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, hb)\n\n\t// 3. Verify the state\n\tassert.Equal(t, types.ExecutorStatusDRAINING, hb.Status)\n\tassert.Equal(t, now, hb.LastHeartbeat)\n\trequire.NotNil(t, assignedFromDB.AssignedShards)\n\tassert.Equal(t, assignState[executorID].AssignedShards, assignedFromDB.AssignedShards)\n\n\t// 4. Test getting a non-existent executor\n\t_, _, err = executorStore.GetHeartbeat(ctx, tc.Namespace, \"executor-non-existent\")\n\trequire.Error(t, err)\n\tassert.ErrorIs(t, err, store.ErrExecutorNotFound)\n}\n\n// TestGetState verifies that the store can accurately retrieve the state of all executors.\nfunc TestGetState(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\texecutorID1 := \"exec-TestGetState-1\"\n\texecutorID2 := \"exec-TestGetState-2\"\n\tshardID1 := \"shard-1\"\n\tshardID2 := \"shard-2\"\n\n\t// Setup: Record heartbeats and assign shards.\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID1, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID2, store.HeartbeatState{Status: types.ExecutorStatusDRAINING}))\n\trequire.NoError(t, executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{\n\t\tNewState: &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\texecutorID1: {AssignedShards: map[string]*types.ShardAssignment{shardID1: {}}},\n\t\t\t\texecutorID2: {AssignedShards: map[string]*types.ShardAssignment{shardID2: {}}},\n\t\t\t},\n\t\t},\n\t}, store.NopGuard()))\n\n\t// Action: Get the state.\n\tnamespaceState, err := executorStore.GetState(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\n\t// Verification:\n\t// Check Executors\n\trequire.Len(t, namespaceState.Executors, 2, \"Should retrieve two heartbeat states\")\n\tassert.Equal(t, types.ExecutorStatusACTIVE, namespaceState.Executors[executorID1].Status)\n\tassert.Equal(t, types.ExecutorStatusDRAINING, namespaceState.Executors[executorID2].Status)\n\n\t// Check ShardAssignments (from executor records)\n\trequire.Len(t, namespaceState.ShardAssignments, 2, \"Should retrieve two assignment states\")\n\tassert.Contains(t, namespaceState.ShardAssignments[executorID1].AssignedShards, shardID1)\n\tassert.Contains(t, namespaceState.ShardAssignments[executorID2].AssignedShards, shardID2)\n}\n\n// TestAssignShards_WithRevisions tests the optimistic locking logic of AssignShards.\nfunc TestAssignShards_WithRevisions(t *testing.T) {\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\texecutorID1 := \"exec-rev-1\"\n\texecutorID2 := \"exec-rev-2\"\n\n\tt.Run(\"Success\", func(t *testing.T) {\n\t\ttc := testhelper.SetupStoreTestCluster(t)\n\t\texecutorStore := createStore(t, tc)\n\t\trecordHeartbeats(ctx, t, executorStore, tc.Namespace, executorID1, executorID2)\n\n\t\t// Define a new state: assign shard1 to exec1\n\t\tnewState := &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\texecutorID1: {AssignedShards: map[string]*types.ShardAssignment{\"shard-1\": {}}},\n\t\t\t},\n\t\t}\n\n\t\t// Assign - should succeed\n\t\terr := executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: newState}, store.NopGuard())\n\t\trequire.NoError(t, err)\n\n\t\t// Verify the assignment\n\t\tstate, err := executorStore.GetState(ctx, tc.Namespace)\n\t\trequire.NoError(t, err)\n\t\tassert.Contains(t, state.ShardAssignments[executorID1].AssignedShards, \"shard-1\")\n\t})\n\n\tt.Run(\"ConflictOnNewShard\", func(t *testing.T) {\n\t\ttc := testhelper.SetupStoreTestCluster(t)\n\t\texecutorStore := createStore(t, tc)\n\t\trecordHeartbeats(ctx, t, executorStore, tc.Namespace, executorID1, executorID2)\n\n\t\t// Process A defines its desired state: assign shard-new to exec1\n\t\tprocessAState := &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\texecutorID1: {AssignedShards: map[string]*types.ShardAssignment{\"shard-new\": {}}},\n\t\t\t\texecutorID2: {},\n\t\t\t},\n\t\t}\n\n\t\t// Process B defines its desired state: assign shard-new to exec2\n\t\tprocessBState := &store.NamespaceState{\n\t\t\tShardAssignments: map[string]store.AssignedState{\n\t\t\t\texecutorID1: {},\n\t\t\t\texecutorID2: {AssignedShards: map[string]*types.ShardAssignment{\"shard-new\": {}}},\n\t\t\t},\n\t\t}\n\n\t\t// Process A succeeds\n\t\terr := executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: processAState}, store.NopGuard())\n\t\trequire.NoError(t, err)\n\n\t\t// Process B tries to commit, but its revision check for shard-new (rev=0) will fail.\n\t\terr = executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: processBState}, store.NopGuard())\n\t\trequire.Error(t, err)\n\t\tassert.ErrorIs(t, err, store.ErrVersionConflict)\n\t})\n\n\tt.Run(\"ConflictOnExistingShard\", func(t *testing.T) {\n\t\ttc := testhelper.SetupStoreTestCluster(t)\n\t\texecutorStore := createStore(t, tc)\n\t\trecordHeartbeats(ctx, t, executorStore, tc.Namespace, executorID1, executorID2)\n\n\t\tshardID := \"shard-to-move\"\n\t\t// 1. Setup: Assign the shard to executor1\n\t\tsetupState, err := executorStore.GetState(ctx, tc.Namespace)\n\t\trequire.NoError(t, err)\n\t\tsetupState.ShardAssignments = map[string]store.AssignedState{\n\t\t\texecutorID1: {AssignedShards: map[string]*types.ShardAssignment{shardID: {}}},\n\t\t}\n\t\trequire.NoError(t, executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: setupState}, store.NopGuard()))\n\n\t\t// 2. Process A reads the state, intending to move the shard to executor2\n\t\tstateForProcA, err := executorStore.GetState(ctx, tc.Namespace)\n\t\trequire.NoError(t, err)\n\t\tstateForProcA.ShardAssignments = map[string]store.AssignedState{\n\t\t\texecutorID1: {ModRevision: stateForProcA.ShardAssignments[executorID1].ModRevision},\n\t\t\texecutorID2: {AssignedShards: map[string]*types.ShardAssignment{shardID: {}}, ModRevision: 0},\n\t\t}\n\n\t\t// 3. In the meantime, another process makes a different change (e.g., re-assigns to same executor, which changes revision)\n\t\tintermediateState, err := executorStore.GetState(ctx, tc.Namespace)\n\t\trequire.NoError(t, err)\n\t\tintermediateState.ShardAssignments = map[string]store.AssignedState{\n\t\t\texecutorID1: {\n\t\t\t\tAssignedShards: map[string]*types.ShardAssignment{shardID: {}},\n\t\t\t\tModRevision:    intermediateState.ShardAssignments[executorID1].ModRevision,\n\t\t\t},\n\t\t}\n\t\trequire.NoError(t, executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: intermediateState}, store.NopGuard()))\n\n\t\t// 4. Process A tries to commit its change. It will fail because its stored revision for the shard is now stale.\n\t\terr = executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: stateForProcA}, store.NopGuard())\n\t\trequire.Error(t, err)\n\t\tassert.ErrorIs(t, err, store.ErrVersionConflict)\n\t})\n\n\tt.Run(\"NoChanges\", func(t *testing.T) {\n\t\ttc := testhelper.SetupStoreTestCluster(t)\n\t\texecutorStore := createStore(t, tc)\n\t\trecordHeartbeats(ctx, t, executorStore, tc.Namespace, executorID1, executorID2)\n\n\t\t// Get the current state\n\t\tstate, err := executorStore.GetState(ctx, tc.Namespace)\n\t\trequire.NoError(t, err)\n\n\t\t// Call AssignShards with the same assignments\n\t\terr = executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: state}, store.NopGuard())\n\t\trequire.NoError(t, err, \"Assigning with no changes should succeed\")\n\t})\n}\n\n// TestGuardedOperations verifies that AssignShards and DeleteExecutors respect the leader guard.\nfunc TestGuardedOperations(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)\n\tdefer cancel()\n\n\tnamespace := \"test-guarded-ns\"\n\texecutorID := \"exec-to-delete\"\n\n\t// 1. Create two potential leaders\n\t// FIX: Use the correct constructor for the leader elector.\n\telector, err := leaderstore.NewLeaderStore(leaderstore.StoreParams{Client: tc.Client, Cfg: tc.LeaderCfg, Lifecycle: fxtest.NewLifecycle(t)})\n\trequire.NoError(t, err)\n\telection1, err := elector.CreateElection(ctx, namespace)\n\tdefer election1.Cleanup(ctx)\n\tdefer func() { _ = election1.Cleanup(ctx) }()\n\telection2, err := elector.CreateElection(ctx, namespace)\n\tdefer election2.Cleanup(ctx)\n\tdefer func() { _ = election2.Cleanup(ctx) }()\n\n\t// 2. First node becomes leader\n\trequire.NoError(t, election1.Campaign(ctx, \"host-1\"))\n\tvalidGuard := election1.Guard()\n\n\t// 3. Use the valid guard to assign shards - should succeed\n\tassignState := map[string]store.AssignedState{\"exec-1\": {}}\n\terr = executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: &store.NamespaceState{ShardAssignments: assignState}}, validGuard)\n\trequire.NoError(t, err, \"Assigning shards with a valid leader guard should succeed\")\n\n\t// 4. First node resigns, second node becomes leader\n\trequire.NoError(t, election1.Resign(ctx))\n\trequire.NoError(t, election2.Campaign(ctx, \"host-2\"))\n\n\t// 5. Use the now-invalid guard from the first leader - should fail\n\terr = executorStore.AssignShards(ctx, tc.Namespace, store.AssignShardsRequest{NewState: &store.NamespaceState{ShardAssignments: assignState}}, validGuard)\n\trequire.Error(t, err, \"Assigning shards with a stale leader guard should fail\")\n\n\t// 6. Use the NopGuard to delete an executor - should succeed\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\terr = executorStore.DeleteExecutors(ctx, tc.Namespace, []string{executorID}, store.NopGuard())\n\trequire.NoError(t, err, \"Deleting an executor without a guard should succeed\")\n\n\t// Verify deletion\n\tnewState, err := executorStore.GetState(ctx, namespace)\n\trequire.NoError(t, err)\n\t_, ok := newState.ShardAssignments[executorID]\n\trequire.False(t, ok, \"Executor should have been deleted\")\n}\n\n// TestSubscribe verifies that the subscription channel receives notifications for significant changes.\nfunc TestSubscribeToExecutorStatusChanges(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\texecutorID := \"exec-sub\"\n\n\t// Start subscription\n\tsub, err := executorStore.SubscribeToExecutorStatusChanges(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\n\t// Test case #1: Update heartbeat without changing status or reported shards - should NOT trigger notification\n\t{\n\t\t// Manually put a heartbeat update, which is an insignificant change\n\t\theartbeatKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, \"heartbeat\")\n\t\t_, err = tc.Client.Put(ctx, heartbeatKey, \"timestamp\")\n\t\trequire.NoError(t, err)\n\n\t\tselect {\n\t\tcase <-sub:\n\t\t\tt.Fatal(\"Should not receive notification for a heartbeat-only update\")\n\t\tcase <-time.After(100 * time.Millisecond):\n\t\t\t// Expected behavior\n\t\t}\n\t}\n\n\t// Test case #2: Update reported shards without changing status - should NOT trigger notification\n\t{\n\t\t// Manually put a reported shards update, which is an insignificant change\n\t\treportedShardsKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, \"reported_shards\")\n\t\twriter, err := common.NewRecordWriter(tc.Compression)\n\t\trequire.NoError(t, err)\n\t\tcompressedShards, err := writer.Write([]byte(`{\"shard-1\":{\"status\":\"running\"}}`))\n\t\trequire.NoError(t, err)\n\t\t_, err = tc.Client.Put(ctx, reportedShardsKey, string(compressedShards))\n\t\trequire.NoError(t, err)\n\n\t\tselect {\n\t\tcase <-sub:\n\t\t\tt.Fatal(\"Should not receive notification for a reported-shards-only update\")\n\t\tcase <-time.After(100 * time.Millisecond):\n\t\t\t// Expected behavior\n\t\t}\n\t}\n\n\t// Test case #3: Update status without prevKV - should trigger notification\n\t{\n\t\tstatusKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, \"status\")\n\t\t_, err = tc.Client.Put(ctx, statusKey, stringStatus(types.ExecutorStatusDRAINING))\n\t\trequire.NoError(t, err)\n\n\t\tselect {\n\t\tcase rev, ok := <-sub:\n\t\t\trequire.True(t, ok, \"Channel should be open\")\n\t\t\tassert.Greater(t, rev, int64(0), \"Should receive a valid revision for status change\")\n\t\tcase <-time.After(1 * time.Second):\n\t\t\tt.Fatal(\"Should have received a notification for a status change\")\n\t\t}\n\t}\n\n\t// Test case #4: Update status with prevKV but the same value - should NOT trigger notification\n\t{\n\t\tstatusKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, \"status\")\n\t\t_, err = tc.Client.Put(ctx, statusKey, stringStatus(types.ExecutorStatusDRAINING))\n\t\trequire.NoError(t, err)\n\n\t\tselect {\n\t\tcase <-sub:\n\t\t\tt.Fatal(\"Should not receive notification\")\n\t\tcase <-time.After(100 * time.Millisecond):\n\t\t\t// Expected behavior\n\t\t}\n\t}\n\n\t// Test case #5: Update status with prevKV - should trigger notification\n\t{\n\t\tstatusKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, \"status\")\n\t\t_, err = tc.Client.Put(ctx, statusKey, stringStatus(types.ExecutorStatusACTIVE))\n\t\trequire.NoError(t, err)\n\n\t\tselect {\n\t\tcase rev, ok := <-sub:\n\t\t\trequire.True(t, ok, \"Channel should be open\")\n\t\t\tassert.Greater(t, rev, int64(0), \"Should receive a valid revision for status change\")\n\t\tcase <-time.After(1 * time.Second):\n\t\t\tt.Fatal(\"Should have received a notification for a status change\")\n\t\t}\n\t}\n}\n\nfunc TestDeleteExecutors_Empty(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\terr := executorStore.DeleteExecutors(ctx, tc.Namespace, []string{}, store.NopGuard())\n\trequire.NoError(t, err)\n}\n\n// TestDeleteExecutors covers various scenarios for the DeleteExecutors method.\nfunc TestDeleteExecutors(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\t// Setup: Create two active executors for the tests.\n\texecutorID1 := \"executor-to-delete-1\"\n\texecutorID2 := \"executor-to-delete-2\"\n\tsurvivingExecutorID := \"executor-survivor\"\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID1, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID2, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, survivingExecutorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\n\tt.Run(\"SucceedsForNonExistentExecutor\", func(t *testing.T) {\n\t\t// Action: Delete a non-existent executor.\n\t\terr := executorStore.DeleteExecutors(ctx, tc.Namespace, []string{\"non-existent-executor\"}, store.NopGuard())\n\t\t// Verification: Should not return an error.\n\t\trequire.NoError(t, err)\n\t})\n\n\tt.Run(\"DeletesMultipleExecutors\", func(t *testing.T) {\n\t\t// Setup: Create and assign shards to multiple executors.\n\t\texecToDelete1 := \"multi-delete-1\"\n\t\texecToDelete2 := \"multi-delete-2\"\n\t\texecToKeep := \"multi-keep-1\"\n\t\tshardOfDeletedExecutor1 := \"multi-shard-1\"\n\t\tshardOfDeletedExecutor2 := \"multi-shard-2\"\n\t\tshardOfSurvivingExecutor := \"multi-shard-keep\"\n\n\t\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, execToDelete1, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\t\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, execToDelete2, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\t\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, execToKeep, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\n\t\trequire.NoError(t, executorStore.AssignShard(ctx, tc.Namespace, shardOfDeletedExecutor1, execToDelete1))\n\t\trequire.NoError(t, executorStore.AssignShard(ctx, tc.Namespace, shardOfDeletedExecutor2, execToDelete2))\n\t\trequire.NoError(t, executorStore.AssignShard(ctx, tc.Namespace, shardOfSurvivingExecutor, execToKeep))\n\n\t\t// Action: Delete two of the three executors in one call.\n\t\terr := executorStore.DeleteExecutors(ctx, tc.Namespace, []string{execToDelete1, execToDelete2}, store.NopGuard())\n\t\trequire.NoError(t, err)\n\n\t\t// Verification:\n\t\t// 1. Check deleted executors are gone.\n\t\t_, _, err = executorStore.GetHeartbeat(ctx, tc.Namespace, execToDelete1)\n\t\tassert.ErrorIs(t, err, store.ErrExecutorNotFound, \"Executor 1 should be gone\")\n\n\t\t_, _, err = executorStore.GetHeartbeat(ctx, tc.Namespace, execToDelete2)\n\t\tassert.ErrorIs(t, err, store.ErrExecutorNotFound, \"Executor 2 should be gone\")\n\n\t\t// 2. Check that the surviving executor remain.\n\t\t_, _, err = executorStore.GetHeartbeat(ctx, tc.Namespace, execToKeep)\n\t\tassert.NoError(t, err, \"Surviving executor should still exist\")\n\t})\n}\n\nfunc TestParseExecutorKey_Errors(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\n\t_, _, err := etcdkeys.ParseExecutorKey(tc.EtcdPrefix, tc.Namespace, \"/wrong/prefix/exec/heartbeat\")\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"does not have expected prefix\")\n\n\tkey := etcdkeys.BuildExecutorsPrefix(tc.EtcdPrefix, tc.Namespace) + \"too/many/parts\"\n\t_, _, err = etcdkeys.ParseExecutorKey(tc.EtcdPrefix, tc.Namespace, key)\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"unexpected key format\")\n}\n\n// TestAssignAndGetShardOwnerRoundtrip verifies the successful assignment and retrieval of a shard owner.\nfunc TestAssignAndGetShardOwnerRoundtrip(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc).(*executorStoreImpl)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnow := executorStore.timeSource.Now().UTC()\n\texecutorID := \"executor-roundtrip\"\n\tshardID := \"shard-roundtrip\"\n\n\t// Setup: Create an active executor.\n\terr := executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE})\n\trequire.NoError(t, err)\n\n\t// 1. Assign a shard to the active executor.\n\terr = executorStore.AssignShard(ctx, tc.Namespace, shardID, executorID)\n\trequire.NoError(t, err, \"Should successfully assign shard to an active executor\")\n\n\t// 2. Get the owner and verify it's the correct executor.\n\tstate, err := executorStore.GetState(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\tassert.Contains(t, state.ShardAssignments[executorID].AssignedShards, shardID)\n\tassert.Equal(t, now, state.ShardAssignments[executorID].LastUpdated)\n}\n\n// TestAssignShardErrors tests the various error conditions when assigning a shard.\nfunc TestAssignShardErrors(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tactiveExecutorID := \"executor-active-errors\"\n\tdrainingExecutorID := \"executor-draining-errors\"\n\tshardID1 := \"shard-err-1\"\n\tshardID2 := \"shard-err-2\"\n\n\t// Setup: Create an active and a draining executor, and assign one shard.\n\terr := executorStore.RecordHeartbeat(ctx, tc.Namespace, activeExecutorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE})\n\trequire.NoError(t, err)\n\terr = executorStore.RecordHeartbeat(ctx, tc.Namespace, drainingExecutorID, store.HeartbeatState{Status: types.ExecutorStatusDRAINING})\n\trequire.NoError(t, err)\n\terr = executorStore.AssignShard(ctx, tc.Namespace, shardID1, activeExecutorID)\n\trequire.NoError(t, err)\n\n\t// Case 1: Assigning an already-assigned shard.\n\terr = executorStore.AssignShard(ctx, tc.Namespace, shardID1, activeExecutorID)\n\trequire.Error(t, err, \"Should fail to assign an already-assigned shard\")\n\tvar alreadyAssigned *store.ErrShardAlreadyAssigned\n\trequire.ErrorAs(t, err, &alreadyAssigned)\n\tassert.Equal(t, shardID1, alreadyAssigned.ShardID)\n\tassert.Equal(t, activeExecutorID, alreadyAssigned.AssignedTo)\n\tassert.NotNil(t, alreadyAssigned.Metadata)\n\n\t// Case 2: Assigning to a non-existent executor.\n\terr = executorStore.AssignShard(ctx, tc.Namespace, shardID2, \"non-existent-executor\")\n\trequire.Error(t, err, \"Should fail to assign to a non-existent executor\")\n\tassert.ErrorIs(t, err, store.ErrExecutorNotFound, \"Error should be ErrExecutorNotFound\")\n\n\t// Case 3: Assigning to a non-active (draining) executor.\n\terr = executorStore.AssignShard(ctx, tc.Namespace, shardID2, drainingExecutorID)\n\trequire.Error(t, err, \"Should fail to assign to a draining executor\")\n\tassert.ErrorIs(t, err, store.ErrVersionConflict, \"Error should be ErrVersionConflict for non-active executor\")\n}\n\n// TestShardStatisticsPersistence verifies that shard statistics are preserved on assignment\n// when they already exist, and that GetState exposes them.\nfunc TestShardStatisticsPersistence(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\texecutorStore.(*executorStoreImpl).cfg = &config.Config{\n\t\tLoadBalancingMode: func(namespace string) string {\n\t\t\treturn config.LoadBalancingModeGREEDY\n\t\t},\n\t\tMaxEtcdTxnOps: dynamicproperties.GetIntPropertyFn(128),\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\texecutorID := \"exec-stats\"\n\tshardID := \"shard-stats\"\n\n\t// 1. Setup: ensure executor is ACTIVE\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, tc.Namespace, executorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\n\t// 2. Pre-create shard statistics as if coming from prior history\n\tstats := store.ShardStatistics{SmoothedLoad: 12.5, LastUpdateTime: time.Unix(1234, 0).UTC(), LastMoveTime: time.Unix(5678, 0).UTC()}\n\t// The stats the executor should have after assignment\n\texecutorStats := map[string]etcdtypes.ShardStatistics{\n\t\tshardID: *etcdtypes.FromShardStatistics(&stats),\n\t}\n\tpayload, err := json.Marshal(executorStats)\n\trequire.NoError(t, err)\n\tstatsKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorShardStatisticsKey)\n\t// Write those stats to etcd under the executor (exec-stats) stats key\n\t_, err = tc.Client.Put(ctx, statsKey, string(payload))\n\trequire.NoError(t, err)\n\n\t// 3. Assign the shard via AssignShard (should not clobber existing metrics)\n\trequire.NoError(t, executorStore.AssignShard(ctx, tc.Namespace, shardID, executorID))\n\n\t// 4. Verify via GetState that metrics are preserved and exposed\n\tnsState, err := executorStore.GetState(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\trequire.Contains(t, nsState.ShardStats, shardID)\n\tupdatedStats := nsState.ShardStats[shardID]\n\tassert.Equal(t, stats.SmoothedLoad, updatedStats.SmoothedLoad)\n\tassert.Equal(t, stats.LastUpdateTime, updatedStats.LastUpdateTime)\n\t// This should be greater than the last move time\n\tassert.Greater(t, updatedStats.LastMoveTime, stats.LastMoveTime)\n\n\t// 5. Also ensure assignment recorded correctly\n\trequire.Contains(t, nsState.ShardAssignments[executorID].AssignedShards, shardID)\n}\n\n// TestGetShardStatisticsForMissingShard verifies GetState does not report statistics for unknown shards.\nfunc TestGetShardStatisticsForMissingShard(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\t// No metrics are written; GetState should not contain unknown shard\n\tst, err := executorStore.GetState(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\tassert.NotContains(t, st.ShardStats, \"unknown\")\n}\n\n// TestDeleteShardStatsDeletesAllStats verifies that shard statistics are correctly deleted.\nfunc TestDeleteShardStatsDeletesAllStats(t *testing.T) {\n\ttc := testhelper.SetupStoreTestCluster(t)\n\texecutorStore := createStore(t, tc)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)\n\tdefer cancel()\n\n\ttotalShardStats := 135 // number of stats to add and make stale\n\tshardIDs := make([]string, 0, totalShardStats)\n\texecutorID := \"exec-delete-stats\"\n\n\t// ensure executor exists\n\tctxHeartbeat, cancelHb := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancelHb()\n\trequire.NoError(t, executorStore.RecordHeartbeat(ctxHeartbeat, tc.Namespace, executorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\n\t// Create stale stats\n\texecutorStats := make(map[string]etcdtypes.ShardStatistics)\n\tfor i := 0; i < totalShardStats; i++ {\n\t\tshardID := \"stale-stats-\" + strconv.Itoa(i)\n\t\tshardIDs = append(shardIDs, shardID)\n\n\t\tstats := store.ShardStatistics{\n\t\t\tSmoothedLoad:   float64(i),\n\t\t\tLastUpdateTime: time.Unix(int64(i), 0).UTC(),\n\t\t\tLastMoveTime:   time.Unix(int64(i), 0).UTC(),\n\t\t}\n\t\texecutorStats[shardID] = *etcdtypes.FromShardStatistics(&stats)\n\t}\n\n\tstatsKey := etcdkeys.BuildExecutorKey(tc.EtcdPrefix, tc.Namespace, executorID, etcdkeys.ExecutorShardStatisticsKey)\n\tpayload, err := json.Marshal(executorStats)\n\trequire.NoError(t, err)\n\t_, err = tc.Client.Put(ctx, statsKey, string(payload))\n\trequire.NoError(t, err)\n\n\trequire.NoError(t, executorStore.DeleteShardStats(ctx, tc.Namespace, shardIDs, store.NopGuard()))\n\n\tnsState, err := executorStore.GetState(ctx, tc.Namespace)\n\trequire.NoError(t, err)\n\t// All stats should be deleted\n\tassert.Empty(t, nsState.ShardStats)\n}\n\n// --- Test Setup ---\n\nfunc stringStatus(s types.ExecutorStatus) string {\n\tres, err := json.Marshal(s)\n\tif err != nil {\n\t\tpanic(err)\n\t}\n\treturn string(res)\n}\n\nfunc recordHeartbeats(ctx context.Context, t *testing.T, executorStore store.Store, namespace string, executorIDs ...string) {\n\tt.Helper()\n\n\tfor _, executorID := range executorIDs {\n\t\trequire.NoError(t, executorStore.RecordHeartbeat(ctx, namespace, executorID, store.HeartbeatState{Status: types.ExecutorStatusACTIVE}))\n\t}\n}\n\n// trackingTxn implements clientv3.Txn to record operations per batch for testing.\ntype trackingTxn struct {\n\topsCount int\n\tcommitFn func(numOps int) (*clientv3.TxnResponse, error)\n}\n\nfunc (t *trackingTxn) If(_ ...clientv3.Cmp) clientv3.Txn    { return t }\nfunc (t *trackingTxn) Else(_ ...clientv3.Op) clientv3.Txn   { return t }\nfunc (t *trackingTxn) Then(ops ...clientv3.Op) clientv3.Txn { t.opsCount += len(ops); return t }\nfunc (t *trackingTxn) Commit() (*clientv3.TxnResponse, error) {\n\treturn t.commitFn(t.opsCount)\n}\n\nfunc TestCommitGuardedOps_Batching(t *testing.T) {\n\tconst testMaxTxnOps = 128\n\tconst testMaxOpsPerBatch = testMaxTxnOps - guardOpOverhead\n\n\ttests := []struct {\n\t\tname            string\n\t\tnumOps          int\n\t\texpectedBatches int\n\t}{\n\t\t{\n\t\t\tname:            \"ZeroOps\",\n\t\t\tnumOps:          0,\n\t\t\texpectedBatches: 0,\n\t\t},\n\t\t{\n\t\t\tname:            \"SingleOp\",\n\t\t\tnumOps:          1,\n\t\t\texpectedBatches: 1,\n\t\t},\n\t\t{\n\t\t\tname:            \"BelowLimit\",\n\t\t\tnumOps:          50,\n\t\t\texpectedBatches: 1,\n\t\t},\n\t\t{\n\t\t\tname:            \"ExactlyAtLimit\",\n\t\t\tnumOps:          testMaxOpsPerBatch,\n\t\t\texpectedBatches: 1,\n\t\t},\n\t\t{\n\t\t\tname:            \"OneOverLimit\",\n\t\t\tnumOps:          testMaxOpsPerBatch + 1,\n\t\t\texpectedBatches: 2,\n\t\t},\n\t\t{\n\t\t\tname:            \"ExactlyTwoBatches\",\n\t\t\tnumOps:          testMaxOpsPerBatch * 2,\n\t\t\texpectedBatches: 2,\n\t\t},\n\t\t{\n\t\t\tname:            \"MultipleBatches\",\n\t\t\tnumOps:          testMaxOpsPerBatch*3 + 10,\n\t\t\texpectedBatches: 4,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockClient := etcdclient.NewMockClient(ctrl)\n\n\t\t\tvar batchSizes []int\n\n\t\t\tops := make([]clientv3.Op, tt.numOps)\n\t\t\tfor i := range ops {\n\t\t\t\tops[i] = clientv3.OpDelete(fmt.Sprintf(\"/test/key/%d\", i))\n\t\t\t}\n\n\t\t\tfor range tt.expectedBatches {\n\t\t\t\ttxn := &trackingTxn{\n\t\t\t\t\tcommitFn: func(numOps int) (*clientv3.TxnResponse, error) {\n\t\t\t\t\t\tbatchSizes = append(batchSizes, numOps)\n\t\t\t\t\t\treturn &clientv3.TxnResponse{Succeeded: true}, nil\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tmockClient.EXPECT().Txn(gomock.Any()).Return(txn)\n\t\t\t}\n\n\t\t\ts := &executorStoreImpl{\n\t\t\t\tclient: mockClient,\n\t\t\t\tcfg:    &config.Config{MaxEtcdTxnOps: dynamicproperties.GetIntPropertyFn(testMaxTxnOps)},\n\t\t\t}\n\t\t\terr := s.commitGuardedOps(context.Background(), ops, store.NopGuard())\n\t\t\trequire.NoError(t, err)\n\n\t\t\trequire.Len(t, batchSizes, tt.expectedBatches)\n\n\t\t\ttotalOps := 0\n\t\t\tfor _, size := range batchSizes {\n\t\t\t\tassert.LessOrEqual(t, size, testMaxOpsPerBatch)\n\t\t\t\ttotalOps += size\n\t\t\t}\n\t\t\tassert.Equal(t, tt.numOps, totalOps)\n\t\t})\n\t}\n}\n\nfunc TestCommitGuardedOps_CommitError_StopsEarly(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockClient := etcdclient.NewMockClient(ctrl)\n\n\tcommitCount := 0\n\tmakeTxn := func(err error) *trackingTxn {\n\t\treturn &trackingTxn{\n\t\t\tcommitFn: func(numOps int) (*clientv3.TxnResponse, error) {\n\t\t\t\tcommitCount++\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\treturn &clientv3.TxnResponse{Succeeded: true}, nil\n\t\t\t},\n\t\t}\n\t}\n\n\t// Use a small limit so we get multiple batches with fewer ops\n\tconst testMaxTxnOps = 10\n\tops := make([]clientv3.Op, testMaxTxnOps+5)\n\tfor i := range ops {\n\t\tops[i] = clientv3.OpDelete(fmt.Sprintf(\"/test/key/%d\", i))\n\t}\n\n\tmockClient.EXPECT().Txn(gomock.Any()).Return(makeTxn(fmt.Errorf(\"etcd unavailable\")))\n\n\ts := &executorStoreImpl{\n\t\tclient: mockClient,\n\t\tcfg:    &config.Config{MaxEtcdTxnOps: dynamicproperties.GetIntPropertyFn(testMaxTxnOps)},\n\t}\n\terr := s.commitGuardedOps(context.Background(), ops, store.NopGuard())\n\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"commit batch\")\n\tassert.Equal(t, 1, commitCount, \"should stop after first failing batch\")\n}\n\nfunc TestCommitGuardedOps_LeadershipLost_StopsEarly(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockClient := etcdclient.NewMockClient(ctrl)\n\n\tcommitCount := 0\n\tmakeTxn := func(succeeded bool) *trackingTxn {\n\t\treturn &trackingTxn{\n\t\t\tcommitFn: func(numOps int) (*clientv3.TxnResponse, error) {\n\t\t\t\tcommitCount++\n\t\t\t\treturn &clientv3.TxnResponse{Succeeded: succeeded}, nil\n\t\t\t},\n\t\t}\n\t}\n\n\tconst testMaxTxnOps = 10\n\tops := make([]clientv3.Op, testMaxTxnOps+5)\n\tfor i := range ops {\n\t\tops[i] = clientv3.OpDelete(fmt.Sprintf(\"/test/key/%d\", i))\n\t}\n\n\tmockClient.EXPECT().Txn(gomock.Any()).Return(makeTxn(false))\n\n\ts := &executorStoreImpl{\n\t\tclient: mockClient,\n\t\tcfg:    &config.Config{MaxEtcdTxnOps: dynamicproperties.GetIntPropertyFn(testMaxTxnOps)},\n\t}\n\terr := s.commitGuardedOps(context.Background(), ops, store.NopGuard())\n\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"leadership may have changed\")\n\tassert.Equal(t, 1, commitCount, \"should stop after first leadership failure\")\n}\n\nfunc TestCommitGuardedOps_GuardError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockClient := etcdclient.NewMockClient(ctrl)\n\n\ttxn := &trackingTxn{\n\t\tcommitFn: func(numOps int) (*clientv3.TxnResponse, error) {\n\t\t\tt.Fatal(\"should not reach commit\")\n\t\t\treturn nil, nil\n\t\t},\n\t}\n\tmockClient.EXPECT().Txn(gomock.Any()).Return(txn)\n\n\tfailingGuard := func(txn store.Txn) (store.Txn, error) {\n\t\treturn nil, fmt.Errorf(\"guard failure\")\n\t}\n\n\tops := []clientv3.Op{clientv3.OpDelete(\"/test/key\")}\n\ts := &executorStoreImpl{\n\t\tclient: mockClient,\n\t\tcfg:    &config.Config{MaxEtcdTxnOps: dynamicproperties.GetIntPropertyFn(128)},\n\t}\n\terr := s.commitGuardedOps(context.Background(), ops, failingGuard)\n\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"apply transaction guard\")\n}\n\nfunc createStore(t *testing.T, tc *testhelper.StoreTestCluster) store.Store {\n\tt.Helper()\n\n\tetcdConfig, err := NewETCDConfig(tc.LeaderCfg)\n\trequire.NoError(t, err)\n\n\tstore, err := NewStore(ExecutorStoreParams{\n\t\tClient:        tc.Client,\n\t\tETCDConfig:    etcdConfig,\n\t\tLifecycle:     fxtest.NewLifecycle(t),\n\t\tLogger:        testlogger.New(t),\n\t\tTimeSource:    clock.NewMockedTimeSourceAt(time.Now()),\n\t\tMetricsClient: metrics.NewNoopMetricsClient(),\n\t\tConfig: &config.Config{\n\t\t\tLoadBalancingMode: func(namespace string) string { return config.LoadBalancingModeNAIVE },\n\t\t\tMaxEtcdTxnOps:     dynamicproperties.GetIntPropertyFn(128),\n\t\t},\n\t})\n\trequire.NoError(t, err)\n\treturn store\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/executorstore_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: etcdstore.go\n//\n// Generated by this command:\n//\n//\tmockgen -package executorstore -source etcdstore.go -destination=executorstore_mock.go ExecutorStore\n//\n\n// Package executorstore is a generated GoMock package.\npackage executorstore\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/module.go",
    "content": "package executorstore\n\nimport \"go.uber.org/fx\"\n\nvar Module = fx.Module(\"executorstore\",\n\tfx.Provide(NewStore),\n\tfx.Provide(NewClient),\n\tfx.Provide(NewETCDConfig),\n)\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/namespaceshardcache.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdkeys\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdtypes\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore/common\"\n)\n\nconst (\n\t// RetryInterval for watch failures is between 50ms to 150ms\n\tnamespaceRefreshLoopWatchJitterCoeff   = 0.5\n\tnamespaceRefreshLoopWatchRetryInterval = 100 * time.Millisecond\n)\n\ntype namespaceShardToExecutor struct {\n\tsync.RWMutex\n\n\tshardToExecutor  map[string]*store.ShardOwner   // shardID -> shardOwner\n\tshardOwners      map[string]*store.ShardOwner   // executorID -> shardOwner\n\texecutorState    map[*store.ShardOwner][]string // executor -> shardIDs\n\texecutorRevision map[string]int64\n\tnamespace        string\n\tetcdPrefix       string\n\tstopCh           chan struct{}\n\tlogger           log.Logger\n\tclient           etcdclient.Client\n\ttimeSource       clock.TimeSource\n\tpubSub           *executorStatePubSub\n\tmetricsClient    metrics.Client\n}\n\nfunc newNamespaceShardToExecutor(etcdPrefix, namespace string, client etcdclient.Client, stopCh chan struct{}, logger log.Logger, timeSource clock.TimeSource, metricsClient metrics.Client) (*namespaceShardToExecutor, error) {\n\treturn &namespaceShardToExecutor{\n\t\tshardToExecutor:  make(map[string]*store.ShardOwner),\n\t\texecutorState:    make(map[*store.ShardOwner][]string),\n\t\texecutorRevision: make(map[string]int64),\n\t\tshardOwners:      make(map[string]*store.ShardOwner),\n\t\tnamespace:        namespace,\n\t\tetcdPrefix:       etcdPrefix,\n\t\tstopCh:           stopCh,\n\t\tlogger:           logger.WithTags(tag.ShardNamespace(namespace)),\n\t\tclient:           client,\n\t\ttimeSource:       timeSource,\n\t\tpubSub:           newExecutorStatePubSub(logger, namespace),\n\t\tmetricsClient:    metricsClient,\n\t}, nil\n}\n\nfunc (n *namespaceShardToExecutor) Start(wg *sync.WaitGroup) {\n\twg.Add(1)\n\tgo func() {\n\t\tdefer wg.Done()\n\t\tn.namespaceRefreshLoop()\n\t}()\n}\n\nfunc (n *namespaceShardToExecutor) GetShardOwner(ctx context.Context, shardID string) (*store.ShardOwner, error) {\n\tshardOwner, err := n.getShardOwnerInMap(ctx, &n.shardToExecutor, shardID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get shard owner in map: %w\", err)\n\t}\n\tif shardOwner != nil {\n\t\treturn shardOwner, nil\n\t}\n\n\treturn nil, store.ErrShardNotFound\n}\n\nfunc (n *namespaceShardToExecutor) GetExecutor(ctx context.Context, executorID string) (*store.ShardOwner, error) {\n\tshardOwner, err := n.getShardOwnerInMap(ctx, &n.shardOwners, executorID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get shard owner in map: %w\", err)\n\t}\n\tif shardOwner != nil {\n\t\treturn shardOwner, nil\n\t}\n\n\treturn nil, store.ErrExecutorNotFound\n}\n\nfunc (n *namespaceShardToExecutor) GetExecutorModRevisionCmp() ([]clientv3.Cmp, error) {\n\tn.RLock()\n\tdefer n.RUnlock()\n\tcomparisons := []clientv3.Cmp{}\n\tfor executor, revision := range n.executorRevision {\n\t\texecutorAssignedStateKey := etcdkeys.BuildExecutorKey(n.etcdPrefix, n.namespace, executor, etcdkeys.ExecutorAssignedStateKey)\n\t\tcomparisons = append(comparisons, clientv3.Compare(clientv3.ModRevision(executorAssignedStateKey), \"=\", revision))\n\t}\n\n\treturn comparisons, nil\n}\n\nfunc (n *namespaceShardToExecutor) Subscribe(ctx context.Context) (<-chan map[*store.ShardOwner][]string, func()) {\n\tsubCh, unSub := n.pubSub.subscribe(ctx)\n\n\t// The go routine sends the initial state to the subscriber.\n\tgo func() {\n\t\tinitialState := n.getExecutorState()\n\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\tn.logger.Warn(\"context finished before initial state was sent\")\n\t\tcase subCh <- initialState:\n\t\t\tn.logger.Info(\"initial state sent to subscriber\", tag.Value(initialState))\n\t\t}\n\n\t}()\n\n\treturn subCh, unSub\n}\n\nfunc (n *namespaceShardToExecutor) namespaceRefreshLoop() {\n\ttriggerCh := n.runWatchLoop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-n.stopCh:\n\t\t\tn.logger.Info(\"stop channel closed, exiting namespaceRefreshLoop\")\n\t\t\treturn\n\n\t\tcase _, ok := <-triggerCh:\n\t\t\tif !ok {\n\t\t\t\tn.logger.Info(\"trigger channel closed, exiting namespaceRefreshLoop\")\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif err := n.refresh(context.Background()); err != nil {\n\t\t\t\tn.logger.Error(\"failed to refresh namespace shard to executor\", tag.Error(err))\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (n *namespaceShardToExecutor) runWatchLoop() <-chan struct{} {\n\ttriggerCh := make(chan struct{}, 1)\n\n\tgo func() {\n\t\tdefer close(triggerCh)\n\n\t\tfor {\n\t\t\tif err := n.watch(triggerCh); err != nil {\n\t\t\t\tn.logger.Error(\"error watching in namespaceRefreshLoop, retrying...\", tag.Error(err))\n\t\t\t\tn.timeSource.Sleep(backoff.JitDuration(\n\t\t\t\t\tnamespaceRefreshLoopWatchRetryInterval,\n\t\t\t\t\tnamespaceRefreshLoopWatchJitterCoeff,\n\t\t\t\t))\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tn.logger.Info(\"namespaceRefreshLoop is exiting\")\n\t\t\treturn\n\t\t}\n\t}()\n\n\treturn triggerCh\n}\n\nfunc (n *namespaceShardToExecutor) watch(triggerCh chan<- struct{}) error {\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\tscope := n.metricsClient.Scope(metrics.ShardDistributorWatchScope).\n\t\tTagged(metrics.NamespaceTag(n.namespace)).\n\t\tTagged(metrics.ShardDistributorWatchTypeTag(\"cache_refresh\"))\n\n\twatchChan := n.client.Watch(\n\t\t// WithRequireLeader ensures that the etcd cluster has a leader\n\t\tclientv3.WithRequireLeader(ctx),\n\t\tetcdkeys.BuildExecutorsPrefix(n.etcdPrefix, n.namespace),\n\t\tclientv3.WithPrefix(),\n\t\tclientv3.WithPrevKV(),\n\t)\n\n\tfor {\n\t\tselect {\n\t\tcase <-n.stopCh:\n\t\t\tn.logger.Info(\"stop channel closed, exiting watch loop\")\n\t\t\treturn nil\n\n\t\tcase watchResp, ok := <-watchChan:\n\t\t\tif err := watchResp.Err(); err != nil {\n\t\t\t\treturn fmt.Errorf(\"watch response: %w\", err)\n\t\t\t}\n\t\t\tif !ok {\n\t\t\t\treturn fmt.Errorf(\"watch channel closed\")\n\t\t\t}\n\n\t\t\t// Track watch metrics\n\t\t\tsw := scope.StartTimer(metrics.ShardDistributorWatchProcessingLatency)\n\t\t\tscope.AddCounter(metrics.ShardDistributorWatchEventsReceived, int64(len(watchResp.Events)))\n\n\t\t\t// Only trigger refresh if the change is related to executor assigned state or metadata\n\t\t\tif !n.hasExecutorStateChanged(watchResp) {\n\t\t\t\tsw.Stop()\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tselect {\n\t\t\tcase triggerCh <- struct{}{}:\n\t\t\tdefault:\n\t\t\t\tn.logger.Info(\"Cache is being refreshed, skipping trigger\")\n\t\t\t}\n\t\t\tsw.Stop()\n\t\t}\n\t}\n}\n\n// hasExecutorStateChanged checks if any of the events in the watch response indicate a change to executor assigned state or metadata,\n// and if the value actually changed (not just same value written again)\nfunc (n *namespaceShardToExecutor) hasExecutorStateChanged(watchResp clientv3.WatchResponse) bool {\n\tfor _, event := range watchResp.Events {\n\t\t_, keyType, keyErr := etcdkeys.ParseExecutorKey(n.etcdPrefix, n.namespace, string(event.Kv.Key))\n\t\tif keyErr != nil {\n\t\t\tn.logger.Warn(\"Received watch event with unrecognized key format\", tag.Value(keyErr))\n\t\t\tcontinue\n\t\t}\n\n\t\t// Only refresh on changes to assigned state or metadata, ignore other changes under the executor prefix such as executor heartbeat keys\n\t\tif keyType != etcdkeys.ExecutorAssignedStateKey && keyType != etcdkeys.ExecutorMetadataKey {\n\t\t\tcontinue\n\t\t}\n\n\t\t// Check if value actually changed (skip if same value written again)\n\t\tif event.PrevKv != nil && string(event.Kv.Value) == string(event.PrevKv.Value) {\n\t\t\tcontinue\n\t\t}\n\n\t\treturn true\n\t}\n\n\treturn false\n}\n\nfunc (n *namespaceShardToExecutor) refresh(ctx context.Context) error {\n\terr := n.refreshExecutorState(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"refresh executor state: %w\", err)\n\t}\n\n\tn.pubSub.publish(n.getExecutorState())\n\treturn nil\n}\n\nfunc (n *namespaceShardToExecutor) getExecutorState() map[*store.ShardOwner][]string {\n\tn.RLock()\n\tdefer n.RUnlock()\n\texecutorState := make(map[*store.ShardOwner][]string)\n\tfor executor, shardIDs := range n.executorState {\n\t\texecutorState[executor] = make([]string, len(shardIDs))\n\t\tcopy(executorState[executor], shardIDs)\n\t}\n\n\treturn executorState\n}\n\nfunc (n *namespaceShardToExecutor) refreshExecutorState(ctx context.Context) error {\n\texecutorPrefix := etcdkeys.BuildExecutorsPrefix(n.etcdPrefix, n.namespace)\n\n\tresp, err := n.client.Get(ctx, executorPrefix, clientv3.WithPrefix())\n\tif err != nil {\n\t\treturn fmt.Errorf(\"get executor prefix for namespace %s: %w\", n.namespace, err)\n\t}\n\n\tn.Lock()\n\tdefer n.Unlock()\n\t// Clear the cache, so we don't have any stale data\n\tn.shardToExecutor = make(map[string]*store.ShardOwner)\n\tn.executorState = make(map[*store.ShardOwner][]string)\n\tn.executorRevision = make(map[string]int64)\n\tn.shardOwners = make(map[string]*store.ShardOwner)\n\n\tfor _, kv := range resp.Kvs {\n\t\texecutorID, keyType, keyErr := etcdkeys.ParseExecutorKey(n.etcdPrefix, n.namespace, string(kv.Key))\n\t\tif keyErr != nil {\n\t\t\tcontinue\n\t\t}\n\t\tswitch keyType {\n\t\tcase etcdkeys.ExecutorAssignedStateKey:\n\t\t\tshardOwner := getOrCreateShardOwner(n.shardOwners, executorID)\n\n\t\t\tvar assignedState etcdtypes.AssignedState\n\t\t\terr = common.DecompressAndUnmarshal(kv.Value, &assignedState)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"parse assigned state: %w\", err)\n\t\t\t}\n\n\t\t\t// Build both shard->executor and executor->shards mappings\n\t\t\tshardIDs := make([]string, 0, len(assignedState.AssignedShards))\n\t\t\tfor shardID := range assignedState.AssignedShards {\n\t\t\t\tn.shardToExecutor[shardID] = shardOwner\n\t\t\t\tshardIDs = append(shardIDs, shardID)\n\t\t\t\tn.executorRevision[executorID] = kv.ModRevision\n\t\t\t}\n\t\t\tn.executorState[shardOwner] = shardIDs\n\n\t\tcase etcdkeys.ExecutorMetadataKey:\n\t\t\tshardOwner := getOrCreateShardOwner(n.shardOwners, executorID)\n\t\t\tmetadataKey := strings.TrimPrefix(string(kv.Key), etcdkeys.BuildMetadataKey(n.etcdPrefix, n.namespace, executorID, \"\"))\n\t\t\tshardOwner.Metadata[metadataKey] = string(kv.Value)\n\n\t\tdefault:\n\t\t\tcontinue\n\t\t}\n\t}\n\n\treturn nil\n}\n\n// getOrCreateShardOwner retrieves an existing ShardOwner from the map or creates a new one if it doesn't exist\nfunc getOrCreateShardOwner(shardOwners map[string]*store.ShardOwner, executorID string) *store.ShardOwner {\n\tshardOwner, ok := shardOwners[executorID]\n\tif !ok {\n\t\tshardOwner = &store.ShardOwner{\n\t\t\tExecutorID: executorID,\n\t\t\tMetadata:   make(map[string]string),\n\t\t}\n\t\tshardOwners[executorID] = shardOwner\n\t}\n\treturn shardOwner\n}\n\n// getShardOwnerInMap retrieves a shard owner from the map if it exists, otherwise it refreshes the cache and tries again\n// it takes a pointer to the map. When the cache is refreshed, the map is updated, so we need to pass a pointer to the map\nfunc (n *namespaceShardToExecutor) getShardOwnerInMap(ctx context.Context, m *map[string]*store.ShardOwner, key string) (*store.ShardOwner, error) {\n\tn.RLock()\n\tshardOwner, ok := (*m)[key]\n\tn.RUnlock()\n\tif ok {\n\t\treturn shardOwner, nil\n\t}\n\n\t// Force refresh the cache\n\terr := n.refresh(ctx)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"refresh for namespace %s: %w\", n.namespace, err)\n\t}\n\n\t// Check the cache again after refresh\n\tn.RLock()\n\tshardOwner, ok = (*m)[key]\n\tn.RUnlock()\n\tif ok {\n\t\treturn shardOwner, nil\n\t}\n\treturn nil, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/namespaceshardcache_test.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.etcd.io/etcd/api/v3/mvccpb\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdkeys\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdtypes\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/testhelper\"\n)\n\nfunc TestNamespaceShardToExecutor_Lifecycle(t *testing.T) {\n\ttestCluster := testhelper.SetupStoreTestCluster(t)\n\tlogger := testlogger.New(t)\n\tstopCh := make(chan struct{})\n\tdefer close(stopCh)\n\n\t// Setup: Create executor-1 with shard-1\n\tsetupExecutorWithShards(t, testCluster, \"executor-1\", []string{\"shard-1\"}, map[string]string{\n\t\t\"hostname\": \"executor-1-host\",\n\t\t\"version\":  \"v1.0.0\",\n\t})\n\n\t// Start the cache\n\tnamespaceShardToExecutor, err := newNamespaceShardToExecutor(testCluster.EtcdPrefix, testCluster.Namespace, testCluster.Client, stopCh, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\tassert.NoError(t, err)\n\tnamespaceShardToExecutor.Start(&sync.WaitGroup{})\n\ttime.Sleep(50 * time.Millisecond)\n\n\t// Verify executor-1 owns shard-1 with correct metadata\n\tverifyShardOwner(t, namespaceShardToExecutor, \"shard-1\", \"executor-1\", map[string]string{\n\t\t\"hostname\": \"executor-1-host\",\n\t\t\"version\":  \"v1.0.0\",\n\t})\n\n\t// Check the cache is populated\n\tnamespaceShardToExecutor.RLock()\n\t_, ok := namespaceShardToExecutor.executorRevision[\"executor-1\"]\n\tassert.True(t, ok)\n\tassert.Equal(t, \"executor-1\", namespaceShardToExecutor.shardToExecutor[\"shard-1\"].ExecutorID)\n\tnamespaceShardToExecutor.RUnlock()\n\n\t// Add executor-2 with shard-2 to trigger watch update\n\tsetupExecutorWithShards(t, testCluster, \"executor-2\", []string{\"shard-2\"}, map[string]string{\n\t\t\"hostname\": \"executor-2-host\",\n\t\t\"region\":   \"us-west\",\n\t})\n\ttime.Sleep(100 * time.Millisecond)\n\n\t// Check that executor-2 and shard-2 is in the cache\n\tnamespaceShardToExecutor.RLock()\n\t_, ok = namespaceShardToExecutor.executorRevision[\"executor-2\"]\n\tassert.True(t, ok)\n\tassert.Equal(t, \"executor-2\", namespaceShardToExecutor.shardToExecutor[\"shard-2\"].ExecutorID)\n\tnamespaceShardToExecutor.RUnlock()\n\n\t// Verify executor-2 owns shard-2 with correct metadata\n\tverifyShardOwner(t, namespaceShardToExecutor, \"shard-2\", \"executor-2\", map[string]string{\n\t\t\"hostname\": \"executor-2-host\",\n\t\t\"region\":   \"us-west\",\n\t})\n}\n\nfunc TestNamespaceShardToExecutor_Subscribe(t *testing.T) {\n\ttestCluster := testhelper.SetupStoreTestCluster(t)\n\tlogger := testlogger.New(t)\n\tstopCh := make(chan struct{})\n\tdefer close(stopCh)\n\n\t// Setup: Create executor-1 with shard-1\n\tsetupExecutorWithShards(t, testCluster, \"executor-1\", []string{\"shard-1\"}, map[string]string{\n\t\t\"hostname\": \"executor-1-host\",\n\t\t\"version\":  \"v1.0.0\",\n\t})\n\n\t// Start the cache\n\tnamespaceShardToExecutor, err := newNamespaceShardToExecutor(testCluster.EtcdPrefix, testCluster.Namespace, testCluster.Client, stopCh, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\tassert.NoError(t, err)\n\tnamespaceShardToExecutor.Start(&sync.WaitGroup{})\n\n\t// Refresh the cache to get the initial state\n\terr = namespaceShardToExecutor.refresh(context.Background())\n\trequire.NoError(t, err)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\tsubCh, unSub := namespaceShardToExecutor.Subscribe(ctx)\n\tdefer unSub()\n\n\tvar wg sync.WaitGroup\n\twg.Add(1)\n\n\t// start listener\n\tgo func() {\n\t\tdefer wg.Done()\n\t\t// Check that we get the initial state\n\t\tstate := <-subCh\n\t\tassert.Len(t, state, 1)\n\t\tverifyExecutorInState(t, state, \"executor-1\", []string{\"shard-1\"}, map[string]string{\n\t\t\t\"hostname\": \"executor-1-host\",\n\t\t\t\"version\":  \"v1.0.0\",\n\t\t})\n\n\t\t// Check that we get the updated state\n\t\tstate = <-subCh\n\t\tassert.Len(t, state, 2)\n\t\tverifyExecutorInState(t, state, \"executor-1\", []string{\"shard-1\"}, map[string]string{\n\t\t\t\"hostname\": \"executor-1-host\",\n\t\t\t\"version\":  \"v1.0.0\",\n\t\t})\n\t\tverifyExecutorInState(t, state, \"executor-2\", []string{\"shard-2\"}, map[string]string{\n\t\t\t\"hostname\": \"executor-2-host\",\n\t\t\t\"region\":   \"us-west\",\n\t\t})\n\t}()\n\ttime.Sleep(10 * time.Millisecond)\n\n\t// Add executor-2 with shard-2 to trigger new subscription update\n\tsetupExecutorWithShards(t, testCluster, \"executor-2\", []string{\"shard-2\"}, map[string]string{\n\t\t\"hostname\": \"executor-2-host\",\n\t\t\"region\":   \"us-west\",\n\t})\n\n\twg.Wait()\n}\n\nfunc TestNamespaceShardToExecutor_watch_watchChanErrors(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tlogger := testlogger.New(t)\n\tmockClient := etcdclient.NewMockClient(ctrl)\n\tstopCh := make(chan struct{})\n\ttestPrefix := \"/test-prefix\"\n\ttestNamespace := \"test-namespace\"\n\n\t// Mock the Watch call to return our watch channel\n\twatchChan := make(chan clientv3.WatchResponse)\n\tmockClient.EXPECT().\n\t\tWatch(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(watchChan).\n\t\tAnyTimes()\n\n\te, err := newNamespaceShardToExecutor(testPrefix, testNamespace, mockClient, stopCh, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\trequire.NoError(t, err)\n\n\ttriggerChan := make(chan struct{}, 1)\n\n\t// Test Case #1\n\t// Test received compact revision error from watch channel\n\t{\n\t\tgo func() {\n\t\t\twatchChan <- clientv3.WatchResponse{\n\t\t\t\tCompactRevision: 100,\n\t\t\t}\n\t\t}()\n\n\t\terr = e.watch(triggerChan)\n\t\trequire.Error(t, err)\n\t\tassert.ErrorContains(t, err, \"etcdserver: mvcc: required revision has been compacted\")\n\t}\n\n\t// Test Case #2\n\t// Test closed watch channel\n\t{\n\t\tclose(watchChan)\n\t\terr = e.watch(triggerChan)\n\t\trequire.Error(t, err)\n\t\tassert.ErrorContains(t, err, \"watch channel closed\")\n\t}\n}\n\nfunc TestNamespaceShardToExecutor_watch_triggerChBlocking(t *testing.T) {\n\ttc := setupNamespaceShardToExecutorTestCase(t)\n\tdefer tc.ctrl.Finish()\n\tdefer goleak.VerifyNone(t)\n\n\t// Create a triggerCh with buffer size 1, but never read from it\n\ttriggerChan := make(chan struct{}, 1)\n\n\texecutorKey := etcdkeys.BuildExecutorKey(tc.prefix, tc.namespace, tc.executorID, etcdkeys.ExecutorAssignedStateKey)\n\n\t// Start watch in a goroutine\n\twatchDone := make(chan error, 1)\n\tgo func() {\n\t\twatchDone <- tc.e.watch(triggerChan)\n\t}()\n\n\t// Send many events - the loop should not block even though triggerCh is full\n\tfor i := 0; i < 100; i++ {\n\t\tselect {\n\t\tcase tc.watchChan <- clientv3.WatchResponse{\n\t\t\tEvents: []*clientv3.Event{\n\t\t\t\t{\n\t\t\t\t\tType: clientv3.EventTypePut,\n\t\t\t\t\tKv: &mvccpb.KeyValue{\n\t\t\t\t\t\tKey: []byte(executorKey),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}:\n\t\tcase <-time.After(100 * time.Millisecond):\n\t\t\tt.Fatal(\"watch loop is stuck - could not send event to watchChan\")\n\t\t}\n\t}\n\n\t// Close stopCh to exit the watch loop\n\tclose(tc.stopCh)\n\n\tselect {\n\tcase err := <-watchDone:\n\t\tassert.NoError(t, err)\n\tcase <-time.After(1 * time.Second):\n\t\tt.Fatal(\"watch loop did not exit after stopCh was closed\")\n\t}\n}\n\nfunc TestNamespaceShardToExecutor_namespaceRefreshLoop_notTriggersRefresh_reportedShards(t *testing.T) {\n\ttc := setupNamespaceShardToExecutorTestCase(t)\n\tdefer tc.ctrl.Finish()\n\tdefer goleak.VerifyNone(t)\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\n\tgo func() {\n\t\tdefer wg.Done()\n\t\ttc.e.namespaceRefreshLoop()\n\t}()\n\n\tkey := etcdkeys.BuildExecutorKey(\n\t\ttc.prefix,\n\t\ttc.namespace,\n\t\ttc.executorID,\n\t\tetcdkeys.ExecutorReportedShardsKey,\n\t)\n\n\ttc.watchChan <- clientv3.WatchResponse{\n\t\tEvents: []*clientv3.Event{\n\t\t\t{\n\t\t\t\tType: clientv3.EventTypePut,\n\t\t\t\tKv: &mvccpb.KeyValue{\n\t\t\t\t\tKey: []byte(key),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\t// mock for refresh should not be called, so no need to set expectation on etcdClient.EXPECT().Get()\n\t// use Never with condition that checks shardOwners is still empty to verify that refresh is not triggered\n\trequire.Neverf(t, func() bool {\n\t\ttc.e.RLock()\n\t\tdefer tc.e.RUnlock()\n\n\t\treturn len(tc.e.shardOwners) > 0\n\t}, 100*time.Millisecond, 1*time.Millisecond, \"expected no refresh to be triggered for reported shards change\")\n\n\t// Close stopCh to exit the loop\n\tclose(tc.stopCh)\n\twg.Wait()\n}\n\nfunc TestNamespaceShardToExecutor_namespaceRefreshLoop_notTriggersRefresh_noUpdates(t *testing.T) {\n\ttc := setupNamespaceShardToExecutorTestCase(t)\n\tdefer tc.ctrl.Finish()\n\tdefer goleak.VerifyNone(t)\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\n\tgo func() {\n\t\tdefer wg.Done()\n\t\ttc.e.namespaceRefreshLoop()\n\t}()\n\n\tmetadataValue := \"metadata-value\"\n\tmetadataKey := \"metadata-key\"\n\tkey := etcdkeys.BuildMetadataKey(\n\t\ttc.prefix,\n\t\ttc.namespace,\n\t\ttc.executorID,\n\t\tmetadataKey,\n\t)\n\n\ttc.watchChan <- clientv3.WatchResponse{\n\t\tEvents: []*clientv3.Event{\n\t\t\t{\n\t\t\t\tType: clientv3.EventTypePut,\n\t\t\t\tKv: &mvccpb.KeyValue{\n\t\t\t\t\tKey:   []byte(key),\n\t\t\t\t\tValue: []byte(metadataValue),\n\t\t\t\t},\n\t\t\t\tPrevKv: &mvccpb.KeyValue{\n\t\t\t\t\tKey:   []byte(key),\n\t\t\t\t\tValue: []byte(metadataValue),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\t// mock for refresh should not be called, so no need to set expectation on etcdClient.EXPECT().Get()\n\t// use Never with condition that checks shardOwners is still empty to verify that refresh is not triggered\n\trequire.Neverf(t, func() bool {\n\t\ttc.e.RLock()\n\t\tdefer tc.e.RUnlock()\n\n\t\treturn len(tc.e.shardOwners) > 0\n\t}, 100*time.Millisecond, 1*time.Millisecond, \"expected no refresh to be triggered for the same metadata value\")\n\n\t// Close stopCh to exit the loop\n\tclose(tc.stopCh)\n\twg.Wait()\n}\n\nfunc TestNamespaceShardToExecutor_namespaceRefreshLoop_triggersRefresh(t *testing.T) {\n\ttc := setupNamespaceShardToExecutorTestCase(t)\n\tdefer tc.ctrl.Finish()\n\tdefer goleak.VerifyNone(t)\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\n\tgo func() {\n\t\tdefer wg.Done()\n\t\ttc.e.namespaceRefreshLoop()\n\t}()\n\n\tmetadataValue := \"metadata-value\"\n\tmetadataKey := \"metadata-key\"\n\tkey := etcdkeys.BuildMetadataKey(\n\t\ttc.prefix,\n\t\ttc.namespace,\n\t\ttc.executorID,\n\t\tmetadataKey,\n\t)\n\n\t// Mock Get call for refresh\n\ttc.etcdClient.EXPECT().\n\t\tGet(gomock.Any(), tc.executorPrefix, gomock.Any()).\n\t\tReturn(\n\t\t\t&clientv3.GetResponse{Kvs: []*mvccpb.KeyValue{\n\t\t\t\t{\n\t\t\t\t\tKey:   []byte(key),\n\t\t\t\t\tValue: []byte(metadataValue),\n\t\t\t\t},\n\t\t\t}},\n\t\t\tnil,\n\t\t)\n\n\t// Send a watch event for metadata change which should trigger the refresh\n\ttc.watchChan <- clientv3.WatchResponse{\n\t\tEvents: []*clientv3.Event{\n\t\t\t{\n\t\t\t\tType: clientv3.EventTypePut,\n\t\t\t\tKv: &mvccpb.KeyValue{\n\t\t\t\t\tKey:   []byte(key),\n\t\t\t\t\tValue: []byte(metadataValue),\n\t\t\t\t},\n\t\t\t\tPrevKv: &mvccpb.KeyValue{\n\t\t\t\t\tKey:   []byte(key),\n\t\t\t\t\tValue: []byte(\"previous value\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\t// Wait for the refresh to be triggered and the shard owner to be updated with the new metadata value\n\trequire.Eventually(t, func() bool {\n\t\ttc.e.RLock()\n\t\tdefer tc.e.RUnlock()\n\n\t\tshardOwner, ok := tc.e.shardOwners[tc.executorID]\n\t\tif !ok {\n\t\t\treturn false\n\t\t}\n\n\t\treturn shardOwner.Metadata[metadataKey] == metadataValue\n\t}, time.Second, 1*time.Millisecond, \"expected metadata value to be updated in shard owner after refresh\")\n\n\t// Close stopCh to exit the loop\n\tclose(tc.stopCh)\n\twg.Wait()\n}\n\nfunc TestNamespaceShardToExecutor_namespaceRefreshLoop_watchError(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tlogger := testlogger.New(t)\n\tmockClient := etcdclient.NewMockClient(ctrl)\n\ttimeSource := clock.NewMockedTimeSource()\n\tstopCh := make(chan struct{})\n\ttestPrefix := \"/test-prefix\"\n\ttestNamespace := \"test-namespace\"\n\n\t// mock for first watch call that receives error\n\twatchChanRcvErr := make(chan clientv3.WatchResponse)\n\tmockClient.EXPECT().\n\t\tWatch(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(watchChanRcvErr)\n\n\t// mock for second watch call that receives closed channel\n\twatchChanClosed := make(chan clientv3.WatchResponse)\n\tmockClient.EXPECT().\n\t\tWatch(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(watchChanClosed)\n\n\t// mock for third watch call that will be used when stopCh is closed\n\t// maybe called or not if stopCh is closed before retry interval\n\tmockClient.EXPECT().\n\t\tWatch(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(make(chan clientv3.WatchResponse)).\n\t\tMinTimes(0).\n\t\tMaxTimes(1)\n\n\te, err := newNamespaceShardToExecutor(testPrefix, testNamespace, mockClient, stopCh, logger, timeSource, metrics.NewNoopMetricsClient())\n\trequire.NoError(t, err)\n\n\twg := sync.WaitGroup{}\n\twg.Add(1)\n\tfinished := atomic.Bool{}\n\n\tgo func() {\n\t\tdefer wg.Done()\n\t\te.namespaceRefreshLoop()\n\t\tfinished.Store(true)\n\t}()\n\n\t// Test Case #1: watchChan receives error\n\t{\n\t\t// Sends a response containing compact revision to simulate error\n\t\twatchChanRcvErr <- clientv3.WatchResponse{\n\t\t\tCompactRevision: 100,\n\t\t}\n\n\t\ttimeSource.BlockUntil(1)\n\t\trequire.False(t, finished.Load(), \"namespaceRefreshLoop should not exit on watch error\")\n\t}\n\n\t// Test Case #2: watchChan is closed\n\t{\n\t\ttimeSource.Advance(2 * namespaceRefreshLoopWatchRetryInterval)\n\n\t\t// Sends a response containing compact revision to simulate error\n\t\tclose(watchChanClosed)\n\n\t\ttimeSource.BlockUntil(1)\n\t\trequire.False(t, finished.Load(), \"namespaceRefreshLoop should not exit on watch error\")\n\t}\n\n\t// Test Case #3: stopCh is closed\n\t{\n\t\ttimeSource.Advance(2 * namespaceRefreshLoopWatchRetryInterval)\n\n\t\tclose(stopCh)\n\t\twg.Wait()\n\t\trequire.True(t, finished.Load(), \"namespaceRefreshLoop should exit on watch error\")\n\t}\n}\n\n// setupExecutorWithShards creates an executor in etcd with assigned shards and metadata\nfunc setupExecutorWithShards(t *testing.T, testCluster *testhelper.StoreTestCluster, executorID string, shards []string, metadata map[string]string) {\n\t// Create assigned state\n\tassignedState := &etcdtypes.AssignedState{\n\t\tAssignedShards: make(map[string]*types.ShardAssignment),\n\t}\n\tfor _, shardID := range shards {\n\t\tassignedState.AssignedShards[shardID] = &types.ShardAssignment{Status: types.AssignmentStatusREADY}\n\t}\n\tassignedStateJSON, err := json.Marshal(assignedState)\n\trequire.NoError(t, err)\n\n\tvar operations []clientv3.Op\n\n\texecutorAssignedStateKey := etcdkeys.BuildExecutorKey(testCluster.EtcdPrefix, testCluster.Namespace, executorID, etcdkeys.ExecutorAssignedStateKey)\n\toperations = append(operations, clientv3.OpPut(executorAssignedStateKey, string(assignedStateJSON)))\n\n\t// Add metadata\n\tfor key, value := range metadata {\n\t\tmetadataKey := etcdkeys.BuildMetadataKey(testCluster.EtcdPrefix, testCluster.Namespace, executorID, key)\n\t\toperations = append(operations, clientv3.OpPut(metadataKey, value))\n\t}\n\n\ttxnResp, err := testCluster.Client.Txn(context.Background()).Then(operations...).Commit()\n\trequire.NoError(t, err)\n\trequire.True(t, txnResp.Succeeded)\n}\n\nfunc verifyExecutorInState(t *testing.T, state map[*store.ShardOwner][]string, executorID string, shards []string, metadata map[string]string) {\n\texecutorInState := false\n\tfor executor, executorShards := range state {\n\t\tif executor.ExecutorID == executorID {\n\t\t\tassert.Equal(t, shards, executorShards)\n\t\t\tassert.Equal(t, metadata, executor.Metadata)\n\t\t\texecutorInState = true\n\t\t\tbreak\n\t\t}\n\t}\n\tassert.True(t, executorInState)\n}\n\n// verifyShardOwner checks that a shard has the expected owner and metadata\nfunc verifyShardOwner(t *testing.T, cache *namespaceShardToExecutor, shardID, expectedExecutorID string, expectedMetadata map[string]string) {\n\towner, err := cache.GetShardOwner(context.Background(), shardID)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, owner)\n\tassert.Equal(t, expectedExecutorID, owner.ExecutorID)\n\tfor key, expectedValue := range expectedMetadata {\n\t\tassert.Equal(t, expectedValue, owner.Metadata[key])\n\t}\n\n\texecutor, err := cache.GetExecutor(context.Background(), expectedExecutorID)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, executor)\n\tassert.Equal(t, expectedExecutorID, executor.ExecutorID)\n\tfor key, expectedValue := range expectedMetadata {\n\t\tassert.Equal(t, expectedValue, executor.Metadata[key])\n\t}\n}\n\ntype namespaceShardToExecutorTestCase struct {\n\tctrl       *gomock.Controller\n\te          *namespaceShardToExecutor\n\tetcdClient *etcdclient.MockClient\n\ttimeSource clock.TimeSource\n\n\twatchChan chan clientv3.WatchResponse\n\tstopCh    chan struct{}\n\n\texecutorID string\n\tprefix     string\n\tnamespace  string\n\n\texecutorPrefix string\n}\n\nfunc setupNamespaceShardToExecutorTestCase(t *testing.T) *namespaceShardToExecutorTestCase {\n\tvar tc = new(namespaceShardToExecutorTestCase)\n\n\ttc.ctrl = gomock.NewController(t)\n\tlogger := testlogger.New(t)\n\n\ttc.etcdClient = etcdclient.NewMockClient(tc.ctrl)\n\ttc.stopCh = make(chan struct{})\n\ttc.prefix = \"/test-prefix\"\n\ttc.namespace = \"test-namespace\"\n\ttc.executorID = \"executor-1\"\n\ttc.executorPrefix = etcdkeys.BuildExecutorsPrefix(tc.prefix, tc.namespace)\n\n\t// Mock the Watch call to return our watch channel\n\ttc.watchChan = make(chan clientv3.WatchResponse)\n\ttc.etcdClient.EXPECT().\n\t\tWatch(gomock.Any(), gomock.Any(), gomock.Any()).\n\t\tReturn(tc.watchChan).\n\t\tAnyTimes()\n\n\te, err := newNamespaceShardToExecutor(tc.prefix, tc.namespace, tc.etcdClient, tc.stopCh, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\trequire.NoError(t, err)\n\ttc.e = e\n\treturn tc\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/pubsub.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"sync\"\n\n\t\"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n// executorStatePubSub manages subscriptions to executor state changes\ntype executorStatePubSub struct {\n\tmu          sync.RWMutex\n\tsubscribers map[string]chan<- map[*store.ShardOwner][]string\n\tlogger      log.Logger\n\tnamespace   string\n}\n\nfunc newExecutorStatePubSub(logger log.Logger, namespace string) *executorStatePubSub {\n\treturn &executorStatePubSub{\n\t\tsubscribers: make(map[string]chan<- map[*store.ShardOwner][]string),\n\t\tlogger:      logger,\n\t\tnamespace:   namespace,\n\t}\n}\n\n// Subscribe returns a channel that receives executor state updates.\nfunc (p *executorStatePubSub) subscribe(ctx context.Context) (chan map[*store.ShardOwner][]string, func()) {\n\tch := make(chan map[*store.ShardOwner][]string)\n\tuniqueID := uuid.New().String()\n\n\tp.mu.Lock()\n\tdefer p.mu.Unlock()\n\tp.subscribers[uniqueID] = ch\n\n\tunSub := func() {\n\t\tp.unSubscribe(uniqueID)\n\t}\n\n\treturn ch, unSub\n}\n\nfunc (p *executorStatePubSub) unSubscribe(uniqueID string) {\n\tp.mu.Lock()\n\tdefer p.mu.Unlock()\n\tdelete(p.subscribers, uniqueID)\n}\n\n// Publish sends the state to all subscribers (non-blocking)\nfunc (p *executorStatePubSub) publish(state map[*store.ShardOwner][]string) {\n\tp.mu.RLock()\n\tdefer p.mu.RUnlock()\n\n\tfor _, sub := range p.subscribers {\n\t\tselect {\n\t\tcase sub <- state:\n\t\tdefault:\n\t\t\t// Subscriber is not reading fast enough, skip this update\n\t\t\tp.logger.Warn(\"Subscriber not keeping up with state updates, dropping update\", tag.ShardNamespace(p.namespace))\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/pubsub_test.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nfunc TestExecutorStatePubSub_SubscribeUnsubscribe(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\tpubsub := newExecutorStatePubSub(testlogger.New(t), \"test-ns\")\n\n\tch, unsub := pubsub.subscribe(context.Background())\n\tassert.NotNil(t, ch)\n\tassert.Len(t, pubsub.subscribers, 1)\n\n\tunsub()\n\tassert.Len(t, pubsub.subscribers, 0)\n\n\t// Unsubscribe is idempotent\n\tunsub()\n\tassert.Len(t, pubsub.subscribers, 0)\n}\n\nfunc TestExecutorStatePubSub_Publish(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tt.Run(\"no subscribers doesn't panic\", func(t *testing.T) {\n\t\tpubsub := newExecutorStatePubSub(testlogger.New(t), \"test-ns\")\n\t\trequire.NotPanics(t, func() {\n\t\t\tpubsub.publish(map[*store.ShardOwner][]string{})\n\t\t})\n\t})\n\n\tt.Run(\"multiple subscribers receive updates\", func(t *testing.T) {\n\t\tpubsub := newExecutorStatePubSub(testlogger.New(t), \"test-ns\")\n\t\tch1, unsub1 := pubsub.subscribe(context.Background())\n\t\tch2, unsub2 := pubsub.subscribe(context.Background())\n\t\tdefer unsub1()\n\t\tdefer unsub2()\n\n\t\ttestState := map[*store.ShardOwner][]string{\n\t\t\t{ExecutorID: \"exec-1\", Metadata: map[string]string{}}: {\"shard-1\"},\n\t\t}\n\n\t\tvar wg sync.WaitGroup\n\t\twg.Add(2)\n\t\tgo func() {\n\t\t\tstate := <-ch1\n\t\t\tassert.Equal(t, testState, state)\n\t\t\twg.Done()\n\t\t}()\n\t\tgo func() {\n\t\t\tstate := <-ch2\n\t\t\tassert.Equal(t, testState, state)\n\t\t\twg.Done()\n\t\t}()\n\t\ttime.Sleep(10 * time.Millisecond)\n\n\t\tpubsub.publish(testState)\n\n\t\twg.Wait()\n\t})\n\n\tt.Run(\"non-blocking publish to slow consumer\", func(t *testing.T) {\n\t\tpubsub := newExecutorStatePubSub(testlogger.New(t), \"test-ns\")\n\n\t\t// We create a subscriber that doesn't read from the channel, this should still not block\n\t\t_, slowUnsub := pubsub.subscribe(context.Background())\n\t\tdefer slowUnsub()\n\n\t\ttestState := map[*store.ShardOwner][]string{\n\t\t\t{ExecutorID: \"exec-1\", Metadata: map[string]string{}}: {\"shard-1\"},\n\t\t}\n\n\t\t// We do not read from the slow channel, this should still not block\n\t\tfor range 10 {\n\t\t\tpubsub.publish(testState)\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/shardcache.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/etcdclient\"\n)\n\ntype NamespaceToShards map[string]*namespaceShardToExecutor\ntype ShardToExecutorCache struct {\n\tsync.RWMutex\n\tnamespaceToShards NamespaceToShards\n\ttimeSource        clock.TimeSource\n\tclient            etcdclient.Client\n\tstopC             chan struct{}\n\tlogger            log.Logger\n\tprefix            string\n\twg                sync.WaitGroup\n\tmetricsClient     metrics.Client\n}\n\nfunc NewShardToExecutorCache(\n\tprefix string,\n\tclient etcdclient.Client,\n\tlogger log.Logger,\n\ttimeSource clock.TimeSource,\n\tmetricsClient metrics.Client,\n) *ShardToExecutorCache {\n\tshardCache := &ShardToExecutorCache{\n\t\tnamespaceToShards: make(NamespaceToShards),\n\t\ttimeSource:        timeSource,\n\t\tstopC:             make(chan struct{}),\n\t\tlogger:            logger,\n\t\tprefix:            prefix,\n\t\tclient:            client,\n\t\twg:                sync.WaitGroup{},\n\t\tmetricsClient:     metricsClient,\n\t}\n\n\treturn shardCache\n}\n\nfunc (s *ShardToExecutorCache) Start() {}\n\nfunc (s *ShardToExecutorCache) Stop() {\n\tclose(s.stopC)\n\ts.wg.Wait()\n}\n\nfunc (s *ShardToExecutorCache) GetShardOwner(ctx context.Context, namespace, shardID string) (*store.ShardOwner, error) {\n\tnamespaceShardToExecutor, err := s.getNamespaceShardToExecutor(namespace)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get namespace shard to executor: %w\", err)\n\t}\n\treturn namespaceShardToExecutor.GetShardOwner(ctx, shardID)\n}\n\nfunc (s *ShardToExecutorCache) GetExecutor(ctx context.Context, namespace, executorID string) (*store.ShardOwner, error) {\n\tnamespaceShardToExecutor, err := s.getNamespaceShardToExecutor(namespace)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get namespace shard to executor: %w\", err)\n\t}\n\treturn namespaceShardToExecutor.GetExecutor(ctx, executorID)\n}\n\nfunc (s *ShardToExecutorCache) GetExecutorModRevisionCmp(namespace string) ([]clientv3.Cmp, error) {\n\tnamespaceShardToExecutor, err := s.getNamespaceShardToExecutor(namespace)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get namespace shard to executor: %w\", err)\n\t}\n\treturn namespaceShardToExecutor.GetExecutorModRevisionCmp()\n}\n\nfunc (s *ShardToExecutorCache) Subscribe(ctx context.Context, namespace string) (<-chan map[*store.ShardOwner][]string, func(), error) {\n\tnamespaceShardToExecutor, err := s.getNamespaceShardToExecutor(namespace)\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"get namespace shard to executor: %w\", err)\n\t}\n\n\tch, unSub := namespaceShardToExecutor.Subscribe(ctx)\n\treturn ch, unSub, nil\n}\n\nfunc (s *ShardToExecutorCache) getNamespaceShardToExecutor(namespace string) (*namespaceShardToExecutor, error) {\n\ts.RLock()\n\tnamespaceShardToExecutor, ok := s.namespaceToShards[namespace]\n\ts.RUnlock()\n\n\tif ok {\n\t\treturn namespaceShardToExecutor, nil\n\t}\n\n\ts.Lock()\n\tdefer s.Unlock()\n\n\tnamespaceShardToExecutor, err := newNamespaceShardToExecutor(s.prefix, namespace, s.client, s.stopC, s.logger, s.timeSource, s.metricsClient)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"new namespace shard to executor: %w\", err)\n\t}\n\tnamespaceShardToExecutor.Start(&s.wg)\n\n\ts.namespaceToShards[namespace] = namespaceShardToExecutor\n\treturn namespaceShardToExecutor, nil\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/executorstore/shardcache/shardcache_test.go",
    "content": "package shardcache\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/testhelper\"\n)\n\nfunc TestNewShardToExecutorCache(t *testing.T) {\n\tlogger := testlogger.New(t)\n\n\tclient := &clientv3.Client{}\n\tcache := NewShardToExecutorCache(\"some-prefix\", client, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\n\tassert.NotNil(t, cache)\n\n\tassert.NotNil(t, cache.namespaceToShards)\n\tassert.NotNil(t, cache.stopC)\n\n\tassert.Equal(t, logger, cache.logger)\n\tassert.Equal(t, \"some-prefix\", cache.prefix)\n\tassert.Equal(t, client, cache.client)\n}\n\nfunc TestShardExecutorCacheForwarding(t *testing.T) {\n\ttestCluster := testhelper.SetupStoreTestCluster(t)\n\tlogger := testlogger.New(t)\n\n\t// Setup: Create executor-1 with shard-1 and metadata\n\tsetupExecutorWithShards(t, testCluster, \"executor-1\", []string{\"shard-1\"}, map[string]string{\n\t\t\"datacenter\": \"dc1\",\n\t\t\"rack\":       \"rack-42\",\n\t})\n\n\tcache := NewShardToExecutorCache(testCluster.EtcdPrefix, testCluster.Client, logger, clock.NewRealTimeSource(), metrics.NewNoopMetricsClient())\n\tcache.Start()\n\tdefer cache.Stop()\n\n\t// This will read the namespace from the store as the cache is empty\n\towner, err := cache.GetShardOwner(context.Background(), testCluster.Namespace, \"shard-1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-1\", owner.ExecutorID)\n\tassert.Equal(t, \"dc1\", owner.Metadata[\"datacenter\"])\n\tassert.Equal(t, \"rack-42\", owner.Metadata[\"rack\"])\n\n\t// Check the cache is populated\n\tassert.Greater(t, cache.namespaceToShards[testCluster.Namespace].executorRevision[\"executor-1\"], int64(0))\n\tassert.Equal(t, \"executor-1\", cache.namespaceToShards[testCluster.Namespace].shardToExecutor[\"shard-1\"].ExecutorID)\n\n\t// Check the executor is also cached\n\texecutor, err := cache.GetExecutor(context.Background(), testCluster.Namespace, \"executor-1\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"executor-1\", executor.ExecutorID)\n\tassert.Equal(t, \"dc1\", executor.Metadata[\"datacenter\"])\n\tassert.Equal(t, \"rack-42\", executor.Metadata[\"rack\"])\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/leaderstore/etcdleaderstore.go",
    "content": "package leaderstore\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.etcd.io/etcd/client/v3/concurrency\"\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t// defaultElectionTTL is the default time-to-live for an election session.\n\t// If the leader does not renew its lease within this time, it will lose leadership.\n\tdefaultElectionTTL = 10 * time.Second\n)\n\ntype LeaderStore struct {\n\tclient         *clientv3.Client\n\telectionConfig etcdCfg\n}\n\ntype etcdCfg struct {\n\tEndpoints   []string      `yaml:\"endpoints\"`\n\tDialTimeout time.Duration `yaml:\"dialTimeout\"`\n\tPrefix      string        `yaml:\"prefix\"`\n\tElectionTTL time.Duration `yaml:\"electionTTL\"`\n}\n\n// StoreParams defines the dependencies for the etcd store, for use with fx.\ntype StoreParams struct {\n\tfx.In\n\n\tClient    *clientv3.Client `optional:\"true\"`\n\tCfg       config.ShardDistribution\n\tLifecycle fx.Lifecycle\n\tLogger    log.Logger\n}\n\n// NewLeaderStore creates a new leaderstore backed by ETCD.\nfunc NewLeaderStore(p StoreParams) (store.Elector, error) {\n\tvar err error\n\n\tvar out etcdCfg\n\tif err := p.Cfg.LeaderStore.StorageParams.Decode(&out); err != nil {\n\t\treturn nil, fmt.Errorf(\"bad config: %w\", err)\n\t}\n\n\tetcdClient := p.Client\n\tif etcdClient == nil {\n\t\tetcdClient, err = clientv3.New(clientv3.Config{\n\t\t\tEndpoints:   out.Endpoints,\n\t\t\tDialTimeout: out.DialTimeout,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\tif out.ElectionTTL == 0 {\n\t\tout.ElectionTTL = defaultElectionTTL\n\t}\n\n\tp.Lifecycle.Append(fx.StopHook(etcdClient.Close))\n\n\treturn &LeaderStore{\n\t\tclient:         etcdClient,\n\t\telectionConfig: out,\n\t}, nil\n}\n\nfunc (ls *LeaderStore) CreateElection(ctx context.Context, namespace string) (el store.Election, err error) {\n\t// Create a new session for election\n\tsession, err := concurrency.NewSession(ls.client,\n\t\tconcurrency.WithTTL(int(ls.electionConfig.ElectionTTL.Seconds())),\n\t\tconcurrency.WithContext(ctx))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create session: %w\", err)\n\t}\n\n\tnamespacePrefix := fmt.Sprintf(\"%s/%s\", ls.electionConfig.Prefix, namespace)\n\telectionKey := fmt.Sprintf(\"%s/leader\", namespacePrefix)\n\tetcdElection := concurrency.NewElection(session, electionKey)\n\n\treturn &election{election: etcdElection, session: session, prefix: namespacePrefix}, nil\n}\n\n// election is a wrapper around etcd.concurrency.Election to abstract implementation from etcd types.\ntype election struct {\n\tsession  *concurrency.Session\n\telection *concurrency.Election\n\tprefix   string\n}\n\nfunc (e *election) Resign(ctx context.Context) error {\n\treturn e.election.Resign(ctx)\n}\n\nfunc (e *election) Cleanup(ctx context.Context) error {\n\terr := e.session.Close()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"close session: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc (e *election) Campaign(ctx context.Context, host string) error {\n\treturn e.election.Campaign(ctx, host)\n}\n\nfunc (e *election) Done() <-chan struct{} {\n\treturn e.session.Done()\n}\n\nfunc (e *election) Guard() store.GuardFunc {\n\treturn func(txn store.Txn) (store.Txn, error) {\n\t\t// The guard receives the generic Txn and asserts it to the concrete type it expects.\n\t\tetcdTxn, ok := txn.(clientv3.Txn)\n\t\tif !ok {\n\t\t\treturn nil, fmt.Errorf(\"invalid transaction type for etcd guard: expected clientv3.Txn, got %T\", txn)\n\t\t}\n\t\t// It applies the etcd-specific condition and returns the modified generic Txn.\n\t\treturn etcdTxn.If(clientv3.Compare(clientv3.ModRevision(e.election.Key()), \"=\", e.election.Rev())), nil\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/leaderstore/etcdleaderstore_test.go",
    "content": "package leaderstore\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"go.uber.org/fx/fxtest\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\n// TestCreateElection tests that an election can be created successfully\nfunc TestCreateElection(t *testing.T) {\n\ttc := setupETCDCluster(t)\n\n\ttimeout := 5 * time.Second\n\tctx, cancel := context.WithTimeout(context.Background(), timeout)\n\tdefer cancel()\n\n\tnamespace := \"test-namespace\"\n\telect, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\trequire.NotNil(t, elect)\n\n\t// Clean up\n\terr = elect.Cleanup(ctx)\n\trequire.NoError(t, err)\n}\n\n// TestCampaign tests that a node can campaign for leadership\nfunc TestCampaign(t *testing.T) {\n\ttc := setupETCDCluster(t)\n\n\t// Use a separate context for test operations\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnamespace := \"test-namespace-campaign\"\n\telection, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election.Cleanup(ctx)\n\n\t// Start campaigning for leadership\n\thost := \"test-host-1\"\n\terr = election.Campaign(ctx, host)\n\trequire.NoError(t, err)\n\n\t// Verify leadership was obtained by checking the key in etcd\n\t// First create a client to connect to etcd\n\tclient, err := clientv3.New(clientv3.Config{\n\t\tEndpoints:   tc.endpoints,\n\t\tDialTimeout: 5 * time.Second,\n\t})\n\trequire.NoError(t, err)\n\tdefer client.Close()\n\n\t// Get the key and verify it exists\n\tkey := fmt.Sprintf(\"%s/%s/leader\", tc.storeConfig.Prefix, namespace)\n\tresp, err := client.Get(ctx, key)\n\trequire.NoError(t, err)\n\trequire.NotEqual(t, 0, resp.Count, \"Leader key should exist\")\n}\n\n// TestResign tests resigning leadership\nfunc TestResign(t *testing.T) {\n\ttc := setupETCDCluster(t)\n\n\t// Use a separate context for test operations\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnamespace := \"test-namespace-resign\"\n\telection, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election.Cleanup(ctx)\n\n\t// Start campaigning for leadership\n\thost := \"test-host-1\"\n\terr = election.Campaign(ctx, host)\n\trequire.NoError(t, err)\n\n\t// Resign the leadership\n\terr = election.Resign(ctx)\n\trequire.NoError(t, err)\n\n\t// Verify leadership was resigned by checking that someone else can become leader\n\telection2, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election2.Cleanup(ctx)\n\n\terr = election2.Campaign(ctx, \"host-2\")\n\trequire.NoError(t, err, \"Second host should be able to become leader after first resigned\")\n}\n\n// TestMultipleNodes tests multiple nodes competing for leadership\nfunc TestMultipleNodes(t *testing.T) {\n\ttc := setupETCDCluster(t)\n\n\t// Use a separate context for test operations\n\tctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)\n\tdefer cancel()\n\n\tnamespace := \"test-namespace-multiple\"\n\n\t// Create first election\n\telection1, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election1.Cleanup(ctx)\n\n\t// Create second election\n\telection2, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election2.Cleanup(ctx)\n\n\t// First node campaigns\n\terr = election1.Campaign(ctx, \"host1\")\n\trequire.NoError(t, err)\n\n\t// Second node campaigns - this should block as first node already has leadership\n\t// We'll use a channel to track when it's done and a shorter context to timeout\n\tcampaignDone := make(chan struct{})\n\tcampaignErr := make(chan error, 1)\n\n\tctxTimeout, cancelTimeout := context.WithTimeout(ctx, 1*time.Second)\n\tdefer cancelTimeout()\n\n\tgo func() {\n\t\terr := election2.Campaign(ctxTimeout, \"host2\")\n\t\tif err != nil {\n\t\t\tcampaignErr <- err\n\t\t}\n\t\tclose(campaignDone)\n\t}()\n\n\t// Verify second node is blocked (should timeout)\n\tselect {\n\tcase err := <-campaignErr:\n\t\t// Expected to get a timeout error\n\t\trequire.Error(t, err, \"Expected a timeout error for the second campaign\")\n\t\trequire.Contains(t, err.Error(), \"context deadline exceeded\", \"Expected a context deadline error\")\n\tcase <-campaignDone:\n\t\tt.Error(\"Second node should not have been able to become leader while first node holds leadership\")\n\tcase <-time.After(2 * time.Second):\n\t\tt.Error(\"Expected the second campaign to timeout quickly\")\n\t}\n\n\t// First node resigns\n\terr = election1.Resign(ctx)\n\trequire.NoError(t, err)\n\n\t// Now the second node should be able to become leader\n\t// Create a new election for the second host\n\telection3, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\tdefer election3.Cleanup(ctx)\n\n\terr = election3.Campaign(ctx, \"host3\")\n\trequire.NoError(t, err, \"Third host should be able to become leader after first host resigned\")\n}\n\n// TestSessionDone tests the Done channel behavior\nfunc TestSessionDone(t *testing.T) {\n\ttc := setupETCDCluster(t)\n\n\t// Use a separate context for test operations\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tnamespace := \"test-namespace-done\"\n\telection, err := tc.store.CreateElection(ctx, namespace)\n\trequire.NoError(t, err)\n\n\tdoneCh := election.Done()\n\trequire.NotNil(t, doneCh)\n\n\t// Clean up should close the session, which should close the Done channel\n\terr = election.Cleanup(ctx)\n\trequire.NoError(t, err)\n\n\t// Verify the Done channel is closed\n\tselect {\n\tcase <-doneCh:\n\t\t// Expected - channel should be closed\n\tcase <-time.After(2 * time.Second):\n\t\tt.Error(\"Done channel should be closed after Cleanup\")\n\t}\n}\n\n// testCluster represents a test etcd cluster with its resources\ntype testCluster struct {\n\tstore       store.Elector\n\tstoreConfig etcdCfg\n\tendpoints   []string\n}\n\n// setupETCDCluster initializes an etcd cluster for testing\nfunc setupETCDCluster(t *testing.T) *testCluster {\n\tt.Helper()\n\n\tflag.Parse()\n\ttestflags.RequireEtcd(t)\n\n\tendpoints := strings.Split(os.Getenv(\"ETCD_ENDPOINTS\"), \",\")\n\tif endpoints == nil || len(endpoints) == 0 || endpoints[0] == \"\" {\n\t\t// default etcd port\n\t\tendpoints = []string{\"localhost:2379\"}\n\t}\n\n\tt.Logf(\"ETCD endpoints: %v\", endpoints)\n\n\ttestConfig := etcdCfg{\n\t\tEndpoints:   endpoints,\n\t\tDialTimeout: 5 * time.Second,\n\t\tPrefix:      fmt.Sprintf(\"/election/%s\", t.Name()),\n\t\tElectionTTL: 5 * time.Second,\n\t}\n\n\t// Create store\n\tstoreParams := StoreParams{\n\t\tCfg:       config.ShardDistribution{LeaderStore: config.Store{StorageParams: createConfig(t, testConfig)}},\n\t\tLifecycle: fxtest.NewLifecycle(t),\n\t}\n\n\tstore, err := NewLeaderStore(storeParams)\n\trequire.NoError(t, err)\n\n\treturn &testCluster{\n\t\tstore:       store,\n\t\tstoreConfig: testConfig,\n\t\tendpoints:   endpoints,\n\t}\n}\n\nfunc createConfig(t *testing.T, cfg etcdCfg) *config.YamlNode {\n\tt.Helper()\n\n\tyamlCfg, err := yaml.Marshal(cfg)\n\trequire.NoError(t, err)\n\n\tvar res *config.YamlNode\n\n\terr = yaml.Unmarshal(yamlCfg, &res)\n\trequire.NoError(t, err)\n\n\treturn res\n}\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/module.go",
    "content": "package etcd\n\nimport (\n\t\"go.uber.org/fx\"\n\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/executorstore\"\n\t\"github.com/uber/cadence/service/sharddistributor/store/etcd/leaderstore\"\n)\n\nvar Module = fx.Module(\"etcd\",\n\texecutorstore.Module,\n\tfx.Provide(leaderstore.NewLeaderStore),\n)\n"
  },
  {
    "path": "service/sharddistributor/store/etcd/testhelper/testhelper.go",
    "content": "package testhelper\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\tclientv3 \"go.etcd.io/etcd/client/v3\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/service/sharddistributor/config\"\n\t\"github.com/uber/cadence/testflags\"\n)\n\ntype StoreTestCluster struct {\n\tEtcdPrefix  string\n\tNamespace   string\n\tLeaderCfg   config.ShardDistribution\n\tClient      *clientv3.Client\n\tCompression string\n}\n\nfunc SetupStoreTestCluster(t *testing.T) *StoreTestCluster {\n\tt.Helper()\n\tflag.Parse()\n\ttestflags.RequireEtcd(t)\n\n\tnamespace := fmt.Sprintf(\"ns-%s\", strings.ToLower(t.Name()))\n\n\tendpoints := strings.Split(os.Getenv(\"ETCD_ENDPOINTS\"), \",\")\n\tif len(endpoints) == 0 || endpoints[0] == \"\" {\n\t\tendpoints = []string{\"localhost:2379\"}\n\t}\n\tt.Logf(\"ETCD endpoints: %v\", endpoints)\n\n\tetcdPrefix := fmt.Sprintf(\"/test-shard-store/%s\", t.Name())\n\tetcdConfigRaw := map[string]interface{}{\n\t\t\"endpoints\":   endpoints,\n\t\t\"dialTimeout\": \"5s\",\n\t\t\"prefix\":      etcdPrefix,\n\t\t\"electionTTL\": \"5s\", // Needed for leader config part\n\t\t\"compression\": \"snappy\",\n\t}\n\n\tyamlCfg, err := yaml.Marshal(etcdConfigRaw)\n\trequire.NoError(t, err)\n\tvar yamlNode *config.YamlNode\n\terr = yaml.Unmarshal(yamlCfg, &yamlNode)\n\trequire.NoError(t, err)\n\n\tleaderCfg := config.ShardDistribution{\n\t\tStore:       config.Store{StorageParams: yamlNode},\n\t\tLeaderStore: config.Store{StorageParams: yamlNode},\n\t}\n\n\tclient, err := clientv3.New(clientv3.Config{Endpoints: endpoints, DialTimeout: 5 * time.Second})\n\trequire.NoError(t, err)\n\tt.Cleanup(func() { client.Close() })\n\n\t_, err = client.Delete(context.Background(), etcdPrefix, clientv3.WithPrefix())\n\trequire.NoError(t, err)\n\n\treturn &StoreTestCluster{\n\t\tNamespace:  namespace,\n\t\tEtcdPrefix: etcdPrefix,\n\t\tLeaderCfg:  leaderCfg,\n\t\tClient:     client,\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/leaderstore.go",
    "content": "package store\n\nimport (\n\t\"context\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=leaderstore_mock.go Elector,Election\n\n// Elector is an interface that provides a way to establish a session for election.\n// It establishes connection and a session and provides Election to run for leader.\ntype Elector interface {\n\tCreateElection(ctx context.Context, namespace string) (Election, error)\n}\n\n// Election is an interface that establishes leader campaign.\ntype Election interface {\n\t// Campaign is a blocking call that will block until either leadership is acquired and return nil or block.\n\tCampaign(ctx context.Context, hostname string) error\n\t// Resign resigns from leadership.\n\tResign(ctx context.Context) error\n\t// Done returns a channel that notifies that the election session closed.\n\tDone() <-chan struct{}\n\t// Cleanup stops internal processes and releases keys.\n\tCleanup(ctx context.Context) error\n\t// Guard returns a transaction guard representing the current leadership term.\n\t// This guard can be passed to the generic store to perform leader-protected writes.\n\tGuard() GuardFunc\n}\n"
  },
  {
    "path": "service/sharddistributor/store/leaderstore_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: leaderstore.go\n//\n// Generated by this command:\n//\n//\tmockgen -package store -source leaderstore.go -destination=leaderstore_mock.go Elector,Election\n//\n\n// Package store is a generated GoMock package.\npackage store\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockElector is a mock of Elector interface.\ntype MockElector struct {\n\tctrl     *gomock.Controller\n\trecorder *MockElectorMockRecorder\n\tisgomock struct{}\n}\n\n// MockElectorMockRecorder is the mock recorder for MockElector.\ntype MockElectorMockRecorder struct {\n\tmock *MockElector\n}\n\n// NewMockElector creates a new mock instance.\nfunc NewMockElector(ctrl *gomock.Controller) *MockElector {\n\tmock := &MockElector{ctrl: ctrl}\n\tmock.recorder = &MockElectorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockElector) EXPECT() *MockElectorMockRecorder {\n\treturn m.recorder\n}\n\n// CreateElection mocks base method.\nfunc (m *MockElector) CreateElection(ctx context.Context, namespace string) (Election, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateElection\", ctx, namespace)\n\tret0, _ := ret[0].(Election)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// CreateElection indicates an expected call of CreateElection.\nfunc (mr *MockElectorMockRecorder) CreateElection(ctx, namespace any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateElection\", reflect.TypeOf((*MockElector)(nil).CreateElection), ctx, namespace)\n}\n\n// MockElection is a mock of Election interface.\ntype MockElection struct {\n\tctrl     *gomock.Controller\n\trecorder *MockElectionMockRecorder\n\tisgomock struct{}\n}\n\n// MockElectionMockRecorder is the mock recorder for MockElection.\ntype MockElectionMockRecorder struct {\n\tmock *MockElection\n}\n\n// NewMockElection creates a new mock instance.\nfunc NewMockElection(ctrl *gomock.Controller) *MockElection {\n\tmock := &MockElection{ctrl: ctrl}\n\tmock.recorder = &MockElectionMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockElection) EXPECT() *MockElectionMockRecorder {\n\treturn m.recorder\n}\n\n// Campaign mocks base method.\nfunc (m *MockElection) Campaign(ctx context.Context, hostname string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Campaign\", ctx, hostname)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Campaign indicates an expected call of Campaign.\nfunc (mr *MockElectionMockRecorder) Campaign(ctx, hostname any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Campaign\", reflect.TypeOf((*MockElection)(nil).Campaign), ctx, hostname)\n}\n\n// Cleanup mocks base method.\nfunc (m *MockElection) Cleanup(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Cleanup\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Cleanup indicates an expected call of Cleanup.\nfunc (mr *MockElectionMockRecorder) Cleanup(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Cleanup\", reflect.TypeOf((*MockElection)(nil).Cleanup), ctx)\n}\n\n// Done mocks base method.\nfunc (m *MockElection) Done() <-chan struct{} {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Done\")\n\tret0, _ := ret[0].(<-chan struct{})\n\treturn ret0\n}\n\n// Done indicates an expected call of Done.\nfunc (mr *MockElectionMockRecorder) Done() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Done\", reflect.TypeOf((*MockElection)(nil).Done))\n}\n\n// Guard mocks base method.\nfunc (m *MockElection) Guard() GuardFunc {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Guard\")\n\tret0, _ := ret[0].(GuardFunc)\n\treturn ret0\n}\n\n// Guard indicates an expected call of Guard.\nfunc (mr *MockElectionMockRecorder) Guard() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Guard\", reflect.TypeOf((*MockElection)(nil).Guard))\n}\n\n// Resign mocks base method.\nfunc (m *MockElection) Resign(ctx context.Context) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Resign\", ctx)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Resign indicates an expected call of Resign.\nfunc (mr *MockElectionMockRecorder) Resign(ctx any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Resign\", reflect.TypeOf((*MockElection)(nil).Resign), ctx)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/state.go",
    "content": "package store\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype HeartbeatState struct {\n\t// LastHeartbeat is the time of the last heartbeat received from the executor\n\tLastHeartbeat  time.Time\n\tStatus         types.ExecutorStatus\n\tReportedShards map[string]*types.ShardStatusReport\n\tMetadata       map[string]string\n}\n\ntype AssignedState struct {\n\t// AssignedShards holds the current assignment of shards to this executor\n\t// Key: ShardID\n\tAssignedShards map[string]*types.ShardAssignment\n\n\t// ShardHandoverStats holds handover statistics of all shards experienced handovers to this executor\n\t// Mostly all shards in AssignedShards will have corresponding entries here\n\t// But if a shard was assigned but never had a handover (e.g., first assignment), it does not have an entry here\n\t// Key: ShardID\n\tShardHandoverStats map[string]ShardHandoverStats\n\n\t// LastUpdated is the time when this assignment state was last updated\n\t// Used to calculate assignment distribution latency for newly assigned shards\n\tLastUpdated time.Time\n\tModRevision int64\n}\n\n// ShardHandoverStats holds statistics related to the latest handover of a shard\ntype ShardHandoverStats struct {\n\t// PreviousExecutorLastHeartbeatTime is the last heartbeat time received\n\t// from the previous executor before the shard was reassigned.\n\tPreviousExecutorLastHeartbeatTime time.Time\n\n\t// HandoverType indicates the type of handover that occurred during the last shard reassignment.\n\tHandoverType types.HandoverType\n}\n\ntype NamespaceState struct {\n\t// Executors holds the heartbeat states of all executors in the namespace.\n\t// Key: ExecutorID\n\tExecutors map[string]HeartbeatState\n\n\t// ShardStats holds the statistics of all shards in the namespace.\n\t// Only loaded for namespace which types.LoadBalancingMode is types.LoadBalancingModeGREEDY\n\t// Key: ShardID\n\tShardStats map[string]ShardStatistics\n\n\t// ShardAssignments holds the assignment states of all shards in the namespace.\n\t// Key: ExecutorID\n\tShardAssignments map[string]AssignedState\n}\n\ntype ShardState struct {\n\tExecutorID string\n}\n\ntype ShardStatistics struct {\n\t// EWMA of shard load that persists across executor changes\n\tSmoothedLoad float64\n\n\t// LastUpdateTime is the heartbeat timestamp that last updated the EWMA\n\tLastUpdateTime time.Time\n\n\t// LastMoveTime is the timestamp when this shard was last reassigned\n\tLastMoveTime time.Time\n}\n\ntype ShardOwner struct {\n\tExecutorID string\n\tMetadata   map[string]string\n}\n\n// CountExecutorsByStatus returns a map of executor status to the count of executors with that status\nfunc (ns *NamespaceState) CountExecutorsByStatus() map[types.ExecutorStatus]int {\n\tcounts := make(map[types.ExecutorStatus]int)\n\tfor _, executor := range ns.Executors {\n\t\tcounts[executor.Status]++\n\t}\n\treturn counts\n}\n"
  },
  {
    "path": "service/sharddistributor/store/state_test.go",
    "content": "package store\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestNamespaceState_CountExecutorsByStatus(t *testing.T) {\n\ttests := []struct {\n\t\tname      string\n\t\texecutors map[string]HeartbeatState\n\t\texpected  map[types.ExecutorStatus]int\n\t}{\n\t\t{\n\t\t\tname:      \"empty executors\",\n\t\t\texecutors: map[string]HeartbeatState{},\n\t\t\texpected:  map[types.ExecutorStatus]int{},\n\t\t},\n\t\t{\n\t\t\tname: \"single active executor\",\n\t\t\texecutors: map[string]HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t},\n\t\t\texpected: map[types.ExecutorStatus]int{\n\t\t\t\ttypes.ExecutorStatusACTIVE: 1,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"multiple executors same status\",\n\t\t\texecutors: map[string]HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\"exec-3\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t},\n\t\t\texpected: map[types.ExecutorStatus]int{\n\t\t\t\ttypes.ExecutorStatusACTIVE: 3,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"all statuses\",\n\t\t\texecutors: map[string]HeartbeatState{\n\t\t\t\t\"exec-1\": {Status: types.ExecutorStatusINVALID},\n\t\t\t\t\"exec-2\": {Status: types.ExecutorStatusACTIVE},\n\t\t\t\t\"exec-3\": {Status: types.ExecutorStatusDRAINING},\n\t\t\t\t\"exec-4\": {Status: types.ExecutorStatusDRAINED},\n\t\t\t},\n\t\t\texpected: map[types.ExecutorStatus]int{\n\t\t\t\ttypes.ExecutorStatusINVALID:  1,\n\t\t\t\ttypes.ExecutorStatusACTIVE:   1,\n\t\t\t\ttypes.ExecutorStatusDRAINING: 1,\n\t\t\t\ttypes.ExecutorStatusDRAINED:  1,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tns := &NamespaceState{\n\t\t\t\tExecutors: tt.executors,\n\t\t\t}\n\t\t\tresult := ns.CountExecutorsByStatus()\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/store.go",
    "content": "package store\n\nimport (\n\t\"context\"\n\t\"fmt\"\n)\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination=store_mock.go Store\n//go:generate gowrap gen -g -p . -i Store -t ./wrappers/templates/metered.tmpl -o ./wrappers/metered/store_generated.go -v handler=Wrapped\n\nvar (\n\t// ErrExecutorNotFound is an error that is returned when queries executor is not registered in the storage.\n\tErrExecutorNotFound = fmt.Errorf(\"executor not found\")\n\n\t// ErrShardNotFound is an error that is returned when a shard does not exist.\n\tErrShardNotFound = fmt.Errorf(\"shard not found\")\n\n\t// ErrVersionConflict is an error that is returned if during operations some precondition failed.\n\tErrVersionConflict = fmt.Errorf(\"version conflict\")\n\n\t// ErrExecutorNotRunning is an error that is returned when shard is attempted to be assigned to a not running executor.\n\tErrExecutorNotRunning = fmt.Errorf(\"executor not running\")\n)\n\ntype ErrShardAlreadyAssigned struct {\n\tShardID    string\n\tAssignedTo string\n\tMetadata   map[string]string\n}\n\nfunc (e *ErrShardAlreadyAssigned) Error() string {\n\treturn fmt.Sprintf(\"shard %s is already assigned to %s\", e.ShardID, e.AssignedTo)\n}\n\n// Txn represents a generic, backend-agnostic transaction.\n// It is used as a vehicle for the GuardFunc to operate on.\ntype Txn interface{}\n\n// GuardFunc is a function that applies a transactional precondition.\n// It takes a generic transaction, applies a backend-specific guard,\n// and returns the modified transaction.\ntype GuardFunc func(Txn) (Txn, error)\n\n// NopGuard is a no-op guard that can be used when no transactional\n// check is required. It simply returns the transaction as-is.\nfunc NopGuard() GuardFunc {\n\treturn func(txn Txn) (Txn, error) {\n\t\treturn txn, nil\n\t}\n}\n\n// AssignShardsRequest is a request to assign shards to executors, and remove unused shards.\ntype AssignShardsRequest struct {\n\t// NewState is the new state of the namespace, containing the new assignments of shards to executors.\n\tNewState *NamespaceState\n\t// ExecutorsToDelete maps executor IDs to their expected ModRevision for deletion.\n\t// The ModRevision is used to ensure the executor's assigned state hasn't changed since we decided to delete it.\n\tExecutorsToDelete map[string]int64\n}\n\ntype Store interface {\n\t// GetState retrieves the current state of a namespace, including executors,\n\t// shard statistics, and shard assignments.\n\tGetState(ctx context.Context, namespace string) (*NamespaceState, error)\n\n\t// AssignShards assigns multiple shards to executors within a namespace.\n\t// It also updates shard statistics and deletes specified executors\n\t// The operation is atomic and guarded by the provided GuardFunc.\n\tAssignShards(ctx context.Context, namespace string, request AssignShardsRequest, guard GuardFunc) error\n\n\t// AssignShard assigns a single shard to an executor within a namespace.\n\tAssignShard(ctx context.Context, namespace string, shardID string, executorID string) error\n\n\t// SubscribeToExecutorStatusChanges subscribes to changes of executors' status key within a namespace.\n\tSubscribeToExecutorStatusChanges(ctx context.Context, namespace string) (<-chan int64, error)\n\tDeleteExecutors(ctx context.Context, namespace string, executorIDs []string, guard GuardFunc) error\n\n\t// DeleteAssignedStates deletes the assigned states of multiple executors within a namespace.\n\tDeleteAssignedStates(ctx context.Context, namespace string, executorIDs []string, guard GuardFunc) error\n\n\t// GetShardOwner retrieves the owner of a specific shard within a namespace.\n\t// It returns ErrShardNotFound if the shard does not exist.\n\tGetShardOwner(ctx context.Context, namespace, shardID string) (*ShardOwner, error)\n\tSubscribeToAssignmentChanges(ctx context.Context, namespace string) (<-chan map[*ShardOwner][]string, func(), error)\n\n\t// GetExecutor retrieves an executor within a namespace.\n\tGetExecutor(ctx context.Context, namespace string, executorID string) (*ShardOwner, error)\n\n\tGetHeartbeat(ctx context.Context, namespace string, executorID string) (*HeartbeatState, *AssignedState, error)\n\tRecordHeartbeat(ctx context.Context, namespace, executorID string, state HeartbeatState) error\n\n\tDeleteShardStats(ctx context.Context, namespace string, shardIDs []string, guard GuardFunc) error\n}\n"
  },
  {
    "path": "service/sharddistributor/store/store_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: store.go\n//\n// Generated by this command:\n//\n//\tmockgen -package store -source store.go -destination=store_mock.go Store\n//\n\n// Package store is a generated GoMock package.\npackage store\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockTxn is a mock of Txn interface.\ntype MockTxn struct {\n\tctrl     *gomock.Controller\n\trecorder *MockTxnMockRecorder\n\tisgomock struct{}\n}\n\n// MockTxnMockRecorder is the mock recorder for MockTxn.\ntype MockTxnMockRecorder struct {\n\tmock *MockTxn\n}\n\n// NewMockTxn creates a new mock instance.\nfunc NewMockTxn(ctrl *gomock.Controller) *MockTxn {\n\tmock := &MockTxn{ctrl: ctrl}\n\tmock.recorder = &MockTxnMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockTxn) EXPECT() *MockTxnMockRecorder {\n\treturn m.recorder\n}\n\n// MockStore is a mock of Store interface.\ntype MockStore struct {\n\tctrl     *gomock.Controller\n\trecorder *MockStoreMockRecorder\n\tisgomock struct{}\n}\n\n// MockStoreMockRecorder is the mock recorder for MockStore.\ntype MockStoreMockRecorder struct {\n\tmock *MockStore\n}\n\n// NewMockStore creates a new mock instance.\nfunc NewMockStore(ctrl *gomock.Controller) *MockStore {\n\tmock := &MockStore{ctrl: ctrl}\n\tmock.recorder = &MockStoreMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockStore) EXPECT() *MockStoreMockRecorder {\n\treturn m.recorder\n}\n\n// AssignShard mocks base method.\nfunc (m *MockStore) AssignShard(ctx context.Context, namespace, shardID, executorID string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AssignShard\", ctx, namespace, shardID, executorID)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AssignShard indicates an expected call of AssignShard.\nfunc (mr *MockStoreMockRecorder) AssignShard(ctx, namespace, shardID, executorID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AssignShard\", reflect.TypeOf((*MockStore)(nil).AssignShard), ctx, namespace, shardID, executorID)\n}\n\n// AssignShards mocks base method.\nfunc (m *MockStore) AssignShards(ctx context.Context, namespace string, request AssignShardsRequest, guard GuardFunc) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"AssignShards\", ctx, namespace, request, guard)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// AssignShards indicates an expected call of AssignShards.\nfunc (mr *MockStoreMockRecorder) AssignShards(ctx, namespace, request, guard any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"AssignShards\", reflect.TypeOf((*MockStore)(nil).AssignShards), ctx, namespace, request, guard)\n}\n\n// DeleteAssignedStates mocks base method.\nfunc (m *MockStore) DeleteAssignedStates(ctx context.Context, namespace string, executorIDs []string, guard GuardFunc) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteAssignedStates\", ctx, namespace, executorIDs, guard)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteAssignedStates indicates an expected call of DeleteAssignedStates.\nfunc (mr *MockStoreMockRecorder) DeleteAssignedStates(ctx, namespace, executorIDs, guard any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteAssignedStates\", reflect.TypeOf((*MockStore)(nil).DeleteAssignedStates), ctx, namespace, executorIDs, guard)\n}\n\n// DeleteExecutors mocks base method.\nfunc (m *MockStore) DeleteExecutors(ctx context.Context, namespace string, executorIDs []string, guard GuardFunc) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteExecutors\", ctx, namespace, executorIDs, guard)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteExecutors indicates an expected call of DeleteExecutors.\nfunc (mr *MockStoreMockRecorder) DeleteExecutors(ctx, namespace, executorIDs, guard any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteExecutors\", reflect.TypeOf((*MockStore)(nil).DeleteExecutors), ctx, namespace, executorIDs, guard)\n}\n\n// DeleteShardStats mocks base method.\nfunc (m *MockStore) DeleteShardStats(ctx context.Context, namespace string, shardIDs []string, guard GuardFunc) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DeleteShardStats\", ctx, namespace, shardIDs, guard)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DeleteShardStats indicates an expected call of DeleteShardStats.\nfunc (mr *MockStoreMockRecorder) DeleteShardStats(ctx, namespace, shardIDs, guard any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DeleteShardStats\", reflect.TypeOf((*MockStore)(nil).DeleteShardStats), ctx, namespace, shardIDs, guard)\n}\n\n// GetExecutor mocks base method.\nfunc (m *MockStore) GetExecutor(ctx context.Context, namespace, executorID string) (*ShardOwner, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetExecutor\", ctx, namespace, executorID)\n\tret0, _ := ret[0].(*ShardOwner)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetExecutor indicates an expected call of GetExecutor.\nfunc (mr *MockStoreMockRecorder) GetExecutor(ctx, namespace, executorID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetExecutor\", reflect.TypeOf((*MockStore)(nil).GetExecutor), ctx, namespace, executorID)\n}\n\n// GetHeartbeat mocks base method.\nfunc (m *MockStore) GetHeartbeat(ctx context.Context, namespace, executorID string) (*HeartbeatState, *AssignedState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetHeartbeat\", ctx, namespace, executorID)\n\tret0, _ := ret[0].(*HeartbeatState)\n\tret1, _ := ret[1].(*AssignedState)\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// GetHeartbeat indicates an expected call of GetHeartbeat.\nfunc (mr *MockStoreMockRecorder) GetHeartbeat(ctx, namespace, executorID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetHeartbeat\", reflect.TypeOf((*MockStore)(nil).GetHeartbeat), ctx, namespace, executorID)\n}\n\n// GetShardOwner mocks base method.\nfunc (m *MockStore) GetShardOwner(ctx context.Context, namespace, shardID string) (*ShardOwner, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetShardOwner\", ctx, namespace, shardID)\n\tret0, _ := ret[0].(*ShardOwner)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetShardOwner indicates an expected call of GetShardOwner.\nfunc (mr *MockStoreMockRecorder) GetShardOwner(ctx, namespace, shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetShardOwner\", reflect.TypeOf((*MockStore)(nil).GetShardOwner), ctx, namespace, shardID)\n}\n\n// GetState mocks base method.\nfunc (m *MockStore) GetState(ctx context.Context, namespace string) (*NamespaceState, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"GetState\", ctx, namespace)\n\tret0, _ := ret[0].(*NamespaceState)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// GetState indicates an expected call of GetState.\nfunc (mr *MockStoreMockRecorder) GetState(ctx, namespace any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"GetState\", reflect.TypeOf((*MockStore)(nil).GetState), ctx, namespace)\n}\n\n// RecordHeartbeat mocks base method.\nfunc (m *MockStore) RecordHeartbeat(ctx context.Context, namespace, executorID string, state HeartbeatState) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"RecordHeartbeat\", ctx, namespace, executorID, state)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// RecordHeartbeat indicates an expected call of RecordHeartbeat.\nfunc (mr *MockStoreMockRecorder) RecordHeartbeat(ctx, namespace, executorID, state any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RecordHeartbeat\", reflect.TypeOf((*MockStore)(nil).RecordHeartbeat), ctx, namespace, executorID, state)\n}\n\n// SubscribeToAssignmentChanges mocks base method.\nfunc (m *MockStore) SubscribeToAssignmentChanges(ctx context.Context, namespace string) (<-chan map[*ShardOwner][]string, func(), error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SubscribeToAssignmentChanges\", ctx, namespace)\n\tret0, _ := ret[0].(<-chan map[*ShardOwner][]string)\n\tret1, _ := ret[1].(func())\n\tret2, _ := ret[2].(error)\n\treturn ret0, ret1, ret2\n}\n\n// SubscribeToAssignmentChanges indicates an expected call of SubscribeToAssignmentChanges.\nfunc (mr *MockStoreMockRecorder) SubscribeToAssignmentChanges(ctx, namespace any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SubscribeToAssignmentChanges\", reflect.TypeOf((*MockStore)(nil).SubscribeToAssignmentChanges), ctx, namespace)\n}\n\n// SubscribeToExecutorStatusChanges mocks base method.\nfunc (m *MockStore) SubscribeToExecutorStatusChanges(ctx context.Context, namespace string) (<-chan int64, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SubscribeToExecutorStatusChanges\", ctx, namespace)\n\tret0, _ := ret[0].(<-chan int64)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// SubscribeToExecutorStatusChanges indicates an expected call of SubscribeToExecutorStatusChanges.\nfunc (mr *MockStoreMockRecorder) SubscribeToExecutorStatusChanges(ctx, namespace any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SubscribeToExecutorStatusChanges\", reflect.TypeOf((*MockStore)(nil).SubscribeToExecutorStatusChanges), ctx, namespace)\n}\n"
  },
  {
    "path": "service/sharddistributor/store/wrappers/metered/base.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\ntype base struct {\n\tmetricClient metrics.Client\n\tlogger       log.Logger\n\ttimeSource   clock.TimeSource\n}\n\nfunc (p *base) updateErrorMetricPerNamespace(err error, scopeWithNamespaceTags metrics.Scope) {\n\tif errors.Is(err, store.ErrExecutorNotFound) {\n\t\tscopeWithNamespaceTags.IncCounter(metrics.ShardDistributorStoreExecutorNotFound)\n\t}\n\n\tscopeWithNamespaceTags.IncCounter(metrics.ShardDistributorStoreFailuresPerNamespace)\n}\n\nfunc (p *base) call(scope metrics.ScopeIdx, op func() error, tags ...metrics.Tag) error {\n\tmetricsScope := p.metricClient.Scope(scope, tags...)\n\n\tmetricsScope.IncCounter(metrics.ShardDistributorStoreRequestsPerNamespace)\n\tbefore := p.timeSource.Now()\n\terr := op()\n\tduration := p.timeSource.Since(before)\n\tmetricsScope.RecordHistogramDuration(metrics.ShardDistributorStoreLatencyHistogramPerNamespace, duration)\n\n\tif err != nil {\n\t\tp.updateErrorMetricPerNamespace(err, metricsScope)\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "service/sharddistributor/store/wrappers/metered/metered_test.go",
    "content": "package metered\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\nconst (\n\t_testNamespace  = \"test_namespace\"\n\t_testExecutorID = \"test_executorID\"\n)\n\nfunc TestMeteredStore_GetHeartbeat(t *testing.T) {\n\theartbeatRes := &store.HeartbeatState{\n\t\tLastHeartbeat: time.Now().UTC(),\n\t}\n\tassignedState := &store.AssignedState{\n\t\tLastUpdated: time.Now().UTC(),\n\t}\n\n\ttests := []struct {\n\t\tname  string\n\t\terror error\n\t}{\n\t\t{\n\t\t\tname:  \"Success\",\n\t\t\terror: nil,\n\t\t},\n\t\t{\n\t\t\tname:  \"NotFound\",\n\t\t\terror: store.ErrExecutorNotFound,\n\t\t},\n\t\t{\n\t\t\tname:  \"Failure\",\n\t\t\terror: &types.InternalServiceError{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.ShardDistributor, metrics.HistogramMigration{})\n\t\t\ttimeSource := clock.NewMockedTimeSource()\n\t\t\tmockHandler := store.NewMockStore(ctrl)\n\n\t\t\tmockHandler.EXPECT().GetHeartbeat(gomock.Any(), _testNamespace, _testExecutorID).Do(func(ctx context.Context, namespace string, executorID string) {\n\t\t\t\ttimeSource.Advance(time.Second)\n\t\t\t}).Return(heartbeatRes, assignedState, tt.error)\n\n\t\t\tmockLogger := log.NewMockLogger(ctrl)\n\t\t\tmockLogger.EXPECT().Helper().Return(mockLogger).AnyTimes()\n\n\t\t\twrapped := NewStore(mockHandler, metricsClient, mockLogger, timeSource).(*meteredStore)\n\n\t\t\tgotHeartbeat, gotAssignedState, err := wrapped.GetHeartbeat(context.Background(), _testNamespace, _testExecutorID)\n\n\t\t\tassert.Equal(t, heartbeatRes, gotHeartbeat)\n\t\t\tassert.Equal(t, assignedState, gotAssignedState)\n\t\t\tassert.Equal(t, tt.error, err)\n\n\t\t\t// check that the metrics were emitted for this method\n\t\t\trequestCounterName := \"test.shard_distributor_store_requests_per_namespace+namespace=test_namespace,operation=StoreGetHeartbeat\"\n\t\t\tassert.Contains(t, testScope.Snapshot().Counters(), requestCounterName)\n\t\t\trequestCounter := testScope.Snapshot().Counters()[requestCounterName]\n\t\t\tassert.Equal(t, int64(1), requestCounter.Value())\n\n\t\t\tlatencyHistogramName := \"test.shard_distributor_store_latency_histogram_per_namespace+namespace=test_namespace,operation=StoreGetHeartbeat\"\n\t\t\tallHistograms := testScope.Snapshot().Histograms()\n\t\t\tassert.Contains(t, allHistograms, latencyHistogramName)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/sharddistributor/store/wrappers/metered/store_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n// meteredStore implements store.Store interface instrumented with metrics.\ntype meteredStore struct {\n\tbase\n\twrapped store.Store\n}\n\n// NewStore creates a new instance of Store with metrics.\nfunc NewStore(\n\twrapped store.Store,\n\tmetricClient metrics.Client,\n\tlogger log.Logger,\n\ttimeSource clock.TimeSource,\n) store.Store {\n\tif wrapped == nil {\n\t\treturn nil\n\t}\n\treturn &meteredStore{\n\t\twrapped: wrapped,\n\t\tbase: base{\n\t\t\tmetricClient: metricClient,\n\t\t\tlogger:       logger,\n\t\t\ttimeSource:   timeSource,\n\t\t},\n\t}\n}\n\nfunc (c *meteredStore) AssignShard(ctx context.Context, namespace string, shardID string, executorID string) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.AssignShard(ctx, namespace, shardID, executorID)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreAssignShardScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) AssignShards(ctx context.Context, namespace string, request store.AssignShardsRequest, guard store.GuardFunc) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.AssignShards(ctx, namespace, request, guard)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreAssignShardsScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) DeleteAssignedStates(ctx context.Context, namespace string, executorIDs []string, guard store.GuardFunc) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteAssignedStates(ctx, namespace, executorIDs, guard)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreDeleteAssignedStatesScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) DeleteExecutors(ctx context.Context, namespace string, executorIDs []string, guard store.GuardFunc) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteExecutors(ctx, namespace, executorIDs, guard)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreDeleteExecutorsScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) DeleteShardStats(ctx context.Context, namespace string, shardIDs []string, guard store.GuardFunc) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.DeleteShardStats(ctx, namespace, shardIDs, guard)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreDeleteShardStatsScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) GetExecutor(ctx context.Context, namespace string, executorID string) (sp1 *store.ShardOwner, err error) {\n\top := func() error {\n\t\tsp1, err = c.wrapped.GetExecutor(ctx, namespace, executorID)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreGetExecutorScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) GetHeartbeat(ctx context.Context, namespace string, executorID string) (hp1 *store.HeartbeatState, ap1 *store.AssignedState, err error) {\n\top := func() error {\n\t\thp1, ap1, err = c.wrapped.GetHeartbeat(ctx, namespace, executorID)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreGetHeartbeatScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) GetShardOwner(ctx context.Context, namespace string, shardID string) (sp1 *store.ShardOwner, err error) {\n\top := func() error {\n\t\tsp1, err = c.wrapped.GetShardOwner(ctx, namespace, shardID)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreGetShardOwnerScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) GetState(ctx context.Context, namespace string) (np1 *store.NamespaceState, err error) {\n\top := func() error {\n\t\tnp1, err = c.wrapped.GetState(ctx, namespace)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreGetStateScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) RecordHeartbeat(ctx context.Context, namespace string, executorID string, state store.HeartbeatState) (err error) {\n\top := func() error {\n\t\terr = c.wrapped.RecordHeartbeat(ctx, namespace, executorID, state)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreRecordHeartbeatScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) SubscribeToAssignmentChanges(ctx context.Context, namespace string) (ch1 <-chan map[*store.ShardOwner][]string, f1 func(), err error) {\n\top := func() error {\n\t\tch1, f1, err = c.wrapped.SubscribeToAssignmentChanges(ctx, namespace)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreSubscribeToAssignmentChangesScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n\nfunc (c *meteredStore) SubscribeToExecutorStatusChanges(ctx context.Context, namespace string) (ch1 <-chan int64, err error) {\n\top := func() error {\n\t\tch1, err = c.wrapped.SubscribeToExecutorStatusChanges(ctx, namespace)\n\t\treturn err\n\t}\n\n\terr = c.call(metrics.ShardDistributorStoreSubscribeToExecutorStatusChangesScope, op, metrics.NamespaceTag(namespace))\n\treturn\n}\n"
  },
  {
    "path": "service/sharddistributor/store/wrappers/templates/metered.tmpl",
    "content": "import (\n\t\"context\"\n\n    \"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/sharddistributor/store\"\n)\n\n{{ $decorator := (printf \"metered%s\" .Interface.Name) }}\n{{ $interfaceName := .Interface.Name }}\n\n// {{$decorator}} implements {{.Interface.Type}} interface instrumented with metrics.\ntype {{$decorator}} struct {\n    base\n    wrapped     {{.Interface.Type}}\n}\n\n// New{{.Interface.Name}} creates a new instance of {{.Interface.Name}} with metrics.\nfunc New{{.Interface.Name}}(\n\twrapped      store.{{.Interface.Name}},\n\tmetricClient metrics.Client,\n\tlogger       log.Logger,\n    timeSource   clock.TimeSource,\n) store.{{.Interface.Name}} {\n    if wrapped == nil {\n        return nil\n    }\n    return &{{$decorator}}{\n        wrapped: wrapped,\n        base:    base{\n\t\t\tmetricClient: metricClient,\n\t\t\tlogger:     logger,\n            timeSource: timeSource,\n        },\n    }\n}\n\n{{range $methodName, $method := .Interface.Methods}}\n    {{- if (and $method.AcceptsContext $method.ReturnsError)}}\n        func (c *{{$decorator}}) {{$method.Declaration}} {\n\t        op := func() error {\n\t\t        {{$method.ResultsNames}} = c.wrapped.{{$method.Call}}\n\t\t        return err\n\t        }\n\t        {{$scopeName := printf \"metrics.ShardDistributorStore%sScope\" $methodName}}\n\n            {{ $namespaceParam := (index $method.Params 1).Name }}\n            {{$extraTags := printf \", metrics.NamespaceTag(%s)\" $namespaceParam}}\n\n\t        err = c.call({{$scopeName}}, op{{$extraTags}})\n\t        return\n        }\n    {{else}}\n           func (c *{{$decorator}}) {{$method.Declaration}} {\n               {{ $method.Pass \"c.wrapped.\" }}\n           }\n    {{end}}\n{{end}}\n"
  },
  {
    "path": "service/sharddistributor/templates/metered.tmpl",
    "content": "\nimport (\n    \"context\"\n\n    \"github.com/uber/cadence/common/log\"\n    \"github.com/uber/cadence/common/log/tag\"\n    \"github.com/uber/cadence/common/metrics\"\n    \"github.com/uber/cadence/common/types\"\n    \"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\n{{- $interfaceName := .Interface.Name}}\n{{- $interfaceType := .Interface.Type}}\n{{- $handlerName := (index .Vars \"handler\")}}\n{{- $decorator := (printf \"%s%s\" (down $handlerName) $interfaceName) }}\n{{- $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n\ntype {{$decorator}} struct {\n    handler {{.Interface.Type}}\n    logger log.Logger\n    metricsClient metrics.Client\n}\n\nfunc New{{$Decorator}}(handler {{$.Interface.Type}}, logger log.Logger, metricsClient metrics.Client) {{.Interface.Type}} {\n    return &{{$decorator}}{\n        handler: handler,\n        logger: logger,\n        metricsClient: metricsClient,\n    }\n}\n\n{{range $method := .Interface.Methods}}\nfunc (h *{{$decorator}}) {{$method.Declaration}} {\n    {{- if has $method.Name (list \"Health\" \"Start\" \"Stop\")}}\n        {{ $method.Pass \"h.handler.\" }}\n    {{- else}}\n        defer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\n        {{- $namespace := \"\" }}\n        {{- range $method.Params }}\n            {{- if contains \"Request\" .Type  }}\n                {{- $namespace = printf \"%s.GetNamespace()\" .Name }}\n            {{- end }}\n        {{- end }}\n\n        scope := h.metricsClient.Scope({{ printf \"metrics.ShardDistributor%sScope\" $method.Name }})\n        scope = scope.Tagged(metrics.NamespaceTag({{ $namespace }}))\n        scope.IncCounter(metrics.ShardDistributorRequests)\n        sw := scope.StartTimer(metrics.ShardDistributorLatency)\n        defer sw.Stop()\n        logger := h.logger.WithTags(tag.ShardNamespace({{ $namespace }}))\n\n        {{$method.ResultsNames}} = h.handler.{{ $method.Call }}\n\n        if err != nil {\n            handleErr(err, scope, logger)\n        }\n\n        return {{ $method.ResultsNames }}\n    {{- end}}\n}\n{{end}}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/grpc/grpc_executor_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\ntype ExecutorGRPCExecutor struct {\n\th handler.Executor\n}\n\nfunc NewExecutorGRPCExecutor(h handler.Executor) ExecutorGRPCExecutor {\n\treturn ExecutorGRPCExecutor{h}\n}\n\nfunc (g ExecutorGRPCExecutor) Heartbeat(ctx context.Context, request *sharddistributorv1.HeartbeatRequest) (*sharddistributorv1.HeartbeatResponse, error) {\n\tresponse, err := g.h.Heartbeat(ctx, proto.ToShardDistributorExecutorHeartbeatRequest(request))\n\treturn proto.FromShardDistributorExecutorHeartbeatResponse(response), proto.FromError(err)\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/grpc/grpc_handler.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage grpc\n\nimport (\n\t\"context\"\n\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\nfunc (g GRPCHandler) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(sharddistributorv1.BuildShardDistributorAPIYARPCProcedures(g))\n}\n\nfunc (g GRPCHandler) Health(ctx context.Context, _ *apiv1.HealthRequest) (*apiv1.HealthResponse, error) {\n\tresponse, err := g.h.Health(ctx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn proto.FromHealthResponse(response), proto.FromError(err)\n}\n\nfunc (g ExecutorGRPCExecutor) Register(dispatcher *yarpc.Dispatcher) {\n\tdispatcher.Register(sharddistributorv1.BuildShardDistributorExecutorAPIYARPCProcedures(g))\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/grpc/grpc_handler_generated.go",
    "content": "package grpc\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../../templates/grpc.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\tsharddistributorv1 \"github.com/uber/cadence/.gen/proto/sharddistributor/v1\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\ntype GRPCHandler struct {\n\th handler.Handler\n}\n\nfunc NewGRPCHandler(h handler.Handler) GRPCHandler {\n\treturn GRPCHandler{h}\n}\n\nfunc (g GRPCHandler) GetShardOwner(ctx context.Context, request *sharddistributorv1.GetShardOwnerRequest) (*sharddistributorv1.GetShardOwnerResponse, error) {\n\tresponse, err := g.h.GetShardOwner(ctx, proto.ToShardDistributorGetShardOwnerRequest(request))\n\treturn proto.FromShardDistributorGetShardOwnerResponse(response), proto.FromError(err)\n}\n\nfunc (g GRPCHandler) WatchNamespaceState(request *sharddistributorv1.WatchNamespaceStateRequest, server sharddistributorv1.ShardDistributorAPIServiceWatchNamespaceStateYARPCServer) error {\n\terr := g.h.WatchNamespaceState(proto.ToShardDistributorWatchNamespaceStateRequest(request), &watchnamespacestateServer{server: server})\n\treturn err\n}\n\ntype watchnamespacestateServer struct {\n\tserver sharddistributorv1.ShardDistributorAPIServiceWatchNamespaceStateYARPCServer\n}\n\nfunc (s *watchnamespacestateServer) Context() context.Context {\n\treturn s.server.Context()\n}\n\nfunc (s *watchnamespacestateServer) Send(response *types.WatchNamespaceStateResponse) error {\n\treturn s.server.Send(proto.FromShardDistributorWatchNamespaceStateResponse(response))\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/metered/api_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\ntype metricsHandler struct {\n\thandler       handler.Handler\n\tlogger        log.Logger\n\tmetricsClient metrics.Client\n}\n\nfunc NewMetricsHandler(handler handler.Handler, logger log.Logger, metricsClient metrics.Client) handler.Handler {\n\treturn &metricsHandler{\n\t\thandler:       handler,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (h *metricsHandler) GetShardOwner(ctx context.Context, gp1 *types.GetShardOwnerRequest) (gp2 *types.GetShardOwnerResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\n\tscope := h.metricsClient.Scope(metrics.ShardDistributorGetShardOwnerScope)\n\tscope = scope.Tagged(metrics.NamespaceTag(gp1.GetNamespace()))\n\tscope.IncCounter(metrics.ShardDistributorRequests)\n\tsw := scope.StartTimer(metrics.ShardDistributorLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tag.ShardNamespace(gp1.GetNamespace()))\n\n\tgp2, err = h.handler.GetShardOwner(ctx, gp1)\n\n\tif err != nil {\n\t\thandleErr(err, scope, logger)\n\t}\n\n\treturn gp2, err\n}\n\nfunc (h *metricsHandler) Health(ctx context.Context) (hp1 *types.HealthStatus, err error) {\n\treturn h.handler.Health(ctx)\n}\n\nfunc (h *metricsHandler) Start() {\n\th.handler.Start()\n\treturn\n}\n\nfunc (h *metricsHandler) Stop() {\n\th.handler.Stop()\n\treturn\n}\n\nfunc (h *metricsHandler) WatchNamespaceState(wp1 *types.WatchNamespaceStateRequest, w1 handler.WatchNamespaceStateServer) (err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\n\tscope := h.metricsClient.Scope(metrics.ShardDistributorWatchNamespaceStateScope)\n\tscope = scope.Tagged(metrics.NamespaceTag(wp1.GetNamespace()))\n\tscope.IncCounter(metrics.ShardDistributorRequests)\n\tsw := scope.StartTimer(metrics.ShardDistributorLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tag.ShardNamespace(wp1.GetNamespace()))\n\n\terr = h.handler.WatchNamespaceState(wp1, w1)\n\n\tif err != nil {\n\t\thandleErr(err, scope, logger)\n\t}\n\n\treturn err\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/metered/executor_generated.go",
    "content": "package metered\n\n// Code generated by gowrap. DO NOT EDIT.\n// template: ../../templates/metered.tmpl\n// gowrap: http://github.com/hexdigest/gowrap\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\ntype executormetricsExecutor struct {\n\thandler       handler.Executor\n\tlogger        log.Logger\n\tmetricsClient metrics.Client\n}\n\nfunc NewExecutorMetricsExecutor(handler handler.Executor, logger log.Logger, metricsClient metrics.Client) handler.Executor {\n\treturn &executormetricsExecutor{\n\t\thandler:       handler,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t}\n}\n\nfunc (h *executormetricsExecutor) Heartbeat(ctx context.Context, ep1 *types.ExecutorHeartbeatRequest) (ep2 *types.ExecutorHeartbeatResponse, err error) {\n\tdefer func() { log.CapturePanic(recover(), h.logger, &err) }()\n\n\tscope := h.metricsClient.Scope(metrics.ShardDistributorHeartbeatScope)\n\tscope = scope.Tagged(metrics.NamespaceTag(ep1.GetNamespace()))\n\tscope.IncCounter(metrics.ShardDistributorRequests)\n\tsw := scope.StartTimer(metrics.ShardDistributorLatency)\n\tdefer sw.Stop()\n\tlogger := h.logger.WithTags(tag.ShardNamespace(ep1.GetNamespace()))\n\n\tep2, err = h.handler.Heartbeat(ctx, ep1)\n\n\tif err != nil {\n\t\thandleErr(err, scope, logger)\n\t}\n\n\treturn ep2, err\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/metered/metered.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc handleErr(err error, scope metrics.Scope, logger log.Logger) error {\n\tlogger = logger.Helper()\n\tswitch {\n\tcase errors.As(err, new(*types.InternalServiceError)):\n\t\tscope.IncCounter(metrics.ShardDistributorFailures)\n\t\tlogger.Error(\"Internal service error\", tag.Error(err))\n\t\treturn err\n\tcase errors.As(err, new(*types.NamespaceNotFoundError)):\n\t\tscope.IncCounter(metrics.ShardDistributorErrNamespaceNotFound)\n\t\treturn err\n\tcase errors.As(err, new(*types.ShardNotFoundError)):\n\t\tscope.IncCounter(metrics.ShardDistributorErrShardNotFound)\n\t\treturn err\n\t}\n\tif errors.Is(err, context.DeadlineExceeded) {\n\t\tlogger.Error(\"request timeout\", tag.Error(err))\n\t\tscope.IncCounter(metrics.ShardDistributorErrContextTimeoutCounter)\n\t\treturn err\n\t}\n\n\tlogger.Error(\"internal uncategorized error\", tag.Error(err))\n\tscope.IncCounter(metrics.ShardDistributorFailures)\n\treturn err\n}\n"
  },
  {
    "path": "service/sharddistributor/wrappers/metered/metered_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage metered\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/sharddistributor/handler\"\n)\n\nfunc TestMetricsHandler_GetShardOwner(t *testing.T) {\n\trequest := &types.GetShardOwnerRequest{\n\t\tShardKey:  \"test-shard-key\",\n\t\tNamespace: \"test-namespace\",\n\t}\n\tresponse := &types.GetShardOwnerResponse{\n\t\tOwner:     \"test-owner\",\n\t\tNamespace: \"test-namespace\",\n\t}\n\n\ttests := []struct {\n\t\tname       string\n\t\tsetupMocks func(logger *log.MockLogger)\n\t\terror      error\n\t}{\n\t\t{\n\t\t\tname:       \"Success\",\n\t\t\tsetupMocks: func(logger *log.MockLogger) {},\n\t\t\terror:      nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Failure\",\n\t\t\tsetupMocks: func(logger *log.MockLogger) {\n\t\t\t\tlogger.EXPECT().Helper().Return(logger)\n\t\t\t\tlogger.EXPECT().Error(\n\t\t\t\t\t\"Internal service error\",\n\t\t\t\t\t[]tag.Tag{tag.Error(&types.InternalServiceError{})},\n\t\t\t\t).Times(1)\n\t\t\t},\n\t\t\terror: &types.InternalServiceError{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.ShardDistributor, metrics.HistogramMigration{})\n\t\t\tmockHandler := handler.NewMockHandler(ctrl)\n\n\t\t\tmockHandler.EXPECT().GetShardOwner(gomock.Any(), request).Return(response, tt.error)\n\n\t\t\tmockLogger := log.NewMockLogger(ctrl)\n\t\t\tmockLogger.EXPECT().WithTags([]tag.Tag{tag.ShardNamespace(\"test-namespace\")}).Return(mockLogger)\n\n\t\t\tmockMetricHandler := NewMetricsHandler(mockHandler, mockLogger, metricsClient).(*metricsHandler)\n\t\t\ttt.setupMocks(mockLogger)\n\n\t\t\tgotResponse, err := mockMetricHandler.GetShardOwner(context.Background(), request)\n\n\t\t\tassert.Equal(t, response, gotResponse)\n\t\t\tassert.Equal(t, tt.error, err)\n\n\t\t\t// check that the metrics were emitted for this method\n\t\t\trequestCounterName := \"test.shard_distributor_requests+namespace=test-namespace,operation=GetShardOwner\"\n\t\t\trequestCounter := testScope.Snapshot().Counters()[requestCounterName]\n\t\t\tassert.NotNil(t, requestCounter)\n\t\t\tassert.Equal(t, int64(1), requestCounter.Value())\n\n\t\t\tlatencyTimerName := \"test.shard_distributor_latency+namespace=test-namespace,operation=GetShardOwner\"\n\t\t\tlatencyTimer := testScope.Snapshot().Timers()[latencyTimerName]\n\t\t\tassert.NotNil(t, latencyTimer)\n\n\t\t\t// check that the latency timer was run, and is not 0\n\t\t\tassert.Greater(t, latencyTimer.Values()[0], int64(0))\n\t\t})\n\t}\n}\n\n// For these methods we expect no metrics nor logs to be emitted\nfunc TestPassThroughMethods(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tcall func(*testing.T, *metricsHandler, *handler.MockHandler)\n\t}{\n\t\t{\n\t\t\tname: \"Health\",\n\t\t\tcall: func(t *testing.T, h *metricsHandler, handlerMock *handler.MockHandler) {\n\t\t\t\tctx := context.Background()\n\t\t\t\thealthStatus := &types.HealthStatus{\n\t\t\t\t\tOk:  true,\n\t\t\t\t\tMsg: \"test\",\n\t\t\t\t}\n\t\t\t\thandlerMock.EXPECT().Health(ctx).Return(healthStatus, nil)\n\n\t\t\t\thealthStatusRet, err := h.Health(ctx)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, healthStatus, healthStatusRet)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Start\",\n\t\t\tcall: func(t *testing.T, h *metricsHandler, handlerMock *handler.MockHandler) {\n\t\t\t\thandlerMock.EXPECT().Start()\n\t\t\t\th.Start()\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Stop\",\n\t\t\tcall: func(t *testing.T, h *metricsHandler, handlerMock *handler.MockHandler) {\n\t\t\t\thandlerMock.EXPECT().Stop()\n\t\t\t\th.Stop()\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.ShardDistributor, metrics.HistogramMigration{})\n\t\t\tmockHandler := handler.NewMockHandler(ctrl)\n\n\t\t\t// We expect _no_ log calls\n\t\t\tmockLogger := log.NewMockLogger(ctrl)\n\n\t\t\ttestMetricsHandler := NewMetricsHandler(mockHandler, mockLogger, metricsClient).(*metricsHandler)\n\n\t\t\ttt.call(t, testMetricsHandler, mockHandler)\n\n\t\t\t// check that no metrics were emitted for this method\n\t\t\tassert.Empty(t, testScope.Snapshot().Counters())\n\t\t\tassert.Empty(t, testScope.Snapshot().Timers())\n\t\t\tassert.Empty(t, testScope.Snapshot().Gauges())\n\t\t\tassert.Empty(t, testScope.Snapshot().Histograms())\n\t\t})\n\t}\n}\n\nfunc TestHandleErr(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\terr           error\n\t\texpectedError error\n\t\tsetupMocks    func(mockLogger *log.MockLogger)\n\t\tmetricName    string\n\t}{\n\t\t{\n\t\t\tname:          \"InternalServiceError\",\n\t\t\terr:           &types.InternalServiceError{},\n\t\t\texpectedError: &types.InternalServiceError{},\n\t\t\tsetupMocks: func(mockLogger *log.MockLogger) {\n\t\t\t\tmockLogger.EXPECT().Error(\n\t\t\t\t\t\"Internal service error\",\n\t\t\t\t\t[]tag.Tag{tag.Error(&types.InternalServiceError{})},\n\t\t\t\t).Times(1)\n\t\t\t},\n\t\t\tmetricName: \"shard_distributor_failures\",\n\t\t},\n\t\t{\n\t\t\tname:          \"NamespaceNotFoundError\",\n\t\t\terr:           &types.NamespaceNotFoundError{},\n\t\t\texpectedError: &types.NamespaceNotFoundError{},\n\t\t\tsetupMocks:    func(mockLogger *log.MockLogger) {},\n\t\t\tmetricName:    \"shard_distributor_err_namespace_not_found\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ShardNotFoundError\",\n\t\t\terr:           &types.ShardNotFoundError{},\n\t\t\texpectedError: &types.ShardNotFoundError{},\n\t\t\tsetupMocks:    func(mockLogger *log.MockLogger) {},\n\t\t\tmetricName:    \"shard_distributor_err_shard_not_found\",\n\t\t},\n\t\t{\n\t\t\tname:          \"ContextDeadlineExceeded\",\n\t\t\terr:           context.DeadlineExceeded,\n\t\t\texpectedError: context.DeadlineExceeded,\n\t\t\tsetupMocks: func(mockLogger *log.MockLogger) {\n\t\t\t\tmockLogger.EXPECT().Error(\n\t\t\t\t\t\"request timeout\",\n\t\t\t\t\t[]tag.Tag{tag.Error(context.DeadlineExceeded)},\n\t\t\t\t).Times(1)\n\t\t\t},\n\t\t\tmetricName: \"shard_distributor_err_context_timeout\",\n\t\t},\n\t\t{\n\t\t\tname:          \"UncategorizedError\",\n\t\t\terr:           errors.New(\"uncategorized error\"),\n\t\t\texpectedError: errors.New(\"uncategorized error\"),\n\t\t\tsetupMocks: func(mockLogger *log.MockLogger) {\n\t\t\t\tmockLogger.EXPECT().Error(\n\t\t\t\t\t\"internal uncategorized error\",\n\t\t\t\t\t[]tag.Tag{tag.Error(errors.New(\"uncategorized error\"))},\n\t\t\t\t).Times(1)\n\t\t\t},\n\t\t\tmetricName: \"shard_distributor_failures\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttestScope := tally.NewTestScope(\"test\", nil)\n\t\t\tmetricsClient := metrics.NewClient(testScope, metrics.ShardDistributor, metrics.HistogramMigration{})\n\t\t\tmockLogger := log.NewMockLogger(gomock.NewController(t))\n\t\t\tmockLogger.EXPECT().Helper().Return(mockLogger)\n\n\t\t\ttt.setupMocks(mockLogger)\n\t\t\terr := handleErr(tt.err, metricsClient.Scope(metrics.ShardDistributorGetShardOwnerScope), mockLogger)\n\t\t\trequire.Equal(t, tt.expectedError, err)\n\n\t\t\t// check metrics\n\t\t\tfullName := fmt.Sprintf(\"test.%s+operation=GetShardOwner\", tt.metricName)\n\t\t\tcounter := testScope.Snapshot().Counters()[fullName]\n\t\t\tassert.NotNil(t, counter)\n\t\t\tassert.Equal(t, int64(1), counter.Value())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/templates/grpc.tmpl",
    "content": "{{$packagePath := (index .Vars \"path\")}}\n{{$package := (index .Vars \"package\")}}\n{{$prefix := (index .Vars \"prefix\")}}\nimport (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t{{$package}} \"{{$packagePath}}\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/proto\"\n)\n\n{{$interfaceName := .Interface.Name}}\n{{$handlerName := (index .Vars \"handler\")}}\n{{ $Decorator := (printf \"%s%s\" $handlerName $interfaceName) }}\n{{$denylist := list \"Start\" \"Stop\" \"PrepareToStop\" \"Health\"}}\n\ntype {{$Decorator}} struct {\n\th {{.Interface.Type}}\n}\n\nfunc New{{$Decorator}}(h {{.Interface.Type}}) {{$Decorator}} {\n\treturn {{$Decorator}}{h}\n}\n\n{{range $method := .Interface.Methods}}\n{{if not (has $method.Name $denylist)}}\n{{$Request := printf \"%sRequest\" $method.Name}}\n{{$Response := printf \"%sResponse\" $method.Name}}\n{{- $isStreaming := false}}\n{{- range $method.Params}}\n\t{{- if contains \"Server\" .Type}}\n\t\t{{- $isStreaming = true}}\n\t{{- end}}\n{{- end}}\n{{- if $isStreaming}}\nfunc (g {{$Decorator}}) {{$method.Name}}(request *{{$package}}.{{$Request}}, server {{$package}}.{{$prefix}}APIService{{$method.Name}}YARPCServer) error {\n\terr := g.h.{{$method.Name}}(proto.To{{$prefix}}{{$Request}}(request), &{{lower $method.Name}}Server{server: server})\n\treturn err\n}\n\ntype {{lower $method.Name}}Server struct {\n\tserver {{$package}}.{{$prefix}}APIService{{$method.Name}}YARPCServer\n}\n\nfunc (s *{{lower $method.Name}}Server) Context() context.Context {\n\treturn s.server.Context()\n}\n\nfunc (s *{{lower $method.Name}}Server) Send(response *types.{{$Response}}) error {\n\treturn s.server.Send(proto.From{{$prefix}}{{$Response}}(response))\n}\n{{- else}}\nfunc (g {{$Decorator}}) {{$method.Name}}(ctx context.Context, request *{{$package}}.{{$Request}}) (*{{$package}}.{{$Response}}, error) {\n\t{{- if eq (len $method.Params) 1}}\n\t{{- if eq (len $method.Results) 1}}\n\terr := g.h.{{$method.Call}}\n\t{{- else}}\n\tresponse, err := g.h.{{$method.Name}}(ctx)\n\t{{- end}}\n\t{{- else}}\n\t{{- if eq (len $method.Results) 1}}\n\terr := g.h.{{$method.Name}}(ctx, proto.To{{$prefix}}{{$Request}}(request))\n\t{{- else}}\n\tresponse, err := g.h.{{$method.Name}}(ctx, proto.To{{$prefix}}{{$Request}}(request))\n\t{{- end}}\n\t{{- end}}\n\n\t{{- if eq (len $method.Results) 1}}\n\treturn &{{$package}}.{{$Response}}{}, proto.FromError(err)\n\t{{- else}}\n\treturn proto.From{{$prefix}}{{$Response}}(response), proto.FromError(err)\n\t{{- end}}\n}\n{{- end}}\n{{end}}\n{{end}}\n"
  },
  {
    "path": "service/templates/thrift.tmpl",
    "content": "import (\n\t\"context\"\n\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/.gen/go/admin\"\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/matching\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n)\n\n{{$prefix := (index .Vars \"prefix\")}}\n{{$handlerName := (index .Vars \"handler\")}}\n{{ $Decorator := (printf \"%sHandler\" $handlerName) }}\n\n{{range $method := .Interface.Methods}}\n{{$Request := printf \"%sRequest\" $method.Name}}\n{{$Response := printf \"%sResponse\" $method.Name}}\nfunc (g {{$Decorator}}) {{$method.Declaration}} {\n\t{{- if eq (len $method.Params) 1}}\n\t{{- if eq (len $method.Results) 1}}\n\t{{(index $method.Results 0).Name}} = g.h.{{$method.Call}}\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.h.{{$method.Call}}\n\t{{- end}}\n\t{{- else}}\n\t{{- if eq (len $method.Results) 1}}\n\t{{- if or (eq $method.Name \"AddDecisionTask\") (eq $method.Name \"AddActivityTask\")}}\n\t_, {{(index $method.Results 0).Name}} = g.h.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.To{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}))\n\t{{- else}}\n\t{{(index $method.Results 0).Name}} = g.h.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.To{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}))\n\t{{- end}}\n\t{{- else}}\n\tresponse, {{(index $method.Results 1).Name}} := g.h.{{$method.Name}}({{(index $method.Params 0).Name}}, thrift.To{{$prefix}}{{$Request}}({{(index $method.Params 1).Name}}))\n\t{{- end}}\n\t{{- end}}\n\n\t{{- if eq (len $method.Results) 1}}\n\treturn thrift.FromError({{(index $method.Results 0).Name}})\n\t{{- else}}\n\treturn thrift.From{{$prefix}}{{$Response}}(response), thrift.FromError({{(index $method.Results 1).Name}})\n\t{{- end}}\n}\n{{end}}\n"
  },
  {
    "path": "service/worker/README.md",
    "content": "Cadence Worker\n==============\n\nCadence Worker is a new role for Cadence service used for hosting any\ncomponents responsible for performing background processing on the Cadence\ncluster.\n\nReplicator\n----------\n\nReplicator is a background worker responsible for consuming replication tasks\ngenerated by remote Cadence clusters and pass it down to processor so they\ncan be applied to local Cadence cluster.\n\nQuickstart for local development with multiple Cadence clusters and replication\n====================================\n1. Start dependency using docker if you don't have one running:\n```\ndocker compose -f docker/dev/cassandra.yml up\n```\nThen install the schemas:\n```\nmake install-schema-xdc\n```\n\n2. Start Cadence development server for cluster0, cluster1 and cluster2:\n```\n./cadence-server --zone xdc_cluster0 start\n./cadence-server --zone xdc_cluster1 start\n./cadence-server --zone xdc_cluster2 start\n```\n\n3. Create a global Cadence domain that replicates data across clusters\n```\ncadence --do samples-domain domain register --ac cluster0 --cl cluster0,cluster1,cluster2\n```\nThen run a helloworld from [Go Client Sample](https://github.com/cadence-workflow/cadence-samples/) or [Java Client Sample](https://github.com/cadence-workflow/cadence-java-samples)\n\n4. Failover a domain between clusters:\n\nFailover to cluster1:\n```\ncadence --do samples-domain domain update --ac cluster1\n```\nor failover to cluster2:\n   ```\n   cadence --do samples-domain domain update --ac cluster2\n   ```\nFailback to cluster0:\n```\ncadence --do samples-domain domain update --ac cluster0\n```\n\n## Multiple region setup\nIn a multiple region setup, use another set of config instead.\n\n```\n./cadence-server --zone cross_region_cluster0 start\n./cadence-server --zone cross_region_cluster1 start\n./cadence-server --zone cross_region_cluster2 start\n```\nRight now the only difference is at clusterGroupMetadata.clusterRedirectionPolicy.\nIn multiple region setup, network communication overhead between clusters is high so should use \"selected-apis-forwarding\". workflow/activity workers need to be connected to each cluster to keep high availability.\n\n\nArchiver\n--------\n\nArchiver is used to handle archival of workflow execution histories. It does this by hosting a cadence client worker\nand running an archival system workflow. The archival client gets used to initiate archival through signal sending. The archiver\nshards work across several workflows.\n"
  },
  {
    "path": "service/worker/archiver/activities.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\n\t\"github.com/uber/cadence/common\"\n\tcarchiver \"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\tuploadHistoryActivityFnName     = \"uploadHistoryActivity\"\n\tdeleteHistoryActivityFnName     = \"deleteHistoryActivity\"\n\tarchiveVisibilityActivityFnName = \"archiveVisibilityActivity\"\n)\n\nvar (\n\terrUploadNonRetriable            = errors.New(\"upload non-retriable error\")\n\terrDeleteNonRetriable            = errors.New(\"delete non-retriable error\")\n\terrArchiveVisibilityNonRetriable = errors.New(\"archive visibility non-retriable error\")\n\n\tuploadHistoryActivityNonRetryableErrors = []string{\"cadenceInternal:Panic\", errUploadNonRetriable.Error()}\n\tdeleteHistoryActivityNonRetryableErrors = []string{\"cadenceInternal:Panic\", errDeleteNonRetriable.Error()}\n)\n\nfunc uploadHistoryActivity(ctx context.Context, request ArchiveRequest) (err error) {\n\tcontainer := ctx.Value(bootstrapContainerKey).(*BootstrapContainer)\n\tscope := container.MetricsClient.Scope(metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(request.DomainName))\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif err.Error() == errUploadNonRetriable.Error() {\n\t\t\t\tscope.IncCounter(metrics.ArchiverNonRetryableErrorCount)\n\t\t\t}\n\t\t\terr = cadence.NewCustomError(err.Error())\n\t\t}\n\t}()\n\tlogger := tagLoggerWithHistoryRequest(tagLoggerWithActivityInfo(container.Logger, activity.GetInfo(ctx)), &request)\n\tURI, err := carchiver.NewURI(request.URI)\n\tif err != nil {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(\"failed to get history archival uri\"), tag.ArchivalURI(request.URI), tag.Error(err))\n\t\treturn errUploadNonRetriable\n\t}\n\thistoryArchiver, err := container.ArchiverProvider.GetHistoryArchiver(URI.Scheme(), service.Worker)\n\tif err != nil {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(\"failed to get history archiver\"), tag.Error(err))\n\t\treturn errUploadNonRetriable\n\t}\n\n\tallowArchivingIncompleteHistoryOpt := carchiver.GetArchivingIncompleteHistoryOption(container.Config.AllowArchivingIncompleteHistory)\n\terr = historyArchiver.Archive(ctx, URI, &carchiver.ArchiveHistoryRequest{\n\t\tShardID:              request.ShardID,\n\t\tDomainID:             request.DomainID,\n\t\tDomainName:           request.DomainName,\n\t\tWorkflowID:           request.WorkflowID,\n\t\tRunID:                request.RunID,\n\t\tBranchToken:          request.BranchToken,\n\t\tNextEventID:          request.NextEventID,\n\t\tCloseFailoverVersion: request.CloseFailoverVersion,\n\t}, carchiver.GetHeartbeatArchiveOption(), carchiver.GetNonRetriableErrorOption(errUploadNonRetriable), allowArchivingIncompleteHistoryOpt)\n\tif err == nil {\n\t\treturn nil\n\t}\n\tif errors.Is(err, errUploadNonRetriable) {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg,\n\t\t\ttag.ArchivalArchiveFailReason(\"got non-retryable error from history archiver\"),\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn errUploadNonRetriable\n\t}\n\tlogger.Error(carchiver.ArchiveTransientErrorMsg,\n\t\ttag.ArchivalArchiveFailReason(\"got retryable error from history archiver\"),\n\t\ttag.Error(err),\n\t)\n\treturn err\n}\n\nfunc deleteHistoryActivity(ctx context.Context, request ArchiveRequest) (err error) {\n\tcontainer := ctx.Value(bootstrapContainerKey).(*BootstrapContainer)\n\tscope := container.MetricsClient.Scope(metrics.ArchiverDeleteHistoryActivityScope, metrics.DomainTag(request.DomainName))\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif err.Error() == errDeleteNonRetriable.Error() {\n\t\t\t\tscope.IncCounter(metrics.ArchiverNonRetryableErrorCount)\n\t\t\t}\n\t\t\terr = cadence.NewCustomError(err.Error())\n\t\t}\n\t}()\n\terr = container.HistoryV2Manager.DeleteHistoryBranch(ctx, &persistence.DeleteHistoryBranchRequest{\n\t\tBranchToken: request.BranchToken,\n\t\tShardID:     common.IntPtr(request.ShardID),\n\t\tDomainName:  request.DomainName,\n\t})\n\tif err == nil {\n\t\treturn nil\n\t}\n\tlogger := tagLoggerWithHistoryRequest(tagLoggerWithActivityInfo(container.Logger, activity.GetInfo(ctx)), &request)\n\tlogger.Error(\"failed to delete history events\", tag.Error(err))\n\tif !persistence.IsTransientError(err) {\n\t\treturn errDeleteNonRetriable\n\t}\n\treturn err\n}\n\nfunc archiveVisibilityActivity(ctx context.Context, request ArchiveRequest) (err error) {\n\tcontainer := ctx.Value(bootstrapContainerKey).(*BootstrapContainer)\n\tscope := container.MetricsClient.Scope(metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(request.DomainName))\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer func() {\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tif err.Error() == errArchiveVisibilityNonRetriable.Error() {\n\t\t\t\tscope.IncCounter(metrics.ArchiverNonRetryableErrorCount)\n\t\t\t}\n\t\t\terr = cadence.NewCustomError(err.Error())\n\t\t}\n\t}()\n\tlogger := tagLoggerWithVisibilityRequest(tagLoggerWithActivityInfo(container.Logger, activity.GetInfo(ctx)), &request)\n\tURI, err := carchiver.NewURI(request.VisibilityURI)\n\tif err != nil {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(\"failed to get visibility archival uri\"), tag.ArchivalURI(request.VisibilityURI), tag.Error(err))\n\t\treturn errArchiveVisibilityNonRetriable\n\t}\n\tvisibilityArchiver, err := container.ArchiverProvider.GetVisibilityArchiver(URI.Scheme(), service.Worker)\n\tif err != nil {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(\"failed to get visibility archiver\"), tag.Error(err))\n\t\treturn errArchiveVisibilityNonRetriable\n\t}\n\terr = visibilityArchiver.Archive(ctx, URI, &carchiver.ArchiveVisibilityRequest{\n\t\tDomainID:           request.DomainID,\n\t\tDomainName:         request.DomainName,\n\t\tWorkflowID:         request.WorkflowID,\n\t\tRunID:              request.RunID,\n\t\tWorkflowTypeName:   request.WorkflowTypeName,\n\t\tStartTimestamp:     request.StartTimestamp,\n\t\tExecutionTimestamp: request.ExecutionTimestamp,\n\t\tCloseTimestamp:     request.CloseTimestamp,\n\t\tCloseStatus:        request.CloseStatus,\n\t\tHistoryLength:      request.HistoryLength,\n\t\tMemo:               request.Memo,\n\t\tSearchAttributes:   convertSearchAttributesToString(request.SearchAttributes),\n\t\tHistoryArchivalURI: request.URI,\n\t}, carchiver.GetNonRetriableErrorOption(errArchiveVisibilityNonRetriable))\n\tif err == nil {\n\t\treturn nil\n\t}\n\tif err.Error() == errArchiveVisibilityNonRetriable.Error() {\n\t\tlogger.Error(carchiver.ArchiveNonRetriableErrorMsg, tag.ArchivalArchiveFailReason(\"got non-retryable error from visibility archiver\"))\n\t\treturn errArchiveVisibilityNonRetriable\n\t}\n\tlogger.Error(carchiver.ArchiveTransientErrorMsg, tag.ArchivalArchiveFailReason(\"got retryable error from visibility archiver\"), tag.Error(err))\n\treturn err\n}\n"
  },
  {
    "path": "service/worker/archiver/activities_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\tcarchiver \"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\ttestDomainID             = \"test-domain-id\"\n\ttestDomainName           = \"test-domain-name\"\n\ttestWorkflowID           = \"test-workflow-id\"\n\ttestRunID                = \"test-run-id\"\n\ttestNextEventID          = 1800\n\ttestCloseFailoverVersion = 100\n\ttestScheme               = \"testScheme\"\n\ttestArchivalURI          = testScheme + \"://history/archival\"\n)\n\nvar (\n\ttestBranchToken = []byte{1, 2, 3}\n\n\terrPersistenceNonRetryable = errors.New(\"persistence non-retryable error\")\n)\n\ntype activitiesSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\n\tcontroller         *gomock.Controller\n\tlogger             log.Logger\n\tmetricsClient      *mmocks.Client\n\tmetricsScope       *mmocks.Scope\n\tarchiverProvider   *provider.MockArchiverProvider\n\thistoryArchiver    *carchiver.HistoryArchiverMock\n\tvisibilityArchiver *carchiver.VisibilityArchiverMock\n}\n\nfunc TestActivitiesSuite(t *testing.T) {\n\tsuite.Run(t, new(activitiesSuite))\n}\n\nfunc (s *activitiesSuite) SetupTest() {\n\ts.controller = gomock.NewController(s.T())\n\ts.logger = testlogger.New(s.T())\n\ts.metricsClient = &mmocks.Client{}\n\ts.metricsScope = &mmocks.Scope{}\n\ts.archiverProvider = provider.NewMockArchiverProvider(s.controller)\n\ts.historyArchiver = &carchiver.HistoryArchiverMock{}\n\ts.visibilityArchiver = &carchiver.VisibilityArchiverMock{}\n\ts.metricsScope.On(\"StartTimer\", metrics.CadenceLatency).Return(metrics.NewTestStopwatch()).Maybe()\n\ts.metricsScope.On(\"RecordTimer\", mock.Anything, mock.Anything).Maybe()\n}\n\nfunc (s *activitiesSuite) TearDownTest() {\n\ts.metricsClient.AssertExpectations(s.T())\n\ts.metricsScope.AssertExpectations(s.T())\n\ts.historyArchiver.AssertExpectations(s.T())\n\ts.visibilityArchiver.AssertExpectations(s.T())\n}\n\nfunc (s *activitiesSuite) TestUploadHistory_Fail_InvalidURI() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\tcontainer := &BootstrapContainer{\n\t\tLogger:        s.logger,\n\t\tMetricsClient: s.metricsClient,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  \"some invalid URI without scheme\",\n\t}\n\t_, err := env.ExecuteActivity(uploadHistoryActivity, request)\n\ts.Equal(errUploadNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestUploadHistory_Fail_GetArchiverError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), service.Worker).Return(nil, errors.New(\"failed to get archiver\"))\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(uploadHistoryActivity, request)\n\ts.Equal(errUploadNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestUploadHistory_Fail_ArchiveNonRetriableError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errUploadNonRetriable)\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), service.Worker).Return(s.historyArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(uploadHistoryActivity, request)\n\ts.Equal(errUploadNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestUploadHistory_Fail_ArchiveRetriableError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ttestArchiveErr := errors.New(\"some transient error\")\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(testArchiveErr)\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), service.Worker).Return(s.historyArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(uploadHistoryActivity, request)\n\ts.Equal(testArchiveErr.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestUploadHistory_Success() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverUploadHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil)\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), service.Worker).Return(s.historyArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(uploadHistoryActivity, request)\n\ts.NoError(err)\n}\n\nfunc (s *activitiesSuite) TestDeleteHistoryActivity_Fail_DeleteFromV2NonRetryableError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverDeleteHistoryActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\tmockHistoryV2Manager := &mocks.HistoryV2Manager{}\n\tmockHistoryV2Manager.On(\"DeleteHistoryBranch\", mock.Anything, mock.Anything).Return(errPersistenceNonRetryable)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tHistoryV2Manager: mockHistoryV2Manager,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:             testDomainID,\n\t\tDomainName:           testDomainName,\n\t\tWorkflowID:           testWorkflowID,\n\t\tRunID:                testRunID,\n\t\tBranchToken:          testBranchToken,\n\t\tNextEventID:          testNextEventID,\n\t\tCloseFailoverVersion: testCloseFailoverVersion,\n\t\tURI:                  testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(deleteHistoryActivity, request)\n\ts.Equal(errDeleteNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestArchiveVisibilityActivity_Fail_InvalidURI() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\tcontainer := &BootstrapContainer{\n\t\tLogger:        s.logger,\n\t\tMetricsClient: s.metricsClient,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:      testDomainID,\n\t\tDomainName:    testDomainName,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tVisibilityURI: \"some invalid URI without scheme\",\n\t}\n\t_, err := env.ExecuteActivity(archiveVisibilityActivity, request)\n\ts.Equal(errArchiveVisibilityNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestArchiveVisibilityActivity_Fail_GetArchiverError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), service.Worker).Return(nil, errors.New(\"failed to get archiver\"))\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:      testDomainID,\n\t\tDomainName:    testDomainName,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tVisibilityURI: testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(archiveVisibilityActivity, request)\n\ts.Equal(errArchiveVisibilityNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestArchiveVisibilityActivity_Fail_ArchiveNonRetriableError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.metricsScope.On(\"IncCounter\", metrics.ArchiverNonRetryableErrorCount).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errArchiveVisibilityNonRetriable)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), service.Worker).Return(s.visibilityArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:      testDomainID,\n\t\tDomainName:    testDomainName,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tVisibilityURI: testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(archiveVisibilityActivity, request)\n\ts.Equal(errArchiveVisibilityNonRetriable.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestArchiveVisibilityActivity_Fail_ArchiveRetriableError() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ttestArchiveErr := errors.New(\"some transient error\")\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(testArchiveErr)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), service.Worker).Return(s.visibilityArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:      testDomainID,\n\t\tDomainName:    testDomainName,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tVisibilityURI: testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(archiveVisibilityActivity, request)\n\ts.Equal(testArchiveErr.Error(), err.Error())\n}\n\nfunc (s *activitiesSuite) TestArchiveVisibilityActivity_Success() {\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverArchiveVisibilityActivityScope, metrics.DomainTag(testDomainName)).Return(s.metricsScope).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), service.Worker).Return(s.visibilityArchiver, nil)\n\tcontainer := &BootstrapContainer{\n\t\tLogger:           s.logger,\n\t\tMetricsClient:    s.metricsClient,\n\t\tArchiverProvider: s.archiverProvider,\n\t\tConfig: &Config{\n\t\t\tAllowArchivingIncompleteHistory: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t}\n\tenv := s.NewTestActivityEnvironment()\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), bootstrapContainerKey, container),\n\t})\n\trequest := ArchiveRequest{\n\t\tDomainID:      testDomainID,\n\t\tDomainName:    testDomainName,\n\t\tWorkflowID:    testWorkflowID,\n\t\tRunID:         testRunID,\n\t\tVisibilityURI: testArchivalURI,\n\t}\n\t_, err := env.ExecuteActivity(archiveVisibilityActivity, request)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "service/worker/archiver/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tcclient \"go.uber.org/cadence/client\"\n\n\tcarchiver \"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/quotas\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n//go:generate mockgen -package=$GOPACKAGE -destination=client_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Client\n\ntype (\n\t// ClientRequest is the archive request sent to the archiver client\n\tClientRequest struct {\n\t\tArchiveRequest       *ArchiveRequest\n\t\tCallerService        string\n\t\tAttemptArchiveInline bool\n\t}\n\n\t// ClientResponse is the archive response returned from the archiver client\n\tClientResponse struct {\n\t\tHistoryArchivedInline bool\n\t}\n\n\t// ArchiveRequest is the request signal sent to the archival workflow\n\tArchiveRequest struct {\n\t\tDomainID   string\n\t\tDomainName string\n\t\tWorkflowID string\n\t\tRunID      string\n\n\t\t// history archival\n\t\tShardID              int\n\t\tBranchToken          []byte\n\t\tNextEventID          int64\n\t\tCloseFailoverVersion int64\n\t\tURI                  string // should be historyURI, but keep the existing name for backward compatibility\n\n\t\t// visibility archival\n\t\tWorkflowTypeName   string\n\t\tStartTimestamp     int64\n\t\tExecutionTimestamp int64\n\t\tCloseTimestamp     int64\n\t\tCloseStatus        types.WorkflowExecutionCloseStatus\n\t\tHistoryLength      int64\n\t\tMemo               *types.Memo\n\t\tSearchAttributes   map[string][]byte\n\t\tVisibilityURI      string\n\n\t\t// archival targets: history and/or visibility\n\t\tTargets []ArchivalTarget\n\t}\n\n\t// Client is used to archive workflow histories\n\tClient interface {\n\t\tArchive(context.Context, *ClientRequest) (*ClientResponse, error)\n\t}\n\n\tclient struct {\n\t\tmetricsScope                metrics.Scope\n\t\tlogger                      log.Logger\n\t\tcadenceClient               cclient.Client\n\t\tnumWorkflows                dynamicproperties.IntPropertyFn\n\t\trateLimiter                 quotas.Limiter\n\t\tinlineHistoryRateLimiter    quotas.Limiter\n\t\tinlineVisibilityRateLimiter quotas.Limiter\n\t\tarchiverProvider            provider.ArchiverProvider\n\t\tarchivingIncompleteHistory  dynamicproperties.BoolPropertyFn\n\t}\n\n\t// ArchivalTarget is either history or visibility\n\tArchivalTarget int\n)\n\nconst (\n\tsignalTimeout = 300 * time.Millisecond\n\n\ttooManyRequestsErrMsg = \"too many requests to archival workflow\"\n)\n\nconst (\n\t// ArchiveTargetHistory is the archive target for workflow history\n\tArchiveTargetHistory ArchivalTarget = iota\n\t// ArchiveTargetVisibility is the archive target for workflow visibility record\n\tArchiveTargetVisibility\n)\n\nvar (\n\terrInlineArchivalThrottled = errors.New(\"inline archival throttled\")\n)\n\n// NewClient creates a new Client\nfunc NewClient(\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tpublicClient workflowserviceclient.Interface,\n\tnumWorkflows dynamicproperties.IntPropertyFn,\n\trequestRateLimiter quotas.Limiter,\n\tinlineHistoryRateLimiter quotas.Limiter,\n\tinlineVisibilityRateLimiter quotas.Limiter,\n\tarchiverProvider provider.ArchiverProvider,\n\tarchivingIncompleteHistory dynamicproperties.BoolPropertyFn,\n) Client {\n\treturn &client{\n\t\tmetricsScope:                metricsClient.Scope(metrics.ArchiverClientScope),\n\t\tlogger:                      logger,\n\t\tcadenceClient:               cclient.NewClient(publicClient, constants.SystemLocalDomainName, &cclient.Options{}),\n\t\tnumWorkflows:                numWorkflows,\n\t\trateLimiter:                 requestRateLimiter,\n\t\tinlineHistoryRateLimiter:    inlineHistoryRateLimiter,\n\t\tinlineVisibilityRateLimiter: inlineVisibilityRateLimiter,\n\t\tarchiverProvider:            archiverProvider,\n\t\tarchivingIncompleteHistory:  archivingIncompleteHistory,\n\t}\n}\n\n// Archive starts an archival task\nfunc (c *client) Archive(ctx context.Context, request *ClientRequest) (*ClientResponse, error) {\n\tscopeWithDomainTag := c.metricsScope.Tagged(metrics.DomainTag(request.ArchiveRequest.DomainName))\n\tfor _, target := range request.ArchiveRequest.Targets {\n\t\tswitch target {\n\t\tcase ArchiveTargetHistory:\n\t\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientHistoryRequestCountPerDomain)\n\t\tcase ArchiveTargetVisibility:\n\t\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientVisibilityRequestCountPerDomain)\n\t\t}\n\t}\n\tlogger := c.logger.WithTags(\n\t\ttag.ArchivalCallerServiceName(request.CallerService),\n\t\ttag.ArchivalArchiveAttemptedInline(request.AttemptArchiveInline),\n\t)\n\tresp := &ClientResponse{\n\t\tHistoryArchivedInline: false,\n\t}\n\tif request.AttemptArchiveInline {\n\t\tresults := []chan error{}\n\t\tfor _, target := range request.ArchiveRequest.Targets {\n\t\t\tch := make(chan error)\n\t\t\tresults = append(results, ch)\n\t\t\tswitch target {\n\t\t\tcase ArchiveTargetHistory:\n\t\t\t\tgo c.archiveHistoryInline(ctx, request, logger, ch)\n\t\t\tcase ArchiveTargetVisibility:\n\t\t\t\tgo c.archiveVisibilityInline(ctx, request, logger, ch)\n\t\t\tdefault:\n\t\t\t\tclose(ch)\n\t\t\t}\n\t\t}\n\n\t\ttargets := []ArchivalTarget{}\n\t\tfor i, target := range request.ArchiveRequest.Targets {\n\t\t\tif <-results[i] != nil {\n\t\t\t\ttargets = append(targets, target)\n\t\t\t} else if target == ArchiveTargetHistory {\n\t\t\t\tresp.HistoryArchivedInline = true\n\t\t\t}\n\t\t}\n\t\trequest.ArchiveRequest.Targets = targets\n\t}\n\tif len(request.ArchiveRequest.Targets) != 0 {\n\t\tif err := c.sendArchiveSignal(ctx, request.ArchiveRequest, logger); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn resp, nil\n}\n\nfunc (c *client) archiveHistoryInline(ctx context.Context, request *ClientRequest, logger log.Logger, errCh chan error) {\n\tlogger = tagLoggerWithHistoryRequest(logger, request.ArchiveRequest)\n\tscopeWithDomainTag := c.metricsScope.Tagged(metrics.DomainTag(request.ArchiveRequest.DomainName))\n\tif !c.inlineHistoryRateLimiter.Allow() {\n\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientHistoryInlineArchiveThrottledCountPerDomain)\n\t\tlogger.Debug(\"inline history archival throttled\")\n\t\terrCh <- errInlineArchivalThrottled\n\t\treturn\n\t}\n\n\tvar err error\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientHistoryInlineArchiveFailureCountPerDomain)\n\t\t\tlogger.Info(\"failed to perform workflow history archival inline\", tag.Error(err))\n\t\t}\n\t\terrCh <- err\n\t}()\n\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain)\n\tURI, err := carchiver.NewURI(request.ArchiveRequest.URI)\n\tif err != nil {\n\t\treturn\n\t}\n\n\thistoryArchiver, err := c.archiverProvider.GetHistoryArchiver(URI.Scheme(), request.CallerService)\n\tif err != nil {\n\t\treturn\n\t}\n\n\tallowArchivingIncompleteHistoryOpt := carchiver.GetArchivingIncompleteHistoryOption(c.archivingIncompleteHistory)\n\terr = historyArchiver.Archive(ctx, URI, &carchiver.ArchiveHistoryRequest{\n\t\tShardID:              request.ArchiveRequest.ShardID,\n\t\tDomainID:             request.ArchiveRequest.DomainID,\n\t\tDomainName:           request.ArchiveRequest.DomainName,\n\t\tWorkflowID:           request.ArchiveRequest.WorkflowID,\n\t\tRunID:                request.ArchiveRequest.RunID,\n\t\tBranchToken:          request.ArchiveRequest.BranchToken,\n\t\tNextEventID:          request.ArchiveRequest.NextEventID,\n\t\tCloseFailoverVersion: request.ArchiveRequest.CloseFailoverVersion,\n\t}, allowArchivingIncompleteHistoryOpt)\n}\n\nfunc (c *client) archiveVisibilityInline(ctx context.Context, request *ClientRequest, logger log.Logger, errCh chan error) {\n\tlogger = tagLoggerWithVisibilityRequest(logger, request.ArchiveRequest)\n\tscopeWithDomainTag := c.metricsScope.Tagged(metrics.DomainTag(request.ArchiveRequest.DomainName))\n\tif !c.inlineVisibilityRateLimiter.Allow() {\n\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientVisibilityInlineArchiveThrottledCountPerDomain)\n\t\tlogger.Debug(\"inline visibility archival throttled\")\n\t\terrCh <- errInlineArchivalThrottled\n\t\treturn\n\t}\n\n\tvar err error\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientVisibilityInlineArchiveFailureCountPerDomain)\n\t\t\tlogger.Info(\"failed to perform visibility archival inline\", tag.Error(err))\n\t\t}\n\t\terrCh <- err\n\t}()\n\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain)\n\tURI, err := carchiver.NewURI(request.ArchiveRequest.VisibilityURI)\n\tif err != nil {\n\t\treturn\n\t}\n\n\tvisibilityArchiver, err := c.archiverProvider.GetVisibilityArchiver(URI.Scheme(), request.CallerService)\n\tif err != nil {\n\t\treturn\n\t}\n\n\terr = visibilityArchiver.Archive(ctx, URI, &carchiver.ArchiveVisibilityRequest{\n\t\tDomainID:           request.ArchiveRequest.DomainID,\n\t\tDomainName:         request.ArchiveRequest.DomainName,\n\t\tWorkflowID:         request.ArchiveRequest.WorkflowID,\n\t\tRunID:              request.ArchiveRequest.RunID,\n\t\tWorkflowTypeName:   request.ArchiveRequest.WorkflowTypeName,\n\t\tStartTimestamp:     request.ArchiveRequest.StartTimestamp,\n\t\tExecutionTimestamp: request.ArchiveRequest.ExecutionTimestamp,\n\t\tCloseTimestamp:     request.ArchiveRequest.CloseTimestamp,\n\t\tCloseStatus:        request.ArchiveRequest.CloseStatus,\n\t\tHistoryLength:      request.ArchiveRequest.HistoryLength,\n\t\tMemo:               request.ArchiveRequest.Memo,\n\t\tSearchAttributes:   convertSearchAttributesToString(request.ArchiveRequest.SearchAttributes),\n\t\tHistoryArchivalURI: request.ArchiveRequest.URI,\n\t})\n}\n\nfunc (c *client) sendArchiveSignal(ctx context.Context, request *ArchiveRequest, taggedLogger log.Logger) error {\n\tscopeWithDomainTag := c.metricsScope.Tagged(metrics.DomainTag(request.DomainName))\n\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientSendSignalCountPerDomain)\n\tif ok := c.rateLimiter.Allow(); !ok {\n\t\tc.logger.Error(tooManyRequestsErrMsg)\n\t\tscopeWithDomainTag.IncCounter(metrics.CadenceErrServiceBusyCounter)\n\t\treturn errors.New(tooManyRequestsErrMsg)\n\t}\n\n\tworkflowID := fmt.Sprintf(\"%v-%v\", workflowIDPrefix, rand.Intn(c.numWorkflows()))\n\tworkflowOptions := cclient.StartWorkflowOptions{\n\t\tID:                              workflowID,\n\t\tTaskList:                        decisionTaskList,\n\t\tExecutionStartToCloseTimeout:    workflowStartToCloseTimeout,\n\t\tDecisionTaskStartToCloseTimeout: workflowTaskStartToCloseTimeout,\n\t\tWorkflowIDReusePolicy:           cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n\tsignalCtx, cancel := context.WithTimeout(context.Background(), signalTimeout)\n\tdefer cancel()\n\t_, err := c.cadenceClient.SignalWithStartWorkflow(signalCtx, workflowID, signalName, *request, workflowOptions, archivalWorkflowFnName, nil)\n\tif err != nil {\n\t\ttaggedLogger = taggedLogger.WithTags(\n\t\t\ttag.ArchivalRequestDomainID(request.DomainID),\n\t\t\ttag.ArchivalRequestDomainName(request.DomainName),\n\t\t\ttag.ArchivalRequestWorkflowID(request.WorkflowID),\n\t\t\ttag.ArchivalRequestRunID(request.RunID),\n\t\t\ttag.WorkflowID(workflowID),\n\t\t\ttag.Error(err),\n\t\t)\n\t\ttaggedLogger.Error(\"failed to send signal to archival system workflow\")\n\t\tscopeWithDomainTag.IncCounter(metrics.ArchiverClientSendSignalFailureCountPerDomain)\n\t\treturn err\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/archiver/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/worker/archiver (interfaces: Client)\n//\n// Generated by this command:\n//\n//\tmockgen -package=archiver -destination=client_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Client\n//\n\n// Package archiver is a generated GoMock package.\npackage archiver\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// Archive mocks base method.\nfunc (m *MockClient) Archive(arg0 context.Context, arg1 *ClientRequest) (*ClientResponse, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Archive\", arg0, arg1)\n\tret0, _ := ret[0].(*ClientResponse)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Archive indicates an expected call of Archive.\nfunc (mr *MockClientMockRecorder) Archive(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Archive\", reflect.TypeOf((*MockClient)(nil).Archive), arg0, arg1)\n}\n"
  },
  {
    "path": "service/worker/archiver/client_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/mocks\"\n\t\"go.uber.org/mock/gomock\"\n\t\"golang.org/x/time/rate\"\n\n\tcarchiver \"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n)\n\ntype clientSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tcontroller         *gomock.Controller\n\tarchiverProvider   *provider.MockArchiverProvider\n\thistoryArchiver    *carchiver.HistoryArchiverMock\n\tvisibilityArchiver *carchiver.VisibilityArchiverMock\n\tmetricsClient      *mmocks.Client\n\tmetricsScope       *mmocks.Scope\n\tcadenceClient      *mocks.Client\n\tclient             *client\n}\n\nfunc TestClientSuite(t *testing.T) {\n\tsuite.Run(t, new(clientSuite))\n}\n\nfunc (s *clientSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\ts.archiverProvider = provider.NewMockArchiverProvider(s.controller)\n\ts.historyArchiver = &carchiver.HistoryArchiverMock{}\n\ts.visibilityArchiver = &carchiver.VisibilityArchiverMock{}\n\ts.metricsClient = &mmocks.Client{}\n\ts.metricsScope = &mmocks.Scope{}\n\ts.cadenceClient = &mocks.Client{}\n\ts.metricsClient.On(\"Scope\", metrics.ArchiverClientScope, mock.Anything).Return(s.metricsScope).Once()\n\ts.client = NewClient(\n\t\ts.metricsClient,\n\t\tlog.NewNoop(),\n\t\tnil,\n\t\tdynamicproperties.GetIntPropertyFn(1000),\n\t\tclock.NewRatelimiter(rate.Limit(1000), 1),\n\t\tclock.NewRatelimiter(rate.Limit(1), 1),\n\t\tclock.NewRatelimiter(rate.Limit(1), 1),\n\t\ts.archiverProvider,\n\t\tdynamicproperties.GetBoolPropertyFn(false),\n\t).(*client)\n\ts.client.cadenceClient = s.cadenceClient\n}\n\nfunc (s *clientSuite) TearDownTest() {\n\ts.historyArchiver.AssertExpectations(s.T())\n\ts.visibilityArchiver.AssertExpectations(s.T())\n\ts.metricsClient.AssertExpectations(s.T())\n\ts.metricsScope.AssertExpectations(s.T())\n}\n\nfunc (s *clientSuite) TestArchiveVisibilityInlineSuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveVisibilityInlineThrottled() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Times(2)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveThrottledCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetVisibility\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tfor i := 0; i < 2; i++ {\n\t\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\t\tArchiveRequest: &ArchiveRequest{\n\t\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\t\tTargets:       []ArchivalTarget{ArchiveTargetVisibility},\n\t\t\t\tDomainName:    \"test_domain_name\",\n\t\t\t},\n\t\t\tAttemptArchiveInline: true,\n\t\t})\n\t\ts.NoError(err)\n\t\ts.NotNil(resp)\n\t}\n}\n\nfunc (s *clientSuite) TestArchiveVisibilityInlineFail_SendSignalSuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetVisibility\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveVisibilityInlineFail_SendSignalFail() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalFailureCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetVisibility\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, errors.New(\"some random error\"))\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.Error(err)\n\ts.Nil(resp)\n}\n\nfunc (s *clientSuite) TestArchiveHistoryInlineSuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:        \"test:///history/archival\",\n\t\t\tTargets:    []ArchivalTarget{ArchiveTargetHistory},\n\t\t\tDomainName: \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.True(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveHistoryInlineThrottled() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Times(2)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveThrottledCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetHistory\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tfor i := 0; i < 2; i++ {\n\t\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\t\tArchiveRequest: &ArchiveRequest{\n\t\t\t\tURI:        \"test:///history/archival\",\n\t\t\t\tTargets:    []ArchivalTarget{ArchiveTargetHistory},\n\t\t\t\tDomainName: \"test_domain_name\",\n\t\t\t},\n\t\t\tAttemptArchiveInline: true,\n\t\t})\n\t\ts.NoError(err)\n\t\ts.NotNil(resp)\n\t}\n}\n\nfunc (s *clientSuite) TestArchiveHistoryInlineFail_SendSignalSuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetHistory\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:        \"test:///history/archival\",\n\t\t\tTargets:    []ArchivalTarget{ArchiveTargetHistory},\n\t\t\tDomainName: \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveHistoryInlineFail_SendSignalFail() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalFailureCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetHistory\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, errors.New(\"some random error\"))\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:        \"test:///history/archival\",\n\t\t\tTargets:    []ArchivalTarget{ArchiveTargetHistory},\n\t\t\tDomainName: \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.Error(err)\n\ts.Nil(resp)\n}\n\nfunc (s *clientSuite) TestArchiveInline_HistoryFail_VisibilitySuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetHistory\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:           \"test:///history/archival\",\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveInline_VisibilityFail_HistorySuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 1 && v.Targets[0] == ArchiveTargetVisibility\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:           \"test:///history/archival\",\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.True(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveInline_VisibilityFail_HistoryFail() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything).Return(errors.New(\"some random error\")).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveFailureCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 2\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:           \"test:///history/archival\",\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveInline_VisibilitySuccess_HistorySuccess() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.archiverProvider.EXPECT().GetHistoryArchiver(gomock.Any(), gomock.Any()).Return(s.historyArchiver, nil).Times(1)\n\ts.archiverProvider.EXPECT().GetVisibilityArchiver(gomock.Any(), gomock.Any()).Return(s.visibilityArchiver, nil).Times(1)\n\ts.historyArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.visibilityArchiver.On(\"Archive\", mock.Anything, mock.Anything, mock.Anything).Return(nil).Once()\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryInlineArchiveAttemptCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityInlineArchiveAttemptCountPerDomain).Once()\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:           \"test:///history/archival\",\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.True(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveSendSignal_Success() {\n\tscopeDomain := &mmocks.Scope{}\n\ts.cadenceClient.On(\"SignalWithStartWorkflow\", mock.Anything, mock.Anything, mock.Anything, mock.MatchedBy(func(v ArchiveRequest) bool {\n\t\treturn len(v.Targets) == 2\n\t}), mock.Anything, mock.Anything, mock.Anything).Return(nil, nil)\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(scopeDomain)\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientHistoryRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientVisibilityRequestCountPerDomain).Once()\n\tscopeDomain.On(\"IncCounter\", metrics.ArchiverClientSendSignalCountPerDomain).Once()\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tURI:           \"test:///history/archival\",\n\t\t\tVisibilityURI: \"test:///visibility/archival\",\n\t\t\tTargets:       []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\tDomainName:    \"test_domain_name\",\n\t\t},\n\t\tAttemptArchiveInline: false,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n\nfunc (s *clientSuite) TestArchiveUnknownTarget() {\n\ts.metricsScope.On(\"Tagged\", mock.Anything).Return(&mmocks.Scope{})\n\tresp, err := s.client.Archive(context.Background(), &ClientRequest{\n\t\tArchiveRequest: &ArchiveRequest{\n\t\t\tTargets: []ArchivalTarget{3},\n\t\t},\n\t\tAttemptArchiveInline: true,\n\t})\n\ts.NoError(err)\n\ts.NotNil(resp)\n\ts.False(resp.HistoryArchivedInline)\n}\n"
  },
  {
    "path": "service/worker/archiver/client_worker.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\t// ClientWorker is a cadence client worker\n\tClientWorker interface {\n\t\tStart() error\n\t\tStop()\n\t}\n\n\tclientWorker struct {\n\t\tworker      worker.Worker\n\t\tdomainCache cache.DomainCache\n\t}\n\n\t// BootstrapContainer contains everything need for bootstrapping\n\tBootstrapContainer struct {\n\t\tPublicClient     workflowserviceclient.Interface\n\t\tMetricsClient    metrics.Client\n\t\tLogger           log.Logger\n\t\tHistoryV2Manager persistence.HistoryManager\n\t\tDomainCache      cache.DomainCache\n\t\tConfig           *Config\n\t\tArchiverProvider provider.ArchiverProvider\n\t}\n\n\t// Config for ClientWorker\n\tConfig struct {\n\t\tArchiverConcurrency             dynamicproperties.IntPropertyFn\n\t\tArchivalsPerIteration           dynamicproperties.IntPropertyFn\n\t\tTimeLimitPerArchivalIteration   dynamicproperties.DurationPropertyFn\n\t\tAllowArchivingIncompleteHistory dynamicproperties.BoolPropertyFn\n\t}\n\n\tcontextKey int\n)\n\nconst (\n\tworkflowIDPrefix                = \"cadence-archival\"\n\tdecisionTaskList                = \"cadence-archival-tl\"\n\tsignalName                      = \"cadence-archival-signal\"\n\tarchivalWorkflowFnName          = \"archivalWorkflow\"\n\tworkflowStartToCloseTimeout     = time.Hour * 24 * 30\n\tworkflowTaskStartToCloseTimeout = time.Minute\n\n\tbootstrapContainerKey contextKey = iota\n)\n\n// these globals exist as a work around because no primitive exists to pass such objects to workflow code\nvar (\n\tglobalLogger        log.Logger\n\tglobalMetricsClient metrics.Client\n\tglobalConfig        *Config\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(archivalWorkflow, workflow.RegisterOptions{Name: archivalWorkflowFnName})\n\tactivity.RegisterWithOptions(uploadHistoryActivity, activity.RegisterOptions{Name: uploadHistoryActivityFnName})\n\tactivity.RegisterWithOptions(deleteHistoryActivity, activity.RegisterOptions{Name: deleteHistoryActivityFnName})\n\tactivity.RegisterWithOptions(archiveVisibilityActivity, activity.RegisterOptions{Name: archiveVisibilityActivityFnName})\n}\n\n// NewClientWorker returns a new ClientWorker\nfunc NewClientWorker(container *BootstrapContainer) ClientWorker {\n\tglobalLogger = container.Logger.WithTags(tag.ComponentArchiver, tag.WorkflowDomainName(constants.SystemLocalDomainName))\n\tglobalMetricsClient = container.MetricsClient\n\tglobalConfig = container.Config\n\tactCtx := context.WithValue(context.Background(), bootstrapContainerKey, container)\n\two := worker.Options{\n\t\tBackgroundActivityContext: actCtx,\n\t}\n\treturn &clientWorker{\n\t\tworker:      worker.New(container.PublicClient, constants.SystemLocalDomainName, decisionTaskList, wo),\n\t\tdomainCache: container.DomainCache,\n\t}\n}\n\n// Start the ClientWorker\nfunc (w *clientWorker) Start() error {\n\tif err := w.worker.Start(); err != nil {\n\t\tw.worker.Stop()\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// Stop the ClientWorker\nfunc (w *clientWorker) Stop() {\n\tw.worker.Stop()\n\tw.domainCache.Stop()\n}\n"
  },
  {
    "path": "service/worker/archiver/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n//go:generate mockgen -package=$GOPACKAGE -destination=handler_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Handler\n\ntype (\n\t// Handler is used to process archival requests\n\tHandler interface {\n\t\tStart()\n\t\tFinished() []uint64\n\t}\n\n\thandler struct {\n\t\tctx           workflow.Context\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tconcurrency   int\n\t\trequestCh     workflow.Channel\n\t\tresultCh      workflow.Channel\n\t}\n)\n\n// NewHandler returns a new Handler\nfunc NewHandler(\n\tctx workflow.Context,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconcurrency int,\n\trequestCh workflow.Channel,\n) Handler {\n\treturn &handler{\n\t\tctx:           ctx,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t\tconcurrency:   concurrency,\n\t\trequestCh:     requestCh,\n\t\tresultCh:      workflow.NewChannel(ctx),\n\t}\n}\n\n// Start spawns concurrency count of coroutine to handle archivals (does not block).\nfunc (h *handler) Start() {\n\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverStartedCount)\n\tfor i := 0; i < h.concurrency; i++ {\n\t\tworkflow.Go(h.ctx, func(ctx workflow.Context) {\n\t\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverCoroutineStartedCount)\n\t\t\tvar handledHashes []uint64\n\t\t\tfor {\n\t\t\t\tvar request ArchiveRequest\n\t\t\t\tif more := h.requestCh.Receive(ctx, &request); !more {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\th.handleRequest(ctx, &request)\n\t\t\t\thandledHashes = append(handledHashes, hash(request))\n\t\t\t}\n\t\t\th.resultCh.Send(ctx, handledHashes)\n\t\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverCoroutineStoppedCount)\n\t\t})\n\t}\n}\n\n// Finished will block until all work has been finished.\n// Returns hashes of requests handled.\nfunc (h *handler) Finished() []uint64 {\n\tvar handledHashes []uint64\n\tfor i := 0; i < h.concurrency; i++ {\n\t\tvar subResult []uint64\n\t\th.resultCh.Receive(h.ctx, &subResult)\n\t\thandledHashes = append(handledHashes, subResult...)\n\t}\n\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverStoppedCount)\n\treturn handledHashes\n}\n\nfunc (h *handler) handleRequest(ctx workflow.Context, request *ArchiveRequest) {\n\t// For backward compatibility\n\ttargets := request.Targets\n\tif len(targets) == 0 {\n\t\ttargets = append(targets, ArchiveTargetHistory)\n\t}\n\tpendingRequests := []workflow.Channel{}\n\tfor _, target := range targets {\n\t\tdoneCh := workflow.NewChannel(ctx)\n\t\tpendingRequests = append(pendingRequests, doneCh)\n\t\tswitch target {\n\t\tcase ArchiveTargetHistory:\n\t\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\t\th.handleHistoryRequest(ctx, request)\n\t\t\t\tdoneCh.Close()\n\t\t\t})\n\t\tcase ArchiveTargetVisibility:\n\t\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\t\th.handleVisibilityRequest(ctx, request)\n\t\t\t\tdoneCh.Close()\n\t\t\t})\n\t\tdefault:\n\t\t\tdoneCh.Close()\n\t\t}\n\t}\n\n\tfor _, doneCh := range pendingRequests {\n\t\tdoneCh.Receive(ctx, nil)\n\t}\n}\n\nfunc (h *handler) handleHistoryRequest(ctx workflow.Context, request *ArchiveRequest) {\n\tsw := h.metricsClient.StartTimer(metrics.ArchiverScope, metrics.ArchiverHandleHistoryRequestLatency)\n\tlogger := tagLoggerWithHistoryRequest(h.logger, request)\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\tStartToCloseTimeout:    1 * time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2.0,\n\t\t\tExpirationInterval:       5 * time.Minute,\n\t\t\tNonRetriableErrorReasons: uploadHistoryActivityNonRetryableErrors,\n\t\t},\n\t}\n\tactCtx := workflow.WithActivityOptions(ctx, ao)\n\tuploadSW := h.metricsClient.StartTimer(metrics.ArchiverScope, metrics.ArchiverUploadWithRetriesLatency)\n\terr := workflow.ExecuteActivity(actCtx, uploadHistoryActivityFnName, *request).Get(actCtx, nil)\n\tif err != nil {\n\t\tlogger.Error(\"failed to archive history, will move on to deleting history without archiving\", tag.Error(err))\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverUploadFailedAllRetriesCount)\n\t} else {\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverUploadSuccessCount)\n\t}\n\tuploadSW.Stop()\n\n\tlao := workflow.LocalActivityOptions{\n\t\tScheduleToCloseTimeout: 1 * time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2.0,\n\t\t\tExpirationInterval:       5 * time.Minute,\n\t\t\tNonRetriableErrorReasons: deleteHistoryActivityNonRetryableErrors,\n\t\t},\n\t}\n\tdeleteSW := h.metricsClient.StartTimer(metrics.ArchiverScope, metrics.ArchiverDeleteWithRetriesLatency)\n\tlocalActCtx := workflow.WithLocalActivityOptions(ctx, lao)\n\terr = workflow.ExecuteLocalActivity(localActCtx, deleteHistoryActivity, *request).Get(localActCtx, nil)\n\tif err != nil {\n\t\tlogger.Error(\"deleting history failed, this means zombie histories are left\", tag.Error(err))\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverDeleteFailedAllRetriesCount)\n\t} else {\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount)\n\t}\n\tdeleteSW.Stop()\n\tsw.Stop()\n}\n\nfunc (h *handler) handleVisibilityRequest(ctx workflow.Context, request *ArchiveRequest) {\n\tsw := h.metricsClient.StartTimer(metrics.ArchiverScope, metrics.ArchiverHandleVisibilityRequestLatency)\n\tlogger := tagLoggerWithVisibilityRequest(h.logger, request)\n\tao := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 1 * time.Minute,\n\t\tStartToCloseTimeout:    1 * time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       2.0,\n\t\t\tExpirationInterval:       5 * time.Minute,\n\t\t\tNonRetriableErrorReasons: uploadHistoryActivityNonRetryableErrors,\n\t\t},\n\t}\n\tactCtx := workflow.WithActivityOptions(ctx, ao)\n\terr := workflow.ExecuteActivity(actCtx, archiveVisibilityActivityFnName, *request).Get(actCtx, nil)\n\tif err != nil {\n\t\tlogger.Error(\"failed to archive workflow visibility record\", tag.Error(err))\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverHandleVisibilityFailedAllRetiresCount)\n\t} else {\n\t\th.metricsClient.IncCounter(metrics.ArchiverScope, metrics.ArchiverHandleVisibilitySuccessCount)\n\t}\n\tsw.Stop()\n}\n"
  },
  {
    "path": "service/worker/archiver/handler_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/worker/archiver (interfaces: Handler)\n//\n// Generated by this command:\n//\n//\tmockgen -package=archiver -destination=handler_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Handler\n//\n\n// Package archiver is a generated GoMock package.\npackage archiver\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockHandler is a mock of Handler interface.\ntype MockHandler struct {\n\tctrl     *gomock.Controller\n\trecorder *MockHandlerMockRecorder\n\tisgomock struct{}\n}\n\n// MockHandlerMockRecorder is the mock recorder for MockHandler.\ntype MockHandlerMockRecorder struct {\n\tmock *MockHandler\n}\n\n// NewMockHandler creates a new mock instance.\nfunc NewMockHandler(ctrl *gomock.Controller) *MockHandler {\n\tmock := &MockHandler{ctrl: ctrl}\n\tmock.recorder = &MockHandlerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockHandler) EXPECT() *MockHandlerMockRecorder {\n\treturn m.recorder\n}\n\n// Finished mocks base method.\nfunc (m *MockHandler) Finished() []uint64 {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Finished\")\n\tret0, _ := ret[0].([]uint64)\n\treturn ret0\n}\n\n// Finished indicates an expected call of Finished.\nfunc (mr *MockHandlerMockRecorder) Finished() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Finished\", reflect.TypeOf((*MockHandler)(nil).Finished))\n}\n\n// Start mocks base method.\nfunc (m *MockHandler) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockHandlerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockHandler)(nil).Start))\n}\n"
  },
  {
    "path": "service/worker/archiver/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n)\n\nvar (\n\thandlerTestMetrics *mmocks.Client\n\thandlerTestLogger  *log.MockLogger\n)\n\ntype handlerSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestHandlerSuite(t *testing.T) {\n\tsuite.Run(t, new(handlerSuite))\n}\n\nfunc (s *handlerSuite) SetupSuite() {\n\tworkflow.Register(handleHistoryRequestWorkflow)\n\tworkflow.Register(handleVisibilityRequestWorkflow)\n\tworkflow.Register(startAndFinishArchiverWorkflow)\n}\n\nfunc (s *handlerSuite) SetupTest() {\n\thandlerTestMetrics = &mmocks.Client{}\n\thandlerTestMetrics.On(\"StartTimer\", mock.Anything, mock.Anything).Return(metrics.NopStopwatch())\n\thandlerTestLogger = log.NewMockLogger(gomock.NewController(s.T()))\n\thandlerTestLogger.EXPECT().WithTags(gomock.Any()).Return(handlerTestLogger).AnyTimes()\n}\n\nfunc (s *handlerSuite) TearDownTest() {\n\thandlerTestMetrics.AssertExpectations(s.T())\n}\n\nfunc (s *handlerSuite) TestHandleHistoryRequest_UploadFails_NonRetryableError() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadFailedAllRetriesCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount).Once()\n\thandlerTestLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(errors.New(\"some random error\"))\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(handleHistoryRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleHistoryRequest_UploadFails_ExpireRetryTimeout() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadFailedAllRetriesCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount).Once()\n\thandlerTestLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\n\ttimeoutErr := workflow.NewTimeoutError(shared.TimeoutTypeStartToClose)\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(timeoutErr)\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(handleHistoryRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleHistoryRequest_UploadSuccess() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadSuccessCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(handleHistoryRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleHistoryRequest_DeleteFails_NonRetryableError() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadSuccessCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteFailedAllRetriesCount).Once()\n\thandlerTestLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(func(context.Context, ArchiveRequest) error {\n\t\treturn cadence.NewCustomError(errDeleteNonRetriable.Error())\n\t})\n\tenv.ExecuteWorkflow(handleHistoryRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleHistoryRequest_DeleteFailsThenSucceeds() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadSuccessCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tfirstRun := true\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(func(context.Context, ArchiveRequest) error {\n\t\tif firstRun {\n\t\t\tfirstRun = false\n\t\t\treturn errors.New(\"some retryable error\")\n\t\t}\n\t\treturn nil\n\t})\n\tenv.ExecuteWorkflow(handleHistoryRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleVisibilityRequest_Fail() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverHandleVisibilityFailedAllRetiresCount).Once()\n\thandlerTestLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(archiveVisibilityActivityFnName, mock.Anything, mock.Anything).Return(errors.New(\"some random error\"))\n\tenv.ExecuteWorkflow(handleVisibilityRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestHandleVisibilityRequest_Success() {\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverHandleVisibilitySuccessCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(archiveVisibilityActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(handleVisibilityRequestWorkflow, ArchiveRequest{})\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *handlerSuite) TestRunArchiver() {\n\tnumRequests := 1000\n\tconcurrency := 10\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverUploadSuccessCount).Times(numRequests)\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverDeleteSuccessCount).Times(numRequests)\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverHandleVisibilitySuccessCount).Times(numRequests)\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverStartedCount).Once()\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverCoroutineStartedCount).Times(concurrency)\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverCoroutineStoppedCount).Times(concurrency)\n\thandlerTestMetrics.On(\"IncCounter\", metrics.ArchiverScope, metrics.ArchiverStoppedCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(uploadHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.OnActivity(deleteHistoryActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.OnActivity(archiveVisibilityActivityFnName, mock.Anything, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(startAndFinishArchiverWorkflow, concurrency, numRequests)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc handleHistoryRequestWorkflow(ctx workflow.Context, request ArchiveRequest) error {\n\thandler := NewHandler(ctx, handlerTestLogger, handlerTestMetrics, 0, nil).(*handler)\n\thandler.handleHistoryRequest(ctx, &request)\n\treturn nil\n}\n\nfunc handleVisibilityRequestWorkflow(ctx workflow.Context, request ArchiveRequest) error {\n\thandler := NewHandler(ctx, handlerTestLogger, handlerTestMetrics, 0, nil).(*handler)\n\thandler.handleVisibilityRequest(ctx, &request)\n\treturn nil\n}\n\nfunc startAndFinishArchiverWorkflow(ctx workflow.Context, concurrency int, numRequests int) error {\n\trequestCh := workflow.NewBufferedChannel(ctx, numRequests)\n\thandler := NewHandler(ctx, handlerTestLogger, handlerTestMetrics, concurrency, requestCh)\n\thandler.Start()\n\tsentHashes := make([]uint64, numRequests)\n\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\tfor i := 0; i < numRequests; i++ {\n\t\t\tar, hash := randomArchiveRequest()\n\t\t\trequestCh.Send(ctx, ar)\n\t\t\tsentHashes[i] = hash\n\t\t}\n\t\trequestCh.Close()\n\t})\n\thandledHashes := handler.Finished()\n\tif !hashesEqual(handledHashes, sentHashes) {\n\t\treturn errors.New(\"handled hashes does not equal sent hashes\")\n\t}\n\treturn nil\n}\n\nfunc randomArchiveRequest() (ArchiveRequest, uint64) {\n\tar := ArchiveRequest{\n\t\tDomainID:   fmt.Sprintf(\"%v\", rand.Intn(1000)),\n\t\tWorkflowID: fmt.Sprintf(\"%v\", rand.Intn(1000)),\n\t\tRunID:      fmt.Sprintf(\"%v\", rand.Intn(1000)),\n\t\tTargets:    []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t}\n\treturn ar, hash(ar)\n}\n"
  },
  {
    "path": "service/worker/archiver/pump.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n//go:generate mockgen -package=$GOPACKAGE -destination=pump_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Pump\n\ntype (\n\t// Pump pumps archival requests into request channel\n\tPump interface {\n\t\tRun() PumpResult\n\t}\n\n\t// PumpResult is the result of pumping requests into request channel\n\tPumpResult struct {\n\t\tPumpedHashes          []uint64\n\t\tUnhandledCarryover    []ArchiveRequest\n\t\tTimeoutWithoutSignals bool\n\t}\n\n\tpump struct {\n\t\tctx           workflow.Context\n\t\tlogger        log.Logger\n\t\tmetricsClient metrics.Client\n\t\tcarryover     []ArchiveRequest\n\t\ttimeout       time.Duration\n\t\trequestLimit  int\n\t\trequestCh     workflow.Channel\n\t\tsignalCh      workflow.Channel\n\t}\n)\n\n// NewPump returns a new Pump\nfunc NewPump(\n\tctx workflow.Context,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tcarryover []ArchiveRequest,\n\ttimeout time.Duration,\n\trequestLimit int,\n\trequestCh workflow.Channel,\n\tsignalCh workflow.Channel,\n) Pump {\n\treturn &pump{\n\t\tctx:           ctx,\n\t\tlogger:        logger,\n\t\tmetricsClient: metricsClient,\n\t\tcarryover:     carryover,\n\t\ttimeout:       timeout,\n\t\trequestLimit:  requestLimit,\n\t\trequestCh:     requestCh,\n\t\tsignalCh:      signalCh,\n\t}\n}\n\n// Run pumps requests into request channel.\n// Blocks until either timout occurs or request limit is satisfied.\n// Returns a PumpResult which contains a summary of what was pumped.\n// Upon returning request channel is closed.\nfunc (p *pump) Run() PumpResult {\n\tsw := p.metricsClient.StartTimer(metrics.ArchiverPumpScope, metrics.CadenceLatency)\n\n\tcarryoverBoundIndex := len(p.carryover)\n\tif carryoverBoundIndex > p.requestLimit {\n\t\tcarryoverBoundIndex = p.requestLimit\n\t}\n\tvar unhandledCarryover []ArchiveRequest\n\tfor i := carryoverBoundIndex; i < len(p.carryover); i++ {\n\t\tunhandledCarryover = append(unhandledCarryover, p.carryover[i])\n\t}\n\tp.metricsClient.UpdateGauge(metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(len(unhandledCarryover)))\n\tpumpResult := PumpResult{\n\t\tUnhandledCarryover: unhandledCarryover,\n\t}\n\tfor i := 0; i < carryoverBoundIndex; i++ {\n\t\trequest := p.carryover[i]\n\t\tp.requestCh.Send(p.ctx, request)\n\t\tpumpResult.PumpedHashes = append(pumpResult.PumpedHashes, hash(request))\n\t}\n\tif len(pumpResult.PumpedHashes) == p.requestLimit {\n\t\tsw.Stop()\n\t\tp.requestCh.Close()\n\t\treturn pumpResult\n\t}\n\tselector := workflow.NewSelector(p.ctx)\n\tfinished := false\n\tselector.AddFuture(workflow.NewTimer(p.ctx, p.timeout), func(_ workflow.Future) {\n\t\tp.metricsClient.IncCounter(metrics.ArchiverPumpScope, metrics.ArchiverPumpTimeoutCount)\n\t\tif len(p.carryover) == len(pumpResult.PumpedHashes) {\n\t\t\tp.metricsClient.IncCounter(metrics.ArchiverPumpScope, metrics.ArchiverPumpTimeoutWithoutSignalsCount)\n\t\t\tpumpResult.TimeoutWithoutSignals = true\n\t\t}\n\t\tfinished = true\n\t})\n\tselector.AddReceive(p.signalCh, func(ch workflow.Channel, more bool) {\n\t\tif !more {\n\t\t\tp.logger.Error(\"signal channel channel closed unexpectedly\")\n\t\t\tp.metricsClient.IncCounter(metrics.ArchiverPumpScope, metrics.ArchiverPumpSignalChannelClosedCount)\n\t\t\tfinished = true\n\t\t\treturn\n\t\t}\n\t\tvar request ArchiveRequest\n\t\tch.Receive(p.ctx, &request)\n\t\tp.requestCh.Send(p.ctx, request)\n\t\tpumpResult.PumpedHashes = append(pumpResult.PumpedHashes, hash(request))\n\t\tfinished = len(pumpResult.PumpedHashes) == p.requestLimit\n\t\tif finished {\n\t\t\tp.metricsClient.IncCounter(metrics.ArchiverPumpScope, metrics.ArchiverPumpSignalThresholdCount)\n\t\t}\n\t})\n\tfor !finished {\n\t\tselector.Select(p.ctx)\n\t}\n\tsw.Stop()\n\tp.requestCh.Close()\n\treturn pumpResult\n}\n"
  },
  {
    "path": "service/worker/archiver/pump_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/worker/archiver (interfaces: Pump)\n//\n// Generated by this command:\n//\n//\tmockgen -package=archiver -destination=pump_mock.go -self_package=github.com/uber/cadence/service/worker/archiver github.com/uber/cadence/service/worker/archiver Pump\n//\n\n// Package archiver is a generated GoMock package.\npackage archiver\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockPump is a mock of Pump interface.\ntype MockPump struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPumpMockRecorder\n\tisgomock struct{}\n}\n\n// MockPumpMockRecorder is the mock recorder for MockPump.\ntype MockPumpMockRecorder struct {\n\tmock *MockPump\n}\n\n// NewMockPump creates a new mock instance.\nfunc NewMockPump(ctrl *gomock.Controller) *MockPump {\n\tmock := &MockPump{ctrl: ctrl}\n\tmock.recorder = &MockPumpMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPump) EXPECT() *MockPumpMockRecorder {\n\treturn m.recorder\n}\n\n// Run mocks base method.\nfunc (m *MockPump) Run() PumpResult {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Run\")\n\tret0, _ := ret[0].(PumpResult)\n\treturn ret0\n}\n\n// Run indicates an expected call of Run.\nfunc (mr *MockPumpMockRecorder) Run() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Run\", reflect.TypeOf((*MockPump)(nil).Run))\n}\n"
  },
  {
    "path": "service/worker/archiver/pump_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n)\n\nvar (\n\tpumpTestMetrics *mmocks.Client\n\tpumpTestLogger  *log.MockLogger\n)\n\ntype pumpSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestPumpSuite(t *testing.T) {\n\tsuite.Run(t, new(pumpSuite))\n}\n\nfunc (s *pumpSuite) SetupSuite() {\n\tworkflow.Register(carryoverSatisfiesLimitWorkflow)\n\tworkflow.Register(pumpWorkflow)\n\tworkflow.Register(signalChClosePumpWorkflow)\n\tworkflow.Register(signalAndCarryoverPumpWorkflow)\n}\n\nfunc (s *pumpSuite) SetupTest() {\n\tpumpTestMetrics = &mmocks.Client{}\n\tpumpTestMetrics.On(\"StartTimer\", mock.Anything, mock.Anything).Return(metrics.NopStopwatch()).Once()\n\tpumpTestLogger = log.NewMockLogger(gomock.NewController(s.T()))\n}\n\nfunc (s *pumpSuite) TearDownTest() {\n\tpumpTestMetrics.AssertExpectations(s.T())\n}\n\nfunc (s *pumpSuite) TestPumpRun_CarryoverLargerThanLimit() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(1)).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(carryoverSatisfiesLimitWorkflow, 10, 11)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_CarryoverExactlyMatchesLimit() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(carryoverSatisfiesLimitWorkflow, 10, 10)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_TimeoutWithoutSignals() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpTimeoutCount).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpTimeoutWithoutSignalsCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(pumpWorkflow, 10, 0)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_TimeoutWithSignals() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpTimeoutCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(pumpWorkflow, 10, 5)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_SignalsGottenSatisfyLimit() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpSignalThresholdCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(pumpWorkflow, 10, 10)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_SignalsAndCarryover() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpSignalThresholdCount).Once()\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(signalAndCarryoverPumpWorkflow, 10, 5, 5)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc (s *pumpSuite) TestPumpRun_SignalChannelClosedUnexpectedly() {\n\tpumpTestMetrics.On(\"UpdateGauge\", metrics.ArchiverPumpScope, metrics.ArchiverBacklogSizeGauge, float64(0)).Once()\n\tpumpTestMetrics.On(\"IncCounter\", metrics.ArchiverPumpScope, metrics.ArchiverPumpSignalChannelClosedCount).Once()\n\tpumpTestLogger.EXPECT().Error(gomock.Any(), gomock.Any()).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(signalChClosePumpWorkflow, 10, 5)\n\n\tenv.AssertExpectations(s.T())\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc carryoverSatisfiesLimitWorkflow(ctx workflow.Context, requestLimit int, carryoverSize int) error {\n\tunhandledCarryoverSize := carryoverSize - requestLimit\n\tcarryover, carryoverHashes := randomCarryover(carryoverSize)\n\trequestCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tpump := NewPump(ctx, pumpTestLogger, pumpTestMetrics, carryover, time.Nanosecond, requestLimit, requestCh, nil)\n\tactual := pump.Run()\n\texpected := PumpResult{\n\t\tPumpedHashes:          carryoverHashes[:len(carryoverHashes)-unhandledCarryoverSize],\n\t\tUnhandledCarryover:    carryover[len(carryover)-unhandledCarryoverSize:],\n\t\tTimeoutWithoutSignals: false,\n\t}\n\tif !pumpResultsEqual(expected, actual) {\n\t\treturn errors.New(\"did not get expected pump result\")\n\t}\n\tif !channelContainsExpected(ctx, requestCh, carryover[:len(carryover)-unhandledCarryoverSize]) {\n\t\treturn errors.New(\"request channel was not populated with expected values\")\n\t}\n\treturn nil\n}\n\nfunc runPumpWorkflow(ctx workflow.Context, requestLimit int, numRequests int, signalCh workflow.Channel, signalHashes []uint64, signalsSent []ArchiveRequest) error {\n\trequestCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tpump := NewPump(ctx, pumpTestLogger, pumpTestMetrics, nil, time.Nanosecond, requestLimit, requestCh, signalCh)\n\tactual := pump.Run()\n\texpected := PumpResult{\n\t\tPumpedHashes:          signalHashes,\n\t\tUnhandledCarryover:    nil,\n\t\tTimeoutWithoutSignals: numRequests == 0,\n\t}\n\tif !pumpResultsEqual(expected, actual) {\n\t\treturn errors.New(\"did not get expected pump result\")\n\t}\n\tif !channelContainsExpected(ctx, requestCh, signalsSent) {\n\t\treturn errors.New(\"request channel was not populated with expected values\")\n\t}\n\treturn nil\n}\n\nfunc pumpWorkflow(ctx workflow.Context, requestLimit int, numRequests int) error {\n\tsignalCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tsignalsSent, signalHashes := sendRequestsToChannel(ctx, signalCh, numRequests)\n\treturn runPumpWorkflow(ctx, requestLimit, numRequests, signalCh, signalHashes, signalsSent)\n}\n\nfunc signalChClosePumpWorkflow(ctx workflow.Context, requestLimit int, numRequests int) error {\n\tsignalCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tsignalsSent, signalHashes := sendRequestsToChannelBlocking(ctx, signalCh, numRequests)\n\tsignalCh.Close()\n\treturn runPumpWorkflow(ctx, requestLimit, numRequests, signalCh, signalHashes, signalsSent)\n}\n\nfunc signalAndCarryoverPumpWorkflow(ctx workflow.Context, requestLimit int, carryoverSize, numSignals int) error {\n\tsignalCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tsignalsSent, signalHashes := sendRequestsToChannel(ctx, signalCh, numSignals)\n\tcarryover, carryoverHashes := randomCarryover(carryoverSize)\n\trequestCh := workflow.NewBufferedChannel(ctx, requestLimit)\n\tpump := NewPump(ctx, pumpTestLogger, pumpTestMetrics, carryover, time.Nanosecond, requestLimit, requestCh, signalCh)\n\tactual := pump.Run()\n\texpected := PumpResult{\n\t\tPumpedHashes:          append(carryoverHashes, signalHashes...),\n\t\tUnhandledCarryover:    nil,\n\t\tTimeoutWithoutSignals: false,\n\t}\n\tif !pumpResultsEqual(expected, actual) {\n\t\treturn errors.New(\"did not get expected pump result\")\n\t}\n\tif !channelContainsExpected(ctx, requestCh, append(carryover, signalsSent...)) {\n\t\treturn errors.New(\"request channel was not populated with expected values\")\n\t}\n\treturn nil\n}\n\nfunc sendRequestsToChannel(ctx workflow.Context, ch workflow.Channel, numRequests int) ([]ArchiveRequest, []uint64) {\n\trequests := make([]ArchiveRequest, numRequests)\n\thashes := make([]uint64, numRequests)\n\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\tfor i := 0; i < numRequests; i++ {\n\t\t\trequests[i], hashes[i] = randomArchiveRequest()\n\t\t\tch.Send(ctx, requests[i])\n\t\t}\n\t})\n\treturn requests, hashes\n}\n\nfunc sendRequestsToChannelBlocking(ctx workflow.Context, ch workflow.Channel, numRequests int) ([]ArchiveRequest, []uint64) {\n\trequests := make([]ArchiveRequest, numRequests)\n\thashes := make([]uint64, numRequests)\n\tfor i := 0; i < numRequests; i++ {\n\t\trequests[i], hashes[i] = randomArchiveRequest()\n\t\tch.Send(ctx, requests[i])\n\t}\n\treturn requests, hashes\n}\n\nfunc channelContainsExpected(ctx workflow.Context, ch workflow.Channel, expected []ArchiveRequest) bool {\n\tfor i := 0; i < len(expected); i++ {\n\t\tvar actual ArchiveRequest\n\t\tif !ch.Receive(ctx, &actual) {\n\t\t\treturn false\n\t\t}\n\t\tif hash(expected[i]) != hash(actual) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn !ch.Receive(ctx, nil)\n}\n\nfunc randomCarryover(count int) ([]ArchiveRequest, []uint64) {\n\tcarryover := make([]ArchiveRequest, count)\n\thashes := make([]uint64, count)\n\tfor i := 0; i < count; i++ {\n\t\tcarryover[i], hashes[i] = randomArchiveRequest()\n\t}\n\treturn carryover, hashes\n}\n\nfunc pumpResultsEqual(expected PumpResult, actual PumpResult) bool {\n\treturn expected.TimeoutWithoutSignals == actual.TimeoutWithoutSignals &&\n\t\trequestsEqual(expected.UnhandledCarryover, actual.UnhandledCarryover) &&\n\t\thashesEqual(expected.PumpedHashes, actual.PumpedHashes)\n}\n\nfunc requestsEqual(expected []ArchiveRequest, actual []ArchiveRequest) bool {\n\tif len(expected) != len(actual) {\n\t\treturn false\n\t}\n\tfor i := 0; i < len(expected); i++ {\n\t\tif hash(expected[i]) != hash(actual[i]) {\n\t\t\treturn false\n\t\t}\n\t}\n\treturn true\n}\n"
  },
  {
    "path": "service/worker/archiver/replay_metrics_client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\treplayMetricsClient struct {\n\t\tclient metrics.Client\n\t\tctx    workflow.Context\n\t}\n\n\treplayMetricsScope struct {\n\t\tscope metrics.Scope\n\t\tctx   workflow.Context\n\t}\n)\n\n// NewReplayMetricsClient creates a metrics client which is aware of cadence's replay mode\nfunc NewReplayMetricsClient(client metrics.Client, ctx workflow.Context) metrics.Client {\n\treturn &replayMetricsClient{\n\t\tclient: client,\n\t\tctx:    ctx,\n\t}\n}\n\n// IncCounter increments a counter metric\nfunc (r *replayMetricsClient) IncCounter(scope metrics.ScopeIdx, counter metrics.MetricIdx) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.client.IncCounter(scope, counter)\n}\n\n// AddCounter adds delta to the counter metric\nfunc (r *replayMetricsClient) AddCounter(scope metrics.ScopeIdx, counter metrics.MetricIdx, delta int64) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.client.AddCounter(scope, counter, delta)\n}\n\n// StartTimer starts a timer for the given metric name. Time will be recorded when stopwatch is stopped.\nfunc (r *replayMetricsClient) StartTimer(scope metrics.ScopeIdx, timer metrics.MetricIdx) tally.Stopwatch {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn metrics.NopStopwatch()\n\t}\n\treturn r.client.StartTimer(scope, timer)\n}\n\n// RecordTimer starts a timer for the given metric name\nfunc (r *replayMetricsClient) RecordTimer(scope metrics.ScopeIdx, timer metrics.MetricIdx, d time.Duration) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.client.RecordTimer(scope, timer, d)\n}\n\n// RecordHistogramDuration record and emit a duration\nfunc (r *replayMetricsClient) RecordHistogramDuration(scope metrics.ScopeIdx, timer metrics.MetricIdx, d time.Duration) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.client.RecordHistogramDuration(scope, timer, d)\n}\n\n// UpdateGauge reports Gauge type absolute value metric\nfunc (r *replayMetricsClient) UpdateGauge(scope metrics.ScopeIdx, gauge metrics.MetricIdx, value float64) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.client.UpdateGauge(scope, gauge, value)\n}\n\n// Scope returns a client that adds the given tags to all metrics\nfunc (r *replayMetricsClient) Scope(scope metrics.ScopeIdx, tags ...metrics.Tag) metrics.Scope {\n\treturn NewReplayMetricsScope(r.client.Scope(scope, tags...), r.ctx)\n}\n\n// NewReplayMetricsScope creates a metrics scope which is aware of cadence's replay mode\nfunc NewReplayMetricsScope(scope metrics.Scope, ctx workflow.Context) metrics.Scope {\n\treturn &replayMetricsScope{\n\t\tscope: scope,\n\t\tctx:   ctx,\n\t}\n}\n\n// IncCounter increments a counter metric\nfunc (r *replayMetricsScope) IncCounter(counter metrics.MetricIdx) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.IncCounter(counter)\n}\n\n// AddCounter adds delta to the counter metric\nfunc (r *replayMetricsScope) AddCounter(counter metrics.MetricIdx, delta int64) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.AddCounter(counter, delta)\n}\n\n// StartTimer starts a timer for the given metric name. Time will be recorded when stopwatch is stopped.\nfunc (r *replayMetricsScope) StartTimer(timer metrics.MetricIdx) metrics.Stopwatch {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn metrics.NewTestStopwatch()\n\t}\n\treturn r.scope.StartTimer(timer)\n}\n\n// RecordTimer starts a timer for the given metric name\nfunc (r *replayMetricsScope) RecordTimer(timer metrics.MetricIdx, d time.Duration) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.RecordTimer(timer, d)\n}\n\n// RecordHistogramDuration records a duration value in a histogram\nfunc (r *replayMetricsScope) RecordHistogramDuration(timer metrics.MetricIdx, d time.Duration) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.RecordHistogramDuration(timer, d)\n}\n\n// RecordHistogramValue records a value in a histogram\nfunc (r *replayMetricsScope) RecordHistogramValue(timer metrics.MetricIdx, value float64) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.RecordHistogramValue(timer, value)\n}\n\nfunc (r *replayMetricsScope) ExponentialHistogram(hist metrics.MetricIdx, d time.Duration) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.ExponentialHistogram(hist, d)\n}\n\nfunc (r *replayMetricsScope) IntExponentialHistogram(hist metrics.MetricIdx, value int) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.IntExponentialHistogram(hist, value)\n}\n\n// UpdateGauge reports Gauge type absolute value metric\nfunc (r *replayMetricsScope) UpdateGauge(gauge metrics.MetricIdx, value float64) {\n\tif workflow.IsReplaying(r.ctx) {\n\t\treturn\n\t}\n\tr.scope.UpdateGauge(gauge, value)\n}\n\n// Tagged return an internal replay aware scope that can be used to add additional information to metrics\nfunc (r *replayMetricsScope) Tagged(tags ...metrics.Tag) metrics.Scope {\n\treturn NewReplayMetricsScope(r.scope.Tagged(tags...), r.ctx)\n}\n"
  },
  {
    "path": "service/worker/archiver/testdata/archival_workflow_history_v1.json",
    "content": "[{\"eventId\":1,\"timestamp\":1569606123623079000,\"eventType\":\"WorkflowExecutionStarted\",\"version\":-24,\"taskId\":6291586,\"workflowExecutionStartedEventAttributes\":{\"workflowType\":{\"name\":\"archivalWorkflow\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"bnVsbAo=\",\"executionStartToCloseTimeoutSeconds\":2592000,\"taskStartToCloseTimeoutSeconds\":60,\"originalExecutionRunId\":\"2dbedb91-4d8f-4c74-a979-4b7b9c204077\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\",\"firstExecutionRunId\":\"2dbedb91-4d8f-4c74-a979-4b7b9c204077\",\"attempt\":0,\"cronSchedule\":\"\",\"firstDecisionTaskBackoffSeconds\":0,\"header\":{}}},{\"eventId\":2,\"timestamp\":1569606123623099000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291587,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzQwZjhlZTY3LTAyMmYtNDA4ZC04M2U1LTFkYjU3ZTJjNWJmZiIsIlJ1bklEIjoiZjJiMzYwYTAtZDkwYS00YWZhLWFkODgtYmEwNDFmYWQ2YTQyIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JtTW1Jek5qQmhNQzFrT1RCaExUUmhabUV0WVdRNE9DMWlZVEEwTVdaaFpEWmhORElMQUJRQUFBQWtPRFF3TXpBM1lqa3RPVEEzTmkwMFpXVXlMVGd5WVRBdE5EVm1NakZrTmpGa056RTVEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":3,\"timestamp\":1569606123623106000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291590,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"cadence-archival-tl\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":4,\"timestamp\":1569606123627344000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291595,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":3,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"de9215e7-2bcc-4294-9fac-1dc75ed159fc\"}},{\"eventId\":5,\"timestamp\":1569606123634126000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291598,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":3,\"startedEventId\":4,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":6,\"timestamp\":1569606123634168000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291599,\"markerRecordedEventAttributes\":{\"markerName\":\"SideEffect\",\"details\":\"MAoiZXlKQmNtTm9hWFpsY2tOdmJtTjFjbkpsYm1ONUlqbzFNQ3dpUVhKamFHbDJZV3h6VUdWeVNYUmxjbUYwYVc5dUlqb3hNREF3TENKVWFXMWxiR2x0YVhSUVpYSkpkR1Z5WVhScGIyNGlPakV5T1RZd01EQXdNREF3TURBd01EQjlDZz09Igo=\",\"decisionTaskCompletedEventId\":5}},{\"eventId\":7,\"timestamp\":1569606123634180000,\"eventType\":\"TimerStarted\",\"version\":-24,\"taskId\":6291600,\"timerStartedEventAttributes\":{\"timerId\":\"1\",\"startToFireTimeoutSeconds\":1296000,\"decisionTaskCompletedEventId\":5}},{\"eventId\":8,\"timestamp\":1569606123634200000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291601,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"2\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzQwZjhlZTY3LTAyMmYtNDA4ZC04M2U1LTFkYjU3ZTJjNWJmZiIsIlJ1bklEIjoiZjJiMzYwYTAtZDkwYS00YWZhLWFkODgtYmEwNDFmYWQ2YTQyIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JtTW1Jek5qQmhNQzFrT1RCaExUUmhabUV0WVdRNE9DMWlZVEEwTVdaaFpEWmhORElMQUJRQUFBQWtPRFF3TXpBM1lqa3RPVEEzTmkwMFpXVXlMVGd5WVRBdE5EVm1NakZrTmpGa056RTVEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":5,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":9,\"timestamp\":1569606123638659000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291607,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":8,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"7dccfd5f-fb84-450f-b452-3a5dd5f382e9\",\"attempt\":0}},{\"eventId\":10,\"timestamp\":1569606123643937000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291608,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":8,\"startedEventId\":9,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":11,\"timestamp\":1569606123643948000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291611,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":12,\"timestamp\":1569606123647878000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291615,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":11,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"774fa75d-7cef-47c8-96be-4a6db8a4cb99\"}},{\"eventId\":13,\"timestamp\":1569606123654654000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291618,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":11,\"startedEventId\":12,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":14,\"timestamp\":1569606123654683000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291619,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMyIsImFjdGl2aXR5VHlwZSI6ImRlbGV0ZUhpc3RvcnlBY3Rpdml0eSIsInJlcGxheVRpbWUiOiIyMDE5LTA5LTI3VDEwOjQyOjAzLjY0OTczMTYxMy0wNzowMCJ9Cg==\",\"decisionTaskCompletedEventId\":13}},{\"eventId\":15,\"timestamp\":1569606123918543000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291621,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzBjMGQxNDkxLWI1YzAtNGIxZC04NjliLTc0OWEwNWUwNDMyNiIsIlJ1bklEIjoiNGIwYjFiMjYtZjllMi00OGEwLWJjZGQtMTdlMmNlMTQwNzVkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1EwWWpCaU1XSXlOaTFtT1dVeUxUUTRZVEF0WW1Oa1pDMHhOMlV5WTJVeE5EQTNOV1FMQUJRQUFBQWtZekptWVdKak4yWXROelJpT1MwMFlqYzFMVGhsT0RrdE1EbGlaakJsTlRJeU1qWmhEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":16,\"timestamp\":1569606123918557000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291623,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":17,\"timestamp\":1569606123922726000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291627,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":16,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"081890a8-8942-4434-ac52-9e668a406ee2\"}},{\"eventId\":18,\"timestamp\":1569606123928159000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291630,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":16,\"startedEventId\":17,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":19,\"timestamp\":1569606123928190000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291631,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"4\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzBjMGQxNDkxLWI1YzAtNGIxZC04NjliLTc0OWEwNWUwNDMyNiIsIlJ1bklEIjoiNGIwYjFiMjYtZjllMi00OGEwLWJjZGQtMTdlMmNlMTQwNzVkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1EwWWpCaU1XSXlOaTFtT1dVeUxUUTRZVEF0WW1Oa1pDMHhOMlV5WTJVeE5EQTNOV1FMQUJRQUFBQWtZekptWVdKak4yWXROelJpT1MwMFlqYzFMVGhsT0RrdE1EbGlaakJsTlRJeU1qWmhEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":18,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":20,\"timestamp\":1569606123932704000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291636,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":19,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"d8b8a27d-e66d-4ae5-89ac-058cd336b660\",\"attempt\":0}},{\"eventId\":21,\"timestamp\":1569606123938439000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291637,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":19,\"startedEventId\":20,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":22,\"timestamp\":1569606123938450000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291640,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":23,\"timestamp\":1569606123942512000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291644,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":22,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-processor-parent-close-policy\",\"requestId\":\"b66d91d8-f367-4ddd-aad3-1071f937bc48\"}},{\"eventId\":24,\"timestamp\":1569606123950344000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291647,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":22,\"startedEventId\":23,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":25,\"timestamp\":1569606123950371000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291648,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiNSIsImFjdGl2aXR5VHlwZSI6ImRlbGV0ZUhpc3RvcnlBY3Rpdml0eSIsInJlcGxheVRpbWUiOiIyMDE5LTA5LTI3VDEwOjQyOjAzLjk0NDIyNDE3NS0wNzowMCJ9Cg==\",\"decisionTaskCompletedEventId\":24}},{\"eventId\":26,\"timestamp\":1569606124919680000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291650,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzAzYzI4MDRkLWNjZGMtNDI0MS05MmM2LWMzNzM5MGJhM2NkZSIsIlJ1bklEIjoiNDk4MWQzMTUtYjEyZi00YWE4LWJhOWQtNzYwNzczZmMzYTllIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1EwT1RneFpETXhOUzFpTVRKbUxUUmhZVGd0WW1FNVpDMDNOakEzTnpObVl6TmhPV1VMQUJRQUFBQWtOakZrTnpBelltSXRaR0V5T1MwME56VmhMV0ZtWlRNdE9EVTJaRGN5TXpkaU1ETTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":27,\"timestamp\":1569606124919689000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291652,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":28,\"timestamp\":1569606124922147000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291656,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzI0ODFiMWMzLTkyZDMtNGEyZi04NmZiLWUyM2MyNTI2ZDhmMCIsIlJ1bklEIjoiODM1YTFhZGItMTkxNy00ZWU2LTgzMDAtZGE0YmQyMTUyYzgxIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1E0TXpWaE1XRmtZaTB4T1RFM0xUUmxaVFl0T0RNd01DMWtZVFJpWkRJeE5USmpPREVMQUJRQUFBQWtOMlJoWXpSak5qTXRZelk1WmkwME56QTJMVGd6TlRndFlXTTFNbU00TkRSbFpESTREd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":29,\"timestamp\":1569606124924481000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291658,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2RmYzZiODBlLTA0NmEtNGU0Yy1iZTQyLWVlZmY0YmFlMjMxNyIsIlJ1bklEIjoiZjlmMzQwY2ItNGRkNS00YmU2LThmYTEtNDI4NWUwYTUyOTFkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JtT1dZek5EQmpZaTAwWkdRMUxUUmlaVFl0T0daaE1TMDBNamcxWlRCaE5USTVNV1FMQUJRQUFBQWtaRE13WmpFMVpUQXRPVFEzTkMwME9HVTVMVGcyWVdVdE56RTVOR1ExWVRreE1tUmpEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":30,\"timestamp\":1569606124927549000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291660,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":27,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"a4207044-03be-4913-857c-1caadff199f6\"}},{\"eventId\":31,\"timestamp\":1569606124934407000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291663,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":27,\"startedEventId\":30,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":32,\"timestamp\":1569606124934430000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291664,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"6\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzAzYzI4MDRkLWNjZGMtNDI0MS05MmM2LWMzNzM5MGJhM2NkZSIsIlJ1bklEIjoiNDk4MWQzMTUtYjEyZi00YWE4LWJhOWQtNzYwNzczZmMzYTllIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1EwT1RneFpETXhOUzFpTVRKbUxUUmhZVGd0WW1FNVpDMDNOakEzTnpObVl6TmhPV1VMQUJRQUFBQWtOakZrTnpBelltSXRaR0V5T1MwME56VmhMV0ZtWlRNdE9EVTJaRGN5TXpkaU1ETTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":31,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":33,\"timestamp\":1569606124934440000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291665,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"7\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzI0ODFiMWMzLTkyZDMtNGEyZi04NmZiLWUyM2MyNTI2ZDhmMCIsIlJ1bklEIjoiODM1YTFhZGItMTkxNy00ZWU2LTgzMDAtZGE0YmQyMTUyYzgxIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1E0TXpWaE1XRmtZaTB4T1RFM0xUUmxaVFl0T0RNd01DMWtZVFJpWkRJeE5USmpPREVMQUJRQUFBQWtOMlJoWXpSak5qTXRZelk1WmkwME56QTJMVGd6TlRndFlXTTFNbU00TkRSbFpESTREd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":31,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":34,\"timestamp\":1569606124934446000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291666,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"8\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2RmYzZiODBlLTA0NmEtNGU0Yy1iZTQyLWVlZmY0YmFlMjMxNyIsIlJ1bklEIjoiZjlmMzQwY2ItNGRkNS00YmU2LThmYTEtNDI4NWUwYTUyOTFkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JtT1dZek5EQmpZaTAwWkdRMUxUUmlaVFl0T0daaE1TMDBNamcxWlRCaE5USTVNV1FMQUJRQUFBQWtaRE13WmpFMVpUQXRPVFEzTkMwME9HVTVMVGcyWVdVdE56RTVOR1ExWVRreE1tUmpEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":31,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":35,\"timestamp\":1569606124938736000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291679,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":32,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"57e8b0cb-14b3-46c5-9549-5910a4c17cc4\",\"attempt\":0}},{\"eventId\":36,\"timestamp\":1569606124945961000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291680,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":32,\"startedEventId\":35,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":37,\"timestamp\":1569606124945974000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291683,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":38,\"timestamp\":1569606124940672000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291690,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":33,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"de068cc4-efd0-4ddf-afd8-c4e2272c134c\",\"attempt\":0}},{\"eventId\":39,\"timestamp\":1569606124949415000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291691,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":33,\"startedEventId\":38,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":40,\"timestamp\":1569606124942620000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291698,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":34,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"6bfef50a-28cb-4f05-b8d5-915d58347891\",\"attempt\":0}},{\"eventId\":41,\"timestamp\":1569606124953872000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291699,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":34,\"startedEventId\":40,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":42,\"timestamp\":1569606124957961000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291703,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":37,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"9008323e-b89d-4c43-967f-e23846522466\"}},{\"eventId\":43,\"timestamp\":1569606124964261000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291713,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":37,\"startedEventId\":42,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":44,\"timestamp\":1569606124964293000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291714,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTEiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNC45NTkwODgxNDgtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":43}},{\"eventId\":45,\"timestamp\":1569606124964297000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291715,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiOSIsImFjdGl2aXR5VHlwZSI6ImRlbGV0ZUhpc3RvcnlBY3Rpdml0eSIsInJlcGxheVRpbWUiOiIyMDE5LTA5LTI3VDEwOjQyOjA0Ljk1OTQ2MzQyNi0wNzowMCJ9Cg==\",\"decisionTaskCompletedEventId\":43}},{\"eventId\":46,\"timestamp\":1569606124964301000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291716,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTAiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNC45NTk3MTM3MzYtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":43}},{\"eventId\":47,\"timestamp\":1569606124979355000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291728,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjozLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzc0NmFkOGI5LTAyYTEtNGNlNy1iNzkzLTI3ZDgwNjllZGU3MyIsIlJ1bklEIjoiNTY0YTY2OWYtYmM4OS00M2I5LWE2ZjEtN2IwYzc4NGEyNmViIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1ExTmpSaE5qWTVaaTFpWXpnNUxUUXpZamt0WVRabU1TMDNZakJqTnpnMFlUSTJaV0lMQUJRQUFBQWtORGt6TVRKbFpERXRNR1kyTkMwME0yVmpMVGcwWXpRdFpXVXlaREZrWVRaak5tUmtEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":48,\"timestamp\":1569606124979363000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291730,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":49,\"timestamp\":1569606124983474000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291734,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":48,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-processor-parent-close-policy\",\"requestId\":\"82a5d9c2-1deb-494c-98ae-bb0c3681abd3\"}},{\"eventId\":50,\"timestamp\":1569606124988413000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291737,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":48,\"startedEventId\":49,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":51,\"timestamp\":1569606124988446000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291738,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"12\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjozLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzc0NmFkOGI5LTAyYTEtNGNlNy1iNzkzLTI3ZDgwNjllZGU3MyIsIlJ1bklEIjoiNTY0YTY2OWYtYmM4OS00M2I5LWE2ZjEtN2IwYzc4NGEyNmViIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1ExTmpSaE5qWTVaaTFpWXpnNUxUUXpZamt0WVRabU1TMDNZakJqTnpnMFlUSTJaV0lMQUJRQUFBQWtORGt6TVRKbFpERXRNR1kyTkMwME0yVmpMVGcwWXpRdFpXVXlaREZrWVRaak5tUmtEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":50,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":52,\"timestamp\":1569606124992421000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291743,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":51,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"640f192d-c9b3-4ff7-8e71-3234d7bb09c5\",\"attempt\":0}},{\"eventId\":53,\"timestamp\":1569606124997683000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291744,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":51,\"startedEventId\":52,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":54,\"timestamp\":1569606124997696000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291747,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":55,\"timestamp\":1569606125001619000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291751,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":54,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"0f0ce318-bc46-4439-81c9-c6288e063df5\"}},{\"eventId\":56,\"timestamp\":1569606125007998000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291754,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":54,\"startedEventId\":55,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":57,\"timestamp\":1569606125008025000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291755,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTMiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNS4wMDI4NTU2NjItMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":56}},{\"eventId\":58,\"timestamp\":1569606125218433000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291757,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2NmMzNkNzJhLWZhYzktNDA4Zi1hZDJmLThjMTFlZjFkNTQ3NSIsIlJ1bklEIjoiMDJmMjE3M2QtY2EzZC00MDMyLWI5NjEtZDNjNTYyOGRiYjIwIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1F3TW1ZeU1UY3paQzFqWVROa0xUUXdNekl0WWprMk1TMWtNMk0xTmpJNFpHSmlNakFMQUJRQUFBQWtZV1kzTlRCa01UQXRaVFl4TkMwME56QmtMVGhtT1RVdE1UaGtOekU0WWpjeVlqbGhEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":59,\"timestamp\":1569606125218443000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291759,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":60,\"timestamp\":1569606125221695000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291763,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":59,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"8ee8cc2f-88ce-476d-b40f-e82544a9a3f9\"}},{\"eventId\":61,\"timestamp\":1569606125226189000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291766,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":59,\"startedEventId\":60,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":62,\"timestamp\":1569606125226225000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291767,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"14\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2NmMzNkNzJhLWZhYzktNDA4Zi1hZDJmLThjMTFlZjFkNTQ3NSIsIlJ1bklEIjoiMDJmMjE3M2QtY2EzZC00MDMyLWI5NjEtZDNjNTYyOGRiYjIwIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1F3TW1ZeU1UY3paQzFqWVROa0xUUXdNekl0WWprMk1TMWtNMk0xTmpJNFpHSmlNakFMQUJRQUFBQWtZV1kzTlRCa01UQXRaVFl4TkMwME56QmtMVGhtT1RVdE1UaGtOekU0WWpjeVlqbGhEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":61,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":63,\"timestamp\":1569606125229902000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291772,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":62,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"870d2b5f-3286-4992-9b83-911d9cb10842\",\"attempt\":0}},{\"eventId\":64,\"timestamp\":1569606125234358000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291773,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":62,\"startedEventId\":63,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":65,\"timestamp\":1569606125234368000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291776,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":66,\"timestamp\":1569606125238253000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291780,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":65,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-processor-parent-close-policy\",\"requestId\":\"44177a34-efe8-4efc-a406-4018f9a46a87\"}},{\"eventId\":67,\"timestamp\":1569606125244383000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291783,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":65,\"startedEventId\":66,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":68,\"timestamp\":1569606125244421000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291784,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTUiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNS4yMzk2NTkyOTUtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":67}},{\"eventId\":69,\"timestamp\":1569606125920400000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291786,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzZhN2MxYjI5LTEzMjktNDgyYi04NTM3LTk4MmE5M2JjNjM3ZiIsIlJ1bklEIjoiNWQzMzQ4MDQtNzQ3ZC00ODAyLTkwYjctZmNiNjI2OWFiMDVkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1ExWkRNek5EZ3dOQzAzTkRka0xUUTRNREl0T1RCaU55MW1ZMkkyTWpZNVlXSXdOV1FMQUJRQUFBQWtPREF6T0dKaU16Y3RPRE14TUMwME9UVm1MVGxqTlRNdE9HVmpPVEl5WVdVeE1tRTFEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":70,\"timestamp\":1569606125920408000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291788,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":71,\"timestamp\":1569606125923245000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291792,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":70,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"c7ee25d5-3d85-464d-a3ad-197f30ad4c13\"}},{\"eventId\":72,\"timestamp\":1569606125926950000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291795,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":70,\"startedEventId\":71,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":73,\"timestamp\":1569606125926971000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291796,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"16\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzZhN2MxYjI5LTEzMjktNDgyYi04NTM3LTk4MmE5M2JjNjM3ZiIsIlJ1bklEIjoiNWQzMzQ4MDQtNzQ3ZC00ODAyLTkwYjctZmNiNjI2OWFiMDVkIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1ExWkRNek5EZ3dOQzAzTkRka0xUUTRNREl0T1RCaU55MW1ZMkkyTWpZNVlXSXdOV1FMQUJRQUFBQWtPREF6T0dKaU16Y3RPRE14TUMwME9UVm1MVGxqTlRNdE9HVmpPVEl5WVdVeE1tRTFEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":72,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":74,\"timestamp\":1569606125930046000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291801,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":73,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"3a35e9d7-790f-41df-a518-0f4d6a5bc0f2\",\"attempt\":0}},{\"eventId\":75,\"timestamp\":1569606125933614000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291802,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":73,\"startedEventId\":74,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":76,\"timestamp\":1569606125933623000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291805,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":77,\"timestamp\":1569606125936580000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291809,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":76,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"3bd96a83-c2fa-4979-8abb-c3ae80cc6ebb\"}},{\"eventId\":78,\"timestamp\":1569606125941102000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291812,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":76,\"startedEventId\":77,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":79,\"timestamp\":1569606125941122000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291813,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTciLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNS45Mzc0NDI0NTEtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":78}},{\"eventId\":80,\"timestamp\":1569606126048936000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291846,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjozLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2I1YWJjYmNmLTZiZjAtNGE2My05YTM4LTU0NzhhM2MzOWM5MCIsIlJ1bklEIjoiY2VkYWRkZGEtZTM1My00MTQwLWI5MzktZjQ5OWFjMThlOGQ0IiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JqWldSaFpHUmtZUzFsTXpVekxUUXhOREF0WWprek9TMW1ORGs1WVdNeE9HVTRaRFFMQUJRQUFBQWtaVGM0WXpBMFpURXRaRFJpTmkwME4yWTBMVGt5Wm1RdE1tTTRZell3WTJNeVkyUTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":81,\"timestamp\":1569606126048949000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291848,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":82,\"timestamp\":1569606126052440000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291852,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":81,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-processor-parent-close-policy\",\"requestId\":\"304f5dac-44a4-4af2-a759-c206b5a3ec46\"}},{\"eventId\":83,\"timestamp\":1569606126056194000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291855,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":81,\"startedEventId\":82,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":84,\"timestamp\":1569606126056216000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291856,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"18\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjozLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2I1YWJjYmNmLTZiZjAtNGE2My05YTM4LTU0NzhhM2MzOWM5MCIsIlJ1bklEIjoiY2VkYWRkZGEtZTM1My00MTQwLWI5MzktZjQ5OWFjMThlOGQ0IiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JqWldSaFpHUmtZUzFsTXpVekxUUXhOREF0WWprek9TMW1ORGs1WVdNeE9HVTRaRFFMQUJRQUFBQWtaVGM0WXpBMFpURXRaRFJpTmkwME4yWTBMVGt5Wm1RdE1tTTRZell3WTJNeVkyUTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":83,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":85,\"timestamp\":1569606126058989000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291861,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":84,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"b247c133-bc34-472b-9bbd-a4e5e1f942d3\",\"attempt\":0}},{\"eventId\":86,\"timestamp\":1569606126062977000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291862,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":84,\"startedEventId\":85,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":87,\"timestamp\":1569606126062987000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291865,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":88,\"timestamp\":1569606126066232000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291869,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":87,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"26054030-971f-486e-99d8-4151f7f3535b\"}},{\"eventId\":89,\"timestamp\":1569606126071558000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291872,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":87,\"startedEventId\":88,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":90,\"timestamp\":1569606126071585000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291873,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMTkiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNi4wNjcyNDQ4OTEtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":89}},{\"eventId\":91,\"timestamp\":1569606126219398000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291875,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2U5YTE2MDRiLWY1YjEtNGY3Yi1hNjY0LTQ1Yjg4ZjU5YjY2OSIsIlJ1bklEIjoiYzNkMTVhZWQtMWViNS00NGU2LWJmYTMtMWNlNmY2OTIzZDEzIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JqTTJReE5XRmxaQzB4WldJMUxUUTBaVFl0WW1aaE15MHhZMlUyWmpZNU1qTmtNVE1MQUJRQUFBQWtPRGMyTjJFNVpqVXROMlJsWlMwME5HWTBMV0U0Wm1RdFkyWmpZVEEyTUdVeFpXWTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":92,\"timestamp\":1569606126219405000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291877,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":93,\"timestamp\":1569606126221920000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291881,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":92,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"d0a8f7d9-ed21-4c69-8656-348d497ae0ac\"}},{\"eventId\":94,\"timestamp\":1569606126225163000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291884,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":92,\"startedEventId\":93,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":95,\"timestamp\":1569606126225187000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291885,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"20\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoyLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkX2U5YTE2MDRiLWY1YjEtNGY3Yi1hNjY0LTQ1Yjg4ZjU5YjY2OSIsIlJ1bklEIjoiYzNkMTVhZWQtMWViNS00NGU2LWJmYTMtMWNlNmY2OTIzZDEzIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JqTTJReE5XRmxaQzB4WldJMUxUUTBaVFl0WW1aaE15MHhZMlUyWmpZNU1qTmtNVE1MQUJRQUFBQWtPRGMyTjJFNVpqVXROMlJsWlMwME5HWTBMV0U0Wm1RdFkyWmpZVEEyTUdVeFpXWTJEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":94,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":96,\"timestamp\":1569606126228158000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291890,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":95,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"8b9abe6e-e31f-4fd0-ba44-761d7ec7f4b7\",\"attempt\":0}},{\"eventId\":97,\"timestamp\":1569606126231778000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291891,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":95,\"startedEventId\":96,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":98,\"timestamp\":1569606126231788000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291894,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":99,\"timestamp\":1569606126234682000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291898,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":98,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-processor-parent-close-policy\",\"requestId\":\"0cd5dc7b-2a41-49d7-9db8-e952f8ce5cc2\"}},{\"eventId\":100,\"timestamp\":1569606126239125000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291901,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":98,\"startedEventId\":99,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":101,\"timestamp\":1569606126239145000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291902,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMjEiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNi4yMzU0Njg2OTMtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":100}},{\"eventId\":102,\"timestamp\":1569606126923765000,\"eventType\":\"WorkflowExecutionSignaled\",\"version\":-24,\"taskId\":6291904,\"workflowExecutionSignaledEventAttributes\":{\"signalName\":\"cadence-archival-signal\",\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzlmY2U4ZGEzLTMwNzMtNGY1Yi1iMjIxLWU3ZTFlZTM1ZWE0MCIsIlJ1bklEIjoiZGI3OTlhNWMtMmJkYS00OTZjLWI3MDMtNWY1ODNjN2IzZTkzIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JrWWpjNU9XRTFZeTB5WW1SaExUUTVObU10WWpjd015MDFaalU0TTJNM1lqTmxPVE1MQUJRQUFBQWtNMkkwTW1SbU0yVXRNelJpWkMwME5USTNMV0k0T1RRdE1qQXlPRGhpWVdOaU1UZGpEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"identity\":\"59844@ycyang-C02XW41WJGH6@\"}},{\"eventId\":103,\"timestamp\":1569606126923772000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291906,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":104,\"timestamp\":1569606126927973000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291910,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":103,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-sys-tl-scanner-tasklist-0\",\"requestId\":\"9ed3f383-bfe8-4f05-b7af-3ebe0763f3ec\"}},{\"eventId\":105,\"timestamp\":1569606126931546000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291913,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":103,\"startedEventId\":104,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":106,\"timestamp\":1569606126931568000,\"eventType\":\"ActivityTaskScheduled\",\"version\":-24,\"taskId\":6291914,\"activityTaskScheduledEventAttributes\":{\"activityId\":\"22\",\"activityType\":{\"name\":\"uploadHistoryActivity\"},\"taskList\":{\"name\":\"cadence-archival-tl\"},\"input\":\"eyJTaGFyZElEIjoxLCJEb21haW5JRCI6IjhlNjI3NmQ4LWZhOTQtNDZjMS1hYjY5LWFkOGRhOTEyZWE4MCIsIkRvbWFpbk5hbWUiOiJzYW1wbGVzLWRvbWFpbiIsIldvcmtmbG93SUQiOiJoZWxsb3dvcmxkXzlmY2U4ZGEzLTMwNzMtNGY1Yi1iMjIxLWU3ZTFlZTM1ZWE0MCIsIlJ1bklEIjoiZGI3OTlhNWMtMmJkYS00OTZjLWI3MDMtNWY1ODNjN2IzZTkzIiwiRXZlbnRTdG9yZVZlcnNpb24iOjIsIkJyYW5jaFRva2VuIjoiV1FzQUNnQUFBQ1JrWWpjNU9XRTFZeTB5WW1SaExUUTVObU10WWpjd015MDFaalU0TTJNM1lqTmxPVE1MQUJRQUFBQWtNMkkwTW1SbU0yVXRNelJpWkMwME5USTNMV0k0T1RRdE1qQXlPRGhpWVdOaU1UZGpEd0FlREFBQUFBQUEiLCJOZXh0RXZlbnRJRCI6MTIsIkNsb3NlRmFpbG92ZXJWZXJzaW9uIjotMjQsIlVSSSI6ImZpbGU6Ly8vdG1wL2NhZGVuY2VfYXJjaGl2YWwvZGV2ZWxvcG1lbnQifQo=\",\"scheduleToCloseTimeoutSeconds\":300,\"scheduleToStartTimeoutSeconds\":300,\"startToCloseTimeoutSeconds\":60,\"heartbeatTimeoutSeconds\":0,\"decisionTaskCompletedEventId\":105,\"retryPolicy\":{\"initialIntervalInSeconds\":1,\"backoffCoefficient\":2,\"maximumIntervalInSeconds\":100,\"maximumAttempts\":0,\"nonRetriableErrorReasons\":[\"cadenceInternal:Panic\",\"upload non-retriable error\"],\"expirationIntervalInSeconds\":300},\"header\":{}}},{\"eventId\":107,\"timestamp\":1569606126934664000,\"eventType\":\"ActivityTaskStarted\",\"version\":-24,\"taskId\":6291919,\"activityTaskStartedEventAttributes\":{\"scheduledEventId\":106,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"0e6ba111-0876-423c-9962-6f20948f48fa\",\"attempt\":0}},{\"eventId\":108,\"timestamp\":1569606126938683000,\"eventType\":\"ActivityTaskCompleted\",\"version\":-24,\"taskId\":6291920,\"activityTaskCompletedEventAttributes\":{\"scheduledEventId\":106,\"startedEventId\":107,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\"}},{\"eventId\":109,\"timestamp\":1569606126938692000,\"eventType\":\"DecisionTaskScheduled\",\"version\":-24,\"taskId\":6291923,\"decisionTaskScheduledEventAttributes\":{\"taskList\":{\"name\":\"ycyang-C02XW41WJGH6:7ce34128-dace-4a8a-a3ca-6eaa9c97dd13\"},\"startToCloseTimeoutSeconds\":60,\"attempt\":0}},{\"eventId\":110,\"timestamp\":1569606126941424000,\"eventType\":\"DecisionTaskStarted\",\"version\":-24,\"taskId\":6291927,\"decisionTaskStartedEventAttributes\":{\"scheduledEventId\":109,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"requestId\":\"c0bbf240-9a0e-4323-a182-a06decb5b2f0\"}},{\"eventId\":111,\"timestamp\":1569606126945774000,\"eventType\":\"DecisionTaskCompleted\",\"version\":-24,\"taskId\":6291930,\"decisionTaskCompletedEventAttributes\":{\"scheduledEventId\":109,\"startedEventId\":110,\"identity\":\"59844@ycyang-C02XW41WJGH6@cadence-archival-tl\",\"binaryChecksum\":\"a3f549be2faf4819fc3dcb4732046568\"}},{\"eventId\":112,\"timestamp\":1569606126945794000,\"eventType\":\"MarkerRecorded\",\"version\":-24,\"taskId\":6291931,\"markerRecordedEventAttributes\":{\"markerName\":\"LocalActivity\",\"details\":\"eyJhY3Rpdml0eUlkIjoiMjMiLCJhY3Rpdml0eVR5cGUiOiJkZWxldGVIaXN0b3J5QWN0aXZpdHkiLCJyZXBsYXlUaW1lIjoiMjAxOS0wOS0yN1QxMDo0MjowNi45NDIzMTU2MjMtMDc6MDAifQo=\",\"decisionTaskCompletedEventId\":111}}]"
  },
  {
    "path": "service/worker/archiver/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"time\"\n\n\t\"github.com/dgryski/go-farm\"\n\t\"go.uber.org/cadence/activity\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n)\n\n// MaxArchivalIterationTimeout returns the max allowed timeout for a single iteration of archival workflow\nfunc MaxArchivalIterationTimeout() time.Duration {\n\treturn workflowStartToCloseTimeout / 2\n}\n\nfunc hash(i interface{}) uint64 {\n\tvar b bytes.Buffer\n\t// please make sure encoder is deterministic (especially when encoding map objects)\n\t// use json not gob here as json will sort map keys, while gob is non-deterministic\n\tjson.NewEncoder(&b).Encode(i) //nolint:errcheck\n\treturn farm.Fingerprint64(b.Bytes())\n}\n\nfunc hashesEqual(a []uint64, b []uint64) bool {\n\tif len(a) != len(b) {\n\t\treturn false\n\t}\n\taMap := make(map[uint64]int)\n\tfor _, elem := range a {\n\t\taMap[elem] = aMap[elem] + 1\n\t}\n\tfor _, elem := range b {\n\t\tcount := aMap[elem]\n\t\tif count == 0 {\n\t\t\treturn false\n\t\t}\n\t\taMap[elem] = aMap[elem] - 1\n\t}\n\treturn true\n}\n\nfunc tagLoggerWithHistoryRequest(logger log.Logger, request *ArchiveRequest) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.ShardID(request.ShardID),\n\t\ttag.ArchivalRequestDomainID(request.DomainID),\n\t\ttag.ArchivalRequestDomainName(request.DomainName),\n\t\ttag.ArchivalRequestWorkflowID(request.WorkflowID),\n\t\ttag.ArchivalRequestRunID(request.RunID),\n\t\ttag.ArchivalRequestBranchToken(request.BranchToken),\n\t\ttag.ArchivalRequestNextEventID(request.NextEventID),\n\t\ttag.ArchivalRequestCloseFailoverVersion(request.CloseFailoverVersion),\n\t\ttag.ArchivalURI(request.URI),\n\t)\n}\n\nfunc tagLoggerWithVisibilityRequest(logger log.Logger, request *ArchiveRequest) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.ArchivalRequestDomainID(request.DomainID),\n\t\ttag.ArchivalRequestDomainName(request.DomainName),\n\t\ttag.ArchivalRequestWorkflowID(request.WorkflowID),\n\t\ttag.ArchivalRequestRunID(request.RunID),\n\t\ttag.ArchivalURI(request.URI),\n\t)\n}\n\nfunc tagLoggerWithActivityInfo(logger log.Logger, activityInfo activity.Info) log.Logger {\n\treturn logger.WithTags(\n\t\ttag.WorkflowID(activityInfo.WorkflowExecution.ID),\n\t\ttag.WorkflowRunID(activityInfo.WorkflowExecution.RunID),\n\t\ttag.Attempt(activityInfo.Attempt))\n}\n\nfunc convertSearchAttributesToString(searchAttr map[string][]byte) map[string]string {\n\tsearchAttrStr := make(map[string]string)\n\tfor k, v := range searchAttr {\n\t\tsearchAttrStr[k] = string(v)\n\t}\n\treturn searchAttrStr\n}\n"
  },
  {
    "path": "service/worker/archiver/util_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype UtilSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n}\n\nfunc TestUtilSuite(t *testing.T) {\n\tsuite.Run(t, new(UtilSuite))\n}\n\nfunc (s *UtilSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *UtilSuite) TestHashDeterminism() {\n\ttestCases := []struct {\n\t\tinstance interface{}\n\t}{\n\t\t{\n\t\t\tinstance: \"some random string\",\n\t\t},\n\t\t{\n\t\t\tinstance: map[string]string{\n\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t\"key2\": \"value2\",\n\t\t\t\t\"key3\": \"value3\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tinstance: []string{\"value1\", \"value2\", \"value3\"},\n\t\t},\n\t\t{\n\t\t\tinstance: ArchiveRequest{\n\t\t\t\tDomainID:    \"some random domainID\",\n\t\t\t\tShardID:     0,\n\t\t\t\tBranchToken: []byte{1, 2, 3},\n\t\t\t\tNextEventID: int64(123),\n\t\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\tFields: map[string][]byte{\n\t\t\t\t\t\t\"memoKey1\": []byte{1, 2, 3},\n\t\t\t\t\t\t\"memoKey2\": []byte{4, 5, 6},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tSearchAttributes: map[string][]byte{\n\t\t\t\t\t\"customKey1\": []byte{1, 2, 3},\n\t\t\t\t\t\"customKey2\": []byte{4, 5, 6},\n\t\t\t\t},\n\t\t\t\tTargets: []ArchivalTarget{ArchiveTargetHistory, ArchiveTargetVisibility},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\texpectedHash := hash(tc.instance)\n\t\tfor i := 0; i != 100; i++ {\n\t\t\ts.Equal(expectedHash, hash(tc.instance))\n\t\t}\n\t}\n}\n\nfunc (s *UtilSuite) TestHashesEqual() {\n\ttestCases := []struct {\n\t\ta     []uint64\n\t\tb     []uint64\n\t\tequal bool\n\t}{\n\t\t{\n\t\t\ta:     nil,\n\t\t\tb:     nil,\n\t\t\tequal: true,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2, 3},\n\t\t\tb:     []uint64{1, 2, 3},\n\t\t\tequal: true,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2},\n\t\t\tb:     []uint64{1, 2, 3},\n\t\t\tequal: false,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2, 3},\n\t\t\tb:     []uint64{1, 2},\n\t\t\tequal: false,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2, 5, 5, 5},\n\t\t\tb:     []uint64{1, 2, 5, 5, 5},\n\t\t\tequal: true,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2, 5, 5},\n\t\t\tb:     []uint64{1, 2, 5, 5, 5},\n\t\t\tequal: false,\n\t\t},\n\t\t{\n\t\t\ta:     []uint64{1, 2, 5, 5, 5, 5},\n\t\t\tb:     []uint64{1, 2, 5, 5, 5},\n\t\t\tequal: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.equal, hashesEqual(tc.a, tc.b))\n\t}\n}\n"
  },
  {
    "path": "service/worker/archiver/workflow.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype dynamicConfigResult struct {\n\tArchiverConcurrency   int\n\tArchivalsPerIteration int\n\tTimelimitPerIteration time.Duration\n}\n\nfunc archivalWorkflow(ctx workflow.Context, carryover []ArchiveRequest) error {\n\treturn archivalWorkflowHelper(ctx, globalLogger, globalMetricsClient, globalConfig, nil, nil, carryover)\n}\n\nfunc archivalWorkflowHelper(\n\tctx workflow.Context,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tconfig *Config,\n\thandler Handler, // enables tests to inject mocks\n\tpump Pump, // enables tests to inject mocks\n\tcarryover []ArchiveRequest,\n) error {\n\tmetricsClient = NewReplayMetricsClient(metricsClient, ctx)\n\tmetricsClient.IncCounter(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStartedCount)\n\tsw := metricsClient.StartTimer(metrics.ArchiverArchivalWorkflowScope, metrics.CadenceLatency)\n\tworkflowInfo := workflow.GetInfo(ctx)\n\tlogger = logger.WithTags(\n\t\ttag.WorkflowID(workflowInfo.WorkflowExecution.ID),\n\t\ttag.WorkflowRunID(workflowInfo.WorkflowExecution.RunID),\n\t\ttag.WorkflowTaskListName(workflowInfo.TaskListName),\n\t\ttag.WorkflowType(workflowInfo.WorkflowType.Name))\n\tlogger = log.NewReplayLogger(logger, ctx, false)\n\n\tlogger.Info(\"archival system workflow started\")\n\tvar dcResult dynamicConfigResult\n\t_ = workflow.SideEffect(\n\t\tctx,\n\t\tfunc(ctx workflow.Context) interface{} {\n\t\t\ttimeLimit := config.TimeLimitPerArchivalIteration()\n\t\t\tmaxTimeLimit := MaxArchivalIterationTimeout()\n\t\t\tif timeLimit > maxTimeLimit {\n\t\t\t\ttimeLimit = maxTimeLimit\n\t\t\t}\n\t\t\treturn dynamicConfigResult{\n\t\t\t\tArchiverConcurrency:   config.ArchiverConcurrency(),\n\t\t\t\tArchivalsPerIteration: config.ArchivalsPerIteration(),\n\t\t\t\tTimelimitPerIteration: timeLimit,\n\t\t\t}\n\t\t}).Get(&dcResult)\n\trequestCh := workflow.NewBufferedChannel(ctx, dcResult.ArchivalsPerIteration)\n\tif handler == nil {\n\t\thandler = NewHandler(ctx, logger, metricsClient, dcResult.ArchiverConcurrency, requestCh)\n\t}\n\thandlerSW := metricsClient.StartTimer(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverHandleAllRequestsLatency)\n\thandler.Start()\n\tsignalCh := workflow.GetSignalChannel(ctx, signalName)\n\tif pump == nil {\n\t\tpump = NewPump(ctx, logger, metricsClient, carryover, dcResult.TimelimitPerIteration, dcResult.ArchivalsPerIteration, requestCh, signalCh)\n\t}\n\tpumpResult := pump.Run()\n\tmetricsClient.AddCounter(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumPumpedRequestsCount, int64(len(pumpResult.PumpedHashes)))\n\thandledHashes := handler.Finished()\n\thandlerSW.Stop()\n\tmetricsClient.AddCounter(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumHandledRequestsCount, int64(len(handledHashes)))\n\tif !hashesEqual(pumpResult.PumpedHashes, handledHashes) {\n\t\tlogger.Error(\"handled archival requests do not match pumped archival requests\")\n\t\tmetricsClient.IncCounter(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverPumpedNotEqualHandledCount)\n\t}\n\tif pumpResult.TimeoutWithoutSignals {\n\t\tlogger.Info(\"workflow stopping because pump did not get any signals within timeout threshold\")\n\t\tmetricsClient.IncCounter(metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStoppingCount)\n\t\tsw.Stop()\n\t\treturn nil\n\t}\n\tfor {\n\t\tvar request ArchiveRequest\n\t\tif ok := signalCh.ReceiveAsync(&request); !ok {\n\t\t\tbreak\n\t\t}\n\t\tpumpResult.UnhandledCarryover = append(pumpResult.UnhandledCarryover, request)\n\t}\n\tlogger.Info(\"archival system workflow continue as new\")\n\tctx = workflow.WithExecutionStartToCloseTimeout(ctx, workflowStartToCloseTimeout)\n\tctx = workflow.WithWorkflowTaskStartToCloseTimeout(ctx, workflowTaskStartToCloseTimeout)\n\tsw.Stop()\n\treturn workflow.NewContinueAsNewError(ctx, archivalWorkflowFnName, pumpResult.UnhandledCarryover)\n}\n"
  },
  {
    "path": "service/worker/archiver/workflow_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage archiver\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n)\n\nvar (\n\tworkflowTestMetrics *mmocks.Client\n\tworkflowTestLogger  log.Logger\n\tworkflowTestHandler *MockHandler\n\tworkflowTestPump    *MockPump\n\tworkflowTestConfig  *Config\n)\n\ntype workflowSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc (s *workflowSuite) SetupSuite() {\n\tworkflow.Register(archivalWorkflowTest)\n}\n\nfunc TestWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(workflowSuite))\n}\n\nfunc (s *workflowSuite) SetupTest() {\n\tctrl := gomock.NewController(s.T())\n\tworkflowTestMetrics = &mmocks.Client{}\n\tworkflowTestLogger = testlogger.New(s.T())\n\tworkflowTestHandler = NewMockHandler(ctrl)\n\tworkflowTestPump = NewMockPump(ctrl)\n\tworkflowTestConfig = &Config{\n\t\tArchiverConcurrency:           dynamicproperties.GetIntPropertyFn(0),\n\t\tArchivalsPerIteration:         dynamicproperties.GetIntPropertyFn(0),\n\t\tTimeLimitPerArchivalIteration: dynamicproperties.GetDurationPropertyFn(MaxArchivalIterationTimeout()),\n\t}\n}\n\nfunc (s *workflowSuite) TestArchivalWorkflow_Fail_HashesDoNotEqual() {\n\tworkflowTestMetrics.On(\"IncCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStartedCount).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.CadenceLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverHandleAllRequestsLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumPumpedRequestsCount, int64(3)).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumHandledRequestsCount, int64(3)).Once()\n\tworkflowTestMetrics.On(\"IncCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverPumpedNotEqualHandledCount).Once()\n\tworkflowTestHandler.EXPECT().Start().Times(1)\n\tworkflowTestHandler.EXPECT().Finished().Return([]uint64{9, 7, 0}).Times(1)\n\tworkflowTestPump.EXPECT().Run().Return(PumpResult{\n\t\tPumpedHashes: []uint64{8, 7, 0},\n\t}).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(archivalWorkflowTest)\n\n\ts.True(env.IsWorkflowCompleted())\n\tvar continueAsNewError *workflow.ContinueAsNewError\n\tok := errors.As(env.GetWorkflowError(), &continueAsNewError)\n\ts.True(ok, \"Called ContinueAsNew\")\n\tenv.AssertExpectations(s.T())\n}\n\nfunc (s *workflowSuite) TestArchivalWorkflow_Exit_TimeoutWithoutSignals() {\n\tworkflowTestMetrics.On(\"IncCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStartedCount).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.CadenceLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverHandleAllRequestsLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumPumpedRequestsCount, int64(0)).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumHandledRequestsCount, int64(0)).Once()\n\tworkflowTestMetrics.On(\"IncCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStoppingCount).Once()\n\tworkflowTestHandler.EXPECT().Start().Times(1)\n\tworkflowTestHandler.EXPECT().Finished().Return([]uint64{}).Times(1)\n\tworkflowTestPump.EXPECT().Run().Return(PumpResult{\n\t\tPumpedHashes:          []uint64{},\n\t\tTimeoutWithoutSignals: true,\n\t}).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(archivalWorkflowTest)\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n\tenv.AssertExpectations(s.T())\n}\n\nfunc (s *workflowSuite) TestArchivalWorkflow_Success() {\n\tworkflowTestMetrics.On(\"IncCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverWorkflowStartedCount).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.CadenceLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"StartTimer\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverHandleAllRequestsLatency).Return(metrics.NopStopwatch()).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumPumpedRequestsCount, int64(5)).Once()\n\tworkflowTestMetrics.On(\"AddCounter\", metrics.ArchiverArchivalWorkflowScope, metrics.ArchiverNumHandledRequestsCount, int64(5)).Once()\n\tworkflowTestHandler.EXPECT().Start().Times(1)\n\tworkflowTestHandler.EXPECT().Finished().Return([]uint64{1, 2, 3, 4, 5}).Times(1)\n\tworkflowTestPump.EXPECT().Run().Return(PumpResult{\n\t\tPumpedHashes: []uint64{1, 2, 3, 4, 5},\n\t}).Times(1)\n\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.ExecuteWorkflow(archivalWorkflowTest)\n\n\ts.True(env.IsWorkflowCompleted())\n\tvar continueAsNewError *workflow.ContinueAsNewError\n\tok := errors.As(env.GetWorkflowError(), &continueAsNewError)\n\ts.True(ok, \"Called ContinueAsNew\")\n\tenv.AssertExpectations(s.T())\n}\n\nfunc (s *workflowSuite) TestReplayArchiveHistoryWorkflow() {\n\tlogger := testlogger.NewZap(s.T())\n\tglobalLogger = workflowTestLogger\n\tglobalMetricsClient = metrics.NewClient(tally.NewTestScope(\"replay\", nil), metrics.Worker, metrics.HistogramMigration{})\n\tglobalConfig = &Config{\n\t\tArchiverConcurrency:           dynamicproperties.GetIntPropertyFn(50),\n\t\tArchivalsPerIteration:         dynamicproperties.GetIntPropertyFn(1000),\n\t\tTimeLimitPerArchivalIteration: dynamicproperties.GetDurationPropertyFn(MaxArchivalIterationTimeout()),\n\t}\n\terr := worker.ReplayWorkflowHistoryFromJSONFile(logger, \"testdata/archival_workflow_history_v1.json\")\n\ts.NoError(err)\n}\n\nfunc archivalWorkflowTest(ctx workflow.Context) error {\n\treturn archivalWorkflowHelper(ctx, workflowTestLogger, workflowTestMetrics, workflowTestConfig, workflowTestHandler, workflowTestPump, nil)\n}\n"
  },
  {
    "path": "service/worker/asyncworkflow/async_workflow_consumer_manager.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage asyncworkflow\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tdefaultRefreshInterval = 1 * time.Minute\n\tdefaultShutdownTimeout = 5 * time.Second\n)\n\ntype ConsumerManagerOptions func(*ConsumerManager)\n\nfunc WithTimeSource(timeSrc clock.TimeSource) ConsumerManagerOptions {\n\treturn func(c *ConsumerManager) {\n\t\tc.timeSrc = timeSrc\n\t}\n}\n\nfunc WithRefreshInterval(interval time.Duration) ConsumerManagerOptions {\n\treturn func(c *ConsumerManager) {\n\t\tc.refreshInterval = interval\n\t}\n}\n\nfunc WithEnabledPropertyFn(enabledFn dynamicproperties.BoolPropertyFn) ConsumerManagerOptions {\n\treturn func(c *ConsumerManager) {\n\t\tc.enabledFn = enabledFn\n\t}\n}\n\nfunc WithEmitConsumerCountMetrifFn(fn func(int)) ConsumerManagerOptions {\n\treturn func(c *ConsumerManager) {\n\t\tc.emitConsumerCountMetricFn = fn\n\t}\n}\n\nfunc NewConsumerManager(\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\tdomainCache cache.DomainCache,\n\tqueueProvider queue.Provider,\n\tfrontendClient frontend.Client,\n\toptions ...ConsumerManagerOptions,\n) *ConsumerManager {\n\tctx, cancel := context.WithCancel(context.Background())\n\tcm := &ConsumerManager{\n\t\tenabledFn:       dynamicproperties.GetBoolPropertyFn(true),\n\t\tlogger:          logger.WithTags(tag.ComponentAsyncWFConsumptionManager),\n\t\tmetricsClient:   metricsClient,\n\t\tdomainCache:     domainCache,\n\t\tqueueProvider:   queueProvider,\n\t\tfrontendClient:  frontendClient,\n\t\trefreshInterval: defaultRefreshInterval,\n\t\tshutdownTimeout: defaultShutdownTimeout,\n\t\tctx:             ctx,\n\t\tcancelFn:        cancel,\n\t\tactiveConsumers: make(map[string]provider.Consumer),\n\t\ttimeSrc:         clock.NewRealTimeSource(),\n\t}\n\n\tcm.emitConsumerCountMetricFn = cm.emitConsumerCountMetric\n\n\tfor _, opt := range options {\n\t\topt(cm)\n\t}\n\treturn cm\n}\n\ntype ConsumerManager struct {\n\t// all member variables are accessed without any mutex with the assumption that they are only accessed by the background loop\n\tenabledFn                 dynamicproperties.BoolPropertyFn\n\tlogger                    log.Logger\n\tmetricsClient             metrics.Client\n\ttimeSrc                   clock.TimeSource\n\tdomainCache               cache.DomainCache\n\tqueueProvider             queue.Provider\n\tfrontendClient            frontend.Client\n\trefreshInterval           time.Duration\n\tshutdownTimeout           time.Duration\n\tctx                       context.Context\n\tcancelFn                  context.CancelFunc\n\twg                        sync.WaitGroup\n\tactiveConsumers           map[string]provider.Consumer\n\temitConsumerCountMetricFn func(int)\n}\n\nfunc (c *ConsumerManager) Start() {\n\tc.logger.Info(\"Starting ConsumerManager\")\n\tc.wg.Add(1)\n\tgo c.run()\n}\n\nfunc (c *ConsumerManager) Stop() {\n\tc.logger.Info(\"Stopping ConsumerManager\")\n\tc.cancelFn()\n\tc.wg.Wait()\n\tif !common.AwaitWaitGroup(&c.wg, c.shutdownTimeout) {\n\t\tc.logger.Warn(\"ConsumerManager timed out on shutdown\", tag.Dynamic(\"timeout\", c.shutdownTimeout))\n\t\treturn\n\t}\n\n\tc.stopConsumers()\n\n\tc.logger.Info(\"Stopped ConsumerManager\")\n}\n\nfunc (c *ConsumerManager) run() {\n\tdefer c.wg.Done()\n\n\tticker := c.timeSrc.NewTicker(c.refreshInterval)\n\tdefer ticker.Stop()\n\tc.logger.Info(\"ConsumerManager background loop started\", tag.Dynamic(\"refresh-interval\", c.refreshInterval))\n\n\tenabled := c.enabledFn()\n\tif enabled {\n\t\tc.refreshConsumers()\n\t} else {\n\t\tc.logger.Info(\"ConsumerManager is disabled at the moment so skipping initial refresh\")\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tpreviouslyEnabled := enabled\n\t\t\tenabled = c.enabledFn()\n\t\t\tif enabled != previouslyEnabled {\n\t\t\t\tc.logger.Info(\"ConsumerManager enabled state changed\", tag.Dynamic(\"enabled\", enabled))\n\t\t\t}\n\n\t\t\tif enabled {\n\t\t\t\t// refresh consumers every round when consumer is enabled\n\t\t\t\tc.refreshConsumers()\n\t\t\t} else {\n\t\t\t\t// stop consumers when consumer is disabled\n\t\t\t\tc.stopConsumers()\n\t\t\t}\n\n\t\tcase <-c.ctx.Done():\n\t\t\tc.logger.Info(\"ConsumerManager background loop stopped because context is done\")\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (c *ConsumerManager) refreshConsumers() {\n\tdomains := c.domainCache.GetAllDomain()\n\tc.logger.Info(\"Refreshing consumers\", tag.Dynamic(\"domain-count\", len(domains)), tag.Dynamic(\"consumer-count\", len(c.activeConsumers)))\n\trefCounts := make(map[string]int, len(c.activeConsumers))\n\n\tfor _, domain := range domains {\n\t\tselect {\n\t\tdefault:\n\t\tcase <-c.ctx.Done():\n\t\t\tc.logger.Info(\"refreshConsumers is terminating because context is done\")\n\t\t\treturn\n\t\t}\n\n\t\tc.logger.Debug(\"Refreshing consumers for domain\", tag.WorkflowDomainName(domain.GetInfo().Name), tag.Dynamic(\"domain-config\", domain.GetConfig()))\n\n\t\t// domain config is not set or async workflow config is not set\n\t\tif domain.GetConfig() == nil || domain.GetConfig().AsyncWorkflowConfig == (types.AsyncWorkflowConfiguration{}) {\n\t\t\tcontinue\n\t\t}\n\n\t\tcfg := domain.GetConfig().AsyncWorkflowConfig\n\t\tqueue, err := c.getQueue(cfg)\n\t\tif err != nil {\n\t\t\tc.logger.Error(\"Failed to get queue\", tag.Error(err), tag.WorkflowDomainName(domain.GetInfo().Name))\n\t\t\tcontinue\n\t\t}\n\n\t\tif !cfg.Enabled {\n\t\t\t// Already running active consumers for such queues will be stopped in the next loop\n\t\t\tcontinue\n\t\t}\n\n\t\t// async workflow config is enabled. check if consumer is already running\n\t\tif c.activeConsumers[queue.ID()] != nil {\n\t\t\tc.logger.Debug(\"Consumer already running\", tag.WorkflowDomainName(domain.GetInfo().Name), tag.AsyncWFQueueID(queue.ID()))\n\t\t\trefCounts[queue.ID()]++\n\t\t\tcontinue\n\t\t}\n\n\t\tc.logger.Info(\"Starting consumer\", tag.WorkflowDomainName(domain.GetInfo().Name), tag.AsyncWFQueueID(queue.ID()))\n\t\tconsumer, err := queue.CreateConsumer(&provider.Params{\n\t\t\tLogger:         c.logger,\n\t\t\tMetricsClient:  c.metricsClient,\n\t\t\tFrontendClient: c.frontendClient,\n\t\t})\n\t\tif err != nil {\n\t\t\tc.logger.Error(\"Failed to create consumer\", tag.Error(err), tag.WorkflowDomainName(domain.GetInfo().Name), tag.AsyncWFQueueID(queue.ID()))\n\t\t\tcontinue\n\t\t}\n\n\t\tif err := consumer.Start(); err != nil {\n\t\t\tc.logger.Error(\"Failed to start consumer\", tag.Error(err), tag.WorkflowDomainName(domain.GetInfo().Name), tag.AsyncWFQueueID(queue.ID()))\n\t\t\tcontinue\n\t\t}\n\n\t\tc.activeConsumers[queue.ID()] = consumer\n\t\trefCounts[queue.ID()]++\n\t\tc.logger.Info(\"Created and started consumer\", tag.WorkflowDomainName(domain.GetInfo().Name), tag.AsyncWFQueueID(queue.ID()))\n\t}\n\n\t// stop consumers that are not needed\n\tfor qID, consumer := range c.activeConsumers {\n\t\tif refCounts[qID] > 0 {\n\t\t\tcontinue\n\t\t}\n\n\t\tc.logger.Info(\"Stopping consumer because it's not needed\", tag.AsyncWFQueueID(qID))\n\t\tconsumer.Stop()\n\t\tdelete(c.activeConsumers, qID)\n\t\tc.logger.Info(\"Stopped consumer\", tag.AsyncWFQueueID(qID))\n\t}\n\n\tc.logger.Info(\"Refreshed consumers\", tag.Dynamic(\"consumer-count\", len(c.activeConsumers)))\n\tc.emitConsumerCountMetricFn(len(c.activeConsumers))\n}\n\nfunc (c *ConsumerManager) emitConsumerCountMetric(count int) {\n\tc.metricsClient.Scope(metrics.AsyncWorkflowConsumerScope).UpdateGauge(metrics.AsyncWorkflowConsumerCount, float64(count))\n}\n\nfunc (c *ConsumerManager) stopConsumers() {\n\tif len(c.activeConsumers) == 0 {\n\t\treturn\n\t}\n\n\tc.logger.Info(\"Stopping all active consumers\", tag.Dynamic(\"consumer-count\", len(c.activeConsumers)))\n\tfor qID, consumer := range c.activeConsumers {\n\t\tconsumer.Stop()\n\t\tc.logger.Info(\"Stopped consumer\", tag.AsyncWFQueueID(qID))\n\t\tdelete(c.activeConsumers, qID)\n\t}\n\n\tc.emitConsumerCountMetricFn(len(c.activeConsumers))\n\tc.logger.Info(\"Stopped all active consumers\", tag.Dynamic(\"consumer-count\", len(c.activeConsumers)))\n}\n\nfunc (c *ConsumerManager) getQueue(cfg types.AsyncWorkflowConfiguration) (provider.Queue, error) {\n\tif cfg.PredefinedQueueName != \"\" {\n\t\treturn c.queueProvider.GetPredefinedQueue(cfg.PredefinedQueueName)\n\t}\n\n\treturn c.queueProvider.GetQueue(cfg.QueueType, cfg.QueueConfig)\n}\n"
  },
  {
    "path": "service/worker/asyncworkflow/async_workflow_consumer_manager_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage asyncworkflow\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"sort\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/asyncworkflow/queue\"\n\t\"github.com/uber/cadence/common/asyncworkflow/queue/provider\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype domainWithConfig struct {\n\tname                 string\n\tasyncWFCfg           types.AsyncWorkflowConfiguration\n\tfailQueueCreation    bool\n\tfailConsumerCreation bool\n\tfailConsumerStart    bool\n}\n\nfunc TestConsumerManager(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\t// firstRoundDomains will be returned by the domain cache when the consumer manager starts\n\t\tfirstRoundDomains []domainWithConfig\n\t\t// wantFirstRoundConsumers is the list of consumers that should be created after the first round\n\t\twantFirstRoundConsumers []string\n\t\t// secondRoundDomains will be returned by the domain cache after the first refresh interval\n\t\tsecondRoundDomains []domainWithConfig\n\t\t// wantSecondRoundConsumers is the list of consumers that should be created after the second round\n\t\twantSecondRoundConsumers []string\n\t}{\n\t\t{\n\t\t\tname: \"no domains\",\n\t\t},\n\t\t{\n\t\t\tname: \"empty queue config\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname:       \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"queue creation fails\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t\tfailQueueCreation: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"consumer creation fails\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t\tfailConsumerCreation: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"consumer start fails\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t\tfailConsumerStart: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"queue disabled\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             false,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"queue disabled in second round\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{\"queue1\"},\n\t\t\tsecondRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             false,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantSecondRoundConsumers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"same consumers for both rounds\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{\"queue1\"},\n\t\t\tsecondRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"queue1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantSecondRoundConsumers: []string{\"queue1\"},\n\t\t},\n\t\t{\n\t\t\tname: \"shared queue by multiple domains and different enable-disable states\",\n\t\t\tfirstRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"shared_queue\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"domain2\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             false,\n\t\t\t\t\t\tPredefinedQueueName: \"shared_queue\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"domain3\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\tQueueType: \"kafka\",\n\t\t\t\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantFirstRoundConsumers: []string{\n\t\t\t\t\"shared_queue\",\n\t\t\t\t`queuetype:kafka,queueconfig:{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`,\n\t\t\t},\n\t\t\tsecondRoundDomains: []domainWithConfig{\n\t\t\t\t{\n\t\t\t\t\tname: \"domain1\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"shared_queue\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tname: \"domain2\",\n\t\t\t\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:             true,\n\t\t\t\t\t\tPredefinedQueueName: \"shared_queue\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantSecondRoundConsumers: []string{\"shared_queue\"},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tmockTimeSrc := clock.NewMockedTimeSource()\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\n\t\t\t// setup mocks for 2 rounds of domain cache refresh\n\t\t\tmockQueueProvider := queue.NewMockProvider(ctrl)\n\t\t\tfor _, domainsForRound := range [][]domainWithConfig{tc.firstRoundDomains, tc.secondRoundDomains} {\n\t\t\t\tmockDomainCache.EXPECT().\n\t\t\t\t\tGetAllDomain().\n\t\t\t\t\tReturn(toDomainCacheEntries(domainsForRound)).\n\t\t\t\t\tTimes(1)\n\t\t\t\tfor _, dwc := range domainsForRound {\n\t\t\t\t\tif dwc.asyncWFCfg == (types.AsyncWorkflowConfiguration{}) {\n\t\t\t\t\t\tcontinue\n\t\t\t\t\t}\n\n\t\t\t\t\tif dwc.failQueueCreation {\n\t\t\t\t\t\tmockQueueProvider.EXPECT().\n\t\t\t\t\t\t\tGetPredefinedQueue(dwc.asyncWFCfg.PredefinedQueueName).\n\t\t\t\t\t\t\tReturn(nil, errors.New(\"queue creation failed\"))\n\t\t\t\t\t} else {\n\t\t\t\t\t\tqueueMock := provider.NewMockQueue(ctrl)\n\t\t\t\t\t\tqueueMock.EXPECT().ID().Return(queueID(dwc.asyncWFCfg)).AnyTimes()\n\n\t\t\t\t\t\tif dwc.asyncWFCfg.PredefinedQueueName != \"\" {\n\t\t\t\t\t\t\tmockQueueProvider.EXPECT().\n\t\t\t\t\t\t\t\tGetPredefinedQueue(dwc.asyncWFCfg.PredefinedQueueName).\n\t\t\t\t\t\t\t\tReturn(queueMock, nil).AnyTimes()\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tmockQueueProvider.EXPECT().\n\t\t\t\t\t\t\t\tGetQueue(gomock.Any(), gomock.Any()).\n\t\t\t\t\t\t\t\tReturn(queueMock, nil).AnyTimes()\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif !dwc.asyncWFCfg.Enabled {\n\t\t\t\t\t\t\tcontinue\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\tif dwc.failConsumerCreation {\n\t\t\t\t\t\t\tqueueMock.EXPECT().CreateConsumer(gomock.Any()).\n\t\t\t\t\t\t\t\tReturn(nil, errors.New(\"consumer creation failed\")).AnyTimes()\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tmockConsumer := provider.NewMockConsumer(ctrl)\n\t\t\t\t\t\t\tvar startErr error\n\t\t\t\t\t\t\tif dwc.failConsumerStart {\n\t\t\t\t\t\t\t\tstartErr = errors.New(\"consumer start failed\")\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tmockConsumer.EXPECT().Start().Return(startErr).AnyTimes()\n\t\t\t\t\t\t\tmockConsumer.EXPECT().Stop().AnyTimes()\n\t\t\t\t\t\t\tqueueMock.EXPECT().CreateConsumer(gomock.Any()).Return(mockConsumer, nil).AnyTimes()\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\n\t\t\t// create consumer manager\n\t\t\tcm := NewConsumerManager(\n\t\t\t\ttestlogger.New(t),\n\t\t\t\tmetrics.NewNoopMetricsClient(),\n\t\t\t\tmockDomainCache,\n\t\t\t\tmockQueueProvider,\n\t\t\t\tnil,\n\t\t\t\tWithTimeSource(mockTimeSrc),\n\t\t\t)\n\n\t\t\tcm.Start()\n\t\t\tdefer cm.Stop()\n\n\t\t\t// wait for the first round of consumers to be created\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\t\t\t// verify consumers\n\t\t\tt.Log(\"first round comparison\")\n\t\t\tif diff := cmpQueueIDs(cm.activeConsumers, tc.wantFirstRoundConsumers); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Consumer mismatch after first round (-want +got):\\n%s\", diff)\n\t\t\t}\n\n\t\t\t// wait for the second round of consumers to be created\n\t\t\tmockTimeSrc.Advance(defaultRefreshInterval)\n\t\t\ttime.Sleep(50 * time.Millisecond)\n\t\t\t// verify consumers\n\t\t\tt.Log(\"second round comparison\")\n\t\t\tif diff := cmpQueueIDs(cm.activeConsumers, tc.wantSecondRoundConsumers); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Consumer mismatch after second round (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestConsumerManagerEnabledDisabled(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockTimeSrc := clock.NewMockedTimeSource()\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockQueueProvider := queue.NewMockProvider(ctrl)\n\tdwc := domainWithConfig{\n\t\tname: \"domain1\",\n\t\tasyncWFCfg: types.AsyncWorkflowConfiguration{\n\t\t\tEnabled:   true,\n\t\t\tQueueType: \"kafka\",\n\t\t\tQueueConfig: &types.DataBlob{\n\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\tData:         []byte(`{\"brokers\":[\"localhost:9092\"],\"topics\":[\"test-topic\"]}`),\n\t\t\t},\n\t\t},\n\t}\n\n\tmockDomainCache.EXPECT().GetAllDomain().Return(toDomainCacheEntries([]domainWithConfig{dwc})).AnyTimes()\n\tqueueMock := provider.NewMockQueue(ctrl)\n\tqueueMock.EXPECT().ID().Return(queueID(dwc.asyncWFCfg)).AnyTimes()\n\tmockQueueProvider.EXPECT().GetQueue(gomock.Any(), gomock.Any()).Return(queueMock, nil).AnyTimes()\n\tmockConsumer := provider.NewMockConsumer(ctrl)\n\tmockConsumer.EXPECT().Start().Return(nil).AnyTimes()\n\tmockConsumer.EXPECT().Stop().AnyTimes()\n\tqueueMock.EXPECT().CreateConsumer(gomock.Any()).Return(mockConsumer, nil).AnyTimes()\n\n\tvar consumerMgrEnabled, consumerCount int32\n\n\t// create consumer manager\n\tcm := NewConsumerManager(\n\t\ttestlogger.New(t),\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tmockDomainCache,\n\t\tmockQueueProvider,\n\t\tnil,\n\t\tWithTimeSource(mockTimeSrc),\n\t\tWithEnabledPropertyFn(func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\treturn atomic.LoadInt32(&consumerMgrEnabled) == 1\n\t\t}),\n\t\tWithEmitConsumerCountMetrifFn(func(count int) {\n\t\t\tatomic.StoreInt32(&consumerCount, int32(count))\n\t\t}),\n\t)\n\n\tcm.Start()\n\tdefer cm.Stop()\n\n\t// wait for the first round of consumers to be created and verify consumer count\n\tatomic.StoreInt32(&consumerMgrEnabled, 1)\n\ttime.Sleep(50 * time.Millisecond)\n\tt.Log(\"first round comparison\")\n\tgot := atomic.LoadInt32(&consumerCount)\n\twant := 1 // consumer manager is enabled\n\tif got != int32(want) {\n\t\tt.Fatalf(\"Consumer count mismatch after first round, want: %v, got: %v\", want, got)\n\t}\n\n\t// disable consumer manager and wait for the second round of refresh\n\tatomic.StoreInt32(&consumerMgrEnabled, 0)\n\tmockTimeSrc.Advance(defaultRefreshInterval)\n\ttime.Sleep(50 * time.Millisecond)\n\tgot = atomic.LoadInt32(&consumerCount)\n\twant = 0 // all consumers should be stopped when consumer manager is disabled\n\tif got != int32(want) {\n\t\tt.Fatalf(\"Consumer count mismatch after second round, want: %v, got: %v\", want, got)\n\t}\n\n\t// enable consumer manager and wait for the third round of refresh\n\tatomic.StoreInt32(&consumerMgrEnabled, 1)\n\tmockTimeSrc.Advance(defaultRefreshInterval)\n\ttime.Sleep(50 * time.Millisecond)\n\tgot = atomic.LoadInt32(&consumerCount)\n\twant = 1 // consumer manager is enabled\n\tif got != int32(want) {\n\t\tt.Fatalf(\"Consumer count mismatch after third round, want: %v, got: %v\", want, got)\n\t}\n}\n\nfunc toDomainCacheEntries(domains []domainWithConfig) map[string]*cache.DomainCacheEntry {\n\tresult := make(map[string]*cache.DomainCacheEntry, len(domains))\n\tfor _, d := range domains {\n\t\tresult[d.name] = cache.NewGlobalDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{\n\t\t\t\tName: d.name,\n\t\t\t},\n\t\t\t&persistence.DomainConfig{\n\t\t\t\tAsyncWorkflowConfig: d.asyncWFCfg,\n\t\t\t},\n\t\t\tnil,\n\t\t\t0,\n\t\t)\n\t}\n\treturn result\n}\n\nfunc queueID(asyncWFCfg types.AsyncWorkflowConfiguration) string {\n\tif asyncWFCfg.PredefinedQueueName != \"\" {\n\t\treturn asyncWFCfg.PredefinedQueueName\n\t}\n\n\tif asyncWFCfg.QueueConfig == nil {\n\t\treturn \"\"\n\t}\n\n\treturn fmt.Sprintf(\"queuetype:%s,queueconfig:%s\", asyncWFCfg.QueueType, string(asyncWFCfg.QueueConfig.Data))\n}\n\nfunc cmpQueueIDs(activeConsumers map[string]provider.Consumer, want []string) string {\n\tgot := make([]string, 0, len(activeConsumers))\n\tfor qID := range activeConsumers {\n\t\tgot = append(got, qID)\n\t}\n\n\tif want == nil {\n\t\twant = []string{}\n\t}\n\n\tsort.Strings(got)\n\tsort.Strings(want)\n\n\treturn cmp.Diff(want, got)\n}\n"
  },
  {
    "path": "service/worker/batcher/batcher.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"context\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/worker\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// Config defines the configuration for batcher\n\tConfig struct {\n\t\tAdminOperationToken dynamicproperties.StringPropertyFn\n\t\t// ClusterMetadata contains the metadata for this cluster\n\t\tClusterMetadata cluster.Metadata\n\t}\n\n\t// BootstrapParams contains the set of params needed to bootstrap\n\t// the batcher sub-system\n\tBootstrapParams struct {\n\t\t// Config contains the configuration for scanner\n\t\tConfig Config\n\t\t// ServiceClient is an instance of cadence service client\n\t\tServiceClient workflowserviceclient.Interface\n\t\t// MetricsClient is an instance of metrics object for emitting stats\n\t\tMetricsClient metrics.Client\n\t\tLogger        log.Logger\n\t\t// TallyScope is an instance of tally metrics scope\n\t\tTallyScope tally.Scope\n\t\t// ClientBean is an instance of client.Bean for a collection of clients\n\t\tClientBean client.Bean\n\t}\n\n\t// Batcher is the background sub-system that execute workflow for batch operations\n\t// It is also the context object that get's passed around within the scanner workflows / activities\n\tBatcher struct {\n\t\tcfg           Config\n\t\tsvcClient     workflowserviceclient.Interface\n\t\tclientBean    client.Bean\n\t\tmetricsClient metrics.Client\n\t\ttallyScope    tally.Scope\n\t\tlogger        log.Logger\n\t}\n)\n\n// New returns a new instance of batcher daemon Batcher\nfunc New(params *BootstrapParams) *Batcher {\n\tcfg := params.Config\n\treturn &Batcher{\n\t\tcfg:           cfg,\n\t\tsvcClient:     params.ServiceClient,\n\t\tmetricsClient: params.MetricsClient,\n\t\ttallyScope:    params.TallyScope,\n\t\tlogger:        params.Logger.WithTags(tag.ComponentBatcher),\n\t\tclientBean:    params.ClientBean,\n\t}\n}\n\n// Start starts the scanner\nfunc (s *Batcher) Start() error {\n\t// start worker for batch operation workflows\n\tctx := context.WithValue(context.Background(), BatcherContextKey, s)\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              s.tallyScope,\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\tbatchWorker := worker.New(s.svcClient, constants.BatcherLocalDomainName, BatcherTaskListName, workerOpts)\n\treturn batchWorker.Start()\n}\n"
  },
  {
    "path": "service/worker/batcher/batcher_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nfunc Test__Start(t *testing.T) {\n\tbatcher, mockResource := setuptest(t)\n\terr := batcher.Start()\n\trequire.NoError(t, err)\n\tmockResource.Finish(t)\n}\n\nfunc setuptest(t *testing.T) (*Batcher, *resource.Test) {\n\tctrl := gomock.NewController(t)\n\tmockResource := resource.NewTest(t, ctrl, metrics.Worker)\n\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockResource.SDKClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.DescribeDomainResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForDecisionTaskResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForActivityTaskResponse{}, nil).AnyTimes()\n\tsdkClient := mockResource.GetSDKClient()\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockResource.FrontendClient).AnyTimes()\n\tmockClientBean.EXPECT().GetRemoteAdminClient(gomock.Any()).Return(mockResource.RemoteAdminClient, nil).AnyTimes()\n\n\treturn New(&BootstrapParams{\n\t\tLogger:        testlogger.New(t),\n\t\tServiceClient: sdkClient,\n\t\tClientBean:    mockClientBean,\n\t\tTallyScope:    tally.TestScope(nil),\n\t\tConfig: Config{\n\t\t\tClusterMetadata: cluster.NewMetadata(\n\t\t\t\tconfig.ClusterGroupMetadata{\n\t\t\t\t\tFailoverVersionIncrement: 12,\n\t\t\t\t\tPrimaryClusterName:       \"test-primary-cluster\",\n\t\t\t\t\tCurrentClusterName:       \"test-primary-cluster\",\n\t\t\t\t\tClusterGroup: map[string]config.ClusterInformation{\n\t\t\t\t\t\t\"test-primary-cluster\":   {},\n\t\t\t\t\t\t\"test-secondary-cluster\": {},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tnil,\n\t\t\t\tmetrics.NewClient(tally.NoopScope, metrics.Worker, metrics.HistogramMigration{}),\n\t\t\t\ttestlogger.New(t),\n\t\t\t),\n\t\t},\n\t}), mockResource\n}\n"
  },
  {
    "path": "service/worker/batcher/entities.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// TerminateParams is the parameters for terminating workflow\ntype TerminateParams struct {\n\t// this indicates whether to terminate children workflow. Default to true.\n\t// TODO https://github.com/uber/cadence/issues/2159\n\t// Ideally default should be childPolicy of the workflow. But it's currently totally broken.\n\tTerminateChildren *bool\n}\n\n// CancelParams is the parameters for canceling workflow\ntype CancelParams struct {\n\t// this indicates whether to cancel children workflow. Default to true.\n\t// TODO https://github.com/uber/cadence/issues/2159\n\t// Ideally default should be childPolicy of the workflow. But it's currently totally broken.\n\tCancelChildren *bool\n}\n\n// SignalParams is the parameters for signaling workflow\ntype SignalParams struct {\n\tSignalName string\n\tInput      string\n}\n\n// ReplicateParams is the parameters for replicating workflow\ntype ReplicateParams struct {\n\tSourceCluster string\n\tTargetCluster string\n}\n\n// BatchParams is the parameters for batch operation workflow\ntype BatchParams struct {\n\t// Target domain to execute batch operation\n\tDomainName string\n\t// To get the target workflows for processing\n\tQuery string\n\t// Reason for the operation\n\tReason string\n\t// Supporting: reset,terminate\n\tBatchType string\n\n\t// Below are all optional\n\t// TerminateParams is params only for BatchTypeTerminate\n\tTerminateParams TerminateParams\n\t// CancelParams is params only for BatchTypeCancel\n\tCancelParams CancelParams\n\t// SignalParams is params only for BatchTypeSignal\n\tSignalParams SignalParams\n\t// ReplicateParams is params only for BatchTypeReplicate\n\tReplicateParams ReplicateParams\n\t// RPS of processing. Default to DefaultRPS\n\t// TODO we will implement smarter way than this static rate limiter: https://github.com/uber/cadence/issues/2138\n\tRPS int\n\t// Number of goroutines running in parallel to process\n\tConcurrency int\n\t// Number of workflows processed in a batch\n\tPageSize int\n\t// Number of attempts for each workflow to process in case of retryable error before giving up\n\tAttemptsOnRetryableError int\n\t// timeout for activity heartbeat\n\tActivityHeartBeatTimeout time.Duration\n\t// Max number of attempts for the activity to retry. Default to 0 (unlimited)\n\tMaxActivityRetries int\n\t// errors that will not retry which consumes AttemptsOnRetryableError. Default to empty\n\tNonRetryableErrors []string\n\t// internal conversion for NonRetryableErrors\n\t_nonRetryableErrors map[string]struct{}\n}\n\n// HeartBeatDetails is the struct for heartbeat details\ntype HeartBeatDetails struct {\n\tPageToken   []byte\n\tCurrentPage int\n\t// This is just an estimation for visibility\n\tTotalEstimate int64\n\t// Number of workflows processed successfully\n\tSuccessCount int\n\t// Number of workflows that give up due to errors.\n\tErrorCount int\n}\n\ntype taskDetail struct {\n\texecution types.WorkflowExecution\n\tattempts  int\n\t// passing along the current heartbeat details to make heartbeat within a task so that it won't timeout\n\thbd HeartBeatDetails\n}\n"
  },
  {
    "path": "service/worker/batcher/workflow.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tcontextKey string\n)\n\nconst (\n\tBatcherContextKey contextKey = \"batcherContext\"\n\t// BatcherTaskListName is the tasklist name\n\tBatcherTaskListName = \"cadence-sys-batcher-tasklist\"\n\t// BatchWFTypeName is the workflow type\n\tBatchWFTypeName   = \"cadence-sys-batch-workflow\"\n\tbatchActivityName = \"cadence-sys-batch-activity\"\n\t// InfiniteDuration is a long duration(20 yrs) we used for infinite workflow running\n\tInfiniteDuration = 20 * 365 * 24 * time.Hour\n\n\t_nonRetriableReason = \"non-retriable-error\"\n\n\t// DefaultRPS is the default RPS\n\tDefaultRPS = 50\n\t// DefaultConcurrency is the default concurrency\n\tDefaultConcurrency = 5\n\t// DefaultPageSize is the default page size\n\tDefaultPageSize = 1000\n\t// DefaultAttemptsOnRetryableError is the default value for AttemptsOnRetryableError\n\tDefaultAttemptsOnRetryableError = 50\n\t// DefaultActivityHeartBeatTimeout is the default value for ActivityHeartBeatTimeout\n\tDefaultActivityHeartBeatTimeout = time.Second * 10\n\t// DefaultMaxActivityRetries is the default value for MaxActivityRetries\n\tDefaultMaxActivityRetries = 4\n)\n\nconst (\n\t// BatchTypeTerminate is batch type for terminating workflows\n\tBatchTypeTerminate = \"terminate\"\n\t// BatchTypeCancel is the batch type for canceling workflows\n\tBatchTypeCancel = \"cancel\"\n\t// BatchTypeSignal is batch type for signaling workflows\n\tBatchTypeSignal = \"signal\"\n\t// BatchTypeReplicate is batch type for replicating workflows\n\tBatchTypeReplicate = \"replicate\"\n)\n\n// AllBatchTypes is the batch types we supported\nvar AllBatchTypes = []string{BatchTypeTerminate, BatchTypeCancel, BatchTypeSignal, BatchTypeReplicate}\n\nvar (\n\tBatchActivityRetryPolicy = cadence.RetryPolicy{\n\t\tInitialInterval:          10 * time.Second,\n\t\tBackoffCoefficient:       1.7,\n\t\tMaximumInterval:          5 * time.Minute,\n\t\tExpirationInterval:       InfiniteDuration,\n\t\tNonRetriableErrorReasons: []string{_nonRetriableReason},\n\t}\n\n\tbatchActivityOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 5 * time.Minute,\n\t\tStartToCloseTimeout:    InfiniteDuration,\n\t\tRetryPolicy:            &BatchActivityRetryPolicy,\n\t}\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(BatchWorkflow, workflow.RegisterOptions{Name: BatchWFTypeName})\n\tactivity.RegisterWithOptions(BatchActivity, activity.RegisterOptions{Name: batchActivityName})\n}\n\n// BatchWorkflow is the workflow that runs a batch job of resetting workflows\nfunc BatchWorkflow(ctx workflow.Context, batchParams BatchParams) (HeartBeatDetails, error) {\n\tbatchParams = setDefaultParams(batchParams)\n\terr := validateParams(batchParams)\n\tif err != nil {\n\t\treturn HeartBeatDetails{}, err\n\t}\n\tbatchActivityOptions.HeartbeatTimeout = batchParams.ActivityHeartBeatTimeout\n\tbatchActivityOptions.RetryPolicy.MaximumAttempts = int32(batchParams.MaxActivityRetries)\n\topt := workflow.WithActivityOptions(ctx, batchActivityOptions)\n\tvar result HeartBeatDetails\n\terr = workflow.ExecuteActivity(opt, batchActivityName, batchParams).Get(ctx, &result)\n\treturn result, err\n}\n\nfunc validateParams(params BatchParams) error {\n\tif params.BatchType == \"\" ||\n\t\tparams.Reason == \"\" ||\n\t\tparams.DomainName == \"\" ||\n\t\tparams.Query == \"\" {\n\t\treturn fmt.Errorf(\"must provide required parameters: BatchType/Reason/DomainName/Query\")\n\t}\n\tswitch params.BatchType {\n\tcase BatchTypeSignal:\n\t\tif params.SignalParams.SignalName == \"\" {\n\t\t\treturn fmt.Errorf(\"must provide signal name\")\n\t\t}\n\t\treturn nil\n\tcase BatchTypeReplicate:\n\t\tif params.ReplicateParams.SourceCluster == \"\" {\n\t\t\treturn fmt.Errorf(\"must provide source cluster\")\n\t\t}\n\t\tif params.ReplicateParams.TargetCluster == \"\" {\n\t\t\treturn fmt.Errorf(\"must provide target cluster\")\n\t\t}\n\t\treturn nil\n\tcase BatchTypeCancel:\n\t\tfallthrough\n\tcase BatchTypeTerminate:\n\t\treturn nil\n\tdefault:\n\t\treturn fmt.Errorf(\"not supported batch type: %v\", params.BatchType)\n\t}\n}\n\nfunc setDefaultParams(params BatchParams) BatchParams {\n\tif params.RPS <= 0 {\n\t\tparams.RPS = DefaultRPS\n\t}\n\tif params.Concurrency <= 0 {\n\t\tparams.Concurrency = DefaultConcurrency\n\t}\n\tif params.PageSize <= 0 {\n\t\tparams.PageSize = DefaultPageSize\n\t}\n\tif params.AttemptsOnRetryableError <= 0 {\n\t\tparams.AttemptsOnRetryableError = DefaultAttemptsOnRetryableError\n\t}\n\tif params.ActivityHeartBeatTimeout <= 0 {\n\t\tparams.ActivityHeartBeatTimeout = DefaultActivityHeartBeatTimeout\n\t}\n\tif len(params.NonRetryableErrors) > 0 {\n\t\tparams._nonRetryableErrors = make(map[string]struct{}, len(params.NonRetryableErrors))\n\t\tfor _, estr := range params.NonRetryableErrors {\n\t\t\tparams._nonRetryableErrors[estr] = struct{}{}\n\t\t}\n\t}\n\tif params.TerminateParams.TerminateChildren == nil {\n\t\tparams.TerminateParams.TerminateChildren = common.BoolPtr(true)\n\t}\n\tif params.MaxActivityRetries < 0 {\n\t\tparams.MaxActivityRetries = DefaultMaxActivityRetries\n\t}\n\treturn params\n}\n\n// BatchActivity is activity for processing batch operation\nfunc BatchActivity(ctx context.Context, batchParams BatchParams) (HeartBeatDetails, error) {\n\tbatcher := ctx.Value(BatcherContextKey).(*Batcher)\n\tclient := batcher.clientBean.GetFrontendClient()\n\tvar adminClient admin.Client\n\tif batchParams.BatchType == BatchTypeReplicate {\n\t\tcurrentCluster := batcher.cfg.ClusterMetadata.GetCurrentClusterName()\n\t\tif currentCluster != batchParams.ReplicateParams.SourceCluster {\n\t\t\treturn HeartBeatDetails{}, cadence.NewCustomError(_nonRetriableReason, fmt.Sprintf(\"the activity must run in the source cluster, current cluster is %s\", currentCluster))\n\t\t}\n\t\tvar err error\n\t\tadminClient, err = batcher.clientBean.GetRemoteAdminClient(batchParams.ReplicateParams.TargetCluster)\n\t\tif err != nil {\n\t\t\treturn HeartBeatDetails{}, cadence.NewCustomError(_nonRetriableReason, err.Error())\n\t\t}\n\t}\n\n\tdomainResp, err := client.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: &batchParams.DomainName,\n\t})\n\tif err != nil {\n\t\treturn HeartBeatDetails{}, err\n\t}\n\tdomainID := domainResp.GetDomainInfo().GetUUID()\n\thbd, ok := getHeartBeatDetails(ctx)\n\n\tif !ok {\n\t\tresp, err := client.CountWorkflowExecutions(ctx, &types.CountWorkflowExecutionsRequest{\n\t\t\tDomain: batchParams.DomainName,\n\t\t\tQuery:  batchParams.Query,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn HeartBeatDetails{}, err\n\t\t}\n\t\thbd.TotalEstimate = resp.GetCount()\n\t}\n\trateLimiter := rate.NewLimiter(rate.Limit(batchParams.RPS), batchParams.RPS)\n\ttaskCh := make(chan taskDetail, batchParams.PageSize)\n\trespCh := make(chan error, batchParams.PageSize)\n\tfor i := 0; i < batchParams.Concurrency; i++ {\n\t\tgo startTaskProcessor(ctx, batchParams, domainID, taskCh, respCh, rateLimiter, client, adminClient)\n\t}\n\n\tfor {\n\t\t// TODO https://github.com/uber/cadence/issues/2154\n\t\t//  Need to improve scan concurrency because it will hold an ES resource until the workflow finishes.\n\t\t//  And we can't use list API because terminate / reset will mutate the result.\n\t\tresp, err := client.ScanWorkflowExecutions(ctx, &types.ListWorkflowExecutionsRequest{\n\t\t\tDomain:        batchParams.DomainName,\n\t\t\tPageSize:      int32(batchParams.PageSize),\n\t\t\tNextPageToken: hbd.PageToken,\n\t\t\tQuery:         batchParams.Query,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn HeartBeatDetails{}, err\n\t\t}\n\t\tbatchCount := len(resp.Executions)\n\t\tif batchCount <= 0 {\n\t\t\tbreak\n\t\t}\n\n\t\t// send all tasks\n\t\tfor _, wf := range resp.Executions {\n\t\t\ttaskCh <- taskDetail{\n\t\t\t\texecution: *wf.Execution,\n\t\t\t\tattempts:  0,\n\t\t\t\thbd:       hbd,\n\t\t\t}\n\t\t}\n\n\t\tsuccCount := 0\n\t\terrCount := 0\n\t\t// wait for counters indicate this batch is done\n\tLoop:\n\t\tfor {\n\t\t\tselect {\n\t\t\tcase err := <-respCh:\n\t\t\t\tif err == nil {\n\t\t\t\t\tsuccCount++\n\t\t\t\t} else {\n\t\t\t\t\terrCount++\n\t\t\t\t}\n\t\t\t\tif succCount+errCount == batchCount {\n\t\t\t\t\tbreak Loop\n\t\t\t\t}\n\t\t\tcase <-ctx.Done():\n\t\t\t\treturn HeartBeatDetails{}, ctx.Err()\n\t\t\t}\n\t\t}\n\n\t\thbd.CurrentPage++\n\t\thbd.PageToken = resp.NextPageToken\n\t\thbd.SuccessCount += succCount\n\t\thbd.ErrorCount += errCount\n\t\tactivity.RecordHeartbeat(ctx, hbd)\n\n\t\tif len(hbd.PageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t}\n\n\treturn hbd, nil\n}\n\nfunc getHeartBeatDetails(ctx context.Context) (hbd HeartBeatDetails, ok bool) {\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\tif err := activity.GetHeartbeatDetails(ctx, &hbd); err != nil {\n\t\t\tbatcher := ctx.Value(BatcherContextKey).(*Batcher)\n\t\t\tbatcher.metricsClient.IncCounter(metrics.BatcherScope, metrics.BatcherProcessorFailures)\n\t\t\tgetActivityLogger(ctx).Error(\"Failed to recover from last heartbeat, start over from beginning\", tag.Error(err))\n\t\t\treturn HeartBeatDetails{}, false\n\t\t}\n\t\treturn hbd, true\n\t}\n\n\treturn hbd, false\n}\n\nfunc startTaskProcessor(\n\tctx context.Context,\n\tbatchParams BatchParams,\n\tdomainID string,\n\ttaskCh chan taskDetail,\n\trespCh chan error,\n\tlimiter *rate.Limiter,\n\tclient frontend.Client,\n\tadminClient admin.Client,\n) {\n\tbatcher := ctx.Value(BatcherContextKey).(*Batcher)\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn\n\t\tcase task := <-taskCh:\n\t\t\tif isDone(ctx) {\n\t\t\t\treturn\n\t\t\t}\n\t\t\tvar err error\n\t\t\trequestID := uuid.New().String()\n\n\t\t\tswitch batchParams.BatchType {\n\t\t\tcase BatchTypeTerminate:\n\t\t\t\terr = processTask(ctx, limiter, task, batchParams, client,\n\t\t\t\t\tbatchParams.TerminateParams.TerminateChildren,\n\t\t\t\t\tfunc(workflowID, runID string) error {\n\t\t\t\t\t\treturn client.TerminateWorkflowExecution(ctx, &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: batchParams.DomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tReason:   batchParams.Reason,\n\t\t\t\t\t\t\tIdentity: BatchWFTypeName,\n\t\t\t\t\t\t})\n\t\t\t\t\t})\n\t\t\tcase BatchTypeCancel:\n\t\t\t\terr = processTask(ctx, limiter, task, batchParams, client,\n\t\t\t\t\tbatchParams.CancelParams.CancelChildren,\n\t\t\t\t\tfunc(workflowID, runID string) error {\n\t\t\t\t\t\treturn client.RequestCancelWorkflowExecution(ctx, &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: batchParams.DomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tIdentity:  BatchWFTypeName,\n\t\t\t\t\t\t\tRequestID: requestID,\n\t\t\t\t\t\t})\n\t\t\t\t\t})\n\t\t\tcase BatchTypeSignal:\n\t\t\t\terr = processTask(ctx, limiter, task, batchParams, client, common.BoolPtr(false),\n\t\t\t\t\tfunc(workflowID, runID string) error {\n\t\t\t\t\t\treturn client.SignalWorkflowExecution(ctx, &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: batchParams.DomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: workflowID,\n\t\t\t\t\t\t\t\tRunID:      runID,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tIdentity:   BatchWFTypeName,\n\t\t\t\t\t\t\tRequestID:  requestID,\n\t\t\t\t\t\t\tSignalName: batchParams.SignalParams.SignalName,\n\t\t\t\t\t\t\tInput:      []byte(batchParams.SignalParams.Input),\n\t\t\t\t\t\t})\n\t\t\t\t\t})\n\t\t\tcase BatchTypeReplicate:\n\t\t\t\terr = processTask(ctx, limiter, task, batchParams, client, common.BoolPtr(false),\n\t\t\t\t\tfunc(workflowID, runID string) error {\n\t\t\t\t\t\treturn adminClient.ResendReplicationTasks(ctx, &types.ResendReplicationTasksRequest{\n\t\t\t\t\t\t\tDomainID:      domainID,\n\t\t\t\t\t\t\tWorkflowID:    workflowID,\n\t\t\t\t\t\t\tRunID:         runID,\n\t\t\t\t\t\t\tRemoteCluster: batchParams.ReplicateParams.SourceCluster,\n\t\t\t\t\t\t})\n\t\t\t\t\t})\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\tbatcher.metricsClient.IncCounter(metrics.BatcherScope, metrics.BatcherProcessorFailures)\n\t\t\t\tgetActivityLogger(ctx).Error(\"Failed to process batch operation task\", tag.Error(err))\n\n\t\t\t\t_, ok := batchParams._nonRetryableErrors[err.Error()]\n\t\t\t\tif ok || task.attempts >= batchParams.AttemptsOnRetryableError {\n\t\t\t\t\trespCh <- err\n\t\t\t\t} else {\n\t\t\t\t\t// put back to the channel if less than attemptsOnError\n\t\t\t\t\ttask.attempts++\n\t\t\t\t\ttaskCh <- task\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tbatcher.metricsClient.IncCounter(metrics.BatcherScope, metrics.BatcherProcessorSuccess)\n\t\t\t\trespCh <- nil\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc processTask(\n\tctx context.Context,\n\tlimiter *rate.Limiter,\n\ttask taskDetail,\n\tbatchParams BatchParams,\n\tclient frontend.Client,\n\tapplyOnChild *bool,\n\tprocFn func(string, string) error,\n) error {\n\twfs := []types.WorkflowExecution{task.execution}\n\tfor len(wfs) > 0 {\n\t\twf := wfs[0]\n\t\twfs = wfs[1:]\n\n\t\terr := limiter.Wait(ctx)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tactivity.RecordHeartbeat(ctx, task.hbd)\n\n\t\terr = procFn(wf.GetWorkflowID(), wf.GetRunID())\n\t\tif err != nil {\n\t\t\t// EntityNotExistsError means wf is not running or deleted\n\t\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\t\tresp, err := client.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: batchParams.DomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wf.WorkflowID,\n\t\t\t\tRunID:      wf.RunID,\n\t\t\t},\n\t\t})\n\t\tif err != nil {\n\t\t\t// EntityNotExistsError means wf is deleted\n\t\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\treturn err\n\t\t}\n\n\t\t// TODO https://github.com/uber/cadence/issues/2159\n\t\t// By default should use ChildPolicy, but it is totally broken in Cadence, we need to fix it before using\n\t\tif applyOnChild != nil && *applyOnChild && len(resp.PendingChildren) > 0 {\n\t\t\tgetActivityLogger(ctx).Info(\"Found more child workflows to process\", tag.Number(int64(len(resp.PendingChildren))))\n\t\t\tfor _, ch := range resp.PendingChildren {\n\t\t\t\twfs = append(wfs, types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: ch.WorkflowID,\n\t\t\t\t\tRunID:      ch.RunID,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc isDone(ctx context.Context) bool {\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n\nfunc getActivityLogger(ctx context.Context) log.Logger {\n\tbatcher := ctx.Value(BatcherContextKey).(*Batcher)\n\twfInfo := activity.GetInfo(ctx)\n\treturn batcher.logger.WithTags(\n\t\ttag.WorkflowID(wfInfo.WorkflowExecution.ID),\n\t\ttag.WorkflowRunID(wfInfo.WorkflowExecution.RunID),\n\t\ttag.WorkflowDomainName(wfInfo.WorkflowDomain),\n\t)\n}\n"
  },
  {
    "path": "service/worker/batcher/workflow_retry_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype workflowRetrySuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tworkflowEnv  *testsuite.TestWorkflowEnvironment\n\tactivityEnv  *testsuite.TestActivityEnvironment\n\tmockResource *resource.Test\n\tmetricsMock  *mmocks.Client\n}\n\nfunc TestWorkflowRetrySuite(t *testing.T) {\n\tsuite.Run(t, new(workflowRetrySuite))\n}\n\nfunc (s *workflowRetrySuite) SetupTest() {\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\ts.workflowEnv.RegisterWorkflow(BatchWorkflow)\n\n\ts.activityEnv = s.NewTestActivityEnvironment()\n\ts.activityEnv.RegisterActivity(BatchActivity)\n\n\tbatcher, mockResource := setuptest(s.T())\n\ts.mockResource = mockResource\n\n\ts.metricsMock = &mmocks.Client{}\n\tbatcher.metricsClient = s.metricsMock\n\n\tmockResource.FrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{}, nil).AnyTimes()\n\n\tmockResource.FrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{Count: 1}, nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{}, nil).AnyTimes()\n\n\tctx := context.WithValue(context.Background(), BatcherContextKey, batcher)\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              tally.TestScope(nil),\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\ts.workflowEnv.SetWorkerOptions(workerOpts)\n\ts.activityEnv.SetWorkerOptions(workerOpts)\n\n}\n\nfunc (s *workflowRetrySuite) TestActivityRetries() {\n\tparams := createParams(BatchTypeCancel)\n\tparams.MaxActivityRetries = 4\n\n\ts.metricsMock.On(\"IncCounter\", metrics.BatcherScope, metrics.BatcherProcessorSuccess).Times(2)\n\ts.metricsMock.On(\"IncCounter\", metrics.BatcherScope, metrics.BatcherProcessorFailures).Times(5)\n\n\t// First call to scan workflow executions succeeds, but returns a bad page token\n\ts.mockResource.FrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        params.DomainName,\n\t\tPageSize:      int32(params.PageSize),\n\t\tNextPageToken: nil,\n\t\tQuery:         params.Query,\n\t}).\n\t\tReturn(&types.ListWorkflowExecutionsResponse{\n\t\t\tExecutions:    []*types.WorkflowExecutionInfo{{Execution: &types.WorkflowExecution{WorkflowID: \"wid\", RunID: \"rid\"}}},\n\t\t\tNextPageToken: []byte(\"bad-page-token\"),\n\t\t}, nil).Times(1)\n\n\t// Max attempt is set to 4, so we try 5 times, 0,1,2,3,4\n\ts.mockResource.FrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        params.DomainName,\n\t\tPageSize:      int32(params.PageSize),\n\t\tNextPageToken: []byte(\"bad-page-token\"),\n\t\tQuery:         params.Query,\n\t}).\n\t\tReturn(&types.ListWorkflowExecutionsResponse{\n\t\t\tExecutions:    []*types.WorkflowExecutionInfo{{Execution: &types.WorkflowExecution{WorkflowID: \"wid\", RunID: \"rid\"}}},\n\t\t\tNextPageToken: []byte(\"bad-page-token\"),\n\t\t}, assert.AnError).Times(5)\n\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc (s *workflowRetrySuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n"
  },
  {
    "path": "service/worker/batcher/workflow_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage batcher\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tmmocks \"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype workflowSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tworkflowEnv *testsuite.TestWorkflowEnvironment\n\tactivityEnv *testsuite.TestActivityEnvironment\n}\n\nfunc TestWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(workflowSuite))\n}\n\nfunc (s *workflowSuite) SetupTest() {\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\ts.workflowEnv.RegisterWorkflow(BatchWorkflow)\n\n\ts.activityEnv = s.NewTestActivityEnvironment()\n\ts.activityEnv.RegisterActivity(BatchActivity)\n\n\tbatcher, mockResource := setuptest(s.T())\n\n\tmetricsMock := &mmocks.Client{}\n\tmetricsMock.On(\"IncCounter\", metrics.BatcherScope, metrics.BatcherProcessorSuccess).Once()\n\tbatcher.metricsClient = metricsMock\n\n\tmockResource.FrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{}, nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{\n\t\tExecutions:    []*types.WorkflowExecutionInfo{{Execution: &types.WorkflowExecution{WorkflowID: \"wid\", RunID: \"rid\"}}},\n\t\tNextPageToken: nil,\n\t}, nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{Count: 1}, nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{}, nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\tmockResource.FrontendClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\n\tmockResource.RemoteAdminClient.EXPECT().ResendReplicationTasks(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()\n\n\tctx := context.WithValue(context.Background(), BatcherContextKey, batcher)\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              tally.TestScope(nil),\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\ts.activityEnv.SetWorkerOptions(workerOpts)\n\n}\n\nfunc (s *workflowSuite) TestWorkflow() {\n\tparams := createParams(BatchTypeCancel)\n\tactivityHeartBeatDeatils := HeartBeatDetails{}\n\ts.workflowEnv.OnActivity(batchActivityName, mock.Anything, mock.Anything).Return(activityHeartBeatDeatils, nil)\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.NoError(s.workflowEnv.GetWorkflowError())\n}\n\nfunc (s *workflowSuite) TestActivity_BatchCancel() {\n\tparams := createParams(BatchTypeCancel)\n\t_, err := s.activityEnv.ExecuteActivity(BatchActivity, params)\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestActivity_BatchTerminate() {\n\tparams := createParams(BatchTypeTerminate)\n\t_, err := s.activityEnv.ExecuteActivity(BatchActivity, params)\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestActivity_BatchSignal() {\n\tparams := createParams(BatchTypeSignal)\n\t_, err := s.activityEnv.ExecuteActivity(BatchActivity, params)\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestActivity_BatchReplicate() {\n\tparams := createParams(BatchTypeReplicate)\n\t_, err := s.activityEnv.ExecuteActivity(BatchActivity, params)\n\ts.NoError(err)\n}\n\nfunc (s *workflowSuite) TestWorkflow_BatchTypeCancelValidationError() {\n\tparams := createParams(BatchTypeCancel)\n\tparams.Query = \"\"\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), \"must provide required parameters: BatchType/Reason/DomainName/Query\")\n}\n\nfunc (s *workflowSuite) TestWorkflow_UnsupportedBatchType() {\n\tparams := createParams(\"invalid-batch-type\")\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), \"not supported batch type\")\n}\n\nfunc (s *workflowSuite) TestWorkflow_BatchTypeSignalValidation() {\n\tparams := createParams(BatchTypeSignal)\n\tparams.SignalParams.SignalName = \"\"\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), \"must provide signal name\")\n}\n\nfunc (s *workflowSuite) TestWorkflow_BatchTypeReplicateSourceCLusterValidation() {\n\tparams := createParams(BatchTypeReplicate)\n\tparams.ReplicateParams.SourceCluster = \"\"\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), \"must provide source cluster\")\n}\n\nfunc (s *workflowSuite) TestWorkflow_BatchTypeReplicateTargetCLusterValidation() {\n\tparams := createParams(BatchTypeReplicate)\n\tparams.ReplicateParams.TargetCluster = \"\"\n\ts.workflowEnv.ExecuteWorkflow(BatchWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), \"must provide target cluster\")\n}\n\nfunc (s *workflowSuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc createParams(batchType string) BatchParams {\n\treturn BatchParams{\n\t\tDomainName: \"test-domain\",\n\t\tQuery:      \"Closetime=missing\",\n\t\tReason:     \"unit-test\",\n\t\tBatchType:  batchType,\n\t\tTerminateParams: TerminateParams{\n\t\t\tTerminateChildren: common.BoolPtr(true),\n\t\t},\n\t\tCancelParams: CancelParams{\n\t\t\tCancelChildren: common.BoolPtr(true),\n\t\t},\n\t\tSignalParams: SignalParams{\n\t\t\tSignalName: \"test-signal-name\",\n\t\t},\n\t\tReplicateParams: ReplicateParams{\n\t\t\tSourceCluster: \"test-primary-cluster\",\n\t\t\tTargetCluster: \"test-secondary-cluster\",\n\t\t},\n\t\tRPS:                      5,\n\t\tConcurrency:              5,\n\t\tPageSize:                 10,\n\t\tAttemptsOnRetryableError: 0,\n\t\tActivityHeartBeatTimeout: 0,\n\t\tMaxActivityRetries:       0,\n\t\tNonRetryableErrors:       []string{\"HeartbeatTimeoutError\"},\n\t\t_nonRetryableErrors:      nil,\n\t}\n}\n"
  },
  {
    "path": "service/worker/diagnostics/activities.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/analytics\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\nconst (\n\tlinkToTimeoutsRunbook = \"https://cadenceworkflow.io/docs/workflow-troubleshooting/timeouts/\"\n\tlinkToFailuresRunbook = \"https://cadenceworkflow.io/docs/workflow-troubleshooting/activity-failures/\"\n\tlinkToRetriesRunbook  = \"https://cadenceworkflow.io/docs/workflow-troubleshooting/retries\"\n\tWfDiagnosticsAppName  = \"workflow-diagnostics\"\n\n\t_maxPageSize           = 1000            // current maximum page size for fetching workflow history\n\t_contextTimeout        = 1 * time.Minute // timeout to fetch the whole execution history\n\t_maxIssuesPerInvariant = 10              // maximum number of issues to return per invariant check\n)\n\ntype identifyIssuesParams struct {\n\tExecution *types.WorkflowExecution\n\tDomain    string\n}\n\nfunc (w *dw) identifyIssues(ctx context.Context, info identifyIssuesParams) ([]invariant.InvariantCheckResult, error) {\n\tresult := make([]invariant.InvariantCheckResult, 0)\n\n\thistory, err := w.getWorkflowExecutionHistory(ctx, info.Execution, info.Domain)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tfor _, inv := range w.invariants {\n\t\tissues, err := inv.Check(ctx, invariant.InvariantCheckInput{\n\t\t\tWorkflowExecutionHistory: history,\n\t\t\tDomain:                   info.Domain,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult = append(result, capNumberOfIssues(issues)...)\n\t}\n\n\treturn result, nil\n}\n\nfunc (w *dw) getWorkflowExecutionHistory(ctx context.Context, execution *types.WorkflowExecution, domain string) (*types.GetWorkflowExecutionHistoryResponse, error) {\n\tfrontendClient := w.clientBean.GetFrontendClient()\n\tvar nextPageToken []byte\n\tvar history []*types.HistoryEvent\n\n\tctx, cancel := context.WithTimeout(ctx, _contextTimeout)\n\tdefer cancel()\n\n\tfor {\n\n\t\tresponse, err := frontendClient.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain:                 domain,\n\t\t\tExecution:              execution,\n\t\t\tMaximumPageSize:        _maxPageSize,\n\t\t\tNextPageToken:          nextPageToken,\n\t\t\tWaitForNewEvent:        false,\n\t\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t\tSkipArchival:           true,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to get history: %w\", err)\n\t\t}\n\n\t\tfor _, event := range response.GetHistory().GetEvents() {\n\t\t\tif event != nil {\n\t\t\t\thistory = append(history, event)\n\t\t\t}\n\t\t}\n\n\t\tif response.NextPageToken == nil {\n\t\t\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\t\t\tHistory: &types.History{\n\t\t\t\t\tEvents: history,\n\t\t\t\t}}, nil\n\t\t}\n\n\t\tnextPageToken = response.NextPageToken\n\t}\n}\n\ntype rootCauseIssuesParams struct {\n\tDomain string\n\tIssues []invariant.InvariantCheckResult\n}\n\nfunc (w *dw) rootCauseIssues(ctx context.Context, info rootCauseIssuesParams) ([]invariant.InvariantRootCauseResult, error) {\n\tresult := make([]invariant.InvariantRootCauseResult, 0)\n\n\tfor _, inv := range w.invariants {\n\t\trootCause, err := inv.RootCause(ctx, invariant.InvariantRootCauseInput{\n\t\t\tDomain: info.Domain,\n\t\t\tIssues: info.Issues,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult = append(result, rootCause...)\n\t}\n\n\treturn result, nil\n}\n\nfunc (w *dw) emitUsageLogs(ctx context.Context, info analytics.WfDiagnosticsUsageData) error {\n\tif w.messagingClient == nil {\n\t\t// skip emitting logs if messaging client is not provided since it is optional\n\t\tw.logger.Error(\"messaging client is not provided, skipping emitting wf-diagnostics usage logs\", tag.WorkflowDomainName(info.Domain))\n\t\treturn nil\n\t}\n\treturn w.emit(ctx, info, w.messagingClient)\n}\n\nfunc (w *dw) emit(ctx context.Context, info analytics.WfDiagnosticsUsageData, client messaging.Client) error {\n\tproducer, err := client.NewProducer(WfDiagnosticsAppName)\n\tif err != nil {\n\t\t// skip emitting logs if producer cannot be created since it is optional\n\t\tw.logger.Error(\"producer creation failed, skipping emitting wf-diagnostics usage logs\", tag.WorkflowDomainName(info.Domain))\n\t\treturn nil\n\t}\n\temitter := analytics.NewEmitter(analytics.EmitterParams{\n\t\tProducer: producer,\n\t})\n\treturn emitter.EmitUsageData(ctx, info)\n}\n\n// capNumberOfIssues limits the number of issues to avoid overwhelming the result with too many issues.\nfunc capNumberOfIssues(issues []invariant.InvariantCheckResult) []invariant.InvariantCheckResult {\n\tif len(issues) > _maxIssuesPerInvariant {\n\t\treturn issues[:_maxIssuesPerInvariant]\n\t}\n\treturn issues\n}\n"
  },
  {
    "path": "service/worker/diagnostics/activities_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/analytics\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/failure\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/retry\"\n)\n\nconst (\n\tworkflowTimeoutSecond = int32(110)\n\ttestTimeStamp         = int64(2547596872371000000)\n\ttimeUnit              = time.Second\n)\n\nfunc Test__identifyIssues(t *testing.T) {\n\tdwtest := testDiagnosticWorkflow(t, testWorkflowExecutionHistoryResponseWithMultipleIssues())\n\tactMetadata := failure.FailureIssuesMetadata{\n\t\tIdentity:            \"localhost\",\n\t\tActivityType:        \"test-activity\",\n\t\tActivityScheduledID: 2,\n\t\tActivityStartedID:   3,\n\t}\n\tactMetadataInBytes, err := json.Marshal(actMetadata)\n\trequire.NoError(t, err)\n\texpectedResult := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       0,\n\t\t\tInvariantType: failure.ActivityFailed.String(),\n\t\t\tReason:        failure.GenericError.String(),\n\t\t\tMetadata:      actMetadataInBytes,\n\t\t},\n\t}\n\tfor i := 0; i < _maxIssuesPerInvariant; i++ {\n\t\tretryMetadata := retry.RetryMetadata{\n\t\t\tEventID: int64(i),\n\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\tMaximumAttempts:          1,\n\t\t\t},\n\t\t}\n\t\tretryMetadataInBytes, err := json.Marshal(retryMetadata)\n\t\trequire.NoError(t, err)\n\t\texpectedResult = append(expectedResult, invariant.InvariantCheckResult{\n\t\t\tIssueID:       i,\n\t\t\tInvariantType: retry.ActivityRetryIssue.String(),\n\t\t\tReason:        retry.RetryPolicyValidationMaxAttempts.String(),\n\t\t\tMetadata:      retryMetadataInBytes,\n\t\t})\n\t}\n\tresult, err := dwtest.identifyIssues(context.Background(), identifyIssuesParams{Execution: &types.WorkflowExecution{\n\t\tWorkflowID: \"123\",\n\t\tRunID:      \"abc\",\n\t}})\n\trequire.NoError(t, err)\n\trequire.Equal(t, _maxIssuesPerInvariant+1, len(result)) // retry invariant returns 10 issues (capped) , failure invariant returns 1 issue\n\trequire.Equal(t, expectedResult, result)\n}\n\nfunc Test__rootCauseIssues(t *testing.T) {\n\tdwtest := testDiagnosticWorkflow(t, testWorkflowExecutionHistoryResponse())\n\tactMetadata := failure.FailureIssuesMetadata{\n\t\tIdentity:            \"localhost\",\n\t\tActivityScheduledID: 1,\n\t\tActivityStartedID:   2,\n\t}\n\tactMetadataInBytes, err := json.Marshal(actMetadata)\n\trequire.NoError(t, err)\n\tissues := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       0,\n\t\t\tInvariantType: failure.ActivityFailed.String(),\n\t\t\tReason:        failure.CustomError.String(),\n\t\t\tMetadata:      actMetadataInBytes,\n\t\t},\n\t}\n\texpectedRootCause := []invariant.InvariantRootCauseResult{\n\t\t{\n\t\t\tIssueID:   0,\n\t\t\tRootCause: invariant.RootCauseTypeServiceSideCustomError,\n\t\t\tMetadata:  actMetadataInBytes,\n\t\t},\n\t}\n\tresult, err := dwtest.rootCauseIssues(context.Background(), rootCauseIssuesParams{Domain: \"test-domain\", Issues: issues})\n\trequire.NoError(t, err)\n\trequire.Equal(t, expectedRootCause, result)\n}\n\nfunc Test__emit(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdwtest := testDiagnosticWorkflow(t, testWorkflowExecutionHistoryResponse())\n\tmockClient := messaging.NewMockClient(ctrl)\n\tmockProducer := messaging.NewMockProducer(ctrl)\n\tmockProducer.EXPECT().Publish(gomock.Any(), gomock.Any()).Return(nil)\n\tmockClient.EXPECT().NewProducer(WfDiagnosticsAppName).Return(mockProducer, nil)\n\terr := dwtest.emit(context.Background(), analytics.WfDiagnosticsUsageData{}, mockClient)\n\trequire.NoError(t, err)\n}\n\nfunc testDiagnosticWorkflow(t *testing.T, history *types.GetWorkflowExecutionHistoryResponse) *dw {\n\tctrl := gomock.NewController(t)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockFrontendClient := frontend.NewMockClient(ctrl)\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockFrontendClient).AnyTimes()\n\tmockFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(history, nil).AnyTimes()\n\treturn &dw{\n\t\tclientBean: mockClientBean,\n\t\tinvariants: []invariant.Invariant{failure.NewInvariant(), retry.NewInvariant()},\n\t}\n}\n\nfunc testWorkflowExecutionHistoryResponse() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 2,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityID:                 \"101\",\n\t\t\t\t\t\tActivityType:               &types.ActivityType{Name: \"test-activity\"},\n\t\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(int32(10)),\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:    common.Int32Ptr(int32(5)),\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\t\t\t\tMaximumAttempts:          1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 3,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tIdentity: \"localhost\",\n\t\t\t\t\t\tAttempt:  0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        4,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tReason:           common.StringPtr(\"cadenceInternal:Generic\"),\n\t\t\t\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tIdentity:         \"localhost\",\n\t\t\t\t\t\tScheduledEventID: 2,\n\t\t\t\t\t\tStartedEventID:   3,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:                                       5,\n\t\t\t\t\tTimestamp:                                common.Int64Ptr(testTimeStamp + int64(workflowTimeoutSecond)*timeUnit.Nanoseconds()),\n\t\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{TimeoutType: types.TimeoutTypeStartToClose.Ptr()},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc testWorkflowExecutionHistoryResponseWithMultipleIssues() *types.GetWorkflowExecutionHistoryResponse {\n\ttestResponse := &types.GetWorkflowExecutionHistoryResponse{History: &types.History{\n\t\tEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}}\n\tfor i := 0; i <= 20; i++ {\n\t\ttestResponse.History.Events = append(testResponse.History.Events, &types.HistoryEvent{\n\t\t\tID: int64(i),\n\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\tActivityID:                 string(rune(i)),\n\t\t\t\tActivityType:               &types.ActivityType{Name: \"test-activity\"},\n\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(int32(10)),\n\t\t\t\tHeartbeatTimeoutSeconds:    common.Int32Ptr(int32(5)),\n\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\t\tMaximumAttempts:          1,\n\t\t\t\t},\n\t\t\t},\n\t\t})\n\n\t}\n\ttestResponse.History.Events = append(testResponse.History.Events, &types.HistoryEvent{\n\t\tID:        4,\n\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\tReason:           common.StringPtr(\"cadenceInternal:Generic\"),\n\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\tIdentity:         \"localhost\",\n\t\t\tScheduledEventID: 2,\n\t\t\tStartedEventID:   3,\n\t\t},\n\t})\n\n\treturn testResponse\n}\n\nfunc Test__identifyIssuesWithPaginatedHistory(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockFrontendClient := frontend.NewMockClient(ctrl)\n\ttoken := []byte(\"next-page-token\")\n\ttestExecution := &types.WorkflowExecution{\n\t\tWorkflowID: \"123\",\n\t\tRunID:      \"abc\",\n\t}\n\tpartialWFHistoryResponse := &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\t\t\t\tMaximumAttempts:          1,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tAttempt: 0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: token,\n\t}\n\tremainingWFHistoryResponse := &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\t\tReason:                       common.StringPtr(\"cadenceInternal:Timeout START_TO_CLOSE\"),\n\t\t\t\t\t\tDecisionTaskCompletedEventID: 10,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockFrontendClient).AnyTimes()\n\tfirstCall := mockFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution:              testExecution,\n\t\tMaximumPageSize:        1000,\n\t\tNextPageToken:          nil,\n\t\tWaitForNewEvent:        false,\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\tSkipArchival:           true,\n\t}).Return(partialWFHistoryResponse, nil)\n\tmockFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tExecution:              testExecution,\n\t\tMaximumPageSize:        1000,\n\t\tNextPageToken:          token,\n\t\tWaitForNewEvent:        false,\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\tSkipArchival:           true,\n\t}).Return(remainingWFHistoryResponse, nil).After(firstCall)\n\n\tretryMetadata := retry.RetryMetadata{\n\t\tEventID: 1,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds: 1,\n\t\t\tMaximumAttempts:          1,\n\t\t},\n\t}\n\tretryMetadataInBytes, err := json.Marshal(retryMetadata)\n\trequire.NoError(t, err)\n\tfailureMetadataInBytes, err := json.Marshal(failure.FailureIssuesMetadata{})\n\trequire.NoError(t, err)\n\texpectedResult := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       0,\n\t\t\tInvariantType: retry.WorkflowRetryIssue.String(),\n\t\t\tReason:        \"MaximumAttempts set to 1 will not retry since maximum attempts includes the first attempt.\",\n\t\t\tMetadata:      retryMetadataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       0,\n\t\t\tInvariantType: failure.WorkflowFailed.String(),\n\t\t\tReason:        \"The failure is caused by a timeout during the execution\",\n\t\t\tMetadata:      failureMetadataInBytes,\n\t\t},\n\t}\n\n\tdwtest := &dw{\n\t\tclientBean: mockClientBean,\n\t\tinvariants: []invariant.Invariant{retry.NewInvariant(), failure.NewInvariant()},\n\t}\n\n\tresult, err := dwtest.identifyIssues(context.Background(), identifyIssuesParams{Execution: testExecution})\n\trequire.NoError(t, err)\n\trequire.Equal(t, expectedResult, result)\n}\n"
  },
  {
    "path": "service/worker/diagnostics/analytics/emitter.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage analytics\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/messaging\"\n)\n\ntype Emitter DataEmitter\n\ntype emitter struct {\n\tproducer messaging.Producer\n}\n\ntype EmitterParams struct {\n\tProducer messaging.Producer\n}\n\nfunc NewEmitter(p EmitterParams) DataEmitter {\n\treturn &emitter{\n\t\tproducer: p.Producer,\n\t}\n}\n\nfunc (et *emitter) EmitUsageData(ctx context.Context, data WfDiagnosticsUsageData) error {\n\tmsg := make(map[string]interface{})\n\tmsg[Domain] = data.Domain\n\tmsg[WorkflowID] = data.WorkflowID\n\tmsg[RunID] = data.RunID\n\tmsg[Identity] = data.Identity\n\tmsg[SatisfactionFeedback] = data.SatisfactionFeedback\n\tmsg[IssueType] = data.IssueType\n\tmsg[DiagnosticsWfID] = data.DiagnosticsWorkflowID\n\tmsg[DiagnosticsWfRunID] = data.DiagnosticsRunID\n\tmsg[Environment] = data.Environment\n\tmsg[DiagnosticsStartTime] = data.DiagnosticsStartTime.UTC().UnixMilli()\n\tmsg[DiagnosticsEndTime] = data.DiagnosticsEndTime.UTC().UnixMilli()\n\n\tserializedMsg, err := json.Marshal(msg)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tpinotMsg := &indexer.PinotMessage{\n\t\tWorkflowID: common.StringPtr(data.DiagnosticsWorkflowID),\n\t\tPayload:    serializedMsg,\n\t}\n\treturn et.producer.Publish(ctx, pinotMsg)\n}\n"
  },
  {
    "path": "service/worker/diagnostics/analytics/emitter_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage analytics\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/mocks\"\n)\n\nfunc Test__EmitUsageData(t *testing.T) {\n\n\ttestdata := WfDiagnosticsUsageData{\n\t\tDomain:                \"test-domain\",\n\t\tWorkflowID:            \"wid\",\n\t\tRunID:                 \"rid\",\n\t\tIdentity:              \"test@uber.com\",\n\t\tIssueType:             \"timeout\",\n\t\tDiagnosticsWorkflowID: \"diagnostics-wid\",\n\t\tDiagnosticsRunID:      \"diagnostics-rid\",\n\t\tEnvironment:           \"test-env\",\n\t\tDiagnosticsStartTime:  time.Now(),\n\t\tDiagnosticsEndTime:    time.Now().Add(1 * time.Minute),\n\t\tSatisfactionFeedback:  false,\n\t}\n\tmockErr := errors.New(\"mockErr\")\n\ttests := map[string]struct {\n\t\tdata                   WfDiagnosticsUsageData\n\t\tproducerMockAffordance func(mockProducer *mocks.KafkaProducer)\n\t\texpectedError          error\n\t}{\n\t\t\"Case1: normal case\": {\n\t\t\tdata: testdata,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\trequire.Equal(t, testdata.DiagnosticsWorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(nil).Once()\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t\"Case1: error case\": {\n\t\t\tdata: testdata,\n\t\t\tproducerMockAffordance: func(mockProducer *mocks.KafkaProducer) {\n\t\t\t\tmockProducer.On(\"Publish\", mock.Anything, mock.MatchedBy(func(input *indexer.PinotMessage) bool {\n\t\t\t\t\trequire.Equal(t, testdata.DiagnosticsWorkflowID, input.GetWorkflowID())\n\t\t\t\t\treturn true\n\t\t\t\t})).Return(mockErr).Once()\n\t\t\t},\n\t\t\texpectedError: mockErr,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tmockProducer := &mocks.KafkaProducer{}\n\t\t\temitter := NewEmitter(EmitterParams{\n\t\t\t\tProducer: mockProducer,\n\t\t\t})\n\t\t\ttest.producerMockAffordance(mockProducer)\n\n\t\t\terr := emitter.EmitUsageData(context.Background(), test.data)\n\t\t\tif test.expectedError != nil {\n\t\t\t\trequire.Error(t, err)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/diagnostics/analytics/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage analytics\n\nimport (\n\t\"context\"\n\t\"time\"\n)\n\ntype WfDiagnosticsUsageData struct {\n\tDomain                string\n\tWorkflowID            string\n\tRunID                 string\n\tIdentity              string\n\tIssueType             string\n\tDiagnosticsWorkflowID string\n\tDiagnosticsRunID      string\n\tEnvironment           string\n\tDiagnosticsStartTime  time.Time\n\tDiagnosticsEndTime    time.Time\n\tSatisfactionFeedback  bool\n}\n\n// DataEmitter is the interface to emit workflow diagnostics data\ntype DataEmitter interface {\n\tEmitUsageData(context.Context, WfDiagnosticsUsageData) error\n}\n"
  },
  {
    "path": "service/worker/diagnostics/analytics/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage analytics\n\nconst (\n\tDomain               = \"domain\"\n\tWorkflowID           = \"workflowID\"\n\tRunID                = \"runID\"\n\tIdentity             = \"identity\"\n\tIssueType            = \"issue_type\"\n\tDiagnosticsWfID      = \"diagnostics_workflowID\"\n\tDiagnosticsWfRunID   = \"diagnostics_workflow_runID\"\n\tSatisfactionFeedback = \"satisfaction_feedback\"\n\tEnvironment          = \"environment\"\n\tDiagnosticsStartTime = \"diagnostics_start_time\"\n\tDiagnosticsEndTime   = \"diagnostics_end_time\"\n)\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/failure/failure.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage failure\n\nimport (\n\t\"context\"\n\t\"strings\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\n// Failure is an invariant that will be used to identify the different failures in the workflow execution history\ntype Failure invariant.Invariant\n\ntype failure struct{}\n\nfunc NewInvariant() Failure {\n\treturn &failure{}\n}\n\nfunc (f *failure) Check(ctx context.Context, params invariant.InvariantCheckInput) ([]invariant.InvariantCheckResult, error) {\n\tresult := make([]invariant.InvariantCheckResult, 0)\n\tevents := params.WorkflowExecutionHistory.GetHistory().GetEvents()\n\tissueID := 0\n\tfor _, event := range events {\n\t\tif event.GetWorkflowExecutionFailedEventAttributes() != nil && event.WorkflowExecutionFailedEventAttributes.Reason != nil {\n\t\t\tattr := event.WorkflowExecutionFailedEventAttributes\n\t\t\treason := attr.Reason\n\t\t\tidentity := fetchIdentity(attr, events)\n\t\t\tif *reason == common.FailureReasonDecisionBlobSizeExceedsLimit {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: DecisionCausedFailure.String(),\n\t\t\t\t\tReason:        DecisionBlobSizeLimit.String(),\n\t\t\t\t\tMetadata:      invariant.MarshalData(FailureIssuesMetadata{Identity: identity}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t} else {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: WorkflowFailed.String(),\n\t\t\t\t\tReason:        ErrorTypeFromReason(*reason).String(),\n\t\t\t\t\tMetadata:      invariant.MarshalData(FailureIssuesMetadata{Identity: identity}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t}\n\n\t\t}\n\t\tif event.GetActivityTaskFailedEventAttributes() != nil && event.ActivityTaskFailedEventAttributes.Reason != nil {\n\t\t\tattr := event.ActivityTaskFailedEventAttributes\n\t\t\treason := attr.Reason\n\t\t\tscheduled := fetchScheduledEvent(attr, events)\n\t\t\tif *reason == common.FailureReasonHeartbeatExceedsLimit {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        HeartBeatBlobSizeLimit.String(),\n\t\t\t\t\tMetadata: invariant.MarshalData(FailureIssuesMetadata{\n\t\t\t\t\t\tIdentity:            attr.Identity,\n\t\t\t\t\t\tActivityType:        scheduled.ActivityType.GetName(),\n\t\t\t\t\t\tActivityScheduledID: attr.ScheduledEventID,\n\t\t\t\t\t\tActivityStartedID:   attr.StartedEventID,\n\t\t\t\t\t}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t} else if *reason == common.FailureReasonCompleteResultExceedsLimit {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        ActivityOutputBlobSizeLimit.String(),\n\t\t\t\t\tMetadata: invariant.MarshalData(FailureIssuesMetadata{\n\t\t\t\t\t\tIdentity:            attr.Identity,\n\t\t\t\t\t\tActivityType:        scheduled.ActivityType.GetName(),\n\t\t\t\t\t\tActivityScheduledID: attr.ScheduledEventID,\n\t\t\t\t\t\tActivityStartedID:   attr.StartedEventID,\n\t\t\t\t\t}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t} else {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        ErrorTypeFromReason(*reason).String(),\n\t\t\t\t\tMetadata: invariant.MarshalData(FailureIssuesMetadata{\n\t\t\t\t\t\tIdentity:            attr.Identity,\n\t\t\t\t\t\tActivityType:        scheduled.ActivityType.GetName(),\n\t\t\t\t\t\tActivityScheduledID: attr.ScheduledEventID,\n\t\t\t\t\t\tActivityStartedID:   attr.StartedEventID,\n\t\t\t\t\t}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t}\n\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc ErrorTypeFromReason(reason string) ErrorType {\n\tif strings.Contains(reason, \"Generic\") {\n\t\treturn GenericError\n\t}\n\tif strings.Contains(reason, \"Panic\") {\n\t\treturn PanicError\n\t}\n\tif strings.Contains(reason, \"Timeout\") {\n\t\treturn TimeoutError\n\t}\n\treturn CustomError\n}\n\nfunc fetchScheduledEvent(attr *types.ActivityTaskFailedEventAttributes, events []*types.HistoryEvent) *types.ActivityTaskScheduledEventAttributes {\n\tfor _, event := range events {\n\t\tif event.ID == attr.GetScheduledEventID() {\n\t\t\treturn event.GetActivityTaskScheduledEventAttributes()\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc fetchIdentity(attr *types.WorkflowExecutionFailedEventAttributes, events []*types.HistoryEvent) string {\n\tfor _, event := range events {\n\t\tif event.ID == attr.DecisionTaskCompletedEventID {\n\t\t\treturn event.GetDecisionTaskCompletedEventAttributes().Identity\n\t\t}\n\t}\n\treturn \"\"\n}\n\nfunc (f *failure) RootCause(ctx context.Context, params invariant.InvariantRootCauseInput) ([]invariant.InvariantRootCauseResult, error) {\n\tresult := make([]invariant.InvariantRootCauseResult, 0)\n\tfor _, issue := range params.Issues {\n\t\tswitch issue.Reason {\n\t\tcase GenericError.String():\n\t\t\tresult = append(result, invariant.InvariantRootCauseResult{\n\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSideIssue,\n\t\t\t\tMetadata:  issue.Metadata,\n\t\t\t})\n\t\tcase PanicError.String():\n\t\t\tresult = append(result, invariant.InvariantRootCauseResult{\n\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSidePanic,\n\t\t\t\tMetadata:  issue.Metadata,\n\t\t\t})\n\t\tcase CustomError.String():\n\t\t\tresult = append(result, invariant.InvariantRootCauseResult{\n\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSideCustomError,\n\t\t\t\tMetadata:  issue.Metadata,\n\t\t\t})\n\t\t}\n\t}\n\treturn result, nil\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/failure/failure_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage failure\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\nconst (\n\ttestDomain = \"test-domain\"\n)\n\nfunc Test__Check(t *testing.T) {\n\tmetadata := FailureIssuesMetadata{\n\t\tIdentity: \"localhost\",\n\t}\n\tmetadataInBytes, err := json.Marshal(metadata)\n\trequire.NoError(t, err)\n\tactMetadata := FailureIssuesMetadata{\n\t\tIdentity:            \"localhost\",\n\t\tActivityType:        \"test-activity\",\n\t\tActivityScheduledID: 1,\n\t\tActivityStartedID:   2,\n\t}\n\tactMetadataInBytes, err := json.Marshal(actMetadata)\n\trequire.NoError(t, err)\n\ttestCases := []struct {\n\t\tname           string\n\t\ttestData       *types.GetWorkflowExecutionHistoryResponse\n\t\texpectedResult []invariant.InvariantCheckResult\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname:     \"workflow execution timeout\",\n\t\t\ttestData: failedWfHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        GenericError.String(),\n\t\t\t\t\tMetadata:      actMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       1,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        PanicError.String(),\n\t\t\t\t\tMetadata:      actMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       2,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        CustomError.String(),\n\t\t\t\t\tMetadata:      actMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       3,\n\t\t\t\t\tInvariantType: WorkflowFailed.String(),\n\t\t\t\t\tReason:        TimeoutError.String(),\n\t\t\t\t\tMetadata:      metadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"blob size limit exceeded\",\n\t\t\ttestData: blobSizeLimitExceededHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        ActivityOutputBlobSizeLimit.String(),\n\t\t\t\t\tMetadata:      actMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       1,\n\t\t\t\t\tInvariantType: DecisionCausedFailure.String(),\n\t\t\t\t\tReason:        DecisionBlobSizeLimit.String(),\n\t\t\t\t\tMetadata:      metadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tinv := NewInvariant()\n\t\tresult, err := inv.Check(context.Background(), invariant.InvariantCheckInput{\n\t\t\tWorkflowExecutionHistory: tc.testData,\n\t\t\tDomain:                   testDomain,\n\t\t})\n\t\trequire.Equal(t, tc.err, err)\n\t\trequire.Equal(t, len(tc.expectedResult), len(result))\n\t\trequire.ElementsMatch(t, tc.expectedResult, result)\n\t}\n}\n\nfunc failedWfHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityID:   \"101\",\n\t\t\t\t\t\tActivityType: &types.ActivityType{Name: \"test-activity\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 2,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tIdentity: \"localhost\",\n\t\t\t\t\t\tAttempt:  0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tReason:           common.StringPtr(\"cadenceInternal:Generic\"),\n\t\t\t\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tIdentity:         \"localhost\",\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tReason:           common.StringPtr(\"cadenceInternal:Panic\"),\n\t\t\t\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tIdentity:         \"localhost\",\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tReason:           common.StringPtr(\"custom error\"),\n\t\t\t\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tIdentity:         \"localhost\",\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 10,\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tIdentity: \"localhost\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\t\tReason:                       common.StringPtr(\"cadenceInternal:Timeout START_TO_CLOSE\"),\n\t\t\t\t\t\tDetails:                      []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tDecisionTaskCompletedEventID: 10,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc blobSizeLimitExceededHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tActivityID:   \"101\",\n\t\t\t\t\t\tActivityType: &types.ActivityType{Name: \"test-activity\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 2,\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tIdentity: \"localhost\",\n\t\t\t\t\t\tAttempt:  0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{\n\t\t\t\t\t\tReason:           common.StringPtr(\"COMPLETE_RESULT_EXCEEDS_LIMIT\"),\n\t\t\t\t\t\tDetails:          []byte(\"test-activity-failure\"),\n\t\t\t\t\t\tIdentity:         \"localhost\",\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 10,\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tIdentity: \"localhost\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\t\tReason:                       common.StringPtr(\"DECISION_BLOB_SIZE_EXCEEDS_LIMIT\"),\n\t\t\t\t\t\tDetails:                      []byte(\"test-wf-failure\"),\n\t\t\t\t\t\tDecisionTaskCompletedEventID: 10,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc Test__RootCause(t *testing.T) {\n\tmetadata := FailureIssuesMetadata{\n\t\tIdentity: \"localhost\",\n\t}\n\tmetadataInBytes, err := json.Marshal(metadata)\n\trequire.NoError(t, err)\n\ttestCases := []struct {\n\t\tname           string\n\t\tinput          []invariant.InvariantCheckResult\n\t\texpectedResult []invariant.InvariantRootCauseResult\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname: \"customer side known failure\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        CustomError.String(),\n\t\t\t\t\tMetadata:      metadataInBytes,\n\t\t\t\t}},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{{\n\t\t\t\tIssueID:   0,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSideCustomError,\n\t\t\t\tMetadata:  metadataInBytes,\n\t\t\t}},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"customer side error\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        GenericError.String(),\n\t\t\t\t\tMetadata:      metadataInBytes,\n\t\t\t\t}},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{{\n\t\t\t\tIssueID:   0,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSideIssue,\n\t\t\t\tMetadata:  metadataInBytes,\n\t\t\t}},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"customer side panic\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: ActivityFailed.String(),\n\t\t\t\t\tReason:        PanicError.String(),\n\t\t\t\t\tMetadata:      metadataInBytes,\n\t\t\t\t}},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{{\n\t\t\t\tIssueID:   0,\n\t\t\t\tRootCause: invariant.RootCauseTypeServiceSidePanic,\n\t\t\t\tMetadata:  metadataInBytes,\n\t\t\t}},\n\t\t\terr: nil,\n\t\t},\n\t}\n\tinv := NewInvariant()\n\tfor _, tc := range testCases {\n\t\tresult, err := inv.RootCause(context.Background(), invariant.InvariantRootCauseInput{\n\t\t\tIssues: tc.input,\n\t\t})\n\t\trequire.Equal(t, tc.err, err)\n\t\trequire.Equal(t, len(tc.expectedResult), len(result))\n\t\trequire.ElementsMatch(t, tc.expectedResult, result)\n\t}\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/failure/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage failure\n\ntype ErrorType string\n\nconst (\n\tCustomError                 ErrorType = \"The failure is caused by a specific custom error returned from the service code\"\n\tGenericError                ErrorType = \"The failure is because of an error returned from the service code\"\n\tPanicError                  ErrorType = \"The failure is caused by a panic in the service code\"\n\tTimeoutError                ErrorType = \"The failure is caused by a timeout during the execution\"\n\tHeartBeatBlobSizeLimit      ErrorType = \"Heartbeat details has exceeded the blob size limit\"\n\tActivityOutputBlobSizeLimit ErrorType = \"Activity output has exceeded the blob size limit\"\n\tDecisionBlobSizeLimit       ErrorType = \"Decision result caused to exceed blob size limit\"\n)\n\nfunc (e ErrorType) String() string {\n\treturn string(e)\n}\n\ntype FailureType string\n\nconst (\n\tActivityFailed        FailureType = \"Activity Failed\"\n\tWorkflowFailed        FailureType = \"Workflow Failed\"\n\tDecisionCausedFailure FailureType = \"Decision caused failure\"\n)\n\nfunc (f FailureType) String() string {\n\treturn string(f)\n}\n\ntype FailureIssuesMetadata struct {\n\tIdentity            string\n\tActivityType        string\n\tActivityScheduledID int64\n\tActivityStartedID   int64\n}\n\n// BlobSizeMetadata includes the details of blob size limits\ntype BlobSizeMetadata struct {\n\tBlobSizeWarnLimit  int32\n\tBlobSizeErrorLimit int32\n}\n\ntype FailureRootcauseMetadata struct {\n\tBlobSizeMetadata *BlobSizeMetadata\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage invariant\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// InvariantCheckResult is the result from the invariant check\ntype InvariantCheckResult struct {\n\tIssueID       int\n\tInvariantType string\n\tReason        string\n\tMetadata      []byte\n}\n\n// InvariantRootCauseResult is the root cause for the issues identified in the invariant check\ntype InvariantRootCauseResult struct {\n\tIssueID   int\n\tRootCause RootCause\n\tMetadata  []byte\n}\n\ntype InvariantCheckInput struct {\n\tWorkflowExecutionHistory *types.GetWorkflowExecutionHistoryResponse\n\tDomain                   string\n}\n\ntype InvariantRootCauseInput struct {\n\tDomain string\n\tIssues []InvariantCheckResult\n}\ntype RootCause string\n\nconst (\n\tRootCauseTypeMissingPollers                        RootCause = \"There are no pollers for the tasklist\"\n\tRootCauseTypePollersStatus                         RootCause = \"There are pollers for the tasklist. Check backlog status\"\n\tRootCauseTypeNoHeartBeatTimeoutNoRetryPolicy       RootCause = \"Heartbeat timeout and retry policy are not configured\"\n\tRootCauseTypeHeartBeatingNotEnabledWithRetryPolicy RootCause = \"Heartbeat timeout not enabled for activity but there is a retry policy configured\"\n\tRootCauseTypeHeartBeatingEnabledWithoutRetryPolicy RootCause = \"Heartbeat timeout enabled for activity but there is no retry policy configured\"\n\tRootCauseTypeHeartBeatingEnabledMissingHeartbeat   RootCause = \"Heartbeat timeout enabled for activity but timed out due to missing heartbeat\"\n\tRootCauseTypeServiceSideIssue                      RootCause = \"There is an issue in the worker service that is causing a failure. Check identity for service logs\"\n\tRootCauseTypeServiceSidePanic                      RootCause = \"There is a panic in the activity/workflow that is causing a failure\"\n\tRootCauseTypeServiceSideCustomError                RootCause = \"Customised error returned by the activity/workflow\"\n\tRootCauseTypeBlobSizeLimit                         RootCause = \"Workflow has exceeded the blob size limits configured for the domain\"\n)\n\nfunc (r RootCause) String() string {\n\treturn string(r)\n}\n\n// Invariant represents a condition of a workflow execution.\ntype Invariant interface {\n\tCheck(context.Context, InvariantCheckInput) ([]InvariantCheckResult, error)\n\tRootCause(context.Context, InvariantRootCauseInput) ([]InvariantRootCauseResult, error)\n}\n\nfunc MarshalData(rc any) []byte {\n\tdata, _ := json.Marshal(rc)\n\treturn data\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/retry/retry.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage retry\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\n// Retry is an invariant that will be used to identify the issues regarding retries in the workflow execution history\ntype Retry invariant.Invariant\n\ntype retry struct {\n}\n\nfunc NewInvariant() Retry {\n\treturn &retry{}\n}\n\nfunc (r *retry) Check(ctx context.Context, params invariant.InvariantCheckInput) ([]invariant.InvariantCheckResult, error) {\n\tresult := make([]invariant.InvariantCheckResult, 0)\n\tevents := params.WorkflowExecutionHistory.GetHistory().GetEvents()\n\tissueID := 0\n\tstartedEvent := fetchWfStartedEvent(events)\n\n\tif issue := checkRetryPolicy(startedEvent.RetryPolicy); issue != \"\" {\n\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\tIssueID:       issueID,\n\t\t\tInvariantType: WorkflowRetryIssue.String(),\n\t\t\tReason:        issue.String(),\n\t\t\tMetadata: invariant.MarshalData(RetryMetadata{\n\t\t\t\tEventID:     1,\n\t\t\t\tRetryPolicy: startedEvent.RetryPolicy,\n\t\t\t}),\n\t\t})\n\t\tissueID++\n\t}\n\n\tfor _, event := range events {\n\t\tif event.GetActivityTaskScheduledEventAttributes() != nil {\n\t\t\tattr := event.GetActivityTaskScheduledEventAttributes()\n\t\t\tif issue := checkRetryPolicy(attr.RetryPolicy); issue != \"\" {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: ActivityRetryIssue.String(),\n\t\t\t\t\tReason:        issue.String(),\n\t\t\t\t\tMetadata: invariant.MarshalData(RetryMetadata{\n\t\t\t\t\t\tEventID:     event.ID,\n\t\t\t\t\t\tRetryPolicy: attr.RetryPolicy,\n\t\t\t\t\t}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t}\n\t\t\tif attr.GetStartToCloseTimeoutSeconds() <= attr.GetHeartbeatTimeoutSeconds() {\n\t\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\t\tIssueID:       issueID,\n\t\t\t\t\tInvariantType: ActivityHeartbeatIssue.String(),\n\t\t\t\t\tReason:        HeartBeatTimeoutEqualToStartToCloseTimeout.String(),\n\t\t\t\t\tMetadata: invariant.MarshalData(RetryMetadata{\n\t\t\t\t\t\tEventID: event.ID,\n\t\t\t\t\t}),\n\t\t\t\t})\n\t\t\t\tissueID++\n\t\t\t}\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\nfunc fetchWfStartedEvent(events []*types.HistoryEvent) *types.WorkflowExecutionStartedEventAttributes {\n\tfor _, event := range events {\n\t\tif event.GetWorkflowExecutionStartedEventAttributes() != nil {\n\t\t\treturn event.GetWorkflowExecutionStartedEventAttributes()\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc checkRetryPolicy(policy *types.RetryPolicy) IssueType {\n\tif policy == nil {\n\t\treturn \"\"\n\t}\n\tif policy.GetExpirationIntervalInSeconds() == 0 && policy.GetMaximumAttempts() == 1 {\n\t\treturn RetryPolicyValidationMaxAttempts\n\t}\n\tif policy.GetMaximumAttempts() == 0 && policy.GetExpirationIntervalInSeconds() < policy.GetInitialIntervalInSeconds() {\n\t\treturn RetryPolicyValidationExpInterval\n\t}\n\treturn \"\"\n}\n\nfunc (r *retry) RootCause(ctx context.Context, params invariant.InvariantRootCauseInput) ([]invariant.InvariantRootCauseResult, error) {\n\t// Not implemented since this invariant does not have any root cause.\n\t// Issue identified in Check() are the root cause.\n\tresult := make([]invariant.InvariantRootCauseResult, 0)\n\treturn result, nil\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/retry/retry_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage retry\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\nfunc Test__Check(t *testing.T) {\n\tinvalidAttemptsMetadata := RetryMetadata{\n\t\tEventID: 5,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds: 1,\n\t\t\tMaximumAttempts:          1,\n\t\t},\n\t}\n\tinvalidAttemptsMetadataInBytes, err := json.Marshal(invalidAttemptsMetadata)\n\trequire.NoError(t, err)\n\tinvalidExpIntervalMetadata := RetryMetadata{\n\t\tEventID: 1,\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds:    10,\n\t\t\tExpirationIntervalInSeconds: 5,\n\t\t},\n\t}\n\tinvalidExpIntervalMetadataInBytes, err := json.Marshal(invalidExpIntervalMetadata)\n\trequire.NoError(t, err)\n\tinvalidHBMetadata := RetryMetadata{\n\t\tEventID: 6,\n\t}\n\tinvalidHBMetadataInBytes, err := json.Marshal(invalidHBMetadata)\n\trequire.NoError(t, err)\n\ttestCases := []struct {\n\t\tname           string\n\t\ttestData       *types.GetWorkflowExecutionHistoryResponse\n\t\texpectedResult []invariant.InvariantCheckResult\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname:     \"invalid retry policy\",\n\t\t\ttestData: invalidRetryPolicyWfHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       1,\n\t\t\t\t\tInvariantType: ActivityRetryIssue.String(),\n\t\t\t\t\tReason:        RetryPolicyValidationMaxAttempts.String(),\n\t\t\t\t\tMetadata:      invalidAttemptsMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: WorkflowRetryIssue.String(),\n\t\t\t\t\tReason:        RetryPolicyValidationExpInterval.String(),\n\t\t\t\t\tMetadata:      invalidExpIntervalMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       2,\n\t\t\t\t\tInvariantType: ActivityHeartbeatIssue.String(),\n\t\t\t\t\tReason:        HeartBeatTimeoutEqualToStartToCloseTimeout.String(),\n\t\t\t\t\tMetadata:      invalidHBMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tinv := NewInvariant()\n\t\tresult, err := inv.Check(context.Background(), invariant.InvariantCheckInput{\n\t\t\tWorkflowExecutionHistory: tc.testData,\n\t\t})\n\t\trequire.Equal(t, tc.err, err)\n\t\trequire.Equal(t, len(tc.expectedResult), len(result))\n\t\trequire.ElementsMatch(t, tc.expectedResult, result)\n\t}\n}\n\nfunc retriedWfHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\t\t\t\tMaximumAttempts:          2,\n\t\t\t\t\t\t},\n\t\t\t\t\t\tAttempt: 0,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{\n\t\t\t\t\t\tFailureReason:                common.StringPtr(\"cadenceInternal:Timeout START_TO_CLOSE\"),\n\t\t\t\t\t\tDecisionTaskCompletedEventID: 10,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc invalidRetryPolicyWfHistory() *types.GetWorkflowExecutionHistoryResponse {\n\ttimeout := int32(5)\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds:    10,\n\t\t\t\t\t\t\tExpirationIntervalInSeconds: 5,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 5,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:    common.Int32Ptr(timeout),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(timeout * 2),\n\t\t\t\t\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\t\t\t\t\tInitialIntervalInSeconds: 1,\n\t\t\t\t\t\t\tMaximumAttempts:          1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 6,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tHeartbeatTimeoutSeconds:    common.Int32Ptr(timeout),\n\t\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(timeout),\n\t\t\t\t\t\tRetryPolicy:                nil,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/retry/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage retry\n\nimport \"github.com/uber/cadence/common/types\"\n\ntype RetryType string\n\nconst (\n\tWorkflowRetryIssue     RetryType = \"Workflow Retry configured but invalid\"\n\tActivityRetryIssue     RetryType = \"Activity Retry configured but invalid\"\n\tActivityHeartbeatIssue RetryType = \"Activity Heartbeat configured but invalid\"\n)\n\nfunc (r RetryType) String() string {\n\treturn string(r)\n}\n\ntype IssueType string\n\nconst (\n\tRetryPolicyValidationMaxAttempts           IssueType = \"MaximumAttempts set to 1 will not retry since maximum attempts includes the first attempt.\"\n\tRetryPolicyValidationExpInterval           IssueType = \"ExpirationIntervalInSeconds less than  InitialIntervalInSeconds  will not retry.\"\n\tHeartBeatTimeoutEqualToStartToCloseTimeout IssueType = \"Heartbeat timeout being equal or higher than StartToClose timeout will not provide any benefit.\"\n)\n\nfunc (i IssueType) String() string {\n\treturn string(i)\n}\n\ntype RetryMetadata struct {\n\tEventID     int64\n\tRetryPolicy *types.RetryPolicy\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/timeout/timeout.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\ntype Timeout invariant.Invariant\n\ntype timeout struct {\n\tclient workflowserviceclient.Interface\n}\n\ntype Params struct {\n\tClient workflowserviceclient.Interface\n}\n\nfunc NewInvariant(p Params) invariant.Invariant {\n\treturn &timeout{\n\t\tclient: p.Client,\n\t}\n}\n\nfunc (t *timeout) Check(ctx context.Context, params invariant.InvariantCheckInput) ([]invariant.InvariantCheckResult, error) {\n\tresult := make([]invariant.InvariantCheckResult, 0)\n\tevents := params.WorkflowExecutionHistory.GetHistory().GetEvents()\n\tissueID := 0\n\tfor _, event := range events {\n\t\tif event.WorkflowExecutionTimedOutEventAttributes != nil {\n\t\t\ttimeoutLimit := getWorkflowExecutionConfiguredTimeout(events)\n\t\t\tdata := ExecutionTimeoutMetadata{\n\t\t\t\tExecutionTime:     getExecutionTime(1, event.ID, events),\n\t\t\t\tConfiguredTimeout: time.Duration(timeoutLimit) * time.Second,\n\t\t\t\tLastOngoingEvent:  events[len(events)-2],\n\t\t\t\tTasklist:          getWorkflowExecutionTasklist(events),\n\t\t\t}\n\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\tIssueID:       issueID,\n\t\t\t\tInvariantType: TimeoutTypeExecution.String(),\n\t\t\t\tReason:        event.GetWorkflowExecutionTimedOutEventAttributes().GetTimeoutType().String(),\n\t\t\t\tMetadata:      invariant.MarshalData(data),\n\t\t\t})\n\t\t\tissueID++\n\t\t}\n\t\tif event.ActivityTaskTimedOutEventAttributes != nil {\n\t\t\tmetadata, err := getActivityTaskMetadata(event, events)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\tIssueID:       issueID,\n\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\tReason:        event.GetActivityTaskTimedOutEventAttributes().GetTimeoutType().String(),\n\t\t\t\tMetadata:      invariant.MarshalData(metadata),\n\t\t\t})\n\t\t\tissueID++\n\t\t}\n\t\tif event.DecisionTaskTimedOutEventAttributes != nil {\n\t\t\treason, metadata := reasonForDecisionTaskTimeouts(event, events)\n\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\tIssueID:       issueID,\n\t\t\t\tInvariantType: TimeoutTypeDecision.String(),\n\t\t\t\tReason:        reason,\n\t\t\t\tMetadata:      invariant.MarshalData(metadata),\n\t\t\t})\n\t\t\tissueID++\n\t\t}\n\t\tif event.ChildWorkflowExecutionTimedOutEventAttributes != nil {\n\t\t\ttimeoutLimit := getChildWorkflowExecutionConfiguredTimeout(event, events)\n\t\t\tdata := ChildWfTimeoutMetadata{\n\t\t\t\tExecutionTime:     getExecutionTime(event.GetChildWorkflowExecutionTimedOutEventAttributes().StartedEventID, event.ID, events),\n\t\t\t\tConfiguredTimeout: time.Duration(timeoutLimit) * time.Second,\n\t\t\t\tExecution:         event.GetChildWorkflowExecutionTimedOutEventAttributes().WorkflowExecution,\n\t\t\t}\n\t\t\tresult = append(result, invariant.InvariantCheckResult{\n\t\t\t\tIssueID:       issueID,\n\t\t\t\tInvariantType: TimeoutTypeChildWorkflow.String(),\n\t\t\t\tReason:        event.GetChildWorkflowExecutionTimedOutEventAttributes().TimeoutType.String(),\n\t\t\t\tMetadata:      invariant.MarshalData(data),\n\t\t\t})\n\t\t\tissueID++\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc (t *timeout) RootCause(ctx context.Context, params invariant.InvariantRootCauseInput) ([]invariant.InvariantRootCauseResult, error) {\n\tresult := make([]invariant.InvariantRootCauseResult, 0)\n\tfor _, issue := range params.Issues {\n\t\tif issue.InvariantType == TimeoutTypeActivity.String() || issue.InvariantType == TimeoutTypeExecution.String() {\n\t\t\tpollerStatus, err := t.checkTasklist(ctx, issue, params.Domain)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, pollerStatus)\n\t\t}\n\n\t\tif issue.InvariantType == TimeoutTypeActivity.String() {\n\t\t\theartbeatStatus, err := checkHeartbeatStatus(issue)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, heartbeatStatus...)\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc (t *timeout) checkTasklist(ctx context.Context, issue invariant.InvariantCheckResult, domain string) (invariant.InvariantRootCauseResult, error) {\n\tvar taskList *types.TaskList\n\tvar tasklistType *shared.TaskListType\n\tswitch issue.InvariantType {\n\tcase TimeoutTypeExecution.String():\n\t\tvar metadata ExecutionTimeoutMetadata\n\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\tif err != nil {\n\t\t\treturn invariant.InvariantRootCauseResult{}, err\n\t\t}\n\t\ttaskList = metadata.Tasklist\n\t\ttasklistType = shared.TaskListTypeDecision.Ptr()\n\tcase TimeoutTypeActivity.String():\n\t\tvar metadata ActivityTimeoutMetadata\n\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\tif err != nil {\n\t\t\treturn invariant.InvariantRootCauseResult{}, err\n\t\t}\n\t\ttaskList = metadata.Tasklist\n\t\ttasklistType = shared.TaskListTypeActivity.Ptr()\n\t}\n\tif taskList == nil {\n\t\treturn invariant.InvariantRootCauseResult{}, fmt.Errorf(\"tasklist not set\")\n\t}\n\n\tresp, err := t.client.DescribeTaskList(ctx, &shared.DescribeTaskListRequest{\n\t\tDomain: &domain,\n\t\tTaskList: &shared.TaskList{\n\t\t\tName: &taskList.Name,\n\t\t\tKind: taskListKind(taskList.GetKind()),\n\t\t},\n\t\tTaskListType: tasklistType,\n\t})\n\tif err != nil {\n\t\treturn invariant.InvariantRootCauseResult{}, err\n\t}\n\n\ttasklistBacklog := resp.GetTaskListStatus().GetBacklogCountHint()\n\tpolllersMetadataInBytes := invariant.MarshalData(PollersMetadata{\n\t\tTaskListName:    taskList.Name,\n\t\tTaskListBacklog: tasklistBacklog,\n\t})\n\tif len(resp.GetPollers()) == 0 {\n\t\treturn invariant.InvariantRootCauseResult{\n\t\t\tIssueID:   issue.IssueID,\n\t\t\tRootCause: invariant.RootCauseTypeMissingPollers,\n\t\t\tMetadata:  polllersMetadataInBytes,\n\t\t}, nil\n\t}\n\treturn invariant.InvariantRootCauseResult{\n\t\tIssueID:   issue.IssueID,\n\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\tMetadata:  polllersMetadataInBytes,\n\t}, nil\n\n}\n\nfunc taskListKind(kind types.TaskListKind) *shared.TaskListKind {\n\tif kind.String() == shared.TaskListKindNormal.String() {\n\t\treturn shared.TaskListKindNormal.Ptr()\n\t}\n\n\treturn shared.TaskListKindSticky.Ptr()\n}\n\nfunc checkHeartbeatStatus(issue invariant.InvariantCheckResult) ([]invariant.InvariantRootCauseResult, error) {\n\tvar metadata ActivityTimeoutMetadata\n\terr := json.Unmarshal(issue.Metadata, &metadata)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\theartbeatingMetadataInBytes := invariant.MarshalData(HeartbeatingMetadata{TimeElapsed: metadata.TimeElapsed, RetryPolicy: metadata.RetryPolicy})\n\n\tif metadata.HeartBeatTimeout == 0 && activityStarted(metadata) {\n\t\tif metadata.RetryPolicy != nil {\n\t\t\treturn []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeHeartBeatingNotEnabledWithRetryPolicy,\n\t\t\t\t\tMetadata:  heartbeatingMetadataInBytes,\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\t\treturn []invariant.InvariantRootCauseResult{\n\t\t\t{\n\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\tRootCause: invariant.RootCauseTypeNoHeartBeatTimeoutNoRetryPolicy,\n\t\t\t\tMetadata:  heartbeatingMetadataInBytes,\n\t\t\t},\n\t\t}, nil\n\t}\n\n\tif metadata.HeartBeatTimeout > 0 && metadata.TimeoutType.String() == types.TimeoutTypeHeartbeat.String() {\n\t\tif metadata.RetryPolicy == nil {\n\t\t\treturn []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeHeartBeatingEnabledWithoutRetryPolicy,\n\t\t\t\t\tMetadata:  heartbeatingMetadataInBytes,\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\t\treturn []invariant.InvariantRootCauseResult{\n\t\t\t{\n\t\t\t\tIssueID:   issue.IssueID,\n\t\t\t\tRootCause: invariant.RootCauseTypeHeartBeatingEnabledMissingHeartbeat,\n\t\t\t\tMetadata:  heartbeatingMetadataInBytes,\n\t\t\t},\n\t\t}, nil\n\t}\n\n\treturn nil, nil\n}\n\nfunc activityStarted(metadata ActivityTimeoutMetadata) bool {\n\treturn metadata.TimeoutType.String() != types.TimeoutTypeScheduleToStart.String()\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/timeout/timeout_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/golang/mock/gomock\"\n\t\"github.com/stretchr/testify/require\"\n\tpublicservicetest \"go.uber.org/cadence/.gen/go/cadence/workflowservicetest\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\nconst (\n\tworkflowTimeoutSecond = int32(110)\n\ttaskTimeoutSecond     = int32(50)\n\ttestTimeStamp         = int64(2547596872371000000)\n\ttimeUnit              = time.Second\n\ttestTasklist          = \"test-tasklist\"\n\ttestDomain            = \"test-domain\"\n\ttestTaskListBacklog   = int64(10)\n)\n\nfunc Test__Check(t *testing.T) {\n\tdecisionTimeoutMetadata := DecisionTimeoutMetadata{ConfiguredTimeout: 50 * time.Second}\n\tdecisionTimeoutMetadataInBytes, err := json.Marshal(decisionTimeoutMetadata)\n\trequire.NoError(t, err)\n\ttestCases := []struct {\n\t\tname           string\n\t\ttestData       *types.GetWorkflowExecutionHistoryResponse\n\t\texpectedResult []invariant.InvariantCheckResult\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname:     \"workflow execution timeout\",\n\t\t\ttestData: wfTimeoutHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeExecution.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      wfTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"child workflow execution timeout\",\n\t\t\ttestData: childWfTimeoutHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeChildWorkflow.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      childWfTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"activity timeout\",\n\t\t\ttestData: activityTimeoutHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"SCHEDULE_TO_START\",\n\t\t\t\t\tMetadata:      activityScheduleToStartTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       1,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"HEARTBEAT\",\n\t\t\t\t\tMetadata:      activityHeartBeatTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname:     \"decision timeout\",\n\t\t\ttestData: decisionTimeoutHistory(),\n\t\t\texpectedResult: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeDecision.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      decisionTimeoutMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:       1,\n\t\t\t\t\tInvariantType: TimeoutTypeDecision.String(),\n\t\t\t\t\tReason:        \"workflow reset - New run ID: new run ID\",\n\t\t\t\t\tMetadata:      decisionTimeoutMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\tctrl := gomock.NewController(t)\n\tmockClient := publicservicetest.NewMockClient(ctrl)\n\tfor _, tc := range testCases {\n\t\tinv := NewInvariant(Params{\n\t\t\tClient: mockClient,\n\t\t})\n\t\tresult, err := inv.Check(context.Background(), invariant.InvariantCheckInput{\n\t\t\tWorkflowExecutionHistory: tc.testData,\n\t\t\tDomain:                   testDomain,\n\t\t})\n\t\trequire.Equal(t, tc.err, err)\n\t\trequire.Equal(t, len(tc.expectedResult), len(result))\n\t\tfor i := range result {\n\t\t\trequire.Equal(t, tc.expectedResult[i], result[i])\n\t\t}\n\n\t}\n}\n\nfunc wfTimeoutHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\tName: testTasklist,\n\t\t\t\t\t\t\tKind: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:                                       2,\n\t\t\t\t\tTimestamp:                                common.Int64Ptr(testTimeStamp + int64(workflowTimeoutSecond)*timeUnit.Nanoseconds()),\n\t\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{TimeoutType: types.TimeoutTypeStartToClose.Ptr()},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc childWfTimeoutHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 1,\n\t\t\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: &types.StartChildWorkflowExecutionInitiatedEventAttributes{\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        3,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp + int64(workflowTimeoutSecond)*timeUnit.Nanoseconds()),\n\t\t\t\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\t\tInitiatedEventID: 1,\n\t\t\t\t\t\tStartedEventID:   2,\n\t\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"123\",\n\t\t\t\t\t\t\tRunID:      \"abc\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc activityTimeoutHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tScheduleToStartTimeoutSeconds: common.Int32Ptr(taskTimeoutSecond),\n\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\tName: testTasklist,\n\t\t\t\t\t\t\tKind: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp + int64(taskTimeoutSecond)*timeUnit.Nanoseconds()),\n\t\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t\tTimeoutType:      types.TimeoutTypeScheduleToStart.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 3,\n\t\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{\n\t\t\t\t\t\tHeartbeatTimeoutSeconds: common.Int32Ptr(taskTimeoutSecond),\n\t\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\t\tName: testTasklist,\n\t\t\t\t\t\t\tKind: nil,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        4,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 21,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        5,\n\t\t\t\t\tTimestamp: common.Int64Ptr(testTimeStamp + int64(taskTimeoutSecond)*timeUnit.Nanoseconds()),\n\t\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 3,\n\t\t\t\t\t\tStartedEventID:   4,\n\t\t\t\t\t\tTimeoutType:      types.TimeoutTypeHeartbeat.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc decisionTimeoutHistory() *types.GetWorkflowExecutionHistoryResponse {\n\treturn &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 13,\n\t\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(taskTimeoutSecond),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 13,\n\t\t\t\t\t\tStartedEventID:   14,\n\t\t\t\t\t\tCause:            types.DecisionTaskTimedOutCauseTimeout.Ptr(),\n\t\t\t\t\t\tTimeoutType:      types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID: 23,\n\t\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{\n\t\t\t\t\t\tStartToCloseTimeoutSeconds: common.Int32Ptr(taskTimeoutSecond),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 23,\n\t\t\t\t\t\tCause:            types.DecisionTaskTimedOutCauseReset.Ptr(),\n\t\t\t\t\t\tReason:           \"workflow reset\",\n\t\t\t\t\t\tNewRunID:         \"new run ID\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc wfTimeoutDataInBytes(t *testing.T) []byte {\n\tdata := ExecutionTimeoutMetadata{\n\t\tExecutionTime:     110 * time.Second,\n\t\tConfiguredTimeout: 110 * time.Second,\n\t\tLastOngoingEvent: &types.HistoryEvent{\n\t\t\tID:        1,\n\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\tName: testTasklist,\n\t\t\t\t\tKind: nil,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tTasklist: &types.TaskList{\n\t\t\tName: testTasklist,\n\t\t\tKind: nil,\n\t\t},\n\t}\n\tdataInBytes, err := json.Marshal(data)\n\trequire.NoError(t, err)\n\treturn dataInBytes\n}\n\nfunc activityScheduleToStartTimeoutData() ActivityTimeoutMetadata {\n\treturn ActivityTimeoutMetadata{\n\t\tTimeoutType:       types.TimeoutTypeScheduleToStart.Ptr(),\n\t\tConfiguredTimeout: 50 * time.Second,\n\t\tTimeElapsed:       50 * time.Second,\n\t\tRetryPolicy:       nil,\n\t\tHeartBeatTimeout:  0,\n\t\tTasklist: &types.TaskList{\n\t\t\tName: testTasklist,\n\t\t\tKind: nil,\n\t\t},\n\t}\n}\n\nfunc activityStartToCloseTimeoutData() ActivityTimeoutMetadata {\n\treturn ActivityTimeoutMetadata{\n\t\tTimeoutType:       types.TimeoutTypeStartToClose.Ptr(),\n\t\tConfiguredTimeout: 50 * time.Second,\n\t\tTimeElapsed:       50 * time.Second,\n\t\tRetryPolicy:       nil,\n\t\tHeartBeatTimeout:  0,\n\t\tTasklist: &types.TaskList{\n\t\t\tName: testTasklist,\n\t\t\tKind: nil,\n\t\t},\n\t}\n}\n\nfunc activityScheduleToStartTimeoutDataInBytes(t *testing.T) []byte {\n\tdata := activityScheduleToStartTimeoutData()\n\tdataInBytes, err := json.Marshal(data)\n\trequire.NoError(t, err)\n\treturn dataInBytes\n}\n\nfunc activityStartToCloseTimeoutDataInBytes(t *testing.T) []byte {\n\tdata := activityStartToCloseTimeoutData()\n\tdataInBytes, err := json.Marshal(data)\n\trequire.NoError(t, err)\n\treturn dataInBytes\n}\n\nfunc activityHeartBeatTimeoutDataInBytes(t *testing.T) []byte {\n\tactTimeoutData := activityStartToCloseTimeoutData()\n\tactTimeoutData.TimeoutType = types.TimeoutTypeHeartbeat.Ptr()\n\tactTimeoutData.HeartBeatTimeout = 50 * time.Second\n\tactHeartBeatTimeoutDataInBytes, err := json.Marshal(actTimeoutData)\n\trequire.NoError(t, err)\n\treturn actHeartBeatTimeoutDataInBytes\n}\n\nfunc activityHeartBeatTimeoutDataWithRetryPolicyInBytes(t *testing.T) []byte {\n\tactTimeoutData := activityStartToCloseTimeoutData()\n\tactTimeoutData.TimeoutType = types.TimeoutTypeHeartbeat.Ptr()\n\tactTimeoutData.HeartBeatTimeout = 50 * time.Second\n\tactTimeoutData.RetryPolicy = &types.RetryPolicy{\n\t\tMaximumAttempts: 3,\n\t}\n\tactHeartBeatTimeoutDataInBytes, err := json.Marshal(actTimeoutData)\n\trequire.NoError(t, err)\n\treturn actHeartBeatTimeoutDataInBytes\n}\n\nfunc childWfTimeoutDataInBytes(t *testing.T) []byte {\n\tdata := ChildWfTimeoutMetadata{\n\t\tExecutionTime:     110 * time.Second,\n\t\tConfiguredTimeout: 110 * time.Second,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"123\",\n\t\t\tRunID:      \"abc\",\n\t\t},\n\t}\n\tdataInBytes, err := json.Marshal(data)\n\trequire.NoError(t, err)\n\treturn dataInBytes\n}\n\nfunc Test__RootCause(t *testing.T) {\n\tactStartToCloseTimeoutData := activityStartToCloseTimeoutData()\n\tpollersMetadataInBytes, err := json.Marshal(PollersMetadata{TaskListName: testTasklist, TaskListBacklog: testTaskListBacklog})\n\trequire.NoError(t, err)\n\theartBeatingMetadataInBytes, err := json.Marshal(HeartbeatingMetadata{TimeElapsed: actStartToCloseTimeoutData.TimeElapsed})\n\trequire.NoError(t, err)\n\theartBeatingMetadataWithRetryPolicyInBytes, err := json.Marshal(HeartbeatingMetadata{TimeElapsed: actStartToCloseTimeoutData.TimeElapsed, RetryPolicy: &types.RetryPolicy{MaximumAttempts: 3}})\n\trequire.NoError(t, err)\n\ttestCases := []struct {\n\t\tname           string\n\t\tinput          []invariant.InvariantCheckResult\n\t\tclientExpects  func(client *publicservicetest.MockClient)\n\t\texpectedResult []invariant.InvariantRootCauseResult\n\t\terr            error\n\t}{\n\t\t{\n\t\t\tname: \"workflow execution timeout without pollers\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeExecution.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      wfTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: nil,\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeMissingPollers,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"workflow execution timeout with pollers\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeExecution.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      wfTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*shared.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: common.StringPtr(\"dca24-xy\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"activity timeout and heart beating not enabled\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      activityStartToCloseTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*shared.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: common.StringPtr(\"dca24-xy\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeNoHeartBeatTimeoutNoRetryPolicy,\n\t\t\t\t\tMetadata:  heartBeatingMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"activity schedule to start timeout\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"SCHEDULE_TO_START\",\n\t\t\t\t\tMetadata:      activityScheduleToStartTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*shared.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: common.StringPtr(\"dca24-xy\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"activity timeout and heart beating enabled with retry policy\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      activityHeartBeatTimeoutDataWithRetryPolicyInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*shared.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: common.StringPtr(\"dca24-xy\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeHeartBeatingEnabledMissingHeartbeat,\n\t\t\t\t\tMetadata:  heartBeatingMetadataWithRetryPolicyInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"activity timeout and heart beating enabled without retry policy\",\n\t\t\tinput: []invariant.InvariantCheckResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:       0,\n\t\t\t\t\tInvariantType: TimeoutTypeActivity.String(),\n\t\t\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\t\t\tMetadata:      activityHeartBeatTimeoutDataInBytes(t),\n\t\t\t\t},\n\t\t\t},\n\t\t\tclientExpects: func(client *publicservicetest.MockClient) {\n\t\t\t\tclient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(&shared.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*shared.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: common.StringPtr(\"dca24-xy\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tTaskListStatus: &shared.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: common.Int64Ptr(testTaskListBacklog),\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedResult: []invariant.InvariantRootCauseResult{\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tIssueID:   0,\n\t\t\t\t\tRootCause: invariant.RootCauseTypeHeartBeatingEnabledWithoutRetryPolicy,\n\t\t\t\t\tMetadata:  heartBeatingMetadataInBytes,\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t}\n\tctrl := gomock.NewController(t)\n\tmockClient := publicservicetest.NewMockClient(ctrl)\n\tinv := NewInvariant(Params{\n\t\tClient: mockClient,\n\t})\n\tfor _, tc := range testCases {\n\t\ttc.clientExpects(mockClient)\n\t\tresult, err := inv.RootCause(context.Background(), invariant.InvariantRootCauseInput{\n\t\t\tDomain: testDomain,\n\t\t\tIssues: tc.input,\n\t\t})\n\t\trequire.Equal(t, tc.err, err)\n\t\trequire.Equal(t, len(tc.expectedResult), len(result))\n\t\tfor i := range result {\n\t\t\trequire.Equal(t, tc.expectedResult[i], result[i])\n\t\t}\n\n\t}\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/timeout/timeout_utils.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc reasonForDecisionTaskTimeouts(event *types.HistoryEvent, allEvents []*types.HistoryEvent) (string, DecisionTimeoutMetadata) {\n\teventScheduledID := event.GetDecisionTaskTimedOutEventAttributes().GetScheduledEventID()\n\tattr := event.GetDecisionTaskTimedOutEventAttributes()\n\tcause := attr.GetCause()\n\tvar reason string\n\tswitch cause {\n\tcase types.DecisionTaskTimedOutCauseTimeout:\n\t\treason = attr.TimeoutType.String()\n\tcase types.DecisionTaskTimedOutCauseReset:\n\t\tnewRunID := attr.GetNewRunID()\n\t\treason = fmt.Sprintf(\"%s - New run ID: %s\", attr.Reason, newRunID)\n\t}\n\treturn reason, DecisionTimeoutMetadata{\n\t\tConfiguredTimeout: time.Duration(getDecisionTaskConfiguredTimeout(eventScheduledID, allEvents)) * time.Second,\n\t}\n}\n\nfunc getWorkflowExecutionConfiguredTimeout(events []*types.HistoryEvent) int32 {\n\tfor _, event := range events {\n\t\tif event.ID == 1 { // event 1 is workflow execution started event\n\t\t\treturn event.GetWorkflowExecutionStartedEventAttributes().GetExecutionStartToCloseTimeoutSeconds()\n\t\t}\n\t}\n\treturn 0\n}\n\nfunc getWorkflowExecutionTasklist(events []*types.HistoryEvent) *types.TaskList {\n\tfor _, event := range events {\n\t\tif event.ID == 1 { // event 1 is workflow execution started event\n\t\t\treturn event.GetWorkflowExecutionStartedEventAttributes().TaskList\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc getActivityTaskMetadata(e *types.HistoryEvent, events []*types.HistoryEvent) (ActivityTimeoutMetadata, error) {\n\teventScheduledID := e.GetActivityTaskTimedOutEventAttributes().GetScheduledEventID()\n\teventStartedID := e.GetActivityTaskTimedOutEventAttributes().StartedEventID\n\ttimeoutType := e.GetActivityTaskTimedOutEventAttributes().GetTimeoutType()\n\tvar configuredTimeout int32\n\tvar timeElapsed time.Duration\n\tfor _, event := range events {\n\t\tif event.ID == eventScheduledID {\n\t\t\tattr := event.GetActivityTaskScheduledEventAttributes()\n\t\t\tswitch timeoutType {\n\t\t\tcase types.TimeoutTypeHeartbeat:\n\t\t\t\tconfiguredTimeout = attr.GetHeartbeatTimeoutSeconds()\n\t\t\t\ttimeElapsed = getExecutionTime(eventStartedID, e.ID, events)\n\t\t\tcase types.TimeoutTypeScheduleToClose:\n\t\t\t\tconfiguredTimeout = attr.GetScheduleToCloseTimeoutSeconds()\n\t\t\t\ttimeElapsed = getExecutionTime(eventScheduledID, e.ID, events)\n\t\t\tcase types.TimeoutTypeScheduleToStart:\n\t\t\t\tconfiguredTimeout = attr.GetScheduleToStartTimeoutSeconds()\n\t\t\t\ttimeElapsed = getExecutionTime(eventScheduledID, e.ID, events)\n\t\t\tcase types.TimeoutTypeStartToClose:\n\t\t\t\tconfiguredTimeout = attr.GetStartToCloseTimeoutSeconds()\n\t\t\t\ttimeElapsed = getExecutionTime(eventStartedID, e.ID, events)\n\t\t\tdefault:\n\t\t\t\treturn ActivityTimeoutMetadata{}, fmt.Errorf(\"unknown timeout type\")\n\t\t\t}\n\t\t\treturn ActivityTimeoutMetadata{\n\t\t\t\tTimeoutType:       timeoutType.Ptr(),\n\t\t\t\tConfiguredTimeout: time.Duration(configuredTimeout) * time.Second,\n\t\t\t\tTimeElapsed:       timeElapsed,\n\t\t\t\tRetryPolicy:       attr.RetryPolicy,\n\t\t\t\tHeartBeatTimeout:  time.Duration(attr.GetHeartbeatTimeoutSeconds()) * time.Second,\n\t\t\t\tTasklist:          attr.TaskList,\n\t\t\t}, nil\n\t\t}\n\n\t}\n\treturn ActivityTimeoutMetadata{}, fmt.Errorf(\"activity scheduled event not found\")\n}\n\nfunc getDecisionTaskConfiguredTimeout(eventScheduledID int64, events []*types.HistoryEvent) int32 {\n\tfor _, event := range events {\n\t\tif event.ID == eventScheduledID {\n\t\t\treturn event.GetDecisionTaskScheduledEventAttributes().GetStartToCloseTimeoutSeconds()\n\t\t}\n\t}\n\treturn 0\n}\n\nfunc getChildWorkflowExecutionConfiguredTimeout(e *types.HistoryEvent, events []*types.HistoryEvent) int32 {\n\twfInitiatedID := e.GetChildWorkflowExecutionTimedOutEventAttributes().GetInitiatedEventID()\n\tfor _, event := range events {\n\t\tif event.ID == wfInitiatedID {\n\t\t\treturn event.GetStartChildWorkflowExecutionInitiatedEventAttributes().GetExecutionStartToCloseTimeoutSeconds()\n\t\t}\n\t}\n\treturn 0\n}\n\nfunc getExecutionTime(startID, timeoutID int64, events []*types.HistoryEvent) time.Duration {\n\tsort.SliceStable(events, func(i, j int) bool {\n\t\treturn events[i].ID < events[j].ID\n\t})\n\n\tfirstEvent := events[startID-1]\n\tlastEvent := events[timeoutID-1]\n\treturn time.Unix(0, common.Int64Default(lastEvent.Timestamp)).Sub(time.Unix(0, common.Int64Default(firstEvent.Timestamp)))\n}\n"
  },
  {
    "path": "service/worker/diagnostics/invariant/timeout/types.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timeout\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype TimeoutType string\n\nconst (\n\tTimeoutTypeExecution     TimeoutType = \"The Workflow Execution has timed out\"\n\tTimeoutTypeActivity      TimeoutType = \"Activity task has timed out\"\n\tTimeoutTypeDecision      TimeoutType = \"Decision task has timed out\"\n\tTimeoutTypeChildWorkflow TimeoutType = \"Child Workflow Execution has timed out\"\n)\n\nfunc (tt TimeoutType) String() string {\n\treturn string(tt)\n}\n\ntype ExecutionTimeoutMetadata struct {\n\tExecutionTime     time.Duration\n\tConfiguredTimeout time.Duration\n\tTasklist          *types.TaskList\n\tLastOngoingEvent  *types.HistoryEvent\n}\n\ntype ChildWfTimeoutMetadata struct {\n\tExecutionTime     time.Duration\n\tConfiguredTimeout time.Duration\n\tExecution         *types.WorkflowExecution\n}\n\ntype ActivityTimeoutMetadata struct {\n\tTimeoutType       *types.TimeoutType\n\tConfiguredTimeout time.Duration\n\tTimeElapsed       time.Duration\n\tRetryPolicy       *types.RetryPolicy\n\tHeartBeatTimeout  time.Duration\n\tTasklist          *types.TaskList\n}\n\ntype DecisionTimeoutMetadata struct {\n\tConfiguredTimeout time.Duration\n}\n\ntype PollersMetadata struct {\n\tTaskListName    string\n\tTaskListBacklog int64\n}\n\ntype HeartbeatingMetadata struct {\n\tTimeElapsed time.Duration\n\tRetryPolicy *types.RetryPolicy\n}\n\ntype TimeoutIssuesMetadata struct {\n\tExecutionTimeout *ExecutionTimeoutMetadata\n\tActivityTimeout  *ActivityTimeoutMetadata\n\tChildWfTimeout   *ChildWfTimeoutMetadata\n\tDecisionTimeout  *DecisionTimeoutMetadata\n}\n\ntype TimeoutRootcauseMetadata struct {\n\tPollersMetadata      *PollersMetadata\n\tHeartBeatingMetadata *HeartbeatingMetadata\n}\n"
  },
  {
    "path": "service/worker/diagnostics/module.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"context\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n)\n\ntype DiagnosticsWorkflow interface {\n\tStart() error\n\tStop()\n}\n\ntype dw struct {\n\tsvcClient       workflowserviceclient.Interface\n\tclientBean      client.Bean\n\tmetricsClient   metrics.Client\n\tmessagingClient messaging.Client\n\tlogger          log.Logger\n\ttallyScope      tally.Scope\n\tworker          worker.Worker\n\tinvariants      []invariant.Invariant\n\tclusterMetadata cluster.Metadata\n}\n\ntype Params struct {\n\tServiceClient   workflowserviceclient.Interface\n\tClientBean      client.Bean\n\tMetricsClient   metrics.Client\n\tMessagingClient messaging.Client\n\tLogger          log.Logger\n\tTallyScope      tally.Scope\n\tInvariants      []invariant.Invariant\n\tClusterMetadata cluster.Metadata\n}\n\n// New creates a new diagnostics workflow.\nfunc New(params Params) DiagnosticsWorkflow {\n\treturn &dw{\n\t\tsvcClient:       params.ServiceClient,\n\t\tmetricsClient:   params.MetricsClient,\n\t\tmessagingClient: params.MessagingClient,\n\t\ttallyScope:      params.TallyScope,\n\t\tclientBean:      params.ClientBean,\n\t\tlogger:          params.Logger,\n\t\tinvariants:      params.Invariants,\n\t\tclusterMetadata: params.ClusterMetadata,\n\t}\n}\n\n// Start starts the worker\nfunc (w *dw) Start() error {\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:                     w.tallyScope,\n\t\tBackgroundActivityContext:        context.Background(),\n\t\tTracer:                           opentracing.GlobalTracer(),\n\t\tMaxConcurrentActivityTaskPollers: 10,\n\t\tMaxConcurrentDecisionTaskPollers: 10,\n\t}\n\tnewWorker := worker.New(w.svcClient, constants.SystemLocalDomainName, tasklist, workerOpts)\n\tnewWorker.RegisterWorkflowWithOptions(w.DiagnosticsWorkflow, workflow.RegisterOptions{Name: diagnosticsWorkflow})\n\tnewWorker.RegisterWorkflowWithOptions(w.DiagnosticsStarterWorkflow, workflow.RegisterOptions{Name: diagnosticsStarterWorkflow})\n\tnewWorker.RegisterActivityWithOptions(w.identifyIssues, activity.RegisterOptions{Name: identifyIssuesActivity})\n\tnewWorker.RegisterActivityWithOptions(w.rootCauseIssues, activity.RegisterOptions{Name: rootCauseIssuesActivity})\n\tnewWorker.RegisterActivityWithOptions(w.emitUsageLogs, activity.RegisterOptions{Name: emitUsageLogsActivity})\n\tw.worker = newWorker\n\treturn newWorker.Start()\n}\n\nfunc (w *dw) Stop() {\n\tw.worker.Stop()\n}\n"
  },
  {
    "path": "service/worker/diagnostics/module_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/failure\"\n)\n\nfunc Test__Start(t *testing.T) {\n\tdwTest, mockResource := setuptest(t)\n\terr := dwTest.Start()\n\trequire.NoError(t, err)\n\tdwTest.Stop()\n\tmockResource.Finish(t)\n}\n\nfunc setuptest(t *testing.T) (DiagnosticsWorkflow, *resource.Test) {\n\tctrl := gomock.NewController(t)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockResource := resource.NewTest(t, ctrl, metrics.Worker)\n\tsdkClient := mockResource.GetSDKClient()\n\tmockResource.SDKClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.DescribeDomainResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForDecisionTaskResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForActivityTaskResponse{}, nil).AnyTimes()\n\treturn New(Params{\n\t\tServiceClient: sdkClient,\n\t\tClientBean:    mockClientBean,\n\t\tMetricsClient: nil,\n\t\tTallyScope:    tally.TestScope(nil),\n\t\tInvariants:    []invariant.Invariant{failure.NewInvariant()},\n\t}), mockResource\n}\n"
  },
  {
    "path": "service/worker/diagnostics/parent_workflow.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/analytics\"\n)\n\nconst (\n\tdiagnosticsStarterWorkflow = \"diagnostics-starter-workflow\"\n\temitUsageLogsActivity      = \"emitUsageLogs\"\n\tqueryDiagnosticsReport     = \"query-diagnostics-report\"\n\n\tissueTypeTimeouts = \"Timeout\"\n\tissueTypeFailures = \"Failure\"\n\tissueTypeRetry    = \"Retry\"\n)\n\ntype DiagnosticsStarterWorkflowInput struct {\n\tDomain     string\n\tIdentity   string\n\tWorkflowID string\n\tRunID      string\n}\n\ntype DiagnosticsStarterWorkflowResult struct {\n\tDiagnosticsResult    *DiagnosticsWorkflowResult\n\tDiagnosticsCompleted bool\n}\n\nfunc (w *dw) DiagnosticsStarterWorkflow(ctx workflow.Context, params DiagnosticsStarterWorkflowInput) (*DiagnosticsStarterWorkflowResult, error) {\n\tvar diagWfResult DiagnosticsWorkflowResult\n\tworkflowResult := DiagnosticsStarterWorkflowResult{\n\t\tDiagnosticsResult: &diagWfResult,\n\t}\n\terr := workflow.SetQueryHandler(ctx, queryDiagnosticsReport, func() (DiagnosticsStarterWorkflowResult, error) {\n\t\treturn workflowResult, nil\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tfuture := workflow.ExecuteChildWorkflow(ctx, w.DiagnosticsWorkflow, DiagnosticsWorkflowInput{\n\t\tDomain:     params.Domain,\n\t\tWorkflowID: params.WorkflowID,\n\t\tRunID:      params.RunID,\n\t})\n\n\tvar childWfExec workflow.Execution\n\tvar childWfStart, childWfEnd time.Time\n\tif err = future.GetChildWorkflowExecution().Get(ctx, &childWfExec); err != nil {\n\t\treturn nil, fmt.Errorf(\"Workflow Diagnostics start failed: %w\", err)\n\t}\n\tchildWfStart = workflow.Now(ctx)\n\n\terr = future.Get(ctx, &diagWfResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Workflow Diagnostics failed: %w\", err)\n\t}\n\tworkflowResult.DiagnosticsCompleted = true\n\tchildWfEnd = workflow.Now(ctx)\n\n\tinfo := workflow.GetInfo(ctx)\n\tactivityOptions := workflow.ActivityOptions{\n\t\tScheduleToCloseTimeout: time.Second * 10,\n\t\tScheduleToStartTimeout: time.Second * 5,\n\t\tStartToCloseTimeout:    time.Second * 5,\n\t}\n\tactivityCtx := workflow.WithActivityOptions(ctx, activityOptions)\n\terr = workflow.ExecuteActivity(activityCtx, emitUsageLogsActivity, analytics.WfDiagnosticsUsageData{\n\t\tDomain:                params.Domain,\n\t\tWorkflowID:            params.WorkflowID,\n\t\tRunID:                 params.RunID,\n\t\tIdentity:              params.Identity,\n\t\tIssueType:             getIssueType(diagWfResult),\n\t\tEnvironment:           w.clusterMetadata.GetCurrentClusterName(),\n\t\tDiagnosticsWorkflowID: childWfExec.ID,\n\t\tDiagnosticsRunID:      childWfExec.RunID,\n\t\tDiagnosticsStartTime:  childWfStart,\n\t\tDiagnosticsEndTime:    childWfEnd,\n\t}).Get(ctx, nil)\n\tif err != nil {\n\t\tw.logger.Error(\"wf-diagnostics usage logs emission failed\",\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowID(info.WorkflowExecution.ID),\n\t\t\ttag.WorkflowRunID(info.WorkflowExecution.RunID))\n\t}\n\n\treturn &workflowResult, nil\n}\n\nfunc getIssueType(result DiagnosticsWorkflowResult) string {\n\tvar issueType string\n\tif result.Timeouts != nil {\n\t\tissueType = fmt.Sprintf(\"%s-%s\", issueType, issueTypeTimeouts)\n\t}\n\tif result.Failures != nil {\n\t\tissueType = fmt.Sprintf(\"%s-%s\", issueType, issueTypeFailures)\n\t}\n\tif result.Retries != nil {\n\t\tissueType = fmt.Sprintf(\"%s-%s\", issueType, issueTypeRetry)\n\t}\n\treturn issueType\n}\n"
  },
  {
    "path": "service/worker/diagnostics/workflow.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/failure\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/retry\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/timeout\"\n)\n\nconst (\n\tdiagnosticsWorkflow = \"diagnostics-workflow\"\n\ttasklist            = \"diagnostics-wf-tasklist\"\n\n\tidentifyIssuesActivity  = \"identifyIssues\"\n\trootCauseIssuesActivity = \"rootCauseIssues\"\n)\n\ntype DiagnosticsWorkflowInput struct {\n\tDomain     string\n\tWorkflowID string\n\tRunID      string\n}\n\ntype DiagnosticsWorkflowResult struct {\n\tTimeouts *timeoutDiagnostics\n\tFailures *failureDiagnostics\n\tRetries  *retryDiagnostics\n}\n\ntype timeoutDiagnostics struct {\n\tIssues    []*timeoutIssuesResult\n\tRootCause []*timeoutRootCauseResult\n\tRunbook   string\n}\n\ntype timeoutIssuesResult struct {\n\tIssueID       int\n\tInvariantType string\n\tReason        string\n\tMetadata      *timeout.TimeoutIssuesMetadata\n}\n\ntype timeoutRootCauseResult struct {\n\tIssueID       int\n\tRootCauseType string\n\tMetadata      *timeout.TimeoutRootcauseMetadata\n}\n\ntype failureDiagnostics struct {\n\tIssues    []*failureIssuesResult\n\tRootCause []*failureRootCauseResult\n\tRunbook   string\n}\n\ntype failureIssuesResult struct {\n\tIssueID       int\n\tInvariantType string\n\tReason        string\n\tMetadata      *failure.FailureIssuesMetadata\n}\n\ntype failureRootCauseResult struct {\n\tIssueID       int\n\tRootCauseType string\n\tMetadata      *failure.FailureRootcauseMetadata\n}\n\ntype retryDiagnostics struct {\n\tIssues  []*retryIssuesResult\n\tRunbook string\n}\n\ntype retryIssuesResult struct {\n\tIssueID       int\n\tInvariantType string\n\tReason        string\n\tMetadata      retry.RetryMetadata\n}\n\nfunc (w *dw) DiagnosticsWorkflow(ctx workflow.Context, params DiagnosticsWorkflowInput) (*DiagnosticsWorkflowResult, error) {\n\tscope := w.metricsClient.Scope(metrics.DiagnosticsWorkflowScope, metrics.DomainTag(params.Domain))\n\tscope.IncCounter(metrics.DiagnosticsWorkflowStartedCount)\n\tsw := scope.StartTimer(metrics.DiagnosticsWorkflowExecutionLatency)\n\tdefer sw.Stop()\n\n\tvar timeoutsResult *timeoutDiagnostics\n\tvar failureResult *failureDiagnostics\n\tvar retryResult *retryDiagnostics\n\tvar checkResult []invariant.InvariantCheckResult\n\tvar rootCauseResult []invariant.InvariantRootCauseResult\n\n\tactivityOptions := workflow.ActivityOptions{\n\t\tScheduleToCloseTimeout: time.Second * 10,\n\t\tScheduleToStartTimeout: time.Second * 5,\n\t\tStartToCloseTimeout:    time.Second * 5,\n\t}\n\tactivityCtx := workflow.WithActivityOptions(ctx, activityOptions)\n\n\terr := workflow.ExecuteActivity(activityCtx, identifyIssuesActivity, identifyIssuesParams{\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: params.WorkflowID,\n\t\t\tRunID:      params.RunID,\n\t\t},\n\t\tDomain: params.Domain,\n\t}).Get(ctx, &checkResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"IdentifyIssues: %w\", err)\n\t}\n\n\terr = workflow.ExecuteActivity(activityCtx, rootCauseIssuesActivity, rootCauseIssuesParams{\n\t\tDomain: params.Domain,\n\t\tIssues: checkResult,\n\t}).Get(ctx, &rootCauseResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"RootCauseIssues: %w\", err)\n\t}\n\n\ttimeoutIssues, err := retrieveTimeoutIssues(checkResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"RetrieveTimeoutIssues: %w\", err)\n\t}\n\n\tif len(timeoutIssues) > 0 {\n\t\ttimeoutRootCause, err := retrieveTimeoutRootCause(rootCauseResult)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"RetrieveTimeoutRootCause: %w\", err)\n\t\t}\n\t\ttimeoutsResult = &timeoutDiagnostics{\n\t\t\tIssues:    timeoutIssues,\n\t\t\tRootCause: timeoutRootCause,\n\t\t\tRunbook:   linkToTimeoutsRunbook,\n\t\t}\n\t}\n\n\tfailureIssues, err := retrieveFailureIssues(checkResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"RetrieveFailureIssues: %w\", err)\n\t}\n\n\tif len(failureIssues) > 0 {\n\t\tfailureRootCause, err := retrieveFailureRootCause(rootCauseResult)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"RetrieveFailureRootCause: %w\", err)\n\t\t}\n\t\tfailureResult = &failureDiagnostics{\n\t\t\tIssues:    failureIssues,\n\t\t\tRootCause: failureRootCause,\n\t\t\tRunbook:   linkToFailuresRunbook,\n\t\t}\n\t}\n\n\tretryIssues, err := retrieveRetryIssues(checkResult)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"RetrieveRetryIssues: %w\", err)\n\t}\n\n\tif len(retryIssues) > 0 {\n\t\tretryResult = &retryDiagnostics{\n\t\t\tIssues:  retryIssues,\n\t\t\tRunbook: linkToRetriesRunbook,\n\t\t}\n\t}\n\n\tscope.IncCounter(metrics.DiagnosticsWorkflowSuccess)\n\treturn &DiagnosticsWorkflowResult{\n\t\tTimeouts: timeoutsResult,\n\t\tFailures: failureResult,\n\t\tRetries:  retryResult,\n\t}, nil\n}\n\nfunc retrieveTimeoutIssues(issues []invariant.InvariantCheckResult) ([]*timeoutIssuesResult, error) {\n\tresult := make([]*timeoutIssuesResult, 0)\n\tfor _, issue := range issues {\n\t\tswitch issue.InvariantType {\n\t\tcase timeout.TimeoutTypeExecution.String():\n\t\t\tvar metadata timeout.ExecutionTimeoutMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\t\tExecutionTimeout: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\tcase timeout.TimeoutTypeActivity.String():\n\t\t\tvar metadata timeout.ActivityTimeoutMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\t\tActivityTimeout: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\tcase timeout.TimeoutTypeChildWorkflow.String():\n\t\t\tvar metadata timeout.ChildWfTimeoutMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\t\tChildWfTimeout: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\tcase timeout.TimeoutTypeDecision.String():\n\t\t\tvar metadata timeout.DecisionTimeoutMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\t\tDecisionTimeout: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc retrieveTimeoutRootCause(rootCause []invariant.InvariantRootCauseResult) ([]*timeoutRootCauseResult, error) {\n\tresult := make([]*timeoutRootCauseResult, 0)\n\tfor _, rc := range rootCause {\n\t\tif rootCausePollersRelated(rc.RootCause) {\n\t\t\tvar metadata timeout.PollersMetadata\n\t\t\terr := json.Unmarshal(rc.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutRootCauseResult{\n\t\t\t\tIssueID:       rc.IssueID,\n\t\t\t\tRootCauseType: rc.RootCause.String(),\n\t\t\t\tMetadata: &timeout.TimeoutRootcauseMetadata{\n\t\t\t\t\tPollersMetadata: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\t} else if rootCauseHeartBeatRelated(rc.RootCause) {\n\t\t\tvar metadata timeout.HeartbeatingMetadata\n\t\t\terr := json.Unmarshal(rc.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &timeoutRootCauseResult{\n\t\t\t\tIssueID:       rc.IssueID,\n\t\t\t\tRootCauseType: rc.RootCause.String(),\n\t\t\t\tMetadata: &timeout.TimeoutRootcauseMetadata{\n\t\t\t\t\tHeartBeatingMetadata: &metadata,\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\n\treturn result, nil\n}\n\nfunc retrieveFailureIssues(issues []invariant.InvariantCheckResult) ([]*failureIssuesResult, error) {\n\tresult := make([]*failureIssuesResult, 0)\n\tfor _, issue := range issues {\n\t\tif issue.InvariantType == failure.ActivityFailed.String() || issue.InvariantType == failure.WorkflowFailed.String() || issue.InvariantType == failure.DecisionCausedFailure.String() {\n\t\t\tvar data failure.FailureIssuesMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &data)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &failureIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata:      &data,\n\t\t\t})\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc retrieveFailureRootCause(rootCause []invariant.InvariantRootCauseResult) ([]*failureRootCauseResult, error) {\n\tresult := make([]*failureRootCauseResult, 0)\n\tfor _, rc := range rootCause {\n\t\tif rc.RootCause == invariant.RootCauseTypeServiceSideIssue || rc.RootCause == invariant.RootCauseTypeServiceSidePanic || rc.RootCause == invariant.RootCauseTypeServiceSideCustomError {\n\t\t\tresult = append(result, &failureRootCauseResult{\n\t\t\t\tIssueID:       rc.IssueID,\n\t\t\t\tRootCauseType: rc.RootCause.String(),\n\t\t\t})\n\t\t}\n\t\tif rc.RootCause == invariant.RootCauseTypeBlobSizeLimit {\n\t\t\tvar metadata failure.FailureRootcauseMetadata\n\t\t\terr := json.Unmarshal(rc.Metadata, &metadata)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &failureRootCauseResult{\n\t\t\t\tIssueID:       rc.IssueID,\n\t\t\t\tRootCauseType: rc.RootCause.String(),\n\t\t\t\tMetadata: &failure.FailureRootcauseMetadata{\n\t\t\t\t\tBlobSizeMetadata: metadata.BlobSizeMetadata,\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc retrieveRetryIssues(issues []invariant.InvariantCheckResult) ([]*retryIssuesResult, error) {\n\tresult := make([]*retryIssuesResult, 0)\n\tfor _, issue := range issues {\n\t\tif issueRetryRelated(issue) {\n\t\t\tvar data retry.RetryMetadata\n\t\t\terr := json.Unmarshal(issue.Metadata, &data)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t\tresult = append(result, &retryIssuesResult{\n\t\t\t\tIssueID:       issue.IssueID,\n\t\t\t\tInvariantType: issue.InvariantType,\n\t\t\t\tReason:        issue.Reason,\n\t\t\t\tMetadata:      data,\n\t\t\t})\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc rootCauseHeartBeatRelated(rootCause invariant.RootCause) bool {\n\tfor _, rc := range []invariant.RootCause{invariant.RootCauseTypeNoHeartBeatTimeoutNoRetryPolicy,\n\t\tinvariant.RootCauseTypeHeartBeatingNotEnabledWithRetryPolicy,\n\t\tinvariant.RootCauseTypeHeartBeatingEnabledWithoutRetryPolicy,\n\t\tinvariant.RootCauseTypeHeartBeatingEnabledMissingHeartbeat} {\n\t\tif rc == rootCause {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc rootCausePollersRelated(rootCause invariant.RootCause) bool {\n\tfor _, rc := range []invariant.RootCause{invariant.RootCauseTypePollersStatus, invariant.RootCauseTypeMissingPollers} {\n\t\tif rc == rootCause {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc issueRetryRelated(issue invariant.InvariantCheckResult) bool {\n\tfor _, i := range []string{retry.WorkflowRetryIssue.String(), retry.ActivityRetryIssue.String(), retry.ActivityHeartbeatIssue.String()} {\n\t\tif issue.InvariantType == i {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "service/worker/diagnostics/workflow_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage diagnostics\n\nimport (\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/failure\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/retry\"\n\t\"github.com/uber/cadence/service/worker/diagnostics/invariant/timeout\"\n)\n\ntype diagnosticsWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tworkflowEnv *testsuite.TestWorkflowEnvironment\n\tdw          *dw\n}\n\nfunc TestDiagnosticsWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(diagnosticsWorkflowTestSuite))\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) SetupTest() {\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\tpublicClient := mockResource.GetSDKClient()\n\ts.dw = &dw{\n\t\tlogger:        mockResource.GetLogger(),\n\t\tsvcClient:     publicClient,\n\t\tclientBean:    mockResource.ClientBean,\n\t\tmetricsClient: mockResource.GetMetricsClient(),\n\t\tinvariants:    []invariant.Invariant{timeout.NewInvariant(timeout.Params{Client: publicClient}), failure.NewInvariant(), retry.NewInvariant()},\n\t}\n\n\ts.T().Cleanup(func() {\n\t\tmockResource.Finish(s.T())\n\t})\n\n\ts.workflowEnv.RegisterWorkflowWithOptions(s.dw.DiagnosticsStarterWorkflow, workflow.RegisterOptions{Name: diagnosticsStarterWorkflow})\n\ts.workflowEnv.RegisterWorkflowWithOptions(s.dw.DiagnosticsWorkflow, workflow.RegisterOptions{Name: diagnosticsWorkflow})\n\ts.workflowEnv.RegisterActivityWithOptions(s.dw.identifyIssues, activity.RegisterOptions{Name: identifyIssuesActivity})\n\ts.workflowEnv.RegisterActivityWithOptions(s.dw.rootCauseIssues, activity.RegisterOptions{Name: rootCauseIssuesActivity})\n\ts.workflowEnv.RegisterActivityWithOptions(s.dw.emitUsageLogs, activity.RegisterOptions{Name: emitUsageLogsActivity})\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) TestWorkflow() {\n\tparams := &DiagnosticsStarterWorkflowInput{\n\t\tDomain:     \"test\",\n\t\tWorkflowID: \"123\",\n\t\tRunID:      \"abc\",\n\t}\n\tworkflowTimeoutData := timeout.ExecutionTimeoutMetadata{\n\t\tExecutionTime:     110 * time.Second,\n\t\tConfiguredTimeout: 110 * time.Second,\n\t\tLastOngoingEvent: &types.HistoryEvent{\n\t\t\tID:        1,\n\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t},\n\t\t},\n\t}\n\tworkflowTimeoutDataInBytes, err := json.Marshal(workflowTimeoutData)\n\ts.NoError(err)\n\tactMetadata := failure.FailureIssuesMetadata{\n\t\tIdentity:            \"localhost\",\n\t\tActivityScheduledID: 1,\n\t\tActivityStartedID:   2,\n\t}\n\tactMetadataInBytes, err := json.Marshal(actMetadata)\n\ts.NoError(err)\n\tissues := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: timeout.TimeoutTypeExecution.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata:      workflowTimeoutDataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: failure.ActivityFailed.String(),\n\t\t\tReason:        failure.ActivityOutputBlobSizeLimit.String(),\n\t\t\tMetadata:      actMetadataInBytes,\n\t\t},\n\t}\n\ttimeoutIssues := []*timeoutIssuesResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: timeout.TimeoutTypeExecution.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\tExecutionTimeout: &workflowTimeoutData,\n\t\t\t},\n\t\t},\n\t}\n\ttaskListBacklog := int64(10)\n\tpollersMetadataInBytes, err := json.Marshal(timeout.PollersMetadata{TaskListName: \"test\", TaskListBacklog: taskListBacklog})\n\ts.NoError(err)\n\tblobSizeMetadataInBytes, err := json.Marshal(failure.FailureRootcauseMetadata{\n\t\tBlobSizeMetadata: &failure.BlobSizeMetadata{\n\t\t\tBlobSizeWarnLimit:  5,\n\t\t\tBlobSizeErrorLimit: 10,\n\t\t},\n\t})\n\ts.NoError(err)\n\trootCause := []invariant.InvariantRootCauseResult{\n\t\t{\n\t\t\tIssueID:   1,\n\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:   2,\n\t\t\tRootCause: invariant.RootCauseTypeBlobSizeLimit,\n\t\t\tMetadata:  blobSizeMetadataInBytes,\n\t\t},\n\t}\n\tfailureRootCause := []*failureRootCauseResult{\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tRootCauseType: invariant.RootCauseTypeBlobSizeLimit.String(),\n\t\t\tMetadata: &failure.FailureRootcauseMetadata{\n\t\t\t\tBlobSizeMetadata: &failure.BlobSizeMetadata{\n\t\t\t\t\tBlobSizeWarnLimit:  5,\n\t\t\t\t\tBlobSizeErrorLimit: 10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\ttimeoutRootCause := []*timeoutRootCauseResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tRootCauseType: invariant.RootCauseTypePollersStatus.String(),\n\t\t\tMetadata: &timeout.TimeoutRootcauseMetadata{\n\t\t\t\tPollersMetadata: &timeout.PollersMetadata{TaskListName: \"test\", TaskListBacklog: taskListBacklog},\n\t\t\t},\n\t\t}}\n\ts.workflowEnv.OnActivity(identifyIssuesActivity, mock.Anything, mock.Anything).Return(issues, nil)\n\ts.workflowEnv.OnActivity(rootCauseIssuesActivity, mock.Anything, mock.Anything).Return(rootCause, nil)\n\ts.workflowEnv.OnActivity(emitUsageLogsActivity, mock.Anything, mock.Anything).Return(nil)\n\ts.workflowEnv.ExecuteWorkflow(diagnosticsStarterWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\tvar result DiagnosticsStarterWorkflowResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.ElementsMatch(timeoutIssues, result.DiagnosticsResult.Timeouts.Issues)\n\ts.ElementsMatch(timeoutRootCause, result.DiagnosticsResult.Timeouts.RootCause)\n\ts.ElementsMatch(failureRootCause, result.DiagnosticsResult.Failures.RootCause)\n\ts.True(result.DiagnosticsCompleted)\n\n\tqueriedResult := s.queryDiagnostics()\n\ts.ElementsMatch(queriedResult.DiagnosticsResult.Timeouts.Issues, result.DiagnosticsResult.Timeouts.Issues)\n\ts.ElementsMatch(queriedResult.DiagnosticsResult.Timeouts.RootCause, result.DiagnosticsResult.Timeouts.RootCause)\n\ts.True(queriedResult.DiagnosticsCompleted)\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) TestWorkflow_Error() {\n\tparams := &DiagnosticsWorkflowInput{\n\t\tDomain:     \"test\",\n\t\tWorkflowID: \"123\",\n\t\tRunID:      \"abc\",\n\t}\n\tmockErr := errors.New(\"mockErr\")\n\terrExpected := fmt.Errorf(\"IdentifyIssues: %w\", mockErr)\n\ts.workflowEnv.OnActivity(identifyIssuesActivity, mock.Anything, mock.Anything).Return(nil, mockErr)\n\ts.workflowEnv.ExecuteWorkflow(diagnosticsWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n\ts.EqualError(s.workflowEnv.GetWorkflowError(), errExpected.Error())\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) TestWorkflow_NoErrorIfEmitLogsActivityFails() {\n\tparams := &DiagnosticsWorkflowInput{\n\t\tDomain:     \"test\",\n\t\tWorkflowID: \"123\",\n\t\tRunID:      \"abc\",\n\t}\n\tmockErr := errors.New(\"mockErr\")\n\ts.workflowEnv.OnActivity(identifyIssuesActivity, mock.Anything, mock.Anything).Return(nil, nil)\n\ts.workflowEnv.OnActivity(rootCauseIssuesActivity, mock.Anything, mock.Anything).Return(nil, nil)\n\ts.workflowEnv.OnActivity(emitUsageLogsActivity, mock.Anything, mock.Anything).Return(mockErr)\n\ts.workflowEnv.ExecuteWorkflow(diagnosticsStarterWorkflow, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.NoError(s.workflowEnv.GetWorkflowError())\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) queryDiagnostics() DiagnosticsStarterWorkflowResult {\n\tqueryFuture, err := s.workflowEnv.QueryWorkflow(queryDiagnosticsReport)\n\ts.NoError(err)\n\n\tvar result DiagnosticsStarterWorkflowResult\n\terr = queryFuture.Get(&result)\n\ts.NoError(err)\n\treturn result\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) Test__retrieveTimeoutIssues() {\n\tworkflowTimeoutData := timeout.ExecutionTimeoutMetadata{\n\t\tExecutionTime:     110 * time.Second,\n\t\tConfiguredTimeout: 110 * time.Second,\n\t\tLastOngoingEvent: &types.HistoryEvent{\n\t\t\tID:        1,\n\t\t\tTimestamp: common.Int64Ptr(testTimeStamp),\n\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeoutSecond),\n\t\t\t},\n\t\t},\n\t}\n\tworkflowTimeoutDataInBytes, err := json.Marshal(workflowTimeoutData)\n\ts.NoError(err)\n\tchildWorkflowTimeoutData := timeout.ChildWfTimeoutMetadata{\n\t\tExecutionTime:     110 * time.Second,\n\t\tConfiguredTimeout: 110 * time.Second,\n\t}\n\tchildWorkflowTimeoutDataInBytes, err := json.Marshal(childWorkflowTimeoutData)\n\ts.NoError(err)\n\tactivityTimeoutData := timeout.ActivityTimeoutMetadata{\n\t\tTimeoutType:       types.TimeoutTypeStartToClose.Ptr(),\n\t\tConfiguredTimeout: 5 * time.Second,\n\t\tTimeElapsed:       5 * time.Second,\n\t\tHeartBeatTimeout:  0,\n\t}\n\tactivityTimeoutDataInBytes, err := json.Marshal(activityTimeoutData)\n\ts.NoError(err)\n\tdescTimeoutData := timeout.DecisionTimeoutMetadata{\n\t\tConfiguredTimeout: 5 * time.Second,\n\t}\n\tdescTimeoutDataInBytes, err := json.Marshal(activityTimeoutData)\n\ts.NoError(err)\n\tissues := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: timeout.TimeoutTypeExecution.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata:      workflowTimeoutDataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tInvariantType: timeout.TimeoutTypeActivity.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata:      activityTimeoutDataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       3,\n\t\t\tInvariantType: timeout.TimeoutTypeDecision.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata:      descTimeoutDataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       4,\n\t\t\tInvariantType: timeout.TimeoutTypeChildWorkflow.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata:      childWorkflowTimeoutDataInBytes,\n\t\t},\n\t}\n\ttimeoutIssues := []*timeoutIssuesResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: timeout.TimeoutTypeExecution.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\tExecutionTimeout: &workflowTimeoutData,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tInvariantType: timeout.TimeoutTypeActivity.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\tActivityTimeout: &activityTimeoutData,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tIssueID:       3,\n\t\t\tInvariantType: timeout.TimeoutTypeDecision.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\tDecisionTimeout: &descTimeoutData,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tIssueID:       4,\n\t\t\tInvariantType: timeout.TimeoutTypeChildWorkflow.String(),\n\t\t\tReason:        \"START_TO_CLOSE\",\n\t\t\tMetadata: &timeout.TimeoutIssuesMetadata{\n\t\t\t\tChildWfTimeout: &childWorkflowTimeoutData,\n\t\t\t},\n\t\t},\n\t}\n\tresult, err := retrieveTimeoutIssues(issues)\n\ts.NoError(err)\n\ts.ElementsMatch(timeoutIssues, result)\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) Test__retrieveTimeoutRootCause() {\n\ttaskListBacklog := int64(10)\n\tpollersMetadataInBytes, err := json.Marshal(timeout.PollersMetadata{TaskListBacklog: taskListBacklog})\n\ts.NoError(err)\n\theartBeatingMetadataInBytes, err := json.Marshal(timeout.HeartbeatingMetadata{TimeElapsed: 5 * time.Second})\n\ts.NoError(err)\n\trootCause := []invariant.InvariantRootCauseResult{\n\t\t{\n\t\t\tIssueID:   1,\n\t\t\tRootCause: invariant.RootCauseTypePollersStatus,\n\t\t\tMetadata:  pollersMetadataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:   2,\n\t\t\tRootCause: invariant.RootCauseTypeNoHeartBeatTimeoutNoRetryPolicy,\n\t\t\tMetadata:  heartBeatingMetadataInBytes,\n\t\t},\n\t}\n\ttimeoutRootCause := []*timeoutRootCauseResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tRootCauseType: invariant.RootCauseTypePollersStatus.String(),\n\t\t\tMetadata: &timeout.TimeoutRootcauseMetadata{\n\t\t\t\tPollersMetadata: &timeout.PollersMetadata{TaskListBacklog: taskListBacklog},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tRootCauseType: invariant.RootCauseTypeNoHeartBeatTimeoutNoRetryPolicy.String(),\n\t\t\tMetadata: &timeout.TimeoutRootcauseMetadata{\n\t\t\t\tHeartBeatingMetadata: &timeout.HeartbeatingMetadata{TimeElapsed: 5 * time.Second},\n\t\t\t},\n\t\t},\n\t}\n\tresult, err := retrieveTimeoutRootCause(rootCause)\n\ts.NoError(err)\n\ts.ElementsMatch(timeoutRootCause, result)\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) Test__retrieveFailureIssues() {\n\tactMetadata := failure.FailureIssuesMetadata{\n\t\tIdentity:            \"localhost\",\n\t\tActivityScheduledID: 1,\n\t\tActivityStartedID:   2,\n\t}\n\tactMetadataInBytes, err := json.Marshal(actMetadata)\n\ts.NoError(err)\n\tissues := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: failure.ActivityFailed.String(),\n\t\t\tReason:        failure.CustomError.String(),\n\t\t\tMetadata:      actMetadataInBytes,\n\t\t},\n\t}\n\tfailureIssues := []*failureIssuesResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: failure.ActivityFailed.String(),\n\t\t\tReason:        failure.CustomError.String(),\n\t\t\tMetadata:      &actMetadata,\n\t\t},\n\t}\n\tresult, err := retrieveFailureIssues(issues)\n\ts.NoError(err)\n\ts.ElementsMatch(failureIssues, result)\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) Test__retrieveFailureRootCause() {\n\tblobSizeMetadataInBytes, err := json.Marshal(failure.FailureRootcauseMetadata{\n\t\tBlobSizeMetadata: &failure.BlobSizeMetadata{\n\t\t\tBlobSizeWarnLimit:  5,\n\t\t\tBlobSizeErrorLimit: 10,\n\t\t},\n\t})\n\ts.NoError(err)\n\trootCause := []invariant.InvariantRootCauseResult{\n\t\t{\n\t\t\tIssueID:   1,\n\t\t\tRootCause: invariant.RootCauseTypeServiceSideIssue,\n\t\t},\n\t\t{\n\t\t\tIssueID:   2,\n\t\t\tRootCause: invariant.RootCauseTypeBlobSizeLimit,\n\t\t\tMetadata:  blobSizeMetadataInBytes,\n\t\t},\n\t}\n\tfailureRootCause := []*failureRootCauseResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tRootCauseType: invariant.RootCauseTypeServiceSideIssue.String(),\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tRootCauseType: invariant.RootCauseTypeBlobSizeLimit.String(),\n\t\t\tMetadata: &failure.FailureRootcauseMetadata{\n\t\t\t\tBlobSizeMetadata: &failure.BlobSizeMetadata{\n\t\t\t\t\tBlobSizeWarnLimit:  5,\n\t\t\t\t\tBlobSizeErrorLimit: 10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tresult, err := retrieveFailureRootCause(rootCause)\n\ts.NoError(err)\n\ts.ElementsMatch(failureRootCause, result)\n}\n\nfunc (s *diagnosticsWorkflowTestSuite) Test__retrieveRetryIssues() {\n\tretryMetadata := retry.RetryMetadata{\n\t\tRetryPolicy: &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds: 1,\n\t\t\tMaximumAttempts:          1,\n\t\t},\n\t}\n\tretryMetadataInBytes, err := json.Marshal(retryMetadata)\n\ts.NoError(err)\n\tissues := []invariant.InvariantCheckResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: retry.ActivityRetryIssue.String(),\n\t\t\tReason:        retry.RetryPolicyValidationMaxAttempts.String(),\n\t\t\tMetadata:      retryMetadataInBytes,\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tInvariantType: retry.WorkflowRetryIssue.String(),\n\t\t\tReason:        retry.RetryPolicyValidationMaxAttempts.String(),\n\t\t\tMetadata:      retryMetadataInBytes,\n\t\t},\n\t}\n\tretryIssues := []*retryIssuesResult{\n\t\t{\n\t\t\tIssueID:       1,\n\t\t\tInvariantType: retry.ActivityRetryIssue.String(),\n\t\t\tReason:        retry.RetryPolicyValidationMaxAttempts.String(),\n\t\t\tMetadata:      retryMetadata,\n\t\t},\n\t\t{\n\t\t\tIssueID:       2,\n\t\t\tInvariantType: retry.WorkflowRetryIssue.String(),\n\t\t\tReason:        retry.RetryPolicyValidationMaxAttempts.String(),\n\t\t\tMetadata:      retryMetadata,\n\t\t},\n\t}\n\tresult, err := retrieveRetryIssues(issues)\n\ts.NoError(err)\n\ts.ElementsMatch(retryIssues, result)\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/activities.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"go.uber.org/cadence\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// DisableArchivalActivity disables archival for the domain\nfunc (w *domainDeprecator) DisableArchivalActivity(ctx context.Context, params DomainDeprecationParams) error {\n\tclient := w.clientBean.GetFrontendClient()\n\tdisabled := types.ArchivalStatusDisabled\n\n\tdescribeRequest := &types.DescribeDomainRequest{\n\t\tName: &params.DomainName,\n\t}\n\tdomainResp, err := client.DescribeDomain(ctx, describeRequest)\n\tif err != nil {\n\t\tvar entityNotExistsError *types.EntityNotExistsError\n\t\tif errors.As(err, &entityNotExistsError) {\n\t\t\treturn cadence.NewCustomError(ErrDomainDoesNotExistNonRetryable)\n\t\t}\n\t\treturn fmt.Errorf(\"failed to describe domain: %v\", err)\n\t}\n\n\t// Check if archival is already disabled\n\tif *domainResp.Configuration.VisibilityArchivalStatus == disabled &&\n\t\t*domainResp.Configuration.HistoryArchivalStatus == disabled {\n\t\tw.logger.Info(\"Archival is already disabled for domain\", tag.WorkflowDomainName(params.DomainName))\n\t\treturn nil\n\t}\n\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:                     params.DomainName,\n\t\tHistoryArchivalStatus:    &disabled,\n\t\tVisibilityArchivalStatus: &disabled,\n\t\tSecurityToken:            params.SecurityToken,\n\t}\n\tupdateResp, err := client.UpdateDomain(ctx, updateRequest)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to update domain: %v\", err)\n\t}\n\n\tif *updateResp.Configuration.VisibilityArchivalStatus != disabled ||\n\t\t*updateResp.Configuration.HistoryArchivalStatus != disabled {\n\t\treturn fmt.Errorf(\"failed to disable archival for domain %s\", params.DomainName)\n\t}\n\n\tw.logger.Info(\"Disabled archival for domain\", tag.WorkflowDomainName(params.DomainName))\n\treturn nil\n}\n\n// DeprecateDomainActivity deprecates the domain\nfunc (w *domainDeprecator) DeprecateDomainActivity(ctx context.Context, params DomainDeprecationParams) error {\n\tclient := w.clientBean.GetFrontendClient()\n\n\terr := client.DeprecateDomain(ctx, &types.DeprecateDomainRequest{\n\t\tName:          params.DomainName,\n\t\tSecurityToken: params.SecurityToken,\n\t})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to deprecate domain: %v\", err)\n\t}\n\n\treturn nil\n}\n\n// CheckOpenWorkflowsActivity checks if there are any open workflows in the domain\nfunc (w *domainDeprecator) CheckOpenWorkflowsActivity(ctx context.Context, params DomainDeprecationParams) (bool, error) {\n\tclient := w.clientBean.GetFrontendClient()\n\n\tcountRequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: params.DomainName,\n\t\tQuery:  \"CloseTime = missing\",\n\t}\n\n\tcountResp, err := client.CountWorkflowExecutions(ctx, countRequest)\n\tif err != nil {\n\t\treturn false, fmt.Errorf(\"failed to count open workflows: %v\", err)\n\t}\n\n\thasOpenWorkflows := countResp.Count > 0\n\tif hasOpenWorkflows {\n\t\tw.logger.Info(\"Found open workflows in domain\",\n\t\t\ttag.WorkflowDomainName(params.DomainName),\n\t\t\ttag.Number(countResp.Count))\n\t} else {\n\t\tw.logger.Info(\"No open workflows found in domain\",\n\t\t\ttag.WorkflowDomainName(params.DomainName))\n\t}\n\n\treturn hasOpenWorkflows, nil\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/activities_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestDisableArchivalActivity(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockClient := frontend.NewMockClient(ctrl)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockClient).AnyTimes()\n\n\tdeprecator := &domainDeprecator{\n\t\tcfg: Config{\n\t\t\tAdminOperationToken: dynamicproperties.GetStringPropertyFn(\"\"),\n\t\t},\n\t\tclientBean: mockClientBean,\n\t\tlogger:     testlogger.New(t),\n\t}\n\n\ttestDomain := \"test-domain\"\n\tsecurityToken := \"token\"\n\tdisabled := types.ArchivalStatusDisabled\n\tenabled := types.ArchivalStatusEnabled\n\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMocks    func()\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"Success - Disable archival\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&types.DescribeDomainResponse{\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tVisibilityArchivalStatus: &enabled,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    &enabled,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t\tmockClient.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&types.UpdateDomainResponse{\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tVisibilityArchivalStatus: &disabled,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    &disabled,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - Archival already disabled\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&types.DescribeDomainResponse{\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tVisibilityArchivalStatus: &disabled,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    &disabled,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - Describe domain fails\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\tnil, assert.AnError)\n\t\t\t},\n\t\t\texpectedError: assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - Update domain fails\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\t&types.DescribeDomainResponse{\n\t\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\t\tVisibilityArchivalStatus: &enabled,\n\t\t\t\t\t\t\tHistoryArchivalStatus:    &enabled,\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil)\n\t\t\t\tmockClient.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\tnil, assert.AnError)\n\t\t\t},\n\t\t\texpectedError: assert.AnError,\n\t\t},\n\t\t{\n\t\t\tname: \"Error - Domain does not exist\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(\n\t\t\t\t\tnil, types.EntityNotExistsError{},\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedError: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttt.setupMocks()\n\t\t\terr := deprecator.DisableArchivalActivity(context.Background(), DomainDeprecationParams{\n\t\t\t\tDomainName:    testDomain,\n\t\t\t\tSecurityToken: securityToken,\n\t\t\t})\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDeprecateDomainActivity(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockClient := frontend.NewMockClient(ctrl)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockClient).AnyTimes()\n\n\tdeprecator := &domainDeprecator{\n\t\tcfg: Config{\n\t\t\tAdminOperationToken: dynamicproperties.GetStringPropertyFn(\"\"),\n\t\t},\n\t\tclientBean: mockClientBean,\n\t\tlogger:     testlogger.New(t),\n\t}\n\n\ttestDomain := \"test-domain\"\n\tsecurityToken := \"token\"\n\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMocks    func()\n\t\texpectedError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DeprecateDomain(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\texpectedError: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().DeprecateDomain(gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t\t\t},\n\t\t\texpectedError: assert.AnError,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttt.setupMocks()\n\t\t\terr := deprecator.DeprecateDomainActivity(context.Background(), DomainDeprecationParams{\n\t\t\t\tDomainName:    testDomain,\n\t\t\t\tSecurityToken: securityToken,\n\t\t\t})\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestCheckOpenWorkflowsActivity(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tmockClient := frontend.NewMockClient(ctrl)\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockClientBean.EXPECT().GetFrontendClient().Return(mockClient).AnyTimes()\n\n\tdeprecator := &domainDeprecator{\n\t\tcfg: Config{\n\t\t\tAdminOperationToken: dynamicproperties.GetStringPropertyFn(\"\"),\n\t\t},\n\t\tclientBean: mockClientBean,\n\t\tlogger:     testlogger.New(t),\n\t}\n\n\ttestDomain := \"test-domain\"\n\n\ttests := []struct {\n\t\tname           string\n\t\tsetupMocks     func()\n\t\texpectedResult bool\n\t\texpectedError  error\n\t}{\n\t\t{\n\t\t\tname: \"Success - no open workflows\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{Count: 0}, nil)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - has open workflows\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{Count: 100}, nil)\n\t\t\t},\n\t\t\texpectedResult: true,\n\t\t\texpectedError:  nil,\n\t\t},\n\t\t{\n\t\t\tname: \"Error\",\n\t\t\tsetupMocks: func() {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{Count: 0}, assert.AnError)\n\t\t\t},\n\t\t\texpectedResult: false,\n\t\t\texpectedError:  assert.AnError,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttt.setupMocks()\n\t\t\thasOpenWorkflows, err := deprecator.CheckOpenWorkflowsActivity(context.Background(), DomainDeprecationParams{\n\t\t\t\tDomainName: testDomain,\n\t\t\t})\n\t\t\tif tt.expectedError != nil {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.expectedResult, hasOpenWorkflows)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/entities.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\n// DomainDeprecationParams contains the parameters required for domain deprecation workflow.\ntype DomainDeprecationParams struct {\n\tDomainName    string `json:\"domain_name\"`\n\tSecurityToken string `json:\"security_token\"`\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/helpers.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport \"time\"\n\nconst (\n\t// ErrDomainDoesNotExistNonRetryable is error reason used for Cadence non-retryable errors\n\tErrDomainDoesNotExistNonRetryable = \"domain does not exist\"\n\t// ErrAccessDeniedNonRetryable is permission error that require manual intervention\n\tErrAccessDeniedNonRetryable = \"AccessDeniedError\"\n\t// ErrWorkflowAlreadyCompletedNonRetryable is error that indicates workflow execution already completed\n\tErrWorkflowAlreadyCompletedNonRetryable = \"WorkflowExecutionAlreadyCompletedError\"\n\n\t// DefaultRPS is the default RPS\n\tDefaultRPS = 50\n\t// DefaultConcurrency is the default concurrency\n\tDefaultConcurrency = 5\n\t// DefaultPageSize is the default page size\n\tDefaultPageSize = 1000\n\t// DefaultAttemptsOnRetryableError is the default value for AttemptsOnRetryableError\n\tDefaultAttemptsOnRetryableError = 50\n\t// DefaultActivityHeartBeatTimeout is the default value for ActivityHeartBeatTimeout\n\tDefaultActivityHeartBeatTimeout = time.Second * 10\n\t// DefaultMaxActivityRetries is the default value for MaxActivityRetries\n\tDefaultMaxActivityRetries = 4\n)\n"
  },
  {
    "path": "service/worker/domaindeprecation/module.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"context\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n)\n\ntype (\n\tDomainDeprecationWorker interface {\n\t\tStart() error\n\t\tStop()\n\t}\n\n\t// Config defines the configuration for domain deprecator\n\tConfig struct {\n\t\t// AdminOperationToken is a dynamic config that provides the security token for admin operations\n\t\tAdminOperationToken dynamicproperties.StringPropertyFn\n\t}\n\n\tdomainDeprecator struct {\n\t\tcfg           Config\n\t\tsvcClient     workflowserviceclient.Interface\n\t\tclientBean    client.Bean\n\t\tmetricsClient metrics.Client\n\t\tworker        worker.Worker\n\t\ttally         tally.Scope\n\t\tlogger        log.Logger\n\t}\n\n\tParams struct {\n\t\tConfig        Config\n\t\tServiceClient workflowserviceclient.Interface\n\t\tClientBean    client.Bean\n\t\tMetricsClient metrics.Client\n\t\tTally         tally.Scope\n\t\tLogger        log.Logger\n\t}\n)\n\n// New creates a new domain deprecation workflow.\nfunc New(params Params) DomainDeprecationWorker {\n\treturn &domainDeprecator{\n\t\tcfg:           params.Config,\n\t\tsvcClient:     params.ServiceClient,\n\t\tclientBean:    params.ClientBean,\n\t\tmetricsClient: params.MetricsClient,\n\t\ttally:         params.Tally,\n\t\tlogger:        params.Logger,\n\t}\n}\n\n// Start starts the worker\nfunc (w *domainDeprecator) Start() error {\n\tbatcherParams := &batcher.BootstrapParams{\n\t\tConfig: batcher.Config{\n\t\t\tAdminOperationToken: w.cfg.AdminOperationToken,\n\t\t},\n\t\tServiceClient: w.svcClient,\n\t\tClientBean:    w.clientBean,\n\t\tMetricsClient: w.metricsClient,\n\t\tLogger:        w.logger,\n\t\tTallyScope:    w.tally,\n\t}\n\tbatcherInstance := batcher.New(batcherParams)\n\n\tctx := context.WithValue(context.Background(), batcher.BatcherContextKey, batcherInstance)\n\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:                     w.tally,\n\t\tBackgroundActivityContext:        ctx,\n\t\tTracer:                           opentracing.GlobalTracer(),\n\t\tMaxConcurrentActivityTaskPollers: 10,\n\t\tMaxConcurrentDecisionTaskPollers: 10,\n\t}\n\tnewWorker := worker.New(w.svcClient, constants.SystemLocalDomainName, DomainDeprecationTaskListName, workerOpts)\n\tnewWorker.RegisterWorkflowWithOptions(w.DomainDeprecationWorkflow, workflow.RegisterOptions{Name: DomainDeprecationWorkflowTypeName})\n\tnewWorker.RegisterActivityWithOptions(w.DisableArchivalActivity, activity.RegisterOptions{Name: disableArchivalActivity, EnableAutoHeartbeat: true})\n\tnewWorker.RegisterActivityWithOptions(w.CheckOpenWorkflowsActivity, activity.RegisterOptions{Name: checkOpenWorkflowsActivity, EnableAutoHeartbeat: true})\n\tnewWorker.RegisterActivityWithOptions(w.DeprecateDomainActivity, activity.RegisterOptions{Name: deprecateDomainActivity, EnableAutoHeartbeat: true})\n\tw.worker = newWorker\n\treturn newWorker.Start()\n}\n\nfunc (w *domainDeprecator) Stop() {\n\tw.worker.Stop()\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/module_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nfunc Test__Start(t *testing.T) {\n\tdomainDeprecationWorkerTest, mockResource := setupTest(t)\n\terr := domainDeprecationWorkerTest.Start()\n\trequire.NoError(t, err)\n\n\tdomainDeprecationWorkerTest.Stop()\n\tmockResource.Finish(t)\n}\n\nfunc setupTest(t *testing.T) (DomainDeprecationWorker, *resource.Test) {\n\tctrl := gomock.NewController(t)\n\n\tmockResource := resource.NewTest(t, ctrl, metrics.Worker)\n\tmockResource.SDKClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.DescribeDomainResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForDecisionTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForDecisionTaskResponse{}, nil).AnyTimes()\n\tmockResource.SDKClient.EXPECT().PollForActivityTask(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(&shared.PollForActivityTaskResponse{}, nil).AnyTimes()\n\n\tmockClientBean := client.NewMockBean(ctrl)\n\tmockSvcClient := mockResource.GetSDKClient()\n\n\treturn New(Params{\n\t\tConfig: Config{\n\t\t\tAdminOperationToken: dynamicproperties.GetStringPropertyFn(\"\"),\n\t\t},\n\t\tServiceClient: mockSvcClient,\n\t\tClientBean:    mockClientBean,\n\t\tTally:         tally.TestScope(nil),\n\t\tLogger:        mockResource.GetLogger(),\n\t}), mockResource\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/workflow.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n)\n\nconst (\n\tDomainDeprecationWorkflowTypeName = \"domain-deprecation-workflow\"\n\tDomainDeprecationTaskListName     = \"domain-deprecation-tasklist\"\n\tdomainDeprecationBatchPrefix      = \"domain-deprecation-batch\"\n\n\tdisableArchivalActivity    = \"disableArchival\"\n\tdeprecateDomainActivity    = \"deprecateDomain\"\n\tcheckOpenWorkflowsActivity = \"checkOpenWorkflows\"\n\n\tworkflowStartToCloseTimeout     = time.Hour * 24 * 30\n\tworkflowTaskStartToCloseTimeout = 5 * time.Minute\n\n\t// MaxBatchWorkflowAttempts is the maximum number of attempts to terminate open workflows\n\tMaxBatchWorkflowAttempts = 3\n\n\t// VisibilityRefreshWaitTime is the time to wait for visibility storage to refresh after workflow termination\n\tVisibilityRefreshWaitTime = 3 * time.Minute\n)\n\nvar (\n\tretryPolicy = cadence.RetryPolicy{\n\t\tInitialInterval:    10 * time.Second,\n\t\tBackoffCoefficient: 1.7,\n\t\tMaximumInterval:    5 * time.Minute,\n\t\tExpirationInterval: 24 * time.Hour,\n\t\tNonRetriableErrorReasons: []string{\n\t\t\tErrDomainDoesNotExistNonRetryable,\n\t\t\tErrAccessDeniedNonRetryable,\n\t\t},\n\t}\n\n\tactivityOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 5 * time.Minute,\n\t\tStartToCloseTimeout:    5 * time.Minute,\n\t\tHeartbeatTimeout:       5 * time.Minute,\n\t\tRetryPolicy:            &retryPolicy,\n\t}\n)\n\n// DomainDeprecationWorkflow is the workflow that handles domain deprecation process\nfunc (w *domainDeprecator) DomainDeprecationWorkflow(ctx workflow.Context, params DomainDeprecationParams) error {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"Starting domain deprecation workflow\", zap.String(\"domain\", params.DomainName))\n\n\t// Step 1: Activity disables archival\n\terr := workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, activityOptions),\n\t\tw.DisableArchivalActivity,\n\t\tparams,\n\t).Get(ctx, nil)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// Step 2: Deprecate a domain\n\terr = workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, activityOptions),\n\t\tw.DeprecateDomainActivity,\n\t\tparams,\n\t).Get(ctx, nil)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\t// Step 3: Start child batch workflow to terminate open workflows of a domain\n\tbatchParams := batcher.BatchParams{\n\t\tDomainName: params.DomainName,\n\t\tQuery:      \"CloseTime = missing\",\n\t\tReason:     \"domain is deprecated\",\n\t\tBatchType:  batcher.BatchTypeTerminate,\n\t\tTerminateParams: batcher.TerminateParams{\n\t\t\tTerminateChildren: common.BoolPtr(true),\n\t\t},\n\t\tRPS:                      DefaultRPS,\n\t\tConcurrency:              DefaultConcurrency,\n\t\tPageSize:                 DefaultPageSize,\n\t\tAttemptsOnRetryableError: DefaultAttemptsOnRetryableError,\n\t\tActivityHeartBeatTimeout: DefaultActivityHeartBeatTimeout,\n\t\tMaxActivityRetries:       DefaultMaxActivityRetries,\n\t\tNonRetryableErrors: []string{\n\t\t\tErrAccessDeniedNonRetryable,\n\t\t\tErrWorkflowAlreadyCompletedNonRetryable,\n\t\t},\n\t}\n\n\tfor attempt := 1; attempt <= MaxBatchWorkflowAttempts; attempt++ {\n\t\tlogger.Info(\"Starting batch workflow\",\n\t\t\tzap.String(\"domain\", params.DomainName),\n\t\t\tzap.Int(\"attempt\", attempt),\n\t\t\tzap.Int(\"max_attempts\", MaxBatchWorkflowAttempts))\n\n\t\tworkflowID := uuid.New()\n\t\tchildWorkflowOptions := workflow.WithChildOptions(ctx, workflow.ChildWorkflowOptions{\n\t\t\tWorkflowID:                   fmt.Sprintf(\"%s-%s-%s-%d\", domainDeprecationBatchPrefix, params.DomainName, workflowID, attempt),\n\t\t\tExecutionStartToCloseTimeout: workflowStartToCloseTimeout,\n\t\t\tTaskStartToCloseTimeout:      workflowTaskStartToCloseTimeout,\n\t\t})\n\n\t\tvar result batcher.HeartBeatDetails\n\t\terr = workflow.ExecuteChildWorkflow(childWorkflowOptions, batcher.BatchWorkflow, batchParams).Get(ctx, &result)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"batch workflow failed on attempt %d: %v\", attempt, err)\n\t\t}\n\n\t\t// Wait for visibility storage to refresh\n\t\terr = workflow.Sleep(ctx, VisibilityRefreshWaitTime)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"workflow sleep failed on attempt %d: %v\", attempt, err)\n\t\t}\n\n\t\t// Check if there are still open workflows\n\t\tvar hasOpenWorkflows bool\n\t\terr = workflow.ExecuteActivity(\n\t\t\tworkflow.WithActivityOptions(ctx, activityOptions),\n\t\t\tw.CheckOpenWorkflowsActivity,\n\t\t\tparams,\n\t\t).Get(ctx, &hasOpenWorkflows)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to check open workflows on attempt %d: %v\", attempt, err)\n\t\t}\n\n\t\tif !hasOpenWorkflows {\n\t\t\tlogger.Info(\"No open workflows found after batch workflow\",\n\t\t\t\tzap.String(\"domain\", params.DomainName),\n\t\t\t\tzap.Int(\"attempt\", attempt))\n\t\t\tbreak\n\t\t}\n\n\t\tif attempt == MaxBatchWorkflowAttempts {\n\t\t\treturn fmt.Errorf(\"failed to terminate all workflows after %d attempts\", MaxBatchWorkflowAttempts)\n\t\t}\n\n\t\tlogger.Info(\"Found open workflows after batch workflow, will retry\",\n\t\t\tzap.String(\"domain\", params.DomainName),\n\t\t\tzap.Int(\"attempt\", attempt))\n\t}\n\n\tlogger.Info(\"Domain deprecation workflow completed successfully\", zap.String(\"domain\", params.DomainName))\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/domaindeprecation/workflow_test.go",
    "content": "// Copyright (c) 2024 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage domaindeprecation\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n)\n\nvar (\n\ttestDomain    = \"test-domain\"\n\tdefaultParams = DomainDeprecationParams{\n\t\tDomainName:    testDomain,\n\t\tSecurityToken: \"token\",\n\t}\n)\n\ntype domainDeprecationWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tworkflowEnv *testsuite.TestWorkflowEnvironment\n\tdeprecator  *domainDeprecator\n}\n\nfunc TestDomainDeprecationWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(domainDeprecationWorkflowTestSuite))\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) SetupTest() {\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\tpublicClient := mockResource.GetSDKClient()\n\ts.deprecator = &domainDeprecator{\n\t\tcfg: Config{\n\t\t\tAdminOperationToken: dynamicproperties.GetStringPropertyFn(\"\"),\n\t\t},\n\t\tsvcClient:     publicClient,\n\t\tclientBean:    mockResource.ClientBean,\n\t\tmetricsClient: metrics.NewNoopMetricsClient(),\n\t\ttally:         tally.NoopScope,\n\t\tlogger:        mockResource.GetLogger(),\n\t}\n\n\ts.T().Cleanup(func() {\n\t\tmockResource.Finish(s.T())\n\t})\n\n\ts.workflowEnv.RegisterWorkflowWithOptions(s.deprecator.DomainDeprecationWorkflow, workflow.RegisterOptions{Name: DomainDeprecationWorkflowTypeName})\n\ts.workflowEnv.RegisterActivityWithOptions(s.deprecator.DisableArchivalActivity, activity.RegisterOptions{Name: disableArchivalActivity})\n\ts.workflowEnv.RegisterActivityWithOptions(s.deprecator.DeprecateDomainActivity, activity.RegisterOptions{Name: deprecateDomainActivity})\n\ts.workflowEnv.RegisterActivityWithOptions(s.deprecator.CheckOpenWorkflowsActivity, activity.RegisterOptions{Name: checkOpenWorkflowsActivity})\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TestWorkflow_Success() {\n\ts.workflowEnv.OnActivity(disableArchivalActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnActivity(deprecateDomainActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnWorkflow(batcher.BatchWorkflow, mock.Anything, mock.Anything).Return(\n\t\tbatcher.HeartBeatDetails{\n\t\t\tSuccessCount: 10,\n\t\t\tErrorCount:   0,\n\t\t}, nil).Once()\n\ts.workflowEnv.OnActivity(checkOpenWorkflowsActivity, mock.Anything, defaultParams).Return(false, nil)\n\n\ts.workflowEnv.ExecuteWorkflow(DomainDeprecationWorkflowTypeName, defaultParams)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.NoError(s.workflowEnv.GetWorkflowError())\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TestWorkflow_Disable_Archival_Error() {\n\tmockErr := errors.New(\"error\")\n\ts.workflowEnv.OnActivity(disableArchivalActivity, mock.Anything, defaultParams).Return(mockErr)\n\ts.workflowEnv.ExecuteWorkflow(DomainDeprecationWorkflowTypeName, defaultParams)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), mockErr.Error())\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TestWorkflow_Deprecate_Domain_Error() {\n\tmockErr := errors.New(\"error\")\n\ts.workflowEnv.OnActivity(disableArchivalActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnActivity(deprecateDomainActivity, mock.Anything, defaultParams).Return(mockErr)\n\ts.workflowEnv.ExecuteWorkflow(DomainDeprecationWorkflowTypeName, defaultParams)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), mockErr.Error())\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TestWorkflow_BatchTerminate_ChildWorkflowError() {\n\tmockErr := errors.New(\"batch workflow failed\")\n\ts.workflowEnv.OnActivity(disableArchivalActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnActivity(deprecateDomainActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnWorkflow(batcher.BatchWorkflow, mock.Anything, mock.Anything).Return(\n\t\tbatcher.HeartBeatDetails{}, mockErr)\n\n\ts.workflowEnv.ExecuteWorkflow(DomainDeprecationWorkflowTypeName, defaultParams)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), mockErr.Error())\n}\n\nfunc (s *domainDeprecationWorkflowTestSuite) TestWorkflow_CheckOpenWorkflows_Error() {\n\tmockErr := errors.New(\"error\")\n\ts.workflowEnv.OnActivity(disableArchivalActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnActivity(deprecateDomainActivity, mock.Anything, defaultParams).Return(nil)\n\ts.workflowEnv.OnWorkflow(batcher.BatchWorkflow, mock.Anything, mock.Anything).Return(\n\t\tbatcher.HeartBeatDetails{\n\t\t\tSuccessCount: 10,\n\t\t\tErrorCount:   0,\n\t\t}, nil).Once()\n\n\ts.workflowEnv.OnActivity(checkOpenWorkflowsActivity, mock.Anything, defaultParams).Return(true, mockErr)\n\n\ts.workflowEnv.ExecuteWorkflow(DomainDeprecationWorkflowTypeName, defaultParams)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n\ts.ErrorContains(s.workflowEnv.GetWorkflowError(), mockErr.Error())\n}\n"
  },
  {
    "path": "service/worker/esanalyzer/analyzer.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esanalyzer\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/worker/workercommon\"\n)\n\ntype readMode string\n\nconst (\n\tPinot readMode = \"pinot\"\n\tES    readMode = \"es\"\n)\n\ntype (\n\t// Analyzer is the background sub-system to query ElasticSearch and execute mitigations\n\tAnalyzer struct {\n\t\tsvcClient           workflowserviceclient.Interface\n\t\tfrontendClient      frontend.Client\n\t\tclientBean          client.Bean\n\t\tesClient            es.GenericClient\n\t\tpinotClient         pinot.GenericClient\n\t\treadMode            readMode\n\t\tlogger              log.Logger\n\t\ttallyScope          tally.Scope\n\t\tvisibilityIndexName string\n\t\tpinotTableName      string\n\t\tresource            resource.Resource\n\t\tdomainCache         cache.DomainCache\n\t\tconfig              *Config\n\t}\n\n\t// Config contains all configs for ElasticSearch Analyzer\n\tConfig struct {\n\t\tESAnalyzerPause                          dynamicproperties.BoolPropertyFn\n\t\tESAnalyzerTimeWindow                     dynamicproperties.DurationPropertyFn\n\t\tESAnalyzerMaxNumDomains                  dynamicproperties.IntPropertyFn\n\t\tESAnalyzerMaxNumWorkflowTypes            dynamicproperties.IntPropertyFn\n\t\tESAnalyzerLimitToTypes                   dynamicproperties.StringPropertyFn\n\t\tESAnalyzerEnableAvgDurationBasedChecks   dynamicproperties.BoolPropertyFn\n\t\tESAnalyzerLimitToDomains                 dynamicproperties.StringPropertyFn\n\t\tESAnalyzerNumWorkflowsToRefresh          dynamicproperties.IntPropertyFnWithWorkflowTypeFilter\n\t\tESAnalyzerBufferWaitTime                 dynamicproperties.DurationPropertyFnWithWorkflowTypeFilter\n\t\tESAnalyzerMinNumWorkflowsForAvg          dynamicproperties.IntPropertyFnWithWorkflowTypeFilter\n\t\tESAnalyzerWorkflowDurationWarnThresholds dynamicproperties.StringPropertyFn\n\t\tESAnalyzerWorkflowVersionDomains         dynamicproperties.StringPropertyFn\n\t\tESAnalyzerWorkflowTypeDomains            dynamicproperties.StringPropertyFn\n\t}\n\n\tWorkflow struct {\n\t\tanalyzer *Analyzer\n\t}\n\n\tEsAggregateCount struct {\n\t\tAggregateKey   string `json:\"key\"`\n\t\tAggregateCount int64  `json:\"doc_count\"`\n\t}\n)\n\nconst (\n\ttaskListName = \"cadence-sys-es-analyzer\"\n\n\tstartUpDelay = time.Second * 10\n)\n\n// New returns a new instance as daemon\nfunc New(\n\tsvcClient workflowserviceclient.Interface,\n\tfrontendClient frontend.Client,\n\tclientBean client.Bean,\n\tesClient es.GenericClient,\n\tpinotClient pinot.GenericClient,\n\tesConfig *config.ElasticSearchConfig,\n\tpinotConfig *config.PinotVisibilityConfig,\n\tlogger log.Logger,\n\ttallyScope tally.Scope,\n\tresource resource.Resource,\n\tdomainCache cache.DomainCache,\n\tconfig *Config,\n) *Analyzer {\n\tvar mode readMode\n\tvar indexName string\n\tvar pinotTableName string\n\n\tif esClient != nil {\n\t\tmode = ES\n\t\tindexName = esConfig.Indices[constants.VisibilityAppName]\n\t\tpinotTableName = \"\"\n\t} else if pinotClient != nil {\n\t\tmode = Pinot\n\t\tindexName = \"\"\n\t\tpinotTableName = pinotConfig.Table\n\t}\n\n\treturn &Analyzer{\n\t\tsvcClient:           svcClient,\n\t\tfrontendClient:      frontendClient,\n\t\tclientBean:          clientBean,\n\t\tesClient:            esClient,\n\t\tpinotClient:         pinotClient,\n\t\treadMode:            mode,\n\t\tlogger:              logger,\n\t\ttallyScope:          tallyScope,\n\t\tvisibilityIndexName: indexName,\n\t\tpinotTableName:      pinotTableName,\n\t\tresource:            resource,\n\t\tdomainCache:         domainCache,\n\t\tconfig:              config,\n\t}\n}\n\n// Start starts the scanner\nfunc (a *Analyzer) Start() error {\n\tctx := context.Background()\n\ta.StartWorkflow(ctx)\n\tctx = context.Background()\n\ta.StartDomainWFTypeCountWorkflow(ctx)\n\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              a.tallyScope,\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\tesWorker := worker.New(a.svcClient, constants.SystemLocalDomainName, taskListName, workerOpts)\n\terr := esWorker.Start()\n\treturn err\n}\n\nfunc (a *Analyzer) StartWorkflow(ctx context.Context) {\n\tinitWorkflow(a)\n\tgo workercommon.StartWorkflowWithRetry(esanalyzerWFTypeName, startUpDelay, a.resource, func(client cclient.Client) error {\n\t\t_, err := client.StartWorkflow(ctx, wfOptions, esanalyzerWFTypeName)\n\t\tswitch err.(type) {\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn nil\n\t\tdefault:\n\t\t\ta.logger.Error(\"Failed to start ElasticSearch Analyzer\", tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t})\n}\n\nfunc (a *Analyzer) StartDomainWFTypeCountWorkflow(ctx context.Context) {\n\tinitDomainWorkflowTypeCountWorkflow(a)\n\tgo workercommon.StartWorkflowWithRetry(domainWFTypeCountWorkflowTypeName, startUpDelay, a.resource, func(client cclient.Client) error {\n\t\t_, err := client.StartWorkflow(ctx, domainWfTypeCountStartOptions, domainWFTypeCountWorkflowTypeName)\n\t\tswitch err.(type) {\n\t\tcase *shared.WorkflowExecutionAlreadyStartedError:\n\t\t\treturn nil\n\t\tdefault:\n\t\t\ta.logger.Error(\"Failed to start domain wf type count workflow\", tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t})\n}\n"
  },
  {
    "path": "service/worker/esanalyzer/analyzer_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABxILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esanalyzer\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\tesMocks \"github.com/uber/cadence/common/elasticsearch/mocks\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/pinot\"\n\t\"github.com/uber/cadence/service/history/resource\"\n)\n\ntype esanalyzerWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tactivityEnv        *testsuite.TestActivityEnvironment\n\tworkflowEnv        *testsuite.TestWorkflowEnvironment\n\tcontroller         *gomock.Controller\n\tresource           *resource.Test\n\tmockAdminClient    *admin.MockClient\n\tmockDomainCache    *cache.MockDomainCache\n\tclientBean         *client.MockBean\n\tlogger             *log.MockLogger\n\tmockMetricClient   *mocks.Client\n\tscopedMetricClient *mocks.Scope\n\tmockESClient       *esMocks.GenericClient\n\ttallyScope         tally.Scope\n\tanalyzer           *Analyzer\n\tworkflow           *Workflow\n\tconfig             Config\n\tDomainID           string\n\tDomainName         string\n\tWorkflowType       string\n\tWorkflowID         string\n\tRunID              string\n\tWorkflowVersion    string\n}\n\nfunc TestESAnalyzerWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(esanalyzerWorkflowTestSuite))\n}\n\nfunc (s *esanalyzerWorkflowTestSuite) SetupTest() {\n\ts.DomainID = \"deadbeef-0123-4567-890a-bcdef0123460\"\n\ts.DomainName = \"test-domain\"\n\ts.WorkflowType = \"test-workflow-type\"\n\ts.WorkflowID = \"test-workflow_id\"\n\ts.RunID = \"test-run_id\"\n\ts.WorkflowVersion = \"test-workflow-version\"\n\ts.tallyScope = tally.NoopScope\n\tactiveDomainCache := cache.NewGlobalDomainCacheEntryForTest(\n\t\t&persistence.DomainInfo{ID: s.DomainID, Name: s.DomainName},\n\t\t&persistence.DomainConfig{Retention: 1},\n\t\t&persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: cluster.TestCurrentClusterName,\n\t\t\tClusters: []*persistence.ClusterReplicationConfig{\n\t\t\t\t{ClusterName: cluster.TestCurrentClusterName},\n\t\t\t\t{ClusterName: cluster.TestAlternativeClusterName},\n\t\t\t},\n\t\t},\n\t\t1234,\n\t)\n\n\ts.config = Config{\n\t\tESAnalyzerPause:                          dynamicproperties.GetBoolPropertyFn(false),\n\t\tESAnalyzerTimeWindow:                     dynamicproperties.GetDurationPropertyFn(time.Hour * 24 * 30),\n\t\tESAnalyzerMaxNumDomains:                  dynamicproperties.GetIntPropertyFn(500),\n\t\tESAnalyzerMaxNumWorkflowTypes:            dynamicproperties.GetIntPropertyFn(100),\n\t\tESAnalyzerLimitToTypes:                   dynamicproperties.GetStringPropertyFn(\"\"),\n\t\tESAnalyzerLimitToDomains:                 dynamicproperties.GetStringPropertyFn(\"\"),\n\t\tESAnalyzerNumWorkflowsToRefresh:          dynamicproperties.GetIntPropertyFilteredByWorkflowType(2),\n\t\tESAnalyzerBufferWaitTime:                 dynamicproperties.GetDurationPropertyFilteredByWorkflowType(time.Minute * 30),\n\t\tESAnalyzerMinNumWorkflowsForAvg:          dynamicproperties.GetIntPropertyFilteredByWorkflowType(100),\n\t\tESAnalyzerWorkflowDurationWarnThresholds: dynamicproperties.GetStringPropertyFn(\"\"),\n\t}\n\n\ts.activityEnv = s.NewTestActivityEnvironment()\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\ts.controller = gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(s.controller)\n\ts.resource = resource.NewTest(s.T(), s.controller, metrics.Worker)\n\ts.mockAdminClient = admin.NewMockClient(s.controller)\n\ts.clientBean = client.NewMockBean(s.controller)\n\ts.logger = log.NewMockLogger(s.controller)\n\ts.mockMetricClient = &mocks.Client{}\n\ts.scopedMetricClient = &mocks.Scope{}\n\ts.mockESClient = &esMocks.GenericClient{}\n\n\t//\n\t// s.mockDomainCache.EXPECT().GetDomainByID(s.DomainID).Return(activeDomainCache, nil).AnyTimes()\n\ts.mockDomainCache.EXPECT().GetDomain(s.DomainName).Return(activeDomainCache, nil).AnyTimes()\n\n\t// SET UP ANALYZER\n\ts.analyzer = &Analyzer{\n\t\tsvcClient:   s.resource.GetSDKClient(),\n\t\tclientBean:  s.clientBean,\n\t\tdomainCache: s.mockDomainCache,\n\t\tlogger:      s.logger,\n\t\tesClient:    s.mockESClient,\n\t\tconfig:      &s.config,\n\t\ttallyScope:  s.tallyScope,\n\t}\n\ts.activityEnv.SetTestTimeout(time.Second * 5)\n\ts.activityEnv.SetWorkerOptions(worker.Options{BackgroundActivityContext: context.Background()})\n\n\t// REGISTER WORKFLOWS AND ACTIVITIES\n\ts.workflow = &Workflow{analyzer: s.analyzer}\n\ts.workflowEnv.RegisterWorkflowWithOptions(\n\t\ts.workflow.workflowFunc,\n\t\tworkflow.RegisterOptions{Name: esanalyzerWFTypeName})\n\ts.workflowEnv.RegisterActivityWithOptions(\n\t\ts.workflow.emitWorkflowVersionMetrics,\n\t\tactivity.RegisterOptions{Name: emitWorkflowVersionMetricsActivity},\n\t)\n\ts.activityEnv.RegisterActivityWithOptions(\n\t\ts.workflow.emitWorkflowVersionMetrics,\n\t\tactivity.RegisterOptions{Name: emitWorkflowVersionMetricsActivity},\n\t)\n\n\ts.workflowEnv.RegisterWorkflowWithOptions(\n\t\ts.workflow.emitWorkflowTypeCount,\n\t\tworkflow.RegisterOptions{Name: domainWFTypeCountWorkflowTypeName})\n\ts.activityEnv.RegisterActivityWithOptions(\n\t\ts.workflow.emitWorkflowTypeCountMetrics,\n\t\tactivity.RegisterOptions{Name: emitDomainWorkflowTypeCountMetricsActivity},\n\t)\n}\n\nfunc (s *esanalyzerWorkflowTestSuite) TearDownTest() {\n\tdefer s.controller.Finish()\n\tdefer s.resource.Finish(s.T())\n\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc (s *esanalyzerWorkflowTestSuite) TestExecuteWorkflow() {\n\ts.workflowEnv.OnActivity(emitWorkflowVersionMetricsActivity, mock.Anything).Return(nil).Times(1)\n\n\ts.workflowEnv.ExecuteWorkflow(esanalyzerWFTypeName, mock.Anything)\n\terr := s.workflowEnv.GetWorkflowResult(nil)\n\ts.NoError(err)\n}\n\nfunc (s *esanalyzerWorkflowTestSuite) TestEmitWorkflowVersionMetricsActivity() {\n\ts.config.ESAnalyzerWorkflowVersionDomains = dynamicproperties.GetStringPropertyFn(\n\t\tfmt.Sprintf(`[\"%s\"]`, s.DomainName),\n\t)\n\tesRaw := `\n{\n  \"took\": 642,\n  \"timed_out\": false,\n  \"_shards\": {\n    \"total\": 20,\n    \"successful\": 20,\n    \"skipped\": 0,\n    \"failed\": 0\n  },\n  \"hits\": {\n    \"total\": 4588,\n    \"max_score\": 0,\n    \"hits\": [\n\n    ]\n  },\n  \"aggregations\": {\n    \"wftypes\": {\n      \"doc_count_error_upper_bound\": 0,\n      \"sum_other_doc_count\": 0,\n      \"buckets\": [\n        {\n          \"key\": \"CourierUpdateWorkflow\",\n          \"doc_count\": 4539,\n          \"versions\": {\n            \"doc_count_error_upper_bound\": 0,\n            \"sum_other_doc_count\": 0,\n            \"buckets\": [\n\n            ]\n          }\n        },\n        {\n          \"key\": \"UpdateTipWorkflow\",\n          \"doc_count\": 33,\n          \"versions\": {\n            \"doc_count_error_upper_bound\": 0,\n            \"sum_other_doc_count\": 0,\n            \"buckets\": [\n              {\n                \"key\": \"update tip efp flow-1\",\n                \"doc_count\": 3\n              }\n            ]\n          }\n        },\n        {\n          \"key\": \"RoboCourierWorkflow\",\n          \"doc_count\": 12,\n          \"versions\": {\n            \"doc_count_error_upper_bound\": 0,\n            \"sum_other_doc_count\": 0,\n            \"buckets\": [\n\n            ]\n          }\n        },\n        {\n          \"key\": \"ImproveDropoffPinWorkflow\",\n          \"doc_count\": 2,\n          \"versions\": {\n            \"doc_count_error_upper_bound\": 0,\n            \"sum_other_doc_count\": 0,\n            \"buckets\": [\n\n            ]\n          }\n        },\n        {\n          \"key\": \"OrderStateUpdatedWorkflow\",\n          \"doc_count\": 2,\n          \"versions\": {\n            \"doc_count_error_upper_bound\": 0,\n            \"sum_other_doc_count\": 0,\n            \"buckets\": [\n\n            ]\n          }\n        }\n      ]\n    }\n  }\n}\n\t`\n\tvar rawEs elasticsearch.RawResponse\n\terr := json.Unmarshal([]byte(esRaw), &rawEs)\n\ts.NoError(err)\n\ts.mockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t&rawEs, nil).Times(1)\n\t_, err = s.activityEnv.ExecuteActivity(s.workflow.emitWorkflowVersionMetrics)\n\ts.NoError(err)\n\n}\n\nfunc (s *esanalyzerWorkflowTestSuite) TestEmitWorkflowTypeCountMetricsActivity() {\n\ts.config.ESAnalyzerWorkflowTypeDomains = dynamicproperties.GetStringPropertyFn(\n\t\tfmt.Sprintf(`[\"%s\"]`, s.DomainName),\n\t)\n\tesRaw := `\n{\n  \"took\": 642,\n  \"timed_out\": false,\n  \"_shards\": {\n    \"total\": 20,\n    \"successful\": 20,\n    \"skipped\": 0,\n    \"failed\": 0\n  },\n  \"hits\": {\n    \"total\": 4588,\n    \"max_score\": 0,\n    \"hits\": [\n\n    ]\n  },\n  \"aggregations\": {\n    \"wftypes\": {\n      \"doc_count_error_upper_bound\": 0,\n      \"sum_other_doc_count\": 0,\n      \"buckets\": [\n        {\n          \"key\": \"CourierUpdateWorkflow\",\n          \"doc_count\": 4539\n        },\n        {\n          \"key\": \"UpdateTipWorkflow\",\n          \"doc_count\": 33\n        },\n        {\n          \"key\": \"RoboCourierWorkflow\",\n          \"doc_count\": 12\n        },\n        {\n          \"key\": \"ImproveDropoffPinWorkflow\",\n          \"doc_count\": 2\n        },\n        {\n          \"key\": \"OrderStateUpdatedWorkflow\",\n          \"doc_count\": 2\n        }\n      ]\n    }\n  }\n}\n\t`\n\tvar rawEs elasticsearch.RawResponse\n\terr := json.Unmarshal([]byte(esRaw), &rawEs)\n\ts.NoError(err)\n\ts.mockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t&rawEs, nil).Times(1)\n\t_, err = s.activityEnv.ExecuteActivity(s.workflow.emitWorkflowTypeCountMetrics)\n\ts.NoError(err)\n\n}\n\nfunc TestNewAnalyzer(t *testing.T) {\n\tmockESConfig := &config.ElasticSearchConfig{\n\t\tIndices: map[string]string{\n\t\t\tconstants.VisibilityAppName: \"test\",\n\t\t},\n\t}\n\tmockPinotConfig := &config.PinotVisibilityConfig{\n\t\tTable: \"test\",\n\t}\n\n\tmockESClient := &esMocks.GenericClient{}\n\ttestAnalyzer1 := New(nil, nil, nil, mockESClient, nil, mockESConfig, mockPinotConfig, nil, nil, nil, nil, nil)\n\n\tmockPinotClient := &pinot.MockGenericClient{}\n\ttestAnalyzer2 := New(nil, nil, nil, nil, mockPinotClient, mockESConfig, mockPinotConfig, nil, nil, nil, nil, nil)\n\n\tassert.Equal(t, testAnalyzer1.readMode, ES)\n\tassert.Equal(t, testAnalyzer2.readMode, Pinot)\n}\n\nfunc TestEmitWorkflowTypeCountMetricsESErrorCases(t *testing.T) {\n\tmockESConfig := &config.ElasticSearchConfig{\n\t\tIndices: map[string]string{\n\t\t\tconstants.VisibilityAppName: \"test\",\n\t\t},\n\t}\n\tmockPinotConfig := &config.PinotVisibilityConfig{\n\t\tTable: \"test\",\n\t}\n\n\tctrl := gomock.NewController(t)\n\tmockESClient := &esMocks.GenericClient{}\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestAnalyzer := New(nil, nil, nil, mockESClient, nil, mockESConfig, mockPinotConfig, log.NewNoop(), tally.NoopScope, nil, mockDomainCache, nil)\n\ttestWorkflow := &Workflow{analyzer: testAnalyzer}\n\n\ttests := map[string]struct {\n\t\tdomainCacheAffordance func(mockDomainCache *cache.MockDomainCache)\n\t\tESClientAffordance    func(mockESClient *esMocks.GenericClient)\n\t\texpectedErr           error\n\t}{\n\t\t\"Case1: error getting domain\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {},\n\t\t\texpectedErr:        fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: error ES searchRaw\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\tnil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case3: foundAggregation is false\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&elasticsearch.RawResponse{\n\t\t\t\t\t\tAggregations: map[string]json.RawMessage{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"aggregation failed for domain in ES: test-domain\"),\n\t\t},\n\t\t\"Case4: error unmarshalling aggregation\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&elasticsearch.RawResponse{\n\t\t\t\t\t\tAggregations: map[string]json.RawMessage{\n\t\t\t\t\t\t\t\"wftypes\": []byte(\"invalid\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"invalid character 'i' looking for beginning of value\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\ttest.domainCacheAffordance(mockDomainCache)\n\t\t\ttest.ESClientAffordance(mockESClient)\n\n\t\t\terr := testWorkflow.emitWorkflowTypeCountMetricsES(context.Background(), \"test-domain\", zap.NewNop())\n\t\t\tif err == nil {\n\t\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestEmitWorkflowVersionMetricsESErrorCases(t *testing.T) {\n\tmockESConfig := &config.ElasticSearchConfig{\n\t\tIndices: map[string]string{\n\t\t\tconstants.VisibilityAppName: \"test\",\n\t\t},\n\t}\n\tmockPinotConfig := &config.PinotVisibilityConfig{\n\t\tTable: \"test\",\n\t}\n\n\tctrl := gomock.NewController(t)\n\tmockESClient := &esMocks.GenericClient{}\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestAnalyzer := New(nil, nil, nil, mockESClient, nil, mockESConfig, mockPinotConfig, log.NewNoop(), tally.NoopScope, nil, mockDomainCache, nil)\n\ttestWorkflow := &Workflow{analyzer: testAnalyzer}\n\n\ttests := map[string]struct {\n\t\tdomainCacheAffordance func(mockDomainCache *cache.MockDomainCache)\n\t\tESClientAffordance    func(mockESClient *esMocks.GenericClient)\n\t\texpectedErr           error\n\t}{\n\t\t\"Case1: error getting domain\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {},\n\t\t\texpectedErr:        fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case2: error ES searchRaw\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\tnil, fmt.Errorf(\"error\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error\"),\n\t\t},\n\t\t\"Case3: foundAggregation is false\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&elasticsearch.RawResponse{\n\t\t\t\t\t\tAggregations: map[string]json.RawMessage{},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"aggregation failed for domain in ES: test-domain\"),\n\t\t},\n\t\t\"Case4: error unmarshalling aggregation\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tESClientAffordance: func(mockESClient *esMocks.GenericClient) {\n\t\t\t\tmockESClient.On(\"SearchRaw\", mock.Anything, mock.Anything, mock.Anything).Return(\n\t\t\t\t\t&elasticsearch.RawResponse{\n\t\t\t\t\t\tAggregations: map[string]json.RawMessage{\n\t\t\t\t\t\t\t\"wftypes\": []byte(\"invalid\"),\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"invalid character 'i' looking for beginning of value\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\ttest.domainCacheAffordance(mockDomainCache)\n\t\t\ttest.ESClientAffordance(mockESClient)\n\n\t\t\terr := testWorkflow.emitWorkflowVersionMetricsES(context.Background(), \"test-domain\", zap.NewNop())\n\t\t\tif err == nil {\n\t\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestEmitWorkflowTypeCountMetricsPinot(t *testing.T) {\n\tmockPinotConfig := &config.PinotVisibilityConfig{\n\t\tTable: \"test\",\n\t}\n\tmockESConfig := &config.ElasticSearchConfig{\n\t\tIndices: map[string]string{\n\t\t\tconstants.VisibilityAppName: \"test\",\n\t\t},\n\t}\n\n\tctrl := gomock.NewController(t)\n\n\tmockPinotClient := pinot.NewMockGenericClient(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestAnalyzer := New(nil, nil, nil, nil, mockPinotClient, mockESConfig, mockPinotConfig, log.NewNoop(), tally.NoopScope, nil, mockDomainCache, nil)\n\ttestWorkflow := &Workflow{analyzer: testAnalyzer}\n\n\ttests := map[string]struct {\n\t\tdomainCacheAffordance func(mockDomainCache *cache.MockDomainCache)\n\t\tPinotClientAffordance func(mockPinotClient *pinot.MockGenericClient)\n\t\texpectedErr           error\n\t}{\n\t\t\"Case0: success\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test0\", float64(1)},\n\t\t\t\t\t{\"test1\", float64(2)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t\"Case1: error getting domain\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, fmt.Errorf(\"domain error\")).Times(1)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {},\n\t\t\texpectedErr:           fmt.Errorf(\"domain error\"),\n\t\t},\n\t\t\"Case2: error Pinot query\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return(nil, fmt.Errorf(\"pinot error\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"pinot error\"),\n\t\t},\n\t\t\"Case3: Aggregation is empty\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"aggregation failed for domain in Pinot: test-domain\"),\n\t\t},\n\t\t\"Case4: error parsing workflow count\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test\", \"invalid\"},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error parsing workflow count for workflow type test\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\ttest.domainCacheAffordance(mockDomainCache)\n\t\t\ttest.PinotClientAffordance(mockPinotClient)\n\n\t\t\terr := testWorkflow.emitWorkflowTypeCountMetricsPinot(\"test-domain\", zap.NewNop())\n\t\t\tif err == nil {\n\t\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestEmitWorkflowVersionMetricsPinot(t *testing.T) {\n\tmockPinotConfig := &config.PinotVisibilityConfig{\n\t\tTable: \"test\",\n\t}\n\tmockESConfig := &config.ElasticSearchConfig{\n\t\tIndices: map[string]string{\n\t\t\tconstants.VisibilityAppName: \"test\",\n\t\t},\n\t}\n\n\tctrl := gomock.NewController(t)\n\n\tmockPinotClient := pinot.NewMockGenericClient(ctrl)\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\ttestAnalyzer := New(nil, nil, nil, nil, mockPinotClient, mockESConfig, mockPinotConfig, log.NewNoop(), tally.NoopScope, nil, mockDomainCache, nil)\n\ttestWorkflow := &Workflow{analyzer: testAnalyzer}\n\n\ttests := map[string]struct {\n\t\tdomainCacheAffordance func(mockDomainCache *cache.MockDomainCache)\n\t\tPinotClientAffordance func(mockPinotClient *pinot.MockGenericClient)\n\t\texpectedErr           error\n\t}{\n\t\t\"Case0: success\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil).Times(3)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-type0\", float64(100)},\n\t\t\t\t\t{\"test-wf-type1\", float64(200)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-version0\", float64(10)},\n\t\t\t\t\t{\"test-wf-version1\", float64(20)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-version3\", float64(10)},\n\t\t\t\t\t{\"test-wf-version4\", float64(2)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t\"Case1: error getting domain\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, fmt.Errorf(\"domain error\")).Times(1)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {},\n\t\t\texpectedErr:           fmt.Errorf(\"domain error\"),\n\t\t},\n\t\t\"Case2: error Pinot query\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return(nil, fmt.Errorf(\"pinot error\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"failed to query Pinot to find workflow type count Info: test-domain, error: pinot error\"),\n\t\t},\n\t\t\"Case3: Aggregation is empty\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"aggregation failed for domain in Pinot: test-domain\"),\n\t\t},\n\t\t\"Case4: error parsing workflow count\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test\", \"invalid\"},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error parsing workflow count for workflow type test\"),\n\t\t},\n\t\t\"Case5-1: failure case in queryWorkflowVersionsWithType: getWorkflowVersionPinotQuery error\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil).Times(1)\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(nil, fmt.Errorf(\"domain error\")).Times(1)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-type0\", float64(100)},\n\t\t\t\t\t{\"test-wf-type1\", float64(200)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error querying workflow versions for workflow type: test-wf-type0: error: domain error\"),\n\t\t},\n\t\t\"Case5-2: failure case in queryWorkflowVersionsWithType: SearchAggr error\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil).Times(2)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-type0\", float64(100)},\n\t\t\t\t\t{\"test-wf-type1\", float64(200)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return(nil, fmt.Errorf(\"pinot error\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error querying workflow versions for workflow type: test-wf-type0: error: pinot error\"),\n\t\t},\n\t\t\"Case5-3: failure case in queryWorkflowVersionsWithType: error parsing workflow count\": {\n\t\t\tdomainCacheAffordance: func(mockDomainCache *cache.MockDomainCache) {\n\t\t\t\tmockDomainCache.EXPECT().GetDomain(gomock.Any()).Return(cache.NewDomainCacheEntryForTest(\n\t\t\t\t\t&persistence.DomainInfo{ID: \"test-id\"}, nil, false, nil, 0, nil, 0, 0, 0), nil).Times(2)\n\t\t\t},\n\t\t\tPinotClientAffordance: func(mockPinotClient *pinot.MockGenericClient) {\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-type0\", float64(100)},\n\t\t\t\t\t{\"test-wf-type1\", float64(200)},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockPinotClient.EXPECT().SearchAggr(gomock.Any()).Return([][]interface{}{\n\t\t\t\t\t{\"test-wf-version0\", float64(3.14)},\n\t\t\t\t\t{\"test-wf-version1\", 20},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedErr: fmt.Errorf(\"error querying workflow versions for workflow type: \" +\n\t\t\t\t\"test-wf-type0: error: error parsing workflow count for cadence version test-wf-version1\"),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\t// Set up mocks\n\t\t\ttest.domainCacheAffordance(mockDomainCache)\n\t\t\ttest.PinotClientAffordance(mockPinotClient)\n\n\t\t\terr := testWorkflow.emitWorkflowVersionMetricsPinot(\"test-domain\", zap.NewNop())\n\t\t\tif err == nil {\n\t\t\t\tassert.Equal(t, test.expectedErr, err)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, test.expectedErr.Error(), err.Error())\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/esanalyzer/domainWorkflowTypeCountWorkflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esanalyzer\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/pinot\"\n)\n\nconst (\n\tworkflowTypeCountMetrics = \"workflow_type_count\"\n\n\t// workflow constants\n\tdomainWFTypeCountWorkflowID                = \"cadence-sys-tl-esanalyzer-domain-wf-type-count\"\n\tdomainWFTypeCountWorkflowTypeName          = \"cadence-sys-es-analyzer-domain-wf-type-count-workflow\"\n\temitDomainWorkflowTypeCountMetricsActivity = \"cadence-sys-es-analyzer-emit-domain-workflow-type-count-metrics\"\n)\n\ntype (\n\tDomainWorkflowTypeCount struct {\n\t\tWorkflowTypes []EsAggregateCount `json:\"buckets\"`\n\t}\n)\n\nvar (\n\tworkflowActivityOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    5 * time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    10 * time.Second,\n\t\t\tBackoffCoefficient: 1.7,\n\t\t\tMaximumInterval:    5 * time.Minute,\n\t\t\tExpirationInterval: 10 * time.Minute,\n\t\t},\n\t}\n\n\tdomainWfTypeCountStartOptions = cclient.StartWorkflowOptions{\n\t\tID:                           domainWFTypeCountWorkflowID,\n\t\tTaskList:                     taskListName,\n\t\tExecutionStartToCloseTimeout: 5 * time.Minute,\n\t\tCronSchedule:                 \"*/5 * * * *\",\n\t}\n)\n\nfunc initDomainWorkflowTypeCountWorkflow(a *Analyzer) {\n\tw := Workflow{analyzer: a}\n\tworkflow.RegisterWithOptions(w.emitWorkflowTypeCount, workflow.RegisterOptions{Name: domainWFTypeCountWorkflowTypeName})\n\tactivity.RegisterWithOptions(\n\t\tw.emitWorkflowTypeCountMetrics,\n\t\tactivity.RegisterOptions{Name: emitDomainWorkflowTypeCountMetricsActivity},\n\t)\n}\n\n// emitWorkflowTypeCount queries ElasticSearch for workflow count per type and emit metrics\nfunc (w *Workflow) emitWorkflowTypeCount(ctx workflow.Context) error {\n\tif w.analyzer.config.ESAnalyzerPause() {\n\t\tlogger := workflow.GetLogger(ctx)\n\t\tlogger.Info(\"Skipping ESAnalyzer execution cycle since it was paused\")\n\t\treturn nil\n\t}\n\tvar err error\n\terr = workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, workflowActivityOptions),\n\t\temitDomainWorkflowTypeCountMetricsActivity,\n\t).Get(ctx, &err)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\n// get open workflows count per workflow type for specified domain\nfunc (w *Workflow) getDomainWorkflowTypeCountQuery(domainName string) (string, error) {\n\tdomain, err := w.analyzer.domainCache.GetDomain(domainName)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// exclude uninitialized workflow executions by checking whether record has start time field\n\treturn fmt.Sprintf(`\n{\n    \"aggs\" : {\n        \"wftypes\" : {\n            \"terms\" : { \"field\" : \"WorkflowType\"}\n        }\n    },\n    \"query\": {\n        \"bool\": {\n            \"must_not\": {\n                \"exists\": {\n                    \"field\": \"CloseTime\"\n                }\n            },\n            \"must\": [\n                {\n                    \"match\" : {\n                        \"DomainID\" : \"%s\"\n                    }\n                },\n\t\t\t\t{\n\t\t\t\t\t\"exists\": {\n\t\t\t\t\t\t\"field\": \"StartTime\"\n\t\t\t\t\t}\n\t\t\t\t}\n            ]\n        }\n    },\n    \"size\": 0\n}\n    `, domain.GetInfo().ID), nil\n}\n\n// emitWorkflowTypeCountMetrics is an activity that emits the running workflow type counts of a domain\n// it will switch between ES and Pinot based on the readMode\nfunc (w *Workflow) emitWorkflowTypeCountMetrics(ctx context.Context) error {\n\tlogger := activity.GetLogger(ctx)\n\tvar workflowMetricDomainNames []string\n\tworkflowMetricDomains := w.analyzer.config.ESAnalyzerWorkflowTypeDomains()\n\tif len(workflowMetricDomains) > 0 {\n\t\terr := json.Unmarshal([]byte(workflowMetricDomains), &workflowMetricDomainNames)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tvar failedDomains []string\n\t\tfor _, domainName := range workflowMetricDomainNames {\n\t\t\tswitch w.analyzer.readMode {\n\t\t\tcase ES:\n\t\t\t\terr = w.emitWorkflowTypeCountMetricsES(ctx, domainName, logger)\n\t\t\tcase Pinot:\n\t\t\t\terr = w.emitWorkflowTypeCountMetricsPinot(domainName, logger)\n\t\t\tdefault:\n\t\t\t\terr = w.emitWorkflowTypeCountMetricsES(ctx, domainName, logger)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\tlogger.Error(fmt.Sprintf(\"failed to emit workflow type metrics for domain %s\", domainName), zap.Error(err))\n\t\t\t\tfailedDomains = append(failedDomains, domainName)\n\t\t\t}\n\t\t}\n\t\tif len(failedDomains) == len(workflowMetricDomainNames) {\n\t\t\treturn fmt.Errorf(\"failed to emit workflow type metrics for all domains\")\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (w *Workflow) getDomainWorkflowTypeCountPinotQuery(domainName string) (string, error) {\n\tdomain, err := w.analyzer.domainCache.GetDomain(domainName)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// exclude uninitialized workflow executions by checking whether record has start time field\n\t// there's a \"LIMIT 10\" because in ES, Aggr clause by default returns the top 10 results\n\treturn fmt.Sprintf(`\nSELECT WorkflowType, COUNT(*) AS count\nFROM %s\nWHERE DomainID = '%s'\nAND CloseStatus = -1\nAND StartTime > 0\nGROUP BY WorkflowType\nORDER BY count\nLIMIT 10\n    `, w.analyzer.pinotTableName, domain.GetInfo().ID), nil\n}\n\nfunc (w *Workflow) emitWorkflowTypeCountMetricsPinot(domainName string, logger *zap.Logger) error {\n\twfTypeCountPinotQuery, err := w.getDomainWorkflowTypeCountPinotQuery(domainName)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get Pinot query to find domain workflow type Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tresponse, err := w.analyzer.pinotClient.SearchAggr(&pinot.SearchRequest{Query: wfTypeCountPinotQuery})\n\tif err != nil {\n\t\tlogger.Error(\"Failed to query Pinot to find workflow type count Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"VisibilityQuery\", wfTypeCountPinotQuery),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tfoundAggregation := len(response) > 0\n\n\tif !foundAggregation {\n\t\tlogger.Error(\"Pinot error: aggregation failed.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", fmt.Sprintf(\"%v\", response)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfTypeCountPinotQuery),\n\t\t)\n\t\treturn fmt.Errorf(\"aggregation failed for domain in Pinot: %s\", domainName)\n\t}\n\tvar domainWorkflowTypeCount DomainWorkflowTypeCount\n\tfor _, row := range response {\n\t\tworkflowType := row[0].(string)\n\n\t\t// even though the count is a int, it is returned as a float64, need to pay attention to this\n\t\tworkflowCount, ok := row[1].(float64)\n\t\tif !ok {\n\t\t\tlogger.Error(\"Error parsing workflow count\",\n\t\t\t\tzap.Error(err),\n\t\t\t\tzap.String(\"WorkflowType\", workflowType),\n\t\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\t\tzap.Float64(\"WorkflowCount\", workflowCount),\n\t\t\t\tzap.String(\"WorkflowCountType\", fmt.Sprintf(\"%T\", row[1])),\n\t\t\t\tzap.String(\"raw data\", fmt.Sprintf(\"%#v\", response)),\n\t\t\t)\n\t\t\treturn fmt.Errorf(\"error parsing workflow count for workflow type %s\", workflowType)\n\t\t}\n\t\tdomainWorkflowTypeCount.WorkflowTypes = append(domainWorkflowTypeCount.WorkflowTypes, EsAggregateCount{\n\t\t\tAggregateKey:   workflowType,\n\t\t\tAggregateCount: int64(workflowCount),\n\t\t})\n\t}\n\tfor _, workflowType := range domainWorkflowTypeCount.WorkflowTypes {\n\t\tw.analyzer.tallyScope.Tagged(\n\t\t\tmap[string]string{domainTag: domainName, workflowTypeTag: workflowType.AggregateKey},\n\t\t).Gauge(workflowTypeCountMetrics).Update(float64(workflowType.AggregateCount))\n\t}\n\treturn nil\n}\n\nfunc (w *Workflow) emitWorkflowTypeCountMetricsES(ctx context.Context, domainName string, logger *zap.Logger) error {\n\twfTypeCountEsQuery, err := w.getDomainWorkflowTypeCountQuery(domainName)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get ElasticSearch query to find domain workflow type Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tresponse, err := w.analyzer.esClient.SearchRaw(ctx, w.analyzer.visibilityIndexName, wfTypeCountEsQuery)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to query ElasticSearch to find workflow type count Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"VisibilityQuery\", wfTypeCountEsQuery),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tagg, foundAggregation := response.Aggregations[workflowTypesAggKey]\n\n\tif !foundAggregation {\n\t\tlogger.Error(\"ElasticSearch error: aggregation failed.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", string(agg)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfTypeCountEsQuery),\n\t\t)\n\t\treturn fmt.Errorf(\"aggregation failed for domain in ES: %s\", domainName)\n\t}\n\tvar domainWorkflowTypeCount DomainWorkflowTypeCount\n\terr = json.Unmarshal(agg, &domainWorkflowTypeCount)\n\tif err != nil {\n\t\tlogger.Error(\"ElasticSearch error parsing aggregation.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", string(agg)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfTypeCountEsQuery),\n\t\t)\n\t\treturn err\n\t}\n\tfor _, workflowType := range domainWorkflowTypeCount.WorkflowTypes {\n\t\tw.analyzer.tallyScope.Tagged(\n\t\t\tmap[string]string{domainTag: domainName, workflowTypeTag: workflowType.AggregateKey},\n\t\t).Gauge(workflowTypeCountMetrics).Update(float64(workflowType.AggregateCount))\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/esanalyzer/workflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage esanalyzer\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/pinot\"\n)\n\nconst (\n\tworkflowTypesAggKey         = \"wftypes\"\n\tdomainTag                   = \"domain\"\n\tworkflowVersionTag          = \"workflowVersion\"\n\tworkflowVersionCountMetrics = \"workflow_version_count\"\n\tworkflowTypeTag             = \"workflowType\"\n\n\t// workflow constants\n\tesAnalyzerWFID                     = \"cadence-sys-tl-esanalyzer\"\n\tesanalyzerWFTypeName               = \"cadence-sys-es-analyzer-workflow\"\n\temitWorkflowVersionMetricsActivity = \"cadence-sys-es-analyzer-emit-workflow-version-metrics\"\n)\n\ntype (\n\tDomainWorkflowVersionCount struct {\n\t\tWorkflowTypes []WorkflowTypeCount `json:\"buckets\"`\n\t}\n\tWorkflowTypeCount struct {\n\t\tEsAggregateCount\n\t\tWorkflowVersions WorkflowVersionCount `json:\"versions\"`\n\t}\n\tWorkflowVersionCount struct {\n\t\tWorkflowVersions []EsAggregateCount `json:\"buckets\"`\n\t}\n)\n\nvar (\n\tretryPolicy = cadence.RetryPolicy{\n\t\tInitialInterval:    10 * time.Second,\n\t\tBackoffCoefficient: 1.7,\n\t\tMaximumInterval:    5 * time.Minute,\n\t\tExpirationInterval: 10 * time.Minute,\n\t}\n\n\tgetWorkflowMetricsOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    5 * time.Minute,\n\t\tRetryPolicy:            &retryPolicy,\n\t}\n\n\twfOptions = cclient.StartWorkflowOptions{\n\t\tID:                           esAnalyzerWFID,\n\t\tTaskList:                     taskListName,\n\t\tExecutionStartToCloseTimeout: 10 * time.Minute,\n\t\tCronSchedule:                 \"*/10 * * * *\",\n\t}\n)\n\nfunc initWorkflow(a *Analyzer) {\n\tw := Workflow{analyzer: a}\n\tworkflow.RegisterWithOptions(w.workflowFunc, workflow.RegisterOptions{Name: esanalyzerWFTypeName})\n\tactivity.RegisterWithOptions(\n\t\tw.emitWorkflowVersionMetrics,\n\t\tactivity.RegisterOptions{Name: emitWorkflowVersionMetricsActivity},\n\t)\n}\n\n// workflowFunc queries ElasticSearch for information and do something with it\nfunc (w *Workflow) workflowFunc(ctx workflow.Context) error {\n\tif w.analyzer.config.ESAnalyzerPause() {\n\t\tlogger := workflow.GetLogger(ctx)\n\t\tlogger.Info(\"Skipping ESAnalyzer execution cycle since it was paused\")\n\t\treturn nil\n\t}\n\tvar err error\n\terr = workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, getWorkflowMetricsOptions),\n\t\temitWorkflowVersionMetricsActivity,\n\t).Get(ctx, &err)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc (w *Workflow) getWorkflowVersionQuery(domainName string) (string, error) {\n\tdomain, err := w.analyzer.domainCache.GetDomain(domainName)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(`\n{\n    \"aggs\" : {\n        \"wftypes\" : {\n            \"terms\" : { \"field\" : \"WorkflowType\"},\n            \"aggs\": {\n                \"versions\": {\n                    \"terms\" : { \"field\" : \"Attr.CadenceChangeVersion\"}\n                }\n            }\n        }\n    },\n    \"query\": {\n        \"bool\": {\n            \"must_not\": {\n                \"exists\": {\n                    \"field\": \"CloseTime\"\n                }\n            },\n            \"must\": [\n                {\n                    \"match\" : {\n                        \"DomainID\" : \"%s\"\n                    }\n                }\n            ]\n        }\n    },\n    \"size\": 0\n}\n    `, domain.GetInfo().ID), nil\n}\n\nfunc (w *Workflow) getWorkflowTypePinotQuery(domainName string) (string, error) {\n\tdomain, err := w.analyzer.domainCache.GetDomain(domainName)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// exclude uninitialized workflow executions by checking whether record has start time field\n\t// there's a \"LIMIT 10\" because in ES, Aggr clause by default returns the top 10 results\n\treturn fmt.Sprintf(`\nSELECT WorkflowType, COUNT(*) AS count\nFROM %s\nWHERE DomainID = '%s'\n  AND CloseStatus = -1\n  AND StartTime > 0\nGROUP BY WorkflowType\nORDER BY count DESC\nLIMIT 10\n    `, w.analyzer.pinotTableName, domain.GetInfo().ID), nil\n}\n\nfunc (w *Workflow) getWorkflowVersionPinotQuery(domainName string, wfType string) (string, error) {\n\tdomain, err := w.analyzer.domainCache.GetDomain(domainName)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\t// exclude uninitialized workflow executions by checking whether record has start time field\n\t// there's a \"LIMIT 10\" because in ES, Aggr clause by default returns the top 10 results\n\treturn fmt.Sprintf(`\nSELECT JSON_EXTRACT_SCALAR(Attr, '$.CadenceChangeVersion', 'STRING_ARRAY') AS CadenceChangeVersion, COUNT(*) AS count\nFROM %s\nWHERE DomainID = '%s'\n  AND CloseStatus = -1\n  AND StartTime > 0\n  AND WorkflowType = '%s'\nGROUP BY JSON_EXTRACT_SCALAR(Attr, '$.CadenceChangeVersion', 'STRING_ARRAY')\nORDER BY count DESC\nLIMIT 10\n    `, w.analyzer.pinotTableName, domain.GetInfo().ID, wfType), nil\n}\n\n// emitWorkflowVersionMetrics is an activity that emits the running WF versions of a domain\nfunc (w *Workflow) emitWorkflowVersionMetrics(ctx context.Context) error {\n\tlogger := activity.GetLogger(ctx)\n\tvar workflowMetricDomainNames []string\n\tworkflowMetricDomains := w.analyzer.config.ESAnalyzerWorkflowVersionDomains()\n\tif len(workflowMetricDomains) > 0 {\n\t\terr := json.Unmarshal([]byte(workflowMetricDomains), &workflowMetricDomainNames)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tvar failedDomains []string\n\t\tfor _, domainName := range workflowMetricDomainNames {\n\t\t\tswitch w.analyzer.readMode {\n\t\t\tcase ES:\n\t\t\t\terr = w.emitWorkflowVersionMetricsES(ctx, domainName, logger)\n\t\t\tcase Pinot:\n\t\t\t\terr = w.emitWorkflowVersionMetricsPinot(domainName, logger)\n\t\t\tdefault:\n\t\t\t\terr = w.emitWorkflowVersionMetricsES(ctx, domainName, logger)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\tlogger.Error(fmt.Sprintf(\"failed to emit workflow version metrics for domain %s\", domainName), zap.Error(err))\n\t\t\t\tfailedDomains = append(failedDomains, domainName)\n\t\t\t}\n\t\t}\n\t\tif len(failedDomains) == len(workflowMetricDomainNames) {\n\t\t\treturn fmt.Errorf(\"failed to emit workflow version metrics for all domains\")\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (w *Workflow) emitWorkflowVersionMetricsPinot(domainName string, logger *zap.Logger) error {\n\twfVersionPinotQuery, err := w.getWorkflowTypePinotQuery(domainName)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get Pinot query to find workflow type Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tresponse, err := w.analyzer.pinotClient.SearchAggr(&pinot.SearchRequest{Query: wfVersionPinotQuery})\n\tif err != nil {\n\t\tlogger.Error(\"Failed to query Pinot to find workflow type count Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionPinotQuery),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn fmt.Errorf(\"failed to query Pinot to find workflow type count Info: %s, error: %s\", domainName, err.Error())\n\t}\n\tfoundAggregation := len(response) > 0\n\n\tif !foundAggregation {\n\t\tlogger.Error(\"Pinot error: aggregation failed.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", fmt.Sprintf(\"%v\", response)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionPinotQuery),\n\t\t)\n\t\treturn fmt.Errorf(\"aggregation failed for domain in Pinot: %s\", domainName)\n\t}\n\tvar domainWorkflowVersionCount DomainWorkflowVersionCount\n\tfor _, row := range response {\n\t\tworkflowType := row[0].(string)\n\t\tworkflowCount, ok := row[1].(float64)\n\t\tif !ok {\n\t\t\tlogger.Error(\"error parsing workflow count for cadence version\",\n\t\t\t\tzap.Error(err),\n\t\t\t\tzap.String(\"WorkflowType\", workflowType),\n\t\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\t\tzap.Float64(\"WorkflowCount\", workflowCount),\n\t\t\t\tzap.String(\"WorkflowCountType\", fmt.Sprintf(\"%T\", row[1])),\n\t\t\t\tzap.String(\"raw data\", fmt.Sprintf(\"%#v\", response)),\n\t\t\t)\n\t\t\treturn fmt.Errorf(\"error parsing workflow count for workflow type %s\", workflowType)\n\t\t}\n\t\tworkflowVersions, err := w.queryWorkflowVersionsWithType(domainName, workflowType, logger)\n\n\t\tif err != nil {\n\t\t\tlogger.Error(\"Error querying workflow versions\",\n\t\t\t\tzap.Error(err),\n\t\t\t\tzap.String(\"WorkflowType\", workflowType),\n\t\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\t)\n\t\t\treturn fmt.Errorf(\"error querying workflow versions for workflow type: %s: error: %s\", workflowType, err.Error())\n\t\t}\n\n\t\tdomainWorkflowVersionCount.WorkflowTypes = append(domainWorkflowVersionCount.WorkflowTypes, WorkflowTypeCount{\n\t\t\tEsAggregateCount: EsAggregateCount{\n\t\t\t\tAggregateKey:   workflowType,\n\t\t\t\tAggregateCount: int64(workflowCount),\n\t\t\t},\n\t\t\tWorkflowVersions: workflowVersions,\n\t\t})\n\t}\n\n\tfor _, workflowType := range domainWorkflowVersionCount.WorkflowTypes {\n\t\tfor _, workflowVersion := range workflowType.WorkflowVersions.WorkflowVersions {\n\t\t\tw.analyzer.tallyScope.Tagged(\n\t\t\t\tmap[string]string{domainTag: domainName, workflowVersionTag: workflowVersion.AggregateKey, workflowTypeTag: workflowType.AggregateKey},\n\t\t\t).Gauge(workflowVersionCountMetrics).Update(float64(workflowVersion.AggregateCount))\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (w *Workflow) queryWorkflowVersionsWithType(domainName string, wfType string, logger *zap.Logger) (WorkflowVersionCount, error) {\n\twfVersionPinotQuery, err := w.getWorkflowVersionPinotQuery(domainName, wfType)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get Pinot query to find workflow version Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn WorkflowVersionCount{}, err\n\t}\n\n\tresponse, err := w.analyzer.pinotClient.SearchAggr(&pinot.SearchRequest{Query: wfVersionPinotQuery})\n\tif err != nil {\n\t\tlogger.Error(\"Failed to query Pinot to find workflow type count Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionPinotQuery),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn WorkflowVersionCount{}, err\n\t}\n\tfoundAggregation := len(response) > 0\n\n\t// if no CadenceChangeVersion is found, return an empty WorkflowVersionCount, no errors\n\tif !foundAggregation {\n\t\treturn WorkflowVersionCount{}, nil\n\t}\n\n\tvar workflowVersions WorkflowVersionCount\n\tfor _, row := range response {\n\t\tworkflowVersion := row[0].(string)\n\t\tworkflowCount, ok := row[1].(float64)\n\t\tif !ok {\n\t\t\tlogger.Error(\"error parsing workflow count for cadence version\",\n\t\t\t\tzap.Error(err),\n\t\t\t\tzap.String(\"WorkflowVersion\", workflowVersion),\n\t\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\t\tzap.Float64(\"WorkflowCount\", workflowCount),\n\t\t\t\tzap.String(\"WorkflowCountType\", fmt.Sprintf(\"%T\", row[1])),\n\t\t\t\tzap.String(\"raw data\", fmt.Sprintf(\"%#v\", response)),\n\t\t\t)\n\t\t\treturn WorkflowVersionCount{}, fmt.Errorf(\"error parsing workflow count for cadence version %s\", workflowVersion)\n\t\t}\n\t\tworkflowVersions.WorkflowVersions = append(workflowVersions.WorkflowVersions, EsAggregateCount{\n\t\t\tAggregateKey:   workflowVersion,\n\t\t\tAggregateCount: int64(workflowCount),\n\t\t})\n\t}\n\treturn workflowVersions, nil\n}\n\nfunc (w *Workflow) emitWorkflowVersionMetricsES(ctx context.Context, domainName string, logger *zap.Logger) error {\n\twfVersionEsQuery, err := w.getWorkflowVersionQuery(domainName)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to get ElasticSearch query to find workflow version Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tresponse, err := w.analyzer.esClient.SearchRaw(ctx, w.analyzer.visibilityIndexName, wfVersionEsQuery)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to query ElasticSearch to find workflow version Info\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionEsQuery),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t)\n\t\treturn err\n\t}\n\tagg, foundAggregation := response.Aggregations[workflowTypesAggKey]\n\n\tif !foundAggregation {\n\t\tlogger.Error(\"ElasticSearch error: aggregation failed.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", string(agg)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionEsQuery),\n\t\t)\n\t\treturn fmt.Errorf(\"aggregation failed for domain in ES: %s\", domainName)\n\t}\n\tvar domainWorkflowVersionCount DomainWorkflowVersionCount\n\terr = json.Unmarshal(agg, &domainWorkflowVersionCount)\n\tif err != nil {\n\t\tlogger.Error(\"ElasticSearch error parsing aggregation.\",\n\t\t\tzap.Error(err),\n\t\t\tzap.String(\"Aggregation\", string(agg)),\n\t\t\tzap.String(\"DomainName\", domainName),\n\t\t\tzap.String(\"VisibilityQuery\", wfVersionEsQuery),\n\t\t)\n\t\treturn err\n\t}\n\tfor _, workflowType := range domainWorkflowVersionCount.WorkflowTypes {\n\t\tfor _, workflowVersion := range workflowType.WorkflowVersions.WorkflowVersions {\n\t\t\tw.analyzer.tallyScope.Tagged(\n\t\t\t\tmap[string]string{domainTag: domainName, workflowVersionTag: workflowVersion.AggregateKey, workflowTypeTag: workflowType.AggregateKey},\n\t\t\t).Gauge(workflowVersionCountMetrics).Update(float64(workflowVersion.AggregateCount))\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/failovermanager/rebalance_workflow.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage failovermanager\n\nimport (\n\t\"context\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// RebalanceParams contains parameters for rebalance workflow\n\tRebalanceParams struct {\n\t\t// BatchFailoverSize is number of domains to failover in one batch\n\t\tBatchFailoverSize int\n\t\t// BatchFailoverWaitTimeInSeconds is the waiting time between batch failover\n\t\tBatchFailoverWaitTimeInSeconds int\n\t}\n\n\t// RebalanceResult contains the result from the rebalance workflow\n\tRebalanceResult struct {\n\t\tSuccessDomains []string\n\t\tFailedDomains  []string\n\t}\n\n\t// DomainRebalanceData contains the result from getRebalanceDomains activity\n\tDomainRebalanceData struct {\n\t\tDomainName       string\n\t\tPreferredCluster string\n\t}\n)\n\n// RebalanceWorkflow is to rebalance domains across clusters based on rebalance policy.\nfunc RebalanceWorkflow(ctx workflow.Context, params *RebalanceParams) (*RebalanceResult, error) {\n\t// get rebalance domains\n\tao := workflow.WithActivityOptions(ctx, getGetDomainsActivityOptions())\n\tvar domainData []*DomainRebalanceData\n\terr := workflow.ExecuteActivity(ao, getRebalanceDomainsActivityName).Get(ctx, &domainData)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomainPerCluster := getDomainsByCluster(domainData)\n\n\tresult := &RebalanceResult{\n\t\tSuccessDomains: []string{},\n\t\tFailedDomains:  []string{},\n\t}\n\t// failover domains for rebalance\n\tfor cluster, domains := range domainPerCluster {\n\t\tfailoverParams := &FailoverParams{\n\t\t\tTargetCluster:                  cluster,\n\t\t\tBatchFailoverSize:              params.BatchFailoverSize,\n\t\t\tBatchFailoverWaitTimeInSeconds: params.BatchFailoverWaitTimeInSeconds,\n\t\t}\n\t\tsuccessDomains, failedDomains := failoverDomainsByBatch(\n\t\t\tctx,\n\t\t\tdomains,\n\t\t\tfailoverParams,\n\t\t\tfunc() {},\n\t\t\tfalse,\n\t\t)\n\t\tresult.SuccessDomains = append(result.SuccessDomains, successDomains...)\n\t\tresult.FailedDomains = append(result.FailedDomains, failedDomains...)\n\t}\n\n\treturn result, nil\n}\n\n// GetDomainsForRebalanceActivity activity fetch domains for rebalance\nfunc GetDomainsForRebalanceActivity(ctx context.Context) ([]*DomainRebalanceData, error) {\n\tdomains, err := getAllDomains(ctx, nil)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar res []*DomainRebalanceData\n\tfor _, domain := range domains {\n\t\tif shouldAllowRebalance(domain) {\n\t\t\tdomainName := domain.GetDomainInfo().GetName()\n\t\t\tres = append(res, &DomainRebalanceData{\n\t\t\t\tDomainName:       domainName,\n\t\t\t\tPreferredCluster: getPreferredClusterName(domain),\n\t\t\t})\n\t\t}\n\t}\n\treturn res, nil\n}\n\nfunc getPreferredClusterName(domain *types.DescribeDomainResponse) string {\n\treturn domain.GetDomainInfo().GetData()[constants.DomainDataKeyForPreferredCluster]\n}\n\nfunc shouldAllowRebalance(domain *types.DescribeDomainResponse) bool {\n\tpreferredCluster := getPreferredClusterName(domain)\n\treturn isDomainFailoverManagedByCadence(domain) &&\n\t\tdomain.IsGlobalDomain &&\n\t\tdomain.GetDomainInfo().GetStatus() == types.DomainStatusRegistered &&\n\t\tlen(preferredCluster) != 0 &&\n\t\tpreferredCluster != domain.ReplicationConfiguration.GetActiveClusterName() &&\n\t\tisPreferredClusterInClusterListForDomain(preferredCluster, domain)\n}\n\nfunc getDomainsByCluster(domainData []*DomainRebalanceData) map[string][]string {\n\tdomainPerCluster := make(map[string][]string)\n\tfor _, domain := range domainData {\n\t\tdomainPerCluster[domain.PreferredCluster] = append(domainPerCluster[domain.PreferredCluster], domain.DomainName)\n\t}\n\n\treturn domainPerCluster\n}\n\nfunc isPreferredClusterInClusterListForDomain(preferredCluster string, domain *types.DescribeDomainResponse) bool {\n\tfor _, cluster := range domain.ReplicationConfiguration.Clusters {\n\t\tif cluster.ClusterName == preferredCluster {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "service/worker/failovermanager/rebalance_workflow_test.go",
    "content": "// Copyright (c) 2017-2021 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage failovermanager\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype rebalanceWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tactivityEnv *testsuite.TestActivityEnvironment\n\tworkflowEnv *testsuite.TestWorkflowEnvironment\n}\n\nfunc TestRebalanceWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(rebalanceWorkflowTestSuite))\n}\n\nfunc (s *rebalanceWorkflowTestSuite) SetupTest() {\n\ts.activityEnv = s.NewTestActivityEnvironment()\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\ts.workflowEnv.RegisterWorkflowWithOptions(RebalanceWorkflow, workflow.RegisterOptions{Name: RebalanceWorkflowTypeName})\n\ts.workflowEnv.RegisterActivityWithOptions(FailoverActivity, activity.RegisterOptions{Name: failoverActivityName})\n\ts.workflowEnv.RegisterActivityWithOptions(GetDomainsForRebalanceActivity, activity.RegisterOptions{Name: getRebalanceDomainsActivityName})\n\ts.activityEnv.RegisterActivityWithOptions(FailoverActivity, activity.RegisterOptions{Name: failoverActivityName})\n\ts.activityEnv.RegisterActivityWithOptions(GetDomainsForRebalanceActivity, activity.RegisterOptions{Name: getRebalanceDomainsActivityName})\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestGetDomainsForRebalanceActivity_ReturnOne() {\n\tactEnv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d2\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"false\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d3\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d4\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"false\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d5\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(domains, nil)\n\n\tactFuture, err := actEnv.ExecuteActivity(GetDomainsForRebalanceActivity)\n\ts.NoError(err)\n\tvar domainData []*DomainRebalanceData\n\terr = actFuture.Get(&domainData)\n\ts.NoError(err)\n\ts.Equal(1, len(domainData))\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestGetDomainsForRebalanceActivity_Error() {\n\tactEnv, mockResource := s.prepareTestActivityEnv()\n\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).\n\t\tReturn(nil, fmt.Errorf(\"test\"))\n\n\t_, err := actEnv.ExecuteActivity(GetDomainsForRebalanceActivity)\n\ts.Error(err)\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestWorkflow_Success() {\n\tparams := &RebalanceParams{\n\t\tBatchFailoverWaitTimeInSeconds: 10,\n\t\tBatchFailoverSize:              10,\n\t}\n\tdomainData := []*DomainRebalanceData{\n\t\t{\n\t\t\tDomainName:       \"d1\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d2\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d3\",\n\t\t\tPreferredCluster: \"c2\",\n\t\t},\n\t}\n\ts.workflowEnv.OnActivity(getRebalanceDomainsActivityName, mock.Anything).Return(domainData, nil)\n\tfailoverActivityParams1 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d1\", \"d2\"},\n\t\tTargetCluster: \"c1\",\n\t}\n\tfailoverActivityResult1 := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\", \"d2\"},\n\t}\n\tfailoverActivityParams2 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d3\"},\n\t\tTargetCluster: \"c2\",\n\t}\n\tfailoverActivityResult2 := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d3\"},\n\t}\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, failoverActivityParams1).Return(failoverActivityResult1, nil).Times(1)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, failoverActivityParams2).Return(failoverActivityResult2, nil).Times(1)\n\ts.workflowEnv.ExecuteWorkflow(RebalanceWorkflowTypeName, params)\n\tvar result RebalanceResult\n\terr := s.workflowEnv.GetWorkflowResult(&result)\n\ts.NoError(err)\n\ts.Equal(3, len(result.SuccessDomains))\n\ts.Equal(0, len(result.FailedDomains))\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestWorkflow_HalfFailoverActivityError_NoWorkflowError() {\n\tparams := &RebalanceParams{\n\t\tBatchFailoverWaitTimeInSeconds: 10,\n\t\tBatchFailoverSize:              10,\n\t}\n\tdomainData := []*DomainRebalanceData{\n\t\t{\n\t\t\tDomainName:       \"d1\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d2\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d3\",\n\t\t\tPreferredCluster: \"c2\",\n\t\t},\n\t}\n\ts.workflowEnv.OnActivity(getRebalanceDomainsActivityName, mock.Anything).Return(domainData, nil)\n\tfailoverActivityParams1 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d1\", \"d2\"},\n\t\tTargetCluster: \"c1\",\n\t}\n\tfailoverActivityResult1 := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\", \"d2\"},\n\t}\n\tfailoverActivityParams2 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d3\"},\n\t\tTargetCluster: \"c2\",\n\t}\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, failoverActivityParams1).Return(failoverActivityResult1, nil).Times(1)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, failoverActivityParams2).\n\t\tReturn(nil, fmt.Errorf(\"test\")).Times(1)\n\ts.workflowEnv.ExecuteWorkflow(RebalanceWorkflowTypeName, params)\n\tvar result RebalanceResult\n\terr := s.workflowEnv.GetWorkflowResult(&result)\n\ts.NoError(err)\n\ts.Equal(2, len(result.SuccessDomains))\n\ts.Equal(1, len(result.FailedDomains))\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestWorkflow_FailoverActivityError_NoWorkflowError() {\n\tparams := &RebalanceParams{\n\t\tBatchFailoverWaitTimeInSeconds: 10,\n\t\tBatchFailoverSize:              10,\n\t}\n\tdomainData := []*DomainRebalanceData{\n\t\t{\n\t\t\tDomainName:       \"d1\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d2\",\n\t\t\tPreferredCluster: \"c1\",\n\t\t},\n\t\t{\n\t\t\tDomainName:       \"d3\",\n\t\t\tPreferredCluster: \"c2\",\n\t\t},\n\t}\n\ts.workflowEnv.OnActivity(getRebalanceDomainsActivityName, mock.Anything).Return(domainData, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(nil, fmt.Errorf(\"test\"))\n\ts.workflowEnv.ExecuteWorkflow(RebalanceWorkflowTypeName, params)\n\tvar result RebalanceResult\n\terr := s.workflowEnv.GetWorkflowResult(&result)\n\ts.NoError(err)\n\ts.Equal(0, len(result.SuccessDomains))\n\ts.Equal(3, len(result.FailedDomains))\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestWorkflow_GetRebalanceDomainsActivityError_WorkflowError() {\n\tparams := &RebalanceParams{\n\t\tBatchFailoverWaitTimeInSeconds: 10,\n\t\tBatchFailoverSize:              10,\n\t}\n\ts.workflowEnv.OnActivity(getRebalanceDomainsActivityName, mock.Anything).Return(nil, fmt.Errorf(\"test\"))\n\ts.workflowEnv.ExecuteWorkflow(RebalanceWorkflowTypeName, params)\n\tvar result RebalanceResult\n\terr := s.workflowEnv.GetWorkflowResult(&result)\n\ts.Error(err)\n}\n\nfunc (s *rebalanceWorkflowTestSuite) TestShouldAllowRebalance() {\n\ttestCases := []struct {\n\t\tname   string\n\t\tdomain *types.DescribeDomainResponse\n\t\texpect bool\n\t}{\n\t\t{\n\t\t\tname: \"allow rebalance\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\texpect: true,\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because domain failover is not managed by cadence\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"false\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\texpect: false,\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because domain is not global domain\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: false,\n\t\t\t},\n\t\t\texpect: false,\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because domain status is not registered\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c2\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusDeprecated.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because domain has no preferred cluster\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because preferred cluster is the same as active cluster\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c1\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"not allow rebalance because preferred cluster is not in cluster list\",\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover:  \"true\",\n\t\t\t\t\t\tconstants.DomainDataKeyForPreferredCluster: \"c3\",\n\t\t\t\t\t},\n\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ts.Equal(tc.expect, shouldAllowRebalance(tc.domain))\n\t\t})\n\t}\n}\n\nfunc (s *rebalanceWorkflowTestSuite) prepareTestActivityEnv() (*testsuite.TestActivityEnvironment, *resource.Test) {\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\n\tctx := &FailoverManager{\n\t\tsvcClient:  mockResource.GetSDKClient(),\n\t\tclientBean: mockResource.ClientBean,\n\t}\n\ts.activityEnv.SetTestTimeout(time.Second * 5)\n\ts.activityEnv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), failoverManagerContextKey, ctx),\n\t})\n\n\ts.T().Cleanup(func() {\n\t\tmockResource.Finish(s.T())\n\t})\n\n\treturn s.activityEnv, mockResource\n}\n"
  },
  {
    "path": "service/worker/failovermanager/starter.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage failovermanager\n\nimport (\n\t\"context\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\n\t// Config defines the configuration for failover\n\tConfig struct {\n\t\tAdminOperationToken dynamicproperties.StringPropertyFn\n\t\t// ClusterMetadata contains the metadata for this cluster\n\t\tClusterMetadata cluster.Metadata\n\t}\n\n\t// BootstrapParams contains the set of params needed to bootstrap\n\t// failover manager\n\tBootstrapParams struct {\n\t\t// Config contains the configuration for scanner\n\t\tConfig Config\n\t\t// ServiceClient is an instance of cadence service client\n\t\tServiceClient workflowserviceclient.Interface\n\t\t// MetricsClient is an instance of metrics object for emitting stats\n\t\tMetricsClient metrics.Client\n\t\tLogger        log.Logger\n\t\t// TallyScope is an instance of tally metrics scope\n\t\tTallyScope tally.Scope\n\t\t// ClientBean is an instance of client.Bean for a collection of clients\n\t\tClientBean client.Bean\n\t}\n\n\t// FailoverManager of cadence worker service\n\tFailoverManager struct {\n\t\tcfg           Config\n\t\tsvcClient     workflowserviceclient.Interface\n\t\tclientBean    client.Bean\n\t\tmetricsClient metrics.Client\n\t\ttallyScope    tally.Scope\n\t\tlogger        log.Logger\n\t\tworker        worker.Worker\n\t}\n)\n\n// New returns a new instance of FailoverManager\nfunc New(params *BootstrapParams) *FailoverManager {\n\treturn &FailoverManager{\n\t\tcfg:           params.Config,\n\t\tsvcClient:     params.ServiceClient,\n\t\tmetricsClient: params.MetricsClient,\n\t\ttallyScope:    params.TallyScope,\n\t\tlogger:        params.Logger.WithTags(tag.ComponentBatcher),\n\t\tclientBean:    params.ClientBean,\n\t}\n}\n\n// Start starts the worker\nfunc (s *FailoverManager) Start() error {\n\tctx := context.WithValue(context.Background(), failoverManagerContextKey, s)\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              s.tallyScope,\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\tfailoverWorker := worker.New(s.svcClient, constants.SystemLocalDomainName, TaskListName, workerOpts)\n\tfailoverWorker.RegisterWorkflowWithOptions(FailoverWorkflow, workflow.RegisterOptions{Name: FailoverWorkflowTypeName})\n\tfailoverWorker.RegisterWorkflowWithOptions(RebalanceWorkflow, workflow.RegisterOptions{Name: RebalanceWorkflowTypeName})\n\tfailoverWorker.RegisterActivityWithOptions(FailoverActivity, activity.RegisterOptions{Name: failoverActivityName})\n\tfailoverWorker.RegisterActivityWithOptions(GetDomainsActivity, activity.RegisterOptions{Name: getDomainsActivityName})\n\tfailoverWorker.RegisterActivityWithOptions(GetDomainsForRebalanceActivity, activity.RegisterOptions{Name: getRebalanceDomainsActivityName})\n\ts.worker = failoverWorker\n\treturn failoverWorker.Start()\n}\n\n// Stop stops the worker\nfunc (s *FailoverManager) Stop() {\n\ts.worker.Stop()\n}\n"
  },
  {
    "path": "service/worker/failovermanager/workflow.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage failovermanager\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tcontextKey string\n)\n\nconst (\n\tfailoverManagerContextKey contextKey = \"failoverManagerContext\"\n\t// TaskListName tasklist\n\tTaskListName = \"cadence-sys-failoverManager-tasklist\"\n\t// FailoverWorkflowTypeName workflow type name\n\tFailoverWorkflowTypeName = \"cadence-sys-failoverManager-workflow\"\n\t// RebalanceWorkflowTypeName is rebalance workflow type name\n\tRebalanceWorkflowTypeName = \"cadence-sys-rebalance-workflow\"\n\t// WorkflowID will be reused to ensure only one workflow running\n\tFailoverWorkflowID              = \"cadence-failover-manager\"\n\tRebalanceWorkflowID             = \"cadence-rebalance-workflow\"\n\tDrillWorkflowID                 = FailoverWorkflowID + \"-drill\"\n\tfailoverActivityName            = \"cadence-sys-failover-activity\"\n\tgetDomainsActivityName          = \"cadence-sys-getDomains-activity\"\n\tgetRebalanceDomainsActivityName = \"cadence-sys-getRebalanceDomains-activity\"\n\n\tdefaultBatchFailoverSize              = 20\n\tdefaultBatchFailoverWaitTimeInSeconds = 30\n\n\terrMsgParamsIsNil                 = \"params is nil\"\n\terrMsgTargetClusterIsEmpty        = \"targetCluster is empty\"\n\terrMsgSourceClusterIsEmpty        = \"sourceCluster is empty\"\n\terrMsgTargetClusterIsSameAsSource = \"targetCluster is same as sourceCluster\"\n\n\t// QueryType for failover workflow\n\tQueryType = \"state\"\n\t// PauseSignal signal name for pause\n\tPauseSignal = \"pause\"\n\t// ResumeSignal signal name for resume\n\tResumeSignal = \"resume\"\n\n\t// workflow states for query\n\n\t// WorkflowInitialized state\n\tWorkflowInitialized = \"initialized\"\n\t// WorkflowRunning state\n\tWorkflowRunning = \"running\"\n\t// WorkflowPaused state\n\tWorkflowPaused = \"paused\"\n\t// WorkflowCompleted state\n\tWorkflowCompleted = \"complete\"\n\t// WorkflowAborted state\n\tWorkflowAborted = \"aborted\"\n\n\tunknownOperator = \"unknown\"\n)\n\ntype (\n\t// FailoverParams is the arg for failoverWorkflow\n\tFailoverParams struct {\n\t\t// TargetCluster is the destination of failover\n\t\tTargetCluster string\n\t\t// SourceCluster is from which cluster the domains are active before failover\n\t\tSourceCluster string\n\t\t// BatchFailoverSize is number of domains to failover in one batch\n\t\tBatchFailoverSize int\n\t\t// BatchFailoverWaitTimeInSeconds is the waiting time between batch failover\n\t\tBatchFailoverWaitTimeInSeconds int\n\t\t// Domains candidates to be failover\n\t\tDomains []string\n\t\t// DrillWaitTime defines the wait time of a failover drill\n\t\tDrillWaitTime time.Duration\n\t\t// GracefulFailoverTimeoutInSeconds\n\t\tGracefulFailoverTimeoutInSeconds *int32\n\t}\n\n\t// FailoverResult is workflow result\n\tFailoverResult struct {\n\t\tSuccessDomains      []string\n\t\tFailedDomains       []string\n\t\tSuccessResetDomains []string\n\t\tFailedResetDomains  []string\n\t}\n\n\t// GetDomainsActivityParams params for activity\n\tGetDomainsActivityParams struct {\n\t\tTargetCluster string\n\t\tSourceCluster string\n\t\tDomains       []string\n\t}\n\n\t// FailoverActivityParams params for activity\n\tFailoverActivityParams struct {\n\t\tDomains                          []string\n\t\tTargetCluster                    string\n\t\tGracefulFailoverTimeoutInSeconds *int32\n\t}\n\n\t// FailoverActivityResult result for failover activity\n\tFailoverActivityResult struct {\n\t\tSuccessDomains []string\n\t\tFailedDomains  []string\n\t}\n\n\t// QueryResult for failover progress\n\tQueryResult struct {\n\t\tTotalDomains        int\n\t\tSuccess             int\n\t\tFailed              int\n\t\tState               string\n\t\tTargetCluster       string\n\t\tSourceCluster       string\n\t\tSuccessDomains      []string // SuccessDomains are guaranteed succeed processed\n\t\tFailedDomains       []string // FailedDomains contains false positive\n\t\tSuccessResetDomains []string // SuccessResetDomains are domains successfully reset in drill mode\n\t\tFailedResetDomains  []string // FailedResetDomains contains false positive in drill mode\n\t\tOperator            string\n\t}\n)\n\n// FailoverWorkflow is the workflow that managed failover all domains with IsManagedByCadence=true\nfunc FailoverWorkflow(ctx workflow.Context, params *FailoverParams) (*FailoverResult, error) {\n\terr := validateParams(params)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// define query properties\n\tvar failedDomains []string\n\tvar successDomains []string\n\tvar successResetDomains []string\n\tvar failedResetDomains []string\n\tvar totalNumOfDomains int\n\twfState := WorkflowInitialized\n\toperator := getOperator(ctx)\n\terr = workflow.SetQueryHandler(ctx, QueryType, func(input []byte) (*QueryResult, error) {\n\t\treturn &QueryResult{\n\t\t\tTotalDomains:        totalNumOfDomains,\n\t\t\tSuccess:             len(successDomains),\n\t\t\tFailed:              len(failedDomains),\n\t\t\tState:               wfState,\n\t\t\tTargetCluster:       params.TargetCluster,\n\t\t\tSourceCluster:       params.SourceCluster,\n\t\t\tSuccessDomains:      successDomains,\n\t\t\tFailedDomains:       failedDomains,\n\t\t\tSuccessResetDomains: successResetDomains,\n\t\t\tFailedResetDomains:  failedResetDomains,\n\t\t\tOperator:            operator,\n\t\t}, nil\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\t// get target domains\n\tao := workflow.WithActivityOptions(ctx, getGetDomainsActivityOptions())\n\tgetDomainsParams := &GetDomainsActivityParams{\n\t\tTargetCluster: params.TargetCluster,\n\t\tSourceCluster: params.SourceCluster,\n\t\tDomains:       params.Domains,\n\t}\n\tvar domains []string\n\terr = workflow.ExecuteActivity(ao, GetDomainsActivity, getDomainsParams).Get(ctx, &domains)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\ttotalNumOfDomains = len(domains)\n\n\tpauseCh := workflow.GetSignalChannel(ctx, PauseSignal)\n\tresumeCh := workflow.GetSignalChannel(ctx, ResumeSignal)\n\tvar shouldPause bool\n\tcheckPauseSignal := func() {\n\t\tshouldPause = pauseCh.ReceiveAsync(nil)\n\t\tif shouldPause {\n\t\t\twfState = WorkflowPaused\n\t\t\tresumeCh.Receive(ctx, nil)\n\t\t\t// clean up all pending pause signal\n\t\t\tcleanupChannel(pauseCh)\n\t\t}\n\t\twfState = WorkflowRunning\n\t}\n\n\t// failover in batch\n\tsuccessDomains, failedDomains = failoverDomainsByBatch(ctx, domains, params, checkPauseSignal, false)\n\n\tif params.DrillWaitTime == 0 {\n\t\t// This is a normal failover\n\t\twfState = WorkflowCompleted\n\t\treturn &FailoverResult{\n\t\t\tSuccessDomains: successDomains,\n\t\t\tFailedDomains:  failedDomains,\n\t\t}, nil\n\t}\n\n\tworkflow.Sleep(ctx, params.DrillWaitTime)\n\t// Reset domains to original cluster\n\tsuccessResetDomains, failedResetDomains = failoverDomainsByBatch(ctx, domains, params, checkPauseSignal, true)\n\twfState = WorkflowCompleted\n\n\treturn &FailoverResult{\n\t\tSuccessDomains:      successDomains,\n\t\tFailedDomains:       failedDomains,\n\t\tSuccessResetDomains: successResetDomains,\n\t\tFailedResetDomains:  failedResetDomains,\n\t}, nil\n}\n\nfunc failoverDomainsByBatch(\n\tctx workflow.Context,\n\tdomains []string,\n\tparams *FailoverParams,\n\tpauseSignalHandler func(),\n\treverseFailover bool,\n) (successDomains []string, failedDomains []string) {\n\n\ttotalNumOfDomains := len(domains)\n\tbatchSize := params.BatchFailoverSize\n\ttimes := totalNumOfDomains/batchSize + 1\n\tao := workflow.WithActivityOptions(ctx, getFailoverActivityOptions())\n\ttargetCluster := params.TargetCluster\n\tif reverseFailover {\n\t\ttargetCluster = params.SourceCluster\n\t}\n\tfor i := 0; i < times; i++ {\n\t\tpauseSignalHandler()\n\n\t\tfailoverActivityParams := &FailoverActivityParams{\n\t\t\tDomains:                          domains[i*batchSize : min((i+1)*batchSize, totalNumOfDomains)],\n\t\t\tTargetCluster:                    targetCluster,\n\t\t\tGracefulFailoverTimeoutInSeconds: params.GracefulFailoverTimeoutInSeconds,\n\t\t}\n\t\tvar actResult FailoverActivityResult\n\t\terr := workflow.ExecuteActivity(ao, FailoverActivity, failoverActivityParams).Get(ctx, &actResult)\n\t\tif err != nil {\n\t\t\t// Domains in failed activity can be either failovered or not, but we treated them as failed.\n\t\t\t// This makes the query result for FailedDomains contains false positive results.\n\t\t\tfailedDomains = append(failedDomains, failoverActivityParams.Domains...)\n\t\t} else {\n\t\t\tsuccessDomains = append(successDomains, actResult.SuccessDomains...)\n\t\t\tfailedDomains = append(failedDomains, actResult.FailedDomains...)\n\t\t}\n\n\t\tif i != times-1 {\n\t\t\tworkflow.Sleep(ctx, time.Duration(params.BatchFailoverWaitTimeInSeconds)*time.Second)\n\t\t}\n\t}\n\treturn\n}\n\nfunc getOperator(ctx workflow.Context) string {\n\tmemo := workflow.GetInfo(ctx).Memo\n\tif memo == nil || len(memo.Fields) == 0 {\n\t\treturn unknownOperator\n\t}\n\topBytes, ok := memo.Fields[constants.MemoKeyForOperator]\n\tif !ok {\n\t\treturn unknownOperator\n\t}\n\tvar operator string\n\terr := json.Unmarshal(opBytes, &operator)\n\tif err != nil {\n\t\treturn unknownOperator\n\t}\n\treturn operator\n}\n\nfunc getGetDomainsActivityOptions() workflow.ActivityOptions {\n\treturn workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 10 * time.Second,\n\t\tStartToCloseTimeout:    20 * time.Second,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:    2 * time.Second,\n\t\t\tBackoffCoefficient: 2,\n\t\t\tMaximumInterval:    1 * time.Minute,\n\t\t\tExpirationInterval: 10 * time.Minute,\n\t\t\tNonRetriableErrorReasons: []string{\n\t\t\t\terrMsgParamsIsNil,\n\t\t\t\terrMsgTargetClusterIsEmpty,\n\t\t\t\terrMsgSourceClusterIsEmpty,\n\t\t\t\terrMsgTargetClusterIsSameAsSource},\n\t\t},\n\t}\n}\n\nfunc getFailoverActivityOptions() workflow.ActivityOptions {\n\treturn workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 10 * time.Second,\n\t\tStartToCloseTimeout:    10 * time.Second,\n\t}\n}\n\nfunc validateParams(params *FailoverParams) error {\n\tif params == nil {\n\t\treturn errors.New(errMsgParamsIsNil)\n\t}\n\tif params.BatchFailoverSize <= 0 {\n\t\tparams.BatchFailoverSize = defaultBatchFailoverSize\n\t}\n\tif params.BatchFailoverWaitTimeInSeconds <= 0 {\n\t\tparams.BatchFailoverWaitTimeInSeconds = defaultBatchFailoverWaitTimeInSeconds\n\t}\n\treturn validateTargetAndSourceCluster(params.TargetCluster, params.SourceCluster)\n}\n\n// GetDomainsActivity activity def\nfunc GetDomainsActivity(ctx context.Context, params *GetDomainsActivityParams) ([]string, error) {\n\terr := validateGetDomainsActivityParams(params)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdomains, err := getAllDomains(ctx, params.Domains)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar res []string\n\tfor _, domain := range domains {\n\t\tif shouldFailover(domain, params.SourceCluster) {\n\t\t\tdomainName := domain.GetDomainInfo().GetName()\n\t\t\tres = append(res, domainName)\n\t\t}\n\t}\n\treturn res, nil\n}\n\nfunc validateGetDomainsActivityParams(params *GetDomainsActivityParams) error {\n\tif params == nil {\n\t\treturn errors.New(errMsgParamsIsNil)\n\t}\n\treturn validateTargetAndSourceCluster(params.TargetCluster, params.SourceCluster)\n}\n\nfunc validateTargetAndSourceCluster(targetCluster, sourceCluster string) error {\n\tif len(targetCluster) == 0 {\n\t\treturn errors.New(errMsgTargetClusterIsEmpty)\n\t}\n\tif len(sourceCluster) == 0 {\n\t\treturn errors.New(errMsgSourceClusterIsEmpty)\n\t}\n\tif sourceCluster == targetCluster {\n\t\treturn errors.New(errMsgTargetClusterIsSameAsSource)\n\t}\n\treturn nil\n}\n\nfunc shouldFailover(domain *types.DescribeDomainResponse, sourceCluster string) bool {\n\tif !domain.GetIsGlobalDomain() {\n\t\treturn false\n\t}\n\t// Skip deprecated and deleted domains\n\tif domain.DomainInfo != nil && domain.DomainInfo.Status != nil {\n\t\tstatus := *domain.DomainInfo.Status\n\t\tif status == types.DomainStatusDeprecated || status == types.DomainStatusDeleted {\n\t\t\treturn false\n\t\t}\n\t}\n\n\t// TODO(active-active): Remove this check once failover drills are supported for\n\t// active-active workflows\n\n\tif domain.ReplicationConfiguration.ActiveClusters != nil &&\n\t\tlen(domain.ReplicationConfiguration.ActiveClusters.AttributeScopes) > 0 {\n\t\treturn false\n\t}\n\n\tcurrentActiveCluster := domain.ReplicationConfiguration.GetActiveClusterName()\n\tisDomainTarget := currentActiveCluster == sourceCluster\n\treturn isDomainTarget && isDomainFailoverManagedByCadence(domain)\n}\n\nfunc isDomainFailoverManagedByCadence(domain *types.DescribeDomainResponse) bool {\n\tdomainData := domain.DomainInfo.GetData()\n\treturn strings.ToLower(strings.TrimSpace(domainData[constants.DomainDataKeyForManagedFailover])) == \"true\"\n}\n\nfunc getClient(ctx context.Context) frontend.Client {\n\tmanager := ctx.Value(failoverManagerContextKey).(*FailoverManager)\n\tfeClient := manager.clientBean.GetFrontendClient()\n\treturn feClient\n}\n\nfunc getRemoteClient(ctx context.Context, clusterName string) (frontend.Client, error) {\n\tmanager := ctx.Value(failoverManagerContextKey).(*FailoverManager)\n\treturn manager.clientBean.GetRemoteFrontendClient(clusterName)\n}\n\nfunc getAllDomains(ctx context.Context, targetDomains []string) ([]*types.DescribeDomainResponse, error) {\n\tfeClient := getClient(ctx)\n\tvar res []*types.DescribeDomainResponse\n\n\tisTargetDomainsProvided := len(targetDomains) > 0\n\ttargetDomainsSet := make(map[string]struct{})\n\tif isTargetDomainsProvided {\n\t\tfor _, domain := range targetDomains {\n\t\t\ttargetDomainsSet[domain] = struct{}{}\n\t\t}\n\t}\n\n\tpagesize := int32(200)\n\tvar token []byte\n\tfor more := true; more; more = len(token) > 0 {\n\t\tlistRequest := &types.ListDomainsRequest{\n\t\t\tPageSize:      pagesize,\n\t\t\tNextPageToken: token,\n\t\t}\n\t\tlistResp, err := feClient.ListDomains(ctx, listRequest)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\ttoken = listResp.GetNextPageToken()\n\n\t\tif isTargetDomainsProvided {\n\t\t\tfor _, domain := range listResp.GetDomains() {\n\t\t\t\tif _, ok := targetDomainsSet[domain.DomainInfo.GetName()]; ok {\n\t\t\t\t\tres = append(res, domain)\n\t\t\t\t}\n\t\t\t}\n\t\t} else {\n\t\t\tres = append(res, listResp.GetDomains()...)\n\t\t}\n\n\t\tactivity.RecordHeartbeat(ctx, len(res))\n\t}\n\treturn res, nil\n}\n\n// FailoverActivity activity def\nfunc FailoverActivity(ctx context.Context, params *FailoverActivityParams) (*FailoverActivityResult, error) {\n\n\tlogger := activity.GetLogger(ctx)\n\tfrontendClient := getClient(ctx)\n\tdomains := params.Domains\n\tvar successDomains []string\n\tvar failedDomains []string\n\tfor _, domain := range domains {\n\t\t// Check if poller exist\n\t\tif err := validateTaskListPollerInfo(ctx, params.TargetCluster, domain); err != nil {\n\t\t\tlogger.Error(\"Failed to validate task list poller info\", zap.Error(err))\n\t\t\tfailedDomains = append(failedDomains, domain)\n\t\t\tcontinue\n\t\t}\n\t\tupdateRequest := &types.UpdateDomainRequest{\n\t\t\tName:              domain,\n\t\t\tActiveClusterName: common.StringPtr(params.TargetCluster),\n\t\t}\n\t\tif params.GracefulFailoverTimeoutInSeconds != nil {\n\t\t\tupdateRequest.FailoverTimeoutInSeconds = params.GracefulFailoverTimeoutInSeconds\n\t\t}\n\n\t\t_, err := frontendClient.UpdateDomain(ctx, updateRequest)\n\t\tif err != nil {\n\t\t\tfailedDomains = append(failedDomains, domain)\n\t\t} else {\n\t\t\tsuccessDomains = append(successDomains, domain)\n\t\t}\n\t}\n\treturn &FailoverActivityResult{\n\t\tSuccessDomains: successDomains,\n\t\tFailedDomains:  failedDomains,\n\t}, nil\n}\n\nfunc cleanupChannel(channel workflow.Channel) {\n\tfor {\n\t\tif hasValue := channel.ReceiveAsync(nil); !hasValue {\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc validateTaskListPollerInfo(ctx context.Context, targetCluster string, domain string) error {\n\tremoteFrontendClient, err := getRemoteClient(ctx, targetCluster)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfrontendClient := getClient(ctx)\n\tlocalTaskListResponse, err := frontendClient.GetTaskListsByDomain(ctx, &types.GetTaskListsByDomainRequest{Domain: domain})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to get task list for domain %s\", domain)\n\t}\n\n\tremoteTaskListRepsonse, err := remoteFrontendClient.GetTaskListsByDomain(ctx, &types.GetTaskListsByDomainRequest{Domain: domain})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to get task list for domain %s\", domain)\n\t}\n\tfor name, tl := range localTaskListResponse.GetDecisionTaskListMap() {\n\t\tif len(tl.GetPollers()) != 0 {\n\t\t\tremoteTaskList, ok := remoteTaskListRepsonse.GetDecisionTaskListMap()[name]\n\t\t\tif !ok || len(remoteTaskList.GetPollers()) == 0 {\n\t\t\t\treturn fmt.Errorf(\"received zero poller in decision task list %s with domain %s\", name, domain)\n\t\t\t}\n\t\t}\n\t}\n\tfor name, tl := range localTaskListResponse.GetActivityTaskListMap() {\n\t\tif len(tl.GetPollers()) != 0 {\n\t\t\tremoteTaskList, ok := remoteTaskListRepsonse.GetActivityTaskListMap()[name]\n\t\t\tif !ok || len(remoteTaskList.GetPollers()) == 0 {\n\t\t\t\treturn fmt.Errorf(\"received zero poller in decision task list %s with domain %s\", name, domain)\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/failovermanager/workflow_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage failovermanager\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar clusters = []*types.ClusterReplicationConfiguration{\n\t{\n\t\tClusterName: \"c1\",\n\t},\n\t{\n\t\tClusterName: \"c2\",\n\t},\n}\n\ntype failoverWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tactivityEnv *testsuite.TestActivityEnvironment\n\tworkflowEnv *testsuite.TestWorkflowEnvironment\n}\n\nfunc TestFailoverWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(failoverWorkflowTestSuite))\n}\n\nfunc (s *failoverWorkflowTestSuite) SetupTest() {\n\ts.activityEnv = s.NewTestActivityEnvironment()\n\ts.workflowEnv = s.NewTestWorkflowEnvironment()\n\ts.workflowEnv.RegisterWorkflowWithOptions(FailoverWorkflow, workflow.RegisterOptions{Name: FailoverWorkflowTypeName})\n\ts.workflowEnv.RegisterActivityWithOptions(FailoverActivity, activity.RegisterOptions{Name: failoverActivityName})\n\ts.workflowEnv.RegisterActivityWithOptions(GetDomainsActivity, activity.RegisterOptions{Name: getDomainsActivityName})\n\ts.activityEnv.RegisterActivityWithOptions(FailoverActivity, activity.RegisterOptions{Name: failoverActivityName})\n\ts.activityEnv.RegisterActivityWithOptions(GetDomainsActivity, activity.RegisterOptions{Name: getDomainsActivityName})\n}\n\nfunc (s *failoverWorkflowTestSuite) TearDownTest() {\n\ts.workflowEnv.AssertExpectations(s.T())\n}\n\nfunc (s *failoverWorkflowTestSuite) TestValidateParams() {\n\ts.Error(validateParams(nil))\n\tparams := &FailoverParams{}\n\ts.Error(validateParams(params))\n\tparams.TargetCluster = \"t\"\n\ts.Error(validateParams(params))\n\tparams.SourceCluster = \"t\"\n\ts.Error(validateParams(params))\n\tparams.SourceCluster = \"s\"\n\ts.NoError(validateParams(params))\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_InvalidParams() {\n\tparams := &FailoverParams{}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Error(s.workflowEnv.GetWorkflowError())\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_GetDomainActivityError() {\n\terr := errors.New(\"mockErr\")\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(nil, err)\n\tparams := &FailoverParams{\n\t\tTargetCluster: \"t\",\n\t\tSourceCluster: \"s\",\n\t}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\ts.True(s.workflowEnv.IsWorkflowCompleted())\n\ts.Equal(\"mockErr\", s.workflowEnv.GetWorkflowError().Error())\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_FailoverActivityError() {\n\tdomains := []string{\"d1\"}\n\terr := errors.New(\"mockErr\")\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(domains, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(nil, err)\n\tparams := &FailoverParams{\n\t\tTargetCluster: \"t\",\n\t\tSourceCluster: \"s\",\n\t}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.Equal(0, len(result.SuccessDomains))\n\ts.Equal(domains, result.FailedDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_Success() {\n\tdomains := []string{\"d1\"}\n\tmockFailoverActivityResult := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\"},\n\t}\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(domains, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(mockFailoverActivityResult, nil)\n\tparams := &FailoverParams{\n\t\tTargetCluster: \"t\",\n\t\tSourceCluster: \"s\",\n\t}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.Equal(mockFailoverActivityResult.SuccessDomains, result.SuccessDomains)\n\ts.Equal(mockFailoverActivityResult.FailedDomains, result.FailedDomains)\n\n\tqueryResult, err := s.workflowEnv.QueryWorkflow(QueryType)\n\ts.NoError(err)\n\tvar res QueryResult\n\ts.NoError(queryResult.Get(&res))\n\ts.Equal(len(domains), res.TotalDomains)\n\ts.Equal(len(domains), res.Success)\n\ts.Equal(0, res.Failed)\n\ts.Equal(WorkflowCompleted, res.State)\n\ts.Equal(\"t\", res.TargetCluster)\n\ts.Equal(\"s\", res.SourceCluster)\n\ts.Equal(domains, res.SuccessDomains)\n\ts.Equal(0, len(res.FailedDomains))\n\ts.Equal(unknownOperator, res.Operator)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_Success_Batches() {\n\tdomains := []string{\"d1\", \"d2\", \"d3\"}\n\texpectFailoverActivityParams1 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d1\", \"d2\"},\n\t\tTargetCluster: \"t\",\n\t}\n\tmockFailoverActivityResult1 := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\", \"d2\"},\n\t}\n\texpectFailoverActivityParams2 := &FailoverActivityParams{\n\t\tDomains:       []string{\"d3\"},\n\t\tTargetCluster: \"t\",\n\t}\n\tmockFailoverActivityResult2 := &FailoverActivityResult{\n\t\tFailedDomains: []string{\"d3\"},\n\t}\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(domains, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, expectFailoverActivityParams1).Return(mockFailoverActivityResult1, nil).Once()\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, expectFailoverActivityParams2).Return(mockFailoverActivityResult2, nil).Once()\n\n\tparams := &FailoverParams{\n\t\tTargetCluster:     \"t\",\n\t\tSourceCluster:     \"s\",\n\t\tBatchFailoverSize: 2,\n\t\tDomains:           domains,\n\t}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.Equal(mockFailoverActivityResult1.SuccessDomains, result.SuccessDomains)\n\ts.Equal(mockFailoverActivityResult2.FailedDomains, result.FailedDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_Pause() {\n\tdomains := []string{\"d1\"}\n\tmockFailoverActivityResult := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\"},\n\t}\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(domains, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(mockFailoverActivityResult, nil).Once()\n\n\ts.workflowEnv.RegisterDelayedCallback(func() {\n\t\ts.workflowEnv.SignalWorkflow(PauseSignal, nil)\n\t}, time.Millisecond*0)\n\ts.workflowEnv.RegisterDelayedCallback(func() {\n\t\ts.assertQueryState(s.workflowEnv, WorkflowPaused)\n\t}, time.Millisecond*100)\n\ts.workflowEnv.RegisterDelayedCallback(func() {\n\t\ts.workflowEnv.SignalWorkflow(ResumeSignal, nil)\n\t}, time.Millisecond*200)\n\ts.workflowEnv.RegisterDelayedCallback(func() {\n\t\ts.assertQueryState(s.workflowEnv, WorkflowRunning)\n\t}, time.Millisecond*300)\n\n\tparams := &FailoverParams{\n\t\tTargetCluster:     \"t\",\n\t\tSourceCluster:     \"s\",\n\t\tBatchFailoverSize: 2,\n\t\tDomains:           domains,\n\t}\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.Equal(mockFailoverActivityResult.SuccessDomains, result.SuccessDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestWorkflow_WithDrillWaitTime_Success() {\n\tdomains := []string{\"d1\"}\n\tmockFailoverActivityResult := &FailoverActivityResult{\n\t\tSuccessDomains: []string{\"d1\"},\n\t}\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(domains, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(mockFailoverActivityResult, nil)\n\tparams := &FailoverParams{\n\t\tTargetCluster: \"t\",\n\t\tSourceCluster: \"s\",\n\t\tDrillWaitTime: 1 * time.Second,\n\t}\n\tvar timerCount int\n\ts.workflowEnv.SetOnTimerScheduledListener(func(timerID string, duration time.Duration) {\n\t\ttimerCount++\n\t\tif duration != time.Second && duration != 30*time.Second {\n\t\t\ts.Fail(\"Receive unknown timer.\")\n\t\t}\n\t})\n\ts.workflowEnv.SetOnTimerFiredListener(func(timerID string) {\n\t\ttimerCount--\n\t})\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\ts.Equal(mockFailoverActivityResult.SuccessDomains, result.SuccessDomains)\n\ts.Equal(mockFailoverActivityResult.FailedDomains, result.FailedDomains)\n\n\tqueryResult, err := s.workflowEnv.QueryWorkflow(QueryType)\n\ts.NoError(err)\n\ts.Equal(0, timerCount)\n\tvar res QueryResult\n\ts.NoError(queryResult.Get(&res))\n\ts.Equal(len(domains), res.TotalDomains)\n\ts.Equal(len(domains), res.Success)\n\ts.Equal(0, res.Failed)\n\ts.Equal(WorkflowCompleted, res.State)\n\ts.Equal(\"t\", res.TargetCluster)\n\ts.Equal(\"s\", res.SourceCluster)\n\ts.Equal(domains, res.SuccessDomains)\n\ts.Equal(0, len(res.FailedDomains))\n\ts.Equal(unknownOperator, res.Operator)\n\ts.Equal(domains, res.SuccessResetDomains)\n\ts.Equal(0, len(res.FailedResetDomains))\n}\n\nfunc (s *failoverWorkflowTestSuite) TestShouldFailover() {\n\n\ttests := []struct {\n\t\tdomain        *types.DescribeDomainResponse\n\t\tsourceCluster string\n\t\texpected      bool\n\t}{\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: false,\n\t\t\t},\n\t\t\tsourceCluster: \"c1\",\n\t\t\texpected:      false,\n\t\t},\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsourceCluster: \"c2\",\n\t\t\texpected:      false,\n\t\t},\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c2\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t},\n\t\t\tsourceCluster: \"c2\",\n\t\t\texpected:      false,\n\t\t},\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c2\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsourceCluster: \"c2\",\n\t\t\texpected:      true,\n\t\t},\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c2\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"c1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsourceCluster: \"c2\",\n\t\t\texpected:      false,\n\t\t},\n\t\t{\n\t\t\tdomain: &types.DescribeDomainResponse{\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c2\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tsourceCluster: \"c2\",\n\t\t\texpected:      false,\n\t\t},\n\t}\n\tfor _, t := range tests {\n\t\ts.Equal(t.expected, shouldFailover(t.domain, t.sourceCluster))\n\t}\n}\n\nfunc (s *failoverWorkflowTestSuite) TestGetDomainsActivity() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(domains, nil)\n\n\tparams := &GetDomainsActivityParams{\n\t\tTargetCluster: \"c2\",\n\t\tSourceCluster: \"c1\",\n\t}\n\tactResult, err := env.ExecuteActivity(getDomainsActivityName, params)\n\ts.NoError(err)\n\tvar result []string\n\ts.NoError(actResult.Get(&result))\n\ts.Equal([]string{\"d1\"}, result)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestGetDomainsActivity_WithTargetDomains() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d1\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d2\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"d3\",\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(domains, nil)\n\n\tparams := &GetDomainsActivityParams{\n\t\tTargetCluster: \"c2\",\n\t\tSourceCluster: \"c1\",\n\t\tDomains:       []string{\"d1\", \"d3\"}, // only target d1 and d3\n\t}\n\tactResult, err := env.ExecuteActivity(getDomainsActivityName, params)\n\ts.NoError(err)\n\tvar result []string\n\ts.NoError(actResult.Get(&result))\n\ts.Equal([]string{\"d1\"}, result) // d3 filtered out because not managed\n}\n\nfunc (s *failoverWorkflowTestSuite) TestGetDomainsActivity_FiltersActiveActiveDomains() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"regular-domain\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"active-active-domain\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"empty-active-clusters-domain\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(domains, nil)\n\n\tparams := &GetDomainsActivityParams{\n\t\tTargetCluster: \"c2\",\n\t\tSourceCluster: \"c1\",\n\t}\n\tactResult, err := env.ExecuteActivity(getDomainsActivityName, params)\n\ts.NoError(err)\n\tvar result []string\n\ts.NoError(actResult.Get(&result))\n\ts.Equal([]string{\"regular-domain\"}, result)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestGetDomainsActivity_WithActiveActiveTargetDomains() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"regular-domain\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName: \"active-active-domain\",\n\t\t\t\t\tData: map[string]string{constants.DomainDataKeyForManagedFailover: \"true\"},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters:          clusters,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\"region2\": {\n\t\t\t\t\t\t\t\t\t\tActiveClusterName: \"c2\",\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tIsGlobalDomain: true,\n\t\t\t},\n\t\t},\n\t}\n\tmockResource.FrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(domains, nil)\n\n\tparams := &GetDomainsActivityParams{\n\t\tTargetCluster: \"c2\",\n\t\tSourceCluster: \"c1\",\n\t\tDomains:       []string{\"regular-domain\", \"active-active-domain\"},\n\t}\n\tactResult, err := env.ExecuteActivity(getDomainsActivityName, params)\n\ts.NoError(err)\n\tvar result []string\n\ts.NoError(actResult.Get(&result))\n\ts.Equal([]string{\"regular-domain\"}, result)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestFailoverActivity_ForceFailover_Success() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := []string{\"d1\", \"d2\"}\n\tdescribeTaskListResp := &types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{\n\t\t{\n\t\t\tIdentity: \"test\",\n\t\t},\n\t}}\n\ttaskListMap := map[string]*types.DescribeTaskListResponse{\n\t\t\"tl\": describeTaskListResp,\n\t}\n\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil, nil).Times(len(domains))\n\tmockResource.FrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\tmockResource.RemoteFrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\n\tparams := &FailoverActivityParams{\n\t\tDomains:       domains,\n\t\tTargetCluster: \"c2\",\n\t}\n\n\tactResult, err := env.ExecuteActivity(failoverActivityName, params)\n\ts.NoError(err)\n\tvar result FailoverActivityResult\n\ts.NoError(actResult.Get(&result))\n\ts.Equal(domains, result.SuccessDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestFailoverActivity_GracefulFailover_Success() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := []string{\"d1\", \"d2\"}\n\tdescribeTaskListResp := &types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{\n\t\t{\n\t\t\tIdentity: \"test\",\n\t\t},\n\t}}\n\ttaskListMap := map[string]*types.DescribeTaskListResponse{\n\t\t\"tl\": describeTaskListResp,\n\t}\n\tparams := &FailoverActivityParams{\n\t\tDomains:                          domains,\n\t\tTargetCluster:                    \"c2\",\n\t\tGracefulFailoverTimeoutInSeconds: common.Int32Ptr(int32(10)),\n\t}\n\n\tupdateRequest1 := &types.UpdateDomainRequest{\n\t\tName:                     \"d1\",\n\t\tActiveClusterName:        common.StringPtr(\"c2\"),\n\t\tFailoverTimeoutInSeconds: params.GracefulFailoverTimeoutInSeconds,\n\t}\n\tupdateRequest2 := &types.UpdateDomainRequest{\n\t\tName:                     \"d2\",\n\t\tActiveClusterName:        common.StringPtr(\"c2\"),\n\t\tFailoverTimeoutInSeconds: params.GracefulFailoverTimeoutInSeconds,\n\t}\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), updateRequest1).Return(nil, nil).Times(1)\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), updateRequest2).Return(nil, nil).Times(1)\n\tmockResource.FrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\tmockResource.RemoteFrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\n\tactResult, err := env.ExecuteActivity(failoverActivityName, params)\n\ts.NoError(err)\n\tvar result FailoverActivityResult\n\ts.NoError(actResult.Get(&result))\n\ts.Equal(domains, result.SuccessDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestFailoverActivity_Error() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := []string{\"d1\", \"d2\"}\n\ttargetCluster := \"c2\"\n\tupdateRequest1 := &types.UpdateDomainRequest{\n\t\tName:              \"d1\",\n\t\tActiveClusterName: common.StringPtr(targetCluster),\n\t}\n\tupdateRequest2 := &types.UpdateDomainRequest{\n\t\tName:              \"d2\",\n\t\tActiveClusterName: common.StringPtr(targetCluster),\n\t}\n\tdescribeTaskListResp := &types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{\n\t\t{\n\t\t\tIdentity: \"test\",\n\t\t},\n\t}}\n\ttaskListMap := map[string]*types.DescribeTaskListResponse{\n\t\t\"tl\": describeTaskListResp,\n\t}\n\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), updateRequest1).Return(nil, nil)\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), updateRequest2).Return(nil, errors.New(\"mockErr\"))\n\tmockResource.FrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\tmockResource.RemoteFrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap,\n\t\tActivityTaskListMap: taskListMap,\n\t}, nil).Times(len(domains))\n\n\tparams := &FailoverActivityParams{\n\t\tDomains:       domains,\n\t\tTargetCluster: targetCluster,\n\t}\n\n\tactResult, err := env.ExecuteActivity(failoverActivityName, params)\n\ts.NoError(err)\n\tvar result FailoverActivityResult\n\ts.NoError(actResult.Get(&result))\n\ts.Equal([]string{\"d1\"}, result.SuccessDomains)\n\ts.Equal([]string{\"d2\"}, result.FailedDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestFailoverActivity_NoPoller_Error() {\n\tenv, mockResource := s.prepareTestActivityEnv()\n\n\tdomains := []string{\"d1\", \"d2\"}\n\ttargetCluster := \"c2\"\n\tdescribeTaskListResp1 := &types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{\n\t\t{\n\t\t\tIdentity: \"test\",\n\t\t},\n\t}}\n\ttaskListMap1 := map[string]*types.DescribeTaskListResponse{\n\t\t\"tl\": describeTaskListResp1,\n\t}\n\tdescribeTaskListResp2 := &types.DescribeTaskListResponse{Pollers: []*types.PollerInfo{}}\n\ttaskListMap2 := map[string]*types.DescribeTaskListResponse{\n\t\t\"tl\": describeTaskListResp2,\n\t}\n\n\tmockResource.FrontendClient.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Times(0)\n\tmockResource.FrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap1,\n\t\tActivityTaskListMap: taskListMap1,\n\t}, nil).Times(len(domains))\n\tmockResource.RemoteFrontendClient.EXPECT().GetTaskListsByDomain(gomock.Any(), gomock.Any()).Return(&types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: taskListMap2,\n\t\tActivityTaskListMap: taskListMap2,\n\t}, nil).Times(len(domains))\n\n\tparams := &FailoverActivityParams{\n\t\tDomains:       domains,\n\t\tTargetCluster: targetCluster,\n\t}\n\n\tactResult, err := env.ExecuteActivity(failoverActivityName, params)\n\ts.NoError(err)\n\tvar result FailoverActivityResult\n\ts.NoError(actResult.Get(&result))\n\ts.Equal(0, len(result.SuccessDomains))\n\ts.Equal([]string{\"d1\", \"d2\"}, result.FailedDomains)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestGetOperator() {\n\toperator := \"testOperator\"\n\ts.workflowEnv.SetMemoOnStart(map[string]interface{}{\n\t\tconstants.MemoKeyForOperator: operator,\n\t})\n\n\ts.workflowEnv.OnActivity(getDomainsActivityName, mock.Anything, mock.Anything).Return(nil, nil)\n\ts.workflowEnv.OnActivity(failoverActivityName, mock.Anything, mock.Anything).Return(nil, nil)\n\tparams := &FailoverParams{\n\t\tTargetCluster: \"t\",\n\t\tSourceCluster: \"s\",\n\t}\n\n\ts.workflowEnv.ExecuteWorkflow(FailoverWorkflowTypeName, params)\n\tvar result FailoverResult\n\ts.NoError(s.workflowEnv.GetWorkflowResult(&result))\n\n\tqueryResult, err := s.workflowEnv.QueryWorkflow(QueryType)\n\ts.NoError(err)\n\tvar res QueryResult\n\ts.NoError(queryResult.Get(&res))\n\n\ts.Equal(operator, res.Operator)\n}\n\nfunc (s *failoverWorkflowTestSuite) TestShouldFailover_DeprecatedDomain() {\n\tdeprecatedDomain := &types.DescribeDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:   \"test-deprecated-domain\",\n\t\t\tStatus: types.DomainStatusDeprecated.Ptr(),\n\t\t\tData: map[string]string{\n\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t},\n\t\t},\n\t\tIsGlobalDomain: true,\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: \"cluster1\",\n\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t{ClusterName: \"cluster1\"},\n\t\t\t\t{ClusterName: \"cluster2\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tresult := shouldFailover(deprecatedDomain, \"cluster1\")\n\ts.False(result, \"Deprecated domains should not be included in failover\")\n}\n\nfunc (s *failoverWorkflowTestSuite) TestShouldFailover_DeletedDomain() {\n\tdeletedDomain := &types.DescribeDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:   \"test-deleted-domain\",\n\t\t\tStatus: types.DomainStatusDeleted.Ptr(),\n\t\t\tData: map[string]string{\n\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t},\n\t\t},\n\t\tIsGlobalDomain: true,\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: \"cluster1\",\n\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t{ClusterName: \"cluster1\"},\n\t\t\t\t{ClusterName: \"cluster2\"},\n\t\t\t},\n\t\t},\n\t}\n\n\tresult := shouldFailover(deletedDomain, \"cluster1\")\n\ts.False(result, \"Deleted domains should not be included in failover\")\n}\n\nfunc (s *failoverWorkflowTestSuite) assertQueryState(env *testsuite.TestWorkflowEnvironment, expectedState string) {\n\tqueryResult, err := env.QueryWorkflow(QueryType)\n\ts.NoError(err)\n\tvar res QueryResult\n\ts.NoError(queryResult.Get(&res))\n\ts.Equal(expectedState, res.State)\n}\n\nfunc (s *failoverWorkflowTestSuite) prepareTestActivityEnv() (*testsuite.TestActivityEnvironment, *resource.Test) {\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\n\tctx := &FailoverManager{\n\t\tsvcClient:  mockResource.GetSDKClient(),\n\t\tclientBean: mockResource.ClientBean,\n\t}\n\ts.activityEnv.SetTestTimeout(time.Second * 5)\n\ts.activityEnv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: context.WithValue(context.Background(), failoverManagerContextKey, ctx),\n\t})\n\n\ts.T().Cleanup(func() {\n\t\tmockResource.Finish(s.T())\n\t})\n\n\treturn s.activityEnv, mockResource\n}\n"
  },
  {
    "path": "service/worker/indexer/esProcessor.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage indexer\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/definition\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst (\n\t// retry configs for es bulk bulkProcessor\n\tesProcessorInitialRetryInterval = 200 * time.Millisecond\n\tesProcessorMaxRetryInterval     = 20 * time.Second\n)\n\ntype (\n\t// ESProcessorImpl implements ESProcessor, it's an agent of GenericBulkProcessor\n\tESProcessorImpl struct {\n\t\tbulkProcessor bulk.GenericBulkProcessor\n\t\tmapToKafkaMsg collection.ConcurrentTxMap // used to map ES request to kafka message\n\t\tconfig        *Config\n\t\tlogger        log.Logger\n\t\tscope         metrics.Scope\n\t\tmsgEncoder    codec.BinaryEncoder\n\t}\n\n\tkafkaMessageWithMetrics struct { // value of ESProcessorImpl.mapToKafkaMsg\n\t\tmessage        messaging.Message\n\t\tswFromAddToAck *metrics.Stopwatch // metric from message add to process, to message ack/nack\n\t}\n)\n\n// newESProcessor creates new ESProcessor\nfunc newESProcessor(\n\tname string,\n\tconfig *Config,\n\tclient es.GenericClient,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) (*ESProcessorImpl, error) {\n\tp := &ESProcessorImpl{\n\t\tconfig:     config,\n\t\tlogger:     logger.WithTags(tag.ComponentIndexerESProcessor),\n\t\tscope:      metricsClient.Scope(metrics.ESProcessorScope),\n\t\tmsgEncoder: defaultEncoder,\n\t}\n\n\tparams := &bulk.BulkProcessorParameters{\n\t\tName:          name,\n\t\tNumOfWorkers:  config.ESProcessorNumOfWorkers(),\n\t\tBulkActions:   config.ESProcessorBulkActions(),\n\t\tBulkSize:      config.ESProcessorBulkSize(),\n\t\tFlushInterval: config.ESProcessorFlushInterval(),\n\t\tBackoff:       bulk.NewExponentialBackoff(esProcessorInitialRetryInterval, esProcessorMaxRetryInterval),\n\t\tBeforeFunc:    p.bulkBeforeAction,\n\t\tAfterFunc:     p.bulkAfterAction,\n\t}\n\tprocessor, err := client.RunBulkProcessor(context.Background(), params)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tp.bulkProcessor = processor\n\tp.mapToKafkaMsg = collection.NewShardedConcurrentTxMap(1024, p.hashFn)\n\treturn p, nil\n}\n\nfunc (p *ESProcessorImpl) Start() {\n\t// current implementation (v6 and v7) allows to invoke Start() multiple times\n\tp.bulkProcessor.Start(context.Background())\n\n}\nfunc (p *ESProcessorImpl) Stop() {\n\tp.bulkProcessor.Stop() //nolint:errcheck\n\tp.mapToKafkaMsg = nil\n}\n\n// Add an ES request, and an map item for kafka message\nfunc (p *ESProcessorImpl) Add(request *bulk.GenericBulkableAddRequest, key string, kafkaMsg messaging.Message) {\n\tactionWhenFoundDuplicates := func(key interface{}, value interface{}) error {\n\t\treturn kafkaMsg.Ack()\n\t}\n\tsw := p.scope.StartTimer(metrics.ESProcessorProcessMsgLatency)\n\tmapVal := newKafkaMessageWithMetrics(kafkaMsg, &sw)\n\t_, isDup, _ := p.mapToKafkaMsg.PutOrDo(key, mapVal, actionWhenFoundDuplicates)\n\tif isDup {\n\t\treturn\n\t}\n\tp.bulkProcessor.Add(request)\n}\n\n// bulkBeforeAction is triggered before bulk bulkProcessor commit\nfunc (p *ESProcessorImpl) bulkBeforeAction(executionID int64, requests []bulk.GenericBulkableRequest) {\n\tp.scope.AddCounter(metrics.ESProcessorRequests, int64(len(requests)))\n}\n\n// bulkAfterAction is triggered after bulk bulkProcessor commit\nfunc (p *ESProcessorImpl) bulkAfterAction(id int64, requests []bulk.GenericBulkableRequest, response *bulk.GenericBulkResponse, err *bulk.GenericError) {\n\tif err != nil {\n\t\tisRetryable := isResponseRetriable(err.Status)\n\t\tfor _, request := range requests {\n\t\t\tif isRetryable {\n\t\t\t\t// This happens after configured retry, which means something bad happens on cluster or index\n\t\t\t\t// When cluster back to live, bulkProcessor will re-commit those failure requests\n\t\t\t\t// retryable errors will be retried by the bulk processor\n\t\t\t\tp.logger.Error(\"Error commit bulk request. ES request failed and is retryable\", tag.Error(err.Details), tag.ESRequest(request.String()))\n\t\t\t} else {\n\t\t\t\tkey := p.retrieveKafkaKey(request)\n\t\t\t\tif key == \"\" {\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\twid, rid, domainID := p.getMsgWithInfo(key)\n\n\t\t\t\t// check if it is a delete request and status code\n\t\t\t\t// 404 means the document does not exist\n\t\t\t\t// 409 means means the document's version does not match (or if the document has been updated or deleted by another process)\n\t\t\t\t// this can happen during the data migration, the doc was deleted in the old index but not exists in the new index\n\t\t\t\tif err.Status == 409 {\n\t\t\t\t\t// Sample version conflict logs to reduce spam (log every 100th conflict)\n\t\t\t\t\tif rand.Intn(100) == 0 {\n\t\t\t\t\t\tp.logger.Info(\"Request encountered a version conflict. Acknowledging to prevent retry.\",\n\t\t\t\t\t\t\ttag.ESResponseStatus(err.Status), tag.ESRequest(request.String()),\n\t\t\t\t\t\t\ttag.WorkflowID(wid),\n\t\t\t\t\t\t\ttag.WorkflowRunID(rid),\n\t\t\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\t\t}\n\t\t\t\t\tp.ackKafkaMsg(key)\n\t\t\t\t\tcontinue\n\t\t\t\t} else if err.Status == 404 {\n\t\t\t\t\treq, err := request.Source()\n\t\t\t\t\tif err == nil && p.isDeleteRequest(req) {\n\t\t\t\t\t\tp.logger.Info(\"Document has been deleted. Acknowledging to prevent retry.\",\n\t\t\t\t\t\t\ttag.ESResponseStatus(404), tag.ESRequest(request.String()),\n\t\t\t\t\t\t\ttag.WorkflowID(wid),\n\t\t\t\t\t\t\ttag.WorkflowRunID(rid),\n\t\t\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\t\t\tp.ackKafkaMsg(key)\n\t\t\t\t\t\tcontinue\n\t\t\t\t\t} else {\n\t\t\t\t\t\tp.logger.Error(\"Get request source err.\", tag.Error(err), tag.ESRequest(request.String()))\n\t\t\t\t\t\tp.scope.IncCounter(metrics.ESProcessorCorruptedData)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tp.logger.Error(\"Error commit bulk request. ES request failed and is not retryable\",\n\t\t\t\t\ttag.ESResponseStatus(err.Status),\n\t\t\t\t\ttag.ESRequest(request.String()),\n\t\t\t\t\ttag.WorkflowID(wid),\n\t\t\t\t\ttag.WorkflowRunID(rid),\n\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\tp.nackKafkaMsg(key)\n\t\t\t}\n\t\t\tp.scope.IncCounter(metrics.ESProcessorFailures)\n\t\t}\n\t\treturn\n\t}\n\n\tresponseItems := response.Items\n\tfor i := 0; i < len(requests); i++ {\n\t\tkey := p.retrieveKafkaKey(requests[i])\n\t\tif key == \"\" {\n\t\t\tcontinue\n\t\t}\n\t\tresponseItem := responseItems[i]\n\t\tfor _, resp := range responseItem {\n\t\t\tswitch {\n\t\t\tcase isResponseSuccess(resp.Status):\n\t\t\t\tp.ackKafkaMsg(key)\n\t\t\tcase !isResponseRetriable(resp.Status):\n\t\t\t\twid, rid, domainID := p.getMsgWithInfo(key)\n\t\t\t\tp.logger.Error(\"ES request failed.\",\n\t\t\t\t\ttag.ESResponseStatus(resp.Status), tag.ESResponseError(getErrorMsgFromESResp(resp)), tag.WorkflowID(wid), tag.WorkflowRunID(rid),\n\t\t\t\t\ttag.WorkflowDomainID(domainID))\n\t\t\t\tp.nackKafkaMsg(key)\n\t\t\tdefault: // bulk bulkProcessor will retry\n\t\t\t\tp.logger.Info(\"ES request retried.\", tag.ESResponseStatus(resp.Status))\n\t\t\t\tp.scope.IncCounter(metrics.ESProcessorRetries)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (p *ESProcessorImpl) ackKafkaMsg(key string) {\n\tp.ackKafkaMsgHelper(key, false)\n}\n\nfunc (p *ESProcessorImpl) nackKafkaMsg(key string) {\n\tp.ackKafkaMsgHelper(key, true)\n}\n\nfunc (p *ESProcessorImpl) ackKafkaMsgHelper(key string, nack bool) {\n\tkafkaMsg, ok := p.getKafkaMsg(key)\n\tif !ok {\n\t\treturn\n\t}\n\n\tif nack {\n\t\tkafkaMsg.Nack()\n\t} else {\n\t\tkafkaMsg.Ack()\n\t}\n\n\tp.mapToKafkaMsg.Remove(key)\n}\n\nfunc (p *ESProcessorImpl) getKafkaMsg(key string) (kafkaMsg *kafkaMessageWithMetrics, ok bool) {\n\tmsg, ok := p.mapToKafkaMsg.Get(key)\n\tif !ok {\n\t\treturn // duplicate kafka message\n\t}\n\tkafkaMsg, ok = msg.(*kafkaMessageWithMetrics)\n\tif !ok { // must be bug in code and bad deployment\n\t\tp.logger.Fatal(\"Message is not kafka message.\", tag.ESKey(key))\n\t}\n\treturn kafkaMsg, ok\n}\n\nfunc (p *ESProcessorImpl) retrieveKafkaKey(request bulk.GenericBulkableRequest) string {\n\treq, err := request.Source()\n\tif err != nil {\n\t\tp.logger.Error(\"Get request source err.\", tag.Error(err), tag.ESRequest(request.String()))\n\t\tp.scope.IncCounter(metrics.ESProcessorCorruptedData)\n\t\treturn \"\"\n\t}\n\n\tvar key string\n\tif !p.isDeleteRequest(req) { // index or update requests\n\t\tvar body map[string]interface{}\n\t\tif err := json.Unmarshal([]byte(req[1]), &body); err != nil {\n\t\t\tp.logger.Error(\"Unmarshal index request body err.\", tag.Error(err))\n\t\t\tp.scope.IncCounter(metrics.ESProcessorCorruptedData)\n\t\t\treturn \"\"\n\t\t}\n\n\t\tk, ok := body[definition.KafkaKey]\n\t\tif !ok {\n\t\t\t// must be bug in code and bad deployment, check processor that add es requests\n\t\t\tpanic(definition.KafkaKey + \" not found\")\n\t\t}\n\t\tkey, ok = k.(string)\n\t\tif !ok {\n\t\t\t// must be bug in code and bad deployment, check processor that add es requests\n\t\t\tpanic(\"kafkaKey is not string\")\n\t\t}\n\t} else { // delete requests\n\t\tvar body map[string]map[string]interface{}\n\t\tif err := json.Unmarshal([]byte(req[0]), &body); err != nil {\n\t\t\tp.logger.Error(\"Unmarshal delete request body err.\", tag.Error(err))\n\t\t\tp.scope.IncCounter(metrics.ESProcessorCorruptedData)\n\t\t\treturn \"\"\n\t\t}\n\t\topMap, ok := body[\"delete\"]\n\t\tif !ok {\n\t\t\t// must be bug, check if dependency changed\n\t\t\tpanic(\"delete key not found in request\")\n\t\t}\n\t\tk, ok := opMap[\"_id\"]\n\t\tif !ok {\n\t\t\t// must be bug in code and bad deployment, check processor that add es requests\n\t\t\tpanic(\"_id not found in request opMap\")\n\t\t}\n\t\tkey, _ = k.(string)\n\t}\n\treturn key\n}\n\nfunc (p *ESProcessorImpl) getMsgWithInfo(key string) (wid string, rid string, domainID string) {\n\tkafkaMsg, ok := p.getKafkaMsg(key)\n\tif !ok {\n\t\treturn\n\t}\n\n\tvar msg indexer.Message\n\tif err := p.msgEncoder.Decode(kafkaMsg.message.Value(), &msg); err != nil {\n\t\tp.logger.Error(\"failed to deserialize kafka message.\", tag.Error(err))\n\t\treturn\n\t}\n\treturn msg.GetWorkflowID(), msg.GetRunID(), msg.GetDomainID()\n}\n\nfunc (p *ESProcessorImpl) hashFn(key interface{}) uint32 {\n\tid, ok := key.(string)\n\tif !ok {\n\t\treturn 0\n\t}\n\tnumOfShards := p.config.IndexerConcurrency()\n\treturn uint32(common.WorkflowIDToHistoryShard(id, numOfShards))\n}\n\nfunc (p *ESProcessorImpl) isDeleteRequest(request []string) bool {\n\t// The Source() method typically returns a slice of strings, where each string represents a part of the bulk request in JSON format.\n\t// For delete operations, the Source() method typically returns only one part\n\t// The metadata that specifies the delete action, including _index and _id.\n\t// \"{\\\"delete\\\":{\\\"_index\\\":\\\"my-index\\\",\\\"_id\\\":\\\"1\\\"}}\"\n\t// For index/update operations, the Source() method typically returns two parts\n\t// reference: https://www.elastic.co/guide/en/elasticsearch/reference/6.8/docs-bulk.html\n\treturn len(request) == 1\n}\n\n// 409 - Version Conflict\n// 404 - Not Found\nfunc isResponseSuccess(status int) bool {\n\tif status >= 200 && status < 300 || status == 409 || status == 404 {\n\t\treturn true\n\t}\n\treturn false\n}\n\n// isResponseRetriable is complaint with GenericBulkProcessorService.RetryItemStatusCodes\n// responses with these status will be kept in queue and retried until success\n// 408 - Request Timeout\n// 429 - Too Many Requests\n// 500 - Node not connected\n// 503 - Service Unavailable\n// 507 - Insufficient Storage\nvar retryableStatusCode = map[int]struct{}{408: {}, 429: {}, 500: {}, 503: {}, 507: {}}\n\nfunc isResponseRetriable(status int) bool {\n\t_, ok := retryableStatusCode[status]\n\treturn ok\n}\n\nfunc getErrorMsgFromESResp(resp *bulk.GenericBulkResponseItem) string {\n\tvar errMsg string\n\tif resp.Error != nil {\n\t\terrMsg = fmt.Sprintf(\"%v\", resp.Error)\n\t}\n\treturn errMsg\n}\n\nfunc newKafkaMessageWithMetrics(kafkaMsg messaging.Message, stopwatch *metrics.Stopwatch) *kafkaMessageWithMetrics {\n\treturn &kafkaMessageWithMetrics{\n\t\tmessage:        kafkaMsg,\n\t\tswFromAddToAck: stopwatch,\n\t}\n}\n\nfunc (km *kafkaMessageWithMetrics) Ack() {\n\tkm.message.Ack() // nolint:errcheck\n\tif km.swFromAddToAck != nil {\n\t\tkm.swFromAddToAck.Stop()\n\t}\n}\n\nfunc (km *kafkaMessageWithMetrics) Nack() {\n\tkm.message.Nack() //nolint:errcheck\n\tif km.swFromAddToAck != nil {\n\t\tkm.swFromAddToAck.Stop()\n\t}\n}\n"
  },
  {
    "path": "service/worker/indexer/esProcessor_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage indexer\n\nimport (\n\t\"fmt\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\tmocks2 \"github.com/uber/cadence/common/elasticsearch/bulk/mocks\"\n\tesMocks \"github.com/uber/cadence/common/elasticsearch/mocks\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\tmsgMocks \"github.com/uber/cadence/common/messaging/mocks\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/metrics/mocks\"\n)\n\ntype esProcessorSuite struct {\n\tsuite.Suite\n\tesProcessor       *ESProcessorImpl\n\tmockBulkProcessor *mocks2.GenericBulkProcessor\n\tmockESClient      *esMocks.GenericClient\n\tmockScope         *mocks.Scope\n}\n\nvar (\n\ttestIndex     = \"test-index\"\n\ttestType      = elasticsearch.GetESDocType()\n\ttestID        = \"test-doc-id\"\n\ttestStopWatch = metrics.NoopScope.StartTimer(metrics.ESProcessorProcessMsgLatency)\n\ttestScope     = metrics.ESProcessorScope\n\ttestMetric    = metrics.ESProcessorProcessMsgLatency\n)\n\nfunc TestESProcessorSuite(t *testing.T) {\n\ts := new(esProcessorSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *esProcessorSuite) SetupSuite() {\n}\n\nfunc (s *esProcessorSuite) SetupTest() {\n\tconfig := &Config{\n\t\tIndexerConcurrency:       dynamicproperties.GetIntPropertyFn(32),\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t}\n\ts.mockBulkProcessor = &mocks2.GenericBulkProcessor{}\n\ts.mockScope = &mocks.Scope{}\n\n\tp := &ESProcessorImpl{\n\t\tconfig:     config,\n\t\tlogger:     testlogger.New(s.T()),\n\t\tscope:      s.mockScope,\n\t\tmsgEncoder: defaultEncoder,\n\t}\n\tp.mapToKafkaMsg = collection.NewShardedConcurrentTxMap(1024, p.hashFn)\n\tp.bulkProcessor = s.mockBulkProcessor\n\n\ts.esProcessor = p\n\n\ts.mockESClient = &esMocks.GenericClient{}\n}\n\nfunc (s *esProcessorSuite) TearDownTest() {\n\ts.mockBulkProcessor.AssertExpectations(s.T())\n\ts.mockESClient.AssertExpectations(s.T())\n}\n\nfunc (s *esProcessorSuite) TestNewESProcessorAndStart() {\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t}\n\tprocessorName := \"test-bulkProcessor\"\n\n\ts.mockESClient.On(\"RunBulkProcessor\", mock.Anything, mock.MatchedBy(func(input *bulk.BulkProcessorParameters) bool {\n\t\ts.Equal(processorName, input.Name)\n\t\ts.Equal(config.ESProcessorNumOfWorkers(), input.NumOfWorkers)\n\t\ts.Equal(config.ESProcessorBulkActions(), input.BulkActions)\n\t\ts.Equal(config.ESProcessorBulkSize(), input.BulkSize)\n\t\ts.Equal(config.ESProcessorFlushInterval(), input.FlushInterval)\n\t\ts.NotNil(input.Backoff)\n\t\ts.NotNil(input.AfterFunc)\n\t\treturn true\n\t})).Return(&mocks2.GenericBulkProcessor{}, nil).Once()\n\tprocessor, err := newESProcessor(processorName, config, s.mockESClient, s.esProcessor.logger, metrics.NewNoopMetricsClient())\n\ts.NoError(err)\n\n\ts.NotNil(processor.mapToKafkaMsg)\n}\n\nfunc (s *esProcessorSuite) TestStop() {\n\ts.mockBulkProcessor.On(\"Stop\").Return(nil).Once()\n\ts.esProcessor.Stop()\n\ts.Nil(s.esProcessor.mapToKafkaMsg)\n}\n\nfunc (s *esProcessorSuite) TestAdd() {\n\trequest := &bulk.GenericBulkableAddRequest{RequestType: bulk.BulkableIndexRequest}\n\tmockKafkaMsg := &msgMocks.Message{}\n\tkey := \"test-key\"\n\ts.Equal(0, s.esProcessor.mapToKafkaMsg.Len())\n\n\ts.mockBulkProcessor.On(\"Add\", request).Return().Once()\n\ts.mockScope.On(\"StartTimer\", testMetric).Return(testStopWatch).Once()\n\ts.esProcessor.Add(request, key, mockKafkaMsg)\n\ts.Equal(1, s.esProcessor.mapToKafkaMsg.Len())\n\tmockKafkaMsg.AssertExpectations(s.T())\n\n\t// handle duplicate\n\tmockKafkaMsg.On(\"Ack\").Return(nil).Once()\n\ts.mockScope.On(\"StartTimer\", testMetric).Return(testStopWatch).Once()\n\ts.esProcessor.Add(request, key, mockKafkaMsg)\n\ts.Equal(1, s.esProcessor.mapToKafkaMsg.Len())\n\tmockKafkaMsg.AssertExpectations(s.T())\n}\n\nfunc (s *esProcessorSuite) TestAdd_ConcurrentAdd() {\n\trequest := &bulk.GenericBulkableAddRequest{RequestType: bulk.BulkableIndexRequest}\n\tmockKafkaMsg := &msgMocks.Message{}\n\tkey := \"test-key\"\n\n\taddFunc := func(wg *sync.WaitGroup) {\n\t\ts.mockScope.On(\"StartTimer\", testMetric).Return(testStopWatch).Once()\n\t\ts.esProcessor.Add(request, key, mockKafkaMsg)\n\t\twg.Done()\n\t}\n\tduplicates := 5\n\twg := &sync.WaitGroup{}\n\twg.Add(duplicates)\n\ts.mockBulkProcessor.On(\"Add\", request).Return().Once()\n\tmockKafkaMsg.On(\"Ack\").Return(nil).Times(duplicates - 1)\n\tfor i := 0; i < duplicates; i++ {\n\t\taddFunc(wg)\n\t}\n\twg.Wait()\n\tmockKafkaMsg.AssertExpectations(s.T())\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterActionX() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{string(`{\"delete\":{\"_id\":\"testKey\"}}`)}, nil)\n\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmSuccess := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"index\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  200,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: false,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mSuccess},\n\t}\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\tmockKafkaMsg.On(\"Ack\").Return(nil).Once()\n\ts.esProcessor.bulkAfterAction(0, requests, response, nil)\n\tmockKafkaMsg.AssertExpectations(s.T())\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterAction_Nack() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{string(`{\"delete\":{\"_id\":\"testKey\"}}`)}, nil)\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmFailed := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"index\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  400,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: false,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mFailed},\n\t}\n\n\twid := \"test-workflowID\"\n\trid := \"test-runID\"\n\tdomainID := \"test-domainID\"\n\tpayload := s.getEncodedMsg(wid, rid, domainID)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\t// s.mockBulkProcessor.On(\"RetrieveKafkaKey\", request, mock.Anything, mock.Anything).Return(testKey)\n\ts.esProcessor.bulkAfterAction(0, requests, response, nil)\n\tmockKafkaMsg.AssertExpectations(s.T())\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterAction_Error() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{string(`{\"delete\":{\"_id\":\"testKey\"}}`)}, nil)\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmFailed := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"index\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  400,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: true,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mFailed},\n\t}\n\n\twid := \"test-workflowID\"\n\trid := \"test-runID\"\n\tdomainID := \"test-domainID\"\n\tpayload := s.getEncodedMsg(wid, rid, domainID)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\ts.mockScope.On(\"IncCounter\", metrics.ESProcessorFailures).Once()\n\ts.esProcessor.bulkAfterAction(0, requests, response, &bulk.GenericError{Details: fmt.Errorf(\"some error\")})\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterAction_Error_Nack() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{`{\"delete\":{\"_index\":\"test-index\",\"_id\":\"testKey\"}}`}, nil)\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmFailed := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"delete\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  409,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: true,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mFailed},\n\t}\n\n\twid := \"test-workflowID\"\n\trid := \"test-runID\"\n\tdomainID := \"test-domainID\"\n\tpayload := s.getEncodedMsg(wid, rid, domainID)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\tmockKafkaMsg.On(\"Ack\").Return(nil).Once() // Expect Ack to be called\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\ts.mockScope.On(\"IncCounter\", metrics.ESProcessorFailures).Once()\n\ts.esProcessor.bulkAfterAction(0, requests, response, &bulk.GenericError{Status: 404, Details: fmt.Errorf(\"some error\")})\n}\n\nfunc (s *esProcessorSuite) TestAckKafkaMsg() {\n\tkey := \"test-key\"\n\t// no msg in map, nothing called\n\ts.esProcessor.ackKafkaMsg(key)\n\n\trequest := &bulk.GenericBulkableAddRequest{}\n\tmockKafkaMsg := &msgMocks.Message{}\n\ts.mockScope.On(\"StartTimer\", testMetric).Return(testStopWatch).Once()\n\ts.mockBulkProcessor.On(\"Add\", request).Return().Once()\n\ts.esProcessor.Add(request, key, mockKafkaMsg)\n\ts.Equal(1, s.esProcessor.mapToKafkaMsg.Len())\n\n\tmockKafkaMsg.On(\"Ack\").Return(nil).Once()\n\ts.esProcessor.ackKafkaMsg(key)\n\tmockKafkaMsg.AssertExpectations(s.T())\n\ts.Equal(0, s.esProcessor.mapToKafkaMsg.Len())\n}\n\nfunc (s *esProcessorSuite) TestNackKafkaMsg() {\n\tkey := \"test-key-nack\"\n\t// no msg in map, nothing called\n\ts.esProcessor.nackKafkaMsg(key)\n\n\trequest := &bulk.GenericBulkableAddRequest{}\n\tmockKafkaMsg := &msgMocks.Message{}\n\ts.mockBulkProcessor.On(\"Add\", request).Return().Once()\n\ts.mockScope.On(\"StartTimer\", testMetric).Return(testStopWatch).Once()\n\ts.esProcessor.Add(request, key, mockKafkaMsg)\n\ts.Equal(1, s.esProcessor.mapToKafkaMsg.Len())\n\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\ts.esProcessor.nackKafkaMsg(key)\n\tmockKafkaMsg.AssertExpectations(s.T())\n\ts.Equal(0, s.esProcessor.mapToKafkaMsg.Len())\n}\n\nfunc (s *esProcessorSuite) TestHashFn() {\n\ts.Equal(uint32(0), s.esProcessor.hashFn(0))\n\ts.NotEqual(uint32(0), s.esProcessor.hashFn(\"test\"))\n}\n\nfunc (s *esProcessorSuite) getEncodedMsg(wid string, rid string, domainID string) []byte {\n\tindexMsg := &indexer.Message{\n\t\tDomainID:   common.StringPtr(domainID),\n\t\tWorkflowID: common.StringPtr(wid),\n\t\tRunID:      common.StringPtr(rid),\n\t}\n\tpayload, err := s.esProcessor.msgEncoder.Encode(indexMsg)\n\ts.NoError(err)\n\treturn payload\n}\n\nfunc (s *esProcessorSuite) TestGetMsgWithInfo() {\n\ttestKey := \"test-key\"\n\ttestWid := \"test-workflowID\"\n\ttestRid := \"test-runID\"\n\ttestDomainid := \"test-domainID\"\n\tpayload := s.getEncodedMsg(testWid, testRid, testDomainid)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\twid, rid, domainID := s.esProcessor.getMsgWithInfo(testKey)\n\ts.Equal(testWid, wid)\n\ts.Equal(testRid, rid)\n\ts.Equal(testDomainid, domainID)\n}\n\nfunc (s *esProcessorSuite) TestGetMsgInfo_Error() {\n\ttestKey := \"test-key\"\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmockKafkaMsg.On(\"Value\").Return([]byte{}).Once()\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\twid, rid, domainID := s.esProcessor.getMsgWithInfo(testKey)\n\ts.Equal(\"\", wid)\n\ts.Equal(\"\", rid)\n\ts.Equal(\"\", domainID)\n}\n\nfunc (s *esProcessorSuite) TestIsResponseSuccess() {\n\tfor i := 200; i < 300; i++ {\n\t\ts.True(isResponseSuccess(i))\n\t}\n\tstatus := []int{409, 404}\n\tfor _, code := range status {\n\t\ts.True(isResponseSuccess(code))\n\t}\n\tstatus = []int{100, 199, 300, 400, 500, 408, 429, 503, 507}\n\tfor _, code := range status {\n\t\ts.False(isResponseSuccess(code))\n\t}\n}\n\nfunc (s *esProcessorSuite) TestIsResponseRetriable() {\n\tstatus := []int{408, 429, 500, 503, 507}\n\tfor _, code := range status {\n\t\ts.True(isResponseRetriable(code))\n\t}\n}\n\nfunc (s *esProcessorSuite) TestIsErrorRetriable() {\n\ttests := []struct {\n\t\tinput    *bulk.GenericError\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tinput:    &bulk.GenericError{Status: 400},\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tinput:    &bulk.GenericError{Status: 408},\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tinput:    &bulk.GenericError{},\n\t\t\texpected: false,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\ts.Equal(test.expected, isResponseRetriable(test.input.Status))\n\t}\n}\n\nfunc (s *esProcessorSuite) TestIsDeleteRequest() {\n\ttests := []struct {\n\t\trequest   bulk.GenericBulkableRequest\n\t\tbIsDelete bool\n\t}{\n\t\t{\n\t\t\trequest: bulk.NewBulkIndexRequest().\n\t\t\t\tID(\"request.ID\").\n\t\t\t\tIndex(\"request.Index\").\n\t\t\t\tVersion(int64(0)).\n\t\t\t\tVersionType(\"request.VersionType\").Doc(\"request.Doc\"),\n\t\t\tbIsDelete: false,\n\t\t},\n\t\t{\n\t\t\trequest: bulk.NewBulkDeleteRequest().\n\t\t\t\tID(\"request.ID\").\n\t\t\t\tIndex(\"request.Index\"),\n\t\t\tbIsDelete: true,\n\t\t},\n\t}\n\tfor _, test := range tests {\n\t\treq, _ := test.request.Source()\n\t\ts.Equal(test.bIsDelete, s.esProcessor.isDeleteRequest(req))\n\t}\n}\n\nfunc (s *esProcessorSuite) TestIsDeleteRequest_Error() {\n\trequest := &MockBulkableRequest{}\n\ts.mockScope.On(\"IncCounter\", mock.AnythingOfType(\"int\")).Return()\n\treq, err := request.Source()\n\ts.False(s.esProcessor.isDeleteRequest(req))\n\ts.Error(err)\n}\n\n// MockBulkableRequest is a mock implementation of the GenericBulkableRequest interface\ntype MockBulkableRequest struct{}\n\n// String returns a mock string\nfunc (m *MockBulkableRequest) String() string {\n\treturn \"mock request\"\n}\n\n// Source returns an error to simulate a failure\nfunc (m *MockBulkableRequest) Source() ([]string, error) {\n\treturn nil, fmt.Errorf(\"simulated source error\")\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterAction_Nack_Shadow_WithError() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{string(`{\"delete\":{\"_id\":\"testKey\"}}`)}, nil)\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmFailed := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"index\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  400,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: false,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mFailed},\n\t}\n\n\t// Mock error to be passed to the after action functions\n\tmockErr := &bulk.GenericError{\n\t\tStatus:  500,\n\t\tDetails: fmt.Errorf(\"Test error occurred\"),\n\t}\n\n\twid := \"test-workflowID\"\n\trid := \"test-runID\"\n\tdomainID := \"test-domainID\"\n\tpayload := s.getEncodedMsg(wid, rid, domainID)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\n\t// Mock Kafka message Nack and Value\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\ts.mockScope.On(\"IncCounter\", mock.AnythingOfType(\"metrics.MetricIdx\")).Return()\n\t// Execute bulkAfterAction for primary processor with error\n\ts.esProcessor.bulkAfterAction(0, requests, response, mockErr)\n}\n\nfunc (s *esProcessorSuite) TestBulkAfterAction_Shadow_Fail_WithoutError() {\n\tversion := int64(3)\n\ttestKey := \"testKey\"\n\trequest := &mocks2.GenericBulkableRequest{}\n\trequest.On(\"String\").Return(\"\")\n\trequest.On(\"Source\").Return([]string{string(`{\"delete\":{\"_id\":\"testKey\"}}`)}, nil)\n\trequests := []bulk.GenericBulkableRequest{request}\n\n\tmFailed := map[string]*bulk.GenericBulkResponseItem{\n\t\t\"index\": {\n\t\t\tIndex:   testIndex,\n\t\t\tType:    testType,\n\t\t\tID:      testID,\n\t\t\tVersion: version,\n\t\t\tStatus:  400,\n\t\t},\n\t}\n\tresponse := &bulk.GenericBulkResponse{\n\t\tTook:   3,\n\t\tErrors: false,\n\t\tItems:  []map[string]*bulk.GenericBulkResponseItem{mFailed},\n\t}\n\n\twid := \"test-workflowID\"\n\trid := \"test-runID\"\n\tdomainID := \"test-domainID\"\n\tpayload := s.getEncodedMsg(wid, rid, domainID)\n\n\tmockKafkaMsg := &msgMocks.Message{}\n\tmapVal := newKafkaMessageWithMetrics(mockKafkaMsg, &testStopWatch)\n\ts.esProcessor.mapToKafkaMsg.Put(testKey, mapVal)\n\n\t// Mock Kafka message Nack and Value\n\tmockKafkaMsg.On(\"Nack\").Return(nil).Once()\n\tmockKafkaMsg.On(\"Value\").Return(payload).Once()\n\ts.mockScope.On(\"IncCounter\", mock.AnythingOfType(\"int\")).Return()\n\t// Execute bulkAfterAction for primary processor with error\n\ts.esProcessor.bulkAfterAction(0, requests, response, nil)\n}\n"
  },
  {
    "path": "service/worker/indexer/indexer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination indexer_mock.go github.com/uber/cadence/service/worker/indexer ESProcessor\n\npackage indexer\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tversionTypeExternal    = \"external\"\n\tprocessorName          = \"visibility-processor\"\n\tmigrationProcessorName = \"migration-visibility-processor\"\n)\n\nvar (\n\terrUnknownMessageType = &types.BadRequestError{Message: \"unknown message type\"}\n\tdefaultEncoder        = codec.NewThriftRWEncoder()\n)\n\ntype (\n\tESProcessor interface {\n\t\tcommon.Daemon\n\t\tAdd(request *bulk.GenericBulkableAddRequest, key string, kafkaMsg messaging.Message)\n\t}\n\t// Indexer used to consumer data from kafka then send to ElasticSearch\n\tIndexer struct {\n\t\tesIndexName         string\n\t\tconsumer            messaging.Consumer\n\t\tvisibilityProcessor ESProcessor\n\t\tconfig              *Config\n\t\tlogger              log.Logger\n\t\tscope               metrics.Scope\n\t\tmsgEncoder          codec.BinaryEncoder\n\n\t\tisStarted  int32\n\t\tisStopped  int32\n\t\tshutdownWG sync.WaitGroup\n\t\tshutdownCh chan struct{}\n\t}\n\n\tDualIndexer struct {\n\t\tSourceIndexer *Indexer\n\t\tDestIndexer   *Indexer\n\t}\n\n\t// Config contains all configs for indexer\n\tConfig struct {\n\t\tIndexerConcurrency             dynamicproperties.IntPropertyFn\n\t\tESProcessorNumOfWorkers        dynamicproperties.IntPropertyFn\n\t\tESProcessorBulkActions         dynamicproperties.IntPropertyFn // max number of requests in bulk\n\t\tESProcessorBulkSize            dynamicproperties.IntPropertyFn // max total size of bytes in bulk\n\t\tESProcessorFlushInterval       dynamicproperties.DurationPropertyFn\n\t\tValidSearchAttributes          dynamicproperties.MapPropertyFn\n\t\tEnableQueryAttributeValidation dynamicproperties.BoolPropertyFn\n\t}\n)\n\n// NewIndexer create a new Indexer\nfunc NewIndexer(\n\tconfig *Config,\n\tclient messaging.Client,\n\tvisibilityClient es.GenericClient,\n\tvisibilityName string,\n\tconsumerName string,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n) *Indexer {\n\tlogger = logger.WithTags(tag.ComponentIndexer)\n\n\tvisibilityProcessor, err := newESProcessor(processorName, config, visibilityClient, logger, metricsClient)\n\tif err != nil {\n\t\tlogger.Fatal(\"Index ES processor state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\tif consumerName == \"\" {\n\t\tconsumerName = getConsumerName(visibilityName)\n\t}\n\n\tconsumer, err := client.NewConsumer(constants.VisibilityAppName, consumerName)\n\tif err != nil {\n\t\tlogger.Fatal(\"Index consumer state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\treturn &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         visibilityName,\n\t\tconsumer:            consumer,\n\t\tlogger:              logger.WithTags(tag.ComponentIndexerProcessor),\n\t\tscope:               metricsClient.Scope(metrics.IndexProcessorScope),\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: visibilityProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n}\n\nfunc getConsumerName(topic string) string {\n\treturn fmt.Sprintf(\"%s-consumer\", topic)\n}\n\nfunc (i *Indexer) Start() error {\n\tif !atomic.CompareAndSwapInt32(&i.isStarted, 0, 1) {\n\t\treturn nil\n\t}\n\n\tif err := i.consumer.Start(); err != nil {\n\t\ti.logger.Info(\"Index consumer state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t\treturn err\n\t}\n\n\ti.visibilityProcessor.Start()\n\n\ti.shutdownWG.Add(1)\n\tgo i.processorPump()\n\n\ti.logger.Info(\"Index bulkProcessor state changed\", tag.LifeCycleStarted)\n\treturn nil\n}\n\nfunc (i *Indexer) Stop() {\n\tif !atomic.CompareAndSwapInt32(&i.isStopped, 0, 1) {\n\t\treturn\n\t}\n\n\ti.logger.Info(\"Index bulkProcessor state changed\", tag.LifeCycleStopping)\n\tdefer i.logger.Info(\"Index bulkProcessor state changed\", tag.LifeCycleStopped)\n\n\tif atomic.LoadInt32(&i.isStarted) == 1 {\n\t\tclose(i.shutdownCh)\n\t}\n\n\tif success := common.AwaitWaitGroup(&i.shutdownWG, time.Minute); !success {\n\t\ti.logger.Info(\"Index bulkProcessor state changed\", tag.LifeCycleStopTimedout)\n\t}\n}\n\nfunc (i *Indexer) processorPump() {\n\tdefer i.shutdownWG.Done()\n\n\tvar workerWG sync.WaitGroup\n\tfor workerID := 0; workerID < i.config.IndexerConcurrency(); workerID++ {\n\t\tworkerWG.Add(1)\n\t\tgo i.messageProcessLoop(&workerWG)\n\t}\n\n\t<-i.shutdownCh\n\t// Processor is shutting down, close the underlying consumer and esProcessor\n\ti.consumer.Stop()\n\ti.visibilityProcessor.Stop()\n\n\ti.logger.Info(\"Index bulkProcessor pump shutting down.\")\n\tif success := common.AwaitWaitGroup(&workerWG, 10*time.Second); !success {\n\t\ti.logger.Warn(\"Index bulkProcessor timed out on worker shutdown.\")\n\t}\n}\n\nfunc (i *Indexer) messageProcessLoop(workerWG *sync.WaitGroup) {\n\tdefer workerWG.Done()\n\n\tfor msg := range i.consumer.Messages() {\n\t\tsw := i.scope.StartTimer(metrics.IndexProcessorProcessMsgLatency)\n\t\terr := i.process(msg)\n\t\tsw.Stop()\n\t\tif err != nil {\n\t\t\tmsg.Nack() //nolint:errcheck\n\t\t}\n\t}\n}\n\nfunc (i *Indexer) process(kafkaMsg messaging.Message) error {\n\tlogger := i.logger.WithTags(tag.KafkaPartition(kafkaMsg.Partition()), tag.KafkaOffset(kafkaMsg.Offset()), tag.AttemptStart(time.Now()))\n\n\tindexMsg, err := i.deserialize(kafkaMsg.Value())\n\tif err != nil {\n\t\tlogger.Error(\"Failed to deserialize index messages.\", tag.Error(err))\n\t\ti.scope.IncCounter(metrics.IndexProcessorCorruptedData)\n\t\treturn err\n\t}\n\n\treturn i.addMessageToES(indexMsg, kafkaMsg, logger)\n}\n\nfunc (i *Indexer) deserialize(payload []byte) (*indexer.Message, error) {\n\tvar msg indexer.Message\n\tif err := i.msgEncoder.Decode(payload, &msg); err != nil {\n\t\treturn nil, err\n\t}\n\treturn &msg, nil\n}\n\nfunc (i *Indexer) addMessageToES(indexMsg *indexer.Message, kafkaMsg messaging.Message, logger log.Logger) error {\n\tdocID := es.GenerateDocID(indexMsg.GetWorkflowID(), indexMsg.GetRunID())\n\t// check and skip invalid docID\n\tif len(docID) >= es.GetESDocIDSizeLimit() {\n\t\tlogger.Error(\"Index message is too long\",\n\t\t\ttag.WorkflowDomainID(indexMsg.GetDomainID()),\n\t\t\ttag.WorkflowID(indexMsg.GetWorkflowID()),\n\t\t\ttag.WorkflowRunID(indexMsg.GetRunID()))\n\t\tkafkaMsg.Nack()\n\t\treturn nil\n\t}\n\n\tvar keyToKafkaMsg string\n\treq := &bulk.GenericBulkableAddRequest{\n\t\tIndex:       i.esIndexName,\n\t\tType:        es.GetESDocType(),\n\t\tID:          docID,\n\t\tVersionType: versionTypeExternal,\n\t\tVersion:     indexMsg.GetVersion(),\n\t}\n\tswitch indexMsg.GetMessageType() {\n\tcase indexer.MessageTypeIndex:\n\t\tkeyToKafkaMsg = fmt.Sprintf(\"%v-%v\", kafkaMsg.Partition(), kafkaMsg.Offset())\n\t\treq.Doc = i.generateESDoc(indexMsg, keyToKafkaMsg)\n\t\treq.RequestType = bulk.BulkableIndexRequest\n\tcase indexer.MessageTypeDelete:\n\t\tkeyToKafkaMsg = docID\n\t\treq.RequestType = bulk.BulkableDeleteRequest\n\tcase indexer.MessageTypeCreate:\n\t\tkeyToKafkaMsg = fmt.Sprintf(\"%v-%v\", kafkaMsg.Partition(), kafkaMsg.Offset())\n\t\treq.Doc = i.generateESDoc(indexMsg, keyToKafkaMsg)\n\t\treq.RequestType = bulk.BulkableCreateRequest\n\tdefault:\n\t\tlogger.Error(\"Unknown message type\")\n\t\ti.scope.IncCounter(metrics.IndexProcessorCorruptedData)\n\t\treturn errUnknownMessageType\n\t}\n\n\ti.visibilityProcessor.Add(req, keyToKafkaMsg, kafkaMsg)\n\n\treturn nil\n}\n\nfunc (i *Indexer) generateESDoc(msg *indexer.Message, keyToKafkaMsg string) map[string]interface{} {\n\tdoc := i.dumpFieldsToMap(msg.Fields, msg.GetDomainID())\n\tfulfillDoc(doc, msg, keyToKafkaMsg)\n\treturn doc\n}\n\nfunc (i *Indexer) decodeSearchAttrBinary(bytes []byte, key string) interface{} {\n\tvar val interface{}\n\terr := json.Unmarshal(bytes, &val)\n\tif err != nil {\n\t\ti.logger.Error(\"Error when decode search attributes values.\", tag.Error(err), tag.ESField(key))\n\t\ti.scope.IncCounter(metrics.IndexProcessorCorruptedData)\n\t}\n\treturn val\n}\n\nfunc (i *Indexer) dumpFieldsToMap(fields map[string]*indexer.Field, domainID string) map[string]interface{} {\n\tdoc := make(map[string]interface{})\n\tattr := make(map[string]interface{})\n\tfor k, v := range fields {\n\t\tif !i.isValidFieldToES(k) {\n\t\t\ti.logger.Error(\"Unregistered field.\", tag.ESField(k), tag.WorkflowDomainID(domainID))\n\t\t\ti.scope.IncCounter(metrics.IndexProcessorCorruptedData)\n\t\t\tcontinue\n\t\t}\n\n\t\t// skip VisibilityOperation since it’s not being used for advanced visibility\n\t\tif k == es.VisibilityOperation {\n\t\t\tcontinue\n\t\t}\n\n\t\tswitch v.GetType() {\n\t\tcase indexer.FieldTypeString:\n\t\t\tdoc[k] = v.GetStringData()\n\t\tcase indexer.FieldTypeInt:\n\t\t\tdoc[k] = v.GetIntData()\n\t\tcase indexer.FieldTypeBool:\n\t\t\tdoc[k] = v.GetBoolData()\n\t\tcase indexer.FieldTypeBinary:\n\t\t\tif k == definition.Memo {\n\t\t\t\tdoc[k] = v.GetBinaryData()\n\t\t\t} else { // custom search attributes\n\t\t\t\tattr[k] = i.decodeSearchAttrBinary(v.GetBinaryData(), k)\n\t\t\t}\n\t\tdefault:\n\t\t\t// there must be bug in code and bad deployment, check data sent from producer\n\t\t\ti.logger.Fatal(\"Unknown field type\")\n\t\t}\n\t}\n\tdoc[definition.Attr] = attr\n\treturn doc\n}\n\nfunc (i *Indexer) isValidFieldToES(field string) bool {\n\tif !i.config.EnableQueryAttributeValidation() {\n\t\treturn true\n\t}\n\tif _, ok := i.config.ValidSearchAttributes()[field]; ok {\n\t\treturn true\n\t}\n\tif field == definition.Memo || field == definition.KafkaKey || field == definition.Encoding || field == es.VisibilityOperation {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc fulfillDoc(doc map[string]interface{}, msg *indexer.Message, keyToKafkaMsg string) {\n\tdoc[definition.DomainID] = msg.GetDomainID()\n\tdoc[definition.WorkflowID] = msg.GetWorkflowID()\n\tdoc[definition.RunID] = msg.GetRunID()\n\tdoc[definition.KafkaKey] = keyToKafkaMsg\n}\n"
  },
  {
    "path": "service/worker/indexer/indexer_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: indexer.go\n//\n// Generated by this command:\n//\n//\tmockgen -package indexer -source indexer.go -destination indexer_mock.go github.com/uber/cadence/service/worker/indexer ESProcessor\n//\n\n// Package indexer is a generated GoMock package.\npackage indexer\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tbulk \"github.com/uber/cadence/common/elasticsearch/bulk\"\n\tmessaging \"github.com/uber/cadence/common/messaging\"\n)\n\n// MockESProcessor is a mock of ESProcessor interface.\ntype MockESProcessor struct {\n\tctrl     *gomock.Controller\n\trecorder *MockESProcessorMockRecorder\n\tisgomock struct{}\n}\n\n// MockESProcessorMockRecorder is the mock recorder for MockESProcessor.\ntype MockESProcessorMockRecorder struct {\n\tmock *MockESProcessor\n}\n\n// NewMockESProcessor creates a new mock instance.\nfunc NewMockESProcessor(ctrl *gomock.Controller) *MockESProcessor {\n\tmock := &MockESProcessor{ctrl: ctrl}\n\tmock.recorder = &MockESProcessorMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockESProcessor) EXPECT() *MockESProcessorMockRecorder {\n\treturn m.recorder\n}\n\n// Add mocks base method.\nfunc (m *MockESProcessor) Add(request *bulk.GenericBulkableAddRequest, key string, kafkaMsg messaging.Message) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Add\", request, key, kafkaMsg)\n}\n\n// Add indicates an expected call of Add.\nfunc (mr *MockESProcessorMockRecorder) Add(request, key, kafkaMsg any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Add\", reflect.TypeOf((*MockESProcessor)(nil).Add), request, key, kafkaMsg)\n}\n\n// Start mocks base method.\nfunc (m *MockESProcessor) Start() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Start\")\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockESProcessorMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockESProcessor)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockESProcessor) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockESProcessorMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockESProcessor)(nil).Stop))\n}\n"
  },
  {
    "path": "service/worker/indexer/indexer_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage indexer\n\nimport (\n\t\"fmt\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/definition\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/elasticsearch/bulk\"\n\tmocks2 \"github.com/uber/cadence/common/elasticsearch/bulk/mocks\"\n\tesMocks \"github.com/uber/cadence/common/elasticsearch/mocks\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nfunc TestNewDualIndexer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t}\n\tprocessorName := \"test-bulkProcessor\"\n\tconsumerName := \"test-bulkProcessor-os-consumer\"\n\tmockESClient := &esMocks.GenericClient{}\n\tmockESClient.On(\"RunBulkProcessor\", mock.Anything, mock.MatchedBy(func(input *bulk.BulkProcessorParameters) bool {\n\t\treturn true\n\t})).Return(&mocks2.GenericBulkProcessor{}, nil).Times(2)\n\n\tmockMessagingClient := messaging.NewMockClient(ctrl)\n\tmockMessagingClient.EXPECT().NewConsumer(\"visibility\", \"test-bulkProcessor-consumer\").Return(nil, nil).Times(1)\n\tmockMessagingClient.EXPECT().NewConsumer(\"visibility\", \"test-bulkProcessor-os-consumer\").Return(nil, nil).Times(1)\n\n\tindexer := NewMigrationDualIndexer(config, mockMessagingClient, mockESClient, mockESClient, processorName, processorName, \"\", consumerName, testlogger.New(t), metrics.NewNoopMetricsClient())\n\tassert.NotNil(t, indexer)\n}\n\nfunc TestNewIndexer(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tdefer ctrl.Finish()\n\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t}\n\tprocessorName := \"test-bulkProcessor\"\n\tmockESClient := &esMocks.GenericClient{}\n\tmockESClient.On(\"RunBulkProcessor\", mock.Anything, mock.MatchedBy(func(input *bulk.BulkProcessorParameters) bool {\n\t\treturn true\n\t})).Return(&mocks2.GenericBulkProcessor{}, nil).Times(2)\n\n\tmockMessagingClient := messaging.NewMockClient(ctrl)\n\tmockMessagingClient.EXPECT().NewConsumer(\"visibility\", \"test-bulkProcessor-consumer\").Return(nil, nil).Times(1)\n\n\tindexer := NewIndexer(config, mockMessagingClient, mockESClient, processorName, \"\", testlogger.New(t), metrics.NewNoopMetricsClient())\n\tassert.NotNil(t, indexer)\n}\n\n// TestIndexerStart tests the Start method of Indexer\nfunc TestIndexerStart(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t\tIndexerConcurrency:       dynamicproperties.GetIntPropertyFn(1),\n\t}\n\tmockConsumer := messaging.NewMockConsumer(ctrl)\n\tmockConsumer.EXPECT().Start().Return(nil).Times(1)\n\tmessageChan := make(chan messaging.Message)\n\tmockConsumer.EXPECT().Messages().Return((<-chan messaging.Message)(messageChan)).Times(1)\n\tmockConsumer.EXPECT().Stop().Return().Times(1)\n\tmockvisibiltyProcessor := NewMockESProcessor(ctrl)\n\tmockvisibiltyProcessor.EXPECT().Start().Return().Times(1)\n\tmockvisibiltyProcessor.EXPECT().Stop().Return().Times(1)\n\n\tindexer := &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         \"test-index\",\n\t\tconsumer:            mockConsumer,\n\t\tlogger:              log.NewNoop(),\n\t\tscope:               metrics.NoopScope,\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: mockvisibiltyProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n\terr := indexer.Start()\n\tassert.NoError(t, err)\n\tclose(messageChan)\n\n\tindexer.Stop()\n\tdefer goleak.VerifyNone(t)\n}\n\n// TestIndexerStart_ConsumerError tests the Start method when consumer.Start returns an error\nfunc TestIndexerStart_ConsumerError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t\tIndexerConcurrency:       dynamicproperties.GetIntPropertyFn(1),\n\t}\n\tmockConsumer := messaging.NewMockConsumer(ctrl)\n\tmockConsumer.EXPECT().Start().Return(fmt.Errorf(\"some error\")).Times(1)\n\tmockvisibiltyProcessor := NewMockESProcessor(ctrl)\n\n\tindexer := &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         \"test-index\",\n\t\tconsumer:            mockConsumer,\n\t\tlogger:              log.NewNoop(),\n\t\tscope:               metrics.NoopScope,\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: mockvisibiltyProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n\terr := indexer.Start()\n\tassert.ErrorContains(t, err, \"some error\")\n\n}\n\nfunc TestIndexerStop(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\tconfig := &Config{\n\t\tESProcessorNumOfWorkers:  dynamicproperties.GetIntPropertyFn(1),\n\t\tESProcessorBulkActions:   dynamicproperties.GetIntPropertyFn(10),\n\t\tESProcessorBulkSize:      dynamicproperties.GetIntPropertyFn(2 << 20),\n\t\tESProcessorFlushInterval: dynamicproperties.GetDurationPropertyFn(1 * time.Minute),\n\t\tIndexerConcurrency:       dynamicproperties.GetIntPropertyFn(1),\n\t}\n\n\t// Mock the messaging consumer\n\tmockConsumer := messaging.NewMockConsumer(ctrl)\n\tmessageChan := make(chan messaging.Message)\n\tmockConsumer.EXPECT().Messages().Return((<-chan messaging.Message)(messageChan)).AnyTimes()\n\t// No specific expectations for Start or Stop since they're not called in Stop()\n\n\t// Mock the visibility processor\n\tmockVisibilityProcessor := NewMockESProcessor(ctrl)\n\t// Create the Indexer instance with mocks\n\tindexer := &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         \"test-index\",\n\t\tconsumer:            mockConsumer,\n\t\tlogger:              log.NewNoop(),\n\t\tscope:               metrics.NoopScope,\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: mockVisibilityProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n\n\t// Simulate that the indexer was started\n\tatomic.StoreInt32(&indexer.isStarted, 1)\n\n\t// Simulate the processorPump goroutine that waits on shutdownCh\n\tindexer.shutdownWG.Add(1)\n\tgo func() {\n\t\tdefer indexer.shutdownWG.Done()\n\t\t<-indexer.shutdownCh\n\t}()\n\n\t// Call Stop and verify behavior\n\tindexer.Stop()\n\n\t// Verify that shutdownCh is closed\n\tselect {\n\tcase <-indexer.shutdownCh:\n\t\t// Expected: shutdownCh should be closed\n\tdefault:\n\t\tt.Error(\"shutdownCh is not closed\")\n\t}\n\n\t// Verify that the WaitGroup has completed\n\tsuccess := common.AwaitWaitGroup(&indexer.shutdownWG, time.Second)\n\tassert.True(t, success)\n\n\t// Verify that isStopped flag is set\n\tassert.Equal(t, int32(1), atomic.LoadInt32(&indexer.isStopped))\n\n\t// Call Stop again to ensure idempotency\n\tindexer.Stop()\n\tdefer goleak.VerifyNone(t)\n}\n\nfunc TestIsValidFieldToES(t *testing.T) {\n\ttests := map[string]struct {\n\t\ttestIndexer *Indexer\n\t\tfield       string\n\t\texpectedRes bool\n\t}{\n\t\t\"not EnableQueryAttributeValidation\": {\n\t\t\ttestIndexer: &Indexer{\n\t\t\t\tconfig: &Config{\n\t\t\t\t\tEnableQueryAttributeValidation: dynamicproperties.GetBoolPropertyFn(false),\n\t\t\t\t},\n\t\t\t},\n\t\t\tfield:       \"someField\",\n\t\t\texpectedRes: true,\n\t\t},\n\t\t\"field is valid\": {\n\t\t\ttestIndexer: &Indexer{\n\t\t\t\tconfig: &Config{\n\t\t\t\t\tEnableQueryAttributeValidation: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\t\tValidSearchAttributes:          dynamicproperties.GetMapPropertyFn(map[string]interface{}{\"someField\": \"ok\"}),\n\t\t\t\t},\n\t\t\t},\n\t\t\tfield:       \"someField\",\n\t\t\texpectedRes: true,\n\t\t},\n\t\t\"field is not valid, but meet definition\": {\n\t\t\ttestIndexer: &Indexer{\n\t\t\t\tconfig: &Config{\n\t\t\t\t\tEnableQueryAttributeValidation: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\t\tValidSearchAttributes:          dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t\t},\n\t\t\t},\n\t\t\tfield:       definition.Memo,\n\t\t\texpectedRes: true,\n\t\t},\n\t\t\"false\": {\n\t\t\ttestIndexer: &Indexer{\n\t\t\t\tconfig: &Config{\n\t\t\t\t\tEnableQueryAttributeValidation: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\t\tValidSearchAttributes:          dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t\t\t},\n\t\t\t},\n\t\t\tfield:       \"stuff\",\n\t\t\texpectedRes: false,\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := tc.testIndexer.isValidFieldToES(tc.field)\n\t\t\tassert.Equal(t, tc.expectedRes, res)\n\t\t})\n\t}\n}\n\nfunc TestFulfillDoc_AllFieldsPresent(t *testing.T) {\n\tdomainID := \"domain1\"\n\tworkflowID := \"workflow1\"\n\trunID := \"run1\"\n\tkeyToKafkaMsg := \"kafka-key-1\"\n\n\tdoc := map[string]interface{}{}\n\tmsg := &indexer.Message{\n\t\tDomainID:   &domainID,\n\t\tWorkflowID: &workflowID,\n\t\tRunID:      &runID,\n\t}\n\n\texpectedDoc := map[string]interface{}{\n\t\tdefinition.DomainID:   domainID,\n\t\tdefinition.WorkflowID: workflowID,\n\t\tdefinition.RunID:      runID,\n\t\tdefinition.KafkaKey:   keyToKafkaMsg,\n\t}\n\n\tfulfillDoc(doc, msg, keyToKafkaMsg)\n\tassert.Equal(t, expectedDoc, doc, \"fulfillDoc() result mismatch\")\n}\n\nfunc TestDumpFieldsToMap(t *testing.T) {\n\ttestIndexer := &Indexer{\n\t\tconfig: &Config{\n\t\t\tEnableQueryAttributeValidation: dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tValidSearchAttributes:          dynamicproperties.GetMapPropertyFn(map[string]interface{}{}),\n\t\t},\n\t\tlogger: log.NewNoop(),\n\t\tscope:  metrics.NoopScope,\n\t}\n\n\tstringPtr := \"string\"\n\n\ttests := map[string]struct {\n\t\tfields   map[string]*indexer.Field\n\t\texpected map[string]interface{}\n\t}{\n\t\t\"empty fields\": {\n\t\t\tfields: map[string]*indexer.Field{},\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"Attr\": map[string]interface{}{},\n\t\t\t},\n\t\t},\n\t\t\"different fields\": {\n\t\t\tfields: map[string]*indexer.Field{\n\t\t\t\t\"invalid\": {\n\t\t\t\t\tStringData: &stringPtr,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpected: map[string]interface{}{\n\t\t\t\t\"Attr\": map[string]interface{}{},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres := testIndexer.dumpFieldsToMap(tc.fields, \"id\")\n\t\t\tassert.Equal(t, tc.expected, res)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/indexer/migration_indexer.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage indexer\n\nimport (\n\t\"github.com/uber/cadence/common/constants\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/messaging\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n// NewMigrationDualIndexer create a new Indexer that will be used during visibility migration\n// When migrate from ES to OS, we will have this indexer to index to both ES and OS\nfunc NewMigrationDualIndexer(config *Config,\n\tclient messaging.Client,\n\tprimaryClient es.GenericClient,\n\tsecondaryClient es.GenericClient,\n\tprimaryVisibilityName string,\n\tsecondaryVisibilityName string,\n\tprimaryConsumerName string,\n\tsecondaryConsumerName string,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client) *DualIndexer {\n\n\tlogger = logger.WithTags(tag.ComponentIndexer)\n\n\tvisibilityProcessor, err := newESProcessor(processorName, config, primaryClient, logger, metricsClient)\n\tif err != nil {\n\t\tlogger.Fatal(\"Index ES processor state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\tif primaryConsumerName == \"\" {\n\t\tprimaryConsumerName = getConsumerName(primaryVisibilityName)\n\t}\n\tconsumer, err := client.NewConsumer(constants.VisibilityAppName, primaryConsumerName)\n\tif err != nil {\n\t\tlogger.Fatal(\"Index consumer state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\tsourceIndexer := &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         primaryVisibilityName,\n\t\tconsumer:            consumer,\n\t\tlogger:              logger.WithTags(tag.ComponentIndexerProcessor),\n\t\tscope:               metricsClient.Scope(metrics.IndexProcessorScope),\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: visibilityProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n\n\tsecondaryVisibilityProcessor, err := newESProcessor(migrationProcessorName, config, secondaryClient, logger, metricsClient)\n\tif err != nil {\n\t\tlogger.Fatal(\"Migration Index ES processor state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\tif secondaryConsumerName == \"\" {\n\t\tsecondaryConsumerName = getConsumerName(primaryVisibilityName)\n\t}\n\tsecondaryConsumer, err := client.NewConsumer(constants.VisibilityAppName, secondaryConsumerName)\n\tif err != nil {\n\t\tlogger.Fatal(\"Migration Index consumer state changed\", tag.LifeCycleStartFailed, tag.Error(err))\n\t}\n\n\tdestIndexer := &Indexer{\n\t\tconfig:              config,\n\t\tesIndexName:         secondaryVisibilityName,\n\t\tconsumer:            secondaryConsumer,\n\t\tlogger:              logger.WithTags(tag.ComponentIndexerProcessor),\n\t\tscope:               metricsClient.Scope(metrics.IndexProcessorScope),\n\t\tshutdownCh:          make(chan struct{}),\n\t\tvisibilityProcessor: secondaryVisibilityProcessor,\n\t\tmsgEncoder:          defaultEncoder,\n\t}\n\n\treturn &DualIndexer{\n\t\tSourceIndexer: sourceIndexer,\n\t\tDestIndexer:   destIndexer,\n\t}\n}\n\nfunc (i *DualIndexer) Start() error {\n\tif err := i.SourceIndexer.Start(); err != nil {\n\t\ti.SourceIndexer.Stop()\n\t\treturn err\n\t}\n\n\tif err := i.DestIndexer.Start(); err != nil {\n\t\ti.DestIndexer.Stop()\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc (i *DualIndexer) Stop() {\n\ti.SourceIndexer.Stop()\n\ti.DestIndexer.Stop()\n}\n"
  },
  {
    "path": "service/worker/parentclosepolicy/client.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage parentclosepolicy\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tcclient \"go.uber.org/cadence/client\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\n//go:generate mockgen -package=$GOPACKAGE -destination=client_mock.go -self_package=github.com/uber/cadence/service/worker/parentclosepolicy github.com/uber/cadence/service/worker/parentclosepolicy Client\n\ntype (\n\n\t// Client is used to send request to processor workflow\n\tClient interface {\n\t\tSendParentClosePolicyRequest(context.Context, Request) error\n\t}\n\n\tclientImpl struct {\n\t\tmetricsClient metrics.Client\n\t\tlogger        log.Logger\n\t\tcadenceClient cclient.Client\n\t\tnumWorkflows  int\n\t}\n)\n\nvar _ Client = (*clientImpl)(nil)\n\nconst (\n\tsignalTimeout    = 400 * time.Millisecond\n\tworkflowIDPrefix = \"parent-close-policy-workflow\"\n)\n\n// NewClient creates a new Client\nfunc NewClient(\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tpublicClient workflowserviceclient.Interface,\n\tnumWorkflows int,\n) Client {\n\treturn &clientImpl{\n\t\tmetricsClient: metricsClient,\n\t\tlogger:        logger,\n\t\tcadenceClient: cclient.NewClient(publicClient, constants.SystemLocalDomainName, &cclient.Options{}),\n\t\tnumWorkflows:  numWorkflows,\n\t}\n}\n\nfunc (c *clientImpl) SendParentClosePolicyRequest(\n\tctx context.Context,\n\trequest Request,\n) error {\n\trandomID := rand.Intn(c.numWorkflows)\n\tworkflowID := fmt.Sprintf(\"%v-%v\", workflowIDPrefix, randomID)\n\tworkflowOptions := cclient.StartWorkflowOptions{\n\t\tID:                              workflowID,\n\t\tTaskList:                        processorTaskListName,\n\t\tExecutionStartToCloseTimeout:    infiniteDuration,\n\t\tDecisionTaskStartToCloseTimeout: time.Minute,\n\t\tWorkflowIDReusePolicy:           cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t}\n\tsignalCtx, cancel := context.WithTimeout(ctx, signalTimeout)\n\tdefer cancel()\n\t_, err := c.cadenceClient.SignalWithStartWorkflow(signalCtx, workflowID, processorChannelName, request, workflowOptions, processorWFTypeName, nil)\n\treturn err\n}\n"
  },
  {
    "path": "service/worker/parentclosepolicy/client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/service/worker/parentclosepolicy (interfaces: Client)\n//\n// Generated by this command:\n//\n//\tmockgen -package=parentclosepolicy -destination=client_mock.go -self_package=github.com/uber/cadence/service/worker/parentclosepolicy github.com/uber/cadence/service/worker/parentclosepolicy Client\n//\n\n// Package parentclosepolicy is a generated GoMock package.\npackage parentclosepolicy\n\nimport (\n\tcontext \"context\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockClient is a mock of Client interface.\ntype MockClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientMockRecorder is the mock recorder for MockClient.\ntype MockClientMockRecorder struct {\n\tmock *MockClient\n}\n\n// NewMockClient creates a new mock instance.\nfunc NewMockClient(ctrl *gomock.Controller) *MockClient {\n\tmock := &MockClient{ctrl: ctrl}\n\tmock.recorder = &MockClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClient) EXPECT() *MockClientMockRecorder {\n\treturn m.recorder\n}\n\n// SendParentClosePolicyRequest mocks base method.\nfunc (m *MockClient) SendParentClosePolicyRequest(arg0 context.Context, arg1 Request) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"SendParentClosePolicyRequest\", arg0, arg1)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// SendParentClosePolicyRequest indicates an expected call of SendParentClosePolicyRequest.\nfunc (mr *MockClientMockRecorder) SendParentClosePolicyRequest(arg0, arg1 any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"SendParentClosePolicyRequest\", reflect.TypeOf((*MockClient)(nil).SendParentClosePolicyRequest), arg0, arg1)\n}\n"
  },
  {
    "path": "service/worker/parentclosepolicy/processor.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage parentclosepolicy\n\nimport (\n\t\"context\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/worker\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// BootstrapParams contains the set of params needed to bootstrap\n\t// the sub-system\n\tBootstrapParams struct {\n\t\t// Config contains the configuration for scanner\n\t\t// ServiceClient is an instance of cadence service client\n\t\tServiceClient workflowserviceclient.Interface\n\t\t// MetricsClient is an instance of metrics object for emitting stats\n\t\tMetricsClient metrics.Client\n\t\tLogger        log.Logger\n\t\t// TallyScope is an instance of tally metrics scope\n\t\tTallyScope tally.Scope\n\t\t// ClientBean is the collection of clients\n\t\tClientBean client.Bean\n\t\t// DomainCache is the cache for domain information and configuration\n\t\tDomainCache cache.DomainCache\n\t\t// NumWorkflows is the total number of workflows for processing parent close policy\n\t\tNumWorkflows int\n\t}\n\n\t// Processor is the background sub-system that execute workflow for ParentClosePolicy\n\tProcessor struct {\n\t\tsvcClient     workflowserviceclient.Interface\n\t\tclientBean    client.Bean\n\t\tdomainCache   cache.DomainCache\n\t\tnumWorkflows  int\n\t\tmetricsClient metrics.Client\n\t\ttallyScope    tally.Scope\n\t\tlogger        log.Logger\n\t}\n)\n\n// New returns a new instance as daemon\nfunc New(params *BootstrapParams) *Processor {\n\treturn &Processor{\n\t\tsvcClient:     params.ServiceClient,\n\t\tclientBean:    params.ClientBean,\n\t\tdomainCache:   params.DomainCache,\n\t\tnumWorkflows:  params.NumWorkflows,\n\t\tmetricsClient: params.MetricsClient,\n\t\ttallyScope:    params.TallyScope,\n\t\tlogger:        params.Logger.WithTags(tag.ComponentBatcher),\n\t}\n}\n\n// Start starts the scanner\nfunc (s *Processor) Start() error {\n\tctx := context.WithValue(context.Background(), processorContextKey, s)\n\tworkerOpts := worker.Options{\n\t\tMetricsScope:              s.tallyScope,\n\t\tBackgroundActivityContext: ctx,\n\t\tTracer:                    opentracing.GlobalTracer(),\n\t}\n\tprocessorWorker := worker.New(s.svcClient, constants.SystemLocalDomainName, processorTaskListName, workerOpts)\n\treturn processorWorker.Start()\n}\n"
  },
  {
    "path": "service/worker/parentclosepolicy/workflow.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage parentclosepolicy\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/encoded\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tcontextKey string\n)\n\nconst (\n\tprocessorContextKey contextKey = \"processorContext\"\n\t// processorTaskListName is the tasklist name\n\tprocessorTaskListName = \"cadence-sys-processor-parent-close-policy\"\n\t// processorWFTypeName is the workflow type\n\tprocessorWFTypeName   = \"cadence-sys-parent-close-policy-workflow\"\n\tprocessorActivityName = \"cadence-sys-parent-close-policy-activity\"\n\tinfiniteDuration      = 20 * 365 * 24 * time.Hour\n\tprocessorChannelName  = \"ParentClosePolicyProcessorChannelName\"\n)\n\ntype (\n\t// RequestDetail defines detail of each workflow to process\n\tRequestDetail struct {\n\t\tDomainID   string\n\t\tDomainName string\n\t\tWorkflowID string\n\t\tRunID      string\n\t\tPolicy     types.ParentClosePolicy\n\t}\n\n\t// Request defines the request for parent close policy\n\tRequest struct {\n\t\tParentExecution types.WorkflowExecution\n\t\tExecutions      []RequestDetail\n\n\t\t// DEPRECATED: the following field is deprecated since childworkflow\n\t\t// might in a different domain, use the DomainName field in RequestDetail\n\t\tDomainName string\n\t}\n)\n\nvar (\n\tretryPolicy = cadence.RetryPolicy{\n\t\tInitialInterval:    10 * time.Second,\n\t\tBackoffCoefficient: 1.7,\n\t\tMaximumInterval:    5 * time.Minute,\n\t\tExpirationInterval: infiniteDuration,\n\t}\n\n\tactivityOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    5 * time.Minute,\n\t\tRetryPolicy:            &retryPolicy,\n\t}\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(ProcessorWorkflow, workflow.RegisterOptions{Name: processorWFTypeName})\n\tactivity.RegisterWithOptions(ProcessorActivity, activity.RegisterOptions{Name: processorActivityName})\n}\n\n// ProcessorWorkflow is the workflow that performs actions for ParentClosePolicy\nfunc ProcessorWorkflow(ctx workflow.Context) error {\n\trequestCh := workflow.GetSignalChannel(ctx, processorChannelName)\n\tfor {\n\t\tvar request Request\n\t\tif !requestCh.ReceiveAsync(&request) {\n\t\t\t// no more request\n\t\t\tbreak\n\t\t}\n\n\t\topt := workflow.WithActivityOptions(ctx, activityOptions)\n\t\t_ = workflow.ExecuteActivity(opt, processorActivityName, request).Get(ctx, nil)\n\t}\n\treturn nil\n}\n\n// ProcessorActivity is activity for processing batch operation\nfunc ProcessorActivity(ctx context.Context, request Request) error {\n\tprocessor := ctx.Value(processorContextKey).(*Processor)\n\tdomainCache := processor.domainCache\n\thistoryClient := processor.clientBean.GetHistoryClient()\n\tlogger := getActivityLogger(ctx)\n\tscope := processor.metricsClient.Scope(metrics.ParentClosePolicyProcessorScope)\n\n\tchildWorkflowOnly := false\n\tif request.ParentExecution.WorkflowID != \"\" && request.ParentExecution.RunID != \"\" {\n\t\t// this is for backward compatibility\n\t\t// ideally we should always set childWorkflowOnly = true\n\t\t// however if ParentExecution is not specified, setting it to true\n\t\t// will cause terminate or cancel request to return mismatch error\n\t\tchildWorkflowOnly = true\n\t}\n\n\tremoteExecutions := make(map[string][]RequestDetail)\n\tfor _, execution := range request.Executions {\n\t\tdomainName := execution.DomainName\n\t\tif domainName == \"\" {\n\t\t\t// for backward compatibility\n\t\t\tdomainName = request.DomainName\n\t\t}\n\n\t\tvar err error\n\t\tdomainID := execution.DomainID\n\t\tif domainID == \"\" {\n\t\t\t// for backward compatibility\n\t\t\tdomainID, err = domainCache.GetDomainID(domainName)\n\t\t\tif err != nil {\n\t\t\t\tif common.IsEntityNotExistsError(err) {\n\t\t\t\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorSuccess)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorFailures)\n\t\t\t\tlogger.Error(\"Failed to process parent close policy\", tag.Error(err))\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\n\t\tswitch execution.Policy {\n\t\tcase types.ParentClosePolicyAbandon:\n\t\t\t// no-op\n\t\t\tcontinue\n\t\tcase types.ParentClosePolicyTerminate:\n\t\t\tterminateReq := &types.HistoryTerminateWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: domainID,\n\t\t\t\tTerminateRequest: &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\tDomain: domainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\t\t\t},\n\t\t\t\t\tReason:              \"by parent close policy\",\n\t\t\t\t\tIdentity:            processorWFTypeName,\n\t\t\t\t\tFirstExecutionRunID: execution.RunID,\n\t\t\t\t},\n\t\t\t}\n\t\t\tif childWorkflowOnly {\n\t\t\t\tterminateReq.ChildWorkflowOnly = true\n\t\t\t\tterminateReq.ExternalWorkflowExecution = &request.ParentExecution\n\t\t\t}\n\t\t\terr = historyClient.TerminateWorkflowExecution(ctx, terminateReq)\n\t\tcase types.ParentClosePolicyRequestCancel:\n\t\t\tcancelReq := &types.HistoryRequestCancelWorkflowExecutionRequest{\n\t\t\t\tDomainUUID: domainID,\n\t\t\t\tCancelRequest: &types.RequestCancelWorkflowExecutionRequest{\n\t\t\t\t\tDomain: domainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\t\t\t},\n\t\t\t\t\tIdentity:            processorWFTypeName,\n\t\t\t\t\tFirstExecutionRunID: execution.RunID,\n\t\t\t\t},\n\t\t\t}\n\t\t\tif childWorkflowOnly {\n\t\t\t\tcancelReq.ChildWorkflowOnly = true\n\t\t\t\tcancelReq.ExternalWorkflowExecution = &request.ParentExecution\n\t\t\t}\n\t\t\terr = historyClient.RequestCancelWorkflowExecution(ctx, cancelReq)\n\t\tdefault:\n\t\t\terr = fmt.Errorf(\"unknown parent close policy: %v\", execution.Policy)\n\t\t}\n\t\tif err != nil {\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.EntityNotExistsError,\n\t\t\t\t*types.WorkflowExecutionAlreadyCompletedError,\n\t\t\t\t*types.CancellationAlreadyRequestedError:\n\t\t\t\terr = nil\n\t\t\tcase *types.DomainNotActiveError:\n\t\t\t\tvar domainEntry *cache.DomainCacheEntry\n\t\t\t\tif domainEntry, err = domainCache.GetDomainByID(domainID); err == nil {\n\t\t\t\t\tcluster := domainEntry.GetReplicationConfig().ActiveClusterName\n\t\t\t\t\tremoteExecutions[cluster] = append(remoteExecutions[cluster], execution)\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tif err != nil {\n\t\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorFailures)\n\t\t\tlogger.Error(\"Failed to process parent close policy\", tag.Error(err))\n\t\t\treturn err\n\t\t}\n\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorSuccess)\n\t}\n\n\tif err := signalRemoteCluster(\n\t\tctx,\n\t\tprocessor.clientBean,\n\t\trequest.ParentExecution,\n\t\tremoteExecutions,\n\t\tprocessor.numWorkflows,\n\t); err != nil {\n\t\tscope.IncCounter(metrics.ParentClosePolicyProcessorFailures)\n\t\tlogger.Error(\"Failed to signal remote parent close policy workflow\", tag.Error(err))\n\t\treturn err\n\t}\n\n\treturn nil\n}\n\nfunc signalRemoteCluster(\n\tctx context.Context,\n\tclientBean client.Bean,\n\tparentExecution types.WorkflowExecution,\n\tremoteExecutions map[string][]RequestDetail,\n\tnumWorkflows int,\n) error {\n\tfor cluster, executions := range remoteExecutions {\n\t\tremoteClient, err := clientBean.GetRemoteFrontendClient(cluster)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tsignalCtx, cancel := context.WithTimeout(ctx, signalTimeout)\n\t\tsignalValue := Request{\n\t\t\tParentExecution: parentExecution,\n\t\t\tExecutions:      executions,\n\t\t}\n\n\t\tdc := encoded.GetDefaultDataConverter()\n\t\tsignalInput, err := dc.ToData(signalValue)\n\t\tif err != nil {\n\t\t\tcancel()\n\t\t\treturn err\n\t\t}\n\n\t\t_, err = remoteClient.SignalWithStartWorkflowExecution(signalCtx, &types.SignalWithStartWorkflowExecutionRequest{\n\t\t\tDomain:                              constants.SystemLocalDomainName,\n\t\t\tRequestID:                           uuid.New(),\n\t\t\tWorkflowID:                          fmt.Sprintf(\"%v-%v\", workflowIDPrefix, rand.Intn(numWorkflows)),\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: processorWFTypeName},\n\t\t\tTaskList:                            &types.TaskList{Name: processorTaskListName},\n\t\t\tInput:                               nil,\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32(infiniteDuration.Seconds())),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(60),\n\t\t\tIdentity:                            \"cadence-worker\",\n\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t\tSignalName:                          processorChannelName,\n\t\t\tSignalInput:                         signalInput,\n\t\t})\n\t\tcancel()\n\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc getActivityLogger(ctx context.Context) log.Logger {\n\tprocessor := ctx.Value(processorContextKey).(*Processor)\n\twfInfo := activity.GetInfo(ctx)\n\treturn processor.logger.WithTags(\n\t\ttag.WorkflowID(wfInfo.WorkflowExecution.ID),\n\t\ttag.WorkflowRunID(wfInfo.WorkflowExecution.RunID),\n\t\ttag.WorkflowDomainName(wfInfo.WorkflowDomain),\n\t)\n}\n"
  },
  {
    "path": "service/worker/replicator/domain_replication_processor.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replicator\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tfetchTaskRequestTimeout                   = 10 * time.Second\n\tpollTimerJitterCoefficient                = 0.2\n\tpollIntervalSecs                          = 1\n\ttaskProcessorErrorRetryWait               = time.Second\n\ttaskProcessorErrorRetryBackoffCoefficient = 1\n)\n\ntype (\n\tdomainReplicationProcessor struct {\n\t\thostInfo               membership.HostInfo\n\t\tmembershipResolver     membership.Resolver\n\t\tstatus                 int32\n\t\tsourceCluster          string\n\t\tcurrentCluster         string\n\t\tlogger                 log.Logger\n\t\tremotePeer             admin.Client\n\t\ttaskExecutor           domain.ReplicationTaskExecutor\n\t\tmetricsClient          metrics.Client\n\t\tthrottleRetry          *backoff.ThrottleRetry\n\t\tlastProcessedMessageID int64\n\t\tlastRetrievedMessageID int64\n\t\tctx                    context.Context\n\t\tcancelFn               context.CancelFunc\n\t\twg                     sync.WaitGroup\n\t\ttimeSource             clock.TimeSource\n\t\tdomainReplicationQueue domain.ReplicationQueue\n\t}\n)\n\nfunc newDomainReplicationProcessor(\n\tsourceCluster string,\n\tcurrentCluster string,\n\tlogger log.Logger,\n\tremotePeer admin.Client,\n\tmetricsClient metrics.Client,\n\ttaskExecutor domain.ReplicationTaskExecutor,\n\thostInfo membership.HostInfo,\n\tresolver membership.Resolver,\n\tdomainReplicationQueue domain.ReplicationQueue,\n\treplicationMaxRetry time.Duration,\n\ttimeSource clock.TimeSource,\n) *domainReplicationProcessor {\n\tretryPolicy := backoff.NewExponentialRetryPolicy(taskProcessorErrorRetryWait)\n\tretryPolicy.SetBackoffCoefficient(taskProcessorErrorRetryBackoffCoefficient)\n\tretryPolicy.SetExpirationInterval(replicationMaxRetry)\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(isTransientRetryableError),\n\t)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\treturn &domainReplicationProcessor{\n\t\thostInfo:               hostInfo,\n\t\tmembershipResolver:     resolver,\n\t\tstatus:                 common.DaemonStatusInitialized,\n\t\tsourceCluster:          sourceCluster,\n\t\tcurrentCluster:         currentCluster,\n\t\tlogger:                 logger,\n\t\tremotePeer:             remotePeer,\n\t\ttaskExecutor:           taskExecutor,\n\t\tmetricsClient:          metricsClient,\n\t\tthrottleRetry:          throttleRetry,\n\t\tlastProcessedMessageID: -1,\n\t\tlastRetrievedMessageID: -1,\n\t\tctx:                    ctx,\n\t\tcancelFn:               cancel,\n\t\ttimeSource:             timeSource,\n\t\tdomainReplicationQueue: domainReplicationQueue,\n\t}\n}\n\nfunc (p *domainReplicationProcessor) Start() {\n\tif !atomic.CompareAndSwapInt32(&p.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\n\tp.wg.Add(1)\n\tgo p.processorLoop()\n\tp.logger.Info(\"Domain replication processor started.\")\n}\n\nfunc (p *domainReplicationProcessor) processorLoop() {\n\tdefer p.wg.Done()\n\tdur := getWaitDuration()\n\ttimer := p.timeSource.NewTimer(dur)\n\tdefer timer.Stop()\n\n\tfor {\n\t\tselect {\n\t\tcase <-timer.Chan():\n\t\t\tp.fetchDomainReplicationTasks()\n\t\t\ttimer.Reset(getWaitDuration())\n\t\tcase <-p.ctx.Done():\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (p *domainReplicationProcessor) fetchDomainReplicationTasks() {\n\t// The following is a best effort to make sure only one worker is processing tasks for a\n\t// particular source cluster. When the ring is under reconfiguration, it is possible that\n\t// for a small period of time two or more workers think they are the owner and try to execute\n\t// the processing logic. This will not result in correctness issue as domain replication task\n\t// processing will be protected by version check.\n\tinfo, err := p.membershipResolver.Lookup(service.Worker, p.sourceCluster)\n\tif err != nil {\n\t\tp.logger.Info(\"Failed to lookup host info. Skip current run.\")\n\t\treturn\n\t}\n\n\tif info.Identity() != p.hostInfo.Identity() {\n\t\tp.logger.Debug(fmt.Sprintf(\"Worker not responsible for source cluster %v.\", p.sourceCluster))\n\t\treturn\n\t}\n\n\tctx, cancel := context.WithTimeout(p.ctx, fetchTaskRequestTimeout)\n\trequest := &types.GetDomainReplicationMessagesRequest{\n\t\tLastRetrievedMessageID: common.Int64Ptr(p.lastRetrievedMessageID),\n\t\tLastProcessedMessageID: common.Int64Ptr(p.lastProcessedMessageID),\n\t\tClusterName:            p.currentCluster,\n\t}\n\tresponse, err := p.remotePeer.GetDomainReplicationMessages(ctx, request)\n\tdefer cancel()\n\n\tif err != nil {\n\t\tp.logger.Error(\"Failed to get replication tasks\", tag.Error(err))\n\t\treturn\n\t}\n\n\tp.logger.Debug(\"Successfully fetched domain replication tasks.\", tag.Counter(len(response.Messages.ReplicationTasks)))\n\n\tfor taskIndex := range response.Messages.ReplicationTasks {\n\t\ttask := response.Messages.ReplicationTasks[taskIndex]\n\t\terr := p.throttleRetry.Do(p.ctx, func(ctx context.Context) error {\n\t\t\treturn p.handleDomainReplicationTask(task)\n\t\t})\n\n\t\tif err != nil {\n\t\t\tp.logger.Error(\"Failed to apply domain replication tasks\", tag.Error(err))\n\t\t\tdlqErr := p.throttleRetry.Do(context.Background(), func(ctx context.Context) error {\n\t\t\t\treturn p.putDomainReplicationTaskToDLQ(task)\n\t\t\t})\n\t\t\tif dlqErr != nil {\n\t\t\t\tp.logger.Error(\"Failed to put replication tasks to DLQ\", tag.Error(dlqErr))\n\t\t\t\tp.metricsClient.IncCounter(metrics.DomainReplicationTaskScope, metrics.ReplicatorDLQFailures)\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t}\n\n\tp.lastProcessedMessageID = response.Messages.GetLastRetrievedMessageID()\n\tp.lastRetrievedMessageID = response.Messages.GetLastRetrievedMessageID()\n}\n\nfunc (p *domainReplicationProcessor) putDomainReplicationTaskToDLQ(task *types.ReplicationTask) error {\n\tdomainAttribute := task.GetDomainTaskAttributes()\n\tif domainAttribute == nil {\n\t\treturn &types.InternalServiceError{\n\t\t\tMessage: \"Domain replication task does not set domain task attribute\",\n\t\t}\n\t}\n\tp.metricsClient.Scope(\n\t\tmetrics.DomainReplicationTaskScope,\n\t\tmetrics.DomainTag(domainAttribute.GetInfo().GetName()),\n\t).IncCounter(metrics.DomainReplicationEnqueueDLQCount)\n\treturn p.domainReplicationQueue.PublishToDLQ(context.Background(), task)\n}\n\nfunc (p *domainReplicationProcessor) handleDomainReplicationTask(\n\ttask *types.ReplicationTask,\n) error {\n\tp.metricsClient.IncCounter(metrics.DomainReplicationTaskScope, metrics.ReplicatorMessages)\n\tsw := p.metricsClient.StartTimer(metrics.DomainReplicationTaskScope, metrics.ReplicatorLatency)\n\tdefer sw.Stop()\n\n\terr := p.taskExecutor.Execute(task.DomainTaskAttributes)\n\tif err != nil {\n\t\tp.metricsClient.IncCounter(metrics.DomainReplicationTaskScope, metrics.ReplicatorFailures)\n\t}\n\treturn err\n}\n\nfunc (p *domainReplicationProcessor) Stop() {\n\tp.logger.Info(\"Domain replication processor stopping.\")\n\tp.cancelFn()\n\tp.wg.Wait()\n\tp.logger.Info(\"Domain replication processor stopped.\")\n}\n\nfunc getWaitDuration() time.Duration {\n\treturn backoff.JitDuration(time.Duration(pollIntervalSecs)*time.Second, pollTimerJitterCoefficient)\n}\n\nfunc isTransientRetryableError(err error) bool {\n\tswitch err.(type) {\n\tcase *types.BadRequestError:\n\t\treturn false\n\tdefault:\n\t\treturn true\n\t}\n}\n"
  },
  {
    "path": "service/worker/replicator/domain_replication_processor_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\npackage replicator\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype domainReplicationSuite struct {\n\tsuite.Suite\n\t*require.Assertions\n\tcontroller *gomock.Controller\n\n\tsourceCluster          string\n\tcurrentCluster         string\n\ttaskExecutor           *domain.MockReplicationTaskExecutor\n\tremoteClient           *admin.MockClient\n\tdomainReplicationQueue *domain.MockReplicationQueue\n\treplicationProcessor   *domainReplicationProcessor\n\ttimeSource             clock.MockedTimeSource\n}\n\nfunc TestDomainReplicationSuite(t *testing.T) {\n\ts := new(domainReplicationSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *domainReplicationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\tresource := resource.NewTest(s.T(), s.controller, metrics.Worker)\n\n\ts.sourceCluster = \"active\"\n\ts.currentCluster = \"standby\"\n\ts.taskExecutor = domain.NewMockReplicationTaskExecutor(s.controller)\n\ts.domainReplicationQueue = domain.NewMockReplicationQueue(s.controller)\n\ts.remoteClient = resource.RemoteAdminClient\n\tserviceResolver := resource.MembershipResolver\n\tserviceResolver.EXPECT().Lookup(service.Worker, s.sourceCluster).Return(resource.GetHostInfo(), nil).AnyTimes()\n\ts.timeSource = clock.NewMockedTimeSource()\n\ts.replicationProcessor = newDomainReplicationProcessor(\n\t\ts.sourceCluster,\n\t\ts.currentCluster,\n\t\tresource.GetLogger(),\n\t\ts.remoteClient,\n\t\tresource.GetMetricsClient(),\n\t\ts.taskExecutor,\n\t\tresource.GetHostInfo(),\n\t\tserviceResolver,\n\t\ts.domainReplicationQueue,\n\t\ttime.Millisecond,\n\t\ts.timeSource,\n\t)\n\tretryPolicy := backoff.NewExponentialRetryPolicy(time.Nanosecond)\n\tretryPolicy.SetMaximumAttempts(1)\n\ts.replicationProcessor.throttleRetry = backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryPolicy),\n\t\tbackoff.WithRetryableError(isTransientRetryableError),\n\t)\n}\n\nfunc (s *domainReplicationSuite) TearDownTest() {\n}\n\nfunc (s *domainReplicationSuite) TestStartStop() {\n\n\ts.replicationProcessor.Start()\n\n\t// second call should be no-op\n\ts.replicationProcessor.Start()\n\t// yield execution so background goroutine starts\n\ttime.Sleep(50 * time.Millisecond)\n\n\ts.remoteClient.EXPECT().\n\t\tGetDomainReplicationMessages(gomock.Any(), gomock.Any()).\n\t\tReturn(&types.GetDomainReplicationMessagesResponse{\n\t\t\tMessages: &types.ReplicationMessages{},\n\t\t}, nil)\n\n\t// advance time to let it run one iteration\n\ts.timeSource.Advance(pollIntervalSecs * 5 * time.Second)\n\t// yield execution\n\ttime.Sleep(50 * time.Millisecond)\n\n\t// stop processor\n\ts.replicationProcessor.Stop()\n\n\t// validate no goroutines left\n\tgoleak.VerifyNone(s.T())\n}\n\nfunc (s *domainReplicationSuite) TestHandleDomainReplicationTask() {\n\tdomainID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\tID: domainID,\n\t\t},\n\t}\n\n\ts.taskExecutor.EXPECT().Execute(task.DomainTaskAttributes).Return(nil).Times(1)\n\terr := s.replicationProcessor.handleDomainReplicationTask(task)\n\ts.NoError(err)\n\n\ts.taskExecutor.EXPECT().Execute(task.DomainTaskAttributes).Return(errors.New(\"test\")).Times(1)\n\terr = s.replicationProcessor.handleDomainReplicationTask(task)\n\ts.Error(err)\n\n}\n\nfunc (s *domainReplicationSuite) TestPutDomainReplicationTaskToDLQ() {\n\tdomainID := uuid.New()\n\ttask := &types.ReplicationTask{\n\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t}\n\n\terr := s.replicationProcessor.putDomainReplicationTaskToDLQ(task)\n\ts.Error(err)\n\n\ttask.DomainTaskAttributes = &types.DomainTaskAttributes{\n\t\tID: domainID,\n\t}\n\n\ts.domainReplicationQueue.EXPECT().PublishToDLQ(gomock.Any(), task).Return(nil).Times(1)\n\terr = s.replicationProcessor.putDomainReplicationTaskToDLQ(task)\n\ts.NoError(err)\n\n\ts.domainReplicationQueue.EXPECT().PublishToDLQ(gomock.Any(), task).Return(errors.New(\"test\")).Times(1)\n\terr = s.replicationProcessor.putDomainReplicationTaskToDLQ(task)\n\ts.Error(err)\n}\n\nfunc (s *domainReplicationSuite) TestFetchDomainReplicationTasks() {\n\tdomainID1 := uuid.New()\n\tdomainID2 := uuid.New()\n\tlastMessageID := int64(1000)\n\tresp := &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: &types.ReplicationMessages{\n\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tLastRetrievedMessageID: lastMessageID,\n\t\t},\n\t}\n\ts.remoteClient.EXPECT().GetDomainReplicationMessages(gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.taskExecutor.EXPECT().Execute(resp.Messages.ReplicationTasks[0].DomainTaskAttributes).Return(nil).Times(1)\n\ts.taskExecutor.EXPECT().Execute(resp.Messages.ReplicationTasks[1].DomainTaskAttributes).Return(nil).Times(1)\n\n\ts.replicationProcessor.fetchDomainReplicationTasks()\n\ts.Equal(lastMessageID, s.replicationProcessor.lastProcessedMessageID)\n\ts.Equal(lastMessageID, s.replicationProcessor.lastRetrievedMessageID)\n}\n\nfunc (s *domainReplicationSuite) TestFetchDomainReplicationTasks_Failed() {\n\tlastMessageID := int64(1000)\n\ts.remoteClient.EXPECT().GetDomainReplicationMessages(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test\"))\n\ts.replicationProcessor.fetchDomainReplicationTasks()\n\ts.NotEqual(lastMessageID, s.replicationProcessor.lastProcessedMessageID)\n\ts.NotEqual(lastMessageID, s.replicationProcessor.lastRetrievedMessageID)\n}\n\nfunc (s *domainReplicationSuite) TestFetchDomainReplicationTasks_FailedOnExecution() {\n\tdomainID1 := uuid.New()\n\tdomainID2 := uuid.New()\n\tlastMessageID := int64(1000)\n\tresp := &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: &types.ReplicationMessages{\n\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tLastRetrievedMessageID: lastMessageID,\n\t\t},\n\t}\n\ts.remoteClient.EXPECT().GetDomainReplicationMessages(gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.taskExecutor.EXPECT().Execute(gomock.Any()).Return(errors.New(\"test\")).AnyTimes()\n\ts.domainReplicationQueue.EXPECT().PublishToDLQ(gomock.Any(), gomock.Any()).Return(nil).Times(2)\n\n\ts.replicationProcessor.fetchDomainReplicationTasks()\n\ts.Equal(lastMessageID, s.replicationProcessor.lastProcessedMessageID)\n\ts.Equal(lastMessageID, s.replicationProcessor.lastRetrievedMessageID)\n}\n\nfunc (s *domainReplicationSuite) TestFetchDomainReplicationTasks_FailedOnDLQ() {\n\tdomainID1 := uuid.New()\n\tdomainID2 := uuid.New()\n\tlastMessageID := int64(1001)\n\tresp := &types.GetDomainReplicationMessagesResponse{\n\t\tMessages: &types.ReplicationMessages{\n\t\t\tReplicationTasks: []*types.ReplicationTask{\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tTaskType: types.ReplicationTaskTypeDomain.Ptr(),\n\t\t\t\t\tDomainTaskAttributes: &types.DomainTaskAttributes{\n\t\t\t\t\t\tID: domainID2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tLastRetrievedMessageID: lastMessageID,\n\t\t},\n\t}\n\ts.remoteClient.EXPECT().GetDomainReplicationMessages(gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.taskExecutor.EXPECT().Execute(gomock.Any()).Return(nil).AnyTimes()\n\ts.domainReplicationQueue.EXPECT().PublishToDLQ(gomock.Any(), gomock.Any()).Return(errors.New(\"test\")).Times(0)\n\n\ts.replicationProcessor.fetchDomainReplicationTasks()\n\ts.Equal(lastMessageID, s.replicationProcessor.lastProcessedMessageID)\n\ts.Equal(lastMessageID, s.replicationProcessor.lastRetrievedMessageID)\n}\n"
  },
  {
    "path": "service/worker/replicator/replicator.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage replicator\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/client\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// Replicator is the processor for replication tasks\n\tReplicator struct {\n\t\tclusterMetadata               cluster.Metadata\n\t\tdomainReplicationTaskExecutor domain.ReplicationTaskExecutor\n\t\tclientBean                    client.Bean\n\t\tdomainProcessors              []*domainReplicationProcessor\n\t\tlogger                        log.Logger\n\t\tmetricsClient                 metrics.Client\n\t\thostInfo                      membership.HostInfo\n\t\tmembershipResolver            membership.Resolver\n\t\tdomainReplicationQueue        domain.ReplicationQueue\n\t\treplicationMaxRetry           time.Duration\n\t}\n)\n\n// NewReplicator creates a new replicator for processing replication tasks\nfunc NewReplicator(\n\tclusterMetadata cluster.Metadata,\n\tclientBean client.Bean,\n\tlogger log.Logger,\n\tmetricsClient metrics.Client,\n\thostInfo membership.HostInfo,\n\tmembership membership.Resolver,\n\tdomainReplicationQueue domain.ReplicationQueue,\n\tdomainReplicationTaskExecutor domain.ReplicationTaskExecutor,\n\treplicationMaxRetry time.Duration,\n) *Replicator {\n\n\tlogger = logger.WithTags(tag.ComponentReplicator)\n\treturn &Replicator{\n\t\thostInfo:                      hostInfo,\n\t\tmembershipResolver:            membership,\n\t\tclusterMetadata:               clusterMetadata,\n\t\tdomainReplicationTaskExecutor: domainReplicationTaskExecutor,\n\t\tclientBean:                    clientBean,\n\t\tlogger:                        logger,\n\t\tmetricsClient:                 metricsClient,\n\t\tdomainReplicationQueue:        domainReplicationQueue,\n\t\treplicationMaxRetry:           replicationMaxRetry,\n\t}\n}\n\n// Start is called to start replicator\nfunc (r *Replicator) Start() error {\n\tcurrentClusterName := r.clusterMetadata.GetCurrentClusterName()\n\tfor clusterName := range r.clusterMetadata.GetRemoteClusterInfo() {\n\t\tadminClient, err := r.clientBean.GetRemoteAdminClient(clusterName)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tprocessor := newDomainReplicationProcessor(\n\t\t\tclusterName,\n\t\t\tcurrentClusterName,\n\t\t\tr.logger.WithTags(tag.ComponentReplicationTaskProcessor, tag.SourceCluster(clusterName)),\n\t\t\tadminClient,\n\t\t\tr.metricsClient,\n\t\t\tr.domainReplicationTaskExecutor,\n\t\t\tr.hostInfo,\n\t\t\tr.membershipResolver,\n\t\t\tr.domainReplicationQueue,\n\t\t\tr.replicationMaxRetry,\n\t\t\tclock.NewRealTimeSource(),\n\t\t)\n\t\tr.domainProcessors = append(r.domainProcessors, processor)\n\t}\n\n\tfor _, domainProcessor := range r.domainProcessors {\n\t\tdomainProcessor.Start()\n\t}\n\n\treturn nil\n}\n\n// Stop is called to stop replicator\nfunc (r *Replicator) Stop() {\n\n\tfor _, domainProcessor := range r.domainProcessors {\n\t\tdomainProcessor.Stop()\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/README.md",
    "content": "# A brief* overview of Scanner and Fixer and how to control them\n\nFirst and foremost: most code in this folder is disabled by default dynamic config,\nand it should be considered beta-like quality in general.  Understand what you are\nenabling before enabling it, and run it at your own risk.  It is shared primarily so\nothers can learn from and leverage what we have already encountered, if they end up\nneeding it or something similar.\n\n**Any _actually recommended_ fixing processes will be explicitly called out in release\nnotes or similar.**  This document is **not** recommending any, merely describing.\n\nThere are a variety of problems with the Scanner and Fixer related code, so\nplease do not take this document as a sign that it is a structure we want to _keep_.\nIt has just been confusing enough that it took substantial time to understand,\nand now some of that effort is written down to save people the full effort in\nthe future.\n\n## What is this folder for\n\nThis folder as a whole contains a variety of data-cleanup workflows.\n\nStuff like:\n- Find old, unnecessary data and delete it.\n- Find data caused by old bugs and fix it.\n- Clean up and remove abandoned tasklists so they do not continue taking up space.\n\nAs a general rule, these all scan the full database (for one kind of data), check\nsome data, and clean it up if necessary.  \n*How their code does that* varies quite a bit, however.\n\nE.g. the \"history scavenger\" finds old branches of history that lost their CAS race\nto update the workflow's official history.  It is not possible to guarantee that\nthese are cleaned up while a workflow runs, because any cleanup could have failed.  \nInstead, the history scavenger periodically walks through the whole database and\nlooks for these orphaned history branches, and deletes them.\n\nThe most complex of these processes are based around `Scanner` and `Fixer`, and\nso **this readme is almost exclusively written for them**.  \nFor others, just read their code, it's probably faster than reading any doc about\ntheir code.\n\n## Basic structure of `Scanner` and `Fixer` workflows\n\n- \"Invariants\" define `Check` and `Fix` methods that make sure our invariants hold,\n  and fixes them if they did not for some reason.\n  - These are in the `common/reconciliation/invariant` folder, e.g.\n    [concreteExecutionExists.go](../../../common/reconciliation/invariant/concreteExecutionExists.go)\n    checks (`Check`) that a current execution record points to a concrete record\n    that exists.  If it does not, it deletes the current record (the `Fix`).\n  - Some of these have a paired \"invariant collection\" which is currently a 1:1\n    relationship, and is used elsewhere to refer to the invariants by a name where\n    it isn't unique per data-type (e.g. timer only has one).\n  - These are often bundled together in an `InvariantManager`, which runs multiple and\n    aggregates the results.\n  - Invariants are _almost_ exclusively used by Scanner and Fixer.  One is also used\n    as part of replication processing, in [ndc/history_resender.go](../../../common/ndc/history_resender.go).\n    That use is essentially completely unrelated to Scanner or Fixer, and is ignored for this doc.\n- [Scanner](shardscanner/scanner.go) runs only the `Check`, and pushes all failing checks to the blobstore.\n  - It gets its core data through an Iterator, whose implementation depends on your datastore.\n  - On each Iterator item, it runs through its list of Invariants (via an `InvariantManager`).\n  - Per item, the aggregated result from the InvariantManager is collected, and pushed to a blob-store. \n- [Fixer](shardscanner/fixer.go) runs only `Fix`, on the downloaded results from the most-recent Scanner.\n  - Structurally it's extremely similar to Scanner, but it calls `Fix` instead, and\n    it gets its data from a different Iterator (this one iterates over the scanner\n    results in your blobstore).\n  - _All_ configured invariants run, not just the ones that failed `Check` previously.\n  - Because only `Fix` is called, invariants should likely `Check` first internally.\n- ^ The workflows that run Scanner and Fixer are started at service startup _if enabled_,\n  and are run as continuous every-minute crons that essentially never expire.\n  - Each scanner / fixer type uses its own tasklist, and these workers are only\n    started if enabled.\n    - This effectively means that enabling one works immediately after a service start,\n      but disabling _only pauses_, which may be undesirable if they are resumed after\n      a lengthy delay.\n    - Because these are crons, only the original start arguments are retained, not new\n      ones if changes are made.\n    - **Manually cancel or terminate the cron workflows if you change relevant fields,\n      and (re)start a worker to start the new versions.**\n  - Each workflow that runs processes *only one data-source*, primarily via its Iterator.\n    - This means all invariants within a scanner/fixer run handle the same type of data.\n- The bulk of all ^ this is plugged together by `*shardscanner.ScannerConfig` instances,\n  which contain everything that customizes each particular _kind_ of scanner/fixer,\n  which is stored in / retrieved from the context by the workflow's type as a key.\n  - E.g. see the [concrete_execution.go config](executions/concrete_execution.go).\n  - The workflow type (registered function name) is in there, as are the start args,\n    some high level dynamic config to control the scanner/fixer workflows (enabled,\n    concurrency, etc), and \"hooks\" for both scanner and fixer.\n  - The workflows largely do not care about this config, they just run the same\n    activities each time and let the activities figure out what to do.\n  - Activities get their config and other dependencies from [Get{Scanner,Fixer}Context](shardscanner/types.go)),\n    which contain this config.\n- Hooks (fields in the `ScannerConfig`) are where much of the non-workflow behavior comes from.\n  - E.g. the concrete scanner hooks bundle up a \"manager\" (InvariantManager), \"iterator\"\n    (walks the datasource and yields individual items to check), and \"custom config\"\n    (dynamic config to control which invariants are enabled) as part of the `create*Hooks`\n    funcs in [concrete_exeuction.go](executions/concrete_execution.go).\n- And so, ultimately the workflows essentially take this config and create a Scanner or Fixer\n  out of them (in activities), as you can see in e.g. [the scanner workflow](shardscanner/scanner_workflow.go).\n  - It reads some config through an activity (which gets its per-workflow-type context).\n    - This is used to decide parallelism / pass additional args to the scan activities.\n  - The `scanShardActivity` (in [shardscanner/activities.go](shardscanner/activities.go)\n    essentially iterates over shards, and runs `scanShard` on each one, which creates\n    a `NewScanner` that gets its config and behavior from args / environment / hooks.\n  - Fixer is **very** similar, except it also queries the previous Scanner run to get the\n    list of blobstore files that it needs to process. \n\nLast but not least: there are other workflows in this folder which _do not_ follow these patterns!\n- E.g. the [tasklist scanner and history scavenger](workflow.go), and [CheckDataCorruptionWorkflow](data_corruption_workflow.go).\n- These are MUCH more localized in behavior and simpler, so they are not covered in this document.\n  Just read the code :)\n- Their workflows are still started in the main entry-point, [scanner.go](scanner.go).\n  See the `Start` function, most are pretty easily found in there.\n\nHopefully that made sense.\n\n## Config\n\nScanners and fixers are generally disabled by default, as they can consume\nsubstantial resources and may delete or modify data to correct issues.  \nBecause of this, you generally need to modify your dynamic config to run them.\n\nAt the time of writing, you can enable these with config like the following.\n\nEnable scanner workflows (these are per data source / per record type, like\n\"concrete executions\" and \"timers\"):\n```yaml\nworker.executionsScannerEnabled:\n  - value: true        # default false\nworker.currentExecutionsScannerEnabled:\n  - value: true        # default false\nworker.timersScannerEnabled:\n  - value: true        # default false\nworker.historyScannerEnabled:\n  - value: true        # default false\nworker.taskListScannerEnabled:\n  - value: true        # default true, only used on sql stores\n```\nEnable scanner invariants (currently each one only supports one data source /\nrecord type, but there may be multiple invariants for the data source):\n```yaml\n# concretes\nworker.executionsScannerInvariantCollectionStale:\n  - value: true         # default false\nworker.executionsScannerInvariantCollectionMutableState:\n  - value: true         # default true\nworker.executionsScannerInvariantCollectionHistory:\n  - value: true         # default true\n\n# timer invariant is implied as there is only one.\n# to enable it, enable the workflow.\n\n# currents, NONE OF THESE WORK because of type mismatch\nworker.currentExecutionsScannerInvariantCollectionHistory:\n  - value: true         # default true\nworker.currentExecutionsInvariantCollectionMutableState:\n  - value: true         # default true\n```\nEnable fixer workflows (also one per type):\n```yaml\nworker.concreteExecutionFixerEnabled:\n  - value: true       # default false\nworker.currentExecutionFixerEnabled:\n  - value: true       # default false\nworker.timersFixerEnabled:\n  - value: true       # default false\n```\nEnable fixer to run on a domain (required to do anything to a domain's data,\nwhich also means nothing will be fixed without this):\n```yaml\nworker.currentExecutionFixerDomainAllow:\n  - value: true         # default false\n    constraints: {domainName: \"your-domain\"}  # for example, or have no constraints to enable for all domains\nworker.concreteExecutionFixerDomainAllow:\n  - value: true         # default false\nworker.timersFixerDomainAllow:\n  - value: true         # default false\n```\nEnable fixer invariants:\n```yaml\n# concretes\nworker.executionsFixerInvariantCollectionStale:\n  - value: true         # default false\nworker.executionsFixerInvariantCollectionMutableState:\n  - value: true         # default true\nworker.executionsFixerInvariantCollectionHistory:\n  - value: true         # default true\n\n# timer invariant is enabled if timer-fixer is enabled, as there is only one\n\n# current execution fixer has never worked and does not currently support dynamic config\n```\n\n## Verifying locally\n\nThere are a few ways to run local clusters and make changes and test things out,\nbut this is the way I did things when reading and changing this code:\n\n1. Run the default docker-compose cluster ([docker-compose.yml](../../../docker/docker-compose.yml))\n2. Make your config/code/etc changes locally for scanner / fixer.\n3. `make cadence-server` to ensure it builds\n4. `./cadence-server start --services worker` to start up a worker that will\n   connect to your docker compose cluster.\n5. Browse history via the web UI, usually http://localhost:8088/domains/cadence-system/workflows\n\nThe default docker compose setup starts a worker instance, but due to the default\ndynamic config setup where all but `worker.taskListScannerEnabled` are disabled,\nthe in-docker worker will not run (most) scanner/fixer tasklists and will not steal\nany tasks from a local worker.\n\nSo you can often simply run it without any changes, start up your local (customized)\nworker service outside of docker, and everything will Just Work™.\n\nThis way you can leverage the normal docker compose yaml files with minimal effort,\nuse the web UI to observe the results, and rapidly change/rebuild/rerun/debug/etc\nwithout needing to deal with docker.\n\nIf you **do** need to debug the tasklist scanner, I would recommend making a custom build,\nand modifying the docker compose file to use your build.  Details on that are below,\nbut they **are not necessary** for other scanners/fixers.\n\n### Docker compose changes (tasklist scanner only)\n\nThere are a few ways to achieve this, but I like modifying the `docker-compose*.yaml`\nfile to use a custom local build, and just changing its dynamic config to disable\nthe tasklist scanner.\n\nTo do that, see the [docker/README.md file](../../../docker/README.md) for instructions.\nPersonally I prefer making a unique auto-setup tag so it does not replace any non-customized\ndocker compose runs in the future.  E.g.:\n```yaml\nservices:\n  # ...\n  cadence:\n    image: ubercadence/server:my-auto-setup  # use your new tag\n    # ...\n    environment:\n      # note this env var, it is the file you need to change\n      - \"DYNAMIC_CONFIG_FILE_PATH=config/dynamicconfig/development.yaml\"\n```\nAnd just make a build after changing the dynamic config file.  This will copy the file\ninto the docker image, and any local changes won't affect the running container.\n```yaml\nworker.taskListScannerEnabled:\n  - value: false        # default true, only used on sql stores\n```\nJust set ^ this, and make sure the others are not explicitly enabled as they are\ndisabled by default, and you're likely done.\n\n### Other things to change outside docker\n\n- Config, as running the server locally will generally use the `config/dynamicconfig/development.yaml`\n  file, so you likely want to change that one to enable your code.\n- Data, to give your invariant / scanner something to notice\n  - Running some workflows and then deleting / modifying the data by hand in the\n    database is fairly simple.\n- Your invariants, to prevent prematurely purging any manual data changes you made.\n  - I'm fond of this trick:\n    ```go\n    func (i *invariant) Check(...) {\n        x := true    // go vet doesn't currently complain\n        if x {       // about dead code with this.  handy!\n            return CheckResult{Failed} // fake failure, so all records go to fixer\n        }\n        // ... the rest of your normal code, unchanged\n    }\n    func (i *invariant) Fix(...) {\n        // print the fix rather than doing it, so the next runs try too.\n        // or do the `if x {` trick here too\n    }\n    ```\n- Your IDE, to launch with only the worker, as other services are not necessary\n  and it will just slow down startup/shutdown.\n  - Just make sure you have `start --services worker` in its start-args.\n\n### Running it all and checking the results\n\nAdd some breakpoints or print statements, run it and see what happens :)\n\nIf Scanner found anything interesting, you should now have a `/tmp/blobstore` folder with\nfiles like `{uuid()}_0.corrupted`.  These uuids are _random_, and the `_0.corrupted` suffix\nmarks them as page 0 (out of N), and that they refer to corrupted entries.  You'll have\none uuid per Scanner shard (configurable, for concurrency) that found issues, and multiple\npages per shard if the results exceeded the paging size limit.\n\nIf Fixer found anything, `/tmp/blobstore` should now have `{uuid()}_0.skipped` and/or\n`{uuid()}_0.fixed` files.  These uuids are _also random_, and do not refer to the Scanner\nfile that their data came from, and the uuid and paging patterns are the same as Scanner.\n\nYou may also have `*.failed` files, following the same pattern.  Similarly, these are cases\nwhere an invariant returned a Failure result (they can come from scanner _or_ fixer).\nOnly `*.corrupted` files from scanner will be processed by fixer, however.\n\nNote that `*.failed` files can contain invariant results of _all_ statuses, as the status\nof a record _trends towards_ \"failed\", and only the final status is recorded.\nFor details, see the behavior in the [InvariantManger](../../../common/reconciliation/invariant/invariantManager.go).\n\nYou can also see the results in the scanner and fixer workflows in the UI.  In particular:\n- Scanner:\n  - Each scanner type has a unique ID, like `concreteExecutionsScannerWFID = \"cadence-sys-executions-scanner\"`.\n  - Check the activities to see how many corruptions were found per shard\n  - Query its `aggregate_report` (works in UI, others require arguments and you currently\n    need to use the CLI) to get overall counts.\n  - The activities return results with UUIDs and page numbers.\n    These are the UUIDs and page ranges for files in `/tmp/blobstore`, so you can look up\n    specific detailed results.\n    - Otherwise tbh I grab a known workflow ID and grep the most-recent batch of files for it.\n- Fixer:\n  - Check the recent fixer workflows for fix results.\n    - If it has completed already, it is likely _not_ the most-recent or the currently running one.\n      Check older ones until you find some with more than ~ a dozen events, as those are no-ops.\n  - The activities accept UUIDs and page ranges from scanner (these match the return values in\n    scanner, and refer to the scan-result files in `/tmp/blobstore`), and return the same kind of\n    structure (new random UUIDs, new page ranges, referring to new files in `/tmp/blobstore`).\n    - Again, I would also recommend just grepping for known IDs that should have been processed.\n\nIf you are not printing or debugging whatever info you are looking for, check the\ncontents of these files to verify they're doing what you expect!\n\n#### An example working scanner/fixer setup, visible in /tmp/blobstore files\n\nThis is the new \"stale\" invariant working in its concrete execution scanner -> the\nconcrete execution fixer also working, with faked results to simplify testing.\nI also ran all the other concrete invariants because I was curious.\n\nI made a change like this:\n```go\nfunc (c *staleWorkflowCheck) Check(\n\tctx context.Context,\n\texecution interface{},\n) CheckResult {\n\tx := true\n\tif x {\n\t\treturn CheckResult{\n\t\t\tCheckResultType: CheckResultTypeCorrupted,\n\t\t\tInvariantName:   c.Name(),\n\t\t\tInfo:            \"fake corrupt\",\n\t\t}\n\t}\n\t_, result := c.check(ctx, execution)\n\treturn result\n}\n```\nso the `Check` call would always fail, and the `Fix` call would run the real check.\n\nAnd added dynamic config like this:\n```yaml\n# enable these invariants\nworker.executionsScannerInvariantCollectionStale:\n  - value: true         # default false\nworker.executionsScannerInvariantCollectionMutableState:\n  - value: true         # default true\nworker.executionsScannerInvariantCollectionHistory:\n  - value: true         # default true\nworker.executionsFixerInvariantCollectionStale:\n  - value: true         # default false\nworker.executionsFixerInvariantCollectionMutableState:\n  - value: true         # default true\nworker.executionsFixerInvariantCollectionHistory:\n  - value: true         # default true\n\n# these invariants are all run by the concrete execution scanner/fixer\nworker.executionsScannerEnabled:  # note slightly different name\n  - value: true         # default false\nworker.concreteExecutionFixerEnabled:\n  - value: true         # default false\nworker.concreteExecutionFixerDomainAllow:\n  - value: true         # default false\n```\n\nAfter a scanner and fixer run, `/tmp/blobstore` contained `*.corrupted` and `*.skipped` files.\nThe `*.corrupted` files contained data like this:\n```json\n{\n  \"Input\": {\n    \"Execution\": { ... },\n    \"Result\": {\n      \"CheckResultType\": \"corrupted\",\n      \"DeterminingInvariantType\": \"stale_workflow\",\n      \"CheckResults\": [\n        {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"history_exists\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"open_current_execution\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        {\n          \"CheckResultType\": \"corrupted\",\n          \"InvariantName\": \"stale_workflow\",\n          \"Info\": \"fake corrupt\",\n          \"InfoDetails\": \"\"\n        }\n      ]\n    }\n  },\n  \"Result\": {\n    \"FixResultType\": \"skipped\",\n    \"DeterminingInvariantName\": null,\n    \"FixResults\": null\n  }\n}\n```\nYou can see the two healthy invariants, and the one I faked.\n\nWhen fixer then ran I got this in a `*.skipped` file:\n```json\n{\n  \"Execution\": { ... },\n  \"Input\": {\n    \"Execution\": { ... },\n    \"Result\": {\n      \"CheckResultType\": \"corrupted\",\n      \"DeterminingInvariantType\": \"stale_workflow\",\n      \"CheckResults\": [\n        {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"history_exists\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"open_current_execution\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        {\n          \"CheckResultType\": \"corrupted\",\n          \"InvariantName\": \"stale_workflow\",\n          \"Info\": \"fake corrupt\",\n          \"InfoDetails\": \"\"\n        }\n      ]\n    }\n  },\n  \"Result\": {\n    \"FixResultType\": \"skipped\",\n    \"DeterminingInvariantName\": null,\n    \"FixResults\": [\n      {\n        \"FixResultType\": \"skipped\",\n        \"InvariantName\": \"history_exists\",\n        \"CheckResult\": {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"history_exists\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        \"Info\": \"skipped fix because execution was healthy\",\n        \"InfoDetails\": \"\"\n      },\n      {\n        \"FixResultType\": \"skipped\",\n        \"InvariantName\": \"open_current_execution\",\n        \"CheckResult\": {\n          \"CheckResultType\": \"healthy\",\n          \"InvariantName\": \"open_current_execution\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        \"Info\": \"skipped fix because execution was healthy\",\n        \"InfoDetails\": \"\"\n      },\n      {\n        \"FixResultType\": \"skipped\",\n        \"InvariantName\": \"stale_workflow\",\n        \"CheckResult\": {\n          \"CheckResultType\": \"\",\n          \"InvariantName\": \"\",\n          \"Info\": \"\",\n          \"InfoDetails\": \"\"\n        },\n        \"Info\": \"no need to fix: completed workflow still within retention + 10-day buffer\",\n        \"InfoDetails\": \"completed workflow still within retention + 10-day buffer, closed 2023-09-20 20:26:04.924876012 -0500 CDT and allowed to exist until 2023-10-07\"\n      }\n    ]\n  }\n}\n```\nNotice that _all three_ invariants ran in the fixer (all three were enabled), and\nall three fixes were skipped because they did not find any problems.\n\nIf I had also faked the stale workflow invariant `Fix`, you would see a `FixResultType`\nof \"fixed\" on that invariant in fixer, and a file named `*.fixed` rather than `*.skipped`.\n\n#### An example bad scanner/fixer setup\n\nThis is the current-execution scanner working -> the current-execution fixer misbehaving\nand using the wrong type, and creating `*.failed` files, as of commit `eb55629d`.\n\nI faked the current execution invariant to always say \"corrupt\" in `Check`, and panic in `Fix`,\nand enabled the current execution scanner and fixer in dynamic config, and ran the worker.\n\nFirst, the scanner run creates `*.corrupted` files with entries like this:\n```json\n{\n  \"Input\": {\n    \"Execution\": { ... },\n    \"Result\": {\n      \"CheckResultType\": \"corrupted\",\n      \"DeterminingInvariantType\": \"concrete_execution_exists\",\n      \"CheckResults\": [\n        {\n          \"CheckResultType\": \"corrupted\",\n          \"InvariantName\": \"concrete_execution_exists\",\n          \"Info\": \"execution is open without having concrete execution\",\n          \"InfoDetails\": \"concrete execution not found. WorkflowId: e905c98f-108a-4191-9ef2-ca07a1361f9c, RunId: 6bc5386b-c043-4eb1-a332-c3bb7b5188f0\"\n        }\n      ]\n    }\n  },\n  \"Result\": {\n    \"FixResultType\": \"skipped\",\n    \"DeterminingInvariantName\": null,\n    \"FixResults\": null\n  }\n}\n```\nThis shows that it identified my \"always corrupt\" results correctly, in the\n[concrete_execution_exists invariant](../../../common/reconciliation/invariant/concreteExecutionExists.go).\n\nThis is later consumed by the current execution fixer, which produces `*.failed` files with\ncontents like this:\n```json\n{\n  \"Execution\": { ... },\n  \"Input\": {\n    \"Execution\": { ... },\n    \"Result\": {\n      \"CheckResultType\": \"corrupted\",\n      \"DeterminingInvariantType\": \"concrete_execution_exists\",\n      \"CheckResults\": [\n        {\n          \"CheckResultType\": \"corrupted\",\n          \"InvariantName\": \"concrete_execution_exists\",\n          \"Info\": \"execution is open without having concrete execution\",\n          \"InfoDetails\": \"concrete execution not found. WorkflowId: e905c98f-108a-4191-9ef2-ca07a1361f9c, RunId: 6bc5386b-c043-4eb1-a332-c3bb7b5188f0\"\n        }\n      ]\n    }\n  },\n  \"Result\": {\n    \"FixResultType\": \"failed\",\n    \"DeterminingInvariantName\": \"history_exists\",\n    \"FixResults\": [\n      {\n        \"FixResultType\": \"failed\",\n        \"InvariantName\": \"history_exists\",\n        \"CheckResult\": {\n          \"CheckResultType\": \"failed\",\n          \"InvariantName\": \"history_exists\",\n          \"Info\": \"failed to check: expected concrete execution\",\n          \"InfoDetails\": \"\"\n        },\n        \"Info\": \"failed fix because check failed\",\n        \"InfoDetails\": \"\"\n      },\n      {\n        \"FixResultType\": \"failed\",\n        \"InvariantName\": \"open_current_execution\",\n        \"CheckResult\": {\n          \"CheckResultType\": \"failed\",\n          \"InvariantName\": \"open_current_execution\",\n          \"Info\": \"failed to check: expected concrete execution\",\n          \"InfoDetails\": \"\"\n        },\n        \"Info\": \"failed fix because check failed\",\n        \"InfoDetails\": \"\"\n      }\n    ]\n  }\n}\n```\nYou can see scanner's data as the input-result, and _completely different invariants_\nrunning and failing as the fixer result.\n\n## Thoughts and prayers\n\n\n\nMost notably:\n- The current implementation is extremely difficult to extend externally, as the\n  code depends heavily on constants that cannot be added to or changed.\n  - Changing this likely requires rewriting a significant amount of the code, but\n    does seem worth doing.\n  - Future versions really should try to fix this.  Custom database plugins may\n    have unique problems and need unique scan/fix tools, and those are unlikely\n    to be open-source friendly and useful to run on _every_ database.\n- The current-execution _fixer_ has never been run successfully anywhere.\n  Beware drawing any conclusions from its code or config.\n  - The scanner appears to work, and the invariant seems correct, but the fixer\n    was written to use the _concrete_ execution invariants, and that has caused\n    it to always fail when run.\n  - This might be fixed and (optionally) enabled or deleted in the future.\n- Current, concrete, and timer scan/fix code is _very similar_ but not identical.\n  - Maybe this is good, maybe not, but it can be confusing.  Read carefully.\n- Config keys, configurability, all things config-like vary widely between scan/fix implementations.\n  - Copy/paste, don't guess.  And verify it before calling it \"done\".\n- The current \"scan everything, then fix everything\" structure is problematically slow\n  on large clusters, and quite resource-wasteful due to doing everything at least 2x.\n  - Generally speaking it's probably not a bad thing to slow it down and re-check, but\n    large clusters can take _weeks_ before fixer starts.  That's horrible for verifying\n    fixes, or addressing any issues quickly.\n- Queries feel like an odd way to pass data from scanner to fixer.\n  - I _suspect_ they were done to avoid returning too much data, violating a blob-size\n    limit constraint.  This... might be valid?  The data it actually uses is saved by\n    fixer's querying activity though, so it is still bound by that limit.\n  - Due to results being spread between many activities and many queries, overviews\n    are hard to get.  We could instead push it all to a new blobstore file, for human use.\n- There's a lot of indirection in general for unknown reasons, and that makes control-flow\n  extremely hard to figure out.\n  - Possibly it was intended for more flexibility than it has seen, possibly it would\n    simply benefit from some modernizing (instance fields rather than background\n    activity context), possibly it's a functional-like lament about the lack of\n    generics in Go when it was written, I'm really not sure.\n\nOverall it seems like it probably deserves a rewrite, though some of the parts are\npretty clearly reusable (invariants, iterators, etc).\n\n---\n\n[*]: may not actually be brief\n"
  },
  {
    "path": "service/worker/scanner/data_corruption_workflow.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\tc \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/fetcher\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(\n\t\tCheckDataCorruptionWorkflow,\n\t\tworkflow.RegisterOptions{Name: reconciliation.CheckDataCorruptionWorkflowType},\n\t)\n\tactivity.Register(ExecutionFixerActivity)\n\tactivity.Register(EmitResultMetricsActivity)\n}\n\nconst (\n\tworkflowTimer        = 5 * time.Minute\n\tmaxSignalNumber      = 1000\n\tfixerActivityTimeout = time.Minute\n)\n\n// CheckDataCorruptionWorkflow is invoked by remote cluster via signals\nfunc CheckDataCorruptionWorkflow(ctx workflow.Context, fixList []entity.Execution) error {\n\n\tlogger := workflow.GetLogger(ctx)\n\tsignalCh := workflow.GetSignalChannel(ctx, reconciliation.CheckDataCorruptionWorkflowSignalName)\n\tsignalCount := 0\n\tmaxReceivedSignalNumber := workflow.SideEffect(ctx, func(ctx workflow.Context) interface{} {\n\t\treturn maxSignalNumber\n\t})\n\n\tfor {\n\t\tselector := workflow.NewSelector(ctx)\n\t\t// timer\n\t\ttimerCtx, timerCancel := workflow.WithCancel(ctx)\n\t\twaitTimer := workflow.NewTimer(timerCtx, workflowTimer)\n\t\tselector.AddFuture(waitTimer, func(f workflow.Future) {\n\t\t\t// do nothing. Unblock the selector\n\t\t})\n\n\t\t// signal\n\t\tselector.AddReceive(signalCh, func(c workflow.Channel, more bool) {\n\t\t\tvar fixExecution entity.Execution\n\t\t\tfor c.ReceiveAsync(&fixExecution) {\n\t\t\t\tsignalCount++\n\t\t\t\tfixList = append(fixList, fixExecution)\n\t\t\t}\n\t\t})\n\n\t\tselector.Select(ctx)\n\t\tif len(fixList) == 0 {\n\t\t\treturn nil\n\t\t}\n\t\ttimerCancel()\n\n\t\ttimeout := fixerActivityTimeout * time.Duration(len(fixList))\n\t\tactivityOptions = workflow.ActivityOptions{\n\t\t\tScheduleToStartTimeout: time.Minute,\n\t\t\tStartToCloseTimeout:    timeout,\n\t\t\tHeartbeatTimeout:       fixerActivityTimeout,\n\t\t\tRetryPolicy:            &activityRetryPolicy,\n\t\t}\n\t\tactivityCtx := workflow.WithActivityOptions(ctx, activityOptions)\n\t\tvar fixResults []invariant.FixResult\n\t\terr := workflow.ExecuteActivity(activityCtx, ExecutionFixerActivity, fixList).Get(activityCtx, &fixResults)\n\t\tif err != nil {\n\t\t\tlogger.Error(\"failed to run execution fixer activity\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\n\t\tactivityOptions = workflow.ActivityOptions{\n\t\t\tScheduleToStartTimeout: time.Minute,\n\t\t\tStartToCloseTimeout:    time.Minute,\n\t\t}\n\t\tactivityCtx = workflow.WithActivityOptions(ctx, activityOptions)\n\t\terr = workflow.ExecuteActivity(activityCtx, EmitResultMetricsActivity, fixResults).Get(activityCtx, nil)\n\t\tif err != nil {\n\t\t\tlogger.Error(\"failed to run execution fixer activity\", zap.Error(err))\n\t\t\treturn err\n\t\t}\n\n\t\tfixList = []entity.Execution{}\n\t\tworkflow.GetMetricsScope(ctx)\n\t\tvar maxSignalCount int\n\t\tif err := maxReceivedSignalNumber.Get(&maxSignalCount); err != nil {\n\t\t\tlogger.Error(\"failed to get max supported signal number\")\n\t\t\treturn err\n\t\t}\n\n\t\tif signalCount > maxSignalCount {\n\t\t\tvar fixExecution entity.Execution\n\t\t\tfor signalCh.ReceiveAsync(&fixExecution) {\n\t\t\t\tfixList = append(fixList, fixExecution)\n\t\t\t}\n\t\t\treturn workflow.NewContinueAsNewError(ctx, reconciliation.CheckDataCorruptionWorkflowType, fixList)\n\t\t}\n\t}\n}\n\nfunc ExecutionFixerActivity(ctx context.Context, fixList []entity.Execution) ([]invariant.FixResult, error) {\n\tvar result []invariant.FixResult\n\tindex := 0\n\tif activity.HasHeartbeatDetails(ctx) {\n\t\tactivity.GetHeartbeatDetails(ctx, &index)\n\t}\n\n\tfor index < len(fixList) {\n\t\texecution := fixList[index]\n\t\tpr, cache, err := getDefaultDAO(ctx, execution.ShardID)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\trequest := fetcher.ExecutionRequest{\n\t\t\tDomainID:   execution.DomainID,\n\t\t\tWorkflowID: execution.WorkflowID,\n\t\t\tRunID:      execution.RunID,\n\t\t}\n\t\tconcreteExecution, err := fetcher.ConcreteExecution(ctx, pr, request)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tcurrentExecutionInvariant := invariant.NewOpenCurrentExecution(pr, cache)\n\t\tfixResult := currentExecutionInvariant.Fix(ctx, concreteExecution)\n\t\tresult = append(result, fixResult)\n\t\thistoryInvariant := invariant.NewHistoryExists(pr, cache)\n\t\tfixResult = historyInvariant.Fix(ctx, concreteExecution)\n\t\tresult = append(result, fixResult)\n\t\tactivity.RecordHeartbeat(ctx, index)\n\t\tindex++\n\t}\n\treturn result, nil\n}\n\nfunc getDefaultDAO(\n\tctx context.Context,\n\tshardID int,\n) (persistence.Retryer, cache.DomainCache, error) {\n\tsc, err := getScannerContext(ctx)\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"cannot find key %v in context\", reconciliation.CheckDataCorruptionWorkflowType)\n\t}\n\tres := sc.resource\n\tcache := res.GetDomainCache()\n\texecManager, err := res.GetExecutionManager(shardID)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\tpr := persistence.NewPersistenceRetryer(execManager, res.GetHistoryManager(), c.CreatePersistenceRetryPolicy())\n\treturn pr, cache, nil\n}\n\nfunc EmitResultMetricsActivity(ctx context.Context, fixResults []invariant.FixResult) error {\n\tsc, err := getScannerContext(ctx)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"cannot find key %v in context\", reconciliation.CheckDataCorruptionWorkflowType)\n\t}\n\tres := sc.resource\n\n\tfor _, result := range fixResults {\n\t\tscope := res.GetMetricsClient().Scope(\n\t\t\tmetrics.CheckDataCorruptionWorkflowScope,\n\t\t\tmetrics.InvariantTypeTag(string(result.InvariantName)))\n\n\t\tscope.IncCounter(metrics.DataCorruptionWorkflowCount)\n\t\tswitch result.FixResultType {\n\t\tcase invariant.FixResultTypeFailed:\n\t\t\tscope.IncCounter(metrics.DataCorruptionWorkflowFailure)\n\t\tcase invariant.FixResultTypeFixed:\n\t\t\tscope.IncCounter(metrics.DataCorruptionWorkflowSuccessCount)\n\t\tcase invariant.FixResultTypeSkipped:\n\t\t\tscope.IncCounter(metrics.DataCorruptionWorkflowSkipCount)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc NewDataCorruptionWorkflowWorker(\n\tresource resource.Resource,\n\tparams *BootstrapParams,\n) *Scanner {\n\n\tzapLogger, err := zap.NewProduction()\n\tif err != nil {\n\t\tresource.GetLogger().Fatal(\"failed to initialize zap logger\", tag.Error(err))\n\t}\n\treturn &Scanner{\n\t\tcontext: scannerContext{\n\t\t\tresource: resource,\n\t\t},\n\t\ttallyScope: params.TallyScope,\n\t\tzapLogger:  zapLogger.Named(\"data-corruption-workflow\"),\n\t}\n\n}\n\nfunc (s *Scanner) StartDataCorruptionWorkflowWorker() error {\n\tctx := context.WithValue(context.Background(), contextKey(reconciliation.CheckDataCorruptionWorkflowType), s.context)\n\tworkerOpts := worker.Options{\n\t\tLogger:                                 s.zapLogger,\n\t\tMetricsScope:                           s.tallyScope,\n\t\tMaxConcurrentActivityExecutionSize:     maxConcurrentActivityExecutionSize,\n\t\tMaxConcurrentDecisionTaskExecutionSize: maxConcurrentDecisionTaskExecutionSize,\n\t\tBackgroundActivityContext:              ctx,\n\t}\n\n\terr := worker.New(\n\t\ts.context.resource.GetSDKClient(),\n\t\tconstants.SystemLocalDomainName,\n\t\treconciliation.CheckDataCorruptionWorkflowTaskList,\n\t\tworkerOpts,\n\t).Start()\n\treturn err\n}\n"
  },
  {
    "path": "service/worker/scanner/data_corruption_workflow_test.go",
    "content": "// Copyright (c) 2021 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/history/constants\"\n)\n\nconst testWorkflowName = \"default-test-workflow-type-name\"\n\nvar validBranchToken = []byte{89, 11, 0, 10, 0, 0, 0, 12, 116, 101, 115, 116, 45, 116, 114, 101, 101, 45, 105, 100, 11, 0, 20, 0, 0, 0, 14, 116, 101, 115, 116, 45, 98, 114, 97, 110, 99, 104, 45, 105, 100, 0}\n\ntype dataCorruptionWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestDataCorruptionWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(dataCorruptionWorkflowTestSuite))\n}\n\nfunc (s *dataCorruptionWorkflowTestSuite) TestWorkflow_SignalWorkflow_Success() {\n\tsignalOnce := false\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(ExecutionFixerActivity, mock.Anything, mock.Anything).Return([]invariant.FixResult{}, nil).Times(1)\n\tenv.OnActivity(EmitResultMetricsActivity, mock.Anything, mock.Anything).Return(nil).Times(1)\n\tenv.SetOnTimerScheduledListener(func(timerID string, duration time.Duration) {\n\t\tif !signalOnce {\n\t\t\tenv.SignalWorkflow(reconciliation.CheckDataCorruptionWorkflowSignalName, entity.Execution{\n\t\t\t\tShardID:    0,\n\t\t\t\tDomainID:   uuid.New(),\n\t\t\t\tWorkflowID: uuid.New(),\n\t\t\t\tRunID:      uuid.New(),\n\t\t\t})\n\t\t\tsignalOnce = true\n\t\t}\n\t})\n\tenv.ExecuteWorkflow(reconciliation.CheckDataCorruptionWorkflowType, nil)\n\ts.True(env.IsWorkflowCompleted())\n\tenv.AssertExpectations(s.T())\n}\n\nfunc (s *dataCorruptionWorkflowTestSuite) TestWorkflow_TimerFire_Success() {\n\tisTimerFire := false\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.SetOnTimerFiredListener(func(timerID string) {\n\t\tisTimerFire = true\n\t})\n\tenv.ExecuteWorkflow(reconciliation.CheckDataCorruptionWorkflowType, nil)\n\ts.True(env.IsWorkflowCompleted())\n\ts.True(isTimerFire)\n\tenv.AssertExpectations(s.T())\n\tenv.AssertNotCalled(s.T(), \"ExecutionFixerActivity\", mock.Anything, mock.Anything)\n\tenv.AssertNotCalled(s.T(), \"EmitResultMetricsActivity\", mock.Anything, mock.Anything)\n}\n\nfunc (s *dataCorruptionWorkflowTestSuite) TestExecutionFixerActivity_Success() {\n\tenv := s.NewTestActivityEnvironment()\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\tdefer mockResource.Finish(s.T())\n\tfixList := []entity.Execution{\n\t\t{\n\t\t\tShardID:    0,\n\t\t\tDomainID:   uuid.New(),\n\t\t\tWorkflowID: uuid.New(),\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t}\n\tmockResource.ExecutionMgr.On(\"GetShardID\").Return(0)\n\tmockResource.ExecutionMgr.On(\"GetCurrentExecution\", mock.Anything, mock.Anything).Return(&p.GetCurrentExecutionResponse{\n\t\tRunID:            fixList[0].RunID,\n\t\tState:            2,\n\t\tCloseStatus:      1,\n\t\tLastWriteVersion: 0,\n\t}, nil)\n\tmockResource.ExecutionMgr.On(\"GetWorkflowExecution\", mock.Anything, mock.Anything).Return(&p.GetWorkflowExecutionResponse{\n\t\tState: &p.WorkflowMutableState{\n\t\t\tExecutionInfo: &p.WorkflowExecutionInfo{\n\t\t\t\tBranchToken: validBranchToken,\n\t\t\t},\n\t\t\tVersionHistories: &p.VersionHistories{\n\t\t\t\tCurrentVersionHistoryIndex: 0,\n\t\t\t\tHistories: []*p.VersionHistory{\n\t\t\t\t\t{\n\t\t\t\t\t\tBranchToken: validBranchToken,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil)\n\tmockResource.ExecutionMgr.On(\"DeleteWorkflowExecution\", mock.Anything, mock.Anything).Return(nil)\n\tmockResource.ExecutionMgr.On(\"DeleteCurrentWorkflowExecution\", mock.Anything, mock.Anything).Return(nil)\n\tmockResource.HistoryMgr.On(\"ReadHistoryBranch\", mock.Anything, mock.Anything).Return(&p.ReadHistoryBranchResponse{}, nil)\n\tctx := context.WithValue(context.Background(), contextKey(testWorkflowName), scannerContext{resource: mockResource})\n\tenv.SetTestTimeout(time.Second * 5)\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: ctx,\n\t})\n\ttlScavengerHBInterval = time.Millisecond * 10\n\tmockResource.DomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain-name\", nil).AnyTimes()\n\tmockResource.DomainCache.EXPECT().GetDomainByID(gomock.Any()).Return(constants.TestGlobalDomainEntry, nil).AnyTimes()\n\t_, err := env.ExecuteActivity(ExecutionFixerActivity, fixList)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/concrete_execution.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"time\"\n\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\nconst (\n\t// ConcreteExecutionsScannerWFTypeName defines workflow type name for concrete executions scanner\n\tConcreteExecutionsScannerWFTypeName   = \"cadence-sys-executions-scanner-workflow\"\n\tconcreteExecutionsScannerWFID         = \"cadence-sys-executions-scanner\"\n\tconcreteExecutionsScannerTaskListName = \"cadence-sys-executions-scanner-tasklist-0\"\n\n\t// ConcreteExecutionsFixerWFTypeName defines workflow type name for concrete executions fixer\n\tConcreteExecutionsFixerWFTypeName   = \"cadence-sys-executions-fixer-workflow\"\n\tconcreteExecutionsFixerWFID         = \"cadence-sys-executions-fixer\"\n\tconcreteExecutionsFixerTaskListName = \"cadence-sys-executions-fixer-tasklist-0\"\n)\n\n// ConcreteScannerWorkflow starts concrete executions scanner.\nfunc ConcreteScannerWorkflow(ctx workflow.Context, params shardscanner.ScannerWorkflowParams) error {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"Starting ConcreteExecutionsScannerWorkflow\", zap.Any(\"Params\", params))\n\n\twf, err := shardscanner.NewScannerWorkflow(ctx, ConcreteExecutionsScannerWFTypeName, params)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to create new scanner workflow\", zap.Error(err))\n\t\treturn err\n\t}\n\n\terr = wf.Start(ctx)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to start scanner workflow\", zap.Error(err))\n\t}\n\treturn err\n}\n\n// ConcreteFixerWorkflow starts concrete executions fixer.\nfunc ConcreteFixerWorkflow(ctx workflow.Context, params shardscanner.FixerWorkflowParams) error {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"Starting ConcreteExecutionsFixerWorkflow\", zap.Any(\"Params\", params))\n\n\twf, err := shardscanner.NewFixerWorkflow(ctx, ConcreteExecutionsFixerWFTypeName, params)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to create new fixer workflow\", zap.Error(err))\n\t\treturn err\n\t}\n\n\terr = wf.Start(ctx)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to start fixer workflow\", zap.Error(err))\n\t}\n\treturn err\n}\n\n// concreteExecutionScannerHooks provides hooks for concrete executions scanner\nfunc concreteExecutionScannerHooks() *shardscanner.ScannerHooks {\n\th, err := shardscanner.NewScannerHooks(concreteExecutionScannerManager, concreteExecutionScannerIterator, concreteExecutionCustomScannerConfig)\n\tif err != nil {\n\t\treturn nil\n\t}\n\treturn h\n}\n\n// concreteExecutionFixerHooks provides hooks needed for concrete executions fixer.\nfunc concreteExecutionFixerHooks() *shardscanner.FixerHooks {\n\th, err := shardscanner.NewFixerHooks(concreteExecutionFixerManager, concreteExecutionFixerIterator, concreteExecutionCustomFixerConfig)\n\tif err != nil {\n\t\treturn nil\n\t}\n\treturn h\n}\n\n// concreteExecutionScannerManager provides invariant manager for concrete execution scanner\nfunc concreteExecutionScannerManager(\n\tctx context.Context,\n\tpr persistence.Retryer,\n\tparams shardscanner.ScanShardActivityParams,\n\tdomainCache cache.DomainCache,\n) invariant.Manager {\n\n\tcollections := ParseCollections(params.ScannerConfig)\n\n\tvar ivs []invariant.Invariant\n\tfor _, fn := range ConcreteExecutionType.ToInvariants(collections, zap.NewNop()) {\n\t\tivs = append(ivs, fn(pr, domainCache))\n\t}\n\n\treturn invariant.NewInvariantManager(ivs)\n}\n\n// concreteExecutionScannerIterator provides iterator for concrete execution scanner.\nfunc concreteExecutionScannerIterator(\n\tctx context.Context,\n\tpr persistence.Retryer,\n\tparams shardscanner.ScanShardActivityParams,\n) pagination.Iterator {\n\tit := ConcreteExecutionType.ToIterator()\n\treturn it(ctx, pr, params.PageSize)\n\n}\n\n// concreteExecutionFixerIterator provides iterator for concrete execution fixer.\nfunc concreteExecutionFixerIterator(ctx context.Context, client blobstore.Client, keys store.Keys, _ shardscanner.FixShardActivityParams) store.ScanOutputIterator {\n\treturn store.NewBlobstoreIterator(ctx, client, keys, ConcreteExecutionType.ToBlobstoreEntity())\n}\n\n// concreteExecutionFixerManager provides invariant manager for concrete execution fixer.\nfunc concreteExecutionFixerManager(_ context.Context, pr persistence.Retryer, params shardscanner.FixShardActivityParams, domainCache cache.DomainCache) invariant.Manager {\n\t// convert to invariants.\n\t// this may produce an empty list if it all fixers are intentionally disabled,\n\t// or if the list came from a previous version of the server which lacked this config.\n\tvar collections []invariant.Collection\n\tfor k, v := range params.EnabledInvariants {\n\t\tif v == strconv.FormatBool(true) {\n\t\t\tivc, err := invariant.CollectionString(k)\n\t\t\tif err != nil {\n\t\t\t\tpanic(fmt.Sprintf(\"invalid collection name: %v\", err)) // error includes the name\n\t\t\t}\n\t\t\tcollections = append(collections, ivc)\n\t\t}\n\t}\n\n\tvar ivs []invariant.Invariant\n\tfor _, fn := range ConcreteExecutionType.ToInvariants(collections, zap.NewNop()) {\n\t\tivs = append(ivs, fn(pr, domainCache))\n\t}\n\treturn invariant.NewInvariantManager(ivs)\n}\n\n// concreteExecutionCustomScannerConfig resolves dynamic config for concrete executions scanner.\nfunc concreteExecutionCustomScannerConfig(ctx shardscanner.ScannerContext) shardscanner.CustomScannerConfig {\n\tres := shardscanner.CustomScannerConfig{}\n\n\tif ctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsScannerInvariantCollectionHistory)() {\n\t\tres[invariant.CollectionHistory.String()] = strconv.FormatBool(true)\n\t}\n\tif ctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsScannerInvariantCollectionMutableState)() {\n\t\tres[invariant.CollectionMutableState.String()] = strconv.FormatBool(true)\n\t}\n\tif ctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsScannerInvariantCollectionStale)() {\n\t\tres[invariant.CollectionStale.String()] = strconv.FormatBool(true)\n\t}\n\n\treturn res\n}\n\n// concreteExecutionCustomFixerConfig resolves dynamic config for concrete executions scanner.\nfunc concreteExecutionCustomFixerConfig(ctx shardscanner.FixerContext) shardscanner.CustomScannerConfig {\n\tres := shardscanner.CustomScannerConfig{}\n\n\t// unlike scanner, fixer expects keys to exist when both true and false, to differentiate from pre-config behavior\n\tres[invariant.CollectionHistory.String()] = strconv.FormatBool(\n\t\tctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsFixerInvariantCollectionHistory)(),\n\t)\n\tres[invariant.CollectionMutableState.String()] = strconv.FormatBool(\n\t\tctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsFixerInvariantCollectionMutableState)(),\n\t)\n\tres[invariant.CollectionStale.String()] = strconv.FormatBool(\n\t\tctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.ConcreteExecutionsFixerInvariantCollectionStale)(),\n\t)\n\n\treturn res\n}\n\n// ConcreteExecutionConfig configures concrete execution scanner\nfunc ConcreteExecutionConfig(dc *dynamicconfig.Collection) *shardscanner.ScannerConfig {\n\treturn &shardscanner.ScannerConfig{\n\t\tScannerWFTypeName: ConcreteExecutionsScannerWFTypeName,\n\t\tFixerWFTypeName:   ConcreteExecutionsFixerWFTypeName,\n\t\tDynamicParams: shardscanner.DynamicParams{\n\t\t\tScannerEnabled:          dc.GetBoolProperty(dynamicproperties.ConcreteExecutionsScannerEnabled),\n\t\t\tFixerEnabled:            dc.GetBoolProperty(dynamicproperties.ConcreteExecutionFixerEnabled),\n\t\t\tConcurrency:             dc.GetIntProperty(dynamicproperties.ConcreteExecutionsScannerConcurrency),\n\t\t\tPageSize:                dc.GetIntProperty(dynamicproperties.ConcreteExecutionsScannerPersistencePageSize),\n\t\t\tBlobstoreFlushThreshold: dc.GetIntProperty(dynamicproperties.ConcreteExecutionsScannerBlobstoreFlushThreshold),\n\t\t\tActivityBatchSize:       dc.GetIntProperty(dynamicproperties.ConcreteExecutionsScannerActivityBatchSize),\n\t\t\tAllowDomain:             dc.GetBoolPropertyFilteredByDomain(dynamicproperties.ConcreteExecutionFixerDomainAllow),\n\t\t},\n\t\tDynamicCollection: dc,\n\t\tScannerHooks:      concreteExecutionScannerHooks,\n\t\tFixerHooks:        concreteExecutionFixerHooks,\n\t\tStartWorkflowOptions: cclient.StartWorkflowOptions{\n\t\t\tID:                           concreteExecutionsScannerWFID,\n\t\t\tTaskList:                     concreteExecutionsScannerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t\tStartFixerOptions: cclient.StartWorkflowOptions{\n\t\t\tID:                           concreteExecutionsFixerWFID,\n\t\t\tTaskList:                     concreteExecutionsFixerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/concrete_execution_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/common/types/testdata\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\ntype concreteExectionsWorkflowsSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestConcreteScannerWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(concreteExectionsWorkflowsSuite))\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) SetupSuite() {\n\tworkflow.Register(ConcreteScannerWorkflow)\n\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) TestScannerWorkflow_Success() {\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(shardscanner.ActivityScannerConfig, mock.Anything, mock.Anything).Return(shardscanner.ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: shardscanner.GenericScannerConfig{\n\t\t\tEnabled:           true,\n\t\t\tConcurrency:       3,\n\t\t\tActivityBatchSize: 5,\n\t\t},\n\t}, nil)\n\tenv.OnActivity(shardscanner.ActivityScannerEmitMetrics, mock.Anything, mock.Anything).Return(nil)\n\tshards := shardscanner.Shards{\n\t\tRange: &shardscanner.ShardRange{\n\t\t\tMin: 0,\n\t\t\tMax: 30,\n\t\t},\n\t}\n\n\tbatches := [][]int{\n\t\t{0, 3, 6, 9, 12},\n\t\t{15, 18, 21, 24, 27},\n\t\t{1, 4, 7, 10, 13},\n\t\t{16, 19, 22, 25, 28},\n\t\t{2, 5, 8, 11, 14},\n\t\t{17, 20, 23, 26, 29},\n\t}\n\n\tfor _, batch := range batches {\n\t\tvar reports []shardscanner.ScanReport\n\t\tfor i := range batch {\n\t\t\tif i == 0 {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t} else {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\tvar customc shardscanner.CustomScannerConfig\n\t\tenv.OnActivity(shardscanner.ActivityScanShard, mock.Anything, shardscanner.ScanShardActivityParams{\n\t\t\tShards:        batch,\n\t\t\tScannerConfig: customc,\n\t\t}).Return(reports, nil)\n\t}\n\n\tenv.ExecuteWorkflow(ConcreteScannerWorkflow, shardscanner.ScannerWorkflowParams{\n\t\tShards: shards,\n\t})\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n\n\taggValue, err := env.QueryWorkflow(shardscanner.AggregateReportQuery)\n\ts.NoError(err)\n\tvar agg shardscanner.AggregateScanReportResult\n\ts.NoError(aggValue.Get(&agg))\n\ts.Equal(shardscanner.AggregateScanReportResult{\n\t\tEntitiesCount:    240,\n\t\tCorruptedCount:   48,\n\t\tCheckFailedCount: 24,\n\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\tinvariant.HistoryExists: 24,\n\t\t},\n\t}, agg)\n\n\tfor i := 0; i < 30; i++ {\n\t\tshardReportValue, err := env.QueryWorkflow(shardscanner.ShardReportQuery, i)\n\t\ts.NoError(err)\n\t\tvar shardReport *shardscanner.ScanReport\n\t\ts.NoError(shardReportValue.Get(&shardReport))\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t} else {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t}\n\t}\n\n\tstatusValue, err := env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar status *shardscanner.ShardStatusQueryResult\n\ts.NoError(statusValue.Get(&status))\n\texpected := make(map[int]shardscanner.ShardStatus)\n\tfor i := 0; i < 30; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.True(status.ShardQueryPaginationToken.IsDone)\n\ts.Nil(status.ShardQueryPaginationToken.NextShardID)\n\n\t// check for paginated query result\n\tstatusValue, err = env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{\n\t\tStartingShardID: common.IntPtr(5),\n\t\tLimitShards:     common.IntPtr(10),\n\t})\n\ts.NoError(err)\n\tstatus = &shardscanner.ShardStatusQueryResult{}\n\ts.NoError(statusValue.Get(&status))\n\texpected = make(map[int]shardscanner.ShardStatus)\n\tfor i := 5; i < 15; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.False(status.ShardQueryPaginationToken.IsDone)\n\ts.Equal(15, *status.ShardQueryPaginationToken.NextShardID)\n\n\tcorruptionKeysValue, err := env.QueryWorkflow(shardscanner.ShardCorruptKeysQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar shardCorruptKeysResult *shardscanner.ShardCorruptKeysQueryResult\n\ts.NoError(corruptionKeysValue.Get(&shardCorruptKeysResult))\n\texpectedCorrupted := make(map[int]store.Keys)\n\tfor i := 0; i < 30; i++ {\n\t\tif i != 0 && i != 1 && i != 2 && i != 15 && i != 16 && i != 17 {\n\t\t\texpectedCorrupted[i] = store.Keys{\n\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\tMinPage: 0,\n\t\t\t\tMaxPage: 10,\n\t\t\t}\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardCorruptKeysResult(expectedCorrupted), shardCorruptKeysResult.Result)\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) TestConcreteScannerWorkflow_NewScannerWorkflow_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tparams := shardscanner.ScannerWorkflowParams{\n\t\tShards: shardscanner.Shards{\n\t\t\tRange: &shardscanner.ShardRange{\n\t\t\t\tMin: 0,\n\t\t\t},\n\t\t},\n\t}\n\n\tenv.ExecuteWorkflow(ConcreteScannerWorkflow, params)\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), \"empty Range provided\")\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) TestConcreteScannerWorkflow_Start_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tparams := shardscanner.ScannerWorkflowParams{\n\t\tShards: shardscanner.Shards{\n\t\t\tRange: &shardscanner.ShardRange{\n\t\t\t\tMin: 0,\n\t\t\t\tMax: 30,\n\t\t\t},\n\t\t},\n\t\tScannerWorkflowConfigOverwrites: shardscanner.ScannerWorkflowConfigOverwrites{},\n\t}\n\n\tcfg := shardscanner.ScannerConfigActivityParams{\n\t\tOverwrites: params.ScannerWorkflowConfigOverwrites,\n\t}\n\n\tenv.OnActivity(shardscanner.ActivityScannerConfig, mock.Anything, cfg).Return(shardscanner.ResolvedScannerWorkflowConfig{}, assert.AnError)\n\n\tenv.ExecuteWorkflow(ConcreteScannerWorkflow, params)\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) TestConcreteFixerWorkflow_NewScannerWorkflow_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tparams := shardscanner.FixerWorkflowParams{\n\t\tScannerWorkflowWorkflowID: constants.TestWorkflowID,\n\t\tScannerWorkflowRunID:      constants.TestRunID,\n\t}\n\n\tfixerCorruptedKeysActivityParams := shardscanner.FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: params.ScannerWorkflowWorkflowID,\n\t\tScannerWorkflowRunID:      params.ScannerWorkflowRunID,\n\t\tStartingShardID:           nil,\n\t}\n\n\tenv.OnActivity(shardscanner.ActivityFixerCorruptedKeys, mock.Anything, fixerCorruptedKeysActivityParams).Return(&shardscanner.FixerCorruptedKeysActivityResult{}, assert.AnError)\n\n\tenv.ExecuteWorkflow(ConcreteFixerWorkflow, params)\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc (s *concreteExectionsWorkflowsSuite) TestConcreteFixerWorkflow_Start_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tparams := shardscanner.FixerWorkflowParams{\n\t\tScannerWorkflowWorkflowID: constants.TestWorkflowID,\n\t\tScannerWorkflowRunID:      constants.TestRunID,\n\t}\n\n\tfixerCorruptedKeysActivityParams := shardscanner.FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: params.ScannerWorkflowWorkflowID,\n\t\tScannerWorkflowRunID:      params.ScannerWorkflowRunID,\n\t\tStartingShardID:           nil,\n\t}\n\n\tenv.OnActivity(shardscanner.ActivityFixerCorruptedKeys, mock.Anything, fixerCorruptedKeysActivityParams).Return(&shardscanner.FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []shardscanner.CorruptedKeysEntry{\n\t\t\t{\n\t\t\t\tShardID: 2,\n\t\t\t\tCorruptedKeys: store.Keys{\n\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\tMinPage: 0,\n\t\t\t\t\tMaxPage: 10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tMinShard: common.IntPtr(0),\n\t\tMaxShard: common.IntPtr(30),\n\t}, nil)\n\n\tenv.OnActivity(shardscanner.ActivityFixerConfig, mock.Anything, shardscanner.FixShardConfigParams{}).Return(&shardscanner.FixShardConfigResults{}, assert.AnError)\n\n\tenv.ExecuteWorkflow(ConcreteFixerWorkflow, params)\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc Test_concreteExecutionScannerHooks(t *testing.T) {\n\th := concreteExecutionScannerHooks()\n\n\tassert.NotNil(t, h)\n}\n\nfunc Test_concreteExecutionFixerHooks(t *testing.T) {\n\th := concreteExecutionFixerHooks()\n\n\tassert.NotNil(t, h)\n}\n\nfunc Test_concreteExecutionScannerManager(t *testing.T) {\n\tparams := shardscanner.ScanShardActivityParams{\n\t\tShards: []int{1, 2, 3},\n\t\tScannerConfig: shardscanner.CustomScannerConfig{\n\t\t\t\"CollectionHistory\": strconv.FormatBool(true),\n\t\t},\n\t}\n\n\tm := concreteExecutionScannerManager(context.Background(), nil, params, nil)\n\n\tassert.NotNil(t, m)\n}\n\nfunc Test_concreteExecutionScannerIterator(t *testing.T) {\n\tparams := shardscanner.ScanShardActivityParams{\n\t\tShards: []int{1, 2, 3},\n\t\tScannerConfig: shardscanner.CustomScannerConfig{\n\t\t\t\"CollectionHistory\": strconv.FormatBool(true),\n\t\t},\n\t\tPageSize: 1,\n\t}\n\n\tctrl := gomock.NewController(t)\n\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\tmockRetryer.EXPECT().ListConcreteExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListConcreteExecutionsResponse{\n\t\tExecutions: []*persistence.ListConcreteExecutionsEntity{\n\t\t\t{\n\t\t\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\t\t\tDomainID:    constants.TestDomainID,\n\t\t\t\t\tWorkflowID:  constants.TestWorkflowID,\n\t\t\t\t\tRunID:       constants.TestRunID,\n\t\t\t\t\tBranchToken: testdata.BranchToken,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\n\ti := concreteExecutionScannerIterator(context.Background(), mockRetryer, params)\n\n\tassert.NotNil(t, i)\n}\n\nfunc Test_concreteExecutionFixerIterator(t *testing.T) {\n\tctx := context.Background()\n\tmockClient := blobstore.NewMockClient(gomock.NewController(t))\n\treq := &blobstore.GetRequest{\n\t\tKey: concreteExecutionsFixerTaskListName + \"_0.\",\n\t}\n\n\tmockClient.EXPECT().Get(ctx, req).Return(&blobstore.GetResponse{}, nil).Times(1)\n\n\tit := concreteExecutionFixerIterator(\n\t\tctx,\n\t\tmockClient,\n\t\tstore.Keys{UUID: concreteExecutionsFixerTaskListName},\n\t\tshardscanner.FixShardActivityParams{})\n\n\tassert.NotNil(t, it)\n}\n\nfunc Test_concreteExecutionFixerManager(t *testing.T) {\n\tmockRetryer := persistence.NewMockRetryer(gomock.NewController(t))\n\n\tparams := shardscanner.FixShardActivityParams{\n\t\tEnabledInvariants: shardscanner.CustomScannerConfig{\n\t\t\t\"CollectionHistory\": strconv.FormatBool(true),\n\t\t},\n\t}\n\n\tm := concreteExecutionFixerManager(context.Background(), mockRetryer, params, nil)\n\n\tassert.NotNil(t, m)\n}\n\nfunc Test_concreteExecutionFixerManager_Panic(t *testing.T) {\n\tmockRetryer := persistence.NewMockRetryer(gomock.NewController(t))\n\n\tparams := shardscanner.FixShardActivityParams{\n\t\tEnabledInvariants: shardscanner.CustomScannerConfig{\n\t\t\t\"NotAnInvariantKey\": strconv.FormatBool(true),\n\t\t},\n\t}\n\tassert.Panics(t, func() {\n\t\tconcreteExecutionFixerManager(context.Background(), mockRetryer, params, nil)\n\t})\n}\n\nfunc Test_concreteExecutionCustomScannerConfig(t *testing.T) {\n\tmockClient := dynamicconfig.NewMockClient(gomock.NewController(t))\n\n\tcollection := dynamicconfig.NewCollection(mockClient, log.NewNoop())\n\n\tmockClient.EXPECT().GetBoolValue(gomock.Any(), gomock.Any()).Return(true, nil).Times(3)\n\n\tctx := shardscanner.ScannerContext{\n\t\tConfig: &shardscanner.ScannerConfig{\n\t\t\tDynamicCollection: collection,\n\t\t},\n\t}\n\n\tcfg := concreteExecutionCustomScannerConfig(ctx)\n\n\tassert.NotNil(t, cfg)\n\tassert.Len(t, cfg, 3)\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionHistory.String()])\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionMutableState.String()])\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionStale.String()])\n}\n\nfunc Test_concreteExecutionCustomFixerConfig(t *testing.T) {\n\tmockClient := dynamicconfig.NewMockClient(gomock.NewController(t))\n\n\tcollection := dynamicconfig.NewCollection(mockClient, log.NewNoop())\n\n\tmockClient.EXPECT().GetBoolValue(gomock.Any(), gomock.Any()).Return(true, nil).Times(3)\n\n\tctx := shardscanner.FixerContext{\n\t\tConfig: &shardscanner.ScannerConfig{\n\t\t\tDynamicCollection: collection,\n\t\t},\n\t}\n\n\tcfg := concreteExecutionCustomFixerConfig(ctx)\n\n\tassert.NotNil(t, cfg)\n\tassert.Len(t, cfg, 3)\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionHistory.String()])\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionMutableState.String()])\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionStale.String()])\n}\n\nfunc TestConcreteExecutionConfig(t *testing.T) {\n\tmockClient := dynamicconfig.NewMockClient(gomock.NewController(t))\n\n\tcollection := dynamicconfig.NewCollection(mockClient, log.NewNoop())\n\n\tcfg := ConcreteExecutionConfig(collection)\n\n\tassert.NotNil(t, cfg)\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/current_execution.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"time\"\n\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\nconst (\n\t// currentExecutionsScannerWFID is the current execution scanner workflow ID\n\tcurrentExecutionsScannerWFID = \"cadence-sys-current-executions-scanner\"\n\t// CurrentExecutionsScannerWFTypeName is the current execution scanner workflow type\n\tCurrentExecutionsScannerWFTypeName = \"cadence-sys-current-executions-scanner-workflow\"\n\t// CurrentExecutionsScannerTaskListName is the current execution scanner workflow tasklist\n\tCurrentExecutionsScannerTaskListName = \"cadence-sys-current-executions-scanner-tasklist-0\"\n\n\t// CurrentExecutionsFixerWFTypeName is the current execution fixer workflow ID\n\tCurrentExecutionsFixerWFTypeName = \"cadence-sys-current-executions-fixer-workflow\"\n\tcurrentExecutionsFixerWFID       = \"cadence-sys-current-executions-fixer\"\n\t// CurrentExecutionsFixerTaskListName is the current execution fixer workflow tasklist\n\tCurrentExecutionsFixerTaskListName = \"cadence-sys-current-executions-fixer-tasklist-0\"\n)\n\n/*\n\n!!!!!!!!!!!!!!\nNOTE: Current execution fixers have never been run.\nBeware drawing any conclusions from current-execution scanner/fixer code.\n!!!!!!!!!!!!!!\n\nWhile this code appears structurally complete, the wrong fixer manager is being\nused, and we have apparently never fully enabled it in our production clusters.\n\nIt likely needs further checks and possibly a rewrite before attempting to use.\n\n*/\n\n// CurrentScannerWorkflow is the workflow that scans over all current executions\nfunc CurrentScannerWorkflow(\n\tctx workflow.Context,\n\tparams shardscanner.ScannerWorkflowParams,\n) error {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"Starting CurrentScannerWorkflow\", zap.Any(\"Params\", params))\n\n\twf, err := shardscanner.NewScannerWorkflow(ctx, CurrentExecutionsScannerWFTypeName, params)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to start new scanner workflow\", zap.Error(err))\n\t\treturn err\n\t}\n\n\terr = wf.Start(ctx)\n\tif err != nil {\n\t\tlogger.Error(\"Failed to execute scanner workflow\", zap.Error(err))\n\t}\n\treturn err\n}\n\n// currentExecutionScannerHooks provides hooks for current executions scanner.\nfunc currentExecutionScannerHooks() *shardscanner.ScannerHooks {\n\twf, err := shardscanner.NewScannerHooks(currentExecutionScannerManager, currentExecutionScannerIterator, currentExecutionCustomScannerConfig)\n\tif err != nil {\n\t\treturn nil\n\t}\n\treturn wf\n}\n\n// currentExecutionScannerManager is the current execution scanner manager\nfunc currentExecutionScannerManager(\n\tctx context.Context,\n\tpr persistence.Retryer,\n\tparams shardscanner.ScanShardActivityParams,\n\tdomainCache cache.DomainCache,\n) invariant.Manager {\n\tlogger := zap.L()\n\tlogger.Info(\"Creating invariant manager for current execution scanner\", zap.Any(\"Params\", params))\n\tvar ivs []invariant.Invariant\n\tcollections := ParseCollections(params.ScannerConfig)\n\tfor _, fn := range CurrentExecutionType.ToInvariants(collections, zap.NewNop()) {\n\t\tivs = append(ivs, fn(pr, domainCache))\n\t}\n\treturn invariant.NewInvariantManager(ivs)\n}\n\n// CurrentFixerWorkflow starts current executions fixer.\nfunc CurrentFixerWorkflow(\n\tctx workflow.Context,\n\tparams shardscanner.FixerWorkflowParams,\n) error {\n\twf, err := shardscanner.NewFixerWorkflow(ctx, CurrentExecutionsFixerWFTypeName, params)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn wf.Start(ctx)\n}\n\n// currentExecutionCustomScannerConfig resolves dynamic config for current executions scanner.\nfunc currentExecutionCustomScannerConfig(ctx shardscanner.ScannerContext) shardscanner.CustomScannerConfig {\n\tres := shardscanner.CustomScannerConfig{}\n\n\tif ctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.CurrentExecutionsScannerInvariantCollectionHistory)() {\n\t\tres[invariant.CollectionHistory.String()] = strconv.FormatBool(true)\n\t}\n\tif ctx.Config.DynamicCollection.GetBoolProperty(dynamicproperties.CurrentExecutionsScannerInvariantCollectionMutableState)() {\n\t\tres[invariant.CollectionMutableState.String()] = strconv.FormatBool(true)\n\t}\n\n\treturn res\n}\n\n// currentExecutionFixerHooks provides hooks for current executions fixer.\nfunc currentExecutionFixerHooks() *shardscanner.FixerHooks {\n\tnoCustomConfig := func(fixer shardscanner.FixerContext) shardscanner.CustomScannerConfig {\n\t\treturn nil\n\t}\n\t// TODO: yes, this DOES incorrectly use the concrete execution fixer manager, which does not work.\n\t// It is retained for now to avoid making a lot of mostly-unrelated changes / fixes / cleanup.\n\th, err := shardscanner.NewFixerHooks(concreteExecutionFixerManager, currentExecutionFixerIterator, noCustomConfig)\n\tif err != nil {\n\t\treturn nil\n\t}\n\treturn h\n}\n\n// CurrentExecutionConfig configures current execution scanner\nfunc CurrentExecutionConfig(dc *dynamicconfig.Collection) *shardscanner.ScannerConfig {\n\treturn &shardscanner.ScannerConfig{\n\t\tScannerWFTypeName: CurrentExecutionsScannerWFTypeName,\n\t\tFixerWFTypeName:   CurrentExecutionsFixerWFTypeName,\n\t\tDynamicCollection: dc,\n\t\tDynamicParams: shardscanner.DynamicParams{\n\t\t\tScannerEnabled:          dc.GetBoolProperty(dynamicproperties.CurrentExecutionsScannerEnabled),\n\t\t\tFixerEnabled:            dc.GetBoolProperty(dynamicproperties.CurrentExecutionFixerEnabled),\n\t\t\tConcurrency:             dc.GetIntProperty(dynamicproperties.CurrentExecutionsScannerConcurrency),\n\t\t\tPageSize:                dc.GetIntProperty(dynamicproperties.CurrentExecutionsScannerPersistencePageSize),\n\t\t\tBlobstoreFlushThreshold: dc.GetIntProperty(dynamicproperties.CurrentExecutionsScannerBlobstoreFlushThreshold),\n\t\t\tActivityBatchSize:       dc.GetIntProperty(dynamicproperties.CurrentExecutionsScannerActivityBatchSize),\n\t\t\tAllowDomain:             dc.GetBoolPropertyFilteredByDomain(dynamicproperties.CurrentExecutionFixerDomainAllow),\n\t\t},\n\t\tScannerHooks: currentExecutionScannerHooks,\n\t\tFixerHooks:   currentExecutionFixerHooks,\n\t\tStartWorkflowOptions: cclient.StartWorkflowOptions{\n\t\t\tID:                           currentExecutionsScannerWFID,\n\t\t\tTaskList:                     CurrentExecutionsScannerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t\tStartFixerOptions: cclient.StartWorkflowOptions{\n\t\t\tID:                           currentExecutionsFixerWFID,\n\t\t\tTaskList:                     CurrentExecutionsFixerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t}\n}\n\n// currentExecutionScannerIterator is the iterator of current executions\nfunc currentExecutionScannerIterator(\n\tctx context.Context,\n\tpr persistence.Retryer,\n\tparams shardscanner.ScanShardActivityParams,\n) pagination.Iterator {\n\treturn CurrentExecutionType.ToIterator()(ctx, pr, params.PageSize)\n}\n\n// currentExecutionFixerIterator is the iterator of fixer execution\nfunc currentExecutionFixerIterator(\n\tctx context.Context,\n\tclient blobstore.Client,\n\tkeys store.Keys,\n\t_ shardscanner.FixShardActivityParams,\n) store.ScanOutputIterator {\n\treturn store.NewBlobstoreIterator(ctx, client, keys, CurrentExecutionType.ToBlobstoreEntity())\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/current_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/history/constants\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\ntype currentExecutionsWorkflowsSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestCurrentScannerWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(currentExecutionsWorkflowsSuite))\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) SetupSuite() {\n\tworkflow.Register(CurrentScannerWorkflow)\n\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestScannerWorkflow_Success() {\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(shardscanner.ActivityScannerConfig, mock.Anything, mock.Anything).Return(shardscanner.ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: shardscanner.GenericScannerConfig{\n\t\t\tEnabled:           true,\n\t\t\tConcurrency:       3,\n\t\t\tActivityBatchSize: 5,\n\t\t},\n\t}, nil)\n\tenv.OnActivity(shardscanner.ActivityScannerEmitMetrics, mock.Anything, mock.Anything).Return(nil)\n\tshards := shardscanner.Shards{\n\t\tRange: &shardscanner.ShardRange{\n\t\t\tMin: 0,\n\t\t\tMax: 30,\n\t\t},\n\t}\n\n\tbatches := [][]int{\n\t\t{0, 3, 6, 9, 12},\n\t\t{15, 18, 21, 24, 27},\n\t\t{1, 4, 7, 10, 13},\n\t\t{16, 19, 22, 25, 28},\n\t\t{2, 5, 8, 11, 14},\n\t\t{17, 20, 23, 26, 29},\n\t}\n\n\tfor _, batch := range batches {\n\t\tvar reports []shardscanner.ScanReport\n\t\tfor i := range batch {\n\t\t\tif i == 0 {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t} else {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\tvar customScannerConfig shardscanner.CustomScannerConfig\n\t\tenv.OnActivity(shardscanner.ActivityScanShard, mock.Anything, shardscanner.ScanShardActivityParams{\n\t\t\tShards:        batch,\n\t\t\tScannerConfig: customScannerConfig,\n\t\t}).Return(reports, nil)\n\t}\n\n\tenv.ExecuteWorkflow(CurrentScannerWorkflow, shardscanner.ScannerWorkflowParams{\n\t\tShards: shards,\n\t})\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n\n\taggValue, err := env.QueryWorkflow(shardscanner.AggregateReportQuery)\n\ts.NoError(err)\n\tvar agg shardscanner.AggregateScanReportResult\n\ts.NoError(aggValue.Get(&agg))\n\ts.Equal(shardscanner.AggregateScanReportResult{\n\t\tEntitiesCount:    240,\n\t\tCorruptedCount:   48,\n\t\tCheckFailedCount: 24,\n\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\tinvariant.HistoryExists: 24,\n\t\t},\n\t}, agg)\n\n\tfor i := 0; i < 30; i++ {\n\t\tshardReportValue, err := env.QueryWorkflow(shardscanner.ShardReportQuery, i)\n\t\ts.NoError(err)\n\t\tvar shardReport *shardscanner.ScanReport\n\t\ts.NoError(shardReportValue.Get(&shardReport))\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t} else {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t}\n\t}\n\n\tstatusValue, err := env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar status *shardscanner.ShardStatusQueryResult\n\ts.NoError(statusValue.Get(&status))\n\texpected := make(map[int]shardscanner.ShardStatus)\n\tfor i := 0; i < 30; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.True(status.ShardQueryPaginationToken.IsDone)\n\ts.Nil(status.ShardQueryPaginationToken.NextShardID)\n\n\t// check for paginated query result\n\tstatusValue, err = env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{\n\t\tStartingShardID: common.IntPtr(5),\n\t\tLimitShards:     common.IntPtr(10),\n\t})\n\ts.NoError(err)\n\tstatus = &shardscanner.ShardStatusQueryResult{}\n\ts.NoError(statusValue.Get(&status))\n\texpected = make(map[int]shardscanner.ShardStatus)\n\tfor i := 5; i < 15; i++ {\n\t\tif i == 0 || i == 1 || i == 2 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.False(status.ShardQueryPaginationToken.IsDone)\n\ts.Equal(15, *status.ShardQueryPaginationToken.NextShardID)\n\n\tcorruptionKeysValue, err := env.QueryWorkflow(shardscanner.ShardCorruptKeysQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar shardCorruptKeysResult *shardscanner.ShardCorruptKeysQueryResult\n\ts.NoError(corruptionKeysValue.Get(&shardCorruptKeysResult))\n\texpectedCorrupted := make(map[int]store.Keys)\n\tfor i := 0; i < 30; i++ {\n\t\tif i != 0 && i != 1 && i != 2 && i != 15 && i != 16 && i != 17 {\n\t\t\texpectedCorrupted[i] = store.Keys{\n\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\tMinPage: 0,\n\t\t\t\tMaxPage: 10,\n\t\t\t}\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardCorruptKeysResult(expectedCorrupted), shardCorruptKeysResult.Result)\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestScannerWorkflow_NewScannerWorkflow_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tenv.ExecuteWorkflow(CurrentScannerWorkflow, shardscanner.ScannerWorkflowParams{})\n\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), \"must provide either List or Range\")\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestScannerWorkflow_Start_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(shardscanner.ActivityScannerConfig, mock.Anything, mock.Anything).Return(shardscanner.ResolvedScannerWorkflowConfig{}, assert.AnError)\n\tenv.ExecuteWorkflow(CurrentScannerWorkflow, shardscanner.ScannerWorkflowParams{\n\t\tShards: shardscanner.Shards{\n\t\t\tList: []int{1},\n\t\t},\n\t})\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc Test_currentExecutionScannerHooks(t *testing.T) {\n\thooks := currentExecutionScannerHooks()\n\tassert.NotNil(t, hooks)\n}\n\nfunc Test_currentExecutionScannerManager(t *testing.T) {\n\tmockRetryer := persistence.NewMockRetryer(gomock.NewController(t))\n\tparams := shardscanner.ScanShardActivityParams{\n\t\tScannerConfig: shardscanner.CustomScannerConfig{\n\t\t\t\"CollectionMutableState\": \"true\",\n\t\t},\n\t}\n\n\tmanager := currentExecutionScannerManager(context.Background(), mockRetryer, params, nil)\n\tassert.NotNil(t, manager)\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestCurrentFixerWorkflow_NewFixerWorkflow_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tenv.OnActivity(shardscanner.ActivityFixerCorruptedKeys, mock.Anything, mock.Anything).Return(&shardscanner.FixerCorruptedKeysActivityResult{}, assert.AnError)\n\n\tenv.ExecuteWorkflow(CurrentFixerWorkflow, shardscanner.FixerWorkflowParams{})\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestCurrentFixerWorkflow_Start_Error() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tenv.OnActivity(shardscanner.ActivityFixerCorruptedKeys, mock.Anything, mock.Anything).\n\t\tReturn(&shardscanner.FixerCorruptedKeysActivityResult{\n\t\t\tCorruptedKeys: []shardscanner.CorruptedKeysEntry{{ShardID: 1, CorruptedKeys: store.Keys{UUID: uuid.New()}}},\n\t\t\tMinShard:      common.IntPtr(1),\n\t\t\tMaxShard:      common.IntPtr(3),\n\t\t}, nil)\n\n\tenv.OnActivity(shardscanner.ActivityFixerConfig, mock.Anything, mock.Anything).Return(&shardscanner.FixShardConfigResults{}, assert.AnError)\n\n\tenv.ExecuteWorkflow(CurrentFixerWorkflow, shardscanner.FixerWorkflowParams{})\n\ts.True(env.IsWorkflowCompleted())\n\ts.ErrorContains(env.GetWorkflowError(), assert.AnError.Error())\n}\n\nfunc (s *currentExecutionsWorkflowsSuite) TestCurrentFixerWorkflow_Success() {\n\tenv := s.NewTestWorkflowEnvironment()\n\n\tenv.OnActivity(shardscanner.ActivityFixerCorruptedKeys, mock.Anything, mock.Anything).\n\t\tReturn(&shardscanner.FixerCorruptedKeysActivityResult{\n\t\t\tCorruptedKeys: []shardscanner.CorruptedKeysEntry{{ShardID: 1, CorruptedKeys: store.Keys{UUID: uuid.New()}}},\n\t\t\tMinShard:      common.IntPtr(1),\n\t\t\tMaxShard:      common.IntPtr(3),\n\t\t}, nil)\n\n\tenv.OnActivity(shardscanner.ActivityFixerConfig, mock.Anything, mock.Anything).Return(&shardscanner.FixShardConfigResults{}, nil)\n\n\tenv.OnActivity(shardscanner.ActivityFixShard, mock.Anything, mock.Anything).Return([]shardscanner.FixReport{{ShardID: 1}}, nil)\n\n\tenv.ExecuteWorkflow(CurrentFixerWorkflow, shardscanner.FixerWorkflowParams{})\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n}\n\nfunc Test_currentExecutionCustomScannerConfig(t *testing.T) {\n\tmockClient := dynamicconfig.NewMockClient(gomock.NewController(t))\n\n\tcollection := dynamicconfig.NewCollection(mockClient, log.NewNoop())\n\n\tmockClient.EXPECT().GetBoolValue(gomock.Any(), gomock.Any()).Return(true, nil).Times(2)\n\n\tctx := shardscanner.ScannerContext{\n\t\tConfig: &shardscanner.ScannerConfig{\n\t\t\tDynamicCollection: collection,\n\t\t},\n\t}\n\n\tcfg := currentExecutionCustomScannerConfig(ctx)\n\n\tassert.NotNil(t, cfg)\n\tassert.Len(t, cfg, 2)\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionHistory.String()])\n\tassert.Equal(t, \"true\", cfg[invariant.CollectionMutableState.String()])\n}\n\nfunc Test_currentExecutionFixerHooks(t *testing.T) {\n\th := currentExecutionFixerHooks()\n\n\tassert.NotNil(t, h)\n}\n\nfunc TestCurrentExecutionConfig(t *testing.T) {\n\tmockClient := dynamicconfig.NewMockClient(gomock.NewController(t))\n\n\tcollection := dynamicconfig.NewCollection(mockClient, log.NewNoop())\n\n\tcfg := CurrentExecutionConfig(collection)\n\n\tassert.NotNil(t, cfg)\n}\n\nfunc Test_currentExecutionScannerIterator(t *testing.T) {\n\tparams := shardscanner.ScanShardActivityParams{\n\t\tShards: []int{1, 2, 3},\n\t\tScannerConfig: shardscanner.CustomScannerConfig{\n\t\t\t\"CollectionHistory\": strconv.FormatBool(true),\n\t\t},\n\t\tPageSize: 1,\n\t}\n\n\tctrl := gomock.NewController(t)\n\tmockRetryer := persistence.NewMockRetryer(ctrl)\n\n\tmockRetryer.EXPECT().ListCurrentExecutions(gomock.Any(), gomock.Any()).Return(&persistence.ListCurrentExecutionsResponse{\n\t\tExecutions: []*persistence.CurrentWorkflowExecution{\n\t\t\t{\n\t\t\t\tDomainID:   constants.TestDomainID,\n\t\t\t\tWorkflowID: constants.TestWorkflowID,\n\t\t\t\tRunID:      constants.TestRunID,\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\n\tmockRetryer.EXPECT().GetShardID().Return(2)\n\n\ti := currentExecutionScannerIterator(context.Background(), mockRetryer, params)\n\n\tassert.NotNil(t, i)\n}\n\nfunc Test_currentExecutionFixerIterator(t *testing.T) {\n\tctx := context.Background()\n\tmockClient := blobstore.NewMockClient(gomock.NewController(t))\n\treq := &blobstore.GetRequest{\n\t\tKey: CurrentExecutionsFixerTaskListName + \"_0.\",\n\t}\n\n\tmockClient.EXPECT().Get(ctx, req).Return(&blobstore.GetResponse{}, nil).Times(1)\n\n\tit := currentExecutionFixerIterator(\n\t\tctx,\n\t\tmockClient,\n\t\tstore.Keys{UUID: CurrentExecutionsFixerTaskListName},\n\t\tshardscanner.FixShardActivityParams{})\n\n\tassert.NotNil(t, it)\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/scantype_enumer_generated.go",
    "content": "// Code generated by \"enumer -type=ScanType -output scantype_enumer_generated.go\"; DO NOT EDIT.\n\npackage executions\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\nconst _ScanTypeName = \"ConcreteExecutionTypeCurrentExecutionType\"\n\nvar _ScanTypeIndex = [...]uint8{0, 21, 41}\n\nconst _ScanTypeLowerName = \"concreteexecutiontypecurrentexecutiontype\"\n\nfunc (i ScanType) String() string {\n\tif i < 0 || i >= ScanType(len(_ScanTypeIndex)-1) {\n\t\treturn fmt.Sprintf(\"ScanType(%d)\", i)\n\t}\n\treturn _ScanTypeName[_ScanTypeIndex[i]:_ScanTypeIndex[i+1]]\n}\n\n// An \"invalid array index\" compiler error signifies that the constant values have changed.\n// Re-run the stringer command to generate them again.\nfunc _ScanTypeNoOp() {\n\tvar x [1]struct{}\n\t_ = x[ConcreteExecutionType-(0)]\n\t_ = x[CurrentExecutionType-(1)]\n}\n\nvar _ScanTypeValues = []ScanType{ConcreteExecutionType, CurrentExecutionType}\n\nvar _ScanTypeNameToValueMap = map[string]ScanType{\n\t_ScanTypeName[0:21]:       ConcreteExecutionType,\n\t_ScanTypeLowerName[0:21]:  ConcreteExecutionType,\n\t_ScanTypeName[21:41]:      CurrentExecutionType,\n\t_ScanTypeLowerName[21:41]: CurrentExecutionType,\n}\n\nvar _ScanTypeNames = []string{\n\t_ScanTypeName[0:21],\n\t_ScanTypeName[21:41],\n}\n\n// ScanTypeString retrieves an enum value from the enum constants string name.\n// Throws an error if the param is not part of the enum.\nfunc ScanTypeString(s string) (ScanType, error) {\n\tif val, ok := _ScanTypeNameToValueMap[s]; ok {\n\t\treturn val, nil\n\t}\n\n\tif val, ok := _ScanTypeNameToValueMap[strings.ToLower(s)]; ok {\n\t\treturn val, nil\n\t}\n\treturn 0, fmt.Errorf(\"%s does not belong to ScanType values\", s)\n}\n\n// ScanTypeValues returns all values of the enum\nfunc ScanTypeValues() []ScanType {\n\treturn _ScanTypeValues\n}\n\n// ScanTypeStrings returns a slice of all String values of the enum\nfunc ScanTypeStrings() []string {\n\tstrs := make([]string, len(_ScanTypeNames))\n\tcopy(strs, _ScanTypeNames)\n\treturn strs\n}\n\n// IsAScanType returns \"true\" if the value is listed in the enum definition. \"false\" otherwise\nfunc (i ScanType) IsAScanType() bool {\n\tfor _, v := range _ScanTypeValues {\n\t\tif i == v {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n\n//go:generate enumer -type=ScanType -output scantype_enumer_generated.go\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/fetcher\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\nconst (\n\t// ConcreteExecutionType concrete execution entity\n\tConcreteExecutionType ScanType = iota\n\t// CurrentExecutionType current execution entity\n\tCurrentExecutionType\n)\n\n// ScanType is the enum for representing different entity types to scan\ntype ScanType int\n\ntype (\n\t// InvariantFactory represents a function which returns Invariant\n\tInvariantFactory func(retryer persistence.Retryer, domainCache cache.DomainCache) invariant.Invariant\n\n\t// ExecutionFetcher represents a function which returns specific execution entity\n\tExecutionFetcher func(ctx context.Context, retryer persistence.Retryer, request fetcher.ExecutionRequest) (entity.Entity, error)\n)\n\n// ToBlobstoreEntity picks struct depending on scanner type.\nfunc (st ScanType) ToBlobstoreEntity() entity.Entity {\n\tswitch st {\n\tcase ConcreteExecutionType:\n\t\treturn &entity.ConcreteExecution{}\n\tcase CurrentExecutionType:\n\t\treturn &entity.CurrentExecution{}\n\t}\n\tpanic(\"unknown scan type\")\n}\n\n// ToIterator selects appropriate iterator. It will panic if scan type is unknown.\nfunc (st ScanType) ToIterator() func(ctx context.Context, retryer persistence.Retryer, pageSize int) pagination.Iterator {\n\tswitch st {\n\tcase ConcreteExecutionType:\n\t\treturn fetcher.ConcreteExecutionIterator\n\tcase CurrentExecutionType:\n\t\treturn fetcher.CurrentExecutionIterator\n\tdefault:\n\t\tpanic(\"unknown scan type\")\n\t}\n}\n\n// ToExecutionFetcher selects appropriate execution fetcher. Fetcher returns single execution entity. It will panic if scan type is unknown.\nfunc (st ScanType) ToExecutionFetcher() ExecutionFetcher {\n\tswitch st {\n\tcase ConcreteExecutionType:\n\t\treturn fetcher.ConcreteExecution\n\tcase CurrentExecutionType:\n\t\treturn fetcher.CurrentExecution\n\tdefault:\n\t\tpanic(\"unknown scan type\")\n\t}\n}\n\n// ToInvariants returns list of invariants to be checked depending on scan type.\nfunc (st ScanType) ToInvariants(collections []invariant.Collection, logger *zap.Logger) []InvariantFactory {\n\tvar fns []InvariantFactory\n\tswitch st {\n\tcase ConcreteExecutionType:\n\t\tfor _, collection := range collections {\n\t\t\tswitch collection {\n\t\t\tcase invariant.CollectionDomain:\n\t\t\t\tfns = append(fns, invariant.NewInactiveDomainExists)\n\t\t\tcase invariant.CollectionHistory:\n\t\t\t\tfns = append(fns, invariant.NewHistoryExists)\n\t\t\tcase invariant.CollectionStale:\n\t\t\t\tfns = append(fns, func(pr persistence.Retryer, dc cache.DomainCache) invariant.Invariant {\n\t\t\t\t\treturn invariant.NewStaleWorkflow(pr, dc, logger.Named(string(invariant.StaleWorkflow)))\n\t\t\t\t})\n\t\t\tcase invariant.CollectionMutableState:\n\t\t\t\tfns = append(fns, invariant.NewOpenCurrentExecution)\n\t\t\t}\n\t\t}\n\t\treturn fns\n\tcase CurrentExecutionType:\n\t\tfor _, collection := range collections {\n\t\t\tswitch collection {\n\t\t\tcase invariant.CollectionMutableState:\n\t\t\t\tfns = append(fns, invariant.NewConcreteExecutionExists)\n\t\t\t}\n\t\t}\n\t\treturn fns\n\tdefault:\n\t\tpanic(\"unknown scan type\")\n\t}\n}\n\n// ParseCollections converts string based map to list of collections\nfunc ParseCollections(params shardscanner.CustomScannerConfig) []invariant.Collection {\n\tvar collections []invariant.Collection\n\n\tfor k, v := range params {\n\t\tc, e := invariant.CollectionString(k)\n\t\tif e != nil {\n\t\t\tcontinue\n\t\t}\n\t\tenabled, err := strconv.ParseBool(v)\n\t\tif enabled && err == nil {\n\t\t\tcollections = append(collections, c)\n\t\t}\n\t}\n\treturn collections\n}\n"
  },
  {
    "path": "service/worker/scanner/executions/workflows_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage executions\n"
  },
  {
    "path": "service/worker/scanner/executor/executor.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage executor\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\t// Task defines the interface for a runnable task\n\tTask interface {\n\t\t// Run should execute the task and return well known status codes\n\t\tRun() TaskStatus\n\t}\n\n\t// Executor defines the interface for any executor which can\n\t// accept tasks and execute them based on some policy\n\tExecutor interface {\n\t\t// Start starts the executor\n\t\tStart()\n\t\t// Stop stops the executor\n\t\tStop()\n\t\t// Submit is a blocking call that accepts a task to execute\n\t\tSubmit(task Task) bool\n\t\t// TaskCount returns the number of outstanding tasks in the executor\n\t\tTaskCount() int64\n\t}\n\n\t// fixedPoolExecutor is an implementation of an executor that uses fixed size\n\t// goroutine pool. This executor also supports deferred execution of tasks\n\t// for fairness\n\tfixedPoolExecutor struct {\n\t\tsize        int\n\t\tmaxDeferred int\n\t\trunQ        *runQueue\n\t\toutstanding int64\n\t\tstatus      int32\n\t\tmetrics     metrics.Client\n\t\tmetricScope metrics.ScopeIdx\n\t\tstopC       chan struct{}\n\t\tstopWG      sync.WaitGroup\n\t}\n\n\t// TaskStatus is the return code from a Task\n\tTaskStatus int\n)\n\nconst (\n\t// TaskStatusDone indicates task is finished successfully\n\tTaskStatusDone TaskStatus = iota\n\t// TaskStatusDefer indicates task should be scheduled again for execution at later time\n\tTaskStatusDefer\n\t// TaskStatusErr indicates task is finished with errors\n\tTaskStatusErr\n)\n\n// NewFixedSizePoolExecutor returns an implementation of task executor that maintains\n// a fixed size pool of goroutines.The returned executor also allows task processing to\n// to be deferred for fairness. To defer processing of a task, simply return TaskStatsDefer\n// from your task.Run method. When a task is deferred, it will be added to the tail of a\n// deferredTaskQ which in turn will be processed after the current runQ is drained\nfunc NewFixedSizePoolExecutor(size int, maxDeferred int, metrics metrics.Client, scope metrics.ScopeIdx) Executor {\n\tstopC := make(chan struct{})\n\treturn &fixedPoolExecutor{\n\t\tsize:        size,\n\t\tmaxDeferred: maxDeferred,\n\t\trunQ:        newRunQueue(size, stopC),\n\t\tmetrics:     metrics,\n\t\tmetricScope: scope,\n\t\tstopC:       stopC,\n\t}\n}\n\n// Start starts the executor\nfunc (e *fixedPoolExecutor) Start() {\n\tif !atomic.CompareAndSwapInt32(&e.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tfor i := 0; i < e.size; i++ {\n\t\te.stopWG.Add(1)\n\t\tgo e.worker()\n\t}\n}\n\n// Stop stops the executor\nfunc (e *fixedPoolExecutor) Stop() {\n\tif !atomic.CompareAndSwapInt32(&e.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\tclose(e.stopC)\n\te.stopWG.Wait()\n}\n\n// Submit is a blocking call that accepts a task for execution\nfunc (e *fixedPoolExecutor) Submit(task Task) bool {\n\tif !e.alive() {\n\t\treturn false\n\t}\n\tadded := e.runQ.add(task)\n\tif added {\n\t\tatomic.AddInt64(&e.outstanding, 1)\n\t}\n\treturn added\n}\n\n// TaskCount returns the total of number of tasks currently outstanding\nfunc (e *fixedPoolExecutor) TaskCount() int64 {\n\treturn atomic.LoadInt64(&e.outstanding)\n}\n\nfunc (e *fixedPoolExecutor) worker() {\n\tdefer e.stopWG.Done()\n\tfor e.alive() {\n\t\ttask, ok := e.runQ.remove()\n\t\tif !ok {\n\t\t\treturn\n\t\t}\n\t\tstatus := task.Run()\n\t\tif status == TaskStatusDefer {\n\t\t\tif e.runQ.deferredCount() < e.maxDeferred {\n\t\t\t\te.runQ.addAndDefer(task)\n\t\t\t\te.metrics.IncCounter(e.metricScope, metrics.ExecutorTasksDeferredCount)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\te.metrics.IncCounter(e.metricScope, metrics.ExecutorTasksDroppedCount)\n\t\t}\n\t\tatomic.AddInt64(&e.outstanding, -1)\n\t}\n}\n\nfunc (e *fixedPoolExecutor) alive() bool {\n\treturn atomic.LoadInt32(&e.status) == common.DaemonStatusStarted\n}\n"
  },
  {
    "path": "service/worker/scanner/executor/executor_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage executor\n\nimport (\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\ntype (\n\tExecutorTestSuite struct {\n\t\tsuite.Suite\n\t}\n\ttestTask struct {\n\t\tnext    TaskStatus\n\t\tcounter *int64\n\t}\n)\n\nfunc TestExecutionTestSuite(t *testing.T) {\n\tsuite.Run(t, new(ExecutorTestSuite))\n}\n\nfunc (s *ExecutorTestSuite) TestStartStop() {\n\te := NewFixedSizePoolExecutor(\n\t\t4, 4, metrics.NewClient(tally.NoopScope, metrics.Worker, metrics.HistogramMigration{}), metrics.TaskListScavengerScope)\n\te.Start()\n\te.Stop()\n}\n\nfunc (s *ExecutorTestSuite) TestTaskExecution() {\n\te := NewFixedSizePoolExecutor(\n\t\t32, 100, metrics.NewClient(tally.NoopScope, metrics.Worker, metrics.HistogramMigration{}), metrics.TaskListScavengerScope)\n\te.Start()\n\tvar runCounter int64\n\tvar startWG sync.WaitGroup\n\tfor i := 0; i < 5; i++ {\n\t\tstartWG.Add(1)\n\t\tgo func() {\n\t\t\tdefer startWG.Done()\n\t\t\tfor i := 0; i < 20; i++ {\n\t\t\t\tif i%2 == 0 {\n\t\t\t\t\te.Submit(&testTask{TaskStatusDefer, &runCounter})\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\te.Submit(&testTask{TaskStatusDone, &runCounter})\n\t\t\t}\n\t\t}()\n\t}\n\ts.True(common.AwaitWaitGroup(&startWG, time.Second*10))\n\ts.True(s.awaitCompletion(e))\n\ts.Equal(int64(150), runCounter)\n\te.Stop()\n}\n\nfunc (s *ExecutorTestSuite) awaitCompletion(e Executor) bool {\n\texpiry := time.Now().Add(time.Second * 10)\n\tfor time.Now().Before(expiry) {\n\t\tif e.TaskCount() == 0 {\n\t\t\treturn true\n\t\t}\n\t\ttime.Sleep(time.Millisecond * 50)\n\t}\n\treturn false\n}\n\nfunc (tt *testTask) Run() TaskStatus {\n\tatomic.AddInt64(tt.counter, 1)\n\tstatus := tt.next\n\tif status == TaskStatusDefer {\n\t\ttt.next = TaskStatusDone\n\t}\n\treturn status\n}\n"
  },
  {
    "path": "service/worker/scanner/executor/runq.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage executor\n\nimport (\n\t\"container/list\"\n\t\"sync\"\n)\n\ntype (\n\trunQueue struct {\n\t\ttasks         chan Task\n\t\tdeferredTasks *threadSafeList\n\t\tstopC         chan struct{}\n\t}\n\n\t// threadSafeList is a wrapper around list.List which\n\t// guarantees thread safety\n\tthreadSafeList struct {\n\t\tsync.Mutex\n\t\tlist *list.List\n\t}\n)\n\n// newRunQueue returns an instance of task run queue\nfunc newRunQueue(size int, stopC chan struct{}) *runQueue {\n\treturn &runQueue{\n\t\ttasks:         make(chan Task, size),\n\t\tdeferredTasks: newThreadSafeList(),\n\t\tstopC:         stopC,\n\t}\n}\n\nfunc (rq *runQueue) add(task Task) bool {\n\tselect {\n\tcase rq.tasks <- task:\n\t\treturn true\n\tcase <-rq.stopC:\n\t\treturn false\n\t}\n}\n\nfunc (rq *runQueue) addAndDefer(task Task) {\n\trq.deferredTasks.add(task)\n}\n\n// remove returns the next task from run queue\n// if there are no active tasks in runq, a task\n// from deferredQ is returned\nfunc (rq *runQueue) remove() (Task, bool) {\n\t// Give 1st priority to taskQ -if there are no more tasks, handout tasks from deferredQ\n\tselect {\n\tcase job, ok := <-rq.tasks:\n\t\tif !ok {\n\t\t\treturn nil, false\n\t\t}\n\t\treturn job, true\n\tdefault:\n\t\tif task := rq.deferredTasks.remove(); task != nil {\n\t\t\treturn task.Value.(Task), true\n\t\t}\n\t}\n\t// at this point, there are no tasks either in taskQ or deferredQ\n\t// simply block on taskQ until we have tasks\n\tselect {\n\tcase job, ok := <-rq.tasks:\n\t\tif !ok {\n\t\t\treturn nil, false\n\t\t}\n\t\treturn job, true\n\tcase <-rq.stopC:\n\t\treturn nil, false\n\t}\n}\n\nfunc (rq *runQueue) deferredCount() int {\n\treturn rq.deferredTasks.len()\n}\n\n// newThreadSafeList returns a new thread safe linked list\nfunc newThreadSafeList() *threadSafeList {\n\treturn &threadSafeList{\n\t\tlist: list.New(),\n\t}\n}\n\nfunc (tl *threadSafeList) add(elem interface{}) {\n\ttl.Lock()\n\tdefer tl.Unlock()\n\ttl.list.PushBack(elem)\n}\n\nfunc (tl *threadSafeList) len() int {\n\ttl.Lock()\n\tdefer tl.Unlock()\n\treturn tl.list.Len()\n}\n\nfunc (tl *threadSafeList) remove() *list.Element {\n\ttl.Lock()\n\tdefer tl.Unlock()\n\te := tl.list.Front()\n\tif e != nil {\n\t\ttl.list.Remove(e)\n\t}\n\treturn e\n}\n"
  },
  {
    "path": "service/worker/scanner/history/scavenger.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\t// ScavengerHeartbeatDetails is the heartbeat detail for HistoryScavengerActivity\n\tScavengerHeartbeatDetails struct {\n\t\tNextPageToken []byte\n\t\tCurrentPage   int\n\t\tSkipCount     int\n\t\tErrorCount    int\n\t\tSuccCount     int\n\t}\n\n\t// Scavenger is the type that holds the state for history scavenger daemon\n\tScavenger struct {\n\t\tdb                         p.HistoryManager\n\t\tclient                     history.Client\n\t\thbd                        ScavengerHeartbeatDetails\n\t\trps                        int\n\t\tlimiter                    *rate.Limiter\n\t\tmaxWorkflowRetentionInDays dynamicproperties.IntPropertyFn\n\t\tmetrics                    metrics.Client\n\t\tlogger                     log.Logger\n\t\tisInTest                   bool\n\t\tdomainCache                cache.DomainCache\n\t}\n\n\ttaskDetail struct {\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\t\ttreeID     string\n\t\tbranchID   string\n\n\t\t// passing along the current heartbeat details to make heartbeat within a task so that it won't timeout\n\t\thbd ScavengerHeartbeatDetails\n\t}\n)\n\nconst (\n\t// used this to decide how many goroutines to process\n\trpsPerConcurrency = 50\n\tpageSize          = 1000\n)\n\n// only clean up history branches that older than this threshold\n// we double the MaxWorkflowRetentionPeriodInDays to avoid racing condition with history archival.\n// Our history archiver delete mutable state, and then upload history to blob store and then delete history.\n// This scanner will face racing condition with archiver because it relys on describe mutable state returning entityNotExist error.\n// That's why we need to keep MaxWorkflowRetentionPeriodInDays stable and not decreasing all the time.\nfunc getHistoryCleanupThreshold(maxWorkflowRetentionInDays int) time.Duration {\n\treturn time.Hour * 24 * time.Duration(maxWorkflowRetentionInDays) * 2\n}\n\n// NewScavenger returns an instance of history scavenger daemon\n// The Scavenger can be started by calling the Run() method on the\n// returned object. Calling the Run() method will result in one\n// complete iteration over all of the history branches in the system. For\n// each branch, the scavenger will attempt\n//   - describe the corresponding workflow execution\n//   - deletion of history itself, if there are no workflow execution\nfunc NewScavenger(\n\tdb p.HistoryManager,\n\trps int,\n\tclient history.Client,\n\thbd ScavengerHeartbeatDetails,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\tmaxWorkflowRetentionInDays dynamicproperties.IntPropertyFn,\n\tdomainCache cache.DomainCache,\n) *Scavenger {\n\n\trateLimiter := rate.NewLimiter(rate.Limit(rps), rps)\n\n\treturn &Scavenger{\n\t\tdb:                         db,\n\t\tclient:                     client,\n\t\thbd:                        hbd,\n\t\trps:                        rps,\n\t\tlimiter:                    rateLimiter,\n\t\tmaxWorkflowRetentionInDays: maxWorkflowRetentionInDays,\n\t\tmetrics:                    metricsClient,\n\t\tlogger:                     logger,\n\t\tdomainCache:                domainCache,\n\t}\n}\n\n// Run runs the scavenger\nfunc (s *Scavenger) Run(ctx context.Context) (ScavengerHeartbeatDetails, error) {\n\ttaskCh := make(chan taskDetail, pageSize)\n\trespCh := make(chan error, pageSize)\n\tconcurrency := s.rps/rpsPerConcurrency + 1\n\n\tfor i := 0; i < concurrency; i++ {\n\t\tgo s.startTaskProcessor(ctx, taskCh, respCh)\n\t}\n\n\tfor {\n\t\tresp, err := s.db.GetAllHistoryTreeBranches(ctx, &p.GetAllHistoryTreeBranchesRequest{\n\t\t\tPageSize:      pageSize,\n\t\t\tNextPageToken: s.hbd.NextPageToken,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn s.hbd, err\n\t\t}\n\t\tbatchCount := len(resp.Branches)\n\n\t\tskips := 0\n\t\terrorsOnSplitting := 0\n\t\t// send all tasks\n\t\tfor _, br := range resp.Branches {\n\t\t\tif time.Now().Add(-1 * getHistoryCleanupThreshold(s.maxWorkflowRetentionInDays())).Before(br.ForkTime) {\n\t\t\t\tbatchCount--\n\t\t\t\tskips++\n\t\t\t\ts.metrics.IncCounter(metrics.HistoryScavengerScope, metrics.HistoryScavengerSkipCount)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tdomainID, wid, rid, err := p.SplitHistoryGarbageCleanupInfo(br.Info)\n\t\t\tif err != nil {\n\t\t\t\tbatchCount--\n\t\t\t\terrorsOnSplitting++\n\t\t\t\ts.logger.Error(\"scavenger: unable to parse the history cleanup info\", tag.WorkflowTreeID(br.TreeID), tag.WorkflowBranchID(br.BranchID), tag.DetailInfo(br.Info))\n\t\t\t\ts.metrics.IncCounter(metrics.HistoryScavengerScope, metrics.HistoryScavengerErrorCount)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\ttaskCh <- taskDetail{\n\t\t\t\tdomainID:   domainID,\n\t\t\t\tworkflowID: wid,\n\t\t\t\trunID:      rid,\n\t\t\t\ttreeID:     br.TreeID,\n\t\t\t\tbranchID:   br.BranchID,\n\n\t\t\t\thbd: s.hbd,\n\t\t\t}\n\t\t}\n\n\t\tsuccCount := 0\n\t\terrCount := 0\n\t\tif batchCount > 0 {\n\t\t\t// wait for counters indicate this batch is done\n\t\tLoop:\n\t\t\tfor {\n\t\t\t\tselect {\n\t\t\t\tcase err := <-respCh:\n\t\t\t\t\tif err == nil {\n\t\t\t\t\t\ts.metrics.IncCounter(metrics.HistoryScavengerScope, metrics.HistoryScavengerSuccessCount)\n\t\t\t\t\t\tsuccCount++\n\t\t\t\t\t} else {\n\t\t\t\t\t\ts.metrics.IncCounter(metrics.HistoryScavengerScope, metrics.HistoryScavengerErrorCount)\n\t\t\t\t\t\terrCount++\n\t\t\t\t\t}\n\t\t\t\t\tif succCount+errCount == batchCount {\n\t\t\t\t\t\tbreak Loop\n\t\t\t\t\t}\n\t\t\t\tcase <-ctx.Done():\n\t\t\t\t\treturn s.hbd, ctx.Err()\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\ts.hbd.CurrentPage++\n\t\ts.hbd.NextPageToken = resp.NextPageToken\n\t\ts.hbd.SuccCount += succCount\n\t\ts.hbd.ErrorCount += errCount + errorsOnSplitting\n\t\ts.hbd.SkipCount += skips\n\t\tif !s.isInTest {\n\t\t\tactivity.RecordHeartbeat(ctx, s.hbd)\n\t\t}\n\n\t\tif len(s.hbd.NextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn s.hbd, nil\n}\n\nfunc (s *Scavenger) startTaskProcessor(\n\tctx context.Context,\n\ttaskCh chan taskDetail,\n\trespCh chan error,\n) {\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\treturn\n\t\tcase task := <-taskCh:\n\t\t\tif isDone(ctx) {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tif !s.isInTest {\n\t\t\t\tactivity.RecordHeartbeat(ctx, s.hbd)\n\t\t\t}\n\n\t\t\terr := s.limiter.Wait(ctx)\n\t\t\tif err != nil {\n\t\t\t\trespCh <- err\n\t\t\t\ts.logger.Error(\"encounter error when wait for rate limiter\",\n\t\t\t\t\tgetTaskLoggingTags(err, task)...)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\t// this checks if the mutableState still exists\n\t\t\t// if not then the history branch is garbage, we need to delete the history branch\n\t\t\t_, err = s.client.DescribeMutableState(ctx, &types.DescribeMutableStateRequest{\n\t\t\t\tDomainUUID: task.domainID,\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: task.workflowID,\n\t\t\t\t\tRunID:      task.runID,\n\t\t\t\t},\n\t\t\t})\n\n\t\t\tif err != nil {\n\t\t\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\t\t\t// deleting history branch\n\t\t\t\t\tvar branchToken []byte\n\t\t\t\t\tbranchToken, err = p.NewHistoryBranchTokenByBranchID(task.treeID, task.branchID)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\trespCh <- err\n\t\t\t\t\t\ts.logger.Error(\"encounter error when creating branch token\",\n\t\t\t\t\t\t\tgetTaskLoggingTags(err, task)...)\n\t\t\t\t\t\tcontinue\n\t\t\t\t\t}\n\t\t\t\t\tdomainName, err := s.domainCache.GetDomainName(task.domainID)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\trespCh <- err\n\t\t\t\t\t\ts.logger.Error(\"Unexpected: Encountered error while fetching domain name\",\n\t\t\t\t\t\t\tgetTaskLoggingTags(err, task)...)\n\t\t\t\t\t\tcontinue\n\t\t\t\t\t}\n\t\t\t\t\terr = s.db.DeleteHistoryBranch(ctx, &p.DeleteHistoryBranchRequest{\n\t\t\t\t\t\tBranchToken: branchToken,\n\t\t\t\t\t\t// This is a required argument but it is not needed for Cassandra.\n\t\t\t\t\t\t// Since this scanner is only for Cassandra,\n\t\t\t\t\t\t// we can fill any number here to let to code go through\n\t\t\t\t\t\tShardID:    common.IntPtr(1),\n\t\t\t\t\t\tDomainName: domainName,\n\t\t\t\t\t})\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\trespCh <- err\n\t\t\t\t\t\ts.logger.Error(\"encounter error when deleting garbage history branch\",\n\t\t\t\t\t\t\tgetTaskLoggingTags(err, task)...)\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// deleted garbage\n\t\t\t\t\t\ts.logger.Info(\"deleted history garbage\",\n\t\t\t\t\t\t\tgetTaskLoggingTags(nil, task)...)\n\n\t\t\t\t\t\trespCh <- nil\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\ts.logger.Error(\"encounter error when describing the mutable state\",\n\t\t\t\t\t\tgetTaskLoggingTags(err, task)...)\n\t\t\t\t\trespCh <- err\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\t// no garbage\n\t\t\t\trespCh <- nil\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc getTaskLoggingTags(err error, task taskDetail) []tag.Tag {\n\tif err != nil {\n\t\treturn []tag.Tag{\n\t\t\ttag.Error(err),\n\t\t\ttag.WorkflowDomainID(task.domainID),\n\t\t\ttag.WorkflowID(task.workflowID),\n\t\t\ttag.WorkflowRunID(task.runID),\n\t\t\ttag.WorkflowTreeID(task.treeID),\n\t\t\ttag.WorkflowBranchID(task.branchID),\n\t\t}\n\t}\n\treturn []tag.Tag{\n\t\ttag.WorkflowDomainID(task.domainID),\n\t\ttag.WorkflowID(task.workflowID),\n\t\ttag.WorkflowRunID(task.runID),\n\t\ttag.WorkflowTreeID(task.treeID),\n\t\ttag.WorkflowBranchID(task.branchID),\n\t}\n}\n\nfunc isDone(ctx context.Context) bool {\n\tselect {\n\tcase <-ctx.Done():\n\t\treturn true\n\tdefault:\n\t\treturn false\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/history/scavenger_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage history\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype (\n\tScavengerTestSuite struct {\n\t\tsuite.Suite\n\t\tlogger    log.Logger\n\t\tmetric    metrics.Client\n\t\tmockCache *cache.MockDomainCache\n\t}\n)\n\nfunc TestScavengerTestSuite(t *testing.T) {\n\tsuite.Run(t, new(ScavengerTestSuite))\n}\n\nfunc (s *ScavengerTestSuite) SetupTest() {\n\ts.logger = testlogger.New(s.T())\n\ts.metric = metrics.NewClient(tally.NoopScope, metrics.Worker, metrics.HistogramMigration{})\n\tcontroller := gomock.NewController(s.T())\n\ts.mockCache = cache.NewMockDomainCache(controller)\n}\n\nfunc (s *ScavengerTestSuite) createTestScavenger(rps int) (*mocks.HistoryV2Manager, *history.MockClient, *Scavenger) {\n\tdb := &mocks.HistoryV2Manager{}\n\tcontroller := gomock.NewController(s.T())\n\tworkflowClient := history.NewMockClient(controller)\n\tmaxWorkflowRetentionInDays := dynamicproperties.GetIntPropertyFn(dynamicproperties.MaxRetentionDays.DefaultInt())\n\tscvgr := NewScavenger(db, rps, workflowClient, ScavengerHeartbeatDetails{}, s.metric, s.logger, maxWorkflowRetentionInDays, s.mockCache)\n\tscvgr.isInTest = true\n\treturn db, workflowClient, scvgr\n}\n\nfunc (s *ScavengerTestSuite) TestAllSkipTasksTwoPages() {\n\tdb, _, scvgr := s.createTestScavenger(100)\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: pageSize,\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tNextPageToken: []byte(\"page1\"),\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID1\",\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\tForkTime: time.Now(),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID1\", \"workflowID1\", \"runID1\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID2\",\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\tForkTime: time.Now(),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID2\", \"workflowID2\", \"runID2\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: []byte(\"page1\"),\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID3\",\n\t\t\t\tBranchID: \"branchID3\",\n\t\t\t\tForkTime: time.Now(),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID3\", \"workflowID3\", \"runID3\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID4\",\n\t\t\t\tBranchID: \"branchID4\",\n\t\t\t\tForkTime: time.Now(),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID4\", \"workflowID4\", \"runID4\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\thbd, err := scvgr.Run(ctx)\n\ts.Nil(err)\n\ts.Equal(4, hbd.SkipCount)\n\ts.Equal(0, hbd.SuccCount)\n\ts.Equal(0, hbd.ErrorCount)\n\ts.Equal(2, hbd.CurrentPage)\n\ts.Equal(0, len(hbd.NextPageToken))\n}\n\nfunc (s *ScavengerTestSuite) TestAllErrorSplittingTasksTwoPages() {\n\tdb, _, scvgr := s.createTestScavenger(100)\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: pageSize,\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tNextPageToken: []byte(\"page1\"),\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID1\",\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     \"error-info\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID2\",\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     \"error-info\",\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: []byte(\"page1\"),\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID3\",\n\t\t\t\tBranchID: \"branchID3\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     \"error-info\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID4\",\n\t\t\t\tBranchID: \"branchID4\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     \"error-info\",\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\thbd, err := scvgr.Run(ctx)\n\ts.Nil(err)\n\ts.Equal(0, hbd.SkipCount)\n\ts.Equal(0, hbd.SuccCount)\n\ts.Equal(4, hbd.ErrorCount)\n\ts.Equal(2, hbd.CurrentPage)\n\ts.Equal(0, len(hbd.NextPageToken))\n}\n\nfunc (s *ScavengerTestSuite) TestNoGarbageTwoPages() {\n\tdb, client, scvgr := s.createTestScavenger(100)\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: pageSize,\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tNextPageToken: []byte(\"page1\"),\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID1\",\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID1\", \"workflowID1\", \"runID1\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID2\",\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID2\", \"workflowID2\", \"runID2\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: []byte(\"page1\"),\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID3\",\n\t\t\t\tBranchID: \"branchID3\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID3\", \"workflowID3\", \"runID3\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID4\",\n\t\t\t\tBranchID: \"branchID4\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID4\", \"workflowID4\", \"runID4\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID1\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID1\",\n\t\t\tRunID:      \"runID1\",\n\t\t},\n\t}).Return(nil, nil)\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID2\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID2\",\n\t\t\tRunID:      \"runID2\",\n\t\t},\n\t}).Return(nil, nil)\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID3\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID3\",\n\t\t\tRunID:      \"runID3\",\n\t\t},\n\t}).Return(nil, nil)\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID4\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID4\",\n\t\t\tRunID:      \"runID4\",\n\t\t},\n\t}).Return(nil, nil)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\thbd, err := scvgr.Run(ctx)\n\ts.Nil(err)\n\ts.Equal(0, hbd.SkipCount)\n\ts.Equal(4, hbd.SuccCount)\n\ts.Equal(0, hbd.ErrorCount)\n\ts.Equal(2, hbd.CurrentPage)\n\ts.Equal(0, len(hbd.NextPageToken))\n}\n\nfunc (s *ScavengerTestSuite) TestDeletingBranchesTwoPages() {\n\tdb, client, scvgr := s.createTestScavenger(100)\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: pageSize,\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tNextPageToken: []byte(\"page1\"),\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID1\",\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID1\", \"workflowID1\", \"runID1\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID2\",\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID2\", \"workflowID2\", \"runID2\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: []byte(\"page1\"),\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID3\",\n\t\t\t\tBranchID: \"branchID3\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID3\", \"workflowID3\", \"runID3\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\tTreeID:   \"treeID4\",\n\t\t\t\tBranchID: \"branchID4\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID4\", \"workflowID4\", \"runID4\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID1\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID1\",\n\t\t\tRunID:      \"runID1\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID2\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID2\",\n\t\t\tRunID:      \"runID2\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID3\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID3\",\n\t\t\tRunID:      \"runID3\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID4\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID4\",\n\t\t\tRunID:      \"runID4\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\tdomainName := \"test-domainName\"\n\ts.mockCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\tbranchToken1, err := p.NewHistoryBranchTokenByBranchID(\"treeID1\", \"branchID1\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken1,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(nil).Once()\n\tbranchToken2, err := p.NewHistoryBranchTokenByBranchID(\"treeID2\", \"branchID2\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken2,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(nil).Once()\n\tbranchToken3, err := p.NewHistoryBranchTokenByBranchID(\"treeID3\", \"branchID3\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken3,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(nil).Once()\n\tbranchToken4, err := p.NewHistoryBranchTokenByBranchID(\"treeID4\", \"branchID4\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken4,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(nil).Once()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\thbd, err := scvgr.Run(ctx)\n\ts.Nil(err)\n\ts.Equal(0, hbd.SkipCount)\n\ts.Equal(4, hbd.SuccCount)\n\ts.Equal(0, hbd.ErrorCount)\n\ts.Equal(2, hbd.CurrentPage)\n\ts.Equal(0, len(hbd.NextPageToken))\n}\n\nfunc (s *ScavengerTestSuite) TestMixesTwoPages() {\n\tdb, client, scvgr := s.createTestScavenger(100)\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize: pageSize,\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tNextPageToken: []byte(\"page1\"),\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\t// skip\n\t\t\t\tTreeID:   \"treeID1\",\n\t\t\t\tBranchID: \"branchID1\",\n\t\t\t\tForkTime: time.Now(),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID1\", \"workflowID1\", \"runID1\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\t// split error\n\t\t\t\tTreeID:   \"treeID2\",\n\t\t\t\tBranchID: \"branchID2\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     \"error-info\",\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\tdb.On(\"GetAllHistoryTreeBranches\", mock.Anything, &p.GetAllHistoryTreeBranchesRequest{\n\t\tPageSize:      pageSize,\n\t\tNextPageToken: []byte(\"page1\"),\n\t}).Return(&p.GetAllHistoryTreeBranchesResponse{\n\t\tBranches: []p.HistoryBranchDetail{\n\t\t\t{\n\t\t\t\t// delete succ\n\t\t\t\tTreeID:   \"treeID3\",\n\t\t\t\tBranchID: \"branchID3\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID3\", \"workflowID3\", \"runID3\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\t// delete fail\n\t\t\t\tTreeID:   \"treeID4\",\n\t\t\t\tBranchID: \"branchID4\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID4\", \"workflowID4\", \"runID4\"),\n\t\t\t},\n\t\t\t{\n\t\t\t\t// not delete\n\t\t\t\tTreeID:   \"treeID5\",\n\t\t\t\tBranchID: \"branchID5\",\n\t\t\t\tForkTime: time.Now().Add(-getHistoryCleanupThreshold(dynamicproperties.MaxRetentionDays.DefaultInt()) * 2),\n\t\t\t\tInfo:     p.BuildHistoryGarbageCleanupInfo(\"domainID5\", \"workflowID5\", \"runID5\"),\n\t\t\t},\n\t\t},\n\t}, nil).Once()\n\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID3\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID3\",\n\t\t\tRunID:      \"runID3\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID4\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID4\",\n\t\t\tRunID:      \"runID4\",\n\t\t},\n\t}).Return(nil, &types.EntityNotExistsError{})\n\tclient.EXPECT().DescribeMutableState(gomock.Any(), &types.DescribeMutableStateRequest{\n\t\tDomainUUID: \"domainID5\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"workflowID5\",\n\t\t\tRunID:      \"runID5\",\n\t\t},\n\t}).Return(nil, nil)\n\tdomainName := \"test-domainName\"\n\ts.mockCache.EXPECT().GetDomainName(gomock.Any()).Return(domainName, nil).AnyTimes()\n\tbranchToken3, err := p.NewHistoryBranchTokenByBranchID(\"treeID3\", \"branchID3\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken3,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(nil).Once()\n\n\tbranchToken4, err := p.NewHistoryBranchTokenByBranchID(\"treeID4\", \"branchID4\")\n\ts.Nil(err)\n\tdb.On(\"DeleteHistoryBranch\", mock.Anything, &p.DeleteHistoryBranchRequest{\n\t\tBranchToken: branchToken4,\n\t\tShardID:     common.IntPtr(1),\n\t\tDomainName:  domainName,\n\t}).Return(fmt.Errorf(\"failed to delete history\")).Once()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\thbd, err := scvgr.Run(ctx)\n\ts.Nil(err)\n\ts.Equal(1, hbd.SkipCount)\n\ts.Equal(2, hbd.SuccCount)\n\ts.Equal(2, hbd.ErrorCount)\n\ts.Equal(2, hbd.CurrentPage)\n\ts.Equal(0, len(hbd.NextPageToken))\n}\n"
  },
  {
    "path": "service/worker/scanner/scanner.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n\t\"github.com/uber/cadence/service/worker/scanner/tasklist\"\n\t\"github.com/uber/cadence/service/worker/workercommon\"\n)\n\nconst (\n\t// scannerStartUpDelay is to let services warm up\n\tscannerStartUpDelay = time.Second * 4\n)\n\ntype (\n\tcontextKey string\n\n\t// Config defines the configuration for scanner\n\tConfig struct {\n\t\t// ScannerPersistenceMaxQPS the max rate of calls to persistence\n\t\t// Right now is being used by historyScanner to determine the rate of persistence API calls\n\t\tScannerPersistenceMaxQPS dynamicproperties.IntPropertyFn\n\t\t// TaskListScannerEnabled indicates if taskList scanner should be started as part of scanner\n\t\tTaskListScannerEnabled dynamicproperties.BoolPropertyFn\n\t\t// TaskListScannerOptions contains options for TaskListScanner\n\t\tTaskListScannerOptions tasklist.Options\n\t\t// Persistence contains the persistence configuration\n\t\tPersistence *config.Persistence\n\t\t// ClusterMetadata contains the metadata for this cluster\n\t\tClusterMetadata cluster.Metadata\n\t\t// HistoryScannerEnabled indicates if history scanner should be started as part of scanner\n\t\tHistoryScannerEnabled dynamicproperties.BoolPropertyFn\n\t\t// ShardScanners is a list of shard scanner configs\n\t\tShardScanners              []*shardscanner.ScannerConfig\n\t\tMaxWorkflowRetentionInDays dynamicproperties.IntPropertyFn\n\t}\n\n\t// BootstrapParams contains the set of params needed to bootstrap\n\t// the scanner sub-system\n\tBootstrapParams struct {\n\t\t// Config contains the configuration for scanner\n\t\tConfig Config\n\t\t// TallyScope is an instance of tally metrics scope\n\t\tTallyScope tally.Scope\n\t}\n\n\t// scannerContext is the context object that get's\n\t// passed around within the scanner workflows / activities\n\tscannerContext struct {\n\t\tresource resource.Resource\n\t\tcfg      Config\n\t}\n\n\t// Scanner is the background sub-system that does full scans\n\t// of database tables to cleanup resources, monitor anomalies\n\t// and emit stats for analytics\n\tScanner struct {\n\t\tcontext                  scannerContext\n\t\ttallyScope               tally.Scope\n\t\tzapLogger                *zap.Logger\n\t\tstartWorkflowWithRetryFn func(string, time.Duration, resource.Resource, func(client client.Client) error) error\n\t\tnewWorkerFn              func(workflowserviceclient.Interface, string, string, worker.Options) worker.Worker\n\t}\n)\n\n// New returns a new instance of scanner daemon\n// Scanner is the background sub-system that does full\n// scans of database tables in an attempt to cleanup\n// resources, monitor system anamolies and emit stats\n// for analysis and alerting\nfunc New(\n\tresource resource.Resource,\n\tparams *BootstrapParams,\n) *Scanner {\n\n\tzapLogger, err := zap.NewProduction()\n\tif err != nil {\n\t\tresource.GetLogger().Fatal(\"failed to initialize zap logger\", tag.Error(err))\n\t}\n\tzapLogger.Info(\"Initializing new scanner\")\n\treturn &Scanner{\n\t\tcontext: scannerContext{\n\t\t\tresource: resource,\n\t\t\tcfg:      params.Config,\n\t\t},\n\t\ttallyScope:               params.TallyScope,\n\t\tzapLogger:                zapLogger.Named(\"scanner\"),\n\t\tstartWorkflowWithRetryFn: workercommon.StartWorkflowWithRetry,\n\t\tnewWorkerFn:              worker.New,\n\t}\n\n}\n\n// Start starts the scanner\nfunc (s *Scanner) Start() error {\n\tctx := context.Background()\n\tvar workerTaskListNames []string\n\tvar wtl []string\n\n\tfor _, sc := range s.context.cfg.ShardScanners {\n\t\tctx, wtl = s.startShardScanner(ctx, sc)\n\t\tworkerTaskListNames = append(workerTaskListNames, wtl...)\n\t}\n\n\tif s.context.cfg.Persistence.DefaultStoreType() == config.StoreTypeSQL {\n\t\tif s.context.cfg.TaskListScannerEnabled() {\n\t\t\tctx = s.startScanner(\n\t\t\t\tctx,\n\t\t\t\ttlScannerWFStartOptions,\n\t\t\t\ttlScannerWFTypeName)\n\t\t\tworkerTaskListNames = append(workerTaskListNames, tlScannerTaskListName)\n\t\t}\n\t}\n\tif s.context.cfg.HistoryScannerEnabled() {\n\t\tctx = s.startScanner(\n\t\t\tctx,\n\t\t\thistoryScannerWFStartOptions,\n\t\t\thistoryScannerWFTypeName)\n\t\tworkerTaskListNames = append(workerTaskListNames, historyScannerTaskListName)\n\t}\n\n\tworkerOpts := worker.Options{\n\t\tLogger:                                 s.zapLogger,\n\t\tMetricsScope:                           s.tallyScope,\n\t\tMaxConcurrentActivityExecutionSize:     maxConcurrentActivityExecutionSize,\n\t\tMaxConcurrentDecisionTaskExecutionSize: maxConcurrentDecisionTaskExecutionSize,\n\t\tBackgroundActivityContext:              ctx,\n\t}\n\n\tfor _, tl := range workerTaskListNames {\n\t\ts.zapLogger.Info(\"Starting worker for task list\", zap.String(\"TaskList\", tl))\n\t\tif err := s.newWorkerFn(s.context.resource.GetSDKClient(), constants.SystemLocalDomainName, tl, workerOpts).Start(); err != nil {\n\t\t\ts.zapLogger.Error(\"Failed to start worker\", zap.String(\"TaskList\", tl), zap.Error(err))\n\t\t\treturn err\n\t\t}\n\t}\n\ts.zapLogger.Info(\"Scanner started successfully\", zap.Strings(\"workerTaskListNames\", workerTaskListNames))\n\treturn nil\n}\n\nfunc (s *Scanner) startScanner(ctx context.Context, options client.StartWorkflowOptions, workflowName string) context.Context {\n\tgo s.startWorkflowWithRetryFn(workflowName, scannerStartUpDelay, s.context.resource, func(client client.Client) error {\n\t\treturn s.startWorkflow(client, options, workflowName, nil)\n\t})\n\treturn NewScannerContext(ctx, workflowName, s.context)\n}\n\nfunc (s *Scanner) startShardScanner(\n\tctx context.Context,\n\tconfig *shardscanner.ScannerConfig,\n) (context.Context, []string) {\n\tvar workerTaskListNames []string\n\tif config.DynamicParams.ScannerEnabled() {\n\t\tctx = shardscanner.NewScannerContext(\n\t\t\tctx,\n\t\t\tconfig.ScannerWFTypeName,\n\t\t\tshardscanner.NewShardScannerContext(s.context.resource, config),\n\t\t)\n\t\tgo s.startWorkflowWithRetryFn(\n\t\t\tconfig.ScannerWFTypeName,\n\t\t\tscannerStartUpDelay,\n\t\t\ts.context.resource,\n\t\t\tfunc(client client.Client) error {\n\t\t\t\treturn s.startWorkflow(client, config.StartWorkflowOptions, config.ScannerWFTypeName, shardscanner.ScannerWorkflowParams{\n\t\t\t\t\tShards: shardscanner.Shards{\n\t\t\t\t\t\tRange: &shardscanner.ShardRange{\n\t\t\t\t\t\t\tMin: 0,\n\t\t\t\t\t\t\tMax: s.context.cfg.Persistence.NumHistoryShards,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t})\n\n\t\tworkerTaskListNames = append(workerTaskListNames, config.StartWorkflowOptions.TaskList)\n\t}\n\n\tif config.DynamicParams.FixerEnabled() {\n\t\tctx = shardscanner.NewFixerContext(\n\t\t\tctx,\n\t\t\tconfig.FixerWFTypeName,\n\t\t\tshardscanner.NewShardFixerContext(s.context.resource, config),\n\t\t)\n\t\tgo s.startWorkflowWithRetryFn(\n\t\t\tconfig.FixerWFTypeName,\n\t\t\tscannerStartUpDelay,\n\t\t\ts.context.resource,\n\t\t\tfunc(client client.Client) error {\n\t\t\t\treturn s.startWorkflow(client, config.StartFixerOptions, config.FixerWFTypeName,\n\t\t\t\t\tshardscanner.FixerWorkflowParams{\n\t\t\t\t\t\tScannerWorkflowWorkflowID: config.StartWorkflowOptions.ID,\n\t\t\t\t\t})\n\t\t\t})\n\n\t\tworkerTaskListNames = append(workerTaskListNames, config.StartFixerOptions.TaskList)\n\t}\n\n\treturn ctx, workerTaskListNames\n}\n\nfunc (s *Scanner) startWorkflow(\n\tclient client.Client,\n\toptions client.StartWorkflowOptions,\n\tworkflowType string,\n\tworkflowArg interface{},\n) error {\n\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)\n\tvar err error\n\tif workflowArg != nil {\n\t\t_, err = client.StartWorkflow(ctx, options, workflowType, workflowArg)\n\t} else {\n\t\t_, err = client.StartWorkflow(ctx, options, workflowType)\n\t}\n\n\tcancel()\n\n\tif cadence.IsWorkflowExecutionAlreadyStartedError(err) {\n\t\ts.zapLogger.Error(\"Workflow had already started\", zap.String(\"workflowType\", workflowType), zap.Error(err))\n\t\treturn nil\n\t}\n\tif err != nil {\n\t\ts.zapLogger.Error(\"Failed to start workflow\", zap.String(\"workflowType\", workflowType), zap.Error(err))\n\t} else {\n\t\ts.zapLogger.Info(\"Workflow started\", zap.String(\"workflowType\", workflowType))\n\t}\n\treturn err\n}\n\n// NewScannerContext provides context to be used as background activity context\n// it uses typed, private key to reduce access scope\nfunc NewScannerContext(ctx context.Context, workflowName string, scannerContext scannerContext) context.Context {\n\treturn context.WithValue(ctx, contextKey(workflowName), scannerContext)\n}\n\n// getScannerContext extracts scanner context from activity context\n// it uses typed, private key to reduce access scope\nfunc getScannerContext(ctx context.Context) (scannerContext, error) {\n\tinfo := activity.GetInfo(ctx)\n\tif info.WorkflowType == nil {\n\t\treturn scannerContext{}, fmt.Errorf(\"workflowType is nil\")\n\t}\n\tval, ok := ctx.Value(contextKey(info.WorkflowType.Name)).(scannerContext)\n\tif !ok {\n\t\treturn scannerContext{}, fmt.Errorf(\"context type is not %T for a key %q\", val, info.WorkflowType.Name)\n\t}\n\treturn val, nil\n}\n"
  },
  {
    "path": "service/worker/scanner/scanner_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/mocks\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tcommonResource \"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/history/resource\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n\tworker2 \"github.com/uber/cadence/service/worker/worker\"\n)\n\ntype scannerTestSuite struct {\n\tsuite.Suite\n\tmockCtrl     *gomock.Controller\n\tmockResource *resource.Test\n\tmockScope    tally.Scope\n\tscanner      *Scanner\n\tmockWorker   *worker2.MockWorker\n\tmockClient   *mocks.Client\n}\n\nfunc TestScannerSuite(t *testing.T) {\n\tsuite.Run(t, new(scannerTestSuite))\n}\n\nfunc (s *scannerTestSuite) SetupTest() {\n\ts.mockCtrl = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.mockCtrl, metrics.Worker)\n\ts.mockScope = tally.NoopScope\n\ts.mockWorker = worker2.NewMockWorker(s.mockCtrl)\n\ts.mockClient = new(mocks.Client)\n\n\ts.scanner = &Scanner{\n\t\tcontext: scannerContext{\n\t\t\tresource: s.mockResource,\n\t\t\tcfg:      Config{},\n\t\t},\n\t\ttallyScope: s.mockScope,\n\t\tzapLogger:  zap.NewNop(),\n\t\tstartWorkflowWithRetryFn: func(_ string, _ time.Duration, _ commonResource.Resource, _ func(client client.Client) error) error {\n\t\t\treturn nil\n\t\t},\n\t\tnewWorkerFn: func(_ workflowserviceclient.Interface, _ string, _ string, _ worker.Options) worker.Worker {\n\t\t\treturn s.mockWorker\n\t\t},\n\t}\n}\n\nfunc (s *scannerTestSuite) TearDownTest() {\n\ts.mockCtrl.Finish()\n}\n\nfunc (s *scannerTestSuite) TestNew() {\n\tparams := &BootstrapParams{\n\t\tConfig:     Config{},\n\t\tTallyScope: s.mockScope,\n\t}\n\n\ts.NotNil(New(s.mockResource, params))\n}\n\nfunc (s *scannerTestSuite) TestStart() {\n\ttestCases := []struct {\n\t\tname       string\n\t\tcfg        Config\n\t\tsetupMocks func()\n\t\terr        error\n\t}{\n\t\t{\n\t\t\tname: \"with shard scanners\",\n\t\t\tcfg: Config{\n\t\t\t\tShardScanners: []*shardscanner.ScannerConfig{\n\t\t\t\t\t{\n\t\t\t\t\t\tScannerWFTypeName: \"testScannerWF1\",\n\t\t\t\t\t\tDynamicParams: shardscanner.DynamicParams{\n\t\t\t\t\t\t\tScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\t\t\t\treturn true\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFixerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\t\t\t\treturn true\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartWorkflowOptions: client.StartWorkflowOptions{\n\t\t\t\t\t\t\tTaskList: \"shard-scanner-task-list-1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tScannerHooks: func() *shardscanner.ScannerHooks {\n\t\t\t\t\t\t\treturn &shardscanner.ScannerHooks{\n\t\t\t\t\t\t\t\tGetScannerConfig: func(scanner shardscanner.ScannerContext) shardscanner.CustomScannerConfig {\n\t\t\t\t\t\t\t\t\treturn map[string]string{}\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t},\n\t\t\t\t\t\tFixerHooks: func() *shardscanner.FixerHooks {\n\t\t\t\t\t\t\treturn &shardscanner.FixerHooks{\n\t\t\t\t\t\t\t\tGetFixerConfig: func(fixer shardscanner.FixerContext) shardscanner.CustomScannerConfig {\n\t\t\t\t\t\t\t\t\treturn map[string]string{}\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tPersistence: &config.Persistence{\n\t\t\t\t\tDefaultStore: \"nosql\",\n\t\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\t\"nosql\": {\n\t\t\t\t\t\t\tNoSQL: &config.NoSQL{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tHistoryScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn false\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func() {\n\t\t\t\t// this is mocking the worker being instantiated and started\n\t\t\t\t// in here the mockWorker is being started twice, but in reality\n\t\t\t\t// each worker gets started only once\n\t\t\t\ts.mockWorker.EXPECT().Start().Return(nil).Times(1)\n\t\t\t\ts.mockWorker.EXPECT().Start().Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"with default store type SQL with TaskListScanner enabled\",\n\t\t\tcfg: Config{\n\t\t\t\tPersistence: &config.Persistence{\n\t\t\t\t\tDefaultStore: \"sql\",\n\t\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\t\"sql\": {\n\t\t\t\t\t\t\tSQL: &config.SQL{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTaskListScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn true\n\t\t\t\t},\n\t\t\t\tHistoryScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn false\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockWorker.EXPECT().Start().Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"with HistoryScanner enabled\",\n\t\t\tcfg: Config{\n\t\t\t\tPersistence: &config.Persistence{\n\t\t\t\t\tDefaultStore: \"nosql\",\n\t\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\t\"nosql\": {\n\t\t\t\t\t\t\tNoSQL: &config.NoSQL{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTaskListScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn false\n\t\t\t\t},\n\t\t\t\tHistoryScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn true\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockWorker.EXPECT().Start().Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"failed to start worker\",\n\t\t\tcfg: Config{\n\t\t\t\tPersistence: &config.Persistence{\n\t\t\t\t\tDefaultStore: \"nosql\",\n\t\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\t\"nosql\": {\n\t\t\t\t\t\t\tNoSQL: &config.NoSQL{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tTaskListScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn false\n\t\t\t\t},\n\t\t\t\tHistoryScannerEnabled: func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\t\t\treturn true\n\t\t\t\t},\n\t\t\t},\n\t\t\tsetupMocks: func() {\n\t\t\t\ts.mockWorker.EXPECT().Start().Return(errors.New(\"some new error\")).Times(1)\n\t\t\t},\n\t\t\terr: errors.New(\"some new error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\tdefer goleak.VerifyNone(s.T())\n\n\t\t\ts.scanner.context.cfg = tc.cfg\n\t\t\ttc.setupMocks()\n\n\t\t\terr := s.scanner.Start()\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *scannerTestSuite) Test__startWorkflow() {\n\ttestCases := []struct {\n\t\tname         string\n\t\toptions      client.StartWorkflowOptions\n\t\tworkflowType string\n\t\targs         interface{}\n\t\tsetupMocks   func(client.StartWorkflowOptions, string, interface{})\n\t\terr          error\n\t}{\n\t\t{\n\t\t\tname: \"success with args\",\n\t\t\toptions: client.StartWorkflowOptions{\n\t\t\t\tID: \"testIDWithArgs\",\n\t\t\t},\n\t\t\tworkflowType: \"workflowTypeWithArgs\",\n\t\t\targs:         \"testArgs\",\n\t\t\tsetupMocks: func(options client.StartWorkflowOptions, workflowType string, args interface{}) {\n\t\t\t\ts.mockClient.On(\"StartWorkflow\", mock.Anything, options, workflowType, args).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"success with no args\",\n\t\t\toptions: client.StartWorkflowOptions{\n\t\t\t\tID: \"testIDWithNoArgs\",\n\t\t\t},\n\t\t\tworkflowType: \"workflowTypeWithArgs\",\n\t\t\tsetupMocks: func(options client.StartWorkflowOptions, workflowType string, args interface{}) {\n\t\t\t\ts.mockClient.On(\"StartWorkflow\", mock.Anything, options, workflowType).Return(nil, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionAlreadyStarted error\",\n\t\t\toptions: client.StartWorkflowOptions{\n\t\t\t\tID: \"testIDWorkflowExecutionAlreadyStarted\",\n\t\t\t},\n\t\t\tworkflowType: \"workflowTypeWorkflowExecutionAlreadyStarted\",\n\t\t\tsetupMocks: func(options client.StartWorkflowOptions, workflowType string, args interface{}) {\n\t\t\t\ts.mockClient.On(\"StartWorkflow\", mock.Anything, options, workflowType).Return(nil, &shared.WorkflowExecutionAlreadyStartedError{})\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"another error\",\n\t\t\toptions: client.StartWorkflowOptions{\n\t\t\t\tID: \"testIDAnotherError\",\n\t\t\t},\n\t\t\tworkflowType: \"workflowTypeAnotherError\",\n\t\t\tsetupMocks: func(options client.StartWorkflowOptions, workflowType string, args interface{}) {\n\t\t\t\ts.mockClient.On(\"StartWorkflow\", mock.Anything, options, workflowType).Return(nil, errors.New(\"some random error\"))\n\t\t\t},\n\t\t\terr: errors.New(\"some random error\"),\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.T().Run(tc.name, func(t *testing.T) {\n\t\t\ttc.setupMocks(tc.options, tc.workflowType, tc.args)\n\n\t\t\terr := s.scanner.startWorkflow(s.mockClient, tc.options, tc.workflowType, tc.args)\n\n\t\t\tif tc.err != nil {\n\t\t\t\ts.Error(err)\n\t\t\t\ts.Equal(tc.err, err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/activities.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\n\tc \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\nconst (\n\t// ActivityScannerEmitMetrics is the activity name for scannerEmitMetricsActivity\n\tActivityScannerEmitMetrics = \"cadence-sys-shardscanner-emit-metrics-activity\"\n\t// ActivityScannerConfig is the activity name scannerConfigActivity\n\tActivityScannerConfig = \"cadence-sys-shardscanner-config-activity\"\n\t// ActivityFixerConfig is the activity name fixerConfigActivity\n\tActivityFixerConfig = \"cadence-sys-shardscanner-fixer-config-activity\"\n\t// ActivityScanShard is the activity name for scanShardActivity\n\tActivityScanShard = \"cadence-sys-shardscanner-scanshard-activity\"\n\t// ActivityFixerCorruptedKeys is the activity name for fixerCorruptedKeysActivity\n\tActivityFixerCorruptedKeys = \"cadence-sys-shardscanner-corruptedkeys-activity\"\n\t// ActivityFixShard is the activity name for fixShardActivity\n\tActivityFixShard = \"cadence-sys-shardscanner-fixshard-activity\"\n\t// ShardCorruptKeysQuery is the query name for the query used to get all completed shards with at least one corruption\n\tShardCorruptKeysQuery = \"shard_corrupt_keys\"\n)\n\n// scannerConfigActivity will read dynamic config, apply overwrites and return a resolved config.\nfunc scannerConfigActivity(\n\tactivityCtx context.Context,\n\tparams ScannerConfigActivityParams,\n) (ResolvedScannerWorkflowConfig, error) {\n\tctx, err := GetScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn ResolvedScannerWorkflowConfig{}, err\n\t}\n\tdc := ctx.Config.DynamicParams\n\n\tresult := ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\tEnabled:                 dc.ScannerEnabled(),\n\t\t\tConcurrency:             dc.Concurrency(),\n\t\t\tPageSize:                dc.PageSize(),\n\t\t\tBlobstoreFlushThreshold: dc.BlobstoreFlushThreshold(),\n\t\t\tActivityBatchSize:       dc.ActivityBatchSize(),\n\t\t},\n\t}\n\n\tif ctx.Hooks != nil && ctx.Hooks.GetScannerConfig != nil {\n\t\tresult.CustomScannerConfig = ctx.Hooks.GetScannerConfig(ctx)\n\t}\n\n\toverwrites := params.Overwrites.GenericScannerConfig\n\tif overwrites.Enabled != nil {\n\t\tresult.GenericScannerConfig.Enabled = *overwrites.Enabled\n\t}\n\tif overwrites.Concurrency != nil {\n\t\tresult.GenericScannerConfig.Concurrency = *overwrites.Concurrency\n\t}\n\tif overwrites.PageSize != nil {\n\t\tresult.GenericScannerConfig.PageSize = *overwrites.PageSize\n\t}\n\tif overwrites.BlobstoreFlushThreshold != nil {\n\t\tresult.GenericScannerConfig.BlobstoreFlushThreshold = *overwrites.BlobstoreFlushThreshold\n\t}\n\n\tif overwrites.ActivityBatchSize != nil {\n\t\tresult.GenericScannerConfig.ActivityBatchSize = *overwrites.ActivityBatchSize\n\t}\n\n\tif params.Overwrites.CustomScannerConfig != nil {\n\t\tresult.CustomScannerConfig = *params.Overwrites.CustomScannerConfig\n\t}\n\n\treturn result, nil\n}\n\n// scanShardActivity will scan a collection of shards for invariant violations.\nfunc scanShardActivity(\n\tactivityCtx context.Context,\n\tparams ScanShardActivityParams,\n) ([]ScanReport, error) {\n\theartbeatDetails := ScanShardHeartbeatDetails{\n\t\tLastShardIndexHandled: -1,\n\t\tReports:               nil,\n\t}\n\tctx, err := GetScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif activity.HasHeartbeatDetails(activityCtx) {\n\t\tif err := activity.GetHeartbeatDetails(activityCtx, &heartbeatDetails); err != nil {\n\t\t\tctx.Logger.Error(\"getting heartbeat details\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tfor i := heartbeatDetails.LastShardIndexHandled + 1; i < len(params.Shards); i++ {\n\t\tcurrentShardID := params.Shards[i]\n\t\tshardReport, err := scanShard(activityCtx, params, currentShardID, heartbeatDetails)\n\t\tif err != nil {\n\t\t\tctx.Logger.Error(\"scanning shard\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t\theartbeatDetails = ScanShardHeartbeatDetails{\n\t\t\tLastShardIndexHandled: i,\n\t\t\tReports:               append(heartbeatDetails.Reports, *shardReport),\n\t\t}\n\t}\n\treturn heartbeatDetails.Reports, nil\n}\n\nfunc scanShard(\n\tactivityCtx context.Context,\n\tparams ScanShardActivityParams,\n\tshardID int,\n\theartbeatDetails ScanShardHeartbeatDetails,\n) (*ScanReport, error) {\n\tctx, err := GetScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tinfo := activity.GetInfo(activityCtx)\n\n\tscope := ctx.Scope.Tagged(\n\t\tmetrics.ActivityTypeTag(ActivityScanShard),\n\t\tmetrics.WorkflowTypeTag(info.WorkflowType.Name),\n\t\tmetrics.DomainTag(constants.SystemLocalDomainName),\n\t)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\n\tif ctx.Hooks == nil {\n\t\treturn nil, cadence.NewCustomError(ErrMissingHooks)\n\t}\n\n\tresources := ctx.Resource\n\texecManager, err := resources.GetExecutionManager(shardID)\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\treturn nil, err\n\t}\n\n\tpr := persistence.NewPersistenceRetryer(execManager, resources.GetHistoryManager(), c.CreatePersistenceRetryPolicy())\n\n\tscanner := NewScanner(\n\t\tshardID,\n\t\tctx.Hooks.Iterator(activityCtx, pr, params),\n\t\tresources.GetBlobstoreClient(),\n\t\tparams.BlobstoreFlushThreshold,\n\t\tctx.Hooks.Manager(activityCtx, pr, params, resources.GetDomainCache()),\n\t\tfunc() { activity.RecordHeartbeat(activityCtx, heartbeatDetails) },\n\t\tscope,\n\t\tresources.GetDomainCache(),\n\t)\n\treport := scanner.Scan(activityCtx)\n\tif report.Result.ControlFlowFailure != nil {\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t}\n\treturn &report, nil\n}\n\n// fixerCorruptedKeysActivity will fetch the keys of blobs from shards with corruptions from a completed scan workflow.\n// If scan workflow is not closed or if query fails activity will return an error.\n// Accepts as input the shard to start query at and returns a next page token, therefore this activity can\n// be used to do pagination.\nfunc fixerCorruptedKeysActivity(\n\tactivityCtx context.Context,\n\tparams FixerCorruptedKeysActivityParams,\n) (*FixerCorruptedKeysActivityResult, error) {\n\tctx, err := GetFixerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tclient := ctx.Resource.GetSDKClient()\n\tif params.ScannerWorkflowRunID == \"\" {\n\t\tlistResp, err := client.ListClosedWorkflowExecutions(activityCtx, &shared.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          c.StringPtr(constants.SystemLocalDomainName),\n\t\t\tMaximumPageSize: c.Int32Ptr(10),\n\t\t\tNextPageToken:   nil,\n\t\t\tStartTimeFilter: &shared.StartTimeFilter{\n\t\t\t\tEarliestTime: c.Int64Ptr(0),\n\t\t\t\tLatestTime:   c.Int64Ptr(time.Now().UnixNano()),\n\t\t\t},\n\t\t\tExecutionFilter: &shared.WorkflowExecutionFilter{\n\t\t\t\tWorkflowId: c.StringPtr(params.ScannerWorkflowWorkflowID),\n\t\t\t},\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tif len(listResp.Executions) > 10 {\n\t\t\treturn nil, errors.New(\"got unexpected number of executions back from list\")\n\t\t}\n\t\t// ListClosedWorkflowExecutions API doesn't support querying by workflow ID and status filter at the same time,\n\t\t// and we want to avoid using a scan result with Terminated state.\n\t\tfor _, executionInfo := range listResp.Executions {\n\t\t\tif *executionInfo.CloseStatus == shared.WorkflowExecutionCloseStatusContinuedAsNew {\n\t\t\t\tparams.ScannerWorkflowRunID = *executionInfo.Execution.RunId\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tif len(params.ScannerWorkflowRunID) == 0 {\n\t\t\treturn nil, errors.New(\"failed to find a recent scanner workflow execution with ContinuedAsNew status\")\n\t\t}\n\t}\n\n\tdescResp, err := client.DescribeWorkflowExecution(activityCtx, &shared.DescribeWorkflowExecutionRequest{\n\t\tDomain: c.StringPtr(constants.SystemLocalDomainName),\n\t\tExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: c.StringPtr(params.ScannerWorkflowWorkflowID),\n\t\t\tRunId:      c.StringPtr(params.ScannerWorkflowRunID),\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif descResp.WorkflowExecutionInfo.CloseStatus == nil {\n\t\treturn nil, cadence.NewCustomError(ErrScanWorkflowNotClosed)\n\t}\n\tqueryArgs := PaginatedShardQueryRequest{\n\t\tStartingShardID: params.StartingShardID,\n\t}\n\tqueryArgsBytes, err := json.Marshal(queryArgs)\n\tif err != nil {\n\t\treturn nil, cadence.NewCustomError(ErrSerialization)\n\t}\n\tqueryResp, err := client.QueryWorkflow(activityCtx, &shared.QueryWorkflowRequest{\n\t\tDomain: c.StringPtr(constants.SystemLocalDomainName),\n\t\tExecution: &shared.WorkflowExecution{\n\t\t\tWorkflowId: c.StringPtr(params.ScannerWorkflowWorkflowID),\n\t\t\tRunId:      c.StringPtr(params.ScannerWorkflowRunID),\n\t\t},\n\t\tQuery: &shared.WorkflowQuery{\n\t\t\tQueryType: c.StringPtr(ShardCorruptKeysQuery),\n\t\t\tQueryArgs: queryArgsBytes,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tqueryResult := &ShardCorruptKeysQueryResult{}\n\tif err := json.Unmarshal(queryResp.QueryResult, &queryResult); err != nil {\n\t\treturn nil, cadence.NewCustomError(ErrSerialization)\n\t}\n\tvar corrupted []CorruptedKeysEntry\n\tvar minShardID *int\n\tvar maxShardID *int\n\tfor sid, keys := range queryResult.Result {\n\t\tif minShardID == nil || *minShardID > sid {\n\t\t\tminShardID = c.IntPtr(sid)\n\t\t}\n\t\tif maxShardID == nil || *maxShardID < sid {\n\t\t\tmaxShardID = c.IntPtr(sid)\n\t\t}\n\t\tcorrupted = append(corrupted, CorruptedKeysEntry{\n\t\t\tShardID:       sid,\n\t\t\tCorruptedKeys: keys,\n\t\t})\n\t}\n\treturn &FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys:             corrupted,\n\t\tMinShard:                  minShardID,\n\t\tMaxShard:                  maxShardID,\n\t\tShardQueryPaginationToken: queryResult.ShardQueryPaginationToken,\n\t}, nil\n}\n\ntype (\n\tFixShardConfigParams struct {\n\t\t// intentionally empty, no args needed currently.  just reserving arg space for future needs.\n\t}\n\tFixShardConfigResults struct {\n\t\tEnabledInvariants CustomScannerConfig\n\t}\n)\n\n// fixerConfigActivity returns a list of all enabled invariants for this fixer.\n// The type of the workflow determines the type of the fixer (concrete, current, etc).\n//\n// It essentially mirrors scannerConfigActivity, but does not try to merge into a common structure.\nfunc fixerConfigActivity(activityCtx context.Context, params FixShardConfigParams) (*FixShardConfigResults, error) {\n\tctx, err := GetFixerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcfg := ctx.Hooks.GetFixerConfig(ctx)\n\tif len(cfg) == 0 {\n\t\t// sanity check for new code.  historically this field did not exist, now it is required to be populated.\n\t\treturn nil, fmt.Errorf(`invalid empty fixer config, you must explicitly specify \"true\" or \"false\" for all relevant invariants`)\n\t}\n\n\treturn &FixShardConfigResults{\n\t\tEnabledInvariants: cfg,\n\t}, nil\n}\n\n// fixShardActivity will fix a collection of shards.\nfunc fixShardActivity(\n\tactivityCtx context.Context,\n\tparams FixShardActivityParams,\n) ([]FixReport, error) {\n\tctx, err := GetFixerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\theartbeatDetails := FixShardHeartbeatDetails{\n\t\tLastShardIndexHandled: -1,\n\t\tReports:               nil,\n\t}\n\tif activity.HasHeartbeatDetails(activityCtx) {\n\t\tif err := activity.GetHeartbeatDetails(activityCtx, &heartbeatDetails); err != nil {\n\t\t\tctx.Logger.Error(\"getting heartbeat details\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t}\n\tfor i := heartbeatDetails.LastShardIndexHandled + 1; i < len(params.CorruptedKeysEntries); i++ {\n\t\tcurrentShardID := params.CorruptedKeysEntries[i].ShardID\n\t\tcurrentKeys := params.CorruptedKeysEntries[i].CorruptedKeys\n\t\tshardReport, err := fixShard(activityCtx, params, currentShardID, currentKeys, heartbeatDetails)\n\t\tif err != nil {\n\t\t\tctx.Logger.Error(\"fixing shard\", tag.Error(err))\n\t\t\treturn nil, err\n\t\t}\n\t\theartbeatDetails = FixShardHeartbeatDetails{\n\t\t\tLastShardIndexHandled: i,\n\t\t\tReports:               append(heartbeatDetails.Reports, *shardReport),\n\t\t}\n\t}\n\treturn heartbeatDetails.Reports, nil\n}\n\nfunc fixShard(\n\tactivityCtx context.Context,\n\tparams FixShardActivityParams,\n\tshardID int,\n\tcorruptedKeys store.Keys,\n\theartbeatDetails FixShardHeartbeatDetails,\n) (*FixReport, error) {\n\tctx, err := GetFixerContext(activityCtx)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tresource := ctx.Resource\n\tinfo := activity.GetInfo(activityCtx)\n\tscope := ctx.Scope.Tagged(\n\t\tmetrics.ActivityTypeTag(ActivityFixShard),\n\t\tmetrics.WorkflowTypeTag(info.WorkflowType.Name),\n\t\tmetrics.DomainTag(constants.SystemLocalDomainName),\n\t)\n\tsw := scope.StartTimer(metrics.CadenceLatency)\n\tdefer sw.Stop()\n\n\tif ctx.Hooks == nil {\n\t\treturn nil, cadence.NewCustomError(ErrMissingHooks)\n\t}\n\n\texecManager, err := resource.GetExecutionManager(shardID)\n\tif err != nil {\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t\treturn nil, err\n\t}\n\n\tpr := persistence.NewPersistenceRetryer(execManager, resource.GetHistoryManager(), c.CreatePersistenceRetryPolicy())\n\n\tfixer := NewFixer(\n\t\tactivityCtx,\n\t\tshardID,\n\t\tctx.Hooks.InvariantManager(activityCtx, pr, params, resource.GetDomainCache()),\n\t\tctx.Hooks.Iterator(activityCtx, resource.GetBlobstoreClient(), corruptedKeys, params),\n\t\tresource.GetBlobstoreClient(),\n\t\tparams.ResolvedFixerWorkflowConfig.BlobstoreFlushThreshold,\n\t\tfunc() { activity.RecordHeartbeat(activityCtx, heartbeatDetails) },\n\t\tresource.GetDomainCache(),\n\t\tctx.Config.DynamicParams.AllowDomain,\n\t\tscope,\n\t)\n\treport := fixer.Fix()\n\tif report.Result.ControlFlowFailure != nil {\n\t\tscope.IncCounter(metrics.CadenceFailures)\n\t}\n\treturn &report, nil\n}\n\n// scannerEmitMetricsActivity will emit metrics for a complete run of ShardScanner\nfunc scannerEmitMetricsActivity(\n\tactivityCtx context.Context,\n\tparams ScannerEmitMetricsActivityParams,\n) error {\n\tctx, err := GetScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tinfo := activity.GetInfo(activityCtx)\n\tscope := ctx.Scope.Tagged(\n\t\tmetrics.ActivityTypeTag(ActivityScannerEmitMetrics),\n\t\tmetrics.WorkflowTypeTag(info.WorkflowType.Name),\n\t\tmetrics.DomainTag(constants.SystemLocalDomainName),\n\t)\n\tscope.UpdateGauge(metrics.CadenceShardSuccessGauge, float64(params.ShardSuccessCount))\n\tscope.UpdateGauge(metrics.CadenceShardFailureGauge, float64(params.ShardControlFlowFailureCount))\n\n\tagg := params.AggregateReportResult\n\tscope.UpdateGauge(metrics.ScannerExecutionsGauge, float64(agg.EntitiesCount))\n\tscope.UpdateGauge(metrics.ScannerCorruptedGauge, float64(agg.CorruptedCount))\n\tscope.UpdateGauge(metrics.ScannerCheckFailedGauge, float64(agg.CheckFailedCount))\n\tfor k, v := range agg.CorruptionByType {\n\t\tscope.Tagged(metrics.InvariantTypeTag(string(k))).UpdateGauge(metrics.ScannerCorruptionByTypeGauge, float64(v))\n\t}\n\tshardStats := params.ShardDistributionStats\n\tscope.UpdateGauge(metrics.ScannerShardSizeMaxGauge, float64(shardStats.Max))\n\tscope.UpdateGauge(metrics.ScannerShardSizeMedianGauge, float64(shardStats.Median))\n\tscope.UpdateGauge(metrics.ScannerShardSizeMinGauge, float64(shardStats.Min))\n\tscope.UpdateGauge(metrics.ScannerShardSizeNinetyGauge, float64(shardStats.P90))\n\tscope.UpdateGauge(metrics.ScannerShardSizeSeventyFiveGauge, float64(shardStats.P75))\n\tscope.UpdateGauge(metrics.ScannerShardSizeTwentyFiveGauge, float64(shardStats.P25))\n\tscope.UpdateGauge(metrics.ScannerShardSizeTenGauge, float64(shardStats.P10))\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/activities_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nconst testWorkflowName = \"default-test-workflow-type-name\"\n\ntype activitiesSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\n\tcontroller   *gomock.Controller\n\tmockResource *resource.Test\n}\n\nfunc TestActivitiesSuite(t *testing.T) {\n\tsuite.Run(t, new(activitiesSuite))\n}\n\nfunc (s *activitiesSuite) SetupSuite() {\n\tactivity.Register(scannerConfigActivity)\n\tactivity.Register(fixerCorruptedKeysActivity)\n\tactivity.Register(scanShardActivity)\n\tactivity.Register(fixShardActivity)\n}\n\nfunc (s *activitiesSuite) SetupTest() {\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Worker)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\ts.mockResource.DomainCache = domainCache\n}\n\nfunc (s *activitiesSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *activitiesSuite) TestScanShardActivity() {\n\n\ttestCases := []struct {\n\t\tparams       ScanShardActivityParams\n\t\twantErr      bool\n\t\tmanagerHook  func(ctx context.Context, pr persistence.Retryer, params ScanShardActivityParams, cache cache.DomainCache) invariant.Manager\n\t\titHook       func(ctx context.Context, pr persistence.Retryer, params ScanShardActivityParams) pagination.Iterator\n\t\tworkflowName string\n\t}{\n\t\t{\n\t\t\tparams: ScanShardActivityParams{\n\t\t\t\tShards: []int{0},\n\t\t\t},\n\t\t\tmanagerHook: func(ctx context.Context, pr persistence.Retryer, params ScanShardActivityParams, cache cache.DomainCache) invariant.Manager {\n\t\t\t\tmanager := invariant.NewMockManager(s.controller)\n\t\t\t\tmanager.EXPECT().RunChecks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tAnyTimes().\n\t\t\t\t\tReturn(\n\t\t\t\t\t\tinvariant.ManagerCheckResult{CheckResultType: invariant.CheckResultTypeHealthy},\n\t\t\t\t\t)\n\t\t\t\treturn manager\n\t\t\t},\n\t\t\titHook: func(ctx context.Context, pr persistence.Retryer, params ScanShardActivityParams) pagination.Iterator {\n\t\t\t\tit := pagination.NewMockIterator(s.controller)\n\t\t\t\tcalls := 0\n\t\t\t\tit.EXPECT().HasNext().DoAndReturn(\n\t\t\t\t\tfunc() bool {\n\t\t\t\t\t\tif calls > 1 {\n\t\t\t\t\t\t\treturn false\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcalls++\n\t\t\t\t\t\treturn true\n\t\t\t\t\t},\n\t\t\t\t).AnyTimes()\n\t\t\t\tit.EXPECT().Next().Return(&entity.ConcreteExecution{}, nil).Times(2)\n\t\t\t\treturn it\n\t\t\t},\n\t\t\tworkflowName: testWorkflowName,\n\t\t\twantErr:      false,\n\t\t},\n\t\t{\n\t\t\tparams: ScanShardActivityParams{\n\t\t\t\tShards: []int{0},\n\t\t\t},\n\t\t\tworkflowName: \"wrong-test-name\",\n\t\t\twantErr:      true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\n\t\tenv := s.NewTestActivityEnvironment()\n\t\thooks, _ := NewScannerHooks(tc.managerHook, tc.itHook, func(scanner ScannerContext) CustomScannerConfig {\n\t\t\treturn nil // no config overrides\n\t\t})\n\t\tsc := NewShardScannerContext(s.mockResource, &ScannerConfig{\n\t\t\tScannerHooks: func() *ScannerHooks { return hooks },\n\t\t})\n\t\tctx := NewScannerContext(context.Background(), tc.workflowName, sc)\n\t\tenv.SetWorkerOptions(worker.Options{\n\t\t\tBackgroundActivityContext: ctx,\n\t\t})\n\t\treport, err := env.ExecuteActivity(scanShardActivity, tc.params)\n\t\tif tc.wantErr {\n\t\t\ts.Error(err)\n\t\t\tbreak\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t\tvar reports []ScanReport\n\t\ts.NoError(report.Get(&reports))\n\n\t\tfor _, v := range s.mockResource.MetricsScope.Snapshot().Timers() {\n\t\t\ttags := v.Tags()\n\t\t\ts.Equal(\"cadence-sys-shardscanner-scanshard-activity\", tags[\"activityType\"])\n\t\t\ts.Equal(tc.workflowName, tags[\"workflowType\"])\n\t\t}\n\n\t}\n}\n\nfunc (s *activitiesSuite) TestFixShardActivity() {\n\n\ttestCases := []struct {\n\t\tname        string\n\t\tparams      FixShardActivityParams\n\t\twantErr     bool\n\t\tmanagerHook FixerManagerCB\n\t\titHook      FixerIteratorCB\n\t}{\n\t\t{\n\t\t\tname:    \"run fixer\",\n\t\t\twantErr: false,\n\t\t\tparams: FixShardActivityParams{\n\t\t\t\tCorruptedKeysEntries: []CorruptedKeysEntry{\n\t\t\t\t\t{ShardID: 1, CorruptedKeys: struct {\n\t\t\t\t\t\tUUID      string\n\t\t\t\t\t\tMinPage   int\n\t\t\t\t\t\tMaxPage   int\n\t\t\t\t\t\tExtension store.Extension\n\t\t\t\t\t}{UUID: \"test-uuid\", MinPage: 0, MaxPage: 1, Extension: \"test-ext\"}},\n\t\t\t\t},\n\t\t\t\tResolvedFixerWorkflowConfig: ResolvedFixerWorkflowConfig{},\n\t\t\t},\n\t\t\tmanagerHook: func(ctx context.Context, pr persistence.Retryer, p FixShardActivityParams, cache cache.DomainCache) invariant.Manager {\n\t\t\t\tmanager := invariant.NewMockManager(s.controller)\n\t\t\t\tmanager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).\n\t\t\t\t\tAnyTimes().\n\t\t\t\t\tReturn(\n\t\t\t\t\t\tinvariant.ManagerFixResult{FixResultType: invariant.FixResultTypeFixed},\n\t\t\t\t\t)\n\t\t\t\treturn manager\n\t\t\t},\n\t\t\titHook: func(ctx context.Context, client blobstore.Client, k store.Keys, params FixShardActivityParams) store.ScanOutputIterator {\n\t\t\t\tit := store.NewMockScanOutputIterator(s.controller)\n\t\t\t\tcalls := 0\n\t\t\t\tit.EXPECT().HasNext().DoAndReturn(\n\t\t\t\t\tfunc() bool {\n\t\t\t\t\t\tif calls > 1 {\n\t\t\t\t\t\t\treturn false\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcalls++\n\t\t\t\t\t\treturn true\n\t\t\t\t\t},\n\t\t\t\t).AnyTimes()\n\t\t\t\tit.EXPECT().Next().Return(&store.ScanOutputEntity{\n\t\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\t\tDomainID: \"test_domain\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\t\t\t\treturn it\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\ts.mockResource.BlobstoreClient.\n\t\t\t\tEXPECT().\n\t\t\t\tPut(gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(&blobstore.PutResponse{}, nil).\n\t\t\t\tTimes(2)\n\t\t\tdomainCache := cache.NewMockDomainCache(s.controller)\n\t\t\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\t\t\ts.mockResource.DomainCache = domainCache\n\t\t\tcfg := &ScannerConfig{\n\t\t\t\tDynamicParams: DynamicParams{\n\t\t\t\t\tAllowDomain: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t\t\t},\n\t\t\t}\n\t\t\tif tc.itHook != nil && tc.managerHook != nil {\n\t\t\t\tcfg.FixerHooks = func() *FixerHooks {\n\t\t\t\t\treturn &FixerHooks{\n\t\t\t\t\t\tInvariantManager: tc.managerHook,\n\t\t\t\t\t\tIterator:         tc.itHook,\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tfc := NewShardFixerContext(s.mockResource, cfg)\n\n\t\t\tenv := s.NewTestActivityEnvironment()\n\n\t\t\tenv.SetWorkerOptions(worker.Options{\n\t\t\t\tBackgroundActivityContext: NewFixerContext(context.Background(), testWorkflowName, fc),\n\t\t\t})\n\t\t\treport, execErr := env.ExecuteActivity(fixShardActivity, tc.params)\n\t\t\tif tc.wantErr {\n\t\t\t\ts.Error(execErr)\n\t\t\t} else {\n\t\t\t\tvar reports []FixReport\n\t\t\t\ts.NoError(report.Get(&reports))\n\t\t\t}\n\t\t})\n\t}\n\n}\n\nfunc (s *activitiesSuite) TestScannerConfigActivity() {\n\ttestCases := []struct {\n\t\tdynamicParams *DynamicParams\n\t\tparams        ScannerConfigActivityParams\n\t\tresolved      ResolvedScannerWorkflowConfig\n\t\taddHook       bool\n\t}{\n\t\t{\n\t\t\tdynamicParams: &DynamicParams{\n\t\t\t\tScannerEnabled:          dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\tConcurrency:             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\tPageSize:                dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tActivityBatchSize:       dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\tBlobstoreFlushThreshold: dynamicproperties.GetIntPropertyFn(1000),\n\t\t\t},\n\t\t\tparams: ScannerConfigActivityParams{\n\t\t\t\tOverwrites: ScannerWorkflowConfigOverwrites{},\n\t\t\t},\n\t\t\taddHook: true,\n\t\t\tresolved: ResolvedScannerWorkflowConfig{\n\t\t\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\t\t\tEnabled:                 true,\n\t\t\t\t\tConcurrency:             10,\n\t\t\t\t\tActivityBatchSize:       10,\n\t\t\t\t\tPageSize:                100,\n\t\t\t\t\tBlobstoreFlushThreshold: 1000,\n\t\t\t\t},\n\t\t\t\tCustomScannerConfig: CustomScannerConfig{\n\t\t\t\t\t\"test-key\": \"test-value\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdynamicParams: &DynamicParams{\n\t\t\t\tScannerEnabled:          dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\tConcurrency:             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\tPageSize:                dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tActivityBatchSize:       dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\tBlobstoreFlushThreshold: dynamicproperties.GetIntPropertyFn(1000),\n\t\t\t},\n\t\t\tparams: ScannerConfigActivityParams{\n\t\t\t\tOverwrites: ScannerWorkflowConfigOverwrites{},\n\t\t\t},\n\t\t\tresolved: ResolvedScannerWorkflowConfig{\n\t\t\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\t\t\tEnabled:                 true,\n\t\t\t\t\tConcurrency:             10,\n\t\t\t\t\tActivityBatchSize:       10,\n\t\t\t\t\tPageSize:                100,\n\t\t\t\t\tBlobstoreFlushThreshold: 1000,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdynamicParams: &DynamicParams{\n\t\t\t\tScannerEnabled:          dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\tConcurrency:             dynamicproperties.GetIntPropertyFn(10),\n\t\t\t\tActivityBatchSize:       dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tPageSize:                dynamicproperties.GetIntPropertyFn(100),\n\t\t\t\tBlobstoreFlushThreshold: dynamicproperties.GetIntPropertyFn(1000),\n\t\t\t},\n\t\t\tparams: ScannerConfigActivityParams{\n\t\t\t\tOverwrites: ScannerWorkflowConfigOverwrites{\n\t\t\t\t\tGenericScannerConfig: GenericScannerConfigOverwrites{\n\t\t\t\t\t\tEnabled:                 common.BoolPtr(false),\n\t\t\t\t\t\tActivityBatchSize:       common.IntPtr(1),\n\t\t\t\t\t\tBlobstoreFlushThreshold: common.IntPtr(100),\n\t\t\t\t\t},\n\t\t\t\t\tCustomScannerConfig: &CustomScannerConfig{\n\t\t\t\t\t\t\"test\": \"test\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\tresolved: ResolvedScannerWorkflowConfig{\n\t\t\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\t\t\tEnabled:                 false,\n\t\t\t\t\tConcurrency:             10,\n\t\t\t\t\tActivityBatchSize:       1,\n\t\t\t\t\tPageSize:                100,\n\t\t\t\t\tBlobstoreFlushThreshold: 100,\n\t\t\t\t},\n\t\t\t\tCustomScannerConfig: CustomScannerConfig{\n\t\t\t\t\t\"test\": \"test\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tenv := s.NewTestActivityEnvironment()\n\n\t\tcfg := &ScannerConfig{\n\t\t\tDynamicParams: *tc.dynamicParams,\n\t\t\tScannerHooks: func() *ScannerHooks {\n\t\t\t\treturn &ScannerHooks{}\n\t\t\t},\n\t\t}\n\t\tif tc.addHook {\n\t\t\tcfg.ScannerHooks = func() *ScannerHooks {\n\t\t\t\treturn &ScannerHooks{\n\t\t\t\t\tGetScannerConfig: func(scanner ScannerContext) CustomScannerConfig {\n\t\t\t\t\t\treturn map[string]string{\"test-key\": \"test-value\"}\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tenv.SetWorkerOptions(worker.Options{\n\t\t\tBackgroundActivityContext: NewScannerContext(\n\t\t\t\tcontext.Background(),\n\t\t\t\ttestWorkflowName,\n\t\t\t\tNewShardScannerContext(s.mockResource, cfg),\n\t\t\t),\n\t\t},\n\t\t)\n\n\t\tresolvedValue, err := env.ExecuteActivity(scannerConfigActivity, tc.params)\n\t\ts.NoError(err)\n\t\tvar resolved ResolvedScannerWorkflowConfig\n\t\ts.NoError(resolvedValue.Get(&resolved))\n\t\ts.Equal(tc.resolved, resolved)\n\t}\n}\n\nfunc (s *activitiesSuite) TestFixerCorruptedKeysActivity() {\n\ts.mockResource.SDKClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&shared.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &shared.WorkflowExecutionInfo{\n\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t},\n\t}, nil)\n\tqueryResult := &ShardCorruptKeysQueryResult{\n\t\tResult: map[int]store.Keys{\n\t\t\t1: {UUID: \"first\"},\n\t\t\t2: {UUID: \"second\"},\n\t\t\t3: {UUID: \"third\"},\n\t\t},\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: common.IntPtr(4),\n\t\t\tIsDone:      false,\n\t\t},\n\t}\n\tqueryResultData, err := json.Marshal(queryResult)\n\tresponse := &shared.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions: []*shared.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tExecution: &shared.WorkflowExecution{\n\t\t\t\t\tWorkflowId: common.StringPtr(\"test-list-workflow-id\"),\n\t\t\t\t\tRunId:      common.StringPtr(uuid.New()),\n\t\t\t\t},\n\t\t\t\tType: &shared.WorkflowType{\n\t\t\t\t\tName: common.StringPtr(\"test-list-workflow-type\"),\n\t\t\t\t},\n\t\t\t\tStartTime:     common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\tCloseTime:     common.Int64Ptr(time.Now().Add(time.Hour).UnixNano()),\n\t\t\t\tCloseStatus:   shared.WorkflowExecutionCloseStatusContinuedAsNew.Ptr(),\n\t\t\t\tHistoryLength: common.Int64Ptr(12),\n\t\t\t},\n\t\t},\n\t}\n\ts.NoError(err)\n\ts.mockResource.SDKClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).Times(1).Return(response, nil)\n\n\ts.mockResource.SDKClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(&shared.QueryWorkflowResponse{\n\t\tQueryResult: queryResultData,\n\t}, nil)\n\tenv := s.getFixerActivityEnvironment()\n\tfixerResultValue, err := env.ExecuteActivity(fixerCorruptedKeysActivity, FixerCorruptedKeysActivityParams{})\n\ts.NoError(err)\n\tfixerResult := &FixerCorruptedKeysActivityResult{}\n\ts.NoError(fixerResultValue.Get(&fixerResult))\n\ts.Equal(1, *fixerResult.MinShard)\n\ts.Equal(3, *fixerResult.MaxShard)\n\ts.Equal(ShardQueryPaginationToken{\n\t\tNextShardID: common.IntPtr(4),\n\t\tIsDone:      false,\n\t}, fixerResult.ShardQueryPaginationToken)\n\ts.Contains(fixerResult.CorruptedKeys, CorruptedKeysEntry{\n\t\tShardID: 1,\n\t\tCorruptedKeys: store.Keys{\n\t\t\tUUID: \"first\",\n\t\t},\n\t})\n\ts.Contains(fixerResult.CorruptedKeys, CorruptedKeysEntry{\n\t\tShardID: 2,\n\t\tCorruptedKeys: store.Keys{\n\t\t\tUUID: \"second\",\n\t\t},\n\t})\n\ts.Contains(fixerResult.CorruptedKeys, CorruptedKeysEntry{\n\t\tShardID: 3,\n\t\tCorruptedKeys: store.Keys{\n\t\t\tUUID: \"third\",\n\t\t},\n\t})\n}\n\nfunc (s *activitiesSuite) TestFixerCorruptedKeysActivity_Fails_WhenNoSuitableExecutionsAreFound() {\n\tresponse := &shared.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions: []*shared.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusCanceled.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusTimedOut.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusTerminated.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tCloseStatus: shared.WorkflowExecutionCloseStatusFailed.Ptr(),\n\t\t\t},\n\t\t},\n\t}\n\ts.mockResource.SDKClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any(), gomock.Any()).Times(1).Return(response, nil)\n\n\tenv := s.getFixerActivityEnvironment()\n\tfixerResultValue, err := env.ExecuteActivity(fixerCorruptedKeysActivity, FixerCorruptedKeysActivityParams{})\n\ts.Nil(fixerResultValue)\n\ts.EqualError(err, \"failed to find a recent scanner workflow execution with ContinuedAsNew status\")\n}\n\nfunc (s *activitiesSuite) getFixerActivityEnvironment() *testsuite.TestActivityEnvironment {\n\tenv := s.NewTestActivityEnvironment()\n\tcfg := &ScannerConfig{\n\t\tDynamicParams: DynamicParams{\n\t\t\tAllowDomain: dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\t},\n\t\tFixerHooks: func() *FixerHooks {\n\t\t\treturn &FixerHooks{}\n\t\t},\n\t}\n\tfc := NewShardFixerContext(s.mockResource, cfg)\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: NewFixerContext(context.Background(), testWorkflowName, fc),\n\t})\n\treturn env\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/aggregators.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"fmt\"\n\t\"sort\"\n\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\nconst (\n\n\t// ShardStatusSuccess indicates the scan on the shard ran successfully\n\tShardStatusSuccess ShardStatus = \"success\"\n\t// ShardStatusControlFlowFailure indicates the scan on the shard failed\n\tShardStatusControlFlowFailure ShardStatus = \"control_flow_failure\"\n\t// ShardStatusRunning indicates the shard has not completed yet\n\tShardStatusRunning ShardStatus = \"running\"\n\n\tmaxShardQueryResult = 1000\n)\n\ntype (\n\t// ShardStatus is the type which indicates the status of a shard scan.\n\tShardStatus string\n\n\t// ShardStatusResult indicates the status for all shards\n\tShardStatusResult map[int]ShardStatus\n\n\t// ShardStatusSummaryResult indicates the counts of shards in each status\n\tShardStatusSummaryResult map[ShardStatus]int\n\n\t// AggregateScanReportResult indicates the result of summing together all\n\t// shard reports which have finished scan.\n\tAggregateScanReportResult ScanStats\n\n\t// AggregateFixReportResult indicates the result of summing together all\n\t// shard reports that have finished for fix.\n\tAggregateFixReportResult FixStats\n\n\t// ShardCorruptKeysResult is a map of all shards which have finished scan successfully and have at least one corruption\n\tShardCorruptKeysResult map[int]store.Keys\n\n\t// ScanReportError is a type that is used to send either error or report on a channel.\n\t// Exactly one of Report and ErrorStr should be non-nil.\n\tScanReportError struct {\n\t\tReports  []ScanReport\n\t\tErrorStr *string\n\t}\n\n\t// FixerCorruptedKeysActivityResult is the result of fixerCorruptedKeysActivity\n\tFixerCorruptedKeysActivityResult struct {\n\t\tCorruptedKeys             []CorruptedKeysEntry\n\t\tMinShard                  *int\n\t\tMaxShard                  *int\n\t\tShardQueryPaginationToken ShardQueryPaginationToken\n\t}\n\n\t// CorruptedKeysEntry is a pair of shardID and corrupted keys\n\tCorruptedKeysEntry struct {\n\t\tShardID       int\n\t\tCorruptedKeys store.Keys\n\t}\n\n\t// ScanShardHeartbeatDetails is the heartbeat details for scan shard\n\tScanShardHeartbeatDetails struct {\n\t\tLastShardIndexHandled int\n\t\tReports               []ScanReport\n\t}\n\n\t// FixShardHeartbeatDetails is the heartbeat details for the fix shard\n\tFixShardHeartbeatDetails struct {\n\t\tLastShardIndexHandled int\n\t\tReports               []FixReport\n\t}\n\n\t// FixReportError is a type that is used to send either error or report on a channel.\n\t// Exactly one of Report and ErrorStr should be non-nil.\n\tFixReportError struct {\n\t\tReports  []FixReport\n\t\tErrorStr *string\n\t}\n\n\t// ShardSizeQueryRequest is the request used for ShardSizeQuery.\n\t// The following must be true: 0 <= StartIndex < EndIndex <= len(shards successfully finished)\n\t// The following must be true: EndIndex - StartIndex <= maxShardQueryResult.\n\t// StartIndex is inclusive, EndIndex is exclusive.\n\tShardSizeQueryRequest struct {\n\t\tStartIndex int\n\t\tEndIndex   int\n\t}\n\n\t// PaginatedShardQueryRequest is the request used for queries which return results over all shards\n\tPaginatedShardQueryRequest struct {\n\t\t// StartingShardID is the first shard to start iteration from.\n\t\t// Setting to nil will start iteration from the beginning of the shards.\n\t\tStartingShardID *int\n\t\t// LimitShards indicates the maximum number of results that can be returned.\n\t\t// If nil or larger than allowed maximum, will default to maximum allowed.\n\t\tLimitShards *int\n\t}\n\n\t// ShardQueryPaginationToken is used to return information used to make the next query\n\tShardQueryPaginationToken struct {\n\t\t// NextShardID is one greater than the highest shard returned in the current query.\n\t\t// NextShardID is nil if IsDone is true.\n\t\t// It is possible to get NextShardID != nil and on the next call to get an empty result with IsDone = true.\n\t\tNextShardID *int\n\t\tIsDone      bool\n\t}\n\n\t// ShardStatusQueryResult is the query result for ShardStatusQuery\n\tShardStatusQueryResult struct {\n\t\tResult                    ShardStatusResult\n\t\tShardQueryPaginationToken ShardQueryPaginationToken\n\t}\n\n\t// ShardCorruptKeysQueryResult is the query result for ShardCorruptKeysQuery\n\tShardCorruptKeysQueryResult struct {\n\t\tResult                    ShardCorruptKeysResult\n\t\tShardQueryPaginationToken ShardQueryPaginationToken\n\t}\n\n\t// ShardSizeQueryResult is the result from ShardSizeQuery.\n\t// Contains sorted list of shards, sorted by the number of executions per shard.\n\tShardSizeQueryResult []ShardSizeTuple\n\n\t// ShardSizeTuple indicates the size and sorted index of a single shard\n\tShardSizeTuple struct {\n\t\tShardID       int\n\t\tEntitiesCount int64\n\t}\n\t// ShardDistributionStats contains stats on the distribution of executions in shards.\n\t// It is used by the ScannerEmitMetricsActivityParams.\n\tShardDistributionStats struct {\n\t\tMax    int64\n\t\tMedian int64\n\t\tMin    int64\n\t\tP90    int64\n\t\tP75    int64\n\t\tP25    int64\n\t\tP10    int64\n\t}\n\n\t// ShardFixResultAggregator is used to keep aggregated fix metrics\n\tShardFixResultAggregator struct {\n\t\tminShard int\n\t\tmaxShard int\n\n\t\treports       map[int]FixReport\n\t\tdomainStats   map[string]*FixStats\n\t\tstatus        ShardStatusResult\n\t\tstatusSummary ShardStatusSummaryResult\n\t\taggregation   AggregateFixReportResult\n\t}\n\n\t// ShardScanResultAggregator is used to keep aggregated scan metrics\n\tShardScanResultAggregator struct {\n\t\tminShard int\n\t\tmaxShard int\n\n\t\treports        map[int]ScanReport\n\t\tdomainStats    map[string]*ScanStats\n\t\tstatus         ShardStatusResult\n\t\tstatusSummary  ShardStatusSummaryResult\n\t\taggregation    AggregateScanReportResult\n\t\tshardSizes     ShardSizeQueryResult\n\t\tcorruptionKeys map[int]store.Keys\n\t}\n\n\t// DomainReportQueryRequest is the request used for queries which return stats broken by domains\n\tDomainReportQueryRequest struct {\n\t\t// DomainID specifies a single domain for which the stats are requested\n\t\t// Setting to nil indicates stats for all domains are requested\n\t\tDomainID *string\n\t}\n\n\t// DomainReportQueryResult is the query result for DomainReportQuery in the scanner workflow\n\tDomainScanReportQueryResult struct {\n\t\tReports []DomainScanStats\n\t}\n\n\t// DomainFixReportQueryResult is the query result for DomainReportQuery in the fixer workflow\n\tDomainFixReportQueryResult struct {\n\t\tReports []DomainFixStats\n\t}\n)\n\n// NewShardFixResultAggregator returns an instance of ShardFixResultAggregator\nfunc NewShardFixResultAggregator(\n\tcorruptKeys []CorruptedKeysEntry,\n\tminShard int,\n\tmaxShard int,\n) *ShardFixResultAggregator {\n\tstatus := make(map[int]ShardStatus, len(corruptKeys))\n\tfor _, s := range corruptKeys {\n\t\tstatus[s.ShardID] = ShardStatusRunning\n\t}\n\tstatusSummary := map[ShardStatus]int{\n\t\tShardStatusRunning:            len(corruptKeys),\n\t\tShardStatusSuccess:            0,\n\t\tShardStatusControlFlowFailure: 0,\n\t}\n\treturn &ShardFixResultAggregator{\n\t\tminShard: minShard,\n\t\tmaxShard: maxShard,\n\n\t\treports:       make(map[int]FixReport),\n\t\tdomainStats:   make(map[string]*FixStats),\n\t\tstatus:        status,\n\t\tstatusSummary: statusSummary,\n\t\taggregation:   AggregateFixReportResult{},\n\t}\n}\n\n// GetStatusSummary returns scan status summary.\nfunc (a *ShardScanResultAggregator) GetStatusSummary() ShardStatusSummaryResult {\n\treturn a.statusSummary\n}\n\n// GetStatusSummary returns fix status summary.\nfunc (a *ShardFixResultAggregator) GetStatusSummary() ShardStatusSummaryResult {\n\treturn a.statusSummary\n}\n\n// GetAggregation returns scan aggregation.\nfunc (a *ShardFixResultAggregator) GetAggregation() AggregateFixReportResult {\n\treturn a.aggregation\n}\n\n// GetStatusResult returns paginated results for a range of shards\nfunc (a *ShardFixResultAggregator) GetStatusResult(req PaginatedShardQueryRequest) (*ShardStatusQueryResult, error) {\n\treturn getStatusResult(a.minShard, a.maxShard, req, a.status)\n}\n\n// GetStatusResult returns stats broken by domain IDs\nfunc (a *ShardFixResultAggregator) GetDomainStatus(req DomainReportQueryRequest) (*DomainFixReportQueryResult, error) {\n\tresponseStats := []DomainFixStats{}\n\n\tif req.DomainID == nil {\n\t\tdomainIDs := make([]string, 0)\n\t\tfor domainID := range a.domainStats {\n\t\t\tdomainIDs = append(domainIDs, domainID)\n\t\t}\n\t\tsort.Strings(domainIDs)\n\t\tfor _, domainID := range domainIDs {\n\t\t\tresponseStats = append(responseStats, DomainFixStats{\n\t\t\t\tDomainID: domainID,\n\t\t\t\tStats:    *a.domainStats[domainID],\n\t\t\t})\n\t\t}\n\t} else if domainStats, ok := a.domainStats[*req.DomainID]; ok {\n\t\tresponseStats = append(responseStats, DomainFixStats{\n\t\t\tDomainID: *req.DomainID,\n\t\t\tStats:    *domainStats,\n\t\t})\n\t} else {\n\t\treturn nil, fmt.Errorf(\"domain %v does not exist or relevant stats have not been reported yet\", req.DomainID)\n\t}\n\n\tresult := &DomainFixReportQueryResult{\n\t\tReports: responseStats,\n\t}\n\n\treturn result, nil\n}\n\n// AddReport adds fix report for a shard.\nfunc (a *ShardFixResultAggregator) AddReport(report FixReport) {\n\ta.reports[report.ShardID] = report\n\ta.statusSummary[ShardStatusRunning]--\n\tif report.Result.ControlFlowFailure != nil {\n\t\ta.status[report.ShardID] = ShardStatusControlFlowFailure\n\t\ta.statusSummary[ShardStatusControlFlowFailure]++\n\t} else {\n\t\ta.status[report.ShardID] = ShardStatusSuccess\n\t\ta.statusSummary[ShardStatusSuccess]++\n\t}\n\tif report.Result.ShardFixKeys != nil {\n\t\ta.adjustAggregation(report.Stats, func(a, b int64) int64 { return a + b })\n\t}\n\tif report.DomainStats != nil {\n\t\ta.updateDomainStats(report)\n\t}\n}\n\nfunc (a *ShardFixResultAggregator) updateDomainStats(report FixReport) {\n\tfor domainID, domainStats := range report.DomainStats {\n\t\tif _, ok := a.domainStats[domainID]; !ok {\n\t\t\ta.domainStats[domainID] = &FixStats{}\n\t\t}\n\t\taggregateStats := a.domainStats[domainID]\n\t\taggregateStats.EntitiesCount += domainStats.EntitiesCount\n\t\taggregateStats.FixedCount += domainStats.FixedCount\n\t\taggregateStats.SkippedCount += domainStats.SkippedCount\n\t\taggregateStats.FailedCount += domainStats.FailedCount\n\t}\n}\n\n// GetReport returns fix report for a shard.\nfunc (a *ShardFixResultAggregator) GetReport(shardID int) (*FixReport, error) {\n\tif _, ok := a.status[shardID]; !ok {\n\t\treturn nil, fmt.Errorf(\"shard %v is not included in shards which will be processed\", shardID)\n\t}\n\tif report, ok := a.reports[shardID]; ok {\n\t\treturn &report, nil\n\t}\n\treturn nil, fmt.Errorf(\"shard %v has not finished yet, check back later for report\", shardID)\n}\n\nfunc (a *ShardFixResultAggregator) GetAllFixResults() (map[int]FixResult, error) {\n\tresult := make(map[int]FixResult, len(a.reports))\n\tfor k, v := range a.reports {\n\t\tif v.Result.Empty() {\n\t\t\tcontinue\n\t\t}\n\t\tresult[k] = v.Result\n\t}\n\treturn result, nil\n}\n\nfunc (a *ShardFixResultAggregator) adjustAggregation(stats FixStats, fn func(a, b int64) int64) {\n\ta.aggregation.EntitiesCount = fn(a.aggregation.EntitiesCount, stats.EntitiesCount)\n\ta.aggregation.SkippedCount = fn(a.aggregation.SkippedCount, stats.SkippedCount)\n\ta.aggregation.FailedCount = fn(a.aggregation.FailedCount, stats.FailedCount)\n\ta.aggregation.FixedCount = fn(a.aggregation.FixedCount, stats.FixedCount)\n}\n\n// NewShardScanResultAggregator returns aggregator for a scan result.\nfunc NewShardScanResultAggregator(\n\tshards []int,\n\tminShard int,\n\tmaxShard int,\n) *ShardScanResultAggregator {\n\tstatus := make(map[int]ShardStatus)\n\tfor _, s := range shards {\n\t\tstatus[s] = ShardStatusRunning\n\t}\n\tstatusSummary := map[ShardStatus]int{\n\t\tShardStatusSuccess:            0,\n\t\tShardStatusControlFlowFailure: 0,\n\t\tShardStatusRunning:            len(shards),\n\t}\n\treturn &ShardScanResultAggregator{\n\t\tminShard: minShard,\n\t\tmaxShard: maxShard,\n\n\t\treports:       make(map[int]ScanReport),\n\t\tdomainStats:   map[string]*ScanStats{},\n\t\tstatus:        status,\n\t\tstatusSummary: statusSummary,\n\t\tshardSizes:    nil,\n\t\taggregation: AggregateScanReportResult{\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tcorruptionKeys: make(map[int]store.Keys),\n\t}\n}\n\n// GetAggregateReport returns aggregated scan report.\nfunc (a *ShardScanResultAggregator) GetAggregateReport() AggregateScanReportResult {\n\treturn a.aggregation\n}\n\n// GetShardSizeQueryResult returns shard size statistics.\nfunc (a *ShardScanResultAggregator) GetShardSizeQueryResult(req ShardSizeQueryRequest) (ShardSizeQueryResult, error) {\n\tif req.StartIndex < 0 || req.StartIndex >= req.EndIndex || req.EndIndex > len(a.shardSizes) {\n\t\treturn nil, fmt.Errorf(\"index out of bounds exception (required startIndex >= 0 && startIndex < endIndex && endIndex <= %v)\", len(a.shardSizes))\n\t}\n\tif req.EndIndex-req.StartIndex > maxShardQueryResult {\n\t\treturn nil, fmt.Errorf(\"too many shards requested, the limit is %v\", maxShardQueryResult)\n\t}\n\treturn a.shardSizes[req.StartIndex:req.EndIndex], nil\n}\n\n// GetCorruptionKeys returns a list of corrupt keys\nfunc (a *ShardScanResultAggregator) GetCorruptionKeys(req PaginatedShardQueryRequest) (*ShardCorruptKeysQueryResult, error) {\n\tstartingShardID := a.minShard\n\tif req.StartingShardID != nil {\n\t\tstartingShardID = *req.StartingShardID\n\t}\n\tif err := shardInBounds(a.minShard, a.maxShard, startingShardID); err != nil {\n\t\treturn nil, err\n\t}\n\tlimit := maxShardQueryResult\n\tif req.LimitShards != nil && *req.LimitShards > 0 && *req.LimitShards < maxShardQueryResult {\n\t\tlimit = *req.LimitShards\n\t}\n\tresult := make(map[int]store.Keys)\n\tcurrentShardID := startingShardID\n\tfor len(result) < limit && currentShardID <= a.maxShard {\n\t\tkeys, ok := a.corruptionKeys[currentShardID]\n\t\tif !ok {\n\t\t\tcurrentShardID++\n\t\t\tcontinue\n\t\t}\n\t\tresult[currentShardID] = keys\n\t\tcurrentShardID++\n\t}\n\tif currentShardID > a.maxShard {\n\t\treturn &ShardCorruptKeysQueryResult{\n\t\t\tResult: result,\n\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\tNextShardID: nil,\n\t\t\t\tIsDone:      true,\n\t\t\t},\n\t\t}, nil\n\t}\n\treturn &ShardCorruptKeysQueryResult{\n\t\tResult: result,\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: &currentShardID,\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil\n}\n\n// GetStatusResult returns scan status for a range of shards.\nfunc (a *ShardScanResultAggregator) GetStatusResult(req PaginatedShardQueryRequest) (*ShardStatusQueryResult, error) {\n\treturn getStatusResult(a.minShard, a.maxShard, req, a.status)\n}\n\n// AddReport adds scan report for a shard.\nfunc (a *ShardScanResultAggregator) AddReport(report ScanReport) {\n\tif report.Result.ShardScanKeys != nil {\n\t\ta.insertReportIntoSizes(report)\n\t}\n\ta.reports[report.ShardID] = report\n\ta.statusSummary[ShardStatusRunning]--\n\tif report.Result.ControlFlowFailure != nil {\n\t\ta.status[report.ShardID] = ShardStatusControlFlowFailure\n\t\ta.statusSummary[ShardStatusControlFlowFailure]++\n\t} else {\n\t\ta.status[report.ShardID] = ShardStatusSuccess\n\t\ta.statusSummary[ShardStatusSuccess]++\n\t}\n\tif report.Result.ShardScanKeys != nil {\n\t\ta.adjustAggregation(report.Stats, func(a, b int64) int64 { return a + b })\n\t\tif report.Result.ShardScanKeys.Corrupt != nil {\n\t\t\ta.corruptionKeys[report.ShardID] = *report.Result.ShardScanKeys.Corrupt\n\t\t}\n\t}\n\tif report.DomainStats != nil {\n\t\ta.updateDomainStats(report)\n\t}\n}\n\nfunc (a *ShardScanResultAggregator) updateDomainStats(report ScanReport) {\n\tfor domainID, domainStats := range report.DomainStats {\n\t\tif _, ok := a.domainStats[domainID]; !ok {\n\t\t\ta.domainStats[domainID] = &ScanStats{\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t}\n\t\t}\n\t\taggregateStats := a.domainStats[domainID]\n\t\taggregateStats.CorruptedCount += domainStats.CorruptedCount\n\t\taggregateStats.CheckFailedCount += domainStats.CheckFailedCount\n\t\taggregateStats.EntitiesCount += domainStats.EntitiesCount\n\t\tfor corruptionType, count := range domainStats.CorruptionByType {\n\t\t\taggregateStats.CorruptionByType[corruptionType] += count\n\t\t}\n\n\t}\n}\n\nfunc (a *ShardScanResultAggregator) insertReportIntoSizes(report ScanReport) {\n\ttuple := ShardSizeTuple{\n\t\tShardID:       report.ShardID,\n\t\tEntitiesCount: report.Stats.EntitiesCount,\n\t}\n\tinsertIndex := 0\n\tfor insertIndex < len(a.shardSizes) {\n\t\tif a.shardSizes[insertIndex].EntitiesCount < tuple.EntitiesCount {\n\t\t\tbreak\n\t\t}\n\t\tinsertIndex++\n\t}\n\ta.shardSizes = append(a.shardSizes, ShardSizeTuple{})\n\tcopy(a.shardSizes[insertIndex+1:], a.shardSizes[insertIndex:])\n\ta.shardSizes[insertIndex] = tuple\n}\n\n// GetShardDistributionStats returns aggregated size statistics\nfunc (a *ShardScanResultAggregator) GetShardDistributionStats() ShardDistributionStats {\n\treturn ShardDistributionStats{\n\t\tMax:    a.shardSizes[0].EntitiesCount,\n\t\tMedian: a.shardSizes[int(float64(len(a.shardSizes))*.5)].EntitiesCount,\n\t\tMin:    a.shardSizes[len(a.shardSizes)-1].EntitiesCount,\n\t\tP90:    a.shardSizes[int(float64(len(a.shardSizes))*.1)].EntitiesCount,\n\t\tP75:    a.shardSizes[int(float64(len(a.shardSizes))*.25)].EntitiesCount,\n\t\tP25:    a.shardSizes[int(float64(len(a.shardSizes))*.75)].EntitiesCount,\n\t\tP10:    a.shardSizes[int(float64(len(a.shardSizes))*.9)].EntitiesCount,\n\t}\n}\n\n// GetReport returns a report for a single shard.\nfunc (a *ShardScanResultAggregator) GetReport(shardID int) (*ScanReport, error) {\n\tif _, ok := a.status[shardID]; !ok {\n\t\treturn nil, fmt.Errorf(\"shard %v is not included in shards which will be processed\", shardID)\n\t}\n\tif report, ok := a.reports[shardID]; ok {\n\t\treturn &report, nil\n\t}\n\treturn nil, fmt.Errorf(\"shard %v has not finished yet, check back later for report\", shardID)\n}\n\n// GetDomainStatus returns stats broken by domain IDs\nfunc (a *ShardScanResultAggregator) GetDomainStatus(req DomainReportQueryRequest) (*DomainScanReportQueryResult, error) {\n\tresponseStats := []DomainScanStats{}\n\n\tif req.DomainID == nil {\n\t\tdomainIDs := make([]string, 0)\n\t\tfor domainID := range a.domainStats {\n\t\t\tdomainIDs = append(domainIDs, domainID)\n\t\t}\n\t\tsort.Strings(domainIDs)\n\t\tfor _, domainID := range domainIDs {\n\t\t\tresponseStats = append(responseStats, DomainScanStats{\n\t\t\t\tDomainID: domainID,\n\t\t\t\tStats:    *a.domainStats[domainID],\n\t\t\t})\n\t\t}\n\t} else if domainStats, ok := a.domainStats[*req.DomainID]; ok {\n\t\tresponseStats = append(responseStats, DomainScanStats{\n\t\t\tDomainID: *req.DomainID,\n\t\t\tStats:    *domainStats,\n\t\t})\n\t} else {\n\t\treturn nil, fmt.Errorf(\"domain %v does not exist or relevant stats have not been reported yet\", req.DomainID)\n\t}\n\n\tresult := &DomainScanReportQueryResult{\n\t\tReports: responseStats,\n\t}\n\n\treturn result, nil\n}\n\nfunc (a *ShardScanResultAggregator) adjustAggregation(stats ScanStats, fn func(a, b int64) int64) {\n\ta.aggregation.EntitiesCount = fn(a.aggregation.EntitiesCount, stats.EntitiesCount)\n\ta.aggregation.CorruptedCount = fn(a.aggregation.CorruptedCount, stats.CorruptedCount)\n\ta.aggregation.CheckFailedCount = fn(a.aggregation.CheckFailedCount, stats.CheckFailedCount)\n\tfor k, v := range stats.CorruptionByType {\n\t\ta.aggregation.CorruptionByType[k] = fn(a.aggregation.CorruptionByType[k], v)\n\t}\n}\n\nfunc (a *ShardScanResultAggregator) GetAllScanResults() (map[int]ScanResult, error) {\n\tresult := make(map[int]ScanResult, len(a.reports))\n\tfor k, v := range a.reports {\n\t\tif v.Result.Empty() {\n\t\t\tcontinue\n\t\t}\n\t\tresult[k] = v.Result\n\t}\n\treturn result, nil\n}\n\nfunc getStatusResult(\n\tminShardID int,\n\tmaxShardID int,\n\treq PaginatedShardQueryRequest,\n\tstatus ShardStatusResult,\n) (*ShardStatusQueryResult, error) {\n\tstartingShardID := minShardID\n\tif req.StartingShardID != nil {\n\t\tstartingShardID = *req.StartingShardID\n\t}\n\tif err := shardInBounds(minShardID, maxShardID, startingShardID); err != nil {\n\t\treturn nil, err\n\t}\n\tlimit := maxShardQueryResult\n\tif req.LimitShards != nil && *req.LimitShards > 0 && *req.LimitShards < maxShardQueryResult {\n\t\tlimit = *req.LimitShards\n\t}\n\tresult := make(map[int]ShardStatus)\n\tcurrentShardID := startingShardID\n\tfor len(result) < limit && currentShardID <= maxShardID {\n\t\tstatus, ok := status[currentShardID]\n\t\tif !ok {\n\t\t\tcurrentShardID++\n\t\t\tcontinue\n\t\t}\n\t\tresult[currentShardID] = status\n\t\tcurrentShardID++\n\t}\n\tif currentShardID > maxShardID {\n\t\treturn &ShardStatusQueryResult{\n\t\t\tResult: result,\n\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\tNextShardID: nil,\n\t\t\t\tIsDone:      true,\n\t\t\t},\n\t\t}, nil\n\t}\n\treturn &ShardStatusQueryResult{\n\t\tResult: result,\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: &currentShardID,\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil\n}\n\nfunc shardInBounds(minShardID, maxShardID, shardID int) error {\n\tif shardID > maxShardID || shardID < minShardID {\n\t\treturn fmt.Errorf(\"requested shard %v is outside of bounds (min: %v and max: %v)\", shardID, minShardID, maxShardID)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/aggregators_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"math/rand\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\tc \"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\ntype aggregatorsSuite struct {\n\tsuite.Suite\n}\n\nfunc TestAggregatorSuite(t *testing.T) {\n\tsuite.Run(t, new(aggregatorsSuite))\n}\n\nfunc (s *aggregatorsSuite) TestShardScanResultAggregator() {\n\tagg := NewShardScanResultAggregator([]int{1, 2, 3}, 1, 3)\n\texpected := &ShardScanResultAggregator{\n\t\tminShard:    1,\n\t\tmaxShard:    3,\n\t\treports:     map[int]ScanReport{},\n\t\tdomainStats: map[string]*ScanStats{},\n\t\tstatus: map[int]ShardStatus{\n\t\t\t1: ShardStatusRunning,\n\t\t\t2: ShardStatusRunning,\n\t\t\t3: ShardStatusRunning,\n\t\t},\n\t\taggregation: AggregateScanReportResult{\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tcorruptionKeys: make(map[int]store.Keys),\n\t\tstatusSummary: map[ShardStatus]int{\n\t\t\tShardStatusRunning:            3,\n\t\t\tShardStatusControlFlowFailure: 0,\n\t\t\tShardStatusSuccess:            0,\n\t\t},\n\t\tshardSizes: nil,\n\t}\n\ts.Equal(expected, agg)\n\treport, err := agg.GetReport(1)\n\ts.Nil(report)\n\ts.Equal(\"shard 1 has not finished yet, check back later for report\", err.Error())\n\treport, err = agg.GetReport(5)\n\ts.Nil(report)\n\ts.Equal(\"shard 5 is not included in shards which will be processed\", err.Error())\n\tfirstReport := ScanReport{\n\t\tShardID: 1,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    10,\n\t\t\tCorruptedCount:   3,\n\t\t\tCheckFailedCount: 1,\n\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\tinvariant.HistoryExists:        2,\n\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t},\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tShardScanKeys: &ScanKeys{\n\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\tUUID: \"test_uuid\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"ABC\": &ScanStats{\n\t\t\t\tEntitiesCount:    5,\n\t\t\t\tCorruptedCount:   2,\n\t\t\t\tCheckFailedCount: 1,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.HistoryExists:        1,\n\t\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"DEF\": &ScanStats{\n\t\t\t\tEntitiesCount:  5,\n\t\t\t\tCorruptedCount: 1,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tagg.AddReport(firstReport)\n\texpected.status[1] = ShardStatusSuccess\n\texpected.statusSummary[ShardStatusRunning] = 2\n\texpected.statusSummary[ShardStatusSuccess] = 1\n\texpected.reports[1] = firstReport\n\texpected.shardSizes = []ShardSizeTuple{\n\t\t{\n\t\t\tShardID:       1,\n\t\t\tEntitiesCount: 10,\n\t\t},\n\t}\n\texpected.aggregation.EntitiesCount = 10\n\texpected.aggregation.CorruptedCount = 3\n\texpected.aggregation.CheckFailedCount = 1\n\texpected.aggregation.CorruptionByType = map[invariant.Name]int64{\n\t\tinvariant.HistoryExists:        2,\n\t\tinvariant.OpenCurrentExecution: 1,\n\t}\n\texpected.domainStats[\"ABC\"] = &ScanStats{\n\t\tEntitiesCount:    5,\n\t\tCorruptedCount:   2,\n\t\tCheckFailedCount: 1,\n\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\tinvariant.HistoryExists:        1,\n\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t},\n\t}\n\texpected.domainStats[\"DEF\"] = &ScanStats{\n\t\tEntitiesCount:  5,\n\t\tCorruptedCount: 1,\n\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\tinvariant.HistoryExists: 1,\n\t\t},\n\t}\n\n\texpected.corruptionKeys = map[int]store.Keys{\n\t\t1: {\n\t\t\tUUID: \"test_uuid\",\n\t\t},\n\t}\n\ts.Equal(expected, agg)\n\treport, err = agg.GetReport(1)\n\ts.NoError(err)\n\ts.Equal(firstReport, *report)\n\tsecondReport := ScanReport{\n\t\tShardID: 2,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    10,\n\t\t\tCorruptedCount:   3,\n\t\t\tCheckFailedCount: 1,\n\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\tinvariant.HistoryExists:        2,\n\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t},\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"ABC\": &ScanStats{\n\t\t\t\tEntitiesCount:    5,\n\t\t\t\tCorruptedCount:   2,\n\t\t\t\tCheckFailedCount: 1,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.HistoryExists:        1,\n\t\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"DEF\": &ScanStats{\n\t\t\t\tEntitiesCount:  5,\n\t\t\t\tCorruptedCount: 1,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.HistoryExists: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tagg.AddReport(secondReport)\n\texpected.status[2] = ShardStatusControlFlowFailure\n\texpected.statusSummary[ShardStatusRunning] = 1\n\texpected.statusSummary[ShardStatusControlFlowFailure] = 1\n\texpected.reports[2] = secondReport\n\texpected.shardSizes = []ShardSizeTuple{\n\t\t{\n\t\t\tShardID:       1,\n\t\t\tEntitiesCount: 10,\n\t\t},\n\t}\n\texpected.domainStats[\"ABC\"].EntitiesCount *= 2\n\texpected.domainStats[\"ABC\"].CorruptedCount *= 2\n\texpected.domainStats[\"ABC\"].CheckFailedCount *= 2\n\texpected.domainStats[\"ABC\"].CorruptionByType[invariant.HistoryExists] *= 2\n\texpected.domainStats[\"ABC\"].CorruptionByType[invariant.OpenCurrentExecution] *= 2\n\texpected.domainStats[\"DEF\"].EntitiesCount *= 2\n\texpected.domainStats[\"DEF\"].CorruptedCount *= 2\n\texpected.domainStats[\"DEF\"].CorruptionByType[invariant.HistoryExists] *= 2\n\ts.Equal(expected, agg)\n\tshardStatus, err := agg.GetStatusResult(PaginatedShardQueryRequest{\n\t\tStartingShardID: c.IntPtr(1),\n\t\tLimitShards:     c.IntPtr(2),\n\t})\n\ts.NoError(err)\n\ts.Equal(&ShardStatusQueryResult{\n\t\tResult: map[int]ShardStatus{\n\t\t\t1: ShardStatusSuccess,\n\t\t\t2: ShardStatusControlFlowFailure,\n\t\t},\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: c.IntPtr(3),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, shardStatus)\n\tcorruptedKeys, err := agg.GetCorruptionKeys(PaginatedShardQueryRequest{\n\t\tStartingShardID: c.IntPtr(1),\n\t\tLimitShards:     c.IntPtr(3),\n\t})\n\ts.NoError(err)\n\ts.Equal(&ShardCorruptKeysQueryResult{\n\t\tResult: map[int]store.Keys{\n\t\t\t1: {\n\t\t\t\tUUID: \"test_uuid\",\n\t\t\t},\n\t\t},\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: nil,\n\t\t\tIsDone:      true,\n\t\t},\n\t}, corruptedKeys)\n\n\tdomainReport, err := agg.GetDomainStatus(DomainReportQueryRequest{\n\t\tDomainID: c.StringPtr(\"ABC\"),\n\t})\n\ts.NoError(err)\n\ts.Equal(&DomainScanReportQueryResult{\n\t\tReports: []DomainScanStats{\n\t\t\t{\n\t\t\t\tDomainID: \"ABC\",\n\t\t\t\tStats: ScanStats{\n\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\tCorruptedCount:   4,\n\t\t\t\t\tCheckFailedCount: 2,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\tinvariant.HistoryExists:        2,\n\t\t\t\t\t\tinvariant.OpenCurrentExecution: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, domainReport)\n\tallDomainsReport, err := agg.GetDomainStatus(DomainReportQueryRequest{})\n\ts.NoError(err)\n\ts.Equal(&DomainScanReportQueryResult{\n\t\tReports: []DomainScanStats{\n\t\t\t{\n\t\t\t\tDomainID: \"ABC\",\n\t\t\t\tStats: ScanStats{\n\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\tCorruptedCount:   4,\n\t\t\t\t\tCheckFailedCount: 2,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\tinvariant.HistoryExists:        2,\n\t\t\t\t\t\tinvariant.OpenCurrentExecution: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\t{\n\t\t\t\tDomainID: \"DEF\",\n\t\t\t\tStats: ScanStats{\n\t\t\t\t\tEntitiesCount:  10,\n\t\t\t\t\tCorruptedCount: 2,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\t\tinvariant.HistoryExists: 2,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, allDomainsReport)\n}\n\nfunc (s *aggregatorsSuite) TestShardFixResultAggregator() {\n\tagg := NewShardFixResultAggregator([]CorruptedKeysEntry{{ShardID: 1}, {ShardID: 2}, {ShardID: 3}}, 1, 3)\n\texpected := &ShardFixResultAggregator{\n\t\tminShard:    1,\n\t\tmaxShard:    3,\n\t\treports:     map[int]FixReport{},\n\t\tdomainStats: map[string]*FixStats{},\n\t\tstatus: map[int]ShardStatus{\n\t\t\t1: ShardStatusRunning,\n\t\t\t2: ShardStatusRunning,\n\t\t\t3: ShardStatusRunning,\n\t\t},\n\t\tstatusSummary: map[ShardStatus]int{\n\t\t\tShardStatusRunning:            3,\n\t\t\tShardStatusControlFlowFailure: 0,\n\t\t\tShardStatusSuccess:            0,\n\t\t},\n\t\taggregation: AggregateFixReportResult{},\n\t}\n\ts.Equal(expected, agg)\n\treport, err := agg.GetReport(1)\n\ts.Nil(report)\n\ts.Equal(\"shard 1 has not finished yet, check back later for report\", err.Error())\n\treport, err = agg.GetReport(5)\n\ts.Nil(report)\n\ts.Equal(\"shard 5 is not included in shards which will be processed\", err.Error())\n\tfirstReport := FixReport{\n\t\tShardID: 1,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 10,\n\t\t\tFixedCount:    3,\n\t\t\tFailedCount:   1,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tShardFixKeys: &FixKeys{\n\t\t\t\tFixed: &store.Keys{\n\t\t\t\t\tUUID: \"test_uuid\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\tagg.AddReport(firstReport)\n\texpected.status[1] = ShardStatusSuccess\n\texpected.statusSummary[ShardStatusSuccess] = 1\n\texpected.statusSummary[ShardStatusRunning] = 2\n\texpected.reports[1] = firstReport\n\texpected.aggregation.EntitiesCount = 10\n\texpected.aggregation.FixedCount = 3\n\texpected.aggregation.FailedCount = 1\n\ts.Equal(expected, agg)\n\treport, err = agg.GetReport(1)\n\ts.NoError(err)\n\ts.Equal(firstReport, *report)\n\tsecondReport := FixReport{\n\t\tShardID: 2,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 10,\n\t\t\tFixedCount:    3,\n\t\t\tFailedCount:   1,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{},\n\t\t},\n\t}\n\tagg.AddReport(secondReport)\n\texpected.status[2] = ShardStatusControlFlowFailure\n\texpected.statusSummary[ShardStatusControlFlowFailure] = 1\n\texpected.statusSummary[ShardStatusRunning] = 1\n\texpected.reports[2] = secondReport\n\ts.Equal(expected, agg)\n\tshardStatus, err := agg.GetStatusResult(PaginatedShardQueryRequest{\n\t\tStartingShardID: c.IntPtr(1),\n\t\tLimitShards:     c.IntPtr(2),\n\t})\n\ts.NoError(err)\n\ts.Equal(&ShardStatusQueryResult{\n\t\tResult: map[int]ShardStatus{\n\t\t\t1: ShardStatusSuccess,\n\t\t\t2: ShardStatusControlFlowFailure,\n\t\t},\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: c.IntPtr(3),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, shardStatus)\n}\n\nfunc (s *aggregatorsSuite) TestGetStatusResult() {\n\ttestCases := []struct {\n\t\tminShardID     int\n\t\tmaxShardID     int\n\t\treq            PaginatedShardQueryRequest\n\t\tstatus         ShardStatusResult\n\t\texpectedResult *ShardStatusQueryResult\n\t\texpectedError  bool\n\t}{\n\t\t{\n\t\t\tminShardID: 0,\n\t\t\tmaxShardID: 5,\n\t\t\treq: PaginatedShardQueryRequest{\n\t\t\t\tStartingShardID: c.IntPtr(6),\n\t\t\t},\n\t\t\texpectedResult: nil,\n\t\t\texpectedError:  true,\n\t\t},\n\t\t{\n\t\t\tminShardID: 0,\n\t\t\tmaxShardID: 5,\n\t\t\treq: PaginatedShardQueryRequest{\n\t\t\t\tStartingShardID: c.IntPtr(0),\n\t\t\t\tLimitShards:     c.IntPtr(10),\n\t\t\t},\n\t\t\tstatus: map[int]ShardStatus{\n\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t3: ShardStatusSuccess,\n\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t},\n\t\t\texpectedResult: &ShardStatusQueryResult{\n\t\t\t\tResult: map[int]ShardStatus{\n\t\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t\t3: ShardStatusSuccess,\n\t\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t\t},\n\t\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\t\tNextShardID: nil,\n\t\t\t\t\tIsDone:      true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tminShardID: 0,\n\t\t\tmaxShardID: 5,\n\t\t\treq: PaginatedShardQueryRequest{\n\t\t\t\tStartingShardID: c.IntPtr(0),\n\t\t\t\tLimitShards:     c.IntPtr(2),\n\t\t\t},\n\t\t\tstatus: map[int]ShardStatus{\n\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t3: ShardStatusSuccess,\n\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t},\n\t\t\texpectedResult: &ShardStatusQueryResult{\n\t\t\t\tResult: map[int]ShardStatus{\n\t\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t},\n\t\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\t\tNextShardID: c.IntPtr(3),\n\t\t\t\t\tIsDone:      false,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tminShardID: 0,\n\t\t\tmaxShardID: 5,\n\t\t\treq: PaginatedShardQueryRequest{\n\t\t\t\tStartingShardID: c.IntPtr(0),\n\t\t\t\tLimitShards:     c.IntPtr(3),\n\t\t\t},\n\t\t\tstatus: map[int]ShardStatus{\n\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t},\n\t\t\texpectedResult: &ShardStatusQueryResult{\n\t\t\t\tResult: map[int]ShardStatus{\n\t\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t},\n\t\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\t\tNextShardID: c.IntPtr(5),\n\t\t\t\t\tIsDone:      false,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tminShardID: 0,\n\t\t\tmaxShardID: 5,\n\t\t\treq: PaginatedShardQueryRequest{\n\t\t\t\tStartingShardID: c.IntPtr(2),\n\t\t\t\tLimitShards:     c.IntPtr(3),\n\t\t\t},\n\t\t\tstatus: map[int]ShardStatus{\n\t\t\t\t1: ShardStatusRunning,\n\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t},\n\t\t\texpectedResult: &ShardStatusQueryResult{\n\t\t\t\tResult: map[int]ShardStatus{\n\t\t\t\t\t2: ShardStatusRunning,\n\t\t\t\t\t4: ShardStatusSuccess,\n\t\t\t\t\t5: ShardStatusControlFlowFailure,\n\t\t\t\t},\n\t\t\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\t\t\tNextShardID: nil,\n\t\t\t\t\tIsDone:      true,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedError: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tresult, err := getStatusResult(tc.minShardID, tc.maxShardID, tc.req, tc.status)\n\t\ts.Equal(tc.expectedResult, result)\n\t\tif tc.expectedError {\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t}\n}\n\nfunc (s *aggregatorsSuite) TestGetShardSizeQueryResult() {\n\ttestCases := []struct {\n\t\tshardSizes       []ShardSizeTuple\n\t\treq              ShardSizeQueryRequest\n\t\texpectedErrorStr *string\n\t\texpectedResult   ShardSizeQueryResult\n\t}{\n\t\t{\n\t\t\tshardSizes: nil,\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: -1,\n\t\t\t},\n\t\t\texpectedErrorStr: c.StringPtr(\"index out of bounds exception (required startIndex >= 0 && startIndex < endIndex && endIndex <= 0)\"),\n\t\t\texpectedResult:   nil,\n\t\t},\n\t\t{\n\t\t\tshardSizes: nil,\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 1,\n\t\t\t\tEndIndex:   1,\n\t\t\t},\n\t\t\texpectedErrorStr: c.StringPtr(\"index out of bounds exception (required startIndex >= 0 && startIndex < endIndex && endIndex <= 0)\"),\n\t\t\texpectedResult:   nil,\n\t\t},\n\t\t{\n\t\t\tshardSizes: nil,\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 0,\n\t\t\t\tEndIndex:   1,\n\t\t\t},\n\t\t\texpectedErrorStr: c.StringPtr(\"index out of bounds exception (required startIndex >= 0 && startIndex < endIndex && endIndex <= 0)\"),\n\t\t\texpectedResult:   nil,\n\t\t},\n\t\t{\n\t\t\tshardSizes: make([]ShardSizeTuple, 10),\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 0,\n\t\t\t\tEndIndex:   11,\n\t\t\t},\n\t\t\texpectedErrorStr: c.StringPtr(\"index out of bounds exception (required startIndex >= 0 && startIndex < endIndex && endIndex <= 10)\"),\n\t\t\texpectedResult:   nil,\n\t\t},\n\t\t{\n\t\t\tshardSizes: make([]ShardSizeTuple, 10000),\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 0,\n\t\t\t\tEndIndex:   maxShardQueryResult + 1,\n\t\t\t},\n\t\t\texpectedErrorStr: c.StringPtr(\"too many shards requested, the limit is 1000\"),\n\t\t\texpectedResult:   nil,\n\t\t},\n\t\t{\n\t\t\tshardSizes: []ShardSizeTuple{\n\t\t\t\t{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 2,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 3,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 4,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 5,\n\t\t\t\t},\n\t\t\t},\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 0,\n\t\t\t\tEndIndex:   1,\n\t\t\t},\n\t\t\texpectedErrorStr: nil,\n\t\t\texpectedResult: []ShardSizeTuple{\n\t\t\t\t{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tshardSizes: []ShardSizeTuple{\n\t\t\t\t{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 2,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 3,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 4,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 5,\n\t\t\t\t},\n\t\t\t},\n\t\t\treq: ShardSizeQueryRequest{\n\t\t\t\tStartIndex: 0,\n\t\t\t\tEndIndex:   5,\n\t\t\t},\n\t\t\texpectedErrorStr: nil,\n\t\t\texpectedResult: []ShardSizeTuple{\n\t\t\t\t{\n\t\t\t\t\tShardID: 1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 2,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 3,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 4,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tShardID: 5,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tagg := &ShardScanResultAggregator{\n\t\t\tshardSizes: tc.shardSizes,\n\t\t}\n\t\tresult, err := agg.GetShardSizeQueryResult(tc.req)\n\t\tif tc.expectedErrorStr != nil {\n\t\t\ts.Equal(*tc.expectedErrorStr, err.Error())\n\t\t} else {\n\t\t\ts.Equal(tc.expectedResult, result)\n\t\t}\n\t}\n}\n\nfunc (s *aggregatorsSuite) TestInsertReportIntoSizes() {\n\trandomReport := func() ScanReport {\n\t\treturn ScanReport{\n\t\t\tShardID: 0,\n\t\t\tStats: ScanStats{\n\t\t\t\tEntitiesCount: int64(rand.Intn(10)),\n\t\t\t},\n\t\t}\n\t}\n\tagg := &ShardScanResultAggregator{}\n\tfor i := 0; i < 1000; i++ {\n\t\tagg.insertReportIntoSizes(randomReport())\n\t}\n\tfor i := 0; i < 999; i++ {\n\t\ts.GreaterOrEqual(agg.shardSizes[i].EntitiesCount, agg.shardSizes[i+1].EntitiesCount)\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/fixer.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\n// Fixer is used to fix entities in a shard. It is responsible for three things:\n// 1. Confirming that each entity it scans is corrupted.\n// 2. Attempting to fix any confirmed corrupted executions.\n// 3. Recording skipped entities, failed to fix entities and successfully fix entities to durable store.\n// 4. Producing a FixReport\ntype Fixer interface {\n\tFix() FixReport\n}\n\ntype (\n\t// ShardFixer is a generic fixer which iterates over entities provided by iterator\n\t// implementations of this fixer have to provided invariant manager and iterator.\n\tShardFixer struct {\n\t\tctx              context.Context\n\t\tshardID          int\n\t\titr              store.ScanOutputIterator\n\t\tskippedWriter    store.ExecutionWriter\n\t\tfailedWriter     store.ExecutionWriter\n\t\tfixedWriter      store.ExecutionWriter\n\t\tinvariantManager invariant.Manager\n\t\tprogressReportFn func()\n\t\tdomainCache      cache.DomainCache\n\t\tallowDomain      dynamicproperties.BoolPropertyFnWithDomainFilter\n\t\tscope            metrics.Scope\n\t}\n)\n\n// NewFixer constructs a new shard fixer.\nfunc NewFixer(\n\tctx context.Context,\n\tshardID int,\n\tmanager invariant.Manager,\n\titerator store.ScanOutputIterator,\n\tblobstoreClient blobstore.Client,\n\tblobstoreFlushThreshold int,\n\tprogressReportFn func(),\n\tdomainCache cache.DomainCache,\n\tallowDomain dynamicproperties.BoolPropertyFnWithDomainFilter,\n\tscope metrics.Scope,\n) *ShardFixer {\n\tid := uuid.New()\n\n\treturn &ShardFixer{\n\t\tctx:              ctx,\n\t\tshardID:          shardID,\n\t\titr:              iterator,\n\t\tskippedWriter:    store.NewBlobstoreWriter(id, store.SkippedExtension, blobstoreClient, blobstoreFlushThreshold),\n\t\tfailedWriter:     store.NewBlobstoreWriter(id, store.FailedExtension, blobstoreClient, blobstoreFlushThreshold),\n\t\tfixedWriter:      store.NewBlobstoreWriter(id, store.FixedExtension, blobstoreClient, blobstoreFlushThreshold),\n\t\tinvariantManager: manager,\n\t\tprogressReportFn: progressReportFn,\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      allowDomain,\n\t\tscope:            scope,\n\t}\n}\n\n// Fix scans over all executions in shard and runs invariant fixes per execution.\nfunc (f *ShardFixer) Fix() FixReport {\n\n\tresult := FixReport{\n\t\tShardID:     f.shardID,\n\t\tDomainStats: map[string]*FixStats{},\n\t}\n\n\tfor f.itr.HasNext() {\n\t\tf.progressReportFn()\n\t\tsoe, err := f.itr.Next()\n\t\tif err != nil {\n\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore iterator returned error\",\n\t\t\t\tInfoDetails: err.Error(),\n\t\t\t}\n\t\t\treturn result\n\t\t}\n\n\t\tdomainID := soe.Execution.(entity.Entity).GetDomainID()\n\t\tdomainName, err := f.domainCache.GetDomainName(domainID)\n\t\tif err != nil {\n\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to get domain name\",\n\t\t\t\tInfoDetails: err.Error(),\n\t\t\t}\n\t\t\treturn result\n\t\t}\n\t\tif _, ok := result.DomainStats[domainID]; !ok {\n\t\t\tresult.DomainStats[domainID] = &FixStats{}\n\t\t}\n\n\t\tvar fixResult invariant.ManagerFixResult\n\n\t\tif f.allowDomain(domainName) {\n\t\t\tfixResult = f.invariantManager.RunFixes(f.ctx, soe.Execution)\n\t\t} else {\n\t\t\tfixResult = invariant.ManagerFixResult{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t}\n\t\t}\n\t\tresult.Stats.EntitiesCount++\n\t\tresult.DomainStats[domainID].EntitiesCount++\n\t\tfoe := store.FixOutputEntity{\n\t\t\tExecution: soe.Execution,\n\t\t\tInput:     *soe,\n\t\t\tResult:    fixResult,\n\t\t}\n\n\t\tinvariantName := \"\"\n\t\tif fixResult.DeterminingInvariantName != nil {\n\t\t\tinvariantName = string(*fixResult.DeterminingInvariantName)\n\t\t}\n\n\t\tf.scope.Tagged(\n\t\t\tmetrics.DomainTag(domainName),\n\t\t\tmetrics.InvariantTypeTag(invariantName),\n\t\t\tmetrics.ShardScannerFixResult(string(fixResult.FixResultType)),\n\t\t).IncCounter(metrics.ShardScannerFix)\n\n\t\tswitch fixResult.FixResultType {\n\t\tcase invariant.FixResultTypeFixed:\n\t\t\tif err := f.fixedWriter.Add(foe); err != nil {\n\t\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\t\tInfo:        \"blobstore add failed for fixed execution fix\",\n\t\t\t\t\tInfoDetails: err.Error(),\n\t\t\t\t}\n\t\t\t\treturn result\n\t\t\t}\n\t\t\tresult.Stats.FixedCount++\n\t\t\tresult.DomainStats[domainID].FixedCount++\n\t\tcase invariant.FixResultTypeSkipped:\n\t\t\tif err := f.skippedWriter.Add(foe); err != nil {\n\t\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\t\tInfo:        \"blobstore add failed for skipped execution fix\",\n\t\t\t\t\tInfoDetails: err.Error(),\n\t\t\t\t}\n\t\t\t\treturn result\n\t\t\t}\n\t\t\tresult.Stats.SkippedCount++\n\t\t\tresult.DomainStats[domainID].SkippedCount++\n\t\tcase invariant.FixResultTypeFailed:\n\t\t\tif err := f.failedWriter.Add(foe); err != nil {\n\t\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\t\tInfo:        \"blobstore add failed for failed execution fix\",\n\t\t\t\t\tInfoDetails: err.Error(),\n\t\t\t\t}\n\t\t\t\treturn result\n\t\t\t}\n\t\t\tresult.Stats.FailedCount++\n\t\t\tresult.DomainStats[domainID].FailedCount++\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown FixResultType: %v\", fixResult.FixResultType))\n\t\t}\n\t}\n\tif err := f.fixedWriter.Flush(); err != nil {\n\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\tInfo:        \"failed to flush for fixed execution fixes\",\n\t\t\tInfoDetails: err.Error(),\n\t\t}\n\t\treturn result\n\t}\n\tif err := f.skippedWriter.Flush(); err != nil {\n\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\tInfo:        \"failed to flush for skipped execution fixes\",\n\t\t\tInfoDetails: err.Error(),\n\t\t}\n\t\treturn result\n\t}\n\tif err := f.failedWriter.Flush(); err != nil {\n\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\tInfo:        \"failed to flush for failed execution fixes\",\n\t\t\tInfoDetails: err.Error(),\n\t\t}\n\t\treturn result\n\t}\n\tresult.Result.ShardFixKeys = &FixKeys{\n\t\tFixed:   f.fixedWriter.FlushedKeys(),\n\t\tFailed:  f.failedWriter.FlushedKeys(),\n\t\tSkipped: f.skippedWriter.FlushedKeys(),\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/fixer_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\ntype FixerSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\n\tmockResource *resource.Test\n\tcontroller   *gomock.Controller\n}\n\nfunc TestFixerSuite(t *testing.T) {\n\tsuite.Run(t, new(FixerSuite))\n}\n\nfunc (s *FixerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n\ts.mockResource = resource.NewTest(s.T(), s.controller, metrics.Worker)\n}\n\nfunc (s *FixerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *FixerSuite) TestFix_Failure_FirstIteratorError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(nil, errors.New(\"iterator error\")).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore iterator returned error\",\n\t\t\t\tInfoDetails: \"iterator error\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_NonFirstError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\titeratorCallNumber := 0\n\tmockItr.EXPECT().HasNext().DoAndReturn(func() bool {\n\t\treturn iteratorCallNumber < 5\n\t}).Times(5)\n\tmockItr.EXPECT().Next().DoAndReturn(func() (*store.ScanOutputEntity, error) {\n\t\tdefer func() {\n\t\t\titeratorCallNumber++\n\t\t}()\n\t\tif iteratorCallNumber < 4 {\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"test_domain\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\t}\n\t\treturn nil, fmt.Errorf(\"iterator got error on: %v\", iteratorCallNumber)\n\t}).Times(5)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFixed,\n\t}).Times(4)\n\tfixedWriter := store.NewMockExecutionWriter(s.controller)\n\tfixedWriter.EXPECT().Add(gomock.Any()).Return(nil).Times(4)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).Times(4)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tinvariantManager: mockInvariantManager,\n\t\tfixedWriter:      fixedWriter,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 4,\n\t\t\tFixedCount:    4,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore iterator returned error\",\n\t\t\t\tInfoDetails: \"iterator got error on: 4\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{\n\t\t\t\"test_domain\": {\n\t\t\t\tEntitiesCount: 4,\n\t\t\t\tFixedCount:    4,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_SkippedWriterError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(&store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"test_domain\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t}).Times(1)\n\tskippedWriter := store.NewMockExecutionWriter(s.controller)\n\tskippedWriter.EXPECT().Add(gomock.Any()).Return(errors.New(\"skipped writer error\")).Times(1)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tskippedWriter:    skippedWriter,\n\t\tinvariantManager: mockInvariantManager,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 1,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore add failed for skipped execution fix\",\n\t\t\t\tInfoDetails: \"skipped writer error\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{\n\t\t\t\"test_domain\": {\n\t\t\t\tEntitiesCount: 1,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_FailedWriterError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(&store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"test_domain\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFailed,\n\t}).Times(1)\n\tfailedWriter := store.NewMockExecutionWriter(s.controller)\n\tfailedWriter.EXPECT().Add(gomock.Any()).Return(errors.New(\"failed writer error\")).Times(1)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfailedWriter:     failedWriter,\n\t\tinvariantManager: mockInvariantManager,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 1,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore add failed for failed execution fix\",\n\t\t\t\tInfoDetails: \"failed writer error\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{\n\t\t\t\"test_domain\": {\n\t\t\t\tEntitiesCount: 1,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_FixedWriterError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(&store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"test_domain\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFixed,\n\t}).Times(1)\n\tfixedWriter := store.NewMockExecutionWriter(s.controller)\n\tfixedWriter.EXPECT().Add(gomock.Any()).Return(errors.New(\"fixed writer error\")).Times(1)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfixedWriter:      fixedWriter,\n\t\tinvariantManager: mockInvariantManager,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      dynamicproperties.GetBoolPropertyFnFilteredByDomain(true),\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 1,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore add failed for fixed execution fix\",\n\t\t\t\tInfoDetails: \"fixed writer error\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{\n\t\t\t\"test_domain\": {\n\t\t\t\tEntitiesCount: 1,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_FixedWriterFlushError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(false).Times(1)\n\tfixedWriter := store.NewMockExecutionWriter(s.controller)\n\tfixedWriter.EXPECT().Flush().Return(errors.New(\"fix writer flush failed\")).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfixedWriter:      fixedWriter,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to flush for fixed execution fixes\",\n\t\t\t\tInfoDetails: \"fix writer flush failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_SkippedWriterFlushError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(false).Times(1)\n\tfixedWriter := store.NewMockExecutionWriter(s.controller)\n\tfixedWriter.EXPECT().Flush().Return(nil)\n\tskippedWriter := store.NewMockExecutionWriter(s.controller)\n\tskippedWriter.EXPECT().Flush().Return(errors.New(\"skip writer flush failed\")).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfixedWriter:      fixedWriter,\n\t\tskippedWriter:    skippedWriter,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to flush for skipped execution fixes\",\n\t\t\t\tInfoDetails: \"skip writer flush failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Failure_FailedWriterFlushError() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(false).Times(1)\n\tfixedWriter := store.NewMockExecutionWriter(s.controller)\n\tfixedWriter.EXPECT().Flush().Return(nil)\n\tskippedWriter := store.NewMockExecutionWriter(s.controller)\n\tskippedWriter.EXPECT().Flush().Return(nil).Times(1)\n\tfailedWriter := store.NewMockExecutionWriter(s.controller)\n\tfailedWriter.EXPECT().Flush().Return(errors.New(\"fail writer flush failed\")).Times(1)\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfixedWriter:      fixedWriter,\n\t\tskippedWriter:    skippedWriter,\n\t\tfailedWriter:     failedWriter,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tResult: FixResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to flush for failed execution fixes\",\n\t\t\t\tInfoDetails: \"fail writer flush failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{},\n\t}, result)\n}\n\nfunc (s *FixerSuite) TestFix_Success() {\n\tmockItr := store.NewMockScanOutputIterator(s.controller)\n\titeratorCallNumber := 0\n\tmockItr.EXPECT().HasNext().DoAndReturn(func() bool {\n\t\treturn iteratorCallNumber < 12\n\t}).Times(13)\n\tmockItr.EXPECT().Next().DoAndReturn(func() (*store.ScanOutputEntity, error) {\n\t\tdefer func() {\n\t\t\titeratorCallNumber++\n\t\t}()\n\t\tswitch iteratorCallNumber {\n\t\tcase 0, 1, 2, 3:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"skipped\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 4, 5:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"history_missing\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 6:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"first_history_event\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 7:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"orphan_execution\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 8, 9:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"failed\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 10, 11:\n\t\t\treturn &store.ScanOutputEntity{\n\t\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\t\tDomainID: \"disallow_domain\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, nil\n\t\tdefault:\n\t\t\tpanic(\"should not get here\")\n\t\t}\n\t}).Times(12)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"skipped\",\n\t\t},\n\t}).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\tFixResults: []invariant.FixResult{\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t},\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t},\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\tInvariantName: invariant.OpenCurrentExecution,\n\t\t\t},\n\t\t},\n\t}).Times(4)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"history_missing\",\n\t\t},\n\t}).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\tFixResults: []invariant.FixResult{\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\tInfo:          \"history did not exist\",\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"first_history_event\",\n\t\t},\n\t}).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\tFixResults: []invariant.FixResult{\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t},\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\tInfo:          \"first event is not valid\",\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"orphan_execution\",\n\t\t\tState:    persistence.WorkflowStateCreated,\n\t\t},\n\t}).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\tFixResults: []invariant.FixResult{\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t},\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t},\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\tInvariantName: invariant.OpenCurrentExecution,\n\t\t\t\tInfo:          \"execution was orphan\",\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"failed\",\n\t\t},\n\t}).Return(invariant.ManagerFixResult{\n\t\tFixResultType: invariant.FixResultTypeFailed,\n\t\tFixResults: []invariant.FixResult{\n\t\t\t{\n\t\t\t\tFixResultType: invariant.FixResultTypeFailed,\n\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\tInfo:          \"failed to check if history exists\",\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\n\tmockFixedWriter := store.NewMockExecutionWriter(s.controller)\n\tmockFixedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"history_missing\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"history_missing\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\tFixResults: []invariant.FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\t\tInfo:          \"history did not exist\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\tmockFixedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"first_history_event\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"first_history_event\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\tFixResults: []invariant.FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\t\tInfo:          \"first event is not valid\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockFixedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"orphan_execution\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"orphan_execution\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\tFixResults: []invariant.FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeFixed,\n\t\t\t\t\tInvariantName: invariant.OpenCurrentExecution,\n\t\t\t\t\tInfo:          \"execution was orphan\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockFailedWriter := store.NewMockExecutionWriter(s.controller)\n\tmockFailedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"failed\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"failed\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeFailed,\n\t\t\tFixResults: []invariant.FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeFailed,\n\t\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\t\tInfo:          \"failed to check if history exists\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\tmockSkippedWriter := store.NewMockExecutionWriter(s.controller)\n\tmockSkippedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"skipped\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"skipped\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\tFixResults: []invariant.FixResult{\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: invariant.HistoryExists,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t\t\t\tInvariantName: invariant.OpenCurrentExecution,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(4)\n\tmockSkippedWriter.EXPECT().Add(store.FixOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"disallow_domain\",\n\t\t\t},\n\t\t},\n\t\tInput: store.ScanOutputEntity{\n\t\t\tExecution: &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"disallow_domain\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tResult: invariant.ManagerFixResult{\n\t\t\tFixResultType: invariant.FixResultTypeSkipped,\n\t\t},\n\t}).Times(2)\n\tmockSkippedWriter.EXPECT().Flush().Return(nil)\n\tmockFailedWriter.EXPECT().Flush().Return(nil)\n\tmockFixedWriter.EXPECT().Flush().Return(nil)\n\tmockSkippedWriter.EXPECT().FlushedKeys().Return(&store.Keys{UUID: \"skipped_keys_uuid\"})\n\tmockFailedWriter.EXPECT().FlushedKeys().Return(&store.Keys{UUID: \"failed_keys_uuid\"})\n\tmockFixedWriter.EXPECT().FlushedKeys().Return(&store.Keys{UUID: \"fixed_keys_uuid\"})\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(\"skipped\").Return(\"skipped\", nil).Times(4)\n\tdomainCache.EXPECT().GetDomainName(\"history_missing\").Return(\"history_missing\", nil).Times(2)\n\tdomainCache.EXPECT().GetDomainName(\"first_history_event\").Return(\"first_history_event\", nil).Times(1)\n\tdomainCache.EXPECT().GetDomainName(\"orphan_execution\").Return(\"orphan_execution\", nil).Times(1)\n\tdomainCache.EXPECT().GetDomainName(\"failed\").Return(\"failed\", nil).Times(2)\n\tdomainCache.EXPECT().GetDomainName(\"disallow_domain\").Return(\"disallow_domain\", nil).Times(2)\n\n\tallowDomain := func(domain string) bool {\n\t\treturn domain != \"disallow_domain\"\n\t}\n\tfixer := &ShardFixer{\n\t\tshardID:          0,\n\t\tinvariantManager: mockInvariantManager,\n\t\tskippedWriter:    mockSkippedWriter,\n\t\tfailedWriter:     mockFailedWriter,\n\t\tfixedWriter:      mockFixedWriter,\n\t\titr:              mockItr,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tallowDomain:      allowDomain,\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := fixer.Fix()\n\ts.Equal(FixReport{\n\t\tShardID: 0,\n\t\tStats: FixStats{\n\t\t\tEntitiesCount: 12,\n\t\t\tFixedCount:    4,\n\t\t\tSkippedCount:  6,\n\t\t\tFailedCount:   2,\n\t\t},\n\t\tResult: FixResult{\n\t\t\tShardFixKeys: &FixKeys{\n\t\t\t\tFixed:   &store.Keys{UUID: \"fixed_keys_uuid\"},\n\t\t\t\tFailed:  &store.Keys{UUID: \"failed_keys_uuid\"},\n\t\t\t\tSkipped: &store.Keys{UUID: \"skipped_keys_uuid\"},\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*FixStats{\n\t\t\t\"disallow_domain\": {\n\t\t\t\tEntitiesCount: 2,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  2,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t\t\"failed\": {\n\t\t\t\tEntitiesCount: 2,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   2,\n\t\t\t},\n\t\t\t\"first_history_event\": {\n\t\t\t\tEntitiesCount: 1,\n\t\t\t\tFixedCount:    1,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t\t\"history_missing\": {\n\t\t\t\tEntitiesCount: 2,\n\t\t\t\tFixedCount:    2,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t\t\"orphan_execution\": {\n\t\t\t\tEntitiesCount: 1,\n\t\t\t\tFixedCount:    1,\n\t\t\t\tSkippedCount:  0,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t\t\"skipped\": {\n\t\t\t\tEntitiesCount: 4,\n\t\t\t\tFixedCount:    0,\n\t\t\t\tSkippedCount:  4,\n\t\t\t\tFailedCount:   0,\n\t\t\t},\n\t\t},\n\t}, result)\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/fixer_workflow.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\nconst (\n\tfixShardReportChan = \"fixShardReportChan\"\n)\n\nvar (\n\terrQueryNotReady = errors.New(\"query is not yet ready to be handled, please try again shortly\")\n)\n\ntype (\n\t// FixerManagerCB is a function which returns invariant manager for fixer.\n\tFixerManagerCB func(\n\t\tcontext.Context,\n\t\tpersistence.Retryer,\n\t\tFixShardActivityParams,\n\t\tcache.DomainCache,\n\t) invariant.Manager\n\n\t// FixerIteratorCB is a function which returns ScanOutputIterator for fixer.\n\tFixerIteratorCB func(\n\t\tcontext.Context,\n\t\tblobstore.Client,\n\t\tstore.Keys,\n\t\tFixShardActivityParams,\n\t) store.ScanOutputIterator\n\n\t// FixerHooks holds callback functions for shard scanner workflow implementation.\n\tFixerHooks struct {\n\t\tInvariantManager FixerManagerCB\n\t\tIterator         FixerIteratorCB\n\t\tGetFixerConfig   func(fixer FixerContext) CustomScannerConfig\n\t}\n\n\t// FixerWorkflow is the workflow that fixes all entities from a scan output.\n\tFixerWorkflow struct {\n\t\tAggregator *ShardFixResultAggregator\n\t\tParams     FixerWorkflowParams\n\t\tKeys       *FixerCorruptedKeysActivityResult\n\t}\n)\n\n// NewFixerHooks returns initialized callbacks for shard scanner workflow implementation.\nfunc NewFixerHooks(\n\tmanager FixerManagerCB,\n\titerator FixerIteratorCB,\n\tconfig func(fixer FixerContext) CustomScannerConfig,\n) (*FixerHooks, error) {\n\tif manager == nil || iterator == nil || config == nil {\n\t\treturn nil, errors.New(\"all fixer hooks args are required\")\n\t}\n\treturn &FixerHooks{\n\t\tInvariantManager: manager,\n\t\tIterator:         iterator,\n\t\tGetFixerConfig:   config,\n\t}, nil\n}\n\n// NewFixerWorkflow returns a new instance of fixer workflow\nfunc NewFixerWorkflow(\n\tctx workflow.Context,\n\tname string,\n\tparams FixerWorkflowParams,\n) (*FixerWorkflow, error) {\n\n\tif len(name) < 1 {\n\t\treturn nil, errors.New(\"workflow name is not provided\")\n\t}\n\n\twf := FixerWorkflow{\n\t\tParams: params,\n\t}\n\n\tcorruptKeys, err := GetCorruptedKeys(ctx, wf.Params)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif corruptKeys.CorruptedKeys == nil {\n\t\treturn nil, errors.New(\"corrupted keys not found\")\n\t}\n\n\twf.Keys = corruptKeys\n\twf.Aggregator = NewShardFixResultAggregator(corruptKeys.CorruptedKeys, *corruptKeys.MinShard, *corruptKeys.MaxShard)\n\n\tfor name, fn := range setHandlers(wf.Aggregator) {\n\t\tif err := workflow.SetQueryHandler(ctx, name, fn); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn &wf, nil\n}\n\nfunc supportsFixerConfig(ctx workflow.Context) bool {\n\t// this can probably be removed after a version or three, it just prevents a one-time\n\t// non-determinism failure when resuming a previous version's fixer run.\n\treturn workflow.GetVersion(ctx, \"dynamic fixer config\", workflow.DefaultVersion, 1) == 1\n}\n\n// Start starts a shard fixer workflow.\nfunc (fx *FixerWorkflow) Start(ctx workflow.Context) error {\n\n\tresolvedConfig := resolveFixerConfig(fx.Params.FixerWorkflowConfigOverwrites)\n\n\tvar enabled CustomScannerConfig\n\tif supportsFixerConfig(ctx) {\n\t\tactivityCtx := getShortActivityContext(ctx)\n\t\tvar out FixShardConfigResults\n\t\tif err := workflow.ExecuteActivity(activityCtx, ActivityFixerConfig, FixShardConfigParams{}).Get(activityCtx, &out); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tenabled = out.EnabledInvariants\n\t}\n\n\tshardReportChan := workflow.GetSignalChannel(ctx, fixShardReportChan)\n\n\tfor i := 0; i < resolvedConfig.Concurrency; i++ {\n\t\tidx := i\n\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\tbatches := getCorruptedKeysBatches(resolvedConfig.ActivityBatchSize, resolvedConfig.Concurrency, fx.Keys.CorruptedKeys, idx)\n\t\t\tfor _, batch := range batches {\n\t\t\t\tactivityCtx := getLongActivityContext(ctx)\n\t\t\t\tvar reports []FixReport\n\t\t\t\tif err := workflow.ExecuteActivity(activityCtx, ActivityFixShard, FixShardActivityParams{\n\t\t\t\t\tCorruptedKeysEntries:        batch,\n\t\t\t\t\tResolvedFixerWorkflowConfig: resolvedConfig,\n\t\t\t\t\tEnabledInvariants:           enabled,\n\t\t\t\t}).Get(ctx, &reports); err != nil {\n\t\t\t\t\terrStr := err.Error()\n\t\t\t\t\tshardReportChan.Send(ctx, FixReportError{\n\t\t\t\t\t\tReports:  nil,\n\t\t\t\t\t\tErrorStr: &errStr,\n\t\t\t\t\t})\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tshardReportChan.Send(ctx, FixReportError{\n\t\t\t\t\tReports:  reports,\n\t\t\t\t\tErrorStr: nil,\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n\n\tfor i := 0; i < len(fx.Keys.CorruptedKeys); {\n\t\tvar reportErr FixReportError\n\t\tshardReportChan.Receive(ctx, &reportErr)\n\t\tif reportErr.ErrorStr != nil {\n\t\t\treturn errors.New(*reportErr.ErrorStr)\n\t\t}\n\t\tfor _, report := range reportErr.Reports {\n\t\t\tfx.Aggregator.AddReport(report)\n\t\t\ti++\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc resolveFixerConfig(overwrites FixerWorkflowConfigOverwrites) ResolvedFixerWorkflowConfig {\n\tresolvedConfig := ResolvedFixerWorkflowConfig{\n\t\tConcurrency:             25,\n\t\tBlobstoreFlushThreshold: 1000,\n\t\tActivityBatchSize:       200,\n\t}\n\tif overwrites.Concurrency != nil {\n\t\tresolvedConfig.Concurrency = *overwrites.Concurrency\n\t}\n\tif overwrites.BlobstoreFlushThreshold != nil {\n\t\tresolvedConfig.BlobstoreFlushThreshold = *overwrites.BlobstoreFlushThreshold\n\t}\n\tif overwrites.ActivityBatchSize != nil {\n\t\tresolvedConfig.ActivityBatchSize = *overwrites.ActivityBatchSize\n\t}\n\treturn resolvedConfig\n}\n\nfunc setHandlers(aggregator *ShardFixResultAggregator) map[string]interface{} {\n\treturn map[string]interface{}{\n\t\tShardReportQuery: func(shardID int) (*FixReport, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn nil, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetReport(shardID)\n\t\t},\n\t\tShardStatusQuery: func(req PaginatedShardQueryRequest) (*ShardStatusQueryResult, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn nil, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetStatusResult(req)\n\t\t},\n\t\tShardStatusSummaryQuery: func() (ShardStatusSummaryResult, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn nil, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetStatusSummary(), nil\n\t\t},\n\t\tAggregateReportQuery: func() (AggregateFixReportResult, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn AggregateFixReportResult{}, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetAggregation(), nil\n\t\t},\n\t\tDomainReportQuery: func(req DomainReportQueryRequest) (*DomainFixReportQueryResult, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn nil, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetDomainStatus(req)\n\t\t},\n\t\tAllResultsQuery: func() (map[int]FixResult, error) {\n\t\t\tif aggregator == nil {\n\t\t\t\treturn nil, errQueryNotReady\n\t\t\t}\n\t\t\treturn aggregator.GetAllFixResults()\n\t\t},\n\t}\n}\n\nfunc getCorruptedKeysBatches(\n\tbatchSize int,\n\tconcurrency int,\n\tcorruptedKeys []CorruptedKeysEntry,\n\tworkerIdx int,\n) [][]CorruptedKeysEntry {\n\tbatchIndices := getBatchIndices(batchSize, concurrency, len(corruptedKeys), workerIdx)\n\tvar result [][]CorruptedKeysEntry\n\tfor _, batch := range batchIndices {\n\t\tvar curr []CorruptedKeysEntry\n\t\tfor _, i := range batch {\n\t\t\tcurr = append(curr, corruptedKeys[i])\n\t\t}\n\t\tresult = append(result, curr)\n\t}\n\treturn result\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/fixer_workflow_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\ntype fixerWorkflowSuite struct {\n\tsuite.Suite\n}\n\nfunc TestFixerWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(fixerWorkflowSuite))\n}\n\nfunc (s *fixerWorkflowSuite) TestResolveFixerConfig() {\n\tresult := resolveFixerConfig(FixerWorkflowConfigOverwrites{\n\t\tConcurrency: common.IntPtr(1000),\n\t})\n\ts.Equal(ResolvedFixerWorkflowConfig{\n\t\tConcurrency:             1000,\n\t\tBlobstoreFlushThreshold: 1000,\n\t\tActivityBatchSize:       200,\n\t}, result)\n}\n\nfunc (s *fixerWorkflowSuite) TestGetCorruptedKeysBatches() {\n\tvar keys []CorruptedKeysEntry\n\tfor i := 5; i < 50; i += 2 {\n\t\tkeys = append(keys, CorruptedKeysEntry{\n\t\t\tShardID: i,\n\t\t})\n\t}\n\tbatches := getCorruptedKeysBatches(5, 3, keys, 1)\n\ts.Equal([][]CorruptedKeysEntry{\n\t\t{\n\t\t\t{ShardID: 7},\n\t\t\t{ShardID: 13},\n\t\t\t{ShardID: 19},\n\t\t\t{ShardID: 25},\n\t\t\t{ShardID: 31},\n\t\t},\n\t\t{\n\t\t\t{ShardID: 37},\n\t\t\t{ShardID: 43},\n\t\t\t{ShardID: 49},\n\t\t},\n\t}, batches)\n}\n\nfunc (s *fixerWorkflowSuite) TestNewFixerHooks() {\n\ttestCases := []struct {\n\t\tname     string\n\t\tmanager  FixerManagerCB\n\t\titerator FixerIteratorCB\n\t\twantErr  bool\n\t}{\n\t\t{\n\t\t\tname:     \"both arguments are nil\",\n\t\t\tmanager:  nil,\n\t\t\titerator: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"invariant is nil\",\n\t\t\tmanager: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams FixShardActivityParams,\n\t\t\t\tcache cache.DomainCache,\n\t\t\t) invariant.Manager {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\titerator: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"both provided\",\n\t\t\tmanager: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams FixShardActivityParams,\n\t\t\t\tcache cache.DomainCache,\n\t\t\t) invariant.Manager {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\titerator: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tclient blobstore.Client,\n\t\t\t\tkeys store.Keys,\n\t\t\t\tparams FixShardActivityParams,\n\t\t\t) store.ScanOutputIterator {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\t_, err := NewFixerHooks(tc.manager, tc.iterator, func(fixer FixerContext) CustomScannerConfig {\n\t\t\t\treturn nil // no config overrides\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/scanner.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/pborman/uuid\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\n// Scanner is used to scan over given iterator. It is responsible for three things:\n// 1. Checking invariants for each entity.\n// 2. Recording corruption and failures to durable store.\n// 3. Producing a ScanReport\ntype Scanner interface {\n\tScan() ScanReport\n}\n\ntype (\n\t// ShardScanner is a generic scanner which iterates over entities provided by iterator\n\t// implementations of this scanner have to provided invariant manager and iterator\n\tShardScanner struct {\n\t\tshardID          int\n\t\titr              pagination.Iterator\n\t\tfailedWriter     store.ExecutionWriter\n\t\tcorruptedWriter  store.ExecutionWriter\n\t\tinvariantManager invariant.Manager\n\t\tprogressReportFn func()\n\t\tscope            metrics.Scope\n\t\tdomainCache      cache.DomainCache\n\t}\n)\n\n// NewScanner constructs a new ShardScanner\nfunc NewScanner(\n\tshardID int,\n\titerator pagination.Iterator,\n\tblobstoreClient blobstore.Client,\n\tblobstoreFlushThreshold int,\n\tmanager invariant.Manager,\n\tprogressReportFn func(),\n\tscope metrics.Scope,\n\tdomainCache cache.DomainCache,\n) *ShardScanner {\n\tid := uuid.New()\n\n\treturn &ShardScanner{\n\t\tshardID:          shardID,\n\t\titr:              iterator,\n\t\tfailedWriter:     store.NewBlobstoreWriter(id, store.FailedExtension, blobstoreClient, blobstoreFlushThreshold),\n\t\tcorruptedWriter:  store.NewBlobstoreWriter(id, store.CorruptedExtension, blobstoreClient, blobstoreFlushThreshold),\n\t\tinvariantManager: manager,\n\t\tprogressReportFn: progressReportFn,\n\t\tscope:            scope,\n\t\tdomainCache:      domainCache,\n\t}\n}\n\n// Scan scans over all executions in shard and runs invariant checks per execution.\nfunc (s *ShardScanner) Scan(ctx context.Context) ScanReport {\n\tresult := ScanReport{\n\t\tShardID: s.shardID,\n\t\tStats: ScanStats{\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{},\n\t}\n\tfor s.itr.HasNext() {\n\t\ts.progressReportFn()\n\t\texecution, err := s.itr.Next()\n\t\tif err != nil {\n\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\tInfo:        \"persistence iterator returned error\",\n\t\t\t\tInfoDetails: err.Error(),\n\t\t\t}\n\t\t\treturn result\n\t\t}\n\t\tcheckResult := s.invariantManager.RunChecks(ctx, execution)\n\t\tdomainID, err := s.getDomainIDFromEntity(execution)\n\t\tif err != nil {\n\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to get domainID from entity\",\n\t\t\t\tInfoDetails: err.Error(),\n\t\t\t}\n\t\t\treturn result\n\t\t}\n\t\tdomainName, err := s.domainCache.GetDomainName(*domainID)\n\t\tif err != nil {\n\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to get domain name from cache\",\n\t\t\t\tInfoDetails: err.Error(),\n\t\t\t}\n\t\t\treturn result\n\t\t}\n\n\t\tif _, ok := result.DomainStats[*domainID]; !ok {\n\t\t\tresult.DomainStats[*domainID] = &ScanStats{\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t}\n\t\t}\n\t\tresult.DomainStats[*domainID].EntitiesCount++\n\t\tresult.Stats.EntitiesCount++\n\t\tinvariantName := \"\"\n\t\tif checkResult.DeterminingInvariantType != nil {\n\t\t\tinvariantName = string(*checkResult.DeterminingInvariantType)\n\t\t}\n\t\ts.scope.Tagged(\n\t\t\tmetrics.DomainTag(domainName),\n\t\t\tmetrics.InvariantTypeTag(invariantName),\n\t\t\tmetrics.ShardScannerScanResult(string(checkResult.CheckResultType)),\n\t\t).IncCounter(metrics.ShardScannerScan)\n\n\t\tswitch checkResult.CheckResultType {\n\t\tcase invariant.CheckResultTypeHealthy:\n\t\t\t// do nothing if execution is healthy\n\t\tcase invariant.CheckResultTypeCorrupted:\n\t\t\tif err := s.corruptedWriter.Add(store.ScanOutputEntity{\n\t\t\t\tExecution: execution,\n\t\t\t\tResult:    checkResult,\n\t\t\t}); err != nil {\n\t\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\t\tInfo:        \"blobstore add failed for corrupted execution check\",\n\t\t\t\t\tInfoDetails: err.Error(),\n\t\t\t\t}\n\t\t\t\treturn result\n\t\t\t}\n\t\t\tresult.Stats.CorruptedCount++\n\t\t\tresult.Stats.CorruptionByType[*checkResult.DeterminingInvariantType]++\n\t\t\tresult.DomainStats[*domainID].CorruptedCount++\n\t\t\tresult.DomainStats[*domainID].CorruptionByType[*checkResult.DeterminingInvariantType]++\n\t\tcase invariant.CheckResultTypeFailed:\n\t\t\tif err := s.failedWriter.Add(store.ScanOutputEntity{\n\t\t\t\tExecution: execution,\n\t\t\t\tResult:    checkResult,\n\t\t\t}); err != nil {\n\t\t\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\t\t\tInfo:        \"blobstore add failed for failed execution check\",\n\t\t\t\t\tInfoDetails: err.Error(),\n\t\t\t\t}\n\t\t\t\treturn result\n\t\t\t}\n\t\t\tresult.Stats.CheckFailedCount++\n\t\t\tresult.DomainStats[*domainID].CheckFailedCount++\n\t\tdefault:\n\t\t\tpanic(fmt.Sprintf(\"unknown CheckResultType: %v\", checkResult.CheckResultType))\n\t\t}\n\t}\n\n\tif err := s.failedWriter.Flush(); err != nil {\n\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\tInfo:        \"failed to flush for failed execution checks\",\n\t\t\tInfoDetails: err.Error(),\n\t\t}\n\t\treturn result\n\t}\n\tif err := s.corruptedWriter.Flush(); err != nil {\n\t\tresult.Result.ControlFlowFailure = &ControlFlowFailure{\n\t\t\tInfo:        \"failed to flush for corrupted execution checks\",\n\t\t\tInfoDetails: err.Error(),\n\t\t}\n\t\treturn result\n\t}\n\n\tresult.Result.ShardScanKeys = &ScanKeys{\n\t\tCorrupt: s.corruptedWriter.FlushedKeys(),\n\t\tFailed:  s.failedWriter.FlushedKeys(),\n\t}\n\treturn result\n}\n\nfunc (s *ShardScanner) getDomainIDFromEntity(e interface{}) (*string, error) {\n\tconcreteExecution, ok := e.(*entity.ConcreteExecution)\n\tif ok {\n\t\treturn &concreteExecution.DomainID, nil\n\t}\n\n\tcurrentExecution, ok := e.(*entity.CurrentExecution)\n\tif ok {\n\t\treturn &currentExecution.DomainID, nil\n\t}\n\n\ttimer, ok := e.(*entity.Timer)\n\tif ok {\n\t\treturn &timer.DomainID, nil\n\t}\n\n\treturn nil, fmt.Errorf(\"unknown entity type in scanner: %T\", e)\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/scanner_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\ntype ScannerSuite struct {\n\t*require.Assertions\n\tsuite.Suite\n\tcontroller *gomock.Controller\n}\n\nfunc TestScannerSuite(t *testing.T) {\n\tsuite.Run(t, new(ScannerSuite))\n}\n\nfunc (s *ScannerSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n\ts.controller = gomock.NewController(s.T())\n}\n\nfunc (s *ScannerSuite) TearDownTest() {\n\ts.controller.Finish()\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_FirstIteratorError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(nil, errors.New(\"iterator error\")).Times(1)\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"persistence iterator returned error\",\n\t\t\t\tInfoDetails: \"iterator error\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: make(map[string]*ScanStats),\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_NonFirstError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\titeratorCallNumber := 0\n\tmockItr.EXPECT().HasNext().DoAndReturn(func() bool {\n\t\treturn iteratorCallNumber < 5\n\t}).Times(5)\n\tmockItr.EXPECT().Next().DoAndReturn(func() (*entity.ConcreteExecution, error) {\n\t\tdefer func() {\n\t\t\titeratorCallNumber++\n\t\t}()\n\t\tif iteratorCallNumber < 4 {\n\t\t\treturn &entity.ConcreteExecution{Execution: entity.Execution{DomainID: \"ABC\"}}, nil\n\t\t}\n\t\treturn nil, fmt.Errorf(\"iterator got error on: %v\", iteratorCallNumber)\n\t}).Times(5)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunChecks(gomock.Any(), gomock.Any()).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeHealthy,\n\t}).Times(4)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).Times(4)\n\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tinvariantManager: mockInvariantManager,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    4,\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"persistence iterator returned error\",\n\t\t\t\tInfoDetails: \"iterator got error on: 4\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"ABC\": &ScanStats{\n\t\t\t\tEntitiesCount:    4,\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_CorruptedWriterError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(&entity.ConcreteExecution{Execution: entity.Execution{DomainID: \"ABC\"}}, nil).Times(1)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunChecks(gomock.Any(), gomock.Any()).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t}).Times(1)\n\tcorruptedWriter := store.NewMockExecutionWriter(s.controller)\n\tcorruptedWriter.EXPECT().Add(gomock.Any()).Return(errors.New(\"corrupted writer add failed\")).Times(1)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tinvariantManager: mockInvariantManager,\n\t\tcorruptedWriter:  corruptedWriter,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    1,\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore add failed for corrupted execution check\",\n\t\t\t\tInfoDetails: \"corrupted writer add failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"ABC\": &ScanStats{\n\t\t\t\tEntitiesCount:    1,\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_FailedWriterError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(true).Times(1)\n\tmockItr.EXPECT().Next().Return(&entity.ConcreteExecution{Execution: entity.Execution{DomainID: \"ABC\"}}, nil).Times(1)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunChecks(gomock.Any(), gomock.Any()).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeFailed,\n\t}).Times(1)\n\tfailedWriter := store.NewMockExecutionWriter(s.controller)\n\tfailedWriter.EXPECT().Add(gomock.Any()).Return(errors.New(\"failed writer add failed\")).Times(1)\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tinvariantManager: mockInvariantManager,\n\t\tfailedWriter:     failedWriter,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    1,\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"blobstore add failed for failed execution check\",\n\t\t\t\tInfoDetails: \"failed writer add failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"ABC\": &ScanStats{\n\t\t\t\tEntitiesCount:    1,\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_FailedWriterFlushError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(false).Times(1)\n\tfailedWriter := store.NewMockExecutionWriter(s.controller)\n\tfailedWriter.EXPECT().Flush().Return(errors.New(\"failed writer flush failed\")).Times(1)\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tfailedWriter:     failedWriter,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    0,\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to flush for failed execution checks\",\n\t\t\t\tInfoDetails: \"failed writer flush failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: make(map[string]*ScanStats),\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Failure_CorruptedWriterFlushError() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\tmockItr.EXPECT().HasNext().Return(false).Times(1)\n\tcorruptedWriter := store.NewMockExecutionWriter(s.controller)\n\tcorruptedWriter.EXPECT().Flush().Return(errors.New(\"corrupted writer flush failed\")).Times(1)\n\tfailedWriter := store.NewMockExecutionWriter(s.controller)\n\tfailedWriter.EXPECT().Flush().Return(nil).Times(1)\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\titr:              mockItr,\n\t\tcorruptedWriter:  corruptedWriter,\n\t\tfailedWriter:     failedWriter,\n\t\tprogressReportFn: func() {},\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    0,\n\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\tInfo:        \"failed to flush for corrupted execution checks\",\n\t\t\t\tInfoDetails: \"corrupted writer flush failed\",\n\t\t\t},\n\t\t},\n\t\tDomainStats: make(map[string]*ScanStats),\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestScan_Success() {\n\tmockItr := pagination.NewMockIterator(s.controller)\n\titeratorCallNumber := 0\n\tmockItr.EXPECT().HasNext().DoAndReturn(func() bool {\n\t\treturn iteratorCallNumber < 10\n\t}).Times(11)\n\tmockItr.EXPECT().Next().DoAndReturn(func() (*entity.ConcreteExecution, error) {\n\t\tdefer func() {\n\t\t\titeratorCallNumber++\n\t\t}()\n\t\tswitch iteratorCallNumber {\n\t\tcase 0, 1, 2, 3:\n\t\t\treturn &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"healthy\",\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 4, 5, 6:\n\t\t\treturn &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"history_missing\",\n\t\t\t\t\tState:    persistence.WorkflowStateCompleted,\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 7:\n\t\t\treturn &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"orphan_execution\",\n\t\t\t\t\tState:    persistence.WorkflowStateCreated,\n\t\t\t\t},\n\t\t\t}, nil\n\t\tcase 8, 9:\n\t\t\treturn &entity.ConcreteExecution{\n\t\t\t\tExecution: entity.Execution{\n\t\t\t\t\tDomainID: \"failed\",\n\t\t\t\t},\n\t\t\t}, nil\n\t\tdefault:\n\t\t\tpanic(\"should not get here\")\n\t\t}\n\t}).Times(10)\n\tmockInvariantManager := invariant.NewMockManager(s.controller)\n\tmockInvariantManager.EXPECT().RunChecks(context.Background(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"healthy\",\n\t\t},\n\t}).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType: invariant.CheckResultTypeHealthy,\n\t}).Times(4)\n\tmockInvariantManager.EXPECT().RunChecks(context.Background(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"history_missing\",\n\t\t\tState:    persistence.WorkflowStateCompleted,\n\t\t},\n\t}).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType:          invariant.CheckResultTypeCorrupted,\n\t\tDeterminingInvariantType: invariant.NamePtr(invariant.HistoryExists),\n\t\tCheckResults: []invariant.CheckResult{\n\t\t\t{\n\t\t\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t\tInfo:            \"history did not exist\",\n\t\t\t},\n\t\t},\n\t}).Times(3)\n\tmockInvariantManager.EXPECT().RunChecks(context.Background(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"orphan_execution\",\n\t\t\tState:    persistence.WorkflowStateCreated,\n\t\t}}).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType:          invariant.CheckResultTypeCorrupted,\n\t\tDeterminingInvariantType: invariant.NamePtr(invariant.OpenCurrentExecution),\n\t\tCheckResults: []invariant.CheckResult{\n\t\t\t{\n\t\t\t\tCheckResultType: invariant.CheckResultTypeHealthy,\n\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t},\n\t\t\t{\n\t\t\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t\t\t\tInvariantName:   invariant.OpenCurrentExecution,\n\t\t\t\tInfo:            \"execution was orphan\",\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockInvariantManager.EXPECT().RunChecks(context.Background(), &entity.ConcreteExecution{\n\t\tExecution: entity.Execution{\n\t\t\tDomainID: \"failed\",\n\t\t}}).Return(invariant.ManagerCheckResult{\n\t\tCheckResultType:          invariant.CheckResultTypeFailed,\n\t\tDeterminingInvariantType: invariant.NamePtr(invariant.HistoryExists),\n\t\tCheckResults: []invariant.CheckResult{\n\t\t\t{\n\t\t\t\tCheckResultType: invariant.CheckResultTypeFailed,\n\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t\tInfo:            \"failed to check if history exists\",\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\n\tmockCorruptedWriter := store.NewMockExecutionWriter(s.controller)\n\tmockCorruptedWriter.EXPECT().Add(store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"history_missing\",\n\t\t\t\tState:    persistence.WorkflowStateCompleted,\n\t\t\t}},\n\t\tResult: invariant.ManagerCheckResult{\n\t\t\tCheckResultType:          invariant.CheckResultTypeCorrupted,\n\t\t\tDeterminingInvariantType: invariant.NamePtr(invariant.HistoryExists),\n\t\t\tCheckResults: []invariant.CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t\t\tInfo:            \"history did not exist\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(3)\n\tmockCorruptedWriter.EXPECT().Add(store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"orphan_execution\",\n\t\t\t\tState:    persistence.WorkflowStateCreated,\n\t\t\t}},\n\t\tResult: invariant.ManagerCheckResult{\n\t\t\tCheckResultType:          invariant.CheckResultTypeCorrupted,\n\t\t\tDeterminingInvariantType: invariant.NamePtr(invariant.OpenCurrentExecution),\n\t\t\tCheckResults: []invariant.CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: invariant.CheckResultTypeHealthy,\n\t\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: invariant.CheckResultTypeCorrupted,\n\t\t\t\t\tInvariantName:   invariant.OpenCurrentExecution,\n\t\t\t\t\tInfo:            \"execution was orphan\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(1)\n\tmockFailedWriter := store.NewMockExecutionWriter(s.controller)\n\tmockFailedWriter.EXPECT().Add(store.ScanOutputEntity{\n\t\tExecution: &entity.ConcreteExecution{\n\t\t\tExecution: entity.Execution{\n\t\t\t\tDomainID: \"failed\",\n\t\t\t}},\n\t\tResult: invariant.ManagerCheckResult{\n\t\t\tCheckResultType:          invariant.CheckResultTypeFailed,\n\t\t\tDeterminingInvariantType: invariant.NamePtr(invariant.HistoryExists),\n\t\t\tCheckResults: []invariant.CheckResult{\n\t\t\t\t{\n\t\t\t\t\tCheckResultType: invariant.CheckResultTypeFailed,\n\t\t\t\t\tInvariantName:   invariant.HistoryExists,\n\t\t\t\t\tInfo:            \"failed to check if history exists\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}).Times(2)\n\tmockCorruptedWriter.EXPECT().Flush().Return(nil)\n\tmockFailedWriter.EXPECT().Flush().Return(nil)\n\tmockCorruptedWriter.EXPECT().FlushedKeys().Return(&store.Keys{UUID: \"corrupt_keys_uuid\"})\n\tmockFailedWriter.EXPECT().FlushedKeys().Return(&store.Keys{UUID: \"failed_keys_uuid\"})\n\tdomainCache := cache.NewMockDomainCache(s.controller)\n\tdomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test-domain\", nil).AnyTimes()\n\n\tscanner := &ShardScanner{\n\t\tshardID:          0,\n\t\tinvariantManager: mockInvariantManager,\n\t\tcorruptedWriter:  mockCorruptedWriter,\n\t\tfailedWriter:     mockFailedWriter,\n\t\titr:              mockItr,\n\t\tprogressReportFn: func() {},\n\t\tdomainCache:      domainCache,\n\t\tscope:            metrics.NoopScope,\n\t}\n\tresult := scanner.Scan(context.Background())\n\ts.Equal(ScanReport{\n\t\tShardID: 0,\n\t\tStats: ScanStats{\n\t\t\tEntitiesCount:    10,\n\t\t\tCorruptedCount:   4,\n\t\t\tCheckFailedCount: 2,\n\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\tinvariant.HistoryExists:        3,\n\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t},\n\t\t},\n\t\tResult: ScanResult{\n\t\t\tShardScanKeys: &ScanKeys{\n\t\t\t\tCorrupt: &store.Keys{UUID: \"corrupt_keys_uuid\"},\n\t\t\t\tFailed:  &store.Keys{UUID: \"failed_keys_uuid\"},\n\t\t\t},\n\t\t},\n\t\tDomainStats: map[string]*ScanStats{\n\t\t\t\"failed\": &ScanStats{\n\t\t\t\tEntitiesCount:    2,\n\t\t\t\tCheckFailedCount: 2,\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t},\n\t\t\t\"healthy\": &ScanStats{\n\t\t\t\tEntitiesCount:    4,\n\t\t\t\tCorruptionByType: make(map[invariant.Name]int64),\n\t\t\t},\n\t\t\t\"history_missing\": &ScanStats{\n\t\t\t\tEntitiesCount:  3,\n\t\t\t\tCorruptedCount: 3,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.HistoryExists: 3,\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"orphan_execution\": &ScanStats{\n\t\t\t\tEntitiesCount:  1,\n\t\t\t\tCorruptedCount: 1,\n\t\t\t\tCorruptionByType: map[invariant.Name]int64{\n\t\t\t\t\tinvariant.OpenCurrentExecution: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, result)\n}\n\nfunc (s *ScannerSuite) TestGetDomainIDFromEntity() {\n\tscanner := &ShardScanner{}\n\n\tconcreteExecution := entity.ConcreteExecution{Execution: entity.Execution{DomainID: \"A\"}}\n\tconcreteExecutionDomain, err := scanner.getDomainIDFromEntity(&concreteExecution)\n\ts.NoError(err)\n\ts.Equal(\"A\", *concreteExecutionDomain)\n\n\tcurrentExecution := entity.CurrentExecution{Execution: entity.Execution{DomainID: \"B\"}}\n\tcurrentExecutionDomain, err := scanner.getDomainIDFromEntity(&currentExecution)\n\ts.NoError(err)\n\ts.Equal(\"B\", *currentExecutionDomain)\n\n\ttimer := entity.CurrentExecution{Execution: entity.Execution{DomainID: \"C\"}}\n\ttimerDomain, err := scanner.getDomainIDFromEntity(&timer)\n\ts.NoError(err)\n\ts.Equal(\"C\", *timerDomain)\n\n\tbaseExecution := entity.Execution{DomainID: \"D\"}\n\tresult, err := scanner.getDomainIDFromEntity(&baseExecution)\n\ts.Nil(result)\n\ts.Error(err)\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/scanner_workflow.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"errors\"\n\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n)\n\nconst (\n\t// ShardReportQuery is the query name for the query used to get a single shard's report\n\tShardReportQuery = \"shard_report\"\n\t// ShardStatusQuery is the query name for the query used to get the status of all shards\n\tShardStatusQuery = \"shard_status\"\n\t// ShardStatusSummaryQuery is the query name for the query used to get the shard status -> counts map\n\tShardStatusSummaryQuery = \"shard_status_summary\"\n\t// AggregateReportQuery is the query name for the query used to get the aggregate result of all finished shards\n\tAggregateReportQuery = \"aggregate_report\"\n\t// ShardSizeQuery is the query name for the query used to get the number of executions per shard in sorted order\n\tShardSizeQuery = \"shard_size\"\n\t// DomainReportQuery is the query name for the query used to get the reports per domains for all finished shards\n\tDomainReportQuery = \"domain_report\"\n\t// AllResultsQuery returns filenames / paginating data for corruptions and failures in this workflow,\n\t// for shards which have finished processing.  This works for both scanner and fixer, and the return structures\n\t// are very similar.\n\t//\n\t// This data is also available for a single shard under ShardReportQuery, but using that requires\n\t// re-querying repeatedly if more than that single shard's data is desired, e.g. for manual\n\t// troubleshooting purposes.\n\tAllResultsQuery = \"all_results\"\n\n\tscanShardReportChan = \"scanShardReportChan\"\n)\n\n// ManagerCB is a function which returns invariant manager for scanner.\ntype ManagerCB func(\n\tcontext.Context,\n\tpersistence.Retryer,\n\tScanShardActivityParams,\n\tcache.DomainCache,\n) invariant.Manager\n\n// IteratorCB is a function which returns iterator for scanner.\ntype IteratorCB func(\n\tcontext.Context,\n\tpersistence.Retryer,\n\tScanShardActivityParams,\n) pagination.Iterator\n\n// ScannerWorkflow is a workflow which scans and checks entities in a shard.\ntype ScannerWorkflow struct {\n\tAggregator *ShardScanResultAggregator\n\tParams     ScannerWorkflowParams\n\tName       string\n\tShards     []int\n}\n\n// ScannerHooks allows provide manager and iterator for different types of scanners.\ntype ScannerHooks struct {\n\tManager          ManagerCB\n\tIterator         IteratorCB\n\tGetScannerConfig func(scanner ScannerContext) CustomScannerConfig\n}\n\n// NewScannerWorkflow creates instance of shard scanner\nfunc NewScannerWorkflow(\n\tctx workflow.Context,\n\tname string,\n\tparams ScannerWorkflowParams,\n) (*ScannerWorkflow, error) {\n\n\tif name == \"\" {\n\t\treturn nil, errors.New(\"workflow name is not provided\")\n\t}\n\n\tif err := params.Shards.Validate(); err != nil {\n\t\treturn nil, err\n\t}\n\n\tshards, minShard, maxShard := params.Shards.Flatten()\n\twf := ScannerWorkflow{\n\t\tName:       name,\n\t\tParams:     params,\n\t\tAggregator: NewShardScanResultAggregator(shards, minShard, maxShard),\n\t\tShards:     shards,\n\t}\n\n\tfor name, fn := range getScanHandlers(wf.Aggregator) {\n\t\tif err := workflow.SetQueryHandler(ctx, name, fn); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\n\treturn &wf, nil\n}\n\n// Start starts a shard scanner workflow.\nfunc (wf *ScannerWorkflow) Start(ctx workflow.Context) error {\n\tactivityCtx := getShortActivityContext(ctx)\n\n\tvar resolvedConfig ResolvedScannerWorkflowConfig\n\tif err := workflow.ExecuteActivity(activityCtx, ActivityScannerConfig, ScannerConfigActivityParams{\n\t\tOverwrites: wf.Params.ScannerWorkflowConfigOverwrites,\n\t}).Get(ctx, &resolvedConfig); err != nil {\n\t\treturn err\n\t}\n\n\tif !resolvedConfig.GenericScannerConfig.Enabled {\n\t\treturn nil\n\t}\n\n\tshardReportChan := workflow.GetSignalChannel(ctx, scanShardReportChan)\n\tfor i := 0; i < resolvedConfig.GenericScannerConfig.Concurrency; i++ {\n\t\tidx := i\n\t\tworkflow.Go(ctx, func(ctx workflow.Context) {\n\t\t\tbatches := getShardBatches(resolvedConfig.GenericScannerConfig.ActivityBatchSize, resolvedConfig.GenericScannerConfig.Concurrency, wf.Shards, idx)\n\t\t\tfor _, batch := range batches {\n\t\t\t\tactivityCtx = getLongActivityContext(ctx)\n\t\t\t\tvar reports []ScanReport\n\t\t\t\tif err := workflow.ExecuteActivity(activityCtx, ActivityScanShard, ScanShardActivityParams{\n\t\t\t\t\tShards:                  batch,\n\t\t\t\t\tPageSize:                resolvedConfig.GenericScannerConfig.PageSize,\n\t\t\t\t\tBlobstoreFlushThreshold: resolvedConfig.GenericScannerConfig.BlobstoreFlushThreshold,\n\t\t\t\t\tScannerConfig:           resolvedConfig.CustomScannerConfig,\n\t\t\t\t}).Get(ctx, &reports); err != nil {\n\t\t\t\t\terrStr := err.Error()\n\t\t\t\t\tshardReportChan.Send(ctx, ScanReportError{\n\t\t\t\t\t\tReports:  nil,\n\t\t\t\t\t\tErrorStr: &errStr,\n\t\t\t\t\t})\n\t\t\t\t\treturn\n\t\t\t\t}\n\t\t\t\tshardReportChan.Send(ctx, ScanReportError{\n\t\t\t\t\tReports:  reports,\n\t\t\t\t\tErrorStr: nil,\n\t\t\t\t})\n\t\t\t}\n\t\t})\n\t}\n\n\tfor i := 0; i < len(wf.Shards); {\n\t\tvar reportErr ScanReportError\n\n\t\tshardReportChan.Receive(ctx, &reportErr)\n\n\t\tif reportErr.ErrorStr != nil {\n\t\t\treturn errors.New(*reportErr.ErrorStr)\n\t\t}\n\t\tfor _, report := range reportErr.Reports {\n\t\t\twf.Aggregator.AddReport(report)\n\t\t\ti++\n\t\t}\n\t}\n\n\tactivityCtx = getShortActivityContext(ctx)\n\tsummary := wf.Aggregator.GetStatusSummary()\n\n\treturn workflow.ExecuteActivity(activityCtx, ActivityScannerEmitMetrics, ScannerEmitMetricsActivityParams{\n\t\tShardSuccessCount:            summary[ShardStatusSuccess],\n\t\tShardControlFlowFailureCount: summary[ShardStatusControlFlowFailure],\n\t\tAggregateReportResult:        wf.Aggregator.GetAggregateReport(),\n\t\tShardDistributionStats:       wf.Aggregator.GetShardDistributionStats(),\n\t}).Get(ctx, nil)\n\n}\n\nfunc getScanHandlers(aggregator *ShardScanResultAggregator) map[string]interface{} {\n\treturn map[string]interface{}{\n\t\tShardReportQuery: func(shardID int) (*ScanReport, error) {\n\t\t\treturn aggregator.GetReport(shardID)\n\t\t},\n\t\tShardStatusQuery: func(req PaginatedShardQueryRequest) (*ShardStatusQueryResult, error) {\n\t\t\treturn aggregator.GetStatusResult(req)\n\t\t},\n\t\tShardStatusSummaryQuery: func() (ShardStatusSummaryResult, error) {\n\t\t\treturn aggregator.GetStatusSummary(), nil\n\t\t},\n\t\tAggregateReportQuery: func() (AggregateScanReportResult, error) {\n\t\t\treturn aggregator.GetAggregateReport(), nil\n\t\t},\n\t\tShardCorruptKeysQuery: func(req PaginatedShardQueryRequest) (*ShardCorruptKeysQueryResult, error) {\n\t\t\treturn aggregator.GetCorruptionKeys(req)\n\t\t},\n\t\tShardSizeQuery: func(req ShardSizeQueryRequest) (ShardSizeQueryResult, error) {\n\t\t\treturn aggregator.GetShardSizeQueryResult(req)\n\t\t},\n\t\tDomainReportQuery: func(req DomainReportQueryRequest) (*DomainScanReportQueryResult, error) {\n\t\t\treturn aggregator.GetDomainStatus(req)\n\t\t},\n\t\tAllResultsQuery: func() (map[int]ScanResult, error) {\n\t\t\treturn aggregator.GetAllScanResults()\n\t\t},\n\t}\n}\n\nfunc getShardBatches(\n\tbatchSize int,\n\tconcurrency int,\n\tshards []int,\n\tworkerIdx int,\n) [][]int {\n\tbatchIndices := getBatchIndices(batchSize, concurrency, len(shards), workerIdx)\n\tvar result [][]int\n\tfor _, batch := range batchIndices {\n\t\tvar curr []int\n\t\tfor _, i := range batch {\n\t\t\tcurr = append(curr, shards[i])\n\t\t}\n\t\tresult = append(result, curr)\n\t}\n\treturn result\n}\n\n// NewScannerHooks is used to have per scanner iterator and invariant manager\nfunc NewScannerHooks(manager ManagerCB, iterator IteratorCB, config func(scanner ScannerContext) CustomScannerConfig) (*ScannerHooks, error) {\n\tif manager == nil || iterator == nil || config == nil {\n\t\treturn nil, errors.New(\"all scanner hooks args are required\")\n\t}\n\n\treturn &ScannerHooks{\n\t\tManager:          manager,\n\t\tIterator:         iterator,\n\t\tGetScannerConfig: config,\n\t}, nil\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/scanner_workflow_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n)\n\ntype scannerWorkflowsSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestScannerWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(scannerWorkflowsSuite))\n}\n\nfunc (s *scannerWorkflowsSuite) TestGetBatchIndices() {\n\ttestCases := []struct {\n\t\tbatchSize   int\n\t\tconcurrency int\n\t\tsliceLength int\n\t\tworkerIdx   int\n\t\tbatches     [][]int\n\t}{\n\t\t{\n\t\t\tbatchSize:   1,\n\t\t\tconcurrency: 1,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   0,\n\t\t\tbatches:     [][]int{{0}, {1}, {2}, {3}, {4}, {5}, {6}, {7}, {8}, {9}, {10}, {11}, {12}, {13}, {14}, {15}, {16}, {17}, {18}, {19}},\n\t\t},\n\t\t{\n\t\t\tbatchSize:   2,\n\t\t\tconcurrency: 1,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   0,\n\t\t\tbatches:     [][]int{{0, 1}, {2, 3}, {4, 5}, {6, 7}, {8, 9}, {10, 11}, {12, 13}, {14, 15}, {16, 17}, {18, 19}},\n\t\t},\n\t\t{\n\t\t\tbatchSize:   7,\n\t\t\tconcurrency: 1,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   0,\n\t\t\tbatches:     [][]int{{0, 1, 2, 3, 4, 5, 6}, {7, 8, 9, 10, 11, 12, 13}, {14, 15, 16, 17, 18, 19}},\n\t\t},\n\t\t{\n\t\t\tbatchSize:   5,\n\t\t\tconcurrency: 3,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   0,\n\t\t\tbatches:     [][]int{{0, 3, 6, 9, 12}, {15, 18}},\n\t\t},\n\t\t{\n\t\t\tbatchSize:   5,\n\t\t\tconcurrency: 3,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   1,\n\t\t\tbatches:     [][]int{{1, 4, 7, 10, 13}, {16, 19}},\n\t\t},\n\t\t{\n\t\t\tbatchSize:   5,\n\t\t\tconcurrency: 3,\n\t\t\tsliceLength: 20,\n\t\t\tworkerIdx:   2,\n\t\t\tbatches:     [][]int{{2, 5, 8, 11, 14}, {17}},\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Equal(tc.batches, getBatchIndices(tc.batchSize, tc.concurrency, tc.sliceLength, tc.workerIdx))\n\t}\n}\n\nfunc (s *scannerWorkflowsSuite) TestGetShardBatches() {\n\tvar shards []int\n\tfor i := 5; i < 50; i += 2 {\n\t\tshards = append(shards, i)\n\t}\n\tbatches := getShardBatches(5, 3, shards, 1)\n\ts.Equal([][]int{\n\t\t{7, 13, 19, 25, 31},\n\t\t{37, 43, 49},\n\t}, batches)\n}\n\nfunc (s *scannerWorkflowsSuite) TestFlattenShards() {\n\ttestCases := []struct {\n\t\tinput        Shards\n\t\texpectedList []int\n\t\texpectedMin  int\n\t\texpectedMax  int\n\t\tmsg          string\n\t}{\n\t\t{\n\t\t\tinput: Shards{\n\t\t\t\tList: []int{1, 2, 3},\n\t\t\t},\n\t\t\texpectedList: []int{1, 2, 3},\n\t\t\texpectedMin:  1,\n\t\t\texpectedMax:  3,\n\t\t\tmsg:          \"Shard list provided\",\n\t\t},\n\t\t{\n\t\t\tinput: Shards{\n\t\t\t\tRange: &ShardRange{\n\t\t\t\t\tMin: 5,\n\t\t\t\t\tMax: 10,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedList: []int{5, 6, 7, 8, 9},\n\t\t\texpectedMin:  5,\n\t\t\texpectedMax:  9,\n\t\t\tmsg:          \"Shard range provided\",\n\t\t},\n\t\t{\n\t\t\tinput: Shards{\n\t\t\t\tList: []int{1, 90, 2, 3},\n\t\t\t},\n\t\t\texpectedList: []int{1, 90, 2, 3},\n\t\t\texpectedMin:  1,\n\t\t\texpectedMax:  90,\n\t\t\tmsg:          \"Unordered shard list provided\",\n\t\t},\n\t}\n\tfor _, tc := range testCases {\n\t\tshardList, min, max := tc.input.Flatten()\n\t\ts.Equal(tc.expectedList, shardList)\n\t\ts.Equal(tc.expectedMin, min)\n\t\ts.Equal(tc.expectedMax, max)\n\t}\n}\n\nfunc (s *scannerWorkflowsSuite) TestValidateShards() {\n\ttestCases := []struct {\n\t\tshards    Shards\n\t\texpectErr bool\n\t}{\n\t\t{\n\t\t\tshards:    Shards{},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tshards: Shards{\n\t\t\t\tList:  []int{},\n\t\t\t\tRange: &ShardRange{},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tshards: Shards{\n\t\t\t\tList: []int{},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tshards: Shards{\n\t\t\t\tRange: &ShardRange{\n\t\t\t\t\tMin: 0,\n\t\t\t\t\tMax: 0,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: true,\n\t\t},\n\t\t{\n\t\t\tshards: Shards{\n\t\t\t\tRange: &ShardRange{\n\t\t\t\t\tMin: 0,\n\t\t\t\t\tMax: 1,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\terr := tc.shards.Validate()\n\t\tif tc.expectErr {\n\t\t\ts.Error(err)\n\t\t} else {\n\t\t\ts.NoError(err)\n\t\t}\n\t}\n}\n\nfunc (s *fixerWorkflowSuite) TestNewScannerHooks() {\n\ttestCases := []struct {\n\t\tname     string\n\t\tmanager  ManagerCB\n\t\titerator IteratorCB\n\t\twantErr  bool\n\t}{\n\t\t{\n\t\t\tname:     \"both arguments are nil\",\n\t\t\tmanager:  nil,\n\t\t\titerator: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tname:    \"manager is not provided\",\n\t\t\tmanager: nil,\n\t\t\titerator: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams ScanShardActivityParams,\n\t\t\t) pagination.Iterator {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\n\t\t{\n\t\t\tname: \"iterator is not provided\",\n\t\t\tmanager: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams ScanShardActivityParams,\n\t\t\t\tcache cache.DomainCache,\n\t\t\t) invariant.Manager {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\titerator: nil,\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tname: \"both provided\",\n\t\t\tmanager: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams ScanShardActivityParams,\n\t\t\t\tcache cache.DomainCache,\n\t\t\t) invariant.Manager {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\titerator: func(\n\t\t\t\tctx context.Context,\n\t\t\t\tretryer persistence.Retryer,\n\t\t\t\tparams ScanShardActivityParams,\n\t\t\t) pagination.Iterator {\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\twantErr: false,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\ts.Run(tc.name, func() {\n\t\t\t_, err := NewScannerHooks(tc.manager, tc.iterator, func(scanner ScannerContext) CustomScannerConfig {\n\t\t\t\treturn nil // no config overrides\n\t\t\t})\n\t\t\tif tc.wantErr {\n\t\t\t\ts.Error(err)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/types.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nconst (\n\t// ErrScanWorkflowNotClosed indicates fix was attempted on scan workflow which was not finished\n\tErrScanWorkflowNotClosed = \"scan workflow is not closed, only can run fix on output of finished scan workflow\"\n\t// ErrSerialization indicates a serialization or deserialization error occurred\n\tErrSerialization = \"encountered serialization error\"\n\t// ErrMissingHooks indicates scanner is not providing hooks to Invariant manager or Iterator\n\tErrMissingHooks = \"hooks are not provided for this scanner\"\n)\n\ntype (\n\tcontextKey string\n\n\t// ScannerContext is the resource that is available in activities under ShardScanner context key\n\tScannerContext struct {\n\t\tResource resource.Resource\n\t\tHooks    *ScannerHooks\n\t\tScope    metrics.Scope\n\t\tConfig   *ScannerConfig\n\t\tLogger   log.Logger\n\t}\n\n\t// FixerContext is the resource that is available to activities under ShardFixer key\n\tFixerContext struct {\n\t\tResource resource.Resource\n\t\tHooks    *FixerHooks\n\t\tScope    metrics.Scope\n\t\tConfig   *ScannerConfig\n\t\tLogger   log.Logger\n\t}\n\n\t// ScannerEmitMetricsActivityParams is the parameter for scannerEmitMetricsActivity\n\tScannerEmitMetricsActivityParams struct {\n\t\tShardSuccessCount            int\n\t\tShardControlFlowFailureCount int\n\t\tAggregateReportResult        AggregateScanReportResult\n\t\tShardDistributionStats       ShardDistributionStats\n\t}\n\n\t// ShardRange identifies a set of shards based on min (inclusive) and max (exclusive)\n\tShardRange struct {\n\t\tMin int\n\t\tMax int\n\t}\n\n\t// Shards identify the shards that should be scanned.\n\t// Exactly one of List or Range should be non-nil.\n\tShards struct {\n\t\tList  []int\n\t\tRange *ShardRange\n\t}\n\n\t// ScannerWorkflowParams are the parameters to the scan workflow\n\tScannerWorkflowParams struct {\n\t\tShards                          Shards\n\t\tScannerWorkflowConfigOverwrites ScannerWorkflowConfigOverwrites\n\t}\n\n\t// ScannerConfigActivityParams is the parameter for scannerConfigActivity\n\tScannerConfigActivityParams struct {\n\t\tOverwrites ScannerWorkflowConfigOverwrites\n\t}\n\n\t// ScanShardActivityParams is the parameter for scanShardActivity\n\tScanShardActivityParams struct {\n\t\tShards                  []int\n\t\tPageSize                int\n\t\tBlobstoreFlushThreshold int\n\t\tScannerConfig           CustomScannerConfig\n\t}\n\n\t// FixerWorkflowParams are the parameters to the fix workflow\n\tFixerWorkflowParams struct {\n\t\tScannerWorkflowWorkflowID     string\n\t\tScannerWorkflowRunID          string\n\t\tFixerWorkflowConfigOverwrites FixerWorkflowConfigOverwrites\n\t}\n\n\t// ScanReport is the report of running Scan on a single shard.\n\tScanReport struct {\n\t\tShardID     int\n\t\tStats       ScanStats\n\t\tResult      ScanResult\n\t\tDomainStats map[string]*ScanStats\n\t}\n\n\t// DomainStats is the report of stats for one domain\n\tDomainScanStats struct {\n\t\tDomainID string\n\t\tStats    ScanStats\n\t}\n\n\t// DomainStats is the report of stats for one domain\n\tDomainFixStats struct {\n\t\tDomainID string\n\t\tStats    FixStats\n\t}\n\n\t// ScanStats indicates the stats of entities which were handled by shard Scan.\n\tScanStats struct {\n\t\tEntitiesCount    int64\n\t\tCorruptedCount   int64\n\t\tCheckFailedCount int64\n\t\tCorruptionByType map[invariant.Name]int64\n\t}\n\n\t// ScanResult indicates the result of running scan on a shard.\n\t// Exactly one of ControlFlowFailure or ScanKeys will be non-nil\n\tScanResult struct {\n\t\tShardScanKeys      *ScanKeys\n\t\tControlFlowFailure *ControlFlowFailure\n\t}\n\n\t// ScanKeys are the keys to the blobs that were uploaded during scan.\n\t// Keys can be nil if there were no uploads.\n\tScanKeys struct {\n\t\tCorrupt *store.Keys\n\t\tFailed  *store.Keys\n\t}\n\n\t// FixReport is the report of running Fix on a single shard\n\tFixReport struct {\n\t\tShardID     int\n\t\tStats       FixStats\n\t\tResult      FixResult\n\t\tDomainStats map[string]*FixStats\n\t}\n\n\t// FixStats indicates the stats of executions that were handled by shard Fix.\n\tFixStats struct {\n\t\tEntitiesCount int64\n\t\tFixedCount    int64\n\t\tSkippedCount  int64\n\t\tFailedCount   int64\n\t}\n\n\t// FixResult indicates the result of running fix on a shard.\n\t// Exactly one of ControlFlowFailure or FixKeys will be non-nil.\n\tFixResult struct {\n\t\tShardFixKeys       *FixKeys\n\t\tControlFlowFailure *ControlFlowFailure\n\t}\n\n\t// FixKeys are the keys to the blobs that were uploaded during fix.\n\t// Keys can be nil if there were no uploads.\n\tFixKeys struct {\n\t\tSkipped *store.Keys\n\t\tFailed  *store.Keys\n\t\tFixed   *store.Keys\n\t}\n\n\t// ControlFlowFailure indicates an error occurred which makes it impossible to\n\t// even attempt to check or fix one or more execution(s). Note that it is not a ControlFlowFailure\n\t// if a check or fix fails, it is only a ControlFlowFailure if\n\t// an error is encountered which makes even attempting to check or fix impossible.\n\tControlFlowFailure struct {\n\t\tInfo        string\n\t\tInfoDetails string\n\t}\n\n\t// FixerCorruptedKeysActivityParams is the parameter for fixerCorruptedKeysActivity\n\tFixerCorruptedKeysActivityParams struct {\n\t\tScannerWorkflowWorkflowID string\n\t\tScannerWorkflowRunID      string\n\t\tStartingShardID           *int\n\t}\n\n\t// FixShardActivityParams is the parameter for fixShardActivity\n\tFixShardActivityParams struct {\n\t\tCorruptedKeysEntries        []CorruptedKeysEntry\n\t\tResolvedFixerWorkflowConfig ResolvedFixerWorkflowConfig\n\n\t\t// EnabledInvariants contains all known invariants for fixer, with \"true\" or \"false\" values.\n\t\t// In current code it should never be empty.\n\t\t//\n\t\t// If empty, EnabledInvariants came from old serialized data prior to this field existing,\n\t\t// and the historical list of invariants should be used.  This should be a one-time event\n\t\t// after upgrading.\n\t\tEnabledInvariants CustomScannerConfig\n\t}\n\n\t// CustomScannerConfig is used to pass key/value parameters between shardscanner activity and scanner/fixer implementations.\n\t// this is used to have activities with better consistency, as the workflow records one config and uses it for all shards,\n\t// even after config / code changes.\n\t//\n\t// It is currently only used to pass invariant names to control whether they are enabled or not.\n\t// Please do not use this for other purposes.\n\tCustomScannerConfig map[string]string\n\n\t// GenericScannerConfig is a generic params for all shard scanners\n\tGenericScannerConfig struct {\n\t\tEnabled                 bool\n\t\tConcurrency             int\n\t\tPageSize                int\n\t\tBlobstoreFlushThreshold int\n\t\tActivityBatchSize       int\n\t}\n\n\t// GenericScannerConfigOverwrites allows to override generic params\n\tGenericScannerConfigOverwrites struct {\n\t\tEnabled                 *bool\n\t\tConcurrency             *int\n\t\tPageSize                *int\n\t\tBlobstoreFlushThreshold *int\n\t\tActivityBatchSize       *int\n\t}\n\n\t// ResolvedScannerWorkflowConfig is the resolved config after reading dynamic config\n\t// and applying overwrites.\n\tResolvedScannerWorkflowConfig struct {\n\t\tGenericScannerConfig GenericScannerConfig\n\t\tCustomScannerConfig  CustomScannerConfig\n\t}\n\n\t// ScannerWorkflowConfigOverwrites enables overwriting the values in dynamic config.\n\t// If provided workflow will favor overwrites over dynamic config.\n\t// Any overwrites that are left as nil will fall back to using dynamic config.\n\tScannerWorkflowConfigOverwrites struct {\n\t\tGenericScannerConfig GenericScannerConfigOverwrites\n\t\tCustomScannerConfig  *CustomScannerConfig\n\t}\n\n\t// DynamicParams is the dynamic config for scanner workflow.\n\tDynamicParams struct {\n\t\tScannerEnabled          dynamicproperties.BoolPropertyFn\n\t\tFixerEnabled            dynamicproperties.BoolPropertyFn\n\t\tConcurrency             dynamicproperties.IntPropertyFn\n\t\tPageSize                dynamicproperties.IntPropertyFn\n\t\tBlobstoreFlushThreshold dynamicproperties.IntPropertyFn\n\t\tActivityBatchSize       dynamicproperties.IntPropertyFn\n\t\tAllowDomain             dynamicproperties.BoolPropertyFnWithDomainFilter\n\t}\n\n\t// ScannerConfig is the config for ShardScanner workflow\n\tScannerConfig struct {\n\t\tScannerWFTypeName    string\n\t\tFixerWFTypeName      string\n\t\tScannerHooks         func() *ScannerHooks\n\t\tFixerHooks           func() *FixerHooks\n\t\tDynamicParams        DynamicParams\n\t\tDynamicCollection    *dynamicconfig.Collection\n\t\tStartWorkflowOptions client.StartWorkflowOptions\n\t\tStartFixerOptions    client.StartWorkflowOptions\n\t}\n\n\t// FixerWorkflowConfigOverwrites enables overwriting the default values.\n\t// If provided workflow will favor overwrites over defaults.\n\t// Any overwrites that are left as nil will fall back to defaults.\n\tFixerWorkflowConfigOverwrites struct {\n\t\tConcurrency             *int\n\t\tBlobstoreFlushThreshold *int\n\t\tActivityBatchSize       *int\n\t}\n\n\t// ResolvedFixerWorkflowConfig is the resolved config after reading defaults and applying overwrites.\n\tResolvedFixerWorkflowConfig struct {\n\t\tConcurrency             int\n\t\tBlobstoreFlushThreshold int\n\t\tActivityBatchSize       int\n\t}\n)\n\nfunc getShortActivityContext(ctx workflow.Context) workflow.Context {\n\tactivityOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    5 * time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       1.7,\n\t\t\tExpirationInterval:       10 * time.Minute,\n\t\t\tNonRetriableErrorReasons: []string{ErrScanWorkflowNotClosed, ErrSerialization, ErrMissingHooks},\n\t\t},\n\t}\n\treturn workflow.WithActivityOptions(ctx, activityOptions)\n}\n\nfunc getLongActivityContext(ctx workflow.Context) workflow.Context {\n\tactivityOptions := workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: time.Minute,\n\t\tStartToCloseTimeout:    48 * time.Hour,\n\t\tHeartbeatTimeout:       time.Minute,\n\t\tRetryPolicy: &cadence.RetryPolicy{\n\t\t\tInitialInterval:          time.Second,\n\t\t\tBackoffCoefficient:       1.7,\n\t\t\tExpirationInterval:       48 * time.Hour,\n\t\t\tNonRetriableErrorReasons: []string{ErrScanWorkflowNotClosed, ErrSerialization, ErrMissingHooks},\n\t\t},\n\t}\n\treturn workflow.WithActivityOptions(ctx, activityOptions)\n}\n\n// GetCorruptedKeys is a workflow which is used to retrieve keys for fixer workflow.\nfunc GetCorruptedKeys(\n\tctx workflow.Context,\n\tparams FixerWorkflowParams,\n) (*FixerCorruptedKeysActivityResult, error) {\n\tfixerCorruptedKeysActivityParams := FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: params.ScannerWorkflowWorkflowID,\n\t\tScannerWorkflowRunID:      params.ScannerWorkflowRunID,\n\t\tStartingShardID:           nil,\n\t}\n\tvar minShardID *int\n\tvar maxShardID *int\n\tvar corruptKeys []CorruptedKeysEntry\n\tisFirst := true\n\tfor isFirst || fixerCorruptedKeysActivityParams.StartingShardID != nil {\n\t\tisFirst = false\n\t\tcorruptedKeysResult := &FixerCorruptedKeysActivityResult{}\n\t\tactivityCtx := getShortActivityContext(ctx)\n\t\tif err := workflow.ExecuteActivity(\n\t\t\tactivityCtx,\n\t\t\tActivityFixerCorruptedKeys,\n\t\t\tfixerCorruptedKeysActivityParams).Get(ctx, &corruptedKeysResult); err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tfixerCorruptedKeysActivityParams.StartingShardID = corruptedKeysResult.ShardQueryPaginationToken.NextShardID\n\t\tif len(corruptedKeysResult.CorruptedKeys) == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tcorruptKeys = append(corruptKeys, corruptedKeysResult.CorruptedKeys...)\n\t\tif corruptedKeysResult.MinShard != nil && (minShardID == nil || *minShardID > *corruptedKeysResult.MinShard) {\n\t\t\tminShardID = corruptedKeysResult.MinShard\n\t\t}\n\t\tif corruptedKeysResult.MaxShard != nil && (maxShardID == nil || *maxShardID < *corruptedKeysResult.MaxShard) {\n\t\t\tmaxShardID = corruptedKeysResult.MaxShard\n\t\t}\n\t}\n\treturn &FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: corruptKeys,\n\t\tMinShard:      minShardID,\n\t\tMaxShard:      maxShardID,\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: nil,\n\t\t\tIsDone:      true,\n\t\t},\n\t}, nil\n}\n\nfunc getBatchIndices(\n\tbatchSize int,\n\tconcurrency int,\n\tsliceLength int,\n\tworkerIdx int,\n) [][]int {\n\tvar batches [][]int\n\tvar currBatch []int\n\tfor i := 0; i < sliceLength; i++ {\n\t\tif i%concurrency == workerIdx {\n\t\t\tcurrBatch = append(currBatch, i)\n\t\t\tif len(currBatch) == batchSize {\n\t\t\t\tbatches = append(batches, currBatch)\n\t\t\t\tcurrBatch = nil\n\t\t\t}\n\t\t}\n\t}\n\tif len(currBatch) > 0 {\n\t\tbatches = append(batches, currBatch)\n\t}\n\treturn batches\n}\n\n// Validate validates shard list or range\nfunc (s Shards) Validate() error {\n\tif s.List == nil && s.Range == nil {\n\t\treturn errors.New(\"must provide either List or Range\")\n\t}\n\tif s.List != nil && s.Range != nil {\n\t\treturn errors.New(\"only one of List or Range can be provided\")\n\t}\n\tif s.List != nil && len(s.List) == 0 {\n\t\treturn errors.New(\"empty List provided\")\n\t}\n\tif s.Range != nil && s.Range.Max <= s.Range.Min {\n\t\treturn errors.New(\"empty Range provided\")\n\t}\n\treturn nil\n}\n\n// Flatten flattens Shards to a list of shard IDs and finds the min/max shardID\nfunc (s Shards) Flatten() ([]int, int, int) {\n\tshardList := s.List\n\tif len(shardList) == 0 {\n\t\tshardList = []int{}\n\t\tfor i := s.Range.Min; i < s.Range.Max; i++ {\n\t\t\tshardList = append(shardList, i)\n\t\t}\n\t}\n\tmin := shardList[0]\n\tmax := shardList[0]\n\tfor i := 1; i < len(shardList); i++ {\n\t\tif shardList[i] < min {\n\t\t\tmin = shardList[i]\n\t\t}\n\t\tif shardList[i] > max {\n\t\t\tmax = shardList[i]\n\t\t}\n\t}\n\n\treturn shardList, min, max\n}\n\n// NewShardScannerContext sets scanner context up\nfunc NewShardScannerContext(\n\tres resource.Resource,\n\tconfig *ScannerConfig,\n) ScannerContext {\n\treturn ScannerContext{\n\t\tResource: res,\n\t\tScope:    res.GetMetricsClient().Scope(metrics.ExecutionsScannerScope),\n\t\tConfig:   config,\n\t\tLogger:   res.GetLogger().WithTags(tag.ComponentShardScanner),\n\t\tHooks:    config.ScannerHooks(),\n\t}\n}\n\n// NewShardFixerContext sets fixer context up\nfunc NewShardFixerContext(\n\tres resource.Resource,\n\tconfig *ScannerConfig,\n) FixerContext {\n\treturn FixerContext{\n\t\tResource: res,\n\t\tScope:    res.GetMetricsClient().Scope(metrics.ExecutionsFixerScope),\n\t\tConfig:   config,\n\t\tHooks:    config.FixerHooks(),\n\t\tLogger:   res.GetLogger().WithTags(tag.ComponentShardFixer),\n\t}\n}\n\n// NewContext provides context to be used as background activity context\nfunc NewFixerContext(\n\tctx context.Context,\n\tworkflowName string,\n\tfixerContext FixerContext,\n) context.Context {\n\treturn context.WithValue(ctx, contextKey(workflowName), fixerContext)\n}\n\n// NewContext provides context to be used as background activity context\nfunc NewScannerContext(\n\tctx context.Context,\n\tworkflowName string,\n\tscannerContext ScannerContext,\n) context.Context {\n\treturn context.WithValue(ctx, contextKey(workflowName), scannerContext)\n}\n\n// GetScannerContext extracts scanner context from activity context\nfunc GetScannerContext(\n\tctx context.Context,\n) (ScannerContext, error) {\n\tinfo := activity.GetInfo(ctx)\n\tif info.WorkflowType == nil {\n\t\treturn ScannerContext{}, fmt.Errorf(\"workflowType is nil\")\n\t}\n\tval, ok := ctx.Value(contextKey(info.WorkflowType.Name)).(ScannerContext)\n\tif !ok {\n\t\treturn ScannerContext{}, fmt.Errorf(\"context type is not %T for a key %q\", val, info.WorkflowType.Name)\n\t}\n\treturn val, nil\n}\n\n// GetFixerContext extracts fixer context from activity context\n// it uses typed, private key to reduce access scope\nfunc GetFixerContext(\n\tctx context.Context,\n) (FixerContext, error) {\n\tinfo := activity.GetInfo(ctx)\n\tif info.WorkflowType == nil {\n\t\treturn FixerContext{}, fmt.Errorf(\"workflowType is nil\")\n\t}\n\tval, ok := ctx.Value(contextKey(info.WorkflowType.Name)).(FixerContext)\n\tif !ok {\n\t\treturn FixerContext{}, fmt.Errorf(\"context type is not %T for a key %q\", val, info.WorkflowType.Name)\n\t}\n\treturn val, nil\n}\n\n// Empty returns true if this ScanResult has no \"real\" data, e.g. only nils or empty values.\nfunc (s *ScanResult) Empty() bool {\n\tif s == nil {\n\t\treturn true\n\t}\n\tif s.ControlFlowFailure != nil && (*s.ControlFlowFailure != ControlFlowFailure{}) {\n\t\treturn false // at least control flow failure has data\n\t}\n\tif s.ShardScanKeys != nil {\n\t\tif s.ShardScanKeys.Corrupt != nil && (*s.ShardScanKeys.Corrupt != store.Keys{}) {\n\t\t\treturn false // corrupt data exists\n\t\t}\n\t\tif s.ShardScanKeys.Failed != nil && (*s.ShardScanKeys.Failed != store.Keys{}) {\n\t\t\treturn false // failed data exists\n\t\t}\n\t}\n\treturn true // both empty\n}\n\n// Empty returns true if this FixResult has no \"real\" data, e.g. only nils or empty values.\nfunc (f *FixResult) Empty() bool {\n\tif f == nil {\n\t\treturn true\n\t}\n\tif f.ControlFlowFailure != nil && (*f.ControlFlowFailure != ControlFlowFailure{}) {\n\t\treturn false // at least control flow failure has data\n\t}\n\tif f.ShardFixKeys != nil {\n\t\tif f.ShardFixKeys.Fixed != nil && (*f.ShardFixKeys.Fixed != store.Keys{}) {\n\t\t\treturn false // fixed data exists\n\t\t}\n\t\tif f.ShardFixKeys.Failed != nil && (*f.ShardFixKeys.Failed != store.Keys{}) {\n\t\t\treturn false // failed data exists\n\t\t}\n\t\tif f.ShardFixKeys.Skipped != nil && (*f.ShardFixKeys.Skipped != store.Keys{}) {\n\t\t\treturn false // skipped data exists\n\t\t}\n\t}\n\treturn true // both empty\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/types_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"context\"\n\t\"testing\"\n)\n\nfunc TestGetFixerContextPanicsWhenNonActivityContextProvided(t *testing.T) {\n\tdefer func() { recover() }()\n\tGetFixerContext(context.Background())\n\tt.Errorf(\"should have panicked\")\n}\n\nfunc TestGetScannerContextPanicsWhenNonActivityContextProvided(t *testing.T) {\n\tdefer func() { recover() }()\n\tGetScannerContext(context.Background())\n\tt.Errorf(\"should have panicked\")\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/workflows.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport \"go.uber.org/cadence/activity\"\n\nfunc init() {\n\t// shardscanner activities\n\tactivity.RegisterWithOptions(scannerEmitMetricsActivity, activity.RegisterOptions{Name: ActivityScannerEmitMetrics})\n\tactivity.RegisterWithOptions(scanShardActivity, activity.RegisterOptions{Name: ActivityScanShard})\n\tactivity.RegisterWithOptions(scannerConfigActivity, activity.RegisterOptions{Name: ActivityScannerConfig})\n\tactivity.RegisterWithOptions(fixerConfigActivity, activity.RegisterOptions{Name: ActivityFixerConfig})\n\tactivity.RegisterWithOptions(fixerCorruptedKeysActivity, activity.RegisterOptions{Name: ActivityFixerCorruptedKeys})\n\tactivity.RegisterWithOptions(fixShardActivity, activity.RegisterOptions{Name: ActivityFixShard})\n}\n"
  },
  {
    "path": "service/worker/scanner/shardscanner/workflows_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage shardscanner\n\nimport (\n\t\"errors\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n)\n\ntype workflowsSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tenv *testsuite.TestWorkflowEnvironment\n}\n\nfunc TestWorkflowsSuite(t *testing.T) {\n\tsuite.Run(t, new(workflowsSuite))\n}\n\nfunc (s *workflowsSuite) SetupTest() {\n\ts.env = s.WorkflowTestSuite.NewTestWorkflowEnvironment()\n\ts.env.RegisterWorkflow(NewTestWorkflow)\n\ts.env.RegisterWorkflow(NewTestFixerWorkflow)\n\ts.env.RegisterWorkflow(GetCorruptedKeys)\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Failure_ScanShard() {\n\ts.env.OnActivity(ActivityScannerConfig, mock.Anything, mock.Anything).Return(ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\tEnabled:           true,\n\t\t\tConcurrency:       3,\n\t\t\tActivityBatchSize: 5,\n\t\t},\n\t}, nil)\n\tshards := Shards{\n\t\tRange: &ShardRange{\n\t\t\tMin: 0,\n\t\t\tMax: 30,\n\t\t},\n\t}\n\n\tbatches := [][]int{\n\t\t{0, 3, 6, 9, 12},\n\t\t{15, 18, 21, 24, 27},\n\t\t{1, 4, 7, 10, 13},\n\t\t{16, 19, 22, 25, 28},\n\t\t{2, 5, 8, 11, 14},\n\t\t{17, 20, 23, 26, 29},\n\t}\n\n\tfor i, batch := range batches {\n\n\t\tvar reports []ScanReport\n\t\tvar err error\n\t\tif i == len(batches)-1 {\n\t\t\treports = nil\n\t\t\terr = errors.New(\"scan shard activity got error\")\n\t\t} else {\n\t\t\terr = nil\n\t\t\tfor _, s := range batch {\n\t\t\t\treports = append(reports, ScanReport{\n\t\t\t\t\tShardID: s,\n\t\t\t\t\tStats: ScanStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t},\n\t\t\t\t\tResult: ScanResult{\n\t\t\t\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\ts.env.OnActivity(ActivityScanShard, mock.Anything, ScanShardActivityParams{\n\t\t\tShards: batch,\n\t\t}).Return(reports, err)\n\t}\n\ts.env.ExecuteWorkflow(NewTestWorkflow, \"test-workflow\", ScannerWorkflowParams{\n\t\tShards: shards,\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Equal(\"scan shard activity got error\", s.env.GetWorkflowError().Error())\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Failure_ScannerConfigActivity() {\n\ts.env.OnActivity(ActivityScannerConfig, mock.Anything, mock.Anything).Return(ResolvedScannerWorkflowConfig{}, errors.New(\"got error getting config\"))\n\ts.env.ExecuteWorkflow(NewTestWorkflow, \"test-workflow\", ScannerWorkflowParams{\n\t\tShards: Shards{\n\t\t\tList: []int{1, 2, 3},\n\t\t},\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Equal(\"got error getting config\", s.env.GetWorkflowError().Error())\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Requires_Name() {\n\ts.env.OnActivity(ActivityScannerConfig, mock.Anything, mock.Anything).Return(ResolvedScannerWorkflowConfig{}, errors.New(\"got error getting config\"))\n\ts.env.ExecuteWorkflow(NewTestWorkflow, \"\", ScannerWorkflowParams{\n\t\tShards: Shards{\n\t\t\tList: []int{1, 2, 3},\n\t\t},\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Equal(\"workflow name is not provided\", s.env.GetWorkflowError().Error())\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Requires_Valid_ShardConfig() {\n\ts.env.OnActivity(ActivityScannerConfig, mock.Anything, mock.Anything).Return(ResolvedScannerWorkflowConfig{}, errors.New(\"got error getting config\"))\n\ts.env.ExecuteWorkflow(NewTestWorkflow, \"test-workflow\", ScannerWorkflowParams{})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Equal(\"must provide either List or Range\", s.env.GetWorkflowError().Error())\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Success_Disabled() {\n\ts.env.OnActivity(ActivityScannerConfig, mock.Anything, mock.Anything).Return(ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: GenericScannerConfig{\n\t\t\tEnabled: false,\n\t\t},\n\t}, nil)\n\n\ts.env.ExecuteWorkflow(NewTestWorkflow, \"test-workflow\", ScannerWorkflowParams{\n\t\tShards: Shards{\n\t\t\tList: []int{1, 2, 3},\n\t\t},\n\t})\n\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowsSuite) TestFixerWorkflow_Success() {\n\tcorruptedKeys := make([]CorruptedKeysEntry, 30)\n\tfor i := 0; i < 30; i++ {\n\t\tcorruptedKeys[i] = CorruptedKeysEntry{\n\t\t\tShardID: i,\n\t\t}\n\t}\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, mock.Anything).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: corruptedKeys,\n\t\tMinShard:      common.IntPtr(0),\n\t\tMaxShard:      common.IntPtr(29),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tIsDone:      true,\n\t\t\tNextShardID: nil,\n\t\t},\n\t}, nil)\n\n\tenabledFixInvariants := CustomScannerConfig{\n\t\t// historically enabled by default\n\t\tinvariant.CollectionHistory.String():      \"true\",\n\t\tinvariant.CollectionMutableState.String(): \"true\",\n\t\t// disabled by default\n\t\tinvariant.CollectionStale.String(): \"false\",\n\t}\n\ts.env.OnActivity(ActivityFixerConfig, mock.Anything, FixShardConfigParams{ /* no contents currently */ }).Return(&FixShardConfigResults{\n\t\tEnabledInvariants: enabledFixInvariants,\n\t}, nil)\n\n\tfixerWorkflowConfigOverwrites := FixerWorkflowConfigOverwrites{\n\t\tConcurrency:             common.IntPtr(3),\n\t\tBlobstoreFlushThreshold: common.IntPtr(1000),\n\t\tActivityBatchSize:       common.IntPtr(5),\n\t}\n\tresolvedFixerWorkflowConfig := ResolvedFixerWorkflowConfig{\n\t\tConcurrency:             3,\n\t\tActivityBatchSize:       5,\n\t\tBlobstoreFlushThreshold: 1000,\n\t}\n\tbatches := [][]int{\n\t\t{0, 3, 6, 9, 12},\n\t\t{15, 18, 21, 24, 27},\n\t\t{1, 4, 7, 10, 13},\n\t\t{16, 19, 22, 25, 28},\n\t\t{2, 5, 8, 11, 14},\n\t\t{17, 20, 23, 26, 29},\n\t}\n\n\tfor _, batch := range batches {\n\t\tvar corruptedKeys []CorruptedKeysEntry\n\t\tfor _, shard := range batch {\n\t\t\tcorruptedKeys = append(corruptedKeys, CorruptedKeysEntry{\n\t\t\t\tShardID: shard,\n\t\t\t})\n\t\t}\n\t\tvar reports []FixReport\n\t\tfor i, s := range batch {\n\t\t\tif i == 0 {\n\t\t\t\treports = append(reports, FixReport{\n\t\t\t\t\tShardID: s,\n\t\t\t\t\tStats: FixStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t},\n\t\t\t\t\tResult: FixResult{\n\t\t\t\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t} else {\n\t\t\t\treports = append(reports, FixReport{\n\t\t\t\t\tShardID: s,\n\t\t\t\t\tStats: FixStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t\tFixedCount:    2,\n\t\t\t\t\t\tSkippedCount:  1,\n\t\t\t\t\t\tFailedCount:   1,\n\t\t\t\t\t},\n\t\t\t\t\tResult: FixResult{\n\t\t\t\t\t\tShardFixKeys: &FixKeys{\n\t\t\t\t\t\t\tSkipped: &store.Keys{\n\t\t\t\t\t\t\t\tUUID: \"skipped_keys\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFailed: &store.Keys{\n\t\t\t\t\t\t\t\tUUID: \"failed_keys\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tFixed: &store.Keys{\n\t\t\t\t\t\t\t\tUUID: \"fixed_keys\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\ts.env.OnActivity(ActivityFixShard, mock.Anything, FixShardActivityParams{\n\t\t\tCorruptedKeysEntries:        corruptedKeys,\n\t\t\tResolvedFixerWorkflowConfig: resolvedFixerWorkflowConfig,\n\t\t\tEnabledInvariants:           enabledFixInvariants,\n\t\t}).Return(reports, nil)\n\t}\n\n\ts.env.ExecuteWorkflow(NewTestFixerWorkflow, FixerWorkflowParams{\n\t\tScannerWorkflowWorkflowID:     \"test_wid\",\n\t\tScannerWorkflowRunID:          \"test_rid\",\n\t\tFixerWorkflowConfigOverwrites: fixerWorkflowConfigOverwrites,\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n\n\taggValue, err := s.env.QueryWorkflow(AggregateReportQuery)\n\ts.NoError(err)\n\tvar agg AggregateFixReportResult\n\ts.NoError(aggValue.Get(&agg))\n\ts.Equal(AggregateFixReportResult{\n\t\tEntitiesCount: 240,\n\t\tFixedCount:    48,\n\t\tFailedCount:   24,\n\t\tSkippedCount:  24,\n\t}, agg)\n\n\tfor i := 0; i < 30; i++ {\n\t\tshardReportValue, err := s.env.QueryWorkflow(ShardReportQuery, i)\n\t\ts.NoError(err)\n\t\tvar shardReport *FixReport\n\t\ts.NoError(shardReportValue.Get(&shardReport))\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\ts.Equal(&FixReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: FixStats{\n\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t},\n\t\t\t\tResult: FixResult{\n\t\t\t\t\tControlFlowFailure: &ControlFlowFailure{\n\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t} else {\n\t\t\ts.Equal(&FixReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: FixStats{\n\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\tFixedCount:    2,\n\t\t\t\t\tFailedCount:   1,\n\t\t\t\t\tSkippedCount:  1,\n\t\t\t\t},\n\t\t\t\tResult: FixResult{\n\t\t\t\t\tShardFixKeys: &FixKeys{\n\t\t\t\t\t\tSkipped: &store.Keys{\n\t\t\t\t\t\t\tUUID: \"skipped_keys\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tFailed: &store.Keys{\n\t\t\t\t\t\t\tUUID: \"failed_keys\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tFixed: &store.Keys{\n\t\t\t\t\t\t\tUUID: \"fixed_keys\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t}\n\t}\n\n\tstatusValue, err := s.env.QueryWorkflow(ShardStatusQuery, PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar status *ShardStatusQueryResult\n\ts.NoError(statusValue.Get(&status))\n\texpected := make(map[int]ShardStatus)\n\tfor i := 0; i < 30; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(ShardStatusResult(expected), status.Result)\n\n\t// check for paginated query result\n\tstatusValue, err = s.env.QueryWorkflow(ShardStatusQuery, PaginatedShardQueryRequest{\n\t\tStartingShardID: common.IntPtr(5),\n\t\tLimitShards:     common.IntPtr(10),\n\t})\n\ts.NoError(err)\n\tstatus = &ShardStatusQueryResult{}\n\ts.NoError(statusValue.Get(&status))\n\texpected = make(map[int]ShardStatus)\n\tfor i := 5; i < 15; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(ShardStatusResult(expected), status.Result)\n\ts.False(status.ShardQueryPaginationToken.IsDone)\n\ts.Equal(15, *status.ShardQueryPaginationToken.NextShardID)\n}\n\nfunc (s *workflowsSuite) TestGetCorruptedKeys_Success() {\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           nil,\n\t}).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{{ShardID: 1}, {ShardID: 5}, {ShardID: 10}},\n\t\tMinShard:      common.IntPtr(1),\n\t\tMaxShard:      common.IntPtr(10),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: common.IntPtr(11),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil)\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           common.IntPtr(11),\n\t}).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{{ShardID: 11}, {ShardID: 12}},\n\t\tMinShard:      common.IntPtr(11),\n\t\tMaxShard:      common.IntPtr(12),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: common.IntPtr(13),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil)\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           common.IntPtr(13),\n\t}).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{{ShardID: 20}, {ShardID: 41}},\n\t\tMinShard:      common.IntPtr(20),\n\t\tMaxShard:      common.IntPtr(41),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: common.IntPtr(42),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil)\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           common.IntPtr(42),\n\t}).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{},\n\t\tMinShard:      nil,\n\t\tMaxShard:      nil,\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: nil,\n\t\t\tIsDone:      true,\n\t\t},\n\t}, nil)\n\n\ts.env.ExecuteWorkflow(GetCorruptedKeys, FixerWorkflowParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.NoError(s.env.GetWorkflowError())\n\tvar result *FixerCorruptedKeysActivityResult\n\ts.NoError(s.env.GetWorkflowResult(&result))\n\ts.Equal(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{\n\t\t\t{ShardID: 1},\n\t\t\t{ShardID: 5},\n\t\t\t{ShardID: 10},\n\t\t\t{ShardID: 11},\n\t\t\t{ShardID: 12},\n\t\t\t{ShardID: 20},\n\t\t\t{ShardID: 41},\n\t\t},\n\t\tMinShard: common.IntPtr(1),\n\t\tMaxShard: common.IntPtr(41),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: nil,\n\t\t\tIsDone:      true,\n\t\t},\n\t}, result)\n}\n\nfunc (s *workflowsSuite) TestGetCorruptedKeys_Error() {\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           nil,\n\t}).Return(&FixerCorruptedKeysActivityResult{\n\t\tCorruptedKeys: []CorruptedKeysEntry{{ShardID: 1}, {ShardID: 5}, {ShardID: 10}},\n\t\tMinShard:      common.IntPtr(1),\n\t\tMaxShard:      common.IntPtr(10),\n\t\tShardQueryPaginationToken: ShardQueryPaginationToken{\n\t\t\tNextShardID: common.IntPtr(11),\n\t\t\tIsDone:      false,\n\t\t},\n\t}, nil)\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, FixerCorruptedKeysActivityParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t\tStartingShardID:           common.IntPtr(11),\n\t}).Return(nil, errors.New(\"got error\"))\n\ts.env.ExecuteWorkflow(GetCorruptedKeys, FixerWorkflowParams{\n\t\tScannerWorkflowWorkflowID: \"test_wid\",\n\t\tScannerWorkflowRunID:      \"test_rid\",\n\t})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Error(s.env.GetWorkflowError())\n}\n\nfunc (s *workflowsSuite) TestScannerWorkflow_Failure_CorruptedKeysActivity() {\n\ts.env.OnActivity(ActivityFixerCorruptedKeys, mock.Anything, mock.Anything).Return(nil, errors.New(\"got error getting corrupted keys\"))\n\ts.env.ExecuteWorkflow(NewTestFixerWorkflow, FixerWorkflowParams{})\n\ts.True(s.env.IsWorkflowCompleted())\n\ts.Equal(\"got error getting corrupted keys\", s.env.GetWorkflowError().Error())\n}\n\nfunc NewTestWorkflow(ctx workflow.Context, name string, params ScannerWorkflowParams) error {\n\twf, err := NewScannerWorkflow(ctx, name, params)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn wf.Start(ctx)\n}\n\nfunc NewTestFixerWorkflow(ctx workflow.Context, params FixerWorkflowParams) error {\n\twf, err := NewFixerWorkflow(ctx, \"test-fixer\", params)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn wf.Start(ctx)\n\n}\n"
  },
  {
    "path": "service/worker/scanner/tasklist/db.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar retryForeverPolicy = newRetryForeverPolicy()\n\nfunc (s *Scavenger) completeTasks(info *p.TaskListInfo, taskID int64, limit int) (int, error) {\n\tvar resp *p.CompleteTasksLessThanResponse\n\tvar err error\n\tdomainName, errorDomain := s.cache.GetDomainName(info.DomainID)\n\tif errorDomain != nil {\n\t\treturn 0, errorDomain\n\t}\n\terr = s.retryForever(func(ctx context.Context) error {\n\t\tresp, err = s.db.CompleteTasksLessThan(ctx, &p.CompleteTasksLessThanRequest{\n\t\t\tDomainID:     info.DomainID,\n\t\t\tTaskListName: info.Name,\n\t\t\tTaskType:     info.TaskType,\n\t\t\tTaskID:       taskID,\n\t\t\tLimit:        limit,\n\t\t\tDomainName:   domainName,\n\t\t})\n\t\treturn err\n\t})\n\tif resp != nil {\n\t\treturn resp.TasksCompleted, err\n\t}\n\treturn 0, err\n}\n\nfunc (s *Scavenger) getOrphanTasks(limit int) (*p.GetOrphanTasksResponse, error) {\n\tvar tasks *p.GetOrphanTasksResponse\n\tvar err error\n\terr = s.retryForever(func(ctx context.Context) error {\n\t\ttasks, err = s.db.GetOrphanTasks(ctx, &p.GetOrphanTasksRequest{\n\t\t\tLimit: limit,\n\t\t})\n\t\treturn err\n\t})\n\treturn tasks, err\n}\n\nfunc (s *Scavenger) completeTask(info *p.TaskListInfo, taskid int64) error {\n\tvar err error\n\tdomainName, errorDomain := s.cache.GetDomainName(info.DomainID)\n\tif errorDomain != nil {\n\t\treturn errorDomain\n\t}\n\terr = s.retryForever(func(ctx context.Context) error {\n\t\terr = s.db.CompleteTask(ctx, &p.CompleteTaskRequest{\n\t\t\tTaskList:   info,\n\t\t\tTaskID:     taskid,\n\t\t\tDomainName: domainName,\n\t\t})\n\t\treturn err\n\t})\n\treturn err\n}\n\nfunc (s *Scavenger) getTasks(info *p.TaskListInfo, batchSize int) (*p.GetTasksResponse, error) {\n\tvar err error\n\tvar resp *p.GetTasksResponse\n\tdomainName, errorDomain := s.cache.GetDomainName(info.DomainID)\n\tif errorDomain != nil {\n\t\treturn nil, errorDomain\n\t}\n\terr = s.retryForever(func(ctx context.Context) error {\n\t\tresp, err = s.db.GetTasks(ctx, &p.GetTasksRequest{\n\t\t\tDomainID:   info.DomainID,\n\t\t\tTaskList:   info.Name,\n\t\t\tTaskType:   info.TaskType,\n\t\t\tReadLevel:  -1, // get the first N tasks sorted by taskID\n\t\t\tBatchSize:  batchSize,\n\t\t\tDomainName: domainName,\n\t\t})\n\t\treturn err\n\t})\n\treturn resp, err\n}\n\nfunc (s *Scavenger) listTaskList(pageSize int, pageToken []byte) (*p.ListTaskListResponse, error) {\n\tvar err error\n\tvar resp *p.ListTaskListResponse\n\terr = s.retryForever(func(ctx context.Context) error {\n\t\tresp, err = s.db.ListTaskList(ctx, &p.ListTaskListRequest{\n\t\t\tPageSize:  pageSize,\n\t\t\tPageToken: pageToken,\n\t\t})\n\t\treturn err\n\t})\n\treturn resp, err\n}\n\nfunc (s *Scavenger) deleteTaskList(info *p.TaskListInfo) error {\n\tdomainName, errorDomain := s.cache.GetDomainName(info.DomainID)\n\tif errorDomain != nil {\n\t\treturn errorDomain\n\t}\n\top := func(ctx context.Context) error {\n\t\treturn s.db.DeleteTaskList(ctx, &p.DeleteTaskListRequest{\n\t\t\tDomainID:     info.DomainID,\n\t\t\tTaskListName: info.Name,\n\t\t\tTaskListType: info.TaskType,\n\t\t\tRangeID:      info.RangeID,\n\t\t\tDomainName:   domainName,\n\t\t})\n\t}\n\t// retry only on service busy errors\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryForeverPolicy),\n\t\tbackoff.WithRetryableError(func(err error) bool {\n\t\t\t_, ok := err.(*types.ServiceBusyError)\n\t\t\treturn ok\n\t\t}),\n\t)\n\treturn throttleRetry.Do(s.ctx, op)\n}\n\nfunc (s *Scavenger) retryForever(op func(ctx context.Context) error) error {\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(retryForeverPolicy),\n\t\tbackoff.WithRetryableError(s.isRetryable),\n\t)\n\treturn throttleRetry.Do(s.ctx, op)\n}\n\nfunc newRetryForeverPolicy() backoff.RetryPolicy {\n\tpolicy := backoff.NewExponentialRetryPolicy(250 * time.Millisecond)\n\tpolicy.SetExpirationInterval(backoff.NoInterval)\n\tpolicy.SetMaximumInterval(30 * time.Second)\n\treturn policy\n}\n\nfunc (s *Scavenger) isRetryable(err error) bool {\n\treturn s.Alive()\n}\n"
  },
  {
    "path": "service/worker/scanner/tasklist/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"strings\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/wrappers/ratelimited\"\n\t\"github.com/uber/cadence/service/worker/scanner/executor\"\n)\n\ntype handlerStatus = executor.TaskStatus\n\nconst (\n\thandlerStatusDone  = executor.TaskStatusDone\n\thandlerStatusErr   = executor.TaskStatusErr\n\thandlerStatusDefer = executor.TaskStatusDefer\n)\n\nconst scannerTaskListPrefix = \"cadence-sys-tl-scanner\"\n\n// deleteHandler handles deletions for a given task list\n// this handler limits the amount of tasks deleted to maxTasksPerJob\n// for fairness among all the task-list in the system - when there\n// is more work to do subsequently, this handler will return StatusDefer\n// with the assumption that the executor will schedule this task later\n//\n// Each loop of the handler proceeds as follows\n//   - Retrieve the next batch of tasks sorted by task_id for this task-list from persistence\n//   - If there are 0 tasks for this task-list, try deleting the task-list if its idle\n//   - If any of the tasks in the batch isn't expired, we are done. Since tasks are retrieved\n//     in sorted order, if one of the tasks isn't expired, chances are, none of the tasks above\n//     it are expired as well - so, we give up and wait for the next run\n//   - Delete the entire batch of tasks\n//   - If the number of tasks retrieved is less than batchSize, there are no more tasks in the task-list\n//     Try deleting the task-list if its idle\nfunc (s *Scavenger) deleteHandler(taskListInfo *p.TaskListInfo) handlerStatus {\n\tvar err error\n\tvar nProcessed, nDeleted int\n\n\tdefer func() { s.deleteHandlerLog(taskListInfo, nProcessed, nDeleted, err) }()\n\ttaskBatchSize := s.taskBatchSizeFn()\n\tmaxTasksPerJob := s.maxTasksPerJobFn()\n\n\tfor nProcessed < maxTasksPerJob {\n\t\tresp, err1 := s.getTasks(taskListInfo, taskBatchSize)\n\t\tif err1 != nil {\n\t\t\terr = err1\n\t\t\treturn handlerStatusErr\n\t\t}\n\n\t\tnTasks := len(resp.Tasks)\n\t\tif nTasks == 0 {\n\t\t\ts.tryDeleteTaskList(taskListInfo)\n\t\t\treturn handlerStatusDone\n\t\t}\n\n\t\tfor _, task := range resp.Tasks {\n\t\t\tnProcessed++\n\t\t\tif !s.isTaskExpired(task) {\n\t\t\t\treturn handlerStatusDone\n\t\t\t}\n\t\t}\n\n\t\ttaskID := resp.Tasks[nTasks-1].TaskID\n\t\tif _, err = s.completeTasks(taskListInfo, taskID, nTasks); err != nil {\n\t\t\treturn handlerStatusErr\n\t\t}\n\n\t\tnDeleted += nTasks\n\t\tif nTasks < taskBatchSize {\n\t\t\ts.tryDeleteTaskList(taskListInfo)\n\t\t\treturn handlerStatusDone\n\t\t}\n\t}\n\n\treturn handlerStatusDefer\n}\n\nfunc (s *Scavenger) tryDeleteTaskList(info *p.TaskListInfo) {\n\tif strings.HasPrefix(info.Name, scannerTaskListPrefix) {\n\t\treturn // avoid deleting our own task list\n\t}\n\tdelta := time.Since(info.LastUpdated)\n\tif delta < taskListGracePeriod {\n\t\treturn\n\t}\n\t// usually, matching engine is the authoritative owner of a tasklist\n\t// and its incorrect for any other entity to mutate executorTask lists (including deleting it)\n\t// the delete here is safe because of two reasons:\n\t//   - we delete the executorTask list only if the lastUpdated is > 48H. If a executorTask list is idle for\n\t//     this amount of time, it will no longer be owned by any host in matching engine (because\n\t//     of idle timeout). If any new host has to take ownership of this at this time, it can only\n\t//     do so by updating the rangeID\n\t//   - deleteTaskList is a conditional delete where condition is the rangeID\n\tif err := s.deleteTaskList(info); err != nil {\n\t\ts.logger.Error(\"deleteTaskList error\", tag.Error(err))\n\t\treturn\n\t}\n\tatomic.AddInt64(&s.stats.tasklist.nDeleted, 1)\n\ts.logger.Info(\"tasklist deleted\", tag.WorkflowDomainID(info.DomainID), tag.WorkflowTaskListName(info.Name), tag.TaskType(info.TaskType))\n}\n\nfunc (s *Scavenger) deleteHandlerLog(info *p.TaskListInfo, nProcessed int, nDeleted int, err error) {\n\tatomic.AddInt64(&s.stats.task.nDeleted, int64(nDeleted))\n\tatomic.AddInt64(&s.stats.task.nProcessed, int64(nProcessed))\n\tif err != nil {\n\t\ts.logger.Error(\"scavenger.deleteHandler processed.\",\n\t\t\ttag.Error(err), tag.WorkflowDomainID(info.DomainID), tag.WorkflowTaskListName(info.Name), tag.TaskType(info.TaskType), tag.NumberProcessed(nProcessed), tag.NumberDeleted(nDeleted))\n\t\treturn\n\t}\n\tif nProcessed > 0 {\n\t\ts.logger.Info(\"scavenger.deleteHandler processed.\",\n\t\t\ttag.WorkflowDomainID(info.DomainID), tag.WorkflowTaskListName(info.Name), tag.TaskType(info.TaskType), tag.NumberProcessed(nProcessed), tag.NumberDeleted(nDeleted))\n\t}\n}\n\nfunc (s *Scavenger) isTaskExpired(t *p.TaskInfo) bool {\n\treturn t.Expiry.After(time.Unix(0, 0)) && time.Now().After(t.Expiry)\n}\n\nfunc (s *Scavenger) completeOrphanTasksHandler() handlerStatus {\n\tvar nDeleted int\n\tbatchSize := s.getOrphanTasksPageSizeFn()\n\tresp, err := s.getOrphanTasks(batchSize)\n\tif err == ratelimited.ErrPersistenceLimitExceeded {\n\t\ts.logger.Info(\"scavenger.completeOrphanTasksHandler query was ratelimited; will retry\")\n\t\treturn handlerStatusDefer\n\t}\n\tif err != nil {\n\t\ts.logger.Error(\"scavenger.completeOrphanTasksHandler error getting orphan tasks\")\n\t\treturn handlerStatusErr\n\t}\n\tfor _, taskKey := range resp.Tasks {\n\t\terr = s.completeTask(&p.TaskListInfo{\n\t\t\tDomainID: taskKey.DomainID,\n\t\t\tName:     taskKey.TaskListName,\n\t\t\tTaskType: taskKey.TaskType,\n\t\t}, taskKey.TaskID)\n\t\tif err == ratelimited.ErrPersistenceLimitExceeded {\n\t\t\ts.logger.Info(\"scavenger.completeOrphanTasksHandler query was ratelimited; will retry\")\n\t\t\treturn handlerStatusDefer\n\t\t}\n\t\tif err != nil {\n\t\t\ts.logger.Error(\"scavenger.completeOrphanTasksHandler error getting orphan tasks\")\n\t\t\treturn handlerStatusErr\n\t\t}\n\t\tnDeleted++\n\t\tatomic.AddInt64(&s.stats.task.nDeleted, 1)\n\t\tatomic.AddInt64(&s.stats.task.nProcessed, 1)\n\t}\n\ts.logger.Info(\"scavenger.completeOrphanTasksHandler deleted.\", tag.NumberDeleted(nDeleted))\n\tif len(resp.Tasks) < batchSize {\n\t\treturn handlerStatusDone\n\t}\n\treturn handlerStatusDefer\n}\n"
  },
  {
    "path": "service/worker/scanner/tasklist/mocks_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"encoding/binary\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\n\tp \"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tmockTaskTable struct {\n\t\tdomainID   string\n\t\tworkflowID string\n\t\trunID      string\n\t\tnextTaskID int64\n\t\ttasks      []*p.TaskInfo\n\t}\n\tmockTaskListTable struct {\n\t\tsync.Mutex\n\t\tinfo []p.TaskListInfo\n\t}\n)\n\nfunc newMockTaskTable() *mockTaskTable {\n\treturn &mockTaskTable{\n\t\tdomainID:   uuid.New(),\n\t\tworkflowID: uuid.New(),\n\t\trunID:      uuid.New(),\n\t}\n}\n\nfunc (tbl *mockTaskListTable) generate(name string, idle bool) {\n\ttl := p.TaskListInfo{\n\t\tDomainID:    uuid.New(),\n\t\tName:        name,\n\t\tRangeID:     22,\n\t\tLastUpdated: time.Now(),\n\t}\n\tif idle {\n\t\ttl.LastUpdated = time.Unix(1000, 1000)\n\t}\n\ttbl.info = append(tbl.info, tl)\n}\n\nfunc (tbl *mockTaskListTable) list(token []byte, count int) ([]p.TaskListInfo, []byte) {\n\ttbl.Lock()\n\tdefer tbl.Unlock()\n\tif tbl.info == nil {\n\t\treturn nil, nil\n\t}\n\tvar off int\n\tif token != nil {\n\t\toff = int(binary.BigEndian.Uint32(token))\n\t}\n\trem := len(tbl.info) - count + 1\n\tif rem < count {\n\t\treturn tbl.info[off:], nil\n\t}\n\ttoken = make([]byte, 4)\n\tbinary.BigEndian.PutUint32(token, uint32(off+count))\n\treturn tbl.info[off : off+count], token\n}\n\nfunc (tbl *mockTaskListTable) delete(name string) {\n\ttbl.Lock()\n\tdefer tbl.Unlock()\n\tvar newInfo []p.TaskListInfo\n\tfor _, tl := range tbl.info {\n\t\tif tl.Name != name {\n\t\t\tnewInfo = append(newInfo, tl)\n\t\t}\n\t}\n\ttbl.info = newInfo\n}\n\nfunc (tbl *mockTaskListTable) get(name string) *p.TaskListInfo {\n\ttbl.Lock()\n\tdefer tbl.Unlock()\n\tfor _, tl := range tbl.info {\n\t\tif tl.Name == name {\n\t\t\treturn &tl\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (tbl *mockTaskTable) generate(count int, expired bool) {\n\tfor i := 0; i < count; i++ {\n\t\tti := &p.TaskInfo{\n\t\t\tDomainID:                      tbl.domainID,\n\t\t\tWorkflowID:                    tbl.workflowID,\n\t\t\tRunID:                         tbl.runID,\n\t\t\tTaskID:                        tbl.nextTaskID,\n\t\t\tScheduleID:                    3,\n\t\t\tScheduleToStartTimeoutSeconds: 30,\n\t\t\tExpiry:                        time.Now().Add(time.Hour),\n\t\t}\n\t\tif expired {\n\t\t\tti.ScheduleToStartTimeoutSeconds = -33\n\t\t\tti.Expiry = time.Unix(0, time.Now().UnixNano()-int64(time.Second*33))\n\t\t}\n\t\ttbl.tasks = append(tbl.tasks, ti)\n\t\ttbl.nextTaskID++\n\t}\n}\n\nfunc (tbl *mockTaskTable) get(count int) []*p.TaskInfo {\n\tif len(tbl.tasks) >= count {\n\t\treturn tbl.tasks[:count]\n\t}\n\treturn tbl.tasks[:]\n}\n\nfunc (tbl *mockTaskTable) deleteLessThan(id int64, limit int) int {\n\tcount := 0\n\tfor _, t := range tbl.tasks {\n\t\tif t.TaskID <= id && count < limit {\n\t\t\tcount++\n\t\t\tcontinue\n\t\t}\n\t\tbreak\n\t}\n\tswitch {\n\tcase count == 0:\n\tcase count == len(tbl.tasks):\n\t\ttbl.tasks = nil\n\tdefault:\n\t\ttbl.tasks = tbl.tasks[count:]\n\t}\n\treturn count\n}\n"
  },
  {
    "path": "service/worker/scanner/tasklist/scavenger.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/service/worker/scanner/executor\"\n)\n\nconst (\n\texecutorMaxDeferredTasks = 10000\n\ttaskListBatchSize        = 32 // maximum number of task list we process concurrently\n\ttaskBatchSize            = 16\n\ttaskListGracePeriod      = 48 * time.Hour // amount of time a executorTask list has to be idle before it becomes a candidate for deletion\n)\n\ntype (\n\t// Scavenger is the type that holds the state for task list scavenger daemon\n\tScavenger struct {\n\t\tctx                      context.Context\n\t\tdb                       p.TaskManager\n\t\tcache                    cache.DomainCache\n\t\texecutor                 executor.Executor\n\t\tscope                    metrics.Scope\n\t\tlogger                   log.Logger\n\t\tstats                    stats\n\t\tstatus                   int32\n\t\tgetOrphanTasksPageSizeFn dynamicproperties.IntPropertyFn\n\t\ttaskBatchSizeFn          dynamicproperties.IntPropertyFn\n\t\tmaxTasksPerJobFn         dynamicproperties.IntPropertyFn\n\t\tcleanOrphans             dynamicproperties.BoolPropertyFn\n\t\tpollInterval             time.Duration\n\t\ttimeSource               clock.TimeSource\n\n\t\t// stopC is used to signal the scavenger to stop\n\t\tstopC chan struct{}\n\t\t// stopWG is used to wait for the scavenger to stop\n\t\tstopWG sync.WaitGroup\n\t\t// stoppedC is closed once the scavenger is stopped\n\t\tstopped chan struct{}\n\t}\n\n\tstats struct {\n\t\ttasklist struct {\n\t\t\tnProcessed int64\n\t\t\tnDeleted   int64\n\t\t}\n\t\ttask struct {\n\t\t\tnProcessed int64\n\t\t\tnDeleted   int64\n\t\t}\n\t}\n\t// Options is used to customize scavenger operations\n\tOptions struct {\n\t\tGetOrphanTasksPageSizeFn dynamicproperties.IntPropertyFn\n\t\tTaskBatchSizeFn          dynamicproperties.IntPropertyFn\n\t\tEnableCleaning           dynamicproperties.BoolPropertyFn\n\t\tMaxTasksPerJobFn         dynamicproperties.IntPropertyFn\n\t\tExecutorPollInterval     time.Duration\n\t}\n\n\t// executorTask is a runnable task that adheres to the executor.Task interface\n\t// for the scavenger, each of this task processes a single task list\n\texecutorTask struct {\n\t\ttaskListInfo p.TaskListInfo\n\t\tscvg         *Scavenger\n\t}\n\n\t// orphanExecutorTask is a runnable task that processes a limited block of\n\t// orphans\n\torphanExecutorTask struct {\n\t\tscvg *Scavenger\n\t}\n)\n\n// NewScavenger returns an instance of executorTask list scavenger daemon\n// The Scavenger can be started by calling the Start() method on the\n// returned object. Calling the Start() method will result in one\n// complete iteration over all of the task lists in the system. For\n// each task list, the scavenger will attempt\n//   - deletion of expired tasks in the task lists\n//   - deletion of task list itself, if there are no tasks and the task list hasn't been updated for a grace period\n//\n// The scavenger will retry on all persistence errors infinitely and will only stop under\n// two conditions\n//   - either all task lists are processed successfully (or)\n//   - Stop() method is called to stop the scavenger\nfunc NewScavenger(\n\tctx context.Context,\n\tdb p.TaskManager,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n\topts *Options,\n\tcache cache.DomainCache,\n) *Scavenger {\n\ttaskExecutor := executor.NewFixedSizePoolExecutor(\n\t\ttaskListBatchSize,\n\t\texecutorMaxDeferredTasks,\n\t\tmetricsClient,\n\t\tmetrics.TaskListScavengerScope,\n\t)\n\n\tif opts == nil {\n\t\topts = &Options{}\n\t}\n\n\tcleanOrphans := opts.EnableCleaning\n\tif cleanOrphans == nil {\n\t\tcleanOrphans = func(opts ...dynamicproperties.FilterOption) bool {\n\t\t\treturn false\n\t\t}\n\t}\n\n\tgetOrphanTasksPageSize := opts.GetOrphanTasksPageSizeFn\n\tif getOrphanTasksPageSize == nil {\n\t\tgetOrphanTasksPageSize = func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn dynamicproperties.ScannerGetOrphanTasksPageSize.DefaultInt()\n\t\t}\n\t}\n\n\ttaskBatchSizeFn := opts.TaskBatchSizeFn\n\tif taskBatchSizeFn == nil {\n\t\ttaskBatchSizeFn = func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn taskBatchSize\n\t\t}\n\t}\n\n\tmaxTasksPerJobFn := opts.MaxTasksPerJobFn\n\tif maxTasksPerJobFn == nil {\n\t\tmaxTasksPerJobFn = func(opts ...dynamicproperties.FilterOption) int {\n\t\t\treturn dynamicproperties.ScannerMaxTasksProcessedPerTasklistJob.DefaultInt()\n\t\t}\n\t}\n\n\tpollInterval := opts.ExecutorPollInterval\n\tif pollInterval == 0 {\n\t\tpollInterval = time.Minute\n\t}\n\treturn &Scavenger{\n\t\tctx:                      ctx,\n\t\tdb:                       db,\n\t\tcache:                    cache,\n\t\tscope:                    metricsClient.Scope(metrics.TaskListScavengerScope),\n\t\tlogger:                   logger,\n\t\tstopC:                    make(chan struct{}),\n\t\tstopped:                  make(chan struct{}),\n\t\texecutor:                 taskExecutor,\n\t\tcleanOrphans:             cleanOrphans,\n\t\ttaskBatchSizeFn:          taskBatchSizeFn,\n\t\tpollInterval:             pollInterval,\n\t\tmaxTasksPerJobFn:         maxTasksPerJobFn,\n\t\tgetOrphanTasksPageSizeFn: getOrphanTasksPageSize,\n\t\ttimeSource:               clock.NewRealTimeSource(),\n\t}\n}\n\n// Start starts the scavenger\nfunc (s *Scavenger) Start() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\ts.logger.Info(\"Tasklist scavenger starting\")\n\ts.stopWG.Add(1)\n\ts.executor.Start()\n\tgo s.run()\n\ts.scope.IncCounter(metrics.StartedCount)\n\ts.logger.Info(\"Tasklist scavenger started\")\n}\n\n// Stop stops the scavenger\nfunc (s *Scavenger) Stop() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\ts.scope.IncCounter(metrics.StoppedCount)\n\ts.logger.Info(\"Tasklist scavenger stopping\")\n\tclose(s.stopC)\n\ts.executor.Stop()\n\ts.stopWG.Wait()\n\ts.logger.Info(\"Tasklist scavenger stopped\")\n\tclose(s.stopped)\n}\n\n// Alive returns true if the scavenger is still running\nfunc (s *Scavenger) Alive() bool {\n\treturn atomic.LoadInt32(&s.status) == common.DaemonStatusStarted\n}\n\n// run does a single run over all executorTask lists\nfunc (s *Scavenger) run() {\n\tdefer func() {\n\t\ts.emitStats()\n\t\tgo s.Stop()\n\t\ts.stopWG.Done()\n\t}()\n\n\t// Start a task to delete orphaned tasks from the tasks table, if enabled\n\tif s.cleanOrphans() {\n\t\ts.executor.Submit(&orphanExecutorTask{scvg: s})\n\t}\n\n\tvar pageToken []byte\n\tfor {\n\t\tresp, err := s.listTaskList(taskListBatchSize, pageToken)\n\t\tif err != nil {\n\t\t\ts.logger.Error(\"listTaskList error\", tag.Error(err))\n\t\t\treturn\n\t\t}\n\n\t\tfor _, item := range resp.Items {\n\t\t\tatomic.AddInt64(&s.stats.tasklist.nProcessed, 1)\n\t\t\tif !s.executor.Submit(s.newTask(&item)) {\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\n\t\tpageToken = resp.NextPageToken\n\t\tif pageToken == nil {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ts.awaitExecutor()\n}\n\n// process is a callback function that gets invoked from within the executor.Run() method\nfunc (s *Scavenger) process(taskListInfo *p.TaskListInfo) executor.TaskStatus {\n\treturn s.deleteHandler(taskListInfo)\n}\n\nfunc (s *Scavenger) awaitExecutor() {\n\toutstanding := s.executor.TaskCount()\n\tticker := s.timeSource.NewTicker(s.pollInterval)\n\tdefer ticker.Stop()\n\tfor outstanding > 0 {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\toutstanding = s.executor.TaskCount()\n\t\t\ts.scope.UpdateGauge(metrics.TaskListOutstandingCount, float64(outstanding))\n\t\tcase <-s.stopC:\n\t\t\treturn\n\t\t}\n\t}\n}\n\nfunc (s *Scavenger) emitStats() {\n\ts.scope.UpdateGauge(metrics.TaskProcessedCount, float64(s.stats.task.nProcessed))\n\ts.scope.UpdateGauge(metrics.TaskDeletedCount, float64(s.stats.task.nDeleted))\n\ts.scope.UpdateGauge(metrics.TaskListProcessedCount, float64(s.stats.tasklist.nProcessed))\n\ts.scope.UpdateGauge(metrics.TaskListDeletedCount, float64(s.stats.tasklist.nDeleted))\n}\n\n// newTask returns a new instance of an executable task which will process a single task list\nfunc (s *Scavenger) newTask(info *p.TaskListInfo) executor.Task {\n\treturn &executorTask{\n\t\ttaskListInfo: *info,\n\t\tscvg:         s,\n\t}\n}\n\n// Run runs the task\nfunc (t *executorTask) Run() executor.TaskStatus {\n\treturn t.scvg.process(&t.taskListInfo)\n}\n\nfunc (t *orphanExecutorTask) Run() executor.TaskStatus {\n\treturn t.scvg.completeOrphanTasksHandler()\n}\n"
  },
  {
    "path": "service/worker/scanner/tasklist/scavenger_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage tasklist\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/uber-go/tally\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\tp \"github.com/uber/cadence/common/persistence\"\n)\n\ntype (\n\tScavengerTestSuite struct {\n\t\tsuite.Suite\n\t\ttaskListTable   *mockTaskListTable\n\t\ttaskTables      map[string]*mockTaskTable\n\t\ttaskMgr         *mocks.TaskManager\n\t\tscvgr           *Scavenger\n\t\tscvgrCancelFn   context.CancelFunc\n\t\tmockDomainCache *cache.MockDomainCache\n\t}\n)\n\nconst (\n\tscavengerTestTimeout = 10 * time.Second\n)\n\nvar errTest = errors.New(\"transient error\")\n\nfunc TestScavengerTestSuite(t *testing.T) {\n\tsuite.Run(t, new(ScavengerTestSuite))\n}\n\nfunc (s *ScavengerTestSuite) SetupTest() {\n\ts.taskMgr = &mocks.TaskManager{}\n\ts.taskListTable = &mockTaskListTable{}\n\ts.taskTables = make(map[string]*mockTaskTable)\n\tlogger := testlogger.New(s.T())\n\tctrl := gomock.NewController(s.T())\n\ts.mockDomainCache = cache.NewMockDomainCache(ctrl)\n\tscvgrCtx, scvgrCancelFn := context.WithTimeout(context.Background(), scavengerTestTimeout)\n\ts.scvgr = NewScavenger(\n\t\tscvgrCtx,\n\t\ts.taskMgr,\n\t\tmetrics.NewClient(tally.NoopScope, metrics.Worker, metrics.HistogramMigration{}),\n\t\tlogger,\n\t\t&Options{\n\t\t\tEnableCleaning:           dynamicproperties.GetBoolPropertyFn(true),\n\t\t\tTaskBatchSizeFn:          dynamicproperties.GetIntPropertyFn(16),\n\t\t\tGetOrphanTasksPageSizeFn: dynamicproperties.GetIntPropertyFn(16),\n\t\t\tExecutorPollInterval:     time.Millisecond * 50,\n\t\t},\n\t\ts.mockDomainCache,\n\t)\n\ts.scvgrCancelFn = scvgrCancelFn\n}\n\nfunc (s *ScavengerTestSuite) TestAllExpiredTasks() {\n\tnTasks := 32\n\tnTaskLists := 3\n\tfor i := 0; i < nTaskLists; i++ {\n\t\tname := fmt.Sprintf(\"test-expired-tl-%v\", i)\n\t\ts.taskListTable.generate(name, true)\n\t\ttt := newMockTaskTable()\n\t\ttt.generate(nTasks, true)\n\t\ts.taskTables[name] = tt\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\ts.setupTaskMgrMocks()\n\ts.runScavenger()\n\tfor tl, tbl := range s.taskTables {\n\t\ttasks := tbl.get(100)\n\t\ts.Equal(0, len(tasks), \"failed to delete all expired tasks\")\n\t\ts.Nil(s.taskListTable.get(tl), \"failed to delete expired executorTask list\")\n\t}\n}\n\nfunc (s *ScavengerTestSuite) TestAllAliveTasks() {\n\tnTasks := 32\n\tnTaskLists := 3\n\tfor i := 0; i < nTaskLists; i++ {\n\t\tname := fmt.Sprintf(\"test-Alive-tl-%v\", i)\n\t\ts.taskListTable.generate(name, true)\n\t\ttt := newMockTaskTable()\n\t\ttt.generate(nTasks, false)\n\t\ts.taskTables[name] = tt\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\ts.setupTaskMgrMocks()\n\ts.runScavenger()\n\tfor tl, tbl := range s.taskTables {\n\t\ttasks := tbl.get(100)\n\t\ts.Equal(nTasks, len(tasks), \"scavenger deleted a non-expired executorTask\")\n\t\ts.NotNil(s.taskListTable.get(tl), \"scavenger deleted a non-expired executorTask list\")\n\t}\n}\n\nfunc (s *ScavengerTestSuite) TestExpiredTasksFollowedByAlive() {\n\tnTasks := 32\n\tnTaskLists := 3\n\tfor i := 0; i < nTaskLists; i++ {\n\t\tname := fmt.Sprintf(\"test-Alive-tl-%v\", i)\n\t\ts.taskListTable.generate(name, true)\n\t\ttt := newMockTaskTable()\n\t\ttt.generate(nTasks/2, true)\n\t\ttt.generate(nTasks/2, false)\n\t\ts.taskTables[name] = tt\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\ts.setupTaskMgrMocks()\n\ts.runScavenger()\n\tfor tl, tbl := range s.taskTables {\n\t\ttasks := tbl.get(100)\n\t\ts.Equal(nTasks/2, len(tasks), \"scavenger deleted non-expired tasks\")\n\t\ts.Equal(int64(nTasks/2), tasks[0].TaskID, \"scavenger deleted wrong set of tasks\")\n\t\ts.NotNil(s.taskListTable.get(tl), \"scavenger deleted a non-expired executorTask list\")\n\t}\n}\n\nfunc (s *ScavengerTestSuite) TestAliveTasksFollowedByExpired() {\n\tnTasks := 32\n\tnTaskLists := 3\n\tfor i := 0; i < nTaskLists; i++ {\n\t\tname := fmt.Sprintf(\"test-Alive-tl-%v\", i)\n\t\ts.taskListTable.generate(name, true)\n\t\ttt := newMockTaskTable()\n\t\ttt.generate(nTasks/2, false)\n\t\ttt.generate(nTasks/2, true)\n\t\ts.taskTables[name] = tt\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\ts.setupTaskMgrMocks()\n\ts.runScavenger()\n\tfor tl, tbl := range s.taskTables {\n\t\ttasks := tbl.get(100)\n\t\ts.Equal(nTasks, len(tasks), \"scavenger deleted non-expired tasks\")\n\t\ts.NotNil(s.taskListTable.get(tl), \"scavenger deleted a non-expired executorTask list\")\n\t}\n}\n\nfunc (s *ScavengerTestSuite) TestAllExpiredTasksWithErrors() {\n\tnTasks := 32\n\tnTaskLists := 3\n\tfor i := 0; i < nTaskLists; i++ {\n\t\tname := fmt.Sprintf(\"test-expired-tl-%v\", i)\n\t\ts.taskListTable.generate(name, true)\n\t\ttt := newMockTaskTable()\n\t\ttt.generate(nTasks, true)\n\t\ts.taskTables[name] = tt\n\t}\n\ts.mockDomainCache.EXPECT().GetDomainName(gomock.Any()).Return(\"test_domain_name\", nil).AnyTimes()\n\ts.setupTaskMgrMocksWithErrors()\n\ts.runScavenger()\n\tfor _, tbl := range s.taskTables {\n\t\ttasks := tbl.get(100)\n\t\ts.Equal(0, len(tasks), \"failed to delete all expired tasks\")\n\t}\n\tresult, _ := s.taskListTable.list(nil, 10)\n\ts.Equal(1, len(result), \"expected partial deletion due to transient errors\")\n}\n\nfunc (s *ScavengerTestSuite) runScavenger() {\n\ts.scvgr.Start()\n\tdefer s.scvgr.Stop()\n\ttimer := time.NewTimer(scavengerTestTimeout)\n\tselect {\n\tcase <-s.scvgr.stopped:\n\t\ttimer.Stop()\n\t\treturn\n\tcase <-timer.C:\n\t\ts.Fail(\"timed out waiting for scavenger to finish\")\n\t}\n}\n\nfunc (s *ScavengerTestSuite) setupTaskMgrMocks() {\n\ts.taskMgr.On(\"ListTaskList\", mock.Anything, mock.Anything).Return(\n\t\tfunc(_ context.Context, req *p.ListTaskListRequest) *p.ListTaskListResponse {\n\t\t\titems, next := s.taskListTable.list(req.PageToken, req.PageSize)\n\t\t\treturn &p.ListTaskListResponse{Items: items, NextPageToken: next}\n\t\t}, nil)\n\ts.taskMgr.On(\"DeleteTaskList\", mock.Anything, mock.Anything).Return(\n\t\tfunc(_ context.Context, req *p.DeleteTaskListRequest) error {\n\t\t\ts.taskListTable.delete(req.TaskListName)\n\t\t\treturn nil\n\t\t})\n\ts.taskMgr.On(\"GetTasks\", mock.Anything, mock.Anything).Return(\n\t\tfunc(_ context.Context, req *p.GetTasksRequest) *p.GetTasksResponse {\n\t\t\tresult := s.taskTables[req.TaskList].get(req.BatchSize)\n\t\t\treturn &p.GetTasksResponse{Tasks: result}\n\t\t}, nil)\n\ts.taskMgr.On(\"CompleteTasksLessThan\", mock.Anything, mock.Anything).Return(\n\t\tfunc(_ context.Context, req *p.CompleteTasksLessThanRequest) *p.CompleteTasksLessThanResponse {\n\t\t\trowsDeleted := s.taskTables[req.TaskListName].deleteLessThan(req.TaskID, req.Limit)\n\t\t\treturn &p.CompleteTasksLessThanResponse{TasksCompleted: rowsDeleted}\n\t\t}, nil)\n\ts.taskMgr.On(\"GetOrphanTasks\", mock.Anything, mock.Anything).Return(\n\t\tfunc(_ context.Context, req *p.GetOrphanTasksRequest) *p.GetOrphanTasksResponse {\n\t\t\treturn &p.GetOrphanTasksResponse{\n\t\t\t\tTasks: make([]*p.TaskKey, 0),\n\t\t\t}\n\t\t}, nil)\n}\n\nfunc (s *ScavengerTestSuite) setupTaskMgrMocksWithErrors() {\n\ts.taskMgr.On(\"ListTaskList\", mock.Anything, mock.Anything).Return(nil, errTest).Once()\n\ts.taskMgr.On(\"GetTasks\", mock.Anything, mock.Anything).Return(nil, errTest).Once()\n\ts.taskMgr.On(\"CompleteTasksLessThan\", mock.Anything, mock.Anything).Return(nil, errTest).Once()\n\ts.taskMgr.On(\"DeleteTaskList\", mock.Anything, mock.Anything).Return(errTest).Once()\n\ts.taskMgr.On(\"GetOrphanTasks\", mock.Anything, mock.Anything).Return(nil, errTest).Once()\n\ts.setupTaskMgrMocks()\n}\n"
  },
  {
    "path": "service/worker/scanner/timers/timers.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timers\n\nimport (\n\t\"context\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/blobstore\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/fetcher\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\nconst (\n\t// ScannerWFTypeName defines workflow type name for concrete executions scanner\n\tScannerWFTypeName   = \"cadence-sys-timers-scanner-workflow\"\n\twfid                = \"cadence-sys-timers-scanner\"\n\tscannerTaskListName = \"cadence-sys-timers-scanner-tasklist-0\"\n\n\t// FixerWFTypeName defines workflow type name for timers fixer\n\tFixerWFTypeName   = \"cadence-sys-timers-fixer-workflow\"\n\tfixerTaskListName = \"cadence-sys-timers-fixer-tasklist-0\"\n\tfixerwfid         = \"cadence-sys-timers-fixer\"\n\tperiodStartKey    = \"period_start\"\n\tperiodEndKey      = \"period_end\"\n)\n\n// ScannerWorkflow starts timers scanner.\nfunc ScannerWorkflow(\n\tctx workflow.Context,\n\tparams shardscanner.ScannerWorkflowParams,\n) error {\n\twf, err := shardscanner.NewScannerWorkflow(ctx, ScannerWFTypeName, params)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn wf.Start(ctx)\n}\n\n// FixerWorkflow starts timers executions fixer.\nfunc FixerWorkflow(\n\tctx workflow.Context,\n\tparams shardscanner.FixerWorkflowParams,\n) error {\n\twf, err := shardscanner.NewFixerWorkflow(ctx, FixerWFTypeName, params)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn wf.Start(ctx)\n}\n\n// ScannerHooks provides hooks for timers scanner.\nfunc ScannerHooks() *shardscanner.ScannerHooks {\n\th, err := shardscanner.NewScannerHooks(Manager, Iterator, Config)\n\tif err != nil {\n\t\treturn nil\n\t}\n\n\treturn h\n}\n\n// FixerHooks provides hooks needed for timers fixer.\nfunc FixerHooks() *shardscanner.FixerHooks {\n\th, err := shardscanner.NewFixerHooks(FixerManager, FixerIterator, timerCustomConfig)\n\tif err != nil {\n\t\treturn nil\n\t}\n\treturn h\n}\n\nfunc timerCustomConfig(_ shardscanner.FixerContext) shardscanner.CustomScannerConfig {\n\t// must be non-empty to pass backwards-compat check,\n\t// and this allows safely adding more invariants in the future.\n\t//\n\t// currently this is not read anywhere because \"fixer enabled\"\n\t// means \"run this one invariant's fixes\".\n\treturn map[string]string{\n\t\tinvariant.TimerInvalidName: \"true\",\n\t}\n}\n\n// Manager provides invariant manager for timers scanner.\nfunc Manager(\n\t_ context.Context,\n\tpr persistence.Retryer,\n\t_ shardscanner.ScanShardActivityParams,\n\tcache cache.DomainCache,\n) invariant.Manager {\n\treturn invariant.NewInvariantManager(getInvariants(pr, cache))\n}\n\n// Iterator provides iterator for timers scanner.\nfunc Iterator(\n\tctx context.Context,\n\tpr persistence.Retryer,\n\tparams shardscanner.ScanShardActivityParams,\n) pagination.Iterator {\n\tstart, err := strconv.Atoi(params.ScannerConfig[periodStartKey])\n\tif err != nil {\n\t\treturn nil\n\t}\n\tend, err := strconv.Atoi(params.ScannerConfig[periodEndKey])\n\tif err != nil {\n\t\treturn nil\n\t}\n\n\tstartAt := time.Now().Add(time.Hour * time.Duration(start))\n\tendAt := startAt.Add(time.Hour * time.Duration(end))\n\n\treturn fetcher.TimerIterator(ctx, pr, startAt, endAt, params.PageSize)\n\n}\n\n// FixerIterator provides iterator for timers fixer.\nfunc FixerIterator(\n\tctx context.Context,\n\tclient blobstore.Client,\n\tkeys store.Keys,\n\t_ shardscanner.FixShardActivityParams,\n) store.ScanOutputIterator {\n\treturn store.NewBlobstoreIterator(ctx, client, keys, &entity.Timer{})\n}\n\n// FixerManager provides invariant manager for timers fixer.\nfunc FixerManager(\n\t_ context.Context,\n\tpr persistence.Retryer,\n\t_ shardscanner.FixShardActivityParams,\n\tcache cache.DomainCache,\n) invariant.Manager {\n\treturn invariant.NewInvariantManager(getInvariants(pr, cache))\n}\n\n// Config resolves dynamic config for timers scanner.\nfunc Config(ctx shardscanner.ScannerContext) shardscanner.CustomScannerConfig {\n\tres := shardscanner.CustomScannerConfig{}\n\tres[periodStartKey] = strconv.Itoa(ctx.Config.DynamicCollection.GetIntProperty(dynamicproperties.TimersScannerPeriodStart)())\n\tres[periodEndKey] = strconv.Itoa(ctx.Config.DynamicCollection.GetIntProperty(dynamicproperties.TimersScannerPeriodEnd)())\n\treturn res\n}\n\n// ScannerConfig configures timers scanner\nfunc ScannerConfig(dc *dynamicconfig.Collection) *shardscanner.ScannerConfig {\n\treturn &shardscanner.ScannerConfig{\n\t\tScannerWFTypeName: ScannerWFTypeName,\n\t\tFixerWFTypeName:   FixerWFTypeName,\n\t\tDynamicParams: shardscanner.DynamicParams{\n\t\t\tScannerEnabled:          dc.GetBoolProperty(dynamicproperties.TimersScannerEnabled),\n\t\t\tFixerEnabled:            dc.GetBoolProperty(dynamicproperties.TimersFixerEnabled),\n\t\t\tConcurrency:             dc.GetIntProperty(dynamicproperties.TimersScannerConcurrency),\n\t\t\tPageSize:                dc.GetIntProperty(dynamicproperties.TimersScannerPersistencePageSize),\n\t\t\tBlobstoreFlushThreshold: dc.GetIntProperty(dynamicproperties.TimersScannerBlobstoreFlushThreshold),\n\t\t\tActivityBatchSize:       dc.GetIntProperty(dynamicproperties.TimersScannerActivityBatchSize),\n\t\t\tAllowDomain:             dc.GetBoolPropertyFilteredByDomain(dynamicproperties.TimersFixerDomainAllow),\n\t\t},\n\t\tDynamicCollection: dc,\n\t\tScannerHooks:      ScannerHooks,\n\t\tFixerHooks:        FixerHooks,\n\n\t\tStartWorkflowOptions: client.StartWorkflowOptions{\n\t\t\tID:                           wfid,\n\t\t\tTaskList:                     scannerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        client.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t\tStartFixerOptions: client.StartWorkflowOptions{\n\t\t\tID:                           fixerwfid,\n\t\t\tTaskList:                     fixerTaskListName,\n\t\t\tExecutionStartToCloseTimeout: 20 * 365 * 24 * time.Hour,\n\t\t\tWorkflowIDReusePolicy:        client.WorkflowIDReusePolicyAllowDuplicate,\n\t\t\tCronSchedule:                 \"*/5 * * * *\",\n\t\t},\n\t}\n}\n\nfunc getInvariants(pr persistence.Retryer, cache cache.DomainCache) []invariant.Invariant {\n\treturn []invariant.Invariant{\n\t\tinvariant.NewTimerInvalid(pr, cache),\n\t}\n}\n"
  },
  {
    "path": "service/worker/scanner/timers/timers_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage timers\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n)\n\ntype timersWorkflowsSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n\tcontroller *gomock.Controller\n}\n\nfunc TestScannerWorkflowSuite(t *testing.T) {\n\tsuite.Run(t, new(timersWorkflowsSuite))\n}\n\nfunc (s *timersWorkflowsSuite) SetupSuite() {\n\tworkflow.Register(ScannerWorkflow)\n\ts.controller = gomock.NewController(s.T())\n\n}\nfunc (s *timersWorkflowsSuite) TestScannerWorkflow_SetsHooks() {\n\tdcClient := dynamicconfig.NewMockClient(s.controller)\n\tlogger := log.NewNoop()\n\n\tdc := dynamicconfig.NewCollection(dcClient, logger)\n\tcfg := ScannerConfig(dc)\n\ts.Equal(ScannerWFTypeName, cfg.ScannerWFTypeName, \"scanner wf type name is set\")\n\ts.NotNil(cfg.FixerHooks)\n\ts.NotNil(cfg.ScannerHooks)\n}\nfunc (s *timersWorkflowsSuite) TestScannerWorkflow_Success() {\n\tenv := s.NewTestWorkflowEnvironment()\n\tcconfig := shardscanner.CustomScannerConfig{\n\t\tperiodStartKey: \"1\",\n\t\tperiodEndKey:   \"2\",\n\t}\n\tenv.OnActivity(shardscanner.ActivityScannerConfig, mock.Anything, mock.Anything).Return(shardscanner.ResolvedScannerWorkflowConfig{\n\t\tGenericScannerConfig: shardscanner.GenericScannerConfig{\n\t\t\tEnabled:           true,\n\t\t\tConcurrency:       3,\n\t\t\tActivityBatchSize: 5,\n\t\t},\n\t\tCustomScannerConfig: cconfig,\n\t}, nil)\n\tenv.OnActivity(shardscanner.ActivityScannerEmitMetrics, mock.Anything, mock.Anything).Return(nil)\n\tshards := shardscanner.Shards{\n\t\tRange: &shardscanner.ShardRange{\n\t\t\tMin: 0,\n\t\t\tMax: 30,\n\t\t},\n\t}\n\n\tbatches := [][]int{\n\t\t{0, 3, 6, 9, 12},\n\t\t{15, 18, 21, 24, 27},\n\t\t{1, 4, 7, 10, 13},\n\t\t{16, 19, 22, 25, 28},\n\t\t{2, 5, 8, 11, 14},\n\t\t{17, 20, 23, 26, 29},\n\t}\n\n\tfor _, batch := range batches {\n\t\tvar reports []shardscanner.ScanReport\n\t\tfor i := range batch {\n\t\t\tif i == 0 {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t} else {\n\t\t\t\treports = append(reports, shardscanner.ScanReport{\n\t\t\t\t\tShardID: batch[i],\n\t\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{},\n\t\t\t\t\t},\n\t\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\t// var customc shardscanner.CustomScannerConfig\n\t\tenv.OnActivity(shardscanner.ActivityScanShard, mock.Anything, shardscanner.ScanShardActivityParams{\n\t\t\tShards:        batch,\n\t\t\tScannerConfig: cconfig,\n\t\t}).Return(reports, nil)\n\t}\n\n\tenv.ExecuteWorkflow(ScannerWorkflow, shardscanner.ScannerWorkflowParams{\n\t\tShards: shards,\n\t})\n\ts.True(env.IsWorkflowCompleted())\n\ts.NoError(env.GetWorkflowError())\n\n\taggValue, err := env.QueryWorkflow(shardscanner.AggregateReportQuery)\n\ts.NoError(err)\n\tvar agg shardscanner.AggregateScanReportResult\n\ts.NoError(aggValue.Get(&agg))\n\ts.Equal(shardscanner.AggregateScanReportResult{\n\t\tEntitiesCount:    240,\n\t\tCorruptedCount:   48,\n\t\tCheckFailedCount: 24,\n\t\tCorruptionByType: map[invariant.Name]int64{},\n\t}, agg)\n\n\tfor i := 0; i < 30; i++ {\n\t\tshardReportValue, err := env.QueryWorkflow(shardscanner.ShardReportQuery, i)\n\t\ts.NoError(err)\n\t\tvar shardReport *shardscanner.ScanReport\n\t\ts.NoError(shardReportValue.Get(&shardReport))\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount: 10,\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tControlFlowFailure: &shardscanner.ControlFlowFailure{\n\t\t\t\t\t\tInfo: \"got control flow failure\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t} else {\n\t\t\ts.Equal(&shardscanner.ScanReport{\n\t\t\t\tShardID: i,\n\t\t\t\tStats: shardscanner.ScanStats{\n\t\t\t\t\tEntitiesCount:    10,\n\t\t\t\t\tCorruptedCount:   2,\n\t\t\t\t\tCheckFailedCount: 1,\n\t\t\t\t\tCorruptionByType: map[invariant.Name]int64{},\n\t\t\t\t},\n\t\t\t\tResult: shardscanner.ScanResult{\n\t\t\t\t\tShardScanKeys: &shardscanner.ScanKeys{\n\t\t\t\t\t\tCorrupt: &store.Keys{\n\t\t\t\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\t\t\t\tMinPage: 0,\n\t\t\t\t\t\t\tMaxPage: 10,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}, shardReport)\n\t\t}\n\t}\n\n\tstatusValue, err := env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar status *shardscanner.ShardStatusQueryResult\n\ts.NoError(statusValue.Get(&status))\n\texpected := make(map[int]shardscanner.ShardStatus)\n\tfor i := 0; i < 30; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.True(status.ShardQueryPaginationToken.IsDone)\n\ts.Nil(status.ShardQueryPaginationToken.NextShardID)\n\n\t// check for paginated query result\n\tstatusValue, err = env.QueryWorkflow(shardscanner.ShardStatusQuery, shardscanner.PaginatedShardQueryRequest{\n\t\tStartingShardID: common.IntPtr(5),\n\t\tLimitShards:     common.IntPtr(10),\n\t})\n\ts.NoError(err)\n\tstatus = &shardscanner.ShardStatusQueryResult{}\n\ts.NoError(statusValue.Get(&status))\n\texpected = make(map[int]shardscanner.ShardStatus)\n\tfor i := 5; i < 15; i++ {\n\t\tif i == 0 || i == 1 || i == 2 || i == 15 || i == 16 || i == 17 {\n\t\t\texpected[i] = shardscanner.ShardStatusControlFlowFailure\n\t\t} else {\n\t\t\texpected[i] = shardscanner.ShardStatusSuccess\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardStatusResult(expected), status.Result)\n\ts.False(status.ShardQueryPaginationToken.IsDone)\n\ts.Equal(15, *status.ShardQueryPaginationToken.NextShardID)\n\n\tcorruptionKeysValue, err := env.QueryWorkflow(shardscanner.ShardCorruptKeysQuery, shardscanner.PaginatedShardQueryRequest{})\n\ts.NoError(err)\n\tvar shardCorruptKeysResult *shardscanner.ShardCorruptKeysQueryResult\n\ts.NoError(corruptionKeysValue.Get(&shardCorruptKeysResult))\n\texpectedCorrupted := make(map[int]store.Keys)\n\tfor i := 0; i < 30; i++ {\n\t\tif i != 0 && i != 1 && i != 2 && i != 15 && i != 16 && i != 17 {\n\t\t\texpectedCorrupted[i] = store.Keys{\n\t\t\t\tUUID:    \"test_uuid\",\n\t\t\t\tMinPage: 0,\n\t\t\t\tMaxPage: 10,\n\t\t\t}\n\t\t}\n\t}\n\ts.Equal(shardscanner.ShardCorruptKeysResult(expectedCorrupted), shardCorruptKeysResult.Result)\n}\n"
  },
  {
    "path": "service/worker/scanner/workflow.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"context\"\n\t\"time\"\n\n\t\"go.uber.org/cadence\"\n\t\"go.uber.org/cadence/activity\"\n\tcclient \"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/service/worker/scanner/executions\"\n\t\"github.com/uber/cadence/service/worker/scanner/history\"\n\t\"github.com/uber/cadence/service/worker/scanner/tasklist\"\n\t\"github.com/uber/cadence/service/worker/scanner/timers\"\n)\n\nconst (\n\tmaxConcurrentActivityExecutionSize     = 10\n\tmaxConcurrentDecisionTaskExecutionSize = 10\n\tinfiniteDuration                       = 20 * 365 * 24 * time.Hour\n\n\ttlScannerWFID                 = \"cadence-sys-tl-scanner\"\n\ttlScannerWFTypeName           = \"cadence-sys-tl-scanner-workflow\"\n\ttlScannerTaskListName         = \"cadence-sys-tl-scanner-tasklist-0\"\n\ttaskListScavengerActivityName = \"cadence-sys-tl-scanner-scvg-activity\"\n\n\thistoryScannerWFID           = \"cadence-sys-history-scanner\"\n\thistoryScannerWFTypeName     = \"cadence-sys-history-scanner-workflow\"\n\thistoryScannerTaskListName   = \"cadence-sys-history-scanner-tasklist-0\"\n\thistoryScavengerActivityName = \"cadence-sys-history-scanner-scvg-activity\"\n)\n\nvar (\n\ttlScavengerHBInterval = 10 * time.Second\n\n\tactivityRetryPolicy = cadence.RetryPolicy{\n\t\tInitialInterval:    10 * time.Second,\n\t\tBackoffCoefficient: 1.7,\n\t\tMaximumInterval:    5 * time.Minute,\n\t\tExpirationInterval: infiniteDuration,\n\t}\n\tactivityOptions = workflow.ActivityOptions{\n\t\tScheduleToStartTimeout: 5 * time.Minute,\n\t\tStartToCloseTimeout:    infiniteDuration,\n\t\tHeartbeatTimeout:       5 * time.Minute,\n\t\tRetryPolicy:            &activityRetryPolicy,\n\t}\n\ttlScannerWFStartOptions = cclient.StartWorkflowOptions{\n\t\tID:                           tlScannerWFID,\n\t\tTaskList:                     tlScannerTaskListName,\n\t\tExecutionStartToCloseTimeout: 5 * 24 * time.Hour,\n\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\tCronSchedule:                 \"0 */12 * * *\",\n\t}\n\thistoryScannerWFStartOptions = cclient.StartWorkflowOptions{\n\t\tID:                           historyScannerWFID,\n\t\tTaskList:                     historyScannerTaskListName,\n\t\tExecutionStartToCloseTimeout: infiniteDuration,\n\t\tWorkflowIDReusePolicy:        cclient.WorkflowIDReusePolicyAllowDuplicate,\n\t\tCronSchedule:                 \"0 */12 * * *\",\n\t}\n)\n\nfunc init() {\n\tworkflow.RegisterWithOptions(TaskListScannerWorkflow, workflow.RegisterOptions{Name: tlScannerWFTypeName})\n\tactivity.RegisterWithOptions(TaskListScavengerActivity, activity.RegisterOptions{Name: taskListScavengerActivityName})\n\n\tworkflow.RegisterWithOptions(HistoryScannerWorkflow, workflow.RegisterOptions{Name: historyScannerWFTypeName})\n\tactivity.RegisterWithOptions(HistoryScavengerActivity, activity.RegisterOptions{Name: historyScavengerActivityName})\n\n\tworkflow.RegisterWithOptions(executions.ConcreteScannerWorkflow, workflow.RegisterOptions{Name: executions.ConcreteExecutionsScannerWFTypeName})\n\tworkflow.RegisterWithOptions(executions.CurrentScannerWorkflow, workflow.RegisterOptions{Name: executions.CurrentExecutionsScannerWFTypeName})\n\tworkflow.RegisterWithOptions(executions.ConcreteFixerWorkflow, workflow.RegisterOptions{Name: executions.ConcreteExecutionsFixerWFTypeName})\n\tworkflow.RegisterWithOptions(executions.CurrentFixerWorkflow, workflow.RegisterOptions{Name: executions.CurrentExecutionsFixerWFTypeName})\n\tworkflow.RegisterWithOptions(timers.ScannerWorkflow, workflow.RegisterOptions{Name: timers.ScannerWFTypeName})\n\tworkflow.RegisterWithOptions(timers.FixerWorkflow, workflow.RegisterOptions{Name: timers.FixerWFTypeName})\n}\n\n// TaskListScannerWorkflow is the workflow that runs the task-list scanner background daemon\nfunc TaskListScannerWorkflow(\n\tctx workflow.Context,\n) error {\n\n\tfuture := workflow.ExecuteActivity(workflow.WithActivityOptions(ctx, activityOptions), taskListScavengerActivityName)\n\treturn future.Get(ctx, nil)\n}\n\n// HistoryScannerWorkflow is the workflow that runs the history scanner background daemon\nfunc HistoryScannerWorkflow(\n\tctx workflow.Context,\n) error {\n\n\tfuture := workflow.ExecuteActivity(\n\t\tworkflow.WithActivityOptions(ctx, activityOptions),\n\t\thistoryScavengerActivityName,\n\t)\n\treturn future.Get(ctx, nil)\n}\n\n// HistoryScavengerActivity is the activity that runs history scavenger\nfunc HistoryScavengerActivity(\n\tactivityCtx context.Context,\n) (history.ScavengerHeartbeatDetails, error) {\n\n\tctx, err := getScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn history.ScavengerHeartbeatDetails{}, err\n\t}\n\n\trps := ctx.cfg.ScannerPersistenceMaxQPS()\n\tres := ctx.resource\n\n\thbd := history.ScavengerHeartbeatDetails{}\n\tif activity.HasHeartbeatDetails(activityCtx) {\n\t\tif err := activity.GetHeartbeatDetails(activityCtx, &hbd); err != nil {\n\t\t\tres.GetLogger().Error(\"Failed to recover from last heartbeat, start over from beginning\", tag.Error(err))\n\t\t}\n\t}\n\tcache := res.GetDomainCache()\n\tscavenger := history.NewScavenger(\n\t\tres.GetHistoryManager(),\n\t\trps,\n\t\tres.GetHistoryClient(),\n\t\thbd,\n\t\tres.GetMetricsClient(),\n\t\tres.GetLogger(),\n\t\tctx.cfg.MaxWorkflowRetentionInDays,\n\t\tcache,\n\t)\n\treturn scavenger.Run(activityCtx)\n}\n\n// TaskListScavengerActivity is the activity that runs task list scavenger\nfunc TaskListScavengerActivity(\n\tactivityCtx context.Context,\n) error {\n\tctx, err := getScannerContext(activityCtx)\n\tif err != nil {\n\t\treturn err\n\t}\n\tres := ctx.resource\n\tscavenger := tasklist.NewScavenger(\n\t\tactivityCtx,\n\t\tres.GetTaskManager(),\n\t\tres.GetMetricsClient(),\n\t\tres.GetLogger(),\n\t\t&ctx.cfg.TaskListScannerOptions,\n\t\tres.GetDomainCache(),\n\t)\n\n\tres.GetLogger().Info(\"Starting task list scavenger\")\n\tscavenger.Start()\n\tfor scavenger.Alive() {\n\t\tactivity.RecordHeartbeat(activityCtx)\n\t\tif activityCtx.Err() != nil {\n\t\t\tres.GetLogger().Info(\"activity context error, stopping scavenger\", tag.Error(activityCtx.Err()))\n\t\t\tscavenger.Stop()\n\t\t\treturn activityCtx.Err()\n\t\t}\n\t\ttime.Sleep(tlScavengerHBInterval)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "service/worker/scanner/workflow_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scanner\n\nimport (\n\t\"context\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/cadence/testsuite\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/metrics\"\n\tp \"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/service/worker/scanner/tasklist\"\n)\n\ntype scannerWorkflowTestSuite struct {\n\tsuite.Suite\n\ttestsuite.WorkflowTestSuite\n}\n\nfunc TestScannerWorkflowTestSuite(t *testing.T) {\n\tsuite.Run(t, new(scannerWorkflowTestSuite))\n}\n\nfunc (s *scannerWorkflowTestSuite) TestWorkflow() {\n\tenv := s.NewTestWorkflowEnvironment()\n\tenv.OnActivity(taskListScavengerActivityName, mock.Anything).Return(nil)\n\tenv.ExecuteWorkflow(tlScannerWFTypeName)\n\ts.True(env.IsWorkflowCompleted())\n}\n\nfunc (s *scannerWorkflowTestSuite) TestScavengerActivity() {\n\tenv := s.NewTestActivityEnvironment()\n\tcontroller := gomock.NewController(s.T())\n\tmockResource := resource.NewTest(s.T(), controller, metrics.Worker)\n\tdefer mockResource.Finish(s.T())\n\n\tmockResource.TaskMgr.On(\"ListTaskList\", mock.Anything, mock.Anything).Return(&p.ListTaskListResponse{}, nil)\n\tmockResource.TaskMgr.On(\"GetOrphanTasks\", mock.Anything, mock.Anything).Return(&p.GetOrphanTasksResponse{}, nil)\n\tctx := scannerContext{\n\t\tresource: mockResource,\n\t\tcfg: Config{\n\t\t\tTaskListScannerOptions: tasklist.Options{\n\t\t\t\tGetOrphanTasksPageSizeFn: dynamicproperties.GetIntPropertyFn(dynamicproperties.ScannerGetOrphanTasksPageSize.DefaultInt()),\n\t\t\t\tEnableCleaning:           dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\tExecutorPollInterval:     time.Millisecond * 50,\n\t\t\t},\n\t\t},\n\t}\n\tenv.SetTestTimeout(time.Second * 5)\n\tenv.SetWorkerOptions(worker.Options{\n\t\tBackgroundActivityContext: NewScannerContext(context.Background(), \"default-test-workflow-type-name\", ctx),\n\t})\n\ttlScavengerHBInterval = time.Millisecond * 10\n\t_, err := env.ExecuteActivity(taskListScavengerActivityName)\n\ts.NoError(err)\n}\n"
  },
  {
    "path": "service/worker/scheduler/activity.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// schedulerRequestIDNamespace is a stable UUID namespace used to derive\n// deterministic RequestIDs. Cassandra's schema stores create_request_id as\n// a uuid column, so plain strings are rejected by the gocql driver.\nvar schedulerRequestIDNamespace = uuid.NewSHA1(uuid.NameSpaceDNS, []byte(\"cadence.scheduler\"))\n\ntype contextKey string\n\nconst schedulerContextKey contextKey = \"schedulerContext\"\n\n// schedulerContext is the context passed to activities via BackgroundActivityContext.\ntype schedulerContext struct {\n\tFrontendClient frontend.Client\n}\n\n// startWorkflowActivity starts a workflow execution as specified by the schedule's action.\n// It generates a deterministic workflow ID from the schedule ID and scheduled time\n// to ensure idempotent execution.\nfunc startWorkflowActivity(ctx context.Context, req StartWorkflowRequest) (*StartWorkflowResult, error) {\n\tsc, ok := ctx.Value(schedulerContextKey).(schedulerContext)\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"scheduler context not found in activity context\")\n\t}\n\n\tworkflowID := generateWorkflowID(req.Action.WorkflowIDPrefix, req.ScheduleID, req.ScheduledTime)\n\n\treusePolicy := types.WorkflowIDReusePolicyAllowDuplicate\n\tstartReq := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              req.Domain,\n\t\tWorkflowID:                          workflowID,\n\t\tWorkflowType:                        req.Action.WorkflowType,\n\t\tTaskList:                            req.Action.TaskList,\n\t\tInput:                               req.Action.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: req.Action.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      req.Action.TaskStartToCloseTimeoutSeconds,\n\t\tRequestID:                           generateRequestID(req.ScheduleID, req.ScheduledTime.UnixNano(), req.TriggerSource),\n\t\tWorkflowIDReusePolicy:               &reusePolicy,\n\t\tRetryPolicy:                         req.Action.RetryPolicy,\n\t\tMemo:                                req.Action.Memo,\n\t\tSearchAttributes:                    req.Action.SearchAttributes,\n\t}\n\n\tresp, err := sc.FrontendClient.StartWorkflowExecution(ctx, startReq)\n\tif err != nil {\n\t\tif isAlreadyStartedError(err) {\n\t\t\treturn &StartWorkflowResult{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tSkipped:    true,\n\t\t\t}, nil\n\t\t}\n\t\treturn nil, fmt.Errorf(\"failed to start workflow: %w\", err)\n\t}\n\n\treturn &StartWorkflowResult{\n\t\tWorkflowID: workflowID,\n\t\tRunID:      resp.GetRunID(),\n\t\tStarted:    true,\n\t}, nil\n}\n\n// generateWorkflowID creates a deterministic workflow ID from the\n// schedule's prefix (or schedule ID) and the scheduled time.\n// Example: \"my-prefix-2026-01-15T10:00:00Z\"\nfunc generateWorkflowID(prefix, scheduleID string, scheduledTime time.Time) string {\n\tif prefix == \"\" {\n\t\tprefix = scheduleID\n\t}\n\treturn fmt.Sprintf(\"%s-%s\", prefix, scheduledTime.UTC().Format(time.RFC3339))\n}\n\n// generateRequestID produces a deterministic UUID from the schedule ID,\n// scheduled time, and trigger source. Including the trigger source ensures\n// that a backfill for the same timestamp as a normal schedule fire produces\n// a distinct RequestID, avoiding unintended server-side deduplication.\nfunc generateRequestID(scheduleID string, scheduledTimeNanos int64, source TriggerSource) string {\n\tname := fmt.Sprintf(\"%s-%d-%s\", scheduleID, scheduledTimeNanos, source)\n\treturn uuid.NewSHA1(schedulerRequestIDNamespace, []byte(name)).String()\n}\n\nfunc isAlreadyStartedError(err error) bool {\n\t_, ok := err.(*types.WorkflowExecutionAlreadyStartedError)\n\treturn ok\n}\n"
  },
  {
    "path": "service/worker/scheduler/activity_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/google/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestGenerateWorkflowID(t *testing.T) {\n\tts := time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC)\n\ttests := []struct {\n\t\tname       string\n\t\tprefix     string\n\t\tscheduleID string\n\t\ttime       time.Time\n\t\twant       string\n\t}{\n\t\t{\n\t\t\tname:       \"uses prefix when provided\",\n\t\t\tprefix:     \"my-workflow\",\n\t\t\tscheduleID: \"sched-123\",\n\t\t\ttime:       ts,\n\t\t\twant:       \"my-workflow-2026-01-15T10:00:00Z\",\n\t\t},\n\t\t{\n\t\t\tname:       \"falls back to scheduleID when prefix is empty\",\n\t\t\tprefix:     \"\",\n\t\t\tscheduleID: \"sched-456\",\n\t\t\ttime:       ts,\n\t\t\twant:       \"sched-456-2026-01-15T10:00:00Z\",\n\t\t},\n\t\t{\n\t\t\tname:       \"deterministic for same inputs\",\n\t\t\tprefix:     \"wf\",\n\t\t\tscheduleID: \"sched-789\",\n\t\t\ttime:       ts,\n\t\t\twant:       \"wf-2026-01-15T10:00:00Z\",\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot := generateWorkflowID(tc.prefix, tc.scheduleID, tc.time)\n\t\t\tassert.Equal(t, tc.want, got)\n\t\t})\n\t}\n}\n\nfunc TestGenerateRequestID(t *testing.T) {\n\tt.Run(\"returns valid UUID\", func(t *testing.T) {\n\t\tid := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\t_, err := uuid.Parse(id)\n\t\tassert.NoError(t, err)\n\t})\n\tt.Run(\"deterministic for same inputs\", func(t *testing.T) {\n\t\ta := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\tb := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\tassert.Equal(t, a, b)\n\t})\n\tt.Run(\"different for different scheduleID\", func(t *testing.T) {\n\t\ta := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\tb := generateRequestID(\"sched-2\", 1000000000, TriggerSourceSchedule)\n\t\tassert.NotEqual(t, a, b)\n\t})\n\tt.Run(\"different for different time\", func(t *testing.T) {\n\t\ta := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\tb := generateRequestID(\"sched-1\", 2000000000, TriggerSourceSchedule)\n\t\tassert.NotEqual(t, a, b)\n\t})\n\tt.Run(\"different for different trigger source\", func(t *testing.T) {\n\t\ta := generateRequestID(\"sched-1\", 1000000000, TriggerSourceSchedule)\n\t\tb := generateRequestID(\"sched-1\", 1000000000, TriggerSourceBackfill)\n\t\tassert.NotEqual(t, a, b)\n\t})\n}\n\nfunc TestIsAlreadyStartedError(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\terr  error\n\t\twant bool\n\t}{\n\t\t{\n\t\t\tname: \"WorkflowExecutionAlreadyStartedError returns true\",\n\t\t\terr:  &types.WorkflowExecutionAlreadyStartedError{Message: \"already started\"},\n\t\t\twant: true,\n\t\t},\n\t\t{\n\t\t\tname: \"other error returns false\",\n\t\t\terr:  errors.New(\"some other error\"),\n\t\t\twant: false,\n\t\t},\n\t\t{\n\t\t\tname: \"nil error returns false\",\n\t\t\terr:  nil,\n\t\t\twant: false,\n\t\t},\n\t}\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tgot := isAlreadyStartedError(tc.err)\n\t\t\tassert.Equal(t, tc.want, got)\n\t\t})\n\t}\n}\n\nfunc TestStartWorkflowActivity(t *testing.T) {\n\tscheduledTime := time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC)\n\tint32Ptr := func(v int32) *int32 { return &v }\n\tbaseReq := StartWorkflowRequest{\n\t\tDomain:     \"test-domain\",\n\t\tScheduleID: \"sched-1\",\n\t\tAction: types.StartWorkflowAction{\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"my-workflow\"},\n\t\t\tTaskList:                            &types.TaskList{Name: \"my-tasklist\"},\n\t\t\tInput:                               []byte(`{\"key\":\"value\"}`),\n\t\t\tWorkflowIDPrefix:                    \"my-prefix\",\n\t\t\tExecutionStartToCloseTimeoutSeconds: int32Ptr(3600),\n\t\t\tTaskStartToCloseTimeoutSeconds:      int32Ptr(60),\n\t\t},\n\t\tScheduledTime: scheduledTime,\n\t\tTriggerSource: TriggerSourceSchedule,\n\t}\n\n\ttests := []struct {\n\t\tname       string\n\t\treq        StartWorkflowRequest\n\t\tsetupMock  func(m *frontend.MockClient)\n\t\twantResult *StartWorkflowResult\n\t\twantErr    bool\n\t}{\n\t\t{\n\t\t\tname: \"successful start\",\n\t\t\treq:  baseReq,\n\t\t\tsetupMock: func(m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, req *types.StartWorkflowExecutionRequest, _ ...interface{}) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, \"test-domain\", req.Domain)\n\t\t\t\t\t\tassert.Equal(t, \"my-prefix-\"+formatTime(scheduledTime), req.WorkflowID)\n\t\t\t\t\t\tassert.Equal(t, \"my-workflow\", req.WorkflowType.Name)\n\t\t\t\t\t\tassert.Equal(t, \"my-tasklist\", req.TaskList.Name)\n\t\t\t\t\t\t_, uuidErr := uuid.Parse(req.RequestID)\n\t\t\t\t\t\tassert.NoError(t, uuidErr, \"RequestID must be a valid UUID\")\n\t\t\t\t\t\tassert.Equal(t, generateRequestID(\"sched-1\", scheduledTime.UnixNano(), TriggerSourceSchedule), req.RequestID, \"RequestID must be deterministic\")\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: \"run-abc\"}, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\twantResult: &StartWorkflowResult{\n\t\t\t\tWorkflowID: \"my-prefix-\" + formatTime(scheduledTime),\n\t\t\t\tRunID:      \"run-abc\",\n\t\t\t\tStarted:    true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"already started returns skipped\",\n\t\t\treq:  baseReq,\n\t\t\tsetupMock: func(m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, &types.WorkflowExecutionAlreadyStartedError{Message: \"already started\"})\n\t\t\t},\n\t\t\twantResult: &StartWorkflowResult{\n\t\t\t\tWorkflowID: \"my-prefix-\" + formatTime(scheduledTime),\n\t\t\t\tSkipped:    true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"transient error propagated\",\n\t\t\treq:  baseReq,\n\t\t\tsetupMock: func(m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"connection refused\"))\n\t\t\t},\n\t\t\twantErr: true,\n\t\t},\n\t\t{\n\t\t\tname: \"empty prefix falls back to scheduleID\",\n\t\t\treq: func() StartWorkflowRequest {\n\t\t\t\tr := baseReq\n\t\t\t\tr.Action.WorkflowIDPrefix = \"\"\n\t\t\t\treturn r\n\t\t\t}(),\n\t\t\tsetupMock: func(m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, req *types.StartWorkflowExecutionRequest, _ ...interface{}) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, \"sched-1-\"+formatTime(scheduledTime), req.WorkflowID)\n\t\t\t\t\t\treturn &types.StartWorkflowExecutionResponse{RunID: \"run-def\"}, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\twantResult: &StartWorkflowResult{\n\t\t\t\tWorkflowID: \"sched-1-\" + formatTime(scheduledTime),\n\t\t\t\tRunID:      \"run-def\",\n\t\t\t\tStarted:    true,\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockClient := frontend.NewMockClient(ctrl)\n\t\t\ttc.setupMock(mockClient)\n\n\t\t\tctx := context.WithValue(context.Background(), schedulerContextKey, schedulerContext{\n\t\t\t\tFrontendClient: mockClient,\n\t\t\t})\n\n\t\t\tresult, err := startWorkflowActivity(ctx, tc.req)\n\t\t\tif tc.wantErr {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\trequire.NoError(t, err)\n\t\t\tassert.Equal(t, tc.wantResult, result)\n\t\t})\n\t}\n}\n\nfunc TestStartWorkflowActivity_MissingContext(t *testing.T) {\n\tctx := context.Background()\n\t_, err := startWorkflowActivity(ctx, StartWorkflowRequest{})\n\trequire.Error(t, err)\n\tassert.Contains(t, err.Error(), \"scheduler context not found\")\n}\n\nfunc formatTime(t time.Time) string {\n\treturn t.UTC().Format(time.RFC3339)\n}\n"
  },
  {
    "path": "service/worker/scheduler/client_worker.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"context\"\n\t\"sync\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\tcadenceworker \"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nconst (\n\tdefaultRefreshInterval = 1 * time.Minute\n\tdefaultShutdownTimeout = 5 * time.Second\n)\n\n// BootstrapParams contains the parameters needed to create a scheduler worker manager.\ntype BootstrapParams struct {\n\tServiceClient      workflowserviceclient.Interface\n\tFrontendClient     frontend.Client\n\tLogger             log.Logger\n\tDomainCache        cache.DomainCache\n\tMembershipResolver membership.Resolver\n\tHostInfo           membership.HostInfo\n}\n\n// workerHandle is the subset of cadenceworker.Worker used by the manager,\n// extracted to allow unit testing without starting real pollers.\ntype workerHandle interface {\n\tStop()\n}\n\n// workerFactory creates a worker for a given domain. Returns a workerHandle\n// and an error. The default factory creates a real Cadence SDK worker.\ntype workerFactory func(domainName string) (workerHandle, error)\n\n// WorkerManager manages per-domain scheduler workers. It periodically scans\n// the domain cache and uses the membership hashring to determine which domains\n// this host owns. For each owned domain it starts a Cadence SDK worker polling\n// the scheduler task list in that domain.\ntype WorkerManager struct {\n\tenabledFn          dynamicproperties.BoolPropertyFn\n\tserviceClient      workflowserviceclient.Interface\n\tfrontendClient     frontend.Client\n\tlogger             log.Logger\n\tdomainCache        cache.DomainCache\n\tmembershipResolver membership.Resolver\n\thostInfo           membership.HostInfo\n\ttimeSrc            clock.TimeSource\n\trefreshInterval    time.Duration\n\tshutdownTimeout    time.Duration\n\tctx                context.Context\n\tcancelFn           context.CancelFunc\n\twg                 sync.WaitGroup\n\tactiveWorkers      map[string]workerHandle // domain name -> worker\n\tcreateWorker       workerFactory\n}\n\n// NewWorkerManager creates a new per-domain scheduler worker manager.\nfunc NewWorkerManager(params *BootstrapParams, enabledFn dynamicproperties.BoolPropertyFn) *WorkerManager {\n\tctx, cancel := context.WithCancel(context.Background())\n\twm := &WorkerManager{\n\t\tenabledFn:          enabledFn,\n\t\tserviceClient:      params.ServiceClient,\n\t\tfrontendClient:     params.FrontendClient,\n\t\tlogger:             params.Logger.WithTags(tag.ComponentScheduler),\n\t\tdomainCache:        params.DomainCache,\n\t\tmembershipResolver: params.MembershipResolver,\n\t\thostInfo:           params.HostInfo,\n\t\ttimeSrc:            clock.NewRealTimeSource(),\n\t\trefreshInterval:    defaultRefreshInterval,\n\t\tshutdownTimeout:    defaultShutdownTimeout,\n\t\tctx:                ctx,\n\t\tcancelFn:           cancel,\n\t\tactiveWorkers:      make(map[string]workerHandle),\n\t}\n\twm.createWorker = wm.defaultCreateWorker\n\treturn wm\n}\n\n// Start begins the background loop that manages per-domain workers.\nfunc (m *WorkerManager) Start() {\n\tm.logger.Info(\"scheduler worker manager starting\")\n\tm.wg.Add(1)\n\tgo m.run()\n}\n\n// Stop signals the background loop to stop and waits for it to finish.\n// It then stops all active workers.\nfunc (m *WorkerManager) Stop() {\n\tm.logger.Info(\"scheduler worker manager stopping\")\n\tm.cancelFn()\n\tif !common.AwaitWaitGroup(&m.wg, m.shutdownTimeout) {\n\t\tm.logger.Warn(\"scheduler worker manager timed out on shutdown\")\n\t}\n\tm.stopAllWorkers()\n\tm.logger.Info(\"scheduler worker manager stopped\")\n}\n\nfunc (m *WorkerManager) run() {\n\tdefer m.wg.Done()\n\n\tticker := m.timeSrc.NewTicker(m.refreshInterval)\n\tdefer ticker.Stop()\n\n\tenabled := m.enabledFn()\n\tif enabled {\n\t\tm.refreshWorkers()\n\t} else {\n\t\tm.logger.Info(\"scheduler worker manager is disabled, skipping initial refresh\")\n\t}\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker.Chan():\n\t\t\tpreviouslyEnabled := enabled\n\t\t\tenabled = m.enabledFn()\n\t\t\tif enabled != previouslyEnabled {\n\t\t\t\tm.logger.Info(\"scheduler worker manager enabled state changed\",\n\t\t\t\t\ttag.Dynamic(\"enabled\", enabled),\n\t\t\t\t)\n\t\t\t}\n\n\t\t\tif enabled {\n\t\t\t\tm.refreshWorkers()\n\t\t\t} else {\n\t\t\t\tm.stopAllWorkers()\n\t\t\t}\n\t\tcase <-m.ctx.Done():\n\t\t\tm.logger.Info(\"scheduler worker manager background loop stopped\")\n\t\t\treturn\n\t\t}\n\t}\n}\n\n// refreshWorkers scans all domains and reconciles the set of active workers\n// with the domains this host owns via the membership hashring.\nfunc (m *WorkerManager) refreshWorkers() {\n\tdomains := m.domainCache.GetAllDomain()\n\townedDomains := make(map[string]struct{}, len(domains))\n\tlookupFailed := make(map[string]struct{})\n\n\tfor _, domainEntry := range domains {\n\t\tselect {\n\t\tcase <-m.ctx.Done():\n\t\t\treturn\n\t\tdefault:\n\t\t}\n\n\t\tif domainEntry.IsDeprecatedOrDeleted() {\n\t\t\tcontinue\n\t\t}\n\n\t\tdomainName := domainEntry.GetInfo().Name\n\n\t\towner, err := m.membershipResolver.Lookup(service.Worker, domainName)\n\t\tif err != nil {\n\t\t\tm.logger.Warn(\"failed to look up domain owner, skipping\",\n\t\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\t\ttag.Error(err),\n\t\t\t)\n\t\t\tlookupFailed[domainName] = struct{}{}\n\t\t\tcontinue\n\t\t}\n\n\t\tif owner.Identity() != m.hostInfo.Identity() {\n\t\t\tcontinue\n\t\t}\n\n\t\townedDomains[domainName] = struct{}{}\n\n\t\tif _, exists := m.activeWorkers[domainName]; exists {\n\t\t\tcontinue\n\t\t}\n\n\t\tm.startWorkerForDomain(domainName)\n\t}\n\n\tfor domainName, w := range m.activeWorkers {\n\t\tif _, owned := ownedDomains[domainName]; owned {\n\t\t\tcontinue\n\t\t}\n\t\t// Keep workers running for domains where lookup failed to avoid\n\t\t// unnecessary churn during transient membership ring issues.\n\t\tif _, failed := lookupFailed[domainName]; failed {\n\t\t\tcontinue\n\t\t}\n\t\tm.logger.Info(\"stopping scheduler worker for domain no longer owned\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\t\tw.Stop()\n\t\tdelete(m.activeWorkers, domainName)\n\t}\n\n\tm.logger.Debug(\"scheduler workers refreshed\",\n\t\ttag.Dynamic(\"active-worker-count\", len(m.activeWorkers)),\n\t)\n}\n\nfunc (m *WorkerManager) startWorkerForDomain(domainName string) {\n\tw, err := m.createWorker(domainName)\n\tif err != nil {\n\t\tm.logger.Error(\"failed to start scheduler worker for domain\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t\ttag.Error(err),\n\t\t)\n\t\treturn\n\t}\n\n\tm.activeWorkers[domainName] = w\n\tm.logger.Info(\"started scheduler worker for domain\",\n\t\ttag.WorkflowDomainName(domainName),\n\t)\n}\n\nfunc (m *WorkerManager) defaultCreateWorker(domainName string) (workerHandle, error) {\n\tactCtx := context.WithValue(context.Background(), schedulerContextKey, schedulerContext{\n\t\tFrontendClient: m.frontendClient,\n\t})\n\n\tw := cadenceworker.New(m.serviceClient, domainName, TaskListName, cadenceworker.Options{\n\t\tBackgroundActivityContext: actCtx,\n\t})\n\tw.RegisterWorkflowWithOptions(SchedulerWorkflow, workflow.RegisterOptions{Name: WorkflowTypeName})\n\n\tif err := w.Start(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn w, nil\n}\n\nfunc (m *WorkerManager) stopAllWorkers() {\n\tfor domainName, w := range m.activeWorkers {\n\t\tw.Stop()\n\t\tm.logger.Info(\"stopped scheduler worker for domain\",\n\t\t\ttag.WorkflowDomainName(domainName),\n\t\t)\n\t\tdelete(m.activeWorkers, domainName)\n\t}\n}\n"
  },
  {
    "path": "service/worker/scheduler/client_worker_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/testlogger\"\n\t\"github.com/uber/cadence/common/membership\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n)\n\nfunc TestRefreshWorkers(t *testing.T) {\n\tselfHost := membership.NewDetailedHostInfo(\"10.0.0.1:7933\", \"self\", nil)\n\totherHost := membership.NewDetailedHostInfo(\"10.0.0.2:7933\", \"other\", nil)\n\n\tmakeDomainEntry := func(name string) *cache.DomainCacheEntry {\n\t\treturn cache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: name},\n\t\t\tnil, false, nil, 0, nil, 0, 0, 0,\n\t\t)\n\t}\n\n\ttests := []struct {\n\t\tname               string\n\t\tdomains            map[string]*cache.DomainCacheEntry\n\t\tlookupResults      map[string]membership.HostInfo\n\t\tlookupErrors       map[string]error\n\t\texistingWorkers    []string\n\t\twantActiveWorkers  []string\n\t\twantStoppedWorkers []string\n\t\twantStartedWorkers []string\n\t}{\n\t\t{\n\t\t\tname: \"starts workers for owned domains\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-a\": makeDomainEntry(\"domain-a\"),\n\t\t\t\t\"domain-b\": makeDomainEntry(\"domain-b\"),\n\t\t\t},\n\t\t\tlookupResults: map[string]membership.HostInfo{\n\t\t\t\t\"domain-a\": selfHost,\n\t\t\t\t\"domain-b\": selfHost,\n\t\t\t},\n\t\t\twantActiveWorkers:  []string{\"domain-a\", \"domain-b\"},\n\t\t\twantStartedWorkers: []string{\"domain-a\", \"domain-b\"},\n\t\t},\n\t\t{\n\t\t\tname: \"skips domains owned by other hosts\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-a\": makeDomainEntry(\"domain-a\"),\n\t\t\t\t\"domain-b\": makeDomainEntry(\"domain-b\"),\n\t\t\t},\n\t\t\tlookupResults: map[string]membership.HostInfo{\n\t\t\t\t\"domain-a\": selfHost,\n\t\t\t\t\"domain-b\": otherHost,\n\t\t\t},\n\t\t\twantActiveWorkers:  []string{\"domain-a\"},\n\t\t\twantStartedWorkers: []string{\"domain-a\"},\n\t\t},\n\t\t{\n\t\t\tname: \"stops workers for domains no longer owned\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-a\": makeDomainEntry(\"domain-a\"),\n\t\t\t},\n\t\t\tlookupResults: map[string]membership.HostInfo{\n\t\t\t\t\"domain-a\": otherHost,\n\t\t\t},\n\t\t\texistingWorkers:    []string{\"domain-a\"},\n\t\t\twantActiveWorkers:  []string{},\n\t\t\twantStoppedWorkers: []string{\"domain-a\"},\n\t\t},\n\t\t{\n\t\t\tname: \"stops workers for domains that disappeared from cache\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-b\": makeDomainEntry(\"domain-b\"),\n\t\t\t},\n\t\t\tlookupResults: map[string]membership.HostInfo{\n\t\t\t\t\"domain-b\": selfHost,\n\t\t\t},\n\t\t\texistingWorkers:    []string{\"domain-a\"},\n\t\t\twantActiveWorkers:  []string{\"domain-b\"},\n\t\t\twantStoppedWorkers: []string{\"domain-a\"},\n\t\t\twantStartedWorkers: []string{\"domain-b\"},\n\t\t},\n\t\t{\n\t\t\tname:              \"no domains means no workers\",\n\t\t\tdomains:           map[string]*cache.DomainCacheEntry{},\n\t\t\twantActiveWorkers: []string{},\n\t\t},\n\t\t{\n\t\t\tname: \"lookup error skips domain without stopping existing worker\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-a\": makeDomainEntry(\"domain-a\"),\n\t\t\t},\n\t\t\tlookupErrors: map[string]error{\n\t\t\t\t\"domain-a\": fmt.Errorf(\"ring not ready\"),\n\t\t\t},\n\t\t\texistingWorkers:   []string{\"domain-a\"},\n\t\t\twantActiveWorkers: []string{\"domain-a\"},\n\t\t},\n\t\t{\n\t\t\tname: \"does not restart already running worker\",\n\t\t\tdomains: map[string]*cache.DomainCacheEntry{\n\t\t\t\t\"domain-a\": makeDomainEntry(\"domain-a\"),\n\t\t\t},\n\t\t\tlookupResults: map[string]membership.HostInfo{\n\t\t\t\t\"domain-a\": selfHost,\n\t\t\t},\n\t\t\texistingWorkers:   []string{\"domain-a\"},\n\t\t\twantActiveWorkers: []string{\"domain-a\"},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\t\t\tmockDomainCache.EXPECT().GetAllDomain().Return(tc.domains)\n\n\t\t\tmockResolver := membership.NewMockResolver(ctrl)\n\t\t\tfor domainName, host := range tc.lookupResults {\n\t\t\t\tmockResolver.EXPECT().Lookup(service.Worker, domainName).Return(host, nil)\n\t\t\t}\n\t\t\tfor domainName, err := range tc.lookupErrors {\n\t\t\t\tmockResolver.EXPECT().Lookup(service.Worker, domainName).Return(membership.HostInfo{}, err)\n\t\t\t}\n\n\t\t\tstopped := make(map[string]bool)\n\t\t\tstarted := make(map[string]bool)\n\n\t\t\tctx, cancel := context.WithCancel(context.Background())\n\t\t\tdefer cancel()\n\n\t\t\twm := &WorkerManager{\n\t\t\t\tenabledFn:          dynamicproperties.GetBoolPropertyFn(true),\n\t\t\t\tlogger:             testlogger.New(t),\n\t\t\t\tdomainCache:        mockDomainCache,\n\t\t\t\tmembershipResolver: mockResolver,\n\t\t\t\thostInfo:           selfHost,\n\t\t\t\tactiveWorkers:      make(map[string]workerHandle),\n\t\t\t\tctx:                ctx,\n\t\t\t\tcreateWorker: func(domainName string) (workerHandle, error) {\n\t\t\t\t\tstarted[domainName] = true\n\t\t\t\t\treturn &fakeWorker{\n\t\t\t\t\t\tstopFn: func() { stopped[domainName] = true },\n\t\t\t\t\t}, nil\n\t\t\t\t},\n\t\t\t}\n\n\t\t\tfor _, d := range tc.existingWorkers {\n\t\t\t\tdomain := d\n\t\t\t\twm.activeWorkers[d] = &fakeWorker{\n\t\t\t\t\tstopFn: func() { stopped[domain] = true },\n\t\t\t\t}\n\t\t\t}\n\n\t\t\twm.refreshWorkers()\n\n\t\t\tassert.Equal(t, len(tc.wantActiveWorkers), len(wm.activeWorkers),\n\t\t\t\t\"active worker count mismatch\")\n\t\t\tfor _, d := range tc.wantActiveWorkers {\n\t\t\t\t_, exists := wm.activeWorkers[d]\n\t\t\t\tassert.True(t, exists, \"expected active worker for domain %s\", d)\n\t\t\t}\n\n\t\t\tfor _, d := range tc.wantStoppedWorkers {\n\t\t\t\tassert.True(t, stopped[d], \"expected worker for domain %s to be stopped\", d)\n\t\t\t}\n\n\t\t\tfor _, d := range tc.wantStartedWorkers {\n\t\t\t\tassert.True(t, started[d], \"expected worker for domain %s to be started\", d)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRefreshWorkersHandlesCreateWorkerError(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\tselfHost := membership.NewDetailedHostInfo(\"10.0.0.1:7933\", \"self\", nil)\n\n\tmockDomainCache := cache.NewMockDomainCache(ctrl)\n\tmockDomainCache.EXPECT().GetAllDomain().Return(map[string]*cache.DomainCacheEntry{\n\t\t\"domain-a\": cache.NewDomainCacheEntryForTest(\n\t\t\t&persistence.DomainInfo{Name: \"domain-a\"},\n\t\t\tnil, false, nil, 0, nil, 0, 0, 0,\n\t\t),\n\t})\n\n\tmockResolver := membership.NewMockResolver(ctrl)\n\tmockResolver.EXPECT().Lookup(service.Worker, \"domain-a\").Return(selfHost, nil)\n\n\tctx, cancel := context.WithCancel(context.Background())\n\tdefer cancel()\n\n\twm := &WorkerManager{\n\t\tenabledFn:          dynamicproperties.GetBoolPropertyFn(true),\n\t\tlogger:             testlogger.New(t),\n\t\tdomainCache:        mockDomainCache,\n\t\tmembershipResolver: mockResolver,\n\t\thostInfo:           selfHost,\n\t\tactiveWorkers:      make(map[string]workerHandle),\n\t\tctx:                ctx,\n\t\tcreateWorker: func(domainName string) (workerHandle, error) {\n\t\t\treturn nil, fmt.Errorf(\"connection refused\")\n\t\t},\n\t}\n\n\twm.refreshWorkers()\n\n\tassert.Empty(t, wm.activeWorkers, \"worker should not be added on creation error\")\n}\n\nfunc TestStopAllWorkers(t *testing.T) {\n\twm := &WorkerManager{\n\t\tlogger:        testlogger.New(t),\n\t\tactiveWorkers: make(map[string]workerHandle),\n\t}\n\n\tstoppedDomains := make(map[string]bool)\n\tfor _, d := range []string{\"domain-a\", \"domain-b\", \"domain-c\"} {\n\t\tdomain := d\n\t\twm.activeWorkers[d] = &fakeWorker{\n\t\t\tstopFn: func() { stoppedDomains[domain] = true },\n\t\t}\n\t}\n\n\twm.stopAllWorkers()\n\n\trequire.Empty(t, wm.activeWorkers)\n\tassert.True(t, stoppedDomains[\"domain-a\"])\n\tassert.True(t, stoppedDomains[\"domain-b\"])\n\tassert.True(t, stoppedDomains[\"domain-c\"])\n}\n\ntype fakeWorker struct {\n\tstopFn func()\n}\n\nfunc (f *fakeWorker) Stop() {\n\tif f.stopFn != nil {\n\t\tf.stopFn()\n\t}\n}\n"
  },
  {
    "path": "service/worker/scheduler/types.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tWorkflowTypeName = \"cadence-scheduler\"\n\tTaskListName     = \"cadence-scheduler\"\n\n\tSignalNamePause    = \"scheduler-pause\"\n\tSignalNameUnpause  = \"scheduler-unpause\"\n\tSignalNameUpdate   = \"scheduler-update\"\n\tSignalNameBackfill = \"scheduler-backfill\"\n\tSignalNameDelete   = \"scheduler-delete\"\n\n\tQueryTypeDescribe = \"scheduler-describe\"\n\n\tmaxIterationsBeforeContinueAsNew = 500\n\tmaxCatchUpFiresPerExecution      = 10\n\tmaxBackfillFiresPerExecution     = 10\n\tmaxPendingBackfills              = 10\n\n\tlocalActivityScheduleToCloseTimeout = 60 * time.Second\n\tlocalActivityMaxRetries             = 3\n\tlocalActivityRetryInitialInterval   = time.Second\n\tlocalActivityRetryMaxInterval       = 10 * time.Second\n)\n\n// SchedulerWorkflowInput is the input to the scheduler workflow.\n// It carries the schedule definition and any prior state (for ContinueAsNew).\ntype SchedulerWorkflowInput struct {\n\tDomain     string                 `json:\"domain\"`\n\tScheduleID string                 `json:\"scheduleId\"`\n\tSpec       types.ScheduleSpec     `json:\"spec\"`\n\tAction     types.ScheduleAction   `json:\"action\"`\n\tPolicies   types.SchedulePolicies `json:\"policies\"`\n\tState      SchedulerWorkflowState `json:\"state\"`\n}\n\n// SchedulerWorkflowState is the mutable runtime state that survives ContinueAsNew.\ntype SchedulerWorkflowState struct {\n\tPaused            bool              `json:\"paused\"`\n\tPauseReason       string            `json:\"pauseReason,omitempty\"`\n\tPausedBy          string            `json:\"pausedBy,omitempty\"`\n\tDeleted           bool              `json:\"-\"`                           // transient flag, not persisted across ContinueAsNew\n\tLastRunTime       time.Time         `json:\"lastRunTime,omitempty\"`       // last time a workflow was actually started\n\tLastProcessedTime time.Time         `json:\"lastProcessedTime,omitempty\"` // catch-up watermark: latest missed fire we've processed (fired or skipped)\n\tNextRunTime       time.Time         `json:\"nextRunTime,omitempty\"`\n\tTotalRuns         int64             `json:\"totalRuns\"`\n\tMissedRuns        int64             `json:\"missedRuns\"`\n\tSkippedRuns       int64             `json:\"skippedRuns\"`\n\tIterations        int               `json:\"iterations\"`\n\tBufferedRuns      int               `json:\"bufferedRuns\"`\n\tPendingBackfills  []BackfillRequest `json:\"pendingBackfills,omitempty\"`\n}\n\n// BackfillRequest is a queued backfill that persists across ContinueAsNew.\ntype BackfillRequest struct {\n\tStartTime     time.Time                   `json:\"startTime\"`\n\tEndTime       time.Time                   `json:\"endTime\"`\n\tOverlapPolicy types.ScheduleOverlapPolicy `json:\"overlapPolicy\"`\n\tBackfillID    string                      `json:\"backfillId,omitempty\"`\n}\n\n// PauseSignal is the payload sent with a pause signal.\ntype PauseSignal struct {\n\tReason   string `json:\"reason,omitempty\"`\n\tPausedBy string `json:\"pausedBy,omitempty\"`\n}\n\n// UnpauseSignal is the payload sent with an unpause signal.\ntype UnpauseSignal struct {\n\tReason        string                      `json:\"reason,omitempty\"`\n\tCatchUpPolicy types.ScheduleCatchUpPolicy `json:\"catchUpPolicy,omitempty\"`\n}\n\n// UpdateSignal is the payload sent with an update signal.\ntype UpdateSignal struct {\n\tSpec     *types.ScheduleSpec     `json:\"spec,omitempty\"`\n\tAction   *types.ScheduleAction   `json:\"action,omitempty\"`\n\tPolicies *types.SchedulePolicies `json:\"policies,omitempty\"`\n}\n\n// BackfillSignal is the payload sent with a backfill signal.\ntype BackfillSignal struct {\n\tStartTime     time.Time                   `json:\"startTime\"`\n\tEndTime       time.Time                   `json:\"endTime\"`\n\tOverlapPolicy types.ScheduleOverlapPolicy `json:\"overlapPolicy\"`\n\tBackfillID    string                      `json:\"backfillId,omitempty\"`\n}\n\n// ScheduleDescription is the query result returned by the describe query handler.\n// It provides a snapshot of the schedule's current configuration and runtime state.\ntype ScheduleDescription struct {\n\tScheduleID  string                 `json:\"scheduleId\"`\n\tDomain      string                 `json:\"domain\"`\n\tSpec        types.ScheduleSpec     `json:\"spec\"`\n\tAction      types.ScheduleAction   `json:\"action\"`\n\tPolicies    types.SchedulePolicies `json:\"policies\"`\n\tPaused      bool                   `json:\"paused\"`\n\tPauseReason string                 `json:\"pauseReason,omitempty\"`\n\tPausedBy    string                 `json:\"pausedBy,omitempty\"`\n\tLastRunTime time.Time              `json:\"lastRunTime,omitempty\"`\n\tNextRunTime time.Time              `json:\"nextRunTime,omitempty\"`\n\tTotalRuns   int64                  `json:\"totalRuns\"`\n\tMissedRuns  int64                  `json:\"missedRuns\"`\n\tSkippedRuns int64                  `json:\"skippedRuns\"`\n}\n\n// TriggerSource identifies what caused a schedule fire, used to differentiate\n// RequestIDs so that e.g. a backfill for the same timestamp as a normal fire\n// does not collide in server-side deduplication.\ntype TriggerSource string\n\nconst (\n\tTriggerSourceSchedule TriggerSource = \"schedule\"\n\tTriggerSourceBackfill TriggerSource = \"backfill\"\n)\n\n// StartWorkflowRequest is the input to the start-workflow activity.\ntype StartWorkflowRequest struct {\n\tDomain        string                    `json:\"domain\"`\n\tScheduleID    string                    `json:\"scheduleId\"`\n\tAction        types.StartWorkflowAction `json:\"action\"`\n\tScheduledTime time.Time                 `json:\"scheduledTime\"`\n\tTriggerSource TriggerSource             `json:\"triggerSource\"`\n}\n\n// StartWorkflowResult is the output of the start-workflow activity.\ntype StartWorkflowResult struct {\n\tWorkflowID string `json:\"workflowId\"`\n\tRunID      string `json:\"runId\"`\n\tStarted    bool   `json:\"started\"`\n\tSkipped    bool   `json:\"skipped\"`\n}\n"
  },
  {
    "path": "service/worker/scheduler/workflow.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"fmt\"\n\t\"time\"\n\n\t\"github.com/robfig/cron/v3\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// signalChannels bundles the signal channels used by the scheduler workflow\ntype signalChannels struct {\n\tpause    workflow.Channel\n\tunpause  workflow.Channel\n\tupdate   workflow.Channel\n\tbackfill workflow.Channel\n\tdelete   workflow.Channel\n}\n\n// SchedulerWorkflow is a long-running workflow that manages a single schedule.\n// It computes the next fire time from the cron expression, waits via a timer,\n// and dispatches the configured action. Signals control pause/unpause, update,\n// backfill, and deletion.\n//\n// The main loop follows a state-machine pattern: all inputs (signals and timer)\n// uniformly mutate state, and then a single decision point inspects the resulting\n// state to determine what to do next. ContinueAsNew is triggered on any\n// state-changing signal (pause, unpause, update) so the new execution's input\n// is always the authoritative source of truth.\nfunc SchedulerWorkflow(ctx workflow.Context, input SchedulerWorkflowInput) error {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Info(\"scheduler workflow started\",\n\t\tzap.String(\"domain\", input.Domain),\n\t\tzap.String(\"scheduleId\", input.ScheduleID),\n\t\tzap.Bool(\"paused\", input.State.Paused),\n\t)\n\n\tstate := &input.State\n\n\terr := workflow.SetQueryHandler(ctx, QueryTypeDescribe, func() (*ScheduleDescription, error) {\n\t\treturn buildScheduleDescription(&input, state), nil\n\t})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to register query handler: %w\", err)\n\t}\n\n\tchs := signalChannels{\n\t\tpause:    workflow.GetSignalChannel(ctx, SignalNamePause),\n\t\tunpause:  workflow.GetSignalChannel(ctx, SignalNameUnpause),\n\t\tupdate:   workflow.GetSignalChannel(ctx, SignalNameUpdate),\n\t\tbackfill: workflow.GetSignalChannel(ctx, SignalNameBackfill),\n\t\tdelete:   workflow.GetSignalChannel(ctx, SignalNameDelete),\n\t}\n\n\tsched, err := cron.ParseStandard(input.Spec.CronExpression)\n\tif err != nil {\n\t\tlogger.Error(\"invalid cron expression, terminating\", zap.String(\"cron\", input.Spec.CronExpression), zap.Error(err))\n\t\treturn fmt.Errorf(\"invalid cron expression %q: %w\", input.Spec.CronExpression, err)\n\t}\n\n\t// On the first iteration (after ContinueAsNew or fresh start), check for\n\t// fires that were missed during the transition gap or prior pause period.\n\t// Subsequent iterations don't need this because the timer handles fire times.\n\t// If more missed fires remain beyond the per-execution cap, ContinueAsNew\n\t// immediately so each batch runs in its own decision task.\n\tif moreMissed := processMissedRuns(ctx, logger, sched, &input, state); moreMissed {\n\t\treturn safeContinueAsNew(ctx, logger, chs.delete, input, state)\n\t}\n\n\t// Process any pending backfill requests carried over from a previous execution.\n\tif moreBackfills := processBackfills(ctx, logger, sched, &input, state); moreBackfills {\n\t\treturn safeContinueAsNew(ctx, logger, chs.delete, input, state)\n\t}\n\n\tfor {\n\t\tstate.Iterations++\n\n\t\t// Set up timer only when not paused. When paused, applyAllInputs\n\t\t// blocks on signals alone until an unpause or delete arrives.\n\t\tvar timerFuture workflow.Future\n\t\tvar timerCancel func()\n\t\tif !state.Paused {\n\t\t\tnow := workflow.Now(ctx)\n\t\t\tnextRun := computeNextRunTime(sched, now, input.Spec)\n\t\t\tif nextRun.IsZero() {\n\t\t\t\tlogger.Info(\"schedule has no more runs (past end time), completing\")\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\tstate.NextRunTime = nextRun\n\n\t\t\tdur := nextRun.Sub(now)\n\t\t\tif dur < 0 {\n\t\t\t\tdur = 0\n\t\t\t}\n\t\t\tvar timerCtx workflow.Context\n\t\t\ttimerCtx, timerCancel = workflow.WithCancel(ctx)\n\t\t\ttimerFuture = workflow.NewTimer(timerCtx, dur)\n\t\t}\n\n\t\tchanged, timerFired := applyAllInputs(ctx, logger, timerFuture, chs, state, &input)\n\n\t\tif timerCancel != nil {\n\t\t\ttimerCancel()\n\t\t}\n\n\t\tif state.Deleted {\n\t\t\tlogger.Info(\"schedule deleted\")\n\t\t\treturn nil\n\t\t}\n\n\t\tif timerFired && !state.Paused {\n\t\t\tprocessScheduleFire(ctx, logger, &input, state, state.NextRunTime, TriggerSourceSchedule)\n\t\t}\n\n\t\tif changed || state.Iterations >= maxIterationsBeforeContinueAsNew {\n\t\t\treturn safeContinueAsNew(ctx, logger, chs.delete, input, state)\n\t\t}\n\t}\n}\n\n// applyAllInputs blocks until at least one input (signal or timer) arrives,\n// processes it, then drains any remaining buffered signals.\n// Signals and the timer are treated uniformly: each mutates state without\n// triggering side effects (no timer cancellation, no ContinueAsNew).\n// Returns (stateChanged, timerFired): stateChanged is true if a state-changing\n// signal (pause, unpause, update) was received; timerFired is true if the timer\n// completed successfully.\nfunc applyAllInputs(\n\tctx workflow.Context,\n\tlogger *zap.Logger,\n\ttimerFuture workflow.Future,\n\tchs signalChannels,\n\tstate *SchedulerWorkflowState,\n\tinput *SchedulerWorkflowInput,\n) (bool, bool) {\n\tselector := workflow.NewSelector(ctx)\n\tstateChanged := false\n\n\ttimerFired := false\n\tif timerFuture != nil {\n\t\tselector.AddFuture(timerFuture, func(f workflow.Future) {\n\t\t\tif f.Get(ctx, nil) != nil {\n\t\t\t\treturn\n\t\t\t}\n\t\t\ttimerFired = true\n\t\t})\n\t}\n\n\tselector.AddReceive(chs.pause, func(c workflow.Channel, more bool) {\n\t\tvar sig PauseSignal\n\t\tc.Receive(ctx, &sig)\n\t\tif handlePause(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t})\n\n\tselector.AddReceive(chs.unpause, func(c workflow.Channel, more bool) {\n\t\tvar sig UnpauseSignal\n\t\tc.Receive(ctx, &sig)\n\t\tif handleUnpause(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t})\n\n\tselector.AddReceive(chs.update, func(c workflow.Channel, more bool) {\n\t\tvar sig UpdateSignal\n\t\tc.Receive(ctx, &sig)\n\t\tif handleUpdate(logger, sig, input, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t})\n\n\tselector.AddReceive(chs.backfill, func(c workflow.Channel, more bool) {\n\t\tvar sig BackfillSignal\n\t\tc.Receive(ctx, &sig)\n\t\tif handleBackfill(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t})\n\n\tselector.AddReceive(chs.delete, func(c workflow.Channel, more bool) {\n\t\tc.Receive(ctx, nil)\n\t\tstate.Deleted = true\n\t})\n\n\tselector.Select(ctx)\n\n\tif drainBufferedSignals(logger, chs, state, input) {\n\t\tstateChanged = true\n\t}\n\n\treturn stateChanged, timerFired\n}\n\n// drainBufferedSignals processes any remaining buffered signals without blocking.\n// Delete signals are checked first to prevent signal loss across ContinueAsNew boundaries.\n// Returns true if a state-changing signal was found.\nfunc drainBufferedSignals(\n\tlogger *zap.Logger,\n\tchs signalChannels,\n\tstate *SchedulerWorkflowState,\n\tinput *SchedulerWorkflowInput,\n) bool {\n\tif chs.delete.ReceiveAsync(nil) {\n\t\tstate.Deleted = true\n\t\treturn false\n\t}\n\n\tstateChanged := false\n\tfor {\n\t\tvar sig PauseSignal\n\t\tif !chs.pause.ReceiveAsync(&sig) {\n\t\t\tbreak\n\t\t}\n\t\tif handlePause(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t}\n\tfor {\n\t\tvar sig UnpauseSignal\n\t\tif !chs.unpause.ReceiveAsync(&sig) {\n\t\t\tbreak\n\t\t}\n\t\tif handleUnpause(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t}\n\tfor {\n\t\tvar sig UpdateSignal\n\t\tif !chs.update.ReceiveAsync(&sig) {\n\t\t\tbreak\n\t\t}\n\t\tif handleUpdate(logger, sig, input, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t}\n\tfor {\n\t\tvar sig BackfillSignal\n\t\tif !chs.backfill.ReceiveAsync(&sig) {\n\t\t\tbreak\n\t\t}\n\t\tif handleBackfill(logger, sig, state) {\n\t\t\tstateChanged = true\n\t\t}\n\t}\n\n\treturn stateChanged\n}\n\nfunc handlePause(logger *zap.Logger, sig PauseSignal, state *SchedulerWorkflowState) bool {\n\tstate.Paused = true\n\tstate.PauseReason = sig.Reason\n\tstate.PausedBy = sig.PausedBy\n\tlogger.Info(\"schedule paused\", zap.String(\"reason\", sig.Reason), zap.String(\"pausedBy\", sig.PausedBy))\n\treturn true\n}\n\nfunc handleUnpause(logger *zap.Logger, sig UnpauseSignal, state *SchedulerWorkflowState) bool {\n\tif !state.Paused {\n\t\tlogger.Info(\"ignoring unpause signal, schedule is not paused\")\n\t\treturn false\n\t}\n\tstate.Paused = false\n\tstate.PauseReason = \"\"\n\tstate.PausedBy = \"\"\n\tlogger.Info(\"schedule unpaused\", zap.String(\"reason\", sig.Reason), zap.String(\"catchUpPolicy\", sig.CatchUpPolicy.String()))\n\treturn true\n}\n\nfunc handleUpdate(logger *zap.Logger, sig UpdateSignal, input *SchedulerWorkflowInput, state *SchedulerWorkflowState) bool {\n\tif sig.Spec == nil && sig.Action == nil && sig.Policies == nil {\n\t\tlogger.Info(\"ignoring empty update signal\")\n\t\treturn false\n\t}\n\tchanged := false\n\tif sig.Spec != nil {\n\t\tif _, err := cron.ParseStandard(sig.Spec.CronExpression); err != nil {\n\t\t\tlogger.Error(\"ignoring update with invalid cron expression\",\n\t\t\t\tzap.String(\"cron\", sig.Spec.CronExpression), zap.Error(err))\n\t\t} else {\n\t\t\tinput.Spec = *sig.Spec\n\t\t\tchanged = true\n\t\t\tif len(state.PendingBackfills) > 0 {\n\t\t\t\tlogger.Warn(\"spec change cleared pending backfills\",\n\t\t\t\t\tzap.Int(\"clearedCount\", len(state.PendingBackfills)))\n\t\t\t\tstate.PendingBackfills = nil\n\t\t\t}\n\t\t}\n\t}\n\tif sig.Action != nil {\n\t\tinput.Action = *sig.Action\n\t\tchanged = true\n\t}\n\tif sig.Policies != nil {\n\t\tinput.Policies = *sig.Policies\n\t\tchanged = true\n\t}\n\tif changed {\n\t\tlogger.Info(\"schedule updated\")\n\t}\n\treturn changed\n}\n\nfunc handleBackfill(logger *zap.Logger, sig BackfillSignal, state *SchedulerWorkflowState) bool {\n\tif !sig.EndTime.After(sig.StartTime) {\n\t\tlogger.Warn(\"ignoring backfill with invalid time range\",\n\t\t\tzap.Time(\"startTime\", sig.StartTime),\n\t\t\tzap.Time(\"endTime\", sig.EndTime),\n\t\t)\n\t\treturn false\n\t}\n\tif len(state.PendingBackfills) >= maxPendingBackfills {\n\t\tlogger.Warn(\"ignoring backfill: pending backfill queue is full\",\n\t\t\tzap.String(\"backfillId\", sig.BackfillID),\n\t\t\tzap.Int(\"queueSize\", len(state.PendingBackfills)),\n\t\t\tzap.Int(\"maxPendingBackfills\", maxPendingBackfills),\n\t\t)\n\t\treturn false\n\t}\n\tfor _, existing := range state.PendingBackfills {\n\t\tif sig.StartTime.Before(existing.EndTime) && sig.EndTime.After(existing.StartTime) {\n\t\t\tlogger.Warn(\"backfill window overlaps with pending backfill, fires for overlapping times will be deduplicated\",\n\t\t\t\tzap.String(\"newBackfillId\", sig.BackfillID),\n\t\t\t\tzap.String(\"existingBackfillId\", existing.BackfillID),\n\t\t\t\tzap.Time(\"overlapStart\", maxTime(sig.StartTime, existing.StartTime)),\n\t\t\t\tzap.Time(\"overlapEnd\", minTime(sig.EndTime, existing.EndTime)),\n\t\t\t)\n\t\t}\n\t}\n\tstate.PendingBackfills = append(state.PendingBackfills, BackfillRequest{\n\t\tStartTime:     sig.StartTime,\n\t\tEndTime:       sig.EndTime,\n\t\tOverlapPolicy: sig.OverlapPolicy,\n\t\tBackfillID:    sig.BackfillID,\n\t})\n\tlogger.Info(\"backfill queued\",\n\t\tzap.Time(\"startTime\", sig.StartTime),\n\t\tzap.Time(\"endTime\", sig.EndTime),\n\t\tzap.String(\"overlapPolicy\", sig.OverlapPolicy.String()),\n\t\tzap.String(\"backfillId\", sig.BackfillID),\n\t\tzap.Int(\"pendingCount\", len(state.PendingBackfills)),\n\t)\n\treturn true\n}\n\n// processScheduleFire executes the configured action for a single schedule fire.\n// It calls the start-workflow activity, updates state counters, and logs the outcome.\n// Activity failures do not terminate the schedule, they are logged and counted as missed runs.\nfunc processScheduleFire(ctx workflow.Context, logger *zap.Logger, input *SchedulerWorkflowInput, state *SchedulerWorkflowState, scheduledTime time.Time, trigger TriggerSource) {\n\tstate.LastRunTime = scheduledTime\n\tstate.TotalRuns++\n\n\tlogger.Info(\"schedule fired\",\n\t\tzap.Time(\"scheduledTime\", scheduledTime),\n\t\tzap.Int64(\"totalRuns\", state.TotalRuns),\n\t)\n\n\tactivityOpts := workflow.LocalActivityOptions{\n\t\tScheduleToCloseTimeout: localActivityScheduleToCloseTimeout,\n\t\tRetryPolicy: &workflow.RetryPolicy{\n\t\t\tInitialInterval:    localActivityRetryInitialInterval,\n\t\t\tMaximumInterval:    localActivityRetryMaxInterval,\n\t\t\tMaximumAttempts:    localActivityMaxRetries,\n\t\t\tBackoffCoefficient: 2,\n\t\t},\n\t}\n\tactCtx := workflow.WithLocalActivityOptions(ctx, activityOpts)\n\n\tif input.Action.StartWorkflow == nil {\n\t\tstate.MissedRuns++\n\t\tlogger.Error(\"schedule action has no StartWorkflow configuration\")\n\t\treturn\n\t}\n\n\treq := StartWorkflowRequest{\n\t\tDomain:        input.Domain,\n\t\tScheduleID:    input.ScheduleID,\n\t\tAction:        *input.Action.StartWorkflow,\n\t\tScheduledTime: scheduledTime,\n\t\tTriggerSource: trigger,\n\t}\n\n\tvar result StartWorkflowResult\n\terr := workflow.ExecuteLocalActivity(actCtx, startWorkflowActivity, req).Get(ctx, &result)\n\tif err != nil {\n\t\tstate.MissedRuns++\n\t\tlogger.Error(\"scheduled action failed\",\n\t\t\tzap.Time(\"scheduledTime\", scheduledTime),\n\t\t\tzap.Error(err),\n\t\t)\n\t\treturn\n\t}\n\n\tif result.Skipped {\n\t\tstate.SkippedRuns++\n\t\tlogger.Info(\"scheduled action skipped (already running)\",\n\t\t\tzap.String(\"workflowId\", result.WorkflowID),\n\t\t)\n\t\treturn\n\t}\n\n\tlogger.Info(\"scheduled workflow started\",\n\t\tzap.String(\"workflowId\", result.WorkflowID),\n\t\tzap.String(\"runId\", result.RunID),\n\t)\n}\n\n// computeNextRunTime determines the next fire time for the cron schedule,\n// respecting the spec's StartTime and EndTime boundaries.\nfunc computeNextRunTime(sched cron.Schedule, now time.Time, spec types.ScheduleSpec) time.Time {\n\tif !spec.StartTime.IsZero() && now.Before(spec.StartTime) {\n\t\tnow = spec.StartTime.Add(-time.Second)\n\t}\n\tnext := sched.Next(now)\n\tif !spec.EndTime.IsZero() && next.After(spec.EndTime) {\n\t\treturn time.Time{}\n\t}\n\treturn next\n}\n\n// missedFiresResult holds the output of computeMissedFireTimes.\ntype missedFiresResult struct {\n\ttimes     []time.Time\n\ttruncated bool // true if the result was capped at maxCatchUpFires\n}\n\n// computeMissedFireTimes returns all cron fire times between (lastRun, now].\n// It caps the result at maxCatchUpFires to prevent unbounded iteration\n// for very frequent schedules that were paused for a long time.\n// The truncated flag signals that more fires exist beyond the cap.\nfunc computeMissedFireTimes(sched cron.Schedule, lastRun, now time.Time, spec types.ScheduleSpec) missedFiresResult {\n\tconst maxCatchUpFires = 1000\n\tvar missed []time.Time\n\tt := lastRun\n\tfor len(missed) < maxCatchUpFires {\n\t\tnext := computeNextRunTime(sched, t, spec)\n\t\tif next.IsZero() || next.After(now) {\n\t\t\treturn missedFiresResult{times: missed, truncated: false}\n\t\t}\n\t\tmissed = append(missed, next)\n\t\tt = next\n\t}\n\treturn missedFiresResult{times: missed, truncated: true}\n}\n\n// missedRunPolicyResult is the output of applyMissedRunPolicy.\ntype missedRunPolicyResult struct {\n\ttoFire  []time.Time // fire times to execute, in order\n\tskipped int64       // fires to count as skipped\n}\n\n// applyMissedRunPolicy is a pure function that determines which missed fires\n// to execute and how many to skip, given the catch-up policy and window.\n// It is separated from processMissedRuns to allow direct unit testing.\nfunc applyMissedRunPolicy(policy types.ScheduleCatchUpPolicy, window time.Duration, missed []time.Time, now time.Time, logger *zap.Logger) missedRunPolicyResult {\n\tvar eligible []time.Time\n\tfor _, t := range missed {\n\t\tif window <= 0 || now.Sub(t) <= window {\n\t\t\teligible = append(eligible, t)\n\t\t}\n\t}\n\toutOfWindow := int64(len(missed) - len(eligible))\n\n\tswitch policy {\n\tcase types.ScheduleCatchUpPolicyOne:\n\t\tif len(eligible) == 0 {\n\t\t\treturn missedRunPolicyResult{skipped: int64(len(missed))}\n\t\t}\n\t\treturn missedRunPolicyResult{\n\t\t\ttoFire:  []time.Time{eligible[len(eligible)-1]},\n\t\t\tskipped: outOfWindow + int64(len(eligible)-1),\n\t\t}\n\tcase types.ScheduleCatchUpPolicyAll:\n\t\treturn missedRunPolicyResult{\n\t\t\ttoFire:  eligible,\n\t\t\tskipped: outOfWindow,\n\t\t}\n\tcase types.ScheduleCatchUpPolicySkip:\n\t\treturn missedRunPolicyResult{skipped: int64(len(missed))}\n\tdefault:\n\t\tlogger.Warn(\"unknown catch-up policy, defaulting to skip\",\n\t\t\tzap.Int32(\"policy\", int32(policy)),\n\t\t)\n\t\treturn missedRunPolicyResult{skipped: int64(len(missed))}\n\t}\n}\n\n// processMissedRuns checks for and processes any cron fires that were missed\n// while the schedule was paused or during ContinueAsNew transitions.\n// The catch-up policy determines how missed fires are handled:\n//   - Skip: all missed fires are counted as skipped\n//   - One: only the most recent eligible fire (within CatchUpWindow) is executed\n//   - All: all eligible fires within the CatchUpWindow are executed\n//\n// To avoid exceeding the decision task timeout, at most maxCatchUpFiresPerExecution\n// fires are executed per workflow execution. Returns true if there are more missed\n// fires remaining, signalling the caller to ContinueAsNew for the next batch.\nfunc processMissedRuns(ctx workflow.Context, logger *zap.Logger, sched cron.Schedule, input *SchedulerWorkflowInput, state *SchedulerWorkflowState) bool {\n\t// Use LastProcessedTime as the catch-up watermark; fall back to\n\t// LastRunTime for schedules created before this field existed.\n\twatermark := state.LastProcessedTime\n\tif watermark.IsZero() {\n\t\twatermark = state.LastRunTime\n\t}\n\tif state.Paused || watermark.IsZero() {\n\t\treturn false\n\t}\n\tnow := workflow.Now(ctx)\n\tfires := computeMissedFireTimes(sched, watermark, now, input.Spec)\n\tif len(fires.times) == 0 {\n\t\treturn false\n\t}\n\n\tif fires.truncated {\n\t\tlogger.Warn(\"missed fires truncated, remaining will be caught up after ContinueAsNew\",\n\t\t\tzap.Int(\"count\", len(fires.times)),\n\t\t\tzap.Time(\"lastProcessedTime\", watermark),\n\t\t\tzap.Time(\"now\", now),\n\t\t)\n\t}\n\n\tresult := applyMissedRunPolicy(input.Policies.CatchUpPolicy, input.Policies.CatchUpWindow, fires.times, now, logger)\n\n\tfired := 0\n\tfor _, t := range result.toFire {\n\t\tif fired >= maxCatchUpFiresPerExecution {\n\t\t\tbreak\n\t\t}\n\t\tprocessScheduleFire(ctx, logger, input, state, t, TriggerSourceSchedule)\n\t\tfired++\n\t}\n\tunfired := int64(len(result.toFire) - fired)\n\n\tif result.skipped > 0 {\n\t\tstate.SkippedRuns += result.skipped\n\t\tlogger.Info(\"catch-up skipped missed fires\",\n\t\t\tzap.Int64(\"skipped\", result.skipped),\n\t\t\tzap.Int(\"total_missed\", len(fires.times)),\n\t\t\tzap.String(\"policy\", input.Policies.CatchUpPolicy.String()),\n\t\t)\n\t}\n\n\t// Advance watermark past all fires we've fully processed (fired or\n\t// skipped) to avoid re-discovering them after ContinueAsNew.\n\t// If we capped fires via maxCatchUpFiresPerExecution, only advance\n\t// to the last one we actually fired so the rest are retried.\n\tif unfired > 0 {\n\t\tstate.LastProcessedTime = result.toFire[fired-1]\n\t} else if last := fires.times[len(fires.times)-1]; last.After(state.LastProcessedTime) {\n\t\tstate.LastProcessedTime = last\n\t}\n\n\treturn unfired > 0 || fires.truncated\n}\n\n// processBackfills drains pending backfill requests from state, computing\n// cron fire times for each request's time range and executing them.\n// Like processMissedRuns, it caps fires per execution and returns true\n// if more work remains (signalling the caller to ContinueAsNew).\nfunc processBackfills(ctx workflow.Context, logger *zap.Logger, sched cron.Schedule, input *SchedulerWorkflowInput, state *SchedulerWorkflowState) bool {\n\t// Backfills respect the pause state: an explicit user request to replay a time\n\t// range should not fire workflows while the schedule is paused. The pending\n\t// backfills are preserved in state and will execute once the schedule is unpaused.\n\tif state.Paused || len(state.PendingBackfills) == 0 {\n\t\treturn false\n\t}\n\n\tfired := 0\n\tfor len(state.PendingBackfills) > 0 {\n\t\tbf := &state.PendingBackfills[0]\n\n\t\tfires := computeMissedFireTimes(sched, bf.StartTime.Add(-time.Second), bf.EndTime, input.Spec)\n\n\t\tfor _, t := range fires.times {\n\t\t\tif fired >= maxBackfillFiresPerExecution {\n\t\t\t\tbf.StartTime = t\n\t\t\t\tlogger.Info(\"backfill batch cap reached, continuing after ContinueAsNew\",\n\t\t\t\t\tzap.String(\"backfillId\", bf.BackfillID),\n\t\t\t\t\tzap.Time(\"resumeFrom\", t),\n\t\t\t\t\tzap.Int(\"firedThisBatch\", fired),\n\t\t\t\t)\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tprocessScheduleFire(ctx, logger, input, state, t, TriggerSourceBackfill)\n\t\t\tfired++\n\t\t}\n\n\t\tif fires.truncated {\n\t\t\t// More fires exist beyond the 1000-fire scan cap.\n\t\t\t// Advance start past the last processed fire so it isn't replayed.\n\t\t\tif len(fires.times) > 0 {\n\t\t\t\tbf.StartTime = fires.times[len(fires.times)-1].Add(time.Second)\n\t\t\t}\n\t\t\tlogger.Info(\"backfill range has more fires beyond scan cap, continuing after ContinueAsNew\",\n\t\t\t\tzap.String(\"backfillId\", bf.BackfillID),\n\t\t\t\tzap.Int(\"firedThisBatch\", fired),\n\t\t\t)\n\t\t\treturn true\n\t\t}\n\n\t\tlogger.Info(\"backfill completed\",\n\t\t\tzap.String(\"backfillId\", bf.BackfillID),\n\t\t\tzap.Int(\"firedTotal\", fired),\n\t\t)\n\t\tstate.PendingBackfills = state.PendingBackfills[1:]\n\t}\n\n\treturn false\n}\n\n// buildScheduleDescription creates a snapshot of the current schedule\n// configuration and runtime state for the describe query handler.\nfunc buildScheduleDescription(input *SchedulerWorkflowInput, state *SchedulerWorkflowState) *ScheduleDescription {\n\treturn &ScheduleDescription{\n\t\tScheduleID:  input.ScheduleID,\n\t\tDomain:      input.Domain,\n\t\tSpec:        input.Spec,\n\t\tAction:      input.Action,\n\t\tPolicies:    input.Policies,\n\t\tPaused:      state.Paused,\n\t\tPauseReason: state.PauseReason,\n\t\tPausedBy:    state.PausedBy,\n\t\tLastRunTime: state.LastRunTime,\n\t\tNextRunTime: state.NextRunTime,\n\t\tTotalRuns:   state.TotalRuns,\n\t\tMissedRuns:  state.MissedRuns,\n\t\tSkippedRuns: state.SkippedRuns,\n\t}\n}\n\n// safeContinueAsNew drains the delete channel before performing ContinueAsNew.\n// Buffered signals are not carried across ContinueAsNew boundaries, so a delete\n// signal that arrived alongside a state-changing signal would be lost without this check.\nfunc safeContinueAsNew(ctx workflow.Context, logger *zap.Logger, deleteCh workflow.Channel, input SchedulerWorkflowInput, state *SchedulerWorkflowState) error {\n\tif deleteCh.ReceiveAsync(nil) {\n\t\tlogger.Info(\"schedule deleted (caught before ContinueAsNew)\")\n\t\treturn nil\n\t}\n\tstate.Iterations = 0\n\tinput.State = *state\n\treturn workflow.NewContinueAsNewError(ctx, WorkflowTypeName, input)\n}\n\nfunc minTime(a, b time.Time) time.Time {\n\tif a.Before(b) {\n\t\treturn a\n\t}\n\treturn b\n}\n\nfunc maxTime(a, b time.Time) time.Time {\n\tif a.After(b) {\n\t\treturn a\n\t}\n\treturn b\n}\n"
  },
  {
    "path": "service/worker/scheduler/workflow_test.go",
    "content": "// Copyright (c) 2026 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage scheduler\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/robfig/cron/v3\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nvar testLogger = zap.NewNop()\n\nfunc mustParseCron(t *testing.T, expr string) cron.Schedule {\n\tt.Helper()\n\ts, err := cron.ParseStandard(expr)\n\trequire.NoError(t, err)\n\treturn s\n}\n\nfunc TestComputeNextRunTime(t *testing.T) {\n\tnow := time.Date(2026, 1, 15, 10, 30, 0, 0, time.UTC)\n\n\ttests := []struct {\n\t\tname     string\n\t\tcron     string\n\t\tnow      time.Time\n\t\tspec     types.ScheduleSpec\n\t\twantZero bool\n\t\twantTime time.Time\n\t}{\n\t\t{\n\t\t\tname:     \"every hour - next on the hour\",\n\t\t\tcron:     \"0 * * * *\",\n\t\t\tnow:      now,\n\t\t\tspec:     types.ScheduleSpec{},\n\t\t\twantTime: time.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:     \"every day at midnight\",\n\t\t\tcron:     \"0 0 * * *\",\n\t\t\tnow:      now,\n\t\t\tspec:     types.ScheduleSpec{},\n\t\t\twantTime: time.Date(2026, 1, 16, 0, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:     \"now is before startTime - uses startTime as base\",\n\t\t\tcron:     \"0 * * * *\",\n\t\t\tnow:      time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\tspec:     types.ScheduleSpec{StartTime: time.Date(2026, 6, 1, 0, 0, 0, 0, time.UTC)},\n\t\t\twantTime: time.Date(2026, 6, 1, 0, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:     \"next run is after endTime - returns zero\",\n\t\t\tcron:     \"0 0 * * *\",\n\t\t\tnow:      time.Date(2026, 1, 15, 23, 0, 0, 0, time.UTC),\n\t\t\tspec:     types.ScheduleSpec{EndTime: time.Date(2026, 1, 15, 23, 59, 0, 0, time.UTC)},\n\t\t\twantZero: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"next run is before endTime - returns next\",\n\t\t\tcron:     \"0 * * * *\",\n\t\t\tnow:      now,\n\t\t\tspec:     types.ScheduleSpec{EndTime: time.Date(2026, 12, 31, 0, 0, 0, 0, time.UTC)},\n\t\t\twantTime: time.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname: \"startTime and endTime together - within window\",\n\t\t\tcron: \"0 * * * *\",\n\t\t\tnow:  time.Date(2026, 6, 1, 5, 30, 0, 0, time.UTC),\n\t\t\tspec: types.ScheduleSpec{\n\t\t\t\tStartTime: time.Date(2026, 6, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:   time.Date(2026, 6, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t\twantTime: time.Date(2026, 6, 1, 6, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:     \"every minute\",\n\t\t\tcron:     \"* * * * *\",\n\t\t\tnow:      now,\n\t\t\tspec:     types.ScheduleSpec{},\n\t\t\twantTime: time.Date(2026, 1, 15, 10, 31, 0, 0, time.UTC),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tsched := mustParseCron(t, tt.cron)\n\t\t\tgot := computeNextRunTime(sched, tt.now, tt.spec)\n\t\t\tif tt.wantZero {\n\t\t\t\tassert.True(t, got.IsZero(), \"expected zero time, got %v\", got)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.wantTime, got)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestComputeMissedFireTimes(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcron          string\n\t\tlastRun       time.Time\n\t\tnow           time.Time\n\t\tspec          types.ScheduleSpec\n\t\twantTimes     []time.Time\n\t\twantTruncated bool\n\t}{\n\t\t{\n\t\t\tname:      \"no missed fires - now is before next fire\",\n\t\t\tcron:      \"0 * * * *\",\n\t\t\tlastRun:   time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:       time.Date(2026, 1, 15, 10, 30, 0, 0, time.UTC),\n\t\t\twantTimes: nil,\n\t\t},\n\t\t{\n\t\t\tname:    \"one missed fire\",\n\t\t\tcron:    \"0 * * * *\",\n\t\t\tlastRun: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:     time.Date(2026, 1, 15, 11, 30, 0, 0, time.UTC),\n\t\t\twantTimes: []time.Time{\n\t\t\t\ttime.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"multiple missed fires\",\n\t\t\tcron:    \"0 * * * *\",\n\t\t\tlastRun: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:     time.Date(2026, 1, 15, 13, 30, 0, 0, time.UTC),\n\t\t\twantTimes: []time.Time{\n\t\t\t\ttime.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\t\ttime.Date(2026, 1, 15, 12, 0, 0, 0, time.UTC),\n\t\t\t\ttime.Date(2026, 1, 15, 13, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"missed fire exactly at now is included\",\n\t\t\tcron:    \"0 * * * *\",\n\t\t\tlastRun: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:     time.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\twantTimes: []time.Time{\n\t\t\t\ttime.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"respects endTime - no fires past end\",\n\t\t\tcron:    \"0 * * * *\",\n\t\t\tlastRun: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:     time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t\tspec:    types.ScheduleSpec{EndTime: time.Date(2026, 1, 15, 12, 30, 0, 0, time.UTC)},\n\t\t\twantTimes: []time.Time{\n\t\t\t\ttime.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\t\ttime.Date(2026, 1, 15, 12, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:      \"lastRun equals now - no missed fires\",\n\t\t\tcron:      \"0 * * * *\",\n\t\t\tlastRun:   time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tnow:       time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\twantTimes: nil,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tsched := mustParseCron(t, tt.cron)\n\t\t\tgot := computeMissedFireTimes(sched, tt.lastRun, tt.now, tt.spec)\n\t\t\tassert.Equal(t, tt.wantTimes, got.times)\n\t\t\tassert.Equal(t, tt.wantTruncated, got.truncated)\n\t\t})\n\t}\n}\n\nfunc TestCatchUpOrchestration(t *testing.T) {\n\tnow := time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC)\n\tlastProcessed := time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC)\n\tcronExpr := \"0 * * * *\"\n\n\ttests := []struct {\n\t\tname                   string\n\t\tpolicy                 types.ScheduleCatchUpPolicy\n\t\twindow                 time.Duration\n\t\twantFiredCount         int\n\t\twantSkipped            int64\n\t\twantLastProcessedAfter time.Time\n\t}{\n\t\t{\n\t\t\tname:                   \"Skip advances watermark past all missed, fires nothing\",\n\t\t\tpolicy:                 types.ScheduleCatchUpPolicySkip,\n\t\t\twantFiredCount:         0,\n\t\t\twantSkipped:            4, // 11:00, 12:00, 13:00, 14:00\n\t\t\twantLastProcessedAfter: time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:                   \"One fires most recent, skips rest, advances watermark\",\n\t\t\tpolicy:                 types.ScheduleCatchUpPolicyOne,\n\t\t\twantFiredCount:         1,\n\t\t\twantSkipped:            3, // 11:00, 12:00, 13:00 skipped\n\t\t\twantLastProcessedAfter: time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:                   \"All fires everything, skips nothing, advances watermark\",\n\t\t\tpolicy:                 types.ScheduleCatchUpPolicyAll,\n\t\t\twantFiredCount:         4,\n\t\t\twantSkipped:            0,\n\t\t\twantLastProcessedAfter: time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:                   \"One with window excludes old fires\",\n\t\t\tpolicy:                 types.ScheduleCatchUpPolicyOne,\n\t\t\twindow:                 90 * time.Minute,\n\t\t\twantFiredCount:         1,\n\t\t\twantSkipped:            3, // 11:00, 12:00 out of window + 13:00 skipped eligible\n\t\t\twantLastProcessedAfter: time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t},\n\t\t{\n\t\t\tname:                   \"All with tight window fires only recent\",\n\t\t\tpolicy:                 types.ScheduleCatchUpPolicyAll,\n\t\t\twindow:                 90 * time.Minute,\n\t\t\twantFiredCount:         2, // 13:00 and 14:00 within 90min of 14:00\n\t\t\twantSkipped:            2, // 11:00 and 12:00 out of window\n\t\t\twantLastProcessedAfter: time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tsched := mustParseCron(t, cronExpr)\n\t\t\tfires := computeMissedFireTimes(sched, lastProcessed, now, types.ScheduleSpec{})\n\t\t\trequire.False(t, fires.truncated)\n\t\t\trequire.Equal(t, 4, len(fires.times)) // 11:00, 12:00, 13:00, 14:00\n\n\t\t\tresult := applyMissedRunPolicy(tt.policy, tt.window, fires.times, now, testLogger)\n\t\t\tassert.Equal(t, tt.wantFiredCount, len(result.toFire), \"fired count\")\n\t\t\tassert.Equal(t, tt.wantSkipped, result.skipped, \"skipped count\")\n\n\t\t\tlastMissed := fires.times[len(fires.times)-1]\n\t\t\tassert.True(t, !lastMissed.Before(tt.wantLastProcessedAfter), \"watermark should advance to at least %v\", tt.wantLastProcessedAfter)\n\t\t})\n\t}\n}\n\nfunc TestApplyMissedRunPolicy(t *testing.T) {\n\tnow := time.Date(2026, 1, 15, 14, 0, 0, 0, time.UTC)\n\tfires := []time.Time{\n\t\ttime.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC), // 3h ago\n\t\ttime.Date(2026, 1, 15, 12, 0, 0, 0, time.UTC), // 2h ago\n\t\ttime.Date(2026, 1, 15, 13, 0, 0, 0, time.UTC), // 1h ago\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\tpolicy      types.ScheduleCatchUpPolicy\n\t\twindow      time.Duration\n\t\twantToFire  []time.Time\n\t\twantSkipped int64\n\t}{\n\t\t{\n\t\t\tname:        \"Skip - all missed fires are skipped\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicySkip,\n\t\t\twantToFire:  nil,\n\t\t\twantSkipped: 3,\n\t\t},\n\t\t{\n\t\t\tname:        \"Invalid (zero value) - defaults to skip\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyInvalid,\n\t\t\twantToFire:  nil,\n\t\t\twantSkipped: 3,\n\t\t},\n\t\t{\n\t\t\tname:        \"One - no window, fires most recent\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyOne,\n\t\t\twantToFire:  []time.Time{fires[2]},\n\t\t\twantSkipped: 2,\n\t\t},\n\t\t{\n\t\t\tname:        \"One - window excludes two oldest, fires most recent eligible\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyOne,\n\t\t\twindow:      90 * time.Minute,\n\t\t\twantToFire:  []time.Time{fires[2]}, // only 13:00 is within 90min of 14:00\n\t\t\twantSkipped: 2,                     // 2 out-of-window, 0 skipped eligible\n\t\t},\n\t\t{\n\t\t\tname:        \"One - window excludes all, nothing fired\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyOne,\n\t\t\twindow:      30 * time.Minute,\n\t\t\twantToFire:  nil,\n\t\t\twantSkipped: 3,\n\t\t},\n\t\t{\n\t\t\tname:        \"All - no window, fires all\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyAll,\n\t\t\twantToFire:  fires,\n\t\t\twantSkipped: 0,\n\t\t},\n\t\t{\n\t\t\tname:        \"All - window filters two oldest, fires most recent only\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyAll,\n\t\t\twindow:      90 * time.Minute,\n\t\t\twantToFire:  fires[2:], // only 13:00 is within 90min of 14:00\n\t\t\twantSkipped: 2,\n\t\t},\n\t\t{\n\t\t\tname:        \"All - window excludes all, nothing fired\",\n\t\t\tpolicy:      types.ScheduleCatchUpPolicyAll,\n\t\t\twindow:      30 * time.Minute,\n\t\t\twantToFire:  nil,\n\t\t\twantSkipped: 3,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := applyMissedRunPolicy(tt.policy, tt.window, fires, now, testLogger)\n\t\t\tassert.Equal(t, tt.wantToFire, got.toFire)\n\t\t\tassert.Equal(t, tt.wantSkipped, got.skipped)\n\t\t})\n\t}\n}\n\nfunc TestBuildScheduleDescription(t *testing.T) {\n\tlastRun := time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC)\n\tnextRun := time.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC)\n\n\ttests := []struct {\n\t\tname  string\n\t\tinput SchedulerWorkflowInput\n\t\tstate SchedulerWorkflowState\n\t\twant  *ScheduleDescription\n\t}{\n\t\t{\n\t\t\tname: \"running schedule with counters\",\n\t\t\tinput: SchedulerWorkflowInput{\n\t\t\t\tScheduleID: \"sched-1\",\n\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\tSpec:       types.ScheduleSpec{CronExpression: \"0 * * * *\"},\n\t\t\t\tAction: types.ScheduleAction{\n\t\t\t\t\tStartWorkflow: &types.StartWorkflowAction{\n\t\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"my-wf\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tPolicies: types.SchedulePolicies{OverlapPolicy: types.ScheduleOverlapPolicySkipNew},\n\t\t\t},\n\t\t\tstate: SchedulerWorkflowState{\n\t\t\t\tLastRunTime: lastRun,\n\t\t\t\tNextRunTime: nextRun,\n\t\t\t\tTotalRuns:   42,\n\t\t\t\tMissedRuns:  1,\n\t\t\t\tSkippedRuns: 3,\n\t\t\t},\n\t\t\twant: &ScheduleDescription{\n\t\t\t\tScheduleID: \"sched-1\",\n\t\t\t\tDomain:     \"test-domain\",\n\t\t\t\tSpec:       types.ScheduleSpec{CronExpression: \"0 * * * *\"},\n\t\t\t\tAction: types.ScheduleAction{\n\t\t\t\t\tStartWorkflow: &types.StartWorkflowAction{\n\t\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"my-wf\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tPolicies:    types.SchedulePolicies{OverlapPolicy: types.ScheduleOverlapPolicySkipNew},\n\t\t\t\tLastRunTime: lastRun,\n\t\t\t\tNextRunTime: nextRun,\n\t\t\t\tTotalRuns:   42,\n\t\t\t\tMissedRuns:  1,\n\t\t\t\tSkippedRuns: 3,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"paused schedule\",\n\t\t\tinput: SchedulerWorkflowInput{\n\t\t\t\tScheduleID: \"sched-2\",\n\t\t\t\tDomain:     \"prod\",\n\t\t\t\tSpec:       types.ScheduleSpec{CronExpression: \"0 0 * * *\"},\n\t\t\t},\n\t\t\tstate: SchedulerWorkflowState{\n\t\t\t\tPaused:      true,\n\t\t\t\tPauseReason: \"maintenance\",\n\t\t\t\tPausedBy:    \"admin@test.com\",\n\t\t\t\tTotalRuns:   10,\n\t\t\t},\n\t\t\twant: &ScheduleDescription{\n\t\t\t\tScheduleID:  \"sched-2\",\n\t\t\t\tDomain:      \"prod\",\n\t\t\t\tSpec:        types.ScheduleSpec{CronExpression: \"0 0 * * *\"},\n\t\t\t\tPaused:      true,\n\t\t\t\tPauseReason: \"maintenance\",\n\t\t\t\tPausedBy:    \"admin@test.com\",\n\t\t\t\tTotalRuns:   10,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:  \"fresh schedule with no runs\",\n\t\t\tinput: SchedulerWorkflowInput{ScheduleID: \"sched-new\", Domain: \"dev\"},\n\t\t\tstate: SchedulerWorkflowState{},\n\t\t\twant:  &ScheduleDescription{ScheduleID: \"sched-new\", Domain: \"dev\"},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tgot := buildScheduleDescription(&tt.input, &tt.state)\n\t\t\tassert.Equal(t, tt.want, got)\n\t\t})\n\t}\n}\n\nfunc TestHandlePause(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tinitial      SchedulerWorkflowState\n\t\tsig          PauseSignal\n\t\twantPaused   bool\n\t\twantReason   string\n\t\twantPausedBy string\n\t\twantChanged  bool\n\t}{\n\t\t{\n\t\t\tname:         \"pause from running\",\n\t\t\tinitial:      SchedulerWorkflowState{},\n\t\t\tsig:          PauseSignal{Reason: \"maintenance\", PausedBy: \"admin@test.com\"},\n\t\t\twantPaused:   true,\n\t\t\twantReason:   \"maintenance\",\n\t\t\twantPausedBy: \"admin@test.com\",\n\t\t\twantChanged:  true,\n\t\t},\n\t\t{\n\t\t\tname:         \"pause overwrites previous pause reason\",\n\t\t\tinitial:      SchedulerWorkflowState{Paused: true, PauseReason: \"old\", PausedBy: \"old-user\"},\n\t\t\tsig:          PauseSignal{Reason: \"new reason\", PausedBy: \"new-user\"},\n\t\t\twantPaused:   true,\n\t\t\twantReason:   \"new reason\",\n\t\t\twantPausedBy: \"new-user\",\n\t\t\twantChanged:  true,\n\t\t},\n\t\t{\n\t\t\tname:         \"pause with empty reason\",\n\t\t\tinitial:      SchedulerWorkflowState{},\n\t\t\tsig:          PauseSignal{},\n\t\t\twantPaused:   true,\n\t\t\twantReason:   \"\",\n\t\t\twantPausedBy: \"\",\n\t\t\twantChanged:  true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tstate := tt.initial\n\t\t\tchanged := handlePause(testLogger, tt.sig, &state)\n\t\t\tassert.Equal(t, tt.wantChanged, changed)\n\t\t\tassert.Equal(t, tt.wantPaused, state.Paused)\n\t\t\tassert.Equal(t, tt.wantReason, state.PauseReason)\n\t\t\tassert.Equal(t, tt.wantPausedBy, state.PausedBy)\n\t\t})\n\t}\n}\n\nfunc TestHandleUnpause(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tinitial      SchedulerWorkflowState\n\t\tsig          UnpauseSignal\n\t\twantPaused   bool\n\t\twantReason   string\n\t\twantPausedBy string\n\t\twantChanged  bool\n\t}{\n\t\t{\n\t\t\tname:         \"unpause from paused\",\n\t\t\tinitial:      SchedulerWorkflowState{Paused: true, PauseReason: \"maintenance\", PausedBy: \"admin\"},\n\t\t\tsig:          UnpauseSignal{Reason: \"maintenance done\"},\n\t\t\twantPaused:   false,\n\t\t\twantReason:   \"\",\n\t\t\twantPausedBy: \"\",\n\t\t\twantChanged:  true,\n\t\t},\n\t\t{\n\t\t\tname:         \"unpause when not paused is a no-op\",\n\t\t\tinitial:      SchedulerWorkflowState{Paused: false},\n\t\t\tsig:          UnpauseSignal{Reason: \"shouldn't matter\"},\n\t\t\twantPaused:   false,\n\t\t\twantReason:   \"\",\n\t\t\twantPausedBy: \"\",\n\t\t\twantChanged:  false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tstate := tt.initial\n\t\t\tchanged := handleUnpause(testLogger, tt.sig, &state)\n\t\t\tassert.Equal(t, tt.wantChanged, changed)\n\t\t\tassert.Equal(t, tt.wantPaused, state.Paused)\n\t\t\tassert.Equal(t, tt.wantReason, state.PauseReason)\n\t\t\tassert.Equal(t, tt.wantPausedBy, state.PausedBy)\n\t\t})\n\t}\n}\n\nfunc TestHandleUpdate(t *testing.T) {\n\toriginal := SchedulerWorkflowInput{\n\t\tDomain:     \"test-domain\",\n\t\tScheduleID: \"sched-1\",\n\t\tSpec:       types.ScheduleSpec{CronExpression: \"0 * * * *\"},\n\t\tAction: types.ScheduleAction{\n\t\t\tStartWorkflow: &types.StartWorkflowAction{\n\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"old-workflow\"},\n\t\t\t},\n\t\t},\n\t\tPolicies: types.SchedulePolicies{OverlapPolicy: types.ScheduleOverlapPolicySkipNew},\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\tsig         UpdateSignal\n\t\twantCron    string\n\t\twantWF      string\n\t\twantPol     types.ScheduleOverlapPolicy\n\t\twantChanged bool\n\t}{\n\t\t{\n\t\t\tname: \"update spec only\",\n\t\t\tsig: UpdateSignal{\n\t\t\t\tSpec: &types.ScheduleSpec{CronExpression: \"*/5 * * * *\"},\n\t\t\t},\n\t\t\twantCron:    \"*/5 * * * *\",\n\t\t\twantWF:      \"old-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicySkipNew,\n\t\t\twantChanged: true,\n\t\t},\n\t\t{\n\t\t\tname: \"update action only\",\n\t\t\tsig: UpdateSignal{\n\t\t\t\tAction: &types.ScheduleAction{\n\t\t\t\t\tStartWorkflow: &types.StartWorkflowAction{\n\t\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: \"new-workflow\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\twantCron:    \"0 * * * *\",\n\t\t\twantWF:      \"new-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicySkipNew,\n\t\t\twantChanged: true,\n\t\t},\n\t\t{\n\t\t\tname: \"update policies only\",\n\t\t\tsig: UpdateSignal{\n\t\t\t\tPolicies: &types.SchedulePolicies{OverlapPolicy: types.ScheduleOverlapPolicyConcurrent},\n\t\t\t},\n\t\t\twantCron:    \"0 * * * *\",\n\t\t\twantWF:      \"old-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicyConcurrent,\n\t\t\twantChanged: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"nil fields leave input unchanged\",\n\t\t\tsig:         UpdateSignal{},\n\t\t\twantCron:    \"0 * * * *\",\n\t\t\twantWF:      \"old-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicySkipNew,\n\t\t\twantChanged: false,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid cron expression is rejected, spec unchanged\",\n\t\t\tsig: UpdateSignal{\n\t\t\t\tSpec: &types.ScheduleSpec{CronExpression: \"not-a-cron\"},\n\t\t\t},\n\t\t\twantCron:    \"0 * * * *\",\n\t\t\twantWF:      \"old-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicySkipNew,\n\t\t\twantChanged: false,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid cron rejected but action and policies still applied\",\n\t\t\tsig: UpdateSignal{\n\t\t\t\tSpec:     &types.ScheduleSpec{CronExpression: \"bad\"},\n\t\t\t\tAction:   &types.ScheduleAction{StartWorkflow: &types.StartWorkflowAction{WorkflowType: &types.WorkflowType{Name: \"new-workflow\"}}},\n\t\t\t\tPolicies: &types.SchedulePolicies{OverlapPolicy: types.ScheduleOverlapPolicyConcurrent},\n\t\t\t},\n\t\t\twantCron:    \"0 * * * *\",\n\t\t\twantWF:      \"new-workflow\",\n\t\t\twantPol:     types.ScheduleOverlapPolicyConcurrent,\n\t\t\twantChanged: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tinput := original\n\t\t\tstate := &SchedulerWorkflowState{}\n\t\t\tchanged := handleUpdate(testLogger, tt.sig, &input, state)\n\t\t\tassert.Equal(t, tt.wantChanged, changed)\n\t\t\tassert.Equal(t, tt.wantCron, input.Spec.CronExpression)\n\t\t\tassert.Equal(t, tt.wantWF, input.Action.StartWorkflow.WorkflowType.Name)\n\t\t\tassert.Equal(t, tt.wantPol, input.Policies.OverlapPolicy)\n\t\t})\n\t}\n\n\tt.Run(\"spec change clears pending backfills\", func(t *testing.T) {\n\t\tinput := original\n\t\tstate := &SchedulerWorkflowState{\n\t\t\tPendingBackfills: []BackfillRequest{\n\t\t\t\t{StartTime: time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC), EndTime: time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC)},\n\t\t\t\t{StartTime: time.Date(2026, 1, 3, 0, 0, 0, 0, time.UTC), EndTime: time.Date(2026, 1, 4, 0, 0, 0, 0, time.UTC)},\n\t\t\t},\n\t\t}\n\t\tchanged := handleUpdate(testLogger, UpdateSignal{\n\t\t\tSpec: &types.ScheduleSpec{CronExpression: \"*/5 * * * *\"},\n\t\t}, &input, state)\n\t\tassert.True(t, changed)\n\t\tassert.Equal(t, \"*/5 * * * *\", input.Spec.CronExpression)\n\t\tassert.Empty(t, state.PendingBackfills)\n\t})\n\n\tt.Run(\"action-only update preserves pending backfills\", func(t *testing.T) {\n\t\tinput := original\n\t\tstate := &SchedulerWorkflowState{\n\t\t\tPendingBackfills: []BackfillRequest{\n\t\t\t\t{StartTime: time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC), EndTime: time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC)},\n\t\t\t},\n\t\t}\n\t\tchanged := handleUpdate(testLogger, UpdateSignal{\n\t\t\tAction: &types.ScheduleAction{StartWorkflow: &types.StartWorkflowAction{WorkflowType: &types.WorkflowType{Name: \"new-workflow\"}}},\n\t\t}, &input, state)\n\t\tassert.True(t, changed)\n\t\tassert.Len(t, state.PendingBackfills, 1)\n\t})\n\n\tt.Run(\"invalid cron does not clear pending backfills\", func(t *testing.T) {\n\t\tinput := original\n\t\tstate := &SchedulerWorkflowState{\n\t\t\tPendingBackfills: []BackfillRequest{\n\t\t\t\t{StartTime: time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC), EndTime: time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC)},\n\t\t\t},\n\t\t}\n\t\tchanged := handleUpdate(testLogger, UpdateSignal{\n\t\t\tSpec: &types.ScheduleSpec{CronExpression: \"not-a-cron\"},\n\t\t}, &input, state)\n\t\tassert.False(t, changed)\n\t\tassert.Len(t, state.PendingBackfills, 1)\n\t})\n}\n\nfunc TestHandleBackfill(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsig            BackfillSignal\n\t\tinitialPending int\n\t\twantQueued     bool\n\t\twantPendingLen int\n\t}{\n\t\t{\n\t\t\tname: \"valid backfill is queued\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime:     time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:       time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t\tOverlapPolicy: types.ScheduleOverlapPolicyConcurrent,\n\t\t\t\tBackfillID:    \"bf-1\",\n\t\t\t},\n\t\t\twantQueued:     true,\n\t\t\twantPendingLen: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"invalid range (end <= start) is rejected\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime: time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:   time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t\twantQueued:     false,\n\t\t\twantPendingLen: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"equal start and end is rejected\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime: time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:   time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t},\n\t\t\twantQueued:     false,\n\t\t\twantPendingLen: 0,\n\t\t},\n\t\t{\n\t\t\tname: \"multiple backfills accumulate\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime:  time.Date(2026, 1, 3, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:    time.Date(2026, 1, 4, 0, 0, 0, 0, time.UTC),\n\t\t\t\tBackfillID: \"bf-2\",\n\t\t\t},\n\t\t\tinitialPending: 1,\n\t\t\twantQueued:     true,\n\t\t\twantPendingLen: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"overlapping backfill is queued with warning\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime:  time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:    time.Date(2026, 1, 3, 0, 0, 0, 0, time.UTC),\n\t\t\t\tBackfillID: \"bf-overlap\",\n\t\t\t},\n\t\t\tinitialPending: 1,\n\t\t\twantQueued:     true,\n\t\t\twantPendingLen: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"backfill rejected when queue is full\",\n\t\t\tsig: BackfillSignal{\n\t\t\t\tStartTime:  time.Date(2026, 2, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:    time.Date(2026, 2, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t\tBackfillID: \"bf-over-cap\",\n\t\t\t},\n\t\t\tinitialPending: maxPendingBackfills,\n\t\t\twantQueued:     false,\n\t\t\twantPendingLen: maxPendingBackfills,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tstate := &SchedulerWorkflowState{}\n\t\t\tfor i := 0; i < tt.initialPending; i++ {\n\t\t\t\tstate.PendingBackfills = append(state.PendingBackfills, BackfillRequest{\n\t\t\t\t\tStartTime: time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC),\n\t\t\t\t\tEndTime:   time.Date(2026, 1, 2, 0, 0, 0, 0, time.UTC),\n\t\t\t\t})\n\t\t\t}\n\t\t\tgot := handleBackfill(testLogger, tt.sig, state)\n\t\t\tassert.Equal(t, tt.wantQueued, got)\n\t\t\tassert.Equal(t, tt.wantPendingLen, len(state.PendingBackfills))\n\t\t})\n\t}\n}\n\nfunc TestProcessBackfillsRespectsPause(t *testing.T) {\n\tsched := mustParseCron(t, \"0 * * * *\")\n\tinput := &SchedulerWorkflowInput{\n\t\tSpec: types.ScheduleSpec{CronExpression: \"0 * * * *\"},\n\t}\n\tstate := &SchedulerWorkflowState{\n\t\tPaused: true,\n\t\tPendingBackfills: []BackfillRequest{\n\t\t\t{\n\t\t\t\tStartTime:  time.Date(2026, 1, 1, 10, 0, 0, 0, time.UTC),\n\t\t\t\tEndTime:    time.Date(2026, 1, 1, 13, 0, 0, 0, time.UTC),\n\t\t\t\tBackfillID: \"bf-paused\",\n\t\t\t},\n\t\t},\n\t}\n\t// processBackfills should short-circuit without touching PendingBackfills\n\tmoreWork := processBackfills(nil, testLogger, sched, input, state)\n\tassert.False(t, moreWork, \"paused schedule should not process backfills\")\n\tassert.Len(t, state.PendingBackfills, 1, \"pending backfills should be preserved while paused\")\n}\n\nfunc TestBackfillFireComputation(t *testing.T) {\n\tsched := mustParseCron(t, \"0 * * * *\")\n\n\ttests := []struct {\n\t\tname      string\n\t\tstartTime time.Time\n\t\tendTime   time.Time\n\t\twantFires int\n\t}{\n\t\t{\n\t\t\tname:      \"3-hour window [10:00, 13:00] produces 4 fires (inclusive both ends)\",\n\t\t\tstartTime: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tendTime:   time.Date(2026, 1, 15, 13, 0, 0, 0, time.UTC),\n\t\t\twantFires: 4, // 10:00, 11:00, 12:00, 13:00\n\t\t},\n\t\t{\n\t\t\tname:      \"exact boundary [10:00, 11:00] includes both endpoints\",\n\t\t\tstartTime: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tendTime:   time.Date(2026, 1, 15, 11, 0, 0, 0, time.UTC),\n\t\t\twantFires: 2, // 10:00, 11:00\n\t\t},\n\t\t{\n\t\t\tname:      \"sub-hour window [10:00, 10:30] includes start fire only\",\n\t\t\tstartTime: time.Date(2026, 1, 15, 10, 0, 0, 0, time.UTC),\n\t\t\tendTime:   time.Date(2026, 1, 15, 10, 30, 0, 0, time.UTC),\n\t\t\twantFires: 1, // 10:00\n\t\t},\n\t\t{\n\t\t\tname:      \"24-hour window [00:00, 00:00+1d] produces 25 fires\",\n\t\t\tstartTime: time.Date(2026, 1, 15, 0, 0, 0, 0, time.UTC),\n\t\t\tendTime:   time.Date(2026, 1, 16, 0, 0, 0, 0, time.UTC),\n\t\t\twantFires: 25, // 00:00 through 00:00 next day, inclusive\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tfires := computeMissedFireTimes(sched, tt.startTime.Add(-time.Second), tt.endTime, types.ScheduleSpec{})\n\t\t\tassert.Equal(t, tt.wantFires, len(fires.times))\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "service/worker/service.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage worker\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"sync/atomic\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/resource\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/archiver\"\n\t\"github.com/uber/cadence/service/worker/asyncworkflow\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n\t\"github.com/uber/cadence/service/worker/diagnostics\"\n\t\"github.com/uber/cadence/service/worker/domaindeprecation\"\n\t\"github.com/uber/cadence/service/worker/esanalyzer\"\n\t\"github.com/uber/cadence/service/worker/failovermanager\"\n\t\"github.com/uber/cadence/service/worker/indexer\"\n\t\"github.com/uber/cadence/service/worker/parentclosepolicy\"\n\t\"github.com/uber/cadence/service/worker/replicator\"\n\t\"github.com/uber/cadence/service/worker/scanner\"\n\t\"github.com/uber/cadence/service/worker/scanner/executions\"\n\t\"github.com/uber/cadence/service/worker/scanner/shardscanner\"\n\t\"github.com/uber/cadence/service/worker/scanner/tasklist\"\n\t\"github.com/uber/cadence/service/worker/scanner/timers\"\n\t\"github.com/uber/cadence/service/worker/scheduler\"\n)\n\ntype (\n\t// Service represents the cadence-worker service. This service hosts all background processing needed for cadence cluster:\n\t// 1. Replicator: Handles applying replication tasks generated by remote clusters.\n\t// 2. Indexer: Handles uploading of visibility records to elastic search.\n\t// 3. Archiver: Handles archival of workflow histories.\n\tService struct {\n\t\tresource.Resource\n\n\t\tstatus int32\n\t\tstopC  chan struct{}\n\t\tparams *resource.Params\n\t\tconfig *Config\n\t}\n\n\t// Config contains all the service config for worker\n\tConfig struct {\n\t\tAdminOperationToken                 dynamicproperties.StringPropertyFn\n\t\tKafkaCfg                            config.KafkaConfig\n\t\tArchiverConfig                      *archiver.Config\n\t\tIndexerCfg                          *indexer.Config\n\t\tScannerCfg                          *scanner.Config\n\t\tBatcherCfg                          *batcher.Config\n\t\tESAnalyzerCfg                       *esanalyzer.Config\n\t\tfailoverManagerCfg                  *failovermanager.Config\n\t\tThrottledLogRPS                     dynamicproperties.IntPropertyFn\n\t\tPersistenceGlobalMaxQPS             dynamicproperties.IntPropertyFn\n\t\tPersistenceMaxQPS                   dynamicproperties.IntPropertyFn\n\t\tEnableBatcher                       dynamicproperties.BoolPropertyFn\n\t\tEnableScheduler                     dynamicproperties.BoolPropertyFn\n\t\tEnableParentClosePolicyWorker       dynamicproperties.BoolPropertyFn\n\t\tNumParentClosePolicySystemWorkflows dynamicproperties.IntPropertyFn\n\t\tEnableFailoverManager               dynamicproperties.BoolPropertyFn\n\t\tDomainReplicationMaxRetryDuration   dynamicproperties.DurationPropertyFn\n\t\tEnableESAnalyzer                    dynamicproperties.BoolPropertyFn\n\t\tEnableAsyncWorkflowConsumption      dynamicproperties.BoolPropertyFn\n\t\tEnableDomainAuditLogging            dynamicproperties.BoolPropertyFn\n\t\tHostName                            string\n\t}\n)\n\n// NewService builds a new cadence-worker service\nfunc NewService(params *resource.Params) (resource.Resource, error) {\n\tserviceConfig := NewConfig(params)\n\tserviceResource, err := resource.New(\n\t\tparams,\n\t\tservice.Worker,\n\t\t&service.Config{\n\t\t\tPersistenceMaxQPS:        serviceConfig.PersistenceMaxQPS,\n\t\t\tPersistenceGlobalMaxQPS:  serviceConfig.PersistenceGlobalMaxQPS,\n\t\t\tThrottledLoggerMaxRPS:    serviceConfig.ThrottledLogRPS,\n\t\t\tIsErrorRetryableFunction: common.IsServiceTransientError,\n\t\t\t// worker service doesn't need visibility config as it never call visibilityManager API\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Service{\n\t\tResource: serviceResource,\n\t\tstatus:   common.DaemonStatusInitialized,\n\t\tconfig:   serviceConfig,\n\t\tparams:   params,\n\t\tstopC:    make(chan struct{}),\n\t}, nil\n}\n\n// NewConfig builds the new Config for cadence-worker service\nfunc NewConfig(params *resource.Params) *Config {\n\tdc := dynamicconfig.NewCollection(\n\t\tparams.DynamicConfig,\n\t\tparams.Logger,\n\t\tdynamicproperties.ClusterNameFilter(params.ClusterMetadata.GetCurrentClusterName()),\n\t)\n\tconfig := &Config{\n\t\tAdminOperationToken: dc.GetStringProperty(dynamicproperties.AdminOperationToken),\n\t\tArchiverConfig: &archiver.Config{\n\t\t\tArchiverConcurrency:             dc.GetIntProperty(dynamicproperties.WorkerArchiverConcurrency),\n\t\t\tArchivalsPerIteration:           dc.GetIntProperty(dynamicproperties.WorkerArchivalsPerIteration),\n\t\t\tTimeLimitPerArchivalIteration:   dc.GetDurationProperty(dynamicproperties.WorkerTimeLimitPerArchivalIteration),\n\t\t\tAllowArchivingIncompleteHistory: dc.GetBoolProperty(dynamicproperties.AllowArchivingIncompleteHistory),\n\t\t},\n\t\tScannerCfg: &scanner.Config{\n\t\t\tScannerPersistenceMaxQPS: dc.GetIntProperty(dynamicproperties.ScannerPersistenceMaxQPS),\n\t\t\tTaskListScannerOptions: tasklist.Options{\n\t\t\t\tGetOrphanTasksPageSizeFn: dc.GetIntProperty(dynamicproperties.ScannerGetOrphanTasksPageSize),\n\t\t\t\tTaskBatchSizeFn:          dc.GetIntProperty(dynamicproperties.ScannerBatchSizeForTasklistHandler),\n\t\t\t\tEnableCleaning:           dc.GetBoolProperty(dynamicproperties.EnableCleaningOrphanTaskInTasklistScavenger),\n\t\t\t\tMaxTasksPerJobFn:         dc.GetIntProperty(dynamicproperties.ScannerMaxTasksProcessedPerTasklistJob),\n\t\t\t},\n\t\t\tPersistence:            &params.PersistenceConfig,\n\t\t\tClusterMetadata:        params.ClusterMetadata,\n\t\t\tTaskListScannerEnabled: dc.GetBoolProperty(dynamicproperties.TaskListScannerEnabled),\n\t\t\tHistoryScannerEnabled:  dc.GetBoolProperty(dynamicproperties.HistoryScannerEnabled),\n\t\t\tShardScanners: []*shardscanner.ScannerConfig{\n\t\t\t\texecutions.ConcreteExecutionConfig(dc),\n\t\t\t\texecutions.CurrentExecutionConfig(dc),\n\t\t\t\ttimers.ScannerConfig(dc),\n\t\t\t},\n\t\t\tMaxWorkflowRetentionInDays: dc.GetIntProperty(dynamicproperties.MaxRetentionDays),\n\t\t},\n\t\tKafkaCfg: params.KafkaConfig,\n\t\tBatcherCfg: &batcher.Config{\n\t\t\tAdminOperationToken: dc.GetStringProperty(dynamicproperties.AdminOperationToken),\n\t\t\tClusterMetadata:     params.ClusterMetadata,\n\t\t},\n\t\tfailoverManagerCfg: &failovermanager.Config{\n\t\t\tAdminOperationToken: dc.GetStringProperty(dynamicproperties.AdminOperationToken),\n\t\t\tClusterMetadata:     params.ClusterMetadata,\n\t\t},\n\t\tESAnalyzerCfg: &esanalyzer.Config{\n\t\t\tESAnalyzerPause:                          dc.GetBoolProperty(dynamicproperties.ESAnalyzerPause),\n\t\t\tESAnalyzerTimeWindow:                     dc.GetDurationProperty(dynamicproperties.ESAnalyzerTimeWindow),\n\t\t\tESAnalyzerMaxNumDomains:                  dc.GetIntProperty(dynamicproperties.ESAnalyzerMaxNumDomains),\n\t\t\tESAnalyzerMaxNumWorkflowTypes:            dc.GetIntProperty(dynamicproperties.ESAnalyzerMaxNumWorkflowTypes),\n\t\t\tESAnalyzerLimitToTypes:                   dc.GetStringProperty(dynamicproperties.ESAnalyzerLimitToTypes),\n\t\t\tESAnalyzerEnableAvgDurationBasedChecks:   dc.GetBoolProperty(dynamicproperties.ESAnalyzerEnableAvgDurationBasedChecks),\n\t\t\tESAnalyzerLimitToDomains:                 dc.GetStringProperty(dynamicproperties.ESAnalyzerLimitToDomains),\n\t\t\tESAnalyzerNumWorkflowsToRefresh:          dc.GetIntPropertyFilteredByWorkflowType(dynamicproperties.ESAnalyzerNumWorkflowsToRefresh),\n\t\t\tESAnalyzerBufferWaitTime:                 dc.GetDurationPropertyFilteredByWorkflowType(dynamicproperties.ESAnalyzerBufferWaitTime),\n\t\t\tESAnalyzerMinNumWorkflowsForAvg:          dc.GetIntPropertyFilteredByWorkflowType(dynamicproperties.ESAnalyzerMinNumWorkflowsForAvg),\n\t\t\tESAnalyzerWorkflowDurationWarnThresholds: dc.GetStringProperty(dynamicproperties.ESAnalyzerWorkflowDurationWarnThresholds),\n\t\t\tESAnalyzerWorkflowVersionDomains:         dc.GetStringProperty(dynamicproperties.ESAnalyzerWorkflowVersionMetricDomains),\n\t\t\tESAnalyzerWorkflowTypeDomains:            dc.GetStringProperty(dynamicproperties.ESAnalyzerWorkflowTypeMetricDomains),\n\t\t},\n\t\tEnableBatcher:                       dc.GetBoolProperty(dynamicproperties.EnableBatcher),\n\t\tEnableScheduler:                     dc.GetBoolProperty(dynamicproperties.EnableScheduler),\n\t\tEnableParentClosePolicyWorker:       dc.GetBoolProperty(dynamicproperties.EnableParentClosePolicyWorker),\n\t\tNumParentClosePolicySystemWorkflows: dc.GetIntProperty(dynamicproperties.NumParentClosePolicySystemWorkflows),\n\t\tEnableESAnalyzer:                    dc.GetBoolProperty(dynamicproperties.EnableESAnalyzer),\n\t\tEnableFailoverManager:               dc.GetBoolProperty(dynamicproperties.EnableFailoverManager),\n\t\tThrottledLogRPS:                     dc.GetIntProperty(dynamicproperties.WorkerThrottledLogRPS),\n\t\tPersistenceGlobalMaxQPS:             dc.GetIntProperty(dynamicproperties.WorkerPersistenceGlobalMaxQPS),\n\t\tPersistenceMaxQPS:                   dc.GetIntProperty(dynamicproperties.WorkerPersistenceMaxQPS),\n\t\tDomainReplicationMaxRetryDuration:   dc.GetDurationProperty(dynamicproperties.WorkerReplicationTaskMaxRetryDuration),\n\t\tEnableAsyncWorkflowConsumption:      dc.GetBoolProperty(dynamicproperties.EnableAsyncWorkflowConsumption),\n\t\tEnableDomainAuditLogging:            dc.GetBoolProperty(dynamicproperties.EnableDomainAuditLogging),\n\t\tHostName:                            params.HostName,\n\t}\n\tadvancedVisWritingMode := dc.GetStringProperty(\n\t\tdynamicproperties.WriteVisibilityStoreName,\n\t)\n\n\tif shouldStartIndexer(params, advancedVisWritingMode) {\n\t\tconfig.IndexerCfg = &indexer.Config{\n\t\t\tIndexerConcurrency:             dc.GetIntProperty(dynamicproperties.WorkerIndexerConcurrency),\n\t\t\tESProcessorNumOfWorkers:        dc.GetIntProperty(dynamicproperties.WorkerESProcessorNumOfWorkers),\n\t\t\tESProcessorBulkActions:         dc.GetIntProperty(dynamicproperties.WorkerESProcessorBulkActions),\n\t\t\tESProcessorBulkSize:            dc.GetIntProperty(dynamicproperties.WorkerESProcessorBulkSize),\n\t\t\tESProcessorFlushInterval:       dc.GetDurationProperty(dynamicproperties.WorkerESProcessorFlushInterval),\n\t\t\tValidSearchAttributes:          dc.GetMapProperty(dynamicproperties.ValidSearchAttributes),\n\t\t\tEnableQueryAttributeValidation: dc.GetBoolProperty(dynamicproperties.EnableQueryAttributeValidation),\n\t\t}\n\t}\n\n\treturn config\n}\n\n// Start is called to start the service\nfunc (s *Service) Start() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusInitialized, common.DaemonStatusStarted) {\n\t\treturn\n\t}\n\tlogger := s.GetLogger()\n\tlogger.Info(\"worker starting\", tag.ComponentWorker)\n\n\ts.Resource.Start()\n\ts.Resource.GetDomainReplicationQueue().Start()\n\n\ts.ensureDomainExists(constants.SystemLocalDomainName)\n\ts.startScanner()\n\ts.startFixerWorkflowWorker()\n\tif s.config.IndexerCfg != nil {\n\t\tif shouldStartMigrationIndexer(s.params) {\n\t\t\ts.startMigrationDualIndexer()\n\t\t} else {\n\t\t\ts.startIndexer()\n\t\t}\n\t}\n\n\ts.startReplicator()\n\ts.startDiagnostics()\n\ts.startDomainDeprecation()\n\n\tif s.GetArchivalMetadata().GetHistoryConfig().ClusterConfiguredForArchival() {\n\t\ts.startArchiver()\n\t}\n\tif s.config.EnableBatcher() {\n\t\ts.ensureDomainExists(constants.BatcherLocalDomainName)\n\t\ts.startBatcher()\n\t}\n\tif s.config.EnableScheduler() {\n\t\tsm := s.startSchedulerWorkerManager()\n\t\tdefer sm.Stop()\n\t}\n\tif s.config.EnableParentClosePolicyWorker() {\n\t\ts.startParentClosePolicyProcessor()\n\t}\n\tif s.config.EnableESAnalyzer() {\n\t\ts.startESAnalyzer()\n\t}\n\tif s.config.EnableFailoverManager() {\n\t\ts.startFailoverManager()\n\t}\n\n\tcm := s.startAsyncWorkflowConsumerManager()\n\tdefer cm.Stop()\n\n\tlogger.Info(\"worker started\", tag.ComponentWorker)\n\t<-s.stopC\n}\n\n// Stop is called to stop the service\nfunc (s *Service) Stop() {\n\tif !atomic.CompareAndSwapInt32(&s.status, common.DaemonStatusStarted, common.DaemonStatusStopped) {\n\t\treturn\n\t}\n\n\ts.GetLogger().Info(\"worker stopping\", tag.ComponentWorker)\n\n\tclose(s.stopC)\n\n\ts.Resource.Stop()\n\ts.Resource.GetDomainReplicationQueue().Stop()\n\n\ts.GetLogger().Info(\"worker stopped\", tag.ComponentWorker)\n}\n\nfunc (s *Service) startParentClosePolicyProcessor() {\n\tparams := &parentclosepolicy.BootstrapParams{\n\t\tServiceClient: s.params.PublicClient,\n\t\tMetricsClient: s.GetMetricsClient(),\n\t\tLogger:        s.GetLogger(),\n\t\tTallyScope:    s.params.MetricScope,\n\t\tClientBean:    s.GetClientBean(),\n\t\tDomainCache:   s.GetDomainCache(),\n\t\tNumWorkflows:  s.config.NumParentClosePolicySystemWorkflows(),\n\t}\n\tprocessor := parentclosepolicy.New(params)\n\tif err := processor.Start(); err != nil {\n\t\ts.GetLogger().Fatal(\"error starting parentclosepolicy processor\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startESAnalyzer() {\n\tesClient := s.params.ESClient\n\tesConfig := s.params.ESConfig\n\t// when es client is not configured, use os client, this can happen during os migration, we will use os client to read from os\n\t// there is another case is pinot migration and migration mode is off, in this case nil es config/client is expected\n\tif esClient == nil && s.params.OSClient != nil {\n\t\tesClient = s.params.OSClient\n\t\tesConfig = s.params.OSConfig\n\t}\n\n\tanalyzer := esanalyzer.New(\n\t\ts.params.PublicClient,\n\t\ts.GetFrontendClient(),\n\t\ts.GetClientBean(),\n\t\tesClient,\n\t\ts.params.PinotClient,\n\t\tesConfig,\n\t\ts.params.PinotConfig,\n\t\ts.GetLogger(),\n\t\ts.params.MetricScope,\n\t\ts.Resource,\n\t\ts.GetDomainCache(),\n\t\ts.config.ESAnalyzerCfg,\n\t)\n\n\tif err := analyzer.Start(); err != nil {\n\t\ts.GetLogger().Fatal(\"error starting esanalyzer\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startBatcher() {\n\tparams := &batcher.BootstrapParams{\n\t\tConfig:        *s.config.BatcherCfg,\n\t\tServiceClient: s.params.PublicClient,\n\t\tMetricsClient: s.GetMetricsClient(),\n\t\tLogger:        s.GetLogger(),\n\t\tTallyScope:    s.params.MetricScope,\n\t\tClientBean:    s.GetClientBean(),\n\t}\n\tif err := batcher.New(params).Start(); err != nil {\n\t\ts.GetLogger().Fatal(\"error starting batcher\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startSchedulerWorkerManager() *scheduler.WorkerManager {\n\tparams := &scheduler.BootstrapParams{\n\t\tServiceClient:      s.params.PublicClient,\n\t\tFrontendClient:     s.GetClientBean().GetFrontendClient(),\n\t\tLogger:             s.GetLogger(),\n\t\tDomainCache:        s.GetDomainCache(),\n\t\tMembershipResolver: s.GetMembershipResolver(),\n\t\tHostInfo:           s.GetHostInfo(),\n\t}\n\twm := scheduler.NewWorkerManager(params, s.config.EnableScheduler)\n\twm.Start()\n\treturn wm\n}\n\nfunc (s *Service) startScanner() {\n\tparams := &scanner.BootstrapParams{\n\t\tConfig:     *s.config.ScannerCfg,\n\t\tTallyScope: s.params.MetricScope,\n\t}\n\tif err := scanner.New(s.Resource, params).Start(); err != nil {\n\t\ts.GetLogger().Fatal(\"error starting scanner\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startFixerWorkflowWorker() {\n\tparams := &scanner.BootstrapParams{\n\t\tConfig:     *s.config.ScannerCfg,\n\t\tTallyScope: s.params.MetricScope,\n\t}\n\tif err := scanner.NewDataCorruptionWorkflowWorker(s.Resource, params).StartDataCorruptionWorkflowWorker(); err != nil {\n\t\ts.GetLogger().Fatal(\"error starting fixer workflow worker\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startDiagnostics() {\n\tparams := diagnostics.Params{\n\t\tServiceClient:   s.params.PublicClient,\n\t\tMetricsClient:   s.GetMetricsClient(),\n\t\tMessagingClient: s.GetMessagingClient(),\n\t\tTallyScope:      s.params.MetricScope,\n\t\tClientBean:      s.GetClientBean(),\n\t\tLogger:          s.GetLogger(),\n\t\tInvariants:      s.params.DiagnosticsInvariants,\n\t\tClusterMetadata: s.GetClusterMetadata(),\n\t}\n\tif err := diagnostics.New(params).Start(); err != nil {\n\t\ts.Stop()\n\t\ts.GetLogger().Fatal(\"error starting diagnostics\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startReplicator() {\n\tdomainReplicationTaskExecutor := domain.NewReplicationTaskExecutor(\n\t\ts.Resource.GetDomainManager(),\n\t\ts.Resource.GetDomainAuditManager(),\n\t\ts.Resource.GetTimeSource(),\n\t\ts.Resource.GetLogger(),\n\t\ts.config.EnableDomainAuditLogging,\n\t)\n\tmsgReplicator := replicator.NewReplicator(\n\t\ts.GetClusterMetadata(),\n\t\ts.GetClientBean(),\n\t\ts.GetLogger(),\n\t\ts.GetMetricsClient(),\n\t\ts.GetHostInfo(),\n\t\ts.GetMembershipResolver(),\n\t\ts.GetDomainReplicationQueue(),\n\t\tdomainReplicationTaskExecutor,\n\t\ts.config.DomainReplicationMaxRetryDuration(),\n\t)\n\tif err := msgReplicator.Start(); err != nil {\n\t\tmsgReplicator.Stop()\n\t\ts.GetLogger().Fatal(\"fail to start replicator\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startIndexer() {\n\tvisibilityIndexer := indexer.NewIndexer(\n\t\ts.config.IndexerCfg,\n\t\ts.GetMessagingClient(),\n\t\ts.params.ESClient,\n\t\ts.params.ESConfig.Indices[constants.VisibilityAppName],\n\t\ts.params.ESConfig.ConsumerName,\n\t\ts.GetLogger(),\n\t\ts.GetMetricsClient(),\n\t)\n\tif err := visibilityIndexer.Start(); err != nil {\n\t\tvisibilityIndexer.Stop()\n\t\ts.GetLogger().Fatal(\"fail to start indexer\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startMigrationDualIndexer() {\n\tvisibilityDualIndexer := indexer.NewMigrationDualIndexer(\n\t\ts.config.IndexerCfg,\n\t\ts.GetMessagingClient(),\n\t\ts.params.ESClient,\n\t\ts.params.OSClient,\n\t\ts.params.ESConfig.Indices[constants.VisibilityAppName],\n\t\ts.params.OSConfig.Indices[constants.VisibilityAppName],\n\t\ts.params.ESConfig.ConsumerName,\n\t\ts.params.OSConfig.ConsumerName,\n\t\ts.GetLogger(),\n\t\ts.GetMetricsClient(),\n\t)\n\tif err := visibilityDualIndexer.Start(); err != nil {\n\t\t// not need to call visibilityDualIndexer.Stop() since it has been called inside Start()\n\t\ts.GetLogger().Fatal(\"fail to start indexer\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startArchiver() {\n\tbc := &archiver.BootstrapContainer{\n\t\tPublicClient:     s.GetSDKClient(),\n\t\tMetricsClient:    s.GetMetricsClient(),\n\t\tLogger:           s.GetLogger(),\n\t\tHistoryV2Manager: s.GetHistoryManager(),\n\t\tDomainCache:      s.GetDomainCache(),\n\t\tConfig:           s.config.ArchiverConfig,\n\t\tArchiverProvider: s.GetArchiverProvider(),\n\t}\n\tclientWorker := archiver.NewClientWorker(bc)\n\tif err := clientWorker.Start(); err != nil {\n\t\tclientWorker.Stop()\n\t\ts.GetLogger().Fatal(\"failed to start archiver\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startFailoverManager() {\n\tparams := &failovermanager.BootstrapParams{\n\t\tConfig:        *s.config.failoverManagerCfg,\n\t\tServiceClient: s.params.PublicClient,\n\t\tMetricsClient: s.GetMetricsClient(),\n\t\tLogger:        s.GetLogger(),\n\t\tTallyScope:    s.params.MetricScope,\n\t\tClientBean:    s.GetClientBean(),\n\t}\n\tif err := failovermanager.New(params).Start(); err != nil {\n\t\ts.Stop()\n\t\ts.GetLogger().Fatal(\"error starting failoverManager\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) startAsyncWorkflowConsumerManager() common.Daemon {\n\tcm := asyncworkflow.NewConsumerManager(\n\t\ts.GetLogger(),\n\t\ts.GetMetricsClient(),\n\t\ts.GetDomainCache(),\n\t\ts.Resource.GetAsyncWorkflowQueueProvider(),\n\t\ts.GetFrontendClient(),\n\t\tasyncworkflow.WithEnabledPropertyFn(s.config.EnableAsyncWorkflowConsumption),\n\t)\n\tcm.Start()\n\treturn cm\n}\n\nfunc (s *Service) startDomainDeprecation() {\n\tparams := domaindeprecation.Params{\n\t\tConfig: domaindeprecation.Config{\n\t\t\tAdminOperationToken: s.config.AdminOperationToken,\n\t\t},\n\t\tServiceClient: s.params.PublicClient,\n\t\tClientBean:    s.GetClientBean(),\n\t\tMetricsClient: s.GetMetricsClient(),\n\t\tTally:         s.params.MetricScope,\n\t\tLogger:        s.GetLogger(),\n\t}\n\n\tif err := domaindeprecation.New(params).Start(); err != nil {\n\t\ts.Stop()\n\t\ts.GetLogger().Fatal(\"error starting domain deprecator\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) ensureDomainExists(domain string) {\n\t_, err := s.GetDomainManager().GetDomain(context.Background(), &persistence.GetDomainRequest{Name: domain})\n\tswitch err.(type) {\n\tcase nil:\n\t\t// noop\n\tcase *types.EntityNotExistsError:\n\t\ts.GetLogger().Info(fmt.Sprintf(\"domain %s does not exist, attempting to register domain\", domain))\n\t\ts.registerSystemDomain(domain)\n\tdefault:\n\t\ts.GetLogger().Fatal(\"failed to verify if system domain exists\", tag.Error(err))\n\t}\n}\n\nfunc (s *Service) registerSystemDomain(domain string) {\n\n\tcurrentClusterName := s.GetClusterMetadata().GetCurrentClusterName()\n\t_, err := s.GetDomainManager().CreateDomain(context.Background(), &persistence.CreateDomainRequest{\n\t\tInfo: &persistence.DomainInfo{\n\t\t\tID:          getDomainID(domain),\n\t\t\tName:        domain,\n\t\t\tDescription: \"Cadence internal system domain\",\n\t\t},\n\t\tConfig: &persistence.DomainConfig{\n\t\t\tRetention:  constants.SystemDomainRetentionDays,\n\t\t\tEmitMetric: true,\n\t\t},\n\t\tReplicationConfig: &persistence.DomainReplicationConfig{\n\t\t\tActiveClusterName: currentClusterName,\n\t\t\tClusters:          cluster.GetOrUseDefaultClusters(currentClusterName, nil),\n\t\t},\n\t\tIsGlobalDomain:  false,\n\t\tFailoverVersion: constants.EmptyVersion,\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*types.DomainAlreadyExistsError); ok {\n\t\t\treturn\n\t\t}\n\t\ts.GetLogger().Fatal(\"failed to register system domain\", tag.Error(err))\n\t}\n}\n\nfunc getDomainID(domain string) string {\n\tvar domainID string\n\tswitch domain {\n\tcase constants.SystemLocalDomainName:\n\t\tdomainID = constants.SystemDomainID\n\tcase constants.BatcherLocalDomainName:\n\t\tdomainID = constants.BatcherDomainID\n\tcase constants.ShadowerLocalDomainName:\n\t\tdomainID = constants.ShadowerDomainID\n\t}\n\treturn domainID\n}\n\nfunc shouldStartIndexer(params *resource.Params, advancedWritingMode dynamicproperties.StringPropertyFn) bool {\n\t// only start indexer when advanced visibility writing mode is set to on and advanced visibility store is configured\n\tif !common.IsAdvancedVisibilityWritingEnabled(advancedWritingMode(), params.PersistenceConfig.IsAdvancedVisibilityConfigExist()) {\n\t\treturn false\n\t}\n\n\t// when it is using pinot and not in migration mode, indexer should not be started since Pinot will direclty ingest from kafka\n\tif params.PersistenceConfig.AdvancedVisibilityStore == constants.PinotVisibilityStoreName &&\n\t\tparams.PinotConfig != nil &&\n\t\t!params.PinotConfig.Migration.Enabled {\n\t\treturn false\n\t}\n\n\treturn true\n}\n\nfunc shouldStartMigrationIndexer(params *resource.Params) bool {\n\t// not need to check IsAdvancedVisibilityWritingEnabled here since it was already checked or the s.config.IndexerCfg will be nil\n\t// when it is using OS and in migration mode, we should start dual indexer to write to both ES and OS\n\tif params.PersistenceConfig.AdvancedVisibilityStore == constants.OSVisibilityStoreName &&\n\t\tparams.OSConfig != nil &&\n\t\tparams.OSConfig.Migration.Enabled {\n\t\treturn true\n\t}\n\n\treturn false\n}\n"
  },
  {
    "path": "service/worker/worker/worker_interface.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2024 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage worker\n\nimport (\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\n//go:generate mockgen -source $GOFILE -destination worker_mock.go -package worker github.com/uber/cadence/service/worker/worker Worker\n\ntype Worker interface {\n\tRegisterActivity(activity interface{})\n\tRegisterActivityWithOptions(activity interface{}, options activity.RegisterOptions)\n\tRegisterWorkflow(workflow interface{})\n\tRegisterWorkflowWithOptions(workflow interface{}, options workflow.RegisterOptions)\n\tStart() error\n\tStop()\n\tRun() error\n}\n"
  },
  {
    "path": "service/worker/worker/worker_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: worker_interface.go\n//\n// Generated by this command:\n//\n//\tmockgen -source worker_interface.go -destination worker_mock.go -package worker github.com/uber/cadence/service/worker/worker Worker\n//\n\n// Package worker is a generated GoMock package.\npackage worker\n\nimport (\n\treflect \"reflect\"\n\n\tactivity \"go.uber.org/cadence/activity\"\n\tworkflow \"go.uber.org/cadence/workflow\"\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockWorker is a mock of Worker interface.\ntype MockWorker struct {\n\tctrl     *gomock.Controller\n\trecorder *MockWorkerMockRecorder\n\tisgomock struct{}\n}\n\n// MockWorkerMockRecorder is the mock recorder for MockWorker.\ntype MockWorkerMockRecorder struct {\n\tmock *MockWorker\n}\n\n// NewMockWorker creates a new mock instance.\nfunc NewMockWorker(ctrl *gomock.Controller) *MockWorker {\n\tmock := &MockWorker{ctrl: ctrl}\n\tmock.recorder = &MockWorkerMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockWorker) EXPECT() *MockWorkerMockRecorder {\n\treturn m.recorder\n}\n\n// RegisterActivity mocks base method.\nfunc (m *MockWorker) RegisterActivity(activity any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RegisterActivity\", activity)\n}\n\n// RegisterActivity indicates an expected call of RegisterActivity.\nfunc (mr *MockWorkerMockRecorder) RegisterActivity(activity any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterActivity\", reflect.TypeOf((*MockWorker)(nil).RegisterActivity), activity)\n}\n\n// RegisterActivityWithOptions mocks base method.\nfunc (m *MockWorker) RegisterActivityWithOptions(activity any, options activity.RegisterOptions) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RegisterActivityWithOptions\", activity, options)\n}\n\n// RegisterActivityWithOptions indicates an expected call of RegisterActivityWithOptions.\nfunc (mr *MockWorkerMockRecorder) RegisterActivityWithOptions(activity, options any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterActivityWithOptions\", reflect.TypeOf((*MockWorker)(nil).RegisterActivityWithOptions), activity, options)\n}\n\n// RegisterWorkflow mocks base method.\nfunc (m *MockWorker) RegisterWorkflow(workflow any) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RegisterWorkflow\", workflow)\n}\n\n// RegisterWorkflow indicates an expected call of RegisterWorkflow.\nfunc (mr *MockWorkerMockRecorder) RegisterWorkflow(workflow any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterWorkflow\", reflect.TypeOf((*MockWorker)(nil).RegisterWorkflow), workflow)\n}\n\n// RegisterWorkflowWithOptions mocks base method.\nfunc (m *MockWorker) RegisterWorkflowWithOptions(workflow any, options workflow.RegisterOptions) {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"RegisterWorkflowWithOptions\", workflow, options)\n}\n\n// RegisterWorkflowWithOptions indicates an expected call of RegisterWorkflowWithOptions.\nfunc (mr *MockWorkerMockRecorder) RegisterWorkflowWithOptions(workflow, options any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"RegisterWorkflowWithOptions\", reflect.TypeOf((*MockWorker)(nil).RegisterWorkflowWithOptions), workflow, options)\n}\n\n// Run mocks base method.\nfunc (m *MockWorker) Run() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Run\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Run indicates an expected call of Run.\nfunc (mr *MockWorkerMockRecorder) Run() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Run\", reflect.TypeOf((*MockWorker)(nil).Run))\n}\n\n// Start mocks base method.\nfunc (m *MockWorker) Start() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Start\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Start indicates an expected call of Start.\nfunc (mr *MockWorkerMockRecorder) Start() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Start\", reflect.TypeOf((*MockWorker)(nil).Start))\n}\n\n// Stop mocks base method.\nfunc (m *MockWorker) Stop() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Stop\")\n}\n\n// Stop indicates an expected call of Stop.\nfunc (mr *MockWorkerMockRecorder) Stop() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Stop\", reflect.TypeOf((*MockWorker)(nil).Stop))\n}\n"
  },
  {
    "path": "service/worker/workercommon/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage workercommon\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/client\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/resource\"\n)\n\nfunc StartWorkflowWithRetry(\n\tworkflowType string,\n\tstartUpDelay time.Duration,\n\tresource resource.Resource,\n\tstartWorkflow func(client client.Client) error,\n) error {\n\t// let history / matching service warm up\n\ttime.Sleep(startUpDelay)\n\tsdkClient := client.NewClient(\n\t\tresource.GetSDKClient(),\n\t\tconstants.SystemLocalDomainName,\n\t\tnil, /* &client.Options{} */\n\t)\n\tpolicy := backoff.NewExponentialRetryPolicy(time.Second)\n\tpolicy.SetMaximumInterval(time.Minute)\n\tpolicy.SetExpirationInterval(backoff.NoInterval)\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(policy),\n\t\tbackoff.WithRetryableError(func(_ error) bool { return true }),\n\t)\n\terr := throttleRetry.Do(context.Background(), func(ctx context.Context) error {\n\t\treturn startWorkflow(sdkClient)\n\t})\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"unreachable: %#v\", err))\n\t} else {\n\t\tresource.GetLogger().Info(\"starting workflow\", tag.WorkflowType(workflowType))\n\t}\n\treturn err\n}\n"
  },
  {
    "path": "simulation/README.md",
    "content": "# Simulation Tests\n\nSimulation tests are black box tests that validate complex multi-component scenarios against a Cadence cluster. They enable systematic testing of workflows that would otherwise require manual sanity checks.\n\n## Overview\n\nEach suite of simulation tests:\n- instantiates a local docker cluster\n- makes requests to the local cluster, based on the test configuration\n- analyses the results\n\n### Test Types\n\n- **replication** - Tests the edge-case behaviour of workflows replicated to multiple clusters, e.g during failover, replication lag, or in active-active domains. \n- **matching** - Tests the performance of the cadence-matching service.\n- **history** - Tests the behaviour of the history service.\n\nSee individual subdirectory READMEs for specific test details.\n\n## Running Simulations\n\nRunning the tests requires that the developer has \n- [setup their development environment](../CONTRIBUTING.md#development-environment)\n- has docker running\n\n### Quick Start\n\n```bash\n# Run a basic replication test\n./simulation/replication/run.sh --scenario default\n\n# Run a matching performance test\n./simulation/matching/run.sh --scenario throughput\n\n# Run a history analysis test\n./simulation/history/run.sh --scenario default\n```\n\n### Local Development\n\nSome contributors may require a custom dockerfile to be passed to allow the simulation tests to build locally.\nAdd your dockerfile to `docker/github_actions/` with a custom suffix, and then pass the `--dockerfile-suffix` paramater to use it:\n\n```bash\n# For a file named Dockerfile.local in docker/github_actions:\n./simulation/replication/run.sh --scenario default --dockerfile-suffix .local\n```\n\n## Adding New Simulations\n\nAny time you are writing code that is complex, touches multiple components, or requires manual testing to sanity check its behaviour it is a candidate for a simulation test. To add a new simulation:\n\n1. Create scenario file: `{type}/testdata/{type}_simulation_{name}.yaml`\n2. Add config overrides: `config/dynamicconfig/{type}_simulation_{name}.yml` (if needed)\n3. Test locally: `./simulation/{type}/run.sh --scenario {name}`\n4. Add to CI matrix in `.github/workflows/{type}-simulation.yml`\n\n### CI/CD\n\nNote: only replication simulations run in GitHub Actions currently. To add a new replication scenario to CI, update the matrix in `.github/workflows/replication-simulation.yml`.\n"
  },
  {
    "path": "simulation/history/dynamicconfig/default.yaml",
    "content": "system.workflowDeletionJitterRange:\n- value: 0\n  constraints: {}\n"
  },
  {
    "path": "simulation/history/dynamicconfig/queuev2.yaml",
    "content": "system.workflowDeletionJitterRange:\n- value: 0\n  constraints: {}\nhistory.enableTimerQueueV2:\n- value: true\n  constraints: {}\nhistory.enableTransferQueueV2:\n- value: true\n  constraints: {}\nhistory.queueMaxPendingTaskCount:\n- value: 10000\n  constraints: {}\nhistory.timerTaskBatchSize:\n- value: 100\n  constraints: {}\nhistory.timerProcessorUpdateAckInterval:\n- value: 30s\n  constraints: {}\nhistory.timerProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.timerProcessorMaxPollRPS:\n- value: 20\n  constraints: {}\nhistory.transferTaskBatchSize:\n- value: 100\n  constraints: {}\nhistory.transferProcessorUpdateAckInterval:\n- value: 30s\n  constraints: {}\nhistory.transferProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.transferProcessorMaxPollRPS:\n- value: 20\n  constraints: {}\n"
  },
  {
    "path": "simulation/history/dynamicconfig/queuev2_split.yaml",
    "content": "system.workflowDeletionJitterRange:\n- value: 0\n  constraints: {}\nhistory.enableTimerQueueV2:\n- value: true\n  constraints: {}\nhistory.enableTransferQueueV2:\n- value: true\n  constraints: {}\nhistory.timerTaskBatchSize:\n- value: 100\n  constraints: {}\nhistory.timerProcessorUpdateAckInterval:\n- value: 5s\n  constraints: {}\nhistory.timerProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.timerProcessorMaxPollRPS:\n- value: 20\n  constraints: {}\nhistory.transferTaskBatchSize:\n- value: 100\n  constraints: {}\nhistory.transferProcessorUpdateAckInterval:\n- value: 5s\n  constraints: {}\nhistory.transferProcessorUpdateAckIntervalJitterCoefficient:\n- value: 0\n  constraints: {}\nhistory.transferProcessorMaxPollRPS:\n- value: 20\n  constraints: {}\nhistory.shardUpdateMinInterval:\n- value: 3s\n  constraints: {}\nhistory.queueProcessorPollBackoffInterval:\n- value: 5s\n  constraints: {}\nhistory.virtualSliceForceAppendInterval:\n- value: 5ms\n  constraints: {}\n# Only Enable 1 level of split, which has been tested manually\nhistory.queueMaxVirtualQueueCount:\n- value: 2\n  constraints: {}\n# Enable task rate limiter so that the number of pending tasks increases\nhistory.taskSchedulerEnableRateLimiter:\n- value: true\n  constraints: {}\nhistory.taskSchedulerEnableRateLimiterShadowMode:\n- value: false\n  constraints: {}\nhistory.taskSchedulerGlobalDomainRPS:\n- value: 5\n  constraints: {}\nhistory.enableTransferQueueV2PendingTaskCountAlert:\n- value: false\n  constraints: {}\n- value: true\n  constraints:\n    shardID: 0\nhistory.enableTimerQueueV2PendingTaskCountAlert:\n- value: false\n  constraints: {}\n- value: true\n  constraints:\n    shardID: 1\n# Set a low number to trigger queue pause with load from tests\nhistory.queueMaxPendingTaskCount:\n- value: 20\n  constraints: {}\n# Set a low number to trigger queue split with load from tests\nhistory.queueCriticalPendingTaskCount:\n- value: 18\n  constraints: {}\n"
  },
  {
    "path": "simulation/history/history_simulation_test.go",
    "content": "package history\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/client\"\n\t\"go.uber.org/cadence/compatibility\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log/tag\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/host\"\n\t\"github.com/uber/cadence/simulation/history/workflow\"\n)\n\nconst (\n\tdefaultTestCase = \"testdata/history_simulation_default.yaml\"\n)\n\ntype HistorySimulationSuite struct {\n\t*require.Assertions\n\t*host.IntegrationBase\n\twfService workflowserviceclient.Interface\n\twfClient  client.Client\n\tworker    worker.Worker\n\ttaskList  string\n}\n\nfunc TestHistorySimulation(t *testing.T) {\n\tflag.Parse()\n\n\tconfPath := os.Getenv(\"HISTORY_SIMULATION_CONFIG\")\n\tif confPath == \"\" {\n\t\tconfPath = defaultTestCase\n\t}\n\tclusterConfig, err := host.GetTestClusterConfig(confPath)\n\tif err != nil {\n\t\tt.Fatalf(\"failed creating cluster config from %s, err: %v\", confPath, err)\n\t}\n\ttestCluster := host.NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(HistorySimulationSuite)\n\tparams := host.IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = host.NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *HistorySimulationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := host.NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tReadNoSQLHistoryTaskFromDataBlob:         dynamicproperties.GetBoolPropertyFn(false),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t\tReadNoSQLShardFromDataBlob:               dynamicproperties.GetBoolPropertyFn(true),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := host.NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ttime.Sleep(2 * time.Second)\n\n\ts.wfService = s.buildServiceClient()\n\ts.wfClient = client.NewClient(s.wfService, s.DomainName, nil)\n\n\ts.taskList = \"history-simulation-tasklist\"\n\ts.worker = worker.New(s.wfService, s.DomainName, s.taskList, worker.Options{})\n\tworkflow.RegisterWorker(s.worker)\n\tif err := s.worker.Start(); err != nil {\n\t\ts.Logger.Fatal(\"Error when start worker\", tag.Error(err))\n\t} else {\n\t\ts.Logger.Info(\"Worker started\")\n\t}\n}\n\nfunc (s *HistorySimulationSuite) buildServiceClient() workflowserviceclient.Interface {\n\tcadenceClientName := \"cadence-client\"\n\thostPort := \"127.0.0.1:7114\" // use grpc port\n\tif host.TestFlags.FrontendAddr != \"\" {\n\t\thostPort = host.TestFlags.FrontendAddr\n\t}\n\tch := grpc.NewTransport(\n\t\tgrpc.ServerMaxRecvMsgSize(32*1024*1024),\n\t\tgrpc.ClientMaxRecvMsgSize(32*1024*1024),\n\t)\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: cadenceClientName,\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\tservice.Frontend: {Unary: ch.NewSingleOutbound(hostPort)},\n\t\t},\n\t})\n\tif err := dispatcher.Start(); err != nil {\n\t\ts.Logger.Fatal(\"Failed to create outbound transport channel\", tag.Error(err))\n\t}\n\tcc := dispatcher.ClientConfig(service.Frontend)\n\treturn compatibility.NewThrift2ProtoAdapter(\n\t\tapiv1.NewDomainAPIYARPCClient(cc),\n\t\tapiv1.NewWorkflowAPIYARPCClient(cc),\n\t\tapiv1.NewWorkerAPIYARPCClient(cc),\n\t\tapiv1.NewVisibilityAPIYARPCClient(cc),\n\t)\n}\n\nfunc (s *HistorySimulationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *HistorySimulationSuite) TearDownSuite() {\n\ts.worker.Stop()\n\t// Sleep for a while to ensure all metrics are emitted/scraped by prometheus\n\ttime.Sleep(5 * time.Second)\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *HistorySimulationSuite) TestHistorySimulation() {\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)\n\tdefer cancel()\n\tvar runs []client.WorkflowRun\n\tfor i := 0; i < 100; i++ {\n\t\t// set a short timeout so that timer tasks can be executed before complete\n\t\tworkflowOptions := client.StartWorkflowOptions{\n\t\t\tTaskList:                        s.taskList,\n\t\t\tExecutionStartToCloseTimeout:    120 * time.Second,\n\t\t\tDecisionTaskStartToCloseTimeout: 5 * time.Second,\n\t\t}\n\t\twe, err := s.wfClient.ExecuteWorkflow(ctx, workflowOptions, workflow.NoopWorkflow)\n\t\tif err != nil {\n\t\t\ts.Logger.Fatal(\"Start workflow with err\", tag.Error(err))\n\t\t}\n\t\ts.NotNil(we)\n\t\ts.True(we.GetRunID() != \"\")\n\t\ts.Logger.Info(\"successfully start a workflow\", tag.WorkflowID(we.GetID()), tag.WorkflowRunID(we.GetRunID()))\n\t\truns = append(runs, we)\n\t}\n\tfor _, we := range runs {\n\t\ts.NoError(we.Get(ctx, nil))\n\t}\n\ttime.Sleep(120 * time.Second)\n}\n"
  },
  {
    "path": "simulation/history/run.sh",
    "content": "#!/bin/bash\n\n# Cadence History Simulation Test Script\n#\n# Usage:\n#   ./simulation/history/run.sh [OPTIONS]\n#\n# Examples:\n#   # Run default scenario\n#   ./simulation/history/run.sh --scenario default\n#\n#   # Run specific scenario\n#   ./simulation/history/run.sh --scenario queuev2\n#\n#   # Run with custom timestamp\n#   ./simulation/history/run.sh --scenario default --timestamp 2024-01-15-10-30-00\n#\n#   # Run with custom dockerfile\n#   ./simulation/history/run.sh --scenario queuev2 --dockerfile-suffix .local\n#\n# TODO: Add simulation/history/README.md.  \n\nset -eo pipefail\n\nshow_help() {\n    cat << EOF\nCadence History Simulation Test Script\n\nUSAGE:\n    $0 [OPTIONS]\n\nOPTIONS:\n    -s, --scenario SCENARIO      Test scenario to run (required)\n                                Corresponds to testdata/history_simulation_SCENARIO.yaml\n\n    -t, --timestamp TIMESTAMP   Custom timestamp for test naming (default: current time)\n                                Format: YYYY-MM-DD-HH-MM-SS\n\n    -d, --dockerfile-suffix SUFFIX  Dockerfile suffix for custom builds (default: empty)\n                                   Example: .local for Dockerfile.local\n\n    -h, --help                  Show this help message\n\nEXAMPLES:\n    # Run default scenario\n    $0 --scenario default\n\n    # Run specific scenario\n    $0 --scenario queuev2\n\n    # Run with custom timestamp\n    $0 --scenario default --timestamp 2024-01-15-10-30-00\n\n    # Run with custom dockerfile\n    $0 --scenario queuev2 --dockerfile-suffix .local\n\nEOF\n}\n\n# Default values\ntestCase=\"\"\ntimestamp=\"\"\ndockerFileSuffix=\"\"\n\n# Parse command line arguments\nwhile [[ $# -gt 0 ]]; do\n    case $1 in\n        -s|--scenario)\n            testCase=\"$2\"\n            shift 2\n            ;;\n        -t|--timestamp)\n            timestamp=\"$2\"\n            shift 2\n            ;;\n        -d|--dockerfile-suffix)\n            dockerFileSuffix=\"$2\"\n            shift 2\n            ;;\n        -h|--help)\n            show_help\n            exit 0\n            ;;\n        --)\n            shift\n            break\n            ;;\n        -*)\n            echo \"Unknown option: $1\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n        *)\n            echo \"Unexpected positional argument: $1\" >&2\n            echo \"Use --scenario to specify the test scenario.\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n    esac\ndone\n\n# Require scenario parameter\nif [[ -z \"$testCase\" ]]; then\n    echo \"Error: --scenario parameter is required\" >&2\n    echo \"\" >&2\n    show_help\n    exit 1\nfi\n\n# Set default timestamp if not provided\nif [[ -z \"$timestamp\" ]]; then\n    timestamp=\"$(date '+%Y-%m-%d-%H-%M-%S')\"\nfi\n\ntestCfg=\"testdata/history_simulation_$testCase.yaml\"\ntestName=\"test-$testCase-$timestamp\"\nresultFolder=\"history-simulator-output\"\nmkdir -p \"$resultFolder\"\neventLogsFile=\"$resultFolder/$testName-events.json\"\ntestSummaryFile=\"$resultFolder/$testName-summary.txt\"\n\necho \"Building test image\"\nDOCKERFILE_SUFFIX=$dockerFileSuffix docker compose -f docker/github_actions/docker-compose-local-history-simulation.yml \\\n  build history-simulator\n\nfunction check_test_failure()\n{\n  faillog=$(grep 'FAIL: TestHistorySimulationSuite' -B 10 test.log 2>/dev/null || true)\n  timeoutlog=$(grep 'test timed out' test.log 2>/dev/null || true)\n  if [ -z \"$faillog\" ] && [ -z \"$timeoutlog\" ]; then\n    echo \"Passed\"\n  else\n    echo 'Test failed!!!'\n    echo \"Fail log: $faillog\"\n    echo \"Timeout log: $timeoutlog\"\n    echo \"Check test.log file for more details\"\n    exit 1\n  fi\n}\n\ntrap check_test_failure EXIT\n\necho \"Running the test $testCase\"\nDOCKERFILE_SUFFIX=$dockerFileSuffix docker compose \\\n  -f docker/github_actions/docker-compose-local-history-simulation.yml \\\n  run -e HISTORY_SIMULATION_CONFIG=$testCfg --rm --remove-orphans --service-ports --use-aliases \\\n  history-simulator \\\n  | grep -a --line-buffered \"History New Event\" \\\n  | sed \"s/History New Event: //\" \\\n  | jq . > \"$eventLogsFile\"\n\necho \"---- Simulation Summary ----\"\n\nformat_counts() {\n  event_name=\"$1\"\n  header=\"$2\"\n  echo \"$header:\"\n\n  jq -r --arg event \"$event_name\" '\n    select(.EventName == $event) |\n    \"\\(.ShardID)|\\(.Payload.task_category)|\\(.Payload.task_type)\"\n  ' \"$eventLogsFile\" | sort | uniq -c | sort -k2,2n -k3,3 -k4,4 | awk '\n  BEGIN { last_shard = \"__unset__\" }\n  {\n    count = $1\n    sub(/^[ \\t]*[0-9]+[ \\t]*/, \"\", $0)\n    split($0, parts, \"|\")\n    shard = parts[1]\n    category = parts[2]\n    type = parts[3]\n\n    if (shard != last_shard) {\n      print \"ShardID: \" shard\n      last_shard = shard\n    }\n    printf \"  TaskCategory: %s, TaskType: %s, Count: %d\\n\", category, type, count\n  }'\n\n  echo \"\"\n}\n\nformat_counts \"Create History Task\" \"Tasks created\" | tee -a \"$testSummaryFile\"\nformat_counts \"Execute History Task\" \"Tasks executed\" | tee -a \"$testSummaryFile\"\n\ncreate_tasks=$(jq -r '\n  select(.EventName == \"Create History Task\") |\n    \"\\(.ShardID)|\\(.Payload.task_category)|\\(.Payload.task_type)|\\(.Payload.task_key.taskID)|\\(.Payload.task_key.scheduledTime)\"' \"$eventLogsFile\"\n)\n\nexecute_tasks=$(jq -r '\n  select(.EventName == \"Execute History Task\") |\n    \"\\(.ShardID)|\\(.Payload.task_category)|\\(.Payload.task_type)|\\(.Payload.task_key.taskID)|\\(.Payload.task_key.scheduledTime)\"' \"$eventLogsFile\"\n)\n\nmissing=$(echo \"$create_tasks\" | while IFS= read -r line; do\n  if ! echo \"$execute_tasks\" | grep -Fxq \"$line\"; then\n    echo \"$line\"\n  fi\ndone)\n\necho \"Tasks that were created but not executed:\" | tee -a \"$testSummaryFile\"\n# Group and print nicely\necho \"$missing\" | sort -t '|' -k1,1n -k2,2 -k3,3n -k4,4n | awk -F'|' '\nBEGIN { last_key = \"__none__\"; last_shard = \"__none__\"}\n{\n  shard = $1\n  category = $2\n  type = $3\n  taskid = $4\n  sched = $5\n\n  key = shard \"|\" category \"|\" type\n  if (key != last_key) {\n    if (shard != last_shard) {\n      print \"ShardID: \" shard\n      last_shard = shard\n    }\n    print \"  TaskCategory: \" category \", TaskType: \" type\n    last_key = key\n  }\n  print \"    TaskID: \" taskid \", ScheduledTime: \" sched\n}' | tee -a \"$testSummaryFile\"\n\n\nprintf \"\\nResults are saved in %s\\n\" \"$testSummaryFile\"\nprintf \"For further ad-hoc analysis, please check %s via jq queries\\n\" \"$eventLogsFile\"\nprintf \"Visit http://localhost:3000/ to view Cadence History grafana dashboard\\n\"\n"
  },
  {
    "path": "simulation/history/testdata/history_simulation_default.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enableasyncwfconsumer: false\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"dynamicconfig/default.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "simulation/history/testdata/history_simulation_queuev2.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enableasyncwfconsumer: false\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"dynamicconfig/queuev2.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "simulation/history/testdata/history_simulation_queuev2_split.yaml",
    "content": "enablearchival: false\nclusterno: 0\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 1\nworkerconfig:\n  enableasyncwfconsumer: false\n  enablearchiver: false\n  enablereplicator: false\n  enableindexer: false\ndynamicclientconfig:\n  filepath: \"dynamicconfig/queuev2_split.yaml\"\n  pollInterval: \"10s\"\n"
  },
  {
    "path": "simulation/history/workflow/workflow.go",
    "content": "package workflow\n\nimport (\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n)\n\nfunc RegisterWorker(w worker.Registry) {\n\tw.RegisterWorkflow(NoopWorkflow)\n}\n\nfunc NoopWorkflow(ctx workflow.Context) error {\n\treturn nil\n}\n"
  },
  {
    "path": "simulation/matching/comparison/README.md",
    "content": "\nThis tool runs a set of matching simulation tests, extracts stats from their output and generates a csv to compare them easily.\n\nNote: The parsing logic might break in the future if the `simulation/matching/run.sh` starts spitting different shaped lines. Alternative is to load all the event logs into a sqlite table and then run queries on top instead of parsing outputs of jq in this tool.\n\n\nRun all the scenarios and compare:\n```\ngo run simulation/matching/comparison/*.go\n```\n\nRun subset of scenarios and compare:\n```\ngo run simulation/matching/comparison/*.go \\\n    --scenarios \"fluctuating\"\n```\n\nIf you have already run some scenarios before and made changes in the csv output then run in Compare mode\n```\ngo run simulation/matching/comparison/*.go \\\n    --ts 2024-11-27-21-29-55 \\\n    --mode Compare\n```\n"
  },
  {
    "path": "simulation/matching/comparison/main.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage main\n\nimport (\n\t\"bytes\"\n\t\"encoding/csv\"\n\t\"flag\"\n\t\"fmt\"\n\t\"log\"\n\t\"os\"\n\t\"os/exec\"\n\t\"path\"\n\t\"regexp\"\n\t\"sort\"\n\t\"strings\"\n\t\"time\"\n)\n\nvar (\n\tscenarioFilter = flag.String(\"scenarios\", \".*\", \"Regex to filter the tests to execute by name\")\n\tmode           = flag.String(\"mode\", \"RunAndCompare\", \"Mode to run the tool in. Options: RunAndCompare, Compare. \"+\n\t\t\"RunAndCompare runs the simulation and compares the results. \"+\n\t\t\"Compare only compares the results of previously run results. Compare requires --ts flag to be set.\")\n\ttimestamp = flag.String(\"ts\", \"\", \"Timestamp of the simulation run to compare in following format '2006-01-02-15-04-05'. Required when mode is Compare\")\n)\n\nconst (\n\toutputFolder = \"matching-simulator-output\"\n)\n\nvar (\n\tsimNameRegex = regexp.MustCompile(`matching_simulation_(?P<name>.*).yaml`)\n\n\t// oneLineStatRegex extracts key value pairs that represent a single line stat from the summary file\n\t// e.g. Max Task latency (ms): 8273\n\t//\n\t// See https://regex101.com/r/ZqeiWC/1 for example matches\n\toneLineStatRegex = regexp.MustCompile(`^(?P<key>[a-zA-Z]+[a-zA-Z0-9\\s\\(\\)]+):[\\s]+(?P<val>[+-]?[0-9]*[.]?[0-9]+[s]?)$`)\n\n\t// multiLineStatRegex matches with the start of a multi line stat in the summary file.\n\t// Subsequent lines that start with a space are part of the same stat and parsed in the code.\n\t// e.g.\n\t// Per tasklist sync matches:\n\t//   179 \"/__cadence_sys/my-tasklist/1\"\n\t//   375 \"/__cadence_sys/my-tasklist/2\"\n\t//   470 \"/__cadence_sys/my-tasklist/3\"\n\t//   3222 \"my-tasklist\"\n\t//\n\t// See https://regex101.com/r/Un3yLo/1 for example matches\n\tmultiLineStatRegex = regexp.MustCompile(`(?m)^(?P<multilinestart>[a-zA-Z]+[a-zA-Z0-9\\s\\(\\)]+):[\\s]?$`)\n)\n\nfunc main() {\n\tvalidateAndParseFlags()\n\n\troot := mustGetRootDir()\n\n\tscenarios := mustGetSimulationScenarios(root)\n\n\tts := time.Now().UTC().Format(\"2006-01-02-15-04-05\")\n\tif *timestamp != \"\" {\n\t\tts = *timestamp\n\t}\n\n\tif *mode == \"RunAndCompare\" {\n\t\tfor _, scenario := range scenarios {\n\t\t\tmustRunScenario(root, scenario, ts)\n\t\t}\n\t}\n\n\tmustGenerateReports(root, scenarios, ts)\n}\n\nfunc mustGenerateReports(root string, scenarios []string, ts string) {\n\t// outer key is scenario name, inner key is stat name, value is stat value\n\tcsvData := make(map[string]map[string]string)\n\n\tvar missingScenarios []string\n\tvar missingScenariosReasons []string\n\tfor _, scenario := range scenarios {\n\t\tif reason, ok := scenarioHasRun(root, scenario, ts); !ok {\n\t\t\tmissingScenarios = append(missingScenarios, scenario)\n\t\t\tmissingScenariosReasons = append(missingScenariosReasons, reason)\n\t\t\tcontinue\n\t\t}\n\n\t\tsummaryFile := scenarioSummaryFile(root, scenario, ts)\n\t\tcsvData[scenario] = mustParseSummaryFile(summaryFile, scenario)\n\t}\n\n\tif len(missingScenarios) == len(scenarios) {\n\t\tlog.Fatalf(\"No simulation results found for any of the scenarios for timestamp: %s, reasons:\\n%s\", ts, strings.Join(missingScenariosReasons, \"\\n\"))\n\t}\n\n\tallStatKeys := mustGetAllStatKeys(csvData)\n\theaders := append([]string{\"scenario\"}, allStatKeys...)\n\tvar data [][]string\n\toutputFile := csvFilePath(root)\n\twriter := mustNewCSVWriter(csvFilePath(root))\n\tfor scenario, stats := range csvData {\n\t\trow := []string{scenario}\n\t\tfor _, key := range allStatKeys {\n\t\t\trow = append(row, stats[key])\n\t\t}\n\n\t\tdata = append(data, row)\n\t}\n\n\twriter.Write(headers)\n\tfor _, row := range data {\n\t\twriter.Write(row)\n\t}\n\n\twriter.Flush()\n\tif err := writer.Error(); err != nil {\n\t\tlog.Fatalf(\"Error writing to output file, err: %v\", err)\n\t}\n\n\tfmt.Printf(\"Comparison CSV generated at: %s\\n\", outputFile)\n}\n\nfunc csvFilePath(root string) string {\n\treturn path.Join(root, outputFolder, \"comparison.csv\")\n}\n\nfunc mustGetAllStatKeys(csvData map[string]map[string]string) []string {\n\tallStatKeys := make(map[string]bool)\n\tfor _, stats := range csvData {\n\t\tfor k := range stats {\n\t\t\tallStatKeys[k] = true\n\t\t}\n\t}\n\n\tvar allStatKeysSlice []string\n\tfor k := range allStatKeys {\n\t\tallStatKeysSlice = append(allStatKeysSlice, k)\n\t}\n\n\tsort.Strings(allStatKeysSlice)\n\n\treturn allStatKeysSlice\n}\n\nfunc mustParseSummaryFile(path, scenario string) map[string]string {\n\tcontent, err := os.ReadFile(path)\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not read file %s, err: %v\", path, err)\n\t}\n\tstrContent := string(content)\n\tstats := make(map[string]string)\n\n\t// extract one line stats\n\tfor _, line := range strings.Split(strContent, \"\\n\") {\n\t\tmatches := oneLineStatRegex.FindStringSubmatch(line)\n\t\tif len(matches) == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tstats[matches[oneLineStatRegex.SubexpIndex(\"key\")]] = matches[oneLineStatRegex.SubexpIndex(\"val\")]\n\t}\n\n\tfmt.Printf(\"Scenario %q has %d oneline stats\\n\", scenario, len(stats))\n\n\t// extract multi line stats\n\tindices := multiLineStatRegex.FindAllStringIndex(strContent, -1)\n\tfmt.Printf(\"Scenario %q has %d multiline stats\\n\", scenario, len(indices))\n\tfor _, idx := range indices {\n\t\tstart := idx[0]\n\t\tend := idx[1]\n\t\tkey := strContent[start:end]\n\n\t\t// value is all lines that start with a space after the key\n\t\tvar b bytes.Buffer\n\t\trest := strContent[end:]\n\t\tfor i := strings.Index(rest, \"\\n\"); i < len(rest); i++ {\n\t\t\tif rest[i] == '\\n' && i+1 < len(rest) && rest[i+1] != ' ' {\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\tb.WriteByte(rest[i])\n\t\t}\n\n\t\tstats[key] = b.String()\n\t}\n\n\treturn stats\n}\n\nfunc mustNewCSVWriter(outputFile string) *csv.Writer {\n\tfile, err := os.Create(outputFile)\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not create output file, err: %v\", err)\n\t}\n\n\treturn csv.NewWriter(file)\n}\n\nfunc scenarioSummaryFile(root, scenario, ts string) string {\n\t// e.g. matching-simulator-output/test-default-2024-09-12-18-16-44-summary.txt\n\treturn path.Join(root, fmt.Sprintf(\"matching-simulator-output/test-%s-%s-summary.txt\", scenario, ts))\n}\n\nfunc scenarioRawEventsFile(root, scenario, ts string) string {\n\t// e.g. matching-simulator-output/test-default-2024-09-12-18-16-44-events.json\n\treturn path.Join(root, fmt.Sprintf(\"matching-simulator-output/test-%s-%s-events.json\", scenario, ts))\n}\n\nfunc mustRunScenario(root, scenario, ts string) {\n\tif _, ok := scenarioHasRun(root, scenario, ts); ok {\n\t\tfmt.Printf(\"Scenario %s already ran for timestamp %s, skipping\\n\", scenario, ts)\n\t\treturn\n\t}\n\n\tfmt.Printf(\"Running scenario: %s\\n\", scenario)\n\tstart := time.Now()\n\tcmd := exec.Command(\"bash\", path.Join(root, \"simulation/matching/run.sh\"), \"--scenario\", scenario, \"--timestamp\", ts)\n\tvar stdout, stderr bytes.Buffer\n\tcmd.Stdout = &stdout\n\tcmd.Stderr = &stderr\n\terr := cmd.Run()\n\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not run scenario %s, err: %v,\\n-----stdout:-----\\n%s\\n----stderr:----\\n%s\\nMore details can be found in test.log file.\\n\", scenario, err, stdout.String(), stderr.String())\n\t}\n\n\tfmt.Printf(\"Finished running scenario: %s in %v seconds\\n\", scenario, time.Since(start).Seconds())\n}\n\nfunc mustGetSimulationScenarios(root string) []string {\n\tpath := path.Join(root, \"simulation/matching/testdata\")\n\tentries, err := os.ReadDir(path)\n\tif err != nil {\n\t\tlog.Fatal(err)\n\t}\n\n\tscenarioFilterRegex, err := regexp.Compile(*scenarioFilter)\n\tif err != nil {\n\t\tlog.Fatalf(\"Error parsing scenarios flag: %v\", err)\n\t}\n\n\tvar scenarios []string\n\tfor _, entry := range entries {\n\t\tif entry.IsDir() {\n\t\t\tcontinue\n\t\t}\n\n\t\tmatches := simNameRegex.FindStringSubmatch(entry.Name())\n\t\tif len(matches) == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tscenarioName := matches[simNameRegex.SubexpIndex(\"name\")]\n\t\tif scenarioFilterRegex.MatchString(scenarioName) {\n\t\t\tscenarios = append(scenarios, scenarioName)\n\t\t} else {\n\t\t\tlog.Printf(\"Skipping %s due to scenario filter\", scenarioName)\n\t\t}\n\n\t}\n\n\tfmt.Println(\"Simulation scenarios found: \\n  \", strings.Join(scenarios, \"\\n   \"))\n\treturn scenarios\n}\n\nfunc validateAndParseFlags() {\n\tflag.Parse()\n\n\tif *mode != \"RunAndCompare\" && *mode != \"Compare\" {\n\t\tfmt.Println(\"--mode must be RunAndCompare or Compare\")\n\t\tos.Exit(1)\n\t}\n\n\tif *mode == \"Compare\" && *timestamp == \"\" {\n\t\tfmt.Println(\"--ts is required when mode is Compare\")\n\t\tos.Exit(1)\n\t}\n}\n\nfunc scenarioHasRun(root, scenario, ts string) (string, bool) {\n\t// check if summary file exists and complete\n\tsummaryFilePath := scenarioSummaryFile(root, scenario, ts)\n\t_, err := os.Stat(summaryFilePath)\n\tif os.IsNotExist(err) {\n\t\treturn fmt.Sprintf(\"summary file %s doesn't exist\", summaryFilePath), false\n\t}\n\n\tcontent, err := os.ReadFile(summaryFilePath)\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not read summary file %s, err: %v\", summaryFilePath, err)\n\t}\n\n\tif !strings.Contains(string(content), \"End of summary\") {\n\t\treturn fmt.Sprintf(\"summary file %s doesn't contain 'End of summary'\", summaryFilePath), false\n\t}\n\n\t// check if raw events file exists\n\teventFilePath := scenarioRawEventsFile(root, scenario, ts)\n\t_, err = os.Stat(eventFilePath)\n\tif os.IsNotExist(err) {\n\t\treturn fmt.Sprintf(\"event file %s doesn't exist\", eventFilePath), false\n\t}\n\n\treturn \"\", true\n}\n\nfunc mustGetRootDir() string {\n\troot, err := os.Getwd()\n\tif err != nil {\n\t\tlog.Fatalf(\"Could not get executable path, err: %v\", err)\n\t}\n\n\treturn root\n}\n"
  },
  {
    "path": "simulation/matching/matching_simulation_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n/*\nTo run locally:\n\n1. Pick a scenario from the existing config files simulation/matching/testdata/matching_simulation_.*.yaml or add a new one\n\n2. Run the scenario\n`./simulation/matching/run.sh default`\n\nFull test logs can be found at test.log file. Event json logs can be found at matching-simulator-output folder.\nSee the run.sh script for more details about how to parse events.\n\nIf you want to run multiple scenarios and compare them refer to simulation/matching/comparison/README.md\n*/\npackage matching\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"math/rand\"\n\t\"os\"\n\t\"reflect\"\n\t\"sort\"\n\t\"strings\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\t\"golang.org/x/exp/slices\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/client/history\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/isolationgroup\"\n\t\"github.com/uber/cadence/common/persistence\"\n\tpt \"github.com/uber/cadence/common/persistence/persistence-tests\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/host\"\n\n\t_ \"github.com/uber/cadence/common/asyncworkflow/queue/kafka\" // needed to load kafka asyncworkflow queue\n)\n\ntype operation string\n\nconst (\n\toperationPollForDecisionTask operation = \"PollForDecisionTask\"\n\toperationPollReceivedTask    operation = \"PollReceivedTask\"\n\toperationAddDecisionTask     operation = \"AddDecisionTask\"\n\tdefaultTestCase                        = \"testdata/matching_simulation_default.yaml\"\n)\n\ntype operationStats struct {\n\top        operation\n\tdur       time.Duration\n\terr       error\n\ttimestamp time.Time\n}\n\ntype operationAggStats struct {\n\tsuccessCnt    int\n\tfailCnt       int\n\ttotalDuration time.Duration\n\tmaxDuration   time.Duration\n\tlastUpdated   time.Time\n}\n\ntype MatchingSimulationSuite struct {\n\t*require.Assertions\n\t*host.IntegrationBase\n}\n\nfunc TestMatchingSimulation(t *testing.T) {\n\tflag.Parse()\n\n\tconfPath := os.Getenv(\"MATCHING_SIMULATION_CONFIG\")\n\tif confPath == \"\" {\n\t\tconfPath = defaultTestCase\n\t}\n\tclusterConfig, err := host.GetTestClusterConfig(confPath)\n\tif err != nil {\n\t\tt.Fatalf(\"failed creating cluster config from %s, err: %v\", confPath, err)\n\t}\n\n\tisolationGroups := getIsolationGroups(clusterConfig.MatchingConfig.SimulationConfig)\n\n\tclusterConfig.MatchingDynamicConfigOverrides = map[dynamicproperties.Key]interface{}{\n\t\tdynamicproperties.MatchingNumTasklistWritePartitions:           getPartitions(clusterConfig.MatchingConfig.SimulationConfig.TaskListWritePartitions),\n\t\tdynamicproperties.MatchingNumTasklistReadPartitions:            getPartitions(clusterConfig.MatchingConfig.SimulationConfig.TaskListReadPartitions),\n\t\tdynamicproperties.MatchingForwarderMaxOutstandingPolls:         getForwarderMaxOutstandingPolls(clusterConfig.MatchingConfig.SimulationConfig.ForwarderMaxOutstandingPolls),\n\t\tdynamicproperties.MatchingForwarderMaxOutstandingTasks:         getForwarderMaxOutstandingTasks(clusterConfig.MatchingConfig.SimulationConfig.ForwarderMaxOutstandingTasks),\n\t\tdynamicproperties.MatchingForwarderMaxRatePerSecond:            getForwarderMaxRPS(clusterConfig.MatchingConfig.SimulationConfig.ForwarderMaxRatePerSecond),\n\t\tdynamicproperties.MatchingForwarderMaxChildrenPerNode:          getForwarderMaxChildPerNode(clusterConfig.MatchingConfig.SimulationConfig.ForwarderMaxChildrenPerNode),\n\t\tdynamicproperties.LocalPollWaitTime:                            clusterConfig.MatchingConfig.SimulationConfig.LocalPollWaitTime,\n\t\tdynamicproperties.LocalTaskWaitTime:                            clusterConfig.MatchingConfig.SimulationConfig.LocalTaskWaitTime,\n\t\tdynamicproperties.EnableTasklistIsolation:                      len(isolationGroups) > 0,\n\t\tdynamicproperties.AllIsolationGroups:                           isolationGroups,\n\t\tdynamicproperties.TasklistLoadBalancerStrategy:                 getTasklistLoadBalancerStrategy(clusterConfig.MatchingConfig.SimulationConfig.TasklistLoadBalancerStrategy),\n\t\tdynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache: clusterConfig.MatchingConfig.SimulationConfig.GetPartitionConfigFromDB,\n\t\tdynamicproperties.MatchingEnableAdaptiveScaler:                 clusterConfig.MatchingConfig.SimulationConfig.EnableAdaptiveScaler,\n\t\tdynamicproperties.MatchingPartitionDownscaleFactor:             clusterConfig.MatchingConfig.SimulationConfig.PartitionDownscaleFactor,\n\t\tdynamicproperties.MatchingPartitionUpscaleRPS:                  clusterConfig.MatchingConfig.SimulationConfig.PartitionUpscaleRPS,\n\t\tdynamicproperties.MatchingPartitionUpscaleSustainedDuration:    clusterConfig.MatchingConfig.SimulationConfig.PartitionUpscaleSustainedDuration,\n\t\tdynamicproperties.MatchingPartitionDownscaleSustainedDuration:  clusterConfig.MatchingConfig.SimulationConfig.PartitionDownscaleSustainedDuration,\n\t\tdynamicproperties.MatchingAdaptiveScalerUpdateInterval:         clusterConfig.MatchingConfig.SimulationConfig.AdaptiveScalerUpdateInterval,\n\t\tdynamicproperties.MatchingQPSTrackerInterval:                   getQPSTrackerInterval(clusterConfig.MatchingConfig.SimulationConfig.QPSTrackerInterval),\n\t\tdynamicproperties.TaskIsolationDuration:                        clusterConfig.MatchingConfig.SimulationConfig.TaskIsolationDuration,\n\t}\n\n\tctrl := gomock.NewController(t)\n\tmockHistoryCl := history.NewMockClient(ctrl)\n\tmockHistoryCl.EXPECT().RecordDecisionTaskStarted(gomock.Any(), gomock.Any()).\n\t\tDoAndReturn(func(ctx context.Context, req *types.RecordDecisionTaskStartedRequest, opts ...yarpc.CallOption) (*types.RecordDecisionTaskStartedResponse, error) {\n\t\t\ttime.Sleep(getRecordDecisionTaskStartedTime(clusterConfig.MatchingConfig.SimulationConfig.RecordDecisionTaskStartedTime))\n\t\t\treturn &types.RecordDecisionTaskStartedResponse{\n\t\t\t\tScheduledEventID: req.ScheduleID,\n\t\t\t}, nil\n\t\t}).AnyTimes()\n\tclusterConfig.HistoryConfig.MockClient = mockHistoryCl\n\n\ttestCluster := host.NewPersistenceTestCluster(t, clusterConfig)\n\n\ts := new(MatchingSimulationSuite)\n\tparams := host.IntegrationBaseParams{\n\t\tDefaultTestCluster:    testCluster,\n\t\tVisibilityTestCluster: testCluster,\n\t\tTestClusterConfig:     clusterConfig,\n\t}\n\ts.IntegrationBase = host.NewIntegrationBase(params)\n\tsuite.Run(t, s)\n}\n\nfunc (s *MatchingSimulationSuite) SetupSuite() {\n\ts.SetupLogger()\n\n\ts.Logger.Info(\"Running integration test against test cluster\")\n\tclusterMetadata := host.NewClusterMetadata(s.T(), s.TestClusterConfig)\n\tdc := persistence.DynamicConfiguration{\n\t\tEnableCassandraAllConsistencyLevelDelete: dynamicproperties.GetBoolPropertyFn(true),\n\t\tPersistenceSampleLoggingRate:             dynamicproperties.GetIntPropertyFn(100),\n\t\tEnableShardIDMetrics:                     dynamicproperties.GetBoolPropertyFn(true),\n\t\tEnableHistoryTaskDualWriteMode:           dynamicproperties.GetBoolPropertyFn(true),\n\t\tSerializationEncoding:                    dynamicproperties.GetStringPropertyFn(string(constants.EncodingTypeThriftRW)),\n\t}\n\tparams := pt.TestBaseParams{\n\t\tDefaultTestCluster:    s.DefaultTestCluster,\n\t\tVisibilityTestCluster: s.VisibilityTestCluster,\n\t\tClusterMetadata:       clusterMetadata,\n\t\tDynamicConfiguration:  dc,\n\t}\n\tcluster, err := host.NewCluster(s.T(), s.TestClusterConfig, s.Logger, params)\n\ts.Require().NoError(err)\n\ts.TestCluster = cluster\n\ts.Engine = s.TestCluster.GetFrontendClient()\n\ts.AdminClient = s.TestCluster.GetAdminClient()\n\n\ts.DomainName = s.RandomizeStr(\"integration-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.DomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\ts.SecondaryDomainName = s.RandomizeStr(\"unused-test-domain\")\n\ts.Require().NoError(s.RegisterDomain(s.SecondaryDomainName, 1, types.ArchivalStatusDisabled, \"\", types.ArchivalStatusDisabled, \"\", nil))\n\n\ttime.Sleep(2 * time.Second)\n}\n\nfunc (s *MatchingSimulationSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *MatchingSimulationSuite) TearDownSuite() {\n\t// Sleep for a while to ensure all metrics are emitted/scraped by prometheus\n\ttime.Sleep(5 * time.Second)\n\ts.TearDownBaseSuite()\n}\n\nfunc (s *MatchingSimulationSuite) TestMatchingSimulation() {\n\tmatchingClients := s.TestCluster.GetMatchingClients()\n\n\tctx, cancel := context.WithCancel(context.Background())\n\n\tdomainID := s.domainID(ctx)\n\ttasklist := \"my-tasklist\"\n\n\tif s.TestClusterConfig.MatchingConfig.SimulationConfig.GetPartitionConfigFromDB &&\n\t\t!s.TestClusterConfig.MatchingConfig.SimulationConfig.EnableAdaptiveScaler {\n\t\t_, err := s.TestCluster.GetMatchingClient().UpdateTaskListPartitionConfig(ctx, &types.MatchingUpdateTaskListPartitionConfigRequest{\n\t\t\tDomainUUID:   domainID,\n\t\t\tTaskList:     &types.TaskList{Name: tasklist, Kind: types.TaskListKindNormal.Ptr()},\n\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\tReadPartitions:  getPartitions(s.TestClusterConfig.MatchingConfig.SimulationConfig.TaskListReadPartitions),\n\t\t\t\tWritePartitions: getPartitions(s.TestClusterConfig.MatchingConfig.SimulationConfig.TaskListWritePartitions),\n\t\t\t},\n\t\t})\n\t\ts.NoError(err)\n\t}\n\n\t// Start stat collector\n\tstatsCh := make(chan *operationStats, 200000)\n\taggStats := make(map[operation]*operationAggStats)\n\tvar collectorWG sync.WaitGroup\n\tcollectorWG.Add(1)\n\tgo s.collectStats(statsCh, aggStats, &collectorWG)\n\n\ttotalTaskCount := getTotalTasks(s.TestClusterConfig.MatchingConfig.SimulationConfig.Tasks)\n\tseed := time.Now().UnixNano()\n\trand.Seed(seed)\n\ttotalBacklogCount := 0\n\tfor idx, backlogConfig := range s.TestClusterConfig.MatchingConfig.SimulationConfig.Backlogs {\n\t\ttotalBacklogCount += backlogConfig.BacklogCount\n\t\tpartition := getPartitionTaskListName(tasklist, backlogConfig.Partition)\n\t\tfor i := 0; i < backlogConfig.BacklogCount; i++ {\n\t\t\tisolationGroup := \"\"\n\t\t\tif len(backlogConfig.IsolationGroups) > 0 {\n\t\t\t\tisolationGroup = randomlyPickKey(backlogConfig.IsolationGroups)\n\t\t\t}\n\t\t\tdecisionTask := newDecisionTask(domainID, partition, isolationGroup, idx)\n\t\t\treqCtx, cancel := context.WithTimeout(ctx, 2*time.Second)\n\t\t\t_, err := matchingClients[0].AddDecisionTask(reqCtx, decisionTask)\n\t\t\tcancel()\n\t\t\tif err != nil {\n\t\t\t\ts.log(\"Error when adding decision task, err: %v\", err)\n\t\t\t}\n\t\t}\n\t}\n\n\t// Start pollers\n\tnumPollers := 0\n\tvar tasksToReceive sync.WaitGroup\n\ttasksToReceive.Add(totalTaskCount + totalBacklogCount)\n\tvar pollerWG sync.WaitGroup\n\tfor idx, pollerConfig := range s.TestClusterConfig.MatchingConfig.SimulationConfig.Pollers {\n\t\tfor i := 0; i < getNumPollers(pollerConfig); i++ {\n\t\t\tnumPollers++\n\t\t\tpollerWG.Add(1)\n\t\t\tpollerID := fmt.Sprintf(\"[%d]-%s-%d\", idx, getIsolationGroup(pollerConfig), i)\n\t\t\tconfig := pollerConfig\n\t\t\tgo s.poll(ctx, matchingClients[i%len(matchingClients)], domainID, tasklist, pollerID, &pollerWG, statsCh, &tasksToReceive, config)\n\t\t}\n\t}\n\n\t// wait a bit for pollers to start.\n\ttime.Sleep(300 * time.Millisecond)\n\n\tstartTime := time.Now()\n\t// Start task generators\n\tnumGenerators := 0\n\tvar generatorWG sync.WaitGroup\n\tlastTaskScheduleID := int32(0)\n\tfor _, taskConfig := range s.TestClusterConfig.MatchingConfig.SimulationConfig.Tasks {\n\t\ttasksGenerated := int32(0)\n\t\trateLimiter := newSimulationRateLimiter(taskConfig, startTime, clock.NewRealTimeSource(), s.log)\n\t\tfor i := 0; i < getNumTaskGenerators(taskConfig); i++ {\n\t\t\tnumGenerators++\n\t\t\tgeneratorWG.Add(1)\n\t\t\tconfig := taskConfig\n\t\t\tgo s.generate(\n\t\t\t\tctx,\n\t\t\t\tmatchingClients[i%len(matchingClients)],\n\t\t\t\tdomainID,\n\t\t\t\ttasklist,\n\t\t\t\t&tasksGenerated,\n\t\t\t\t&lastTaskScheduleID,\n\t\t\t\t&generatorWG,\n\t\t\t\tstatsCh,\n\t\t\t\tconfig,\n\t\t\t\trateLimiter,\n\t\t\t)\n\t\t}\n\t}\n\n\t// Let it run until all tasks have been polled.\n\t// There's a test timeout configured in docker/github_actions/docker-compose-local-matching-simulation.yml that you\n\t// can change if your test case needs more time\n\ts.log(\"Waiting until all tasks are received\")\n\ttasksToReceive.Wait()\n\texecutionTime := time.Since(startTime)\n\ts.log(\"Completed benchmark in %v\", executionTime)\n\ts.log(\"Canceling context to stop pollers and task generators\")\n\tcancel()\n\tpollerWG.Wait()\n\ts.log(\"Pollers stopped\")\n\tgeneratorWG.Wait()\n\ts.log(\"Generators stopped\")\n\ts.log(\"Stopping stats collector\")\n\tclose(statsCh)\n\tcollectorWG.Wait()\n\ts.log(\"Stats collector stopped\")\n\n\t// Print the test summary.\n\t// Don't change the start/end line format as it is used by scripts to parse the summary info\n\ttestSummary := []string{}\n\ttestSummary = append(testSummary, \"Simulation Summary:\")\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Random seed: %v\", seed))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Task generate Duration: %v\", aggStats[operationAddDecisionTask].lastUpdated.Sub(startTime)))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Simulation Duration: %v\", executionTime))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Num of Pollers: %d\", numPollers))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Num of Task Generators: %d\", numGenerators))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Record Decision Task Started Time: %v\", s.TestClusterConfig.MatchingConfig.SimulationConfig.RecordDecisionTaskStartedTime))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Num of Write Partitions: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingNumTasklistWritePartitions]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Num of Read Partitions: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingNumTasklistReadPartitions]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Get Num of Partitions from DB: %v\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingEnableGetNumberOfPartitionsFromCache]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Tasklist load balancer strategy: %v\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.TasklistLoadBalancerStrategy]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Forwarder Max Outstanding Polls: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingForwarderMaxOutstandingPolls]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Forwarder Max Outstanding Tasks: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingForwarderMaxOutstandingTasks]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Forwarder Max RPS: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingForwarderMaxRatePerSecond]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Forwarder Max Children per Node: %d\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.MatchingForwarderMaxChildrenPerNode]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Local Poll Wait Time: %v\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.LocalPollWaitTime]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Local Task Wait Time: %v\", s.TestClusterConfig.MatchingDynamicConfigOverrides[dynamicproperties.LocalTaskWaitTime]))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Tasks generated: %d\", aggStats[operationAddDecisionTask].successCnt))\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Tasks polled: %d\", aggStats[operationPollReceivedTask].successCnt))\n\n\ttestSummary = appendMetric(testSummary, operationPollForDecisionTask, aggStats)\n\ttestSummary = appendMetric(testSummary, operationAddDecisionTask, aggStats)\n\n\ttestSummary = append(testSummary, \"End of Simulation Summary\")\n\tfmt.Println(strings.Join(testSummary, \"\\n\"))\n}\n\nfunc (s *MatchingSimulationSuite) log(msg string, args ...interface{}) {\n\tmsg = time.Now().Format(time.RFC3339Nano) + \"\\t\" + msg\n\ts.T().Logf(msg, args...)\n}\n\nfunc (s *MatchingSimulationSuite) generate(\n\tctx context.Context,\n\tmatchingClient host.MatchingClient,\n\tdomainID, tasklist string,\n\ttasksGenerated *int32,\n\tlastTaskScheduleID *int32,\n\twg *sync.WaitGroup,\n\tstatsCh chan *operationStats,\n\ttaskConfig host.SimulationTaskConfiguration,\n\trateLimiter *simulationRateLimiter,\n) {\n\tdefer wg.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\ts.log(\"Generator done\")\n\t\t\treturn\n\t\tdefault:\n\t\t\tif err := rateLimiter.Wait(ctx); err != nil {\n\t\t\t\tif !errors.Is(err, context.Canceled) {\n\t\t\t\t\ts.T().Error(\"Rate limiter failed: \", err)\n\t\t\t\t}\n\t\t\t\treturn\n\t\t\t}\n\t\t\tnewTasksGenerated := int(atomic.AddInt32(tasksGenerated, 1))\n\t\t\tif newTasksGenerated > getMaxTasksToGenerate(taskConfig) {\n\t\t\t\ts.log(\"Generated %d tasks so generator will stop\", newTasksGenerated)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tisolationGroup := \"\"\n\t\t\tif igs := getTaskIsolationGroups(taskConfig); len(igs) > 0 {\n\t\t\t\tisolationGroup = igs[newTasksGenerated%len(igs)]\n\t\t\t}\n\t\t\tscheduleID := int(atomic.AddInt32(lastTaskScheduleID, 1))\n\t\t\tstart := time.Now()\n\t\t\tdecisionTask := newDecisionTask(domainID, tasklist, isolationGroup, scheduleID)\n\t\t\treqCtx, cancel := context.WithTimeout(ctx, 2*time.Second)\n\t\t\t_, err := matchingClient.AddDecisionTask(reqCtx, decisionTask)\n\t\t\tstatsCh <- &operationStats{\n\t\t\t\top:        operationAddDecisionTask,\n\t\t\t\tdur:       time.Since(start),\n\t\t\t\terr:       err,\n\t\t\t\ttimestamp: time.Now(),\n\t\t\t}\n\t\t\tcancel()\n\t\t\tif err != nil {\n\t\t\t\ts.log(\"Error when adding decision task, err: %v\", err)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc (s *MatchingSimulationSuite) poll(\n\tctx context.Context,\n\tmatchingClient host.MatchingClient,\n\tdomainID, tasklist, pollerID string,\n\twg *sync.WaitGroup,\n\tstatsCh chan *operationStats,\n\ttasksToReceive *sync.WaitGroup,\n\tpollerConfig host.SimulationPollerConfiguration,\n) {\n\tdefer wg.Done()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ctx.Done():\n\t\t\ts.log(\"Poller done\")\n\t\t\treturn\n\t\tdefault:\n\t\t\ts.log(\"Poller will initiate a poll\")\n\t\t\treqCtx, cancel := context.WithTimeout(ctx, getPollTimeout(pollerConfig))\n\t\t\tstart := time.Now()\n\t\t\tresp, err := matchingClient.PollForDecisionTask(reqCtx, &types.MatchingPollForDecisionTaskRequest{\n\t\t\t\tDomainUUID: domainID,\n\t\t\t\tPollerID:   pollerID,\n\t\t\t\tPollRequest: &types.PollForDecisionTaskRequest{\n\t\t\t\t\tTaskList: &types.TaskList{\n\t\t\t\t\t\tName: tasklist,\n\t\t\t\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t\t\t\t},\n\t\t\t\t\tIdentity: pollerID,\n\t\t\t\t},\n\t\t\t\tIsolationGroup: getIsolationGroup(pollerConfig),\n\t\t\t})\n\t\t\tcancel()\n\n\t\t\tstatsCh <- &operationStats{\n\t\t\t\top:        operationPollForDecisionTask,\n\t\t\t\tdur:       time.Since(start),\n\t\t\t\terr:       err,\n\t\t\t\ttimestamp: time.Now(),\n\t\t\t}\n\n\t\t\tif err != nil {\n\t\t\t\ts.log(\"PollForDecisionTask failed: %v\", err)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tempty := &types.MatchingPollForDecisionTaskResponse{}\n\n\t\t\tif reflect.DeepEqual(empty, resp) {\n\t\t\t\ts.log(\"PollForDecisionTask response is empty\")\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tstatsCh <- &operationStats{\n\t\t\t\top:        operationPollReceivedTask,\n\t\t\t\ttimestamp: time.Now(),\n\t\t\t}\n\n\t\t\ts.log(\"PollForDecisionTask got a task with startedid: %d. resp: %+v\", resp.StartedEventID, resp)\n\t\t\ttasksToReceive.Done()\n\t\t\ttime.Sleep(getTaskProcessTime(pollerConfig))\n\t\t}\n\t}\n}\n\nfunc (s *MatchingSimulationSuite) collectStats(statsCh chan *operationStats, aggStats map[operation]*operationAggStats, wg *sync.WaitGroup) {\n\tdefer wg.Done()\n\tfor stat := range statsCh {\n\t\topAggStats, ok := aggStats[stat.op]\n\t\tif !ok {\n\t\t\topAggStats = &operationAggStats{}\n\t\t\taggStats[stat.op] = opAggStats\n\t\t}\n\t\tif stat.timestamp.After(opAggStats.lastUpdated) {\n\t\t\topAggStats.lastUpdated = stat.timestamp\n\t\t}\n\t\tif stat.err != nil {\n\t\t\topAggStats.failCnt++\n\t\t} else {\n\t\t\topAggStats.successCnt++\n\t\t}\n\n\t\topAggStats.totalDuration += stat.dur\n\t\tif stat.dur > opAggStats.maxDuration {\n\t\t\topAggStats.maxDuration = stat.dur\n\t\t}\n\t}\n\n\ts.log(\"Stats collector done\")\n}\n\nfunc (s *MatchingSimulationSuite) domainID(ctx context.Context) string {\n\treqCtx, cancel := context.WithTimeout(ctx, 250*time.Millisecond)\n\tdefer cancel()\n\tdomainDesc, err := s.TestCluster.GetFrontendClient().DescribeDomain(reqCtx, &types.DescribeDomainRequest{\n\t\tName: &s.DomainName,\n\t})\n\ts.Require().NoError(err, \"Error when describing domain\")\n\n\tdomainID := domainDesc.GetDomainInfo().UUID\n\ts.T().Logf(\"DomainID: %s\", domainID)\n\treturn domainID\n}\n\nfunc newDecisionTask(domainID, tasklist, isolationGroup string, i int) *types.AddDecisionTaskRequest {\n\treturn &types.AddDecisionTaskRequest{\n\t\tDomainUUID: domainID,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      uuid.New(),\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: tasklist,\n\t\t\tKind: types.TaskListKindNormal.Ptr(),\n\t\t},\n\t\tScheduleID: int64(i),\n\t\tPartitionConfig: map[string]string{\n\t\t\tisolationgroup.GroupKey: isolationGroup,\n\t\t},\n\t}\n}\n\nfunc appendMetric(testSummary []string, op operation, aggStats map[operation]*operationAggStats) []string {\n\ttotal := 0\n\tif pollStats, ok := aggStats[op]; ok {\n\t\ttotal = pollStats.successCnt + pollStats.failCnt\n\t}\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Operation Summary (%v): \", op))\n\n\tif total == 0 {\n\t\ttestSummary = append(testSummary, \"  N/A\")\n\t} else {\n\t\ttestSummary = append(testSummary, fmt.Sprintf(\"  Total: %d\", total))\n\t\ttestSummary = append(testSummary, fmt.Sprintf(\"  Failure rate %%: %d\", 100*aggStats[op].failCnt/total))\n\t\ttestSummary = append(testSummary, fmt.Sprintf(\"  Avg duration (ms): %d\", (aggStats[op].totalDuration/time.Duration(total)).Milliseconds()))\n\t\ttestSummary = append(testSummary, fmt.Sprintf(\"  Max duration (ms): %d\", aggStats[op].maxDuration.Milliseconds()))\n\t}\n\treturn testSummary\n}\n\nfunc getTotalTasks(tasks []host.SimulationTaskConfiguration) int {\n\ttotal := 0\n\tfor _, taskConfiguration := range tasks {\n\t\ttotal += getMaxTasksToGenerate(taskConfiguration)\n\t}\n\treturn total\n}\n\nfunc getIsolationGroups(c host.MatchingSimulationConfig) []any {\n\tgroups := make(map[string]struct{})\n\tfor _, poller := range c.Pollers {\n\t\tif getIsolationGroup(poller) != \"\" {\n\t\t\tgroups[getIsolationGroup(poller)] = struct{}{}\n\t\t}\n\t}\n\tfor _, task := range c.Tasks {\n\t\tfor _, group := range getTaskIsolationGroups(task) {\n\t\t\tgroups[group] = struct{}{}\n\t\t}\n\t}\n\tvar uniqueGroups []any\n\tfor group := range groups {\n\t\tuniqueGroups = append(uniqueGroups, group)\n\t}\n\treturn uniqueGroups\n}\n\nfunc getNumTaskGenerators(c host.SimulationTaskConfiguration) int {\n\tif c.NumTaskGenerators == 0 {\n\t\treturn 1\n\t}\n\treturn c.NumTaskGenerators\n}\n\nfunc getMaxTasksToGenerate(c host.SimulationTaskConfiguration) int {\n\tif c.MaxTaskToGenerate == 0 {\n\t\treturn 2000\n\t}\n\treturn c.MaxTaskToGenerate\n}\n\nfunc getTasksPerSecond(c host.SimulationTaskConfiguration) int {\n\tif c.TasksPerSecond == 0 {\n\t\treturn 40\n\t}\n\treturn c.TasksPerSecond\n}\n\nfunc getTasksBurst(c host.SimulationTaskConfiguration) int {\n\tif c.TasksBurst == 0 {\n\t\treturn 1\n\t}\n\treturn c.TasksBurst\n}\n\nfunc getTaskIsolationGroups(c host.SimulationTaskConfiguration) []string {\n\treturn c.IsolationGroups\n}\n\nfunc getPartitions(i int) map[int]*types.TaskListPartition {\n\tif i == 0 {\n\t\treturn map[int]*types.TaskListPartition{\n\t\t\t0: {},\n\t\t}\n\t}\n\tresult := make(map[int]*types.TaskListPartition)\n\tfor j := 0; j < i; j++ {\n\t\tresult[j] = &types.TaskListPartition{}\n\t}\n\treturn result\n}\n\nfunc getForwarderMaxOutstandingPolls(i int) int {\n\treturn i\n}\n\nfunc getForwarderMaxOutstandingTasks(i int) int {\n\treturn i\n}\n\nfunc getForwarderMaxRPS(i int) int {\n\tif i == 0 {\n\t\treturn 10\n\t}\n\treturn i\n}\n\nfunc getForwarderMaxChildPerNode(i int) int {\n\tif i == 0 {\n\t\treturn 20\n\t}\n\treturn i\n}\n\nfunc getNumPollers(c host.SimulationPollerConfiguration) int {\n\tif c.NumPollers == 0 {\n\t\treturn 1\n\t}\n\treturn c.NumPollers\n}\n\nfunc getTaskProcessTime(c host.SimulationPollerConfiguration) time.Duration {\n\tif c.TaskProcessTime == 0 {\n\t\treturn time.Millisecond\n\t}\n\treturn c.TaskProcessTime\n}\n\nfunc getPollTimeout(c host.SimulationPollerConfiguration) time.Duration {\n\tif c.PollTimeout == 0 {\n\t\treturn 15 * time.Second\n\t}\n\treturn c.PollTimeout\n}\n\nfunc getIsolationGroup(c host.SimulationPollerConfiguration) string {\n\treturn c.IsolationGroup\n}\n\nfunc getRecordDecisionTaskStartedTime(duration time.Duration) time.Duration {\n\tif duration == 0 {\n\t\treturn time.Millisecond\n\t}\n\n\treturn duration\n}\n\nfunc getTasklistLoadBalancerStrategy(strategy string) string {\n\tif strategy == \"\" {\n\t\treturn \"random\"\n\t}\n\treturn strategy\n}\n\nfunc getPartitionTaskListName(root string, partition int) string {\n\tif partition <= 0 {\n\t\treturn root\n\t}\n\treturn fmt.Sprintf(\"%v%v/%v\", constants.ReservedTaskListPrefix, root, partition)\n}\n\nfunc getQPSTrackerInterval(duration time.Duration) time.Duration {\n\tif duration == 0 {\n\t\treturn 10 * time.Second\n\t}\n\treturn duration\n}\n\nfunc randomlyPickKey(weights map[string]int) string {\n\t// Calculate the total weight\n\ttotalWeight := 0\n\tfor _, weight := range weights {\n\t\ttotalWeight += weight\n\t}\n\n\t// Generate a random number between 0 and totalWeight - 1\n\trandomWeight := rand.Intn(totalWeight)\n\n\t// Iterate through the map to find the key corresponding to the random weight\n\tfor key, weight := range weights {\n\t\tif randomWeight < weight {\n\t\t\treturn key\n\t\t}\n\t\trandomWeight -= weight\n\t}\n\n\t// Return an empty string as a fallback (should not happen if weights are positive)\n\treturn \"\"\n}\n\ntype rateLimiterForTimeRange struct {\n\tlimiter    *rate.Limiter\n\tstart, end int\n}\n\nfunc (r *rateLimiterForTimeRange) String() string {\n\treturn fmt.Sprintf(\"{start: %d, end: %d}\", r.start, r.end)\n}\n\ntype simulationRateLimiter struct {\n\tstartTime    time.Time\n\ttimeSrc      clock.TimeSource\n\trateLimiters []*rateLimiterForTimeRange\n\tlogFn        func(msg string, args ...interface{})\n}\n\nfunc newSimulationRateLimiter(\n\ttaskConfig host.SimulationTaskConfiguration,\n\tstartTime time.Time,\n\ttimeSrc clock.TimeSource,\n\tlogFn func(msg string, args ...interface{}),\n) *simulationRateLimiter {\n\tvar rateLimiters []*rateLimiterForTimeRange\n\tif len(taskConfig.OverTime) == 0 {\n\t\tl := rate.NewLimiter(rate.Limit(getTasksPerSecond(taskConfig)), getTasksBurst(taskConfig))\n\t\trateLimiters = append(rateLimiters, &rateLimiterForTimeRange{limiter: l, start: 0, end: -1})\n\t} else {\n\t\tstart := 0\n\t\tfor _, spec := range taskConfig.OverTime {\n\t\t\tl := rate.NewLimiter(rate.Limit(spec.TasksPerSecond), spec.TasksBurst)\n\t\t\tend := -1\n\t\t\tif spec.Duration != nil {\n\t\t\t\tend = start + int(spec.Duration.Seconds())\n\t\t\t}\n\t\t\trateLimiters = append(rateLimiters, &rateLimiterForTimeRange{limiter: l, start: start, end: end})\n\t\t\tstart = end\n\t\t}\n\t}\n\n\tsort.Slice(rateLimiters, func(i, j int) bool {\n\t\treturn rateLimiters[i].start < rateLimiters[j].start\n\t})\n\n\tlogFn(\"Rate limiters: %v\", rateLimiters)\n\n\treturn &simulationRateLimiter{\n\t\tstartTime:    startTime,\n\t\ttimeSrc:      timeSrc,\n\t\trateLimiters: rateLimiters,\n\t\tlogFn:        logFn,\n\t}\n}\n\nfunc (r *simulationRateLimiter) Wait(ctx context.Context) error {\n\tlimiter, err := r.getLimiter()\n\tif err != nil {\n\t\treturn err\n\t}\n\n\treturn limiter.limiter.Wait(ctx)\n}\n\nfunc (r *simulationRateLimiter) getLimiter() (*rateLimiterForTimeRange, error) {\n\telapsed := int(r.timeSrc.Since(r.startTime).Seconds())\n\tidx, ok := slices.BinarySearchFunc(r.rateLimiters, elapsed, func(r *rateLimiterForTimeRange, t int) int {\n\t\tif t >= r.start && (r.end == -1 || t < r.end) {\n\t\t\treturn 0\n\t\t}\n\n\t\tif r.start > t {\n\t\t\treturn 1\n\t\t}\n\n\t\treturn -1\n\t})\n\n\tif !ok {\n\t\treturn nil, fmt.Errorf(\"rate limiter not found, elapsed: %ds\", elapsed)\n\t}\n\n\tr.logFn(\"Elapsed %vs so using rate limiter at index %d\", elapsed, idx)\n\treturn r.rateLimiters[idx], nil\n}\n\nfunc TestMatchingSimulation_RateLimiterBST(t *testing.T) {\n\tmockTimeSrc := clock.NewMockedTimeSource()\n\tsrl := &simulationRateLimiter{\n\t\tstartTime: mockTimeSrc.Now(),\n\t\ttimeSrc:   mockTimeSrc,\n\t\trateLimiters: []*rateLimiterForTimeRange{\n\t\t\t{limiter: rate.NewLimiter(rate.Limit(10), 1), start: 0, end: 5},\n\t\t\t{limiter: rate.NewLimiter(rate.Limit(10), 1), start: 5, end: -1},\n\t\t},\n\t\tlogFn: t.Logf,\n\t}\n\n\t// t = 0\n\tl, err := srl.getLimiter()\n\trequire.NoError(t, err)\n\trequire.Equal(t, 0, l.start) // limiter at index 0 should be used\n\n\t// t = 3\n\tmockTimeSrc.Advance(time.Second * 3)\n\tl, err = srl.getLimiter()\n\trequire.NoError(t, err)\n\trequire.Equal(t, 0, l.start) // limiter at index 0 should be used\n\n\t// t = 5\n\tmockTimeSrc.Advance(time.Second * 2)\n\tl, err = srl.getLimiter()\n\trequire.NoError(t, err)\n\trequire.Equal(t, 5, l.start) // limiter at index 1 should be used\n\n\t// t = 10\n\tmockTimeSrc.Advance(time.Second * 5)\n\tl, err = srl.getLimiter()\n\trequire.NoError(t, err)\n\trequire.Equal(t, 5, l.start) // limiter at index 1 should be used\n}\n"
  },
  {
    "path": "simulation/matching/run.sh",
    "content": "#!/bin/bash\n\n# Cadence Matching Simulation Test Script\n#\n# Usage:\n#   ./simulation/matching/run.sh [OPTIONS]\n#\n# Examples:\n#   # Run default scenario\n#   ./simulation/matching/run.sh --scenario default\n#\n#   # Run specific scenario\n#   ./simulation/matching/run.sh --scenario throughput\n#\n#   # Run with custom timestamp\n#   ./simulation/matching/run.sh --scenario default --timestamp 2024-01-15-10-30-00\n#\n#   # Run with custom dockerfile\n#   ./simulation/matching/run.sh --scenario throughput --dockerfile-suffix .local\n#\n# TODO: Add README with more information on how to analyse the test results, as well as more detailed information on the comparison tests.  \n\nset -eo pipefail\n\nshow_help() {\n    cat << EOF\nCadence Matching Simulation Test Script\n\nUSAGE:\n    $0 [OPTIONS]\n\nOPTIONS:\n    -s, --scenario SCENARIO      Test scenario to run (required)\n                                Corresponds to testdata/matching_simulation_SCENARIO.yaml\n\n    -t, --timestamp TIMESTAMP   Custom timestamp for test naming (default: current time)\n                                Format: YYYY-MM-DD-HH-MM-SS\n\n    -d, --dockerfile-suffix SUFFIX  Dockerfile suffix for custom builds (default: empty)\n                                   Example: .local for Dockerfile.local\n\n    -h, --help                  Show this help message\n\nEXAMPLES:\n    # Run default scenario\n    $0 --scenario default\n\n    # Run specific scenario\n    $0 --scenario throughput\n\n    # Run with custom timestamp\n    $0 --scenario default --timestamp 2024-01-15-10-30-00\n\n    # Run with custom dockerfile\n    $0 --scenario throughput --dockerfile-suffix .local\n\nEOF\n}\n\n# Default values\ntestCase=\"\"\ntimestamp=\"\"\ndockerFileSuffix=\"\"\n\n# Parse command line arguments\nwhile [[ $# -gt 0 ]]; do\n    case $1 in\n        -s|--scenario)\n            testCase=\"$2\"\n            shift 2\n            ;;\n        -t|--timestamp)\n            timestamp=\"$2\"\n            shift 2\n            ;;\n        -d|--dockerfile-suffix)\n            dockerFileSuffix=\"$2\"\n            shift 2\n            ;;\n        -h|--help)\n            show_help\n            exit 0\n            ;;\n        --)\n            shift\n            break\n            ;;\n        -*)\n            echo \"Unknown option: $1\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n        *)\n            echo \"Unexpected positional argument: $1\" >&2\n            echo \"Use --scenario to specify the test scenario.\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n    esac\ndone\n\n# Require scenario parameter\nif [[ -z \"$testCase\" ]]; then\n    echo \"Error: --scenario parameter is required\" >&2\n    echo \"\" >&2\n    show_help\n    exit 1\nfi\n\n# Set default timestamp if not provided\nif [[ -z \"$timestamp\" ]]; then\n    timestamp=\"$(date '+%Y-%m-%d-%H-%M-%S')\"\nfi\n\ntestCfg=\"testdata/matching_simulation_$testCase.yaml\"\ntestName=\"test-$testCase-$timestamp\"\nresultFolder=\"matching-simulator-output\"\nmkdir -p \"$resultFolder\"\neventLogsFile=\"$resultFolder/$testName-events.json\"\ntestSummaryFile=\"$resultFolder/$testName-summary.txt\"\n\necho \"Building test image\"\nDOCKERFILE_SUFFIX=$dockerFileSuffix docker compose -f docker/github_actions/docker-compose-local-matching-simulation.yml \\\n  build matching-simulator\n\nfunction check_test_failure()\n{\n  faillog=$(grep 'FAIL: TestMatchingSimulationSuite' -B 10 test.log 2>/dev/null || true)\n  timeoutlog=$(grep 'test timed out' test.log 2>/dev/null || true)\n  if [ -z \"$faillog\" ] && [ -z \"$timeoutlog\" ]; then\n    echo \"Passed\"\n  else\n    echo 'Test failed!!!'\n    echo \"Fail log: $faillog\"\n    echo \"Timeout log: $timeoutlog\"\n    echo \"Check test.log file for more details\"\n    exit 1\n  fi\n}\n\ntrap check_test_failure EXIT\n\necho \"Running the test $testCase\"\nDOCKERFILE_SUFFIX=$dockerFileSuffix docker compose \\\n  -f docker/github_actions/docker-compose-local-matching-simulation.yml \\\n  run -e MATCHING_SIMULATION_CONFIG=$testCfg --rm --remove-orphans --service-ports --use-aliases \\\n  matching-simulator \\\n  | grep -a --line-buffered \"Matching New Event\" \\\n  | sed \"s/Matching New Event: //\" \\\n  | jq . > \"$eventLogsFile\"\n\necho \"---- Simulation Summary ----\"\ncat test.log \\\n  | sed -n '/Simulation Summary/,/End of Simulation Summary/p' \\\n  | grep -v \"Simulation Summary\" \\\n  | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | awk '{s+=$0}END{print s/NR}')\necho \"Avg Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | sort -n | awk '{a[NR]=$0}END{print a[int(NR*0.50)]}')\necho \"P50 Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | sort -n | awk '{a[NR]=$0}END{print a[int(NR*0.75)]}')\necho \"P75 Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | sort -n | awk '{a[NR]=$0}END{print a[int(NR*0.95)]}')\necho \"P95 Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | sort -n | awk '{a[NR]=$0}END{print a[int(NR*0.99)]}')\necho \"P99 Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq .Payload.Latency | sort -n | tail -n 1)\necho \"Max Task latency (ms): $tmp\" | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq -s 'map(if .Payload.IsolationGroup == .PartitionConfig.\"original-isolation-group\" then 1 else 0 end) | add / length')\necho \"Task Containment: $tmp\" | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Worker Polls that returned a task: $tmp\" | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\" and .Payload.TaskIsForwarded == true)' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Worker Polls that returned a forwarded task: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returned no tasks\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Worker Polls that returned NO task: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Matcher Falling Back to Non-Local Polling\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Worker Polls that falled back to non-local polling: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Attempting to Forward Poll\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Poll forward attempts: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Forwarded Poll returned task\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Forwarded poll returned task: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Task Written to DB\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Tasks Written to DB: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Attempting to Forward Task\")' \\\n  | jq '{ScheduleID,TaskListName,EventName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Task forward attempts: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName | contains(\"Matched Task\"))' \\\n  | jq -c 'select((.Payload.SyncMatched == true) and (.Payload.TaskIsForwarded == true))' \\\n  | jq '{ScheduleID,TaskListName}' \\\n  | jq -c '.' | wc -l)\necho \"Sync matches - task is forwarded: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName | contains(\"Matched Task\"))' \\\n  | jq -c 'select((.Payload.SyncMatched == true) and (.Payload.TaskIsForwarded == false))' \\\n  | jq '{ScheduleID,TaskListName}' \\\n  | jq -c '.' | wc -l)\necho \"Sync matches - task is not forwarded: $tmp\" | tee -a $testSummaryFile\n\n\necho \"Per tasklist sync matches:\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"SyncMatched so not persisted\")' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Could not SyncMatched Forwarded Task so not persisted\")' \\\n  | jq '{ScheduleID,TaskListName}' \\\n  | jq -c '.' | wc -l)\necho \"Forwarded Task failed to sync match: $tmp\" | tee -a $testSummaryFile\n\ntmp=$(cat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName | contains(\"Matched Task\"))' \\\n  | jq -c 'select(.Payload.SyncMatched != true)' \\\n  | jq '{ScheduleID,TaskListName,Payload}' \\\n  | jq -c '.' | wc -l)\necho \"Async matches: $tmp\" | tee -a $testSummaryFile\n\necho \"Matched tasks per tasklist:\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName | contains(\"Matched Task\"))' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"AddDecisionTask request per tasklist (excluding forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received AddDecisionTask\" and .Payload.RequestForwardedFrom == \"\")' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"AddDecisionTask request per tasklist (forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received AddDecisionTask\" and .Payload.RequestForwardedFrom != \"\")' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\n\necho \"PollForDecisionTask request per tasklist (excluding forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received PollForDecisionTask\" and .Payload.RequestForwardedFrom == \"\")' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\n\necho \"PollForDecisionTask request per tasklist (forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received PollForDecisionTask\" and .Payload.RequestForwardedFrom != \"\")' \\\n  | jq '.TaskListName' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"AddDecisionTask request per isolation group (excluding forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received AddDecisionTask\" and .Payload.RequestForwardedFrom == \"\")' \\\n  | jq '.PartitionConfig.\"original-isolation-group\" // .PartitionConfig.\"isolation-group\"' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"AddDecisionTask request per isolation group (forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received AddDecisionTask\" and .Payload.RequestForwardedFrom != \"\")' \\\n  | jq '.PartitionConfig.\"original-isolation-group\" // .PartitionConfig.\"isolation-group\"' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"PollForDecisionTask request per isolation group  (excluding forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received PollForDecisionTask\" and .Payload.RequestForwardedFrom == \"\")' \\\n  | jq '.Payload.IsolationGroup' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"PollForDecisionTask request per isolation group  (forwarded):\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"Received PollForDecisionTask\" and .Payload.RequestForwardedFrom != \"\")' \\\n  | jq '.Payload.IsolationGroup' \\\n  | jq -c '.' | sort -n | uniq -c | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"Latency per isolation group:\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq -s 'group_by(.PartitionConfig.\"original-isolation-group\")[] | {\"IsolationGroup\": .[0].PartitionConfig.\"original-isolation-group\",\n   \"Avg\": (map(.Payload.Latency | tonumber) | add / length), \"Median\": (map(.Payload.Latency | tonumber) | sort | .[length/2]), \"Max\":(map(.Payload.Latency | tonumber) | max) }'\\\n  | jq -s 'sort_by(.IsolationGroup)[]'\\\n  | jq -r '[.IsolationGroup, .Median, .Avg, .Max] | @tsv' \\\n  | sort -n | column -t  | sed -e 's/^/     /' | tee -a $testSummaryFile\n\n\necho \"Latency per isolation group and task list:\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\" \\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq -s 'group_by(.PartitionConfig.\"original-isolation-group\", .TaskListName)[] | {\"IsolationGroup\": .[0].PartitionConfig.\"original-isolation-group\", \"TaskListName\": .[0].TaskListName,\n   \"Avg\": (map(.Payload.Latency | tonumber) | add / length), \"Median\": (map(.Payload.Latency | tonumber) | sort | .[length/2]), \"Max\":(map(.Payload.Latency | tonumber) | max) }'\\\n  | jq -s 'sort_by(.TaskListName, .IsolationGroup)[]'\\\n  | jq -r '[.TaskListName, .IsolationGroup, .Median, .Avg, .Max] | @tsv' \\\n  | sort -n | column -t  | sed -e 's/^/     /' | tee -a $testSummaryFile\n\necho \"Task Containment per isolation group and task list:\" | tee -a $testSummaryFile\ncat \"$eventLogsFile\"\\\n  | jq -c 'select(.EventName == \"PollForDecisionTask returning task\")' \\\n  | jq -s 'group_by(.PartitionConfig.\"original-isolation-group\", (.Payload.RequestForwardedFrom // .TaskListName))[] | {\"IsolationGroup\": .[0].PartitionConfig.\"original-isolation-group\", \"TaskListName\": (.[0].Payload.RequestForwardedFrom // .[0].TaskListName),\n   \"Containment\": (map(if .Payload.IsolationGroup == .PartitionConfig.\"original-isolation-group\" then 1 else 0 end) | add / length) }'\\\n  | jq -s 'sort_by(.TaskListName, .IsolationGroup)[]'\\\n  | jq -r '[.TaskListName, .IsolationGroup, .Containment] | @tsv' \\\n  | sort -n | column -t  | sed -e 's/^/     /' | tee -a $testSummaryFile\n\n\necho \"End of summary\" | tee -a $testSummaryFile\n\nprintf \"\\nResults are saved in $testSummaryFile\\n\"\nprintf \"For further ad-hoc analysis, please check $eventLogsFile via jq queries\\n\"\nprintf \"Visit http://localhost:3000/ to view Cadence Matching grafana dashboard\\n\"\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_burst.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 0ms\n    localtaskwaittime: 0ms\n    tasks:\n      - numtaskgenerators: 10\n        taskspersecond: 500\n        maxtasktogenerate: 10000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_burst_adaptive.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 0 # this doesn't matter. adaptive scaler will start from 1\n    tasklistreadpartitions: 0  # this doesn't matter. adaptive scaler will start from 1\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 10\n        taskspersecond: 500\n        maxtasktogenerate: 10000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\n    getpartitionconfigfromdb: true\n    enableadaptivescaler: true\n    partitiondownscalefactor: 0.75\n    partitionupscalerps: 200\n    partitionupscalesustainedduration: 5s\n    partitiondownscalesustainedduration: 5s\n    adaptivescalerupdateinterval: 1s\n    qpstrackerinterval: 2s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_default.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_fluctuating.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 1\n    tasklistreadpartitions: 1\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 4\n        maxtasktogenerate: 5000\n        overtime:\n        - taskspersecond: 10\n          tasksburst: 10\n          duration: 30s\n        - taskspersecond: 250\n          tasksburst: 250\n          duration: 15s\n        - taskspersecond: 10\n          tasksburst: 10\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_fluctuating_adaptive.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 0 # this doesn't matter. adaptive scaler will start from 1\n    tasklistreadpartitions: 0 # this doesn't matter. adaptive scaler will start from 1\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 3\n        maxtasktogenerate: 8000\n        overtime:\n        - taskspersecond: 10\n          tasksburst: 10\n          duration: 15s\n        - taskspersecond: 250\n          tasksburst: 250\n          duration: 30s\n        - taskspersecond: 10\n          tasksburst: 10\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\n    getpartitionconfigfromdb: true\n    enableadaptivescaler: true\n    partitiondownscalefactor: 0.7\n    partitionupscalerps: 120\n    partitionupscalesustainedduration: 3s\n    partitiondownscalesustainedduration: 3s\n    adaptivescalerupdateinterval: 1s\n    qpstrackerinterval: 2s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_get_partition_config_from_db.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    getpartitionconfigfromdb: true\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_more_read_partitions.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 8\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_no_forwarding.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 0\n    forwardermaxoutstandingtasks: 0\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_no_wait_time_for_addtask.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 0ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_no_wait_time_for_polltask.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 0ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_no_wait_time_for_polltask_and_addtask.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 0ms\n    localtaskwaittime: 0ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_round_robin_load_balancer.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 8\n  simulationconfig:\n    tasklistwritepartitions: 2\n    tasklistreadpartitions: 2\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 0ms\n    localtaskwaittime: 0ms\n    recorddecisiontaskstartedtime: 13ms\n    tasklistloadbalancerstrategy: round-robin\n    tasks:\n      - numtaskgenerators: 100\n        taskspersecond: 300\n        maxtasktogenerate: 30000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_throughput.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 1\n    tasklistreadpartitions: 1\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_weighted_load_balancer_with_backlog.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasklistloadbalancerstrategy: weighted\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\n    backlogs:\n      - partition: 1\n        backlogcount: 1000\n      - partition: 2\n        backlogcount: 2000\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_with_backlog.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    tasks:\n      - numtaskgenerators: 2\n        taskspersecond: 80\n        maxtasktogenerate: 3000\n    pollers:\n      - taskprocesstime: 1ms\n        numpollers: 8\n        polltimeout: 60s\n    backlogs:\n      - partition: 1\n        backlogcount: 1000\n      - partition: 2\n        backlogcount: 2000\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasks:\n      - numtaskgenerators: 30\n        taskspersecond: 500\n        maxtasktogenerate:  5000\n        isolationgroups: ['a', 'b', 'c', 'd']\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_few_pollers.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 30\n        taskspersecond: 500\n        maxtasktogenerate:  5000\n        isolationgroups: ['a', 'b', 'c', 'd']\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_many_pollers.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 30\n        taskspersecond: 500\n        maxtasktogenerate:  5000\n        isolationgroups: ['a', 'b', 'c', 'd']\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 1ms\n        numpollers: 16\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 1ms\n        numpollers: 16\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 1ms\n        numpollers: 16\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 1ms\n        numpollers: 16\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_single_partition.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 1\n    tasklistreadpartitions: 1\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 30\n        taskspersecond: 500\n        maxtasktogenerate:  5000\n        isolationgroups: ['a', 'b', 'c', 'd']\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 1ms\n        numpollers: 2\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_skew.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 10\n        taskspersecond: 180\n        maxtasktogenerate:  1800\n        isolationgroups: ['a', 'b', 'c']\n      - numtaskgenerators: 20\n        taskspersecond: 320\n        maxtasktogenerate: 3200\n        isolationgroups: [ 'd' ]\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_skew_extreme.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 3\n        taskspersecond: 50\n        maxtasktogenerate:  500\n        isolationgroups: ['a', 'b', 'c']\n      - numtaskgenerators: 27\n        taskspersecond: 450\n        maxtasktogenerate: 4500\n        isolationgroups: [ 'd' ]\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/matching/testdata/matching_simulation_zonal_isolation_skew_forwarding.yaml",
    "content": "enablearchival: false\nclusterno: 1\nmessagingclientconfig:\n  usemock: true\nhistoryconfig:\n  numhistoryshards: 4\n  numhistoryhosts: 1\nmatchingconfig:\n  nummatchinghosts: 4\n  simulationconfig:\n    tasklistwritepartitions: 4\n    tasklistreadpartitions: 4\n    forwardermaxoutstandingpolls: 1\n    forwardermaxoutstandingtasks: 1\n    forwardermaxratepersecond: 10000\n    forwardermaxchildrenpernode: 20\n    localpollwaittime: 10ms\n    localtaskwaittime: 10ms\n    taskisolationduration: 1s\n    tasklistloadbalancerstrategy: isolation\n    tasks:\n      - numtaskgenerators: 10\n        taskspersecond: 180\n        maxtasktogenerate:  1800\n        isolationgroups: ['a', 'b', 'c']\n      - numtaskgenerators: 20\n        taskspersecond: 320\n        maxtasktogenerate: 3200\n        isolationgroups: [ 'd' ]\n    pollers:\n      - isolationgroup: 'a'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'b'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'c'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\n      - isolationgroup: 'd'\n        taskprocesstime: 25ms\n        numpollers: 8\n        polltimeout: 60s\nworkerconfig:\n  enableasyncwfconsumer: false\n"
  },
  {
    "path": "simulation/replication/README.md",
    "content": "# Replication Simulation\n\nThese simulations test replication of workflows across clusters. As replication is handled at the application layer there are many edge case scenarios that lead to conflict resolution when:\n\n- there is a delay in replication\n- a domain is failed over from one cluster to another\n- active-active is enabled and a customer is making requests to both clusters\n\nThere are also sanity checks for the request routing layer. \n\n## Quick Run\n\n```bash\n# Basic test\n./simulation/replication/run.sh --scenario default\n\n# Active-active test\n./simulation/replication/run.sh --scenario activeactive\n\n# With custom dockerfile\n./simulation/replication/run.sh --scenario default --dockerfile-suffix .local\n\n# Rerun without rebuilding\n./simulation/replication/run.sh --scenario default --rerun\n```\n\n### Results\n\nResults are output to the following files:\n- **Test logs**: `test.log` contains the summary of the test run\n- **Summary**: `replication-simulator-output/test-{scenario}-{timestamp}-summary.txt` contains a summary of the test run \n\nYou can also use the [Cadence UI](http://localhost:8088) to debug the workflows that ran during your test. \n\nTo further debug, you can query for logs against the running docker containers.\n\n## Configuration\n\nScenarios are written in `testdata/replication_simulation_{scenario}.yaml`. \nThis naming convention is required by `run.sh` to find test scenarios to run. \n\nTo configure the cadence instances that are running for the test use a dynamic config file at `config/dynamicconfig/replication_simulation_{scenario}.yml`.\nDynamic config can change any feature flag supported by Cadence - these feature flags can be used to hide full features, or to hide test-specific implementations that expose additional data required by your test.\n\n## CI/CD\n\nScenarios run automatically in GitHub Actions via `.github/workflows/replication-simulation.yml`. Add new scenarios to the matrix once they pass locally.\n"
  },
  {
    "path": "simulation/replication/replication_simulation_test.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n/*\nTo run locally:\n\n1. Pick a scenario from the existing config files simulation/replication/testdata/replication_simulation_${scenario}.yaml or add a new one\n\n2. Run the scenario\n`./simulation/replication/run.sh default`\n\nFull test logs can be found at test.log file. Event json logs can be found at replication-simulator-output folder.\nSee the run.sh script for more details about how to parse events.\n*/\npackage replication\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"flag\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"reflect\"\n\t\"sort\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\tsimTypes \"github.com/uber/cadence/simulation/replication/types\"\n)\n\nfunc TestReplicationSimulation(t *testing.T) {\n\tflag.Parse()\n\n\tsimTypes.Logf(t, \"Starting Replication Simulation\")\n\n\tsimTypes.Logf(t, \"Sleeping for 30 seconds to allow services to start/warmup\")\n\ttime.Sleep(30 * time.Second)\n\n\t// load config\n\tsimCfg, err := simTypes.LoadConfig()\n\trequire.NoError(t, err, \"failed to load config\")\n\n\t// initialize replication simulation\n\tsim := simTypes.NewReplicationSimulation()\n\n\t// initialize cadence clients\n\tfor clusterName := range simCfg.Clusters {\n\t\tsimCfg.MustInitClientsFor(t, clusterName)\n\t}\n\n\tsimTypes.Logf(t, \"Registering domains\")\n\tfor domainName, domainCfg := range simCfg.Domains {\n\t\tsimTypes.Logf(t, \"Domain: %s\", domainName)\n\t\tsimCfg.MustRegisterDomain(t, domainName, domainCfg)\n\t}\n\n\t// wait for domain data to be replicated and workers to start.\n\twaitUntilWorkersReady(t)\n\n\tsort.Slice(simCfg.Operations, func(i, j int) bool {\n\t\treturn simCfg.Operations[i].At < simCfg.Operations[j].At\n\t})\n\n\tstartTime := time.Now().UTC()\n\tsimTypes.Logf(t, \"Simulation start time: %v\", startTime)\n\tfor i, op := range simCfg.Operations {\n\t\twaitForOpTime(t, op, startTime)\n\t\tvar err error\n\t\tswitch op.Type {\n\t\tcase simTypes.ReplicationSimulationOperationStartWorkflow:\n\t\t\terr = startWorkflow(t, op, simCfg, sim)\n\t\tcase simTypes.ReplicationSimulationOperationResetWorkflow:\n\t\t\terr = resetWorkflow(t, op, simCfg)\n\t\tcase simTypes.ReplicationSimulationOperationChangeActiveClusters:\n\t\t\terr = changeActiveClusters(t, op, simCfg)\n\t\tcase simTypes.ReplicationSimulationOperationValidate:\n\t\t\terr = validate(t, op, simCfg, sim)\n\t\tcase simTypes.ReplicationSimulationOperationQueryWorkflow:\n\t\t\terr = queryWorkflow(t, op, simCfg, sim)\n\t\tcase simTypes.ReplicationSimulationOperationSignalWithStartWorkflow:\n\t\t\terr = signalWithStartWorkflow(t, op, simCfg, sim)\n\t\tcase simTypes.ReplicationSimulationOperationValidateWorkflowReplication:\n\t\t\terr = validateWorkflowReplication(t, op, simCfg)\n\t\tdefault:\n\t\t\trequire.Failf(t, \"unknown operation type\", \"operation type: %s\", op.Type)\n\t\t}\n\n\t\tif err != nil {\n\t\t\tt.Fatalf(\"Operation %d failed: %v\", i, err)\n\t\t}\n\t}\n\n\t// Print the test summary.\n\t// Don't change the start/end line format as it is used by scripts to parse the summary info\n\texecutionTime := time.Since(startTime)\n\tvar testSummary []string\n\ttestSummary = append(testSummary, \"Simulation Summary:\")\n\ttestSummary = append(testSummary, fmt.Sprintf(\"Simulation Duration: %v\", executionTime))\n\ttestSummary = append(testSummary, \"End of Simulation Summary\")\n\tfmt.Println(strings.Join(testSummary, \"\\n\"))\n}\n\nfunc startWorkflow(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n\tsim *simTypes.ReplicationSimulation,\n) error {\n\tt.Helper()\n\n\tsimTypes.Logf(t, \"Starting workflow: %s on domain %s on cluster: %s\", op.WorkflowID, op.Domain, op.Cluster)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\n\tif op.WorkflowExecutionStartToCloseTimeout == 0 || op.WorkflowExecutionStartToCloseTimeout < op.WorkflowDuration {\n\t\treturn fmt.Errorf(\"workflow execution start to close timeout must be specified and should be greater than workflow duration\")\n\t}\n\n\tinput := mustJSON(t, &simTypes.WorkflowInput{\n\t\tDuration:             op.WorkflowDuration,\n\t\tActivityCount:        op.ActivityCount,\n\t\tChildWorkflowID:      op.ChildWorkflowID,\n\t\tChildWorkflowTimeout: op.ChildWorkflowTimeout,\n\t})\n\tworkflowIDReusePolicy := types.WorkflowIDReusePolicyAllowDuplicate.Ptr()\n\tif op.WorkflowIDReusePolicy != nil {\n\t\tworkflowIDReusePolicy = op.WorkflowIDReusePolicy\n\t}\n\tresp, err := simCfg.MustGetFrontendClient(t, op.Cluster).StartWorkflowExecution(ctx,\n\t\t&types.StartWorkflowExecutionRequest{\n\t\t\tRequestID:                           uuid.New(),\n\t\t\tDomain:                              op.Domain,\n\t\t\tWorkflowID:                          op.WorkflowID,\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: op.WorkflowType},\n\t\t\tTaskList:                            &types.TaskList{Name: simTypes.TasklistName},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32((op.WorkflowExecutionStartToCloseTimeout).Seconds())),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(5),\n\t\t\tInput:                               input,\n\t\t\tWorkflowIDReusePolicy:               workflowIDReusePolicy,\n\t\t\tDelayStartSeconds:                   common.Int32Ptr(op.DelayStartSeconds),\n\t\t\tCronSchedule:                        op.CronSchedule,\n\t\t\tActiveClusterSelectionPolicy:        op.ActiveClusterSelectionPolicy,\n\t\t})\n\n\tif err != nil {\n\t\tif op.Want.Error != \"\" {\n\t\t\tif strings.Contains(err.Error(), op.Want.Error) {\n\t\t\t\tsimTypes.Logf(t, \"Start workflow got expected error: %s on domain: %s on cluster: %s. Error: %s\", op.WorkflowID, op.Domain, op.Cluster, err.Error())\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn fmt.Errorf(\"expected error: %s, but got: %s\", op.Want.Error, err.Error())\n\t\t}\n\t\treturn err\n\t}\n\n\trunID := resp.GetRunID()\n\tsimTypes.Logf(t, \"Started workflow: %s on domain: %s on cluster: %s. RunID: %s\", op.WorkflowID, op.Domain, op.Cluster, runID)\n\n\t// Store RunID if runIDKey is specified\n\tif op.RunIDKey != \"\" {\n\t\tsim.StoreRunID(op.RunIDKey, runID)\n\t\tsimTypes.Logf(t, \"Stored RunID %s with key: %s\", runID, op.RunIDKey)\n\t}\n\n\treturn nil\n}\n\nfunc resetWorkflow(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n) error {\n\tt.Helper()\n\n\tsimTypes.Logf(t, \"Resetting workflow: %s on domain %s on cluster: %s\", op.WorkflowID, op.Domain, op.Cluster)\n\tctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)\n\tdefer cancel()\n\n\tresp, err := simCfg.MustGetFrontendClient(t, op.Cluster).DescribeWorkflowExecution(ctx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: op.Domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: op.WorkflowID,\n\t\t\t},\n\t\t})\n\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to describe workflow %s: %w\", op.WorkflowID, err)\n\t}\n\n\tsimTypes.Logf(t, \"DescribeWorkflowExecution workflowID: %s on domain: %s on cluster: %s. RunID: %s\", op.WorkflowID, op.Domain, op.Cluster, resp.GetWorkflowExecutionInfo().Execution.RunID)\n\n\tresetResp, err := simCfg.MustGetFrontendClient(t, op.Cluster).ResetWorkflowExecution(ctx,\n\t\t&types.ResetWorkflowExecutionRequest{\n\t\t\tDomain: op.Domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: op.WorkflowID,\n\t\t\t\tRunID:      resp.GetWorkflowExecutionInfo().Execution.RunID,\n\t\t\t},\n\t\t\tReason:                \"resetting workflow\",\n\t\t\tDecisionFinishEventID: op.EventID,\n\t\t})\n\n\tif err != nil {\n\t\tif op.Want.Error != \"\" {\n\t\t\tif strings.Contains(err.Error(), op.Want.Error) {\n\t\t\t\tsimTypes.Logf(t, \"Expected error: %s\", op.Want.Error)\n\t\t\t\treturn nil\n\t\t\t}\n\t\t\treturn fmt.Errorf(\"expected error: %s, but got: %s\", op.Want.Error, err.Error())\n\t\t}\n\t\tsimTypes.Logf(t, err.Error())\n\t\treturn err\n\t}\n\n\tif op.Want.Error != \"\" {\n\t\treturn fmt.Errorf(\"expected error: %s, but got nil\", op.Want.Error)\n\t}\n\n\tsimTypes.Logf(t, \"Reset workflowID: %s on domain: %s on cluster: %s. RunID: %s\", op.WorkflowID, op.Domain, op.Cluster, resetResp.GetRunID())\n\n\treturn nil\n}\n\n// changeActiveClusters modifies the active clusters for a domain\n// It can be used to change the active cluster for a domain or a sub-set of the AttributeScopes for the domain\nfunc changeActiveClusters(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n) error {\n\tt.Helper()\n\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\tdescResp, err := simCfg.MustGetFrontendClient(t, simCfg.PrimaryCluster).DescribeDomain(ctx, &types.DescribeDomainRequest{Name: common.StringPtr(op.Domain)})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to describe domain %s: %w\", op.Domain, err)\n\t}\n\n\tupdateDomainRequest := &types.UpdateDomainRequest{\n\t\tName:                     op.Domain,\n\t\tActiveClusters:           &types.ActiveClusters{},\n\t\tFailoverTimeoutInSeconds: op.FailoverTimeout,\n\t}\n\n\tif op.NewActiveCluster != \"\" {\n\t\tfromCluster := descResp.ReplicationConfiguration.ActiveClusterName\n\t\ttoCluster := op.NewActiveCluster\n\t\tsimTypes.Logf(t, \"Changing active clusters for domain %s from %s to %s\", op.Domain, fromCluster, toCluster)\n\t\tupdateDomainRequest.ActiveClusterName = &toCluster\n\t}\n\n\tif !op.NewClusterAttributes.IsEmpty() {\n\t\tupdateDomainRequest.ActiveClusters.AttributeScopes = op.NewClusterAttributes.ToAttributeScopes()\n\t\tsimTypes.Logf(t, \"Changing cluster attributes for domain %s from %+v to %+v\", op.Domain, descResp.ReplicationConfiguration.ActiveClusters, updateDomainRequest.ActiveClusters)\n\t}\n\n\tctx, cancel = context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\t_, err = simCfg.MustGetFrontendClient(t, simCfg.PrimaryCluster).UpdateDomain(ctx, updateDomainRequest)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to update ActiveClusters, err: %w\", err)\n\t}\n\n\tsimTypes.Logf(t, \"Completed change to ActiveClusters\")\n\treturn nil\n}\n\nfunc queryWorkflow(t *testing.T, op *simTypes.Operation, simCfg *simTypes.ReplicationSimulationConfig, sim *simTypes.ReplicationSimulation) error {\n\tt.Helper()\n\n\tsimTypes.Logf(t, \"Querying workflow: %s on domain %s on cluster: %s\", op.WorkflowID, op.Domain, op.Cluster)\n\n\tfrontendCl := simCfg.MustGetFrontendClient(t, op.Cluster)\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\n\tconsistencyLevel := types.QueryConsistencyLevelEventual.Ptr()\n\tif op.ConsistencyLevel == \"strong\" {\n\t\tconsistencyLevel = types.QueryConsistencyLevelStrong.Ptr()\n\t}\n\n\t// Prepare workflow execution - use specific RunID if provided via runIDKey\n\texecutionRequest := &types.WorkflowExecution{\n\t\tWorkflowID: op.WorkflowID,\n\t}\n\tif op.RunIDKey != \"\" {\n\t\tif runID, err := sim.GetRunID(op.RunIDKey); err == nil && runID != \"\" {\n\t\t\texecutionRequest.RunID = runID\n\t\t\tsimTypes.Logf(t, \"Using stored RunID %s for query (key: %s)\", runID, op.RunIDKey)\n\t\t} else {\n\t\t\treturn fmt.Errorf(\"runIDKey %s specified but no RunID found in registry\", op.RunIDKey)\n\t\t}\n\t}\n\n\tqueryResp, err := frontendCl.QueryWorkflow(ctx, &types.QueryWorkflowRequest{\n\t\tDomain:    op.Domain,\n\t\tExecution: executionRequest,\n\t\tQuery: &types.WorkflowQuery{\n\t\t\tQueryType: op.Query,\n\t\t},\n\t\tQueryConsistencyLevel: consistencyLevel,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tgot := mustParseJSON(t, queryResp.GetQueryResult())\n\twant := op.Want.QueryResult\n\tif !reflect.DeepEqual(want, got) {\n\t\treturn fmt.Errorf(\"query result mismatch. want: %v (type: %T), got: %v (type: %T)\", want, want, got, got)\n\t}\n\n\tsimTypes.Logf(t, \"Query workflow: %s on domain: %s on cluster: %s. Query result: %v\", op.WorkflowID, op.Domain, op.Cluster, got)\n\n\treturn nil\n}\n\nfunc signalWithStartWorkflow(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n\tsim *simTypes.ReplicationSimulation,\n) error {\n\tt.Helper()\n\tsimTypes.Logf(t, \"SignalWithStart workflow: %s on domain %s on cluster: %s\", op.WorkflowID, op.Domain, op.Cluster)\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\n\tfrontendCl := simCfg.MustGetFrontendClient(t, op.Cluster)\n\n\tworkflowIDReusePolicy := types.WorkflowIDReusePolicyAllowDuplicate.Ptr()\n\tif op.WorkflowIDReusePolicy != nil {\n\t\tworkflowIDReusePolicy = op.WorkflowIDReusePolicy\n\t}\n\tsignalResp, err := frontendCl.SignalWithStartWorkflowExecution(ctx, &types.SignalWithStartWorkflowExecutionRequest{\n\t\tRequestID:                           uuid.New(),\n\t\tDomain:                              op.Domain,\n\t\tWorkflowID:                          op.WorkflowID,\n\t\tWorkflowIDReusePolicy:               workflowIDReusePolicy,\n\t\tWorkflowType:                        &types.WorkflowType{Name: op.WorkflowType},\n\t\tSignalName:                          op.SignalName,\n\t\tSignalInput:                         mustJSON(t, op.SignalInput),\n\t\tTaskList:                            &types.TaskList{Name: simTypes.TasklistName},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32((op.WorkflowExecutionStartToCloseTimeout).Seconds())),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(5),\n\t\tInput:                               mustJSON(t, &simTypes.WorkflowInput{Duration: op.WorkflowDuration, ActivityCount: op.ActivityCount}),\n\t\tActiveClusterSelectionPolicy:        op.ActiveClusterSelectionPolicy,\n\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\trunID := signalResp.GetRunID()\n\tsimTypes.Logf(t, \"SignalWithStart workflow: %s on domain %s on cluster: %s. RunID: %s\", op.WorkflowID, op.Domain, op.Cluster, runID)\n\n\t// Store RunID if runIDKey is specified\n\tif op.RunIDKey != \"\" {\n\t\tsim.StoreRunID(op.RunIDKey, runID)\n\t\tsimTypes.Logf(t, \"Stored RunID %s with key: %s\", runID, op.RunIDKey)\n\t}\n\n\treturn nil\n}\n\n// validate performs validation based on given operation config.\n// validate function does not fail the test via t.Fail (or require.X).\n// It runs in separate goroutine. It should return an error.\nfunc validate(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n\tsim *simTypes.ReplicationSimulation,\n) error {\n\tt.Helper()\n\n\tsimTypes.Logf(t, \"Validating workflow: %s on cluster: %s\", op.WorkflowID, op.Cluster)\n\n\tconsistencyLevel := types.QueryConsistencyLevelEventual.Ptr()\n\tif op.ConsistencyLevel == \"strong\" {\n\t\tconsistencyLevel = types.QueryConsistencyLevelStrong.Ptr()\n\t}\n\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\n\t// Prepare workflow execution - use specific RunID if provided via runIDKey\n\texecutionRequest := &types.WorkflowExecution{\n\t\tWorkflowID: op.WorkflowID,\n\t}\n\tif op.RunIDKey != \"\" {\n\t\tif runID, err := sim.GetRunID(op.RunIDKey); err == nil && runID != \"\" {\n\t\t\texecutionRequest.RunID = runID\n\t\t\tsimTypes.Logf(t, \"Using stored RunID %s for validation (key: %s)\", runID, op.RunIDKey)\n\t\t} else {\n\t\t\treturn fmt.Errorf(\"runIDKey %s specified but no RunID found in registry\", op.RunIDKey)\n\t\t}\n\t}\n\n\tresp, err := simCfg.MustGetFrontendClient(t, op.Cluster).DescribeWorkflowExecution(ctx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain:                op.Domain,\n\t\t\tExecution:             executionRequest,\n\t\t\tQueryConsistencyLevel: consistencyLevel,\n\t\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tworkflowStatus := resp.GetWorkflowExecutionInfo().GetCloseStatus()\n\tworkflowCloseTime := resp.GetWorkflowExecutionInfo().GetCloseTime()\n\tswitch op.Want.Status {\n\tcase \"completed\":\n\t\t// Validate workflow completed\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusCompleted || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not completed. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tcase \"failed\":\n\t\t// Validate workflow failed\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusFailed || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not failed. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tcase \"canceled\":\n\t\t// Validate workflow canceled\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusCanceled || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not canceled. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tcase \"terminated\":\n\t\t// Validate workflow terminated\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusTerminated || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not terminated. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tcase \"continued-as-new\":\n\t\t// Validate workflow continued as new\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusContinuedAsNew || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not continued as new. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tcase \"timed-out\":\n\t\t// Validate workflow timed out\n\t\tif workflowStatus != types.WorkflowExecutionCloseStatusTimedOut || workflowCloseTime == 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not timed out. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\tdefault:\n\t\t// Validate workflow is running\n\t\tif workflowCloseTime != 0 {\n\t\t\treturn fmt.Errorf(\"workflow %s not running. status: %s, close time: %v\", op.WorkflowID, workflowStatus, time.Unix(0, workflowCloseTime))\n\t\t}\n\t}\n\n\tsimTypes.Logf(t, \"Validated workflow: %s on cluster: %s. Status: %s, CloseTime: %v\", op.WorkflowID, op.Cluster, resp.GetWorkflowExecutionInfo().GetCloseStatus(), time.Unix(0, resp.GetWorkflowExecutionInfo().GetCloseTime()))\n\n\t// Get history to validate the worker identity that started and completed the workflow\n\t// Some workflows start in cluster0 and complete in cluster1. This is to validate that\n\tvar runID string\n\tif op.RunIDKey != \"\" {\n\t\trunID, err = sim.GetRunID(op.RunIDKey)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\thistory, err := getAllHistory(t, simCfg, op.Cluster, op.Domain, op.WorkflowID, runID)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif len(history) == 0 {\n\t\treturn fmt.Errorf(\"no history events found for workflow %s\", op.WorkflowID)\n\t}\n\n\tstartedWorker, err := firstDecisionTaskWorker(history)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif op.Want.StartedByWorkersInCluster != \"\" && startedWorker != simTypes.WorkerIdentityFor(op.Want.StartedByWorkersInCluster, op.Domain) {\n\t\treturn fmt.Errorf(\"workflow %s started by worker %s, expected %s\", op.WorkflowID, startedWorker, simTypes.WorkerIdentityFor(op.Want.StartedByWorkersInCluster, op.Domain))\n\t}\n\n\tcompletedWorker, err := lastDecisionTaskWorker(history)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif op.Want.CompletedByWorkersInCluster != \"\" && completedWorker != simTypes.WorkerIdentityFor(op.Want.CompletedByWorkersInCluster, op.Domain) {\n\t\treturn fmt.Errorf(\"workflow %s completed by worker %s, expected %s\", op.WorkflowID, completedWorker, simTypes.WorkerIdentityFor(op.Want.CompletedByWorkersInCluster, op.Domain))\n\t}\n\n\treturn nil\n}\n\nfunc validateWorkflowReplication(\n\tt *testing.T,\n\top *simTypes.Operation,\n\tsimCfg *simTypes.ReplicationSimulationConfig,\n) error {\n\tt.Helper()\n\n\tsimTypes.Logf(t, \"Describing workflow: %s on domain %s on cluster: %s\", op.WorkflowID, op.Domain, op.Cluster)\n\n\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\tresp, err := simCfg.MustGetFrontendClient(t, op.SourceCluster).DescribeWorkflowExecution(ctx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: op.Domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: op.WorkflowID,\n\t\t\t},\n\t\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsourceClusterWorkflowExecution := resp.GetWorkflowExecutionInfo().GetExecution()\n\n\tsimTypes.Logf(t, \"Described workflow: %s on domain: %s on cluster: %s. Status: %s, CloseTime: %v\", op.WorkflowID, op.Domain, op.Cluster, resp.GetWorkflowExecutionInfo().GetCloseStatus(), time.Unix(0, resp.GetWorkflowExecutionInfo().GetCloseTime()))\n\n\tctx, cancel = context.WithTimeout(context.Background(), 2*time.Second)\n\tdefer cancel()\n\tresp, err = simCfg.MustGetFrontendClient(t, op.TargetCluster).DescribeWorkflowExecution(ctx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: op.Domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: op.WorkflowID,\n\t\t\t},\n\t\t})\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttargetClusterWorkflowExecution := resp.GetWorkflowExecutionInfo().GetExecution()\n\n\tif !reflect.DeepEqual(sourceClusterWorkflowExecution, targetClusterWorkflowExecution) {\n\t\treturn fmt.Errorf(\"workflow execution info mismatch between source cluster %s and target cluster %s for workflow %s. \\nSource: %+v\\nTarget: %+v\", op.SourceCluster, op.TargetCluster, op.WorkflowID, *sourceClusterWorkflowExecution, *targetClusterWorkflowExecution)\n\t}\n\n\treturn nil\n}\n\nfunc firstDecisionTaskWorker(history []types.HistoryEvent) (string, error) {\n\tfor _, event := range history {\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\treturn event.GetDecisionTaskCompletedEventAttributes().Identity, nil\n\t\t}\n\t}\n\treturn \"\", fmt.Errorf(\"failed to find first decision task worker because there's no DecisionTaskCompleted event found in history\")\n}\n\nfunc lastDecisionTaskWorker(history []types.HistoryEvent) (string, error) {\n\tfor i := len(history) - 1; i >= 0; i-- {\n\t\tevent := history[i]\n\t\tif event.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\treturn event.GetDecisionTaskCompletedEventAttributes().Identity, nil\n\t\t}\n\t}\n\treturn \"\", fmt.Errorf(\"failed to find lastDecisionTaskWorker because there's no DecisionTaskCompleted event found in history\")\n}\n\nfunc waitForOpTime(t *testing.T, op *simTypes.Operation, startTime time.Time) {\n\tt.Helper()\n\td := startTime.Add(op.At).Sub(time.Now().UTC())\n\tif d > 0 {\n\t\tsimTypes.Logf(t, \"Waiting for next operation time (t + %ds). Will sleep for %ds\", int(op.At.Seconds()), int(d.Seconds()))\n\t\t<-time.After(d)\n\t}\n\n\tsimTypes.Logf(t, \"Operation time (t + %ds) reached: %v\", int(op.At.Seconds()), startTime.Add(op.At))\n}\n\nfunc getAllHistory(t *testing.T, simCfg *simTypes.ReplicationSimulationConfig, clusterName, domainName, wfID, runID string) ([]types.HistoryEvent, error) {\n\tfrontendCl := simCfg.MustGetFrontendClient(t, clusterName)\n\tvar nextPageToken []byte\n\tvar history []types.HistoryEvent\n\n\texecutionRequest := &types.WorkflowExecution{\n\t\tWorkflowID: wfID,\n\t}\n\tif runID != \"\" {\n\t\texecutionRequest.RunID = runID\n\t}\n\n\tfor {\n\t\tctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)\n\t\tresponse, err := frontendCl.GetWorkflowExecutionHistory(ctx, &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain:                 domainName,\n\t\t\tExecution:              executionRequest,\n\t\t\tMaximumPageSize:        1000,\n\t\t\tNextPageToken:          nextPageToken,\n\t\t\tWaitForNewEvent:        false,\n\t\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t\tSkipArchival:           true,\n\t\t})\n\t\tcancel()\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to get history: %w\", err)\n\t\t}\n\n\t\tfor _, event := range response.GetHistory().GetEvents() {\n\t\t\tif event != nil {\n\t\t\t\thistory = append(history, *event)\n\t\t\t}\n\t\t}\n\n\t\tif response.NextPageToken == nil {\n\t\t\treturn history, nil\n\t\t}\n\n\t\tnextPageToken = response.NextPageToken\n\t\ttime.Sleep(10 * time.Millisecond) // sleep to avoid throttling\n\t}\n}\n\nfunc mustParseJSON(t *testing.T, v []byte) any {\n\tvar result interface{}\n\terr := json.Unmarshal(v, &result)\n\trequire.NoError(t, err, \"failed to unmarshal from json\")\n\treturn result\n}\n\nfunc mustJSON(t *testing.T, v interface{}) []byte {\n\tdata, err := json.Marshal(v)\n\trequire.NoError(t, err, \"failed to marshal to json\")\n\treturn data\n}\n\nfunc waitUntilWorkersReady(t *testing.T) {\n\t// workers expose :6060/health endpoint. Poll on them to check if they are healthy\n\tsimTypes.Logf(t, \"Waiting for workers to start and report healthy\")\n\tworkerEndpoints := []string{\n\t\t\"http://cadence-worker0:6060/health\",\n\t\t\"http://cadence-worker1:6060/health\",\n\t}\n\n\tfor {\n\t\tallHealthy := true\n\t\tfor _, endpoint := range workerEndpoints {\n\t\t\tresp, err := http.Get(endpoint)\n\t\t\tif err != nil || resp.StatusCode != http.StatusOK {\n\t\t\t\tallHealthy = false\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\n\t\tif allHealthy {\n\t\t\tbreak\n\t\t}\n\n\t\tsimTypes.Logf(t, \"Workers are not reporting healthy yet. Sleep for 2s and try again\")\n\t\ttime.Sleep(2 * time.Second)\n\t}\n\n\tsimTypes.Logf(t, \"All workers are healthy\")\n}\n"
  },
  {
    "path": "simulation/replication/run.sh",
    "content": "#!/bin/bash\n\n# Cadence Replication Simulation Test Script\n#\n# Usage:\n#   ./simulation/replication/run.sh [OPTIONS]\n#\n# Examples:\n#   # Run default scenario\n#   ./simulation/replication/run.sh --scenario default\n#\n#   # Run specific scenario with custom dockerfile\n#   ./simulation/replication/run.sh --scenario activeactive --dockerfile-suffix .local\n#\n#   # Rerun without rebuilding images\n#   ./simulation/replication/run.sh --scenario default --rerun\n#\n#   # Run with custom timestamp\n#   ./simulation/replication/run.sh --scenario activeactive --timestamp 2024-01-15-10-30-00\n#\n# See simulation/replication/README.md for scenario details and output information.\n\nset -eo pipefail\n\nshow_help() {\n    cat << EOF\nCadence Replication Simulation Test Script\n\nUSAGE:\n    $0 [OPTIONS]\n\nOPTIONS:\n    -s, --scenario SCENARIO          Test scenario to run (required)\n                                   Corresponds to testdata/replication_simulation_SCENARIO.yaml\n\n    -r, --rerun                     Skip rebuilding images and reuse existing containers\n                                   Useful for faster reruns during development\n\n    -t, --timestamp TIMESTAMP       Custom timestamp for test naming (default: current time)\n                                   Format: YYYY-MM-DD-HH-MM-SS\n\n    -d, --dockerfile-suffix SUFFIX  Dockerfile suffix for custom builds (default: empty)\n                                   Example: .local for Dockerfile.local\n\n    -h, --help                      Show this help message\n\nEXAMPLES:\n    # Run default scenario\n    $0 --scenario default\n\n    # Run specific scenario with custom dockerfile\n    $0 --scenario activeactive --dockerfile-suffix .local\n\n    # Rerun without rebuilding images\n    $0 --scenario default --rerun\n\n    # Run with all options\n    $0 --scenario activeactive --rerun --timestamp 2024-01-15-10-30-00 --dockerfile-suffix .local\n\nEOF\n}\n\n# Default values\ntestCase=\"\"\nrerun=\"\"\ntimestamp=\"\"\ndockerFileSuffix=\"\"\n\n# Parse command line arguments\nwhile [[ $# -gt 0 ]]; do\n    case $1 in\n        -s|--scenario)\n            testCase=\"$2\"\n            shift 2\n            ;;\n        -r|--rerun)\n            rerun=\"rerun\"\n            shift\n            ;;\n        -t|--timestamp)\n            timestamp=\"$2\"\n            shift 2\n            ;;\n        -d|--dockerfile-suffix)\n            dockerFileSuffix=\"$2\"\n            shift 2\n            ;;\n        -h|--help)\n            show_help\n            exit 0\n            ;;\n        --)\n            shift\n            break\n            ;;\n        -*)\n            echo \"Unknown option: $1\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n        *)\n            echo \"Unexpected positional argument: $1\" >&2\n            echo \"Use --scenario to specify the test scenario.\" >&2\n            echo \"Use --help for usage information.\" >&2\n            exit 1\n            ;;\n    esac\ndone\n\n# Require scenario parameter\nif [[ -z \"$testCase\" ]]; then\n    echo \"Error: --scenario parameter is required\" >&2\n    echo \"\" >&2\n    show_help\n    exit 1\nfi\n\n# Set default timestamp if not provided\nif [[ -z \"$timestamp\" ]]; then\n    timestamp=\"$(date '+%Y-%m-%d-%H-%M-%S')\"\nfi\n\ntestCfg=\"testdata/replication_simulation_$testCase.yaml\"\ntestName=\"test-$testCase-$timestamp\"\nresultFolder=\"replication-simulator-output\"\nmkdir -p \"$resultFolder\"\ntestSummaryFile=\"$resultFolder/$testName-summary.txt\"\n\n# Prune everything and rebuild images unless rerun is specified\nif [ \"$rerun\" != \"rerun\" ]; then\n  echo \"Removing some of the previous containers (if exists) to start fresh\"\n  SCENARIO=$testCase DOCKERFILE_SUFFIX=$dockerFileSuffix docker compose -f docker/github_actions/docker-compose-local-replication-simulation.yml \\\n    down cassandra cadence-cluster0 cadence-cluster1 cadence-worker0 cadence-worker1 replication-simulator\n\n  echo \"Each simulation run creates multiple new giant container images. Running docker system prune to avoid disk space issues\"\n  docker system prune -f\n\n  echo \"Building test images\"\n  SCENARIO=$testCase DOCKERFILE_SUFFIX=$dockerFileSuffix docker compose -f docker/github_actions/docker-compose-local-replication-simulation.yml \\\n    build cadence-cluster0 cadence-cluster1 cadence-worker0 cadence-worker1 replication-simulator\nfi\n\nfunction check_test_failure()\n{\n  faillog=$(grep 'FAIL: TestReplicationSimulation' -B 10 test.log 2>/dev/null || true)\n  timeoutlog=$(grep 'test timed out' test.log 2>/dev/null || true)\n  if [ -z \"$faillog\" ] && [ -z \"$timeoutlog\" ]; then\n    echo \"Passed\"\n  else\n    echo 'Test failed!!!'\n    echo \"Fail log: $faillog\"\n    echo \"Timeout log: $timeoutlog\"\n    echo \"Check test.log file for more details\"\n    exit 1\n  fi\n}\n\ntrap check_test_failure EXIT\n\necho \"Running the scenario '$testCase' with dockerfile suffix: '$dockerFileSuffix'\"\necho \"Test name: $testName\"\n\nSCENARIO=$testCase DOCKERFILE_SUFFIX=$dockerFileSuffix docker compose \\\n  -f docker/github_actions/docker-compose-local-replication-simulation.yml \\\n  run \\\n  -e REPLICATION_SIMULATION_CONFIG=$testCfg \\\n  --rm --remove-orphans --service-ports --use-aliases \\\n  replication-simulator\n\n\necho \"---- Simulation Summary ----\"\ncat test.log \\\n  | sed -n '/Simulation Summary/,/End of Simulation Summary/p' \\\n  | grep -v \"Simulation Summary\" \\\n  | tee -a $testSummaryFile\n\necho \"End of summary\" | tee -a $testSummaryFile\n\nprintf \"\\nResults are saved in $testSummaryFile\\n\"\nprintf \"For further ad-hoc analysis, please check $eventLogsFile via jq queries\\n\"\nprintf \"Visit http://localhost:3000/ to view Cadence replication grafana dashboard\\n\"\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive.yml\n# Tests that active-active allows workflows to be created in multiple clusters\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  # start workflow in cluster1\n  - op: start_workflow\n    at: 0s\n    workflowID: wf2\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n  # validate that wf1 is started in cluster0 and completed in cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  # validate that wf2 is started and completed in cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_child.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_child\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_child.yml\n# Tests that active-active allows workflows to be created in multiple clusters\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: child-activity-loop-workflow\n    cluster: cluster0\n    childWorkflowID: wf1-child\n    childWorkflowTimeout: 65s\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  # start workflow in cluster1\n  - op: start_workflow\n    at: 0s\n    workflowID: wf2\n    workflowType: child-activity-loop-workflow\n    cluster: cluster1\n    childWorkflowID: wf2-child\n    childWorkflowTimeout: 65s\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n  # validate that wf1 is started in cluster0 and completed in cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf1-child\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n  - op: validate\n    at: 70s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  # validate that wf2 is started and completed in cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf2-child\n    cluster: cluster1\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_cron.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_cron\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_cron.yml\n# Tests active-active behaviour with cron workflows\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start a cron workflow that runs every minute in cluster0\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 55s\n    workflowDuration: 35s\n    cronSchedule: \"*/1 * * * *\"\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  # start a cron workflow that runs every minute in cluster1\n  - op: start_workflow\n    at: 0s\n    workflowID: wf2\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 55s\n    workflowDuration: 35s\n    cronSchedule: \"*/1 * * * *\"\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n\n  # Failover from cluster0 to cluster1\n  - op: change_active_clusters\n    at: 120s\n    domain: test-domain-aa\n    newClusterAttributes:\n      region:\n        region0: cluster1 # this is changed from cluster0 to cluster1\n        region1: cluster1\n\n\n# TODO(active-active): validating cron is not as trivial as checking completion status.\n# It keeps creating new runs every minute.\n# Add a custom validation for cron workflows:\n# - Let it run for 3 minutes\n# - List the workflows by id\n# - Validate first 2 continued as new and last one is running (waiting for initial decision task)\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_invalid_cluster_attribute.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_invalid_cluster_attribute\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_invalid_cluster_attribute.yml\n# Tests that active-active allows workflows to be created in multiple clusters with invalid cluster attribute\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: invalid\n        name: invalid\n    want:\n      error: \"Cannot start workflow with a cluster attribute that is not found in the domain's metadata.\"\n\n  # start workflow in cluster1\n  - op: start_workflow\n    at: 0s\n    workflowID: wf2\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: invalid\n        name: invalid\n    want:\n      error: \"Cannot start workflow with a cluster attribute that is not found in the domain's metadata.\"\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_regional_failover.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_regional_failover\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_regional_failover.yml\n# Tests active-active behaviour during a failover\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # Start wf1 on cluster0 before failover. It should start in cluster0 and complete in cluster1.\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 60s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  # Failover from cluster0 to cluster1\n  - op: change_active_clusters\n    at: 10s\n    domain: test-domain-aa\n    newClusterAttributes:\n      region:\n        region0: cluster1 # this is changed from cluster0 to cluster1\n        region1: cluster1\n\n  # Start wf2 on cluster0 right before failover. It will be started by cluster0 workers and completed by cluster1 workers.\n  - op: start_workflow\n    at: 9s\n    workflowID: wf2\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 70s\n    workflowID: wf2\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  # Start a few more workflows on cluster0 right at/after failover time with 10s delay start.\n  # They will be processed by cluster1 workers.\n  - op: start_workflow\n    at: 10s\n    workflowID: wf3\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    delayStartSeconds: 10\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 70s\n    workflowID: wf3\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n  - op: start_workflow\n    at: 11s\n    workflowID: wf4\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    delayStartSeconds: 10\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 71s\n    workflowID: wf4\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n  - op: start_workflow\n    at: 12s\n    workflowID: wf5\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    delayStartSeconds: 10\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 72s\n    workflowID: wf5\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n  - op: start_workflow\n    at: 13s\n    workflowID: wf6\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    delayStartSeconds: 10\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 73s\n    workflowID: wf6\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n\n  # Start a workflow on cluster0 after domain cache is updated (failover time + 10s refresh interval). Request should be forwarded to cluster1.\n  - op: start_workflow\n    at: 25s\n    workflowID: wf7\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 45s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n  - op: validate\n    at: 75s\n    workflowID: wf7\n    cluster: cluster0\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_regional_failover_start_same_wfid.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_regional_failover_start_same_wfid\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_regional_failover_start_same_wfid.yml\n# When a domain is configured as active-active\n# And a workflow is running in one cluster\n# And a failover occurs\n# It should not be possible to start another workflow with the same workflowID\n# And the request should be rejected with error \"Workflow execution is already running\"\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # Start wf1 on cluster1 before failover.\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 70s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n\n  # Failover from cluster1 to cluster0\n  - op: change_active_clusters\n    at: 10s\n    domain: test-domain-aa\n    newClusterAttributes:\n      region:\n        region1: cluster0\n\n  # Attempt to start wf1 on cluster1 again. It will be forwarded to cluster0 and cluster0 should reject with error \"Workflow execution is already running\".\n  - op: start_workflow\n    at: 30s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 70s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n    want:\n      error: \"Workflow execution is already running\"\n\n  # Validate that wf1 is completed in cluster0.\n  - op: validate\n    at: 80s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_regional_failover_start_same_wfid_2.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_regional_failover_start_same_wfid_2\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_regional_failover_start_same_wfid_2.yml\n# This test is a variant of the activeactive_regional_failover_start_same_wfid test\n# It tests that when a workflow is already running in one cluster the same workflow cannot be started in another cluster\n# even after failover\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # Start wf1 on cluster1 before failover.\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 70s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n\n  # Failover from cluster1 to cluster0\n  - op: change_active_clusters\n    at: 10s\n    domain: test-domain-aa\n    newClusterAttributes:\n      region:\n        region1: cluster0\n\n  # Attempt to start wf1 on cluster0. It should reject with error \"Workflow execution is already running\".\n  - op: start_workflow\n    at: 30s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 70s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n    want:\n      error: \"Workflow execution is already running\"\n\n  # Validate that wf1 is completed in cluster0.\n  - op: validate\n    at: 80s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain-aa\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_same_wfid.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_same_wfid\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_same_wfid.yml\n# Tests that the same wfid cannot be created in multiple clusters in active-active mode\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: cluster0\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  # start workflow in cluster1\n  - op: start_workflow\n    at: 2s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: cluster1\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n  # validate that wf1 is started in cluster0 and terminated\n  - op: validate\n    at: 70s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    runIDKey: cluster0\n    want:\n      status: terminated\n\n  # validate that wf2 is started and completed in cluster1\n  - op: validate\n    at: 70s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain-aa\n    runIDKey: cluster1\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_same_wfid_signalwithstart.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_same_wfid_signalwithstart\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_same_wfid_signalwithstart.yml\n# When a domain is configured as active-active\n# And the same workflow ID is used to SignalWithStart in multiple clusters\n# Then the 'earlier' workflow should be terminated\n# And the 'later' workflow should complete\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa-conflict:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  - op: signal_with_start_workflow\n    at: 0s\n    workflowID: conflict-wf\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa-conflict\n    signalName: custom-signal\n    signalInput: \"cluster0-signal-data\"\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: cluster0-run\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  - op: signal_with_start_workflow\n    at: 0s\n    workflowID: conflict-wf\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa-conflict\n    signalName: custom-signal\n    signalInput: \"cluster1-signal-data\"\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: cluster1-run\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n  # Query the cluster0 run to validate it was started with the correct signal\n  # Note that if this query is delayed it will eventually see the signal from the 'later' workflow, as conflict resolution will have occurred\n  - op: query_workflow\n    at: 0s\n    workflowID: conflict-wf\n    cluster: cluster0\n    domain: test-domain-aa-conflict\n    query: latest-signal-content\n    runIDKey: cluster0-run\n    want:\n      queryResult: [\"cluster0-signal-data\"]\n\n  # Query the cluster1 run to validate it was started with the correct signal\n  - op: query_workflow\n    at: 0s\n    workflowID: conflict-wf\n    cluster: cluster1\n    domain: test-domain-aa-conflict\n    query: latest-signal-content\n    runIDKey: cluster1-run\n    want:\n      queryResult: [\"cluster1-signal-data\"]\n\n  - op: validate\n    at: 30s\n    workflowID: conflict-wf\n    runIDKey: cluster0-run\n    cluster: cluster0\n    domain: test-domain-aa-conflict\n    want:\n      status: terminated\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  - op: validate\n    at: 30s\n    workflowID: conflict-wf\n    runIDKey: cluster1-run\n    cluster: cluster0\n    domain: test-domain-aa-conflict\n    want:\n      status: running\n      startedByWorkersInCluster: cluster1\n\n  - op: validate\n    at: 70s\n    workflowID: conflict-wf\n    runIDKey: cluster1-run\n    cluster: cluster0\n    domain: test-domain-aa-conflict\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_same_wfid_signalwithstart_delayed.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_same_wfid_signalwithstart_delayed\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_same_wfid_signalwithstart_delayed.yml\n# When a domain is configured as active-active\n# And the same workflow ID is used to SignalWithStart in multiple clusters with a 20s delay\n# Then there should be no duplicate run - a single workflow should contain both signals\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa-delayed:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  - op: signal_with_start_workflow\n    at: 0s\n    workflowID: delayed-wf\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa-delayed\n    signalName: custom-signal\n    signalInput: \"cluster0-signal-data\"\n    workflowExecutionStartToCloseTimeout: 90s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n\n  - op: signal_with_start_workflow\n    at: 20s\n    workflowID: delayed-wf\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa-delayed\n    signalName: custom-signal\n    signalInput: \"cluster1-signal-data\"\n    workflowExecutionStartToCloseTimeout: 90s\n    workflowDuration: 60s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n\n  # Query after the second signal to validate both signals are in the same workflow\n  - op: query_workflow\n    at: 25s\n    workflowID: delayed-wf\n    cluster: cluster0\n    domain: test-domain-aa-delayed\n    query: latest-signal-content\n    want:\n      queryResult: [\"cluster0-signal-data\", \"cluster1-signal-data\"]\n\n  - op: validate\n    at: 45s\n    workflowID: delayed-wf\n    cluster: cluster0\n    domain: test-domain-aa-delayed\n    want:\n      status: running\n      startedByWorkersInCluster: cluster0\n\n  # Validate workflow completes successfully with both signals\n  - op: validate\n    at: 70s\n    workflowID: delayed-wf\n    cluster: cluster1\n    domain: test-domain-aa-delayed\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_signalwithstart_terminateifrunning.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_signalwithstart_terminateifrunning\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_signalwithstart_terminateifrunning.yml\n# Tests that active-active allows workflows to be created in multiple clusters with terminate if running policy.\n# If the existing workflow is active in the same cluster with the new workflow, the existing workflow should be terminated.\n# If the existing workflow is active in a different cluster with the same active cluster selection policy, the start workflow should fail.\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0 with no active cluster selection policy\n  - op: signal_with_start_workflow\n    at: 0s\n    workflowID: wf1\n    signalName: custom-signal\n    signalInput: \"cluster0-signal-data\"\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: wf1-0\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n\n  # start workflow in cluster0 with active cluster selection policy and terminate if running policy\n  - op: signal_with_start_workflow\n    at: 5s\n    workflowID: wf1\n    signalName: custom-signal\n    signalInput: \"cluster1-signal-data\"\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n    runIDKey: wf1-1\n\n  # start workflow in cluster1 with different active cluster selection policy and terminate if running policy, should fail\n  - op: signal_with_start_workflow\n    at: 10s\n    workflowID: wf1\n    signalName: custom-signal\n    signalInput: \"cluster2-signal-data\"\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n    want:\n      error: \"Cannot terminate the existing workflow and start a new workflow because it is active in a different cluster with a different active cluster selection policy.\"\n\n  # validate that wf1-0 is started in cluster0 and terminated\n  - op: validate\n    at: 20s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    runIDKey: wf1-0\n    want:\n      status: terminated\n      startedByWorkersInCluster: cluster0\n\n  # validate that wf1-1 is started and completed in cluster0\n  - op: validate\n    at: 75s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain-aa\n    runIDKey: wf1-1\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activeactive_start_terminateifrunning.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activeactive_start_terminateifrunning\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activeactive_start_terminateifrunning.yml\n# Tests that active-active allows workflows to be created in multiple clusters with terminate if running policy.\n# If the existing workflow is active in the same cluster with the new workflow, the existing workflow should be terminated.\n# If the existing workflow is active in a different cluster with the same active cluster selection policy, the start workflow should fail.\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain-aa:\n    activeClusterName: cluster0\n    clusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\noperations:\n  # start workflow in cluster0 with no active cluster selection policy\n  - op: start_workflow\n    at: 0s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    runIDKey: wf1-0\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n\n  # start workflow in cluster0 with active cluster selection policy and terminate if running policy\n  - op: start_workflow\n    at: 5s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster0\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region0\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n    runIDKey: wf1-1\n\n  # start workflow in cluster1 with different active cluster selection policy and terminate if running policy, should fail\n  - op: start_workflow\n    at: 10s\n    workflowID: wf1\n    workflowType: timer-activity-loop-workflow\n    cluster: cluster1\n    domain: test-domain-aa\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n    workflowIDReusePolicy: TERMINATEIFRUNNING\n    want:\n      error: \"Cannot terminate the existing workflow and start a new workflow because it is active in a different cluster with a different active cluster selection policy.\"\n\n  # validate that wf1-0 is started in cluster0 and terminated\n  - op: validate\n    at: 20s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain-aa\n    runIDKey: wf1-0\n    want:\n      status: terminated\n      startedByWorkersInCluster: cluster0\n\n  # validate that wf1-1 is started and completed in cluster0\n  - op: validate\n    at: 75s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain-aa\n    runIDKey: wf1-1\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_activepassive_to_activeactive.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario activepassive_to_activeactive\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_activepassive_to_activeactive.yml\n# Validates behaviour of workflows in a domain converted from active-passive to active-active\n# When a domain is migrated from active-passive to active-active\n# It should run new workflows in region sticky mode\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\ndomains:\n  test-domain:\n    activeClusterName: cluster0\n\n\noperations:\n  # start wf1 for active-passive domain and validate it runs in cluster0 smoothly\n  # while the domain is migrating to active-active\n  - op: start_workflow\n    at: 0s\n    workflowType: timer-activity-loop-workflow\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 90s\n    workflowDuration: 60s\n  - op: validate\n    at: 100s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  # migrate domain to active-active\n  - op: change_active_clusters\n    at: 40s\n    domain: test-domain\n    newClusterAttributes:\n      region:\n        region0: cluster0\n        region1: cluster1\n\n  # wait a bit so domain data is replicated to cluster1,\n  # and start wf2 for active-active domain and validate it runs in cluster1\n  - op: start_workflow\n    at: 75s\n    workflowType: timer-activity-loop-workflow\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 60s\n    workflowDuration: 30s\n  - op: validate\n    at: 110s\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  # and start wf3 for active-active domain and validate it runs in cluster1\n  - op: start_workflow\n    at: 75s\n    workflowType: timer-activity-loop-workflow\n    workflowID: wf3\n    cluster: cluster1\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 60s\n    workflowDuration: 30s\n    activeClusterSelectionPolicy:\n      clusterAttribute:\n        scope: region\n        name: region1\n  - op: validate\n    at: 110s\n    workflowID: wf3\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster1\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_budget_manager.yaml",
    "content": "# This file is a replication simulation scenario spec for testing the budget manager.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulations can be run via ./simulation/replication/run.sh --scenario budget_manager\n#\n# Test replication with budget manager enabled\n# When workflows are running in a cluster with budget manager\n# And a failover occurs\n# Then tasks should be cached within budget limits\n# And workflows should continue to run and complete in the new cluster\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\n\ndomains:\n  test-domain:\n    activeClusterName: cluster0\n\noperations:\n  - op: start_workflow\n    at: 0s\n    workflowType: timer-activity-loop-workflow\n    workflowID: timer-activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n\n  - op: change_active_clusters # failover from cluster0 to cluster1\n    at: 20s\n    domain: test-domain\n    newActiveCluster: cluster1\n    # failoverTimeoutSec: 5 # unset means force failover. setting it means graceful failover request\n\n  - op: validate\n    at: 41s # with the current workflow implementation, it's possible that the workflow may finish right after the 40s mark\n    workflowID: timer-activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  # Validate replication occurred by checking the workflow exists in both clusters\n  - op: validate_workflow_replication\n    at: 42s\n    workflowID: timer-activity-loop-workflow1\n    sourceCluster: cluster1\n    targetCluster: cluster0\n    domain: test-domain\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_clusterredirection.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario clusterredirection\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_clusterredirection.yml\n# Ensures redirection of RPCs behaves as expected\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\n\ndomains:\n  test-domain:\n    activeClusterName: cluster0\n\noperations:\n  # Start wf1 in cluster1 will be handled by cluster0 (active cluster)\n  - op: start_workflow\n    at: 0s\n    workflowType: query-workflow\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 30s\n    workflowDuration: 20s\n\n  # Query the wf1 in cluster0 and expect cluster0 worker handles query\n  - op: query_workflow\n    at: 10s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain\n    query: cluster-name\n    consistencyLevel: eventual\n    want:\n      queryResult: cluster0\n\n  # Query the wf1 in cluster1 with eventual consistency and expect cluster1 worker handles query\n  - op: query_workflow\n    at: 10s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain\n    query: cluster-name\n    consistencyLevel: eventual\n    want:\n      queryResult: cluster1\n\n  # Query the wf1 in cluster1 with strong consistency and expect cluster0 (active cluster) worker handles query\n  - op: query_workflow\n    at: 10s\n    workflowID: wf1\n    cluster: cluster1\n    domain: test-domain\n    query: cluster-name\n    consistencyLevel: strong\n    want:\n      queryResult: cluster0\n\n  # Validate wf1 completed in cluster0\n  - op: validate\n    at: 25s\n    workflowID: wf1\n    cluster: cluster0\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  # SignalWithStart wf2 in cluster1, will be handled by cluster0 (active cluster)\n  - op: signal_with_start_workflow\n    at: 30s\n    workflowType: query-workflow\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain\n    signalName: custom-signal\n    signalInput: \"x\"\n    workflowExecutionStartToCloseTimeout: 30s\n    workflowDuration: 20s\n\n  # SignalWithStart wf2 in cluster1 again and signal will be forwarded to cluster0 (active cluster) and wf2 will be started in cluster0\n  - op: signal_with_start_workflow\n    at: 31s\n    workflowType: query-workflow\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain\n    signalName: custom-signal\n    signalInput: \"y\"\n    workflowExecutionStartToCloseTimeout: 30s\n    workflowDuration: 20s\n\n\n  # Query wf2 in cluster1 with strong consistency and expect cluster0 (active cluster) worker handles query\n  # Should return signal count = 2\n  - op: query_workflow\n    at: 40s\n    workflowID: wf2\n    cluster: cluster1\n    domain: test-domain\n    query: signal-count\n    consistencyLevel: strong\n    want:\n      queryResult: 2.0 # query result unmarshalled to float64 so we need to use float64 here for comparison to work\n\n  # Validate wf2 completed in cluster0\n  - op: validate\n    at: 60s\n    workflowID: wf2\n    cluster: cluster0\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_default.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulations can be run via ./simulation/replication/run.sh --scenario default\n#\n# Test workflow behaviour during a failover\n# When a workflow is running in a cluster\n# And a failover occurs\n# Then the workflow should continue to run and complete in the new cluster\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\n\ndomains:\n  test-domain:\n    activeClusterName: cluster0\n\noperations:\n  - op: start_workflow\n    at: 0s\n    workflowType: timer-activity-loop-workflow\n    workflowID: timer-activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 65s\n    workflowDuration: 35s\n\n  - op: change_active_clusters # failover from cluster0 to cluster1\n    at: 20s\n    domain: test-domain\n    newActiveCluster: cluster1\n    # failoverTimeoutSec: 5 # unset means force failover. setting it means graceful failover request\n\n  - op: validate\n    at: 41s # with the current workflow implementation, it's possible that the workflow may finish right after the 40s mark\n    workflowID: timer-activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n"
  },
  {
    "path": "simulation/replication/testdata/replication_simulation_reset.yaml",
    "content": "# This file is a replication simulation scenario spec.\n# It is parsed into ReplicationSimulationConfig struct.\n# Replication simulation for this file can be run via ./simulation/replication/run.sh --scenario reset\n# Dynamic config overrides can be set via config/dynamicconfig/replication_simulation_reset.yml\n# Tests the behaviour of workflows when failovers and reset commands are issued\n\nclusters:\n  cluster0:\n    grpcEndpoint: \"cadence-cluster0:7833\"\n  cluster1:\n    grpcEndpoint: \"cadence-cluster1:7833\"\n\n# primaryCluster is where domain data is written to and replicates to others. e.g. domain registration\nprimaryCluster: \"cluster0\"\n\n\ndomains:\n  test-domain:\n    activeClusterName: cluster0\n  test-domain-2:\n    activeClusterName: cluster0\n\noperations:\n  # Start two workflows in different domains\n  - op: start_workflow\n    at: 0s\n    workflowType: activity-loop-workflow\n    workflowID: activity-loop-workflow1\n    activityCount: 45\n    cluster: cluster0\n    domain: test-domain\n    workflowExecutionStartToCloseTimeout: 60s\n\n  - op: start_workflow\n    at: 0s\n    workflowType: activity-loop-workflow\n    workflowID: activity-loop-workflow1\n    activityCount: 45\n    cluster: cluster0\n    domain: test-domain-2\n    workflowExecutionStartToCloseTimeout: 60s\n\n  # Failover from cluster0 to cluster1\n  - op: change_active_clusters \n    at: 20s\n    domain: test-domain\n    newActiveCluster: cluster1\n    # failoverTimeoutSec: 5 # unset means force failover. setting it means graceful failover request\n\n  - op: validate\n    at: 62s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain-2\n    want:\n      status: timed-out\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  - op: validate_workflow_replication\n    at: 62s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster0\n    targetCluster: cluster1\n    domain: test-domain-2\n\n  - op: validate\n    at: 62s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain\n    want:\n      status: timed-out\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  - op: validate_workflow_replication\n    at: 62s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster0\n    targetCluster: cluster1\n    domain: test-domain\n\n  - op: reset_workflow\n    at: 70s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain-2\n    eventID: 94\n\n  - op: reset_workflow\n    at: 70s\n    workflowID: activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    eventID: 94\n\n  - op: validate\n    at: 132s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain-2\n    want:\n      status: timed-out\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n  - op: validate_workflow_replication\n    at: 132s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster0\n    targetCluster: cluster1\n    domain: test-domain-2\n\n  - op: validate\n    at: 132s\n    workflowID: activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: timed-out\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  - op: validate_workflow_replication\n    at: 132s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster1\n    targetCluster: cluster0\n    domain: test-domain\n\n  - op: reset_workflow\n    at: 140s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain-2\n    eventID: 205\n\n  - op: reset_workflow\n    at: 140s\n    workflowID: activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    eventID: 205\n\n  - op: validate\n    at: 202s\n    workflowID: activity-loop-workflow1\n    cluster: cluster0\n    domain: test-domain-2\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster0\n\n\n  - op: validate_workflow_replication\n    at: 202s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster0\n    targetCluster: cluster1\n    domain: test-domain-2\n\n  - op: validate\n    at: 202s\n    workflowID: activity-loop-workflow1\n    cluster: cluster1\n    domain: test-domain\n    want:\n      status: completed\n      startedByWorkersInCluster: cluster0\n      completedByWorkersInCluster: cluster1\n\n  - op: validate_workflow_replication\n    at: 202s\n    workflowID: activity-loop-workflow1\n    sourceCluster: cluster1\n    targetCluster: cluster0\n    domain: test-domain"
  },
  {
    "path": "simulation/replication/types/repl_sim_config.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/require\"\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"gopkg.in/yaml.v2\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\tgrpcClient \"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\ntype ReplicationSimulationOperation string\n\nconst (\n\tReplicationSimulationOperationStartWorkflow               ReplicationSimulationOperation = \"start_workflow\"\n\tReplicationSimulationOperationResetWorkflow               ReplicationSimulationOperation = \"reset_workflow\"\n\tReplicationSimulationOperationChangeActiveClusters        ReplicationSimulationOperation = \"change_active_clusters\"\n\tReplicationSimulationOperationValidate                    ReplicationSimulationOperation = \"validate\"\n\tReplicationSimulationOperationQueryWorkflow               ReplicationSimulationOperation = \"query_workflow\"\n\tReplicationSimulationOperationSignalWithStartWorkflow     ReplicationSimulationOperation = \"signal_with_start_workflow\"\n\tReplicationSimulationOperationValidateWorkflowReplication ReplicationSimulationOperation = \"validate_workflow_replication\"\n)\n\ntype ReplicationSimulationConfig struct {\n\t// Clusters is the map of all clusters\n\tClusters map[string]*Cluster `yaml:\"clusters\"`\n\n\t// PrimaryCluster is used for domain registration\n\tPrimaryCluster string `yaml:\"primaryCluster\"`\n\n\tDomains map[string]ReplicationDomainConfig `yaml:\"domains\"`\n\n\tOperations []*Operation `yaml:\"operations\"`\n}\n\ntype ReplicationDomainConfig struct {\n\tActiveClusterName string `yaml:\"activeClusterName\"`\n\n\t// ClusterAttributes specifies the Attributes for a domain and is passed to ActiveClusters in the RegisterDomainRequest\n\t// It is a simplified expression for the AttributeScopes type\n\t// The format should be expressed as follows:\n\t// clusterAttributes: // This will be mapped to AttributeScopes\n\t//   region: // This is the key for a ClusterAttributeScope. All children are mapped to ClusterAttributes\n\t//\t \tus-west1: cluster0 // us-west1 is the key within a ClusterAttributeScope\n\t//      us-east1: cluster1 // cluster1 is the ActiveClusterName for the corresponding ActiveClusterInfo\n\t//   cityID:\n\t//      ...\n\tClusterAttributes ClusterAttributesMap `yaml:\"clusterAttributes\"`\n}\n\ntype Operation struct {\n\tType          ReplicationSimulationOperation `yaml:\"op\"`\n\tAt            time.Duration                  `yaml:\"at\"`\n\tCluster       string                         `yaml:\"cluster\"`\n\tSourceCluster string                         `yaml:\"sourceCluster\"`\n\tTargetCluster string                         `yaml:\"targetCluster\"`\n\n\tWorkflowType                         string                              `yaml:\"workflowType\"`\n\tWorkflowID                           string                              `yaml:\"workflowID\"`\n\tWorkflowExecutionStartToCloseTimeout time.Duration                       `yaml:\"workflowExecutionStartToCloseTimeout\"`\n\tWorkflowDuration                     time.Duration                       `yaml:\"workflowDuration\"`\n\tChildWorkflowID                      string                              `yaml:\"childWorkflowID\"`\n\tChildWorkflowTimeout                 time.Duration                       `yaml:\"childWorkflowTimeout\"`\n\tActivityCount                        int                                 `yaml:\"activityCount\"`\n\tDelayStartSeconds                    int32                               `yaml:\"delayStartSeconds\"`\n\tCronSchedule                         string                              `yaml:\"cronSchedule\"`\n\tActiveClusterSelectionPolicy         *types.ActiveClusterSelectionPolicy `yaml:\"activeClusterSelectionPolicy\"`\n\tWorkflowIDReusePolicy                *types.WorkflowIDReusePolicy        `yaml:\"workflowIDReusePolicy\"`\n\n\tQuery            string `yaml:\"query\"`\n\tConsistencyLevel string `yaml:\"consistencyLevel\"`\n\n\tSignalName  string `yaml:\"signalName\"`\n\tSignalInput any    `yaml:\"signalInput\"`\n\n\tEventID int64 `yaml:\"eventID\"`\n\n\tDomain           string `yaml:\"domain\"`\n\tNewActiveCluster string `yaml:\"newActiveCluster\"`\n\t// NewClusterAttributes specifies the AttributeScopes to change for the domain\n\t// This can be a sub-set of the total AttributeScopes for the domain\n\tNewClusterAttributes ClusterAttributesMap `yaml:\"newClusterAttributes\"`\n\tFailoverTimeout      *int32               `yaml:\"failoverTimeoutSec\"`\n\n\t// RunIDKey specifies a key to store/retrieve RunID for this operation\n\tRunIDKey string `yaml:\"runIDKey\"`\n\n\tWant Validation `yaml:\"want\"`\n}\n\ntype Validation struct {\n\tStatus                      string `yaml:\"status\"`\n\tStartedByWorkersInCluster   string `yaml:\"startedByWorkersInCluster\"`\n\tCompletedByWorkersInCluster string `yaml:\"completedByWorkersInCluster\"`\n\tError                       string `yaml:\"error\"`\n\tQueryResult                 any    `yaml:\"queryResult\"`\n}\n\ntype Cluster struct {\n\tGRPCEndpoint string `yaml:\"grpcEndpoint\"`\n\n\tAdminClient    admin.Client    `yaml:\"-\"`\n\tFrontendClient frontend.Client `yaml:\"-\"`\n}\n\nfunc LoadConfig() (*ReplicationSimulationConfig, error) {\n\tpath := os.Getenv(\"REPLICATION_SIMULATION_CONFIG\")\n\tif path == \"\" {\n\t\tpath = DefaultTestCase\n\t}\n\tconfContent, err := os.ReadFile(path)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to read config file: %w\", err)\n\t}\n\n\tvar cfg ReplicationSimulationConfig\n\terr = yaml.Unmarshal(confContent, &cfg)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to unmarshal config: %w\", err)\n\t}\n\n\tfmt.Printf(\"Loaded config from path: %s\\n\", path)\n\treturn &cfg, nil\n}\n\nfunc (s *ReplicationSimulationConfig) MustGetFrontendClient(t *testing.T, clusterName string) frontend.Client {\n\tt.Helper()\n\tcluster, ok := s.Clusters[clusterName]\n\trequire.True(t, ok, \"Cluster %s not found in the config\", clusterName)\n\trequire.NotNil(t, cluster.FrontendClient, \"Cluster %s frontend client not initialized\", clusterName)\n\treturn cluster.FrontendClient\n}\n\nfunc (s *ReplicationSimulationConfig) MustInitClientsFor(t *testing.T, clusterName string) {\n\tt.Helper()\n\tcluster, ok := s.Clusters[clusterName]\n\trequire.True(t, ok, \"Cluster %s not found in the config\", clusterName)\n\toutbounds := transport.Outbounds{Unary: grpc.NewTransport().NewSingleOutbound(cluster.GRPCEndpoint)}\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: \"cadence-client\",\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\"cadence-frontend\": outbounds,\n\t\t},\n\t})\n\n\tif err := dispatcher.Start(); err != nil {\n\t\tdispatcher.Stop()\n\t\trequire.NoError(t, err, \"failed to create outbound transport channel\")\n\t}\n\n\tclientConfig := dispatcher.ClientConfig(\"cadence-frontend\")\n\tcluster.FrontendClient = grpcClient.NewFrontendClient(\n\t\tapiv1.NewDomainAPIYARPCClient(clientConfig),\n\t\tapiv1.NewWorkflowAPIYARPCClient(clientConfig),\n\t\tapiv1.NewWorkerAPIYARPCClient(clientConfig),\n\t\tapiv1.NewVisibilityAPIYARPCClient(clientConfig),\n\t)\n\n\tcluster.AdminClient = grpcClient.NewAdminClient(adminv1.NewAdminAPIYARPCClient(clientConfig))\n\tLogf(t, \"Initialized clients for cluster %s\", clusterName)\n}\n\nfunc (s *ReplicationSimulationConfig) IsActiveActiveDomain(domainName string) bool {\n\tdomainCfg, ok := s.Domains[domainName]\n\tif !ok {\n\t\treturn false\n\t}\n\treturn len(domainCfg.ClusterAttributes.attributeScopes) > 0\n}\n\nfunc (s *ReplicationSimulationConfig) MustRegisterDomain(\n\tt *testing.T,\n\tdomainName string,\n\tdomainCfg ReplicationDomainConfig,\n) {\n\tLogf(t, \"Registering domain: %s\", domainName)\n\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tfor name := range s.Clusters {\n\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\tClusterName: name,\n\t\t})\n\t}\n\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\tdefer cancel()\n\treq := &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tClusters:                               clusters,\n\t\tWorkflowExecutionRetentionPeriodInDays: 1,\n\t\tIsGlobalDomain:                         true,\n\t}\n\n\tif len(domainCfg.ActiveClusterName) > 0 {\n\t\treq.ActiveClusterName = domainCfg.ActiveClusterName\n\t} else {\n\t\t// ActiveClusterName is required for all global domains\n\t\trequire.Fail(t, \"activeClusterName is required but missing for domain %s\", domainName)\n\t}\n\n\tif !domainCfg.ClusterAttributes.IsEmpty() {\n\t\treq.ActiveClusters = &types.ActiveClusters{}\n\t\treq.ActiveClusters.AttributeScopes = domainCfg.ClusterAttributes.ToAttributeScopes()\n\t}\n\n\terr := s.MustGetFrontendClient(t, s.PrimaryCluster).RegisterDomain(ctx, req)\n\n\tif err != nil {\n\t\tif _, ok := err.(*types.DomainAlreadyExistsError); !ok {\n\t\t\trequire.NoError(t, err, \"failed to register domain\")\n\t\t} else {\n\t\t\tLogf(t, \"Domains already exists: %s\", domainName)\n\t\t}\n\t\treturn\n\t}\n\n\tLogf(t, \"Registered domain: %s\", domainName)\n\n}\n\n// ClusterAttributesMap is a custom type for YAML unmarshalling of cluster attributes.\ntype ClusterAttributesMap struct {\n\tattributeScopes map[string]types.ClusterAttributeScope\n}\n\n// UnmarshalYAML implements custom YAML unmarshalling for ClusterAttributesMap.\nfunc (c *ClusterAttributesMap) UnmarshalYAML(unmarshal func(interface{}) error) error {\n\t// Unmarshal into the simplified map structure\n\tvar clusterAttributes map[string]map[string]string\n\tif err := unmarshal(&clusterAttributes); err != nil {\n\t\treturn err\n\t}\n\n\t// Convert to AttributeScopes\n\tc.attributeScopes = make(map[string]types.ClusterAttributeScope)\n\n\t// scopeType is the key for ClusterAttributeScope\n\t// It is the name of the scope, e.g region, datacenter, city, etc.\n\tfor scopeType, clusterAttributeScope := range clusterAttributes {\n\t\tattributeScope := types.ClusterAttributeScope{\n\t\t\tClusterAttributes: make(map[string]types.ActiveClusterInfo),\n\t\t}\n\t\t// attributeName is the ClusterAttribute key, e.g seattle for a city scope, us-west for a region scope, etc.\n\t\t// activeClusterName is the name of a cluster corresponding to the clusterMetadata setup\n\t\tfor attributeName, activeClusterName := range clusterAttributeScope {\n\t\t\tattributeScope.ClusterAttributes[attributeName] = types.ActiveClusterInfo{\n\t\t\t\tActiveClusterName: activeClusterName,\n\t\t\t}\n\t\t}\n\n\t\tc.attributeScopes[scopeType] = attributeScope\n\t}\n\n\treturn nil\n}\n\n// ToAttributeScopes is a convenience method to return the parsed AttributeScopes map.\nfunc (c *ClusterAttributesMap) ToAttributeScopes() map[string]types.ClusterAttributeScope {\n\tif c == nil {\n\t\treturn nil\n\t}\n\treturn c.attributeScopes\n}\n\nfunc (c *ClusterAttributesMap) IsEmpty() bool {\n\treturn c == nil || len(c.attributeScopes) == 0\n}\n"
  },
  {
    "path": "simulation/replication/types/types.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage types\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n)\n\nconst (\n\tDefaultTestCase = \"testdata/replication_simulation_default.yaml\"\n\tTasklistName    = \"test-tasklist\"\n\n\tTimerInterval = 5 * time.Second\n)\n\ntype OperationFunction func(t *testing.T, op *Operation, simCfg *ReplicationSimulationConfig) error\n\ntype WorkflowInput struct {\n\tDuration             time.Duration\n\tActivityCount        int\n\tChildWorkflowID      string\n\tChildWorkflowTimeout time.Duration\n}\n\ntype WorkflowOutput struct {\n\tCount int\n}\n\ntype ReplicationSimulation struct {\n\tRunIDRegistry map[string]string\n}\n\nfunc NewReplicationSimulation() *ReplicationSimulation {\n\treturn &ReplicationSimulation{\n\t\tRunIDRegistry: make(map[string]string),\n\t}\n}\n\nfunc (s *ReplicationSimulation) StoreRunID(key, runID string) error {\n\tif s.RunIDRegistry == nil {\n\t\treturn fmt.Errorf(\"runIDRegistry is nil\")\n\t}\n\ts.RunIDRegistry[key] = runID\n\treturn nil\n}\n\nfunc (s *ReplicationSimulation) GetRunID(key string) (string, error) {\n\tif s.RunIDRegistry == nil {\n\t\treturn \"\", fmt.Errorf(\"runIDRegistry is nil\")\n\t}\n\treturn s.RunIDRegistry[key], nil\n}\n\nfunc WorkerIdentityFor(clusterName string, domainName string) string {\n\tif domainName == \"\" {\n\t\treturn fmt.Sprintf(\"worker-%s\", clusterName)\n\t}\n\treturn fmt.Sprintf(\"worker-%s-%s\", domainName, clusterName)\n}\n\nfunc Logf(t *testing.T, msg string, args ...interface{}) {\n\tt.Helper()\n\tmsg = time.Now().Format(time.RFC3339Nano) + \"\\t\" + msg\n\tt.Logf(msg, args...)\n}\n"
  },
  {
    "path": "simulation/replication/worker/cmd/main.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage main\n\nimport (\n\t\"context\"\n\t\"flag\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"os\"\n\t\"os/signal\"\n\t\"sync\"\n\t\"sync/atomic\"\n\t\"syscall\"\n\t\"time\"\n\n\t\"github.com/uber-go/tally\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"go.uber.org/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"go.uber.org/cadence/.gen/go/shared\"\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/compatibility\"\n\t\"go.uber.org/cadence/worker\"\n\t\"go.uber.org/cadence/workflow\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/zap\"\n\t\"go.uber.org/zap/zapcore\"\n\n\t\"github.com/uber/cadence/common\"\n\tsimTypes \"github.com/uber/cadence/simulation/replication/types\"\n\t\"github.com/uber/cadence/simulation/replication/workflows\"\n)\n\nvar (\n\tclusterName = flag.String(\"cluster\", \"\", \"cluster name\")\n\n\tready int32\n)\n\nfunc main() {\n\tconfig := zap.NewDevelopmentConfig()\n\tconfig.Level.SetLevel(zapcore.InfoLevel)\n\n\tlogger, err := config.Build()\n\tif err != nil {\n\t\tpanic(fmt.Sprintf(\"failed to create logger: %v\", err))\n\t}\n\n\tflag.Parse()\n\tif *clusterName == \"\" {\n\t\tlogger.Fatal(\"cluster name is not set\")\n\t}\n\n\tsimCfg, err := simTypes.LoadConfig()\n\tif err != nil {\n\t\tlogger.Fatal(\"failed to load simulation config\", zap.Error(err))\n\t}\n\n\tcluster, ok := simCfg.Clusters[*clusterName]\n\tif !ok {\n\t\tlogger.Fatal(\"cluster not found in config\", zap.String(\"cluster\", *clusterName))\n\t}\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName: simTypes.WorkerIdentityFor(*clusterName, \"\"),\n\t\tOutbounds: yarpc.Outbounds{\n\t\t\t\"cadence-frontend\": {Unary: grpc.NewTransport().NewSingleOutbound(cluster.GRPCEndpoint)},\n\t\t},\n\t})\n\terr = dispatcher.Start()\n\tif err != nil {\n\t\tlogger.Fatal(\"Failed to start dispatcher\", zap.Error(err))\n\t}\n\tdefer dispatcher.Stop()\n\n\tclientConfig := dispatcher.ClientConfig(\"cadence-frontend\")\n\n\tcadenceClient := compatibility.NewThrift2ProtoAdapter(\n\t\tapiv1.NewDomainAPIYARPCClient(clientConfig),\n\t\tapiv1.NewWorkflowAPIYARPCClient(clientConfig),\n\t\tapiv1.NewWorkerAPIYARPCClient(clientConfig),\n\t\tapiv1.NewVisibilityAPIYARPCClient(clientConfig),\n\t)\n\n\thttp.HandleFunc(\"/health\", func(w http.ResponseWriter, r *http.Request) {\n\t\tif atomic.LoadInt32(&ready) == int32(len(simCfg.Domains)) {\n\t\t\tw.WriteHeader(http.StatusOK)\n\t\t} else {\n\t\t\tw.WriteHeader(http.StatusServiceUnavailable)\n\t\t}\n\t})\n\tgo http.ListenAndServe(\":6060\", nil)\n\n\twg := sync.WaitGroup{}\n\tfor domainName := range simCfg.Domains {\n\t\twg.Add(1)\n\n\t\tgo func(name string) {\n\t\t\twaitUntilDomainReady(logger, cadenceClient, name)\n\t\t\twg.Done()\n\t\t}(domainName)\n\t}\n\twg.Wait()\n\n\t// Create a channel to receive termination signals\n\tsigs := make(chan os.Signal, 1)\n\tsignal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)\n\n\t// Create a slice to hold the started workers\n\tvar workers []worker.Worker\n\n\tfor domainName := range simCfg.Domains {\n\t\tworkerOptions := worker.Options{\n\t\t\tIdentity:     simTypes.WorkerIdentityFor(*clusterName, domainName),\n\t\t\tLogger:       logger,\n\t\t\tMetricsScope: tally.NewTestScope(simTypes.TasklistName, map[string]string{\"cluster\": *clusterName}),\n\t\t}\n\n\t\tw := worker.New(\n\t\t\tcadenceClient,\n\t\t\tdomainName,\n\t\t\tsimTypes.TasklistName,\n\t\t\tworkerOptions,\n\t\t)\n\n\t\tfor name, wf := range workflows.Workflows(*clusterName) {\n\t\t\tw.RegisterWorkflowWithOptions(wf, workflow.RegisterOptions{Name: name})\n\t\t}\n\n\t\tfor name, act := range workflows.Activities {\n\t\t\tw.RegisterActivityWithOptions(act, activity.RegisterOptions{Name: name})\n\t\t}\n\n\t\terr := w.Start()\n\t\tif err != nil {\n\t\t\tlogger.Fatal(\"Failed to start worker\", zap.Error(err))\n\t\t}\n\t\tworkers = append(workers, w) // Add the worker to the slice\n\n\t\tfmt.Printf(\"Started worker for domain: %s\\n\", domainName)\n\t\tlogger.Info(\"Started worker\", zap.String(\"cluster\", *clusterName), zap.String(\"endpoint\", cluster.GRPCEndpoint))\n\t}\n\n\tlogger.Info(\"All workers started. Waiting for SIGINT or SIGTERM\")\n\tsig := <-sigs\n\tlogger.Sugar().Infof(\"Received signal: %v so terminating\", sig)\n\n\t// Stop each worker gracefully\n\tlogger.Info(\"Stopping workers...\")\n\tfor _, w := range workers {\n\t\tw.Stop()\n\t\tlogger.Info(\"Stopped worker\")\n\t}\n}\n\nfunc waitUntilDomainReady(logger *zap.Logger, client workflowserviceclient.Interface, domainName string) {\n\tfor {\n\t\tctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)\n\t\t_, err := client.DescribeDomain(ctx, &shared.DescribeDomainRequest{\n\t\t\tName: common.StringPtr(domainName),\n\t\t})\n\n\t\tcancel()\n\t\tif err == nil {\n\t\t\tlogger.Info(\"Domains is ready\", zap.String(\"domain\", domainName))\n\t\t\tatomic.AddInt32(&ready, 1)\n\t\t\treturn\n\t\t}\n\n\t\tlogger.Info(\"Domains not ready\", zap.String(\"domain\", domainName), zap.Error(err))\n\t\ttime.Sleep(2 * time.Second)\n\t}\n}\n"
  },
  {
    "path": "simulation/replication/workflows/activityloop/workflow.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage activityloop\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/simulation/replication/types\"\n)\n\nfunc Workflow(ctx workflow.Context, input types.WorkflowInput) (types.WorkflowOutput, error) {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Sugar().Infof(\"single-serial-activity-workflow started with input: %+v\", input)\n\n\tcount := 0\n\n\tfor count < input.ActivityCount {\n\t\tlogger.Sugar().Infof(\"single-serial-activity-workflow iteration %d\", count)\n\t\tselector := workflow.NewSelector(ctx)\n\t\tactivityFuture := workflow.ExecuteActivity(workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\t\tTaskList:               types.TasklistName,\n\t\t\tScheduleToStartTimeout: 10 * time.Second,\n\t\t\tStartToCloseTimeout:    10 * time.Second,\n\t\t}), FormatStringActivity, \"World\")\n\t\tselector.AddFuture(activityFuture, func(f workflow.Future) {\n\t\t\tlogger.Info(\"single-serial-activity-workflow completed activity\")\n\t\t})\n\n\t\tselector.Select(ctx)\n\t\tcount++\n\t}\n\n\tlogger.Info(\"single-serial-activity-workflow completed\")\n\treturn types.WorkflowOutput{Count: count}, nil\n}\n\nfunc FormatStringActivity(ctx context.Context, input string) (string, error) {\n\tlogger := activity.GetLogger(ctx)\n\tlogger.Info(\"single-serial-activity-workflow format-string-activity started\")\n\n\ttime.Sleep(3 * time.Second)\n\n\treturn fmt.Sprintf(\"Hello, %s!\", input), nil\n}\n"
  },
  {
    "path": "simulation/replication/workflows/childactivityloop/workflow.go",
    "content": "package childactivityloop\n\nimport (\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/simulation/replication/types\"\n)\n\nfunc Workflow(ctx workflow.Context, input types.WorkflowInput) (types.WorkflowOutput, error) {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Sugar().Infof(\"child-activity-loop-workflow started with input: %+v\", input)\n\n\tcwo := workflow.ChildWorkflowOptions{\n\t\tWorkflowID:                   input.ChildWorkflowID,\n\t\tExecutionStartToCloseTimeout: input.ChildWorkflowTimeout,\n\t}\n\tctx = workflow.WithChildOptions(ctx, cwo)\n\tvar output types.WorkflowOutput\n\terr := workflow.ExecuteChildWorkflow(ctx, \"timer-activity-loop-workflow\", input).Get(ctx, &output)\n\tif err != nil {\n\t\tlogger.Sugar().Errorf(\"failed to execute child workflow: %v\", err)\n\t\treturn types.WorkflowOutput{}, err\n\t}\n\n\treturn output, nil\n}\n"
  },
  {
    "path": "simulation/replication/workflows/query/workflow.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage query\n\nimport (\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/simulation/replication/types\"\n)\n\nconst (\n\tsignalName       = \"custom-signal\"\n\tclusterNameQuery = \"cluster-name\"\n\tsignalCountQuery = \"signal-count\"\n)\n\ntype Runner struct {\n\tClusterName string\n}\n\nfunc (r *Runner) Workflow(ctx workflow.Context, input types.WorkflowInput) (types.WorkflowOutput, error) {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Sugar().Infof(\"query workflow started with input: %+v\", input)\n\n\terr := workflow.SetQueryHandler(ctx, clusterNameQuery, func() (string, error) {\n\t\tlogger.Sugar().Infof(\"query handler called. returning cluster name: %s\", r.ClusterName)\n\t\treturn r.ClusterName, nil\n\t})\n\tif err != nil {\n\t\tlogger.Sugar().Errorf(\"failed to set query handler: %v\", err)\n\t\treturn types.WorkflowOutput{}, err\n\t}\n\n\tsignalCount := 0\n\terr = workflow.SetQueryHandler(ctx, signalCountQuery, func() (int, error) {\n\t\tlogger.Sugar().Infof(\"query handler called. returning signal count: %d\", signalCount)\n\t\treturn signalCount, nil\n\t})\n\tif err != nil {\n\t\tlogger.Sugar().Errorf(\"failed to set query handler: %v\", err)\n\t\treturn types.WorkflowOutput{}, err\n\t}\n\n\tendTime := workflow.Now(ctx).Add(input.Duration)\n\tsignalCh := workflow.GetSignalChannel(ctx, signalName)\n\tdone := false\n\tfor {\n\t\tselector := workflow.NewSelector(ctx)\n\n\t\t// timer\n\t\ttimerCtx, timerCancel := workflow.WithCancel(ctx)\n\t\twaitTimer := workflow.NewTimer(timerCtx, endTime.Sub(workflow.Now(ctx)))\n\t\tselector.AddFuture(waitTimer, func(f workflow.Future) {\n\t\t\tdone = true\n\t\t})\n\n\t\t// signal\n\t\tselector.AddReceive(signalCh, func(c workflow.Channel, more bool) {\n\t\t\tvar signal string\n\t\t\tfor c.ReceiveAsync(&signal) {\n\t\t\t\tsignalCount++\n\t\t\t\tlogger.Sugar().Infof(\"signal received: %s\", signal)\n\t\t\t}\n\t\t})\n\n\t\tselector.Select(ctx)\n\t\ttimerCancel()\n\t\tif done {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tlogger.Info(\"query workflow completed\")\n\treturn types.WorkflowOutput{}, nil\n}\n"
  },
  {
    "path": "simulation/replication/workflows/timeractivityloop/workflow.go",
    "content": "// Copyright (c) 2018 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage timeractivityloop\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"time\"\n\n\t\"go.uber.org/cadence/activity\"\n\t\"go.uber.org/cadence/workflow\"\n\n\t\"github.com/uber/cadence/simulation/replication/types\"\n)\n\nconst (\n\tlatestSignalContentQuery = \"latest-signal-content\"\n\tsignalName               = \"custom-signal\"\n)\n\nfunc Workflow(ctx workflow.Context, input types.WorkflowInput) (types.WorkflowOutput, error) {\n\tlogger := workflow.GetLogger(ctx)\n\tlogger.Sugar().Infof(\"timer-activity-loop-workflow started with input: %+v\", input)\n\n\tsignalContent := make([]string, 0)\n\terr := workflow.SetQueryHandler(ctx, latestSignalContentQuery, func() ([]string, error) {\n\t\tlogger.Sugar().Infof(\"query handler called. returning all signal content: %s\", signalContent)\n\t\treturn signalContent, nil\n\t})\n\tif err != nil {\n\t\tlogger.Sugar().Errorf(\"failed to set query handler: %v\", err)\n\t\treturn types.WorkflowOutput{}, err\n\t}\n\n\tsignalCh := workflow.GetSignalChannel(ctx, signalName)\n\tendTime := workflow.Now(ctx).Add(input.Duration)\n\tcount := 0\n\tfor {\n\t\tlogger.Sugar().Infof(\"timer-activity-loop-workflow iteration %d\", count)\n\t\tselector := workflow.NewSelector(ctx)\n\t\tactivityFuture := workflow.ExecuteActivity(workflow.WithActivityOptions(ctx, workflow.ActivityOptions{\n\t\t\tTaskList:               types.TasklistName,\n\t\t\tScheduleToStartTimeout: 10 * time.Second,\n\t\t\tStartToCloseTimeout:    10 * time.Second,\n\t\t}), FormatStringActivity, \"World\")\n\t\tselector.AddFuture(activityFuture, func(f workflow.Future) {\n\t\t\tlogger.Info(\"timer-activity-loop-workflow completed activity\")\n\t\t})\n\n\t\t// use timer future to send notification email if processing takes too long\n\t\ttimerFuture := workflow.NewTimer(ctx, types.TimerInterval)\n\t\tselector.AddFuture(timerFuture, func(f workflow.Future) {\n\t\t\tlogger.Info(\"timer-activity-loop-workflow timer fired\")\n\t\t})\n\n\t\tselector.AddReceive(signalCh, func(c workflow.Channel, more bool) {\n\t\t\tvar signal string\n\t\t\tfor c.ReceiveAsync(&signal) {\n\t\t\t\tlogger.Sugar().Infof(\"signal received: %s\", signal)\n\t\t\t\tsignalContent = append(signalContent, signal)\n\t\t\t}\n\t\t})\n\n\t\t// wait for both activity and timer to complete\n\t\tselector.Select(ctx)\n\t\tselector.Select(ctx)\n\t\tcount++\n\n\t\tnow := workflow.Now(ctx)\n\t\tif now.Before(endTime) {\n\t\t\tlogger.Sugar().Infof(\"timer-activity-loop-workflow will continue iteration because [now %v] < [endTime %v]\", now, endTime)\n\t\t} else {\n\t\t\tlogger.Sugar().Infof(\"timer-activity-loop-workflow will exit because [now %v] >= [endTime %v]\", now, endTime)\n\t\t\tbreak\n\t\t}\n\t}\n\n\tlogger.Info(\"timer-activity-loop-workflow completed\")\n\treturn types.WorkflowOutput{Count: count}, nil\n}\n\nfunc FormatStringActivity(ctx context.Context, input string) (string, error) {\n\tlogger := activity.GetLogger(ctx)\n\tlogger.Info(\"timer-activity-loop-workflow format-string-activity started\")\n\treturn fmt.Sprintf(\"Hello, %s!\", input), nil\n}\n"
  },
  {
    "path": "simulation/replication/workflows/workflows.go",
    "content": "package workflows\n\nimport (\n\t\"github.com/uber/cadence/simulation/replication/workflows/activityloop\"\n\t\"github.com/uber/cadence/simulation/replication/workflows/childactivityloop\"\n\t\"github.com/uber/cadence/simulation/replication/workflows/query\"\n\t\"github.com/uber/cadence/simulation/replication/workflows/timeractivityloop\"\n)\n\n// Add workflows and activities to this map to register them with the worker.\nvar (\n\tWorkflows = func(clusterName string) map[string]any {\n\t\tqueryWFRunner := &query.Runner{ClusterName: clusterName}\n\t\treturn map[string]any{\n\t\t\t\"timer-activity-loop-workflow\": timeractivityloop.Workflow,\n\t\t\t\"activity-loop-workflow\":       activityloop.Workflow,\n\t\t\t\"query-workflow\":               queryWFRunner.Workflow,\n\t\t\t\"child-activity-loop-workflow\": childactivityloop.Workflow,\n\t\t}\n\t}\n\tActivities = map[string]any{\n\t\t\"timer-activity-loop-format-string-activity\": timeractivityloop.FormatStringActivity,\n\t\t\"activity-loop-format-string-activity\":       activityloop.FormatStringActivity,\n\t}\n)\n"
  },
  {
    "path": "testflags/testflags.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage testflags\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n)\n\n// Pass these testflags in as environment variables in order to run tests marked as needing an\n// external dependency.\n// Tests can be marked as having an external dependency by calling the Require* functions below.\n// In order to start external dependencies, see documentation at https://github.com/uber/cadence\n\nvar (\n\tcassandra = \"CASSANDRA\"\n\tmongodb   = \"MONGODB\"\n\tmysql     = \"MYSQL\"\n\tpostgres  = \"POSTGRES\"\n\tetcd      = \"ETCD\"\n)\n\n// NOTE: We are using environment variables instead of go testflags or go build directives, because we:\n// 1) Want tests to be built all the time (go build directives don't give us that)\n// 2) Want tests to be marked as skipped (go build directives don't give us that)\n// 3) Want to be able to run individual tests (go testflags errors if the test doesn't import\n// something that defines the testflags.)\n\nfunc RequireMySQL(t *testing.T) {\n\trequire(t, mysql)\n}\n\nfunc RequirePostgres(t *testing.T) {\n\trequire(t, postgres)\n}\n\nfunc RequireMongoDB(t *testing.T) {\n\trequire(t, mongodb)\n}\n\nfunc RequireCassandra(t *testing.T) {\n\trequire(t, cassandra)\n}\n\nfunc RequireEtcd(t *testing.T) {\n\trequire(t, etcd)\n}\n\nfunc require(t *testing.T, name string) {\n\tif !checkEnv(name) {\n\t\tt.Skip(fmt.Sprintf(\"Skipping test that requires %s to run - start %s and set '%s=1' environment variable to run this test.\",\n\t\t\tname, name, name))\n\t}\n}\n\nfunc checkEnv(name string) bool {\n\treturn os.Getenv(name) != \"\"\n}\n"
  },
  {
    "path": "tools/cassandra/README.md",
    "content": "## Using the cassandra schema tool\nThis package contains the tooling for cadence cassandra operations.\n\n## For localhost development\n```\nmake install-schema\n```\n> NOTE: See [CONTRIBUTING](/CONTRIBUTING.md) for prerequisite of make command.\n\n## For production\n\n### Get the Cassandra Schema tool\n* Use brew to install CLI: `brew install cadence-workflow` which includes `cadence-cassandra-tool`\n  * The schema files are located at `/usr/local/etc/cadence/schema/`.\n  * Follow the [instructions](https://github.com/cadence-workflow/cadence/discussions/4457) if you need to install older versions of schema tools via homebrew.\n However, easier way is to use new versions of schema tools with old versions of schemas.\n All you need is to check out the older version of schemas from this repo. Run `git checkout v0.21.3` to get the v0.21.3 schemas in [the schema folder](/schema).\n* Or build yourself, with `make cadence-cassandra-tool`. See [CONTRIBUTING](/CONTRIBUTING.md) for prerequisite of make command.\n\n> Note: The binaries can also be found in the `ubercadence/server` docker images\n\n### Do one time database creation and schema setup for a new cluster\nThis uses Cassandra's SimpleStratagey for replication. For production, we recommend using a replication factor of 3 with NetworkTopologyStrategy.\n\n```\ncadence-cassandra-tool --ep $CASSANDRA_SEEDS create -k $KEYSPACE --rf $RF\n```\n\nSee https://www.ecyrd.com/cassandracalculator for an easy way to determine how many nodes and what replication factor you will want to use.  Note that Cadence by default uses `Quorum` for read and write consistency.\n\n```\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence setup-schema -v 0.0 # this sets up just the schema version tables with initial version of 0.0\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence update-schema -d ./schema/cassandra/cadence/versioned # upgrades your schema to the latest version\n\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence_visibility setup-schema -v 0.0 # this sets up just the schema version tables with initial version of 0.0 for visibility\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence_visibility update-schema -d ./schema/cassandra/visibility/versioned # upgrades your schema to the latest version for visibility\n```\n\n### Update schema as part of a release\nYou can only upgrade to a new version after the initial setup done above.\n\n```\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence update-schema -d ./schema/cassandra/cadence/versioned -v x.xx -dryrun # executes a dryrun of upgrade to version x.xx\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence update-schema -d ./schema/cassandra/cadence/versioned -v x.xx # actually executes the upgrade to version x.xx\n\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence_visibility update-schema -d ./schema/cassandra/visibility/versioned -v x.xx --dryrun # executes a dryrun of upgrade to version x.xx\n./cadence-cassandra-tool -ep 127.0.0.1 -k cadence_visibility update-schema -d ./schema/cassandra/visibility/versioned -v x.xx  # actually executes the upgrade to version x.xx\n```\n"
  },
  {
    "path": "tools/cassandra/cqlclient.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"log\"\n\t\"time\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\ntype (\n\tCqlClient interface {\n\t\tCreateDatabase(name string) error\n\t\tDropDatabase(name string) error\n\t\tCreateKeyspace(name string) error\n\t\tCreateNTSKeyspace(name string, datacenter string) error\n\t\tDropKeyspace(name string) error\n\t\tDropAllTables() error\n\t\tCreateSchemaVersionTables() error\n\t\tReadSchemaVersion() (string, error)\n\t\tUpdateSchemaVersion(newVersion string, minCompatibleVersion string) error\n\t\tWriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error\n\t\tExecDDLQuery(stmt string, args ...interface{}) error\n\t\tClose()\n\t\tListTables() ([]string, error)\n\t\tListTypes() ([]string, error)\n\t\tDropTable(name string) error\n\t\tDropType(name string) error\n\t\tDropAllTablesTypes() error\n\t}\n\n\tCqlClientImpl struct {\n\t\tnReplicas int\n\t\tsession   gocql.Session\n\t\tcfg       *CQLClientConfig\n\t}\n\n\t// CQLClientConfig contains the configuration for cql client\n\tCQLClientConfig struct {\n\t\tHosts                 string\n\t\tPort                  int\n\t\tUser                  string\n\t\tPassword              string\n\t\tAllowedAuthenticators []string\n\t\tKeyspace              string\n\t\tTimeout               int\n\t\tConnectTimeout        int\n\t\tNumReplicas           int\n\t\tProtoVersion          int\n\t\tTLS                   *config.TLS\n\t}\n)\n\nconst (\n\tDefaultTimeout        = 30 // Timeout in seconds\n\tDefaultConnectTimeout = 2  // Connect timeout in seconds\n\tDefaultCassandraPort  = 9042\n\tSystemKeyspace        = \"system\"\n)\n\nconst (\n\t// Retry policy constants\n\tClientCreationRetryInitialInterval = time.Second\n\tClientCreationRetryMaximumInterval = 10 * time.Second\n\tClientCreationRetryMaximumAttempts = 3\n)\n\nconst (\n\treadSchemaVersionCQL        = `SELECT curr_version from schema_version where keyspace_name=?`\n\tlistTablesCQL               = `SELECT table_name from system_schema.tables where keyspace_name=?`\n\tlistTypesCQL                = `SELECT type_name from system_schema.types where keyspace_name=?`\n\twriteSchemaVersionCQL       = `INSERT into schema_version(keyspace_name, creation_time, curr_version, min_compatible_version) VALUES (?,?,?,?)`\n\twriteSchemaUpdateHistoryCQL = `INSERT into schema_update_history(year, month, update_time, old_version, new_version, manifest_md5, description) VALUES(?,?,?,?,?,?,?)`\n\n\tcreateSchemaVersionTableCQL = `CREATE TABLE schema_version(keyspace_name text PRIMARY KEY, ` +\n\t\t`creation_time timestamp, ` +\n\t\t`curr_version text, ` +\n\t\t`min_compatible_version text);`\n\n\tcreateSchemaUpdateHistoryTableCQL = `CREATE TABLE schema_update_history(` +\n\t\t`year int, ` +\n\t\t`month int, ` +\n\t\t`update_time timestamp, ` +\n\t\t`description text, ` +\n\t\t`manifest_md5 text, ` +\n\t\t`new_version text, ` +\n\t\t`old_version text, ` +\n\t\t`PRIMARY KEY ((year, month), update_time));`\n\n\tcreateKeyspaceCQL = `CREATE KEYSPACE IF NOT EXISTS %v ` +\n\t\t`WITH replication = { 'class' : 'SimpleStrategy', 'replication_factor' : %v};`\n\n\tcreateNTSKeyspaceCQL = `CREATE KEYSPACE IF NOT EXISTS %v ` +\n\t\t`WITH replication = { 'class' : 'NetworkTopologyStrategy', '%v' : %v};`\n)\n\nvar _ schema.SchemaClient = (*CqlClientImpl)(nil)\n\nfunc NewCQLClient(cfg *CQLClientConfig, expectedConsistency gocql.Consistency) (CqlClient, error) {\n\tretrier := createClientCreationRetrier()\n\tcassClient := gocql.GetRegisteredClient()\n\treturn NewCQLClientWithRetry(cfg, expectedConsistency, cassClient, retrier)\n}\n\n// NewCQLClient returns a new instance of CQLClient\nfunc NewCQLClientWithRetry(cfg *CQLClientConfig, expectedConsistency gocql.Consistency, cassClient gocql.Client, retrier *backoff.ThrottleRetry) (CqlClient, error) {\n\tcqlClient := new(CqlClientImpl)\n\tcqlClient.cfg = cfg\n\tcqlClient.nReplicas = cfg.NumReplicas\n\n\terr := retrier.Do(context.Background(), func(ctx context.Context) error {\n\t\tvar err error\n\t\tcqlClient.session, err = cassClient.CreateSession(gocql.ClusterConfig{\n\t\t\tHosts:                 cfg.Hosts,\n\t\t\tPort:                  cfg.Port,\n\t\t\tUser:                  cfg.User,\n\t\t\tPassword:              cfg.Password,\n\t\t\tAllowedAuthenticators: cfg.AllowedAuthenticators,\n\t\t\tKeyspace:              cfg.Keyspace,\n\t\t\tTLS:                   cfg.TLS,\n\t\t\tTimeout:               time.Duration(cfg.Timeout) * time.Second,\n\t\t\tConnectTimeout:        time.Duration(cfg.ConnectTimeout) * time.Second,\n\t\t\tProtoVersion:          cfg.ProtoVersion,\n\t\t\tConsistency:           expectedConsistency,\n\t\t})\n\t\treturn err\n\t})\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn cqlClient, nil\n}\n\nfunc (client *CqlClientImpl) CreateDatabase(name string) error {\n\treturn client.CreateKeyspace(name)\n}\n\nfunc (client *CqlClientImpl) DropDatabase(name string) error {\n\treturn client.DropKeyspace(name)\n}\n\n// CreateNTSKeyspace creates a cassandra Keyspace if it doesn't exist using network topology strategy\nfunc (client *CqlClientImpl) CreateKeyspace(name string) error {\n\treturn client.ExecDDLQuery(fmt.Sprintf(createKeyspaceCQL, name, client.nReplicas))\n}\n\n// CreateNTSKeyspace creates a cassandra Keyspace if it doesn't exist using network topology strategy\nfunc (client *CqlClientImpl) CreateNTSKeyspace(name string, datacenter string) error {\n\treturn client.ExecDDLQuery(fmt.Sprintf(createNTSKeyspaceCQL, name, datacenter, client.nReplicas))\n}\n\n// DropKeyspace drops a Keyspace\nfunc (client *CqlClientImpl) DropKeyspace(name string) error {\n\treturn client.ExecDDLQuery(fmt.Sprintf(\"DROP KEYSPACE %v\", name))\n}\n\nfunc (client *CqlClientImpl) DropAllTables() error {\n\treturn client.DropAllTablesTypes()\n}\n\n// CreateSchemaVersionTables sets up the schema version tables\nfunc (client *CqlClientImpl) CreateSchemaVersionTables() error {\n\tif err := client.ExecDDLQuery(createSchemaVersionTableCQL); err != nil {\n\t\treturn err\n\t}\n\treturn client.ExecDDLQuery(createSchemaUpdateHistoryTableCQL)\n}\n\n// ReadSchemaVersion returns the current schema version for the Keyspace\nfunc (client *CqlClientImpl) ReadSchemaVersion() (string, error) {\n\tquery := client.session.Query(readSchemaVersionCQL, client.cfg.Keyspace)\n\titer := query.Iter()\n\tvar version string\n\tif !iter.Scan(&version) {\n\t\terr := iter.Close()\n\t\treturn \"\", fmt.Errorf(\"reading schema version: %w\", err)\n\t}\n\tif err := iter.Close(); err != nil {\n\t\treturn \"\", err\n\t}\n\treturn version, nil\n}\n\n// UpdateSchemaVersion updates the schema version for the Keyspace\nfunc (client *CqlClientImpl) UpdateSchemaVersion(newVersion string, minCompatibleVersion string) error {\n\tquery := client.session.Query(writeSchemaVersionCQL, client.cfg.Keyspace, time.Now(), newVersion, minCompatibleVersion)\n\treturn query.Exec()\n}\n\n// WriteSchemaUpdateLog adds an entry to the schema update history table\nfunc (client *CqlClientImpl) WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error {\n\tnow := time.Now().UTC()\n\tquery := client.session.Query(writeSchemaUpdateHistoryCQL)\n\tquery.Bind(now.Year(), int(now.Month()), now, oldVersion, newVersion, manifestMD5, desc)\n\treturn query.Exec()\n}\n\n// ExecDDLQuery executes a cql statement\nfunc (client *CqlClientImpl) ExecDDLQuery(stmt string, args ...interface{}) error {\n\treturn client.session.Query(stmt, args...).Exec()\n}\n\n// Close closes the cql client\nfunc (client *CqlClientImpl) Close() {\n\tif client.session != nil {\n\t\tclient.session.Close()\n\t}\n}\n\n// ListTables lists the table names in a Keyspace\nfunc (client *CqlClientImpl) ListTables() ([]string, error) {\n\tquery := client.session.Query(listTablesCQL, client.cfg.Keyspace)\n\titer := query.Iter()\n\tvar names []string\n\tvar name string\n\tfor iter.Scan(&name) {\n\t\tnames = append(names, name)\n\t}\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn names, nil\n}\n\n// ListTypes lists the User defined types in a Keyspace.\nfunc (client *CqlClientImpl) ListTypes() ([]string, error) {\n\tqry := client.session.Query(listTypesCQL, client.cfg.Keyspace)\n\titer := qry.Iter()\n\tvar names []string\n\tvar name string\n\tfor iter.Scan(&name) {\n\t\tnames = append(names, name)\n\t}\n\tif err := iter.Close(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn names, nil\n}\n\n// DropTable drops a given table from the Keyspace\nfunc (client *CqlClientImpl) DropTable(name string) error {\n\treturn client.ExecDDLQuery(fmt.Sprintf(\"DROP TABLE %v\", name))\n}\n\n// DropType drops a given type from the Keyspace\nfunc (client *CqlClientImpl) DropType(name string) error {\n\treturn client.ExecDDLQuery(fmt.Sprintf(\"DROP TYPE %v\", name))\n}\n\n// DropAllTablesTypes deletes all tables/types in the\n// Keyspace without deleting the Keyspace\nfunc (client *CqlClientImpl) DropAllTablesTypes() error {\n\ttables, err := client.ListTables()\n\tif err != nil {\n\t\treturn err\n\t}\n\tlog.Printf(\"Dropping following tables: %v\\n\", tables)\n\tfor _, table := range tables {\n\t\terr1 := client.DropTable(table)\n\t\tif err1 != nil {\n\t\t\tlog.Printf(\"Error dropping table %v, err=%v\\n\", table, err1)\n\t\t}\n\t}\n\n\ttypes, err := client.ListTypes()\n\tif err != nil {\n\t\treturn err\n\t}\n\tlog.Printf(\"Dropping following types: %v\\n\", types)\n\tnumOfTypes := len(types)\n\tfor i := 0; i < numOfTypes && len(types) > 0; i++ {\n\t\tvar erroredTypes []string\n\t\tfor _, t := range types {\n\t\t\terr = client.DropType(t)\n\t\t\tif err != nil {\n\t\t\t\tlog.Printf(\"Error dropping type %v, err=%v\\n\", t, err)\n\t\t\t\terroredTypes = append(erroredTypes, t)\n\t\t\t}\n\t\t}\n\t\ttypes = erroredTypes\n\t}\n\tif len(types) > 0 {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc createClientCreationRetrier() *backoff.ThrottleRetry {\n\tpolicy := backoff.NewExponentialRetryPolicy(ClientCreationRetryInitialInterval)\n\tpolicy.SetMaximumInterval(ClientCreationRetryMaximumInterval)\n\tpolicy.SetMaximumAttempts(ClientCreationRetryMaximumAttempts)\n\n\tretrier := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(policy),\n\t\tbackoff.WithRetryableError(func(err error) bool { return true }),\n\t)\n\treturn retrier\n}\n"
  },
  {
    "path": "tools/cassandra/cqlclient_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"go.uber.org/goleak\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n)\n\nfunc TestNewCQLClientWithRetry(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tmockClient, mockSession, mockClock := setUpMocks(t)\n\tcfg := configuration()\n\texpectedCfg := expectedConfig()\n\n\t// Fail the first two times, then succeed.\n\tmockClient.EXPECT().CreateSession(expectedCfg).Return(mockSession, assert.AnError)\n\tmockClient.EXPECT().CreateSession(expectedCfg).Return(mockSession, assert.AnError)\n\tmockClient.EXPECT().CreateSession(expectedCfg).Return(mockSession, nil)\n\n\tgo func() {\n\t\t// First retry happens after 1 second\n\t\tmockClock.BlockUntil(1)\n\t\tmockClock.Advance(1 * time.Second)\n\t\t// Second retry happens after 2 seconds\n\t\tmockClock.BlockUntil(1)\n\t\tmockClock.Advance(2 * time.Second)\n\t}()\n\n\ttestRetrier := createClientCreationRetrier()\n\tbackoff.WithClock(mockClock)(testRetrier)\n\n\tcqlClient, err := NewCQLClientWithRetry(cfg, gocql.One, mockClient, testRetrier)\n\tassert.NoError(t, err)\n\n\tcqlClientImpl, ok := cqlClient.(*CqlClientImpl)\n\trequire.True(t, ok)\n\n\tassert.Equal(t, 333, cqlClientImpl.nReplicas)\n\tassert.Equal(t, cfg, cqlClientImpl.cfg)\n\tassert.Equal(t, mockSession, cqlClientImpl.session)\n}\n\nfunc TestNewCQLClientWithRetry_Fail(t *testing.T) {\n\tdefer goleak.VerifyNone(t)\n\n\tmockClient, mockSession, mockClock := setUpMocks(t)\n\tcfg := configuration()\n\texpectedCfg := expectedConfig()\n\n\t// There is a off by one error in the retrier implementation, so we need to add one more call\n\t// if this is fixed, it's fine to remove the +1\n\tmockClient.EXPECT().CreateSession(expectedCfg).Return(mockSession, assert.AnError).Times(ClientCreationRetryMaximumAttempts + 1)\n\n\tgo func() {\n\t\t// We double the wait time each iteration\n\t\tnextAdvance := 1 * time.Second\n\n\t\t// Wait until someone sleeps, then advance the clock more than the backoff would be.\n\t\tfor i := 0; i < ClientCreationRetryMaximumAttempts; i++ {\n\t\t\tmockClock.BlockUntil(1)\n\t\t\tmockClock.Advance(nextAdvance)\n\t\t\tnextAdvance *= 2\n\t\t}\n\t}()\n\n\ttestRetrier := createClientCreationRetrier()\n\tbackoff.WithClock(mockClock)(testRetrier)\n\n\t_, err := NewCQLClientWithRetry(cfg, gocql.One, mockClient, testRetrier)\n\tassert.ErrorIs(t, err, assert.AnError)\n}\n\nfunc setUpMocks(t *testing.T) (*gocql.MockClient, *gocql.MockSession, clock.MockedTimeSource) {\n\tctrl := gomock.NewController(t)\n\tmockClient := gocql.NewMockClient(ctrl)\n\tmockSession := gocql.NewMockSession(ctrl)\n\tmockClock := clock.NewMockedTimeSource()\n\treturn mockClient, mockSession, mockClock\n}\n\nfunc configuration() *CQLClientConfig {\n\treturn &CQLClientConfig{\n\t\tHosts:                 \"testHost\",\n\t\tPort:                  1234,\n\t\tUser:                  \"testUser\",\n\t\tPassword:              \"testPassword\",\n\t\tAllowedAuthenticators: []string{\"testAuthenticator\"},\n\t\tKeyspace:              \"testKeyspace\",\n\t\tTimeout:               111,\n\t\tConnectTimeout:        222,\n\t\tNumReplicas:           333,\n\t\tProtoVersion:          444,\n\t\tTLS:                   &config.TLS{Enabled: true},\n\t}\n}\n\nfunc expectedConfig() gocql.ClusterConfig {\n\treturn gocql.ClusterConfig{\n\t\tHosts:                 \"testHost\",\n\t\tPort:                  1234,\n\t\tUser:                  \"testUser\",\n\t\tPassword:              \"testPassword\",\n\t\tAllowedAuthenticators: []string{\"testAuthenticator\"},\n\t\tKeyspace:              \"testKeyspace\",\n\t\tTLS:                   &config.TLS{Enabled: true},\n\t\tTimeout:               111 * time.Second,\n\t\tConnectTimeout:        222 * time.Second,\n\t\tProtoVersion:          444,\n\t\tConsistency:           gocql.One,\n\t}\n}\n"
  },
  {
    "path": "tools/cassandra/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/schema/cassandra\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\nconst defaultNumReplicas = 1\n\n// SetupSchemaConfig contains the configuration params needed to setup schema tables\ntype SetupSchemaConfig struct {\n\tCQLClientConfig\n\tschema.SetupConfig\n}\n\n// VerifyCompatibleVersion ensures that the installed version of cadence and visibility keyspaces\n// is greater than or equal to the expected version.\n// In most cases, the versions should match. However if after a schema upgrade there is a code\n// rollback, the code version (expected version) would fall lower than the actual version in\n// cassandra.\nfunc VerifyCompatibleVersion(\n\tcfg config.Persistence, expectedConsistency gocql.Consistency,\n) error {\n\n\tif ds, ok := cfg.DataStores[cfg.DefaultStore]; ok {\n\t\tif err := verifyCompatibleVersion(ds, cassandra.Version, expectedConsistency); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif ds, ok := cfg.DataStores[cfg.VisibilityStore]; ok {\n\t\tif err := verifyCompatibleVersion(ds, cassandra.VisibilityVersion, expectedConsistency); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\treturn nil\n}\n\nfunc verifyCompatibleVersion(\n\tds config.DataStore,\n\texpectedCassandraVersion string, expectedConsistency gocql.Consistency,\n) error {\n\tif ds.NoSQL != nil {\n\t\treturn verifyPluginVersion(ds.NoSQL, expectedCassandraVersion, expectedConsistency)\n\t}\n\tif ds.ShardedNoSQL != nil {\n\t\tfor shardName, connection := range ds.ShardedNoSQL.Connections {\n\t\t\terr := verifyPluginVersion(connection.NoSQLPlugin, expectedCassandraVersion, expectedConsistency)\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"Failed to verify version for DB shard: %v. Error: %v\", shardName, err.Error())\n\t\t\t}\n\t\t}\n\t}\n\n\t// not using nosql\n\treturn nil\n}\n\nfunc verifyPluginVersion(plugin *config.NoSQL, expectedCassandraVersion string, expectedConsistency gocql.Consistency) error {\n\t// Use hardcoded instead of constant because of cycle dependency issue.\n\t// However, this file will be refactor to support NoSQL soon. After the refactoring, cycle dependency issue\n\t// should be gone and we can use constant at that time\n\tif plugin.PluginName != \"cassandra\" {\n\t\treturn fmt.Errorf(\"unknown NoSQL plugin name: %q\", plugin.PluginName)\n\t}\n\n\treturn CheckCompatibleVersion(*plugin, expectedCassandraVersion, expectedConsistency)\n}\n\n// CheckCompatibleVersion check the version compatibility\nfunc CheckCompatibleVersion(\n\tcfg config.Cassandra,\n\texpectedVersion string,\n\texpectedConsistency gocql.Consistency,\n) error {\n\n\tclient, err := NewCQLClient(&CQLClientConfig{\n\t\tHosts:                 cfg.Hosts,\n\t\tPort:                  cfg.Port,\n\t\tUser:                  cfg.User,\n\t\tPassword:              cfg.Password,\n\t\tKeyspace:              cfg.Keyspace,\n\t\tAllowedAuthenticators: cfg.AllowedAuthenticators,\n\t\tTimeout:               DefaultTimeout,\n\t\tConnectTimeout:        DefaultConnectTimeout,\n\t\tTLS:                   cfg.TLS,\n\t\tProtoVersion:          cfg.ProtoVersion,\n\t}, expectedConsistency)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"creating CQL client: %w\", err)\n\t}\n\tdefer client.Close()\n\n\treturn schema.VerifyCompatibleVersion(client, cfg.Keyspace, expectedVersion)\n}\n\n// setupSchema executes the setupSchemaTask\n// using the given command line arguments\n// as input\nfunc setupSchema(cli *cli.Context) error {\n\tconfig, err := newCQLClientConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tclient, err := NewCQLClient(config, gocql.All)\n\tif err != nil {\n\t\treturn handleErr(err)\n\t}\n\tdefer client.Close()\n\tif err := schema.Setup(cli, client); err != nil {\n\t\treturn handleErr(err)\n\t}\n\treturn nil\n}\n\n// updateSchema executes the updateSchemaTask\n// using the given command lien args as input\nfunc updateSchema(cli *cli.Context) error {\n\tconfig, err := newCQLClientConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tclient, err := NewCQLClient(config, gocql.All)\n\tif err != nil {\n\t\treturn handleErr(err)\n\t}\n\tdefer client.Close()\n\tif err := schema.Update(cli, client); err != nil {\n\t\treturn handleErr(err)\n\t}\n\treturn nil\n}\n\n// createKeyspace creates a cassandra Keyspace\nfunc createKeyspace(cli *cli.Context) error {\n\tconfig, err := newCQLClientConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tkeyspace := cli.String(schema.CLIOptKeyspace)\n\tif keyspace == \"\" {\n\t\treturn handleErr(schema.NewConfigError(\"missing \" + flag(schema.CLIOptKeyspace) + \" argument \"))\n\t}\n\tdatacenter := cli.String(schema.CLIOptDatacenter)\n\terr = doCreateKeyspace(*config, keyspace, datacenter)\n\tif err != nil {\n\t\treturn handleErr(fmt.Errorf(\"error creating Keyspace:%v\", err))\n\t}\n\treturn nil\n}\n\nfunc doCreateKeyspace(cfg CQLClientConfig, name string, datacenter string) error {\n\tcfg.Keyspace = SystemKeyspace\n\tclient, err := NewCQLClient(&cfg, gocql.All)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer client.Close()\n\tif datacenter != \"\" {\n\t\treturn client.CreateNTSKeyspace(name, datacenter)\n\t}\n\treturn client.CreateKeyspace(name)\n}\n\nfunc newCQLClientConfig(cli *cli.Context) (*CQLClientConfig, error) {\n\tcqlConfig := new(CQLClientConfig)\n\tcqlConfig.Hosts = cli.String(schema.CLIOptEndpoint)\n\tcqlConfig.Port = cli.Int(schema.CLIOptPort)\n\tcqlConfig.User = cli.String(schema.CLIOptUser)\n\tcqlConfig.Password = cli.String(schema.CLIOptPassword)\n\tcqlConfig.AllowedAuthenticators = cli.StringSlice(schema.CLIOptAllowedAuthenticators)\n\tcqlConfig.Timeout = cli.Int(schema.CLIOptTimeout)\n\tcqlConfig.ConnectTimeout = cli.Int(schema.CLIOptConnectTimeout)\n\tcqlConfig.Keyspace = cli.String(schema.CLIOptKeyspace)\n\tcqlConfig.NumReplicas = cli.Int(schema.CLIOptReplicationFactor)\n\tcqlConfig.ProtoVersion = cli.Int(schema.CLIOptProtoVersion)\n\n\tif cli.Bool(schema.CLIFlagEnableTLS) {\n\t\tcqlConfig.TLS = &config.TLS{\n\t\t\tEnabled:                true,\n\t\t\tCertFile:               cli.String(schema.CLIFlagTLSCertFile),\n\t\t\tKeyFile:                cli.String(schema.CLIFlagTLSKeyFile),\n\t\t\tCaFile:                 cli.String(schema.CLIFlagTLSCaFile),\n\t\t\tEnableHostVerification: cli.Bool(schema.CLIFlagTLSEnableHostVerification),\n\t\t\tServerName:             cli.String(schema.CLIFlagTLSServerName),\n\t\t}\n\t}\n\n\tif err := validateCQLClientConfig(cqlConfig); err != nil {\n\t\treturn nil, err\n\t}\n\treturn cqlConfig, nil\n}\n\nfunc validateCQLClientConfig(config *CQLClientConfig) error {\n\tif len(config.Hosts) == 0 {\n\t\treturn schema.NewConfigError(\"missing cassandra endpoint argument \" + flag(schema.CLIOptEndpoint))\n\t}\n\tif config.Keyspace == \"\" {\n\t\treturn schema.NewConfigError(\"missing \" + flag(schema.CLIOptKeyspace) + \" argument \")\n\t}\n\tif config.Port == 0 {\n\t\tconfig.Port = DefaultCassandraPort\n\t}\n\tif config.NumReplicas == 0 {\n\t\tconfig.NumReplicas = defaultNumReplicas\n\t}\n\n\treturn nil\n}\n\nfunc flag(opt string) string {\n\treturn \"(-\" + opt + \")\"\n}\n\nfunc handleErr(err error) error {\n\tlog.Println(err)\n\treturn err\n}\n"
  },
  {
    "path": "tools/cassandra/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\n\t\"github.com/uber/cadence/environment\"\n)\n\ntype (\n\tHandlerTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestHandlerTestSuite(t *testing.T) {\n\tsuite.Run(t, new(HandlerTestSuite))\n}\n\nfunc (s *HandlerTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *HandlerTestSuite) TestValidateCQLClientConfig() {\n\tconfig := new(CQLClientConfig)\n\ts.NotNil(validateCQLClientConfig(config))\n\n\tconfig.Hosts = environment.GetCassandraAddress()\n\ts.NotNil(validateCQLClientConfig(config))\n\n\tconfig.Keyspace = \"foobar\"\n\ts.Nil(validateCQLClientConfig(config))\n}\n"
  },
  {
    "path": "tools/cassandra/main.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cassandra\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra/gocql\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\n// RunTool runs the cadence-cassandra-tool command line tool\nfunc RunTool(args []string) error {\n\tapp := BuildCLIOptions()\n\treturn app.Run(args) // exits on error\n}\n\n// SetupSchema setups the cassandra schema\nfunc SetupSchema(config *SetupSchemaConfig) error {\n\tif err := validateCQLClientConfig(&config.CQLClientConfig); err != nil {\n\t\treturn err\n\t}\n\tdb, err := NewCQLClient(&config.CQLClientConfig, gocql.All)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn schema.SetupFromConfig(&config.SetupConfig, db)\n}\n\n// root handler for all cli commands\nfunc cliHandler(c *cli.Context, handler func(c *cli.Context) error) error {\n\tquiet := c.Bool(schema.CLIOptQuiet)\n\terr := handler(c)\n\tif err != nil {\n\t\tif quiet { // if quiet, don't return error\n\t\t\tfmt.Println(\"fail to run tool: \", err)\n\t\t\treturn nil\n\t\t}\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc BuildCLIOptions() *cli.App {\n\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence-cassandra-tool\"\n\tapp.Usage = \"Command line tool for cadence cassandra operations\"\n\tapp.Version = \"0.0.1\"\n\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagEndpoint,\n\t\t\tAliases: []string{\"ep\"},\n\t\t\tValue:   \"127.0.0.1\",\n\t\t\tUsage:   \"hostname or ip address of cassandra host to connect to\",\n\t\t\tEnvVars: []string{\"CASSANDRA_HOST\"},\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    schema.CLIFlagPort,\n\t\t\tAliases: []string{\"p\"},\n\t\t\tValue:   DefaultCassandraPort,\n\t\t\tUsage:   \"Port of cassandra host to connect to\",\n\t\t\tEnvVars: []string{\"CASSANDRA_DB_PORT\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagUser,\n\t\t\tAliases: []string{\"u\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"User name used for authentication for connecting to cassandra host\",\n\t\t\tEnvVars: []string{\"CASSANDRA_USER\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagPassword,\n\t\t\tAliases: []string{\"pw\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"Password used for authentication for connecting to cassandra host\",\n\t\t\tEnvVars: []string{\"CASSANDRA_PASSWORD\"},\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    schema.CLIFlagAllowedAuthenticators,\n\t\t\tAliases: []string{\"aa\"},\n\t\t\tValue:   cli.NewStringSlice(\"\"),\n\t\t\tUsage:   \"Set allowed authenticators for servers with custom authenticators\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    schema.CLIFlagTimeout,\n\t\t\tAliases: []string{\"t\"},\n\t\t\tValue:   DefaultTimeout,\n\t\t\tUsage:   \"request Timeout in seconds used for cql client\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TIMEOUT\"},\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  schema.CLIOptConnectTimeout,\n\t\t\tValue: DefaultConnectTimeout,\n\t\t\tUsage: \"Connection Timeout in seconds used for cql client\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagKeyspace,\n\t\t\tAliases: []string{\"k\"},\n\t\t\tValue:   \"cadence\",\n\t\t\tUsage:   \"name of the cassandra Keyspace\",\n\t\t\tEnvVars: []string{\"CASSANDRA_KEYSPACE\"},\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagQuiet,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"Don't set exit status to 1 on error\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    schema.CLIFlagProtoVersion,\n\t\t\tAliases: []string{\"pv\"},\n\t\t\tUsage:   \"Protocol Version to connect to cassandra host\",\n\t\t\tEnvVars: []string{\"CASSANDRA_PROTO_VERSION\"},\n\t\t},\n\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagEnableTLS,\n\t\t\tUsage:   \"enable TLS\",\n\t\t\tEnvVars: []string{\"CASSANDRA_ENABLE_TLS\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSCertFile,\n\t\t\tUsage:   \"TLS cert file\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TLS_CERT\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSKeyFile,\n\t\t\tUsage:   \"TLS key file\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TLS_KEY\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSCaFile,\n\t\t\tUsage:   \"TLS CA file\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TLS_CA\"},\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagTLSEnableHostVerification,\n\t\t\tUsage:   \"TLS host verification\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TLS_VERIFY_HOST\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSServerName,\n\t\t\tUsage:   \"TLS ServerName\",\n\t\t\tEnvVars: []string{\"CASSANDRA_TLS_SERVER_NAME\"},\n\t\t},\n\t}\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:    \"setup-schema\",\n\t\t\tAliases: []string{\"setup\"},\n\t\t\tUsage:   \"setup initial version of cassandra schema\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagVersion,\n\t\t\t\t\tAliases: []string{\"v\"},\n\t\t\t\t\tUsage:   \"initial version of the schema, cannot be used with disable-versioning\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagSchemaFile,\n\t\t\t\t\tAliases: []string{\"f\"},\n\t\t\t\t\tUsage:   \"path to the .cql schema file; if un-specified, will just setup versioning tables\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    schema.CLIFlagDisableVersioning,\n\t\t\t\t\tAliases: []string{\"d\"},\n\t\t\t\t\tUsage:   \"disable setup of schema versioning\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    schema.CLIFlagOverwrite,\n\t\t\t\t\tAliases: []string{\"o\"},\n\t\t\t\t\tUsage:   \"drop all existing tables before setting up new schema\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, setupSchema)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"update-schema\",\n\t\t\tAliases: []string{\"update\"},\n\t\t\tUsage:   \"update cassandra schema to a specific version\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagTargetVersion,\n\t\t\t\t\tAliases: []string{\"v\"},\n\t\t\t\t\tUsage:   \"target version for the schema update, defaults to latest\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagSchemaDir,\n\t\t\t\t\tAliases: []string{\"d\"},\n\t\t\t\t\tUsage:   \"path to directory containing versioned schema\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  schema.CLIFlagDryrun,\n\t\t\t\t\tUsage: \"do a dryrun\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, updateSchema)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"create-Keyspace\",\n\t\t\tAliases: []string{\"create\"},\n\t\t\tUsage:   \"creates a Keyspace with simple strategy. If datacenter is provided, will use network topology strategy\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagKeyspace,\n\t\t\t\t\tAliases: []string{\"k\"},\n\t\t\t\t\tUsage:   \"name of the Keyspace\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagDatacenter,\n\t\t\t\t\tAliases: []string{\"dc\"},\n\t\t\t\t\tValue:   \"\",\n\t\t\t\t\tUsage:   \"name of the cassandra datacenter, used when creating the keyspace with network topology strategy\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    schema.CLIFlagReplicationFactor,\n\t\t\t\t\tAliases: []string{\"rf\"},\n\t\t\t\t\tValue:   1,\n\t\t\t\t\tUsage:   \"replication factor for the Keyspace\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, createKeyspace)\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n"
  },
  {
    "path": "tools/cli/README.md",
    "content": "Documentation for the Cadence command line interface is located at our [main site](https://cadenceworkflow.io/docs/cli/).\n\n### Build CLI binary locally\n\nTo build the CLI tool locally check out the version tag (e.g. `git checkout v0.21.3`) and run `make tools`. \nThis produces an executable called cadence.\n\nRun help command with a local build:\n````\n./cadence --help\n````\n\nCommand to describe a domain would look like this:\n````\n./cadence --domain samples-domain domain describe\n````"
  },
  {
    "path": "tools/cli/admin.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/service/worker/scanner/executions\"\n)\n\nfunc newAdminWorkflowCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:  \"show\",\n\t\t\tUsage: \"show workflow history from database\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t// v2 history events\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagTreeID,\n\t\t\t\t\tUsage: \"TreeID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagBranchID,\n\t\t\t\t\tUsage: \"BranchID\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:  FlagMinEventID,\n\t\t\t\t\tValue: 1,\n\t\t\t\t\tUsage: \"MinEventID\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:  FlagMaxEventID,\n\t\t\t\t\tValue: 10000,\n\t\t\t\t\tUsage: \"MaxEventID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagOutputFilename,\n\t\t\t\t\tAliases: []string{\"of\"},\n\t\t\t\t\tUsage:   \"output file\",\n\t\t\t\t},\n\t\t\t\t// support mysql query\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagShardID,\n\t\t\t\t\tAliases: []string{\"sid\"},\n\t\t\t\t\tUsage:   \"ShardID\",\n\t\t\t\t}),\n\t\t\tAction: AdminShowWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe internal information of workflow execution\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminDescribeWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"refresh-tasks\",\n\t\t\tAliases: []string{\"rt\"},\n\t\t\tUsage:   \"Refreshes all the tasks of a workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminRefreshWorkflowTasks,\n\t\t},\n\t\t{\n\t\t\tName:    \"delete\",\n\t\t\tAliases: []string{\"del\"},\n\t\t\tUsage:   \"Delete current workflow execution and the mutableState record\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagSkipErrorMode,\n\t\t\t\t\tAliases: []string{\"serr\"},\n\t\t\t\t\tUsage:   \"skip errors when deleting history\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagRemote,\n\t\t\t\t\tUsage: \"Executes deletion on server side\",\n\t\t\t\t}),\n\t\t\tAction: AdminDeleteWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"fix_corruption\",\n\t\t\tAliases: []string{\"fc\"},\n\t\t\tUsage:   \"Checks if workflow record is corrupted in database and cleans up\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagSkipErrorMode,\n\t\t\t\t\tAliases: []string{\"serr\"},\n\t\t\t\t\tUsage:   \"Skip errors and tries to delete as much as possible from the DB\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminMaintainCorruptWorkflow,\n\t\t},\n\t}\n}\n\nfunc newAdminShardManagementCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"d\"},\n\t\t\tUsage:   \"Describe shard by Id\",\n\t\t\tFlags: append(\n\t\t\t\tgetDBFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagShardID,\n\t\t\t\t\tUsage: \"The Id of the shard to describe\",\n\t\t\t\t},\n\t\t\t),\n\t\t\tAction: AdminDescribeShard,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"List shard distribution\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagPageSize,\n\t\t\t\t\tValue: 100,\n\t\t\t\t\tUsage: \"Max number of results to return\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagPageID,\n\t\t\t\t\tValue: 0,\n\t\t\t\t\tUsage: \"Option to show results offset from pagesize * page_id\",\n\t\t\t\t},\n\t\t\t\tgetFormatFlag(),\n\t\t\t},\n\t\t\tAction: AdminDescribeShardDistribution,\n\t\t},\n\t\t{\n\t\t\tName:    \"setRangeID\",\n\t\t\tAliases: []string{\"srid\"},\n\t\t\tUsage:   \"Force update shard rangeID\",\n\t\t\tFlags: append(\n\t\t\t\tgetDBFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagShardID,\n\t\t\t\t\tAliases: []string{\"sid\"},\n\t\t\t\t\tUsage:   \"ID of the shard to reset\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:    FlagRangeID,\n\t\t\t\t\tAliases: []string{\"rid\"},\n\t\t\t\t\tUsage:   \"new shard rangeID\",\n\t\t\t\t},\n\t\t\t),\n\t\t\tAction: AdminSetShardRangeID,\n\t\t},\n\t\t{\n\t\t\tName:    \"closeShard\",\n\t\t\tAliases: []string{\"clsh\"},\n\t\t\tUsage:   \"close a shard given a shard id\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagShardID,\n\t\t\t\t\tUsage: \"ShardID for the cadence cluster to manage\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminCloseShard,\n\t\t},\n\t\t{\n\t\t\tName:    \"removeTask\",\n\t\t\tAliases: []string{\"rmtk\"},\n\t\t\tUsage:   \"remove a task based on shardID, task type, taskID, and task visibility timestamp\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagShardID,\n\t\t\t\t\tUsage: \"shardID\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:  FlagTaskID,\n\t\t\t\t\tUsage: \"taskID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagTaskType,\n\t\t\t\t\tUsage: \"task type: 2 (transfer task), 3 (timer task), 4 (replication task) or 6 (cross-cluster task)\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:  FlagTaskVisibilityTimestamp,\n\t\t\t\t\tUsage: \"task visibility timestamp in nano (required for removing timer task)\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagCluster,\n\t\t\t\t\tUsage: \"target cluster of the task (required for removing cross-cluster task)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminRemoveTask,\n\t\t},\n\t\t{\n\t\t\tName:  \"timers\",\n\t\t\tUsage: \"get scheduled timers for a given time range\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagShardID,\n\t\t\t\t\tUsage: \"shardID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagPageSize,\n\t\t\t\t\tUsage: \"page size used to query db executions table\",\n\t\t\t\t\tValue: 500,\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagStartDate,\n\t\t\t\t\tUsage: \"start date\",\n\t\t\t\t\tValue: time.Now().UTC().Format(time.RFC3339),\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagEndDate,\n\t\t\t\t\tUsage: \"end date\",\n\t\t\t\t\tValue: time.Now().UTC().Add(24 * time.Hour).Format(time.RFC3339),\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDomainID,\n\t\t\t\t\tUsage: \"filter tasks by DomainID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntSliceFlag{\n\t\t\t\t\tName: FlagTimerType,\n\t\t\t\t\tUsage: \"timer types: 0 - DecisionTimeoutTask, 1 - TaskTypeActivityTimeout, \" +\n\t\t\t\t\t\t\"2 - TaskTypeUserTimer, 3 - TaskTypeWorkflowTimeout, 4 - TaskTypeDeleteHistoryEvent, \" +\n\t\t\t\t\t\t\"5 - TaskTypeActivityRetryTimer, 6 - TaskTypeWorkflowBackoffTimer\",\n\t\t\t\t\tValue: cli.NewIntSlice(-1),\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagPrintJSON,\n\t\t\t\t\tUsage: \"print raw json data instead of histogram\",\n\t\t\t\t},\n\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagSkipErrorMode,\n\t\t\t\t\tUsage: \"skip errors\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagInputFile,\n\t\t\t\t\tUsage: \"file to use, will not connect to persistence\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDateFormat,\n\t\t\t\t\tUsage: \"create buckets using time format. Use Go reference time: Mon Jan 2 15:04:05 MST 2006. If set, --\" + FlagBucketSize + \" is ignored\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagBucketSize,\n\t\t\t\t\tValue: \"hour\",\n\t\t\t\t\tUsage: \"group timers by time bucket. Available: day, hour, minute, second\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagShardMultiplier,\n\t\t\t\t\tUsage: \"multiply timer counters for histogram\",\n\t\t\t\t\tValue: 16384,\n\t\t\t\t},\n\t\t\t),\n\t\t\tAction: AdminTimers,\n\t\t},\n\t}\n}\n\nfunc newAdminHistoryHostCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe internal information of history host\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagHistoryAddress,\n\t\t\t\t\tAliases: []string{\"had\"},\n\t\t\t\t\tUsage:   \"History Host address(IP:PORT)\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagShardID,\n\t\t\t\t\tAliases: []string{\"sid\"},\n\t\t\t\t\tUsage:   \"ShardID\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagPrintFullyDetail,\n\t\t\t\t\tAliases: []string{\"pf\"},\n\t\t\t\t\tUsage:   \"Print fully detail\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminDescribeHistoryHost,\n\t\t},\n\t\t{\n\t\t\tName:    \"getshard\",\n\t\t\tAliases: []string{\"gsh\"},\n\t\t\tUsage:   \"Get shardID for a workflowID\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagNumberOfShards,\n\t\t\t\t\tUsage: \"NumberOfShards for the cadence cluster(see config for numHistoryShards)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminGetShardID,\n\t\t},\n\t}\n}\n\n// may be better to do this inside an \"ensure setup\" in each method,\n// but this reduces branches for testing.\nfunc withDomainClient(c *cli.Context, admin bool, cb func(dc *domainCLIImpl) error) error {\n\tdc, err := newDomainCLI(c, admin)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn cb(dc)\n}\n\nfunc newAdminDomainCommands() []*cli.Command {\n\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"register\",\n\t\t\tAliases: []string{\"re\"},\n\t\t\tUsage:   \"Register workflow domain\",\n\t\t\tFlags:   adminRegisterDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, true, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.RegisterDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"update\",\n\t\t\tAliases: []string{\"up\", \"u\"},\n\t\t\tUsage:   \"Update existing workflow domain\",\n\t\t\tFlags:   adminUpdateDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, true, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.UpdateDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"delete\",\n\t\t\tAliases: []string{\"del\"},\n\t\t\tUsage:   \"Delete existing workflow domain\",\n\t\t\tFlags:   adminDeleteDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, true, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DeleteDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"deprecate\",\n\t\t\tAliases: []string{\"dep\"},\n\t\t\tUsage:   \"Deprecate existing workflow domain\",\n\t\t\tFlags:   adminDeprecateDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, true, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DeprecateDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe existing workflow domain\",\n\t\t\tFlags:   adminDescribeDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, true, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DescribeDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"getdomainidorname\",\n\t\t\tAliases: []string{\"getdn\"},\n\t\t\tUsage:   \"Get domainID or domainName\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDomainID,\n\t\t\t\t\tUsage: \"Domain ID(uuid)\",\n\t\t\t\t}),\n\t\t\tAction: AdminGetDomainIDOrName,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"List all domains in the cluster\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagPageSize,\n\t\t\t\t\tAliases: []string{\"ps\"},\n\t\t\t\t\tValue:   10,\n\t\t\t\t\tUsage:   \"Result page size\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagAll,\n\t\t\t\t\tAliases: []string{\"a\"},\n\t\t\t\t\tUsage:   \"List all domains, by default only domains in REGISTERED status are listed\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagDeprecated,\n\t\t\t\t\tAliases: []string{\"dep\"},\n\t\t\t\t\tUsage:   \"List deprecated domains only, by default only domains in REGISTERED status are listed\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagPrefix,\n\t\t\t\t\tUsage: \"List domains that are matching to the given prefix\",\n\t\t\t\t\tValue: \"\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagPrintFullyDetail,\n\t\t\t\t\tAliases: []string{\"pf\"},\n\t\t\t\t\tUsage:   \"Print full domain detail\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagPrintJSON,\n\t\t\t\t\tAliases: []string{\"pjson\"},\n\t\t\t\t\tUsage:   \"Print in raw json format (DEPRECATED: instead use --format json)\",\n\t\t\t\t},\n\t\t\t\tgetFormatFlag(),\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.ListDomains(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc newAdminKafkaCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\t// TODO: do we still need this command given that kafka replication has been deprecated?\n\t\t\tName:    \"parse\",\n\t\t\tAliases: []string{\"par\"},\n\t\t\tUsage:   \"Parse replication tasks from kafka messages\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage:   \"Input file to use, if not present assumes piping\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagOutputFilename,\n\t\t\t\t\tAliases: []string{\"of\"},\n\t\t\t\t\tUsage:   \"Output file to write to, if not provided output is written to stdout\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagSkipErrorMode,\n\t\t\t\t\tAliases: []string{\"serr\"},\n\t\t\t\t\tUsage:   \"Skip errors in parsing messages\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagHeadersMode,\n\t\t\t\t\tAliases: []string{\"he\"},\n\t\t\t\t\tUsage:   \"Output headers of messages in format: DomainID, WorkflowID, RunID, FirstEventID, NextEventID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagMessageType,\n\t\t\t\t\tAliases: []string{\"mt\"},\n\t\t\t\t\tUsage:   \"Kafka message type (0: replicationTasks; 1: visibility)\",\n\t\t\t\t\tValue:   0,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminKafkaParse,\n\t\t},\n\t\t{\n\t\t\t// TODO: move this command be a subcommand of admin workflow\n\t\t\tName:    \"rereplicate\",\n\t\t\tAliases: []string{\"rrp\"},\n\t\t\tUsage:   \"Rereplicate replication tasks from history tables\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagSourceCluster,\n\t\t\t\t\tUsage: \"Name of source cluster to resend the replication task\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDomainID,\n\t\t\t\t\tUsage: \"DomainID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.Int64Flag{\n\t\t\t\t\tName:  FlagMaxEventID,\n\t\t\t\t\tUsage: \"MaxEventID Optional, default to all events\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagEndEventVersion,\n\t\t\t\t\tUsage: \"Workflow end event version, required if MaxEventID is specified\",\n\t\t\t\t}},\n\t\t\tAction: AdminRereplicate,\n\t\t},\n\t}\n}\n\nfunc newAdminElasticSearchCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"catIndex\",\n\t\t\tAliases: []string{\"cind\"},\n\t\t\tUsage:   \"Cat Indices on ElasticSearch\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagURL,\n\t\t\t\t\tUsage: \"URL of ElasticSearch cluster\",\n\t\t\t\t},\n\t\t\t\tgetFormatFlag(),\n\t\t\t},\n\t\t\tAction: AdminCatIndices,\n\t\t},\n\t\t{\n\t\t\tName:    \"index\",\n\t\t\tAliases: []string{\"ind\"},\n\t\t\tUsage:   \"Index docs on ElasticSearch\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagURL,\n\t\t\t\t\tUsage: \"URL of ElasticSearch cluster\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagIndex,\n\t\t\t\t\tUsage: \"ElasticSearch target index\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage:   \"Input file of indexer.Message in json format, separated by newline\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagBatchSize,\n\t\t\t\t\tAliases: []string{\"bs\"},\n\t\t\t\t\tUsage:   \"Optional batch size of actions for bulk operations\",\n\t\t\t\t\tValue:   1000,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminIndex,\n\t\t},\n\t\t{\n\t\t\tName:    \"delete\",\n\t\t\tAliases: []string{\"del\"},\n\t\t\tUsage:   \"Delete docs on ElasticSearch\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagURL,\n\t\t\t\t\tUsage: \"URL of ElasticSearch cluster\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagIndex,\n\t\t\t\t\tUsage: \"ElasticSearch target index\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage: \"Input file name. Redirect cadence wf list result (with tale format) to a file and use as delete input. \" +\n\t\t\t\t\t\t\"First line should be table header like WORKFLOW TYPE | WORKFLOW ID | RUN ID | ...\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagBatchSize,\n\t\t\t\t\tAliases: []string{\"bs\"},\n\t\t\t\t\tUsage:   \"Optional batch size of actions for bulk operations\",\n\t\t\t\t\tValue:   1000,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagRPS,\n\t\t\t\t\tUsage: \"Optional batch request rate per second\",\n\t\t\t\t\tValue: 30,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminDelete,\n\t\t},\n\t\t{\n\t\t\tName:    \"report\",\n\t\t\tAliases: []string{\"rep\"},\n\t\t\tUsage:   \"Generate Report by Aggregation functions on ElasticSearch\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagURL,\n\t\t\t\t\tUsage: \"URL of ElasticSearch cluster\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagIndex,\n\t\t\t\t\tUsage: \"ElasticSearch target index\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagListQuery,\n\t\t\t\t\tUsage: \"SQL query of the report\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagOutputFormat,\n\t\t\t\t\tUsage: \"Additional output format (html or csv)\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagOutputFilename,\n\t\t\t\t\tUsage: \"Additional output filename with path\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: GenerateReport,\n\t\t},\n\t}\n}\n\nfunc newAdminTaskListCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe pollers and status information of tasklist\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskList,\n\t\t\t\t\tAliases: []string{\"tl\"},\n\t\t\t\t\tUsage:   \"TaskList description\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskListType,\n\t\t\t\t\tAliases: []string{\"tlt\"},\n\t\t\t\t\tValue:   \"decision\",\n\t\t\t\t\tUsage:   \"Optional TaskList type [decision|activity]\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminDescribeTaskList,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"List active tasklist under a domain\",\n\t\t\tAction:  AdminListTaskList,\n\t\t},\n\t\t{\n\t\t\tName:    \"update-partition\",\n\t\t\tAliases: []string{\"up\"},\n\t\t\tUsage:   \"Update tasklist's partition config\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskList,\n\t\t\t\t\tAliases: []string{\"tl\"},\n\t\t\t\t\tUsage:   \"TaskList Name\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskListType,\n\t\t\t\t\tAliases: []string{\"tlt\"},\n\t\t\t\t\tUsage:   \"TaskList type [decision|activity]\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagNumReadPartitions,\n\t\t\t\t\tAliases: []string{\"nrp\"},\n\t\t\t\t\tUsage:   \"Number of read partitions\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagNumWritePartitions,\n\t\t\t\t\tAliases: []string{\"nwp\"},\n\t\t\t\t\tUsage:   \"Number of write partitions\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagForce,\n\t\t\t\t\tAliases: []string{\"f\"},\n\t\t\t\t\tUsage:   \"Force an update operation that may be unsafe\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminUpdateTaskListPartitionConfig,\n\t\t},\n\t}\n}\n\nfunc newAdminClusterCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"add-search-attr\",\n\t\t\tAliases: []string{\"asa\"},\n\t\t\tUsage:   \"whitelist search attribute\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagSearchAttributesKey,\n\t\t\t\t\tUsage: \"Search Attribute key to be whitelisted\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagSearchAttributesType,\n\t\t\t\t\tValue: -1,\n\t\t\t\t\tUsage: \"Search Attribute value type. [0:String, 1:Keyword, 2:Int, 3:Double, 4:Bool, 5:Datetime]\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagSecurityToken,\n\t\t\t\t\tAliases: []string{\"st\"},\n\t\t\t\t\tUsage:   \"Optional token for security check\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminAddSearchAttribute,\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"d\"},\n\t\t\tUsage:   \"Describe cluster information\",\n\t\t\tAction:  AdminDescribeCluster,\n\t\t},\n\t\t{\n\t\t\tName:        \"failover\",\n\t\t\tAliases:     []string{\"fo\"},\n\t\t\tUsage:       \"Failover domains with domain data IsManagedByCadence=true to target cluster\",\n\t\t\tSubcommands: newAdminFailoverCommands(),\n\t\t},\n\t\t{\n\t\t\tName:    \"failover_fast\",\n\t\t\tAliases: []string{\"fof\"},\n\t\t\tUsage:   \"Failover domains with domain data IsManagedByCadence=true to target cluster\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTargetCluster,\n\t\t\t\t\tAliases: []string{\"tc\"},\n\t\t\t\t\tUsage:   \"Target active cluster name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.FailoverDomains(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:        \"rebalance\",\n\t\t\tAliases:     []string{\"rb\"},\n\t\t\tUsage:       \"Rebalance the domains active cluster\",\n\t\t\tSubcommands: newAdminRebalanceCommands(),\n\t\t},\n\t}\n}\n\nfunc getDLQFlags() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagShards,\n\t\t\tUsage: \"Comma separated shard IDs or inclusive ranges. Example: \\\"2,5-6,10\\\".  Alternatively, feed one shard ID per line via STDIN.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagDLQType,\n\t\t\tAliases: []string{\"dt\"},\n\t\t\tUsage:   \"Type of DLQ to manage. (Options: domain, history)\",\n\t\t\tValue:   \"history\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagSourceCluster,\n\t\t\tUsage: \"The cluster where the task is generated\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagLastMessageID,\n\t\t\tAliases: []string{\"lm\"},\n\t\t\tUsage:   \"The upper boundary of the read message\",\n\t\t},\n\t}\n}\n\nfunc newAdminDLQCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"count\",\n\t\t\tAliases: []string{\"c\"},\n\t\t\tUsage:   \"Count DLQ Messages\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\tgetFormatFlag(),\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagDLQType,\n\t\t\t\t\tAliases: []string{\"dt\"},\n\t\t\t\t\tUsage:   \"Type of DLQ to manage. (Options: domain, history)\",\n\t\t\t\t\tValue:   \"history\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagForce,\n\t\t\t\t\tUsage: \"Force fetch latest counts (will put additional stress on DB)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminCountDLQMessages,\n\t\t},\n\t\t{\n\t\t\tName:    \"read\",\n\t\t\tAliases: []string{\"r\"},\n\t\t\tUsage:   \"Read DLQ Messages\",\n\t\t\tFlags: append(getDLQFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagMaxMessageCount,\n\t\t\t\t\tAliases: []string{\"mmc\"},\n\t\t\t\t\tUsage:   \"Max message size to fetch\",\n\t\t\t\t},\n\t\t\t\tgetFormatFlag(),\n\t\t\t),\n\t\t\tAction: AdminGetDLQMessages,\n\t\t},\n\t\t{\n\t\t\tName:    \"purge\",\n\t\t\tAliases: []string{\"p\"},\n\t\t\tUsage:   \"Delete DLQ messages with equal or smaller ids than the provided task id\",\n\t\t\tFlags:   getDLQFlags(),\n\t\t\tAction:  AdminPurgeDLQMessages,\n\t\t},\n\t\t{\n\t\t\tName:    \"merge\",\n\t\t\tAliases: []string{\"m\"},\n\t\t\tUsage:   \"Merge DLQ messages with equal or smaller ids than the provided task id\",\n\t\t\tFlags:   getDLQFlags(),\n\t\t\tAction:  AdminMergeDLQMessages,\n\t\t},\n\t}\n}\n\nfunc newAdminQueueCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:   \"reset\",\n\t\t\tUsage:  \"reset processing queue states for transfer or timer queue processor\",\n\t\t\tFlags:  getQueueCommandFlags(),\n\t\t\tAction: AdminResetQueue,\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"describe processing queue states for transfer or timer queue processor\",\n\t\t\tFlags:   getQueueCommandFlags(),\n\t\t\tAction:  AdminDescribeQueue,\n\t\t},\n\t}\n}\n\nfunc newAdminAsyncQueueCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:   \"get\",\n\t\t\tUsage:  \"get async workflow queue configuration of a domain\",\n\t\t\tAction: AdminGetAsyncWFConfig,\n\t\t},\n\t\t{\n\t\t\tName:  \"update\",\n\t\t\tUsage: \"upsert async workflow queue configuration of a domain\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagJSON,\n\t\t\t\t\tUsage:    `AsyncWorkflowConfiguration in json format. Schema can be found in https://github.com/uber/cadence/blob/master/common/types/admin.go`,\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminUpdateAsyncWFConfig,\n\t\t},\n\t}\n}\n\nfunc newDBCommands() []*cli.Command {\n\tvar collections cli.StringSlice = *cli.NewStringSlice(invariant.CollectionStrings()...)\n\n\tscanFlag := &cli.StringFlag{\n\t\tName:     FlagScanType,\n\t\tUsage:    \"Scan type to use: \" + strings.Join(executions.ScanTypeStrings(), \", \"),\n\t\tRequired: true,\n\t}\n\n\tcollectionsFlag := &cli.StringSliceFlag{\n\t\tName:  FlagInvariantCollection,\n\t\tUsage: \"Scan collection type to use: \" + strings.Join(collections.Value(), \", \"),\n\t\tValue: &collections,\n\t}\n\n\tverboseFlag := &cli.BoolFlag{\n\t\tName:     FlagVerbose,\n\t\tUsage:    \"verbose output\",\n\t\tRequired: false,\n\t}\n\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:  \"scan\",\n\t\t\tUsage: \"scan executions in database and detect corruptions\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:     FlagNumberOfShards,\n\t\t\t\t\tUsage:    \"NumberOfShards for the cadence cluster (see config for numHistoryShards)\",\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t\tscanFlag,\n\t\t\t\tcollectionsFlag,\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage:   \"Input file of executions to scan in JSON format {\\\"DomainID\\\":\\\"x\\\",\\\"WorkflowID\\\":\\\"x\\\",\\\"RunID\\\":\\\"x\\\"} separated by a newline\",\n\t\t\t\t},\n\t\t\t\tverboseFlag,\n\t\t\t),\n\n\t\t\tAction: AdminDBScan,\n\t\t},\n\t\t{\n\t\t\tName:  \"unsupported-workflow\",\n\t\t\tUsage: \"use this command when upgrade the Cadence server from version less than 0.16.0. This scan database and detect unsupported workflow type.\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagRPS,\n\t\t\t\t\tUsage: \"NumberOfShards for the cadence cluster (see config for numHistoryShards)\",\n\t\t\t\t\tValue: 1000,\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagOutputFilename,\n\t\t\t\t\tAliases: []string{\"of\"},\n\t\t\t\t\tUsage:   \"Output file to write to, if not provided output is written to stdout\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:     FlagLowerShardBound,\n\t\t\t\t\tUsage:    \"FlagLowerShardBound for the start shard to scan. (Default: 0)\",\n\t\t\t\t\tValue:    0,\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:     FlagUpperShardBound,\n\t\t\t\t\tUsage:    \"FlagLowerShardBound for the end shard to scan. (Default: 16383)\",\n\t\t\t\t\tValue:    16383,\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t),\n\n\t\t\tAction: AdminDBScanUnsupportedWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:  \"clean\",\n\t\t\tUsage: \"clean up corrupted workflows\",\n\t\t\tFlags: append(getDBFlags(),\n\t\t\t\tscanFlag,\n\t\t\t\tcollectionsFlag,\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage:   \"Input file of execution to clean in JSON format. Use `scan` command to generate list of executions.\",\n\t\t\t\t},\n\t\t\t\tverboseFlag,\n\t\t\t),\n\t\t\tAction: AdminDBClean,\n\t\t},\n\t\t{\n\t\t\tName:  \"decode_thrift\",\n\t\t\tUsage: \"decode thrift object, print into JSON if the data is matching with any supported struct\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInput,\n\t\t\t\t\tAliases: []string{\"i\"},\n\t\t\t\t\tEnvVars: []string{\"Input\"},\n\t\t\t\t\tUsage:   \"Input of Thrift encoded data structure.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputEncoding,\n\t\t\t\t\tAliases: []string{\"enc\"},\n\t\t\t\t\tUsage:   \"Encoding of the input: [hex|base64] (Default: hex)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminDBDataDecodeThrift,\n\t\t},\n\t}\n}\n\nfunc getQueueCommandFlags() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagShardID,\n\t\t\tAliases: []string{\"sid\"},\n\t\t\tUsage:   \"shardID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagCluster,\n\t\t\tUsage: \"cluster the task processor is responsible for\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagQueueType,\n\t\t\tUsage: \"queue type: 2 (transfer queue), 3 (timer queue) or 6 (cross-cluster queue)\",\n\t\t},\n\t}\n}\n\nfunc newAdminFailoverCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"start\",\n\t\t\tAliases: []string{\"s\"},\n\t\t\tUsage:   \"start failover workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTargetCluster,\n\t\t\t\t\tAliases: []string{\"tc\"},\n\t\t\t\t\tUsage:   \"Target cluster name\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagSourceCluster,\n\t\t\t\t\tAliases: []string{\"sc\"},\n\t\t\t\t\tUsage:   \"Source cluster name\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverTimeout,\n\t\t\t\t\tAliases: []string{\"fts\"},\n\t\t\t\t\tUsage:   \"Optional graceful failover timeout in seconds. If this field is define, the failover will use graceful failover.\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagExecutionTimeout,\n\t\t\t\t\tAliases: []string{\"et\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow timeout in seconds\",\n\t\t\t\t\tValue:   defaultFailoverWorkflowTimeoutInSeconds,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverWaitTime,\n\t\t\t\t\tAliases: []string{\"fwts\"},\n\t\t\t\t\tUsage:   \"Optional Failover wait time after each batch in seconds\",\n\t\t\t\t\tValue:   defaultBatchFailoverWaitTimeInSeconds,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverBatchSize,\n\t\t\t\t\tAliases: []string{\"fbs\"},\n\t\t\t\t\tUsage:   \"Optional number of domains to failover in one batch\",\n\t\t\t\t\tValue:   defaultBatchFailoverSize,\n\t\t\t\t},\n\t\t\t\t&cli.StringSliceFlag{\n\t\t\t\t\tName: FlagFailoverDomains,\n\t\t\t\t\tUsage: \"Optional domains to failover, eg d1,d2..,dn. \" +\n\t\t\t\t\t\t\"Only provided domains in source cluster will be failover.\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverDrillWaitTime,\n\t\t\t\t\tAliases: []string{\"fdws\"},\n\t\t\t\t\tUsage: \"Optional failover drill wait time. \" +\n\t\t\t\t\t\t\"After the wait time, the domains will be reset to original regions.\" +\n\t\t\t\t\t\t\"This field is required if the cron schedule is specified.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName: FlagCronSchedule,\n\t\t\t\t\tUsage: \"Optional cron schedule on failover drill. Please specify failover drill wait time \" +\n\t\t\t\t\t\t\"if this field is specific\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverStart,\n\t\t},\n\t\t{\n\t\t\tName:    \"pause\",\n\t\t\tAliases: []string{\"p\"},\n\t\t\tUsage:   \"pause failover workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow runID, default is latest runID\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagFailoverDrill,\n\t\t\t\t\tAliases: []string{\"fd\"},\n\t\t\t\t\tUsage: \"Optional to pause failover workflow or failover drill workflow.\" +\n\t\t\t\t\t\t\" The default is normal failover workflow\",\n\t\t\t\t},\n\t\t\t},\n\n\t\t\tAction: AdminFailoverPause,\n\t\t},\n\t\t{\n\t\t\tName:    \"resume\",\n\t\t\tAliases: []string{\"re\"},\n\t\t\tUsage:   \"resume paused failover workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow runID, default is latest runID\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagFailoverDrill,\n\t\t\t\t\tAliases: []string{\"fd\"},\n\t\t\t\t\tUsage: \"Optional to resume failover workflow or failover drill workflow.\" +\n\t\t\t\t\t\t\" The default is normal failover workflow\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverResume,\n\t\t},\n\t\t{\n\t\t\tName:    \"query\",\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"query failover workflow state\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagFailoverDrill,\n\t\t\t\t\tAliases: []string{\"fd\"},\n\t\t\t\t\tUsage: \"Optional to query failover workflow or failover drill workflow.\" +\n\t\t\t\t\t\t\" The default is normal failover workflow\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow runID, default is latest runID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverQuery,\n\t\t},\n\t\t{\n\t\t\tName:    \"abort\",\n\t\t\tAliases: []string{\"a\"},\n\t\t\tUsage:   \"abort failover workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow runID, default is latest runID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagReason,\n\t\t\t\t\tAliases: []string{\"re\"},\n\t\t\t\t\tUsage:   \"Optional reason why abort\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagFailoverDrill,\n\t\t\t\t\tAliases: []string{\"fd\"},\n\t\t\t\t\tUsage: \"Optional to abort failover workflow or failover drill workflow.\" +\n\t\t\t\t\t\t\" The default is normal failover workflow\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverAbort,\n\t\t},\n\t\t{\n\t\t\tName:    \"rollback\",\n\t\t\tAliases: []string{\"ro\"},\n\t\t\tUsage:   \"rollback failover workflow\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow runID, default is latest runID\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverTimeout,\n\t\t\t\t\tAliases: []string{\"fts\"},\n\t\t\t\t\tUsage:   \"Optional graceful failover timeout in seconds. If this field is define, the failover will use graceful failover.\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagExecutionTimeout,\n\t\t\t\t\tAliases: []string{\"et\"},\n\t\t\t\t\tUsage:   \"Optional Failover workflow timeout in seconds\",\n\t\t\t\t\tValue:   defaultFailoverWorkflowTimeoutInSeconds,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverWaitTime,\n\t\t\t\t\tAliases: []string{\"fwts\"},\n\t\t\t\t\tUsage:   \"Optional Failover wait time after each batch in seconds\",\n\t\t\t\t\tValue:   defaultBatchFailoverWaitTimeInSeconds,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagFailoverBatchSize,\n\t\t\t\t\tAliases: []string{\"fbs\"},\n\t\t\t\t\tUsage:   \"Optional number of domains to failover in one batch\",\n\t\t\t\t\tValue:   defaultBatchFailoverSize,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverRollback,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"list failover workflow runs closed/open. This is just a simplified list cmd\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagOpen,\n\t\t\t\t\tAliases: []string{\"op\"},\n\t\t\t\t\tUsage:   \"List for open workflow executions, default is to list for closed ones\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagPageSize,\n\t\t\t\t\tAliases: []string{\"ps\"},\n\t\t\t\t\tValue:   10,\n\t\t\t\t\tUsage:   \"Result page size\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\t\t\tUsage:   \"Ignore this. It is a dummy flag which will be forced overwrite\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagFailoverDrill,\n\t\t\t\t\tAliases: []string{\"fd\"},\n\t\t\t\t\tUsage: \"Optional to query failover workflow or failover drill workflow.\" +\n\t\t\t\t\t\t\" The default is normal failover workflow\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminFailoverList,\n\t\t},\n\t}\n}\n\nfunc newAdminRebalanceCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"start\",\n\t\t\tAliases: []string{\"s\"},\n\t\t\tUsage:   \"start rebalance workflow\",\n\t\t\tFlags:   []cli.Flag{},\n\t\t\tAction:  AdminRebalanceStart,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"list rebalance workflow runs closed/open.\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    FlagOpen,\n\t\t\t\t\tAliases: []string{\"op\"},\n\t\t\t\t\tUsage:   \"List for open workflow executions, default is to list for closed ones\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagPageSize,\n\t\t\t\t\tAliases: []string{\"ps\"},\n\t\t\t\t\tValue:   10,\n\t\t\t\t\tUsage:   \"Result page size\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminRebalanceList,\n\t\t},\n\t}\n}\n\nfunc newAdminConfigStoreCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"get\",\n\t\t\tAliases: []string{\"g\"},\n\t\t\tUsage:   \"Get Dynamic Config Value\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagDynamicConfigName,\n\t\t\t\t\tUsage:    \"Name of Dynamic Config parameter to get value of\",\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDynamicConfigFilter,\n\t\t\t\t\tUsage: fmt.Sprintf(`Optional. ex: --%s '{\"domainName\":\"global-samples-domain\", \"shardID\":1, \"isEnabled\": true}'`, FlagDynamicConfigFilter),\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminGetDynamicConfig,\n\t\t},\n\t\t{\n\t\t\tName:    \"update\",\n\t\t\tAliases: []string{\"u\"},\n\t\t\tUsage:   \"Update Dynamic Config Value\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagDynamicConfigName,\n\t\t\t\t\tUsage:    \"Name of Dynamic Config parameter to update value of\",\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t\t&cli.StringSliceFlag{\n\t\t\t\t\tName:     FlagDynamicConfigValue,\n\t\t\t\t\tUsage:    fmt.Sprintf(`Can be specified multiple times for multiple values. ex: --%s '{\"Value\":true,\"Filters\":[]}'`, FlagDynamicConfigValue),\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminUpdateDynamicConfig,\n\t\t},\n\t\t{\n\t\t\tName:    \"restore\",\n\t\t\tAliases: []string{\"r\"},\n\t\t\tUsage:   \"Restore Dynamic Config Value\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagDynamicConfigName,\n\t\t\t\t\tUsage:    \"Name of Dynamic Config parameter to restore\",\n\t\t\t\t\tRequired: true,\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDynamicConfigFilter,\n\t\t\t\t\tUsage: fmt.Sprintf(`Optional. ex: --%s '{\"domainName\":\"global-samples-domain\", \"shardID\":1, \"isEnabled\": true}'`, FlagDynamicConfigFilter),\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminRestoreDynamicConfig,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"List Dynamic Config Value\",\n\t\t\tFlags:   []cli.Flag{},\n\t\t\tAction:  AdminListDynamicConfig,\n\t\t},\n\t\t{\n\t\t\tName:    \"listall\",\n\t\t\tAliases: []string{\"la\"},\n\t\t\tUsage:   \"List all available configuration keys\",\n\t\t\tFlags:   []cli.Flag{getFormatFlag()},\n\t\t\tAction:  AdminListConfigKeys,\n\t\t},\n\t}\n}\n\nfunc newAdminIsolationGroupCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:  \"get-global\",\n\t\t\tUsage: \"gets the global isolation groups\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagFormat,\n\t\t\t\t\tUsage: `output format`,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminGetGlobalIsolationGroups,\n\t\t},\n\t\t{\n\t\t\tName:  \"update-global\",\n\t\t\tUsage: \"sets the global isolation groups\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagJSON,\n\t\t\t\t\tUsage:    `the configurations to upsert: eg: [{\"Name\": \"zone-1\": \"State\": 2}]. To remove groups, specify an empty configuration`,\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t\t&cli.StringSliceFlag{\n\t\t\t\t\tName:     FlagIsolationGroupSetDrains,\n\t\t\t\t\tUsage:    \"Use to upsert the configuration for all drains. Note that this is an upsert operation and will overwrite all existing configuration\",\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:     FlagIsolationGroupsRemoveAllDrains,\n\t\t\t\t\tUsage:    \"Removes all drains\",\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminUpdateGlobalIsolationGroups,\n\t\t},\n\t\t{\n\t\t\tName:  \"get-domain\",\n\t\t\tUsage: \"gets the domain isolation groups\",\n\t\t\tFlags: []cli.Flag{\n\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagFormat,\n\t\t\t\t\tUsage: `output format`,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminGetDomainIsolationGroups,\n\t\t},\n\t\t{\n\t\t\tName:  \"update-domain\",\n\t\t\tUsage: \"sets the domain isolation groups\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:     FlagJSON,\n\t\t\t\t\tUsage:    `the configurations to upsert: eg: [{\"Name\": \"zone-1\": \"State\": 2}]. To remove groups, specify an empty configuration`,\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t\t&cli.StringSliceFlag{\n\t\t\t\t\tName:     FlagIsolationGroupSetDrains,\n\t\t\t\t\tUsage:    \"Use to upsert the configuration for all drains. Note that this is an upsert operation and will overwrite all existing configuration\",\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:     FlagIsolationGroupsRemoveAllDrains,\n\t\t\t\t\tUsage:    \"Removes all drains\",\n\t\t\t\t\tRequired: false,\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: AdminUpdateDomainIsolationGroups,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_async_queue_commands.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nfunc AdminGetAsyncWFConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not present:\", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\treq := &types.GetDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain: domainName,\n\t}\n\n\tresp, err := adminClient.GetDomainAsyncWorkflowConfiguraton(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get async wf queue config\", err)\n\t}\n\n\tif resp == nil || resp.Configuration == nil {\n\t\tfmt.Printf(\"Async workflow queue config not found for domain %s\\n\", domainName)\n\t\treturn nil\n\t}\n\n\tfmt.Printf(\"Async workflow queue config for domain %s:\\n\", domainName)\n\tprettyPrintJSONObject(getDeps(c).Output(), resp.Configuration)\n\treturn nil\n}\n\nfunc AdminUpdateAsyncWFConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not present:\", err)\n\t}\n\tasyncWFCfgJSON, err := getRequiredOption(c, FlagJSON)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not present:\", err)\n\t}\n\n\tvar cfg types.AsyncWorkflowConfiguration\n\terr = json.Unmarshal([]byte(asyncWFCfgJSON), &cfg)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to parse async workflow config\", err)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\treq := &types.UpdateDomainAsyncWorkflowConfiguratonRequest{\n\t\tDomain:        domainName,\n\t\tConfiguration: &cfg,\n\t}\n\n\t_, err = adminClient.UpdateDomainAsyncWorkflowConfiguraton(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to update async workflow queue config\", err)\n\t}\n\n\tfmt.Printf(\"Successfully updated async workflow queue config for domain %s\\n\", domainName)\n\treturn nil\n}\n"
  },
  {
    "path": "tools/cli/admin_async_queue_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestAdminGetAsyncWFConfig(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\n\t// Define table-driven tests\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\texpectedStr      string\n\t\tcmdline          string\n\t\tmockDepsError    error\n\t\tmockContextError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\texpectedResponse := &types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\t\tConfiguration: &types.AsyncWorkflowConfiguration{\n\t\t\t\t\t\tEnabled:   true,\n\t\t\t\t\t\tQueueType: \"queueType\",\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\texpectedStr:   \"PredefinedQueueName\",\n\t\t\tcmdline:       \"cadence --domain test-domain admin async-wf-queue get\",\n\t\t},\n\t\t{\n\t\t\tname: \"Required flag not present\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call to the mock admin client is expected\n\t\t\t},\n\t\t\texpectedError: \"Required flag not present:\",\n\t\t\tcmdline:       \"cadence admin async-wf-queue get\", // --domain is missing\n\t\t},\n\t\t{\n\t\t\tname: \"Config not found (resp.Configuration == nil)\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.GetDomainAsyncWorkflowConfiguratonResponse{\n\t\t\t\t\t\tConfiguration: nil,\n\t\t\t\t\t}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tcmdline:       \"cadence --domain test-domain admin async-wf-queue get\",\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to get async wf config\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"failed to get async config\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"Failed to get async wf queue config\",\n\t\t\tcmdline:       \"cadence --domain test-domain admin async-wf-queue get\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Create mock app with clientFactoryMock\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\n\t\t\terr := clitest.RunCommandLine(t, app, tt.cmdline)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Contains(t, ioHandler.outputBytes.String(), tt.expectedStr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateAsyncWFConfig(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tdefer mockCtrl.Finish()\n\n\t// Define table-driven tests\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\tcmdline          string\n\t\tmockContextError error\n\t\tunmarshalError   error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateDomainAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.UpdateDomainAsyncWorkflowConfiguratonResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tcmdline:       `cadence --domain test-domain admin async-wf-queue update --json '{\"Enabled\": true}'`,\n\t\t},\n\t\t{\n\t\t\tname: \"Required flag not present for domain\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call to the mock admin client is expected\n\t\t\t},\n\t\t\texpectedError: \"Required flag not present:\",\n\t\t\tcmdline:       `cadence admin async-wf-queue update --json '{\"Enabled\": true}'`, // --domain is missing\n\t\t},\n\t\t{\n\t\t\tname: \"Required flag not present for JSON\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call to the mock admin client is expected\n\t\t\t},\n\t\t\texpectedError: \"Required flag not present:\",\n\t\t\tcmdline:       `cadence --domain test-domain admin async-wf-queue update --json \"\"`, // empty --json flag\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to parse async workflow config\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call setup for this test case as JSON parsing fails\n\t\t\t},\n\t\t\texpectedError:  \"Failed to parse async workflow config\",\n\t\t\tcmdline:        `cadence --domain test-domain admin async-wf-queue update --json invalid-json`,\n\t\t\tunmarshalError: fmt.Errorf(\"unmarshal error\"),\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to update async workflow config\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateDomainAsyncWorkflowConfiguraton(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"update failed\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"Failed to update async workflow queue config\",\n\t\t\tcmdline:       `cadence --domain test-domain admin async-wf-queue update --json '{\"Enabled\": true}'`,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\n\t\t\t// Create mock app with clientFactoryMock\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t})\n\n\t\t\terr := clitest.RunCommandLine(t, app, tt.cmdline)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_cluster_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/fatih/color\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/visibility\"\n\t\"github.com/uber/cadence/service/worker/failovermanager\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// An indirection for the prompt function so that it can be mocked in the unit tests\nvar promptFn = prompt\n\n// AdminAddSearchAttribute to whitelist search attribute\nfunc AdminAddSearchAttribute(c *cli.Context) error {\n\tkey, err := getRequiredOption(c, FlagSearchAttributesKey)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not present:\", err)\n\t}\n\tif err := visibility.ValidateSearchAttributeKey(key); err != nil {\n\t\treturn commoncli.Problem(\"Invalid search-attribute key.\", err)\n\t}\n\n\tvalType, err := getRequiredIntOption(c, FlagSearchAttributesType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not present:\", err)\n\t}\n\tif !isValueTypeValid(valType) {\n\t\treturn commoncli.Problem(\"Unknown Search Attributes value type.\", nil)\n\t}\n\n\t// ask user for confirmation\n\tpromptMsg := fmt.Sprintf(\"Are you trying to add key [%s] with Type [%s]? y/N\",\n\t\tcolor.YellowString(key), color.YellowString(intValTypeToString(valType)))\n\tpromptFn(promptMsg)\n\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tctx, cancel, err := newContext(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tdefer cancel()\n\trequest := &types.AddSearchAttributeRequest{\n\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\tkey: types.IndexedValueType(valType),\n\t\t},\n\t\tSecurityToken: c.String(FlagSecurityToken),\n\t}\n\n\terr = adminClient.AddSearchAttribute(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Add search attribute failed.\", err)\n\t}\n\tfmt.Println(\"Success. Note that for a multi-node Cadence cluster, DynamicConfig MUST be updated separately to whitelist the new attributes.\")\n\treturn nil\n}\n\n// AdminDescribeCluster is used to dump information about the cluster\nfunc AdminDescribeCluster(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tresponse, err := adminClient.DescribeCluster(ctx)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation DescribeCluster failed.\", err)\n\t}\n\n\tprettyPrintJSONObject(getDeps(c).Output(), response)\n\treturn nil\n}\n\nfunc AdminRebalanceStart(c *cli.Context) error {\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Context not created:\", err)\n\t}\n\tworkflowID := failovermanager.RebalanceWorkflowID\n\trbParams := &failovermanager.RebalanceParams{\n\t\tBatchFailoverSize:              100,\n\t\tBatchFailoverWaitTimeInSeconds: 10,\n\t}\n\tinput, err := json.Marshal(rbParams)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to serialize params for failover workflow\", err)\n\t}\n\top, err := getOperator()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get operator\", err)\n\t}\n\tmemo, err := getWorkflowMemo(map[string]interface{}{\n\t\tconstants.MemoKeyForOperator: op,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to serialize memo\", err)\n\t}\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              constants.SystemLocalDomainName,\n\t\tWorkflowID:                          workflowID,\n\t\tRequestID:                           uuid.New(),\n\t\tIdentity:                            getCliIdentity(),\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(int32(defaultDecisionTimeoutInSeconds)),\n\t\tInput:                               input,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: failovermanager.TaskListName,\n\t\t},\n\t\tMemo: memo,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: failovermanager.RebalanceWorkflowTypeName,\n\t\t},\n\t}\n\n\tresp, err := client.StartWorkflowExecution(tcCtx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to start failover workflow\", err)\n\t}\n\n\toutput := getDeps(c).Output()\n\toutput.Write([]byte(\"Rebalance workflow started\\n\"))\n\toutput.Write([]byte(\"wid: \" + workflowID + \"\\n\"))\n\toutput.Write([]byte(\"rid: \" + resp.GetRunID() + \"\\n\"))\n\n\treturn nil\n}\n\nfunc AdminRebalanceList(c *cli.Context) error {\n\tif err := c.Set(FlagWorkflowID, failovermanager.RebalanceWorkflowID); err != nil {\n\t\treturn err\n\t}\n\tif err := c.Set(FlagDomain, constants.SystemLocalDomainName); err != nil {\n\t\treturn err\n\t}\n\treturn ListWorkflow(c)\n}\n\nfunc intValTypeToString(valType int) string {\n\tswitch valType {\n\tcase 0:\n\t\treturn \"String\"\n\tcase 1:\n\t\treturn \"Keyword\"\n\tcase 2:\n\t\treturn \"Int\"\n\tcase 3:\n\t\treturn \"Double\"\n\tcase 4:\n\t\treturn \"Bool\"\n\tcase 5:\n\t\treturn \"Datetime\"\n\tdefault:\n\t\treturn \"\"\n\t}\n}\n\nfunc isValueTypeValid(valType int) bool {\n\treturn valType >= 0 && valType <= 5\n}\n"
  },
  {
    "path": "tools/cli/admin_cluster_commands_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/visibility\"\n\t\"github.com/uber/cadence/service/worker/failovermanager\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestAdminAddSearchAttribute_isValueTypeValid(t *testing.T) {\n\ttestCases := []struct {\n\t\tname     string\n\t\tinput    int\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname:     \"negative\",\n\t\t\tinput:    -1,\n\t\t\texpected: false,\n\t\t},\n\t\t{\n\t\t\tname:     \"valid\",\n\t\t\tinput:    0,\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"valid\",\n\t\t\tinput:    5,\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"unknown\",\n\t\t\tinput:    6,\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, testCase := range testCases {\n\t\tassert.Equal(t, testCase.expected, isValueTypeValid(testCase.input))\n\t}\n}\n\nfunc TestAdminFailover(t *testing.T) {\n\tvar listDomainsResponse = &types.ListDomainsResponse{\n\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t{\n\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\tName:        \"test-domain\",\n\t\t\t\t\tDescription: \"a test domain\",\n\t\t\t\t\tOwnerEmail:  \"test@uber.com\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\tconstants.DomainDataKeyForManagedFailover: \"true\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\t\t\tActiveClusterName: \"active\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"active\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tClusterName: \"standby\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tt.Run(\"standby cluster\", func(t *testing.T) {\n\t\ttd := newCLITestData(t)\n\t\tcliCtx := clitest.NewCLIContext(\n\t\t\tt,\n\t\t\ttd.app,\n\t\t\tclitest.StringArgument(FlagActiveClusterName, \"standby\"),\n\t\t)\n\n\t\ttd.mockFrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(listDomainsResponse, nil).Times(1)\n\t\ttd.mockFrontendClient.EXPECT().UpdateDomain(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\n\t\tdomainCLI := &domainCLIImpl{\n\t\t\tfrontendClient: td.mockFrontendClient,\n\t\t}\n\n\t\tsucceed, failed, err := domainCLI.failoverDomains(cliCtx)\n\t\tassert.Equal(t, []string{\"test-domain\"}, succeed)\n\t\tassert.Equal(t, 0, len(failed))\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"active cluster\", func(t *testing.T) {\n\t\ttd := newCLITestData(t)\n\t\tcliCtx := clitest.NewCLIContext(\n\t\t\tt,\n\t\t\ttd.app,\n\t\t\tclitest.StringArgument(FlagActiveClusterName, \"active\"),\n\t\t)\n\n\t\ttd.mockFrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(listDomainsResponse, nil).Times(1)\n\n\t\tdomainCLI := &domainCLIImpl{\n\t\t\tfrontendClient: td.mockFrontendClient,\n\t\t}\n\n\t\tsucceed, failed, err := domainCLI.failoverDomains(cliCtx)\n\t\tassert.Equal(t, 0, len(succeed))\n\t\tassert.Equal(t, 0, len(failed))\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc TestValidSearchAttributeKey(t *testing.T) {\n\tassert.NoError(t, visibility.ValidateSearchAttributeKey(\"city\"))\n\tassert.NoError(t, visibility.ValidateSearchAttributeKey(\"cityId\"))\n\tassert.NoError(t, visibility.ValidateSearchAttributeKey(\"paymentProfileUUID\"))\n\tassert.NoError(t, visibility.ValidateSearchAttributeKey(\"job_type\"))\n\n\tassert.Error(t, visibility.ValidateSearchAttributeKey(\"payments-biling-invoices-TransactionUUID\"))\n\tassert.Error(t, visibility.ValidateSearchAttributeKey(\"9lives\"))\n\tassert.Error(t, visibility.ValidateSearchAttributeKey(\"tax%\"))\n}\n\nfunc TestAdminDescribeCluster(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmockSetup      func(td *cliTestData)\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\t// Expected response from DescribeCluster\n\t\t\t\texpectedResponse := &types.DescribeClusterResponse{\n\t\t\t\t\tSupportedClientVersions: &types.SupportedClientVersions{\n\t\t\t\t\t\tGoSdk: \"1.5.0\",\n\t\t\t\t\t},\n\t\t\t\t}\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeCluster(gomock.Any()).Return(expectedResponse, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: `{\n  \"supportedClientVersions\": {\n    \"goSdk\": \"1.5.0\"\n  }\n}\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeClusterError\",\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\t// Mock DescribeCluster to return an error\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeCluster(gomock.Any()).Return(nil, fmt.Errorf(\"DescribeCluster failed\")).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Operation DescribeCluster failed.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\t// Set up mock based on the specific test case\n\t\t\ttt.mockSetup(td)\n\t\t\tcliCtx := clitest.NewCLIContext(t, td.app)\n\n\t\t\terr := AdminDescribeCluster(cliCtx)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminRebalanceStart(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tmockSetup      func(td *cliTestData)\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tmockResponse := &types.StartWorkflowExecutionResponse{\n\t\t\t\t\tRunID: \"test-run-id\",\n\t\t\t\t}\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(mockResponse, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: \"Rebalance workflow started\\nwid: cadence-rebalance-workflow\\nrid: test-run-id\\n\",\n\t\t\texpectedError:  \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"StartWorkflowExecutionError\",\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\t// Mock StartWorkflowExecution to return an error\n\t\t\t\ttd.mockFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"failed to start workflow\")).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Failed to start failover workflow\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.mockSetup(td)\n\n\t\t\tcliCtx := clitest.NewCLIContext(t, td.app, clitest.StringArgument(FlagDomain, testDomain))\n\n\t\t\terr := AdminRebalanceStart(cliCtx)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestIntValTypeToString(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tvalType  int\n\t\texpected string\n\t}{\n\t\t{\n\t\t\tname:     \"StringType\",\n\t\t\tvalType:  0,\n\t\t\texpected: \"String\",\n\t\t},\n\t\t{\n\t\t\tname:     \"KeywordType\",\n\t\t\tvalType:  1,\n\t\t\texpected: \"Keyword\",\n\t\t},\n\t\t{\n\t\t\tname:     \"IntType\",\n\t\t\tvalType:  2,\n\t\t\texpected: \"Int\",\n\t\t},\n\t\t{\n\t\t\tname:     \"DoubleType\",\n\t\t\tvalType:  3,\n\t\t\texpected: \"Double\",\n\t\t},\n\t\t{\n\t\t\tname:     \"BoolType\",\n\t\t\tvalType:  4,\n\t\t\texpected: \"Bool\",\n\t\t},\n\t\t{\n\t\t\tname:     \"DatetimeType\",\n\t\t\tvalType:  5,\n\t\t\texpected: \"Datetime\",\n\t\t},\n\t\t{\n\t\t\tname:     \"UnknownType\",\n\t\t\tvalType:  999,\n\t\t\texpected: \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Call the function and check the result\n\t\t\tresult := intValTypeToString(tt.valType)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestAdminRebalanceList(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tprepareEnv     func(td *cliTestData) *cli.Context\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tprepareEnv: func(td *cliTestData) *cli.Context {\n\t\t\t\t// Mock successful ListWorkflow call\n\t\t\t\ttd.mockFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{}, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListClosedWorkflowExecutionsResponse{}, nil).Times(1)\n\n\t\t\t\treturn clitest.NewCLIContext(t,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, failovermanager.RebalanceWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagDomain, constants.SystemLocalDomainName),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedOutput: \"\\n\", // Example output for success case\n\t\t\texpectedError:  \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"SetWorkflowIDError\",\n\n\t\t\tprepareEnv: func(td *cliTestData) *cli.Context {\n\t\t\t\t// Create CLI app and set up flag set without FlagWorkflowID\n\t\t\t\treturn clitest.NewCLIContext(t,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, constants.SystemLocalDomainName),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"no such flag -workflow_id\",\n\t\t},\n\t\t{\n\t\t\tname: \"SetDomainError\",\n\t\t\tprepareEnv: func(td *cliTestData) *cli.Context {\n\t\t\t\t// Create CLI app and set up flag set without FlagDomain\n\t\t\t\treturn clitest.NewCLIContext(t,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, failovermanager.RebalanceWorkflowID),\n\t\t\t\t)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"no such flag -domain\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\t// Prepare the test environment for the specific test case\n\t\t\tc := tt.prepareEnv(td)\n\n\t\t\terr := AdminRebalanceList(c)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminAddSearchAttribute_errors(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupContext   func(app *cli.App) *cli.Context\n\t\texpectedOutput string\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"MissingSearchAttributesKey\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, app /* missing FlagSearchAttributesKey argument */)\n\t\t\t},\n\t\t\texpectedOutput: \"\", // In this case, likely no output\n\t\t\texpectedError:  \"Required flag not present:\",\n\t\t},\n\t\t{\n\t\t\tname: \"InvalidSearchAttributeKey\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tconst invalidSearchAttr = \"123_invalid_key\" // Invalid key, starts with number\n\t\t\t\treturn clitest.NewCLIContext(t, app, clitest.StringArgument(FlagSearchAttributesKey, invalidSearchAttr))\n\t\t\t},\n\t\t\texpectedOutput: \"\", // In this case, likely no output\n\t\t\texpectedError:  \"Invalid search-attribute key.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\n\t\t\t// Set up the CLI context for the specific test case\n\t\t\tcliCtx := tt.setupContext(td.app)\n\n\t\t\terr := AdminAddSearchAttribute(cliCtx)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_commands.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io/ioutil\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\ttableRenderSize = 10\n\tremoteHint      = \" (Hint: use --remote flag to run this operation via the server API instead of direct DB access)\"\n)\n\n// AdminShowWorkflow shows history\nfunc AdminShowWorkflow(c *cli.Context) error {\n\ttid := c.String(FlagTreeID)\n\tbid := c.String(FlagBranchID)\n\tsid := c.Int(FlagShardID)\n\tminEventID := c.Int64(FlagMinEventID)\n\tmaxEventID := c.Int64(FlagMaxEventID)\n\toutputFileName := c.String(FlagOutputFilename)\n\tdomainName := c.String(FlagDomain)\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tserializer := persistence.NewPayloadSerializer()\n\tvar history []*persistence.DataBlob\n\tif len(tid) != 0 {\n\t\tthriftrwEncoder := codec.NewThriftRWEncoder()\n\t\thistV2, err := getDeps(c).initializeHistoryManager(c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in Admin delete WF: \", err)\n\t\t}\n\t\tbranchToken, err := thriftrwEncoder.Encode(&shared.HistoryBranch{\n\t\t\tTreeID:   &tid,\n\t\t\tBranchID: &bid,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"encoding branch token err\", err)\n\t\t}\n\t\tresp, err := histV2.ReadRawHistoryBranch(ctx, &persistence.ReadHistoryBranchRequest{\n\t\t\tBranchToken: branchToken,\n\t\t\tMinEventID:  minEventID,\n\t\t\tMaxEventID:  maxEventID,\n\t\t\tPageSize:    int(maxEventID - minEventID + 1),\n\t\t\tShardID:     &sid,\n\t\t\tDomainName:  domainName,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"ReadHistoryBranch err\", err)\n\t\t}\n\n\t\thistory = resp.HistoryEventBlobs\n\t} else {\n\t\treturn commoncli.Problem(\"need to specify TreeID/BranchID/ShardID\", nil)\n\t}\n\n\tif len(history) == 0 {\n\t\treturn commoncli.Problem(\"no events\", nil)\n\t}\n\tallEvents := &shared.History{}\n\ttotalSize := 0\n\tfor idx, b := range history {\n\t\ttotalSize += len(b.Data)\n\t\tfmt.Printf(\"======== batch %v, blob len: %v ======\\n\", idx+1, len(b.Data))\n\t\tinternalHistoryBatch, err := serializer.DeserializeBatchEvents(b)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"DeserializeBatchEvents err\", err)\n\t\t}\n\t\thistoryBatch := thrift.FromHistoryEventArray(internalHistoryBatch)\n\t\tallEvents.Events = append(allEvents.Events, historyBatch...)\n\t\tfor _, e := range historyBatch {\n\t\t\tjsonstr, err := json.Marshal(e)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"json.Marshal err\", err)\n\t\t\t}\n\t\t\tfmt.Println(string(jsonstr))\n\t\t}\n\t}\n\tfmt.Printf(\"======== total batches %v, total blob len: %v ======\\n\", len(history), totalSize)\n\n\tif outputFileName != \"\" {\n\t\tdata, err := json.Marshal(allEvents.Events)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to serialize history data.\", err)\n\t\t}\n\t\tif err := ioutil.WriteFile(outputFileName, data, 0666); err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to export history data file.\", err)\n\t\t}\n\t}\n\treturn nil\n}\n\n// AdminDescribeWorkflow describe a new workflow execution for admin\nfunc AdminDescribeWorkflow(c *cli.Context) error {\n\n\tresp, err := describeMutableState(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), resp)\n\n\tif resp != nil {\n\t\tmsStr := resp.GetMutableStateInDatabase()\n\t\tms := persistence.WorkflowMutableState{}\n\t\terr := json.Unmarshal([]byte(msStr), &ms)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"json.Unmarshal err\", err)\n\t\t}\n\t\tcurrentBranchToken := ms.ExecutionInfo.BranchToken\n\t\tif ms.VersionHistories != nil {\n\t\t\t// if VersionHistories is set, then all branch infos are stored in VersionHistories\n\t\t\tcurrentVersionHistory, err := ms.VersionHistories.GetCurrentVersionHistory()\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"ms.VersionHistories.GetCurrentVersionHistory err\", err)\n\t\t\t}\n\t\t\tcurrentBranchToken = currentVersionHistory.GetBranchToken()\n\t\t}\n\n\t\tbranchInfo := shared.HistoryBranch{}\n\t\tthriftrwEncoder := codec.NewThriftRWEncoder()\n\t\terr = thriftrwEncoder.Decode(currentBranchToken, &branchInfo)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"thriftrwEncoder.Decode err\", err)\n\t\t}\n\t\tprettyPrintJSONObject(getDeps(c).Output(), branchInfo)\n\t\tif ms.ExecutionInfo.AutoResetPoints != nil {\n\t\t\tgetDeps(c).Output().Write([]byte(\"auto-reset-points:\"))\n\t\t\tfor _, p := range ms.ExecutionInfo.AutoResetPoints.Points {\n\t\t\t\tcreateT := time.Unix(0, p.GetCreatedTimeNano())\n\t\t\t\texpireT := time.Unix(0, p.GetExpiringTimeNano())\n\t\t\t\tgetDeps(c).Output().Write([]byte(fmt.Sprintln(p.GetBinaryChecksum(), p.GetRunID(), p.GetFirstDecisionCompletedID(), p.GetResettable(), createT, expireT)))\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc describeMutableState(c *cli.Context) (*types.AdminDescribeWorkflowExecutionResponse, error) {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found\", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found\", err)\n\t}\n\trid := c.String(FlagRunID)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tresp, err := adminClient.DescribeWorkflowExecution(\n\t\tctx,\n\t\t&types.AdminDescribeWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t},\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Get workflow mutableState failed\", err)\n\t}\n\treturn resp, nil\n}\n\n// AdminMaintainCorruptWorkflow deletes workflow from DB if it's corrupt\nfunc AdminMaintainCorruptWorkflow(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tworkflowID := c.String(FlagWorkflowID)\n\trunID := c.String(FlagRunID)\n\tskipErrors := c.Bool(FlagSkipErrorMode)\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\trequest := &types.AdminMaintainWorkflowRequest{\n\t\tDomain: domainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tSkipErrors: skipErrors,\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\t_, err = adminClient.MaintainCorruptWorkflow(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation AdminMaintainCorruptWorkflow failed.\", err)\n\t}\n\n\treturn err\n}\n\n// AdminDeleteWorkflow delete a workflow execution for admin\nfunc AdminDeleteWorkflow(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\trid := c.String(FlagRunID)\n\tremote := c.Bool(FlagRemote)\n\tskipError := c.Bool(FlagSkipErrorMode)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\t// With remote flag, we run the command on the server side using existing APIs\n\t// Without remote, commands are run directly through some DB clients. This is\n\t// useful if server is down somehow. However, we only support couple DB clients\n\t// currently. If the server side hosts working, remote is a cleaner approach\n\tif remote {\n\t\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\trequest := &types.AdminDeleteWorkflowRequest{\n\t\t\tDomain: domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t},\n\t\t\tSkipErrors: skipError,\n\t\t}\n\t\t_, err = adminClient.DeleteWorkflow(ctx, request)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Operation AdminMaintainCorruptWorkflow failed.\", err)\n\t\t}\n\t\treturn nil\n\t}\n\n\tresp, err := describeMutableState(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get workflow mutable state\"+remoteHint, err)\n\t}\n\tmsStr := resp.GetMutableStateInDatabase()\n\tms := persistence.WorkflowMutableState{}\n\terr = json.Unmarshal([]byte(msStr), &ms)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"json.Unmarshal err\", err)\n\t}\n\tdomainID := ms.ExecutionInfo.DomainID\n\n\tshardID := resp.GetShardID()\n\tshardIDInt, err := strconv.Atoi(shardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"strconv.Atoi(shardID) err\", err)\n\t}\n\thistV2, err := getDeps(c).initializeHistoryManager(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Unable to initialize history manager\"+remoteHint, err)\n\t}\n\tdefer histV2.Close()\n\texeStore, err := getDeps(c).initializeExecutionManager(c, shardIDInt)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Unable to initialize execution manager\"+remoteHint, err)\n\t}\n\tbranchInfo := shared.HistoryBranch{}\n\tthriftrwEncoder := codec.NewThriftRWEncoder()\n\tbranchTokens := [][]byte{ms.ExecutionInfo.BranchToken}\n\tif ms.VersionHistories != nil {\n\t\t// if VersionHistories is set, then all branch infos are stored in VersionHistories\n\t\tbranchTokens = [][]byte{}\n\t\tfor _, versionHistory := range ms.VersionHistories.ToInternalType().Histories {\n\t\t\tbranchTokens = append(branchTokens, versionHistory.BranchToken)\n\t\t}\n\t}\n\n\tfor _, branchToken := range branchTokens {\n\t\terr = thriftrwEncoder.Decode(branchToken, &branchInfo)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"thriftrwEncoder.Decode err\", err)\n\t\t}\n\t\tfmt.Println(\"deleting history events for ...\")\n\t\tprettyPrintJSONObject(getDeps(c).Output(), branchInfo)\n\t\terr = histV2.DeleteHistoryBranch(ctx, &persistence.DeleteHistoryBranchRequest{\n\t\t\tBranchToken: branchToken,\n\t\t\tShardID:     &shardIDInt,\n\t\t\tDomainName:  domain,\n\t\t})\n\t\tif err != nil {\n\t\t\tif skipError {\n\t\t\t\tfmt.Println(\"failed to delete history, \", err)\n\t\t\t} else {\n\t\t\t\treturn commoncli.Problem(\"DeleteHistoryBranch err\"+remoteHint, err)\n\t\t\t}\n\t\t}\n\t}\n\n\treq := &persistence.DeleteWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t\tDomainName: domain,\n\t}\n\n\terr = exeStore.DeleteWorkflowExecution(ctx, req)\n\tif err != nil {\n\t\tif skipError {\n\t\t\tfmt.Println(\"delete mutableState row failed, \", err)\n\t\t} else {\n\t\t\treturn commoncli.Problem(\"delete mutableState row failed\"+remoteHint, err)\n\t\t}\n\t}\n\tfmt.Println(\"delete mutableState row successfully\")\n\n\tdeleteCurrentReq := &persistence.DeleteCurrentWorkflowExecutionRequest{\n\t\tDomainID:   domainID,\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t}\n\n\terr = exeStore.DeleteCurrentWorkflowExecution(ctx, deleteCurrentReq)\n\tif err != nil {\n\t\tif skipError {\n\t\t\tfmt.Println(\"delete current row failed, \", err)\n\t\t} else {\n\t\t\treturn commoncli.Problem(\"delete current row failed\"+remoteHint, err)\n\t\t}\n\t}\n\tfmt.Println(\"delete current row successfully\")\n\treturn nil\n}\n\n// AdminGetDomainIDOrName map domain\nfunc AdminGetDomainIDOrName(c *cli.Context) error {\n\tdomainID := c.String(FlagDomainID)\n\tdomainName := c.String(FlagDomain)\n\n\tif (len(domainID) == 0 && len(domainName) == 0) || (len(domainID) != 0 && len(domainName) != 0) {\n\t\treturn commoncli.Problem(\"Need either domainName or domainID\", nil)\n\t}\n\n\tdomainManager, err := getDeps(c).initializeDomainManager(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in Admin delete WF: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\toutput := getDeps(c).Output()\n\tif len(domainID) > 0 {\n\t\tres, err := domainManager.GetDomain(ctx, &persistence.GetDomainRequest{ID: domainID})\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"GetDomain error\", err)\n\t\t}\n\t\tfmt.Fprintf(output, \"domainName for domainID %v is %v\\n\", domainID, res.Info.Name)\n\t} else {\n\t\tres, err := domainManager.GetDomain(ctx, &persistence.GetDomainRequest{Name: domainName})\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"GetDomain error\", err)\n\t\t}\n\t\tfmt.Fprintf(output, \"domainID for domainName %v is %v\\n\", domainName, res.Info.ID)\n\t}\n\treturn nil\n}\n\n// AdminGetShardID get shardID\nfunc AdminGetShardID(c *cli.Context) error {\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tnumberOfShards := c.Int(FlagNumberOfShards)\n\n\tif numberOfShards <= 0 {\n\t\treturn commoncli.Problem(\"numberOfShards is required\", nil)\n\t}\n\tshardID := common.WorkflowIDToHistoryShard(wid, numberOfShards)\n\n\tfmt.Fprintf(getDeps(c).Output(), \"ShardID for workflowID: %v is %v\\n\", wid, shardID)\n\treturn nil\n}\n\n// AdminRemoveTask describes history host\nfunc AdminRemoveTask(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tshardID, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\ttaskID, err := getRequiredInt64Option(c, FlagTaskID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\ttypeID, err := getRequiredIntOption(c, FlagTaskType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tvar visibilityTimestamp int64\n\tif constants.TaskType(typeID) == constants.TaskTypeTimer {\n\t\tvisibilityTimestamp, err = getRequiredInt64Option(c, FlagTaskVisibilityTimestamp)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t\t}\n\t}\n\tvar clusterName string\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\treq := &types.RemoveTaskRequest{\n\t\tShardID:             int32(shardID),\n\t\tType:                common.Int32Ptr(int32(typeID)),\n\t\tTaskID:              taskID,\n\t\tVisibilityTimestamp: common.Int64Ptr(visibilityTimestamp),\n\t\tClusterName:         clusterName,\n\t}\n\n\terr = adminClient.RemoveTask(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Remove task has failed\", err)\n\t}\n\treturn nil\n}\n\n// AdminDescribeShard describes shard by shard id\nfunc AdminDescribeShard(c *cli.Context) error {\n\tsid, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tshardManager, err := getDeps(c).initializeShardManager(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in describe shard: \", err)\n\t}\n\tgetShardReq := &persistence.GetShardRequest{ShardID: sid}\n\tresp, err := shardManager.GetShard(ctx, getShardReq)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to describe shard.\", err)\n\t}\n\n\tprettyPrintJSONObject(getDeps(c).Output(), resp)\n\treturn nil\n}\n\n// AdminSetShardRangeID set shard rangeID by shard id\nfunc AdminSetShardRangeID(c *cli.Context) error {\n\tsid, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\trid, err := getRequiredInt64Option(c, FlagRangeID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tshardManager, err := getDeps(c).initializeShardManager(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in Admin SetShardRangeID: \", err)\n\t}\n\tgetShardResp, err := shardManager.GetShard(ctx, &persistence.GetShardRequest{ShardID: sid})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get shardInfo.\", err)\n\t}\n\n\tpreviousRangeID := getShardResp.ShardInfo.RangeID\n\tupdatedShardInfo := getShardResp.ShardInfo\n\tupdatedShardInfo.RangeID = rid\n\tupdatedShardInfo.StolenSinceRenew++\n\tupdatedShardInfo.Owner = \"\"\n\tupdatedShardInfo.UpdatedAt = time.Now()\n\n\terr = shardManager.UpdateShard(ctx, &persistence.UpdateShardRequest{\n\t\tPreviousRangeID: previousRangeID,\n\t\tShardInfo:       updatedShardInfo,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to reset shard rangeID.\", err)\n\t}\n\n\tfmt.Fprintf(getDeps(c).Output(), \"Successfully updated rangeID from %v to %v for shard %v.\\n\", previousRangeID, rid, sid)\n\treturn nil\n}\n\n// AdminCloseShard closes shard by shard id\nfunc AdminCloseShard(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tsid, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required option not found\", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context\", err)\n\t}\n\treq := &types.CloseShardRequest{}\n\treq.ShardID = int32(sid)\n\n\terr = adminClient.CloseShard(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Close shard task has failed\", err)\n\t}\n\treturn nil\n}\n\ntype ShardRow struct {\n\tShardID  int32  `header:\"ShardID\"`\n\tIdentity string `header:\"Identity\"`\n}\n\n// AdminDescribeShardDistribution describes shard distribution\nfunc AdminDescribeShardDistribution(c *cli.Context) error {\n\toutput := getDeps(c).Output()\n\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context\", err)\n\t}\n\treq := &types.DescribeShardDistributionRequest{\n\t\tPageSize: int32(c.Int(FlagPageSize)),\n\t\tPageID:   int32(c.Int(FlagPageID)),\n\t}\n\n\tresp, err := adminClient.DescribeShardDistribution(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Shard list failed\", err)\n\t}\n\n\tfmt.Fprintf(output, \"Total Number of Shards: %d\\n\", resp.NumberOfShards)\n\tfmt.Fprintf(output, \"Number of Shards Returned: %d\\n\", len(resp.Shards))\n\n\tif len(resp.Shards) == 0 {\n\t\treturn nil\n\t}\n\n\ttable := []ShardRow{}\n\topts := RenderOptions{DefaultTemplate: templateTable, Color: true}\n\toutputPageSize := tableRenderSize\n\tfor shardID, identity := range resp.Shards {\n\t\tif outputPageSize == 0 {\n\t\t\tif err := Render(c, table, opts); err != nil {\n\t\t\t\treturn fmt.Errorf(\"error rendering: %w\", err)\n\t\t\t}\n\t\t\ttable = []ShardRow{}\n\t\t\tif !showNextPage(output) {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\toutputPageSize = tableRenderSize\n\t\t}\n\t\ttable = append(table, ShardRow{ShardID: shardID, Identity: identity})\n\t\toutputPageSize--\n\t}\n\t// output the remaining rows\n\treturn Render(c, table, opts)\n}\n\n// AdminDescribeHistoryHost describes history host\nfunc AdminDescribeHistoryHost(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\twid := c.String(FlagWorkflowID)\n\tsid := c.Int(FlagShardID)\n\taddr := c.String(FlagHistoryAddress)\n\tprintFully := c.Bool(FlagPrintFullyDetail)\n\n\tif len(wid) == 0 && !c.IsSet(FlagShardID) && len(addr) == 0 {\n\t\treturn commoncli.Problem(\"at least one of them is required to provide to lookup host: workflowID, shardID and host address\", nil)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context\", err)\n\t}\n\treq := &types.DescribeHistoryHostRequest{}\n\tif len(wid) > 0 {\n\t\treq.ExecutionForHost = &types.WorkflowExecution{WorkflowID: wid}\n\t}\n\tif c.IsSet(FlagShardID) {\n\t\treq.ShardIDForHost = common.Int32Ptr(int32(sid))\n\t}\n\tif len(addr) > 0 {\n\t\treq.HostAddress = common.StringPtr(addr)\n\t}\n\n\tresp, err := adminClient.DescribeHistoryHost(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Describe history host failed\", err)\n\t}\n\n\tif !printFully {\n\t\tresp.ShardIDs = nil\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), resp)\n\treturn nil\n}\n\n// AdminRefreshWorkflowTasks refreshes all the tasks of a workflow\nfunc AdminRefreshWorkflowTasks(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\trid := c.String(FlagRunID)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\terr = adminClient.RefreshWorkflowTasks(ctx, &types.RefreshWorkflowTasksRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Refresh workflow task failed\", err)\n\t}\n\tfmt.Fprintln(getDeps(c).Output(), \"Refresh workflow task succeeded.\")\n\treturn nil\n}\n\n// AdminResetQueue resets task processing queue states\nfunc AdminResetQueue(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tshardID, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tclusterName, err := getRequiredOption(c, FlagCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\ttypeID, err := getRequiredIntOption(c, FlagQueueType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\treq := &types.ResetQueueRequest{\n\t\tShardID:     int32(shardID),\n\t\tClusterName: clusterName,\n\t\tType:        common.Int32Ptr(int32(typeID)),\n\t}\n\n\terr = adminClient.ResetQueue(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to reset queue\", err)\n\t}\n\n\tfmt.Fprintln(getDeps(c).Output(), \"Reset queue state succeeded\")\n\treturn nil\n}\n\n// AdminDescribeQueue describes task processing queue states\nfunc AdminDescribeQueue(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tshardID, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tclusterName, err := getRequiredOption(c, FlagCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\ttypeID, err := getRequiredIntOption(c, FlagQueueType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\treq := &types.DescribeQueueRequest{\n\t\tShardID:     int32(shardID),\n\t\tClusterName: clusterName,\n\t\tType:        common.Int32Ptr(int32(typeID)),\n\t}\n\n\tresp, err := adminClient.DescribeQueue(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to describe queue\", err)\n\t}\n\n\toutput := getDeps(c).Output()\n\tfor _, state := range resp.ProcessingQueueStates {\n\t\tfmt.Fprintln(output, state)\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "tools/cli/admin_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nconst (\n\ttestShardID      = 1234\n\ttestCluster      = \"test-cluster\"\n\ttestDomain       = \"test-domain\"\n\ttestDomainID     = \"0000001-0000002-0000003\"\n\ttestTaskList     = \"test-tasklist\"\n\ttestTaskListType = \"decision\"\n\ttestQueueType    = 2 // transfer queue\n\ttestWorkflowID   = \"test-workflow-id\"\n\ttestRunID        = \"test-run-id\"\n)\n\ntype cliTestData struct {\n\tctrl               *gomock.Controller\n\tmockFrontendClient *frontend.MockClient\n\tmockAdminClient    *admin.MockClient\n\tioHandler          *testIOHandler\n\tapp                *cli.App\n\tmockManagerFactory *MockManagerFactory\n}\n\nfunc newCLITestData(t *testing.T) *cliTestData {\n\tvar td cliTestData\n\n\ttd.ctrl = gomock.NewController(t)\n\n\ttd.mockFrontendClient = frontend.NewMockClient(td.ctrl)\n\ttd.mockAdminClient = admin.NewMockClient(td.ctrl)\n\ttd.mockManagerFactory = NewMockManagerFactory(td.ctrl)\n\ttd.ioHandler = &testIOHandler{}\n\n\t// Create a new CLI app with client factory and persistence manager factory\n\ttd.app = NewCliApp(\n\t\t&clientFactoryMock{\n\t\t\tserverFrontendClient: td.mockFrontendClient,\n\t\t\tserverAdminClient:    td.mockAdminClient,\n\t\t},\n\t\tWithIOHandler(td.ioHandler),\n\t\tWithManagerFactory(td.mockManagerFactory), // Inject the mocked persistence manager factory\n\t)\n\treturn &td\n}\n\nfunc (td *cliTestData) consoleOutput() string {\n\treturn td.ioHandler.outputBytes.String()\n}\n\nfunc TestAdminResetQueue(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no shardID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing cluster argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\t/* no cluster argument */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing queue type argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\t/* no queue type argument */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\tclitest.IntArgument(FlagQueueType, testQueueType),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().ResetQueue(gomock.Any(), &types.ResetQueueRequest{\n\t\t\t\t\tShardID:     testShardID,\n\t\t\t\t\tClusterName: testCluster,\n\t\t\t\t\tType:        common.Int32Ptr(testQueueType),\n\t\t\t\t})\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: \"Reset queue state succeeded\\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"ResetQueue returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\tclitest.IntArgument(FlagQueueType, testQueueType),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().ResetQueue(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to reset queue\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminResetQueue(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeQueue(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no shardID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing cluster argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\t/* no cluster argument */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing queue type argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\t/* no queue type argument */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\tclitest.IntArgument(FlagQueueType, testQueueType),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeQueue(gomock.Any(), &types.DescribeQueueRequest{\n\t\t\t\t\tShardID:     testShardID,\n\t\t\t\t\tClusterName: testCluster,\n\t\t\t\t\tType:        common.Int32Ptr(testQueueType),\n\t\t\t\t}).Return(&types.DescribeQueueResponse{\n\t\t\t\t\tProcessingQueueStates: []string{\"state1\", \"state2\"},\n\t\t\t\t}, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: \"state1\\nstate2\\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeQueue returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagCluster, testCluster),\n\t\t\t\t\tclitest.IntArgument(FlagQueueType, testQueueType),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeQueue(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to describe queue\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminDescribeQueue(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminRefreshWorkflowTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no domain argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing workflowID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\t/* no workflowID argument */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().RefreshWorkflowTasks(gomock.Any(), &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\tDomain: testDomain,\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: \"Refresh workflow task succeeded.\\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"RefreshWorkflowTasks returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Refresh workflow task failed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminRefreshWorkflowTasks(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeHistoryHost(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"at least one of them is required to provide to lookup host: workflowID, shardID and host address\",\n\t\t},\n\t\t{\n\t\t\tname: \"workflow-id provided, but empty\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"\"))\n\t\t\t},\n\t\t\terrContains: \"at least one of them is required to provide to lookup host: workflowID, shardID and host address\",\n\t\t},\n\t\t{\n\t\t\tname: \"addr provided, but empty\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(FlagHistoryAddress, \"\"))\n\t\t\t},\n\t\t\terrContains: \"at least one of them is required to provide to lookup host: workflowID, shardID and host address\",\n\t\t},\n\t\t{\n\t\t\tname: \"calling with all arguments\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagHistoryAddress, \"host:port\"),\n\t\t\t\t\tclitest.BoolArgument(FlagPrintFullyDetail, false),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeHistoryHost(gomock.Any(),\n\t\t\t\t\t&types.DescribeHistoryHostRequest{\n\t\t\t\t\t\tExecutionForHost: &types.WorkflowExecution{WorkflowID: testWorkflowID},\n\t\t\t\t\t\tShardIDForHost:   common.Int32Ptr(int32(testShardID)),\n\t\t\t\t\t\tHostAddress:      common.StringPtr(\"host:port\"),\n\t\t\t\t\t}).Return(\n\t\t\t\t\t&types.DescribeHistoryHostResponse{NumberOfShards: 12, Address: \"host:port\"},\n\t\t\t\t\tnil,\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t\texpectedOutput: `{\n  \"numberOfShards\": 12,\n  \"address\": \"host:port\"\n}\n`,\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeHistoryHost returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeHistoryHost(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Describe history host failed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminDescribeHistoryHost(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminCloseShard(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"no arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required option not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"calling with all arguments\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().CloseShard(gomock.Any(),\n\t\t\t\t\t&types.CloseShardRequest{\n\t\t\t\t\t\tShardID: testShardID,\n\t\t\t\t\t}).Return(nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"CloseShard returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().CloseShard(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Close shard task has failed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminCloseShard(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminRemoveTask(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"no arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing ShardID\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, 1), // type is provided\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing TaskID\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, 1),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, 1), // type is provided\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"missing TaskType\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, 1),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"calling with all arguments\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, 1), // some valid type\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().RemoveTask(gomock.Any(),\n\t\t\t\t\t&types.RemoveTaskRequest{\n\t\t\t\t\t\tShardID:             int32(testShardID),\n\t\t\t\t\t\tType:                common.Int32Ptr(1),\n\t\t\t\t\t\tTaskID:              123,\n\t\t\t\t\t\tVisibilityTimestamp: common.Int64Ptr(0),\n\t\t\t\t\t\tClusterName:         \"\",\n\t\t\t\t\t}).Return(nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"RemoveTask returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, 1), // some valid type\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().RemoveTask(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Remove task has failed\",\n\t\t},\n\t\t{\n\t\t\tname: \"calling with Timer task requiring visibility timestamp\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, int(constants.TaskTypeTimer)),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskVisibilityTimestamp, 1616161616), // visibility timestamp\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().RemoveTask(gomock.Any(),\n\t\t\t\t\t&types.RemoveTaskRequest{\n\t\t\t\t\t\tShardID:             int32(testShardID),\n\t\t\t\t\t\tType:                common.Int32Ptr(int32(constants.TaskTypeTimer)),\n\t\t\t\t\t\tTaskID:              123,\n\t\t\t\t\t\tVisibilityTimestamp: common.Int64Ptr(1616161616),\n\t\t\t\t\t\tClusterName:         \"\",\n\t\t\t\t\t}).Return(nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"calling with Timer task requiring visibility timestamp, but not provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagTaskID, 123),\n\t\t\t\t\tclitest.IntArgument(FlagTaskType, int(constants.TaskTypeTimer)),\n\t\t\t\t\t// visibility timestamp is missing though FlagTaskType is common.TaskTypeTimer\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminRemoveTask(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminGetShardID(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\texpectedOutput string // expected output to check against\n\t\terrContains    string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"no WorkflowID provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"numberOfShards not provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"some-workflow-id\"),\n\t\t\t\t\t/* numberOfShards is missing */\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"numberOfShards is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"numberOfShards is zero\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"some-workflow-id\"),\n\t\t\t\t\tclitest.IntArgument(FlagNumberOfShards, 0), // zero is invalid\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"numberOfShards is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"valid inputs\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.IntArgument(FlagNumberOfShards, 10), // valid number of shards\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\texpectedOutput: \"ShardID for workflowID: test-workflow-id is 6\\n\",\n\t\t\terrContains:    \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminGetShardID(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\n\t\t\t// If there is no error, check the output\n\t\t\tif tt.expectedOutput != \"\" {\n\t\t\t\tassert.Contains(t, tt.expectedOutput, td.consoleOutput())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeShardDistribution(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagPageSize, 10),\n\t\t\t\t\tclitest.IntArgument(FlagPageID, 1),\n\t\t\t\t\tclitest.StringArgument(FlagFormat, formatJSON),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeShardDistribution(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&types.DescribeShardDistributionRequest{\n\t\t\t\t\t\tPageSize: 10,\n\t\t\t\t\t\tPageID:   1,\n\t\t\t\t\t}).Return(\n\t\t\t\t\t&types.DescribeShardDistributionResponse{\n\t\t\t\t\t\tNumberOfShards: 1,\n\t\t\t\t\t\tShards: map[int32]string{\n\t\t\t\t\t\t\t1: \"identity1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t}, nil,\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t\texpectedOutput: `Total Number of Shards: 1\nNumber of Shards Returned: 1\n[\n  {\n    \"ShardID\": 1,\n    \"Identity\": \"identity1\"\n  }\n]\n`,\n\t\t},\n\t\t{\n\t\t\tname: \"no shards are returned\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagPageSize, 10),\n\t\t\t\t\tclitest.IntArgument(FlagPageID, 1),\n\t\t\t\t\tclitest.StringArgument(FlagFormat, formatJSON),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeShardDistribution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(\n\t\t\t\t\t\t&types.DescribeShardDistributionResponse{\n\t\t\t\t\t\t\tNumberOfShards: 10,\n\t\t\t\t\t\t\tShards:         nil, // no shards\n\t\t\t\t\t\t}, nil,\n\t\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: \"Total Number of Shards: 10\\nNumber of Shards Returned: 0\\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"DescribeShardDistribution returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagPageSize, 10),\n\t\t\t\t\tclitest.IntArgument(FlagPageID, 1),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DescribeShardDistribution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Shard list failed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminDescribeShardDistribution(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminMaintainCorruptWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"no domain argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t\tclitest.BoolArgument(FlagSkipErrorMode, true),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().MaintainCorruptWorkflow(gomock.Any(), &types.AdminMaintainWorkflowRequest{\n\t\t\t\t\tDomain: testDomain,\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tSkipErrors: true,\n\t\t\t\t}).Return(nil, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"MaintainCorruptWorkflow returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t\tclitest.BoolArgument(FlagSkipErrorMode, false),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().MaintainCorruptWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Operation AdminMaintainCorruptWorkflow failed.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminMaintainCorruptWorkflow(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminDescribeShard(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t\tcheckOutput func(td *cliTestData)\n\t}{\n\t\t{\n\t\t\tname: \"no ShardID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\t\t\t\tmockShardManager.EXPECT().GetShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetShardRequest{ShardID: testShardID},\n\t\t\t\t).Return(&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID: testShardID,\n\t\t\t\t\t\tOwner:   \"host-abc\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t\tcheckOutput: func(td *cliTestData) {\n\t\t\t\t// We must have visualised GetShardResponse in console.\n\t\t\t\t// Check it is a valid JSON and important fields are there\n\t\t\t\tvar resp persistence.GetShardResponse\n\n\t\t\t\trequire.NoError(t, json.Unmarshal([]byte(td.consoleOutput()), &resp))\n\t\t\t\tassert.Equal(t, testShardID, resp.ShardInfo.ShardID)\n\t\t\t\tassert.Equal(t, \"host-abc\", resp.ShardInfo.Owner)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\t\t\t\tmockShardManager.EXPECT().GetShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetShardRequest{ShardID: testShardID},\n\t\t\t\t).Return(&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID: testShardID,\n\t\t\t\t\t\tOwner:   \"host-abc\",\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t\tcheckOutput: func(td *cliTestData) {\n\t\t\t\t// We must have visualised GetShardResponse in console.\n\t\t\t\t// Check it is a valid JSON and important fields are there\n\t\t\t\tvar resp persistence.GetShardResponse\n\n\t\t\t\trequire.NoError(t, json.Unmarshal([]byte(td.consoleOutput()), &resp))\n\t\t\t\tassert.Equal(t, testShardID, resp.ShardInfo.ShardID)\n\t\t\t\tassert.Equal(t, \"host-abc\", resp.ShardInfo.Owner)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"GetShard returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\t\t\t\tmockShardManager.EXPECT().GetShard(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to describe shard\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to initializeShardManager\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"failed to initializeShardManager\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"failed to initializeShardManager\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminDescribeShard(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\trequire.NotNil(t, tt.checkOutput)\n\t\t\t\ttt.checkOutput(td)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminSetShardRangeID(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no ShardID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"no RangeID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\t// FlagRangeID is missing\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagRangeID, 133),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\n\t\t\t\tmockShardManager.EXPECT().GetShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetShardRequest{ShardID: testShardID},\n\t\t\t\t).Return(&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID:          testShardID,\n\t\t\t\t\t\tOwner:            \"host-abc\",\n\t\t\t\t\t\tRangeID:          123,\n\t\t\t\t\t\tStolenSinceRenew: 100, // this supposed to be incremented\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockShardManager.EXPECT().UpdateShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Do(func(ctx context.Context, req *persistence.UpdateShardRequest) {\n\t\t\t\t\t// we can't use input arguments matching as it contains current time\n\t\t\t\t\tassert.Equal(t, int64(123), req.PreviousRangeID)\n\t\t\t\t\tassert.Equal(t, 101, req.ShardInfo.StolenSinceRenew)\n\t\t\t\t\tassert.Equal(t, int64(133), req.ShardInfo.RangeID)\n\n\t\t\t\t\tnow := time.Now()\n\n\t\t\t\t\t// check time is really updated\n\t\t\t\t\tassert.WithinRange(\n\t\t\t\t\t\tt,\n\t\t\t\t\t\treq.ShardInfo.UpdatedAt,\n\t\t\t\t\t\tnow.Add(-time.Minute),\n\t\t\t\t\t\ttime.Now(),\n\t\t\t\t\t\t\"didn't update UpdatedAt?\",\n\t\t\t\t\t)\n\t\t\t\t}).Return(nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: \"Successfully updated rangeID from 123 to 133 for shard 1234.\\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"all arguments provided, but UpdateShard fails\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagRangeID, 133),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\n\t\t\t\tmockShardManager.EXPECT().GetShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(&persistence.GetShardResponse{\n\t\t\t\t\tShardInfo: &persistence.ShardInfo{\n\t\t\t\t\t\tShardID: testShardID,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\tmockShardManager.EXPECT().UpdateShard(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t).Return(errors.New(\"critical failure\"))\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to reset shard rangeID.\",\n\t\t},\n\t\t{\n\t\t\tname: \"GetShard returns an error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagRangeID, 123),\n\t\t\t\t)\n\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(td.ctrl)\n\t\t\t\tmockShardManager.EXPECT().GetShard(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(mockShardManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to get shardInfo.\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to initializeShardManager\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.IntArgument(FlagShardID, testShardID),\n\t\t\t\t\tclitest.Int64Argument(FlagRangeID, 123),\n\t\t\t\t)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeShardManager(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"failed to initializeShardManager\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"failed to initializeShardManager\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminSetShardRangeID(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDomainIDOrName(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *cliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"no DomainID or DomainName argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Need either domainName or domainID\",\n\t\t},\n\t\t{\n\t\t\tname: \"DomainID provided, valid response\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomainID, testDomainID),\n\t\t\t\t)\n\n\t\t\t\tmockDomainManager := persistence.NewMockDomainManager(td.ctrl)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetDomainRequest{ID: testDomainID},\n\t\t\t\t).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tID:   testDomainID,\n\t\t\t\t\t\tName: testDomain,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeDomainManager(gomock.Any()).\n\t\t\t\t\tReturn(mockDomainManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: fmt.Sprintf(\"domainName for domainID %v is %v\\n\", testDomainID, testDomain),\n\t\t},\n\t\t{\n\t\t\tname: \"DomainName provided, valid response\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t)\n\n\t\t\t\tmockDomainManager := persistence.NewMockDomainManager(td.ctrl)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(\n\t\t\t\t\tgomock.Any(),\n\t\t\t\t\t&persistence.GetDomainRequest{Name: testDomain},\n\t\t\t\t).Return(&persistence.GetDomainResponse{\n\t\t\t\t\tInfo: &persistence.DomainInfo{\n\t\t\t\t\t\tID:   testDomainID,\n\t\t\t\t\t\tName: testDomain,\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeDomainManager(gomock.Any()).\n\t\t\t\t\tReturn(mockDomainManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains:    \"\",\n\t\t\texpectedOutput: fmt.Sprintf(\"domainID for domainName %v is %v\\n\", testDomain, testDomainID),\n\t\t},\n\t\t{\n\t\t\tname: \"DomainManager returns an error for domainID\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomainID, testDomainID),\n\t\t\t\t)\n\n\t\t\t\tmockDomainManager := persistence.NewMockDomainManager(td.ctrl)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeDomainManager(gomock.Any()).\n\t\t\t\t\tReturn(mockDomainManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"GetDomain error\",\n\t\t},\n\t\t{\n\t\t\tname: \"DomainManager returns an error for domainName\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t)\n\n\t\t\t\tmockDomainManager := persistence.NewMockDomainManager(td.ctrl)\n\t\t\t\tmockDomainManager.EXPECT().GetDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"critical error\"))\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeDomainManager(gomock.Any()).\n\t\t\t\t\tReturn(mockDomainManager, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"GetDomain error\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to initializeDomainManager\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomainID, testDomainID),\n\t\t\t\t)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeDomainManager(gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"failed to initializeDomainManager\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"failed to initializeDomainManager\",\n\t\t},\n\t\t{\n\t\t\tname: \"both DomainID and DomainName provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomainID, testDomainID),\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Need either domainName or domainID\", // Expecting the error when both are provided\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminGetDomainIDOrName(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestAdminDeleteWorkflow(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"no domain argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app /* arguments are missing */)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"no workflow ID argument\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"remote delete success\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t\tclitest.BoolArgument(FlagRemote, true),\n\t\t\t\t\tclitest.BoolArgument(FlagSkipErrorMode, false),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DeleteWorkflow(gomock.Any(), &types.AdminDeleteWorkflowRequest{\n\t\t\t\t\tDomain: testDomain,\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: testWorkflowID,\n\t\t\t\t\t\tRunID:      testRunID,\n\t\t\t\t\t},\n\t\t\t\t\tSkipErrors: false,\n\t\t\t\t}).Return(&types.AdminDeleteWorkflowResponse{}, nil)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"remote delete returns error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagWorkflowID, testWorkflowID),\n\t\t\t\t\tclitest.StringArgument(FlagRunID, testRunID),\n\t\t\t\t\tclitest.BoolArgument(FlagRemote, true),\n\t\t\t\t)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().DeleteWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, errors.New(\"delete failed\"))\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Operation AdminMaintainCorruptWorkflow failed.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\terr := AdminDeleteWorkflow(cliCtx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_config_store_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"sort\"\n\t\"strings\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\ntype cliEntry struct {\n\tName         string\n\tDefaultValue interface{} `json:\"defaultValue,omitempty\"`\n\tValues       []*cliValue\n}\n\ntype cliValue struct {\n\tValue   interface{}\n\tFilters []*cliFilter\n}\n\ntype cliFilter struct {\n\tName  string\n\tValue interface{}\n}\n\n// AdminGetDynamicConfig gets value of specified dynamic config parameter matching specified filter\nfunc AdminGetDynamicConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tconfigName, err := getRequiredOption(c, FlagDynamicConfigName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\n\tfilter := c.String(FlagDynamicConfigFilter)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\tparsedFilters, err := parseInputFilter(filter)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to parse input filter array\", err)\n\t}\n\n\treq := &types.GetDynamicConfigRequest{\n\t\tConfigName: configName,\n\t\tFilters:    parsedFilters,\n\t}\n\n\tval, err := adminClient.GetDynamicConfig(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get dynamic config value\", err)\n\t}\n\n\tvar umVal interface{}\n\terr = json.Unmarshal(val.Value.Data, &umVal)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to unmarshal response\", err)\n\t}\n\n\tif umVal == nil {\n\t\tfmt.Printf(\"No values stored for specified dynamic config.\\n\")\n\t} else {\n\t\tprettyPrintJSONObject(getDeps(c).Output(), umVal)\n\t}\n\n\treturn nil\n}\n\n// AdminUpdateDynamicConfig updates specified dynamic config parameter with specified values\nfunc AdminUpdateDynamicConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdcName, err := getRequiredOption(c, FlagDynamicConfigName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tdcValuesRaw := c.StringSlice(FlagDynamicConfigValue)\n\n\t// WORKAROUND: urfave/cli v2 StringSliceFlag splits on commas by default.\n\t// This breaks JSON values. Try reassembling the split pieces.\n\tvar dcValues []string\n\tif len(dcValuesRaw) > 1 && strings.HasPrefix(dcValuesRaw[0], \"{\") {\n\t\tassembled := strings.Join(dcValuesRaw, \",\")\n\t\tvar test interface{}\n\t\tif json.Unmarshal([]byte(assembled), &test) == nil {\n\t\t\tdcValues = []string{assembled}\n\t\t} else {\n\t\t\tdcValues = dcValuesRaw\n\t\t}\n\t} else {\n\t\tdcValues = dcValuesRaw\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tvar parsedValues []*types.DynamicConfigValue\n\n\tif dcValues != nil {\n\t\tparsedValues = make([]*types.DynamicConfigValue, 0, len(dcValues))\n\n\t\tfor _, valueString := range dcValues {\n\t\t\tvar parsedInputValue *cliValue\n\t\t\terr := json.Unmarshal([]byte(valueString), &parsedInputValue)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Unable to unmarshal value to inputValue\", err)\n\t\t\t}\n\t\t\tparsedValue, err := convertFromInputValue(parsedInputValue)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Unable to convert from inputValue to DynamicConfigValue\", err)\n\t\t\t}\n\t\t\tparsedValues = append(parsedValues, parsedValue)\n\t\t}\n\t} else {\n\t\tparsedValues = nil\n\t}\n\n\treq := &types.UpdateDynamicConfigRequest{\n\t\tConfigName:   dcName,\n\t\tConfigValues: parsedValues,\n\t}\n\n\terr = adminClient.UpdateDynamicConfig(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to update dynamic config value\", err)\n\t}\n\tfmt.Printf(\"Dynamic Config %q updated with %s \\n\", dcName, dcValues)\n\treturn nil\n}\n\n// AdminRestoreDynamicConfig removes values of specified dynamic config parameter matching specified filter\nfunc AdminRestoreDynamicConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdcName, err := getRequiredOption(c, FlagDynamicConfigName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tfilter := c.String(FlagDynamicConfigFilter)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\tparsedFilters, err := parseInputFilter(filter)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to parse input filter\", err)\n\t}\n\n\treq := &types.RestoreDynamicConfigRequest{\n\t\tConfigName: dcName,\n\t\tFilters:    parsedFilters,\n\t}\n\n\terr = adminClient.RestoreDynamicConfig(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to restore dynamic config value\", err)\n\t}\n\tfmt.Printf(\"Dynamic Config %q restored\\n\", dcName)\n\treturn nil\n}\n\n// AdminListDynamicConfig lists all values associated with specified dynamic config parameter or all values for all dc parameter if none is specified.\nfunc AdminListDynamicConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\treq := &types.ListDynamicConfigRequest{\n\t\tConfigName: \"\", // empty string means all config values\n\t}\n\n\tval, err := adminClient.ListDynamicConfig(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to list dynamic config value(s)\", err)\n\t}\n\n\tif val == nil || val.Entries == nil || len(val.Entries) == 0 {\n\t\tfmt.Printf(\"No dynamic config values stored to list.\\n\")\n\t} else {\n\t\tcliEntries := make([]*cliEntry, 0, len(val.Entries))\n\t\tfor _, dcEntry := range val.Entries {\n\t\t\tcliEntry, err := convertToInputEntry(dcEntry)\n\t\t\tif err != nil {\n\t\t\t\tfmt.Printf(\"Cannot parse list response.\\n\")\n\t\t\t}\n\t\t\tcliEntries = append(cliEntries, cliEntry)\n\t\t}\n\t\tprettyPrintJSONObject(getDeps(c).Output(), cliEntries)\n\t}\n\treturn nil\n}\n\n// AdminListConfigKeys lists all available dynamic config keys with description and default value\nfunc AdminListConfigKeys(c *cli.Context) error {\n\n\ttype ConfigRow struct {\n\t\tName        string      `header:\"Name\" json:\"name\"`\n\t\tDescription string      `header:\"Description\" json:\"description\"`\n\t\tDefault     interface{} `header:\"Default value\" json:\"default\"`\n\t}\n\n\tvar rows []ConfigRow\n\n\tfor name, k := range dynamicproperties.GetAllKeys() {\n\t\trows = append(rows, ConfigRow{\n\t\t\tName:        name,\n\t\t\tDescription: k.Description(),\n\t\t\tDefault:     k.DefaultValue(),\n\t\t})\n\t}\n\t// sorting config key names alphabetically\n\tsort.SliceStable(rows, func(i, j int) bool {\n\t\treturn rows[i].Name < rows[j].Name\n\t})\n\n\treturn Render(c, rows, RenderOptions{\n\t\tDefaultTemplate: templateTable,\n\t\tColor:           true,\n\t\tBorder:          true,\n\t\tColumnAlignment: []int{tablewriter.ALIGN_LEFT, tablewriter.ALIGN_LEFT, tablewriter.ALIGN_RIGHT}},\n\t)\n}\n\nfunc convertToInputEntry(dcEntry *types.DynamicConfigEntry) (*cliEntry, error) {\n\tnewValues := make([]*cliValue, 0, len(dcEntry.Values))\n\tfor _, value := range dcEntry.Values {\n\t\tnewValue, err := convertToInputValue(value)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tnewValues = append(newValues, newValue)\n\t}\n\treturn &cliEntry{\n\t\tName:   dcEntry.Name,\n\t\tValues: newValues,\n\t}, nil\n}\n\nfunc convertToInputValue(dcValue *types.DynamicConfigValue) (*cliValue, error) {\n\tnewFilters := make([]*cliFilter, 0, len(dcValue.Filters))\n\tfor _, filter := range dcValue.Filters {\n\t\tnewFilter, err := convertToInputFilter(filter)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tnewFilters = append(newFilters, newFilter)\n\t}\n\n\tvar val interface{}\n\terr := json.Unmarshal(dcValue.Value.Data, &val)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &cliValue{\n\t\tValue:   val,\n\t\tFilters: newFilters,\n\t}, nil\n}\n\nfunc convertToInputFilter(dcFilter *types.DynamicConfigFilter) (*cliFilter, error) {\n\tvar val interface{}\n\terr := json.Unmarshal(dcFilter.Value.Data, &val)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &cliFilter{\n\t\tName:  dcFilter.Name,\n\t\tValue: val,\n\t}, nil\n}\n\nfunc convertFromInputValue(inputValue *cliValue) (*types.DynamicConfigValue, error) {\n\tencodedValue, err := json.Marshal(inputValue.Value)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tblob := &types.DataBlob{\n\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\tData:         encodedValue,\n\t}\n\n\tdcFilters := make([]*types.DynamicConfigFilter, 0, len(inputValue.Filters))\n\tfor _, inputFilter := range inputValue.Filters {\n\t\tdcFilter, err := convertFromInputFilter(inputFilter)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tdcFilters = append(dcFilters, dcFilter)\n\t}\n\n\treturn &types.DynamicConfigValue{\n\t\tValue:   blob,\n\t\tFilters: dcFilters,\n\t}, nil\n}\n\nfunc convertFromInputFilter(inputFilter *cliFilter) (*types.DynamicConfigFilter, error) {\n\tencodedValue, err := json.Marshal(inputFilter.Value)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &types.DynamicConfigFilter{\n\t\tName: inputFilter.Name,\n\t\tValue: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\tData:         encodedValue,\n\t\t},\n\t}, nil\n}\n\nfunc parseInputFilter(inputFilter string) ([]*types.DynamicConfigFilter, error) {\n\tif inputFilter == \"\" || inputFilter == \"{}\" {\n\t\treturn nil, nil\n\t}\n\n\tvar mapFilter = make(map[string]interface{})\n\tif err := json.Unmarshal([]byte(inputFilter), &mapFilter); err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar parsedFilters = make([]*types.DynamicConfigFilter, 0, len(mapFilter))\n\n\tfor name, value := range mapFilter {\n\t\tparsedFilter, err := convertFromInputFilter(&cliFilter{\n\t\t\tName:  name,\n\t\t\tValue: value,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\n\t\tparsedFilters = append(parsedFilters, parsedFilter)\n\t}\n\n\treturn parsedFilters, nil\n}\n"
  },
  {
    "path": "tools/cli/admin_config_store_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestAdminGetDynamicConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tcmdline     string\n\t\tsetupMock   func(td *cliTestData)\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tcmdline: `cadence admin config get --name \"\"`,\n\t\t\tname:    \"no arguments provided\",\n\t\t\tsetupMock: func(td *cliTestData) {\n\t\t\t\t// empty since arguments are missing\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname:    \"failed to get dynamic config values\",\n\t\t\tcmdline: `cadence admin config get --name test-dynamic-config-name`,\n\t\t\tsetupMock: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().GetDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.GetDynamicConfigRequest, _ ...yarpc.CallOption) (*types.GetDynamicConfigResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, \"test-dynamic-config-name\", request.ConfigName)\n\t\t\t\t\t\treturn nil, assert.AnError\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"Failed to get dynamic config value\",\n\t\t},\n\t\t{\n\t\t\tname:    \"received a dynamic config value successfully\",\n\t\t\tcmdline: `cadence admin config get --name test-dynamic-config-name`,\n\t\t\tsetupMock: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().GetDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.GetDynamicConfigRequest, _ ...yarpc.CallOption) (*types.GetDynamicConfigResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, \"test-dynamic-config-name\", request.ConfigName)\n\t\t\t\t\t\treturn &types.GetDynamicConfigResponse{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`\"config-value\"`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"received a dynamic config value with filters successfully\",\n\t\t\tcmdline: `cadence admin config get --name test-dynamic-config-name --filter '{\"domainName\":\"test-domain\", \"shardID\": 1, \"isEnabled\": true}'`,\n\t\t\tsetupMock: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().GetDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.GetDynamicConfigRequest, _ ...yarpc.CallOption) (*types.GetDynamicConfigResponse, error) {\n\t\t\t\t\t\tassert.Equal(t, \"test-dynamic-config-name\", request.ConfigName)\n\t\t\t\t\t\tassert.ElementsMatch(t, request.Filters, []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"domainName\",\n\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\tData:         []byte(`\"test-domain\"`),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"shardID\",\n\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\tData:         []byte(`1`),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tName: \"isEnabled\",\n\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t\t\t\t\t\t\t\tData:         []byte(`true`),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t})\n\t\t\t\t\t\treturn &types.GetDynamicConfigResponse{\n\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\tData:         []byte(`\"config-value\"`),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.setupMock(td)\n\n\t\t\terr := clitest.RunCommandLine(t, td.app, tt.cmdline)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateDynamicConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tcmdline     string\n\t\tsetupMocks  func(td *cliTestData)\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname:    \"no arguments provided\",\n\t\t\tcmdline: `cadence admin config update --name \"\" --value \"\"`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\t// empty since arguments are missing\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname:    \"calling with required arguments\",\n\t\t\tcmdline: `cadence admin config update --name test-dynamic-config-name --value \"{}\"`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"failed to update dynamic config values\",\n\t\t\tcmdline: `cadence admin config update --name test-dynamic-config-name --value \"{}\"`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t\t\t},\n\t\t\terrContains: \"Failed to update dynamic config value\",\n\t\t},\n\t\t{\n\t\t\tname:    \"update with simple boolean value\",\n\t\t\tcmdline: `cadence admin config update --name test.config --value ` + `'{\"Value\":true,\"Filters\":[]}'`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.UpdateDynamicConfigRequest, _ ...yarpc.CallOption) error {\n\t\t\t\t\t\tassert.Equal(t, \"test.config\", request.ConfigName)\n\t\t\t\t\t\tassert.Len(t, request.ConfigValues, 1)\n\n\t\t\t\t\t\t// Verify the value is correctly parsed\n\t\t\t\t\t\tvar actualValue interface{}\n\t\t\t\t\t\terr := json.Unmarshal(request.ConfigValues[0].Value.Data, &actualValue)\n\t\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\t\tassert.Equal(t, true, actualValue)\n\n\t\t\t\t\t\t// Verify no filters\n\t\t\t\t\t\tassert.Empty(t, request.ConfigValues[0].Filters)\n\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"update with map value and no filters\",\n\t\t\tcmdline: `cadence admin config update --name frontend.validSearchAttributes --value ` + `'{\"Value\":{\"DomainID\":1,\"WorkflowID\":1},\"Filters\":[]}'`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.UpdateDynamicConfigRequest, _ ...yarpc.CallOption) error {\n\t\t\t\t\t\tassert.Equal(t, \"frontend.validSearchAttributes\", request.ConfigName)\n\t\t\t\t\t\tassert.Len(t, request.ConfigValues, 1)\n\n\t\t\t\t\t\t// Verify the value is correctly parsed\n\t\t\t\t\t\tvar actualValue interface{}\n\t\t\t\t\t\terr := json.Unmarshal(request.ConfigValues[0].Value.Data, &actualValue)\n\t\t\t\t\t\tassert.NoError(t, err)\n\n\t\t\t\t\t\texpectedValue := map[string]interface{}{\n\t\t\t\t\t\t\t\"DomainID\":   float64(1), // JSON numbers unmarshal to float64\n\t\t\t\t\t\t\t\"WorkflowID\": float64(1),\n\t\t\t\t\t\t}\n\t\t\t\t\t\tassert.Equal(t, expectedValue, actualValue)\n\n\t\t\t\t\t\t// Verify no filters\n\t\t\t\t\t\tassert.Empty(t, request.ConfigValues[0].Filters)\n\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"update with map value and filters\",\n\t\t\tcmdline: `cadence admin config update --name frontend.validSearchAttributes --value ` + `'{\"Value\":{\"DomainID\":1,\"WorkflowID\":1},\"Filters\":[{\"Name\":\"domainName\",\"Value\":\"test-domain\"}]}'`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().UpdateDynamicConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(_ context.Context, request *types.UpdateDynamicConfigRequest, _ ...yarpc.CallOption) error {\n\t\t\t\t\t\tassert.Equal(t, \"frontend.validSearchAttributes\", request.ConfigName)\n\t\t\t\t\t\tassert.Len(t, request.ConfigValues, 1)\n\n\t\t\t\t\t\t// Verify the value is correctly parsed\n\t\t\t\t\t\tvar actualValue interface{}\n\t\t\t\t\t\terr := json.Unmarshal(request.ConfigValues[0].Value.Data, &actualValue)\n\t\t\t\t\t\tassert.NoError(t, err)\n\n\t\t\t\t\t\texpectedValue := map[string]interface{}{\n\t\t\t\t\t\t\t\"DomainID\":   float64(1),\n\t\t\t\t\t\t\t\"WorkflowID\": float64(1),\n\t\t\t\t\t\t}\n\t\t\t\t\t\tassert.Equal(t, expectedValue, actualValue)\n\n\t\t\t\t\t\t// Verify filters\n\t\t\t\t\t\tassert.Len(t, request.ConfigValues[0].Filters, 1)\n\t\t\t\t\t\tassert.Equal(t, \"domainName\", request.ConfigValues[0].Filters[0].Name)\n\n\t\t\t\t\t\tvar filterValue interface{}\n\t\t\t\t\t\terr = json.Unmarshal(request.ConfigValues[0].Filters[0].Value.Data, &filterValue)\n\t\t\t\t\t\tassert.NoError(t, err)\n\t\t\t\t\t\tassert.Equal(t, \"test-domain\", filterValue)\n\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t})\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.setupMocks(td)\n\n\t\t\terr := clitest.RunCommandLine(t, td.app, tt.cmdline)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminRestoreDynamicConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tcmdline     string\n\t\tsetupMocks  func(td *cliTestData)\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname:    \"no arguments provided\",\n\t\t\tcmdline: `cadence admin config restore --name ''`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\t// empty since args are missing\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname:    \"calling with required arguments\",\n\t\t\tcmdline: `cadence admin config restore --name test-dynamic-config-name --filter '{\"domainName\":\"test-domain\"}'`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().RestoreDynamicConfig(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname:    \"failed to update dynamic config values\",\n\t\t\tcmdline: `cadence admin config restore --name test-dynamic-config-name --filter '{\"Value\":\"some-value\",\"Filters\":[]}'`,\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().RestoreDynamicConfig(gomock.Any(), gomock.Any()).Return(assert.AnError)\n\t\t\t},\n\t\t\terrContains: \"Failed to restore dynamic config value\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.setupMocks(td)\n\n\t\t\terr := clitest.RunCommandLine(t, td.app, tt.cmdline)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminListDynamicConfig(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tsetupMocks  func(td *cliTestData)\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"failed with no dynamic config values stored to list\",\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().ListDynamicConfig(gomock.Any(), gomock.Any()).Return(nil, nil)\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to list dynamic config values\",\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().ListDynamicConfig(gomock.Any(), gomock.Any()).Return(nil, assert.AnError)\n\t\t\t},\n\t\t\terrContains: \"Failed to list dynamic config value(s)\",\n\t\t},\n\t\t{\n\t\t\tname: \"succeeded to list dynamic config values\",\n\t\t\tsetupMocks: func(td *cliTestData) {\n\t\t\t\ttd.mockAdminClient.EXPECT().ListDynamicConfig(gomock.Any(), gomock.Any()).Return(&types.ListDynamicConfigResponse{\n\t\t\t\t\tEntries: []*types.DynamicConfigEntry{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tName: \"test-dynamic-config-name\",\n\t\t\t\t\t\t\tValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\tFilters: []*types.DynamicConfigFilter{\n\t\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\tName: \"Filter1\",\n\t\t\t\t\t\t\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\t\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\t\t\t\t\t\t\tData:         []byte(\"filter-value\"),\n\t\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t}}, nil)\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.setupMocks(td)\n\n\t\t\terr := clitest.RunCommandLine(t, td.app, \"cadence admin config list\")\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminListConfigKeys(t *testing.T) {\n\tt.Run(\"list config keys\", func(t *testing.T) {\n\t\ttd := newCLITestData(t)\n\t\tassert.NoError(t, clitest.RunCommandLine(t, td.app, \"cadence admin config listall\"))\n\t})\n}\n"
  },
  {
    "path": "tools/cli/admin_db_clean_command.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/entity\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/executions\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// AdminDBClean is the command to clean up unhealthy executions.\n// Input is a JSON stream provided via STDIN or a file.\nfunc AdminDBClean(c *cli.Context) error {\n\tflagscantype, err := getRequiredOption(c, FlagScanType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tscanType, err := executions.ScanTypeString(flagscantype)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"unknown scan type\", err)\n\t}\n\tcollectionSlice := c.StringSlice(FlagInvariantCollection)\n\tblob := scanType.ToBlobstoreEntity()\n\n\tvar collections []invariant.Collection\n\tfor _, v := range collectionSlice {\n\t\tcollection, err := invariant.CollectionString(v)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"unknown invariant collection\", err)\n\t\t}\n\t\tcollections = append(collections, collection)\n\t}\n\n\tlogger := zap.NewNop()\n\tif c.Bool(FlagVerbose) {\n\t\tlogger, err = zap.NewDevelopment()\n\t\tif err != nil {\n\t\t\t// probably impossible with default config\n\t\t\treturn commoncli.Problem(\"could not construct logger\", err)\n\t\t}\n\t}\n\n\tinvariants := scanType.ToInvariants(collections, logger)\n\tif len(invariants) < 1 {\n\t\treturn commoncli.Problem(\n\t\t\tfmt.Sprintf(\"no invariants for scantype %q and collections %q\",\n\t\t\t\tscanType.String(),\n\t\t\t\tcollectionSlice),\n\t\t\tnil,\n\t\t)\n\t}\n\n\tinput, err := getInputFile(c.String(FlagInputFile))\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tdec := json.NewDecoder(input)\n\tvar data []*store.ScanOutputEntity\n\n\tfor {\n\t\tsoe := &store.ScanOutputEntity{\n\t\t\tExecution: blob.Clone(),\n\t\t}\n\n\t\tif err := dec.Decode(&soe); err != nil {\n\t\t\tif err == io.EOF || err == io.ErrUnexpectedEOF {\n\t\t\t\tbreak\n\t\t\t}\n\t\t} else {\n\t\t\tdata = append(data, soe)\n\t\t}\n\t}\n\n\tfor _, e := range data {\n\t\tresult, err := fixExecution(c, invariants, e)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in fix execution: \", err)\n\t\t}\n\t\tout := store.FixOutputEntity{\n\t\t\tExecution: e.Execution,\n\t\t\tInput:     *e,\n\t\t\tResult:    result,\n\t\t}\n\t\tdata, err := json.Marshal(out)\n\t\tif err != nil {\n\t\t\tfmt.Fprintln(os.Stderr, err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\toutput := getDeps(c).Output()\n\t\toutput.Write([]byte(string(data) + \"\\n\"))\n\t}\n\treturn nil\n}\n\nfunc fixExecution(\n\tc *cli.Context,\n\tinvariants []executions.InvariantFactory,\n\texecution *store.ScanOutputEntity,\n) (invariant.ManagerFixResult, error) {\n\texecManager, err := getDeps(c).initializeExecutionManager(c, execution.Execution.(entity.Entity).GetShardID())\n\tif err != nil {\n\t\treturn invariant.ManagerFixResult{}, err\n\t}\n\tdefer execManager.Close()\n\n\thistoryV2Mgr, err := getDeps(c).initializeHistoryManager(c)\n\tif err != nil {\n\t\treturn invariant.ManagerFixResult{}, err\n\t}\n\tdefer historyV2Mgr.Close()\n\n\tpr := persistence.NewPersistenceRetryer(\n\t\texecManager,\n\t\thistoryV2Mgr,\n\t\tcommon.CreatePersistenceRetryPolicy(),\n\t)\n\n\tvar ivs []invariant.Invariant\n\n\tfor _, fn := range invariants {\n\t\tivs = append(ivs, fn(pr, cache.NewNoOpDomainCache()))\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn invariant.ManagerFixResult{}, err\n\t}\n\tinvariantManager, err := getDeps(c).initializeInvariantManager(ivs)\n\tif err != nil {\n\t\treturn invariant.ManagerFixResult{}, err\n\t}\n\n\treturn invariantManager.RunFixes(ctx, execution.Execution.(entity.Entity)), nil\n}\n"
  },
  {
    "path": "tools/cli/admin_db_clean_command_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n)\n\nfunc TestAdminDBClean_noFixExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupContext   func(app *cli.App) *cli.Context\n\t\tinputFileData  string // Simulate the content of the input file\n\t\texpectedOutput string\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"SuccessCase_emptyResult\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define flags\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\t// Use a StringSliceFlag to simulate multiple collection values\n\t\t\t\tset.Var(cli.NewStringSlice(\"CollectionHistory\", \"CollectionDomain\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\t// Set actual values for the flags\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, \"input.json\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputFileData:  `[{\"Execution\": {\"ShardID\": 1}}]`, // Simulate the content of input file\n\t\t\texpectedOutput: ``,\n\t\t\texpectedError:  \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"MissingRequiredFlagScanType\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\t// Missing FlagScanType\n\t\t\t\t_ = set.Set(FlagInputFile, \"input.json\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputFileData:  ``,\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Required flag not found:\",\n\t\t},\n\t\t{\n\t\t\tname: \"UnknownScanType\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"unknown\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, \"input.json\"))\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputFileData:  ``,\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"unknown scan type\",\n\t\t},\n\t\t{\n\t\t\tname: \"InvalidInvariantCollection\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define FlagScanType and FlagInputFile\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\n\t\t\t\t// Simulate the collection slice with multiple collections (including an invalid one)\n\t\t\t\tset.Var(cli.NewStringSlice(\"invalid_collection\", \"history\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\t// Set actual values for the flags\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, \"input.json\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputFileData:  ``,\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"unknown invariant collection\",\n\t\t},\n\t\t{\n\t\t\tname: \"NoInvariantsError\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInvariantCollection, \"\", \"invariant collection flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, \"input.json\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInvariantCollection, \"invalid_collection\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputFileData:  `[{\"Execution\": {\"ShardID\": 1}}]`,\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"no invariants for scantype\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Create temp input file with the test's input data\n\t\t\tinputFile, err := os.CreateTemp(\"\", \"test_input_*.json\")\n\t\t\tassert.NoError(t, err)\n\t\t\tdefer os.Remove(inputFile.Name()) // Clean up after test\n\n\t\t\t// Write input data to the temp file\n\t\t\tif tt.inputFileData != \"\" {\n\t\t\t\t_, err = inputFile.WriteString(tt.inputFileData)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tinputFile.Close()\n\n\t\t\t// Create test IO handler to capture output\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Set up the CLI app\n\t\t\tapp := NewCliApp(nil, WithIOHandler(ioHandler))\n\n\t\t\t// Set up the CLI context\n\t\t\tc := tt.setupContext(app)\n\n\t\t\t// Overwrite the FlagInputFile with the actual temp file path\n\t\t\t_ = c.Set(FlagInputFile, inputFile.Name())\n\n\t\t\t// Call AdminDBClean and validate output and errors\n\t\t\terr = AdminDBClean(c)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\t// Validate the captured output\n\t\t\tassert.Contains(t, ioHandler.outputBytes.String(), tt.expectedOutput)\n\t\t})\n\t}\n}\n\nfunc TestAdminDBClean_inFixExecution(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcontextSetup   func(td *cliTestData, inputFilePath string) *cli.Context\n\t\tmockSetup      func(td *cliTestData)\n\t\tinputFileData  string\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tcontextSetup: func(td *cliTestData, inputFilePath string) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\tset.Var(cli.NewStringSlice(\"CollectionHistory\", \"CollectionDomain\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, inputFilePath))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\t\t\t\tmockInvariantManager := invariant.NewMockManager(ctrl)\n\n\t\t\t\tmockExecManager.EXPECT().Close().Times(1)\n\t\t\t\tmockHistoryManager.EXPECT().Close().Times(1)\n\n\t\t\t\tmockInvariantManager.EXPECT().RunFixes(gomock.Any(), gomock.Any()).Return(invariant.ManagerFixResult{}).Times(1)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), gomock.Any()).Return(mockExecManager, nil).Times(1)\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeHistoryManager(gomock.Any()).Return(mockHistoryManager, nil).Times(1)\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeInvariantManager(gomock.Any()).Return(mockInvariantManager, nil).Times(1)\n\n\t\t\t},\n\t\t\tinputFileData: `{\"Execution\": {\"ShardID\": 1}, \"Result\": {}}`,\n\t\t\texpectedOutput: `{\"Execution\":{\"BranchToken\":null,\"TreeID\":\"\",\"BranchID\":\"\",\"ShardID\":1,\"DomainID\":\"\",\"WorkflowID\":\"\",\"RunID\":\"\",\"State\":0},\"Input\":{\"Execution\":{\"BranchToken\":null,\"TreeID\":\"\",\"BranchID\":\"\",\"ShardID\":1,\"DomainID\":\"\",\"WorkflowID\":\"\",\"RunID\":\"\",\"State\":0},\"Result\":{\"CheckResultType\":\"\",\"DeterminingInvariantType\":null,\"CheckResults\":null}},\"Result\":{\"FixResultType\":\"\",\"DeterminingInvariantName\":null,\"FixResults\":null}}\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"init invariant manager error\",\n\t\t\tcontextSetup: func(td *cliTestData, inputFilePath string) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\tset.Var(cli.NewStringSlice(\"CollectionHistory\", \"CollectionDomain\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, inputFilePath))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\n\t\t\t\tmockExecManager.EXPECT().Close().Times(1)\n\t\t\t\tmockHistoryManager.EXPECT().Close().Times(1)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), gomock.Any()).Return(mockExecManager, nil).Times(1)\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeHistoryManager(gomock.Any()).Return(mockHistoryManager, nil).Times(1)\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeInvariantManager(gomock.Any()).Return(nil, fmt.Errorf(\"init invariant manager error\")).Times(1)\n\n\t\t\t},\n\t\t\tinputFileData:  `{\"Execution\": {\"ShardID\": 1}, \"Result\": {}}`,\n\t\t\texpectedOutput: ``,\n\t\t\texpectedError:  \"Error in fix execution: : init invariant manager error\",\n\t\t},\n\t\t{\n\t\t\tname: \"init execution manager error\",\n\t\t\tcontextSetup: func(td *cliTestData, inputFilePath string) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\tset.Var(cli.NewStringSlice(\"CollectionHistory\", \"CollectionDomain\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, inputFilePath))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"init execution manager error\")).Times(1)\n\t\t\t},\n\t\t\tinputFileData:  `{\"Execution\": {\"ShardID\": 1}, \"Result\": {}}`,\n\t\t\texpectedOutput: ``,\n\t\t\texpectedError:  \"Error in fix execution: : init execution manager error\",\n\t\t},\n\t\t{\n\t\t\tname: \"init history manager error\",\n\t\t\tcontextSetup: func(td *cliTestData, inputFilePath string) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagScanType, \"\", \"scan type flag\")\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\t\t\t\tset.Var(cli.NewStringSlice(\"CollectionHistory\", \"CollectionDomain\"), FlagInvariantCollection, \"invariant collection flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagScanType, \"ConcreteExecutionType\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, inputFilePath))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockExecManager.EXPECT().Close().Times(1)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), gomock.Any()).Return(mockExecManager, nil).Times(1)\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeHistoryManager(gomock.Any()).Return(nil, fmt.Errorf(\"init history manager error\")).Times(1)\n\t\t\t},\n\t\t\tinputFileData:  `{\"Execution\": {\"ShardID\": 1}, \"Result\": {}}`,\n\t\t\texpectedOutput: ``,\n\t\t\texpectedError:  \"Error in fix execution: : init history manager error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttt.mockSetup(td)\n\n\t\t\tinputFile, err := os.CreateTemp(\"\", \"test_input_*.json\")\n\t\t\tassert.NoError(t, err)\n\t\t\tdefer os.Remove(inputFile.Name())\n\n\t\t\tif tt.inputFileData != \"\" {\n\t\t\t\t_, err = inputFile.WriteString(tt.inputFileData)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t\tinputFile.Close()\n\n\t\t\tc := tt.contextSetup(td, inputFile.Name())\n\n\t\t\terr = AdminDBClean(c)\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_db_decode_thrift.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"encoding/base64\"\n\t\"encoding/hex\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/davecgh/go-spew/spew\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/.gen/go/config\"\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common/codec\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nvar decodingTypes = map[string]func() codec.ThriftObject{\n\t\"shared.History\":                      func() codec.ThriftObject { return &shared.History{} },\n\t\"shared.HistoryEvent\":                 func() codec.ThriftObject { return &shared.HistoryEvent{} },\n\t\"shared.HistoryBranch\":                func() codec.ThriftObject { return &shared.HistoryBranch{} },\n\t\"shared.Memo\":                         func() codec.ThriftObject { return &shared.Memo{} },\n\t\"shared.ResetPoints\":                  func() codec.ThriftObject { return &shared.ResetPoints{} },\n\t\"shared.BadBinaries\":                  func() codec.ThriftObject { return &shared.BadBinaries{} },\n\t\"shared.VersionHistories\":             func() codec.ThriftObject { return &shared.VersionHistories{} },\n\t\"replicator.FailoverMarkers\":          func() codec.ThriftObject { return &replicator.FailoverMarkers{} },\n\t\"history.ProcessingQueueStates\":       func() codec.ThriftObject { return &history.ProcessingQueueStates{} },\n\t\"config.DynamicConfigBlob\":            func() codec.ThriftObject { return &config.DynamicConfigBlob{} },\n\t\"sqlblobs.ShardInfo\":                  func() codec.ThriftObject { return &sqlblobs.ShardInfo{} },\n\t\"sqlblobs.DomainInfo\":                 func() codec.ThriftObject { return &sqlblobs.DomainInfo{} },\n\t\"sqlblobs.HistoryTreeInfo\":            func() codec.ThriftObject { return &sqlblobs.HistoryTreeInfo{} },\n\t\"sqlblobs.WorkflowExecutionInfo\":      func() codec.ThriftObject { return &sqlblobs.WorkflowExecutionInfo{} },\n\t\"sqlblobs.ActivityInfo\":               func() codec.ThriftObject { return &sqlblobs.ActivityInfo{} },\n\t\"sqlblobs.ChildExecutionInfo\":         func() codec.ThriftObject { return &sqlblobs.ChildExecutionInfo{} },\n\t\"sqlblobs.SignalInfo\":                 func() codec.ThriftObject { return &sqlblobs.SignalInfo{} },\n\t\"sqlblobs.RequestCancelInfo\":          func() codec.ThriftObject { return &sqlblobs.RequestCancelInfo{} },\n\t\"sqlblobs.TimerInfo\":                  func() codec.ThriftObject { return &sqlblobs.TimerInfo{} },\n\t\"sqlblobs.TaskInfo\":                   func() codec.ThriftObject { return &sqlblobs.TaskInfo{} },\n\t\"sqlblobs.TaskListInfo\":               func() codec.ThriftObject { return &sqlblobs.TaskListInfo{} },\n\t\"sqlblobs.TransferTaskInfo\":           func() codec.ThriftObject { return &sqlblobs.TransferTaskInfo{} },\n\t\"sqlblobs.TimerTaskInfo\":              func() codec.ThriftObject { return &sqlblobs.TimerTaskInfo{} },\n\t\"sqlblobs.ReplicationTaskInfo\":        func() codec.ThriftObject { return &sqlblobs.ReplicationTaskInfo{} },\n\t\"shared.AsyncWorkflowConfiguration\":   func() codec.ThriftObject { return &shared.AsyncWorkflowConfiguration{} },\n\t\"shared.ActiveClusters\":               func() codec.ThriftObject { return &shared.ActiveClusters{} },\n\t\"shared.ActiveClusterSelectionPolicy\": func() codec.ThriftObject { return &shared.ActiveClusterSelectionPolicy{} },\n}\n\ntype decodeError struct {\n\tshortMsg string\n\terr      error\n}\n\n// AdminDBDataDecodeThrift is the command to decode thrift binary into JSON\nfunc AdminDBDataDecodeThrift(c *cli.Context) error {\n\tinput, err := getRequiredOption(c, FlagInput)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tencoding := c.String(FlagInputEncoding)\n\tdata, err := decodeUserInput(input, encoding)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to decode input\", err)\n\t}\n\n\tif _, err := decodeThriftPayload(data); err != nil {\n\t\treturn commoncli.Problem(\"failed to decode thrift payload\", err.err)\n\t}\n\treturn nil\n}\n\nfunc decodeThriftPayload(data []byte) (codec.ThriftObject, *decodeError) {\n\tencoder := codec.NewThriftRWEncoder()\n\t// this is an inconsistency in the code base, some place use ThriftRWEncoder(version0Thriftrw.go) some use thriftEncoder(thrift_encoder.go)\n\tdataWithPrepend := []byte{0x59}\n\tdataWithPrepend = append(dataWithPrepend, data...)\n\tdatas := [][]byte{data, dataWithPrepend}\n\n\tfor _, data := range datas {\n\t\tfor typeName, objFn := range decodingTypes {\n\t\t\tt := objFn()\n\t\t\tif err := encoder.Decode(data, t); err != nil {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\t// encoding back to confirm\n\t\t\tdata2, err := encoder.Encode(t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, &decodeError{\n\t\t\t\t\tshortMsg: \"cannot encode back to confirm\",\n\t\t\t\t\terr:      err,\n\t\t\t\t}\n\t\t\t}\n\t\t\tif !bytes.Equal(data, data2) {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tfmt.Printf(\"======= Decode into type %v ========\\n\", typeName)\n\t\t\tspew.Dump(t)\n\t\t\t// json-ify it for easier mechanical use\n\t\t\tjs, err := json.Marshal(t)\n\t\t\tif err == nil {\n\t\t\t\tfmt.Println(\"======= As JSON ========\")\n\t\t\t\tfmt.Println(string(js))\n\t\t\t}\n\t\t\treturn t, nil\n\t\t}\n\t}\n\n\treturn nil, &decodeError{\n\t\tshortMsg: \"input data cannot be decoded into any struct\",\n\t\terr:      nil,\n\t}\n}\n\nfunc decodeUserInput(input, encoding string) ([]byte, error) {\n\tswitch encoding {\n\tcase \"\", \"hex\":\n\t\t// remove \"0x\" from the beginning of the input. hex library doesn't expect it but that's how it's printed it out by csql\n\t\tinput = strings.TrimPrefix(input, \"0x\")\n\t\treturn hex.DecodeString(input)\n\tcase \"base64\":\n\t\treturn base64.StdEncoding.DecodeString(input)\n\t}\n\n\treturn nil, fmt.Errorf(\"unknown input encoding: %s\", encoding)\n}\n"
  },
  {
    "path": "tools/cli/admin_db_decode_thrift_test.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\n\t\"github.com/uber/cadence/.gen/go/history\"\n\t\"github.com/uber/cadence/.gen/go/shared\"\n\t\"github.com/uber/cadence/.gen/go/sqlblobs\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/codec\"\n)\n\nfunc TestThriftDecodeHelper(t *testing.T) {\n\ttests := []struct {\n\t\tdesc      string\n\t\tinput     string\n\t\tencoding  string\n\t\twantObjFn func(*testing.T) codec.ThriftObject\n\t\twantErr   bool\n\t}{\n\t\t{\n\t\t\tdesc:     \"invalid base64 input\",\n\t\t\tinput:    \"not-a-valid-base64\",\n\t\t\tencoding: \"base64\",\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tdesc:     \"invalid hex input\",\n\t\t\tinput:    \"not-a-valid-hex\",\n\t\t\tencoding: \"hex\",\n\t\t\twantErr:  true,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"VersionHistories hex with '0x' prefix\",\n\t\t\tinput:     \"0x5908000a000000000f00140c000000010b000a0000004c590b000a000000067472656569640b0014000000086272616e636869640f001e0c000000010b000a000000086272616e636869640a001400000000000000010a001e000000000000006400000000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestVersionHistories,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"VersionHistories hex without '0x' prefix\",\n\t\t\tinput:     \"5908000a000000000f00140c000000010b000a0000004c590b000a000000067472656569640b0014000000086272616e636869640f001e0c000000010b000a000000086272616e636869640a001400000000000000010a001e000000000000006400000000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestVersionHistories,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"VersionHistories base64\",\n\t\t\tinput:     \"WQgACgAAAAAPABQMAAAAAQsACgAAAExZCwAKAAAABnRyZWVpZAsAFAAAAAhicmFuY2hpZA8AHgwAAAABCwAKAAAACGJyYW5jaGlkCgAUAAAAAAAAAAEKAB4AAAAAAAAAZAAAAAA=\",\n\t\t\tencoding:  \"base64\",\n\t\t\twantObjFn: generateTestVersionHistories,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"ResetPoints hex\",\n\t\t\tinput:     \"590f000a0c000000010b000a00000008636865636b73756d0b00140000000572756e69640a001e00000000000000010a002800000000000000010a0032000000000000000102003c010000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestResetPoints,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"ProcessingQueueStates hex\",\n\t\t\tinput:     \"590d000a0b0f0000000100000008636c7573746572310c0000000208000a000000000a001400000000000003e80a001e00000000000007d00c00280f000a0b0000000100000006646f6d61696e000008000a000000010a0014000000000000012c0a001e00000000000001900c00280f000a0b0000000100000006646f6d61696e000000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestProcessingQueueStates,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"DomainInfo hex\",\n\t\t\tinput:     \"590b000a000000046e616d650b000c0000000b6465736372697074696f6e0b000e000000056f776e65720800100000000306001200070d00260b0b0000000100000007646174616b6579000000096461746176616c756500\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestDomainInfo,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"HistoryTreeInfo hex\",\n\t\t\tinput:     \"590a000a0000000218711a000f000c0c000000010b000a000000086272616e636869640a001400000000000000010a001e0000000000000064000b000e00000009736f6d6520696e666f00\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestHistoryTreeInfo,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"TimerInfo hex\",\n\t\t\tinput:     \"590a000a00000000000000010a000c00000000000000010a000e00000000000003e80a0010000000000000000500\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestTimerInfo,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"Active clusters config\",\n\t\t\tinput:     \"0x590d000b0b0c00000001000000086c6f636174696f6e0d000a0b0c00000001000000066c6f6e646f6e0b000a00000008636c7573746572320a00140000000000000002000000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestActiveClustersConfig,\n\t\t},\n\t\t{\n\t\t\tdesc:      \"Active cluster selection policy\",\n\t\t\tinput:     \"5908000a000000000b001400000007726567696f6e300b001e000000000b00280000000000\",\n\t\t\tencoding:  \"hex\",\n\t\t\twantObjFn: generateTestActiveClusterSelectionPolicy,\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\ttc := tc\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tt.Parallel()\n\t\t\tdata, err := decodeUserInput(tc.input, tc.encoding)\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"decodeUserInput() error: %v, wantErr: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\tgotObj, decodeErr := decodeThriftPayload(data)\n\t\t\tif (decodeErr != nil) != tc.wantErr {\n\t\t\t\tt.Fatalf(\"decodeUserInput() error: %v, wantErr: %v\", decodeErr, tc.wantErr)\n\t\t\t}\n\t\t\tif decodeErr != nil {\n\t\t\t\treturn\n\t\t\t}\n\n\t\t\twant := tc.wantObjFn(t)\n\n\t\t\tif diff := cmp.Diff(want, gotObj); diff != \"\" {\n\t\t\t\tt.Fatalf(\"Object mismatch (-want +got):\\n%s\", diff)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mustThriftEncode(t *testing.T, obj codec.ThriftObject) []byte {\n\tt.Helper()\n\tdata, err := codec.NewThriftRWEncoder().Encode(obj)\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to encode thrift obj, err: %v\", err)\n\t}\n\treturn data\n}\n\nfunc generateTestVersionHistories(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\thistoryBranch := &shared.HistoryBranch{\n\t\tTreeID:   common.StringPtr(\"treeid\"),\n\t\tBranchID: common.StringPtr(\"branchid\"),\n\t\tAncestors: []*shared.HistoryBranchRange{\n\t\t\t{\n\t\t\t\tBranchID:    common.StringPtr(\"branchid\"),\n\t\t\t\tBeginNodeID: common.Int64Ptr(1),\n\t\t\t\tEndNodeID:   common.Int64Ptr(100),\n\t\t\t},\n\t\t},\n\t}\n\treturn &shared.VersionHistories{\n\t\tCurrentVersionHistoryIndex: common.Int32Ptr(0),\n\t\tHistories: []*shared.VersionHistory{\n\t\t\t{\n\t\t\t\tBranchToken: mustThriftEncode(t, historyBranch),\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateTestResetPoints(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &shared.ResetPoints{\n\t\tPoints: []*shared.ResetPointInfo{\n\t\t\t{\n\t\t\t\tBinaryChecksum:           common.StringPtr(\"checksum\"),\n\t\t\t\tRunId:                    common.StringPtr(\"runid\"),\n\t\t\t\tFirstDecisionCompletedId: common.Int64Ptr(1),\n\t\t\t\tCreatedTimeNano:          common.Int64Ptr(1),\n\t\t\t\tResettable:               common.BoolPtr(true),\n\t\t\t\tExpiringTimeNano:         common.Int64Ptr(1),\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateTestProcessingQueueStates(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &history.ProcessingQueueStates{\n\t\tStatesByCluster: map[string][]*history.ProcessingQueueState{\n\t\t\t\"cluster1\": {\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(0),\n\t\t\t\t\tAckLevel: common.Int64Ptr(1000),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(2000),\n\t\t\t\t\tDomainFilter: &history.DomainFilter{\n\t\t\t\t\t\tDomainIDs: []string{\"domain\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tLevel:    common.Int32Ptr(1),\n\t\t\t\t\tAckLevel: common.Int64Ptr(300),\n\t\t\t\t\tMaxLevel: common.Int64Ptr(400),\n\t\t\t\t\tDomainFilter: &history.DomainFilter{\n\t\t\t\t\t\tDomainIDs: []string{\"domain\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n\nfunc generateTestDomainInfo(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &sqlblobs.DomainInfo{\n\t\tName:          common.StringPtr(\"name\"),\n\t\tOwner:         common.StringPtr(\"owner\"),\n\t\tStatus:        common.Int32Ptr(3),\n\t\tRetentionDays: common.Int16Ptr(7),\n\t\tDescription:   common.StringPtr(\"description\"),\n\t\tData: map[string]string{\n\t\t\t\"datakey\": \"datavalue\",\n\t\t},\n\t}\n}\n\nfunc generateTestHistoryTreeInfo(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &sqlblobs.HistoryTreeInfo{\n\t\tCreatedTimeNanos: common.Int64Ptr(9000000000),\n\t\tAncestors: []*shared.HistoryBranchRange{\n\t\t\t{\n\t\t\t\tBranchID:    common.StringPtr(\"branchid\"),\n\t\t\t\tBeginNodeID: common.Int64Ptr(1),\n\t\t\t\tEndNodeID:   common.Int64Ptr(100),\n\t\t\t},\n\t\t},\n\t\tInfo: common.StringPtr(\"some info\"),\n\t}\n}\n\nfunc generateTestTimerInfo(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &sqlblobs.TimerInfo{\n\t\tVersion:         common.Int64Ptr(1),\n\t\tStartedID:       common.Int64Ptr(1),\n\t\tExpiryTimeNanos: common.Int64Ptr(1000),\n\t\tTaskID:          common.Int64Ptr(5),\n\t}\n}\n\nfunc generateTestActiveClusterSelectionPolicy(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &shared.ActiveClusterSelectionPolicy{\n\t\tStrategy:           shared.ActiveClusterSelectionStrategyRegionSticky.Ptr(),\n\t\tStickyRegion:       common.StringPtr(\"region0\"),\n\t\tExternalEntityType: common.StringPtr(\"\"),\n\t\tExternalEntityKey:  common.StringPtr(\"\"),\n\t}\n}\n\nfunc generateTestActiveClustersConfig(t *testing.T) codec.ThriftObject {\n\tt.Helper()\n\treturn &shared.ActiveClusters{\n\t\tActiveClustersByClusterAttribute: map[string]*shared.ClusterAttributeScope{\n\t\t\t\"location\": &shared.ClusterAttributeScope{\n\t\t\t\tClusterAttributes: map[string]*shared.ActiveClusterInfo{\n\t\t\t\t\t\"london\": {\n\t\t\t\t\t\tActiveClusterName: common.Ptr(\"cluster2\"),\n\t\t\t\t\t\tFailoverVersion:   common.Int64Ptr(2),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_db_scan_command.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/zap\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/cache\"\n\t\"github.com/uber/cadence/common/collection\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/reconciliation/fetcher\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/common/reconciliation/store\"\n\t\"github.com/uber/cadence/service/worker/scanner/executions\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tlistContextTimeout = time.Minute\n)\n\n// AdminDBScan is used to scan over executions in database and detect corruptions.\nfunc AdminDBScan(c *cli.Context) error {\n\tscanType, err := executions.ScanTypeString(c.String(FlagScanType))\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"unknown scan type\", err)\n\t}\n\n\tnumberOfShards, err := getRequiredIntOption(c, FlagNumberOfShards)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tcollectionSlice := c.StringSlice(FlagInvariantCollection)\n\n\tvar collections []invariant.Collection\n\tfor _, v := range collectionSlice {\n\t\tcollection, err := invariant.CollectionString(v)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"unknown invariant collection\", err)\n\t\t}\n\t\tcollections = append(collections, collection)\n\t}\n\n\tlogger := zap.NewNop()\n\tif c.Bool(FlagVerbose) {\n\t\tlogger, err = zap.NewDevelopment()\n\t\tif err != nil {\n\t\t\t// probably impossible with default config\n\t\t\treturn commoncli.Problem(\"could not construct logger\", err)\n\t\t}\n\t}\n\n\tinvariants := scanType.ToInvariants(collections, logger)\n\tif len(invariants) < 1 {\n\t\treturn commoncli.Problem(\n\t\t\tfmt.Sprintf(\"no invariants for scan type %q and collections %q\",\n\t\t\t\tscanType.String(),\n\t\t\t\tcollectionSlice),\n\t\t\tnil,\n\t\t)\n\t}\n\tef := scanType.ToExecutionFetcher()\n\n\tinput, err := getInputFile(c.String(FlagInputFile))\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Input file not found\", err)\n\t}\n\tdec := json.NewDecoder(input)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"\", err)\n\t}\n\tvar data []fetcher.ExecutionRequest\n\n\tfor {\n\t\tvar exec fetcher.ExecutionRequest\n\t\tif err := dec.Decode(&exec); err != nil {\n\t\t\tif err == io.EOF || err == io.ErrUnexpectedEOF {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\treturn commoncli.Problem(\"Error decoding input file\", err)\n\t\t}\n\t\tdata = append(data, exec)\n\t}\n\n\tif len(data) == 0 {\n\t\treturn commoncli.Problem(\"Input file contained no data to scan\", nil)\n\t}\n\n\tfor _, e := range data {\n\t\texecution, result, err := checkExecution(c, numberOfShards, e, invariants, ef)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Execution check failed\", err)\n\t\t}\n\t\tout := store.ScanOutputEntity{\n\t\t\tExecution: execution,\n\t\t\tResult:    result,\n\t\t}\n\t\tdata, err := json.Marshal(out)\n\t\tif err != nil {\n\t\t\tfmt.Fprintln(os.Stderr, err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tgetDeps(c).Output().Write(data)\n\t}\n\treturn nil\n}\n\nfunc checkExecution(\n\tc *cli.Context,\n\tnumberOfShards int,\n\treq fetcher.ExecutionRequest,\n\tinvariants []executions.InvariantFactory,\n\tfetcher executions.ExecutionFetcher,\n) (interface{}, invariant.ManagerCheckResult, error) {\n\texecManager, err := getDeps(c).initializeExecutionManager(c, common.WorkflowIDToHistoryShard(req.WorkflowID, numberOfShards))\n\tif err != nil {\n\t\treturn nil, invariant.ManagerCheckResult{}, fmt.Errorf(\"initialize execution manager: %w\", err)\n\t}\n\tdefer execManager.Close()\n\n\thistoryV2Mgr, err := getDeps(c).initializeHistoryManager(c)\n\tif err != nil {\n\t\treturn nil, invariant.ManagerCheckResult{}, fmt.Errorf(\"initialize history manager: %w\", err)\n\t}\n\tdefer historyV2Mgr.Close()\n\n\tpr := persistence.NewPersistenceRetryer(\n\t\texecManager,\n\t\thistoryV2Mgr,\n\t\tcommon.CreatePersistenceRetryPolicy(),\n\t)\n\n\tctx, cancel, err := newContext(c)\n\tif err != nil {\n\t\treturn nil, invariant.ManagerCheckResult{}, fmt.Errorf(\"Error in creating context: %w\", err)\n\t}\n\tdefer cancel()\n\texecution, err := fetcher(ctx, pr, req)\n\n\tif err != nil {\n\t\treturn nil, invariant.ManagerCheckResult{}, fmt.Errorf(\"fetching execution: %w\", err)\n\t}\n\n\tvar ivs []invariant.Invariant\n\n\tfor _, fn := range invariants {\n\t\tivs = append(ivs, fn(pr, cache.NewNoOpDomainCache()))\n\t}\n\n\treturn execution, invariant.NewInvariantManager(ivs).RunChecks(ctx, execution), nil\n}\n\n// AdminDBScanUnsupportedWorkflow is to scan DB for unsupported workflow for a new release\nfunc AdminDBScanUnsupportedWorkflow(c *cli.Context) error {\n\toutputFile, err := getOutputFile(c.String(FlagOutputFilename))\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in admin db scan unsupported wf: \", err)\n\t}\n\tstartShardID := c.Int(FlagLowerShardBound)\n\tendShardID := c.Int(FlagUpperShardBound)\n\n\tdefer outputFile.Close()\n\tfor i := startShardID; i <= endShardID; i++ {\n\t\tif err := listExecutionsByShardID(c, i, outputFile); err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfmt.Printf(\"Shard %v scan operation is completed.\\n\", i)\n\t}\n\treturn nil\n}\n\nfunc listExecutionsByShardID(\n\tc *cli.Context,\n\tshardID int,\n\toutputFile *os.File,\n) error {\n\n\tclient, err := getDeps(c).initializeExecutionManager(c, shardID)\n\tif err != nil {\n\t\tcommoncli.Problem(\"initialize execution manager:\", err)\n\t}\n\tdefer client.Close()\n\tpaginationFunc := func(paginationToken []byte) ([]interface{}, []byte, error) {\n\t\tctx, cancel := context.WithTimeout(c.Context, listContextTimeout)\n\t\tdefer cancel()\n\n\t\tresp, err := client.ListConcreteExecutions(\n\t\t\tctx,\n\t\t\t&persistence.ListConcreteExecutionsRequest{\n\t\t\t\tPageSize:  1000,\n\t\t\t\tPageToken: paginationToken,\n\t\t\t},\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\tvar paginateItems []interface{}\n\t\tfor _, history := range resp.Executions {\n\t\t\tpaginateItems = append(paginateItems, history)\n\t\t}\n\t\treturn paginateItems, resp.PageToken, nil\n\t}\n\n\texecutionIterator := collection.NewPagingIterator(paginationFunc)\n\tfor executionIterator.HasNext() {\n\t\tresult, err := executionIterator.Next()\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Failed to scan shard ID: %v for unsupported workflow. Please retry.\", shardID), err)\n\t\t}\n\t\texecution := result.(*persistence.ListConcreteExecutionsEntity)\n\t\texecutionInfo := execution.ExecutionInfo\n\t\tif executionInfo != nil && executionInfo.CloseStatus == 0 && execution.VersionHistories == nil {\n\n\t\t\toutStr := fmt.Sprintf(\"cadence --address <host>:<port> --domain <%v> workflow reset --wid %v --rid %v --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\\n\",\n\t\t\t\texecutionInfo.DomainID,\n\t\t\t\texecutionInfo.WorkflowID,\n\t\t\t\texecutionInfo.RunID,\n\t\t\t)\n\t\t\tif _, err = outputFile.WriteString(outStr); err != nil {\n\t\t\t\treturn commoncli.Problem(\"Failed to write data to file\", err)\n\t\t\t}\n\t\t\tif err = outputFile.Sync(); err != nil {\n\t\t\t\treturn commoncli.Problem(\"Failed to sync data to file\", err)\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "tools/cli/admin_db_scan_command_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestAdminDBScanErrorCases(t *testing.T) {\n\tcases := []struct {\n\t\tname        string\n\t\ttestSetup   func(td *cliTestData) *cli.Context\n\t\terrContains string // empty if no error is expected\n\t}{\n\t\t{\n\t\t\tname: \"scan type not provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app)\n\t\t\t},\n\t\t\terrContains: \"unknown scan type\",\n\t\t},\n\t\t{\n\t\t\tname: \"unknown scan type provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"some_unknown_scan_type\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"unknown scan type\",\n\t\t},\n\t\t{\n\t\t\tname: \"number of shards not provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"invariant collection not provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"no invariants for scan type \\\"ConcreteExecutionType\\\" and collections []\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid invariant collection provided\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"some_unknown_invariant_collection\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"unknown invariant collection: some_unknown_invariant_collection\",\n\t\t},\n\t\t{\n\t\t\tname: \"input file not found\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/non-existant-file.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Input file not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"input file is empty\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input_empty.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Input file contained no data to scan\",\n\t\t},\n\t\t{\n\t\t\tname: \"bad data in input file\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input_bad_data.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Error decoding input file\",\n\t\t},\n\t\t{\n\t\t\tname: \"execution manager initialization error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeExecutionManager(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, assert.AnError)\n\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Execution check failed: initialize execution manager: assert.AnError general error for testing\",\n\t\t},\n\t\t{\n\t\t\tname: \"historyV2 manager initialization error\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tshardID1 := common.WorkflowIDToHistoryShard(\"test-workflow-id1\", 16384)\n\t\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(td.ctrl)\n\t\t\t\tmockExecutionManager.EXPECT().Close()\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeExecutionManager(gomock.Any(), shardID1).\n\t\t\t\t\tReturn(mockExecutionManager, nil)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeHistoryManager(gomock.Any()).\n\t\t\t\t\tReturn(nil, assert.AnError)\n\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Execution check failed: initialize history manager: assert.AnError general error for testing\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to fetch execution\",\n\t\t\ttestSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tshardID1 := common.WorkflowIDToHistoryShard(\"test-workflow-id1\", 16384)\n\t\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(td.ctrl)\n\t\t\t\tmockExecutionManager.EXPECT().Close()\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeExecutionManager(gomock.Any(), shardID1).\n\t\t\t\t\tReturn(mockExecutionManager, nil)\n\n\t\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(td.ctrl)\n\t\t\t\tmockHistoryManager.EXPECT().Close()\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeHistoryManager(gomock.Any()).\n\t\t\t\t\tReturn(mockHistoryManager, nil)\n\n\t\t\t\tmockExecutionManager.EXPECT().GetWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, assert.AnError)\n\n\t\t\t\treturn clitest.NewCLIContext(t, td.app,\n\t\t\t\t\tclitest.StringArgument(\"scan_type\", \"ConcreteExecutionType\"),\n\t\t\t\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\t\t\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionHistory\"),\n\t\t\t\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input.json\"),\n\t\t\t\t)\n\t\t\t},\n\t\t\terrContains: \"Execution check failed: fetching execution: assert.AnError general error for testing\",\n\t\t},\n\t}\n\n\tfor _, tc := range cases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\ttestCtx := tc.testSetup(td)\n\t\t\terr := AdminDBScan(testCtx)\n\t\t\tif tc.errContains != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.ErrorContains(t, err, tc.errContains)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminDBScan(t *testing.T) {\n\ttd := newCLITestData(t)\n\n\texpectWorkFlow(td, \"test-workflow-id1\")\n\texpectWorkFlow(td, \"test-workflow-id2\")\n\texpectWorkFlow(td, \"test-workflow-id3\")\n\n\tcliCtx := clitest.NewCLIContext(t, td.app,\n\t\tclitest.StringArgument(\"scan_type\", \"CurrentExecutionType\"),\n\t\tclitest.IntArgument(\"number_of_shards\", 16384),\n\t\tclitest.StringSliceArgument(\"invariant_collection\", \"CollectionMutableState\"),\n\t\tclitest.StringArgument(\"input_file\", \"testdata/scan_input.json\"),\n\t)\n\n\terr := AdminDBScan(cliCtx)\n\tassert.NoError(t, err)\n\n\tassert.Equal(t, expectedAdminDBScanOutput, td.ioHandler.outputBytes.String())\n}\n\n// The expected output does not have any newlines or tabs\n// so we use strings.Join(strings.Fields()) to remove them\nvar expectedAdminDBScanOutput = strings.Join(strings.Fields(`\n{\n\t\"Execution\":{\n\t\t\"CurrentRunID\":\"test-run-id1\",\n\t\t\"ShardID\":8946,\n\t\t\"DomainID\":\"test-domain-id1\",\n\t\t\"WorkflowID\":\"test-workflow-id1\",\n\t\t\"RunID\":\"test-run-id1\",\n\t\t\"State\":2\n\t},\n\t\"Result\":{\n\t\t\"CheckResultType\":\"healthy\",\n\t\t\"DeterminingInvariantType\":null,\n\t\t\"CheckResults\":[{\"CheckResultType\":\"healthy\",\"InvariantName\":\"concrete_execution_exists\",\"Info\":\"\",\"InfoDetails\":\"\"}]\n\t}\n}{\n\t\"Execution\":{\n\t\t\"CurrentRunID\":\"test-run-id1\",\n\t\t\"ShardID\":14767,\n\t\t\"DomainID\":\"test-domain-id2\",\n\t\t\"WorkflowID\":\"test-workflow-id2\",\n\t\t\"RunID\":\"test-run-id1\",\n\t\t\"State\":2\n\t},\n\t\"Result\":{\n\t\t\"CheckResultType\":\"healthy\",\n\t\t\"DeterminingInvariantType\":null,\n\t\t\"CheckResults\":[{\"CheckResultType\":\"healthy\",\"InvariantName\":\"concrete_execution_exists\",\"Info\":\"\",\"InfoDetails\":\"\"}]\n\t}\n}{\n\t\"Execution\":{\n\t\t\"CurrentRunID\":\"test-run-id1\",\n\t\t\"ShardID\":14582,\n\t\t\"DomainID\":\"test-domain-id3\",\n\t\t\"WorkflowID\":\"test-workflow-id3\",\n\t\t\"RunID\":\"test-run-id1\",\"State\":2\n\t},\n\t\"Result\":{\n\t\t\"CheckResultType\":\"healthy\",\n\t\t\"DeterminingInvariantType\":null,\n\t\t\"CheckResults\":[{\"CheckResultType\":\"healthy\",\"InvariantName\":\"concrete_execution_exists\",\"Info\":\"\",\"InfoDetails\":\"\"}]\n\t}\n}`,\n), \"\")\n\nfunc expectWorkFlow(td *cliTestData, workflowID string) {\n\tshardID1 := common.WorkflowIDToHistoryShard(workflowID, 16384)\n\tmockExecutionManager := persistence.NewMockExecutionManager(td.ctrl)\n\tmockExecutionManager.EXPECT().Close().Times(1)\n\ttd.mockManagerFactory.EXPECT().\n\t\tinitializeExecutionManager(gomock.Any(), shardID1).\n\t\tReturn(mockExecutionManager, nil).\n\t\tTimes(1)\n\n\tmockHistoryManager := persistence.NewMockHistoryManager(td.ctrl)\n\tmockHistoryManager.EXPECT().Close().Times(1)\n\ttd.mockManagerFactory.EXPECT().\n\t\tinitializeHistoryManager(gomock.Any()).\n\t\tReturn(mockHistoryManager, nil).\n\t\tTimes(1)\n\n\tmockExecutionManager.EXPECT().GetCurrentExecution(gomock.Any(), gomock.Any()).\n\t\tReturn(&persistence.GetCurrentExecutionResponse{\n\t\t\tRunID: \"test-run-id1\",\n\t\t\tState: persistence.WorkflowStateCompleted,\n\t\t}, nil).\n\t\tTimes(1)\n\tmockExecutionManager.EXPECT().GetShardID().Return(shardID1).Times(1)\n\tmockExecutionManager.EXPECT().IsWorkflowExecutionExists(gomock.Any(), gomock.Any()).\n\t\tReturn(&persistence.IsWorkflowExecutionExistsResponse{\n\t\t\tExists: true,\n\t\t}, nil).\n\t\tTimes(1)\n}\n\nfunc TestAdminDBScanUnsupportedWorkflow(t *testing.T) {\n\ttd := newCLITestData(t)\n\n\toutPutFile := createTempFileWithContent(t, \"\")\n\n\texpectShard(td, 123)\n\texpectShard(td, 124)\n\texpectShard(td, 125)\n\n\tcliCtx := clitest.NewCLIContext(t, td.app,\n\t\tclitest.StringArgument(\"output_filename\", outPutFile),\n\t\tclitest.IntArgument(\"lower_shard_bound\", 123),\n\t\tclitest.IntArgument(\"upper_shard_bound\", 125),\n\t)\n\n\terr := AdminDBScanUnsupportedWorkflow(cliCtx)\n\tassert.NoError(t, err)\n\n\tactual, err := os.ReadFile(outPutFile)\n\trequire.NoError(t, err)\n\tassert.Equal(t, expectedAdminDBScanUnsupportedOutput, string(actual))\n}\n\nfunc expectShard(td *cliTestData, shardID int) {\n\tmockExecutionManager := persistence.NewMockExecutionManager(td.ctrl)\n\tmockExecutionManager.EXPECT().Close().Times(1)\n\n\t// Return 2 executions in the first call and a page token\n\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(),\n\t\t&persistence.ListConcreteExecutionsRequest{\n\t\t\tPageSize:  1000,\n\t\t\tPageToken: nil,\n\t\t},\n\t).\n\t\tReturn(&persistence.ListConcreteExecutionsResponse{\n\t\t\tExecutions: []*persistence.ListConcreteExecutionsEntity{\n\t\t\t\tcreateListConcreteExecutionsEntity(1, shardID),\n\t\t\t\tcreateListConcreteExecutionsEntity(2, shardID),\n\t\t\t},\n\t\t\tPageToken: []byte(\"some-next-page-token\"),\n\t\t}, nil).Times(1)\n\n\t// Return 1 execution in the second call and no page token\n\tmockExecutionManager.EXPECT().ListConcreteExecutions(gomock.Any(),\n\t\t&persistence.ListConcreteExecutionsRequest{\n\t\t\tPageSize:  1000,\n\t\t\tPageToken: []byte(\"some-next-page-token\"),\n\t\t},\n\t).\n\t\tReturn(&persistence.ListConcreteExecutionsResponse{\n\t\t\tExecutions: []*persistence.ListConcreteExecutionsEntity{\n\t\t\t\tcreateListConcreteExecutionsEntity(3, shardID),\n\t\t\t},\n\t\t\tPageToken: nil,\n\t\t}, nil).Times(1)\n\n\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), shardID).\n\t\tReturn(mockExecutionManager, nil).Times(1)\n}\n\nfunc createListConcreteExecutionsEntity(number int, shardID int) *persistence.ListConcreteExecutionsEntity {\n\treturn &persistence.ListConcreteExecutionsEntity{\n\t\tExecutionInfo: &persistence.WorkflowExecutionInfo{\n\t\t\tDomainID:   fmt.Sprintf(\"%d-test-domain-id%d\", shardID, number),\n\t\t\tWorkflowID: fmt.Sprintf(\"%d-test-workflow-id%d\", shardID, number),\n\t\t\tRunID:      fmt.Sprintf(\"%d-test-run-id%d\", shardID, number),\n\t\t},\n\t\tVersionHistories: nil,\n\t}\n}\n\nconst expectedAdminDBScanUnsupportedOutput = `cadence --address <host>:<port> --domain <123-test-domain-id1> workflow reset --wid 123-test-workflow-id1 --rid 123-test-run-id1 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <123-test-domain-id2> workflow reset --wid 123-test-workflow-id2 --rid 123-test-run-id2 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <123-test-domain-id3> workflow reset --wid 123-test-workflow-id3 --rid 123-test-run-id3 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <124-test-domain-id1> workflow reset --wid 124-test-workflow-id1 --rid 124-test-run-id1 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <124-test-domain-id2> workflow reset --wid 124-test-workflow-id2 --rid 124-test-run-id2 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <124-test-domain-id3> workflow reset --wid 124-test-workflow-id3 --rid 124-test-run-id3 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <125-test-domain-id1> workflow reset --wid 125-test-workflow-id1 --rid 125-test-run-id1 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <125-test-domain-id2> workflow reset --wid 125-test-workflow-id2 --rid 125-test-run-id2 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\ncadence --address <host>:<port> --domain <125-test-domain-id3> workflow reset --wid 125-test-workflow-id3 --rid 125-test-run-id3 --reset_type LastDecisionCompleted --reason 'release 0.16 upgrade'\n`\n"
  },
  {
    "path": "tools/cli/admin_dlq_commands.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tdefaultPageSize = 1000\n)\n\ntype DLQRow struct {\n\tShardID         int                        `header:\"Shard ID\" json:\"shardID\"`\n\tDomainName      string                     `header:\"Domain Name\" json:\"domainName\"`\n\tDomainID        string                     `header:\"Domain ID\" json:\"domainID\"`\n\tWorkflowID      string                     `header:\"Workflow ID\" json:\"workflowID\"`\n\tRunID           string                     `header:\"Run ID\" json:\"runID\"`\n\tTaskID          int64                      `header:\"Task ID\" json:\"taskID\"`\n\tTaskType        *types.ReplicationTaskType `header:\"Task Type\" json:\"taskType\"`\n\tVersion         int64                      `json:\"version\"`\n\tFirstEventID    int64                      `json:\"firstEventID\"`\n\tNextEventID     int64                      `json:\"nextEventID\"`\n\tScheduledID     int64                      `json:\"scheduledID\"`\n\tReplicationTask *types.ReplicationTask     `json:\"replicationTask\"`\n\n\t// Those are deserialized variants from history replications task\n\tEvents       []*types.HistoryEvent `json:\"events\"`\n\tNewRunEvents []*types.HistoryEvent `json:\"newRunEvents,omitempty\"`\n\n\t// Only event IDs for compact table representation\n\tEventIDs       []int64 `header:\"Event IDs\"`\n\tNewRunEventIDs []int64 `header:\"New Run Event IDs\"`\n}\n\ntype HistoryDLQCountRow struct {\n\tSourceCluster string `header:\"Source Cluster\" json:\"sourceCluster\"`\n\tShardID       int32  `header:\"Shard ID\" json:\"shardID\"`\n\tCount         int64  `header:\"Count\" json:\"count\"`\n}\n\n// AdminCountDLQMessages returns info how many and where DLQ messages are queued\nfunc AdminCountDLQMessages(c *cli.Context) error {\n\tforce := c.Bool(FlagForce)\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresponse, err := adminClient.CountDLQMessages(ctx, &types.CountDLQMessagesRequest{ForceFetch: force})\n\tif err != nil {\n\t\treturn fmt.Errorf(\"Error occurred while getting DLQ count, results may be partial: %w\", err)\n\t}\n\n\tif c.String(FlagDLQType) == \"domain\" {\n\t\tfmt.Println(response.Domain)\n\t\treturn nil\n\t}\n\n\ttable := []HistoryDLQCountRow{}\n\tfor key, count := range response.History {\n\t\ttable = append(table, HistoryDLQCountRow{\n\t\t\tSourceCluster: key.SourceCluster,\n\t\t\tShardID:       key.ShardID,\n\t\t\tCount:         count,\n\t\t})\n\t}\n\tsort.Slice(table, func(i, j int) bool {\n\t\t// First sort by source cluster\n\t\tswitch strings.Compare(table[i].SourceCluster, table[j].SourceCluster) {\n\t\tcase -1:\n\t\t\treturn true\n\t\tcase 1:\n\t\t\treturn false\n\t\t}\n\n\t\t// Then by count in decreasing order\n\t\tdiff := table[i].Count - table[j].Count\n\t\tif diff > 0 {\n\t\t\treturn true\n\t\t}\n\t\tif diff < 0 {\n\t\t\treturn false\n\t\t}\n\n\t\t// Finally by shard in increasing order\n\t\treturn table[i].ShardID < table[j].ShardID\n\t})\n\n\treturn Render(c, table, RenderOptions{Color: true, DefaultTemplate: templateTable})\n}\n\n// AdminGetDLQMessages gets DLQ metadata\nfunc AdminGetDLQMessages(c *cli.Context) error {\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\tclient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfdlqtype, err := getRequiredOption(c, FlagDLQType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tdlqType, err := toQueueType(fdlqtype)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to convert queue type\", err)\n\t}\n\tsourceCluster, err := getRequiredOption(c, FlagSourceCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tremainingMessageCount := constants.InclusiveEndMessageID\n\tif c.IsSet(FlagMaxMessageCount) {\n\t\tremainingMessageCount = c.Int64(FlagMaxMessageCount)\n\t}\n\tlastMessageID := constants.InclusiveEndMessageID\n\tif c.IsSet(FlagLastMessageID) {\n\t\tlastMessageID = c.Int64(FlagLastMessageID)\n\t}\n\n\t// Cache for domain names\n\tdomainNames := map[string]string{}\n\tgetDomainName := func(domainId string) (string, error) {\n\t\tif domainName, ok := domainNames[domainId]; ok {\n\t\t\treturn domainName, nil\n\t\t}\n\n\t\tresp, err := client.DescribeDomain(ctx, &types.DescribeDomainRequest{UUID: common.StringPtr(domainId)})\n\t\tif err != nil {\n\t\t\treturn \"\", commoncli.Problem(\"failed to describe domain\", err)\n\t\t}\n\t\tdomainNames[domainId] = resp.DomainInfo.Name\n\t\treturn resp.DomainInfo.Name, nil\n\t}\n\n\treadShard := func(shardID int) ([]DLQRow, error) {\n\t\tvar rows []DLQRow\n\t\tvar pageToken []byte\n\n\t\tfor {\n\t\t\tresp, err := adminClient.ReadDLQMessages(ctx, &types.ReadDLQMessagesRequest{\n\t\t\t\tType:                  dlqType,\n\t\t\t\tSourceCluster:         sourceCluster,\n\t\t\t\tShardID:               int32(shardID),\n\t\t\t\tInclusiveEndMessageID: common.Int64Ptr(lastMessageID),\n\t\t\t\tMaximumPageSize:       defaultPageSize,\n\t\t\t\tNextPageToken:         pageToken,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn nil, commoncli.Problem(fmt.Sprintf(\"fail to read dlq message for shard: %d\", shardID), err)\n\t\t\t}\n\n\t\t\treplicationTasks := map[int64]*types.ReplicationTask{}\n\t\t\tfor _, task := range resp.ReplicationTasks {\n\t\t\t\treplicationTasks[task.SourceTaskID] = task\n\t\t\t}\n\n\t\t\tfor _, info := range resp.ReplicationTasksInfo {\n\t\t\t\ttask := replicationTasks[info.TaskID]\n\n\t\t\t\tvar taskType *types.ReplicationTaskType\n\t\t\t\tif task != nil {\n\t\t\t\t\ttaskType = task.TaskType\n\t\t\t\t}\n\n\t\t\t\tevents, err := deserializeBatchEvents(task.GetHistoryTaskV2Attributes().GetEvents())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, fmt.Errorf(\"Error in deserializing batch events: %w\", err)\n\t\t\t\t}\n\t\t\t\tnewRunEvents, err := deserializeBatchEvents(task.GetHistoryTaskV2Attributes().GetNewRunEvents())\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, fmt.Errorf(\"Error in deserializing new run batch events: %w\", err)\n\t\t\t\t}\n\t\t\t\tdomainName, err := getDomainName(info.DomainID)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn nil, err\n\t\t\t\t}\n\t\t\t\trows = append(rows, DLQRow{\n\t\t\t\t\tShardID:         shardID,\n\t\t\t\t\tDomainName:      domainName,\n\t\t\t\t\tDomainID:        info.DomainID,\n\t\t\t\t\tWorkflowID:      info.WorkflowID,\n\t\t\t\t\tRunID:           info.RunID,\n\t\t\t\t\tTaskType:        taskType,\n\t\t\t\t\tTaskID:          info.TaskID,\n\t\t\t\t\tVersion:         info.Version,\n\t\t\t\t\tFirstEventID:    info.FirstEventID,\n\t\t\t\t\tNextEventID:     info.NextEventID,\n\t\t\t\t\tScheduledID:     info.ScheduledID,\n\t\t\t\t\tReplicationTask: task,\n\t\t\t\t\tEvents:          events,\n\t\t\t\t\tEventIDs:        collectEventIDs(events),\n\t\t\t\t\tNewRunEvents:    newRunEvents,\n\t\t\t\t\tNewRunEventIDs:  collectEventIDs(newRunEvents),\n\t\t\t\t})\n\n\t\t\t\tremainingMessageCount--\n\t\t\t\tif remainingMessageCount <= 0 {\n\t\t\t\t\treturn rows, nil\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif len(resp.NextPageToken) == 0 {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tpageToken = resp.NextPageToken\n\t\t}\n\t\treturn rows, nil\n\t}\n\n\ttable := []DLQRow{}\n\tfor shardID := range getShards(c) {\n\t\tif remainingMessageCount <= 0 {\n\t\t\tbreak\n\t\t}\n\t\ttablesInShard, err := readShard(shardID)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to read DLQ messages in shard %v: %w\", shardID, err)\n\t\t}\n\t\ttable = append(table, tablesInShard...)\n\t}\n\n\treturn Render(c, table, RenderOptions{DefaultTemplate: templateTable, Color: true})\n}\n\n// AdminPurgeDLQMessages deletes messages from DLQ\nfunc AdminPurgeDLQMessages(c *cli.Context) error {\n\tfdlqtype, err := getRequiredOption(c, FlagDLQType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tdlqType, err := toQueueType(fdlqtype)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to convert queue type\", err)\n\t}\n\tsourceCluster, err := getRequiredOption(c, FlagSourceCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required option not found\", err)\n\t}\n\tvar lastMessageID *int64\n\tif c.IsSet(FlagLastMessageID) {\n\t\tlastMessageID = common.Int64Ptr(c.Int64(FlagLastMessageID))\n\t}\n\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor shardID := range getShards(c) {\n\t\tctx, cancel, err := newContext(c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t\t}\n\t\terr = adminClient.PurgeDLQMessages(ctx, &types.PurgeDLQMessagesRequest{\n\t\t\tType:                  dlqType,\n\t\t\tSourceCluster:         sourceCluster,\n\t\t\tShardID:               int32(shardID),\n\t\t\tInclusiveEndMessageID: lastMessageID,\n\t\t})\n\t\tcancel()\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"Failed to purge DLQ message in shard %v with error: %v.\\n\", shardID, err)\n\t\t\tcontinue\n\t\t}\n\t\ttime.Sleep(10 * time.Millisecond)\n\t\tfmt.Printf(\"Successfully purge DLQ Messages in shard %v.\\n\", shardID)\n\t}\n\treturn nil\n}\n\n// AdminMergeDLQMessages merges message from DLQ\nfunc AdminMergeDLQMessages(c *cli.Context) error {\n\tfdlqtype, err := getRequiredOption(c, FlagDLQType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found\", err)\n\t}\n\tdlqType, err := toQueueType(fdlqtype)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to convert queue type\", err)\n\t}\n\tsourceCluster, err := getRequiredOption(c, FlagSourceCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required option not found\", err)\n\t}\n\tvar lastMessageID *int64\n\tif c.IsSet(FlagLastMessageID) {\n\t\tlastMessageID = common.Int64Ptr(c.Int64(FlagLastMessageID))\n\t}\n\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\nShardIDLoop:\n\tfor shardID := range getShards(c) {\n\t\trequest := &types.MergeDLQMessagesRequest{\n\t\t\tType:                  dlqType,\n\t\t\tSourceCluster:         sourceCluster,\n\t\t\tShardID:               int32(shardID),\n\t\t\tInclusiveEndMessageID: lastMessageID,\n\t\t\tMaximumPageSize:       defaultPageSize,\n\t\t}\n\n\t\tfor {\n\t\t\tctx, cancel, err := newContext(c)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t\t\t}\n\t\t\tresponse, err := adminClient.MergeDLQMessages(ctx, request)\n\t\t\tcancel()\n\t\t\tif err != nil {\n\t\t\t\tfmt.Printf(\"Failed to merge DLQ message in shard %v with error: %v.\\n\", shardID, err)\n\t\t\t\tcontinue ShardIDLoop\n\t\t\t}\n\n\t\t\tif len(response.NextPageToken) == 0 {\n\t\t\t\tbreak\n\t\t\t}\n\n\t\t\trequest.NextPageToken = response.NextPageToken\n\t\t}\n\t\tfmt.Printf(\"Successfully merged all messages in shard %v.\\n\", shardID)\n\t}\n\treturn nil\n}\n\nfunc getShards(c *cli.Context) chan int {\n\t// Check if we have stdin available\n\tstat, err := os.Stdin.Stat()\n\tif err == nil && (stat.Mode()&os.ModeCharDevice) == 0 {\n\t\treturn readShardsFromStdin()\n\t}\n\n\treturn generateShardRangeFromFlags(c)\n}\n\nfunc generateShardRangeFromFlags(c *cli.Context) chan int {\n\tshards := make(chan int)\n\tgo func() {\n\t\tshardRange, err := parseIntMultiRange(c.String(FlagShards))\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"failed to parse shard range: %q\\n\", c.String(FlagShards))\n\t\t} else {\n\t\t\tfor _, shard := range shardRange {\n\t\t\t\tshards <- shard\n\t\t\t}\n\t\t}\n\t\tclose(shards)\n\t}()\n\treturn shards\n}\n\nfunc readShardsFromStdin() chan int {\n\tshards := make(chan int)\n\tgo func() {\n\t\treader := bufio.NewReader(os.Stdin)\n\t\tfor {\n\t\t\tline, err := reader.ReadString('\\n')\n\t\t\tif err == io.EOF {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tif err != nil {\n\t\t\t\tfmt.Printf(\"Unable to read from stdin: %v\", err)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tshard, err := strconv.ParseInt(strings.TrimSpace(line), 10, 64)\n\t\t\tif err != nil {\n\t\t\t\tfmt.Printf(\"Failed to parse shard id: %q\\n\", line)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tshards <- int(shard)\n\t\t}\n\t\tclose(shards)\n\t}()\n\treturn shards\n}\n\nfunc toQueueType(dlqType string) (*types.DLQType, error) {\n\tswitch dlqType {\n\tcase \"domain\":\n\t\treturn types.DLQTypeDomain.Ptr(), nil\n\tcase \"history\":\n\t\treturn types.DLQTypeReplication.Ptr(), nil\n\tdefault:\n\t\treturn nil, fmt.Errorf(\"the queue type is not supported. Type: %v\", dlqType)\n\t}\n}\n\nfunc deserializeBatchEvents(blob *types.DataBlob) ([]*types.HistoryEvent, error) {\n\tif blob == nil {\n\t\treturn nil, nil\n\t}\n\tserializer := persistence.NewPayloadSerializer()\n\tevents, err := serializer.DeserializeBatchEvents(persistence.NewDataBlobFromInternal(blob))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to decode DLQ history replication events: %w\", err)\n\t}\n\treturn events, nil\n}\n\nfunc collectEventIDs(events []*types.HistoryEvent) []int64 {\n\tids := make([]int64, 0, len(events))\n\tfor _, event := range events {\n\t\tids = append(ids, event.ID)\n\t}\n\treturn ids\n}\n"
  },
  {
    "path": "tools/cli/admin_elastic_search_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"math\"\n\t\"os\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/olivere/elastic\"\n\t\"github.com/urfave/cli/v2\"\n\t\"golang.org/x/time/rate\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n\tes \"github.com/uber/cadence/common/elasticsearch\"\n\t\"github.com/uber/cadence/common/elasticsearch/esql\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tversionTypeExternal = \"external\"\n)\n\nvar timeKeys = map[string]bool{\n\t\"StartTime\":     true,\n\t\"CloseTime\":     true,\n\t\"ExecutionTime\": true,\n}\n\nfunc timeKeyFilter(key string) bool {\n\treturn timeKeys[key]\n}\n\nfunc timeValProcess(timeStr string) (string, error) {\n\t// first check if already in int64 format\n\tif _, err := strconv.ParseInt(timeStr, 10, 64); err == nil {\n\t\treturn timeStr, nil\n\t}\n\t// try to parse time\n\tparsedTime, err := time.Parse(defaultDateTimeFormat, timeStr)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn fmt.Sprintf(\"%v\", parsedTime.UnixNano()), nil\n}\n\ntype ESIndexRow struct {\n\tHealth             string `header:\"Health\"`\n\tStatus             string `header:\"Status\"`\n\tIndex              string `header:\"Index\"`\n\tPrimaryShards      int    `header:\"Pri\"`\n\tReplicaShards      int    `header:\"Rep\"`\n\tDocsCount          int    `header:\"Docs Count\"`\n\tDocsDeleted        int    `header:\"Docs Deleted\"`\n\tStorageSize        string `header:\"Store Size\"`\n\tPrimaryStorageSize string `header:\"Pri Store Size\"`\n}\n\n// AdminCatIndices cat indices for ES cluster\nfunc AdminCatIndices(c *cli.Context) error {\n\tesClient, err := getDeps(c).ElasticSearchClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx := c.Context\n\tresp, err := esClient.CatIndices().Do(ctx)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Unable to cat indices\", err)\n\t}\n\n\ttable := []ESIndexRow{}\n\tfor _, row := range resp {\n\t\ttable = append(table, ESIndexRow{\n\t\t\tHealth:             row.Health,\n\t\t\tStatus:             row.Status,\n\t\t\tIndex:              row.Index,\n\t\t\tPrimaryShards:      row.Pri,\n\t\t\tReplicaShards:      row.Rep,\n\t\t\tDocsCount:          row.DocsCount,\n\t\t\tDocsDeleted:        row.DocsDeleted,\n\t\t\tStorageSize:        row.StoreSize,\n\t\t\tPrimaryStorageSize: row.PriStoreSize,\n\t\t})\n\t}\n\treturn Render(c, table, RenderOptions{DefaultTemplate: templateTable, Color: true, Border: true})\n}\n\n// AdminIndex used to bulk insert message from kafka parse\nfunc AdminIndex(c *cli.Context) error {\n\tesClient, err := getDeps(c).ElasticSearchClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tindexName, err := getRequiredOption(c, FlagIndex)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tinputFileName, err := getRequiredOption(c, FlagInputFile)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tbatchSize := c.Int(FlagBatchSize)\n\n\tmessages, err := parseIndexerMessage(inputFileName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Unable to parse indexer message\", err)\n\t}\n\n\tbulkRequest := esClient.Bulk()\n\tbulkConductFn := func() error {\n\t\t_, err := bulkRequest.Do(c.Context)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Bulk failed\", err)\n\t\t}\n\t\tif bulkRequest.NumberOfActions() != 0 {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Bulk request not done, %d\", bulkRequest.NumberOfActions()), err)\n\t\t}\n\t\treturn nil\n\t}\n\tfor i, message := range messages {\n\t\tdocID := message.GetWorkflowID() + elasticsearch.GetESDocDelimiter() + message.GetRunID()\n\t\tvar req elastic.BulkableRequest\n\t\tswitch message.GetMessageType() {\n\t\tcase indexer.MessageTypeIndex:\n\t\t\tdoc, err := generateESDoc(message)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Error in Admin Index: \", err)\n\t\t\t}\n\t\t\treq = elastic.NewBulkIndexRequest().\n\t\t\t\tIndex(indexName).\n\t\t\t\tType(elasticsearch.GetESDocType()).\n\t\t\t\tId(docID).\n\t\t\t\tVersionType(versionTypeExternal).\n\t\t\t\tVersion(message.GetVersion()).\n\t\t\t\tDoc(doc)\n\t\tcase indexer.MessageTypeDelete:\n\t\t\treq = elastic.NewBulkDeleteRequest().\n\t\t\t\tIndex(indexName).\n\t\t\t\tType(elasticsearch.GetESDocType()).\n\t\t\t\tId(docID).\n\t\t\t\tVersionType(versionTypeExternal).\n\t\t\t\tVersion(message.GetVersion())\n\t\tcase indexer.MessageTypeCreate:\n\t\t\treq = elastic.NewBulkIndexRequest().\n\t\t\t\tOpType(\"create\").\n\t\t\t\tIndex(indexName).\n\t\t\t\tType(elasticsearch.GetESDocType()).\n\t\t\t\tId(docID).\n\t\t\t\tVersionType(\"internal\")\n\t\tdefault:\n\t\t\treturn commoncli.Problem(\"Unknown message type\", nil)\n\t\t}\n\t\tbulkRequest.Add(req)\n\n\t\tif i%batchSize == batchSize-1 {\n\t\t\tif err := bulkConductFn(); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t}\n\tif bulkRequest.NumberOfActions() != 0 {\n\t\tif err := bulkConductFn(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// AdminDelete used to delete documents from ElasticSearch with input of list result\nfunc AdminDelete(c *cli.Context) error {\n\tesClient, err := getDeps(c).ElasticSearchClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tindexName, err := getRequiredOption(c, FlagIndex)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tinputFileName, err := getRequiredOption(c, FlagInputFile)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tbatchSize := c.Int(FlagBatchSize)\n\trps := c.Int(FlagRPS)\n\tif rps <= 0 {\n\t\terr = fmt.Errorf(\"FlagRPS must be positive value but got %v\", rps)\n\t\treturn commoncli.Problem(\"Invalid RPS value: \", err)\n\t}\n\tratelimiter := clock.NewRatelimiter(rate.Limit(rps), rps)\n\n\t// This is only executed from the CLI by an admin user\n\t// #nosec\n\tfile, err := os.Open(inputFileName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Cannot open input file\", nil)\n\t}\n\tdefer file.Close()\n\tscanner := bufio.NewScanner(file)\n\n\tscanner.Scan() // skip first line\n\ti := 0\n\n\tbulkRequest := esClient.Bulk()\n\tbulkConductFn := func() error {\n\t\t// Wait for rate limiter token to become available\n\t\tif err = ratelimiter.Wait(c.Context); err != nil {\n\t\t\treturn err\n\t\t}\n\t\t_, err = bulkRequest.Do(c.Context)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Bulk failed, current processed row %d\", i), err)\n\t\t}\n\t\tif bulkRequest.NumberOfActions() != 0 {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Bulk request not done, current processed row %d\", i), err)\n\t\t}\n\t\treturn nil\n\t}\n\n\tfor scanner.Scan() {\n\t\tline := strings.Split(scanner.Text(), \"|\")\n\t\tdocID := strings.TrimSpace(line[1]) + elasticsearch.GetESDocDelimiter() + strings.TrimSpace(line[2])\n\t\treq := elastic.NewBulkDeleteRequest().\n\t\t\tIndex(indexName).\n\t\t\tType(elasticsearch.GetESDocType()).\n\t\t\tId(docID).\n\t\t\tVersionType(versionTypeExternal).\n\t\t\tVersion(math.MaxInt64)\n\t\tbulkRequest.Add(req)\n\t\tif i%batchSize == batchSize-1 {\n\t\t\tif err = bulkConductFn(); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\ti++\n\t}\n\tif bulkRequest.NumberOfActions() != 0 {\n\t\treturn bulkConductFn()\n\t}\n\treturn nil\n}\n\nfunc parseIndexerMessage(fileName string) (messages []*indexer.Message, err error) {\n\t// Executed from the CLI to parse existing elastiseach files\n\t// #nosec\n\tfile, err := os.Open(fileName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer file.Close()\n\n\tscanner := bufio.NewScanner(file)\n\tidx := 0\n\tfor scanner.Scan() {\n\t\tidx++\n\t\tline := strings.TrimSpace(scanner.Text())\n\t\tif len(line) == 0 {\n\t\t\tfmt.Printf(\"line %v is empty, skipped\\n\", idx)\n\t\t\tcontinue\n\t\t}\n\n\t\tmsg := &indexer.Message{}\n\t\terr := json.Unmarshal([]byte(line), msg)\n\t\tif err != nil {\n\t\t\tfmt.Printf(\"line %v cannot be deserialized to indexer message: %v.\\n\", idx, line)\n\t\t\treturn nil, err\n\t\t}\n\t\tmessages = append(messages, msg)\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\treturn nil, err\n\t}\n\treturn messages, nil\n}\n\nfunc generateESDoc(msg *indexer.Message) (map[string]interface{}, error) {\n\tdoc := make(map[string]interface{})\n\tdoc[es.DomainID] = msg.GetDomainID()\n\tdoc[es.WorkflowID] = msg.GetWorkflowID()\n\tdoc[es.RunID] = msg.GetRunID()\n\n\tfor k, v := range msg.Fields {\n\t\tswitch v.GetType() {\n\t\tcase indexer.FieldTypeString:\n\t\t\tdoc[k] = v.GetStringData()\n\t\tcase indexer.FieldTypeInt:\n\t\t\tdoc[k] = v.GetIntData()\n\t\tcase indexer.FieldTypeBool:\n\t\t\tdoc[k] = v.GetBoolData()\n\t\tcase indexer.FieldTypeBinary:\n\t\t\tdoc[k] = v.GetBinaryData()\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"Unknown field type %v\", nil)\n\t\t}\n\t}\n\treturn doc, nil\n}\n\n// This function is used to trim unnecessary tag in returned json for table header\nfunc trimBucketKey(k string) string {\n\t// group key is in form of \"group_key\", we only need \"key\" as the column name\n\tk = strings.TrimPrefix(k, \"group_\")\n\tk = strings.TrimPrefix(k, \"Attr_\")\n\treturn fmt.Sprintf(`%v(*)`, k)\n}\n\n// parse the returned time to readable string if time is in int64 format\nfunc toTimeStr(s interface{}) string {\n\tfloatTime, err := strconv.ParseFloat(s.(string), 64)\n\tintTime := int64(floatTime)\n\tif err != nil {\n\t\treturn s.(string)\n\t}\n\tt := time.Unix(0, intTime)\n\treturn t.Format(time.RFC3339)\n}\n\n// GenerateReport generate report for an aggregation query to ES\nfunc GenerateReport(c *cli.Context) error {\n\toutput := getDeps(c).Output()\n\n\t// use url command argument to create client\n\tindex, err := getRequiredOption(c, FlagIndex)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tsql, err := getRequiredOption(c, FlagListQuery)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tvar reportFormat, reportFilePath string\n\tif c.IsSet(FlagOutputFormat) {\n\t\treportFormat = c.String(FlagOutputFormat)\n\t}\n\tif c.IsSet(FlagOutputFilename) {\n\t\treportFilePath = c.String(FlagOutputFilename)\n\t} else {\n\t\treportFilePath = \"./report.\" + reportFormat\n\t}\n\tesClient, err := getDeps(c).ElasticSearchClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tctx := c.Context\n\n\t// convert sql to dsl\n\te := esql.NewESql()\n\te.SetCadence(true)\n\te.ProcessQueryValue(timeKeyFilter, timeValProcess)\n\tdsl, sortFields, err := e.ConvertPrettyCadence(sql, \"\")\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Fail to convert sql to dsl\", err)\n\t}\n\n\t// query client\n\tresp, err := esClient.Search(index).Source(dsl).Do(ctx)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Fail to talk with ES\", err)\n\t}\n\n\t// Show result to terminal\n\ttable := tablewriter.NewWriter(output)\n\tvar headers []string\n\tvar groupby, bucket map[string]interface{}\n\tvar buckets []interface{}\n\terr = json.Unmarshal(*resp.Aggregations[\"groupby\"], &groupby)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Fail to parse groupby\", err)\n\t}\n\tbuckets = groupby[\"buckets\"].([]interface{})\n\tif len(buckets) == 0 {\n\t\toutput.Write([]byte(\"no matching bucket\\n\"))\n\t\treturn nil\n\t}\n\n\t// get the FIRST bucket in bucket list to extract all tags. These extracted tags are to be used as table heads\n\tbucket = buckets[0].(map[string]interface{})\n\t// record the column position in the table of each returned item\n\tids := make(map[string]int)\n\t// We want these 3 columns shows at leftmost of the table in cadence report usage. It can be changed in future.\n\tprimaryCols := []string{\"group_DomainID\", \"group_WorkflowType\", \"group_CloseStatus\"}\n\tprimaryColsMap := map[string]int{\n\t\t\"group_DomainID\":     1,\n\t\t\"group_WorkflowType\": 1,\n\t\t\"group_CloseStatus\":  1,\n\t}\n\tbuckKeys := 0 // number of bucket keys, used for table collapsing in html report\n\tif v, exist := bucket[\"key\"]; exist {\n\t\tvmap := v.(map[string]interface{})\n\t\t// first search whether primaryCols keys exist, if found, put them at the table beginning\n\t\tfor _, k := range primaryCols {\n\t\t\tif _, exist := vmap[k]; exist {\n\t\t\t\tk = trimBucketKey(k) // trim the unnecessary prefix\n\t\t\t\theaders = append(headers, k)\n\t\t\t\tids[k] = len(ids)\n\t\t\t\tbuckKeys++\n\t\t\t}\n\t\t}\n\t\t// extract all remaining bucket keys\n\t\tfor k := range vmap {\n\t\t\tif _, exist := primaryColsMap[k]; !exist {\n\t\t\t\tk = trimBucketKey(k)\n\t\t\t\theaders = append(headers, k)\n\t\t\t\tids[k] = len(ids)\n\t\t\t\tbuckKeys++\n\t\t\t}\n\t\t}\n\t}\n\t// extract all other non-key items and set the table head accordingly\n\tfor k := range bucket {\n\t\tif k != \"key\" {\n\t\t\tif k == \"doc_count\" {\n\t\t\t\tk = \"count\"\n\t\t\t}\n\t\t\theaders = append(headers, k)\n\t\t\tids[k] = len(ids)\n\t\t}\n\t}\n\ttable.SetHeader(headers)\n\n\t// read each bucket and fill the table, use map ids to find the correct spot\n\tvar tableData [][]string\n\tfor _, b := range buckets {\n\t\tbucket = b.(map[string]interface{})\n\t\tdata := make([]string, len(headers))\n\t\tfor k, v := range bucket {\n\t\t\tswitch k {\n\t\t\tcase \"key\": // fill group key\n\t\t\t\tvmap := v.(map[string]interface{})\n\t\t\t\tfor kk, vv := range vmap {\n\t\t\t\t\tkk = trimBucketKey(kk)\n\t\t\t\t\tdata[ids[kk]] = fmt.Sprintf(\"%v\", vv)\n\t\t\t\t}\n\t\t\tcase \"doc_count\": // fill bucket size count\n\t\t\t\tdata[ids[\"count\"]] = fmt.Sprintf(\"%v\", v)\n\t\t\tdefault:\n\t\t\t\tvar datum string\n\t\t\t\tvmap := v.(map[string]interface{})\n\t\t\t\tif strings.Contains(k, \"Attr_CustomDatetimeField\") {\n\t\t\t\t\tdatum = fmt.Sprintf(\"%v\", vmap[\"value_as_string\"])\n\t\t\t\t} else {\n\t\t\t\t\tdatum = fmt.Sprintf(\"%v\", vmap[\"value\"])\n\t\t\t\t\t// convert Cadence stored time (unix nano) to readable format\n\t\t\t\t\tif strings.Contains(k, \"Time\") && !strings.Contains(k, \"Attr_\") {\n\t\t\t\t\t\tdatum = toTimeStr(datum)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tdata[ids[k]] = datum\n\t\t\t}\n\t\t}\n\t\ttable.Append(data)\n\t\ttableData = append(tableData, data)\n\t}\n\ttable.Render()\n\n\tswitch reportFormat {\n\tcase \"html\", \"HTML\":\n\t\tsorted := len(sortFields) > 0 || strings.Contains(sql, \"ORDER BY\") || strings.Contains(sql, \"order by\")\n\t\treturn generateHTMLReport(reportFilePath, buckKeys, sorted, headers, tableData)\n\tcase \"csv\", \"CSV\":\n\t\treturn generateCSVReport(reportFilePath, headers, tableData)\n\tdefault:\n\t\treturn commoncli.Problem(fmt.Sprintf(`Report format %v not supported.`, reportFormat), nil)\n\t}\n}\n\nfunc generateCSVReport(reportFileName string, headers []string, tableData [][]string) error {\n\t// write csv report\n\tf, err := os.Create(reportFileName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Fail to create csv report file\", err)\n\t}\n\tcsvContent := strings.Join(headers, \",\") + \"\\n\"\n\tfor _, data := range tableData {\n\t\tcsvContent += strings.Join(data, \",\") + \"\\n\"\n\t}\n\t_, err = f.WriteString(csvContent)\n\tif err != nil {\n\t\tfmt.Printf(\"Error write to file, err: %v\", err)\n\t}\n\treturn f.Close()\n}\n\nfunc generateHTMLReport(reportFileName string, numBuckKeys int, sorted bool, headers []string, tableData [][]string) error {\n\t// write html report\n\tf, err := os.Create(reportFileName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Fail to create html report file\", err)\n\t}\n\tvar htmlContent string\n\tm, n := len(headers), len(tableData)\n\trowSpan := make([]int, m) // record the collapsing size of each column\n\tfor i := 0; i < m; i++ {\n\t\trowSpan[i] = 1\n\t\tcell := wrapWithTag(headers[i], \"td\", \"\")\n\t\thtmlContent += cell\n\t}\n\thtmlContent = wrapWithTag(htmlContent, \"tr\", \"\")\n\n\tfor row := 0; row < n; row++ {\n\t\tvar rowData string\n\t\tfor col := 0; col < m; col++ {\n\t\t\trowSpan[col]--\n\t\t\t// don't do collapsing if sorted\n\t\t\tif col < numBuckKeys-1 && !sorted {\n\t\t\t\tif rowSpan[col] == 0 {\n\t\t\t\t\tfor i := row; i < n; i++ {\n\t\t\t\t\t\tif tableData[i][col] == tableData[row][col] {\n\t\t\t\t\t\t\trowSpan[col]++\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tbreak\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t\tvar property string\n\t\t\t\t\tif rowSpan[col] > 1 {\n\t\t\t\t\t\tproperty = fmt.Sprintf(`rowspan=\"%d\"`, rowSpan[col])\n\t\t\t\t\t}\n\t\t\t\t\tcell := wrapWithTag(tableData[row][col], \"td\", property)\n\t\t\t\t\trowData += cell\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tcell := wrapWithTag(tableData[row][col], \"td\", \"\")\n\t\t\t\trowData += cell\n\t\t\t}\n\t\t}\n\t\trowData = wrapWithTag(rowData, \"tr\", \"\")\n\t\thtmlContent += rowData\n\t}\n\thtmlContent = wrapWithTag(htmlContent, \"table\", \"\")\n\thtmlContent = wrapWithTag(htmlContent, \"body\", \"\")\n\thtmlContent = wrapWithTag(htmlContent, \"html\", \"\")\n\n\t//nolint:errcheck\n\tf.WriteString(\"<!DOCTYPE html>\\n\")\n\tf.WriteString(`<head>\n\t<style>\n\ttable, th, td {\n\t  border: 1px solid black;\n\t  padding: 5px;\n\t}\n\ttable {\n\t  border-spacing: 2px;\n\t}\n\t</style>\n\t</head>` + \"\\n\")\n\tf.WriteString(htmlContent)\n\treturn f.Close()\n}\n\n// return a string that use tag to wrap content\nfunc wrapWithTag(content string, tag string, property string) string {\n\tif property != \"\" {\n\t\tproperty = \" \" + property\n\t}\n\tif tag != \"td\" {\n\t\tcontent = \"\\n\" + content\n\t}\n\treturn \"<\" + tag + property + \">\" + content + \"</\" + tag + \">\\n\"\n}\n"
  },
  {
    "path": "tools/cli/admin_elastic_search_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"flag\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"net/http/httptest\"\n\t\"os\"\n\t\"regexp\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/common/elasticsearch\"\n)\n\n// Tests for timeKeyFilter function\nfunc TestTimeKeyFilter(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tkey      string\n\t\texpected bool\n\t}{\n\t\t{\n\t\t\tname:     \"ValidTimeKeyStartTime\",\n\t\t\tkey:      \"StartTime\",\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"ValidTimeKeyCloseTime\",\n\t\t\tkey:      \"CloseTime\",\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"ValidTimeKeyExecutionTime\",\n\t\t\tkey:      \"ExecutionTime\",\n\t\t\texpected: true,\n\t\t},\n\t\t{\n\t\t\tname:     \"InvalidTimeKey\",\n\t\t\tkey:      \"SomeOtherKey\",\n\t\t\texpected: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := timeKeyFilter(tt.key)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\n// Tests for timeValProcess function\nfunc TestTimeValProcess(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\ttimeStr     string\n\t\texpected    string\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname:        \"ValidInt64TimeString\",\n\t\t\ttimeStr:     \"1630425600000000000\", // Already in int64 format\n\t\t\texpected:    \"1630425600000000000\",\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"ValidDateTimeString\",\n\t\t\ttimeStr:     \"2021-09-01T00:00:00Z\", // A valid time string\n\t\t\texpected:    fmt.Sprintf(\"%v\", time.Date(2021, 9, 1, 0, 0, 0, 0, time.UTC).UnixNano()),\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"InvalidTimeString\",\n\t\t\ttimeStr:     \"invalid-time\",\n\t\t\texpected:    \"\",\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult, err := timeValProcess(tt.timeStr)\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Helper function to remove ANSI color codes from the output\nfunc removeANSIColors(text string) string {\n\tansiEscapePattern := `\\x1b\\[[0-9;]*m`\n\tre := regexp.MustCompile(ansiEscapePattern)\n\treturn re.ReplaceAllString(text, \"\")\n}\n\nfunc TestAdminCatIndices(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\thandler        http.HandlerFunc\n\t\texpectedOutput string\n\t\texpectedError  string\n\t\thandlerCalled  bool\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful response from Elasticsearch CatIndices API\n\t\t\t\tif r.URL.Path == \"/_cat/indices\" && r.Method == \"GET\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`[{\"health\":\"green\",\"status\":\"open\",\"index\":\"test-index\",\"pri\":\"5\",\"rep\":\"1\",\"docs.count\":\"1000\",\"docs.deleted\":\"50\",\"store.size\":\"10gb\",\"pri.store.size\":\"5gb\"}]`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\texpectedOutput: `+--------+--------+------------+-----+-----+------------+--------------+------------+----------------+\n| HEALTH | STATUS |   INDEX    | PRI | REP | DOCS COUNT | DOCS DELETED | STORE SIZE | PRI STORE SIZE |\n+--------+--------+------------+-----+-----+------------+--------------+------------+----------------+\n| green  | open   | test-index |   5 |   1 |       1000 |           50 | 10gb       | 5gb            |\n+--------+--------+------------+-----+-----+------------+--------------+------------+----------------+\n\n`,\n\t\t\texpectedError: \"\",\n\t\t\thandlerCalled: true,\n\t\t},\n\t\t{\n\t\t\tname: \"CatIndices Error\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate an error response\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t}),\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Unable to cat indices\",\n\t\t\thandlerCalled:  true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\thandlerCalled := false\n\n\t\t\t// Wrap the test case's handler to track if it was called\n\t\t\twrappedHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\thandlerCalled = true\n\t\t\t\ttt.handler.ServeHTTP(w, r)\n\t\t\t})\n\n\t\t\t// Create mock Elasticsearch client and server\n\t\t\tesClient, testServer := getMockClient(t, wrappedHandler)\n\t\t\tdefer testServer.Close()\n\n\t\t\t// Initialize mock controller\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\t// Create mock Cadence client factory\n\t\t\tmockClientFactory := NewMockClientFactory(mockCtrl)\n\n\t\t\t// Create test IO handler to capture output\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Set up the CLI app\n\t\t\tapp := NewCliApp(mockClientFactory, WithIOHandler(ioHandler))\n\n\t\t\t// Expect ElasticSearchClient to return the mock client created by getMockClient\n\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\n\t\t\t// Create a mock CLI context\n\t\t\tc := setContextMock(app)\n\n\t\t\t// Call AdminCatIndices\n\t\t\terr := AdminCatIndices(c)\n\n\t\t\t// Validate handler was called\n\t\t\tassert.Equal(t, tt.handlerCalled, handlerCalled, \"Expected handler to be called\")\n\n\t\t\t// Check for expected error or success\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\t// Remove ANSI color codes from the captured output\n\t\t\t\tactualOutput := removeANSIColors(ioHandler.outputBytes.String())\n\n\t\t\t\t// Validate the output captured by testIOHandler\n\t\t\t\tassert.Equal(t, tt.expectedOutput, actualOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc setContextMock(app *cli.App) *cli.Context {\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"Domain flag\")\n\tc := cli.NewContext(app, set, nil)\n\treturn c\n}\n\n// getMockClient creates a mock elastic.Client using the provided HTTP handler and returns the client and the test server\nfunc getMockClient(t *testing.T, handler http.HandlerFunc) (*elastic.Client, *httptest.Server) {\n\t// Create a mock HTTP test server\n\ttestServer := httptest.NewTLSServer(handler)\n\n\t// Create an Elasticsearch client using the test server's URL\n\tmockClient, err := elastic.NewClient(\n\t\telastic.SetURL(testServer.URL),\n\t\telastic.SetSniff(false),\n\t\telastic.SetHealthcheck(false),\n\t\telastic.SetHttpClient(testServer.Client()),\n\t)\n\t// Ensure no error occurred while creating the mock client\n\tassert.NoError(t, err)\n\n\t// Return the elastic.Client and the test server\n\treturn mockClient, testServer\n}\n\nfunc TestAdminIndex(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\thandler         http.HandlerFunc\n\t\tcreateInputFile bool\n\t\tmessageType     indexer.MessageType\n\t\texpectedOutput  string\n\t\texpectedError   string\n\t}{\n\t\t{\n\t\t\tname: \"SuccessIndexMessage\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful Bulk request\n\t\t\t\tif r.URL.Path == \"/_bulk\" && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\"took\": 30, \"errors\": false}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\tmessageType:     indexer.MessageTypeIndex,\n\t\t\texpectedOutput:  \"\", // Example output for success case\n\t\t\texpectedError:   \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"SuccessCreateMessage\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful Bulk request\n\t\t\t\tif r.URL.Path == \"/_bulk\" && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\"took\": 30, \"errors\": false}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\tmessageType:     indexer.MessageTypeCreate,\n\t\t\texpectedOutput:  \"\", // Example output for create case\n\t\t\texpectedError:   \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"SuccessDeleteMessage\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful Bulk request\n\t\t\t\tif r.URL.Path == \"/_bulk\" && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\"took\": 30, \"errors\": false}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\tmessageType:     indexer.MessageTypeDelete,\n\t\t\texpectedOutput:  \"\", // Example output for delete case\n\t\t\texpectedError:   \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"UnknownMessageType\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// No bulk request needed for this case\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\tmessageType:     indexer.MessageType(9999),\n\t\t\texpectedOutput:  \"\",\n\t\t\texpectedError:   \"Unknown message type\",\n\t\t},\n\t\t{\n\t\t\tname: \"BulkRequestFailure\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a Bulk request failure\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tw.Write([]byte(`{\"error\": \"Bulk request failed\"}`))\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\tmessageType:     indexer.MessageTypeIndex,\n\t\t\texpectedOutput:  \"\",\n\t\t\texpectedError:   \"Bulk failed\",\n\t\t},\n\t\t{\n\t\t\tname: \"ParseIndexerMessageError\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// In this test case, we are simulating a parse error, so no bulk request needed.\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t}),\n\t\t\tcreateInputFile: false, // No valid input file created\n\t\t\tmessageType:     indexer.MessageTypeIndex,\n\t\t\texpectedOutput:  \"\",\n\t\t\texpectedError:   \"Unable to parse indexer message\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar inputFileName string\n\t\t\tvar err error\n\t\t\tif tt.createInputFile {\n\t\t\t\t// Create a temporary input file with a valid message\n\t\t\t\tinputFileName, err = createTempIndexerInputFileWithMessageType(tt.messageType, false)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tdefer os.Remove(inputFileName) // Clean up after test\n\t\t\t}\n\n\t\t\t// Create mock Elasticsearch client and server\n\t\t\tesClient, testServer := getMockClient(t, tt.handler)\n\t\t\tdefer testServer.Close()\n\n\t\t\t// Initialize mock controller\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\t// Create mock client factory\n\t\t\tmockClientFactory := NewMockClientFactory(mockCtrl)\n\n\t\t\t// Create test IO handler to capture output\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Set up the CLI app\n\t\t\tapp := NewCliApp(mockClientFactory, WithIOHandler(ioHandler))\n\n\t\t\t// Expect ElasticSearchClient to return the mock client created by getMockClient\n\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\n\t\t\t// Setup flag values for the CLI context\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagIndex, \"test-index\", \"Index flag\")\n\t\t\tif tt.createInputFile {\n\t\t\t\tset.String(FlagInputFile, inputFileName, \"Input file flag\")\n\t\t\t} else {\n\t\t\t\tset.String(FlagInputFile, \"invalid-input-file\", \"Input file flag\")\n\t\t\t}\n\t\t\tset.Int(FlagBatchSize, 1, \"Batch size flag\")\n\n\t\t\t// Create a mock CLI context\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call AdminIndex\n\t\t\terr = AdminIndex(c)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\t// Validate the output captured by testIOHandler\n\t\t\t\tassert.Equal(t, tt.expectedOutput, ioHandler.outputBytes.String())\n\t\t\t}\n\t\t})\n\t}\n}\n\n// Helper function to create a temporary input file for AdminIndex or AdminDelete with valid data\nfunc createTempIndexerInputFileWithMessageType(messageType indexer.MessageType, forDelete bool) (string, error) {\n\tfile, err := os.CreateTemp(\"\", \"indexer_input_*.txt\")\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tdefer file.Close()\n\n\twriter := bufio.NewWriter(file)\n\n\tif forDelete {\n\t\t// For AdminDelete, we need to simulate workflow-id|run-id format\n\t\t_, err = writer.WriteString(\"Header\\n\") // First line is skipped in AdminDelete\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\t_, err = writer.WriteString(\"some-value|workflow-id|run-id\\n\") // Simulate document deletion data\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t} else {\n\t\t// For AdminIndex, we need to generate a JSON message format\n\t\tmessage := `{\"WorkflowID\": \"test-workflow-id\", \"RunID\": \"test-run-id\", \"Version\": 1, \"MessageType\": ` + fmt.Sprintf(\"%d\", messageType) + `}`\n\t\t_, err = writer.WriteString(message + \"\\n\")\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t}\n\n\twriter.Flush()\n\n\treturn file.Name(), nil\n}\n\nfunc TestAdminDelete(t *testing.T) {\n\ttests := []struct {\n\t\tname            string\n\t\thandler         http.HandlerFunc\n\t\tcreateInputFile bool\n\t\texpectedOutput  string\n\t\texpectedError   string\n\t}{\n\t\t{\n\t\t\tname: \"SuccessDelete\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful Bulk delete request\n\t\t\t\tif r.URL.Path == \"/_bulk\" && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\"took\": 30, \"errors\": false}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\texpectedOutput:  \"\", // Example output for delete case\n\t\t\texpectedError:   \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"BulkRequestFailure\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate an error in the Bulk delete request\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tw.Write([]byte(`{\"error\": \"Bulk request failed\"}`))\n\t\t\t}),\n\t\t\tcreateInputFile: true,\n\t\t\texpectedOutput:  \"\",\n\t\t\texpectedError:   \"Bulk failed\",\n\t\t},\n\t\t{\n\t\t\tname: \"ParseFileError\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// No bulk request needed in this case, just simulating a file parsing error\n\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t}),\n\t\t\tcreateInputFile: false, // No valid input file created\n\t\t\texpectedOutput:  \"\",\n\t\t\texpectedError:   \"Cannot open input file\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar inputFileName string\n\t\t\tvar err error\n\t\t\tif tt.createInputFile {\n\t\t\t\t// Reuse the temp input file creation helper from previous tests\n\t\t\t\tinputFileName, err = createTempIndexerInputFileWithMessageType(indexer.MessageTypeDelete, true)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tdefer os.Remove(inputFileName) // Clean up after test\n\t\t\t}\n\n\t\t\t// Create mock Elasticsearch client and server\n\t\t\tesClient, testServer := getMockClient(t, tt.handler)\n\t\t\tdefer testServer.Close()\n\n\t\t\t// Initialize mock controller\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\t// Create mock client factory\n\t\t\tmockClientFactory := NewMockClientFactory(mockCtrl)\n\n\t\t\t// Expect ElasticSearchClient to return the mock client created by getMockClient\n\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\n\t\t\t// Create test IO handler to capture output\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Set up the CLI app\n\t\t\tapp := NewCliApp(mockClientFactory, WithIOHandler(ioHandler))\n\n\t\t\t// Setup flag values for the CLI context\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagIndex, \"test-index\", \"Index flag\")\n\t\t\tif tt.createInputFile {\n\t\t\t\tset.String(FlagInputFile, inputFileName, \"Input file flag\")\n\t\t\t} else {\n\t\t\t\tset.String(FlagInputFile, \"invalid-input-file\", \"Input file flag\")\n\t\t\t}\n\t\t\tset.Int(FlagBatchSize, 1, \"Batch size flag\")\n\t\t\tset.Int(FlagRPS, 10, \"RPS flag\")\n\n\t\t\t// Create a mock CLI context\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call AdminDelete\n\t\t\terr = AdminDelete(c)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tif err != nil {\n\t\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t\t} else {\n\t\t\t\t\tt.Errorf(\"Expected error: %s, but got no error\", tt.expectedError)\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\t// Validate the output captured by testIOHandler\n\t\t\t\tassert.Equal(t, tt.expectedOutput, ioHandler.outputBytes.String())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestParseIndexerMessage(t *testing.T) {\n\tworkflowID := \"test-workflow-id\"\n\trunID := \"test-run-id\"\n\tversion := int64(1)\n\tmessageType := indexer.MessageTypeIndex\n\n\ttests := []struct {\n\t\tname            string\n\t\tmessageType     indexer.MessageType\n\t\tcreateInputFile bool\n\t\texpectedError   string\n\t\texpectedResult  []*indexer.Message\n\t}{\n\t\t{\n\t\t\tname:            \"SuccessParse\",\n\t\t\tmessageType:     indexer.MessageTypeIndex,\n\t\t\tcreateInputFile: true,\n\t\t\texpectedError:   \"\",\n\t\t\texpectedResult: []*indexer.Message{\n\t\t\t\t{\n\t\t\t\t\tWorkflowID:  &workflowID,\n\t\t\t\t\tRunID:       &runID,\n\t\t\t\t\tVersion:     &version,\n\t\t\t\t\tMessageType: &messageType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:            \"FileNotExist\",\n\t\t\tmessageType:     0,\n\t\t\tcreateInputFile: false, // No file created\n\t\t\texpectedError:   \"open nonexistent-file.txt: no such file or directory\",\n\t\t\texpectedResult:  nil,\n\t\t},\n\t\t{\n\t\t\tname:            \"SkipEmptyLines\",\n\t\t\tmessageType:     indexer.MessageTypeIndex,\n\t\t\tcreateInputFile: true,\n\t\t\texpectedError:   \"\",\n\t\t\texpectedResult: []*indexer.Message{\n\t\t\t\t{\n\t\t\t\t\tWorkflowID:  &workflowID,\n\t\t\t\t\tRunID:       &runID,\n\t\t\t\t\tVersion:     &version,\n\t\t\t\t\tMessageType: &messageType,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar fileName string\n\t\t\tvar err error\n\t\t\tif tt.createInputFile {\n\t\t\t\t// Use the existing createTempIndexerInputFileWithMessageType function\n\t\t\t\tfileName, err = createTempIndexerInputFileWithMessageType(tt.messageType, false) // forDelete=false for AdminIndex\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tdefer os.Remove(fileName) // Clean up after test\n\t\t\t} else {\n\t\t\t\t// Simulate file not found\n\t\t\t\tfileName = \"nonexistent-file.txt\"\n\t\t\t}\n\n\t\t\t// Call the function being tested\n\t\t\tmessages, err := parseIndexerMessage(fileName)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t\tassert.Nil(t, messages)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedResult, messages)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGenerateESDoc(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tmessage       *indexer.Message\n\t\texpectedDoc   map[string]interface{}\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"SuccessWithAllFieldTypes\",\n\t\t\tmessage: &indexer.Message{\n\t\t\t\tDomainID:   &[]string{\"domain1\"}[0],\n\t\t\t\tWorkflowID: &[]string{\"workflow1\"}[0],\n\t\t\t\tRunID:      &[]string{\"run1\"}[0],\n\t\t\t\tFields: map[string]*indexer.Field{\n\t\t\t\t\t\"field_string\": {\n\t\t\t\t\t\tType:       &[]indexer.FieldType{indexer.FieldTypeString}[0],\n\t\t\t\t\t\tStringData: &[]string{\"string_value\"}[0],\n\t\t\t\t\t},\n\t\t\t\t\t\"field_int\": {\n\t\t\t\t\t\tType:    &[]indexer.FieldType{indexer.FieldTypeInt}[0],\n\t\t\t\t\t\tIntData: &[]int64{123}[0],\n\t\t\t\t\t},\n\t\t\t\t\t\"field_bool\": {\n\t\t\t\t\t\tType:     &[]indexer.FieldType{indexer.FieldTypeBool}[0],\n\t\t\t\t\t\tBoolData: &[]bool{true}[0],\n\t\t\t\t\t},\n\t\t\t\t\t\"field_binary\": {\n\t\t\t\t\t\tType:       &[]indexer.FieldType{indexer.FieldTypeBinary}[0],\n\t\t\t\t\t\tBinaryData: []byte(\"binary_value\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDoc: map[string]interface{}{\n\t\t\t\telasticsearch.DomainID:   \"domain1\",\n\t\t\t\telasticsearch.WorkflowID: \"workflow1\",\n\t\t\t\telasticsearch.RunID:      \"run1\",\n\t\t\t\t\"field_string\":           \"string_value\",\n\t\t\t\t\"field_int\":              int64(123),\n\t\t\t\t\"field_bool\":             true,\n\t\t\t\t\"field_binary\":           []byte(\"binary_value\"),\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"UnknownFieldType\",\n\t\t\tmessage: &indexer.Message{\n\t\t\t\tDomainID:   &[]string{\"domain1\"}[0],\n\t\t\t\tWorkflowID: &[]string{\"workflow1\"}[0],\n\t\t\t\tRunID:      &[]string{\"run1\"}[0],\n\t\t\t\tFields: map[string]*indexer.Field{\n\t\t\t\t\t\"unknown_field\": {\n\t\t\t\t\t\tType: &[]indexer.FieldType{9999}[0], // Invalid field type\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedDoc:   nil,\n\t\t\texpectedError: \"Unknown field type\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Call the function being tested\n\t\t\tdoc, err := generateESDoc(tt.message)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t\tassert.Nil(t, doc)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedDoc, doc)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestGenerateReport(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\thandler        http.HandlerFunc\n\t\tsetupContext   func(app *cli.App) *cli.Context\n\t\tsetupMocks     func(mockClientFactory *MockClientFactory, esClient *elastic.Client)\n\t\texpectedOutput string\n\t\texpectedError  string\n\t}{\n\t\t{\n\t\t\tname: \"SuccessCSVReportWithExtraKey\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful response with extra key not in primaryColsMap\n\t\t\t\texpectedPath := \"/test-index/_search\"\n\t\t\t\tif r.URL.Path == expectedPath && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"groupby\": {\n\t\t\t\t\t\t\t\t\"buckets\": [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\"key\": {\n\t\t\t\t\t\t\t\t\t\t\t\"group_DomainID\": \"domain1\",\n\t\t\t\t\t\t\t\t\t\t\t\"group_CustomKey\": \"custom-value\"\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\"Attr_CustomDatetimeField\": {\n\t\t\t\t\t\t\t\t\t\t\t\"value_as_string\": \"2023-10-01T12:34:56.789Z\"\n\t\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t]\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\tset.String(FlagOutputFormat, \"\", \"Output format flag\")\n\t\t\t\tset.String(FlagOutputFilename, \"\", \"Output file flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\t_ = set.Set(FlagOutputFormat, \"csv\")\n\t\t\t\t_ = set.Set(FlagOutputFilename, \"test-report.csv\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: `+-------------+--------------+--------------------------+\n| DOMAINID(*) | CUSTOMKEY(*) | ATTR CUSTOMDATETIMEFIELD |\n+-------------+--------------+--------------------------+\n| domain1     | custom-value | 2023-10-01T12:34:56.789Z |\n+-------------+--------------+--------------------------+\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t// can't put all keys all together because keys generated in reports are in a random order, thus will fail tests\n\t\t{\n\t\t\tname: \"SuccessCSVReportWithOtherExtraKey\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful response with extra key not in primaryColsMap\n\t\t\t\texpectedPath := \"/test-index/_search\"\n\t\t\t\tif r.URL.Path == expectedPath && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"groupby\": {\n\t\t\t\t\t\t\t\t\"buckets\": [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\"key\": {\n\t\t\t\t\t\t\t\t\t\t\t\"group_DomainID\": \"domain1\",\n\t\t\t\t\t\t\t\t\t\t\t\"group_CustomKey\": \"custom-value\"\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\"Attr_CustomStringField\": {\n\t\t\t\t\t\t\t\t\t\t\t\"value\": \"test-string\"\n\t\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t]\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\tset.String(FlagOutputFormat, \"\", \"Output format flag\")\n\t\t\t\tset.String(FlagOutputFilename, \"\", \"Output file flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\t_ = set.Set(FlagOutputFormat, \"csv\")\n\t\t\t\t_ = set.Set(FlagOutputFilename, \"test-report.csv\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: `+-------------+--------------+------------------------+\n| DOMAINID(*) | CUSTOMKEY(*) | ATTR CUSTOMSTRINGFIELD |\n+-------------+--------------+------------------------+\n| domain1     | custom-value | test-string            |\n+-------------+--------------+------------------------+\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"SuccessHTMLReportWithExtraKey\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful response with extra key not in primaryColsMap\n\t\t\t\texpectedPath := \"/test-index/_search\"\n\t\t\t\tif r.URL.Path == expectedPath && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"groupby\": {\n\t\t\t\t\t\t\t\t\"buckets\": [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\"key\": {\n\t\t\t\t\t\t\t\t\t\t\t\"group_DomainID\": \"domain1\",\n\t\t\t\t\t\t\t\t\t\t\t\"group_CustomKey\": \"custom-value\"\n\t\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t\t\t\"doc_count\": 10\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t]\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\tset.String(FlagOutputFormat, \"\", \"Output format flag\")\n\t\t\t\tset.String(FlagOutputFilename, \"\", \"Output file flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\t_ = set.Set(FlagOutputFormat, \"html\")\n\t\t\t\t_ = set.Set(FlagOutputFilename, \"test-report.csv\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: `+-------------+--------------+-------+\n| DOMAINID(*) | CUSTOMKEY(*) | COUNT |\n+-------------+--------------+-------+\n| domain1     | custom-value |    10 |\n+-------------+--------------+-------+\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"EmptyBucket\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate a successful response with extra key not in primaryColsMap\n\t\t\t\texpectedPath := \"/test-index/_search\"\n\t\t\t\tif r.URL.Path == expectedPath && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"groupby\": {\n\t\t\t\t\t\t\t\t\"buckets\": []\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\tset.String(FlagOutputFormat, \"\", \"Output format flag\")\n\t\t\t\tset.String(FlagOutputFilename, \"\", \"Output file flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\t_ = set.Set(FlagOutputFormat, \"html\")\n\t\t\t\t_ = set.Set(FlagOutputFilename, \"test-report.csv\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: `no matching bucket\n`,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"UnsupportedReportFormat\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Elasticsearch request returns successful response\n\t\t\t\tif r.URL.Path == \"/test-index/_search\" && r.Method == \"POST\" {\n\t\t\t\t\tw.WriteHeader(http.StatusOK)\n\t\t\t\t\tw.Write([]byte(`{\n\t\t\t\t\t\t\"aggregations\": {\n\t\t\t\t\t\t\t\"groupby\": {\n\t\t\t\t\t\t\t\t\"buckets\": [\n\t\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\t\t\"key\": {\"group_DomainID\": \"domain1\"},\n\t\t\t\t\t\t\t\t\t\t\"doc_count\": 10\n\t\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\t]\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t}`))\n\t\t\t\t} else {\n\t\t\t\t\tw.WriteHeader(http.StatusNotFound)\n\t\t\t\t}\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\tset.String(FlagOutputFormat, \"\", \"Output format flag\")\n\t\t\t\tset.String(FlagOutputFilename, \"\", \"Output file flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\t_ = set.Set(FlagOutputFormat, \"unsupported-format\")\n\t\t\t\t_ = set.Set(FlagOutputFilename, \"test-report.unsupported\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Report format unsupported-format not supported.\",\n\t\t},\n\t\t{\n\t\t\tname: \"ElasticsearchQueryError\",\n\t\t\thandler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n\t\t\t\t// Simulate an error response from Elasticsearch\n\t\t\t\tw.WriteHeader(http.StatusInternalServerError)\n\t\t\t\tw.Write([]byte(`{\"error\": \"query failed\"}`))\n\t\t\t}),\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Define and set flags\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\t// Set the actual values\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks: func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {\n\t\t\t\tmockClientFactory.EXPECT().ElasticSearchClient(gomock.Any()).Return(esClient, nil).Times(1)\n\t\t\t},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Fail to talk with ES\",\n\t\t},\n\t\t{\n\t\t\tname:    \"MissingRequiredFlagIndex\",\n\t\t\thandler: nil, // No handler needed since the error occurs before any Elasticsearch interaction\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Only setting FlagListQuery, but missing FlagIndex to trigger the error\n\t\t\t\tset.String(FlagListQuery, \"\", \"List query flag\")\n\t\t\t\t_ = set.Set(FlagListQuery, \"SELECT * FROM logs\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks:     func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Required flag not found: \",\n\t\t},\n\t\t{\n\t\t\tname:    \"MissingRequiredFlagListQuery\",\n\t\t\thandler: nil, // No handler needed since the error occurs before any Elasticsearch interaction\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\t// Only setting FlagIndex, but missing FlagListQuery to trigger the error\n\t\t\t\tset.String(FlagIndex, \"\", \"Index flag\")\n\t\t\t\t_ = set.Set(FlagIndex, \"test-index\")\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tsetupMocks:     func(mockClientFactory *MockClientFactory, esClient *elastic.Client) {},\n\t\t\texpectedOutput: \"\",\n\t\t\texpectedError:  \"Required flag not found: \",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Create mock Elasticsearch client and server\n\t\t\tesClient, testServer := getMockClient(t, tt.handler)\n\t\t\tdefer testServer.Close()\n\n\t\t\t// Initialize mock controller\n\t\t\tmockCtrl := gomock.NewController(t)\n\n\t\t\t// Create mock client factory\n\t\t\tmockClientFactory := NewMockClientFactory(mockCtrl)\n\n\t\t\t// Create test IO handler to capture output\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Set up the CLI app\n\t\t\tapp := NewCliApp(mockClientFactory, WithIOHandler(ioHandler))\n\n\t\t\t// Expect ElasticSearchClient to return the mock client created by getMockClient\n\t\t\ttt.setupMocks(mockClientFactory, esClient)\n\n\t\t\t// Set up the context for the specific test case\n\t\t\tc := tt.setupContext(app)\n\n\t\t\t// Call GenerateReport\n\t\t\terr := GenerateReport(c)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\t// Validate the output captured by testIOHandler\n\t\t\t\tassert.Equal(t, tt.expectedOutput, ioHandler.outputBytes.String())\n\t\t\t}\n\n\t\t\tos.Remove(\"test-report.csv\")\n\t\t})\n\t}\n}\n\nfunc TestGenerateHTMLReport_RowSpanLogic(t *testing.T) {\n\t// Prepare headers and tableData to trigger the rowspan logic\n\theaders := []string{\"Domain\", \"Status\", \"Count\"}\n\ttableData := [][]string{\n\t\t{\"domain1\", \"open\", \"10\"},\n\t\t{\"domain1\", \"open\", \"15\"},\n\t\t{\"domain2\", \"closed\", \"20\"},\n\t\t{\"domain2\", \"closed\", \"25\"},\n\t}\n\n\t// Prepare temp file to write HTML report\n\ttempFile, err := os.CreateTemp(\"\", \"test_report_*.html\")\n\tassert.NoError(t, err)\n\tdefer os.Remove(tempFile.Name()) // Clean up\n\n\t// Call generateHTMLReport with numBuckKeys to control the column collapsing\n\terr = generateHTMLReport(tempFile.Name(), 2, false, headers, tableData)\n\tassert.NoError(t, err)\n\n\t// Read and validate the generated HTML content\n\tcontent, err := os.ReadFile(tempFile.Name())\n\tassert.NoError(t, err)\n\n\t// Remove all newlines and spaces to simplify comparison\n\tactualContent := string(content)\n\tactualContent = removeWhitespace(actualContent)\n\n\t// Expected HTML content (also simplified by removing whitespace)\n\texpectedHTMLStructure := removeWhitespace(`<tr>\n\t<td rowspan=\"2\">domain1</td><td>open</td><td>10</td>\n\t</tr>\n\t<tr><td>open</td><td>15</td></tr>`)\n\n\t// Validate the rowspan logic was applied correctly\n\tassert.Contains(t, actualContent, expectedHTMLStructure)\n}\n\n// Helper function to remove all whitespace from a string\nfunc removeWhitespace(input string) string {\n\treturn strings.ReplaceAll(strings.ReplaceAll(input, \"\\n\", \"\"), \"\\t\", \"\")\n}\n"
  },
  {
    "path": "tools/cli/admin_failover_commands.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os/user\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/failovermanager\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tdefaultAbortReason                      = \"Failover aborted through admin CLI\"\n\tdefaultBatchFailoverSize                = 20\n\tdefaultBatchFailoverWaitTimeInSeconds   = 30\n\tdefaultFailoverWorkflowTimeoutInSeconds = 1200\n)\n\nvar (\n\tuuidFn        = uuid.New\n\tgetOperatorFn = getOperator\n)\n\ntype startParams struct {\n\ttargetCluster                  string\n\tsourceCluster                  string\n\tbatchFailoverSize              int\n\tbatchFailoverWaitTimeInSeconds int\n\tfailoverWorkflowTimeout        int\n\tfailoverTimeout                int\n\tdomains                        []string\n\tdrillWaitTime                  int\n\tcron                           string\n}\n\n// AdminFailoverStart start failover workflow\nfunc AdminFailoverStart(c *cli.Context) error {\n\ttc, err := getRequiredOption(c, FlagTargetCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tsc, err := getRequiredOption(c, FlagSourceCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tparams := &startParams{\n\t\ttargetCluster:                  tc,\n\t\tsourceCluster:                  sc,\n\t\tbatchFailoverSize:              c.Int(FlagFailoverBatchSize),\n\t\tbatchFailoverWaitTimeInSeconds: c.Int(FlagFailoverWaitTime),\n\t\tfailoverTimeout:                c.Int(FlagFailoverTimeout),\n\t\tfailoverWorkflowTimeout:        c.Int(FlagExecutionTimeout),\n\t\tdomains:                        c.StringSlice(FlagFailoverDomains),\n\t\tdrillWaitTime:                  c.Int(FlagFailoverDrillWaitTime),\n\t\tcron:                           c.String(FlagCronSchedule),\n\t}\n\treturn failoverStart(c, params)\n}\n\n// AdminFailoverPause pause failover workflow\nfunc AdminFailoverPause(c *cli.Context) error {\n\terr := executePauseOrResume(c, getFailoverWorkflowID(c), true)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to pause failover workflow\", err)\n\t}\n\tfmt.Println(\"Failover paused on \" + getFailoverWorkflowID(c))\n\treturn nil\n}\n\n// AdminFailoverResume resume a paused failover workflow\nfunc AdminFailoverResume(c *cli.Context) error {\n\terr := executePauseOrResume(c, getFailoverWorkflowID(c), false)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to resume failover workflow\", err)\n\t}\n\tfmt.Println(\"Failover resumed on \" + getFailoverWorkflowID(c))\n\treturn nil\n}\n\n// AdminFailoverQuery query a failover workflow\nfunc AdminFailoverQuery(c *cli.Context) error {\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tworkflowID := getFailoverWorkflowID(c)\n\trunID := getRunID(c)\n\tresult, err := query(tcCtx, client, workflowID, runID)\n\tif err != nil {\n\t\treturn err\n\t}\n\trequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: constants.SystemLocalDomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t}\n\n\tdescResp, err := client.DescribeWorkflowExecution(tcCtx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to describe workflow\", err)\n\t}\n\tif isWorkflowTerminated(descResp) {\n\t\tresult.State = failovermanager.WorkflowAborted\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), result)\n\treturn nil\n}\n\n// AdminFailoverAbort abort a failover workflow\nfunc AdminFailoverAbort(c *cli.Context) error {\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\treason := c.String(FlagReason)\n\tif len(reason) == 0 {\n\t\treason = defaultAbortReason\n\t}\n\tworkflowID := getFailoverWorkflowID(c)\n\trunID := getRunID(c)\n\trequest := &types.TerminateWorkflowExecutionRequest{\n\t\tDomain: constants.SystemLocalDomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tReason: reason,\n\t}\n\n\terr = client.TerminateWorkflowExecution(tcCtx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to abort failover workflow\", err)\n\t}\n\n\tfmt.Println(\"Failover aborted\")\n\treturn nil\n}\n\n// AdminFailoverRollback rollback a failover run\nfunc AdminFailoverRollback(c *cli.Context) error {\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\trunID := getRunID(c)\n\n\tqueryResult, err := query(tcCtx, client, failovermanager.FailoverWorkflowID, runID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif isWorkflowRunning(queryResult) {\n\t\trequest := &types.TerminateWorkflowExecutionRequest{\n\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tReason:   \"Rollback\",\n\t\t\tIdentity: getCliIdentity(),\n\t\t}\n\n\t\terr := client.TerminateWorkflowExecution(tcCtx, request)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to terminate failover workflow\", err)\n\t\t}\n\t}\n\t// query again\n\tqueryResult, err = query(tcCtx, client, failovermanager.FailoverWorkflowID, runID)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar rollbackDomains []string\n\t// rollback includes both success and failed domains to make sure no leftover domains\n\trollbackDomains = append(rollbackDomains, queryResult.SuccessDomains...)\n\trollbackDomains = append(rollbackDomains, queryResult.FailedDomains...)\n\n\tparams := &startParams{\n\t\ttargetCluster:                  queryResult.SourceCluster,\n\t\tsourceCluster:                  queryResult.TargetCluster,\n\t\tdomains:                        rollbackDomains,\n\t\tbatchFailoverSize:              c.Int(FlagFailoverBatchSize),\n\t\tbatchFailoverWaitTimeInSeconds: c.Int(FlagFailoverWaitTime),\n\t\tfailoverTimeout:                c.Int(FlagFailoverTimeout),\n\t\tfailoverWorkflowTimeout:        c.Int(FlagExecutionTimeout),\n\t}\n\treturn failoverStart(c, params)\n}\n\n// AdminFailoverList list failover runs\nfunc AdminFailoverList(c *cli.Context) error {\n\tif err := c.Set(FlagWorkflowID, getFailoverWorkflowID(c)); err != nil {\n\t\treturn err\n\t}\n\tif err := c.Set(FlagDomain, constants.SystemLocalDomainName); err != nil {\n\t\treturn err\n\t}\n\treturn ListWorkflow(c)\n}\n\nfunc query(\n\ttcCtx context.Context,\n\tclient frontend.Client,\n\tworkflowID string,\n\trunID string) (*failovermanager.QueryResult, error) {\n\n\trequest := &types.QueryWorkflowRequest{\n\t\tDomain: constants.SystemLocalDomainName,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tQuery: &types.WorkflowQuery{\n\t\t\tQueryType: failovermanager.QueryType,\n\t\t},\n\t}\n\tqueryResp, err := client.QueryWorkflow(tcCtx, request)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Failed to query failover workflow\", err)\n\t}\n\n\tif queryResp.GetQueryResult() == nil {\n\t\treturn nil, commoncli.Problem(\"QueryResult has no value\", nil)\n\t}\n\tvar queryResult failovermanager.QueryResult\n\terr = json.Unmarshal(queryResp.GetQueryResult(), &queryResult)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Unable to deserialize QueryResult\", nil)\n\t}\n\treturn &queryResult, nil\n}\n\nfunc isWorkflowRunning(queryResult *failovermanager.QueryResult) bool {\n\treturn queryResult.State == failovermanager.WorkflowRunning ||\n\t\tqueryResult.State == failovermanager.WorkflowPaused\n}\n\nfunc getCadenceClient(c *cli.Context) (frontend.Client, error) {\n\tsvcClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn svcClient, nil\n}\n\nfunc getRunID(c *cli.Context) string {\n\tif c.IsSet(FlagRunID) {\n\t\treturn c.String(FlagRunID)\n\t}\n\treturn \"\"\n}\n\nfunc failoverStart(c *cli.Context, params *startParams) error {\n\tif err := validateStartParams(params); err != nil {\n\t\treturn commoncli.Problem(\"Invalid input parameters\", err)\n\t}\n\n\tworkflowID := failovermanager.FailoverWorkflowID\n\ttargetCluster := params.targetCluster\n\tsourceCluster := params.sourceCluster\n\tbatchFailoverSize := params.batchFailoverSize\n\tbatchFailoverWaitTimeInSeconds := params.batchFailoverWaitTimeInSeconds\n\tworkflowTimeout := int32(params.failoverWorkflowTimeout)\n\tdomains := params.domains\n\tdrillWaitTime := time.Duration(params.drillWaitTime) * time.Second\n\tvar gracefulFailoverTimeoutInSeconds *int32\n\tif params.failoverTimeout > 0 {\n\t\tgracefulFailoverTimeoutInSeconds = common.Int32Ptr(int32(params.failoverTimeout))\n\t}\n\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\top, err := getOperatorFn()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in getting operator: \", err)\n\t}\n\tmemo, err := getWorkflowMemo(map[string]interface{}{\n\t\tconstants.MemoKeyForOperator: op,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to serialize memo\", err)\n\t}\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              constants.SystemLocalDomainName,\n\t\tRequestID:                           uuidFn(),\n\t\tWorkflowID:                          workflowID,\n\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\tTaskList:                            &types.TaskList{Name: failovermanager.TaskListName},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(workflowTimeout),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(defaultDecisionTimeoutInSeconds),\n\t\tMemo:                                memo,\n\t\tWorkflowType:                        &types.WorkflowType{Name: failovermanager.FailoverWorkflowTypeName},\n\t}\n\tif params.drillWaitTime > 0 {\n\t\trequest.WorkflowID = failovermanager.DrillWorkflowID\n\t\trequest.CronSchedule = params.cron\n\t} else {\n\t\tif len(params.cron) > 0 {\n\t\t\treturn commoncli.Problem(\"The drill wait time is required when cron is specified.\", nil)\n\t\t}\n\n\t\t// block if there is an on-going failover drill\n\t\tif err := executePauseOrResume(c, failovermanager.DrillWorkflowID, true); err != nil {\n\t\t\tswitch err.(type) {\n\t\t\tcase *types.EntityNotExistsError:\n\t\t\t\tbreak\n\t\t\tcase *types.WorkflowExecutionAlreadyCompletedError:\n\t\t\t\tbreak\n\t\t\tdefault:\n\t\t\t\treturn commoncli.Problem(\"Failed to send pause signal to drill workflow\", err)\n\t\t\t}\n\t\t}\n\t}\n\n\tfoParams := failovermanager.FailoverParams{\n\t\tTargetCluster:                    targetCluster,\n\t\tSourceCluster:                    sourceCluster,\n\t\tBatchFailoverSize:                batchFailoverSize,\n\t\tBatchFailoverWaitTimeInSeconds:   batchFailoverWaitTimeInSeconds,\n\t\tDomains:                          domains,\n\t\tDrillWaitTime:                    drillWaitTime,\n\t\tGracefulFailoverTimeoutInSeconds: gracefulFailoverTimeoutInSeconds,\n\t}\n\tinput, err := json.Marshal(foParams)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to serialize Failover Params\", err)\n\t}\n\trequest.Input = input\n\twf, err := client.StartWorkflowExecution(tcCtx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to start failover workflow\", err)\n\t}\n\tfmt.Println(\"Failover workflow started\")\n\tfmt.Println(\"wid: \" + workflowID)\n\tfmt.Println(\"rid: \" + wf.GetRunID())\n\treturn nil\n}\n\nfunc getFailoverWorkflowID(c *cli.Context) string {\n\tif c.Bool(FlagFailoverDrill) {\n\t\treturn failovermanager.DrillWorkflowID\n\t}\n\treturn failovermanager.FailoverWorkflowID\n}\n\nfunc getOperator() (string, error) {\n\tuser, err := user.Current()\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"failed to get operator info %w\", err)\n\t}\n\n\treturn fmt.Sprintf(\"%s (username: %s)\", user.Name, user.Username), nil\n}\n\nfunc isWorkflowTerminated(descResp *types.DescribeWorkflowExecutionResponse) bool {\n\treturn types.WorkflowExecutionCloseStatusTerminated.String() == descResp.GetWorkflowExecutionInfo().GetCloseStatus().String()\n}\n\nfunc executePauseOrResume(c *cli.Context, workflowID string, isPause bool) error {\n\tclient, err := getCadenceClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\trunID := getRunID(c)\n\tvar signalName string\n\tif isPause {\n\t\tsignalName = failovermanager.PauseSignal\n\t} else {\n\t\tsignalName = failovermanager.ResumeSignal\n\t}\n\n\trequest := &types.SignalWorkflowExecutionRequest{\n\t\tDomain: constants.SystemLocalDomainName,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tSignalName: signalName,\n\t\tIdentity:   getCliIdentity(),\n\t}\n\n\treturn client.SignalWorkflowExecution(tcCtx, request)\n}\n\nfunc validateStartParams(params *startParams) error {\n\tif len(params.targetCluster) == 0 {\n\t\treturn fmt.Errorf(\"targetCluster is not provided: %v\", nil)\n\t}\n\tif len(params.sourceCluster) == 0 {\n\t\treturn fmt.Errorf(\"sourceCluster is not provided: %v\", nil)\n\t}\n\tif params.targetCluster == params.sourceCluster {\n\t\treturn fmt.Errorf(\"targetCluster is same as sourceCluster: %v\", nil)\n\t}\n\tif params.batchFailoverSize <= 0 {\n\t\tparams.batchFailoverSize = defaultBatchFailoverSize\n\t}\n\tif params.batchFailoverWaitTimeInSeconds <= 0 {\n\t\tparams.batchFailoverWaitTimeInSeconds = defaultBatchFailoverWaitTimeInSeconds\n\t}\n\tif params.failoverWorkflowTimeout <= 0 {\n\t\tparams.failoverWorkflowTimeout = defaultFailoverWorkflowTimeoutInSeconds\n\t}\n\treturn nil\n}\n"
  },
  {
    "path": "tools/cli/admin_failover_commands_test.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strconv\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/google/go-cmp/cmp\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/yarpc\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/failovermanager\"\n)\n\nfunc TestAdminFailoverStart(t *testing.T) {\n\toldUUIDFn := uuidFn\n\tuuidFn = func() string { return \"test-uuid\" }\n\toldGetOperatorFn := getOperatorFn\n\tgetOperatorFn = func() (string, error) { return \"test-user\", nil }\n\tdefer func() {\n\t\tuuidFn = oldUUIDFn\n\t\tgetOperatorFn = oldGetOperatorFn\n\t}()\n\n\ttests := []struct {\n\t\tdesc                    string\n\t\tsourceCluster           string\n\t\ttargetCluster           string\n\t\tfailoverBatchSize       int\n\t\tfailoverWaitTime        int\n\t\tgracefulFailoverTimeout int\n\t\tfailoverWFTimeout       int\n\t\tfailoverDomains         []string\n\t\tfailoverDrillWaitTime   int\n\t\tfailoverCron            string\n\t\trunID                   string\n\t\tmockFn                  func(*testing.T, *frontend.MockClient)\n\t\twantErr                 bool\n\t}{\n\t\t{\n\t\t\tdesc:                    \"success\",\n\t\t\tsourceCluster:           \"cluster1\",\n\t\t\ttargetCluster:           \"cluster2\",\n\t\t\tfailoverBatchSize:       10,\n\t\t\tfailoverWaitTime:        120,\n\t\t\tgracefulFailoverTimeout: 300,\n\t\t\tfailoverWFTimeout:       600,\n\t\t\tfailoverDomains:         []string{\"domain1\", \"domain2\"},\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// first drill workflow will be signalled to pause in case it is running.\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\n\t\t\t\t// then failover workflow will be started\n\t\t\t\twantReq := &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.SystemLocalDomainName,\n\t\t\t\t\tRequestID:                           \"test-uuid\",\n\t\t\t\t\tWorkflowID:                          failovermanager.FailoverWorkflowID,\n\t\t\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: failovermanager.TaskListName},\n\t\t\t\t\tInput:                               []byte(`{\"TargetCluster\":\"cluster2\",\"SourceCluster\":\"cluster1\",\"BatchFailoverSize\":10,\"BatchFailoverWaitTimeInSeconds\":120,\"Domains\":[\"domain1\",\"domain2\"],\"DrillWaitTime\":0,\"GracefulFailoverTimeoutInSeconds\":300}`),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(600), // == failoverWFTimeout\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(defaultDecisionTimeoutInSeconds),\n\t\t\t\t\tMemo: mustGetWorkflowMemo(t, map[string]interface{}{\n\t\t\t\t\t\tconstants.MemoKeyForOperator: \"test-user\",\n\t\t\t\t\t}),\n\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: failovermanager.FailoverWorkflowTypeName},\n\t\t\t\t}\n\t\t\t\tresp := &types.StartWorkflowExecutionResponse{}\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.StartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn resp, nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"startworkflow fails\",\n\t\t\twantErr:       true,\n\t\t\tsourceCluster: \"cluster1\",\n\t\t\ttargetCluster: \"cluster2\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// first drill workflow will be signalled to pause in case it is running.\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\n\t\t\t\t// then failover workflow will be started\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.StartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"failed to start workflow\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"source and target cluster same\",\n\t\t\twantErr:       true,\n\t\t\tsourceCluster: \"cluster1\",\n\t\t\ttargetCluster: \"cluster1\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// no frontend calls due to validation failure\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"no source cluster specified\",\n\t\t\twantErr:       true,\n\t\t\tsourceCluster: \"\",\n\t\t\ttargetCluster: \"cluster2\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// no frontend calls due to validation failure\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"no target cluster specified\",\n\t\t\twantErr:       true,\n\t\t\tsourceCluster: \"cluster1\",\n\t\t\ttargetCluster: \"\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// no frontend calls due to validation failure\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:                    \"success with cron\",\n\t\t\tsourceCluster:           \"cluster1\",\n\t\t\ttargetCluster:           \"cluster2\",\n\t\t\tfailoverBatchSize:       10,\n\t\t\tfailoverWaitTime:        120,\n\t\t\tgracefulFailoverTimeout: 300,\n\t\t\tfailoverWFTimeout:       600,\n\t\t\tfailoverDrillWaitTime:   30,\n\t\t\tfailoverCron:            \"0 0 * * *\",\n\t\t\tfailoverDomains:         []string{\"domain1\", \"domain2\"},\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// failover drill workflow will be started\n\t\t\t\twantReq := &types.StartWorkflowExecutionRequest{\n\t\t\t\t\tDomain:                              constants.SystemLocalDomainName,\n\t\t\t\t\tRequestID:                           \"test-uuid\",\n\t\t\t\t\tCronSchedule:                        \"0 0 * * *\",\n\t\t\t\t\tWorkflowID:                          failovermanager.DrillWorkflowID,\n\t\t\t\t\tWorkflowIDReusePolicy:               types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t\t\t\tTaskList:                            &types.TaskList{Name: failovermanager.TaskListName},\n\t\t\t\t\tInput:                               []byte(`{\"TargetCluster\":\"cluster2\",\"SourceCluster\":\"cluster1\",\"BatchFailoverSize\":10,\"BatchFailoverWaitTimeInSeconds\":120,\"Domains\":[\"domain1\",\"domain2\"],\"DrillWaitTime\":30000000000,\"GracefulFailoverTimeoutInSeconds\":300}`),\n\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(600), // == failoverWFTimeout\n\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(defaultDecisionTimeoutInSeconds),\n\t\t\t\t\tMemo: mustGetWorkflowMemo(t, map[string]interface{}{\n\t\t\t\t\t\tconstants.MemoKeyForOperator: \"test-user\",\n\t\t\t\t\t}),\n\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: failovermanager.FailoverWorkflowTypeName},\n\t\t\t\t}\n\t\t\t\tresp := &types.StartWorkflowExecutionResponse{}\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.StartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn resp, nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tfrontendCl := frontend.NewMockClient(ctrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttc.mockFn(t, frontendCl)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: frontendCl,\n\t\t\t})\n\n\t\t\targs := []string{\"\", \"admin\", \"cluster\", \"failover\", \"start\",\n\t\t\t\t\"--sc\", tc.sourceCluster,\n\t\t\t\t\"--tc\", tc.targetCluster,\n\t\t\t\t\"--failover_batch_size\", strconv.Itoa(tc.failoverBatchSize),\n\t\t\t\t\"--failover_wait_time_second\", strconv.Itoa(tc.failoverWaitTime),\n\t\t\t\t\"--failover_timeout_seconds\", strconv.Itoa(tc.gracefulFailoverTimeout),\n\t\t\t\t\"--execution_timeout\", strconv.Itoa(tc.failoverWFTimeout),\n\t\t\t\t\"--domains\", strings.Join(tc.failoverDomains, \",\"),\n\t\t\t\t\"--failover_drill_wait_second\", strconv.Itoa(tc.failoverDrillWaitTime),\n\t\t\t\t\"--cron\", tc.failoverCron,\n\t\t\t}\n\t\t\terr := app.Run(args)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminFailoverPauseResume(t *testing.T) {\n\ttests := []struct {\n\t\tdesc          string\n\t\trunID         string\n\t\tpauseOrResume string\n\t\tmockFn        func(*testing.T, *frontend.MockClient)\n\t\twantErr       bool\n\t}{\n\t\t{\n\t\t\tdesc:          \"pause success\",\n\t\t\tpauseOrResume: \"pause\",\n\t\t\trunID:         \"runid1\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.SignalWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\twantReq := &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"runid1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tSignalName: failovermanager.PauseSignal,\n\t\t\t\t\t\t\tIdentity:   getCliIdentity(),\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"pause signal workflow fails\",\n\t\t\tpauseOrResume: \"pause\",\n\t\t\twantErr:       true,\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, r *types.SignalWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\treturn fmt.Errorf(\"failed to signal workflow\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"resume success\",\n\t\t\tpauseOrResume: \"resume\",\n\t\t\trunID:         \"runid1\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\twantReq := &types.SignalWorkflowExecutionRequest{\n\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\tRunID:      \"runid1\",\n\t\t\t\t\t},\n\t\t\t\t\tSignalName: failovermanager.ResumeSignal,\n\t\t\t\t\tIdentity:   getCliIdentity(),\n\t\t\t\t}\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.SignalWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:          \"resume signal workflow fails\",\n\t\t\tpauseOrResume: \"resume\",\n\t\t\twantErr:       true,\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, r *types.SignalWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\treturn fmt.Errorf(\"failed to signal workflow\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tfrontendCl := frontend.NewMockClient(ctrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttc.mockFn(t, frontendCl)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: frontendCl,\n\t\t\t})\n\n\t\t\targs := []string{\"\", \"admin\", \"cluster\", \"failover\", tc.pauseOrResume,\n\t\t\t\t\"--rid\", tc.runID,\n\t\t\t}\n\t\t\terr := app.Run(args)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminFailoverQuery(t *testing.T) {\n\tqueryResult := failovermanager.QueryResult{\n\t\tTotalDomains: 10,\n\t\tSuccess:      2,\n\t\tFailed:       3,\n\t}\n\ttests := []struct {\n\t\tdesc    string\n\t\tmockFn  func(*testing.T, *frontend.MockClient)\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tdesc: \"success\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.QueryWorkflowRequest, opts ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\t\t\t\t\t\twantReq := &types.QueryWorkflowRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\t\tQueryType: failovermanager.QueryType,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.QueryWorkflowResponse{\n\t\t\t\t\t\t\tQueryResult: mustMarshalQueryResult(t, queryResult),\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\tm.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.DescribeWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error) {\n\t\t\t\t\t\twantReq := &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusTerminated.Ptr(),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:    \"query failed\",\n\t\t\twantErr: true,\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.QueryWorkflowRequest, opts ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"failed to query workflow\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tdesc:    \"describe failed\",\n\t\t\twantErr: true,\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.QueryWorkflowRequest, opts ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\t\t\t\t\t\twantReq := &types.QueryWorkflowRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\t\tQueryType: failovermanager.QueryType,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.QueryWorkflowResponse{\n\t\t\t\t\t\t\tQueryResult: mustMarshalQueryResult(t, queryResult),\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\tm.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.DescribeWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.DescribeWorkflowExecutionResponse, error) {\n\t\t\t\t\t\treturn nil, fmt.Errorf(\"failed to describe workflow\")\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tfrontendCl := frontend.NewMockClient(ctrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttc.mockFn(t, frontendCl)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: frontendCl,\n\t\t\t})\n\n\t\t\targs := []string{\"\", \"admin\", \"cluster\", \"failover\", \"query\"}\n\t\t\terr := app.Run(args)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminFailoverAbort(t *testing.T) {\n\ttests := []struct {\n\t\tdesc    string\n\t\tmockFn  func(*testing.T, *frontend.MockClient)\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tdesc: \"success\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\tm.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.TerminateWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\twantReq := &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tReason: \"Failover aborted through admin CLI\",\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tfrontendCl := frontend.NewMockClient(ctrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttc.mockFn(t, frontendCl)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: frontendCl,\n\t\t\t})\n\n\t\t\targs := []string{\"\", \"admin\", \"cluster\", \"failover\", \"abort\"}\n\t\t\terr := app.Run(args)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminFailoverRollback(t *testing.T) {\n\toldUUIDFn := uuidFn\n\tuuidFn = func() string { return \"test-uuid\" }\n\toldGetOperatorFn := getOperatorFn\n\tgetOperatorFn = func() (string, error) { return \"test-user\", nil }\n\tdefer func() {\n\t\tuuidFn = oldUUIDFn\n\t\tgetOperatorFn = oldGetOperatorFn\n\t}()\n\n\ttests := []struct {\n\t\tdesc    string\n\t\tmockFn  func(*testing.T, *frontend.MockClient)\n\t\twantErr bool\n\t}{\n\t\t{\n\t\t\tdesc: \"success\",\n\t\t\tmockFn: func(t *testing.T, m *frontend.MockClient) {\n\t\t\t\t// query to check if it's running.\n\t\t\t\tm.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.QueryWorkflowRequest, opts ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\t\t\t\t\t\twantReq := &types.QueryWorkflowRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\t\tQueryType: failovermanager.QueryType,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.QueryWorkflowResponse{\n\t\t\t\t\t\t\tQueryResult: mustMarshalQueryResult(t, failovermanager.QueryResult{\n\t\t\t\t\t\t\t\tState: failovermanager.WorkflowRunning,\n\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\t// terminate since it's running\n\t\t\t\tm.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.TerminateWorkflowExecutionRequest, opts ...yarpc.CallOption) error {\n\t\t\t\t\t\twantReq := &types.TerminateWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tReason:   \"Rollback\",\n\t\t\t\t\t\t\tIdentity: getCliIdentity(),\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\t// query again to get domains\n\t\t\t\tm.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.QueryWorkflowRequest, opts ...yarpc.CallOption) (*types.QueryWorkflowResponse, error) {\n\t\t\t\t\t\twantReq := &types.QueryWorkflowRequest{\n\t\t\t\t\t\t\tDomain: constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\t\tRunID:      \"\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\t\tQueryType: failovermanager.QueryType,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn &types.QueryWorkflowResponse{\n\t\t\t\t\t\t\tQueryResult: mustMarshalQueryResult(t, failovermanager.QueryResult{\n\t\t\t\t\t\t\t\tState:          failovermanager.WorkflowAborted,\n\t\t\t\t\t\t\t\tSourceCluster:  \"cluster1\",\n\t\t\t\t\t\t\t\tTargetCluster:  \"cluster2\",\n\t\t\t\t\t\t\t\tSuccessDomains: []string{\"domain1\", \"domain2\"},\n\t\t\t\t\t\t\t\tFailedDomains:  []string{\"domain3\"},\n\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t}, nil\n\t\t\t\t\t}).Times(1)\n\n\t\t\t\t// failback for success+failed domains\n\t\t\t\t//\n\t\t\t\t// first drill workflow will be signalled to pause in case it is running.\n\t\t\t\tm.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t\t// then failover workflow will be started to perform failback\n\t\t\t\tresp := &types.StartWorkflowExecutionResponse{}\n\t\t\t\tm.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\t\tDoAndReturn(func(ctx context.Context, gotReq *types.StartWorkflowExecutionRequest, opts ...yarpc.CallOption) (*types.StartWorkflowExecutionResponse, error) {\n\t\t\t\t\t\twantReq := &types.StartWorkflowExecutionRequest{\n\t\t\t\t\t\t\tDomain:                constants.SystemLocalDomainName,\n\t\t\t\t\t\t\tRequestID:             \"test-uuid\",\n\t\t\t\t\t\t\tWorkflowID:            failovermanager.FailoverWorkflowID,\n\t\t\t\t\t\t\tWorkflowIDReusePolicy: types.WorkflowIDReusePolicyAllowDuplicate.Ptr(),\n\t\t\t\t\t\t\tTaskList:              &types.TaskList{Name: failovermanager.TaskListName},\n\t\t\t\t\t\t\tInput: mustMarshalFailoverParams(t, failovermanager.FailoverParams{\n\t\t\t\t\t\t\t\tSourceCluster:                  \"cluster2\",\n\t\t\t\t\t\t\t\tTargetCluster:                  \"cluster1\",\n\t\t\t\t\t\t\t\tDomains:                        []string{\"domain1\", \"domain2\", \"domain3\"},\n\t\t\t\t\t\t\t\tBatchFailoverSize:              20, // default value will be used\n\t\t\t\t\t\t\t\tBatchFailoverWaitTimeInSeconds: 30, // default value will be used\n\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(1200), // default value will be used\n\t\t\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(defaultDecisionTimeoutInSeconds),\n\t\t\t\t\t\t\tMemo: mustGetWorkflowMemo(t, map[string]interface{}{\n\t\t\t\t\t\t\t\tconstants.MemoKeyForOperator: \"test-user\",\n\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t\tWorkflowType: &types.WorkflowType{Name: failovermanager.FailoverWorkflowTypeName},\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif diff := cmp.Diff(wantReq, gotReq); diff != \"\" {\n\t\t\t\t\t\t\tt.Fatalf(\"Request mismatch (-want +got):\\n%s\", diff)\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn resp, nil\n\t\t\t\t\t}).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.desc, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tfrontendCl := frontend.NewMockClient(ctrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttc.mockFn(t, frontendCl)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: frontendCl,\n\t\t\t})\n\n\t\t\targs := []string{\"\", \"admin\", \"cluster\", \"failover\", \"rollback\"}\n\t\t\terr := app.Run(args)\n\n\t\t\tif (err != nil) != tc.wantErr {\n\t\t\t\tt.Errorf(\"Got error: %v, wantErr?: %v\", err, tc.wantErr)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc mustGetWorkflowMemo(t *testing.T, input map[string]interface{}) *types.Memo {\n\tmemo, err := getWorkflowMemo(input)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to get workflow memo: %v\", err)\n\t}\n\treturn memo\n}\n\nfunc mustMarshalQueryResult(t *testing.T, queryResult failovermanager.QueryResult) []byte {\n\tres, err := json.Marshal(queryResult)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to marshal query result: %v\", err)\n\t}\n\treturn res\n}\n\nfunc mustMarshalFailoverParams(t *testing.T, p failovermanager.FailoverParams) []byte {\n\tres, err := json.Marshal(p)\n\tif err != nil {\n\t\tt.Fatalf(\"failed to marshal failover params: %v\", err)\n\t}\n\treturn res\n}\n"
  },
  {
    "path": "tools/cli/admin_kafka_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"regexp\"\n\t\"sync/atomic\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\t\"go.uber.org/thriftrw/wire\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/common/types/mapper/thrift\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\ntype (\n\tfilterFn              func(*types.ReplicationTask) bool\n\tfilterFnForVisibility func(*indexer.Message) bool\n\n\tkafkaMessageType int\n\n\thistoryV2Task struct {\n\t\tTask         *types.ReplicationTask\n\t\tEvents       []*types.HistoryEvent\n\t\tNewRunEvents []*types.HistoryEvent\n\t}\n)\n\nconst (\n\tkafkaMessageTypeReplicationTask kafkaMessageType = iota\n\tkafkaMessageTypeVisibilityMsg\n)\n\nconst (\n\tbufferSize                       = 8192\n\tpreambleVersion0            byte = 0x59\n\tmalformedMessage                 = \"Input was malformed\"\n\tchanBufferSize                   = 10000\n\tmaxRereplicateEventID            = 999999\n\tdefaultResendContextTimeout      = 30 * time.Second\n)\n\nvar (\n\tr = regexp.MustCompile(`Partition: .*?, Offset: .*?, Key: .*?`)\n)\n\ntype writerChannel struct {\n\tType                   kafkaMessageType\n\tReplicationTaskChannel chan *types.ReplicationTask\n\tVisibilityMsgChannel   chan *indexer.Message\n}\n\nfunc newWriterChannel(messageType kafkaMessageType) *writerChannel {\n\tch := &writerChannel{\n\t\tType: messageType,\n\t}\n\tswitch messageType {\n\tcase kafkaMessageTypeReplicationTask:\n\t\tch.ReplicationTaskChannel = make(chan *types.ReplicationTask, chanBufferSize)\n\tcase kafkaMessageTypeVisibilityMsg:\n\t\tch.VisibilityMsgChannel = make(chan *indexer.Message, chanBufferSize)\n\t}\n\treturn ch\n}\n\nfunc (ch *writerChannel) Close() {\n\tif ch.ReplicationTaskChannel != nil {\n\t\tclose(ch.ReplicationTaskChannel)\n\t}\n\tif ch.VisibilityMsgChannel != nil {\n\t\tclose(ch.VisibilityMsgChannel)\n\t}\n}\n\n// AdminKafkaParse parses the output of k8read and outputs replication tasks\nfunc AdminKafkaParse(c *cli.Context) error {\n\tinputFile, err := getInputFile(c.String(FlagInputFile))\n\tdefer inputFile.Close()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in Admin kafka parse: \", err)\n\t}\n\toutputFile, err := getOutputFile(c.String(FlagOutputFilename))\n\tdefer outputFile.Close()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in Admin kafka parse: \", err)\n\t}\n\treaderCh := make(chan []byte, chanBufferSize)\n\twriterCh := newWriterChannel(kafkaMessageType(c.Int(FlagMessageType)))\n\tdoneCh := make(chan struct{})\n\tserializer := persistence.NewPayloadSerializer()\n\n\tvar skippedCount int32\n\tskipErrMode := c.Bool(FlagSkipErrorMode)\n\n\tgo startReader(inputFile, readerCh)\n\tgo startParser(readerCh, writerCh, skipErrMode, &skippedCount)\n\tgo startWriter(outputFile, writerCh, doneCh, &skippedCount, serializer, c)\n\n\t<-doneCh\n\n\tif skipErrMode {\n\t\tfmt.Printf(\"%v messages were skipped due to errors in parsing\", atomic.LoadInt32(&skippedCount))\n\t}\n\treturn nil\n}\n\nfunc buildFilterFn(workflowID, runID string) filterFn {\n\treturn func(task *types.ReplicationTask) bool {\n\t\tif len(workflowID) != 0 || len(runID) != 0 {\n\t\t\tif task.GetHistoryTaskV2Attributes() == nil {\n\t\t\t\treturn false\n\t\t\t}\n\t\t}\n\t\tif len(workflowID) != 0 && task.GetHistoryTaskV2Attributes().WorkflowID != workflowID {\n\t\t\treturn false\n\t\t}\n\t\tif len(runID) != 0 && task.GetHistoryTaskV2Attributes().RunID != runID {\n\t\t\treturn false\n\t\t}\n\t\treturn true\n\t}\n}\n\nfunc buildFilterFnForVisibility(workflowID, runID string) filterFnForVisibility {\n\treturn func(msg *indexer.Message) bool {\n\t\tif len(workflowID) != 0 && msg.GetWorkflowID() != workflowID {\n\t\t\treturn false\n\t\t}\n\t\tif len(runID) != 0 && msg.GetRunID() != runID {\n\t\t\treturn false\n\t\t}\n\t\treturn true\n\t}\n}\n\nfunc getOutputFile(outputFile string) (*os.File, error) {\n\tif len(outputFile) == 0 {\n\t\treturn os.Stdout, nil\n\t}\n\tf, err := os.Create(outputFile)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create output file %w\", err)\n\t}\n\n\treturn f, nil\n}\n\nfunc startReader(file io.Reader, readerCh chan<- []byte) error {\n\tdefer close(readerCh)\n\treader := bufio.NewReader(file)\n\n\tfor {\n\t\tbuf := make([]byte, bufferSize)\n\t\tn, err := reader.Read(buf)\n\t\tif err != nil {\n\t\t\tif err != io.EOF {\n\t\t\t\treturn fmt.Errorf(\"failed to read from reader: %w\", err)\n\t\t\t}\n\t\t\tbreak\n\t\t}\n\t\tbuf = buf[:n]\n\t\treaderCh <- buf\n\t}\n\treturn nil\n}\n\nfunc startParser(readerCh <-chan []byte, writerCh *writerChannel, skipErrors bool, skippedCount *int32) error {\n\tdefer writerCh.Close()\n\n\tvar buffer []byte\n\tfor data := range readerCh {\n\t\tbuffer = append(buffer, data...)\n\t\tparsedData, nextBuffer, err := splitBuffer(buffer)\n\t\tif err != nil {\n\t\t\tif skipErrors {\n\t\t\t\tatomic.AddInt32(skippedCount, 1)\n\t\t\t\tcontinue\n\t\t\t} else {\n\t\t\t\treturn fmt.Errorf(\"error in splitBuffer: %w\", err)\n\t\t\t}\n\t\t}\n\t\tbuffer = nextBuffer\n\t\tparse(parsedData, skipErrors, skippedCount, writerCh)\n\t}\n\tparse(buffer, skipErrors, skippedCount, writerCh)\n\treturn nil\n}\n\nfunc startWriter(\n\toutputFile *os.File,\n\twriterCh *writerChannel,\n\tdoneCh chan struct{},\n\tskippedCount *int32,\n\tserializer persistence.PayloadSerializer,\n\tc *cli.Context,\n) error {\n\tdefer close(doneCh)\n\n\tskipErrMode := c.Bool(FlagSkipErrorMode)\n\theaderMode := c.Bool(FlagHeadersMode)\n\n\tswitch writerCh.Type {\n\tcase kafkaMessageTypeReplicationTask:\n\t\treturn writeReplicationTask(outputFile, writerCh, skippedCount, skipErrMode, headerMode, serializer, c)\n\tcase kafkaMessageTypeVisibilityMsg:\n\t\treturn writeVisibilityMessage(outputFile, writerCh, skippedCount, skipErrMode, headerMode, c)\n\tdefault:\n\t\treturn fmt.Errorf(\"unsupported message type: %v\", writerCh.Type)\n\t}\n}\n\nfunc writeReplicationTask(\n\toutputFile *os.File,\n\twriterCh *writerChannel,\n\tskippedCount *int32,\n\tskipErrMode bool,\n\theaderMode bool,\n\tserializer persistence.PayloadSerializer,\n\tc *cli.Context,\n) error {\n\tfilter := buildFilterFn(c.String(FlagWorkflowID), c.String(FlagRunID))\n\tfor task := range writerCh.ReplicationTaskChannel {\n\t\tif filter(task) {\n\t\t\tjsonStr, err := decodeReplicationTask(task, serializer)\n\t\t\tif err != nil {\n\t\t\t\tif !skipErrMode {\n\t\t\t\t\treturn fmt.Errorf(\"malformed message, failed to encode into json: %w\", err)\n\t\t\t\t}\n\t\t\t\tatomic.AddInt32(skippedCount, 1)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tvar outStr string\n\t\t\tif !headerMode {\n\t\t\t\toutStr = string(jsonStr)\n\t\t\t} else {\n\t\t\t\toutStr = fmt.Sprintf(\n\t\t\t\t\t\"%v, %v, %v\",\n\t\t\t\t\ttask.GetHistoryTaskV2Attributes().DomainID,\n\t\t\t\t\ttask.GetHistoryTaskV2Attributes().WorkflowID,\n\t\t\t\t\ttask.GetHistoryTaskV2Attributes().RunID,\n\t\t\t\t)\n\t\t\t}\n\t\t\t_, err = outputFile.WriteString(fmt.Sprintf(\"%v\\n\", outStr))\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"failed to write to file: %w\", err)\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc writeVisibilityMessage(\n\toutputFile *os.File,\n\twriterCh *writerChannel,\n\tskippedCount *int32,\n\tskipErrMode bool,\n\theaderMode bool,\n\tc *cli.Context,\n) error {\n\tfilter := buildFilterFnForVisibility(c.String(FlagWorkflowID), c.String(FlagRunID))\n\n\tfor msg := range writerCh.VisibilityMsgChannel {\n\t\tif filter(msg) {\n\t\t\tjsonStr, err := json.Marshal(msg)\n\t\t\tif err != nil {\n\t\t\t\tif !skipErrMode {\n\t\t\t\t\treturn fmt.Errorf(\"malformed message: failed to encode into json, err: %w\", err)\n\t\t\t\t}\n\t\t\t\tatomic.AddInt32(skippedCount, 1)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\tvar outStr string\n\t\t\tif !headerMode {\n\t\t\t\toutStr = string(jsonStr)\n\t\t\t} else {\n\t\t\t\toutStr = fmt.Sprintf(\n\t\t\t\t\t\"%v, %v, %v, %v, %v\",\n\t\t\t\t\tmsg.GetDomainID(),\n\t\t\t\t\tmsg.GetWorkflowID(),\n\t\t\t\t\tmsg.GetRunID(),\n\t\t\t\t\tmsg.GetMessageType().String(),\n\t\t\t\t\tmsg.GetVersion(),\n\t\t\t\t)\n\t\t\t}\n\t\t\t_, err = outputFile.WriteString(fmt.Sprintf(\"%v\\n\", outStr))\n\t\t\tif err != nil {\n\t\t\t\treturn fmt.Errorf(\"failed to write to file: %w\", err)\n\t\t\t}\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc splitBuffer(buffer []byte) ([]byte, []byte, error) {\n\tmatches := r.FindAllIndex(buffer, -1)\n\tif len(matches) == 0 {\n\t\treturn nil, nil, fmt.Errorf(\"header not found, did you generate dump with -v\")\n\t}\n\tsplitIndex := matches[len(matches)-1][0]\n\treturn buffer[:splitIndex], buffer[splitIndex:], nil\n}\n\nfunc parse(bytes []byte, skipErrors bool, skippedCount *int32, writerCh *writerChannel) error {\n\tmessages, skippedGetMsgCount, err := getMessages(bytes, skipErrors)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"parsing failed: %w\", err)\n\t}\n\tswitch writerCh.Type {\n\tcase kafkaMessageTypeReplicationTask:\n\t\tmsgs, skippedDeserializeCount, err := deserializeMessages(messages, skipErrors)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"parsing failed: %w\", err)\n\t\t}\n\t\tatomic.AddInt32(skippedCount, skippedGetMsgCount+skippedDeserializeCount)\n\t\tfor _, msg := range msgs {\n\t\t\twriterCh.ReplicationTaskChannel <- msg\n\t\t}\n\tcase kafkaMessageTypeVisibilityMsg:\n\t\tmsgs, skippedDeserializeCount, err := deserializeVisibilityMessages(messages, skipErrors)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"parsing failed: %w\", err)\n\t\t}\n\t\tatomic.AddInt32(skippedCount, skippedGetMsgCount+skippedDeserializeCount)\n\t\tfor _, msg := range msgs {\n\t\t\twriterCh.VisibilityMsgChannel <- msg\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc getMessages(data []byte, skipErrors bool) ([][]byte, int32, error) {\n\tstr := string(data)\n\tmessagesWithHeaders := r.Split(str, -1)\n\tif len(messagesWithHeaders[0]) != 0 {\n\t\treturn nil, 0, fmt.Errorf(malformedMessage+\"Error: %v\", errors.New(\"got data chunk to handle that does not start with valid header\"))\n\t}\n\tmessagesWithHeaders = messagesWithHeaders[1:]\n\tvar rawMessages [][]byte\n\tvar skipped int32\n\tfor _, m := range messagesWithHeaders {\n\t\tif len(m) == 0 {\n\t\t\treturn nil, 0, fmt.Errorf(malformedMessage+\"Error: %v\", errors.New(\"got empty message between valid headers\"))\n\t\t}\n\t\tcurr := []byte(m)\n\t\tmessageStart := bytes.Index(curr, []byte{preambleVersion0})\n\t\tif messageStart == -1 {\n\t\t\tif !skipErrors {\n\t\t\t\treturn nil, 0, fmt.Errorf(malformedMessage+\"Error: %v\", errors.New(\"failed to find message preamble\"))\n\t\t\t}\n\t\t\tskipped++\n\t\t\tcontinue\n\t\t}\n\t\trawMessages = append(rawMessages, curr[messageStart:])\n\t}\n\treturn rawMessages, skipped, nil\n}\n\nfunc deserializeMessages(messages [][]byte, skipErrors bool) ([]*types.ReplicationTask, int32, error) {\n\tvar replicationTasks []*types.ReplicationTask\n\tvar skipped int32\n\tfor _, m := range messages {\n\t\tvar task replicator.ReplicationTask\n\t\terr := decode(m, &task)\n\t\tif err != nil {\n\t\t\tif !skipErrors {\n\t\t\t\treturn nil, 0, fmt.Errorf(malformedMessage+\"Error: %v\", err)\n\t\t\t}\n\t\t\tskipped++\n\t\t\tcontinue\n\t\t}\n\t\treplicationTasks = append(replicationTasks, thrift.ToReplicationTask(&task))\n\t}\n\treturn replicationTasks, skipped, nil\n}\n\nfunc decode(message []byte, val *replicator.ReplicationTask) error {\n\treader := bytes.NewReader(message[1:])\n\twireVal, err := binary.Default.Decode(reader, wire.TStruct)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn val.FromWire(wireVal)\n}\n\nfunc deserializeVisibilityMessages(messages [][]byte, skipErrors bool) ([]*indexer.Message, int32, error) {\n\tvar visibilityMessages []*indexer.Message\n\tvar skipped int32\n\tfor _, m := range messages {\n\t\tvar msg indexer.Message\n\t\terr := decodeVisibility(m, &msg)\n\t\tif err != nil {\n\t\t\tif !skipErrors {\n\t\t\t\treturn nil, 0, fmt.Errorf(malformedMessage+\"Error: %v\", err)\n\t\t\t}\n\t\t\tskipped++\n\t\t\tcontinue\n\t\t}\n\t\tvisibilityMessages = append(visibilityMessages, &msg)\n\t}\n\treturn visibilityMessages, skipped, nil\n}\n\nfunc decodeVisibility(message []byte, val *indexer.Message) error {\n\treader := bytes.NewReader(message[1:])\n\twireVal, err := binary.Default.Decode(reader, wire.TStruct)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn val.FromWire(wireVal)\n}\n\n// ClustersConfig describes the kafka clusters\ntype ClustersConfig struct {\n\tClusters map[string]config.ClusterConfig\n\tTLS      config.TLS\n}\n\nfunc doRereplicate(\n\tctx context.Context,\n\tdomainID string,\n\twid string,\n\trid string,\n\tendEventID *int64,\n\tendEventVersion *int64,\n\tsourceCluster string,\n\tadminClient admin.Client,\n) error {\n\tfmt.Printf(\"Start rereplication for wid: %v, rid:%v \\n\", wid, rid)\n\tif err := adminClient.ResendReplicationTasks(\n\t\tctx,\n\t\t&types.ResendReplicationTasksRequest{\n\t\t\tDomainID:      domainID,\n\t\t\tWorkflowID:    wid,\n\t\t\tRunID:         rid,\n\t\t\tRemoteCluster: sourceCluster,\n\t\t\tEndEventID:    endEventID,\n\t\t\tEndVersion:    endEventVersion,\n\t\t},\n\t); err != nil {\n\t\treturn commoncli.Problem(\"Failed to resend ndc workflow\", err)\n\t}\n\tfmt.Printf(\"Done rereplication for wid: %v, rid:%v \\n\", wid, rid)\n\treturn nil\n}\n\n// AdminRereplicate parses will re-publish replication tasks to topic\nfunc AdminRereplicate(c *cli.Context) error {\n\tsourceCluster, err := getRequiredOption(c, FlagSourceCluster)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar endEventID, endVersion *int64\n\tif c.IsSet(FlagMaxEventID) {\n\t\tendEventID = common.Int64Ptr(c.Int64(FlagMaxEventID) + 1)\n\t}\n\tif c.IsSet(FlagEndEventVersion) {\n\t\tendVersion = common.Int64Ptr(c.Int64(FlagEndEventVersion))\n\t}\n\tdomainID, err := getRequiredOption(c, FlagDomainID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid, err := getRequiredOption(c, FlagRunID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tcontextTimeout := defaultResendContextTimeout\n\n\tif c.IsSet(FlagContextTimeout) {\n\t\tcontextTimeout = time.Duration(c.Int(FlagContextTimeout)) * time.Second\n\t}\n\tctx, cancel := context.WithTimeout(c.Context, contextTimeout)\n\tdefer cancel()\n\n\treturn doRereplicate(\n\t\tctx,\n\t\tdomainID,\n\t\twid,\n\t\trid,\n\t\tendEventID,\n\t\tendVersion,\n\t\tsourceCluster,\n\t\tadminClient,\n\t)\n}\n\nfunc decodeReplicationTask(\n\ttask *types.ReplicationTask,\n\tserializer persistence.PayloadSerializer,\n) ([]byte, error) {\n\tswitch task.GetTaskType() {\n\tcase types.ReplicationTaskTypeHistoryV2:\n\t\thistoryV2 := task.GetHistoryTaskV2Attributes()\n\t\tevents, err := serializer.DeserializeBatchEvents(\n\t\t\tpersistence.NewDataBlobFromInternal(historyV2.Events),\n\t\t)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tvar newRunEvents []*types.HistoryEvent\n\t\tif historyV2.NewRunEvents != nil {\n\t\t\tnewRunEvents, err = serializer.DeserializeBatchEvents(\n\t\t\t\tpersistence.NewDataBlobFromInternal(historyV2.NewRunEvents),\n\t\t\t)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, err\n\t\t\t}\n\t\t}\n\t\thistoryV2.Events = nil\n\t\thistoryV2.NewRunEvents = nil\n\t\thistoryV2Attributes := &historyV2Task{\n\t\t\tTask:         task,\n\t\t\tEvents:       events,\n\t\t\tNewRunEvents: newRunEvents,\n\t\t}\n\t\treturn json.Marshal(historyV2Attributes)\n\tdefault:\n\t\treturn json.Marshal(task)\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_kafka_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"errors\"\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\t\"go.uber.org/thriftrw/protocol/binary\"\n\t\"go.uber.org/thriftrw/ptr\"\n\n\t\"github.com/uber/cadence/.gen/go/indexer\"\n\t\"github.com/uber/cadence/.gen/go/replicator\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nfunc TestWriterChannel(t *testing.T) {\n\tt.Run(\"replication task type\", func(t *testing.T) {\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\n\t\tassert.Equal(t, kafkaMessageType(0), ch.Type)\n\t\tassert.Nil(t, ch.VisibilityMsgChannel)\n\t\tassert.NotNil(t, ch.ReplicationTaskChannel)\n\t\tassert.Equal(t, 10000, cap(ch.ReplicationTaskChannel))\n\t})\n\n\tt.Run(\"msg visibility type\", func(t *testing.T) {\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\n\t\tassert.Equal(t, kafkaMessageType(1), ch.Type)\n\t\tassert.Nil(t, ch.ReplicationTaskChannel)\n\t\tassert.NotNil(t, ch.VisibilityMsgChannel)\n\t\tassert.Equal(t, 10000, cap(ch.VisibilityMsgChannel))\n\t})\n\n\tt.Run(\"replication task type close channel\", func(t *testing.T) {\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tch.ReplicationTaskChannel <- nil\n\t\t_, ok := <-ch.ReplicationTaskChannel\n\t\tassert.True(t, ok)\n\n\t\tch.Close()\n\t\t_, ok = <-ch.ReplicationTaskChannel\n\t\tassert.False(t, ok)\n\t})\n\tt.Run(\"msg visibility type close channel\", func(t *testing.T) {\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\t\tch.VisibilityMsgChannel <- nil\n\t\t_, ok := <-ch.VisibilityMsgChannel\n\t\tassert.True(t, ok)\n\n\t\tch.Close()\n\t\t_, ok = <-ch.VisibilityMsgChannel\n\t\tassert.False(t, ok)\n\t})\n}\n\nfunc TestBuildFilterFn(t *testing.T) {\n\tallTasks := []*types.ReplicationTask{\n\t\tnil,\n\t\t{HistoryTaskV2Attributes: nil},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{RunID: \"run-id-1\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{RunID: \"run-id-2\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-1\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-2\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"alien-workflow-id\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"alien-workflow-id\", RunID: \"run-id-1\"}},\n\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"alien-workflow-id\", RunID: \"run-id-2\"}},\n\t}\n\n\ttests := []struct {\n\t\tname                     string\n\t\tfilterFn                 filterFn\n\t\texpectedFilteredTaskList []*types.ReplicationTask\n\t}{\n\t\t{\"empty filter always return true\", buildFilterFn(\"\", \"\"), allTasks},\n\t\t{\n\t\t\t\"filter with workflowId only\",\n\t\t\tbuildFilterFn(\"my-workflow-id\", \"\"),\n\t\t\t[]*types.ReplicationTask{\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\"}},\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-1\"}},\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-2\"}},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"filter with runId only\",\n\t\t\tbuildFilterFn(\"\", \"run-id-1\"),\n\t\t\t[]*types.ReplicationTask{\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{RunID: \"run-id-1\"}},\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-1\"}},\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"alien-workflow-id\", RunID: \"run-id-1\"}},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"filter with workflow and runId\",\n\t\t\tbuildFilterFn(\"my-workflow-id\", \"run-id-1\"),\n\t\t\t[]*types.ReplicationTask{\n\t\t\t\t{HistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{WorkflowID: \"my-workflow-id\", RunID: \"run-id-1\"}},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar result []*types.ReplicationTask\n\n\t\t\tfor _, task := range allTasks {\n\t\t\t\tif tt.filterFn(task) {\n\t\t\t\t\tresult = append(result, task)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedFilteredTaskList, result)\n\t\t})\n\t}\n}\n\nfunc TestBuildFilterFnForVisibility(t *testing.T) {\n\tallMessages := []*indexer.Message{\n\t\tnil,\n\t\t{},\n\t\t{RunID: ptr.String(\"run-id-1\")},\n\t\t{RunID: ptr.String(\"run-id-2\")},\n\t\t{WorkflowID: ptr.String(\"my-workflow-id\")},\n\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-2\")},\n\t\t{WorkflowID: ptr.String(\"alien-workflow-id\")},\n\t\t{WorkflowID: ptr.String(\"alien-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t{WorkflowID: ptr.String(\"alien-workflow-id\"), RunID: ptr.String(\"run-id-2\")},\n\t}\n\n\ttests := []struct {\n\t\tname                     string\n\t\tfilterFn                 filterFnForVisibility\n\t\texpectedFilteredTaskList []*indexer.Message\n\t}{\n\t\t{\"empty filter always return true\", buildFilterFnForVisibility(\"\", \"\"), allMessages},\n\t\t{\n\t\t\t\"filter with workflowId only\",\n\t\t\tbuildFilterFnForVisibility(\"my-workflow-id\", \"\"),\n\t\t\t[]*indexer.Message{\n\t\t\t\t{WorkflowID: ptr.String(\"my-workflow-id\")},\n\t\t\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-2\")},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"filter with runId only\",\n\t\t\tbuildFilterFnForVisibility(\"\", \"run-id-1\"),\n\t\t\t[]*indexer.Message{\n\t\t\t\t{RunID: ptr.String(\"run-id-1\")},\n\t\t\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t\t\t{WorkflowID: ptr.String(\"alien-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"filter with workflow and runId\",\n\t\t\tbuildFilterFnForVisibility(\"my-workflow-id\", \"run-id-1\"),\n\t\t\t[]*indexer.Message{\n\t\t\t\t{WorkflowID: ptr.String(\"my-workflow-id\"), RunID: ptr.String(\"run-id-1\")},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar result []*indexer.Message\n\n\t\t\tfor _, task := range allMessages {\n\t\t\t\tif tt.filterFn(task) {\n\t\t\t\t\tresult = append(result, task)\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tassert.Equal(t, tt.expectedFilteredTaskList, result)\n\t\t})\n\t}\n}\n\nfunc TestGetOutputFile(t *testing.T) {\n\tt.Run(\"returns stdout if filename is empty\", func(t *testing.T) {\n\t\tfile, err := getOutputFile(\"\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, os.Stdout, file)\n\t})\n\n\tt.Run(\"creates a file\", func(t *testing.T) {\n\t\tfile, err := getOutputFile(\"test.txt\")\n\t\tdefer func() {\n\t\t\tif file != nil {\n\t\t\t\t_ = file.Close()\n\t\t\t\t_ = os.Remove(\"test.txt\")\n\t\t\t}\n\t\t}()\n\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, file)\n\n\t\tinfo, err := file.Stat()\n\n\t\tassert.Equal(t, \"test.txt\", info.Name())\n\t})\n\n\tt.Run(\"fails to create a file\", func(t *testing.T) {\n\t\tfile, err := getOutputFile(\"/non/existent/directory/test.txt\")\n\n\t\tassert.EqualError(t, err, \"failed to create output file open /non/existent/directory/test.txt: no such file or directory\")\n\t\tassert.Nil(t, file)\n\t})\n}\n\ntype FailingIOReader struct{}\n\nfunc (*FailingIOReader) Read(_ []byte) (n int, err error) {\n\treturn 0, errors.New(\"failed to read\")\n}\n\nfunc TestStartReader(t *testing.T) {\n\tt.Run(\"starts reader successfully\", func(t *testing.T) {\n\t\tch := make(chan []byte)\n\t\tfile := bytes.NewBuffer([]byte{1, 2, 3, 4})\n\n\t\tgo func() {\n\t\t\terr := startReader(file, ch)\n\t\t\tassert.NoError(t, err)\n\t\t}()\n\n\t\treadBytes := <-ch\n\n\t\tassert.Equal(t, []byte{1, 2, 3, 4}, readBytes)\n\t})\n\n\tt.Run(\"reader stops successfully at the EOF\", func(t *testing.T) {\n\t\tch := make(chan []byte)\n\t\tfile := bytes.NewBuffer([]byte{1, 2, 3, 4})\n\n\t\tgo func() {\n\t\t\t<-ch\n\t\t}()\n\n\t\terr := startReader(file, ch)\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"returns error if couldn't read\", func(t *testing.T) {\n\t\terr := startReader(&FailingIOReader{}, make(chan<- []byte))\n\t\tassert.EqualError(t, err, \"failed to read from reader: failed to read\")\n\t})\n}\n\nfunc TestStartParser(t *testing.T) {\n\tt.Run(\"skipErrors ignores invalid data\", func(t *testing.T) {\n\t\treaderChannel := make(chan []byte)\n\t\twriterChannel := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tskippedMessagesCount := int32(0)\n\n\t\tgo func() {\n\t\t\treaderChannel <- []byte{1, 2, 3, 4}\n\t\t\tclose(readerChannel)\n\t\t}()\n\n\t\terr := startParser(readerChannel, writerChannel, true, &skippedMessagesCount)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, int32(1), skippedMessagesCount)\n\t})\n\n\tt.Run(\"without skipErrors the invalid data returns error\", func(t *testing.T) {\n\t\treaderChannel := make(chan []byte)\n\t\twriterChannel := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tcount := int32(0)\n\n\t\tgo func() {\n\t\t\treaderChannel <- []byte{1, 2, 3, 4}\n\t\t\tclose(readerChannel)\n\t\t}()\n\n\t\terr := startParser(readerChannel, writerChannel, false, &count)\n\t\tassert.EqualError(t, err, \"error in splitBuffer: header not found, did you generate dump with -v\")\n\t})\n\n\tt.Run(\"successfully start parser\", func(t *testing.T) {\n\t\treaderChannel := make(chan []byte)\n\t\twriterChannel := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tcount := int32(0)\n\n\t\tgo func() {\n\t\t\treaderChannel <- []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\t\tclose(readerChannel)\n\t\t}()\n\n\t\tgo func() {\n\t\t\terr := startParser(readerChannel, writerChannel, false, &count)\n\t\t\tassert.NoError(t, err)\n\t\t}()\n\n\t\tresult := <-writerChannel.ReplicationTaskChannel\n\t\tassert.Equal(t, (*types.ReplicationTask)(nil), result)\n\t})\n}\n\nfunc TestDeserializeMessage(t *testing.T) {\n\tt.Run(\"skipErrors ignores malformed messages\", func(t *testing.T) {\n\t\ttasks, skipped, err := deserializeMessages([][]byte{{1, 2, 3, 4}}, true)\n\n\t\tassert.Equal(t, []*types.ReplicationTask(nil), tasks)\n\t\tassert.Equal(t, int32(1), skipped)\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"without skipErrors malformed messages return error\", func(t *testing.T) {\n\t\ttasks, skipped, err := deserializeMessages([][]byte{{1, 2, 3, 4}}, false)\n\n\t\tassert.Nil(t, tasks)\n\t\tassert.Equal(t, int32(0), skipped)\n\t\tassert.EqualError(t, err, \"Input was malformedError: unexpected EOF\")\n\t})\n\n\tt.Run(\"successful deserialization\", func(t *testing.T) {\n\t\twireVal, _ := (&replicator.ReplicationTask{CreationTime: ptr.Int64(123)}).ToWire()\n\n\t\tencoded := bytes.NewBuffer([]byte{})\n\t\t_ = binary.Default.Encode(wireVal, encoded)\n\n\t\ttasks, skipped, err := deserializeMessages([][]byte{encoded.Bytes()}, false)\n\n\t\tassert.Equal(t, []*types.ReplicationTask{{}}, tasks)\n\t\tassert.Equal(t, int32(0), skipped)\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc TestDeserializeVisibilityMessage(t *testing.T) {\n\tt.Run(\"skipErrors ignores malformed messages\", func(t *testing.T) {\n\t\ttasks, skipped, err := deserializeVisibilityMessages([][]byte{{1, 2, 3, 4}}, true)\n\n\t\tassert.Equal(t, []*indexer.Message(nil), tasks)\n\t\tassert.Equal(t, int32(1), skipped)\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"without skipErrors malformed messages return error\", func(t *testing.T) {\n\t\ttasks, skipped, err := deserializeVisibilityMessages([][]byte{{1, 2, 3, 4}}, false)\n\n\t\tassert.Nil(t, tasks)\n\t\tassert.Equal(t, int32(0), skipped)\n\t\tassert.EqualError(t, err, \"Input was malformedError: unexpected EOF\")\n\t})\n\n\tt.Run(\"successful deserialization\", func(t *testing.T) {\n\t\twireVal, _ := (&indexer.Message{Version: ptr.Int64(123)}).ToWire()\n\n\t\tencoded := bytes.NewBuffer([]byte{})\n\t\t_ = binary.Default.Encode(wireVal, encoded)\n\n\t\ttasks, skipped, err := deserializeVisibilityMessages([][]byte{encoded.Bytes()}, false)\n\n\t\tassert.Equal(t, []*indexer.Message{{}}, tasks)\n\t\tassert.Equal(t, int32(0), skipped)\n\t\tassert.NoError(t, err)\n\t})\n}\n\nfunc TestDecodeReplicationTask(t *testing.T) {\n\tt.Run(\"decode replication task\", func(t *testing.T) {\n\t\tresult, err := decodeReplicationTask(&types.ReplicationTask{TaskType: types.ReplicationTaskTypeHistory.Ptr()}, persistence.NewPayloadSerializer())\n\t\texpectedTaskJSONBytes := []byte(\"{\\\"taskType\\\":\\\"History\\\"}\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, expectedTaskJSONBytes, result)\n\t})\n\n\tt.Run(\"decode replication task type HistoryV2\", func(t *testing.T) {\n\t\ttask := &types.ReplicationTask{\n\t\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\tEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: nil,\n\t\t\t\t\tData:         nil,\n\t\t\t\t},\n\t\t\t\tNewRunEvents: &types.DataBlob{\n\t\t\t\t\tEncodingType: nil,\n\t\t\t\t\tData:         nil,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tresult, err := decodeReplicationTask(task, persistence.NewPayloadSerializer())\n\t\texpectedTaskJSONBytes := []byte(\"{\\\"Task\\\":{\\\"taskType\\\":\\\"HistoryV2\\\",\\\"historyTaskV2Attributes\\\":{}},\\\"Events\\\":null,\\\"NewRunEvents\\\":null}\")\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, expectedTaskJSONBytes, result)\n\t})\n\n\tt.Run(\"deserialize returns an error\", func(t *testing.T) {\n\t\ttask := &types.ReplicationTask{\n\t\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\tEvents: &types.DataBlob{EncodingType: nil, Data: []byte{}},\n\t\t\t},\n\t\t}\n\n\t\teventsBlob := persistence.NewDataBlobFromInternal(&types.DataBlob{EncodingType: nil, Data: []byte{}})\n\t\tserializerMock := persistence.NewMockPayloadSerializer(gomock.NewController(t))\n\t\tserializerMock.EXPECT().DeserializeBatchEvents(eventsBlob).Return(nil, assert.AnError).Times(1)\n\n\t\tresult, err := decodeReplicationTask(task, serializerMock)\n\n\t\tassert.Equal(t, err, assert.AnError)\n\t\tassert.Nil(t, result)\n\t})\n\n\tt.Run(\"deserialize returns an error for NewRunEvents\", func(t *testing.T) {\n\t\ttask := &types.ReplicationTask{\n\t\t\tTaskType: types.ReplicationTaskTypeHistoryV2.Ptr(),\n\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\tEvents:       &types.DataBlob{EncodingType: nil, Data: []byte{}},\n\t\t\t\tNewRunEvents: &types.DataBlob{EncodingType: nil, Data: []byte{}},\n\t\t\t},\n\t\t}\n\n\t\teventsBlob := persistence.NewDataBlobFromInternal(&types.DataBlob{EncodingType: nil, Data: []byte{}})\n\t\tnewRunEventsBlob := persistence.NewDataBlobFromInternal(&types.DataBlob{EncodingType: nil, Data: []byte{}})\n\t\tserializerMock := persistence.NewMockPayloadSerializer(gomock.NewController(t))\n\t\tserializerMock.EXPECT().DeserializeBatchEvents(eventsBlob).Return(nil, nil).Times(1)\n\t\tserializerMock.EXPECT().DeserializeBatchEvents(newRunEventsBlob).Return(nil, assert.AnError).Times(1)\n\n\t\tresult, err := decodeReplicationTask(task, serializerMock)\n\n\t\tassert.Equal(t, err, assert.AnError)\n\t\tassert.Nil(t, result)\n\t})\n}\n\nfunc TestDoRereplicate(t *testing.T) {\n\tt.Run(\"successful rereplication\", func(t *testing.T) {\n\t\tclientMock := admin.NewMockClient(gomock.NewController(t))\n\t\tclientMock.EXPECT().ResendReplicationTasks(context.Background(), &types.ResendReplicationTasksRequest{\n\t\t\tDomainID:      \"domainID\",\n\t\t\tWorkflowID:    \"wid\",\n\t\t\tRunID:         \"rid\",\n\t\t\tRemoteCluster: \"sourceCluster\",\n\t\t\tEndEventID:    ptr.Int64(1),\n\t\t\tEndVersion:    ptr.Int64(1),\n\t\t}).Return(nil).Times(1)\n\n\t\terr := doRereplicate(context.Background(), \"domainID\", \"wid\", \"rid\", ptr.Int64(1), ptr.Int64(1), \"sourceCluster\", clientMock)\n\n\t\tassert.NoError(t, err)\n\t})\n\n\tt.Run(\"returns err\", func(t *testing.T) {\n\t\tclientMock := admin.NewMockClient(gomock.NewController(t))\n\t\tclientMock.EXPECT().ResendReplicationTasks(context.Background(), &types.ResendReplicationTasksRequest{\n\t\t\tDomainID:      \"domainID\",\n\t\t\tWorkflowID:    \"wid\",\n\t\t\tRunID:         \"rid\",\n\t\t\tRemoteCluster: \"sourceCluster\",\n\t\t\tEndEventID:    ptr.Int64(1),\n\t\t\tEndVersion:    ptr.Int64(1),\n\t\t}).Return(assert.AnError).Times(1)\n\n\t\terr := doRereplicate(context.Background(), \"domainID\", \"wid\", \"rid\", ptr.Int64(1), ptr.Int64(1), \"sourceCluster\", clientMock)\n\n\t\tassert.EqualError(t, err, \"Failed to resend ndc workflow: assert.AnError general error for testing\")\n\t})\n}\n\nfunc TestAdminRereplicate(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\targs []clitest.CliArgument\n\t\terr  error\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagSourceCluster, \"sourceCluster\"),\n\t\t\t\tclitest.IntArgument(FlagMaxEventID, 123),\n\t\t\t\tclitest.IntArgument(FlagEndEventVersion, 321),\n\t\t\t\tclitest.StringArgument(FlagDomainID, \"domainID\"),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"workflowID\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"runID\"),\n\t\t\t\tclitest.StringArgument(FlagContextTimeout, \"10s\"),\n\t\t\t},\n\t\t\terr: nil,\n\t\t},\n\t\t{\n\t\t\tname: \"source cluster is missing\",\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.IntArgument(FlagMaxEventID, 123),\n\t\t\t\tclitest.IntArgument(FlagEndEventVersion, 321),\n\t\t\t\tclitest.StringArgument(FlagDomainID, \"domainID\"),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"workflowID\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"runID\"),\n\t\t\t\tclitest.StringArgument(FlagContextTimeout, \"10s\"),\n\t\t\t},\n\t\t\terr: commoncli.Problem(\"Required flag not found: \", errors.New(\"option source_cluster is required\")),\n\t\t},\n\t\t{\n\t\t\tname: \"domain ID is missing\",\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagSourceCluster, \"sourceCluster\"),\n\t\t\t\tclitest.IntArgument(FlagMaxEventID, 123),\n\t\t\t\tclitest.IntArgument(FlagEndEventVersion, 321),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"workflowID\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"runID\"),\n\t\t\t\tclitest.StringArgument(FlagContextTimeout, \"10s\"),\n\t\t\t},\n\t\t\terr: commoncli.Problem(\"Required flag not found: \", errors.New(\"option domain_id is required\")),\n\t\t},\n\t\t{\n\t\t\tname: \"workflow ID is missing\",\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagSourceCluster, \"sourceCluster\"),\n\t\t\t\tclitest.IntArgument(FlagMaxEventID, 123),\n\t\t\t\tclitest.IntArgument(FlagEndEventVersion, 321),\n\t\t\t\tclitest.StringArgument(FlagDomainID, \"domainID\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"runID\"),\n\t\t\t\tclitest.StringArgument(FlagContextTimeout, \"10s\"),\n\t\t\t},\n\t\t\terr: commoncli.Problem(\"Required flag not found: \", errors.New(\"option workflow_id is required\")),\n\t\t},\n\t\t{\n\t\t\tname: \"run ID is missing\",\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagSourceCluster, \"sourceCluster\"),\n\t\t\t\tclitest.IntArgument(FlagMaxEventID, 123),\n\t\t\t\tclitest.IntArgument(FlagEndEventVersion, 321),\n\t\t\t\tclitest.StringArgument(FlagDomainID, \"domainID\"),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"workflowID\"),\n\t\t\t\tclitest.StringArgument(FlagContextTimeout, \"10s\"),\n\t\t\t},\n\t\t\terr: commoncli.Problem(\"Required flag not found: \", errors.New(\"option run_id is required\")),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttestData := newCLITestData(t)\n\t\t\ttestData.mockAdminClient.EXPECT().ResendReplicationTasks(gomock.Any(), &types.ResendReplicationTasksRequest{\n\t\t\t\tDomainID:      \"domainID\",\n\t\t\t\tWorkflowID:    \"workflowID\",\n\t\t\t\tRunID:         \"runID\",\n\t\t\t\tRemoteCluster: \"sourceCluster\",\n\t\t\t\tEndEventID:    ptr.Int64(124),\n\t\t\t\tEndVersion:    ptr.Int64(321),\n\t\t\t}).Return(nil).MaxTimes(1)\n\n\t\t\tcliContext := clitest.NewCLIContext(t, testData.app, tt.args...)\n\n\t\t\terr := AdminRereplicate(cliContext)\n\t\t\tassert.Equal(t, tt.err, err)\n\t\t})\n\t}\n}\n\nfunc TestParse(t *testing.T) {\n\tt.Run(\"successful parse replication task\", func(t *testing.T) {\n\t\twireVal, _ := (&replicator.ReplicationTask{CreationTime: ptr.Int64(123)}).ToWire()\n\t\tencoded := bytes.NewBuffer([]byte{})\n\t\t_ = binary.Default.Encode(wireVal, encoded)\n\t\tdata := []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\tdata = append(data, preambleVersion0)\n\t\tdata = append(data, encoded.Bytes()...)\n\n\t\tskipped := ptr.Int32(0)\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\n\t\tgo func() {\n\t\t\terr := parse(data, false, skipped, ch)\n\t\t\tassert.NoError(t, err)\n\t\t\tclose(ch.ReplicationTaskChannel)\n\t\t}()\n\n\t\tresult := <-ch.ReplicationTaskChannel\n\n\t\tassert.NotNil(t, result)\n\t})\n\n\tt.Run(\"successful parse visibility task\", func(t *testing.T) {\n\t\twireVal, _ := (&indexer.Message{Version: ptr.Int64(123)}).ToWire()\n\t\tencoded := bytes.NewBuffer([]byte{})\n\t\t_ = binary.Default.Encode(wireVal, encoded)\n\t\tdata := []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\tdata = append(data, preambleVersion0)\n\t\tdata = append(data, encoded.Bytes()...)\n\n\t\tskipped := ptr.Int32(0)\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\n\t\tgo func() {\n\t\t\terr := parse(data, false, skipped, ch)\n\t\t\tassert.NoError(t, err)\n\t\t\tclose(ch.VisibilityMsgChannel)\n\t\t}()\n\n\t\tresult := <-ch.VisibilityMsgChannel\n\n\t\tassert.NotNil(t, result)\n\t})\n\n\tt.Run(\"invalid replication task\", func(t *testing.T) {\n\t\tdata := []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\tdata = append(data, preambleVersion0)\n\t\tdata = append(data, []byte{1, 2, 3, 4}...)\n\n\t\tskipped := ptr.Int32(0)\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\n\t\tgo func() {\n\t\t\terr := parse(data, false, skipped, ch)\n\t\t\tassert.EqualError(t, err, \"parsing failed: Input was malformedError: unknown ttype Type(1)\")\n\t\t\tclose(ch.ReplicationTaskChannel)\n\t\t}()\n\n\t\tresult := <-ch.ReplicationTaskChannel\n\n\t\tassert.Nil(t, result)\n\t})\n\n\tt.Run(\"invalid visibility task\", func(t *testing.T) {\n\t\tdata := []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\tdata = append(data, preambleVersion0)\n\t\tdata = append(data, []byte{1, 2, 3, 4}...)\n\n\t\tskipped := ptr.Int32(0)\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\n\t\tgo func() {\n\t\t\terr := parse(data, false, skipped, ch)\n\t\t\tassert.EqualError(t, err, \"parsing failed: Input was malformedError: unknown ttype Type(1)\")\n\t\t\tclose(ch.VisibilityMsgChannel)\n\t\t}()\n\n\t\tresult := <-ch.VisibilityMsgChannel\n\n\t\tassert.Nil(t, result)\n\t})\n}\n\nfunc TestAdminKafkaParse(t *testing.T) {\n\tt.Run(\"success\", func(t *testing.T) {\n\t\twireVal, _ := (&replicator.ReplicationTask{CreationTime: ptr.Int64(123)}).ToWire()\n\t\tencoded := bytes.NewBuffer([]byte{})\n\t\t_ = binary.Default.Encode(wireVal, encoded)\n\t\tdata := []byte(\"Partition: abc, Offset: 0, Key: 123\")\n\t\tdata = append(data, preambleVersion0)\n\t\tdata = append(data, encoded.Bytes()...)\n\n\t\tinputFileName := createTempFileWithContent(t, string(data))\n\t\toutputFileName := createTempFileWithContent(t, \"\")\n\n\t\ttestData := newCLITestData(t)\n\n\t\tcliContext := clitest.NewCLIContext(t, testData.app,\n\t\t\tclitest.StringArgument(FlagInputFile, inputFileName),\n\t\t\tclitest.StringArgument(FlagOutputFilename, outputFileName),\n\t\t)\n\n\t\terr := AdminKafkaParse(cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, err := os.ReadFile(outputFileName)\n\n\t\tassert.Equal(t, \"{\\\"creationTime\\\":123}\\n\", string(result))\n\t})\n\n\tt.Run(\"can't open input file\", func(t *testing.T) {\n\t\ttestData := newCLITestData(t)\n\n\t\tcliContext := clitest.NewCLIContext(t, testData.app,\n\t\t\tclitest.StringArgument(FlagInputFile, \"/path/do/not/exist/input.txt\"),\n\t\t\tclitest.StringArgument(FlagOutputFilename, \"output.txt\"),\n\t\t)\n\n\t\terr := AdminKafkaParse(cliContext)\n\n\t\tassert.EqualError(t, err, \"Error in Admin kafka parse: : failed to open input file for reading: /path/do/not/exist/input.txt: open /path/do/not/exist/input.txt: no such file or directory\")\n\t})\n\n\tt.Run(\"can't open output file\", func(t *testing.T) {\n\t\ttestData := newCLITestData(t)\n\n\t\tinputFileName := createTempFileWithContent(t, \"\")\n\n\t\tcliContext := clitest.NewCLIContext(t, testData.app,\n\t\t\tclitest.StringArgument(FlagInputFile, inputFileName),\n\t\t\tclitest.StringArgument(FlagOutputFilename, \"/path/do/not/exist/output.txt\"),\n\t\t)\n\n\t\terr := AdminKafkaParse(cliContext)\n\n\t\tassert.EqualError(t, err, \"Error in Admin kafka parse: : failed to create output file open /path/do/not/exist/output.txt: no such file or directory\")\n\t})\n}\n\nfunc TestWriteVisibilityMessage(t *testing.T) {\n\tt.Run(\"successful write\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.VisibilityMsgChannel <- &indexer.Message{\n\t\t\t\tDomainID:   ptr.String(\"domain_id\"),\n\t\t\t\tWorkflowID: ptr.String(\"workflow_id\"),\n\t\t\t\tRunID:      ptr.String(\"run_id\"),\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := writeVisibilityMessage(file, ch, skippedCount, false, false, cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"{\\\"domainID\\\":\\\"domain_id\\\",\\\"workflowID\\\":\\\"workflow_id\\\",\\\"runID\\\":\\\"run_id\\\"}\\n\", string(result))\n\t})\n\n\tt.Run(\"successful write with header mode\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.VisibilityMsgChannel <- &indexer.Message{\n\t\t\t\tDomainID:   ptr.String(\"domain_id\"),\n\t\t\t\tWorkflowID: ptr.String(\"workflow_id\"),\n\t\t\t\tRunID:      ptr.String(\"run_id\"),\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := writeVisibilityMessage(file, ch, skippedCount, false, true, cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"domain_id, workflow_id, run_id, Index, 0\\n\", string(result))\n\t})\n\n\tt.Run(\"returns error if can't write to the file\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.VisibilityMsgChannel <- &indexer.Message{}\n\t\t\tch.Close()\n\t\t}()\n\n\t\t_ = file.Close()\n\n\t\terr := writeVisibilityMessage(file, ch, skippedCount, false, true, cliContext)\n\n\t\tassert.Errorf(t, err, \"failed to write to file: write %v: file already closed\", filename)\n\t})\n}\n\nfunc TestWriteReplicationTask(t *testing.T) {\n\tt.Run(\"successful write\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.ReplicationTaskChannel <- &types.ReplicationTask{\n\t\t\t\tSourceTaskID: 123,\n\t\t\t\tCreationTime: ptr.Int64(123),\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := writeReplicationTask(file, ch, skippedCount, false, false, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"{\\\"sourceTaskId\\\":123,\\\"creationTime\\\":123}\\n\", string(result))\n\t})\n\n\tt.Run(\"successful write with header mode\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.ReplicationTaskChannel <- &types.ReplicationTask{\n\t\t\t\tSourceTaskID: 123,\n\t\t\t\tCreationTime: ptr.Int64(123),\n\t\t\t\tHistoryTaskV2Attributes: &types.HistoryTaskV2Attributes{\n\t\t\t\t\tDomainID:   \"domain_id\",\n\t\t\t\t\tWorkflowID: \"workflow_id\",\n\t\t\t\t\tRunID:      \"run_id\",\n\t\t\t\t},\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := writeReplicationTask(file, ch, skippedCount, false, true, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"domain_id, workflow_id, run_id\\n\", string(result))\n\t})\n\n\tt.Run(\"returns error if can't write to the file\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.ReplicationTaskChannel <- &types.ReplicationTask{}\n\t\t\tch.Close()\n\t\t}()\n\n\t\tfile.Close()\n\n\t\terr := writeReplicationTask(file, ch, skippedCount, false, false, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.Errorf(t, err, \"failed to write to file: write %v: file already closed\", filename)\n\t})\n}\n\nfunc TestStartWriter(t *testing.T) {\n\tt.Run(\"successful write to replication task\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeReplicationTask)\n\t\tdoneCh := make(chan struct{})\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.ReplicationTaskChannel <- &types.ReplicationTask{\n\t\t\t\tSourceTaskID: 123,\n\t\t\t\tCreationTime: ptr.Int64(123),\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := startWriter(file, ch, doneCh, skippedCount, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"{\\\"sourceTaskId\\\":123,\\\"creationTime\\\":123}\\n\", string(result))\n\t})\n\n\tt.Run(\"successful write with visibility task\", func(t *testing.T) {\n\t\tfilename := createTempFileWithContent(t, \"\")\n\t\tfile, _ := os.OpenFile(filename, os.O_WRONLY, 0600)\n\t\tdefer file.Close()\n\n\t\tch := newWriterChannel(kafkaMessageTypeVisibilityMsg)\n\t\tdoneCh := make(chan struct{})\n\t\tskippedCount := ptr.Int32(0)\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\tgo func() {\n\t\t\tch.VisibilityMsgChannel <- &indexer.Message{\n\t\t\t\tDomainID:   ptr.String(\"domain_id\"),\n\t\t\t\tWorkflowID: ptr.String(\"workflow_id\"),\n\t\t\t\tRunID:      ptr.String(\"run_id\"),\n\t\t\t}\n\t\t\tch.Close()\n\t\t}()\n\n\t\terr := startWriter(file, ch, doneCh, skippedCount, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.NoError(t, err)\n\n\t\tresult, _ := os.ReadFile(filename)\n\n\t\tassert.Equal(t, \"{\\\"domainID\\\":\\\"domain_id\\\",\\\"workflowID\\\":\\\"workflow_id\\\",\\\"runID\\\":\\\"run_id\\\"}\\n\", string(result))\n\t})\n\n\tt.Run(\"unsupported type\", func(t *testing.T) {\n\t\tch := newWriterChannel(kafkaMessageType(321))\n\t\tdoneCh := make(chan struct{})\n\t\tcliContext := clitest.NewCLIContext(t, newCLITestData(t).app)\n\n\t\terr := startWriter(nil, ch, doneCh, nil, persistence.NewPayloadSerializer(), cliContext)\n\n\t\tassert.EqualError(t, err, \"unsupported message type: 321\")\n\t})\n}\n"
  },
  {
    "path": "tools/cli/admin_task_list_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"slices\"\n\t\"strings\"\n\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/multierr\"\n\t\"golang.org/x/exp/maps\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\ntype (\n\tTaskListRow struct {\n\t\tName                string `header:\"Task List Name\"`\n\t\tActivityPollerCount int    `header:\"Activity Poller Count\"`\n\t\tDecisionPollerCount int    `header:\"Decision Poller Count\"`\n\t}\n\tTaskListStatusRow struct {\n\t\tType        string  `header:\"Type\"`\n\t\tPollerCount int     `header:\"PollerCount\"`\n\t\tReadLevel   int64   `header:\"Read Level\"`\n\t\tAckLevel    int64   `header:\"Ack Level\"`\n\t\tBacklog     int64   `header:\"Backlog\"`\n\t\tRPS         float64 `header:\"RPS\"`\n\t\tStartID     int64   `header:\"Lease Start TaskID\"`\n\t\tEndID       int64   `header:\"Lease End TaskID\"`\n\t}\n\tTaskListPartitionConfigRow struct {\n\t\tType            string                           `header:\"Type\"`\n\t\tVersion         int64                            `header:\"Version\"`\n\t\tReadPartitions  map[int]*types.TaskListPartition `header:\"Read Partitions\"`\n\t\tWritePartitions map[int]*types.TaskListPartition `header:\"Write Partitions\"`\n\t}\n)\n\n// AdminDescribeTaskList displays poller and status information of task list.\nfunc AdminDescribeTaskList(c *cli.Context) error {\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskList, err := getRequiredOption(c, FlagTaskList)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskListTypes, err := getTaskListTypes(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\tresponses := make(map[types.TaskListType]*types.DescribeTaskListResponse)\n\tfor _, tlType := range taskListTypes {\n\t\trequest := &types.DescribeTaskListRequest{\n\t\t\tDomain:                domain,\n\t\t\tTaskList:              &types.TaskList{Name: taskList},\n\t\t\tTaskListType:          tlType.Ptr(),\n\t\t\tIncludeTaskListStatus: true,\n\t\t}\n\n\t\tresponse, err := frontendClient.DescribeTaskList(ctx, request)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Operation DescribeTaskList failed for type: \"+tlType.String(), err)\n\t\t}\n\t\tresponses[tlType] = response\n\t}\n\tif c.String(FlagFormat) == formatJSON {\n\t\tprettyPrintJSONObject(getDeps(c).Output(), responses)\n\t\treturn nil\n\t}\n\n\tif err := printTaskListStatus(getDeps(c).Output(), responses); err != nil {\n\t\treturn fmt.Errorf(\"failed to print task list status: %w\", err)\n\t}\n\tgetDeps(c).Output().Write([]byte(\"\\n\"))\n\tif err := printTaskListPartitionConfig(getDeps(c).Output(), responses); err != nil {\n\t\treturn fmt.Errorf(\"failed to print task list partition config: %w\", err)\n\t}\n\tgetDeps(c).Output().Write([]byte(\"\\n\"))\n\n\treturn nil\n}\n\n// AdminListTaskList displays all task lists under a domain.\nfunc AdminListTaskList(c *cli.Context) error {\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\trequest := &types.GetTaskListsByDomainRequest{\n\t\tDomain: domain,\n\t}\n\n\tresponse, err := frontendClient.GetTaskListsByDomain(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation GetTaskListByDomain failed.\", err)\n\t}\n\n\tif c.String(FlagFormat) == formatJSON {\n\t\tprettyPrintJSONObject(getDeps(c).Output(), response)\n\t\treturn nil\n\t}\n\n\t_, _ = fmt.Fprintln(getDeps(c).Output(), \"Task Lists for domain \", domain, \":\")\n\ttlByName := make(map[string]TaskListRow)\n\tfor name, taskList := range response.GetDecisionTaskListMap() {\n\t\trow := tlByName[name]\n\t\trow.Name = name\n\t\trow.DecisionPollerCount = len(taskList.GetPollers())\n\t\ttlByName[name] = row\n\t}\n\tfor name, taskList := range response.GetActivityTaskListMap() {\n\t\trow := tlByName[name]\n\t\trow.Name = name\n\t\trow.ActivityPollerCount = len(taskList.GetPollers())\n\t\ttlByName[name] = row\n\t}\n\ttable := maps.Values(tlByName)\n\tslices.SortFunc(table, func(a, b TaskListRow) int {\n\t\treturn strings.Compare(a.Name, b.Name)\n\t})\n\treturn RenderTable(os.Stdout, table, RenderOptions{Color: true, Border: true})\n}\n\nfunc printTaskListStatus(w io.Writer, responses map[types.TaskListType]*types.DescribeTaskListResponse) error {\n\tvar table []TaskListStatusRow\n\tfor tlType, response := range responses {\n\t\ttaskListStatus := response.TaskListStatus\n\t\ttable = append(table, TaskListStatusRow{\n\t\t\tType:        tlType.String(),\n\t\t\tPollerCount: len(response.Pollers),\n\t\t\tReadLevel:   taskListStatus.GetReadLevel(),\n\t\t\tAckLevel:    taskListStatus.GetAckLevel(),\n\t\t\tBacklog:     taskListStatus.GetBacklogCountHint(),\n\t\t\tRPS:         taskListStatus.GetRatePerSecond(),\n\t\t\tStartID:     taskListStatus.GetTaskIDBlock().GetStartID(),\n\t\t\tEndID:       taskListStatus.GetTaskIDBlock().GetEndID(),\n\t\t})\n\t}\n\tslices.SortFunc(table, func(a, b TaskListStatusRow) int {\n\t\treturn strings.Compare(a.Type, b.Type)\n\t})\n\treturn RenderTable(w, table, RenderOptions{Color: true})\n}\n\nfunc printTaskListPartitionConfig(w io.Writer, responses map[types.TaskListType]*types.DescribeTaskListResponse) error {\n\tvar table []TaskListPartitionConfigRow\n\tfor tlType, response := range responses {\n\t\tconfig := response.PartitionConfig\n\t\tif config == nil {\n\t\t\tconfig = &types.TaskListPartitionConfig{\n\t\t\t\tVersion:         0,\n\t\t\t\tReadPartitions:  createPartitions(1),\n\t\t\t\tWritePartitions: createPartitions(1),\n\t\t\t}\n\t\t}\n\t\ttable = append(table, TaskListPartitionConfigRow{\n\t\t\tType:            tlType.String(),\n\t\t\tVersion:         config.Version,\n\t\t\tReadPartitions:  config.ReadPartitions,\n\t\t\tWritePartitions: config.WritePartitions,\n\t\t})\n\t}\n\tslices.SortFunc(table, func(a, b TaskListPartitionConfigRow) int {\n\t\treturn strings.Compare(a.Type, b.Type)\n\t})\n\treturn RenderTable(w, table, RenderOptions{Color: true})\n}\n\nfunc AdminUpdateTaskListPartitionConfig(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskList, err := getRequiredOption(c, FlagTaskList)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tforce := c.Bool(FlagForce)\n\ttaskListTypes, err := getTaskListTypes(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tnumReadPartitions, err := getRequiredIntOption(c, FlagNumReadPartitions)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tnumWritePartitions, err := getRequiredIntOption(c, FlagNumWritePartitions)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\tcfg := &types.TaskListPartitionConfig{\n\t\tReadPartitions:  createPartitions(numReadPartitions),\n\t\tWritePartitions: createPartitions(numWritePartitions),\n\t}\n\ttl := &types.TaskList{Name: taskList, Kind: types.TaskListKindNormal.Ptr()}\n\tif !force {\n\t\thasPollers := false\n\t\tvar errors []error\n\t\tfor _, tlType := range taskListTypes {\n\t\t\ttypeHasPollers, typeErr := validateChange(ctx, frontendClient, domain, tl, tlType.Ptr(), cfg)\n\t\t\tif typeErr != nil {\n\t\t\t\terrors = append(errors, fmt.Errorf(\"%s:%s failed validation: %w\", tl.Name, tlType, typeErr))\n\t\t\t}\n\t\t\thasPollers = typeHasPollers || hasPollers\n\t\t}\n\t\tif len(errors) > 0 {\n\t\t\treturn commoncli.Problem(\"Potentially unsafe operation. Specify '--force' to proceed anyway\", multierr.Combine(errors...))\n\t\t}\n\t\tif !hasPollers {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Operation is safe but %s has no pollers of the specified types. Is the name correct? Specify '--force' to proceed anyway\", tl.Name), nil)\n\t\t}\n\t}\n\tfor _, tlType := range taskListTypes {\n\t\t_, err = adminClient.UpdateTaskListPartitionConfig(ctx, &types.UpdateTaskListPartitionConfigRequest{\n\t\t\tDomain:          domain,\n\t\t\tTaskList:        tl,\n\t\t\tTaskListType:    tlType.Ptr(),\n\t\t\tPartitionConfig: cfg,\n\t\t})\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Operation UpdateTaskListPartitionConfig failed for type: \"+tlType.String(), err)\n\t\t}\n\t\t_, _ = fmt.Fprintln(getDeps(c).Output(), \"Successfully updated \", tl.Name, \":\", tlType)\n\t}\n\treturn nil\n}\n\nfunc validateChange(ctx context.Context, client frontend.Client, domain string, tl *types.TaskList, tlt *types.TaskListType, newCfg *types.TaskListPartitionConfig) (bool, error) {\n\tdescription, err := client.DescribeTaskList(ctx, &types.DescribeTaskListRequest{\n\t\tDomain:       domain,\n\t\tTaskList:     tl,\n\t\tTaskListType: tlt,\n\t})\n\tif err != nil {\n\t\treturn false, fmt.Errorf(\"DescribeTaskList failed: %w\", err)\n\t}\n\t// Illegal operations are rejected by the server (read < write), but unsafe ones are still allowed\n\tif description.PartitionConfig != nil {\n\t\toldCfg := description.PartitionConfig\n\t\t// Ensure they're not removing active write partitions\n\t\tif len(newCfg.ReadPartitions) < len(oldCfg.WritePartitions) {\n\t\t\treturn false, fmt.Errorf(\"remove write partitions, then read partitions. Removing an active write partition risks losing tasks. Proposed read count is less than current write count (%d < %d)\", len(newCfg.ReadPartitions), len(oldCfg.WritePartitions))\n\t\t}\n\t\t// Ensure removed read partitions are drained\n\t\tfor i := len(newCfg.ReadPartitions); i < len(oldCfg.ReadPartitions); i++ {\n\t\t\tpartition, err := client.DescribeTaskList(ctx, &types.DescribeTaskListRequest{\n\t\t\t\tDomain:                domain,\n\t\t\t\tTaskList:              &types.TaskList{Name: getPartitionTaskListName(tl.Name, i), Kind: tl.Kind},\n\t\t\t\tTaskListType:          tlt,\n\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\treturn false, fmt.Errorf(\"DescribeTaskList failed for partition %d: %w\", i, err)\n\t\t\t}\n\t\t\tif partition.TaskListStatus.BacklogCountHint != 0 {\n\t\t\t\treturn false, fmt.Errorf(\"partition %d still has %d tasks remaining\", i, partition.TaskListStatus.BacklogCountHint)\n\t\t\t}\n\t\t}\n\t}\n\t// If it's otherwise valid but there are no pollers, they might have mistyped the name\n\treturn len(description.Pollers) > 0, nil\n}\n\nfunc createPartitions(num int) map[int]*types.TaskListPartition {\n\tresult := make(map[int]*types.TaskListPartition, num)\n\tfor i := 0; i < num; i++ {\n\t\tresult[i] = &types.TaskListPartition{}\n\t}\n\treturn result\n}\n\nfunc getPartitionTaskListName(root string, partition int) string {\n\tif partition <= 0 {\n\t\treturn root\n\t}\n\treturn fmt.Sprintf(\"%v%v/%v\", constants.ReservedTaskListPrefix, root, partition)\n}\n\nfunc getTaskListTypes(c *cli.Context) ([]types.TaskListType, error) {\n\tvar taskListTypes []types.TaskListType\n\tif strings.ToLower(c.String(FlagTaskListType)) == \"activity\" {\n\t\ttaskListTypes = []types.TaskListType{types.TaskListTypeActivity}\n\t} else if strings.ToLower(c.String(FlagTaskListType)) == \"decision\" {\n\t\ttaskListTypes = []types.TaskListType{types.TaskListTypeDecision}\n\t} else if c.String(FlagTaskListType) == \"\" {\n\t\ttaskListTypes = []types.TaskListType{types.TaskListTypeActivity, types.TaskListTypeDecision}\n\t} else {\n\t\treturn nil, commoncli.Problem(\"Invalid task list type: valid types are 'activity', 'decision', or empty (both)\", nil)\n\t}\n\treturn taskListTypes, nil\n}\n"
  },
  {
    "path": "tools/cli/admin_task_list_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestAdminDescribeTaskList(t *testing.T) {\n\tresponse := &types.DescribeTaskListResponse{\n\t\tPollers: []*types.PollerInfo{\n\t\t\t{\n\t\t\t\tIdentity: \"test-poller\",\n\t\t\t},\n\t\t},\n\t\tTaskListStatus: &types.TaskListStatus{\n\t\t\tBacklogCountHint: 10,\n\t\t},\n\t}\n\ttaskList := &types.TaskList{\n\t\tName: testTaskList,\n\t}\n\n\ttests := []struct {\n\t\tname        string\n\t\tallowance   func(td *cliTestData)\n\t\ttlType      string\n\t\tformat      string\n\t\tbaseArgs    []clitest.CliArgument\n\t\tassertions  func(t *testing.T, td *cliTestData)\n\t\texpectedErr string\n\t}{\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tallowance: func(td *cliTestData) {\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(response, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(response, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:   \"success - decision only\",\n\t\t\ttlType: \"decision\",\n\t\t\tallowance: func(td *cliTestData) {\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(response, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:   \"success - activity only\",\n\t\t\ttlType: \"activity\",\n\t\t\tallowance: func(td *cliTestData) {\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(response, nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:   \"json\",\n\t\t\ttlType: \"decision\",\n\t\t\tformat: \"json\",\n\t\t\tallowance: func(td *cliTestData) {\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(response, nil).Times(1)\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, td *cliTestData) {\n\t\t\t\texpected := map[types.TaskListType]*types.DescribeTaskListResponse{\n\t\t\t\t\ttypes.TaskListTypeDecision: response,\n\t\t\t\t}\n\t\t\t\toutput := td.consoleOutput()\n\t\t\t\tvar result map[types.TaskListType]*types.DescribeTaskListResponse\n\t\t\t\terr := json.Unmarshal([]byte(output), &result)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, expected, result)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"error - missing domain\",\n\t\t\texpectedErr: \"Required flag not found\",\n\t\t\tbaseArgs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagTaskList, testTaskList),\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:        \"error - missing tasklist\",\n\t\t\texpectedErr: \"Required flag not found\",\n\t\t\tbaseArgs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t},\n\t\t},\n\n\t\t{\n\t\t\tname:   \"error - from server\",\n\t\t\ttlType: \"decision\",\n\t\t\tallowance: func(td *cliTestData) {\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                testDomain,\n\t\t\t\t\tTaskList:              taskList,\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(nil, errors.New(\"oh no\")).Times(1)\n\t\t\t},\n\t\t\texpectedErr: \"oh no\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\n\t\t\tvar args []clitest.CliArgument\n\t\t\tif tc.baseArgs == nil {\n\t\t\t\targs = []clitest.CliArgument{\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagTaskList, testTaskList),\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\targs = tc.baseArgs\n\t\t\t}\n\t\t\tif tc.tlType != \"\" {\n\t\t\t\targs = append(args, clitest.StringArgument(FlagTaskListType, tc.tlType))\n\t\t\t}\n\t\t\tif tc.format != \"\" {\n\t\t\t\targs = append(args, clitest.StringArgument(FlagFormat, tc.format))\n\t\t\t}\n\n\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\tt,\n\t\t\t\ttd.app,\n\t\t\t\targs...,\n\t\t\t)\n\n\t\t\tif tc.allowance != nil {\n\t\t\t\ttc.allowance(td)\n\t\t\t}\n\n\t\t\terr := AdminDescribeTaskList(cliCtx)\n\n\t\t\tif tc.expectedErr == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedErr)\n\t\t\t}\n\n\t\t\tif tc.assertions != nil {\n\t\t\t\ttc.assertions(t, td)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminListTaskList(t *testing.T) {\n\texpectedResponse := &types.GetTaskListsByDomainResponse{\n\t\tDecisionTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\"decision-tasklist-1\": {\n\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t{Identity: \"poller1\"},\n\t\t\t\t\t{Identity: \"poller2\"},\n\t\t\t\t},\n\t\t\t},\n\t\t\t\"decision-tasklist-2\": {\n\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t{Identity: \"poller3\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tActivityTaskListMap: map[string]*types.DescribeTaskListResponse{\n\t\t\t\"activity-tasklist-1\": {\n\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t{Identity: \"poller4\"},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\t// Define table of test cases\n\ttests := []struct {\n\t\tname          string\n\t\tsetupMocks    func(*frontend.MockClient)\n\t\tassertions    func(t *testing.T, td *cliTestData)\n\t\texpectedError string\n\t\tdomainFlag    string\n\t\tformat        string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetTaskListsByDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\tdomainFlag: \"test-domain\",\n\t\t},\n\t\t{\n\t\t\tname:   \"Success - json\",\n\t\t\tformat: \"json\",\n\t\t\tsetupMocks: func(client *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetTaskListsByDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\tassertions: func(t *testing.T, td *cliTestData) {\n\t\t\t\toutput := td.consoleOutput()\n\t\t\t\tvar result *types.GetTaskListsByDomainResponse\n\t\t\t\terr := json.Unmarshal([]byte(output), &result)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.Equal(t, expectedResponse, result)\n\t\t\t},\n\t\t\tdomainFlag: \"test-domain\",\n\t\t},\n\t\t{\n\t\t\tname: \"GetTaskListsByDomainFails\",\n\t\t\tsetupMocks: func(client *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetTaskListsByDomain(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"GetTaskListsByDomain failed\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"Operation GetTaskListByDomain failed\",\n\t\t\tdomainFlag:    \"test-domain\",\n\t\t},\n\t\t{\n\t\t\tname:          \"NoDomainFlag\",\n\t\t\tsetupMocks:    func(client *frontend.MockClient) {},\n\t\t\texpectedError: \"Required flag not found\",\n\t\t\tdomainFlag:    \"\", // Omit Domain flag\n\t\t},\n\t}\n\n\t// Loop through test cases\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(td.mockFrontendClient)\n\n\t\t\tvar cliCtx *cli.Context\n\t\t\tif tt.domainFlag == \"\" {\n\t\t\t\tcliCtx = clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\t/* omit the domain flag */\n\t\t\t\t\tclitest.StringArgument(FlagFormat, tt.format),\n\t\t\t\t)\n\t\t\t} else {\n\t\t\t\tcliCtx = clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagFormat, tt.format),\n\t\t\t\t)\n\t\t\t}\n\n\t\t\terr := AdminListTaskList(cliCtx)\n\t\t\tif tt.expectedError == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t}\n\n\t\t\tif tt.assertions != nil {\n\t\t\t\ttt.assertions(t, td)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateTaskListPartitionConfig(t *testing.T) {\n\t// Define table of test cases\n\ttests := []struct {\n\t\tname               string\n\t\tsetupMocks         func(*admin.MockClient, *frontend.MockClient)\n\t\texpectedError      string\n\t\tdomainFlag         string\n\t\ttaskListFlag       string\n\t\ttaskListType       string\n\t\tnumReadPartitions  int\n\t\tnumWritePartitions int\n\t\tforce              bool\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:      \"\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - both types\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:      \"\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Success - force\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:      \"\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t\tforce:              true,\n\t\t},\n\t\t{\n\t\t\tname: \"UpdateTaskListPartitionConfigFails\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"API failed\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:      \"API failed\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname:               \"NoDomainFlag\",\n\t\t\texpectedError:      \"Required flag not found\",\n\t\t\tdomainFlag:         \"\", // Omit Domain flag\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname:               \"NoTaskListFlag\",\n\t\t\texpectedError:      \"Required flag not found\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname:               \"Invalid task list type\",\n\t\t\texpectedError:      \"Invalid task list type: valid types are 'activity', 'decision', or empty (both)\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"ihsdajhi\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname:               \"NoReadPartitionFlag\",\n\t\t\texpectedError:      \"Required flag not found\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname:              \"NoWritePartitionFlag\",\n\t\t\texpectedError:     \"Required flag not found\",\n\t\t\tdomainFlag:        \"test-domain\",\n\t\t\ttaskListFlag:      \"test-tasklist\",\n\t\t\ttaskListType:      \"decision\",\n\t\t\tnumReadPartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"safe - removing drained partition\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\t// Check the root for pollers and config safety\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  createPartitions(3),\n\t\t\t\t\t\tWritePartitions: createPartitions(1),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\t// Check the partitions being drained\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                \"test-domain\",\n\t\t\t\t\tTaskList:              &types.TaskList{Name: getPartitionTaskListName(\"test-tasklist\", 2), Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: nil,\n\t\t\t\t\tTaskListStatus: &types.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: 0,\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                \"test-domain\",\n\t\t\t\t\tTaskList:              &types.TaskList{Name: getPartitionTaskListName(\"test-tasklist\", 1), Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: nil,\n\t\t\t\t\tTaskListStatus: &types.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: 0,\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\t// do the update\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(1),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(1),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:      \"\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  1,\n\t\t\tnumWritePartitions: 1,\n\t\t},\n\t\t{\n\t\t\tname: \"Safe - only one type missing pollers\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:         nil,\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateTaskListPartitionConfig(gomock.Any(), &types.UpdateTaskListPartitionConfigRequest{\n\t\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t\t},\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(&types.UpdateTaskListPartitionConfigResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Unsafe - no pollers\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:         nil,\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError:      \"no pollers\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Unsafe - both types no pollers\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:         nil,\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers:         nil,\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError:      \"no pollers\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Unsafe - removing active write partition\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  createPartitions(3),\n\t\t\t\t\t\tWritePartitions: createPartitions(3),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError:      \"remove write partitions, then read partitions\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Unsafe - removing non-drained partition\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  createPartitions(3),\n\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:                \"test-domain\",\n\t\t\t\t\tTaskList:              &types.TaskList{Name: getPartitionTaskListName(\"test-tasklist\", 2), Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType:          types.TaskListTypeDecision.Ptr(),\n\t\t\t\t\tIncludeTaskListStatus: true,\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: nil,\n\t\t\t\t\tTaskListStatus: &types.TaskListStatus{\n\t\t\t\t\t\tBacklogCountHint: 1,\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: nil,\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\texpectedError:      \"partition 2 still has 1 tasks remaining\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\ttaskListType:       \"decision\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t\t{\n\t\t\tname: \"Unsafe - one type fails validation\",\n\t\t\tsetupMocks: func(client *admin.MockClient, f *frontend.MockClient) {\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeDecision.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  createPartitions(3),\n\t\t\t\t\t\tWritePartitions: createPartitions(3),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tf.EXPECT().DescribeTaskList(gomock.Any(), &types.DescribeTaskListRequest{\n\t\t\t\t\tDomain:       \"test-domain\",\n\t\t\t\t\tTaskList:     &types.TaskList{Name: \"test-tasklist\", Kind: types.TaskListKindNormal.Ptr()},\n\t\t\t\t\tTaskListType: types.TaskListTypeActivity.Ptr(),\n\t\t\t\t}).Return(&types.DescribeTaskListResponse{\n\t\t\t\t\tPollers: []*types.PollerInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tIdentity: \"poller\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPartitionConfig: &types.TaskListPartitionConfig{\n\t\t\t\t\t\tVersion:         1,\n\t\t\t\t\t\tReadPartitions:  createPartitions(2),\n\t\t\t\t\t\tWritePartitions: createPartitions(2),\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError:      \"test-tasklist:Decision failed validation:\",\n\t\t\tdomainFlag:         \"test-domain\",\n\t\t\ttaskListFlag:       \"test-tasklist\",\n\t\t\tnumReadPartitions:  2,\n\t\t\tnumWritePartitions: 2,\n\t\t},\n\t}\n\n\t// Loop through test cases\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\n\t\t\tif tt.setupMocks != nil {\n\t\t\t\ttt.setupMocks(td.mockAdminClient, td.mockFrontendClient)\n\t\t\t}\n\n\t\t\tvar cliArgs []clitest.CliArgument\n\t\t\tif tt.domainFlag != \"\" {\n\t\t\t\tcliArgs = append(cliArgs, clitest.StringArgument(FlagDomain, tt.domainFlag))\n\t\t\t}\n\t\t\tif tt.taskListFlag != \"\" {\n\t\t\t\tcliArgs = append(cliArgs, clitest.StringArgument(FlagTaskList, tt.taskListFlag))\n\t\t\t}\n\t\t\tif tt.taskListType != \"\" {\n\t\t\t\tcliArgs = append(cliArgs, clitest.StringArgument(FlagTaskListType, tt.taskListType))\n\t\t\t}\n\t\t\tif tt.numReadPartitions != 0 {\n\t\t\t\tcliArgs = append(cliArgs, clitest.IntArgument(FlagNumReadPartitions, tt.numReadPartitions))\n\t\t\t}\n\t\t\tif tt.numWritePartitions != 0 {\n\t\t\t\tcliArgs = append(cliArgs, clitest.IntArgument(FlagNumWritePartitions, tt.numWritePartitions))\n\t\t\t}\n\t\t\tif tt.force {\n\t\t\t\tcliArgs = append(cliArgs, clitest.BoolArgument(FlagForce, true))\n\t\t\t}\n\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\tt,\n\t\t\t\ttd.app,\n\t\t\t\tcliArgs...,\n\t\t\t)\n\n\t\t\terr := AdminUpdateTaskListPartitionConfig(cliCtx)\n\t\t\tif tt.expectedError == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/admin_timers.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination admin_timers_mock.go github.com/uber/cadence/tools/cli LoadCloser,Printer\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/backoff\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// LoadCloser loads timer task information\ntype LoadCloser interface {\n\tLoad() ([]*persistence.TimerTaskInfo, error)\n\tClose()\n}\n\n// Printer prints timer task information\ntype Printer interface {\n\tPrint(output io.Writer, timers []*persistence.TimerTaskInfo) error\n}\n\n// Reporter wraps LoadCloser, Printer and a filter on time task type and domainID\ntype Reporter struct {\n\tdomainID   string\n\ttimerTypes []int\n\tloader     LoadCloser\n\tprinter    Printer\n}\n\ntype dbLoadCloser struct {\n\tctx              *cli.Context\n\texecutionManager persistence.ExecutionManager\n}\n\ntype fileLoadCloser struct {\n\tfile *os.File\n}\n\ntype histogramPrinter struct {\n\tctx        *cli.Context\n\ttimeFormat string\n}\n\ntype jsonPrinter struct {\n\tctx *cli.Context\n}\n\n// NewDBLoadCloser creates a new LoadCloser to load timer task information from database\nfunc NewDBLoadCloser(c *cli.Context) (LoadCloser, error) {\n\tshardID, err := getRequiredIntOption(c, FlagShardID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error in NewDBLoadCloser: failed to get shard ID: %w\", err)\n\t}\n\n\texecutionManager, err := getDeps(c).initializeExecutionManager(c, shardID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error in NewDBLoadCloser: failed to initialize execution store: %w\", err)\n\t}\n\treturn &dbLoadCloser{\n\t\tctx:              c,\n\t\texecutionManager: executionManager,\n\t}, nil\n}\n\nfunc NewFileLoadCloser(c *cli.Context) (LoadCloser, error) {\n\tfile, err := os.Open(c.String(FlagInputFile))\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error in NewFileLoadCloser: cannot open file: %w\", err)\n\t}\n\treturn &fileLoadCloser{\n\t\tfile: file,\n\t}, nil\n}\n\n// NewReporter creates a new Reporter\nfunc NewReporter(domain string, timerTypes []int, loader LoadCloser, printer Printer) *Reporter {\n\treturn &Reporter{\n\t\ttimerTypes: timerTypes,\n\t\tdomainID:   domain,\n\t\tloader:     loader,\n\t\tprinter:    printer,\n\t}\n}\n\n// NewHistogramPrinter creates a new Printer to display timer task information in a histogram\nfunc NewHistogramPrinter(c *cli.Context, timeFormat string) Printer {\n\treturn &histogramPrinter{\n\t\tctx:        c,\n\t\ttimeFormat: timeFormat,\n\t}\n}\n\n// NewJSONPrinter creates a new Printer to display timer task information in a JSON format\nfunc NewJSONPrinter(c *cli.Context) Printer {\n\treturn &jsonPrinter{\n\t\tctx: c,\n\t}\n}\n\nfunc (r *Reporter) filter(timers []*persistence.TimerTaskInfo) []*persistence.TimerTaskInfo {\n\ttaskTypes := intSliceToSet(r.timerTypes)\n\n\tfor i, t := range timers {\n\t\tif len(r.domainID) > 0 && t.DomainID != r.domainID {\n\t\t\ttimers[i] = nil\n\t\t\tcontinue\n\t\t}\n\t\tif _, ok := taskTypes[t.TaskType]; !ok {\n\t\t\ttimers[i] = nil\n\t\t\tcontinue\n\n\t\t}\n\t}\n\n\treturn timers\n}\n\n// Report loads, filters and prints timer tasks\nfunc (r *Reporter) Report(output io.Writer) error {\n\tloader, err := r.loader.Load()\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn r.printer.Print(output, r.filter(loader))\n}\n\n// AdminTimers is used to list scheduled timers.\nfunc AdminTimers(c *cli.Context) error {\n\ttimerTypes := c.IntSlice(FlagTimerType)\n\tif !c.IsSet(FlagTimerType) || (len(timerTypes) == 1 && timerTypes[0] == -1) {\n\t\ttimerTypes = []int{\n\t\t\tpersistence.TaskTypeDecisionTimeout,\n\t\t\tpersistence.TaskTypeActivityTimeout,\n\t\t\tpersistence.TaskTypeUserTimer,\n\t\t\tpersistence.TaskTypeWorkflowTimeout,\n\t\t\tpersistence.TaskTypeDeleteHistoryEvent,\n\t\t\tpersistence.TaskTypeActivityRetryTimer,\n\t\t\tpersistence.TaskTypeWorkflowBackoffTimer,\n\t\t}\n\t}\n\n\t// setup loader\n\tvar loader LoadCloser\n\tvar err error\n\tif !c.IsSet(FlagInputFile) {\n\t\tloader, err = NewDBLoadCloser(c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in timer: \", err)\n\t\t}\n\t} else {\n\t\tloader, err = NewFileLoadCloser(c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in timer: \", err)\n\t\t}\n\t}\n\tdefer loader.Close()\n\n\t// setup printer\n\tvar printer Printer\n\tif !c.Bool(FlagPrintJSON) {\n\t\tvar timerFormat string\n\t\tif c.IsSet(FlagDateFormat) {\n\t\t\ttimerFormat = c.String(FlagDateFormat)\n\t\t} else {\n\t\t\tswitch c.String(FlagBucketSize) {\n\t\t\tcase \"day\":\n\t\t\t\ttimerFormat = \"2006-01-02\"\n\t\t\tcase \"hour\":\n\t\t\t\ttimerFormat = \"2006-01-02T15\"\n\t\t\tcase \"minute\":\n\t\t\t\ttimerFormat = \"2006-01-02T15:04\"\n\t\t\tcase \"second\":\n\t\t\t\ttimerFormat = \"2006-01-02T15:04:05\"\n\t\t\tdefault:\n\t\t\t\treturn commoncli.Problem(\"unknown bucket size: \"+c.String(FlagBucketSize), nil)\n\t\t\t}\n\t\t}\n\t\tprinter = NewHistogramPrinter(c, timerFormat)\n\t} else {\n\t\tprinter = NewJSONPrinter(c)\n\t}\n\n\treporter := NewReporter(c.String(FlagDomainID), timerTypes, loader, printer)\n\toutput := getDeps(c).Output()\n\tif err := reporter.Report(output); err != nil {\n\t\treturn commoncli.Problem(\"Reporter failed\", err)\n\t}\n\treturn nil\n}\n\nfunc (jp *jsonPrinter) Print(output io.Writer, timers []*persistence.TimerTaskInfo) error {\n\tfor _, t := range timers {\n\t\tif t == nil {\n\t\t\tcontinue\n\t\t}\n\t\tdata, err := json.Marshal(t)\n\t\tif err != nil {\n\t\t\tif !jp.ctx.Bool(FlagSkipErrorMode) {\n\t\t\t\treturn commoncli.Problem(\"cannot marshal timer to json\", err)\n\t\t\t}\n\t\t\toutput.Write([]byte(fmt.Sprintf(\"%s\\n\", err.Error())))\n\t\t} else {\n\t\t\toutput.Write([]byte(fmt.Sprintf(\"%s\\n\", data)))\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (cl *dbLoadCloser) Load() ([]*persistence.TimerTaskInfo, error) {\n\tbatchSize := cl.ctx.Int(FlagBatchSize)\n\tstartDate := cl.ctx.String(FlagStartDate)\n\tendDate := cl.ctx.String(FlagEndDate)\n\n\tst, err := parseSingleTs(startDate)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"wrong date format for \"+FlagEndDate+\" Error: %v\", err)\n\t}\n\tet, err := parseSingleTs(endDate)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"wrong date format for \"+FlagEndDate+\" Error: %v\", err)\n\t}\n\n\tvar timers []*persistence.TimerTaskInfo\n\n\tisRetryable := func(err error) bool {\n\t\treturn persistence.IsTransientError(err) || common.IsContextTimeoutError(err)\n\t}\n\n\tthrottleRetry := backoff.NewThrottleRetry(\n\t\tbackoff.WithRetryPolicy(common.CreatePersistenceRetryPolicy()),\n\t\tbackoff.WithRetryableError(isRetryable),\n\t)\n\tvar token []byte\n\tisFirstIteration := true\n\tfor isFirstIteration || len(token) != 0 {\n\t\tisFirstIteration = false\n\t\treq := persistence.GetHistoryTasksRequest{\n\t\t\tTaskCategory:        persistence.HistoryTaskCategoryTimer,\n\t\t\tInclusiveMinTaskKey: persistence.NewHistoryTaskKey(st, 0),\n\t\t\tExclusiveMaxTaskKey: persistence.NewHistoryTaskKey(et, 0),\n\t\t\tPageSize:            batchSize,\n\t\t\tNextPageToken:       token,\n\t\t}\n\n\t\tresp := &persistence.GetHistoryTasksResponse{}\n\n\t\top := func(ctx context.Context) error {\n\t\t\tctx, cancel, err := newContext(cl.ctx)\n\t\t\tdefer cancel()\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t\t\t}\n\t\t\tresp, err = cl.executionManager.GetHistoryTasks(ctx, &req)\n\t\t\treturn err\n\t\t}\n\n\t\terr = throttleRetry.Do(cl.ctx.Context, op)\n\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"cannot get timer tasks for shard: %v\", err)\n\t\t}\n\n\t\ttoken = resp.NextPageToken\n\t\tfor _, task := range resp.Tasks {\n\t\t\ttimer, err := task.ToTimerTaskInfo()\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"error converting task to timer task info: %w\", err)\n\t\t\t}\n\t\t\ttimers = append(timers, timer)\n\t\t}\n\t}\n\treturn timers, nil\n}\n\nfunc (cl *dbLoadCloser) Close() {\n\tif cl.executionManager != nil {\n\t\tcl.executionManager.Close()\n\t}\n}\n\nfunc (fl *fileLoadCloser) Load() ([]*persistence.TimerTaskInfo, error) {\n\tvar data []*persistence.TimerTaskInfo\n\tdec := json.NewDecoder(fl.file)\n\n\tfor {\n\t\tvar timer persistence.TimerTaskInfo\n\t\tif err := dec.Decode(&timer); err != nil {\n\t\t\tif err == io.EOF || err == io.ErrUnexpectedEOF {\n\t\t\t\tbreak\n\t\t\t}\n\t\t\treturn nil, fmt.Errorf(\"error decoding timer: %w\", err)\n\t\t}\n\t\tdata = append(data, &timer)\n\t}\n\treturn data, nil\n}\n\nfunc (fl *fileLoadCloser) Close() {\n\tif fl.file != nil {\n\t\tfl.file.Close()\n\t}\n}\n\nfunc (hp *histogramPrinter) Print(output io.Writer, timers []*persistence.TimerTaskInfo) error {\n\th := NewHistogram()\n\tfor _, t := range timers {\n\t\tif t == nil {\n\t\t\tcontinue\n\t\t}\n\t\th.Add(t.VisibilityTimestamp.Format(hp.timeFormat))\n\t}\n\n\treturn h.Print(output, hp.ctx.Int(FlagShardMultiplier))\n}\n"
  },
  {
    "path": "tools/cli/admin_timers_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/tools/cli (interfaces: LoadCloser,Printer)\n//\n// Generated by this command:\n//\n//\tmockgen -package cli -destination admin_timers_mock.go github.com/uber/cadence/tools/cli LoadCloser,Printer\n//\n\n// Package cli is a generated GoMock package.\npackage cli\n\nimport (\n\tio \"io\"\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n)\n\n// MockLoadCloser is a mock of LoadCloser interface.\ntype MockLoadCloser struct {\n\tctrl     *gomock.Controller\n\trecorder *MockLoadCloserMockRecorder\n\tisgomock struct{}\n}\n\n// MockLoadCloserMockRecorder is the mock recorder for MockLoadCloser.\ntype MockLoadCloserMockRecorder struct {\n\tmock *MockLoadCloser\n}\n\n// NewMockLoadCloser creates a new mock instance.\nfunc NewMockLoadCloser(ctrl *gomock.Controller) *MockLoadCloser {\n\tmock := &MockLoadCloser{ctrl: ctrl}\n\tmock.recorder = &MockLoadCloserMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockLoadCloser) EXPECT() *MockLoadCloserMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockLoadCloser) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockLoadCloserMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockLoadCloser)(nil).Close))\n}\n\n// Load mocks base method.\nfunc (m *MockLoadCloser) Load() ([]*persistence.TimerTaskInfo, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Load\")\n\tret0, _ := ret[0].([]*persistence.TimerTaskInfo)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// Load indicates an expected call of Load.\nfunc (mr *MockLoadCloserMockRecorder) Load() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Load\", reflect.TypeOf((*MockLoadCloser)(nil).Load))\n}\n\n// MockPrinter is a mock of Printer interface.\ntype MockPrinter struct {\n\tctrl     *gomock.Controller\n\trecorder *MockPrinterMockRecorder\n\tisgomock struct{}\n}\n\n// MockPrinterMockRecorder is the mock recorder for MockPrinter.\ntype MockPrinterMockRecorder struct {\n\tmock *MockPrinter\n}\n\n// NewMockPrinter creates a new mock instance.\nfunc NewMockPrinter(ctrl *gomock.Controller) *MockPrinter {\n\tmock := &MockPrinter{ctrl: ctrl}\n\tmock.recorder = &MockPrinterMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockPrinter) EXPECT() *MockPrinterMockRecorder {\n\treturn m.recorder\n}\n\n// Print mocks base method.\nfunc (m *MockPrinter) Print(output io.Writer, timers []*persistence.TimerTaskInfo) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"Print\", output, timers)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// Print indicates an expected call of Print.\nfunc (mr *MockPrinterMockRecorder) Print(output, timers any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Print\", reflect.TypeOf((*MockPrinter)(nil).Print), output, timers)\n}\n"
  },
  {
    "path": "tools/cli/admin_timers_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common/persistence\"\n)\n\nfunc TestNewDBLoadCloser(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcontextSetup  func(td *cliTestData) *cli.Context\n\t\tmockSetup     func(td *cliTestData)\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Int(FlagShardID, 1, \"shard ID flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagShardID, \"1\"))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(gomock.NewController(t))\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), 1).Return(mockExecManager, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"Missing ShardID Error\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup:     func(td *cliTestData) {},\n\t\t\texpectedError: \"error in NewDBLoadCloser: failed to get shard ID\",\n\t\t},\n\t\t{\n\t\t\tname: \"Execution Manager Initialization Error\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Int(FlagShardID, 1, \"shard ID flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagShardID, \"1\"))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\ttd.mockManagerFactory.EXPECT().initializeExecutionManager(gomock.Any(), 1).Return(nil, fmt.Errorf(\"failed to initialize execution store\")).Times(1)\n\t\t\t},\n\t\t\texpectedError: \"error in NewDBLoadCloser: failed to initialize execution store\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tc := tt.contextSetup(td)\n\t\t\ttt.mockSetup(td)\n\n\t\t\t_, err := NewDBLoadCloser(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNewFileLoadCloser(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tcontextSetup  func(td *cliTestData) (*cli.Context, string) // Return context and temp file path\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\t// Setup context with a temporary file to simulate successful file opening\n\t\t\tcontextSetup: func(td *cliTestData) (*cli.Context, string) {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagInputFile, \"\", \"Input file flag\")\n\n\t\t\t\t// Create a temporary file for testing\n\t\t\t\ttempFile, err := os.CreateTemp(\"\", \"testfile_*.txt\")\n\t\t\t\trequire.NoError(t, err)\n\n\t\t\t\t// Close the file but do not remove it yet, so it remains available during the test\n\t\t\t\trequire.NoError(t, tempFile.Close())\n\n\t\t\t\t// Set FlagInputFile to the path of the temporary file\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, tempFile.Name()))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil), tempFile.Name() // Return context and file path\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"File Open Error\",\n\t\t\t// Setup context with a non-existent file path to simulate file open failure\n\t\t\tcontextSetup: func(td *cliTestData) (*cli.Context, string) {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagInputFile, \"non_existent_file.txt\", \"input file flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, \"non_existent_file.txt\"))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil), \"\" // No file to delete after test\n\t\t\t},\n\t\t\texpectedError: \"error in NewFileLoadCloser: cannot open file\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Initialize test data\n\t\t\ttd := newCLITestData(t)\n\t\t\t// Setup context and get temp file path\n\t\t\tc, tempFilePath := tt.contextSetup(td)\n\n\t\t\t// Defer file deletion after the test completes\n\t\t\tif tempFilePath != \"\" {\n\t\t\t\tdefer os.Remove(tempFilePath) // Clean up the temporary file\n\t\t\t}\n\n\t\t\t// Call NewFileLoadCloser and capture result\n\t\t\tcloser, err := NewFileLoadCloser(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\t// Verify error message when expected\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t\tassert.Nil(t, closer)\n\t\t\t} else {\n\t\t\t\t// Verify no error and a non-nil LoadCloser instance on success\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\tassert.NotNil(t, closer)\n\t\t\t\tcloser.Close() // Close the file to prevent resource leakage\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestNewReporter(t *testing.T) {\n\t// Mock dependencies\n\tmockLoader := &MockLoadCloser{} // Substitute with actual mock creation\n\tmockPrinter := &MockPrinter{}   // Substitute with actual mock creation\n\n\t// Define test inputs\n\tdomain := \"testDomain\"\n\ttimerTypes := []int{1, 2, 3}\n\n\t// Call the function under test\n\treporter := NewReporter(domain, timerTypes, mockLoader, mockPrinter)\n\n\t// Validate the output\n\trequire.NotNil(t, reporter)\n\tassert.Equal(t, domain, reporter.domainID)\n\tassert.Equal(t, timerTypes, reporter.timerTypes)\n\tassert.Equal(t, mockLoader, reporter.loader)\n\tassert.Equal(t, mockPrinter, reporter.printer)\n}\n\nfunc TestNewHistogramPrinter(t *testing.T) {\n\t// Mock CLI context\n\ttd := newCLITestData(t) // Setup test data for CLI context\n\tc := cli.NewContext(td.app, flag.NewFlagSet(\"test\", 0), nil)\n\n\t// Define test inputs\n\ttimeFormat := \"2006-01-02 15:04:05\"\n\n\t// Call the function under test\n\tprinter := NewHistogramPrinter(c, timeFormat)\n\n\t// Validate the output\n\trequire.NotNil(t, printer)\n\thistPrinter, ok := printer.(*histogramPrinter)\n\trequire.True(t, ok, \"Expected Printer to be of type *histogramPrinter\")\n\tassert.Equal(t, c, histPrinter.ctx)\n\tassert.Equal(t, timeFormat, histPrinter.timeFormat)\n}\n\nfunc TestNewJSONPrinter(t *testing.T) {\n\t// Mock CLI context\n\ttd := newCLITestData(t) // Setup test data for CLI context\n\tc := cli.NewContext(td.app, flag.NewFlagSet(\"test\", 0), nil)\n\n\t// Call the function under test\n\tprinter := NewJSONPrinter(c)\n\n\t// Validate the output\n\trequire.NotNil(t, printer)\n\tjsonPrinter, ok := printer.(*jsonPrinter)\n\trequire.True(t, ok, \"Expected Printer to be of type *jsonPrinter\")\n\tassert.Equal(t, c, jsonPrinter.ctx)\n}\n\nfunc TestReporter_Report(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tdomainID       string\n\t\ttimerTypes     []int\n\t\tloaderOutput   []*persistence.TimerTaskInfo\n\t\tloaderError    error\n\t\texpectedOutput []*persistence.TimerTaskInfo\n\t\texpectedError  string\n\t\tmockSetup      func(mockLoader *MockLoadCloser, mockPrinter *MockPrinter)\n\t}{\n\t\t{\n\t\t\tname:       \"Successful Report with Filtering\",\n\t\t\tdomainID:   \"testDomain\",\n\t\t\ttimerTypes: []int{1, 2},\n\t\t\tloaderOutput: []*persistence.TimerTaskInfo{\n\t\t\t\t{TaskType: 1, DomainID: \"testDomain\"},\n\t\t\t\t{TaskType: 3, DomainID: \"testDomain\"},\n\t\t\t\t{TaskType: 2, DomainID: \"otherDomain\"},\n\t\t\t},\n\t\t\texpectedOutput: []*persistence.TimerTaskInfo{\n\t\t\t\t{TaskType: 1, DomainID: \"testDomain\"},\n\t\t\t\tnil, // TaskType 3 filtered out\n\t\t\t\tnil, // DomainID otherDomain filtered out\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tmockSetup: func(mockLoader *MockLoadCloser, mockPrinter *MockPrinter) {\n\t\t\t\tmockLoader.EXPECT().Load().Return([]*persistence.TimerTaskInfo{\n\t\t\t\t\t{TaskType: 1, DomainID: \"testDomain\"},\n\t\t\t\t\t{TaskType: 3, DomainID: \"testDomain\"},\n\t\t\t\t\t{TaskType: 2, DomainID: \"otherDomain\"},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\tmockPrinter.EXPECT().Print(gomock.Any(), gomock.Any()).Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:           \"Loader Error\",\n\t\t\tdomainID:       \"testDomain\",\n\t\t\ttimerTypes:     []int{1},\n\t\t\tloaderOutput:   nil,\n\t\t\tloaderError:    fmt.Errorf(\"failed to load tasks\"),\n\t\t\texpectedOutput: nil,\n\t\t\texpectedError:  \"failed to load tasks\",\n\t\t\tmockSetup: func(mockLoader *MockLoadCloser, mockPrinter *MockPrinter) {\n\t\t\t\tmockLoader.EXPECT().Load().Return(nil, fmt.Errorf(\"failed to load tasks\")).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\t// Initialize mocks\n\t\t\tmockLoader := NewMockLoadCloser(ctrl)\n\t\t\tmockPrinter := NewMockPrinter(ctrl)\n\n\t\t\t// Set up mock expectations\n\t\t\ttt.mockSetup(mockLoader, mockPrinter)\n\n\t\t\t// Create reporter instance\n\t\t\treporter := NewReporter(tt.domainID, tt.timerTypes, mockLoader, mockPrinter)\n\n\t\t\t// Call Report and capture error\n\t\t\terr := reporter.Report(nil)\n\n\t\t\t// Check results based on expected error\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminTimers(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tcontextSetup   func(td *cliTestData) *cli.Context\n\t\tmockSetup      func(td *cliTestData)\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Default Timer Types with DB Loader and Histogram Printer\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Var(cli.NewIntSlice(-1), FlagTimerType, \"timer type flag\") // Default timer types\n\t\t\t\tset.Bool(FlagPrintJSON, false, \"print JSON flag\")              // Use histogram printer\n\t\t\t\tset.String(FlagBucketSize, \"\", \"bucket size flag\")\n\t\t\t\tset.Int(FlagShardID, 1, \"shard ID flag\")                 // Set shard ID for DB loader\n\t\t\t\tset.Int(FlagShardMultiplier, 1, \"shard multiplier flag\") // Set a non-zero multiplier\n\t\t\t\tset.String(FlagEndDate, \"\", \"end date flag\")             // Explicitly set end date format\n\t\t\t\tset.String(FlagStartDate, \"\", \"start date flag\")         // Explicitly set start date format\n\n\t\t\t\trequire.NoError(t, set.Set(FlagTimerType, \"-1\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagPrintJSON, \"false\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagBucketSize, \"hour\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagShardID, \"1\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagShardMultiplier, \"1\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagEndDate, \"2023-01-01T00:00:00Z\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagStartDate, \"2022-01-01T00:00:00Z\"))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(ctrl)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeExecutionManager(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecManager, nil).Times(1)\n\n\t\t\t\tmockExecManager.EXPECT().\n\t\t\t\t\tGetHistoryTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&persistence.GetHistoryTasksResponse{\n\t\t\t\t\t\tTasks: []persistence.Task{\n\t\t\t\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\t\t\tDomainID:   \"testDomain\",\n\t\t\t\t\t\t\t\t\tWorkflowID: \"testWorkflow1\",\n\t\t\t\t\t\t\t\t\tRunID:      \"testRun1\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\t\t\tTaskID:              1,\n\t\t\t\t\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t&persistence.UserTimerTask{\n\t\t\t\t\t\t\t\tWorkflowIdentifier: persistence.WorkflowIdentifier{\n\t\t\t\t\t\t\t\t\tDomainID:   \"testDomain\",\n\t\t\t\t\t\t\t\t\tWorkflowID: \"testWorkflow2\",\n\t\t\t\t\t\t\t\t\tRunID:      \"testRun2\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\tTaskData: persistence.TaskData{\n\t\t\t\t\t\t\t\t\tTaskID:              2,\n\t\t\t\t\t\t\t\t\tVisibilityTimestamp: time.Now().Add(time.Hour),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\tNextPageToken: nil,\n\t\t\t\t\t}, nil).Times(1)\n\n\t\t\t\tmockExecManager.EXPECT().Close().Times(1)\n\t\t\t},\n\t\t\texpectedError:  \"\",\n\t\t\texpectedOutput: \"Bucket        Count\\n\", // Customize this with expected histogram output\n\t\t},\n\t\t{\n\t\t\tname: \"File Loader with JSON Printer\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Var(cli.NewIntSlice(persistence.TaskTypeUserTimer), FlagTimerType, \"timer type flag\")\n\t\t\t\tset.Bool(FlagPrintJSON, true, \"print JSON flag\") // Use JSON printer\n\t\t\t\tset.String(FlagInputFile, \"\", \"input file flag\") // Set input file for file loader\n\n\t\t\t\t// Create a temporary file and write test timer data to it\n\t\t\t\ttempFile, err := os.CreateTemp(\"\", \"testfile_*.txt\")\n\t\t\t\trequire.NoError(t, err)\n\n\t\t\t\t// Write a sample TimerTaskInfo JSON object to the file\n\t\t\t\ttimerTask := persistence.TimerTaskInfo{\n\t\t\t\t\tDomainID:            \"testDomain\",\n\t\t\t\t\tWorkflowID:          \"testWorkflow1\",\n\t\t\t\t\tRunID:               \"testRun1\",\n\t\t\t\t\tVisibilityTimestamp: time.Now(),\n\t\t\t\t\tTaskID:              1,\n\t\t\t\t\tTaskType:            persistence.TaskTypeUserTimer,\n\t\t\t\t}\n\t\t\t\ttimerData, err := json.Marshal(timerTask)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\t_, err = tempFile.Write(timerData)\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\ttempFile.Close() // Close after writing to ensure data is saved\n\n\t\t\t\t// Set flag values\n\t\t\t\trequire.NoError(t, set.Set(FlagTimerType, fmt.Sprint(persistence.TaskTypeUserTimer)))\n\t\t\t\trequire.NoError(t, set.Set(FlagPrintJSON, \"true\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagInputFile, tempFile.Name()))\n\n\t\t\t\t// Ensure file is removed after the test\n\t\t\t\tt.Cleanup(func() {\n\t\t\t\t\tos.Remove(tempFile.Name())\n\t\t\t\t})\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup:      func(td *cliTestData) {},\n\t\t\texpectedError:  \"\",\n\t\t\texpectedOutput: `{\"DomainID\":\"testDomain\",\"WorkflowID\":\"testWorkflow1\",\"RunID\":\"testRun1\"`, // Partial match to validate JSON structure\n\t\t},\n\t\t{\n\t\t\tname: \"Error with Unknown Bucket Size\",\n\t\t\tcontextSetup: func(td *cliTestData) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagBucketSize, \"\", \"bucket size flag\")\n\t\t\t\tset.Int(FlagShardID, 1, \"shard ID flag\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagBucketSize, \"unknown\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagShardID, \"1\"))\n\n\t\t\t\treturn cli.NewContext(td.app, set, nil)\n\t\t\t},\n\t\t\tmockSetup: func(td *cliTestData) {\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockExecManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockExecManager.EXPECT().Close().Times(1)\n\n\t\t\t\ttd.mockManagerFactory.EXPECT().\n\t\t\t\t\tinitializeExecutionManager(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(mockExecManager, nil).Times(1)\n\t\t\t},\n\t\t\texpectedError: \"unknown bucket size\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newCLITestData(t)\n\t\t\tc := tt.contextSetup(td)\n\n\t\t\ttt.mockSetup(td)\n\n\t\t\terr := AdminTimers(c)\n\n\t\t\t// Validate expected error\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\trequire.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\trequire.NoError(t, err)\n\t\t\t\t// Validate the output using testIOHandler's outputBytes\n\t\t\t\toutput := td.ioHandler.outputBytes.String()\n\t\t\t\tassert.Contains(t, output, tt.expectedOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/app.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"io\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/metrics\"\n)\n\nconst depsKey = \"deps\"\n\ntype CLIAppOptions func(*cli.App)\n\n// WithIOHandler sets the IOHandler for the CLI app. By default the app uses urfave's default Reader/Writer/ErrorWriter.\nfunc WithIOHandler(h IOHandler) CLIAppOptions {\n\treturn func(app *cli.App) {\n\t\tif app.Metadata == nil {\n\t\t\treturn\n\t\t}\n\n\t\td, ok := app.Metadata[depsKey].(*deps)\n\t\tif !ok {\n\t\t\treturn\n\t\t}\n\n\t\td.IOHandler = h\n\t}\n}\n\n// WithManagerFactory sets the ManagerFactory for the CLI app.\nfunc WithManagerFactory(factory ManagerFactory) CLIAppOptions {\n\treturn func(app *cli.App) {\n\t\tif app.Metadata == nil {\n\t\t\treturn\n\t\t}\n\n\t\td, ok := app.Metadata[depsKey].(*deps)\n\t\tif !ok {\n\t\t\treturn\n\t\t}\n\n\t\td.ManagerFactory = factory\n\t}\n}\n\n// NewCliApp instantiates a new instance of the CLI application\nfunc NewCliApp(cf ClientFactory, opts ...CLIAppOptions) *cli.App {\n\tversion := fmt.Sprintf(\"CLI feature version: %v \\n\"+\n\t\t\"   Release version: %v\\n\"+\n\t\t\"   Build commit: %v\\n\"+\n\t\t\"   Note: CLI feature version is for compatibility checking between server and CLI if enabled feature checking. Server is always backward compatible to older CLI versions, but not accepting newer than it can support.\",\n\t\tclient.SupportedCLIVersion, metrics.ReleaseVersion, metrics.Revision)\n\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence\"\n\tapp.Usage = \"A command-line tool for cadence users\"\n\tapp.Version = version\n\tapp.Metadata = map[string]any{\n\t\tdepsKey: &deps{ClientFactory: cf, IOHandler: &defaultIOHandler{app: app}, ManagerFactory: &defaultManagerFactory{}},\n\t}\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagAddress,\n\t\t\tAliases: []string{\"ad\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"host:port for cadence frontend service\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_ADDRESS\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagDomain,\n\t\t\tAliases: []string{\"do\"},\n\t\t\tUsage:   \"cadence workflow domain\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_DOMAIN\"},\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagContextTimeout,\n\t\t\tAliases: []string{\"ct\"},\n\t\t\tValue:   defaultContextTimeoutInSeconds,\n\t\t\tUsage:   \"optional timeout for context of RPC call in seconds\",\n\t\t\tEnvVars: []string{\"CADENCE_CONTEXT_TIMEOUT\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagJWT,\n\t\t\tUsage:   \"optional JWT for authorization. Either this or --jwt-private-key is needed for jwt authorization\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_JWT\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagJWTPrivateKey,\n\t\t\tAliases: []string{\"jwt-pk\"},\n\t\t\tUsage:   \"optional private key path to create JWT. Either this or --jwt is needed for jwt authorization. --jwt flag has priority over this one if both provided\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_JWT_PRIVATE_KEY\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagTransport,\n\t\t\tAliases: []string{\"t\"},\n\t\t\tUsage:   \"optional argument for transport protocol format, either 'grpc' or 'tchannel'. Defaults to tchannel if not provided\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_TRANSPORT_PROTOCOL\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagTLSCertPath,\n\t\t\tAliases: []string{\"tcp\"},\n\t\t\tUsage:   \"optional argument for path to TLS certificate. Defaults to an empty string if not provided\",\n\t\t\tEnvVars: []string{\"CADENCE_CLI_TLS_CERT_PATH\"},\n\t\t},\n\t}\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:        \"domain\",\n\t\t\tAliases:     []string{\"d\"},\n\t\t\tUsage:       \"Operate cadence domain\",\n\t\t\tSubcommands: newDomainCommands(),\n\t\t},\n\t\t{\n\t\t\tName:        \"workflow\",\n\t\t\tAliases:     []string{\"wf\"},\n\t\t\tUsage:       \"Operate cadence workflow\",\n\t\t\tSubcommands: newWorkflowCommands(),\n\t\t},\n\t\t{\n\t\t\tName:        \"tasklist\",\n\t\t\tAliases:     []string{\"tl\"},\n\t\t\tUsage:       \"Operate cadence tasklist\",\n\t\t\tSubcommands: newTaskListCommands(),\n\t\t},\n\t\t{\n\t\t\tName:    \"admin\",\n\t\t\tAliases: []string{\"adm\"},\n\t\t\tUsage:   \"Run admin operation\",\n\t\t\tSubcommands: []*cli.Command{\n\t\t\t\t{\n\t\t\t\t\tName:        \"workflow\",\n\t\t\t\t\tAliases:     []string{\"wf\"},\n\t\t\t\t\tUsage:       \"Run admin operation on workflow\",\n\t\t\t\t\tSubcommands: newAdminWorkflowCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"shard\",\n\t\t\t\t\tAliases:     []string{\"shar\"},\n\t\t\t\t\tUsage:       \"Run admin operation on specific shard\",\n\t\t\t\t\tSubcommands: newAdminShardManagementCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"history_host\",\n\t\t\t\t\tAliases:     []string{\"hist\"},\n\t\t\t\t\tUsage:       \"Run admin operation on history host\",\n\t\t\t\t\tSubcommands: newAdminHistoryHostCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"kafka\",\n\t\t\t\t\tAliases:     []string{\"ka\"},\n\t\t\t\t\tUsage:       \"Run admin operation on kafka messages\",\n\t\t\t\t\tSubcommands: newAdminKafkaCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"domain\",\n\t\t\t\t\tAliases:     []string{\"d\"},\n\t\t\t\t\tUsage:       \"Run admin operation on domain\",\n\t\t\t\t\tSubcommands: newAdminDomainCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"elasticsearch\",\n\t\t\t\t\tAliases:     []string{\"es\"},\n\t\t\t\t\tUsage:       \"Run admin operation on ElasticSearch\",\n\t\t\t\t\tSubcommands: newAdminElasticSearchCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"tasklist\",\n\t\t\t\t\tAliases:     []string{\"tl\"},\n\t\t\t\t\tUsage:       \"Run admin operation on taskList\",\n\t\t\t\t\tSubcommands: newAdminTaskListCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"cluster\",\n\t\t\t\t\tAliases:     []string{\"cl\"},\n\t\t\t\t\tUsage:       \"Run admin operation on cluster\",\n\t\t\t\t\tSubcommands: newAdminClusterCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"isolation-groups\",\n\t\t\t\t\tAliases:     []string{\"ig\"},\n\t\t\t\t\tUsage:       \"Run admin operation on isolation-groups\",\n\t\t\t\t\tSubcommands: newAdminIsolationGroupCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"dlq\",\n\t\t\t\t\tUsage:       \"Run admin operation on DLQ\",\n\t\t\t\t\tSubcommands: newAdminDLQCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"database\",\n\t\t\t\t\tAliases:     []string{\"db\"},\n\t\t\t\t\tUsage:       \"Run admin operations on database\",\n\t\t\t\t\tSubcommands: newDBCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"queue\",\n\t\t\t\t\tAliases:     []string{\"q\"},\n\t\t\t\t\tUsage:       \"Run admin operations on queue\",\n\t\t\t\t\tSubcommands: newAdminQueueCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"async-wf-queue\",\n\t\t\t\t\tAliases:     []string{\"aq\"},\n\t\t\t\t\tUsage:       \"Run admin operations on async workflow queues\",\n\t\t\t\t\tSubcommands: newAdminAsyncQueueCommands(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tName:        \"config\",\n\t\t\t\t\tAliases:     []string{\"c\"},\n\t\t\t\t\tUsage:       \"Run admin operation on config store\",\n\t\t\t\t\tSubcommands: newAdminConfigStoreCommands(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:        \"cluster\",\n\t\t\tAliases:     []string{\"cl\"},\n\t\t\tUsage:       \"Operate cadence cluster\",\n\t\t\tSubcommands: newClusterCommands(),\n\t\t},\n\t}\n\tapp.CommandNotFound = func(context *cli.Context, command string) {\n\t\toutput := getDeps(context).Output()\n\t\tprintMessage(output, \"command not found: \"+command)\n\t}\n\n\tfor _, opt := range opts {\n\t\topt(app)\n\t}\n\n\treturn app\n}\n\nfunc getDeps(ctx *cli.Context) cliDeps {\n\t// currently Metadata is completely unused by urfave/cli/v2, and it has fewer ways to fail\n\t// than using the ctx.Context (as you must use RunContext to supply dependencies via the Context).\n\t//\n\t// this is fairly easy to move to ctx.Context if needed, it just leads to slightly more complex code.\n\n\t// intentionally panics when an invalid context is not passed in, to help collapse logic branches.\n\t// generally speaking this should not be possible to trigger without doing something obviously questionable.\n\treturn ctx.App.Metadata[depsKey].(cliDeps)\n}\n\n// cliDeps is an interface primarily to allow it to be mocked in tests,\n// so individual client-getter funcs can be asserted as used or unused.\n//\n// exposing a struct may be good enough, it just hasn't been done yet.\ntype cliDeps interface {\n\tClientFactory\n\tIOHandler\n\tManagerFactory\n}\n\ntype IOHandler interface {\n\t// cli.Context does not contain readers/writers, they are only in cli.App.\n\t// which isn't passed to commands.\n\t//\n\t// since needing to output something is extremely common, this wrapper adds it to the context.\n\n\t// Input is the \"primary\" input to read from.\n\t//\n\t// This is currently always os.Stdin, as files are handled separately.\n\tInput() io.Reader\n\t// Output should be used to write to the primary output target.\n\t// This may be os.Stdout or a file (or an in-memory writer),\n\t// and it should only be used for the \"results\" of a command so values\n\t// can be redirected / grepped / etc reasonably.\n\t//\n\t// For progress or info-like output, use Progress() instead.\n\t//\n\t// For errors, return an error value instead.\n\tOutput() io.Writer\n\t// Progress should be used for any \"non-result\" output,\n\t// e.g. \"loading X\" or \"press enter to continue\" or similar.\n\t//\n\t// This generally writes to os.Stderr so any displayed text will\n\t// be visible when the CLI is piped, and will not be sent into \"result\" files.\n\t//\n\t// For error reporting, return an error instead, such as a commoncli.Problem\n\tProgress() io.Writer\n}\n\ntype defaultIOHandler struct {\n\tapp *cli.App\n}\n\nfunc (d *defaultIOHandler) Input() io.Reader {\n\treturn d.app.Reader\n}\nfunc (d *defaultIOHandler) Output() io.Writer {\n\treturn d.app.Writer\n}\nfunc (d *defaultIOHandler) Progress() io.Writer {\n\treturn d.app.ErrWriter\n}\n\nvar _ cliDeps = &deps{}\n\ntype deps struct {\n\tClientFactory\n\tIOHandler\n\tManagerFactory\n}\n"
  },
  {
    "path": "tools/cli/app_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/olivere/elastic\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\ntype (\n\tcliAppSuite struct {\n\t\tsuite.Suite\n\t\tapp                  *cli.App\n\t\tmockCtrl             *gomock.Controller\n\t\tserverFrontendClient *frontend.MockClient\n\t\tserverAdminClient    *admin.MockClient\n\t\ttestIOHandler        *testIOHandler\n\t}\n\n\ttestcase struct {\n\t\tname    string\n\t\tcommand string\n\t\terr     string\n\t\tmock    func()\n\t}\n)\n\nvar _ ClientFactory = (*clientFactoryMock)(nil)\n\ntype clientFactoryMock struct {\n\tserverFrontendClient frontend.Client\n\tserverAdminClient    admin.Client\n\tconfig               *config.Config\n}\n\nfunc (m *clientFactoryMock) ServerFrontendClient(c *cli.Context) (frontend.Client, error) {\n\treturn m.serverFrontendClient, nil\n}\n\nfunc (m *clientFactoryMock) ServerAdminClient(c *cli.Context) (admin.Client, error) {\n\treturn m.serverAdminClient, nil\n}\n\nfunc (m *clientFactoryMock) ServerFrontendClientForMigration(c *cli.Context) (frontend.Client, error) {\n\tpanic(\"not implemented\")\n}\n\nfunc (m *clientFactoryMock) ServerAdminClientForMigration(c *cli.Context) (admin.Client, error) {\n\tpanic(\"not implemented\")\n}\n\nfunc (m *clientFactoryMock) ElasticSearchClient(c *cli.Context) (*elastic.Client, error) {\n\tpanic(\"not implemented\")\n}\n\nfunc (m *clientFactoryMock) ServerConfig(c *cli.Context) (*config.Config, error) {\n\tif m.config != nil {\n\t\treturn m.config, nil\n\t}\n\treturn nil, fmt.Errorf(\"config not set\")\n}\n\nvar commands = []string{\n\t\"domain\", \"d\",\n\t\"workflow\", \"wf\",\n\t\"tasklist\", \"tl\",\n}\n\nvar domainName = \"cli-test-domain\"\n\n// Implements IOHandler to be used for validation in tests\ntype testIOHandler struct {\n\toutputBytes bytes.Buffer\n}\n\nfunc (t *testIOHandler) Input() io.Reader {\n\treturn os.Stdin\n}\n\nfunc (t *testIOHandler) Output() io.Writer {\n\treturn &t.outputBytes\n}\n\nfunc (t *testIOHandler) Progress() io.Writer {\n\treturn os.Stdout\n}\n\nfunc TestCLIAppSuite(t *testing.T) {\n\ts := new(cliAppSuite)\n\tsuite.Run(t, s)\n}\n\nfunc (s *cliAppSuite) SetupTest() {\n\ts.mockCtrl = gomock.NewController(s.T())\n\ts.serverFrontendClient = frontend.NewMockClient(s.mockCtrl)\n\ts.serverAdminClient = admin.NewMockClient(s.mockCtrl)\n\ts.testIOHandler = &testIOHandler{}\n\ts.app = NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: s.serverFrontendClient,\n\t\tserverAdminClient:    s.serverAdminClient,\n\t}, WithIOHandler(s.testIOHandler))\n}\n\nfunc (s *cliAppSuite) TearDownTest() {\n\ts.mockCtrl.Finish() // assert mock’s expectations\n}\n\nfunc (s *cliAppSuite) SetupSubTest() {\n\t// this is required to re-establish mocks on subtest-level\n\t// otherwise you will confusingly see expectations from previous test-cases on errors\n\ts.SetupTest()\n}\n\nfunc (s *cliAppSuite) TearDownSubTest() {\n\ts.TearDownTest()\n}\n\nfunc (s *cliAppSuite) runTestCase(tt testcase) {\n\tif tt.mock != nil {\n\t\ttt.mock()\n\t}\n\n\terr := clitest.RunCommandLine(s.T(), s.app, tt.command)\n\tif tt.err != \"\" {\n\t\tassert.ErrorContains(s.T(), err, tt.err)\n\t} else {\n\t\tassert.NoError(s.T(), err)\n\t}\n}\n\nfunc (s *cliAppSuite) TestAppCommands() {\n\tfor _, test := range commands {\n\t\tcmd := s.app.Command(test)\n\t\ts.NotNil(cmd)\n\t}\n}\n\nvar describeDomainResponseServer = &types.DescribeDomainResponse{\n\tDomainInfo: &types.DomainInfo{\n\t\tName:        \"test-domain\",\n\t\tDescription: \"a test domain\",\n\t\tOwnerEmail:  \"test@uber.com\",\n\t},\n\tConfiguration: &types.DomainConfiguration{\n\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\tEmitMetric:                             true,\n\t},\n\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\tActiveClusterName: \"active\",\n\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t{\n\t\t\t\tClusterName: \"active\",\n\t\t\t},\n\t\t\t{\n\t\t\t\tClusterName: \"standby\",\n\t\t\t},\n\t\t},\n\t},\n}\n\n// This is used to mock the domain deletion confirmation\nfunc (s *cliAppSuite) mockStdinInput(input string) func() {\n\toldStdin := os.Stdin\n\tr, w, err := os.Pipe()\n\ts.Require().NoError(err)\n\tos.Stdin = r\n\n\tgo func() {\n\t\t_, wrErr := w.Write([]byte(input + \"\\n\"))\n\t\ts.Require().NoError(wrErr)\n\t\terr = w.Close()\n\t\ts.Require().NoError(err)\n\t}()\n\n\treturn func() {\n\t\tos.Stdin = oldStdin\n\t\tr.Close()\n\t}\n}\n\nfunc (s *cliAppSuite) TestDomainDelete() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.serverFrontendClient.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(nil)\n\n\tdefer s.mockStdinInput(\"Yes\")()\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"delete\", \"--st\", \"test-token\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDomainDelete_DomainNotExist() {\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"delete\", \"--st\", \"test-token\"}))\n}\n\nfunc (s *cliAppSuite) TestDomainDelete_Failed() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.serverFrontendClient.EXPECT().DeleteDomain(gomock.Any(), gomock.Any()).Return(&types.BadRequestError{\"mock error\"})\n\n\tdefer s.mockStdinInput(\"Yes\")()\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"delete\", \"--st\", \"test-token\"}))\n}\n\nfunc (s *cliAppSuite) TestDomainDelete_Cancelled() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil)\n\n\toldStdout := os.Stdout\n\tr, w, _ := os.Pipe()\n\tos.Stdout = w\n\n\tdefer func() {\n\t\tos.Stdout = oldStdout\n\t}()\n\n\tdefer s.mockStdinInput(\"No\")()\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"delete\", \"--st\", \"test-token\"})\n\ts.Nil(err)\n\n\tw.Close()\n\tvar stdoutBuf bytes.Buffer\n\tio.Copy(&stdoutBuf, r)\n\n\ts.Contains(stdoutBuf.String(), \"Domain deletion cancelled\")\n}\n\nfunc (s *cliAppSuite) TestDomainDeprecate() {\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{\n\t\tRunID: \"run-id-example\",\n\t}, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"deprecate\", \"--st\", \"secretToken\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDomainDeprecate_FailedToStartDeprecationWorkflow() {\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.BadRequestError{\"mock error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"deprecate\", \"--st\", \"secretToken\"}))\n}\n\nfunc (s *cliAppSuite) TestDomainDescribe() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"describe\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDomainDescribe_DomainNotExist() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, &types.EntityNotExistsError{})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"describe\"}))\n}\n\nfunc (s *cliAppSuite) TestDomainDescribe_Failed() {\n\tresp := describeDomainResponseServer\n\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(resp, &types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"domain\", \"describe\"}))\n}\n\nvar (\n\teventType = types.EventTypeWorkflowExecutionStarted\n\n\tgetWorkflowExecutionHistoryResponse = &types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: &eventType,\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"TestWorkflow\"},\n\t\t\t\t\t\tTaskList:                            &types.TaskList{Name: \"taskList\"},\n\t\t\t\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\t\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\t\t\t\tIdentity:                            \"tester\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: nil,\n\t}\n)\n\nfunc (s *cliAppSuite) TestShowHistory() {\n\tresp := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(resp, nil)\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}\n\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(describeResp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"show\", \"-w\", \"wid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestShowHistoryWithID() {\n\tresp := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(resp, nil)\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}\n\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(describeResp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"showid\", \"wid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestShowHistoryWithQueryConsistencyLevel() {\n\tresp := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().\n\t\tGetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: domainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"wid\",\n\t\t\t},\n\t\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\t\tQueryConsistencyLevel:  types.QueryConsistencyLevelStrong.Ptr(),\n\t\t}).\n\t\tReturn(resp, nil)\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}\n\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(describeResp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"show\", \"-w\", \"wid\", \"--query_consistency_level\", \"strong\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestShowHistoryWithInvalidQueryConsistencyLevel() {\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"show\", \"-w\", \"wid\", \"--query_consistency_level\", \"invalid\"})\n\ts.Error(err)\n\ts.Contains(err.Error(), \"invalid query consistency level\")\n}\n\nfunc (s *cliAppSuite) TestShowHistory_PrintRawTime() {\n\tresp := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(resp, nil)\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}\n\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(describeResp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"show\", \"-w\", \"wid\", \"-prt\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestShowHistory_PrintDateTime() {\n\tresp := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(resp, nil)\n\tdescribeResp := &types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}\n\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(describeResp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"show\", \"-w\", \"wid\", \"-pdt\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestRestartWorkflow() {\n\tresp := &types.RestartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().RestartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"restart\", \"-w\", \"wid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestRestartWorkflow_Failed() {\n\tresp := &types.RestartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().RestartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, &types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"restart\", \"-w\", \"wid\"}))\n}\n\nfunc (s *cliAppSuite) TestDiagnoseWorkflow() {\n\tresp := &types.DiagnoseWorkflowExecutionResponse{Domain: \"test\", DiagnosticWorkflowExecution: &types.WorkflowExecution{WorkflowID: \"123\", RunID: uuid.New()}}\n\ts.serverFrontendClient.EXPECT().DiagnoseWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil).Times(1)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"diagnose\", \"-w\", \"wid\", \"-r\", \"rid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDiagnoseWorkflow_Failed() {\n\ts.serverFrontendClient.EXPECT().DiagnoseWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"diagnose\", \"-w\", \"wid\", \"-r\", \"rid\"}))\n}\n\nfunc (s *cliAppSuite) TestStartWorkflow() {\n\tresp := &types.StartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil).Times(2)\n\t// start with wid\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"start\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"-w\", \"wid\", \"wrp\", \"2\"})\n\ts.Nil(err)\n\t// start without wid\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"start\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"wrp\", \"2\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestStartWorkflow_Failed() {\n\tresp := &types.StartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, &types.BadRequestError{\"faked error\"})\n\t// start with wid\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"start\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"-w\", \"wid\"}))\n}\n\nfunc (s *cliAppSuite) TestRunWorkflow() {\n\tresp := &types.StartWorkflowExecutionResponse{RunID: uuid.New()}\n\thistory := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil).Times(2)\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(history, nil).Times(2)\n\t// start with wid\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"run\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"-w\", \"wid\", \"wrp\", \"2\"})\n\ts.Nil(err)\n\t// start without wid\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"run\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"wrp\", \"2\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestRunWorkflow_Failed() {\n\tresp := &types.StartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, &types.BadRequestError{\"faked error\"})\n\t// start with wid\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"run\", \"-tl\", \"testTaskList\", \"-wt\", \"testWorkflowType\", \"-et\", \"60\", \"-w\", \"wid\"}))\n}\n\nfunc (s *cliAppSuite) TestTerminateWorkflow() {\n\ts.serverFrontendClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"terminate\", \"-w\", \"wid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestTerminateWorkflow_Failed() {\n\ts.serverFrontendClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"terminate\", \"-w\", \"wid\"}))\n}\n\nfunc (s *cliAppSuite) TestCancelWorkflow() {\n\ts.serverFrontendClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"cancel\", \"-w\", \"wid\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestCancelWorkflow_Failed() {\n\ts.serverFrontendClient.EXPECT().RequestCancelWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"cancel\", \"-w\", \"wid\"}))\n}\n\nfunc (s *cliAppSuite) TestSignalWorkflow() {\n\ts.serverFrontendClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"signal\", \"-w\", \"wid\", \"-n\", \"signal-name\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestSignalWorkflow_Failed() {\n\ts.serverFrontendClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"signal\", \"-w\", \"wid\", \"-n\", \"signal-name\"}))\n}\n\nfunc (s *cliAppSuite) TestQueryWorkflowUsingStackTrace() {\n\tresp := &types.QueryWorkflowResponse{\n\t\tQueryResult: []byte(\"query-result\"),\n\t}\n\ts.serverFrontendClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"stack\", \"-w\", \"wid\"})\n\ts.Nil(err)\n}\n\nvar (\n\tcloseStatus = types.WorkflowExecutionCloseStatusCompleted\n\n\tlistClosedWorkflowExecutionsResponse = &types.ListClosedWorkflowExecutionsResponse{\n\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-list-workflow-id\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\tName: \"test-list-workflow-type\",\n\t\t\t\t},\n\t\t\t\tStartTime:     common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\tCloseTime:     common.Int64Ptr(time.Now().Add(time.Hour).UnixNano()),\n\t\t\t\tCloseStatus:   &closeStatus,\n\t\t\t\tHistoryLength: 12,\n\t\t\t},\n\t\t},\n\t}\n\n\tlistOpenWorkflowExecutionsResponse = &types.ListOpenWorkflowExecutionsResponse{\n\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-list-open-workflow-id\",\n\t\t\t\t\tRunID:      uuid.New(),\n\t\t\t\t},\n\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\tName: \"test-list-open-workflow-type\",\n\t\t\t\t},\n\t\t\t\tStartTime:     common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\tCloseTime:     common.Int64Ptr(time.Now().Add(time.Hour).UnixNano()),\n\t\t\t\tHistoryLength: 12,\n\t\t\t},\n\t\t},\n\t}\n)\n\nfunc (s *cliAppSuite) TestListWorkflow() {\n\tresp := listClosedWorkflowExecutionsResponse\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_WithWorkflowID() {\n\tresp := &types.ListClosedWorkflowExecutionsResponse{}\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-wid\", \"nothing\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_WithWorkflowType() {\n\tresp := &types.ListClosedWorkflowExecutionsResponse{}\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-wt\", \"no-type\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_PrintDateTime() {\n\tresp := listClosedWorkflowExecutionsResponse\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-pdt\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_PrintRawTime() {\n\tresp := listClosedWorkflowExecutionsResponse\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-prt\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_Open() {\n\tresp := listOpenWorkflowExecutionsResponse\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-op\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_Open_WithWorkflowID() {\n\tresp := &types.ListOpenWorkflowExecutionsResponse{}\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-op\", \"-wid\", \"nothing\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListWorkflow_Open_WithWorkflowType() {\n\tresp := &types.ListOpenWorkflowExecutionsResponse{}\n\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\ts.serverFrontendClient.EXPECT().ListOpenWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"list\", \"-op\", \"-wt\", \"no-type\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestListArchivedWorkflow() {\n\tresp := &types.ListArchivedWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().ListArchivedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"listarchived\", \"-q\", \"some query string\", \"--ps\", \"200\", \"--all\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestCountWorkflow() {\n\tresp := &types.CountWorkflowExecutionsResponse{}\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"count\"})\n\ts.Nil(err)\n\n\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"count\", \"-q\", \"'CloseTime = missing'\"})\n\ts.Nil(err)\n}\n\nvar describeTaskListResponse = &types.DescribeTaskListResponse{\n\tPollers: []*types.PollerInfo{\n\t\t{\n\t\t\tLastAccessTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\tIdentity:       \"tester\",\n\t\t},\n\t},\n}\n\nfunc (s *cliAppSuite) TestAdminDescribeWorkflow() {\n\tresp := &types.AdminDescribeWorkflowExecutionResponse{\n\t\tShardID:                \"test-shard-id\",\n\t\tHistoryAddr:            \"ip:port\",\n\t\tMutableStateInDatabase: \"{\\\"ExecutionInfo\\\":{\\\"BranchToken\\\":\\\"WQsACgAAACQ2MzI5YzEzMi1mMGI0LTQwZmUtYWYxMS1hODVmMDA3MzAzODQLABQAAAAkOWM5OWI1MjItMGEyZi00NTdmLWEyNDgtMWU0OTA0ZDg4YzVhDwAeDAAAAAAA\\\"}}\",\n\t}\n\n\ts.serverAdminClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"admin\", \"wf\", \"describe\", \"-w\", \"test-wf-id\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestAdminDescribeWorkflow_Failed() {\n\ts.serverAdminClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.BadRequestError{\"faked error\"})\n\ts.Error(s.app.Run(([]string{\"\", \"--do\", domainName, \"admin\", \"wf\", \"describe\", \"-w\", \"test-wf-id\"})))\n}\n\nfunc (s *cliAppSuite) TestAdminAddSearchAttribute() {\n\tvar promptMsg string\n\tpromptFn = func(msg string) {\n\t\tpromptMsg = msg\n\t}\n\trequest := &types.AddSearchAttributeRequest{\n\t\tSearchAttribute: map[string]types.IndexedValueType{\n\t\t\t\"testKey\": types.IndexedValueType(1),\n\t\t},\n\t}\n\ts.serverAdminClient.EXPECT().AddSearchAttribute(gomock.Any(), request).Return(nil)\n\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"admin\", \"cl\", \"asa\", \"--search_attr_key\", \"testKey\", \"--search_attr_type\", \"1\"})\n\ts.Equal(\"Are you trying to add key [testKey] with Type [Keyword]? y/N\", promptMsg)\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestAdminFailover() {\n\tresp := &types.StartWorkflowExecutionResponse{RunID: uuid.New()}\n\ts.serverFrontendClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Return(resp, nil)\n\ts.serverFrontendClient.EXPECT().SignalWorkflowExecution(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)\n\terr := s.app.Run([]string{\"\", \"admin\", \"cl\", \"fo\", \"start\", \"--tc\", \"standby\", \"--sc\", \"active\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDescribeTaskList() {\n\tresp := describeTaskListResponse\n\ts.serverFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"tasklist\", \"describe\", \"-tl\", \"test-taskList\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestDescribeTaskList_Activity() {\n\tresp := describeTaskListResponse\n\ts.serverFrontendClient.EXPECT().DescribeTaskList(gomock.Any(), gomock.Any()).Return(resp, nil)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"tasklist\", \"describe\", \"-tl\", \"test-taskList\", \"-tlt\", \"activity\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestObserveWorkflow() {\n\thistory := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(history, nil).Times(2)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"observe\", \"-w\", \"wid\"})\n\ts.Nil(err)\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"observe\", \"-w\", \"wid\", \"-sd\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestObserveWorkflowWithID() {\n\thistory := getWorkflowExecutionHistoryResponse\n\ts.serverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(history, nil).Times(2)\n\terr := s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"observeid\", \"wid\"})\n\ts.Nil(err)\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"workflow\", \"observeid\", \"wid\", \"-sd\"})\n\ts.Nil(err)\n}\n\n// TestParseTime tests the parsing of date argument in UTC and UnixNano formats\nfunc (s *cliAppSuite) TestParseTime() {\n\tpt, err := parseTime(\"\", 100)\n\ts.NoError(err)\n\ts.Equal(int64(100), pt)\n\tpt, err = parseTime(\"2018-06-07T15:04:05+00:00\", 0)\n\ts.Equal(int64(1528383845000000000), pt)\n\tpt, err = parseTime(\"1528383845000000000\", 0)\n\ts.Equal(int64(1528383845000000000), pt)\n}\n\n// TestParseTimeDateRange tests the parsing of date argument in time range format, N<duration>\n// where N is the integral multiplier, and duration can be second/minute/hour/day/week/month/year\nfunc (s *cliAppSuite) TestParseTimeDateRange() {\n\ttests := []struct {\n\t\ttimeStr  string // input\n\t\tdefVal   int64  // input\n\t\texpected int64  // expected unix nano (approx)\n\t}{\n\t\t{\n\t\t\ttimeStr:  \"1s\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-time.Second).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"100second\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-100 * time.Second).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"2m\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-2 * time.Minute).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"200minute\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-200 * time.Minute).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"3h\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-3 * time.Hour).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"1000hour\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-1000 * time.Hour).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"5d\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-5 * day).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"25day\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-25 * day).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"5w\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-5 * week).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"52week\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-52 * week).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"3M\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-3 * month).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"6month\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-6 * month).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"1y\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-year).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"7year\",\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Now().Add(-7 * year).UnixNano(),\n\t\t},\n\t\t{\n\t\t\ttimeStr:  \"100y\", // epoch time will be returned as that's the minimum unix timestamp possible\n\t\t\tdefVal:   int64(0),\n\t\t\texpected: time.Unix(0, 0).UnixNano(),\n\t\t},\n\t}\n\tdelta := int64(50 * time.Millisecond)\n\tfor _, te := range tests {\n\t\tpt, err := parseTime(te.timeStr, te.defVal)\n\t\ts.NoError(err)\n\t\ts.True(te.expected <= pt)\n\t\tpt, err = parseTime(te.timeStr, te.defVal)\n\t\ts.True(te.expected+delta >= pt)\n\t}\n}\n\nfunc (s *cliAppSuite) TestBreakLongWords() {\n\ts.Equal(\"111 222 333 4\", breakLongWords(\"1112223334\", 3))\n\ts.Equal(\"111 2 223\", breakLongWords(\"1112 223\", 3))\n\ts.Equal(\"11 122 23\", breakLongWords(\"11 12223\", 3))\n\ts.Equal(\"111\", breakLongWords(\"111\", 3))\n\ts.Equal(\"\", breakLongWords(\"\", 3))\n\ts.Equal(\"111  222\", breakLongWords(\"111 222\", 3))\n}\n\nfunc (s *cliAppSuite) TestAnyToString() {\n\targ := strings.Repeat(\"LongText\", 80)\n\tevent := &types.HistoryEvent{\n\t\tID:        1,\n\t\tEventType: &eventType,\n\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\tWorkflowType:                        &types.WorkflowType{Name: \"code.uber.internal/devexp/cadence-samples.git/cmd/samples/recipes/helloworld.Workflow\"},\n\t\t\tTaskList:                            &types.TaskList{Name: \"taskList\"},\n\t\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(60),\n\t\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(10),\n\t\t\tIdentity:                            \"tester\",\n\t\t\tInput:                               []byte(arg),\n\t\t},\n\t}\n\tres := anyToString(event, false, defaultMaxFieldLength)\n\tss, l := tablewriter.WrapString(res, 10)\n\ts.Equal(9, len(ss))\n\ts.Equal(147, l)\n}\n\nfunc (s *cliAppSuite) TestAnyToString_DecodeMapValues() {\n\tfields := map[string][]byte{\n\t\t\"TestKey\": []byte(\"testValue\"),\n\t}\n\texecution := &types.WorkflowExecutionInfo{\n\t\tMemo: &types.Memo{Fields: fields},\n\t}\n\ts.Equal(\"{HistoryLength:0, Memo:{Fields:map{TestKey:testValue}}, IsCron:false, PartitionConfig:map{}}\", anyToString(execution, true, 0))\n\n\tfields[\"TestKey2\"] = []byte(`anotherTestValue`)\n\texecution.Memo = &types.Memo{Fields: fields}\n\tgot := anyToString(execution, true, 0)\n\texpected := got == \"{HistoryLength:0, Memo:{Fields:map{TestKey2:anotherTestValue, TestKey:testValue}}, IsCron:false, PartitionConfig:map{}}\" ||\n\t\tgot == \"{HistoryLength:0, Memo:{Fields:map{TestKey:testValue, TestKey2:anotherTestValue}}, IsCron:false, PartitionConfig:map{}}\"\n\ts.True(expected)\n}\n\nfunc (s *cliAppSuite) TestIsAttributeName() {\n\ts.True(isAttributeName(\"WorkflowExecutionStartedEventAttributes\"))\n\ts.False(isAttributeName(\"workflowExecutionStartedEventAttributes\"))\n}\n\nfunc (s *cliAppSuite) TestGetWorkflowIdReusePolicy() {\n\tres, err := getWorkflowIDReusePolicy(2)\n\ts.NoError(err)\n\ts.Equal(res.String(), types.WorkflowIDReusePolicyRejectDuplicate.String())\n}\n\nfunc (s *cliAppSuite) TestGetWorkflowIdReusePolicy_Failed_ExceedRange() {\n\t_, err := getWorkflowIDReusePolicy(2147483647)\n\ts.Error(err)\n}\n\nfunc (s *cliAppSuite) TestGetWorkflowIdReusePolicy_Failed_Negative() {\n\t_, err := getWorkflowIDReusePolicy(-1)\n\ts.Error(err)\n}\n\nfunc (s *cliAppSuite) TestGetSearchAttributes() {\n\tresp := &types.GetSearchAttributesResponse{}\n\ts.serverFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(resp, nil).Times(2)\n\terr := s.app.Run([]string{\"\", \"cluster\", \"get-search-attr\"})\n\ts.Nil(err)\n\terr = s.app.Run([]string{\"\", \"--do\", domainName, \"cluster\", \"get-search-attr\"})\n\ts.Nil(err)\n}\n\nfunc (s *cliAppSuite) TestParseBool() {\n\tres, err := parseBool(\"true\")\n\ts.NoError(err)\n\ts.True(res)\n\n\tres, err = parseBool(\"false\")\n\ts.NoError(err)\n\ts.False(res)\n\n\tfor _, v := range []string{\"True, TRUE, False, FALSE, T, F\"} {\n\t\tres, err = parseBool(v)\n\t\ts.Error(err)\n\t\ts.False(res)\n\t}\n}\n\nfunc (s *cliAppSuite) TestConvertStringToRealType() {\n\tvar res interface{}\n\n\t// int\n\tres = convertStringToRealType(\"1\")\n\ts.Equal(int64(1), res)\n\n\t// bool\n\tres = convertStringToRealType(\"true\")\n\ts.Equal(true, res)\n\tres = convertStringToRealType(\"false\")\n\ts.Equal(false, res)\n\n\t// double\n\tres = convertStringToRealType(\"1.0\")\n\ts.Equal(float64(1.0), res)\n\n\t// datetime\n\tres = convertStringToRealType(\"2019-01-01T01:01:01Z\")\n\ts.Equal(time.Date(2019, 1, 1, 1, 1, 1, 0, time.UTC), res)\n\n\t// array\n\tres = convertStringToRealType(`[\"a\", \"b\", \"c\"]`)\n\ts.Equal([]interface{}{\"a\", \"b\", \"c\"}, res)\n\n\t// string\n\tres = convertStringToRealType(\"test string\")\n\ts.Equal(\"test string\", res)\n}\n\nfunc (s *cliAppSuite) TestConvertArray() {\n\tt1, _ := time.Parse(defaultDateTimeFormat, \"2019-06-07T16:16:34-08:00\")\n\tt2, _ := time.Parse(defaultDateTimeFormat, \"2019-06-07T17:16:34-08:00\")\n\ttestCases := []struct {\n\t\tname     string\n\t\tinput    string\n\t\texpected interface{}\n\t}{\n\t\t{\n\t\t\tname:     \"string\",\n\t\t\tinput:    `[\"a\", \"b\", \"c\"]`,\n\t\t\texpected: []interface{}{\"a\", \"b\", \"c\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"int\",\n\t\t\tinput:    `[1, 2, 3]`,\n\t\t\texpected: []interface{}{\"1\", \"2\", \"3\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"double\",\n\t\t\tinput:    `[1.1, 2.2, 3.3]`,\n\t\t\texpected: []interface{}{\"1.1\", \"2.2\", \"3.3\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"bool\",\n\t\t\tinput:    `[\"true\", \"false\"]`,\n\t\t\texpected: []interface{}{\"true\", \"false\"},\n\t\t},\n\t\t{\n\t\t\tname:     \"datetime\",\n\t\t\tinput:    `[\"2019-06-07T16:16:34-08:00\", \"2019-06-07T17:16:34-08:00\"]`,\n\t\t\texpected: []interface{}{t1, t2},\n\t\t},\n\t}\n\tfor _, testCase := range testCases {\n\t\tres, err := parseArray(testCase.input)\n\t\ts.Nil(err)\n\t\ts.Equal(testCase.expected, res)\n\t}\n\n\ttestCases2 := []struct {\n\t\tname     string\n\t\tinput    string\n\t\texpected error\n\t}{\n\t\t{\n\t\t\tname:  \"not array\",\n\t\t\tinput: \"normal string\",\n\t\t},\n\t\t{\n\t\t\tname:  \"empty string\",\n\t\t\tinput: \"\",\n\t\t},\n\t\t{\n\t\t\tname:  \"not json array\",\n\t\t\tinput: \"[a, b, c]\",\n\t\t},\n\t}\n\tfor _, testCase := range testCases2 {\n\t\tres, err := parseArray(testCase.input)\n\t\ts.NotNil(err)\n\t\ts.Nil(res)\n\t}\n}\n\nfunc TestWithManagerFactory(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tapp            *cli.App\n\t\texpectFactory  bool\n\t\texpectNilMeta  bool\n\t\texpectDepsType bool\n\t}{\n\t\t{\n\t\t\tname: \"Valid Metadata with deps key\",\n\t\t\tapp: &cli.App{\n\t\t\t\tMetadata: map[string]interface{}{\n\t\t\t\t\tdepsKey: &deps{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectFactory:  true,\n\t\t\texpectNilMeta:  false,\n\t\t\texpectDepsType: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"No Metadata\",\n\t\t\tapp:            &cli.App{Metadata: nil},\n\t\t\texpectFactory:  false,\n\t\t\texpectNilMeta:  true,\n\t\t\texpectDepsType: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Incorrect deps key type\",\n\t\t\tapp: &cli.App{\n\t\t\t\tMetadata: map[string]interface{}{\n\t\t\t\t\tdepsKey: \"invalid_type\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectFactory:  false,\n\t\t\texpectNilMeta:  false,\n\t\t\texpectDepsType: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockFactory := NewMockManagerFactory(ctrl)\n\n\t\t\toption := WithManagerFactory(mockFactory)\n\t\t\toption(tt.app)\n\n\t\t\tif tt.expectNilMeta {\n\t\t\t\tassert.Nil(t, tt.app.Metadata, \"Expected Metadata to be nil\")\n\t\t\t} else {\n\t\t\t\td, ok := tt.app.Metadata[depsKey].(*deps)\n\t\t\t\tif tt.expectDepsType {\n\t\t\t\t\tassert.True(t, ok, \"Expected depsKey to be of type *deps\")\n\t\t\t\t\tif tt.expectFactory {\n\t\t\t\t\t\tassert.Equal(t, mockFactory, d.ManagerFactory, \"Expected ManagerFactory to be set correctly\")\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tassert.False(t, ok, \"Expected depsKey not to be of type *deps\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestWithIOHandler(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tapp            *cli.App\n\t\texpectHandler  bool\n\t\texpectNilMeta  bool\n\t\texpectDepsType bool\n\t}{\n\t\t{\n\t\t\tname: \"Valid Metadata with deps key\",\n\t\t\tapp: &cli.App{\n\t\t\t\tMetadata: map[string]interface{}{\n\t\t\t\t\tdepsKey: &deps{},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectHandler:  true,\n\t\t\texpectNilMeta:  false,\n\t\t\texpectDepsType: true,\n\t\t},\n\t\t{\n\t\t\tname:           \"No Metadata\",\n\t\t\tapp:            &cli.App{Metadata: nil},\n\t\t\texpectHandler:  false,\n\t\t\texpectNilMeta:  true,\n\t\t\texpectDepsType: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Incorrect deps key type\",\n\t\t\tapp: &cli.App{\n\t\t\t\tMetadata: map[string]interface{}{\n\t\t\t\t\tdepsKey: \"invalid_type\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectHandler:  false,\n\t\t\texpectNilMeta:  false,\n\t\t\texpectDepsType: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttestHandler := testIOHandler{}\n\n\t\t\toption := WithIOHandler(&testHandler)\n\t\t\toption(tt.app)\n\n\t\t\tif tt.expectNilMeta {\n\t\t\t\tassert.Nil(t, tt.app.Metadata, \"Expected Metadata to be nil\")\n\t\t\t} else {\n\t\t\t\td, ok := tt.app.Metadata[depsKey].(*deps)\n\t\t\t\tif tt.expectDepsType {\n\t\t\t\t\tassert.True(t, ok, \"Expected depsKey to be of type *deps\")\n\t\t\t\t\tif tt.expectHandler {\n\t\t\t\t\t\tassert.Equal(t, &testHandler, d.IOHandler, \"Expected IOHandler to be set correctly\")\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tassert.False(t, ok, \"Expected depsKey not to be of type *deps\")\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/clitest/context.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clitest\n\nimport (\n\t\"flag\"\n\t\"strconv\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n)\n\n// CliArgument is a way to apply argument to given flagset/context\ntype CliArgument func(t *testing.T, flags *flag.FlagSet, c *cli.Context)\n\n// StringArgument introduces a new string argument for cli context\nfunc StringArgument(name, value string) CliArgument {\n\treturn func(t *testing.T, flags *flag.FlagSet, c *cli.Context) {\n\t\tt.Helper()\n\t\tflags.String(name, \"\", \"\")\n\t\trequire.NoError(t, c.Set(name, value))\n\t}\n}\n\n// IntArgument introduces a new int argument for cli context\nfunc IntArgument(name string, value int) CliArgument {\n\treturn func(t *testing.T, flags *flag.FlagSet, c *cli.Context) {\n\t\tt.Helper()\n\t\tflags.Int(name, 0, \"\")\n\t\trequire.NoError(t, c.Set(name, strconv.Itoa(value)))\n\t}\n}\n\n// BoolArgument introduces a new boolean argument for cli context\nfunc BoolArgument(name string, value bool) CliArgument {\n\treturn func(t *testing.T, flags *flag.FlagSet, c *cli.Context) {\n\t\tt.Helper()\n\t\tflags.Bool(name, value, \"\")\n\t\trequire.NoError(t, c.Set(name, strconv.FormatBool(value)))\n\t}\n}\n\n// Int64Argument introduces a new int64 argument for cli context\nfunc Int64Argument(name string, value int64) CliArgument {\n\treturn func(t *testing.T, flags *flag.FlagSet, c *cli.Context) {\n\t\tt.Helper()\n\t\tflags.Int64(name, value, \"\")\n\t\trequire.NoError(t, c.Set(name, strconv.FormatInt(value, 10)))\n\t}\n}\n\n// StringSliceArgument introduces a new string slice argument for cli context\nfunc StringSliceArgument(name string, values ...string) CliArgument {\n\treturn func(t *testing.T, flags *flag.FlagSet, c *cli.Context) {\n\t\tt.Helper()\n\t\tflags.Var(&cli.StringSlice{}, name, \"\")\n\t\tfor _, v := range values {\n\t\t\trequire.NoError(t, c.Set(name, v))\n\t\t}\n\t}\n}\n\n// NewCLIContext creates a new cli context with optional arguments\n// this is a useful to make testing of commands compact\nfunc NewCLIContext(t *testing.T, app *cli.App, args ...CliArgument) *cli.Context {\n\tt.Helper()\n\tflags := flag.NewFlagSet(\"test\", 0)\n\tcliCtx := cli.NewContext(app, flags, nil)\n\tfor _, arg := range args {\n\t\targ(t, flags, cliCtx)\n\t}\n\treturn cliCtx\n}\n"
  },
  {
    "path": "tools/cli/clitest/context_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage clitest\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n)\n\nfunc TestNewCLIContext(t *testing.T) {\n\tapp := cli.NewApp()\n\tctx := NewCLIContext(\n\t\tt,\n\t\tapp,\n\t\tStringArgument(\"region\", \"dca\"),\n\t\tIntArgument(\"shards\", 1024),\n\t\tBoolArgument(\"verbose\", true),\n\t\tBoolArgument(\"exit-if-error\", false),\n\t\tInt64Argument(\"bytes-per-minute\", 999876543210),\n\t\tStringSliceArgument(\"tags\", \"tag1\", \"tag2\"),\n\t)\n\n\tassert.True(t, ctx.IsSet(\"region\"))\n\tassert.Equal(t, \"dca\", ctx.String(\"region\"))\n\n\tassert.True(t, ctx.IsSet(\"shards\"))\n\tassert.Equal(t, 1024, ctx.Int(\"shards\"))\n\n\tassert.True(t, ctx.IsSet(\"verbose\"))\n\tassert.True(t, ctx.Bool(\"verbose\"))\n\n\tassert.True(t, ctx.IsSet(\"exit-if-error\"))\n\tassert.False(t, ctx.Bool(\"exit-if-error\"))\n\n\tassert.True(t, ctx.IsSet(\"bytes-per-minute\"))\n\tassert.Equal(t, int64(999876543210), ctx.Int64(\"bytes-per-minute\"))\n\n\tassert.True(t, ctx.IsSet(\"tags\"))\n\tassert.Equal(t, []string{\"tag1\", \"tag2\"}, ctx.StringSlice(\"tags\"))\n\n\tassert.False(t, ctx.IsSet(\"should-not-exist\"))\n}\n"
  },
  {
    "path": "tools/cli/clitest/runner.go",
    "content": "package clitest\n\nimport (\n\t\"testing\"\n\n\t\"github.com/flynn/go-shlex\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n)\n\n// RunCommandLine runs cmdline (a whole line like you'd type it in Shell)\nfunc RunCommandLine(t *testing.T, app *cli.App, cmdline string) error {\n\tt.Helper()\n\n\targs, err := shlex.Split(cmdline)\n\trequire.NoError(t, err)\n\treturn app.Run(args)\n}\n"
  },
  {
    "path": "tools/cli/clitest/runner_test.go",
    "content": "package clitest\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n)\n\nfunc TestRunCommandLine(t *testing.T) {\n\tvar output string\n\n\ttestApp := &cli.App{\n\t\tName: \"devmngr\",\n\t\tCommands: []*cli.Command{\n\t\t\t{\n\t\t\t\tName: \"upgrade\",\n\t\t\t\tFlags: []cli.Flag{\n\t\t\t\t\t&cli.StringFlag{Name: \"device\"},\n\t\t\t\t\t&cli.BoolFlag{Name: \"enabled\"},\n\t\t\t\t\t&cli.StringFlag{Name: \"reason\"},\n\t\t\t\t},\n\t\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\t\toutput = fmt.Sprintf(\n\t\t\t\t\t\t\"upgrading device: \\\"%s\\\"; enabled: %v; reason: \\\"%s\\\"\",\n\t\t\t\t\t\tc.String(\"device\"),\n\t\t\t\t\t\tc.Bool(\"enabled\"),\n\t\t\t\t\t\tc.String(\"reason\"),\n\t\t\t\t\t)\n\t\t\t\t\treturn nil\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\terr := RunCommandLine(\n\t\tt,\n\t\ttestApp,\n\t\t\"devmngr upgrade --device video-card --enabled --reason \\\"Doom: The Dark Ages is released\\\"\",\n\t)\n\trequire.NoError(t, err)\n\tassert.Equal(t,\n\t\t\"upgrading device: \\\"video-card\\\"; enabled: true; reason: \\\"Doom: The Dark Ages is released\\\"\",\n\t\toutput,\n\t)\n}\n"
  },
  {
    "path": "tools/cli/cluster.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport \"github.com/urfave/cli/v2\"\n\nfunc newClusterCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:   \"get-search-attr\",\n\t\t\tUsage:  \"get list of legal search attributes that can be used in list workflow query.\",\n\t\t\tAction: GetSearchAttributes,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/cluster_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"os\"\n\t\"sort\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\ntype (\n\tSearchAttributesRow struct {\n\t\tKey       string `header:\"Key\"`\n\t\tValueType string `header:\"Value type\"`\n\t}\n\tSearchAttributesTable []SearchAttributesRow\n)\n\nfunc (s SearchAttributesTable) Len() int {\n\treturn len(s)\n}\nfunc (s SearchAttributesTable) Swap(i, j int) {\n\ts[i], s[j] = s[j], s[i]\n}\nfunc (s SearchAttributesTable) Less(i, j int) bool {\n\treturn s[i].Key < s[j].Key\n}\n\n// GetSearchAttributes get valid search attributes\nfunc GetSearchAttributes(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\tresp, err := wfClient.GetSearchAttributes(ctx)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to get search attributes.\", err)\n\t}\n\n\ttable := SearchAttributesTable{}\n\tfor k, v := range resp.Keys {\n\t\ttable = append(table, SearchAttributesRow{Key: k, ValueType: v.String()})\n\t}\n\tsort.Sort(table)\n\treturn RenderTable(os.Stdout, table, RenderOptions{Color: true, Border: true})\n}\n"
  },
  {
    "path": "tools/cli/database.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -destination mock_manager_factory.go -self_package github.com/uber/cadence/tools/cli github.com/uber/cadence/tools/cli ManagerFactory\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"net\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\t\"github.com/uber/cadence/tools/common/flag\"\n)\n\nvar supportedDBs = append(sql.GetRegisteredPluginNames(), \"cassandra\")\n\nfunc getDBFlags() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagServiceConfigDir,\n\t\t\tAliases: []string{\"scd\"},\n\t\t\tValue:   \"config\",\n\t\t\tUsage:   \"service configuration dir\",\n\t\t\tEnvVars: []string{config.EnvKeyConfigDir},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagServiceEnv,\n\t\t\tAliases: []string{\"se\"},\n\t\t\tUsage:   \"service env for loading service configuration\",\n\t\t\tEnvVars: []string{config.EnvKeyEnvironment},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagServiceZone,\n\t\t\tAliases: []string{\"sz\"},\n\t\t\tUsage:   \"service zone for loading service configuration\",\n\t\t\tEnvVars: []string{config.EnvKeyAvailabilityZone},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDBType,\n\t\t\tValue: \"cassandra\",\n\t\t\tUsage: fmt.Sprintf(\"persistence type. Current supported options are %v\", supportedDBs),\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDBAddress,\n\t\t\tValue: \"127.0.0.1\",\n\t\t\tUsage: \"persistence address (right now only cassandra is fully supported)\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagDBPort,\n\t\t\tValue: 9042,\n\t\t\tUsage: \"persistence port\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDBRegion,\n\t\t\tUsage: \"persistence region\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagDBShard,\n\t\t\tUsage: \"number of db shards in a sharded SQL database\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagUsername,\n\t\t\tUsage: \"persistence username\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagPassword,\n\t\t\tUsage: \"persistence password\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagKeyspace,\n\t\t\tValue: \"cadence\",\n\t\t\tUsage: \"cassandra keyspace\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDatabaseName,\n\t\t\tValue: \"cadence\",\n\t\t\tUsage: \"sql database name\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagEncodingType,\n\t\t\tValue: \"thriftrw\",\n\t\t\tUsage: \"sql database encoding type\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:  FlagDecodingTypes,\n\t\t\tValue: cli.NewStringSlice(\"thriftrw\"),\n\t\t\tUsage: \"sql database decoding types\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagProtoVersion,\n\t\t\tValue: 4,\n\t\t\tUsage: \"cassandra protocol version\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:  FlagEnableTLS,\n\t\t\tUsage: \"enable TLS over cassandra connection\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagTLSCertPath,\n\t\t\tUsage: \"cassandra tls client cert path (tls must be enabled)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagTLSKeyPath,\n\t\t\tUsage: \"cassandra tls client key path (tls must be enabled)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagTLSCaPath,\n\t\t\tUsage: \"cassandra tls client ca path (tls must be enabled)\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:  FlagTLSEnableHostVerification,\n\t\t\tUsage: \"cassandra tls verify hostname and server cert (tls must be enabled)\",\n\t\t},\n\t\t&cli.GenericFlag{\n\t\t\tName:  FlagConnectionAttributes,\n\t\t\tUsage: \"a key-value set of sql database connection attributes (must be in key1=value1,key2=value2,...,keyN=valueN format, e.g. cluster=dca or cluster=dca,instance=cadence)\",\n\t\t\tValue: &flag.StringMap{},\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagRPS,\n\t\t\tUsage: \"target rps of database queries\",\n\t\t\tValue: 100,\n\t\t},\n\t}\n}\n\ntype ManagerFactory interface {\n\tinitializeExecutionManager(c *cli.Context, shardID int) (persistence.ExecutionManager, error)\n\tinitializeHistoryManager(c *cli.Context) (persistence.HistoryManager, error)\n\tinitializeShardManager(c *cli.Context) (persistence.ShardManager, error)\n\tinitializeDomainManager(c *cli.Context) (persistence.DomainManager, error)\n\tinitPersistenceFactory(c *cli.Context) (client.Factory, error)\n\tinitializeInvariantManager(ivs []invariant.Invariant) (invariant.Manager, error)\n}\n\ntype defaultManagerFactory struct {\n\tpersistenceFactory client.Factory\n}\n\nfunc (f *defaultManagerFactory) initializeExecutionManager(c *cli.Context, shardID int) (persistence.ExecutionManager, error) {\n\tfactory, err := f.getPersistenceFactory(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to get persistence factory: %w\", err)\n\t}\n\texecutionManager, err := factory.NewExecutionManager(shardID)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to initialize history manager %w\", err)\n\t}\n\treturn executionManager, nil\n}\n\nfunc (f *defaultManagerFactory) initializeHistoryManager(c *cli.Context) (persistence.HistoryManager, error) {\n\tfactory, err := f.getPersistenceFactory(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to get persistence factory: %w\", err)\n\t}\n\thistoryManager, err := factory.NewHistoryManager()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to initialize history manager %w\", err)\n\t}\n\treturn historyManager, nil\n}\n\nfunc (f *defaultManagerFactory) initializeShardManager(c *cli.Context) (persistence.ShardManager, error) {\n\tfactory, err := f.getPersistenceFactory(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to get persistence factory: %w\", err)\n\t}\n\tshardManager, err := factory.NewShardManager()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to initialize shard manager %w\", err)\n\t}\n\treturn shardManager, nil\n}\n\nfunc (f *defaultManagerFactory) initializeDomainManager(c *cli.Context) (persistence.DomainManager, error) {\n\tfactory, err := f.getPersistenceFactory(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to get persistence factory: %w\", err)\n\t}\n\tdomainManager, err := factory.NewDomainManager()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Failed to initialize domain manager: %w\", err)\n\t}\n\treturn domainManager, nil\n}\n\nfunc (f *defaultManagerFactory) getPersistenceFactory(c *cli.Context) (client.Factory, error) {\n\tvar err error\n\tif f.persistenceFactory == nil {\n\t\tf.persistenceFactory, err = getDeps(c).initPersistenceFactory(c)\n\t\tif err != nil {\n\t\t\treturn f.persistenceFactory, fmt.Errorf(\"%w\", err)\n\t\t}\n\t}\n\treturn f.persistenceFactory, nil\n}\n\nfunc (f *defaultManagerFactory) initPersistenceFactory(c *cli.Context) (client.Factory, error) {\n\tcfg, err := getDeps(c).ServerConfig(c)\n\n\tif err != nil {\n\t\tcfg = &config.Config{\n\t\t\tPersistence: config.Persistence{\n\t\t\t\tDefaultStore: \"default\",\n\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\"default\": {NoSQL: &config.NoSQL{PluginName: cassandra.PluginName}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tClusterGroupMetadata: &config.ClusterGroupMetadata{\n\t\t\t\tCurrentClusterName: \"current-cluster\",\n\t\t\t},\n\t\t}\n\t}\n\n\t// If there are any overrides provided via CLI flags, apply them here\n\tdefaultStore := cfg.Persistence.DataStores[cfg.Persistence.DefaultStore]\n\tdefaultStore, err = overrideDataStore(c, defaultStore)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in init persistence factory: %w\", err)\n\t}\n\tcfg.Persistence.DataStores[cfg.Persistence.DefaultStore] = defaultStore\n\n\tcfg.Persistence.TransactionSizeLimit = dynamicproperties.GetIntPropertyFn(constants.DefaultTransactionSizeLimit)\n\tcfg.Persistence.ErrorInjectionRate = dynamicproperties.GetFloatPropertyFn(0.0)\n\n\trps := c.Float64(FlagRPS)\n\n\treturn client.NewFactory(\n\t\t&cfg.Persistence,\n\t\tfunc() float64 { return rps },\n\t\tcfg.ClusterGroupMetadata.CurrentClusterName,\n\t\tmetrics.NewNoopMetricsClient(),\n\t\tlog.NewNoop(),\n\t\t&persistence.DynamicConfiguration{\n\t\t\tEnableSQLAsyncTransaction: dynamicproperties.GetBoolPropertyFn(false),\n\t\t},\n\t), nil\n}\n\nfunc (f *defaultManagerFactory) initializeInvariantManager(ivs []invariant.Invariant) (invariant.Manager, error) {\n\treturn invariant.NewInvariantManager(ivs), nil\n}\n\nfunc overrideDataStore(c *cli.Context, ds config.DataStore) (config.DataStore, error) {\n\tif c.IsSet(FlagDBType) {\n\t\t// overriding DBType will wipe out all settings, everything will be set from flags only\n\t\tvar err error\n\t\tds, err = createDataStore(c)\n\t\tif err != nil {\n\t\t\treturn config.DataStore{}, fmt.Errorf(\"Error in overriding data store: %w\", err)\n\t\t}\n\t}\n\n\tif ds.NoSQL != nil {\n\t\toverrideNoSQLDataStore(c, ds.NoSQL)\n\t}\n\tif ds.SQL != nil {\n\t\toverrideSQLDataStore(c, ds.SQL)\n\t}\n\n\treturn ds, nil\n}\n\nfunc createDataStore(c *cli.Context) (config.DataStore, error) {\n\tdbType := c.String(FlagDBType)\n\tswitch dbType {\n\tcase cassandra.PluginName:\n\t\treturn config.DataStore{NoSQL: &config.NoSQL{PluginName: cassandra.PluginName}}, nil\n\tdefault:\n\t\tif sql.PluginRegistered(dbType) {\n\t\t\treturn config.DataStore{SQL: &config.SQL{PluginName: dbType}}, nil\n\t\t}\n\t}\n\tfmt.Errorf(\"The DB type is not supported. Options are: %s. Error %v\", supportedDBs, nil)\n\treturn config.DataStore{}, nil\n}\n\nfunc overrideNoSQLDataStore(c *cli.Context, cfg *config.NoSQL) {\n\tif c.IsSet(FlagDBAddress) || cfg.Hosts == \"\" {\n\t\tcfg.Hosts = c.String(FlagDBAddress)\n\t}\n\tif c.IsSet(FlagDBPort) || cfg.Port == 0 {\n\t\tcfg.Port = c.Int(FlagDBPort)\n\t}\n\tif c.IsSet(FlagDBRegion) || cfg.Region == \"\" {\n\t\tcfg.Region = c.String(FlagDBRegion)\n\t}\n\tif c.IsSet(FlagUsername) || cfg.User == \"\" {\n\t\tcfg.User = c.String(FlagUsername)\n\t}\n\tif c.IsSet(FlagPassword) || cfg.Password == \"\" {\n\t\tcfg.Password = c.String(FlagPassword)\n\t}\n\tif c.IsSet(FlagKeyspace) || cfg.Keyspace == \"\" {\n\t\tcfg.Keyspace = c.String(FlagKeyspace)\n\t}\n\tif c.IsSet(FlagProtoVersion) || cfg.ProtoVersion == 0 {\n\t\tcfg.ProtoVersion = c.Int(FlagProtoVersion)\n\t}\n\n\tif cfg.TLS == nil {\n\t\tcfg.TLS = &config.TLS{}\n\t}\n\toverrideTLS(c, cfg.TLS)\n}\n\nfunc overrideSQLDataStore(c *cli.Context, cfg *config.SQL) {\n\thost, port, _ := net.SplitHostPort(cfg.ConnectAddr)\n\tif c.IsSet(FlagDBAddress) || cfg.ConnectAddr == \"\" {\n\t\thost = c.String(FlagDBAddress)\n\t}\n\tif c.IsSet(FlagDBPort) || cfg.ConnectAddr == \"\" {\n\t\tport = c.String(FlagDBPort)\n\t}\n\tcfg.ConnectAddr = net.JoinHostPort(host, port)\n\n\tif c.IsSet(FlagDBType) || cfg.PluginName == \"\" {\n\t\tcfg.PluginName = c.String(FlagDBType)\n\t}\n\tif c.IsSet(FlagDBShard) || cfg.NumShards == 0 {\n\t\tcfg.NumShards = c.Int(FlagDBShard)\n\t}\n\tif c.IsSet(FlagUsername) || cfg.User == \"\" {\n\t\tcfg.User = c.String(FlagUsername)\n\t}\n\tif c.IsSet(FlagPassword) || cfg.Password == \"\" {\n\t\tcfg.Password = c.String(FlagPassword)\n\t}\n\tif c.IsSet(FlagDatabaseName) || cfg.DatabaseName == \"\" {\n\t\tcfg.DatabaseName = c.String(FlagDatabaseName)\n\t}\n\tif c.IsSet(FlagEncodingType) || cfg.EncodingType == \"\" {\n\t\tcfg.EncodingType = c.String(FlagEncodingType)\n\t}\n\tif c.IsSet(FlagDecodingTypes) || len(cfg.DecodingTypes) == 0 {\n\t\tcfg.DecodingTypes = c.StringSlice(FlagDecodingTypes)\n\t}\n\tif c.IsSet(FlagConnectionAttributes) || cfg.ConnectAttributes == nil {\n\t\tcfg.ConnectAttributes = c.Generic(FlagConnectionAttributes).(*flag.StringMap).Value()\n\t}\n\n\tif cfg.TLS == nil {\n\t\tcfg.TLS = &config.TLS{}\n\t}\n\toverrideTLS(c, cfg.TLS)\n}\n\nfunc overrideTLS(c *cli.Context, tls *config.TLS) {\n\ttls.Enabled = c.Bool(FlagEnableTLS)\n\n\tif c.IsSet(FlagTLSCertPath) {\n\t\ttls.CertFile = c.String(FlagTLSCertPath)\n\t}\n\tif c.IsSet(FlagTLSKeyPath) {\n\t\ttls.KeyFile = c.String(FlagTLSKeyPath)\n\t}\n\tif c.IsSet(FlagTLSCaPath) {\n\t\ttls.CaFile = c.String(FlagTLSCaPath)\n\t}\n\tif c.IsSet(FlagTLSEnableHostVerification) {\n\t\ttls.EnableHostVerification = c.Bool(FlagTLSEnableHostVerification)\n\t}\n}\n"
  },
  {
    "path": "tools/cli/database_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"flag\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/persistence/client\"\n\t\"github.com/uber/cadence/common/persistence/nosql/nosqlplugin/cassandra\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/common/reconciliation/invariant\"\n\tcommonFlag \"github.com/uber/cadence/tools/common/flag\"\n)\n\nfunc TestDefaultManagerFactory(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tsetupMocks   func(*client.MockFactory, *gomock.Controller) interface{}\n\t\tmethodToTest func(*defaultManagerFactory, *cli.Context) (interface{}, error)\n\t\texpectError  bool\n\t}{\n\t\t{\n\t\t\tname: \"initializeExecutionManager - success\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockExecutionManager := persistence.NewMockExecutionManager(ctrl)\n\t\t\t\tmockFactory.EXPECT().NewExecutionManager(gomock.Any()).Return(mockExecutionManager, nil)\n\t\t\t\treturn mockExecutionManager\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeExecutionManager(ctx, 1)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeExecutionManager - error\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockFactory.EXPECT().NewExecutionManager(gomock.Any()).Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeExecutionManager(ctx, 1)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeHistoryManager - success\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockHistoryManager := persistence.NewMockHistoryManager(ctrl)\n\t\t\t\tmockFactory.EXPECT().NewHistoryManager().Return(mockHistoryManager, nil)\n\t\t\t\treturn mockHistoryManager\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeHistoryManager(ctx)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeHistoryManager - error\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockFactory.EXPECT().NewHistoryManager().Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeHistoryManager(ctx)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeShardManager - success\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockShardManager := persistence.NewMockShardManager(ctrl)\n\t\t\t\tmockFactory.EXPECT().NewShardManager().Return(mockShardManager, nil)\n\t\t\t\treturn mockShardManager\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeShardManager(ctx)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeShardManager - error\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockFactory.EXPECT().NewShardManager().Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeShardManager(ctx)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeDomainManager - success\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockDomainManager := persistence.NewMockDomainManager(ctrl)\n\t\t\t\tmockFactory.EXPECT().NewDomainManager().Return(mockDomainManager, nil)\n\t\t\t\treturn mockDomainManager\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeDomainManager(ctx)\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"initializeDomainManager - error\",\n\t\t\tsetupMocks: func(mockFactory *client.MockFactory, ctrl *gomock.Controller) interface{} {\n\t\t\t\tmockFactory.EXPECT().NewDomainManager().Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t\treturn nil\n\t\t\t},\n\t\t\tmethodToTest: func(f *defaultManagerFactory, ctx *cli.Context) (interface{}, error) {\n\t\t\t\treturn f.initializeDomainManager(ctx)\n\t\t\t},\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\t\t\tdefer ctrl.Finish()\n\n\t\t\tmockFactory := client.NewMockFactory(ctrl)\n\t\t\texpectedManager := tt.setupMocks(mockFactory, ctrl)\n\n\t\t\tf := &defaultManagerFactory{persistenceFactory: mockFactory}\n\t\t\tapp := &cli.App{}\n\t\t\tctx := cli.NewContext(app, nil, nil)\n\n\t\t\tresult, err := tt.methodToTest(f, ctx)\n\n\t\t\tif tt.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, expectedManager, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestInitPersistenceFactory(t *testing.T) {\n\tctrl := gomock.NewController(t)\n\n\t// Mock the ManagerFactory and ClientFactory\n\tmockClientFactory := NewMockClientFactory(ctrl)\n\tmockPersistenceFactory := client.NewMockFactory(ctrl)\n\n\t// Set up the context and app\n\tset := flag.NewFlagSet(\"test\", 0)\n\tapp := NewCliApp(mockClientFactory)\n\tc := cli.NewContext(app, set, nil)\n\n\t// Mock ServerConfig to return an error\n\tmockClientFactory.EXPECT().ServerConfig(gomock.Any()).Return(nil, fmt.Errorf(\"config error\")).Times(1)\n\n\t// Initialize the ManagerFactory with the mock ClientFactory\n\tmanagerFactory := defaultManagerFactory{\n\t\tpersistenceFactory: mockPersistenceFactory,\n\t}\n\n\t// Call initPersistenceFactory and validate results\n\tfactory, err := managerFactory.initPersistenceFactory(c)\n\n\t// Assert that no error occurred and a default config was used\n\tassert.NoError(t, err)\n\tassert.NotNil(t, factory)\n}\n\nfunc TestInitializeInvariantManager(t *testing.T) {\n\t// Create an instance of defaultManagerFactory\n\tfactory := &defaultManagerFactory{}\n\n\t// Define some fake invariants for testing\n\tinvariants := []invariant.Invariant{}\n\n\t// Call initializeInvariantManager\n\tmanager, err := factory.initializeInvariantManager(invariants)\n\n\t// Check that no error is returned\n\trequire.NoError(t, err, \"Expected no error from initializeInvariantManager\")\n\n\t// Check that the returned Manager is not nil\n\trequire.NotNil(t, manager, \"Expected non-nil invariant.Manager\")\n}\n\nfunc TestOverrideDataStore(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetupContext   func(app *cli.App) *cli.Context\n\t\tinputDataStore config.DataStore\n\t\texpectedError  string\n\t\texpectedSQL    *config.SQL\n\t}{\n\t\t{\n\t\t\tname: \"OverrideDBType_Cassandra\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagDBType, cassandra.PluginName, \"DB type flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagDBType, cassandra.PluginName)) // Set DBType to Cassandra\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\tinputDataStore: config.DataStore{}, // Empty DataStore to trigger createDataStore\n\t\t\texpectedError:  \"\",\n\t\t\texpectedSQL:    nil, // No SQL expected for Cassandra\n\t\t},\n\t\t{\n\t\t\tname: \"OverrideSQLDataStore\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\t// Create a new mock SQL plugin using gomock\n\t\t\t\tctrl := gomock.NewController(t)\n\t\t\t\tmockSQLPlugin := sqlplugin.NewMockPlugin(ctrl)\n\n\t\t\t\t// Register the mock SQL plugin for \"mysql\"\n\t\t\t\tsql.RegisterPlugin(\"mysql\", mockSQLPlugin)\n\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.String(FlagDBType, \"mysql\", \"DB type flag\") // Set SQL database type\n\t\t\t\tset.String(FlagDBAddress, \"127.0.0.1\", \"DB address flag\")\n\t\t\t\tset.String(FlagDBPort, \"3306\", \"DB port flag\")\n\t\t\t\tset.String(FlagUsername, \"testuser\", \"DB username flag\")\n\t\t\t\tset.String(FlagPassword, \"testpass\", \"DB password flag\")\n\t\t\t\tconnAttr := &commonFlag.StringMap{}\n\t\t\t\trequire.NoError(t, connAttr.Set(\"attr1=value1\"))\n\t\t\t\trequire.NoError(t, connAttr.Set(\"attr2=value2\"))\n\t\t\t\tset.Var(connAttr, FlagConnectionAttributes, \"Connection attributes flag\")\n\t\t\t\trequire.NoError(t, set.Set(FlagDBType, \"mysql\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagDBAddress, \"127.0.0.1\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagDBPort, \"3306\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagUsername, \"testuser\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagPassword, \"testpass\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\texpectedSQL: &config.SQL{\n\t\t\t\tPluginName:  \"mysql\",\n\t\t\t\tConnectAddr: \"127.0.0.1:3306\",\n\t\t\t\tUser:        \"testuser\",\n\t\t\t\tPassword:    \"testpass\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Set up app and context\n\t\t\tapp := cli.NewApp()\n\t\t\tc := tt.setupContext(app)\n\n\t\t\t// Call overrideDataStore with initial DataStore and capture result\n\t\t\tresult, err := overrideDataStore(c, tt.inputDataStore)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.ErrorContains(t, err, tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\t// Validate SQL DataStore settings if expected\n\t\t\t\tif tt.expectedSQL != nil && result.SQL != nil {\n\t\t\t\t\tassert.Equal(t, tt.expectedSQL.PluginName, result.SQL.PluginName)\n\t\t\t\t\tassert.Equal(t, tt.expectedSQL.ConnectAddr, result.SQL.ConnectAddr)\n\t\t\t\t\tassert.Equal(t, tt.expectedSQL.User, result.SQL.User)\n\t\t\t\t\tassert.Equal(t, tt.expectedSQL.Password, result.SQL.Password)\n\t\t\t\t}\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestOverrideTLS(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tsetupContext func(app *cli.App) *cli.Context\n\t\texpectedTLS  config.TLS\n\t}{\n\t\t{\n\t\t\tname: \"AllTLSFlagsSet\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Bool(FlagEnableTLS, true, \"Enable TLS flag\")\n\t\t\t\tset.String(FlagTLSCertPath, \"/path/to/cert\", \"TLS Cert Path\")\n\t\t\t\tset.String(FlagTLSKeyPath, \"/path/to/key\", \"TLS Key Path\")\n\t\t\t\tset.String(FlagTLSCaPath, \"/path/to/ca\", \"TLS CA Path\")\n\t\t\t\tset.Bool(FlagTLSEnableHostVerification, true, \"Enable Host Verification\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagEnableTLS, \"true\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagTLSCertPath, \"/path/to/cert\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagTLSKeyPath, \"/path/to/key\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagTLSCaPath, \"/path/to/ca\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagTLSEnableHostVerification, \"true\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\texpectedTLS: config.TLS{\n\t\t\t\tEnabled:                true,\n\t\t\t\tCertFile:               \"/path/to/cert\",\n\t\t\t\tKeyFile:                \"/path/to/key\",\n\t\t\t\tCaFile:                 \"/path/to/ca\",\n\t\t\t\tEnableHostVerification: true,\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"PartialTLSFlagsSet\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\tset.Bool(FlagEnableTLS, true, \"Enable TLS flag\")\n\t\t\t\tset.String(FlagTLSCertPath, \"/path/to/cert\", \"TLS Cert Path\")\n\n\t\t\t\trequire.NoError(t, set.Set(FlagEnableTLS, \"true\"))\n\t\t\t\trequire.NoError(t, set.Set(FlagTLSCertPath, \"/path/to/cert\"))\n\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\texpectedTLS: config.TLS{\n\t\t\t\tEnabled:  true,\n\t\t\t\tCertFile: \"/path/to/cert\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"NoTLSFlagsSet\",\n\t\t\tsetupContext: func(app *cli.App) *cli.Context {\n\t\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\t\treturn cli.NewContext(app, set, nil)\n\t\t\t},\n\t\t\texpectedTLS: config.TLS{},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Set up app and context\n\t\t\tapp := cli.NewApp()\n\t\t\tc := tt.setupContext(app)\n\n\t\t\t// Initialize an empty TLS config and apply overrideTLS\n\t\t\ttlsConfig := &config.TLS{}\n\t\t\toverrideTLS(c, tlsConfig)\n\n\t\t\t// Validate TLS config settings\n\t\t\tassert.Equal(t, tt.expectedTLS.Enabled, tlsConfig.Enabled)\n\t\t\tassert.Equal(t, tt.expectedTLS.CertFile, tlsConfig.CertFile)\n\t\t\tassert.Equal(t, tt.expectedTLS.KeyFile, tlsConfig.KeyFile)\n\t\t\tassert.Equal(t, tt.expectedTLS.CaFile, tlsConfig.CaFile)\n\t\t\tassert.Equal(t, tt.expectedTLS.EnableHostVerification, tlsConfig.EnableHostVerification)\n\t\t})\n\t}\n}\n\nfunc newClientFactoryMock() *clientFactoryMock {\n\treturn &clientFactoryMock{\n\t\tserverFrontendClient: frontend.NewMockClient(gomock.NewController(nil)),\n\t\tserverAdminClient:    admin.NewMockClient(gomock.NewController(nil)),\n\t\tconfig: &config.Config{\n\t\t\tPersistence: config.Persistence{\n\t\t\t\tDefaultStore: \"default\",\n\t\t\t\tDataStores: map[string]config.DataStore{\n\t\t\t\t\t\"default\": {NoSQL: &config.NoSQL{PluginName: cassandra.PluginName}},\n\t\t\t\t},\n\t\t\t},\n\t\t\tClusterGroupMetadata: &config.ClusterGroupMetadata{\n\t\t\t\tCurrentClusterName: \"current-cluster\",\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/defs.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/fatih/color\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\tlocalHost    = \"127.0.0.1\"\n\ttchannelPort = localHost + \":7933\"\n\tgrpcPort     = localHost + \":7833\"\n\n\tgrpcTransport   = \"grpc\"\n\tthriftTransport = \"tchannel\"\n\n\tmaxOutputStringLength = 200 // max length for output string\n\tmaxWorkflowTypeLength = 32  // max item length for output workflow type in table\n\tdefaultMaxFieldLength = 500 // default max length for each attribute field\n\tmaxWordLength         = 120 // if text length is larger than maxWordLength, it will be inserted spaces\n\n\t// regex expression for parsing time durations, shorter, longer notations and numeric value respectively\n\tdefaultDateTimeRangeShortRE = \"^[1-9][0-9]*[smhdwMy]$\"                                // eg. 1s, 20m, 300h etc.\n\tdefaultDateTimeRangeLongRE  = \"^[1-9][0-9]*(second|minute|hour|day|week|month|year)$\" // eg. 1second, 20minute, 300hour etc.\n\tdefaultDateTimeRangeNum     = \"^[1-9][0-9]*\"                                          // eg. 1, 20, 300 etc.\n\n\t// time ranges\n\tday   = 24 * time.Hour\n\tweek  = 7 * day\n\tmonth = 30 * day\n\tyear  = 365 * day\n\n\tdefaultTimeFormat                            = \"15:04:05\"   // used for converting UnixNano to string like 16:16:36 (only time)\n\tdefaultDateTimeFormat                        = time.RFC3339 // used for converting UnixNano to string like 2018-02-15T16:16:36-08:00\n\tdefaultDomainRetentionDays                   = 3\n\tdefaultContextTimeoutInSeconds               = 5\n\tdefaultContextTimeout                        = defaultContextTimeoutInSeconds * time.Second\n\tdefaultContextTimeoutForLongPoll             = 2 * time.Minute\n\tdefaultContextTimeoutForListArchivedWorkflow = 3 * time.Minute\n\n\tdefaultDecisionTimeoutInSeconds = 10\n\tdefaultPageSizeForList          = 500\n\tdefaultPageSizeForScan          = 2000\n\tdefaultWorkflowIDReusePolicy    = types.WorkflowIDReusePolicyAllowDuplicateFailedOnly\n\n\tworkflowStatusNotSet = -1\n\tshowErrorStackEnv    = `CADENCE_CLI_SHOW_STACKS`\n\n\tsearchAttrInputSeparator = \"|\"\n\n\tdefaultGracefulFailoverTimeoutInSeconds = 60\n)\n\nvar envKeysForUserName = []string{\n\t\"USER\",\n\t\"LOGNAME\",\n\t\"HOME\",\n}\n\nconst resetTypeFirstDecisionCompleted = \"FirstDecisionCompleted\"\nconst resetTypeLastDecisionCompleted = \"LastDecisionCompleted\"\nconst resetTypeLastContinuedAsNew = \"LastContinuedAsNew\"\nconst resetTypeBadBinary = \"BadBinary\"\nconst resetTypeDecisionCompletedTime = \"DecisionCompletedTime\"\nconst resetTypeFirstDecisionScheduled = \"FirstDecisionScheduled\"\nconst resetTypeLastDecisionScheduled = \"LastDecisionScheduled\"\n\nvar resetTypesMap = map[string]string{\n\tresetTypeFirstDecisionCompleted: \"\",\n\tresetTypeLastDecisionCompleted:  \"\",\n\tresetTypeLastContinuedAsNew:     \"\",\n\tresetTypeBadBinary:              FlagResetBadBinaryChecksum,\n\tresetTypeDecisionCompletedTime:  FlagEarliestTime,\n\tresetTypeFirstDecisionScheduled: \"\",\n\tresetTypeLastDecisionScheduled:  \"\",\n}\n\ntype jsonType int\n\nconst (\n\tjsonTypeInput jsonType = iota\n\tjsonTypeMemo\n\tjsonTypeHeader\n\tjsonTypeSignal\n)\n\nvar (\n\tcolorRed     = color.New(color.FgRed).SprintFunc()\n\tcolorMagenta = color.New(color.FgMagenta).SprintFunc()\n\tcolorGreen   = color.New(color.FgGreen).SprintFunc()\n\n\toptionErr               = \"there is something wrong with your command options\"\n\tosExit                  = os.Exit\n\tworkflowClosedStatusMap = map[string]types.WorkflowExecutionCloseStatus{\n\t\t\"completed\":        types.WorkflowExecutionCloseStatusCompleted,\n\t\t\"failed\":           types.WorkflowExecutionCloseStatusFailed,\n\t\t\"canceled\":         types.WorkflowExecutionCloseStatusCanceled,\n\t\t\"terminated\":       types.WorkflowExecutionCloseStatusTerminated,\n\t\t\"continued_as_new\": types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\"timed_out\":        types.WorkflowExecutionCloseStatusTimedOut,\n\t\t// below are some alias\n\t\t\"c\":              types.WorkflowExecutionCloseStatusCompleted,\n\t\t\"complete\":       types.WorkflowExecutionCloseStatusCompleted,\n\t\t\"f\":              types.WorkflowExecutionCloseStatusFailed,\n\t\t\"fail\":           types.WorkflowExecutionCloseStatusFailed,\n\t\t\"cancel\":         types.WorkflowExecutionCloseStatusCanceled,\n\t\t\"terminate\":      types.WorkflowExecutionCloseStatusTerminated,\n\t\t\"term\":           types.WorkflowExecutionCloseStatusTerminated,\n\t\t\"continue\":       types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\"cont\":           types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\"continuedasnew\": types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\"continueasnew\":  types.WorkflowExecutionCloseStatusContinuedAsNew,\n\t\t\"timedout\":       types.WorkflowExecutionCloseStatusTimedOut,\n\t\t\"timeout\":        types.WorkflowExecutionCloseStatusTimedOut,\n\t}\n)\n"
  },
  {
    "path": "tools/cli/domain.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// by default we don't require any domain data. But this can be overridden by calling SetRequiredDomainDataKeys()\nvar requiredDomainDataKeys = []string{}\n\n// SetRequiredDomainDataKeys will set requiredDomainDataKeys\nfunc SetRequiredDomainDataKeys(keys []string) {\n\trequiredDomainDataKeys = keys\n}\n\nfunc checkRequiredDomainDataKVs(domainData map[string]string) error {\n\t// check requiredDomainDataKeys\n\tfor _, k := range requiredDomainDataKeys {\n\t\t_, ok := domainData[k]\n\t\tif !ok {\n\t\t\treturn fmt.Errorf(\"domain data error, missing required key %v . All required keys: %v\", k, requiredDomainDataKeys)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc checkNoAdditionalArgsPassed(c *cli.Context) error {\n\tif c.NArg() > 0 {\n\t\treturn fmt.Errorf(\"Domain commands cannot have arguments: <%v>\\nClusters are now specified as --clusters c1,c2 see help for more info\", strings.Join(c.Args().Slice(), \" \"))\n\t}\n\treturn nil\n}\n\nfunc newDomainCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"register\",\n\t\t\tAliases: []string{\"re\"},\n\t\t\tUsage:   \"Register workflow domain\",\n\t\t\tFlags:   registerDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.RegisterDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"update\",\n\t\t\tAliases: []string{\"up\", \"u\"},\n\t\t\tUsage:   \"Update existing workflow domain\",\n\t\t\tFlags:   updateDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.UpdateDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"delete\",\n\t\t\tAliases: []string{\"del\"},\n\t\t\tUsage:   \"Delete existing workflow domain\",\n\t\t\tFlags:   deleteDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DeleteDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"deprecate\",\n\t\t\tAliases: []string{\"dep\"},\n\t\t\tUsage:   \"Deprecate existing workflow domain\",\n\t\t\tFlags:   deprecateDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DeprecateDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe existing workflow domain\",\n\t\t\tFlags:   describeDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.DescribeDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"migration\",\n\t\t\tAliases: []string{\"mi\"},\n\t\t\tUsage:   \"Migrate existing domain to new domain. This command only validates the settings. It does not perform actual data migration\",\n\t\t\tFlags:   migrateDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\t// exit on error already handled in the command\n\t\t\t\t// TODO best practice is to return error if validation fails\n\t\t\t\terr = NewDomainMigrationCommand(c).Validation(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn commoncli.Problem(\"Failed validation: \", err)\n\t\t\t\t}\n\t\t\t\treturn nil\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"failover\",\n\t\t\tAliases: []string{\"fo\"},\n\t\t\tUsage:   \"Failover workflow domain to target active cluster\",\n\t\t\tFlags:   failoverDomainFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.FailoverDomain(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"list-failover-history\",\n\t\t\tAliases: []string{\"lfh\"},\n\t\t\tUsage:   \"List failover history for a domain\",\n\t\t\tFlags:   listFailoverHistoryFlags,\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\terr := checkNoAdditionalArgsPassed(c)\n\t\t\t\tif err != nil {\n\t\t\t\t\treturn err\n\t\t\t\t}\n\t\t\t\treturn withDomainClient(c, false, func(dc *domainCLIImpl) error {\n\t\t\t\t\treturn dc.ListFailoverHistory(c)\n\t\t\t\t})\n\t\t\t},\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/domain_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"os\"\n\t\"regexp\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/domaindeprecation\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n\t\"github.com/uber/cadence/tools/common/flag\"\n)\n\nconst (\n\tdecisionTimeoutInSeconds    = 5 * 60\n\tworkflowStartToCloseTimeout = 24 * 30 * 60 * 60 // 30 days\n)\n\nvar (\n\tgracefulFailoverType = \"grace\"\n)\n\ntype (\n\tdomainCLIImpl struct {\n\t\t// used when making RPC call to frontend service\n\t\tfrontendClient      frontend.Client\n\t\tfrontendAdminClient admin.Client\n\n\t\t// act as admin to modify domain in DB directly\n\t\tdomainHandler domain.Handler\n\t}\n)\n\n// newDomainCLI creates a domain CLI\nfunc newDomainCLI(\n\tc *cli.Context,\n\tisAdminMode bool,\n) (*domainCLIImpl, error) {\n\td := &domainCLIImpl{}\n\tvar err error\n\td.frontendClient, err = initializeFrontendClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif isAdminMode {\n\t\td.frontendAdminClient, err = initializeFrontendAdminClient(c)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\td.domainHandler, err = initializeAdminDomainHandler(c)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t}\n\treturn d, nil\n}\n\n// RegisterDomain register a domain\nfunc (d *domainCLIImpl) RegisterDomain(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tdescription := c.String(FlagDescription)\n\townerEmail := c.String(FlagOwnerEmail)\n\tretentionDays := defaultDomainRetentionDays\n\n\tif c.IsSet(FlagRetentionDays) {\n\t\tretentionDays = c.Int(FlagRetentionDays)\n\t}\n\tsecurityToken := c.String(FlagSecurityToken)\n\n\tisGlobalDomain := true\n\tif c.IsSet(FlagIsGlobalDomain) {\n\t\tisGlobalDomain, err = strconv.ParseBool(c.String(FlagIsGlobalDomain))\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Option %s format is invalid.\", FlagIsGlobalDomain), err)\n\t\t}\n\t}\n\n\tvar domainData *flag.StringMap\n\tif c.IsSet(FlagDomainData) {\n\t\tdomainData = c.Generic(FlagDomainData).(*flag.StringMap)\n\t}\n\tif len(requiredDomainDataKeys) > 0 {\n\t\terr = checkRequiredDomainDataKVs(domainData.Value())\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Domain data missed required data.\", err)\n\t\t}\n\t}\n\n\tactiveClusterName := \"\"\n\tif c.IsSet(FlagActiveClusterName) {\n\t\tactiveClusterName = c.String(FlagActiveClusterName)\n\t}\n\n\tvar clusters []*types.ClusterReplicationConfiguration\n\tif c.IsSet(FlagClusters) {\n\t\tfor _, clusterStr := range c.StringSlice(FlagClusters) {\n\t\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\t\tClusterName: clusterStr,\n\t\t\t})\n\t\t}\n\t}\n\n\thas, err := archivalStatus(c, FlagHistoryArchivalStatus)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to parse %s flag: %w\", FlagHistoryArchivalStatus, err)\n\t}\n\tvas, err := archivalStatus(c, FlagVisibilityArchivalStatus)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to parse %s flag: %w\", FlagVisibilityArchivalStatus, err)\n\t}\n\n\tvar activeClusters *types.ActiveClusters\n\tif c.IsSet(FlagActiveClusters) {\n\t\tac, err := parseActiveClustersByClusterAttribute(c.String(FlagActiveClusters))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tactiveClusters = &ac\n\t}\n\n\trequest := &types.RegisterDomainRequest{\n\t\tName:                                   domainName,\n\t\tDescription:                            description,\n\t\tOwnerEmail:                             ownerEmail,\n\t\tData:                                   domainData.Value(),\n\t\tWorkflowExecutionRetentionPeriodInDays: int32(retentionDays),\n\t\tClusters:                               clusters,\n\t\tActiveClusterName:                      activeClusterName,\n\t\tActiveClusters:                         activeClusters,\n\t\tSecurityToken:                          securityToken,\n\t\tHistoryArchivalStatus:                  has,\n\t\tHistoryArchivalURI:                     c.String(FlagHistoryArchivalURI),\n\t\tVisibilityArchivalStatus:               vas,\n\t\tVisibilityArchivalURI:                  c.String(FlagVisibilityArchivalURI),\n\t\tIsGlobalDomain:                         isGlobalDomain,\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\terr = d.registerDomain(ctx, request)\n\tif err != nil {\n\t\tif _, ok := err.(*types.DomainAlreadyExistsError); !ok {\n\t\t\treturn commoncli.Problem(\"Register Domain operation failed.\", err)\n\t\t}\n\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s already registered.\", domainName), err)\n\t}\n\tfmt.Printf(\"Domain %s successfully registered.\\n\", domainName)\n\treturn nil\n}\n\n// UpdateDomain updates a domain\nfunc (d *domainCLIImpl) UpdateDomain(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tvar updateRequest *types.UpdateDomainRequest\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tif c.IsSet(FlagActiveClusterName) { // active-passive domain failover\n\t\tactiveCluster := c.String(FlagActiveClusterName)\n\t\tfmt.Printf(\"Will set active cluster name to: %s, other flag will be omitted.\\n\", activeCluster)\n\n\t\tvar failoverTimeout *int32\n\t\tif c.String(FlagFailoverType) == gracefulFailoverType {\n\t\t\ttimeout := int32(c.Int(FlagFailoverTimeout))\n\t\t\tfailoverTimeout = &timeout\n\t\t}\n\n\t\tupdateRequest = &types.UpdateDomainRequest{\n\t\t\tName:                     domainName,\n\t\t\tActiveClusterName:        common.StringPtr(activeCluster),\n\t\t\tFailoverTimeoutInSeconds: failoverTimeout,\n\t\t}\n\t} else if c.IsSet(FlagActiveClusters) { // active-active domain failover\n\t\tactiveClusters, err := parseActiveClustersByClusterAttribute(c.String(FlagActiveClusters))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tupdateRequest = &types.UpdateDomainRequest{\n\t\t\tName: domainName,\n\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\tAttributeScopes: activeClusters.AttributeScopes,\n\t\t\t},\n\t\t}\n\t} else {\n\t\tresp, err := d.describeDomain(ctx, &types.DescribeDomainRequest{\n\t\t\tName: common.StringPtr(domainName),\n\t\t})\n\t\tif err != nil {\n\t\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\t\treturn commoncli.Problem(\"Operation UpdateDomain failed.\", err)\n\t\t\t}\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t\t}\n\n\t\tdescription := resp.DomainInfo.GetDescription()\n\t\townerEmail := resp.DomainInfo.GetOwnerEmail()\n\t\tretentionDays := resp.Configuration.GetWorkflowExecutionRetentionPeriodInDays()\n\t\temitMetric := resp.Configuration.GetEmitMetric()\n\t\tvar clusters []*types.ClusterReplicationConfiguration\n\n\t\tif c.IsSet(FlagDescription) {\n\t\t\tdescription = c.String(FlagDescription)\n\t\t}\n\t\tif c.IsSet(FlagOwnerEmail) {\n\t\t\townerEmail = c.String(FlagOwnerEmail)\n\t\t}\n\t\tvar domainData *flag.StringMap\n\t\tif c.IsSet(FlagDomainData) {\n\t\t\tdomainData = c.Generic(FlagDomainData).(*flag.StringMap)\n\t\t}\n\t\tif c.IsSet(FlagRetentionDays) {\n\t\t\tretentionDays = int32(c.Int(FlagRetentionDays))\n\t\t}\n\t\tif c.IsSet(FlagClusters) {\n\t\t\tfor _, clusterStr := range c.StringSlice(FlagClusters) {\n\t\t\t\tclusters = append(clusters, &types.ClusterReplicationConfiguration{\n\t\t\t\t\tClusterName: clusterStr,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\n\t\tvar binBinaries *types.BadBinaries\n\t\tif c.IsSet(FlagAddBadBinary) {\n\t\t\tif !c.IsSet(FlagReason) {\n\t\t\t\treturn commoncli.Problem(\"Must provide a reason.\", nil)\n\t\t\t}\n\t\t\tbinChecksum := c.String(FlagAddBadBinary)\n\t\t\treason := c.String(FlagReason)\n\t\t\toperator := getCurrentUserFromEnv()\n\t\t\tbinBinaries = &types.BadBinaries{\n\t\t\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\t\t\tbinChecksum: {\n\t\t\t\t\t\tReason:   reason,\n\t\t\t\t\t\tOperator: operator,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t}\n\t\t}\n\n\t\tvar badBinaryToDelete *string\n\t\tif c.IsSet(FlagRemoveBadBinary) {\n\t\t\tbadBinaryToDelete = common.StringPtr(c.String(FlagRemoveBadBinary))\n\t\t}\n\n\t\thas, err := archivalStatus(c, FlagHistoryArchivalStatus)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to parse %s flag: %w\", FlagHistoryArchivalStatus, err)\n\t\t}\n\t\tvas, err := archivalStatus(c, FlagVisibilityArchivalStatus)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"failed to parse %s flag: %w\", FlagVisibilityArchivalStatus, err)\n\t\t}\n\n\t\tupdateRequest = &types.UpdateDomainRequest{\n\t\t\tName:                                   domainName,\n\t\t\tDescription:                            common.StringPtr(description),\n\t\t\tOwnerEmail:                             common.StringPtr(ownerEmail),\n\t\t\tData:                                   domainData.Value(),\n\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(retentionDays),\n\t\t\tEmitMetric:                             common.BoolPtr(emitMetric),\n\t\t\tHistoryArchivalStatus:                  has,\n\t\t\tHistoryArchivalURI:                     common.StringPtr(c.String(FlagHistoryArchivalURI)),\n\t\t\tVisibilityArchivalStatus:               vas,\n\t\t\tVisibilityArchivalURI:                  common.StringPtr(c.String(FlagVisibilityArchivalURI)),\n\t\t\tBadBinaries:                            binBinaries,\n\t\t\tClusters:                               clusters,\n\t\t\tDeleteBadBinary:                        badBinaryToDelete,\n\t\t}\n\t}\n\n\tsecurityToken := c.String(FlagSecurityToken)\n\tupdateRequest.SecurityToken = securityToken\n\t_, err = d.updateDomain(ctx, updateRequest)\n\tif err != nil {\n\t\tvar entityNotExistsErr *types.EntityNotExistsError\n\t\tif errors.As(err, &entityNotExistsErr) {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t\t}\n\t\tvar accessDeniedErr *types.AccessDeniedError\n\t\tif errors.As(err, &accessDeniedErr) {\n\t\t\tfmt.Fprintf(os.Stderr, \"WARNING: Update domain operation may not be available for general user use and is reserved for administrative operations.\\n\")\n\t\t\tfmt.Fprintf(os.Stderr, \"Please use the 'cadence domain failover' cli-command instead for domain management.\\n\")\n\t\t\treturn commoncli.Problem(\"Operation UpdateDomain failed due to authorization.\", err)\n\t\t}\n\t\treturn commoncli.Problem(\"Operation UpdateDomain failed.\", err)\n\t}\n\tfmt.Printf(\"Domain %s successfully updated.\\n\", domainName)\n\treturn nil\n}\n\nfunc (d *domainCLIImpl) DeleteDomain(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tsecurityToken := c.String(FlagSecurityToken)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\t// First describe the domain to show its details\n\tdescribeRequest := &types.DescribeDomainRequest{\n\t\tName: &domainName,\n\t}\n\tresp, err := d.describeDomain(ctx, describeRequest)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn commoncli.Problem(\"Operation DescribeDomain failed.\", err)\n\t\t}\n\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t}\n\n\tif err := Render(c, newDomainRow(resp), RenderOptions{\n\t\tDefaultTemplate: templateDomain,\n\t\tColor:           true,\n\t\tBorder:          true,\n\t\tPrintDateTime:   true,\n\t}); err != nil {\n\t\treturn fmt.Errorf(\"failed to render domain details: %w\", err)\n\t}\n\n\tfmt.Print(\"Do you want to delete this domain? \\n\")\n\treader := bufio.NewReader(os.Stdin)\n\tfor {\n\t\tfmt.Print(\"Please confirm[Yes/No]:\")\n\t\ttext, err := reader.ReadString('\\n')\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to get confirmation for domain deletion\", err)\n\t\t}\n\t\tif strings.EqualFold(strings.TrimSpace(text), \"yes\") {\n\t\t\tbreak\n\t\t} else {\n\t\t\tfmt.Println(\"Domain deletion cancelled\")\n\t\t\treturn nil\n\t\t}\n\t}\n\n\terr = d.deleteDomain(ctx, &types.DeleteDomainRequest{\n\t\tName:          domainName,\n\t\tSecurityToken: securityToken,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation Delete domain failed.\", err)\n\t}\n\tfmt.Printf(\"Domain %s successfully deleted.\\n\", domainName)\n\treturn nil\n}\n\nfunc (d *domainCLIImpl) DeprecateDomain(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not provided: \", err)\n\t}\n\n\tsecurityToken := c.String(FlagSecurityToken)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tparams := domaindeprecation.DomainDeprecationParams{\n\t\tDomainName:    domainName,\n\t\tSecurityToken: securityToken,\n\t}\n\tinput, err := json.Marshal(params)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to encode domain deprecation parameters\", err)\n\t}\n\n\tworkflowID := uuid.NewRandom().String()\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:     constants.SystemLocalDomainName,\n\t\tWorkflowID: fmt.Sprintf(\"domain-deprecation-%s-%s\", domainName, workflowID),\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: domaindeprecation.DomainDeprecationWorkflowTypeName,\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: domaindeprecation.DomainDeprecationTaskListName,\n\t\t},\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32(workflowStartToCloseTimeout)),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(decisionTimeoutInSeconds),\n\t\tRequestID:                           uuid.New(),\n\t\tInput:                               input,\n\t}\n\n\tresp, err := frontendClient.StartWorkflowExecution(ctx, startRequest)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to start domain deprecation workflow\", err)\n\t}\n\n\tfmt.Printf(\"Domain deprecation is in progress. Workflow ID: %s, Run ID: %s\\n\", startRequest.WorkflowID, resp.GetRunID())\n\treturn nil\n}\n\n// FailoverDomain fails over a single domain to a target cluster\nfunc (d *domainCLIImpl) FailoverDomain(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\n\tfailoverRequest := &types.FailoverDomainRequest{\n\t\tDomainName: domainName,\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\tif c.IsSet(FlagActiveClustersJSON) && c.IsSet(FlagActiveClusters) {\n\t\treturn commoncli.Problem(\"Cannot use both --active_clusters_json and --active_clusters flags.\", nil)\n\t}\n\n\tif !c.IsSet(FlagActiveClusterName) && !c.IsSet(FlagActiveClusters) && !c.IsSet(FlagActiveClustersJSON) {\n\t\treturn commoncli.Problem(\"At least one of the flags --active_cluster, --active_clusters or --active_clusters_json must be provided.\", nil)\n\t}\n\n\tif c.IsSet(FlagActiveClusterName) { // active-passive domain failover\n\t\tactiveCluster := c.String(FlagActiveClusterName)\n\t\tfailoverRequest.DomainActiveClusterName = common.StringPtr(activeCluster)\n\t}\n\tif c.IsSet(FlagActiveClusters) { // active-active domain failover\n\t\tac, err := parseActiveClustersByClusterAttribute(c.String(FlagActiveClusters))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfailoverRequest.ActiveClusters = &ac\n\t}\n\n\tif c.IsSet(FlagActiveClustersJSON) {\n\t\tac, err := parseActiveClustersByClusterAttributeFromJSON(c.String(FlagActiveClustersJSON))\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tfailoverRequest.ActiveClusters = &ac\n\t}\n\t// Set the reason for failover (will use default value if not specified)\n\treason := c.String(FlagFailoverReason)\n\tif reason != \"\" {\n\t\tfailoverRequest.Reason = common.StringPtr(reason)\n\t}\n\n\t_, err = d.failoverDomain(ctx, failoverRequest)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t\t}\n\t\treturn commoncli.Problem(\"Operation FailoverDomain failed.\", err)\n\t}\n\tfmt.Printf(\"Domain %s successfully failed over.\\n\", domainName)\n\treturn nil\n}\n\n// FailoverDomains is used for managed failover all domains with domain data IsManagedByCadence=true\nfunc (d *domainCLIImpl) FailoverDomains(c *cli.Context) error {\n\t// ask user for confirmation\n\tprompt(\"You are trying to failover all managed domains, continue? y/N\")\n\t_, _, err := d.failoverDomains(c)\n\treturn err\n}\n\n// return succeed and failed domains for testing purpose\nfunc (d *domainCLIImpl) failoverDomains(c *cli.Context) ([]string, []string, error) {\n\ttargetCluster, err := getRequiredOption(c, FlagActiveClusterName)\n\tif err != nil {\n\t\treturn nil, nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tdomains, err := d.getAllDomains(c)\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"Failed to list domains: %w\", err)\n\t}\n\tshouldFailover := func(domain *types.DescribeDomainResponse) bool {\n\t\tisDomainNotActiveInTargetCluster := domain.ReplicationConfiguration.GetActiveClusterName() != targetCluster\n\t\treturn isDomainNotActiveInTargetCluster && isDomainFailoverManagedByCadence(domain)\n\t}\n\tvar succeedDomains []string\n\tvar failedDomains []string\n\tfor _, domain := range domains {\n\t\tif shouldFailover(domain) {\n\t\t\tdomainName := domain.GetDomainInfo().GetName()\n\t\t\terr := d.failover(c, domainName, targetCluster)\n\t\t\tif err != nil {\n\t\t\t\tprintError(getDeps(c).Output(), fmt.Sprintf(\"Failed failover domain: %s\\n\", domainName), err)\n\t\t\t\tfailedDomains = append(failedDomains, domainName)\n\t\t\t} else {\n\t\t\t\tfmt.Printf(\"Success failover domain: %s\\n\", domainName)\n\t\t\t\tsucceedDomains = append(succeedDomains, domainName)\n\t\t\t}\n\t\t}\n\t}\n\tfmt.Printf(\"Succeed %d: %v\\n\", len(succeedDomains), succeedDomains)\n\tfmt.Printf(\"Failed  %d: %v\\n\", len(failedDomains), failedDomains)\n\treturn succeedDomains, failedDomains, nil\n}\n\nfunc (d *domainCLIImpl) getAllDomains(c *cli.Context) ([]*types.DescribeDomainResponse, error) {\n\tvar res []*types.DescribeDomainResponse\n\tpagesize := int32(200)\n\tvar token []byte\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tfor more := true; more; more = len(token) > 0 {\n\t\tlistRequest := &types.ListDomainsRequest{\n\t\t\tPageSize:      pagesize,\n\t\t\tNextPageToken: token,\n\t\t}\n\t\tlistResp, err := d.listDomains(ctx, listRequest)\n\t\tif err != nil {\n\t\t\treturn nil, commoncli.Problem(\"Error when list domains info\", err)\n\t\t}\n\t\ttoken = listResp.GetNextPageToken()\n\t\tres = append(res, listResp.GetDomains()...)\n\t}\n\treturn res, nil\n}\n\nfunc isDomainFailoverManagedByCadence(domain *types.DescribeDomainResponse) bool {\n\tdomainData := domain.DomainInfo.GetData()\n\treturn strings.ToLower(strings.TrimSpace(domainData[constants.DomainDataKeyForManagedFailover])) == \"true\"\n}\n\nfunc (d *domainCLIImpl) failover(c *cli.Context, domainName string, targetCluster string) error {\n\tupdateRequest := &types.UpdateDomainRequest{\n\t\tName:              domainName,\n\t\tActiveClusterName: common.StringPtr(targetCluster),\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\t_, err = d.updateDomain(ctx, updateRequest)\n\treturn err\n}\n\nvar templateDomain = `Name: {{.Name}}\nUUID: {{.UUID}}\nDescription: {{.Description}}\nOwnerEmail: {{.OwnerEmail}}\nDomainData: {{.DomainData}}\nStatus: {{.Status}}\nRetentionInDays: {{.RetentionDays}}\nEmitMetrics: {{.EmitMetrics}}\nIsGlobal(XDC)Domain: {{.IsGlobal}}\nActiveClusterName: {{.ActiveCluster}}\nIsActiveActiveDomain: {{.IsActiveActiveDomain}}\nClusters: {{if .IsGlobal}}{{.Clusters}}{{else}}N/A, Not a global domain{{end}}\nHistoryArchivalStatus: {{.HistoryArchivalStatus}}{{with .HistoryArchivalURI}}\nHistoryArchivalURI: {{.}}{{end}}\nVisibilityArchivalStatus: {{.VisibilityArchivalStatus}}{{with .VisibilityArchivalURI}}\nVisibilityArchivalURI: {{.}}{{end}}\n{{with .BadBinaries}}Bad binaries to reset:\n{{table .}}{{end}}\n{{with .FailoverInfo}}Graceful failover info:\n{{table .}}{{end}}\n{{if .IsActiveActiveDomain}}To see active clusters by cluster attribute use --print-json.\n{{end}}`\n\n// DescribeDomain updates a domain\nfunc (d *domainCLIImpl) DescribeDomain(c *cli.Context) error {\n\tdomainName := c.String(FlagDomain)\n\tdomainID := c.String(FlagDomainID)\n\tprintJSON := c.Bool(FlagPrintJSON)\n\n\trequest := types.DescribeDomainRequest{}\n\tif domainID != \"\" {\n\t\trequest.UUID = &domainID\n\t}\n\tif domainName != \"\" {\n\t\trequest.Name = &domainName\n\t}\n\tif domainID == \"\" && domainName == \"\" {\n\t\treturn commoncli.Problem(\"At least domainID or domainName must be provided.\", nil)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\tresp, err := d.describeDomain(ctx, &request)\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); !ok {\n\t\t\treturn commoncli.Problem(\"Operation DescribeDomain failed.\", err)\n\t\t}\n\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t}\n\n\tif printJSON {\n\t\toutput, err := json.Marshal(resp)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to encode domain response into JSON.\", err)\n\t\t}\n\t\tfmt.Println(string(output))\n\t\treturn nil\n\t}\n\n\treturn Render(c, newDomainRow(resp), RenderOptions{\n\t\tDefaultTemplate: templateDomain,\n\t\tColor:           true,\n\t\tBorder:          true,\n\t\tPrintDateTime:   true,\n\t})\n}\n\ntype BadBinaryRow struct {\n\tChecksum  string    `header:\"Binary Checksum\"`\n\tOperator  string    `header:\"Operator\"`\n\tStartTime time.Time `header:\"Start Time\"`\n\tReason    string    `header:\"Reason\"`\n}\n\ntype FailoverInfoRow struct {\n\tFailoverVersion     int64     `header:\"Failover Version\"`\n\tStartTime           time.Time `header:\"Start Time\"`\n\tExpireTime          time.Time `header:\"Expire Time\"`\n\tCompletedShardCount int32     `header:\"Completed Shard Count\"`\n\tPendingShard        []int32   `header:\"Pending Shard\"`\n}\n\ntype FailoverHistoryRow struct {\n\tEventID     string    `header:\"Event ID\"`\n\tCreatedTime time.Time `header:\"Created Time\"`\n\tFromCluster string    `header:\"From Cluster\"`\n\tToCluster   string    `header:\"To Cluster\"`\n\tAttribute   string    `header:\"Cluster Attribute\"`\n}\n\ntype DomainRow struct {\n\tName                     string `header:\"Name\"`\n\tUUID                     string `header:\"UUID\"`\n\tDescription              string\n\tOwnerEmail               string\n\tDomainData               map[string]string  `header:\"Domain Data\"`\n\tStatus                   types.DomainStatus `header:\"Status\"`\n\tIsGlobal                 bool               `header:\"Is Global Domain\"`\n\tActiveCluster            string             `header:\"Active Cluster\"`\n\tClusters                 []string           `header:\"Clusters\"`\n\tRetentionDays            int32              `header:\"Retention Days\"`\n\tEmitMetrics              bool\n\tHistoryArchivalStatus    types.ArchivalStatus `header:\"History Archival Status\"`\n\tHistoryArchivalURI       string               `header:\"History Archival URI\"`\n\tVisibilityArchivalStatus types.ArchivalStatus `header:\"Visibility Archival Status\"`\n\tVisibilityArchivalURI    string               `header:\"Visibility Archival URI\"`\n\tBadBinaries              []BadBinaryRow\n\tFailoverInfo             *FailoverInfoRow\n\tLongRunningWorkFlowNum   *int\n\tIsActiveActiveDomain     bool\n}\n\ntype DomainMigrationRow struct {\n\tValidationCheck   string `header:\"Validation Checker\"`\n\tValidationResult  bool   `header:\"Validation Result\"`\n\tValidationDetails ValidationDetails\n}\n\ntype ValidationDetails struct {\n\tCurrentDomainRow            *types.DescribeDomainResponse\n\tNewDomainRow                *types.DescribeDomainResponse\n\tMismatchedDomainMetaData    string\n\tLongRunningWorkFlowNum      *int\n\tMismatchedDynamicConfig     []MismatchedDynamicConfig\n\tMissingCurrSearchAttributes []string\n\tMissingNewSearchAttributes  []string\n}\n\ntype MismatchedDynamicConfig struct {\n\tKey        dynamicproperties.Key\n\tCurrValues []*types.DynamicConfigValue\n\tNewValues  []*types.DynamicConfigValue\n}\n\nfunc newDomainRow(domain *types.DescribeDomainResponse) DomainRow {\n\treturn DomainRow{\n\t\tName:                     domain.DomainInfo.Name,\n\t\tUUID:                     domain.DomainInfo.UUID,\n\t\tDescription:              domain.DomainInfo.Description,\n\t\tOwnerEmail:               domain.DomainInfo.OwnerEmail,\n\t\tDomainData:               domain.DomainInfo.GetData(),\n\t\tStatus:                   domain.DomainInfo.GetStatus(),\n\t\tIsGlobal:                 domain.IsGlobalDomain,\n\t\tActiveCluster:            domain.ReplicationConfiguration.GetActiveClusterName(),\n\t\tClusters:                 clustersToStrings(domain.ReplicationConfiguration.GetClusters()),\n\t\tRetentionDays:            domain.Configuration.GetWorkflowExecutionRetentionPeriodInDays(),\n\t\tEmitMetrics:              domain.Configuration.GetEmitMetric(),\n\t\tHistoryArchivalStatus:    domain.Configuration.GetHistoryArchivalStatus(),\n\t\tHistoryArchivalURI:       domain.Configuration.GetHistoryArchivalURI(),\n\t\tVisibilityArchivalStatus: domain.Configuration.GetVisibilityArchivalStatus(),\n\t\tVisibilityArchivalURI:    domain.Configuration.GetVisibilityArchivalURI(),\n\t\tBadBinaries:              newBadBinaryRows(domain.Configuration.BadBinaries),\n\t\tFailoverInfo:             newFailoverInfoRow(domain.FailoverInfo),\n\t\tIsActiveActiveDomain:     domain.ReplicationConfiguration.IsActiveActive(),\n\t}\n}\n\nfunc newFailoverInfoRow(info *types.FailoverInfo) *FailoverInfoRow {\n\tif info == nil {\n\t\treturn nil\n\t}\n\treturn &FailoverInfoRow{\n\t\tFailoverVersion:     info.GetFailoverVersion(),\n\t\tStartTime:           time.Unix(0, info.GetFailoverStartTimestamp()),\n\t\tExpireTime:          time.Unix(0, info.GetFailoverExpireTimestamp()),\n\t\tCompletedShardCount: info.GetCompletedShardCount(),\n\t\tPendingShard:        info.GetPendingShards(),\n\t}\n}\n\nfunc newBadBinaryRows(bb *types.BadBinaries) []BadBinaryRow {\n\tif bb == nil {\n\t\treturn nil\n\t}\n\trows := []BadBinaryRow{}\n\tfor cs, bin := range bb.Binaries {\n\t\trows = append(rows, BadBinaryRow{\n\t\t\tChecksum:  cs,\n\t\t\tOperator:  bin.GetOperator(),\n\t\t\tStartTime: time.Unix(0, bin.GetCreatedTimeNano()),\n\t\t\tReason:    bin.GetReason(),\n\t\t})\n\t}\n\treturn rows\n}\n\nfunc renderFailoverHistoryTable(response *types.ListFailoverHistoryResponse) {\n\trenderFailoverHistoryTableToWriter(os.Stdout, response)\n}\n\nfunc renderFailoverHistoryTableToWriter(writer interface{ Write([]byte) (int, error) }, response *types.ListFailoverHistoryResponse) {\n\ttable := tablewriter.NewWriter(writer)\n\ttable.SetRowLine(true)\n\ttable.SetBorder(true)\n\ttable.SetAutoWrapText(false)\n\ttable.SetAutoFormatHeaders(true)\n\ttable.SetAutoMergeCells(true)\n\ttable.SetHeaderAlignment(tablewriter.ALIGN_LEFT)\n\ttable.SetAlignment(tablewriter.ALIGN_LEFT)\n\n\tdisplayClusterAttributeCol := false\n\tvar columnsToRender [][]string\n\n\tfor _, event := range response.GetFailoverEvents() {\n\t\teventID := event.GetID()\n\t\tcreatedTime := time.Unix(0, event.GetCreatedTime()).Format(time.RFC3339)\n\t\tclusterFailovers := event.GetClusterFailovers()\n\n\t\tif len(clusterFailovers) == 0 {\n\t\t\t// If no cluster failovers, show event info only\n\t\t\ttable.Append([]string{eventID, createdTime, \"-\", \"-\", \"-\"})\n\t\t\tcontinue\n\t\t}\n\n\t\tfor _, failover := range clusterFailovers {\n\t\t\tfromCluster := \"\"\n\t\t\tif fromInfo := failover.GetFromCluster(); fromInfo != nil {\n\t\t\t\tfromCluster = fromInfo.ActiveClusterName\n\t\t\t}\n\t\t\ttoCluster := \"\"\n\t\t\tif toInfo := failover.GetToCluster(); toInfo != nil {\n\t\t\t\ttoCluster = toInfo.ActiveClusterName\n\t\t\t}\n\t\t\tdomainFailover := fmt.Sprintf(\"%s -> %s\", fromCluster, toCluster)\n\n\t\t\tattribute := \"\"\n\t\t\tif attr := failover.GetClusterAttribute(); attr != nil {\n\t\t\t\tif attr.Scope != \"\" && attr.Name != \"\" {\n\t\t\t\t\tattribute = fmt.Sprintf(\"%s.%s\", attr.Scope, attr.Name)\n\t\t\t\t\tdisplayClusterAttributeCol = true\n\t\t\t\t}\n\t\t\t}\n\t\t\tif displayClusterAttributeCol {\n\t\t\t\tcolumnsToRender = append(columnsToRender, []string{eventID, createdTime, domainFailover, attribute})\n\t\t\t} else {\n\t\t\t\tcolumnsToRender = append(columnsToRender, []string{eventID, createdTime, domainFailover})\n\t\t\t}\n\t\t}\n\t}\n\ttable.AppendBulk(columnsToRender)\n\n\tif displayClusterAttributeCol {\n\t\ttable.SetHeader([]string{\"Event ID\", \"Failover Timestamp\", \"Failover\", \"Cluster Attribute\"})\n\t} else {\n\t\ttable.SetHeader([]string{\"Event ID\", \"Failover Timestamp\", \"Failover\"})\n\t}\n\n\ttable.Render()\n}\n\nfunc domainTableOptions(c *cli.Context) RenderOptions {\n\tprintAll := c.Bool(FlagAll)\n\tprintFull := c.Bool(FlagPrintFullyDetail)\n\n\treturn RenderOptions{\n\t\tDefaultTemplate: templateTable,\n\t\tColor:           true,\n\t\tOptionalColumns: map[string]bool{\n\t\t\t\"Status\":                     printAll || printFull,\n\t\t\t\"Clusters\":                   printFull,\n\t\t\t\"Retention Days\":             printFull,\n\t\t\t\"History Archival Status\":    printFull,\n\t\t\t\"History Archival URI\":       printFull,\n\t\t\t\"Visibility Archival Status\": printFull,\n\t\t\t\"Visibility Archival URI\":    printFull,\n\t\t},\n\t}\n}\n\nfunc (d *domainCLIImpl) ListDomains(c *cli.Context) error {\n\toutput := getDeps(c).Output()\n\n\tpageSize := c.Int(FlagPageSize)\n\tprefix := c.String(FlagPrefix)\n\tprintAll := c.Bool(FlagAll)\n\tprintDeprecated := c.Bool(FlagDeprecated)\n\tprintJSON := c.Bool(FlagPrintJSON)\n\n\tif printAll && printDeprecated {\n\t\treturn commoncli.Problem(fmt.Sprintf(\"Cannot specify %s and %s flags at the same time.\", FlagAll, FlagDeprecated), nil)\n\t}\n\n\tdomains, err := d.getAllDomains(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar filteredDomains []*types.DescribeDomainResponse\n\n\t// Only list domains that are matching to the prefix if prefix is provided\n\tif len(prefix) > 0 {\n\t\tvar prefixDomains []*types.DescribeDomainResponse\n\t\tfor _, domain := range domains {\n\t\t\tif strings.Index(domain.DomainInfo.Name, prefix) == 0 {\n\t\t\t\tprefixDomains = append(prefixDomains, domain)\n\t\t\t}\n\t\t}\n\t\tdomains = prefixDomains\n\t}\n\n\tif printAll {\n\t\tfilteredDomains = domains\n\t} else {\n\t\tfilteredDomains = make([]*types.DescribeDomainResponse, 0, len(domains))\n\t\tfor _, domain := range domains {\n\t\t\tif printDeprecated && *domain.DomainInfo.Status == types.DomainStatusDeprecated {\n\t\t\t\tfilteredDomains = append(filteredDomains, domain)\n\t\t\t} else if !printDeprecated && *domain.DomainInfo.Status == types.DomainStatusRegistered {\n\t\t\t\tfilteredDomains = append(filteredDomains, domain)\n\t\t\t}\n\t\t}\n\t}\n\n\tif printJSON {\n\t\toutput, err := json.Marshal(filteredDomains)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to encode domain results into JSON.\", err)\n\t\t}\n\t\tfmt.Println(string(output))\n\t\treturn nil\n\t}\n\n\ttable := make([]DomainRow, 0, pageSize)\n\n\tcurrentPageSize := 0\n\tfor i, domain := range filteredDomains {\n\t\ttable = append(table, newDomainRow(domain))\n\t\tcurrentPageSize++\n\n\t\tif currentPageSize != pageSize {\n\t\t\tcontinue\n\t\t}\n\n\t\t// page is full\n\t\tif err := Render(c, table, domainTableOptions(c)); err != nil {\n\t\t\treturn fmt.Errorf(\"failed to render domain list: %w\", err)\n\t\t}\n\t\tif i == len(domains)-1 || !showNextPage(output) {\n\t\t\treturn nil\n\t\t}\n\t\ttable = make([]DomainRow, 0, pageSize)\n\t\tcurrentPageSize = 0\n\t}\n\n\treturn Render(c, table, domainTableOptions(c))\n}\n\nfunc (d *domainCLIImpl) listDomains(\n\tctx context.Context,\n\trequest *types.ListDomainsRequest,\n) (*types.ListDomainsResponse, error) {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.ListDomains(ctx, request)\n\t}\n\n\treturn d.domainHandler.ListDomains(ctx, request)\n}\n\nfunc (d *domainCLIImpl) registerDomain(\n\tctx context.Context,\n\trequest *types.RegisterDomainRequest,\n) error {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.RegisterDomain(ctx, request)\n\t}\n\n\treturn d.domainHandler.RegisterDomain(ctx, request)\n}\n\nfunc (d *domainCLIImpl) updateDomain(\n\tctx context.Context,\n\trequest *types.UpdateDomainRequest,\n) (*types.UpdateDomainResponse, error) {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.UpdateDomain(ctx, request)\n\t}\n\n\treturn d.domainHandler.UpdateDomain(ctx, request)\n}\n\nfunc (d *domainCLIImpl) deleteDomain(\n\tctx context.Context,\n\trequest *types.DeleteDomainRequest,\n) error {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.DeleteDomain(ctx, request)\n\t}\n\n\treturn d.domainHandler.DeleteDomain(ctx, request)\n}\n\n// ListFailoverHistory lists the failover history for a domain\nfunc (d *domainCLIImpl) ListFailoverHistory(c *cli.Context) error {\n\tdomainName, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\n\t// Get domain ID by describing the domain first\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t}\n\n\tdescribeResp, err := d.describeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: common.StringPtr(domainName),\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\treturn commoncli.Problem(fmt.Sprintf(\"Domain %s does not exist.\", domainName), err)\n\t\t}\n\t\treturn commoncli.Problem(\"Failed to describe domain.\", err)\n\t}\n\n\tdomainID := describeResp.DomainInfo.GetUUID()\n\n\tlimit := 10\n\tif c.Bool(FlagAll) {\n\t\tlimit = -1\n\t}\n\n\tprintJSON := c.Bool(FlagPrintJSON)\n\tallResponses := &types.ListFailoverHistoryResponse{}\n\n\tvar nextPageToken []byte\n\n\tfor {\n\t\trequest := &types.ListFailoverHistoryRequest{\n\t\t\tFilters: &types.ListFailoverHistoryRequestFilters{\n\t\t\t\tDomainID: domainID,\n\t\t\t},\n\t\t\tPagination: &types.PaginationOptions{\n\t\t\t\tPageSize:      common.Int32Ptr(int32(10)),\n\t\t\t\tNextPageToken: nextPageToken,\n\t\t\t},\n\t\t}\n\n\t\tctx, cancel, err = newContext(c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error in creating context: \", err)\n\t\t}\n\n\t\tresp, err := d.frontendClient.ListFailoverHistory(ctx, request)\n\t\tcancel()\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to list failover history.\", err)\n\t\t}\n\t\tallResponses.FailoverEvents = append(allResponses.FailoverEvents, resp.FailoverEvents...)\n\t\tnextPageToken = resp.NextPageToken\n\t\tif len(nextPageToken) == 0 || nextPageToken == nil {\n\t\t\tbreak\n\t\t}\n\t\tif limit > 0 && len(allResponses.FailoverEvents) >= int(limit) {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif printJSON {\n\t\toutput, err := json.Marshal(allResponses)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to encode failover history into JSON.\", err)\n\t\t}\n\t\tfmt.Println(string(output))\n\t\treturn nil\n\t}\n\n\tif len(allResponses.GetFailoverEvents()) == 0 {\n\t\tfmt.Println(\"No failover history found for domain:\", domainName)\n\t\treturn nil\n\t}\n\n\trenderFailoverHistoryTable(allResponses)\n\treturn nil\n}\n\nfunc (d *domainCLIImpl) describeDomain(\n\tctx context.Context,\n\trequest *types.DescribeDomainRequest,\n) (*types.DescribeDomainResponse, error) {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.DescribeDomain(ctx, request)\n\t}\n\n\treturn d.domainHandler.DescribeDomain(ctx, request)\n}\n\nfunc (d *domainCLIImpl) failoverDomain(\n\tctx context.Context,\n\trequest *types.FailoverDomainRequest,\n) (*types.FailoverDomainResponse, error) {\n\n\tif d.frontendClient != nil {\n\t\treturn d.frontendClient.FailoverDomain(ctx, request)\n\t}\n\n\treturn d.domainHandler.FailoverDomain(ctx, request)\n}\n\nfunc archivalStatus(c *cli.Context, statusFlagName string) (*types.ArchivalStatus, error) {\n\tif c.IsSet(statusFlagName) {\n\t\tswitch c.String(statusFlagName) {\n\t\tcase \"disabled\":\n\t\t\treturn types.ArchivalStatusDisabled.Ptr(), nil\n\t\tcase \"enabled\":\n\t\t\treturn types.ArchivalStatusEnabled.Ptr(), nil\n\t\tdefault:\n\t\t\treturn nil, commoncli.Problem(fmt.Sprintf(\"Option %s format is invalid.\", statusFlagName), errors.New(\"invalid status, valid values are \\\"disabled\\\" and \\\"enabled\\\"\"))\n\t\t}\n\t}\n\treturn nil, nil\n}\n\nfunc clustersToStrings(clusters []*types.ClusterReplicationConfiguration) []string {\n\tvar res []string\n\tfor _, cluster := range clusters {\n\t\tres = append(res, cluster.GetClusterName())\n\t}\n\treturn res\n}\n\nfunc parseActiveClustersByClusterAttributeFromJSON(jsonStr string) (types.ActiveClusters, error) {\n\tac := types.ActiveClusters{}\n\tif err := json.Unmarshal([]byte(jsonStr), &ac); err != nil {\n\t\treturn types.ActiveClusters{}, fmt.Errorf(\"couldn't parse the JSON: %w. Got %s\", err, jsonStr)\n\t}\n\treturn ac, nil\n}\n\nfunc parseActiveClustersByClusterAttribute(clusters string) (types.ActiveClusters, error) {\n\tsplit := regexp.MustCompile(`(?P<attribute>[a-zA-Z0-9_-]+).(?P<scope>[a-zA-Z0-9_-]+):(?P<name>[a-zA-Z0-9_-]+)`)\n\tmatches := split.FindAllStringSubmatch(clusters, -1)\n\tif len(matches) == 0 {\n\t\treturn types.ActiveClusters{}, fmt.Errorf(\"option %s format is invalid. Expected format is 'region.dca:dev2_dca,region.phx:dev2_phx'\", FlagActiveClusters)\n\t}\n\n\tout := types.ActiveClusters{\n\t\tAttributeScopes: map[string]types.ClusterAttributeScope{},\n\t}\n\n\tfor _, match := range matches {\n\t\tif len(match) != 4 {\n\t\t\treturn types.ActiveClusters{}, fmt.Errorf(\"option %s format is invalid. Expected format is 'region.dca:dev2_dca,region.phx:dev2_phx'\", FlagActiveClusters)\n\t\t}\n\t\tattribute := match[1]\n\t\tscope := match[2]\n\t\tname := match[3]\n\n\t\texisting, ok := out.AttributeScopes[attribute]\n\t\tif !ok {\n\t\t\tout.AttributeScopes[attribute] = types.ClusterAttributeScope{\n\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\tscope: {ActiveClusterName: name},\n\t\t\t\t},\n\t\t\t}\n\t\t} else {\n\t\t\tif _, ok := existing.ClusterAttributes[scope]; ok {\n\t\t\t\treturn types.ActiveClusters{}, fmt.Errorf(\"option active_clusters format is invalid. the key %q was duplicated. This can only map to a single active cluster\", scope)\n\t\t\t}\n\t\t\texisting.ClusterAttributes[scope] = types.ActiveClusterInfo{ActiveClusterName: name}\n\t\t\tout.AttributeScopes[attribute] = existing\n\t\t}\n\n\t\tout.AttributeScopes[attribute].ClusterAttributes[scope] = types.ActiveClusterInfo{ActiveClusterName: name}\n\t}\n\n\treturn out, nil\n}\n"
  },
  {
    "path": "tools/cli/domain_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc (s *cliAppSuite) TestDomainRegister() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\t\"local\",\n\t\t\t\"cadence --do test-domain domain register --global_domain false\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\t\tIsGlobalDomain:                         false,\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"global\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"active-active domain\",\n\t\t\t\"cadence --do test-domain domain register --active_clusters region.region1:cluster1,region.region2:cluster2\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"active-active domain with invalid active clusters by region\",\n\t\t\t\"cadence --do test-domain domain register --active_clusters region1=cluster1\",\n\t\t\t\"option active_clusters format is invalid. Expected format is 'region.dca:dev2_dca,region.phx:dev2_phx\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"domain with other options\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true --retention 5 --desc description --active_cluster c1 --clusters c1,c2 --domain_data key1=value1,key2=value2\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 5,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t\tDescription:                            \"description\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t\t\t\"key2\": \"value2\",\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"c1\"}, {ClusterName: \"c2\"},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"domain with other options and clusters mentioned as multiple options\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true --retention 5 --desc description --active_cluster c1 --cl c1 --cl c2 --domain_data key1=value1,key2=value2 --cl c3\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 5,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t\tDescription:                            \"description\",\n\t\t\t\t\tData: map[string]string{\n\t\t\t\t\t\t\"key1\": \"value1\",\n\t\t\t\t\t\t\"key2\": \"value2\",\n\t\t\t\t\t},\n\t\t\t\t\tActiveClusterName: \"c1\",\n\t\t\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t\t\t{ClusterName: \"c1\"}, {ClusterName: \"c2\"}, {ClusterName: \"c3\"},\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"domain exists\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true\",\n\t\t\t\"Domain test-domain already registered\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t}).Return(&types.DomainAlreadyExistsError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"failed\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true\",\n\t\t\t\"Register Domain operation failed\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RegisterDomain(gomock.Any(), &types.RegisterDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t\t\t\tIsGlobalDomain:                         true,\n\t\t\t\t}).Return(&types.BadRequestError{Message: \"fake error\"})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"missing flag\",\n\t\t\t\"cadence domain register\",\n\t\t\t\"option domain is required\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"fail on extra arguments at end\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true --retention 5 --desc description --active_cluster c1 --clusters c1,c2 unused_arg\",\n\t\t\t\"Domain commands cannot have arguments: <unused_arg>\\nClusters are now specified as --clusters c1,c2 see help for more info\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"fail on extra arguments at in command\",\n\t\t\t\"cadence --do test-domain domain register --global_domain true --retention 5 --desc description --active_cluster c1 unused_arg --clusters c1,c2\",\n\t\t\t\"Domain commands cannot have arguments: <unused_arg --clusters c1,c2>\\nClusters are now specified as --clusters c1,c2 see help for more info\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"invalid global domain flag\",\n\t\t\t\"cadence --do test-domain domain register --global_domain invalid\",\n\t\t\t\"format is invalid\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"invalid history archival status\",\n\t\t\t\"cadence --do test-domain domain register --global_domain false --history_archival_status invalid\",\n\t\t\t\"failed to parse\",\n\t\t\tnil,\n\t\t},\n\t\t{\n\t\t\t\"invalid visibility archival status\",\n\t\t\t\"cadence --do test-domain domain register --global_domain false --visibility_archival_status invalid\",\n\t\t\t\"failed to parse\",\n\t\t\tnil,\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestDomainUpdate() {\n\tdescribeResponse := &types.DescribeDomainResponse{\n\t\tDomainInfo: &types.DomainInfo{\n\t\t\tName:        \"test-domain\",\n\t\t\tDescription: \"a test domain\",\n\t\t\tOwnerEmail:  \"test@cadence.io\",\n\t\t\tData: map[string]string{\n\t\t\t\t\"key1\": \"value1\",\n\t\t\t},\n\t\t},\n\t\tConfiguration: &types.DomainConfiguration{\n\t\t\tWorkflowExecutionRetentionPeriodInDays: 3,\n\t\t},\n\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{\n\t\t\tActiveClusterName: \"c1\",\n\t\t\tClusters: []*types.ClusterReplicationConfiguration{\n\t\t\t\t{\n\t\t\t\t\tClusterName: \"c1\",\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tClusterName: \"c2\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\ttestCases := []testcase{\n\t\t{\n\t\t\t\"update nothing\",\n\t\t\t\"cadence --do test-domain domain update\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: common.StringPtr(\"test-domain\"),\n\t\t\t\t}).Return(describeResponse, nil)\n\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tDescription:                            common.StringPtr(\"a test domain\"),\n\t\t\t\t\tOwnerEmail:                             common.StringPtr(\"test@cadence.io\"),\n\t\t\t\t\tData:                                   nil,\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(3),\n\t\t\t\t\tEmitMetric:                             common.BoolPtr(false),\n\t\t\t\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\t\t\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\t\t\t\tActiveClusterName:                      nil,\n\t\t\t\t\tClusters:                               nil,\n\t\t\t\t}).Return(&types.UpdateDomainResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"update description\",\n\t\t\t\"cadence --do test-domain domain update --desc new-description\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: common.StringPtr(\"test-domain\"),\n\t\t\t\t}).Return(describeResponse, nil)\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tDescription:                            common.StringPtr(\"new-description\"),\n\t\t\t\t\tOwnerEmail:                             common.StringPtr(\"test@cadence.io\"),\n\t\t\t\t\tData:                                   nil,\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(3),\n\t\t\t\t\tEmitMetric:                             common.BoolPtr(false),\n\t\t\t\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\t\t\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\t\t\t\tActiveClusterName:                      nil,\n\t\t\t\t\tClusters:                               nil,\n\t\t\t\t}).Return(&types.UpdateDomainResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"active-passive domain failover\",\n\t\t\t\"cadence --do test-domain domain update --ac c2\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName:              \"test-domain\",\n\t\t\t\t\tActiveClusterName: common.StringPtr(\"c2\"),\n\t\t\t\t}).Return(&types.UpdateDomainResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"active-passive domain graceful failover\",\n\t\t\t\"cadence --do test-domain domain update --ac c2 --failover_type grace --failover_timeout_seconds 10\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName:                     \"test-domain\",\n\t\t\t\t\tActiveClusterName:        common.StringPtr(\"c2\"),\n\t\t\t\t\tFailoverTimeoutInSeconds: common.Int32Ptr(10),\n\t\t\t\t}).Return(&types.UpdateDomainResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"active-active domain failover\",\n\t\t\t\"cadence --do test-domain domain update --active_clusters region.region1:c1,region.region2:c2\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName: \"test-domain\",\n\t\t\t\t\tActiveClusters: &types.ActiveClusters{\n\t\t\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\t\t\"region\": {\n\t\t\t\t\t\t\t\tClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\t\t\t\"region1\": {ActiveClusterName: \"c1\"},\n\t\t\t\t\t\t\t\t\t\"region2\": {ActiveClusterName: \"c2\"},\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}).Return(&types.UpdateDomainResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"domain not exist\",\n\t\t\t\"cadence --do test-domain domain update --desc new-description\",\n\t\t\t\"does not exist\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: common.StringPtr(\"test-domain\"),\n\t\t\t\t}).Return(nil, &types.EntityNotExistsError{})\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"describe failure\",\n\t\t\t\"cadence --do test-domain domain update --desc new-description\",\n\t\t\t\"describe error\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: common.StringPtr(\"test-domain\"),\n\t\t\t\t}).Return(nil, fmt.Errorf(\"describe error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"update failure\",\n\t\t\t\"cadence --do test-domain domain update --desc new-description\",\n\t\t\t\"update error\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: common.StringPtr(\"test-domain\"),\n\t\t\t\t}).Return(describeResponse, nil)\n\t\t\t\ts.serverFrontendClient.EXPECT().UpdateDomain(gomock.Any(), &types.UpdateDomainRequest{\n\t\t\t\t\tName:                                   \"test-domain\",\n\t\t\t\t\tDescription:                            common.StringPtr(\"new-description\"),\n\t\t\t\t\tOwnerEmail:                             common.StringPtr(\"test@cadence.io\"),\n\t\t\t\t\tData:                                   nil,\n\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: common.Int32Ptr(3),\n\t\t\t\t\tEmitMetric:                             common.BoolPtr(false),\n\t\t\t\t\tHistoryArchivalURI:                     common.StringPtr(\"\"),\n\t\t\t\t\tVisibilityArchivalURI:                  common.StringPtr(\"\"),\n\t\t\t\t\tActiveClusterName:                      nil,\n\t\t\t\t\tClusters:                               nil,\n\t\t\t\t}).Return(nil, fmt.Errorf(\"update error\"))\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestListDomains() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\t\"list domains by default\",\n\t\t\t\"cadence admin domain list\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\ts.serverFrontendClient.EXPECT().ListDomains(gomock.Any(), gomock.Any()).Return(&types.ListDomainsResponse{\n\t\t\t\t\tDomains: []*types.DescribeDomainResponse{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tDomainInfo: &types.DomainInfo{\n\t\t\t\t\t\t\t\tName:   \"test-domain\",\n\t\t\t\t\t\t\t\tStatus: types.DomainStatusRegistered.Ptr(),\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tReplicationConfiguration: &types.DomainReplicationConfiguration{},\n\t\t\t\t\t\t\tConfiguration:            &types.DomainConfiguration{},\n\t\t\t\t\t\t\tFailoverInfo:             &types.FailoverInfo{},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc TestParseActiveClustersByClusterAttribute(t *testing.T) {\n\n\ttestCases := map[string]struct {\n\t\tclusters      string\n\t\texpected      types.ActiveClusters\n\t\texpectedError error\n\t}{\n\t\t\"valid active clusters by cluster attribute\": {\n\t\t\tclusters: \"region.newyork:cluster0,region.manilla:cluster1\",\n\t\t\texpected: types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"newyork\": {ActiveClusterName: \"cluster0\"},\n\t\t\t\t\t\t\"manilla\": {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"valid active clusters by cluster attribute with multiple scopes\": {\n\t\t\tclusters: \"region.newyork:cluster0,location.brussels:cluster2,region.madrid:cluster1\",\n\t\t\texpected: types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"newyork\": {ActiveClusterName: \"cluster0\"},\n\t\t\t\t\t\t\"madrid\":  {ActiveClusterName: \"cluster1\"},\n\t\t\t\t\t}},\n\t\t\t\t\t\"location\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"brussels\": {ActiveClusterName: \"cluster2\"},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"clusters apparently can contain dashes\": {\n\t\t\tclusters: \"region.newyork:cluster-0-us-east-1\",\n\t\t\texpected: types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"newyork\": {ActiveClusterName: \"cluster-0-us-east-1\"},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"other things can use dashes too\": {\n\t\t\tclusters: \"region-us-east1.new-york:cluster-0-us-east-1\",\n\t\t\texpected: types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region-us-east1\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"new-york\": {ActiveClusterName: \"cluster-0-us-east-1\"},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"duplicate keys consistutes an error in parsing and shouldn't be allowed\": {\n\t\t\tclusters:      \"region.newyork:cluster0,region.newyork:cluster1\",\n\t\t\texpectedError: fmt.Errorf(`option active_clusters format is invalid. the key \"newyork\" was duplicated. This can only map to a single active cluster`),\n\t\t},\n\t\t\"Some invalid input\": {\n\t\t\tclusters:      \"bad-data\",\n\t\t\texpectedError: fmt.Errorf(\"option active_clusters format is invalid. Expected format is 'region.dca:dev2_dca,region.phx:dev2_phx'\"),\n\t\t},\n\t\t\"empty input\": {\n\t\t\tclusters:      \"\",\n\t\t\texpectedError: fmt.Errorf(\"option active_clusters format is invalid. Expected format is 'region.dca:dev2_dca,region.phx:dev2_phx'\"),\n\t\t},\n\t}\n\n\tfor name, td := range testCases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactiveClusters, err := parseActiveClustersByClusterAttribute(td.clusters)\n\t\t\tassert.Equal(t, td.expected, activeClusters)\n\t\t\tassert.Equal(t, td.expectedError, err)\n\t\t})\n\t}\n}\n\nfunc TestParseActiveClustersByClusterAttributeFromJSON(t *testing.T) {\n\ttestCases := map[string]struct {\n\t\tjsonStr       string\n\t\texpected      types.ActiveClusters\n\t\texpectedError bool\n\t}{\n\t\t\"valid JSON\": {\n\t\t\tjsonStr: `{\"attributeScopes\":{\"region-us-east1\":{\"clusterAttributes\":{\"new-york\":{\"activeClusterName\":\"cluster-0-us-east-1\",\"failoverVersion\":0}}}}}`,\n\t\t\texpected: types.ActiveClusters{\n\t\t\t\tAttributeScopes: map[string]types.ClusterAttributeScope{\n\t\t\t\t\t\"region-us-east1\": {ClusterAttributes: map[string]types.ActiveClusterInfo{\n\t\t\t\t\t\t\"new-york\": {ActiveClusterName: \"cluster-0-us-east-1\", FailoverVersion: 0},\n\t\t\t\t\t}},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t\"making nerds upset with brackets\": {\n\t\t\tjsonStr:       `{\"`,\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor name, td := range testCases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tactiveClusters, err := parseActiveClustersByClusterAttributeFromJSON(td.jsonStr)\n\t\t\tassert.Equal(t, td.expected, activeClusters)\n\t\t\tif td.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRenderFailoverHistoryTable(t *testing.T) {\n\ttestCases := map[string]struct {\n\t\tresponse       *types.ListFailoverHistoryResponse\n\t\texpectedOutput []string // Substrings that should be present in output\n\t}{\n\t\t\"single failover event with one cluster failover\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-1\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000), // 2023-11-14T22:13:20Z\n\t\t\t\t\t\tFailoverType: func() *types.FailoverType {\n\t\t\t\t\t\t\tt := types.FailoverTypeGraceful\n\t\t\t\t\t\t\treturn &t\n\t\t\t\t\t\t}(),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-a\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-b\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\t\t\tName:  \"us-west\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-1\",\n\t\t\t\t\"2023-11-14\", // Just check date, not full timestamp with timezone\n\t\t\t\t\"cluster-a -> cluster-b\",\n\t\t\t\t\"region.us-west\",\n\t\t\t\t\"FAILOVER TIMESTAMP\", // New column header (uppercase)\n\t\t\t\t\"CLUSTER ATTRIBUTE\",  // Conditional column header (uppercase)\n\t\t\t},\n\t\t},\n\t\t\"single failover event with multiple cluster failovers\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-2\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tFailoverType: func() *types.FailoverType {\n\t\t\t\t\t\t\tt := types.FailoverTypeForce\n\t\t\t\t\t\t\treturn &t\n\t\t\t\t\t\t}(),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-a\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-b\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\t\t\tName:  \"us-west\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-c\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-d\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"zone\",\n\t\t\t\t\t\t\t\t\tName:  \"az-1\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-2\",\n\t\t\t\t\"cluster-a -> cluster-b\",\n\t\t\t\t\"region.us-west\",\n\t\t\t\t\"cluster-c -> cluster-d\",\n\t\t\t\t\"zone.az-1\",\n\t\t\t},\n\t\t},\n\t\t\"failover event with nil ActiveClusterName\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-3\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tFailoverType: func() *types.FailoverType {\n\t\t\t\t\t\t\tt := types.FailoverTypeGraceful\n\t\t\t\t\t\t\treturn &t\n\t\t\t\t\t\t}(),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\t\t\tName:  \"us-east\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-3\",\n\t\t\t\t\" -> \",\n\t\t\t\t\"region.us-east\",\n\t\t\t},\n\t\t},\n\t\t\"failover event with nil ClusterAttribute\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-4\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tFailoverType: func() *types.FailoverType {\n\t\t\t\t\t\t\tt := types.FailoverTypeForce\n\t\t\t\t\t\t\treturn &t\n\t\t\t\t\t\t}(),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster:      &types.ActiveClusterInfo{ActiveClusterName: \"cluster-x\"},\n\t\t\t\t\t\t\t\tToCluster:        &types.ActiveClusterInfo{ActiveClusterName: \"cluster-y\"},\n\t\t\t\t\t\t\t\tClusterAttribute: nil,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-4\",\n\t\t\t\t\"cluster-x -> cluster-y\",\n\t\t\t\t\"FAILOVER\", // Should have 3-column table (no Cluster Attribute column, uppercase)\n\t\t\t},\n\t\t},\n\t\t\"failover event with empty ClusterAttribute scope and name\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-5\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tFailoverType: func() *types.FailoverType {\n\t\t\t\t\t\t\tt := types.FailoverTypeGraceful\n\t\t\t\t\t\t\treturn &t\n\t\t\t\t\t\t}(),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-m\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-n\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"\",\n\t\t\t\t\t\t\t\t\tName:  \"\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-5\",\n\t\t\t\t\"cluster-m -> cluster-n\",\n\t\t\t},\n\t\t},\n\t\t\"failover event with no cluster failovers\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:               common.StringPtr(\"event-6\"),\n\t\t\t\t\t\tCreatedTime:      common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-6\",\n\t\t\t\t\"2023-11-14\", // Just check date, not full timestamp with timezone\n\t\t\t},\n\t\t},\n\t\t\"failover event with nil FromCluster and ToCluster\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-7\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: nil,\n\t\t\t\t\t\t\t\tToCluster:   nil,\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\t\t\t\tName:  \"us-central\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-7\",\n\t\t\t\t\" -> \",\n\t\t\t\t\"region.us-central\",\n\t\t\t},\n\t\t},\n\t\t\"multiple failover events\": {\n\t\t\tresponse: &types.ListFailoverHistoryResponse{\n\t\t\t\tFailoverEvents: []*types.FailoverEvent{\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-8\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700000000000000000),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-1\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-2\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"dc\",\n\t\t\t\t\t\t\t\t\tName:  \"dc1\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\t{\n\t\t\t\t\t\tID:          common.StringPtr(\"event-9\"),\n\t\t\t\t\t\tCreatedTime: common.Int64Ptr(1700001000000000000),\n\t\t\t\t\t\tClusterFailovers: []*types.ClusterFailover{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tFromCluster: &types.ActiveClusterInfo{ActiveClusterName: \"cluster-3\"},\n\t\t\t\t\t\t\t\tToCluster:   &types.ActiveClusterInfo{ActiveClusterName: \"cluster-4\"},\n\t\t\t\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\t\t\t\tScope: \"dc\",\n\t\t\t\t\t\t\t\t\tName:  \"dc2\",\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: []string{\n\t\t\t\t\"event-8\",\n\t\t\t\t\"cluster-1 -> cluster-2\",\n\t\t\t\t\"dc.dc1\",\n\t\t\t\t\"event-9\",\n\t\t\t\t\"cluster-3 -> cluster-4\",\n\t\t\t\t\"dc.dc2\",\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, tc := range testCases {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tvar output strings.Builder\n\t\t\trenderFailoverHistoryTableToWriter(&output, tc.response)\n\n\t\t\tresult := output.String()\n\t\t\tfor _, expected := range tc.expectedOutput {\n\t\t\t\tassert.Contains(t, result, expected, \"output should contain '%s'\", expected)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/domain_migration_command.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"reflect\"\n\t\"strconv\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nconst (\n\t// workflows running longer than 14 days are considered long running workflows\n\tlongRunningDuration = 14 * 24 * time.Hour\n)\n\nvar (\n\tdomainMigrationTemplate = `Validation Check:\n{{- range .}}\n- {{.ValidationCheck}}: {{.ValidationResult}}\n{{- with .ValidationDetails}}\n  {{- with .CurrentDomainRow}}\n    Current Domain:\n      Name: {{.DomainInfo.Name}}\n      UUID: {{.DomainInfo.UUID}}\n  {{- end}}\n  {{- with .NewDomainRow}}\n    New Domain:\n      Name: {{.DomainInfo.Name}}\n      UUID: {{.DomainInfo.UUID}}\n  {{- end}}\n  {{- if ne (len .MismatchedDomainMetaData) 0 }}\n    Mismatched Domain Meta Data: {{.MismatchedDomainMetaData}}\n  {{- end }}\n  {{- if .LongRunningWorkFlowNum}}\n    Long Running Workflow Num (> 14 days): {{.LongRunningWorkFlowNum}}\n  {{- end}}\n  {{- if .MissingCurrSearchAttributes}}\n    Missing Search Attributes in Current Domain:\n    {{- range .MissingCurrSearchAttributes}}\n      - {{.}}\n    {{- end}}\n  {{- end}}\n  {{- if .MissingNewSearchAttributes}}\n    Missing Search Attributes in New Domain:\n    {{- range .MissingNewSearchAttributes}}\n      - {{.}}\n    {{- end}}\n  {{- end}}\n  {{- range .MismatchedDynamicConfig}}\n    {{- $dynamicConfig := . }}\n    - Config Key: {{.Key}}\n      {{- range $i, $v := .CurrValues}}\n        Current Response:\n          Data: {{ printf \"%s\" (index $dynamicConfig.CurrValues $i).Value.Data }}\n          Filters:\n          {{- range $filter := (index $dynamicConfig.CurrValues $i).Filters}}\n            - Name: {{ $filter.Name }}\n              Value: {{ printf \"%s\" $filter.Value.Data }}\n          {{- end}}\n        New Response:\n          Data: {{ printf \"%s\" (index $dynamicConfig.NewValues $i).Value.Data }}\n          Filters:\n          {{- range $filter := (index $dynamicConfig.NewValues $i).Filters}}\n            - Name: {{ $filter.Name }}\n              Value: {{ printf \"%s\" $filter.Value.Data }}\n          {{- end}}\n        {{- end}}\n      {{- end}}\n  {{- end}}\n{{- end}}\n`\n\temptyGetDynamicConfigRequest = &types.GetDynamicConfigResponse{\n\t\tValue: &types.DataBlob{\n\t\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\t},\n\t}\n)\n\ntype DomainMigrationCommand interface {\n\tValidation(c *cli.Context) error\n\tDomainMetaDataCheck(c *cli.Context) (DomainMigrationRow, error)\n\tDomainWorkFlowCheck(c *cli.Context) (DomainMigrationRow, error)\n\tSearchAttributesChecker(c *cli.Context) (DomainMigrationRow, error)\n\tDynamicConfigCheck(c *cli.Context) (DomainMigrationRow, error)\n}\n\nfunc (d *domainMigrationCLIImpl) NewDomainMigrationCLIImpl(c *cli.Context) (*domainMigrationCLIImpl, error) {\n\tfc, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfcm, err := getDeps(c).ServerFrontendClientForMigration(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tac, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tacm, err := getDeps(c).ServerAdminClientForMigration(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &domainMigrationCLIImpl{\n\t\tfrontendClient:         fc,\n\t\tdestinationClient:      fcm,\n\t\tfrontendAdminClient:    ac,\n\t\tdestinationAdminClient: acm,\n\t}, nil\n}\n\n// Export a function to create an instance of the domainMigrationCLIImpl.\nfunc NewDomainMigrationCommand(c *cli.Context) DomainMigrationCommand {\n\treturn &domainMigrationCLIImpl{}\n}\n\ntype domainMigrationCLIImpl struct {\n\tfrontendClient, destinationClient           frontend.Client\n\tfrontendAdminClient, destinationAdminClient admin.Client\n}\n\nfunc (d *domainMigrationCLIImpl) Validation(c *cli.Context) error {\n\tcheckers := []func(*cli.Context) (DomainMigrationRow, error){\n\t\td.DomainMetaDataCheck,\n\t\td.DomainWorkFlowCheck,\n\t\td.DynamicConfigCheck,\n\t\td.SearchAttributesChecker,\n\t}\n\twg := &sync.WaitGroup{}\n\terrCh := make(chan error, len(checkers)) // Channel to capture errors\n\tresults := make([]DomainMigrationRow, len(checkers))\n\tvar err error\n\tfor i := range checkers {\n\t\tgo func(i int) {\n\t\t\tdefer wg.Done()\n\t\t\tresults[i], err = checkers[i](c)\n\t\t\tif err != nil {\n\t\t\t\terrCh <- fmt.Errorf(\"Error in checkers: %w\", err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\terrCh <- nil\n\t\t}(i)\n\t}\n\n\twg.Wait()\n\tclose(errCh)\n\tfor err := range errCh {\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\trenderOpts := RenderOptions{\n\t\tDefaultTemplate: domainMigrationTemplate,\n\t\tColor:           true,\n\t\tBorder:          true,\n\t\tPrintDateTime:   true,\n\t}\n\n\tif err := Render(c, results, renderOpts); err != nil {\n\t\treturn fmt.Errorf(\"failed to render: %w\", err)\n\t}\n\treturn nil\n}\n\nfunc (d *domainMigrationCLIImpl) DomainMetaDataCheck(c *cli.Context) (DomainMigrationRow, error) {\n\tdomain := c.String(FlagDomain)\n\tnewDomain := c.String(FlagDestinationDomain)\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Error in Domain meta data check: %w\", err)\n\t}\n\tcurrResp, err := d.frontendClient.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: &domain,\n\t})\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Could not describe old domain, Please check to see if old domain exists before migrating. %w\", err)\n\t}\n\tnewResp, err := d.destinationClient.DescribeDomain(ctx, &types.DescribeDomainRequest{\n\t\tName: &newDomain,\n\t})\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Could not describe new domain, Please check to see if new domain exists before migrating. %w\", err)\n\t}\n\tvalidationResult, mismatchedMetaData := metaDataValidation(currResp, newResp)\n\tvalidationRow := DomainMigrationRow{\n\t\tValidationCheck:  \"Domain Meta Data\",\n\t\tValidationResult: validationResult,\n\t\tValidationDetails: ValidationDetails{\n\t\t\tCurrentDomainRow:         currResp,\n\t\t\tNewDomainRow:             newResp,\n\t\t\tMismatchedDomainMetaData: mismatchedMetaData,\n\t\t},\n\t}\n\treturn validationRow, nil\n}\n\nfunc metaDataValidation(currResp *types.DescribeDomainResponse, newResp *types.DescribeDomainResponse) (bool, string) {\n\tif !reflect.DeepEqual(currResp.Configuration, newResp.Configuration) {\n\t\treturn false, \"mismatched DomainConfiguration\"\n\t}\n\n\tif currResp.DomainInfo.OwnerEmail != newResp.DomainInfo.OwnerEmail {\n\t\treturn false, \"mismatched OwnerEmail\"\n\t}\n\treturn true, \"\"\n}\n\nfunc (d *domainMigrationCLIImpl) DomainWorkFlowCheck(c *cli.Context) (DomainMigrationRow, error) {\n\tcountWorkFlows, err := d.countLongRunningWorkflow(c)\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Error in Domain flow check: %w\", err)\n\t}\n\tcheck := countWorkFlows == 0\n\tdmr := DomainMigrationRow{\n\t\tValidationCheck:  \"Workflow Check\",\n\t\tValidationResult: check,\n\t\tValidationDetails: ValidationDetails{\n\t\t\tLongRunningWorkFlowNum: &countWorkFlows,\n\t\t},\n\t}\n\treturn dmr, nil\n}\n\nfunc (d *domainMigrationCLIImpl) countLongRunningWorkflow(c *cli.Context) (int, error) {\n\tdomain := c.String(FlagDomain)\n\tthresholdOfLongRunning := time.Now().Add(-longRunningDuration)\n\trequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: domain,\n\t\tQuery:  \"CloseTime=missing AND StartTime < \" + strconv.FormatInt(thresholdOfLongRunning.UnixNano(), 10),\n\t}\n\tctx, cancel, err := newContextForLongPoll(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn 0, fmt.Errorf(\"Error in creating long context: %w\", err)\n\t}\n\tresponse, err := d.frontendClient.CountWorkflowExecutions(ctx, request)\n\tif err != nil {\n\t\treturn 0, fmt.Errorf(\"Failed to count workflow. %w\", err)\n\t}\n\treturn int(response.GetCount()), nil\n}\n\nfunc (d *domainMigrationCLIImpl) SearchAttributesChecker(c *cli.Context) (DomainMigrationRow, error) {\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Error in creating long context: %w\", err)\n\t}\n\t// getting user provided search attributes\n\tsearchAttributes := c.StringSlice(FlagSearchAttribute)\n\tif len(searchAttributes) == 0 {\n\t\tdmr := DomainMigrationRow{\n\t\t\tValidationCheck:  \"Search Attributes Check\",\n\t\t\tValidationResult: true,\n\t\t}\n\t\treturn dmr, nil\n\t}\n\n\t// Parse the provided search attributes into a map[string]IndexValueType\n\trequiredAttributes := make(map[string]types.IndexedValueType)\n\tfor _, attr := range searchAttributes {\n\t\tparts := strings.SplitN(attr, \":\", 2)\n\t\tif len(parts) != 2 {\n\t\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Invalid search attribute format: %s, Error: %v\", attr, nil)\n\t\t}\n\t\tkey, valueType := parts[0], parts[1]\n\t\tivt, err := parseIndexedValueType(valueType)\n\t\tif err != nil {\n\t\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Invalid search attribute type for %s: %s , Error: %v\", key, valueType, err)\n\t\t}\n\t\trequiredAttributes[key] = ivt\n\t}\n\n\t// getting search attributes for current domain\n\tcurrentSearchAttributes, err := d.frontendClient.GetSearchAttributes(ctx)\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Unable to get search attributes for old domain. %w\", err)\n\t}\n\n\t// getting search attributes for new domain\n\tdestinationSearchAttributes, err := d.destinationClient.GetSearchAttributes(ctx)\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Unable to get search attributes for new domain. %w\", err)\n\t}\n\n\tcurrentSearchAttrs := currentSearchAttributes.Keys\n\tdestinationSearchAttrs := destinationSearchAttributes.Keys\n\n\t// checking to see if search attributes exist\n\tmissingInCurrent := findMissingAttributes(requiredAttributes, currentSearchAttrs)\n\tmissingInNew := findMissingAttributes(requiredAttributes, destinationSearchAttrs)\n\n\tvalidationResult := len(missingInCurrent) == 0 && len(missingInNew) == 0\n\n\tvalidationRow := DomainMigrationRow{\n\t\tValidationCheck:  \"Search Attributes Check\",\n\t\tValidationResult: validationResult,\n\t\tValidationDetails: ValidationDetails{\n\t\t\tMissingCurrSearchAttributes: missingInCurrent,\n\t\t\tMissingNewSearchAttributes:  missingInNew,\n\t\t},\n\t}\n\n\treturn validationRow, nil\n}\n\n// helper to parse types.IndexedValueType from string\nfunc parseIndexedValueType(valueType string) (types.IndexedValueType, error) {\n\tvar result types.IndexedValueType\n\tvalueTypeBytes := []byte(valueType)\n\tif err := result.UnmarshalText(valueTypeBytes); err != nil {\n\t\treturn 0, err\n\t}\n\treturn result, nil\n}\n\n// finds missing attributed in a map of existing attributed based on required attributes\nfunc findMissingAttributes(requiredAttributes map[string]types.IndexedValueType, existingAttributes map[string]types.IndexedValueType) []string {\n\tmissingAttributes := make([]string, 0)\n\tfor key, requiredType := range requiredAttributes {\n\t\texistingType, ok := existingAttributes[key]\n\t\tif !ok || existingType != requiredType {\n\t\t\t// construct the key:type string format\n\t\t\tattr := fmt.Sprintf(\"%s:%s\", key, requiredType)\n\t\t\tmissingAttributes = append(missingAttributes, attr)\n\t\t}\n\t}\n\treturn missingAttributes\n}\n\nfunc (d *domainMigrationCLIImpl) DynamicConfigCheck(c *cli.Context) (DomainMigrationRow, error) {\n\tvar mismatchedConfigs []MismatchedDynamicConfig\n\tcheck := true\n\n\tresp := dynamicproperties.ListAllProductionKeys()\n\n\tcurrDomain := c.String(FlagDomain)\n\tnewDomain := c.String(FlagDestinationDomain)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Failed to create context: %w\", err)\n\t}\n\tcurrentDomainID, err1 := getDomainID(ctx, currDomain, d.frontendClient)\n\tdestinationDomainID, err2 := getDomainID(ctx, newDomain, d.destinationClient)\n\tif currentDomainID == \"\" || err1 != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Failed to get domainID for the current domain. %v\", nil)\n\t}\n\n\tif destinationDomainID == \"\" || err2 != nil {\n\t\treturn DomainMigrationRow{}, fmt.Errorf(\"Failed to get domainID for the destination domain. %v\", nil)\n\t}\n\n\tfor _, configKey := range resp {\n\t\tif len(configKey.Filters()) == 1 && configKey.Filters()[0] == dynamicproperties.DomainName {\n\t\t\t// Validate dynamic configs with only domainName filter\n\t\t\tcurrRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\tdynamicproperties.DomainFilter(currDomain),\n\t\t\t})\n\n\t\t\tnewRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\tdynamicproperties.DomainFilter(newDomain),\n\t\t\t})\n\n\t\t\tcurrResp, err := d.frontendAdminClient.GetDynamicConfig(ctx, currRequest)\n\t\t\tif err != nil {\n\t\t\t\t// empty to indicate N/A\n\t\t\t\tcurrResp = emptyGetDynamicConfigRequest\n\t\t\t}\n\t\t\tnewResp, err := d.destinationAdminClient.GetDynamicConfig(ctx, newRequest)\n\t\t\tif err != nil {\n\t\t\t\t// empty to indicate N/A\n\t\t\t\tnewResp = emptyGetDynamicConfigRequest\n\t\t\t}\n\n\t\t\tif !reflect.DeepEqual(currResp.Value, newResp.Value) {\n\t\t\t\tcheck = false\n\t\t\t\tmismatchedConfigs = append(mismatchedConfigs, MismatchedDynamicConfig{\n\t\t\t\t\tKey: configKey,\n\t\t\t\t\tCurrValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\ttoDynamicConfigValue(currResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\t\tdynamicproperties.DomainName: currDomain,\n\t\t\t\t\t\t}),\n\t\t\t\t\t},\n\t\t\t\t\tNewValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\ttoDynamicConfigValue(newResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\t\tdynamicproperties.DomainName: newDomain,\n\t\t\t\t\t\t}),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\n\t\t} else if len(configKey.Filters()) == 1 && configKey.Filters()[0] == dynamicproperties.DomainID {\n\t\t\t// Validate dynamic configs with only domainID filter\n\t\t\tcurrRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\tdynamicproperties.DomainIDFilter(currentDomainID),\n\t\t\t})\n\n\t\t\tnewRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\tdynamicproperties.DomainIDFilter(destinationDomainID),\n\t\t\t})\n\n\t\t\tcurrResp, err := d.frontendAdminClient.GetDynamicConfig(ctx, currRequest)\n\t\t\tif err != nil {\n\t\t\t\t// empty to indicate N/A\n\t\t\t\tcurrResp = emptyGetDynamicConfigRequest\n\t\t\t}\n\t\t\tnewResp, err := d.destinationAdminClient.GetDynamicConfig(ctx, newRequest)\n\t\t\tif err != nil {\n\t\t\t\t// empty to indicate N/A\n\t\t\t\tnewResp = emptyGetDynamicConfigRequest\n\t\t\t}\n\n\t\t\tif !reflect.DeepEqual(currResp.Value, newResp.Value) {\n\t\t\t\tcheck = false\n\t\t\t\tmismatchedConfigs = append(mismatchedConfigs, MismatchedDynamicConfig{\n\t\t\t\t\tKey: configKey,\n\t\t\t\t\tCurrValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\ttoDynamicConfigValue(currResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\t\tdynamicproperties.DomainID: currentDomainID,\n\t\t\t\t\t\t}),\n\t\t\t\t\t},\n\t\t\t\t\tNewValues: []*types.DynamicConfigValue{\n\t\t\t\t\t\ttoDynamicConfigValue(newResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\t\tdynamicproperties.DomainID: destinationDomainID,\n\t\t\t\t\t\t}),\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\n\t\t} else if containsFilter(configKey, dynamicproperties.DomainName.String()) && containsFilter(configKey, dynamicproperties.TaskListName.String()) {\n\t\t\t// Validate dynamic configs with only domainName and TaskList filters\n\t\t\ttaskLists := c.StringSlice(FlagTaskList)\n\t\t\tvar mismatchedCurValues []*types.DynamicConfigValue\n\t\t\tvar mismatchedNewValues []*types.DynamicConfigValue\n\t\t\tfor _, taskList := range taskLists {\n\n\t\t\t\tcurrRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\t\tdynamicproperties.DomainFilter(currDomain),\n\t\t\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\t\t})\n\n\t\t\t\tnewRequest := dynamicproperties.ToGetDynamicConfigFilterRequest(configKey.String(), []dynamicproperties.FilterOption{\n\t\t\t\t\tdynamicproperties.DomainFilter(newDomain),\n\t\t\t\t\tdynamicproperties.TaskListFilter(taskList),\n\t\t\t\t})\n\n\t\t\t\tcurrResp, err := d.frontendAdminClient.GetDynamicConfig(ctx, currRequest)\n\t\t\t\tif err != nil {\n\t\t\t\t\t// empty to indicate N/A\n\t\t\t\t\tcurrResp = emptyGetDynamicConfigRequest\n\t\t\t\t}\n\t\t\t\tnewResp, err := d.destinationAdminClient.GetDynamicConfig(ctx, newRequest)\n\t\t\t\tif err != nil {\n\t\t\t\t\t// empty to indicate N/A\n\t\t\t\t\tnewResp = emptyGetDynamicConfigRequest\n\t\t\t\t}\n\n\t\t\t\tif !reflect.DeepEqual(currResp.Value, newResp.Value) {\n\t\t\t\t\tcheck = false\n\t\t\t\t\tmismatchedCurValues = append(mismatchedCurValues, toDynamicConfigValue(currResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\tdynamicproperties.DomainName:   currDomain,\n\t\t\t\t\t\tdynamicproperties.TaskListName: taskLists,\n\t\t\t\t\t}))\n\t\t\t\t\tmismatchedNewValues = append(mismatchedNewValues, toDynamicConfigValue(newResp.Value, map[dynamicproperties.Filter]interface{}{\n\t\t\t\t\t\tdynamicproperties.DomainName:   newDomain,\n\t\t\t\t\t\tdynamicproperties.TaskListName: taskLists,\n\t\t\t\t\t}))\n\n\t\t\t\t}\n\t\t\t}\n\t\t\tif len(mismatchedCurValues) > 0 && len(mismatchedNewValues) > 0 {\n\t\t\t\tmismatchedConfigs = append(mismatchedConfigs, MismatchedDynamicConfig{\n\t\t\t\t\tKey:        configKey,\n\t\t\t\t\tCurrValues: mismatchedCurValues,\n\t\t\t\t\tNewValues:  mismatchedNewValues,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t}\n\n\tvalidationRow := DomainMigrationRow{\n\t\tValidationCheck:  \"Dynamic Config Check\",\n\t\tValidationResult: check,\n\t\tValidationDetails: ValidationDetails{\n\t\t\tMismatchedDynamicConfig: mismatchedConfigs,\n\t\t},\n\t}\n\n\treturn validationRow, nil\n}\n\nfunc getDomainID(c context.Context, domain string, client frontend.Client) (string, error) {\n\tresp, err := client.DescribeDomain(c, &types.DescribeDomainRequest{Name: &domain})\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"Failed to describe domain. %w\", err)\n\t}\n\n\treturn resp.DomainInfo.GetUUID(), nil\n}\n\nfunc valueToDataBlob(value interface{}) *types.DataBlob {\n\tif value == nil {\n\t\treturn nil\n\t}\n\t// No need to handle error as this is a private helper method\n\t// where the correct value will always be passed regardless\n\tdata, _ := json.Marshal(value)\n\n\treturn &types.DataBlob{\n\t\tEncodingType: types.EncodingTypeJSON.Ptr(),\n\t\tData:         data,\n\t}\n}\n\nfunc toDynamicConfigValue(value *types.DataBlob, filterMaps map[dynamicproperties.Filter]interface{}) *types.DynamicConfigValue {\n\tvar configFilters []*types.DynamicConfigFilter\n\tfor filter, filterValue := range filterMaps {\n\t\tconfigFilters = append(configFilters, &types.DynamicConfigFilter{\n\t\t\tName:  filter.String(),\n\t\t\tValue: valueToDataBlob(filterValue),\n\t\t})\n\t}\n\n\treturn &types.DynamicConfigValue{\n\t\tValue:   value,\n\t\tFilters: configFilters,\n\t}\n}\n\nfunc containsFilter(key dynamicproperties.Key, value string) bool {\n\tfilters := key.Filters()\n\tfor _, filter := range filters {\n\t\tif filter.String() == value {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "tools/cli/domain_migration_command_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"testing\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nconst (\n\ttestDomainName    = \"test-domain\"\n\ttestNewDomainName = \"test-new-domain\"\n\ttestDomainOwner   = \"test-owner\"\n)\n\ntype domainMigrationCliTestData struct {\n\tmockFrontendClient     *frontend.MockClient\n\tmockAdminClient        *admin.MockClient\n\tioHandler              *testIOHandler\n\tapp                    *cli.App\n\tmockManagerFactory     *MockManagerFactory\n\tdomainMigrationCLIImpl DomainMigrationCommand\n}\n\nfunc newDomainMigrationCliTestData(t *testing.T) *domainMigrationCliTestData {\n\tvar td domainMigrationCliTestData\n\n\tctrl := gomock.NewController(t)\n\n\ttd.mockFrontendClient = frontend.NewMockClient(ctrl)\n\ttd.mockAdminClient = admin.NewMockClient(ctrl)\n\ttd.mockManagerFactory = NewMockManagerFactory(ctrl)\n\ttd.ioHandler = &testIOHandler{}\n\n\t// Create a new CLI app with client factory and persistence manager factory\n\ttd.app = NewCliApp(\n\t\t&clientFactoryMock{\n\t\t\tserverFrontendClient: td.mockFrontendClient,\n\t\t\tserverAdminClient:    td.mockAdminClient,\n\t\t},\n\t\tWithIOHandler(td.ioHandler),\n\t\tWithManagerFactory(td.mockManagerFactory), // Inject the mocked persistence manager factory\n\t)\n\n\ttd.domainMigrationCLIImpl = &domainMigrationCLIImpl{\n\t\tfrontendClient:         td.mockFrontendClient,\n\t\tdestinationClient:      td.mockFrontendClient,\n\t\tfrontendAdminClient:    td.mockAdminClient,\n\t\tdestinationAdminClient: td.mockAdminClient,\n\t}\n\treturn &td\n}\n\nfunc (td *domainMigrationCliTestData) consoleOutput() string {\n\treturn td.ioHandler.outputBytes.String()\n}\n\nfunc TestDomainMetaDataCheck(t *testing.T) {\n\ttestDomain := testDomainName\n\ttestNewDomain := testNewDomainName\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *domainMigrationCliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomain),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{OwnerEmail: testDomainOwner},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 30,\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{\n\t\t\t\t\tName: &testNewDomain,\n\t\t\t\t}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{OwnerEmail: testDomainOwner},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 30,\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"could not describe old domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomain),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(\n\t\t\t\t\tgomock.Any(), gomock.Any(),\n\t\t\t\t).Return(nil, assert.AnError).Times(1)\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Could not describe old domain, Please check to see if old domain exists before migrating.\",\n\t\t},\n\t\t{\n\t\t\tname: \"could not describe new domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomain),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), gomock.Any()).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{OwnerEmail: testDomainOwner},\n\t\t\t\t\tConfiguration: &types.DomainConfiguration{\n\t\t\t\t\t\tWorkflowExecutionRetentionPeriodInDays: 30,\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(\n\t\t\t\t\tgomock.Any(), gomock.Any(),\n\t\t\t\t).Return(nil, assert.AnError).Times(1)\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Could not describe new domain, Please check to see if new domain exists before migrating.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newDomainMigrationCliTestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\t_, err := td.domainMigrationCLIImpl.DomainMetaDataCheck(cliCtx)\n\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestDomainWorkflowCheck(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *domainMigrationCliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomainName),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{\n\t\t\t\t\tCount: 25,\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"could not count workflows\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomainName),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().CountWorkflowExecutions(\n\t\t\t\t\tgomock.Any(), gomock.Any(),\n\t\t\t\t).Return(nil, assert.AnError).Times(1)\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to count workflow.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newDomainMigrationCliTestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\t_, err := td.domainMigrationCLIImpl.DomainWorkFlowCheck(cliCtx)\n\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestSearchAttributesChecker(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *domainMigrationCliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"all arguments provided\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringSliceArgument(FlagSearchAttribute, \"Domain:STRING\", \"WorkflowID:INT\"),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(&types.GetSearchAttributesResponse{\n\t\t\t\t\tKeys: map[string]types.IndexedValueType{\n\t\t\t\t\t\t\"Domain\":     types.IndexedValueTypeString,\n\t\t\t\t\t\t\"WorkflowID\": types.IndexedValueTypeInt,\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(2)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"without the search attributes argument\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid search attribute format\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringSliceArgument(FlagSearchAttribute, \"Domain\"),\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Invalid search attribute format:\",\n\t\t},\n\t\t{\n\t\t\tname: \"invalid search attribute type\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringSliceArgument(FlagSearchAttribute, \"DomainData:Blob\"),\n\t\t\t\t)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Invalid search attribute type for DomainData: Blob\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get search attributes for old domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringSliceArgument(FlagSearchAttribute, \"Domain:STRING\", \"WorkflowID:INT\"),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(nil, assert.AnError).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Unable to get search attributes for old domain.\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get search attributes for new domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringSliceArgument(FlagSearchAttribute, \"Domain:STRING\", \"WorkflowID:INT\"),\n\t\t\t\t)\n\t\t\t\ttd.mockFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(&types.GetSearchAttributesResponse{\n\t\t\t\t\tKeys: map[string]types.IndexedValueType{\n\t\t\t\t\t\t\"Domain\":     types.IndexedValueTypeString,\n\t\t\t\t\t\t\"WorkflowID\": types.IndexedValueTypeInt,\n\t\t\t\t\t},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(nil, assert.AnError).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Unable to get search attributes for new domain.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newDomainMigrationCliTestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\t_, err := td.domainMigrationCLIImpl.SearchAttributesChecker(cliCtx)\n\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n\nfunc TestDomainConfigCheck(t *testing.T) {\n\ttestDomain := testDomainName\n\ttestNewDomain := testNewDomainName\n\ttestDomainUUID := uuid.New()\n\ttestNewDomainUUID := uuid.New()\n\n\ttests := []struct {\n\t\tname           string\n\t\ttestSetup      func(td *domainMigrationCliTestData) *cli.Context\n\t\terrContains    string // empty if no error is expected\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"dynamic configs are equal\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomainName),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testDomainUUID},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testNewDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testNewDomainUUID},\n\t\t\t\t}, nil).Times(1)\n\n\t\t\t\ttd.mockAdminClient.EXPECT().GetDynamicConfig(gomock.Any(), gomock.Any()).Return(&types.GetDynamicConfigResponse{\n\t\t\t\t\tValue: &types.DataBlob{\n\t\t\t\t\t\tEncodingType: types.EncodingTypeThriftRW.Ptr(),\n\t\t\t\t\t\tData:         []byte(\"config-value\"),\n\t\t\t\t\t},\n\t\t\t\t}, nil).AnyTimes()\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get domainID fo the current domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomainName),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testDomainUUID},\n\t\t\t\t}, assert.AnError).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testNewDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testNewDomainUUID},\n\t\t\t\t}, assert.AnError).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to get domainID for the current domain.\",\n\t\t},\n\t\t{\n\t\t\tname: \"failed to get domainID fo the destination domain\",\n\t\t\ttestSetup: func(td *domainMigrationCliTestData) *cli.Context {\n\t\t\t\tcliCtx := clitest.NewCLIContext(\n\t\t\t\t\tt,\n\t\t\t\t\ttd.app,\n\t\t\t\t\tclitest.StringArgument(FlagDomain, testDomain),\n\t\t\t\t\tclitest.StringArgument(FlagDestinationDomain, testNewDomainName),\n\t\t\t\t)\n\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testDomainUUID},\n\t\t\t\t}, nil).Times(1)\n\t\t\t\ttd.mockFrontendClient.EXPECT().DescribeDomain(gomock.Any(), &types.DescribeDomainRequest{Name: &testNewDomain}).Return(&types.DescribeDomainResponse{\n\t\t\t\t\tDomainInfo: &types.DomainInfo{UUID: testNewDomainUUID},\n\t\t\t\t}, assert.AnError).Times(1)\n\n\t\t\t\treturn cliCtx\n\t\t\t},\n\t\t\terrContains: \"Failed to get domainID for the destination domain.\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\ttd := newDomainMigrationCliTestData(t)\n\t\t\tcliCtx := tt.testSetup(td)\n\n\t\t\t_, err := td.domainMigrationCLIImpl.DynamicConfigCheck(cliCtx)\n\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t\tassert.Equal(t, tt.expectedOutput, td.consoleOutput())\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/domain_utils.go",
    "content": "// Copyright (c) 2019 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/stretchr/testify/mock\"\n\t\"github.com/uber-go/tally\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common/archiver\"\n\t\"github.com/uber/cadence/common/archiver/provider\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/cluster\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/domain\"\n\t\"github.com/uber/cadence/common/dynamicconfig\"\n\t\"github.com/uber/cadence/common/dynamicconfig/dynamicproperties\"\n\t\"github.com/uber/cadence/common/log\"\n\t\"github.com/uber/cadence/common/metrics\"\n\t\"github.com/uber/cadence/common/mocks\"\n\t\"github.com/uber/cadence/common/persistence\"\n\t\"github.com/uber/cadence/common/service\"\n\t\"github.com/uber/cadence/tools/common/flag\"\n)\n\nconst (\n\tdependencyMaxQPS = 100\n)\n\nvar (\n\tregisterDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagDescription,\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Domain description\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagOwnerEmail,\n\t\t\tAliases: []string{\"oe\"},\n\t\t\tUsage:   \"Owner email\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagRetentionDays,\n\t\t\tAliases: []string{\"rd\"},\n\t\t\tUsage:   \"Workflow execution retention in days\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagActiveClusterName,\n\t\t\tAliases: []string{\"ac\"},\n\t\t\tUsage:   \"Active cluster name\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    FlagActiveClusters,\n\t\t\tAliases: []string{\"acs\"},\n\t\t\tUsage:   \"Active clusters by cluster attribute in the format '<cluster-attr>.<scope>:<name> ie: region.manilla:cluster0,region.newyork:cluster1'\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    FlagClusters,\n\t\t\tAliases: []string{\"cl\"},\n\t\t\tUsage:   FlagClustersUsage,\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagIsGlobalDomain,\n\t\t\tAliases: []string{\"gd\"},\n\t\t\tUsage:   \"Flag to indicate whether domain is a global domain (active-passive domain). Default to true. Local domain is now legacy.\",\n\t\t\tValue:   \"true\",\n\t\t},\n\t\t&cli.GenericFlag{\n\t\t\tName:    FlagDomainData,\n\t\t\tAliases: []string{\"dmd\"},\n\t\t\tUsage:   \"Domain data of key value pairs (must be in key1=value1,key2=value2,...,keyN=valueN format, e.g. cluster=dca or cluster=dca,instance=cadence)\",\n\t\t\tValue:   &flag.StringMap{},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSecurityToken,\n\t\t\tAliases: []string{\"st\"},\n\t\t\tUsage:   \"Optional token for security check\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagHistoryArchivalStatus,\n\t\t\tAliases: []string{\"has\"},\n\t\t\tUsage:   \"Flag to set history archival status, valid values are \\\"disabled\\\" and \\\"enabled\\\"\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagHistoryArchivalURI,\n\t\t\tAliases: []string{\"huri\"},\n\t\t\tUsage:   \"Optionally specify history archival URI (cannot be changed after first time archival is enabled)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagVisibilityArchivalStatus,\n\t\t\tAliases: []string{\"vas\"},\n\t\t\tUsage:   \"Flag to set visibility archival status, valid values are \\\"disabled\\\" and \\\"enabled\\\"\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagVisibilityArchivalURI,\n\t\t\tAliases: []string{\"vuri\"},\n\t\t\tUsage:   \"Optionally specify visibility archival URI (cannot be changed after first time archival is enabled)\",\n\t\t},\n\t}\n\n\tupdateDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagDescription,\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Domain description\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagOwnerEmail,\n\t\t\tAliases: []string{\"oe\"},\n\t\t\tUsage:   \"Owner email\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagRetentionDays,\n\t\t\tAliases: []string{\"rd\"},\n\t\t\tUsage:   \"Workflow execution retention in days\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagActiveClusterName,\n\t\t\tAliases: []string{\"ac\"},\n\t\t\tUsage:   \"Active cluster name\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    FlagActiveClusters,\n\t\t\tAliases: []string{\"acs\"},\n\t\t\tUsage:   \"Active clusters by cluster attribute in the format '<cluster-attr>.<scope>:<name> ie: region.manilla:cluster0,region.newyork:cluster1'\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    FlagClusters,\n\t\t\tAliases: []string{\"cl\"},\n\t\t\tUsage:   FlagClustersUsage,\n\t\t},\n\t\t&cli.GenericFlag{\n\t\t\tName:  FlagDomainData,\n\t\t\tUsage: \"Domain data of key value pairs (must be in key1=value1,key2=value2,...,keyN=valueN format, e.g. cluster=dca or cluster=dca,instance=cadence)\",\n\t\t\tValue: &flag.StringMap{},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSecurityToken,\n\t\t\tAliases: []string{\"st\"},\n\t\t\tUsage:   \"Optional token for security check\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagHistoryArchivalStatus,\n\t\t\tAliases: []string{\"has\"},\n\t\t\tUsage:   \"Flag to set history archival status, valid values are \\\"disabled\\\" and \\\"enabled\\\"\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagHistoryArchivalURI,\n\t\t\tAliases: []string{\"huri\"},\n\t\t\tUsage:   \"Optionally specify history archival URI (cannot be changed after first time archival is enabled)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagVisibilityArchivalStatus,\n\t\t\tAliases: []string{\"vas\"},\n\t\t\tUsage:   \"Flag to set visibility archival status, valid values are \\\"disabled\\\" and \\\"enabled\\\"\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagVisibilityArchivalURI,\n\t\t\tAliases: []string{\"vuri\"},\n\t\t\tUsage:   \"Optionally specify visibility archival URI (cannot be changed after first time archival is enabled)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagAddBadBinary,\n\t\t\tUsage: \"Binary checksum to add for resetting workflow\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagRemoveBadBinary,\n\t\t\tUsage: \"Binary checksum to remove for resetting workflow\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagReason,\n\t\t\tUsage: \"Reason for the operation\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagFailoverType,\n\t\t\tAliases: []string{\"ft\"},\n\t\t\tUsage:   \"Domain failover type. Default value: force. Options: [force,grace]\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagFailoverTimeout,\n\t\t\tAliases: []string{\"fts\"},\n\t\t\tValue:   defaultGracefulFailoverTimeoutInSeconds,\n\t\t\tUsage:   \"[Optional] Domain failover timeout in seconds.\",\n\t\t},\n\t}\n\n\tdeleteDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSecurityToken,\n\t\t\tAliases: []string{\"st\"},\n\t\t\tUsage:   \"Optional token for security check\",\n\t\t},\n\t}\n\n\tdeprecateDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSecurityToken,\n\t\t\tAliases: []string{\"st\"},\n\t\t\tUsage:   \"Optional token for security check\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:  FlagForce,\n\t\t\tUsage: \"Deprecate domain regardless of domain history.\",\n\t\t},\n\t}\n\n\tdescribeDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDomainID,\n\t\t\tUsage: \"Domain UUID (required if not specify domainName)\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintJSON,\n\t\t\tAliases: []string{\"pjson\"},\n\t\t\tUsage:   \"Print in raw JSON format\",\n\t\t},\n\t\tgetFormatFlag(),\n\t}\n\n\tmigrateDomainFlags = []cli.Flag{\n\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDestinationAddress,\n\t\t\tUsage: \"Destination cadence-frontend address in <host>:<port> format\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagDestinationDomain,\n\t\t\tUsage: \"Destination domain name\",\n\t\t},\n\n\t\t&cli.StringSliceFlag{\n\t\t\tName:  FlagTaskList,\n\t\t\tUsage: \"All tasklists in the current domain\",\n\t\t},\n\n\t\t&cli.StringSliceFlag{\n\t\t\tName:  FlagSearchAttribute,\n\t\t\tUsage: \"Specify search attributes in the format key:type, available types are STRING, KEYWORD, INT, DOUBLE, BOOL, DATETIME\",\n\t\t},\n\n\t\tgetFormatFlag(),\n\t}\n\n\tfailoverDomainFlags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagActiveClusterName,\n\t\t\tAliases: []string{\"ac\"},\n\t\t\tUsage:   \"Active cluster name\",\n\t\t},\n\t\t&cli.StringSliceFlag{\n\t\t\tName:    FlagActiveClusters,\n\t\t\tAliases: []string{\"acs\"},\n\t\t\tUsage:   \"Active clusters by cluster attribute in the format '<cluster-attr>.<scope>:<name> ie: region.manilla:cluster0,region.newyork:cluster1'\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagActiveClustersJSON,\n\t\t\tAliases: []string{\"acs-json\"},\n\t\t\tUsage:   `Active clusters by cluster attribute in JSON format. Eg {\"attributeScopes\":{\"region-us-east1\":{\"clusterAttributes\":{\"new-york\":{\"activeClusterName\":\"cluster1\"}}}}}`,\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagFailoverReason,\n\t\t\tAliases: []string{\"r\"},\n\t\t\tUsage:   \"Reason for failover (for tracking and transparency)\",\n\t\t},\n\t}\n\n\tlistFailoverHistoryFlags = []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagAll,\n\t\t\tAliases: []string{\"a\"},\n\t\t\tUsage:   \"List all failover history events\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintJSON,\n\t\t\tAliases: []string{\"pjson\"},\n\t\t\tUsage:   \"Print in raw JSON format\",\n\t\t},\n\t\tgetFormatFlag(),\n\t}\n\n\tadminDomainCommonFlags = getDBFlags()\n\n\tadminRegisterDomainFlags = append(\n\t\tregisterDomainFlags,\n\t\tadminDomainCommonFlags...,\n\t)\n\n\tadminUpdateDomainFlags = append(\n\t\tupdateDomainFlags,\n\t\tadminDomainCommonFlags...,\n\t)\n\n\tadminDeleteDomainFlags = append(\n\t\tdeleteDomainFlags,\n\t\tadminDomainCommonFlags...,\n\t)\n\n\tadminDeprecateDomainFlags = append(\n\t\tdeprecateDomainFlags,\n\t\tadminDomainCommonFlags...,\n\t)\n\n\tadminDescribeDomainFlags = append(\n\t\tupdateDomainFlags,\n\t\tadminDomainCommonFlags...,\n\t)\n)\n\nfunc initializeFrontendClient(c *cli.Context) (frontend.Client, error) {\n\treturn getDeps(c).ServerFrontendClient(c)\n}\n\nfunc initializeFrontendAdminClient(c *cli.Context) (admin.Client, error) {\n\treturn getDeps(c).ServerAdminClient(c)\n}\n\nfunc initializeAdminDomainHandler(c *cli.Context) (domain.Handler, error) {\n\n\tconfiguration, err := getDeps(c).ServerConfig(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tmetricsClient := initializeMetricsClient()\n\tlogger, err := initializeLogger(configuration)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in init admin domain handler: %w\", err)\n\t}\n\tclusterMetadata := initializeClusterMetadata(configuration, metricsClient, logger)\n\tmetadataMgr, err := getDeps(c).initializeDomainManager(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in init admin domain handler: %w\", err)\n\t}\n\tdynamicConfig, err := initializeDynamicConfig(configuration, logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in init admin domain handler: %w\", err)\n\t}\n\tarchivalprovider, err := initializeArchivalProvider(configuration, clusterMetadata, metricsClient, logger)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in init admin domain handler: %w\", err)\n\t}\n\tdomainhandler := initializeDomainHandler(\n\t\tlogger,\n\t\tmetadataMgr,\n\t\tclusterMetadata,\n\t\tinitializeArchivalMetadata(configuration, dynamicConfig),\n\t\tarchivalprovider,\n\t)\n\treturn domainhandler, nil\n}\n\nfunc loadConfig(\n\tcontext *cli.Context,\n) (*config.Config, error) {\n\tenv := getEnvironment(context)\n\tzone := getZone(context)\n\tconfigDir, err := getConfigDir(context)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Unable to load config. %w\", err)\n\t}\n\tvar cfg config.Config\n\terr = config.Load(env, configDir, zone, &cfg)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Unable to load config. %w\", err)\n\t}\n\treturn &cfg, nil\n}\n\nfunc initializeDomainHandler(\n\tlogger log.Logger,\n\tdomainManager persistence.DomainManager,\n\tclusterMetadata cluster.Metadata,\n\tarchivalMetadata archiver.ArchivalMetadata,\n\tarchiverProvider provider.ArchiverProvider,\n) domain.Handler {\n\n\tdomainConfig := domain.Config{\n\t\tMinRetentionDays:  dynamicproperties.GetIntPropertyFn(dynamicproperties.MinRetentionDays.DefaultInt()),\n\t\tMaxBadBinaryCount: dynamicproperties.GetIntPropertyFilteredByDomain(dynamicproperties.FrontendMaxBadBinaries.DefaultInt()),\n\t\tFailoverCoolDown:  dynamicproperties.GetDurationPropertyFnFilteredByDomain(dynamicproperties.FrontendFailoverCoolDown.DefaultDuration()),\n\t}\n\treturn domain.NewHandler(\n\t\tdomainConfig,\n\t\tlogger,\n\t\tdomainManager,\n\t\tnil, // domainAuditManager not needed for CLI tools\n\t\tclusterMetadata,\n\t\tinitializeDomainReplicator(logger),\n\t\tarchivalMetadata,\n\t\tarchiverProvider,\n\t\tclock.NewRealTimeSource(),\n\t)\n}\n\nfunc initializeLogger(\n\tserviceConfig *config.Config,\n) (log.Logger, error) {\n\tzapLogger, err := serviceConfig.Log.NewZapLogger()\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to create zap logger, err: %w\", err)\n\t}\n\treturn log.NewLogger(zapLogger), nil\n}\n\nfunc initializeClusterMetadata(serviceConfig *config.Config, metrics metrics.Client, logger log.Logger) cluster.Metadata {\n\tclusterGroupMetadata := serviceConfig.ClusterGroupMetadata\n\treturn cluster.NewMetadata(\n\t\t*clusterGroupMetadata,\n\t\tfunc(d string) bool { return false },\n\t\tmetrics,\n\t\tlogger,\n\t)\n}\n\nfunc initializeArchivalMetadata(\n\tserviceConfig *config.Config,\n\tdynamicConfig *dynamicconfig.Collection,\n) archiver.ArchivalMetadata {\n\n\treturn archiver.NewArchivalMetadata(\n\t\tdynamicConfig,\n\t\tserviceConfig.Archival.History.Status,\n\t\tserviceConfig.Archival.History.EnableRead,\n\t\tserviceConfig.Archival.Visibility.Status,\n\t\tserviceConfig.Archival.Visibility.EnableRead,\n\t\t&serviceConfig.DomainDefaults.Archival,\n\t)\n}\n\nfunc initializeArchivalProvider(\n\tserviceConfig *config.Config,\n\tclusterMetadata cluster.Metadata,\n\tmetricsClient metrics.Client,\n\tlogger log.Logger,\n) (provider.ArchiverProvider, error) {\n\n\tarchiverProvider := provider.NewArchiverProvider(\n\t\tserviceConfig.Archival.History.Provider,\n\t\tserviceConfig.Archival.Visibility.Provider,\n\t)\n\n\thistoryArchiverBootstrapContainer := &archiver.HistoryBootstrapContainer{\n\t\tHistoryV2Manager: nil, // not used\n\t\tLogger:           logger,\n\t\tMetricsClient:    metricsClient,\n\t\tClusterMetadata:  clusterMetadata,\n\t\tDomainCache:      nil, // not used\n\t}\n\tvisibilityArchiverBootstrapContainer := &archiver.VisibilityBootstrapContainer{\n\t\tLogger:          logger,\n\t\tMetricsClient:   metricsClient,\n\t\tClusterMetadata: clusterMetadata,\n\t\tDomainCache:     nil, // not used\n\t}\n\n\terr := archiverProvider.RegisterBootstrapContainer(\n\t\tservice.Frontend,\n\t\thistoryArchiverBootstrapContainer,\n\t\tvisibilityArchiverBootstrapContainer,\n\t)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error initializing archival provider. %w\", err)\n\t}\n\treturn archiverProvider, nil\n}\n\nfunc initializeDomainReplicator(\n\tlogger log.Logger,\n) domain.Replicator {\n\n\treplicationMessageSink := &mocks.KafkaProducer{}\n\treplicationMessageSink.On(\"Publish\", mock.Anything, mock.Anything).Return(nil)\n\treturn domain.NewDomainReplicator(replicationMessageSink, logger)\n}\n\nfunc initializeDynamicConfig(\n\tserviceConfig *config.Config,\n\tlogger log.Logger,\n) (*dynamicconfig.Collection, error) {\n\n\t// the done channel is used by dynamic config to stop refreshing\n\t// and CLI does not need that, so just close the done channel\n\tdoneChan := make(chan struct{})\n\tclose(doneChan)\n\tdynamicConfigClient, err := dynamicconfig.NewFileBasedClient(\n\t\t&serviceConfig.DynamicConfig.FileBased,\n\t\tlogger,\n\t\tdoneChan,\n\t)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error initializing dynamic config. %w\", err)\n\t}\n\treturn dynamicconfig.NewCollection(dynamicConfigClient, logger), nil\n}\n\nfunc initializeMetricsClient() metrics.Client {\n\treturn metrics.NewClient(tally.NoopScope, metrics.Common, metrics.HistogramMigration{})\n}\n\nfunc getEnvironment(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(FlagServiceEnv))\n}\n\nfunc getZone(c *cli.Context) string {\n\treturn strings.TrimSpace(c.String(FlagServiceZone))\n}\n\nfunc getConfigDir(c *cli.Context) (string, error) {\n\tdirPath := c.String(FlagServiceConfigDir)\n\tif len(dirPath) == 0 {\n\t\treturn \"\", fmt.Errorf(\"Must provide service configuration dir path. %v\", nil)\n\t}\n\treturn dirPath, nil\n}\n"
  },
  {
    "path": "tools/cli/factory.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\n//go:generate mockgen -package $GOPACKAGE -source $GOFILE -destination factory_mock.go -self_package github.com/uber/cadence/tools/cli\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"crypto/tls\"\n\t\"crypto/x509\"\n\t\"fmt\"\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/olivere/elastic\"\n\tadminv1 \"github.com/uber/cadence-idl/go/proto/admin/v1\"\n\tapiv1 \"github.com/uber/cadence-idl/go/proto/api/v1\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/yarpc\"\n\t\"go.uber.org/yarpc/api/transport\"\n\t\"go.uber.org/yarpc/peer\"\n\t\"go.uber.org/yarpc/peer/hostport\"\n\t\"go.uber.org/yarpc/transport/grpc\"\n\t\"go.uber.org/yarpc/transport/tchannel\"\n\t\"go.uber.org/zap\"\n\t\"google.golang.org/grpc/credentials\"\n\n\tserverAdmin \"github.com/uber/cadence/.gen/go/admin/adminserviceclient\"\n\tserverFrontend \"github.com/uber/cadence/.gen/go/cadence/workflowserviceclient\"\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\tgrpcClient \"github.com/uber/cadence/client/wrappers/grpc\"\n\t\"github.com/uber/cadence/client/wrappers/thrift\"\n\t\"github.com/uber/cadence/common\"\n\tcc \"github.com/uber/cadence/common/client\"\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nconst (\n\tcadenceClientName      = \"cadence-client\"\n\tcadenceFrontendService = \"cadence-frontend\"\n)\n\n// ContextKey is an alias for string, used as context key\ntype ContextKey string\n\nconst (\n\t// CtxKeyJWT is the name of the context key for the JWT\n\tCtxKeyJWT = ContextKey(\"ctxKeyJWT\")\n)\n\n// ClientFactory is used to construct rpc clients\ntype ClientFactory interface {\n\tServerFrontendClient(c *cli.Context) (frontend.Client, error)\n\tServerAdminClient(c *cli.Context) (admin.Client, error)\n\n\t// ServerFrontendClientForMigration frontend client of the migration destination\n\tServerFrontendClientForMigration(c *cli.Context) (frontend.Client, error)\n\t// ServerAdminClientForMigration admin client of the migration destination\n\tServerAdminClientForMigration(c *cli.Context) (admin.Client, error)\n\n\tElasticSearchClient(c *cli.Context) (*elastic.Client, error)\n\n\tServerConfig(c *cli.Context) (*config.Config, error)\n}\n\ntype clientFactory struct {\n\tdispatcher          *yarpc.Dispatcher // lazy, via ensureDispatcher\n\tdispatcherMigration *yarpc.Dispatcher // lazy, via ensureDispatcherForMigration\n\tlogger              *zap.Logger\n}\n\n// NewClientFactory creates a new ClientFactory\nfunc NewClientFactory(logger *zap.Logger) ClientFactory {\n\treturn &clientFactory{\n\t\tlogger: logger,\n\t}\n}\n\n// ServerConfig returns Cadence server configs.\n// Use in some CLI admin operations (e.g. accessing DB directly)\nfunc (b *clientFactory) ServerConfig(c *cli.Context) (*config.Config, error) {\n\tenv := c.String(FlagServiceEnv)\n\tzone := c.String(FlagServiceZone)\n\tconfigDir := c.String(FlagServiceConfigDir)\n\n\tvar cfg config.Config\n\terr := config.Load(env, configDir, zone, &cfg)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\n\t\t\tfmt.Sprintf(\n\t\t\t\t\"failed to load config (for --%v %q --%v %q --%v %q)\",\n\t\t\t\tFlagServiceEnv, env, FlagServiceZone, zone, FlagServiceConfigDir, configDir,\n\t\t\t),\n\t\t\terr,\n\t\t)\n\t}\n\treturn &cfg, nil\n}\n\n// ServerFrontendClient builds a frontend client (based on server side thrift interface)\nfunc (b *clientFactory) ServerFrontendClient(c *cli.Context) (frontend.Client, error) {\n\terr := b.ensureDispatcher(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"failed to create frontend client dependency\", err)\n\t}\n\tclientConfig := b.dispatcher.ClientConfig(cadenceFrontendService)\n\tif c.String(FlagTransport) == grpcTransport {\n\t\treturn grpcClient.NewFrontendClient(\n\t\t\tapiv1.NewDomainAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewWorkflowAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewWorkerAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewVisibilityAPIYARPCClient(clientConfig),\n\t\t), nil\n\t}\n\treturn thrift.NewFrontendClient(serverFrontend.New(clientConfig)), nil\n}\n\n// ServerAdminClient builds an admin client (based on server side thrift interface)\nfunc (b *clientFactory) ServerAdminClient(c *cli.Context) (admin.Client, error) {\n\terr := b.ensureDispatcher(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"failed to create admin client dependency\", err)\n\t}\n\tclientConfig := b.dispatcher.ClientConfig(cadenceFrontendService)\n\tif c.String(FlagTransport) == grpcTransport {\n\t\treturn grpcClient.NewAdminClient(adminv1.NewAdminAPIYARPCClient(clientConfig)), nil\n\t}\n\treturn thrift.NewAdminClient(serverAdmin.New(clientConfig)), nil\n}\n\n// ServerFrontendClientForMigration builds a frontend client (based on server side thrift interface)\nfunc (b *clientFactory) ServerFrontendClientForMigration(c *cli.Context) (frontend.Client, error) {\n\terr := b.ensureDispatcherForMigration(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"failed to create frontend client dependency\", err)\n\t}\n\tclientConfig := b.dispatcherMigration.ClientConfig(cadenceFrontendService)\n\tif c.String(FlagTransport) == grpcTransport {\n\t\treturn grpcClient.NewFrontendClient(\n\t\t\tapiv1.NewDomainAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewWorkflowAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewWorkerAPIYARPCClient(clientConfig),\n\t\t\tapiv1.NewVisibilityAPIYARPCClient(clientConfig),\n\t\t), nil\n\t}\n\treturn thrift.NewFrontendClient(serverFrontend.New(clientConfig)), nil\n}\n\n// ServerAdminClientForMigration builds an admin client (based on server side thrift interface)\nfunc (b *clientFactory) ServerAdminClientForMigration(c *cli.Context) (admin.Client, error) {\n\terr := b.ensureDispatcherForMigration(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"failed to create admin client dependency\", err)\n\t}\n\tclientConfig := b.dispatcherMigration.ClientConfig(cadenceFrontendService)\n\tif c.String(FlagTransport) == grpcTransport {\n\t\treturn grpcClient.NewAdminClient(adminv1.NewAdminAPIYARPCClient(clientConfig)), nil\n\t}\n\treturn thrift.NewAdminClient(serverAdmin.New(clientConfig)), nil\n}\n\n// ElasticSearchClient builds an ElasticSearch client\nfunc (b *clientFactory) ElasticSearchClient(c *cli.Context) (*elastic.Client, error) {\n\n\turl, err := getRequiredOption(c, FlagURL)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Required flag not present %w\", err)\n\t}\n\n\tretrier := elastic.NewBackoffRetrier(elastic.NewExponentialBackoff(128*time.Millisecond, 513*time.Millisecond))\n\n\tclient, err := elastic.NewClient(\n\t\telastic.SetURL(url),\n\t\telastic.SetRetrier(retrier),\n\t)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\n\t\t\tfmt.Sprintf(\"failed to create ElasticSearch client (for --%v %q)\", FlagURL, url),\n\t\t\terr,\n\t\t)\n\t}\n\n\treturn client, nil\n}\n\nfunc (b *clientFactory) ensureDispatcher(c *cli.Context) error {\n\tif b.dispatcher != nil {\n\t\treturn nil\n\t}\n\td, err := b.newClientDispatcher(c, c.String(FlagAddress))\n\tif err != nil {\n\t\treturn commoncli.Problem(\n\t\t\tfmt.Sprintf(\"failed to create dispatcher (for --%v %q)\", FlagAddress, c.String(FlagAddress)),\n\t\t\terr,\n\t\t)\n\t}\n\tb.dispatcher = d\n\treturn nil\n}\n\nfunc (b *clientFactory) ensureDispatcherForMigration(c *cli.Context) error {\n\tif b.dispatcherMigration != nil {\n\t\treturn nil\n\t}\n\tdm, err := b.newClientDispatcher(c, c.String(FlagDestinationAddress))\n\tif err != nil {\n\t\treturn fmt.Errorf(\n\t\t\t\"failed to create dispatcher for migration (for --%v %q): %w\",\n\t\t\tFlagDestinationAddress, c.String(FlagDestinationAddress),\n\t\t\terr,\n\t\t)\n\t}\n\tb.dispatcherMigration = dm\n\treturn nil\n}\n\nfunc (b *clientFactory) newClientDispatcher(c *cli.Context, hostPortOverride string) (*yarpc.Dispatcher, error) {\n\tshouldUseGrpc := c.String(FlagTransport) == grpcTransport\n\n\thostPort := tchannelPort\n\tif shouldUseGrpc {\n\t\thostPort = grpcPort\n\t}\n\tif hostPortOverride != \"\" {\n\t\thostPort = hostPortOverride\n\t}\n\tvar outbounds transport.Outbounds\n\tif shouldUseGrpc {\n\t\tgrpcTransport := grpc.NewTransport()\n\t\toutbounds = transport.Outbounds{Unary: grpc.NewTransport().NewSingleOutbound(hostPort)}\n\n\t\ttlsCertificatePath := c.String(FlagTLSCertPath)\n\t\tif tlsCertificatePath != \"\" {\n\t\t\tcaCert, err := os.ReadFile(tlsCertificatePath)\n\t\t\tif err != nil {\n\t\t\t\treturn nil, commoncli.Problem(\n\t\t\t\t\tfmt.Sprintf(\"unable to find server CA certificate, from --%s %q\", FlagTLSCertPath, tlsCertificatePath),\n\t\t\t\t\terr,\n\t\t\t\t)\n\t\t\t}\n\t\t\tcaCertPool := x509.NewCertPool()\n\t\t\tif !caCertPool.AppendCertsFromPEM(caCert) {\n\t\t\t\tb.logger.Fatal(\"Failed to add server CA certificate\", zap.Error(err))\n\t\t\t}\n\t\t\ttlsConfig := tls.Config{\n\t\t\t\tRootCAs: caCertPool,\n\t\t\t}\n\t\t\ttlsCreds := credentials.NewTLS(&tlsConfig)\n\t\t\ttlsChooser := peer.NewSingle(hostport.Identify(hostPort), grpcTransport.NewDialer(grpc.DialerCredentials(tlsCreds)))\n\t\t\toutbounds = transport.Outbounds{Unary: grpc.NewTransport().NewOutbound(tlsChooser)}\n\t\t}\n\t} else {\n\t\tch, err := tchannel.NewChannelTransport(tchannel.ServiceName(cadenceClientName), tchannel.ListenAddr(\"127.0.0.1:0\"))\n\t\tif err != nil {\n\t\t\treturn nil, commoncli.Problem(\"failed create tchannel client transport\", err)\n\t\t}\n\t\toutbounds = transport.Outbounds{Unary: ch.NewSingleOutbound(hostPort)}\n\t}\n\n\tdispatcher := yarpc.NewDispatcher(yarpc.Config{\n\t\tName:      cadenceClientName,\n\t\tOutbounds: yarpc.Outbounds{cadenceFrontendService: outbounds},\n\t\tOutboundMiddleware: yarpc.OutboundMiddleware{\n\t\t\tUnary: &versionMiddleware{},\n\t\t},\n\t})\n\n\tif err := dispatcher.Start(); err != nil {\n\t\tif stoperr := dispatcher.Stop(); stoperr != nil {\n\t\t\treturn nil, commoncli.Problem(\n\t\t\t\t\"failed to start (and stop) tchannel dispatcher\",\n\t\t\t\t// currently does not print very well due to joined errors, but that can be improved\n\t\t\t\tfmt.Errorf(\"start err: %w, stop err: %w\", err, stoperr),\n\t\t\t)\n\t\t}\n\t\treturn nil, commoncli.Problem(\"failed to start tchannel dispatcher\", err)\n\t}\n\treturn dispatcher, nil\n}\n\ntype versionMiddleware struct {\n}\n\nfunc (vm *versionMiddleware) Call(ctx context.Context, request *transport.Request, out transport.UnaryOutbound) (*transport.Response, error) {\n\trequest.Headers = request.Headers.\n\t\tWith(common.ClientImplHeaderName, cc.CLI).\n\t\tWith(common.FeatureVersionHeaderName, cc.SupportedCLIVersion).\n\t\tWith(common.ClientFeatureFlagsHeaderName, cc.FeatureFlagsHeader(cc.DefaultCLIFeatureFlags)).\n\t\tWith(common.CallerTypeHeaderName, types.CallerTypeCLI.String())\n\tif jwtKey, ok := ctx.Value(CtxKeyJWT).(string); ok {\n\t\trequest.Headers = request.Headers.With(common.AuthorizationTokenHeaderName, jwtKey)\n\t}\n\treturn out.Call(ctx, request)\n}\n\nfunc getJWT(c *cli.Context) string {\n\treturn c.String(FlagJWT)\n}\n\nfunc getJWTPrivateKey(c *cli.Context) string {\n\treturn c.String(FlagJWTPrivateKey)\n}\n"
  },
  {
    "path": "tools/cli/factory_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: factory.go\n//\n// Generated by this command:\n//\n//\tmockgen -package cli -source factory.go -destination factory_mock.go -self_package github.com/uber/cadence/tools/cli\n//\n\n// Package cli is a generated GoMock package.\npackage cli\n\nimport (\n\treflect \"reflect\"\n\n\telastic \"github.com/olivere/elastic\"\n\tcli \"github.com/urfave/cli/v2\"\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tadmin \"github.com/uber/cadence/client/admin\"\n\tfrontend \"github.com/uber/cadence/client/frontend\"\n\tconfig \"github.com/uber/cadence/common/config\"\n)\n\n// MockClientFactory is a mock of ClientFactory interface.\ntype MockClientFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockClientFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockClientFactoryMockRecorder is the mock recorder for MockClientFactory.\ntype MockClientFactoryMockRecorder struct {\n\tmock *MockClientFactory\n}\n\n// NewMockClientFactory creates a new mock instance.\nfunc NewMockClientFactory(ctrl *gomock.Controller) *MockClientFactory {\n\tmock := &MockClientFactory{ctrl: ctrl}\n\tmock.recorder = &MockClientFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockClientFactory) EXPECT() *MockClientFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// ElasticSearchClient mocks base method.\nfunc (m *MockClientFactory) ElasticSearchClient(c *cli.Context) (*elastic.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ElasticSearchClient\", c)\n\tret0, _ := ret[0].(*elastic.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ElasticSearchClient indicates an expected call of ElasticSearchClient.\nfunc (mr *MockClientFactoryMockRecorder) ElasticSearchClient(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ElasticSearchClient\", reflect.TypeOf((*MockClientFactory)(nil).ElasticSearchClient), c)\n}\n\n// ServerAdminClient mocks base method.\nfunc (m *MockClientFactory) ServerAdminClient(c *cli.Context) (admin.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ServerAdminClient\", c)\n\tret0, _ := ret[0].(admin.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ServerAdminClient indicates an expected call of ServerAdminClient.\nfunc (mr *MockClientFactoryMockRecorder) ServerAdminClient(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ServerAdminClient\", reflect.TypeOf((*MockClientFactory)(nil).ServerAdminClient), c)\n}\n\n// ServerAdminClientForMigration mocks base method.\nfunc (m *MockClientFactory) ServerAdminClientForMigration(c *cli.Context) (admin.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ServerAdminClientForMigration\", c)\n\tret0, _ := ret[0].(admin.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ServerAdminClientForMigration indicates an expected call of ServerAdminClientForMigration.\nfunc (mr *MockClientFactoryMockRecorder) ServerAdminClientForMigration(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ServerAdminClientForMigration\", reflect.TypeOf((*MockClientFactory)(nil).ServerAdminClientForMigration), c)\n}\n\n// ServerConfig mocks base method.\nfunc (m *MockClientFactory) ServerConfig(c *cli.Context) (*config.Config, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ServerConfig\", c)\n\tret0, _ := ret[0].(*config.Config)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ServerConfig indicates an expected call of ServerConfig.\nfunc (mr *MockClientFactoryMockRecorder) ServerConfig(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ServerConfig\", reflect.TypeOf((*MockClientFactory)(nil).ServerConfig), c)\n}\n\n// ServerFrontendClient mocks base method.\nfunc (m *MockClientFactory) ServerFrontendClient(c *cli.Context) (frontend.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ServerFrontendClient\", c)\n\tret0, _ := ret[0].(frontend.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ServerFrontendClient indicates an expected call of ServerFrontendClient.\nfunc (mr *MockClientFactoryMockRecorder) ServerFrontendClient(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ServerFrontendClient\", reflect.TypeOf((*MockClientFactory)(nil).ServerFrontendClient), c)\n}\n\n// ServerFrontendClientForMigration mocks base method.\nfunc (m *MockClientFactory) ServerFrontendClientForMigration(c *cli.Context) (frontend.Client, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ServerFrontendClientForMigration\", c)\n\tret0, _ := ret[0].(frontend.Client)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ServerFrontendClientForMigration indicates an expected call of ServerFrontendClientForMigration.\nfunc (mr *MockClientFactoryMockRecorder) ServerFrontendClientForMigration(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ServerFrontendClientForMigration\", reflect.TypeOf((*MockClientFactory)(nil).ServerFrontendClientForMigration), c)\n}\n"
  },
  {
    "path": "tools/cli/flags.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport \"github.com/urfave/cli/v2\"\n\n// Flags used to specify cli command line arguments\nconst (\n\tFlagVerbose                        = \"verbose\"\n\tFlagUsername                       = \"username\"\n\tFlagPassword                       = \"password\"\n\tFlagKeyspace                       = \"keyspace\"\n\tFlagDatabaseName                   = \"db_name\"\n\tFlagEncodingType                   = \"encoding_type\"\n\tFlagDecodingTypes                  = \"decoding_types\"\n\tFlagAddress                        = \"address\"\n\tFlagDestinationAddress             = \"destination_address\"\n\tFlagHistoryAddress                 = \"history_address\"\n\tFlagDBType                         = \"db_type\"\n\tFlagDBAddress                      = \"db_address\"\n\tFlagDBPort                         = \"db_port\"\n\tFlagDBRegion                       = \"db_region\"\n\tFlagDBShard                        = \"db_shard\"\n\tFlagProtoVersion                   = \"protocol_version\"\n\tFlagDomainID                       = \"domain_id\"\n\tFlagDomain                         = \"domain\"\n\tFlagDestinationDomain              = \"destination_domain\"\n\tFlagShardID                        = \"shard_id\"\n\tFlagShards                         = \"shards\"\n\tFlagRangeID                        = \"range_id\"\n\tFlagWorkflowID                     = \"workflow_id\"\n\tFlagRunID                          = \"run_id\"\n\tFlagTreeID                         = \"tree_id\"\n\tFlagBranchID                       = \"branch_id\"\n\tFlagNumberOfShards                 = \"number_of_shards\"\n\tFlagTargetCluster                  = \"target_cluster\"\n\tFlagSourceCluster                  = \"source_cluster\"\n\tFlagMinEventID                     = \"min_event_id\"\n\tFlagMaxEventID                     = \"max_event_id\"\n\tFlagEndEventVersion                = \"end_event_version\"\n\tFlagTaskList                       = \"tasklist\"\n\tFlagTaskListType                   = \"tasklisttype\"\n\tFlagWorkflowIDReusePolicy          = \"workflowidreusepolicy\"\n\tFlagCronSchedule                   = \"cron\"\n\tFlagWorkflowType                   = \"workflow_type\"\n\tFlagWorkflowStatus                 = \"status\"\n\tFlagExecutionTimeout               = \"execution_timeout\"\n\tFlagDecisionTimeout                = \"decision_timeout\"\n\tFlagContextTimeout                 = \"context_timeout\"\n\tFlagInput                          = \"input\"\n\tFlagInputFile                      = \"input_file\"\n\tFlagInputEncoding                  = \"encoding\"\n\tFlagSignalInput                    = \"signal_input\"\n\tFlagSignalInputFile                = \"signal_input_file\"\n\tFlagExcludeFile                    = \"exclude_file\"\n\tFlagInputSeparator                 = \"input_separator\"\n\tFlagParallelism                    = \"input_parallelism\"\n\tFlagParallismDeprecated            = \"input_parallism\" // typo, replaced by FlagParallelism\n\tFlagScanType                       = \"scan_type\"\n\tFlagInvariantCollection            = \"invariant_collection\"\n\tFlagSkipCurrentOpen                = \"skip_current_open\"\n\tFlagSkipCurrentCompleted           = \"skip_current_completed\"\n\tFlagSkipBaseIsNotCurrent           = \"skip_base_is_not_current\"\n\tFlagDryRun                         = \"dry_run\"\n\tFlagNonDeterministicOnly           = \"only_non_deterministic\"\n\tFlagInputTopic                     = \"input_topic\"\n\tFlagHostFile                       = \"host_file\"\n\tFlagCluster                        = \"cluster\"\n\tFlagInputCluster                   = \"input_cluster\"\n\tFlagStartOffset                    = \"start_offset\"\n\tFlagTopic                          = \"topic\"\n\tFlagGroup                          = \"group\"\n\tFlagResult                         = \"result\"\n\tFlagIdentity                       = \"identity\"\n\tFlagDetail                         = \"detail\"\n\tFlagReason                         = \"reason\"\n\tFlagOpen                           = \"open\"\n\tFlagMore                           = \"more\"\n\tFlagAll                            = \"all\"\n\tFlagPrefix                         = \"prefix\"\n\tFlagDeprecated                     = \"deprecated\"\n\tFlagForce                          = \"force\"\n\tFlagPageID                         = \"page_id\"\n\tFlagPageSize                       = \"pagesize\"\n\tFlagLimit                          = \"limit\"\n\tFlagEarliestTime                   = \"earliest_time\"\n\tFlagLatestTime                     = \"latest_time\"\n\tFlagPrintEventVersion              = \"print_event_version\"\n\tFlagPrintFullyDetail               = \"print_full\"\n\tFlagPrintRawTime                   = \"print_raw_time\"\n\tFlagPrintRaw                       = \"print_raw\"\n\tFlagPrintDateTime                  = \"print_datetime\"\n\tFlagPrintMemo                      = \"print_memo\"\n\tFlagPrintSearchAttr                = \"print_search_attr\"\n\tFlagPrintCron                      = \"print_cron\"\n\tFlagPrintJSON                      = \"print_json\" // Deprecated: use --format json\n\tFlagDescription                    = \"description\"\n\tFlagOwnerEmail                     = \"owner_email\"\n\tFlagRetentionDays                  = \"retention\"\n\tFlagHistoryArchivalStatus          = \"history_archival_status\"\n\tFlagHistoryArchivalURI             = \"history_uri\"\n\tFlagVisibilityArchivalStatus       = \"visibility_archival_status\"\n\tFlagVisibilityArchivalURI          = \"visibility_uri\"\n\tFlagName                           = \"name\"\n\tFlagOutputFilename                 = \"output_filename\"\n\tFlagOutputFormat                   = \"output\"\n\tFlagQueryType                      = \"query_type\"\n\tFlagQueryRejectCondition           = \"query_reject_condition\"\n\tFlagQueryConsistencyLevel          = \"query_consistency_level\"\n\tFlagShowDetail                     = \"show_detail\"\n\tFlagActiveClusterName              = \"active_cluster\"\n\tFlagActiveClusters                 = \"active_clusters\"\n\tFlagActiveClustersJSON             = \"active_clusters_json\"\n\tFlagClusters                       = \"clusters\"\n\tFlagFailoverReason                 = \"reason\"\n\tFlagIsGlobalDomain                 = \"global_domain\" // active-passive domain\n\tFlagDomainData                     = \"domain_data\"\n\tFlagEventID                        = \"event_id\"\n\tFlagActivityID                     = \"activity_id\"\n\tFlagMaxFieldLength                 = \"max_field_length\"\n\tFlagSecurityToken                  = \"security_token\"\n\tFlagSkipErrorMode                  = \"skip_errors\"\n\tFlagRemote                         = \"remote\"\n\tFlagTimerType                      = \"timer_type\"\n\tFlagHeadersMode                    = \"headers\"\n\tFlagMessageType                    = \"message_type\"\n\tFlagURL                            = \"url\"\n\tFlagIndex                          = \"index\"\n\tFlagBatchSize                      = \"batch_size\"\n\tFlagMemoKey                        = \"memo_key\"\n\tFlagMemo                           = \"memo\"\n\tFlagMemoFile                       = \"memo_file\"\n\tFlagSearchAttributesKey            = \"search_attr_key\"\n\tFlagSearchAttributesVal            = \"search_attr_value\"\n\tFlagSearchAttributesType           = \"search_attr_type\"\n\tFlagAddBadBinary                   = \"add_bad_binary\"\n\tFlagRemoveBadBinary                = \"remove_bad_binary\"\n\tFlagResetType                      = \"reset_type\"\n\tFlagDecisionOffset                 = \"decision_offset\"\n\tFlagResetPointsOnly                = \"reset_points_only\"\n\tFlagResetBadBinaryChecksum         = \"reset_bad_binary_checksum\"\n\tFlagSkipSignalReapply              = \"skip_signal_reapply\"\n\tFlagListQuery                      = \"query\"\n\tFlagExcludeWorkflowIDByQuery       = \"exclude_query\"\n\tFlagBatchType                      = \"batch_type\"\n\tFlagSignalName                     = \"signal_name\"\n\tFlagTaskID                         = \"task_id\"\n\tFlagTaskType                       = \"task_type\"\n\tFlagTaskVisibilityTimestamp        = \"task_timestamp\"\n\tFlagQueueType                      = \"queue_type\"\n\tFlagStartingRPS                    = \"starting_rps\"\n\tFlagRPS                            = \"rps\"\n\tFlagRPSScaleUpSeconds              = \"rps_scale_up_seconds\"\n\tFlagJobID                          = \"job_id\"\n\tFlagYes                            = \"yes\"\n\tFlagServiceConfigDir               = \"service_config_dir\"\n\tFlagServiceEnv                     = \"service_env\"\n\tFlagServiceZone                    = \"service_zone\"\n\tFlagEnableTLS                      = \"tls\"\n\tFlagTLSCertPath                    = \"tls_cert_path\"\n\tFlagTLSKeyPath                     = \"tls_key_path\"\n\tFlagTLSCaPath                      = \"tls_ca_path\"\n\tFlagTLSEnableHostVerification      = \"tls_enable_host_verification\"\n\tFlagDLQType                        = \"dlq_type\"\n\tFlagMaxMessageCount                = \"max_message_count\"\n\tFlagLastMessageID                  = \"last_message_id\"\n\tFlagConcurrency                    = \"concurrency\"\n\tFlagReportRate                     = \"report_rate\"\n\tFlagLowerShardBound                = \"lower_shard_bound\"\n\tFlagUpperShardBound                = \"upper_shard_bound\"\n\tFlagInputDirectory                 = \"input_directory\"\n\tFlagSkipHistoryChecks              = \"skip_history_checks\"\n\tFlagFailoverType                   = \"failover_type\"\n\tFlagFailoverTimeout                = \"failover_timeout_seconds\"\n\tFlagActivityHeartBeatTimeout       = \"heart_beat_timeout_seconds\"\n\tFlagFailoverWaitTime               = \"failover_wait_time_second\"\n\tFlagFailoverBatchSize              = \"failover_batch_size\"\n\tFlagFailoverDomains                = \"domains\"\n\tFlagFailoverDrillWaitTime          = \"failover_drill_wait_second\"\n\tFlagFailoverDrill                  = \"failover_drill\"\n\tFlagRetryInterval                  = \"retry_interval\"\n\tFlagRetryAttempts                  = \"retry_attempts\"\n\tFlagMaxActivityRetries             = \"max_activity_retries\"\n\tFlagRetryExpiration                = \"retry_expiration\"\n\tFlagRetryBackoff                   = \"retry_backoff\"\n\tFlagRetryMaxInterval               = \"retry_max_interval\"\n\tFlagHeaderKey                      = \"header_key\"\n\tFlagHeaderValue                    = \"header_value\"\n\tFlagHeaderFile                     = \"header_file\"\n\tFlagStartDate                      = \"start_date\"\n\tFlagEndDate                        = \"end_date\"\n\tFlagDateFormat                     = \"date_format\"\n\tFlagShardMultiplier                = \"shard_multiplier\"\n\tFlagBucketSize                     = \"bucket_size\"\n\tDelayStartSeconds                  = \"delay_start_seconds\"\n\tJitterStartSeconds                 = \"jitter_start_seconds\"\n\tFirstRunAtTime                     = \"first_run_at_time\"\n\tFlagConnectionAttributes           = \"conn_attrs\"\n\tFlagJWT                            = \"jwt\"\n\tFlagJWTPrivateKey                  = \"jwt-private-key\"\n\tFlagDynamicConfigName              = \"name\"\n\tFlagDynamicConfigFilter            = \"filter\"\n\tFlagDynamicConfigValue             = \"value\"\n\tFlagTransport                      = \"transport\"\n\tFlagFormat                         = \"format\"\n\tFlagJSON                           = \"json\"\n\tFlagIsolationGroupSetDrains        = \"set-drains\"\n\tFlagIsolationGroupsRemoveAllDrains = \"remove-all-drains\"\n\tFlagSearchAttribute                = \"search_attr\"\n\tFlagNumReadPartitions              = \"num_read_partitions\"\n\tFlagNumWritePartitions             = \"num_write_partitions\"\n\tFlagCronOverlapPolicy              = \"cron_overlap_policy\"\n\tFlagClusterAttributeScope          = \"cluster_attribute_scope\"\n\tFlagClusterAttributeName           = \"cluster_attribute_name\"\n\n\tFlagClustersUsage = \"Clusters (example: --clusters clusterA,clusterB or --cl clusterA --cl clusterB)\"\n)\n\nvar flagsForExecution = []cli.Flag{\n\t&cli.StringFlag{\n\t\tName:    FlagWorkflowID,\n\t\tAliases: []string{\"w\", \"wid\"},\n\t\tUsage:   \"WorkflowID\",\n\t},\n\t&cli.StringFlag{\n\t\tName:    FlagRunID,\n\t\tAliases: []string{\"r\", \"rid\"},\n\t\tUsage:   \"RunID\",\n\t},\n}\n\nvar flagsOfExecutionForShow = []cli.Flag{\n\t&cli.StringFlag{\n\t\tName:    FlagWorkflowID,\n\t\tAliases: []string{\"w\", \"wid\"},\n\t\tUsage:   \"WorkflowID\",\n\t},\n\t&cli.StringFlag{\n\t\tName:    FlagRunID,\n\t\tAliases: []string{\"r\", \"rid\"},\n\t\tUsage:   \"RunID, required for archived history\",\n\t},\n}\n\nfunc getFlagsForShow() []cli.Flag {\n\treturn append(flagsOfExecutionForShow, getFlagsForShowID()...)\n}\n\nfunc getFlagsForShowID() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintDateTime,\n\t\t\tAliases: []string{\"pdt\"},\n\t\t\tUsage:   \"Print timestamp\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintRawTime,\n\t\t\tAliases: []string{\"prt\"},\n\t\t\tUsage:   \"Print raw timestamp\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagOutputFilename,\n\t\t\tAliases: []string{\"of\"},\n\t\t\tUsage:   \"Serialize history event to a file\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintFullyDetail,\n\t\t\tAliases: []string{\"pf\"},\n\t\t\tUsage:   \"Print fully event detail\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintEventVersion,\n\t\t\tAliases: []string{\"pev\"},\n\t\t\tUsage:   \"Print event version\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagEventID,\n\t\t\tAliases: []string{\"eid\"},\n\t\t\tUsage:   \"Print specific event details\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagMaxFieldLength,\n\t\t\tAliases: []string{\"maxl\"},\n\t\t\tUsage:   \"Maximum length for each attribute field\",\n\t\t\tValue:   defaultMaxFieldLength,\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:  FlagResetPointsOnly,\n\t\t\tUsage: \"Only show events that are eligible for reset\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagQueryConsistencyLevel,\n\t\t\tAliases: []string{\"qcl\"},\n\t\t\tUsage:   \"Optional flag to set query consistency level. Valid values are \\\"eventual\\\" and \\\"strong\\\"\",\n\t\t},\n\t}\n}\n\nfunc getFlagsForStart() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagTaskList,\n\t\t\tAliases: []string{\"tl\"},\n\t\t\tUsage:   \"TaskList\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowID,\n\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\tUsage:   \"WorkflowID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowType,\n\t\t\tAliases: []string{\"wt\"},\n\t\t\tUsage:   \"WorkflowTypeName\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagExecutionTimeout,\n\t\t\tAliases: []string{\"et\"},\n\t\t\tUsage:   \"Execution start to close timeout in seconds\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagDecisionTimeout,\n\t\t\tAliases: []string{\"dt\"},\n\t\t\tValue:   defaultDecisionTimeoutInSeconds,\n\t\t\tUsage:   \"Decision task start to close timeout in seconds\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagCronSchedule,\n\t\t\tUsage: \"Optional cron schedule for the workflow. Cron spec is as following: \\n\" +\n\t\t\t\t\"\\t┌───────────── minute (0 - 59) \\n\" +\n\t\t\t\t\"\\t│ ┌───────────── hour (0 - 23) \\n\" +\n\t\t\t\t\"\\t│ │ ┌───────────── day of the month (1 - 31) \\n\" +\n\t\t\t\t\"\\t│ │ │ ┌───────────── month (1 - 12) \\n\" +\n\t\t\t\t\"\\t│ │ │ │ ┌───────────── day of the week (0 - 6) (Sunday to Saturday) \\n\" +\n\t\t\t\t\"\\t│ │ │ │ │ \\n\" +\n\t\t\t\t\"\\t* * * * *\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagCronOverlapPolicy,\n\t\t\tAliases: []string{\"cop\"},\n\t\t\tUsage: \"Optional cron overlap policy for the workflow when a cron run overlaps with next scheduled run. \" +\n\t\t\t\t\"Available options: 0: Skip running if cron run overlaps, 1: Start new run immediately if previous run overlaps and completes\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagWorkflowIDReusePolicy,\n\t\t\tAliases: []string{\"wrp\"},\n\t\t\tUsage: \"Optional input to configure if the same workflow ID is allow to use for new workflow execution. \" +\n\t\t\t\t\"Available options: 0: AllowDuplicateFailedOnly, 1: AllowDuplicate, 2: RejectDuplicate, 3:TerminateIfRunning\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInput,\n\t\t\tAliases: []string{\"i\"},\n\t\t\tUsage:   \"Optional input for the workflow, in JSON format. If there are multiple parameters, concatenate them and separate by space.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInputFile,\n\t\t\tAliases: []string{\"if\"},\n\t\t\tUsage: \"Optional input for the workflow from JSON file. If there are multiple JSON, concatenate them and separate by space or newline. \" +\n\t\t\t\t\"Input from file will be overwrite by input from command line\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagMemoKey,\n\t\t\tUsage: \"Optional key of memo. If there are multiple keys, concatenate them and separate by space\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagMemo,\n\t\t\tUsage: \"Optional info that can be showed when list workflow, in JSON format. If there are multiple JSON, concatenate them and separate by space. \" +\n\t\t\t\t\"The order must be same as memo_key\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagMemoFile,\n\t\t\tUsage: \"Optional info that can be listed in list workflow, from JSON format file. If there are multiple JSON, concatenate them and separate by space or newline. \" +\n\t\t\t\t\"The order must be same as memo_key\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FlagHeaderKey,\n\t\t\tUsage: \"Optional key of header. If there are multiple keys, concatenate them and separate by space\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagHeaderValue,\n\t\t\tUsage: \"Optional info to propogate via workflow context, in JSON format. If there are multiple JSON, concatenate them and separate by space. \" +\n\t\t\t\t\"The order must be same as \" + FlagHeaderKey,\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagHeaderFile,\n\t\t\tUsage: \"Optional info to propogate via workflow context, from JSON format file. If there are multiple JSON, concatenate them and separate by space or newline. \" +\n\t\t\t\t\"The order must be same as \" + FlagHeaderKey,\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagSearchAttributesKey,\n\t\t\tUsage: \"Optional search attributes keys that can be be used in list query. If there are multiple keys, concatenate them and separate by |. \" +\n\t\t\t\t\"Use 'cluster get-search-attr' cmd to list legal keys.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagSearchAttributesVal,\n\t\t\tUsage: \"Optional search attributes value that can be be used in list query. If there are multiple keys, concatenate them and separate by |. \" +\n\t\t\t\t\"If value is array, use json array like [\\\"a\\\",\\\"b\\\"], [1,2], [\\\"true\\\",\\\"false\\\"], [\\\"2019-06-07T17:16:34-08:00\\\",\\\"2019-06-07T18:16:34-08:00\\\"]. \" +\n\t\t\t\t\"Use 'cluster get-search-attr' cmd to list legal keys and value types\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagRetryExpiration,\n\t\t\tUsage: \"Optional retry expiration in seconds. If set workflow will be retried for the specified period of time. retry_attempts and retry_expiration must not both be 0.\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagRetryAttempts,\n\t\t\tUsage: \"Optional retry attempts. If set workflow will be retried the specified amount of times. retry_attempts and retry_expiration must not both be 0.\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagRetryInterval,\n\t\t\tValue: 10,\n\t\t\tUsage: \"Optional retry interval in seconds.\",\n\t\t},\n\t\t&cli.Float64Flag{\n\t\t\tName:  FlagRetryBackoff,\n\t\t\tValue: 1.0,\n\t\t\tUsage: \"Optional retry backoff coefficient. Must be or equal or greater than 1.\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  FlagRetryMaxInterval,\n\t\t\tUsage: \"Optional retry maximum interval in seconds. If set will give an upper bound for retry interval. Must be equal or greater than retry interval.\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  DelayStartSeconds,\n\t\t\tUsage: \"Optional workflow start delay in seconds. If set workflow start will be delayed this many seconds\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:  JitterStartSeconds,\n\t\t\tUsage: \"Optional workflow start jitter in seconds. If set, workflow start will be jittered between 0-n seconds (after delay)\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:  FirstRunAtTime,\n\t\t\tUsage: \"Optional workflow's first run start time in RFC3339 format, like \\\"1970-01-01T00:00:00Z\\\". If set, first run of the workflow will start at the specified time.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagClusterAttributeScope,\n\t\t\tUsage:   \"Optional cluster attribute to specify how to select the active cluster. Examples might be 'region' or 'location'\",\n\t\t\tAliases: []string{\"cascope\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagClusterAttributeName,\n\t\t\tUsage:   \"Optional cluster attribute name, paired with a cluster attribute scope, to specify how to select the active cluster. This specifies which attribute to tie the workflow to, for example, if the scope is 'region' and the name is 'Lisbon' or 'San Francisco'\",\n\t\t\tAliases: []string{\"caname\"},\n\t\t},\n\t}\n}\n\nfunc getFlagsForRun() []cli.Flag {\n\tflagsForRun := []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagShowDetail,\n\t\t\tAliases: []string{\"sd\"},\n\t\t\tUsage:   \"Show event details\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagMaxFieldLength,\n\t\t\tAliases: []string{\"maxl\"},\n\t\t\tUsage:   \"Maximum length for each attribute field\",\n\t\t},\n\t}\n\tflagsForRun = append(getFlagsForStart(), flagsForRun...)\n\treturn flagsForRun\n}\n\nfunc getFlagsForSignal() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowID,\n\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\tUsage:   \"WorkflowID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagRunID,\n\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\tUsage:   \"RunID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagName,\n\t\t\tAliases: []string{\"n\"},\n\t\t\tUsage:   \"SignalName\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInput,\n\t\t\tAliases: []string{\"i\"},\n\t\t\tUsage:   \"Input for the signal, in JSON format.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInputFile,\n\t\t\tAliases: []string{\"if\"},\n\t\t\tUsage:   \"Input for the signal from JSON file.\",\n\t\t},\n\t}\n}\n\nfunc getFlagsForSignalWithStart() []cli.Flag {\n\treturn append(getFlagsForStart(),\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagName,\n\t\t\tAliases: []string{\"n\"},\n\t\t\tUsage:   \"SignalName\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSignalInput,\n\t\t\tAliases: []string{\"si\"},\n\t\t\tUsage:   \"Input for the signal, in JSON format.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagSignalInputFile,\n\t\t\tAliases: []string{\"sif\"},\n\t\t\tUsage:   \"Input for the signal from JSON file.\",\n\t\t})\n}\n\nfunc getFlagsForTerminate() []cli.Flag {\n\treturn append(flagsForExecution, &cli.StringFlag{\n\t\tName:    FlagReason,\n\t\tAliases: []string{\"re\"},\n\t\tUsage:   \"The reason you want to terminate the workflow\",\n\t})\n}\n\nfunc getFlagsForCancel() []cli.Flag {\n\treturn append(flagsForExecution, &cli.StringFlag{\n\t\tName:    FlagReason,\n\t\tAliases: []string{\"re\"},\n\t\tUsage:   \"The reason you want to cancel the workflow\",\n\t})\n}\n\nfunc getFormatFlag() cli.Flag {\n\treturn &cli.StringFlag{\n\t\tName:  FlagFormat,\n\t\tUsage: \"Format [table|json|<template>]; Use GoLang \\\"text/template\\\" syntax to format the output.\",\n\t}\n}\n\nfunc getCommonFlagsForVisibility() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintRawTime,\n\t\t\tAliases: []string{\"prt\"},\n\t\t\tUsage:   \"Print raw timestamp\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintDateTime,\n\t\t\tAliases: []string{\"pdt\"},\n\t\t\tUsage:   \"Print full date time in '2006-01-02T15:04:05Z07:00' format\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintMemo,\n\t\t\tAliases: []string{\"pme\"},\n\t\t\tUsage:   \"Print memo\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintSearchAttr,\n\t\t\tAliases: []string{\"psa\"},\n\t\t\tUsage:   \"Print search attributes\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintCron,\n\t\t\tAliases: []string{\"pcr\"},\n\t\t\tUsage:   \"Print cron schedule and scheduled execution time\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintFullyDetail,\n\t\t\tAliases: []string{\"pf\"},\n\t\t\tUsage:   \"Print full message without table format\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintJSON,\n\t\t\tAliases: []string{\"pjson\"},\n\t\t\tUsage:   \"Print in raw json format (DEPRECATED: instead use --format json)\",\n\t\t},\n\t\tgetFormatFlag(),\n\t}\n}\n\nfunc getFlagsForList() []cli.Flag {\n\tflagsForList := []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagMore,\n\t\t\tAliases: []string{\"m\"},\n\t\t\tUsage:   \"List more pages, default is to list one page of default page size 10\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagPageSize,\n\t\t\tAliases: []string{\"ps\"},\n\t\t\tValue:   10,\n\t\t\tUsage:   \"Result page size\",\n\t\t},\n\t}\n\tflagsForList = append(getFlagsForListAll(), flagsForList...)\n\treturn flagsForList\n}\n\nfunc getFlagsForListAll() []cli.Flag {\n\tflagsForListAll := []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagOpen,\n\t\t\tAliases: []string{\"op\"},\n\t\t\tUsage:   \"List for open workflow executions, default is to list for closed ones\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagEarliestTime,\n\t\t\tAliases: []string{\"et\"},\n\t\t\tUsage: \"EarliestTime of start time, supported formats are '2006-01-02T15:04:05+07:00', raw UnixNano and \" +\n\t\t\t\t\"time range (N<duration>), where 0 < N < 1000000 and duration (full-notation/short-notation) can be second/s, \" +\n\t\t\t\t\"minute/m, hour/h, day/d, week/w, month/M or year/y. For example, '15minute' or '15m' implies last 15 minutes.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagLatestTime,\n\t\t\tAliases: []string{\"lt\"},\n\t\t\tUsage: \"LatestTime of start time, supported formats are '2006-01-02T15:04:05+07:00', raw UnixNano and \" +\n\t\t\t\t\"time range (N<duration>), where 0 < N < 1000000 and duration (in full-notation/short-notation) can be second/s, \" +\n\t\t\t\t\"minute/m, hour/h, day/d, week/w, month/M or year/y. For example, '15minute' or '15m' implies last 15 minutes\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowID,\n\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\tUsage:   \"WorkflowID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowType,\n\t\t\tAliases: []string{\"wt\"},\n\t\t\tUsage:   \"WorkflowTypeName\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowStatus,\n\t\t\tAliases: []string{\"s\"},\n\t\t\tUsage:   \"Closed workflow status [completed, failed, canceled, terminated, continued_as_new, timed_out]\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagListQuery,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage: \"Optional SQL like query for use of search attributes. NOTE: using query will ignore all other filter flags including: \" +\n\t\t\t\t\"[open, earliest_time, latest_time, workflow_id, workflow_type]\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName: FlagExcludeWorkflowIDByQuery,\n\t\t\tUsage: \"Another optional SQL like query, but for excluding the results by workflowIDs. This is useful because a single query cannot do join operation. One use case is to \" +\n\t\t\t\t\"find failed workflows excluding any workflow that has another run that is open or completed.\",\n\t\t},\n\t}\n\tflagsForListAll = append(getCommonFlagsForVisibility(), flagsForListAll...)\n\treturn flagsForListAll\n}\n\nfunc getFlagsForScan() []cli.Flag {\n\tflagsForScan := []cli.Flag{\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagPageSize,\n\t\t\tAliases: []string{\"ps\"},\n\t\t\tValue:   2000,\n\t\t\tUsage:   \"Page size for each Scan API call\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagListQuery,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"Optional SQL like query\",\n\t\t},\n\t}\n\tflagsForScan = append(getCommonFlagsForVisibility(), flagsForScan...)\n\treturn flagsForScan\n}\n\nfunc getFlagsForListArchived() []cli.Flag {\n\tflagsForListArchived := []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagListQuery,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"SQL like query. Please check the documentation of the visibility archiver used by your domain for detailed instructions\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagPageSize,\n\t\t\tAliases: []string{\"ps\"},\n\t\t\tValue:   100,\n\t\t\tUsage:   \"Count of visibility records included in a single page, default to 100\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagAll,\n\t\t\tAliases: []string{\"a\"},\n\t\t\tUsage:   \"List all pages\",\n\t\t},\n\t}\n\tflagsForListArchived = append(getCommonFlagsForVisibility(), flagsForListArchived...)\n\treturn flagsForListArchived\n}\n\nfunc getFlagsForCount() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagListQuery,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"Optional SQL like query. e.g count all open workflows 'CloseTime = missing'; 'WorkflowType=\\\"wtype\\\" and CloseTime > 0'\",\n\t\t},\n\t}\n}\n\nfunc getFlagsForQuery() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagWorkflowID,\n\t\t\tAliases: []string{\"w\", \"wid\"},\n\t\t\tUsage:   \"WorkflowID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagRunID,\n\t\t\tAliases: []string{\"r\", \"rid\"},\n\t\t\tUsage:   \"RunID\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagQueryType,\n\t\t\tAliases: []string{\"qt\"},\n\t\t\tUsage:   \"The query type you want to run\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInput,\n\t\t\tAliases: []string{\"i\"},\n\t\t\tUsage:   \"Optional input for the query, in JSON format. If there are multiple parameters, concatenate them and separate by space.\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagInputFile,\n\t\t\tAliases: []string{\"if\"},\n\t\t\tUsage: \"Optional input for the query from JSON file. If there are multiple JSON, concatenate them and separate by space or newline. \" +\n\t\t\t\t\"Input from file will be overwrite by input from command line\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagQueryRejectCondition,\n\t\t\tAliases: []string{\"qrc\"},\n\t\t\tUsage:   \"Optional flag to reject queries based on workflow state. Valid values are \\\"not_open\\\" and \\\"not_completed_cleanly\\\"\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagQueryConsistencyLevel,\n\t\t\tAliases: []string{\"qcl\"},\n\t\t\tUsage:   \"Optional flag to set query consistency level. Valid values are \\\"eventual\\\" and \\\"strong\\\"\",\n\t\t},\n\t}\n}\n\n// all flags of query except QueryType\nfunc getFlagsForStack() []cli.Flag {\n\tflags := getFlagsForQuery()\n\tfor i := 0; i < len(flags); i++ {\n\t\tif flags[i].Names()[0] == FlagQueryType {\n\t\t\treturn append(flags[:i], flags[i+1:]...)\n\t\t}\n\t}\n\treturn flags\n}\n\nfunc getFlagsForDescribe() []cli.Flag {\n\treturn append(flagsForExecution, getFlagsForDescribeID()...)\n}\n\nfunc getFlagsForDescribeID() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagPrintRaw,\n\t\t\tAliases: []string{\"praw\"},\n\t\t\tUsage:   \"Print properties as they are stored\",\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:  FlagResetPointsOnly,\n\t\t\tUsage: \"Only show auto-reset points\",\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    FlagQueryConsistencyLevel,\n\t\t\tAliases: []string{\"qcl\"},\n\t\t\tUsage:   \"Optional flag to set query consistency level. Valid values are \\\"eventual\\\" and \\\"strong\\\"\",\n\t\t},\n\t}\n}\n\nfunc getFlagsForObserve() []cli.Flag {\n\treturn append(flagsForExecution, getFlagsForObserveID()...)\n}\n\nfunc getFlagsForObserveID() []cli.Flag {\n\treturn []cli.Flag{\n\t\t&cli.BoolFlag{\n\t\t\tName:    FlagShowDetail,\n\t\t\tAliases: []string{\"sd\"},\n\t\t\tUsage:   \"Optional show event details\",\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    FlagMaxFieldLength,\n\t\t\tAliases: []string{\"maxl\"},\n\t\t\tUsage:   \"Optional maximum length for each attribute field when show details\",\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/histogram.go",
    "content": "// The MIT License (MIT)\n//\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"fmt\"\n\t\"io\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n)\n\nconst (\n\tbar          = \"*\"\n\tdefaultWidth = 100\n)\n\ntype counter struct {\n\tkey   string\n\tcount int\n}\n\n// Histogram holds the occurrence count for each key\ntype Histogram struct {\n\tmaxCount int    // used to format output\n\tmaxKey   string // used to format output\n\n\tcounters []*counter\n}\n\n// NewHistogram creates a new Histogram\nfunc NewHistogram() *Histogram {\n\treturn &Histogram{\n\t\tmaxCount: defaultWidth,\n\t\tmaxKey:   \"Bucket\",\n\t}\n}\n\n// Add will increment occurrence count of the key\nfunc (h *Histogram) Add(key string) {\n\tvar found bool\n\tfor _, c := range h.counters {\n\t\tif c.key == key {\n\t\t\tfound = true\n\t\t\tc.count++\n\t\t\tif c.count > h.maxCount {\n\t\t\t\th.maxCount = c.count\n\t\t\t}\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif !found {\n\t\th.counters = append(h.counters, &counter{\n\t\t\tkey:   key,\n\t\t\tcount: 1,\n\t\t})\n\t}\n\n\tif len(key) > len(h.maxKey) {\n\t\th.maxKey = key\n\t}\n}\n\nfunc (h *Histogram) addMultiplier(multiplier int) {\n\tfor _, counter := range h.counters {\n\t\tcounter.count = counter.count * multiplier\n\t}\n\n\th.maxCount = h.maxCount * multiplier\n}\n\n// Print will output histogram with key and counter information.\nfunc (h *Histogram) Print(output io.Writer, multiplier int) error {\n\th.addMultiplier(multiplier)\n\tsort.Sort(h)\n\n\tkeyLength := len(h.maxKey)\n\tcountLength := len(strconv.FormatInt(int64(h.maxCount), 10))\n\n\toutput.Write([]byte(fmt.Sprintf(\"%-*s %*s\\n\", keyLength, \"Bucket\", countLength, \"Count\")))\n\n\tfor _, c := range h.counters {\n\t\tw := (defaultWidth - keyLength - countLength) * c.count / h.maxCount\n\t\toutput.Write([]byte(fmt.Sprintf(\"%-*s %*d %s\\n\", keyLength, c.key, countLength, c.count, strings.Repeat(bar, w))))\n\t}\n\n\treturn nil\n}\n\nfunc (h *Histogram) Len() int { return len(h.counters) }\n\nfunc (h *Histogram) Less(i, j int) bool {\n\treturn h.counters[i].key < h.counters[j].key\n}\n\nfunc (h *Histogram) Swap(i, j int) {\n\th.counters[j], h.counters[i] = h.counters[i], h.counters[j]\n}\n"
  },
  {
    "path": "tools/cli/histogram_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"strings\"\n\t\"testing\"\n)\n\n// TestNewHistogram tests the creation of a new Histogram\nfunc TestNewHistogram(t *testing.T) {\n\th := NewHistogram()\n\tif h.maxCount != defaultWidth || h.maxKey != \"Bucket\" {\n\t\tt.Errorf(\"NewHistogram() failed. Got maxCount = %d, maxKey = %s, want maxCount = %d, maxKey = Bucket\",\n\t\t\th.maxCount, h.maxKey, defaultWidth)\n\t}\n}\n\n// TestAdd tests the Add function of Histogram\nfunc TestHistogram_Add(t *testing.T) {\n\th := NewHistogram()\n\n\th.Add(\"key1\")\n\tif len(h.counters) != 1 {\n\t\tt.Errorf(\"Add() failed. Expected 1 counter, got %d\", len(h.counters))\n\t}\n\n\th.Add(\"key1\")\n\tif h.counters[0].count != 2 {\n\t\tt.Errorf(\"Add() failed. Expected count of 2, got %d\", h.counters[0].count)\n\t}\n\n\th.Add(\"key2\")\n\tif len(h.counters) != 2 {\n\t\tt.Errorf(\"Add() failed. Expected 2 counters, got %d\", len(h.counters))\n\t}\n\n\t// Check if maxKey is still \"Bucket\" because \"key1\" and \"key2\" are shorter than \"Bucket\"\n\tif h.maxKey != \"Bucket\" {\n\t\tt.Errorf(\"Add() failed. Expected maxKey 'Bucket', got %s\", h.maxKey)\n\t}\n\n\t// Add a longer key to test maxKey update\n\tlongKey := \"longerKeyThanBucket\"\n\th.Add(longKey)\n\n\tif h.maxKey != longKey {\n\t\tt.Errorf(\"Add() failed. Expected maxKey '%s', got %s\", longKey, h.maxKey)\n\t}\n}\n\n// TestAddWithMaxCount tests the Add function and covers the maxCount update\nfunc TestHistogram_AddWithMaxCount(t *testing.T) {\n\th := NewHistogram()\n\n\t// Add \"key1\" multiple times to exceed the initial maxCount (defaultWidth)\n\tfor i := 0; i < defaultWidth+1; i++ {\n\t\th.Add(\"key1\")\n\t}\n\n\t// Check if maxCount has been updated to the new value\n\tif h.maxCount != defaultWidth+1 {\n\t\tt.Errorf(\"Add() failed. Expected maxCount to be %d, got %d\", defaultWidth+1, h.maxCount)\n\t}\n\n\t// Add another key and check that maxCount doesn't change unless exceeded\n\th.Add(\"key2\")\n\tif h.maxCount != defaultWidth+1 {\n\t\tt.Errorf(\"Add() failed. Expected maxCount to remain %d, got %d\", defaultWidth+1, h.maxCount)\n\t}\n}\n\n// TestAddMultiplier tests the addMultiplier function\nfunc TestHistogram_AddMultiplier(t *testing.T) {\n\th := NewHistogram()\n\n\th.Add(\"key1\")\n\th.Add(\"key2\")\n\n\th.addMultiplier(2)\n\n\tif h.counters[0].count != 2 || h.counters[1].count != 2 {\n\t\tt.Errorf(\"addMultiplier() failed. Expected count of 2, got %d and %d\",\n\t\t\th.counters[0].count, h.counters[1].count)\n\t}\n}\n\n// TestPrint tests the Print function with multiplier\nfunc TestHistogram_Print(t *testing.T) {\n\th := NewHistogram()\n\n\t// Add sample keys to the histogram\n\th.Add(\"key1\")\n\th.Add(\"key2\")\n\n\t// Use a buffer to capture the output\n\tvar buf bytes.Buffer\n\n\t// Call the Print function with the buffer as the output writer and a multiplier\n\terr := h.Print(&buf, 2)\n\tif err != nil {\n\t\tt.Errorf(\"Print() failed. Expected no error, got %v\", err)\n\t}\n\n\toutput := buf.String()\n\n\t// Check if the output contains the expected headers and keys\n\tif !strings.Contains(output, \"Bucket\") {\n\t\tt.Errorf(\"Print() failed. Expected header 'Bucket' in output\")\n\t}\n\n\tif !strings.Contains(output, \"key1\") || !strings.Contains(output, \"key2\") {\n\t\tt.Errorf(\"Print() failed. Expected 'key1' and 'key2' in output\")\n\t}\n}\n\n// TestSortFunctions tests Len, Less, and Swap methods\nfunc TestHistogram_SortFunctions(t *testing.T) {\n\th := NewHistogram()\n\n\th.Add(\"key2\")\n\th.Add(\"key1\")\n\th.Add(\"key3\")\n\n\tif !h.Less(1, 0) {\n\t\tt.Errorf(\"Less() failed. Expected key1 < key2\")\n\t}\n\n\th.Swap(0, 2)\n\tif h.counters[0].key != \"key3\" {\n\t\tt.Errorf(\"Swap() failed. Expected key3 at index 0, got %s\", h.counters[0].key)\n\t}\n\n\tif h.Len() != 3 {\n\t\tt.Errorf(\"Len() failed. Expected length 3, got %d\", h.Len())\n\t}\n}\n"
  },
  {
    "path": "tools/cli/isolation-groups.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"text/tabwriter\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\nfunc AdminGetGlobalIsolationGroups(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context:\", err)\n\t}\n\treq := &types.GetGlobalIsolationGroupsRequest{}\n\tigs, err := adminClient.GetGlobalIsolationGroups(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to get isolation-groups:\", err)\n\t}\n\n\tformat := c.String(FlagFormat)\n\tswitch format {\n\tcase \"json\":\n\t\tprettyPrintJSONObject(getDeps(c).Output(), igs.IsolationGroups.ToPartitionList())\n\tdefault:\n\t\tgetDeps(c).Output().Write(renderIsolationGroups(igs.IsolationGroups))\n\t}\n\treturn nil\n}\n\nfunc AdminUpdateGlobalIsolationGroups(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context:\", err)\n\t}\n\terr = validateIsolationGroupUpdateArgs(\n\t\tc.String(FlagDomain),\n\t\tc.StringSlice(FlagIsolationGroupSetDrains),\n\t\tc.String(FlagJSON),\n\t\tc.Bool(FlagIsolationGroupsRemoveAllDrains),\n\t\tfalse,\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"invalid args:\", err)\n\t}\n\n\tcfg, err := parseIsolationGroupCliInputCfg(\n\t\tc.StringSlice(FlagIsolationGroupSetDrains),\n\t\tc.String(FlagJSON),\n\t\tc.Bool(FlagIsolationGroupsRemoveAllDrains),\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to parse input:\", err)\n\t}\n\n\t_, err = adminClient.UpdateGlobalIsolationGroups(ctx, &types.UpdateGlobalIsolationGroupsRequest{\n\t\tIsolationGroups: *cfg,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to update isolation-groups\", fmt.Errorf(\"used %#v, got %v\", cfg, err))\n\t}\n\treturn nil\n}\n\nfunc AdminGetDomainIsolationGroups(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain := c.String(FlagDomain)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context:\", err)\n\t}\n\n\treq := &types.GetDomainIsolationGroupsRequest{\n\t\tDomain: domain,\n\t}\n\tigs, err := adminClient.GetDomainIsolationGroups(ctx, req)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to get isolation-groups:\", err)\n\t}\n\n\tformat := c.String(FlagFormat)\n\tswitch format {\n\tcase \"json\":\n\t\tprettyPrintJSONObject(getDeps(c).Output(), igs.IsolationGroups.ToPartitionList())\n\tdefault:\n\t\tgetDeps(c).Output().Write([]byte(renderIsolationGroups(igs.IsolationGroups)))\n\t}\n\treturn nil\n}\n\nfunc AdminUpdateDomainIsolationGroups(c *cli.Context) error {\n\tadminClient, err := getDeps(c).ServerAdminClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain := c.String(FlagDomain)\n\n\terr = validateIsolationGroupUpdateArgs(\n\t\tc.String(FlagDomain),\n\t\tc.StringSlice(FlagIsolationGroupSetDrains),\n\t\tc.String(FlagJSON),\n\t\tc.Bool(FlagIsolationGroupsRemoveAllDrains),\n\t\ttrue,\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"invalid args:\", err)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context:\", err)\n\t}\n\n\tcfg, err := parseIsolationGroupCliInputCfg(\n\t\tc.StringSlice(FlagIsolationGroupSetDrains),\n\t\tc.String(FlagJSON),\n\t\tc.Bool(FlagIsolationGroupsRemoveAllDrains),\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to parse input:\", err)\n\t}\n\n\treq := &types.UpdateDomainIsolationGroupsRequest{\n\t\tDomain:          domain,\n\t\tIsolationGroups: *cfg,\n\t}\n\t_, err = adminClient.UpdateDomainIsolationGroups(ctx, req)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"failed to update isolation-groups\", fmt.Errorf(\"used %#v, got %v\", req, err))\n\t}\n\treturn nil\n}\n\nfunc validateIsolationGroupUpdateArgs(\n\tdomainArgs string,\n\tsetDrainsArgs []string,\n\tjsonCfgArgs string,\n\tremoveAllDrainsArgs bool,\n\trequiresDomain bool,\n) error {\n\tif requiresDomain {\n\t\tif domainArgs == \"\" {\n\t\t\treturn fmt.Errorf(\"the --domain flag is required\")\n\t\t}\n\t}\n\n\tif len(setDrainsArgs) == 0 &&\n\t\tjsonCfgArgs == \"\" &&\n\t\t!removeAllDrainsArgs {\n\t\treturn fmt.Errorf(\"need to specify either %q, %q or %q flags\",\n\t\t\tFlagIsolationGroupSetDrains,\n\t\t\tFlagJSON,\n\t\t\tFlagIsolationGroupsRemoveAllDrains,\n\t\t)\n\t}\n\n\tif removeAllDrainsArgs && (len(setDrainsArgs) != 0 || jsonCfgArgs != \"\") {\n\t\treturn fmt.Errorf(\"specify either remove or set-drains, not both\")\n\t}\n\n\treturn nil\n}\n\nfunc parseIsolationGroupCliInputCfg(drains []string, jsonInput string, removeAllDrains bool) (*types.IsolationGroupConfiguration, error) {\n\n\treq := types.IsolationGroupConfiguration{}\n\tif removeAllDrains {\n\t\treturn &req, nil\n\t}\n\n\tif len(drains) != 0 {\n\t\treq := types.IsolationGroupConfiguration{}\n\t\tfor _, drain := range drains {\n\t\t\treq[drain] = types.IsolationGroupPartition{\n\t\t\t\tName:  drain,\n\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t}\n\t\t}\n\t\treturn &req, nil\n\t}\n\n\tvar input []types.IsolationGroupPartition\n\terr := json.Unmarshal([]byte(jsonInput), &input)\n\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(`failed to marshal input. Trying to marshal []types.IsolationGroupPartition\n\nexamples:\n- []                                    # will remove all isolation groups\n- [{\"Name\": \"zone-123\", \"State\": 2}]    # drain zone-123\n\n%v`, err)\n\t}\n\tfor _, g := range input {\n\t\treq[g.Name] = g\n\t}\n\n\treturn &req, nil\n}\n\nfunc renderIsolationGroups(igs types.IsolationGroupConfiguration) []byte {\n\toutput := &bytes.Buffer{}\n\tw := tabwriter.NewWriter(output, 0, 0, 1, ' ', 0)\n\tfmt.Fprintln(w, \"Isolation Groups\\tState\")\n\tif len(igs) == 0 {\n\t\treturn []byte(\"-- No groups found --\\n\")\n\t}\n\tfor _, v := range igs.ToPartitionList() {\n\t\tfmt.Fprintf(w, \"%s\\t%s\\n\", v.Name, convertIsolationGroupStateToString(v.State))\n\t}\n\tw.Flush()\n\treturn output.Bytes()\n}\n\nfunc convertIsolationGroupStateToString(state types.IsolationGroupState) string {\n\tswitch state {\n\tcase types.IsolationGroupStateDrained:\n\t\treturn \"Drained\"\n\tcase types.IsolationGroupStateHealthy:\n\t\treturn \"Healthy\"\n\tdefault:\n\t\treturn fmt.Sprintf(\"Unknown state: %d\", state)\n\t}\n}\n"
  },
  {
    "path": "tools/cli/isolation_groups_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestValidateIsolationGroupArgs(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tdomainArgs          string\n\t\tsetDrainsArgs       []string\n\t\tjsonConfigArgs      string\n\t\tremoveAllDrainsArgs bool\n\n\t\trequiresDomain bool\n\t\texpectedErr    error\n\t}{\n\t\t\"valid inputs for doing a drain\": {\n\t\t\tdomainArgs:     \"some-domain\",\n\t\t\tsetDrainsArgs:  []string{\"zone-1\", \"zone-2\"},\n\t\t\tjsonConfigArgs: \"\",\n\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t\"valid json input\": {\n\t\t\tdomainArgs:     \"some-domain\",\n\t\t\tsetDrainsArgs:  nil,\n\t\t\tjsonConfigArgs: \"{}\",\n\n\t\t\texpectedErr: nil,\n\t\t},\n\t\t\"invalid - no domain\": {\n\t\t\tdomainArgs:     \"\",\n\t\t\tsetDrainsArgs:  nil,\n\t\t\tjsonConfigArgs: \"{}\",\n\t\t\trequiresDomain: true,\n\n\t\t\texpectedErr: errors.New(\"the --domain flag is required\"),\n\t\t},\n\t\t\"invalid - no config domain\": {\n\t\t\tdomainArgs:     \"domain\",\n\t\t\tsetDrainsArgs:  nil,\n\t\t\tjsonConfigArgs: \"\",\n\t\t\trequiresDomain: true,\n\n\t\t\texpectedErr: errors.New(\"need to specify either \\\"set-drains\\\", \\\"json\\\" or \\\"remove-all-drains\\\" flags\"),\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expectedErr, validateIsolationGroupUpdateArgs(\n\t\t\t\ttd.domainArgs,\n\t\t\t\ttd.setDrainsArgs,\n\t\t\t\ttd.jsonConfigArgs,\n\t\t\t\ttd.removeAllDrainsArgs,\n\t\t\t\ttd.requiresDomain))\n\t\t})\n\t}\n}\n\nfunc TestParseCliInput(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tsetDrainsArgs  []string\n\t\tjsonConfigArgs string\n\n\t\texpected    *types.IsolationGroupConfiguration\n\t\texpectedErr error\n\t}{\n\t\t\"valid inputs for doing a drain\": {\n\t\t\tsetDrainsArgs:  []string{\"zone-1\", \"zone-2\"},\n\t\t\tjsonConfigArgs: \"\",\n\n\t\t\texpected: &types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateDrained},\n\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: types.IsolationGroupStateDrained},\n\t\t\t},\n\t\t},\n\t\t\"valid json input\": {\n\t\t\tsetDrainsArgs:  nil,\n\t\t\tjsonConfigArgs: \"[{\\\"Name\\\": \\\"zone-1\\\", \\\"State\\\": 2}, {\\\"Name\\\": \\\"zone-2\\\", \\\"State\\\": 1}]\",\n\t\t\texpected: &types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {Name: \"zone-1\", State: types.IsolationGroupStateDrained},\n\t\t\t\t\"zone-2\": {Name: \"zone-2\", State: types.IsolationGroupStateHealthy},\n\t\t\t},\n\n\t\t\texpectedErr: nil,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tres, err := parseIsolationGroupCliInputCfg(\n\t\t\t\ttd.setDrainsArgs,\n\t\t\t\ttd.jsonConfigArgs,\n\t\t\t\tfalse,\n\t\t\t)\n\t\t\tassert.Equal(t, td.expected, res)\n\t\t\tassert.Equal(t, td.expectedErr, err)\n\t\t})\n\t}\n}\n\nfunc TestRenderIsolationGroupNormalOutput(t *testing.T) {\n\n\ttests := map[string]struct {\n\t\tinput          types.IsolationGroupConfiguration\n\t\texpectedOutput string\n\t}{\n\t\t\"valid inputs for doing a drain\": {\n\t\t\tinput: types.IsolationGroupConfiguration{\n\t\t\t\t\"zone-1\": {\n\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t},\n\t\t\t\t\"zone-2\": {\n\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-3-a-very-long-name\": {\n\t\t\t\t\tName:  \"zone-3-a-very-long-name\",\n\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t},\n\t\t\t\t\"zone-4\": {\n\t\t\t\t\tName:  \"zone-4\",\n\t\t\t\t\tState: 5,\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectedOutput: `Isolation Groups        State\nzone-1                  Healthy\nzone-2                  Drained\nzone-3-a-very-long-name Drained\nzone-4                  Unknown state: 5\n`,\n\t\t},\n\t\t\"nothing\": {\n\t\t\tinput: types.IsolationGroupConfiguration{},\n\t\t\texpectedOutput: `-- No groups found --\n`,\n\t\t},\n\t}\n\n\tfor name, td := range tests {\n\t\tt.Run(name, func(t *testing.T) {\n\t\t\tassert.Equal(t, td.expectedOutput, string(renderIsolationGroups(td.input)))\n\t\t})\n\t}\n}\n\nfunc TestAdminGetGlobalIsolationGroups(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\n\t// Table of test cases\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\tflagFormat       string\n\t\tmockDepsError    error\n\t\tmockContextError error\n\t}{\n\t\t{\n\t\t\tname: \"Success with JSON format\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\texpectedResponse := &types.GetGlobalIsolationGroupsResponse{\n\t\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\t\"zone-1\": {\n\t\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"zone-2\": {\n\t\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetGlobalIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tflagFormat:    \"json\",\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to get global isolation groups\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetGlobalIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"failed to get isolation-groups\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"failed to get isolation-groups\",\n\t\t\tflagFormat:    \"json\",\n\t\t},\n\t}\n\n\t// Loop through test cases\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t})\n\n\t\t\t// Create CLI context with flags\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagFormat, tt.flagFormat, \"Format flag\")\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call the function under test\n\t\t\terr := AdminGetGlobalIsolationGroups(c)\n\n\t\t\t// Check the expected outcome\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateGlobalIsolationGroups(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\n\t// Define table-driven tests\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\tflagDomain       string\n\t\tremoveAllDrains  bool\n\t\tmockDepsError    error\n\t\tmockContextError error\n\t\tvalidationError  error\n\t\tparseConfigError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateGlobalIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.UpdateGlobalIsolationGroupsResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:   \"\",\n\t\t\tflagDomain:      \"test-domain\",\n\t\t\tremoveAllDrains: true,\n\t\t},\n\t\t{\n\t\t\tname: \"parse failure\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t},\n\t\t\texpectedError:   \"invalid args:\",\n\t\t\tflagDomain:      \"test-domain\",\n\t\t\tremoveAllDrains: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t})\n\n\t\t\t// Set up CLI context with flags\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagDomain, tt.flagDomain, \"Domain flag\")\n\t\t\tset.Bool(FlagIsolationGroupsRemoveAllDrains, tt.removeAllDrains, \"RemoveAllDrains flag\")\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call the function under test\n\t\t\terr := AdminUpdateGlobalIsolationGroups(c)\n\n\t\t\t// Check the expected outcome\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminGetDomainIsolationGroups(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\n\t// Define table-driven tests\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\tflagDomain       string\n\t\tflagFormat       string\n\t\tmockDepsError    error\n\t\tmockContextError error\n\t}{\n\t\t{\n\t\t\tname: \"Success with JSON format\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\texpectedResponse := &types.GetDomainIsolationGroupsResponse{\n\t\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\t\"zone-1\": {\n\t\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"zone-2\": {\n\t\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tflagDomain:    \"test-domain\",\n\t\t\tflagFormat:    \"json\",\n\t\t},\n\t\t{\n\t\t\tname: \"Success with other format\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\texpectedResponse := &types.GetDomainIsolationGroupsResponse{\n\t\t\t\t\tIsolationGroups: types.IsolationGroupConfiguration{\n\t\t\t\t\t\t\"zone-1\": {\n\t\t\t\t\t\t\tName:  \"zone-1\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateHealthy,\n\t\t\t\t\t\t},\n\t\t\t\t\t\t\"zone-2\": {\n\t\t\t\t\t\t\tName:  \"zone-2\",\n\t\t\t\t\t\t\tState: types.IsolationGroupStateDrained,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(expectedResponse, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\tflagDomain:    \"test-domain\",\n\t\t\tflagFormat:    \"else\",\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to get domain isolation groups\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tGetDomainIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"failed to get isolation-groups\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError: \"failed to get isolation-groups\",\n\t\t\tflagDomain:    \"test-domain\",\n\t\t\tflagFormat:    \"json\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\t\t\tioHandler := &testIOHandler{}\n\n\t\t\t// Create mock app with clientFactoryMock, including any deps errors\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\n\t\t\t// Set up CLI context with flags\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagDomain, tt.flagDomain, \"Domain flag\")\n\t\t\tset.String(FlagFormat, tt.flagFormat, \"Format flag\")\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call the function under test\n\t\t\terr := AdminGetDomainIsolationGroups(c)\n\n\t\t\t// Check the expected outcome\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Contains(t, ioHandler.outputBytes.String(), \"zone-1\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestAdminUpdateDomainIsolationGroups(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\n\t// Define table-driven tests\n\ttests := []struct {\n\t\tname             string\n\t\tsetupMocks       func(*admin.MockClient)\n\t\texpectedError    string\n\t\tflagDomain       string\n\t\tflagJSON         string\n\t\tflagDrains       []string\n\t\tremoveAllDrains  bool\n\t\tmockContextError error\n\t\tvalidationError  error // simulate validation errors\n\t\tparseConfigError error\n\t}{\n\t\t{\n\t\t\tname: \"Success\",\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateDomainIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(&types.UpdateDomainIsolationGroupsResponse{}, nil).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:   \"\",\n\t\t\tflagDomain:      \"test-domain\",\n\t\t\tflagJSON:        `[{\"Name\": \"zone-123\", \"State\": 2}]`,\n\t\t\tflagDrains:      []string{\"zone-1\"},\n\t\t\tremoveAllDrains: false,\n\t\t},\n\t\t{\n\t\t\tname: \"Validation error\", // This simulates a failure due to invalid arguments\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call to mock admin client as validation fails\n\t\t\t},\n\t\t\texpectedError:   \"invalid args:\",\n\t\t\tflagDomain:      \"\",\n\t\t\tflagJSON:        \"\",\n\t\t\tflagDrains:      []string{},\n\t\t\tremoveAllDrains: false,\n\t\t\tvalidationError: fmt.Errorf(\"the --domain flag is required\"), // Simulate validation failure\n\t\t},\n\t\t{\n\t\t\tname: \"Parse config error\", // This simulates an error during parsing of the input config\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\t// No call setup for this test case as parsing fails\n\t\t\t},\n\t\t\texpectedError:    \"failed to parse input:\",\n\t\t\tflagDomain:       \"test-domain\",\n\t\t\tflagJSON:         `[{\"Name\": \"zone-123\", \"State\": \"123123\"}]`,\n\t\t\tflagDrains:       []string{\"zone-1\"},\n\t\t\tremoveAllDrains:  false,\n\t\t\tparseConfigError: fmt.Errorf(\"config parsing failed\"), // Simulate config parsing failure\n\t\t},\n\t\t{\n\t\t\tname: \"Failed to update isolation groups\", // This simulates a failure when updating the isolation groups\n\t\t\tsetupMocks: func(client *admin.MockClient) {\n\t\t\t\tclient.EXPECT().\n\t\t\t\t\tUpdateDomainIsolationGroups(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(nil, fmt.Errorf(\"update failed\")).\n\t\t\t\t\tTimes(1)\n\t\t\t},\n\t\t\texpectedError:   \"failed to update isolation-groups\",\n\t\t\tflagDomain:      \"test-domain\",\n\t\t\tflagJSON:        `[{\"Name\": \"zone-123\", \"State\": 2}]`,\n\t\t\tflagDrains:      []string{\"zone-1\"},\n\t\t\tremoveAllDrains: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Mock the admin client\n\t\t\tadminClient := admin.NewMockClient(mockCtrl)\n\n\t\t\t// Set up mocks for the current test case\n\t\t\ttt.setupMocks(adminClient)\n\n\t\t\t// Create mock app with clientFactoryMock\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverAdminClient: adminClient,\n\t\t\t})\n\n\t\t\t// Set up CLI context with flags\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tset.String(FlagDomain, tt.flagDomain, \"Domain flag\")\n\t\t\tset.String(FlagJSON, tt.flagJSON, \"JSON flag\")\n\t\t\tset.Bool(FlagIsolationGroupsRemoveAllDrains, tt.removeAllDrains, \"RemoveAllDrains flag\")\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call the function under test\n\t\t\terr := AdminUpdateDomainIsolationGroups(c)\n\n\t\t\t// Check the expected outcome\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/mock_manager_factory.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: github.com/uber/cadence/tools/cli (interfaces: ManagerFactory)\n//\n// Generated by this command:\n//\n//\tmockgen -package cli -destination mock_manager_factory.go -self_package github.com/uber/cadence/tools/cli github.com/uber/cadence/tools/cli ManagerFactory\n//\n\n// Package cli is a generated GoMock package.\npackage cli\n\nimport (\n\treflect \"reflect\"\n\n\tcli \"github.com/urfave/cli/v2\"\n\tgomock \"go.uber.org/mock/gomock\"\n\n\tpersistence \"github.com/uber/cadence/common/persistence\"\n\tclient \"github.com/uber/cadence/common/persistence/client\"\n\tinvariant \"github.com/uber/cadence/common/reconciliation/invariant\"\n)\n\n// MockManagerFactory is a mock of ManagerFactory interface.\ntype MockManagerFactory struct {\n\tctrl     *gomock.Controller\n\trecorder *MockManagerFactoryMockRecorder\n\tisgomock struct{}\n}\n\n// MockManagerFactoryMockRecorder is the mock recorder for MockManagerFactory.\ntype MockManagerFactoryMockRecorder struct {\n\tmock *MockManagerFactory\n}\n\n// NewMockManagerFactory creates a new mock instance.\nfunc NewMockManagerFactory(ctrl *gomock.Controller) *MockManagerFactory {\n\tmock := &MockManagerFactory{ctrl: ctrl}\n\tmock.recorder = &MockManagerFactoryMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockManagerFactory) EXPECT() *MockManagerFactoryMockRecorder {\n\treturn m.recorder\n}\n\n// initPersistenceFactory mocks base method.\nfunc (m *MockManagerFactory) initPersistenceFactory(c *cli.Context) (client.Factory, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initPersistenceFactory\", c)\n\tret0, _ := ret[0].(client.Factory)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initPersistenceFactory indicates an expected call of initPersistenceFactory.\nfunc (mr *MockManagerFactoryMockRecorder) initPersistenceFactory(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initPersistenceFactory\", reflect.TypeOf((*MockManagerFactory)(nil).initPersistenceFactory), c)\n}\n\n// initializeDomainManager mocks base method.\nfunc (m *MockManagerFactory) initializeDomainManager(c *cli.Context) (persistence.DomainManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initializeDomainManager\", c)\n\tret0, _ := ret[0].(persistence.DomainManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initializeDomainManager indicates an expected call of initializeDomainManager.\nfunc (mr *MockManagerFactoryMockRecorder) initializeDomainManager(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initializeDomainManager\", reflect.TypeOf((*MockManagerFactory)(nil).initializeDomainManager), c)\n}\n\n// initializeExecutionManager mocks base method.\nfunc (m *MockManagerFactory) initializeExecutionManager(c *cli.Context, shardID int) (persistence.ExecutionManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initializeExecutionManager\", c, shardID)\n\tret0, _ := ret[0].(persistence.ExecutionManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initializeExecutionManager indicates an expected call of initializeExecutionManager.\nfunc (mr *MockManagerFactoryMockRecorder) initializeExecutionManager(c, shardID any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initializeExecutionManager\", reflect.TypeOf((*MockManagerFactory)(nil).initializeExecutionManager), c, shardID)\n}\n\n// initializeHistoryManager mocks base method.\nfunc (m *MockManagerFactory) initializeHistoryManager(c *cli.Context) (persistence.HistoryManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initializeHistoryManager\", c)\n\tret0, _ := ret[0].(persistence.HistoryManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initializeHistoryManager indicates an expected call of initializeHistoryManager.\nfunc (mr *MockManagerFactoryMockRecorder) initializeHistoryManager(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initializeHistoryManager\", reflect.TypeOf((*MockManagerFactory)(nil).initializeHistoryManager), c)\n}\n\n// initializeInvariantManager mocks base method.\nfunc (m *MockManagerFactory) initializeInvariantManager(ivs []invariant.Invariant) (invariant.Manager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initializeInvariantManager\", ivs)\n\tret0, _ := ret[0].(invariant.Manager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initializeInvariantManager indicates an expected call of initializeInvariantManager.\nfunc (mr *MockManagerFactoryMockRecorder) initializeInvariantManager(ivs any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initializeInvariantManager\", reflect.TypeOf((*MockManagerFactory)(nil).initializeInvariantManager), ivs)\n}\n\n// initializeShardManager mocks base method.\nfunc (m *MockManagerFactory) initializeShardManager(c *cli.Context) (persistence.ShardManager, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"initializeShardManager\", c)\n\tret0, _ := ret[0].(persistence.ShardManager)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// initializeShardManager indicates an expected call of initializeShardManager.\nfunc (mr *MockManagerFactoryMockRecorder) initializeShardManager(c any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"initializeShardManager\", reflect.TypeOf((*MockManagerFactory)(nil).initializeShardManager), c)\n}\n"
  },
  {
    "path": "tools/cli/render.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"reflect\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"text/template\"\n\t\"time\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/urfave/cli/v2\"\n)\n\nconst (\n\tformatTable = \"table\"\n\tformatJSON  = \"json\"\n\n\ttemplateTable = \"{{table .}}\\n\"\n\ttemplateJSON  = \"{{json .}}\\n\"\n\n\tdefaultSliceSeparator   = \", \"\n\tdefaultMapSeparator     = \", \"\n\tdefaultMapItemSeparator = \":\"\n)\n\nvar tableHeaderBlue = tablewriter.Colors{tablewriter.FgHiBlueColor}\n\n// RenderOptions allows passing optional flags for altering rendered table\ntype RenderOptions struct {\n\t// OptionalColumns may contain column header names which can be hidden\n\tOptionalColumns map[string]bool\n\n\t// Border specified whether to render table border\n\tBorder bool\n\n\t// Custom, per column alignment\n\tColumnAlignment []int\n\n\t// Color will use coloring characters while printing table\n\tColor bool\n\n\t// PrintRawTime will print time as int64 unix nanos\n\tPrintRawTime bool\n\t// PrintDateTime will print both date & time\n\tPrintDateTime bool\n\n\t// DefaultTemplate (if specified) will be used to render data when not --format flag is given\n\tDefaultTemplate string\n}\n\n// Render is an entry point for presentation layer. It uses --format flag to determine output format.\nfunc Render(c *cli.Context, data interface{}, opts RenderOptions) (err error) {\n\tdefer func() {\n\t\tif err != nil {\n\t\t\tfmt.Errorf(\"failed to render: %w\", err)\n\t\t}\n\t}()\n\n\t// For now always output to stdout\n\tw := getDeps(c).Output()\n\n\ttemplate := opts.DefaultTemplate\n\n\t// Handle template shorthands\n\tswitch format := c.String(FlagFormat); format {\n\tcase formatJSON:\n\t\ttemplate = templateJSON\n\tcase formatTable:\n\t\ttemplate = templateTable\n\tdefault:\n\t\tif len(format) > 0 {\n\t\t\tswitch kind := reflect.ValueOf(data).Kind(); kind {\n\t\t\tcase reflect.Slice:\n\t\t\t\ttemplate = fmt.Sprintf(\"{{range .}}%s\\n{{end}}\", format)\n\t\t\tcase reflect.Struct:\n\t\t\t\ttemplate = fmt.Sprintf(\"%s\\n\", format)\n\t\t\tdefault:\n\t\t\t\treturn fmt.Errorf(\"data must be a struct or a slice, provided: %s\", kind)\n\t\t\t}\n\t\t}\n\t}\n\n\treturn RenderTemplate(w, data, template, opts)\n}\n\n// RenderTemplate uses golang text/template format to render data with user provided template\nfunc RenderTemplate(w io.Writer, data interface{}, tmpl string, opts RenderOptions) error {\n\tfns := map[string]interface{}{\n\t\t\"table\": func(data interface{}) (string, error) {\n\t\t\tsb := &strings.Builder{}\n\t\t\tif err := RenderTable(sb, data, opts); err != nil {\n\t\t\t\treturn \"\", err\n\t\t\t}\n\t\t\treturn sb.String(), nil\n\t\t},\n\t\t\"json\": func(data interface{}) (string, error) {\n\t\t\tencoded, err := json.MarshalIndent(data, \"\", \"  \")\n\t\t\treturn string(encoded), err\n\t\t},\n\t}\n\n\tt, err := template.New(\"\").Funcs(fns).Parse(tmpl)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"invalid template %q: %w\", tmpl, err)\n\t}\n\n\treturn t.Execute(w, data)\n}\n\n// RenderTable is generic function for rendering a slice of structs as a table\nfunc RenderTable(w io.Writer, data interface{}, opts RenderOptions) error {\n\tvalue := reflect.ValueOf(data)\n\n\tif value.Kind() == reflect.Ptr {\n\t\t// Nil pointer - nothing to render\n\t\tif value.IsNil() {\n\t\t\treturn nil\n\t\t}\n\n\t\t// Drop the pointer and start over\n\t\treturn RenderTable(w, value.Elem().Interface(), opts)\n\t}\n\n\tif value.Kind() != reflect.Slice {\n\t\t// If data is not a slice - wrap it with a slice of one element\n\t\tvalue = reflect.MakeSlice(reflect.SliceOf(reflect.TypeOf(data)), 0, 1)\n\t\tvalue = reflect.Append(value, reflect.ValueOf(data))\n\t}\n\n\t// We have a slice at this point\n\tslice := value\n\n\t// No elements - nothing to render\n\tif slice.Len() == 0 {\n\t\treturn nil\n\t}\n\n\tfirstElem := slice.Index(0)\n\tif firstElem.Kind() != reflect.Struct {\n\t\treturn fmt.Errorf(\"table slice element must be a struct, provided: %s\", firstElem.Kind())\n\t}\n\n\ttable := tablewriter.NewWriter(w)\n\ttable.SetBorder(opts.Border)\n\ttable.SetColumnSeparator(\"|\")\n\ttable.SetHeaderLine(opts.Border)\n\tif opts.ColumnAlignment != nil {\n\t\ttable.SetColumnAlignment(opts.ColumnAlignment)\n\t}\n\n\tfor r := 0; r < slice.Len(); r++ {\n\t\tvar row []string\n\t\tvar headers []string\n\t\tvar colors []tablewriter.Colors\n\n\t\telem := slice.Index(r)\n\t\tfor f := 0; f < elem.NumField(); f++ {\n\t\t\ttag := elem.Type().Field(f).Tag\n\n\t\t\theader := columnHeader(tag, opts)\n\t\t\tif header == \"\" {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif r == 0 {\n\t\t\t\theaders = append(headers, header)\n\t\t\t\tcolors = append(colors, tableHeaderBlue)\n\t\t\t}\n\n\t\t\trow = append(row, formatValue(elem.Field(f).Interface(), opts, tag))\n\t\t}\n\t\tif r == 0 {\n\t\t\ttable.SetHeader(headers)\n\t\t\tif opts.Color {\n\t\t\t\ttable.SetHeaderColor(colors...)\n\t\t\t}\n\t\t}\n\n\t\ttable.Append(row)\n\t}\n\n\ttable.Render()\n\n\treturn nil\n}\n\nfunc columnHeader(tag reflect.StructTag, opts RenderOptions) string {\n\theader, ok := tag.Lookup(\"header\")\n\tif !ok {\n\t\t// No header tag - do not display\n\t\treturn \"\"\n\t}\n\n\tif opts.OptionalColumns == nil {\n\t\t// No optional columns defined - display\n\t\treturn header\n\t}\n\n\tinclude, optional := opts.OptionalColumns[header]\n\tif !optional {\n\t\t// Display if it is non-optional\n\t\treturn header\n\t}\n\n\tif include {\n\t\t// Display if it is optional but included\n\t\treturn header\n\t}\n\n\t// Do not display optional and excluded\n\treturn \"\"\n}\n\nfunc formatValue(value interface{}, opts RenderOptions, tag reflect.StructTag) string {\n\tswitch v := value.(type) {\n\tcase time.Time:\n\t\treturn formatTime(v, opts)\n\tcase string:\n\t\treturn formatString(v, tag)\n\t}\n\n\tswitch v := reflect.ValueOf(value); v.Kind() {\n\tcase reflect.Slice:\n\t\treturn formatSlice(value, opts)\n\tcase reflect.Map:\n\t\treturn formatMap(value, opts)\n\t}\n\n\treturn fmt.Sprintf(\"%v\", value)\n}\n\nfunc formatSlice(value interface{}, opts RenderOptions) string {\n\tv := reflect.ValueOf(value)\n\tvalues := []string{}\n\tfor i := 0; i < v.Len(); i++ {\n\t\tvalues = append(values, formatValue(v.Index(i).Interface(), opts, \"\"))\n\t}\n\treturn strings.Join(values, defaultSliceSeparator)\n}\n\nfunc formatMap(value interface{}, opts RenderOptions) string {\n\tv := reflect.ValueOf(value)\n\tvalues := []string{}\n\tfor _, key := range v.MapKeys() {\n\t\tvalues = append(values, formatValue(key, opts, \"\")+defaultMapItemSeparator+formatValue(v.MapIndex(key), opts, \"\"))\n\t}\n\tsort.Strings(values)\n\treturn strings.Join(values, defaultMapSeparator)\n}\n\nfunc formatTime(t time.Time, opts RenderOptions) string {\n\tif opts.PrintRawTime {\n\t\treturn strconv.FormatInt(t.Unix(), 10)\n\t}\n\tif opts.PrintDateTime {\n\t\treturn t.Format(defaultDateTimeFormat)\n\t}\n\treturn t.Format(defaultTimeFormat)\n}\n\nfunc formatString(str string, tag reflect.StructTag) string {\n\tif maxLengthStr, ok := tag.Lookup(\"maxLength\"); ok {\n\t\tmaxLength, _ := strconv.ParseInt(maxLengthStr, 10, 64)\n\t\tstr = trimString(str, int(maxLength))\n\t}\n\n\treturn str\n}\n\nfunc trimString(str string, maxLength int) string {\n\tif len(str) < maxLength {\n\t\treturn str\n\t}\n\n\titems := strings.Split(str, \"/\")\n\tlastItem := items[len(items)-1]\n\tif len(lastItem) < maxLength {\n\t\treturn \".../\" + lastItem\n\t}\n\n\treturn \"...\" + lastItem[len(lastItem)-maxLength:]\n}\n"
  },
  {
    "path": "tools/cli/render_test.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"strings\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc Test_RenderTable(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tdata         interface{}\n\t\topts         RenderOptions\n\t\texpectOutput string\n\t\texpectErr    string\n\t}{\n\t\t{\n\t\t\tname:         \"nil slice\",\n\t\t\tdata:         ([]testRow)(nil),\n\t\t\texpectOutput: \"\",\n\t\t},\n\t\t{\n\t\t\tname:         \"empty slice\",\n\t\t\tdata:         []testRow{},\n\t\t\texpectOutput: \"\",\n\t\t},\n\t\t{\n\t\t\tname:         \"nil struct\",\n\t\t\tdata:         (*testRow)(nil),\n\t\t\texpectOutput: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"single struct\",\n\t\t\tdata: testTable[0],\n\t\t\texpectOutput: \"\" +\n\t\t\t\t\"  STRING | INTEGER | BOOL |   TIME   |    MAP     |  SLICE   \\n\" +\n\t\t\t\t\"  text   |     123 | true | 03:04:05 | A:AA, B:BB | 1, 2, 3  \\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"pointer to single struct\",\n\t\t\tdata: &testTable[0],\n\t\t\texpectOutput: \"\" +\n\t\t\t\t\"  STRING | INTEGER | BOOL |   TIME   |    MAP     |  SLICE   \\n\" +\n\t\t\t\t\"  text   |     123 | true | 03:04:05 | A:AA, B:BB | 1, 2, 3  \\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"a slice\",\n\t\t\tdata: testTable,\n\t\t\texpectOutput: \"\" +\n\t\t\t\t\"        STRING        | INTEGER | BOOL  |   TIME   |    MAP     |  SLICE   \\n\" +\n\t\t\t\t\"  text                |     123 | true  | 03:04:05 | A:AA, B:BB | 1, 2, 3  \\n\" +\n\t\t\t\t\"  .../ string this is |     456 | false | 13:14:15 |            |          \\n\",\n\t\t},\n\t\t{\n\t\t\tname: \"a slice with rendering options\",\n\t\t\tdata: testTable,\n\t\t\topts: RenderOptions{OptionalColumns: map[string]bool{\"map\": true, \"slice\": false}, PrintDateTime: true},\n\t\t\texpectOutput: \"\" +\n\t\t\t\t\"        STRING        | INTEGER | BOOL  |         TIME         |    MAP      \\n\" +\n\t\t\t\t\"  text                |     123 | true  | 2000-01-02T03:04:05Z | A:AA, B:BB  \\n\" +\n\t\t\t\t\"  .../ string this is |     456 | false | 2000-11-12T13:14:15Z |             \\n\",\n\t\t},\n\t\t{\n\t\t\tname:      \"non-struct element\",\n\t\t\tdata:      123,\n\t\t\texpectErr: \"table slice element must be a struct, provided: int\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tbuilder := &strings.Builder{}\n\t\t\terr := RenderTable(builder, tt.data, tt.opts)\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.expectOutput, builder.String())\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_RenderTemplate(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tdata         interface{}\n\t\ttemplate     string\n\t\topts         RenderOptions\n\t\texpectOutput string\n\t\texpectErr    string\n\t}{\n\t\t{\n\t\t\tname:         \"a field from nil struct\",\n\t\t\tdata:         (*testRow)(nil),\n\t\t\ttemplate:     \"{{.StringField}}\",\n\t\t\texpectOutput: \"\",\n\t\t},\n\t\t{\n\t\t\tname:         \"a field from struct\",\n\t\t\tdata:         testTable[0],\n\t\t\ttemplate:     \"{{.StringField}}\",\n\t\t\texpectOutput: \"text\",\n\t\t},\n\t\t{\n\t\t\tname:         \"json function\",\n\t\t\tdata:         testTable[0],\n\t\t\ttemplate:     \"{{json .MapField}}\",\n\t\t\texpectOutput: \"{\\n  \\\"A\\\": \\\"AA\\\",\\n  \\\"B\\\": \\\"BB\\\"\\n}\",\n\t\t},\n\t\t{\n\t\t\tname:     \"table function\",\n\t\t\tdata:     testTable,\n\t\t\ttemplate: \"{{table .}}\",\n\t\t\texpectOutput: \"\" +\n\t\t\t\t\"        STRING        | INTEGER | BOOL  |   TIME   |    MAP     |  SLICE   \\n\" +\n\t\t\t\t\"  text                |     123 | true  | 03:04:05 | A:AA, B:BB | 1, 2, 3  \\n\" +\n\t\t\t\t\"  .../ string this is |     456 | false | 13:14:15 |            |          \\n\",\n\t\t},\n\t\t{\n\t\t\tname:      \"invalid template\",\n\t\t\tdata:      testTable,\n\t\t\ttemplate:  \"{{invalid}}\",\n\t\t\texpectErr: \"invalid template \\\"{{invalid}}\\\": template: :1: function \\\"invalid\\\" not defined\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tbuilder := &strings.Builder{}\n\t\t\terr := RenderTemplate(builder, tt.data, tt.template, tt.opts)\n\t\t\tif tt.expectErr != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectErr)\n\t\t\t} else {\n\t\t\t\tassert.Equal(t, tt.expectOutput, builder.String())\n\t\t\t}\n\t\t})\n\t}\n}\n\ntype testRow struct {\n\tStringField  string            `header:\"string\" maxLength:\"16\"`\n\tIntField     int               `header:\"integer\"`\n\tBoolField    bool              `header:\"bool\"`\n\tTimeField    time.Time         `header:\"time\"`\n\tMapField     map[string]string `header:\"map\"`\n\tSliceField   []int             `header:\"slice\"`\n\tIgnoredField int\n}\n\nvar testTable = []testRow{\n\t{\n\t\tStringField: \"text\",\n\t\tIntField:    123,\n\t\tBoolField:   true,\n\t\tTimeField:   time.Date(2000, 1, 2, 3, 4, 5, 6, time.UTC),\n\t\tMapField:    map[string]string{\"A\": \"AA\", \"B\": \"BB\"},\n\t\tSliceField:  []int{1, 2, 3},\n\t},\n\t{\n\t\tStringField: \"very long/ string this is\",\n\t\tIntField:    456,\n\t\tBoolField:   false,\n\t\tTimeField:   time.Date(2000, 11, 12, 13, 14, 15, 16, time.FixedZone(\"UTC+3\", 0)),\n\t\tMapField:    nil,\n\t\tSliceField:  nil,\n\t},\n}\n"
  },
  {
    "path": "tools/cli/task_list.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport \"github.com/urfave/cli/v2\"\n\nfunc newTaskListCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe pollers info of tasklist\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskList,\n\t\t\t\t\tAliases: []string{\"tl\"},\n\t\t\t\t\tUsage:   \"TaskList description\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskListType,\n\t\t\t\t\tAliases: []string{\"tlt\"},\n\t\t\t\t\tValue:   \"decision\",\n\t\t\t\t\tUsage:   \"Optional TaskList type [decision|activity]\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: DescribeTaskList,\n\t\t},\n\t\t{\n\t\t\tName:    \"list-partition\",\n\t\t\tAliases: []string{\"lp\"},\n\t\t\tUsage:   \"List all the tasklist partitions and the hostname for partitions.\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskList,\n\t\t\t\t\tAliases: []string{\"tl\"},\n\t\t\t\t\tUsage:   \"TaskList description\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTaskListType,\n\t\t\t\t\tAliases: []string{\"tlt\"},\n\t\t\t\t\tValue:   \"decision\",\n\t\t\t\t\tUsage:   \"Optional TaskList type [decision|activity]\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: ListTaskListPartitions,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/task_list_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"io\"\n\t\"os\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\ntype (\n\tTaskListPollerRow struct {\n\t\tActivityIdentity string    `header:\"Activity Poller Identity\"`\n\t\tDecisionIdentity string    `header:\"Decision Poller Identity\"`\n\t\tLastAccessTime   time.Time `header:\"Last Access Time\"`\n\t}\n\tTaskListPartitionRow struct {\n\t\tActivityPartition string `header:\"Activity Task List Partition\"`\n\t\tDecisionPartition string `header:\"Decision Task List Partition\"`\n\t\tHost              string `header:\"Host\"`\n\t}\n)\n\n// DescribeTaskList show pollers info of a given tasklist\nfunc DescribeTaskList(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskList, err := getRequiredOption(c, FlagTaskList)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskListType := strToTaskListType(c.String(FlagTaskListType)) // default type is decision\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\trequest := &types.DescribeTaskListRequest{\n\t\tDomain: domain,\n\t\tTaskList: &types.TaskList{\n\t\t\tName: taskList,\n\t\t},\n\t\tTaskListType: &taskListType,\n\t}\n\tresponse, err := wfClient.DescribeTaskList(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation DescribeTaskList failed.\", err)\n\t}\n\n\tpollers := response.Pollers\n\tif len(pollers) == 0 {\n\t\treturn commoncli.Problem(colorMagenta(\"No poller for tasklist: \"+taskList), nil)\n\t}\n\n\treturn printTaskListPollers(getDeps(c).Output(), pollers, taskListType)\n}\n\n// ListTaskListPartitions gets all the tasklist partition and host information.\nfunc ListTaskListPartitions(c *cli.Context) error {\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskList, err := getRequiredOption(c, FlagTaskList)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskListType := strToTaskListType(c.String(FlagTaskListType)) // default type is decision\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\trequest := &types.ListTaskListPartitionsRequest{\n\t\tDomain:   domain,\n\t\tTaskList: &types.TaskList{Name: taskList},\n\t}\n\n\tresponse, err := frontendClient.ListTaskListPartitions(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Operation ListTaskListPartitions failed.\", err)\n\t}\n\n\tswitch taskListType {\n\tcase types.TaskListTypeActivity:\n\t\treturn printTaskListPartitions(types.TaskListTypeActivity, response.ActivityTaskListPartitions)\n\tcase types.TaskListTypeDecision:\n\t\treturn printTaskListPartitions(types.TaskListTypeDecision, response.DecisionTaskListPartitions)\n\tdefault:\n\t\t// should never happen\n\t\treturn nil\n\t}\n}\n\nfunc printTaskListPollers(w io.Writer, pollers []*types.PollerInfo, taskListType types.TaskListType) error {\n\ttable := []TaskListPollerRow{}\n\tfor _, poller := range pollers {\n\t\ttable = append(table, TaskListPollerRow{\n\t\t\tActivityIdentity: poller.GetIdentity(),\n\t\t\tDecisionIdentity: poller.GetIdentity(),\n\t\t\tLastAccessTime:   time.Unix(0, poller.GetLastAccessTime())})\n\t}\n\treturn RenderTable(w, table, RenderOptions{Color: true, PrintDateTime: true, OptionalColumns: map[string]bool{\n\t\t\"Activity Poller Identity\": taskListType == types.TaskListTypeActivity,\n\t\t\"Decision Poller Identity\": taskListType == types.TaskListTypeDecision,\n\t}})\n}\n\nfunc printTaskListPartitions(taskListType types.TaskListType, partitions []*types.TaskListPartitionMetadata) error {\n\ttable := []TaskListPartitionRow{}\n\tfor _, partition := range partitions {\n\t\ttable = append(table, TaskListPartitionRow{\n\t\t\tActivityPartition: partition.GetKey(),\n\t\t\tDecisionPartition: partition.GetKey(),\n\t\t\tHost:              partition.GetOwnerHostName(),\n\t\t})\n\t}\n\treturn RenderTable(os.Stdout, table, RenderOptions{Color: true, OptionalColumns: map[string]bool{\n\t\t\"Activity Task List Partition\": taskListType == types.TaskListTypeActivity,\n\t\t\"Decision Task List Partition\": taskListType == types.TaskListTypeDecision,\n\t}})\n}\n"
  },
  {
    "path": "tools/cli/testdata/scan_input.json",
    "content": "{\"DomainID\":\"test-domain-id1\",\"WorkflowID\":\"test-workflow-id1\",\"RunID\":\"test-run-id1\",\"DomainName\":\"test-domain-name1\"}\n{\"DomainID\":\"test-domain-id2\",\"WorkflowID\":\"test-workflow-id2\",\"RunID\":\"test-run-id2\",\"DomainName\":\"test-domain-name2\"}\n{\"DomainID\":\"test-domain-id3\",\"WorkflowID\":\"test-workflow-id3\",\"RunID\":\"test-run-id3\",\"DomainName\":\"test-domain-name3\"}\n"
  },
  {
    "path": "tools/cli/testdata/scan_input_bad_data.json",
    "content": "We should error on invalid data, this is invalid data\n{\"DomainID\":\"test-domain-id1\",\"WorkflowID\":\"test-workflow-id1\",\"RunID\":\"test-run-id1\",\"DomainName\":\"test-domain-name1\"}\n"
  },
  {
    "path": "tools/cli/testdata/scan_input_empty.json",
    "content": ""
  },
  {
    "path": "tools/cli/utils.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"reflect\"\n\t\"regexp\"\n\t\"runtime/debug\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/fatih/color\"\n\t\"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/urfave/cli/v2\"\n\t\"github.com/valyala/fastjson\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/pagination\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\n// JSONHistorySerializer is used to encode history event in JSON\ntype JSONHistorySerializer struct{}\n\n// Serialize serializes history.\nfunc (j *JSONHistorySerializer) Serialize(h *types.History) ([]byte, error) {\n\treturn json.Marshal(h.Events)\n}\n\n// Deserialize deserializes history\nfunc (j *JSONHistorySerializer) Deserialize(data []byte) (*types.History, error) {\n\tvar events []*types.HistoryEvent\n\terr := json.Unmarshal(data, &events)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn &types.History{Events: events}, nil\n}\n\n// GetHistory helper method to iterate over all pages and return complete list of history events\nfunc GetHistory(ctx context.Context, workflowClient frontend.Client, domain, workflowID, runID string, consistencyLevel *types.QueryConsistencyLevel) (*types.History, error) {\n\tevents := []*types.HistoryEvent{}\n\titerator, err := GetWorkflowHistoryIterator(ctx, workflowClient, domain, workflowID, runID, false, types.HistoryEventFilterTypeAllEvent.Ptr(), consistencyLevel)\n\tfor iterator.HasNext() {\n\t\tentity, err := iterator.Next()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tevents = append(events, entity.(*types.HistoryEvent))\n\t}\n\thistory := &types.History{}\n\thistory.Events = events\n\treturn history, err\n}\n\n// GetWorkflowHistoryIterator returns a HistoryEvent iterator\nfunc GetWorkflowHistoryIterator(\n\tctx context.Context,\n\tworkflowClient frontend.Client,\n\tdomain,\n\tworkflowID,\n\trunID string,\n\tisLongPoll bool,\n\tfilterType *types.HistoryEventFilterType,\n\tconsistencyLevel *types.QueryConsistencyLevel,\n) (pagination.Iterator, error) {\n\tpaginate := func(ctx context.Context, pageToken pagination.PageToken) (pagination.Page, error) {\n\t\ttcCtx, cancel := context.WithTimeout(ctx, 25*time.Second)\n\t\tdefer cancel()\n\n\t\tvar nextPageToken []byte\n\t\tif pageToken != nil {\n\t\t\tnextPageToken, _ = pageToken.([]byte)\n\t\t}\n\t\trequest := &types.GetWorkflowExecutionHistoryRequest{\n\t\t\tDomain: domain,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: workflowID,\n\t\t\t\tRunID:      runID,\n\t\t\t},\n\t\t\tWaitForNewEvent:        isLongPoll,\n\t\t\tHistoryEventFilterType: filterType,\n\t\t\tNextPageToken:          nextPageToken,\n\t\t\tSkipArchival:           isLongPoll,\n\t\t\tQueryConsistencyLevel:  consistencyLevel,\n\t\t}\n\n\t\tvar resp *types.GetWorkflowExecutionHistoryResponse\n\t\tvar err error\n\tLoop:\n\t\tfor {\n\t\t\tresp, err = workflowClient.GetWorkflowExecutionHistory(tcCtx, request)\n\t\t\tif err != nil {\n\t\t\t\treturn pagination.Page{}, err\n\t\t\t}\n\n\t\t\tif isLongPoll && len(resp.History.Events) == 0 && len(resp.NextPageToken) != 0 {\n\t\t\t\trequest.NextPageToken = resp.NextPageToken\n\t\t\t\tcontinue Loop\n\t\t\t}\n\t\t\tbreak Loop\n\t\t}\n\t\tentities := make([]pagination.Entity, len(resp.History.Events))\n\t\tfor i, e := range resp.History.Events {\n\t\t\tentities[i] = e\n\t\t}\n\t\tvar nextToken interface{} = resp.NextPageToken\n\t\tif len(resp.NextPageToken) == 0 {\n\t\t\tnextToken = nil\n\t\t}\n\t\tpage := pagination.Page{\n\t\t\tCurrentToken: pageToken,\n\t\t\tNextToken:    nextToken,\n\t\t\tEntities:     entities,\n\t\t}\n\t\treturn page, err\n\t}\n\treturn pagination.NewIterator(ctx, nil, paginate), nil\n}\n\n// HistoryEventToString convert HistoryEvent to string\nfunc HistoryEventToString(e *types.HistoryEvent, printFully bool, maxFieldLength int) string {\n\tdata := getEventAttributes(e)\n\treturn anyToString(data, printFully, maxFieldLength)\n}\n\nfunc anyToString(d interface{}, printFully bool, maxFieldLength int) string {\n\t// fields related to schedule are of time.Time type, and we shouldn't dive\n\t// into it with reflection - it's fields are private.\n\ttm, ok := d.(time.Time)\n\tif ok {\n\t\treturn trimText(tm.String(), maxFieldLength)\n\t}\n\n\tv := reflect.ValueOf(d)\n\tswitch v.Kind() {\n\tcase reflect.Ptr:\n\t\treturn anyToString(v.Elem().Interface(), printFully, maxFieldLength)\n\tcase reflect.Struct:\n\t\tvar buf bytes.Buffer\n\t\tt := reflect.TypeOf(d)\n\t\tbuf.WriteString(\"{\")\n\t\tfor i := 0; i < v.NumField(); i++ {\n\t\t\tf := v.Field(i)\n\t\t\tif f.Kind() == reflect.Invalid {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tfieldValue := valueToString(f, printFully, maxFieldLength)\n\t\t\tif len(fieldValue) == 0 {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif buf.Len() > 1 {\n\t\t\t\tbuf.WriteString(\", \")\n\t\t\t}\n\t\t\tfieldName := t.Field(i).Name\n\t\t\tif !isAttributeName(fieldName) {\n\t\t\t\tif !printFully {\n\t\t\t\t\tfieldValue = trimTextAndBreakWords(fieldValue, maxFieldLength)\n\t\t\t\t} else if maxFieldLength != 0 { // for command run workflow and observe history\n\t\t\t\t\tfieldValue = trimText(fieldValue, maxFieldLength)\n\t\t\t\t}\n\t\t\t}\n\t\t\tif fieldName == \"Reason\" || fieldName == \"Details\" || fieldName == \"Cause\" {\n\t\t\t\tbuf.WriteString(fmt.Sprintf(\"%s:%s\", color.RedString(fieldName), color.MagentaString(fieldValue)))\n\t\t\t} else {\n\t\t\t\tbuf.WriteString(fmt.Sprintf(\"%s:%s\", fieldName, fieldValue))\n\t\t\t}\n\t\t}\n\t\tbuf.WriteString(\"}\")\n\t\treturn buf.String()\n\tdefault:\n\t\treturn fmt.Sprint(d)\n\t}\n}\n\nfunc valueToString(v reflect.Value, printFully bool, maxFieldLength int) string {\n\tswitch v.Kind() {\n\tcase reflect.Ptr:\n\t\treturn valueToString(v.Elem(), printFully, maxFieldLength)\n\tcase reflect.Struct:\n\t\treturn anyToString(v.Interface(), printFully, maxFieldLength)\n\tcase reflect.Invalid:\n\t\treturn \"\"\n\tcase reflect.Slice:\n\t\tif v.Type().Elem().Kind() == reflect.Uint8 {\n\t\t\tn := string(v.Bytes())\n\t\t\tif n != \"\" && n[len(n)-1] == '\\n' {\n\t\t\t\treturn fmt.Sprintf(\"[%v]\", n[:len(n)-1])\n\t\t\t}\n\t\t\treturn fmt.Sprintf(\"[%v]\", n)\n\t\t}\n\t\treturn fmt.Sprintf(\"[len=%d]\", v.Len())\n\tcase reflect.Map:\n\t\tstr := \"map{\"\n\t\tfor i, key := range v.MapKeys() {\n\t\t\tstr += key.String() + \":\"\n\t\t\tval := v.MapIndex(key)\n\t\t\tswitch val.Interface().(type) {\n\t\t\tcase []byte:\n\t\t\t\tstr += string(val.Interface().([]byte))\n\t\t\tdefault:\n\t\t\t\tstr += val.String()\n\t\t\t}\n\t\t\tif i != len(v.MapKeys())-1 {\n\t\t\t\tstr += \", \"\n\t\t\t}\n\t\t}\n\t\tstr += \"}\"\n\t\treturn str\n\tdefault:\n\t\treturn fmt.Sprint(v.Interface())\n\t}\n}\n\n// limit the maximum length for each field\nfunc trimText(input string, maxFieldLength int) string {\n\tif len(input) > maxFieldLength {\n\t\tinput = fmt.Sprintf(\"%s ... %s\", input[:maxFieldLength/2], input[(len(input)-maxFieldLength/2):])\n\t}\n\treturn input\n}\n\n// limit the maximum length for each field, and break long words for table item correctly wrap words\nfunc trimTextAndBreakWords(input string, maxFieldLength int) string {\n\tinput = trimText(input, maxFieldLength)\n\treturn breakLongWords(input, maxWordLength)\n}\n\n// long words will make output in table cell looks bad,\n// break long text \"ltltltltllt...\" to \"ltlt ltlt lt...\" will make use of table autowrap so that output is pretty.\nfunc breakLongWords(input string, maxWordLength int) string {\n\tif len(input) <= maxWordLength {\n\t\treturn input\n\t}\n\n\tcnt := 0\n\tfor i := 0; i < len(input); i++ {\n\t\tif cnt == maxWordLength {\n\t\t\tcnt = 0\n\t\t\tinput = input[:i] + \" \" + input[i:]\n\t\t\tcontinue\n\t\t}\n\t\tcnt++\n\t\tif input[i] == ' ' {\n\t\t\tcnt = 0\n\t\t}\n\t}\n\treturn input\n}\n\n// ColorEvent takes an event and return string with color\n// Event with color mapping rules:\n//\n//\tFailed - red\n//\tTimeout - yellow\n//\tCanceled - magenta\n//\tCompleted - green\n//\tStarted - blue\n//\tOthers - default (white/black)\nfunc ColorEvent(e *types.HistoryEvent) string {\n\tf := EventColorFunction(*e.EventType)\n\treturn f(e.EventType.String())\n}\n\nfunc EventColorFunction(eventType types.EventType) func(format string, a ...interface{}) string {\n\tvar colorFunc func(format string, a ...interface{}) string\n\tnoColorFunc := func(format string, a ...interface{}) string {\n\t\treturn format\n\t}\n\tswitch eventType {\n\tcase types.EventTypeWorkflowExecutionStarted,\n\t\ttypes.EventTypeChildWorkflowExecutionStarted:\n\t\tcolorFunc = color.BlueString\n\n\tcase types.EventTypeWorkflowExecutionCompleted,\n\t\ttypes.EventTypeChildWorkflowExecutionCompleted:\n\t\tcolorFunc = color.GreenString\n\n\tcase types.EventTypeWorkflowExecutionFailed,\n\t\ttypes.EventTypeRequestCancelActivityTaskFailed,\n\t\ttypes.EventTypeCancelTimerFailed,\n\t\ttypes.EventTypeStartChildWorkflowExecutionFailed,\n\t\ttypes.EventTypeChildWorkflowExecutionFailed,\n\t\ttypes.EventTypeRequestCancelExternalWorkflowExecutionFailed,\n\t\ttypes.EventTypeSignalExternalWorkflowExecutionFailed,\n\t\ttypes.EventTypeActivityTaskFailed:\n\t\tcolorFunc = color.RedString\n\n\tcase types.EventTypeWorkflowExecutionTimedOut,\n\t\ttypes.EventTypeActivityTaskTimedOut,\n\t\ttypes.EventTypeWorkflowExecutionCanceled,\n\t\ttypes.EventTypeChildWorkflowExecutionTimedOut,\n\t\ttypes.EventTypeDecisionTaskTimedOut:\n\t\tcolorFunc = color.YellowString\n\n\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\tcolorFunc = color.MagentaString\n\n\tdefault:\n\t\tcolorFunc = noColorFunc\n\t}\n\n\treturn colorFunc\n}\n\nfunc getEventAttributes(e *types.HistoryEvent) interface{} {\n\tvar data interface{}\n\tswitch e.GetEventType() {\n\tcase types.EventTypeWorkflowExecutionStarted:\n\t\tdata = e.WorkflowExecutionStartedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tdata = e.WorkflowExecutionCompletedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tdata = e.WorkflowExecutionFailedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tdata = e.WorkflowExecutionTimedOutEventAttributes\n\n\tcase types.EventTypeDecisionTaskScheduled:\n\t\tdata = e.DecisionTaskScheduledEventAttributes\n\n\tcase types.EventTypeDecisionTaskStarted:\n\t\tdata = e.DecisionTaskStartedEventAttributes\n\n\tcase types.EventTypeDecisionTaskCompleted:\n\t\tdata = e.DecisionTaskCompletedEventAttributes\n\n\tcase types.EventTypeDecisionTaskTimedOut:\n\t\tdata = e.DecisionTaskTimedOutEventAttributes\n\n\tcase types.EventTypeActivityTaskScheduled:\n\t\tdata = e.ActivityTaskScheduledEventAttributes\n\n\tcase types.EventTypeActivityTaskStarted:\n\t\tdata = e.ActivityTaskStartedEventAttributes\n\n\tcase types.EventTypeActivityTaskCompleted:\n\t\tdata = e.ActivityTaskCompletedEventAttributes\n\n\tcase types.EventTypeActivityTaskFailed:\n\t\tdata = e.ActivityTaskFailedEventAttributes\n\n\tcase types.EventTypeActivityTaskTimedOut:\n\t\tdata = e.ActivityTaskTimedOutEventAttributes\n\n\tcase types.EventTypeActivityTaskCancelRequested:\n\t\tdata = e.ActivityTaskCancelRequestedEventAttributes\n\n\tcase types.EventTypeRequestCancelActivityTaskFailed:\n\t\tdata = e.RequestCancelActivityTaskFailedEventAttributes\n\n\tcase types.EventTypeActivityTaskCanceled:\n\t\tdata = e.ActivityTaskCanceledEventAttributes\n\n\tcase types.EventTypeTimerStarted:\n\t\tdata = e.TimerStartedEventAttributes\n\n\tcase types.EventTypeTimerFired:\n\t\tdata = e.TimerFiredEventAttributes\n\n\tcase types.EventTypeCancelTimerFailed:\n\t\tdata = e.CancelTimerFailedEventAttributes\n\n\tcase types.EventTypeTimerCanceled:\n\t\tdata = e.TimerCanceledEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionCancelRequested:\n\t\tdata = e.WorkflowExecutionCancelRequestedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tdata = e.WorkflowExecutionCanceledEventAttributes\n\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionInitiated:\n\t\tdata = e.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes\n\n\tcase types.EventTypeRequestCancelExternalWorkflowExecutionFailed:\n\t\tdata = e.RequestCancelExternalWorkflowExecutionFailedEventAttributes\n\n\tcase types.EventTypeExternalWorkflowExecutionCancelRequested:\n\t\tdata = e.ExternalWorkflowExecutionCancelRequestedEventAttributes\n\n\tcase types.EventTypeMarkerRecorded:\n\t\tdata = e.MarkerRecordedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionSignaled:\n\t\tdata = e.WorkflowExecutionSignaledEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionTerminated:\n\t\tdata = e.WorkflowExecutionTerminatedEventAttributes\n\n\tcase types.EventTypeWorkflowExecutionContinuedAsNew:\n\t\tdata = e.WorkflowExecutionContinuedAsNewEventAttributes\n\n\tcase types.EventTypeStartChildWorkflowExecutionInitiated:\n\t\tdata = e.StartChildWorkflowExecutionInitiatedEventAttributes\n\n\tcase types.EventTypeStartChildWorkflowExecutionFailed:\n\t\tdata = e.StartChildWorkflowExecutionFailedEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionStarted:\n\t\tdata = e.ChildWorkflowExecutionStartedEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionCompleted:\n\t\tdata = e.ChildWorkflowExecutionCompletedEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionFailed:\n\t\tdata = e.ChildWorkflowExecutionFailedEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionCanceled:\n\t\tdata = e.ChildWorkflowExecutionCanceledEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionTimedOut:\n\t\tdata = e.ChildWorkflowExecutionTimedOutEventAttributes\n\n\tcase types.EventTypeChildWorkflowExecutionTerminated:\n\t\tdata = e.ChildWorkflowExecutionTerminatedEventAttributes\n\n\tcase types.EventTypeSignalExternalWorkflowExecutionInitiated:\n\t\tdata = e.SignalExternalWorkflowExecutionInitiatedEventAttributes\n\n\tcase types.EventTypeSignalExternalWorkflowExecutionFailed:\n\t\tdata = e.SignalExternalWorkflowExecutionFailedEventAttributes\n\n\tcase types.EventTypeExternalWorkflowExecutionSignaled:\n\t\tdata = e.ExternalWorkflowExecutionSignaledEventAttributes\n\n\tdefault:\n\t\tdata = e\n\t}\n\treturn data\n}\n\nfunc isAttributeName(name string) bool {\n\tfor i := types.EventType(0); i < types.EventType(40); i++ {\n\t\tif name == i.String()+\"EventAttributes\" {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc getCurrentUserFromEnv() string {\n\tfor _, n := range envKeysForUserName {\n\t\tif len(os.Getenv(n)) > 0 {\n\t\t\treturn os.Getenv(n)\n\t\t}\n\t}\n\treturn \"unknown\"\n}\n\nfunc prettyPrintJSONObject(writer io.Writer, o interface{}) {\n\tb, err := json.MarshalIndent(o, \"\", \"  \")\n\tif err != nil {\n\t\twriter.Write([]byte(fmt.Sprintf(\"Error when try to print pretty: %v\\n\", err)))\n\t\twriter.Write([]byte(fmt.Sprintf(\"%+v\\n\", o)))\n\t}\n\twriter.Write(b)\n\twriter.Write([]byte(\"\\n\"))\n}\n\nfunc mapKeysToArray(m map[string]string) []string {\n\tvar out []string\n\tfor k := range m {\n\t\tout = append(out, k)\n\t}\n\treturn out\n}\n\nfunc intSliceToSet(s []int) map[int]struct{} {\n\tvar ret = make(map[int]struct{}, len(s))\n\tfor _, v := range s {\n\t\tret[v] = struct{}{}\n\t}\n\treturn ret\n}\n\nfunc printMessage(output io.Writer, msg string) {\n\toutput.Write([]byte(fmt.Sprintf(\"%s %s\\n\", \"cadence:\", msg)))\n}\n\nfunc printError(output io.Writer, msg string, err error) {\n\tif err != nil {\n\t\toutput.Write([]byte(fmt.Sprintf(\"%s %s\\n%s %+v\\n\", colorRed(\"Error:\"), msg, colorMagenta(\"Error Details:\"), err)))\n\t\tif os.Getenv(showErrorStackEnv) != `` {\n\t\t\tfmt.Printf(\"Stack trace:\\n\")\n\t\t\tdebug.PrintStack()\n\t\t} else {\n\t\t\toutput.Write([]byte(fmt.Sprintf(\"('export %s=1' to see stack traces)\\n\", showErrorStackEnv)))\n\t\t}\n\t} else {\n\t\toutput.Write([]byte(fmt.Sprintf(\"%s %s\\n\", colorRed(\"Error:\"), msg)))\n\t}\n}\n\nfunc getWorkflowClient(c *cli.Context) (frontend.Client, error) {\n\treturn getDeps(c).ServerFrontendClient(c)\n}\n\n// getOptionWithSerializer is a helper function that retrieves a non-required option using a serializer function.\n// To support nil as a valid value, returns nil when the option is not set.\n// Do not use this function with a pointer type (e.g getOptionWithSerializer[*int]).\n// Returns a pointer to the parsed value when the option is serialized.\n// Returns an error when the serializer function is unable to parse the option value.\nfunc getOptionWithSerializer[T any](c *cli.Context, optionName string, serializer func(string) (T, error)) (*T, error) {\n\tif !c.IsSet(optionName) {\n\t\treturn nil, nil\n\t}\n\n\tvalue, err := serializer(c.String(optionName))\n\n\treturn &value, err\n}\n\nfunc getRequiredOption(c *cli.Context, optionName string) (string, error) {\n\tvalue := c.String(optionName)\n\tif len(value) == 0 {\n\t\treturn \"\", fmt.Errorf(\"option %s is required\", optionName)\n\t}\n\treturn value, nil\n}\n\nfunc getRequiredInt64Option(c *cli.Context, optionName string) (int64, error) {\n\tif !c.IsSet(optionName) {\n\t\treturn 0, fmt.Errorf(\"option %s is required\", optionName)\n\t}\n\treturn c.Int64(optionName), nil\n}\n\nfunc getRequiredIntOption(c *cli.Context, optionName string) (int, error) {\n\tif !c.IsSet(optionName) {\n\t\treturn 0, fmt.Errorf(\"option %s is required\", optionName)\n\t}\n\treturn c.Int(optionName), nil\n}\n\nfunc timestampPtrToStringPtr(unixNanoPtr *int64, onlyTime bool) *string {\n\tif unixNanoPtr == nil {\n\t\treturn nil\n\t}\n\treturn common.StringPtr(timestampToString(*unixNanoPtr, onlyTime))\n}\n\nfunc timestampToString(unixNano int64, onlyTime bool) string {\n\tt := time.Unix(0, unixNano).UTC()\n\tvar result string\n\tif onlyTime {\n\t\tresult = t.Format(defaultTimeFormat)\n\t} else {\n\t\tresult = t.Format(defaultDateTimeFormat)\n\t}\n\treturn result\n}\n\nfunc parseTime(timeStr string, defaultValue int64) (int64, error) {\n\tif len(timeStr) == 0 {\n\t\treturn defaultValue, nil\n\t}\n\n\t// try to parse\n\tparsedTime, err := time.Parse(defaultDateTimeFormat, timeStr)\n\tif err == nil {\n\t\treturn parsedTime.UnixNano(), nil\n\t}\n\n\t// treat as raw time\n\tresultValue, err := strconv.ParseInt(timeStr, 10, 64)\n\tif err == nil {\n\t\treturn resultValue, nil\n\t}\n\n\t// treat as time range format\n\tparsedTime, err = parseTimeRange(timeStr)\n\tif err != nil {\n\t\treturn 0, fmt.Errorf(\"cannot parse time '%s', use UTC format '2006-01-02T15:04:05Z', \"+\n\t\t\t\"time range or raw UnixNano directly. see help for more details: %v\", timeStr, err)\n\t}\n\treturn parsedTime.UnixNano(), nil\n}\n\n// parseTimeRange parses a given time duration string (in format X<time-duration>) and\n// returns parsed timestamp given that duration in the past from current time.\n// All valid values must contain a number followed by a time-duration, from the following list (long form/short form):\n// - second/s\n// - minute/m\n// - hour/h\n// - day/d\n// - week/w\n// - month/M\n// - year/y\n// For example, possible input values, and their result:\n// - \"3d\" or \"3day\" --> three days --> time.Now().Add(-3 * 24 * time.Hour)\n// - \"2m\" or \"2minute\" --> two minutes --> time.Now().Add(-2 * time.Minute)\n// - \"1w\" or \"1week\" --> one week --> time.Now().Add(-7 * 24 * time.Hour)\n// - \"30s\" or \"30second\" --> thirty seconds --> time.Now().Add(-30 * time.Second)\n// Note: Duration strings are case-sensitive, and should be used as mentioned above only.\n// Limitation: Value of numerical multiplier, X should be in b/w 0 - 1e6 (1 million), boundary values excluded i.e.\n// 0 < X < 1e6. Also, the maximum time in the past can be 1 January 1970 00:00:00 UTC (epoch time),\n// so giving \"1000y\" will result in epoch time.\nfunc parseTimeRange(timeRange string) (time.Time, error) {\n\tmatch, err := regexp.MatchString(defaultDateTimeRangeShortRE, timeRange)\n\tif !match { // fallback on to check if it's of longer notation\n\t\t_, err = regexp.MatchString(defaultDateTimeRangeLongRE, timeRange)\n\t}\n\tif err != nil {\n\t\treturn time.Time{}, err\n\t}\n\n\tre, _ := regexp.Compile(defaultDateTimeRangeNum)\n\tidx := re.FindStringSubmatchIndex(timeRange)\n\tif idx == nil {\n\t\treturn time.Time{}, fmt.Errorf(\"cannot parse timeRange %s\", timeRange)\n\t}\n\n\tnum, err := strconv.Atoi(timeRange[idx[0]:idx[1]])\n\tif err != nil {\n\t\treturn time.Time{}, fmt.Errorf(\"cannot parse timeRange %s\", timeRange)\n\t}\n\tif num >= 1e6 {\n\t\treturn time.Time{}, fmt.Errorf(\"invalid time-duation multiplier %d, allowed range is 0 < multiplier < 1000000\", num)\n\t}\n\n\tdur, err := parseTimeDuration(timeRange[idx[1]:])\n\tif err != nil {\n\t\treturn time.Time{}, fmt.Errorf(\"cannot parse timeRange %s\", timeRange)\n\t}\n\n\tres := time.Now().Add(time.Duration(-num) * dur) // using server's local timezone\n\tepochTime := time.Unix(0, 0)\n\tif res.Before(epochTime) {\n\t\tres = epochTime\n\t}\n\treturn res, nil\n}\n\nfunc parseSingleTs(ts string) (time.Time, error) {\n\tvar tsOut time.Time\n\tvar err error\n\tformats := []string{\"2006-01-02T15:04:05\", \"2006-01-02T15:04\", \"2006-01-02\", \"2006-01-02T15:04:05+0700\", time.RFC3339}\n\tfor _, format := range formats {\n\t\tif tsOut, err = time.Parse(format, ts); err == nil {\n\t\t\treturn tsOut, err\n\t\t}\n\t}\n\treturn tsOut, err\n}\n\n// parseTimeDuration parses the given time duration in either short or long convention\n// and returns the time.Duration\n// Valid values (long notation/short notation):\n// - second/s\n// - minute/m\n// - hour/h\n// - day/d\n// - week/w\n// - month/M\n// - year/y\n// NOTE: the input \"duration\" is case-sensitive\nfunc parseTimeDuration(duration string) (dur time.Duration, err error) {\n\tswitch duration {\n\tcase \"s\", \"second\":\n\t\tdur = time.Second\n\tcase \"m\", \"minute\":\n\t\tdur = time.Minute\n\tcase \"h\", \"hour\":\n\t\tdur = time.Hour\n\tcase \"d\", \"day\":\n\t\tdur = day\n\tcase \"w\", \"week\":\n\t\tdur = week\n\tcase \"M\", \"month\":\n\t\tdur = month\n\tcase \"y\", \"year\":\n\t\tdur = year\n\tdefault:\n\t\terr = fmt.Errorf(\"unknown time duration %s\", duration)\n\t}\n\treturn\n}\n\n// strToTaskListType converts str to types.TaskListType\n// if it's not matched, it returns types.TaskListTypeDecision\nfunc strToTaskListType(str string) types.TaskListType {\n\tif strings.ToLower(str) == \"activity\" {\n\t\treturn types.TaskListTypeActivity\n\t}\n\treturn types.TaskListTypeDecision\n}\n\nfunc getCliIdentity() string {\n\treturn fmt.Sprintf(\"cadence-cli@%s\", getHostName())\n}\n\nfunc getHostName() string {\n\thostName, err := os.Hostname()\n\tif err != nil {\n\t\thostName = \"UnKnown\"\n\t}\n\treturn hostName\n}\n\nfunc processJWTFlags(ctx context.Context, cliCtx *cli.Context) (context.Context, error) {\n\tpath := getJWTPrivateKey(cliCtx)\n\tt := getJWT(cliCtx)\n\tvar token string\n\tvar err error\n\n\tif t != \"\" {\n\t\ttoken = t\n\t} else if path != \"\" {\n\t\ttoken, err = createJWT(path, clock.NewRealTimeSource())\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"error creating JWT token: %w\", err)\n\t\t}\n\t}\n\n\treturn context.WithValue(ctx, CtxKeyJWT, token), nil\n}\n\nfunc populateContextFromCLIContext(ctx context.Context, cliCtx *cli.Context) (context.Context, error) {\n\tctx, err := processJWTFlags(ctx, cliCtx)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error while populating context from CLI: %w\", err)\n\t}\n\treturn ctx, nil\n}\n\nfunc newContext(c *cli.Context) (context.Context, context.CancelFunc, error) {\n\treturn newTimedContext(c, defaultContextTimeout)\n}\n\nfunc newContextForLongPoll(c *cli.Context) (context.Context, context.CancelFunc, error) {\n\treturn newTimedContext(c, defaultContextTimeoutForLongPoll)\n}\n\nfunc newIndefiniteContext(c *cli.Context) (context.Context, context.CancelFunc, error) {\n\tif c.IsSet(FlagContextTimeout) {\n\t\tctx, cancel, err := newTimedContext(c, time.Duration(c.Int(FlagContextTimeout))*time.Second)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn nil, nil, fmt.Errorf(\"Error in new indifinite context: %w\", err)\n\t\t}\n\t\treturn ctx, cancel, nil\n\t}\n\tctx, err := populateContextFromCLIContext(c.Context, c)\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"error in newIndefiniteContext: %w\", err)\n\t}\n\tctx, cancel := context.WithCancel(ctx)\n\tdefer cancel()\n\n\treturn ctx, cancel, nil\n}\n\nfunc newTimedContext(c *cli.Context, timeout time.Duration) (context.Context, context.CancelFunc, error) {\n\tif overrideTimeout := c.Int(FlagContextTimeout); overrideTimeout > 0 {\n\t\ttimeout = time.Duration(overrideTimeout) * time.Second\n\t}\n\n\tctx, err := populateContextFromCLIContext(c.Context, c)\n\tif err != nil {\n\t\treturn nil, nil, fmt.Errorf(\"error populate context from CLI context: %w\", err)\n\t}\n\tctx, timeOut := context.WithTimeout(ctx, timeout)\n\n\treturn ctx, timeOut, nil\n}\n\n// process and validate input provided through cmd or file\nfunc processJSONInput(c *cli.Context) (string, error) {\n\treturn processJSONInputHelper(c, jsonTypeInput)\n}\n\n// process and validate json\nfunc processJSONInputHelper(c *cli.Context, jType jsonType) (string, error) {\n\tvar flagNameOfRawInput string\n\tvar flagNameOfInputFileName string\n\n\tswitch jType {\n\tcase jsonTypeInput:\n\t\tflagNameOfRawInput = FlagInput\n\t\tflagNameOfInputFileName = FlagInputFile\n\tcase jsonTypeMemo:\n\t\tflagNameOfRawInput = FlagMemo\n\t\tflagNameOfInputFileName = FlagMemoFile\n\tcase jsonTypeHeader:\n\t\tflagNameOfRawInput = FlagHeaderValue\n\t\tflagNameOfInputFileName = FlagHeaderFile\n\tcase jsonTypeSignal:\n\t\tflagNameOfRawInput = FlagSignalInput\n\t\tflagNameOfInputFileName = FlagSignalInputFile\n\tdefault:\n\t\treturn \"\", nil\n\t}\n\n\tvar input string\n\tif c.IsSet(flagNameOfRawInput) {\n\t\tinput = c.String(flagNameOfRawInput)\n\t} else if c.IsSet(flagNameOfInputFileName) {\n\t\tinputFile := c.String(flagNameOfInputFileName)\n\t\t// This method is purely used to parse input from the CLI. The input comes from a trusted user\n\t\t// #nosec\n\t\tdata, err := os.ReadFile(inputFile)\n\t\tif err != nil {\n\t\t\treturn \"\", fmt.Errorf(\"error reading input file: %w\", err)\n\t\t}\n\t\tinput = string(data)\n\t}\n\tif input != \"\" {\n\t\tif err := validateJSONs(input); err != nil {\n\t\t\treturn \"\", fmt.Errorf(\"input is not valid JSON: %w\", err)\n\t\t}\n\t}\n\treturn input, nil\n}\n\nfunc processMultipleKeys(rawKey, separator string) []string {\n\tvar keys []string\n\tif strings.TrimSpace(rawKey) != \"\" {\n\t\tkeys = strings.Split(rawKey, separator)\n\t}\n\treturn keys\n}\n\nfunc processMultipleJSONValues(rawValue string) ([]string, error) {\n\tvar values []string\n\tvar sc fastjson.Scanner\n\tsc.Init(rawValue)\n\tfor sc.Next() {\n\t\tvalues = append(values, sc.Value().String())\n\t}\n\tif err := sc.Error(); err != nil {\n\t\treturn nil, fmt.Errorf(\"parse json error: %w\", err)\n\t}\n\treturn values, nil\n}\n\nfunc mapFromKeysValues(keys, values []string) map[string][]byte {\n\tfields := map[string][]byte{}\n\tfor i, key := range keys {\n\t\tfields[key] = []byte(values[i])\n\t}\n\treturn fields\n}\n\n// validate whether str is a valid json or multi valid json concatenated with spaces/newlines\nfunc validateJSONs(str string) error {\n\tinput := []byte(str)\n\tdec := json.NewDecoder(bytes.NewReader(input))\n\tfor {\n\t\t_, err := dec.Token()\n\t\tif err == io.EOF {\n\t\t\treturn nil // End of input, valid JSON\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err // Invalid input\n\t\t}\n\t}\n}\n\n// use parseBool to ensure all BOOL search attributes only be \"true\" or \"false\"\nfunc parseBool(str string) (bool, error) {\n\tswitch str {\n\tcase \"true\":\n\t\treturn true, nil\n\tcase \"false\":\n\t\treturn false, nil\n\t}\n\treturn false, fmt.Errorf(\"not parseable bool value: %s\", str)\n}\n\nfunc trimSpace(strs []string) []string {\n\tresult := make([]string, len(strs))\n\tfor i, v := range strs {\n\t\tresult[i] = strings.TrimSpace(v)\n\t}\n\treturn result\n}\n\nfunc parseArray(v string) (interface{}, error) {\n\tif len(v) > 0 && v[0] == '[' && v[len(v)-1] == ']' {\n\t\tparsedValues, err := fastjson.Parse(v)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tarr, err := parsedValues.Array()\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tresult := make([]interface{}, len(arr))\n\t\tfor i, item := range arr {\n\t\t\ts := item.String()\n\t\t\tif len(s) >= 2 && s[0] == '\"' && s[len(s)-1] == '\"' { // remove addition quote from json\n\t\t\t\ts = s[1 : len(s)-1]\n\t\t\t\tif sTime, err := time.Parse(defaultDateTimeFormat, s); err == nil {\n\t\t\t\t\tresult[i] = sTime\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t}\n\t\t\tresult[i] = s\n\t\t}\n\t\treturn result, nil\n\t}\n\treturn nil, errors.New(\"not array\")\n}\n\nfunc convertStringToRealType(v string) interface{} {\n\tvar genVal interface{}\n\tvar err error\n\n\tif genVal, err = strconv.ParseInt(v, 10, 64); err == nil {\n\n\t} else if genVal, err = parseBool(v); err == nil {\n\n\t} else if genVal, err = strconv.ParseFloat(v, 64); err == nil {\n\n\t} else if genVal, err = time.Parse(defaultDateTimeFormat, v); err == nil {\n\n\t} else if genVal, err = parseArray(v); err == nil {\n\n\t} else {\n\t\tgenVal = v\n\t}\n\n\treturn genVal\n}\n\nfunc truncate(str string) string {\n\tif len(str) > maxOutputStringLength {\n\t\treturn str[:maxOutputStringLength]\n\t}\n\treturn str\n}\n\n// this only works for ANSI terminal, which means remove existing lines won't work if users redirect to file\n// ref: https://en.wikipedia.org/wiki/ANSI_escape_code\nfunc removePrevious2LinesFromTerminal(output io.Writer) {\n\toutput.Write([]byte(\"\\033[1A\"))\n\toutput.Write([]byte(\"\\033[2K\"))\n\toutput.Write([]byte(\"\\033[1A\"))\n\toutput.Write([]byte(\"\\033[2K\"))\n}\n\nfunc showNextPage(output io.Writer) bool {\n\toutput.Write([]byte(fmt.Sprintf(\"Press %s to show next page, press %s to quit: \",\n\t\tcolor.GreenString(\"Enter\"), color.RedString(\"any other key then Enter\"))))\n\tvar input string\n\tfmt.Scanln(&input)\n\treturn strings.Trim(input, \" \") == \"\"\n}\n\n// prompt will show input msg, then waiting user input y/yes to continue\nfunc prompt(msg string) {\n\treader := bufio.NewReader(os.Stdin)\n\tfmt.Println(msg)\n\ttext, _ := reader.ReadString('\\n')\n\ttextLower := strings.ToLower(strings.TrimRight(text, \"\\n\"))\n\tif textLower != \"y\" && textLower != \"yes\" {\n\t\tos.Exit(0)\n\t}\n}\nfunc getInputFile(inputFile string) (*os.File, error) {\n\tif len(inputFile) == 0 {\n\t\tinfo, err := os.Stdin.Stat()\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to stat stdin file handle: %w\", err)\n\t\t}\n\t\tif info.Mode()&os.ModeCharDevice != 0 || info.Size() <= 0 {\n\t\t\tfmt.Fprintln(os.Stderr, \"Provide a filename or pass data to STDIN\")\n\t\t\tos.Exit(1)\n\t\t}\n\t\treturn os.Stdin, nil\n\t}\n\t// This code is executed from the CLI. All user input is from a CLI user.\n\t// #nosec\n\tf, err := os.Open(inputFile)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"failed to open input file for reading: %v: %w\", inputFile, err)\n\t}\n\treturn f, nil\n}\n\n// createJWT defines the logic to create a JWT\nfunc createJWT(keyPath string, timeSource clock.TimeSource) (string, error) {\n\tprivateKey, err := common.LoadRSAPrivateKey(keyPath)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\n\tttl := int64(60 * 10)\n\tclaims := authorization.JWTClaims{\n\t\tAdmin: true,\n\t\tRegisteredClaims: jwt.RegisteredClaims{\n\t\t\tIssuedAt:  jwt.NewNumericDate(timeSource.Now()),\n\t\t\tExpiresAt: jwt.NewNumericDate(timeSource.Now().Add(time.Second * time.Duration(ttl))),\n\t\t},\n\t}\n\n\treturn jwt.NewWithClaims(jwt.SigningMethodRS256, claims).SignedString(privateKey)\n}\n\nfunc getWorkflowMemo(input map[string]interface{}) (*types.Memo, error) {\n\tif input == nil {\n\t\treturn nil, nil\n\t}\n\n\tmemo := make(map[string][]byte)\n\tfor k, v := range input {\n\t\tmemoBytes, err := json.Marshal(v)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"encode workflow memo error: %w\", err)\n\t\t}\n\t\tmemo[k] = memoBytes\n\t}\n\treturn &types.Memo{Fields: memo}, nil\n}\n\nfunc serializeSearchAttributes(input map[string]interface{}) (*types.SearchAttributes, error) {\n\tif input == nil {\n\t\treturn nil, nil\n\t}\n\n\tattr := make(map[string][]byte)\n\tfor k, v := range input {\n\t\tattrBytes, err := json.Marshal(v)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"encode search attribute [%s] error: %v\", k, err)\n\t\t}\n\t\tattr[k] = attrBytes\n\t}\n\treturn &types.SearchAttributes{IndexedFields: attr}, nil\n}\n\n// parseIntMultiRange will parse string of multiple integer ranges separates by commas.\n// Single range can be an integer or inclusive range separated by dash.\n// The result is a sorted set union of integers.\n// Example: \"3,8-8,5-6\" -> [3,4,5,8]\nfunc parseIntMultiRange(s string) ([]int, error) {\n\tset := map[int]struct{}{}\n\tranges := strings.Split(strings.TrimSpace(s), \",\")\n\tfor _, r := range ranges {\n\t\tr = strings.TrimSpace(r)\n\t\tif len(r) == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tparts := strings.Split(r, \"-\")\n\t\tswitch len(parts) {\n\t\tcase 1:\n\t\t\ti, err := strconv.Atoi(strings.TrimSpace(parts[0]))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"single number %q: %v\", r, err)\n\t\t\t}\n\t\t\tset[i] = struct{}{}\n\t\tcase 2:\n\t\t\tlower, err := strconv.Atoi(strings.TrimSpace(parts[0]))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"lower range of %q: %v\", r, err)\n\t\t\t}\n\t\t\tupper, err := strconv.Atoi(strings.TrimSpace(parts[1]))\n\t\t\tif err != nil {\n\t\t\t\treturn nil, fmt.Errorf(\"upper range of %q: %v\", r, err)\n\t\t\t}\n\t\t\tfor i := lower; i <= upper; i++ {\n\t\t\t\tset[i] = struct{}{}\n\t\t\t}\n\t\tdefault:\n\t\t\treturn nil, fmt.Errorf(\"invalid range %q\", r)\n\t\t}\n\t}\n\n\tresult := []int{}\n\tfor i := range set {\n\t\tresult = append(result, i)\n\t}\n\tsort.Ints(result)\n\treturn result, nil\n}\n"
  },
  {
    "path": "tools/cli/utils_test.go",
    "content": "// Copyright (c) 2022 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bytes\"\n\t\"crypto/rand\"\n\t\"crypto/rsa\"\n\t\"crypto/x509\"\n\t\"encoding/json\"\n\t\"encoding/pem\"\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/fatih/color\"\n\t\"github.com/golang-jwt/jwt/v5\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/authorization\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/testing/testdatagen/idlfuzzedtestdata\"\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc Test_JSONHistorySerializer(t *testing.T) {\n\tserializer := JSONHistorySerializer{}\n\th := &types.History{\n\t\tEvents: []*types.HistoryEvent{\n\t\t\t{\n\t\t\t\tID:        1,\n\t\t\t\tVersion:   1,\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t},\n\t\t\t{\n\t\t\t\tID:        2,\n\t\t\t\tVersion:   1,\n\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t},\n\t\t},\n\t}\n\tdata, err := serializer.Serialize(h)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, data)\n\n\th2, err := serializer.Deserialize(data)\n\tassert.NoError(t, err)\n\tassert.Equal(t, h, h2)\n}\n\nfunc Test_ParseIntMultiRange(t *testing.T) {\n\ttests := []struct {\n\t\tname         string\n\t\tinput        string\n\t\texpectOutput []int\n\t\texpectError  string\n\t}{\n\t\t{\n\t\t\tname:         \"empty\",\n\t\t\tinput:        \"\",\n\t\t\texpectOutput: []int{},\n\t\t},\n\t\t{\n\t\t\tname:         \"single number\",\n\t\t\tinput:        \" 1 \",\n\t\t\texpectOutput: []int{1},\n\t\t},\n\t\t{\n\t\t\tname:         \"single range\",\n\t\t\tinput:        \"1 - 3 \",\n\t\t\texpectOutput: []int{1, 2, 3},\n\t\t},\n\t\t{\n\t\t\tname:         \"multi range\",\n\t\t\tinput:        \"1 - 3 ,,  6\",\n\t\t\texpectOutput: []int{1, 2, 3, 6},\n\t\t},\n\t\t{\n\t\t\tname:         \"overlapping ranges\",\n\t\t\tinput:        \"1-3,2-4\",\n\t\t\texpectOutput: []int{1, 2, 3, 4},\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid single number\",\n\t\t\tinput:       \"1a\",\n\t\t\texpectError: \"single number \\\"1a\\\": strconv.Atoi: parsing \\\"1a\\\": invalid syntax\",\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid lower bound\",\n\t\t\tinput:       \"1a-2\",\n\t\t\texpectError: \"lower range of \\\"1a-2\\\": strconv.Atoi: parsing \\\"1a\\\": invalid syntax\",\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid upper bound\",\n\t\t\tinput:       \"1-2a\",\n\t\t\texpectError: \"upper range of \\\"1-2a\\\": strconv.Atoi: parsing \\\"2a\\\": invalid syntax\",\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid range\",\n\t\t\tinput:       \"1-2-3\",\n\t\t\texpectError: \"invalid range \\\"1-2-3\\\"\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\toutput, err := parseIntMultiRange(tt.input)\n\t\t\tif tt.expectError != \"\" {\n\t\t\t\tassert.EqualError(t, err, tt.expectError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectOutput, output)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_anyToStringWorksWithTime(t *testing.T) {\n\ttm := time.Date(2020, 1, 15, 14, 30, 45, 0, time.UTC)\n\n\tassert.Equal(\n\t\tt,\n\t\t\"2020-01-15 14:30:45 +0000 UTC\",\n\t\tanyToString(tm, false, 100),\n\t)\n\tassert.Equal(\n\t\tt,\n\t\t\"2020-01-15 ...  +0000 UTC\",\n\t\tanyToString(tm, false, 20),\n\t\t\"trimming should work for time.Time as well\",\n\t)\n}\n\nfunc Test_anyToString(t *testing.T) {\n\tinfo := struct {\n\t\tName   string\n\t\tNumber int\n\t\tTime   time.Time\n\t}{\n\t\t\"Joel\",\n\t\t1234,\n\t\ttime.Date(2019, 1, 15, 14, 30, 45, 0, time.UTC),\n\t}\n\n\tres := anyToString(info, false, 100)\n\tassert.Equal(t, \"{Name:Joel, Number:1234, Time:2019-01-15 14:30:45 +0000 UTC}\", res)\n}\n\nfunc TestJSONHistorySerializer_Serialize(t *testing.T) {\n\tgen := idlfuzzedtestdata.NewFuzzerWithIDLTypes(t)\n\th := types.History{}\n\tgen.Fuzz(&h)\n\tserializer := JSONHistorySerializer{}\n\tdata, err := serializer.Serialize(&h)\n\tassert.NoError(t, err)\n\troundTrip, err := serializer.Deserialize(data)\n\tassert.NoError(t, err)\n\tassert.Equal(t, h, *roundTrip)\n}\n\nfunc TestEventColorFunction(t *testing.T) {\n\ttests := []struct {\n\t\teventType     types.EventType\n\t\texpectedColor func(format string, a ...interface{}) string\n\t}{\n\t\t// Blue color for start events\n\t\t{eventType: types.EventTypeWorkflowExecutionStarted, expectedColor: color.BlueString},\n\t\t{eventType: types.EventTypeChildWorkflowExecutionStarted, expectedColor: color.BlueString},\n\n\t\t// Green color for completion events\n\t\t{eventType: types.EventTypeWorkflowExecutionCompleted, expectedColor: color.GreenString},\n\t\t{eventType: types.EventTypeChildWorkflowExecutionCompleted, expectedColor: color.GreenString},\n\n\t\t// Red color for failure events\n\t\t{eventType: types.EventTypeWorkflowExecutionFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeRequestCancelActivityTaskFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeCancelTimerFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeStartChildWorkflowExecutionFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeChildWorkflowExecutionFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeRequestCancelExternalWorkflowExecutionFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeSignalExternalWorkflowExecutionFailed, expectedColor: color.RedString},\n\t\t{eventType: types.EventTypeActivityTaskFailed, expectedColor: color.RedString},\n\n\t\t// Yellow color for timeout and cancel events\n\t\t{eventType: types.EventTypeWorkflowExecutionTimedOut, expectedColor: color.YellowString},\n\t\t{eventType: types.EventTypeActivityTaskTimedOut, expectedColor: color.YellowString},\n\t\t{eventType: types.EventTypeWorkflowExecutionCanceled, expectedColor: color.YellowString},\n\t\t{eventType: types.EventTypeChildWorkflowExecutionTimedOut, expectedColor: color.YellowString},\n\t\t{eventType: types.EventTypeDecisionTaskTimedOut, expectedColor: color.YellowString},\n\n\t\t// Magenta color for child workflow canceled\n\t\t{eventType: types.EventTypeChildWorkflowExecutionCanceled, expectedColor: color.MagentaString},\n\n\t\t// Default (no color)\n\t\t{eventType: types.EventTypeDecisionTaskScheduled, expectedColor: func(format string, a ...interface{}) string { return format }},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.eventType.String(), func(t *testing.T) {\n\t\t\tcolorFunc := EventColorFunction(tt.eventType)\n\t\t\tresult := colorFunc(\"test message %s\", \"color\")\n\n\t\t\t// Apply the expected color function to get expected formatted output\n\t\t\texpected := tt.expectedColor(\"test message %s\", \"color\")\n\n\t\t\tassert.Equal(t, expected, result, \"EventType %v did not return expected color function\", tt.eventType)\n\t\t})\n\t}\n}\n\nfunc TestGetEventAttributes(t *testing.T) {\n\ttests := []struct {\n\t\tname     string\n\t\tevent    *types.HistoryEvent\n\t\texpected interface{}\n\t}{\n\t\t// Workflow execution events\n\t\t{\n\t\t\tname: \"WorkflowExecutionStarted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                               types.EventTypeWorkflowExecutionStarted.Ptr(),\n\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionCompleted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionCompletedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                              types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionFailedEventAttributes{},\n\t\t},\n\t\t// More workflow execution events\n\t\t{\n\t\t\tname: \"WorkflowExecutionTimedOut\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                                types.EventTypeWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionTimedOutEventAttributes{},\n\t\t},\n\t\t// Decision task events\n\t\t{\n\t\t\tname: \"DecisionTaskScheduled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                            types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\tDecisionTaskScheduledEventAttributes: &types.DecisionTaskScheduledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.DecisionTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"DecisionTaskStarted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                          types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\tDecisionTaskStartedEventAttributes: &types.DecisionTaskStartedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.DecisionTaskStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"DecisionTaskCompleted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                            types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.DecisionTaskCompletedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"DecisionTaskTimedOut\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                           types.EventTypeDecisionTaskTimedOut.Ptr(),\n\t\t\t\tDecisionTaskTimedOutEventAttributes: &types.DecisionTaskTimedOutEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.DecisionTaskTimedOutEventAttributes{},\n\t\t},\n\t\t// Activity task events\n\t\t{\n\t\t\tname: \"ActivityTaskScheduled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                            types.EventTypeActivityTaskScheduled.Ptr(),\n\t\t\t\tActivityTaskScheduledEventAttributes: &types.ActivityTaskScheduledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskScheduledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskStarted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                          types.EventTypeActivityTaskStarted.Ptr(),\n\t\t\t\tActivityTaskStartedEventAttributes: &types.ActivityTaskStartedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskCompleted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                            types.EventTypeActivityTaskCompleted.Ptr(),\n\t\t\t\tActivityTaskCompletedEventAttributes: &types.ActivityTaskCompletedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskCompletedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                         types.EventTypeActivityTaskFailed.Ptr(),\n\t\t\t\tActivityTaskFailedEventAttributes: &types.ActivityTaskFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskTimedOut\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                           types.EventTypeActivityTaskTimedOut.Ptr(),\n\t\t\t\tActivityTaskTimedOutEventAttributes: &types.ActivityTaskTimedOutEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskTimedOutEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskCancelRequested\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeActivityTaskCancelRequested.Ptr(),\n\t\t\t\tActivityTaskCancelRequestedEventAttributes: &types.ActivityTaskCancelRequestedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskCancelRequestedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"RequestCancelActivityTaskFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeRequestCancelActivityTaskFailed.Ptr(),\n\t\t\t\tRequestCancelActivityTaskFailedEventAttributes: &types.RequestCancelActivityTaskFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.RequestCancelActivityTaskFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ActivityTaskCanceled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                           types.EventTypeActivityTaskCanceled.Ptr(),\n\t\t\t\tActivityTaskCanceledEventAttributes: &types.ActivityTaskCanceledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ActivityTaskCanceledEventAttributes{},\n\t\t},\n\t\t// Timer events\n\t\t{\n\t\t\tname: \"TimerStarted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                   types.EventTypeTimerStarted.Ptr(),\n\t\t\t\tTimerStartedEventAttributes: &types.TimerStartedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.TimerStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"TimerFired\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                 types.EventTypeTimerFired.Ptr(),\n\t\t\t\tTimerFiredEventAttributes: &types.TimerFiredEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.TimerFiredEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"CancelTimerFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                        types.EventTypeCancelTimerFailed.Ptr(),\n\t\t\t\tCancelTimerFailedEventAttributes: &types.CancelTimerFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.CancelTimerFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"TimerCanceled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                    types.EventTypeTimerCanceled.Ptr(),\n\t\t\t\tTimerCanceledEventAttributes: &types.TimerCanceledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.TimerCanceledEventAttributes{},\n\t\t},\n\t\t// Workflow execution cancellation events\n\t\t{\n\t\t\tname: \"WorkflowExecutionCancelRequested\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionCancelRequested.Ptr(),\n\t\t\t\tWorkflowExecutionCancelRequestedEventAttributes: &types.WorkflowExecutionCancelRequestedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionCancelRequestedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionCanceled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                                types.EventTypeWorkflowExecutionCanceled.Ptr(),\n\t\t\t\tWorkflowExecutionCanceledEventAttributes: &types.WorkflowExecutionCanceledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionCanceledEventAttributes{},\n\t\t},\n\t\t// External workflow execution cancellation events\n\t\t{\n\t\t\tname: \"RequestCancelExternalWorkflowExecutionInitiated\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeRequestCancelExternalWorkflowExecutionInitiated.Ptr(),\n\t\t\t\tRequestCancelExternalWorkflowExecutionInitiatedEventAttributes: &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.RequestCancelExternalWorkflowExecutionInitiatedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"RequestCancelExternalWorkflowExecutionFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeRequestCancelExternalWorkflowExecutionFailed.Ptr(),\n\t\t\t\tRequestCancelExternalWorkflowExecutionFailedEventAttributes: &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.RequestCancelExternalWorkflowExecutionFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ExternalWorkflowExecutionCancelRequested\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeExternalWorkflowExecutionCancelRequested.Ptr(),\n\t\t\t\tExternalWorkflowExecutionCancelRequestedEventAttributes: &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ExternalWorkflowExecutionCancelRequestedEventAttributes{},\n\t\t},\n\t\t// Marker and signaling events\n\t\t{\n\t\t\tname: \"MarkerRecorded\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                     types.EventTypeMarkerRecorded.Ptr(),\n\t\t\t\tMarkerRecordedEventAttributes: &types.MarkerRecordedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.MarkerRecordedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionSignaled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType:                                types.EventTypeWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tWorkflowExecutionSignaledEventAttributes: &types.WorkflowExecutionSignaledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionSignaledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionTerminated\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionTerminated.Ptr(),\n\t\t\t\tWorkflowExecutionTerminatedEventAttributes: &types.WorkflowExecutionTerminatedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionTerminatedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"WorkflowExecutionContinuedAsNew\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\t\t\tWorkflowExecutionContinuedAsNewEventAttributes: &types.WorkflowExecutionContinuedAsNewEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.WorkflowExecutionContinuedAsNewEventAttributes{},\n\t\t},\n\t\t// Child workflow execution events\n\t\t{\n\t\t\tname: \"StartChildWorkflowExecutionInitiated\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeStartChildWorkflowExecutionInitiated.Ptr(),\n\t\t\t\tStartChildWorkflowExecutionInitiatedEventAttributes: &types.StartChildWorkflowExecutionInitiatedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.StartChildWorkflowExecutionInitiatedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"StartChildWorkflowExecutionFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeStartChildWorkflowExecutionFailed.Ptr(),\n\t\t\t\tStartChildWorkflowExecutionFailedEventAttributes: &types.StartChildWorkflowExecutionFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.StartChildWorkflowExecutionFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionStarted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionStarted.Ptr(),\n\t\t\t\tChildWorkflowExecutionStartedEventAttributes: &types.ChildWorkflowExecutionStartedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionStartedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionCompleted\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCompleted.Ptr(),\n\t\t\t\tChildWorkflowExecutionCompletedEventAttributes: &types.ChildWorkflowExecutionCompletedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionCompletedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionFailed.Ptr(),\n\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionCanceled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionCanceled.Ptr(),\n\t\t\t\tChildWorkflowExecutionCanceledEventAttributes: &types.ChildWorkflowExecutionCanceledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionCanceledEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTimedOut\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tChildWorkflowExecutionTimedOutEventAttributes: &types.ChildWorkflowExecutionTimedOutEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionTimedOutEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ChildWorkflowExecutionTerminated\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeChildWorkflowExecutionTerminated.Ptr(),\n\t\t\t\tChildWorkflowExecutionTerminatedEventAttributes: &types.ChildWorkflowExecutionTerminatedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ChildWorkflowExecutionTerminatedEventAttributes{},\n\t\t},\n\t\t// External workflow execution signal events\n\t\t{\n\t\t\tname: \"SignalExternalWorkflowExecutionInitiated\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeSignalExternalWorkflowExecutionInitiated.Ptr(),\n\t\t\t\tSignalExternalWorkflowExecutionInitiatedEventAttributes: &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.SignalExternalWorkflowExecutionInitiatedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"SignalExternalWorkflowExecutionFailed\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeSignalExternalWorkflowExecutionFailed.Ptr(),\n\t\t\t\tSignalExternalWorkflowExecutionFailedEventAttributes: &types.SignalExternalWorkflowExecutionFailedEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.SignalExternalWorkflowExecutionFailedEventAttributes{},\n\t\t},\n\t\t{\n\t\t\tname: \"ExternalWorkflowExecutionSignaled\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeExternalWorkflowExecutionSignaled.Ptr(),\n\t\t\t\tExternalWorkflowExecutionSignaledEventAttributes: &types.ExternalWorkflowExecutionSignaledEventAttributes{},\n\t\t\t},\n\t\t\texpected: &types.ExternalWorkflowExecutionSignaledEventAttributes{},\n\t\t},\n\t\t// Default case: when an unknown EventType is provided\n\t\t{\n\t\t\tname: \"UnknownEventType\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventType(-1).Ptr(), // Undefined EventType\n\t\t\t},\n\t\t\texpected: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventType(-1).Ptr(),\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := getEventAttributes(tt.event)\n\t\t\tassert.Equal(t, tt.expected, result)\n\t\t})\n\t}\n}\n\nfunc TestIntSliceToSet(t *testing.T) {\n\t// Test with a slice containing unique and duplicate elements\n\tinput := []int{1, 2, 2, 3}\n\texpected := map[int]struct{}{\n\t\t1: {},\n\t\t2: {},\n\t\t3: {},\n\t}\n\n\t// Call the function and check the result\n\tresult := intSliceToSet(input)\n\tassert.Equal(t, expected, result)\n}\n\nfunc TestPrintMessage(t *testing.T) {\n\t// Create a buffer to capture the output\n\tvar buf bytes.Buffer\n\n\t// Define the message to print\n\tmsg := \"Test message\"\n\texpectedOutput := fmt.Sprintf(\"cadence: %s\\n\", msg)\n\n\t// Call the function with the buffer as output\n\tprintMessage(&buf, msg)\n\n\t// Assert that the output matches the expected output\n\tassert.Equal(t, expectedOutput, buf.String())\n}\n\nfunc TestGetRequiredInt64Option(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\toptionName    string\n\t\toptionValue   int64\n\t\tisSet         bool\n\t\texpectedValue int64\n\t\texpectedError string\n\t}{\n\t\t{\n\t\t\tname:          \"OptionSet\",\n\t\t\toptionName:    \"test-option\",\n\t\t\toptionValue:   42,\n\t\t\tisSet:         true,\n\t\t\texpectedValue: 42,\n\t\t\texpectedError: \"\",\n\t\t},\n\t\t{\n\t\t\tname:          \"OptionNotSet\",\n\t\t\toptionName:    \"test-option\",\n\t\t\toptionValue:   0,\n\t\t\tisSet:         false,\n\t\t\texpectedValue: 0,\n\t\t\texpectedError: \"option test-option is required\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\t// Create a CLI app and flag set\n\t\t\tapp := cli.NewApp()\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\n\t\t\t// Define the option flag\n\t\t\tset.Int64(tt.optionName, tt.optionValue, \"test option flag\")\n\n\t\t\t// Conditionally set the option based on the test case\n\t\t\tif tt.isSet {\n\t\t\t\trequire.NoError(t, set.Set(tt.optionName, fmt.Sprintf(\"%d\", tt.optionValue)))\n\t\t\t}\n\n\t\t\t// Create the CLI context\n\t\t\tc := cli.NewContext(app, set, nil)\n\n\t\t\t// Call getRequiredInt64Option\n\t\t\tresult, err := getRequiredInt64Option(c, tt.optionName)\n\n\t\t\t// Validate results\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedValue, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestParseSingleTs(t *testing.T) {\n\t// Test with a valid timestamp format\n\tvalidInput := \"2023-10-31T14:45:30\"\n\texpectedTime := time.Date(2023, 10, 31, 14, 45, 30, 0, time.UTC)\n\tresult, err := parseSingleTs(validInput)\n\tassert.NoError(t, err)\n\tassert.Equal(t, expectedTime, result)\n\n\t// Test with an invalid timestamp format\n\tinvalidInput := \"31-10-2023\"\n\tresult, err = parseSingleTs(invalidInput)\n\tassert.Error(t, err)\n}\n\nfunc TestPrompt(t *testing.T) {\n\t// Simulate user input for \"y\" to prevent os.Exit\n\tr, w, _ := os.Pipe()\n\torigStdin := os.Stdin\n\tdefer func() { os.Stdin = origStdin }()\n\tos.Stdin = r\n\n\tfmt.Fprint(w, \"y\\n\") // write simulated input to the pipe\n\tw.Close()            // close the write end to signal EOF\n\n\tprompt(\"Test message\") // if input is \"y\", it will not exit\n}\n\nfunc TestSerializeSearchAttributes(t *testing.T) {\n\t// Test with nil input\n\tresult, err := serializeSearchAttributes(nil)\n\tassert.NoError(t, err)\n\tassert.Nil(t, result)\n\n\t// Test with valid input\n\tinput := map[string]interface{}{\n\t\t\"key1\": \"value1\",\n\t\t\"key2\": 123,\n\t}\n\tresult, err = serializeSearchAttributes(input)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, result)\n\n\t// Verify serialized values\n\texpectedAttr := make(map[string][]byte)\n\tfor k, v := range input {\n\t\tattrBytes, _ := json.Marshal(v)\n\t\texpectedAttr[k] = attrBytes\n\t}\n\tassert.Equal(t, expectedAttr, result.IndexedFields)\n\n\t// Test with input that causes JSON marshalling error\n\tinputWithError := map[string]interface{}{\n\t\t\"key\": make(chan int), // unsupported type for JSON serialization\n\t}\n\tresult, err = serializeSearchAttributes(inputWithError)\n\tassert.Error(t, err)\n\tassert.Nil(t, result)\n}\n\nfunc TestPrintError(t *testing.T) {\n\tshowErrorStackEnv := \"CADENCE_CLI_SHOW_STACKS\"\n\n\ttests := []struct {\n\t\tname        string\n\t\tmsg         string\n\t\terr         error\n\t\tsetEnv      bool\n\t\texpectedOut string\n\t}{\n\t\t{\n\t\t\tname:        \"Error with details and stack trace prompt\",\n\t\t\tmsg:         \"Test message\",\n\t\t\terr:         fmt.Errorf(\"sample error\"),\n\t\t\tsetEnv:      false,\n\t\t\texpectedOut: fmt.Sprintf(\"%s Test message\\n%s sample error\\n('export %s=1' to see stack traces)\\n\", colorRed(\"Error:\"), colorMagenta(\"Error Details:\"), showErrorStackEnv),\n\t\t},\n\t\t{\n\t\t\tname:        \"Error without details\",\n\t\t\tmsg:         \"Test message\",\n\t\t\terr:         nil,\n\t\t\texpectedOut: fmt.Sprintf(\"%s Test message\\n\", colorRed(\"Error:\")),\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tvar buf bytes.Buffer\n\n\t\t\t// Set or unset environment variable as needed\n\t\t\tif tt.setEnv {\n\t\t\t\tos.Setenv(showErrorStackEnv, \"1\")\n\t\t\t} else {\n\t\t\t\tos.Unsetenv(showErrorStackEnv)\n\t\t\t}\n\n\t\t\t// Call the function\n\t\t\tprintError(&buf, tt.msg, tt.err)\n\n\t\t\t// Verify output matches expected output\n\t\t\toutput := buf.String()\n\t\t\tassert.Contains(t, output, tt.expectedOut)\n\n\t\t\t// If stack trace is expected, verify it is included in the output\n\t\t\tif tt.setEnv && tt.err != nil {\n\t\t\t\tassert.Contains(t, output, \"goroutine\")\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestRemovePrevious2LinesFromTerminal(t *testing.T) {\n\t// Create a buffer to capture the output\n\tvar buf bytes.Buffer\n\n\t// Call the function\n\tremovePrevious2LinesFromTerminal(&buf)\n\n\t// Verify the output matches the expected ANSI escape sequences\n\texpectedOutput := \"\\033[1A\\033[2K\\033[1A\\033[2K\"\n\tassert.Equal(t, expectedOutput, buf.String())\n}\n\nfunc TestPrettyPrintJSONObject_Error(t *testing.T) {\n\t// Define a struct containing an unmarshalable field (e.g., chan type)\n\ttype InvalidStruct struct {\n\t\tName string\n\t\tCh   chan int\n\t}\n\n\t// Create an instance of InvalidStruct\n\tinvalidObj := InvalidStruct{\n\t\tName: \"TestName\",\n\t\tCh:   make(chan int),\n\t}\n\n\t// Use bytes.Buffer as writer to capture the output\n\tvar buf bytes.Buffer\n\n\t// Call the function\n\tprettyPrintJSONObject(&buf, invalidObj)\n\n\t// Verify that output contains error message and raw object content\n\toutput := buf.String()\n\tassert.Contains(t, output, \"Error when try to print pretty:\")\n\tassert.Contains(t, output, \"TestName\")\n\tassert.Contains(t, output, \"chan int\") // Ensure unmarshalable field type information is included\n}\n\nfunc TestCreateJWT(t *testing.T) {\n\t// Generate a temporary RSA private key for testing\n\tprivateKey, err := rsa.GenerateKey(rand.Reader, 2048)\n\tassert.NoError(t, err)\n\n\t// Save the private key in PKCS#8 format to a temporary file\n\ttmpFile, err := os.CreateTemp(\"\", \"test_key.pem\")\n\tassert.NoError(t, err)\n\tdefer os.Remove(tmpFile.Name())\n\n\t// Encode the private key in PEM PKCS#8 format and write it to the temp file\n\tkeyBytes, err := x509.MarshalPKCS8PrivateKey(privateKey)\n\tassert.NoError(t, err)\n\terr = pem.Encode(tmpFile, &pem.Block{\n\t\tType:  \"PRIVATE KEY\",\n\t\tBytes: keyBytes,\n\t})\n\tassert.NoError(t, err)\n\ttmpFile.Close()\n\n\ttimeSource := clock.NewMockedTimeSource()\n\n\t// Call createJWT with the path to the temporary private key file\n\ttokenString, err := createJWT(tmpFile.Name(), timeSource)\n\tassert.NoError(t, err)\n\tassert.NotEmpty(t, tokenString)\n\n\t// Parse the token to validate its claims and signature\n\ttoken, err := jwt.ParseWithClaims(tokenString, &authorization.JWTClaims{}, func(token *jwt.Token) (interface{}, error) {\n\t\treturn &privateKey.PublicKey, nil\n\t})\n\tassert.NoError(t, err)\n\tassert.True(t, token.Valid)\n\n\t// Validate the claims\n\tclaims, ok := token.Claims.(*authorization.JWTClaims)\n\tassert.True(t, ok)\n\tassert.True(t, claims.Admin)\n\t// jwt serialization does Truncate(jwt.TimePrecision)\n\tassert.Equal(t, timeSource.Now().Truncate(jwt.TimePrecision), claims.IssuedAt.Time)\n\tassert.Equal(t, timeSource.Now().Add(10*time.Minute).Truncate(jwt.TimePrecision), claims.ExpiresAt.Time)\n}\n"
  },
  {
    "path": "tools/cli/workflow.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/service/worker/batcher\"\n)\n\nfunc newWorkflowCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"restart\",\n\t\t\tAliases: []string{\"res\"},\n\t\t\tUsage:   \"restarts a previous workflow execution\",\n\t\t\tFlags:   flagsForExecution,\n\t\t\tAction:  RestartWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"diagnose\",\n\t\t\tAliases: []string{\"diag\"},\n\t\t\tUsage:   \"diagnoses a previous workflow execution\",\n\t\t\tFlags:   flagsForExecution,\n\t\t\tAction:  DiagnoseWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"refresh-tasks\",\n\t\t\tAliases: []string{\"rt\"},\n\t\t\tUsage:   \"refreshes all the workflow tasks to resume progress\",\n\t\t\tFlags:   flagsForExecution,\n\t\t\tAction:  RefreshWorkflowTasks,\n\t\t},\n\t\t{\n\t\t\tName:        \"activity\",\n\t\t\tAliases:     []string{\"act\"},\n\t\t\tUsage:       \"operate activities of workflow\",\n\t\t\tSubcommands: newActivityCommands(),\n\t\t},\n\t\t{\n\t\t\tName:   \"show\",\n\t\t\tUsage:  \"show workflow history\",\n\t\t\tFlags:  getFlagsForShow(),\n\t\t\tAction: ShowHistory,\n\t\t},\n\t\t{\n\t\t\tName:        \"showid\",\n\t\t\tUsage:       \"show workflow history with given workflow_id and run_id (a shortcut of `show -w <wid> -r <rid>`). run_id is only required for archived history\",\n\t\t\tDescription: \"cadence workflow showid <workflow_id> <run_id>. workflow_id is required; run_id is only required for archived history\",\n\t\t\tFlags:       getFlagsForShowID(),\n\t\t\tAction:      ShowHistoryWithWID,\n\t\t},\n\t\t{\n\t\t\tName:   \"start\",\n\t\t\tUsage:  \"start a new workflow execution\",\n\t\t\tFlags:  getFlagsForStart(),\n\t\t\tAction: StartWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:   \"run\",\n\t\t\tUsage:  \"start a new workflow execution and get workflow progress\",\n\t\t\tFlags:  getFlagsForRun(),\n\t\t\tAction: RunWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"terminate\",\n\t\t\tAliases: []string{\"term\"},\n\t\t\tUsage:   \"force-stop a workflow execution (no cleanup)\",\n\t\t\tFlags:   getFlagsForTerminate(),\n\t\t\tAction:  TerminateWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"cancel\",\n\t\t\tAliases: []string{\"c\"},\n\t\t\tUsage:   \"gracefully stop a workflow execution (similar to terminate, but allows to cleanup)\",\n\t\t\tFlags:   getFlagsForCancel(),\n\t\t\tAction:  CancelWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"signal\",\n\t\t\tAliases: []string{\"s\"},\n\t\t\tUsage:   \"signal a workflow execution\",\n\t\t\tFlags:   getFlagsForSignal(),\n\t\t\tAction:  SignalWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:   \"signalwithstart\",\n\t\t\tUsage:  \"signal the current open workflow if exists, or attempt to start a new run based on IDResuePolicy and signals it\",\n\t\t\tFlags:  getFlagsForSignalWithStart(),\n\t\t\tAction: SignalWithStartWorkflowExecution,\n\t\t},\n\t\t{\n\t\t\tName:        \"list\",\n\t\t\tAliases:     []string{\"l\"},\n\t\t\tUsage:       \"list open or closed workflow executions\",\n\t\t\tDescription: \"list one page (default size 10 items) by default, use flag --pagesize to change page size\",\n\t\t\tFlags:       getFlagsForList(),\n\t\t\tAction:      ListWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"listall\",\n\t\t\tAliases: []string{\"la\"},\n\t\t\tUsage:   \"list all open or closed workflow executions\",\n\t\t\tFlags:   getFlagsForListAll(),\n\t\t\tAction:  ListAllWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:   \"listarchived\",\n\t\t\tUsage:  \"list archived workflow executions\",\n\t\t\tFlags:  getFlagsForListArchived(),\n\t\t\tAction: ListArchivedWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"scan\",\n\t\t\tAliases: []string{\"sc\", \"scanall\"},\n\t\t\tUsage: \"scan workflow executions (need to enable Cadence server on ElasticSearch). \" +\n\t\t\t\t\"It will be faster than listall, but result are not sorted.\",\n\t\t\tFlags:  getFlagsForScan(),\n\t\t\tAction: ScanAllWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:    \"count\",\n\t\t\tAliases: []string{\"cnt\"},\n\t\t\tUsage:   \"count number of workflow executions (need to enable Cadence server on ElasticSearch)\",\n\t\t\tFlags:   getFlagsForCount(),\n\t\t\tAction:  CountWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:        \"query\",\n\t\t\tUsage:       \"query workflow execution\",\n\t\t\tDescription: \"query result will be printed as JSON\",\n\t\t\tFlags:       getFlagsForQuery(),\n\t\t\tAction:      QueryWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:   \"query-types\",\n\t\t\tUsage:  \"list all available query types\",\n\t\t\tFlags:  getFlagsForStack(),\n\t\t\tAction: QueryWorkflowUsingQueryTypes,\n\t\t},\n\t\t{\n\t\t\tName:   \"stack\",\n\t\t\tUsage:  \"query workflow execution with __stack_trace as query type\",\n\t\t\tFlags:  getFlagsForStack(),\n\t\t\tAction: QueryWorkflowUsingStackTrace,\n\t\t},\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"show information of workflow execution\",\n\t\t\tFlags:   getFlagsForDescribe(),\n\t\t\tAction:  DescribeWorkflow,\n\t\t},\n\t\t{\n\t\t\tName:        \"describeid\",\n\t\t\tAliases:     []string{\"descid\"},\n\t\t\tUsage:       \"show information of workflow execution with given workflow_id and optional run_id (a shortcut of `describe -w <wid> -r <rid>`)\",\n\t\t\tDescription: \"cadence workflow describeid <workflow_id> <run_id>. workflow_id is required; run_id is optional\",\n\t\t\tFlags:       getFlagsForDescribeID(),\n\t\t\tAction:      DescribeWorkflowWithID,\n\t\t},\n\t\t{\n\t\t\tName:    \"observe\",\n\t\t\tAliases: []string{\"ob\"},\n\t\t\tUsage:   \"show the progress of workflow history\",\n\t\t\tFlags:   getFlagsForObserve(),\n\t\t\tAction:  ObserveHistory,\n\t\t},\n\t\t{\n\t\t\tName:    \"observeid\",\n\t\t\tAliases: []string{\"obid\"},\n\t\t\tUsage:   \"show the progress of workflow history with given workflow_id and optional run_id (a shortcut of `observe -w <wid> -r <rid>`)\",\n\t\t\tFlags:   getFlagsForObserveID(),\n\t\t\tAction:  ObserveHistoryWithID,\n\t\t},\n\t\t{\n\t\t\tName:    \"reset\",\n\t\t\tAliases: []string{\"rs\"},\n\t\t\tUsage:   \"reset the workflow, by either eventID or resetType.\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\t\t\tUsage:   \"WorkflowID, required\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"RunID, optional, default to the current/latest RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName: FlagEventID,\n\t\t\t\t\tUsage: \"The eventID of any event after DecisionTaskStarted you want to reset to (this event is exclusive in a new run. The new run \" +\n\t\t\t\t\t\t\"history will fork and continue from the previous eventID of this). It can be DecisionTaskCompleted, DecisionTaskFailed or others\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagReason,\n\t\t\t\t\tUsage: \"reason to do the reset, required for tracking purpose\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagResetType,\n\t\t\t\t\tUsage: \"where to reset. Support one of these: \" + strings.Join(mapKeysToArray(resetTypesMap), \",\"),\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDecisionOffset,\n\t\t\t\t\tUsage: \"based on the reset point calculated by resetType, this offset will move/offset the point by decision. Currently only negative number is supported, and only works with LastDecisionCompleted.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagResetBadBinaryChecksum,\n\t\t\t\t\tUsage: \"Binary checksum for resetType of BadBinary\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagEarliestTime,\n\t\t\t\t\tAliases: []string{\"et\"},\n\t\t\t\t\tUsage: \"EarliestTime of decision start time, required for resetType of DecisionCompletedTime.\" +\n\t\t\t\t\t\t\"Supported formats are '2006-01-02T15:04:05+07:00', raw UnixNano and \" +\n\t\t\t\t\t\t\"time range (N<duration>), where 0 < N < 1000000 and duration (full-notation/short-notation) can be second/s, \" +\n\t\t\t\t\t\t\"minute/m, hour/h, day/d, week/w, month/M or year/y. For example, '15minute' or '15m' implies last 15 minutes, \" +\n\t\t\t\t\t\t\"meaning that workflow will be reset to the first decision that completed in last 15 minutes.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagSkipSignalReapply,\n\t\t\t\t\tUsage: \"whether or not skipping signals reapply after the reset point\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: ResetWorkflow,\n\t\t},\n\t\t{\n\t\t\tName: \"reset-batch\",\n\t\t\tUsage: \"reset workflow in batch by resetType: \" + strings.Join(mapKeysToArray(resetTypesMap), \",\") +\n\t\t\t\t\"To get base workflowIDs/runIDs to reset, source is from input file or visibility query.\",\n\t\t\tArgsUsage: \"\\n\\t To reset workflows specify --input_file <csv_file> of workflow_id and run_id and run: cadence wf reset-batch --input_file <csv_file>\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInputFile,\n\t\t\t\t\tAliases: []string{\"if\"},\n\t\t\t\t\tUsage:   \"Input file to use for resetting, one workflow per line of WorkflowID and RunID. RunID is optional, default to current runID if not specified. \",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagListQuery,\n\t\t\t\t\tAliases: []string{\"lq\"},\n\t\t\t\t\tUsage:   \"visibility query to get workflows to reset\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagExcludeFile,\n\t\t\t\t\tValue: \"\",\n\t\t\t\t\tUsage: \"Another input file to use for excluding from resetting, only workflowID is needed.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName: FlagExcludeWorkflowIDByQuery,\n\t\t\t\t\tUsage: \"Another visibility SQL like query, but for excluding the results by workflowIDs. This is useful because a single query cannot do join operation. One use case is to \" +\n\t\t\t\t\t\t\"find failed workflows excluding any workflow that has another run that is open or completed.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagInputSeparator,\n\t\t\t\t\tValue: \"\\t\",\n\t\t\t\t\tUsage: \"Separator for input file(default to tab)\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagReason,\n\t\t\t\t\tUsage: \"Reason for reset, required for tracking purpose\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:   FlagParallismDeprecated,\n\t\t\t\t\tValue:  1,\n\t\t\t\t\tUsage:  \"Number of goroutines to run in parallel. Each goroutine would process one line for every second.\",\n\t\t\t\t\tHidden: true,\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagParallelism,\n\t\t\t\t\tValue: 1,\n\t\t\t\t\tUsage: \"Number of goroutines to run in parallel. Each goroutine would process one line for every second.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagSkipCurrentOpen,\n\t\t\t\t\tUsage: \"Skip the workflow if the current run is open for the same workflowID as base.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagSkipCurrentCompleted,\n\t\t\t\t\tUsage: \"Skip the workflow if the current run is completed for the same workflowID as base.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName: FlagSkipBaseIsNotCurrent,\n\t\t\t\t\t// TODO https://github.com/uber/cadence/issues/2930\n\t\t\t\t\t// The right way to prevent needs server side implementation .\n\t\t\t\t\t// This client side is only best effort\n\t\t\t\t\tUsage: \"Skip if base run is not current run.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagNonDeterministicOnly,\n\t\t\t\t\tUsage: \"Only apply onto workflows whose last event is decisionTaskFailed with non deterministic error.\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagDryRun,\n\t\t\t\t\tUsage: \"Not do real action of reset(just logging in STDOUT)\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagResetType,\n\t\t\t\t\tUsage: \"where to reset. Support one of these: \" + strings.Join(mapKeysToArray(resetTypesMap), \",\"),\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName: FlagDecisionOffset,\n\t\t\t\t\tUsage: \"based on the reset point calculated by resetType, this offset will move/offset the point by decision. \" +\n\t\t\t\t\t\t\"Limitation: currently only negative number is supported, and only works with LastDecisionCompleted.\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagResetBadBinaryChecksum,\n\t\t\t\t\tUsage: \"Binary checksum for resetType of BadBinary\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagSkipSignalReapply,\n\t\t\t\t\tUsage: \"whether or not skipping signals reapply after the reset point\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagEarliestTime,\n\t\t\t\t\tAliases: []string{\"et\"},\n\t\t\t\t\tUsage: \"EarliestTime of decision start time, required for resetType of DecisionCompletedTime.\" +\n\t\t\t\t\t\t\"Supported formats are '2006-01-02T15:04:05+07:00', raw UnixNano and \" +\n\t\t\t\t\t\t\"time range (N<duration>), where 0 < N < 1000000 and duration (full-notation/short-notation) can be second/s, \" +\n\t\t\t\t\t\t\"minute/m, hour/h, day/d, week/w, month/M or year/y. For example, '15minute' or '15m' implies last 15 minutes, \" +\n\t\t\t\t\t\t\"meaning that workflow will be reset to the first decision that completed in last 15 minutes.\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: ResetInBatch,\n\t\t},\n\t\t{\n\t\t\tName:        \"batch\",\n\t\t\tUsage:       \"batch operation on a list of workflows from query.\",\n\t\t\tSubcommands: newBatchCommands(),\n\t\t\tArgsUsage: \"\\n\\t To make a batch operation use wf batch start command and specify --batch_type to terminate/signal/cancel workflows.\\n\" +\n\t\t\t\t\"\\t ex: to batch terminate workflows run: cadence batch start --batch_type terminate --query <targeted_workflows_query>\\n\" +\n\t\t\t\t\"\\t cadence wf batch terminate - is used to terminate a batch operation not workflows.\\n\" +\n\t\t\t\t\"\\t To inspect the progress run: cadence wf batch desc --job_id <your_job_id>\",\n\t\t},\n\t}\n}\n\nfunc newActivityCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"complete\",\n\t\t\tAliases: []string{\"comp\"},\n\t\t\tUsage:   \"complete an activity\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagActivityID,\n\t\t\t\t\tAliases: []string{\"aid\"},\n\t\t\t\t\tUsage:   \"The activityID to operate on\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagResult,\n\t\t\t\t\tUsage: \"Result of the activity\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagIdentity,\n\t\t\t\t\tUsage: \"Identity of the operator\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: CompleteActivity,\n\t\t},\n\t\t{\n\t\t\tName:  \"fail\",\n\t\t\tUsage: \"fail an activity\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagWorkflowID,\n\t\t\t\t\tAliases: []string{\"wid\", \"w\"},\n\t\t\t\t\tUsage:   \"WorkflowID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagRunID,\n\t\t\t\t\tAliases: []string{\"rid\", \"r\"},\n\t\t\t\t\tUsage:   \"RunID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagActivityID,\n\t\t\t\t\tAliases: []string{\"aid\"},\n\t\t\t\t\tUsage:   \"The activityID to operate on\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagReason,\n\t\t\t\t\tUsage: \"Reason to fail the activity\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagDetail,\n\t\t\t\t\tUsage: \"Detail to fail the activity\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:  FlagIdentity,\n\t\t\t\t\tUsage: \"Identity of the operator\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: FailActivity,\n\t\t},\n\t}\n}\n\nfunc newBatchCommands() []*cli.Command {\n\treturn []*cli.Command{\n\t\t{\n\t\t\tName:    \"describe\",\n\t\t\tAliases: []string{\"desc\"},\n\t\t\tUsage:   \"Describe a batch operation job\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagJobID,\n\t\t\t\t\tAliases: []string{\"jid\"},\n\t\t\t\t\tUsage:   \"Batch Job ID\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: DescribeBatchJob,\n\t\t},\n\t\t{\n\t\t\tName:  \"terminate\",\n\t\t\tUsage: \"terminate a batch operation job\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagJobID,\n\t\t\t\t\tAliases: []string{\"jid\"},\n\t\t\t\t\tUsage:   \"Batch Job ID\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagReason,\n\t\t\t\t\tAliases: []string{\"re\"},\n\t\t\t\t\tUsage:   \"Reason to stop this batch job\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: TerminateBatchJob,\n\t\t},\n\t\t{\n\t\t\tName:    \"list\",\n\t\t\tAliases: []string{\"l\"},\n\t\t\tUsage:   \"Describe a batch operation job\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagPageSize,\n\t\t\t\t\tAliases: []string{\"ps\"},\n\t\t\t\t\tValue:   30,\n\t\t\t\t\tUsage:   \"Result page size\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: ListBatchJobs,\n\t\t},\n\t\t{\n\t\t\tName:  \"start\",\n\t\t\tUsage: \"Start a batch operation job\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagListQuery,\n\t\t\t\t\tAliases: []string{\"q\"},\n\t\t\t\t\tUsage:   \"Query to get workflows for being executed this batch operation\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagReason,\n\t\t\t\t\tAliases: []string{\"re\"},\n\t\t\t\t\tUsage:   \"Reason to run this batch job\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagBatchType,\n\t\t\t\t\tAliases: []string{\"bt\"},\n\t\t\t\t\tUsage:   \"Types supported: \" + strings.Join(batcher.AllBatchTypes, \",\"),\n\t\t\t\t},\n\t\t\t\t// below are optional\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagSignalName,\n\t\t\t\t\tAliases: []string{\"sn\"},\n\t\t\t\t\tUsage:   \"Required for batch signal\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagInput,\n\t\t\t\t\tAliases: []string{\"in\"},\n\t\t\t\t\tUsage:   \"Optional input of signal\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagSourceCluster,\n\t\t\t\t\tAliases: []string{\"sc\"},\n\t\t\t\t\tUsage:   \"Required for batch replicate\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    FlagTargetCluster,\n\t\t\t\t\tAliases: []string{\"tc\"},\n\t\t\t\t\tUsage:   \"Required for batch replicate\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagRPS,\n\t\t\t\t\tValue: batcher.DefaultRPS,\n\t\t\t\t\tUsage: \"RPS of processing\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  FlagYes,\n\t\t\t\t\tUsage: \"Optional flag to disable confirmation prompt\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagPageSize,\n\t\t\t\t\tValue: batcher.DefaultPageSize,\n\t\t\t\t\tUsage: \"PageSize of processiing\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagRetryAttempts,\n\t\t\t\t\tValue: batcher.DefaultAttemptsOnRetryableError,\n\t\t\t\t\tUsage: \"Retry attempts for retriable errors\",\n\t\t\t\t},\n\t\t\t\t// TODO duration should use DurationFlag instead of IntFlag\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:    FlagActivityHeartBeatTimeout,\n\t\t\t\t\tAliases: []string{\"hbt\"},\n\t\t\t\t\tValue:   int(batcher.DefaultActivityHeartBeatTimeout / time.Second),\n\t\t\t\t\tUsage:   \"Heartbeat timeout for batcher activity in seconds\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagConcurrency,\n\t\t\t\t\tValue: batcher.DefaultConcurrency,\n\t\t\t\t\tUsage: \"Concurrency of batch activity\",\n\t\t\t\t},\n\t\t\t\t&cli.IntFlag{\n\t\t\t\t\tName:  FlagMaxActivityRetries,\n\t\t\t\t\tValue: batcher.DefaultMaxActivityRetries,\n\t\t\t\t\tUsage: \"Max retries of batch activity, before retrying the whole workflow (0 means unlimited)\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: StartBatchJob,\n\t\t},\n\t}\n}\n"
  },
  {
    "path": "tools/cli/workflow_batch_commands.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/pborman/uuid\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/constants\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// TerminateBatchJob stops abatch job\nfunc TerminateBatchJob(c *cli.Context) error {\n\tjobID, err := getRequiredOption(c, FlagJobID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\treason, err := getRequiredOption(c, FlagReason)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tsvcClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\terr = svcClient.TerminateWorkflowExecution(\n\t\ttcCtx,\n\t\t&types.TerminateWorkflowExecutionRequest{\n\t\t\tDomain: constants.BatcherLocalDomainName,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: jobID,\n\t\t\t\tRunID:      \"\",\n\t\t\t},\n\t\t\tReason:   reason,\n\t\t\tIdentity: getCliIdentity(),\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to terminate batch job\", err)\n\t}\n\toutput := map[string]interface{}{\n\t\t\"msg\": \"batch job is terminated\",\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), output)\n\treturn nil\n}\n\n// DescribeBatchJob describe the status of the batch job\nfunc DescribeBatchJob(c *cli.Context) error {\n\tjobID, err := getRequiredOption(c, FlagJobID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tsvcClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\twf, err := svcClient.DescribeWorkflowExecution(\n\t\ttcCtx,\n\t\t&types.DescribeWorkflowExecutionRequest{\n\t\t\tDomain: constants.BatcherLocalDomainName,\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: jobID,\n\t\t\t\tRunID:      \"\",\n\t\t\t},\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to describe batch job\", err)\n\t}\n\n\toutput := map[string]interface{}{}\n\tif wf.WorkflowExecutionInfo.CloseStatus != nil {\n\t\tif wf.WorkflowExecutionInfo.GetCloseStatus() != types.WorkflowExecutionCloseStatusCompleted {\n\t\t\toutput[\"msg\"] = \"batch job stopped status: \" + wf.WorkflowExecutionInfo.GetCloseStatus().String()\n\t\t} else {\n\t\t\toutput[\"msg\"] = \"batch job is finished successfully\"\n\t\t}\n\t} else {\n\t\toutput[\"msg\"] = \"batch job is running\"\n\t\tif len(wf.PendingActivities) > 0 {\n\t\t\thbdBinary := wf.PendingActivities[0].HeartbeatDetails\n\t\t\thbd := batcher.HeartBeatDetails{}\n\t\t\terr := json.Unmarshal(hbdBinary, &hbd)\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Failed to describe batch job\", err)\n\t\t\t}\n\t\t\toutput[\"progress\"] = hbd\n\t\t}\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), output)\n\treturn nil\n}\n\n// ListBatchJobs list the started batch jobs\nfunc ListBatchJobs(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tpageSize := c.Int(FlagPageSize)\n\tsvcClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\tresp, err := svcClient.ListWorkflowExecutions(\n\t\ttcCtx,\n\t\t&types.ListWorkflowExecutionsRequest{\n\t\t\tDomain:   constants.BatcherLocalDomainName,\n\t\t\tPageSize: int32(pageSize),\n\t\t\tQuery:    fmt.Sprintf(\"CustomDomain = '%v'\", domain),\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to list batch jobs\", err)\n\t}\n\toutput := make([]interface{}, 0, len(resp.Executions))\n\tfor _, wf := range resp.Executions {\n\t\tjob := map[string]string{\n\t\t\t\"jobID\":     wf.Execution.GetWorkflowID(),\n\t\t\t\"startTime\": timestampToString(wf.GetStartTime(), false),\n\t\t\t\"reason\":    string(wf.Memo.Fields[\"Reason\"]),\n\t\t\t\"operator\":  string(wf.SearchAttributes.IndexedFields[\"Operator\"]),\n\t\t}\n\n\t\tif wf.CloseStatus != nil {\n\t\t\tjob[\"status\"] = wf.CloseStatus.String()\n\t\t\tjob[\"closeTime\"] = timestampToString(wf.GetCloseTime(), false)\n\t\t} else {\n\t\t\tjob[\"status\"] = \"RUNNING\"\n\t\t}\n\n\t\toutput = append(output, job)\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), output)\n\treturn nil\n}\n\n// StartBatchJob starts a batch job\nfunc StartBatchJob(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tquery, err := getRequiredOption(c, FlagListQuery)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\treason, err := getRequiredOption(c, FlagReason)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tbatchType, err := getRequiredOption(c, FlagBatchType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tif !validateBatchType(batchType) {\n\t\treturn commoncli.Problem(\"batchType is not valid, supported:\"+strings.Join(batcher.AllBatchTypes, \",\"), nil)\n\t}\n\toperator := getCurrentUserFromEnv()\n\tvar sigName, sigVal string\n\tif batchType == batcher.BatchTypeSignal {\n\t\tsigName, err = getRequiredOption(c, FlagSignalName)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t\t}\n\t\tsigVal, err = getRequiredOption(c, FlagInput)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t\t}\n\t}\n\tvar sourceCluster, targetCluster string\n\tif batchType == batcher.BatchTypeReplicate {\n\t\tsourceCluster, err = getRequiredOption(c, FlagSourceCluster)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t\t}\n\t\ttargetCluster, err = getRequiredOption(c, FlagTargetCluster)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t\t}\n\t}\n\trps := c.Int(FlagRPS)\n\tpageSize := c.Int(FlagPageSize)\n\tconcurrency := c.Int(FlagConcurrency)\n\tretryAttempt := c.Int(FlagRetryAttempts)\n\theartBeatTimeout := time.Duration(c.Int(FlagActivityHeartBeatTimeout)) * time.Second\n\tmaxActivityRetries := c.Int(FlagMaxActivityRetries)\n\n\tsvcClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\tresp, err := svcClient.CountWorkflowExecutions(\n\t\ttcCtx,\n\t\t&types.CountWorkflowExecutionsRequest{\n\t\t\tDomain: domain,\n\t\t\tQuery:  query,\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to count impacting workflows for starting a batch job\", err)\n\t}\n\tfmt.Printf(\"This batch job will be operating on %v workflows.\\n\", resp.GetCount())\n\tif !c.Bool(FlagYes) {\n\t\treader := bufio.NewReader(os.Stdin)\n\t\tfor {\n\t\t\tfmt.Print(\"Please confirm[Yes/No]:\")\n\t\t\ttext, err := reader.ReadString('\\n')\n\t\t\tif err != nil {\n\t\t\t\treturn commoncli.Problem(\"Failed to  get confirmation for starting a batch job\", err)\n\t\t\t}\n\t\t\tif strings.EqualFold(strings.TrimSpace(text), \"yes\") {\n\t\t\t\tbreak\n\t\t\t} else {\n\t\t\t\tfmt.Println(\"Batch job is not started\")\n\t\t\t\treturn nil\n\t\t\t}\n\t\t}\n\n\t}\n\ttcCtx, cancel, err = newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in creating context:\", err)\n\t}\n\n\tparams := batcher.BatchParams{\n\t\tDomainName: domain,\n\t\tQuery:      query,\n\t\tReason:     reason,\n\t\tBatchType:  batchType,\n\t\tSignalParams: batcher.SignalParams{\n\t\t\tSignalName: sigName,\n\t\t\tInput:      sigVal,\n\t\t},\n\t\tReplicateParams: batcher.ReplicateParams{\n\t\t\tSourceCluster: sourceCluster,\n\t\t\tTargetCluster: targetCluster,\n\t\t},\n\t\tRPS:                      rps,\n\t\tConcurrency:              concurrency,\n\t\tPageSize:                 pageSize,\n\t\tAttemptsOnRetryableError: retryAttempt,\n\t\tActivityHeartBeatTimeout: heartBeatTimeout,\n\t\tMaxActivityRetries:       maxActivityRetries,\n\t}\n\tinput, err := json.Marshal(params)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to encode batch job parameters\", err)\n\t}\n\tmemo, err := getWorkflowMemo(map[string]interface{}{\n\t\t\"Reason\": reason,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to encode batch job memo\", err)\n\t}\n\tsearchAttributes, err := serializeSearchAttributes(map[string]interface{}{\n\t\t\"CustomDomain\": domain,\n\t\t\"Operator\":     operator,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to encode batch job search attributes\", err)\n\t}\n\tworkflowID := uuid.NewRandom().String()\n\trequest := &types.StartWorkflowExecutionRequest{\n\t\tDomain:                              constants.BatcherLocalDomainName,\n\t\tRequestID:                           uuid.New(),\n\t\tWorkflowID:                          workflowID,\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32(batcher.InfiniteDuration.Seconds())),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(int32(defaultDecisionTimeoutInSeconds)),\n\t\tTaskList:                            &types.TaskList{Name: batcher.BatcherTaskListName},\n\t\tMemo:                                memo,\n\t\tSearchAttributes:                    searchAttributes,\n\t\tRetryPolicy:                         copyRetryPolicyFromWorkflow(),\n\t\tWorkflowType:                        &types.WorkflowType{Name: batcher.BatchWFTypeName},\n\t\tInput:                               input,\n\t}\n\t_, err = svcClient.StartWorkflowExecution(tcCtx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to start batch job\", err)\n\t}\n\toutput := map[string]interface{}{\n\t\t\"msg\":   \"batch job is started\",\n\t\t\"jobID\": workflowID,\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), output)\n\treturn nil\n}\n\nfunc copyRetryPolicyFromWorkflow() *types.RetryPolicy {\n\treturn &types.RetryPolicy{\n\t\tInitialIntervalInSeconds:    int32(batcher.BatchActivityRetryPolicy.InitialInterval.Seconds()),\n\t\tBackoffCoefficient:          batcher.BatchActivityRetryPolicy.BackoffCoefficient,\n\t\tMaximumIntervalInSeconds:    int32(batcher.BatchActivityRetryPolicy.MaximumInterval.Seconds()),\n\t\tNonRetriableErrorReasons:    batcher.BatchActivityRetryPolicy.NonRetriableErrorReasons,\n\t\tExpirationIntervalInSeconds: int32(batcher.BatchActivityRetryPolicy.ExpirationInterval.Seconds()),\n\t}\n}\n\nfunc validateBatchType(bt string) bool {\n\tfor _, b := range batcher.AllBatchTypes {\n\t\tif b == bt {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
  },
  {
    "path": "tools/cli/workflow_batch_commands_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"encoding/json\"\n\t\"errors\"\n\t\"flag\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/worker/batcher\"\n)\n\nfunc TestStartBatchJob(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetup          func(*frontend.MockClient)\n\t\tflags          map[string]interface{}\n\t\texpectedError  string\n\t\texpectedOutput string\n\t}{\n\t\t{\n\t\t\tname: \"Valid Start Batch Job\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{\n\t\t\t\t\tCount: 100,\n\t\t\t\t}, nil)\n\t\t\t\tmockClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{\n\t\t\t\t\tRunID: \"run-id-example\",\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:     \"test-domain\",\n\t\t\t\tFlagListQuery:  \"workflowType='batch'\",\n\t\t\t\tFlagReason:     \"Testing batch job\",\n\t\t\t\tFlagBatchType:  batcher.BatchTypeSignal,\n\t\t\t\tFlagSignalName: \"test-signal\",\n\t\t\t\tFlagInput:      \"test-input\",\n\t\t\t\tFlagYes:        true,\n\t\t\t},\n\t\t\texpectedError:  \"\",\n\t\t\texpectedOutput: \"batch job is started\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing Domain\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagListQuery: \"workflowType='batch'\",\n\t\t\t\tFlagReason:    \"Testing batch job\",\n\t\t\t\tFlagBatchType: batcher.BatchTypeSignal,\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option domain is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing ListQuery\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:    \"test-domain\",\n\t\t\t\tFlagReason:    \"Testing batch job\",\n\t\t\t\tFlagBatchType: batcher.BatchTypeSignal,\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option query is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing Signal\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:    \"test-domain\",\n\t\t\t\tFlagListQuery: \"workflowType='batch'\",\n\t\t\t\tFlagReason:    \"Testing batch job\",\n\t\t\t\tFlagBatchType: batcher.BatchTypeSignal,\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option signal_name is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing Input\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:     \"test-domain\",\n\t\t\t\tFlagListQuery:  \"workflowType='batch'\",\n\t\t\t\tFlagReason:     \"Testing batch job\",\n\t\t\t\tFlagBatchType:  batcher.BatchTypeSignal,\n\t\t\t\tFlagSignalName: \"test-signal\",\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option input is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing Reason\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:    \"test-domain\",\n\t\t\t\tFlagListQuery: \"workflowType='batch'\",\n\t\t\t\tFlagBatchType: batcher.BatchTypeSignal,\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option reason is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Invalid Batch Type\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:    \"test-domain\",\n\t\t\t\tFlagListQuery: \"workflowType='batch'\",\n\t\t\t\tFlagReason:    \"Testing batch job\",\n\t\t\t\tFlagBatchType: \"invalidBatchType\",\n\t\t\t},\n\t\t\texpectedError: \"batchType is not valid, supported:terminate,cancel,signal,replicate\",\n\t\t},\n\t\t{\n\t\t\tname: \"Count Workflow Executions Failure\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"count error\"))\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:     \"test-domain\",\n\t\t\t\tFlagListQuery:  \"workflowType='batch'\",\n\t\t\t\tFlagReason:     \"Testing batch job\",\n\t\t\t\tFlagBatchType:  batcher.BatchTypeSignal,\n\t\t\t\tFlagSignalName: \"test-signal\",\n\t\t\t\tFlagInput:      \"test-input\",\n\t\t\t\tFlagYes:        true,\n\t\t\t},\n\t\t\texpectedError: \"Failed to count impacting workflows for starting a batch job: count error\",\n\t\t},\n\t\t{\n\t\t\tname: \"Start Workflow Execution Failure\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{\n\t\t\t\t\tCount: 100,\n\t\t\t\t}, nil)\n\t\t\t\tmockClient.EXPECT().StartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"start error\"))\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:     \"test-domain\",\n\t\t\t\tFlagListQuery:  \"workflowType='batch'\",\n\t\t\t\tFlagReason:     \"Testing batch job\",\n\t\t\t\tFlagBatchType:  batcher.BatchTypeSignal,\n\t\t\t\tFlagSignalName: \"test-signal\",\n\t\t\t\tFlagInput:      \"test-input\",\n\t\t\t\tFlagYes:        true,\n\t\t\t},\n\t\t\texpectedError: \"Failed to start batch job: start error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\t\t\tmockClient := frontend.NewMockClient(mockCtrl)\n\t\t\tioHandler := &testIOHandler{}\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: mockClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tfor k, v := range tt.flags {\n\t\t\t\tswitch val := v.(type) {\n\t\t\t\tcase string:\n\t\t\t\t\t_ = set.String(k, val, \"\")\n\t\t\t\tcase int:\n\t\t\t\t\t_ = set.Int(k, val, \"\")\n\t\t\t\tcase bool:\n\t\t\t\t\t_ = set.Bool(k, val, \"\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tc := cli.NewContext(app, set, nil)\n\t\t\ttt.setup(mockClient)\n\n\t\t\terr := StartBatchJob(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualOutput map[string]interface{}\n\t\t\t\terr = json.Unmarshal(ioHandler.outputBytes.Bytes(), &actualOutput)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedOutput, actualOutput[\"msg\"])\n\t\t\t\tassert.Regexp(t, `^[a-f0-9\\-]{36}$`, actualOutput[\"jobID\"])\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestTerminateBatchJob(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetup          func(*frontend.MockClient)\n\t\tflags          map[string]interface{}\n\t\texpectedError  string\n\t\texpectedOutput map[string]interface{}\n\t}{\n\t\t{\n\t\t\tname: \"Valid Termination\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID:  \"example-workflow-1\",\n\t\t\t\tFlagReason: \"Testing termination\",\n\t\t\t},\n\t\t\texpectedError:  \"\",\n\t\t\texpectedOutput: map[string]interface{}{\"msg\": \"batch job is terminated\"},\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing JobID\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagReason: \"Testing termination\",\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option job_id is required\",\n\t\t},\n\t\t{\n\t\t\tname:  \"Missing Reason\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID: \"example-workflow-1\",\n\t\t\t},\n\t\t\texpectedError: \"Required flag not found: : option reason is required\",\n\t\t},\n\t\t{\n\t\t\tname: \"Terminate Failure\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().TerminateWorkflowExecution(gomock.Any(), gomock.Any()).Return(errors.New(\"termination error\"))\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID:  \"example-workflow-1\",\n\t\t\t\tFlagReason: \"Testing termination\",\n\t\t\t},\n\t\t\texpectedError: \"Failed to terminate batch job: termination error\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\t\t\tmockClient := frontend.NewMockClient(mockCtrl)\n\t\t\tioHandler := &testIOHandler{}\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: mockClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tfor k, v := range tt.flags {\n\t\t\t\tswitch val := v.(type) {\n\t\t\t\tcase string:\n\t\t\t\t\t_ = set.String(k, val, \"\")\n\t\t\t\tcase int:\n\t\t\t\t\t_ = set.Int(k, val, \"\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tc := cli.NewContext(app, set, nil)\n\t\t\ttt.setup(mockClient)\n\n\t\t\terr := TerminateBatchJob(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualOutput map[string]interface{}\n\t\t\t\terr = json.Unmarshal(ioHandler.outputBytes.Bytes(), &actualOutput)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedOutput, actualOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestDescribeBatchJob(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetup          func(*frontend.MockClient)\n\t\tflags          map[string]interface{}\n\t\texpectedError  string\n\t\texpectedOutput map[string]interface{}\n\t}{\n\t\t{\n\t\t\tname: \"Valid Batch Job\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"example-workflow-1\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartTime: common.Int64Ptr(1697018400),\n\t\t\t\t\t},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tHeartbeatDetails: json.RawMessage(`{\"PageToken\": null, \"CurrentPage\": 50, \"TotalEstimate\": 100, \"SuccessCount\": 0, \"ErrorCount\": 0}`),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID: \"example-workflow-1\",\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\texpectedOutput: map[string]interface{}{\n\t\t\t\t\"msg\": \"batch job is finished successfully\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Batch Job Running\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\tCloseStatus: nil,\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"example-workflow-2\",\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tHeartbeatDetails: json.RawMessage(`{\"PageToken\": null, \"CurrentPage\": 30, \"TotalEstimate\": 100, \"SuccessCount\": 10, \"ErrorCount\": 0}`),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID: \"example-workflow-2\",\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\texpectedOutput: map[string]interface{}{\n\t\t\t\t\"msg\": \"batch job is running\",\n\t\t\t\t\"progress\": batcher.HeartBeatDetails{\n\t\t\t\t\tPageToken:     nil,\n\t\t\t\t\tCurrentPage:   30,\n\t\t\t\t\tTotalEstimate: 100,\n\t\t\t\t\tSuccessCount:  10,\n\t\t\t\t\tErrorCount:    0,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"Error when describing job\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"service error\"))\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagJobID: \"error-job\",\n\t\t\t},\n\t\t\texpectedError: \"Failed to describe batch job: service error\",\n\t\t},\n\t\t{\n\t\t\tname:          \"Missing Job ID\",\n\t\t\tsetup:         func(mockClient *frontend.MockClient) {},\n\t\t\tflags:         map[string]interface{}{},\n\t\t\texpectedError: \"Required flag not found: : option job_id is required\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tdefer mockCtrl.Finish()\n\t\t\tmockClient := frontend.NewMockClient(mockCtrl)\n\t\t\tioHandler := &testIOHandler{}\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: mockClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tfor k, v := range tt.flags {\n\t\t\t\tswitch val := v.(type) {\n\t\t\t\tcase string:\n\t\t\t\t\t_ = set.String(k, val, \"\")\n\t\t\t\tcase int:\n\t\t\t\t\t_ = set.Int(k, val, \"\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tc := cli.NewContext(app, set, nil)\n\t\t\ttt.setup(mockClient)\n\n\t\t\terr := DescribeBatchJob(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualOutput map[string]interface{}\n\t\t\t\terr = json.Unmarshal(ioHandler.outputBytes.Bytes(), &actualOutput)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tif progressRaw, exists := actualOutput[\"progress\"]; exists {\n\t\t\t\t\tvar progress batcher.HeartBeatDetails\n\t\t\t\t\tprogressBytes, err := json.Marshal(progressRaw)\n\t\t\t\t\tif err == nil {\n\t\t\t\t\t\tjson.Unmarshal(progressBytes, &progress)\n\t\t\t\t\t\tactualOutput[\"progress\"] = progress\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tassert.Equal(t, tt.expectedOutput, actualOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestListBatchJobs(t *testing.T) {\n\ttests := []struct {\n\t\tname           string\n\t\tsetup          func(*frontend.MockClient)\n\t\tflags          map[string]interface{}\n\t\texpectedError  string\n\t\texpectedOutput []map[string]string\n\t}{\n\t\t{\n\t\t\tname: \"Valid Batch Job\",\n\t\t\tsetup: func(mockClient *frontend.MockClient) {\n\t\t\t\tmockClient.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{\n\t\t\t\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\t\tWorkflowID: \"example-workflow-1\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tStartTime: common.Int64Ptr(1697018400),\n\t\t\t\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\t\t\t\tFields: map[string][]byte{\n\t\t\t\t\t\t\t\t\t\"Reason\": []byte(\"Testing reason\"),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\t\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\t\t\t\t\"Operator\": []byte(\"test-operator\"),\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t\tNextPageToken: []byte(\"test-next-token\"),\n\t\t\t\t}, nil)\n\t\t\t},\n\t\t\tflags: map[string]interface{}{\n\t\t\t\tFlagDomain:   \"test-domain\",\n\t\t\t\tFlagPageSize: 100,\n\t\t\t},\n\t\t\texpectedError: \"\",\n\t\t\texpectedOutput: []map[string]string{\n\t\t\t\t{\n\t\t\t\t\t\"jobID\":     \"example-workflow-1\",\n\t\t\t\t\t\"startTime\": \"1970-01-01T00:00:01Z\",\n\t\t\t\t\t\"reason\":    \"Testing reason\",\n\t\t\t\t\t\"operator\":  \"test-operator\",\n\t\t\t\t\t\"status\":    \"COMPLETED\",\n\t\t\t\t\t\"closeTime\": \"1970-01-01T00:00:00Z\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:          \"Missing Domain\",\n\t\t\tsetup:         func(mockClient *frontend.MockClient) {},\n\t\t\tflags:         map[string]interface{}{},\n\t\t\texpectedError: \"Required flag not found: : option domain is required\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tmockClient := frontend.NewMockClient(mockCtrl)\n\t\t\tioHandler := &testIOHandler{}\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: mockClient,\n\t\t\t}, WithIOHandler(ioHandler))\n\t\t\tset := flag.NewFlagSet(\"test\", 0)\n\t\t\tfor k, v := range tt.flags {\n\t\t\t\tswitch val := v.(type) {\n\t\t\t\tcase string:\n\t\t\t\t\t_ = set.String(k, val, \"\")\n\t\t\t\tcase int:\n\t\t\t\t\t_ = set.Int(k, val, \"\")\n\t\t\t\t}\n\t\t\t}\n\t\t\tc := cli.NewContext(app, set, nil)\n\t\t\ttt.setup(mockClient)\n\n\t\t\terr := ListBatchJobs(c)\n\n\t\t\tif tt.expectedError != \"\" {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tt.expectedError)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tvar actualOutput []map[string]string\n\t\t\t\terr = json.Unmarshal(ioHandler.outputBytes.Bytes(), &actualOutput)\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expectedOutput, actualOutput)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestValidateBatchType(t *testing.T) {\n\t// Mock batch types for testing\n\tbatcher.AllBatchTypes = []string{\"signal\", \"replicate\", \"terminate\"}\n\n\ttests := []struct {\n\t\tname      string\n\t\tbatchType string\n\t\texpected  bool\n\t}{\n\t\t{\n\t\t\tname:      \"Valid batch type - signal\",\n\t\t\tbatchType: \"signal\",\n\t\t\texpected:  true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Valid batch type - replicate\",\n\t\t\tbatchType: \"replicate\",\n\t\t\texpected:  true,\n\t\t},\n\t\t{\n\t\t\tname:      \"Invalid batch type\",\n\t\t\tbatchType: \"invalid\",\n\t\t\texpected:  false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tresult := validateBatchType(tt.batchType)\n\t\t\tif result != tt.expected {\n\t\t\t\tt.Errorf(\"expected %v, got %v for batch type %v\", tt.expected, result, tt.batchType)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/cli/workflow_commands.go",
    "content": "// Copyright (c) 2017-2020 Uber Technologies Inc.\n// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage cli\n\nimport (\n\t\"bufio\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"math/rand\"\n\t\"os\"\n\t\"regexp\"\n\t\"strconv\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/olekukonko/tablewriter\"\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/clock\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/service/history/execution\"\n\t\"github.com/uber/cadence/tools/common/commoncli\"\n)\n\n// RestartWorkflow restarts a workflow execution\nfunc RestartWorkflow(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tresp, err := wfClient.RestartWorkflowExecution(\n\t\tctx,\n\t\t&types.RestartWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t}, Identity: getCliIdentity(),\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Restart workflow failed.\", err)\n\t}\n\tfmt.Printf(\"Restarted Workflow Id: %s, run Id: %s\\n\", wid, resp.GetRunID())\n\treturn nil\n}\n\n// DiagnoseWorkflow diagnoses a workflow execution\nfunc DiagnoseWorkflow(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid, err := getRequiredOption(c, FlagRunID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tresp, err := wfClient.DiagnoseWorkflowExecution(\n\t\tctx,\n\t\t&types.DiagnoseWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t},\n\t\t\tIdentity: getCliIdentity(),\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Diagnose workflow failed.\", err)\n\t}\n\tfmt.Println(\"Workflow diagnosis started. Query the diagnostic workflow to get diagnostics report.\")\n\tfmt.Println(\"============Diagnostic Workflow details============\")\n\tfmt.Printf(\"Domain: %s, Workflow Id: %s, Run Id: %s\\n\", resp.GetDomain(), resp.GetDiagnosticWorkflowExecution().GetWorkflowID(), resp.GetDiagnosticWorkflowExecution().GetRunID())\n\treturn nil\n}\n\n// RefreshWorkflowTasks refreshes all the tasks of a workflow\nfunc RefreshWorkflowTasks(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\n\tctx, cancel, err := newContext(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tdefer cancel()\n\n\terr = wfClient.RefreshWorkflowTasks(ctx, &types.RefreshWorkflowTasksRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Refresh workflow tasks failed.\", err)\n\t}\n\tfmt.Println(\"Refresh workflow tasks succeeded.\")\n\treturn nil\n}\n\n// ShowHistory shows the history of given workflow execution based on workflowID and runID.\nfunc ShowHistory(c *cli.Context) error {\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\treturn showHistoryHelper(c, wid, rid)\n}\n\n// ShowHistoryWithWID shows the history of given workflow with workflow_id\nfunc ShowHistoryWithWID(c *cli.Context) error {\n\tif !c.Args().Present() {\n\t\treturn commoncli.Problem(\"Argument workflow_id is required.\", nil)\n\t}\n\twid := c.Args().First()\n\trid := \"\"\n\tif c.NArg() >= 2 {\n\t\trid = c.Args().Get(1)\n\t}\n\treturn showHistoryHelper(c, wid, rid)\n}\n\nfunc showHistoryHelper(c *cli.Context, wid, rid string) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tprintDateTime := c.Bool(FlagPrintDateTime)\n\tprintRawTime := c.Bool(FlagPrintRawTime)\n\tprintFully := c.Bool(FlagPrintFullyDetail)\n\tprintVersion := c.Bool(FlagPrintEventVersion)\n\toutputFileName := c.String(FlagOutputFilename)\n\tvar maxFieldLength int\n\tif c.IsSet(FlagMaxFieldLength) || !printFully {\n\t\tmaxFieldLength = c.Int(FlagMaxFieldLength)\n\t}\n\tresetPointsOnly := c.Bool(FlagResetPointsOnly)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\tconsistencyLevel, err := getOptionWithSerializer[types.QueryConsistencyLevel](c, FlagQueryConsistencyLevel, mapQueryConsistencyLevelFromFlag)\n\tif err != nil {\n\t\treturn commoncli.Problem(err.Error(), nil)\n\t}\n\n\thistory, err := GetHistory(ctx, wfClient, domain, wid, rid, consistencyLevel)\n\tif err != nil {\n\t\treturn commoncli.Problem(fmt.Sprintf(\"Failed to get history on workflow id: %s, run id: %s.\", wid, rid), err)\n\t}\n\n\tprevEvent := types.HistoryEvent{}\n\tif printFully { // dump everything\n\t\tfor _, e := range history.Events {\n\t\t\tif resetPointsOnly {\n\t\t\t\tif prevEvent.GetEventType() != types.EventTypeDecisionTaskStarted {\n\t\t\t\t\tprevEvent = *e\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tprevEvent = *e\n\t\t\t}\n\t\t\tfmt.Println(anyToString(e, true, maxFieldLength))\n\t\t}\n\t} else if c.IsSet(FlagEventID) { // only dump that event\n\t\teventID := c.Int(FlagEventID)\n\t\tif eventID <= 0 || eventID > len(history.Events) {\n\t\t\treturn commoncli.Problem(\"EventId out of range.\", fmt.Errorf(\"number should be 1 - %d inclusive\", len(history.Events)))\n\t\t}\n\t\te := history.Events[eventID-1]\n\t\tfmt.Println(anyToString(e, true, 0))\n\t} else { // use table to pretty output, will trim long text\n\t\ttable := tablewriter.NewWriter(os.Stdout)\n\t\ttable.SetBorder(false)\n\t\ttable.SetColumnSeparator(\"\")\n\t\tfor _, e := range history.Events {\n\t\t\tif resetPointsOnly {\n\t\t\t\tif prevEvent.GetEventType() != types.EventTypeDecisionTaskStarted {\n\t\t\t\t\tprevEvent = *e\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tprevEvent = *e\n\t\t\t}\n\n\t\t\tcolumns := []string{}\n\t\t\tcolumns = append(columns, strconv.FormatInt(e.ID, 10))\n\n\t\t\tif printRawTime {\n\t\t\t\tcolumns = append(columns, strconv.FormatInt(e.GetTimestamp(), 10))\n\t\t\t} else if printDateTime {\n\t\t\t\tcolumns = append(columns, timestampToString(e.GetTimestamp(), false))\n\t\t\t}\n\t\t\tif printVersion {\n\t\t\t\tcolumns = append(columns, fmt.Sprintf(\"(Version: %v)\", e.Version))\n\t\t\t}\n\n\t\t\tcolumns = append(columns, ColorEvent(e), HistoryEventToString(e, false, maxFieldLength))\n\t\t\ttable.Append(columns)\n\t\t}\n\t\ttable.Render()\n\t}\n\n\tif outputFileName != \"\" {\n\t\tserializer := &JSONHistorySerializer{}\n\t\tdata, err := serializer.Serialize(history)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to serialize history data.\", err)\n\t\t}\n\t\tif err := os.WriteFile(outputFileName, data, 0666); err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to export history data file.\", err)\n\t\t}\n\t}\n\n\t// finally append activities with retry\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresp, err := frontendClient.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t})\n\tif err != nil {\n\t\tif _, ok := err.(*types.EntityNotExistsError); ok {\n\t\t\treturn commoncli.Problem(\"workflow not exist\", err)\n\t\t}\n\t\treturn commoncli.Problem(\"Describe workflow execution failed, cannot get information of pending activities\", err)\n\t}\n\tfmt.Println(\"History Source: Default Storage\")\n\n\tdescOutput, err := convertDescribeWorkflowExecutionResponse(resp, frontendClient, c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error in convert describe wf: \", err)\n\t}\n\tif len(descOutput.PendingActivities) > 0 {\n\t\tgetDeps(c).Output().Write([]byte(\"============Workflow Pending activities============\\n\"))\n\t\tprettyPrintJSONObject(getDeps(c).Output(), descOutput.PendingActivities)\n\t\tgetDeps(c).Output().Write([]byte(\"NOTE: ActivityStartedEvent with retry policy will be written into history when the activity is finished.\\n\"))\n\t}\n\treturn nil\n}\n\n// StartWorkflow starts a new workflow execution\nfunc StartWorkflow(c *cli.Context) error {\n\treturn startWorkflowHelper(c, false)\n}\n\n// RunWorkflow starts a new workflow execution and print workflow progress and result\nfunc RunWorkflow(c *cli.Context) error {\n\treturn startWorkflowHelper(c, true)\n}\n\nfunc startWorkflowHelper(c *cli.Context, shouldPrintProgress bool) error {\n\tserviceClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tstartRequest, err := constructStartWorkflowRequest(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain := startRequest.GetDomain()\n\twid := startRequest.GetWorkflowID()\n\tworkflowType := startRequest.WorkflowType.GetName()\n\ttaskList := startRequest.TaskList.GetName()\n\tinput := string(startRequest.Input)\n\n\tstartFn := func() error {\n\t\ttcCtx, cancel, err := newContext(c)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t\t}\n\t\tresp, err := serviceClient.StartWorkflowExecution(tcCtx, startRequest)\n\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to create workflow.\", replaceRetryPropertiesInErrorMessageWithRetryArguments(err))\n\t\t}\n\t\tfmt.Printf(\"Started Workflow Id: %s, run Id: %s\\n\", wid, resp.GetRunID())\n\t\treturn nil\n\t}\n\n\trunFn := func() error {\n\t\ttcCtx, cancel, err := newContextForLongPoll(c)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t\t}\n\t\tresp, err := serviceClient.StartWorkflowExecution(tcCtx, startRequest)\n\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Failed to run workflow.\", replaceRetryPropertiesInErrorMessageWithRetryArguments(err))\n\t\t}\n\n\t\t// print execution summary\n\t\tfmt.Println(colorMagenta(\"Running execution:\"))\n\t\ttable := tablewriter.NewWriter(os.Stdout)\n\t\texecutionData := [][]string{\n\t\t\t{\"Workflow Id\", wid},\n\t\t\t{\"Run Id\", resp.GetRunID()},\n\t\t\t{\"Type\", workflowType},\n\t\t\t{\"Domain\", domain},\n\t\t\t{\"Task List\", taskList},\n\t\t\t{\"Args\", truncate(input)}, // in case of large input\n\t\t}\n\t\ttable.SetBorder(false)\n\t\ttable.SetColumnSeparator(\":\")\n\t\ttable.AppendBulk(executionData) // Add Bulk Data\n\t\ttable.Render()\n\n\t\tprintWorkflowProgress(c, domain, wid, resp.GetRunID())\n\t\treturn nil\n\t}\n\n\tif shouldPrintProgress {\n\t\treturn runFn()\n\t}\n\treturn startFn()\n}\n\nfunc replaceRetryPropertiesInErrorMessageWithRetryArguments(err error) error {\n\terrMsg := err.Error()\n\t// This mapping is built based on the implementation of the function constructStartWorkflowRequest\n\tfor requestField, cliFlag := range map[string]string{\n\t\t\"InitialIntervalInSeconds\":    FlagRetryInterval,\n\t\t\"BackoffCoefficient\":          FlagRetryBackoff,\n\t\t\"MaximumIntervalInSeconds\":    FlagRetryMaxInterval,\n\t\t\"MaximumAttempts\":             FlagRetryAttempts,\n\t\t\"ExpirationIntervalInSeconds\": FlagRetryExpiration,\n\t} {\n\t\terrMsg = strings.ReplaceAll(errMsg, requestField, cliFlag)\n\t}\n\treturn errors.New(errMsg)\n}\n\nfunc constructStartWorkflowRequest(c *cli.Context) (*types.StartWorkflowExecutionRequest, error) {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\ttaskList, err := getRequiredOption(c, FlagTaskList)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tworkflowType, err := getRequiredOption(c, FlagWorkflowType)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tet := c.Int(FlagExecutionTimeout)\n\tif et == 0 {\n\t\treturn nil, commoncli.Problem(fmt.Sprintf(\"Option %s format is invalid.\", FlagExecutionTimeout), nil)\n\t}\n\tdt := c.Int(FlagDecisionTimeout)\n\twid := c.String(FlagWorkflowID)\n\tif len(wid) == 0 {\n\t\twid = uuid.New()\n\t}\n\treusePolicy := defaultWorkflowIDReusePolicy.Ptr()\n\twfidreusepolicy, err := getWorkflowIDReusePolicy(c.Int(FlagWorkflowIDReusePolicy))\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error getting wf reuse policy: \", err)\n\t}\n\tif c.IsSet(FlagWorkflowIDReusePolicy) {\n\t\treusePolicy = wfidreusepolicy\n\t}\n\n\tinput, err := processJSONInput(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error in starting wf request: \", err)\n\t}\n\tactiveClusterSelectionPolicy, err := parseClusterAttributes(c.String(FlagClusterAttributeScope), c.String(FlagClusterAttributeName))\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error parsing cluster attributes: \", err)\n\t}\n\tstartRequest := &types.StartWorkflowExecutionRequest{\n\t\tRequestID:  uuid.New(),\n\t\tDomain:     domain,\n\t\tWorkflowID: wid,\n\t\tWorkflowType: &types.WorkflowType{\n\t\t\tName: workflowType,\n\t\t},\n\t\tTaskList: &types.TaskList{\n\t\t\tName: taskList,\n\t\t},\n\t\tInput:                               []byte(input),\n\t\tExecutionStartToCloseTimeoutSeconds: common.Int32Ptr(int32(et)),\n\t\tTaskStartToCloseTimeoutSeconds:      common.Int32Ptr(int32(dt)),\n\t\tIdentity:                            getCliIdentity(),\n\t\tWorkflowIDReusePolicy:               reusePolicy,\n\t\tActiveClusterSelectionPolicy:        activeClusterSelectionPolicy,\n\t}\n\tif c.IsSet(FlagCronSchedule) {\n\t\tstartRequest.CronSchedule = c.String(FlagCronSchedule)\n\t}\n\n\tif c.IsSet(FlagRetryAttempts) || c.IsSet(FlagRetryExpiration) {\n\t\tstartRequest.RetryPolicy = &types.RetryPolicy{\n\t\t\tInitialIntervalInSeconds: int32(c.Int(FlagRetryInterval)),\n\t\t\tBackoffCoefficient:       c.Float64(FlagRetryBackoff),\n\t\t}\n\n\t\tif c.IsSet(FlagRetryAttempts) {\n\t\t\tstartRequest.RetryPolicy.MaximumAttempts = int32(c.Int(FlagRetryAttempts))\n\t\t}\n\t\tif c.IsSet(FlagRetryExpiration) {\n\t\t\tstartRequest.RetryPolicy.ExpirationIntervalInSeconds = int32(c.Int(FlagRetryExpiration))\n\t\t}\n\t\tif c.IsSet(FlagRetryMaxInterval) {\n\t\t\tstartRequest.RetryPolicy.MaximumIntervalInSeconds = int32(c.Int(FlagRetryMaxInterval))\n\t\t}\n\t}\n\n\tif c.IsSet(DelayStartSeconds) {\n\t\tstartRequest.DelayStartSeconds = common.Int32Ptr(int32(c.Int(DelayStartSeconds)))\n\t}\n\n\tif c.IsSet(JitterStartSeconds) {\n\t\tstartRequest.JitterStartSeconds = common.Int32Ptr(int32(c.Int(JitterStartSeconds)))\n\t}\n\n\tif c.IsSet(FirstRunAtTime) {\n\t\tt, err := time.Parse(time.RFC3339, c.String(FirstRunAtTime))\n\t\tif err != nil {\n\t\t\treturn nil, commoncli.Problem(\"First_run_at time format invalid, please use RFC3339\", err)\n\t\t}\n\t\tstartRequest.FirstRunAtTimeStamp = common.Int64Ptr(t.UnixNano())\n\t}\n\n\tif c.IsSet(FlagCronOverlapPolicy) {\n\t\tstartRequest.CronOverlapPolicy = types.CronOverlapPolicy(c.Int(FlagCronOverlapPolicy)).Ptr()\n\t}\n\n\theaderFields, err := processHeader(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error when process header: %w\", err)\n\t}\n\tif len(headerFields) != 0 {\n\t\tstartRequest.Header = &types.Header{Fields: headerFields}\n\t}\n\n\tmemoFields, err := processMemo(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"error processing memo: \", err)\n\t}\n\tif len(memoFields) != 0 {\n\t\tstartRequest.Memo = &types.Memo{Fields: memoFields}\n\t}\n\n\tsearchAttrFields, err := processSearchAttr(c)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"error processing search attributes: \", err)\n\t}\n\tif len(searchAttrFields) != 0 {\n\t\tstartRequest.SearchAttributes = &types.SearchAttributes{IndexedFields: searchAttrFields}\n\t}\n\n\treturn startRequest, nil\n}\n\nfunc processSearchAttr(c *cli.Context) (map[string][]byte, error) {\n\trawSearchAttrKey := c.String(FlagSearchAttributesKey)\n\tvar searchAttrKeys []string\n\tif strings.TrimSpace(rawSearchAttrKey) != \"\" {\n\t\tsearchAttrKeys = trimSpace(strings.Split(rawSearchAttrKey, searchAttrInputSeparator))\n\t}\n\n\trawSearchAttrVal := c.String(FlagSearchAttributesVal)\n\tvar searchAttrVals []interface{}\n\tif strings.TrimSpace(rawSearchAttrVal) != \"\" {\n\t\tsearchAttrValsStr := trimSpace(strings.Split(rawSearchAttrVal, searchAttrInputSeparator))\n\n\t\tfor _, v := range searchAttrValsStr {\n\t\t\tsearchAttrVals = append(searchAttrVals, convertStringToRealType(v))\n\t\t}\n\t}\n\n\tif len(searchAttrKeys) != len(searchAttrVals) {\n\t\treturn nil, fmt.Errorf(\"Number of search attributes keys and values are not equal. %v\", nil)\n\t}\n\n\tfields := map[string][]byte{}\n\tfor i, key := range searchAttrKeys {\n\t\tval, err := json.Marshal(searchAttrVals[i])\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"Encode value %v error. Error %v\", val, err)\n\t\t}\n\t\tfields[key] = val\n\t}\n\n\treturn fields, nil\n}\n\nfunc processHeader(c *cli.Context) (map[string][]byte, error) {\n\t// CLI flag input headers\n\theaderKeys := processMultipleKeys(c.String(FlagHeaderKey), \" \")\n\tjsoninputhandler, err := processJSONInputHelper(c, jsonTypeHeader)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in process header: %w\", err)\n\t}\n\theaderValues, err := processMultipleJSONValues(jsoninputhandler)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in process header: %w\", err)\n\t}\n\tif len(headerKeys) != len(headerValues) {\n\t\treturn nil, commoncli.Problem(\"Number of header keys and values are not equal.\", nil)\n\t}\n\n\theaders := mapFromKeysValues(headerKeys, headerValues)\n\t// append context headers if exist\n\tif span := opentracing.SpanFromContext(c.Context); span != nil && span.Context() != nil {\n\t\tspanHeaders := make(map[string]string)\n\t\tif err := span.Tracer().Inject(span.Context(), opentracing.TextMap, opentracing.TextMapCarrier(spanHeaders)); err != nil {\n\t\t\treturn nil, fmt.Errorf(\"error in inject span context: %w\", err)\n\t\t}\n\t\tfor k, v := range spanHeaders {\n\t\t\tif _, exist := headers[k]; !exist { // append only if not exist\n\t\t\t\theaders[k] = []byte(v)\n\t\t\t}\n\t\t}\n\t}\n\treturn headers, nil\n}\n\nfunc processMemo(c *cli.Context) (map[string][]byte, error) {\n\tmemoKeys := processMultipleKeys(c.String(FlagMemoKey), \" \")\n\tjsoninputhandler, err := processJSONInputHelper(c, jsonTypeMemo)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error in process header: %w\", err)\n\t}\n\tmemoValues, err := processMultipleJSONValues(jsoninputhandler)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error in process header: %w\", err)\n\t}\n\tif len(memoKeys) != len(memoValues) {\n\t\treturn nil, fmt.Errorf(\"number of memo keys and values are not equal\")\n\t}\n\n\treturn mapFromKeysValues(memoKeys, memoValues), nil\n}\n\n// helper function to print workflow progress with time refresh every second\nfunc printWorkflowProgress(c *cli.Context, domain, wid, rid string) error {\n\toutput := getDeps(c).Output()\n\n\tfmt.Println(colorMagenta(\"Progress:\"))\n\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\ttimeElapse := 1\n\tisTimeElapseExist := false\n\tdoneChan := make(chan bool)       // Channel to signal completion\n\terrChan := make(chan error)       // Separate channel for errors\n\tvar lastEvent *types.HistoryEvent // Used to print result of this run\n\tticker := time.NewTicker(time.Second).C\n\n\ttcCtx, cancel, err := newIndefiniteContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"error creating context: %w\", err)\n\t}\n\n\tshowDetails := c.Bool(FlagShowDetail)\n\tvar maxFieldLength int\n\tif c.IsSet(FlagMaxFieldLength) {\n\t\tmaxFieldLength = c.Int(FlagMaxFieldLength)\n\t}\n\n\tgo func() {\n\t\titerator, err := GetWorkflowHistoryIterator(tcCtx, wfClient, domain, wid, rid, true, types.HistoryEventFilterTypeAllEvent.Ptr(), nil)\n\t\tif err != nil {\n\t\t\terrChan <- fmt.Errorf(\"unable to get history events: %w\", err)\n\t\t\treturn\n\t\t}\n\t\tfor iterator.HasNext() {\n\t\t\tentity, err := iterator.Next()\n\t\t\tif err != nil {\n\t\t\t\terrChan <- fmt.Errorf(\"unable to read event: %w\", err)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tevent := entity.(*types.HistoryEvent)\n\n\t\t\tif isTimeElapseExist {\n\t\t\t\tremovePrevious2LinesFromTerminal(output)\n\t\t\t\tisTimeElapseExist = false\n\t\t\t}\n\t\t\tif showDetails {\n\t\t\t\tfmt.Printf(\"  %d, %s, %s, %s\\n\", event.ID, timestampToString(event.GetTimestamp(), false), ColorEvent(event), HistoryEventToString(event, true, maxFieldLength))\n\t\t\t} else {\n\t\t\t\tfmt.Printf(\"  %d, %s, %s\\n\", event.ID, timestampToString(event.GetTimestamp(), false), ColorEvent(event))\n\t\t\t}\n\t\t\tlastEvent = event\n\t\t}\n\t\tdoneChan <- true // Signal completion\n\t}()\n\n\tfor {\n\t\tselect {\n\t\tcase <-ticker:\n\t\t\tif isTimeElapseExist {\n\t\t\t\tremovePrevious2LinesFromTerminal(output)\n\t\t\t}\n\t\t\tfmt.Printf(\"\\nTime elapse: %ds\\n\", timeElapse)\n\t\t\tisTimeElapseExist = true\n\t\t\ttimeElapse++\n\t\tcase err := <-errChan: // Check for errors\n\t\t\treturn err\n\t\tcase <-doneChan: // Print result of this run\n\t\t\tfmt.Println(colorMagenta(\"\\nResult:\"))\n\t\t\tfmt.Printf(\"  Run Time: %d seconds\\n\", timeElapse)\n\t\t\tprintRunStatus(lastEvent)\n\t\t\treturn nil\n\t\t}\n\t}\n}\n\n// TerminateWorkflow terminates a workflow execution\nfunc TerminateWorkflow(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\treason := c.String(FlagReason)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\terr = wfClient.TerminateWorkflowExecution(\n\t\tctx,\n\t\t&types.TerminateWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tReason: reason,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t}, Identity: getCliIdentity(),\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Terminate workflow failed.\", err)\n\t}\n\tfmt.Println(\"Terminate workflow succeeded.\")\n\n\treturn nil\n}\n\n// CancelWorkflow cancels a workflow execution\nfunc CancelWorkflow(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\treason := c.String(FlagReason)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\terr = wfClient.RequestCancelWorkflowExecution(\n\t\tctx,\n\t\t&types.RequestCancelWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t},\n\t\t\tIdentity:  getCliIdentity(),\n\t\t\tCause:     reason,\n\t\t\tRequestID: uuid.New(),\n\t\t},\n\t)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Cancel workflow failed.\", err)\n\t}\n\tfmt.Println(\"Cancel workflow succeeded.\")\n\treturn nil\n}\n\n// SignalWorkflow signals a workflow execution\nfunc SignalWorkflow(c *cli.Context) error {\n\tserviceClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\tname, err := getRequiredOption(c, FlagName)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tinput, err := processJSONInput(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error proccessing JSON input: \", err)\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\terr = serviceClient.SignalWorkflowExecution(\n\t\ttcCtx,\n\t\t&types.SignalWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t},\n\t\t\tSignalName: name,\n\t\t\tInput:      []byte(input),\n\t\t\tIdentity:   getCliIdentity(),\n\t\t\tRequestID:  uuid.New(),\n\t\t},\n\t)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Signal workflow failed.\", err)\n\t}\n\tfmt.Println(\"Signal workflow succeeded.\")\n\treturn nil\n}\n\n// SignalWithStartWorkflowExecution starts a workflow execution if not already exists and signals it\nfunc SignalWithStartWorkflowExecution(c *cli.Context) error {\n\tserviceClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tsignalWithStartRequest, err := constructSignalWithStartWorkflowRequest(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\tresp, err := serviceClient.SignalWithStartWorkflowExecution(tcCtx, signalWithStartRequest)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"SignalWithStart workflow failed.\", err)\n\t}\n\tfmt.Printf(\"SignalWithStart workflow succeeded. Workflow Id: %s, run Id: %s\\n\", signalWithStartRequest.GetWorkflowID(), resp.GetRunID())\n\treturn nil\n}\n\nfunc constructSignalWithStartWorkflowRequest(c *cli.Context) (*types.SignalWithStartWorkflowExecutionRequest, error) {\n\tstartRequest, err := constructStartWorkflowRequest(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsignalname, err := getRequiredOption(c, FlagName)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"required flag not present %w\", err)\n\t}\n\tjsoninputsignal, err := processJSONInputSignal(c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error processing json input signal: %w\", err)\n\t}\n\treturn &types.SignalWithStartWorkflowExecutionRequest{\n\t\tDomain:                              startRequest.Domain,\n\t\tWorkflowID:                          startRequest.WorkflowID,\n\t\tWorkflowType:                        startRequest.WorkflowType,\n\t\tTaskList:                            startRequest.TaskList,\n\t\tInput:                               startRequest.Input,\n\t\tExecutionStartToCloseTimeoutSeconds: startRequest.ExecutionStartToCloseTimeoutSeconds,\n\t\tTaskStartToCloseTimeoutSeconds:      startRequest.TaskStartToCloseTimeoutSeconds,\n\t\tIdentity:                            startRequest.Identity,\n\t\tRequestID:                           startRequest.RequestID,\n\t\tWorkflowIDReusePolicy:               startRequest.WorkflowIDReusePolicy,\n\t\tRetryPolicy:                         startRequest.RetryPolicy,\n\t\tCronSchedule:                        startRequest.CronSchedule,\n\t\tCronOverlapPolicy:                   startRequest.CronOverlapPolicy,\n\t\tMemo:                                startRequest.Memo,\n\t\tSearchAttributes:                    startRequest.SearchAttributes,\n\t\tHeader:                              startRequest.Header,\n\t\tSignalName:                          signalname,\n\t\tSignalInput:                         []byte(jsoninputsignal),\n\t\tDelayStartSeconds:                   startRequest.DelayStartSeconds,\n\t\tJitterStartSeconds:                  startRequest.JitterStartSeconds,\n\t\tFirstRunAtTimestamp:                 startRequest.FirstRunAtTimeStamp,\n\t\tActiveClusterSelectionPolicy:        startRequest.ActiveClusterSelectionPolicy,\n\t}, nil\n}\n\nfunc processJSONInputSignal(c *cli.Context) (string, error) {\n\treturn processJSONInputHelper(c, jsonTypeSignal)\n}\n\n// QueryWorkflow query workflow execution\nfunc QueryWorkflow(c *cli.Context) error {\n\tqueryType, err := getRequiredOption(c, FlagQueryType)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found for query type: \", err)\n\t}\n\treturn queryWorkflowHelper(c, queryType)\n}\n\n// QueryWorkflowUsingStackTrace query workflow execution using __stack_trace as query type\nfunc QueryWorkflowUsingStackTrace(c *cli.Context) error {\n\treturn queryWorkflowHelper(c, \"__stack_trace\")\n}\n\n// QueryWorkflowUsingQueryTypes list all query types of the workflow using __query_types as query type\nfunc QueryWorkflowUsingQueryTypes(c *cli.Context) error {\n\treturn queryWorkflowHelper(c, \"__query_types\")\n}\n\nfunc queryWorkflowHelper(c *cli.Context, queryType string) error {\n\tserviceClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\tinput, err := processJSONInput(c)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error processing json \", err)\n\t}\n\ttcCtx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tqueryRequest := &types.QueryWorkflowRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t\tQuery: &types.WorkflowQuery{\n\t\t\tQueryType: queryType,\n\t\t},\n\t}\n\tif input != \"\" {\n\t\tqueryRequest.Query.QueryArgs = []byte(input)\n\t}\n\n\trejectCondition, err := getOptionWithSerializer(c, FlagQueryRejectCondition, mapQueryRejectConditionFromFlag)\n\tif err != nil {\n\t\treturn commoncli.Problem(err.Error(), nil)\n\t}\n\tqueryRequest.QueryRejectCondition = rejectCondition\n\n\tconsistencyLevel, err := getOptionWithSerializer(c, FlagQueryConsistencyLevel, mapQueryConsistencyLevelFromFlag)\n\tif err != nil {\n\t\treturn commoncli.Problem(err.Error(), nil)\n\t}\n\tqueryRequest.QueryConsistencyLevel = consistencyLevel\n\n\tqueryResponse, err := serviceClient.QueryWorkflow(tcCtx, queryRequest)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Query workflow failed.\", err)\n\t}\n\n\tif queryResponse.QueryRejected != nil {\n\t\tfmt.Printf(\"Query was rejected, workflow is in state: %v\\n\", *queryResponse.QueryRejected.CloseStatus)\n\t} else {\n\t\t// assume it is json encoded\n\t\tfmt.Print(string(queryResponse.QueryResult))\n\t}\n\treturn nil\n}\n\n// ListWorkflow list workflow executions based on filters\nfunc ListWorkflow(c *cli.Context) error {\n\tlistWF, err := listWorkflows(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tlistExcludedWF, err := filterExcludedWorkflows(c, listWF)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn displayPagedWorkflows(c, listExcludedWF, !c.Bool(FlagMore))\n}\n\n// ListAllWorkflow list all workflow executions based on filters\nfunc ListAllWorkflow(c *cli.Context) error {\n\tlistWF, err := listWorkflows(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tlistExcludedWF, err := filterExcludedWorkflows(c, listWF)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn displayAllWorkflows(c, listExcludedWF)\n}\n\n// ScanAllWorkflow list all workflow executions using Scan API.\n// It should be faster than ListAllWorkflow, but result are not sorted.\nfunc ScanAllWorkflow(c *cli.Context) error {\n\tpagefn, err := scanWorkflows(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn displayAllWorkflows(c, pagefn)\n}\n\nfunc isQueryOpen(query string) bool {\n\tvar openWFPattern = regexp.MustCompile(`CloseTime[ ]*=[ ]*missing`)\n\treturn openWFPattern.MatchString(query)\n}\n\n// CountWorkflow count number of workflows\nfunc CountWorkflow(c *cli.Context) error {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tquery := c.String(FlagListQuery)\n\trequest := &types.CountWorkflowExecutionsRequest{\n\t\tDomain: domain,\n\t\tQuery:  query,\n\t}\n\n\tctx, cancel, err := newContextForLongPoll(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tresponse, err := wfClient.CountWorkflowExecutions(ctx, request)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failed to count workflow.\", err)\n\t}\n\n\tfmt.Println(response.GetCount())\n\treturn nil\n}\n\n// ListArchivedWorkflow lists archived workflow executions based on filters\nfunc ListArchivedWorkflow(c *cli.Context) error {\n\tprintAll := c.Bool(FlagAll)\n\tpagefn, err := listArchivedWorkflows(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif printAll {\n\t\treturn displayAllWorkflows(c, pagefn)\n\t}\n\treturn displayPagedWorkflows(c, pagefn, false)\n}\n\n// DescribeWorkflow show information about the specified workflow execution\nfunc DescribeWorkflow(c *cli.Context) error {\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\n\treturn describeWorkflowHelper(c, wid, rid)\n}\n\n// DescribeWorkflowWithID show information about the specified workflow execution\nfunc DescribeWorkflowWithID(c *cli.Context) error {\n\tif !c.Args().Present() {\n\t\treturn commoncli.Problem(\"Argument workflow_id is required.\", nil)\n\t}\n\twid := c.Args().First()\n\trid := \"\"\n\tif c.NArg() >= 2 {\n\t\trid = c.Args().Get(1)\n\t}\n\n\treturn describeWorkflowHelper(c, wid, rid)\n}\n\nfunc describeWorkflowHelper(c *cli.Context, wid, rid string) error {\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tprintRaw := c.Bool(FlagPrintRaw) // printRaw is false by default,\n\t// and will show datetime and decoded search attributes instead of raw timestamp and byte arrays\n\tprintResetPointsOnly := c.Bool(FlagResetPointsOnly)\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\trequest := &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t}\n\n\tconsistencyLevel, err := getOptionWithSerializer[types.QueryConsistencyLevel](c, FlagQueryConsistencyLevel, mapQueryConsistencyLevelFromFlag)\n\tif err != nil {\n\t\treturn commoncli.Problem(err.Error(), nil)\n\t}\n\trequest.QueryConsistencyLevel = consistencyLevel\n\n\tresp, err := frontendClient.DescribeWorkflowExecution(ctx, request)\n\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Describe workflow execution failed\", err)\n\t}\n\n\tif printResetPointsOnly {\n\t\treturn printAutoResetPoints(resp)\n\t}\n\n\tvar o interface{}\n\tif printRaw {\n\t\to = resp\n\t} else {\n\t\to, err = convertDescribeWorkflowExecutionResponse(resp, frontendClient, c)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"convert describe workflow response failed: \", err)\n\t\t}\n\t}\n\n\tprettyPrintJSONObject(getDeps(c).Output(), o)\n\treturn nil\n}\n\ntype AutoResetPointRow struct {\n\tBinaryChecksum string    `header:\"Binary Checksum\"`\n\tCreateTime     time.Time `header:\"Create Time\"`\n\tRunID          string    `header:\"RunID\"`\n\tEventID        int64     `header:\"EventID\"`\n}\n\nfunc printAutoResetPoints(resp *types.DescribeWorkflowExecutionResponse) error {\n\tfmt.Println(\"Auto Reset Points:\")\n\ttable := []AutoResetPointRow{}\n\tif resp.WorkflowExecutionInfo.AutoResetPoints == nil || len(resp.WorkflowExecutionInfo.AutoResetPoints.Points) == 0 {\n\t\treturn nil\n\t}\n\tfor _, pt := range resp.WorkflowExecutionInfo.AutoResetPoints.Points {\n\t\ttable = append(table, AutoResetPointRow{\n\t\t\tBinaryChecksum: pt.GetBinaryChecksum(),\n\t\t\tCreateTime:     time.Unix(0, pt.GetCreatedTimeNano()),\n\t\t\tRunID:          pt.GetRunID(),\n\t\t\tEventID:        pt.GetFirstDecisionCompletedID(),\n\t\t})\n\t}\n\treturn RenderTable(os.Stdout, table, RenderOptions{Color: true, Border: true, PrintDateTime: true})\n}\n\n// describeWorkflowExecutionResponse is used to print datetime instead of print raw time\ntype describeWorkflowExecutionResponse struct {\n\tExecutionConfiguration *types.WorkflowExecutionConfiguration\n\tWorkflowExecutionInfo  workflowExecutionInfo\n\tPendingActivities      []*pendingActivityInfo\n\tPendingChildren        []*types.PendingChildExecutionInfo\n\tPendingDecision        *pendingDecisionInfo\n}\n\n// workflowExecutionInfo has same fields as types.WorkflowExecutionInfo, but has datetime instead of raw time\ntype workflowExecutionInfo struct {\n\tExecution                    *types.WorkflowExecution\n\tType                         *types.WorkflowType\n\tStartTime                    *string // change from *int64\n\tCloseTime                    *string // change from *int64\n\tCloseStatus                  *types.WorkflowExecutionCloseStatus\n\tHistoryLength                int64\n\tParentDomainID               *string\n\tParentExecution              *types.WorkflowExecution\n\tMemo                         *types.Memo\n\tSearchAttributes             map[string]interface{}\n\tAutoResetPoints              *types.ResetPoints\n\tPartitionConfig              map[string]string\n\tCronOverlapPolicy            *types.CronOverlapPolicy\n\tActiveClusterSelectionPolicy *types.ActiveClusterSelectionPolicy\n}\n\n// pendingActivityInfo has same fields as types.PendingActivityInfo, but different field type for better display\ntype pendingActivityInfo struct {\n\tActivityID             string\n\tActivityType           *types.ActivityType\n\tState                  *types.PendingActivityState\n\tScheduledTimestamp     *string `json:\",omitempty\"` // change from *int64\n\tLastStartedTimestamp   *string `json:\",omitempty\"` // change from *int64\n\tHeartbeatDetails       *string `json:\",omitempty\"` // change from []byte\n\tLastHeartbeatTimestamp *string `json:\",omitempty\"` // change from *int64\n\tAttempt                int32   `json:\",omitempty\"`\n\tMaximumAttempts        int32   `json:\",omitempty\"`\n\tExpirationTimestamp    *string `json:\",omitempty\"` // change from *int64\n\tLastFailureReason      *string `json:\",omitempty\"`\n\tLastWorkerIdentity     string  `json:\",omitempty\"`\n\tLastFailureDetails     *string `json:\",omitempty\"` // change from []byte\n\tScheduleID             int64   `json:\",omitempty\"`\n}\n\ntype pendingDecisionInfo struct {\n\tState                      *types.PendingDecisionState\n\tOriginalScheduledTimestamp *string `json:\",omitempty\"` // change from *int64\n\tScheduledTimestamp         *string `json:\",omitempty\"` // change from *int64\n\tStartedTimestamp           *string `json:\",omitempty\"` // change from *int64\n\tAttempt                    int64   `json:\",omitempty\"`\n\tScheduleID                 int64   `json:\",omitempty\"`\n}\n\nfunc convertDescribeWorkflowExecutionResponse(resp *types.DescribeWorkflowExecutionResponse,\n\twfClient frontend.Client, c *cli.Context) (*describeWorkflowExecutionResponse, error) {\n\n\tinfo := resp.WorkflowExecutionInfo\n\tsearchattributes, err := convertSearchAttributesToMapOfInterface(info.SearchAttributes, wfClient, c)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error converting search attributes: %w\", err)\n\t}\n\texecutionInfo := workflowExecutionInfo{\n\t\tExecution:                    info.Execution,\n\t\tType:                         info.Type,\n\t\tStartTime:                    common.StringPtr(timestampToString(info.GetStartTime(), false)),\n\t\tCloseTime:                    common.StringPtr(timestampToString(info.GetCloseTime(), false)),\n\t\tCloseStatus:                  info.CloseStatus,\n\t\tHistoryLength:                info.HistoryLength,\n\t\tParentDomainID:               info.ParentDomainID,\n\t\tParentExecution:              info.ParentExecution,\n\t\tMemo:                         info.Memo,\n\t\tSearchAttributes:             searchattributes,\n\t\tAutoResetPoints:              info.AutoResetPoints,\n\t\tPartitionConfig:              info.PartitionConfig,\n\t\tCronOverlapPolicy:            info.CronOverlapPolicy,\n\t\tActiveClusterSelectionPolicy: info.ActiveClusterSelectionPolicy,\n\t}\n\n\tvar pendingActs []*pendingActivityInfo\n\tvar tmpAct *pendingActivityInfo\n\tfor _, pa := range resp.PendingActivities {\n\t\ttmpAct = &pendingActivityInfo{\n\t\t\tActivityID:             pa.ActivityID,\n\t\t\tActivityType:           pa.ActivityType,\n\t\t\tState:                  pa.State,\n\t\t\tScheduledTimestamp:     timestampPtrToStringPtr(pa.ScheduledTimestamp, false),\n\t\t\tLastStartedTimestamp:   timestampPtrToStringPtr(pa.LastStartedTimestamp, false),\n\t\t\tLastHeartbeatTimestamp: timestampPtrToStringPtr(pa.LastHeartbeatTimestamp, false),\n\t\t\tAttempt:                pa.Attempt,\n\t\t\tMaximumAttempts:        pa.MaximumAttempts,\n\t\t\tExpirationTimestamp:    timestampPtrToStringPtr(pa.ExpirationTimestamp, false),\n\t\t\tLastFailureReason:      pa.LastFailureReason,\n\t\t\tLastWorkerIdentity:     pa.LastWorkerIdentity,\n\t\t\tScheduleID:             pa.ScheduleID,\n\t\t}\n\t\tif pa.HeartbeatDetails != nil {\n\t\t\ttmpAct.HeartbeatDetails = common.StringPtr(string(pa.HeartbeatDetails))\n\t\t}\n\t\tif pa.LastFailureDetails != nil {\n\t\t\ttmpAct.LastFailureDetails = common.StringPtr(string(pa.LastFailureDetails))\n\t\t}\n\t\tpendingActs = append(pendingActs, tmpAct)\n\t}\n\n\tvar pendingDecision *pendingDecisionInfo\n\tif resp.PendingDecision != nil {\n\t\tpendingDecision = &pendingDecisionInfo{\n\t\t\tState:              resp.PendingDecision.State,\n\t\t\tScheduledTimestamp: timestampPtrToStringPtr(resp.PendingDecision.ScheduledTimestamp, false),\n\t\t\tStartedTimestamp:   timestampPtrToStringPtr(resp.PendingDecision.StartedTimestamp, false),\n\t\t\tAttempt:            resp.PendingDecision.Attempt,\n\t\t\tScheduleID:         resp.PendingDecision.ScheduleID,\n\t\t}\n\t\t// TODO: Idea here is only display decision task original scheduled timestamp if user are\n\t\t// using decision heartbeat. And we should be able to tell whether a decision task has heartbeat\n\t\t// or not by comparing the original scheduled timestamp and scheduled timestamp.\n\t\t// However, currently server may assign different value to original scheduled timestamp and\n\t\t// scheduled time even if there's no decision heartbeat.\n\t\t// if resp.PendingDecision.OriginalScheduledTimestamp != nil &&\n\t\t// \tresp.PendingDecision.ScheduledTimestamp != nil &&\n\t\t// \t*resp.PendingDecision.OriginalScheduledTimestamp != *resp.PendingDecision.ScheduledTimestamp {\n\t\t// \tpendingDecision.OriginalScheduledTimestamp = timestampPtrToStringPtr(resp.PendingDecision.OriginalScheduledTimestamp, false)\n\t\t// }\n\t}\n\n\treturn &describeWorkflowExecutionResponse{\n\t\tExecutionConfiguration: resp.ExecutionConfiguration,\n\t\tWorkflowExecutionInfo:  executionInfo,\n\t\tPendingActivities:      pendingActs,\n\t\tPendingChildren:        resp.PendingChildren,\n\t\tPendingDecision:        pendingDecision,\n\t}, nil\n}\n\nfunc convertSearchAttributesToMapOfInterface(searchAttributes *types.SearchAttributes,\n\twfClient frontend.Client, c *cli.Context) (map[string]interface{}, error) {\n\n\tif searchAttributes == nil || len(searchAttributes.GetIndexedFields()) == 0 {\n\t\treturn nil, nil\n\t}\n\n\tresult := make(map[string]interface{})\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tvalidSearchAttributes, err := wfClient.GetSearchAttributes(ctx)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error when get search attributes\", err)\n\t}\n\tvalidKeys := validSearchAttributes.GetKeys()\n\n\tindexedFields := searchAttributes.GetIndexedFields()\n\tfor k, v := range indexedFields {\n\t\tvalueType := validKeys[k]\n\t\tdeserializedValue, err := common.DeserializeSearchAttributeValue(v, valueType)\n\t\tif err != nil {\n\t\t\treturn nil, commoncli.Problem(\"Error deserializing search attribute value\", err)\n\t\t}\n\t\tresult[k] = deserializedValue\n\t}\n\n\treturn result, nil\n}\n\nfunc getAllWorkflowIDsByQuery(c *cli.Context, query string) (map[string]bool, error) {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tpageSize := 1000\n\tvar nextPageToken []byte\n\tresult := map[string]bool{}\n\tfor {\n\t\tinfo, nextPageToken, err := scanWorkflowExecutions(wfClient, pageSize, nextPageToken, query, c)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tfor _, we := range info {\n\t\t\twid := we.Execution.GetWorkflowID()\n\t\t\tresult[wid] = true\n\t\t}\n\n\t\tif nextPageToken == nil {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn result, nil\n}\n\nfunc printRunStatus(event *types.HistoryEvent) {\n\tswitch event.GetEventType() {\n\tcase types.EventTypeWorkflowExecutionCompleted:\n\t\tfmt.Printf(\"  Status: %s\\n\", colorGreen(\"COMPLETED\"))\n\t\tfmt.Printf(\"  Output: %s\\n\", string(event.WorkflowExecutionCompletedEventAttributes.Result))\n\tcase types.EventTypeWorkflowExecutionFailed:\n\t\tfmt.Printf(\"  Status: %s\\n\", colorRed(\"FAILED\"))\n\t\tfmt.Printf(\"  Reason: %s\\n\", event.WorkflowExecutionFailedEventAttributes.GetReason())\n\t\tfmt.Printf(\"  Detail: %s\\n\", string(event.WorkflowExecutionFailedEventAttributes.Details))\n\tcase types.EventTypeWorkflowExecutionTimedOut:\n\t\tfmt.Printf(\"  Status: %s\\n\", colorRed(\"TIMEOUT\"))\n\t\tfmt.Printf(\"  Timeout Type: %s\\n\", event.WorkflowExecutionTimedOutEventAttributes.GetTimeoutType())\n\tcase types.EventTypeWorkflowExecutionCanceled:\n\t\tfmt.Printf(\"  Status: %s\\n\", colorRed(\"CANCELED\"))\n\t\tfmt.Printf(\"  Detail: %s\\n\", string(event.WorkflowExecutionCanceledEventAttributes.Details))\n\t}\n}\n\n// WorkflowRow is a presentation layer entity use to render a table of workflows\ntype WorkflowRow struct {\n\tWorkflowType           string                 `header:\"Workflow Type\" maxLength:\"32\"`\n\tWorkflowID             string                 `header:\"Workflow ID\"`\n\tRunID                  string                 `header:\"Run ID\"`\n\tTaskList               string                 `header:\"Task List\"`\n\tIsCron                 bool                   `header:\"Is Cron\"`\n\tStartTime              time.Time              `header:\"Start Time\"`\n\tExecutionTime          time.Time              `header:\"Execution Time\"`\n\tEndTime                time.Time              `header:\"End Time\"`\n\tCloseStatus            string                 `header:\"Close Status\"`\n\tHistoryLength          int64                  `header:\"History Length\"`\n\tUpdateTime             time.Time              `header:\"Update Time\"`\n\tMemo                   map[string]string      `header:\"Memo\"`\n\tSearchAttributes       map[string]interface{} `header:\"Search Attributes\"`\n\tExecutionStatus        string                 `header:\"Execution Status\"`\n\tCronSchedule           string                 `header:\"Cron Schedule\"`\n\tScheduledExecutionTime time.Time              `header:\"Scheduled Execution Time\"`\n}\n\nfunc newWorkflowRow(workflow *types.WorkflowExecutionInfo) (WorkflowRow, error) {\n\tmemo := map[string]string{}\n\tfor k, v := range workflow.Memo.GetFields() {\n\t\tmemo[k] = string(v)\n\t}\n\n\tsa := map[string]interface{}{}\n\tfor k, v := range workflow.SearchAttributes.GetIndexedFields() {\n\t\tvar decodedVal interface{}\n\t\tif err := json.Unmarshal(v, &decodedVal); err != nil {\n\t\t\treturn WorkflowRow{}, fmt.Errorf(\"error decoding search attribute %s: %w\", k, err)\n\t\t}\n\t\tsa[k] = decodedVal\n\t}\n\n\t// Convert ExecutionStatus enum to string\n\texecutionStatus := \"\"\n\tif workflow.ExecutionStatus != nil {\n\t\tswitch workflow.GetExecutionStatus() {\n\t\tcase types.WorkflowExecutionStatusPending:\n\t\t\texecutionStatus = \"PENDING\"\n\t\tcase types.WorkflowExecutionStatusStarted:\n\t\t\texecutionStatus = \"STARTED\"\n\t\tcase types.WorkflowExecutionStatusCompleted:\n\t\t\texecutionStatus = \"COMPLETED\"\n\t\tcase types.WorkflowExecutionStatusFailed:\n\t\t\texecutionStatus = \"FAILED\"\n\t\tcase types.WorkflowExecutionStatusCanceled:\n\t\t\texecutionStatus = \"CANCELED\"\n\t\tcase types.WorkflowExecutionStatusTerminated:\n\t\t\texecutionStatus = \"TERMINATED\"\n\t\tcase types.WorkflowExecutionStatusContinuedAsNew:\n\t\t\texecutionStatus = \"CONTINUED_AS_NEW\"\n\t\tcase types.WorkflowExecutionStatusTimedOut:\n\t\t\texecutionStatus = \"TIMED_OUT\"\n\t\t}\n\t}\n\n\treturn WorkflowRow{\n\t\tWorkflowType:           workflow.Type.GetName(),\n\t\tWorkflowID:             workflow.Execution.GetWorkflowID(),\n\t\tRunID:                  workflow.Execution.GetRunID(),\n\t\tTaskList:               workflow.TaskList.GetName(),\n\t\tIsCron:                 workflow.IsCron,\n\t\tStartTime:              time.Unix(0, workflow.GetStartTime()),\n\t\tExecutionTime:          time.Unix(0, workflow.GetExecutionTime()),\n\t\tEndTime:                time.Unix(0, workflow.GetCloseTime()),\n\t\tUpdateTime:             time.Unix(0, workflow.GetUpdateTime()),\n\t\tCloseStatus:            workflow.GetCloseStatus().String(),\n\t\tHistoryLength:          workflow.HistoryLength,\n\t\tMemo:                   memo,\n\t\tSearchAttributes:       sa,\n\t\tExecutionStatus:        executionStatus,\n\t\tCronSchedule:           workflow.GetCronSchedule(),\n\t\tScheduledExecutionTime: time.Unix(0, workflow.GetScheduledExecutionTime()),\n\t}, nil\n}\n\nfunc workflowTableOptions(c *cli.Context) RenderOptions {\n\tisScanQueryOpen := isQueryOpen(c.String(FlagListQuery))\n\n\treturn RenderOptions{\n\t\tDefaultTemplate: templateTable,\n\t\tColor:           true,\n\t\tPrintDateTime:   c.Bool(FlagPrintDateTime),\n\t\tPrintRawTime:    c.Bool(FlagPrintRawTime),\n\t\tOptionalColumns: map[string]bool{\n\t\t\t\"End Time\":                 !(c.Bool(FlagOpen) || isScanQueryOpen),\n\t\t\t\"Memo\":                     c.Bool(FlagPrintMemo),\n\t\t\t\"Search Attributes\":        c.Bool(FlagPrintSearchAttr),\n\t\t\t\"Cron Schedule\":            c.Bool(FlagPrintCron),\n\t\t\t\"Scheduled Execution Time\": c.Bool(FlagPrintCron),\n\t\t},\n\t}\n}\n\ntype getWorkflowPageFn func([]byte) ([]*types.WorkflowExecutionInfo, []byte, error)\n\nfunc getAllWorkflows(getWorkflowPage getWorkflowPageFn) ([]*types.WorkflowExecutionInfo, error) {\n\tvar all, page []*types.WorkflowExecutionInfo\n\tvar nextPageToken []byte\n\tvar err error\n\tfor {\n\t\tpage, nextPageToken, err = getWorkflowPage(nextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tall = append(all, page...)\n\t\tif len(nextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn all, nil\n}\n\nfunc filterExcludedWorkflows(c *cli.Context, getWorkflowPage getWorkflowPageFn) (getWorkflowPageFn, error) {\n\texcludeWIDs := map[string]bool{}\n\tvar err error\n\tif c.IsSet(FlagListQuery) && c.IsSet(FlagExcludeWorkflowIDByQuery) {\n\t\texcludeQuery := c.String(FlagExcludeWorkflowIDByQuery)\n\t\texcludeWIDs, err = getAllWorkflowIDsByQuery(c, excludeQuery)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tfmt.Printf(\"found %d workflowIDs to exclude\\n\", len(excludeWIDs))\n\t}\n\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\tpage, nextPageToken, err := getWorkflowPage(nextPageToken)\n\t\tif err != nil {\n\t\t\treturn nil, nil, err\n\t\t}\n\t\tfiltered := make([]*types.WorkflowExecutionInfo, 0, len(page))\n\t\tfor _, workflow := range page {\n\t\t\tif excludeWIDs[workflow.GetExecution().GetWorkflowID()] {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tfiltered = append(filtered, workflow)\n\t\t}\n\t\treturn filtered, nextPageToken, nil\n\t}, nil\n}\n\nfunc displayPagedWorkflows(c *cli.Context, getWorkflowPage getWorkflowPageFn, firstPageOnly bool) error {\n\toutput := getDeps(c).Output()\n\n\tvar page []*types.WorkflowExecutionInfo\n\tvar nextPageToken []byte\n\tvar err error\n\tfor {\n\t\tpage, nextPageToken, err = getWorkflowPage(nextPageToken)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tif err := displayWorkflows(c, page); err != nil {\n\t\t\treturn fmt.Errorf(\"error displaying workflows: %w\", err)\n\t\t}\n\n\t\tif firstPageOnly {\n\t\t\tbreak\n\t\t}\n\t\tif len(nextPageToken) == 0 {\n\t\t\tbreak\n\t\t}\n\t\tif !showNextPage(output) {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc displayAllWorkflows(c *cli.Context, getWorkflowsPage getWorkflowPageFn) error {\n\twfs, err := getAllWorkflows(getWorkflowsPage)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn displayWorkflows(c, wfs)\n}\n\nfunc displayWorkflows(c *cli.Context, workflows []*types.WorkflowExecutionInfo) error {\n\tprintJSON := c.Bool(FlagPrintJSON)\n\tprintDecodedRaw := c.Bool(FlagPrintFullyDetail)\n\n\tif printJSON || printDecodedRaw {\n\t\tfmt.Println(\"[\")\n\t\tprintListResults(workflows, printJSON, false)\n\t\tfmt.Println(\"]\")\n\t\treturn nil\n\t}\n\ttableOptions := workflowTableOptions(c)\n\tvar table []WorkflowRow\n\tfor _, workflow := range workflows {\n\t\trow, err := newWorkflowRow(workflow)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\ttable = append(table, row)\n\t}\n\treturn Render(c, table, tableOptions)\n}\n\nfunc listWorkflowExecutions(client frontend.Client, pageSize int, domain, query string, c *cli.Context) getWorkflowPageFn {\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\trequest := &types.ListWorkflowExecutionsRequest{\n\t\t\tDomain:        domain,\n\t\t\tPageSize:      int32(pageSize),\n\t\t\tNextPageToken: nextPageToken,\n\t\t\tQuery:         query,\n\t\t}\n\n\t\tctx, cancel, err := newContextForLongPoll(c)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Error creating context: \", err)\n\t\t}\n\t\tresponse, err := client.ListWorkflowExecutions(ctx, request)\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list workflow.\", err)\n\t\t}\n\t\treturn response.Executions, response.NextPageToken, nil\n\t}\n}\n\nfunc listOpenWorkflow(client frontend.Client, pageSize int, earliestTime, latestTime int64, domain, workflowID, workflowType string, c *cli.Context) getWorkflowPageFn {\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\trequest := &types.ListOpenWorkflowExecutionsRequest{\n\t\t\tDomain:          domain,\n\t\t\tMaximumPageSize: int32(pageSize),\n\t\t\tNextPageToken:   nextPageToken,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(earliestTime),\n\t\t\t\tLatestTime:   common.Int64Ptr(latestTime),\n\t\t\t},\n\t\t}\n\t\tif len(workflowID) > 0 {\n\t\t\trequest.ExecutionFilter = &types.WorkflowExecutionFilter{WorkflowID: workflowID}\n\t\t}\n\t\tif len(workflowType) > 0 {\n\t\t\trequest.TypeFilter = &types.WorkflowTypeFilter{Name: workflowType}\n\t\t}\n\n\t\tctx, cancel, err := newContextForLongPoll(c)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list open workflow.\", err)\n\t\t}\n\t\tresponse, err := client.ListOpenWorkflowExecutions(ctx, request)\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list open workflow.\", err)\n\t\t}\n\t\treturn response.Executions, response.NextPageToken, nil\n\t}\n}\n\nfunc listClosedWorkflow(client frontend.Client, pageSize int, earliestTime, latestTime int64, domain, workflowID, workflowType string, workflowStatus types.WorkflowExecutionCloseStatus, c *cli.Context) getWorkflowPageFn {\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\trequest := &types.ListClosedWorkflowExecutionsRequest{\n\t\t\tDomain:          domain,\n\t\t\tMaximumPageSize: int32(pageSize),\n\t\t\tNextPageToken:   nextPageToken,\n\t\t\tStartTimeFilter: &types.StartTimeFilter{\n\t\t\t\tEarliestTime: common.Int64Ptr(earliestTime),\n\t\t\t\tLatestTime:   common.Int64Ptr(latestTime),\n\t\t\t},\n\t\t}\n\t\tif len(workflowID) > 0 {\n\t\t\trequest.ExecutionFilter = &types.WorkflowExecutionFilter{WorkflowID: workflowID}\n\t\t}\n\t\tif len(workflowType) > 0 {\n\t\t\trequest.TypeFilter = &types.WorkflowTypeFilter{Name: workflowType}\n\t\t}\n\t\tif workflowStatus != workflowStatusNotSet {\n\t\t\trequest.StatusFilter = &workflowStatus\n\t\t}\n\n\t\tctx, cancel, err := newContextForLongPoll(c)\n\t\tdefer cancel()\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list closed workflow.\", err)\n\t\t}\n\t\tresponse, err := client.ListClosedWorkflowExecutions(ctx, request)\n\t\tif err != nil {\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list closed workflow.\", err)\n\t\t}\n\t\treturn response.Executions, response.NextPageToken, nil\n\t}\n}\n\nfunc listWorkflows(c *cli.Context) (getWorkflowPageFn, error) {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tearliestTime, err := parseTime(c.String(FlagEarliestTime), 0)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in listing workflows: %w\", err)\n\t}\n\tlatestTime, err := parseTime(c.String(FlagLatestTime), time.Now().UnixNano())\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"Error in listing workflows: %w\", err)\n\t}\n\tworkflowID := c.String(FlagWorkflowID)\n\tworkflowType := c.String(FlagWorkflowType)\n\tqueryOpen := c.Bool(FlagOpen)\n\tpageSize := c.Int(FlagPageSize)\n\tif pageSize <= 0 {\n\t\tpageSize = defaultPageSizeForList\n\t}\n\n\tvar workflowStatus types.WorkflowExecutionCloseStatus\n\tif c.IsSet(FlagWorkflowStatus) {\n\t\tif queryOpen {\n\t\t\treturn nil, commoncli.Problem(optionErr, errors.New(\"you can only filter on status for closed workflow, not open workflow\"))\n\t\t}\n\t\tworkflowStatus, err = getWorkflowStatus(c.String(FlagWorkflowStatus))\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"failed to parse workflow status: %w\", err)\n\t\t}\n\t} else {\n\t\tworkflowStatus = workflowStatusNotSet\n\t}\n\n\tif len(workflowID) > 0 && len(workflowType) > 0 {\n\t\treturn nil, commoncli.Problem(optionErr, errors.New(\"you can filter on workflow_id or workflow_type, but not on both\"))\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tresp, err := wfClient.CountWorkflowExecutions(\n\t\tctx,\n\t\t&types.CountWorkflowExecutionsRequest{\n\t\t\tDomain: domain,\n\t\t\tQuery:  c.String(FlagListQuery),\n\t\t},\n\t)\n\tif err == nil {\n\t\t_, err = fmt.Fprintf(os.Stderr, \"Fetching %v workflows...\\n\", resp.GetCount())\n\t\tif err != nil {\n\t\t\treturn nil, commoncli.Problem(\"Failed to print to stderr\", err)\n\t\t}\n\t}\n\n\tif c.IsSet(FlagListQuery) {\n\t\tlistQuery := c.String(FlagListQuery)\n\t\treturn listWorkflowExecutions(wfClient, pageSize, domain, listQuery, c), nil\n\t} else if queryOpen {\n\t\treturn listOpenWorkflow(wfClient, pageSize, earliestTime, latestTime, domain, workflowID, workflowType, c), nil\n\t} else {\n\t\treturn listClosedWorkflow(wfClient, pageSize, earliestTime, latestTime, domain, workflowID, workflowType, workflowStatus, c), nil\n\t}\n}\n\nfunc listArchivedWorkflows(c *cli.Context) (getWorkflowPageFn, error) {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tpageSize := c.Int(FlagPageSize)\n\tlistQuery, err := getRequiredOption(c, FlagListQuery)\n\tif err != nil {\n\t\treturn nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tif pageSize <= 0 {\n\t\tpageSize = defaultPageSizeForList\n\t}\n\n\tcontextTimeout := defaultContextTimeoutForListArchivedWorkflow\n\tif c.IsSet(FlagContextTimeout) {\n\t\tcontextTimeout = time.Duration(c.Int(FlagContextTimeout)) * time.Second\n\t}\n\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\trequest := &types.ListArchivedWorkflowExecutionsRequest{\n\t\t\tDomain:        domain,\n\t\t\tPageSize:      int32(pageSize),\n\t\t\tQuery:         listQuery,\n\t\t\tNextPageToken: nextPageToken,\n\t\t}\n\n\t\tctx, cancel := context.WithTimeout(c.Context, contextTimeout)\n\t\tdefer cancel()\n\n\t\tresult, err := wfClient.ListArchivedWorkflowExecutions(ctx, request)\n\t\tif err != nil {\n\t\t\tcancel()\n\t\t\treturn nil, nil, commoncli.Problem(\"Failed to list archived workflow.\", err)\n\t\t}\n\t\treturn result.Executions, result.NextPageToken, nil\n\t}, nil\n}\n\nfunc scanWorkflows(c *cli.Context) (getWorkflowPageFn, error) {\n\twfClient, err := getWorkflowClient(c)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tlistQuery := c.String(FlagListQuery)\n\tpageSize := c.Int(FlagPageSize)\n\tif pageSize <= 0 {\n\t\tpageSize = defaultPageSizeForScan\n\t}\n\n\treturn func(nextPageToken []byte) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\t\treturn scanWorkflowExecutions(wfClient, pageSize, nextPageToken, listQuery, c)\n\t}, nil\n}\n\nfunc scanWorkflowExecutions(client frontend.Client, pageSize int, nextPageToken []byte, query string, c *cli.Context) ([]*types.WorkflowExecutionInfo, []byte, error) {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn nil, nil, commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trequest := &types.ListWorkflowExecutionsRequest{\n\t\tDomain:        domain,\n\t\tPageSize:      int32(pageSize),\n\t\tNextPageToken: nextPageToken,\n\t\tQuery:         query,\n\t}\n\tctx, cancel, err := newContextForLongPoll(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn nil, nil, commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tresponse, err := client.ScanWorkflowExecutions(ctx, request)\n\tif err != nil {\n\t\treturn nil, nil, commoncli.Problem(\"Failed to list workflow.\", err)\n\t}\n\treturn response.Executions, response.NextPageToken, nil\n}\n\nfunc getWorkflowStatus(statusStr string) (types.WorkflowExecutionCloseStatus, error) {\n\tif status, ok := workflowClosedStatusMap[strings.ToLower(statusStr)]; ok {\n\t\treturn status, nil\n\t}\n\treturn -1, commoncli.Problem(optionErr, errors.New(\"option status is not one of allowed values \"+\n\t\t\"[completed, failed, canceled, terminated, continued_as_new, timed_out]\"))\n}\n\nfunc getWorkflowIDReusePolicy(value int) (*types.WorkflowIDReusePolicy, error) {\n\tif value >= 0 && types.WorkflowIDReusePolicy(value) <= types.WorkflowIDReusePolicyTerminateIfRunning {\n\t\treturn types.WorkflowIDReusePolicy(value).Ptr(), nil\n\t}\n\t// At this point, the policy should return if the value is valid\n\treturn nil, fmt.Errorf(\"Option %v value is not in supported range. %v\", FlagWorkflowIDReusePolicy, nil)\n}\n\n// default will print decoded raw\nfunc printListResults(executions []*types.WorkflowExecutionInfo, inJSON bool, more bool) {\n\tfor i, execution := range executions {\n\t\tif inJSON {\n\t\t\tj, _ := json.Marshal(execution)\n\t\t\tif more || i < len(executions)-1 {\n\t\t\t\tfmt.Println(string(j) + \",\")\n\t\t\t} else {\n\t\t\t\tfmt.Println(string(j))\n\t\t\t}\n\t\t} else {\n\t\t\tif more || i < len(executions)-1 {\n\t\t\t\tfmt.Println(anyToString(execution, true, 0) + \",\")\n\t\t\t} else {\n\t\t\t\tfmt.Println(anyToString(execution, true, 0))\n\t\t\t}\n\t\t}\n\t}\n}\n\n// ObserveHistory show the process of running workflow\nfunc ObserveHistory(c *cli.Context) error {\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid := c.String(FlagRunID)\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\n\tprintWorkflowProgress(c, domain, wid, rid)\n\treturn nil\n}\n\n// ResetWorkflow reset workflow\nfunc ResetWorkflow(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\treason, err := getRequiredOption(c, FlagReason)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tif len(reason) == 0 {\n\t\treturn commoncli.Problem(\"wrong reason\", fmt.Errorf(\"reason cannot be empty\"))\n\t}\n\teventID := c.Int64(FlagEventID)\n\tresetType := c.String(FlagResetType)\n\tdecisionOffset := c.Int(FlagDecisionOffset)\n\tif decisionOffset > 0 {\n\t\treturn commoncli.Problem(\"Only decision offset <=0 is supported\", nil)\n\t}\n\n\textraForResetType, ok := resetTypesMap[resetType]\n\tif !ok && eventID <= 0 {\n\t\treturn commoncli.Problem(\"Must specify valid eventID or valid resetType\", nil)\n\t}\n\tif ok && len(extraForResetType) > 0 {\n\t\tgetRequiredOption(c, extraForResetType)\n\t}\n\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\trid := c.String(FlagRunID)\n\tif rid == \"\" {\n\t\trid, err = getCurrentRunID(ctx, domain, wid, frontendClient)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Cannot get latest RunID as default\", err)\n\t\t}\n\t}\n\n\tresetBaseRunID := rid\n\tdecisionFinishID := eventID\n\tif resetType != \"\" {\n\t\tresetBaseRunID, decisionFinishID, err = getResetEventIDByType(ctx, c, resetType, decisionOffset, domain, wid, rid, frontendClient)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"getResetEventIDByType failed\", err)\n\t\t}\n\t}\n\tresp, err := frontendClient.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      resetBaseRunID,\n\t\t},\n\t\tReason:                fmt.Sprintf(\"%v:%v\", getCurrentUserFromEnv(), reason),\n\t\tDecisionFinishEventID: decisionFinishID,\n\t\tRequestID:             uuid.New(),\n\t\tSkipSignalReapply:     c.Bool(FlagSkipSignalReapply),\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"reset failed\", err)\n\t}\n\tprettyPrintJSONObject(getDeps(c).Output(), resp)\n\treturn nil\n}\n\nfunc processResets(c *cli.Context, domain string, wes chan types.WorkflowExecution, done chan bool, wg *sync.WaitGroup, params batchResetParamsType) {\n\tfor {\n\t\tselect {\n\t\tcase we := <-wes:\n\t\t\tfmt.Println(\"received: \", we.GetWorkflowID(), we.GetRunID())\n\t\t\twid := we.GetWorkflowID()\n\t\t\trid := we.GetRunID()\n\t\t\tvar err error\n\t\t\tfor i := 0; i < 3; i++ {\n\t\t\t\terr = doReset(c, domain, wid, rid, params)\n\t\t\t\tif err == nil {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\tif _, ok := err.(*types.BadRequestError); ok {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\tfmt.Println(\"failed and retry...: \", wid, rid, err)\n\t\t\t\ttime.Sleep(time.Millisecond * time.Duration(rand.Intn(2000)))\n\t\t\t}\n\t\t\ttime.Sleep(time.Millisecond * time.Duration(rand.Intn(1000)))\n\t\t\tif err != nil {\n\t\t\t\tfmt.Println(\"[ERROR] failed processing: \", wid, rid, err.Error())\n\t\t\t}\n\t\tcase <-done:\n\t\t\twg.Done()\n\t\t\treturn\n\t\t}\n\t}\n}\n\ntype batchResetParamsType struct {\n\treason               string\n\tskipCurrentOpen      bool\n\tskipCurrentCompleted bool\n\tnonDeterministicOnly bool\n\tskipBaseNotCurrent   bool\n\tdryRun               bool\n\tresetType            string\n\tdecisionOffset       int\n\tskipSignalReapply    bool\n}\n\n// ResetInBatch resets workflow in batch\nfunc ResetInBatch(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tresetType, err := getRequiredOption(c, FlagResetType)\n\tdecisionOffset := c.Int(FlagDecisionOffset)\n\tif decisionOffset > 0 {\n\t\treturn commoncli.Problem(\"Only decision offset <=0 is supported\", nil)\n\t}\n\n\tinFileName := c.String(FlagInputFile)\n\tquery := c.String(FlagListQuery)\n\texcludeFileName := c.String(FlagExcludeFile)\n\texcludeQuery := c.String(FlagExcludeWorkflowIDByQuery)\n\tseparator := c.String(FlagInputSeparator)\n\tparallel := c.Int(FlagParallismDeprecated)\n\tif parallel == 1 {\n\t\tparallel = c.Int(FlagParallelism)\n\t}\n\n\textraForResetType, ok := resetTypesMap[resetType]\n\tif !ok {\n\t\treturn commoncli.Problem(\"Not supported reset type\", nil)\n\t} else if len(extraForResetType) > 0 {\n\t\tgetRequiredOption(c, extraForResetType)\n\t}\n\n\tif excludeFileName != \"\" && excludeQuery != \"\" {\n\t\treturn commoncli.Problem(\"Only one of the excluding option is allowed\", nil)\n\t}\n\trsn, err := getRequiredOption(c, FlagReason)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tbatchResetParams := batchResetParamsType{\n\t\treason:               rsn,\n\t\tskipCurrentOpen:      c.Bool(FlagSkipCurrentOpen),\n\t\tskipCurrentCompleted: c.Bool(FlagSkipCurrentCompleted),\n\t\tnonDeterministicOnly: c.Bool(FlagNonDeterministicOnly),\n\t\tskipBaseNotCurrent:   c.Bool(FlagSkipBaseIsNotCurrent),\n\t\tdryRun:               c.Bool(FlagDryRun),\n\t\tresetType:            resetType,\n\t\tdecisionOffset:       decisionOffset,\n\t\tskipSignalReapply:    c.Bool(FlagSkipSignalReapply),\n\t}\n\n\tif inFileName == \"\" && query == \"\" {\n\t\treturn commoncli.Problem(\"Must provide input file or list query to get target workflows to reset\", nil)\n\t}\n\n\twg := &sync.WaitGroup{}\n\n\twes := make(chan types.WorkflowExecution)\n\tdone := make(chan bool)\n\tfor i := 0; i < parallel; i++ {\n\t\twg.Add(1)\n\t\tgo processResets(c, domain, wes, done, wg, batchResetParams)\n\t}\n\n\t// read excluded workflowIDs\n\texcludeWIDs := map[string]bool{}\n\tif excludeFileName != \"\" {\n\t\texcludeWIDs, err = loadWorkflowIDsFromFile(excludeFileName, separator)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Error loading WF Ids from file: \", err)\n\t\t}\n\t}\n\tif excludeQuery != \"\" {\n\t\tvar err error\n\t\texcludeWIDs, err = getAllWorkflowIDsByQuery(c, excludeQuery)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tfmt.Println(\"num of excluded WorkflowIDs:\", len(excludeWIDs))\n\n\tif len(inFileName) > 0 {\n\t\tinFile, err := os.Open(inFileName)\n\t\tif err != nil {\n\t\t\treturn commoncli.Problem(\"Open failed\", err)\n\t\t}\n\t\tdefer inFile.Close()\n\t\tscanner := bufio.NewScanner(inFile)\n\t\tidx := 0\n\t\tfor scanner.Scan() {\n\t\t\tidx++\n\t\t\tline := strings.TrimSpace(scanner.Text())\n\t\t\tif len(line) == 0 {\n\t\t\t\tfmt.Printf(\"line %v is empty, skipped\\n\", idx)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tcols := strings.Split(line, separator)\n\t\t\tif len(cols) < 1 {\n\t\t\t\treturn commoncli.Problem(\"Split failed\", fmt.Errorf(\"line %v has less than 1 cols separated by comma, only %v \", idx, len(cols)))\n\t\t\t}\n\t\t\tfmt.Printf(\"Start processing line %v ...\\n\", idx)\n\t\t\twid := strings.TrimSpace(cols[0])\n\t\t\trid := \"\"\n\t\t\tif len(cols) > 1 {\n\t\t\t\trid = strings.TrimSpace(cols[1])\n\t\t\t}\n\n\t\t\tif excludeWIDs[wid] {\n\t\t\t\tfmt.Println(\"skip by exclude file: \", wid, rid)\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\twes <- types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      rid,\n\t\t\t}\n\t\t}\n\t} else {\n\t\twfClient, err := getWorkflowClient(c)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tpageSize := 1000\n\t\tvar nextPageToken []byte\n\t\tvar result []*types.WorkflowExecutionInfo\n\t\tfor {\n\t\t\tresult, nextPageToken, err = scanWorkflowExecutions(wfClient, pageSize, nextPageToken, query, c)\n\t\t\tif err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t\tfor _, we := range result {\n\t\t\t\twid := we.Execution.GetWorkflowID()\n\t\t\t\trid := we.Execution.GetRunID()\n\t\t\t\tif excludeWIDs[wid] {\n\t\t\t\t\tfmt.Println(\"skip by exclude file: \", wid, rid)\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\n\t\t\t\twes <- types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: wid,\n\t\t\t\t\tRunID:      rid,\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif nextPageToken == nil {\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n\n\tclose(done)\n\tfmt.Println(\"wait for all goroutines...\")\n\twg.Wait()\n\treturn nil\n}\n\nfunc loadWorkflowIDsFromFile(excludeFileName, separator string) (map[string]bool, error) {\n\texcludeWIDs := map[string]bool{}\n\tif len(excludeFileName) > 0 {\n\t\t// This code is only used in the CLI. The input provided is from a trusted user.\n\t\t// #nosec\n\t\texcFile, err := os.Open(excludeFileName)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"Open failed2 %w\", err)\n\t\t}\n\t\tdefer excFile.Close()\n\t\tscanner := bufio.NewScanner(excFile)\n\t\tidx := 0\n\t\tfor scanner.Scan() {\n\t\t\tidx++\n\t\t\tline := strings.TrimSpace(scanner.Text())\n\t\t\tif len(line) == 0 {\n\t\t\t\tfmt.Printf(\"line %v is empty, skipped\\n\", idx)\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tcols := strings.Split(line, separator)\n\t\t\tif len(cols) < 1 {\n\t\t\t\treturn nil, fmt.Errorf(\"Split failed %w\", fmt.Errorf(\"line %v has less than 1 cols separated by comma, only %v \", idx, len(cols)))\n\t\t\t}\n\t\t\twid := strings.TrimSpace(cols[0])\n\t\t\texcludeWIDs[wid] = true\n\t\t}\n\t}\n\treturn excludeWIDs, nil\n}\n\nfunc printErrorAndReturn(msg string, err error) error {\n\tfmt.Println(msg)\n\treturn err\n}\n\nfunc doReset(c *cli.Context, domain, wid, rid string, params batchResetParamsType) error {\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresp, err := frontendClient.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn printErrorAndReturn(\"DescribeWorkflowExecution failed\", err)\n\t}\n\n\tcurrentRunID := resp.WorkflowExecutionInfo.Execution.GetRunID()\n\tif currentRunID != rid && params.skipBaseNotCurrent {\n\t\tfmt.Println(\"skip because base run is different from current run: \", wid, rid, currentRunID)\n\t\treturn nil\n\t}\n\tif rid == \"\" {\n\t\trid = currentRunID\n\t}\n\n\tif resp.WorkflowExecutionInfo.CloseStatus == nil || resp.WorkflowExecutionInfo.CloseTime == nil {\n\t\tif params.skipCurrentOpen {\n\t\t\tfmt.Println(\"skip because current run is open: \", wid, rid, currentRunID)\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tif resp.WorkflowExecutionInfo.GetCloseStatus() == types.WorkflowExecutionCloseStatusCompleted {\n\t\tif params.skipCurrentCompleted {\n\t\t\tfmt.Println(\"skip because current run is completed: \", wid, rid, currentRunID)\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tif params.nonDeterministicOnly {\n\t\tisLDN, err := isLastEventDecisionTaskFailedWithNonDeterminism(ctx, domain, wid, rid, frontendClient)\n\t\tif err != nil {\n\t\t\treturn printErrorAndReturn(\"check isLastEventDecisionTaskFailedWithNonDeterminism failed\", err)\n\t\t}\n\t\tif !isLDN {\n\t\t\tfmt.Println(\"skip because last event is not DecisionTaskFailedWithNonDeterminism\")\n\t\t\treturn nil\n\t\t}\n\t}\n\n\tresetBaseRunID, decisionFinishID, err := getResetEventIDByType(ctx, c, params.resetType, params.decisionOffset, domain, wid, rid, frontendClient)\n\tif err != nil {\n\t\treturn printErrorAndReturn(\"getResetEventIDByType failed\", err)\n\t}\n\tfmt.Println(\"DecisionFinishEventId for reset:\", wid, rid, resetBaseRunID, decisionFinishID)\n\n\tif params.dryRun {\n\t\tfmt.Printf(\"dry run to reset wid: %v, rid:%v to baseRunID:%v, eventID:%v \\n\", wid, rid, resetBaseRunID, decisionFinishID)\n\t} else {\n\t\tresp2, err := frontendClient.ResetWorkflowExecution(ctx, &types.ResetWorkflowExecutionRequest{\n\t\t\tDomain: domain,\n\t\t\tWorkflowExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: wid,\n\t\t\t\tRunID:      resetBaseRunID,\n\t\t\t},\n\t\t\tDecisionFinishEventID: decisionFinishID,\n\t\t\tRequestID:             uuid.New(),\n\t\t\tReason:                fmt.Sprintf(\"%v:%v\", getCurrentUserFromEnv(), params.reason),\n\t\t\tSkipSignalReapply:     params.skipSignalReapply,\n\t\t})\n\n\t\tif err != nil {\n\t\t\treturn printErrorAndReturn(\"ResetWorkflowExecution failed\", err)\n\t\t}\n\t\tfmt.Println(\"new runID for wid/rid is ,\", wid, rid, resp2.GetRunID())\n\t}\n\n\treturn nil\n}\n\nfunc isLastEventDecisionTaskFailedWithNonDeterminism(ctx context.Context, domain, wid, rid string, frontendClient frontend.Client) (bool, error) {\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   nil,\n\t}\n\n\tvar firstEvent, decisionFailed *types.HistoryEvent\n\tfor {\n\t\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\t\tif err != nil {\n\t\t\treturn false, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t\t}\n\t\tfor _, e := range resp.GetHistory().GetEvents() {\n\t\t\tif firstEvent == nil {\n\t\t\t\tfirstEvent = e\n\t\t\t}\n\t\t\tif e.GetEventType() == types.EventTypeDecisionTaskFailed {\n\t\t\t\tdecisionFailed = e\n\t\t\t} else if e.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\t\tdecisionFailed = nil\n\t\t\t}\n\t\t}\n\t\tif len(resp.NextPageToken) != 0 {\n\t\t\treq.NextPageToken = resp.NextPageToken\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\n\tif decisionFailed != nil {\n\t\tattr := decisionFailed.GetDecisionTaskFailedEventAttributes()\n\t\tif attr.GetCause() == types.DecisionTaskFailedCauseWorkflowWorkerUnhandledFailure ||\n\t\t\tstrings.Contains(string(attr.GetDetails()), \"nondeterministic\") {\n\t\t\tfmt.Printf(\"found non-deterministic workflow wid:%v, rid:%v, originalStartTime:%v \\n\", wid, rid, time.Unix(0, firstEvent.GetTimestamp()))\n\t\t\treturn true, nil\n\t\t}\n\t}\n\n\treturn false, nil\n}\n\nfunc getResetEventIDByType(\n\tctx context.Context,\n\tc *cli.Context,\n\tresetType string, decisionOffset int,\n\tdomain, wid, rid string,\n\tfrontendClient frontend.Client,\n) (resetBaseRunID string, decisionFinishID int64, err error) {\n\t// default to the same runID\n\tresetBaseRunID = rid\n\n\tfmt.Println(\"resetType:\", resetType)\n\tswitch resetType {\n\tcase resetTypeLastDecisionCompleted:\n\t\tdecisionFinishID, err = getLastDecisionTaskByType(ctx, domain, wid, rid, frontendClient, types.EventTypeDecisionTaskCompleted, decisionOffset)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\tcase resetTypeLastContinuedAsNew:\n\t\t// this reset type may change the base runID\n\t\tresetBaseRunID, decisionFinishID, err = getLastContinueAsNewID(ctx, domain, wid, rid, frontendClient)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\tcase resetTypeFirstDecisionCompleted:\n\t\tdecisionFinishID, err = getFirstDecisionTaskByType(ctx, domain, wid, rid, frontendClient, types.EventTypeDecisionTaskCompleted)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\tcase resetTypeBadBinary:\n\t\tbinCheckSum := c.String(FlagResetBadBinaryChecksum)\n\t\tdecisionFinishID, err = getBadDecisionCompletedID(ctx, domain, wid, rid, binCheckSum, frontendClient)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\tcase resetTypeDecisionCompletedTime:\n\t\tearliestTime, err := parseTime(c.String(FlagEarliestTime), 0)\n\t\tif err != nil {\n\t\t\treturn \"\", 0, fmt.Errorf(\"Get reset event id by type failed: %w\", err)\n\t\t}\n\t\tdecisionFinishID, err = getEarliestDecisionID(ctx, domain, wid, rid, earliestTime, frontendClient)\n\t\tif err != nil {\n\t\t\treturn \"\", 0, fmt.Errorf(\"Get reset event id by type failed: %w\", err)\n\t\t}\n\tcase resetTypeFirstDecisionScheduled:\n\t\tdecisionFinishID, err = getFirstDecisionTaskByType(ctx, domain, wid, rid, frontendClient, types.EventTypeDecisionTaskScheduled)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t\t// decisionFinishID is exclusive in reset API\n\t\tdecisionFinishID++\n\tcase resetTypeLastDecisionScheduled:\n\t\tdecisionFinishID, err = getLastDecisionTaskByType(ctx, domain, wid, rid, frontendClient, types.EventTypeDecisionTaskScheduled, decisionOffset)\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t\t// decisionFinishID is exclusive in reset API\n\t\tdecisionFinishID++\n\tdefault:\n\t\tpanic(\"not supported resetType\")\n\t}\n\treturn\n}\n\nfunc getFirstDecisionTaskByType(\n\tctx context.Context,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tfrontendClient frontend.Client,\n\tdecisionType types.EventType,\n) (decisionFinishID int64, err error) {\n\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   nil,\n\t}\n\n\tfor {\n\t\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\t\tif err != nil {\n\t\t\treturn 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t\t}\n\n\t\tfor _, e := range resp.GetHistory().GetEvents() {\n\t\t\tif e.GetEventType() == decisionType {\n\t\t\t\tdecisionFinishID = e.ID\n\t\t\t\treturn decisionFinishID, nil\n\t\t\t}\n\t\t}\n\n\t\tif len(resp.NextPageToken) != 0 {\n\t\t\treq.NextPageToken = resp.NextPageToken\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\treturn decisionFinishID, printErrorAndReturn(\"Get DecisionFinishID failed\", fmt.Errorf(\"no DecisionFinishID\"))\n}\n\nfunc getCurrentRunID(ctx context.Context, domain, wid string, frontendClient frontend.Client) (string, error) {\n\tresp, err := frontendClient.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\treturn resp.WorkflowExecutionInfo.Execution.GetRunID(), nil\n}\n\nfunc getBadDecisionCompletedID(ctx context.Context, domain, wid, rid, binChecksum string, frontendClient frontend.Client) (decisionFinishID int64, err error) {\n\tresp, err := frontendClient.DescribeWorkflowExecution(ctx, &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t})\n\tif err != nil {\n\t\treturn 0, printErrorAndReturn(\"DescribeWorkflowExecution failed\", err)\n\t}\n\n\t_, p := execution.FindAutoResetPoint(clock.NewRealTimeSource(), &types.BadBinaries{\n\t\tBinaries: map[string]*types.BadBinaryInfo{\n\t\t\tbinChecksum: {},\n\t\t},\n\t}, resp.WorkflowExecutionInfo.AutoResetPoints)\n\tif p != nil {\n\t\tdecisionFinishID = p.GetFirstDecisionCompletedID()\n\t}\n\n\tif decisionFinishID == 0 {\n\t\treturn 0, printErrorAndReturn(\"Get DecisionFinishID failed\", &types.BadRequestError{Message: \"no DecisionFinishID\"})\n\t}\n\treturn\n}\n\nfunc getLastDecisionTaskByType(\n\tctx context.Context,\n\tdomain string,\n\tworkflowID string,\n\trunID string,\n\tfrontendClient frontend.Client,\n\tdecisionType types.EventType,\n\tdecisionOffset int,\n) (int64, error) {\n\n\t// this fixedSizeQueue is for remembering the offset decision eventID\n\tfixedSizeQueue := make([]int64, 0)\n\tsize := int(math.Abs(float64(decisionOffset))) + 1\n\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: workflowID,\n\t\t\tRunID:      runID,\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   nil,\n\t}\n\n\tfor {\n\t\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\t\tif err != nil {\n\t\t\treturn 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t\t}\n\n\t\tfor _, e := range resp.GetHistory().GetEvents() {\n\t\t\tif e.GetEventType() == decisionType {\n\t\t\t\tdecisionEventID := e.ID\n\t\t\t\tfixedSizeQueue = append(fixedSizeQueue, decisionEventID)\n\t\t\t\tif len(fixedSizeQueue) > size {\n\t\t\t\t\tfixedSizeQueue = fixedSizeQueue[1:]\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\tif len(resp.NextPageToken) != 0 {\n\t\t\treq.NextPageToken = resp.NextPageToken\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\tif len(fixedSizeQueue) == 0 {\n\t\treturn 0, printErrorAndReturn(\"Get DecisionFinishID failed\", fmt.Errorf(\"no DecisionFinishID\"))\n\t}\n\treturn fixedSizeQueue[0], nil\n}\n\nfunc getLastContinueAsNewID(ctx context.Context, domain, wid, rid string, frontendClient frontend.Client) (resetBaseRunID string, decisionFinishID int64, err error) {\n\t// get first event\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t\tMaximumPageSize: 1,\n\t\tNextPageToken:   nil,\n\t}\n\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\tif err != nil {\n\t\treturn \"\", 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t}\n\tfirstEvent := resp.History.Events[0]\n\tresetBaseRunID = firstEvent.GetWorkflowExecutionStartedEventAttributes().GetContinuedExecutionRunID()\n\tif resetBaseRunID == \"\" {\n\t\treturn \"\", 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", fmt.Errorf(\"cannot get resetBaseRunID\"))\n\t}\n\n\treq = &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      resetBaseRunID,\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   nil,\n\t}\n\tfor {\n\t\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\t\tif err != nil {\n\t\t\treturn \"\", 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t\t}\n\t\tfor _, e := range resp.GetHistory().GetEvents() {\n\t\t\tif e.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\t\tdecisionFinishID = e.ID\n\t\t\t}\n\t\t}\n\t\tif len(resp.NextPageToken) != 0 {\n\t\t\treq.NextPageToken = resp.NextPageToken\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\tif decisionFinishID == 0 {\n\t\treturn \"\", 0, printErrorAndReturn(\"Get DecisionFinishID failed\", fmt.Errorf(\"no DecisionFinishID\"))\n\t}\n\treturn\n}\n\n// CompleteActivity completes an activity\nfunc CompleteActivity(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid, err := getRequiredOption(c, FlagRunID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tactivityID, err := getRequiredOption(c, FlagActivityID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tresult, err := getRequiredOption(c, FlagResult)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tidentity, err := getRequiredOption(c, FlagIdentity)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = frontendClient.RespondActivityTaskCompletedByID(ctx, &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     domain,\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t\tActivityID: activityID,\n\t\tResult:     []byte(result),\n\t\tIdentity:   identity,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Completing activity failed\", err)\n\t}\n\tfmt.Println(\"Complete activity successfully.\")\n\treturn nil\n}\n\n// FailActivity fails an activity\nfunc FailActivity(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\twid, err := getRequiredOption(c, FlagWorkflowID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\trid, err := getRequiredOption(c, FlagRunID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tactivityID, err := getRequiredOption(c, FlagActivityID)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\treason, err := getRequiredOption(c, FlagReason)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tdetail, err := getRequiredOption(c, FlagDetail)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tidentity, err := getRequiredOption(c, FlagIdentity)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tctx, cancel, err := newContext(c)\n\tdefer cancel()\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Error creating context: \", err)\n\t}\n\tfrontendClient, err := getDeps(c).ServerFrontendClient(c)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = frontendClient.RespondActivityTaskFailedByID(ctx, &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     domain,\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t\tActivityID: activityID,\n\t\tReason:     common.StringPtr(reason),\n\t\tDetails:    []byte(detail),\n\t\tIdentity:   identity,\n\t})\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Failing activity failed\", err)\n\t}\n\tfmt.Println(\"Fail activity successfully.\")\n\treturn nil\n}\n\n// ObserveHistoryWithID show the process of running workflow\nfunc ObserveHistoryWithID(c *cli.Context) error {\n\tdomain, err := getRequiredOption(c, FlagDomain)\n\tif err != nil {\n\t\treturn commoncli.Problem(\"Required flag not found: \", err)\n\t}\n\tif !c.Args().Present() {\n\t\treturn commoncli.Problem(\"Argument workflow_id is required.\", nil)\n\t}\n\twid := c.Args().First()\n\trid := \"\"\n\tif c.NArg() >= 2 {\n\t\trid = c.Args().Get(1)\n\t}\n\n\tprintWorkflowProgress(c, domain, wid, rid)\n\treturn nil\n}\n\nfunc getEarliestDecisionID(\n\tctx context.Context,\n\tdomain string, wid string,\n\trid string, earliestTime int64,\n\tfrontendClient frontend.Client,\n) (decisionFinishID int64, err error) {\n\treq := &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: domain,\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: wid,\n\t\t\tRunID:      rid,\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   nil,\n\t}\n\nOuterLoop:\n\tfor {\n\t\tresp, err := frontendClient.GetWorkflowExecutionHistory(ctx, req)\n\t\tif err != nil {\n\t\t\treturn 0, printErrorAndReturn(\"GetWorkflowExecutionHistory failed\", err)\n\t\t}\n\t\tfor _, e := range resp.GetHistory().GetEvents() {\n\t\t\tif e.GetEventType() == types.EventTypeDecisionTaskCompleted {\n\t\t\t\tif e.GetTimestamp() >= earliestTime {\n\t\t\t\t\tdecisionFinishID = e.ID\n\t\t\t\t\tbreak OuterLoop\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tif len(resp.NextPageToken) != 0 {\n\t\t\treq.NextPageToken = resp.NextPageToken\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\tif decisionFinishID == 0 {\n\t\treturn 0, printErrorAndReturn(\"Get DecisionFinishID failed\", fmt.Errorf(\"no DecisionFinishID\"))\n\t}\n\treturn\n}\n\nfunc mapQueryConsistencyLevelFromFlag(flag string) (types.QueryConsistencyLevel, error) {\n\tvar consistencyLevel types.QueryConsistencyLevel\n\tswitch flag {\n\tcase \"eventual\":\n\t\tconsistencyLevel = types.QueryConsistencyLevelEventual\n\tcase \"strong\":\n\t\tconsistencyLevel = types.QueryConsistencyLevelStrong\n\tdefault:\n\t\treturn consistencyLevel, fmt.Errorf(\"invalid query consistency level %q, valid values are \\\"eventual\\\" and \\\"strong\\\"\", flag)\n\t}\n\n\treturn consistencyLevel, nil\n}\n\nfunc mapQueryRejectConditionFromFlag(flag string) (types.QueryRejectCondition, error) {\n\tvar rejectCondition types.QueryRejectCondition\n\tswitch flag {\n\tcase \"not_open\":\n\t\trejectCondition = types.QueryRejectConditionNotOpen\n\tcase \"not_completed_cleanly\":\n\t\trejectCondition = types.QueryRejectConditionNotCompletedCleanly\n\tdefault:\n\t\treturn rejectCondition, fmt.Errorf(\"invalid reject condition %v, valid values are \\\"not_open\\\" and \\\"not_completed_cleanly\\\"\", flag)\n\t}\n\n\treturn rejectCondition, nil\n}\n\nfunc parseClusterAttributes(clusterAttributeScope string, clusterAttributeName string) (*types.ActiveClusterSelectionPolicy, error) {\n\tif clusterAttributeScope == \"\" && clusterAttributeName == \"\" {\n\t\t// default case, these values are optional so most workflows will not use them\n\t\treturn nil, nil\n\t}\n\tif clusterAttributeScope == \"\" || clusterAttributeName == \"\" {\n\t\treturn nil, fmt.Errorf(\"invalid cluster attribute, scope or name is empty, either use both or none to start workflows. got %q.%q\", clusterAttributeScope, clusterAttributeName)\n\t}\n\tpolicy := &types.ActiveClusterSelectionPolicy{\n\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\tScope: clusterAttributeScope,\n\t\t\tName:  clusterAttributeName,\n\t\t},\n\t}\n\treturn policy, nil\n}\n"
  },
  {
    "path": "tools/cli/workflow_commands_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage cli\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"flag\"\n\t\"fmt\"\n\t\"os\"\n\t\"sync\"\n\t\"testing\"\n\t\"time\"\n\n\t\"github.com/opentracing/opentracing-go\"\n\t\"github.com/opentracing/opentracing-go/mocktracer\"\n\t\"github.com/pborman/uuid\"\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/urfave/cli/v2\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/client/admin\"\n\t\"github.com/uber/cadence/client/frontend\"\n\t\"github.com/uber/cadence/common\"\n\t\"github.com/uber/cadence/common/types\"\n\t\"github.com/uber/cadence/tools/cli/clitest\"\n)\n\nfunc TestConstructStartWorkflowRequest(t *testing.T) {\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(FlagTaskList, \"test-task-list\", \"tasklist\")\n\tset.String(FlagWorkflowType, \"test-workflow-type\", \"workflow-type\")\n\tset.Int(\"execution_timeout\", 100, \"execution_timeout\")\n\tset.Int(\"decision_timeout\", 50, \"decision_timeout\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.Int(\"workflow_id_reuse_policy\", 1, \"workflow_id_reuse_policy\")\n\tset.String(\"input\", \"{}\", \"input\")\n\tset.String(\"cron_schedule\", \"* * * * *\", \"cron_schedule\")\n\tset.Int(\"retry_attempts\", 5, \"retry_attempts\")\n\tset.Int(\"retry_expiration\", 600, \"retry_expiration\")\n\tset.Int(\"retry_interval\", 10, \"retry_interval\")\n\tset.Float64(\"retry_backoff\", 2.0, \"retry_backoff\")\n\tset.Int(\"retry_max_interval\", 100, \"retry_max_interval\")\n\tset.Int(DelayStartSeconds, 5, DelayStartSeconds)\n\tset.Int(JitterStartSeconds, 2, JitterStartSeconds)\n\tset.String(\"first_run_at_time\", \"2024-07-24T12:00:00Z\", \"first-run-at-time\")\n\tset.Int(\"cron_overlap_policy\", 0, \"cron_overlap_policy\")\n\n\tc := cli.NewContext(nil, set, nil)\n\t// inject context with span\n\ttracer := mocktracer.New()\n\tspan, ctx := opentracing.StartSpanFromContextWithTracer(context.Background(), tracer, \"test-span\")\n\tspan.SetBaggageItem(\"tracer-test-key\", \"tracer-test-value\")\n\tdefer span.Finish()\n\tc.Context = ctx\n\n\tassert.NoError(t, c.Set(FlagDomain, \"test-domain\"))\n\tassert.NoError(t, c.Set(FlagTaskList, \"test-task-list\"))\n\tassert.NoError(t, c.Set(FlagWorkflowType, \"test-workflow-type\"))\n\tassert.NoError(t, c.Set(\"execution_timeout\", \"100\"))\n\tassert.NoError(t, c.Set(\"decision_timeout\", \"50\"))\n\tassert.NoError(t, c.Set(\"workflow_id\", \"test-workflow-id\"))\n\tassert.NoError(t, c.Set(\"workflow_id_reuse_policy\", \"1\"))\n\tassert.NoError(t, c.Set(\"input\", \"{}\"))\n\tassert.NoError(t, c.Set(\"cron_schedule\", \"* * * * *\"))\n\tassert.NoError(t, c.Set(\"retry_attempts\", \"5\"))\n\tassert.NoError(t, c.Set(\"retry_expiration\", \"600\"))\n\tassert.NoError(t, c.Set(\"retry_interval\", \"10\"))\n\tassert.NoError(t, c.Set(\"retry_backoff\", \"2.0\"))\n\tassert.NoError(t, c.Set(\"retry_max_interval\", \"100\"))\n\tassert.NoError(t, c.Set(DelayStartSeconds, \"5\"))\n\tassert.NoError(t, c.Set(JitterStartSeconds, \"2\"))\n\tassert.NoError(t, c.Set(\"first_run_at_time\", \"2024-07-24T12:00:00Z\"))\n\tassert.NoError(t, c.Set(\"cron_overlap_policy\", \"0\"))\n\trequest, err := constructStartWorkflowRequest(c)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, request)\n\tassert.Equal(t, \"test-domain\", request.Domain)\n\tassert.Equal(t, \"test-task-list\", request.TaskList.Name)\n\tassert.Equal(t, \"test-workflow-type\", request.WorkflowType.Name)\n\tassert.Equal(t, int32(100), *request.ExecutionStartToCloseTimeoutSeconds)\n\tassert.Equal(t, int32(50), *request.TaskStartToCloseTimeoutSeconds)\n\tassert.Equal(t, \"test-workflow-id\", request.WorkflowID)\n\tassert.NotNil(t, request.WorkflowIDReusePolicy)\n\tassert.Equal(t, int32(5), *request.DelayStartSeconds)\n\tassert.Equal(t, int32(2), *request.JitterStartSeconds)\n\tassert.Contains(t, request.Header.Fields, \"mockpfx-baggage-tracer-test-key\")\n\tassert.Equal(t, []byte(\"tracer-test-value\"), request.Header.Fields[\"mockpfx-baggage-tracer-test-key\"])\n\n\tfirstRunAt, err := time.Parse(time.RFC3339, \"2024-07-24T12:00:00Z\")\n\tassert.NoError(t, err)\n\tassert.Equal(t, firstRunAt.UnixNano(), *request.FirstRunAtTimeStamp)\n\tassert.Equal(t, types.CronOverlapPolicySkipped, *request.CronOverlapPolicy)\n}\n\nfunc Test_PrintAutoResetPoints(t *testing.T) {\n\ttests := []struct {\n\t\tname string\n\t\tresp *types.DescribeWorkflowExecutionResponse\n\t}{\n\t\t{\n\t\t\tname: \"empty reset points\",\n\t\t\tresp: &types.DescribeWorkflowExecutionResponse{\n\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tresp: &types.DescribeWorkflowExecutionResponse{\n\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tBinaryChecksum: \"test-binary-checksum\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\terr := printAutoResetPoints(tt.resp)\n\t\t\tassert.NoError(t, err)\n\t\t})\n\t}\n}\n\nfunc Test_DescribeWorkflow(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tc := getMockContext(t, nil, app)\n\terr := DescribeWorkflow(c)\n\tassert.NoError(t, err)\n}\n\nfunc Test_DescribeWorkflowWithID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tc := getMockContext(t, nil, app)\n\terr := DescribeWorkflowWithID(c)\n\tassert.NoError(t, err)\n}\n\nfunc Test_DescribeWorkflowWithID_Error(t *testing.T) {\n\tset := flag.NewFlagSet(\"test\", 0)\n\terr := DescribeWorkflowWithID(cli.NewContext(nil, set, nil))\n\tassert.Error(t, err)\n\n\t// WF helper describe failed\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\"CustomKeywordField\": []byte(\"test-value\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\n\tc := getMockContext(t, nil, app)\n\terr = DescribeWorkflowWithID(c)\n\tassert.Error(t, err)\n}\n\nfunc getMockContext(t *testing.T, set *flag.FlagSet, app *cli.App) *cli.Context {\n\tif set == nil {\n\t\tset = flag.NewFlagSet(\"test\", 0)\n\t\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\t\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\t\tset.String(\"run_id\", \"test-run-id\", \"run_id\")\n\t\tset.Bool(\"print_reset_points\", true, \"print_reset_points\")\n\t\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\t}\n\n\tc := cli.NewContext(app, set, nil)\n\n\treturn c\n}\n\nfunc Test_ListAllWorkflow(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tserverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.CountWorkflowExecutionsResponse{\n\t\tCount: int64(1),\n\t}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListClosedWorkflowExecutionsResponse{}, nil).AnyTimes()\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.String(\"run_id\", \"test-run-id\", \"run_id\")\n\tset.String(\"status\", \"open\", \"status\")\n\tc := getMockContext(t, set, app)\n\terr := ListAllWorkflow(c)\n\tassert.NoError(t, err)\n}\n\nfunc Test_ConvertSearchAttributesToMapOfInterface(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tin            *types.SearchAttributes\n\t\tout           map[string]interface{}\n\t\texpectedError bool\n\t\tmockResponse  *types.GetSearchAttributesResponse\n\t\tmockError     error\n\t}{\n\t\t{\n\t\t\tname:          \"empty search attributes\",\n\t\t\tout:           nil,\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname: \"error when get search attributes\",\n\t\t\tin: &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\"CustomKeywordField\": []byte(\"test-value\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tout:           nil,\n\t\t\tmockError:     errors.New(\"test-error\"),\n\t\t\texpectedError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"error deserialize search attributes\",\n\t\t\tin: &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\"CustomKeywordField\": []byte(\"test-value\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tout:           nil,\n\t\t\tmockError:     nil,\n\t\t\texpectedError: true,\n\t\t},\n\t\t{\n\t\t\tname: \"normal case\",\n\t\t\tin: &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\"CustomKeywordField\": []byte(`\"test-value\"`), // Assuming the value is a serialized string\n\t\t\t\t},\n\t\t\t},\n\t\t\tout: map[string]interface{}{\n\t\t\t\t\"CustomKeywordField\": \"test-value\", // Expected deserialized value\n\t\t\t},\n\t\t\tmockResponse: &types.GetSearchAttributesResponse{\n\t\t\t\tKeys: map[string]types.IndexedValueType{\n\t\t\t\t\t\"CustomKeywordField\": types.IndexedValueTypeKeyword,\n\t\t\t\t},\n\t\t\t},\n\t\t\tmockError:     nil,\n\t\t\texpectedError: false,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\t\t\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: serverFrontendClient,\n\t\t\t\tserverAdminClient:    serverAdminClient,\n\t\t\t})\n\t\t\tc := getMockContext(t, nil, app)\n\t\t\tserverFrontendClient.EXPECT().GetSearchAttributes(gomock.Any()).Return(tt.mockResponse, tt.mockError).AnyTimes()\n\t\t\tout, err := convertSearchAttributesToMapOfInterface(tt.in, serverFrontendClient, c)\n\t\t\tif tt.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.out, out)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_GetAllWorkflowIDsByQuery(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\t// missing required flag\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tc := cli.NewContext(app, set, nil)\n\t_, err := getAllWorkflowIDsByQuery(c, \"WorkflowType='test-workflow-type'\")\n\tassert.Error(t, err)\n\n\tc = getMockContext(t, nil, app)\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), &types.ListWorkflowExecutionsRequest{\n\t\tQuery:    \"WorkflowType='test-workflow-type'\",\n\t\tDomain:   \"test-domain\",\n\t\tPageSize: 1000,\n\t}).Return(&types.ListWorkflowExecutionsResponse{}, nil).Times(1)\n\n\tresp, err := getAllWorkflowIDsByQuery(c, \"WorkflowType='test-workflow-type'\")\n\tassert.NoError(t, err)\n\tassert.NotNil(t, resp)\n\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), &types.ListWorkflowExecutionsRequest{\n\t\tQuery:    \"WorkflowType='test-workflow-type'\",\n\t\tDomain:   \"test-domain\",\n\t\tPageSize: 1000,\n\t}).Return(&types.ListWorkflowExecutionsResponse{}, errors.New(\"test-error\")).Times(1)\n\n\t_, err = getAllWorkflowIDsByQuery(c, \"WorkflowType='test-workflow-type'\")\n\tassert.Error(t, err)\n}\n\nfunc Test_GetWorkflowStatus(t *testing.T) {\n\ttests := []struct {\n\t\tname          string\n\t\tinput         string\n\t\texpected      types.WorkflowExecutionCloseStatus\n\t\texpectedError bool\n\t}{\n\t\t{\n\t\t\tname:          \"Valid status - completed\",\n\t\t\tinput:         \"completed\",\n\t\t\texpected:      types.WorkflowExecutionCloseStatusCompleted,\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Valid alias - fail\",\n\t\t\tinput:         \"fail\",\n\t\t\texpected:      types.WorkflowExecutionCloseStatusFailed,\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Valid status - timed_out\",\n\t\t\tinput:         \"timed_out\",\n\t\t\texpected:      types.WorkflowExecutionCloseStatusTimedOut,\n\t\t\texpectedError: false,\n\t\t},\n\t\t{\n\t\t\tname:          \"Invalid status\",\n\t\t\tinput:         \"invalid\",\n\t\t\texpected:      -1,\n\t\t\texpectedError: true,\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tstatus, err := getWorkflowStatus(tt.input)\n\n\t\t\tif tt.expectedError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Equal(t, -1, int(status))\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tt.expected, status)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ConvertDescribeWorkflowExecutionResponse(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\n\tt.Run(\"basic conversion\", func(t *testing.T) {\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t{\n\t\t\t\t\tActivityID: \"test-activity-id\",\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"test-activity-type\",\n\t\t\t\t\t},\n\t\t\t\t\tHeartbeatDetails:   []byte(\"test-heartbeat-details\"),\n\t\t\t\t\tLastFailureDetails: []byte(\"test-failure-details\"),\n\t\t\t\t},\n\t\t\t},\n\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\tState: nil,\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, \"test-workflow-id\", resp.WorkflowExecutionInfo.Execution.WorkflowID)\n\t\tassert.Equal(t, 1, len(resp.PendingActivities))\n\t\tassert.NotNil(t, resp.PendingDecision)\n\t})\n\n\tt.Run(\"with execution configuration\", func(t *testing.T) {\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tExecutionConfiguration: &types.WorkflowExecutionConfiguration{\n\t\t\t\tTaskList: &types.TaskList{Name: \"test-task-list\"},\n\t\t\t},\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-wf-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\tName: \"test-workflow-type\",\n\t\t\t\t},\n\t\t\t\tHistoryLength: 100,\n\t\t\t},\n\t\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t\t\tPendingChildren:   []*types.PendingChildExecutionInfo{},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, resp.ExecutionConfiguration)\n\t\tassert.Equal(t, \"test-task-list\", resp.ExecutionConfiguration.TaskList.Name)\n\t\tassert.Equal(t, int64(100), resp.WorkflowExecutionInfo.HistoryLength)\n\t})\n\n\tt.Run(\"with pending activities and conversion of timestamps\", func(t *testing.T) {\n\t\tnow := time.Now().UnixNano()\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-wf\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t\tStartTime: common.Int64Ptr(now),\n\t\t\t\tCloseTime: common.Int64Ptr(now + 1000),\n\t\t\t},\n\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t{\n\t\t\t\t\tActivityID: \"activity-1\",\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"ActivityType1\",\n\t\t\t\t\t},\n\t\t\t\t\tState:                  types.PendingActivityStateScheduled.Ptr(),\n\t\t\t\t\tScheduledTimestamp:     common.Int64Ptr(now),\n\t\t\t\t\tLastStartedTimestamp:   common.Int64Ptr(now + 500),\n\t\t\t\t\tLastHeartbeatTimestamp: common.Int64Ptr(now + 600),\n\t\t\t\t\tExpirationTimestamp:    common.Int64Ptr(now + 2000),\n\t\t\t\t\tAttempt:                1,\n\t\t\t\t\tMaximumAttempts:        3,\n\t\t\t\t\tLastFailureReason:      common.StringPtr(\"test-failure\"),\n\t\t\t\t\tLastWorkerIdentity:     \"worker-1\",\n\t\t\t\t\tScheduleID:             1,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tActivityID: \"activity-2\",\n\t\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\t\tName: \"ActivityType2\",\n\t\t\t\t\t},\n\t\t\t\t\tHeartbeatDetails:   []byte(\"heartbeat-data\"),\n\t\t\t\t\tLastFailureDetails: []byte(\"failure-data\"),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, 2, len(resp.PendingActivities))\n\n\t\t// Check first activity has timestamps converted\n\t\tassert.NotNil(t, resp.PendingActivities[0].ScheduledTimestamp)\n\t\tassert.NotNil(t, resp.PendingActivities[0].LastStartedTimestamp)\n\t\tassert.NotNil(t, resp.PendingActivities[0].LastHeartbeatTimestamp)\n\t\tassert.NotNil(t, resp.PendingActivities[0].ExpirationTimestamp)\n\t\tassert.Equal(t, int32(1), resp.PendingActivities[0].Attempt)\n\t\tassert.Equal(t, int32(3), resp.PendingActivities[0].MaximumAttempts)\n\t\tassert.Equal(t, \"test-failure\", *resp.PendingActivities[0].LastFailureReason)\n\n\t\t// Check second activity has byte fields converted\n\t\tassert.NotNil(t, resp.PendingActivities[1].HeartbeatDetails)\n\t\tassert.Equal(t, \"heartbeat-data\", *resp.PendingActivities[1].HeartbeatDetails)\n\t\tassert.NotNil(t, resp.PendingActivities[1].LastFailureDetails)\n\t\tassert.Equal(t, \"failure-data\", *resp.PendingActivities[1].LastFailureDetails)\n\t})\n\n\tt.Run(\"with pending decision and timestamps\", func(t *testing.T) {\n\t\tnow := time.Now().UnixNano()\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-wf\",\n\t\t\t\t\tRunID:      \"test-run\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tPendingDecision: &types.PendingDecisionInfo{\n\t\t\t\tState:              types.PendingDecisionStateScheduled.Ptr(),\n\t\t\t\tScheduledTimestamp: common.Int64Ptr(now),\n\t\t\t\tStartedTimestamp:   common.Int64Ptr(now + 100),\n\t\t\t\tAttempt:            5,\n\t\t\t\tScheduleID:         42,\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, resp.PendingDecision)\n\t\tassert.NotNil(t, resp.PendingDecision.ScheduledTimestamp)\n\t\tassert.NotNil(t, resp.PendingDecision.StartedTimestamp)\n\t\tassert.Equal(t, int64(5), resp.PendingDecision.Attempt)\n\t\tassert.Equal(t, int64(42), resp.PendingDecision.ScheduleID)\n\t})\n\n\tt.Run(\"with pending children\", func(t *testing.T) {\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"parent-wf\",\n\t\t\t\t\tRunID:      \"parent-run\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tPendingChildren: []*types.PendingChildExecutionInfo{\n\t\t\t\t{\n\t\t\t\t\tWorkflowID:        \"child-wf-1\",\n\t\t\t\t\tRunID:             \"child-run-1\",\n\t\t\t\t\tWorkflowTypeName:  \"ChildWorkflowType\",\n\t\t\t\t\tInitiatedID:       10,\n\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyTerminate.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tWorkflowID:        \"child-wf-2\",\n\t\t\t\t\tRunID:             \"child-run-2\",\n\t\t\t\t\tWorkflowTypeName:  \"ChildWorkflowType2\",\n\t\t\t\t\tInitiatedID:       20,\n\t\t\t\t\tParentClosePolicy: types.ParentClosePolicyAbandon.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.Equal(t, 2, len(resp.PendingChildren))\n\t\tassert.Equal(t, \"child-wf-1\", resp.PendingChildren[0].WorkflowID)\n\t\tassert.Equal(t, \"child-wf-2\", resp.PendingChildren[1].WorkflowID)\n\t})\n\n\tt.Run(\"with workflow execution info fields\", func(t *testing.T) {\n\t\tnow := time.Now().UnixNano()\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"run-id\",\n\t\t\t\t},\n\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\tName: \"WorkflowType\",\n\t\t\t\t},\n\t\t\t\tStartTime:      common.Int64Ptr(now),\n\t\t\t\tCloseTime:      common.Int64Ptr(now + 1000),\n\t\t\t\tCloseStatus:    types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\t\tHistoryLength:  50,\n\t\t\t\tParentDomainID: common.StringPtr(\"parent-domain\"),\n\t\t\t\tParentExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"parent-wf\",\n\t\t\t\t\tRunID:      \"parent-run\",\n\t\t\t\t},\n\t\t\t\tMemo: &types.Memo{\n\t\t\t\t\tFields: map[string][]byte{\n\t\t\t\t\t\t\"key1\": []byte(\"value1\"),\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tPartitionConfig: map[string]string{\n\t\t\t\t\t\"config1\": \"value1\",\n\t\t\t\t},\n\t\t\t\tCronOverlapPolicy: types.CronOverlapPolicySkipped.Ptr(),\n\t\t\t\tActiveClusterSelectionPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\t\tScope: \"region\",\n\t\t\t\t\t\tName:  \"us-west-1\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.StartTime)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.CloseTime)\n\t\tassert.Equal(t, types.WorkflowExecutionCloseStatusCompleted, *resp.WorkflowExecutionInfo.CloseStatus)\n\t\tassert.Equal(t, int64(50), resp.WorkflowExecutionInfo.HistoryLength)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.ParentDomainID)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.ParentExecution)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.Memo)\n\t\tassert.Equal(t, map[string]string{\"config1\": \"value1\"}, resp.WorkflowExecutionInfo.PartitionConfig)\n\t\tassert.Equal(t, types.CronOverlapPolicySkipped, *resp.WorkflowExecutionInfo.CronOverlapPolicy)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy)\n\t\tassert.NotNil(t, resp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy.ClusterAttribute)\n\t\tassert.Equal(t, \"region\", resp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy.ClusterAttribute.Scope)\n\t\tassert.Equal(t, \"us-west-1\", resp.WorkflowExecutionInfo.ActiveClusterSelectionPolicy.ClusterAttribute.Name)\n\t})\n\n\tt.Run(\"with no pending activities or decision\", func(t *testing.T) {\n\t\tmockResp := &types.DescribeWorkflowExecutionResponse{\n\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"wf-id\",\n\t\t\t\t\tRunID:      \"run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\n\t\tresp, err := convertDescribeWorkflowExecutionResponse(mockResp, serverFrontendClient, nil)\n\t\tassert.NoError(t, err)\n\t\tassert.Nil(t, resp.PendingDecision)\n\t\tassert.Equal(t, 0, len(resp.PendingActivities))\n\t})\n}\n\nfunc Test_PrintRunStatus(t *testing.T) {\n\t// this method only prints results, no need to test the output\n\ttests := []struct {\n\t\tname  string\n\t\tevent *types.HistoryEvent\n\t}{\n\t\t{\n\t\t\tname: \"COMPLETED\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionCompleted.Ptr(),\n\t\t\t\tWorkflowExecutionCompletedEventAttributes: &types.WorkflowExecutionCompletedEventAttributes{\n\t\t\t\t\tResult: []byte(\"workflow completed successfully\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"FAILED\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionFailed.Ptr(),\n\t\t\t\tWorkflowExecutionFailedEventAttributes: &types.WorkflowExecutionFailedEventAttributes{\n\t\t\t\t\tReason:  nil,\n\t\t\t\t\tDetails: []byte(\"failure details\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"TIMEOUT\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionTimedOut.Ptr(),\n\t\t\t\tWorkflowExecutionTimedOutEventAttributes: &types.WorkflowExecutionTimedOutEventAttributes{\n\t\t\t\t\tTimeoutType: types.TimeoutTypeStartToClose.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname: \"CANCELED\",\n\t\t\tevent: &types.HistoryEvent{\n\t\t\t\tEventType: types.EventTypeWorkflowExecutionCanceled.Ptr(),\n\t\t\t\tWorkflowExecutionCanceledEventAttributes: &types.WorkflowExecutionCanceledEventAttributes{\n\t\t\t\t\tDetails: []byte(\"canceled details\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tassert.NotPanics(t, func() {\n\t\t\t\tprintRunStatus(tt.event)\n\t\t\t})\n\t\t})\n\t}\n}\n\nfunc Test_ListWorkflowExecutions(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\tc := getMockContext(t, nil, app)\n\tlistFn := listWorkflowExecutions(serverFrontendClient, 100, \"test-domain\", \"WorkflowType='test-workflow-type'\", c)\n\tassert.NotNil(t, listFn)\n\texpectedResp := &types.ListWorkflowExecutionsResponse{\n\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t\tNextPageToken: []byte(\"test-next-page-token\"),\n\t}\n\tserverFrontendClient.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(expectedResp, nil).Times(1)\n\texecutions, nextPageToken, err := listFn(nil)\n\tassert.NoError(t, err)\n\tassert.NotNil(t, executions)\n\tassert.Equal(t, expectedResp.Executions, executions)\n\tassert.NotNil(t, nextPageToken)\n\n\tserverFrontendClient.EXPECT().ListWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\t_, _, err = listFn(nil)\n\tassert.Error(t, err)\n}\n\nfunc Test_PrintListResults(t *testing.T) {\n\texecutions := []*types.WorkflowExecutionInfo{\n\t\t{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id-1\",\n\t\t\t\tRunID:      \"test-run-id-1\",\n\t\t\t},\n\t\t\tType: &types.WorkflowType{\n\t\t\t\tName: \"test-workflow-type-1\",\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id-2\",\n\t\t\t\tRunID:      \"test-run-id-2\",\n\t\t\t},\n\t\t\tType: &types.WorkflowType{\n\t\t\t\tName: \"test-workflow-type-2\",\n\t\t\t},\n\t\t},\n\t}\n\n\tassert.NotPanics(t, func() {\n\t\tprintListResults(executions, true, false)\n\t\tprintListResults(executions, false, false)\n\t\tprintListResults(executions, true, true)\n\t\tprintListResults(nil, true, false)\n\t})\n}\n\nfunc Test_ResetWorkflow(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\t// missing domain flag\n\terr := ResetWorkflow(c)\n\tassert.Error(t, err)\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\t// missing workflowID flag\n\terr = ResetWorkflow(c)\n\tassert.Error(t, err)\n\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\t// missing reason flag\n\terr = ResetWorkflow(c)\n\tassert.Error(t, err)\n\n\tset.String(\"reason\", \"test\", \"reason\")\n\tset.String(\"decision_offset\", \"-1\", \"decision_offset\")\n\t// invalid event ID\n\terr = ResetWorkflow(c)\n\tassert.Error(t, err)\n\n\tset.String(\"reset_type\", \"LastDecisionCompleted\", \"reset_type\")\n\tset.String(\"run_id\", \"test-run-id\", \"run_id\")\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}, nil).AnyTimes()\n\terr = ResetWorkflow(c)\n\tassert.NoError(t, err)\n\n\t// reset failed\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\terr = ResetWorkflow(c)\n\tassert.Error(t, err)\n\n\t// getResetEventIDByType failed\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\terr = ResetWorkflow(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetWorkflow_Invalid_Decision_Offset(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\tset.String(\"reason\", \"test\", \"reason\")\n\tset.String(\"decision_offset\", \"100\", \"decision_offset\")\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetWorkflow(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetWorkflow_Missing_RunID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.String(\"reason\", \"test\", \"reason\")\n\tset.String(\"decision_offset\", \"-1\", \"decision_offset\")\n\tset.String(\"reset_type\", \"BadBinary\", \"reset_type\")\n\tset.String(\"reset_bad_binary_checksum\", \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}, errors.New(\"test-error\")).AnyTimes()\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetWorkflow(c)\n\tassert.Error(t, err)\n}\n\nfunc (s *cliAppSuite) TestCompleteActivity() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf activity complete -w wid -r rid -aid 3 -result result --identity tester`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RespondActivityTaskCompletedByID(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestDescribeWorkflow() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\t\"happy\",\n\t\t\t\"cadence --do test-domain wf describe -w wid\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\t\t\tName: \"workflow-type\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\tCloseTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{},\n\t\t\t\t\t},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{},\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\t\"with strong consistency level\",\n\t\t\t\"cadence --do test-domain wf describe -w wid -qcl strong\",\n\t\t\t\"\",\n\t\t\tfunc() {\n\t\t\t\treq := &types.DescribeWorkflowExecutionRequest{\n\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t},\n\t\t\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t\t\t}\n\n\t\t\t\tresp := &types.DescribeWorkflowExecutionResponse{\n\t\t\t\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tType: &types.WorkflowType{\n\t\t\t\t\t\t\tName: \"workflow-type\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tStartTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t\tCloseTime: common.Int64Ptr(time.Now().UnixNano()),\n\t\t\t\t\t},\n\t\t\t\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t\t\t\t{},\n\t\t\t\t\t},\n\t\t\t\t\tPendingDecision: &types.PendingDecisionInfo{},\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), req).Return(resp, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestFailActivity() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf activity fail -w wid -r rid -aid 3 --reason somereason --detail somedetail --identity tester`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().RespondActivityTaskFailedByID(gomock.Any(), gomock.Any()).Return(nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestListAllWorkflow() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf listall`,\n\t\t\tmock: func() {\n\t\t\t\tcountWorkflowResp := &types.CountWorkflowExecutionsResponse{}\n\t\t\t\ts.serverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(countWorkflowResp, nil)\n\t\t\t\ts.serverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(listClosedWorkflowExecutionsResponse, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestQueryWorkflow() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryResult: []byte(\"query-result\"),\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"query with reject not_open\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_reject_condition not_open`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryResult: []byte(\"query-result\"),\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().\n\t\t\t\t\tQueryWorkflow(gomock.Any(), &types.QueryWorkflowRequest{\n\t\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\tQueryType: \"query-type-test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQueryRejectCondition: types.QueryRejectConditionNotOpen.Ptr(),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"query with reject not_completed_cleanly\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_reject_condition not_completed_cleanly`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryResult: []byte(\"query-result\"),\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().\n\t\t\t\t\tQueryWorkflow(gomock.Any(), &types.QueryWorkflowRequest{\n\t\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\tQueryType: \"query-type-test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQueryRejectCondition: types.QueryRejectConditionNotCompletedCleanly.Ptr(),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"query with unknown reject\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_reject_condition unknown`,\n\t\t\terr:     \"invalid reject condition\",\n\t\t},\n\t\t{\n\t\t\tname:    \"query with eventual consistency\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_consistency_level eventual`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryResult: []byte(\"query-result\"),\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().\n\t\t\t\t\tQueryWorkflow(gomock.Any(), &types.QueryWorkflowRequest{\n\t\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\tQueryType: \"query-type-test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelEventual.Ptr(),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"query with strong consistency\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_consistency_level strong`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.QueryWorkflowResponse{\n\t\t\t\t\tQueryResult: []byte(\"query-result\"),\n\t\t\t\t}\n\t\t\t\ts.serverFrontendClient.EXPECT().\n\t\t\t\t\tQueryWorkflow(gomock.Any(), &types.QueryWorkflowRequest{\n\t\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\t\tWorkflowID: \"wid\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQuery: &types.WorkflowQuery{\n\t\t\t\t\t\t\tQueryType: \"query-type-test\",\n\t\t\t\t\t\t},\n\t\t\t\t\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t\t\t\t\t}).\n\t\t\t\t\tReturn(resp, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"query with invalid consistency\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test --query_consistency_level invalid`,\n\t\t\terr:     \"invalid query consistency level\",\n\t\t},\n\t\t{\n\t\t\tname:    \"failed\",\n\t\t\tcommand: `cadence --do test-domain wf query -w wid -qt query-type-test`,\n\t\t\terr:     \"some error\",\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().QueryWorkflow(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"missing flags\",\n\t\t\tcommand: \"cadence wf query\",\n\t\t\terr:     \"Required flag not found\",\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestResetWorkflow() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf reset -w wid -r rid -reason test-reason --event_id 1`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\tresp := &types.ResetWorkflowExecutionResponse{RunID: uuid.New()}\n\t\t\t\ts.serverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(resp, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestScanAllWorkflow() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf scanall`,\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{}, nil)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc (s *cliAppSuite) TestSignalWithStartWorkflowExecution() {\n\ttestCases := []testcase{\n\t\t{\n\t\t\tname:    \"happy\",\n\t\t\tcommand: `cadence --do test-domain wf signalwithstart --et 100 --workflow_type sometype --tasklist tasklist -w wid -n signal-name --signal_input []`,\n\t\t\terr:     \"\",\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.StartWorkflowExecutionResponse{}, nil)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"failed\",\n\t\t\tcommand: `cadence --do test-domain wf signalwithstart --et 100 --workflow_type sometype --tasklist tasklist -w wid -n signal-name --signal_input []`,\n\t\t\terr:     \"some error\",\n\t\t\tmock: func() {\n\t\t\t\ts.serverFrontendClient.EXPECT().SignalWithStartWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"some error\"))\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:    \"missing flags\",\n\t\t\tcommand: \"cadence wf signalwithstart\",\n\t\t\terr:     \"Required flag not found\",\n\t\t},\n\t}\n\n\tfor _, tt := range testCases {\n\t\ts.Run(tt.name, func() {\n\t\t\ts.runTestCase(tt)\n\t\t})\n\t}\n}\n\nfunc Test_DoReset(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\tset.String(\"reason\", \"test\", \"reason\")\n\tset.String(\"decision_offset\", \"-1\", \"decision_offset\")\n\tset.String(\"reset_type\", \"LastDecisionCompleted\", \"reset_type\")\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}, errors.New(\"test-error\")).Times(1)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\n\tc := cli.NewContext(app, set, nil)\n\tparams := &batchResetParamsType{\n\t\treason:    \"test\",\n\t\tresetType: \"LastDecisionCompleted\",\n\t}\n\t// describe workflow execution failed\n\terr := doReset(c, \"test-domain\", \"test-workflow-id\", \"test-run-id\", *params)\n\tassert.Error(t, err)\n\n\t// get reset event id failure\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"test-run-id\", *params)\n\tassert.Error(t, err)\n\n\t// reset failure\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"\", *params)\n\tassert.Error(t, err)\n\n\t// normal case\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"\", *params)\n\tassert.NoError(t, err)\n\n\t// dry run\n\tparams.dryRun = true\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"\", *params)\n\tassert.NoError(t, err)\n\n\t// skip current open\n\tparams.skipCurrentOpen = true\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"\", *params)\n\tassert.NoError(t, err)\n\n\t// current run id not match with input rid\n\tparams.skipBaseNotCurrent = true\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"test-not-matched-rid\", *params)\n\tassert.NoError(t, err)\n\n\t// dry run\n\tparams.dryRun = true\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"\", *params)\n\tassert.NoError(t, err)\n}\n\nfunc Test_DoReset_SkipCurrentCompleted(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverAdminClient := admin.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t\tserverAdminClient:    serverAdminClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"workflow_id\", \"test-workflow-id\", \"workflow_id\")\n\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\tset.String(\"reason\", \"test\", \"reason\")\n\tset.String(\"decision_offset\", \"-1\", \"decision_offset\")\n\tset.String(\"reset_type\", \"LastDecisionCompleted\", \"reset_type\")\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tCloseStatus: types.WorkflowExecutionCloseStatusCompleted.Ptr(),\n\t\t\tCloseTime:   common.Int64Ptr(time.Now().UnixNano()),\n\t\t},\n\t}, nil).AnyTimes()\n\n\tc := cli.NewContext(app, set, nil)\n\tparams := &batchResetParamsType{\n\t\treason:               \"test\",\n\t\tresetType:            \"LastDecisionCompleted\",\n\t\tnonDeterministicOnly: true,\n\t}\n\t// check non determinism failed when get history\n\terr := doReset(c, \"test-domain\", \"test-workflow-id\", \"test-run-id\", *params)\n\tassert.Error(t, err)\n\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"test-run-id\", *params)\n\tassert.NoError(t, err)\n\n\t// describe workflow execution failed\n\tparams.skipCurrentCompleted = true\n\terr = doReset(c, \"test-domain\", \"test-workflow-id\", \"test-run-id\", *params)\n\tassert.NoError(t, err)\n}\n\nfunc createTempFileWithContent(t *testing.T, content string) string {\n\ttmpFile, err := os.CreateTemp(\"\", \"testfile\")\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to create temporary file: %v\", err)\n\t}\n\n\t_, err = tmpFile.Write([]byte(content))\n\tif err != nil {\n\t\tt.Fatalf(\"Failed to write to temporary file: %v\", err)\n\t}\n\n\ttmpFileName := tmpFile.Name()\n\ttmpFile.Close()\n\n\tt.Cleanup(func() {\n\t\t_ = os.Remove(tmpFileName)\n\t})\n\n\treturn tmpFileName\n}\n\nfunc TestLoadWorkflowIDsFromFile_Success(t *testing.T) {\n\tcontent := \"wid1,wid2,wid3\\n\\nwid4,wid5\\nwid6\\n\"\n\tfileName := createTempFileWithContent(t, content)\n\n\tworkflowIDs, err := loadWorkflowIDsFromFile(fileName, \",\")\n\tassert.NoError(t, err)\n\n\texpected := map[string]bool{\n\t\t\"wid1\": true,\n\t\t\"wid4\": true,\n\t\t\"wid6\": true,\n\t}\n\tassert.Equal(t, expected, workflowIDs)\n}\n\nfunc TestLoadWorkflowIDsFromFile_Failure(t *testing.T) {\n\t// open failed\n\t_, err := loadWorkflowIDsFromFile(\"non exist file\", \",\")\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetInBatch_WithQuery(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\t// missing domain flag\n\terr := ResetInBatch(c)\n\tassert.Error(t, err)\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"reset_type\", \"BadBinary\", \"reset_type\")\n\tset.String(\"reset_bad_binary_checksum\", \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tset.String(FlagParallismDeprecated, \"1\", \"input parallism\")\n\tset.String(FlagParallelism, \"2\", \"parallelism\")\n\tset.String(FlagExcludeWorkflowIDByQuery, \"test-workflow-id\", \"exclude query\")\n\n\t// missing reason\n\terr = ResetInBatch(c)\n\tassert.Error(t, err)\n\n\tset.String(FlagReason, \"test\", \"reason\")\n\t// missing query\n\terr = ResetInBatch(c)\n\tassert.Error(t, err)\n\n\tset.String(FlagListQuery, \"WorkflowType='test-workflow-type'\", \"list query\")\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{\n\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\terr = ResetInBatch(c)\n\tassert.NoError(t, err)\n}\n\nfunc Test_ResetInBatch_WithFile(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"reset_type\", \"BadBinary\", \"reset_type\")\n\tset.String(\"reset_bad_binary_checksum\", \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tcontent := \"wid1,wid2,wid3\\n\\nwid4,wid5\\nwid6\\n\"\n\tfileName := createTempFileWithContent(t, content)\n\n\tset.String(FlagInputFile, fileName, \"input file\")\n\tset.String(FlagParallismDeprecated, \"1\", \"input parallism\")\n\tset.String(FlagParallelism, \"2\", \"parallelism\")\n\tset.String(FlagExcludeFile, fileName, \"exclude query\")\n\tset.String(FlagReason, \"test\", \"reason\")\n\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(&types.ListWorkflowExecutionsResponse{\n\t\tExecutions: []*types.WorkflowExecutionInfo{\n\t\t\t{\n\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetInBatch(c)\n\tassert.NoError(t, err)\n\n\t// error when both exclude query and file are provided\n\tset.String(FlagExcludeWorkflowIDByQuery, \"test-workflow-id\", \"exclude query\")\n\terr = ResetInBatch(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetInBatch_InvalidDescisionOffset(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(FlagDecisionOffset, \"100\", \"decision_offset\")\n\tset.String(FlagResetType, \"BadBinary\", \"reset_type\")\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetInBatch(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetInBatch_InvalidResetType(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(FlagDecisionOffset, \"-1\", \"decision_offset\")\n\tset.String(FlagResetType, \"test\", \"reset_type\")\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetInBatch(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetInBatch_InvalidInputFile(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"reset_type\", \"BadBinary\", \"reset_type\")\n\tset.String(\"reset_bad_binary_checksum\", \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tset.String(FlagInputFile, \"non exist file\", \"input file\")\n\tset.String(FlagParallismDeprecated, \"1\", \"input parallism\")\n\tset.String(FlagParallelism, \"2\", \"parallelism\")\n\tset.String(FlagReason, \"test\", \"reason\")\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetInBatch(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_ResetInBatch_InvalidexcludeFile(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\tset.String(\"reset_type\", \"BadBinary\", \"reset_type\")\n\tset.String(\"reset_bad_binary_checksum\", \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tset.String(FlagExcludeFile, \"non exist file\", \"exclude query\")\n\tset.String(FlagParallismDeprecated, \"1\", \"input parallism\")\n\tset.String(FlagParallelism, \"2\", \"parallelism\")\n\tset.String(FlagReason, \"test\", \"reason\")\n\tset.String(FlagListQuery, \"WorkflowType='test-workflow-type'\", \"list query\")\n\tc := cli.NewContext(app, set, nil)\n\terr := ResetInBatch(c)\n\tassert.Error(t, err)\n}\n\nfunc Test_GetFirstDecisionTaskByType(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\t// get desision type successfully\n\tdecisionFinishID, err := getFirstDecisionTaskByType(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient, 6)\n\tassert.Equal(t, int64(1), decisionFinishID)\n\tassert.NoError(t, err)\n\n\t// failed to get workflow history\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(nil, errors.New(\"test-error\")).Times(1)\n\tdecisionFinishID, err = getFirstDecisionTaskByType(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient, 6)\n\tassert.Equal(t, int64(0), decisionFinishID)\n\tassert.Error(t, err)\n\n\t// not found decision task\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\tdecisionFinishID, err = getFirstDecisionTaskByType(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient, 6)\n\tassert.Equal(t, int64(0), decisionFinishID)\n\tassert.Error(t, err)\n\n\t// next page token\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tNextPageToken: []byte(\"test-next-page-token\"),\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   []byte(\"test-next-page-token\"),\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{}, nil).Times(1)\n\tdecisionFinishID, err = getFirstDecisionTaskByType(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient, 6)\n\tassert.Equal(t, int64(0), decisionFinishID)\n\tassert.Error(t, err)\n}\n\nfunc Test_GetCurrentRunID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t},\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tRunID: \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\n\trunID, err := getCurrentRunID(context.Background(), \"test-domain\", \"test-workflow-id\", serverFrontendClient)\n\tassert.NoError(t, err)\n\tassert.Equal(t, \"test-run-id\", runID)\n}\n\nfunc Test_GetBadDecisionCompletedID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}).Return(&types.DescribeWorkflowExecutionResponse{}, errors.New(\"test-error\")).Times(1)\n\n\tdecision, err := getBadDecisionCompletedID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", \"\", serverFrontendClient)\n\tassert.Equal(t, int64(0), decision)\n\tassert.Error(t, err)\n\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tAutoResetPoints: &types.ResetPoints{\n\t\t\t\tPoints: []*types.ResetPointInfo{\n\t\t\t\t\t{\n\t\t\t\t\t\tBinaryChecksum:           \"test-bad-binary-checksum\",\n\t\t\t\t\t\tFirstDecisionCompletedID: 5,\n\t\t\t\t\t\tResettable:               true,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\tdecision, err = getBadDecisionCompletedID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", \"\", serverFrontendClient)\n\tassert.Equal(t, int64(0), decision)\n\tassert.Error(t, err)\n\n\tdecision, err = getBadDecisionCompletedID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", \"test-bad-binary-checksum\", serverFrontendClient)\n\tassert.Equal(t, int64(5), decision)\n\tassert.NoError(t, err)\n}\n\nfunc Test_GetLastContinueAsNewID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1,\n\t}).Return(nil, errors.New(\"test-error\")).Times(1)\n\t// get workflow history failed\n\trunID, decisionID, err := getLastContinueAsNewID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\t// cannot get reset base runID\n\trunID, decisionID, err = getLastContinueAsNewID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tID:        10,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\trunID, decisionID, err = getLastContinueAsNewID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(10), decisionID)\n\tassert.NoError(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(nil, errors.New(\"test-error\")).Times(1)\n\t// fail to get workflow history after getting the reset base runID\n\trunID, decisionID, err = getLastContinueAsNewID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{},\n\t\t},\n\t\tNextPageToken: []byte(\"test-next-page-token\"),\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{},\n\t\t},\n\t}, nil).Times(1)\n\t// fail to get workflow history after getting the reset base runID\n\trunID, decisionID, err = getLastContinueAsNewID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n}\n\nfunc Test_GetEarliestDecisionID(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(nil, errors.New(\"test-error\")).Times(1)\n\n\t// get workflow execution history failed\n\tdecisionID, err := getEarliestDecisionID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", int64(100), serverFrontendClient)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{},\n\t\t},\n\t\tNextPageToken: []byte(\"test-next-page-token\"),\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t\tNextPageToken:   []byte(\"test-next-page-token\"),\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{},\n\t\t},\n\t}, nil).Times(1)\n\t// no DecisionFinishID\n\tdecisionID, err = getEarliestDecisionID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", int64(100), serverFrontendClient)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.Error(t, err)\n\n\ttestEarliestTime := int64(110)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tMaximumPageSize: 1000,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tID:        10,\n\t\t\t\t\tTimestamp: &testEarliestTime,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\t// no DecisionFinishID\n\tdecisionID, err = getEarliestDecisionID(context.Background(), \"test-domain\", \"test-workflow-id\", \"test-run-id\", int64(100), serverFrontendClient)\n\tassert.Equal(t, int64(10), decisionID)\n\tassert.NoError(t, err)\n}\n\nfunc Test_FailActivity_CompleteActivity_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\t// call fail activity and complete activity without required flags, they should return the same error\n\terr := FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\tset.String(FlagWorkflowID, \"test-workflow-id\", \"workflow_id\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagRunID))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagRunID))\n\tset.String(FlagRunID, \"test-run-id\", \"run_id\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagActivityID))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagActivityID))\n\tset.String(FlagActivityID, \"test-activity-id\", \"activity_id\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagReason))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagResult))\n\tset.String(FlagReason, \"test\", \"reason\")\n\tset.String(FlagResult, \"test\", \"result\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDetail))\n\tset.String(FlagDetail, \"test\", \"detail\")\n\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagIdentity))\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagIdentity))\n\tset.String(FlagIdentity, \"test\", \"indentity\")\n\n\tserverFrontendClient.EXPECT().RespondActivityTaskFailedByID(gomock.Any(), &types.RespondActivityTaskFailedByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t\tActivityID: \"test-activity-id\",\n\t\tReason:     common.StringPtr(\"test\"),\n\t\tDetails:    []byte(\"test\"),\n\t\tIdentity:   \"test\",\n\t}).Return(errors.New(\"test-error\")).Times(1)\n\terr = FailActivity(c)\n\tassert.ErrorContains(t, err, \"Failing activity failed\")\n\n\tserverFrontendClient.EXPECT().RespondActivityTaskCompletedByID(gomock.Any(), &types.RespondActivityTaskCompletedByIDRequest{\n\t\tDomain:     \"test-domain\",\n\t\tWorkflowID: \"test-workflow-id\",\n\t\tRunID:      \"test-run-id\",\n\t\tActivityID: \"test-activity-id\",\n\t\tResult:     []byte(\"test\"),\n\t\tIdentity:   \"test\",\n\t}).Return(errors.New(\"test-error\")).Times(1)\n\terr = CompleteActivity(c)\n\tassert.ErrorContains(t, err, \"Completing activity failed\")\n}\n\nfunc Test_GetResetEventIDByType_LastContinuedAsNew(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tc := getMockContext(t, nil, app)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        15,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\t// reset type last continued as new\n\trunID, decisionID, err := getResetEventIDByType(context.Background(), c, resetTypeLastContinuedAsNew, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(15), decisionID)\n\tassert.NoError(t, err)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tChildWorkflowExecutionFailedEventAttributes: &types.ChildWorkflowExecutionFailedEventAttributes{},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\t_, _, err = getResetEventIDByType(context.Background(), c, resetTypeLastContinuedAsNew, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.ErrorContains(t, err, \"cannot get resetBaseRunID\")\n}\n\nfunc Test_GetResetEventIDByType_FirstDecisionCompleted(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tc := getMockContext(t, nil, app)\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        15,\n\t\t\t\t\tEventType: types.EventTypeActivityTaskCompleted.Ptr().Ptr(),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\n\t// reset type last decision completed\n\trunID, decisionID, err := getResetEventIDByType(context.Background(), c, resetTypeFirstDecisionCompleted, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.ErrorContains(t, err, \"no DecisionFinishID\")\n}\n\nfunc Test_GetResetEventIDByType_BadBinary(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagResetBadBinaryChecksum, \"test-bad-binary-checksum\", \"reset_bad_binary_checksum\")\n\tc := getMockContext(t, set, app)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{},\n\t}, nil).Times(1)\n\trunID, decisionID, err := getResetEventIDByType(context.Background(), c, resetTypeBadBinary, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.ErrorContains(t, err, \"no DecisionFinishID\")\n}\n\nfunc Test_GetResetEventIDByType_DecisionCompletedTime(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := getMockContext(t, set, app)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 15,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\t_, decisionID, err := getResetEventIDByType(context.Background(), c, resetTypeDecisionCompletedTime, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.ErrorContains(t, err, \"no DecisionFinishID\")\n\n\tset.String(FlagEarliestTime, \"20201025Test\", \"earliest_time\")\n\t_, _, err = getResetEventIDByType(context.Background(), c, resetTypeDecisionCompletedTime, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.ErrorContains(t, err, \"use UTC format\")\n}\n\nfunc Test_GetResetEventIDByType_FirstDecisionScheduled_LastDecisionScheduled(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tc := getMockContext(t, nil, app)\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID: 15,\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\t// reset type first decision scheduled\n\trunID, decisionID, err := getResetEventIDByType(context.Background(), c, resetTypeFirstDecisionScheduled, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.ErrorContains(t, err, \"no DecisionFinishID\")\n\t// reset type Last decision scheduled\n\trunID, decisionID, err = getResetEventIDByType(context.Background(), c, resetTypeLastDecisionScheduled, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(0), decisionID)\n\tassert.ErrorContains(t, err, \"no DecisionFinishID\")\n\t// unsupported reset type\n\tassert.Panics(t, func() {\n\t\tgetResetEventIDByType(context.Background(), c, \"test reset type\", -1, \"test-domain\", \"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\t})\n\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        15,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskScheduled.Ptr(),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(2)\n\t_, decisionID, err = getResetEventIDByType(context.Background(), c, resetTypeFirstDecisionScheduled, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(16), decisionID)\n\tassert.NoError(t, err)\n\n\t_, decisionID, err = getResetEventIDByType(context.Background(), c, resetTypeLastDecisionScheduled, -1, \"test-domain\",\n\t\t\"test-workflow-id\", \"test-run-id\", serverFrontendClient)\n\tassert.Equal(t, \"test-run-id\", runID)\n\tassert.Equal(t, int64(16), decisionID)\n\tassert.NoError(t, err)\n}\n\nfunc Test_SignalWorkflow_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := SignalWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = SignalWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\n\tset.String(FlagWorkflowID, \"test-workflow-id\", \"workflow_id\")\n\terr = SignalWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagName))\n}\n\nfunc Test_RestartWorkflow_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := RestartWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = RestartWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_DiagnoseWorkflow_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := DiagnoseWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = DiagnoseWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\n\tset.String(FlagWorkflowID, \"test-workflow-id\", \"workflow_id\")\n\terr = DiagnoseWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagRunID))\n}\n\nfunc Test_TerminateWorkflow_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := TerminateWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = TerminateWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_ShowHistory_MissingWorkflowID(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := ShowHistory(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_ShowHistoryWithID_MissingWorkflowID(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := ShowHistoryWithWID(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\n\tset.Parse([]string{\"test-workflow-id\", \"test-run-id\"})\n\terr = ShowHistoryWithWID(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n}\n\nfunc Test_ConstructStartWorkflowRequest(t *testing.T) {\n\tbaseArgs := []clitest.CliArgument{\n\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagTaskList, \"test-tasklist\"),\n\t\tclitest.StringArgument(FlagWorkflowType, \"test-workflow-type\"),\n\t}\n\n\ttests := []struct {\n\t\tname                 string\n\t\tcliArguments         []clitest.CliArgument\n\t\texpectedError        bool\n\t\texpectedErrorMessage string\n\t}{\n\t\t{\n\t\t\tname:                 \"missing domain\",\n\t\t\tcliArguments:         []clitest.CliArgument{},\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: fmt.Sprintf(\"%s is required\", FlagDomain),\n\t\t},\n\t\t{\n\t\t\tname: \"missing task list\",\n\t\t\tcliArguments: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\t\t},\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: fmt.Sprintf(\"%s is required\", FlagTaskList),\n\t\t},\n\t\t{\n\t\t\tname: \"missing workflow type\",\n\t\t\tcliArguments: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\t\t\tclitest.StringArgument(FlagTaskList, \"test-tasklist\"),\n\t\t\t},\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: fmt.Sprintf(\"%s is required\", FlagWorkflowType),\n\t\t},\n\t\t{\n\t\t\tname:                 \"invalid execution timeout\",\n\t\t\tcliArguments:         baseArgs,\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: fmt.Sprintf(\"%s format is invalid\", FlagExecutionTimeout),\n\t\t},\n\t\t{\n\t\t\tname: \"invalid workflowID reuse policy\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, -10),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"value is not in supported range\",\n\t\t},\n\t\t{\n\t\t\tname: \"process JSON error\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagInput, \"invalid json\"),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"input is not valid JSON\",\n\t\t},\n\t\t{\n\t\t\tname: \"error processing first run at\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagCronSchedule, \"* * * * *\"),\n\t\t\t\tclitest.StringArgument(FirstRunAtTime, \"10:00\"),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"time format invalid\",\n\t\t},\n\t\t{\n\t\t\tname: \"error processing header\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagCronSchedule, \"* * * * *\"),\n\t\t\t\tclitest.StringArgument(FlagHeaderFile, \"invalid file\"),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"error when process header\",\n\t\t},\n\t\t{\n\t\t\tname: \"error processing memo\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagCronSchedule, \"* * * * *\"),\n\t\t\t\tclitest.StringArgument(FlagSearchAttributesKey, \"key\"),\n\t\t\t\tclitest.StringArgument(FlagMemoFile, \"invalid file\"),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"error processing memo\",\n\t\t},\n\t\t{\n\t\t\tname: \"error processing search attributes\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagCronSchedule, \"* * * * *\"),\n\t\t\t\tclitest.StringArgument(FlagSearchAttributesKey, \"key\"),\n\t\t\t),\n\t\t\texpectedError:        true,\n\t\t\texpectedErrorMessage: \"error processing search attributes\",\n\t\t},\n\t\t{\n\t\t\tname: \"no error\",\n\t\t\tcliArguments: append(\n\t\t\t\tbaseArgs,\n\t\t\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\t\t\tclitest.IntArgument(FlagWorkflowIDReusePolicy, 1),\n\t\t\t\tclitest.StringArgument(FlagCronSchedule, \"* * * * *\"),\n\t\t\t\tclitest.StringArgument(FlagSearchAttributesKey, \"key\"),\n\t\t\t\tclitest.StringArgument(FlagSearchAttributesVal, \"val\"),\n\t\t\t),\n\t\t\texpectedError:        false,\n\t\t\texpectedErrorMessage: \"\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tapp := NewCliApp(&clientFactoryMock{})\n\t\t\tctx := clitest.NewCLIContext(t, app, tc.cliArguments...)\n\t\t\t_, err := constructStartWorkflowRequest(ctx)\n\n\t\t\tif tc.expectedError {\n\t\t\t\tassert.ErrorContains(t, err, tc.expectedErrorMessage)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_StartWorkflowHelper_RetryErrorMapping(t *testing.T) {\n\trequiredArguments := []clitest.CliArgument{\n\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagTaskList, \"test-tasklist\"),\n\t\tclitest.StringArgument(FlagWorkflowType, \"test-workflow-type\"),\n\t\tclitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t}\n\ttests := []struct {\n\t\tcliArguments          []clitest.CliArgument\n\t\trespondedErrorMessage string\n\t\texpectedErrorMessage  string\n\t}{\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 1), clitest.IntArgument(FlagRetryInterval, -1)),\n\t\t\t\"InitialIntervalInSeconds must be greater than 0 on retry policy.\",\n\t\t\t\"retry_interval must be greater than 0 on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 1), clitest.IntArgument(FlagRetryBackoff, -1)),\n\t\t\t\"BackoffCoefficient cannot be less than 1 on retry policy.\",\n\t\t\t\"retry_backoff cannot be less than 1 on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 1), clitest.IntArgument(FlagRetryMaxInterval, -1)),\n\t\t\t\"MaximumIntervalInSeconds cannot be less than 0 on retry policy.\",\n\t\t\t\"retry_max_interval cannot be less than 0 on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 1), clitest.IntArgument(FlagRetryInterval, 2), clitest.IntArgument(FlagRetryMaxInterval, -1)),\n\t\t\t\"MaximumIntervalInSeconds cannot be less than InitialIntervalInSeconds on retry policy.\",\n\t\t\t\"retry_max_interval cannot be less than retry_interval on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, -1)),\n\t\t\t\"MaximumAttempts cannot be less than 0 on retry policy.\",\n\t\t\t\"retry_attempts cannot be less than 0 on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 1), clitest.IntArgument(FlagRetryExpiration, -1)),\n\t\t\t\"ExpirationIntervalInSeconds cannot be less than 0 on retry policy.\",\n\t\t\t\"retry_expiration cannot be less than 0 on retry policy.\",\n\t\t},\n\t\t{\n\t\t\tappend(requiredArguments, clitest.IntArgument(FlagRetryAttempts, 0), clitest.IntArgument(FlagRetryBackoff, 0)),\n\t\t\t\"MaximumAttempts and ExpirationIntervalInSeconds are both 0. At least one of them must be specified.\",\n\t\t\t\"retry_attempts and retry_expiration are both 0. At least one of them must be specified.\",\n\t\t},\n\t}\n\n\tfor _, tc := range tests {\n\t\tt.Run(tc.expectedErrorMessage, func(t *testing.T) {\n\t\t\tctrl := gomock.NewController(t)\n\n\t\t\tmockServiceClient := frontend.NewMockClient(ctrl)\n\t\t\tmockServiceClient.EXPECT().\n\t\t\t\tStartWorkflowExecution(gomock.Any(), gomock.Any()).\n\t\t\t\tReturn(nil, errors.New(tc.respondedErrorMessage))\n\n\t\t\tapp := NewCliApp(&clientFactoryMock{serverFrontendClient: mockServiceClient})\n\n\t\t\tctx := clitest.NewCLIContext(t, app, tc.cliArguments...)\n\t\t\terr := startWorkflowHelper(ctx, false)\n\t\t\tassert.ErrorContains(t, err, tc.expectedErrorMessage)\n\n\t\t})\n\t}\n}\n\nfunc Test_ProcessSearchAttr(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tset.String(FlagSearchAttributesKey, \"key\", \"search attribute key\")\n\tc := cli.NewContext(app, set, nil)\n\t_, err := processSearchAttr(c)\n\tassert.ErrorContains(t, err, \"keys and values are not equal\")\n\n\tset.String(FlagSearchAttributesVal, \"value\", \"search attribute value\")\n\tresp, err := processSearchAttr(c)\n\tassert.NoError(t, err)\n\texpectedVal, _ := json.Marshal(\"value\")\n\texpectedResp := map[string][]byte{\"key\": expectedVal}\n\tassert.Equal(t, expectedResp, resp)\n}\n\nfunc Test_CancelWorkflow_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := CancelWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = CancelWorkflow(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_QueryWorkflowHelper_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := queryWorkflowHelper(c, \"\")\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tset.String(FlagDomain, \"test-domain\", \"domain\")\n\terr = queryWorkflowHelper(c, \"\")\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n\n\tcontent := \"wid1,wid2,wid3\\n\\nwid4,wid5\\nwid6\\n\"\n\tfileName := createTempFileWithContent(t, content)\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"), clitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"),\n\t\tclitest.StringArgument(FlagInputFile, fileName))\n\terr = QueryWorkflowUsingQueryTypes(ctx)\n\tassert.ErrorContains(t, err, \"Error processing json\")\n}\n\nfunc Test_ProcessJsonInputHelper(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tcontent := \"wid1,wid2,wid3\\n\\nwid4,wid5\\nwid6\\n\"\n\tfileName := createTempFileWithContent(t, content)\n\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagInputFile, fileName))\n\t_, err := processJSONInputHelper(ctx, jsonTypeInput)\n\tassert.ErrorContains(t, err, \"input is not valid JSON\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagInputFile, \"non exist file\"))\n\t_, err = processJSONInputHelper(ctx, jsonTypeInput)\n\tassert.ErrorContains(t, err, \"error reading input file\")\n\n\tresp, err := processJSONInputHelper(ctx, -1)\n\tassert.Equal(t, \"\", resp)\n\tassert.NoError(t, err)\n}\n\nfunc Test_ConstructSignalWithStartWorkflowRequest_Errors(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"), clitest.StringArgument(FlagTaskList, \"test-tasklist\"),\n\t\tclitest.StringArgument(FlagWorkflowType, \"test-workflow-type\"), clitest.StringArgument(FlagExecutionTimeout, \"10\"))\n\n\t_, err := constructSignalWithStartWorkflowRequest(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagName))\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"), clitest.StringArgument(FlagTaskList, \"test-tasklist\"),\n\t\tclitest.StringArgument(FlagWorkflowType, \"test-workflow-type\"), clitest.StringArgument(FlagExecutionTimeout, \"10\"),\n\t\tclitest.StringArgument(FlagName, \"test-signal-name\"), clitest.StringArgument(FlagSignalInputFile, \"invalid json\"))\n\t_, err = constructSignalWithStartWorkflowRequest(ctx)\n\tassert.ErrorContains(t, err, \"error processing json input signal\")\n}\n\nfunc Test_ListWorkflow_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app)\n\terr := ListWorkflow(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tserverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"), clitest.StringArgument(FlagExcludeWorkflowIDByQuery, \"test-exclude\"),\n\t\tclitest.StringArgument(FlagListQuery, \"test-query\"))\n\terr = ListWorkflow(ctx)\n\tassert.ErrorContains(t, err, \"test-error\")\n}\n\nfunc Test_ListAllWorkflow_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app)\n\terr := ListAllWorkflow(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tserverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ListClosedWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ScanWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"), clitest.StringArgument(FlagExcludeWorkflowIDByQuery, \"test-exclude\"),\n\t\tclitest.StringArgument(FlagListQuery, \"test-query\"))\n\terr = ListAllWorkflow(ctx)\n\tassert.ErrorContains(t, err, \"test-error\")\n}\n\nfunc Test_CountWorkflow_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app)\n\terr := CountWorkflow(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"))\n\tserverFrontendClient.EXPECT().CountWorkflowExecutions(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).AnyTimes()\n\terr = CountWorkflow(ctx)\n\tassert.ErrorContains(t, err, \"test-error\")\n}\n\nfunc Test_DescribeWorkflow_Errors(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tctx := clitest.NewCLIContext(t, app)\n\terr := DescribeWorkflow(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_RefreshWorkflowTasks(t *testing.T) {\n\ttests := []struct {\n\t\tname        string\n\t\tsetupMock   func(*frontend.MockClient)\n\t\targs        []clitest.CliArgument\n\t\terrContains string\n\t}{\n\t\t{\n\t\t\tname:        \"missing domain\",\n\t\t\tsetupMock:   func(_ *frontend.MockClient) {},\n\t\t\targs:        []clitest.CliArgument{},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname:      \"missing workflowID\",\n\t\t\tsetupMock: func(_ *frontend.MockClient) {},\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\t\t},\n\t\t\terrContains: \"Required flag not found\",\n\t\t},\n\t\t{\n\t\t\tname: \"success\",\n\t\t\tsetupMock: func(client *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().RefreshWorkflowTasks(gomock.Any(), &types.RefreshWorkflowTasksRequest{\n\t\t\t\t\tDomain: \"test-domain\",\n\t\t\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t}).Return(nil)\n\t\t\t},\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"test-run-id\"),\n\t\t\t},\n\t\t\terrContains: \"\",\n\t\t},\n\t\t{\n\t\t\tname: \"api error\",\n\t\t\tsetupMock: func(client *frontend.MockClient) {\n\t\t\t\tclient.EXPECT().RefreshWorkflowTasks(gomock.Any(), gomock.Any()).\n\t\t\t\t\tReturn(errors.New(\"api error\"))\n\t\t\t},\n\t\t\targs: []clitest.CliArgument{\n\t\t\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\t\t\tclitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"),\n\t\t\t\tclitest.StringArgument(FlagRunID, \"test-run-id\"),\n\t\t\t},\n\t\t\terrContains: \"Refresh workflow tasks failed\",\n\t\t},\n\t}\n\n\tfor _, tt := range tests {\n\t\tt.Run(tt.name, func(t *testing.T) {\n\t\t\tmockCtrl := gomock.NewController(t)\n\t\t\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\t\t\tapp := NewCliApp(&clientFactoryMock{\n\t\t\t\tserverFrontendClient: serverFrontendClient,\n\t\t\t})\n\t\t\ttt.setupMock(serverFrontendClient)\n\t\t\tctx := clitest.NewCLIContext(t, app, tt.args...)\n\t\t\terr := RefreshWorkflowTasks(ctx)\n\t\t\tif tt.errContains == \"\" {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t} else {\n\t\t\t\tassert.ErrorContains(t, err, tt.errContains)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc Test_ObserveHistory_MissingFlags(t *testing.T) {\n\tapp := NewCliApp(&clientFactoryMock{})\n\tset := flag.NewFlagSet(\"test\", 0)\n\tc := cli.NewContext(app, set, nil)\n\terr := ObserveHistory(c)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagWorkflowID))\n}\n\nfunc Test_ShowHistoryHelper(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tWaitForNewEvent:        false,\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\tSkipArchival:           false,\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(5)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, &types.EntityNotExistsError{}).Times(1)\n\t// workflow not exists\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"true\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagOutputFilename, \"test-file\"))\n\tdefer func() {\n\t\terr := os.Remove(\"test-file\")\n\t\tassert.NoError(t, err, \"Expected no error during file cleanup\")\n\t}()\n\terr := showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"workflow not exist\")\n\n\t// error when describe workflow\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).Times(1)\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"cannot get information of pending activities\")\n\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t\tSearchAttributes: &types.SearchAttributes{\n\t\t\t\tIndexedFields: map[string][]byte{\n\t\t\t\t\t\"CustomKeywordField\": []byte(\"test\"),\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().GetSearchAttributes(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"false\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagPrintEventVersion, \"true\"))\n\t// error when converting search attribtues\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"Error in convert describe wf\")\n\n\t// no error when converting search attribtues\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{\n\t\t\t{\n\t\t\t\tActivityID: \"test-activity-id\",\n\t\t\t\tActivityType: &types.ActivityType{\n\t\t\t\t\tName: \"test-activity-type\",\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"false\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagEventID, \"1\"))\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"false\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagEventID, \"test-event-id\"))\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"EventId out of range\")\n\n\t// Test with strong consistency level\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), &types.GetWorkflowExecutionHistoryRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tWaitForNewEvent:        false,\n\t\tHistoryEventFilterType: types.HistoryEventFilterTypeAllEvent.Ptr(),\n\t\tSkipArchival:           false,\n\t\tQueryConsistencyLevel:  types.QueryConsistencyLevelStrong.Ptr(),\n\t}).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t}, nil).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"false\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagQueryConsistencyLevel, \"strong\"))\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n\n\t// Test with eventual consistency level\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        1,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskStarted.Ptr(),\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        2,\n\t\t\t\t\tVersion:   1,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tDecisionTaskCompletedEventAttributes: &types.DecisionTaskCompletedEventAttributes{\n\t\t\t\t\t\tScheduledEventID: 1,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).Times(1)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t}, nil).Times(1)\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagPrintFullyDetail, \"false\"), clitest.StringArgument(FlagResetPointsOnly, \"true\"),\n\t\tclitest.StringArgument(FlagQueryConsistencyLevel, \"eventual\"))\n\terr = showHistoryHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n}\n\nfunc Test_DescribeWorkflowHelper(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\n\t// Test with eventual consistency level\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tQueryConsistencyLevel: types.QueryConsistencyLevelEventual.Ptr(),\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t}, nil).Times(1)\n\n\tctx := clitest.NewCLIContext(t, app,\n\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagQueryConsistencyLevel, \"eventual\"))\n\terr := describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n\n\t// Test with strong consistency level\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t\tQueryConsistencyLevel: types.QueryConsistencyLevelStrong.Ptr(),\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t}, nil).Times(1)\n\n\tctx = clitest.NewCLIContext(t, app,\n\t\tclitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagQueryConsistencyLevel, \"strong\"))\n\terr = describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n\n\t// Test without specifying consistency level (should use default)\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), &types.DescribeWorkflowExecutionRequest{\n\t\tDomain: \"test-domain\",\n\t\tExecution: &types.WorkflowExecution{\n\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\tRunID:      \"test-run-id\",\n\t\t},\n\t}).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t\tPendingActivities: []*types.PendingActivityInfo{},\n\t}, nil).Times(1)\n\n\tctx = clitest.NewCLIContext(t, app,\n\t\tclitest.StringArgument(FlagDomain, \"test-domain\"))\n\terr = describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.NoError(t, err)\n}\n\nfunc Test_DescribeWorkflowHelper_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app)\n\terr := describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"))\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf(\"test-error\")).Times(1)\n\terr = describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"Describe workflow execution failed\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"), clitest.StringArgument(FlagQueryConsistencyLevel, \"invalid\"))\n\terr = describeWorkflowHelper(ctx, \"test-workflow-id\", \"test-run-id\")\n\tassert.ErrorContains(t, err, \"invalid query consistency level\")\n}\n\nfunc Test_ProcessResets(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tserverFrontendClient.EXPECT().DescribeWorkflowExecution(gomock.Any(), gomock.Any()).Return(&types.DescribeWorkflowExecutionResponse{\n\t\tWorkflowExecutionInfo: &types.WorkflowExecutionInfo{\n\t\t\tExecution: &types.WorkflowExecution{\n\t\t\t\tWorkflowID: \"test-workflow-id\",\n\t\t\t\tRunID:      \"test-run-id\",\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().GetWorkflowExecutionHistory(gomock.Any(), gomock.Any()).Return(&types.GetWorkflowExecutionHistoryResponse{\n\t\tHistory: &types.History{\n\t\t\tEvents: []*types.HistoryEvent{\n\t\t\t\t{\n\t\t\t\t\tID:        15,\n\t\t\t\t\tEventType: types.EventTypeWorkflowExecutionContinuedAsNew.Ptr(),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\tID:        17,\n\t\t\t\t\tEventType: types.EventTypeDecisionTaskCompleted.Ptr(),\n\t\t\t\t\tWorkflowExecutionStartedEventAttributes: &types.WorkflowExecutionStartedEventAttributes{\n\t\t\t\t\t\tContinuedExecutionRunID: \"test-run-id\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t},\n\t}, nil).AnyTimes()\n\tserverFrontendClient.EXPECT().ResetWorkflowExecution(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes()\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"))\n\n\t// Initialize channels and WaitGroup\n\twes := make(chan types.WorkflowExecution, 1)\n\tdone := make(chan bool)\n\twg := &sync.WaitGroup{}\n\twg.Add(1)\n\n\twid := \"test-workflow-id\"\n\trid := \"test-run-id\"\n\tparams := batchResetParamsType{\n\t\tresetType: resetTypeLastContinuedAsNew,\n\t\treason:    \"test-reason\",\n\t}\n\n\t// Run processResets in a goroutine to prevent hanging\n\tgo func() {\n\t\tdefer close(wes)\n\t\tdefer close(done)\n\t\tprocessResets(ctx, \"test-domain\", wes, done, wg, params)\n\t}()\n\n\twes <- types.WorkflowExecution{\n\t\tWorkflowID: wid,\n\t\tRunID:      rid,\n\t}\n\tdone <- true\n\twg.Wait()\n\tassert.NoError(t, doReset(ctx, \"test-domain\", wid, rid, params))\n}\n\nfunc Test_ListWorkflows_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagEarliestTime, \"InvalidTime\"))\n\t_, err := listWorkflows(ctx)\n\tassert.ErrorContains(t, err, \"cannot parse timeRange\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagLatestTime, \"InvalidTime\"))\n\t_, err = listWorkflows(ctx)\n\tassert.ErrorContains(t, err, \"cannot parse timeRange\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagWorkflowStatus, \"Closed\"), clitest.StringArgument(FlagOpen, \"true\"))\n\t_, err = listWorkflows(ctx)\n\tassert.ErrorContains(t, err, \"you can only filter on status for closed workflow\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagWorkflowStatus, \"Test\"))\n\t_, err = listWorkflows(ctx)\n\tassert.ErrorContains(t, err, \"failed to parse workflow status\")\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"),\n\t\tclitest.StringArgument(FlagWorkflowID, \"test-workflow-id\"), clitest.StringArgument(FlagWorkflowType, \"test-workflow-type\"))\n\t_, err = listWorkflows(ctx)\n\tassert.ErrorContains(t, err, \"you can filter on workflow_id or workflow_type\")\n}\n\nfunc Test_ListArchivedWorkflows_Errors(t *testing.T) {\n\tmockCtrl := gomock.NewController(t)\n\tserverFrontendClient := frontend.NewMockClient(mockCtrl)\n\tapp := NewCliApp(&clientFactoryMock{\n\t\tserverFrontendClient: serverFrontendClient,\n\t})\n\tctx := clitest.NewCLIContext(t, app)\n\t_, err := listArchivedWorkflows(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagDomain))\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"))\n\t_, err = listArchivedWorkflows(ctx)\n\tassert.ErrorContains(t, err, fmt.Sprintf(\"%s is required\", FlagListQuery))\n\n\tctx = clitest.NewCLIContext(t, app, clitest.StringArgument(FlagDomain, \"test-domain\"), clitest.StringArgument(FlagListQuery, \"test-query\"),\n\t\tclitest.IntArgument(FlagPageSize, -1), clitest.IntArgument(FlagContextTimeout, 10))\n\t_, err = listArchivedWorkflows(ctx)\n\tassert.NoError(t, err)\n}\n\nfunc TestMapQueryConsistencyLevelFromFlag(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tinput       string\n\t\texpected    types.QueryConsistencyLevel\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname:        \"valid strong\",\n\t\t\tinput:       \"strong\",\n\t\t\texpected:    types.QueryConsistencyLevelStrong,\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"valid eventual\",\n\t\t\tinput:       \"eventual\",\n\t\t\texpected:    types.QueryConsistencyLevelEventual,\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"mixed case\",\n\t\t\tinput:       \"Eventual\",\n\t\t\texpected:    types.QueryConsistencyLevelEventual,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"valid with spaces\",\n\t\t\tinput:       \"  eventual  \",\n\t\t\texpected:    types.QueryConsistencyLevelEventual,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"valid uppercase\",\n\t\t\tinput:       \"STRONG\",\n\t\t\texpected:    types.QueryConsistencyLevelStrong,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid value\",\n\t\t\tinput:       \"invalid\",\n\t\t\texpected:    types.QueryConsistencyLevelEventual, // default value\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"empty string\",\n\t\t\tinput:       \"\",\n\t\t\texpected:    types.QueryConsistencyLevelEventual, // default value\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"whitespace only\",\n\t\t\tinput:       \"   \",\n\t\t\texpected:    types.QueryConsistencyLevelEventual, // default value\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := mapQueryConsistencyLevelFromFlag(tc.input)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), \"invalid query consistency level\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestMapQueryRejectConditionFromFlag(t *testing.T) {\n\ttestCases := []struct {\n\t\tname        string\n\t\tinput       string\n\t\texpected    types.QueryRejectCondition\n\t\texpectError bool\n\t}{\n\t\t{\n\t\t\tname:        \"valid not_open\",\n\t\t\tinput:       \"not_open\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen,\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"valid not_completed_cleanly\",\n\t\t\tinput:       \"not_completed_cleanly\",\n\t\t\texpected:    types.QueryRejectConditionNotCompletedCleanly,\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:        \"mixed case\",\n\t\t\tinput:       \"Not_Open\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"with spaces\",\n\t\t\tinput:       \"  not_open  \",\n\t\t\texpected:    types.QueryRejectConditionNotOpen,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"uppercase\",\n\t\t\tinput:       \"NOT_OPEN\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen,\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"invalid value\",\n\t\t\tinput:       \"invalid\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen, // default value\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"empty string\",\n\t\t\tinput:       \"\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen, // default value\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"whitespace only\",\n\t\t\tinput:       \"   \",\n\t\t\texpected:    types.QueryRejectConditionNotOpen, // default value\n\t\t\texpectError: true,\n\t\t},\n\t\t{\n\t\t\tname:        \"partial match\",\n\t\t\tinput:       \"not_open_extra\",\n\t\t\texpected:    types.QueryRejectConditionNotOpen, // default value\n\t\t\texpectError: true,\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := mapQueryRejectConditionFromFlag(tc.input)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), \"invalid reject condition\")\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expected, result)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc TestParseClusterAttributes(t *testing.T) {\n\ttestCases := []struct {\n\t\tname                   string\n\t\tclusterAttributeScope  string\n\t\tclusterAttributeName   string\n\t\texpectedPolicy         *types.ActiveClusterSelectionPolicy\n\t\texpectError            bool\n\t\texpectedErrorSubstring string\n\t}{\n\t\t{\n\t\t\tname:                  \"both empty - should return nil\",\n\t\t\tclusterAttributeScope: \"\",\n\t\t\tclusterAttributeName:  \"\",\n\t\t\texpectedPolicy:        nil,\n\t\t\texpectError:           false,\n\t\t},\n\t\t{\n\t\t\tname:                  \"both provided - should return valid policy\",\n\t\t\tclusterAttributeScope: \"test-scope\",\n\t\t\tclusterAttributeName:  \"test-name\",\n\t\t\texpectedPolicy: &types.ActiveClusterSelectionPolicy{\n\t\t\t\tClusterAttribute: &types.ClusterAttribute{\n\t\t\t\t\tScope: \"test-scope\",\n\t\t\t\t\tName:  \"test-name\",\n\t\t\t\t},\n\t\t\t},\n\t\t\texpectError: false,\n\t\t},\n\t\t{\n\t\t\tname:                   \"empty scope with name provided - should error\",\n\t\t\tclusterAttributeScope:  \"\",\n\t\t\tclusterAttributeName:   \"test-name\",\n\t\t\texpectedPolicy:         nil,\n\t\t\texpectError:            true,\n\t\t\texpectedErrorSubstring: \"invalid cluster attribute\",\n\t\t},\n\t\t{\n\t\t\tname:                   \"scope provided with empty name - should error\",\n\t\t\tclusterAttributeScope:  \"test-scope\",\n\t\t\tclusterAttributeName:   \"\",\n\t\t\texpectedPolicy:         nil,\n\t\t\texpectError:            true,\n\t\t\texpectedErrorSubstring: \"invalid cluster attribute\",\n\t\t},\n\t}\n\n\tfor _, tc := range testCases {\n\t\tt.Run(tc.name, func(t *testing.T) {\n\t\t\tresult, err := parseClusterAttributes(tc.clusterAttributeScope, tc.clusterAttributeName)\n\n\t\t\tif tc.expectError {\n\t\t\t\tassert.Error(t, err)\n\t\t\t\tassert.Contains(t, err.Error(), tc.expectedErrorSubstring)\n\t\t\t\tassert.Nil(t, result)\n\t\t\t} else {\n\t\t\t\tassert.NoError(t, err)\n\t\t\t\tassert.Equal(t, tc.expectedPolicy, result)\n\t\t\t}\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/common/commoncli/cli.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage commoncli\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"os\"\n\t\"strings\"\n\n\t\"github.com/fatih/color\"\n)\n\nvar (\n\tcolorRed     = color.New(color.FgRed).SprintFunc()\n\tcolorMagenta = color.New(color.FgMagenta).SprintFunc()\n)\n\n// ExitHandler converts errors that urfave/cli did not handle into a nicely\n// printed message, and an appropriate os.Exit call to ensure this func never\n// returns from either branch.\n//\n// This should be used instead of an in-CLI ExitErrHandler, as using that still\n// leaves a dangling `err` return value from *cli.App.Run(), which is awkward\n// in a number of ways.\nfunc ExitHandler(err error) {\n\tif err == nil {\n\t\t// no error, which (oddly) includes some command-not-found paths\n\t\tos.Exit(0)\n\t}\n\n\t// print what we can to stderr.\n\t// and ignore errs while doing so, since we have no real alternative.\n\t_ = printErr(err, os.Stderr)\n\n\t// all errors are \"fatal\", unlike default behavior which only fails if you\n\t// return an ExitCoder error.\n\tos.Exit(1)\n}\n\n// prints this (possibly printable) error to the given io.Writer.\n// write-errors will be returned, if any are encountered.\nfunc printErr(err error, to io.Writer) (writeErr error) {\n\t// the way Go does error wrapping is really a massive pain, as the stdlib encourages\n\t// people to wrap *and duplicate* basically every error message, because they don't\n\t// provide any API for getting \"just this error\" content.\n\t//\n\t// we can live with that.\n\t// build strings recursively, stripping out matching suffixes, and hope for the best.\n\t// it's quite wasteful, but it's only done once per process so it's fine.\n\n\t// error-coalescing write-helper.\n\t// if an error is encountered, the first will be saved, and later calls will no-op.\n\twrite := func(format string, a ...any) {\n\t\tif writeErr != nil {\n\t\t\treturn\n\t\t}\n\t\t_, tmpErr := fmt.Fprintf(to, format, a...)\n\t\tif tmpErr != nil {\n\t\t\twriteErr = tmpErr\n\t\t}\n\t}\n\n\tvar topPrintable *printableErr\n\t_ = errors.As(err, &topPrintable)\n\n\tvar unwrapped []error\n\tunwrapped = append(unwrapped, err)\n\tcurrent := err\n\tfor i := 0; i < 1000; i++ { // 1k would be an insane depth, likely a loop\n\t\tnext := errors.Unwrap(current)\n\t\tif next != nil {\n\t\t\tcurrent = next\n\t\t\tunwrapped = append(unwrapped, next)\n\t\t} else {\n\t\t\tbreak\n\t\t}\n\t}\n\n\ttype parsed struct {\n\t\terr error\n\t\tmsg string\n\t}\n\tallParsed := make([]parsed, len(unwrapped))\n\tfor i := range unwrapped {\n\t\tp := parsed{err: unwrapped[i]}\n\t\tif perr, ok := unwrapped[i].(*printableErr); ok {\n\t\t\tp.msg = perr.display\n\t\t\tif i > 0 && perr == topPrintable { // must be same instance somewhere deeper in the stack of errors\n\t\t\t\tp.msg = \"Error (above): \" + p.msg\n\t\t\t} else {\n\t\t\t\tp.msg = \"Error: \" + p.msg\n\t\t\t}\n\t\t} else {\n\t\t\tp.msg = unwrapped[i].Error()\n\t\t}\n\t\tif i > 0 {\n\t\t\t// trim current from previous (needs to be err.Error() instead of p.msg or the special \"Error: \" prefixing above will break trimming),\n\t\t\t// and attempt to trim off any trailing \": \" because it's so common.\n\t\t\t// this is not an exact science, but neither are error messages so it's just a hopeful hack.\n\t\t\tallParsed[i-1].msg = strings.TrimSuffix(\n\t\t\t\tstrings.TrimSpace(\n\t\t\t\t\tstrings.TrimSuffix(\n\t\t\t\t\t\tstrings.TrimSpace(allParsed[i-1].msg),\n\t\t\t\t\t\tstrings.TrimSpace(p.err.Error()),\n\t\t\t\t\t),\n\t\t\t\t),\n\t\t\t\t\":\",\n\t\t\t)\n\t\t}\n\t\tallParsed[i] = p\n\t}\n\n\tif topPrintable != nil {\n\t\twrite(\"%s %s\\n\", colorRed(\"Error:\"), topPrintable.display)\n\t\tif allParsed[0].err == topPrintable {\n\t\t\tallParsed = allParsed[1:] // already printed, skip\n\t\t}\n\t} else { // len(allParsed) > 0\n\t\twrite(\"%s %s\\n\", colorRed(\"Error:\"), allParsed[0].msg)\n\t\tallParsed = allParsed[1:] // already printed, skip\n\t}\n\n\tif len(allParsed) == 0 {\n\t\treturn // no details to print\n\t}\n\n\tindent := \"  \"\n\twrite(\"%s\\n\", colorMagenta(\"Error details:\")) // only the top level gets color\n\t// and now write the rest\n\tneedDetails := false\n\tfor _, this := range allParsed {\n\t\tlines := strings.Split(this.msg, \"\\n\")\n\t\tif needDetails {\n\t\t\twrite(\"%s%s\\n\", indent, \"Error details:\")\n\t\t\tindent += \"  \" // next errors indent further\n\t\t\tneedDetails = false\n\t\t}\n\t\tfor _, line := range lines {\n\t\t\twrite(\"%s%s\\n\", indent, line)\n\t\t}\n\t\tif _, ok := this.err.(*printableErr); ok { // already unwrapped\n\t\t\tneedDetails = true\n\t\t}\n\t}\n\treturn\n}\n\n// Problem returns a typed error that will report this message \"nicely\" to the\n// user if it exits the CLI app.  The message will be used as the top-level\n// \"Error: ...\" string regardless of where in the error stack it is, and other\n// wrapped errors will be printed line by line beneath it.\n//\n// Nested Problem messages will nest structurally, like:\n//\n//\tError: msg\n//\tDetails:\n//\t  some error\n//\t  some other error\n//\t  Error: message\n//\t  ErrorDetails:\n//\t    more nested errors\nfunc Problem(msg string, err error) error {\n\tif err != nil && strings.Contains(err.Error(), \"Request unauthorized\") {\n\t\tmsg = fmt.Sprintf(\"%s. Ensure the domain and cluster are correct.\", msg)\n\t}\n\treturn &printableErr{msg, err}\n}\n\ntype printableErr struct {\n\tdisplay string\n\tcause   error\n}\n\nfunc (p *printableErr) Error() string {\n\tbuf := strings.Builder{}\n\tbuf.WriteString(p.display)\n\tif p.cause != nil {\n\t\tbuf.WriteString(\": \")\n\t\tbuf.WriteString(p.cause.Error())\n\t}\n\treturn buf.String()\n}\n\nfunc (p *printableErr) Unwrap() error {\n\treturn p.cause\n}\n"
  },
  {
    "path": "tools/common/commoncli/cli_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage commoncli\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\n\t\"github.com/uber/cadence/common/types\"\n)\n\nfunc TestPrintErr(t *testing.T) {\n\trun := func(t *testing.T, err error) string {\n\t\tt.Helper()\n\t\tbuf := strings.Builder{}\n\t\t// improves t.Log display by placing it all on a common start column, and easier to read the Equal below\n\t\tbuf.WriteString(\"\\n\")\n\t\tfailed := printErr(err, &buf)\n\t\tassert.NoError(t, failed, \"error during printing\")\n\t\tt.Log(buf.String())\n\t\treturn buf.String()\n\t}\n\tt.Run(\"printable only\", func(t *testing.T) {\n\t\tstr := run(t, Problem(\"a problem\", nil))\n\t\tassert.Equal(t, `\nError: a problem\n`, str)\n\t})\n\tt.Run(\"printable top\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tProblem(\"a problem\",\n\t\t\t\tfmt.Errorf(\"wrapper: %w\",\n\t\t\t\t\terrors.New(\"cause\"))))\n\t\t// causes are nested flat, chains are cleaned up\n\t\tassert.Equal(t, `\nError: a problem\nError details:\n  wrapper\n  cause\n`, str)\n\t})\n\tt.Run(\"printable top with access denied\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tProblem(\"a problem\",\n\t\t\t\tfmt.Errorf(\"wrapper: %w\",\n\t\t\t\t\t&types.AccessDeniedError{Message: \"Request unauthorized.\"})))\n\t\t// causes are nested flat, chains are cleaned up\n\t\tassert.Equal(t, `\nError: a problem. Ensure the domain and cluster are correct.\nError details:\n  wrapper\n  Request unauthorized.\n`, str)\n\t})\n\tt.Run(\"printable top fancy middle\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tProblem(\"a problem\",\n\t\t\t\tfmt.Errorf(\"wrapper caused by -> %w\",\n\t\t\t\t\terrors.New(\"cause\"))))\n\t\t// cleans up other kinds of suffixes\n\t\tassert.Equal(t, `\nError: a problem\nError details:\n  wrapper caused by ->\n  cause\n`, str)\n\t})\n\tt.Run(\"printable top unfixable middle\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tProblem(\"a problem\",\n\t\t\t\tfmt.Errorf(\"wrapper (caused by: %w)\",\n\t\t\t\t\terrors.New(\"cause\"))))\n\t\t// does not clean up something that isn't a suffix\n\t\tassert.Equal(t, `\nError: a problem\nError details:\n  wrapper (caused by: cause)\n  cause\n`, str)\n\t})\n\tt.Run(\"printable bottom\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tfmt.Errorf(\"msg: %w\",\n\t\t\t\tProblem(\"a problem\", nil)))\n\t\t// note: the Problem is the displayed err, even though there is an error wrapper above it.\n\t\t// all contents are visible for troubleshooting purposes, it just tries to make the \"problem\" clearer.\n\t\tassert.Equal(t, `\nError: a problem\nError details:\n  msg\n  Error (above): a problem\n`, str)\n\t})\n\tt.Run(\"printable mid\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tfmt.Errorf(\"msg: %w\",\n\t\t\t\tProblem(\"one layer deep\",\n\t\t\t\t\terrors.New(\"cause\"))))\n\t\tassert.Equal(t, `\nError: one layer deep\nError details:\n  msg\n  Error (above): one layer deep\n  Error details:\n    cause\n`, str)\n\t})\n\tt.Run(\"printable nested\", func(t *testing.T) {\n\t\tstr := run(t,\n\t\t\tProblem(\"top\",\n\t\t\t\tfmt.Errorf(\"wrapper: %w\",\n\t\t\t\t\tProblem(\"bottom\",\n\t\t\t\t\t\terrors.New(\"cause\")))))\n\t\tassert.Equal(t, `\nError: top\nError details:\n  wrapper\n  Error: bottom\n  Error details:\n    cause\n`, str)\n\t})\n\tt.Run(\"multi-line error\", func(t *testing.T) {\n\t\tstr := run(t, fmt.Errorf(`what if\nit has\nmultiple lines: %w`, errors.New(`even\nwhen\nnested`)))\n\t\t// maybe not ideal but it works well enough I think\n\t\tassert.Equal(t, `\nError: what if\nit has\nmultiple lines\nError details:\n  even\n  when\n  nested\n`, str)\n\t})\n}\n"
  },
  {
    "path": "tools/common/flag/flag.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage flag\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\ntype (\n\tStringMap map[string]string\n)\n\nfunc (m *StringMap) Set(value string) error {\n\tif m == nil {\n\t\treturn fmt.Errorf(\"StringMap is nil\")\n\t}\n\tif *m == nil {\n\t\t*m = make(map[string]string)\n\t}\n\tfor _, s := range strings.Split(value, \",\") {\n\t\tkv := strings.Split(s, \"=\")\n\t\tif len(kv) != 2 {\n\t\t\treturn fmt.Errorf(\"should be in 'key1=value1,key2=value2,...,keyN=valueN' format\")\n\t\t}\n\t\t(*m)[kv[0]] = kv[1]\n\t}\n\treturn nil\n}\n\nfunc (m *StringMap) String() string {\n\tif m == nil {\n\t\treturn fmt.Sprintf(\"%v\", map[string]string(nil))\n\t}\n\treturn fmt.Sprintf(\"%v\", *m)\n}\n\nfunc (m *StringMap) Value() map[string]string {\n\tif m == nil {\n\t\treturn nil\n\t}\n\treturn *m\n}\n"
  },
  {
    "path": "tools/common/flag/flag_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage flag\n\nimport (\n\t\"os\"\n\t\"reflect\"\n\t\"testing\"\n\n\t\"github.com/urfave/cli/v2\"\n)\n\nfunc TestParseMultiStringMap(t *testing.T) {\n\t_ = (&cli.App{\n\t\tFlags: []cli.Flag{\n\t\t\t&cli.GenericFlag{Name: \"serve\", Aliases: []string{\"s\"}, Value: &StringMap{}},\n\t\t},\n\t\tAction: func(ctx *cli.Context) error {\n\t\t\tif !reflect.DeepEqual(ctx.Generic(\"serve\"), &StringMap{\"a\": \"b\", \"c\": \"d\"}) {\n\t\t\t\tt.Errorf(\"main name not set\")\n\t\t\t}\n\t\t\tif !reflect.DeepEqual(ctx.Generic(\"s\"), &StringMap{\"a\": \"b\", \"c\": \"d\"}) {\n\t\t\t\tt.Errorf(\"short name not set\")\n\t\t\t}\n\t\t\treturn nil\n\t\t},\n\t}).Run([]string{\"run\", \"-s\", \"a=b\", \"-s\", \"c=d\"})\n}\n\nfunc TestParseMultiStringMapFromEnv(t *testing.T) {\n\tos.Clearenv()\n\t_ = os.Setenv(\"APP_SERVE\", \"x=y,w=v\")\n\t_ = (&cli.App{\n\t\tFlags: []cli.Flag{\n\t\t\t&cli.GenericFlag{Name: \"serve\", Aliases: []string{\"s\"}, Value: &StringMap{}, EnvVars: []string{\"APP_SERVE\"}},\n\t\t},\n\t\tAction: func(ctx *cli.Context) error {\n\t\t\tif !reflect.DeepEqual(ctx.Generic(\"serve\"), &StringMap{\"x\": \"y\", \"w\": \"v\"}) {\n\t\t\t\tt.Errorf(\"main name not set from env\")\n\t\t\t}\n\t\t\tif !reflect.DeepEqual(ctx.Generic(\"s\"), &StringMap{\"x\": \"y\", \"w\": \"v\"}) {\n\t\t\t\tt.Errorf(\"short name not set from env\")\n\t\t\t}\n\t\t\treturn nil\n\t\t},\n\t}).Run([]string{\"run\"})\n}\n\nfunc TestParseMultiStringMapFromEnvCascade(t *testing.T) {\n\tos.Clearenv()\n\t_ = os.Setenv(\"APP_FOO\", \"u=t,r=s\")\n\t_ = (&cli.App{\n\t\tFlags: []cli.Flag{\n\t\t\t&cli.GenericFlag{Name: \"foos\", Value: &StringMap{}, EnvVars: []string{\"COMPAT_FOO\", \"APP_FOO\"}},\n\t\t},\n\t\tAction: func(ctx *cli.Context) error {\n\t\t\tif !reflect.DeepEqual(ctx.Generic(\"foos\"), &StringMap{\"u\": \"t\", \"r\": \"s\"}) {\n\t\t\t\tt.Errorf(\"value not set from env\")\n\t\t\t}\n\t\t\treturn nil\n\t\t},\n\t}).Run([]string{\"run\"})\n}\n"
  },
  {
    "path": "tools/common/schema/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"fmt\"\n\t\"os\"\n\n\t\"github.com/urfave/cli/v2\"\n)\n\n// VerifyCompatibleVersion ensures that the installed version is greater than or equal to the expected version.\nfunc VerifyCompatibleVersion(\n\tdb SchemaClient,\n\tdbName string,\n\texpectedVersion string,\n) error {\n\n\tversion, err := db.ReadSchemaVersion()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"reading schema version keyspace/database: %q, error: %w\", dbName, err)\n\t}\n\t// In most cases, the versions should match. However if after a schema upgrade there is a code\n\t// rollback, the code version (expected version) would fall lower than the actual version in\n\t// cassandra. This check is to allow such rollbacks since we only make backwards compatible schema\n\t// changes\n\tif cmpVersion(version, expectedVersion) < 0 {\n\t\treturn fmt.Errorf(\n\t\t\t\"version mismatch for keyspace/database: %q. Expected version: %s cannot be greater than \"+\n\t\t\t\t\"Actual version: %s\", dbName, expectedVersion, version,\n\t\t)\n\t}\n\treturn nil\n}\n\n// SetupFromConfig sets up schema tables based on the given config\nfunc SetupFromConfig(config *SetupConfig, db SchemaClient) error {\n\tif err := validateSetupConfig(config); err != nil {\n\t\treturn err\n\t}\n\treturn newSetupSchemaTask(db, config).Run()\n}\n\n// Setup sets up schema tables\nfunc Setup(cli *cli.Context, db SchemaClient) error {\n\tcfg, err := newSetupConfig(cli)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn newSetupSchemaTask(db, cfg).Run()\n}\n\n// UpdateFromConfig updates the schema for the specified database based on the given config\nfunc UpdateFromConfig(config *UpdateConfig, db SchemaClient) error {\n\tif err := validateUpdateConfig(config); err != nil {\n\t\treturn err\n\t}\n\treturn NewUpdateSchemaTask(db, config).Run()\n}\n\n// Update updates the schema for the specified database\nfunc Update(cli *cli.Context, db SchemaClient) error {\n\tcfg, err := newUpdateConfig(cli)\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn NewUpdateSchemaTask(db, cfg).Run()\n}\n\nfunc newUpdateConfig(cli *cli.Context) (*UpdateConfig, error) {\n\tconfig := new(UpdateConfig)\n\tschemaDir := cli.String(CLIOptSchemaDir)\n\tif len(schemaDir) == 0 {\n\t\treturn nil, NewConfigError(\"missing \" + flag(CLIOptSchemaDir) + \" argument \")\n\t}\n\tconfig.SchemaFS = os.DirFS(schemaDir)\n\tconfig.IsDryRun = cli.Bool(CLIOptDryrun)\n\tconfig.TargetVersion = cli.String(CLIOptTargetVersion)\n\n\tif err := validateUpdateConfig(config); err != nil {\n\t\treturn nil, err\n\t}\n\treturn config, nil\n}\n\nfunc newSetupConfig(cli *cli.Context) (*SetupConfig, error) {\n\tconfig := new(SetupConfig)\n\tconfig.SchemaFilePath = cli.String(CLIOptSchemaFile)\n\tconfig.InitialVersion = cli.String(CLIOptVersion)\n\tconfig.DisableVersioning = cli.Bool(CLIOptDisableVersioning)\n\tconfig.Overwrite = cli.Bool(CLIOptOverwrite)\n\n\tif err := validateSetupConfig(config); err != nil {\n\t\treturn nil, err\n\t}\n\treturn config, nil\n}\n\nfunc validateSetupConfig(config *SetupConfig) error {\n\tif len(config.SchemaFilePath) == 0 && config.DisableVersioning {\n\t\treturn NewConfigError(\"missing schemaFilePath \" + flag(CLIOptSchemaFile))\n\t}\n\tif (config.DisableVersioning && len(config.InitialVersion) > 0) ||\n\t\t(!config.DisableVersioning && len(config.InitialVersion) == 0) {\n\t\treturn NewConfigError(\"either \" + flag(CLIOptDisableVersioning) + \" or \" +\n\t\t\tflag(CLIOptVersion) + \" but not both must be specified\")\n\t}\n\tif !config.DisableVersioning {\n\t\tver, err := parseValidateVersion(config.InitialVersion)\n\t\tif err != nil {\n\t\t\treturn NewConfigError(\"invalid \" + flag(CLIOptVersion) + \" argument:\" + err.Error())\n\t\t}\n\t\tconfig.InitialVersion = ver\n\t}\n\treturn nil\n}\n\nfunc validateUpdateConfig(config *UpdateConfig) error {\n\tif config.SchemaFS == nil {\n\t\treturn NewConfigError(\"schema file system is not set\")\n\t}\n\tif len(config.TargetVersion) > 0 {\n\t\tver, err := parseValidateVersion(config.TargetVersion)\n\t\tif err != nil {\n\t\t\treturn NewConfigError(\"invalid \" + flag(CLIOptTargetVersion) + \" argument:\" + err.Error())\n\t\t}\n\t\tconfig.TargetVersion = ver\n\t}\n\treturn nil\n}\n\nfunc flag(opt string) string {\n\treturn \"(-\" + opt + \")\"\n}\n"
  },
  {
    "path": "tools/common/schema/handler_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"os\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tHandlerTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestHandlerTestSuite(t *testing.T) {\n\tsuite.Run(t, new(HandlerTestSuite))\n}\n\nfunc (s *HandlerTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *HandlerTestSuite) TestValidateSetupConfig() {\n\n\tconfig := new(SetupConfig)\n\ts.assertValidateSetupFails(config)\n\n\tconfig.InitialVersion = \"0.1\"\n\tconfig.DisableVersioning = true\n\tconfig.SchemaFilePath = \"\"\n\ts.assertValidateSetupFails(config)\n\n\tconfig.InitialVersion = \"0.1\"\n\tconfig.DisableVersioning = true\n\tconfig.SchemaFilePath = \"/tmp/foo.cql\"\n\ts.assertValidateSetupFails(config)\n\n\tconfig.InitialVersion = \"\"\n\tconfig.DisableVersioning = true\n\tconfig.SchemaFilePath = \"\"\n\ts.assertValidateSetupFails(config)\n\n\tconfig.InitialVersion = \"0.1\"\n\tconfig.DisableVersioning = false\n\tconfig.SchemaFilePath = \"/tmp/foo.cql\"\n\ts.assertValidateSetupSucceeds(config)\n\n\tconfig.InitialVersion = \"0.1\"\n\tconfig.DisableVersioning = false\n\tconfig.SchemaFilePath = \"\"\n\ts.assertValidateSetupSucceeds(config)\n\n\tconfig.InitialVersion = \"\"\n\tconfig.DisableVersioning = true\n\tconfig.SchemaFilePath = \"/tmp/foo.cql\"\n\ts.assertValidateSetupSucceeds(config)\n}\n\nfunc (s *HandlerTestSuite) TestValidateUpdateConfig() {\n\n\tconfig := new(UpdateConfig)\n\ts.assertValidateUpdateFails(config)\n\n\tconfig.SchemaFS = os.DirFS(\"/tmp\")\n\tconfig.TargetVersion = \"abc\"\n\ts.assertValidateUpdateFails(config)\n\n\tconfig.SchemaFS = os.DirFS(\"/tmp\")\n\tconfig.TargetVersion = \"\"\n\ts.assertValidateUpdateSucceeds(config)\n\n\tconfig.SchemaFS = os.DirFS(\"/tmp\")\n\tconfig.TargetVersion = \"1.2\"\n\ts.assertValidateUpdateSucceeds(config)\n\n\tconfig.SchemaFS = os.DirFS(\"/tmp\")\n\tconfig.TargetVersion = \"v1.2\"\n\ts.assertValidateUpdateSucceeds(config)\n\ts.Equal(\"1.2\", config.TargetVersion)\n}\n\nfunc (s *HandlerTestSuite) assertValidateSetupSucceeds(input *SetupConfig) {\n\terr := validateSetupConfig(input)\n\ts.Nil(err)\n}\n\nfunc (s *HandlerTestSuite) assertValidateSetupFails(input *SetupConfig) {\n\terr := validateSetupConfig(input)\n\ts.NotNil(err)\n\t_, ok := err.(*ConfigError)\n\ts.True(ok)\n}\n\nfunc (s *HandlerTestSuite) assertValidateUpdateSucceeds(input *UpdateConfig) {\n\terr := validateUpdateConfig(input)\n\ts.Nil(err)\n}\n\nfunc (s *HandlerTestSuite) assertValidateUpdateFails(input *UpdateConfig) {\n\terr := validateUpdateConfig(input)\n\ts.NotNil(err)\n\t_, ok := err.(*ConfigError)\n\ts.True(ok)\n}\n"
  },
  {
    "path": "tools/common/schema/schema_client_mock.go",
    "content": "// Code generated by MockGen. DO NOT EDIT.\n// Source: types.go\n//\n// Generated by this command:\n//\n//\tmockgen -source types.go -destination schema_client_mock.go -package schema github.com/uber/cadence/tools/common/schema SchemaClient\n//\n\n// Package schema is a generated GoMock package.\npackage schema\n\nimport (\n\treflect \"reflect\"\n\n\tgomock \"go.uber.org/mock/gomock\"\n)\n\n// MockSchemaClient is a mock of SchemaClient interface.\ntype MockSchemaClient struct {\n\tctrl     *gomock.Controller\n\trecorder *MockSchemaClientMockRecorder\n\tisgomock struct{}\n}\n\n// MockSchemaClientMockRecorder is the mock recorder for MockSchemaClient.\ntype MockSchemaClientMockRecorder struct {\n\tmock *MockSchemaClient\n}\n\n// NewMockSchemaClient creates a new mock instance.\nfunc NewMockSchemaClient(ctrl *gomock.Controller) *MockSchemaClient {\n\tmock := &MockSchemaClient{ctrl: ctrl}\n\tmock.recorder = &MockSchemaClientMockRecorder{mock}\n\treturn mock\n}\n\n// EXPECT returns an object that allows the caller to indicate expected use.\nfunc (m *MockSchemaClient) EXPECT() *MockSchemaClientMockRecorder {\n\treturn m.recorder\n}\n\n// Close mocks base method.\nfunc (m *MockSchemaClient) Close() {\n\tm.ctrl.T.Helper()\n\tm.ctrl.Call(m, \"Close\")\n}\n\n// Close indicates an expected call of Close.\nfunc (mr *MockSchemaClientMockRecorder) Close() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"Close\", reflect.TypeOf((*MockSchemaClient)(nil).Close))\n}\n\n// CreateSchemaVersionTables mocks base method.\nfunc (m *MockSchemaClient) CreateSchemaVersionTables() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"CreateSchemaVersionTables\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// CreateSchemaVersionTables indicates an expected call of CreateSchemaVersionTables.\nfunc (mr *MockSchemaClientMockRecorder) CreateSchemaVersionTables() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"CreateSchemaVersionTables\", reflect.TypeOf((*MockSchemaClient)(nil).CreateSchemaVersionTables))\n}\n\n// DropAllTables mocks base method.\nfunc (m *MockSchemaClient) DropAllTables() error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"DropAllTables\")\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// DropAllTables indicates an expected call of DropAllTables.\nfunc (mr *MockSchemaClientMockRecorder) DropAllTables() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"DropAllTables\", reflect.TypeOf((*MockSchemaClient)(nil).DropAllTables))\n}\n\n// ExecDDLQuery mocks base method.\nfunc (m *MockSchemaClient) ExecDDLQuery(stmt string, args ...any) error {\n\tm.ctrl.T.Helper()\n\tvarargs := []any{stmt}\n\tfor _, a := range args {\n\t\tvarargs = append(varargs, a)\n\t}\n\tret := m.ctrl.Call(m, \"ExecDDLQuery\", varargs...)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// ExecDDLQuery indicates an expected call of ExecDDLQuery.\nfunc (mr *MockSchemaClientMockRecorder) ExecDDLQuery(stmt any, args ...any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\tvarargs := append([]any{stmt}, args...)\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ExecDDLQuery\", reflect.TypeOf((*MockSchemaClient)(nil).ExecDDLQuery), varargs...)\n}\n\n// ReadSchemaVersion mocks base method.\nfunc (m *MockSchemaClient) ReadSchemaVersion() (string, error) {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"ReadSchemaVersion\")\n\tret0, _ := ret[0].(string)\n\tret1, _ := ret[1].(error)\n\treturn ret0, ret1\n}\n\n// ReadSchemaVersion indicates an expected call of ReadSchemaVersion.\nfunc (mr *MockSchemaClientMockRecorder) ReadSchemaVersion() *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"ReadSchemaVersion\", reflect.TypeOf((*MockSchemaClient)(nil).ReadSchemaVersion))\n}\n\n// UpdateSchemaVersion mocks base method.\nfunc (m *MockSchemaClient) UpdateSchemaVersion(newVersion, minCompatibleVersion string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"UpdateSchemaVersion\", newVersion, minCompatibleVersion)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// UpdateSchemaVersion indicates an expected call of UpdateSchemaVersion.\nfunc (mr *MockSchemaClientMockRecorder) UpdateSchemaVersion(newVersion, minCompatibleVersion any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"UpdateSchemaVersion\", reflect.TypeOf((*MockSchemaClient)(nil).UpdateSchemaVersion), newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog mocks base method.\nfunc (m *MockSchemaClient) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc string) error {\n\tm.ctrl.T.Helper()\n\tret := m.ctrl.Call(m, \"WriteSchemaUpdateLog\", oldVersion, newVersion, manifestMD5, desc)\n\tret0, _ := ret[0].(error)\n\treturn ret0\n}\n\n// WriteSchemaUpdateLog indicates an expected call of WriteSchemaUpdateLog.\nfunc (mr *MockSchemaClientMockRecorder) WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc any) *gomock.Call {\n\tmr.mock.ctrl.T.Helper()\n\treturn mr.mock.ctrl.RecordCallWithMethodType(mr.mock, \"WriteSchemaUpdateLog\", reflect.TypeOf((*MockSchemaClient)(nil).WriteSchemaUpdateLog), oldVersion, newVersion, manifestMD5, desc)\n}\n"
  },
  {
    "path": "tools/common/schema/schema_graph.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage schema\n\nimport (\n\t\"gonum.org/v1/gonum/graph\"\n\t\"gonum.org/v1/gonum/graph/path\"\n\t\"gonum.org/v1/gonum/graph/simple\"\n)\n\n// versionNode is the version representation in the schema graph\n// implements gonum.org/v1/gonum/graph Node\ntype versionNode struct {\n\tid    int64\n\tvalue string\n}\n\nfunc (n *versionNode) ID() int64 {\n\treturn n.id\n}\n\n// versionEdge is the schema update representation in the schema graph\n// implements gonum.org/v1/gonum/graph Edge\ntype versionEdge struct {\n\tfrom    graph.Node\n\tto      graph.Node\n\tdirName string\n}\n\nfunc (e *versionEdge) From() graph.Node {\n\treturn e.from\n}\n\nfunc (e *versionEdge) To() graph.Node {\n\treturn e.to\n}\n\nfunc (e *versionEdge) ReversedEdge() graph.Edge {\n\treturn &versionEdge{to: e.from, from: e.to, dirName: e.dirName}\n}\n\nfunc addNode(version string, g graph.NodeAdder, m map[string]*versionNode) graph.Node {\n\tn := &versionNode{id: int64(len(m)), value: version}\n\tm[version] = n\n\tg.AddNode(n)\n\treturn n\n}\n\nfunc findShortestPath(startVer string, endVer string, increments []string, shortcuts []squashVersion) ([]string, error) {\n\tnodeMap := make(map[string]*versionNode, len(increments)+1)\n\tg := simple.NewDirectedGraph()\n\tstartNode := addNode(startVer, g, nodeMap)\n\tendNode := addNode(endVer, g, nodeMap)\n\tprev := startNode\n\tfor _, inc := range increments {\n\t\tnext := endNode\n\t\tv := dirToVersion(inc)\n\t\tif v != endVer {\n\t\t\tnext = addNode(v, g, nodeMap)\n\t\t}\n\t\te := &versionEdge{from: prev, to: next, dirName: inc}\n\t\tg.SetEdge(e)\n\t\tprev = next\n\t}\n\n\tfor _, s := range shortcuts {\n\t\tl, lok := nodeMap[s.prev]\n\t\tr, rok := nodeMap[s.ver]\n\t\tif lok && rok {\n\t\t\te := &versionEdge{from: l, to: r, dirName: s.dirName}\n\t\t\tg.SetEdge(e)\n\t\t}\n\t}\n\n\tvar res []string\n\tp, _ := path.DijkstraFrom(startNode, g).To(endNode.ID())\n\tfor i := range p {\n\t\t// given the path, populate the directory slice\n\t\tif i == 0 {\n\t\t\tcontinue\n\t\t}\n\t\tres = append(res, g.Edge(p[i-1].ID(), p[i].ID()).(*versionEdge).dirName)\n\t}\n\n\treturn res, nil\n}\n"
  },
  {
    "path": "tools/common/schema/schema_graph_test.go",
    "content": "// Copyright (c) 2020 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage schema\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype VersionGraphTestSuite struct {\n\tsuite.Suite\n}\n\nfunc TestVersionGraphTestSuite(t *testing.T) {\n\tsuite.Run(t, new(VersionGraphTestSuite))\n}\n\nfunc (s *VersionGraphTestSuite) TestFindShortestPath() {\n\tincrements := []string{\"v0.1\", \"v0.2\", \"v0.3\", \"v0.4\", \"v0.5\", \"v1.0\", \"v1.1\", \"v1.2\", \"v1.3\"}\n\ttests := []struct {\n\t\tname      string\n\t\tto        string\n\t\tshortcuts []squashVersion\n\t\twant      []string\n\t}{\n\t\t{\n\t\t\tname: \"increments only\",\n\t\t\twant: []string{\"v0.1\", \"v0.2\", \"v0.3\", \"v0.4\", \"v0.5\", \"v1.0\", \"v1.1\", \"v1.2\", \"v1.3\"},\n\t\t},\n\t\t{\n\t\t\tname:      \"single hop\",\n\t\t\tshortcuts: []squashVersion{{prev: \"0.0\", ver: \"1.3\", dirName: \"foo\"}},\n\t\t\twant:      []string{\"foo\"},\n\t\t},\n\t\t{\n\t\t\tname: \"middle hops\",\n\t\t\tshortcuts: []squashVersion{\n\t\t\t\t{prev: \"0.2\", ver: \"0.4\", dirName: \"foo\"},\n\t\t\t\t{prev: \"1.0\", ver: \"1.2\", dirName: \"bar\"},\n\t\t\t},\n\t\t\twant: []string{\"v0.1\", \"v0.2\", \"foo\", \"v0.5\", \"v1.0\", \"bar\", \"v1.3\"},\n\t\t},\n\t\t{\n\t\t\tname: \"hop at the start\",\n\t\t\tshortcuts: []squashVersion{\n\t\t\t\t{prev: \"0.0\", ver: \"0.4\", dirName: \"foo\"},\n\t\t\t},\n\t\t\twant: []string{\"foo\", \"v0.5\", \"v1.0\", \"v1.1\", \"v1.2\", \"v1.3\"},\n\t\t},\n\t\t{\n\t\t\tname: \"hop at the end\",\n\t\t\tshortcuts: []squashVersion{\n\t\t\t\t{prev: \"1.0\", ver: \"1.3\", dirName: \"foo\"},\n\t\t\t},\n\t\t\twant: []string{\"v0.1\", \"v0.2\", \"v0.3\", \"v0.4\", \"v0.5\", \"v1.0\", \"foo\"},\n\t\t},\n\t\t{\n\t\t\tname: \"out of range\",\n\t\t\tto:   \"v1.4\",\n\t\t\tshortcuts: []squashVersion{\n\t\t\t\t{prev: \"1.0\", ver: \"1.3\", dirName: \"foo\"},\n\t\t\t},\n\t\t},\n\t}\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\tto := increments[len(increments)-1]\n\t\t\tif tt.to != \"\" {\n\t\t\t\tto = tt.to\n\t\t\t}\n\t\t\tp, err := findShortestPath(\"0.0\", dirToVersion(to), increments, tt.shortcuts)\n\t\t\ts.Require().NoError(err)\n\t\t\ts.Equal(tt.want, p)\n\t\t})\n\t}\n}\n"
  },
  {
    "path": "tools/common/schema/setuptask.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"log\"\n\t\"os\"\n)\n\n// SetupTask represents a task\n// that sets up cassandra schema on\n// a specified keyspace\ntype SetupTask struct {\n\tdb     SchemaClient\n\tconfig *SetupConfig\n}\n\nfunc newSetupSchemaTask(db SchemaClient, config *SetupConfig) *SetupTask {\n\treturn &SetupTask{\n\t\tdb:     db,\n\t\tconfig: config,\n\t}\n}\n\n// Run executes the task\nfunc (task *SetupTask) Run() error {\n\tconfig := task.config\n\tlog.Printf(\"Starting schema setup, config=%+v\\n\", config)\n\n\tif config.Overwrite {\n\t\terr := task.db.DropAllTables()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif !config.DisableVersioning {\n\t\tlog.Printf(\"Setting up version tables\\n\")\n\t\tif err := task.db.CreateSchemaVersionTables(); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tif len(config.SchemaFilePath) > 0 {\n\t\tfile, err := os.Open(config.SchemaFilePath)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tstmts, err := ParseFile(file)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tlog.Println(\"----- Creating types and tables -----\")\n\t\tfor _, stmt := range stmts {\n\t\t\tlog.Println(rmspaceRegex.ReplaceAllString(stmt, \" \"))\n\t\t\tif err := task.db.ExecDDLQuery(stmt); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tlog.Println(\"----- Done -----\")\n\t}\n\n\tif !config.DisableVersioning {\n\t\tlog.Printf(\"Setting initial schema version to %v\\n\", config.InitialVersion)\n\t\terr := task.db.UpdateSchemaVersion(config.InitialVersion, config.InitialVersion)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tlog.Printf(\"Updating schema update log\\n\")\n\t\terr = task.db.WriteSchemaUpdateLog(\"0\", config.InitialVersion, \"\", \"initial version\")\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\n\tlog.Println(\"Schema setup complete\")\n\n\treturn nil\n}\n"
  },
  {
    "path": "tools/common/schema/types.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"fmt\"\n\t\"io/fs\"\n\t\"regexp\"\n)\n\n//go:generate mockgen -source $GOFILE -destination schema_client_mock.go -package schema github.com/uber/cadence/tools/common/schema SchemaClient\n\ntype (\n\t// ConfigError is an error type that\n\t// represents a problem with the config\n\tConfigError struct {\n\t\tmsg string\n\t}\n\t// UpdateConfig holds the config\n\t// params for executing a UpdateTask\n\tUpdateConfig struct {\n\t\tDBName        string\n\t\tTargetVersion string\n\t\tSchemaFS      fs.FS\n\t\tIsDryRun      bool\n\t}\n\t// SetupConfig holds the config\n\t// params need by the SetupTask\n\tSetupConfig struct {\n\t\tSchemaFilePath    string\n\t\tInitialVersion    string\n\t\tOverwrite         bool // overwrite previous data\n\t\tDisableVersioning bool // do not use schema versioning\n\t}\n\t// SchemaClient is the database interface that's required to be implemented\n\t// for the schema-tool to work\n\tSchemaClient interface {\n\t\t// ExecDDLQuery executes a schema statement\n\t\tExecDDLQuery(stmt string, args ...interface{}) error\n\t\t// DropAllTables drops all tables\n\t\tDropAllTables() error\n\t\t// CreateSchemaVersionTables sets up the schema version tables\n\t\tCreateSchemaVersionTables() error\n\t\t// ReadSchemaVersion returns the current schema version for the keyspace\n\t\tReadSchemaVersion() (string, error)\n\t\t// UpdateSchemaVersion updates the schema version for the keyspace\n\t\tUpdateSchemaVersion(newVersion string, minCompatibleVersion string) error\n\t\t// WriteSchemaUpdateLog adds an entry to the schema update history table\n\t\tWriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error\n\t\t// Close gracefully closes the client object\n\t\tClose()\n\t}\n)\n\nconst (\n\t// CLIOptEndpoint is the cli option for endpoint\n\tCLIOptEndpoint = \"endpoint\"\n\t// CLIOptPort is the cli option for port\n\tCLIOptPort = \"port\"\n\t// CLIOptUser is the cli option for user\n\tCLIOptUser = \"user\"\n\t// CLIOptPassword is the cli option for password\n\tCLIOptPassword = \"password\"\n\t// CLIOptAllowedAuthenticatorsis the cli option for cassandra allowed authenticators\n\tCLIOptAllowedAuthenticators = \"allowed-authenticators\"\n\t// CLIOptTimeout is the cli option for timeout\n\tCLIOptTimeout = \"timeout\"\n\t// CLIOptConnectTimeout is the cli option for connection timeout\n\tCLIOptConnectTimeout = \"connect-timeout\"\n\t// CLIOptKeyspace is the cli option for keyspace\n\tCLIOptKeyspace = \"keyspace\"\n\t// CLIOptDatabase is the cli option for datacenter\n\tCLIOptDatacenter = \"datacenter\"\n\t// CLIOptDatabase is the cli option for database\n\tCLIOptDatabase = \"database\"\n\t// CLIOptPluginName is the cli option for plugin name\n\tCLIOptPluginName = \"plugin\"\n\t// CLIOptConnectAttributes is the cli option for connect attributes (key/values via a url query string)\n\tCLIOptConnectAttributes = \"connect-attributes\"\n\t// CLIOptVersion is the cli option for version\n\tCLIOptVersion = \"version\"\n\t// CLIOptSchemaFile is the cli option for schema file\n\tCLIOptSchemaFile = \"schema-file\"\n\t// CLIOptOverwrite is the cli option for overwrite\n\tCLIOptOverwrite = \"overwrite\"\n\t// CLIOptDisableVersioning is the cli option to disabling versioning\n\tCLIOptDisableVersioning = \"disable-versioning\"\n\t// CLIOptTargetVersion is the cli option for target version\n\tCLIOptTargetVersion = \"version\"\n\t// CLIOptDryrun is the cli option for enabling dryrun\n\tCLIOptDryrun = \"dryrun\"\n\t// CLIOptSchemaDir is the cli option for schema directory\n\tCLIOptSchemaDir = \"schema-dir\"\n\t// CLIOptReplicationFactor is the cli option for replication factor\n\tCLIOptReplicationFactor = \"replication-factor\"\n\t// CLIOptQuiet is the cli option for quiet mode\n\tCLIOptQuiet = \"quiet\"\n\t// CLIOptProtoVersion is the cli option for protocol version\n\tCLIOptProtoVersion = \"protocol-version\"\n\n\t// CLIFlagEndpoint is the cli flag for endpoint\n\tCLIFlagEndpoint = CLIOptEndpoint\n\t// CLIFlagPort is the cli flag for port\n\tCLIFlagPort = CLIOptPort\n\t// CLIFlagUser is the cli flag for user\n\tCLIFlagUser = CLIOptUser\n\t// CLIFlagPassword is the cli flag for password\n\tCLIFlagPassword = CLIOptPassword\n\t// CLIFlagAllowedAuthenticators is the cli flag for whitelisting custom authenticators\n\tCLIFlagAllowedAuthenticators = CLIOptAllowedAuthenticators\n\t// CLIFlagConnectTimeout is the cli flag for connection timeout\n\tCLIFlagConnectTimeout = CLIOptConnectTimeout\n\t// CLIFlagTimeout is the cli flag for timeout\n\tCLIFlagTimeout = CLIOptTimeout\n\t// CLIFlagKeyspace is the cli flag for keyspace\n\tCLIFlagKeyspace = CLIOptKeyspace\n\t// CLIFlagDatacenter is the cli flag for datacenter\n\tCLIFlagDatacenter = CLIOptDatacenter\n\t// CLIFlagDatabase is the cli flag for database\n\tCLIFlagDatabase = CLIOptDatabase\n\t// CLIFlagPluginName is the cli flag for plugin name\n\tCLIFlagPluginName = CLIOptPluginName\n\t// CLIFlagConnectAttributes allows arbitrary connect attributes\n\tCLIFlagConnectAttributes = CLIOptConnectAttributes\n\t// CLIFlagVersion is the cli flag for version\n\tCLIFlagVersion = CLIOptVersion\n\t// CLIFlagSchemaFile is the cli flag for schema file\n\tCLIFlagSchemaFile = CLIOptSchemaFile\n\t// CLIFlagOverwrite is the cli flag for overwrite\n\tCLIFlagOverwrite = CLIOptOverwrite\n\t// CLIFlagDisableVersioning is the cli flag for disabling versioning\n\tCLIFlagDisableVersioning = CLIOptDisableVersioning\n\t// CLIFlagTargetVersion is the cli flag for target version\n\tCLIFlagTargetVersion = CLIOptTargetVersion\n\t// CLIFlagDryrun is the cli flag for dryrun\n\tCLIFlagDryrun = CLIOptDryrun\n\t// CLIFlagSchemaDir is the cli flag for schema directory\n\tCLIFlagSchemaDir = CLIOptSchemaDir\n\t// CLIFlagReplicationFactor is the cli flag for replication factor\n\tCLIFlagReplicationFactor = CLIOptReplicationFactor\n\t// CLIFlagQuiet is the cli flag for quiet mode\n\tCLIFlagQuiet = CLIOptQuiet\n\t// CLIFlagProtoVersion is the cli flag for protocol version\n\tCLIFlagProtoVersion = CLIOptProtoVersion\n\n\t// CLIFlagEnableTLS enables cassandra client TLS\n\tCLIFlagEnableTLS = \"tls\"\n\t// CLIFlagTLSCertFile is the optional tls cert file (tls must be enabled)\n\tCLIFlagTLSCertFile = \"tls-cert-file\"\n\t// CLIFlagTLSKeyFile is the optional tls key file (tls must be enabled)\n\tCLIFlagTLSKeyFile = \"tls-key-file\"\n\t// CLIFlagTLSCaFile is the optional tls CA file (tls must be enabled)\n\tCLIFlagTLSCaFile = \"tls-ca-file\"\n\t// CLIFlagTLSEnableHostVerification enables tls host verification (tls must be enabled)\n\tCLIFlagTLSEnableHostVerification = \"tls-enable-host-verification\"\n\t// CLIFlagTLSServerName is the Server Name Indication to verify the hostname on the returned certificates.\n\t// It is also included in the client's handshake to support virtual hosting unless it is an IP address.\n\tCLIFlagTLSServerName = \"tls-server-name\"\n)\n\nvar rmspaceRegex = regexp.MustCompile(`\\s+`)\n\n// NewConfigError creates and returns an instance of ConfigError\nfunc NewConfigError(msg string) error {\n\treturn &ConfigError{msg: msg}\n}\n\n// Error returns a string representation of this error\nfunc (e *ConfigError) Error() string {\n\treturn fmt.Sprintf(\"Config Error:%v\", e.msg)\n}\n"
  },
  {
    "path": "tools/common/schema/updatetask.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t// In this context md5 is just used for versioning the current schema. It is a weak cryptographic primitive and\n\t// should not be used for anything more important (password hashes etc.). Marking it as #nosec because of how it's\n\t// being used.\n\t\"crypto/md5\" // #nosec\n\t\"encoding/hex\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"io/fs\"\n\t\"log\"\n\t\"sort\"\n\t\"strings\"\n\t\"time\"\n)\n\ntype (\n\t// UpdateTask represents a task\n\t// that executes a cassandra schema upgrade\n\tUpdateTask struct {\n\t\tdb     SchemaClient\n\t\tconfig *UpdateConfig\n\t}\n\n\t// manifest is a value type that represents\n\t// the deserialized manifest.json file within\n\t// a schema version directory\n\tmanifest struct {\n\t\tCurrVersion          string\n\t\tMinCompatibleVersion string\n\t\tDescription          string\n\t\tSchemaUpdateCqlFiles []string\n\t\tmd5                  string\n\t}\n\n\t// ChangeSet represents all the changes\n\t// corresponding to a single schema version\n\tChangeSet struct {\n\t\tVersion  string\n\t\tmanifest *manifest\n\t\tCqlStmts []string\n\t}\n\n\t// byVersion is a comparator type\n\t// for sorting a set of version\n\t// strings\n\tbyVersion []string\n\n\t// squashVersion represents a squashed statement batch\n\t// for a shortcut between the versions\n\tsquashVersion struct {\n\t\tprev    string\n\t\tver     string\n\t\tdirName string\n\t}\n)\n\nconst (\n\tmanifestFileName = \"manifest.json\"\n)\n\nvar (\n\twhitelistedCQLPrefixes = [4]string{\"CREATE\", \"ALTER\", \"INSERT\", \"DROP\"}\n)\n\n// NewUpdateSchemaTask returns a new instance of UpdateTask.\nfunc NewUpdateSchemaTask(db SchemaClient, config *UpdateConfig) *UpdateTask {\n\treturn &UpdateTask{\n\t\tdb:     db,\n\t\tconfig: config,\n\t}\n}\n\n// Run executes the task\nfunc (task *UpdateTask) Run() error {\n\tconfig := task.config\n\n\tlog.Printf(\"UpdateSchemeTask started, config=%+v\\n\", config)\n\n\tcurrVer, err := task.db.ReadSchemaVersion()\n\tif err != nil {\n\t\treturn fmt.Errorf(\"error reading current schema version:%v\", err.Error())\n\t}\n\n\tupdates, err := task.BuildChangeSet(currVer)\n\tif err != nil {\n\t\treturn err\n\t}\n\n\tif config.IsDryRun {\n\t\tlog.Println(\"In DryRun mode, this command will only print queries without executing.....\")\n\t\tif len(updates) == 0 {\n\t\t\tlog.Println(\"Found zero updates to run\")\n\t\t}\n\t\tfor _, upd := range updates {\n\t\t\tlog.Printf(\"DryRun of updating to version: %s, manifest: %s \\n\", upd.Version, upd.manifest)\n\t\t\tfor _, stmt := range upd.CqlStmts {\n\t\t\t\tlog.Printf(\"DryRun query:%s \\n\", stmt)\n\t\t\t}\n\t\t}\n\t} else {\n\t\terr = task.executeUpdates(currVer, updates)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tlog.Printf(\"UpdateSchemeTask done\\n\")\n\n\treturn nil\n}\n\nfunc (task *UpdateTask) executeUpdates(currVer string, updates []ChangeSet) error {\n\n\tif len(updates) == 0 {\n\t\tlog.Printf(\"found zero updates from current version %v\", currVer)\n\t\treturn nil\n\t}\n\tupdStart := time.Now()\n\tfor _, cs := range updates {\n\t\tcsStart := time.Now()\n\n\t\terr := task.execCQLStmts(cs.Version, cs.CqlStmts)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = task.updateSchemaVersion(currVer, &cs)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\n\t\tlog.Printf(\"Schema updated from %v to %v, elapsed %v\\n\", currVer, cs.Version, time.Since(csStart))\n\t\tcurrVer = cs.Version\n\t}\n\n\tlog.Printf(\"All schema changes completed in %v\\n\", time.Since(updStart))\n\n\treturn nil\n}\n\nfunc (task *UpdateTask) execCQLStmts(ver string, stmts []string) error {\n\tlog.Printf(\"---- Executing updates for version %v ----\\n\", ver)\n\tfor _, stmt := range stmts {\n\t\tlog.Println(rmspaceRegex.ReplaceAllString(stmt, \" \"))\n\t\te := task.db.ExecDDLQuery(stmt)\n\t\tif e != nil {\n\t\t\treturn fmt.Errorf(\"error executing CQL statement:%v\", e)\n\t\t}\n\t}\n\tlog.Printf(\"---- Done ----\\n\")\n\treturn nil\n}\n\nfunc (task *UpdateTask) updateSchemaVersion(oldVer string, cs *ChangeSet) error {\n\n\terr := task.db.UpdateSchemaVersion(cs.Version, cs.manifest.MinCompatibleVersion)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to update schema_version table, err=%v\", err.Error())\n\t}\n\n\terr = task.db.WriteSchemaUpdateLog(oldVer, cs.manifest.CurrVersion, cs.manifest.md5, cs.manifest.Description)\n\tif err != nil {\n\t\treturn fmt.Errorf(\"failed to add entry to schema_update_history, err=%v\", err.Error())\n\t}\n\n\treturn nil\n}\n\nfunc (task *UpdateTask) BuildChangeSet(currVer string) ([]ChangeSet, error) {\n\n\tconfig := task.config\n\n\tverDirs, err := readSchemaDir(config.SchemaFS, currVer, config.TargetVersion)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"error listing schema dir:%v\", err.Error())\n\t}\n\n\tvar result []ChangeSet\n\n\tfor _, vd := range verDirs {\n\n\t\tm, e := readManifest(config.SchemaFS, vd)\n\t\tif e != nil {\n\t\t\treturn nil, fmt.Errorf(\"error processing manifest for version %v:%v\", vd, e.Error())\n\t\t}\n\n\t\tif squashVersionStrRegex.MatchString(vd) {\n\t\t\t_, v := squashDirToVersion(vd)\n\t\t\tif m.CurrVersion != v {\n\t\t\t\treturn nil, fmt.Errorf(\"manifest version doesn't match with dirname, dir=%v,manifest.version=%v\",\n\t\t\t\t\tvd, m.CurrVersion)\n\t\t\t}\n\t\t} else if m.CurrVersion != dirToVersion(vd) {\n\t\t\treturn nil, fmt.Errorf(\"manifest version doesn't match with dirname, dir=%v,manifest.version=%v\",\n\t\t\t\tvd, m.CurrVersion)\n\t\t}\n\n\t\tstmts, e := task.parseSQLStmts(vd, m)\n\t\tif e != nil {\n\t\t\treturn nil, e\n\t\t}\n\n\t\te = validateCQLStmts(stmts)\n\t\tif e != nil {\n\t\t\treturn nil, fmt.Errorf(\"error processing version %v:%v\", vd, e.Error())\n\t\t}\n\n\t\tcs := ChangeSet{}\n\t\tcs.manifest = m\n\t\tcs.CqlStmts = stmts\n\t\tcs.Version = m.CurrVersion\n\t\tresult = append(result, cs)\n\t}\n\n\treturn result, nil\n}\n\nfunc (task *UpdateTask) parseSQLStmts(dir string, manifest *manifest) ([]string, error) {\n\n\tresult := make([]string, 0, 4)\n\n\tfor _, file := range manifest.SchemaUpdateCqlFiles {\n\t\tpath := dir + \"/\" + file\n\t\tf, err := task.config.SchemaFS.Open(path)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"error opening file %v, err=%v\", path, err)\n\t\t}\n\t\tstmts, err := ParseFile(f)\n\t\tif err != nil {\n\t\t\treturn nil, fmt.Errorf(\"error parsing file %v, err=%v\", path, err)\n\t\t}\n\t\tresult = append(result, stmts...)\n\t}\n\n\tif len(result) == 0 {\n\t\treturn nil, fmt.Errorf(\"found 0 updates in dir %v\", dir)\n\t}\n\n\treturn result, nil\n}\n\nfunc validateCQLStmts(stmts []string) error {\n\tfor _, stmt := range stmts {\n\t\tvalid := false\n\t\tfor _, prefix := range whitelistedCQLPrefixes {\n\t\t\tif strings.HasPrefix(stmt, prefix) {\n\t\t\t\tvalid = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tif !valid {\n\t\t\treturn fmt.Errorf(\"CQL prefix not in whitelist, stmt=%v\", stmt)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc readManifest(fileSystem fs.FS, subdir string) (*manifest, error) {\n\tfsys, err := fs.Sub(fileSystem, subdir)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tfile, err := fsys.Open(manifestFileName)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tjsonBlob, err := io.ReadAll(file)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar manifest manifest\n\terr = json.Unmarshal(jsonBlob, &manifest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tcurrVer, err := parseValidateVersion(manifest.CurrVersion)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"invalid CurrVersion in manifest\")\n\t}\n\tmanifest.CurrVersion = currVer\n\n\tminVer, err := parseValidateVersion(manifest.MinCompatibleVersion)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif len(manifest.MinCompatibleVersion) == 0 {\n\t\treturn nil, fmt.Errorf(\"invalid MinCompatibleVersion in manifest\")\n\t}\n\tmanifest.MinCompatibleVersion = minVer\n\n\tif len(manifest.SchemaUpdateCqlFiles) == 0 {\n\t\treturn nil, fmt.Errorf(\"manifest missing SchemaUpdateCqlFiles\")\n\t}\n\n\t// See comment above. This is an appropriate usage of md5.\n\t// #nosec\n\tmd5Bytes := md5.Sum(jsonBlob)\n\tmanifest.md5 = hex.EncodeToString(md5Bytes[:])\n\n\treturn &manifest, nil\n}\n\n// readSchemaDir returns a sorted list of subdir names that hold\n// the schema changes for versions in the range startVer < ver <= endVer\n// when endVer is empty this method returns all subdir names that are greater than startVer\n// this method has an assumption that the subdirs containing the\n// schema changes will be of the form vx.x, where x.x is the version\n// returns error when\n//   - startVer >= endVer\n//   - endVer is empty and no subdirs have version >= startVer\n//   - endVer is non-empty and subdir with version == endVer is not found\nfunc readSchemaDir(fileSystem fs.FS, startVer string, endVer string) ([]string, error) {\n\tsubdirs, err := fs.ReadDir(fileSystem, \".\")\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tvar dirNames []string\n\tfor _, dir := range subdirs {\n\t\tif dir.IsDir() {\n\t\t\tdirNames = append(dirNames, dir.Name())\n\t\t}\n\t}\n\n\tresult, squashes, err := filterDirectories(dirNames, startVer, endVer)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\tif len(squashes) == 0 || len(result) == 0 {\n\t\t// if no shortcuts are found between the versions,\n\t\t// apply them one by one incrementally\n\t\treturn result, nil\n\t}\n\n\treturn findShortestPath(startVer, dirToVersion(result[len(result)-1]), result, squashes)\n}\n\nfunc filterDirectories(dirNames []string, startVer string, endVer string) ([]string, []squashVersion, error) {\n\tvar endFound bool\n\tvar highestVer string\n\tvar result []string\n\tvar squashes []squashVersion\n\thasEndVer := len(endVer) > 0\n\n\tif hasEndVer && cmpVersion(startVer, endVer) >= 0 {\n\t\treturn nil, nil, fmt.Errorf(\"startVer (%v) must be less than endVer (%v)\", startVer, endVer)\n\t}\n\n\tfor _, dirname := range dirNames {\n\n\t\tvar prev, ver string\n\t\tif versionStrRegex.MatchString(dirname) {\n\t\t\tver = dirToVersion(dirname)\n\t\t} else if squashVersionStrRegex.MatchString(dirname) {\n\t\t\tprev, ver = squashDirToVersion(dirname)\n\t\t\tif cmpVersion(prev, ver) >= 0 {\n\t\t\t\treturn nil, nil, fmt.Errorf(\"invalid squashed version %q, %v >= %v\", dirname, prev, ver)\n\t\t\t}\n\t\t} else {\n\t\t\tcontinue\n\t\t}\n\n\t\tif len(highestVer) == 0 || cmpVersion(ver, highestVer) > 0 {\n\t\t\thighestVer = ver\n\t\t}\n\n\t\thighcmp := 0\n\t\tlowcmp := cmpVersion(ver, startVer)\n\t\tif hasEndVer {\n\t\t\thighcmp = cmpVersion(ver, endVer)\n\t\t}\n\n\t\tif lowcmp <= 0 || highcmp > 0 {\n\t\t\tcontinue // out of range\n\t\t}\n\n\t\tif len(prev) > 0 && cmpVersion(prev, startVer) < 0 {\n\t\t\tcontinue // out of range\n\t\t}\n\n\t\tendFound = endFound || (highcmp == 0)\n\t\tif len(prev) == 0 {\n\t\t\tresult = append(result, dirname)\n\t\t} else {\n\t\t\tsquashes = append(squashes, squashVersion{prev: prev, ver: ver, dirName: dirname})\n\t\t}\n\t}\n\n\t// when endVer is specified, atleast one result MUST be found since startVer < endVer\n\tif hasEndVer && !endFound {\n\t\treturn nil, nil, fmt.Errorf(\"version dir not found for target version %v\", endVer)\n\t}\n\n\t// when endVer is empty and no result is found, then the highest version\n\t// found must be equal to startVer, else return error\n\tif !hasEndVer && len(result) == 0 && len(squashes) == 0 {\n\t\tif len(highestVer) == 0 || cmpVersion(startVer, highestVer) != 0 {\n\t\t\treturn nil, nil, fmt.Errorf(\"no subdirs found with version >= %v\", startVer)\n\t\t}\n\t\treturn result, nil, nil\n\t}\n\n\tsort.Sort(byVersion(result))\n\n\treturn result, squashes, nil\n}\n\nfunc dirToVersion(dir string) string {\n\treturn dir[1:]\n}\n\nfunc squashDirToVersion(dir string) (string, string) {\n\tsplits := strings.Split(dir[1:], \"-\")\n\treturn splits[0], splits[1]\n}\n\nfunc (v byVersion) Len() int {\n\treturn len(v)\n}\n\nfunc (v byVersion) Less(i, j int) bool {\n\tv1 := dirToVersion(v[i])\n\tv2 := dirToVersion(v[j])\n\treturn cmpVersion(v1, v2) < 0\n}\n\nfunc (v byVersion) Swap(i, j int) {\n\tv[i], v[j] = v[j], v[i]\n}\n"
  },
  {
    "path": "tools/common/schema/updatetask_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io/fs\"\n\t\"io/ioutil\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n\t\"go.uber.org/mock/gomock\"\n\n\t\"github.com/uber/cadence/schema/cassandra\"\n\t\"github.com/uber/cadence/schema/mysql\"\n\t\"github.com/uber/cadence/schema/postgres\"\n\t\"github.com/uber/cadence/schema/sqlite\"\n)\n\ntype UpdateTaskTestSuite struct {\n\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\tsuite.Suite\n}\n\nfunc TestUpdateTaskTestSuite(t *testing.T) {\n\tsuite.Run(t, new(UpdateTaskTestSuite))\n}\n\nfunc (s *UpdateTaskTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T())\n}\n\nfunc (s *UpdateTaskTestSuite) TestReadSchemaDir() {\n\n\temptyDir := s.T().TempDir()\n\ttmpDir := s.T().TempDir()\n\tsquashDir := s.T().TempDir()\n\n\tsubDirs := []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\"}\n\tfor _, d := range subDirs {\n\t\ts.NoError(os.Mkdir(tmpDir+\"/\"+d, os.FileMode(0444)))\n\t}\n\n\tsquashDirs := []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\", \"s0.5-10.2\", \"s1.5-3.5\"}\n\tfor _, d := range squashDirs {\n\t\ts.NoError(os.Mkdir(squashDir+\"/\"+d, os.FileMode(0444)))\n\t}\n\n\t_, err := readSchemaDir(os.DirFS(tmpDir), \"11.0\", \"11.2\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(tmpDir), \"0.5\", \"10.3\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(tmpDir), \"1.5\", \"1.5\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(tmpDir), \"1.5\", \"0.5\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(tmpDir), \"10.3\", \"\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(emptyDir), \"11.0\", \"\")\n\ts.Error(err)\n\t_, err = readSchemaDir(os.DirFS(emptyDir), \"10.1\", \"\")\n\ts.Error(err)\n\n\tans, err := readSchemaDir(os.DirFS(tmpDir), \"0.4\", \"10.2\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\"}, ans)\n\n\tans, err = readSchemaDir(os.DirFS(tmpDir), \"0.5\", \"3.5\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v1.5\", \"v2.5\", \"v3.5\"}, ans)\n\n\tans, err = readSchemaDir(os.DirFS(tmpDir), \"10.2\", \"\")\n\ts.NoError(err)\n\ts.Equal(0, len(ans))\n\n\tans, err = readSchemaDir(os.DirFS(squashDir), \"0.4\", \"10.2\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.5\", \"s0.5-10.2\"}, ans)\n\n\tans, err = readSchemaDir(os.DirFS(squashDir), \"0.5\", \"3.5\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v1.5\", \"s1.5-3.5\"}, ans)\n\n\tans, err = readSchemaDir(os.DirFS(squashDir), \"10.2\", \"\")\n\ts.NoError(err)\n\ts.Empty(ans)\n}\n\nfunc (s *UpdateTaskTestSuite) TestReadSchemaDirFromEmbeddings() {\n\t// Cassandra\n\tfsys, err := fs.Sub(cassandra.SchemaFS, \"cadence/versioned\")\n\ts.NoError(err)\n\tans, err := readSchemaDir(fsys, \"0.30\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.31\", \"v0.32\", \"v0.33\", \"v0.34\", \"v0.35\", \"v0.36\", \"v0.37\", \"v0.38\", \"v0.39\", \"v0.40\", \"v0.41\", \"v0.42\", \"v0.43\", \"v0.44\", \"v0.45\", \"v0.46\"}, ans)\n\n\tfsys, err = fs.Sub(cassandra.SchemaFS, \"visibility/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.6\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.7\", \"v0.8\", \"v0.9\", \"v0.10\"}, ans)\n\n\t// MySQL\n\tfsys, err = fs.Sub(mysql.SchemaFS, \"v8/cadence/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.3\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.4\", \"v0.5\", \"v0.6\", \"v0.7\"}, ans)\n\n\tfsys, err = fs.Sub(mysql.SchemaFS, \"v8/visibility/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.5\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.6\", \"v0.7\", \"v0.8\"}, ans)\n\n\t// SQLite\n\tfsys, err = fs.Sub(sqlite.SchemaFS, \"cadence/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.1\", \"\")\n\ts.NoError(err)\n\ts.Nil(ans, \"no version dirs found after 0.1\")\n\n\tfsys, err = fs.Sub(sqlite.SchemaFS, \"visibility/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.1\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.2\"}, ans)\n\n\t// Postgres\n\tfsys, err = fs.Sub(postgres.SchemaFS, \"cadence/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.3\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.4\", \"v0.5\", \"v0.6\", \"v0.7\"}, ans)\n\n\tfsys, err = fs.Sub(postgres.SchemaFS, \"visibility/versioned\")\n\ts.NoError(err)\n\tans, err = readSchemaDir(fsys, \"0.5\", \"\")\n\ts.NoError(err)\n\ts.Equal([]string{\"v0.6\", \"v0.7\", \"v0.8\", \"v0.9\"}, ans)\n}\n\nfunc (s *UpdateTaskTestSuite) TestReadManifest() {\n\n\ttmpDir := s.T().TempDir()\n\n\tinput := `{\n\t\t\"CurrVersion\": \"0.4\",\n\t\t\"MinCompatibleVersion\": \"0.1\",\n\t\t\"Description\": \"base version of schema\",\n\t\t\"SchemaUpdateCqlFiles\": [\"base1.cql\", \"base2.cql\", \"base3.cql\"]\n\t}`\n\tfiles := []string{\"base1.cql\", \"base2.cql\", \"base3.cql\"}\n\ts.runReadManifestTest(tmpDir, input, \"0.4\", \"0.1\", \"base version of schema\", files, false)\n\n\terrInputs := []string{\n\t\t`{\n\t\t\t\"MinCompatibleVersion\": \"0.1\",\n\t\t\t\"Description\": \"base\",\n\t\t\t\"SchemaUpdateCqlFiles\": [\"base1.cql\"]\n\t\t }`,\n\t\t`{\n\t\t\t\"CurrVersion\": \"0.4\",\n\t\t\t\"Description\": \"base version of schema\",\n\t\t\t\"SchemaUpdateCqlFiles\": [\"base1.cql\", \"base2.cql\", \"base3.cql\"]\n\t\t }`,\n\t\t`{\n\t\t\t\"CurrVersion\": \"0.4\",\n\t\t\t\"MinCompatibleVersion\": \"0.1\",\n\t\t\t\"Description\": \"base version of schema\",\n\t\t }`,\n\t\t`{\n\t\t\t\"CurrVersion\": \"\",\n\t\t\t\"MinCompatibleVersion\": \"0.1\",\n\t\t\t\"Description\": \"base version of schema\",\n\t\t\t\"SchemaUpdateCqlFiles\": [\"base1.cql\", \"base2.cql\", \"base3.cql\"]\n\t\t }`,\n\t\t`{\n\t\t\t\"CurrVersion\": \"0.4\",\n\t\t\t\"MinCompatibleVersion\": \"\",\n\t\t\t\"Description\": \"base version of schema\",\n\t\t\t\"SchemaUpdateCqlFiles\": [\"base1.cql\", \"base2.cql\", \"base3.cql\"]\n\t\t }`,\n\t\t`{\n\t\t\t\"CurrVersion\": \"\",\n\t\t\t\"MinCompatibleVersion\": \"0.1\",\n\t\t\t\"Description\": \"base version of schema\",\n\t\t\t\"SchemaUpdateCqlFiles\": []\n\t\t }`,\n\t}\n\n\tfor _, in := range errInputs {\n\t\ts.runReadManifestTest(tmpDir, in, \"\", \"\", \"\", nil, true)\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) TestFilterDirectories() {\n\ttests := []struct {\n\t\tname       string\n\t\tstartVer   string\n\t\tendVer     string\n\t\twantErr    string\n\t\temptyDir   bool\n\t\twantResult []string\n\t}{\n\t\t{name: \"endVer highter\", startVer: \"11.0\", endVer: \"11.2\", wantErr: \"version dir not found for target version 11.2\"},\n\t\t{name: \"both outside\", startVer: \"0.5\", endVer: \"10.3\", wantErr: \"version dir not found for target version 10.3\"},\n\t\t{name: \"versions equal\", startVer: \"1.5\", endVer: \"1.5\", wantErr: \"startVer (1.5) must be less than endVer (1.5)\"},\n\t\t{name: \"endVer < startVer\", startVer: \"1.5\", endVer: \"0.5\", wantErr: \"startVer (1.5) must be less than endVer (0.5)\"},\n\t\t{name: \"startVer highter\", startVer: \"10.3\", wantErr: \"no subdirs found with version >= 10.3\"},\n\t\t{name: \"empty set 1\", startVer: \"11.0\", wantErr: \"no subdirs found with version >= 11.0\", emptyDir: true},\n\t\t{name: \"empty set 2\", startVer: \"10.1\", wantErr: \"no subdirs found with version >= 10.1\", emptyDir: true},\n\t\t{name: \"all versions\", startVer: \"0.4\", endVer: \"10.2\", wantResult: []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\"}},\n\t\t{name: \"subset\", startVer: \"0.5\", endVer: \"3.5\", wantResult: []string{\"v1.5\", \"v2.5\", \"v3.5\"}},\n\t\t{name: \"already at last version\", startVer: \"10.2\"},\n\t}\n\tsubDirs := []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\"}\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\tvar dirs []string\n\t\t\tif !tt.emptyDir {\n\t\t\t\tdirs = subDirs\n\t\t\t}\n\t\t\tr, sq, err := filterDirectories(dirs, tt.startVer, tt.endVer)\n\t\t\ts.Empty(sq)\n\t\t\tif tt.wantErr != \"\" {\n\t\t\t\ts.Empty(r)\n\t\t\t\ts.EqualError(err, tt.wantErr)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(tt.wantResult, r)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) TestFilterSquashDirectories() {\n\ttests := []struct {\n\t\tname       string\n\t\tstartVer   string\n\t\tendVer     string\n\t\twantErr    string\n\t\twantResult []string\n\t\twantSq     []squashVersion\n\t\tcustomDirs []string\n\t}{\n\t\t{name: \"endVer highter\", startVer: \"11.0\", endVer: \"11.2\", wantErr: \"version dir not found for target version 11.2\"},\n\t\t{name: \"both outside\", startVer: \"0.5\", endVer: \"10.3\", wantErr: \"version dir not found for target version 10.3\"},\n\t\t{name: \"versions equal\", startVer: \"1.5\", endVer: \"1.5\", wantErr: \"startVer (1.5) must be less than endVer (1.5)\"},\n\t\t{name: \"endVer < startVer\", startVer: \"1.5\", endVer: \"0.5\", wantErr: \"startVer (1.5) must be less than endVer (0.5)\"},\n\t\t{name: \"startVer highter\", startVer: \"10.3\", wantErr: \"no subdirs found with version >= 10.3\"},\n\t\t{\n\t\t\tname:       \"backward version squash\",\n\t\t\tstartVer:   \"1.5\",\n\t\t\twantErr:    \"invalid squashed version \\\"s3.0-2.0\\\", 3.0 >= 2.0\",\n\t\t\tcustomDirs: []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\", \"s0.5-10.2\", \"s1.5-3.5\", \"s3.0-2.0\"},\n\t\t},\n\t\t{\n\t\t\tname:       \"equal version squash\",\n\t\t\tstartVer:   \"1.5\",\n\t\t\twantErr:    \"invalid squashed version \\\"s2.0-2.0\\\", 2.0 >= 2.0\",\n\t\t\tcustomDirs: []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\", \"s0.5-10.2\", \"s1.5-3.5\", \"s2.0-2.0\"},\n\t\t},\n\t\t{\n\t\t\tname:       \"all versions\",\n\t\t\tstartVer:   \"0.4\",\n\t\t\tendVer:     \"10.2\",\n\t\t\twantResult: []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\"},\n\t\t\twantSq: []squashVersion{\n\t\t\t\t{prev: \"1.5\", ver: \"3.5\", dirName: \"s1.5-3.5\"},\n\t\t\t\t{prev: \"0.5\", ver: \"10.2\", dirName: \"s0.5-10.2\"},\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tname:       \"subset\",\n\t\t\tstartVer:   \"0.5\",\n\t\t\tendVer:     \"3.5\",\n\t\t\twantResult: []string{\"v1.5\", \"v2.5\", \"v3.5\"},\n\t\t\twantSq: []squashVersion{\n\t\t\t\t{prev: \"1.5\", ver: \"3.5\", dirName: \"s1.5-3.5\"},\n\t\t\t},\n\t\t},\n\t\t{name: \"already at last version\", startVer: \"10.2\"},\n\t}\n\tsubDirs := []string{\"v0.5\", \"v1.5\", \"v2.5\", \"v3.5\", \"v10.2\", \"abc\", \"2.0\", \"3.0\", \"s0.5-10.2\", \"s1.5-3.5\"}\n\tfor _, tt := range tests {\n\t\ts.Run(tt.name, func() {\n\t\t\tdirs := subDirs\n\t\t\tif len(tt.customDirs) > 0 {\n\t\t\t\tdirs = tt.customDirs\n\t\t\t}\n\t\t\tr, sq, err := filterDirectories(dirs, tt.startVer, tt.endVer)\n\t\t\tif tt.wantErr != \"\" {\n\t\t\t\ts.Empty(r)\n\t\t\t\ts.Empty(sq)\n\t\t\t\ts.EqualError(err, tt.wantErr)\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(tt.wantResult, r)\n\t\t\t\ts.ElementsMatch(tt.wantSq, sq)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) TestSquashDirToVersion() {\n\ttests := []struct {\n\t\tsource string\n\t\tprev   string\n\t\tver    string\n\t}{\n\t\t{source: \"s0.0-0.0\", prev: \"0.0\", ver: \"0.0\"},\n\t\t{source: \"s1.0-0.0\", prev: \"1.0\", ver: \"0.0\"},\n\t\t{source: \"s1.0-2\", prev: \"1.0\", ver: \"2\"},\n\t\t{source: \"s1-2.1\", prev: \"1\", ver: \"2.1\"},\n\t\t{source: \"s1-2\", prev: \"1\", ver: \"2\"},\n\t}\n\tfor _, tt := range tests {\n\t\ts.Run(tt.source, func() {\n\t\t\tprev, ver := squashDirToVersion(tt.source)\n\t\t\ts.Equal(tt.prev, prev)\n\t\t\ts.Equal(tt.ver, ver)\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) TestNewUpdateSchemaTask() {\n\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\tcfg := &UpdateConfig{}\n\n\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\ts.NotNil(updateSchemaTask)\n\ts.Equal(mockSchemaClient, updateSchemaTask.db)\n\ts.Equal(cfg, updateSchemaTask.config)\n}\n\nfunc (s *UpdateTaskTestSuite) TestRun() {\n\ttargetVersion := \"0.4\"\n\n\ttests := map[string]struct {\n\t\tsetupMock     func(client *MockSchemaClient)\n\t\tdryRun        bool\n\t\ttargetVersion string\n\t\tdir           string\n\t\terr           error\n\t}{\n\t\t\"error - ReadSchemaVersion\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"\", assert.AnError).Times(1)\n\t\t\t},\n\t\t\tdir: \"v0.4\",\n\t\t\terr: assert.AnError,\n\t\t},\n\t\t\"error - BuildChangeSet\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"1.0\", nil).Times(1)\n\t\t\t},\n\t\t\tdir:           \"v0.4\",\n\t\t\ttargetVersion: targetVersion,\n\t\t\terr:           fmt.Errorf(\"startVer (1.0) must be less than endVer (%v)\", targetVersion),\n\t\t},\n\t\t\"error - executeUpdates\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"0.3\", nil).Times(1)\n\t\t\t\tclient.EXPECT().ExecDDLQuery(gomock.Any()).Return(assert.AnError).Times(1)\n\t\t\t},\n\t\t\tdir:           \"v0.4\",\n\t\t\ttargetVersion: targetVersion,\n\t\t\terr:           assert.AnError,\n\t\t},\n\t\t\"success - dry run\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"0.3\", nil).Times(1)\n\t\t\t},\n\t\t\tdir:           \"v0.4\",\n\t\t\tdryRun:        true,\n\t\t\ttargetVersion: targetVersion,\n\t\t\terr:           nil,\n\t\t},\n\t\t\"success - dry run no changes squashed version\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"0.4\", nil).Times(1)\n\t\t\t},\n\t\t\tdir:    \"s1-2\",\n\t\t\tdryRun: true,\n\t\t\terr:    nil,\n\t\t},\n\t\t\"success\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().ReadSchemaVersion().Return(\"0.3\", nil).Times(1)\n\t\t\t\tclient.EXPECT().ExecDDLQuery(gomock.Any()).Return(nil).Times(1)\n\t\t\t\tclient.EXPECT().UpdateSchemaVersion(targetVersion, targetVersion).Return(nil).Times(1)\n\t\t\t\tclient.EXPECT().WriteSchemaUpdateLog(\"0.3\", targetVersion, gomock.Any(), \"\").Return(nil).Times(1)\n\t\t\t},\n\t\t\tdir:           \"v0.4\",\n\t\t\ttargetVersion: targetVersion,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\t\t\ttest.setupMock(mockSchemaClient)\n\n\t\t\tdir := s.T().TempDir()\n\t\t\tsubDir := test.dir\n\t\t\ts.NoError(os.Mkdir(filepath.Join(dir, subDir), os.FileMode(0755)))\n\n\t\t\tmanifestFile := filepath.Join(dir, subDir, manifestFileName)\n\t\t\tm := manifest{\n\t\t\t\tCurrVersion:          targetVersion,\n\t\t\t\tMinCompatibleVersion: targetVersion,\n\t\t\t\tSchemaUpdateCqlFiles: []string{\"base.cql\"},\n\t\t\t}\n\t\t\tj, err := json.Marshal(m)\n\t\t\ts.NoError(err)\n\t\t\ts.NoError(os.WriteFile(manifestFile, j, os.FileMode(0644)))\n\t\t\tcqlFile := filepath.Join(dir, subDir, \"base.cql\")\n\t\t\ts.NoError(os.WriteFile(cqlFile, []byte(\"CREATE TABLE;\"), os.FileMode(0644)))\n\n\t\t\tcfg := &UpdateConfig{\n\t\t\t\tSchemaFS:      os.DirFS(dir),\n\t\t\t\tTargetVersion: test.targetVersion,\n\t\t\t\tIsDryRun:      test.dryRun,\n\t\t\t}\n\t\t\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\t\t\terr = updateSchemaTask.Run()\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) Test_executeUpdates() {\n\ttests := map[string]struct {\n\t\tsetupMock func(client *MockSchemaClient)\n\t\tupdates   []ChangeSet\n\t\terr       error\n\t}{\n\t\t\"no updates\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {},\n\t\t\terr:       nil,\n\t\t},\n\t\t\"error - updateSchemaVersion\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().UpdateSchemaVersion(\"0.3\", \"0.3\").Return(assert.AnError).Times(1)\n\t\t\t},\n\t\t\tupdates: []ChangeSet{\n\t\t\t\t{\n\t\t\t\t\tVersion: \"0.3\",\n\t\t\t\t\tmanifest: &manifest{\n\t\t\t\t\t\tCurrVersion:          \"0.3\",\n\t\t\t\t\t\tMinCompatibleVersion: \"0.3\",\n\t\t\t\t\t\tSchemaUpdateCqlFiles: []string{\"base.cql\"},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t\terr: assert.AnError,\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\t\t\ttest.setupMock(mockSchemaClient)\n\n\t\t\tcfg := &UpdateConfig{}\n\n\t\t\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\t\t\terr := updateSchemaTask.executeUpdates(\"0.3\", test.updates)\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) Test_updateSchemaVersion() {\n\tversion := \"0.3\"\n\n\ttests := map[string]struct {\n\t\tsetupMock func(client *MockSchemaClient)\n\t\terr       error\n\t}{\n\t\t\"error - UpdateSchemaVersion\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().UpdateSchemaVersion(version, version).Return(assert.AnError).Times(1)\n\t\t\t},\n\t\t\terr: assert.AnError,\n\t\t},\n\t\t\"error - WriteSchemaVersion\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().UpdateSchemaVersion(version, version).Return(nil).Times(1)\n\t\t\t\tclient.EXPECT().WriteSchemaUpdateLog(\"0.2\", version, \"md5\", \"description\").Return(assert.AnError).Times(1)\n\t\t\t},\n\t\t\terr: assert.AnError,\n\t\t},\n\t\t\"success\": {\n\t\t\tsetupMock: func(client *MockSchemaClient) {\n\t\t\t\tclient.EXPECT().UpdateSchemaVersion(version, version).Return(nil).Times(1)\n\t\t\t\tclient.EXPECT().WriteSchemaUpdateLog(\"0.2\", version, \"md5\", \"description\").Return(nil).Times(1)\n\t\t\t},\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\t\t\ttest.setupMock(mockSchemaClient)\n\n\t\t\tcfg := &UpdateConfig{}\n\n\t\t\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\t\t\tcs := &ChangeSet{\n\t\t\t\tVersion: version,\n\t\t\t\tmanifest: &manifest{\n\t\t\t\t\tCurrVersion:          version,\n\t\t\t\t\tMinCompatibleVersion: version,\n\t\t\t\t\tmd5:                  \"md5\",\n\t\t\t\t\tDescription:          \"description\",\n\t\t\t\t},\n\t\t\t}\n\n\t\t\terr := updateSchemaTask.updateSchemaVersion(\"0.2\", cs)\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) TestBuildChangeSet() {\n\ttests := map[string]struct {\n\t\tmanifestFile    string\n\t\tupdates         []ChangeSet\n\t\tmanifestVersion string\n\t\tcurrentVersion  string\n\t\tcqlFiles        []string\n\t\tdirs            []string\n\t\tcqlFileContent  string\n\t\terr             error\n\t}{\n\t\t\"error - readManifest\": {\n\t\t\tmanifestFile: \"wrong_manifest.json\",\n\t\t\tdirs:         []string{\"v0.4\"},\n\t\t\terr:          errors.New(\"no such file or directory\"),\n\t\t},\n\t\t\"error - squash version - manifest current version different from dir version\": {\n\t\t\tmanifestFile:    manifestFileName,\n\t\t\tdirs:            []string{\"s0.3-0.4\", \"v0.4\"},\n\t\t\tcurrentVersion:  \"0.3\",\n\t\t\tmanifestVersion: \"0.3\",\n\t\t\tcqlFiles:        []string{\"base.cql\"},\n\t\t\terr:             errors.New(\"manifest version doesn't match with dirname, dir=s0.3-0.4,manifest.version=0.3\"),\n\t\t},\n\t\t\"error - manifest current version different from dir version\": {\n\t\t\tmanifestFile:    manifestFileName,\n\t\t\tdirs:            []string{\"v0.4\"},\n\t\t\tcurrentVersion:  \"0.3\",\n\t\t\tmanifestVersion: \"0.3\",\n\t\t\tcqlFiles:        []string{\"base.cql\"},\n\t\t\terr:             errors.New(\"manifest version doesn't match with dirname, dir=v0.4,manifest.version=0.3\"),\n\t\t},\n\t\t\"error - parseSQLStmts\": {\n\t\t\tmanifestFile:    manifestFileName,\n\t\t\tdirs:            []string{\"v0.4\"},\n\t\t\tcurrentVersion:  \"0.3\",\n\t\t\tmanifestVersion: \"0.4\",\n\t\t\tcqlFiles:        []string{\"base.cql\", \"wrong.cql\"},\n\t\t\terr:             errors.New(\"error opening file v0.4/wrong.cql, err=open v0.4/wrong.cql: no such file or directory\"),\n\t\t},\n\t\t\"error - validateCQLStmts\": {\n\t\t\tmanifestFile:    manifestFileName,\n\t\t\tdirs:            []string{\"v0.4\"},\n\t\t\tcurrentVersion:  \"0.3\",\n\t\t\tmanifestVersion: \"0.4\",\n\t\t\tcqlFiles:        []string{\"base.cql\"},\n\t\t\tcqlFileContent:  \"WRONG CQL STATEMENT;\",\n\t\t\terr:             errors.New(\"error processing version v0.4:CQL prefix not in whitelist, stmt=WRONG CQL STATEMENT;\"),\n\t\t},\n\t\t\"success\": {\n\t\t\tmanifestFile:    manifestFileName,\n\t\t\tdirs:            []string{\"v0.4\"},\n\t\t\tcurrentVersion:  \"0.3\",\n\t\t\tmanifestVersion: \"0.4\",\n\t\t\tcqlFiles:        []string{\"base.cql\"},\n\t\t\tcqlFileContent:  \"CREATE TABLE;\",\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\n\t\t\tdir := s.T().TempDir()\n\t\t\tfor _, subDir := range test.dirs {\n\t\t\t\ts.NoError(os.Mkdir(filepath.Join(dir, subDir), os.FileMode(0755)))\n\t\t\t\tmanifestFile := filepath.Join(dir, subDir, test.manifestFile)\n\t\t\t\tm := manifest{\n\t\t\t\t\tCurrVersion:          test.manifestVersion,\n\t\t\t\t\tMinCompatibleVersion: test.manifestVersion,\n\t\t\t\t\tSchemaUpdateCqlFiles: test.cqlFiles,\n\t\t\t\t}\n\t\t\t\tj, err := json.Marshal(m)\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NoError(os.WriteFile(manifestFile, j, os.FileMode(0644)))\n\t\t\t\tcqlFile := filepath.Join(dir, subDir, \"base.cql\")\n\t\t\t\ts.NoError(os.WriteFile(cqlFile, []byte(test.cqlFileContent), os.FileMode(0644)))\n\t\t\t}\n\n\t\t\tcfg := &UpdateConfig{\n\t\t\t\tSchemaFS:      os.DirFS(dir),\n\t\t\t\tTargetVersion: \"0.4\",\n\t\t\t}\n\n\t\t\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\t\t\tcs, err := updateSchemaTask.BuildChangeSet(test.currentVersion)\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.Nil(cs)\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NotNil(cs)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) Test_parseSQLStmts() {\n\tsubDir := \"v0.4\"\n\n\ttests := map[string]struct {\n\t\tcontent  string\n\t\tcqlFiles []string\n\t\tstmts    []string\n\t\terr      error\n\t}{\n\t\t\"success\": {\n\t\t\tstmts:    []string{\"CREATE TABLE;\", \"CREATE TABLE;\"},\n\t\t\tcqlFiles: []string{\"base.cql\"},\n\t\t\tcontent:  \"CREATE TABLE;\\n\\nCREATE TABLE;\\n\",\n\t\t},\n\t\t\"error - failed to open file\": {\n\t\t\tcqlFiles: []string{\"wrong.cql\"},\n\t\t\terr:      errors.New(\"error opening file\"),\n\t\t},\n\t\t\"error - no updates\": {\n\t\t\tcqlFiles: []string{\"base.cql\"},\n\t\t\tcontent:  \"\",\n\t\t\terr:      fmt.Errorf(\"found 0 updates in dir %v\", subDir),\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tmockSchemaClient := NewMockSchemaClient(gomock.NewController(s.T()))\n\n\t\t\tdir := s.T().TempDir()\n\n\t\t\tcfg := &UpdateConfig{\n\t\t\t\tSchemaFS: os.DirFS(dir),\n\t\t\t}\n\n\t\t\ts.NoError(os.Mkdir(filepath.Join(dir, subDir), os.FileMode(0755)))\n\n\t\t\tcqlFile := filepath.Join(dir, subDir, \"base.cql\")\n\t\t\ts.NoError(os.WriteFile(cqlFile, []byte(test.content), os.FileMode(0644)))\n\n\t\t\tupdateSchemaTask := NewUpdateSchemaTask(mockSchemaClient, cfg)\n\n\t\t\tm := &manifest{\n\t\t\t\tSchemaUpdateCqlFiles: test.cqlFiles,\n\t\t\t}\n\n\t\t\tstmts, err := updateSchemaTask.parseSQLStmts(subDir, m)\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.Nil(stmts)\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(test.stmts, stmts)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) Test_readManifest() {\n\ttests := map[string]struct {\n\t\tsubDir        string\n\t\tmanifestFile  string\n\t\tinvalidSubDir bool\n\t\tcqlFiles      []string\n\t\terr           error\n\t}{\n\t\t\"error - invalid sub directory name\": {\n\t\t\tsubDir:        string([]byte{0xff, 0xfe, 0xfd, 0xfc}),\n\t\t\tmanifestFile:  manifestFileName,\n\t\t\tinvalidSubDir: true,\n\t\t\terr:           fs.ErrInvalid, // bad utf8 returns this as of 1.23: https://github.com/golang/go/commit/bf821f65cfd61dcc431922eea2cb97ce0825d60c\n\t\t},\n\t\t\"error - manifest.json file not found\": {\n\t\t\tsubDir:       \"v0.4\",\n\t\t\tmanifestFile: \"wrong_manifest.json\",\n\t\t\terr:          errors.New(\"no such file or directory\"),\n\t\t},\n\t\t\"error - manifest missing SchemaUpdateCqlFiles\": {\n\t\t\tsubDir:       \"v0.4\",\n\t\t\tmanifestFile: manifestFileName,\n\t\t\terr:          fmt.Errorf(\"manifest missing SchemaUpdateCqlFiles\"),\n\t\t},\n\t\t\"success\": {\n\t\t\tsubDir:       \"v0.4\",\n\t\t\tmanifestFile: manifestFileName,\n\t\t\tcqlFiles:     []string{\"base.cql\"},\n\t\t},\n\t}\n\n\tfor name, test := range tests {\n\t\ts.Run(name, func() {\n\t\t\tdir := s.T().TempDir()\n\n\t\t\tvar m manifest\n\n\t\t\tif !test.invalidSubDir {\n\t\t\t\tsubDir := test.subDir\n\n\t\t\t\ts.NoError(os.Mkdir(filepath.Join(dir, subDir), os.FileMode(0755)))\n\t\t\t\tmanifestFile := filepath.Join(dir, subDir, test.manifestFile)\n\t\t\t\tm = manifest{\n\t\t\t\t\tCurrVersion:          \"0.4\",\n\t\t\t\t\tMinCompatibleVersion: \"0.4\",\n\t\t\t\t\tDescription:          \"base version of schema\",\n\t\t\t\t\tSchemaUpdateCqlFiles: test.cqlFiles,\n\t\t\t\t}\n\n\t\t\t\tj, err := json.Marshal(m)\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.NoError(os.WriteFile(manifestFile, j, os.FileMode(0644)))\n\t\t\t}\n\n\t\t\tm2, err := readManifest(os.DirFS(dir), test.subDir)\n\n\t\t\tif test.err != nil {\n\t\t\t\ts.Nil(m2)\n\t\t\t\ts.ErrorContains(err, test.err.Error())\n\t\t\t} else {\n\t\t\t\ts.NoError(err)\n\t\t\t\ts.Equal(m.CurrVersion, m2.CurrVersion)\n\t\t\t\ts.Equal(m.MinCompatibleVersion, m2.MinCompatibleVersion)\n\t\t\t\ts.Equal(m.Description, m2.Description)\n\t\t\t\ts.Equal(m.SchemaUpdateCqlFiles, m2.SchemaUpdateCqlFiles)\n\t\t\t\ts.Empty(m.md5)\n\t\t\t\ts.NotEmpty(m2.md5)\n\t\t\t}\n\t\t})\n\t}\n}\n\nfunc (s *UpdateTaskTestSuite) Test_readSchemaDir_ErrorGettingSubDirs() {\n\tret, err := readSchemaDir(os.DirFS(\"testdata\"), \"0.4\", \"10.2\")\n\ts.Error(err)\n\ts.ErrorContains(err, \"no such file or directory\")\n\ts.Nil(ret)\n}\n\nfunc (s *UpdateTaskTestSuite) runReadManifestTest(dir, input, currVer, minVer, desc string,\n\tfiles []string, isErr bool) {\n\n\tfile := dir + \"/manifest.json\"\n\terr := ioutil.WriteFile(file, []byte(input), os.FileMode(0644))\n\ts.Nil(err)\n\n\tm, err := readManifest(os.DirFS(dir), \".\")\n\tif isErr {\n\t\ts.Error(err)\n\t\treturn\n\t}\n\ts.NoError(err)\n\ts.Equal(currVer, m.CurrVersion)\n\ts.Equal(minVer, m.MinCompatibleVersion)\n\ts.Equal(desc, m.Description)\n\ts.True(len(m.md5) > 0)\n\ts.Equal(files, m.SchemaUpdateCqlFiles)\n}\n"
  },
  {
    "path": "tools/common/schema/util.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"bufio\"\n\t\"io\"\n\t\"io/fs\"\n\t\"strings\"\n)\n\nconst newLineDelim = '\\n'\n\n// ParseFile takes a cql / sql file as input\n// and returns an array of cql / sql statements on\n// success.\nfunc ParseFile(file fs.File) ([]string, error) {\n\treader := bufio.NewReader(file)\n\n\tvar line string\n\tvar currStmt string\n\tvar stmts = make([]string, 0, 4)\n\tvar err error\n\n\tfor err == nil {\n\n\t\tline, err = reader.ReadString(newLineDelim)\n\t\tline = strings.TrimSpace(line)\n\t\tif len(line) < 1 {\n\t\t\tcontinue\n\t\t}\n\n\t\t// Filter out the comment lines, the\n\t\t// only recognized comment line format\n\t\t// is any line that starts with double dashes\n\t\ttokens := strings.Split(line, \"--\")\n\t\tif len(tokens) > 0 && len(tokens[0]) > 0 {\n\t\t\tcurrStmt += tokens[0]\n\t\t\t// semi-colon is the end of statement delim\n\t\t\tif strings.HasSuffix(currStmt, \";\") {\n\t\t\t\tstmts = append(stmts, currStmt)\n\t\t\t\tcurrStmt = \"\"\n\t\t\t}\n\t\t}\n\t}\n\n\tif err == io.EOF {\n\t\treturn stmts, nil\n\t}\n\n\treturn nil, err\n}\n"
  },
  {
    "path": "tools/common/schema/util_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage schema\n\nimport (\n\t\"io\"\n\t\"io/fs\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n)\n\nconst testdata = `\nALTER TYPE domain_config ADD isolation_groups blob;\nALTER TYPE domain_config ADD isolation_groups_encoding text;\n-- a comment\nALTER TYPE domain_config2 ADD isolation_groups_encoding text;\n`\n\ntype mockfile struct {\n\tread bool\n}\n\nfunc (m *mockfile) Stat() (fs.FileInfo, error) {\n\treturn nil, nil\n}\nfunc (m *mockfile) Read(in []byte) (int, error) {\n\tif m.read {\n\t\treturn 0, io.EOF\n\t}\n\tfor i := 0; i < len(in) && i < len([]byte(testdata)); i++ {\n\t\tin[i] = []byte(testdata)[i]\n\t}\n\tm.read = true\n\treturn len(in), nil\n}\nfunc (m *mockfile) Close() error {\n\treturn nil\n}\n\nfunc TestParseFile(t *testing.T) {\n\tres, err := ParseFile(&mockfile{})\n\tassert.NoError(t, err)\n\texpectedOutput := []string{\n\t\t\"ALTER TYPE domain_config ADD isolation_groups blob;\",\n\t\t\"ALTER TYPE domain_config ADD isolation_groups_encoding text;\",\n\t\t\"ALTER TYPE domain_config2 ADD isolation_groups_encoding text;\",\n\t}\n\tassert.Equal(t, expectedOutput, res)\n}\n"
  },
  {
    "path": "tools/common/schema/version.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"fmt\"\n\t\"regexp\"\n\t\"strconv\"\n\t\"strings\"\n)\n\n// represents names of the form vx.x where x.x is a (major, minor) version pair\nvar versionStrRegex = regexp.MustCompile(`^v\\d+(\\.\\d+)?$`)\n\n// represents names of the form x.x where minor version is always single digit\nvar versionNumRegex = regexp.MustCompile(`^\\d+(\\.\\d+)?$`)\n\n// represents names of the form sx.x-y.y where x.x and y.y. are (major, minor) version pairs\nvar squashVersionStrRegex = regexp.MustCompile(`^s\\d+(\\.\\d+)?-\\d+(\\.\\d+)?$`)\n\n// cmpVersion compares two version strings\n// returns 0 if a == b\n// returns < 0 if a < b\n// returns > 0 if a > b\nfunc cmpVersion(a, b string) int {\n\n\taMajor, aMinor, _ := parseVersion(a)\n\tbMajor, bMinor, _ := parseVersion(b)\n\n\tif aMajor != bMajor {\n\t\treturn aMajor - bMajor\n\t}\n\n\treturn aMinor - bMinor\n}\n\n// parseVersion parses a version string and\n// returns the major, minor version pair\nfunc parseVersion(ver string) (major int, minor int, err error) {\n\n\tif len(ver) == 0 {\n\t\treturn\n\t}\n\n\tvals := strings.Split(ver, \".\")\n\tif len(vals) == 0 { // Split returns slice of size=1 on empty string\n\t\treturn major, minor, nil\n\t}\n\n\tif len(vals) > 0 {\n\t\tmajor, err = strconv.Atoi(vals[0])\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\tif len(vals) > 1 {\n\t\tminor, err = strconv.Atoi(vals[1])\n\t\tif err != nil {\n\t\t\treturn\n\t\t}\n\t}\n\n\treturn\n}\n\n// parseValidateVersion validates that the given input conforms to either of vx.x or x.x and\n// returns x.x on success\nfunc parseValidateVersion(ver string) (string, error) {\n\tif len(ver) == 0 {\n\t\treturn \"\", fmt.Errorf(\"version is empty\")\n\t}\n\tif versionStrRegex.MatchString(ver) {\n\t\treturn ver[1:], nil\n\t}\n\tif !versionNumRegex.MatchString(ver) {\n\t\treturn \"\", fmt.Errorf(\"invalid version, expected format is x.x\")\n\t}\n\treturn ver, nil\n}\n"
  },
  {
    "path": "tools/common/schema/version_test.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage schema\n\nimport (\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/require\"\n\t\"github.com/stretchr/testify/suite\"\n)\n\ntype (\n\tVersionTestSuite struct {\n\t\t*require.Assertions // override suite.Suite.Assertions with require.Assertions; this means that s.NotNil(nil) will stop the test, not merely log an error\n\t\tsuite.Suite\n\t}\n)\n\nfunc TestVersionTestSuite(t *testing.T) {\n\tsuite.Run(t, new(VersionTestSuite))\n}\n\nfunc (s *VersionTestSuite) SetupTest() {\n\ts.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil\n}\n\nfunc (s *VersionTestSuite) TestParseVersion() {\n\ts.execParseTest(\"\", 0, 0, false)\n\ts.execParseTest(\"0\", 0, 0, false)\n\ts.execParseTest(\"99\", 99, 0, false)\n\ts.execParseTest(\"0.0\", 0, 0, false)\n\ts.execParseTest(\"0.9\", 0, 9, false)\n\ts.execParseTest(\"0.10\", 0, 10, false)\n\ts.execParseTest(\"1.0\", 1, 0, false)\n\ts.execParseTest(\"9999.0\", 9999, 0, false)\n\ts.execParseTest(\"999.999\", 999, 999, false)\n\ts.execParseTest(\"88.88.88\", 88, 88, false)\n\ts.execParseTest(\"a.b\", 0, 0, true)\n\ts.execParseTest(\"1.5a\", 0, 0, true)\n\ts.execParseTest(\"5.b\", 0, 0, true)\n\ts.execParseTest(\"golang\", 0, 0, true)\n}\n\nfunc (s *VersionTestSuite) TestCmpVersion() {\n\n\ts.Equal(0, cmpVersion(\"0\", \"0\"))\n\ts.Equal(0, cmpVersion(\"999\", \"999\"))\n\ts.Equal(0, cmpVersion(\"0.0\", \"0.0\"))\n\ts.Equal(0, cmpVersion(\"0.999\", \"0.999\"))\n\ts.Equal(0, cmpVersion(\"99.888\", \"99.888\"))\n\n\ts.True(cmpVersion(\"0.1\", \"0\") > 0)\n\ts.True(cmpVersion(\"0.5\", \"0.1\") > 0)\n\ts.True(cmpVersion(\"1.1\", \"0.1\") > 0)\n\ts.True(cmpVersion(\"1.1\", \"0.9\") > 0)\n\ts.True(cmpVersion(\"1.1\", \"1.0\") > 0)\n\n\ts.True(cmpVersion(\"0\", \"0.1\") < 0)\n\ts.True(cmpVersion(\"0.1\", \"0.5\") < 0)\n\ts.True(cmpVersion(\"0.1\", \"1.1\") < 0)\n\ts.True(cmpVersion(\"0.9\", \"1.1\") < 0)\n\ts.True(cmpVersion(\"1.0\", \"1.1\") < 0)\n\n\ts.True(cmpVersion(\"0.1a\", \"0.5\") < 0)\n\ts.True(cmpVersion(\"0.1\", \"0.5a\") > 0)\n\ts.True(cmpVersion(\"ab\", \"cd\") == 0)\n}\n\nfunc (s *VersionTestSuite) TestParseValidateVersion() {\n\n\tinputs := []string{\"0\", \"1000\", \"9999\", \"0.1\", \"0.9\", \"99.9\", \"100.8\"}\n\tfor _, in := range inputs {\n\t\ts.execParseValidateTest(in, in, false)\n\t\ts.execParseValidateTest(\"v\"+in, in, false)\n\t}\n\n\terrInputs := []string{\"1.2a\", \"ab\", \"5.11a\"}\n\tfor _, in := range errInputs {\n\t\ts.execParseValidateTest(in, \"\", true)\n\t\ts.execParseValidateTest(\"v\"+in, \"\", true)\n\t}\n}\n\nfunc (s *VersionTestSuite) execParseValidateTest(input string, output string, isErr bool) {\n\tver, err := parseValidateVersion(input)\n\tif isErr {\n\t\ts.NotNil(err)\n\t\treturn\n\t}\n\ts.Nil(err)\n\ts.Equal(output, ver)\n}\n\nfunc (s *VersionTestSuite) execParseTest(input string, expMajor int, expMinor int, isErr bool) {\n\tmaj, min, err := parseVersion(input)\n\tif isErr {\n\t\ts.NotNil(err)\n\t\treturn\n\t}\n\ts.Nil(err)\n\ts.Equal(expMajor, maj)\n\ts.Equal(expMinor, min)\n}\n"
  },
  {
    "path": "tools/linter/funcorder/analyzer.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage funcorder\n\nimport (\n\t\"os\"\n\t\"sort\"\n\t\"strings\"\n\n\t\"github.com/dave/dst\"\n\t\"github.com/dave/dst/decorator\"\n\t\"golang.org/x/tools/go/analysis\"\n)\n\nconst (\n\tName = \"funcorder\"\n)\n\nvar (\n\tAnalyzer = &analysis.Analyzer{\n\t\tName: Name,\n\t\tDoc:  \"check declaration order and count of types, constants, variables and functions\",\n\t\tRun:  run,\n\t}\n)\n\ntype (\n\trecvFunc struct {\n\t\tIndex    int\n\t\tFuncName string\n\t}\n)\n\nfunc run(pass *analysis.Pass) (interface{}, error) {\n\t// a decorated ast package \"dst\" is used to avoid free floating comment issue\n\t// see https://github.com/golang/go/issues/20744\n\tdec := decorator.NewDecorator(pass.Fset) // holds mapping between ast and dst\n\n\tfor _, f := range pass.Files {\n\t\tdstF, err := dec.DecorateFile(f) // transform ast file to dst file\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\n\t\t// build recv <> []func mapping\n\t\trFuncs := make(map[string][]recvFunc)\n\n\t\tfor i, decl := range dstF.Decls {\n\t\t\tswitch funcDecl := decl.(type) {\n\t\t\tcase *dst.FuncDecl:\n\t\t\t\trecv := getRecvStructName(funcDecl)\n\t\t\t\tif recv == \"\" || isMock(recv) {\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\trFuncs[recv] = append(rFuncs[recv], recvFunc{\n\t\t\t\t\tIndex:    i,\n\t\t\t\t\tFuncName: funcDecl.Name.Name,\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\n\t\t// rearrange dst ordering for each recv\n\t\tfor _, funcList := range rFuncs {\n\t\t\tsort.Slice(funcList, func(i, j int) bool {\n\t\t\t\treturn strings.Compare(funcList[i].FuncName, funcList[j].FuncName) < 0\n\t\t\t})\n\t\t\tselectedIndex := make(map[int]struct{})\n\t\t\tfor i := range funcList {\n\t\t\t\tselectedIndex[funcList[i].Index] = struct{}{}\n\t\t\t}\n\n\t\t\tcursor := 0\n\t\t\tvar newDecls []dst.Decl\n\t\t\tfor i := 0; i < len(dstF.Decls); i++ {\n\t\t\t\tindex := i\n\t\t\t\tif _, ok := selectedIndex[i]; ok {\n\t\t\t\t\tindex = funcList[cursor].Index\n\t\t\t\t\tcursor++\n\t\t\t\t}\n\t\t\t\tnewDecls = append(newDecls, dstF.Decls[index])\n\t\t\t}\n\t\t\tdstF.Decls = newDecls\n\t\t}\n\n\t\t// save cleaned file\n\t\tfName := pass.Fset.Position(f.Pos()).Filename\n\t\twriter, err := os.OpenFile(fName, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\n\t\terr = decorator.Fprint(writer, dstF)\n\t\tif err != nil {\n\t\t\tpanic(err)\n\t\t}\n\t}\n\n\treturn nil, nil\n}\n\nfunc getRecvStructName(funcDecl *dst.FuncDecl) string {\n\tif funcDecl.Recv != nil && len(funcDecl.Recv.List) == 1 && funcDecl.Name.IsExported() {\n\t\tif expr, ok := funcDecl.Recv.List[0].Type.(*dst.StarExpr); ok {\n\t\t\tif ident, ok := expr.X.(*dst.Ident); ok {\n\t\t\t\treturn ident.Name\n\t\t\t}\n\t\t}\n\t}\n\n\treturn \"\"\n}\n\nfunc isMock(s string) bool {\n\treturn len(s) > 4 && s[0:4] == \"Mock\"\n}\n"
  },
  {
    "path": "tools/linter/funcorder/cmd/funcorder/main.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage main\n\nimport (\n\t\"golang.org/x/tools/go/analysis/singlechecker\"\n\n\t\"github.com/uber/cadence/tools/linter/funcorder\"\n)\n\nfunc main() {\n\tsinglechecker.Main(funcorder.Analyzer)\n}\n"
  },
  {
    "path": "tools/mcp/README.md",
    "content": "## Overview\n\nThis folder contains an [MCP server](https://modelcontextprotocol.io/introduction) which exposes useful Cadence tools to Cursor.\n\n## How to install\nThis should be integrated to Cursors inside devpod seamlessly. For now, follow manual steps below:\n\n1. Build the server executable\n```\nmkdir -p .bin && go build -o .bin/cadence_mcp tools/mcp/main.go\n```\n\n\n2. Update .cursor/mcp.json with following entry. Use the full path to the executable:\n```\n{\n\"mcpServers\": {\n  \"cadence-mcp-server\": {\n      \"command\": \"/path/to/repo/.bin/cadence_mcp\",\n      \"args\": [],\n      \"env\": {}\n    }\n  }\n}\n```\n\n3. Enable Agent mode in Cursor.\n\n4. Enable yolo mode if you want tools to be run without confirmation.\n\n5. Restart Cursor\n\n## Usage\n\nAsk a relevant question. For example:\n\n  Is my Cadence domain \"cadence-system\" resilient to regional outages?\n\nFor now, it will tell you \"Yes\" if the domain is global, and \"No\" otherwise.\n\n## How to add a new tool\n\n1. Implement the tool in main.go\n2. Build the server executable\n3. Restart Cursor\n4. Ask a relevant questions and test it out\n"
  },
  {
    "path": "tools/mcp/main.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage main\n\nimport (\n\t\"context\"\n\t\"encoding/hex\"\n\t\"errors\"\n\t\"fmt\"\n\t\"os\"\n\t\"os/exec\"\n\t\"path\"\n\t\"runtime/debug\"\n\t\"strings\"\n\n\t\"github.com/mark3labs/mcp-go/mcp\"\n\t\"github.com/mark3labs/mcp-go/server\"\n)\n\nfunc main() {\n\n\t// Create MCP server\n\ts := server.NewMCPServer(\n\t\t\"Cadence MCP\",\n\t\t\"0.0.1\",\n\t\tserver.WithLogging(),\n\t)\n\n\t// Add tool handlers\n\ts.AddTool(mcp.NewTool(\"domain_rr\",\n\t\tmcp.WithDescription(\"Check if a cadence domain is resilient to regional outages\"),\n\t\tmcp.WithString(\"domain\",\n\t\t\tmcp.Required(),\n\t\t\tmcp.Description(\"Name of the cadence domain to check\"),\n\t\t),\n\t\tmcp.WithString(\"grpc_endpoint\",\n\t\t\tmcp.DefaultString(\"localhost:7833\"),\n\t\t\tmcp.Description(\"gRPC endpoint of the cadence domain\"),\n\t\t),\n\t), domainRRHandler)\n\n\ts.AddTool(mcp.NewTool(\"payload_decoder\",\n\t\tmcp.WithDescription(\"Decode a payload that is encoded by hex or base64. The payload is from Cadence database.\"),\n\t\tmcp.WithString(\"payload\",\n\t\t\tmcp.Required(),\n\t\t\tmcp.Description(\"The payload to decode\"),\n\t\t),\n\t), payloadDecoderHandler)\n\n\ts.AddTool(mcp.NewTool(\"command_generator\",\n\t\tmcp.WithDescription(\"Convert natural language to Cadence CLI commands. Use this tool when you need to generate Cadence CLI commands from natural language descriptions like 'list failed workflows from past 7 days' or 'start workflow with search attributes'.\"),\n\t\tmcp.WithString(\"query\",\n\t\t\tmcp.Required(),\n\t\t\tmcp.Description(\"Natural language description of Cadence CLI command to be generated (e.g., 'list failed workflows from past 7 days', 'start workflow with search attributes')\"),\n\t\t),\n\t\tmcp.WithString(\"domain\",\n\t\t\tmcp.Required(),\n\t\t\tmcp.Description(\"Target domain name for Cadence command\"),\n\t\t),\n\t\tmcp.WithString(\"address\",\n\t\t\tmcp.DefaultString(\"localhost:7833\"),\n\t\t\tmcp.Description(\"gRPC endpoint of cadence domain\"),\n\t\t),\n\t), cadenceCommandGeneratorHandler)\n\n\tdebugLog(\"Cadence MCP started\")\n\n\t// Start the stdio server\n\tif err := server.ServeStdio(s); err != nil {\n\t\tdebugLog(\"Server error: %v\\n\", err)\n\t}\n\n\tdebugLog(\"Cadence MCP stopped\")\n}\n\nfunc domainRRHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {\n\tdefer func() {\n\t\t// recover from panic\n\t\tif r := recover(); r != nil {\n\t\t\t// include the stack trace\n\t\t\tdebugLog(\"Panic: %v\\n\", r)\n\t\t\tdebugLog(\"Stack trace: %s\\n\", string(debug.Stack()))\n\t\t}\n\t}()\n\n\tdomain, ok := request.Params.Arguments[\"domain\"].(string)\n\tif !ok {\n\t\treturn nil, errors.New(\"domain must be a string\")\n\t}\n\n\tendpoint, ok := request.Params.Arguments[\"grpc_endpoint\"].(string)\n\tif !ok {\n\t\tendpoint = \"localhost:7833\"\n\t}\n\n\t// run cadence CLI to check if it's a global domain or not\n\tcmd := exec.Command(\"docker\", \"run\", \"-t\", \"--rm\", \"--network\", \"host\", \"ubercadence/cli:master\",\n\t\t\"--transport\", \"grpc\",\n\t\t\"--address\", endpoint,\n\t\t\"--domain\", domain,\n\t\t\"domain\", \"describe\")\n\t// run the cmd and capture both stdout and stderr\n\toutput, err := cmd.CombinedOutput()\n\tif err != nil {\n\t\tdebugLog(\"Error checking domain resilience: %v, %s\\n\", err, string(output))\n\t\treturn mcp.NewToolResultError(\"Error checking domain resilience: \" + err.Error() + \"\\n\" + string(output)), nil\n\t}\n\n\t// parse the output of the cadence CLI\n\t// if it contains \"IsGlobal(XDC)Domain: true\" then it's a global domain\n\t// otherwise it's not\n\tif strings.Contains(string(output), \"IsGlobal(XDC)Domain: true\") {\n\t\treturn mcp.NewToolResultText(\"Yes, this domain is resilient to regional outages\"), nil\n\t}\n\n\treturn mcp.NewToolResultText(\"No, this domain is not resilient to regional outages. Consider making it a global domain.\"), nil\n}\n\nfunc payloadDecoderHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {\n\tpayload, ok := request.Params.Arguments[\"payload\"].(string)\n\tif !ok {\n\t\treturn nil, errors.New(\"payload must be a string\")\n\t}\n\n\t// check if the payload is encoded by hex or base64\n\tenc := \"base64\"\n\tif isHexEncoded(payload) {\n\t\tenc = \"hex\"\n\t}\n\n\tdebugLog(\"Decoding payload with %s encoding\\n\", enc)\n\n\t// invoke cadence CLI to decode the payload\n\tcmd := exec.Command(\"docker\", \"run\", \"-t\", \"--rm\", \"--network\", \"host\", \"ubercadence/cli:master\",\n\t\t\"admin\", \"db\", \"decode_thrift\",\n\t\t\"--input\", payload,\n\t\t\"--encoding\", enc)\n\n\t// run the cmd and capture both stdout and stderr\n\toutput, err := cmd.CombinedOutput()\n\tif err != nil {\n\t\tdebugLog(\"Error decoding payload: %v, %s\\n\", err, string(output))\n\t\treturn mcp.NewToolResultError(\"Error decoding payload: \" + err.Error() + \"\\n\" + string(output)), nil\n\t}\n\n\treturn mcp.NewToolResultText(string(output)), nil\n}\n\nfunc isHexEncoded(payload string) bool {\n\t_, err := hex.DecodeString(strings.TrimPrefix(payload, \"0x\"))\n\treturn err == nil\n}\n\nfunc debugLog(format string, args ...interface{}) {\n\t// get the path of the binary\n\tbinaryPath, err := os.Executable()\n\tif err != nil {\n\t\tfmt.Println(\"Failed to get executable path:\", err)\n\t\treturn\n\t}\n\tlogFile, err := os.OpenFile(path.Join(path.Dir(binaryPath), \"cadence_mcp.log\"), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)\n\tif err != nil {\n\t\tfmt.Println(\"Failed to open log file:\", err)\n\t\treturn\n\t}\n\tdefer logFile.Close()\n\n\tlogFile.WriteString(fmt.Sprintf(format, args...))\n\tlogFile.WriteString(\"\\n\")\n}\n\nfunc cadenceCommandGeneratorHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {\n\tdefer func() {\n\t\tif r := recover(); r != nil {\n\t\t\tdebugLog(\"Panic in Cadence Command Generator: %v\\n\", r)\n\t\t\tdebugLog(\"Stack trace: %s\\n\", string(debug.Stack()))\n\t\t}\n\t}()\n\n\tquery, ok := request.Params.Arguments[\"query\"].(string)\n\tif !ok {\n\t\treturn nil, errors.New(\"query must be a string\")\n\t}\n\n\tdomain, ok := request.Params.Arguments[\"domain\"].(string)\n\tif !ok {\n\t\treturn nil, errors.New(\"domain must be a string\")\n\t}\n\n\taddress, ok := request.Params.Arguments[\"address\"].(string)\n\tif !ok {\n\t\taddress = \"localhost:7833\"\n\t}\n\n\tcommand, err := generateCadenceCommand(query, domain, address)\n\tif err != nil {\n\t\treturn nil, errors.New(\"error generating cadence command: \" + err.Error())\n\t}\n\n\t// format the command to be displayed to user\n\tformattedCommand := fmt.Sprintf(\"Command Generated: %s\", command)\n\treturn mcp.NewToolResultText(formattedCommand), nil\n}\n"
  },
  {
    "path": "tools/mcp/util.go",
    "content": "package main\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"regexp\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n)\n\n// Main function to generate Cadence command\nfunc generateCadenceCommand(query, domain, address string) (string, error) {\n\tquery = strings.ToLower(query)\n\taddress = strings.ToLower(address)\n\n\tbaseCmd := fmt.Sprintf(\"docker run -t --rm --network host ubercadence/cli:master --transport grpc --address %s --do %s\", address, domain)\n\n\tif strings.Contains(query, \"workflow\") || strings.Contains(query, \"workflows\") {\n\t\treturn generateWorkflowCommand(query, baseCmd)\n\t}\n\n\tif strings.Contains(query, \"domain\") {\n\t\treturn generateDomainCommand(query, baseCmd)\n\t}\n\n\treturn \"\", errors.New(\"unsupported query type\")\n}\n\nfunc generateWorkflowCommand(query, baseCmd string) (string, error) {\n\n\t// Show/History of Workflow Execution\n\tif strings.Contains(query, \"show\") || strings.Contains(query, \"history\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf show --wid %s\", workflowID), nil\n\t\t}\n\t\treturn \"Please provide a workflow ID to show workflow details\", nil\n\t}\n\n\t// Describe Workflow\n\tif strings.Contains(query, \"describe\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf describe -w %s\", workflowID), nil\n\t\t}\n\t\treturn \"Please provide a workflow ID to describe workflow details\", nil\n\t}\n\n\t// Start Workflow\n\tif strings.Contains(query, \"start\") || strings.Contains(query, \"create\") {\n\t\t// Add search attributes if provided\n\t\tif strings.Contains(query, \"search\") || strings.Contains(query, \"custom\") {\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf start --tl <tasklist-name> --wt <workflow-type> --et <execution-timeout> --i <input-json> -search_attr_key CustomIntField | CustomKeywordField | CustomStringField | CustomBoolField | CustomDatetimeField -search_attr_value <value> | keyword | <search-attribute> | true | 2019-06-07T16:16:36-08:00\"), nil\n\t\t}\n\t\treturn fmt.Sprintf(baseCmd, \"wf start --tl <tasklist-name> --wt <workflow-type> --et <execution-timeout> -i <input-json>\"), nil\n\t}\n\n\t// Run Workflow\n\tif strings.Contains(query, \"run\") {\n\t\treturn fmt.Sprintf(baseCmd, \"wf run --tl <tasklist-name> --wt <workflow-type> --et <execution-timeout> -i <input-json>\"), nil\n\t}\n\n\t// Signal Workflow\n\tif strings.Contains(query, \"signal\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") && strings.Contains(query, \"run id\") || strings.Contains(query, \"run\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\trunID := extractRunID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf signal -w %s -r %s -n <signal-name> -i <input-json>\", workflowID, runID), nil\n\t\t}\n\t\treturn \"Please provide a workflow/run ID to signal workflow\", nil\n\t}\n\n\t// Cancel Workflow\n\tif strings.Contains(query, \"cancel\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") && strings.Contains(query, \"run id\") || strings.Contains(query, \"run\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\trunID := extractRunID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf cancel -w %s -r %s\", workflowID, runID), nil\n\t\t}\n\t\treturn \"Please provide a workflow/run ID to cancel workflow\", nil\n\t}\n\n\t// Terminate Workflow\n\tif strings.Contains(query, \"terminate\") || strings.Contains(query, \"stop\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") && strings.Contains(query, \"run id\") || strings.Contains(query, \"run\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\trunID := extractRunID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf terminate -w %s -r %s --reason <reason>\", workflowID, runID), nil\n\t\t}\n\t\treturn \"Please provide a workflow/run ID to terminate workflow\", nil\n\t}\n\n\t// List Workflows with filters\n\tif strings.Contains(query, \"list\") || strings.Contains(query, \"from\") || strings.Contains(query, \"past\") {\n\t\tqueryFilter := buildWorkflowQueryFilter(query)\n\t\tif queryFilter != \"\" {\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf list -q %s\", queryFilter), nil\n\t\t}\n\t\treturn fmt.Sprintf(baseCmd, \"wf list -m\"), nil\n\t}\n\n\t// Count Workflows\n\tif strings.Contains(query, \"count\") {\n\t\tqueryFilter := buildWorkflowQueryFilter(query)\n\t\treturn fmt.Sprintf(baseCmd, \"wf count -q %s\", queryFilter), nil\n\t}\n\n\t// Query Workflows using Stack Trace\n\tif strings.Contains(query, \"stack trace\") || strings.Contains(query, \"stack\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") && strings.Contains(query, \"run id\") || strings.Contains(query, \"run\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\trunID := extractRunID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf stack -w %s -r %s\", workflowID, runID), nil\n\t\t}\n\t\treturn \"Please provide a workflow ID/runID to query stack trace\", nil\n\t}\n\n\t// Reset Workflow\n\tif strings.Contains(query, \"reset\") {\n\t\tif strings.Contains(query, \"workflow id\") || strings.Contains(query, \"id\") && strings.Contains(query, \"run id\") || strings.Contains(query, \"run\") {\n\t\t\tworkflowID := extractWorkflowID(query)\n\t\t\trunID := extractRunID(query)\n\t\t\treturn fmt.Sprintf(baseCmd, \"wf reset -w %s -r %s --reset_type <reset-type> --reason <reason>\", workflowID, runID), nil\n\t\t}\n\t\treturn \"Please provide a workflow ID to reset workflow\", nil\n\t}\n\n\treturn \"\", errors.New(\"unsupported query type\")\n}\n\nfunc generateDomainCommand(query, baseCmd string) (string, error) {\n\tif strings.Contains(query, \"describe\") || strings.Contains(query, \"show\") {\n\t\treturn fmt.Sprintf(baseCmd, \"domain describe\"), nil\n\t} else if strings.Contains(query, \"list\") {\n\t\treturn fmt.Sprintf(baseCmd, \"domain list\"), nil\n\t}\n\treturn \"\", errors.New(\"unsupported query type\")\n}\n\nfunc extractWorkflowID(query string) string {\n\tre := regexp.MustCompile(`(?:workflow\\s+)?id\\s+([a-zA-Z0-9\\-_]+)`)\n\tmatches := re.FindStringSubmatch(query)\n\tif len(matches) > 1 {\n\t\treturn matches[1]\n\t}\n\treturn \"<workflow-id>\"\n}\n\nfunc extractRunID(query string) string {\n\tre := regexp.MustCompile(`(?:run\\s+)?id\\s+([a-zA-Z0-9\\-_]+)`)\n\tmatches := re.FindStringSubmatch(query)\n\tif len(matches) > 1 {\n\t\treturn matches[1]\n\t}\n\treturn \"<run-id>\"\n}\n\nfunc buildWorkflowQueryFilter(query string) string {\n\tvar filters []string\n\n\t// Check for time-based filters\n\tif strings.Contains(query, \"past\") || strings.Contains(query, \"last\") {\n\t\ttimeFilter := buildTimeFilter(query)\n\t\tif timeFilter != \"\" {\n\t\t\tfilters = append(filters, timeFilter)\n\t\t}\n\t}\n\n\t// Check for status filters\n\tstatusFilters := map[string][]string{\n\t\t\"failed\":           {\"failed\", \"failure\"},\n\t\t\"completed\":        {\"completed\", \"success\"},\n\t\t\"canceled\":         {\"canceled\"},\n\t\t\"terminated\":       {\"terminated\"},\n\t\t\"timed_out\":        {\"timed out\", \"timeout\"},\n\t\t\"continued_as_new\": {\"continued as new\"},\n\t}\n\n\tfor status, keywords := range statusFilters {\n\t\tif containsAny(query, keywords) {\n\t\t\tfilters = append(filters, fmt.Sprintf(\"CloseStatus=\\\"%s\\\"\", status))\n\t\t}\n\t}\n\n\t// Check for workflow type filters\n\tworkflowFilters := map[string][]string{\n\t\t\"WorkflowType\": {\"workflow type\", \"type\"},\n\t\t\"WorkflowID\":   {\"workflow id\", \"id\"},\n\t\t\"RunID\":        {\"run id\", \"id\"},\n\t}\n\n\tfor field, keywords := range workflowFilters {\n\t\tif containsAny(query, keywords) {\n\t\t\tfilters = append(filters, fmt.Sprintf(\"%s = \\\"<%s>\\\"\", field, strings.ToLower(field)))\n\t\t}\n\t}\n\n\t// Check for search attribute filters\n\tif strings.Contains(query, \"search\") || strings.Contains(query, \"custom\") {\n\t\tsearchFilters := buildSearchAttributeFilters(query)\n\t\tif len(searchFilters) > 0 {\n\t\t\tfilters = append(filters, searchFilters...)\n\t\t}\n\t}\n\n\t// Combine filters\n\tif len(filters) == 0 {\n\t\treturn \"\"\n\t}\n\n\treturn strings.Join(filters, \" AND \")\n}\n\nfunc buildSearchAttributeFilters(query string) []string {\n\tvar filters []string\n\n\tsearchAttributeFilters := map[string]struct {\n\t\tkeywords []string\n\t\tfilter   string\n\t}{\n\t\t\"CustomIntField\": {\n\t\t\tkeywords: []string{\"customint\", \"int field\"},\n\t\t\tfilter:   \"CustomIntField >= 0\",\n\t\t},\n\t\t\"CustomKeywordField\": {\n\t\t\tkeywords: []string{\"customkeyword\", \"keyword field\"},\n\t\t\tfilter:   \"CustomKeywordField = \\\"<keyword-value>\\\"\",\n\t\t},\n\t\t\"CustomStringField\": {\n\t\t\tkeywords: []string{\"customstring\", \"string field\"},\n\t\t\tfilter:   \"CustomStringField = \\\"<string-value>\\\"\",\n\t\t},\n\t\t\"CustomBoolField\": {\n\t\t\tkeywords: []string{\"custombool\", \"bool field\"},\n\t\t\tfilter:   \"CustomBoolField = true\",\n\t\t},\n\t\t\"CustomDatetimeField\": {\n\t\t\tkeywords: []string{\"customdatetime\", \"datetime field\"},\n\t\t\tfilter:   \"CustomDatetimeField > \\\"<datetime-value>\\\"\",\n\t\t},\n\t}\n\n\tfor _, data := range searchAttributeFilters {\n\t\tif containsAny(query, data.keywords) {\n\t\t\tfilters = append(filters, data.filter)\n\t\t}\n\t}\n\n\treturn filters\n}\n\nfunc containsAny(query string, keywords []string) bool {\n\tfor _, keyword := range keywords {\n\t\tif strings.Contains(query, keyword) {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n\nfunc buildTimeFilter(query string) string {\n\n\tdays := extractDays(query)\n\tif days == 0 {\n\t\treturn \"\"\n\t}\n\n\treferenceDate := extractReferenceDate(query)\n\tvar baseTime time.Time\n\tif !referenceDate.IsZero() {\n\t\tbaseTime = referenceDate\n\t} else {\n\t\tbaseTime = time.Now()\n\t}\n\n\tstartTime := baseTime.AddDate(0, 0, -days)\n\tendTime := baseTime\n\n\t// Format dates as ISO 8601\n\tstartStr := startTime.Format(\"2006-01-02T15:04:05Z\")\n\tendStr := endTime.Format(\"2006-01-02T15:04:05Z\")\n\n\treturn fmt.Sprintf(\"CloseTime between \\\"%s\\\" and \\\"%s\\\"\", startStr, endStr)\n}\n\nfunc extractReferenceDate(query string) time.Time {\n\tpatterns := []string{\n\t\t`today is (january|february|march|april|may|june|july|august|september|october|november|december)\\s+(\\d+)`,\n\t\t`today is (\\d{4})-(\\d{1,2})-(\\d{1,2})`,\n\t\t`reference date is (january|february|march|april|may|june|july|august|september|october|november|december)\\s+(\\d+)`,\n\t\t`reference date is (\\d{4})-(\\d{1,2})-(\\d{1,2})`,\n\t}\n\n\tfor _, pattern := range patterns {\n\t\tre := regexp.MustCompile(pattern)\n\t\tmatches := re.FindStringSubmatch(query)\n\t\tif len(matches) > 1 {\n\t\t\tif len(matches) == 3 {\n\n\t\t\t\tmonthStr := matches[1]\n\t\t\t\tdayStr := matches[2]\n\n\t\t\t\tmonthMap := map[string]int{\n\t\t\t\t\t\"january\": 1, \"february\": 2, \"march\": 3, \"april\": 4,\n\t\t\t\t\t\"may\": 5, \"june\": 6, \"july\": 7, \"august\": 8,\n\t\t\t\t\t\"september\": 9, \"october\": 10, \"november\": 11, \"december\": 12,\n\t\t\t\t}\n\n\t\t\t\tif month, ok := monthMap[strings.ToLower(monthStr)]; ok {\n\t\t\t\t\tif day, err := strconv.Atoi(dayStr); err == nil {\n\t\t\t\t\t\tyear := time.Now().Year()\n\t\t\t\t\t\treturn time.Date(year, time.Month(month), day, 0, 0, 0, 0, time.UTC)\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else if len(matches) == 4 {\n\t\t\t\t// YYYY-MM-DD format\n\t\t\t\tyearStr := matches[1]\n\t\t\t\tmonthStr := matches[2]\n\t\t\t\tdayStr := matches[3]\n\n\t\t\t\tif year, err := strconv.Atoi(yearStr); err == nil {\n\t\t\t\t\tif month, err := strconv.Atoi(monthStr); err == nil {\n\t\t\t\t\t\tif day, err := strconv.Atoi(dayStr); err == nil {\n\t\t\t\t\t\t\treturn time.Date(year, time.Month(month), day, 0, 0, 0, 0, time.UTC)\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\treturn time.Time{} // Return zero time if no reference date found\n}\n\nfunc extractDays(query string) int {\n\t// Look for patterns like \"past 7 days\", \"last 3 days\", \"7 days ago\"\n\tpatterns := []string{\n\t\t`past (\\d+) days?`,\n\t\t`last (\\d+) days?`,\n\t\t`(\\d+) days? ago`,\n\t\t`(\\d+) days?`,\n\t\t`past (\\d+) weeks?`,\n\t\t`last (\\d+) weeks?`,\n\t\t`(\\d+) weeks? ago`,\n\t\t`(\\d+) weeks?`,\n\t\t`past (\\d+) months?`,\n\t\t`last (\\d+) months?`,\n\t\t`(\\d+) months? ago`,\n\t\t`(\\d+) months?`,\n\t}\n\tfor _, pattern := range patterns {\n\t\tre := regexp.MustCompile(pattern)\n\t\tmatches := re.FindStringSubmatch(query)\n\t\tif len(matches) > 1 {\n\t\t\tvalue, err := strconv.Atoi(matches[1])\n\t\t\tif err == nil {\n\t\t\t\t// Convert weeks and months to days\n\t\t\t\tif strings.Contains(pattern, \"weeks\") {\n\t\t\t\t\treturn value * 7\n\t\t\t\t}\n\t\t\t\tif strings.Contains(pattern, \"months\") {\n\t\t\t\t\treturn value * 30\n\t\t\t\t}\n\t\t\t\treturn value\n\t\t\t}\n\t\t}\n\t}\n\n\treturn 0\n}\n"
  },
  {
    "path": "tools/sql/README.md",
    "content": "## Using the SQL schema tool\n\nThis package contains the tooling for cadence sql operations. The tooling itself is agnostic of the storage engine behind\nthe sql interface. So, this same tool can be used against, say, OracleDB and MySQLDB\n\n## For localhost development\n```\nSQL_USER=$USERNAME SQL_PASSWORD=$PASSWD make install-schema-mysql\n```\n> NOTE: See [CONTRIBUTING](/CONTRIBUTING.md) for prerequisite of make command.\n\n## For production\n\n### Get the SQL Schema tool\n* Use brew to install CLI: `brew install cadence-workflow` which includes `cadence-sql-tool`\n  * The schema files are located at `/usr/local/etc/cadence/schema/`.\n  * Follow the [instructions](https://github.com/cadence-workflow/cadence/discussions/4457) if you need to install older versions of schema tools via homebrew.\n However, easier way is to use new versions of schema tools with old versions of schemas.\n All you need is to check out the older version of schemas from this repo. Run `git checkout v0.21.3` to get the v0.21.3 schemas in [the schema folder](/schema).\n* Or build yourself, with `make cadence-sql-tool`. See [CONTRIBUTING](/CONTRIBUTING.md) for prerequisite of make command.\n\n> Note: The binaries can also be found in the `ubercadence/server` docker images\n\n### Do one time database creation and schema setup for a new cluster\n- All command below are taking MySQL as example. For postgres, simply use with \"--plugin postgres\"\n\n```\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql create-database --db cadence\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql create-database --db cadence_visibility\n```\n\n```\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence setup-schema -v 0.0 -- this sets up just the schema version tables with initial version of 0.0\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence update-schema -d ./schema/mysql/v8/cadence/versioned -- upgrades your schema to the latest version\n\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence_visibility setup-schema -v 0.0 -- this sets up just the schema version tables with initial version of 0.0 for visibility\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence_visibility update-schema -d ./schema/mysql/v8/visibility/versioned  -- upgrades your schema to the latest version for visibility\n```\n\n### Update schema as part of a release\nYou can only upgrade to a new version after the initial setup done above.\n\n```\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence update-schema -d ./schema/mysql/v8/cadence/versioned -v x.x --dryrun -- executes a dryrun of upgrade to version x.x\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence update-schema -d ./schema/mysql/v8/cadence/versioned -v x.x    -- actually executes the upgrade to version x.x\n\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence_visibility update-schema -d ./schema/mysql/v8/visibility/versioned -v x.x --dryrun -- executes a dryrun of upgrade to version x.x\n./cadence-sql-tool --ep $SQL_HOST_ADDR -p $port --plugin mysql --db cadence_visibility update-schema -d ./schema/mysql/v8/visibility/versioned -v x.x    -- actually executes the upgrade to version x.x\n```\n"
  },
  {
    "path": "tools/sql/conn.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"context\"\n\n\t\"github.com/uber/cadence/common/config\"\n\t\"github.com/uber/cadence/common/persistence/sql\"\n\t\"github.com/uber/cadence/common/persistence/sql/sqlplugin\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\ntype (\n\t// Connection is the connection to database\n\tConnection struct {\n\t\tdbName  string\n\t\tadminDb sqlplugin.AdminDB\n\t}\n)\n\nvar _ schema.SchemaClient = (*Connection)(nil)\n\n// NewConnection creates a new connection to database\nfunc NewConnection(cfg *config.SQL) (*Connection, error) {\n\tdb, err := sql.NewSQLAdminDB(cfg)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn &Connection{\n\t\tadminDb: db,\n\t\tdbName:  cfg.DatabaseName,\n\t}, nil\n}\n\n// CreateSchemaVersionTables sets up the schema version tables\nfunc (c *Connection) CreateSchemaVersionTables() error {\n\treturn c.adminDb.CreateSchemaVersionTables()\n}\n\n// ReadSchemaVersion returns the current schema version for the keyspace\nfunc (c *Connection) ReadSchemaVersion() (string, error) {\n\treturn c.adminDb.ReadSchemaVersion(c.dbName)\n}\n\n// UpdateSchemaVersion updates the schema version for the keyspace\nfunc (c *Connection) UpdateSchemaVersion(newVersion string, minCompatibleVersion string) error {\n\treturn c.adminDb.UpdateSchemaVersion(c.dbName, newVersion, minCompatibleVersion)\n}\n\n// WriteSchemaUpdateLog adds an entry to the schema update history table\nfunc (c *Connection) WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error {\n\treturn c.adminDb.WriteSchemaUpdateLog(oldVersion, newVersion, manifestMD5, desc)\n}\n\n// ExecDDLQuery executes a sql statement\nfunc (c *Connection) ExecDDLQuery(stmt string, args ...interface{}) error {\n\t// TODO pass in context timeout value from command line param\n\terr := c.adminDb.ExecSchemaOperationQuery(context.Background(), stmt, args...)\n\treturn err\n}\n\n// ListTables returns a list of tables in this database\nfunc (c *Connection) ListTables() ([]string, error) {\n\treturn c.adminDb.ListTables(c.dbName)\n}\n\n// DropTable drops a given table from the database\nfunc (c *Connection) DropTable(name string) error {\n\treturn c.adminDb.DropTable(name)\n}\n\n// DropAllTables drops all tables from this database\nfunc (c *Connection) DropAllTables() error {\n\ttables, err := c.ListTables()\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, tab := range tables {\n\t\tif err := c.DropTable(tab); err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// CreateDatabase creates a database if it doesn't exist\nfunc (c *Connection) CreateDatabase(name string) error {\n\treturn c.adminDb.CreateDatabase(name)\n}\n\n// DropDatabase drops a database\nfunc (c *Connection) DropDatabase(name string) error {\n\treturn c.adminDb.DropDatabase(name)\n}\n\n// Close closes the sql client\nfunc (c *Connection) Close() {\n\tif c.adminDb != nil {\n\t\terr := c.adminDb.Close()\n\t\tif err != nil {\n\t\t\tpanic(\"cannot close connection\")\n\t\t}\n\t}\n}\n"
  },
  {
    "path": "tools/sql/handler.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\t\"log\"\n\t\"net\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tmysql_db \"github.com/uber/cadence/common/persistence/sql/sqlplugin/mysql\"\n\tpostgres_db \"github.com/uber/cadence/common/persistence/sql/sqlplugin/postgres\"\n\tsqlite_db \"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/schema/mysql\"\n\t\"github.com/uber/cadence/schema/postgres\"\n\t\"github.com/uber/cadence/schema/sqlite\"\n\tcliflag \"github.com/uber/cadence/tools/common/flag\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\n// VerifyCompatibleVersion ensures that the installed version of cadence and visibility\n// is greater than or equal to the expected version.\nfunc VerifyCompatibleVersion(\n\tcfg config.Persistence,\n) error {\n\n\tds, ok := cfg.DataStores[cfg.DefaultStore]\n\tif ok && ds.SQL != nil {\n\t\texpectedVersion := mysql.Version\n\t\tswitch ds.SQL.PluginName {\n\t\tcase mysql_db.PluginName:\n\t\t\texpectedVersion = mysql.Version\n\t\tcase postgres_db.PluginName:\n\t\t\texpectedVersion = postgres.Version\n\t\tcase sqlite_db.PluginName:\n\t\t\texpectedVersion = sqlite.Version\n\t\t}\n\t\terr := CheckCompatibleVersion(*ds.SQL, expectedVersion)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tds, ok = cfg.DataStores[cfg.VisibilityStore]\n\tif ok && ds.SQL != nil {\n\t\texpectedVersion := mysql.VisibilityVersion\n\t\tswitch ds.SQL.PluginName {\n\t\tcase mysql_db.PluginName:\n\t\t\texpectedVersion = mysql.VisibilityVersion\n\t\tcase postgres_db.PluginName:\n\t\t\texpectedVersion = postgres.VisibilityVersion\n\t\tcase sqlite_db.PluginName:\n\t\t\texpectedVersion = sqlite.VisibilityVersion\n\n\t\t}\n\t\terr := CheckCompatibleVersion(*ds.SQL, expectedVersion)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\n// CheckCompatibleVersion check the version compatibility\nfunc CheckCompatibleVersion(\n\tcfg config.SQL,\n\texpectedVersion string,\n) error {\n\tif !cfg.UseMultipleDatabases {\n\t\treturn doCheckCompatibleVersion(cfg, expectedVersion)\n\t}\n\t// recover from the original at the end\n\tdefer func() {\n\t\tcfg.User = \"\"\n\t\tcfg.Password = \"\"\n\t\tcfg.DatabaseName = \"\"\n\t\tcfg.ConnectAddr = \"\"\n\t\tcfg.UseMultipleDatabases = true\n\t}()\n\tcfg.UseMultipleDatabases = false\n\t// loop over every database to check schema version\n\tfor idx, entry := range cfg.MultipleDatabasesConfig {\n\t\tcfg.User = entry.User\n\t\tcfg.Password = entry.Password\n\t\tcfg.DatabaseName = entry.DatabaseName\n\t\tcfg.ConnectAddr = entry.ConnectAddr\n\n\t\terr := doCheckCompatibleVersion(cfg, expectedVersion)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"shardID %d fails at check schema version: %v\", idx, err)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc doCheckCompatibleVersion(\n\tcfg config.SQL,\n\texpectedVersion string,\n) error {\n\tconnection, err := NewConnection(&cfg)\n\n\tif err != nil {\n\t\treturn fmt.Errorf(\"unable to create SQL connection: %v\", err.Error())\n\t}\n\tdefer connection.Close()\n\n\treturn schema.VerifyCompatibleVersion(connection, cfg.DatabaseName, expectedVersion)\n}\n\n// setupSchema executes the setupSchemaTask\n// using the given command line arguments\n// as input\nfunc setupSchema(cli *cli.Context) error {\n\tcfg, err := parseConnectConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tconn, err := NewConnection(cfg)\n\tif err != nil {\n\t\treturn handleErr(err)\n\t}\n\tdefer conn.Close()\n\tif err := schema.Setup(cli, conn); err != nil {\n\t\treturn handleErr(err)\n\t}\n\treturn nil\n}\n\n// updateSchema executes the updateSchemaTask\n// using the given command lien args as input\nfunc updateSchema(cli *cli.Context) error {\n\tcfg, err := parseConnectConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tconn, err := NewConnection(cfg)\n\tif err != nil {\n\t\treturn handleErr(err)\n\t}\n\tdefer conn.Close()\n\tif err := schema.Update(cli, conn); err != nil {\n\t\treturn handleErr(err)\n\t}\n\treturn nil\n}\n\n// createDatabase creates a sql database\nfunc createDatabase(cli *cli.Context) error {\n\tcfg, err := parseConnectConfig(cli)\n\tif err != nil {\n\t\treturn handleErr(schema.NewConfigError(err.Error()))\n\t}\n\tdatabase := cli.String(schema.CLIOptDatabase)\n\tif database == \"\" {\n\t\treturn handleErr(schema.NewConfigError(\"missing \" + flag(schema.CLIOptDatabase) + \" argument \"))\n\t}\n\terr = doCreateDatabase(cfg, database)\n\tif err != nil {\n\t\treturn handleErr(fmt.Errorf(\"error creating database:%v\", err))\n\t}\n\treturn nil\n}\n\nfunc doCreateDatabase(cfg *config.SQL, name string) error {\n\tcfg.DatabaseName = \"\"\n\tconn, err := NewConnection(cfg)\n\tif err != nil {\n\t\treturn err\n\t}\n\tdefer conn.Close()\n\treturn conn.CreateDatabase(name)\n}\n\nfunc parseConnectConfig(cli *cli.Context) (*config.SQL, error) {\n\tcfg := new(config.SQL)\n\n\thost := cli.String(schema.CLIOptEndpoint)\n\tport := cli.Int(schema.CLIOptPort)\n\tcfg.ConnectAddr = fmt.Sprintf(\"%s:%v\", host, port)\n\tcfg.User = cli.String(schema.CLIOptUser)\n\tcfg.Password = cli.String(schema.CLIOptPassword)\n\tcfg.DatabaseName = cli.String(schema.CLIOptDatabase)\n\tcfg.PluginName = cli.String(schema.CLIOptPluginName)\n\n\tconnectAttributes := cli.Generic(schema.CLIOptConnectAttributes).(*cliflag.StringMap)\n\tcfg.ConnectAttributes = connectAttributes.Value()\n\n\tif cli.Bool(schema.CLIFlagEnableTLS) {\n\t\tcfg.TLS = &config.TLS{\n\t\t\tEnabled:                true,\n\t\t\tCertFile:               cli.String(schema.CLIFlagTLSCertFile),\n\t\t\tKeyFile:                cli.String(schema.CLIFlagTLSKeyFile),\n\t\t\tCaFile:                 cli.String(schema.CLIFlagTLSCaFile),\n\t\t\tEnableHostVerification: cli.Bool(schema.CLIFlagTLSEnableHostVerification),\n\t\t}\n\t}\n\n\tif err := ValidateConnectConfig(cfg); err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn cfg, nil\n}\n\n// ValidateConnectConfig validates params\nfunc ValidateConnectConfig(cfg *config.SQL) error {\n\thost, _, err := net.SplitHostPort(cfg.ConnectAddr)\n\tif err != nil {\n\t\treturn schema.NewConfigError(\"invalid host and port \" + cfg.ConnectAddr)\n\t}\n\tif len(host) == 0 {\n\t\treturn schema.NewConfigError(\"missing sql endpoint argument \" + flag(schema.CLIOptEndpoint))\n\t}\n\tif cfg.DatabaseName == \"\" {\n\t\treturn schema.NewConfigError(\"missing \" + flag(schema.CLIOptDatabase) + \" argument\")\n\t}\n\tif cfg.TLS != nil && cfg.TLS.Enabled {\n\t\tenabledCaFile := cfg.TLS.CaFile != \"\"\n\t\tenabledCertFile := cfg.TLS.CertFile != \"\"\n\t\tenabledKeyFile := cfg.TLS.KeyFile != \"\"\n\n\t\tif (enabledCertFile && !enabledKeyFile) || (!enabledCertFile && enabledKeyFile) {\n\t\t\treturn schema.NewConfigError(\"must have both CertFile and KeyFile set\")\n\t\t}\n\n\t\tif !enabledCaFile && !enabledCertFile && !enabledKeyFile {\n\t\t\treturn schema.NewConfigError(\"must provide tls certs to use\")\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc flag(opt string) string {\n\treturn \"(-\" + opt + \")\"\n}\n\nfunc handleErr(err error) error {\n\tlog.Println(err)\n\treturn err\n}\n"
  },
  {
    "path": "tools/sql/main.go",
    "content": "// Copyright (c) 2017 Uber Technologies, Inc.\n//\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in\n// all copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n// THE SOFTWARE.\n\npackage sql\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/urfave/cli/v2\"\n\n\tcliflag \"github.com/uber/cadence/tools/common/flag\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n)\n\nconst defaultSQLPort = 3306\n\n// RunTool runs the cadence-cassandra-tool command line tool\nfunc RunTool(args []string) error {\n\tapp := BuildCLIOptions()\n\treturn app.Run(args)\n}\n\n// root handler for all cli commands\nfunc cliHandler(c *cli.Context, handler func(c *cli.Context) error) error {\n\terr := handler(c)\n\tif err != nil {\n\t\tif quiet := c.Bool(schema.CLIOptQuiet); quiet { // return error only if not quiet\n\t\t\tfmt.Println(\"error executing command: \", err)\n\t\t\treturn nil\n\t\t}\n\t\treturn fmt.Errorf(\"error executing command: %w\", err)\n\t}\n\treturn nil\n}\n\n// BuildCLIOptions builds the options for cli\nfunc BuildCLIOptions() *cli.App {\n\n\tapp := cli.NewApp()\n\tapp.Name = \"cadence-sql-tool\"\n\tapp.Usage = \"Command line tool for cadence sql operations\"\n\tapp.Version = \"0.0.1\"\n\n\tapp.Flags = []cli.Flag{\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagEndpoint,\n\t\t\tAliases: []string{\"ep\"},\n\t\t\tValue:   \"127.0.0.1\",\n\t\t\tUsage:   \"hostname or ip address of sql host to connect to\",\n\t\t\tEnvVars: []string{\"SQL_HOST\"},\n\t\t},\n\t\t&cli.IntFlag{\n\t\t\tName:    schema.CLIFlagPort,\n\t\t\tAliases: []string{\"p\"},\n\t\t\tValue:   defaultSQLPort,\n\t\t\tUsage:   \"port of sql host to connect to\",\n\t\t\tEnvVars: []string{\"SQL_PORT\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagUser,\n\t\t\tAliases: []string{\"u\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"user name used for authentication when connecting to sql host\",\n\t\t\tEnvVars: []string{\"SQL_USER\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagPassword,\n\t\t\tAliases: []string{\"pw\"},\n\t\t\tValue:   \"\",\n\t\t\tUsage:   \"password used for authentication when connecting to sql host\",\n\t\t\tEnvVars: []string{\"SQL_PASSWORD\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagDatabase,\n\t\t\tAliases: []string{\"db\"},\n\t\t\tValue:   \"cadence\",\n\t\t\tUsage:   \"name of the sql database\",\n\t\t\tEnvVars: []string{\"SQL_DATABASE\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagPluginName,\n\t\t\tAliases: []string{\"pl\"},\n\t\t\tValue:   \"mysql\",\n\t\t\tUsage:   \"name of the sql plugin\",\n\t\t\tEnvVars: []string{\"SQL_PLUGIN\"},\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagQuiet,\n\t\t\tAliases: []string{\"q\"},\n\t\t\tUsage:   \"Don't set exit status to 1 on error\",\n\t\t},\n\t\t&cli.GenericFlag{\n\t\t\tName:    schema.CLIFlagConnectAttributes,\n\t\t\tAliases: []string{\"ca\"},\n\t\t\tValue:   &cliflag.StringMap{},\n\t\t\tUsage:   \"sql connect attributes (must be in key1=value1,key2=value2,...,keyN=valueN format, e.g. cluster=dca or cluster=dca,instance=cadence)\",\n\t\t\tEnvVars: []string{\"SQL_CONNECT_ATTRIBUTES\"},\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagEnableTLS,\n\t\t\tUsage:   \"enable TLS over sql connection\",\n\t\t\tEnvVars: []string{\"SQL_TLS\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSCertFile,\n\t\t\tUsage:   \"sql tls client cert path (tls must be enabled)\",\n\t\t\tEnvVars: []string{\"SQL_TLS_CERT_FILE\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSKeyFile,\n\t\t\tUsage:   \"sql tls client key path (tls must be enabled)\",\n\t\t\tEnvVars: []string{\"SQL_TLS_KEY_FILE\"},\n\t\t},\n\t\t&cli.StringFlag{\n\t\t\tName:    schema.CLIFlagTLSCaFile,\n\t\t\tUsage:   \"sql tls client ca file (tls must be enabled)\",\n\t\t\tEnvVars: []string{\"SQL_TLS_CA_FILE\"},\n\t\t},\n\t\t&cli.BoolFlag{\n\t\t\tName:    schema.CLIFlagTLSEnableHostVerification,\n\t\t\tUsage:   \"sql tls verify hostname and server cert (tls must be enabled)\",\n\t\t\tEnvVars: []string{\"SQL_TLS_ENABLE_HOST_VERIFICATION\"},\n\t\t},\n\t}\n\n\tapp.Commands = []*cli.Command{\n\t\t{\n\t\t\tName:    \"setup-schema\",\n\t\t\tAliases: []string{\"setup\"},\n\t\t\tUsage:   \"setup initial version of sql schema\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagVersion,\n\t\t\t\t\tAliases: []string{\"v\"},\n\t\t\t\t\tUsage:   \"initial version of the schema, cannot be used with disable-versioning\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagSchemaFile,\n\t\t\t\t\tAliases: []string{\"f\"},\n\t\t\t\t\tUsage:   \"path to the .sql schema file; if un-specified, will just setup versioning tables\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    schema.CLIFlagDisableVersioning,\n\t\t\t\t\tAliases: []string{\"d\"},\n\t\t\t\t\tUsage:   \"disable setup of schema versioning\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:    schema.CLIFlagOverwrite,\n\t\t\t\t\tAliases: []string{\"o\"},\n\t\t\t\t\tUsage:   \"drop all existing tables before setting up new schema\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, setupSchema)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"update-schema\",\n\t\t\tAliases: []string{\"update\"},\n\t\t\tUsage:   \"update sql schema to a specific version\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagTargetVersion,\n\t\t\t\t\tAliases: []string{\"v\"},\n\t\t\t\t\tUsage:   \"target version for the schema update, defaults to latest\",\n\t\t\t\t},\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagSchemaDir,\n\t\t\t\t\tAliases: []string{\"d\"},\n\t\t\t\t\tUsage:   \"path to directory containing versioned schema\",\n\t\t\t\t},\n\t\t\t\t&cli.BoolFlag{\n\t\t\t\t\tName:  schema.CLIFlagDryrun,\n\t\t\t\t\tUsage: \"do a dryrun, which will print queries only without executing them\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, updateSchema)\n\t\t\t},\n\t\t},\n\t\t{\n\t\t\tName:    \"create-database\",\n\t\t\tAliases: []string{\"create\"},\n\t\t\tUsage:   \"creates a database\",\n\t\t\tFlags: []cli.Flag{\n\t\t\t\t&cli.StringFlag{\n\t\t\t\t\tName:    schema.CLIFlagDatabase,\n\t\t\t\t\tAliases: []string{\"db\"},\n\t\t\t\t\tUsage:   \"name of the database\",\n\t\t\t\t},\n\t\t\t},\n\t\t\tAction: func(c *cli.Context) error {\n\t\t\t\treturn cliHandler(c, createDatabase)\n\t\t\t},\n\t\t},\n\t}\n\n\treturn app\n}\n"
  },
  {
    "path": "tools/sql/sqlite/sqlite_test.go",
    "content": "// The MIT License (MIT)\n\n// Copyright (c) 2017-2020 Uber Technologies Inc.\n\n// Permission is hereby granted, free of charge, to any person obtaining a copy\n// of this software and associated documentation files (the \"Software\"), to deal\n// in the Software without restriction, including without limitation the rights\n// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n// copies of the Software, and to permit persons to whom the Software is\n// furnished to do so, subject to the following conditions:\n//\n// The above copyright notice and this permission notice shall be included in all\n// copies or substantial portions of the Software.\n//\n// THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n// SOFTWARE.\n\npackage sqlite\n\nimport (\n\t\"fmt\"\n\t\"testing\"\n\n\t\"github.com/stretchr/testify/assert\"\n\t\"github.com/stretchr/testify/require\"\n\n\t\"github.com/uber/cadence/common/config\"\n\tsqliteplugin \"github.com/uber/cadence/common/persistence/sql/sqlplugin/sqlite\"\n\t\"github.com/uber/cadence/schema/sqlite\"\n\t\"github.com/uber/cadence/tools/common/schema\"\n\t\"github.com/uber/cadence/tools/sql\"\n)\n\n// Test_SetupSchema test that setup schema works for all database sqlite schemas\n// in-memory sqlite database is used for testing\nfunc Test_SetupSchema(t *testing.T) {\n\tfor _, dbName := range listDatabaseNames(t) {\n\t\tt.Run(dbName, func(t *testing.T) {\n\t\t\tconn := newInMemoryDB(t)\n\n\t\t\terr := schema.SetupFromConfig(&schema.SetupConfig{\n\t\t\t\tSchemaFilePath:    fmt.Sprintf(\"../../../schema/sqlite/%s/schema.sql\", dbName),\n\t\t\t\tInitialVersion:    \"0.1\",\n\t\t\t\tOverwrite:         false,\n\t\t\t\tDisableVersioning: false,\n\t\t\t}, conn)\n\n\t\t\tassert.NoError(t, err)\n\t\t})\n\t}\n}\n\n// newInMemoryDB returns a new in-memory sqlite connection\nfunc newInMemoryDB(t *testing.T) *sql.Connection {\n\tt.Helper()\n\n\tconn, err := sql.NewConnection(&config.SQL{\n\t\tPluginName: sqliteplugin.PluginName,\n\t})\n\trequire.NoError(t, err)\n\treturn conn\n}\n\n// listDatabaseSchemaFilePaths returns a list of database schema file paths\nfunc listDatabaseNames(t *testing.T) []string {\n\tt.Helper()\n\n\tdirs, err := sqlite.SchemaFS.ReadDir(\".\")\n\trequire.NoError(t, err)\n\n\tvar databaseNames = make([]string, len(dirs))\n\tfor i, dir := range dirs {\n\t\tdatabaseNames[i] = dir.Name()\n\t}\n\n\treturn databaseNames\n}\n"
  }
]